#include "ntddk.h"
#include "drv03.h"
#if DBG
#define dprintf DbgPrint
#else
#define dprintf(x)
#endif
NTSTATUS DriverDispatch(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp);
VOID DriverUnload(IN PDRIVER_OBJECT DriverObject);
#define NT_DEVICE_NAME L"//Device//drv03"
#define DOS_DEVICE_NAME L"//DosDevices//drv03"
NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject, IN PUNICODE_STRING RegistryPath)
{
NTSTATUS ntStatus;
PDEVICE_OBJECT DeviceObject;
UNICODE_STRING deviceName;
UNICODE_STRING deviceLink;
PDEVICE_EXTENSION deviceExtension;
dprintf("drv03.SYS: entering DriverEntry/n");
RtlInitUnicodeString(& deviceName, NT_DEVICE_NAME);
ntStatus = IoCreateDevice(DriverObject, sizeof(DEVICE_EXTENSION),
& deviceName, FILE_DEVICE_drv03, 0, TRUE,
& DeviceObject);
if ( NT_SUCCESS(ntStatus) )
{
deviceExtension = DeviceObject->DeviceExtension;
RtlInitUnicodeString(& deviceLink, DOS_DEVICE_NAME);
ntStatus = IoCreateSymbolicLink(& deviceLink, & deviceName);
if (! NT_SUCCESS(ntStatus) )
{
dprintf("drv03.SYS: IoCreateSymbolicLink failed/n");
}
DriverObject->MajorFunction[ IRP_MJ_CREATE ] =
DriverObject->MajorFunction[ IRP_MJ_CLOSE ] =
DriverObject->MajorFunction[ IRP_MJ_DEVICE_CONTROL ] = DriverDispatch;
DriverObject->DriverUnload = DriverUnload;
}
if (! NT_SUCCESS(ntStatus) )
{
if (DeviceObject)
IoDeleteDevice(DeviceObject);
}
return ntStatus;
}
NTSTATUS DriverDispatch(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
{
NTSTATUS ntStatus;
PIO_STACK_LOCATION irpStack = NULL;
PDEVICE_EXTENSION deviceExtension;
PVOID ioBuffer;
ULONG inputBufferLength;
ULONG outputBufferLength;
ULONG ioControlCode;
Irp->IoStatus.Status = STATUS_SUCCESS;
Irp->IoStatus.Information = 0;
irpStack = IoGetCurrentIrpStackLocation(Irp);
deviceExtension = DeviceObject->DeviceExtension;
ioBuffer = Irp->AssociatedIrp.SystemBuffer;
inputBufferLength = irpStack->Parameters.DeviceIoControl.InputBufferLength;
outputBufferLength = irpStack->Parameters.DeviceIoControl.OutputBufferLength;
switch (irpStack->MajorFunction)
{
case IRP_MJ_CREATE:
dprintf("drv03.SYS: IRP_MJ_CREATE/n");
break;
case IRP_MJ_CLOSE:
dprintf("drv03.SYS: IRP_MJ_CLOSE/n");
break;
case IRP_MJ_DEVICE_CONTROL:
dprintf("drv03.SYS: IRP_MJ_DEVICE_CONTROL/n");
ioControlCode = irpStack->Parameters.DeviceIoControl.IoControlCode;
switch(ioControlCode)
{
case IOCTL_drv03_HELLO:
break;
default:
Irp->IoStatus.Status = STATUS_INVALID_PARAMETER;
dprintf("drv03.SYS: unknown IRP_MJ_DEVICE_CONTROL/n");
break;
}
break;
}
ntStatus = Irp->IoStatus.Status;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
return ntStatus;
}
VOID DriverUnload(IN PDRIVER_OBJECT DriverObject)
{
UNICODE_STRING deviceLink;
RtlInitUnicodeString(& deviceLink, DOS_DEVICE_NAME);
IoDeleteSymbolicLink(& deviceLink);
IoDeleteDevice(DriverObject->DeviceObject);
dprintf("drv03.SYS: unloading/n");
}