michael@0: // Copyright (c) 2012 The Chromium Authors. All rights reserved.
michael@0: // Use of this source code is governed by a BSD-style license that can be
michael@0: // found in the LICENSE file.
michael@0: 
michael@0: #include "base/win/scoped_handle.h"
michael@0: #include "sandbox/win/src/sandbox.h"
michael@0: #include "sandbox/win/src/sandbox_factory.h"
michael@0: #include "sandbox/win/src/target_services.h"
michael@0: #include "sandbox/win/tests/common/controller.h"
michael@0: #include "testing/gtest/include/gtest/gtest.h"
michael@0: 
michael@0: namespace sandbox {
michael@0: 
michael@0: // Loads and or unloads a DLL passed in the second parameter of argv.
michael@0: // The first parameter of argv is 'L' = load, 'U' = unload or 'B' for both.
michael@0: SBOX_TESTS_COMMAND int UseOneDLL(int argc, wchar_t **argv) {
michael@0:   if (argc != 2)
michael@0:     return SBOX_TEST_FAILED_TO_RUN_TEST;
michael@0:   int rv = SBOX_TEST_FAILED_TO_RUN_TEST;
michael@0: 
michael@0:   wchar_t option = (argv[0])[0];
michael@0:   if ((option == L'L') || (option == L'B')) {
michael@0:     HMODULE module1 = ::LoadLibraryW(argv[1]);
michael@0:     rv = (module1 == NULL) ? SBOX_TEST_FAILED : SBOX_TEST_SUCCEEDED;
michael@0:   }
michael@0: 
michael@0:   if ((option == L'U') || (option == L'B')) {
michael@0:     HMODULE module2 = ::GetModuleHandleW(argv[1]);
michael@0:     rv = ::FreeLibrary(module2) ? SBOX_TEST_SUCCEEDED : SBOX_TEST_FAILED;
michael@0:   }
michael@0:   return rv;
michael@0: }
michael@0: 
michael@0: // Opens an event passed as the first parameter of argv.
michael@0: SBOX_TESTS_COMMAND int SimpleOpenEvent(int argc, wchar_t **argv) {
michael@0:   if (argc != 1)
michael@0:     return SBOX_TEST_FAILED_TO_EXECUTE_COMMAND;
michael@0: 
michael@0:   base::win::ScopedHandle event_open(::OpenEvent(SYNCHRONIZE, FALSE, argv[0]));
michael@0:   return event_open.Get() ? SBOX_TEST_SUCCEEDED : SBOX_TEST_FAILED;
michael@0: }
michael@0: 
michael@0: // Flaky on windows, see http://crbug.com/80569.
michael@0: TEST(UnloadDllTest, DISABLED_BaselineAvicapDll) {
michael@0:   TestRunner runner;
michael@0:   runner.SetTestState(BEFORE_REVERT);
michael@0:   runner.SetTimeout(2000);
michael@0:   // Add a sync rule, because that ensures that the interception agent has
michael@0:   // more than one item in its internal table.
michael@0:   EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_SYNC,
michael@0:                              TargetPolicy::EVENTS_ALLOW_ANY, L"t0001"));
michael@0: 
michael@0:   // Note for the puzzled: avicap32.dll is a 64-bit dll in 64-bit versions of
michael@0:   // windows so this test and the others just work.
michael@0:   EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"UseOneDLL L avicap32.dll"));
michael@0:   EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"UseOneDLL B avicap32.dll"));
michael@0: }
michael@0: 
michael@0: // Flaky on windows, see http://crbug.com/80569.
michael@0: TEST(UnloadDllTest, DISABLED_UnloadAviCapDllNoPatching) {
michael@0:   TestRunner runner;
michael@0:   runner.SetTestState(BEFORE_REVERT);
michael@0:   runner.SetTimeout(2000);
michael@0:   sandbox::TargetPolicy* policy = runner.GetPolicy();
michael@0:   policy->AddDllToUnload(L"avicap32.dll");
michael@0:   EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL L avicap32.dll"));
michael@0:   EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL B avicap32.dll"));
michael@0: }
michael@0: 
michael@0: // Flaky: http://crbug.com/38404
michael@0: TEST(UnloadDllTest, DISABLED_UnloadAviCapDllWithPatching) {
michael@0:   TestRunner runner;
michael@0:   runner.SetTimeout(2000);
michael@0:   runner.SetTestState(BEFORE_REVERT);
michael@0:   sandbox::TargetPolicy* policy = runner.GetPolicy();
michael@0:   policy->AddDllToUnload(L"avicap32.dll");
michael@0: 
michael@0:   base::win::ScopedHandle handle1(::CreateEvent(
michael@0:       NULL, FALSE, FALSE, L"tst0001"));
michael@0: 
michael@0:   // Add a couple of rules that ensures that the interception agent add EAT
michael@0:   // patching on the client which makes sure that the unload dll record does
michael@0:   // not interact badly with them.
michael@0:   EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_REGISTRY,
michael@0:                              TargetPolicy::REG_ALLOW_ANY,
michael@0:                              L"HKEY_LOCAL_MACHINE\\Software\\Microsoft"));
michael@0:   EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_SYNC,
michael@0:                              TargetPolicy::EVENTS_ALLOW_ANY, L"tst0001"));
michael@0: 
michael@0:   EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL L avicap32.dll"));
michael@0: 
michael@0:   runner.SetTestState(AFTER_REVERT);
michael@0:   EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"SimpleOpenEvent tst0001"));
michael@0: }
michael@0: 
michael@0: }  // namespace sandbox