nmidebug.c File Reference

#include <ntifs.h>
#include <ndk/ketypes.h>

Include dependency graph for nmidebug.c:

Go to the source code of this file.

Functions
FORCEINLINE VOID	NmiClearFlag (VOID)
BOOLEAN NTAPI	NmiDbgCallback (IN PVOID Context, IN BOOLEAN Handled)
NTSTATUS NTAPI	DriverEntry (IN PDRIVER_OBJECT DriverObject, IN PUNICODE_STRING RegistryPath)

Variables
PCHAR	NmiBegin = "NMI4NMI@"

Function Documentation

DriverEntry()

NTSTATUS NTAPI DriverEntry	(	IN PDRIVER_OBJECT	DriverObject,
		IN PUNICODE_STRING	RegistryPath
	)

Definition at line 57 of file nmidebug.c.

{
    PAGED_CODE();
 
    /* Register NMI callback */
    KeRegisterNmiCallback(&NmiDbgCallback, NULL);
 
    /* Return success */
    return STATUS_SUCCESS;
}

NmiClearFlag()

FORCEINLINE VOID NmiClearFlag

(

VOID

)

Definition at line 20 of file nmidebug.c.

{
    ((PCHAR)&KiBugCheckData[4])[0] -= (NmiBegin[3] | NmiBegin[7]);
    ((PCHAR)&KiBugCheckData[4])[3] |= 1;
#ifdef _M_IX86
#if defined(_MSC_VER) && !defined(__clang__)
    __asm
    {
        rcr KiBugCheckData[4], 8
    }
#else
    __asm__("rcrl %b[shift], %k[retval]" : [retval] "=rm" (KiBugCheckData[4]) : "[retval]" (KiBugCheckData[4]), [shift] "Nc" (8));
#endif
#endif
}

Referenced by NmiDbgCallback().

NmiDbgCallback()

BOOLEAN NTAPI NmiDbgCallback	(	IN PVOID	Context,
		IN BOOLEAN	Handled
	)

Definition at line 38 of file nmidebug.c.

{
    /* Clear the NMI flag */
    NmiClearFlag();
 
    /* Get NMI status signature */
    __indwordstring(0x80, (PULONG)NmiBegin, 1);
    ((void(*)())&KiBugCheckData[4])();
 
    /* Handle the NMI safely */
#ifdef _M_IX86
    KiEnableTimerWatchdog = (RtlCompareMemory(NmiBegin, NmiBegin + 4, 4) != 4);
#endif
    return TRUE;
}

Referenced by DriverEntry().

Variable Documentation

NmiBegin

PCHAR NmiBegin = "NMI4NMI@"

Definition at line 16 of file nmidebug.c.

Referenced by NmiClearFlag(), and NmiDbgCallback().