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
12namespace 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.
16SBOX_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.
35SBOX_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.
44TEST(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.
60TEST(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
71TEST(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
97