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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 文件操作 > 查看源码
winpeDoc.cpp
资源名称：winpe.zip　[点击查看]
上传用户：ch83438830
上传日期：2007-01-01
资源大小：104k
文件大小：9k
源码类别：

文件操作

开发平台：

Visual C++

winpeDoc.cpp：源码内容
  1. // winpeDoc.cpp : implementation of the CWinpeDoc class
  2. //
  4. #include "stdafx.h"
  5. #include "winpe.h"
  7. #include "winpeDoc.h"
  10. #ifdef _DEBUG
  11. #define new DEBUG_NEW
  12. #undef THIS_FILE
  13. static char THIS_FILE[] = __FILE__;
  14. #endif
  16. /////////////////////////////////////////////////////////////////////////////
  17. // CWinpeDoc
  19. IMPLEMENT_DYNCREATE(CWinpeDoc, CDocument)
  21. BEGIN_MESSAGE_MAP(CWinpeDoc, CDocument)
  22. //{{AFX_MSG_MAP(CWinpeDoc)
  23. ON_UPDATE_COMMAND_UI(ID_VIEWS_DEBUGDIR, OnUpdateViewsDebugdir)
  24. ON_UPDATE_COMMAND_UI(ID_VIEWS_DUMP, OnUpdateViewsDump)
  25. ON_UPDATE_COMMAND_UI(ID_VIEWS_EXPORTS, OnUpdateViewsExports)
  26. ON_UPDATE_COMMAND_UI(ID_VIEWS_HEADERS, OnUpdateViewsHeaders)
  27. ON_UPDATE_COMMAND_UI(ID_VIEWS_IMPORTS, OnUpdateViewsImports)
  28. ON_UPDATE_COMMAND_UI(ID_VIEWS_RESOURCES, OnUpdateViewsResources)
  29. ON_UPDATE_COMMAND_UI(ID_VIEWS_SECTNTBL, OnUpdateViewsSectntbl)
  30. //}}AFX_MSG_MAP
  31. END_MESSAGE_MAP()
  33. /////////////////////////////////////////////////////////////////////////////
  34. // CWinpeDoc construction/destruction
  36. CWinpeDoc::CWinpeDoc()
  37. {
  38. // TODO: add one-time construction code here
  39. m_nDocLength = 0;
  40. m_pFileData = 0;
  41. m_hFileMapping = 0;
  42. m_nFileType = 0;
  43. m_hFile = 0;
  45. m_printerFont = new CFont;
  46. m_screenFont = new CFont;
  48. CClientDC dc (AfxGetMainWnd());
  49. int nHeight = -((dc.GetDeviceCaps (LOGPIXELSY) * 10 ) / 72);
  51. m_screenFont->CreateFont(nHeight,
  52. 0,0,0,
  53. FW_NORMAL,
  54. 0,0,0,
  55. DEFAULT_CHARSET,
  56. OUT_CHARACTER_PRECIS,
  57. CLIP_CHARACTER_PRECIS,
  58. DEFAULT_QUALITY,
  59. DEFAULT_PITCH | FF_DONTCARE,
  60. "Courier New");
  61. nHeight = -((dc.GetDeviceCaps (LOGPIXELSY) * 12 ) / 72);
  63. m_printerFont->CreateFont (nHeight, 0, 0, 0, FW_NORMAL, 0, 0, 0,
  64.         DEFAULT_CHARSET, OUT_CHARACTER_PRECIS, CLIP_CHARACTER_PRECIS,
  65.         DEFAULT_QUALITY, DEFAULT_PITCH | FF_DONTCARE, "Courier New");
  67. TEXTMETRIC tm;
  68. CFont *pOldFont;
  70. pOldFont = dc.SelectObject(m_screenFont);
  71. dc.GetTextMetrics (&tm);
  72. m_cyScreen = tm.tmHeight + tm.tmExternalLeading;
  73. dc.SelectObject(pOldFont);
  76. //pOldFont = dc.SelectObject(&m_printerFont);
  77. //dc.GetTextMetrics (&tm);
  78. //m_cyPrinter = tm.tmHeight + tm.tmExternalLeading;
  79. //dc.SelectObject(pOldFont);
  83. }
  85. CWinpeDoc::~CWinpeDoc()
  86. {
  87. delete m_printerFont;
  88. delete m_screenFont;
  89. }
  91. BOOL CWinpeDoc::OnNewDocument()
  92. {
  93. if (!CDocument::OnNewDocument())
  94. return FALSE;
  96. // TODO: add reinitialization code here
  97. // (SDI documents will reuse this document)
  101. return TRUE;
  102. }
  104. /////////////////////////////////////////////////////////////////////////////
  105. // CWinpeDoc serialization
  107. void CWinpeDoc::Serialize(CArchive& ar)
  108. {
  109. if (ar.IsLoading())
  110. {
  111. // //CFile* pFile = ar.GetFile();
  112. // m_nDocLength = (UINT) pFile->GetLength();
  113. // try
  114. // {
  115. // m_pFileData = new BYTE[m_nDocLength];
  116. // }
  117. // catch (CMemoryException* e)
  118. // {
  119. // m_nDocLength = 0;
  120. // throw e;
  121. // }
  122. //
  123. // try
  124. // {
  125. // pFile->Read (m_pFileData,m_nDocLength);
  126. // }
  127. // catch (CFileException* e)
  128. // {
  129. // delete[] m_pFileData;
  130. // m_pFileData = NULL;
  131. // m_nDocLength = 0;
  132. // throw e;
  133. // }
  136. }
  137. else
  138. {
  139. // TODO: add storing code here
  140. }
  143. }
  145. /////////////////////////////////////////////////////////////////////////////
  146. // CWinpeDoc diagnostics
  148. #ifdef _DEBUG
  149. void CWinpeDoc::AssertValid() const
  150. {
  151. CDocument::AssertValid();
  152. }
  154. void CWinpeDoc::Dump(CDumpContext& dc) const
  155. {
  156. CDocument::Dump(dc);
  157. }
  158. #endif //_DEBUG
  160. /////////////////////////////////////////////////////////////////////////////
  161. // CWinpeDoc commands
  163. UINT CWinpeDoc::GetBytes(UINT nIndex,UINT nCount,PVOID pBuffer)
  164. {
  166. if (nIndex >= m_nDocLength)
  167. return 0;
  169. UINT nLength = nCount;
  170. if((nIndex + nCount) > m_nDocLength)
  171. nLength = m_nDocLength - nIndex;
  173. ::CopyMemory(pBuffer,m_pFileData + nIndex,nLength);
  174. return nLength;
  176. }
  178. UINT CWinpeDoc::GetDocumentLength()
  179. {
  181. return m_nDocLength;
  182. }
  184. void CWinpeDoc::DeleteContents() 
  185. {
  186. // TODO: Add your specialized code here and/or call the base class
  187. //if (m_pFileData != NULL)
  188. //{
  189. // delete[] m_pFileData;
  190. // m_pFileData = NULL;
  191. // m_nDocLength = 0;
  192. //}
  194. if (m_pFileData)
  195. {
  196. UnmapViewOfFile(m_pFileData);
  197. m_pFileData = NULL;
  198. }
  200. if (m_hFileMapping)
  201. {
  202. CloseHandle(m_hFileMapping);
  203. m_hFileMapping = NULL;
  204. }
  206. if (m_hFile)
  207. {
  208. if (m_hFile != INVALID_HANDLE_VALUE)
  209. {
  210. CloseHandle(m_hFile);
  211. m_nDocLength = 0;
  212. m_hFile = NULL;
  213. }
  214. }
  216. CWnd* mywnd = AfxGetMainWnd();
  217. if (mywnd) // after first time thru
  218. // dumpview always default view
  219. mywnd->SendMessage(WM_RESET_FRAMEVIEW,ID_VIEWS_DUMP,0);
  221. // must call base class
  222. CDocument::DeleteContents();
  223. }
  225. BOOL CWinpeDoc::OnOpenDocument(LPCTSTR lpszPathName) 
  226. {
  227. if (!CDocument::OnOpenDocument(lpszPathName))
  228. return FALSE;
  230. // TODO: Add your specialized creation code here
  231. m_pFileData = 0;
  232. m_hFileMapping = 0;
  233. m_nFileType = 0;
  234. m_nDocLength = 0;
  235. m_hFile = CreateFile(lpszPathName,GENERIC_READ,
  236. FILE_SHARE_READ,NULL,
  237. OPEN_EXISTING,
  238. FILE_ATTRIBUTE_NORMAL,
  239. (HANDLE)0);
  240. if (m_hFile == INVALID_HANDLE_VALUE)
  241. return FALSE;
  242. m_hFileMapping = CreateFileMapping(m_hFile,
  243. NULL,
  244. PAGE_READONLY,
  245. 0,0,NULL);
  246. if (m_hFileMapping == 0)
  247. return FALSE;
  248. m_pFileData = (LPBYTE)MapViewOfFile(m_hFileMapping,
  249. FILE_MAP_READ,
  250. 0,0,0);
  251. if (m_pFileData == 0)
  252. return FALSE;
  254. m_nDocLength = GetFileSize (m_hFile, NULL);
  256. m_nFileType = 0;
  257. CHAR *hit;
  258. hit = strstr(lpszPathName,".exe");
  259. if (hit != NULL)
  260. m_nFileType = FILETYPE_EXE;
  261. hit = strstr(lpszPathName,".EXE");
  262. if (hit != NULL)
  263. m_nFileType = FILETYPE_EXE;
  265. hit = strstr(lpszPathName,".dll");
  266. if (hit != NULL)
  267. m_nFileType = FILETYPE_DLL;
  268. hit = strstr(lpszPathName,".DLL");
  269. if (hit != NULL)
  270. m_nFileType = FILETYPE_DLL;
  272. m_Dump_Enabled = TRUE; // dump anything
  274. if (m_nFileType == 0)
  275. {
  276. m_DebugDir_Enabled = FALSE;
  277. m_Exports_Enabled = FALSE;
  278. m_Headers_Enabled = FALSE;
  279. m_Imports_Enabled = FALSE;
  280. m_Resources_Enabled = FALSE;
  281. m_Sections_Enabled = FALSE;
  284. }
  285. else
  286. {
  287. m_DebugDir_Enabled = TRUE;
  288. m_Exports_Enabled = TRUE;
  289. m_Headers_Enabled = TRUE;
  290. m_Imports_Enabled = TRUE;
  291. m_Resources_Enabled = TRUE;
  292. m_Sections_Enabled = TRUE;
  293. }
  295. return TRUE;
  296. }
  298. BYTE* CWinpeDoc::GetDocumentDataP()
  299. {
  300. return m_pFileData;
  302. }
  304. INT CWinpeDoc::GetOpenFileType()
  305. {
  306. return m_nFileType;
  308. }
  312. void CWinpeDoc::OnUpdateViewsDebugdir(CCmdUI* pCmdUI) 
  313. {
  314. // TODO: Add your command update UI handler code here
  315. pCmdUI->Enable(m_DebugDir_Enabled);
  316. pCmdUI->SetCheck(m_DebugDir_Checked);
  318. }
  320. void CWinpeDoc::OnUpdateViewsDump(CCmdUI* pCmdUI) 
  321. {
  322. // TODO: Add your command update UI handler code here
  323. pCmdUI->Enable(m_Dump_Enabled);
  324. pCmdUI->SetCheck(m_Dump_Checked);
  326. }
  328. void CWinpeDoc::OnUpdateViewsExports(CCmdUI* pCmdUI) 
  329. {
  330. // TODO: Add your command update UI handler code here
  331. pCmdUI->Enable(m_Exports_Enabled);
  332. pCmdUI->SetCheck(m_Exports_Checked);
  334. }
  336. void CWinpeDoc::OnUpdateViewsHeaders(CCmdUI* pCmdUI) 
  337. {
  338. // TODO: Add your command update UI handler code here
  339. pCmdUI->Enable(m_Headers_Enabled);
  340. pCmdUI->SetCheck(m_Headers_Checked);
  342. }
  344. void CWinpeDoc::OnUpdateViewsImports(CCmdUI* pCmdUI) 
  345. {
  346. // TODO: Add your command update UI handler code here
  347. pCmdUI->Enable(m_Imports_Enabled);
  348. pCmdUI->SetCheck(m_Imports_Checked);
  350. }
  352. void CWinpeDoc::OnUpdateViewsResources(CCmdUI* pCmdUI) 
  353. {
  354. // TODO: Add your command update UI handler code here
  355. pCmdUI->Enable(m_Resources_Enabled);
  356. pCmdUI->SetCheck(m_Resources_Checked);
  358. }
  360. void CWinpeDoc::OnUpdateViewsSectntbl(CCmdUI* pCmdUI) 
  361. {
  362. // TODO: Add your command update UI handler code here
  363. pCmdUI->Enable(m_Sections_Enabled);
  364. pCmdUI->SetCheck(m_Sections_Checked);
  366. }
  368. void CWinpeDoc::TurnChecksOff()
  369. {
  370. m_Dump_Checked = FALSE;
  371. m_DebugDir_Checked = FALSE;
  372. m_Imports_Checked = FALSE;
  373. m_Exports_Checked = FALSE;
  374. m_Resources_Checked = FALSE;
  375. m_Headers_Checked = FALSE;
  376. m_Sections_Checked = FALSE;
  378. }
  381. PIMAGE_SECTION_HEADER CWinpeDoc::GetEnclosingSectionHeader(DWORD rva,
  382.  PIMAGE_NT_HEADERS pNTHeader)
  383. {
  384. PIMAGE_SECTION_HEADER section = IMAGE_FIRST_SECTION(pNTHeader);
  385.     unsigned i;
  386.     
  387.     for ( i=0; i < pNTHeader->FileHeader.NumberOfSections; i++, section++ )
  388.     {
  389.         // Is the RVA within this section?
  390.         if ( (rva >= section->VirtualAddress) && 
  391.              (rva < (section->VirtualAddress + section->Misc.VirtualSize)))
  392.             return section;
  393.     }
  394.     
  395.     return 0;
  397. }
  399. void CWinpeDoc::SetDocPrtFont(LPLOGFONT lp)
  400. {
  401. delete m_printerFont;
  402. m_printerFont = new CFont;
  403. //m_printerFont.DeleteObject();
  404. m_printerFont->CreateFontIndirect(lp);
  406. }
  408. void CWinpeDoc::SetDocScrnFont(LPLOGFONT lp)
  409. {
  410. delete m_screenFont;
  411. m_screenFont = new CFont;
  412. //m_screenFont.DeleteObject();
  413. m_screenFont->CreateFontIndirect(lp);
  415. }