Ticket #13111: fork-heap-locks-handling.patch

headers/private/libroot/libroot_private.h

@@ -36 +36 @@
 void __init_env_post_heap(void);
 status_t __init_heap(void);
 void __heap_terminate_after(void);
+void __heap_before_fork(void);
+void __heap_after_fork_child(void);
+void __heap_after_fork_parent(void);
 
 void __init_time(addr_t commPageTable);
 void __arch_init_time(struct real_time_data *data, bool setDefaults);

src/system/libroot/posix/malloc/heap.cpp

@@ -132 +132 @@
     , _magic(HEAP_MAGIC)
 #endif
 {
-    // Initialize the per-heap lock.
-    hoardLockInit(_lock, "hoard heap");
+    initLock();
 
     for (int i = 0; i < SUPERBLOCK_FULLNESS_GROUP; i++) {
         for (int j = 0; j < SIZE_CLASSES; j++) {

src/system/libroot/posix/malloc/heap.h

@@ -115 +115 @@
         // Unlock this heap.
         inline void unlock(void);
 
+        // Init this heap lock.
+        inline void initLock(void);
+
         // Set our index number (which heap we are).
         inline void setIndex(int i);
 
@@ -444 +447 @@
 }
 
 
+void
+hoardHeap::initLock(void)
+{
+    // Initialize the per-heap lock.
+    hoardLockInit(_lock, "hoard heap");
+}
+
+
 size_t
 hoardHeap::align(const size_t sz)
 {

src/system/libroot/posix/malloc/processheap.h

@@ -67 +67 @@
         // Get a thread heap index.
         inline int getHeapIndex(void);
 
+        // Get thread heap max.
+        inline int getMaxThreadHeaps(void);
+
         // Get the thread heap with index i.
         inline HEAPTYPE & getHeap(int i);
 
@@ -199 +202 @@
 }
 
 
+// Return the maximum number of heaps.
+
+int
+processHeap::getMaxThreadHeaps(void)
+{
+    return fMaxThreadHeaps;
+}
+
+
 superblock *
 processHeap::acquire(const int sizeclass, hoardHeap * dest)
 {

src/system/libroot/posix/malloc/wrapper.cpp

@@ -256 +256 @@
 }
 
 
+extern "C" void
+__heap_before_fork(void)
+{
+    static processHeap *pHeap = getAllocator();
+    for (int i = 0; i < pHeap->getMaxThreadHeaps(); i++)
+        pHeap->getHeap(i).lock();
+}
+
+
+extern "C" void
+__heap_after_fork_child(void)
+{
+    static processHeap *pHeap = getAllocator();
+    for (int i = 0; i < pHeap->getMaxThreadHeaps(); i++)
+        pHeap->getHeap(i).initLock();
+}
+
+
+extern "C" void
+__heap_after_fork_parent(void)
+{
+    static processHeap *pHeap = getAllocator();
+    for (int i = 0; i < pHeap->getMaxThreadHeaps(); i++)
+        pHeap->getHeap(i).unlock();
+}
+
+
 //  #pragma mark - public functions
 
 

src/system/libroot/posix/malloc_debug/malloc_debug_api.cpp

@@ -202 +202 @@
 }
 
 
+extern "C" void
+__heap_before_fork(void)
+{
+}
+
+
+extern "C" void
+__heap_after_fork_child(void)
+{
+}
+
+
+extern "C" void
+__heap_after_fork_parent(void)
+{
+}
+
+
 // #pragma mark - Public API
 
 

src/system/libroot/posix/unistd/fork.c

@@ -135 +135 @@
 
     // call preparation hooks
     call_fork_hooks(sPrepareHooks);
+    __heap_before_fork();
 
     thread = _kern_fork();
     if (thread < 0) {
@@ -156 +157 @@
             // calling the kernel.
         __gRuntimeLoader->reinit_after_fork();
         __reinit_pwd_backend_after_fork();
+        __heap_after_fork_child();
 
         call_fork_hooks(sChildHooks);
     } else {
         // we are the parent
+        __heap_after_fork_parent();
         call_fork_hooks(sParentHooks);
         mutex_unlock(&sForkLock);
     }