delayimp.c File Reference

#include <stdarg.h>
#include <windef.h>
#include <winbase.h>
#include <delayimp.h>

Include dependency graph for delayimp.c:

Go to the source code of this file.

Functions
FORCEINLINE unsigned	IndexFromPImgThunkData (PCImgThunkData pData, PCImgThunkData pBase)
FORCEINLINE PVOID	PFromRva (RVA rva)
FARPROC WINAPI	__delayLoadHelper2 (PCImgDelayDescr pidd, PImgThunkData pIATEntry)

Variables
PfnDliHook	__pfnDliNotifyHook2
PfnDliHook	__pfnDliFailureHook2
PfnDliHook	__pfnDliNotifyHook2Default = NULL
PfnDliHook	__pfnDliFailureHook2Default = NULL
const IMAGE_DOS_HEADER	__ImageBase

Function Documentation

__delayLoadHelper2()

FARPROC WINAPI __delayLoadHelper2	(	PCImgDelayDescr	pidd,
		PImgThunkData	pIATEntry
	)

Definition at line 60 of file delayimp.c.

{
    DelayLoadInfo dli = {0};
    int index;
    PImgThunkData pIAT;
    PImgThunkData pINT;
    HMODULE *phMod;
 
    pIAT = PFromRva(pidd->rvaIAT);
    pINT = PFromRva(pidd->rvaINT);
    phMod = PFromRva(pidd->rvaHmod);
    index = IndexFromPImgThunkData(pIATEntry, pIAT);
 
    dli.cb = sizeof(dli);
    dli.pidd = pidd;
    dli.ppfn = (FARPROC*)&pIAT[index].u1.Function;
    dli.szDll = PFromRva(pidd->rvaDLLName);
    dli.dlp.fImportByName = !IMAGE_SNAP_BY_ORDINAL(pINT[index].u1.Ordinal);
    if (dli.dlp.fImportByName)
    {
        /* u1.AddressOfData points to a IMAGE_IMPORT_BY_NAME struct */
        PIMAGE_IMPORT_BY_NAME piibn = PFromRva((RVA)pINT[index].u1.AddressOfData);
        dli.dlp.szProcName = (LPCSTR)&piibn->Name;
    }
    else
    {
        dli.dlp.dwOrdinal = IMAGE_ORDINAL(pINT[index].u1.Ordinal);
    }
 
    if (__pfnDliNotifyHook2)
    {
        dli.pfnCur = __pfnDliNotifyHook2(dliStartProcessing, &dli);
        if (dli.pfnCur)
        {
            pIAT[index].u1.Function = (DWORD_PTR)dli.pfnCur;
            if (__pfnDliNotifyHook2)
                __pfnDliNotifyHook2(dliNoteEndProcessing, &dli);
 
            return dli.pfnCur;
        }
    }
 
    dli.hmodCur = *phMod;
 
    if (dli.hmodCur == NULL)
    {
        if (__pfnDliNotifyHook2)
            dli.hmodCur = (HMODULE)__pfnDliNotifyHook2(dliNotePreLoadLibrary, &dli);
        if (dli.hmodCur == NULL)
        {
            dli.hmodCur = LoadLibraryA(dli.szDll);
            if (dli.hmodCur == NULL)
            {
                dli.dwLastError = GetLastError();
                if (__pfnDliFailureHook2)
                    dli.hmodCur = (HMODULE)__pfnDliFailureHook2(dliFailLoadLib, &dli);
 
                if (dli.hmodCur == NULL)
                {
                    ULONG_PTR args[] = { (ULONG_PTR)&dli };
                    RaiseException(VcppException(ERROR_SEVERITY_ERROR, ERROR_MOD_NOT_FOUND), 0, 1, args);
 
                    /* If we survive the exception, we are expected to use pfnCur directly.. */
                    return dli.pfnCur;
                }
            }
        }
        *phMod = dli.hmodCur;
    }
 
    dli.dwLastError = ERROR_SUCCESS;
 
    if (__pfnDliNotifyHook2)
        dli.pfnCur = (FARPROC)__pfnDliNotifyHook2(dliNotePreGetProcAddress, &dli);
    if (dli.pfnCur == NULL)
    {
        /* dli.dlp.szProcName might also contain the ordinal */
        dli.pfnCur = GetProcAddress(dli.hmodCur, dli.dlp.szProcName);
        if (dli.pfnCur == NULL)
        {
            dli.dwLastError = GetLastError();
            if (__pfnDliFailureHook2)
               dli.pfnCur = __pfnDliFailureHook2(dliFailGetProc, &dli);
 
            if (dli.pfnCur == NULL)
            {
                ULONG_PTR args[] = { (ULONG_PTR)&dli };
                RaiseException(VcppException(ERROR_SEVERITY_ERROR, ERROR_PROC_NOT_FOUND), 0, 1, args);
            }
 
            //return NULL;
        }
    }
 
    pIAT[index].u1.Function = (DWORD_PTR)dli.pfnCur;
    dli.dwLastError = ERROR_SUCCESS;
 
    if (__pfnDliNotifyHook2)
        __pfnDliNotifyHook2(dliNoteEndProcessing, &dli);
 
    return dli.pfnCur;
}

IndexFromPImgThunkData()

FORCEINLINE unsigned IndexFromPImgThunkData	(	PCImgThunkData	pData,
		PCImgThunkData	pBase
	)

Definition at line 42 of file delayimp.c.

{
    return pData - pBase;
}

Referenced by __delayLoadHelper2().

PFromRva()

FORCEINLINE PVOID PFromRva

(

RVA

rva

)

Definition at line 51 of file delayimp.c.

{
    return (PVOID)(((ULONG_PTR)(rva)) + ((ULONG_PTR)&__ImageBase));
}

Referenced by __delayLoadHelper2().

Variable Documentation

__ImageBase

const IMAGE_DOS_HEADER __ImageBase

extern

Referenced by PFromRva().

__pfnDliFailureHook2

PfnDliHook __pfnDliFailureHook2

Definition at line 20 of file delayimp.c.

Referenced by __delayLoadHelper2(), and START_TEST().

__pfnDliFailureHook2Default

PfnDliHook __pfnDliFailureHook2Default = NULL

Definition at line 25 of file delayimp.c.

__pfnDliNotifyHook2

PfnDliHook __pfnDliNotifyHook2

Definition at line 19 of file delayimp.c.

Referenced by __delayLoadHelper2(), and START_TEST().

__pfnDliNotifyHook2Default

PfnDliHook __pfnDliNotifyHook2Default = NULL

Definition at line 24 of file delayimp.c.