11e8611bbSAndrii Nakryiko // SPDX-License-Identifier: GPL-2.0 21e8611bbSAndrii Nakryiko #include <test_progs.h> 31e8611bbSAndrii Nakryiko 41e8611bbSAndrii Nakryiko ssize_t get_base_addr() { 51e8611bbSAndrii Nakryiko size_t start; 61e8611bbSAndrii Nakryiko char buf[256]; 71e8611bbSAndrii Nakryiko FILE *f; 81e8611bbSAndrii Nakryiko 91e8611bbSAndrii Nakryiko f = fopen("/proc/self/maps", "r"); 101e8611bbSAndrii Nakryiko if (!f) 111e8611bbSAndrii Nakryiko return -errno; 121e8611bbSAndrii Nakryiko 131e8611bbSAndrii Nakryiko while (fscanf(f, "%zx-%*x %s %*s\n", &start, buf) == 2) { 141e8611bbSAndrii Nakryiko if (strcmp(buf, "r-xp") == 0) { 151e8611bbSAndrii Nakryiko fclose(f); 161e8611bbSAndrii Nakryiko return start; 171e8611bbSAndrii Nakryiko } 181e8611bbSAndrii Nakryiko } 191e8611bbSAndrii Nakryiko 201e8611bbSAndrii Nakryiko fclose(f); 211e8611bbSAndrii Nakryiko return -EINVAL; 221e8611bbSAndrii Nakryiko } 231e8611bbSAndrii Nakryiko 241e8611bbSAndrii Nakryiko void test_attach_probe(void) 251e8611bbSAndrii Nakryiko { 261e8611bbSAndrii Nakryiko const char *kprobe_name = "kprobe/sys_nanosleep"; 271e8611bbSAndrii Nakryiko const char *kretprobe_name = "kretprobe/sys_nanosleep"; 281e8611bbSAndrii Nakryiko const char *uprobe_name = "uprobe/trigger_func"; 291e8611bbSAndrii Nakryiko const char *uretprobe_name = "uretprobe/trigger_func"; 301e8611bbSAndrii Nakryiko const int kprobe_idx = 0, kretprobe_idx = 1; 311e8611bbSAndrii Nakryiko const int uprobe_idx = 2, uretprobe_idx = 3; 321e8611bbSAndrii Nakryiko const char *file = "./test_attach_probe.o"; 331e8611bbSAndrii Nakryiko struct bpf_program *kprobe_prog, *kretprobe_prog; 341e8611bbSAndrii Nakryiko struct bpf_program *uprobe_prog, *uretprobe_prog; 351e8611bbSAndrii Nakryiko struct bpf_object *obj; 361e8611bbSAndrii Nakryiko int err, prog_fd, duration = 0, res; 371e8611bbSAndrii Nakryiko struct bpf_link *kprobe_link = NULL; 381e8611bbSAndrii Nakryiko struct bpf_link *kretprobe_link = NULL; 391e8611bbSAndrii Nakryiko struct bpf_link *uprobe_link = NULL; 401e8611bbSAndrii Nakryiko struct bpf_link *uretprobe_link = NULL; 411e8611bbSAndrii Nakryiko int results_map_fd; 421e8611bbSAndrii Nakryiko size_t uprobe_offset; 431e8611bbSAndrii Nakryiko ssize_t base_addr; 441e8611bbSAndrii Nakryiko 451e8611bbSAndrii Nakryiko base_addr = get_base_addr(); 461e8611bbSAndrii Nakryiko if (CHECK(base_addr < 0, "get_base_addr", 471e8611bbSAndrii Nakryiko "failed to find base addr: %zd", base_addr)) 481e8611bbSAndrii Nakryiko return; 491e8611bbSAndrii Nakryiko uprobe_offset = (size_t)&get_base_addr - base_addr; 501e8611bbSAndrii Nakryiko 511e8611bbSAndrii Nakryiko /* load programs */ 521e8611bbSAndrii Nakryiko err = bpf_prog_load(file, BPF_PROG_TYPE_KPROBE, &obj, &prog_fd); 531e8611bbSAndrii Nakryiko if (CHECK(err, "obj_load", "err %d errno %d\n", err, errno)) 541e8611bbSAndrii Nakryiko return; 551e8611bbSAndrii Nakryiko 561e8611bbSAndrii Nakryiko kprobe_prog = bpf_object__find_program_by_title(obj, kprobe_name); 571e8611bbSAndrii Nakryiko if (CHECK(!kprobe_prog, "find_probe", 581e8611bbSAndrii Nakryiko "prog '%s' not found\n", kprobe_name)) 591e8611bbSAndrii Nakryiko goto cleanup; 601e8611bbSAndrii Nakryiko kretprobe_prog = bpf_object__find_program_by_title(obj, kretprobe_name); 611e8611bbSAndrii Nakryiko if (CHECK(!kretprobe_prog, "find_probe", 621e8611bbSAndrii Nakryiko "prog '%s' not found\n", kretprobe_name)) 631e8611bbSAndrii Nakryiko goto cleanup; 641e8611bbSAndrii Nakryiko uprobe_prog = bpf_object__find_program_by_title(obj, uprobe_name); 651e8611bbSAndrii Nakryiko if (CHECK(!uprobe_prog, "find_probe", 661e8611bbSAndrii Nakryiko "prog '%s' not found\n", uprobe_name)) 671e8611bbSAndrii Nakryiko goto cleanup; 681e8611bbSAndrii Nakryiko uretprobe_prog = bpf_object__find_program_by_title(obj, uretprobe_name); 691e8611bbSAndrii Nakryiko if (CHECK(!uretprobe_prog, "find_probe", 701e8611bbSAndrii Nakryiko "prog '%s' not found\n", uretprobe_name)) 711e8611bbSAndrii Nakryiko goto cleanup; 721e8611bbSAndrii Nakryiko 731e8611bbSAndrii Nakryiko /* load maps */ 741e8611bbSAndrii Nakryiko results_map_fd = bpf_find_map(__func__, obj, "results_map"); 751e8611bbSAndrii Nakryiko if (CHECK(results_map_fd < 0, "find_results_map", 761e8611bbSAndrii Nakryiko "err %d\n", results_map_fd)) 771e8611bbSAndrii Nakryiko goto cleanup; 781e8611bbSAndrii Nakryiko 791e8611bbSAndrii Nakryiko kprobe_link = bpf_program__attach_kprobe(kprobe_prog, 801e8611bbSAndrii Nakryiko false /* retprobe */, 811cb59a60SIlya Leoshkevich SYS_NANOSLEEP_KPROBE_NAME); 821e8611bbSAndrii Nakryiko if (CHECK(IS_ERR(kprobe_link), "attach_kprobe", 831e8611bbSAndrii Nakryiko "err %ld\n", PTR_ERR(kprobe_link))) { 841e8611bbSAndrii Nakryiko kprobe_link = NULL; 851e8611bbSAndrii Nakryiko goto cleanup; 861e8611bbSAndrii Nakryiko } 871e8611bbSAndrii Nakryiko kretprobe_link = bpf_program__attach_kprobe(kretprobe_prog, 881e8611bbSAndrii Nakryiko true /* retprobe */, 891cb59a60SIlya Leoshkevich SYS_NANOSLEEP_KPROBE_NAME); 901e8611bbSAndrii Nakryiko if (CHECK(IS_ERR(kretprobe_link), "attach_kretprobe", 911e8611bbSAndrii Nakryiko "err %ld\n", PTR_ERR(kretprobe_link))) { 921e8611bbSAndrii Nakryiko kretprobe_link = NULL; 931e8611bbSAndrii Nakryiko goto cleanup; 941e8611bbSAndrii Nakryiko } 951e8611bbSAndrii Nakryiko uprobe_link = bpf_program__attach_uprobe(uprobe_prog, 961e8611bbSAndrii Nakryiko false /* retprobe */, 971e8611bbSAndrii Nakryiko 0 /* self pid */, 981e8611bbSAndrii Nakryiko "/proc/self/exe", 991e8611bbSAndrii Nakryiko uprobe_offset); 1001e8611bbSAndrii Nakryiko if (CHECK(IS_ERR(uprobe_link), "attach_uprobe", 1011e8611bbSAndrii Nakryiko "err %ld\n", PTR_ERR(uprobe_link))) { 1021e8611bbSAndrii Nakryiko uprobe_link = NULL; 1031e8611bbSAndrii Nakryiko goto cleanup; 1041e8611bbSAndrii Nakryiko } 1051e8611bbSAndrii Nakryiko uretprobe_link = bpf_program__attach_uprobe(uretprobe_prog, 1061e8611bbSAndrii Nakryiko true /* retprobe */, 1071e8611bbSAndrii Nakryiko -1 /* any pid */, 1081e8611bbSAndrii Nakryiko "/proc/self/exe", 1091e8611bbSAndrii Nakryiko uprobe_offset); 1101e8611bbSAndrii Nakryiko if (CHECK(IS_ERR(uretprobe_link), "attach_uretprobe", 1111e8611bbSAndrii Nakryiko "err %ld\n", PTR_ERR(uretprobe_link))) { 1121e8611bbSAndrii Nakryiko uretprobe_link = NULL; 1131e8611bbSAndrii Nakryiko goto cleanup; 1141e8611bbSAndrii Nakryiko } 1151e8611bbSAndrii Nakryiko 1161e8611bbSAndrii Nakryiko /* trigger & validate kprobe && kretprobe */ 1171e8611bbSAndrii Nakryiko usleep(1); 1181e8611bbSAndrii Nakryiko 1191e8611bbSAndrii Nakryiko err = bpf_map_lookup_elem(results_map_fd, &kprobe_idx, &res); 1201e8611bbSAndrii Nakryiko if (CHECK(err, "get_kprobe_res", 1211e8611bbSAndrii Nakryiko "failed to get kprobe res: %d\n", err)) 1221e8611bbSAndrii Nakryiko goto cleanup; 1231e8611bbSAndrii Nakryiko if (CHECK(res != kprobe_idx + 1, "check_kprobe_res", 1241e8611bbSAndrii Nakryiko "wrong kprobe res: %d\n", res)) 1251e8611bbSAndrii Nakryiko goto cleanup; 1261e8611bbSAndrii Nakryiko 1271e8611bbSAndrii Nakryiko err = bpf_map_lookup_elem(results_map_fd, &kretprobe_idx, &res); 1281e8611bbSAndrii Nakryiko if (CHECK(err, "get_kretprobe_res", 1291e8611bbSAndrii Nakryiko "failed to get kretprobe res: %d\n", err)) 1301e8611bbSAndrii Nakryiko goto cleanup; 1311e8611bbSAndrii Nakryiko if (CHECK(res != kretprobe_idx + 1, "check_kretprobe_res", 1321e8611bbSAndrii Nakryiko "wrong kretprobe res: %d\n", res)) 1331e8611bbSAndrii Nakryiko goto cleanup; 1341e8611bbSAndrii Nakryiko 1351e8611bbSAndrii Nakryiko /* trigger & validate uprobe & uretprobe */ 1361e8611bbSAndrii Nakryiko get_base_addr(); 1371e8611bbSAndrii Nakryiko 1381e8611bbSAndrii Nakryiko err = bpf_map_lookup_elem(results_map_fd, &uprobe_idx, &res); 1391e8611bbSAndrii Nakryiko if (CHECK(err, "get_uprobe_res", 1401e8611bbSAndrii Nakryiko "failed to get uprobe res: %d\n", err)) 1411e8611bbSAndrii Nakryiko goto cleanup; 1421e8611bbSAndrii Nakryiko if (CHECK(res != uprobe_idx + 1, "check_uprobe_res", 1431e8611bbSAndrii Nakryiko "wrong uprobe res: %d\n", res)) 1441e8611bbSAndrii Nakryiko goto cleanup; 1451e8611bbSAndrii Nakryiko 1461e8611bbSAndrii Nakryiko err = bpf_map_lookup_elem(results_map_fd, &uretprobe_idx, &res); 1471e8611bbSAndrii Nakryiko if (CHECK(err, "get_uretprobe_res", 1481e8611bbSAndrii Nakryiko "failed to get uretprobe res: %d\n", err)) 1491e8611bbSAndrii Nakryiko goto cleanup; 1501e8611bbSAndrii Nakryiko if (CHECK(res != uretprobe_idx + 1, "check_uretprobe_res", 1511e8611bbSAndrii Nakryiko "wrong uretprobe res: %d\n", res)) 1521e8611bbSAndrii Nakryiko goto cleanup; 1531e8611bbSAndrii Nakryiko 1541e8611bbSAndrii Nakryiko cleanup: 1551e8611bbSAndrii Nakryiko bpf_link__destroy(kprobe_link); 1561e8611bbSAndrii Nakryiko bpf_link__destroy(kretprobe_link); 1571e8611bbSAndrii Nakryiko bpf_link__destroy(uprobe_link); 1581e8611bbSAndrii Nakryiko bpf_link__destroy(uretprobe_link); 1591e8611bbSAndrii Nakryiko bpf_object__close(obj); 1601e8611bbSAndrii Nakryiko } 161