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
|
#include <string.h>
#include "uapi/piegen-err.h"
#include "piegen.h"
#include "arch_test_handle_binary.h"
extern int launch_test(void *mem, int expected_ret, const char *test_fmt, ...);
extern const size_t test_elf_buf_size;
static uintptr_t elf_addr;
static const char *test_bitness;
#define ASSERT(expected, fmt, ...) launch_test((void *)elf_addr, expected, fmt " %s", ##__VA_ARGS__, test_bitness)
static const unsigned int sections_nr = 1;
static void set_elf_hdr_relocatable(Ehdr_t *hdr)
{
hdr->e_type = ET_REL;
hdr->e_version = EV_CURRENT;
}
static int test_add_strings_section(Ehdr_t *hdr)
{
Shdr_t *sec_strings_hdr;
uintptr_t sections_table = elf_addr + hdr->e_shoff;
size_t sections_table_size = sections_nr * sizeof(hdr->e_shentsize);
hdr->e_shnum = sections_nr;
hdr->e_shstrndx = sections_nr; /* off-by-one */
if (ASSERT(-E_NO_STR_SEC, "strings section's header oob of section table"))
return -1;
hdr->e_shstrndx = 0;
sec_strings_hdr = (void *)sections_table;
sec_strings_hdr->sh_offset = (Off_t)-1;
if (ASSERT(-E_NO_STR_SEC, "strings section oob"))
return -1;
/* Put strings just right after sections table. */
sec_strings_hdr->sh_offset = sections_table - elf_addr + sections_table_size;
return 0;
}
static int test_prepare_section_table(Ehdr_t *hdr)
{
hdr->e_shoff = (Off_t)test_elf_buf_size;
if (ASSERT(-E_NO_STR_SEC, "section table start oob"))
return -1;
/* Lets put sections table right after ELF header. */
hdr->e_shoff = (Off_t)sizeof(Ehdr_t);
hdr->e_shentsize = (Half_t)sizeof(Shdr_t);
hdr->e_shnum = (Half_t)-1;
if (ASSERT(-E_NO_STR_SEC, "too many sections in table"))
return -1;
if (test_add_strings_section(hdr))
return -1;
return 0;
}
static int test_prepare_elf_header(void *elf)
{
memset(elf, 0, sizeof(Ehdr_t));
if (ASSERT(-E_NOT_ELF, "zero ELF header"))
return -1;
arch_test_set_elf_hdr_ident(elf);
if (ASSERT(-E_NOT_ELF, "unsupported ELF header"))
return -1;
arch_test_set_elf_hdr_machine(elf);
if (ASSERT(-E_NOT_ELF, "non-relocatable ELF header"))
return -1;
set_elf_hdr_relocatable(elf);
if (test_prepare_section_table(elf))
return -1;
return 0;
}
int __run_tests(void *mem, const char *msg)
{
elf_addr = (uintptr_t)mem;
test_bitness = msg;
if (test_prepare_elf_header(mem))
return 1;
return 0;
}
|
