1 // SPDX-License-Identifier: GPL-2.0 2 /* Copyright (C) 2023. Huawei Technologies Co., Ltd */ 3 #include <test_progs.h> 4 5 #include "linux/filter.h" 6 #include "kptr_xchg_inline.skel.h" 7 8 void test_kptr_xchg_inline(void) 9 { 10 struct kptr_xchg_inline *skel; 11 struct bpf_insn *insn = NULL; 12 struct bpf_insn exp; 13 unsigned int cnt; 14 int err; 15 16 #if !(defined(__x86_64__) || defined(__aarch64__) || \ 17 (defined(__riscv) && __riscv_xlen == 64) || \ 18 (defined(__loongarch__) && __loongarch_grlen == 64)) 19 test__skip(); 20 return; 21 #endif 22 23 skel = kptr_xchg_inline__open_and_load(); 24 if (!ASSERT_OK_PTR(skel, "open_load")) 25 return; 26 27 err = get_xlated_program(bpf_program__fd(skel->progs.kptr_xchg_inline), &insn, &cnt); 28 if (!ASSERT_OK(err, "prog insn")) 29 goto out; 30 31 /* The original instructions are: 32 * r1 = map[id:xxx][0]+0 33 * r2 = 0 34 * call bpf_kptr_xchg#yyy 35 * 36 * call bpf_kptr_xchg#yyy will be inlined as: 37 * r0 = r2 38 * r0 = atomic64_xchg((u64 *)(r1 +0), r0) 39 */ 40 if (!ASSERT_GT(cnt, 5, "insn cnt")) 41 goto out; 42 43 exp = BPF_MOV64_REG(BPF_REG_0, BPF_REG_2); 44 if (!ASSERT_OK(memcmp(&insn[3], &exp, sizeof(exp)), "mov")) 45 goto out; 46 47 exp = BPF_ATOMIC_OP(BPF_DW, BPF_XCHG, BPF_REG_1, BPF_REG_0, 0); 48 if (!ASSERT_OK(memcmp(&insn[4], &exp, sizeof(exp)), "xchg")) 49 goto out; 50 out: 51 free(insn); 52 kptr_xchg_inline__destroy(skel); 53 } 54