Add ELF header sanity check

2 years ago · 1f326d6ed8
1 changed files with 2 additions and 0 deletions
--- a/pk/elf.c
+++ b/pk/elf.c
@ -76,6 +76,8 @@ void load_elf(const char* fn, elf_info* info)
      panic("not a statically linked ELF program");
    }
    if(ph[i].p_type == PT_LOAD && ph[i].p_memsz) {
+      if (ph[i].p_filesz > ph[i].p_memsz)
+        goto fail;
      uintptr_t prepad = ph[i].p_vaddr % RISCV_PGSIZE;
      uintptr_t vaddr = ph[i].p_vaddr + bias;
      if (vaddr + ph[i].p_memsz > info->brk_min)