summaryrefslogtreecommitdiff
path: root/arch/mips/tools/elf-entry.c
blob: dbd14ff05b4cddd96d05795d37418e32a2dcc37c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// SPDX-License-Identifier: GPL-2.0
#include <byteswap.h>
#include <elf.h>
#include <endian.h>
#include <inttypes.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#ifdef be32toh
/* If libc provides [bl]e{32,64}toh() then we'll use them */
#elif BYTE_ORDER == LITTLE_ENDIAN
# define be32toh(x)	bswap_32(x)
# define le32toh(x)	(x)
# define be64toh(x)	bswap_64(x)
# define le64toh(x)	(x)
#elif BYTE_ORDER == BIG_ENDIAN
# define be32toh(x)	(x)
# define le32toh(x)	bswap_32(x)
# define be64toh(x)	(x)
# define le64toh(x)	bswap_64(x)
#endif

__attribute__((noreturn))
static void die(const char *msg)
{
	fputs(msg, stderr);
	exit(EXIT_FAILURE);
}

int main(int argc, const char *argv[])
{
	uint64_t entry;
	size_t nread;
	FILE *file;
	union {
		Elf32_Ehdr ehdr32;
		Elf64_Ehdr ehdr64;
	} hdr;

	if (argc != 2)
		die("Usage: elf-entry <elf-file>\n");

	file = fopen(argv[1], "r");
	if (!file) {
		perror("Unable to open input file");
		return EXIT_FAILURE;
	}

	nread = fread(&hdr, 1, sizeof(hdr), file);
	if (nread != sizeof(hdr)) {
		perror("Unable to read input file");
		fclose(file);
		return EXIT_FAILURE;
	}

	if (memcmp(hdr.ehdr32.e_ident, ELFMAG, SELFMAG)) {
		fclose(file);
		die("Input is not an ELF\n");
	}

	switch (hdr.ehdr32.e_ident[EI_CLASS]) {
	case ELFCLASS32:
		switch (hdr.ehdr32.e_ident[EI_DATA]) {
		case ELFDATA2LSB:
			entry = le32toh(hdr.ehdr32.e_entry);
			break;
		case ELFDATA2MSB:
			entry = be32toh(hdr.ehdr32.e_entry);
			break;
		default:
			fclose(file);
			die("Invalid ELF encoding\n");
		}

		/* Sign extend to form a canonical address */
		entry = (int64_t)(int32_t)entry;
		break;

	case ELFCLASS64:
		switch (hdr.ehdr32.e_ident[EI_DATA]) {
		case ELFDATA2LSB:
			entry = le64toh(hdr.ehdr64.e_entry);
			break;
		case ELFDATA2MSB:
			entry = be64toh(hdr.ehdr64.e_entry);
			break;
		default:
			fclose(file);
			die("Invalid ELF encoding\n");
		}
		break;

	default:
		fclose(file);
		die("Invalid ELF class\n");
	}

	printf("0x%016" PRIx64 "\n", entry);
	fclose(file);
	return EXIT_SUCCESS;
}