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 struct bpf_testmod_struct_arg_3 { 17 int a; 18 int b[]; 19 }; 20 21 long t1_a_a, t1_a_b, t1_b, t1_c, t1_ret, t1_nregs; 22 __u64 t1_reg0, t1_reg1, t1_reg2, t1_reg3; 23 long t2_a, t2_b_a, t2_b_b, t2_c, t2_ret; 24 long t3_a, t3_b, t3_c_a, t3_c_b, t3_ret; 25 long t4_a_a, t4_b, t4_c, t4_d, t4_e_a, t4_e_b, t4_ret; 26 long t5_ret; 27 int t6; 28 29 SEC("fentry/bpf_testmod_test_struct_arg_1") 30 int BPF_PROG2(test_struct_arg_1, struct bpf_testmod_struct_arg_2, a, int, b, int, c) 31 { 32 t1_a_a = a.a; 33 t1_a_b = a.b; 34 t1_b = b; 35 t1_c = c; 36 return 0; 37 } 38 39 SEC("fexit/bpf_testmod_test_struct_arg_1") 40 int BPF_PROG2(test_struct_arg_2, struct bpf_testmod_struct_arg_2, a, int, b, int, c, int, ret) 41 { 42 t1_nregs = bpf_get_func_arg_cnt(ctx); 43 /* a.a */ 44 bpf_get_func_arg(ctx, 0, &t1_reg0); 45 /* a.b */ 46 bpf_get_func_arg(ctx, 1, &t1_reg1); 47 /* b */ 48 bpf_get_func_arg(ctx, 2, &t1_reg2); 49 t1_reg2 = (int)t1_reg2; 50 /* c */ 51 bpf_get_func_arg(ctx, 3, &t1_reg3); 52 t1_reg3 = (int)t1_reg3; 53 54 t1_ret = ret; 55 return 0; 56 } 57 58 SEC("fentry/bpf_testmod_test_struct_arg_2") 59 int BPF_PROG2(test_struct_arg_3, int, a, struct bpf_testmod_struct_arg_2, b, int, c) 60 { 61 t2_a = a; 62 t2_b_a = b.a; 63 t2_b_b = b.b; 64 t2_c = c; 65 return 0; 66 } 67 68 SEC("fexit/bpf_testmod_test_struct_arg_2") 69 int BPF_PROG2(test_struct_arg_4, int, a, struct bpf_testmod_struct_arg_2, b, int, c, int, ret) 70 { 71 t2_ret = ret; 72 return 0; 73 } 74 75 SEC("fentry/bpf_testmod_test_struct_arg_3") 76 int BPF_PROG2(test_struct_arg_5, int, a, int, b, struct bpf_testmod_struct_arg_2, c) 77 { 78 t3_a = a; 79 t3_b = b; 80 t3_c_a = c.a; 81 t3_c_b = c.b; 82 return 0; 83 } 84 85 SEC("fexit/bpf_testmod_test_struct_arg_3") 86 int BPF_PROG2(test_struct_arg_6, int, a, int, b, struct bpf_testmod_struct_arg_2, c, int, ret) 87 { 88 t3_ret = ret; 89 return 0; 90 } 91 92 SEC("fentry/bpf_testmod_test_struct_arg_4") 93 int BPF_PROG2(test_struct_arg_7, struct bpf_testmod_struct_arg_1, a, int, b, 94 int, c, int, d, struct bpf_testmod_struct_arg_2, e) 95 { 96 t4_a_a = a.a; 97 t4_b = b; 98 t4_c = c; 99 t4_d = d; 100 t4_e_a = e.a; 101 t4_e_b = e.b; 102 return 0; 103 } 104 105 SEC("fexit/bpf_testmod_test_struct_arg_4") 106 int BPF_PROG2(test_struct_arg_8, struct bpf_testmod_struct_arg_1, a, int, b, 107 int, c, int, d, struct bpf_testmod_struct_arg_2, e, int, ret) 108 { 109 t4_ret = ret; 110 return 0; 111 } 112 113 SEC("fentry/bpf_testmod_test_struct_arg_5") 114 int BPF_PROG2(test_struct_arg_9) 115 { 116 return 0; 117 } 118 119 SEC("fexit/bpf_testmod_test_struct_arg_5") 120 int BPF_PROG2(test_struct_arg_10, int, ret) 121 { 122 t5_ret = ret; 123 return 0; 124 } 125 126 SEC("fentry/bpf_testmod_test_struct_arg_6") 127 int BPF_PROG2(test_struct_arg_11, struct bpf_testmod_struct_arg_3 *, a) 128 { 129 t6 = a->b[0]; 130 return 0; 131 } 132 133 char _license[] SEC("license") = "GPL"; 134