]>
Commit | Line | Data |
---|---|---|
688903eb | 1 | /* Copyright (C) 1999-2018 Free Software Foundation, Inc. |
c41041bc UD |
2 | This file is part of the GNU C Library. |
3 | ||
4 | The GNU C Library is free software; you can redistribute it and/or | |
41bdb6e2 AJ |
5 | modify it under the terms of the GNU Lesser General Public |
6 | License as published by the Free Software Foundation; either | |
7 | version 2.1 of the License, or (at your option) any later version. | |
c41041bc UD |
8 | |
9 | The GNU C Library is distributed in the hope that it will be useful, | |
10 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
41bdb6e2 | 12 | Lesser General Public License for more details. |
c41041bc | 13 | |
41bdb6e2 | 14 | You should have received a copy of the GNU Lesser General Public |
59ba27a6 PE |
15 | License along with the GNU C Library; if not, see |
16 | <http://www.gnu.org/licenses/>. */ | |
c41041bc | 17 | |
b6caa3a6 | 18 | #include <assert.h> |
a334319f | 19 | #include <stdlib.h> |
e054f494 | 20 | #include <stdint.h> |
b6caa3a6 | 21 | |
ec999b8e | 22 | #include <libc-lock.h> |
c41041bc | 23 | #include "exit.h" |
a3c88553 | 24 | #include <sysdep.h> |
c41041bc | 25 | |
6b87a564 UD |
26 | #undef __cxa_atexit |
27 | ||
26e70aec PP |
28 | /* See concurrency notes in stdlib/exit.h where this lock is declared. */ |
29 | __libc_lock_define_initialized (, __exit_funcs_lock) | |
30 | ||
610e67ed | 31 | |
c41041bc | 32 | int |
610e67ed UD |
33 | attribute_hidden |
34 | __internal_atexit (void (*func) (void *), void *arg, void *d, | |
35 | struct exit_function_list **listp) | |
c41041bc | 36 | { |
26e70aec PP |
37 | struct exit_function *new; |
38 | ||
39 | __libc_lock_lock (__exit_funcs_lock); | |
40 | new = __new_exitfn (listp); | |
c41041bc UD |
41 | |
42 | if (new == NULL) | |
26e70aec PP |
43 | { |
44 | __libc_lock_unlock (__exit_funcs_lock); | |
45 | return -1; | |
46 | } | |
c41041bc | 47 | |
a3c88553 UD |
48 | #ifdef PTR_MANGLE |
49 | PTR_MANGLE (func); | |
50 | #endif | |
5ef50d00 | 51 | new->func.cxa.fn = (void (*) (void *, int)) func; |
c41041bc UD |
52 | new->func.cxa.arg = arg; |
53 | new->func.cxa.dso_handle = d; | |
a3c88553 | 54 | new->flavor = ef_cxa; |
26e70aec | 55 | __libc_lock_unlock (__exit_funcs_lock); |
c41041bc UD |
56 | return 0; |
57 | } | |
610e67ed UD |
58 | |
59 | ||
60 | /* Register a function to be called by exit or when a shared library | |
61 | is unloaded. This function is only called from code generated by | |
62 | the C++ compiler. */ | |
63 | int | |
64 | __cxa_atexit (void (*func) (void *), void *arg, void *d) | |
65 | { | |
66 | return __internal_atexit (func, arg, d, &__exit_funcs); | |
67 | } | |
cb831245 | 68 | libc_hidden_def (__cxa_atexit) |
c08bc50a UD |
69 | |
70 | ||
c08bc50a UD |
71 | static struct exit_function_list initial; |
72 | struct exit_function_list *__exit_funcs = &initial; | |
a92b1166 | 73 | uint64_t __new_exitfn_called; |
c08bc50a | 74 | |
26e70aec | 75 | /* Must be called with __exit_funcs_lock held. */ |
c08bc50a | 76 | struct exit_function * |
610e67ed | 77 | __new_exitfn (struct exit_function_list **listp) |
c08bc50a | 78 | { |
b6caa3a6 | 79 | struct exit_function_list *p = NULL; |
c08bc50a | 80 | struct exit_function_list *l; |
b6caa3a6 | 81 | struct exit_function *r = NULL; |
c08bc50a UD |
82 | size_t i = 0; |
83 | ||
26e70aec PP |
84 | if (__exit_funcs_done) |
85 | /* Exit code is finished processing all registered exit functions, | |
86 | therefore we fail this registration. */ | |
87 | return NULL; | |
c08bc50a | 88 | |
610e67ed | 89 | for (l = *listp; l != NULL; p = l, l = l->next) |
c08bc50a | 90 | { |
b6caa3a6 UD |
91 | for (i = l->idx; i > 0; --i) |
92 | if (l->fns[i - 1].flavor != ef_free) | |
c08bc50a | 93 | break; |
b6caa3a6 UD |
94 | |
95 | if (i > 0) | |
c08bc50a UD |
96 | break; |
97 | ||
b6caa3a6 UD |
98 | /* This block is completely unused. */ |
99 | l->idx = 0; | |
c08bc50a UD |
100 | } |
101 | ||
b6caa3a6 | 102 | if (l == NULL || i == sizeof (l->fns) / sizeof (l->fns[0])) |
c08bc50a | 103 | { |
b6caa3a6 UD |
104 | /* The last entry in a block is used. Use the first entry in |
105 | the previous block if it exists. Otherwise create a new one. */ | |
106 | if (p == NULL) | |
c08bc50a | 107 | { |
b6caa3a6 UD |
108 | assert (l != NULL); |
109 | p = (struct exit_function_list *) | |
110 | calloc (1, sizeof (struct exit_function_list)); | |
111 | if (p != NULL) | |
112 | { | |
610e67ed UD |
113 | p->next = *listp; |
114 | *listp = p; | |
b6caa3a6 UD |
115 | } |
116 | } | |
c08bc50a | 117 | |
b6caa3a6 UD |
118 | if (p != NULL) |
119 | { | |
120 | r = &p->fns[0]; | |
121 | p->idx = 1; | |
c08bc50a UD |
122 | } |
123 | } | |
b6caa3a6 UD |
124 | else |
125 | { | |
126 | /* There is more room in the block. */ | |
127 | r = &l->fns[i]; | |
128 | l->idx = i + 1; | |
129 | } | |
c08bc50a UD |
130 | |
131 | /* Mark entry as used, but we don't know the flavor now. */ | |
b6caa3a6 | 132 | if (r != NULL) |
a92b1166 UD |
133 | { |
134 | r->flavor = ef_us; | |
135 | ++__new_exitfn_called; | |
136 | } | |
c08bc50a | 137 | |
b6caa3a6 | 138 | return r; |
c08bc50a | 139 | } |