Windows内核编程基础篇之文件读/写操作

浅浅的花香味﹌ 2022-08-04 14:46 304阅读 0赞

打开文件后，最重要是的操作在是对文件的读/写。读写的方法是对称的知识参数输入与输出方向不同。读取文件内容一般用ZwReadFile，写文件一般用ZwWriteFile。

先看看 ZwReadFile 结构吧：

NTSTATUS ZwReadFile(
      _In_     HANDLE           FileHandle,
      _In_opt_ HANDLE           Event,
      _In_opt_ PIO_APC_ROUTINE  ApcRoutine,
      _In_opt_ PVOID            ApcContext,
      _Out_    PIO_STATUS_BLOCK IoStatusBlock,
      _Out_    PVOID            Buffer,
      _In_     ULONG            Length,
      _In_opt_ PLARGE_INTEGER   ByteOffset,
      _In_opt_ PULONG           Key
    );

参数解释：

FileHandle：是ZwCreateFile成功后所得到的FileHandle。若是内核句柄，ZwReadFile和ZwCreateFile并不需要在同一个进程中。句柄是个进程通用的。

Event ： 一个事件，用于异步完成读时。设置为NULL 即可。

ApcRoutine ：回调例程，用于异步完成读时，设置为 NULL即可。

IoStatusBlock ：返回结构状态，同ZwCreateFile中的同名参数。

Buffer ：缓冲区。如果读文件的内容成功，则分内容被读到这个缓冲区里。

Length ：描述缓冲区的长度。这个长度也就是试图读取文件的长度。

ByteOffset ： 要读取的文件的偏移量，也就是要读取的内容在文件中的位置，一般不要设置为NULL，文件句柄不一定支持直接读取当前偏移量。

Key : 读取文件时使用的一种附加信息，一般不用，设置为NULL即可。

ZwReadFile 的返回值： 调用成功，则返回STATUS\_SUCCESSS。只要读取到的任意多个字节，返回值都是STATUS\_SUCCESS。及时试图读取的长度超出了文件本来的大小。但是如果仅是读取文件长度之外的部分，则返回的是：STATUS\_END\_OF\_FILE。

ZwWriteFile 参数与ZwReadFile完全相同。

说了这么多，看看代码,简单的实现了文件的拷贝。

NTSTATUS CopyFile(PUNICODE_STRING target_path, PUNICODE_STRING source_path)
    {
     //---源文件的句柄
    	HANDLE  target = NULL, source = NULL;
    
    	//----拷贝文件的缓冲区
    	PVOID buffer = NULL;
    	LARGE_INTEGER offset = { 0 };
    	IO_STATUS_BLOCK io_status = { 0 };
    
    	do 
    	{
    		//---这里就开始 获取 源文件 和目标路径句柄，并为 buffer分配一个页面。然后就是循环读取文件，再写入文件。
    		while (1)
    		{
    			///--每次读取4KB
    			length = 4 * 1024;
    			//---
    			status = ZwReadFile(
    				source,
    				NULL,
    				NULL,
    				NULL,
    				&my_io_status,
    				buffer,
    				length,
    				&offset,
    				NULL);
    
    			if (!NT_SUCCESS(status))
    			{
    				//---若状态为STATUS_END_OF_FILE，则文件的拷贝结束。
    				if (STATUS_END_OF_FILE == status)
    				{
    					status = STATUS_SUCCESS;
    				}
    				break;
    			}
    			//---得到文件被读取到的长度
    			length = IoStatus.Information;
    			//---下面开始写文件
    			status = ZwWriteFile(
    							target,
    							NULL,
    							NULL,
    							NULL,
    							&my_io_status,
    							buffer,
    							length,
    							&offset,
    							NULL);
    			if (!NT_SUCCESS(status))
    			{
    				break;
    			}
    			///----继续，直到出现 STATUS_END_OF_FILE的时候结束
    			offset.QuadPart += length;
    		}
    	} while (0);
    
    	//---释放资源，关闭句柄
    	if (NULL !=target)
    	{ 
    		ZwClose(target);
    	}
    	if (NULL != source)
    	{
    		ZwClose(source);
    	}
    	if (NULL != buffer)
    	{
    		ExFreePool(buffer);
    	}
    
    	return STATUS_SUCCESS;
    }