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
23static 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
38	scnprintf(filename, PATH_MAX, "%s/pe-file.exe", d);
39	ret = filename__read_build_id(filename, &bid);
40	TEST_ASSERT_VAL("Failed to read build_id",
41			ret == sizeof(expect_build_id));
42	TEST_ASSERT_VAL("Wrong build_id", !memcmp(bid.data, expect_build_id,
43						  sizeof(expect_build_id)));
44
45	ret = filename__read_debuglink(filename, debuglink, PATH_MAX);
46	TEST_ASSERT_VAL("Failed to read debuglink", ret == 0);
47	TEST_ASSERT_VAL("Wrong debuglink",
48			!strcmp(debuglink, expect_debuglink));
49
50	scnprintf(debugfile, PATH_MAX, "%s/%s", d, debuglink);
51	ret = filename__read_build_id(debugfile, &bid);
52	TEST_ASSERT_VAL("Failed to read debug file build_id",
53			ret == sizeof(expect_build_id));
54	TEST_ASSERT_VAL("Wrong build_id", !memcmp(bid.data, expect_build_id,
55						  sizeof(expect_build_id)));
56
57	dso = dso__new(filename);
58	TEST_ASSERT_VAL("Failed to get dso", dso);
59
60	ret = dso__load_bfd_symbols(dso, debugfile);
61	TEST_ASSERT_VAL("Failed to load symbols", ret == 0);
62
63	dso__sort_by_name(dso);
64	sym = dso__find_symbol_by_name(dso, "main");
65	TEST_ASSERT_VAL("Failed to find main", sym);
66	dso__delete(dso);
67
68	return TEST_OK;
69}
70
71int test__pe_file_parsing(struct test *test __maybe_unused,
72			  int subtest __maybe_unused)
73{
74	struct stat st;
75	char path_dir[PATH_MAX];
76
77	/* First try development tree tests. */
78	if (!lstat("./tests", &st))
79		return run_dir("./tests");
80
81	/* Then installed path. */
82	snprintf(path_dir, PATH_MAX, "%s/tests", get_argv_exec_path());
83
84	if (!lstat(path_dir, &st))
85		return run_dir(path_dir);
86
87	return TEST_SKIP;
88}
89
90#else
91
92int test__pe_file_parsing(struct test *test __maybe_unused,
93			  int subtest __maybe_unused)
94{
95	return TEST_SKIP;
96}
97
98#endif