FreeBSD: Enter capabilities mode if restricted

1 files changed, 11 insertions, 0 deletions

diff --git a/chat.c b/chat.c
index 71bfb7b..170094f 100644
--- a/chat.c
+++ b/chat.c
@@ -46,6 +46,10 @@
 #include <tls.h>
 #include <unistd.h>
 
+#ifdef __FreeBSD__
+#include <sys/capsicum.h>
+#endif
+
 #include "chat.h"
 
 #ifndef OPENSSL_BIN
@@ -314,6 +318,13 @@ int main(int argc, char *argv[]) {
 	if (error) err(EX_OSERR, "pledge");
 #endif
 
+#ifdef __FreeBSD__
+	if (self.restricted) {
+		int error = cap_enter();
+		if (error) err(EX_OSERR, "cap_enter");
+	}
+#endif
+
 	if (pass) ircFormat("PASS :%s\r\n", pass);
 	if (sasl) ircFormat("CAP REQ :sasl\r\n");
 	ircFormat("CAP LS\r\n");