OLD | NEW |
1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <windows.h> | 5 #include <windows.h> |
6 #if defined(_WIN32_WINNT_WIN8) && _MSC_VER < 1700 | 6 #if defined(_WIN32_WINNT_WIN8) && _MSC_VER < 1700 |
7 // The Windows 8 SDK defines FACILITY_VISUALCPP in winerror.h, and in | 7 // The Windows 8 SDK defines FACILITY_VISUALCPP in winerror.h, and in |
8 // delayimp.h previous to VS2012. | 8 // delayimp.h previous to VS2012. |
9 #undef FACILITY_VISUALCPP | 9 #undef FACILITY_VISUALCPP |
10 #endif | 10 #endif |
11 #include <DelayIMP.h> | 11 #include <DelayIMP.h> |
12 | 12 |
13 #include "base/basictypes.h" | 13 #include "base/basictypes.h" |
14 #include "base/compiler_specific.h" | 14 #include "base/compiler_specific.h" |
15 #include "base/scoped_native_library.h" | 15 #include "base/scoped_native_library.h" |
16 #include "chrome/app/delay_load_hook_win.h" | 16 #include "chrome/app/delay_load_hook_win.h" |
17 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
18 | 18 |
19 namespace { | 19 namespace { |
20 | 20 |
21 class ChromeDelayLoadHookTest : public testing::Test { | 21 class ChromeDelayLoadHookTest : public testing::Test { |
22 public: | 22 public: |
23 ChromeDelayLoadHookTest() : proc_ptr_(NULL) { | 23 ChromeDelayLoadHookTest() : proc_ptr_(NULL) { |
24 } | 24 } |
25 | 25 |
26 virtual void SetUp() OVERRIDE { | 26 virtual void SetUp() override { |
27 SetupInfo("kernel32.dll"); | 27 SetupInfo("kernel32.dll"); |
28 } | 28 } |
29 | 29 |
30 void SetupInfo(const char* dll_name) { | 30 void SetupInfo(const char* dll_name) { |
31 info_.cb = sizeof(info_); | 31 info_.cb = sizeof(info_); |
32 info_.pidd = NULL; | 32 info_.pidd = NULL; |
33 info_.ppfn = &proc_ptr_; | 33 info_.ppfn = &proc_ptr_; |
34 info_.szDll = dll_name; | 34 info_.szDll = dll_name; |
35 info_.dlp.fImportByName = TRUE; | 35 info_.dlp.fImportByName = TRUE; |
36 info_.dlp.szProcName = "CreateFileA"; | 36 info_.dlp.szProcName = "CreateFileA"; |
(...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
81 TEST_F(ChromeDelayLoadHookTest, IgnoresUnhandledNotifications) { | 81 TEST_F(ChromeDelayLoadHookTest, IgnoresUnhandledNotifications) { |
82 SetupInfo("kernel32-delay.dll"); | 82 SetupInfo("kernel32-delay.dll"); |
83 | 83 |
84 // The hook should ignore all notifications but the preload notifications. | 84 // The hook should ignore all notifications but the preload notifications. |
85 EXPECT_TRUE(ChromeDelayLoadHook(dliNoteStartProcessing, &info_) == NULL); | 85 EXPECT_TRUE(ChromeDelayLoadHook(dliNoteStartProcessing, &info_) == NULL); |
86 EXPECT_TRUE(ChromeDelayLoadHook(dliNotePreGetProcAddress, &info_) == NULL); | 86 EXPECT_TRUE(ChromeDelayLoadHook(dliNotePreGetProcAddress, &info_) == NULL); |
87 EXPECT_TRUE(ChromeDelayLoadHook(dliNoteEndProcessing, &info_) == NULL); | 87 EXPECT_TRUE(ChromeDelayLoadHook(dliNoteEndProcessing, &info_) == NULL); |
88 EXPECT_TRUE(ChromeDelayLoadHook(dliFailLoadLib, &info_) == NULL); | 88 EXPECT_TRUE(ChromeDelayLoadHook(dliFailLoadLib, &info_) == NULL); |
89 EXPECT_TRUE(ChromeDelayLoadHook(dliFailGetProc, &info_) == NULL); | 89 EXPECT_TRUE(ChromeDelayLoadHook(dliFailGetProc, &info_) == NULL); |
90 } | 90 } |
OLD | NEW |