libc: Add a copyright to quick_exit.c
[dragonfly.git] / lib / libc / stdlib / quick_exit.c
1 /*-
2 * Copyright (c) 2012
3 *       The DragonFly Project.  All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 *
9 * 1. Redistributions of source code must retain the above copyright
10 *    notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 *    notice, this list of conditions and the following disclaimer in
13 *    the documentation and/or other materials provided with the
14 *    distribution.
15 * 3. Neither the name of The DragonFly Project nor the names of its
16 *    contributors may be used to endorse or promote products derived
17 *    from this software without specific, prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
22 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
23 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
24 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
25 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
26 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
27 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
28 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
29 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30 * SUCH DAMAGE.
31 */
32
33 #include <stdlib.h>
34 #include <unistd.h>
35
36 #include "libc_private.h"
37 #include "spinlock.h"
38
39
40 struct quick_exit_fn {
41        struct quick_exit_fn    *next;
42        void                    (*func)(void);
43 };
44
45 static struct quick_exit_fn *quick_exit_fns;
46 static spinlock_t quick_exit_spinlock;
47
48 /*
49  * at_quick_exit:
50  *
51  *     Register a function to be called at quick_exit.
52  */
53 int
54 at_quick_exit(void (*func)(void))
55 {
56        struct quick_exit_fn *fn;
57
58        fn = malloc(sizeof(struct quick_exit_fn));
59        if (!fn)
60                return (-1);
61
62        fn->func = func;
63
64        if (__isthreaded)
65                _SPINLOCK(&quick_exit_spinlock);
66        
67        fn->next = quick_exit_fns;
68        quick_exit_fns = fn;
69
70        if (__isthreaded)
71                _SPINUNLOCK(&quick_exit_spinlock);
72
73        return (0);
74 }
75
76 /*
77  * quick_exit:
78  *
79  *     Abandon a process. Execute all quick_exit handlers.
80  */
81 void
82 quick_exit(int status)
83 {
84        struct quick_exit_fn *fn;
85
86        if (__isthreaded)
87                _SPINLOCK(&quick_exit_spinlock);
88        for (fn = quick_exit_fns; fn != NULL; fn = fn->next) {
89                fn->func();
90        }
91        if (__isthreaded)
92                _SPINUNLOCK(&quick_exit_spinlock);
93
94        _exit(status);
95 }