YJX_Driver_028_实战EXE和SYS通信(缓冲模式)

1、

实战EXE和SYS通信
　　A、用户层传入数据-EXE部分代码
　　B、驱动层接收数据并处理-SYS部分代码
　　C、驱动层返回数据至用户层
　　D、用户层获得处理结果

【190】假设驱动部分已经按照我们的思路写好了

【305】创建 VC6 Win32ConsoleApplication 名称为:"test_exe"

　　【345】选择的是“A "Hello, World" application”，非“An empty project”

【400】写代码（应用层的）：

　　【1290】sys和exe 都需要使用 CTL_CODE来构建 控制码，于是新建一个头文件ctl_code.h，【1415】将 CTL_CODE 相关的(宏)代码写在这个头文件里面

“

#include <winioctl.h> // CTL_CODE
#define add_code CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_BUFFERED, FILE_ANY_ACCESS)
#define sub_code CTL_CODE(FILE_DEVICE_UNKNOWN, 0x801, METHOD_BUFFERED, FILE_ANY_ACCESS)

 

#include <windows.h>
#include "ctl_code.h"
int port[2],bufret;
DWORD dwWrite;
int add(HANDLE hDevice,int a,int b)
{
　　port[0]=a;
　　port[1]=b;
　　DeviceIoControl(hDevice, add_code , &port, 8, &bufret, 4, &dwWrite, NULL);
　　return bufret;
}

int main(int argc, char* argv[])
{
　　//CreateFile hDevice
　　HANDLE hDevice = CreateFile(
　　　　"\\\\.\\My_DriverLinkName", // \\??\\My_DriverLinkName
　　　　GENERIC_READ | GENERIC_WRITE,
　　　　0, // share mode none
　　　　NULL, // no security
　　　　OPEN_EXISTING,
　　　　FILE_ATTRIBUTE_NORMAL,
　　　　NULL ); // no template

　　printf("start\n");
　　if (hDevice == INVALID_HANDLE_VALUE)
　　{
　　　　printf("获取驱动句柄失败: %s with Win32 error code: %d\n","MyDriver",
　　　　GetLastError() );
　　　　getchar(); // system("pause");
　　　　return -1;
　　}
　　//add_code
　　int a=55;
　　int b=33;
　　int r= add(hDevice, a, b);
　　printf("%d+%d=%d\n", a, b, r);
　　getchar();
　　return 0;
}

”

【2585】驱动层的代码，以 第25课 的代码为例，进行一些修改

　　【2780】修改符号链接

　　　　RtlInitUnicodeString(&symLinkName,L"\\??\\My_DriverLinkName");

　　【3615】把 刚才 exe项目里面创建的头文件ctl_code.h 复制到 sys项目中 

　　【3935】复制过来的 ctl_code.h 中，要把 “#include <winioctl.h>”注释掉，或者 更通用的做法 这里我们就不讲了...

　　　　【3980】在 NTDDK.h 里面已经有了 CTL_CODE的定义（ZC: 可以用 “#ifdef” 来控制是否需要include winioctl.h/NTDDK.h，也可以将 include winioctl.h/NTDDK.h 写在 ctl_code.h的外面）

 　　【4625】ZC: 看到 sys中 输入/输出 缓冲区的获取，都是“pIrp->AssociatedIrp.SystemBuffer” ! ! ! 它会自动映射相应的内存？读的时候就用DeviceIoControl的输入缓冲区，写的时候就用DeviceIoControl的输出缓冲区？

“

#include "ctl_code.h"

NTSTATUS ddk_DispatchRoutine_CONTROL(IN PDEVICE_OBJECT pDevobj, IN PIRP pIrp)
{
　　ULONG info;
　　//得到当前栈指针
　　PIO_STACK_LOCATION stack = IoGetCurrentIrpStackLocation(pIrp);
　　ULONG mf=stack->MajorFunction; // 区分IRP
　　switch (mf)
　　{
　　　　case IRP_MJ_DEVICE_CONTROL:
　　　　{
　　　　　　KdPrint(("Enter myDriver_DeviceIOControl\n"));
　　　　　　NTSTATUS status = STATUS_SUCCESS;

　　　　　　//得到输入缓冲区大小
　　　　　　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;
　　　　　　　　　　_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;
　　　　　　　　　　_asm
　　　　　　　　　　{
　　　　　　　　　　　　mov eax,a
　　　　　　　　　　　　mov ebx,OutputBuffer
　　　　　　　　　　　　mov [ebx],eax
　　　　　　　　　　}
　　　　　　　　　　KdPrint(("a+b=%d \n",a));

　　　　　　　　　　//设置实际操作输出缓冲区长度
　　　　　　　　　　info = 4;
　　　　　　　　　　break;
　　　　　　　　}
　　　　　　　　case sub_code:
　　　　　　　　{
　　　　　　　　　　break;
　　　　　　　　}
　　　　　　}// switch (code)
　　　　　　break;
　　　　} // IRP_MJ_DEVICE_CONTROL
　　　　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; //返回成功
}

”

 

【5280】sys 和 exe 放到 虚拟机中，进行测试

　　【5455】此时运行，exe计算得到了错误的结果，∵变量未初始化（ZC: 貌似不对啊，exe能打开驱动，说明已经存在了相关的符号链接了啊）

　　【5495】载入驱动

　　　　【5535】此时 exe的计算结果还是错误的，可能是刚才已经载入了驱动。先卸载 之前打开的驱动

　　【5640】此时 报错 获取句柄失败。蓝屏 重新测试

　　【5745】exe计算结果 还是错误的

 

【5760】改代码 加调试信息

【6250】卸载例程 中的 符号链接名称 是错误的 没有改过来

【6855】可能是卸载的时候卸载的不干净，点击 DriverMonitor的"STOP"按钮 虚拟机会蓝屏

 

2、

转载于:https://www.cnblogs.com/debugskill/p/5364191.html