IoCreateFile_drv.c File Reference

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

Include dependency graph for IoCreateFile_drv.c:

Go to the source code of this file.

Classes
struct	_TEST_FCB

Macros
#define	NDEBUG

Typedefs
typedef struct _TEST_FCB	TEST_FCB
typedef struct _TEST_FCB *	PTEST_FCB

Functions
NTSTATUS	TestEntry (_In_ PDRIVER_OBJECT DriverObject, _In_ PCUNICODE_STRING RegistryPath, _Out_ PCWSTR *DeviceName, _Inout_ INT *Flags)
VOID	TestUnload (_In_ PDRIVER_OBJECT DriverObject)
static NTSTATUS	TestIrpHandler (_In_ PDEVICE_OBJECT DeviceObject, _In_ PIRP Irp, _In_ PIO_STACK_LOCATION IoStack)
static NTSTATUS	TestIoCreateFile (IN PUNICODE_STRING Path)
static NTSTATUS	TestMessageHandler (IN PDEVICE_OBJECT DeviceObject, IN ULONG ControlCode, IN PVOID Buffer OPTIONAL, IN SIZE_T InLength, IN OUT PSIZE_T OutLength)

Variables
static KMT_IRP_HANDLER	TestIrpHandler
static KMT_MESSAGE_HANDLER	TestMessageHandler
static PFILE_OBJECT	TestFileObject
static PDEVICE_OBJECT	TestDeviceObject
static volatile long	gNoLinks = FALSE
static UNICODE_STRING	FileObjectFileName = RTL_CONSTANT_STRING(L"\\NonSymlinked")
static UNICODE_STRING	DocumentsAndSettings = RTL_CONSTANT_STRING(L"\\Documents and Settings")

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 10 of file IoCreateFile_drv.c.

Typedef Documentation

PTEST_FCB

typedef struct _TEST_FCB * PTEST_FCB

TEST_FCB

typedef struct _TEST_FCB TEST_FCB

Function Documentation

TestEntry()

NTSTATUS TestEntry	(	_In_ PDRIVER_OBJECT	DriverObject,
		_In_ PCUNICODE_STRING	RegistryPath,
		_Out_ PCWSTR *	DeviceName,
		_Inout_ INT *	Flags
	)

Definition at line 29 of file IoCreateFile_drv.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
 
    PAGED_CODE();
 
    UNREFERENCED_PARAMETER(RegistryPath);
 
    *DeviceName = L"IoCreateFile";
    *Flags = TESTENTRY_NO_EXCLUSIVE_DEVICE |
             TESTENTRY_BUFFERED_IO_DEVICE |
             TESTENTRY_NO_READONLY_DEVICE;
 
    KmtRegisterIrpHandler(IRP_MJ_CREATE, NULL, TestIrpHandler);
    KmtRegisterIrpHandler(IRP_MJ_CLEANUP, NULL, TestIrpHandler);
    KmtRegisterMessageHandler(0, NULL, TestMessageHandler);
 
    return Status;
}

TestIoCreateFile()

static NTSTATUS TestIoCreateFile ( IN PUNICODE_STRING  Path )

static

Definition at line 187 of file IoCreateFile_drv.c.

{
    OBJECT_ATTRIBUTES ObjectAttributes;
    IO_STATUS_BLOCK IoStatusBlock;
    HANDLE Handle;
    NTSTATUS Status;
    PFILE_OBJECT FileObject;
 
    InitializeObjectAttributes(&ObjectAttributes,
                               Path,
                               OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE,
                               NULL,
                               NULL);
    Status = IoCreateFile(&Handle,
                          GENERIC_READ,
                          &ObjectAttributes,
                          &IoStatusBlock,
                          NULL,
                          FILE_ATTRIBUTE_NORMAL,
                          FILE_SHARE_READ | FILE_SHARE_WRITE,
                          FILE_OPEN,
                          FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_ALERT,
                          NULL,
                          0,
                          CreateFileTypeNone,
                          NULL,
                          IO_NO_PARAMETER_CHECKING | (gNoLinks ? IO_STOP_ON_SYMLINK : 0));
    if (NT_SUCCESS(Status))
    {
        NTSTATUS IntStatus;
 
        IntStatus = ObReferenceObjectByHandle(Handle,
                                              FILE_READ_DATA,
                                              *IoFileObjectType,
                                              KernelMode,
                                              (PVOID *)&FileObject,
                                              NULL);
        ok_eq_hex(IntStatus, STATUS_SUCCESS);
        if (NT_SUCCESS(IntStatus))
        {
            ok(RtlCompareUnicodeString(&FileObjectFileName, &FileObject->FileName, TRUE) == 0 ||
               RtlCompareUnicodeString(&DocumentsAndSettings, &FileObject->FileName, TRUE) == 0,
               "Expected: %wZ or %wZ. Opened: %wZ\n", &FileObjectFileName, &DocumentsAndSettings, &FileObject->FileName);
            ObDereferenceObject(FileObject);
        }
 
        IntStatus = ObCloseHandle(Handle, KernelMode);
        ok_eq_hex(IntStatus, STATUS_SUCCESS);
    }
 
    return Status;
}

Referenced by TestMessageHandler().

TestIrpHandler()

static NTSTATUS TestIrpHandler ( _In_ PDEVICE_OBJECT  DeviceObject, _In_ PIRP  Irp, _In_ PIO_STACK_LOCATION  IoStack  )

static

Definition at line 64 of file IoCreateFile_drv.c.

{
    NTSTATUS Status;
    PTEST_FCB Fcb;
    CACHE_UNINITIALIZE_EVENT CacheUninitEvent;
 
    PAGED_CODE();
 
    DPRINT("IRP %x/%x\n", IoStack->MajorFunction, IoStack->MinorFunction);
    ASSERT(IoStack->MajorFunction == IRP_MJ_CREATE ||
           IoStack->MajorFunction == IRP_MJ_CLEANUP);
 
    Status = STATUS_NOT_SUPPORTED;
    Irp->IoStatus.Information = 0;
 
    if (IoStack->MajorFunction == IRP_MJ_CREATE)
    {
        ok((IoStack->Parameters.Create.Options & FILE_OPEN_REPARSE_POINT) == 0, "FILE_OPEN_REPARSE_POINT set\n");
        ok((IoStack->Flags == 0 && !gNoLinks) || (IoStack->Flags == SL_STOP_ON_SYMLINK && gNoLinks), "IoStack->Flags = %lx\n", IoStack->Flags);
 
        if (IoStack->FileObject->FileName.Length >= 2 * sizeof(WCHAR))
        {
            TestDeviceObject = DeviceObject;
            TestFileObject = IoStack->FileObject;
        }
        if (IoStack->FileObject->FileName.Length >= 2 * sizeof(WCHAR) &&
            IoStack->FileObject->FileName.Buffer[1] == 'M')
        {
            PREPARSE_DATA_BUFFER Reparse;
 
            Irp->Tail.Overlay.AuxiliaryBuffer = ExAllocatePoolZero(NonPagedPool, MAXIMUM_REPARSE_DATA_BUFFER_SIZE, 'FwrI');
            Reparse = (PREPARSE_DATA_BUFFER)Irp->Tail.Overlay.AuxiliaryBuffer;
 
            if (!Reparse)
            {
                Status = STATUS_INSUFFICIENT_RESOURCES;
                goto Finish;
            }
 
            Reparse->ReparseTag = IO_REPARSE_TAG_MOUNT_POINT;
            Reparse->ReparseDataLength = 12 + sizeof(L"\\??\\C:\\Documents and Settings");
            Reparse->MountPointReparseBuffer.SubstituteNameLength = sizeof(L"\\??\\C:\\Documents and Settings") - sizeof(UNICODE_NULL);
            Reparse->MountPointReparseBuffer.PrintNameOffset = sizeof(L"\\??\\C:\\Documents and Settings");
            RtlCopyMemory(Reparse->MountPointReparseBuffer.PathBuffer, L"\\??\\C:\\Documents and Settings", sizeof(L"\\??\\C:\\Documents and Settings"));
            Irp->IoStatus.Information = IO_REPARSE_TAG_MOUNT_POINT;
            Status = STATUS_REPARSE;
        }
        else if (IoStack->FileObject->FileName.Length >= 2 * sizeof(WCHAR) &&
            IoStack->FileObject->FileName.Buffer[1] == 'S')
        {
            PREPARSE_DATA_BUFFER Reparse;
 
            if (IoStack->Flags & SL_STOP_ON_SYMLINK)
            {
                Status = STATUS_STOPPED_ON_SYMLINK;
                goto Finish;
            }
 
            Irp->Tail.Overlay.AuxiliaryBuffer = ExAllocatePoolZero(NonPagedPool, MAXIMUM_REPARSE_DATA_BUFFER_SIZE, 'FwrI');
            Reparse = (PREPARSE_DATA_BUFFER)Irp->Tail.Overlay.AuxiliaryBuffer;
 
            if (!Reparse)
            {
                Status = STATUS_INSUFFICIENT_RESOURCES;
                goto Finish;
            }
 
            Reparse->ReparseTag = IO_REPARSE_TAG_SYMLINK;
            Reparse->ReparseDataLength = 12 + sizeof(L"\\??\\C:\\Documents and Settings");
            Reparse->SymbolicLinkReparseBuffer.SubstituteNameLength = sizeof(L"\\??\\C:\\Documents and Settings") - sizeof(UNICODE_NULL);
            Reparse->SymbolicLinkReparseBuffer.PrintNameOffset = sizeof(L"\\??\\C:\\Documents and Settings");
            RtlCopyMemory(Reparse->SymbolicLinkReparseBuffer.PathBuffer, L"\\??\\C:\\Documents and Settings", sizeof(L"\\??\\C:\\Documents and Settings"));
            Irp->IoStatus.Information = IO_REPARSE_TAG_SYMLINK;
            Status = STATUS_REPARSE;
        }
        else
        {
            Fcb = ExAllocatePoolZero(NonPagedPool, sizeof(*Fcb), 'FwrI');
            
            if (!Fcb)
            {
                Status = STATUS_INSUFFICIENT_RESOURCES;
                goto Finish;
            }
 
            ExInitializeFastMutex(&Fcb->HeaderMutex);
            FsRtlSetupAdvancedHeader(&Fcb->Header, &Fcb->HeaderMutex);
            Fcb->Header.AllocationSize.QuadPart = 0;
            Fcb->Header.FileSize.QuadPart = 0;
            Fcb->Header.ValidDataLength.QuadPart = 0;
            IoStack->FileObject->FsContext = Fcb;
            IoStack->FileObject->SectionObjectPointer = &Fcb->SectionObjectPointers;
 
            Irp->IoStatus.Information = FILE_OPENED;
            Status = STATUS_SUCCESS;
        }
    }
    else if (IoStack->MajorFunction == IRP_MJ_CLEANUP)
    {
        KeInitializeEvent(&CacheUninitEvent.Event, NotificationEvent, FALSE);
        CcUninitializeCacheMap(IoStack->FileObject, NULL, &CacheUninitEvent);
        KeWaitForSingleObject(&CacheUninitEvent.Event, Executive, KernelMode, FALSE, NULL);
        Fcb = IoStack->FileObject->FsContext;
        ExFreePoolWithTag(Fcb, 'FwrI');
        IoStack->FileObject->FsContext = NULL;
        Status = STATUS_SUCCESS;
    }
 
Finish:
    Irp->IoStatus.Status = Status;
    IoCompleteRequest(Irp, IO_NO_INCREMENT);
 
    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 243 of file IoCreateFile_drv.c.

{
    NTSTATUS Status = STATUS_SUCCESS;
 
    PAGED_CODE();
 
    switch (ControlCode)
    {
        case IOCTL_DISABLE_SYMLINK:
        {
            if (InterlockedExchange(&gNoLinks, TRUE) == TRUE)
            {
                Status = STATUS_UNSUCCESSFUL;
            }
 
            break;
        }
        case IOCTL_CALL_CREATE:
        {
            ANSI_STRING Path;
            UNICODE_STRING PathW;
 
            ok(Buffer != NULL, "Buffer is NULL\n");
            Path.Length = Path.MaximumLength = (USHORT)InLength;
            Path.Buffer = Buffer;
 
            Status = RtlAnsiStringToUnicodeString(&PathW, &Path, TRUE);
            ok_eq_hex(Status, STATUS_SUCCESS);
 
            Status = TestIoCreateFile(&PathW);
 
            RtlFreeUnicodeString(&PathW);
 
            break;
        }
        default:
            return STATUS_NOT_SUPPORTED;
    }
 
    return Status;
}

TestUnload()

VOID TestUnload

(

_In_ PDRIVER_OBJECT

DriverObject

)

Definition at line 54 of file IoCreateFile_drv.c.

{
    PAGED_CODE();
}

Variable Documentation

DocumentsAndSettings

UNICODE_STRING DocumentsAndSettings = RTL_CONSTANT_STRING(L"\\Documents and Settings")

static

Definition at line 183 of file IoCreateFile_drv.c.

Referenced by TestIoCreateFile().

FileObjectFileName

UNICODE_STRING FileObjectFileName = RTL_CONSTANT_STRING(L"\\NonSymlinked")

static

Definition at line 182 of file IoCreateFile_drv.c.

Referenced by TestIoCreateFile().

gNoLinks

volatile long gNoLinks = FALSE

static

Definition at line 60 of file IoCreateFile_drv.c.

Referenced by TestIoCreateFile(), TestIrpHandler(), and TestMessageHandler().

TestDeviceObject

PDEVICE_OBJECT TestDeviceObject

static

Definition at line 26 of file IoCreateFile_drv.c.

Referenced by TestIrpHandler().

TestFileObject

PFILE_OBJECT TestFileObject

static

Definition at line 25 of file IoCreateFile_drv.c.

Referenced by TestIrpHandler().

TestIrpHandler

KMT_IRP_HANDLER TestIrpHandler

static

Definition at line 22 of file IoCreateFile_drv.c.

Referenced by TestEntry().

TestMessageHandler

KMT_MESSAGE_HANDLER TestMessageHandler

static

Definition at line 23 of file IoCreateFile_drv.c.

Referenced by TestEntry().