2023驱动保护学习 -- 应用层与驱动层读写操作

#include <winioctl.h>
#define 写测试   CTL_CODE(FILE_DEVICE_UNKNOWN,    0x801,   METHOD_BUFFERED,FILE_ANY_ACCESS) 
#define 读测试   CTL_CODE(FILE_DEVICE_UNKNOWN,    0x802,   METHOD_BUFFERED,FILE_ANY_ACCESS) 
#define 读写测试   CTL_CODE(FILE_DEVICE_UNKNOWN, 0x803,   METHOD_BUFFERED,FILE_ANY_ACCESS)

  char 输入缓冲区指针[] = "我是写操作测试！！！";
  int 返回缓冲区指针[] = { 0 };
  DWORD 实际读取字节数;
  DeviceIoControl(
    设备句柄,
    写测试,
    &输入缓冲区指针,
    sizeof(输入缓冲区指针),
    &返回缓冲区指针,
    sizeof(返回缓冲区指针),
    &实际读取字节数,
    NULL);

char 返回数据;
  DWORD 实际读取字节数;
  DeviceIoControl(
    设备句柄,
    读测试,
    NULL,
    0,
    &返回数据,
    sizeof(返回数据),
    &实际读取字节数,
    NULL);
  char 缓存区[256];
  sprintf_s(缓存区, "nxyn:读取到驱动内容：%s\n", 返回数据);
  OutputDebugStringA(缓存区);

  char 缓存区[256];
  sprintf_s(缓存区, "nxyn:应用程序写测试控制为%X\n", 读写测试);
  OutputDebugStringA(缓存区);
  int 输入数据[] = { 1,2,3 };
  int 返回数据;
  DWORD 实际读取字节数;
  DeviceIoControl(
    设备句柄,
    读写测试,
    &输入数据,
    sizeof(输入数据),
    &返回数据,
    sizeof(返回数据),
    &实际读取字节数,
    NULL);
  sprintf_s(缓存区, "nxyn:读取到驱动内容：%d+%d+%d=%d\n", 输入数据[0], 输入数据[1], 输入数据[2],返回数据);
  OutputDebugStringA(缓存区);

void IRP写测试(PIRP IRP指针)
{
  PIO_STACK_LOCATION 栈指针 = IoGetCurrentIrpStackLocation(IRP指针); //通过栈指针获取应用层传来的参数
  char* 缓冲区 = (char*)IRP指针->AssociatedIrp.SystemBuffer;
  KdPrint(("nxyn:应用层缓冲区写入驱动=%s",  缓冲区));
}

void IRP读测试(PIRP IRP指针)
{
  char* 缓冲区 = (char*)IRP指针->AssociatedIrp.SystemBuffer;
  char 返回数据[] = "我是来自驱动层的数据";
  memcpy_s(缓冲区, 256, 返回数据, sizeof(返回数据));
  IRP指针->IoStatus.Information = sizeof(返回数据);//设置操作的字节数
  IRP指针->IoStatus.Status = STATUS_SUCCESS;//返回状态
  IoCompleteRequest(IRP指针, IO_NO_INCREMENT);//完成一个IRP请求
  KdPrint(("nxyn:IRP函数处理完毕"));
}

  int* 缓冲区 = (int*)IRP指针->AssociatedIrp.SystemBuffer;
  int 计算结果=0;
  if(缓冲区)
  {
    计算结果 = 缓冲区[0] + 缓冲区[1] + 缓冲区[2];
    KdPrint(("nxyn:驱动层计算结果%d + %d + %d=%d", 缓冲区[0],缓冲区[1],缓冲区[2],计算结果));
  }
  *缓冲区 = 计算结果;

if (控制码==写测试)
    {
      IRP写测试(IRP指针);
    }
    else if (控制码 == 读测试)
    {
      IRP读测试(IRP指针);
      return STATUS_SUCCESS;
    }
    else if(控制码 == 读写测试)
    {
      IRP读写测试(IRP指针);
    }