51 lines · c
1// SPDX-License-Identifier: GPL-2.02#define _GNU_SOURCE3#include <test_progs.h>4#include "test_task_pt_regs.skel.h"5 6/* uprobe attach point */7static noinline void trigger_func(void)8{9 asm volatile ("");10}11 12void test_task_pt_regs(void)13{14 struct test_task_pt_regs *skel;15 struct bpf_link *uprobe_link;16 ssize_t uprobe_offset;17 bool match;18 19 uprobe_offset = get_uprobe_offset(&trigger_func);20 if (!ASSERT_GE(uprobe_offset, 0, "uprobe_offset"))21 return;22 23 skel = test_task_pt_regs__open_and_load();24 if (!ASSERT_OK_PTR(skel, "skel_open"))25 return;26 if (!ASSERT_OK_PTR(skel->bss, "check_bss"))27 goto cleanup;28 29 uprobe_link = bpf_program__attach_uprobe(skel->progs.handle_uprobe,30 false /* retprobe */,31 0 /* self pid */,32 "/proc/self/exe",33 uprobe_offset);34 if (!ASSERT_OK_PTR(uprobe_link, "attach_uprobe"))35 goto cleanup;36 skel->links.handle_uprobe = uprobe_link;37 38 /* trigger & validate uprobe */39 trigger_func();40 41 if (!ASSERT_EQ(skel->bss->uprobe_res, 1, "check_uprobe_res"))42 goto cleanup;43 44 match = !memcmp(&skel->bss->current_regs, &skel->bss->ctx_regs,45 sizeof(skel->bss->current_regs));46 ASSERT_TRUE(match, "check_regs_match");47 48cleanup:49 test_task_pt_regs__destroy(skel);50}51