首页
社区
课程
招聘
[原创]PE加载器的简单实现
发表于: 2019-1-25 20:44 22314

[原创]PE加载器的简单实现

2019-1-25 20:44
22314

模拟PE加载器加载PE文件,对导入表以及重定位表的操作过程。

 

PE加载器的步骤:

 

1.将PE文件用ReadFile读取数据

char szFileName[] = "1.exe";

//打开文件,设置属性可读可写
HANDLE hFile = CreateFileA(szFileName,
    GENERIC_READ | GENERIC_WRITE,
    FILE_SHARE_READ | FILE_SHARE_WRITE,
    NULL,
    OPEN_EXISTING,
    FILE_ATTRIBUTE_ARCHIVE,
    NULL);

if (INVALID_HANDLE_VALUE == hFile)
{
    printf("文件打开失败\n");
    return 1;
}

//获取文件大小
DWORD dwFileSize = GetFileSize(hFile, NULL);

//申请空间将exe读取到内存中
char *pData = new char[dwFileSize];
if (NULL == pData)
{
    printf("空间申请失败\n");
    return 2;
}

DWORD dwRet = 0;
ReadFile(hFile, pData, dwFileSize, &dwRet, NULL);
CloseHandle(hFile);

2.根据PE结构获取镜像大小,在自己的程序中申请可读可写可执行的内存。

char* chBaseAddress = NULL;

//获取镜像大小
DWORD dwSizeOfImage = GetSizeOfImage(pFileBuff);

//在进程中开辟一块内存空间
chBaseAddress = (char*)VirtualAlloc(NULL,
    dwSizeOfImage,
    MEM_COMMIT | MEM_RESERVE,
    PAGE_EXECUTE_READWRITE);
if (NULL == chBaseAddress)
{
    printf("申请进程空间失败\n");
    return NULL;
}

3.将申请的空间全部填为0

RtlZeroMemory(chBaseAddress, dwSizeOfImage);

4.将用ReadFile读取的数据映射到内存中

 


由上图可知,PE文件在加载到内存中对齐粒度不同,因此在写代码时要注意这个问题

PIMAGE_DOS_HEADER pDos = (PIMAGE_DOS_HEADER)pFileBuff;
PIMAGE_NT_HEADERS pNt = (PIMAGE_NT_HEADERS)(pFileBuff + pDos->e_lfanew);
PIMAGE_SECTION_HEADER pSection = IMAGE_FIRST_SECTION(pNt);
//所有头 + 结表头的大小
DWORD dwSizeOfHeaders = pNt->OptionalHeader.SizeOfHeaders;
//获取区段数量
int nNumerOfSections = pNt->FileHeader.NumberOfSections;

// 将前一部分都拷贝过去
RtlCopyMemory(chBaseAddress, pFileBuff, dwSizeOfHeaders);

char* chSrcMem = NULL;
char* chDestMem = NULL;
DWORD dwSizeOfRawData = 0;
for (int i = 0; i < nNumerOfSections; i++)
{
    if ((0 == pSection->VirtualAddress) ||
        (0 == pSection->SizeOfRawData))
    {
        pSection++;
        continue;
    }

    chSrcMem = (char*)((DWORD)pFileBuff + pSection->PointerToRawData);
    chDestMem = (char*)((DWORD)chBaseAddress + pSection->VirtualAddress);
    dwSizeOfRawData = pSection->SizeOfRawData;
    RtlCopyMemory(chDestMem, chSrcMem, dwSizeOfRawData);

    pSection++;
}

return TRUE;

5.修复重定位

 

重定位后的地址 = 需要重定位的地址 - 默认加载基址 + 当前加载基址

PIMAGE_DOS_HEADER pDos = (PIMAGE_DOS_HEADER)chBaseAddress;
PIMAGE_NT_HEADERS pNt = (PIMAGE_NT_HEADERS)(chBaseAddress + pDos->e_lfanew);
PIMAGE_BASE_RELOCATION pLoc = (PIMAGE_BASE_RELOCATION)(chBaseAddress + pNt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_BASERELOC].VirtualAddress);

//判断是否有重定位表
if ((char*)pLoc == (char*)pDos)
{
    return TRUE;
}

while ((pLoc->VirtualAddress + pLoc->SizeOfBlock) != 0) //开始扫描重定位表
{
    WORD *pLocData = (WORD *)((PBYTE)pLoc + sizeof(IMAGE_BASE_RELOCATION));
    //计算需要修正的重定位项(地址)的数目
    int nNumberOfReloc = (pLoc->SizeOfBlock - sizeof(IMAGE_BASE_RELOCATION)) / sizeof(WORD);

    for (int i = 0; i < nNumberOfReloc; i++)
    {
        // 每个WORD由两部分组成。高4位指出了重定位的类型,WINNT.H中的一系列IMAGE_REL_BASED_xxx定义了重定位类型的取值。
        // 低12位是相对于VirtualAddress域的偏移,指出了必须进行重定位的位置。

        if ((DWORD)(pLocData[i] & 0x0000F000) == 0x00003000) //这是一个需要修正的地址
        {
            DWORD* pAddress = (DWORD *)((PBYTE)pDos + pLoc->VirtualAddress +                                                                    (pLocData[i] & 0x0FFF));    
            DWORD dwDelta = (DWORD)pDos - pNt->OptionalHeader.ImageBase;
            *pAddress += dwDelta;
        }
    }

    //转移到下一个节进行处理
    pLoc = (PIMAGE_BASE_RELOCATION)((PBYTE)pLoc + pLoc->SizeOfBlock);
}

return TRUE;

6.根据PE结构的导入表,加载所需的dll,并获取导入函数的地址并写入导入表中

PIMAGE_DOS_HEADER pDos = (PIMAGE_DOS_HEADER)chBaseAddress;
PIMAGE_NT_HEADERS pNt = (PIMAGE_NT_HEADERS)(chBaseAddress + pDos->e_lfanew);
PIMAGE_IMPORT_DESCRIPTOR pImportTable = (PIMAGE_IMPORT_DESCRIPTOR)((DWORD)pDos +
    pNt->OptionalHeader.DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress);

// 循环遍历DLL导入表中的DLL及获取导入表中的函数地址
char *lpDllName = NULL;
HMODULE hDll = NULL;
PIMAGE_THUNK_DATA lpImportNameArray = NULL;
PIMAGE_IMPORT_BY_NAME lpImportByName = NULL;
PIMAGE_THUNK_DATA lpImportFuncAddrArray = NULL;
FARPROC lpFuncAddress = NULL;
DWORD i = 0;

while (TRUE)
{
    if (0 == pImportTable->OriginalFirstThunk)
    {
        break;
    }

    // 获取导入表中DLL的名称并加载DLL
    lpDllName = (char *)((DWORD)pDos + pImportTable->Name);
    hDll = GetModuleHandleA(lpDllName);
    if (NULL == hDll)
    {
        hDll = LoadLibraryA(lpDllName);
        if (NULL == hDll)
        {
            pImportTable++;
            continue;
        }
    }

    i = 0;
    // 获取OriginalFirstThunk以及对应的导入函数名称表首地址
    lpImportNameArray = (PIMAGE_THUNK_DATA)((DWORD)pDos + pImportTable->OriginalFirstThunk);
    // 获取FirstThunk以及对应的导入函数地址表首地址
    lpImportFuncAddrArray = (PIMAGE_THUNK_DATA)((DWORD)pDos + pImportTable->FirstThunk);
    while (TRUE)
    {
        if (0 == lpImportNameArray[i].u1.AddressOfData)
        {
            break;
        }

        // 获取IMAGE_IMPORT_BY_NAME结构
        lpImportByName = (PIMAGE_IMPORT_BY_NAME)((DWORD)pDos + lpImportNameArray[i].u1.AddressOfData);

        // 判断导出函数是序号导出还是函数名称导出
        if (0x80000000 & lpImportNameArray[i].u1.Ordinal)
        {
            // 序号导出
            lpFuncAddress = GetProcAddress(hDll, (LPCSTR)                           (lpImportNameArray[i].u1.Ordinal & 0x0000FFFF));
        }
        else
        {
            // 名称导出
            lpFuncAddress = GetProcAddress(hDll, (LPCSTR)lpImportByName->Name);
        }
        lpImportFuncAddrArray[i].u1.Function = (DWORD)lpFuncAddress;
        i++;
    }

    pImportTable++;
}

7.修改PE文件的加载基址

PIMAGE_DOS_HEADER pDos = (PIMAGE_DOS_HEADER)chBaseAddress;
PIMAGE_NT_HEADERS pNt = (PIMAGE_NT_HEADERS)(chBaseAddress + pDos->e_lfanew);
pNt->OptionalHeader.ImageBase = (ULONG32)chBaseAddress;

8.跳转到PE的入口点处执行

PIMAGE_DOS_HEADER pDos = (PIMAGE_DOS_HEADER)chBaseAddress;
PIMAGE_NT_HEADERS pNt = (PIMAGE_NT_HEADERS)(chBaseAddress + pDos->e_lfanew);
char* ExeEntry = (char*)(chBaseAddress + pNt->OptionalHeader.AddressOfEntryPoint);
// 跳转到入口点处执行
__asm
{
    mov eax, ExeEntry
    jmp eax
}

通过上述代码,成功加载并运行程序:

 


[课程]Linux pwn 探索篇!

最后于 2019-2-27 19:25 被Iam0x17编辑 ,原因:
上传的附件:
收藏
免费 11
支持
分享
最新回复 (43)
雪    币: 8635
活跃值: (4825)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
jgs
2
普及文章,收藏备用,谢谢楼主
2019-1-25 22:53
1
雪    币: 781
活跃值: (1116)
能力值: ( LV5,RANK:78 )
在线值:
发帖
回帖
粉丝
3
科普文章 感谢 
2019-1-26 20:13
0
雪    币: 45
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
4
谢谢!很实用
2019-1-26 22:47
0
雪    币: 6
活跃值: (3175)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
5
64位PE呢??
2019-1-27 00:18
0
雪    币: 204
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
6
建议打包个源码文件
2019-1-27 08:37
1
雪    币: 13650
活跃值: (19)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
7
咖啡_741298 64位PE呢??
64位PE叫做PE32+,和PE32除了头部分之外基本相同。
2019-1-27 09:06
0
雪    币: 3170
活跃值: (129)
能力值: ( LV3,RANK:30 )
在线值:
发帖
回帖
粉丝
8
顶一下
2019-1-27 09:25
0
雪    币: 6924
活跃值: (3662)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
9
mark顶一下
2019-1-29 20:01
0
雪    币: 1604
活跃值: (640)
能力值: ( LV13,RANK:460 )
在线值:
发帖
回帖
粉丝
10
支持


最后于 2019-1-29 21:02 被shayi编辑 ,原因:
2019-1-29 21:00
0
雪    币: 0
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
11
学习了
2019-1-29 21:11
0
雪    币: 4893
活跃值: (2493)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
12
目测是科锐的学生~ 
2019-1-29 22:16
0
雪    币: 1103
活跃值: (461)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
13
有个64位PE 的加载器就完美了!!!
2019-1-30 00:12
0
雪    币: 728
活跃值: (1369)
能力值: ( LV9,RANK:150 )
在线值:
发帖
回帖
粉丝
14
Iceshara 目测是科锐的学生~
15pb
2019-1-30 09:24
0
雪    币: 4893
活跃值: (2493)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
15
看雪已经成为了科锐和51PB的作业论坛了.....   文章质量下滑严重 
2019-1-30 14:50
2
雪    币: 6400
活跃值: (4160)
能力值: ( LV10,RANK:163 )
在线值:
发帖
回帖
粉丝
16
没有异常处理的PE LOAD都是 渣渣。
2019-1-30 15:51
0
雪    币: 54
能力值: ( LV1,RANK:0 )
在线值:
发帖
回帖
粉丝
17
马克
2019-1-30 16:24
0
雪    币: 728
活跃值: (1369)
能力值: ( LV9,RANK:150 )
在线值:
发帖
回帖
粉丝
18
 
最后于 2019-1-30 17:50 被Iam0x17编辑 ,原因:
2019-1-30 17:48
0
雪    币: 728
活跃值: (1369)
能力值: ( LV9,RANK:150 )
在线值:
发帖
回帖
粉丝
19
 
最后于 2019-1-30 17:50 被Iam0x17编辑 ,原因:
2019-1-30 17:50
0
雪    币: 375
活跃值: (356)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
20
支持,学习了,还少一个GetSizeOfImage(pFileBuff)函数,我怎么总是不成功?有源码吗,发一份给,谢谢!
2019-2-19 15:18
0
雪    币: 433
活跃值: (891)
能力值: ( LV5,RANK:70 )
在线值:
发帖
回帖
粉丝
21
感谢分享
2019-2-25 13:23
0
雪    币: 1258
活跃值: (1434)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
22
Iceshara 看雪已经成为了科锐和51PB的作业论坛了..... 文章质量下滑严重 [em_38]
2019-2-25 17:12
0
雪    币: 1598
活跃值: (3165)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
23
楼主有源码么,可以发一份不,最近在学这部分内容,照着你的思路自己写的,总是失败,排查bug也查了十几个小时也查不到在哪,和你发的源码一点点比对也找不出来,吐血了.
2019-2-26 10:56
0
雪    币: 1598
活跃值: (3165)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
24
你这人怎么关键性的东西留一手?如果这个程序没有重定位表,这种方法根本就没法实现PE加载器,所有全局变量全是错的
2019-2-27 17:33
0
雪    币: 1598
活跃值: (3165)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
25
而大多数.exe都没有重定位表,怎么去实现这个加载器?
2019-2-27 17:45
0
游客
登录 | 注册 方可回帖
返回
//