tools/perf/tests/pe-file-parsing.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 #include <stdbool.h>
   3 #include <inttypes.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <linux/bitops.h>
   7 #include <linux/kernel.h>
   8 #include <linux/types.h>
   9 #include <sys/types.h>
  10 #include <sys/stat.h>
  11 #include <unistd.h>
  12 #include <subcmd/exec-cmd.h>
  13
  14 #include "debug.h"
  15 #include "util/build-id.h"
  16 #include "util/symbol.h"
  17 #include "util/dso.h"
  18
  19 #include "tests.h"
  20
  21 #ifdef HAVE_LIBBFD_SUPPORT
  22
  23 static int run_dir(const char *d)
  24 {
  25         char filename[PATH_MAX];
  26         char debugfile[PATH_MAX];
  27         struct build_id bid;
  28         char debuglink[PATH_MAX];
  29         char expect_build_id[] = {
  30                 0x5a, 0x0f, 0xd8, 0x82, 0xb5, 0x30, 0x84, 0x22,
  31                 0x4b, 0xa4, 0x7b, 0x62, 0x4c, 0x55, 0xa4, 0x69,
  32         };
  33         char expect_debuglink[PATH_MAX] = "pe-file.exe.debug";
  34         struct dso *dso;
  35         struct symbol *sym;
  36         int ret;
  37         size_t idx;
  38
  39         scnprintf(filename, PATH_MAX, "%s/pe-file.exe", d);
  40         ret = filename__read_build_id(filename, &bid);
  41         TEST_ASSERT_VAL("Failed to read build_id",
  42                         ret == sizeof(expect_build_id));
  43         TEST_ASSERT_VAL("Wrong build_id", !memcmp(bid.data, expect_build_id,
  44                                                   sizeof(expect_build_id)));
  45
  46         ret = filename__read_debuglink(filename, debuglink, PATH_MAX);
  47         TEST_ASSERT_VAL("Failed to read debuglink", ret == 0);
  48         TEST_ASSERT_VAL("Wrong debuglink",
  49                         !strcmp(debuglink, expect_debuglink));
  50
  51         scnprintf(debugfile, PATH_MAX, "%s/%s", d, debuglink);
  52         ret = filename__read_build_id(debugfile, &bid);
  53         TEST_ASSERT_VAL("Failed to read debug file build_id",
  54                         ret == sizeof(expect_build_id));
  55         TEST_ASSERT_VAL("Wrong build_id", !memcmp(bid.data, expect_build_id,
  56                                                   sizeof(expect_build_id)));
  57
  58         dso = dso__new(filename);
  59         TEST_ASSERT_VAL("Failed to get dso", dso);
  60
  61         ret = dso__load_bfd_symbols(dso, debugfile);
  62         TEST_ASSERT_VAL("Failed to load symbols", ret == 0);
  63
  64         dso__sort_by_name(dso);
  65         sym = dso__find_symbol_by_name(dso, "main", &idx);
  66         TEST_ASSERT_VAL("Failed to find main", sym);
  67         dso__delete(dso);
  68
  69         return TEST_OK;
  70 }
  71
  72 static int test__pe_file_parsing(struct test_suite *test __maybe_unused,
  73                           int subtest __maybe_unused)
  74 {
  75         struct stat st;
  76         char path_dir[PATH_MAX];
  77
  78         /* First try development tree tests. */
  79         if (!lstat("./tests", &st))
  80                 return run_dir("./tests");
  81
  82         /* Then installed path. */
  83         snprintf(path_dir, PATH_MAX, "%s/tests", get_argv_exec_path());
  84
  85         if (!lstat(path_dir, &st))
  86                 return run_dir(path_dir);
  87
  88         return TEST_SKIP;
  89 }
  90
  91 #else
  92
  93 static int test__pe_file_parsing(struct test_suite *test __maybe_unused,
  94                           int subtest __maybe_unused)
  95 {
  96         return TEST_SKIP;
  97 }
  98
  99 #endif
 100
 101 DEFINE_SUITE("PE file support", pe_file_parsing);