1 // SPDX-License-Identifier: GPL-2.0 2 #define _GNU_SOURCE 3 #include <test_progs.h> 4 #include "test_task_pt_regs.skel.h" 5 6 /* uprobe attach point */ 7 static void trigger_func(void) 8 { 9 asm volatile (""); 10 } 11 12 void 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 48 cleanup: 49 test_task_pt_regs__destroy(skel); 50 } 51