您的位置：首页 > 其它

第九章 HOOK程序测试 windows程序设计王艳平版

2012-08-11 09:45 323 查看

#include <afxwin.h>

class CMyApp : public CWinApp

{

public:

BOOL InitInstance();

};

class CMainDialog : public CDialog

{

public:

CMainDialog(CWnd* pParentWnd = NULL);

protected:

virtual BOOL OnInitDialog();

afx_msg BOOL OnCopyData(CWnd* pWnd, COPYDATASTRUCT *pCopyDataStruct);

DECLARE_MESSAGE_MAP()

};

////////////////////////////////////////////////

// HookTermProApp.cpp文件

#include "resource.h"

#include "HookTermProApp.h"

CMyApp theApp;

// DLL导出函数

BOOL WINAPI SetSysHook(BOOL bInstall, DWORD dwThreadId = 0)

{

typedef (WINAPI *PFNSETSYSHOOK)(BOOL, DWORD);

// 调试的时候可以这样设置szDll[] = "..//09HookTermProLib//debug//09HookTermProLib.dll";

char szDll[] = "09HookTermProLib.dll";

// 加载09HookTermProLib.dll模块

BOOL bNeedFree = FALSE;

HMODULE hModule = ::GetModuleHandle(szDll);

if(hModule == NULL)

{

hModule = ::LoadLibrary(szDll);

bNeedFree = TRUE;

}

// 获取SetSysHook函数的地址

PFNSETSYSHOOK mSetSysHook = (PFNSETSYSHOOK)::GetProcAddress(hModule, "SetSysHook");

if(mSetSysHook == NULL) // 文件不正确?

{

if(bNeedFree)

::FreeLibrary(hModule);

return FALSE;

}

// 调用SetSysHook函数

BOOL bRet = mSetSysHook(bInstall, dwThreadId);

// 如果有必要，释放上面加载的模块

if(bNeedFree)

::FreeLibrary(hModule);

return bRet;

}

BOOL CMyApp::InitInstance()

{

// 安装钩子

if(!SetSysHook(TRUE, 0))

::MessageBox(NULL, "安装钩子出错！", "09HookTermProApp", 0);

// 显示对话框

CMainDialog dlg;

m_pMainWnd = &dlg;

dlg.DoModal();

// 卸载钩子

SetSysHook(FALSE);

return FALSE;

}

CMainDialog::CMainDialog(CWnd* pParentWnd):CDialog(IDD_MAINDLG, pParentWnd)

{

}

BEGIN_MESSAGE_MAP(CMainDialog, CDialog)

ON_WM_COPYDATA()

END_MESSAGE_MAP()

BOOL CMainDialog::OnInitDialog()

{

CDialog::OnInitDialog();

SetIcon(theApp.LoadIcon(IDI_MAIN), FALSE);

::SetWindowPos(m_hWnd, HWND_TOPMOST, 0, 0,

0, 0, SWP_NOSIZE|SWP_NOREDRAW|SWP_NOMOVE);

return TRUE;

}

BOOL CMainDialog::OnCopyData(CWnd* pWnd, COPYDATASTRUCT* pCopyDataStruct)

{

GetDlgItem(IDC_HOOKINFO)->SetWindowText((char*)pCopyDataStruct->lpData);

// 检查是否禁止执行

BOOL bForbid = ((CButton*)GetDlgItem(IDC_FORBIDEXE))->GetCheck();

if(bForbid)

return -1;

return TRUE;

}

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签：

相关文章推荐

新的分享

章节导航

第九章 HOOK程序测试 windows程序设计 王艳平版

第九章 HOOK程序测试 windows程序设计王艳平版