<Windows内核安全与驱动开发>-学习笔记4

codecook

接着复习。
今天写了一个函数，函数的功能是获取文件句柄，可是我不知道通过函数得到的文件句柄，用完再利用ZwClose销毁句柄，这样操作不知道会不会造成内存泄漏，如下：

1. HANDLE GetFileHandle(WCHAR *pwchar_path)
   
2. {       
   
3.         //DbgPrint("%S\n",pwchar_path);
   
4.         USHORT lnFILEPATH_LENGTH = wcslen(pwchar_path);
   
5.         //DbgPrint("%d\n",lnFILEPATH_LENGTH);
   
6.         USHORT lnHEAD =4;
   
7.         WCHAR *pchar =(PWCHAR)ExAllocatePool(NonPagedPool,(lnFILEPATH_LENGTH +lnHEAD)*2 );
   
8.         wcscpy(pchar,L"\\??\");
   
9.         wcscpy(pchar+4,pwchar_path);
   
10.         HANDLE file_handle=NULL;
    
11.         DbgPrint("%S\n",pchar);
    
12.         NTSTATUS status;
    
13.         IO_STATUS_BLOCK io_status_block;
    
14.         OBJECT_ATTRIBUTES object_attributes;
    
15.         UNICODE_STRING ufile_name;
    
16.         ufile_name.Buffer = pchar;
    
17.         //DbgPrint("%d",wcslen(pchar));
    
18.         ufile_name.Length =ufile_name.MaximumLength = wcslen(pchar)*2 ;
    
19.         DbgPrint("%d\n", wcslen(pchar));
    
20.         //DbgPrint("%wZ\n",&ufile_name);
    
21. 
    
22.         DbgPrint("%d,%S,%d\n",ufile_name.Length,ufile_name.Buffer,ufile_name.MaximumLength);
    
23.        
    
24.         InitializeObjectAttributes(&object_attributes,&ufile_name,OBJ_CASE_INSENSITIVE|OBJ_KERNEL_HANDLE,NULL,NULL);
    
25.         status=ZwCreateFile(
    
26.                 &file_handle,
    
27.                 GENERIC_READ|GENERIC_WRITE,
    
28.                 &object_attributes,
    
29.                 &io_status_block,
    
30.                 NULL,
    
31.                 FILE_ATTRIBUTE_NORMAL,
    
32.                 FILE_SHARE_READ,
    
33.                 FILE_OPEN_IF,
    
34.                 FILE_NON_DIRECTORY_FILE|FILE_RANDOM_ACCESS|FILE_SYNCHRONOUS_IO_NONALERT,
    
35.                 NULL,
    
36.                 0
    
37.                 );
    
38.         //ZwClose(file_handle);
    
39.         ExFreePool(pchar);
    
40.         return file_handle;
    
41. }
    
42. NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject, IN PUNICODE_STRING  RegistryPath)
    
43. {
    
44.    HANDLE hSrcFile =GetFileHandle(L"c:\\ABAB.txt");
    
45. .....
    
46. ZwClose(hSrcFile);
    
47. }
    

复制代码

同时今天我也学习了一个wcscpy函数，这个函数用来拼接两个字符串比较方便，不知道安全不？

fishkiss

看起来的确是方便了一点。。但是，就像书上说的那样。。只要你可以保证字符串用结束的话，那就是安全的。。不过我觉得，如果有那个时间都确认这个字符串怎么结束，还不如用那个复杂点的函数。。习惯比较重要一点。。。
至于句柄安全不安全的话，一般是没问题的。。只是最好能自己检查一下。。

你时间到是很多啊。。。我还上班。。