summaryrefslogtreecommitdiff
path: root/lib/libc/internal/init/vdso.c
blob: f94401f6b0db3bb002829667d2010a4d5f90098d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#include <asm/vdso.h>
#include <elf.h>
#include <string.h>
#include <sys/auxv.h>

void __init_vdso(void)
{
	Elf64_Ehdr *ehdr = (Elf64_Ehdr *)getauxval(AT_SYSINFO_EHDR);
	Elf64_Phdr *phdr = (Elf64_Phdr *)(ehdr + ehdr->e_phoff);

	for (int i = 0; i < ehdr->e_phnum; i++) {
		Elf64_Dyn *dyn;
		Elf64_Sym *symtab;
		const char *strtab;

		if (phdr[i].p_type == PT_DYNAMIC) {
			Elf64_Sym *sym;

			dyn = (Elf64_Dyn *)((char *)ehdr + phdr[i].p_offset);
			symtab = 0;
			strtab = 0;

			while (dyn->d_tag != DT_NULL) {
				if (dyn->d_tag == DT_STRTAB) {
					strtab = (const char *)(ehdr + dyn->d_un.d_ptr);
				} else if (dyn->d_tag == DT_SYMTAB) {
					symtab = (Elf64_Sym *)(ehdr + dyn->d_un.d_ptr);
				}

				if (strtab && symtab) {
					break;
				}

				dyn++;
			}

			sym = symtab;
			while ((char *)sym < strtab) {
				if (sym->st_name != 0 && sym->st_value != 0) {
					const char *name = strtab + sym->st_name;

					for (int i = 0; __vdso_symtab[i].name != 0; i++) {
						if (strcmp(name, __vdso_symtab[i].name) == 0) {
							__vdso_symtab[i].func = (void *)(ehdr + sym->st_value);
							break;
						}
					}
				}

				sym = (Elf64_Sym *)((char *)sym + sizeof(Elf64_Sym));
			}

			break;
		}
	}
}