~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

TOMOYO Linux Cross Reference
Linux/tools/testing/selftests/bpf/progs/bpf_iter_setsockopt.c

Version: ~ [ linux-6.11.5 ] ~ [ linux-6.10.14 ] ~ [ linux-6.9.12 ] ~ [ linux-6.8.12 ] ~ [ linux-6.7.12 ] ~ [ linux-6.6.58 ] ~ [ linux-6.5.13 ] ~ [ linux-6.4.16 ] ~ [ linux-6.3.13 ] ~ [ linux-6.2.16 ] ~ [ linux-6.1.114 ] ~ [ linux-6.0.19 ] ~ [ linux-5.19.17 ] ~ [ linux-5.18.19 ] ~ [ linux-5.17.15 ] ~ [ linux-5.16.20 ] ~ [ linux-5.15.169 ] ~ [ linux-5.14.21 ] ~ [ linux-5.13.19 ] ~ [ linux-5.12.19 ] ~ [ linux-5.11.22 ] ~ [ linux-5.10.228 ] ~ [ linux-5.9.16 ] ~ [ linux-5.8.18 ] ~ [ linux-5.7.19 ] ~ [ linux-5.6.19 ] ~ [ linux-5.5.19 ] ~ [ linux-5.4.284 ] ~ [ linux-5.3.18 ] ~ [ linux-5.2.21 ] ~ [ linux-5.1.21 ] ~ [ linux-5.0.21 ] ~ [ linux-4.20.17 ] ~ [ linux-4.19.322 ] ~ [ linux-4.18.20 ] ~ [ linux-4.17.19 ] ~ [ linux-4.16.18 ] ~ [ linux-4.15.18 ] ~ [ linux-4.14.336 ] ~ [ linux-4.13.16 ] ~ [ linux-4.12.14 ] ~ [ linux-4.11.12 ] ~ [ linux-4.10.17 ] ~ [ linux-4.9.337 ] ~ [ linux-4.4.302 ] ~ [ linux-3.10.108 ] ~ [ linux-2.6.32.71 ] ~ [ linux-2.6.0 ] ~ [ linux-2.4.37.11 ] ~ [ unix-v6-master ] ~ [ ccs-tools-1.8.9 ] ~ [ policy-sample ] ~
Architecture: ~ [ i386 ] ~ [ alpha ] ~ [ m68k ] ~ [ mips ] ~ [ ppc ] ~ [ sparc ] ~ [ sparc64 ] ~

  1 // SPDX-License-Identifier: GPL-2.0
  2 /* Copyright (c) 2021 Facebook */
  3 #include "bpf_iter.h"
  4 #include "bpf_tracing_net.h"
  5 #include <bpf/bpf_helpers.h>
  6 #include <bpf/bpf_endian.h>
  7 
  8 #define bpf_tcp_sk(skc) ({                              \
  9         struct sock_common *_skc = skc;                 \
 10         sk = NULL;                                      \
 11         tp = NULL;                                      \
 12         if (_skc) {                                     \
 13                 tp = bpf_skc_to_tcp_sock(_skc);         \
 14                 sk = (struct sock *)tp;                 \
 15         }                                               \
 16         tp;                                             \
 17 })
 18 
 19 unsigned short reuse_listen_hport = 0;
 20 unsigned short listen_hport = 0;
 21 char cubic_cc[TCP_CA_NAME_MAX] = "bpf_cubic";
 22 char dctcp_cc[TCP_CA_NAME_MAX] = "bpf_dctcp";
 23 bool random_retry = false;
 24 
 25 static bool tcp_cc_eq(const char *a, const char *b)
 26 {
 27         int i;
 28 
 29         for (i = 0; i < TCP_CA_NAME_MAX; i++) {
 30                 if (a[i] != b[i])
 31                         return false;
 32                 if (!a[i])
 33                         break;
 34         }
 35 
 36         return true;
 37 }
 38 
 39 SEC("iter/tcp")
 40 int change_tcp_cc(struct bpf_iter__tcp *ctx)
 41 {
 42         char cur_cc[TCP_CA_NAME_MAX];
 43         struct tcp_sock *tp;
 44         struct sock *sk;
 45 
 46         if (!bpf_tcp_sk(ctx->sk_common))
 47                 return 0;
 48 
 49         if (sk->sk_family != AF_INET6 ||
 50             (sk->sk_state != TCP_LISTEN &&
 51              sk->sk_state != TCP_ESTABLISHED) ||
 52             (sk->sk_num != reuse_listen_hport &&
 53              sk->sk_num != listen_hport &&
 54              bpf_ntohs(sk->sk_dport) != listen_hport))
 55                 return 0;
 56 
 57         if (bpf_getsockopt(tp, SOL_TCP, TCP_CONGESTION,
 58                            cur_cc, sizeof(cur_cc)))
 59                 return 0;
 60 
 61         if (!tcp_cc_eq(cur_cc, cubic_cc))
 62                 return 0;
 63 
 64         if (random_retry && bpf_get_prandom_u32() % 4 == 1)
 65                 return 1;
 66 
 67         bpf_setsockopt(tp, SOL_TCP, TCP_CONGESTION, dctcp_cc, sizeof(dctcp_cc));
 68         return 0;
 69 }
 70 
 71 char _license[] SEC("license") = "GPL";
 72 

~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

kernel.org | git.kernel.org | LWN.net | Project Home | SVN repository | Mail admin

Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.

sflogo.php