调试器模拟程序（持续更新）

前言

目前已经完结了Windows内核系列的入门学习，在调试篇章中，每一部分都涉及到一个入门调试器的实验。每一步都是息息相关的，如果放在一起就有太多重复代码了，因此在介绍不同断点以及单步操作时，只是单独列出相应的代码，本篇则将那些代码整合起来，模拟一个简单的调试器程序。

调试器代码

目前仅软件断点功能可用，硬件断点及单步异常有一些问题，内存断点与单步步过功能暂未实现

现在的问题主要是线程发生改变，原先的设置的TF位无法形成单步异常。硬件断点也是如此。

c

// Debugger_OpenProcess.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include "windows.h"

//宏定义与全局变量
#define dbgProcessName "C:\\notepad.exe"
#define SystemInt3 0x7C92120E		//系统断点，根据环境而改变
typedef HANDLE (WINAPI *FnOpenThread) (DWORD dwDesiredAccess,BOOL  bInheritHandle,DWORD dwThreadId);
HANDLE hDebugeeThread;
HANDLE hDebugeeProcess;
BYTE OriginalCode;		//恢复INT3断点时用到
CREATE_PROCESS_DEBUG_INFO processInfo;

//异常分发函数
BOOL ExceptionHandler(DEBUG_EVENT* pDebugEvent);

//异常处理函数
BOOL Int3ExceptionProc(EXCEPTION_DEBUG_INFO* pExceptionInfo);
BOOL SingleStepExceptionProc(EXCEPTION_DEBUG_INFO* pExceptionInfo);

//断点设置函数
VOID SetInt3BreakPoint();
VOID SetHardBreakPoint(PVOID pAddress);
VOID SetSingleStep();

//工具函数
BOOL WaitForUserCommand();
BOOL IsSystemInt3(EXCEPTION_DEBUG_INFO* pExceptionInfo);


int main()
{

	BOOL nIsConinue = TRUE;
	DEBUG_EVENT debugEvent = {0};
	BOOL bRet = TRUE;
	DWORD dwContinue = DBG_CONTINUE;
	
	//1.创建调试进程;
	PROCESS_INFORMATION pInfo = {0};
	STARTUPINFO startupInfo = {0};
	GetStartupInfo(&startupInfo);
	
	//以Create方式创建调试链接
	bRet = CreateProcess(dbgProcessName, NULL, NULL, NULL, TRUE, DEBUG_PROCESS||DEBUG_ONLY_THIS_PROCESS, NULL, NULL, &startupInfo, &pInfo);
	if (bRet == FALSE)
	{
		printf("CreateProcess error:%d\n", GetLastError());
		getchar();
		return 0;
	} 

	//2.调试循环
	while (nIsConinue)
	{
		//取DEBUG_EVENT
		bRet = WaitForDebugEvent(&debugEvent, INFINITE);
		if (!bRet)
		{
			printf("WaitForDebugEvent error:%d\n", GetLastError());
			return 0;
		}

		switch (debugEvent.dwDebugEventCode)
		{
			case EXCEPTION_DEBUG_EVENT:
				// printf("异常：发生异常的地址：%X \n",debugEvent.u.Exception.ExceptionRecord.ExceptionAddress);
				// printf("发生异常调试事件\n");
				bRet = ExceptionHandler(&debugEvent);
				
				if(!bRet)
					dwContinue = DBG_EXCEPTION_NOT_HANDLED;
				break;

			case CREATE_THREAD_DEBUG_EVENT:
				// printf("创建线程调试事件\n");
				break;

			case  CREATE_PROCESS_DEBUG_EVENT:
				// printf("创建进程调试事件\n");

				processInfo = debugEvent.u.CreateProcessInfo;
				hDebugeeProcess = processInfo.hProcess;
				//OEP处下INT3断点
				SetInt3BreakPoint();

				break;

			case EXIT_THREAD_DEBUG_EVENT:
				// printf("退出线程调试事件\n");
				break;

			case EXIT_PROCESS_DEBUG_EVENT:
				// printf("退出进程调试事件\n");
				break;

			case LOAD_DLL_DEBUG_EVENT:
				// printf("加载DLL调试事件\n");
				break;

			case UNLOAD_DLL_DEBUG_EVENT:
				// printf("卸载DLL调试事件\n");
				break;

			default:
				break;
		}

		//DBG_CONTINUE 表示调试器已处理该异常
		//DBG_EXCEPTION_NOT_HANDLED 表示调试器没有处理该异常，转回到用户态中执行，寻找可以处理该异常的异常处理器
		//ContinueDebugEvent 告诉被调试程序让被调试程序继续执行
		bRet = ContinueDebugEvent(debugEvent.dwProcessId, debugEvent.dwThreadId, DBG_CONTINUE);
	}

	return 0;
}


BOOL ExceptionHandler(DEBUG_EVENT* pDebugEvent)
{
	BOOL bRet = TRUE;
	EXCEPTION_DEBUG_INFO* pExceptionInfo = NULL;

	//获取异常事件结构体
	pExceptionInfo = &pDebugEvent->u.Exception;

	//通过线程ID获取线程句柄
	FnOpenThread MyOpenThread = (FnOpenThread)GetProcAddress(LoadLibrary("Kernel32.dll"), "OpenThread");
	hDebugeeThread = MyOpenThread(THREAD_ALL_ACCESS, FALSE, pDebugEvent->dwThreadId);

	switch(pExceptionInfo->ExceptionRecord.ExceptionCode)
	{	
		//INT3异常
		case EXCEPTION_BREAKPOINT:
			bRet = Int3ExceptionProc(pExceptionInfo);
			break;
		
		//访问异常
		case EXCEPTION_ACCESS_VIOLATION:
			break;

		//单步异常
		case EXCEPTION_SINGLE_STEP:
			bRet = SingleStepExceptionProc(pE4xceptionInfo);
			break;
	}

	return bRet;
}


VOID SetInt3BreakPoint()
{
	BYTE INT3 = 0xCC;
	ReadProcessMemory(hDebugeeProcess, processInfo.lpStartAddress, &OriginalCode, 1, NULL);
	WriteProcessMemory(hDebugeeProcess, processInfo.lpStartAddress, &INT3, 1, NULL);
}


BOOL Int3ExceptionProc(EXCEPTION_DEBUG_INFO* pExceptionInfo)
{
	BOOL bRet = FALSE;
	CONTEXT Context;

	//1.将INT3修复为原来的数据(如果是系统断点，不用修复)
	if(IsSystemInt3(pExceptionInfo))
		return TRUE;
	else
		WriteProcessMemory(hDebugeeProcess, pExceptionInfo->ExceptionRecord.ExceptionAddress, &OriginalCode, 1, NULL);

	//2.显示断点位置
	printf("int 3断点: 0x%p \n",pExceptionInfo->ExceptionRecord.ExceptionAddress);

	//3.获取线程上下文
	Context.ContextFlags = CONTEXT_FULL | CONTEXT_DEBUG_REGISTERS;
	GetThreadContext(hDebugeeThread,&Context);

	//4.修复EIP
	Context.Eip--;
	SetThreadContext(hDebugeeThread, &Context);

	//5.显示反汇编


	//这里设置硬件断点
	//SetHardBreakPoint((PVOID)((DWORD)pExceptionInfo->ExceptionRecord.ExceptionAddress+1));


	//6.等待用户命令
	while(bRet == FALSE)
	{
		bRet = WaitForUserCommand();
	}
	return bRet;
}


BOOL IsSystemInt3(EXCEPTION_DEBUG_INFO* pExceptionInfo)
{
	return (pExceptionInfo->ExceptionRecord.ExceptionAddress == (PVOID)SystemInt3);
}


VOID SetHardBreakPoint(PVOID pAddress)
{
	CONTEXT Context;
	//1.获取线程上下文
	Context.ContextFlags = CONTEXT_FULL | CONTEXT_DEBUG_REGISTERS;
	GetThreadContext(hDebugeeThread, &Context);

	//2.设置断点位置
	Context.Dr0 = (DWORD)pAddress;
	Context.Dr6 |= 1;
	Context.Dr7 |= 1;

	//3.设置断点长度
	Context.Dr7 &= 0xfff0ffff;

	//4.设置线程上下文
	SetThreadContext(hDebugeeThread, &Context);
	ResumeThread(hDebugeeThread);
}


VOID SetSingleStep()
{
	CONTEXT Context;
	Context.ContextFlags = CONTEXT_FULL || CONTEXT_DEBUG_REGISTERS;
	GetThreadContext(hDebugeeThread, &Context);
	Context.EFlags |= 0x100;
	SetThreadContext(hDebugeeThread, &Context);
}


BOOL SingleStepExceptionProc(EXCEPTION_DEBUG_INFO* pExceptionInfo)
{
	CONTEXT Context;
	BOOL bRet = FALSE;

	//1.获取线程上下文
	Context.ContextFlags = CONTEXT_FULL || CONTEXT_DEBUG_REGISTERS;
	GetThreadContext(hDebugeeThread, &Context);

	//2.判断是什么导致的单步异常
	if(Context.Dr6 & 0xF){ //硬件断点导致的单步异常
		//显示断点信息
		printf("硬件断点 %d 0x%p\n",Context.Dr7 & 0x00030000,Context.Dr0);

		//去除断点
		Context.Dr0 = 0;
		Context.Dr7 &= 0xfffffffe;
	}else{
		//显示断点信息
		printf("单步异常 0x%p\n",Context.Eip);

		//去除单步
		Context.EFlags &= 0xfffffeff;
	}
	SetThreadContext(hDebugeeThread, &Context);

	while(bRet == FALSE)
	{
		bRet = WaitForUserCommand();
	}

	return bRet;
}


BOOL WaitForUserCommand()
{
	BOOL bRet = FALSE;
	char CMD;

	printf("COMMAND>");
	
	CMD = getchar();
	fflush(stdin);

	switch(CMD)
	{
		//单步步入
		case't':
			bRet = TRUE;
			SetSingleStep();
			break;

		case'g':
			bRet = TRUE;
			break;
	}
	return bRet;
}

参考资料

参考笔记：

• 张嘉杰笔记

参考教程：

• 海哥逆向中级预习班教程