libiberty/xatexit.c

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