Index: sandbox/win/tests/integration_tests/integration_tests_test.cc |
diff --git a/sandbox/win/tests/integration_tests/integration_tests_test.cc b/sandbox/win/tests/integration_tests/integration_tests_test.cc |
deleted file mode 100644 |
index 4acadb90a2881f79353d72aaa2c1d56176f1bff7..0000000000000000000000000000000000000000 |
--- a/sandbox/win/tests/integration_tests/integration_tests_test.cc |
+++ /dev/null |
@@ -1,308 +0,0 @@ |
-// Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-// Some tests for the framework itself. |
- |
-#include <stddef.h> |
- |
-#include "sandbox/win/src/sandbox.h" |
-#include "sandbox/win/src/sandbox_factory.h" |
-#include "sandbox/win/src/target_services.h" |
-#include "sandbox/win/tests/common/controller.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace sandbox { |
- |
-// Returns the current process state. |
-SBOX_TESTS_COMMAND int IntegrationTestsTest_state(int argc, wchar_t **argv) { |
- if (!SandboxFactory::GetTargetServices()->GetState()->InitCalled()) |
- return BEFORE_INIT; |
- |
- if (!SandboxFactory::GetTargetServices()->GetState()->RevertedToSelf()) |
- return BEFORE_REVERT; |
- |
- return AFTER_REVERT; |
-} |
- |
-// Returns the current process state, keeping track of it. |
-SBOX_TESTS_COMMAND int IntegrationTestsTest_state2(int argc, wchar_t **argv) { |
- static SboxTestsState state = MIN_STATE; |
- if (!SandboxFactory::GetTargetServices()->GetState()->InitCalled()) { |
- if (MIN_STATE == state) |
- state = BEFORE_INIT; |
- return state; |
- } |
- |
- if (!SandboxFactory::GetTargetServices()->GetState()->RevertedToSelf()) { |
- if (BEFORE_INIT == state) |
- state = BEFORE_REVERT; |
- return state; |
- } |
- |
- if (BEFORE_REVERT == state) |
- state = AFTER_REVERT; |
- return state; |
-} |
- |
-// Blocks the process for argv[0] milliseconds simulating stuck child. |
-SBOX_TESTS_COMMAND int IntegrationTestsTest_stuck(int argc, wchar_t **argv) { |
- int timeout = 500; |
- if (argc > 0) { |
- timeout = _wtoi(argv[0]); |
- } |
- |
- ::Sleep(timeout); |
- return 1; |
-} |
- |
-// Returns the number of arguments |
-SBOX_TESTS_COMMAND int IntegrationTestsTest_args(int argc, wchar_t **argv) { |
- for (int i = 0; i < argc; i++) { |
- wchar_t argument[20]; |
- size_t argument_bytes = wcslen(argv[i]) * sizeof(wchar_t); |
- memcpy(argument, argv[i], __min(sizeof(argument), argument_bytes)); |
- } |
- |
- return argc; |
-} |
- |
-// Creates a job and tries to run a process inside it. The function can be |
-// called with up to two parameters. The first one if set to "none" means that |
-// the child process should be run with the JOB_NONE JobLevel else it is run |
-// with JOB_LOCKDOWN level. The second if present specifies that the |
-// JOB_OBJECT_LIMIT_BREAKAWAY_OK flag should be set on the job object created |
-// in this function. The return value is either SBOX_TEST_SUCCEEDED if the test |
-// has passed or a value between 0 and 4 indicating which part of the test has |
-// failed. |
-SBOX_TESTS_COMMAND int IntegrationTestsTest_job(int argc, wchar_t **argv) { |
- HANDLE job = ::CreateJobObject(NULL, NULL); |
- if (!job) |
- return 0; |
- |
- JOBOBJECT_EXTENDED_LIMIT_INFORMATION job_limits; |
- if (!::QueryInformationJobObject(job, JobObjectExtendedLimitInformation, |
- &job_limits, sizeof(job_limits), NULL)) { |
- return 1; |
- } |
- // We cheat here and assume no 2-nd parameter means no breakaway flag and any |
- // value for the second param means with breakaway flag. |
- if (argc > 1) { |
- job_limits.BasicLimitInformation.LimitFlags |= |
- JOB_OBJECT_LIMIT_BREAKAWAY_OK; |
- } else { |
- job_limits.BasicLimitInformation.LimitFlags &= |
- ~JOB_OBJECT_LIMIT_BREAKAWAY_OK; |
- } |
- if (!::SetInformationJobObject(job, JobObjectExtendedLimitInformation, |
- &job_limits, sizeof(job_limits))) { |
- return 2; |
- } |
- if (!::AssignProcessToJobObject(job, ::GetCurrentProcess())) |
- return 3; |
- |
- JobLevel job_level = JOB_LOCKDOWN; |
- if (argc > 0 && wcscmp(argv[0], L"none") == 0) |
- job_level = JOB_NONE; |
- |
- TestRunner runner(job_level, USER_RESTRICTED_SAME_ACCESS, USER_LOCKDOWN); |
- runner.SetTimeout(2000); |
- |
- if (1 != runner.RunTest(L"IntegrationTestsTest_args 1")) |
- return 4; |
- |
- // Terminate the job now. |
- ::TerminateJobObject(job, SBOX_TEST_SUCCEEDED); |
- // We should not make it to here but it doesn't mean our test failed. |
- return SBOX_TEST_SUCCEEDED; |
-} |
- |
-TEST(IntegrationTestsTest, CallsBeforeInit) { |
- TestRunner runner; |
- runner.SetTimeout(2000); |
- runner.SetTestState(BEFORE_INIT); |
- ASSERT_EQ(BEFORE_INIT, runner.RunTest(L"IntegrationTestsTest_state")); |
-} |
- |
-TEST(IntegrationTestsTest, CallsBeforeRevert) { |
- TestRunner runner; |
- runner.SetTimeout(2000); |
- runner.SetTestState(BEFORE_REVERT); |
- ASSERT_EQ(BEFORE_REVERT, runner.RunTest(L"IntegrationTestsTest_state")); |
-} |
- |
-TEST(IntegrationTestsTest, CallsAfterRevert) { |
- TestRunner runner; |
- runner.SetTimeout(2000); |
- runner.SetTestState(AFTER_REVERT); |
- ASSERT_EQ(AFTER_REVERT, runner.RunTest(L"IntegrationTestsTest_state")); |
-} |
- |
-TEST(IntegrationTestsTest, CallsEveryState) { |
- TestRunner runner; |
- runner.SetTimeout(2000); |
- runner.SetTestState(EVERY_STATE); |
- ASSERT_EQ(AFTER_REVERT, runner.RunTest(L"IntegrationTestsTest_state2")); |
-} |
- |
-TEST(IntegrationTestsTest, ForwardsArguments) { |
- TestRunner runner; |
- runner.SetTimeout(2000); |
- runner.SetTestState(BEFORE_INIT); |
- ASSERT_EQ(1, runner.RunTest(L"IntegrationTestsTest_args first")); |
- ASSERT_EQ(4, runner.RunTest(L"IntegrationTestsTest_args first second third " |
- L"fourth")); |
-} |
- |
-TEST(IntegrationTestsTest, WaitForStuckChild) { |
- TestRunner runner; |
- runner.SetTimeout(2000); |
- runner.SetAsynchronous(true); |
- runner.SetKillOnDestruction(false); |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner.RunTest(L"IntegrationTestsTest_stuck 100")); |
- ASSERT_EQ(SBOX_ALL_OK, runner.broker()->WaitForAllTargets()); |
-} |
- |
-TEST(IntegrationTestsTest, NoWaitForStuckChildNoJob) { |
- TestRunner runner(JOB_NONE, USER_RESTRICTED_SAME_ACCESS, USER_LOCKDOWN); |
- runner.SetTimeout(2000); |
- runner.SetAsynchronous(true); |
- runner.SetKillOnDestruction(false); |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner.RunTest(L"IntegrationTestsTest_stuck 2000")); |
- ASSERT_EQ(SBOX_ALL_OK, runner.broker()->WaitForAllTargets()); |
- // In this case the processes are not tracked by the broker and should be |
- // still active. |
- DWORD exit_code; |
- ASSERT_TRUE(::GetExitCodeProcess(runner.process(), &exit_code)); |
- ASSERT_EQ(STILL_ACTIVE, exit_code); |
- // Terminate the test process now. |
- ::TerminateProcess(runner.process(), 0); |
-} |
- |
-TEST(IntegrationTestsTest, TwoStuckChildrenSecondOneHasNoJob) { |
- TestRunner runner; |
- runner.SetTimeout(2000); |
- runner.SetAsynchronous(true); |
- runner.SetKillOnDestruction(false); |
- TestRunner runner2(JOB_NONE, USER_RESTRICTED_SAME_ACCESS, USER_LOCKDOWN); |
- runner2.SetTimeout(2000); |
- runner2.SetAsynchronous(true); |
- runner2.SetKillOnDestruction(false); |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner.RunTest(L"IntegrationTestsTest_stuck 100")); |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner2.RunTest(L"IntegrationTestsTest_stuck 2000")); |
- // Actually both runners share the same singleton broker. |
- ASSERT_EQ(SBOX_ALL_OK, runner.broker()->WaitForAllTargets()); |
- // In this case the processes are not tracked by the broker and should be |
- // still active. |
- DWORD exit_code; |
- // Checking the exit code for |runner| is flaky on the slow bots but at |
- // least we know that the wait above has succeeded if we are here. |
- ASSERT_TRUE(::GetExitCodeProcess(runner2.process(), &exit_code)); |
- ASSERT_EQ(STILL_ACTIVE, exit_code); |
- // Terminate the test process now. |
- ::TerminateProcess(runner2.process(), 0); |
-} |
- |
-TEST(IntegrationTestsTest, TwoStuckChildrenFirstOneHasNoJob) { |
- TestRunner runner; |
- runner.SetTimeout(2000); |
- runner.SetAsynchronous(true); |
- runner.SetKillOnDestruction(false); |
- TestRunner runner2(JOB_NONE, USER_RESTRICTED_SAME_ACCESS, USER_LOCKDOWN); |
- runner2.SetTimeout(2000); |
- runner2.SetAsynchronous(true); |
- runner2.SetKillOnDestruction(false); |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner2.RunTest(L"IntegrationTestsTest_stuck 2000")); |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner.RunTest(L"IntegrationTestsTest_stuck 100")); |
- // Actually both runners share the same singleton broker. |
- ASSERT_EQ(SBOX_ALL_OK, runner.broker()->WaitForAllTargets()); |
- // In this case the processes are not tracked by the broker and should be |
- // still active. |
- DWORD exit_code; |
- // Checking the exit code for |runner| is flaky on the slow bots but at |
- // least we know that the wait above has succeeded if we are here. |
- ASSERT_TRUE(::GetExitCodeProcess(runner2.process(), &exit_code)); |
- ASSERT_EQ(STILL_ACTIVE, exit_code); |
- // Terminate the test process now. |
- ::TerminateProcess(runner2.process(), 0); |
-} |
- |
-TEST(IntegrationTestsTest, MultipleStuckChildrenSequential) { |
- TestRunner runner; |
- runner.SetTimeout(2000); |
- runner.SetAsynchronous(true); |
- runner.SetKillOnDestruction(false); |
- TestRunner runner2(JOB_NONE, USER_RESTRICTED_SAME_ACCESS, USER_LOCKDOWN); |
- runner2.SetTimeout(2000); |
- runner2.SetAsynchronous(true); |
- runner2.SetKillOnDestruction(false); |
- |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner.RunTest(L"IntegrationTestsTest_stuck 100")); |
- // Actually both runners share the same singleton broker. |
- ASSERT_EQ(SBOX_ALL_OK, runner.broker()->WaitForAllTargets()); |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner2.RunTest(L"IntegrationTestsTest_stuck 2000")); |
- // Actually both runners share the same singleton broker. |
- ASSERT_EQ(SBOX_ALL_OK, runner.broker()->WaitForAllTargets()); |
- |
- DWORD exit_code; |
- // Checking the exit code for |runner| is flaky on the slow bots but at |
- // least we know that the wait above has succeeded if we are here. |
- ASSERT_TRUE(::GetExitCodeProcess(runner2.process(), &exit_code)); |
- ASSERT_EQ(STILL_ACTIVE, exit_code); |
- // Terminate the test process now. |
- ::TerminateProcess(runner2.process(), 0); |
- |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner.RunTest(L"IntegrationTestsTest_stuck 100")); |
- // Actually both runners share the same singleton broker. |
- ASSERT_EQ(SBOX_ALL_OK, runner.broker()->WaitForAllTargets()); |
-} |
- |
-// Running from inside job that allows us to escape from it should be ok. |
-TEST(IntegrationTestsTest, RunChildFromInsideJob) { |
- TestRunner runner; |
- runner.SetUnsandboxed(true); |
- runner.SetTimeout(2000); |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner.RunTest(L"IntegrationTestsTest_job with_job escape_flag")); |
-} |
- |
-// Running from inside job that doesn't allow us to escape from it should fail |
-// on any windows prior to 8. |
-TEST(IntegrationTestsTest, RunChildFromInsideJobNoEscape) { |
- int expect_result = 4; // Means the runner has failed to execute the child. |
- // Check if we are on Win8 or newer and expect a success as newer windows |
- // versions support nested jobs. |
- OSVERSIONINFOEX version_info = { sizeof version_info }; |
- ::GetVersionEx(reinterpret_cast<OSVERSIONINFO*>(&version_info)); |
- if (version_info.dwMajorVersion > 6 || |
- (version_info.dwMajorVersion == 6 && version_info.dwMinorVersion >= 2)) { |
- expect_result = SBOX_TEST_SUCCEEDED; |
- } |
- |
- TestRunner runner; |
- runner.SetUnsandboxed(true); |
- runner.SetTimeout(2000); |
- ASSERT_EQ(expect_result, |
- runner.RunTest(L"IntegrationTestsTest_job with_job")); |
-} |
- |
-// Running without a job object should be ok regardless of the fact that we are |
-// running inside an outter job. |
-TEST(IntegrationTestsTest, RunJoblessChildFromInsideJob) { |
- TestRunner runner; |
- runner.SetUnsandboxed(true); |
- runner.SetTimeout(2000); |
- ASSERT_EQ(SBOX_TEST_SUCCEEDED, |
- runner.RunTest(L"IntegrationTestsTest_job none")); |
-} |
- |
-} // namespace sandbox |