kmt_test_kernel.h File Reference

Go to the source code of this file.

Classes
struct	_POOL_HEADER

Typedefs
typedef struct _POOL_HEADER	POOL_HEADER
typedef struct _POOL_HEADER *	PPOOL_HEADER

Functions
VOID	KmtSetIrql (IN KIRQL NewIrql)
BOOLEAN	KmtAreInterruptsEnabled (VOID)
ULONG	KmtGetPoolTag (PVOID Memory)
USHORT	KmtGetPoolType (PVOID Memory)
PVOID	KmtGetSystemRoutineAddress (IN PCWSTR RoutineName)
PKTHREAD	KmtStartThread (IN PKSTART_ROUTINE StartRoutine, IN PVOID StartContext OPTIONAL)
VOID	KmtFinishThread (IN PKTHREAD Thread OPTIONAL, IN PKEVENT Event OPTIONAL)
INT __cdecl	KmtVSNPrintF (PSTR Buffer, SIZE_T BufferMaxLength, PCSTR Format, va_list Arguments) KMT_FORMAT(ms_printf

Variables
BOOLEAN	KmtIsCheckedBuild
BOOLEAN	KmtIsMultiProcessorBuild
BOOLEAN	KmtIsVirtualMachine
PCSTR	KmtMajorFunctionNames []

Typedef Documentation

POOL_HEADER

typedef struct _POOL_HEADER POOL_HEADER

PPOOL_HEADER

typedef struct _POOL_HEADER * PPOOL_HEADER

Function Documentation

KmtAreInterruptsEnabled()

BOOLEAN KmtAreInterruptsEnabled

(

VOID

)

Definition at line 59 of file kmt_test_kernel.h.

{
#if defined(_M_IX86) || defined(_M_AMD64)
    return (__readeflags() & EFLAGS_INTERRUPT_MASK) != 0;
#else
#pragma message(__FILE__ ": warning : 'KmtAreInterruptsEnabled()' is UNIMPLEMENTED for this architecture")
    // HACK: trivial stub. Used by ok_bool_true(KmtAreInterruptsEnabled(), ...).
    return FALSE;
#endif
}

Referenced by START_TEST().

KmtFinishThread()

VOID KmtFinishThread	(	IN PKTHREAD Thread	OPTIONAL,
		IN PKEVENT Event	OPTIONAL
	)

Definition at line 178 of file kmt_test_kernel.h.

{
    NTSTATUS Status;
 
    if (skip(Thread != NULL, "No thread\n"))
        return;
 
    if (Event)
        KeSetEvent(Event, IO_NO_INCREMENT, TRUE);
    Status = KeWaitForSingleObject(Thread,
                                   Executive,
                                   KernelMode,
                                   FALSE,
                                   NULL);
    ok_eq_hex(Status, STATUS_SUCCESS);
    ObDereferenceObject(Thread);
}

Referenced by FinishWorkerThread(), PerformTest(), START_TEST(), TestConnect(), TestEventScheduling(), and TestTdi().

KmtGetPoolTag()

ULONG KmtGetPoolTag

(

PVOID

Memory

)

Definition at line 108 of file kmt_test_kernel.h.

{
    PPOOL_HEADER Header;
 
    /* it's not so easy for allocations of PAGE_SIZE */
    if (((ULONG_PTR)Memory & (PAGE_SIZE - 1)) == 0)
        return 'TooL';
 
    Header = Memory;
    Header--;
 
    return Header->PoolTag;
}

Referenced by START_TEST(), TestDriverObject(), TestPoolTags(), and TestPrivateFunctions().

KmtGetPoolType()

USHORT KmtGetPoolType

(

PVOID

Memory

)

Definition at line 122 of file kmt_test_kernel.h.

{
    PPOOL_HEADER Header;
 
    /* it's not so easy for allocations of PAGE_SIZE */
    if (((ULONG_PTR)Memory & (PAGE_SIZE - 1)) == 0)
        return 0;
 
    Header = Memory;
    Header--;
 
    return Header->PoolType;
}

Referenced by TestDriverObject(), and TestPoolQuota().

KmtGetSystemRoutineAddress()

PVOID KmtGetSystemRoutineAddress

(

IN PCWSTR

RoutineName

)

Definition at line 136 of file kmt_test_kernel.h.

{
    UNICODE_STRING RoutineNameString;
    RtlInitUnicodeString(&RoutineNameString, (PWSTR)RoutineName);
    return MmGetSystemRoutineAddress(&RoutineNameString);
}

Referenced by _IRQL_requires_min_(), and START_TEST().

KmtSetIrql()

VOID KmtSetIrql

(

IN KIRQL

NewIrql

)

Definition at line 50 of file kmt_test_kernel.h.

{
    KIRQL Irql = KeGetCurrentIrql();
    if (Irql > NewIrql)
        KeLowerIrql(NewIrql);
    else if (Irql < NewIrql)
        KeRaiseIrql(NewIrql, &Irql);
}

Referenced by START_TEST(), TestEventFunctional(), TestFastMutex(), and TestSpinLock().

KmtStartThread()

PKTHREAD KmtStartThread	(	IN PKSTART_ROUTINE	StartRoutine,
		IN PVOID StartContext	OPTIONAL
	)

Definition at line 143 of file kmt_test_kernel.h.

{
    NTSTATUS Status;
    OBJECT_ATTRIBUTES ObjectAttributes;
    HANDLE ThreadHandle;
    PVOID ThreadObject = NULL;
 
    InitializeObjectAttributes(&ObjectAttributes,
                               NULL,
                               OBJ_KERNEL_HANDLE,
                               NULL,
                               NULL);
    ThreadHandle = INVALID_HANDLE_VALUE;
    Status = PsCreateSystemThread(&ThreadHandle,
                                  SYNCHRONIZE,
                                  &ObjectAttributes,
                                  NULL,
                                  NULL,
                                  StartRoutine,
                                  StartContext);
    ok_eq_hex(Status, STATUS_SUCCESS);
    if (!skip(NT_SUCCESS(Status) && ThreadHandle != NULL && ThreadHandle != INVALID_HANDLE_VALUE, "No thread\n"))
    {
        Status = ObReferenceObjectByHandle(ThreadHandle,
                                           SYNCHRONIZE,
                                           *PsThreadType,
                                           KernelMode,
                                           &ThreadObject,
                                           NULL);
        ok_eq_hex(Status, STATUS_SUCCESS);
        ObCloseHandle(ThreadHandle, KernelMode);
    }
    return ThreadObject;
}

Referenced by PerformTest(), START_TEST(), StartWorkerThread(), TestConnect(), TestEventScheduling(), and TestTdi().

KmtVSNPrintF()

INT __cdecl KmtVSNPrintF	(	PSTR	Buffer,
		SIZE_T	BufferMaxLength,
		PCSTR	Format,
		va_list	Arguments
	)

Variable Documentation

KmtIsCheckedBuild

BOOLEAN KmtIsCheckedBuild

Definition at line 15 of file kmt_test_kernel.h.

Referenced by _IRQL_requires_min_(), DriverEntry(), FsRtlIsDbcsInExpressionTest(), SimpleErrorChecks(), START_TEST(), TestAllInformation(), TestCreateSection(), TestEventFunctional(), TestFastMutex(), TestPhysicalMemorySection(), TestResourceSharedAccess(), TestSpinLock(), and TestUpcaseUnicodeString().

KmtIsMultiProcessorBuild

BOOLEAN KmtIsMultiProcessorBuild

Definition at line 16 of file kmt_test_kernel.h.

Referenced by DriverEntry(), START_TEST(), TestEventFunctional(), and TestSpinLock().

KmtIsVirtualMachine

BOOLEAN KmtIsVirtualMachine

Definition at line 17 of file kmt_test_kernel.h.

Referenced by TestStringIo().

KmtMajorFunctionNames

PCSTR KmtMajorFunctionNames[]

Definition at line 18 of file kmt_test_kernel.h.

Referenced by DriverDispatch(), TestDispatch(), and TestIrpHandler().