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 11namespace sandbox { 12 13// Tests that the IPC is working by issuing a special IPC that is not exposed 14// in the public API. 15SBOX_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. 43TEST(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 50TEST(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