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
test_kptr_xchg_inline(void)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