|
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 |