Driver development code example.

#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");

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值