1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright (c) 2022 Meta Platforms, Inc. and affiliates. */
3 
4 #include <vmlinux.h>
5 #include <bpf/bpf_tracing.h>
6 #include <bpf/bpf_helpers.h>
7 
8 struct bpf_testmod_struct_arg_1 {
9 	int a;
10 };
11 struct bpf_testmod_struct_arg_2 {
12 	long a;
13 	long b;
14 };
15 
16 long t1_a_a, t1_a_b, t1_b, t1_c, t1_ret, t1_nregs;
17 __u64 t1_reg0, t1_reg1, t1_reg2, t1_reg3;
18 long t2_a, t2_b_a, t2_b_b, t2_c, t2_ret;
19 long t3_a, t3_b, t3_c_a, t3_c_b, t3_ret;
20 long t4_a_a, t4_b, t4_c, t4_d, t4_e_a, t4_e_b, t4_ret;
21 long t5_ret;
22 
23 SEC("fentry/bpf_testmod_test_struct_arg_1")
24 int BPF_PROG2(test_struct_arg_1, struct bpf_testmod_struct_arg_2, a, int, b, int, c)
25 {
26 	t1_a_a = a.a;
27 	t1_a_b = a.b;
28 	t1_b = b;
29 	t1_c = c;
30 	return 0;
31 }
32 
33 SEC("fexit/bpf_testmod_test_struct_arg_1")
34 int BPF_PROG2(test_struct_arg_2, struct bpf_testmod_struct_arg_2, a, int, b, int, c, int, ret)
35 {
36 	t1_nregs =  bpf_get_func_arg_cnt(ctx);
37 	/* a.a */
38 	bpf_get_func_arg(ctx, 0, &t1_reg0);
39 	/* a.b */
40 	bpf_get_func_arg(ctx, 1, &t1_reg1);
41 	/* b */
42 	bpf_get_func_arg(ctx, 2, &t1_reg2);
43 	t1_reg2 = (int)t1_reg2;
44 	/* c */
45 	bpf_get_func_arg(ctx, 3, &t1_reg3);
46 	t1_reg3 = (int)t1_reg3;
47 
48 	t1_ret = ret;
49 	return 0;
50 }
51 
52 SEC("fentry/bpf_testmod_test_struct_arg_2")
53 int BPF_PROG2(test_struct_arg_3, int, a, struct bpf_testmod_struct_arg_2, b, int, c)
54 {
55 	t2_a = a;
56 	t2_b_a = b.a;
57 	t2_b_b = b.b;
58 	t2_c = c;
59 	return 0;
60 }
61 
62 SEC("fexit/bpf_testmod_test_struct_arg_2")
63 int BPF_PROG2(test_struct_arg_4, int, a, struct bpf_testmod_struct_arg_2, b, int, c, int, ret)
64 {
65 	t2_ret = ret;
66 	return 0;
67 }
68 
69 SEC("fentry/bpf_testmod_test_struct_arg_3")
70 int BPF_PROG2(test_struct_arg_5, int, a, int, b, struct bpf_testmod_struct_arg_2, c)
71 {
72 	t3_a = a;
73 	t3_b = b;
74 	t3_c_a = c.a;
75 	t3_c_b = c.b;
76 	return 0;
77 }
78 
79 SEC("fexit/bpf_testmod_test_struct_arg_3")
80 int BPF_PROG2(test_struct_arg_6, int, a, int, b, struct bpf_testmod_struct_arg_2, c, int, ret)
81 {
82 	t3_ret = ret;
83 	return 0;
84 }
85 
86 SEC("fentry/bpf_testmod_test_struct_arg_4")
87 int BPF_PROG2(test_struct_arg_7, struct bpf_testmod_struct_arg_1, a, int, b,
88 	     int, c, int, d, struct bpf_testmod_struct_arg_2, e)
89 {
90 	t4_a_a = a.a;
91 	t4_b = b;
92 	t4_c = c;
93 	t4_d = d;
94 	t4_e_a = e.a;
95 	t4_e_b = e.b;
96 	return 0;
97 }
98 
99 SEC("fexit/bpf_testmod_test_struct_arg_4")
100 int BPF_PROG2(test_struct_arg_8, struct bpf_testmod_struct_arg_1, a, int, b,
101 	     int, c, int, d, struct bpf_testmod_struct_arg_2, e, int, ret)
102 {
103 	t4_ret = ret;
104 	return 0;
105 }
106 
107 SEC("fentry/bpf_testmod_test_struct_arg_5")
108 int BPF_PROG2(test_struct_arg_9)
109 {
110 	return 0;
111 }
112 
113 SEC("fexit/bpf_testmod_test_struct_arg_5")
114 int BPF_PROG2(test_struct_arg_10, int, ret)
115 {
116 	t5_ret = ret;
117 	return 0;
118 }
119 
120 char _license[] SEC("license") = "GPL";
121