mirror of
https://github.com/checkpoint-restore/criu
synced 2025-08-22 09:58:09 +00:00
93dd984ca0
Acked-by: Mike Rapoport <rppt@linux.ibm.com> Signed-off-by: Adrian Reber <areber@redhat.com>
96 lines
2.3 KiB
C
96 lines
2.3 KiB
C
#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;
|
|
}
|