Add dso_handle parameter to __register_atfork

* forward.c (struct atfork): Add dso_handle field.
(__register_atfork): Add dso_handle parameter. Store it it dso_handle field.
* sysdeps/generic/pt-atfork.c (__dso_handle): New variable.
(pthread_atfork): Pass address of __dso_handle to __register_atfork.

2 files changed, 9 insertions, 2 deletions

diff --git a/forward.c b/forward.c
index 771b3ca..a2139a2 100644
--- a/forward.c
+++ b/forward.c
@@ -147,6 +147,7 @@ struct atfork {
   void (*prepare) (void);
   void (*parent) (void);
   void (*child) (void);
+  void *dso_handle;
   struct atfork *prev;
   struct atfork *next;
 };
@@ -219,7 +220,8 @@ int
 __register_atfork (
     void (*prepare) (void),
     void (*parent) (void),
-    void (*child) (void))
+    void (*child) (void),
+    void *dso_handle)
 {
   struct atfork *new = malloc (sizeof (*new));
   if (!new)
@@ -228,6 +230,7 @@ __register_atfork (
   new->prepare = prepare;
   new->parent = parent;
   new->child = child;
+  new->dso_handle = dso_handle;
   new->prev = NULL;
 
   __mutex_lock (&atfork_lock);
diff --git a/sysdeps/generic/pt-atfork.c b/sysdeps/generic/pt-atfork.c
index ce72e10..304d7ef 100644
--- a/sysdeps/generic/pt-atfork.c
+++ b/sysdeps/generic/pt-atfork.c
@@ -20,10 +20,14 @@
 #include <pthread.h>
 #include <pt-internal.h>
 
+/* This is defined by newer gcc version unique for each module.  */
+extern void *__dso_handle __attribute__ ((__weak__,
+					  __visibility__ ("hidden")));
+
 int
 pthread_atfork (void (*prepare) (void),
 		void (*parent) (void),
 		void (*child) (void))
 {
-  return __register_atfork (prepare, parent, child);
+  return __register_atfork (prepare, parent, child, &__dso_handle == NULL ? NULL : __dso_handle);
 }