1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (c) 2024 Meta Platforms, Inc. and affiliates. */ 3 4 #include <vmlinux.h> 5 #include <bpf/bpf_tracing.h> 6 #include "bpf_misc.h" 7 #include "../bpf_testmod/bpf_testmod.h" 8 #include "../bpf_testmod/bpf_testmod_kfunc.h" 9 10 char _license[] SEC("license") = "GPL"; 11 12 __success 13 /* prologue */ 14 __xlated("0: r6 = *(u64 *)(r1 +0)") 15 __xlated("1: r7 = *(u64 *)(r6 +0)") 16 __xlated("2: r7 += 1000") 17 __xlated("3: *(u64 *)(r6 +0) = r7") 18 /* main prog */ 19 __xlated("4: if r1 == 0x0 goto pc+5") 20 __xlated("5: if r1 == 0x1 goto pc+2") 21 __xlated("6: r1 = 1") 22 __xlated("7: goto pc-3") 23 __xlated("8: r1 = 0") 24 __xlated("9: goto pc-6") 25 __xlated("10: r0 = 0") 26 __xlated("11: exit") 27 SEC("struct_ops/test_prologue_goto_start") test_prologue_goto_start(void)28__naked int test_prologue_goto_start(void) 29 { 30 asm volatile ( 31 "if r1 == 0 goto +5;" 32 "if r1 == 1 goto +2;" 33 "r1 = 1;" 34 "goto -3;" 35 "r1 = 0;" 36 "goto -6;" 37 "r0 = 0;" 38 "exit;" 39 ::: __clobber_all); 40 } 41 42 __success 43 /* save __u64 *ctx to stack */ 44 __xlated("0: *(u64 *)(r10 -8) = r1") 45 /* main prog */ 46 __xlated("1: if r1 == 0x0 goto pc+5") 47 __xlated("2: if r1 == 0x1 goto pc+2") 48 __xlated("3: r1 = 1") 49 __xlated("4: goto pc-3") 50 __xlated("5: r1 = 0") 51 __xlated("6: goto pc-6") 52 __xlated("7: r0 = 0") 53 /* epilogue */ 54 __xlated("8: r1 = *(u64 *)(r10 -8)") 55 __xlated("9: r1 = *(u64 *)(r1 +0)") 56 __xlated("10: r6 = *(u64 *)(r1 +0)") 57 __xlated("11: r6 += 10000") 58 __xlated("12: *(u64 *)(r1 +0) = r6") 59 __xlated("13: r0 = r6") 60 __xlated("14: r0 *= 2") 61 __xlated("15: exit") 62 SEC("struct_ops/test_epilogue_goto_start") test_epilogue_goto_start(void)63__naked int test_epilogue_goto_start(void) 64 { 65 asm volatile ( 66 "if r1 == 0 goto +5;" 67 "if r1 == 1 goto +2;" 68 "r1 = 1;" 69 "goto -3;" 70 "r1 = 0;" 71 "goto -6;" 72 "r0 = 0;" 73 "exit;" 74 ::: __clobber_all); 75 } 76 77 __success 78 /* prologue */ 79 __xlated("0: r6 = *(u64 *)(r1 +0)") 80 __xlated("1: r7 = *(u64 *)(r6 +0)") 81 __xlated("2: r7 += 1000") 82 __xlated("3: *(u64 *)(r6 +0) = r7") 83 /* save __u64 *ctx to stack */ 84 __xlated("4: *(u64 *)(r10 -8) = r1") 85 /* main prog */ 86 __xlated("5: if r1 == 0x0 goto pc+5") 87 __xlated("6: if r1 == 0x1 goto pc+2") 88 __xlated("7: r1 = 1") 89 __xlated("8: goto pc-3") 90 __xlated("9: r1 = 0") 91 __xlated("10: goto pc-6") 92 __xlated("11: r0 = 0") 93 /* epilogue */ 94 __xlated("12: r1 = *(u64 *)(r10 -8)") 95 __xlated("13: r1 = *(u64 *)(r1 +0)") 96 __xlated("14: r6 = *(u64 *)(r1 +0)") 97 __xlated("15: r6 += 10000") 98 __xlated("16: *(u64 *)(r1 +0) = r6") 99 __xlated("17: r0 = r6") 100 __xlated("18: r0 *= 2") 101 __xlated("19: exit") 102 SEC("struct_ops/test_pro_epilogue_goto_start") test_pro_epilogue_goto_start(void)103__naked int test_pro_epilogue_goto_start(void) 104 { 105 asm volatile ( 106 "if r1 == 0 goto +5;" 107 "if r1 == 1 goto +2;" 108 "r1 = 1;" 109 "goto -3;" 110 "r1 = 0;" 111 "goto -6;" 112 "r0 = 0;" 113 "exit;" 114 ::: __clobber_all); 115 } 116 117 SEC(".struct_ops.link") 118 struct bpf_testmod_st_ops epilogue_goto_start = { 119 .test_prologue = (void *)test_prologue_goto_start, 120 .test_epilogue = (void *)test_epilogue_goto_start, 121 .test_pro_epilogue = (void *)test_pro_epilogue_goto_start, 122 }; 123 124 SEC("syscall") 125 __retval(0) syscall_prologue_goto_start(void * ctx)126int syscall_prologue_goto_start(void *ctx) 127 { 128 struct st_ops_args args = {}; 129 130 return bpf_kfunc_st_ops_test_prologue(&args); 131 } 132 133 SEC("syscall") 134 __retval(20000) /* (EPILOGUE_A [10000]) * 2 */ syscall_epilogue_goto_start(void * ctx)135int syscall_epilogue_goto_start(void *ctx) 136 { 137 struct st_ops_args args = {}; 138 139 return bpf_kfunc_st_ops_test_epilogue(&args); 140 } 141 142 SEC("syscall") 143 __retval(22000) /* (PROLOGUE_A [1000] + EPILOGUE_A [10000]) * 2 */ syscall_pro_epilogue_goto_start(void * ctx)144int syscall_pro_epilogue_goto_start(void *ctx) 145 { 146 struct st_ops_args args = {}; 147 148 return bpf_kfunc_st_ops_test_pro_epilogue(&args); 149 } 150