4 files changed, 42 insertions, 4 deletions

diff --git a/sys/kern/kern_exit.c b/sys/kern/kern_exit.c
index a421e52..b6cbc4e 100644
--- a/sys/kern/kern_exit.c
+++ b/sys/kern/kern_exit.c
@@ -111,6 +111,7 @@ int
 exit1(struct proc *td, int flags)
 {
     struct proc *curtd, *procp;
+    struct proc *parent;
     struct cpu_info *ci;
     pid_t target_pid, curpid;
 
@@ -120,6 +121,7 @@ exit1(struct proc *td, int flags)
 
     curpid = curtd->pid;
     td->flags |= PROC_EXITING;
+    parent = td->parent;
 
     /* If we have any children, kill them too */
     if (td->nleaves > 0) {
@@ -149,6 +151,13 @@ exit1(struct proc *td, int flags)
      */
     if (target_pid == curpid) {
         ci->curtd = NULL;
+        if (parent->pid == 0)
+            sched_enter();
+        if (td->data == NULL)
+            sched_enter();
+
+        sched_enqueue_td(parent);
+        parent->flags &= ~PROC_SLEEP;
         sched_enter();
     }
 
@@ -163,6 +172,7 @@ sys_exit(struct syscall_args *scargs)
 {
     struct proc *td = this_td();
 
+    td->data = scargs->tf;
     td->exit_status = scargs->arg0;
     exit1(td, 0);
     __builtin_unreachable();
diff --git a/sys/kern/kern_spawn.c b/sys/kern/kern_spawn.c
index 3360443..4105668 100644
--- a/sys/kern/kern_spawn.c
+++ b/sys/kern/kern_spawn.c
@@ -111,6 +111,7 @@ spawn(struct proc *cur, void(*func)(void), void *p, int flags, struct proc **new
     struct proc *newproc;
     struct mmap_lgdr *mlgdr;
     int error;
+    pid_t pid;
 
     newproc = dynalloc(sizeof(*newproc));
     if (newproc == NULL) {
@@ -161,7 +162,25 @@ spawn(struct proc *cur, void(*func)(void), void *p, int flags, struct proc **new
     newproc->pid = ++nthreads;
     signals_init(newproc);
     sched_enqueue_td(newproc);
-    return newproc->pid;
+    pid = newproc->pid;
+
+    if (ISSET(flags, SPAWN_WAIT)) {
+        newproc->flags |= PROC_WAITED;
+        cur->flags |= PROC_SLEEP;
+
+        while (ISSET(cur->flags, PROC_SLEEP)) {
+            sched_yield();
+        }
+
+        if (!ISSET(newproc->flags, PROC_ZOMB)) {
+            pr_error("spawn: fatal: %d not zombie\n", newproc->pid);
+            panic("possibly memory corruption\n");
+        }
+
+        proc_reap(newproc);
+    }
+
+    return pid;
 }
 
 /*
diff --git a/sys/kern/kern_stub.c b/sys/kern/kern_stub.c
index fa1be65..17c6e54 100644
--- a/sys/kern/kern_stub.c
+++ b/sys/kern/kern_stub.c
@@ -40,7 +40,9 @@ sigfpe_default(int signo)
     static struct proc *td;
 
     td = this_td();
-    kprintf("Floating point exception (pid=%d)\n", td->pid);
+    syslog_silence(false);
+    kprintf(OMIT_TIMESTAMP "Floating point exception (pid=%d)\n", td->pid);
+    syslog_silence(true);
     exit1(td, 0);
 }
 
@@ -50,7 +52,9 @@ sigkill_default(int signo)
     static struct proc *td;
 
     td = this_td();
-    kprintf("Terminated (pid=%d)\n", td->pid);
+    syslog_silence(false);
+    kprintf(OMIT_TIMESTAMP "Terminated (pid=%d)\n", td->pid);
+    syslog_silence(true);
     exit1(td, 0);
 }
 
@@ -60,7 +64,9 @@ sigsegv_default(int signo)
     static struct proc *td;
 
     td = this_td();
-    kprintf("Segmentation fault (pid=%d)\n", td->pid);
+    syslog_silence(false);
+    kprintf(OMIT_TIMESTAMP "Segmentation fault (pid=%d)\n", td->pid);
+    syslog_silence(true);
     exit1(td, 0);
 }
 
diff --git a/sys/kern/kern_syscall.c b/sys/kern/kern_syscall.c
index 249a04a..8a66f0e 100644
--- a/sys/kern/kern_syscall.c
+++ b/sys/kern/kern_syscall.c
@@ -31,6 +31,7 @@
 #include <sys/sysctl.h>
 #include <sys/reboot.h>
 #include <sys/types.h>
+#include <sys/mman.h>
 #include <sys/proc.h>
 #include <sys/vfs.h>
 
@@ -45,6 +46,8 @@ scret_t(*g_sctab[])(struct syscall_args *) = {
     sys_write,  /* SYS_write */
     sys_spawn,  /* SYS_spawn */
     sys_reboot, /* SYS_reboot */
+    sys_mmap,   /* SYS_mmap */
+    sys_munmap, /* SYS_munap */
 };
 
 const size_t MAX_SYSCALLS = NELEM(g_sctab);