summaryrefslogtreecommitdiff
path: root/lib/libc/stdlib/exit.c
blob: 34b8a0f2b6bbeb3ac5fab60fef03249700cb7370 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#include <io.h>
#include <libc.h>
#include <unistd.h>
#include <stdio.h>

void (*__dummy_atexit_fvec)(void);
weak_reference(__dummy_atexit_fvec, __atexit_fvec);

void (*__dummy_stdio_cleanup)(void);
weak_reference(__dummy_stdio_cleanup, __stdio_cleanup);

static void __fclose(FILE *fp)
{
	if (fp == NULL) {
		return;
	}

	if (fp->buf_len > 0) {
		fflush(fp);
	}

	if (fp->fd > STDERR_FILENO) {
		close(fp->fd);
	}

	if (fp->buf) {
		free(fp->buf);
	}

	if (fp != stdout && fp != stderr && fp != stdin) {
		free(fp);
	}
}

static void __stdio_cleanup_impl(void)
{
	write(1, "HELLO\n", 6);
	fflush(stdout);

	if (stdout->next != NULL) {
		FILE *cur = stdout->next;
		while (cur) {
			struct __FILE *next = cur->next;
			__fclose(cur);
			cur = next;
		}
	}
}

void exit(int status)
{
	void (*fptr)(void);

	/* Only do stdio cleanup if it was referenced (meaning stdio was used)
	 */
	if (__stdio_cleanup) {
		__stdio_cleanup_impl();
	}

	if (__atexit_fvec) {
		fptr = __atexit_fvec;
		while (fptr) {
			fptr();
			fptr++;
		}
	}

	_exit(status);
}