菜鸟之驱动开发13

最新推荐文章于 2024-09-18 16:46:31 发布

iteye_17686

最新推荐文章于 2024-09-18 16:46:31 发布

阅读量72

点赞数

文章标签：驱动开发

在本节，我们将学习在内核模式下操作文件，包括：创建，打开，读取，修改，文件属性读取与修改。

相关的API有：ZwCreateFile, ZwOpenFile, ZwReadFile, ZwWriteFile, ZwQueryInformationFile,ZwSetInformationFile, 它们对应的应用程的API就是去掉前面的Zw.

内核模式操作文件都需要先初始化一个OBJECT_ATTRIBUTES结构如OBJECT_ATTRIBUTES objectAttributes;具体将在下面的源码中看到。

创建文件方法:

VOID CreateFileTest() { NTSTATUS ntStatus; OBJECT_ATTRIBUTES objectAttributes; IO_STATUS_BLOCK iostatus; HANDLE hfile; UNICODE_STRING SymbolFileName; KdPrint(("\n-----------------创建文件测试 --------------------!\n")); //初始化UNICODE_STRING字符串 RtlInitUnicodeString(&SymbolFileName,L"\\Device\\HarddiskVolume1\\CreateFileTest.txt"); //RtlInitUnicodeString( &SymbolFileName, L"\\??\\C:\\\CreateFileTest.txt");//可用此行替换上一行 //初始化objectAttributes InitializeObjectAttributes( &objectAttributes, //指定一个需要OBJECT_ATTRIBUTES结构地址 &SymbolFileName,//是一个UNICODE_STRING字串地址，指定需要操作对象名（在这里可以是符号链接名，或者设备名) OBJ_CASE_INSENSITIVE, //指定此值表示不区分大小写。 NULL, NULL ); //创建文件 ntStatus = ZwCreateFile( &hfile, //PHANDLE类型指针用于返回打开文件的句柄 GENERIC_WRITE, //ACCESS_MASK类型此值用于描述打开文件操作（读，写，或者其它） &objectAttributes, //此值是OBJECT_ATTRIBUTES结构的地址，该结构包含要打开的文件名需要用InitializeObjectAttributes进行初始化 &iostatus, //指向一个IO_STATUS_BLOCK结构，返回值之一用于存放ZwCreateFile操作的结果状态 NULL, //PLARGE_INTEGER 类型（64位整数指针）该数指定文件初始分配时的大小。该参数如果为NULL，那么文件长度将从0开始，随着写入而增长 FILE_ATTRIBUTE_NORMAL, //此参数在驱动下指定为0或者FILE_ATTRIBUTE_NORMAL，如果文件不是被创建和覆盖写入则此参数将被忽略 FILE_SHARE_READ, //指定共享模式有共享读FILE_SHARE_READ,写FILE_SHARE_WRITE,删除FILE_SHARE_DELETE这几中模式 FILE_OPEN_IF, //此值表示文件存在则打开不存在则创建一个新的文件 FILE_SYNCHRONOUS_IO_NONALERT, //指定文件创建或者打开的附加标志 FILE_SYNCHRONOUS_IO_NONALERT表示在文件中的所有操作均同步，并没有警报 NULL, //对于设备和中间驱动程序，此参数必须是NULL 0 ); //对于设备和中间驱动程序，此参数必须是0 if ( NT_SUCCESS(ntStatus)) { KdPrint(("创建文件成功!\n")); }else { KdPrint(("创建文件失败!\n")); } //文件操作 //....... //关闭文件句柄 ZwClose(hfile); }

打开文件方法有几种：可以用ZwCreateFile,可以用ZwOpenFile,在这儿我们用ZwCreateFile, 在完整源码中两种方法都有。

VOID OpenFileTest() { NTSTATUS ntStatus; OBJECT_ATTRIBUTES objectAttributes; IO_STATUS_BLOCK iostatus; HANDLE hfile; UNICODE_STRING SymbolFileName; KdPrint(("\n-----------------打开文件测试OpenFileTest --------------------!\n")); //初始化UNICODE_STRING字符串 RtlInitUnicodeString(&SymbolFileName,L"\\Device\\HarddiskVolume1\\CreateFileTest.txt"); //RtlInitUnicodeString( &SymbolFileName, L"\\??\\C:\\\CreateFileTest.txt");//可用此行替换上一行 //初始化objectAttributes InitializeObjectAttributes( &objectAttributes, //指定一个需要OBJECT_ATTRIBUTES结构地址 &SymbolFileName,//是一个UNICODE_STRING字串地址，指定需要操作对象名（在这里可以是符号链接名，或者设备名) OBJ_CASE_INSENSITIVE, //指定此值表示不区分大小写。 NULL, NULL ); //创建文件 ntStatus = ZwCreateFile( &hfile, //PHANDLE类型指针用于返回打开文件的句柄 GENERIC_WRITE, //ACCESS_MASK类型此值用于描述打开文件操作（读，写，或者其它） &objectAttributes, //此值是OBJECT_ATTRIBUTES结构的地址，该结构包含要打开的文件名需要用InitializeObjectAttributes进行初始化 &iostatus, //指向一个IO_STATUS_BLOCK结构，返回值之一用于存放ZwCreateFile操作的结果状态 NULL, //PLARGE_INTEGER 类型（64位整数指针）该数指定文件初始分配时的大小。该参数如果为NULL，那么文件长度将从0开始，随着写入而增长 FILE_ATTRIBUTE_NORMAL, //此参数在驱动下指定为0或者FILE_ATTRIBUTE_NORMAL，如果文件不是被创建和覆盖写入则此参数将被忽略 FILE_SHARE_READ, //指定共享模式有共享读FILE_SHARE_READ,写FILE_SHARE_WRITE,删除FILE_SHARE_DELETE这几中模式 FILE_OPEN, //此值表示文件存在则打开不存在则创建一个新的文件 FILE_SYNCHRONOUS_IO_NONALERT, //指定文件创建或者打开的附加标志 FILE_SYNCHRONOUS_IO_NONALERT表示在文件中的所有操作均同步，并没有警报 NULL, //对于设备和中间驱动程序，此参数必须是NULL 0 ); //对于设备和中间驱动程序，此参数必须是0 if ( NT_SUCCESS(ntStatus)) { KdPrint(("打开文件成功!hfile=%x\n",hfile)); }else { KdPrint(("打开文件失败!hfile=%x\n",hfile)); } //文件操作 //....... //关闭文件句柄 ZwClose(hfile); KdPrint(("\n-----------------打开文件测试OpenFileTest end --------------------!\n")); }

读取文件：

VOID ReadFileTest() { KdPrint(("\n---------------------ZwReadFile写文件测试--------------------\n")); OBJECT_ATTRIBUTES objectAttributes; IO_STATUS_BLOCK iostatus; HANDLE hfile; UNICODE_STRING SymbolFileName; //初始化UNICODE_STRING字符串 RtlInitUnicodeString( &SymbolFileName, L"\\Device\\HarddiskVolume1\\CreateFileTest.txt"); //RtlInitUnicodeString( &SymbolFileName, L"\\??\\C:\\1.txt"); //初始化objectAttributes InitializeObjectAttributes(&objectAttributes, &SymbolFileName, OBJ_CASE_INSENSITIVE,//对大小写敏感 NULL, NULL ); //创建文件 NTSTATUS ntStatus = ZwCreateFile( &hfile, GENERIC_READ, &objectAttributes, &iostatus, NULL, FILE_ATTRIBUTE_NORMAL, //常规属性 FILE_SHARE_READ,//共享读 FILE_OPEN,//有文件存在则打开，不存在则报错 FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0 ); if (!NT_SUCCESS(ntStatus)) { KdPrint(("文件不存在!\n")); return; } FILE_STANDARD_INFORMATION fsi; //读取文件长度 ntStatus = ZwQueryInformationFile(hfile, &iostatus, &fsi, sizeof(FILE_STANDARD_INFORMATION), FileStandardInformation); KdPrint(("共读取 %d 字节\n",fsi.EndOfFile.QuadPart)); //为读取的文件分配缓冲区 PUCHAR pBuffer = (PUCHAR)ExAllocatePool(PagedPool, (LONG)fsi.EndOfFile.QuadPart); //读取文件 ZwReadFile(hfile,NULL, NULL,NULL, &iostatus, pBuffer, (LONG)fsi.EndOfFile.QuadPart, NULL,NULL); KdPrint(("实际读取 %d 字节\n",iostatus.Information)); //显示实际读取内容 KdPrint(("\n 成功读取内容为:%s \n",pBuffer)); //关闭文件句柄 ZwClose(hfile); //释放缓冲区 ExFreePool(pBuffer); }

修改文件：

VOID WriteFileTest() { KdPrint(("\n---------------------ZwWriteFile写文件测试--------------------\n")); OBJECT_ATTRIBUTES objectAttributes; IO_STATUS_BLOCK iostatus; HANDLE hfile; UNICODE_STRING SymbolFileName; //初始化UNICODE_STRING字符串 RtlInitUnicodeString( &SymbolFileName, L"\\Device\\HarddiskVolume1\\CreateFileTest.txt"); //RtlInitUnicodeString( &SymbolFileName, L"\\??\\C:\\1.txt"); //初始化objectAttributes InitializeObjectAttributes(&objectAttributes, &SymbolFileName, OBJ_CASE_INSENSITIVE,//对大小写敏感 NULL, NULL ); //创建文件 NTSTATUS ntStatus = ZwCreateFile( &hfile, GENERIC_WRITE, &objectAttributes, &iostatus, NULL, FILE_ATTRIBUTE_NORMAL, FILE_SHARE_WRITE, FILE_OPEN_IF,//即使存在该文件，也创建 FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0 ); #define BUFFER_SIZE 30 PCHAR pBuffer = (PCHAR)ExAllocatePool(PagedPool,BUFFER_SIZE); //构造要填充的数据 RtlFillMemory(pBuffer,BUFFER_SIZE,'s');//www.yjxsoft.comssssssssssssxxxxxx //strcpy(pBuffer,"www.yjxsoft.com"); RtlCopyMemory(pBuffer,"www.yjxsoft.com",15); KdPrint(("即将写入 %d 字节\n",BUFFER_SIZE)); //写文件ZwQueryInfomationfile LARGE_INTEGER sl; sl.QuadPart=0i64; ZwWriteFile(hfile,NULL,NULL,NULL,&iostatus,pBuffer,30,&sl,NULL); KdPrint(("成功写入 %d 字节\n",iostatus.Information)); //构造要填充的数据 RtlFillMemory(pBuffer,BUFFER_SIZE,'x'); KdPrint(("成功写入 %d 字节\n",BUFFER_SIZE)); //追加数据 LARGE_INTEGER number; number.QuadPart =30i64;//设置文件指针 //对文件进行附加写 ZwWriteFile(hfile,NULL,NULL,NULL,&iostatus,pBuffer,BUFFER_SIZE,&number,NULL); KdPrint(("追加写入 %d 字节\n",iostatus.Information)); //关闭文件句柄 ZwClose(hfile); ExFreePool(pBuffer); }

文件属性读取与修改：

VOID FileAttributeTest() { KdPrint(("\n-----------------ZwQueryInformationFile ZwSetInformationFile 测试 ---------------------!\n")); OBJECT_ATTRIBUTES objectAttributes; IO_STATUS_BLOCK iostatus; HANDLE hfile; UNICODE_STRING SymbolFileName; //初始化UNICODE_STRING字符串 //RtlInitUnicodeString( &SymbolFileName, L"\\Device\\HarddiskVolume1\\1.txt"); RtlInitUnicodeString( &SymbolFileName, L"\\??\\C:\\CreateFileTest.txt"); //初始化objectAttributes InitializeObjectAttributes(&objectAttributes, &SymbolFileName, OBJ_CASE_INSENSITIVE,//对大小写敏感 NULL, NULL ); //创建文件 NTSTATUS ntStatus = ZwCreateFile( &hfile, GENERIC_ALL, &objectAttributes, &iostatus, 0, FILE_ATTRIBUTE_NORMAL, FILE_SHARE_WRITE, FILE_OPEN_IF,//对文件打开，如果不存在则返回错误 FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0); if (NT_SUCCESS(ntStatus)) { KdPrint(("打开文件成功.\n")); } FILE_STANDARD_INFORMATION fsi; //读取文件长度 ntStatus = ZwQueryInformationFile(hfile, &iostatus, &fsi, sizeof(FILE_STANDARD_INFORMATION), FileStandardInformation); if (NT_SUCCESS(ntStatus)) { KdPrint(("文件长度 :%u\n",fsi.EndOfFile.QuadPart)); } //读出文件属性 FILE_BASIC_INFORMATION fbi; ZwQueryInformationFile(hfile, &iostatus, &fbi, sizeof(FILE_BASIC_INFORMATION), FileBasicInformation); //修改文件属性 fbi.FileAttributes=FILE_ATTRIBUTE_READONLY;//|FILE_ATTRIBUTE_READONLY|FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM|FILE_ATTRIBUTE_NORMAL; ntStatus = ZwSetInformationFile(hfile, &iostatus, &fbi, sizeof(FILE_BASIC_INFORMATION), FileBasicInformation); if (NT_SUCCESS(ntStatus)) { KdPrint(("设置文件属性成功.\n")); } else { KdPrint(("设置文件属性失败ntStatus=%x\n",ntStatus)); } //关闭文件句柄 ZwClose(hfile); }

具体请参看完整源码。