1 // SPDX-License-Identifier: GPL-2.0
2 
3 #include "vmlinux.h"
4 
5 #include <bpf/bpf_helpers.h>
6 #include <bpf/bpf_tracing.h>
7 
8 char _license[] SEC("license") = "GPL";
9 
10 int ca1_cnt = 0;
11 int ca2_cnt = 0;
12 
13 static inline struct tcp_sock *tcp_sk(const struct sock *sk)
14 {
15 	return (struct tcp_sock *)sk;
16 }
17 
18 SEC("struct_ops/ca_update_1_init")
19 void BPF_PROG(ca_update_1_init, struct sock *sk)
20 {
21 	ca1_cnt++;
22 }
23 
24 SEC("struct_ops/ca_update_2_init")
25 void BPF_PROG(ca_update_2_init, struct sock *sk)
26 {
27 	ca2_cnt++;
28 }
29 
30 SEC("struct_ops/ca_update_cong_control")
31 void BPF_PROG(ca_update_cong_control, struct sock *sk,
32 	      const struct rate_sample *rs)
33 {
34 }
35 
36 SEC("struct_ops/ca_update_ssthresh")
37 __u32 BPF_PROG(ca_update_ssthresh, struct sock *sk)
38 {
39 	return tcp_sk(sk)->snd_ssthresh;
40 }
41 
42 SEC("struct_ops/ca_update_undo_cwnd")
43 __u32 BPF_PROG(ca_update_undo_cwnd, struct sock *sk)
44 {
45 	return tcp_sk(sk)->snd_cwnd;
46 }
47 
48 SEC(".struct_ops.link")
49 struct tcp_congestion_ops ca_update_1 = {
50 	.init = (void *)ca_update_1_init,
51 	.cong_control = (void *)ca_update_cong_control,
52 	.ssthresh = (void *)ca_update_ssthresh,
53 	.undo_cwnd = (void *)ca_update_undo_cwnd,
54 	.name = "tcp_ca_update",
55 };
56 
57 SEC(".struct_ops.link")
58 struct tcp_congestion_ops ca_update_2 = {
59 	.init = (void *)ca_update_2_init,
60 	.cong_control = (void *)ca_update_cong_control,
61 	.ssthresh = (void *)ca_update_ssthresh,
62 	.undo_cwnd = (void *)ca_update_undo_cwnd,
63 	.name = "tcp_ca_update",
64 };
65 
66 SEC(".struct_ops.link")
67 struct tcp_congestion_ops ca_wrong = {
68 	.cong_control = (void *)ca_update_cong_control,
69 	.ssthresh = (void *)ca_update_ssthresh,
70 	.undo_cwnd = (void *)ca_update_undo_cwnd,
71 	.name = "tcp_ca_wrong",
72 };
73 
74 SEC(".struct_ops")
75 struct tcp_congestion_ops ca_no_link = {
76 	.cong_control = (void *)ca_update_cong_control,
77 	.ssthresh = (void *)ca_update_ssthresh,
78 	.undo_cwnd = (void *)ca_update_undo_cwnd,
79 	.name = "tcp_ca_no_link",
80 };
81