com32/lib/onexit.c


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

/*
 * onexit.c
 */

#include <stdlib.h>
#include <unistd.h>
#include "atexit.h"

static struct atexit *__atexit_list;

static __noreturn on_exit_exit(int rv)
{
    struct atexit *ap;

    for (ap = __atexit_list; ap; ap = ap->next) {
	ap->fctn(rv, ap->arg);	/* This assumes extra args are harmless */
    }

    _exit(rv);
}

int on_exit(void (*fctn) (int, void *), void *arg)
{
    struct atexit *as = malloc(sizeof(struct atexit));

    if (!as)
	return -1;

    as->fctn = fctn;
    as->arg = arg;

    as->next = __atexit_list;
    __atexit_list = as;

    return 0;
}