Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 | // SPDX-License-Identifier: GPL-2.0 #include <linux/bpf.h> #include <bpf/bpf_helpers.h> #include <bpf/bpf_tracing.h> #include <stdbool.h> #ifdef ENABLE_ATOMICS_TESTS bool skip_tests __attribute((__section__(".data"))) = false; #else bool skip_tests = true; #endif __u32 pid = 0; __u64 add64_value = 1; __u64 add64_result = 0; __u32 add32_value = 1; __u32 add32_result = 0; __u64 add_stack_value_copy = 0; __u64 add_stack_result = 0; __u64 add_noreturn_value = 1; SEC("raw_tp/sys_enter") int add(const void *ctx) { if (pid != (bpf_get_current_pid_tgid() >> 32)) return 0; #ifdef ENABLE_ATOMICS_TESTS __u64 add_stack_value = 1; add64_result = __sync_fetch_and_add(&add64_value, 2); add32_result = __sync_fetch_and_add(&add32_value, 2); add_stack_result = __sync_fetch_and_add(&add_stack_value, 2); add_stack_value_copy = add_stack_value; __sync_fetch_and_add(&add_noreturn_value, 2); #endif return 0; } __s64 sub64_value = 1; __s64 sub64_result = 0; __s32 sub32_value = 1; __s32 sub32_result = 0; __s64 sub_stack_value_copy = 0; __s64 sub_stack_result = 0; __s64 sub_noreturn_value = 1; SEC("raw_tp/sys_enter") int sub(const void *ctx) { if (pid != (bpf_get_current_pid_tgid() >> 32)) return 0; #ifdef ENABLE_ATOMICS_TESTS __u64 sub_stack_value = 1; sub64_result = __sync_fetch_and_sub(&sub64_value, 2); sub32_result = __sync_fetch_and_sub(&sub32_value, 2); sub_stack_result = __sync_fetch_and_sub(&sub_stack_value, 2); sub_stack_value_copy = sub_stack_value; __sync_fetch_and_sub(&sub_noreturn_value, 2); #endif return 0; } __u64 and64_value = (0x110ull << 32); __u64 and64_result = 0; __u32 and32_value = 0x110; __u32 and32_result = 0; __u64 and_noreturn_value = (0x110ull << 32); SEC("raw_tp/sys_enter") int and(const void *ctx) { if (pid != (bpf_get_current_pid_tgid() >> 32)) return 0; #ifdef ENABLE_ATOMICS_TESTS and64_result = __sync_fetch_and_and(&and64_value, 0x011ull << 32); and32_result = __sync_fetch_and_and(&and32_value, 0x011); __sync_fetch_and_and(&and_noreturn_value, 0x011ull << 32); #endif return 0; } __u64 or64_value = (0x110ull << 32); __u64 or64_result = 0; __u32 or32_value = 0x110; __u32 or32_result = 0; __u64 or_noreturn_value = (0x110ull << 32); SEC("raw_tp/sys_enter") int or(const void *ctx) { if (pid != (bpf_get_current_pid_tgid() >> 32)) return 0; #ifdef ENABLE_ATOMICS_TESTS or64_result = __sync_fetch_and_or(&or64_value, 0x011ull << 32); or32_result = __sync_fetch_and_or(&or32_value, 0x011); __sync_fetch_and_or(&or_noreturn_value, 0x011ull << 32); #endif return 0; } __u64 xor64_value = (0x110ull << 32); __u64 xor64_result = 0; __u32 xor32_value = 0x110; __u32 xor32_result = 0; __u64 xor_noreturn_value = (0x110ull << 32); SEC("raw_tp/sys_enter") int xor(const void *ctx) { if (pid != (bpf_get_current_pid_tgid() >> 32)) return 0; #ifdef ENABLE_ATOMICS_TESTS xor64_result = __sync_fetch_and_xor(&xor64_value, 0x011ull << 32); xor32_result = __sync_fetch_and_xor(&xor32_value, 0x011); __sync_fetch_and_xor(&xor_noreturn_value, 0x011ull << 32); #endif return 0; } __u64 cmpxchg64_value = 1; __u64 cmpxchg64_result_fail = 0; __u64 cmpxchg64_result_succeed = 0; __u32 cmpxchg32_value = 1; __u32 cmpxchg32_result_fail = 0; __u32 cmpxchg32_result_succeed = 0; SEC("raw_tp/sys_enter") int cmpxchg(const void *ctx) { if (pid != (bpf_get_current_pid_tgid() >> 32)) return 0; #ifdef ENABLE_ATOMICS_TESTS cmpxchg64_result_fail = __sync_val_compare_and_swap(&cmpxchg64_value, 0, 3); cmpxchg64_result_succeed = __sync_val_compare_and_swap(&cmpxchg64_value, 1, 2); cmpxchg32_result_fail = __sync_val_compare_and_swap(&cmpxchg32_value, 0, 3); cmpxchg32_result_succeed = __sync_val_compare_and_swap(&cmpxchg32_value, 1, 2); #endif return 0; } __u64 xchg64_value = 1; __u64 xchg64_result = 0; __u32 xchg32_value = 1; __u32 xchg32_result = 0; SEC("raw_tp/sys_enter") int xchg(const void *ctx) { if (pid != (bpf_get_current_pid_tgid() >> 32)) return 0; #ifdef ENABLE_ATOMICS_TESTS __u64 val64 = 2; __u32 val32 = 2; xchg64_result = __sync_lock_test_and_set(&xchg64_value, val64); xchg32_result = __sync_lock_test_and_set(&xchg32_value, val32); #endif return 0; } |