tests/unit/ust-elf/ust-elf.c

   1 /*
   2  * SPDX-License-Identifier: LGPL-2.1-or-later
   3  *
   4  * Copyright (C) 2015 Antoine Busque <abusque@efficios.com>
   5  */
   6
   7 #include <inttypes.h>
   8 #include <limits.h>
   9 #include <stdint.h>
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13
  14 #include "common/elf.h"
  15 #include "tap.h"
  16
  17 #define NUM_ARCH 4
  18 #define NUM_TESTS_PER_ARCH 11
  19 #define NUM_TESTS_PIC 3
  20 #define NUM_TESTS (NUM_ARCH * NUM_TESTS_PER_ARCH) + NUM_TESTS_PIC + 1
  21
  22 /*
  23  * Expected memsz were computed using libelf, build ID and debug link
  24  * were determined through readelf.
  25  */
  26 #define X86_MEMSZ 5732
  27 #define X86_64_MEMSZ 2099376
  28 #define ARMEB_MEMSZ 34196
  29 #define AARCH64_BE_MEMSZ 67632
  30
  31 #define X86_CRC 0x1531f73c
  32 #define X86_64_CRC 0xa048a98f
  33 #define ARMEB_CRC 0x9d40261b
  34 #define AARCH64_BE_CRC 0x2b8cedce
  35
  36 #define BUILD_ID_LEN 20
  37 #define DBG_FILE "main.elf.debug"
  38
  39 static const uint8_t x86_build_id[BUILD_ID_LEN] = {
  40         0x27, 0x79, 0x2a, 0xe7, 0xaa, 0xef, 0x72, 0x5c, 0x9c, 0x52,
  41         0x80, 0xec, 0x1e, 0x18, 0xd8, 0x09, 0x02, 0xba, 0xbc, 0x82
  42 };
  43 static const uint8_t x86_64_build_id[BUILD_ID_LEN] = {
  44         0x0f, 0x87, 0xb2, 0xe2, 0x24, 0x9c, 0xe1, 0xc2, 0x24, 0xb1,
  45         0xf8, 0xb6, 0x65, 0x83, 0xa3, 0xc1, 0xcb, 0x30, 0x5c, 0x63
  46 };
  47 static const uint8_t armeb_build_id[BUILD_ID_LEN] = {
  48         0x60, 0x5d, 0x26, 0xa0, 0x0e, 0x30, 0xa4, 0x29, 0xf4, 0xf1,
  49         0x85, 0x53, 0xda, 0x90, 0x68, 0xe1, 0xf5, 0x67, 0xbe, 0x42
  50 };
  51 static const uint8_t aarch64_be_build_id[BUILD_ID_LEN] = {
  52         0xb9, 0x0a, 0xa0, 0xed, 0xd1, 0x41, 0x42, 0xc3, 0x34, 0x85,
  53         0xfa, 0x27, 0x2e, 0xa9, 0x2f, 0xd2, 0xe4, 0xf7, 0xb6, 0x60
  54 };
  55
  56 static
  57 void test_elf(const char *test_dir, const char *arch, uint64_t exp_memsz,
  58                 const uint8_t *exp_build_id, uint32_t exp_crc)
  59 {
  60         char path[PATH_MAX];
  61         struct lttng_ust_elf *elf = NULL;
  62         int ret = 0;
  63         uint64_t memsz = 0;
  64         int has_build_id = 0;
  65         uint8_t *build_id = NULL;
  66         size_t build_id_len = 0;
  67         int has_debug_link = 0;
  68         char *dbg_file = NULL;
  69         uint32_t crc = 0;
  70
  71         diag("Testing %s support", arch);
  72
  73         snprintf(path, PATH_MAX, "%s/data/%s/main.elf", test_dir, arch);
  74         elf = lttng_ust_elf_create(path);
  75         ok(elf != NULL, "lttng_ust_elf_create");
  76
  77         ret = lttng_ust_elf_get_memsz(elf, &memsz);
  78         ok(ret == 0, "lttng_ust_elf_get_memsz returned successfully");
  79         ok(memsz == exp_memsz,
  80                 "memsz - expected: %" PRIu64 ", got: %" PRIu64,
  81                 exp_memsz, memsz);
  82
  83         ret = lttng_ust_elf_get_build_id(elf, &build_id, &build_id_len,
  84                                         &has_build_id);
  85         ok(ret == 0, "lttng_ust_elf_get_build_id returned successfully");
  86         ok(has_build_id == 1, "build id marked as found");
  87         ok(build_id_len == BUILD_ID_LEN,
  88                 "build_id_len - expected: %u, got: %zu",
  89                 BUILD_ID_LEN, build_id_len);
  90         ok(memcmp(build_id, exp_build_id, build_id_len) == 0,
  91                 "build_id has expected value");
  92
  93         ret = lttng_ust_elf_get_debug_link(elf, &dbg_file, &crc,
  94                                         &has_debug_link);
  95         ok(ret == 0, "lttng_ust_elf_get_debug_link returned successfully");
  96         ok(has_debug_link == 1, "debug link marked as found");
  97         ok(dbg_file && strcmp(dbg_file, DBG_FILE) == 0,
  98                 "debug link filename - expected: %s, got: %s",
  99                 DBG_FILE, dbg_file);
 100         ok(crc == exp_crc,
 101                 "debug link crc - expected: %#x, got: %#x",
 102                 exp_crc, crc);
 103
 104         free(build_id);
 105         free(dbg_file);
 106         lttng_ust_elf_destroy(elf);
 107 }
 108
 109 static
 110 void test_pic(const char *test_dir)
 111 {
 112         char exec_path[PATH_MAX];
 113         char pie_path[PATH_MAX];
 114         char pic_path[PATH_MAX];
 115         struct lttng_ust_elf *elf = NULL;
 116         uint8_t is_pic;
 117
 118         snprintf(exec_path, PATH_MAX, "%s/data/pic/hello.exec", test_dir);
 119         snprintf(pie_path, PATH_MAX, "%s/data/pic/hello.pie", test_dir);
 120         snprintf(pic_path, PATH_MAX, "%s/data/pic/hello.pic", test_dir);
 121
 122         elf = lttng_ust_elf_create(exec_path);
 123         is_pic = lttng_ust_elf_is_pic(elf);
 124         ok(is_pic == 0, "hello.exec is not PIC");
 125         lttng_ust_elf_destroy(elf);
 126
 127         elf = lttng_ust_elf_create(pie_path);
 128         is_pic = lttng_ust_elf_is_pic(elf);
 129         ok(is_pic == 1, "hello.pie is PIC");
 130         lttng_ust_elf_destroy(elf);
 131
 132         elf = lttng_ust_elf_create(pic_path);
 133         is_pic = lttng_ust_elf_is_pic(elf);
 134         ok(is_pic == 1, "hello.pic is PIC");
 135         lttng_ust_elf_destroy(elf);
 136 }
 137
 138 int main(int argc, char **argv)
 139 {
 140         const char *test_dir;
 141
 142         plan_tests(NUM_TESTS);
 143
 144         ok(argc == 2, "Invoke as: %s <path>", argv[0]);
 145         if (argc != 2) {
 146                 return EXIT_FAILURE;
 147         } else {
 148                 test_dir = argv[1];
 149         }
 150
 151         test_elf(test_dir, "x86", X86_MEMSZ, x86_build_id, X86_CRC);
 152         test_elf(test_dir, "x86_64", X86_64_MEMSZ, x86_64_build_id, X86_64_CRC);
 153         test_elf(test_dir, "armeb", ARMEB_MEMSZ, armeb_build_id, ARMEB_CRC);
 154         test_elf(test_dir, "aarch64_be", AARCH64_BE_MEMSZ, aarch64_be_build_id,
 155                 AARCH64_BE_CRC);
 156         test_pic(test_dir);
 157
 158         return exit_status();
 159 }