Fix some bugs.

2024-04-24 10:19:58 +08:00
parent a2e5b49b19
commit fc380de895
4 changed files with 32 additions and 94 deletions
--- a/Ubiquitous/XiZi_AIoT/softkernel/syscall/sys_exec.c
+++ b/Ubiquitous/XiZi_AIoT/softkernel/syscall/sys_exec.c
@@ -69,9 +69,15 @@ Modification:
 int task_exec(struct TaskMicroDescriptor* task, char* img_start, char* name, char** argv)
 {
    /* load img to task */
+    if (img_start == NULL) {
+        return -1;
+    }
    /* 1. load elf header */
    struct elfhdr elf;
    memcpy((void*)&elf, img_start, sizeof(elf));
+    if (elf.magic != ELF_MAGIC) {
+        return -1;
+    }
    // pgdir for new task
    struct TopLevelPageDirectory pgdir;
    pgdir.pd_addr = NULL;