This commit is contained in:
lhodges1 2023-08-19 20:58:07 +10:00
parent 4321236a0f
commit 4684dd2158

View file

@ -10,7 +10,7 @@ NTSTATUS ValidateDriverIOCTLDispatchRegion(
) )
{ {
NTSTATUS status; NTSTATUS status;
UINT64 current_function; UINT64 dispatch_function;
UINT64 base = ( UINT64 )Driver->DriverStart; UINT64 base = ( UINT64 )Driver->DriverStart;
UINT64 end = base + Driver->DriverSize; UINT64 end = base + Driver->DriverSize;
@ -22,26 +22,21 @@ NTSTATUS ValidateDriverIOCTLDispatchRegion(
* the module, report it. Basic check but every effective for catching driver * the module, report it. Basic check but every effective for catching driver
* dispatch hooking. * dispatch hooking.
*/ */
dispatch_function = Driver->MajorFunction[ IRP_MJ_DEVICE_CONTROL ];
for ( INT index = 0; index < IRP_MJ_MAXIMUM_FUNCTION + 1; index++ ) if ( dispatch_function == NULL )
return;
DEBUG_LOG( "Current function: %llx", dispatch_function );
if ( dispatch_function >= base && dispatch_function <= end )
{ {
current_function = *(UINT64*) DEBUG_LOG( "THIS ADDRESS IS INSIDE ITS REGIUON :)" );
( ( UINT64 )Driver->MajorFunction + index * sizeof( PVOID ) ); return;
DEBUG_LOG( "Current function: %llx", current_function );
if ( current_function == NULL )
continue;
if ( current_function >= base && current_function <= end )
{
DEBUG_LOG( "THIS ADDRESS IS INSIDE ITS REGIUON :)" );
continue;
}
DEBUG_ERROR( "Driver with invalid IOCTL dispatch routine found" );
*Flag = FALSE;
} }
DEBUG_ERROR( "Driver with invalid IOCTL dispatch routine found" );
*Flag = FALSE;
} }
VOID InitDriverList( VOID InitDriverList(