delayimp.c

Go to the documentation of this file.

/*
 * PROJECT:         ReactOS SDK Library
 * LICENSE:         LGPL, see LGPL.txt in top level directory.
 * FILE:            lib/sdk/delayimp/delayimp.c
 * PURPOSE:         Library for delay importing from dlls
 * PROGRAMMERS:     Timo Kreuzer <timo.kreuzer@reactos.org>
 *
 */

#include <windows.h>
#include <delayimp.h>

/**** load helper ****/

FARPROC WINAPI
__delayLoadHelper2(PCImgDelayDescr pidd, PImgThunkData pIATEntry)
{
    DelayLoadInfo dli;
    int index;
    PImgThunkData pIAT;
    PImgThunkData pINT;
    HMODULE *phMod;
    FARPROC pProc;

    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*)pIATEntry->u1.Function;
    dli.szDll = PFromRva(pidd->rvaDLLName);
    dli.dlp.fImportByName = !(pINT[index].u1.Ordinal & IMAGE_ORDINAL_FLAG);
    if (dli.dlp.fImportByName)
    {
        /* u1.AdressOfData 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 = pINT[index].u1.Ordinal & ~IMAGE_ORDINAL_FLAG;
    }
    dli.hmodCur = *phMod;
    dli.pfnCur = (FARPROC)pIAT[index].u1.Function;
    dli.dwLastError = GetLastError();
    pProc = __pfnDliNotifyHook2(dliStartProcessing, &dli);
    if (pProc)
    {
        pIAT[index].u1.Function = (DWORD_PTR)pProc;
        return pProc;
    }

    if (dli.hmodCur == NULL)
    {
        dli.hmodCur = LoadLibraryA(dli.szDll);
        if (!dli.hmodCur)
        {
            dli.dwLastError = GetLastError();
            __pfnDliFailureHook2(dliFailLoadLib, &dli);
//          if (ret)
//          {
//          }
            // FIXME: raise exception;
            return NULL;
        }
        *phMod = dli.hmodCur;
    }

    /* dli.dlp.szProcName might also contain the ordinal */
    pProc = GetProcAddress(dli.hmodCur, dli.dlp.szProcName);
    if (!pProc)
    {
        dli.dwLastError = GetLastError();
        __pfnDliFailureHook2(dliFailGetProc, &dli);
        // FIXME: handle return value & raise exception
        return NULL;
    }
    pIAT[index].u1.Function = (DWORD_PTR)pProc;

    return pProc;
}

/*** The default hooks ***/

FARPROC WINAPI
DefaultDliNotifyHook2(unsigned dliNotify, PDelayLoadInfo pdli)
{
    return NULL;
}

FARPROC WINAPI
DefaultDliFailureHook2(unsigned dliNotify, PDelayLoadInfo pdli)
{
    return NULL;
}

PfnDliHook __pfnDliNotifyHook2 = DefaultDliNotifyHook2;
PfnDliHook __pfnDliFailureHook2 = DefaultDliFailureHook2;