driver.c File Reference

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

Include dependency graph for driver.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
KSDDKAPI PKSDEVICE NTAPI	KsGetDeviceForDeviceObject (IN PDEVICE_OBJECT FunctionalDeviceObject)
KSDDKAPI PKSDEVICE NTAPI	KsGetDevice (IN PVOID Object)
KSDDKAPI NTSTATUS NTAPI	KsCreateDevice (IN PDRIVER_OBJECT DriverObject, IN PDEVICE_OBJECT PhysicalDeviceObject, IN const KSDEVICE_DESCRIPTOR *Descriptor OPTIONAL, IN ULONG ExtensionSize OPTIONAL, OUT PKSDEVICE *Device OPTIONAL)
KSDDKAPI NTSTATUS NTAPI	KsAddDevice (IN PDRIVER_OBJECT DriverObject, IN PDEVICE_OBJECT PhysicalDeviceObject)
KSDDKAPI NTSTATUS NTAPI	KsInitializeDriver (IN PDRIVER_OBJECT DriverObject, IN PUNICODE_STRING RegistryPath, IN const KSDEVICE_DESCRIPTOR *Descriptor OPTIONAL)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 11 of file driver.c.

Function Documentation

KsAddDevice()

KSDDKAPI NTSTATUS NTAPI KsAddDevice	(	IN PDRIVER_OBJECT	DriverObject,
		IN PDEVICE_OBJECT	PhysicalDeviceObject
	)

Definition at line 129 of file driver.c.

{
    PKS_DRIVER_EXTENSION DriverObjectExtension;
    const KSDEVICE_DESCRIPTOR *Descriptor = NULL;
 
    /* get stored driver object extension */
 
    DriverObjectExtension = IoGetDriverObjectExtension(DriverObject, (PVOID)KsInitializeDriver);
 
    if (DriverObjectExtension)
    {
        /* get the stored descriptor see KsInitializeDriver */
        Descriptor = DriverObjectExtension->Descriptor;
    }
 
    return KsCreateDevice(DriverObject, PhysicalDeviceObject, Descriptor, 0, NULL);
}

Referenced by KsInitializeDriver().

KsCreateDevice()

KSDDKAPI NTSTATUS NTAPI KsCreateDevice	(	IN PDRIVER_OBJECT	DriverObject,
		IN PDEVICE_OBJECT	PhysicalDeviceObject,
		IN const KSDEVICE_DESCRIPTOR *Descriptor	OPTIONAL,
		IN ULONG ExtensionSize	OPTIONAL,
		OUT PKSDEVICE *Device	OPTIONAL
	)

Definition at line 65 of file driver.c.

{
    NTSTATUS Status = STATUS_DEVICE_REMOVED;
    PDEVICE_OBJECT FunctionalDeviceObject= NULL;
    PDEVICE_OBJECT OldHighestDeviceObject;
    if (!ExtensionSize)
        ExtensionSize = sizeof(KSDEVICE_HEADER);
 
    DPRINT("KsCreateDevice Descriptor %p ExtensionSize %lu\n", Descriptor, ExtensionSize);
 
    Status = IoCreateDevice(DriverObject, ExtensionSize, NULL, FILE_DEVICE_KS, FILE_DEVICE_SECURE_OPEN, FALSE, &FunctionalDeviceObject);
    if (!NT_SUCCESS(Status))
        return Status;
 
    OldHighestDeviceObject = IoAttachDeviceToDeviceStack(FunctionalDeviceObject, PhysicalDeviceObject);
    if (OldHighestDeviceObject)
    {
        Status = KsInitializeDevice(FunctionalDeviceObject, PhysicalDeviceObject, OldHighestDeviceObject, Descriptor);
    }
    else
    {
        Status = STATUS_DEVICE_REMOVED;
    }
 
    /* check if all succeeded */
    if (!NT_SUCCESS(Status))
    {
        if (OldHighestDeviceObject)
            IoDetachDevice(OldHighestDeviceObject);
 
        IoDeleteDevice(FunctionalDeviceObject);
        return Status;
    }
 
    /* set device flags */
    FunctionalDeviceObject->Flags |= DO_DIRECT_IO | DO_POWER_PAGABLE;
    FunctionalDeviceObject->Flags &= ~ DO_DEVICE_INITIALIZING;
 
    if (Device)
    {
        /* get PKSDEVICE struct */
        *Device = KsGetDeviceForDeviceObject(FunctionalDeviceObject);
 
        if (ExtensionSize > sizeof(KSDEVICE_HEADER))
        {
            /* caller needs a device extension */
            (*Device)->Context = (PVOID)((ULONG_PTR)FunctionalDeviceObject->DeviceExtension + sizeof(KSDEVICE_HEADER));
        }
    }
 
    return Status;
}

Referenced by KsAddDevice().

KsGetDevice()

KSDDKAPI PKSDEVICE NTAPI KsGetDevice

(

IN PVOID

Object

)

Definition at line 37 of file driver.c.

{
    PKSBASIC_HEADER BasicHeader = (PKSBASIC_HEADER)((ULONG_PTR)Object - sizeof(KSBASIC_HEADER));
 
    DPRINT("KsGetDevice Type %lu KsDevice %p\n", BasicHeader->Type, BasicHeader->KsDevice);
 
    ASSERT(BasicHeader->Type == KsObjectTypeFilterFactory || BasicHeader->Type == KsObjectTypeFilter || BasicHeader->Type == KsObjectTypePin);
    ASSERT(BasicHeader->KsDevice);
    ASSERT(BasicHeader->KsDevice->Descriptor);
    ASSERT(BasicHeader->KsDevice->Bag);
    ASSERT(BasicHeader->KsDevice->Context);
    ASSERT(BasicHeader->KsDevice->FunctionalDeviceObject);
    ASSERT(BasicHeader->KsDevice->PhysicalDeviceObject);
    ASSERT(BasicHeader->KsDevice->NextDeviceObject);
    ASSERT(BasicHeader->KsDevice->Started);
    ASSERT(BasicHeader->KsDevice->SystemPowerState == PowerSystemWorking);
    ASSERT(BasicHeader->KsDevice->DevicePowerState == PowerDeviceD0);
 
    return BasicHeader->KsDevice;
}

KsGetDeviceForDeviceObject()

KSDDKAPI PKSDEVICE NTAPI KsGetDeviceForDeviceObject

(

IN PDEVICE_OBJECT

FunctionalDeviceObject

)

Definition at line 20 of file driver.c.

{
    PDEVICE_EXTENSION DeviceExtension;
 
    /* get device extension */
    DeviceExtension = (PDEVICE_EXTENSION)FunctionalDeviceObject->DeviceExtension;
 
    return &DeviceExtension->DeviceHeader->KsDevice;
}

Referenced by KsCreateDevice().

KsInitializeDriver()

KSDDKAPI NTSTATUS NTAPI KsInitializeDriver	(	IN PDRIVER_OBJECT	DriverObject,
		IN PUNICODE_STRING	RegistryPath,
		IN const KSDEVICE_DESCRIPTOR *Descriptor	OPTIONAL
	)

Definition at line 155 of file driver.c.

{
    PKS_DRIVER_EXTENSION DriverObjectExtension;
    NTSTATUS Status = STATUS_SUCCESS;
 
    DPRINT("KsInitializeDriver\n");
 
    if (Descriptor)
    {
        Status = IoAllocateDriverObjectExtension(DriverObject, (PVOID)KsInitializeDriver, sizeof(KS_DRIVER_EXTENSION), (PVOID*)&DriverObjectExtension);
        if (NT_SUCCESS(Status))
        {
            DriverObjectExtension->Descriptor = Descriptor;
        }
    }
 
    /* sanity check */
    ASSERT(Status == STATUS_SUCCESS);
 
    if (!NT_SUCCESS(Status))
        return Status;
 
    /* Setting our IRP handlers */
    DriverObject->MajorFunction[IRP_MJ_CREATE] = IKsDevice_Create;
    DriverObject->MajorFunction[IRP_MJ_PNP] = IKsDevice_Pnp;
    DriverObject->MajorFunction[IRP_MJ_POWER] = IKsDevice_Power;
    DriverObject->MajorFunction[IRP_MJ_SYSTEM_CONTROL] = KsDefaultForwardIrp;
 
    /* The driver unload routine */
    DriverObject->DriverUnload = KsNullDriverUnload;
 
    /* The driver-supplied AddDevice */
    DriverObject->DriverExtension->AddDevice = KsAddDevice;
 
    /* KS handles these */
    DPRINT1("KsInitializeDriver Setting KS function handlers\n");
    KsSetMajorFunctionHandler(DriverObject, IRP_MJ_CLOSE);
    KsSetMajorFunctionHandler(DriverObject, IRP_MJ_DEVICE_CONTROL);
 
 
    return STATUS_SUCCESS;
}

Referenced by DriverEntry(), KsAddDevice(), and KsInitializeDriver().