libdwarf_reloc.c revision 1.6 1 /* $NetBSD: libdwarf_reloc.c,v 1.6 2024/03/27 21:53:06 christos Exp $ */
2
3 /*-
4 * Copyright (c) 2010 Kai Wang
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include "_libdwarf.h"
30
31 __RCSID("$NetBSD: libdwarf_reloc.c,v 1.6 2024/03/27 21:53:06 christos Exp $");
32 ELFTC_VCSID("Id: libdwarf_reloc.c 3741 2019-06-07 06:32:01Z jkoshy");
33
34 Dwarf_Unsigned
35 _dwarf_get_reloc_type(Dwarf_P_Debug dbg, int is64)
36 {
37
38 assert(dbg != NULL);
39 #ifdef BUILTIN_ELF_HEADERS
40 switch (dbg->dbgp_isa) {
41 case DW_ISA_AARCH64:
42 return (is64 ? R_AARCH64_ABS64 : R_AARCH64_ABS32);
43 case DW_ISA_X86:
44 return (R_386_32);
45 case DW_ISA_X86_64:
46 return (is64 ? R_X86_64_64 : R_X86_64_32);
47 case DW_ISA_SPARC:
48 return (is64 ? R_SPARC_UA64 : R_SPARC_UA32);
49 case DW_ISA_PPC:
50 return (is64 ? R_PPC64_ADDR64 : R_PPC_ADDR32);
51 case DW_ISA_ARM:
52 return (R_ARM_ABS32);
53 case DW_ISA_MIPS:
54 return (is64 ? R_MIPS_64 : R_MIPS_32);
55 case DW_ISA_RISCV:
56 return (is64 ? R_RISCV_64 : R_RISCV_32);
57 case DW_ISA_IA64:
58 return (is64 ? R_IA_64_DIR64LSB : R_IA_64_DIR32LSB);
59 default:
60 break;
61 }
62 #endif
63 return (0); /* NOT REACHED */
64 }
65
66 int
67 _dwarf_get_reloc_size(Dwarf_Debug dbg, Dwarf_Unsigned rel_type)
68 {
69
70 #ifdef BUILTIN_ELF_HEADERS
71 switch (dbg->dbg_machine) {
72 case EM_NONE:
73 break;
74 case EM_AARCH64:
75 if (rel_type == R_AARCH64_ABS32)
76 return (4);
77 else if (rel_type == R_AARCH64_ABS64)
78 return (8);
79 break;
80 case EM_ARM:
81 if (rel_type == R_ARM_ABS32)
82 return (4);
83 break;
84 case EM_386:
85 case EM_IAMCU:
86 if (rel_type == R_386_32)
87 return (4);
88 break;
89 case EM_X86_64:
90 if (rel_type == R_X86_64_32)
91 return (4);
92 else if (rel_type == R_X86_64_64)
93 return (8);
94 break;
95 case EM_SPARC:
96 if (rel_type == R_SPARC_UA32)
97 return (4);
98 else if (rel_type == R_SPARC_UA64)
99 return (8);
100 break;
101 case EM_PPC:
102 if (rel_type == R_PPC_ADDR32)
103 return (4);
104 break;
105 case EM_PPC64:
106 if (rel_type == R_PPC_ADDR32)
107 return (4);
108 else if (rel_type == R_PPC64_ADDR64)
109 return (8);
110 break;
111 case EM_MIPS:
112 if (rel_type == R_MIPS_32)
113 return (4);
114 else if (rel_type == R_MIPS_64)
115 return (8);
116 break;
117 case EM_RISCV:
118 if (rel_type == R_RISCV_32)
119 return (4);
120 else if (rel_type == R_RISCV_64)
121 return (8);
122 break;
123 case EM_IA_64:
124 if (rel_type == R_IA_64_SECREL32LSB)
125 return (4);
126 else if (rel_type == R_IA_64_DIR64LSB)
127 return (8);
128 break;
129 default:
130 break;
131 }
132 #endif
133
134 /* unknown relocation. */
135 return (0);
136 }
137
138 int
139 _dwarf_reloc_section_init(Dwarf_P_Debug dbg, Dwarf_Rel_Section *drsp,
140 Dwarf_P_Section ref, Dwarf_Error *error)
141 {
142 Dwarf_Rel_Section drs;
143 char name[128];
144 int pseudo;
145
146 assert(dbg != NULL && drsp != NULL && ref != NULL);
147
148 if ((drs = calloc(1, sizeof(struct _Dwarf_Rel_Section))) == NULL) {
149 DWARF_SET_ERROR(dbg, error, DW_DLE_MEMORY);
150 return (DW_DLE_MEMORY);
151 }
152
153 drs->drs_ref = ref;
154
155 /*
156 * FIXME The logic here is most likely wrong. It should
157 * be the ISA that determines relocation type.
158 */
159 if (dbg->dbgp_flags & DW_DLC_SIZE_64)
160 drs->drs_addend = 1;
161 else
162 drs->drs_addend = 0;
163
164 if (dbg->dbgp_flags & DW_DLC_SYMBOLIC_RELOCATIONS)
165 pseudo = 1;
166 else
167 pseudo = 0;
168
169 snprintf(name, sizeof(name), "%s%s",
170 drs->drs_addend ? ".rela" : ".rel", ref->ds_name);
171 if (_dwarf_section_init(dbg, &drs->drs_ds, name, pseudo, error) !=
172 DW_DLE_NONE) {
173 free(drs);
174 DWARF_SET_ERROR(dbg, error, DW_DLE_MEMORY);
175 return (DW_DLE_MEMORY);
176 }
177
178 STAILQ_INIT(&drs->drs_dre);
179 STAILQ_INSERT_TAIL(&dbg->dbgp_drslist, drs, drs_next);
180 dbg->dbgp_drscnt++;
181 *drsp = drs;
182
183 return (DW_DLE_NONE);
184 }
185
186 void
187 _dwarf_reloc_section_free(Dwarf_P_Debug dbg, Dwarf_Rel_Section *drsp)
188 {
189 Dwarf_Rel_Section drs, tdrs;
190 Dwarf_Rel_Entry dre, tdre;
191
192 assert(dbg != NULL && drsp != NULL);
193
194 if (*drsp == NULL)
195 return;
196
197 STAILQ_FOREACH_SAFE(drs, &dbg->dbgp_drslist, drs_next, tdrs) {
198 if (drs != *drsp)
199 continue;
200 STAILQ_REMOVE(&dbg->dbgp_drslist, drs, _Dwarf_Rel_Section,
201 drs_next);
202 STAILQ_FOREACH_SAFE(dre, &drs->drs_dre, dre_next, tdre) {
203 STAILQ_REMOVE(&drs->drs_dre, dre, _Dwarf_Rel_Entry,
204 dre_next);
205 free(dre);
206 }
207 if ((dbg->dbgp_flags & DW_DLC_SYMBOLIC_RELOCATIONS) == 0)
208 _dwarf_section_free(dbg, &drs->drs_ds);
209 else {
210 if (drs->drs_ds->ds_name)
211 free(drs->drs_ds->ds_name);
212 free(drs->drs_ds);
213 }
214 free(drs);
215 *drsp = NULL;
216 dbg->dbgp_drscnt--;
217 break;
218 }
219 }
220
221 int
222 _dwarf_reloc_entry_add(Dwarf_P_Debug dbg, Dwarf_Rel_Section drs,
223 Dwarf_P_Section ds, unsigned char type, unsigned char length,
224 Dwarf_Unsigned offset, Dwarf_Unsigned symndx, Dwarf_Unsigned addend,
225 const char *secname, Dwarf_Error *error)
226 {
227 Dwarf_Rel_Entry dre;
228 Dwarf_Unsigned reloff;
229 int ret;
230
231 assert(drs != NULL);
232 assert(offset <= ds->ds_size);
233 reloff = offset;
234
235 /*
236 * If the DW_DLC_SYMBOLIC_RELOCATIONS flag is set or ElfXX_Rel
237 * is used instead of ELfXX_Rela, we need to write the addend
238 * in the storage unit to be relocated. Otherwise write 0 in the
239 * storage unit and the addend will be written into relocation
240 * section later.
241 */
242 if ((dbg->dbgp_flags & DW_DLC_SYMBOLIC_RELOCATIONS) ||
243 drs->drs_addend == 0)
244 ret = dbg->write_alloc(&ds->ds_data, &ds->ds_cap, &offset,
245 addend, length, error);
246 else
247 ret = dbg->write_alloc(&ds->ds_data, &ds->ds_cap, &offset,
248 0, length, error);
249 if (ret != DW_DLE_NONE)
250 return (ret);
251 if (offset > ds->ds_size)
252 ds->ds_size = offset;
253
254 if ((dre = calloc(1, sizeof(struct _Dwarf_Rel_Entry))) == NULL) {
255 DWARF_SET_ERROR(dbg, error, DW_DLE_MEMORY);
256 return (DW_DLE_MEMORY);
257 }
258 STAILQ_INSERT_TAIL(&drs->drs_dre, dre, dre_next);
259 dre->dre_type = type;
260 dre->dre_length = length;
261 dre->dre_offset = reloff;
262 dre->dre_symndx = symndx;
263 dre->dre_addend = addend;
264 dre->dre_secname = secname;
265 drs->drs_drecnt++;
266
267 return (DW_DLE_NONE);
268 }
269
270 int
271 _dwarf_reloc_entry_add_pair(Dwarf_P_Debug dbg, Dwarf_Rel_Section drs,
272 Dwarf_P_Section ds, unsigned char length, Dwarf_Unsigned offset,
273 Dwarf_Unsigned symndx, Dwarf_Unsigned esymndx, Dwarf_Unsigned symoff,
274 Dwarf_Unsigned esymoff, Dwarf_Error *error)
275 {
276 Dwarf_Rel_Entry dre;
277 Dwarf_Unsigned reloff;
278 int ret;
279
280 assert(drs != NULL);
281 assert(offset <= ds->ds_size);
282 assert(dbg->dbgp_flags & DW_DLC_SYMBOLIC_RELOCATIONS);
283 reloff = offset;
284
285 /* Write net offset into section stream. */
286 ret = dbg->write_alloc(&ds->ds_data, &ds->ds_cap, &offset,
287 esymoff - symoff, length, error);
288 if (ret != DW_DLE_NONE)
289 return (ret);
290 if (offset > ds->ds_size)
291 ds->ds_size = offset;
292
293 if ((dre = calloc(2, sizeof(struct _Dwarf_Rel_Entry))) == NULL) {
294 DWARF_SET_ERROR(dbg, error, DW_DLE_MEMORY);
295 return (DW_DLE_MEMORY);
296 }
297 STAILQ_INSERT_TAIL(&drs->drs_dre, &dre[0], dre_next);
298 STAILQ_INSERT_TAIL(&drs->drs_dre, &dre[1], dre_next);
299 dre[0].dre_type = dwarf_drt_first_of_length_pair;
300 dre[0].dre_length = length;
301 dre[0].dre_offset = reloff;
302 dre[0].dre_symndx = symndx;
303 dre[0].dre_addend = 0;
304 dre[0].dre_secname = NULL;
305 dre[1].dre_type = dwarf_drt_second_of_length_pair;
306 dre[1].dre_length = length;
307 dre[1].dre_offset = reloff;
308 dre[1].dre_symndx = esymndx;
309 dre[1].dre_addend = 0;
310 dre[1].dre_secname = NULL;
311 drs->drs_drecnt += 2;
312
313 return (DW_DLE_NONE);
314 }
315
316 int
317 _dwarf_reloc_section_finalize(Dwarf_P_Debug dbg, Dwarf_Rel_Section drs,
318 Dwarf_Error *error)
319 {
320 Dwarf_P_Section ds;
321 Dwarf_Unsigned unit;
322 int ret, size;
323
324 assert(dbg != NULL && drs != NULL && drs->drs_ds != NULL &&
325 drs->drs_ref != NULL);
326
327 ds = drs->drs_ds;
328
329 /*
330 * Calculate the size (in bytes) of the relocation section.
331 */
332 if (dbg->dbgp_flags & DW_DLC_SIZE_64)
333 unit = drs->drs_addend ? sizeof(Elf64_Rela) : sizeof(Elf64_Rel);
334 else
335 unit = drs->drs_addend ? sizeof(Elf32_Rela) : sizeof(Elf32_Rel);
336 assert(ds->ds_size == 0);
337 size = drs->drs_drecnt * unit;
338
339 /*
340 * Discard this relocation section if there is no entry in it.
341 */
342 if (size == 0) {
343 _dwarf_reloc_section_free(dbg, &drs);
344 return (DW_DLE_NONE);
345 }
346
347 /*
348 * If we are under stream mode, realloc the section data block to
349 * this size.
350 */
351 if ((dbg->dbgp_flags & DW_DLC_SYMBOLIC_RELOCATIONS) == 0) {
352 ds->ds_cap = size;
353 if ((ds->ds_data = realloc(ds->ds_data, (size_t) ds->ds_cap)) ==
354 NULL) {
355 DWARF_SET_ERROR(dbg, error, DW_DLE_MEMORY);
356 return (DW_DLE_MEMORY);
357 }
358 }
359
360 /*
361 * Notify the application the creation of this relocation section.
362 * Note that the section link here should point to the .symtab
363 * section, we set it to 0 since we have no way to know .symtab
364 * section index.
365 */
366 ret = _dwarf_pro_callback(dbg, ds->ds_name, size,
367 drs->drs_addend ? SHT_RELA : SHT_REL, 0, 0, drs->drs_ref->ds_ndx,
368 &ds->ds_symndx, NULL);
369 if (ret < 0) {
370 DWARF_SET_ERROR(dbg, error, DW_DLE_ELF_SECT_ERR);
371 return (DW_DLE_ELF_SECT_ERR);
372 }
373 ds->ds_ndx = ret;
374
375 return (DW_DLE_NONE);
376 }
377
378 int
379 _dwarf_reloc_section_gen(Dwarf_P_Debug dbg, Dwarf_Rel_Section drs,
380 Dwarf_Error *error)
381 {
382 Dwarf_Rel_Entry dre;
383 Dwarf_P_Section ds;
384 Dwarf_Unsigned type;
385 int ret;
386
387 assert((dbg->dbgp_flags & DW_DLC_SYMBOLIC_RELOCATIONS) == 0);
388 assert(drs->drs_ds != NULL && drs->drs_ds->ds_size == 0);
389 assert(!STAILQ_EMPTY(&drs->drs_dre));
390 ds = drs->drs_ds;
391
392 STAILQ_FOREACH(dre, &drs->drs_dre, dre_next) {
393 assert(dre->dre_length == 4 || dre->dre_length == 8);
394 type = _dwarf_get_reloc_type(dbg, dre->dre_length == 8);
395 if (dbg->dbgp_flags & DW_DLC_SIZE_64) {
396 /* Write r_offset (8 bytes) */
397 ret = dbg->write_alloc(&ds->ds_data, &ds->ds_cap,
398 &ds->ds_size, dre->dre_offset, 8, error);
399 if (ret != DW_DLE_NONE)
400 return (ret);
401 /* Write r_info (8 bytes) */
402 ret = dbg->write_alloc(&ds->ds_data, &ds->ds_cap,
403 &ds->ds_size, ELF64_R_INFO(dre->dre_symndx, type),
404 8, error);
405 if (ret != DW_DLE_NONE)
406 return (ret);
407 /* Write r_addend (8 bytes) */
408 if (drs->drs_addend) {
409 ret = dbg->write_alloc(&ds->ds_data,
410 &ds->ds_cap, &ds->ds_size, dre->dre_addend,
411 8, error);
412 if (ret != DW_DLE_NONE)
413 return (ret);
414 }
415 } else {
416 /* Write r_offset (4 bytes) */
417 ret = dbg->write_alloc(&ds->ds_data, &ds->ds_cap,
418 &ds->ds_size, dre->dre_offset, 4, error);
419 if (ret != DW_DLE_NONE)
420 return (ret);
421 /* Write r_info (4 bytes) */
422 ret = dbg->write_alloc(&ds->ds_data, &ds->ds_cap,
423 &ds->ds_size, ELF32_R_INFO(dre->dre_symndx, type),
424 4, error);
425 if (ret != DW_DLE_NONE)
426 return (ret);
427 /* Write r_addend (4 bytes) */
428 if (drs->drs_addend) {
429 ret = dbg->write_alloc(&ds->ds_data,
430 &ds->ds_cap, &ds->ds_size, dre->dre_addend,
431 4, error);
432 if (ret != DW_DLE_NONE)
433 return (ret);
434 }
435 }
436 }
437 assert(ds->ds_size == ds->ds_cap);
438
439 return (DW_DLE_NONE);
440 }
441
442 int
443 _dwarf_reloc_gen(Dwarf_P_Debug dbg, Dwarf_Error *error)
444 {
445 Dwarf_Rel_Section drs;
446 Dwarf_Rel_Entry dre;
447 Dwarf_P_Section ds;
448 int ret;
449
450 STAILQ_FOREACH(drs, &dbg->dbgp_drslist, drs_next) {
451 /*
452 * Update relocation entries: translate any section name
453 * reference to section symbol index.
454 */
455 STAILQ_FOREACH(dre, &drs->drs_dre, dre_next) {
456 if (dre->dre_secname == NULL)
457 continue;
458 ds = _dwarf_pro_find_section(dbg, dre->dre_secname);
459 assert(ds != NULL && ds->ds_symndx != 0);
460 dre->dre_symndx = ds->ds_symndx;
461 }
462
463 /*
464 * Generate ELF relocation section if we are under stream
465 * mode.
466 */
467 if ((dbg->dbgp_flags & DW_DLC_SYMBOLIC_RELOCATIONS) == 0) {
468 ret = _dwarf_reloc_section_gen(dbg, drs, error);
469 if (ret != DW_DLE_NONE)
470 return (ret);
471 }
472 }
473
474 return (DW_DLE_NONE);
475 }
476
477 void
478 _dwarf_reloc_cleanup(Dwarf_P_Debug dbg)
479 {
480 Dwarf_Rel_Section drs, tdrs;
481 Dwarf_Rel_Entry dre, tdre;
482
483 assert(dbg != NULL && dbg->dbg_mode == DW_DLC_WRITE);
484
485 STAILQ_FOREACH_SAFE(drs, &dbg->dbgp_drslist, drs_next, tdrs) {
486 STAILQ_REMOVE(&dbg->dbgp_drslist, drs, _Dwarf_Rel_Section,
487 drs_next);
488 free(drs->drs_drd);
489 STAILQ_FOREACH_SAFE(dre, &drs->drs_dre, dre_next, tdre) {
490 STAILQ_REMOVE(&drs->drs_dre, dre, _Dwarf_Rel_Entry,
491 dre_next);
492 free(dre);
493 }
494 if (dbg->dbgp_flags & DW_DLC_SYMBOLIC_RELOCATIONS) {
495 if (drs->drs_ds) {
496 if (drs->drs_ds->ds_name)
497 free(drs->drs_ds->ds_name);
498 free(drs->drs_ds);
499 }
500 }
501 free(drs);
502 }
503 dbg->dbgp_drscnt = 0;
504 dbg->dbgp_drspos = NULL;
505 }
506