97 lines
2.1 KiB
C
97 lines
2.1 KiB
C
/*
|
|
* Copyright (c) 1990 Regents of the University of California.
|
|
* All rights reserved.
|
|
*
|
|
* %sccs.include.redist.c%
|
|
*/
|
|
|
|
|
|
/*
|
|
|
|
@deftypefun int xatexit (void (*@var{fn}) (void))
|
|
|
|
Behaves as the standard @code{atexit} function, but with no limit on
|
|
the number of registered functions. Returns 0 on success, or @minus{}1 on
|
|
failure. If you use @code{xatexit} to register functions, you must use
|
|
@code{xexit} to terminate your program.
|
|
|
|
@end deftypefun
|
|
|
|
*/
|
|
|
|
/* Adapted from newlib/libc/stdlib/{,at}exit.[ch].
|
|
If you use xatexit, you must call xexit instead of exit. */
|
|
|
|
#include "ansidecl.h"
|
|
#include "libiberty.h"
|
|
|
|
#include <stdio.h>
|
|
|
|
#ifdef __STDC__
|
|
#include <stddef.h>
|
|
#else
|
|
#define size_t unsigned long
|
|
#endif
|
|
|
|
/* For systems with larger pointers than ints, this must be declared. */
|
|
PTR malloc PARAMS ((size_t));
|
|
|
|
static void xatexit_cleanup PARAMS ((void));
|
|
|
|
/* Pointer to function run by xexit. */
|
|
extern void (*_xexit_cleanup) PARAMS ((void));
|
|
|
|
#define XATEXIT_SIZE 32
|
|
|
|
struct xatexit {
|
|
struct xatexit *next; /* next in list */
|
|
int ind; /* next index in this table */
|
|
void (*fns[XATEXIT_SIZE]) PARAMS ((void)); /* the table itself */
|
|
};
|
|
|
|
/* Allocate one struct statically to guarantee that we can register
|
|
at least a few handlers. */
|
|
static struct xatexit xatexit_first;
|
|
|
|
/* Points to head of LIFO stack. */
|
|
static struct xatexit *xatexit_head = &xatexit_first;
|
|
|
|
/* Register function FN to be run by xexit.
|
|
Return 0 if successful, -1 if not. */
|
|
|
|
int
|
|
xatexit (fn)
|
|
void (*fn) PARAMS ((void));
|
|
{
|
|
register struct xatexit *p;
|
|
|
|
/* Tell xexit to call xatexit_cleanup. */
|
|
if (!_xexit_cleanup)
|
|
_xexit_cleanup = xatexit_cleanup;
|
|
|
|
p = xatexit_head;
|
|
if (p->ind >= XATEXIT_SIZE)
|
|
{
|
|
if ((p = (struct xatexit *) malloc (sizeof *p)) == NULL)
|
|
return -1;
|
|
p->ind = 0;
|
|
p->next = xatexit_head;
|
|
xatexit_head = p;
|
|
}
|
|
p->fns[p->ind++] = fn;
|
|
return 0;
|
|
}
|
|
|
|
/* Call any cleanup functions. */
|
|
|
|
static void
|
|
xatexit_cleanup ()
|
|
{
|
|
register struct xatexit *p;
|
|
register int n;
|
|
|
|
for (p = xatexit_head; p; p = p->next)
|
|
for (n = p->ind; --n >= 0;)
|
|
(*p->fns[n]) ();
|
|
}
|