Chromium Code Reviews| 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..347cecbf488e098c654fc735155598120c3ed3e6 |
| --- /dev/null |
| +++ b/sandbox/win/tests/integration_tests/hooking_dll.cc |
| @@ -0,0 +1,48 @@ |
| +// 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") |
| + |
| +void SetHook(HHOOK handle) { |
| + hook = handle; |
| + return; |
| +} |
| + |
| +bool WasHookCalled() { |
| + return hook_called; |
| +} |
| + |
| +LRESULT HookProc(int code, WPARAM wParam, LPARAM lParam) { |
| + hook_called = true; |
| + // 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 |
| + // test process. |
| + return CallNextHookEx(hook, code, wParam, lParam); |
| +} |
| + |
| +BOOL WINAPI DllMain(__in HINSTANCE hinstDLL, |
| + __in DWORD reason, |
| + __in LPVOID lpvReserved) { |
|
robertshield
2016/04/14 03:57:29
nit: don't need the __in annotations, naming conve
penny
2016/04/24 18:50:49
Done.
|
| + return TRUE; |
| +} |