驱动文件操作

forwe

1. #include "ntddk.h"
   
2. 
   
3. #define PAGEDCODE code_seg("PAGE")
   
4. #define LOCKEDCODE code_seg()
   
5. #define INITCODE code_seg("INIT")
   
6. 
   
7. #define PAGEDDATA data_seg("PAGE")
   
8. #define LOCKEDDATA data_seg()
   
9. #define INITDATA data_seg("INIT")
   
10. 
    
11. 
    
12. #pragma PAGEDCODE
    
13. NTSTATUS MyCreateFile()
    
14. {
    
15.         NTSTATUS Status;
    
16.         HANDLE hFile;
    
17.         UNICODE_STRING usFileName;
    
18.         OBJECT_ATTRIBUTES FileObjAttr;
    
19.         IO_STATUS_BLOCK IoStatusBlock;
    
20. 
    
21.         RtlInitUnicodeString(&usFileName, L"\\??\\c:\\1.txt");
    
22. 
    
23.         memset(&FileObjAttr, 0, sizeof(OBJECT_ATTRIBUTES));
    
24. 
    
25.         InitializeObjectAttributes(&FileObjAttr, &usFileName, OBJ_CASE_INSENSITIVE, NULL, NULL);
    
26. 
    
27.         Status = ZwCreateFile(&hFile, GENERIC_ALL, &FileObjAttr, &IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL,
    
28.                 FILE_SHARE_READ,  FILE_OPEN_IF, FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT, NULL, 0);
    
29.        
    
30.         if (!NT_SUCCESS(Status))
    
31.         {
    
32.                 return Status;
    
33.         }
    
34. 
    
35.         KdPrint(("create file succcess!"));
    
36. 
    
37.         ZwClose(hFile);
    
38. 
    
39.         return Status;
    
40. }
    
41. 
    
42. 
    
43. #pragma PAGEDCODE
    
44. NTSTATUS MyOpenFile()
    
45. {
    
46.         NTSTATUS Status;
    
47.         HANDLE hFile;
    
48.         UNICODE_STRING usFileName;
    
49.         OBJECT_ATTRIBUTES FileObjAttr;
    
50.         IO_STATUS_BLOCK IoStatusBlock;
    
51. 
    
52.         FILE_BASIC_INFORMATION fbi;
    
53. 
    
54.         LARGE_INTEGER number;
    
55. 
    
56.         PCHAR pBuffer = (PCHAR)ExAllocatePool(PagedPool, 20);
    
57. 
    
58.         RtlInitUnicodeString(&usFileName, L"\\??\\c:\\1.txt");
    
59. 
    
60.         memset(&FileObjAttr, 0, sizeof(OBJECT_ATTRIBUTES));
    
61. 
    
62.         InitializeObjectAttributes(&FileObjAttr, &usFileName, OBJ_CASE_INSENSITIVE, NULL, NULL);
    
63. 
    
64.         Status = ZwOpenFile(&hFile, GENERIC_ALL, &FileObjAttr, &IoStatusBlock, FILE_SHARE_READ, FILE_NON_DIRECTORY_FILE);
    
65. 
    
66.         if (!NT_SUCCESS(Status))
    
67.         {
    
68.                 KdPrint(("open file faild! status:0x%X", Status));
    
69.                 return Status;
    
70.         }
    
71. 
    
72.         KdPrint(("open file succcess!"));
    
73. 
    
74.         Status = ZwQueryInformationFile(hFile, &IoStatusBlock, &fbi,
    
75.                 sizeof(FILE_BASIC_INFORMATION), FileBasicInformation);
    
76. 
    
77.         if (!NT_SUCCESS(Status))
    
78.         {
    
79.                 KdPrint(("Error code: 0x%X",Status));
    
80.                 ZwClose(hFile);
    
81.                 return Status;
    
82.         }
    
83. 
    
84.         //fbi.FileAttributes |= FILE_ATTRIBUTE_READONLY;
    
85. 
    
86.         Status = ZwSetInformationFile(hFile, &IoStatusBlock, &fbi,
    
87.                 sizeof(FILE_BASIC_INFORMATION), FileBasicInformation);
    
88. 
    
89.         if (!NT_SUCCESS(Status))
    
90.         {
    
91.                 KdPrint(("Error code: 0x%X",Status));
    
92.                 ZwClose(hFile);
    
93.                 return Status;
    
94.         }
    
95. 
    
96.         number.QuadPart = 0;
    
97.         Status = ZwReadFile(hFile, NULL, NULL, NULL, &IoStatusBlock, pBuffer, 8, &number, NULL);
    
98. 
    
99.         if (!NT_SUCCESS(Status))
    
100.         {
     
101.                 KdPrint(("Error code: 0x%X",Status));
     
102.                 ExFreePool(pBuffer);
     
103.                 ZwClose(hFile);
     
104.                 return Status;
     
105.         }
     
106. 
     
107.         KdPrint(("%s", pBuffer));
     
108. 
     
109.         RtlCopyMemory(pBuffer, "12345678901234567890", 20);
     
110. 
     
111.         Status = ZwWriteFile(hFile, NULL, NULL, NULL, &IoStatusBlock, pBuffer, 20, &number, NULL);
     
112. 
     
113.         if (!NT_SUCCESS(Status))
     
114.         {
     
115.                 KdPrint(("Error code: 0x%X",Status));
     
116.                 ExFreePool(pBuffer);
     
117.                 ZwClose(hFile);
     
118.                 return Status;
     
119.         }
     
120. 
     
121.         ExFreePool(pBuffer);
     
122.         ZwClose(hFile);
     
123. 
     
124.         return Status;
     
125. }
     
126. 
     
127. 
     
128. #pragma PAGEDCODE
     
129. VOID MyDriverUnload(IN PDRIVER_OBJECT pDriverObject)
     
130. {
     
131. 
     
132.         KdPrint(("DriverEntry unLoading...\n"));
     
133. 
     
134. }
     
135. 
     
136. 
     
137. 
     
138. #pragma INITCODE
     
139. NTSTATUS DriverEntry(IN PDRIVER_OBJECT pDriverObject, IN PUNICODE_STRING RegistryPath)
     
140. {
     
141.         NTSTATUS status = STATUS_SUCCESS;
     
142. 
     
143.         MyCreateFile();       
     
144. 
     
145.         pDriverObject->DriverUnload = MyDriverUnload;
     
146.         return status;
     
147. }

复制代码