Index: sandbox/win/tests/integration_tests/hooking_dll.cc |
diff --git a/sandbox/win/tests/integration_tests/hooking_dll.cc b/sandbox/win/tests/integration_tests/hooking_dll.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f275ae6e940208bc65b71ddeced35199cb7827a7 |
--- /dev/null |
+++ b/sandbox/win/tests/integration_tests/hooking_dll.cc |
@@ -0,0 +1,64 @@ |
+// Copyright 2016 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 <stdio.h> |
+#include <windows.h> |
+ |
+#define _DLL_EXPORTING |
+#include "integration_tests_common.h" |
+ |
+// This data section creates a common area that is accessible |
+// to all instances of the DLL (in every process). They map to |
+// the same physical memory location. |
+// **Note that each instance of this DLL runs in the context of |
+// the process it's injected into, so things like pointers and |
+// addresses won't work. |
+// **Note that any variables must be initialized to put them in |
+// the specified segment, otherwise they will end up in the |
+// default data segment. |
+#pragma data_seg(".hook") |
+HHOOK hook = NULL; |
+bool hook_called = false; |
+#pragma data_seg() |
+#pragma comment(linker, "/SECTION:.hook,RWS") |
+ |
+namespace { |
+ |
+HANDLE event = NULL; |
+} |
+ |
+void SetHook(HHOOK hook_handle) { |
+ hook = hook_handle; |
+ |
+ return; |
+} |
+ |
+bool WasHookCalled() { |
+ return hook_called; |
+} |
+ |
+LRESULT HookProc(int code, WPARAM w_param, LPARAM l_param) { |
+ hook_called = true; |
+ if (event) |
+ ::SetEvent(event); |
+ |
+ // Recent versions of Windows do not require the HHOOK to be passed along, |
+ // but I'm doing it here to show the shared use of the HHOOK variable in |
+ // the shared data segment. It is set by the instance of the DLL in the |
+ // main test process. |
+ return CallNextHookEx(hook, code, w_param, l_param); |
+} |
+ |
+BOOL WINAPI DllMain(HINSTANCE instance, DWORD reason, LPVOID reserved) { |
+ if (reason == DLL_PROCESS_ATTACH) { |
+ // The testing process should have set up this named event already |
+ // (if the test needs this event to be signaled). |
+ event = ::OpenEventW(EVENT_MODIFY_STATE, FALSE, g_hook_event); |
+ } |
+ |
+ if (reason == DLL_PROCESS_DETACH) |
+ ::CloseHandle(event); |
+ |
+ return TRUE; |
+} |