typedef NTSTATUS (FASTCALL *PMYIOCALLDRIVER)(IN PDEVICE_OBJECT,IN PIRP);

PMYIOCALLDRIVER g_Old_IoCallDriver=NULL;

NTSTATUS FASTCALL NewIoCallDriver(IN PDEVICE_OBJECT device,IN PIRP Irp)
{
	NTSTATUS status=STATUS_SUCCESS;
	
	
	if(!g_Old_IoCallDriver)
			return status;
	
	__asm
	{
		 jmp g_Old_IoCallDriver
	}
	
	return status;	
}

PMYIOCALLDRIVER HookRoutine(PMYIOCALLDRIVER ChangeIoCallDriver,BOOLEAN HookOrNot)
{
		PMYIOCALLDRIVER Old_Io_CallDriver=NULL;
		UNICODE_STRING FunName;
		PCHAR address;
		
		RtlInitUnicodeString(&FunName,L"IofCallDriver");
		
		address=MmGetSystemRoutineAddress(&FunName);
		
		if(address==NULL)
				return NULL;
				
		if(HookOrNot)
		{
			 Old_Io_CallDriver=(PMYIOCALLDRIVER)(address+2);
			 g_Old_IoCallDriver=Old_Io_CallDriver;
			 InterlockedExchange((PLONG)(address+2),(LONG)ChangeIoCallDriver);
			 return Old_Io_CallDriver;
		}
		else
		{
			 g_Old_IoCallDriver=NULL;
			 InterlockedExchange((PLONG)(address+2),(LONG)ChangeIoCallDriver);
		}	
		
		return Old_Io_CallDriver; 
}

void HookIoCallUnload(IN  PDRIVER_OBJECT  DriverObject)
{
	 if(g_Old_IoCallDriver)
	 {
	 		HookRoutine(g_Old_IoCallDriver,FALSE);	
   }
}

NTSTATUS DriverEntry(
    IN  PDRIVER_OBJECT  DriverObject,
    IN  PUNICODE_STRING RegistryPath
    )
{
		
		DriverObject->DriverUnload = HookIoCallUnload;
		
		g_Old_IoCallDriver=HookRoutine(NewIoCallDriver,TRUE);
		
    return STATUS_SUCCESS;
}

[招生]科锐逆向工程师培训(2024年11月15日实地，远程教学同时开班, 第51期)