529 lines
15 KiB
C
529 lines
15 KiB
C
|
// SPDX-License-Identifier: GPL-2.0
|
||
|
|
/* Copyright (c) 2021 Facebook */
|
||
|
|
#define _GNU_SOURCE
|
||
|
|
#include <pthread.h>
|
||
|
|
#include <sched.h>
|
||
|
|
#include <sys/syscall.h>
|
||
|
|
#include <sys/mman.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <test_progs.h>
|
||
|
|
#include <network_helpers.h>
|
||
|
|
#include <bpf/btf.h>
|
||
|
|
#include "test_bpf_cookie.skel.h"
|
||
|
|
#include "kprobe_multi.skel.h"
|
||
|
|
|
||
|
|
/* uprobe attach point */
|
||
|
|
static noinline void trigger_func(void)
|
||
|
|
{
|
||
|
|
asm volatile ("");
|
||
|
|
}
|
||
|
|
|
||
|
|
static void kprobe_subtest(struct test_bpf_cookie *skel)
|
||
|
|
{
|
||
|
|
DECLARE_LIBBPF_OPTS(bpf_kprobe_opts, opts);
|
||
|
|
struct bpf_link *link1 = NULL, *link2 = NULL;
|
||
|
|
struct bpf_link *retlink1 = NULL, *retlink2 = NULL;
|
||
|
|
|
||
|
|
/* attach two kprobes */
|
||
|
|
opts.bpf_cookie = 0x1;
|
||
|
|
opts.retprobe = false;
|
||
|
|
link1 = bpf_program__attach_kprobe_opts(skel->progs.handle_kprobe,
|
||
|
|
SYS_NANOSLEEP_KPROBE_NAME, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link1, "link1"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
opts.bpf_cookie = 0x2;
|
||
|
|
opts.retprobe = false;
|
||
|
|
link2 = bpf_program__attach_kprobe_opts(skel->progs.handle_kprobe,
|
||
|
|
SYS_NANOSLEEP_KPROBE_NAME, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link2, "link2"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
/* attach two kretprobes */
|
||
|
|
opts.bpf_cookie = 0x10;
|
||
|
|
opts.retprobe = true;
|
||
|
|
retlink1 = bpf_program__attach_kprobe_opts(skel->progs.handle_kretprobe,
|
||
|
|
SYS_NANOSLEEP_KPROBE_NAME, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(retlink1, "retlink1"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
opts.bpf_cookie = 0x20;
|
||
|
|
opts.retprobe = true;
|
||
|
|
retlink2 = bpf_program__attach_kprobe_opts(skel->progs.handle_kretprobe,
|
||
|
|
SYS_NANOSLEEP_KPROBE_NAME, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(retlink2, "retlink2"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
/* trigger kprobe && kretprobe */
|
||
|
|
usleep(1);
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->kprobe_res, 0x1 | 0x2, "kprobe_res");
|
||
|
|
ASSERT_EQ(skel->bss->kretprobe_res, 0x10 | 0x20, "kretprobe_res");
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
bpf_link__destroy(link1);
|
||
|
|
bpf_link__destroy(link2);
|
||
|
|
bpf_link__destroy(retlink1);
|
||
|
|
bpf_link__destroy(retlink2);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void kprobe_multi_test_run(struct kprobe_multi *skel)
|
||
|
|
{
|
||
|
|
LIBBPF_OPTS(bpf_test_run_opts, topts);
|
||
|
|
int err, prog_fd;
|
||
|
|
|
||
|
|
prog_fd = bpf_program__fd(skel->progs.trigger);
|
||
|
|
err = bpf_prog_test_run_opts(prog_fd, &topts);
|
||
|
|
ASSERT_OK(err, "test_run");
|
||
|
|
ASSERT_EQ(topts.retval, 0, "test_run");
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->kprobe_test1_result, 1, "kprobe_test1_result");
|
||
|
|
ASSERT_EQ(skel->bss->kprobe_test2_result, 1, "kprobe_test2_result");
|
||
|
|
ASSERT_EQ(skel->bss->kprobe_test3_result, 1, "kprobe_test3_result");
|
||
|
|
ASSERT_EQ(skel->bss->kprobe_test4_result, 1, "kprobe_test4_result");
|
||
|
|
ASSERT_EQ(skel->bss->kprobe_test5_result, 1, "kprobe_test5_result");
|
||
|
|
ASSERT_EQ(skel->bss->kprobe_test6_result, 1, "kprobe_test6_result");
|
||
|
|
ASSERT_EQ(skel->bss->kprobe_test7_result, 1, "kprobe_test7_result");
|
||
|
|
ASSERT_EQ(skel->bss->kprobe_test8_result, 1, "kprobe_test8_result");
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->kretprobe_test1_result, 1, "kretprobe_test1_result");
|
||
|
|
ASSERT_EQ(skel->bss->kretprobe_test2_result, 1, "kretprobe_test2_result");
|
||
|
|
ASSERT_EQ(skel->bss->kretprobe_test3_result, 1, "kretprobe_test3_result");
|
||
|
|
ASSERT_EQ(skel->bss->kretprobe_test4_result, 1, "kretprobe_test4_result");
|
||
|
|
ASSERT_EQ(skel->bss->kretprobe_test5_result, 1, "kretprobe_test5_result");
|
||
|
|
ASSERT_EQ(skel->bss->kretprobe_test6_result, 1, "kretprobe_test6_result");
|
||
|
|
ASSERT_EQ(skel->bss->kretprobe_test7_result, 1, "kretprobe_test7_result");
|
||
|
|
ASSERT_EQ(skel->bss->kretprobe_test8_result, 1, "kretprobe_test8_result");
|
||
|
|
}
|
||
|
|
|
||
|
|
static void kprobe_multi_link_api_subtest(void)
|
||
|
|
{
|
||
|
|
int prog_fd, link1_fd = -1, link2_fd = -1;
|
||
|
|
struct kprobe_multi *skel = NULL;
|
||
|
|
LIBBPF_OPTS(bpf_link_create_opts, opts);
|
||
|
|
unsigned long long addrs[8];
|
||
|
|
__u64 cookies[8];
|
||
|
|
|
||
|
|
if (!ASSERT_OK(load_kallsyms(), "load_kallsyms"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
skel = kprobe_multi__open_and_load();
|
||
|
|
if (!ASSERT_OK_PTR(skel, "fentry_raw_skel_load"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
skel->bss->pid = getpid();
|
||
|
|
skel->bss->test_cookie = true;
|
||
|
|
|
||
|
|
#define GET_ADDR(__sym, __addr) ({ \
|
||
|
|
__addr = ksym_get_addr(__sym); \
|
||
|
|
if (!ASSERT_NEQ(__addr, 0, "ksym_get_addr " #__sym)) \
|
||
|
|
goto cleanup; \
|
||
|
|
})
|
||
|
|
|
||
|
|
GET_ADDR("bpf_fentry_test1", addrs[0]);
|
||
|
|
GET_ADDR("bpf_fentry_test3", addrs[1]);
|
||
|
|
GET_ADDR("bpf_fentry_test4", addrs[2]);
|
||
|
|
GET_ADDR("bpf_fentry_test5", addrs[3]);
|
||
|
|
GET_ADDR("bpf_fentry_test6", addrs[4]);
|
||
|
|
GET_ADDR("bpf_fentry_test7", addrs[5]);
|
||
|
|
GET_ADDR("bpf_fentry_test2", addrs[6]);
|
||
|
|
GET_ADDR("bpf_fentry_test8", addrs[7]);
|
||
|
|
|
||
|
|
#undef GET_ADDR
|
||
|
|
|
||
|
|
cookies[0] = 1; /* bpf_fentry_test1 */
|
||
|
|
cookies[1] = 2; /* bpf_fentry_test3 */
|
||
|
|
cookies[2] = 3; /* bpf_fentry_test4 */
|
||
|
|
cookies[3] = 4; /* bpf_fentry_test5 */
|
||
|
|
cookies[4] = 5; /* bpf_fentry_test6 */
|
||
|
|
cookies[5] = 6; /* bpf_fentry_test7 */
|
||
|
|
cookies[6] = 7; /* bpf_fentry_test2 */
|
||
|
|
cookies[7] = 8; /* bpf_fentry_test8 */
|
||
|
|
|
||
|
|
opts.kprobe_multi.addrs = (const unsigned long *) &addrs;
|
||
|
|
opts.kprobe_multi.cnt = ARRAY_SIZE(addrs);
|
||
|
|
opts.kprobe_multi.cookies = (const __u64 *) &cookies;
|
||
|
|
prog_fd = bpf_program__fd(skel->progs.test_kprobe);
|
||
|
|
|
||
|
|
link1_fd = bpf_link_create(prog_fd, 0, BPF_TRACE_KPROBE_MULTI, &opts);
|
||
|
|
if (!ASSERT_GE(link1_fd, 0, "link1_fd"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
cookies[0] = 8; /* bpf_fentry_test1 */
|
||
|
|
cookies[1] = 7; /* bpf_fentry_test3 */
|
||
|
|
cookies[2] = 6; /* bpf_fentry_test4 */
|
||
|
|
cookies[3] = 5; /* bpf_fentry_test5 */
|
||
|
|
cookies[4] = 4; /* bpf_fentry_test6 */
|
||
|
|
cookies[5] = 3; /* bpf_fentry_test7 */
|
||
|
|
cookies[6] = 2; /* bpf_fentry_test2 */
|
||
|
|
cookies[7] = 1; /* bpf_fentry_test8 */
|
||
|
|
|
||
|
|
opts.kprobe_multi.flags = BPF_F_KPROBE_MULTI_RETURN;
|
||
|
|
prog_fd = bpf_program__fd(skel->progs.test_kretprobe);
|
||
|
|
|
||
|
|
link2_fd = bpf_link_create(prog_fd, 0, BPF_TRACE_KPROBE_MULTI, &opts);
|
||
|
|
if (!ASSERT_GE(link2_fd, 0, "link2_fd"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
kprobe_multi_test_run(skel);
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
close(link1_fd);
|
||
|
|
close(link2_fd);
|
||
|
|
kprobe_multi__destroy(skel);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void kprobe_multi_attach_api_subtest(void)
|
||
|
|
{
|
||
|
|
struct bpf_link *link1 = NULL, *link2 = NULL;
|
||
|
|
LIBBPF_OPTS(bpf_kprobe_multi_opts, opts);
|
||
|
|
LIBBPF_OPTS(bpf_test_run_opts, topts);
|
||
|
|
struct kprobe_multi *skel = NULL;
|
||
|
|
const char *syms[8] = {
|
||
|
|
"bpf_fentry_test1",
|
||
|
|
"bpf_fentry_test3",
|
||
|
|
"bpf_fentry_test4",
|
||
|
|
"bpf_fentry_test5",
|
||
|
|
"bpf_fentry_test6",
|
||
|
|
"bpf_fentry_test7",
|
||
|
|
"bpf_fentry_test2",
|
||
|
|
"bpf_fentry_test8",
|
||
|
|
};
|
||
|
|
__u64 cookies[8];
|
||
|
|
|
||
|
|
skel = kprobe_multi__open_and_load();
|
||
|
|
if (!ASSERT_OK_PTR(skel, "fentry_raw_skel_load"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
skel->bss->pid = getpid();
|
||
|
|
skel->bss->test_cookie = true;
|
||
|
|
|
||
|
|
cookies[0] = 1; /* bpf_fentry_test1 */
|
||
|
|
cookies[1] = 2; /* bpf_fentry_test3 */
|
||
|
|
cookies[2] = 3; /* bpf_fentry_test4 */
|
||
|
|
cookies[3] = 4; /* bpf_fentry_test5 */
|
||
|
|
cookies[4] = 5; /* bpf_fentry_test6 */
|
||
|
|
cookies[5] = 6; /* bpf_fentry_test7 */
|
||
|
|
cookies[6] = 7; /* bpf_fentry_test2 */
|
||
|
|
cookies[7] = 8; /* bpf_fentry_test8 */
|
||
|
|
|
||
|
|
opts.syms = syms;
|
||
|
|
opts.cnt = ARRAY_SIZE(syms);
|
||
|
|
opts.cookies = cookies;
|
||
|
|
|
||
|
|
link1 = bpf_program__attach_kprobe_multi_opts(skel->progs.test_kprobe,
|
||
|
|
NULL, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link1, "bpf_program__attach_kprobe_multi_opts"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
cookies[0] = 8; /* bpf_fentry_test1 */
|
||
|
|
cookies[1] = 7; /* bpf_fentry_test3 */
|
||
|
|
cookies[2] = 6; /* bpf_fentry_test4 */
|
||
|
|
cookies[3] = 5; /* bpf_fentry_test5 */
|
||
|
|
cookies[4] = 4; /* bpf_fentry_test6 */
|
||
|
|
cookies[5] = 3; /* bpf_fentry_test7 */
|
||
|
|
cookies[6] = 2; /* bpf_fentry_test2 */
|
||
|
|
cookies[7] = 1; /* bpf_fentry_test8 */
|
||
|
|
|
||
|
|
opts.retprobe = true;
|
||
|
|
|
||
|
|
link2 = bpf_program__attach_kprobe_multi_opts(skel->progs.test_kretprobe,
|
||
|
|
NULL, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link2, "bpf_program__attach_kprobe_multi_opts"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
kprobe_multi_test_run(skel);
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
bpf_link__destroy(link2);
|
||
|
|
bpf_link__destroy(link1);
|
||
|
|
kprobe_multi__destroy(skel);
|
||
|
|
}
|
||
|
|
static void uprobe_subtest(struct test_bpf_cookie *skel)
|
||
|
|
{
|
||
|
|
DECLARE_LIBBPF_OPTS(bpf_uprobe_opts, opts);
|
||
|
|
struct bpf_link *link1 = NULL, *link2 = NULL;
|
||
|
|
struct bpf_link *retlink1 = NULL, *retlink2 = NULL;
|
||
|
|
ssize_t uprobe_offset;
|
||
|
|
|
||
|
|
uprobe_offset = get_uprobe_offset(&trigger_func);
|
||
|
|
if (!ASSERT_GE(uprobe_offset, 0, "uprobe_offset"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
/* attach two uprobes */
|
||
|
|
opts.bpf_cookie = 0x100;
|
||
|
|
opts.retprobe = false;
|
||
|
|
link1 = bpf_program__attach_uprobe_opts(skel->progs.handle_uprobe, 0 /* self pid */,
|
||
|
|
"/proc/self/exe", uprobe_offset, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link1, "link1"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
opts.bpf_cookie = 0x200;
|
||
|
|
opts.retprobe = false;
|
||
|
|
link2 = bpf_program__attach_uprobe_opts(skel->progs.handle_uprobe, -1 /* any pid */,
|
||
|
|
"/proc/self/exe", uprobe_offset, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link2, "link2"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
/* attach two uretprobes */
|
||
|
|
opts.bpf_cookie = 0x1000;
|
||
|
|
opts.retprobe = true;
|
||
|
|
retlink1 = bpf_program__attach_uprobe_opts(skel->progs.handle_uretprobe, -1 /* any pid */,
|
||
|
|
"/proc/self/exe", uprobe_offset, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(retlink1, "retlink1"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
opts.bpf_cookie = 0x2000;
|
||
|
|
opts.retprobe = true;
|
||
|
|
retlink2 = bpf_program__attach_uprobe_opts(skel->progs.handle_uretprobe, 0 /* self pid */,
|
||
|
|
"/proc/self/exe", uprobe_offset, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(retlink2, "retlink2"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
/* trigger uprobe && uretprobe */
|
||
|
|
trigger_func();
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->uprobe_res, 0x100 | 0x200, "uprobe_res");
|
||
|
|
ASSERT_EQ(skel->bss->uretprobe_res, 0x1000 | 0x2000, "uretprobe_res");
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
bpf_link__destroy(link1);
|
||
|
|
bpf_link__destroy(link2);
|
||
|
|
bpf_link__destroy(retlink1);
|
||
|
|
bpf_link__destroy(retlink2);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void tp_subtest(struct test_bpf_cookie *skel)
|
||
|
|
{
|
||
|
|
DECLARE_LIBBPF_OPTS(bpf_tracepoint_opts, opts);
|
||
|
|
struct bpf_link *link1 = NULL, *link2 = NULL, *link3 = NULL;
|
||
|
|
|
||
|
|
/* attach first tp prog */
|
||
|
|
opts.bpf_cookie = 0x10000;
|
||
|
|
link1 = bpf_program__attach_tracepoint_opts(skel->progs.handle_tp1,
|
||
|
|
"syscalls", "sys_enter_nanosleep", &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link1, "link1"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
/* attach second tp prog */
|
||
|
|
opts.bpf_cookie = 0x20000;
|
||
|
|
link2 = bpf_program__attach_tracepoint_opts(skel->progs.handle_tp2,
|
||
|
|
"syscalls", "sys_enter_nanosleep", &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link2, "link2"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
/* trigger tracepoints */
|
||
|
|
usleep(1);
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->tp_res, 0x10000 | 0x20000, "tp_res1");
|
||
|
|
|
||
|
|
/* now we detach first prog and will attach third one, which causes
|
||
|
|
* two internal calls to bpf_prog_array_copy(), shuffling
|
||
|
|
* bpf_prog_array_items around. We test here that we don't lose track
|
||
|
|
* of associated bpf_cookies.
|
||
|
|
*/
|
||
|
|
bpf_link__destroy(link1);
|
||
|
|
link1 = NULL;
|
||
|
|
kern_sync_rcu();
|
||
|
|
skel->bss->tp_res = 0;
|
||
|
|
|
||
|
|
/* attach third tp prog */
|
||
|
|
opts.bpf_cookie = 0x40000;
|
||
|
|
link3 = bpf_program__attach_tracepoint_opts(skel->progs.handle_tp3,
|
||
|
|
"syscalls", "sys_enter_nanosleep", &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link3, "link3"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
/* trigger tracepoints */
|
||
|
|
usleep(1);
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->tp_res, 0x20000 | 0x40000, "tp_res2");
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
bpf_link__destroy(link1);
|
||
|
|
bpf_link__destroy(link2);
|
||
|
|
bpf_link__destroy(link3);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void burn_cpu(void)
|
||
|
|
{
|
||
|
|
volatile int j = 0;
|
||
|
|
cpu_set_t cpu_set;
|
||
|
|
int i, err;
|
||
|
|
|
||
|
|
/* generate some branches on cpu 0 */
|
||
|
|
CPU_ZERO(&cpu_set);
|
||
|
|
CPU_SET(0, &cpu_set);
|
||
|
|
err = pthread_setaffinity_np(pthread_self(), sizeof(cpu_set), &cpu_set);
|
||
|
|
ASSERT_OK(err, "set_thread_affinity");
|
||
|
|
|
||
|
|
/* spin the loop for a while (random high number) */
|
||
|
|
for (i = 0; i < 1000000; ++i)
|
||
|
|
++j;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void pe_subtest(struct test_bpf_cookie *skel)
|
||
|
|
{
|
||
|
|
DECLARE_LIBBPF_OPTS(bpf_perf_event_opts, opts);
|
||
|
|
struct bpf_link *link = NULL;
|
||
|
|
struct perf_event_attr attr;
|
||
|
|
int pfd = -1;
|
||
|
|
|
||
|
|
/* create perf event */
|
||
|
|
memset(&attr, 0, sizeof(attr));
|
||
|
|
attr.size = sizeof(attr);
|
||
|
|
attr.type = PERF_TYPE_SOFTWARE;
|
||
|
|
attr.config = PERF_COUNT_SW_CPU_CLOCK;
|
||
|
|
attr.freq = 1;
|
||
|
|
attr.sample_freq = 1000;
|
||
|
|
pfd = syscall(__NR_perf_event_open, &attr, -1, 0, -1, PERF_FLAG_FD_CLOEXEC);
|
||
|
|
if (!ASSERT_GE(pfd, 0, "perf_fd"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
opts.bpf_cookie = 0x100000;
|
||
|
|
link = bpf_program__attach_perf_event_opts(skel->progs.handle_pe, pfd, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link, "link1"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
burn_cpu(); /* trigger BPF prog */
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->pe_res, 0x100000, "pe_res1");
|
||
|
|
|
||
|
|
/* prevent bpf_link__destroy() closing pfd itself */
|
||
|
|
bpf_link__disconnect(link);
|
||
|
|
/* close BPF link's FD explicitly */
|
||
|
|
close(bpf_link__fd(link));
|
||
|
|
/* free up memory used by struct bpf_link */
|
||
|
|
bpf_link__destroy(link);
|
||
|
|
link = NULL;
|
||
|
|
kern_sync_rcu();
|
||
|
|
skel->bss->pe_res = 0;
|
||
|
|
|
||
|
|
opts.bpf_cookie = 0x200000;
|
||
|
|
link = bpf_program__attach_perf_event_opts(skel->progs.handle_pe, pfd, &opts);
|
||
|
|
if (!ASSERT_OK_PTR(link, "link2"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
burn_cpu(); /* trigger BPF prog */
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->pe_res, 0x200000, "pe_res2");
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
close(pfd);
|
||
|
|
bpf_link__destroy(link);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void tracing_subtest(struct test_bpf_cookie *skel)
|
||
|
|
{
|
||
|
|
__u64 cookie;
|
||
|
|
int prog_fd;
|
||
|
|
int fentry_fd = -1, fexit_fd = -1, fmod_ret_fd = -1;
|
||
|
|
LIBBPF_OPTS(bpf_test_run_opts, opts);
|
||
|
|
LIBBPF_OPTS(bpf_link_create_opts, link_opts);
|
||
|
|
|
||
|
|
skel->bss->fentry_res = 0;
|
||
|
|
skel->bss->fexit_res = 0;
|
||
|
|
|
||
|
|
cookie = 0x10000000000000L;
|
||
|
|
prog_fd = bpf_program__fd(skel->progs.fentry_test1);
|
||
|
|
link_opts.tracing.cookie = cookie;
|
||
|
|
fentry_fd = bpf_link_create(prog_fd, 0, BPF_TRACE_FENTRY, &link_opts);
|
||
|
|
if (!ASSERT_GE(fentry_fd, 0, "fentry.link_create"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
cookie = 0x20000000000000L;
|
||
|
|
prog_fd = bpf_program__fd(skel->progs.fexit_test1);
|
||
|
|
link_opts.tracing.cookie = cookie;
|
||
|
|
fexit_fd = bpf_link_create(prog_fd, 0, BPF_TRACE_FEXIT, &link_opts);
|
||
|
|
if (!ASSERT_GE(fexit_fd, 0, "fexit.link_create"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
cookie = 0x30000000000000L;
|
||
|
|
prog_fd = bpf_program__fd(skel->progs.fmod_ret_test);
|
||
|
|
link_opts.tracing.cookie = cookie;
|
||
|
|
fmod_ret_fd = bpf_link_create(prog_fd, 0, BPF_MODIFY_RETURN, &link_opts);
|
||
|
|
if (!ASSERT_GE(fmod_ret_fd, 0, "fmod_ret.link_create"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
prog_fd = bpf_program__fd(skel->progs.fentry_test1);
|
||
|
|
bpf_prog_test_run_opts(prog_fd, &opts);
|
||
|
|
|
||
|
|
prog_fd = bpf_program__fd(skel->progs.fmod_ret_test);
|
||
|
|
bpf_prog_test_run_opts(prog_fd, &opts);
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->fentry_res, 0x10000000000000L, "fentry_res");
|
||
|
|
ASSERT_EQ(skel->bss->fexit_res, 0x20000000000000L, "fexit_res");
|
||
|
|
ASSERT_EQ(skel->bss->fmod_ret_res, 0x30000000000000L, "fmod_ret_res");
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
if (fentry_fd >= 0)
|
||
|
|
close(fentry_fd);
|
||
|
|
if (fexit_fd >= 0)
|
||
|
|
close(fexit_fd);
|
||
|
|
if (fmod_ret_fd >= 0)
|
||
|
|
close(fmod_ret_fd);
|
||
|
|
}
|
||
|
|
|
||
|
|
int stack_mprotect(void);
|
||
|
|
|
||
|
|
static void lsm_subtest(struct test_bpf_cookie *skel)
|
||
|
|
{
|
||
|
|
__u64 cookie;
|
||
|
|
int prog_fd;
|
||
|
|
int lsm_fd = -1;
|
||
|
|
LIBBPF_OPTS(bpf_link_create_opts, link_opts);
|
||
|
|
int err;
|
||
|
|
|
||
|
|
skel->bss->lsm_res = 0;
|
||
|
|
|
||
|
|
cookie = 0x90000000000090L;
|
||
|
|
prog_fd = bpf_program__fd(skel->progs.test_int_hook);
|
||
|
|
link_opts.tracing.cookie = cookie;
|
||
|
|
lsm_fd = bpf_link_create(prog_fd, 0, BPF_LSM_MAC, &link_opts);
|
||
|
|
if (!ASSERT_GE(lsm_fd, 0, "lsm.link_create"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
err = stack_mprotect();
|
||
|
|
if (!ASSERT_EQ(err, -1, "stack_mprotect") ||
|
||
|
|
!ASSERT_EQ(errno, EPERM, "stack_mprotect"))
|
||
|
|
goto cleanup;
|
||
|
|
|
||
|
|
usleep(1);
|
||
|
|
|
||
|
|
ASSERT_EQ(skel->bss->lsm_res, 0x90000000000090L, "fentry_res");
|
||
|
|
|
||
|
|
cleanup:
|
||
|
|
if (lsm_fd >= 0)
|
||
|
|
close(lsm_fd);
|
||
|
|
}
|
||
|
|
|
||
|
|
void test_bpf_cookie(void)
|
||
|
|
{
|
||
|
|
struct test_bpf_cookie *skel;
|
||
|
|
|
||
|
|
skel = test_bpf_cookie__open_and_load();
|
||
|
|
if (!ASSERT_OK_PTR(skel, "skel_open"))
|
||
|
|
return;
|
||
|
|
|
||
|
|
skel->bss->my_tid = syscall(SYS_gettid);
|
||
|
|
|
||
|
|
if (test__start_subtest("kprobe"))
|
||
|
|
kprobe_subtest(skel);
|
||
|
|
if (test__start_subtest("multi_kprobe_link_api"))
|
||
|
|
kprobe_multi_link_api_subtest();
|
||
|
|
if (test__start_subtest("multi_kprobe_attach_api"))
|
||
|
|
kprobe_multi_attach_api_subtest();
|
||
|
|
if (test__start_subtest("uprobe"))
|
||
|
|
uprobe_subtest(skel);
|
||
|
|
if (test__start_subtest("tracepoint"))
|
||
|
|
tp_subtest(skel);
|
||
|
|
if (test__start_subtest("perf_event"))
|
||
|
|
pe_subtest(skel);
|
||
|
|
if (test__start_subtest("trampoline"))
|
||
|
|
tracing_subtest(skel);
|
||
|
|
if (test__start_subtest("lsm"))
|
||
|
|
lsm_subtest(skel);
|
||
|
|
|
||
|
|
test_bpf_cookie__destroy(skel);
|
||
|
|
}
|