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 __FBSDID("$FreeBSD: stable/12/lib/libthr/thread/thr_clean.c 351010 2019-08-14 09:37:43Z kib $");
35
36 #include "namespace.h"
37 #include <signal.h>
38 #include <errno.h>
39 #include <stdlib.h>
40 #include <pthread.h>
41 #include "un-namespace.h"
42
43 #include "thr_private.h"
44
45 #undef pthread_cleanup_push
46 #undef pthread_cleanup_pop
47
48 /* old binary compatible interfaces */
49 __weak_reference(_thr_cleanup_pop, pthread_cleanup_pop);
50 __weak_reference(_thr_cleanup_pop, _pthread_cleanup_pop);
51 __weak_reference(_thr_cleanup_push, pthread_cleanup_push);
52 __weak_reference(_thr_cleanup_push, _pthread_cleanup_push);
53
54 /* help static linking when libc symbols have preference */
55 __weak_reference(__thr_cleanup_push_imp, __pthread_cleanup_push_imp);
56 __weak_reference(__thr_cleanup_pop_imp, __pthread_cleanup_pop_imp);
57
58 void
__thr_cleanup_push_imp(void (* routine)(void *),void * arg,struct _pthread_cleanup_info * info)59 __thr_cleanup_push_imp(void (*routine)(void *), void *arg,
60 struct _pthread_cleanup_info *info)
61 {
62 struct pthread *curthread = _get_curthread();
63 struct pthread_cleanup *newbuf;
64
65 newbuf = (void *)info;
66 newbuf->routine = routine;
67 newbuf->routine_arg = arg;
68 newbuf->onheap = 0;
69 newbuf->prev = curthread->cleanup;
70 curthread->cleanup = newbuf;
71 }
72
73 void
__thr_cleanup_pop_imp(int execute)74 __thr_cleanup_pop_imp(int execute)
75 {
76 struct pthread *curthread = _get_curthread();
77 struct pthread_cleanup *old;
78
79 if ((old = curthread->cleanup) != NULL) {
80 curthread->cleanup = old->prev;
81 if (execute)
82 old->routine(old->routine_arg);
83 if (old->onheap)
84 free(old);
85 }
86 }
87
88 void
_thr_cleanup_push(void (* routine)(void *),void * arg)89 _thr_cleanup_push(void (*routine)(void *), void *arg)
90 {
91 struct pthread *curthread = _get_curthread();
92 struct pthread_cleanup *newbuf;
93 #ifdef _PTHREAD_FORCED_UNWIND
94 curthread->unwind_disabled = 1;
95 #endif
96 if ((newbuf = (struct pthread_cleanup *)
97 malloc(sizeof(struct pthread_cleanup))) != NULL) {
98 newbuf->routine = routine;
99 newbuf->routine_arg = arg;
100 newbuf->onheap = 1;
101 newbuf->prev = curthread->cleanup;
102 curthread->cleanup = newbuf;
103 }
104 }
105
106 void
_thr_cleanup_pop(int execute)107 _thr_cleanup_pop(int execute)
108 {
109 __pthread_cleanup_pop_imp(execute);
110 }
111