1 files changed, 77 insertions, 0 deletions
diff --git a/kernel/features/load.c b/kernel/features/load.c
index 65d76d4..7d0c3c0 100644
--- a/kernel/features/load.c
+++ b/kernel/features/load.c
@@ -49,3 +49,80 @@ s32 bin_load(const char *path, struct proc *proc)
 	memory_switch_dir(prev);
 	return 0;
 }
+
+s32 elf_load(const char *path, struct proc *proc)
+{
+	if (!path || !memory_valid(path) || !proc)
+		return -EFAULT;
+
+	struct stat s = { 0 };
+	memory_bypass_enable();
+	s32 stat = vfs_stat(path, &s);
+	memory_bypass_disable();
+	if (stat != 0)
+		return stat;
+
+	strcpy(proc->name, path);
+
+	struct elf_header header = { 0 };
+	s32 read = vfs_read(path, &header, 0, sizeof(header));
+	if (read < 0)
+		return read;
+	if (read != sizeof(header))
+		return -ENOEXEC;
+
+	// Valid?
+	u8 *magic = header.ident;
+	u8 valid_magic = magic[ELF_IDENT_MAG0] == ELF_MAG0 && magic[ELF_IDENT_MAG1] == ELF_MAG1 &&
+			 magic[ELF_IDENT_MAG2] == ELF_MAG2 && magic[ELF_IDENT_MAG3] == ELF_MAG3 &&
+			 magic[ELF_IDENT_CLASS] == ELF_IDENT_CLASS_32 &&
+			 magic[ELF_IDENT_DATA] == ELF_IDENT_DATA_LSB;
+	if (!valid_magic || (header.type != ELF_ETYPE_REL && header.type != ELF_ETYPE_EXEC) ||
+	    header.version != 1 || header.machine != ELF_MACHINE_386)
+		return -ENOEXEC;
+
+	// I need to implement ext2 offset reading first
+	panic("TUDU, Marvin!\n");
+
+	for (u32 i = 0; i < header.phnum; i++) {
+		struct elf_program program = { 0 };
+		if (vfs_read(path, &program, header.phoff + header.phentsize * i,
+			     sizeof(program)) != sizeof(program))
+			return -ENOEXEC;
+
+		if (program.vaddr == 0)
+			continue;
+
+		if (program.vaddr <= 0x100000)
+			return -ENOEXEC;
+
+		struct page_dir *prev;
+		memory_backup_dir(&prev);
+		memory_switch_dir(proc->page_dir);
+
+		struct memory_range range = memory_range_around(program.vaddr, program.memsz);
+		struct memory_range prange = physical_alloc(range.size);
+		virtual_map(proc->page_dir, prange, range.base, MEMORY_CLEAR | MEMORY_USER);
+
+		if ((u32)vfs_read(proc->name, (void *)program.vaddr, program.offset,
+				  program.filesz) != program.filesz) {
+			memory_switch_dir(prev);
+			return -ENOEXEC;
+		}
+
+		memory_switch_dir(prev);
+	}
+
+	struct page_dir *prev;
+	memory_backup_dir(&prev);
+	memory_switch_dir(proc->page_dir);
+
+	u32 stack = (u32)memory_alloc(proc->page_dir, PROC_STACK_SIZE, MEMORY_USER | MEMORY_CLEAR);
+	proc->regs.ebp = stack;
+	proc->regs.useresp = stack;
+	proc->regs.eip = header.entry;
+	proc->entry = header.entry;
+
+	memory_switch_dir(prev);
+	return 0;
+}