From 0f23e882fa8ef33d4f44a3fc0a5bba7a32889801 Mon Sep 17 00:00:00 2001
From: m00nbsd <42475391+m00nbsd@users.noreply.github.com>
Date: Thu, 11 Jun 2020 23:16:52 +0200
Subject: sys/netbsd: add support for fault injection (#1817)

---
 executor/common_bsd.h | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)

(limited to 'executor/common_bsd.h')

diff --git a/executor/common_bsd.h b/executor/common_bsd.h
index a04d86a98..82aef4211 100644
--- a/executor/common_bsd.h
+++ b/executor/common_bsd.h
@@ -12,6 +12,7 @@
 #include <sys/syscall.h>
 
 #if GOOS_netbsd
+
 #if SYZ_EXECUTOR || __NR_syz_usb_connect
 #include "common_usb_netbsd.h"
 #endif
@@ -22,6 +23,54 @@ static void setup_usb(void)
 		fail("failed to chmod /dev/vhci");
 }
 #endif
+
+#if SYZ_EXECUTOR || SYZ_FAULT
+#include <fcntl.h>
+#include <sys/fault.h>
+static void setup_fault(void)
+{
+	if (chmod("/dev/fault", 0666))
+		fail("failed to chmod /dev/fault");
+}
+static int inject_fault(int nth)
+{
+	struct fault_ioc_enable en;
+	int fd;
+
+	fd = open("/dev/fault", O_RDWR);
+	if (fd == -1)
+		fail("failed to open /dev/fault");
+
+	en.scope = FAULT_SCOPE_LWP;
+	en.mode = FAULT_MODE_NTH;
+	en.nth = nth;
+	if (ioctl(fd, FAULT_IOC_ENABLE, &en) != 0)
+		fail("FAULT_IOC_ENABLE failed with nth=%d", nth);
+
+	return fd;
+}
+#endif
+#if SYZ_EXECUTOR
+static int fault_injected(int fd)
+{
+	struct fault_ioc_getinfo info;
+	struct fault_ioc_disable dis;
+	int res;
+
+	if (ioctl(fd, FAULT_IOC_GETINFO, &info) != 0)
+		fail("FAULT_IOC_GETINFO failed");
+	res = (info.nfaults > 0);
+
+	dis.scope = FAULT_SCOPE_LWP;
+	if (ioctl(fd, FAULT_IOC_DISABLE, &dis) != 0)
+		fail("FAULT_IOC_DISABLE failed");
+
+	close(fd);
+
+	return res;
+}
+#endif
+
 #endif
 
 #if GOOS_openbsd
-- 
cgit mrf-deployment