worker_process_launcher_unittest.cc revision 9ab5563a3196760eb381d102cbb2bc0f7abc6a50
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/ref_counted.h" 89ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_handle.h" 1090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "base/win/scoped_process_information.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ipc/ipc_channel.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ipc/ipc_channel_proxy.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ipc/ipc_listener.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ipc/ipc_message.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/base/auto_thread_task_runner.h" 162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "remoting/host/chromoting_messages.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/host_exit_codes.h" 1890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "remoting/host/ipc_util.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/win/launch_process_with_token.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/win/worker_process_launcher.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "remoting/host/worker_process_ipc_delegate.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gmock/include/gmock/gmock.h" 232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "testing/gmock_mutant.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using base::win::ScopedHandle; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::_; 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::AnyNumber; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::CreateFunctor; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::DoAll; 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::Expectation; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::Invoke; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::InvokeWithoutArgs; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::Return; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace remoting { 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kIpcSecurityDescriptor[] = "D:(A;;GA;;;AU)"; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class MockProcessLauncherDelegate : public WorkerProcessLauncher::Delegate { 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockProcessLauncherDelegate() {} 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockProcessLauncherDelegate() {} 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // WorkerProcessLauncher::Delegate interface. 4890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) MOCK_METHOD1(LaunchProcess, void(WorkerProcessLauncher*)); 4990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) MOCK_METHOD1(Send, void(IPC::Message*)); 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) MOCK_METHOD0(CloseChannel, void()); 5190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) MOCK_METHOD0(KillProcess, void()); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockProcessLauncherDelegate); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class MockIpcDelegate : public WorkerProcessIpcDelegate { 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockIpcDelegate() {} 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~MockIpcDelegate() {} 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // WorkerProcessIpcDelegate interface. 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MOCK_METHOD1(OnChannelConnected, void(int32)); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MOCK_METHOD1(OnMessageReceived, bool(const IPC::Message&)); 65eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch MOCK_METHOD1(OnPermanentError, void(int)); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockIpcDelegate); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class MockWorkerListener : public IPC::Listener { 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) public: 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) MockWorkerListener() {} 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual ~MockWorkerListener() {} 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) MOCK_METHOD3(OnCrash, void(const std::string&, const std::string&, int)); 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // IPC::Listener implementation 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) private: 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(MockWorkerListener); 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool MockWorkerListener::OnMessageReceived(const IPC::Message& message) { 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool handled = true; 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IPC_BEGIN_MESSAGE_MAP(MockWorkerListener, message) 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IPC_MESSAGE_HANDLER(ChromotingDaemonMsg_Crash, OnCrash) 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IPC_MESSAGE_UNHANDLED(handled = false) 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IPC_END_MESSAGE_MAP() 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(handled); 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return handled; 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)class WorkerProcessLauncherTest 10090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) : public testing::Test, 10190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) public IPC::Listener { 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WorkerProcessLauncherTest(); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~WorkerProcessLauncherTest(); 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUp() OVERRIDE; 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void TearDown() OVERRIDE; 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 10990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // IPC::Listener implementation. 11090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE; 11190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual void OnChannelConnected(int32 peer_pid) OVERRIDE; 11290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual void OnChannelError() OVERRIDE; 11390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // WorkerProcessLauncher::Delegate mocks 11590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) void LaunchProcess( 11690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) WorkerProcessLauncher* event_handler); 11790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) void LaunchProcessAndConnect( 11890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) WorkerProcessLauncher* event_handler); 11990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) void FailLaunchAndStopWorker( 12090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) WorkerProcessLauncher* event_handler); 12190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) void KillProcess(); 12290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 12390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) void TerminateWorker(DWORD exit_code); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Connects the client end of the channel (the worker process's end). 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void ConnectClient(); 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Disconnects the client end of the channel. 1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void DisconnectClient(); 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Disconnects the server end of the channel (the launcher's end). 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void DisconnectServer(); 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Sends a message to the worker process. 13590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) void SendToProcess(IPC::Message* message); 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Sends a fake message to the launcher. 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void SendFakeMessageToLauncher(); 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Requests the worker to crash. 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void CrashWorker(); 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Starts the worker. 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void StartWorker(); 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stops the worker. 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void StopWorker(); 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Quits |message_loop_|. 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void QuitMainMessageLoop(); 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 15390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) void DoLaunchProcess(); 15490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 155c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop message_loop_; 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<AutoThreadTaskRunner> task_runner_; 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Receives messages sent to the worker process. 1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) MockWorkerListener client_listener_; 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Receives messages sent from the worker process. 1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) MockIpcDelegate server_listener_; 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Implements WorkerProcessLauncher::Delegate. 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<MockProcessLauncherDelegate> launcher_delegate_; 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The name of the IPC channel. 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string channel_name_; 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Client and server ends of the IPC channel. 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<IPC::ChannelProxy> channel_client_; 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<IPC::ChannelProxy> channel_server_; 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 17490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) WorkerProcessLauncher* event_handler_; 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The worker process launcher. 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<WorkerProcessLauncher> launcher_; 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 17990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // An event that is used to emulate the worker process's handle. 18090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ScopedHandle worker_process_; 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)WorkerProcessLauncherTest::WorkerProcessLauncherTest() 184c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) : message_loop_(base::MessageLoop::TYPE_IO), 18590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) event_handler_(NULL) { 18690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)WorkerProcessLauncherTest::~WorkerProcessLauncherTest() { 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void WorkerProcessLauncherTest::SetUp() { 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) task_runner_ = new AutoThreadTaskRunner( 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_.message_loop_proxy(), 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&WorkerProcessLauncherTest::QuitMainMessageLoop, 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set up process launcher delegate 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) launcher_delegate_.reset(new MockProcessLauncherDelegate()); 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*launcher_delegate_, Send(_)) 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(AnyNumber()) 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::SendToProcess)); 2022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(*launcher_delegate_, CloseChannel()) 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .Times(AnyNumber()) 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillRepeatedly(Invoke(this, 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &WorkerProcessLauncherTest::DisconnectServer)); 20690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*launcher_delegate_, KillProcess()) 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(AnyNumber()) 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::KillProcess)); 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set up IPC delegate. 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(server_listener_, OnMessageReceived(_)) 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .Times(0); 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void WorkerProcessLauncherTest::TearDown() { 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 21890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)bool WorkerProcessLauncherTest::OnMessageReceived(const IPC::Message& message) { 21990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return event_handler_->OnMessageReceived(message); 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 22290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void WorkerProcessLauncherTest::OnChannelConnected(int32 peer_pid) { 22390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) event_handler_->OnChannelConnected(peer_pid); 22490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 22690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void WorkerProcessLauncherTest::OnChannelError() { 22790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) event_handler_->OnChannelError(); 22890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 23090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void WorkerProcessLauncherTest::LaunchProcess( 23190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) WorkerProcessLauncher* event_handler) { 23290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_FALSE(event_handler_); 23390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) event_handler_ = event_handler; 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 23590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) DoLaunchProcess(); 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 23890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void WorkerProcessLauncherTest::LaunchProcessAndConnect( 23990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) WorkerProcessLauncher* event_handler) { 24090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_FALSE(event_handler_); 24190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) event_handler_ = event_handler; 24290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 24390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) DoLaunchProcess(); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) task_runner_->PostTask( 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&WorkerProcessLauncherTest::ConnectClient, 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 25190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void WorkerProcessLauncherTest::FailLaunchAndStopWorker( 25290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) WorkerProcessLauncher* event_handler) { 25390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_FALSE(event_handler_); 25490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 25590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) event_handler->OnFatalError(); 25690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) task_runner_->PostTask( 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&WorkerProcessLauncherTest::StopWorker, 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 26190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 26290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 26390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void WorkerProcessLauncherTest::KillProcess() { 26490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) event_handler_ = NULL; 26590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 26690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) if (worker_process_.IsValid()) { 26790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) TerminateProcess(worker_process_, CONTROL_C_EXIT); 26890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) worker_process_.Close(); 26990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 27090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 27190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 27290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void WorkerProcessLauncherTest::TerminateWorker(DWORD exit_code) { 27390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) if (worker_process_.IsValid()) 27490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) TerminateProcess(worker_process_, exit_code); 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void WorkerProcessLauncherTest::ConnectClient() { 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) channel_client_.reset(new IPC::ChannelProxy( 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IPC::ChannelHandle(channel_name_), 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IPC::Channel::MODE_CLIENT, 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &client_listener_, 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) task_runner_)); 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Pretend that |kLaunchSuccessTimeoutSeconds| passed since launching 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // the worker process. This will make the backoff algorithm think that this 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // launch attempt was successful and it will not delay the next launch. 28790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) launcher_->RecordSuccessfulLaunchForTest(); 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void WorkerProcessLauncherTest::DisconnectClient() { 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) channel_client_.reset(); 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void WorkerProcessLauncherTest::DisconnectServer() { 2952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) channel_server_.reset(); 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 29890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void WorkerProcessLauncherTest::SendToProcess(IPC::Message* message) { 29990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) if (channel_server_) { 30090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) channel_server_->Send(message); 30190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) return; 30290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) delete message; 3052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void WorkerProcessLauncherTest::SendFakeMessageToLauncher() { 3082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (channel_client_) 3092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) channel_client_->Send(new ChromotingDesktopNetworkMsg_DisconnectSession()); 3102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void WorkerProcessLauncherTest::CrashWorker() { 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) launcher_->Crash(FROM_HERE); 3142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void WorkerProcessLauncherTest::StartWorker() { 31790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) launcher_.reset(new WorkerProcessLauncher(launcher_delegate_.Pass(), 31890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) &server_listener_)); 3192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) launcher_->SetKillProcessTimeoutForTest(base::TimeDelta::FromMilliseconds(0)); 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void WorkerProcessLauncherTest::StopWorker() { 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) launcher_.reset(); 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DisconnectClient(); 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) channel_name_.clear(); 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) channel_server_.reset(); 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) task_runner_ = NULL; 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void WorkerProcessLauncherTest::QuitMainMessageLoop() { 332c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) message_loop_.PostTask(FROM_HERE, base::MessageLoop::QuitClosure()); 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 33590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)void WorkerProcessLauncherTest::DoLaunchProcess() { 33690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_TRUE(event_handler_); 33790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_FALSE(worker_process_.IsValid()); 33890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 33990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) WCHAR notepad[MAX_PATH + 1]; 34090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ASSERT_GT(ExpandEnvironmentStrings( 34190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) L"\045SystemRoot\045\\system32\\notepad.exe", notepad, MAX_PATH), 0u); 34290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 34390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) STARTUPINFOW startup_info = { 0 }; 34490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) startup_info.cb = sizeof(startup_info); 34590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 34690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::win::ScopedProcessInformation process_information; 34790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ASSERT_TRUE(CreateProcess(NULL, 34890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) notepad, 34990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) NULL, // default process attibutes 35090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) NULL, // default thread attibutes 35190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) FALSE, // do not inherit handles 35290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) CREATE_SUSPENDED, 35390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) NULL, // no environment 35490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) NULL, // default current directory 35590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) &startup_info, 35690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) process_information.Receive())); 35790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) worker_process_.Set(process_information.TakeProcessHandle()); 35890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ASSERT_TRUE(worker_process_.IsValid()); 35990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 36090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) channel_name_ = IPC::Channel::GenerateUniqueRandomChannelID(); 36190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ScopedHandle pipe; 36290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ASSERT_TRUE(CreateIpcChannel(channel_name_, kIpcSecurityDescriptor, &pipe)); 36390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 36490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Wrap the pipe into an IPC channel. 36590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) channel_server_.reset(new IPC::ChannelProxy( 36690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) IPC::ChannelHandle(pipe), 36790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) IPC::Channel::MODE_SERVER, 36890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) this, 36990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) task_runner_)); 37090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 37190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ScopedHandle copy; 37290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ASSERT_TRUE(DuplicateHandle(GetCurrentProcess(), 37390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) worker_process_, 37490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) GetCurrentProcess(), 37590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) copy.Receive(), 37690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 0, 37790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) FALSE, 37890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) DUPLICATE_SAME_ACCESS)); 37990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 38090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) event_handler_->OnProcessLaunched(copy.Pass()); 38190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 38290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(WorkerProcessLauncherTest, Start) { 38490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1) 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(Invoke(this, &WorkerProcessLauncherTest::LaunchProcess)); 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(server_listener_, OnChannelConnected(_)) 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 390eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch EXPECT_CALL(server_listener_, OnPermanentError(_)) 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StartWorker(); 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StopWorker(); 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_.Run(); 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Starts and connects to the worker process. Expect OnChannelConnected to be 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// called. 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(WorkerProcessLauncherTest, StartAndConnect) { 40190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1) 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(Invoke( 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(server_listener_, OnChannelConnected(_)) 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1) 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(InvokeWithoutArgs(this, 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &WorkerProcessLauncherTest::StopWorker)); 410eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch EXPECT_CALL(server_listener_, OnPermanentError(_)) 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StartWorker(); 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_.Run(); 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Kills the worker process after the 1st connect and expects it to be 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// restarted. 4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(WorkerProcessLauncherTest, Restart) { 42090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(2) 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(Invoke( 4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Expectation first_connect = 4252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(server_listener_, OnChannelConnected(_)) 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(2) 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(InvokeWithoutArgs(CreateFunctor( 42890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) this, &WorkerProcessLauncherTest::TerminateWorker, 42990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) CONTROL_C_EXIT))) 4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(InvokeWithoutArgs(this, 4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &WorkerProcessLauncherTest::StopWorker)); 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 433eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch EXPECT_CALL(server_listener_, OnPermanentError(_)) 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StartWorker(); 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_.Run(); 4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Drops the IPC channel to the worker process after the 1st connect and expects 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the worker process to be restarted. 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(WorkerProcessLauncherTest, DropIpcChannel) { 44390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(2) 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(Invoke( 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Expectation first_connect = 4492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(server_listener_, OnChannelConnected(_)) 4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(2) 4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(InvokeWithoutArgs( 4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) this, &WorkerProcessLauncherTest::DisconnectClient)) 4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(InvokeWithoutArgs( 4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &WorkerProcessLauncherTest::StopWorker)); 4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 456eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch EXPECT_CALL(server_listener_, OnPermanentError(_)) 4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(0); 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StartWorker(); 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_.Run(); 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns a permanent error exit code and expects OnPermanentError() to be 4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// invoked. 4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(WorkerProcessLauncherTest, PermanentError) { 46690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) 4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1) 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillRepeatedly(Invoke( 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(server_listener_, OnChannelConnected(_)) 4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1) 4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(InvokeWithoutArgs(CreateFunctor( 47490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) this, &WorkerProcessLauncherTest::TerminateWorker, 47590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) kMinPermanentErrorExitCode))); 476eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch EXPECT_CALL(server_listener_, OnPermanentError(_)) 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(1) 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .WillOnce(InvokeWithoutArgs(this, 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &WorkerProcessLauncherTest::StopWorker)); 4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StartWorker(); 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_.Run(); 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Requests the worker to crash and expects it to honor the request. 4862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)TEST_F(WorkerProcessLauncherTest, Crash) { 48790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) 4882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .Times(2) 4892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillRepeatedly(Invoke( 4902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); 4912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(server_listener_, OnChannelConnected(_)) 4932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .Times(2) 4942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillOnce(InvokeWithoutArgs(this, 4952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &WorkerProcessLauncherTest::CrashWorker)) 4962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillOnce(InvokeWithoutArgs(this, 4972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &WorkerProcessLauncherTest::StopWorker)); 4982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(client_listener_, OnCrash(_, _, _)) 5002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .Times(1) 5012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillOnce(InvokeWithoutArgs(CreateFunctor( 50290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) this, &WorkerProcessLauncherTest::TerminateWorker, 5032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXCEPTION_BREAKPOINT))); 5042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) StartWorker(); 5062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) message_loop_.Run(); 5072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Requests the worker to crash and terminates the worker even if it does not 5102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// comply. 5112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)TEST_F(WorkerProcessLauncherTest, CrashAnyway) { 51290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_CALL(*launcher_delegate_, LaunchProcess(_)) 5132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .Times(2) 5142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillRepeatedly(Invoke( 5152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) this, &WorkerProcessLauncherTest::LaunchProcessAndConnect)); 5162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(server_listener_, OnChannelConnected(_)) 5182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .Times(2) 5192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillOnce(InvokeWithoutArgs(this, 5202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &WorkerProcessLauncherTest::CrashWorker)) 5212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillOnce(InvokeWithoutArgs(this, 5222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &WorkerProcessLauncherTest::StopWorker)); 5232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Ignore the crash request and try send another message to the launcher. 5252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_CALL(client_listener_, OnCrash(_, _, _)) 5262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .Times(1) 5272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .WillOnce(InvokeWithoutArgs( 5282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) this, &WorkerProcessLauncherTest::SendFakeMessageToLauncher)); 5292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) StartWorker(); 5312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) message_loop_.Run(); 5322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace remoting 535