#define _DYN_LOADER
#define LDSO_ARCH_IS_RELA_
#include <sys/types.h>
#include <sys/exec_elf.h>
#include <sys/syscall.h>
#include <sys/unistd.h>
#include <machine/reloc.h>
#include "util.h"
#include "resolve.h"
int64_t pcookie __attribute__((section(".openbsd.randomdata"))) __dso_hidden;
void _dl_bind_start(void);
Elf_Addr _dl_bind(elf_object_t *object, int reloff);
#define _RF_S 0x80000000
#define _RF_A 0x40000000
#define _RF_P 0x20000000
#define _RF_G 0x10000000
#define _RF_B 0x08000000
#define _RF_E 0x02000000
#define _RF_SZ(s) (((s) & 0xff) << 8)
#define _RF_RS(s) ((s) & 0xff)
static const int reloc_target_flags[] = {
0,
_RF_S|_RF_A| _RF_SZ(32) | _RF_RS(0),
_RF_S|_RF_P|_RF_A| _RF_SZ(32) | _RF_RS(0),
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_E,
_RF_S| _RF_SZ(32) | _RF_RS(0),
_RF_S|_RF_A| _RF_SZ(32) | _RF_RS(0),
_RF_S| _RF_SZ(32) | _RF_RS(0),
_RF_A| _RF_B| _RF_SZ(32) | _RF_RS(0),
};
#define RELOC_RESOLVE_SYMBOL(t) ((reloc_target_flags[t] & _RF_S) != 0)
#define RELOC_PC_RELATIVE(t) ((reloc_target_flags[t] & _RF_P) != 0)
#define RELOC_BASE_RELATIVE(t) ((reloc_target_flags[t] & _RF_B) != 0)
#define RELOC_USE_ADDEND(t) ((reloc_target_flags[t] & _RF_A) != 0)
#define RELOC_TARGET_SIZE(t) ((reloc_target_flags[t] >> 8) & 0xff)
#define RELOC_VALUE_RIGHTSHIFT(t) (reloc_target_flags[t] & 0xff)
#define RELOC_ERROR(t) \
((t) >= nitems(reloc_target_flags) || (reloc_target_flags[t] & _RF_E))
static const int reloc_target_bitmask[] = {
#define _BM(x) (x == 32? ~0 : ~(-(1UL << (x))))
_BM(0),
_BM(32),
_BM(32),
_BM(8),
_BM(12),
_BM(8),
_BM(8),
_BM(8),
_BM(8),
_BM(8),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(0),
_BM(32),
_BM(0),
_BM(32),
#undef _BM
};
#define RELOC_VALUE_BITMASK(t) (reloc_target_bitmask[t])
#define R_TYPE(x) R_SH_ ## x
void _dl_reloc_plt(Elf_Word *where, Elf_Addr value, Elf_RelA *rel);
void
_dl_reloc_plt(Elf_Word *where, Elf_Addr value, Elf_RelA *rel)
{
*where = value + rel->r_addend;
}
int
_dl_md_reloc(elf_object_t *object, int rel, int relasz)
{
long i;
long numrela;
long relrel;
int fails = 0;
Elf_Addr loff;
Elf_Addr prev_value = 0;
const Elf_Sym *prev_sym = NULL;
Elf_RelA *rels;
loff = object->obj_base;
numrela = object->Dyn.info[relasz] / sizeof(Elf_RelA);
relrel = rel == DT_RELA ? object->relacount : 0;
rels = (Elf_RelA *)(object->Dyn.info[rel]);
if (rels == NULL)
return 0;
if (relrel > numrela)
_dl_die("relacount > numrel: %ld > %ld", relrel, numrela);
for (i = 0; i < relrel; i++, rels++) {
Elf_Addr *where;
where = (Elf_Addr *)(rels->r_offset + loff);
*where = rels->r_addend + loff;
}
for (; i < numrela; i++, rels++) {
Elf_Addr *where, value, mask;
Elf_Word type;
const Elf_Sym *sym;
const char *symn;
type = ELF_R_TYPE(rels->r_info);
if (RELOC_ERROR(type))
_dl_die("bad relocation obj %s %ld %d",
object->load_name, i, type);
if (type == R_TYPE(NONE))
continue;
if (type == R_TYPE(JMP_SLOT) && rel != DT_JMPREL)
continue;
where = (Elf_Addr *)(rels->r_offset + loff);
if (RELOC_USE_ADDEND(type))
#ifdef LDSO_ARCH_IS_RELA_
value = rels->r_addend;
#else
value = *where & RELOC_VALUE_BITMASK(type);
#endif
else
value = 0;
sym = NULL;
symn = NULL;
if (RELOC_RESOLVE_SYMBOL(type)) {
sym = object->dyn.symtab;
sym += ELF_R_SYM(rels->r_info);
symn = object->dyn.strtab + sym->st_name;
if (sym->st_shndx != SHN_UNDEF &&
ELF_ST_BIND(sym->st_info) == STB_LOCAL) {
value += loff;
} else if (sym == prev_sym) {
value += prev_value;
} else {
struct sym_res sr;
sr = _dl_find_symbol(symn,
SYM_SEARCH_ALL|SYM_WARNNOTFOUND|
((type == R_TYPE(JMP_SLOT)) ?
SYM_PLT : SYM_NOTPLT),
sym, object);
if (sr.sym == NULL) {
resolve_failed:
if (ELF_ST_BIND(sym->st_info) !=
STB_WEAK)
fails++;
continue;
}
prev_sym = sym;
prev_value = (Elf_Addr)(sr.obj->obj_base +
sr.sym->st_value);
value += prev_value;
}
}
if (type == R_TYPE(JMP_SLOT)) {
_dl_reloc_plt((Elf_Word *)where, value, rels);
continue;
}
if (type == R_TYPE(COPY)) {
void *dstaddr = where;
const void *srcaddr;
const Elf_Sym *dstsym = sym;
struct sym_res sr;
sr = _dl_find_symbol(symn,
SYM_SEARCH_OTHER|SYM_WARNNOTFOUND|SYM_NOTPLT,
dstsym, object);
if (sr.sym == NULL)
goto resolve_failed;
srcaddr = (void *)(sr.obj->obj_base + sr.sym->st_value);
_dl_bcopy(srcaddr, dstaddr, dstsym->st_size);
continue;
}
if (RELOC_PC_RELATIVE(type))
value -= (Elf_Addr)where;
if (RELOC_BASE_RELATIVE(type))
value += loff;
mask = RELOC_VALUE_BITMASK(type);
value >>= RELOC_VALUE_RIGHTSHIFT(type);
value &= mask;
*where &= ~mask;
*where |= value;
}
return fails;
}
int
_dl_md_reloc_got(elf_object_t *object, int lazy)
{
int fails = 0;
Elf_Addr *pltgot = (Elf_Addr *)object->Dyn.info[DT_PLTGOT];
int i, num;
Elf_RelA *rel;
if (object->Dyn.info[DT_PLTREL] != DT_RELA)
return 0;
if (!lazy) {
fails = _dl_md_reloc(object, DT_JMPREL, DT_PLTRELSZ);
} else {
rel = (Elf_RelA *)(object->Dyn.info[DT_JMPREL]);
num = (object->Dyn.info[DT_PLTRELSZ]) / sizeof(Elf_RelA);
for (i = 0; i < num; i++, rel++) {
Elf_Addr *where, value;
Elf_Word type;
where = (Elf_Addr *)(rel->r_offset + object->obj_base);
type = ELF_R_TYPE(rel->r_info);
if (RELOC_USE_ADDEND(type))
value = rel->r_addend;
else
value = 0;
*where += object->obj_base + value;
}
pltgot[1] = (Elf_Addr)object;
pltgot[2] = (Elf_Addr)_dl_bind_start;
}
return fails;
}
Elf_Addr
_dl_bind(elf_object_t *object, int reloff)
{
Elf_RelA *rel;
const Elf_Sym *sym;
const char *symn;
struct sym_res sr;
uint64_t cookie = pcookie;
struct {
struct __kbind param;
Elf_Addr newval;
} buf;
rel = (Elf_RelA *)(object->Dyn.info[DT_JMPREL] + reloff);
sym = object->dyn.symtab;
sym += ELF_R_SYM(rel->r_info);
symn = object->dyn.strtab + sym->st_name;
sr = _dl_find_symbol(symn, SYM_SEARCH_ALL|SYM_WARNNOTFOUND|SYM_PLT,
sym, object);
if (sr.sym == NULL)
_dl_die("lazy binding failed!");
buf.newval = sr.obj->obj_base + sr.sym->st_value;
if (__predict_false(sr.obj->traced) && _dl_trace_plt(sr.obj, symn))
return buf.newval;
buf.param.kb_addr = (Elf_Addr *)(object->obj_base + rel->r_offset);
buf.param.kb_size = sizeof(Elf_Addr);
{
register long syscall_num __asm("r0") = SYS_kbind;
register void *arg1 __asm("r4") = &buf;
register long arg2 __asm("r5") = sizeof(buf);
register long arg3 __asm("r6") = 0xffffffff & cookie;
register long arg4 __asm("r7") = 0xffffffff & (cookie >> 32);
__asm volatile("trapa #0x80" : "+r" (syscall_num)
: "r" (arg1), "r" (arg2), "r" (arg3), "r" (arg4)
: "r1", "cc", "memory");
}
return buf.newval;
}