kernel32_drv.c File Reference

#include <kmt_test.h>
#include <debug.h>
#include "kernel32_test.h"

Include dependency graph for kernel32_drv.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
NTSTATUS	TestEntry (IN PDRIVER_OBJECT DriverObject, IN PCUNICODE_STRING RegistryPath, OUT PCWSTR *DeviceName, IN OUT INT *Flags)
VOID	TestUnload (IN PDRIVER_OBJECT DriverObject)
static NTSTATUS	TestMessageHandler (IN PDEVICE_OBJECT DeviceObject, IN ULONG ControlCode, IN PVOID Buffer OPTIONAL, IN SIZE_T InLength, IN OUT PSIZE_T OutLength)
static NTSTATUS	TestDirectoryControl (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PIO_STACK_LOCATION IoStackLocation)
static NTSTATUS	TestQueryInformation (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PIO_STACK_LOCATION IoStackLocation)
static NTSTATUS	TestSetInformation (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PIO_STACK_LOCATION IoStackLocation)

Variables
static KMT_MESSAGE_HANDLER	TestMessageHandler
static KMT_IRP_HANDLER	TestDirectoryControl
static KMT_IRP_HANDLER	TestQueryInformation
static KMT_IRP_HANDLER	TestSetInformation
static UNICODE_STRING	ExpectedExpression = RTL_CONSTANT_STRING(L"<not set>")
static WCHAR	ExpressionBuffer [MAX_PATH]
static BOOLEAN	ExpectingSetAttributes = FALSE
static ULONG	ExpectedSetAttributes = -1
static BOOLEAN	ExpectingQueryAttributes = FALSE
static ULONG	ReturnQueryAttributes = -1

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 10 of file kernel32_drv.c.

Function Documentation

TestDirectoryControl()

static NTSTATUS TestDirectoryControl ( IN PDEVICE_OBJECT  DeviceObject, IN PIRP  Irp, IN PIO_STACK_LOCATION  IoStackLocation  )

static

Definition at line 121 of file kernel32_drv.c.

{
    NTSTATUS Status = STATUS_NOT_SUPPORTED;
 
    PAGED_CODE();
 
    DPRINT("IRP %x/%x\n", IoStackLocation->MajorFunction, IoStackLocation->MinorFunction);
    ASSERT(IoStackLocation->MajorFunction == IRP_MJ_DIRECTORY_CONTROL);
 
    ok(IoStackLocation->MinorFunction == IRP_MN_QUERY_DIRECTORY, "Minor function: %u\n", IoStackLocation->MinorFunction);
    if (IoStackLocation->MinorFunction == IRP_MN_QUERY_DIRECTORY)
    {
        ok(IoStackLocation->Parameters.QueryDirectory.FileInformationClass == FileBothDirectoryInformation,
           "FileInformationClass: %d\n", IoStackLocation->Parameters.QueryDirectory.FileInformationClass);
        if (IoStackLocation->Parameters.QueryDirectory.FileInformationClass == FileBothDirectoryInformation)
        {
            ok(RtlEqualUnicodeString(IoStackLocation->Parameters.QueryDirectory.FileName, &ExpectedExpression, FALSE),
               "Expression is '%wZ', expected '%wZ'\n", IoStackLocation->Parameters.QueryDirectory.FileName, &ExpectedExpression);
            RtlZeroMemory(Irp->UserBuffer, IoStackLocation->Parameters.QueryDirectory.Length);
            Status = STATUS_SUCCESS;
        }
    }
 
    Irp->IoStatus.Status = Status;
    Irp->IoStatus.Information = 0;
 
    IoCompleteRequest(Irp, IO_NO_INCREMENT);
 
    return Status;
}

TestEntry()

NTSTATUS TestEntry	(	IN PDRIVER_OBJECT	DriverObject,
		IN PCUNICODE_STRING	RegistryPath,
		OUT PCWSTR *	DeviceName,
		IN OUT INT *	Flags
	)

Definition at line 28 of file kernel32_drv.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
 
    PAGED_CODE();
 
    UNREFERENCED_PARAMETER(RegistryPath);
 
    *DeviceName = L"kernel32";
    *Flags = TESTENTRY_NO_EXCLUSIVE_DEVICE;
 
    KmtRegisterIrpHandler(IRP_MJ_DIRECTORY_CONTROL, NULL, TestDirectoryControl);
    KmtRegisterIrpHandler(IRP_MJ_QUERY_INFORMATION, NULL, TestQueryInformation);
    KmtRegisterIrpHandler(IRP_MJ_SET_INFORMATION, NULL, TestSetInformation);
    KmtRegisterMessageHandler(0, NULL, TestMessageHandler);
 
    return Status;
}

TestMessageHandler()

static NTSTATUS TestMessageHandler ( IN PDEVICE_OBJECT  DeviceObject, IN ULONG  ControlCode, IN PVOID Buffer  OPTIONAL, IN SIZE_T  InLength, IN OUT PSIZE_T  OutLength  )

static

Definition at line 60 of file kernel32_drv.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
 
    PAGED_CODE();
 
    switch (ControlCode)
    {
        case IOCTL_EXPECT_EXPRESSION:
        {
            C_ASSERT(sizeof(ExpressionBuffer) <= UNICODE_STRING_MAX_BYTES);
            DPRINT("IOCTL_EXPECT_EXPRESSION, InLength = %lu\n", InLength);
            if (InLength > sizeof(ExpressionBuffer))
                return STATUS_BUFFER_OVERFLOW;
 
            if (InLength % sizeof(WCHAR) != 0)
                return STATUS_INVALID_PARAMETER;
 
            RtlInitEmptyUnicodeString(&ExpectedExpression, ExpressionBuffer, sizeof(ExpressionBuffer));
            RtlCopyMemory(ExpressionBuffer, Buffer, InLength);
            ExpectedExpression.Length = (USHORT)InLength;
            DPRINT("IOCTL_EXPECT_EXPRESSION: %wZ\n", &ExpectedExpression);
 
            break;
        }
        case IOCTL_RETURN_QUERY_ATTRIBUTES:
        {
            DPRINT("IOCTL_RETURN_QUERY_ATTRIBUTES, InLength = %lu\n", InLength);
            if (InLength != sizeof(ULONG))
                return STATUS_INVALID_PARAMETER;
 
            ReturnQueryAttributes = *(PULONG)Buffer;
            ExpectingQueryAttributes = TRUE;
            DPRINT("IOCTL_RETURN_QUERY_ATTRIBUTES: %lu\n", ReturnQueryAttributes);
            break;
        }
        case IOCTL_EXPECT_SET_ATTRIBUTES:
        {
            DPRINT("IOCTL_EXPECT_SET_ATTRIBUTES, InLength = %lu\n", InLength);
            if (InLength != sizeof(ULONG))
                return STATUS_INVALID_PARAMETER;
 
            ExpectedSetAttributes = *(PULONG)Buffer;
            ExpectingSetAttributes = TRUE;
            DPRINT("IOCTL_EXPECT_SET_ATTRIBUTES: %lu\n", ExpectedSetAttributes);
            break;
        }
        default:
            return STATUS_NOT_SUPPORTED;
    }
 
    return Status;
}

TestQueryInformation()

static NTSTATUS TestQueryInformation ( IN PDEVICE_OBJECT  DeviceObject, IN PIRP  Irp, IN PIO_STACK_LOCATION  IoStackLocation  )

static

Definition at line 157 of file kernel32_drv.c.

{
    NTSTATUS Status = STATUS_NOT_SUPPORTED;
    PFILE_BASIC_INFORMATION BasicInfo;
 
    PAGED_CODE();
 
    DPRINT("IRP %x/%x\n", IoStackLocation->MajorFunction, IoStackLocation->MinorFunction);
    ASSERT(IoStackLocation->MajorFunction == IRP_MJ_QUERY_INFORMATION);
 
    Irp->IoStatus.Information = 0;
 
    ok_eq_ulong(IoStackLocation->Parameters.QueryFile.FileInformationClass, FileBasicInformation);
    if (IoStackLocation->Parameters.QueryFile.FileInformationClass == FileBasicInformation)
    {
        ok(ExpectingQueryAttributes, "Unexpected QUERY_INFORMATION call\n");
        BasicInfo = Irp->AssociatedIrp.SystemBuffer;
        BasicInfo->CreationTime.QuadPart = 126011664000000000;
        BasicInfo->LastAccessTime.QuadPart = 130899112800000000;
        BasicInfo->LastWriteTime.QuadPart = 130899112800000000;
        BasicInfo->ChangeTime.QuadPart = 130899112800000000;
        BasicInfo->FileAttributes = ReturnQueryAttributes;
        ReturnQueryAttributes = -1;
        ExpectingQueryAttributes = FALSE;
        Status = STATUS_SUCCESS;
        Irp->IoStatus.Information = sizeof(*BasicInfo);
    }
 
    Irp->IoStatus.Status = Status;
 
    IoCompleteRequest(Irp, IO_NO_INCREMENT);
 
    return Status;
}

TestSetInformation()

static NTSTATUS TestSetInformation ( IN PDEVICE_OBJECT  DeviceObject, IN PIRP  Irp, IN PIO_STACK_LOCATION  IoStackLocation  )

static

Definition at line 197 of file kernel32_drv.c.

{
    NTSTATUS Status = STATUS_NOT_SUPPORTED;
    PFILE_BASIC_INFORMATION BasicInfo;
 
    PAGED_CODE();
 
    DPRINT("IRP %x/%x\n", IoStackLocation->MajorFunction, IoStackLocation->MinorFunction);
    ASSERT(IoStackLocation->MajorFunction == IRP_MJ_SET_INFORMATION);
 
    ok_eq_ulong(IoStackLocation->Parameters.SetFile.FileInformationClass, FileBasicInformation);
    if (IoStackLocation->Parameters.SetFile.FileInformationClass == FileBasicInformation)
    {
        ok(ExpectingSetAttributes, "Unexpected SET_INFORMATION call\n");
        BasicInfo = Irp->AssociatedIrp.SystemBuffer;
        ok_eq_longlong(BasicInfo->CreationTime.QuadPart, 0LL);
        ok_eq_longlong(BasicInfo->LastAccessTime.QuadPart, 0LL);
        ok_eq_longlong(BasicInfo->LastWriteTime.QuadPart, 0LL);
        ok_eq_longlong(BasicInfo->ChangeTime.QuadPart, 0LL);
        ok_eq_ulong(BasicInfo->FileAttributes, ExpectedSetAttributes);
        ExpectedSetAttributes = -1;
        ExpectingSetAttributes = FALSE;
        Status = STATUS_SUCCESS;
    }
 
    Irp->IoStatus.Status = Status;
    Irp->IoStatus.Information = 0;
 
    IoCompleteRequest(Irp, IO_NO_INCREMENT);
 
    return Status;
}

TestUnload()

VOID TestUnload

(

IN PDRIVER_OBJECT

DriverObject

)

Definition at line 52 of file kernel32_drv.c.

{
    PAGED_CODE();
}

Variable Documentation

ExpectedExpression

UNICODE_STRING ExpectedExpression = RTL_CONSTANT_STRING(L"<not set>")

static

Definition at line 20 of file kernel32_drv.c.

Referenced by START_TEST(), TestDirectoryControl(), and TestMessageHandler().

ExpectedSetAttributes

ULONG ExpectedSetAttributes = -1

static

Definition at line 23 of file kernel32_drv.c.

Referenced by TestMessageHandler(), and TestSetInformation().

ExpectingQueryAttributes

BOOLEAN ExpectingQueryAttributes = FALSE

static

Definition at line 24 of file kernel32_drv.c.

Referenced by TestMessageHandler(), and TestQueryInformation().

ExpectingSetAttributes

BOOLEAN ExpectingSetAttributes = FALSE

static

Definition at line 22 of file kernel32_drv.c.

Referenced by TestMessageHandler(), and TestSetInformation().

ExpressionBuffer

WCHAR ExpressionBuffer[MAX_PATH]

static

Definition at line 21 of file kernel32_drv.c.

Referenced by START_TEST(), and TestMessageHandler().

ReturnQueryAttributes

ULONG ReturnQueryAttributes = -1

static

Definition at line 25 of file kernel32_drv.c.

Referenced by TestMessageHandler(), and TestQueryInformation().

TestDirectoryControl

KMT_IRP_HANDLER TestDirectoryControl

static

Definition at line 16 of file kernel32_drv.c.

Referenced by TestEntry().

TestMessageHandler

KMT_MESSAGE_HANDLER TestMessageHandler

static

Definition at line 15 of file kernel32_drv.c.

Referenced by TestEntry().

TestQueryInformation

KMT_IRP_HANDLER TestQueryInformation

static

Definition at line 17 of file kernel32_drv.c.

Referenced by TestEntry().

TestSetInformation

KMT_IRP_HANDLER TestSetInformation

static

Definition at line 18 of file kernel32_drv.c.

Referenced by TestEntry().