1 | // SPDX-License-Identifier: GPL-2.0 |
2 | #include <errno.h> |
3 | #include <limits.h> |
4 | #include <stdio.h> |
5 | #include <stdlib.h> |
6 | #include <unistd.h> |
7 | #include <sys/epoll.h> |
8 | #include <util/symbol.h> |
9 | #include <linux/filter.h> |
10 | #include "tests.h" |
11 | #include "debug.h" |
12 | #include "probe-file.h" |
13 | #include "build-id.h" |
14 | #include "util.h" |
15 | |
16 | /* To test SDT event, we need libelf support to scan elf binary */ |
17 | #if defined(HAVE_SDT_EVENT) && defined(HAVE_LIBELF_SUPPORT) |
18 | |
19 | #include <sys/sdt.h> |
20 | |
21 | static int target_function(void) |
22 | { |
23 | DTRACE_PROBE(perf, test_target); |
24 | return TEST_OK; |
25 | } |
26 | |
27 | /* Copied from builtin-buildid-cache.c */ |
28 | static int build_id_cache__add_file(const char *filename) |
29 | { |
30 | char sbuild_id[SBUILD_ID_SIZE]; |
31 | struct build_id bid; |
32 | int err; |
33 | |
34 | err = filename__read_build_id(filename, &bid); |
35 | if (err < 0) { |
36 | pr_debug("Failed to read build id of %s\n" , filename); |
37 | return err; |
38 | } |
39 | |
40 | build_id__sprintf(&bid, sbuild_id); |
41 | err = build_id_cache__add_s(sbuild_id, filename, NULL, false, false); |
42 | if (err < 0) |
43 | pr_debug("Failed to add build id cache of %s\n" , filename); |
44 | return err; |
45 | } |
46 | |
47 | static char *get_self_path(void) |
48 | { |
49 | char *buf = calloc(PATH_MAX, sizeof(char)); |
50 | |
51 | if (buf && readlink("/proc/self/exe" , buf, PATH_MAX - 1) < 0) { |
52 | pr_debug("Failed to get correct path of perf\n" ); |
53 | free(buf); |
54 | return NULL; |
55 | } |
56 | return buf; |
57 | } |
58 | |
59 | static int search_cached_probe(const char *target, |
60 | const char *group, const char *event) |
61 | { |
62 | struct probe_cache *cache = probe_cache__new(target, NULL); |
63 | int ret = 0; |
64 | |
65 | if (!cache) { |
66 | pr_debug("Failed to open probe cache of %s\n" , target); |
67 | return -EINVAL; |
68 | } |
69 | |
70 | if (!probe_cache__find_by_name(cache, group, event)) { |
71 | pr_debug("Failed to find %s:%s in the cache\n" , group, event); |
72 | ret = -ENOENT; |
73 | } |
74 | probe_cache__delete(cache); |
75 | |
76 | return ret; |
77 | } |
78 | |
79 | static int test__sdt_event(struct test_suite *test __maybe_unused, int subtests __maybe_unused) |
80 | { |
81 | int ret = TEST_FAIL; |
82 | char __tempdir[] = "./test-buildid-XXXXXX" ; |
83 | char *tempdir = NULL, *myself = get_self_path(); |
84 | |
85 | if (myself == NULL || mkdtemp(__tempdir) == NULL) { |
86 | pr_debug("Failed to make a tempdir for build-id cache\n" ); |
87 | goto error; |
88 | } |
89 | /* Note that buildid_dir must be an absolute path */ |
90 | tempdir = realpath(__tempdir, NULL); |
91 | if (tempdir == NULL) |
92 | goto error_rmdir; |
93 | |
94 | /* At first, scan itself */ |
95 | set_buildid_dir(tempdir); |
96 | if (build_id_cache__add_file(myself) < 0) |
97 | goto error_rmdir; |
98 | |
99 | /* Open a cache and make sure the SDT is stored */ |
100 | if (search_cached_probe(myself, "sdt_perf" , "test_target" ) < 0) |
101 | goto error_rmdir; |
102 | |
103 | /* TBD: probing on the SDT event and collect logs */ |
104 | |
105 | /* Call the target and get an event */ |
106 | ret = target_function(); |
107 | |
108 | error_rmdir: |
109 | /* Cleanup temporary buildid dir */ |
110 | rm_rf(__tempdir); |
111 | error: |
112 | free(tempdir); |
113 | free(myself); |
114 | return ret; |
115 | } |
116 | #else |
117 | static int test__sdt_event(struct test_suite *test __maybe_unused, int subtests __maybe_unused) |
118 | { |
119 | pr_debug("Skip SDT event test because SDT support is not compiled\n" ); |
120 | return TEST_SKIP; |
121 | } |
122 | #endif |
123 | |
124 | DEFINE_SUITE("Probe SDT events" , sdt_event); |
125 | |