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/src/sandbox.h" | |
7 #include "sandbox/src/sandbox_factory.h" | |
8 #include "sandbox/src/target_services.h" | |
9 #include "sandbox/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 // Flaky on windows, see http://crbug.com/80569. | |
44 TEST(UnloadDllTest, DISABLED_BaselineAvicapDll) { | |
45 TestRunner runner; | |
46 runner.SetTestState(BEFORE_REVERT); | |
47 runner.SetTimeout(2000); | |
48 // Add a sync rule, because that ensures that the interception agent has | |
49 // more than one item in its internal table. | |
50 EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_SYNC, | |
51 TargetPolicy::EVENTS_ALLOW_ANY, L"t0001")); | |
52 | |
53 // Note for the puzzled: avicap32.dll is a 64-bit dll in 64-bit versions of | |
54 // windows so this test and the others just work. | |
55 EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"UseOneDLL L avicap32.dll")); | |
56 EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"UseOneDLL B avicap32.dll")); | |
57 } | |
58 | |
59 // Flaky on windows, see http://crbug.com/80569. | |
60 TEST(UnloadDllTest, DISABLED_UnloadAviCapDllNoPatching) { | |
61 TestRunner runner; | |
62 runner.SetTestState(BEFORE_REVERT); | |
63 runner.SetTimeout(2000); | |
64 sandbox::TargetPolicy* policy = runner.GetPolicy(); | |
65 policy->AddDllToUnload(L"avicap32.dll"); | |
66 EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL L avicap32.dll")); | |
67 EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL B avicap32.dll")); | |
68 } | |
69 | |
70 // Flaky: http://crbug.com/38404 | |
71 TEST(UnloadDllTest, DISABLED_UnloadAviCapDllWithPatching) { | |
72 TestRunner runner; | |
73 runner.SetTimeout(2000); | |
74 runner.SetTestState(BEFORE_REVERT); | |
75 sandbox::TargetPolicy* policy = runner.GetPolicy(); | |
76 policy->AddDllToUnload(L"avicap32.dll"); | |
77 | |
78 base::win::ScopedHandle handle1(::CreateEvent( | |
79 NULL, FALSE, FALSE, L"tst0001")); | |
80 | |
81 // Add a couple of rules that ensures that the interception agent add EAT | |
82 // patching on the client which makes sure that the unload dll record does | |
83 // not interact badly with them. | |
84 EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_REGISTRY, | |
85 TargetPolicy::REG_ALLOW_ANY, | |
86 L"HKEY_LOCAL_MACHINE\\Software\\Microsoft")); | |
87 EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_SYNC, | |
88 TargetPolicy::EVENTS_ALLOW_ANY, L"tst0001")); | |
89 | |
90 EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL L avicap32.dll")); | |
91 | |
92 runner.SetTestState(AFTER_REVERT); | |
93 EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"SimpleOpenEvent tst0001")); | |
94 } | |
95 | |
96 } // namespace sandbox | |
OLD | NEW |