1// SPDX-License-Identifier: GPL-2.0-only
  2#define _GNU_SOURCE
  3#include <test_progs.h>
  4
 
 
  5struct inst {
  6	struct bpf_object *obj;
  7	struct bpf_link   *link;
  8};
  9
 10static struct bpf_program *load_prog(char *file, char *name, struct inst *inst)
 11{
 12	struct bpf_object *obj;
 13	struct bpf_program *prog;
 14	int err;
 15
 16	obj = bpf_object__open_file(file, NULL);
 17	if (!ASSERT_OK_PTR(obj, "obj_open_file"))
 18		return NULL;
 19
 20	inst->obj = obj;
 21
 22	err = bpf_object__load(obj);
 23	if (!ASSERT_OK(err, "obj_load"))
 24		return NULL;
 25
 26	prog = bpf_object__find_program_by_name(obj, name);
 27	if (!ASSERT_OK_PTR(prog, "obj_find_prog"))
 28		return NULL;
 29
 30	return prog;
 31}
 32
 33/* TODO: use different target function to run in concurrent mode */
 34void serial_test_trampoline_count(void)
 35{
 36	char *file = "test_trampoline_count.bpf.o";
 37	char *const progs[] = { "fentry_test", "fmod_ret_test", "fexit_test" };
 38	int bpf_max_tramp_links, err, i, prog_fd;
 39	struct bpf_program *prog;
 40	struct bpf_link *link;
 41	struct inst *inst;
 42	LIBBPF_OPTS(bpf_test_run_opts, opts);
 43
 44	bpf_max_tramp_links = get_bpf_max_tramp_links();
 45	if (!ASSERT_GE(bpf_max_tramp_links, 1, "bpf_max_tramp_links"))
 46		return;
 47	inst = calloc(bpf_max_tramp_links + 1, sizeof(*inst));
 48	if (!ASSERT_OK_PTR(inst, "inst"))
 49		return;
 50
 51	/* attach 'allowed' trampoline programs */
 52	for (i = 0; i < bpf_max_tramp_links; i++) {
 53		prog = load_prog(file, progs[i % ARRAY_SIZE(progs)], &inst[i]);
 54		if (!prog)
 55			goto cleanup;
 56
 57		link = bpf_program__attach(prog);
 58		if (!ASSERT_OK_PTR(link, "attach_prog"))
 59			goto cleanup;
 60
 61		inst[i].link = link;
 62	}
 63
 64	/* and try 1 extra.. */
 65	prog = load_prog(file, "fmod_ret_test", &inst[i]);
 66	if (!prog)
 67		goto cleanup;
 68
 69	/* ..that needs to fail */
 70	link = bpf_program__attach(prog);
 71	if (!ASSERT_ERR_PTR(link, "attach_prog")) {
 72		inst[i].link = link;
 73		goto cleanup;
 74	}
 75
 76	/* with E2BIG error */
 77	if (!ASSERT_EQ(libbpf_get_error(link), -E2BIG, "E2BIG"))
 78		goto cleanup;
 79	if (!ASSERT_EQ(link, NULL, "ptr_is_null"))
 80		goto cleanup;
 81
 82	/* and finally execute the probe */
 83	prog_fd = bpf_program__fd(prog);
 84	if (!ASSERT_GE(prog_fd, 0, "bpf_program__fd"))
 85		goto cleanup;
 86
 87	err = bpf_prog_test_run_opts(prog_fd, &opts);
 88	if (!ASSERT_OK(err, "bpf_prog_test_run_opts"))
 89		goto cleanup;
 90
 91	ASSERT_EQ(opts.retval & 0xffff, 33, "bpf_modify_return_test.result");
 92	ASSERT_EQ(opts.retval >> 16, 2, "bpf_modify_return_test.side_effect");
 93
 94cleanup:
 95	for (; i >= 0; i--) {
 96		bpf_link__destroy(inst[i].link);
 97		bpf_object__close(inst[i].obj);
 98	}
 99	free(inst);
100}

 1// SPDX-License-Identifier: GPL-2.0-only
 2#define _GNU_SOURCE
 3#include <test_progs.h>
 4
 5#define MAX_TRAMP_PROGS 38
 6
 7struct inst {
 8	struct bpf_object *obj;
 9	struct bpf_link   *link;
10};
11
12static struct bpf_program *load_prog(char *file, char *name, struct inst *inst)
13{
14	struct bpf_object *obj;
15	struct bpf_program *prog;
16	int err;
17
18	obj = bpf_object__open_file(file, NULL);
19	if (!ASSERT_OK_PTR(obj, "obj_open_file"))
20		return NULL;
21
22	inst->obj = obj;
23
24	err = bpf_object__load(obj);
25	if (!ASSERT_OK(err, "obj_load"))
26		return NULL;
27
28	prog = bpf_object__find_program_by_name(obj, name);
29	if (!ASSERT_OK_PTR(prog, "obj_find_prog"))
30		return NULL;
31
32	return prog;
33}
34
35/* TODO: use different target function to run in concurrent mode */
36void serial_test_trampoline_count(void)
37{
38	char *file = "test_trampoline_count.bpf.o";
39	char *const progs[] = { "fentry_test", "fmod_ret_test", "fexit_test" };
40	struct inst inst[MAX_TRAMP_PROGS + 1] = {};
41	struct bpf_program *prog;
42	struct bpf_link *link;
43	int prog_fd, err, i;
44	LIBBPF_OPTS(bpf_test_run_opts, opts);
45
 
 
 
 
 
 
 
46	/* attach 'allowed' trampoline programs */
47	for (i = 0; i < MAX_TRAMP_PROGS; i++) {
48		prog = load_prog(file, progs[i % ARRAY_SIZE(progs)], &inst[i]);
49		if (!prog)
50			goto cleanup;
51
52		link = bpf_program__attach(prog);
53		if (!ASSERT_OK_PTR(link, "attach_prog"))
54			goto cleanup;
55
56		inst[i].link = link;
57	}
58
59	/* and try 1 extra.. */
60	prog = load_prog(file, "fmod_ret_test", &inst[i]);
61	if (!prog)
62		goto cleanup;
63
64	/* ..that needs to fail */
65	link = bpf_program__attach(prog);
66	if (!ASSERT_ERR_PTR(link, "attach_prog")) {
67		inst[i].link = link;
68		goto cleanup;
69	}
70
71	/* with E2BIG error */
72	if (!ASSERT_EQ(libbpf_get_error(link), -E2BIG, "E2BIG"))
73		goto cleanup;
74	if (!ASSERT_EQ(link, NULL, "ptr_is_null"))
75		goto cleanup;
76
77	/* and finaly execute the probe */
78	prog_fd = bpf_program__fd(prog);
79	if (!ASSERT_GE(prog_fd, 0, "bpf_program__fd"))
80		goto cleanup;
81
82	err = bpf_prog_test_run_opts(prog_fd, &opts);
83	if (!ASSERT_OK(err, "bpf_prog_test_run_opts"))
84		goto cleanup;
85
86	ASSERT_EQ(opts.retval & 0xffff, 4, "bpf_modify_return_test.result");
87	ASSERT_EQ(opts.retval >> 16, 1, "bpf_modify_return_test.side_effect");
88
89cleanup:
90	for (; i >= 0; i--) {
91		bpf_link__destroy(inst[i].link);
92		bpf_object__close(inst[i].obj);
93	}
 
94}