riscv: tools: Add big endian target support to prelink-riscv

Signed-off-by: Marcus Comstedt <marcus@mc.pp.se>
Cc: Rick Chen <rick@andestech.com>
Reviewed-by: Rick Chen <rick@andestech.com>
This commit is contained in:
Marcus Comstedt 2019-08-02 19:45:16 +02:00 committed by Andes
parent e604410d3e
commit 4539926a9c
2 changed files with 62 additions and 32 deletions

View File

@ -47,12 +47,28 @@ const char *argv0;
exit(EXIT_FAILURE); \ exit(EXIT_FAILURE); \
} while (0) } while (0)
#define PRELINK_BYTEORDER le
#define PRELINK_INC_BITS 32 #define PRELINK_INC_BITS 32
#include "prelink-riscv.inc" #include "prelink-riscv.inc"
#undef PRELINK_BYTEORDER
#undef PRELINK_INC_BITS #undef PRELINK_INC_BITS
#define PRELINK_BYTEORDER le
#define PRELINK_INC_BITS 64 #define PRELINK_INC_BITS 64
#include "prelink-riscv.inc" #include "prelink-riscv.inc"
#undef PRELINK_BYTEORDER
#undef PRELINK_INC_BITS
#define PRELINK_BYTEORDER be
#define PRELINK_INC_BITS 32
#include "prelink-riscv.inc"
#undef PRELINK_BYTEORDER
#undef PRELINK_INC_BITS
#define PRELINK_BYTEORDER be
#define PRELINK_INC_BITS 64
#include "prelink-riscv.inc"
#undef PRELINK_BYTEORDER
#undef PRELINK_INC_BITS #undef PRELINK_INC_BITS
int main(int argc, const char *const *argv) int main(int argc, const char *const *argv)
@ -88,11 +104,19 @@ int main(int argc, const char *const *argv)
die("Invalid ELF file %s", argv[1]); die("Invalid ELF file %s", argv[1]);
bool is64 = e_ident[EI_CLASS] == ELFCLASS64; bool is64 = e_ident[EI_CLASS] == ELFCLASS64;
bool isbe = e_ident[EI_DATA] == ELFDATA2MSB;
if (is64) if (is64) {
prelink64(data); if (isbe)
prelink_be64(data);
else else
prelink32(data); prelink_le64(data);
} else {
if (isbe)
prelink_be32(data);
else
prelink_le32(data);
}
return 0; return 0;
} }

View File

@ -12,9 +12,9 @@
#define CONCAT(x, y) CONCAT_IMPL(x, y) #define CONCAT(x, y) CONCAT_IMPL(x, y)
#define CONCAT3(x, y, z) CONCAT(CONCAT(x, y), z) #define CONCAT3(x, y, z) CONCAT(CONCAT(x, y), z)
#define prelink_nn CONCAT(prelink, PRELINK_INC_BITS) #define prelink_bonn CONCAT3(prelink_, PRELINK_BYTEORDER, PRELINK_INC_BITS)
#define uintnn_t CONCAT3(uint, PRELINK_INC_BITS, _t) #define uintnn_t CONCAT3(uint, PRELINK_INC_BITS, _t)
#define get_offset_nn CONCAT(get_offset_, PRELINK_INC_BITS) #define get_offset_bonn CONCAT3(get_offset_, PRELINK_BYTEORDER, PRELINK_INC_BITS)
#define Elf_Ehdr CONCAT3(Elf, PRELINK_INC_BITS, _Ehdr) #define Elf_Ehdr CONCAT3(Elf, PRELINK_INC_BITS, _Ehdr)
#define Elf_Phdr CONCAT3(Elf, PRELINK_INC_BITS, _Phdr) #define Elf_Phdr CONCAT3(Elf, PRELINK_INC_BITS, _Phdr)
#define Elf_Rela CONCAT3(Elf, PRELINK_INC_BITS, _Rela) #define Elf_Rela CONCAT3(Elf, PRELINK_INC_BITS, _Rela)
@ -23,35 +23,38 @@
#define Elf_Addr CONCAT3(Elf, PRELINK_INC_BITS, _Addr) #define Elf_Addr CONCAT3(Elf, PRELINK_INC_BITS, _Addr)
#define ELF_R_TYPE CONCAT3(ELF, PRELINK_INC_BITS, _R_TYPE) #define ELF_R_TYPE CONCAT3(ELF, PRELINK_INC_BITS, _R_TYPE)
#define ELF_R_SYM CONCAT3(ELF, PRELINK_INC_BITS, _R_SYM) #define ELF_R_SYM CONCAT3(ELF, PRELINK_INC_BITS, _R_SYM)
#define lenn_to_cpu CONCAT3(le, PRELINK_INC_BITS, _to_cpu) #define target16_to_cpu CONCAT(PRELINK_BYTEORDER, 16_to_cpu)
#define target32_to_cpu CONCAT(PRELINK_BYTEORDER, 32_to_cpu)
#define target64_to_cpu CONCAT(PRELINK_BYTEORDER, 64_to_cpu)
#define targetnn_to_cpu CONCAT3(PRELINK_BYTEORDER, PRELINK_INC_BITS, _to_cpu)
static void* get_offset_nn (void* data, Elf_Phdr* phdrs, size_t phnum, Elf_Addr addr) static void* get_offset_bonn (void* data, Elf_Phdr* phdrs, size_t phnum, Elf_Addr addr)
{ {
Elf_Phdr *p; Elf_Phdr *p;
for (p = phdrs; p < phdrs + phnum; ++p) for (p = phdrs; p < phdrs + phnum; ++p)
if (lenn_to_cpu(p->p_vaddr) <= addr && lenn_to_cpu(p->p_vaddr) + lenn_to_cpu(p->p_memsz) > addr) if (targetnn_to_cpu(p->p_vaddr) <= addr && targetnn_to_cpu(p->p_vaddr) + targetnn_to_cpu(p->p_memsz) > addr)
return data + lenn_to_cpu(p->p_offset) + (addr - lenn_to_cpu(p->p_vaddr)); return data + targetnn_to_cpu(p->p_offset) + (addr - targetnn_to_cpu(p->p_vaddr));
return NULL; return NULL;
} }
static void prelink_nn(void *data) static void prelink_bonn(void *data)
{ {
Elf_Ehdr *ehdr = data; Elf_Ehdr *ehdr = data;
Elf_Phdr *p; Elf_Phdr *p;
Elf_Dyn *dyn; Elf_Dyn *dyn;
Elf_Rela *r; Elf_Rela *r;
if (le16_to_cpu(ehdr->e_machine) != EM_RISCV) if (target16_to_cpu(ehdr->e_machine) != EM_RISCV)
die("Machine type is not RISC-V"); die("Machine type is not RISC-V");
Elf_Phdr *phdrs = data + lenn_to_cpu(ehdr->e_phoff); Elf_Phdr *phdrs = data + targetnn_to_cpu(ehdr->e_phoff);
Elf_Dyn *dyns = NULL; Elf_Dyn *dyns = NULL;
for (p = phdrs; p < phdrs + le16_to_cpu(ehdr->e_phnum); ++p) { for (p = phdrs; p < phdrs + target16_to_cpu(ehdr->e_phnum); ++p) {
if (le32_to_cpu(p->p_type) == PT_DYNAMIC) { if (target32_to_cpu(p->p_type) == PT_DYNAMIC) {
dyns = data + lenn_to_cpu(p->p_offset); dyns = data + targetnn_to_cpu(p->p_offset);
break; break;
} }
} }
@ -63,14 +66,14 @@ static void prelink_nn(void *data)
size_t rela_count = 0; size_t rela_count = 0;
Elf_Sym *dynsym = NULL; Elf_Sym *dynsym = NULL;
for (dyn = dyns;; ++dyn) { for (dyn = dyns;; ++dyn) {
if (lenn_to_cpu(dyn->d_tag) == DT_NULL) if (targetnn_to_cpu(dyn->d_tag) == DT_NULL)
break; break;
else if (lenn_to_cpu(dyn->d_tag) == DT_RELA) else if (targetnn_to_cpu(dyn->d_tag) == DT_RELA)
rela_dyn = get_offset_nn(data, phdrs, le16_to_cpu(ehdr->e_phnum), + lenn_to_cpu(dyn->d_un.d_ptr)); rela_dyn = get_offset_bonn(data, phdrs, target16_to_cpu(ehdr->e_phnum), + targetnn_to_cpu(dyn->d_un.d_ptr));
else if (lenn_to_cpu(dyn->d_tag) == DT_RELASZ) else if (targetnn_to_cpu(dyn->d_tag) == DT_RELASZ)
rela_count = lenn_to_cpu(dyn->d_un.d_val) / sizeof(Elf_Rela); rela_count = targetnn_to_cpu(dyn->d_un.d_val) / sizeof(Elf_Rela);
else if (lenn_to_cpu(dyn->d_tag) == DT_SYMTAB) else if (targetnn_to_cpu(dyn->d_tag) == DT_SYMTAB)
dynsym = get_offset_nn(data, phdrs, le16_to_cpu(ehdr->e_phnum), + lenn_to_cpu(dyn->d_un.d_ptr)); dynsym = get_offset_bonn(data, phdrs, target16_to_cpu(ehdr->e_phnum), + targetnn_to_cpu(dyn->d_un.d_ptr));
} }
@ -81,23 +84,23 @@ static void prelink_nn(void *data)
die("No .dynsym found"); die("No .dynsym found");
for (r = rela_dyn; r < rela_dyn + rela_count; ++r) { for (r = rela_dyn; r < rela_dyn + rela_count; ++r) {
void* buf = get_offset_nn(data, phdrs, le16_to_cpu(ehdr->e_phnum), lenn_to_cpu(r->r_offset)); void* buf = get_offset_bonn(data, phdrs, target16_to_cpu(ehdr->e_phnum), targetnn_to_cpu(r->r_offset));
if (buf == NULL) if (buf == NULL)
continue; continue;
if (ELF_R_TYPE(lenn_to_cpu(r->r_info)) == R_RISCV_RELATIVE) if (ELF_R_TYPE(targetnn_to_cpu(r->r_info)) == R_RISCV_RELATIVE)
*((uintnn_t*) buf) = r->r_addend; *((uintnn_t*) buf) = r->r_addend;
else if (ELF_R_TYPE(lenn_to_cpu(r->r_info)) == R_RISCV_32) else if (ELF_R_TYPE(targetnn_to_cpu(r->r_info)) == R_RISCV_32)
*((uint32_t*) buf) = dynsym[ELF_R_SYM(lenn_to_cpu(r->r_info))].st_value; *((uint32_t*) buf) = dynsym[ELF_R_SYM(targetnn_to_cpu(r->r_info))].st_value;
else if (ELF_R_TYPE(lenn_to_cpu(r->r_info)) == R_RISCV_64) else if (ELF_R_TYPE(targetnn_to_cpu(r->r_info)) == R_RISCV_64)
*((uint64_t*) buf) = dynsym[ELF_R_SYM(lenn_to_cpu(r->r_info))].st_value; *((uint64_t*) buf) = dynsym[ELF_R_SYM(targetnn_to_cpu(r->r_info))].st_value;
} }
} }
#undef prelink_nn #undef prelink_bonn
#undef uintnn_t #undef uintnn_t
#undef get_offset_nn #undef get_offset_bonn
#undef Elf_Ehdr #undef Elf_Ehdr
#undef Elf_Phdr #undef Elf_Phdr
#undef Elf_Rela #undef Elf_Rela
@ -106,7 +109,10 @@ static void prelink_nn(void *data)
#undef Elf_Addr #undef Elf_Addr
#undef ELF_R_TYPE #undef ELF_R_TYPE
#undef ELF_R_SYM #undef ELF_R_SYM
#undef lenn_to_cpu #undef target16_to_cpu
#undef target32_to_cpu
#undef target64_to_cpu
#undef targetnn_to_cpu
#undef CONCAT_IMPL #undef CONCAT_IMPL
#undef CONCAT #undef CONCAT