ReactOS  0.4.14-dev-115-g4576127
delayimp.cpp File Reference
#include <apitest.h>
#include <strsafe.h>
#include <delayimp.h>
#include <mmsystem.h>
#include <winver.h>
#include <shlwapi.h>
#include <intshcut.h>
#include <sfc.h>
#include <imagehlp.h>
Include dependency graph for delayimp.cpp:

Go to the source code of this file.

Classes

struct  UnProtect
 

Macros

#define FACILITY_VISUALCPP   ((LONG)0x6d)
 
#define VcppException(sev, err)   ((sev) | (FACILITY_VISUALCPP<<16) | err)
 
#define WINMM_DLLNAME   "WINMM.dll"
 
#define LAST_DLI   333
 
#define CheckDli   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : CheckDli_imp
 
#define CheckDliDone   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : CheckDliDone_imp
 

Functions

chartarget (PDelayLoadInfo pdli)
 
static void SetExpectedDli (unsigned *order)
 
static void CheckDli_imp (unsigned dliNotify, PDelayLoadInfo pdli, BOOL ErrorHandler)
 
static void CheckDliDone_imp ()
 
INT_PTR WINAPI MyFunction ()
 
BOOL WINAPI MySfcIsKeyProtected (HKEY KeyHandle, LPCWSTR SubKeyName, REGSAM KeySam)
 
FARPROC WINAPI DliHook (unsigned dliNotify, PDelayLoadInfo pdli)
 
FARPROC WINAPI DliFailHook (unsigned dliNotify, PDelayLoadInfo pdli)
 
LONG ExceptionFilter (IN PEXCEPTION_POINTERS ExceptionInfo, ULONG ExceptionCode)
 
template<typename PTR >
PTR Rva2Addr (PIMAGE_DOS_HEADER dos, RVA rva)
 
 START_TEST (delayimp)
 

Variables

bool g_BreakFunctionName = false
 
bool g_BrokenFunctionName = false
 
bool g_BypassMost = false
 
bool g_ExceptionIsModule = false
 
bool g_ImportByName = true
 
const charg_ExpectedDll = NULL
 
const charg_ExpectedName = NULL
 
char g_Target [100] = { 0 }
 
unsigned * g_DliHookExpected = NULL
 
size_t g_DliHookIndex = 0
 
static HMODULE g_VersionDll
 
PfnDliHook __pfnDliNotifyHook2 = DliHook
 
bool g_UsePointers = false
 
unsigned g_winmm_snd_play_a [] = { dliStartProcessing, dliNotePreLoadLibrary, dliNotePreGetProcAddress, dliNoteEndProcessing, LAST_DLI }
 
unsigned g_winmm_snd_play_w [] = { dliStartProcessing, dliNotePreGetProcAddress, dliNoteEndProcessing, LAST_DLI }
 
unsigned g_winmm_play_w [] = { dliStartProcessing, dliNotePreGetProcAddress, dliFailGetProc, dliNoteEndProcessing, LAST_DLI }
 
unsigned g_sfc_key [] = { dliStartProcessing, dliNotePreLoadLibrary, dliFailLoadLib, dliNotePreGetProcAddress, dliNoteEndProcessing, LAST_DLI }
 
unsigned g_sfc_file [] = { dliStartProcessing, dliNotePreGetProcAddress, dliNoteEndProcessing, LAST_DLI }
 
unsigned g_version_a [] = { dliStartProcessing, dliNotePreLoadLibrary, dliNotePreGetProcAddress, dliFailGetProc, dliNoteEndProcessing, LAST_DLI }
 
unsigned g_version_w [] = { dliStartProcessing, dliNotePreGetProcAddress, dliFailGetProc, dliNoteEndProcessing, LAST_DLI }
 
unsigned g_scard [] = { dliStartProcessing, dliNoteEndProcessing, LAST_DLI }
 
unsigned g_shlwapi [] = { dliStartProcessing, dliNotePreLoadLibrary, dliNotePreGetProcAddress, dliNoteEndProcessing, LAST_DLI }
 
unsigned g_imagehlp [] = { dliStartProcessing, dliNotePreLoadLibrary, dliFailLoadLib, LAST_DLI }
 

Macro Definition Documentation

◆ CheckDli

#define CheckDli   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : CheckDli_imp

Definition at line 125 of file delayimp.cpp.

◆ CheckDliDone

#define CheckDliDone   (winetest_set_location(__FILE__, __LINE__), 0) ? (void)0 : CheckDliDone_imp

Definition at line 126 of file delayimp.cpp.

◆ FACILITY_VISUALCPP

#define FACILITY_VISUALCPP   ((LONG)0x6d)

Definition at line 25 of file delayimp.cpp.

◆ LAST_DLI

#define LAST_DLI   333

Definition at line 85 of file delayimp.cpp.

◆ VcppException

#define VcppException (   sev,
  err 
)    ((sev) | (FACILITY_VISUALCPP<<16) | err)

Definition at line 29 of file delayimp.cpp.

◆ WINMM_DLLNAME

#define WINMM_DLLNAME   "WINMM.dll"

Definition at line 35 of file delayimp.cpp.

Function Documentation

◆ CheckDli_imp()

static void CheckDli_imp ( unsigned  dliNotify,
PDelayLoadInfo  pdli,
BOOL  ErrorHandler 
)
static

Definition at line 94 of file delayimp.cpp.

95 {
96  if (!g_DliHookExpected) return;
97 
98  winetest_ok(dliNotify == g_DliHookExpected[g_DliHookIndex], "Expected dliNotify to be %u, was: %u for %s\n",
99  g_DliHookExpected[g_DliHookIndex], dliNotify, target(pdli));
100  if (ErrorHandler)
101  {
102  winetest_ok(dliNotify == dliFailGetProc || dliNotify == dliFailLoadLib,
103  "Expected code %u to be processed by the Hook, not the ErrorHandler for %s\n", dliNotify, target(pdli));
104  }
105  else
106  {
107  winetest_ok(dliNotify == dliStartProcessing || dliNotify == dliNotePreLoadLibrary ||
108  dliNotify == dliNotePreGetProcAddress || dliNotify == dliNoteEndProcessing,
109  "Expected code %u to be processed by the ErrorHandler, not the Hook for %s\n", dliNotify, target(pdli));
110  }
112  g_DliHookIndex++;
113 }
size_t g_DliHookIndex
Definition: delayimp.cpp:84
void __winetest_cdecl winetest_ok(int condition, const char *msg,...)
#define LAST_DLI
Definition: delayimp.cpp:85
unsigned * g_DliHookExpected
Definition: delayimp.cpp:83
char * target(PDelayLoadInfo pdli)
Definition: delayimp.cpp:47

◆ CheckDliDone_imp()

static void CheckDliDone_imp ( )
static

Definition at line 115 of file delayimp.cpp.

116 {
117  if (!g_DliHookExpected) return;
119  "Expected g_DliHookExpected[g_DliHookIndex] to be %u, was: %u for %s\n",
122  g_Target[0] = '\0';
123 }
char g_Target[100]
Definition: delayimp.cpp:45
size_t g_DliHookIndex
Definition: delayimp.cpp:84
smooth NULL
Definition: ftsmooth.c:416
void __winetest_cdecl winetest_ok(int condition, const char *msg,...)
#define LAST_DLI
Definition: delayimp.cpp:85
unsigned * g_DliHookExpected
Definition: delayimp.cpp:83
char * target(PDelayLoadInfo pdli)
Definition: delayimp.cpp:47

◆ DliFailHook()

FARPROC WINAPI DliFailHook ( unsigned  dliNotify,
PDelayLoadInfo  pdli 
)

Definition at line 255 of file delayimp.cpp.

256 {
257  ok(pdli && pdli->cb >= 36,
258  "Expected a valid pointer with a struct that is big enough: %p, %lu\n", pdli, pdli ? pdli->cb : 0u);
259  if (!pdli || pdli->cb < 36) return NULL;
260 
261  CheckDli(dliNotify, pdli, TRUE);
262 
263  /* Redirections / fixes */
264  if (dliNotify == dliFailLoadLib)
265  {
266  if (!_stricmp(pdli->szDll, "l_m_os.dll"))
267  return (FARPROC)LoadLibraryA("sfc_os.dll");
268  }
269  else if (dliNotify == dliFailGetProc)
270  {
271  if (pdli->dlp.fImportByName && pdli->hmodCur == (HMODULE)1)
272  {
273  return GetProcAddress(g_VersionDll, pdli->dlp.szProcName);
274  }
275  }
276 
277  /* Parameter validation */
278  ok(pdli->ppfn != NULL, "Expected ppfn to be valid, was NULL for %s\n", target(pdli));
279  ok(pdli->szDll != NULL, "Expected szDll to be valid, was NULL for %s\n", target(pdli));
280  if (pdli->dlp.fImportByName)
281  ok(pdli->dlp.szProcName != NULL, "Expected szProcName to be valid, was NULL for %s\n", target(pdli));
282  else
283  ok(pdli->dlp.dwOrdinal != 0, "Expected dwOrdinal to be valid, was NULL for %s\n", target(pdli));
284  switch(dliNotify)
285  {
286  case dliFailLoadLib:
287  ok(pdli->hmodCur == NULL, "Expected hmodCur to be NULL, was: %p for %s\n", pdli->hmodCur, target(pdli));
288  ok(pdli->pfnCur == NULL, "Expected pfnCur to be NULL, was: %p for %s\n", pdli->pfnCur, target(pdli));
290  "Expected dwLastError to be ERROR_MOD_NOT_FOUND, was: %lu for %s\n", pdli->dwLastError, target(pdli));
291  break;
292  case dliFailGetProc:
293  ok(pdli->hmodCur != NULL, "Expected hmodCur to be valid, was NULL for %s\n", target(pdli));
294  ok(pdli->pfnCur == NULL, "Expected pfnCur to be NULL, was: %p for %s\n", pdli->pfnCur, target(pdli));
296  "Expected dwLastError to be ERROR_PROC_NOT_FOUND, was: %lu for %s\n", pdli->dwLastError, target(pdli));
297  break;
298  }
299 
300  return NULL;
301 }
GLsizei GLenum const GLvoid GLsizei GLenum GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLint GLint GLint GLshort GLshort GLshort GLubyte GLubyte GLubyte GLuint GLuint GLuint GLushort GLushort GLushort GLbyte GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLfloat GLint GLint GLint GLint GLshort GLshort GLshort GLshort GLubyte GLubyte GLubyte GLubyte GLuint GLuint GLuint GLuint GLushort GLushort GLushort GLushort GLboolean const GLdouble const GLfloat const GLint const GLshort const GLbyte const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLdouble const GLfloat const GLfloat const GLint const GLint const GLshort const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort GLenum GLenum GLenum GLfloat GLenum GLint GLenum GLenum GLenum GLfloat GLenum GLenum GLint GLenum GLfloat GLenum GLint GLint GLushort GLenum GLenum GLfloat GLenum GLenum GLint GLfloat const GLubyte GLenum GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLint GLint GLsizei GLsizei GLint GLenum GLenum const GLvoid GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLenum const GLdouble GLenum GLenum const GLfloat GLenum GLenum const GLint GLsizei GLuint GLfloat GLuint GLbitfield GLfloat GLint GLuint GLboolean GLenum GLfloat GLenum GLbitfield GLenum GLfloat GLfloat GLint GLint const GLfloat GLenum GLfloat GLfloat GLint GLint GLfloat GLfloat GLint GLint const GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat const GLdouble * u
Definition: glfuncs.h:240
FARPROC pfnCur
Definition: delayimp.h:73
#define TRUE
Definition: types.h:120
BOOL fImportByName
Definition: delayimp.h:57
#define _stricmp
Definition: cat.c:22
#define CheckDli
Definition: delayimp.cpp:125
LPCSTR szProcName
Definition: delayimp.h:60
HINSTANCE WINAPI DECLSPEC_HOTPATCH LoadLibraryA(LPCSTR lpLibFileName)
Definition: loader.c:111
DelayLoadProc dlp
Definition: delayimp.h:71
DWORD dwLastError
Definition: delayimp.h:74
smooth NULL
Definition: ftsmooth.c:416
HMODULE hmodCur
Definition: delayimp.h:72
DWORD cb
Definition: delayimp.h:67
LPCSTR szDll
Definition: delayimp.h:70
DWORD dwOrdinal
Definition: delayimp.h:61
#define ERROR_PROC_NOT_FOUND
Definition: winerror.h:199
#define ok(value,...)
Definition: atltest.h:57
FARPROC * ppfn
Definition: delayimp.h:69
#define ERROR_MOD_NOT_FOUND
Definition: compat.h:94
GLenum target
Definition: glext.h:7315
#define GetProcAddress(x, y)
Definition: compat.h:410
static HMODULE g_VersionDll
Definition: delayimp.cpp:141
int(* FARPROC)()
Definition: compat.h:28

Referenced by START_TEST().

◆ DliHook()

FARPROC WINAPI DliHook ( unsigned  dliNotify,
PDelayLoadInfo  pdli 
)

Definition at line 142 of file delayimp.cpp.

143 {
144  ok(pdli && pdli->cb >= 36, "Expected a valid pointer with a struct that is big enough: %p, %lu\n",
145  pdli, pdli ? pdli->cb : 0u);
146  if (!pdli || pdli->cb < 36) return NULL;
147 
148  CheckDli(dliNotify, pdli, FALSE);
149 
151  {
152  g_BreakFunctionName = false;
153  g_BrokenFunctionName = true;
154  char* procname = (char*)pdli->dlp.szProcName;
155  UnProtect prot(procname);
156  char c = procname[0];
157  procname[0] = isupper(c) ? tolower(c) : toupper(c);
158  }
159 
160  /* Validate dll name when required */
162  {
163  ok(!strcmp(g_ExpectedDll, pdli->szDll), "Expected szDll to be '%s', but was: '%s'\n", g_ExpectedDll, pdli->szDll);
164  ok(pdli->dlp.fImportByName, "Expected import by name (%s!%s)\n", g_ExpectedDll, g_ExpectedName);
165  if (pdli->dlp.fImportByName)
166  {
167  ok(!strcmp(g_ExpectedName, pdli->dlp.szProcName), "Expected szProcName to be '%s', but was: '%s'\n",
168  g_ExpectedName, pdli->dlp.szProcName);
169  }
170  }
171 
172 
173  if (dliNotify == dliStartProcessing)
174  {
175  /* Test loadlib fail */
176  if (!_stricmp(pdli->szDll, "sfc_os.dll"))
177  {
178  char* dll = (char*)pdli->szDll;
179  UnProtect u(dll);
180  dll[0] = 'l'; dll[1] = '_'; dll[2] = 'm';
181  }
182  if (!_stricmp(pdli->szDll, "imagehlp.dll"))
183  {
184  char* dll = (char*)pdli->szDll;
185  UnProtect u(dll);
186  dll[0] = 'x'; dll[1] = 'x'; dll[2] = 'x'; dll[3] = 'x'; dll[4] = 'x';
187  }
188  /* Test bypass */
189  if (!_stricmp(pdli->szDll, "dbghelp.dll"))
190  return MyFunction;
191  }
192  else if (dliNotify == dliNotePreLoadLibrary)
193  {
194  /* Show that this value is actually used! */
195  if (!_stricmp(pdli->szDll, "version.dll"))
196  {
197  g_VersionDll = LoadLibraryA("version.dll");
198  return (FARPROC)1;
199  }
200 
201  }
202  else if (dliNotify == dliNotePreGetProcAddress)
203  {
204  if (pdli->dlp.fImportByName && !strcmp(pdli->dlp.szProcName, "SfcIsKeyProtected"))
205  {
207  }
208  }
209 
210  /* Parameter validation */
211  ok(pdli->ppfn != NULL, "Expected ppfn to be valid, was NULL for %s\n", target(pdli));
212  ok(pdli->szDll != NULL, "Expected szDll to be valid, was NULL for %s\n", target(pdli));
213  ok(pdli->dwLastError == ERROR_SUCCESS,
214  "Expected dwLastError to be ERROR_SUCCESS, was: %lu for %s\n", pdli->dwLastError, target(pdli));
215  ok(g_ImportByName == !!pdli->dlp.fImportByName, "Expected pdli->dlp.fImportByName to equal g_ImportByname\n");
216  if (pdli->dlp.fImportByName)
217  ok(pdli->dlp.szProcName != NULL, "Expected szProcName to be valid, was NULL for %s\n", target(pdli));
218  else
219  ok(pdli->dlp.dwOrdinal != 0, "Expected dwOrdinal to be valid, was NULL for %s\n", target(pdli));
220  switch(dliNotify)
221  {
222  case dliStartProcessing:
223  ok(pdli->hmodCur == NULL, "Expected hmodCur to be NULL, was: %p for %s\n", pdli->hmodCur, target(pdli));
224  ok(pdli->pfnCur == NULL, "Expected pfnCur to be NULL, was: %p for %s\n", pdli->pfnCur, target(pdli));
225  break;
227  ok(pdli->hmodCur == NULL, "Expected hmodCur to be NULL, was: %p for %s\n", pdli->hmodCur, target(pdli));
228  ok(pdli->pfnCur == NULL, "Expected pfnCur to be NULL, was: %p for %s\n", pdli->pfnCur, target(pdli));
229  break;
231  ok(pdli->hmodCur != NULL, "Expected hmodCur to be valid, was NULL for %s\n", target(pdli));
232  ok(pdli->pfnCur == NULL, "Expected pfnCur to be NULL, was: %p for %s\n", pdli->pfnCur, target(pdli));
233  break;
235  if (!g_BypassMost)
236  ok(pdli->hmodCur != NULL, "Expected hmodCur to be valid, was NULL for %s\n", target(pdli));
237  ok(pdli->pfnCur != NULL, "Expected pfnCur to be a valid pointer, was NULL for %s\n", target(pdli));
239  {
241  ok(targetProc != NULL, "This should not happen, the function i need is unavail! (%s!%s)\n",
243  ok(targetProc == pdli->pfnCur, "Expected pfnCur to be %p, was %p for %s\n", targetProc, pdli->pfnCur, target(pdli));
244  ok(pdli->ppfn && targetProc == *pdli->ppfn,
245  "Expected ppfn to be valid and to result in %p, was: %p(%p) for %s\n",
246  target, pdli->ppfn, pdli->ppfn ? *pdli->ppfn : NULL, target(pdli));
247  }
248  break;
249  default:
250  break;
251  }
252  return NULL;
253 }
GLsizei GLenum const GLvoid GLsizei GLenum GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLint GLint GLint GLshort GLshort GLshort GLubyte GLubyte GLubyte GLuint GLuint GLuint GLushort GLushort GLushort GLbyte GLbyte GLbyte GLbyte GLdouble GLdouble GLdouble GLdouble GLfloat GLfloat GLfloat GLfloat GLint GLint GLint GLint GLshort GLshort GLshort GLshort GLubyte GLubyte GLubyte GLubyte GLuint GLuint GLuint GLuint GLushort GLushort GLushort GLushort GLboolean const GLdouble const GLfloat const GLint const GLshort const GLbyte const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLdouble const GLfloat const GLfloat const GLint const GLint const GLshort const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort const GLdouble const GLfloat const GLint const GLshort GLenum GLenum GLenum GLfloat GLenum GLint GLenum GLenum GLenum GLfloat GLenum GLenum GLint GLenum GLfloat GLenum GLint GLint GLushort GLenum GLenum GLfloat GLenum GLenum GLint GLfloat const GLubyte GLenum GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLint GLint GLsizei GLsizei GLint GLenum GLenum const GLvoid GLenum GLenum const GLfloat GLenum GLenum const GLint GLenum GLenum const GLdouble GLenum GLenum const GLfloat GLenum GLenum const GLint GLsizei GLuint GLfloat GLuint GLbitfield GLfloat GLint GLuint GLboolean GLenum GLfloat GLenum GLbitfield GLenum GLfloat GLfloat GLint GLint const GLfloat GLenum GLfloat GLfloat GLint GLint GLfloat GLfloat GLint GLint const GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat GLint GLfloat GLfloat const GLdouble * u
Definition: glfuncs.h:240
FARPROC pfnCur
Definition: delayimp.h:73
BOOL WINAPI MySfcIsKeyProtected(HKEY KeyHandle, LPCWSTR SubKeyName, REGSAM KeySam)
Definition: delayimp.cpp:135
#define ERROR_SUCCESS
Definition: deptool.c:10
bool g_BypassMost
Definition: delayimp.cpp:40
BOOL fImportByName
Definition: delayimp.h:57
#define _stricmp
Definition: cat.c:22
static HMODULE dll
Definition: str.c:188
#define CheckDli
Definition: delayimp.cpp:125
bool g_ImportByName
Definition: delayimp.cpp:42
INT_PTR WINAPI MyFunction()
Definition: delayimp.cpp:130
LPCSTR szProcName
Definition: delayimp.h:60
HINSTANCE WINAPI DECLSPEC_HOTPATCH LoadLibraryA(LPCSTR lpLibFileName)
Definition: loader.c:111
DelayLoadProc dlp
Definition: delayimp.h:71
DWORD dwLastError
Definition: delayimp.h:74
smooth NULL
Definition: ftsmooth.c:416
HMODULE hmodCur
Definition: delayimp.h:72
int toupper(int c)
Definition: utclib.c:881
#define isupper(c)
Definition: acclib.h:71
bool g_BrokenFunctionName
Definition: delayimp.cpp:39
const GLubyte * c
Definition: glext.h:8905
DWORD cb
Definition: delayimp.h:67
LPCSTR szDll
Definition: delayimp.h:70
HMODULE WINAPI DECLSPEC_HOTPATCH GetModuleHandleA(LPCSTR lpModuleName)
Definition: loader.c:821
DWORD dwOrdinal
Definition: delayimp.h:61
#define ok(value,...)
Definition: atltest.h:57
bool g_BreakFunctionName
Definition: delayimp.cpp:38
FARPROC * ppfn
Definition: delayimp.h:69
const char * g_ExpectedName
Definition: delayimp.cpp:44
GLenum target
Definition: glext.h:7315
int strcmp(const char *String1, const char *String2)
Definition: utclib.c:469
#define GetProcAddress(x, y)
Definition: compat.h:410
const char * g_ExpectedDll
Definition: delayimp.cpp:43
static HMODULE g_VersionDll
Definition: delayimp.cpp:141
int tolower(int c)
Definition: utclib.c:902
int(* FARPROC)()
Definition: compat.h:28

Referenced by START_TEST().

◆ ExceptionFilter()

LONG ExceptionFilter ( IN PEXCEPTION_POINTERS  ExceptionInfo,
ULONG  ExceptionCode 
)

Definition at line 304 of file delayimp.cpp.

305 {
307  ok(ExceptionCode == expected, "Expected code to be 0x%lx, was: 0x%lx\n", expected, ExceptionCode);
308  ok(ExceptionInfo != NULL, "Expected to get exception info\n");
309  ok(ExceptionInfo->ExceptionRecord != NULL, "Expected to get a valid record info\n");
310 
311  if (ExceptionCode != expected)
312  {
313  skip("Skipping other checks, this was not the exception we expected!\n");
315  }
316 
317  if (ExceptionInfo && ExceptionInfo->ExceptionRecord)
318  {
319  PEXCEPTION_RECORD ExceptionRecord = ExceptionInfo->ExceptionRecord;
320  ok(ExceptionRecord->ExceptionCode == expected, "Expected ExceptionCode to be 0x%lx, was 0x%lx\n",
321  expected, ExceptionRecord->ExceptionCode);
322  /* We can still continue. */
323  ok(ExceptionRecord->ExceptionFlags == 0, "Expected ExceptionFlags to be 0, was: 0x%lx\n",
324  ExceptionRecord->ExceptionFlags);
325  ok(ExceptionRecord->NumberParameters == 1, "Expected 1 parameter, got %lu\n",
326  ExceptionRecord->NumberParameters);
327  if (ExceptionRecord->NumberParameters == 1)
328  {
329  PDelayLoadInfo LoadInfo = (PDelayLoadInfo)ExceptionRecord->ExceptionInformation[0];
330  ok(LoadInfo && LoadInfo->cb >= 36, "Expected a valid pointer with a struct that is big enough: %p, %lu\n",
331  LoadInfo, LoadInfo ? LoadInfo->cb : 0);
332 
333  if (g_ExpectedDll)
334  ok(!strcmp(g_ExpectedDll, LoadInfo->szDll), "Expected szDll to be '%s', but was: '%s'\n",
335  g_ExpectedDll, LoadInfo->szDll);
336  if (g_ExpectedName)
337  {
338  ok(LoadInfo->dlp.fImportByName, "Expected import by name\n");
339  if (LoadInfo->dlp.fImportByName)
340  {
341  ok(!strcmp(g_ExpectedName, LoadInfo->dlp.szProcName),
342  "Expected szProcName to be '%s', but was: '%s'\n", g_ExpectedName, LoadInfo->dlp.szProcName);
343 
345  {
346  HMODULE mod = LoadLibraryA("imagehlp.dll");
347  LoadInfo->pfnCur = GetProcAddress(mod, g_ExpectedName);
348  }
349  else
350  {
351  char buf[100];
353  sprintf(buf, "%c%s", first, g_ExpectedName + 1);
355  }
357  }
358  }
359  }
360  }
361 
363 }
FARPROC pfnCur
Definition: delayimp.h:73
_Inout_ PIRP _In_ NTSTATUS ExceptionCode
Definition: cdprocs.h:1782
GLenum GLuint GLenum GLsizei const GLchar * buf
Definition: glext.h:7751
const GLint * first
Definition: glext.h:5794
struct DelayLoadInfo * PDelayLoadInfo
BOOL fImportByName
Definition: delayimp.h:57
bool g_ExceptionIsModule
Definition: delayimp.cpp:41
#define sprintf(buf, format,...)
Definition: sprintf.c:55
DWORD ExceptionCode
Definition: compat.h:196
#define ERROR_SEVERITY_ERROR
Definition: ntbasedef.h:603
LPCSTR szProcName
Definition: delayimp.h:60
HINSTANCE WINAPI DECLSPEC_HOTPATCH LoadLibraryA(LPCSTR lpLibFileName)
Definition: loader.c:111
#define EXCEPTION_EXECUTE_HANDLER
Definition: excpt.h:85
DelayLoadProc dlp
Definition: delayimp.h:71
smooth NULL
Definition: ftsmooth.c:416
int toupper(int c)
Definition: utclib.c:881
ULONG_PTR ExceptionInformation[EXCEPTION_MAXIMUM_PARAMETERS]
Definition: compat.h:201
#define isupper(c)
Definition: acclib.h:71
unsigned long DWORD
Definition: ntddk_ex.h:95
DWORD cb
Definition: delayimp.h:67
LPCSTR szDll
Definition: delayimp.h:70
HMODULE WINAPI DECLSPEC_HOTPATCH GetModuleHandleA(LPCSTR lpModuleName)
Definition: loader.c:821
struct _EXCEPTION_RECORD * ExceptionRecord
Definition: compat.h:198
#define ERROR_PROC_NOT_FOUND
Definition: winerror.h:199
#define ok(value,...)
Definition: atltest.h:57
#define skip(...)
Definition: atltest.h:64
const char * g_ExpectedName
Definition: delayimp.cpp:44
#define ERROR_MOD_NOT_FOUND
Definition: compat.h:94
int strcmp(const char *String1, const char *String2)
Definition: utclib.c:469
DWORD ExceptionFlags
Definition: compat.h:197
#define GetProcAddress(x, y)
Definition: compat.h:410
#define VcppException(sev, err)
Definition: delayimp.cpp:29
DWORD NumberParameters
Definition: compat.h:200
const char * g_ExpectedDll
Definition: delayimp.cpp:43
int tolower(int c)
Definition: utclib.c:902
BOOL expected
Definition: store.c:2063
static int mod
Definition: i386-dis.c:1273
#define EXCEPTION_CONTINUE_EXECUTION
Definition: excpt.h:87

Referenced by START_TEST().

◆ MyFunction()

INT_PTR WINAPI MyFunction ( )

Definition at line 130 of file delayimp.cpp.

131 {
132  return 123;
133 }

Referenced by DliHook().

◆ MySfcIsKeyProtected()

BOOL WINAPI MySfcIsKeyProtected ( HKEY  KeyHandle,
LPCWSTR  SubKeyName,
REGSAM  KeySam 
)

Definition at line 135 of file delayimp.cpp.

136 {
137  return 12345;
138 }

Referenced by DliHook().

◆ Rva2Addr()

template<typename PTR >
PTR Rva2Addr ( PIMAGE_DOS_HEADER  dos,
RVA  rva 
)

Definition at line 377 of file delayimp.cpp.

378 {
379  /* Old delayload type */
380  if (g_UsePointers)
381  return reinterpret_cast<PTR>(rva);
382  return reinterpret_cast<PTR>((reinterpret_cast<PBYTE>(dos) + rva));
383 }
bool g_UsePointers
Definition: delayimp.cpp:374
IMAGE_DOS_HEADER dos
Definition: module.c:49

◆ SetExpectedDli()

static void SetExpectedDli ( unsigned *  order)
static

Definition at line 87 of file delayimp.cpp.

88 {
90  g_DliHookIndex = 0;
91  g_Target[0] = '\0';
92 }
GLuint GLdouble GLdouble GLint GLint order
Definition: glext.h:11194
char g_Target[100]
Definition: delayimp.cpp:45
size_t g_DliHookIndex
Definition: delayimp.cpp:84
unsigned * g_DliHookExpected
Definition: delayimp.cpp:83

Referenced by START_TEST().

◆ START_TEST()

START_TEST ( delayimp  )

Definition at line 399 of file delayimp.cpp.

400 {
401  /* Verify that both scenario's work */
402  ok(__pfnDliNotifyHook2 == DliHook, "Expected __pfnDliNotifyHook2 to be DliHook(%p), but was: %p\n",
404  ok(__pfnDliFailureHook2 == NULL, "Expected __pfnDliFailureHook2 to be NULL, but was: %p\n",
406 
408 
409 
411 
412  ok(dos->e_magic == IMAGE_DOS_SIGNATURE, "Expected a DOS header\n");
414  return;
415 
418 
419  /* Test some advanced features (loading / unloading) */
420  if (delaydir->Size != 0)
421  {
422 #if defined(_DELAY_IMP_VER) && _DELAY_IMP_VER == 2 && defined(DELAYLOAD_SUPPORTS_UNLOADING)
423  /* First, before mangling the delayload stuff, let's try some v2 functions */
425  ok(mod == NULL, "Expected mod to be NULL, was %p\n", mod);
426  /* Now, a mistyped module (case sensitive!) */
427  HRESULT hr = __HrLoadAllImportsForDll("WiNmM.DlL");
428  ok(hr == HRESULT_FROM_WIN32(ERROR_MOD_NOT_FOUND), "Expected hr to be HRESULT_FROM_WIN32(ERROR_MOD_NOT_FOUND), was %lu\n", hr);
430  ok(mod == NULL, "Expected mod to be NULL, was %p\n", mod);
431 
432  /* Let's load it */
433  hr = __HrLoadAllImportsForDll(WINMM_DLLNAME);
434  ok(hr == S_OK, "Expected hr to be S_OK, was %lu\n", hr);
436  ok(mod != NULL, "Expected mod to be valid, was NULL\n");
437 
438  BOOL status = __FUnloadDelayLoadedDLL2(WINMM_DLLNAME);
439  ok(status == TRUE, "Expected __FUnloadDelayLoadedDLL2 to succeed\n");
441  ok(mod == NULL, "Expected mod to be NULL, was %p\n", mod);
442 #else
443  trace("Binary compiled without support for unloading\n");
444 #endif
445  }
446  else
447  {
448  skip("No IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT found, some advanced features might not work!\n");
449  }
450 
451  /* Test the normal flow without a dll loaded */
452  BOOL ret = 123;
455  g_ExpectedName = "sndPlaySoundA";
457  ok(ret == TRUE, "Expected ret to be TRUE, was %u\n", ret);
458  CheckDliDone();
459 
460  /* Test the normal flow with a dll loaded */
463  g_ExpectedName = "sndPlaySoundW";
465  ok(ret == TRUE, "Expected ret to be TRUE, was %u\n", ret);
466  CheckDliDone();
467 
468  /* Make sure GetProcAddress fails, also ignore the Failure hook, use the exception to set the address */
471  g_ExpectedName = "playSoundW";
472  g_BreakFunctionName = true;
473  ret = 123;
474  _SEH2_TRY
475  {
477  }
479  {
480  ;
481  }
482  _SEH2_END;
483  ok(ret == TRUE, "Expected ret to be TRUE, was %u\n", ret);
484  CheckDliDone();
485  ok(g_BreakFunctionName == false, "Expected the functionname to be changed\n");
486 
487  /* Make the LoadLib fail, manually load the library in the Failure Hook,
488  Respond to the dliNotePreGetProcAddress with an alternate function address */
491  ok(ret == 12345, "Expected ret to be 12345, was %u\n", ret); /* The original function returns FALSE! */
492  CheckDliDone();
493 
494  /* Show that it works with the manually returned dll */
497  ok(ret == FALSE, "Expected ret to be FALSE, was %u\n", ret);
498  CheckDliDone();
499 
500  /* Return a fake dll handle, so that we can see when it is being used, and manually return a function in the Failure Hook */
503  ok(ret == FALSE, "Expected ret to be FALSE, was %u\n", ret);
504  CheckDliDone();
505 
506  /* Manually return a function in the failure hook, when the module is the previously set bad one */
509  ok(ret == FALSE, "Expected ret to be FALSE, was %u\n", ret);
510  CheckDliDone();
511 
512  if (HIWORD(SymGetOptions) == NULL)
513  {
514  skip("SymGetOptions until CORE-6504 is fixed\n");
515  }
516  else
517  {
518  /* Completely bypass most hooks, by directly replying with a function address */
520  g_BypassMost = true;
521  DWORD opt = SymGetOptions();
522  g_BypassMost = false;
523  ok(opt == 123, "Expected opt to be 123, was %lu\n", opt); /* The original function returns ERROR_INVALID_HANDLE */
524  CheckDliDone();
525  }
526 
527  /* Import by ordinal */
528  g_ImportByName = false;
530  PARSEDURLA pua = { sizeof(pua), 0 };
531  HRESULT hr = ParseURLA("", &pua);
532  ok(hr == URL_E_INVALID_SYNTAX, "Expected tmp to be URL_E_INVALID_SYNTAX, was %lx\n", hr);
533  CheckDliDone();
534  g_ImportByName = true;
535 
536  /* Handle LoadLib failure with an exception handler */
537  if (HIWORD(MapAndLoad) == NULL)
538  {
539  skip("MapAndLoad until CORE-6504 is fixed\n");
540  }
541  else
542  {
544  LOADED_IMAGE img = {0};
545  ret = 123;
546  g_ExceptionIsModule = true;
547  g_ExpectedDll = "xxxxxhlp.dll";
548  g_ExpectedName = "MapAndLoad";
549  _SEH2_TRY
550  {
551  ret = MapAndLoad("some_not_existing_file.aabbcc", NULL, &img, FALSE, TRUE);
552  }
554  {
555  ;
556  }
557  _SEH2_END;
558  g_ExceptionIsModule = false;
559  ok(ret == FALSE, "Expected ret to be FALSE, was %u\n", ret);
560  CheckDliDone();
561  }
562 }
#define TRUE
Definition: types.h:120
#define HRESULT_FROM_WIN32(x)
Definition: winerror.h:92
IMAGE_DATA_DIRECTORY DataDirectory[IMAGE_NUMBEROF_DIRECTORY_ENTRIES]
Definition: ntddk_ex.h:178
HRESULT hr
Definition: shlfolder.c:183
static void SetExpectedDli(unsigned *order)
Definition: delayimp.cpp:87
unsigned g_imagehlp[]
Definition: delayimp.cpp:395
#define CheckDliDone
Definition: delayimp.cpp:126
HRESULT WINAPI ParseURLA(LPCSTR x, PARSEDURLA *y)
Definition: url.c:161
#define WINMM_DLLNAME
Definition: delayimp.cpp:35
bool g_BypassMost
Definition: delayimp.cpp:40
BOOL WINAPI sndPlaySoundW(LPCWSTR pszSound, UINT uFlags)
Definition: playsound.c:548
unsigned g_winmm_snd_play_a[]
Definition: delayimp.cpp:386
PIMAGE_NT_HEADERS32 PIMAGE_NT_HEADERS
Definition: ntddk_ex.h:187
unsigned g_sfc_key[]
Definition: delayimp.cpp:389
unsigned g_version_w[]
Definition: delayimp.cpp:392
IMAGE_NT_HEADERS nt
Definition: module.c:50
BOOL WINAPI MapAndLoad(PCSTR pszImageName, PCSTR pszDllPath, PLOADED_IMAGE pLoadedImage, BOOL bDotDll, BOOL bReadOnly)
Definition: access.c:139
IMAGE_OPTIONAL_HEADER32 OptionalHeader
Definition: ntddk_ex.h:184
unsigned g_version_a[]
Definition: delayimp.cpp:391
FARPROC WINAPI DliFailHook(unsigned dliNotify, PDelayLoadInfo pdli)
Definition: delayimp.cpp:255
_SEH2_TRY
Definition: create.c:4250
bool g_ExceptionIsModule
Definition: delayimp.cpp:41
unsigned g_sfc_file[]
Definition: delayimp.cpp:390
#define IMAGE_DOS_SIGNATURE
Definition: pedump.c:89
LONG ExceptionFilter(IN PEXCEPTION_POINTERS ExceptionInfo, ULONG ExceptionCode)
Definition: delayimp.cpp:304
unsigned g_shlwapi[]
Definition: delayimp.cpp:394
bool g_ImportByName
Definition: delayimp.cpp:42
unsigned int BOOL
Definition: ntddk_ex.h:94
#define SND_NOSTOP
Definition: mmsystem.h:158
#define _SEH2_GetExceptionInformation()
Definition: pseh2_64.h:11
smooth NULL
Definition: ftsmooth.c:416
struct _IMAGE_DOS_HEADER * PIMAGE_DOS_HEADER
#define URL_E_INVALID_SYNTAX
Definition: intshcut.h:32
unsigned g_winmm_play_w[]
Definition: delayimp.cpp:388
unsigned g_scard[]
Definition: delayimp.cpp:393
#define trace
Definition: atltest.h:70
GLint GLvoid * img
Definition: gl.h:1956
LONG HRESULT
Definition: typedefs.h:77
unsigned long DWORD
Definition: ntddk_ex.h:95
BOOL WINAPI SfcIsKeyProtected(HKEY hKey, LPCWSTR lpSubKey, REGSAM samDesired)
Definition: sfc_os.c:117
BOOL WINAPI sndPlaySoundA(LPCSTR pszSoundA, UINT uFlags)
Definition: playsound.c:539
#define IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT
Definition: ntimage.h:488
#define SND_NOWAIT
Definition: mmsystem.h:159
BOOL WINAPI GetFileVersionInfoA(LPCSTR filename, DWORD handle, DWORD datasize, LPVOID data)
Definition: version.c:858
int ret
IMAGE_DOS_HEADER dos
Definition: module.c:49
HMODULE WINAPI DECLSPEC_HOTPATCH GetModuleHandleA(LPCSTR lpModuleName)
Definition: loader.c:821
unsigned g_winmm_snd_play_w[]
Definition: delayimp.cpp:387
FARPROC WINAPI DliHook(unsigned dliNotify, PDelayLoadInfo pdli)
Definition: delayimp.cpp:142
PfnDliHook __pfnDliNotifyHook2
Definition: delayimp.cpp:369
PfnDliHook __pfnDliFailureHook2
Definition: pdb.c:221
#define S_OK
Definition: intsafe.h:59
_SEH2_END
Definition: create.c:4424
#define GetModuleHandle
Definition: winbase.h:3651
#define ok(value,...)
Definition: atltest.h:57
bool g_BreakFunctionName
Definition: delayimp.cpp:38
BOOL WINAPI SfcGetNextProtectedFile(HANDLE RpcHandle, PPROTECTED_FILE_DATA ProtFileData)
Definition: sfc_os.c:142
DWORD WINAPI SymGetOptions(void)
Definition: dbghelp.c:458
BOOL WINAPI GetFileVersionInfoW(LPCWSTR filename, DWORD handle, DWORD datasize, LPVOID data)
Definition: version.c:850
#define skip(...)
Definition: atltest.h:64
const char * g_ExpectedName
Definition: delayimp.cpp:44
#define ERROR_MOD_NOT_FOUND
Definition: compat.h:94
#define HIWORD(l)
Definition: typedefs.h:246
#define _SEH2_EXCEPT(...)
Definition: pseh2_64.h:6
#define _SEH2_GetExceptionCode()
Definition: pseh2_64.h:12
const char * g_ExpectedDll
Definition: delayimp.cpp:43
BYTE * PBYTE
Definition: pedump.c:66
static int mod
Definition: i386-dis.c:1273
BOOL WINAPI PlaySoundW(LPCWSTR pszSoundW, HMODULE hmod, DWORD fdwSound)
Definition: playsound.c:531
Definition: ps.c:97

◆ target()

Definition at line 47 of file delayimp.cpp.

48 {
49  if (g_Target[0] == '\0' && pdli)
50  {
51  if (pdli->dlp.fImportByName)
52  sprintf(g_Target, "%s!%s", pdli->szDll, pdli->dlp.szProcName);
53  else
54  sprintf(g_Target, "%s!#%lu", pdli->szDll, pdli->dlp.dwOrdinal);
55  }
56  return g_Target;
57 }
char g_Target[100]
Definition: delayimp.cpp:45
BOOL fImportByName
Definition: delayimp.h:57
#define sprintf(buf, format,...)
Definition: sprintf.c:55
LPCSTR szProcName
Definition: delayimp.h:60
DelayLoadProc dlp
Definition: delayimp.h:71
LPCSTR szDll
Definition: delayimp.h:70
DWORD dwOrdinal
Definition: delayimp.h:61

Referenced by CheckDli_imp(), and CheckDliDone_imp().

Variable Documentation

◆ __pfnDliNotifyHook2

PfnDliHook __pfnDliNotifyHook2 = DliHook

Definition at line 369 of file delayimp.cpp.

Referenced by START_TEST().

◆ g_BreakFunctionName

bool g_BreakFunctionName = false

Definition at line 38 of file delayimp.cpp.

Referenced by DliHook(), and START_TEST().

◆ g_BrokenFunctionName

bool g_BrokenFunctionName = false

Definition at line 39 of file delayimp.cpp.

Referenced by DliHook().

◆ g_BypassMost

bool g_BypassMost = false

Definition at line 40 of file delayimp.cpp.

Referenced by DliHook(), and START_TEST().

◆ g_DliHookExpected

unsigned* g_DliHookExpected = NULL

Definition at line 83 of file delayimp.cpp.

Referenced by CheckDli_imp(), CheckDliDone_imp(), and SetExpectedDli().

◆ g_DliHookIndex

size_t g_DliHookIndex = 0

Definition at line 84 of file delayimp.cpp.

Referenced by CheckDli_imp(), CheckDliDone_imp(), and SetExpectedDli().

◆ g_ExceptionIsModule

bool g_ExceptionIsModule = false

Definition at line 41 of file delayimp.cpp.

Referenced by ExceptionFilter(), and START_TEST().

◆ g_ExpectedDll

const char* g_ExpectedDll = NULL

Definition at line 43 of file delayimp.cpp.

Referenced by DliHook(), ExceptionFilter(), and START_TEST().

◆ g_ExpectedName

const char* g_ExpectedName = NULL

Definition at line 44 of file delayimp.cpp.

Referenced by DliHook(), ExceptionFilter(), and START_TEST().

◆ g_imagehlp

Definition at line 395 of file delayimp.cpp.

Referenced by START_TEST().

◆ g_ImportByName

bool g_ImportByName = true

Definition at line 42 of file delayimp.cpp.

Referenced by DliHook(), and START_TEST().

◆ g_scard

unsigned g_scard[] = { dliStartProcessing, dliNoteEndProcessing, LAST_DLI }

Definition at line 393 of file delayimp.cpp.

Referenced by START_TEST().

◆ g_sfc_file

Definition at line 390 of file delayimp.cpp.

Referenced by START_TEST().

◆ g_sfc_key

◆ g_shlwapi

Definition at line 394 of file delayimp.cpp.

Referenced by START_TEST().

◆ g_Target

char g_Target[100] = { 0 }

Definition at line 45 of file delayimp.cpp.

Referenced by CheckDliDone_imp(), SetExpectedDli(), and target().

◆ g_UsePointers

bool g_UsePointers = false

Definition at line 374 of file delayimp.cpp.

Referenced by Rva2Addr().

◆ g_version_a

◆ g_version_w

Definition at line 392 of file delayimp.cpp.

Referenced by START_TEST().

◆ g_VersionDll

HMODULE g_VersionDll
static

Definition at line 141 of file delayimp.cpp.

Referenced by DliFailHook(), and DliHook().

◆ g_winmm_play_w

Definition at line 388 of file delayimp.cpp.

Referenced by START_TEST().

◆ g_winmm_snd_play_a

Definition at line 386 of file delayimp.cpp.

Referenced by START_TEST().

◆ g_winmm_snd_play_w

Definition at line 387 of file delayimp.cpp.

Referenced by START_TEST().