Windows驱动
+ -

自定义IRP的完成与处理

2021-07-01 320 0

一般情况下,当我们在windows驱动中收到IRP时,驱动根据实际情况分为几种情况:
第一种是直接完成填充相关的参数后,直接调用IoCompleteRequest完成。
第二种情况是该层做相关的设置,如设置完成例程,然后调用IoCallDriver进行下发。
在第二种情况下,一般是当在IRP_MN_START时,我们设置IoSetCompletionRoutine,并下发:

NTSTATUS
SyncForwardIrp(
    PDEVICE_OBJECT DeviceObject,
    PIRP Irp)
{
    NTSTATUS status = STATUS_SUCCESS;

    KEVENT event;
    KeInitializeEvent(&event, NotificationEvent, FALSE);

    IoCopyCurrentIrpStackLocationToNext(Irp);
    IoSetCompletionRoutine(Irp,
        (PIO_COMPLETION_ROUTINE)IrpCompletionRoutine,
        (PVOID) & event,
        TRUE,
        TRUE,
        TRUE);

    status = IoCallDriver(DeviceObject, Irp);
    if (status == STATUS_PENDING)
    {
        KeWaitForSingleObject(&event,
            Executive,
            KernelMode,
            FALSE,
            NULL);

        status = Irp->IoStatus.Status;
    }

    return status;
}

这里我们等待下层的驱动完成该IRP时,调用我们的回调函数,我们通过返回STATUS_MORE_PROCESSING_REQUIRED重新获取IPR的使用权,当然有的时候我们也会返回STATUS_CONTINUE_COMPLETION继续完成IRP.
这种情况一般都是我们最常见的IRP派遣与完成。

IRP完成例程与IO_STACK_LOCATION揭密

但有的时候,我们的需求变化了,就是当我们收到上层下发的IRP时,需要将该IRP pending,需要重新生成新的IRP,然后再下发给底层驱动,等待其完成,这时,它的完成函数该怎么处理呢?

  1. 使用IoAllocateIrp创建IRP.
  2. 使用IoGetNextIrpStackLocation获取下层的堆栈,并设置对应的IRP参数。
  3. 使用 IoSetCompletionRoutine 加入下层设备完成该IRP时的回调函数。
  4. 使用IoCallDriver下发新创建的IRP到底层设备

我们的初想是当下层完成IRP时,我们只在完成函数中获取相关的数据即可,这时可以返回STATUS_CONTINUE_COMPLETION。不过当设置该标志时,系统会蓝屏。
这是什么原因了?
这主要的原因是我们未设置当前StackLocation相关的参数所致,所以肯定是蓝到IoCompleteRequest中了。所以如果想要处理这类问题,就必须设置为STATUS_MORE_PROCESSING_REQUIRED,重新获取IRP权限,使IoCompleteRequest中断,这样就从下层的StackLocation中直接返回即可。
然后这时可以通过释放该IRP创建的其它资源和调用IoFreeIrp释放IRP即可。

如我们以下发一个USB中断请求为例:

 NewIrp =IoAllocateIrp(DeviceObject->StackSize, FALSE);
...

NewIoStack = IoGetNextIrpStackLocation(NewIrp);
NewIoStack->MajorFunction = IRP_MJ_INTERNAL_DEVICE_CONTROL;
NewIoStack->Parameters.DeviceIoControl.IoControlCode = IOCTL_INTERNAL_USB_SUBMIT_URB;
NewIoStack->Parameters.Others.Argument1 = NewUrb;

...
IoSetCompletionRoutine(NewIrp, NewIrpCompleteRoutine, Context, TRUE, TRUE, TRUE);
status = IoCallDriver(FDODeviceExtension->NextDeviceObject, NewIrp);
...
    }

而完成例程:

NTSTATUS
NewIrpCompleteRoutine(
    PDEVICE_OBJECT DeviceObject,
    PIRP Irp,
    PVOID Context
)
{
...
    ExFreePool(urb);
    IoFreeIrp(Irp);
    return STATUS_MORE_PROCESSING_REQUIRED;
}

0 篇笔记 写笔记

自定义IRP的回收利用
前面说到,我们有时需要自已生成新的IPR并派发到下层驱动,所以这里的IRP派发一般的情况下是频繁的,所以上节中会出现频敏的IoAllocateIrp和IoFreeIrp。这样会造成性能上的损失,那有没有更好的处理方式了。答案是肯定的:在说答案之前,我们先介绍一个函数:IoReuseIrp,其REAC......
IRP的完成IoCompleteRequest
每当一个IRP在下层设备层完成时,是需要调用IoCompleteRequest来实现IRP的完成,这个完成其实是实现对执行的IRP的善后操作,这个操作其实是一个宏,真实函数数是IofCompleteRequest。#define IoCompleteRequest IofCompleteReque......
Windows IRP结构字段解释
 比如说你开发了一个USB接口的设备, 那么为了在windows上使用它, 你需要开发一个驱动程序,也许还需要一个应用程序.有些硬件是"免驱"的.但事实上不是不需要驱动,而是他所需要的驱动已经存在在windows中了,不需要你另外安装而已.  你使用应用程序可以控制该硬件的行为,过程是应用程序发出......
自定义IRP的完成与处理
一般情况下,当我们在windows驱动中收到IRP时,驱动根据实际情况分为几种情况:第一种是直接完成填充相关的参数后,直接调用IoCompleteRequest完成。第二种情况是该层做相关的设置,如设置完成例程,然后调用IoCallDriver进行下发。在第二种情况下,一般是当在IRP_MN_STA......
IRP与LIST_ENTRY的关联使用
IRP是Windows内核中的一个很重要的概念,代表着应用或内核对设备的请求。LIST_ENTRY又是Windows内核提供的一个链表。IRP挂入LIST_ENTRYVOID RecyleFrame(PFDO_DEVICE_EXTENSION deviceExtension, FrameHead......
IoCopyCurrentIrpStackLocationToNext和IoSkipCurrentIrpStackLocation操作的IO_STACK_LOCATION有什么区别
在Windows驱动中,传递IPR一般有两种操作:一种是调用IoSkipCurrentIrpStackLocation,表示跳过本层驱动的操作,直接转发至下层: IoSkipCurrentIrpStackLocation(Irp); return IoCallDriver(FDODeviceEx......
IRP完成APC执行函数IopCompleteRequest
IRP在完成时调用IoCompleteRequest,其最终会执行一个APC调用,该调用的函数名为IopCompleteRequest。其调用APC调用时的代码如下:KeInitializeApc(&Irp->Tail.Apc, &......
IRP完成与IO_STACK_LOCATION
IRP与IO_STACK_LOCATION的关系IPR完成时,有时为了获取数据,我们经常要这样干 IoCopyCurrentIrpStackLocationToNext(Irp); IoSetCompletionRoutine( Irp, Comple......
IRP无法取消的无限等待问题
有时有这样一种功能,就是当上层设备或应用和我们开发的驱动的时候,而我们这时根本无法完成本次的IRP,所以根据WINDOWS 驱动开发规范,我们这时需要对该IRP进行MarkPending,然后返回Pending。NTSTATUSMyDispatch( PDEVICE_OBJECT Dev......
IRP完成例程IoSetCompletionRoutine的设计和实现原理
在进行IRP下层传递时,通过上一节可知道,一种中使用IoCopyCurrentIrpStackLocationToNext,另一种是IoSkipCurrentIrpStackLocation。其中在使用IoCopyCurrentIrpStackLocationToNext表示的是对当前的IRP当留当......
IRP的取消机制IoCancelIrp
在驱动层,可以使用取消例程实现对IRP的取消。IRP的取消主要涉及几个函数,分别是:IoSetCancelRoutine 设置取消IRP例程IoCancelIrp函数看起来比较简单,但过程却是并不简单。REACTOS提供的IoCancelIrp的源代码如下:BOOLEAN NTAPI IoC......
Windows x64 IRP结构体成员偏移地址
4: kd> dt _IRPntdll!_IRP +0x000 Type : Int2B +0x002 Size : Uint2B +0x004 AllocationProcessorNumber : Uint2B +0......
PDO设备的动态创建与卸载IRP_MN_SURPRISE_REMOVAL和IRP_MN_REMOVE_DEVICE
总线驱动创建的PDO在卸载时,因卸载方式不同而不同如果是直接卸载总线,会只调用IRP_MN_REMOVE_DEVICE而如果是动态创建与卸载,在卸载的时候调用IoInvalidateDeviceRelations会导致IRP_MN_SURPRISE_REMOVAL的调用,然后调用IRP_MN_R......
挂起的IPR在卸载驱动中的处理
有时有一种情况,说是会将上层如CAMERA,HID等USB设备下发的IPR进行入链表LIST_ENTRY,当然在入之前,会将此IRP 挂起pending,并设置取消例程。不过在驱动卸载的时候,或者某种情况下需要手动清除这些IRP时,会出现和CANCEL例程竞争的情况。故这些需要进行一些判断再处理:......
IPR超时手动取消
手动发一个IRP下底层,可以设置超时时间,当超时后,可以手动取消该IRP.NTSTATUS HidpCallDriverSynchronous(IN PDEVICE_OBJECT DeviceObject, IN OUT PIRP Irp){ KEVENT event; NTST......
作者信息
我爱内核
Windows驱动开发,网站开发
好好学习,天天向上。
取消
感谢您的支持,我会继续努力的!
扫码支持
扫码打赏,你说多少就多少

打开支付宝扫一扫,即可进行扫码打赏哦

您的支持,是我们前进的动力!