开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 文件操作 > 查看源码
PeClean.c
资源名称：PEClean.rar　[点击查看]
上传用户：kuaidait
上传日期：2013-03-31
资源大小：4k
文件大小：8k
源码类别：

文件操作

开发平台：

C/C++

PeClean.c：源码内容
  1. #include "PeClean.h"
  3. DWORD IsPE(HANDLE hFile)
  4. {
  5. char ReadBuffer[4];
  6. DWORD dwRead;
  7. DWORD dwOffset;
  8. DWORD dwSizeOfHeader;
  10. memset(ReadBuffer, 0, 4);
  11. SetFilePointer(hFile, 0, NULL, FILE_BEGIN);
  12. ReadFile(hFile, ReadBuffer, 4, &dwRead, NULL);
  13. if (ReadBuffer[0]!='M' || ReadBuffer[1]!='Z')
  14. {
  15. return 0;
  16. }
  18. memset(ReadBuffer, 0, 4);
  19. SetFilePointer(hFile, 0x3C, NULL, FILE_BEGIN);
  20. ReadFile(hFile, &dwOffset,  4, &dwRead, NULL);
  21. SetFilePointer(hFile, dwOffset, NULL, FILE_BEGIN);
  22. ReadFile(hFile, ReadBuffer, 4, &dwRead, NULL);
  23. if (ReadBuffer[0]!='P' || ReadBuffer[1]!='E')
  24. {
  25. return 0;
  26. }
  28.     SetFilePointer(hFile, dwOffset+0x54, NULL,  FILE_BEGIN);
  29. ReadFile(hFile, &dwSizeOfHeader, 4, &dwRead, NULL); 
  31. return dwSizeOfHeader;
  32. }
  34. BYTE* ReadHeader(HANDLE hFile, DWORD dwSizeOfHeader)
  35. {
  36. BYTE* hMapOfHeader;
  37. DWORD dwRead;
  39. hMapOfHeader = VirtualAlloc(NULL, dwSizeOfHeader, MEM_COMMIT, PAGE_READWRITE);
  40. if (NULL == hMapOfHeader)
  41. {
  42. return NULL;
  43. }
  44. SetFilePointer(hFile, 0, NULL, FILE_BEGIN);
  45. ReadFile(hFile, hMapOfHeader, dwSizeOfHeader, &dwRead, NULL);
  46. if (dwRead != dwSizeOfHeader)
  47. {
  48. VirtualFree(hMapOfHeader, 0, MEM_RELEASE);
  49. return NULL;
  50. }
  52. return hMapOfHeader;
  53. }
  55. void WriteHeader(HANDLE hFile, BYTE* hMapOfHeader, DWORD dwSizeOfHeader)
  56. {
  57. DWORD dwWritten;
  59. SetFilePointer(hFile, 0, NULL, FILE_BEGIN);
  60. WriteFile(hFile, hMapOfHeader, dwSizeOfHeader, &dwWritten, NULL);
  62. return ;
  63. }
  65. void BackupFile(char* szFileName)
  66. {
  67. BOOL bBackupFile;
  68. char szBackupName[MAX_PATH];
  70. bBackupFile = SendMessage(GetDlgItem(hWnd, IDC_CHECK_BAKFILE), BM_GETCHECK, 0, 0);
  71. if (bBackupFile)
  72. {
  73. strcpy(szBackupName, szFileName);
  74. strcat(szBackupName, ".bak");
  75. CopyFile(szFileName, szBackupName, FALSE);
  76. }
  78. return ;
  79. }
  81. void CleanDosHeader(BYTE* hMapOfHeader)
  82. {
  83. BOOL bMZ;
  84. BYTE* i;
  86. bMZ = SendMessage(GetDlgItem(hWnd, IDC_CHECK_MZ), BM_GETCHECK, 0, 0) ^ 1;
  88. for (i=hMapOfHeader+2; i<(BYTE *)peHeader; i++)
  89. {
  90. (*i) *= bMZ;
  91. }
  92. *(LPWORD)(hMapOfHeader+0x3C) = (WORD)((BYTE *)peHeader-hMapOfHeader);
  94. return ;
  95. }
  97. void CleanFileHeader(BYTE* hMapOfHeader)
  98. {
  99. peHeader->FileHeader.TimeDateStamp        *= 
  100. SendMessage(GetDlgItem(hWnd, IDC_CHECK_FILE1), BM_GETCHECK, 0, 0) ^ 1 ;
  101. peHeader->FileHeader.PointerToSymbolTable *= 
  102. SendMessage(GetDlgItem(hWnd, IDC_CHECK_FILE2), BM_GETCHECK, 0, 0) ^ 1 ;
  103. peHeader->FileHeader.NumberOfSymbols      *= 
  104. SendMessage(GetDlgItem(hWnd, IDC_CHECK_FILE3), BM_GETCHECK, 0, 0) ^ 1 ;
  106. return ;
  107. }
  110. void CleanOptionalHeader(BYTE* hMapOfHeader)
  111. {
  112. peHeader->OptionalHeader.MajorLinkerVersion          *=
  113. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL1), BM_GETCHECK, 0, 0) ^ 1 ;
  114. peHeader->OptionalHeader.MinorLinkerVersion          *=
  115. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL1), BM_GETCHECK, 0, 0) ^ 1 ;
  116. peHeader->OptionalHeader.MajorOperatingSystemVersion *=
  117. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL2), BM_GETCHECK, 0, 0) ^ 1 ;
  118.     peHeader->OptionalHeader.MinorOperatingSystemVersion *=
  119. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL2), BM_GETCHECK, 0, 0) ^ 1 ;
  121. peHeader->OptionalHeader.SizeOfCode              *=
  122. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL3), BM_GETCHECK, 0, 0) ^ 1 ;
  123. peHeader->OptionalHeader.SizeOfInitializedData   *=
  124. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL4), BM_GETCHECK, 0, 0) ^ 1 ;
  125. peHeader->OptionalHeader.SizeOfUninitializedData *=
  126. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL4), BM_GETCHECK, 0, 0) ^ 1 ;
  128. peHeader->OptionalHeader.BaseOfCode   *=
  129. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL5), BM_GETCHECK, 0, 0) ^ 1 ;
  130. peHeader->OptionalHeader.BaseOfData   *=
  131. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL6), BM_GETCHECK, 0, 0) ^ 1 ;
  133.     peHeader->OptionalHeader.MajorImageVersion   *=
  134. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL7), BM_GETCHECK, 0, 0) ^ 1 ;
  135.     peHeader->OptionalHeader.MinorImageVersion   *=
  136.   SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL7), BM_GETCHECK, 0, 0) ^ 1 ;
  137. peHeader->OptionalHeader.Win32VersionValue   *=
  138. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL8), BM_GETCHECK, 0, 0) ^ 1 ;
  140. peHeader->OptionalHeader.CheckSum    *=
  141. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL9), BM_GETCHECK, 0, 0) ^ 1 ;
  143. peHeader->OptionalHeader.LoaderFlags *=
  144. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL10), BM_GETCHECK, 0, 0) ^ 1 ;
  146. peHeader->OptionalHeader.DllCharacteristics *=
  147. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL11), BM_GETCHECK, 0, 0) ^ 1 ;
  149. return ;
  150. }
  152. void CleanFile(char* szFileName)
  153. {
  154. HANDLE hFile;
  155. DWORD dwSizeOfHeader;
  156. BYTE* hMapOfHeader;
  158. hFile = CreateFile(szFileName, 
  159.    GENERIC_READ+GENERIC_WRITE, 
  160.    FILE_SHARE_READ+FILE_SHARE_WRITE,
  161.    0, 
  162.    OPEN_EXISTING, 
  163.    FILE_ATTRIBUTE_NORMAL,NULL);
  165. if (hFile == INVALID_HANDLE_VALUE)
  166. {
  167. MessageBox(hWnd, "Cannot Open the File", "Error", 0);
  168. return ;
  169. }
  170. dwSizeOfHeader = IsPE(hFile);
  171. if (dwSizeOfHeader == 0)
  172. {
  173. CloseHandle(hFile);
  174. MessageBox(hWnd, "Invalid PE File", "Error", 0);
  175. return ;
  176. }
  177. hMapOfHeader = ReadHeader(hFile, dwSizeOfHeader);
  178. if (NULL == hMapOfHeader)
  179. {
  180. CloseHandle(hFile);
  181. MessageBox(hWnd, "Cannot Read PeHeader", "Error", 0);
  182. return ;
  183. }
  185. peHeader = (IMAGE_NT_HEADERS32 *)(hMapOfHeader + *(LPWORD)(hMapOfHeader+0x3C));
  187. BackupFile(szFileName);
  189. CleanDosHeader(hMapOfHeader);
  191. CleanFileHeader(hMapOfHeader);
  193. CleanOptionalHeader(hMapOfHeader);
  195. *(LPWORD)(hMapOfHeader+0x3C) = (WORD)((BYTE *)peHeader-hMapOfHeader);
  196. WriteHeader(hFile, hMapOfHeader, dwSizeOfHeader);
  198. VirtualFree(hMapOfHeader, 0, MEM_RELEASE);
  199. CloseHandle(hFile);
  201. MessageBox(hWnd, "Done!", "ok", 0);
  202. return ;
  203. }
  205. void SetCheck(BOOL bFlag)
  206. {
  207. SendMessage(GetDlgItem(hWnd, IDC_CHECK_BAKFILE)    , BM_SETCHECK, bFlag, 0);
  208. SendMessage(GetDlgItem(hWnd, IDC_CHECK_MZ)         , BM_SETCHECK, bFlag, 0);
  209. SendMessage(GetDlgItem(hWnd, IDC_CHECK_FILE1)      , BM_SETCHECK, bFlag, 0);
  210. SendMessage(GetDlgItem(hWnd, IDC_CHECK_FILE2)      , BM_SETCHECK, bFlag, 0);
  211. SendMessage(GetDlgItem(hWnd, IDC_CHECK_FILE3)      , BM_SETCHECK, bFlag, 0);
  212. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL1)  , BM_SETCHECK, bFlag, 0);
  213. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL2)  , BM_SETCHECK, bFlag, 0);
  214. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL3)  , BM_SETCHECK, bFlag, 0);
  215. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL4)  , BM_SETCHECK, bFlag, 0);
  216. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL5)  , BM_SETCHECK, bFlag, 0);
  217. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL6)  , BM_SETCHECK, bFlag, 0);
  218. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL7)  , BM_SETCHECK, bFlag, 0);
  219. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL8)  , BM_SETCHECK, bFlag, 0);
  220. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL9)  , BM_SETCHECK, bFlag, 0);
  221. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL10) , BM_SETCHECK, bFlag, 0);
  222. SendMessage(GetDlgItem(hWnd, IDC_CHECK_OPTIONAL11) , BM_SETCHECK, bFlag, 0);
  224. SendMessage(GetDlgItem(hWnd, IDC_CHECK_ALL)  , BM_SETCHECK, bFlag,   0);
  225. SendMessage(GetDlgItem(hWnd, IDC_CHECK_NONE) , BM_SETCHECK, bFlag^1, 0);
  226. }
  228. int CALLBACK DlgProc(HWND hDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
  229. {
  230. switch (uMsg)
  231. {
  232. case WM_CLOSE:
  233. PostQuitMessage(0);
  234. break;
  236. case WM_INITDIALOG:
  237. hWnd = hDlg;
  239. hIcon = LoadIcon(hInst, MAKEINTRESOURCE(IDI_ICON1));
  240. SendMessage(hWnd, WM_SETICON, (WPARAM)TRUE, (LPARAM)hIcon);
  241. SetCheck(TRUE);
  242. break;
  244. case WM_DROPFILES:
  245. if (DragQueryFile((HDROP)wParam, 0, szFileName, MAX_PATH))
  246. {
  247. CleanFile(szFileName);
  248. }
  249. break;
  251. case WM_COMMAND:
  252. if (lParam == GetDlgItem(hWnd, IDC_CHECK_ALL))
  253. {
  254. SetCheck(1);
  255. }
  256. else if (lParam == GetDlgItem(hWnd, IDC_CHECK_NONE))
  257. {
  258. SetCheck(0);
  259. }
  260. break;
  261. default:
  262. break;
  263. }
  265. return 0;
  266. }
  269. int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PSTR szCmdLine, int iCmdShow)
  270. {
  271. InitCommonControls();
  273. hInst = hInstance;
  274. DialogBox(hInst, MAKEINTRESOURCE(IDD_DIALOG_PECLEAN), NULL, (DLGPROC)DlgProc);
  276. return 0;
  277. }