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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
#include <apos/elf.h>
#include <apos/vmem.h>
#include <apos/bytes.h>
#include <apos/string.h>
static uint8_t __elf_to_uvflags(uint8_t elf_flags)
{
uint8_t uvflags = VM_V | VM_U;
if(elf_flags & PF_X)
uvflags |= VM_X;
if(elf_flags & PF_W)
uvflags |= VM_W;
if(elf_flags & PF_R)
uvflags |= VM_R;
return uvflags;
}
/* useful bit of info: all segments are sorted in ascending order of p_vaddr */
static void __map_exec(struct tcb *t, vm_t bin, uint8_t ei_c, vm_t phstart, size_t phnum, size_t phsize)
{
/* TODO: take alignment into consideration? */
/* TODO: take overlapping memory regions into account, probably mostly
* by keeping track of previously allocated area and seeing if the
* segment fits into it */
/* TODO: check if p_memsz is larger than p_filesz, the segment should be
* filled with zeroes. */
/* TODO: in general, make this a low more clean. */
vm_t runner = phstart;
for(size_t i = 0; i < phnum; ++i, runner += phsize){
if(program_header_prop(ei_c, runner, p_type) != PT_LOAD)
continue;
vm_t va = program_header_prop(ei_c, runner, p_vaddr);
size_t vsz = program_header_prop(ei_c, runner, p_memsz);
vm_t start = alloc_fixed_region(&t->sp_r, va, vsz, &vsz);
if(!start)
return; /* out of memory or something */
uint8_t elf_flags = program_header_prop(ei_c, runner, p_flags);
uint8_t uvflags = __elf_to_uvflags(elf_flags);
map_allocd_region(t->b_r, start, vsz, VM_V | VM_X | VM_R | VM_W | VM_U);
vm_t vo = bin + program_header_prop(ei_c, runner, p_offset);
vm_t vfz = program_header_prop(ei_c, runner, p_filesz);
memcpy((void *)va, (void *)vo, vfz);
/* skip while testing
* TODO: also fix, this modifies only the first region. Create new
* function?
*
pm_t paddr = 0;
stat_vmem(t->b_r, va, &paddr, 0, 0);
mod_vmem(t->b_r, va, paddr, uvflags);
*/
}
}
static vm_t __map_dyn(struct tcb *t, vm_t bin, uint8_t ei_c, vm_t phstart, size_t phnum, size_t phsize)
{
/* TODO: this path should only be taken when no PT_INTERP is defined, as
* making sure ld is loaded should be done in userspace. Maybe a bit
* hacky, I know.*/
}
static vm_t __prepare_proc(struct tcb *t, uint8_t ei_c, vm_t elf)
{
short e_type = elf_header_prop(ei_c, elf, e_type);
if(e_type != ET_DYN && e_type != ET_EXEC)
return 0;
vm_t phstart = ptradd(elf, elf_header_prop(ei_c, elf, e_phoff));
size_t phnum = elf_header_prop(ei_c, elf, e_phnum);
size_t phsize = elf_header_prop(ei_c, elf, e_phentsize);
vm_t entry = elf_header_prop(ei_c, elf, e_entry);
if(e_type == ET_EXEC){
__map_exec(t, elf, ei_c, phstart, phnum, phsize);
return entry;
} else {
vm_t o = __map_dyn(t, elf, ei_c, phstart, phnum, phsize);
return o + entry;
}
}
/* sets up all memory regions etc, returns the entry address */
vm_t load_elf(struct tcb *t, vm_t b)
{
struct elf_ident *i = (struct elf_ident *)b;
if(i->ei_magic != cpu_to_be32(EI_MAGIC))
return 0;
if(i->ei_class != ELFCLASS32 && i->ei_class != ELFCLASS64)
return 0;
/* more sanity checks? */
return __prepare_proc(t, i->ei_class, b);
}
|
