15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_handle.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sandbox/win/src/sandbox.h" 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sandbox/win/src/sandbox_factory.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sandbox/win/src/target_services.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sandbox/win/tests/common/controller.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace sandbox { 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Loads and or unloads a DLL passed in the second parameter of argv. 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The first parameter of argv is 'L' = load, 'U' = unload or 'B' for both. 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SBOX_TESTS_COMMAND int UseOneDLL(int argc, wchar_t **argv) { 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (argc != 2) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return SBOX_TEST_FAILED_TO_RUN_TEST; 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rv = SBOX_TEST_FAILED_TO_RUN_TEST; 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) wchar_t option = (argv[0])[0]; 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ((option == L'L') || (option == L'B')) { 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HMODULE module1 = ::LoadLibraryW(argv[1]); 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) rv = (module1 == NULL) ? SBOX_TEST_FAILED : SBOX_TEST_SUCCEEDED; 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if ((option == L'U') || (option == L'B')) { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HMODULE module2 = ::GetModuleHandleW(argv[1]); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) rv = ::FreeLibrary(module2) ? SBOX_TEST_SUCCEEDED : SBOX_TEST_FAILED; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return rv; 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Opens an event passed as the first parameter of argv. 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SBOX_TESTS_COMMAND int SimpleOpenEvent(int argc, wchar_t **argv) { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (argc != 1) 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return SBOX_TEST_FAILED_TO_EXECUTE_COMMAND; 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::ScopedHandle event_open(::OpenEvent(SYNCHRONIZE, FALSE, argv[0])); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return event_open.Get() ? SBOX_TEST_SUCCEEDED : SBOX_TEST_FAILED; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Flaky on windows, see http://crbug.com/80569. 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(UnloadDllTest, DISABLED_BaselineAvicapDll) { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestRunner runner; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) runner.SetTestState(BEFORE_REVERT); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) runner.SetTimeout(2000); 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Add a sync rule, because that ensures that the interception agent has 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // more than one item in its internal table. 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_SYNC, 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TargetPolicy::EVENTS_ALLOW_ANY, L"t0001")); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Note for the puzzled: avicap32.dll is a 64-bit dll in 64-bit versions of 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // windows so this test and the others just work. 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"UseOneDLL L avicap32.dll")); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"UseOneDLL B avicap32.dll")); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Flaky on windows, see http://crbug.com/80569. 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(UnloadDllTest, DISABLED_UnloadAviCapDllNoPatching) { 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestRunner runner; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) runner.SetTestState(BEFORE_REVERT); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) runner.SetTimeout(2000); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sandbox::TargetPolicy* policy = runner.GetPolicy(); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) policy->AddDllToUnload(L"avicap32.dll"); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL L avicap32.dll")); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL B avicap32.dll")); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Flaky: http://crbug.com/38404 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(UnloadDllTest, DISABLED_UnloadAviCapDllWithPatching) { 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestRunner runner; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) runner.SetTimeout(2000); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) runner.SetTestState(BEFORE_REVERT); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sandbox::TargetPolicy* policy = runner.GetPolicy(); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) policy->AddDllToUnload(L"avicap32.dll"); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::ScopedHandle handle1(::CreateEvent( 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, FALSE, FALSE, L"tst0001")); 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Add a couple of rules that ensures that the interception agent add EAT 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // patching on the client which makes sure that the unload dll record does 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // not interact badly with them. 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_REGISTRY, 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TargetPolicy::REG_ALLOW_ANY, 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L"HKEY_LOCAL_MACHINE\\Software\\Microsoft")); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(runner.AddRule(TargetPolicy::SUBSYS_SYNC, 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TargetPolicy::EVENTS_ALLOW_ANY, L"tst0001")); 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SBOX_TEST_FAILED, runner.RunTest(L"UseOneDLL L avicap32.dll")); 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) runner.SetTestState(AFTER_REVERT); 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SBOX_TEST_SUCCEEDED, runner.RunTest(L"SimpleOpenEvent tst0001")); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace sandbox 97