[J-u-boot.git] / tools / relocate-rela.c

// SPDX-License-Identifier: GPL-2.0+ OR BSD-2-Clause
/*
 * Copyright 2013 Freescale Semiconductor, Inc.
 *
 * 64-bit and little-endian target only until we need to support a different
 * arch that needs this.
 */

#include <elf.h>
#include <errno.h>
#include <inttypes.h>
#include <stdarg.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "compiler.h"

#ifndef EM_AARCH64
#define EM_AARCH64		183
#endif

#ifndef R_AARCH64_RELATIVE
#define R_AARCH64_RELATIVE	1027
#endif

#ifndef EM_MICROBLAZE
#define EM_MICROBLAZE		189
#endif

#ifndef R_MICROBLAZE_NONE
#define R_MICROBLAZE_NONE	0
#endif

#ifndef R_MICROBLAZE_32
#define R_MICROBLAZE_32		1
#endif

#ifndef R_MICROBLAZE_REL
#define R_MICROBLAZE_REL	16
#endif

#ifndef R_MICROBLAZE_GLOB_DAT
#define R_MICROBLAZE_GLOB_DAT	18
#endif

static int ei_class;
static int ei_data;

static uint64_t rela_start, rela_end, text_base, dyn_start;

static const bool debug_en;

static void debug(const char *fmt, ...)
{
	va_list args;

	if (debug_en) {
		va_start(args, fmt);
		vprintf(fmt, args);
		va_end(args);
	}
}

static uint16_t elf16_to_cpu(uint16_t data)
{
	if (ei_data == ELFDATA2LSB)
		return le16_to_cpu(data);

	return be16_to_cpu(data);
}

static uint32_t elf32_to_cpu(uint32_t data)
{
	if (ei_data == ELFDATA2LSB)
		return le32_to_cpu(data);

	return be32_to_cpu(data);
}

static bool supported_rela(Elf64_Rela *rela)
{
	uint64_t mask = 0xffffffffULL; /* would be different on 32-bit */
	uint32_t type = rela->r_info & mask;

	switch (type) {
	case R_AARCH64_RELATIVE:
		return true;
	default:
		fprintf(stderr, "warning: unsupported relocation type %"
				PRIu32 " at %" PRIx64 "\n",
			type, rela->r_offset);

		return false;
	}
}

static int decode_elf64(FILE *felf, char **argv)
{
	size_t size;
	Elf64_Ehdr header;
	uint64_t section_header_base, section_header_size;
	uint64_t sh_addr, sh_offset, sh_size;
	Elf64_Half sh_index, sh_num;
	Elf64_Shdr *sh_table; /* Elf symbol table */
	int ret, i, machine;
	char *sh_str;

	debug("64bit version\n");

	/* Make sure we are at start */
	rewind(felf);

	size = fread(&header, 1, sizeof(header), felf);
	if (size != sizeof(header)) {
		fclose(felf);
		return 25;
	}

	machine = le16_to_cpu(header.e_machine);
	debug("Machine\t%d\n", machine);

	if (machine != EM_AARCH64) {
		fprintf(stderr, "%s: Not supported machine type\n", argv[0]);
		return 30;
	}

	text_base = le64_to_cpu(header.e_entry);
	section_header_base = le64_to_cpu(header.e_shoff);
	section_header_size = le16_to_cpu(header.e_shentsize) *
			      le16_to_cpu(header.e_shnum);

	sh_table = malloc(section_header_size);
	if (!sh_table) {
		fprintf(stderr, "%s: Cannot allocate space for section header\n",
			argv[0]);
		fclose(felf);
		return 26;
	}

	ret = fseek(felf, section_header_base, SEEK_SET);
	if (ret) {
		fprintf(stderr, "%s: Can't set pointer to section header: %x/%lx\n",
			argv[0], ret, section_header_base);
		free(sh_table);
		fclose(felf);
		return 26;
	}

	size = fread(sh_table, 1, section_header_size, felf);
	if (size != section_header_size) {
		fprintf(stderr, "%s: Can't read section header: %lx/%lx\n",
			argv[0], size, section_header_size);
		free(sh_table);
		fclose(felf);
		return 27;
	}

	sh_index = le16_to_cpu(header.e_shstrndx);
	sh_size = le64_to_cpu(sh_table[sh_index].sh_size);
	debug("e_shstrndx %x, sh_size %lx\n", sh_index, sh_size);

	sh_str = malloc(sh_size);
	if (!sh_str) {
		fprintf(stderr, "malloc failed\n");
		free(sh_table);
		fclose(felf);
		return 28;
	}

	/*
	 * Specifies the byte offset from the beginning of the file
	 * to the first byte in the section.
	 */
	sh_offset = le64_to_cpu(sh_table[sh_index].sh_offset);
	sh_num = le16_to_cpu(header.e_shnum);

	ret = fseek(felf, sh_offset, SEEK_SET);
	if (ret) {
		fprintf(stderr, "Setting up sh_offset failed\n");
		free(sh_str);
		free(sh_table);
		fclose(felf);
		return 29;
	}

	size = fread(sh_str, 1, sh_size, felf);
	if (size != sh_size) {
		fprintf(stderr, "%s: Can't read section: %lx/%lx\n",
			argv[0], size, sh_size);
		free(sh_str);
		free(sh_table);
		fclose(felf);
		return 30;
	}

	for (i = 0; i < sh_num; i++) {
		char *sh_name = sh_str + le32_to_cpu(sh_table[i].sh_name);

		debug("%s\n", sh_name);

		sh_addr = le64_to_cpu(sh_table[i].sh_addr);
		sh_offset = le64_to_cpu(sh_table[i].sh_offset);
		sh_size = le64_to_cpu(sh_table[i].sh_size);

		if (!strcmp(".rela.dyn", sh_name)) {
			debug("Found section\t\".rela_dyn\"\n");
			debug(" at addr\t0x%08x\n", sh_addr);
			debug(" at offset\t0x%08x\n", sh_offset);
			debug(" of size\t0x%08x\n", sh_size);
			rela_start = sh_addr;
			rela_end = rela_start + sh_size;
			break;
		}
	}

	/* Clean up */
	free(sh_str);
	free(sh_table);
	fclose(felf);

	debug("text_base\t0x%08lx\n", text_base);
	debug("rela_start\t0x%08lx\n", rela_start);
	debug("rela_end\t0x%08lx\n", rela_end);

	if (!rela_start)
		return 1;

	return 0;
}

static int decode_elf32(FILE *felf, char **argv)
{
	size_t size;
	Elf32_Ehdr header;
	uint64_t section_header_base, section_header_size;
	uint32_t sh_addr, sh_offset, sh_size;
	Elf32_Half sh_index, sh_num;
	Elf32_Shdr *sh_table; /* Elf symbol table */
	int ret, i, machine;
	char *sh_str;

	debug("32bit version\n");

	/* Make sure we are at start */
	rewind(felf);

	size = fread(&header, 1, sizeof(header), felf);
	if (size != sizeof(header)) {
		fclose(felf);
		return 25;
	}

	machine = elf16_to_cpu(header.e_machine);
	debug("Machine %d\n", machine);

	if (machine != EM_MICROBLAZE) {
		fprintf(stderr, "%s: Not supported machine type\n", argv[0]);
		return 30;
	}

	text_base = elf32_to_cpu(header.e_entry);
	section_header_base = elf32_to_cpu(header.e_shoff);
	section_header_size = elf16_to_cpu(header.e_shentsize) *
			      elf16_to_cpu(header.e_shnum);

	sh_table = malloc(section_header_size);
	if (!sh_table) {
		fprintf(stderr, "%s: Cannot allocate space for section header\n",
			argv[0]);
		fclose(felf);
		return 26;
	}

	ret = fseek(felf, section_header_base, SEEK_SET);
	if (ret) {
		fprintf(stderr, "%s: Can't set pointer to section header: %x/%lx\n",
			argv[0], ret, section_header_base);
		free(sh_table);
		fclose(felf);
		return 26;
	}

	size = fread(sh_table, 1, section_header_size, felf);
	if (size != section_header_size) {
		fprintf(stderr, "%s: Can't read section header: %lx/%lx\n",
			argv[0], size, section_header_size);
		free(sh_table);
		fclose(felf);
		return 27;
	}

	sh_index = elf16_to_cpu(header.e_shstrndx);
	sh_size = elf32_to_cpu(sh_table[sh_index].sh_size);
	debug("e_shstrndx %x, sh_size %lx\n", sh_index, sh_size);

	sh_str = malloc(sh_size);
	if (!sh_str) {
		fprintf(stderr, "malloc failed\n");
		free(sh_table);
		fclose(felf);
		return 28;
	}

	/*
	 * Specifies the byte offset from the beginning of the file
	 * to the first byte in the section.
	 */
	sh_offset = elf32_to_cpu(sh_table[sh_index].sh_offset);
	sh_num = elf16_to_cpu(header.e_shnum);

	ret = fseek(felf, sh_offset, SEEK_SET);
	if (ret) {
		fprintf(stderr, "Setting up sh_offset failed\n");
		free(sh_str);
		free(sh_table);
		fclose(felf);
		return 29;
	}

	size = fread(sh_str, 1, sh_size, felf);
	if (size != sh_size) {
		fprintf(stderr, "%s: Can't read section: %lx/%x\n",
			argv[0], size, sh_size);
		free(sh_str);
		free(sh_table);
		fclose(felf);
		return 30;
	}

	for (i = 0; i < sh_num; i++) {
		char *sh_name = sh_str + elf32_to_cpu(sh_table[i].sh_name);

		debug("%s\n", sh_name);

		sh_addr = elf32_to_cpu(sh_table[i].sh_addr);
		sh_offset = elf32_to_cpu(sh_table[i].sh_offset);
		sh_size = elf32_to_cpu(sh_table[i].sh_size);

		if (!strcmp(".rela.dyn", sh_name)) {
			debug("Found section\t\".rela_dyn\"\n");
			debug(" at addr\t0x%08x\n", sh_addr);
			debug(" at offset\t0x%08x\n", sh_offset);
			debug(" of size\t0x%08x\n", sh_size);
			rela_start = sh_addr;
			rela_end = rela_start + sh_size;
		}
		if (!strcmp(".dynsym", sh_name)) {
			debug("Found section\t\".dynsym\"\n");
			debug(" at addr\t0x%08x\n", sh_addr);
			debug(" at offset\t0x%08x\n", sh_offset);
			debug(" of size\t0x%08x\n", sh_size);
			dyn_start = sh_addr;
		}
	}

	/* Clean up */
	free(sh_str);
	free(sh_table);
	fclose(felf);

	debug("text_base\t0x%08lx\n", text_base);
	debug("rela_start\t0x%08lx\n", rela_start);
	debug("rela_end\t0x%08lx\n", rela_end);
	debug("dyn_start\t0x%08lx\n", dyn_start);

	if (!rela_start)
		return 1;

	return 0;
}

static int decode_elf(char **argv)
{
	FILE *felf;
	size_t size;
	unsigned char e_ident[EI_NIDENT];

	felf = fopen(argv[2], "r+b");
	if (!felf) {
		fprintf(stderr, "%s: Cannot open %s: %s\n",
			argv[0], argv[5], strerror(errno));
		return 2;
	}

	size = fread(e_ident, 1, EI_NIDENT, felf);
	if (size != EI_NIDENT) {
		fclose(felf);
		return 25;
	}

	/* Check if this is really ELF file */
	if (e_ident[0] != 0x7f &&
	    e_ident[1] != 'E' &&
	    e_ident[2] != 'L' &&
	    e_ident[3] != 'F') {
		fclose(felf);
		return 1;
	}

	ei_class = e_ident[4];
	debug("EI_CLASS(1=32bit, 2=64bit) %d\n", ei_class);

	ei_data = e_ident[5];
	debug("EI_DATA(1=little endian, 2=big endian) %d\n", ei_data);

	if (ei_class == 2)
		return decode_elf64(felf, argv);

	return decode_elf32(felf, argv);
}

static int rela_elf64(char **argv, FILE *f)
{
	int i, num;

	if ((rela_end - rela_start) % sizeof(Elf64_Rela)) {
		fprintf(stderr, "%s: rela size isn't a multiple of Elf64_Rela\n", argv[0]);
		return 3;
	}

	num = (rela_end - rela_start) / sizeof(Elf64_Rela);

	for (i = 0; i < num; i++) {
		Elf64_Rela rela, swrela;
		uint64_t pos = rela_start + sizeof(Elf64_Rela) * i;
		uint64_t addr;

		if (fseek(f, pos, SEEK_SET) < 0) {
			fprintf(stderr, "%s: %s: seek to %" PRIx64
					" failed: %s\n",
				argv[0], argv[1], pos, strerror(errno));
		}

		if (fread(&rela, sizeof(rela), 1, f) != 1) {
			fprintf(stderr, "%s: %s: read rela failed at %"
					PRIx64 "\n",
				argv[0], argv[1], pos);
			return 4;
		}

		swrela.r_offset = le64_to_cpu(rela.r_offset);
		swrela.r_info = le64_to_cpu(rela.r_info);
		swrela.r_addend = le64_to_cpu(rela.r_addend);

		if (!supported_rela(&swrela))
			continue;

		debug("Rela %" PRIx64 " %" PRIu64 " %" PRIx64 "\n",
		      swrela.r_offset, swrela.r_info, swrela.r_addend);

		if (swrela.r_offset < text_base) {
			fprintf(stderr, "%s: %s: bad rela at %" PRIx64 "\n",
				argv[0], argv[1], pos);
			return 4;
		}

		addr = swrela.r_offset - text_base;

		if (fseek(f, addr, SEEK_SET) < 0) {
			fprintf(stderr, "%s: %s: seek to %"
					PRIx64 " failed: %s\n",
				argv[0], argv[1], addr, strerror(errno));
		}

		if (fwrite(&rela.r_addend, sizeof(rela.r_addend), 1, f) != 1) {
			fprintf(stderr, "%s: %s: write failed at %" PRIx64 "\n",
				argv[0], argv[1], addr);
			return 4;
		}
	}

	return 0;
}

static bool supported_rela32(Elf32_Rela *rela, uint32_t *type)
{
	uint32_t mask = 0xffULL; /* would be different on 32-bit */
	*type = rela->r_info & mask;

	debug("Type:\t");

	switch (*type) {
	case R_MICROBLAZE_32:
		debug("R_MICROBLAZE_32\n");
		return true;
	case R_MICROBLAZE_GLOB_DAT:
		debug("R_MICROBLAZE_GLOB_DAT\n");
		return true;
	case R_MICROBLAZE_NONE:
		debug("R_MICROBLAZE_NONE - ignoring - do nothing\n");
		return false;
	case R_MICROBLAZE_REL:
		debug("R_MICROBLAZE_REL\n");
		return true;
	default:
		fprintf(stderr, "warning: unsupported relocation type %"
			PRIu32 " at %" PRIx32 "\n", *type, rela->r_offset);

		return false;
	}
}

static int rela_elf32(char **argv, FILE *f)
{
	int i, num, index;
	uint32_t value, type;

	if ((rela_end - rela_start) % sizeof(Elf32_Rela)) {
		fprintf(stderr, "%s: rela size isn't a multiple of Elf32_Rela\n", argv[0]);
		return 3;
	}

	num = (rela_end - rela_start) / sizeof(Elf32_Rela);

	debug("Number of entries: %u\n", num);

	for (i = 0; i < num; i++) {
		Elf32_Rela rela, swrela;
		Elf32_Sym symbols;
		uint32_t pos = rela_start + sizeof(Elf32_Rela) * i;
		uint32_t addr, pos_dyn;

		debug("\nPosition:\t%d/0x%x\n", i, pos);

		if (fseek(f, pos, SEEK_SET) < 0) {
			fprintf(stderr, "%s: %s: seek to %" PRIx32
					" failed: %s\n",
				argv[0], argv[1], pos, strerror(errno));
		}

		if (fread(&rela, sizeof(rela), 1, f) != 1) {
			fprintf(stderr, "%s: %s: read rela failed at %"
					PRIx32 "\n",
				argv[0], argv[1], pos);
			return 4;
		}

		debug("Rela:\toffset:\t%" PRIx32 " r_info:\t%"
		      PRIu32 " r_addend:\t%" PRIx32 "\n",
		      rela.r_offset, rela.r_info, rela.r_addend);

		swrela.r_offset = elf32_to_cpu(rela.r_offset);
		swrela.r_info = elf32_to_cpu(rela.r_info);
		swrela.r_addend = elf32_to_cpu(rela.r_addend);

		debug("SWRela:\toffset:\t%" PRIx32 " r_info:\t%"
		      PRIu32 " r_addend:\t%" PRIx32 "\n",
		      swrela.r_offset, swrela.r_info, swrela.r_addend);

		if (!supported_rela32(&swrela, &type))
			continue;

		if (swrela.r_offset < text_base) {
			fprintf(stderr, "%s: %s: bad rela at %" PRIx32 "\n",
				argv[0], argv[1], pos);
			return 4;
		}

		addr = swrela.r_offset - text_base;

		debug("Addr:\t0x%" PRIx32 "\n", addr);

		switch (type) {
		case R_MICROBLAZE_REL:
			if (fseek(f, addr, SEEK_SET) < 0) {
				fprintf(stderr, "%s: %s: seek to %"
					PRIx32 " failed: %s\n",
					argv[0], argv[1], addr, strerror(errno));
				return 5;
			}

			debug("Write addend\n");

			if (fwrite(&rela.r_addend, sizeof(rela.r_addend), 1, f) != 1) {
				fprintf(stderr, "%s: %s: write failed at %" PRIx32 "\n",
					argv[0], argv[1], addr);
				return 4;
			}
			break;
		case R_MICROBLAZE_32:
		case R_MICROBLAZE_GLOB_DAT:
			/* global symbols read it and add reloc offset */
			index = swrela.r_info >> 8;
			pos_dyn = dyn_start + sizeof(Elf32_Sym) * index;

			debug("Index:\t%d\n", index);
			debug("Pos_dyn:\t0x%x\n", pos_dyn);

			if (fseek(f, pos_dyn, SEEK_SET) < 0) {
				fprintf(stderr, "%s: %s: seek to %"
					PRIx32 " failed: %s\n",
					argv[0], argv[1], pos_dyn, strerror(errno));
				return 5;
			}

			if (fread(&symbols, sizeof(symbols), 1, f) != 1) {
				fprintf(stderr, "%s: %s: read symbols failed at %"
						PRIx32 "\n",
					argv[0], argv[1], pos_dyn);
				return 4;
			}

			debug("Symbol description:\n");
			debug(" st_name:\t0x%x\n", symbols.st_name);
			debug(" st_value:\t0x%x\n", symbols.st_value);
			debug(" st_size:\t0x%x\n", symbols.st_size);

			value = swrela.r_addend + symbols.st_value;

			debug("Value:\t0x%x\n", value);

			if (fseek(f, addr, SEEK_SET) < 0) {
				fprintf(stderr, "%s: %s: seek to %"
					PRIx32 " failed: %s\n",
					argv[0], argv[1], addr, strerror(errno));
				return 5;
			}

			if (fwrite(&value, sizeof(rela.r_addend), 1, f) != 1) {
				fprintf(stderr, "%s: %s: write failed at %" PRIx32 "\n",
					argv[0], argv[1], addr);
				return 4;
			}

			break;
		case R_MICROBLAZE_NONE:
			debug("R_MICROBLAZE_NONE - skip\n");
			break;
		default:
			fprintf(stderr, "warning: unsupported relocation type %"
				PRIu32 " at %" PRIx32 "\n",
				type, rela.r_offset);
		}
	}

	return 0;
}

int main(int argc, char **argv)
{
	FILE *f;
	int ret;
	uint64_t file_size;

	if (argc != 3) {
		fprintf(stderr, "Statically apply ELF rela relocations\n");
		fprintf(stderr, "Usage: %s <bin file> <u-boot ELF>\n",
			argv[0]);
		return 1;
	}

	ret = decode_elf(argv);
	if (ret) {
		fprintf(stderr, "ELF decoding failed\n");
		return ret;
	}

	if (rela_start > rela_end || rela_start < text_base) {
		fprintf(stderr, "%s: bad rela bounds\n", argv[0]);
		return 3;
	}

	rela_start -= text_base;
	rela_end -= text_base;
	dyn_start -= text_base;

	f = fopen(argv[1], "r+b");
	if (!f) {
		fprintf(stderr, "%s: Cannot open %s: %s\n",
			argv[0], argv[1], strerror(errno));
		return 2;
	}

	fseek(f, 0, SEEK_END);
	file_size = ftell(f);
	rewind(f);

	if (rela_end > file_size) {
		// Most likely compiler inserted some section that didn't get
		// objcopy-ed into the final binary
		rela_end = file_size;
	}

	if (ei_class == 2)
		ret = rela_elf64(argv, f);
	else
		ret = rela_elf32(argv, f);

	if (fclose(f) < 0) {
		fprintf(stderr, "%s: %s: close failed: %s\n",
			argv[0], argv[1], strerror(errno));
		return 4;
	}

	return ret;
}
Commit	Line	Data
4549e789	1	// SPDX-License-Identifier: GPL-2.0+ OR BSD-2-Clause
8137af19 SW	2	/*
	3	* Copyright 2013 Freescale Semiconductor, Inc.
	4	*
8137af19 SW	5	* 64-bit and little-endian target only until we need to support a different
	6	* arch that needs this.
	7	*/
	8
	9	#include <elf.h>
	10	#include <errno.h>
	11	#include <inttypes.h>
	12	#include <stdarg.h>
	13	#include <stdbool.h>
	14	#include <stdio.h>
	15	#include <stdlib.h>
	16	#include <string.h>
43db3e3b	17	#include "compiler.h"
8137af19	18
65fc1697 MS	19	#ifndef EM_AARCH64
	20	#define EM_AARCH64 183
	21	#endif
	22
8137af19 SW	23	#ifndef R_AARCH64_RELATIVE
	24	#define R_AARCH64_RELATIVE 1027
	25	#endif
	26
65fc1697 MS	27	#ifndef EM_MICROBLAZE
	28	#define EM_MICROBLAZE 189
	29	#endif
	30
	31	#ifndef R_MICROBLAZE_NONE
	32	#define R_MICROBLAZE_NONE 0
	33	#endif
	34
	35	#ifndef R_MICROBLAZE_32
	36	#define R_MICROBLAZE_32 1
	37	#endif
	38
	39	#ifndef R_MICROBLAZE_REL
	40	#define R_MICROBLAZE_REL 16
	41	#endif
	42
	43	#ifndef R_MICROBLAZE_GLOB_DAT
	44	#define R_MICROBLAZE_GLOB_DAT 18
	45	#endif
	46
4c9e2d64	47	static int ei_class;
424f04fc	48	static int ei_data;
4c9e2d64	49
30fb8d29	50	static uint64_t rela_start, rela_end, text_base, dyn_start;
d8b0444b	51
8137af19 SW	52	static const bool debug_en;
	53
	54	static void debug(const char *fmt, ...)
	55	{
	56	va_list args;
	57
d27e35f2 HS	58	if (debug_en) {
d27e35f2 HS	59	va_start(args, fmt);
8137af19	60	vprintf(fmt, args);
d27e35f2 HS	61	va_end(args);
d27e35f2 HS	62	}
8137af19 SW	63	}
8137af19 SW	64
424f04fc OP	65	static uint16_t elf16_to_cpu(uint16_t data)
	66	{
	67	if (ei_data == ELFDATA2LSB)
	68	return le16_to_cpu(data);
	69
	70	return be16_to_cpu(data);
	71	}
	72
	73	static uint32_t elf32_to_cpu(uint32_t data)
	74	{
	75	if (ei_data == ELFDATA2LSB)
	76	return le32_to_cpu(data);
	77
	78	return be32_to_cpu(data);
	79	}
	80
8137af19 SW	81	static bool supported_rela(Elf64_Rela *rela)
	82	{
	83	uint64_t mask = 0xffffffffULL; /* would be different on 32-bit */
	84	uint32_t type = rela->r_info & mask;
	85
	86	switch (type) {
8137af19 SW	87	case R_AARCH64_RELATIVE:
8137af19 SW	88	return true;
8137af19 SW	89	default:
	90	fprintf(stderr, "warning: unsupported relocation type %"
	91	PRIu32 " at %" PRIx64 "\n",
	92	type, rela->r_offset);
	93
	94	return false;
	95	}
	96	}
	97
4c9e2d64 MS	98	static int decode_elf64(FILE felf, char *argv)
	99	{
	100	size_t size;
	101	Elf64_Ehdr header;
5e0e1a86 SH	102	uint64_t section_header_base, section_header_size;
	103	uint64_t sh_addr, sh_offset, sh_size;
	104	Elf64_Half sh_index, sh_num;
4c9e2d64 MS	105	Elf64_Shdr sh_table; / Elf symbol table */
	106	int ret, i, machine;
	107	char *sh_str;
	108
	109	debug("64bit version\n");
	110
	111	/* Make sure we are at start */
	112	rewind(felf);
	113
	114	size = fread(&header, 1, sizeof(header), felf);
	115	if (size != sizeof(header)) {
	116	fclose(felf);
	117	return 25;
	118	}
	119
5e0e1a86	120	machine = le16_to_cpu(header.e_machine);
4c9e2d64 MS	121	debug("Machine\t%d\n", machine);
4c9e2d64 MS	122
a1405d9c MS	123	if (machine != EM_AARCH64) {
	124	fprintf(stderr, "%s: Not supported machine type\n", argv[0]);
	125	return 30;
	126	}
	127
5e0e1a86 SH	128	text_base = le64_to_cpu(header.e_entry);
	129	section_header_base = le64_to_cpu(header.e_shoff);
	130	section_header_size = le16_to_cpu(header.e_shentsize) *
	131	le16_to_cpu(header.e_shnum);
4c9e2d64 MS	132
	133	sh_table = malloc(section_header_size);
	134	if (!sh_table) {
	135	fprintf(stderr, "%s: Cannot allocate space for section header\n",
	136	argv[0]);
	137	fclose(felf);
	138	return 26;
	139	}
	140
	141	ret = fseek(felf, section_header_base, SEEK_SET);
	142	if (ret) {
	143	fprintf(stderr, "%s: Can't set pointer to section header: %x/%lx\n",
	144	argv[0], ret, section_header_base);
	145	free(sh_table);
	146	fclose(felf);
	147	return 26;
	148	}
	149
	150	size = fread(sh_table, 1, section_header_size, felf);
	151	if (size != section_header_size) {
	152	fprintf(stderr, "%s: Can't read section header: %lx/%lx\n",
	153	argv[0], size, section_header_size);
	154	free(sh_table);
	155	fclose(felf);
	156	return 27;
	157	}
	158
5e0e1a86 SH	159	sh_index = le16_to_cpu(header.e_shstrndx);
	160	sh_size = le64_to_cpu(sh_table[sh_index].sh_size);
	161	debug("e_shstrndx %x, sh_size %lx\n", sh_index, sh_size);
4c9e2d64 MS	162
	163	sh_str = malloc(sh_size);
	164	if (!sh_str) {
	165	fprintf(stderr, "malloc failed\n");
	166	free(sh_table);
	167	fclose(felf);
	168	return 28;
	169	}
	170
	171	/*
	172	* Specifies the byte offset from the beginning of the file
	173	* to the first byte in the section.
	174	*/
5e0e1a86 SH	175	sh_offset = le64_to_cpu(sh_table[sh_index].sh_offset);
5e0e1a86 SH	176	sh_num = le16_to_cpu(header.e_shnum);
4c9e2d64 MS	177
	178	ret = fseek(felf, sh_offset, SEEK_SET);
	179	if (ret) {
	180	fprintf(stderr, "Setting up sh_offset failed\n");
	181	free(sh_str);
	182	free(sh_table);
	183	fclose(felf);
	184	return 29;
	185	}
	186
	187	size = fread(sh_str, 1, sh_size, felf);
	188	if (size != sh_size) {
	189	fprintf(stderr, "%s: Can't read section: %lx/%lx\n",
	190	argv[0], size, sh_size);
	191	free(sh_str);
	192	free(sh_table);
	193	fclose(felf);
	194	return 30;
	195	}
	196
5e0e1a86 SH	197	for (i = 0; i < sh_num; i++) {
	198	char *sh_name = sh_str + le32_to_cpu(sh_table[i].sh_name);
	199
	200	debug("%s\n", sh_name);
	201
	202	sh_addr = le64_to_cpu(sh_table[i].sh_addr);
	203	sh_offset = le64_to_cpu(sh_table[i].sh_offset);
	204	sh_size = le64_to_cpu(sh_table[i].sh_size);
	205
	206	if (!strcmp(".rela.dyn", sh_name)) {
4c9e2d64	207	debug("Found section\t\".rela_dyn\"\n");
5e0e1a86 SH	208	debug(" at addr\t0x%08x\n", sh_addr);
	209	debug(" at offset\t0x%08x\n", sh_offset);
	210	debug(" of size\t0x%08x\n", sh_size);
	211	rela_start = sh_addr;
	212	rela_end = rela_start + sh_size;
4c9e2d64 MS	213	break;
	214	}
	215	}
	216
	217	/* Clean up */
	218	free(sh_str);
	219	free(sh_table);
	220	fclose(felf);
	221
	222	debug("text_base\t0x%08lx\n", text_base);
	223	debug("rela_start\t0x%08lx\n", rela_start);
	224	debug("rela_end\t0x%08lx\n", rela_end);
	225
	226	if (!rela_start)
	227	return 1;
	228
	229	return 0;
	230	}
	231
30fb8d29 MS	232	static int decode_elf32(FILE felf, char *argv)
	233	{
	234	size_t size;
	235	Elf32_Ehdr header;
5e0e1a86 SH	236	uint64_t section_header_base, section_header_size;
	237	uint32_t sh_addr, sh_offset, sh_size;
	238	Elf32_Half sh_index, sh_num;
30fb8d29 MS	239	Elf32_Shdr sh_table; / Elf symbol table */
	240	int ret, i, machine;
	241	char *sh_str;
	242
	243	debug("32bit version\n");
	244
	245	/* Make sure we are at start */
	246	rewind(felf);
	247
	248	size = fread(&header, 1, sizeof(header), felf);
	249	if (size != sizeof(header)) {
	250	fclose(felf);
	251	return 25;
	252	}
	253
02d30e5f	254	machine = elf16_to_cpu(header.e_machine);
30fb8d29 MS	255	debug("Machine %d\n", machine);
	256
	257	if (machine != EM_MICROBLAZE) {
	258	fprintf(stderr, "%s: Not supported machine type\n", argv[0]);
	259	return 30;
	260	}
	261
02d30e5f OP	262	text_base = elf32_to_cpu(header.e_entry);
	263	section_header_base = elf32_to_cpu(header.e_shoff);
	264	section_header_size = elf16_to_cpu(header.e_shentsize) *
	265	elf16_to_cpu(header.e_shnum);
30fb8d29 MS	266
	267	sh_table = malloc(section_header_size);
	268	if (!sh_table) {
	269	fprintf(stderr, "%s: Cannot allocate space for section header\n",
	270	argv[0]);
	271	fclose(felf);
	272	return 26;
	273	}
	274
	275	ret = fseek(felf, section_header_base, SEEK_SET);
	276	if (ret) {
	277	fprintf(stderr, "%s: Can't set pointer to section header: %x/%lx\n",
	278	argv[0], ret, section_header_base);
	279	free(sh_table);
	280	fclose(felf);
	281	return 26;
	282	}
	283
	284	size = fread(sh_table, 1, section_header_size, felf);
	285	if (size != section_header_size) {
	286	fprintf(stderr, "%s: Can't read section header: %lx/%lx\n",
	287	argv[0], size, section_header_size);
	288	free(sh_table);
	289	fclose(felf);
	290	return 27;
	291	}
	292
02d30e5f OP	293	sh_index = elf16_to_cpu(header.e_shstrndx);
02d30e5f OP	294	sh_size = elf32_to_cpu(sh_table[sh_index].sh_size);
5e0e1a86	295	debug("e_shstrndx %x, sh_size %lx\n", sh_index, sh_size);
30fb8d29 MS	296
	297	sh_str = malloc(sh_size);
	298	if (!sh_str) {
	299	fprintf(stderr, "malloc failed\n");
	300	free(sh_table);
	301	fclose(felf);
	302	return 28;
	303	}
	304
	305	/*
	306	* Specifies the byte offset from the beginning of the file
	307	* to the first byte in the section.
	308	*/
02d30e5f OP	309	sh_offset = elf32_to_cpu(sh_table[sh_index].sh_offset);
02d30e5f OP	310	sh_num = elf16_to_cpu(header.e_shnum);
30fb8d29 MS	311
	312	ret = fseek(felf, sh_offset, SEEK_SET);
	313	if (ret) {
	314	fprintf(stderr, "Setting up sh_offset failed\n");
	315	free(sh_str);
	316	free(sh_table);
	317	fclose(felf);
	318	return 29;
	319	}
	320
	321	size = fread(sh_str, 1, sh_size, felf);
	322	if (size != sh_size) {
5e0e1a86	323	fprintf(stderr, "%s: Can't read section: %lx/%x\n",
30fb8d29 MS	324	argv[0], size, sh_size);
	325	free(sh_str);
	326	free(sh_table);
	327	fclose(felf);
	328	return 30;
	329	}
	330
5e0e1a86	331	for (i = 0; i < sh_num; i++) {
02d30e5f	332	char *sh_name = sh_str + elf32_to_cpu(sh_table[i].sh_name);
5e0e1a86 SH	333
	334	debug("%s\n", sh_name);
	335
02d30e5f OP	336	sh_addr = elf32_to_cpu(sh_table[i].sh_addr);
	337	sh_offset = elf32_to_cpu(sh_table[i].sh_offset);
	338	sh_size = elf32_to_cpu(sh_table[i].sh_size);
5e0e1a86 SH	339
5e0e1a86 SH	340	if (!strcmp(".rela.dyn", sh_name)) {
30fb8d29	341	debug("Found section\t\".rela_dyn\"\n");
5e0e1a86 SH	342	debug(" at addr\t0x%08x\n", sh_addr);
	343	debug(" at offset\t0x%08x\n", sh_offset);
	344	debug(" of size\t0x%08x\n", sh_size);
	345	rela_start = sh_addr;
	346	rela_end = rela_start + sh_size;
30fb8d29	347	}
5e0e1a86	348	if (!strcmp(".dynsym", sh_name)) {
30fb8d29	349	debug("Found section\t\".dynsym\"\n");
5e0e1a86 SH	350	debug(" at addr\t0x%08x\n", sh_addr);
	351	debug(" at offset\t0x%08x\n", sh_offset);
	352	debug(" of size\t0x%08x\n", sh_size);
	353	dyn_start = sh_addr;
30fb8d29 MS	354	}
	355	}
	356
	357	/* Clean up */
	358	free(sh_str);
	359	free(sh_table);
	360	fclose(felf);
	361
	362	debug("text_base\t0x%08lx\n", text_base);
	363	debug("rela_start\t0x%08lx\n", rela_start);
	364	debug("rela_end\t0x%08lx\n", rela_end);
	365	debug("dyn_start\t0x%08lx\n", dyn_start);
	366
	367	if (!rela_start)
	368	return 1;
	369
	370	return 0;
	371	}
	372
4c9e2d64	373	static int decode_elf(char **argv)
8137af19	374	{
4c9e2d64 MS	375	FILE *felf;
	376	size_t size;
	377	unsigned char e_ident[EI_NIDENT];
	378
	379	felf = fopen(argv[2], "r+b");
	380	if (!felf) {
	381	fprintf(stderr, "%s: Cannot open %s: %s\n",
	382	argv[0], argv[5], strerror(errno));
	383	return 2;
	384	}
	385
	386	size = fread(e_ident, 1, EI_NIDENT, felf);
	387	if (size != EI_NIDENT) {
	388	fclose(felf);
	389	return 25;
	390	}
	391
	392	/* Check if this is really ELF file */
	393	if (e_ident[0] != 0x7f &&
	394	e_ident[1] != 'E' &&
	395	e_ident[2] != 'L' &&
	396	e_ident[3] != 'F') {
	397	fclose(felf);
	398	return 1;
	399	}
	400
	401	ei_class = e_ident[4];
	402	debug("EI_CLASS(1=32bit, 2=64bit) %d\n", ei_class);
	403
424f04fc OP	404	ei_data = e_ident[5];
	405	debug("EI_DATA(1=little endian, 2=big endian) %d\n", ei_data);
	406
4c9e2d64 MS	407	if (ei_class == 2)
	408	return decode_elf64(felf, argv);
	409
30fb8d29	410	return decode_elf32(felf, argv);
8137af19 SW	411	}
8137af19 SW	412
582ffb5c	413	static int rela_elf64(char *argv, FILE f)
8137af19	414	{
582ffb5c	415	int i, num;
9d3d9816 AD	416
	417	if ((rela_end - rela_start) % sizeof(Elf64_Rela)) {
	418	fprintf(stderr, "%s: rela size isn't a multiple of Elf64_Rela\n", argv[0]);
	419	return 3;
	420	}
	421
8137af19 SW	422	num = (rela_end - rela_start) / sizeof(Elf64_Rela);
	423
	424	for (i = 0; i < num; i++) {
	425	Elf64_Rela rela, swrela;
	426	uint64_t pos = rela_start + sizeof(Elf64_Rela) * i;
	427	uint64_t addr;
	428
	429	if (fseek(f, pos, SEEK_SET) < 0) {
	430	fprintf(stderr, "%s: %s: seek to %" PRIx64
	431	" failed: %s\n",
	432	argv[0], argv[1], pos, strerror(errno));
	433	}
	434
	435	if (fread(&rela, sizeof(rela), 1, f) != 1) {
	436	fprintf(stderr, "%s: %s: read rela failed at %"
	437	PRIx64 "\n",
	438	argv[0], argv[1], pos);
	439	return 4;
	440	}
	441
5e0e1a86 SH	442	swrela.r_offset = le64_to_cpu(rela.r_offset);
	443	swrela.r_info = le64_to_cpu(rela.r_info);
	444	swrela.r_addend = le64_to_cpu(rela.r_addend);
8137af19 SW	445
	446	if (!supported_rela(&swrela))
	447	continue;
	448
	449	debug("Rela %" PRIx64 " %" PRIu64 " %" PRIx64 "\n",
	450	swrela.r_offset, swrela.r_info, swrela.r_addend);
	451
	452	if (swrela.r_offset < text_base) {
	453	fprintf(stderr, "%s: %s: bad rela at %" PRIx64 "\n",
	454	argv[0], argv[1], pos);
	455	return 4;
	456	}
	457
	458	addr = swrela.r_offset - text_base;
	459
	460	if (fseek(f, addr, SEEK_SET) < 0) {
	461	fprintf(stderr, "%s: %s: seek to %"
	462	PRIx64 " failed: %s\n",
	463	argv[0], argv[1], addr, strerror(errno));
	464	}
	465
	466	if (fwrite(&rela.r_addend, sizeof(rela.r_addend), 1, f) != 1) {
	467	fprintf(stderr, "%s: %s: write failed at %" PRIx64 "\n",
	468	argv[0], argv[1], addr);
	469	return 4;
	470	}
	471	}
	472
582ffb5c MS	473	return 0;
	474	}
	475
034944b3 MS	476	static bool supported_rela32(Elf32_Rela rela, uint32_t type)
	477	{
	478	uint32_t mask = 0xffULL; /* would be different on 32-bit */
	479	*type = rela->r_info & mask;
	480
	481	debug("Type:\t");
	482
	483	switch (*type) {
	484	case R_MICROBLAZE_32:
	485	debug("R_MICROBLAZE_32\n");
	486	return true;
	487	case R_MICROBLAZE_GLOB_DAT:
	488	debug("R_MICROBLAZE_GLOB_DAT\n");
	489	return true;
	490	case R_MICROBLAZE_NONE:
	491	debug("R_MICROBLAZE_NONE - ignoring - do nothing\n");
	492	return false;
	493	case R_MICROBLAZE_REL:
	494	debug("R_MICROBLAZE_REL\n");
	495	return true;
	496	default:
	497	fprintf(stderr, "warning: unsupported relocation type %"
	498	PRIu32 " at %" PRIx32 "\n", *type, rela->r_offset);
	499
	500	return false;
	501	}
	502	}
	503
	504	static int rela_elf32(char *argv, FILE f)
	505	{
	506	int i, num, index;
	507	uint32_t value, type;
	508
	509	if ((rela_end - rela_start) % sizeof(Elf32_Rela)) {
	510	fprintf(stderr, "%s: rela size isn't a multiple of Elf32_Rela\n", argv[0]);
	511	return 3;
	512	}
	513
	514	num = (rela_end - rela_start) / sizeof(Elf32_Rela);
	515
	516	debug("Number of entries: %u\n", num);
	517
	518	for (i = 0; i < num; i++) {
	519	Elf32_Rela rela, swrela;
	520	Elf32_Sym symbols;
	521	uint32_t pos = rela_start + sizeof(Elf32_Rela) * i;
	522	uint32_t addr, pos_dyn;
	523
24d32e05	524	debug("\nPosition:\t%d/0x%x\n", i, pos);
034944b3 MS	525
	526	if (fseek(f, pos, SEEK_SET) < 0) {
	527	fprintf(stderr, "%s: %s: seek to %" PRIx32
	528	" failed: %s\n",
	529	argv[0], argv[1], pos, strerror(errno));
	530	}
	531
	532	if (fread(&rela, sizeof(rela), 1, f) != 1) {
	533	fprintf(stderr, "%s: %s: read rela failed at %"
	534	PRIx32 "\n",
	535	argv[0], argv[1], pos);
	536	return 4;
	537	}
	538
	539	debug("Rela:\toffset:\t%" PRIx32 " r_info:\t%"
	540	PRIu32 " r_addend:\t%" PRIx32 "\n",
	541	rela.r_offset, rela.r_info, rela.r_addend);
	542
02d30e5f OP	543	swrela.r_offset = elf32_to_cpu(rela.r_offset);
	544	swrela.r_info = elf32_to_cpu(rela.r_info);
	545	swrela.r_addend = elf32_to_cpu(rela.r_addend);
034944b3 MS	546
	547	debug("SWRela:\toffset:\t%" PRIx32 " r_info:\t%"
	548	PRIu32 " r_addend:\t%" PRIx32 "\n",
	549	swrela.r_offset, swrela.r_info, swrela.r_addend);
	550
	551	if (!supported_rela32(&swrela, &type))
	552	continue;
	553
	554	if (swrela.r_offset < text_base) {
	555	fprintf(stderr, "%s: %s: bad rela at %" PRIx32 "\n",
	556	argv[0], argv[1], pos);
	557	return 4;
	558	}
	559
	560	addr = swrela.r_offset - text_base;
	561
	562	debug("Addr:\t0x%" PRIx32 "\n", addr);
	563
	564	switch (type) {
	565	case R_MICROBLAZE_REL:
	566	if (fseek(f, addr, SEEK_SET) < 0) {
	567	fprintf(stderr, "%s: %s: seek to %"
	568	PRIx32 " failed: %s\n",
	569	argv[0], argv[1], addr, strerror(errno));
	570	return 5;
	571	}
	572
	573	debug("Write addend\n");
	574
	575	if (fwrite(&rela.r_addend, sizeof(rela.r_addend), 1, f) != 1) {
	576	fprintf(stderr, "%s: %s: write failed at %" PRIx32 "\n",
	577	argv[0], argv[1], addr);
	578	return 4;
	579	}
	580	break;
	581	case R_MICROBLAZE_32:
	582	case R_MICROBLAZE_GLOB_DAT:
	583	/* global symbols read it and add reloc offset */
	584	index = swrela.r_info >> 8;
	585	pos_dyn = dyn_start + sizeof(Elf32_Sym) * index;
	586
	587	debug("Index:\t%d\n", index);
	588	debug("Pos_dyn:\t0x%x\n", pos_dyn);
	589
	590	if (fseek(f, pos_dyn, SEEK_SET) < 0) {
	591	fprintf(stderr, "%s: %s: seek to %"
	592	PRIx32 " failed: %s\n",
	593	argv[0], argv[1], pos_dyn, strerror(errno));
	594	return 5;
	595	}
	596
	597	if (fread(&symbols, sizeof(symbols), 1, f) != 1) {
	598	fprintf(stderr, "%s: %s: read symbols failed at %"
	599	PRIx32 "\n",
	600	argv[0], argv[1], pos_dyn);
	601	return 4;
	602	}
	603
	604	debug("Symbol description:\n");
	605	debug(" st_name:\t0x%x\n", symbols.st_name);
	606	debug(" st_value:\t0x%x\n", symbols.st_value);
	607	debug(" st_size:\t0x%x\n", symbols.st_size);
	608
	609	value = swrela.r_addend + symbols.st_value;
610
611	debug("Value:\t0x%x\n", value);
612
613	if (fseek(f, addr, SEEK_SET) < 0) {
614	fprintf(stderr, "%s: %s: seek to %"
615	PRIx32 " failed: %s\n",
616	argv[0], argv[1], addr, strerror(errno));
617	return 5;
618	}
619
620	if (fwrite(&value, sizeof(rela.r_addend), 1, f) != 1) {
621	fprintf(stderr, "%s: %s: write failed at %" PRIx32 "\n",
622	argv[0], argv[1], addr);
623	return 4;
624	}
625
626	break;
627	case R_MICROBLAZE_NONE:
628	debug("R_MICROBLAZE_NONE - skip\n");
629	break;
630	default:
631	fprintf(stderr, "warning: unsupported relocation type %"
632	PRIu32 " at %" PRIx32 "\n",
633	type, rela.r_offset);
634	}
635	}
636
637	return 0;
638	}
639
582ffb5c MS	640	int main(int argc, char **argv)
	641	{
	642	FILE *f;
	643	int ret;
	644	uint64_t file_size;
	645
	646	if (argc != 3) {
	647	fprintf(stderr, "Statically apply ELF rela relocations\n");
	648	fprintf(stderr, "Usage: %s <bin file> <u-boot ELF>\n",
	649	argv[0]);
	650	return 1;
	651	}
	652
	653	ret = decode_elf(argv);
	654	if (ret) {
	655	fprintf(stderr, "ELF decoding failed\n");
	656	return ret;
	657	}
	658
	659	if (rela_start > rela_end \|\| rela_start < text_base) {
	660	fprintf(stderr, "%s: bad rela bounds\n", argv[0]);
	661	return 3;
	662	}
	663
	664	rela_start -= text_base;
	665	rela_end -= text_base;
30fb8d29	666	dyn_start -= text_base;
582ffb5c MS	667
	668	f = fopen(argv[1], "r+b");
	669	if (!f) {
	670	fprintf(stderr, "%s: Cannot open %s: %s\n",
	671	argv[0], argv[1], strerror(errno));
	672	return 2;
	673	}
	674
	675	fseek(f, 0, SEEK_END);
	676	file_size = ftell(f);
	677	rewind(f);
	678
	679	if (rela_end > file_size) {
	680	// Most likely compiler inserted some section that didn't get
	681	// objcopy-ed into the final binary
	682	rela_end = file_size;
	683	}
	684
	685	if (ei_class == 2)
	686	ret = rela_elf64(argv, f);
034944b3 MS	687	else
034944b3 MS	688	ret = rela_elf32(argv, f);
582ffb5c	689
8137af19 SW	690	if (fclose(f) < 0) {
	691	fprintf(stderr, "%s: %s: close failed: %s\n",
	692	argv[0], argv[1], strerror(errno));
	693	return 4;
	694	}
	695
582ffb5c	696	return ret;
8137af19	697	}