chrome: respect --child-clean-exit.

content/browser/child_process_launcher.cc

 // Copyright (c) 2012 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/browser/child_process_launcher.h"
 
 #include <utility>  // For std::pair.
 
 #include "base/bind.h"
 #include "base/command_line.h"
@@ skipping 321 matching lines @@
       client_->OnProcessLaunched();
     } else {
       Terminate();
     }
   }
 
   void Terminate() {
     if (!process_.handle())
       return;
 
-    if (!terminate_child_on_shutdown_)
-      return;
-
-    // On Posix, EnsureProcessTerminated can lead to 2 seconds of sleep!  So
-    // don't this on the UI/IO threads.
-    BrowserThread::PostTask(
-        BrowserThread::PROCESS_LAUNCHER, FROM_HERE,
-        base::Bind(
-            &Context::TerminateInternal,
+    if (CommandLine::ForCurrentProcess()->
+        HasSwitch(switches::kWaitForChildrenBeforeExiting))
+      // When --wait-for-children-before-exiting is provided on the command
+      // line, we want the UI thread to wait for the zygote to exit before
+      // proceeding with the rest of the shut down.
+      TerminateAndWaitForExit(zygote_, process_.handle());
+    else
+      // On Posix, EnsureProcessTerminated can lead to 2 seconds of sleep!  So
+      // don't this on the UI/IO threads.
+      BrowserThread::PostTask(
+          BrowserThread::PROCESS_LAUNCHER, FROM_HERE,
+          base::Bind(
+              &Context::TerminateInternal,
 #if defined(OS_POSIX) && !defined(OS_MACOSX) && !defined(OS_ANDROID)
-            zygote_,
+              zygote_,
 #endif
-            process_.handle()));
+              process_.handle()));
     process_.set_handle(base::kNullProcessHandle);
   }
 
   static void SetProcessBackgrounded(base::ProcessHandle handle,
                                      bool background) {
     base::Process process(handle);
     process.SetProcessBackgrounded(background);
   }
 
+  static void TerminateAndWaitForExit(
+      bool zygote,
+      base::ProcessHandle handle) {
+    base::Process process(handle);
+    if (zygote) {
+      while(ZygoteHostImpl::GetInstance()->GetTerminationStatus(
+          handle,
+          false,
+          NULL) == base::TERMINATION_STATUS_STILL_RUNNING)
+        sleep(2);
+    }
+    process.Close();
+  }
+
   static void TerminateInternal(
 #if defined(OS_POSIX) && !defined(OS_MACOSX) && !defined(OS_ANDROID)
       bool zygote,
 #endif
       base::ProcessHandle handle) {
 #if defined(OS_ANDROID)
     LOG(INFO) << "ChromeProcess: Stopping process with handle " << handle;
     StopChildProcess(handle);
 #else
     base::Process process(handle);
@@ skipping 119 matching lines @@
           GetHandle(), background));
 }
 
 void ChildProcessLauncher::SetTerminateChildOnShutdown(
   bool terminate_on_shutdown) {
   if (context_.get())
     context_->set_terminate_child_on_shutdown(terminate_on_shutdown);
 }
 
 }  // namespace content