misc.c File Reference

#include "serenum.h"
#include <debug.h>

Include dependency graph for misc.c:

Go to the source code of this file.

Functions
NTSTATUS	SerenumInitMultiSzString (OUT PUNICODE_STRING Destination,...)
static NTSTATUS NTAPI	ForwardIrpAndWaitCompletion (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp, IN PVOID Context)
NTSTATUS	ForwardIrpAndWait (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS NTAPI	ForwardIrpToLowerDeviceAndForget (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS NTAPI	ForwardIrpToAttachedFdoAndForget (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS NTAPI	ForwardIrpAndForget (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
NTSTATUS	DuplicateUnicodeString (IN ULONG Flags, IN PCUNICODE_STRING SourceString, OUT PUNICODE_STRING DestinationString)

Variables
static IO_COMPLETION_ROUTINE	ForwardIrpAndWaitCompletion

Function Documentation

DuplicateUnicodeString()

NTSTATUS DuplicateUnicodeString	(	IN ULONG	Flags,
		IN PCUNICODE_STRING	SourceString,
		OUT PUNICODE_STRING	DestinationString
	)

Definition at line 185 of file misc.c.

{
    if (SourceString == NULL || DestinationString == NULL
     || SourceString->Length > SourceString->MaximumLength
     || (SourceString->Length == 0 && SourceString->MaximumLength > 0 && SourceString->Buffer == NULL)
     || Flags == RTL_DUPLICATE_UNICODE_STRING_ALLOCATE_NULL_STRING || Flags >= 4)
    {
        return STATUS_INVALID_PARAMETER;
    }


    if ((SourceString->Length == 0)
     && (Flags != (RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE |
                   RTL_DUPLICATE_UNICODE_STRING_ALLOCATE_NULL_STRING)))
    {
        DestinationString->Length = 0;
        DestinationString->MaximumLength = 0;
        DestinationString->Buffer = NULL;
    }
    else
    {
        USHORT DestMaxLength = SourceString->Length;

        if (Flags & RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE)
            DestMaxLength += sizeof(UNICODE_NULL);

        DestinationString->Buffer = ExAllocatePoolWithTag(PagedPool, DestMaxLength, SERENUM_TAG);
        if (DestinationString->Buffer == NULL)
            return STATUS_NO_MEMORY;

        RtlCopyMemory(DestinationString->Buffer, SourceString->Buffer, SourceString->Length);
        DestinationString->Length = SourceString->Length;
        DestinationString->MaximumLength = DestMaxLength;

        if (Flags & RTL_DUPLICATE_UNICODE_STRING_NULL_TERMINATE)
            DestinationString->Buffer[DestinationString->Length / sizeof(WCHAR)] = 0;
    }

    return STATUS_SUCCESS;
}

ForwardIrpAndForget()

NTSTATUS NTAPI ForwardIrpAndForget	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 170 of file misc.c.

{
    PDEVICE_OBJECT LowerDevice;

    ASSERT(((PCOMMON_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->IsFDO);
    LowerDevice = ((PFDO_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->LowerDevice;
    ASSERT(LowerDevice);

    IoSkipCurrentIrpStackLocation(Irp);
    return IoCallDriver(LowerDevice, Irp);
}

ForwardIrpAndWait()

NTSTATUS ForwardIrpAndWait	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 103 of file misc.c.

{
    PDEVICE_OBJECT LowerDevice;
    KEVENT Event;
    NTSTATUS Status;

    ASSERT(((PCOMMON_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->IsFDO);
    LowerDevice = ((PFDO_DEVICE_EXTENSION)DeviceObject->DeviceExtension)->LowerDevice;

    ASSERT(LowerDevice);

    KeInitializeEvent(&Event, NotificationEvent, FALSE);
    IoCopyCurrentIrpStackLocationToNext(Irp);

    TRACE_(SERENUM, "Calling lower device %p\n", LowerDevice);
    IoSetCompletionRoutine(Irp, ForwardIrpAndWaitCompletion, &Event, TRUE, TRUE, TRUE);

    Status = IoCallDriver(LowerDevice, Irp);
    if (Status == STATUS_PENDING)
    {
        Status = KeWaitForSingleObject(&Event, Suspended, KernelMode, FALSE, NULL);
        if (NT_SUCCESS(Status))
            Status = Irp->IoStatus.Status;
    }

    return Status;
}

ForwardIrpAndWaitCompletion()

static NTSTATUS NTAPI ForwardIrpAndWaitCompletion ( IN PDEVICE_OBJECT  DeviceObject, IN PIRP  Irp, IN PVOID  Context  )

static

Definition at line 92 of file misc.c.

{
    if (Irp->PendingReturned)
        KeSetEvent((PKEVENT)Context, IO_NO_INCREMENT, FALSE);
    return STATUS_MORE_PROCESSING_REQUIRED;
}

ForwardIrpToAttachedFdoAndForget()

NTSTATUS NTAPI ForwardIrpToAttachedFdoAndForget	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 152 of file misc.c.

{
    PPDO_DEVICE_EXTENSION DeviceExtension;
    PDEVICE_OBJECT Fdo;

    DeviceExtension = (PPDO_DEVICE_EXTENSION)DeviceObject->DeviceExtension;
    ASSERT(!DeviceExtension->Common.IsFDO);

    Fdo = DeviceExtension->AttachedFdo;
    ASSERT(Fdo);
    TRACE_(SERENUM, "Calling attached Fdo 0x%p\n", Fdo);
    IoSkipCurrentIrpStackLocation(Irp);
    return IoCallDriver(Fdo, Irp);
}

Referenced by IrpStub(), and SerenumPdoPnp().

ForwardIrpToLowerDeviceAndForget()

NTSTATUS NTAPI ForwardIrpToLowerDeviceAndForget	(	IN PDEVICE_OBJECT	DeviceObject,
		IN PIRP	Irp
	)

Definition at line 134 of file misc.c.

{
    PFDO_DEVICE_EXTENSION DeviceExtension;
    PDEVICE_OBJECT LowerDevice;

    DeviceExtension = (PFDO_DEVICE_EXTENSION)DeviceObject->DeviceExtension;
    ASSERT(DeviceExtension->Common.IsFDO);

    LowerDevice = DeviceExtension->LowerDevice;
    ASSERT(LowerDevice);
    TRACE_(SERENUM, "Calling lower device 0x%p\n", LowerDevice);
    IoSkipCurrentIrpStackLocation(Irp);
    return IoCallDriver(LowerDevice, Irp);
}

Referenced by IrpStub().

SerenumInitMultiSzString()

NTSTATUS SerenumInitMultiSzString	(	OUT PUNICODE_STRING	Destination,
			...
	)

Definition at line 20 of file misc.c.

{
    va_list args;
    PCSZ Source;
    ANSI_STRING AnsiString;
    UNICODE_STRING UnicodeString;
    ULONG DestinationSize = 0;
    NTSTATUS Status = STATUS_SUCCESS;

    ASSERT(Destination);

    /* Calculate length needed for destination unicode string */
    va_start(args, Destination);
    Source = va_arg(args, PCSZ);
    while (Source != NULL)
    {
        RtlInitAnsiString(&AnsiString, Source);
        DestinationSize += RtlAnsiStringToUnicodeSize(&AnsiString)
            + sizeof(WCHAR) /* final NULL */;
        Source = va_arg(args, PCSZ);
    }
    va_end(args);
    if (DestinationSize == 0)
    {
        RtlInitUnicodeString(Destination, NULL);
        return STATUS_SUCCESS;
    }

    /* Initialize destination string */
    DestinationSize += sizeof(WCHAR); // final NULL
    Destination->Buffer = (PWSTR)ExAllocatePoolWithTag(PagedPool, DestinationSize, SERENUM_TAG);
    if (!Destination->Buffer)
        return STATUS_INSUFFICIENT_RESOURCES;
    Destination->Length = 0;
    Destination->MaximumLength = (USHORT)DestinationSize;

    /* Copy arguments to destination string */
    /* Use a temporary unicode string, which buffer is shared with
     * destination string, to copy arguments */
    UnicodeString.Length = Destination->Length;
    UnicodeString.MaximumLength = Destination->MaximumLength;
    UnicodeString.Buffer = Destination->Buffer;
    va_start(args, Destination);
    Source = va_arg(args, PCSZ);
    while (Source != NULL)
    {
        RtlInitAnsiString(&AnsiString, Source);
        Status = RtlAnsiStringToUnicodeString(&UnicodeString, &AnsiString, FALSE);
        if (!NT_SUCCESS(Status))
        {
            ExFreePoolWithTag(Destination->Buffer, SERENUM_TAG);
            break;
        }
        Destination->Length += UnicodeString.Length + sizeof(WCHAR);
        UnicodeString.MaximumLength -= UnicodeString.Length + sizeof(WCHAR);
        UnicodeString.Buffer += UnicodeString.Length / sizeof(WCHAR) + 1;
        UnicodeString.Length = 0;
        Source = va_arg(args, PCSZ);
    }
    va_end(args);
    if (NT_SUCCESS(Status))
    {
        /* Finish multi-sz string */
        Destination->Buffer[Destination->Length / sizeof(WCHAR)] = L'\0';
        Destination->Length += sizeof(WCHAR);
    }
    return Status;
}

Referenced by SerenumDetectLegacyDevice().

Variable Documentation

ForwardIrpAndWaitCompletion

IO_COMPLETION_ROUTINE ForwardIrpAndWaitCompletion

static

Definition at line 14 of file misc.c.

Referenced by ForwardIrpAndWait().