1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * Copyright (C) 2017 Andes Technology
4 * Chih-Mao Chen <cmchen@andestech.com>
5 *
6 * Statically process runtime relocations on RISC-V ELF images
7 * so that it can be directly executed when loaded at LMA
8 * without fixup. Both RV32 and RV64 are supported.
9 */
10
11 #include <errno.h>
12 #include <stdbool.h>
13 #include <stdint.h>
14 #include <stdio.h>
15 #include <stdlib.h>
16 #include <string.h>
17
18 #include <elf.h>
19 #include <fcntl.h>
20 #include <sys/mman.h>
21 #include <sys/stat.h>
22 #include <sys/types.h>
23 #include <unistd.h>
24 #include <compiler.h>
25
26 #ifndef EM_RISCV
27 #define EM_RISCV 243
28 #endif
29
30 #ifndef R_RISCV_32
31 #define R_RISCV_32 1
32 #endif
33
34 #ifndef R_RISCV_64
35 #define R_RISCV_64 2
36 #endif
37
38 #ifndef R_RISCV_RELATIVE
39 #define R_RISCV_RELATIVE 3
40 #endif
41
42 const char *argv0;
43
44 #define die(fmt, ...) \
45 do { \
46 fprintf(stderr, "%s: " fmt "\n", argv0, ## __VA_ARGS__); \
47 exit(EXIT_FAILURE); \
48 } while (0)
49
50 #define PRELINK_BYTEORDER le
51 #define PRELINK_INC_BITS 32
52 #include "prelink-riscv.inc"
53 #undef PRELINK_BYTEORDER
54 #undef PRELINK_INC_BITS
55
56 #define PRELINK_BYTEORDER le
57 #define PRELINK_INC_BITS 64
58 #include "prelink-riscv.inc"
59 #undef PRELINK_BYTEORDER
60 #undef PRELINK_INC_BITS
61
62 #define PRELINK_BYTEORDER be
63 #define PRELINK_INC_BITS 32
64 #include "prelink-riscv.inc"
65 #undef PRELINK_BYTEORDER
66 #undef PRELINK_INC_BITS
67
68 #define PRELINK_BYTEORDER be
69 #define PRELINK_INC_BITS 64
70 #include "prelink-riscv.inc"
71 #undef PRELINK_BYTEORDER
72 #undef PRELINK_INC_BITS
73
main(int argc,const char * const * argv)74 int main(int argc, const char *const *argv)
75 {
76 argv0 = argv[0];
77
78 if (argc < 2) {
79 fprintf(stderr, "Usage: %s <u-boot>\n", argv0);
80 exit(EXIT_FAILURE);
81 }
82
83 int fd = open(argv[1], O_RDWR, 0);
84
85 if (fd < 0)
86 die("Cannot open %s: %s", argv[1], strerror(errno));
87
88 struct stat st;
89
90 if (fstat(fd, &st) < 0)
91 die("Cannot stat %s: %s", argv[1], strerror(errno));
92
93 void *data =
94 mmap(0, st.st_size, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
95
96 if (data == MAP_FAILED)
97 die("Cannot mmap %s: %s", argv[1], strerror(errno));
98
99 close(fd);
100
101 unsigned char *e_ident = (unsigned char *)data;
102
103 if (memcmp(e_ident, ELFMAG, SELFMAG) != 0)
104 die("Invalid ELF file %s", argv[1]);
105
106 bool is64 = e_ident[EI_CLASS] == ELFCLASS64;
107 bool isbe = e_ident[EI_DATA] == ELFDATA2MSB;
108
109 if (is64) {
110 if (isbe)
111 prelink_be64(data);
112 else
113 prelink_le64(data);
114 } else {
115 if (isbe)
116 prelink_be32(data);
117 else
118 prelink_le32(data);
119 }
120
121 return 0;
122 }
123