内存方式加载DLL

【源码目录】

MemLoadDll

├── CalcModule
│   ├── CalcModule.cpp
│   ├── CalcModule.dsp
│   ├── CalcModule.dsw
│   ├── CalcModule.h
│   ├── CalcModule.opt
│   ├── CalcModule.plg
│   ├── CalcModule.sln
│   ├── CalcModule.suo
│   ├── CalcModule.vcproj
│   ├── ReadMe.txt
│   ├── StdAfx.cpp
│   └── StdAfx.h
├── Demo
│   ├── Backup
│   │   ├── memdll.sln
│   │   └── memdll.v12.suo
│   ├── CalcModule32.dll
│   ├── CalcModule64.dll
│   ├── MemLoadDll.cpp
│   ├── MemLoadDll.h
│   ├── ReadMe.txt
│   ├── StdAfx.cpp
│   ├── StdAfx.h
│   ├── UpgradeLog.htm
│   ├── ipch
│   │   └── memdll-34b9df56
│   │       └── debug
│   │           ├── memdll-3c3810e4.ipch
│   │           └── memdll-8ccb9c22.ipch
│   ├── memdll.aps
│   ├── memdll.clw
│   ├── memdll.cpp
│   ├── memdll.dsp
│   ├── memdll.dsw
│   ├── memdll.h
│   ├── memdll.opt
│   ├── memdll.plg
│   ├── memdll.rc
│   ├── memdll.sln
│   ├── memdll.suo
│   ├── memdll.v12.suo
│   ├── memdll.vcproj
│   ├── memdll.vcxproj
│   ├── memdll.vcxproj.filters
│   ├── memdllDlg.cpp
│   ├── memdllDlg.h
│   ├── res
│   │   ├── memdll.ico
│   │   └── memdll.rc2
│   └── resource.h
├── MemLoadDll.cpp
└── MemLoadDll.h

7 directories, 46 files

【核心代码】

BOOL CMemLoadDll::MemLoadLibrary(void* lpFileData, int nDataLength)
{
    if (m_pImageBase != NULL)
    {
        return FALSE;  //已经加载一个dll，还没有释放，不能加载新的dll
    }
    //检查数据有效性，并初始化
    if (!CheckDataValide(lpFileData, nDataLength))
    {
        return FALSE;
    }
    //计算所需的加载空间
    int nImageSize = CalcTotalImageSize();

    if (nImageSize == 0)
    {
        return FALSE;
    }
    // 分配虚拟内存
    void *pMemoryAddress = VirtualAlloc(NULL, nImageSize, MEM_COMMIT|MEM_RESERVE, PAGE_EXECUTE_READWRITE);

    if (pMemoryAddress == NULL)
    {
        return FALSE;
    }
    else
    {
        CopyDllDatas(pMemoryAddress, lpFileData); //复制dll数据，并对齐每个段
        //重定位信息
        if (m_pNTHeader->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress > 0
            && m_pNTHeader->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].Size > 0)
        {
            DoRelocation(pMemoryAddress);
        }
        //填充引入地址表
        if (!FillRavAddress(pMemoryAddress)) //修正引入地址表失败
        {
            VirtualFree(pMemoryAddress, 0, MEM_RELEASE);
            return FALSE;
        }
        //修改页属性。应该根据每个页的属性单独设置其对应内存页的属性。这里简化一下。
        //统一设置成一个属性PAGE_EXECUTE_READWRITE
        unsigned long unOld;

        VirtualProtect(pMemoryAddress, nImageSize, PAGE_EXECUTE_READWRITE, &unOld);
    }
    //修正基地址
#ifdef WIN32
    m_pNTHeader->OptionalHeader.ImageBase = (DWORD)pMemoryAddress;
#else
    m_pNTHeader->OptionalHeader.ImageBase = (ULONGULONG)pMemoryAddress;
#endif
    //接下来要调用一下dll的入口函数，做初始化工作。
    m_pDllMain = (ProcDllMain)(m_pNTHeader->OptionalHeader.AddressOfEntryPoint (PBYTE)pMemoryAddress);

    BOOL InitResult = m_pDllMain((HINSTANCE)pMemoryAddress, DLL_PROCESS_ATTACH, 0);

    if (!InitResult) //初始化失败
    {
        m_pDllMain((HINSTANCE)pMemoryAddress, DLL_PROCESS_DETACH, 0);
        VirtualFree(pMemoryAddress, 0, MEM_RELEASE);
        m_pDllMain = NULL;
        return FALSE;
    }
    
    m_bIsLoadOk = TRUE;
    m_pImageBase = pMemoryAddress;
    return TRUE;
}