迎清明!PE文件处理类

无名侠

    为了方便加载和操作PE文件，我特此封装了一个PE文件类。目前的功能有 载入/修改入口/修改基址/取区块数/添加区块 /地址转换 等等....下面都是类的公开函数：
每一个函数的旁边都有注释。
如果返回值是指针，这个指针是指向PE文件缓冲区里的，可以直接修改，修改后可以用FlushBuffer即时刷新到硬盘文件。
谢谢大家的支持，祝大家清明快乐，我也要去为班主任"上香",在他的门口放点纸钱小小的恶作剧一下吧。
最后我呼吁大家一起来完善这个类，让我们共同进步。我是学生党精力有限你懂的，可以直接在后面回复代码。

1. bool LoadPe(char *FileNam);
   
2.         char *GetBuffer(); //返回PE文件内存映射地址，可以直接进行修改。
   
3.         WORD GetSectionCount(); /*获取区块数量*/
   
4.         DWORD GetImportCount(); /*导出表描述符*/
   
5.         DWORD GetImageBase();        /*获取镜像基址*/
   
6.         DWORD GetEntry();                /*获取入口点*/
   
7.         PIMAGE_NT_HEADERS GetNtHeaders(); /*获取PIMAGE_NT_HEADERS指针*/
   
8.         PIMAGE_IMPORT_DESCRIPTOR GetFirstImport(); /*取导入表指针*/
   
9.         void SetNewBase(DWORD newBase);
   
10.         void SetNewEntry(DWORD newEntry);
    
11. 
    
12.         PIMAGE_DATA_DIRECTORY GetDataDirInfo(int nDir);
    
13. //读/写操作
    
14.         bool FlushBuffer();/*将修改的内容保存到文件*/
    
15. 
    
16.         bool RestView(); /*重新映射文件 如果PE文件扩容后需要调用*/
    
17. 
    
18.         bool WriteDataByRaw(DWORD raw,char *src,int len);
    
19.                 bool WriteDataByRva(DWORD rva,char *src,int len);  //上层封装函数
    
20.         bool ReadDataByRaw(DWORD raw,char *buffer,int len);
    
21.                 bool ReadDataByRva(DWORD rva,char *buffer,int len);
    
22. //地址转换类函数
    
23.         DWORD rva_to_va(DWORD rva);
    
24.         DWORD va_to_rva(DWORD va);
    
25.         DWORD rva_to_raw(DWORD rva);
    
26.         DWORD raw_to_rva(DWORD raw);
    
27. //区块
    
28.         PIMAGE_SECTION_HEADER GetFirstSection();  /*获取起始区块*/
    
29.         PIMAGE_SECTION_HEADER FindSectionByName(char *sectionName); /*通过区块名寻找区块*/
    
30.         PIMAGE_SECTION_HEADER AddSection(char *SectionName,DWORD SectionSize); /*添加区块:成功返回区块描述符指针 失败返回NULL*/
    
31. //对齐函数
    
32.          DWORD static PEAlign(DWORD num,DWORD dwAlignTo);

复制代码

源.zip (2.9 KB, 下载次数: 10)

2015-4-4 18:14 上传

点击文件名下载附件

@牡丹花下死做鬼 @小甲鱼

@福禄娃娃

.......

无名侠

想写添加导入表描述符的，但是有的PE文件数据目录指向的导入表那一块大小刚刚好，所有要新建一个节表来移植所有的导入表描述符，实现起来好像很麻烦的样子。有什么好的办法吗？@小甲鱼
或许实现一个类似内存分配的函数到是不错想法。:titter:大召唤术：
@メ㊣逆ご帅☆  @小名明SIU  @沉ジ寞  @拈花小仙  @怡静  @正在写代码  @swsm

freeparty

路过，过来支持一下。

无名侠

freeparty 发表于 2015-4-4 20:43
路过，过来支持一下。

谢谢支持了。。。。。

freeparty

无名侠 发表于 2015-4-4 21:08
谢谢支持了。。。。。

必须的。

沉ジ寞

。。。技术太渣 代码也看不懂 冏
@小甲鱼  鱼哥 来露一手

wawzz11

感谢楼主的分享啊！！！！