When clearing plugin data at shutdown, wait for it to finish.

chrome/browser/plugin_data_remover.cc

 // Copyright (c) 2010 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 "chrome/browser/plugin_data_remover.h"
 
 #include "base/message_loop_proxy.h"
 #include "base/metrics/histogram.h"
 #include "base/version.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_thread.h"
 #include "chrome/browser/plugin_service.h"
 #include "chrome/common/plugin_messages.h"
 #include "webkit/plugins/npapi/plugin_group.h"
 #include "webkit/plugins/npapi/plugin_list.h"
 
 #if defined(OS_POSIX)
 #include "ipc/ipc_channel_posix.h"
 #endif
 
 namespace {
 const char* kFlashMimeType = "application/x-shockwave-flash";
 // TODO(bauerb): Update minimum required Flash version as soon as there is one
 // implementing the API.
 const char* kMinFlashVersion = "100";
 const int64 kRemovalTimeoutMs = 10000;
 const uint64 kClearAllData = 0;
 }  // namespace
 
 PluginDataRemover::PluginDataRemover()
     : is_removing_(false),
+      has_browser_ref_(false),
+      plugin_process_(NULL),
+      handle_(base::kNullProcessHandle),
       channel_(NULL) { }
 
 PluginDataRemover::~PluginDataRemover() {
   DCHECK(!is_removing_);
   if (channel_)
     BrowserThread::DeleteSoon(BrowserThread::IO, FROM_HERE, channel_);
 }
 
-void PluginDataRemover::StartRemoving(base::Time begin_time, Task* done_task) {
+void PluginDataRemover::StartRemoving(const base::Time& begin_time,
+                                      bool terminate,
+                                      Task* done_task) {
   DCHECK(!done_task_.get());
   DCHECK(!is_removing_);
   remove_start_time_ = base::Time::Now();
   begin_time_ = begin_time;
+  terminate_process_ = terminate;
 
   message_loop_ = base::MessageLoopProxy::CreateForCurrentThread();
   done_task_.reset(done_task);
   is_removing_ = true;
 
+  // Don't quit the main message loop until the process has started.
+  g_browser_process->AddRefModule();
+  has_browser_ref_ = true;
   AddRef();
   PluginService::GetInstance()->OpenChannelToPlugin(
       GURL(), kFlashMimeType, this);
 
   BrowserThread::PostDelayedTask(
       BrowserThread::IO,
       FROM_HERE,
       NewRunnableMethod(this, &PluginDataRemover::OnTimeout),
       kRemovalTimeoutMs);
 }
 
+void PluginDataRemover::WaitForPluginToExit() {
+  DCHECK(terminate_process_);
+  if (!is_removing_)
+    return;
+  DCHECK_NE(base::kNullProcessHandle, handle_);
+  base::Time start_time(base::Time::Now());
+  bool result = base::WaitForSingleProcess(handle_, kRemovalTimeoutMs);
+  UMA_HISTOGRAM_TIMES("ClearPluginData.wait_at_shutdown",
+                      base::Time::Now() - start_time);
+  UMA_HISTOGRAM_TIMES("ClearPluginData.time_at_shutdown",
+                      base::Time::Now() - remove_start_time_);
+  DCHECK(result) << "Error waiting for plugin process";
+}
+
 int PluginDataRemover::ID() {
   // Generate an ID for the browser process.
   return ChildProcessInfo::GenerateChildProcessUniqueId();
 }
 
 bool PluginDataRemover::OffTheRecord() {
   return false;
 }
 
 void PluginDataRemover::SetPluginInfo(
     const webkit::npapi::WebPluginInfo& info) {
 }
 
+void PluginDataRemover::OnPluginProcessFound(PluginProcessHost* process) {
+  plugin_process_ = process;
+}
+
 void PluginDataRemover::OnChannelOpened(const IPC::ChannelHandle& handle) {
   ConnectToChannel(handle);
+  DCHECK(has_browser_ref_);
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      NewRunnableMethod(this, &PluginDataRemover::ReleaseBrowserProcess));
+  has_browser_ref_ = false;
   Release();
 }
 
+void PluginDataRemover::ReleaseBrowserProcess() {
+  g_browser_process->ReleaseModule();
+}
+
 void PluginDataRemover::ConnectToChannel(const IPC::ChannelHandle& handle) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
 
   // If we timed out, don't bother connecting.
   if (!is_removing_)
     return;
 
   DCHECK(!channel_);
   channel_ = new IPC::Channel(handle, IPC::Channel::MODE_CLIENT, this);
   if (!channel_->Connect()) {
-    LOG(DFATAL) << "Couldn't connect to plugin";
+    NOTREACHED() << "Couldn't connect to plugin";
     SignalDone();
     return;
   }
 
   if (!channel_->Send(
           new PluginMsg_ClearSiteData(std::string(),
                                       kClearAllData,
                                       begin_time_))) {
-    LOG(DFATAL) << "Couldn't send ClearSiteData message";
+    NOTREACHED() << "Couldn't send ClearSiteData message";
     SignalDone();
+    return;
+  }
+
+  // Save the process handle so we can wait for the process to exit.
+  handle_ = plugin_process_->handle();
+  if (terminate_process_) {
+    plugin_process_->ForceShutdown();
   }
 }
 
 void PluginDataRemover::OnError() {
-  NOTREACHED() << "Couldn't open plugin channel";
+  LOG(DFATAL) << "Couldn't open plugin channel";
   SignalDone();
+  if (has_browser_ref_) {
+    BrowserThread::PostTask(
+        BrowserThread::UI, FROM_HERE,
+        NewRunnableMethod(this, &PluginDataRemover::ReleaseBrowserProcess));
+    has_browser_ref_ = false;
+  }
   Release();
 }
 
 void PluginDataRemover::OnClearSiteDataResult(bool success) {
-  if (!success)
-    LOG(DFATAL) << "ClearSiteData returned error";
+  LOG_IF(DFATAL, !success) << "ClearSiteData returned error";
   UMA_HISTOGRAM_TIMES("ClearPluginData.time",
                       base::Time::Now() - remove_start_time_);
   SignalDone();
 }
 
+void PluginDataRemover::OnPluginShuttingDown() {
+  VLOG(1) << "Plugin shutting down";
+}
+
 void PluginDataRemover::OnTimeout() {
-  NOTREACHED() << "Timed out";
+  if (is_removing_) {
+    LOG(DFATAL) << "Timed out";
+    if (has_browser_ref_) {
+      BrowserThread::PostTask(
+          BrowserThread::UI, FROM_HERE,
+          NewRunnableMethod(this, &PluginDataRemover::ReleaseBrowserProcess));
+      has_browser_ref_ = false;
+    }
+  }
   SignalDone();
 }
 
 bool PluginDataRemover::OnMessageReceived(const IPC::Message& msg) {
   IPC_BEGIN_MESSAGE_MAP(PluginDataRemover, msg)
     IPC_MESSAGE_HANDLER(PluginHostMsg_ClearSiteDataResult,
                         OnClearSiteDataResult)
+    IPC_MESSAGE_HANDLER(PluginHostMsg_PluginShuttingDown,
+                        OnPluginShuttingDown)
     IPC_MESSAGE_UNHANDLED_ERROR()
   IPC_END_MESSAGE_MAP()
 
   return true;
 }
 
 void PluginDataRemover::OnChannelError() {
-  LOG(DFATAL) << "Channel error";
-  SignalDone();
+  if (is_removing_) {
+    NOTREACHED() << "Channel error";
+    SignalDone();
+  }
 }
 
 void PluginDataRemover::SignalDone() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
   if (!is_removing_)
     return;
   is_removing_ = false;
   if (done_task_.get()) {
     message_loop_->PostTask(FROM_HERE, done_task_.release());
     message_loop_ = NULL;
@@ skipping 11 matching lines @@
     return false;
   }
   scoped_ptr<Version> version(
       webkit::npapi::PluginGroup::CreateVersionFromString(plugin.version));
   scoped_ptr<Version> min_version(
       Version::GetVersionFromString(kMinFlashVersion));
   return plugin.enabled &&
          version.get() &&
          min_version->CompareTo(*version) == -1;
 }