1//===----------------------------------------------------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "abort_message.h"
10#include "cxxabi.h"
11#include <__thread/support.h>
12#ifndef _LIBCXXABI_HAS_NO_THREADS
13#if defined(__ELF__) && defined(_LIBCXXABI_LINK_PTHREAD_LIB)
14#pragma comment(lib, "pthread")
15#endif
16#endif
17
18#include <stdlib.h>
19
20namespace __cxxabiv1 {
21
22 using Dtor = void(*)(void*);
23
24 extern "C"
25#ifndef HAVE___CXA_THREAD_ATEXIT_IMPL
26 // A weak symbol is used to detect this function's presence in the C library
27 // at runtime, even if libc++ is built against an older libc
28 _LIBCXXABI_WEAK
29#endif
30 int __cxa_thread_atexit_impl(Dtor, void*, void*);
31
32#ifndef HAVE___CXA_THREAD_ATEXIT_IMPL
33
34namespace {
35 // This implementation is used if the C library does not provide
36 // __cxa_thread_atexit_impl() for us. It has a number of limitations that are
37 // difficult to impossible to address without ..._impl():
38 //
39 // - dso_symbol is ignored. This means that a shared library may be unloaded
40 // (via dlclose()) before its thread_local destructors have run.
41 //
42 // - thread_local destructors for the main thread are run by the destructor of
43 // a static object. This is later than expected; they should run before the
44 // destructors of any objects with static storage duration.
45 //
46 // - thread_local destructors on non-main threads run on the first iteration
47 // through the __libccpp_tls_key destructors.
48 // std::notify_all_at_thread_exit() and similar functions must be careful to
49 // wait until the second iteration to provide their intended ordering
50 // guarantees.
51 //
52 // Another limitation, though one shared with ..._impl(), is that any
53 // thread_locals that are first initialized after non-thread_local global
54 // destructors begin to run will not be destroyed. [basic.start.term] states
55 // that all thread_local destructors are sequenced before the destruction of
56 // objects with static storage duration, resulting in a contradiction if a
57 // thread_local is constructed after that point. Thus we consider such
58 // programs ill-formed, and don't bother to run those destructors. (If the
59 // program terminates abnormally after such a thread_local is constructed,
60 // the destructor is not expected to run and thus there is no contradiction.
61 // So construction still has to work.)
62
63 struct DtorList {
64 Dtor dtor;
65 void* obj;
66 DtorList* next;
67 };
68
69 // The linked list of thread-local destructors to run
70 __thread DtorList* dtors = nullptr;
71 // True if the destructors are currently scheduled to run on this thread
72 __thread bool dtors_alive = false;
73 // Used to trigger destructors on thread exit; value is ignored
74 std::__libcpp_tls_key dtors_key;
75
76 void run_dtors(void*) {
77 while (auto head = dtors) {
78 dtors = head->next;
79 head->dtor(head->obj);
80 ::free(head);
81 }
82
83 dtors_alive = false;
84 }
85
86 struct DtorsManager {
87 DtorsManager() {
88 // There is intentionally no matching std::__libcpp_tls_delete call, as
89 // __cxa_thread_atexit() may be called arbitrarily late (for example, from
90 // global destructors or atexit() handlers).
91 if (std::__libcpp_tls_create(&dtors_key, run_dtors) != 0) {
92 abort_message("std::__libcpp_tls_create() failed in __cxa_thread_atexit()");
93 }
94 }
95
96 ~DtorsManager() {
97 // std::__libcpp_tls_key destructors do not run on threads that call exit()
98 // (including when the main thread returns from main()), so we explicitly
99 // call the destructor here. This runs at exit time (potentially earlier
100 // if libc++abi is dlclose()'d). Any thread_locals initialized after this
101 // point will not be destroyed.
102 run_dtors(nullptr);
103 }
104 };
105} // namespace
106
107#endif // HAVE___CXA_THREAD_ATEXIT_IMPL
108
109extern "C" {
110
111 _LIBCXXABI_FUNC_VIS int __cxa_thread_atexit(Dtor dtor, void* obj, void* dso_symbol) throw() {
112#ifdef HAVE___CXA_THREAD_ATEXIT_IMPL
113 return __cxa_thread_atexit_impl(dtor, obj, dso_symbol);
114#else
115 if (__cxa_thread_atexit_impl) {
116 return __cxa_thread_atexit_impl(dtor, obj, dso_symbol);
117 } else {
118 // Initialize the dtors std::__libcpp_tls_key (uses __cxa_guard_*() for
119 // one-time initialization and __cxa_atexit() for destruction)
120 static DtorsManager manager;
121
122 if (!dtors_alive) {
123 if (std::__libcpp_tls_set(dtors_key, &dtors_key) != 0) {
124 return -1;
125 }
126 dtors_alive = true;
127 }
128
129 auto head = static_cast<DtorList*>(::malloc(sizeof(DtorList)));
130 if (!head) {
131 return -1;
132 }
133
134 head->dtor = dtor;
135 head->obj = obj;
136 head->next = dtors;
137 dtors = head;
138
139 return 0;
140 }
141#endif // HAVE___CXA_THREAD_ATEXIT_IMPL
142 }
143
144} // extern "C"
145} // namespace __cxxabiv1
146