| 1 | |
| 2 | #include <test_progs.h>
|
| 3 |
|
| 4 | void test_reference_tracking(void)
|
| 5 | {
|
| 6 | const char *file = "test_sk_lookup_kern.bpf.o";
|
| 7 | const char *obj_name = "ref_track";
|
| 8 | DECLARE_LIBBPF_OPTS(bpf_object_open_opts, open_opts,
|
| 9 | .object_name = obj_name,
|
| 10 | .relaxed_maps = true,
|
| 11 | );
|
| 12 | struct bpf_object *obj_iter, *obj = NULL;
|
| 13 | struct bpf_program *prog;
|
| 14 | __u32 duration = 0;
|
| 15 | int err = 0;
|
| 16 |
|
| 17 | obj_iter = bpf_object__open_file(file, &open_opts);
|
| 18 | if (!ASSERT_OK_PTR(obj_iter, "obj_iter_open_file"))
|
| 19 | return;
|
| 20 |
|
| 21 | if (CHECK(strcmp(bpf_object__name(obj_iter), obj_name), "obj_name",
|
| 22 | "wrong obj name '%s', expected '%s'\n",
|
| 23 | bpf_object__name(obj_iter), obj_name))
|
| 24 | goto cleanup;
|
| 25 |
|
| 26 | bpf_object__for_each_program(prog, obj_iter) {
|
| 27 | struct bpf_program *p;
|
| 28 | const char *name;
|
| 29 |
|
| 30 | name = bpf_program__name(prog);
|
| 31 | if (!test__start_subtest(name))
|
| 32 | continue;
|
| 33 |
|
| 34 | obj = bpf_object__open_file(file, &open_opts);
|
| 35 | if (!ASSERT_OK_PTR(obj, "obj_open_file"))
|
| 36 | goto cleanup;
|
| 37 |
|
| 38 | |
| 39 | * autoload to true for the single prog under test
|
| 40 |
|
| 41 | p = bpf_object__find_program_by_name(obj, name);
|
| 42 | bpf_program__set_autoload(p, true);
|
| 43 |
|
| 44 |
|
| 45 | if (strncmp(name, "err_", sizeof("err_") - 1) == 0) {
|
| 46 | libbpf_print_fn_t old_print_fn;
|
| 47 |
|
| 48 | old_print_fn = libbpf_set_print(NULL);
|
| 49 | err = !bpf_object__load(obj);
|
| 50 | libbpf_set_print(old_print_fn);
|
| 51 | } else {
|
| 52 | err = bpf_object__load(obj);
|
| 53 | }
|
| 54 | ASSERT_OK(err, name);
|
| 55 |
|
| 56 | bpf_object__close(obj);
|
| 57 | obj = NULL;
|
| 58 | }
|
| 59 |
|
| 60 | cleanup:
|
| 61 | bpf_object__close(obj);
|
| 62 | bpf_object__close(obj_iter);
|
| 63 | }
|