1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef _SOCK_REUSEPORT_H 3 #define _SOCK_REUSEPORT_H 4 5 #include <linux/filter.h> 6 #include <linux/skbuff.h> 7 #include <linux/types.h> 8 #include <linux/spinlock.h> 9 #include <net/sock.h> 10 11 extern spinlock_t reuseport_lock; 12 13 struct sock_reuseport { 14 struct rcu_head rcu; 15 16 u16 max_socks; /* length of socks */ 17 u16 num_socks; /* elements in socks */ 18 u16 num_closed_socks; /* closed elements in socks */ 19 u16 incoming_cpu; 20 /* The last synq overflow event timestamp of this 21 * reuse->socks[] group. 22 */ 23 unsigned int synq_overflow_ts; 24 /* ID stays the same even after the size of socks[] grows. */ 25 unsigned int reuseport_id; 26 unsigned int bind_inany:1; 27 unsigned int has_conns:1; 28 struct bpf_prog __rcu *prog; /* optional BPF sock selector */ 29 struct sock *socks[] __counted_by(max_socks); 30 }; 31 32 extern int reuseport_alloc(struct sock *sk, bool bind_inany); 33 extern int reuseport_add_sock(struct sock *sk, struct sock *sk2, 34 bool bind_inany); 35 extern void reuseport_detach_sock(struct sock *sk); 36 void reuseport_stop_listen_sock(struct sock *sk); 37 extern struct sock *reuseport_select_sock(struct sock *sk, 38 u32 hash, 39 struct sk_buff *skb, 40 int hdr_len); 41 struct sock *reuseport_migrate_sock(struct sock *sk, 42 struct sock *migrating_sk, 43 struct sk_buff *skb); 44 extern int reuseport_attach_prog(struct sock *sk, struct bpf_prog *prog); 45 extern int reuseport_detach_prog(struct sock *sk); 46 reuseport_has_conns(struct sock * sk)47 static inline bool reuseport_has_conns(struct sock *sk) 48 { 49 struct sock_reuseport *reuse; 50 bool ret = false; 51 52 rcu_read_lock(); 53 reuse = rcu_dereference(sk->sk_reuseport_cb); 54 if (reuse && reuse->has_conns) 55 ret = true; 56 rcu_read_unlock(); 57 58 return ret; 59 } 60 61 void reuseport_has_conns_set(struct sock *sk); 62 void reuseport_update_incoming_cpu(struct sock *sk, int val); 63 64 #endif /* _SOCK_REUSEPORT_H */ 65