Implement __cxa_thread_atexit to support C++11 thread_local scope. The

interface is also made available as __cxa_thread_atexit_impl to satisfy the
needs of GNU libstdc++.

ok guenther@, millert@

2 files changed, 56 insertions, 2 deletions

diff --git a/lib/libc/stdlib/atexit.c b/lib/libc/stdlib/atexit.c
index 83cc1282a98..dc117a37524 100644
--- a/lib/libc/stdlib/atexit.c
+++ b/lib/libc/stdlib/atexit.c
@@ -1,4 +1,4 @@
-/*	$OpenBSD: atexit.c,v 1.24 2015/11/10 04:14:03 guenther Exp $ */
+/*	$OpenBSD: atexit.c,v 1.25 2017/12/05 13:45:31 kettenis Exp $ */
 /*
  * Copyright (c) 2002 Daniel Hartmeier
  * All rights reserved.
@@ -31,12 +31,24 @@
 
 #include <sys/types.h>
 #include <sys/mman.h>
+#include <dlfcn.h>
+#include <elf.h>
+#pragma weak _DYNAMIC
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 #include "atexit.h"
 #include "atfork.h"
 #include "thread_private.h"
+#include "tib.h"
+
+typeof(dlctl) dlctl asm("_dlctl") __attribute__((weak));
+
+struct thread_atexit_fn {
+	void (*func)(void *);
+	void *arg;
+	struct thread_atexit_fn *next;
+};
 
 struct atexit *__atexit;
 static int restartloop;
@@ -121,6 +133,43 @@ atexit(void (*fn)(void))
 }
 DEF_STRONG(atexit);
 
+__weak_alias(__cxa_thread_atexit, __cxa_thread_atexit_impl);
+
+int
+__cxa_thread_atexit_impl(void (*func)(void *), void *arg, void *dso)
+{
+	struct thread_atexit_fn *fnp;
+	struct tib *tib = TIB_GET();
+
+	fnp = calloc(1, sizeof(struct thread_atexit_fn));
+	if (fnp == NULL)
+		return -1;
+
+	if (_DYNAMIC)
+		dlctl(NULL, DL_REFERENCE, dso);
+
+	fnp->func = func;
+	fnp->arg = arg;
+	fnp->next = tib->tib_atexit;
+	tib->tib_atexit = fnp;
+
+	return 0;
+}
+DEF_STRONG(__cxa_thread_atexit_impl);
+
+void
+_thread_finalize(void)
+{
+	struct tib *tib = TIB_GET();
+
+	while (tib->tib_atexit) {
+		struct thread_atexit_fn *fnp = tib->tib_atexit;
+		tib->tib_atexit = fnp->next;
+		fnp->func(fnp->arg);
+		free(fnp);
+	}
+}
+
 /*
  * Call all handlers registered with __cxa_atexit() for the shared
  * object owning 'dso'.
@@ -134,6 +183,9 @@ __cxa_finalize(void *dso)
 	int n, pgsize = getpagesize();
 	static int call_depth;
 
+	if (dso == NULL)
+		_thread_finalize();
+
 	_ATEXIT_LOCK();
 	call_depth++;
 
diff --git a/lib/libc/stdlib/atexit.h b/lib/libc/stdlib/atexit.h
index e2e821de04b..d9bfed852f6 100644
--- a/lib/libc/stdlib/atexit.h
+++ b/lib/libc/stdlib/atexit.h
@@ -1,4 +1,4 @@
-/*	$OpenBSD: atexit.h,v 1.10 2015/10/25 18:01:24 guenther Exp $ */
+/*	$OpenBSD: atexit.h,v 1.11 2017/12/05 13:45:31 kettenis Exp $ */
 
 /*
  * Copyright (c) 2002 Daniel Hartmeier
@@ -46,7 +46,9 @@ extern struct atexit *__atexit;		/* points to head of LIFO stack */
 __END_HIDDEN_DECLS
 
 int	__cxa_atexit(void (*)(void *), void *, void *);
+int	__cxa_thread_atexit(void (*)(void *), void *, void *);
 void	__cxa_finalize(void *);
 
 PROTO_NORMAL(__cxa_atexit);
+PROTO_STD_DEPRECATED(__cxa_thread_atexit);
 PROTO_NORMAL(__cxa_finalize);