Commit | Line | Data |
---|---|---|
22609c7a AB |
1 | /* |
2 | * Copyright (C) 2015 Antoine Busque <abusque@efficios.com> | |
3 | * | |
4 | * This library is free software; you can redistribute it and/or | |
5 | * modify it under the terms of the GNU Lesser General Public | |
6 | * License as published by the Free Software Foundation; either | |
7 | * version 2.1 of the License, or (at your option) any later version. | |
8 | * | |
9 | * This library is distributed in the hope that it will be useful, | |
10 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
12 | * Lesser General Public License for more details. | |
13 | * | |
14 | * You should have received a copy of the GNU Lesser General Public | |
15 | * License along with this library; if not, write to the Free Software | |
16 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
17 | */ | |
18 | ||
98c2069e | 19 | #include <limits.h> |
22609c7a AB |
20 | #include <stdio.h> |
21 | #include <stdlib.h> | |
22 | #include <string.h> | |
23 | ||
24 | #include <lttng/ust-elf.h> | |
25 | #include "tap.h" | |
26 | ||
27 | #define NUM_ARCH 4 | |
28 | #define NUM_TESTS_PER_ARCH 11 | |
968f523d AB |
29 | #define NUM_TESTS_PIC 3 |
30 | #define NUM_TESTS (NUM_ARCH * NUM_TESTS_PER_ARCH) + NUM_TESTS_PIC + 1 | |
22609c7a AB |
31 | |
32 | /* | |
33 | * Expected memsz were computed using libelf, build ID and debug link | |
34 | * were determined through readelf. | |
35 | */ | |
038eca0f AB |
36 | #define X86_MEMSZ 5732 |
37 | #define X86_64_MEMSZ 2099376 | |
38 | #define ARMEB_MEMSZ 34196 | |
39 | #define AARCH64_BE_MEMSZ 67632 | |
22609c7a AB |
40 | |
41 | #define X86_CRC 0x1531f73c | |
42 | #define X86_64_CRC 0xa048a98f | |
43 | #define ARMEB_CRC 0x9d40261b | |
44 | #define AARCH64_BE_CRC 0x2b8cedce | |
45 | ||
46 | #define BUILD_ID_LEN 20 | |
47 | #define DBG_FILE "main.elf.debug" | |
48 | ||
8d8a8c5c MD |
49 | static const uint8_t x86_build_id[BUILD_ID_LEN] = { |
50 | 0x27, 0x79, 0x2a, 0xe7, 0xaa, 0xef, 0x72, 0x5c, 0x9c, 0x52, | |
51 | 0x80, 0xec, 0x1e, 0x18, 0xd8, 0x09, 0x02, 0xba, 0xbc, 0x82 | |
52 | }; | |
53 | static const uint8_t x86_64_build_id[BUILD_ID_LEN] = { | |
54 | 0x0f, 0x87, 0xb2, 0xe2, 0x24, 0x9c, 0xe1, 0xc2, 0x24, 0xb1, | |
55 | 0xf8, 0xb6, 0x65, 0x83, 0xa3, 0xc1, 0xcb, 0x30, 0x5c, 0x63 | |
56 | }; | |
57 | static const uint8_t armeb_build_id[BUILD_ID_LEN] = { | |
58 | 0x60, 0x5d, 0x26, 0xa0, 0x0e, 0x30, 0xa4, 0x29, 0xf4, 0xf1, | |
59 | 0x85, 0x53, 0xda, 0x90, 0x68, 0xe1, 0xf5, 0x67, 0xbe, 0x42 | |
60 | }; | |
61 | static const uint8_t aarch64_be_build_id[BUILD_ID_LEN] = { | |
62 | 0xb9, 0x0a, 0xa0, 0xed, 0xd1, 0x41, 0x42, 0xc3, 0x34, 0x85, | |
63 | 0xfa, 0x27, 0x2e, 0xa9, 0x2f, 0xd2, 0xe4, 0xf7, 0xb6, 0x60 | |
64 | }; | |
65 | ||
22609c7a AB |
66 | static |
67 | void test_elf(const char *test_dir, const char *arch, uint64_t exp_memsz, | |
8d8a8c5c | 68 | const uint8_t *exp_build_id, uint32_t exp_crc) |
22609c7a AB |
69 | { |
70 | char path[PATH_MAX]; | |
71 | struct lttng_ust_elf *elf = NULL; | |
72 | int ret = 0; | |
73 | uint64_t memsz = 0; | |
74 | int has_build_id = 0; | |
75 | uint8_t *build_id = NULL; | |
76 | size_t build_id_len = 0; | |
77 | int has_debug_link = 0; | |
78 | char *dbg_file = NULL; | |
79 | uint32_t crc = 0; | |
80 | ||
81 | diag("Testing %s support", arch); | |
82 | ||
83 | snprintf(path, PATH_MAX, "%s/data/%s/main.elf", test_dir, arch); | |
84 | elf = lttng_ust_elf_create(path); | |
85 | ok(elf != NULL, "lttng_ust_elf_create"); | |
86 | ||
87 | ret = lttng_ust_elf_get_memsz(elf, &memsz); | |
88 | ok(ret == 0, "lttng_ust_elf_get_memsz returned successfully"); | |
89 | ok(memsz == exp_memsz, | |
90 | "memsz - expected: %lu, got: %lu", | |
91 | exp_memsz, memsz); | |
92 | ||
93 | ret = lttng_ust_elf_get_build_id(elf, &build_id, &build_id_len, | |
94 | &has_build_id); | |
95 | ok(ret == 0, "lttng_ust_elf_get_build_id returned successfully"); | |
96 | ok(has_build_id == 1, "build id marked as found"); | |
97 | ok(build_id_len == BUILD_ID_LEN, | |
98 | "build_id_len - expected: %u, got: %u", | |
99 | BUILD_ID_LEN, build_id_len); | |
100 | ok(memcmp(build_id, exp_build_id, build_id_len) == 0, | |
101 | "build_id has expected value"); | |
102 | ||
103 | ret = lttng_ust_elf_get_debug_link(elf, &dbg_file, &crc, | |
104 | &has_debug_link); | |
105 | ok(ret == 0, "lttng_ust_elf_get_debug_link returned successfully"); | |
106 | ok(has_debug_link == 1, "debug link marked as found"); | |
bb4f2490 | 107 | ok(dbg_file && strcmp(dbg_file, DBG_FILE) == 0, |
22609c7a AB |
108 | "debug link filename - expected: %s, got: %s", |
109 | DBG_FILE, dbg_file); | |
110 | ok(crc == exp_crc, | |
111 | "debug link crc - expected: %#x, got: %#x", | |
112 | exp_crc, crc); | |
113 | ||
114 | free(build_id); | |
115 | free(dbg_file); | |
116 | lttng_ust_elf_destroy(elf); | |
117 | } | |
118 | ||
968f523d AB |
119 | static |
120 | void test_pic(const char *test_dir) | |
121 | { | |
122 | char exec_path[PATH_MAX]; | |
123 | char pie_path[PATH_MAX]; | |
124 | char pic_path[PATH_MAX]; | |
125 | struct lttng_ust_elf *elf = NULL; | |
126 | uint8_t is_pic; | |
127 | ||
128 | snprintf(exec_path, PATH_MAX, "%s/data/pic/hello.exec", test_dir); | |
129 | snprintf(pie_path, PATH_MAX, "%s/data/pic/hello.pie", test_dir); | |
130 | snprintf(pic_path, PATH_MAX, "%s/data/pic/hello.pic", test_dir); | |
131 | ||
132 | elf = lttng_ust_elf_create(exec_path); | |
133 | is_pic = lttng_ust_elf_is_pic(elf); | |
134 | ok(is_pic == 0, "hello.exec is not PIC"); | |
135 | lttng_ust_elf_destroy(elf); | |
136 | ||
137 | elf = lttng_ust_elf_create(pie_path); | |
138 | is_pic = lttng_ust_elf_is_pic(elf); | |
139 | ok(is_pic == 1, "hello.pie is PIC"); | |
140 | lttng_ust_elf_destroy(elf); | |
141 | ||
142 | elf = lttng_ust_elf_create(pic_path); | |
143 | is_pic = lttng_ust_elf_is_pic(elf); | |
144 | ok(is_pic == 1, "hello.pic is PIC"); | |
145 | lttng_ust_elf_destroy(elf); | |
146 | } | |
147 | ||
22609c7a AB |
148 | int main(int argc, char **argv) |
149 | { | |
62eb8bca | 150 | const char *test_dir; |
22609c7a AB |
151 | |
152 | plan_tests(NUM_TESTS); | |
153 | ||
62eb8bca MD |
154 | ok(argc == 2, "Invoke as: %s <path>", argv[0]); |
155 | if (argc != 2) { | |
156 | return EXIT_FAILURE; | |
157 | } else { | |
158 | test_dir = argv[1]; | |
159 | } | |
160 | ||
8d8a8c5c MD |
161 | test_elf(test_dir, "x86", X86_MEMSZ, x86_build_id, X86_CRC); |
162 | test_elf(test_dir, "x86_64", X86_64_MEMSZ, x86_64_build_id, X86_64_CRC); | |
163 | test_elf(test_dir, "armeb", ARMEB_MEMSZ, armeb_build_id, ARMEB_CRC); | |
164 | test_elf(test_dir, "aarch64_be", AARCH64_BE_MEMSZ, aarch64_be_build_id, | |
22609c7a | 165 | AARCH64_BE_CRC); |
968f523d | 166 | test_pic(test_dir); |
22609c7a | 167 | |
62eb8bca | 168 | return EXIT_SUCCESS; |
22609c7a | 169 | } |