Allow multiple ChildProcess objects in one process for the --single-process case, when we have bo...

content/common/child_process.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/common/child_process.h"
 
 #if defined(OS_POSIX)
 #include <signal.h>  // For SigUSR1Handler below.
 #endif
 
+#include "base/lazy_instance.h"
 #include "base/message_loop.h"
 #include "base/process_util.h"
 #include "base/string_number_conversions.h"
 #include "base/threading/thread.h"
+#include "base/threading/thread_local.h"
 #include "base/utf_string_conversions.h"
 #include "content/common/child_thread.h"
 
 #if defined(OS_POSIX)
 static void SigUSR1Handler(int signal) { }
 #endif
 
-ChildProcess* ChildProcess::child_process_;
+namespace {
+static base::LazyInstance<base::ThreadLocalPointer<ChildProcess> > lazy_tls(
+    base::LINKER_INITIALIZED);
+}
 
 ChildProcess::ChildProcess()
     : ref_count_(0),
       shutdown_event_(true, false),
       io_thread_("Chrome_ChildIOThread") {
-  DCHECK(!child_process_);
-  child_process_ = this;
+  DCHECK(!lazy_tls.Pointer()->Get());
+  lazy_tls.Pointer()->Set(this);
 
   io_thread_.StartWithOptions(base::Thread::Options(MessageLoop::TYPE_IO, 0));
 }
 
 ChildProcess::~ChildProcess() {
-  DCHECK(child_process_ == this);
+  DCHECK(lazy_tls.Pointer()->Get() == this);
 
   // Signal this event before destroying the child process.  That way all
   // background threads can cleanup.
   // For example, in the renderer the RenderThread instances will be able to
   // notice shutdown before the render process begins waiting for them to exit.
   shutdown_event_.Signal();
 
-  // Kill the main thread object before nulling child_process_, since
+  // Kill the main thread object before nulling lazy_tls, since
   // destruction code might depend on it.
   main_thread_.reset();
 
-  child_process_ = NULL;
+  lazy_tls.Pointer()->Set(NULL);
+}
+
+ChildProcess* ChildProcess::current() {
+  return lazy_tls.Pointer()->Get();
 }
 
 ChildThread* ChildProcess::main_thread() {
   return main_thread_.get();
 }
 
 void ChildProcess::set_main_thread(ChildThread* thread) {
   main_thread_.reset(thread);
 }
 
 void ChildProcess::AddRefProcess() {
   DCHECK(!main_thread_.get() ||  // null in unittests.
          MessageLoop::current() == main_thread_->message_loop());
   ref_count_++;
 }
 
 void ChildProcess::ReleaseProcess() {
   DCHECK(!main_thread_.get() ||  // null in unittests.
          MessageLoop::current() == main_thread_->message_loop());
   DCHECK(ref_count_);
-  DCHECK(child_process_);
   if (--ref_count_)
     return;
 
   if (main_thread_.get())  // null in unittests.
     main_thread_->OnProcessFinalRelease();
 }
 
 base::WaitableEvent* ChildProcess::GetShutDownEvent() {
-  DCHECK(child_process_);
-  return &child_process_->shutdown_event_;
+  return &lazy_tls.Pointer()->Get()->shutdown_event_;
 }
 
 void ChildProcess::WaitForDebugger(const std::string& label) {
 #if defined(OS_WIN)
 #if defined(GOOGLE_CHROME_BUILD)
   std::string title = "Google Chrome";
 #else  // CHROMIUM_BUILD
   std::string title = "Chromium";
 #endif  // CHROMIUM_BUILD
   title += " ";
@@ skipping 13 matching lines @@
              << ") paused waiting for debugger to attach @ pid";
   // Install a signal handler so that pause can be woken.
   struct sigaction sa;
   memset(&sa, 0, sizeof(sa));
   sa.sa_handler = SigUSR1Handler;
   sigaction(SIGUSR1, &sa, NULL);
 
   pause();
 #endif  // defined(OS_POSIX)
 }