WindowsX86内核02.驱动程序
把昨天的程序改用 c++ 编译,改成 .cpp ,发现编译报错原因是名称粉碎,因此可以直接 extern “C”声明一下这个函数 或者用 头文件(推荐)因为 在头文件中 可以把 头文件一起包含进去#pragma onceextern "C" {
#include <Ntddk.h>
/*驱动入口函数*/
NTSTATUS DriverEntry(__in struct _DRIVER_OBJECT* DriverObject,
__in PUNICODE_STRINGRegistryPath);
/*驱动卸载函数*/
VOID Unload(__in struct _DRIVER_OBJECT* DriverObject);
}api在驱动编程里面也可以面向对象编程,可以用class ,但是有很多功能是禁止的 ,例如 try catch 这是3环的机制,内核处理的方法不一样, malloc , printf ,scanf,new 等 基本的库函数 和 3环 的 api都不行 因为 内核有一套独立的内核空间,必须用微软提供的独立的内核api去申请 .操作系统会提供一套独立的 内核专用的 内核 apiReference 里面就是所有的apiWDM(NT)驱动框架我们前面写的算一个一个驱动程序 (有打印功能),但不具备交互性,即不能跟驱动沟通那么怎么让驱动具备交互性呢(跟3环程序作交互)内核驱动一般不做界面功能,有3环程序来做,因为内核不能中出bug因此 内核驱动开发有一个原则,驱动里面3环能解决的问题就在3环解决,驱动里面只做内核才能做的事
[*]想要跟3环做交互,那就要按操作系统的设计来做这个框架
[*]如果不需要跟3环程序交互,可能不需要 按 操作系统的设计来
[*]卸载函数必须有,不然装上了就无法卸载 (因为有些驱动可能是系统一定要用的,防止被误卸载了)如果是开机启动,重启也无法卸载,除非进安全模式
步骤
[*]实现驱动入口函数 DriverEntry
[*]创建设备
[*]注册派函数
[*]注册卸载函数 Unload
系统 =》 驱动接口 =》 硬件因为将来的硬件我们也知道有什么,所以驱动接口必须支持现在的硬件,还要支持将来的硬件,因此驱动接口的设计就十分重要,必须抽象驱动接口 => 抽象 =》 硬件共同的特点 => 文件应用程序 =》 文件API => 系统 => 驱动操作接口(open read write close) Dispatch Routines =》 硬件写驱动就是想操作系统提供 open read write close 的接口,就是文件指针,这些函数就是派遣函数设备对象 DEVICE_OBJECT 存放 open read write close 的接口绑定驱动对象 DRIVER_OBJECT 虚拟出来的,当操作系统加载驱动的时候,就会创建一个驱动对象来描述加载的驱动的所有信息 DriverObject3环程序只需要 操作设备对象 ,设备对象自然会跟驱动对象绑定实现实现驱动入口函数 DriverEntry
/*驱动入口函数*/
NTSTATUS DriverEntry(
__in struct _DRIVER_OBJECT* DriverObject,
__in PUNICODE_STRINGRegistryPath)
{
return STATUS_SUCCESS;
}添加设备操作系统会提供内核API 实现王操作系统添加一个设备ms-help://MS.WDK.v10.7600.091201/Kernel_r/hh/Kernel_r/k104_1e38a631-7e65-4b4b-8d51-3150a8073511.xml.htm最常用的就是 IoCreateDevice参数:
[*]驱动对象 (一个驱动可以创建多个设备对象,他就会挂链表)
[*]设备口拓展大小 (驱动对象结构体有一个空指针成员,可以用来存放用户申请的地址,对结构体进行拓展,用来取代全局变量),一般定义一个结构体,也可以不用
[*]设备名字 (在内核中基本不用 3环的字符串格式char* 即不是'\0'结尾的字符串,而是用 UNICODE_STRING 结构体),名字有要求 前面必须要有 \Device\ 。而且必须唯一,不可以与其他设备相同
[*]设备种类,不能为NULL
[*]是否独占 一般是false true表示独占:该软件使用时,其他软件无法使用
[*]返回的设备对象
不用char* 是因容易字符串一处造成漏洞内核中有些库函数还是可以用的 如: strlen strcpy但不推荐,因为不是 "\0" 结尾windows整个操作系统是由C语言 + 汇编写的,但他整个设计用的是面向对象的思想的内核里面基本返回值统一,这样调 api 可以统一检查 ,通过宏 NT_SUCCESS 检查驱动卸载下次在安装将创建不了,因为名称被占用了,因此需要在upload 中 删除对象设备#define DEVICE_NAMEL"\\Device\\CR42"
UNICODE_STRING ustrDevName;
//ustrDevName.Length = 8;
//ustrDevName.MaximumLength = 10;
//ustrDevName.Buffer = L"CR42";
RtlInitUnicodeString(&ustrDevName, DEVICE_NAME);//该函数就是实现上面3 行代码
PDEVICE_OBJECT pDevObj = NULL;
NTSTATUS Status = IoCreateDevice(DriverObject,
sizeof(MyDeviceExt),
&ustrDevName,
FILE_DEVICE_UNKNOWN,//不知道的设备类型
FILE_DEVICE_SECURE_OPEN,
FALSE, //独占
&pDevObj);
if (!NT_SUCCESS(Status)) { //检查错误
DbgPrint(" IoCreateDevice Status:%p\n", Status);
return Status; //失败返回 状态码
}
注册派遣函数驱动对象最后一项是一个函数指针数组,我们驱动有什么功能就填什么功能,而且通过宏说明了每个下标分别放什么功能一般对一内核驱动来说 必须提供 打开 关闭 控制 3个功能这些函数就是派遣函数,也可以称回调函数NTSTATUS DispatchCreate(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchClose(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchRead(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchWrite(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchControl(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
/*
IRPI/O Request Packet
*/
NTSTATUS DispatchCreate(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
//表示参数暂时不用,不说明一下的话参数没用没报错
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
NTSTATUS DispatchClose(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
//表示参数暂时不用,不说明一下的话参数没用没报错
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
NTSTATUS DispatchRead(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
//表示参数暂时不用,不说明一下的话参数没用没报错
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
NTSTATUS DispatchWrite(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
//表示参数暂时不用,不说明一下的话参数没用没报错
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
NTSTATUS DispatchControl(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
//表示参数暂时不用,不说明一下的话参数没用没报错
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
//3.注册派遣函数
DriverObject->MajorFunction = &DispatchCreate; //打开
DriverObject->MajorFunction = &DispatchClose; //关闭
DriverObject->MajorFunction = &DispatchRead; //读
DriverObject->MajorFunction = &DispatchWrite; //写
DriverObject->MajorFunction = &DispatchControl;//控制注册卸载函数VOID Unload(__in struct _DRIVER_OBJECT* DriverObject){DbgPrint(" Unload! DriverObject:%p\n", DriverObject);//删除设备if (DriverObject->DeviceObject != NULL) IoDeleteDevice(DriverObject->DeviceObject);}
简单框架完成Hello.h#pragma once
extern "C" {
#include <Ntddk.h>
struct MyDeviceExt {
int nVirtualReg;
};
#define DEVICE_NAMEL"\\Device\\CR42"
#define SYMBOL_NAME L"\\DosDevices\\CR42Virtual"
/*驱动入口函数*/
NTSTATUS DriverEntry(__in struct _DRIVER_OBJECT* DriverObject,
__in PUNICODE_STRINGRegistryPath);
VOID Unload(__in struct _DRIVER_OBJECT* DriverObject);
NTSTATUS DispatchCreate(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchClose(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchRead(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchWrite(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchControl(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
}Hello.cpp#include "Hello.h"
/*驱动卸载函数 clean_up*/
VOID Unload(__in struct _DRIVER_OBJECT* DriverObject)
{
DbgPrint(" Unload! DriverObject:%p\n", DriverObject);
//删除设备
if (DriverObject->DeviceObject != NULL)
IoDeleteDevice(DriverObject->DeviceObject);
}
/*1.驱动入口函数*/
NTSTATUS DriverEntry(
__in struct _DRIVER_OBJECT* DriverObject,
__in PUNICODE_STRINGRegistryPath)
{
DbgPrint(" Hello WDK! DriverObject:%p RegistryPath:%wZ\n",
DriverObject, RegistryPath);
//2.创建设备
UNICODE_STRING ustrDevName;
//ustrDevName.Length = 8;
//ustrDevName.MaximumLength = 10;
//ustrDevName.Buffer = L"CR42";
RtlInitUnicodeString(&ustrDevName, DEVICE_NAME);
PDEVICE_OBJECT pDevObj = NULL;
NTSTATUS Status = IoCreateDevice(DriverObject,
sizeof(MyDeviceExt),
&ustrDevName,
FILE_DEVICE_UNKNOWN,//不知道的设备类型
FILE_DEVICE_SECURE_OPEN,
FALSE, //独占
&pDevObj);
if (!NT_SUCCESS(Status)) {
DbgPrint(" IoCreateDevice Status:%p\n", Status);
return Status;
}
//3.注册派遣函数
DriverObject->MajorFunction = &DispatchCreate;
DriverObject->MajorFunction = &DispatchClose;
DriverObject->MajorFunction = &DispatchRead;
DriverObject->MajorFunction = &DispatchWrite;
DriverObject->MajorFunction = &DispatchControl;
//4.注册卸载函数
DriverObject->DriverUnload = Unload;
return STATUS_SUCCESS;
}
/*
IRPI/O Request Packet
*/
NTSTATUS DispatchCreate(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
NTSTATUS DispatchClose(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
NTSTATUS DispatchRead(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
NTSTATUS DispatchWrite(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
NTSTATUS DispatchControl(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
return STATUS_SUCCESS;
}
当设备被打开时,就会找到 DispatchCreate 函数并进行调用 里面一般都是通过 内联汇编代码 跟硬件进行交互3环程序跟驱动进行通讯CreateFile() ReadFile() WriteFile() CloseHandle() DeviceIoControl()分别会对应调到上面的四个函数DispatchCreate() DispatchRead() DispatchWrite() DispatchClose() DispatchControl()实现虚拟硬件因为要在要xp上运行,所以编译成静态库 ,然后再 PE中 把 MajorSubsystemVersion 项 改成 4 就可以了#include <stdio.h>
#include <Windows.h>
#include <stdlib.h>
struct MyDeviceExt {
int nVirtualReg;
};
//显示错误信息FormatMessage
void DisplayErrorText()
{
LPVOID lpMsgBuf;
FormatMessage(
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
(LPTSTR)&lpMsgBuf,
0,
NULL
);
printf((LPCTSTR)lpMsgBuf);
// Free the buffer.
LocalFree(lpMsgBuf);
}
int main()
{
HANDLE hFile = CreateFile("\\Device\\CR42",
GENERIC_ALL,
0,
NULL,
OPEN_EXISTING,//驱动只能打开已存在
FILE_ATTRIBUTE_NORMAL,
NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
DisplayErrorText();
}
else {
printf("hFile = %p\n", hFile);
}
CloseHandle(hFile);
system("pause");
return 0;
}运行可以看到找不到路径 ,这是权限问题每个设备有2个 名字,其中一个名字叫 NT Device Names 还有一个叫 MS-DOS Device NamesNT Device Names 只能在内核中打开 ,3环中无法打开MS-DOS Device Names(符号链接) , 如果要想设被3环访问,必须创建一个 符号链接,就相当于路径的一个别名。如果不想被3环访问,可以没有,而且 符号名和设备名也是唯一的,卸载的时候也要删除#define SYMBOL_NAME L"\\DosDevices\\CR42Virtual"//MS-DOS Device Names创建符号链接UNICODE_STRING ustrSymbolName;RtlInitUnicodeString(&ustrSymbolName, SYMBOL_NAME);Status = IoCreateSymbolicLink(&ustrSymbolName, &ustrDevName);if (!NT_SUCCESS(Status)) { DbgPrint(" IoCreateSymbolicLink Status:%p\n", Status);if (pDevObj != NULL)IoDeleteDevice(pDevObj);return Status;}DbgPrint(" IoCreateSymbolicLink %wZ OK\n", &ustrSymbolName);//删除符号链接UNICODE_STRING ustrSymbolName;RtlInitUnicodeString(&ustrSymbolName, SYMBOL_NAME);IoDeleteSymbolicLink(&ustrSymbolName);
#include <stdio.h>#include <Windows.h>#include <stdlib.h>struct MyDeviceExt { int nVirtualReg;};//显示错误信息FormatMessagevoid DisplayErrorText(){ LPVOID lpMsgBuf; FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, NULL, GetLastError(), MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language (LPTSTR)&lpMsgBuf, 0, NULL ); printf((LPCTSTR)lpMsgBuf); // Free the buffer. LocalFree(lpMsgBuf);}int main(){ HANDLE hFile = CreateFile("\\\\?\\CR42VirtualRegister", // ?= DosDevices GENERIC_ALL, 0, NULL, OPEN_EXISTING,//驱动只能打开已存在 FILE_ATTRIBUTE_NORMAL, NULL); if (hFile == INVALID_HANDLE_VALUE) { DisplayErrorText(); } else { printf("hFile = %p\n", hFile); } CloseHandle(hFile); system("pause"); return 0;}
测试这样3环程序可以指挥驱动程序干活了完善调用四个函数
#include <stdio.h>
#include <Windows.h>
#include <stdlib.h>
struct MyDeviceExt {
int nVirtualReg;
};
void DisplayErrorText()
{
LPVOID lpMsgBuf;
FormatMessage(
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
(LPTSTR)&lpMsgBuf,
0,
NULL
);
printf((LPCTSTR)lpMsgBuf);
LocalFree(lpMsgBuf);
}
int main()
{
HANDLE hFile = CreateFile("\\\\?\\CR42Virtual",
GENERIC_ALL,
0,
NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL,
NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
DisplayErrorText();
}
else {
printf("hFile = %p\n", hFile);
}
char szBuf = {0};
DWORD dwBytes = 0;
ReadFile(hFile, szBuf, sizeof(szBuf), &dwBytes, NULL);
printf("ReadFile dwBytes = %dszBuf:%s\n", dwBytes, szBuf);
WriteFile(hFile, szBuf, sizeof(szBuf), &dwBytes, NULL);
printf("WriteFile dwBytes = %d szBuf:%s\n", dwBytes, szBuf);
//控制
//DeviceIoControl(hFile, 100, szBuf, sizeof(szBuf), szBuf, sizeof(szBuf), &dwBytes, NULL);
CloseHandle(hFile);
system("pause");
return 0;
}实现驱动模拟寄存器驱动程序用一个变量来当作寄存器保存值 ,如果要读寄存器的值,把返变量的值 ,如果要写寄存器的值,就修改变量的值实现的难点就是参数的传递 , 内核如何收到3环的数据3环如果调用WriteFile函数 和 ReadFile 函数时 ,就会把对应的信息打包成一个 Irp 结构体传递给 内核IRP I/O Request Packet 读写请求包我们必须完成这个请求,如果不完成请求,函数可能会卡死在那例如网卡驱动。假如有人发起读请求,读取从别的电脑发送过来的网络数据,这时候给不了,因为没数据发过来,那这个时候要把请求挂起来,也要把线程挂起来,直到收到数据包为止#pragma once
extern "C" {
#include <Ntddk.h>
struct MyDeviceExt {
int nVirtualReg; //用来当作寄存器的值
};
#define DEVICE_NAMEL"\\Device\\CR42"
#define SYMBOL_NAME L"\\DosDevices\\CR42VirtualRegister"
/*驱动入口函数*/
NTSTATUS DriverEntry(__in struct _DRIVER_OBJECT* DriverObject,
__in PUNICODE_STRINGRegistryPath);
VOID Unload(__in struct _DRIVER_OBJECT* DriverObject);
NTSTATUS DispatchCreate(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchClose(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchRead(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchWrite(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
NTSTATUS DispatchControl(
_In_ struct _DEVICE_OBJECT* DeviceObject,
_Inout_ struct _IRP* Irp
);
}#include "Hello.h"
/*驱动卸载函数 clean_up*/
VOID Unload(__in struct _DRIVER_OBJECT* DriverObject)
{
DbgPrint(" Unload! DriverObject:%p\n", DriverObject);
//删除符号链接
UNICODE_STRING ustrSymbolName;
RtlInitUnicodeString(&ustrSymbolName, SYMBOL_NAME);
IoDeleteSymbolicLink(&ustrSymbolName);
//删除设备
if (DriverObject->DeviceObject != NULL)
IoDeleteDevice(DriverObject->DeviceObject);
}
/*1.驱动入口函数*/
NTSTATUS DriverEntry(
__in struct _DRIVER_OBJECT* DriverObject,
__in PUNICODE_STRINGRegistryPath)
{
DbgPrint(" Hello WDK! DriverObject:%p RegistryPath:%wZ\n",
DriverObject, RegistryPath);
//2.创建设备
UNICODE_STRING ustrDevName;
//ustrDevName.Length = 8;
//ustrDevName.MaximumLength = 10;
//ustrDevName.Buffer = L"CR42";
RtlInitUnicodeString(&ustrDevName, DEVICE_NAME);//等于上面的3行代码
PDEVICE_OBJECT pDevObj = NULL;
NTSTATUS Status = IoCreateDevice(DriverObject,
sizeof(MyDeviceExt),
&ustrDevName,
FILE_DEVICE_UNKNOWN,//不知道的设备类型
FILE_DEVICE_SECURE_OPEN,
FALSE, //独占
&pDevObj);
if (!NT_SUCCESS(Status)) {
DbgPrint(" IoCreateDevice Status:%p\n", Status);
return Status;
}
//初始化寄存器变量
RtlZeroMemory(pDevObj->DeviceExtension, sizeof(MyDeviceExt));
MyDeviceExt* pDevExt = (MyDeviceExt*)pDevObj->DeviceExtension;
pDevExt->nVirtualReg = 0;
//MS-DOS Device Names创建符号链接
UNICODE_STRING ustrSymbolName;
RtlInitUnicodeString(&ustrSymbolName, SYMBOL_NAME);
Status = IoCreateSymbolicLink(&ustrSymbolName, &ustrDevName);
if (!NT_SUCCESS(Status)) {
DbgPrint(" IoCreateSymbolicLink Status:%p\n", Status);
if (pDevObj != NULL)
IoDeleteDevice(pDevObj);
return Status;
}
DbgPrint(" IoCreateSymbolicLink %wZ OK\n", &ustrSymbolName);
//3.注册派遣函数
DriverObject->MajorFunction = &DispatchCreate;
DriverObject->MajorFunction = &DispatchClose;
DriverObject->MajorFunction = &DispatchRead;
DriverObject->MajorFunction = &DispatchWrite;
DriverObject->MajorFunction = &DispatchControl;
//4.注册卸载函数
DriverObject->DriverUnload = Unload;
return STATUS_SUCCESS;
}
NTSTATUS DispatchCreate(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
//完成请求如果没完成 3环的程序就会挂起
Irp->IoStatus.Status = STATUS_SUCCESS; //状态,成功
Irp->IoStatus.Information = 0; //成功操作的字节数,非读写一般是0 就算是读写文件 dwBytes的返回值
//第二个参数线程的优先级,内核一般时 IO_NO_INCREMENT 不提升优先级
IoCompleteRequest(Irp, IO_NO_INCREMENT);
return STATUS_SUCCESS;
}
NTSTATUS DispatchClose(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
//完成请求如果没完成 3环的程序就会挂起
Irp->IoStatus.Status = STATUS_SUCCESS; //状态,成功
Irp->IoStatus.Information = 0; //成功操作的字节数,非读写一般是0 就算是读写文件 dwBytes的返回值
//第二个参数线程的优先级,内核一般时 IO_NO_INCREMENT 不提升优先级
IoCompleteRequest(Irp, IO_NO_INCREMENT);
return STATUS_SUCCESS;
}
NTSTATUS DispatchRead(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
//获取用户的缓冲区
PVOID pBuffer = Irp->UserBuffer; //就是用户参数的 缓冲地址
//获取当前IRP堆栈
PIO_STACK_LOCATION pIrpStack = IoGetCurrentIrpStackLocation(Irp);
ULONG nLength = pIrpStack->Parameters.Read.Length; //就是用户参数的 缓冲大小read表示读操作
DbgPrint(" %s pBuffer:%p nLength:%d\n", __FUNCTION__, pBuffer, nLength);
MyDeviceExt* pDevExt = (MyDeviceExt*)DeviceObject->DeviceExtension;
if (nLength > sizeof(pDevExt->nVirtualReg)) {//检查缓冲区大小,防溢出
nLength = sizeof(pDevExt->nVirtualReg);
}
//拷贝信息就等于memcoy
RtlCopyMemory(pBuffer, &pDevExt->nVirtualReg, nLength);//把寄存器数据数据传入用缓冲区
//完成请求如果没完成 3环的程序就会挂起
Irp->IoStatus.Status = STATUS_SUCCESS; //状态,成功
Irp->IoStatus.Information = nLength; //成功操作的字节数,非读写一般是0 就算是读写文件 dwBytes的返回值
//第二个参数线程的优先级,内核一般时 IO_NO_INCREMENT 不提升优先级
IoCompleteRequest(Irp, IO_NO_INCREMENT);
return STATUS_SUCCESS;
}
NTSTATUS DispatchWrite(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
//获取用户的缓冲区
PVOID pBuffer = Irp->UserBuffer; //就是用户参数的 缓冲地址
//获取当前IRP堆栈
PIO_STACK_LOCATION pIrpStack = IoGetCurrentIrpStackLocation(Irp);
ULONG nLength = pIrpStack->Parameters.Write.Length; //就是用户参数的 缓冲大小Write表示写操作
DbgPrint(" %s pBuffer:%p nLength:%d\n", __FUNCTION__, pBuffer, nLength);
MyDeviceExt* pDevExt = (MyDeviceExt*)DeviceObject->DeviceExtension;
if (nLength > sizeof(pDevExt->nVirtualReg)) { //检查缓冲区大小,防溢出
nLength = sizeof(pDevExt->nVirtualReg);
}
RtlCopyMemory(&pDevExt->nVirtualReg, pBuffer, nLength);//将用户数据拷贝进内核的缓冲区
//完成请求如果没完成 3环的程序就会挂起
Irp->IoStatus.Status = STATUS_SUCCESS; //状态,成功
Irp->IoStatus.Information = nLength; //成功操作的字节数,非读写一般是0 就算是读写文件 dwBytes的返回值
//第二个参数线程的优先级,内核一般时 IO_NO_INCREMENT 不提升优先级
IoCompleteRequest(Irp, IO_NO_INCREMENT);
return STATUS_SUCCESS;
}
NTSTATUS DispatchControl(_DEVICE_OBJECT* DeviceObject, _IRP* Irp)
{
UNREFERENCED_PARAMETER(DeviceObject);
UNREFERENCED_PARAMETER(Irp);
DbgPrint(" %s\n", __FUNCTION__);
//完成请求如果没完成 3环的程序就会挂起
Irp->IoStatus.Status = STATUS_SUCCESS; //状态,成功
Irp->IoStatus.Information = 0; //成功操作的字节数,非读写一般是0 就算是读写文件 dwBytes的返回值
//第二个参数线程的优先级,内核一般时 IO_NO_INCREMENT 不提升优先级
IoCompleteRequest(Irp, IO_NO_INCREMENT);
return STATUS_SUCCESS;
}
测试程序
#include <stdio.h>
#include <Windows.h>
#include <stdlib.h>
struct MyDeviceExt {
int nVirtualReg;
};
void DisplayErrorText()
{
LPVOID lpMsgBuf;
FormatMessage(
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS,
NULL,
GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
(LPTSTR)&lpMsgBuf,
0,
NULL
);
printf((LPCTSTR)lpMsgBuf);
LocalFree(lpMsgBuf);
}
int main()
{
HANDLE hFile = CreateFile("\\\\?\\CR42VirtualRegister",
GENERIC_ALL,
0,
NULL,
OPEN_EXISTING,
FILE_ATTRIBUTE_NORMAL,
NULL);
if (hFile == INVALID_HANDLE_VALUE)
{
DisplayErrorText();
}
else {
printf("hFile = %p\n", hFile);
}
MyDeviceExt DevExt = { 0 };
DWORD dwBytes = 0;
DevExt.nVirtualReg = 100;
if (!WriteFile(hFile, &DevExt, 100, &dwBytes, NULL))//往寄存器里面写值
{
DisplayErrorText();
}
else
{
printf("WriteFile dwBytes = %dVirtualReg:%d\n", dwBytes, DevExt.nVirtualReg);
}
DevExt.nVirtualReg = -1;
if (!ReadFile(hFile, &DevExt, sizeof(DevExt), &dwBytes, NULL))//从寄存器里面读值
{
DisplayErrorText();
}
else
{
printf("ReadFile dwBytes = %dVirtualReg:%d\n", dwBytes, DevExt.nVirtualReg);
}
//控制
//DeviceIoControl(hFile, 100, szBuf, sizeof(szBuf), szBuf, sizeof(szBuf), &dwBytes, NULL);
CloseHandle(hFile);
system("pause");
return 0;
}测试因为不是独占,所以多个函数可以同时调用,注意,多函数同时访问存在同步问题漏洞例如: MyDeviceExt DevExt = { 0 }; DWORD dwBytes = 0; DevExt.nVirtualReg = 100; if (!WriteFile(hFile, &DevExt, 10000, &dwBytes, NULL))//往寄存器里面写值 { DisplayErrorText(); } else { printf("WriteFile dwBytes = %dVirtualReg:%d\n", dwBytes, DevExt.nVirtualReg); }
NTSTATUS DispatchWrite(_DEVICE_OBJECT* DeviceObject, _IRP* Irp){ UNREFERENCED_PARAMETER(DeviceObject); UNREFERENCED_PARAMETER(Irp); DbgPrint(" %s\n", __FUNCTION__); //获取用户的缓冲区 PVOID pBuffer = Irp->UserBuffer; //就是用户参数的 缓冲地址 //获取当前IRP堆栈 PIO_STACK_LOCATION pIrpStack = IoGetCurrentIrpStackLocation(Irp); ULONG nLength = pIrpStack->Parameters.Write.Length; //就是用户参数的 缓冲大小Write表示写操作 DbgPrint(" %s pBuffer:%p nLength:%d\n", __FUNCTION__, pBuffer, nLength); MyDeviceExt* pDevExt = (MyDeviceExt*)DeviceObject->DeviceExtension; RtlCopyMemory(&pDevExt->nVirtualReg, pBuffer, nLength);//将用户数据拷贝进内核的缓冲区 //完成请求如果没完成 3环的程序就会挂起 Irp->IoStatus.Status = STATUS_SUCCESS; //状态,成功 Irp->IoStatus.Information = nLength; //成功操作的字节数,非读写一般是0 就算是读写文件 dwBytes的返回值 //第二个参数线程的优先级,内核一般时 IO_NO_INCREMENT 不提升优先级 IoCompleteRequest(Irp, IO_NO_INCREMENT); return STATUS_SUCCESS;}
上面的代码会导致系统蓝屏 因为上面 3成程序 给驱动的缓冲区只有4字节 ,但是大小却指定了 10000,由于没有长度检查 ,这样 驱动去访问那 10000 个字节 就会 c005 ,RtlCopyMemory(&pDevExt->nVirtualReg, pBuffer, nLength); 中 nLength 大小由3环程序控制由于是往内核写数据,因为长度出错,导致代码溢出了,因此需要对长度检查WinObj可以找遍历电脑上已经安装的所有设备https://docs.microsoft.com/zh-cn/sysinternals/downloads/winobj
页:
[1]