lib/libthr/thread/thr_clean.c

   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 }