Porting in chrome/

chrome/common/process_watcher.cc

 // Copyright (c) 2006-2008 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/common/process_watcher.h"
 
+#include "base/basictypes.h"
 #include "base/message_loop.h"
-#include "base/object_watcher.h"
+#include "base/process.h"
+#include "base/process_util.h"
 #include "base/sys_info.h"
+#include "base/timer.h"
+#include "base/worker_pool.h"
 #include "chrome/app/result_codes.h"
 #include "chrome/common/env_vars.h"
 
 // Maximum amount of time (in milliseconds) to wait for the process to exit.
 static const int kWaitInterval = 2000;
 
 namespace {
 
-class TimerExpiredTask : public Task, public base::ObjectWatcher::Delegate {
+class TerminatorTask : public Task {
  public:
-  explicit TimerExpiredTask(base::ProcessHandle process) : process_(process) {
-    watcher_.StartWatching(process_, this);
+  explicit TerminatorTask(base::ProcessHandle process) : process_(process) {
+    timer_.Start(base::TimeDelta::FromMilliseconds(kWaitInterval),
+                 this, &TerminatorTask::KillProcess);
   }
 
-  virtual ~TimerExpiredTask() {
+  virtual ~TerminatorTask() {
     if (process_) {
       KillProcess();
-      DCHECK(!process_) << "Make sure to close the handle.";
+      DCHECK(!process_);
     }
   }
 
-  // Task ---------------------------------------------------------------------
-
   virtual void Run() {
+    base::WaitForSingleProcess(process_, kWaitInterval);
+    timer_.Stop();
     if (process_)
       KillProcess();
   }
 
-  // MessageLoop::Watcher -----------------------------------------------------
-
-  virtual void OnObjectSignaled(HANDLE object) {
-    // When we're called from KillProcess, the ObjectWatcher may still be
-    // watching.  the process handle, so make sure it has stopped.
-    watcher_.StopWatching();
-
-    CloseHandle(process_);
-    process_ = NULL;
-  }
-
  private:
   void KillProcess() {
     if (base::SysInfo::HasEnvVar(env_vars::kHeadless)) {
-     // If running the distributed tests, give the renderer a little time
-     // to figure out that the channel is shutdown and unwind.
-     if (WaitForSingleObject(process_, kWaitInterval) == WAIT_OBJECT_0) {
-       OnObjectSignaled(process_);
-       return;
-     }
+      // If running the distributed tests, give the renderer a little time
+      // to figure out that the channel is shutdown and unwind.
+      if (base::WaitForSingleProcess(process_, kWaitInterval)) {
+        Cleanup();
+        return;
+      }
     }
 
     // OK, time to get frisky.  We don't actually care when the process
-    // terminates.  We just care that it eventually terminates, and that's what
-    // TerminateProcess should do for us. Don't check for the result code since
-    // it fails quite often. This should be investigated eventually.
-    TerminateProcess(process_, ResultCodes::HUNG);
+    // terminates.  We just care that it eventually terminates.
+    base::KillProcess(base::Process(process_).pid(),
+                      ResultCodes::HUNG,
+                      false /* don't wait */);
 
-    // Now, just cleanup as if the process exited normally.
-    OnObjectSignaled(process_);
+    Cleanup();
+  }
+
+  void Cleanup() {
+    timer_.Stop();
+    base::CloseProcessHandle(process_);
+    process_ = NULL;
   }
 
   // The process that we are watching.
   base::ProcessHandle process_;
 
-  base::ObjectWatcher watcher_;
+  base::OneShotTimer<TerminatorTask> timer_;
 
-  DISALLOW_EVIL_CONSTRUCTORS(TimerExpiredTask);
+  DISALLOW_COPY_AND_ASSIGN(TerminatorTask);
 };
 
 }  // namespace
 
 // static
 void ProcessWatcher::EnsureProcessTerminated(base::ProcessHandle process) {
-  DCHECK(process != GetCurrentProcess());
+  DCHECK(base::GetProcId(process) != base::GetCurrentProcId());
 
-  // If already signaled, then we are done!
-  if (WaitForSingleObject(process, 0) == WAIT_OBJECT_0) {
-    CloseHandle(process);
+  // Check if the process has already exited.
+  if (base::WaitForSingleProcess(process, 0)) {
+    base::CloseProcessHandle(process);
     return;
   }
 
-  MessageLoop::current()->PostDelayedTask(FROM_HERE,
-                                          new TimerExpiredTask(process),
-                                          kWaitInterval);
+  WorkerPool::PostTask(FROM_HERE, new TerminatorTask(process), true);
 }
-