lib/libc/stdlib/cxa_thread_atexit_impl.c

   1 /*-
   2  * Copyright (c) 2016 Mahdi Mokhtari <mokhi64@gmail.com>
   3  * Copyright (c) 2016, 2017 The FreeBSD Foundation
   4  * All rights reserved.
   5  *
   6  * Portions of this software were developed by Konstantin Belousov
   7  * under sponsorship from the FreeBSD Foundation.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  19  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  22  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  24  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  27  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  28  * SUCH DAMAGE.
  29  */
  30
  31 #include <sys/cdefs.h>
  32 __FBSDID("$FreeBSD$");
  33
  34 #include <sys/queue.h>
  35 #include "namespace.h"
  36 #include <errno.h>
  37 #include <link.h>
  38 #include <pthread.h>
  39 #include <stddef.h>
  40 #include <stdlib.h>
  41 #include <stdio.h>
  42 #include "un-namespace.h"
  43 #include "libc_private.h"
  44
  45 /*
  46  * C++11 introduces the thread_local scope (like __thread with some
  47  * additions).  As a key-feature it should support non-trivial
  48  * destructors, registered with __cxa_thread_atexit() to be executed
  49  * at the thread termination.
  50  *
  51  * The implemention keeps a _Thread_local list of destructors per each
  52  * thread, and calls __cxa_thread_call_dtors() on each thread's exit
  53  * to do cleanup.  For a thread calling exit(3), in particular, for
  54  * the initial thread returning from main(), we call
  55  * __cxa_thread_call_dtors() inside exit().
  56  *
  57  * It could be possible that a dynamically loaded library, use
  58  * thread_local variable but is dlclose()'d before thread exit.  The
  59  * destructor of this variable will then try to access the address,
  60  * for calling it but it's unloaded, so it'll crash.  We're using
  61  * __elf_phdr_match_addr() to detect and prevent such cases and so
  62  * prevent the crash.
  63  */
  64
  65 #define CXA_DTORS_ITERATIONS 4
  66
  67 struct cxa_thread_dtor {
  68         void *obj;
  69         void (*func)(void *);
  70         void *dso;
  71         LIST_ENTRY(cxa_thread_dtor) entry;
  72 };
  73 static _Thread_local LIST_HEAD(dtor_list, cxa_thread_dtor) dtors =
  74     LIST_HEAD_INITIALIZER(dtors);
  75
  76 int
  77 __cxa_thread_atexit_impl(void (*dtor_func)(void *), void *obj,
  78     void *dso_symbol)
  79 {
  80
  81         return (__cxa_thread_atexit_hidden(dtor_func, obj, dso_symbol));
  82 }
  83
  84 int
  85 __cxa_thread_atexit_hidden(void (*dtor_func)(void *), void *obj,
  86     void *dso_symbol)
  87 {
  88         struct cxa_thread_dtor *new_dtor;
  89
  90         new_dtor = malloc(sizeof(*new_dtor));
  91         if (new_dtor == NULL) {
  92                 errno = ENOMEM; /* forcibly override malloc(3) error */
  93                 return (-1);
  94         }
  95
  96         new_dtor->obj = obj;
  97         new_dtor->func = dtor_func;
  98         new_dtor->dso = dso_symbol;
  99         LIST_INSERT_HEAD(&dtors, new_dtor, entry);
 100         return (0);
 101 }
 102
 103 static void
 104 walk_cb_call(struct cxa_thread_dtor *dtor)
 105 {
 106         struct dl_phdr_info phdr_info;
 107
 108         if (_rtld_addr_phdr(dtor->dso, &phdr_info) &&
 109             __elf_phdr_match_addr(&phdr_info, dtor->func))
 110                 dtor->func(dtor->obj);
 111         else
 112                 fprintf(stderr, "__cxa_thread_call_dtors: dtr %p from "
 113                     "unloaded dso, skipping\n", (void *)(dtor->func));
 114 }
 115
 116 static void
 117 walk_cb_nocall(struct cxa_thread_dtor *dtor __unused)
 118 {
 119 }
 120
 121 static void
 122 cxa_thread_walk(void (*cb)(struct cxa_thread_dtor *))
 123 {
 124         struct cxa_thread_dtor *dtor, *tdtor;
 125
 126         LIST_FOREACH_SAFE(dtor, &dtors, entry, tdtor) {
 127                 LIST_REMOVE(dtor, entry);
 128                 cb(dtor);
 129                 free(dtor);
 130         }
 131 }
 132
 133 /*
 134  * This is the callback function we use to call destructors, once for
 135  * each thread.  It is called in exit(3) in libc/stdlib/exit.c and
 136  * before exit_thread() in libthr/thread/thr_exit.c.
 137  */
 138 void
 139 __cxa_thread_call_dtors(void)
 140 {
 141         int i;
 142
 143         for (i = 0; i < CXA_DTORS_ITERATIONS && !LIST_EMPTY(&dtors); i++)
 144                 cxa_thread_walk(walk_cb_call);
 145
 146         if (!LIST_EMPTY(&dtors)) {
 147                 fprintf(stderr, "Thread %p is exiting with more "
 148                     "thread-specific dtors created after %d iterations "
 149                     "of destructor calls\n",
 150                     _pthread_self(), i);
 151                 cxa_thread_walk(walk_cb_nocall);
 152         }
 153 }