From b2c811ecf70e4fc3a42e1505c5c708e42603dc88 Mon Sep 17 00:00:00 2001
From: arun <arun@x.y>
Date: Sat, 20 Jul 2019 20:18:38 +0530
Subject: [PATCH] Add pledge for OpenBSD

Closes #32.
---
 endlessh.c | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/endlessh.c b/endlessh.c
index 51a1df3..5384ac2 100644
--- a/endlessh.c
+++ b/endlessh.c
@@ -1,4 +1,9 @@
-#define _XOPEN_SOURCE 600
+#if defined(__OpenBSD__)
+#  define _BSD_SOURCE  /* for pledge(2) */
+#else
+#  define _XOPEN_SOURCE 600
+#endif
+
 #include <time.h>
 #include <errno.h>
 #include <stdio.h>
@@ -596,6 +601,11 @@ sendline(struct client *client, int max_line_length, unsigned long *rng)
 int
 main(int argc, char **argv)
 {
+#if (defined(__OpenBSD__))
+    if (pledge("inet stdio rpath", NULL) == -1)
+        die();
+#endif
+
     struct config config = CONFIG_DEFAULT;
     const char *config_file = DEFAULT_CONFIG_FILE;
     config_load(&config, config_file, 1);