include/net/sock_reuseport.h

   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
  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 */