Francois Gouget : shell32/tests: Add traces to the WaitForInputIdle() hooking process.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 759bd96ac227042024481e9634a3d575c18aee01
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=759bd96ac227042024481e9634...

Author: Francois Gouget 
Date:   Fri Jan 29 00:14:38 2016 +0100

shell32/tests: Add traces to the WaitForInputIdle() hooking process.

Signed-off-by: Francois Gouget 
Signed-off-by: Alexandre Julliard

---

dlls/shell32/tests/shlexec.c | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/dlls/shell32/tests/shlexec.c b/dlls/shell32/tests/shlexec.c
index 7844722..af4563e 100644
--- a/dlls/shell32/tests/shlexec.c
+++ b/dlls/shell32/tests/shlexec.c
@@ -2254,6 +2254,8 @@ static dde_tests_t dde_tests[] =

static DWORD WINAPI hooked_WaitForInputIdle(HANDLE process, DWORD timeout)
 {
+    if (winetest_debug > 1)
+        trace("WaitForInputIdle() waiting for dde event\n");
     return WaitForSingleObject(dde_ready_event, timeout);
 }

@@ -2272,6 +2274,7 @@ static void hook_WaitForInputIdle(DWORD (WINAPI *new_func)(HANDLE, DWORD))
     PIMAGE_NT_HEADERS nt_headers;
     DWORD import_directory_rva;
     PIMAGE_IMPORT_DESCRIPTOR import_descriptor;
+    int hook_count = 0;

base = (char *) GetModuleHandleA("shell32.dll");
     nt_headers = (PIMAGE_NT_HEADERS)(base + ((PIMAGE_DOS_HEADER) base)->e_lfanew);
@@ -2309,6 +2312,9 @@ static void hook_WaitForInputIdle(DWORD (WINAPI *new_func)(HANDLE, DWORD))
                         VirtualProtect(&iat_entry->u1.Function, sizeof(ULONG_PTR), PAGE_READWRITE, &old_prot);
                         iat_entry->u1.Function = (ULONG_PTR) new_func;
                         VirtualProtect(&iat_entry->u1.Function, sizeof(ULONG_PTR), old_prot, &old_prot);
+                        if (winetest_debug > 1)
+                            trace("Hooked %s.WaitForInputIdle\n", import_module_name);
+                        hook_count++;
                         break;
                     }
                 }
@@ -2320,6 +2326,7 @@ static void hook_WaitForInputIdle(DWORD (WINAPI *new_func)(HANDLE, DWORD))

import_descriptor++;
     }
+    ok(hook_count, "Could not hook WaitForInputIdle()\n");
 }

static void test_dde(void)