| Index: chrome/app/delay_load_hook_unittest_win.cc
|
| diff --git a/chrome/app/delay_load_hook_unittest_win.cc b/chrome/app/delay_load_hook_unittest_win.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..e599e79bc670a6efa425a78603292d6e1dde228a
|
| --- /dev/null
|
| +++ b/chrome/app/delay_load_hook_unittest_win.cc
|
| @@ -0,0 +1,90 @@
|
| +// Copyright (c) 2013 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include <windows.h>
|
| +#if defined(_WIN32_WINNT_WIN8) && _MSC_VER < 1700
|
| +// The Windows 8 SDK defines FACILITY_VISUALCPP in winerror.h, and in
|
| +// delayimp.h previous to VS2012.
|
| +#undef FACILITY_VISUALCPP
|
| +#endif
|
| +#include <DelayIMP.h>
|
| +
|
| +#include "base/basictypes.h"
|
| +#include "base/compiler_specific.h"
|
| +#include "base/scoped_native_library.h"
|
| +#include "chrome/app/delay_load_hook_win.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +
|
| +namespace {
|
| +
|
| +class ChromeDelayLoadHookTest : public testing::Test {
|
| + public:
|
| + ChromeDelayLoadHookTest() : proc_ptr_(NULL) {
|
| + }
|
| +
|
| + virtual void SetUp() OVERRIDE {
|
| + SetupInfo("kernel32.dll");
|
| + }
|
| +
|
| + void SetupInfo(const char* dll_name) {
|
| + info_.cb = sizeof(info_);
|
| + info_.pidd = NULL;
|
| + info_.ppfn = &proc_ptr_;
|
| + info_.szDll = dll_name;
|
| + info_.dlp.fImportByName = TRUE;
|
| + info_.dlp.szProcName = "CreateFileA";
|
| + info_.hmodCur = NULL;
|
| + info_.pfnCur = NULL;
|
| + info_.dwLastError = 0;
|
| + }
|
| +
|
| + FARPROC proc_ptr_;
|
| + DelayLoadInfo info_;
|
| +};
|
| +
|
| +} // namespace
|
| +
|
| +TEST_F(ChromeDelayLoadHookTest, HooksAreSetAtLinkTime) {
|
| + // This test verifies that referencing the ChromeDelayLoadHook at link
|
| + // time results in overriding the delay loader's hook instances in the CRT
|
| + // ropriately.
|
| + EXPECT_TRUE(__pfnDliNotifyHook2 == ChromeDelayLoadHook);
|
| + EXPECT_TRUE(__pfnDliFailureHook2 == ChromeDelayLoadHook);
|
| +}
|
| +
|
| +TEST_F(ChromeDelayLoadHookTest, NonSuffixedDllsAreNotHandled) {
|
| + // The hook should ignore non-suffixed DLLs.
|
| + SetupInfo("kernel32.dll");
|
| +
|
| + HMODULE none = reinterpret_cast<HMODULE>(
|
| + ChromeDelayLoadHook(dliNotePreLoadLibrary, &info_));
|
| + // Make sure the library is released on exit.
|
| + base::ScopedNativeLibrary lib_holder(none);
|
| +
|
| + ASSERT_TRUE(none == NULL);
|
| +}
|
| +
|
| +TEST_F(ChromeDelayLoadHookTest, SuffixedDllsAreRedirected) {
|
| + // Check that a DLL name of the form "foo-delay.dll" gets redirected to
|
| + // the "foo.dll".
|
| + SetupInfo("kernel32-delay.dll");
|
| + HMODULE kernel32 = reinterpret_cast<HMODULE>(
|
| + ChromeDelayLoadHook(dliNotePreLoadLibrary, &info_));
|
| +
|
| + // Make sure the library is released on exit.
|
| + base::ScopedNativeLibrary lib_holder(kernel32);
|
| +
|
| + ASSERT_TRUE(kernel32 == ::GetModuleHandle(L"kernel32.dll"));
|
| +}
|
| +
|
| +TEST_F(ChromeDelayLoadHookTest, IgnoresUnhandledNotifications) {
|
| + SetupInfo("kernel32-delay.dll");
|
| +
|
| + // The hook should ignore all notifications but the preload notifications.
|
| + EXPECT_TRUE(ChromeDelayLoadHook(dliNoteStartProcessing, &info_) == NULL);
|
| + EXPECT_TRUE(ChromeDelayLoadHook(dliNotePreGetProcAddress, &info_) == NULL);
|
| + EXPECT_TRUE(ChromeDelayLoadHook(dliNoteEndProcessing, &info_) == NULL);
|
| + EXPECT_TRUE(ChromeDelayLoadHook(dliFailLoadLib, &info_) == NULL);
|
| + EXPECT_TRUE(ChromeDelayLoadHook(dliFailGetProc, &info_) == NULL);
|
| +}
|
|
|