[Chromoting] Refactored worker process launching code and speeded up the desktop process launch.

remoting/host/win/worker_process_launcher_unittest.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 "base/basictypes.h"
 #include "base/bind.h"
 #include "base/memory/ref_counted.h"
 #include "base/message_loop.h"
 #include "base/win/scoped_handle.h"
+#include "base/win/scoped_process_information.h"
 #include "ipc/ipc_channel.h"
 #include "ipc/ipc_channel_proxy.h"
 #include "ipc/ipc_listener.h"
 #include "ipc/ipc_message.h"
 #include "remoting/base/auto_thread_task_runner.h"
 #include "remoting/host/chromoting_messages.h"
 #include "remoting/host/host_exit_codes.h"
 #include "remoting/host/win/launch_process_with_token.h"
 #include "remoting/host/win/worker_process_launcher.h"
 #include "remoting/host/worker_process_ipc_delegate.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gmock_mutant.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::win::ScopedHandle;
 using testing::_;
 using testing::AnyNumber;
 using testing::CreateFunctor;
 using testing::DoAll;
 using testing::Expectation;
 using testing::Invoke;
 using testing::InvokeWithoutArgs;
 using testing::Return;
-using testing::ReturnPointee;
 
 namespace remoting {
 
 namespace {
 
 const char kIpcSecurityDescriptor[] = "D:(A;;GA;;;AU)";
 
 class MockProcessLauncherDelegate : public WorkerProcessLauncher::Delegate {
  public:
   MockProcessLauncherDelegate() {}
   virtual ~MockProcessLauncherDelegate() {}
 
-  // IPC::Sender implementation.
-  MOCK_METHOD1(Send, bool(IPC::Message*));
-
   // WorkerProcessLauncher::Delegate interface.
+  MOCK_METHOD1(LaunchProcess, void(WorkerProcessLauncher*));
+  MOCK_METHOD1(Send, void(IPC::Message*));
   MOCK_METHOD0(CloseChannel, void());
-  MOCK_CONST_METHOD0(GetProcessId, DWORD());
-  MOCK_CONST_METHOD1(IsPermanentError, bool(int));
-  MOCK_METHOD1(KillProcess, void(DWORD));
-  MOCK_METHOD2(LaunchProcess, bool(IPC::Listener*, ScopedHandle*));
+  MOCK_METHOD0(KillProcess, void());
 
  private:
   DISALLOW_COPY_AND_ASSIGN(MockProcessLauncherDelegate);
 };
 
 class MockIpcDelegate : public WorkerProcessIpcDelegate {
  public:
   MockIpcDelegate() {}
   virtual ~MockIpcDelegate() {}
 
@@ skipping 27 matching lines @@
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   EXPECT_TRUE(handled);
 
   return handled;
 }
 
 }  // namespace
 
-class WorkerProcessLauncherTest : public testing::Test {
+class WorkerProcessLauncherTest
+    : public testing::Test,
+      public IPC::Listener {
  public:
   WorkerProcessLauncherTest();
   virtual ~WorkerProcessLauncherTest();
 
   virtual void SetUp() OVERRIDE;
   virtual void TearDown() OVERRIDE;
 
+  // IPC::Listener implementation.
+  virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
+  virtual void OnChannelConnected(int32 peer_pid) OVERRIDE;
+  virtual void OnChannelError() OVERRIDE;
+
   // WorkerProcessLauncher::Delegate mocks
-  void KillProcess(DWORD exit_code);
-  bool LaunchProcess(IPC::Listener* delegate,
-                     ScopedHandle* process_exit_event_out);
-  bool LaunchProcessAndConnect(IPC::Listener* delegate,
-                               ScopedHandle* process_exit_event_out);
-  bool FailLaunchAndStopWorker(IPC::Listener* delegate,
-                               ScopedHandle* process_exit_event_out);
+  void LaunchProcess(
+      WorkerProcessLauncher* event_handler);
+  void LaunchProcessAndConnect(
+      WorkerProcessLauncher* event_handler);
+  void FailLaunchAndStopWorker(
+      WorkerProcessLauncher* event_handler);
+  void KillProcess();
+
+  void TerminateWorker(DWORD exit_code);
 
   // Connects the client end of the channel (the worker process's end).
   void ConnectClient();
 
   // Disconnects the client end of the channel.
   void DisconnectClient();
 
   // Disconnects the server end of the channel (the launcher's end).
   void DisconnectServer();
 
   // Sends a message to the worker process.
-  bool SendToProcess(IPC::Message* message);
+  void SendToProcess(IPC::Message* message);
 
   // Sends a fake message to the launcher.
   void SendFakeMessageToLauncher();
 
   // Requests the worker to crash.
   void CrashWorker();
 
   // Starts the worker.
   void StartWorker();
 
   // Stops the worker.
   void StopWorker();
 
   // Quits |message_loop_|.
   void QuitMainMessageLoop();
 
  protected:
+  void DoLaunchProcess();
+
   base::MessageLoop message_loop_;
   scoped_refptr<AutoThreadTaskRunner> task_runner_;
 
   // Receives messages sent to the worker process.
   MockWorkerListener client_listener_;
 
   // Receives messages sent from the worker process.
   MockIpcDelegate server_listener_;
 
   // Implements WorkerProcessLauncher::Delegate.
   scoped_ptr<MockProcessLauncherDelegate> launcher_delegate_;
 
   // The name of the IPC channel.
   std::string channel_name_;
 
   // Client and server ends of the IPC channel.
   scoped_ptr<IPC::ChannelProxy> channel_client_;
   scoped_ptr<IPC::ChannelProxy> channel_server_;
 
-  // Returned as the worker process PID to the launcher.
-  DWORD client_pid_;
+  WorkerProcessLauncher* event_handler_;
 
   // The worker process launcher.
   scoped_ptr<WorkerProcessLauncher> launcher_;
 
-  // The event signalling termination of the worker process.
-  ScopedHandle process_exit_event_;
-
-  // True if a permanent error condition should be emulated.
-  bool permanent_error_;
+  // An event that is used to emulate the worker process's handle.
+  ScopedHandle worker_process_;
 };
 
 WorkerProcessLauncherTest::WorkerProcessLauncherTest()
     : message_loop_(base::MessageLoop::TYPE_IO),
-      client_pid_(GetCurrentProcessId()),
-      permanent_error_(false) {}
+      event_handler_(NULL) {
+}
 
 WorkerProcessLauncherTest::~WorkerProcessLauncherTest() {
 }
 
 void WorkerProcessLauncherTest::SetUp() {
   task_runner_ = new AutoThreadTaskRunner(
       message_loop_.message_loop_proxy(),
       base::Bind(&WorkerProcessLauncherTest::QuitMainMessageLoop,
                  base::Unretained(this)));
 
   // Set up process launcher delegate
   launcher_delegate_.reset(new MockProcessLauncherDelegate());
   EXPECT_CALL(*launcher_delegate_, Send(_))
       .Times(AnyNumber())
       .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::SendToProcess));
   EXPECT_CALL(*launcher_delegate_, CloseChannel())
       .Times(AnyNumber())
       .WillRepeatedly(Invoke(this,
                              &WorkerProcessLauncherTest::DisconnectServer));
-  EXPECT_CALL(*launcher_delegate_, GetProcessId())
-      .Times(AnyNumber())
-      .WillRepeatedly(ReturnPointee(&client_pid_));
-  EXPECT_CALL(*launcher_delegate_, IsPermanentError(_))
-      .Times(AnyNumber())
-      .WillRepeatedly(ReturnPointee(&permanent_error_));
-  EXPECT_CALL(*launcher_delegate_, KillProcess(_))
+  EXPECT_CALL(*launcher_delegate_, KillProcess())
       .Times(AnyNumber())
       .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::KillProcess));
 
   // Set up IPC delegate.
   EXPECT_CALL(server_listener_, OnMessageReceived(_))
       .Times(0);
 }
 
 void WorkerProcessLauncherTest::TearDown() {
 }
 
-void WorkerProcessLauncherTest::KillProcess(DWORD exit_code) {
-  BOOL result = SetEvent(process_exit_event_);
-  EXPECT_TRUE(result);
+bool WorkerProcessLauncherTest::OnMessageReceived(const IPC::Message& message) {
+  return event_handler_->OnMessageReceived(message);
 }
 
-bool WorkerProcessLauncherTest::LaunchProcess(
-    IPC::Listener* delegate,
-    ScopedHandle* process_exit_event_out) {
-  process_exit_event_.Set(CreateEvent(NULL, TRUE, FALSE, NULL));
-  if (!process_exit_event_.IsValid())
-    return false;
-
-  channel_name_ = IPC::Channel::GenerateUniqueRandomChannelID();
-  ScopedHandle pipe;
-  if (!CreateIpcChannel(channel_name_, kIpcSecurityDescriptor, &pipe)) {
-    return false;
-  }
-
-  // Wrap the pipe into an IPC channel.
-  channel_server_.reset(new IPC::ChannelProxy(
-      IPC::ChannelHandle(pipe),
-      IPC::Channel::MODE_SERVER,
-      delegate,
-      task_runner_));
-
-  return DuplicateHandle(GetCurrentProcess(),
-                         process_exit_event_,
-                         GetCurrentProcess(),
-                         process_exit_event_out->Receive(),
-                         0,
-                         FALSE,
-                         DUPLICATE_SAME_ACCESS) != FALSE;
+void WorkerProcessLauncherTest::OnChannelConnected(int32 peer_pid) {
+  event_handler_->OnChannelConnected(peer_pid);
 }
 
-bool WorkerProcessLauncherTest::LaunchProcessAndConnect(
-    IPC::Listener* delegate,
-    ScopedHandle* process_exit_event_out) {
-  if (!LaunchProcess(delegate, process_exit_event_out))
-    return false;
+void WorkerProcessLauncherTest::OnChannelError() {
+  event_handler_->OnChannelError();
+}
+
+void WorkerProcessLauncherTest::LaunchProcess(
+    WorkerProcessLauncher* event_handler) {
+  EXPECT_FALSE(event_handler_);
+  event_handler_ = event_handler;
+
+  DoLaunchProcess();
+}
+
+void WorkerProcessLauncherTest::LaunchProcessAndConnect(
+    WorkerProcessLauncher* event_handler) {
+  EXPECT_FALSE(event_handler_);
+  event_handler_ = event_handler;
+
+  DoLaunchProcess();
 
   task_runner_->PostTask(
       FROM_HERE,
       base::Bind(&WorkerProcessLauncherTest::ConnectClient,
                  base::Unretained(this)));
-  return true;
 }
 
-bool WorkerProcessLauncherTest::FailLaunchAndStopWorker(
-    IPC::Listener* delegate,
-    ScopedHandle* process_exit_event_out) {
+void WorkerProcessLauncherTest::FailLaunchAndStopWorker(
+    WorkerProcessLauncher* event_handler) {
+  EXPECT_FALSE(event_handler_);
+
+  event_handler->OnFatalError();
+
   task_runner_->PostTask(
       FROM_HERE,
       base::Bind(&WorkerProcessLauncherTest::StopWorker,
                  base::Unretained(this)));
-  return false;
+}
+
+void WorkerProcessLauncherTest::KillProcess() {
+  event_handler_ = NULL;
+
+  if (worker_process_.IsValid()) {
+    TerminateProcess(worker_process_, CONTROL_C_EXIT);
+    worker_process_.Close();
+  }
+}
+
+void WorkerProcessLauncherTest::TerminateWorker(DWORD exit_code) {
+  if (worker_process_.IsValid())
+    TerminateProcess(worker_process_, exit_code);
 }
 
 void WorkerProcessLauncherTest::ConnectClient() {
   channel_client_.reset(new IPC::ChannelProxy(
       IPC::ChannelHandle(channel_name_),
       IPC::Channel::MODE_CLIENT,
       &client_listener_,
       task_runner_));
 
   // Pretend that |kLaunchSuccessTimeoutSeconds| passed since launching
   // the worker process. This will make the backoff algorithm think that this
   // launch attempt was successful and it will not delay the next launch.
-  launcher_->ResetLaunchSuccessTimeoutForTest();
+  launcher_->RecordSuccessfulLaunchForTest();
 }
 
 void WorkerProcessLauncherTest::DisconnectClient() {
   channel_client_.reset();
 }
 
 void WorkerProcessLauncherTest::DisconnectServer() {
   channel_server_.reset();
 }
 
-bool WorkerProcessLauncherTest::SendToProcess(IPC::Message* message) {
-  if (channel_server_)
-    return channel_server_->Send(message);
+void WorkerProcessLauncherTest::SendToProcess(IPC::Message* message) {
+  if (channel_server_) {
+    channel_server_->Send(message);
+    return;
+  }
 
   delete message;
-  return false;
 }
 
 void WorkerProcessLauncherTest::SendFakeMessageToLauncher() {
   if (channel_client_)
     channel_client_->Send(new ChromotingDesktopNetworkMsg_DisconnectSession());
 }
 
 void WorkerProcessLauncherTest::CrashWorker() {
   launcher_->Crash(FROM_HERE);
 }
 
 void WorkerProcessLauncherTest::StartWorker() {
-  launcher_.reset(new WorkerProcessLauncher(
-      task_runner_, launcher_delegate_.Pass(), &server_listener_));
+  launcher_.reset(new WorkerProcessLauncher(launcher_delegate_.Pass(),
+                                            &server_listener_));
 
   launcher_->SetKillProcessTimeoutForTest(base::TimeDelta::FromMilliseconds(0));
 }
 
 void WorkerProcessLauncherTest::StopWorker() {
   launcher_.reset();
   DisconnectClient();
   channel_name_.clear();
   channel_server_.reset();
   task_runner_ = NULL;
 }
 
 void WorkerProcessLauncherTest::QuitMainMessageLoop() {
   message_loop_.PostTask(FROM_HERE, base::MessageLoop::QuitClosure());
 }
 
+void WorkerProcessLauncherTest::DoLaunchProcess() {
+  EXPECT_TRUE(event_handler_);
+  EXPECT_FALSE(worker_process_.IsValid());
+
+  WCHAR notepad[MAX_PATH + 1];
+  ASSERT_GT(ExpandEnvironmentStrings(
+      L"\045SystemRoot\045\\system32\\notepad.exe", notepad, MAX_PATH), 0u);
+
+  STARTUPINFOW startup_info = { 0 };
+  startup_info.cb = sizeof(startup_info);
+
+  base::win::ScopedProcessInformation process_information;
+  ASSERT_TRUE(CreateProcess(NULL,
+                            notepad,
+                            NULL,   // default process attibutes
+                            NULL,   // default thread attibutes
+                            FALSE,  // do not inherit handles
+                            CREATE_SUSPENDED,
+                            NULL,   // no environment
+                            NULL,   // default current directory
+                            &startup_info,
+                            process_information.Receive()));
+  worker_process_.Set(process_information.TakeProcessHandle());
+  ASSERT_TRUE(worker_process_.IsValid());
+
+  channel_name_ = IPC::Channel::GenerateUniqueRandomChannelID();
+  ScopedHandle pipe;
+  ASSERT_TRUE(CreateIpcChannel(channel_name_, kIpcSecurityDescriptor, &pipe));
+
+  // Wrap the pipe into an IPC channel.
+  channel_server_.reset(new IPC::ChannelProxy(
+      IPC::ChannelHandle(pipe),
+      IPC::Channel::MODE_SERVER,
+      this,
+      task_runner_));
+
+  ScopedHandle copy;
+  ASSERT_TRUE(DuplicateHandle(GetCurrentProcess(),
+                              worker_process_,
+                              GetCurrentProcess(),
+                              copy.Receive(),
+                              0,
+                              FALSE,
+                              DUPLICATE_SAME_ACCESS));
+
+  event_handler_->OnProcessLaunched(copy.Pass());
+}
+
 TEST_F(WorkerProcessLauncherTest, Start) {
-  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_, _))
+  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_))
       .Times(1)
       .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::LaunchProcess));
 
   EXPECT_CALL(server_listener_, OnChannelConnected(_))
       .Times(0);
   EXPECT_CALL(server_listener_, OnPermanentError())
       .Times(0);
 
   StartWorker();
   StopWorker();
   message_loop_.Run();
 }
 
 // Starts and connects to the worker process. Expect OnChannelConnected to be
 // called.
 TEST_F(WorkerProcessLauncherTest, StartAndConnect) {
-  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_, _))
+  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_))
       .Times(1)
       .WillRepeatedly(Invoke(
           this, &WorkerProcessLauncherTest::LaunchProcessAndConnect));
 
   EXPECT_CALL(server_listener_, OnChannelConnected(_))
       .Times(1)
       .WillOnce(InvokeWithoutArgs(this,
                                   &WorkerProcessLauncherTest::StopWorker));
   EXPECT_CALL(server_listener_, OnPermanentError())
       .Times(0);
 
   StartWorker();
   message_loop_.Run();
 }
 
 // Kills the worker process after the 1st connect and expects it to be
 // restarted.
 TEST_F(WorkerProcessLauncherTest, Restart) {
-  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_, _))
+  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_))
       .Times(2)
       .WillRepeatedly(Invoke(
           this, &WorkerProcessLauncherTest::LaunchProcessAndConnect));
   Expectation first_connect =
       EXPECT_CALL(server_listener_, OnChannelConnected(_))
           .Times(2)
           .WillOnce(InvokeWithoutArgs(CreateFunctor(
-              this, &WorkerProcessLauncherTest::KillProcess, CONTROL_C_EXIT)))
+              this, &WorkerProcessLauncherTest::TerminateWorker,
+              CONTROL_C_EXIT)))
           .WillOnce(InvokeWithoutArgs(this,
                                       &WorkerProcessLauncherTest::StopWorker));
 
   EXPECT_CALL(server_listener_, OnPermanentError())
       .Times(0);
 
   StartWorker();
   message_loop_.Run();
 }
 
 // Drops the IPC channel to the worker process after the 1st connect and expects
 // the worker process to be restarted.
 TEST_F(WorkerProcessLauncherTest, DropIpcChannel) {
-  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_, _))
+  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_))
       .Times(2)
       .WillRepeatedly(Invoke(
           this, &WorkerProcessLauncherTest::LaunchProcessAndConnect));
 
   Expectation first_connect =
       EXPECT_CALL(server_listener_, OnChannelConnected(_))
           .Times(2)
           .WillOnce(InvokeWithoutArgs(
               this, &WorkerProcessLauncherTest::DisconnectClient))
           .WillOnce(InvokeWithoutArgs(
               this, &WorkerProcessLauncherTest::StopWorker));
 
   EXPECT_CALL(server_listener_, OnPermanentError())
       .Times(0);
 
   StartWorker();
   message_loop_.Run();
 }
 
 // Returns a permanent error exit code and expects OnPermanentError() to be
 // invoked.
 TEST_F(WorkerProcessLauncherTest, PermanentError) {
-  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_, _))
+  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_))
       .Times(1)
       .WillRepeatedly(Invoke(
           this, &WorkerProcessLauncherTest::LaunchProcessAndConnect));
 
   EXPECT_CALL(server_listener_, OnChannelConnected(_))
       .Times(1)
       .WillOnce(InvokeWithoutArgs(CreateFunctor(
-          this, &WorkerProcessLauncherTest::KillProcess, CONTROL_C_EXIT)));
+          this, &WorkerProcessLauncherTest::TerminateWorker,
+          kMinPermanentErrorExitCode)));
   EXPECT_CALL(server_listener_, OnPermanentError())
       .Times(1)
       .WillOnce(InvokeWithoutArgs(this,
                                   &WorkerProcessLauncherTest::StopWorker));
 
-  permanent_error_ = true;
-  StartWorker();
-  message_loop_.Run();
-}
-
-// Returns invalid client PID and expects the connection to be rejected.
-TEST_F(WorkerProcessLauncherTest, InvalidClientPid) {
-  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_, _))
-      .Times(2)
-      .WillOnce(Invoke(
-          this, &WorkerProcessLauncherTest::LaunchProcessAndConnect))
-      .WillOnce(Invoke(
-          this, &WorkerProcessLauncherTest::FailLaunchAndStopWorker));
-
-  EXPECT_CALL(server_listener_, OnChannelConnected(_))
-      .Times(0);
-  EXPECT_CALL(server_listener_, OnPermanentError())
-      .Times(0);
-
-  client_pid_ = GetCurrentProcessId() + 4;
   StartWorker();
   message_loop_.Run();
 }
 
 // Requests the worker to crash and expects it to honor the request.
 TEST_F(WorkerProcessLauncherTest, Crash) {
-  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_, _))
+  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_))
       .Times(2)
       .WillRepeatedly(Invoke(
           this, &WorkerProcessLauncherTest::LaunchProcessAndConnect));
 
   EXPECT_CALL(server_listener_, OnChannelConnected(_))
       .Times(2)
       .WillOnce(InvokeWithoutArgs(this,
                                   &WorkerProcessLauncherTest::CrashWorker))
       .WillOnce(InvokeWithoutArgs(this,
                                   &WorkerProcessLauncherTest::StopWorker));
 
   EXPECT_CALL(client_listener_, OnCrash(_, _, _))
       .Times(1)
       .WillOnce(InvokeWithoutArgs(CreateFunctor(
-          this, &WorkerProcessLauncherTest::KillProcess,
+          this, &WorkerProcessLauncherTest::TerminateWorker,
           EXCEPTION_BREAKPOINT)));
 
   StartWorker();
   message_loop_.Run();
 }
 
 // Requests the worker to crash and terminates the worker even if it does not
 // comply.
 TEST_F(WorkerProcessLauncherTest, CrashAnyway) {
-  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_, _))
+  EXPECT_CALL(*launcher_delegate_, LaunchProcess(_))
       .Times(2)
       .WillRepeatedly(Invoke(
           this, &WorkerProcessLauncherTest::LaunchProcessAndConnect));
 
   EXPECT_CALL(server_listener_, OnChannelConnected(_))
       .Times(2)
       .WillOnce(InvokeWithoutArgs(this,
                                   &WorkerProcessLauncherTest::CrashWorker))
       .WillOnce(InvokeWithoutArgs(this,
                                   &WorkerProcessLauncherTest::StopWorker));
 
   // Ignore the crash request and try send another message to the launcher.
   EXPECT_CALL(client_listener_, OnCrash(_, _, _))
       .Times(1)
       .WillOnce(InvokeWithoutArgs(
           this, &WorkerProcessLauncherTest::SendFakeMessageToLauncher));
 
   StartWorker();
   message_loop_.Run();
 }
 
 }  // namespace remoting