Window XP驱动开发(十五) 驱动程序调用驱动程序(以文件句柄形式)

转载请标明是引用于 http://blog.csdn.net/chenyujing1234

欢迎大家提出意见,一起讨论!

代码及EzDriverInstaller下载地址 : http://www.rayfile.com/zh-cn/files/9376d678-b9e1-11e1-9cc9-0015c55db73d/

(编译环境:VS2008+DDK库(参考:Window XP驱动开发(十六) XP下新建驱动程序工程并编译的第二种方法))

在驱动程序开发中,经常需要一个驱动程序调用另一个驱动程序。

例如:

虚拟串口转USB设备的驱动程序,这种驱动程序首先创建一个虚拟串口设备,对这个虚拟串口设备的读写请求会转到一个USB设备上去,

这时需要在虚拟串口驱动程序中调用USB驱动程序。

1、 以文件句柄形式调用其他驱动程序

这种方法类似于在应用程序中调用驱动程序。这种方法使用简单,不需要程序员对Windows底层了解过多知识。

1、1 准备一个标准驱动DriverA

这个“目标”驱动程序创建一个模拟设备,模拟设备支持异步读取操作。事先规定每次对设备读取要耗时3S,因为这样可以很好地演示异步读取操作。

对于这样的驱动程序,应该设置一个定时器,定时器的间隔为3S。另外在IRP_MJ_READ的派遣函数中不结束IRP请求,而是将IRP请求挂起,并且在派遣函数

退出前开启定时器。这样3S后就会进入定时器的回调函数中,并在回调函数中结束IRP请求。

我们把这个“目标”驱动程序命名为DriverA,调用DriverA的驱动程序被命名为DriverB,以下列出DriverA的部分代码:

首先是DriverA的IRP_MJ_READ派遣函数。

/************************************************************************ * 函数名称:HelloDDKRead * 功能描述:对读IRP进行处理 * 参数列表: pDevObj:功能设备对象 pIrp:从IO请求包 * 返回 值:返回状态 *************************************************************************/ #pragma PAGEDCODE NTSTATUS HelloDDKRead(IN PDEVICE_OBJECT pDevObj, IN PIRP pIrp) { KdPrint(("DriverA:Enter A HelloDDKRead\n")); PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION) pDevObj->DeviceExtension; //将IRP设置为挂起 IoMarkIrpPending(pIrp); //将挂起的IRP记录下来 pDevExt->currentPendingIRP = pIrp; //定义3秒后将IRP_MJ_READ的IRP完成 ULONG ulMicroSecond = 3000000; //将32位整数转化成64位整数 LARGE_INTEGER timeout = RtlConvertLongToLargeInteger(-10*ulMicroSecond); // 开启定时器 KeSetTimer( &pDevExt->pollingTimer, timeout, &pDevExt->pollingDPC ); KdPrint(("DriverA:Leave A HelloDDKRead\n")); //返回pending状态 return STATUS_PENDING; }

DriverA的超时回调DPC例程,此例程在CreateDevice创建设备时安装的。

// DriverA的超时回调DPC例程 VOID OnTimerDpc( IN PKDPC pDpc, IN PVOID pContext, IN PVOID SysArg1, IN PVOID SysArg2 ) { PDEVICE_OBJECT pDevObj = (PDEVICE_OBJECT)pContext; PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)pDevObj->DeviceExtension; PIRP currentPendingIRP = pdx->currentPendingIRP; KdPrint(("DriverA:complete the Driver A IRP_MJ_READ irp!\n")); //设置完成状态为STATUS_CANCELLED currentPendingIRP->IoStatus.Status = STATUS_SUCCESS; currentPendingIRP->IoStatus.Information = 0; // bytes xfered IoCompleteRequest( currentPendingIRP, IO_NO_INCREMENT ); }


/************************************************************************ * 函数名称:CreateDevice * 功能描述:初始化设备对象 * 参数列表: pDriverObject:从I/O管理器中传进来的驱动对象 * 返回 值:返回初始化状态 *************************************************************************/ #pragma INITCODE NTSTATUS CreateDevice ( IN PDRIVER_OBJECT pDriverObject) { NTSTATUS status; PDEVICE_OBJECT pDevObj; PDEVICE_EXTENSION pDevExt; //创建设备名称 UNICODE_STRING devName; RtlInitUnicodeString(&devName,L"\\Device\\MyDDKDeviceA"); //创建设备 status = IoCreateDevice( pDriverObject, sizeof(DEVICE_EXTENSION), &(UNICODE_STRING)devName, FILE_DEVICE_UNKNOWN, 0, TRUE, &pDevObj ); if (!NT_SUCCESS(status)) return status; pDevObj->Flags |= DO_BUFFERED_IO; pDevExt = (PDEVICE_EXTENSION)pDevObj->DeviceExtension; pDevExt->pDevice = pDevObj; pDevExt->ustrDeviceName = devName; // 分配定时器及设置超时回调例程 KeInitializeTimer( &pDevExt->pollingTimer ); KeInitializeDpc( &pDevExt->pollingDPC, OnTimerDpc, (PVOID) pDevObj ); //创建符号链接 UNICODE_STRING symLinkName; RtlInitUnicodeString(&symLinkName,L"\\??\\HelloDDKA"); pDevExt->ustrSymLinkName = symLinkName; status = IoCreateSymbolicLink( &symLinkName,&devName ); if (!NT_SUCCESS(status)) { IoDeleteDevice( pDevObj ); return status; } return STATUS_SUCCESS; }


以上就是准备的“目标”驱动程序DriverA ,为了演示DriverB调用DriverA,并使读者可以清楚log消息来源于哪个驱动程序,我们

让DriverA在输出log信息上加“DriverA:”这个前缀,而让DriverB在输出log信息之前加上“DriverB:”这个前缀。

DriverB有多种方法调用DriverA,这些方法可以是同步调用,也可以是异步调用。

1、1、1 DriverA的安装文件Inf的书写
;--------- 版本区域性 --------------------------------------------------- [Version] Signature="$CHICAGO$"; Provider=Zhangfan_DeviceA DriverVer=11/1/2007,3.0.0.3 ; 如果设备是一个标准类别,使用标准类别名称和GUI ; 否则创建一个自定义的类别名称,并自定义它的GUID Class=ZhangfanDeviceB ClassGUID={EF2962F0-0D55-4bff-B8AA-2221EE8A79B1} ;--------- 安装磁盘节 ----------------------- ; 这此节确定安装盘和安装文件的路径 ; 读者可以按照自己的需要修改 [SourceDisksNames] 1 = "HelloDDKA",Disk1,, [SourceDisksFiles] HelloDDKA.sys = 1,MyDriver_Check, ;--------- ClassInstall/ClassInstall32 Section ------------------------------- ; 如果使用标准类别设备,下面是不需要的 ; 9X Style [ClassInstall] Addreg=Class_AddReg ; NT Style [ClassInstall32] Addreg=Class_AddReg [Class_AddReg] HKR,,,,%DeviceClassName% HKR,,Icon,,"-5" ;--------- 目标文件节------------------------------------------- [DestinationDirs] YouMark_Files_Driver = 10,System32\Drivers [DefaultInstall] ;默认的安装。(即右击后点安装后的执行,有些人讲点安装了,可文件还是没有复制过去,就是这个没加的原因。) Copyfiles=YouMark_Files_Driver AddReg=YouMark_9X_AddReg ;--------- 制造商节 ---------------------------------- [Manufacturer] %MfgName%=Mfg0 [Mfg0] ; PCI hardware Ids use the form ; PCI\VEN_aaaa&DEV_bbbb&SUBSYS_cccccccc&REV_dd ;改成你自己的ID %DeviceDesc%=YouMark_DDI, PCI\VEN_9999&DEV_9999 ;---------- DDInstall Sections ----------------------------------------------- ; --------- Windows 9X ----------------- ; 如果在DDInstall中的字符串超过19,将会导致严重的问题 [YouMark_DDI] CopyFiles=YouMark_Files_Driver AddReg=YouMark_9X_AddReg [YouMark_9X_AddReg] HKR,,DevLoader,,*ntkern HKR,,NTMPDriver,,HelloDDKA.sys HKR, "Parameters", "BreakOnEntry", 0x00010001, 0 ; --------- Windows NT ----------------- [YouMark_DDI.NT] CopyFiles=YouMark_Files_Driver AddReg=YouMark_NT_AddReg [YouMark_DDI.NT.Services] Addservice = HelloDDKA, 0x00000002, YouMark_AddService [YouMark_AddService] DisplayName = %SvcDesc% ServiceType = 1 ; SERVICE_KERNEL_DRIVER StartType = 3 ; SERVICE_DEMAND_START ErrorControl = 1 ; SERVICE_ERROR_NORMAL ServiceBinary = %10%\System32\Drivers\HelloDDKA.sys [YouMark_NT_AddReg] HKLM, "System\CurrentControlSet\Services\HelloDDKA\Parameters",\ "BreakOnEntry", 0x00010001, 0 ; --------- 文件 (common) ------------- [YouMark_Files_Driver] HelloDDKA.sys ;--------- 字符串节--------------------------------------------------- [Strings] ProviderName="ZhangfanA." MfgName="Zhangfan SoftA" DeviceDesc="Hello World DDKA!" DeviceClassName="Zhangfan_DeviceA" SvcDesc="ZhangfanA"


1、2 以文件句柄形式调用其他驱动程序的知识点
1、2、1 获得设备句柄的两种方法

(1) 通过ZwCreateFile,以设备名来打开

在驱动程序中,打开设备使用ZwCreateFile内核函数,它会返回设备句柄。这里要讨论一下如何用

ZwCreateFile内核函数打开“同步”设备和“异步”设备。ZwCreateFile内核函数声明如下:

NTSYSAPI NTSTATUS NTAPI ZwCreateFile( OUT PHANDLE FileHandle, IN ACCESS_MASK DesiredAccess, IN POBJECT_ATTRIBUTES ObjectAttributes, OUT PIO_STATUS_BLOCK IoStatusBlock, IN PLARGE_INTEGER AllocationSize OPTIONAL, IN ULONG FileAttributes, IN ULONG ShareAccess, IN ULONG CreateDisposition, IN ULONG CreateOptions, IN PVOID EaBuffer OPTIONAL, IN ULONG EaLength );

(2)

很多情况下,使用都不容易知道具体的设备名,而只知道符号链接。例如“C:"代表第一个硬件分区,而“C:”就是第一个符号链接,

它指向第一个磁盘分区设备。尤其在WDM驱动设备中,通过符号链接打开设备是经常遇到的。

利用ZwOpenSymbolicLinkObject内核函数先得到符号链接的句柄

然后使用ZwQuerySymbolicLinkObject内核函数查找到设备名

通过设备名就可以方便地打开设备。

NTSTATUS HelloDDKRead(IN PDEVICE_OBJECT pDevObj, IN PIRP pIrp) { KdPrint(("DriverB:Enter B HelloDDKRead\n")); NTSTATUS ntStatus = STATUS_SUCCESS; UNICODE_STRING DeviceSymbolicLinkName; RtlInitUnicodeString( &DeviceSymbolicLinkName, L"\\??\\HelloDDKA" ); //初始化objectAttributes OBJECT_ATTRIBUTES objectAttributes; InitializeObjectAttributes(&objectAttributes, &DeviceSymbolicLinkName, OBJ_CASE_INSENSITIVE|OBJ_KERNEL_HANDLE, NULL, NULL ); HANDLE hSymbolic; //设定了FILE_SYNCHRONOUS_IO_NONALERT或者FILE_SYNCHRONOUS_IO_ALERT为同步打开设备 ntStatus = ZwOpenSymbolicLinkObject(&hSymbolic,FILE_ALL_ACCESS,&objectAttributes); #define UNICODE_SIZE 50 UNICODE_STRING LinkTarget; LinkTarget.Buffer = (PWSTR)ExAllocatePool(PagedPool,UNICODE_SIZE); LinkTarget.Length = 0; LinkTarget.MaximumLength = UNICODE_SIZE; ULONG unicode_length; // 通过符号链接得到设备名 ntStatus = ZwQuerySymbolicLinkObject(hSymbolic,&LinkTarget,&unicode_length); KdPrint(("DriverB:The device name is %wZ\n",&LinkTarget)); // 构造objectAttributes InitializeObjectAttributes(&objectAttributes, &LinkTarget, OBJ_CASE_INSENSITIVE, NULL, NULL ); HANDLE hDevice; IO_STATUS_BLOCK status_block; //设定了FILE_SYNCHRONOUS_IO_NONALERT或者FILE_SYNCHRONOUS_IO_ALERT为同步打开设备 ntStatus = ZwCreateFile(&hDevice, FILE_READ_ATTRIBUTES|SYNCHRONIZE, &objectAttributes, &status_block, NULL,FILE_ATTRIBUTE_NORMAL,FILE_SHARE_READ, FILE_OPEN_IF,FILE_SYNCHRONOUS_IO_NONALERT,NULL,0); // 判断是否成功打开设备 if (NT_SUCCESS(ntStatus)) { ZwReadFile(hDevice,NULL,NULL,NULL,&status_block,NULL,0,NULL,NULL); } ZwClose(hDevice); ZwClose(hSymbolic); ExFreePool(LinkTarget.Buffer); ntStatus = STATUS_SUCCESS; // 完成IRP pIrp->IoStatus.Status = ntStatus; pIrp->IoStatus.Information = 0; // bytes xfered IoCompleteRequest( pIrp, IO_NO_INCREMENT ); KdPrint(("DriverB:Leave B HelloDDKRead\n")); return ntStatus; }


1、2、2 打开方式的两种方式

现在我们介绍打开“同步”和“异步”设备的区别。

(1)如果打开“同步”设备,第二个参数DesiredAccess需要设置为SYNCHRONIZE,

且倒数第三个参数CreateOptions需要指定为FILE_SYNCHRONOUS_IO_NONALERT或FILE_SYNCHRONOUS_IO_ALERT

(2) 如果打开“异步”设备,第二个参数DesiredAccess不能设置为SYNCHRONIZE,并且倒数第三个参数CreateOptions不能指定为

FILE_SYNCHRONOUS_IO_NONALERT或FILE_SYNCHRONOUS_IO_ALERT。

//异步打开设备 //没有设定FILE_SYNCHRONOUS_IO_NONALERT或者FILE_SYNCHRONOUS_IO_ALERT为异步打开设备 ntStatus = ZwCreateFile(&hDevice, FILE_READ_ATTRIBUTES, &objectAttributes, &status_block, NULL,FILE_ATTRIBUTE_NORMAL,FILE_SHARE_READ, FILE_OPEN_IF,0,NULL,0);

1、3 以文件句柄形式调用的第一种方法:同步调用

在应用程序中,ReadFile函数既可以来读取文件,又可以来读取设备。同样,在驱动程序中,ZwReadFile内核函数既可以来读文件,也可以来读设备。

如果用ZwReadFile内核函数同步读取设备,操作对应的IRP_MJ_READ请求被结束后,函数才会返回。否则这个函数会一直等待IRP_MJ_READ请求被结束。

如果用ZwReadFile内核函数异步读取设备,即使操作对应的IRP_MJ_READ请求没有被结束,函数也会立即返回。

本节介绍的DriverB是利用同步读取的方法调用DriverA,如用ZwReadFile内核函数同步读取DriverA的设备,它的内部操作过程如下:

(1)在DriverB中用ZwReadFile内核函数读取DriverA的设备对象。ZwReadFile内核函数内部会创建IRP_MJ_READ类型的IRP,然后将这个IRP

当做参数传递给DriverA的派遣函数。

(2)DriverA的派遣函数没有结束IRP请求,而是将IRP请求“挂起”。

(3)ZwReadFile函数会一直等待IRP中的一个事件,此时当前线程进入睡眠状态。

(4)3S后,触发DriverA 的定时器例器,这里IRP请求被结束,IRP中的相关事件也被设置。

(5)由于相应事件被设置,刚才休眠的线程恢复运行,ZwReadFile内核函数退出。

1、3、1 DriverB设计

我们是采用同步打开设备的方式。

NTSTATUS HelloDDKRead(IN PDEVICE_OBJECT pDevObj, IN PIRP pIrp) { KdPrint(("DriverB:Enter B HelloDDKRead\n")); NTSTATUS ntStatus = STATUS_SUCCESS; UNICODE_STRING DeviceName; RtlInitUnicodeString( &DeviceName, L"\\Device\\MyDDKDeviceA" ); //初始化objectAttributes OBJECT_ATTRIBUTES objectAttributes; InitializeObjectAttributes(&objectAttributes, &DeviceName, OBJ_CASE_INSENSITIVE, NULL, NULL ); HANDLE hDevice; IO_STATUS_BLOCK status_block; //同步打开设备 //设定了FILE_SYNCHRONOUS_IO_NONALERT或者FILE_SYNCHRONOUS_IO_ALERT为同步打开设备 ntStatus = ZwCreateFile(&hDevice, FILE_READ_ATTRIBUTES|SYNCHRONIZE, &objectAttributes, &status_block, NULL,FILE_ATTRIBUTE_NORMAL,FILE_SHARE_READ, FILE_OPEN_IF,FILE_SYNCHRONOUS_IO_NONALERT,NULL,0); if (NT_SUCCESS(ntStatus)) { ZwReadFile(hDevice,NULL,NULL,NULL,&status_block,NULL,0,NULL,NULL); } ZwClose(hDevice); // 结束来自应用程序的IRP请求。完成IRP pIrp->IoStatus.Status = ntStatus; pIrp->IoStatus.Information = 0; // bytes xfered IoCompleteRequest( pIrp, IO_NO_INCREMENT ); KdPrint(("DriverB:Leave B HelloDDKRead\n")); return ntStatus; }

1、3、2 应用程序设计

另外,DriverB还需要做应用程序的客户端,因此DriverB的派遣函数还需要结束来自应用程序的IRP请求。

int main() { HANDLE hDevice = CreateFile("\\\\.\\HelloDDKB", GENERIC_READ | GENERIC_WRITE, 0, // share mode none NULL, // no security OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL ); // no template if (hDevice == INVALID_HANDLE_VALUE) { printf("Failed to obtain file handle to device " "with Win32 error code: %d\n", GetLastError() ); return 1; } DWORD dRet; ReadFile(hDevice,NULL,0,&dRet,NULL); CloseHandle(hDevice); return 0; }


1、3、3DriverB的安装文件inf的书写
;--------- 版本区域性 --------------------------------------------------- [Version] Signature="$CHICAGO$"; Provider=Zhangfan_DeviceB DriverVer=11/1/2007,3.0.0.3 ; 如果设备是一个标准类别,使用标准类别名称和GUI ; 否则创建一个自定义的类别名称,并自定义它的GUID Class=ZhangfanDeviceB ClassGUID={EF3962F0-1D55-4bff-B8AA-2221EE8A79B3} ;--------- 安装磁盘节 ----------------------- ; 这此节确定安装盘和安装文件的路径 ; 读者可以按照自己的需要修改 [SourceDisksNames] 1 = "HelloDDKB",Disk2,, [SourceDisksFiles] HelloDDKB.sys = 1,MyDriver_Check, ;--------- ClassInstall/ClassInstall32 Section ------------------------------- ; 如果使用标准类别设备,下面是不需要的 ; 9X Style [ClassInstall] Addreg=Class_AddReg ; NT Style [ClassInstall32] Addreg=Class_AddReg [Class_AddReg] HKR,,,,%DeviceClassName% HKR,,Icon,,"-5" ;--------- 目标文件节------------------------------------------- [DestinationDirs] YouMark_Files_Driver = 10,System32\Drivers [DefaultInstall] ;默认的安装。(即右击后点安装后的执行,有些人讲点安装了,可文件还是没有复制过去,就是这个没加的原因。) Copyfiles=YouMark_Files_Driver AddReg=YouMark_9X_AddReg ;--------- 制造商节 ---------------------------------- [Manufacturer] %MfgName%=Mfg0 [Mfg0] ; PCI hardware Ids use the form ; PCI\VEN_aaaa&DEV_bbbb&SUBSYS_cccccccc&REV_dd ;改成你自己的ID %DeviceDesc%=YouMark_DDI, PCI\VEN_1189&DEV_2398 ;---------- DDInstall Sections ----------------------------------------------- ; --------- Windows 9X ----------------- ; 如果在DDInstall中的字符串超过19,将会导致严重的问题 [YouMark_DDI] CopyFiles=YouMark_Files_Driver AddReg=YouMark_9X_AddReg [YouMark_9X_AddReg] HKR,,DevLoader,,*ntkern HKR,,NTMPDriver,,HelloDDKB.sys HKR, "Parameters", "BreakOnEntry", 0x00010002, 0 ; --------- Windows NT ----------------- [YouMark_DDI.NT] CopyFiles=YouMark_Files_Driver AddReg=YouMark_NT_AddReg [YouMark_DDI.NT.Services] Addservice = HelloDDKB, 0x00000003, YouMark_AddService [YouMark_AddService] DisplayName = %SvcDesc% ServiceType = 1 ; SERVICE_KERNEL_DRIVER StartType = 3 ; SERVICE_DEMAND_START ErrorControl = 1 ; SERVICE_ERROR_NORMAL ServiceBinary = %10%\System32\Drivers\HelloDDKB.sys [YouMark_NT_AddReg] HKLM, "System\CurrentControlSet\Services\HelloDDKB\Parameters",\ "BreakOnEntry", 0x00010001, 0 ; --------- 文件 (common) ------------- [YouMark_Files_Driver] HelloDDKB.sys ;--------- 字符串节--------------------------------------------------- [Strings] ProviderName="ZhangfanB." MfgName="Zhangfan SoftB" DeviceDesc="Hello World DDKB!" DeviceClassName="Zhangfan_DeviceB" SvcDesc="ZhangfanB"

1、3、4 加载驱动测试

由于DriverA与DriverB都是虚拟设备,通过设备管理器里的硬件更新向导加载驱动是可以的。

但是为了方便我们通过EzDriverInstaller工具。它是DriverStudio自带的一个工具软件,用于快速安装WDM程序

(1)打开EzDriverInstaller,选择“File”|“Open”,在弹出的对话框中,选择需要的inf文件。

(2)然后点击“Add New Device”。

此时就会提示成功,并在列表中显示我们的设备名:

打开我们的设备管器看看效果吧:

(3) 运行应用程序,用DebugView查看打印信息。

注:网上很多人反映KdPrint打印不能在DebugView上显示出来。开始我也怀疑KdPrint这个函数不对。

后来才知道是DebugView没设置好。确保Capture菜单下的Log Boot选择已经勾选。

1、4 以文件句柄形式调用的第二种方法:异步调用
1、4、1 异步调用方法一

异步读取主要是指ZwReadFile内核函数没有等待DriverA真正结束IRP请求时,就已经退出。如果用ZwReadFile内核函数异步读取DriverA的设备,

它的内部操作过程如下:

(1)ZwReadFile内核函数内部创建IRP_MJ_READ类型的IRP,然后将这个IRP传递给DriverA的派遣函数。

(2)DriverA派遣函数没有结束IRP请求,而是将IRP"挂起”

(3)ZwReadFile内核函数发现DriverA将IRP_MJ_READ“挂起”,于是它直接返回,返回值是STATUS_PENDING,这就代表读取操作正在进行。

ZwReadFile内核函数退出后,无法得到“挂起”的IRP何时被结束。

因此在调用ZwReadFile内核函数前,可以为IRP设置一个完成例程。当IRP结束时触发这个完成例程。

NTSTATUS HelloDDKRead(IN PDEVICE_OBJECT pDevObj, IN PIRP pIrp) { KdPrint(("DriverB:Enter B HelloDDKRead\n")); NTSTATUS ntStatus = STATUS_SUCCESS; UNICODE_STRING DeviceName; RtlInitUnicodeString( &DeviceName, L"\\Device\\MyDDKDeviceA" ); //初始化objectAttributes OBJECT_ATTRIBUTES objectAttributes; InitializeObjectAttributes(&objectAttributes, &DeviceName, OBJ_CASE_INSENSITIVE, NULL, NULL ); HANDLE hDevice; IO_STATUS_BLOCK status_block; //异步打开设备 //没有设定了FILE_SYNCHRONOUS_IO_NONALERT和FILE_SYNCHRONOUS_IO_ALERT为异步打开设备 ntStatus = ZwCreateFile(&hDevice, FILE_READ_ATTRIBUTES,//没有设SYNCHRONIZE &objectAttributes, &status_block, NULL,FILE_ATTRIBUTE_NORMAL,FILE_SHARE_READ, FILE_OPEN_IF,0,NULL,0); KEVENT event; //初始化事件,用于异步读 KeInitializeEvent(&event,SynchronizationEvent,FALSE); LARGE_INTEGER offset = RtlConvertLongToLargeInteger(0); if (NT_SUCCESS(ntStatus)) { // 用ZwReadFile函数进行异步读取 ntStatus = ZwReadFile(hDevice,NULL,CompleteDriverA_Read,&event,&status_block,NULL,0,&offset,NULL); } // 判断ZwReadFile函数是否返回STATUS_PENDING if (ntStatus==STATUS_PENDING) { KdPrint(("DriverB:ZwReadFile return STATUS_PENDING!\n")); KdPrint(("DriverB:Waiting...")); // 等待IRP被结束 KeWaitForSingleObject(&event,Executive,KernelMode,FALSE,NULL); } // 关闭设备句柄 ZwClose(hDevice); ntStatus = STATUS_SUCCESS; // 设置IRP完成状态 pIrp->IoStatus.Status = ntStatus; pIrp->IoStatus.Information = 0; // bytes xfered // 结束IRP请求 IoCompleteRequest( pIrp, IO_NO_INCREMENT ); KdPrint(("DriverB:Leave B HelloDDKRead\n")); return ntStatus; }


以下是完成例程的代码。完成例程主要是用一个事件通知IRP请求结束。

VOID CompleteDriverA_Read(PVOID context,PIO_STATUS_BLOCK pStatus_block,ULONG) { KdPrint(("DriverB:The Driver A Read completed now!\n")); KeSetEvent((PKEVENT)context,IO_NO_INCREMENT,FALSE); }


1、4、2 异步调用方法二

第二种异步方法和每一种异步方法略有不同。第一种异步读取时,将一个事件的句柄传递给ZwReadFile内核函数,这个事件可以用来通知读取操作何时完成。

而这里介绍的方法不用将事件句柄传递给ZwCreateFile函数,而是将通过文件对象判断读取是否完毕

每打开一个设备,都会伴随存在一个关联的文件对象(FILE_OBJECT)。利用内核函数ObReferenceObjectByHandle可以获得和设备相关的文件对象指针

当IRP_MJ_READ请求被结束后,文件对象的子域Event会被设置,因此用文件对象的Event子域可以当做同步点使用。

这里利用文件对象的子域Event,即FileObject->Event作为同步点。以下是代码:

NTSTATUS HelloDDKRead(IN PDEVICE_OBJECT pDevObj, IN PIRP pIrp) { KdPrint(("DriverB:Enter B HelloDDKRead\n")); NTSTATUS ntStatus = STATUS_SUCCESS; UNICODE_STRING DeviceName; RtlInitUnicodeString( &DeviceName, L"\\Device\\MyDDKDeviceA" ); //初始化objectAttributes OBJECT_ATTRIBUTES objectAttributes; InitializeObjectAttributes(&objectAttributes, &DeviceName, OBJ_CASE_INSENSITIVE, NULL, NULL ); HANDLE hDevice; IO_STATUS_BLOCK status_block; //异步打开设备 ntStatus = ZwCreateFile(&hDevice, FILE_READ_ATTRIBUTES,//没有设SYNCHRONIZE &objectAttributes, &status_block, NULL,FILE_ATTRIBUTE_NORMAL,FILE_SHARE_READ, FILE_OPEN_IF,0,NULL,0); LARGE_INTEGER offset = RtlConvertLongToLargeInteger(0); if (NT_SUCCESS(ntStatus)) { ntStatus = ZwReadFile(hDevice,NULL,NULL,NULL,&status_block,NULL,0,&offset,NULL); } if (ntStatus==STATUS_PENDING) { KdPrint(("DriverB:ZwReadFile return STATUS_PENDING!\n")); PFILE_OBJECT FileObject; // 通过设备句柄找到文件对象指针 ntStatus = ObReferenceObjectByHandle(hDevice, EVENT_MODIFY_STATE, *ExEventObjectType, KernelMode, (PVOID*) &FileObject, NULL); if (NT_SUCCESS(ntStatus)) { KdPrint(("DriverB:Waiting...")); // 等待文件对象中的事件 KeWaitForSingleObject(&FileObject->Event,Executive,KernelMode,FALSE,NULL); KdPrint(("DriverB:Driver A Read IRP completed now!\n")); // 减少文件对象的引用计数 ObDereferenceObject(FileObject); } } ZwClose(hDevice); ntStatus = STATUS_SUCCESS; // 完成IRP pIrp->IoStatus.Status = ntStatus; pIrp->IoStatus.Information = 0; // bytes xfered IoCompleteRequest( pIrp, IO_NO_INCREMENT ); KdPrint(("DriverB:Leave B HelloDDKRead\n")); return ntStatus; }


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值