sandbox/win/src/unload_dll_test.cc

   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