| OLD | NEW |
| (Empty) |
| 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 | |
| OLD | NEW |