mirror of
https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
synced 2026-08-28 22:44:08 -04:00
1b938f42f5
Adding tests for using cookies on tracing multi link. Signed-off-by: Jiri Olsa <jolsa@kernel.org> Link: https://lore.kernel.org/r/20260606123955.345967-25-jolsa@kernel.org Signed-off-by: Alexei Starovoitov <ast@kernel.org>
215 lines
5.4 KiB
C
215 lines
5.4 KiB
C
// SPDX-License-Identifier: GPL-2.0
|
|
#include <vmlinux.h>
|
|
#include <bpf/bpf_helpers.h>
|
|
#include <bpf/bpf_tracing.h>
|
|
|
|
char _license[] SEC("license") = "GPL";
|
|
|
|
int pid = 0;
|
|
bool test_cookies = false;
|
|
|
|
/* bpf_fentry_test1 is exported as kfunc via vmlinux.h */
|
|
extern const void bpf_fentry_test2 __ksym;
|
|
extern const void bpf_fentry_test3 __ksym;
|
|
extern const void bpf_fentry_test4 __ksym;
|
|
extern const void bpf_fentry_test5 __ksym;
|
|
extern const void bpf_fentry_test6 __ksym;
|
|
extern const void bpf_fentry_test7 __ksym;
|
|
extern const void bpf_fentry_test8 __ksym;
|
|
extern const void bpf_fentry_test9 __ksym;
|
|
extern const void bpf_fentry_test10 __ksym;
|
|
|
|
extern const void bpf_testmod_fentry_test1 __ksym;
|
|
extern const void bpf_testmod_fentry_test2 __ksym;
|
|
extern const void bpf_testmod_fentry_test3 __ksym;
|
|
extern const void bpf_testmod_fentry_test7 __ksym;
|
|
extern const void bpf_testmod_fentry_test11 __ksym;
|
|
|
|
int tracing_multi_arg_check(__u64 *ctx, __u64 *test_result, bool is_return)
|
|
{
|
|
void *ip = (void *) bpf_get_func_ip(ctx);
|
|
__u64 value = 0, ret = 0, cookie = 0;
|
|
long err = 0;
|
|
|
|
if (bpf_get_current_pid_tgid() >> 32 != pid)
|
|
return 1;
|
|
|
|
if (is_return)
|
|
err |= bpf_get_func_ret(ctx, &ret);
|
|
if (test_cookies)
|
|
cookie = bpf_get_attach_cookie(ctx);
|
|
|
|
if (ip == &bpf_fentry_test1) {
|
|
int a;
|
|
|
|
err |= bpf_get_func_arg(ctx, 0, &value);
|
|
a = (int) value;
|
|
|
|
err |= is_return ? ret != 2 : 0;
|
|
err |= test_cookies ? cookie != 8 : 0;
|
|
|
|
*test_result += err == 0 && a == 1;
|
|
} else if (ip == &bpf_fentry_test2) {
|
|
__u64 b;
|
|
int a;
|
|
|
|
err |= bpf_get_func_arg(ctx, 0, &value);
|
|
a = (int) value;
|
|
err |= bpf_get_func_arg(ctx, 1, &value);
|
|
b = value;
|
|
|
|
err |= is_return ? ret != 5 : 0;
|
|
err |= test_cookies ? cookie != 9 : 0;
|
|
|
|
*test_result += err == 0 && a == 2 && b == 3;
|
|
} else if (ip == &bpf_fentry_test3) {
|
|
__u64 c;
|
|
char a;
|
|
int b;
|
|
|
|
err |= bpf_get_func_arg(ctx, 0, &value);
|
|
a = (char) value;
|
|
err |= bpf_get_func_arg(ctx, 1, &value);
|
|
b = (int) value;
|
|
err |= bpf_get_func_arg(ctx, 2, &value);
|
|
c = value;
|
|
|
|
err |= is_return ? ret != 15 : 0;
|
|
err |= test_cookies ? cookie != 7 : 0;
|
|
|
|
*test_result += err == 0 && a == 4 && b == 5 && c == 6;
|
|
} else if (ip == &bpf_fentry_test4) {
|
|
void *a;
|
|
char b;
|
|
int c;
|
|
__u64 d;
|
|
|
|
err |= bpf_get_func_arg(ctx, 0, &value);
|
|
a = (void *) value;
|
|
err |= bpf_get_func_arg(ctx, 1, &value);
|
|
b = (char) value;
|
|
err |= bpf_get_func_arg(ctx, 2, &value);
|
|
c = (int) value;
|
|
err |= bpf_get_func_arg(ctx, 3, &value);
|
|
d = value;
|
|
|
|
err |= is_return ? ret != 34 : 0;
|
|
err |= test_cookies ? cookie != 5 : 0;
|
|
|
|
*test_result += err == 0 && a == (void *) 7 && b == 8 && c == 9 && d == 10;
|
|
} else if (ip == &bpf_fentry_test5) {
|
|
__u64 a;
|
|
void *b;
|
|
short c;
|
|
int d;
|
|
__u64 e;
|
|
|
|
err |= bpf_get_func_arg(ctx, 0, &value);
|
|
a = value;
|
|
err |= bpf_get_func_arg(ctx, 1, &value);
|
|
b = (void *) value;
|
|
err |= bpf_get_func_arg(ctx, 2, &value);
|
|
c = (short) value;
|
|
err |= bpf_get_func_arg(ctx, 3, &value);
|
|
d = (int) value;
|
|
err |= bpf_get_func_arg(ctx, 4, &value);
|
|
e = value;
|
|
|
|
err |= is_return ? ret != 65 : 0;
|
|
err |= test_cookies ? cookie != 4 : 0;
|
|
|
|
*test_result += err == 0 && a == 11 && b == (void *) 12 && c == 13 && d == 14 && e == 15;
|
|
} else if (ip == &bpf_fentry_test6) {
|
|
__u64 a;
|
|
void *b;
|
|
short c;
|
|
int d;
|
|
void *e;
|
|
__u64 f;
|
|
|
|
err |= bpf_get_func_arg(ctx, 0, &value);
|
|
a = value;
|
|
err |= bpf_get_func_arg(ctx, 1, &value);
|
|
b = (void *) value;
|
|
err |= bpf_get_func_arg(ctx, 2, &value);
|
|
c = (short) value;
|
|
err |= bpf_get_func_arg(ctx, 3, &value);
|
|
d = (int) value;
|
|
err |= bpf_get_func_arg(ctx, 4, &value);
|
|
e = (void *) value;
|
|
err |= bpf_get_func_arg(ctx, 5, &value);
|
|
f = value;
|
|
|
|
err |= is_return ? ret != 111 : 0;
|
|
err |= test_cookies ? cookie != 2 : 0;
|
|
|
|
*test_result += err == 0 && a == 16 && b == (void *) 17 && c == 18 && d == 19 && e == (void *) 20 && f == 21;
|
|
} else if (ip == &bpf_fentry_test7) {
|
|
err |= is_return ? ret != 0 : 0;
|
|
err |= test_cookies ? cookie != 3 : 0;
|
|
|
|
*test_result += err == 0 ? 1 : 0;
|
|
} else if (ip == &bpf_fentry_test8) {
|
|
err |= is_return ? ret != 0 : 0;
|
|
err |= test_cookies ? cookie != 1 : 0;
|
|
|
|
*test_result += err == 0 ? 1 : 0;
|
|
} else if (ip == &bpf_fentry_test9) {
|
|
err |= is_return ? ret != 0 : 0;
|
|
err |= test_cookies ? cookie != 10 : 0;
|
|
|
|
*test_result += err == 0 ? 1 : 0;
|
|
} else if (ip == &bpf_fentry_test10) {
|
|
err |= is_return ? ret != 0 : 0;
|
|
err |= test_cookies ? cookie != 6 : 0;
|
|
|
|
*test_result += err == 0 ? 1 : 0;
|
|
} else if (ip == &bpf_testmod_fentry_test1) {
|
|
int a;
|
|
|
|
err |= bpf_get_func_arg(ctx, 0, &value);
|
|
a = (int) value;
|
|
|
|
err |= is_return ? ret != 2 : 0;
|
|
|
|
*test_result += err == 0 && a == 1;
|
|
} else if (ip == &bpf_testmod_fentry_test2) {
|
|
int a;
|
|
__u64 b;
|
|
|
|
err |= bpf_get_func_arg(ctx, 0, &value);
|
|
a = (int) value;
|
|
err |= bpf_get_func_arg(ctx, 1, &value);
|
|
b = (__u64) value;
|
|
|
|
err |= is_return ? ret != 5 : 0;
|
|
|
|
*test_result += err == 0 && a == 2 && b == 3;
|
|
} else if (ip == &bpf_testmod_fentry_test3) {
|
|
char a;
|
|
int b;
|
|
__u64 c;
|
|
|
|
err |= bpf_get_func_arg(ctx, 0, &value);
|
|
a = (char) value;
|
|
err |= bpf_get_func_arg(ctx, 1, &value);
|
|
b = (int) value;
|
|
err |= bpf_get_func_arg(ctx, 2, &value);
|
|
c = (__u64) value;
|
|
|
|
err |= is_return ? ret != 15 : 0;
|
|
|
|
*test_result += err == 0 && a == 4 && b == 5 && c == 6;
|
|
} else if (ip == &bpf_testmod_fentry_test7) {
|
|
err |= is_return ? ret != 133 : 0;
|
|
|
|
*test_result += err == 0;
|
|
} else if (ip == &bpf_testmod_fentry_test11) {
|
|
err |= is_return ? ret != 231 : 0;
|
|
|
|
*test_result += err == 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|