diff --git a/lib/libc/stdlib/cxa_thread_atexit.c b/lib/libc/stdlib/cxa_thread_atexit.c new file mode 100644 index 000000000..f06d7f09a --- /dev/null +++ b/lib/libc/stdlib/cxa_thread_atexit.c @@ -0,0 +1,76 @@ +/* $NetBSD: cxa_thread_atexit.c,v 1.1 2017/07/11 15:21:35 joerg Exp $ */ + +/*- + * Copyright (c) 2017 Joerg Sonnenberger + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ + +#include +#include +#include + +#include "atexit.h" + +__dso_hidden bool __cxa_thread_atexit_used; + +struct cxa_dtor { + SLIST_ENTRY(cxa_dtor) link; + void *dso_symbol; + void *obj; + void (*dtor)(void *); +}; + +/* Assumes NULL initialization. */ +static /* __thread */ SLIST_HEAD(, cxa_dtor) cxa_dtors = SLIST_HEAD_INITIALIZER(cxa_dstors); + +void +__cxa_thread_run_atexit(void) +{ + struct cxa_dtor *entry; + + while ((entry = SLIST_FIRST(&cxa_dtors)) != NULL) { + SLIST_REMOVE_HEAD(&cxa_dtors, link); + (*entry->dtor)(entry->obj); + free(entry); + } +} + +int __cxa_thread_atexit(void (*)(void *), void *, void *); + +int +__cxa_thread_atexit_impl(void (*dtor)(void *), void *obj, void *dso_symbol) +{ + struct cxa_dtor *entry; + + __cxa_thread_atexit_used = true; + + entry = malloc(sizeof(*entry)); + if (entry == NULL) + return -1; + entry->dso_symbol = dso_symbol; + entry->obj = obj; + entry->dtor = dtor; + SLIST_INSERT_HEAD(&cxa_dtors, entry, link); + return 0; +} diff --git a/lib/libc/stdlib/exit.c b/lib/libc/stdlib/exit.c index 97be5156b..4c07c4423 100644 --- a/lib/libc/stdlib/exit.c +++ b/lib/libc/stdlib/exit.c @@ -32,6 +32,8 @@ #include #include #include +#include "atexit.h" +#include "reentrant.h" void (*__cleanup)(void); @@ -39,6 +41,9 @@ void (*__cleanup)(void); * Exit, flushing stdio buffers if necessary. */ void exit(int status) { + if (__cxa_thread_atexit_used) + __cxa_thread_run_atexit(); + __cxa_finalize(NULL); if (__cleanup) (*__cleanup)(); _exit(status);