[lttng-ust.git] / tests / unit / ust-elf / ust-elf.c

/*
 * SPDX-License-Identifier: LGPL-2.1-or-later
 *
 * Copyright (C) 2015 Antoine Busque <abusque@efficios.com>
 */

#include <inttypes.h>
#include <limits.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "common/elf.h"
#include "tap.h"

#define NUM_ARCH 4
#define NUM_TESTS_PER_ARCH 11
#define NUM_TESTS_PIC 3
#define NUM_TESTS (NUM_ARCH * NUM_TESTS_PER_ARCH) + NUM_TESTS_PIC + 1

/*
 * Expected memsz were computed using libelf, build ID and debug link
 * were determined through readelf.
 */
#define X86_MEMSZ 5732
#define X86_64_MEMSZ 2099376
#define ARMEB_MEMSZ 34196
#define AARCH64_BE_MEMSZ 67632

#define X86_CRC 0x1531f73c
#define X86_64_CRC 0xa048a98f
#define ARMEB_CRC 0x9d40261b
#define AARCH64_BE_CRC 0x2b8cedce

#define BUILD_ID_LEN 20
#define DBG_FILE "main.elf.debug"

static const uint8_t x86_build_id[BUILD_ID_LEN] = {
	0x27, 0x79, 0x2a, 0xe7, 0xaa, 0xef, 0x72, 0x5c, 0x9c, 0x52,
	0x80, 0xec, 0x1e, 0x18, 0xd8, 0x09, 0x02, 0xba, 0xbc, 0x82
};
static const uint8_t x86_64_build_id[BUILD_ID_LEN] = {
	0x0f, 0x87, 0xb2, 0xe2, 0x24, 0x9c, 0xe1, 0xc2, 0x24, 0xb1,
	0xf8, 0xb6, 0x65, 0x83, 0xa3, 0xc1, 0xcb, 0x30, 0x5c, 0x63
};
static const uint8_t armeb_build_id[BUILD_ID_LEN] = {
	0x60, 0x5d, 0x26, 0xa0, 0x0e, 0x30, 0xa4, 0x29, 0xf4, 0xf1,
	0x85, 0x53, 0xda, 0x90, 0x68, 0xe1, 0xf5, 0x67, 0xbe, 0x42
};
static const uint8_t aarch64_be_build_id[BUILD_ID_LEN] = {
	0xb9, 0x0a, 0xa0, 0xed, 0xd1, 0x41, 0x42, 0xc3, 0x34, 0x85,
	0xfa, 0x27, 0x2e, 0xa9, 0x2f, 0xd2, 0xe4, 0xf7, 0xb6, 0x60
};

static
void test_elf(const char *test_dir, const char *arch, uint64_t exp_memsz,
		const uint8_t *exp_build_id, uint32_t exp_crc)
{
	char path[PATH_MAX];
	struct lttng_ust_elf *elf = NULL;
	int ret = 0;
	uint64_t memsz = 0;
	int has_build_id = 0;
	uint8_t *build_id = NULL;
	size_t build_id_len = 0;
	int has_debug_link = 0;
	char *dbg_file = NULL;
	uint32_t crc = 0;

	diag("Testing %s support", arch);

	snprintf(path, PATH_MAX, "%s/data/%s/main.elf", test_dir, arch);
	elf = lttng_ust_elf_create(path);
	ok(elf != NULL, "lttng_ust_elf_create");

	ret = lttng_ust_elf_get_memsz(elf, &memsz);
	ok(ret == 0, "lttng_ust_elf_get_memsz returned successfully");
	ok(memsz == exp_memsz,
		"memsz - expected: %" PRIu64 ", got: %" PRIu64,
		exp_memsz, memsz);

	ret = lttng_ust_elf_get_build_id(elf, &build_id, &build_id_len,
					&has_build_id);
	ok(ret == 0, "lttng_ust_elf_get_build_id returned successfully");
	ok(has_build_id == 1, "build id marked as found");
	ok(build_id_len == BUILD_ID_LEN,
		"build_id_len - expected: %u, got: %zu",
		BUILD_ID_LEN, build_id_len);
	ok(memcmp(build_id, exp_build_id, build_id_len) == 0,
		"build_id has expected value");

	ret = lttng_ust_elf_get_debug_link(elf, &dbg_file, &crc,
					&has_debug_link);
	ok(ret == 0, "lttng_ust_elf_get_debug_link returned successfully");
	ok(has_debug_link == 1, "debug link marked as found");
	ok(dbg_file && strcmp(dbg_file, DBG_FILE) == 0,
		"debug link filename - expected: %s, got: %s",
		DBG_FILE, dbg_file);
	ok(crc == exp_crc,
		"debug link crc - expected: %#x, got: %#x",
		exp_crc, crc);

	free(build_id);
	free(dbg_file);
	lttng_ust_elf_destroy(elf);
}

static
void test_pic(const char *test_dir)
{
	char exec_path[PATH_MAX];
	char pie_path[PATH_MAX];
	char pic_path[PATH_MAX];
	struct lttng_ust_elf *elf = NULL;
	uint8_t is_pic;

	snprintf(exec_path, PATH_MAX, "%s/data/pic/hello.exec", test_dir);
	snprintf(pie_path, PATH_MAX, "%s/data/pic/hello.pie", test_dir);
	snprintf(pic_path, PATH_MAX, "%s/data/pic/hello.pic", test_dir);

	elf = lttng_ust_elf_create(exec_path);
	is_pic = lttng_ust_elf_is_pic(elf);
	ok(is_pic == 0, "hello.exec is not PIC");
	lttng_ust_elf_destroy(elf);

	elf = lttng_ust_elf_create(pie_path);
	is_pic = lttng_ust_elf_is_pic(elf);
	ok(is_pic == 1, "hello.pie is PIC");
	lttng_ust_elf_destroy(elf);

	elf = lttng_ust_elf_create(pic_path);
	is_pic = lttng_ust_elf_is_pic(elf);
	ok(is_pic == 1, "hello.pic is PIC");
	lttng_ust_elf_destroy(elf);
}

int main(int argc, char **argv)
{
	const char *test_dir;

	plan_tests(NUM_TESTS);

	ok(argc == 2, "Invoke as: %s <path>", argv[0]);
	if (argc != 2) {
		return EXIT_FAILURE;
	} else {
		test_dir = argv[1];
	}

	test_elf(test_dir, "x86", X86_MEMSZ, x86_build_id, X86_CRC);
	test_elf(test_dir, "x86_64", X86_64_MEMSZ, x86_64_build_id, X86_64_CRC);
	test_elf(test_dir, "armeb", ARMEB_MEMSZ, armeb_build_id, ARMEB_CRC);
	test_elf(test_dir, "aarch64_be", AARCH64_BE_MEMSZ, aarch64_be_build_id,
		AARCH64_BE_CRC);
	test_pic(test_dir);

	return exit_status();
}
Commit	Line	Data
22609c7a	1	/*
c0c0989a	2	* SPDX-License-Identifier: LGPL-2.1-or-later
22609c7a	3	*
c0c0989a	4	* Copyright (C) 2015 Antoine Busque <abusque@efficios.com>
22609c7a AB	5	*/
22609c7a AB	6
3c99db22	7	#include <inttypes.h>
98c2069e	8	#include <limits.h>
fb31eb73	9	#include <stdint.h>
22609c7a AB	10	#include <stdio.h>
	11	#include <stdlib.h>
	12	#include <string.h>
	13
9d315d6d	14	#include "common/elf.h"
22609c7a AB	15	#include "tap.h"
	16
	17	#define NUM_ARCH 4
	18	#define NUM_TESTS_PER_ARCH 11
968f523d AB	19	#define NUM_TESTS_PIC 3
968f523d AB	20	#define NUM_TESTS (NUM_ARCH * NUM_TESTS_PER_ARCH) + NUM_TESTS_PIC + 1
22609c7a AB	21
	22	/*
	23	* Expected memsz were computed using libelf, build ID and debug link
	24	* were determined through readelf.
	25	*/
038eca0f AB	26	#define X86_MEMSZ 5732
	27	#define X86_64_MEMSZ 2099376
	28	#define ARMEB_MEMSZ 34196
	29	#define AARCH64_BE_MEMSZ 67632
22609c7a AB	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
8d8a8c5c MD	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
22609c7a AB	56	static
22609c7a AB	57	void test_elf(const char test_dir, const char arch, uint64_t exp_memsz,
8d8a8c5c	58	const uint8_t *exp_build_id, uint32_t exp_crc)
22609c7a AB	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,
3c99db22	80	"memsz - expected: %" PRIu64 ", got: %" PRIu64,
22609c7a AB	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,
3c99db22	88	"build_id_len - expected: %u, got: %zu",
22609c7a AB	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");
bb4f2490	97	ok(dbg_file && strcmp(dbg_file, DBG_FILE) == 0,
22609c7a AB	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
968f523d AB	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
22609c7a AB	138	int main(int argc, char **argv)
22609c7a AB	139	{
62eb8bca	140	const char *test_dir;
22609c7a AB	141
	142	plan_tests(NUM_TESTS);
	143
62eb8bca MD	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
8d8a8c5c MD	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,
22609c7a	155	AARCH64_BE_CRC);
968f523d	156	test_pic(test_dir);
22609c7a	157
bf746e7d	158	return exit_status();
22609c7a	159	}