From f81cad34c0e29384a393bf3aa25dbce3d5962c53 Mon Sep 17 00:00:00 2001
From: Kimplul <kimi.h.kuparinen@gmail.com>
Date: Mon, 26 Aug 2024 19:54:31 +0300
Subject: check for running out of memory during fork

---
 tests/fork-exhaustion/init.c    | 23 +++++++++++++++++++++++
 tests/fork-exhaustion/source.mk |  2 ++
 2 files changed, 25 insertions(+)
 create mode 100644 tests/fork-exhaustion/init.c
 create mode 100644 tests/fork-exhaustion/source.mk

diff --git a/tests/fork-exhaustion/init.c b/tests/fork-exhaustion/init.c
new file mode 100644
index 0000000..0e330f4
--- /dev/null
+++ b/tests/fork-exhaustion/init.c
@@ -0,0 +1,23 @@
+#include <common/test.h>
+
+START(pid, tid, d0, d1, d2, d3)
+{
+	UNUSED(pid);
+	UNUSED(tid);
+	UNUSED(d0);
+	UNUSED(d1);
+	UNUSED(d2);
+	UNUSED(d3);
+
+	check(pid == 0, "illegal pid for init\n");
+	id_t our_id = 0;
+	while (1) {
+		printf("forking\n");
+		id_t new_id = sys_fork(&our_id);
+		if (new_id < 0)
+			break;
+	}
+
+	/* if kernel didn't crash, we consider it a success */
+	ok();
+}
diff --git a/tests/fork-exhaustion/source.mk b/tests/fork-exhaustion/source.mk
new file mode 100644
index 0000000..458973d
--- /dev/null
+++ b/tests/fork-exhaustion/source.mk
@@ -0,0 +1,2 @@
+DO	!= ./scripts/gen-prog -n fork-exhaustion -p init init.c
+DO	!= ./scripts/gen-simple -n fork-exhaustion -p init
-- 
cgit v1.3