| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 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 "base/win/scoped_handle.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 #include "testing/gtest/include/gtest/gtest.h" | |
| 11 | |
| 12 namespace sandbox { | |
| 13 | |
| 14 // Loads and or unloads a DLL passed in the second parameter of argv. | |
| 15 // The first parameter of argv is 'L' = load, 'U' = unload or 'B' for both. | |
| 16 SBOX_TESTS_COMMAND int UseOneDLL(int argc, wchar_t **argv) { | |
| 17 if (argc != 2) | |
| 18 return SBOX_TEST_FAILED_TO_RUN_TEST; | |
| 19 int rv = SBOX_TEST_FAILED_TO_RUN_TEST; | |
| 20 | |
| 21 wchar_t option = (argv[0])[0]; | |
| 22 if ((option == L'L') || (option == L'B')) { | |
| 23 HMODULE module1 = ::LoadLibraryW(argv[1]); | |
| 24 rv = (module1 == NULL) ? SBOX_TEST_FAILED : SBOX_TEST_SUCCEEDED; | |
| 25 } | |
| 26 | |
| 27 if ((option == L'U') || (option == L'B')) { | |
| 28 HMODULE module2 = ::GetModuleHandleW(argv[1]); | |
| 29 rv = ::FreeLibrary(module2) ? SBOX_TEST_SUCCEEDED : SBOX_TEST_FAILED; | |
| 30 } | |
| 31 return rv; | |
| 32 } | |
| 33 | |
| 34 // Opens an event passed as the first parameter of argv. | |
| 35 SBOX_TESTS_COMMAND int SimpleOpenEvent(int argc, wchar_t **argv) { | |
| 36 if (argc != 1) | |
| 37 return SBOX_TEST_FAILED_TO_EXECUTE_COMMAND; | |
| 38 | |
| 39 base::win::ScopedHandle event_open(::OpenEvent(SYNCHRONIZE, FALSE, argv[0])); | |
| 40 return event_open.Get() ? SBOX_TEST_SUCCEEDED : SBOX_TEST_FAILED; | |
| 41 } | |
| 42 | |
| 43 TEST(UnloadDllTest, BaselineAvicapDll) { | |
| 44 TestRunner runner; | |
| 45 runner.SetTestState(BEFORE_REVERT); | |
| 46 runner.SetTimeout(2000); | |
| 47 // Add a sync rule, because that ensures that the interception agent has | |
| 48 // more than one item in its internal table. | |
| 49 EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_SYNC, | |
| 50 TargetPolicy::EVENTS_ALLOW_ANY, L"t0001")); | |
| 51 | |
| 52 // Note for the puzzled: avicap32.dll is a 64-bit dll in 64-bit versions of | |
| 53 // windows so this test and the others just work. | |
| 54 EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"UseOneDLL L avicap32.dll")); | |
| 55 EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"UseOneDLL B avicap32.dll")); | |
| 56 } | |
| 57 | |
| 58 TEST(UnloadDllTest, UnloadAviCapDllNoPatching) { | |
| 59 TestRunner runner; | |
| 60 runner.SetTestState(BEFORE_REVERT); | |
| 61 runner.SetTimeout(2000); | |
| 62 sandbox::TargetPolicy* policy = runner.GetPolicy(); | |
| 63 policy->AddDllToUnload(L"avicap32.dll"); | |
| 64 EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL L avicap32.dll")); | |
| 65 EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL B avicap32.dll")); | |
| 66 } | |
| 67 | |
| 68 TEST(UnloadDllTest, UnloadAviCapDllWithPatching) { | |
| 69 TestRunner runner; | |
| 70 runner.SetTimeout(2000); | |
| 71 runner.SetTestState(BEFORE_REVERT); | |
| 72 sandbox::TargetPolicy* policy = runner.GetPolicy(); | |
| 73 policy->AddDllToUnload(L"avicap32.dll"); | |
| 74 | |
| 75 base::win::ScopedHandle handle1(::CreateEvent( | |
| 76 NULL, FALSE, FALSE, L"tst0001")); | |
| 77 | |
| 78 // Add a couple of rules that ensures that the interception agent add EAT | |
| 79 // patching on the client which makes sure that the unload dll record does | |
| 80 // not interact badly with them. | |
| 81 EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_REGISTRY, | |
| 82 TargetPolicy::REG_ALLOW_ANY, | |
| 83 L"HKEY_LOCAL_MACHINE\\Software\\Microsoft")); | |
| 84 EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_SYNC, | |
| 85 TargetPolicy::EVENTS_ALLOW_ANY, L"tst0001")); | |
| 86 | |
| 87 EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL L avicap32.dll")); | |
| 88 | |
| 89 runner.SetTestState(AFTER_REVERT); | |
| 90 EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"SimpleOpenEvent tst0001")); | |
| 91 } | |
| 92 | |
| 93 } // namespace sandbox | |
| OLD | NEW |