On Tue, Feb 25, 2020 at 02:56 PM CET, Lorenz Bauer wrote: > The sockmap works by overriding some of the callbacks in sk->sk_prot, while > leaving others untouched. This means that we need access to the struct proto > for any protocol we want to support. For IPv4 this is trivial, since both > TCP and UDP are always compiled in. IPv6 may be disabled or compiled as a > module, so the existing TCP sockmap hooks use some trickery to lazily > initialize the modified struct proto for TCPv6. > > Pull this logic into a standalone struct sk_psock_hooks, so that it can be > re-used by UDP sockmap. > > Signed-off-by: Lorenz Bauer <lmb@xxxxxxxxxxxxxx> > --- > include/linux/skmsg.h | 36 ++++++++----- > include/net/tcp.h | 1 - > net/core/skmsg.c | 52 +++++++++++++++++++ > net/core/sock_map.c | 24 ++++----- > net/ipv4/tcp_bpf.c | 114 ++++++++++++------------------------------ > 5 files changed, 116 insertions(+), 111 deletions(-) > > diff --git a/include/linux/skmsg.h b/include/linux/skmsg.h > index c881094387db..70d65ab10b5c 100644 > --- a/include/linux/skmsg.h > +++ b/include/linux/skmsg.h [...] > @@ -424,4 +425,13 @@ static inline void psock_progs_drop(struct sk_psock_progs *progs) > psock_set_prog(&progs->skb_verdict, NULL); > } > > +static inline int sk_psock_hooks_init(struct sk_psock_hooks *hooks, > + struct proto *ipv4_base) > +{ > + hooks->ipv6_lock = __SPIN_LOCK_UNLOCKED(); We will need spin_lock_init instead to play nice with CONFIG_DEBUG_SPINLOCK. > + return hooks->rebuild_proto(hooks->ipv4, ipv4_base); > +} > + > +int sk_psock_hooks_install(struct sk_psock_hooks *hooks, struct sock *sk); > + > #endif /* _LINUX_SKMSG_H */ [...]