Changeset 18989 in vbox for trunk/src/VBox/Runtime/r3

Timestamp:

Apr 17, 2009 1:01:30 PM (16 years ago)

Author:

vboxsync

Message:

iprt/process: update the cached process ID on fork

Location:

trunk/src/VBox/Runtime/r3

Files:

• init.cpp (modified) (2 diffs)
• process.cpp (modified) (2 diffs)

trunk/src/VBox/Runtime/r3/init.cpp

@@ -51 +51 @@
 #include <iprt/string.h>
 #include <iprt/param.h>
+#include <iprt/process.h>
 #if !defined(IN_GUEST) && !defined(RT_NO_GIP)
 # include <iprt/file.h>
@@ -231 +232 @@
      * The Process ID.
      */
-#ifdef _MSC_VER
-    g_ProcessSelf = _getpid(); /* crappy ansi compiler */
-#else
-    g_ProcessSelf = getpid();
-#endif
+    /* The first call to RTProcSelf lazily initialises the cached pid, and
+     * on posix systems also sets a callback to update the cache on fork.
+     * We just do a dummy call to it here rather than duplicating
+     * initialisation code. */
+    /** @todo since we do lazy initialisation anyway, do we really also need
+     * to do it explicitly? */
+    RTProcSelf();
 
     /*

trunk/src/VBox/Runtime/r3/process.cpp

@@ -47 +47 @@
 #endif
 
+#if !defined (RT_OS_OS2) && !defined(RT_OS_WINDOWS)
+# include <pthread.h>
+#endif
+
+#if !defined (RT_OS_OS2) && !defined(RT_OS_WINDOWS)
+/** Update the cached pid on fork on posix systems */
+static void rtProcPThreadAtFork(void)
+{
+    g_ProcessSelf = getpid();
+}
+#endif
 
 /**
@@ -60 +71 @@
 
     /* lazy init. */
+#if !defined (RT_OS_OS2) && !defined(RT_OS_WINDOWS)
+    /* Update the cached pid on fork on posix systems */
+    if (pthread_atfork(NULL, NULL, rtProcPThreadAtFork) != 0)
+        return 0;
+#endif
 #ifdef _MSC_VER
     Self = _getpid(); /* crappy ansi compiler */