blob: 57c0f655d45ce1676fe2e637c07da46b939a6312 [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001// SPDX-License-Identifier: GPL-2.0+
Rick Chen42ac26f2017-12-26 13:55:56 +08002/*
3 * Copyright (C) 2017 Andes Technology
4 * Chih-Mao Chen <cmchen@andestech.com>
5 *
Rick Chen42ac26f2017-12-26 13:55:56 +08006 * 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#define CONCAT_IMPL(x, y) x##y
12#define CONCAT(x, y) CONCAT_IMPL(x, y)
13#define CONCAT3(x, y, z) CONCAT(CONCAT(x, y), z)
14
Marcus Comstedt45399262019-08-02 19:45:16 +020015#define prelink_bonn CONCAT3(prelink_, PRELINK_BYTEORDER, PRELINK_INC_BITS)
Rick Chen42ac26f2017-12-26 13:55:56 +080016#define uintnn_t CONCAT3(uint, PRELINK_INC_BITS, _t)
Marcus Comstedt45399262019-08-02 19:45:16 +020017#define get_offset_bonn CONCAT3(get_offset_, PRELINK_BYTEORDER, PRELINK_INC_BITS)
Rick Chen42ac26f2017-12-26 13:55:56 +080018#define Elf_Ehdr CONCAT3(Elf, PRELINK_INC_BITS, _Ehdr)
19#define Elf_Phdr CONCAT3(Elf, PRELINK_INC_BITS, _Phdr)
20#define Elf_Rela CONCAT3(Elf, PRELINK_INC_BITS, _Rela)
21#define Elf_Sym CONCAT3(Elf, PRELINK_INC_BITS, _Sym)
22#define Elf_Dyn CONCAT3(Elf, PRELINK_INC_BITS, _Dyn)
23#define Elf_Addr CONCAT3(Elf, PRELINK_INC_BITS, _Addr)
24#define ELF_R_TYPE CONCAT3(ELF, PRELINK_INC_BITS, _R_TYPE)
25#define ELF_R_SYM CONCAT3(ELF, PRELINK_INC_BITS, _R_SYM)
Marcus Comstedt45399262019-08-02 19:45:16 +020026#define target16_to_cpu CONCAT(PRELINK_BYTEORDER, 16_to_cpu)
27#define target32_to_cpu CONCAT(PRELINK_BYTEORDER, 32_to_cpu)
28#define target64_to_cpu CONCAT(PRELINK_BYTEORDER, 64_to_cpu)
29#define targetnn_to_cpu CONCAT3(PRELINK_BYTEORDER, PRELINK_INC_BITS, _to_cpu)
Marcus Comstedt71bdfcb2019-08-11 11:29:57 +020030#define cpu_to_target32 CONCAT3(cpu_to_, PRELINK_BYTEORDER, 32)
31#define cpu_to_target64 CONCAT3(cpu_to_, PRELINK_BYTEORDER, 64)
Rick Chen42ac26f2017-12-26 13:55:56 +080032
Bin Mengfb3cc482023-04-13 14:20:02 +080033static void *get_offset_bonn(void *data, Elf_Phdr *phdrs, size_t phnum, Elf_Addr addr)
Rick Chen42ac26f2017-12-26 13:55:56 +080034{
35 Elf_Phdr *p;
36
37 for (p = phdrs; p < phdrs + phnum; ++p)
Marcus Comstedt45399262019-08-02 19:45:16 +020038 if (targetnn_to_cpu(p->p_vaddr) <= addr && targetnn_to_cpu(p->p_vaddr) + targetnn_to_cpu(p->p_memsz) > addr)
39 return data + targetnn_to_cpu(p->p_offset) + (addr - targetnn_to_cpu(p->p_vaddr));
Rick Chen42ac26f2017-12-26 13:55:56 +080040
41 return NULL;
42}
43
Marcus Comstedt45399262019-08-02 19:45:16 +020044static void prelink_bonn(void *data)
Rick Chen42ac26f2017-12-26 13:55:56 +080045{
46 Elf_Ehdr *ehdr = data;
47 Elf_Phdr *p;
48 Elf_Dyn *dyn;
49 Elf_Rela *r;
50
Marcus Comstedt45399262019-08-02 19:45:16 +020051 if (target16_to_cpu(ehdr->e_machine) != EM_RISCV)
Rick Chen42ac26f2017-12-26 13:55:56 +080052 die("Machine type is not RISC-V");
53
Marcus Comstedt45399262019-08-02 19:45:16 +020054 Elf_Phdr *phdrs = data + targetnn_to_cpu(ehdr->e_phoff);
Rick Chen42ac26f2017-12-26 13:55:56 +080055
56 Elf_Dyn *dyns = NULL;
Marcus Comstedt45399262019-08-02 19:45:16 +020057 for (p = phdrs; p < phdrs + target16_to_cpu(ehdr->e_phnum); ++p) {
58 if (target32_to_cpu(p->p_type) == PT_DYNAMIC) {
59 dyns = data + targetnn_to_cpu(p->p_offset);
Rick Chen42ac26f2017-12-26 13:55:56 +080060 break;
61 }
62 }
63
64 if (dyns == NULL)
65 die("No dynamic section found");
66
67 Elf_Rela *rela_dyn = NULL;
68 size_t rela_count = 0;
69 Elf_Sym *dynsym = NULL;
Bin Mengfb3cc482023-04-13 14:20:02 +080070 for (dyn = dyns; ; ++dyn) {
Marcus Comstedt45399262019-08-02 19:45:16 +020071 if (targetnn_to_cpu(dyn->d_tag) == DT_NULL)
Rick Chen42ac26f2017-12-26 13:55:56 +080072 break;
Marcus Comstedt45399262019-08-02 19:45:16 +020073 else if (targetnn_to_cpu(dyn->d_tag) == DT_RELA)
74 rela_dyn = get_offset_bonn(data, phdrs, target16_to_cpu(ehdr->e_phnum), + targetnn_to_cpu(dyn->d_un.d_ptr));
75 else if (targetnn_to_cpu(dyn->d_tag) == DT_RELASZ)
Bin Mengfb3cc482023-04-13 14:20:02 +080076 rela_count = targetnn_to_cpu(dyn->d_un.d_val) / sizeof(Elf_Rela);
Marcus Comstedt45399262019-08-02 19:45:16 +020077 else if (targetnn_to_cpu(dyn->d_tag) == DT_SYMTAB)
78 dynsym = get_offset_bonn(data, phdrs, target16_to_cpu(ehdr->e_phnum), + targetnn_to_cpu(dyn->d_un.d_ptr));
Rick Chen42ac26f2017-12-26 13:55:56 +080079
80 }
81
82 if (rela_dyn == NULL)
83 die("No .rela.dyn found");
84
85 if (dynsym == NULL)
86 die("No .dynsym found");
87
88 for (r = rela_dyn; r < rela_dyn + rela_count; ++r) {
Marcus Comstedt45399262019-08-02 19:45:16 +020089 void* buf = get_offset_bonn(data, phdrs, target16_to_cpu(ehdr->e_phnum), targetnn_to_cpu(r->r_offset));
Rick Chen42ac26f2017-12-26 13:55:56 +080090
91 if (buf == NULL)
92 continue;
93
Marcus Comstedt45399262019-08-02 19:45:16 +020094 if (ELF_R_TYPE(targetnn_to_cpu(r->r_info)) == R_RISCV_RELATIVE)
Bin Mengfb3cc482023-04-13 14:20:02 +080095 *((uintnn_t *)buf) = r->r_addend;
Marcus Comstedt45399262019-08-02 19:45:16 +020096 else if (ELF_R_TYPE(targetnn_to_cpu(r->r_info)) == R_RISCV_32)
Bin Mengfb3cc482023-04-13 14:20:02 +080097 *((uint32_t *)buf) = cpu_to_target32(targetnn_to_cpu(dynsym[ELF_R_SYM(targetnn_to_cpu(r->r_info))].st_value) + targetnn_to_cpu(r->r_addend));
Marcus Comstedt45399262019-08-02 19:45:16 +020098 else if (ELF_R_TYPE(targetnn_to_cpu(r->r_info)) == R_RISCV_64)
Bin Mengfb3cc482023-04-13 14:20:02 +080099 *((uint64_t *)buf) = cpu_to_target64(targetnn_to_cpu(dynsym[ELF_R_SYM(targetnn_to_cpu(r->r_info))].st_value) + targetnn_to_cpu(r->r_addend));
Rick Chen42ac26f2017-12-26 13:55:56 +0800100 }
101}
102
Marcus Comstedt45399262019-08-02 19:45:16 +0200103#undef prelink_bonn
Rick Chen42ac26f2017-12-26 13:55:56 +0800104#undef uintnn_t
Marcus Comstedt45399262019-08-02 19:45:16 +0200105#undef get_offset_bonn
Rick Chen42ac26f2017-12-26 13:55:56 +0800106#undef Elf_Ehdr
107#undef Elf_Phdr
108#undef Elf_Rela
109#undef Elf_Sym
110#undef Elf_Dyn
111#undef Elf_Addr
112#undef ELF_R_TYPE
113#undef ELF_R_SYM
Marcus Comstedt45399262019-08-02 19:45:16 +0200114#undef target16_to_cpu
115#undef target32_to_cpu
116#undef target64_to_cpu
117#undef targetnn_to_cpu
Marcus Comstedt71bdfcb2019-08-11 11:29:57 +0200118#undef cpu_to_target32
119#undef cpu_to_target64
Rick Chen42ac26f2017-12-26 13:55:56 +0800120
121#undef CONCAT_IMPL
122#undef CONCAT
123#undef CONCAT3