[已解决]C语言PE文件filebuffer到imagebuffer的错误

Met32

代码如下，， FileToImage函数 当执行到for循环遍历copy节表的内容.....就报错了  求各位师傅帮忙看一下问题所在~~
顺便问一下 有没有PE方面的书籍 (PE权威指南看不懂~~ 汇编太水了)

1. // 代码节空白添加.cpp : Defines the entry point for the console application.
   
2. //
   
3. 
   
4. #include "stdafx.h"
   
5. #include <windows.h>
   
6. #include <stdlib.h>
   
7. int FileBuffers(PVOID* FileBuffer){//传入的FileBuffer
   
8.         LPVOID PtempBuffer;
   
9.         FILE* f = fopen("C:\\Users\\Administrator\\Desktop\\PETool 1.0.0.5.exe","rb");
   
10.         if(!f)
    
11.         {
    
12.                 printf("文件打开失败\n");
    
13.                 return 0;
    
14.         }
    
15.         fseek(f,0,SEEK_END);
    
16.         int file_size = ftell(f);
    
17.         fseek(f,0,SEEK_SET);
    
18.         PtempBuffer = malloc(file_size);
    
19.         if(!PtempBuffer){
    
20.                 printf("malloc分配失败");
    
21.                 return 0;
    
22.         }
    
23.         size_t n = fread(PtempBuffer,file_size,1,f);
    
24.         if(!n){
    
25.                 printf("内存分配失败");
    
26.                 return 0;
    
27.         }
    
28.         *FileBuffer = PtempBuffer;
    
29.         PtempBuffer = NULL;
    
30.         fclose(f);
    
31.         return file_size;
    
32. }
    
33. int FileToImage(PVOID Filebuffer,PVOID* Imagebuffer){
    
34.         //PE头信息
    
35.         PIMAGE_DOS_HEADER pDosHeader = NULL;
    
36.         PIMAGE_NT_HEADERS pNTHeader = NULL;
    
37.         PIMAGE_FILE_HEADER pPEHeader =NULL;
    
38.         PIMAGE_OPTIONAL_HEADER PoptionHeader = NULL;
    
39.         PIMAGE_SECTION_HEADER pSectionHeader = NULL;
    
40.        
    
41.         PVOID pTempImagebuffer = NULL; //临时的imagebuffer
    
42. 
    
43.         if(!Filebuffer){
    
44.                 printf("读到内存的Filebuffer失效\n");
    
45.                 return 0;
    
46.         }
    
47.         if(*((PWORD)Filebuffer) != IMAGE_DOS_SIGNATURE){
    
48.                 printf("不含MZ标志,不是EXE文件\n");
    
49.                 return 0;
    
50.         }
    
51.         pDosHeader = (PIMAGE_DOS_HEADER)Filebuffer; //强制结构体类型转换pDosHeader
    
52.         pNTHeader = (PIMAGE_NT_HEADERS)((DWORD)Filebuffer+pDosHeader->e_lfanew);//NT头开始的地方
    
53.         if(*((PWORD)((DWORD)Filebuffer+pDosHeader->e_lfanew)) != IMAGE_NT_SIGNATURE){//判断PE标识
    
54.                 printf("不是有效的PE标识\n");
    
55.         }
    
56.         pPEHeader = (PIMAGE_FILE_HEADER)((DWORD)pNTHeader+4);//PE标准头开始
    
57.         PoptionHeader = (PIMAGE_OPTIONAL_HEADER)((DWORD)pPEHeader+0X18);//PE可选头开始  IMAGE_SIZEOF_FILE_HEADER
    
58.         pSectionHeader = (PIMAGE_SECTION_HEADER)((DWORD)PoptionHeader+pPEHeader->SizeOfOptionalHeader);//PE节表开始
    
59.         //分配动态内存
    
60.         pTempImagebuffer = malloc(PoptionHeader->SizeOfImage);
    
61.         if(!pTempImagebuffer){
    
62.                 printf("pTempImagebuffer为NULL");
    
63.                 return 0;
    
64.         }
    
65.         //为临时的imagebuffer初始化
    
66.         memset(pTempImagebuffer,0,PoptionHeader->SizeOfImage);
    
67.         //copy头部
    
68.         memcpy(pTempImagebuffer,pDosHeader,PoptionHeader->SizeOfHeaders);
    
69.         //创建一个临时的PE节表
    
70.         PIMAGE_SECTION_HEADER pTempSectionHeader = pSectionHeader;
    
71. 
    
72.         for(DWORD i=0;i<pPEHeader->NumberOfSections;i++,pTempSectionHeader++){
    
73.                 //copy节从Filebuffer到临时的imagebuffer   
    
74.                 //断点到这里有问题  各位师傅求解决
    
75.                         memcpy((void*)((DWORD)pTempImagebuffer+pTempSectionHeader->VirtualAddress)
    
76.                                 ,(void*)((DWORD)Filebuffer+pTempSectionHeader->PointerToRawData)
    
77.                                 ,pTempSectionHeader->SizeOfRawData);
    
78. 
    
79.                 }
    
80.         *Imagebuffer = pTempImagebuffer;
    
81.         pTempImagebuffer = NULL;
    
82.         return PoptionHeader->SizeOfImage;
    
83. }
    
84. 
    
85. int main(int argc, char* argv[])
    
86. {
    
87.         PVOID FileBuffer;
    
88.         PVOID ImageBuffer;
    
89.         int  x = FileBuffers(&FileBuffer);
    
90.         printf("FileBuffer分配大小为：%d个字节\n",x);
    
91.         int y = FileToImage(FileBuffer,&ImageBuffer);
    
92.         printf("大小为%d\n",y);
    
93.         return 0;
    
94. }
    
95. 
    

复制代码

最佳答案

月排行榜 / 总排行榜

xiao14116

2020-11-4 19:35:20

刚才调试了下，直接原因是因为你申请的内存空间不够，所以出错了

memcpy(pTempImagebuffer,pDosHeader,PoptionHeader->SizeOfHeaders);
PoptionHeader->SizeOfHeaders的值远远大于你申请的内存空间

跳转到最佳答案楼层

xiao14116

刚才调试了下，直接原因是因为你申请的内存空间不够，所以出错了

memcpy(pTempImagebuffer,pDosHeader,PoptionHeader->SizeOfHeaders);
PoptionHeader->SizeOfHeaders的值远远大于你申请的内存空间

Met32

xiao14116 发表于 2020-11-4 19:35
刚才调试了下，直接原因是因为你申请的内存空间不够，所以出错了

memcpy(pTempImagebuffer,pDosHeader,P ...

谢谢表哥，原因是我PE标准头那加的0X18加错了，应该用sizeof算一下的