POTD skeleton #31.

Signed-off-by: lns <matzeton@googlemail.com>

1 files changed, 7 insertions, 3 deletions

diff --git a/src/main.c b/src/main.c
index 0f09095..1d65660 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1,4 +1,6 @@
 #include <stdio.h>
+#include <sys/types.h>
+#include <sys/wait.h>
 
 #include "log.h"
 #include "log_colored.h"
@@ -21,7 +23,7 @@ int main(int argc, char *argv[])
     server_ctx *srv[srv_siz];
     jail_ctx *jail[jail_siz];
     forward_ctx *ssh_fwd = NULL;
-    int epoll_fd;
+    int epoll_fd, proc_status;
     pid_t daemon_pid;
 
     (void) argc;
@@ -60,7 +62,7 @@ int main(int argc, char *argv[])
     epoll_fd = jail_setup_epoll( jail, jail_siz );
     D2("Jail epoll fd: %d", epoll_fd);
     ABORT_ON_FATAL( epoll_fd < 0, "Jail daemon epoll setup" );
-    ABORT_ON_FATAL( jail_daemonize_epoll(epoll_fd, jail, jail_siz) < 0,
+    ABORT_ON_FATAL( jail_daemonize(epoll_fd, jail, jail_siz) < 1,
         "Jail daemon startup" );
 
     {
@@ -97,8 +99,10 @@ int main(int argc, char *argv[])
         "Server dropping privileges" );
 
     N("%s", "Server epoll mainloop");
-    ABORT_ON_FATAL( server_mainloop_epoll( epoll_fd, srv, srv_siz ),
+    ABORT_ON_FATAL( server_daemonize( epoll_fd, srv, srv_siz ) < 1,
         "Server epoll mainloop" );
 
+    wait(&proc_status);
+
     return 0;
 }