multiboot2: fix 32 bit elf section loading

Existing code was using 64 bit elf section header unconditionally. This
commit fixes that :)
This commit is contained in:
Bryce Lanham 2023-05-02 20:09:36 -05:00 committed by mintsuki
parent 4960086ec6
commit b7bca1b000
3 changed files with 46 additions and 26 deletions

View File

@ -76,19 +76,6 @@ struct elf32_phdr {
uint32_t p_align;
};
struct elf32_shdr {
uint32_t sh_name;
uint32_t sh_type;
uint32_t sh_flags;
uint32_t sh_addr;
uint32_t sh_offset;
uint32_t sh_size;
uint32_t sh_link;
uint32_t sh_info;
uint32_t sh_addralign;
uint32_t sh_entsize;
};
struct elf64_rela {
uint64_t r_addr;
uint32_t r_info;

View File

@ -69,6 +69,19 @@ struct elf64_shdr {
uint64_t sh_entsize;
};
struct elf32_shdr {
uint32_t sh_name;
uint32_t sh_type;
uint32_t sh_flags;
uint32_t sh_addr;
uint32_t sh_offset;
uint32_t sh_size;
uint32_t sh_link;
uint32_t sh_info;
uint32_t sh_addralign;
uint32_t sh_entsize;
};
struct elf64_sym {
uint32_t st_name;
uint8_t st_info;

View File

@ -376,22 +376,42 @@ noreturn void multiboot2_load(char *config, char* cmdline) {
memcpy(tag->sections, kernel + section_hdr_info.section_offset, section_hdr_info.section_entry_size * section_hdr_info.num);
int bits = elf_bits(kernel);
for (size_t i = 0; i < section_hdr_info.num; i++) {
struct elf64_shdr *shdr = (void *)tag->sections + i * section_hdr_info.section_entry_size;
if (bits == 64) {
struct elf64_shdr *shdr = (void *)tag->sections + i * section_hdr_info.section_entry_size;
if (shdr->sh_addr != 0 || shdr->sh_size == 0) {
continue;
if (shdr->sh_addr != 0 || shdr->sh_size == 0) {
continue;
}
uint64_t section = (uint64_t)-1; /* no target preference, use top */
if (!elsewhere_append(true /* flexible target */,
ranges, &ranges_count,
kernel + shdr->sh_offset, &section, shdr->sh_size)) {
panic(true, "multiboot2: Cannot allocate elf sections");
}
shdr->sh_addr = section;
} else {
struct elf32_shdr *shdr = (void *)tag->sections + i * section_hdr_info.section_entry_size;
if (shdr->sh_addr != 0 || shdr->sh_size == 0) {
continue;
}
uint64_t section = (uint64_t)-1; /* no target preference, use top */
if (!elsewhere_append(true /* flexible target */,
ranges, &ranges_count,
kernel + shdr->sh_offset, &section, shdr->sh_size)) {
panic(true, "multiboot2: Cannot allocate elf sections");
}
shdr->sh_addr = section;
}
uint64_t section = (uint64_t)-1; /* no target preference, use top */
if (!elsewhere_append(true /* flexible target */,
ranges, &ranges_count,
kernel + shdr->sh_offset, &section, shdr->sh_size)) {
panic(true, "multiboot2: Cannot allocate elf sections");
}
shdr->sh_addr = section;
}
append_tag(info_idx, tag);