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

/*
 * Copyright (C) 2015  Antoine Busque <abusque@efficios.com>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
 */

#include <linux/limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <lttng/ust-elf.h>
#include "tap.h"

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

/*
 * Expected memsz were computed using libelf, build ID and debug link
 * were determined through readelf.
 */
#define X86_MEMSZ 8192
#define X86_64_MEMSZ 4194304
#define ARMEB_MEMSZ 65536
#define AARCH64_BE_MEMSZ 131072

#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
void test_elf(const char *test_dir, const char *arch, uint64_t exp_memsz,
		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: %lu, got: %lu",
		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: %u",
		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(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);
}

int main(int argc, char **argv)
{
	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
	};
	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
	};
	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
	};
	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
	};
	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);

	return EXIT_SUCCESS;
}
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
	19	#include <linux/limits.h>
	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
62eb8bca	29	#define NUM_TESTS (NUM_ARCH * NUM_TESTS_PER_ARCH) + 1
22609c7a AB	30
	31	/*
	32	* Expected memsz were computed using libelf, build ID and debug link
	33	* were determined through readelf.
	34	*/
	35	#define X86_MEMSZ 8192
	36	#define X86_64_MEMSZ 4194304
	37	#define ARMEB_MEMSZ 65536
	38	#define AARCH64_BE_MEMSZ 131072
	39
	40	#define X86_CRC 0x1531f73c
	41	#define X86_64_CRC 0xa048a98f
	42	#define ARMEB_CRC 0x9d40261b
	43	#define AARCH64_BE_CRC 0x2b8cedce
	44
	45	#define BUILD_ID_LEN 20
	46	#define DBG_FILE "main.elf.debug"
	47
	48	static
	49	void test_elf(const char test_dir, const char arch, uint64_t exp_memsz,
	50	uint8_t *exp_build_id, uint32_t exp_crc)
	51	{
	52	char path[PATH_MAX];
	53	struct lttng_ust_elf *elf = NULL;
	54	int ret = 0;
	55	uint64_t memsz = 0;
	56	int has_build_id = 0;
	57	uint8_t *build_id = NULL;
	58	size_t build_id_len = 0;
	59	int has_debug_link = 0;
	60	char *dbg_file = NULL;
	61	uint32_t crc = 0;
	62
	63	diag("Testing %s support", arch);
	64
	65	snprintf(path, PATH_MAX, "%s/data/%s/main.elf", test_dir, arch);
	66	elf = lttng_ust_elf_create(path);
	67	ok(elf != NULL, "lttng_ust_elf_create");
	68
	69	ret = lttng_ust_elf_get_memsz(elf, &memsz);
	70	ok(ret == 0, "lttng_ust_elf_get_memsz returned successfully");
	71	ok(memsz == exp_memsz,
	72	"memsz - expected: %lu, got: %lu",
	73	exp_memsz, memsz);
	74
	75	ret = lttng_ust_elf_get_build_id(elf, &build_id, &build_id_len,
	76	&has_build_id);
	77	ok(ret == 0, "lttng_ust_elf_get_build_id returned successfully");
	78	ok(has_build_id == 1, "build id marked as found");
	79	ok(build_id_len == BUILD_ID_LEN,
	80	"build_id_len - expected: %u, got: %u",
	81	BUILD_ID_LEN, build_id_len);
	82	ok(memcmp(build_id, exp_build_id, build_id_len) == 0,
	83	"build_id has expected value");
	84
	85	ret = lttng_ust_elf_get_debug_link(elf, &dbg_file, &crc,
	86	&has_debug_link);
	87	ok(ret == 0, "lttng_ust_elf_get_debug_link returned successfully");
	88	ok(has_debug_link == 1, "debug link marked as found");
	89	ok(strcmp(dbg_file, DBG_FILE) == 0,
	90	"debug link filename - expected: %s, got: %s",
	91	DBG_FILE, dbg_file);
	92	ok(crc == exp_crc,
	93	"debug link crc - expected: %#x, got: %#x",
94	exp_crc, crc);
95
96	free(build_id);
97	free(dbg_file);
98	lttng_ust_elf_destroy(elf);
99	}
100
101	int main(int argc, char **argv)
102	{
103	uint8_t X86_BUILD_ID[BUILD_ID_LEN] = {
104	0x27, 0x79, 0x2a, 0xe7, 0xaa, 0xef, 0x72, 0x5c, 0x9c, 0x52,
105	0x80, 0xec, 0x1e, 0x18, 0xd8, 0x09, 0x02, 0xba, 0xbc, 0x82
106	};
107	uint8_t X86_64_BUILD_ID[BUILD_ID_LEN] = {
108	0x0f, 0x87, 0xb2, 0xe2, 0x24, 0x9c, 0xe1, 0xc2, 0x24, 0xb1,
109	0xf8, 0xb6, 0x65, 0x83, 0xa3, 0xc1, 0xcb, 0x30, 0x5c, 0x63
110	};
111	uint8_t ARMEB_BUILD_ID[BUILD_ID_LEN] = {
112	0x60, 0x5d, 0x26, 0xa0, 0x0e, 0x30, 0xa4, 0x29, 0xf4, 0xf1,
113	0x85, 0x53, 0xda, 0x90, 0x68, 0xe1, 0xf5, 0x67, 0xbe, 0x42
114	};
115	uint8_t AARCH64_BE_BUILD_ID[BUILD_ID_LEN] = {
116	0xb9, 0x0a, 0xa0, 0xed, 0xd1, 0x41, 0x42, 0xc3, 0x34, 0x85,
117	0xfa, 0x27, 0x2e, 0xa9, 0x2f, 0xd2, 0xe4, 0xf7, 0xb6, 0x60
118	};
62eb8bca	119	const char *test_dir;
22609c7a AB	120
	121	plan_tests(NUM_TESTS);
	122
62eb8bca MD	123	ok(argc == 2, "Invoke as: %s <path>", argv[0]);
	124	if (argc != 2) {
	125	return EXIT_FAILURE;
	126	} else {
	127	test_dir = argv[1];
	128	}
	129
	130	test_elf(test_dir, "x86", X86_MEMSZ, X86_BUILD_ID, X86_CRC);
	131	test_elf(test_dir, "x86_64", X86_64_MEMSZ, X86_64_BUILD_ID, X86_64_CRC);
	132	test_elf(test_dir, "armeb", ARMEB_MEMSZ, ARMEB_BUILD_ID, ARMEB_CRC);
	133	test_elf(test_dir, "aarch64_be", AARCH64_BE_MEMSZ, AARCH64_BE_BUILD_ID,
22609c7a AB	134	AARCH64_BE_CRC);
22609c7a AB	135
62eb8bca	136	return EXIT_SUCCESS;
22609c7a	137	}