Merge git://www.denx.de/git/u-boot-marvell
[platform/kernel/u-boot.git] / tools / relocate-rela.c
1 /*
2  * Copyright 2013 Freescale Semiconductor, Inc.
3  *
4  * SPDX-License-Identifier:     GPL-2.0+ BSD-2-Clause
5  *
6  * 64-bit and little-endian target only until we need to support a different
7  * arch that needs this.
8  */
9
10 #include <elf.h>
11 #include <errno.h>
12 #include <inttypes.h>
13 #include <stdarg.h>
14 #include <stdbool.h>
15 #include <stdio.h>
16 #include <stdlib.h>
17 #include <string.h>
18
19 #ifndef R_AARCH64_RELATIVE
20 #define R_AARCH64_RELATIVE      1027
21 #endif
22
23 static const bool debug_en;
24
25 static void debug(const char *fmt, ...)
26 {
27         va_list args;
28
29         va_start(args, fmt);
30         if (debug_en)
31                 vprintf(fmt, args);
32 }
33
34 static bool supported_rela(Elf64_Rela *rela)
35 {
36         uint64_t mask = 0xffffffffULL; /* would be different on 32-bit */
37         uint32_t type = rela->r_info & mask;
38
39         switch (type) {
40 #ifdef R_AARCH64_RELATIVE
41         case R_AARCH64_RELATIVE:
42                 return true;
43 #endif
44         default:
45                 fprintf(stderr, "warning: unsupported relocation type %"
46                                 PRIu32 " at %" PRIx64 "\n",
47                         type, rela->r_offset);
48
49                 return false;
50         }
51 }
52
53 static inline uint64_t swap64(uint64_t val)
54 {
55         return ((val >> 56) & 0x00000000000000ffULL) |
56                ((val >> 40) & 0x000000000000ff00ULL) |
57                ((val >> 24) & 0x0000000000ff0000ULL) |
58                ((val >>  8) & 0x00000000ff000000ULL) |
59                ((val <<  8) & 0x000000ff00000000ULL) |
60                ((val << 24) & 0x0000ff0000000000ULL) |
61                ((val << 40) & 0x00ff000000000000ULL) |
62                ((val << 56) & 0xff00000000000000ULL);
63 }
64
65 #if __BYTE_ORDER == __LITTLE_ENDIAN
66 static inline uint64_t be64(uint64_t val)
67 {
68         return swap64(val);
69 }
70
71 static inline uint64_t le64(uint64_t val)
72 {
73         return val;
74 }
75 #else
76 static inline uint64_t le64(uint64_t val)
77 {
78         return swap64(val);
79 }
80
81 static inline uint64_t be64(uint64_t val)
82 {
83         return val;
84 }
85 #endif
86
87 static bool read_num(const char *str, uint64_t *num)
88 {
89         char *endptr;
90         *num = strtoull(str, &endptr, 16);
91         return str[0] && !endptr[0];
92 }
93
94 int main(int argc, char **argv)
95 {
96         FILE *f;
97         int i, num;
98         uint64_t rela_start, rela_end, text_base;
99
100         if (argc != 5) {
101                 fprintf(stderr, "Statically apply ELF rela relocations\n");
102                 fprintf(stderr, "Usage: %s <bin file> <text base> " \
103                                 "<rela start> <rela end>\n", argv[0]);
104                 fprintf(stderr, "All numbers in hex.\n");
105                 return 1;
106         }
107
108         f = fopen(argv[1], "r+b");
109         if (!f) {
110                 fprintf(stderr, "%s: Cannot open %s: %s\n",
111                         argv[0], argv[1], strerror(errno));
112                 return 2;
113         }
114
115         if (!read_num(argv[2], &text_base) ||
116             !read_num(argv[3], &rela_start) ||
117             !read_num(argv[4], &rela_end)) {
118                 fprintf(stderr, "%s: bad number\n", argv[0]);
119                 return 3;
120         }
121
122         if (rela_start > rela_end || rela_start < text_base ||
123             (rela_end - rela_start) % sizeof(Elf64_Rela)) {
124                 fprintf(stderr, "%s: bad rela bounds\n", argv[0]);
125                 return 3;
126         }
127
128         rela_start -= text_base;
129         rela_end -= text_base;
130
131         num = (rela_end - rela_start) / sizeof(Elf64_Rela);
132
133         for (i = 0; i < num; i++) {
134                 Elf64_Rela rela, swrela;
135                 uint64_t pos = rela_start + sizeof(Elf64_Rela) * i;
136                 uint64_t addr;
137
138                 if (fseek(f, pos, SEEK_SET) < 0) {
139                         fprintf(stderr, "%s: %s: seek to %" PRIx64
140                                         " failed: %s\n",
141                                 argv[0], argv[1], pos, strerror(errno));
142                 }
143
144                 if (fread(&rela, sizeof(rela), 1, f) != 1) {
145                         fprintf(stderr, "%s: %s: read rela failed at %"
146                                         PRIx64 "\n",
147                                 argv[0], argv[1], pos);
148                         return 4;
149                 }
150
151                 swrela.r_offset = le64(rela.r_offset);
152                 swrela.r_info = le64(rela.r_info);
153                 swrela.r_addend = le64(rela.r_addend);
154
155                 if (!supported_rela(&swrela))
156                         continue;
157
158                 debug("Rela %" PRIx64 " %" PRIu64 " %" PRIx64 "\n",
159                       swrela.r_offset, swrela.r_info, swrela.r_addend);
160
161                 if (swrela.r_offset < text_base) {
162                         fprintf(stderr, "%s: %s: bad rela at %" PRIx64 "\n",
163                                 argv[0], argv[1], pos);
164                         return 4;
165                 }
166
167                 addr = swrela.r_offset - text_base;
168
169                 if (fseek(f, addr, SEEK_SET) < 0) {
170                         fprintf(stderr, "%s: %s: seek to %"
171                                         PRIx64 " failed: %s\n",
172                                 argv[0], argv[1], addr, strerror(errno));
173                 }
174
175                 if (fwrite(&rela.r_addend, sizeof(rela.r_addend), 1, f) != 1) {
176                         fprintf(stderr, "%s: %s: write failed at %" PRIx64 "\n",
177                                 argv[0], argv[1], addr);
178                         return 4;
179                 }
180         }
181
182         if (fclose(f) < 0) {
183                 fprintf(stderr, "%s: %s: close failed: %s\n",
184                         argv[0], argv[1], strerror(errno));
185                 return 4;
186         }
187
188         return 0;
189 }