X-Git-Url: https://git.saurik.com/apple/libpthread.git/blobdiff_plain/f1a1da6cf65a9d0e6858678f6c259025cf5d27fd..11a104536abdb87bd79d71395f5941b03cd722c4:/src/pthread_atfork.c?ds=sidebyside diff --git a/src/pthread_atfork.c b/src/pthread_atfork.c index 724a7fb..edeec16 100644 --- a/src/pthread_atfork.c +++ b/src/pthread_atfork.c @@ -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, @@ -17,28 +17,24 @@ * 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 #include #include #include -PTHREAD_NOEXPORT void pthread_workqueue_atfork_child(void); -PTHREAD_NOEXPORT void __pthread_fork_child_internal(pthread_t); - int pthread_atfork(void (*prepare)(void), void (*parent)(void), void (*child)(void)) { int res = 0; size_t idx; pthread_globals_t globals = _pthread_globals(); - - OSSpinLockLock(&globals->pthread_atfork_lock); + + _PTHREAD_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); - OSSpinLockUnlock(&globals->pthread_atfork_lock); + _PTHREAD_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); - OSSpinLockLock(&globals->pthread_atfork_lock); + _PTHREAD_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. - OSSpinLockUnlock(&globals->pthread_atfork_lock); + _PTHREAD_UNLOCK(globals->pthread_atfork_lock); mach_vm_deallocate(mach_task_self(), storage, size); - OSSpinLockLock(&globals->pthread_atfork_lock); + _PTHREAD_LOCK(globals->pthread_atfork_lock); } } else { res = ENOMEM; @@ -79,27 +75,27 @@ 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; } - OSSpinLockUnlock(&globals->pthread_atfork_lock); + _PTHREAD_UNLOCK(globals->pthread_atfork_lock); return res; } // Called before the fork(2) system call is made in the parent process. // Iterate pthread_atfork prepare handlers. +// Called first in libSystem_atfork_prepare(). void -_pthread_fork_prepare(void) +_pthread_atfork_prepare_handlers(void) { pthread_globals_t globals = _pthread_globals(); - OSSpinLockLock(&globals->pthread_atfork_lock); - + _PTHREAD_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]; @@ -107,21 +103,38 @@ _pthread_fork_prepare(void) e->prepare(); } } +} + +// Take pthread-internal locks. +// Called last in libSystem_atfork_prepare(). +void +_pthread_atfork_prepare(void) +{ + pthread_globals_t globals = _pthread_globals(); - OSSpinLockLock(&globals->psaved_self_global_lock); + _PTHREAD_LOCK(globals->psaved_self_global_lock); globals->psaved_self = pthread_self(); - OSSpinLockLock(&globals->psaved_self->lock); + _PTHREAD_LOCK(globals->psaved_self->lock); } // Called after the fork(2) system call returns to the parent process. -// Iterate pthread_atfork parent handlers. +// 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(); - OSSpinLockUnlock(&globals->psaved_self->lock); - OSSpinLockUnlock(&globals->psaved_self_global_lock); + _PTHREAD_UNLOCK(globals->psaved_self->lock); + _PTHREAD_UNLOCK(globals->psaved_self_global_lock); +} + +// Iterate pthread_atfork parent handlers. +// Called last in libSystem_atfork_parent(). +void +_pthread_atfork_parent_handlers(void) +{ + pthread_globals_t globals = _pthread_globals(); size_t idx; for (idx = 0; idx < globals->atfork_count; ++idx) { @@ -130,25 +143,29 @@ _pthread_fork_parent(void) e->parent(); } } - OSSpinLockUnlock(&globals->pthread_atfork_lock); + _PTHREAD_UNLOCK(globals->pthread_atfork_lock); } // Called after the fork(2) system call returns to the new child process. // Clean up data structures of other threads which no longer exist in the child. // 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(); - globals->psaved_self_global_lock = OS_SPINLOCK_INIT; - __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(®istration_data); } // Iterate pthread_atfork child handlers. +// Called last in libSystem_atfork_child(). void -_pthread_fork_child_postinit(void) +_pthread_atfork_child_handlers(void) { pthread_globals_t globals = _pthread_globals(); size_t idx; @@ -158,5 +175,32 @@ _pthread_fork_child_postinit(void) e->child(); } } - globals->pthread_atfork_lock = OS_SPINLOCK_INIT; + _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(); +} + +void +_pthread_fork_child_postinit(void) +{ + _pthread_atfork_child_handlers(); }