《Windows内核安全与驱动开发》书-学习笔记2

codecook

今天就没那么多废话了，今天学习第四章文件、注册表、线程。
先上一个例子来补充一下书中的ZwCreateFile，为什么要补充，是因为书中的例子省略了一句代码，出现了编译错误：

1. //打开或创建一个文件
   
2.         HANDLE file_handle=NULL;
   
3.         NTSTATUS status;
   
4.         IO_STATUS_BLOCK io_status_block;    //这里，书中的实例缺少了这个定义，造成代码不完整。
   
5.         OBJECT_ATTRIBUTES object_attributes;
   
6.         UNICODE_STRING ufile_name=RTL_CONSTANT_STRING(L"\\??\\C:\\a.txt");  // C:是一个符号链接，符号链接一般都在\\??\\下
   
7.         //OBJ_CASE_INSENSITIVE目标文件不区分大小写  OBJ_KERNEL_HANDLE文件需用内核句柄
   
8.         InitializeObjectAttributes(&object_attributes,&ufile_name,OBJ_CASE_INSENSITIVE|OBJ_KERNEL_HANDLE,NULL,NULL);
   
9.         status=ZwCreateFile(
   
10.                 &file_handle,                                       
    
11.                 GENERIC_READ|GENERIC_WRITE,
    
12.                 &object_attributes,
    
13.                 &io_status_block,
    
14.                 NULL,
    
15.                 FILE_ATTRIBUTE_NORMAL,
    
16.                 FILE_SHARE_READ,
    
17.                 FILE_OPEN_IF,
    
18.                 FILE_NON_DIRECTORY_FILE|FILE_RANDOM_ACCESS|FILE_SYNCHRONOUS_IO_NONALERT,
    
19.                 NULL,
    
20.                 0
    
21.                 );
    
22.         ZwClose(file_handle);

复制代码

接着同样的问题。ZwReadFile与ZwWriteFile书中并没有给出单独用法，而是用了一个函数MyCopyFile，但是这个例子也是不齐全，需要读者自行完善，我简单处理，实现了这个例子的效果：

1. 
   
2. //复制一个已存在的文件到另一个新文件
   
3.         HANDLE aa_file_handle=NULL;
   
4.         NTSTATUS status;
   
5.         IO_STATUS_BLOCK io_status_block;
   
6.         OBJECT_ATTRIBUTES object_attributes;
   
7.         UNICODE_STRING ufile_name=RTL_CONSTANT_STRING(L"\\??\\C:\\aa.txt");
   
8.         //OBJ_CASE_INSENSITIVE目标文件不区分大小写  OBJ_KERNEL_HANDLE文件需用内核句柄
   
9.         InitializeObjectAttributes(&object_attributes,&ufile_name,OBJ_CASE_INSENSITIVE|OBJ_KERNEL_HANDLE,NULL,NULL);
   
10.         status=ZwCreateFile(
    
11.                 &aa_file_handle,
    
12.                 GENERIC_READ|GENERIC_WRITE,
    
13.                 &object_attributes,
    
14.                 &io_status_block,
    
15.                 NULL,
    
16.                 FILE_ATTRIBUTE_NORMAL,
    
17.                 FILE_SHARE_READ,
    
18.                 FILE_OPEN_IF,
    
19.                 FILE_NON_DIRECTORY_FILE|FILE_RANDOM_ACCESS|FILE_SYNCHRONOUS_IO_NONALERT,
    
20.                 NULL,
    
21.                 0
    
22.                 );
    
23. 
    
24.         HANDLE bb_file_handle=NULL;       
    
25.         RtlInitUnicodeString(&ufile_name,L"\\??\\C:\\bb.txt");
    
26.         //OBJ_CASE_INSENSITIVE目标文件不区分大小写  OBJ_KERNEL_HANDLE文件需用内核句柄
    
27.         InitializeObjectAttributes(&object_attributes,&ufile_name,OBJ_CASE_INSENSITIVE|OBJ_KERNEL_HANDLE,NULL,NULL);
    
28.         status=ZwCreateFile(
    
29.                 &bb_file_handle,
    
30.                 GENERIC_READ|GENERIC_WRITE,
    
31.                 &object_attributes,
    
32.                 &io_status_block,
    
33.                 NULL,
    
34.                 FILE_ATTRIBUTE_NORMAL,
    
35.                 FILE_SHARE_READ,
    
36.                 FILE_OPEN_IF,
    
37.                 FILE_NON_DIRECTORY_FILE|FILE_RANDOM_ACCESS|FILE_SYNCHRONOUS_IO_NONALERT,
    
38.                 NULL,
    
39.                 0
    
40.                 );
    
41. 
    
42.         PVOID buffe =NULL;
    
43.         LARGE_INTEGER offset={0};
    
44.         ULONG length = 1;
    
45.     buffe = ExAllocatePool(NonPagedPool, 1);  //书中缺少了这句，不分配内存实例做不下去。
    
46.         while(1)
    
47.         {
    
48.                 status = ZwReadFile(aa_file_handle,NULL,NULL,NULL,&io_status_block,buffe,length,&offset,NULL);
    
49. 
    
50.                 if (!NT_SUCCESS(status))
    
51.                 {
    
52.                         if(status==STATUS_END_OF_FILE)
    
53.                                 status =STATUS_SUCCESS;
    
54.                         break;
    
55.                 }
    
56.                 ZwWriteFile(bb_file_handle,NULL,NULL,NULL,&io_status_block,buffe,length,&offset,NULL);               
    
57.                 offset.QuadPart++;
    
58.         }
    
59.         ZwClose(aa_file_handle);
    
60.         ZwClose(bb_file_handle);
    
61.         if(buffe!=NULL)
    
62.                 ExFreePool(buffe);

复制代码

今天的收获还不错，温故了内存、字符串等，学习了一个创建文件，和复制文件。
今天的问题是不知道如何剪切一个文件。

康小泡

都学内核了啊。