diff options
| -rw-r--r-- | client.c | 9 | ||||
| -rw-r--r-- | server.c | 4 |
2 files changed, 13 insertions, 0 deletions
diff --git a/client.c b/client.c index 7e41b5f..904ded7 100644 --- a/client.c +++ b/client.c @@ -33,6 +33,10 @@ #include <unistd.h> #include <wchar.h> +#ifdef __FreeBSD__ +#include <sys/capsicum.h> +#endif + #include "torus.h" #include "help.h" @@ -694,6 +698,11 @@ int main(int argc, char *argv[]) { int error = connect(client, (struct sockaddr *)&addr, SUN_LEN(&addr)); if (error) err(EX_NOINPUT, "%s", sockPath); +#ifdef __FreeBSD__ + error = cap_enter(); + if (error) err(EX_OSERR, "cap_enter"); +#endif + struct pollfd fds[2] = { { .fd = STDIN_FILENO, .events = POLLIN }, { .fd = client, .events = POLLIN }, diff --git a/server.c b/server.c index cff6d21..a662efd 100644 --- a/server.c +++ b/server.c @@ -35,6 +35,7 @@ #ifdef __FreeBSD__ #include <libutil.h> +#include <sys/capsicum.h> #endif #include "torus.h" @@ -396,6 +397,9 @@ int main(int argc, char *argv[]) { if (error) err(EX_CANTCREAT, "%s", sockPath); #ifdef __FreeBSD__ + error = cap_enter(); + if (error) err(EX_OSERR, "cap_enter"); + if (pid) { error = daemon(0, 0); if (error) err(EX_OSERR, "daemon"); |