Commit | Line | Data |
---|---|---|
252b5132 RH |
1 | /* |
2 | * Copyright (c) 1990 Regents of the University of California. | |
3 | * All rights reserved. | |
4 | * | |
5 | * %sccs.include.redist.c% | |
6 | */ | |
7 | ||
39423523 DD |
8 | |
9 | /* | |
10 | ||
11 | @deftypefun int xatexit (void (*@var{fn}) (void)) | |
12 | ||
13 | Behaves as the standard @code{atexit} function, but with no limit on | |
99b58139 | 14 | the number of registered functions. Returns 0 on success, or @minus{}1 on |
39423523 DD |
15 | failure. If you use @code{xatexit} to register functions, you must use |
16 | @code{xexit} to terminate your program. | |
17 | ||
18 | @end deftypefun | |
19 | ||
20 | */ | |
21 | ||
252b5132 RH |
22 | /* Adapted from newlib/libc/stdlib/{,at}exit.[ch]. |
23 | If you use xatexit, you must call xexit instead of exit. */ | |
24 | ||
fa99459d DD |
25 | #ifdef HAVE_CONFIG_H |
26 | #include "config.h" | |
27 | #endif | |
252b5132 RH |
28 | #include "ansidecl.h" |
29 | #include "libiberty.h" | |
30 | ||
31 | #include <stdio.h> | |
32 | ||
252b5132 | 33 | #include <stddef.h> |
252b5132 | 34 | |
dabc64ea DD |
35 | #if VMS |
36 | #include <stdlib.h> | |
37 | #include <unixlib.h> | |
38 | #else | |
252b5132 | 39 | /* For systems with larger pointers than ints, this must be declared. */ |
49b1fae4 | 40 | PTR malloc (size_t); |
dabc64ea | 41 | #endif |
252b5132 | 42 | |
49b1fae4 | 43 | static void xatexit_cleanup (void); |
252b5132 RH |
44 | |
45 | /* Pointer to function run by xexit. */ | |
49b1fae4 | 46 | extern void (*_xexit_cleanup) (void); |
252b5132 RH |
47 | |
48 | #define XATEXIT_SIZE 32 | |
49 | ||
50 | struct xatexit { | |
51 | struct xatexit *next; /* next in list */ | |
52 | int ind; /* next index in this table */ | |
49b1fae4 | 53 | void (*fns[XATEXIT_SIZE]) (void); /* the table itself */ |
252b5132 RH |
54 | }; |
55 | ||
56 | /* Allocate one struct statically to guarantee that we can register | |
57 | at least a few handlers. */ | |
58 | static struct xatexit xatexit_first; | |
59 | ||
60 | /* Points to head of LIFO stack. */ | |
61 | static struct xatexit *xatexit_head = &xatexit_first; | |
62 | ||
63 | /* Register function FN to be run by xexit. | |
64 | Return 0 if successful, -1 if not. */ | |
65 | ||
66 | int | |
49b1fae4 | 67 | xatexit (void (*fn) (void)) |
252b5132 RH |
68 | { |
69 | register struct xatexit *p; | |
70 | ||
71 | /* Tell xexit to call xatexit_cleanup. */ | |
72 | if (!_xexit_cleanup) | |
73 | _xexit_cleanup = xatexit_cleanup; | |
74 | ||
75 | p = xatexit_head; | |
76 | if (p->ind >= XATEXIT_SIZE) | |
77 | { | |
78 | if ((p = (struct xatexit *) malloc (sizeof *p)) == NULL) | |
79 | return -1; | |
80 | p->ind = 0; | |
81 | p->next = xatexit_head; | |
82 | xatexit_head = p; | |
83 | } | |
84 | p->fns[p->ind++] = fn; | |
85 | return 0; | |
86 | } | |
87 | ||
88 | /* Call any cleanup functions. */ | |
89 | ||
90 | static void | |
49b1fae4 | 91 | xatexit_cleanup (void) |
252b5132 RH |
92 | { |
93 | register struct xatexit *p; | |
94 | register int n; | |
95 | ||
96 | for (p = xatexit_head; p; p = p->next) | |
97 | for (n = p->ind; --n >= 0;) | |
98 | (*p->fns[n]) (); | |
99 | } |