// SPDX-License-Identifier: GPL-2.0 #define _GNU_SOURCE #include #include "test_task_pt_regs.skel.h" void test_task_pt_regs(void) { struct test_task_pt_regs *skel; struct bpf_link *uprobe_link; size_t uprobe_offset; ssize_t base_addr; bool match; base_addr = get_base_addr(); if (!ASSERT_GT(base_addr, 0, "get_base_addr")) return; uprobe_offset = get_uprobe_offset(&get_base_addr, base_addr); skel = test_task_pt_regs__open_and_load(); if (!ASSERT_OK_PTR(skel, "skel_open")) return; if (!ASSERT_OK_PTR(skel->bss, "check_bss")) goto cleanup; uprobe_link = bpf_program__attach_uprobe(skel->progs.handle_uprobe, false /* retprobe */, 0 /* self pid */, "/proc/self/exe", uprobe_offset); if (!ASSERT_OK_PTR(uprobe_link, "attach_uprobe")) goto cleanup; skel->links.handle_uprobe = uprobe_link; /* trigger & validate uprobe */ get_base_addr(); if (!ASSERT_EQ(skel->bss->uprobe_res, 1, "check_uprobe_res")) goto cleanup; match = !memcmp(&skel->bss->current_regs, &skel->bss->ctx_regs, sizeof(skel->bss->current_regs)); ASSERT_TRUE(match, "check_regs_match"); cleanup: test_task_pt_regs__destroy(skel); }