tools/perf/tests/sdt.c

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