模拟windows PE文件加载器

forwe

1. #include "stdio.h"
   
2. #include <WINDOWS.H>
   
3. 
   
4. typedef BOOL (__stdcall *pSetFilePointerEx)(
   
5.    HANDLE hFile,
   
6.    LARGE_INTEGER liDistanceToMove,
   
7.    PLARGE_INTEGER lpNewFilePointer,
   
8.    DWORD dwMoveMethod);
   
9. 
   
10. 
    
11. 
    
12. void ReadPeFile()
    
13. {
    
14.         HANDLE hFile;
    
15.         HMODULE h;
    
16.         BOOL bStatus;
    
17.         ULONG uIndex;
    
18.         DWORD dwRetSize;
    
19.         LARGE_INTEGER FileOffset;
    
20.         IMAGE_DOS_HEADER ImageDosHeader;
    
21.         IMAGE_NT_HEADERS ImageNtHeader;
    
22.         IMAGE_SECTION_HEADER *pImageSectionHeader;
    
23. 
    
24.         pSetFilePointerEx SetFilePointerEx;
    
25. 
    
26.         hFile = CreateFile("c:\\NOTEPAD.EXE", GENERIC_ALL, FILE_SHARE_READ, NULL, OPEN_ALWAYS,
    
27.                 FILE_ATTRIBUTE_NORMAL, NULL);
    
28. 
    
29.         if (hFile == INVALID_HANDLE_VALUE)
    
30.         {
    
31.                 printf("CreateFile Failed!\n");
    
32.                 return;
    
33.         }
    
34. 
    
35.         bStatus = ReadFile(hFile, &ImageDosHeader, sizeof(IMAGE_DOS_HEADER), &dwRetSize, NULL);
    
36. 
    
37.         if (bStatus == FALSE)
    
38.         {
    
39.                 printf("read image_dos_header falied! :%d \n", GetLastError());
    
40.                 CloseHandle(hFile);
    
41.         }
    
42.        
    
43.         printf("e_magic:%s, e_lfanew:0x%X\n", &ImageDosHeader.e_magic, ImageDosHeader.e_lfanew);
    
44. 
    
45. 
    
46.         h = LoadLibrary("Kernel32.dll");
    
47.         if(!h)return;
    
48.        
    
49.         SetFilePointerEx=(pSetFilePointerEx)GetProcAddress(h,"SetFilePointerEx");
    
50. 
    
51.         FileOffset.QuadPart = ImageDosHeader.e_lfanew;
    
52.     bStatus = SetFilePointerEx(hFile, FileOffset, NULL, FILE_BEGIN);
    
53.         if (bStatus == FALSE)
    
54.         {
    
55.                 printf("SetFilePointerEx falied! :%d \n", GetLastError());
    
56.                 CloseHandle(hFile);
    
57.         }
    
58.        
    
59. 
    
60.         bStatus = ReadFile(hFile, &ImageNtHeader, sizeof(IMAGE_NT_HEADERS), &dwRetSize, NULL);
    
61.         if (bStatus == FALSE)
    
62.         {
    
63.                 printf("read image_nt_header falied! :%d \n", GetLastError());
    
64.                 CloseHandle(hFile);
    
65.         }
    
66.         printf("Signature:%s, NumberOfSections: %d\n", &ImageNtHeader.Signature, ImageNtHeader.FileHeader.NumberOfSections);
    
67. 
    
68. 
    
69.         FileOffset.QuadPart += sizeof(IMAGE_NT_HEADERS);
    
70.         bStatus = SetFilePointerEx(hFile, FileOffset, NULL, FILE_BEGIN);
    
71.         if (bStatus == FALSE)
    
72.         {
    
73.                 printf("SetFilePointerEx falied! :%d \n", GetLastError());
    
74.                 CloseHandle(hFile);
    
75.         }       
    
76.        
    
77.         pImageSectionHeader = (IMAGE_SECTION_HEADER *)malloc(sizeof(IMAGE_SECTION_HEADER) * ImageNtHeader.FileHeader.NumberOfSections);
    
78.         if (pImageSectionHeader == 0)
    
79.         {
    
80.                 CloseHandle(hFile);
    
81.                 return;
    
82.         }
    
83.         bStatus = ReadFile(hFile, pImageSectionHeader, sizeof(IMAGE_SECTION_HEADER) * ImageNtHeader.FileHeader.NumberOfSections, &dwRetSize, NULL);
    
84.         if (bStatus == FALSE)
    
85.         {
    
86.                 printf("read IMAGE_SECTION_HEADER falied! :%d \n", GetLastError());
    
87.                 CloseHandle(hFile);
    
88.         }
    
89. 
    
90.         for (uIndex = 0; uIndex < ImageNtHeader.FileHeader.NumberOfSections; uIndex++)
    
91.         {
    
92.                 printf("pImageSectionHeader[%d]:%s\n", uIndex, &pImageSectionHeader[uIndex].Name);
    
93.         }
    
94. 
    
95.         CloseHandle(hFile);
    
96.         free(pImageSectionHeader);
    
97.        
    
98. }
    
99. 
    
100. void ReadFileToMemory()
     
101. {
     
102.         HANDLE hFile;
     
103.         HMODULE h;
     
104.         BOOL bStatus;
     
105.         ULONG uIndex;
     
106.         DWORD dwRetSize;
     
107.         DWORD FileAlign, VirtualAlign;
     
108.         DWORD VirtualSizeOfImage;
     
109.         PVOID pVirtualPoint;
     
110.         DWORD SectionVirtualAddr, SizeOfSection;
     
111.         DWORD PointerToRawData;
     
112.         LARGE_INTEGER FileOffset;
     
113.         IMAGE_DOS_HEADER ImageDosHeader;
     
114.         IMAGE_NT_HEADERS ImageNtHeader;
     
115.         IMAGE_SECTION_HEADER *pImageSectionHeader;
     
116.        
     
117.         pSetFilePointerEx SetFilePointerEx;
     
118.        
     
119.         hFile = CreateFile("c:\\NOTEPAD.EXE", GENERIC_ALL, FILE_SHARE_READ, NULL, OPEN_ALWAYS,
     
120.                 FILE_ATTRIBUTE_NORMAL, NULL);
     
121.        
     
122.         if (hFile == INVALID_HANDLE_VALUE)
     
123.         {
     
124.                 printf("CreateFile Failed!\n");
     
125.                 return;
     
126.         }
     
127.        
     
128.         bStatus = ReadFile(hFile, &ImageDosHeader, sizeof(IMAGE_DOS_HEADER), &dwRetSize, NULL);
     
129.        
     
130.         if (bStatus == FALSE)
     
131.         {
     
132.                 printf("read image_dos_header falied! :%d \n", GetLastError());
     
133.                 CloseHandle(hFile);
     
134.                 return;
     
135.         }
     
136.        
     
137.         printf("e_magic:%s, e_lfanew:0x%X\n", &ImageDosHeader.e_magic, ImageDosHeader.e_lfanew);
     
138.        
     
139.        
     
140.         h = LoadLibrary("Kernel32.dll");
     
141.         if(!h)return;
     
142.        
     
143.         SetFilePointerEx=(pSetFilePointerEx)GetProcAddress(h,"SetFilePointerEx");
     
144.        
     
145.         FileOffset.QuadPart = ImageDosHeader.e_lfanew;
     
146.     bStatus = SetFilePointerEx(hFile, FileOffset, NULL, FILE_BEGIN);
     
147.         if (bStatus == FALSE)
     
148.         {
     
149.                 printf("SetFilePointerEx falied! :%d \n", GetLastError());
     
150.                 CloseHandle(hFile);
     
151.                 return;
     
152.         }
     
153.        
     
154.        
     
155.         bStatus = ReadFile(hFile, &ImageNtHeader, sizeof(IMAGE_NT_HEADERS), &dwRetSize, NULL);
     
156.         if (bStatus == FALSE)
     
157.         {
     
158.                 printf("read image_nt_header falied! :%d \n", GetLastError());
     
159.                 CloseHandle(hFile);
     
160.                 return;
     
161.         }
     
162.         printf("Signature:%s, NumberOfSections: %d\n", &ImageNtHeader.Signature, ImageNtHeader.FileHeader.NumberOfSections);
     
163.        
     
164.        
     
165.         FileOffset.QuadPart += sizeof(IMAGE_NT_HEADERS);
     
166.         bStatus = SetFilePointerEx(hFile, FileOffset, NULL, FILE_BEGIN);
     
167.         if (bStatus == FALSE)
     
168.         {
     
169.                 printf("SetFilePointerEx falied! :%d \n", GetLastError());
     
170.                 CloseHandle(hFile);
     
171.                 return;
     
172.         }       
     
173.        
     
174.         pImageSectionHeader = (IMAGE_SECTION_HEADER *)malloc(sizeof(IMAGE_SECTION_HEADER) * ImageNtHeader.FileHeader.NumberOfSections);
     
175.         if (pImageSectionHeader == 0)
     
176.         {
     
177.                 CloseHandle(hFile);
     
178.                 return;
     
179.         }
     
180.         bStatus = ReadFile(hFile, pImageSectionHeader, sizeof(IMAGE_SECTION_HEADER) * ImageNtHeader.FileHeader.NumberOfSections, &dwRetSize, NULL);
     
181.         if (bStatus == FALSE)
     
182.         {
     
183.                 printf("read IMAGE_SECTION_HEADER falied! :%d \n", GetLastError());
     
184.                 CloseHandle(hFile);
     
185.                 free(pImageSectionHeader);
     
186.                 return;
     
187.         }
     
188.        
     
189.         for (uIndex = 0; uIndex < ImageNtHeader.FileHeader.NumberOfSections; uIndex++)
     
190.         {
     
191.                 printf("pImageSectionHeader[%d]:%s\n", uIndex, &pImageSectionHeader[uIndex].Name);
     
192.         }
     
193.        
     
194. 
     
195.         FileAlign = ImageNtHeader.OptionalHeader.FileAlignment;
     
196.         VirtualAlign = ImageNtHeader.OptionalHeader.SectionAlignment;
     
197.         VirtualSizeOfImage = ImageNtHeader.OptionalHeader.SizeOfImage;
     
198. 
     
199.         pVirtualPoint = malloc(VirtualSizeOfImage);
     
200.         if (pVirtualPoint == 0)
     
201.         {
     
202.                 CloseHandle(hFile);
     
203.                 free(pImageSectionHeader);
     
204.                 return;
     
205.         }
     
206. 
     
207.         memset(pVirtualPoint, 0, VirtualSizeOfImage);
     
208. 
     
209.         memcpy(pVirtualPoint, &ImageDosHeader, sizeof(IMAGE_DOS_HEADER));
     
210.         memcpy((PVOID)((LONG)pVirtualPoint + ImageDosHeader.e_lfanew), &ImageNtHeader,
     
211.                 sizeof(IMAGE_NT_HEADERS));
     
212.         memcpy((PVOID)((LONG)pVirtualPoint + ImageDosHeader.e_lfanew + sizeof(IMAGE_NT_HEADERS)),
     
213.                 pImageSectionHeader, sizeof(IMAGE_SECTION_HEADER) * ImageNtHeader.FileHeader.NumberOfSections);
     
214.                
     
215. 
     
216.         for (uIndex = 0; uIndex < ImageNtHeader.FileHeader.NumberOfSections; uIndex++)
     
217.         {
     
218.                 SectionVirtualAddr = pImageSectionHeader[uIndex].VirtualAddress;
     
219.                 SizeOfSection = pImageSectionHeader[uIndex].SizeOfRawData;
     
220.                 PointerToRawData = pImageSectionHeader[uIndex].PointerToRawData;
     
221. 
     
222.                 FileOffset.QuadPart = PointerToRawData;
     
223.                 bStatus = SetFilePointerEx(hFile, FileOffset, NULL, FILE_BEGIN);
     
224.                 if (bStatus == FALSE)
     
225.                 {
     
226.                         CloseHandle(hFile);
     
227.                         free(pImageSectionHeader);
     
228.                         free(pVirtualPoint);
     
229.                         return;
     
230.                 }
     
231. 
     
232.                 bStatus = ReadFile(hFile, (PVOID)((DWORD)pVirtualPoint + SectionVirtualAddr),
     
233.                         SizeOfSection, &dwRetSize, NULL);
     
234.                 if (bStatus == FALSE)
     
235.                 {
     
236.                         CloseHandle(hFile);
     
237.                         free(pImageSectionHeader);
     
238.                         free(pVirtualPoint);
     
239.                         return;
     
240.                 }
     
241.         }
     
242. 
     
243.         printf("pVirtualPoint:0x%X\n", pVirtualPoint);
     
244. 
     
245.         getchar();
     
246.         getchar();
     
247. 
     
248.         CloseHandle(hFile);
     
249.         free(pImageSectionHeader);
     
250.         free(pVirtualPoint);
     
251. 
     
252. }
     
253. 
     
254. void main(int argc, char** argv)
     
255. {
     
256.         ReadFileToMemory();
     
257. }

复制代码