1 // SPDX-License-Identifier: GPL-2.0-only 2 #define _GNU_SOURCE 3 #include <sched.h> 4 #include <sys/prctl.h> 5 #include <test_progs.h> 6 7 #define MAX_TRAMP_PROGS 38 8 9 struct inst { 10 struct bpf_object *obj; 11 struct bpf_link *link_fentry; 12 struct bpf_link *link_fexit; 13 }; 14 15 static int test_task_rename(void) 16 { 17 int fd, duration = 0, err; 18 char buf[] = "test_overhead"; 19 20 fd = open("/proc/self/comm", O_WRONLY|O_TRUNC); 21 if (CHECK(fd < 0, "open /proc", "err %d", errno)) 22 return -1; 23 err = write(fd, buf, sizeof(buf)); 24 if (err < 0) { 25 CHECK(err < 0, "task rename", "err %d", errno); 26 close(fd); 27 return -1; 28 } 29 close(fd); 30 return 0; 31 } 32 33 static struct bpf_link *load(struct bpf_object *obj, const char *name) 34 { 35 struct bpf_program *prog; 36 int duration = 0; 37 38 prog = bpf_object__find_program_by_title(obj, name); 39 if (CHECK(!prog, "find_probe", "prog '%s' not found\n", name)) 40 return ERR_PTR(-EINVAL); 41 return bpf_program__attach_trace(prog); 42 } 43 44 void test_trampoline_count(void) 45 { 46 const char *fentry_name = "fentry/__set_task_comm"; 47 const char *fexit_name = "fexit/__set_task_comm"; 48 const char *object = "test_trampoline_count.o"; 49 struct inst inst[MAX_TRAMP_PROGS] = {}; 50 int err, i = 0, duration = 0; 51 struct bpf_object *obj; 52 struct bpf_link *link; 53 char comm[16] = {}; 54 55 /* attach 'allowed' trampoline programs */ 56 for (i = 0; i < MAX_TRAMP_PROGS; i++) { 57 obj = bpf_object__open_file(object, NULL); 58 if (CHECK(IS_ERR(obj), "obj_open_file", "err %ld\n", PTR_ERR(obj))) { 59 obj = NULL; 60 goto cleanup; 61 } 62 63 err = bpf_object__load(obj); 64 if (CHECK(err, "obj_load", "err %d\n", err)) 65 goto cleanup; 66 inst[i].obj = obj; 67 obj = NULL; 68 69 if (rand() % 2) { 70 link = load(inst[i].obj, fentry_name); 71 if (CHECK(IS_ERR(link), "attach prog", "err %ld\n", PTR_ERR(link))) { 72 link = NULL; 73 goto cleanup; 74 } 75 inst[i].link_fentry = link; 76 } else { 77 link = load(inst[i].obj, fexit_name); 78 if (CHECK(IS_ERR(link), "attach prog", "err %ld\n", PTR_ERR(link))) { 79 link = NULL; 80 goto cleanup; 81 } 82 inst[i].link_fexit = link; 83 } 84 } 85 86 /* and try 1 extra.. */ 87 obj = bpf_object__open_file(object, NULL); 88 if (CHECK(IS_ERR(obj), "obj_open_file", "err %ld\n", PTR_ERR(obj))) { 89 obj = NULL; 90 goto cleanup; 91 } 92 93 err = bpf_object__load(obj); 94 if (CHECK(err, "obj_load", "err %d\n", err)) 95 goto cleanup_extra; 96 97 /* ..that needs to fail */ 98 link = load(obj, fentry_name); 99 if (CHECK(!IS_ERR(link), "cannot attach over the limit", "err %ld\n", PTR_ERR(link))) { 100 bpf_link__destroy(link); 101 goto cleanup_extra; 102 } 103 104 /* with E2BIG error */ 105 CHECK(PTR_ERR(link) != -E2BIG, "proper error check", "err %ld\n", PTR_ERR(link)); 106 107 /* and finaly execute the probe */ 108 if (CHECK_FAIL(prctl(PR_GET_NAME, comm, 0L, 0L, 0L))) 109 goto cleanup_extra; 110 CHECK_FAIL(test_task_rename()); 111 CHECK_FAIL(prctl(PR_SET_NAME, comm, 0L, 0L, 0L)); 112 113 cleanup_extra: 114 bpf_object__close(obj); 115 cleanup: 116 if (i >= MAX_TRAMP_PROGS) 117 i = MAX_TRAMP_PROGS - 1; 118 for (; i >= 0; i--) { 119 bpf_link__destroy(inst[i].link_fentry); 120 bpf_link__destroy(inst[i].link_fexit); 121 bpf_object__close(inst[i].obj); 122 } 123 } 124