IO_Card_Dev.zip_http://dev-io

/* IO_Card 驱动 1、系统找到DriverEntry()函数，初始化驱动程序。 2、系统调用AddDevice例程添加设备对象。 3、FDO收到 IRP_MN_START_DEVICE IRP，将其设置完成例程（实际设置在下个IO堆栈）后传递给PDO。接着FDO等待同步。（实际上PDO完成资源分配） 4、当PDO完成 IRP_MN_START_DEVICE IRP，设备堆栈回滚时。执行完成例程，并将IRP再次交给FDO。 5、FDO根据IRP提供的IO堆栈得到设备资源，并完成例程。 6、等待各种IRP和中断 6.1、中断（主要用于操作） 6.2、PNP IRP（主要用于配置） 6.3、普通IRP（主要用于操作） 7、当驱动处理完IRP_MN_REMOVE_DEVICE例程后，系统调用HelloWDMUnload例程完成后续工作。 */ #include <initguid.h> #include "Guid.h" #include "IO_Card.h" #include "Ioctls.h" #pragma INITCODE // INIT指明该函数只用于初始化，用完后就释放。（节约内存） // 代码放在代码段，数据放在数据段。 extern "C" NTSTATUS DriverEntry( // IN、OUT和INOUT是空宏定义，主要起注释作用。 IN PDRIVER_OBJECT pDriverObject, // I/O管理器传递过来的驱动对象。 IN PUNICODE_STRING pRegistryPath // 驱动对象注册表路径。 ) // extern "C"用于在CPP模式下引用C代码，保证符号链接的正确性。 { KdPrint(("Enter DriverEntry\n")); // 调试信息打印函数，它只在check版本中生效。 pDriverObject->DriverExtension->AddDevice = IO_CardAddDevice; // 设置添加设备回调函数。 pDriverObject->MajorFunction[IRP_MJ_PNP] = IO_CardPnp; // 设置PNP IRP处理回调函数。 pDriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = DispatchControl; // 设置IO操作回调函数。 pDriverObject->MajorFunction[IRP_MJ_CREATE] = IO_CardDispatchRoutine; pDriverObject->MajorFunction[IRP_MJ_CLOSE] = IO_CardDispatchRoutine; pDriverObject->MajorFunction[IRP_MJ_READ] = IO_CardDispatchRoutine; pDriverObject->MajorFunction[IRP_MJ_WRITE] = IO_CardDispatchRoutine; // 设置缺省IRP处理回调函数。 pDriverObject->DriverUnload = IO_CardUnload; // 卸载驱动，设置删除设备回调函数（实际上在PNP IRP中就被处理了）。 KdPrint(("Leave DriverEntry\n")); // 调试信息打印函数，它只在check版本中生效。 return STATUS_SUCCESS; } #pragma PAGEDCODE // IRQL优先级高的使用分页内存，反之使用非分页内存。（维持系统稳定性） // 代码放在代码段，数据放在数据段。 NTSTATUS IO_CardAddDevice( // IN、OUT和INOUT是空宏定义，主要起注释作用。 IN PDRIVER_OBJECT DriverObject, // I/O管理器传递过来的驱动对象。 IN PDEVICE_OBJECT PhysicalDeviceObject // 从I/O管理器传递过来的物理设备对象。 ) { PAGED_CODE(); // 它会检查当前函数是否运行低于DISPATCH_LEVEL的中断请求级，如果等于或高于这个中断请求级，将产生一个断言。它只在check版本中生效。 KdPrint(("Enter IO_CardAddDevice\n")); // 调试信息打印函数，它只在check版本中生效。 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); // 附着FDO到PDO上，并且把返回的PDO记录在FDO的扩展对象中。 // 创建设备接口。（不能指定设备名） status = IoRegisterDeviceInterface(PhysicalDeviceObject, &IO_CARD_DEVICE, NULL, &pdx->interfaceName); if( !NT_SUCCESS(status)) { IoDeleteDevice(fdo); // 失败则删除FDO。 return status; } IoSetDeviceInterfaceState(&pdx->interfaceName, TRUE); // 使设备名有效。 if( !NT_SUCCESS(status)) { if( !NT_SUCCESS(status)) { return status; } } KdPrint(("%wZ\n",&pdx->interfaceName)); // 调试信息打印函数，它只在check版本中生效。 // 设置标志位。 fdo->Flags |= DO_BUFFERED_IO | DO_POWER_PAGABLE; fdo->Flags &= ~DO_DEVICE_INITIALIZING; KdPrint(("Leave IO_CardAddDevice\n")); // 调试信息打印函数，它只在check版本中生效。 return STATUS_SUCCESS; } // 中断响应例程。 BOOLEAN OnInterrupt(PKINTERRUPT InterruptObject, PDEVICE_EXTENSION pdx) { // 关中断。 UCHAR HSR = READ_PORT_UCHAR(pdx->portbase); HSR = HSR | 0x04; WRITE_PORT_UCHAR(pdx->portbase,HSR); KdPrint(("==============interrupt!!!\n")); // 恢复中断信号电平。 WRITE_REGISTER_UCHAR((PUCHAR)pdx->MemBar0+0x400000,0x10); IoRequestDpc(pdx->fdo, NULL, pdx); // 中断处理标志，DPC处理事件。 return TRUE; } #pragma PAGEDCODE // IRQL优先级高的使用分页内存，反之使用非分页内存。（维持系统稳定性） // 代码放在代码段，数据放在数据段。 NTSTATUS InitMyPCI(IN PDEVICE_EXTENSION pdx,IN PCM_PARTIAL_RESOURCE_LIST list) // 主要用于将系统分配的资源初始化。 { PHYSICAL_ADDRESS portbase; // 端口物理地址。 BOOLEAN gotport = FALSE; ULONG vector; // 中断向量。 KIRQL irql; // 中断请求级。 KINTERRUPT_MODE mode; // 中断模式。 KAFFINITY affinity; // CPU的亲缘关系。 BOOLEAN irqshare; // 是否共享中断。 BOOLEAN gotinterrupt = FALSE; ULONG nres = list->Count; // 资源总数。 PCM_PARTIAL_RESOURCE_DESCRIPTOR resource = &list->PartialDescriptors[0]; // 每次循环获取一种资源。 for (ULONG i = 0; i < nres; ++i, ++resource) { // 判断是何种资源。 switch(resource->Type) { // I/O端口资源。 case CmResourceTypePort: // I/O端口地址。 portbase = resource->u.Port.Start; // I/O端口地址长度。 pdx->nports = resource->u.Port.Length; // 是否需要地址映射。 pdx->mappedport = (resource->Flags & CM_RESOURCE_PORT_IO) == 0; // 表示已经得到I/O端口资源。 gotport = TRUE; break; // 物理内存资源。 case CmResourceTypeMemory: pdx->MemBar0 = (PUCHAR)MmMapIoSpace(resource->u.Memory.Start,resource->u.Memory.Length,MmNonCached); pdx->nMem0 = resource->u.Memory.Length; break; // 中断资源。 case CmResourceTypeInterrupt: // 获得中断请求级。 irql = (KIRQL) resource->u.Interrupt.Level; // 获得中断向量。 vector = resource->u.Interrupt.Vector; // 获取CPU亲缘关系。 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; } } if (!(TRUE&& gotport&& gotinterrupt)) { KdPrint((" Didn't get expected I/O resources\n")); return STATUS_DEVICE_CONFIGURATION_ERROR; } // 判断是够需要I/O端口映射。（为了操作方便，把IO空间映射到MEM空间） if (pdx->mappedport) { // 获得I/O端口地址。 pdx->portbase = (PUCHAR) MmMapIoSpace(portbase, pdx->nports, MmNonCached); if (!pdx->mappedport) { KdPrint(("Unable to map port range %I64X, length %X\n", portbase, pdx->nports)); return STATUS_INSUFFICIENT_RESOURCES; } } else // 获得I/O端口地址。 pdx->portbase = (PUCHAR) portbase.QuadPart; // 链接中断。（中断对象与中断响应例程链接） NTSTATUS status = IoConnectInterrupt(&pdx->InterruptObject, (PKSERVICE_ROUTINE)OnInterrupt, (PVOID)pdx, NULL, vector, irql, irql, LevelSensitive, TRUE, affinity, FALSE); if (!NT_SUCCESS(status)) { KdPrint(("IoConnectInterrupt failed - %X\n", status)); if (pdx->portbase && pdx->mappedport) MmUnmapIoSpace(pdx->portbase, pdx->nports); pdx->portbase = NULL; return status; } return STATUS_SUCCESS; } #pragma LOCKEDCODE // IRQL优先级高的使用分页内存，反之使用非分页内存。（维持系统稳定性） // 代码放在代码段，数据放在数据段。 NTSTATUS OnRequestComplete(PDEVICE_OBJECT junk, PIRP Irp, PKEVENT pev) { KeSetEvent(pev, 0, FALSE); // 把IRP事件设置为完成。 return STATUS_MORE_PROCESSING_REQUIRED; // 再次完成IRP事件 。 } #pragma PAGEDCODE // IRQL优先级高的使用分页内存，反之使用非分页内存。（维持系统稳定性） // 代码放在代码段，数据放在数据段。 // 上层设备的完成例程设置在下层设备IO堆栈。 NTSTATUS ForwardAndWait(PDEVICE_EXTENSION pdx, PIRP Irp) { PAGED_CODE(); // 它会检查当前函数是否运行低于