add c11 quick_exit and at_quick_exit functions

diff --git a/include/stdlib.h b/include/stdlib.h
index 0a8de86..c1d93e7 100644 (file)
--- a/include/stdlib.h
+++ b/include/stdlib.h
@@ -43,6 +43,8 @@ void abort (void);
 int atexit (void (*) (void));
 void exit (int);
 void _Exit (int);
+int at_quick_exit (void (*) (void));
+void quick_exit (int);
 
 char *getenv (const char *);
 

diff --git a/src/exit/at_quick_exit.c b/src/exit/at_quick_exit.c
new file mode 100644 (file)
index 0000000..85c3d26
--- /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 (file)
index 0000000..18d5288
--- /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);
+}