elf-entry.c (2059B)
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)) 26static void die(const char *msg) 27{ 28 fputs(msg, stderr); 29 exit(EXIT_FAILURE); 30} 31 32int 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 fclose(file); 55 return EXIT_FAILURE; 56 } 57 58 if (memcmp(hdr.ehdr32.e_ident, ELFMAG, SELFMAG)) { 59 fclose(file); 60 die("Input is not an ELF\n"); 61 } 62 63 switch (hdr.ehdr32.e_ident[EI_CLASS]) { 64 case ELFCLASS32: 65 switch (hdr.ehdr32.e_ident[EI_DATA]) { 66 case ELFDATA2LSB: 67 entry = le32toh(hdr.ehdr32.e_entry); 68 break; 69 case ELFDATA2MSB: 70 entry = be32toh(hdr.ehdr32.e_entry); 71 break; 72 default: 73 fclose(file); 74 die("Invalid ELF encoding\n"); 75 } 76 77 /* Sign extend to form a canonical address */ 78 entry = (int64_t)(int32_t)entry; 79 break; 80 81 case ELFCLASS64: 82 switch (hdr.ehdr32.e_ident[EI_DATA]) { 83 case ELFDATA2LSB: 84 entry = le64toh(hdr.ehdr64.e_entry); 85 break; 86 case ELFDATA2MSB: 87 entry = be64toh(hdr.ehdr64.e_entry); 88 break; 89 default: 90 fclose(file); 91 die("Invalid ELF encoding\n"); 92 } 93 break; 94 95 default: 96 fclose(file); 97 die("Invalid ELF class\n"); 98 } 99 100 printf("0x%016" PRIx64 "\n", entry); 101 fclose(file); 102 return EXIT_SUCCESS; 103}