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         /* The last synq overflow event timestamp of this
  19          * reuse->socks[] group.
  20          */
  21         unsigned int            synq_overflow_ts;
  22         /* ID stays the same even after the size of socks[] grows. */
  23         unsigned int            reuseport_id;
  24         unsigned int            bind_inany:1;
  25         unsigned int            has_conns:1;
  26         struct bpf_prog __rcu   *prog;          /* optional BPF sock selector */
  27         struct sock             *socks[0];      /* array of sock pointers */
  28 };
  29
  30 extern int reuseport_alloc(struct sock *sk, bool bind_inany);
  31 extern int reuseport_add_sock(struct sock *sk, struct sock *sk2,
  32                               bool bind_inany);
  33 extern void reuseport_detach_sock(struct sock *sk);
  34 extern struct sock *reuseport_select_sock(struct sock *sk,
  35                                           u32 hash,
  36                                           struct sk_buff *skb,
  37                                           int hdr_len);
  38 extern int reuseport_attach_prog(struct sock *sk, struct bpf_prog *prog);
  39
  40 static inline bool reuseport_has_conns(struct sock *sk, bool set)
  41 {
  42         struct sock_reuseport *reuse;
  43         bool ret = false;
  44
  45         rcu_read_lock();
  46         reuse = rcu_dereference(sk->sk_reuseport_cb);
  47         if (reuse) {
  48                 if (set)
  49                         reuse->has_conns = 1;
  50                 ret = reuse->has_conns;
  51         }
  52         rcu_read_unlock();
  53
  54         return ret;
  55 }
  56
  57 int reuseport_get_id(struct sock_reuseport *reuse);
  58
  59 #endif  /* _SOCK_REUSEPORT_H */