Windows驱动开发WDM （16）- 完成例程 （重新获得IRP控制权）

前一次（http://blog.csdn.net/zj510/article/details/8350184）学习了完成例程返回STATUS_SUCCESS的情况，也就是驱动调用IoCallDriver后不会再获得IRP。这次来看看再次获得IRP控制权的情况。

 

设置完成例程

先看看设置完成例程那里的代码：

			IoCopyCurrentIrpStackLocationToNext(Irp);

			KEVENT event;
			KeInitializeEvent(&event, NotificationEvent, FALSE);//创建一个event对象。

			//通过第三个参数将event对象传递给完成例程
			IoSetCompletionRoutine(Irp, MyIoCompletion, &event, TRUE, TRUE, TRUE);

			status = IoCallDriver(pdx->NextStackDevice, Irp);

			if (status == STATUS_PENDING)
			{//如果下层驱动是异步完成irp的话，就等待event被触发（完成例程会触发）
			//如果下层驱动是同步完成irp或者其他错误，那么就无需等待了。直接向上返回。
				KeWaitForSingleObject(&event,Executive,KernelMode ,FALSE,NULL);//等待完成例程触发这个事件，最后一个参数是NULL，表示无限等待

				status = Irp->IoStatus.Status;//获取下层驱动的irp状态
			}

			//获取内核模式下的地址，这个地址一定> 0x7FFFFFFF
			char* outBuf = (char*)MmGetSystemAddressForMdlSafe(Irp->MdlAddress, NormalPagePriority);
			outBuf[1] = outBuf[2];//将第三个字节复制到第二个字节

			//虽然在底层驱动已经将IRP完成了，但是由于完成例程返回的是
			//STATUS_MORE_PROCESSING_REQUIRED,因此需要再次调用IoCompleteRequest!
			IoCompleteRequest (Irp, IO_NO_INCREMENT);

跟前一个版本相比，有3个分别：

1. 创建一个事件对象，初始化，并且在IoCallDriver后面等待这个事件；

2. 事件触发后，将输出缓冲的第三个字节赋值给第二个字节，也就是说过滤驱动再更改一下返回的内容；

3. 调用IoCompleteRequest再次完成IRP（因为完成例程返回了STATUS_MORE_PROCESSING_REQUIRED）

IoSetCompletionRoutine的第三个参数对应完成例程的第三个参数。也就是说我们可以传一些数据给完成例程。

VOID IoSetCompletionRoutine(
  _In_      PIRP Irp,
  _In_opt_  PIO_COMPLETION_ROUTINE CompletionRoutine,
  _In_opt_  PVOID Context,
  _In_      BOOLEAN InvokeOnSuccess,
  _In_      BOOLEAN InvokeOnError,
  _In_      BOOLEAN InvokeOnCancel
);

 

完成例程代码

NTSTATUS MyIoCompletion(IN PDEVICE_OBJECT fdo, IN PIRP irp, IN PVOID context)
{
	KdPrint(("Enter MyIoCompletion\n"));

	//判断是否挂起返回，就是说下层驱动是挂起irp，然后再处理irp（异步irp）
	if(irp->PendingReturned)
	{//设置完成例程那里，只有下层驱动返回pending状态才等待事件，那么我们
		//这里只需要挂起返回的时候才设置事件信号
		KeSetEvent((PKEVENT)context, IO_NO_INCREMENT, FALSE);
		
	}

	KdPrint(("Leave MyIoCompletion\n"));

	//返回STATUS_MORE_PROCESSING_REQUIRED，当前驱动将再次获得irp的控制权，
	//同时也需要再次调用IoCompleteRequest来完成irp
	return STATUS_MORE_PROCESSING_REQUIRED;
}

跟前面的版本比较：

1. 当挂起返回的时候，给事件设置信号，这个事件对象是参数传进来的。

2. 返回STATUS_MORE_PROCESSING_REQUIRED

因为完成例程里面设置了信号，那么KeWaitForSingleObject就会返回。

 

测试代码：

跟前一个文章一样，相当的简单，就是起2个线程，发2个IRP请求：

// TestWDMDriver.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <windows.h>
#include <process.h>

#define DEVICE_NAME L"\\\\.\\HelloWDM"
#define DEVICE_NAME2 L"\\\\.\\HelloWDM2"
#define DEVICE_EX L"\\\\.\\HelloWDM_EX"

void Test(void* pParam)
{
	int index = (int)pParam;
	//设置overlapped标志，表示异步打开
	HANDLE hDevice;
	
	hDevice = CreateFile(DEVICE_NAME,GENERIC_READ | GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE,NULL,OPEN_EXISTING,
			FILE_ATTRIBUTE_NORMAL | FILE_FLAG_OVERLAPPED,NULL);

	wprintf(L"CreateFile, name: %s, ret: %x\n", DEVICE_NAME, hDevice);
	
	

	if (hDevice != INVALID_HANDLE_VALUE)
	{
		char inbuf[100] = {0};
		
		sprintf(inbuf, "hello world %d", index);
		char outbuf[100] = {0};
		DWORD dwBytes = 0;

		DWORD dwStart = GetTickCount();

		printf("input buffer: %s\n", inbuf);
		
		OVERLAPPED ol = {0};
		ol.hEvent = CreateEvent(NULL, FALSE, FALSE, NULL);

		BOOL b = DeviceIoControl(hDevice, CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_IN_DIRECT, FILE_ANY_ACCESS), 
			inbuf, strlen(inbuf), outbuf, 100, &dwBytes, &ol);

		printf("DeviceIoControl thread %d, returns %d, last error: %d, used: %d ms, input: %s\n", index, b, GetLastError(), GetTickCount() - dwStart, inbuf);


		WaitForSingleObject(ol.hEvent, INFINITE);

		DWORD dwEnd = GetTickCount();
		//将输出buffer的数据和'm'亦或，看看是否能够得到初始的字符串。
		for (int i = 0; i < strlen(inbuf); i++)
		{
			outbuf[i] = outbuf[i] ^ 'm';
		}

		printf("Verify thread %d, outbuf: %s, used: %d ms\n", index, outbuf, dwEnd - dwStart);

		CloseHandle(hDevice);

	}
	else
		printf("CreateFile failed, err: %x\n", GetLastError());
}


int _tmain(int argc, _TCHAR* argv[])
{
	HANDLE t1 = (HANDLE)_beginthread(Test, 0, (void*)0);
	_beginthread(Test, 0, (void*)1);

	WaitForSingleObject(t1, INFINITE);

	printf("Test ends\n");
	return 0;
}

看看输出结果：

可以看到原来输出是xello world 0和xello world 1，现在第二个字节变成l了。这是因为过滤驱动在下层驱动完成后，又再次更改了输出内容，就是将第三个字节赋值给第二个字节了。

 

代码：http://download.csdn.net/detail/zj510/4913259

目录1： 功能驱动，从控制面板里面安装

目录2： 过滤驱动，直接右键inf文件安装，同时再改一个注册表，参考http://blog.csdn.net/zj510/article/details/8332658

测试代码：一段简单的测试代码，起2个线程，发2个IRP请求。

WDK7600编译驱动，VS2008编译测试代码。