blob: 681340c5b650ed006a77fc61b62651a69d3aab7f [file] [log] [blame]
// Copyright 2012 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
#include "runtime.h"
#include "textflag.h"
// Look up symbols in the Linux vDSO.
// This code was originally based on the sample Linux vDSO parser at
// https://git.kernel.org/cgit/linux/kernel/git/torvalds/linux.git/tree/Documentation/vDSO/parse_vdso.c
// This implements the ELF dynamic linking spec at
// http://sco.com/developers/gabi/latest/ch5.dynamic.html
// The version section is documented at
// http://refspecs.linuxfoundation.org/LSB_3.2.0/LSB-Core-generic/LSB-Core-generic/symversion.html
#define AT_RANDOM 25
#define AT_SYSINFO_EHDR 33
#define AT_NULL 0 /* End of vector */
#define PT_LOAD 1 /* Loadable program segment */
#define PT_DYNAMIC 2 /* Dynamic linking information */
#define DT_NULL 0 /* Marks end of dynamic section */
#define DT_HASH 4 /* Dynamic symbol hash table */
#define DT_STRTAB 5 /* Address of string table */
#define DT_SYMTAB 6 /* Address of symbol table */
#define DT_VERSYM 0x6ffffff0
#define DT_VERDEF 0x6ffffffc
#define VER_FLG_BASE 0x1 /* Version definition of file itself */
#define SHN_UNDEF 0 /* Undefined section */
#define SHT_DYNSYM 11 /* Dynamic linker symbol table */
#define STT_FUNC 2 /* Symbol is a code object */
#define STB_GLOBAL 1 /* Global symbol */
#define STB_WEAK 2 /* Weak symbol */
/* How to extract and insert information held in the st_info field. */
#define ELF64_ST_BIND(val) (((byte) (val)) >> 4)
#define ELF64_ST_TYPE(val) ((val) & 0xf)
#define EI_NIDENT (16)
typedef uint16 Elf64_Half;
typedef uint32 Elf64_Word;
typedef int32 Elf64_Sword;
typedef uint64 Elf64_Xword;
typedef int64 Elf64_Sxword;
typedef uint64 Elf64_Addr;
typedef uint64 Elf64_Off;
typedef uint16 Elf64_Section;
typedef Elf64_Half Elf64_Versym;
typedef struct Elf64_Sym
{
Elf64_Word st_name;
byte st_info;
byte st_other;
Elf64_Section st_shndx;
Elf64_Addr st_value;
Elf64_Xword st_size;
} Elf64_Sym;
typedef struct Elf64_Verdef
{
Elf64_Half vd_version; /* Version revision */
Elf64_Half vd_flags; /* Version information */
Elf64_Half vd_ndx; /* Version Index */
Elf64_Half vd_cnt; /* Number of associated aux entries */
Elf64_Word vd_hash; /* Version name hash value */
Elf64_Word vd_aux; /* Offset in bytes to verdaux array */
Elf64_Word vd_next; /* Offset in bytes to next verdef entry */
} Elf64_Verdef;
typedef struct Elf64_Ehdr
{
byte e_ident[EI_NIDENT]; /* Magic number and other info */
Elf64_Half e_type; /* Object file type */
Elf64_Half e_machine; /* Architecture */
Elf64_Word e_version; /* Object file version */
Elf64_Addr e_entry; /* Entry point virtual address */
Elf64_Off e_phoff; /* Program header table file offset */
Elf64_Off e_shoff; /* Section header table file offset */
Elf64_Word e_flags; /* Processor-specific flags */
Elf64_Half e_ehsize; /* ELF header size in bytes */
Elf64_Half e_phentsize; /* Program header table entry size */
Elf64_Half e_phnum; /* Program header table entry count */
Elf64_Half e_shentsize; /* Section header table entry size */
Elf64_Half e_shnum; /* Section header table entry count */
Elf64_Half e_shstrndx; /* Section header string table index */
} Elf64_Ehdr;
typedef struct Elf64_Phdr
{
Elf64_Word p_type; /* Segment type */
Elf64_Word p_flags; /* Segment flags */
Elf64_Off p_offset; /* Segment file offset */
Elf64_Addr p_vaddr; /* Segment virtual address */
Elf64_Addr p_paddr; /* Segment physical address */
Elf64_Xword p_filesz; /* Segment size in file */
Elf64_Xword p_memsz; /* Segment size in memory */
Elf64_Xword p_align; /* Segment alignment */
} Elf64_Phdr;
typedef struct Elf64_Shdr
{
Elf64_Word sh_name; /* Section name (string tbl index) */
Elf64_Word sh_type; /* Section type */
Elf64_Xword sh_flags; /* Section flags */
Elf64_Addr sh_addr; /* Section virtual addr at execution */
Elf64_Off sh_offset; /* Section file offset */
Elf64_Xword sh_size; /* Section size in bytes */
Elf64_Word sh_link; /* Link to another section */
Elf64_Word sh_info; /* Additional section information */
Elf64_Xword sh_addralign; /* Section alignment */
Elf64_Xword sh_entsize; /* Entry size if section holds table */
} Elf64_Shdr;
typedef struct Elf64_Dyn
{
Elf64_Sxword d_tag; /* Dynamic entry type */
union
{
Elf64_Xword d_val; /* Integer value */
Elf64_Addr d_ptr; /* Address value */
} d_un;
} Elf64_Dyn;
typedef struct Elf64_Verdaux
{
Elf64_Word vda_name; /* Version or dependency names */
Elf64_Word vda_next; /* Offset in bytes to next verdaux entry */
} Elf64_Verdaux;
typedef struct Elf64_auxv_t
{
uint64 a_type; /* Entry type */
union
{
uint64 a_val; /* Integer value */
} a_un;
} Elf64_auxv_t;
typedef struct symbol_key {
byte* name;
int32 sym_hash;
void** var_ptr;
} symbol_key;
typedef struct version_key {
byte* version;
int32 ver_hash;
} version_key;
struct vdso_info {
bool valid;
/* Load information */
uintptr load_addr;
uintptr load_offset; /* load_addr - recorded vaddr */
/* Symbol table */
Elf64_Sym *symtab;
const byte *symstrings;
Elf64_Word *bucket, *chain;
Elf64_Word nbucket, nchain;
/* Version table */
Elf64_Versym *versym;
Elf64_Verdef *verdef;
};
#pragma dataflag NOPTR
static version_key linux26 = { (byte*)"LINUX_2.6", 0x3ae75f6 };
// initialize with vsyscall fallbacks
#pragma dataflag NOPTR
void* runtime·__vdso_time_sym = (void*)0xffffffffff600400ULL;
#pragma dataflag NOPTR
void* runtime·__vdso_gettimeofday_sym = (void*)0xffffffffff600000ULL;
#pragma dataflag NOPTR
void* runtime·__vdso_clock_gettime_sym = (void*)0;
#pragma dataflag NOPTR
static symbol_key sym_keys[] = {
{ (byte*)"__vdso_time", 0xa33c485, &runtime·__vdso_time_sym },
{ (byte*)"__vdso_gettimeofday", 0x315ca59, &runtime·__vdso_gettimeofday_sym },
{ (byte*)"__vdso_clock_gettime", 0xd35ec75, &runtime·__vdso_clock_gettime_sym },
};
static void
vdso_init_from_sysinfo_ehdr(struct vdso_info *vdso_info, Elf64_Ehdr* hdr)
{
uint64 i;
bool found_vaddr = false;
Elf64_Phdr *pt;
Elf64_Dyn *dyn;
Elf64_Word *hash;
vdso_info->valid = false;
vdso_info->load_addr = (uintptr) hdr;
pt = (Elf64_Phdr*)(vdso_info->load_addr + hdr->e_phoff);
dyn = nil;
// We need two things from the segment table: the load offset
// and the dynamic table.
for(i=0; i<hdr->e_phnum; i++) {
if(pt[i].p_type == PT_LOAD && found_vaddr == false) {
found_vaddr = true;
vdso_info->load_offset = (uintptr)hdr
+ (uintptr)pt[i].p_offset
- (uintptr)pt[i].p_vaddr;
} else if(pt[i].p_type == PT_DYNAMIC) {
dyn = (Elf64_Dyn*)((uintptr)hdr + pt[i].p_offset);
}
}
if(found_vaddr == false || dyn == nil)
return; // Failed
// Fish out the useful bits of the dynamic table.
hash = nil;
vdso_info->symstrings = nil;
vdso_info->symtab = nil;
vdso_info->versym = nil;
vdso_info->verdef = nil;
for(i=0; dyn[i].d_tag!=DT_NULL; i++) {
switch(dyn[i].d_tag) {
case DT_STRTAB:
vdso_info->symstrings = (const byte *)
((uintptr)dyn[i].d_un.d_ptr
+ vdso_info->load_offset);
break;
case DT_SYMTAB:
vdso_info->symtab = (Elf64_Sym *)
((uintptr)dyn[i].d_un.d_ptr
+ vdso_info->load_offset);
break;
case DT_HASH:
hash = (Elf64_Word *)
((uintptr)dyn[i].d_un.d_ptr
+ vdso_info->load_offset);
break;
case DT_VERSYM:
vdso_info->versym = (Elf64_Versym *)
((uintptr)dyn[i].d_un.d_ptr
+ vdso_info->load_offset);
break;
case DT_VERDEF:
vdso_info->verdef = (Elf64_Verdef *)
((uintptr)dyn[i].d_un.d_ptr
+ vdso_info->load_offset);
break;
}
}
if(vdso_info->symstrings == nil || vdso_info->symtab == nil || hash == nil)
return; // Failed
if(vdso_info->verdef == nil)
vdso_info->versym = 0;
// Parse the hash table header.
vdso_info->nbucket = hash[0];
vdso_info->nchain = hash[1];
vdso_info->bucket = &hash[2];
vdso_info->chain = &hash[vdso_info->nbucket + 2];
// That's all we need.
vdso_info->valid = true;
}
static int32
vdso_find_version(struct vdso_info *vdso_info, version_key* ver)
{
if(vdso_info->valid == false) {
return 0;
}
Elf64_Verdef *def = vdso_info->verdef;
while(true) {
if((def->vd_flags & VER_FLG_BASE) == 0) {
Elf64_Verdaux *aux = (Elf64_Verdaux*)((byte *)def + def->vd_aux);
if(def->vd_hash == ver->ver_hash &&
runtime·strcmp(ver->version, vdso_info->symstrings + aux->vda_name) == 0) {
return def->vd_ndx & 0x7fff;
}
}
if(def->vd_next == 0) {
break;
}
def = (Elf64_Verdef *)((byte *)def + def->vd_next);
}
return -1; // can not match any version
}
static void
vdso_parse_symbols(struct vdso_info *vdso_info, int32 version)
{
int32 i;
Elf64_Word chain;
Elf64_Sym *sym;
if(vdso_info->valid == false)
return;
for(i=0; i<nelem(sym_keys); i++) {
for(chain = vdso_info->bucket[sym_keys[i].sym_hash % vdso_info->nbucket];
chain != 0; chain = vdso_info->chain[chain]) {
sym = &vdso_info->symtab[chain];
if(ELF64_ST_TYPE(sym->st_info) != STT_FUNC)
continue;
if(ELF64_ST_BIND(sym->st_info) != STB_GLOBAL &&
ELF64_ST_BIND(sym->st_info) != STB_WEAK)
continue;
if(sym->st_shndx == SHN_UNDEF)
continue;
if(runtime·strcmp(sym_keys[i].name, vdso_info->symstrings + sym->st_name) != 0)
continue;
// Check symbol version.
if(vdso_info->versym != nil && version != 0
&& vdso_info->versym[chain] & 0x7fff != version)
continue;
*sym_keys[i].var_ptr = (void *)(vdso_info->load_offset + sym->st_value);
break;
}
}
}
static void
runtime·linux_setup_vdso(int32 argc, uint8** argv)
{
struct vdso_info vdso_info;
// skip argvc
byte **p = argv;
p = &p[argc+1];
// skip envp to get to ELF auxiliary vector.
for(; *p!=0; p++) {}
// skip NULL separator
p++;
// now, p points to auxv
Elf64_auxv_t *elf_auxv = (Elf64_auxv_t*) p;
for(int32 i=0; elf_auxv[i].a_type!=AT_NULL; i++) {
if(elf_auxv[i].a_type == AT_SYSINFO_EHDR) {
if(elf_auxv[i].a_un.a_val == 0) {
// Something went wrong
continue;
}
vdso_init_from_sysinfo_ehdr(&vdso_info, (Elf64_Ehdr*)elf_auxv[i].a_un.a_val);
vdso_parse_symbols(&vdso_info, vdso_find_version(&vdso_info, &linux26));
continue;
}
if(elf_auxv[i].a_type == AT_RANDOM) {
runtime·startup_random_data = (byte*)elf_auxv[i].a_un.a_val;
runtime·startup_random_data_len = 16;
continue;
}
}
}
void (*runtime·sysargs)(int32, uint8**) = runtime·linux_setup_vdso;