菜鸟之驱动开发9

在上两节，我们分别介绍了驱动与应用程序通信的缓冲区与直接访问模式，今天介绍第三种方式。我们在上一节的代码上做修改即可。

首先修改ctl_code.h为

#ifndef CTL_CODE #pragma message("\n \n-----------EXEģʽ . Include winioctl.h ") #include<winioctl.h> //CTL_CODE ntddk.h wdm.h #else #pragma message("-\n \n---------SYSģʽ .NO Include winioctl.h ") #endif #define add_code CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_NEITHER,FILE_ANY_ACCESS) #define sub_code CTL_CODE(FILE_DEVICE_UNKNOWN, 0x801, METHOD_NEITHER,FILE_ANY_ACCESS)

即修改访问模式为其它模式，第三个参数。

再修改驱动代码， IRP派遣函数如下

//派遣函数 #pragma PAGECODE NTSTATUS ddk_DispatchRoutine_CONTROL(IN PDEVICE_OBJECT pDevobj,IN PIRP pIrp ) { ULONG info; NTSTATUS status = STATUS_SUCCESS; //得到当前栈指针 PIO_STACK_LOCATION stack = IoGetCurrentIrpStackLocation(pIrp); ULONG mf=stack->MajorFunction;//区分IRP switch (mf) { case IRP_MJ_DEVICE_CONTROL: { KdPrint(("Enter myDriver_DeviceIOControl\n")); //得到输入缓冲区大小 ULONG cbin = stack->Parameters.DeviceIoControl.InputBufferLength; //得到输出缓冲区大小 ULONG cbout = stack->Parameters.DeviceIoControl.OutputBufferLength; //得到IOCTL码 ULONG code = stack->Parameters.DeviceIoControl.IoControlCode; switch (code) { case add_code: { int a,b; KdPrint(("add_code 1111111111111111111\n")); //缓冲区方式IOCTL //获取缓冲区数据 a,b //int * InputBuffer = (int*)pIrp->AssociatedIrp.SystemBuffer; int * InputBuffer=(int*)stack->Parameters.DeviceIoControl.Type3InputBuffer; __try { ProbeForRead(InputBuffer,cbin,__alignof(int)); //判断指针是否可读 __asm { mov eax,InputBuffer mov ebx,[eax] mov a,ebx mov ebx,[eax+4] mov b,ebx } KdPrint(("a=%d,b=%d \n", a,b)); a=a+b; //C、驱动层返回数据至用户层 //操作输出缓冲区 //int* OutputBuffer = (int*)pIrp->AssociatedIrp.SystemBuffer; //int OutputBuffer =(int)MmGetSystemAddressForMdlSafe(pIrp->MdlAddress, NormalPagePriority); int* OutputBuffer=(int*)pIrp->UserBuffer; ProbeForWrite(OutputBuffer,cbout,4); //判断指针是否可写 KdPrint(("OutputBuffer=%x",OutputBuffer)); __asm { mov eax,a mov ebx,OutputBuffer mov [ebx],eax //bufferet=a+b } KdPrint(("a+b=%d \n",a)); } __except(EXCEPTION_EXECUTE_HANDLER) { KdPrint(("指定地址不可读 或者 写 \n")); } //设置实际操作输出缓冲区长度 info = 4; break; } case sub_code: { break; } }//end code switch break; } case IRP_MJ_CREATE: { break; } case IRP_MJ_CLOSE: { break; } case IRP_MJ_READ: { break; } } //对相应的IPR进行处理 pIrp->IoStatus.Information=info;//设置操作的字节数为0，这里无实际意义 pIrp->IoStatus.Status=STATUS_SUCCESS;//返回成功 IoCompleteRequest(pIrp,IO_NO_INCREMENT);//指示完成此IRP KdPrint(("离开派遣函数\n"));//调试信息 return STATUS_SUCCESS; //返回成功 }

注意的地方是获取输入buffer与输出buffer的方式都发生变化了。 得到输入buffer的指针后，我们要判断它是否可读，得到输出buffer的指针后，要判断是否可写，因为经常可能得不到正确的指针。

请参看完整源码。