linux/tools/testing/selftests/bpf/prog_tests/kptr_xchg_inline.c

// SPDX-License-Identifier: GPL-2.0
/* Copyright (C) 2023. Huawei Technologies Co., Ltd */
#include <test_progs.h>

#include "linux/filter.h"
#include "kptr_xchg_inline.skel.h"

void test_kptr_xchg_inline(void)
{
	struct kptr_xchg_inline *skel;
	struct bpf_insn *insn = NULL;
	struct bpf_insn exp;
	unsigned int cnt;
	int err;

#if !(defined(__x86_64__) || defined(__aarch64__) || \
      (defined(__riscv) && __riscv_xlen == 64))
	test__skip();
	return;
#endif

	skel = kptr_xchg_inline__open_and_load();
	if (!ASSERT_OK_PTR(skel, "open_load"))
		return;

	err = get_xlated_program(bpf_program__fd(skel->progs.kptr_xchg_inline), &insn, &cnt);
	if (!ASSERT_OK(err, "prog insn"))
		goto out;

	/* The original instructions are:
	 * r1 = map[id:xxx][0]+0
	 * r2 = 0
	 * call bpf_kptr_xchg#yyy
	 *
	 * call bpf_kptr_xchg#yyy will be inlined as:
	 * r0 = r2
	 * r0 = atomic64_xchg((u64 *)(r1 +0), r0)
	 */
	if (!ASSERT_GT(cnt, 5, "insn cnt"))
		goto out;

	exp = BPF_MOV64_REG(BPF_REG_0, BPF_REG_2);
	if (!ASSERT_OK(memcmp(&insn[3], &exp, sizeof(exp)), "mov"))
		goto out;

	exp = BPF_ATOMIC_OP(BPF_DW, BPF_XCHG, BPF_REG_1, BPF_REG_0, 0);
	if (!ASSERT_OK(memcmp(&insn[4], &exp, sizeof(exp)), "xchg"))
		goto out;
out:
	free(insn);
	kptr_xchg_inline__destroy(skel);
}