diff options
-rw-r--r-- | include/stdlib.h | 2 | ||||
-rw-r--r-- | src/exit/at_quick_exit.c | 29 | ||||
-rw-r--r-- | src/exit/quick_exit.c | 15 |
3 files changed, 46 insertions, 0 deletions
diff --git a/include/stdlib.h b/include/stdlib.h index 0a8de866..c1d93e71 100644 --- 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 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); +} |