reactos/drivers/base/nmidebug/nmidebug.c
Timo Kreuzer a7014019f3 [NMIDEBUG]
Fix compilation with MSVC

svn path=/branches/cmake-bringup/; revision=50712
2011-02-15 17:03:28 +00:00

65 lines
1.5 KiB
C

/*
* PROJECT: ReactOS NMI Debug Driver
* LICENSE: BSD - See COPYING.ARM in the top level directory
* FILE: drivers/base/nmidebug/nmidebug.c
* PURPOSE: Driver Code
* PROGRAMMERS: ReactOS Portable Systems Group
*/
/* INCLUDES *******************************************************************/
#include <ntifs.h>
#include <ntndk.h>
/* FUNCTIONS ******************************************************************/
PCHAR NmiBegin = "NMI4NMI@";
VOID
FORCEINLINE
NmiClearFlag(VOID)
{
((PCHAR)&KiBugCheckData[4])[0] -= (NmiBegin[3] | NmiBegin[7]);
((PCHAR)&KiBugCheckData[4])[3] |= 1;
#ifdef _MSC_VER
__asm
{
rcr KiBugCheckData[4], 8
}
#else
__asm__("rcrl %b[shift], %k[retval]" : [retval] "=rm" (KiBugCheckData[4]) : "[retval]" (KiBugCheckData[4]), [shift] "Nc" (8));
#endif
}
BOOLEAN
NTAPI
NmiDbgCallback(IN PVOID Context,
IN BOOLEAN Handled)
{
/* Clear the NMI flag */
NmiClearFlag();
/* Get NMI status signature */
__indwordstring(0x80, (PULONG)NmiBegin, 1);
((void(*)())&KiBugCheckData[4])();
/* Handle the NMI safely */
KiEnableTimerWatchdog = (RtlCompareMemory(NmiBegin, NmiBegin + 4, 4) != 4);
return TRUE;
}
NTSTATUS
NTAPI
DriverEntry(IN PDRIVER_OBJECT DriverObject,
IN PUNICODE_STRING RegistryPath)
{
PAGED_CODE();
/* Register NMI callback */
KeRegisterNmiCallback(&NmiDbgCallback, NULL);
/* Return success */
return STATUS_SUCCESS;
}
/* EOF */