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