stdlib/atexit.c

   1 /* Copyright (C) 1991 Free Software Foundation, Inc.
   2 This file is part of the GNU C Library.
   3
   4 The GNU C Library is free software; you can redistribute it and/or modify
   5 it under the terms of the GNU General Public License as published by
   6 the Free Software Foundation; either version 1, or (at your option)
   7 any later version.
   8
   9 The GNU C Library is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with the GNU C Library; see the file COPYING.  If not, write to
  16 the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17
  18 #include <ansidecl.h>
  19 #include <stdlib.h>
  20 #include "exit.h"
  21
  22
  23 /* Register FUNC to be executed by `exit'.  */
  24 int
  25 DEFUN(atexit, (func), void EXFUN((*func), (NOARGS)))
  26 {
  27   struct exit_function *new = __new_exitfn();
  28
  29   if (new == NULL)
  30     return -1;
  31
  32   new->flavor = ef_at;
  33   new->func.at = func;
  34   return 0;
  35 }
  36
  37
  38 static struct exit_function_list fnlist = { NULL, 0, };
  39 struct exit_function_list *__exit_funcs = &fnlist;
  40
  41 struct exit_function *
  42 DEFUN_VOID(__new_exitfn)
  43 {
  44   register struct exit_function_list *l;
  45
  46   for (l = __exit_funcs; l != NULL; l = l->next)
  47     {
  48       register size_t i;
  49       for (i = 0; i < l->idx; ++i)
  50         if (l->fns[i].flavor == ef_free)
  51           return &l->fns[i];
  52       if (l->idx < sizeof(l->fns) / sizeof(l->fns[0]))
  53         return &l->fns[l->idx++];
  54     }
  55
  56   l = (struct exit_function_list *) malloc(sizeof(struct exit_function_list));
  57   if (l == NULL)
  58     return NULL;
  59   l->next = __exit_funcs;
  60   __exit_funcs = l;
  61
  62   l->idx = 1;
  63   return &l->fns[0];
  64 }