0
回答
菜鸟之驱动开发13
利用AWS快速构建适用于生产的无服务器应用程序,免费试用12个月>>>   

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

相关的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);
}


具体请参看完整源码


原文链接:http://blog.csdn.net/favormm/article/details/6727639
<无标签>
举报
长平狐
发帖于5年前 0回/132阅
顶部