windows10 记事本进程 键盘消息钩子 dll注入

#include <afx.h> //CString的头文件
#include "stdio.h"
#include "windows.h" //要调用的很多windows api函数的头文件

HHOOK g_hHook = NULL; //HHOOK是钩子句柄，如果想搭建钩子链，也可把下一个需要传给的钩子句柄放在这。

CString IsNumber(WPARAM wParam)
{
	CString message;
	switch (wParam) {
	case 0x30: message.Format("按了0"); break;
	case 0x31: message.Format("按了1"); break;
	case 0x32: message.Format("按了2"); break;
	case 0x33: message.Format("按了3"); break;
	case 0x34: message.Format("按了4"); break;
	case 0x35: message.Format("按了5"); break;
	case 0x36: message.Format("按了6"); break;
	case 0x37: message.Format("按了7"); break;
	case 0x38: message.Format("按了8"); break;
	case 0x39: message.Format("按了9"); break;
	default: message.Format("未定义的按键"); break;
	}
	return message;
}
//获取到的wparam是16位的int（也可能是long，这个无所谓），用于标识键盘截取到的消息是哪个键，我简单的
//识别了键盘上的数组键（不是小键盘的数组键），返回CString对象。

LRESULT CALLBACK KeyboardProc(int nCode, WPARAM wParam, LPARAM lParam)
//这是一个键盘钩子消息的回调函数，当设置钩子成功，dll被注入到目标线程，该回调函数会在每次有键盘消息
//传递给目标线程时被调用，第二个参数在这个类型的钩子中放回的是虚拟键盘的信息，其他两个参数我不太清楚
{
	MessageBox(NULL, IsNumber(wParam), _T("Message"), 0);
	return CallNextHookEx(g_hHook, nCode, wParam, lParam);
	//我理解这句代码意思是，如果g_hHook非空，就把消息传给这个句柄，否则就传给应用程序。
}

LIBRARY DLL

EXPORTS
KeyboardProc

#include "stdafx.h"
#include "windows.h"
#include "Psapi.h" //连接了库后引用头文件，EnumProcesses及GetModuleFileNameEx都需要引入这个头文件
#pragma comment(lib,"Psapi.lib") //预编译指令,连接psapi.lib库

DWORD FindProcessByEnumProcess(CString TargetProcessName)
//参数是目标程序名，如notepad.exe
//返回值类型DWORD，是32位的long型，值是找到的目标进程的进程id, 如果打开了多个同名程序，找到的是最后打开的那个进程的进程id
{
	DWORD TargetProcessId = 0; //目标进程初始值是0，没找到时就返回0
	DWORD ProcessesId[1024] = { 0 }; //进程id数组，在之后EnumProcesses函数调用会将当前所有进程id放入数组
	DWORD NeededProcessesId = 0; //在之后EnumProcesses函数调用后会将实际需要的进程数组的大小赋值给它
	LPSTR ProcessName = (LPSTR)malloc((sizeof(char)) * 1024);
	//LPSTR定义是typedef LPSTR char * ，LPSTR被定义成是一个指向以NULL(‘\0’)结尾的32位ANSI字符数组指针
	//用于存储返回到的进程名

	EnumProcesses(ProcessesId, sizeof(ProcessesId), &NeededProcessesId);
	//查询所有当前进程
	//第一个参数是输出参数，返回进程数组存储到ProcessesId[1024]中
	//第二个参数的输入参数，输入需要返回的进程数组的存储大小
	//第三个参数的输出参数，返回实际需要的进程数组的存储大小
	DWORD ProcessNumber = NeededProcessesId / sizeof(DWORD);
	//得到进程个数，用于遍历

	for (unsigned int i = 0; i < ProcessNumber; i++)
	{
		if (ProcessesId[i] != 0)
		{
			//对每个进程id执行下面操作

			HANDLE hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, ProcessesId[i]);
			//HANDLE是进程句柄类型，hProcess存储的就是进程句柄了
			//OpenProcess函数通过进程id获取进程句柄
			//第一个参数是输入参数，标识需要获取的权限，这里我们获取PROCESS_QUERY_INFORMATION和PROCESS_VM_READ权限
			//因为下面的GetModuleFileNameEx函数指定要这两个权限
			//第二个参数是输入参数，用来标识该句柄是否希望被子进程继承，不过不考虑子进程的继承权限则直接赋值为FALSE
			//第三个参数是输入参数，输入需要打开进程的进程id
			//返回值就是得到的句柄了
			if (hProcess != NULL)
			{
				GetModuleFileNameEx(hProcess, NULL, ProcessName, 1024);
				//根据进程句柄获取到进程完整的名称,如C:\Windows\System32\notepad.exe
				//第一个参数是输入参数，输入需要获取进程名的进程句柄
				//第二个参数是输入参数，输入需要获取的模块的模块句柄，为NULL表示获取进程主模块
				//第三个参数是输出参数，输出进程模块完整的名称
				//第四个参数是输入参数，表明ProcessName的存储大小
				CString ProcessFullPathName = (CString)ProcessName;
				//把LPSTR类型转为CString类型，便于进行字符处理
				//CString在ANSI字符集下以存储char数组，在Unicode字符集下以存储wchar_t数组,后者的长度是前者的两倍
				//CString a，则a可作为指向存储的char数组的头部的指针，和LPSTR类型是一样的，所以我用了强转
				//我百度到的转化方法是这么写的：CString ProcessFullPathName(ProcessName); 也可以
				CString ProcessBaseName = ProcessFullPathName.Right(ProcessFullPathName.GetLength() - ProcessFullPathName.ReverseFind('\\') - 1);
				//把路径去掉，留下一个基础名称及C:\Windows\System32\notepad.exe转为notepad.exe

				if (ProcessBaseName == TargetProcessName)
				//如果该进程名与目标进程名相同，那么该进程id就是目标进程id
				{
					TargetProcessId = ProcessesId[i];
				}
				CloseHandle(hProcess);//关闭句柄
				hProcess = NULL;
			}
		}
	}

	return TargetProcessId;
}

void DoInject(DWORD TargetWindowThreadId)
{
	HMODULE hDll = LoadLibrary(_T("DLL.dll"));
	//HMODULE是模块句柄类型
	//LoadLibrary可以显示加载dll
	//这里我没有加路径，所有执行前要将dll放到exe文件同目录下
	if (hDll == NULL) {
		printf("将dll加载到自身进程失败\n");
		exit(0);
	}
	else {
		printf("将dll加载到自身进程成功\n");
	}

	FARPROC KeyboardProc = (FARPROC)GetProcAddress(hDll, "KeyboardProc");
	//通过GetProcAddress函数获取到hDll句柄中的KeyboardProc函数的地址
	if (KeyboardProc == NULL) {
		printf("获取到回调函数地址失败\n");
		exit(0);
	}
	else {
		printf("获取到回调函数地址成功\n");
	}

	HHOOK g_hHook = SetWindowsHookEx(WH_KEYBOARD, (HOOKPROC)KeyboardProc, hDll, TargetWindowThreadId);
	//将dll注入目标线程，设置函数指针指向写好的键盘消息回调函数
	//第一个参数输入钩子类型
	//第二个参数根据不同的钩子类型，要输入不同类型的回调函数地址
	//第三个参数输入dll句柄
	//第四个参数输入目标线程id

	if (g_hHook) {
		printf("向目标线程添加钩子并注入dll成功\n");
	}

	printf("输入q卸载钩子：");
	while (getchar() != 'q');

	if (g_hHook)
	{
		UnhookWindowsHookEx(g_hHook);
		g_hHook = NULL;
	}
	//卸载钩子

	FreeLibrary(hDll);//释放dll
}

BOOL CALLBACK EnumWindowsProc(HWND hwnd, LPARAM lParam)
//EnumWindows设置的回调函数，系统每发现一个窗口都会调用该回调函数
//HWND是窗口句柄类型
//第一个参数返回的是当前窗口句柄，第二个参数类型可以自己定，我传入的是目标进程id
{
	DWORD CurrentWindowProcessId; //当前窗口进程id
	DWORD CurrentWindowThreadId; //当前窗口线程id

	CurrentWindowThreadId = GetWindowThreadProcessId(hwnd, &CurrentWindowProcessId);
	//GetWindowThreadProcessId()可以通过窗口句柄，获取该窗口的所在的进程及线程
	//第一个参数是输入参数，输入目标窗口句柄
	//第二个参数是输出参数，类型是LPDWORD，及指向DWORD的指针，所以要取地址，函数执行成功后CurrentWindowProcessId值就是返回的当前窗口进程id
	//返回值是值传递的，DWORD类型，直接赋值给DWORD类型就行了，值是当前窗口线程id
	if (CurrentWindowProcessId == lParam) {
		//如果当前窗口进程id等于目标进程的进程id
		//则得到的当前窗口线程id就是目标窗口线程id
		DoInject(CurrentWindowThreadId);
		//得到了线程id后就可以注入了
		return false; //当找到后就返回false，这样才会终止遍历
	}

	return true; //不是当前窗口，返回true，继续遍历
}

int main()
{
	CString TargetProcessName(_T("notepad.exe"));
	DWORD TargetProcessId = FindProcessByEnumProcess(TargetProcessName); //先找到目标进程id
	EnumWindows(EnumWindowsProc, TargetProcessId); //根据目标进程id进行遍历，找到目标线程，并注入dll
}