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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 钩子与API截获 > 查看源码
MainFrm.cpp
资源名称：apihook_Code.zip　[点击查看]
上传用户：jstlsd
上传日期：2007-01-13
资源大小：186k
文件大小：4k
源码类别：

钩子与API截获

开发平台：

Visual C++

MainFrm.cpp：源码内容
  1. //---------------------------------------------------------------------------
  2. //
  3. // MainFrm.cpp
  4. //
  5. // SUBSYSTEM:   Hook system
  6. //
  7. // MODULE:      Hook server
  8. //
  9. // DESCRIPTION: Implementation of the CMainFrame class
  10. //             
  11. // AUTHOR: Ivo Ivanov (ivopi@hotmail.com)
  12. // DATE: 2001 December v1.00
  13. //
  14. //---------------------------------------------------------------------------
  15. #include "stdafx.h"
  16. #include "HookSrv.h"
  17. #include "MainFrm.h"
  18. #include "..CommonCustomMessages.h"
  20. //---------------------------------------------------------------------------
  21. //
  22. // Constants
  23. // 
  24. //---------------------------------------------------------------------------
  26. // Message ID used for tray notifications
  27. #define WM_MY_TRAY_NOTIFICATION WM_USER + 0x500
  29. //---------------------------------------------------------------------------
  30. //
  31. // class CMainFrame
  32. // 
  33. //---------------------------------------------------------------------------
  35. IMPLEMENT_DYNAMIC(CMainFrame, CFrameWnd)
  37. BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
  38. //{{AFX_MSG_MAP(CMainFrame)
  39. ON_WM_CREATE()
  40. ON_WM_CLOSE()
  41. ON_COMMAND(ID_APP_EXIT, OnAppExit)
  42. ON_MESSAGE(WM_MY_TRAY_NOTIFICATION, OnTrayNotification)
  43. //}}AFX_MSG_MAP
  44. ON_REGISTERED_MESSAGE(UWM_HOOKTOOL_DLL_LOADED, OnDllLoaded)
  45. ON_REGISTERED_MESSAGE(UWM_HOOKTOOL_DLL_UNLOADED, OnDllUnLoaded)
  46. END_MESSAGE_MAP()
  48. //---------------------------------------------------------------------------
  49. //
  50. // CMainFrame construction
  51. //
  52. //---------------------------------------------------------------------------
  53. CMainFrame::CMainFrame():
  54. m_TrayIcon(IDR_TRAYICON),
  55. m_bShutdown(FALSE),
  56. m_ApplicationScope( CApplicationScope::GetInstance() )
  57. {
  59. }
  61. //---------------------------------------------------------------------------
  62. //
  63. // CMainFrame destruction
  64. //
  65. //---------------------------------------------------------------------------
  66. CMainFrame::~CMainFrame()
  67. {
  68. m_ApplicationScope.InstallHook( FALSE, 0 );
  69. }
  71. //---------------------------------------------------------------------------
  72. //
  73. // OnCreate
  74. //
  75. //---------------------------------------------------------------------------
  76. int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
  77. {
  78. if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
  79. return -1;
  81. // Set up tray icon
  82. m_TrayIcon.SetNotificationWnd(this, WM_MY_TRAY_NOTIFICATION);
  83. m_TrayIcon.SetIcon(IDI_TRAYICON);
  85. //
  86. // Install the hook here
  87. //
  88. m_ApplicationScope.InstallHook( TRUE, m_hWnd );
  90. return 0;
  91. }
  93. //---------------------------------------------------------------------------
  94. //
  95. // OnClose
  96. //
  97. //---------------------------------------------------------------------------
  98. void CMainFrame::OnClose() 
  99. {
  100. if (m_bShutdown)
  101. {
  102. CFrameWnd::OnClose();
  103. }
  104. else
  105. {
  106. ShowWindow(SW_HIDE);
  107. }
  108. }
  110. //---------------------------------------------------------------------------
  111. // OnAppExit
  112. //
  113. // Shut down the process
  114. //---------------------------------------------------------------------------
  115. void CMainFrame::OnAppExit() 
  116. {
  117. m_bShutdown = TRUE;
  118. SendMessage(WM_CLOSE);
  119. }
  121. //---------------------------------------------------------------------------
  122. // OnTrayNotification
  123. //
  124. // Handle notification from tray icon: display a message.
  125. //---------------------------------------------------------------------------
  126. LRESULT CMainFrame::OnTrayNotification(WPARAM uID, LPARAM lEvent)
  127. {
  128. // let tray icon do default stuff
  129. return m_TrayIcon.OnTrayNotification(uID, lEvent);
  130. }
  132. //---------------------------------------------------------------------------
  133. // OnDllLoaded
  134. //
  135. // Fired when a process loads hook tool dll
  136. //---------------------------------------------------------------------------
  137. LRESULT CMainFrame::OnDllLoaded(WPARAM wParam, LPARAM lParam)
  138. {
  139. m_ApplicationScope.OnDllLoaded( lParam );
  140. return 0;
  141. }
  143. //---------------------------------------------------------------------------
  144. // OnDllUnLoaded
  145. //
  146. // Fired when a process unloads hook tool dll
  147. //---------------------------------------------------------------------------
  148. LRESULT CMainFrame::OnDllUnLoaded(WPARAM wParam, LPARAM lParam)
  149. {
  150. m_ApplicationScope.OnDllUnLoaded( lParam );
  151. return 0;
  152. }
  153. //----------------------------End of the file -------------------------------