windows xp下的pci设备驱动程序开发.rar_PCI驱动_pci_windows xp_设备驱动程序

/************************************************************************ * 文件名称:HelloWDM.cpp * 作 者:张帆 * 完成日期:2007-11-1 *************************************************************************/ #include "HelloWDM.h" #include <initguid.h> #include "guid.h" #include "Ioctls.h" /************************************************************************ * 函数名称:DriverEntry * 功能描述:初始化驱动程序，定位和申请硬件资源，创建内核对象 * 参数列表: pDriverObject:从I/O管理器中传进来的驱动对象 pRegistryPath:驱动程序在注册表的中的路径 * 返回 值:返回初始化驱动状态 *************************************************************************/ #pragma INITCODE extern "C" NTSTATUS DriverEntry(IN PDRIVER_OBJECT pDriverObject, IN PUNICODE_STRING pRegistryPath) { KdPrint(("Enter DriverEntry\n")); pDriverObject->DriverExtension->AddDevice = HelloWDMAddDevice; pDriverObject->MajorFunction[IRP_MJ_PNP] = HelloWDMPnp; pDriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = pDriverObject->MajorFunction[IRP_MJ_CREATE] = pDriverObject->MajorFunction[IRP_MJ_CLOSE] = pDriverObject->MajorFunction[IRP_MJ_READ] = pDriverObject->MajorFunction[IRP_MJ_WRITE] = HelloWDMDispatchRoutine; pDriverObject->DriverUnload = HelloWDMUnload; KdPrint(("Leave DriverEntry\n")); return STATUS_SUCCESS; } /************************************************************************ * 函数名称:HelloWDMAddDevice * 功能描述:添加新设备 * 参数列表: DriverObject:从I/O管理器中传进来的驱动对象 PhysicalDeviceObject:从I/O管理器中传进来的物理设备对象 * 返回 值:返回添加新设备状态 *************************************************************************/ #pragma PAGEDCODE NTSTATUS HelloWDMAddDevice(IN PDRIVER_OBJECT DriverObject, IN PDEVICE_OBJECT PhysicalDeviceObject) { PAGED_CODE(); KdPrint(("Enter HelloWDMAddDevice\n")); NTSTATUS status; PDEVICE_OBJECT fdo; status = IoCreateDevice( DriverObject, sizeof(DEVICE_EXTENSION), NULL,//没有指定设备名 FILE_DEVICE_UNKNOWN, 0, FALSE, &fdo); if( !NT_SUCCESS(status)) return status; PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)fdo->DeviceExtension; pdx->fdo = fdo; pdx->NextStackDevice = IoAttachDeviceToDeviceStack(fdo, PhysicalDeviceObject); //创建设备接口 status = IoRegisterDeviceInterface(PhysicalDeviceObject, &MY_WDM_DEVICE, NULL, &pdx->interfaceName); if( !NT_SUCCESS(status)) { IoDeleteDevice(fdo); return status; } KdPrint(("%wZ\n",&pdx->interfaceName)); IoSetDeviceInterfaceState(&pdx->interfaceName, TRUE); if( !NT_SUCCESS(status)) { if( !NT_SUCCESS(status)) { return status; } } fdo->Flags |= DO_BUFFERED_IO | DO_POWER_PAGABLE; fdo->Flags &= ~DO_DEVICE_INITIALIZING; KdPrint(("Leave HelloWDMAddDevice\n")); return STATUS_SUCCESS; } /************************************************************************ * 函数名称:DefaultPnpHandler * 功能描述:对PNP IRP进行缺省处理 * 参数列表: pdx:设备对象的扩展 Irp:从IO请求包 * 返回 值:返回状态 *************************************************************************/ #pragma PAGEDCODE NTSTATUS DefaultPnpHandler(PDEVICE_EXTENSION pdx, PIRP Irp) { PAGED_CODE(); KdPrint(("Enter DefaultPnpHandler\n")); IoSkipCurrentIrpStackLocation(Irp); KdPrint(("Leave DefaultPnpHandler\n")); return IoCallDriver(pdx->NextStackDevice, Irp); } #pragma LOCKEDCODE NTSTATUS OnRequestComplete(PDEVICE_OBJECT junk, PIRP Irp, PKEVENT pev) { // OnRequestComplete //在完成例程中设置等待事件 KeSetEvent(pev, 0, FALSE); //标志本IRP还需要再次被完成 return STATUS_MORE_PROCESSING_REQUIRED; } /////////////////////////////////////////////////////////////////////////////// #pragma PAGEDCODE NTSTATUS ForwardAndWait(PDEVICE_EXTENSION pdx, PIRP Irp) { // ForwardAndWait PAGED_CODE(); KEVENT event; //初始化事件 KeInitializeEvent(&event, NotificationEvent, FALSE); //将本层堆栈拷贝到下一层堆栈 IoCopyCurrentIrpStackLocationToNext(Irp); //设置完成例程 IoSetCompletionRoutine(Irp, (PIO_COMPLETION_ROUTINE) OnRequestComplete, (PVOID) &event, TRUE, TRUE, TRUE); //调用底层驱动，即PDO IoCallDriver(pdx->NextStackDevice, Irp); //等待PDO完成 KeWaitForSingleObject(&event, Executive, KernelMode, FALSE, NULL); return Irp->IoStatus.Status; } // ForwardAndWait /************************************************************************ * 函数名称:HandleRemoveDevice * 功能描述:对IRP_MN_REMOVE_DEVICE IRP进行处理 * 参数列表: fdo:功能设备对象 Irp:从IO请求包 * 返回 值:返回状态 *************************************************************************/ #pragma PAGEDCODE NTSTATUS HandleRemoveDevice(PDEVICE_EXTENSION pdx, PIRP Irp) { PAGED_CODE(); KdPrint(("Enter HandleRemoveDevice\n")); Irp->IoStatus.Status = STATUS_SUCCESS; NTSTATUS status = DefaultPnpHandler(pdx, Irp); IoSetDeviceInterfaceState(&pdx->interfaceName, FALSE); RtlFreeUnicodeString(&pdx->interfaceName); //调用IoDetachDevice()把fdo从设备栈中脱开： if (pdx->NextStackDevice) IoDetachDevice(pdx->NextStackDevice); //删除fdo： IoDeleteDevice(pdx->fdo); //删除中断 IoDisconnectInterrupt(pdx->InterruptObject); KdPrint(("Leave HandleRemoveDevice\n")); return status; } BOOLEAN OnInterrupt(PKINTERRUPT InterruptObject, PDEVICE_EXTENSION pdx) { // OnInterrupt //关中断 UCHAR HSR = READ_PORT_UCHAR(pdx->portbase); HSR = HSR | 0x4; WRITE_PORT_UCHAR(pdx->portbase,HSR); KdPrint(("==============interrupt!!!\n")); //恢复中断信号电平 WRITE_REGISTER_UCHAR((PUCHAR)pdx->MemBar1+0x400000,0x10); IoRequestDpc(pdx->fdo, NULL, pdx); return TRUE; } #pragma PAGEDCODE NTSTATUS InitMyPCI(IN PDEVICE_EXTENSION pdx,IN PCM_PARTIAL_RESOURCE_LIST list) { PDEVICE_OBJECT fdo = pdx->fdo; ULONG vector; KIRQL irql; KINTERRUPT_MODE mode; KAFFINITY affinity; BOOLEAN irqshare; BOOLEAN gotinterrupt = FALSE; PHYSICAL_ADDRESS portbase; BOOLEAN gotport = FALSE; PCM_PARTIAL_RESOURCE_DESCRIPTOR resource = &list->PartialDescriptors[0]; ULONG nres = list->Count; BOOLEAN IsMem0 = TRUE; for (ULONG i = 0; i < nres; ++i, ++resource) { // for each resource switch (resource->Type) { // switch on resource type case CmResourceTypePort: portbase = resource->u.Port.Start; pdx->nports = resource->u.Port.Length; pdx->mappedport = (resource->Flags & CM_RESOURCE_PORT_IO) == 0; gotport = TRUE; break; case CmResourceTypeMemory: if (IsMem0) { pdx->MemBar0 = (PUCHAR)MmMapIoSpace(resource->u.Memory.Start, resource->u.Memory.Length, MmNonCached); pdx->nMem0 = resource->u.Memory.Length; IsMem0 = FALSE; }else { pdx->MemBar1 = (PUCHAR)MmMapIoSpace(resource->u.Memory.Start, resource->u.Memory.Length, MmNonCached); pdx->nMem1 = resource->u.Memory.Length; } break; case CmResourceTypeInterrupt: irql = (KIRQL) resource->u.Interrupt.Level; vector = resource->u.Interrupt.Vector; affinity = resource->u.Interrupt.Affinity; mode = (resource->Flags == CM_RESOURCE_INTERRUPT_LATCHED) ? Latched : LevelSensitive; irqshare = resource->ShareDisposition == CmResourceShareShared; gotinterrupt = TRUE; break; default: KdPrint(("Unexpected I/O resource type %d\n", resource->Type)); break; } // switch on resource type } // for each resource if (!(TRUE&& gotport&& gotinterrupt )) { KdPrint((" Didn't get expected I/O resources\n")); return STATUS_DEVICE_CONFIGURATION_ERROR; } if (pdx->mappedport) { // map port address for RISC platform pdx->portbase = (PUCHAR) MmMapIoSpace(portbase, pdx->nports, MmNonCached); if (!pdx->mappedport) { KdPrint(("Unable to map port r