xref: /freebsd/lib/libthr/thread/thr_clean.c (revision b0b1dbdd)
1 /*
2  * Copyright (c) 1995 John Birrell <jb@cimlogic.com.au>.
3  * 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  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. Neither the name of the author nor the names of any co-contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY JOHN BIRRELL AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  */
29 
30 #include <sys/cdefs.h>
31 __FBSDID("$FreeBSD$");
32 
33 #include "namespace.h"
34 #include <signal.h>
35 #include <errno.h>
36 #include <stdlib.h>
37 #include <pthread.h>
38 #include "un-namespace.h"
39 
40 #include "thr_private.h"
41 
42 #undef pthread_cleanup_push
43 #undef pthread_cleanup_pop
44 
45 /* old binary compatible interfaces */
46 __weak_reference(_pthread_cleanup_push, pthread_cleanup_push);
47 __weak_reference(_pthread_cleanup_pop, pthread_cleanup_pop);
48 
49 void
50 __pthread_cleanup_push_imp(void (*routine)(void *), void *arg,
51 	struct _pthread_cleanup_info *info)
52 {
53 	struct pthread	*curthread = _get_curthread();
54 	struct pthread_cleanup *newbuf;
55 
56 	newbuf = (void *)info;
57 	newbuf->routine = routine;
58 	newbuf->routine_arg = arg;
59 	newbuf->onheap = 0;
60 	newbuf->prev = curthread->cleanup;
61 	curthread->cleanup = newbuf;
62 }
63 
64 void
65 __pthread_cleanup_pop_imp(int execute)
66 {
67 	struct pthread	*curthread = _get_curthread();
68 	struct pthread_cleanup *old;
69 
70 	if ((old = curthread->cleanup) != NULL) {
71 		curthread->cleanup = old->prev;
72 		if (execute)
73 			old->routine(old->routine_arg);
74 		if (old->onheap)
75 			free(old);
76 	}
77 }
78 
79 void
80 _pthread_cleanup_push(void (*routine) (void *), void *arg)
81 {
82 	struct pthread	*curthread = _get_curthread();
83 	struct pthread_cleanup *newbuf;
84 #ifdef _PTHREAD_FORCED_UNWIND
85 	curthread->unwind_disabled = 1;
86 #endif
87 	if ((newbuf = (struct pthread_cleanup *)
88 	    malloc(sizeof(struct pthread_cleanup))) != NULL) {
89 		newbuf->routine = routine;
90 		newbuf->routine_arg = arg;
91 		newbuf->onheap = 1;
92 		newbuf->prev = curthread->cleanup;
93 		curthread->cleanup = newbuf;
94 	}
95 }
96 
97 void
98 _pthread_cleanup_pop(int execute)
99 {
100 	__pthread_cleanup_pop_imp(execute);
101 }
102