add support for init/fini array in main program, and greatly simplify
[musl] / src / exit / exit.c
1 #include <stdlib.h>
2 #include <unistd.h>
3 #include <stdio.h>
4 #include "libc.h"
5 #include "atomic.h"
6 #include "syscall.h"
7
8 static void dummy()
9 {
10 }
11
12 /* __toread.c, __towrite.c, and atexit.c override these */
13 weak_alias(dummy, __funcs_on_exit);
14 weak_alias(dummy, __flush_on_exit);
15 weak_alias(dummy, __seek_on_exit);
16
17 #ifndef SHARED
18 weak_alias(dummy, _fini);
19 extern void (*const __fini_array_start)() __attribute__((weak));
20 extern void (*const __fini_array_end)() __attribute__((weak));
21 #endif
22
23 _Noreturn void exit(int code)
24 {
25         static int lock;
26
27         /* If more than one thread calls exit, hang until _Exit ends it all */
28         while (a_swap(&lock, 1)) __syscall(SYS_pause);
29
30         __funcs_on_exit();
31
32 #ifndef SHARED
33         uintptr_t a = (uintptr_t)&__fini_array_end;
34         for (; a>(uintptr_t)&__fini_array_start; a-=sizeof(void(*)()))
35                 (*(void (**)())(a-sizeof(void(*)())))();
36         _fini();
37 #endif
38
39         __flush_on_exit();
40         __seek_on_exit();
41
42         _Exit(code);
43         for(;;);
44 }