]> git.saurik.com Git - apple/libpthread.git/blobdiff - src/pthread_atfork.c
libpthread-454.100.8.tar.gz
[apple/libpthread.git] / src / pthread_atfork.c
index ec174df1d5907051f79ee38f5aa01cb5ae23f3d6..150c50452f6e7666d0d73dac440fac1bcffdc8cf 100644 (file)
@@ -2,14 +2,14 @@
  * Copyright (c) 1999, 2012 Apple Inc. All rights reserved.
  *
  * @APPLE_LICENSE_HEADER_START@
- * 
+ *
  * This file contains Original Code and/or Modifications of Original Code
  * as defined in and that are subject to the Apple Public Source License
  * Version 2.0 (the 'License'). You may not use this file except in
  * compliance with the License. Please obtain a copy of the License at
  * http://www.opensource.apple.com/apsl/ and read it before using this
  * file.
- * 
+ *
  * The Original Code and all software distributed under the License are
  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  * Please see the License for the specific language governing rights and
  * limitations under the License.
- * 
+ *
  * @APPLE_LICENSE_HEADER_END@
  */
 
 #include "internal.h"
 
-#include <libkern/OSAtomic.h>
 #include <mach/mach_init.h>
 #include <mach/mach_vm.h>
-#include <platform/compat.h>
-
-PTHREAD_NOEXPORT void pthread_workqueue_atfork_child(void);
-PTHREAD_NOEXPORT void __pthread_fork_child_internal(pthread_t);
 
+#if !VARIANT_DYLD
 int
 pthread_atfork(void (*prepare)(void), void (*parent)(void), void (*child)(void))
 {
        int res = 0;
        size_t idx;
        pthread_globals_t globals = _pthread_globals();
-       
-       _PTHREAD_LOCK(globals->pthread_atfork_lock);
+
+       _pthread_lock_lock(&globals->pthread_atfork_lock);
        idx = globals->atfork_count++;
 
        if (idx == 0) {
@@ -49,7 +45,7 @@ pthread_atfork(void (*prepare)(void), void (*parent)(void), void (*child)(void))
                kern_return_t kr;
                mach_vm_address_t storage = 0;
                mach_vm_size_t size = PTHREAD_ATFORK_MAX * sizeof(struct pthread_atfork_entry);
-               _PTHREAD_UNLOCK(globals->pthread_atfork_lock);
+               _pthread_lock_unlock(&globals->pthread_atfork_lock);
                kr = mach_vm_map(mach_task_self(),
                                 &storage,
                                 size,
@@ -61,7 +57,7 @@ pthread_atfork(void (*prepare)(void), void (*parent)(void), void (*child)(void))
                                 VM_PROT_DEFAULT,
                                 VM_PROT_ALL,
                                 VM_INHERIT_DEFAULT);
-               _PTHREAD_LOCK(globals->pthread_atfork_lock);
+               _pthread_lock_lock(&globals->pthread_atfork_lock);
                if (kr == KERN_SUCCESS) {
                        if (globals->atfork == globals->atfork_storage) {
                                globals->atfork = storage;
@@ -69,9 +65,9 @@ pthread_atfork(void (*prepare)(void), void (*parent)(void), void (*child)(void))
                                bzero(globals->atfork_storage, sizeof(globals->atfork_storage));
                        } else {
                                // Another thread did vm_map first.
-                               _PTHREAD_UNLOCK(globals->pthread_atfork_lock);
+                               _pthread_lock_unlock(&globals->pthread_atfork_lock);
                                mach_vm_deallocate(mach_task_self(), storage, size);
-                               _PTHREAD_LOCK(globals->pthread_atfork_lock);
+                               _pthread_lock_lock(&globals->pthread_atfork_lock);
                        }
                } else {
                        res = ENOMEM;
@@ -79,14 +75,14 @@ pthread_atfork(void (*prepare)(void), void (*parent)(void), void (*child)(void))
        } else if (idx >= PTHREAD_ATFORK_MAX) {
                res = ENOMEM;
        }
+
        if (res == 0) {
                struct pthread_atfork_entry *e = &globals->atfork[idx];
                e->prepare = prepare;
                e->parent = parent;
                e->child = child;
        }
-       _PTHREAD_UNLOCK(globals->pthread_atfork_lock);
+       _pthread_lock_unlock(&globals->pthread_atfork_lock);
 
        return res;
 }
@@ -99,7 +95,7 @@ _pthread_atfork_prepare_handlers(void)
 {
        pthread_globals_t globals = _pthread_globals();
 
-       _PTHREAD_LOCK(globals->pthread_atfork_lock);
+       _pthread_lock_lock(&globals->pthread_atfork_lock);
        size_t idx;
        for (idx = globals->atfork_count; idx > 0; --idx) {
                struct pthread_atfork_entry *e = &globals->atfork[idx-1];
@@ -112,25 +108,25 @@ _pthread_atfork_prepare_handlers(void)
 // Take pthread-internal locks.
 // Called last in libSystem_atfork_prepare().
 void
-_pthread_fork_prepare(void)
+_pthread_atfork_prepare(void)
 {
        pthread_globals_t globals = _pthread_globals();
 
-       _PTHREAD_LOCK(globals->psaved_self_global_lock);
+       _pthread_lock_lock(&globals->psaved_self_global_lock);
        globals->psaved_self = pthread_self();
-       _PTHREAD_LOCK(globals->psaved_self->lock);
+       _pthread_lock_lock(&globals->psaved_self->lock);
 }
 
 // Called after the fork(2) system call returns to the parent process.
 // Release pthread-internal locks
 // Called first in libSystem_atfork_parent().
 void
-_pthread_fork_parent(void)
+_pthread_atfork_parent(void)
 {
        pthread_globals_t globals = _pthread_globals();
 
-       _PTHREAD_UNLOCK(globals->psaved_self->lock);
-       _PTHREAD_UNLOCK(globals->psaved_self_global_lock);
+       _pthread_lock_unlock(&globals->psaved_self->lock);
+       _pthread_lock_unlock(&globals->psaved_self_global_lock);
 }
 
 // Iterate pthread_atfork parent handlers.
@@ -147,7 +143,7 @@ _pthread_atfork_parent_handlers(void)
                        e->parent();
                }
        }
-       _PTHREAD_UNLOCK(globals->pthread_atfork_lock);
+       _pthread_lock_unlock(&globals->pthread_atfork_lock);
 }
 
 // Called after the fork(2) system call returns to the new child process.
@@ -155,13 +151,15 @@ _pthread_atfork_parent_handlers(void)
 // Make the current thread the main thread.
 // Called first in libSystem_atfork_child() (after _dyld_fork_child)
 void
-_pthread_fork_child(void)
+_pthread_atfork_child(void)
 {
        pthread_globals_t globals = _pthread_globals();
-       _PTHREAD_LOCK_INIT(globals->psaved_self_global_lock);
-       __pthread_fork_child_internal(globals->psaved_self);
+       _pthread_lock_init(&globals->psaved_self_global_lock);
        __is_threaded = 0;
-       pthread_workqueue_atfork_child();
+       _pthread_main_thread_postfork_init(globals->psaved_self);
+
+       struct _pthread_registration_data registration_data;
+       _pthread_bsdthread_init(&registration_data);
 }
 
 // Iterate pthread_atfork child handlers.
@@ -177,12 +175,33 @@ _pthread_atfork_child_handlers(void)
                        e->child();
                }
        }
-       _PTHREAD_LOCK_INIT(globals->pthread_atfork_lock);
+       _pthread_lock_init(&globals->pthread_atfork_lock);
+}
+
+// Preserve legacy symbols for older iOS simulators
+void
+_pthread_fork_prepare(void)
+{
+       _pthread_atfork_prepare_handlers();
+       _pthread_atfork_prepare();
+}
+
+void
+_pthread_fork_parent(void)
+{
+       _pthread_atfork_parent();
+       _pthread_atfork_parent_handlers();
+}
+
+void
+_pthread_fork_child(void)
+{
+       _pthread_atfork_child();
 }
 
-// Preserve legacy symbol in case somebody depends on it
 void
 _pthread_fork_child_postinit(void)
 {
        _pthread_atfork_child_handlers();
 }
+#endif // !VARIANT_DYLD