From 2be1c778a4dfe2c6ebb371b457c1e1dc9e89c98d Mon Sep 17 00:00:00 2001
From: Ian Moffett <ian@osmora.org>
Date: Wed, 17 Jul 2024 00:24:32 -0400
Subject: kernel: exec: Save program state within thread

Signed-off-by: Ian Moffett <ian@osmora.org>
---
 sys/kern/exec_elf64.c | 1 +
 sys/kern/kern_exec.c  | 4 ++++
 2 files changed, 5 insertions(+)

(limited to 'sys/kern')

diff --git a/sys/kern/exec_elf64.c b/sys/kern/exec_elf64.c
index b207fbb..d9a3bb9 100644
--- a/sys/kern/exec_elf64.c
+++ b/sys/kern/exec_elf64.c
@@ -31,6 +31,7 @@
 #include <sys/exec.h>
 #include <sys/param.h>
 #include <sys/namei.h>
+#include <sys/proc.h>
 #include <sys/vnode.h>
 #include <sys/errno.h>
 #include <vm/pmap.h>
diff --git a/sys/kern/kern_exec.c b/sys/kern/kern_exec.c
index 1466159..ceb08f6 100644
--- a/sys/kern/kern_exec.c
+++ b/sys/kern/kern_exec.c
@@ -36,6 +36,7 @@
 #include <vm/map.h>
 #include <vm/physmem.h>
 #include <machine/pcb.h>
+#include <string.h>
 
 /*
  * Release the memory of the old stack
@@ -84,6 +85,9 @@ execve(struct proc *td, const struct execve_args *args)
     if (td->stack_base != 0)
         release_stack(td);
 
+    /* Save program state */
+    memcpy(&td->exec, &prog, sizeof(td->exec));
+
     /* Set new stack and map it to userspace */
     td->stack_base = stack;
     vm_map(pcbp->addrsp, td->stack_base, td->stack_base,
-- 
cgit v1.2.3