about summary refs log tree commit diff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/exit/at_quick_exit.c29
-rw-r--r--src/exit/quick_exit.c15
2 files changed, 44 insertions, 0 deletions
diff --git a/src/exit/at_quick_exit.c b/src/exit/at_quick_exit.c
new file mode 100644
index 00000000..85c3d26e
--- /dev/null
+++ b/src/exit/at_quick_exit.c
@@ -0,0 +1,29 @@
+#include <stdlib.h>
+#include "libc.h"
+
+#define COUNT 32
+
+static void (*funcs[COUNT])(void);
+static int count;
+static int lock[2];
+
+void __funcs_on_quick_exit()
+{
+	void (*func)(void);
+	LOCK(lock);
+	while (count > 0) {
+		func = funcs[--count];
+		UNLOCK(lock);
+		func();
+		LOCK(lock);
+	}
+}
+
+int at_quick_exit(void (*func)(void))
+{
+	if (count == 32) return -1;
+	LOCK(lock);
+	funcs[count++] = func;
+	UNLOCK(lock);
+	return 0;
+}
diff --git a/src/exit/quick_exit.c b/src/exit/quick_exit.c
new file mode 100644
index 00000000..18d5288a
--- /dev/null
+++ b/src/exit/quick_exit.c
@@ -0,0 +1,15 @@
+#include <stdlib.h>
+#include "syscall.h"
+#include "atomic.h"
+#include "libc.h"
+
+static void dummy() { }
+weak_alias(dummy, __funcs_on_quick_exit);
+
+void quick_exit(int code)
+{
+	static int lock;
+	while (a_swap(&lock, 1)) __syscall(SYS_pause);
+	__funcs_on_quick_exit();
+	_Exit(code);
+}