1 // SPDX-License-Identifier: GPL-2.0 2 #include <byteswap.h> 3 #include <elf.h> 4 #include <endian.h> 5 #include <inttypes.h> 6 #include <stdint.h> 7 #include <stdio.h> 8 #include <stdlib.h> 9 #include <string.h> 10 11 #ifdef be32toh 12 /* If libc provides [bl]e{32,64}toh() then we'll use them */ 13 #elif BYTE_ORDER == LITTLE_ENDIAN 14 # define be32toh(x) bswap_32(x) 15 # define le32toh(x) (x) 16 # define be64toh(x) bswap_64(x) 17 # define le64toh(x) (x) 18 #elif BYTE_ORDER == BIG_ENDIAN 19 # define be32toh(x) (x) 20 # define le32toh(x) bswap_32(x) 21 # define be64toh(x) (x) 22 # define le64toh(x) bswap_64(x) 23 #endif 24 25 __attribute__((noreturn)) 26 static void die(const char *msg) 27 { 28 fputs(msg, stderr); 29 exit(EXIT_FAILURE); 30 } 31 32 int main(int argc, const char *argv[]) 33 { 34 uint64_t entry; 35 size_t nread; 36 FILE *file; 37 union { 38 Elf32_Ehdr ehdr32; 39 Elf64_Ehdr ehdr64; 40 } hdr; 41 42 if (argc != 2) 43 die("Usage: elf-entry <elf-file>\n"); 44 45 file = fopen(argv[1], "r"); 46 if (!file) { 47 perror("Unable to open input file"); 48 return EXIT_FAILURE; 49 } 50 51 nread = fread(&hdr, 1, sizeof(hdr), file); 52 if (nread != sizeof(hdr)) { 53 perror("Unable to read input file"); 54 return EXIT_FAILURE; 55 } 56 57 if (memcmp(hdr.ehdr32.e_ident, ELFMAG, SELFMAG)) 58 die("Input is not an ELF\n"); 59 60 switch (hdr.ehdr32.e_ident[EI_CLASS]) { 61 case ELFCLASS32: 62 switch (hdr.ehdr32.e_ident[EI_DATA]) { 63 case ELFDATA2LSB: 64 entry = le32toh(hdr.ehdr32.e_entry); 65 break; 66 case ELFDATA2MSB: 67 entry = be32toh(hdr.ehdr32.e_entry); 68 break; 69 default: 70 die("Invalid ELF encoding\n"); 71 } 72 73 /* Sign extend to form a canonical address */ 74 entry = (int64_t)(int32_t)entry; 75 break; 76 77 case ELFCLASS64: 78 switch (hdr.ehdr32.e_ident[EI_DATA]) { 79 case ELFDATA2LSB: 80 entry = le64toh(hdr.ehdr64.e_entry); 81 break; 82 case ELFDATA2MSB: 83 entry = be64toh(hdr.ehdr64.e_entry); 84 break; 85 default: 86 die("Invalid ELF encoding\n"); 87 } 88 break; 89 90 default: 91 die("Invalid ELF class\n"); 92 } 93 94 printf("0x%016" PRIx64 "\n", entry); 95 return EXIT_SUCCESS; 96 } 97