1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "testing/gtest/include/gtest/gtest.h" 6 #include "sandbox/win/src/sandbox.h" 7 #include "sandbox/win/src/sandbox_factory.h" 8 #include "sandbox/win/src/target_services.h" 9 #include "sandbox/win/tests/common/controller.h" 10 11 namespace sandbox { 12 13 // Tests that the IPC is working by issuing a special IPC that is not exposed 14 // in the public API. 15 SBOX_TESTS_COMMAND int IPC_Ping(int argc, wchar_t **argv) { 16 if (argc != 1) 17 return SBOX_TEST_FAILED; 18 19 TargetServices* ts = SandboxFactory::GetTargetServices(); 20 if (NULL == ts) 21 return SBOX_TEST_FAILED; 22 23 // Downcast because we have internal knowledge of the object returned. 24 TargetServicesBase* ts_base = reinterpret_cast<TargetServicesBase*>(ts); 25 26 int version = 0; 27 if (L'1' == argv[0][0]) 28 version = 1; 29 else 30 version = 2; 31 32 if (!ts_base->TestIPCPing(version)) 33 return SBOX_TEST_FAILED; 34 35 ::Sleep(1); 36 if (!ts_base->TestIPCPing(version)) 37 return SBOX_TEST_FAILED; 38 39 return SBOX_TEST_SUCCEEDED; 40 } 41 42 // The IPC ping test should work before and after the token drop. 43 TEST(IPCTest, IPCPingTestSimple) { 44 TestRunner runner; 45 runner.SetTimeout(2000); 46 runner.SetTestState(EVERY_STATE); 47 EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"IPC_Ping 1")); 48 } 49 50 TEST(IPCTest, IPCPingTestWithOutput) { 51 TestRunner runner; 52 runner.SetTimeout(2000); 53 runner.SetTestState(EVERY_STATE); 54 EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"IPC_Ping 2")); 55 EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"IPC_Ping 2")); 56 } 57 58 } // namespace sandbox 59
