diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/exit/at_quick_exit.c | 29 | ||||
| -rw-r--r-- | src/exit/quick_exit.c | 15 | 
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); +} | 
