FirstDriver

forwe

驱动层：

1. #include "ntddk.h"
   
2. 
   
3. #define INITCODE        code_seg("INIT")
   
4. #define PAGECODE        code_seg("PAGE")
   
5. 
   
6. 
   
7. #pragma PAGECODE
   
8. void MyDriverUnload(PDRIVER_OBJECT pDriverObject)
   
9. {
   
10.         UNICODE_STRING symName;
    
11. 
    
12.         RtlInitUnicodeString(&symName, L"\\??\\FirstDeviceLinkName");
    
13. 
    
14.         if (pDriverObject->DeviceObject != NULL)
    
15.         {
    
16.                 IoDeleteSymbolicLink(&symName);
    
17.                 IoDeleteDevice(pDriverObject->DeviceObject);
    
18.                 KdPrint(("Delete device success!"));
    
19.         }
    
20. 
    
21. }
    
22. 
    
23. #pragma INITCODE
    
24. NTSTATUS CreateMyDevice(PDRIVER_OBJECT pDriverObject)
    
25. {
    
26.         NTSTATUS Status;
    
27.         PDEVICE_OBJECT pDevObj;
    
28.         UNICODE_STRING DevName;
    
29.         UNICODE_STRING symName;
    
30. 
    
31.         RtlInitUnicodeString(&DevName, L"\\Device\\FirstDevice");
    
32. 
    
33.         Status = IoCreateDevice(pDriverObject, 0, &DevName, FILE_DEVICE_UNKNOWN,
    
34.                 0, TRUE, &pDevObj);
    
35. 
    
36.         if (!NT_SUCCESS(Status))
    
37.         {
    
38.                 return Status;
    
39.         }
    
40.        
    
41.         pDevObj->Flags |= DO_BUFFERED_IO;
    
42. 
    
43.         RtlInitUnicodeString(&symName, L"\\??\\FirstDeviceLinkName");
    
44.         Status = IoCreateSymbolicLink(&symName, &DevName);
    
45. 
    
46.         if (!NT_SUCCESS(Status))
    
47.         {
    
48.                 IoDeleteDevice(pDevObj);
    
49.                 return Status;
    
50.         }
    
51.         return STATUS_SUCCESS;
    
52. }
    
53. 
    
54. #pragma PAGECODE
    
55. NTSTATUS CreateCompleteRoutine(PDEVICE_OBJECT pDeviceObj, PIRP pIrp)
    
56. {
    
57.         NTSTATUS Status;
    
58.         Status = STATUS_SUCCESS;
    
59. 
    
60.         KdPrint(("Create"));
    
61. 
    
62.         pIrp->IoStatus.Status = Status;
    
63.         pIrp->IoStatus.Information = 0;
    
64.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
    
65.         return Status;
    
66. }
    
67. 
    
68. #pragma PAGECODE
    
69. NTSTATUS CloseCompleteRoutine(PDEVICE_OBJECT pDeviceObj, PIRP pIrp)
    
70. {
    
71.         NTSTATUS Status;
    
72.         Status = STATUS_SUCCESS;
    
73. 
    
74.         KdPrint(("Close"));
    
75. 
    
76.         pIrp->IoStatus.Status = Status;
    
77.         pIrp->IoStatus.Information = 0;
    
78.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
    
79.         return Status;
    
80. }
    
81. 
    
82. #pragma PAGECODE
    
83. NTSTATUS ReadCompleteRoutine(PDEVICE_OBJECT pDeviceObj, PIRP pIrp)
    
84. {
    
85.         NTSTATUS Status;
    
86.         Status = STATUS_SUCCESS;
    
87. 
    
88.         KdPrint(("Read"));
    
89. 
    
90.         pIrp->IoStatus.Status = Status;
    
91.         pIrp->IoStatus.Information = 0;
    
92.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
    
93.         return Status;
    
94. }
    
95. 
    
96. #pragma PAGECODE
    
97. NTSTATUS WriteCompleteRoutine(PDEVICE_OBJECT pDeviceObj, PIRP pIrp)
    
98. {
    
99.         NTSTATUS Status;
    
100.         Status = STATUS_SUCCESS;
     
101. 
     
102.         KdPrint(("Write"));
     
103. 
     
104.         pIrp->IoStatus.Status = Status;
     
105.         pIrp->IoStatus.Information = 0;
     
106.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
     
107.         return Status;
     
108. }
     
109. 
     
110. #pragma INITCODE
     
111. NTSTATUS DriverEntry(PDRIVER_OBJECT pDriverObject, PUNICODE_STRING pRegistryPath)
     
112. {
     
113.         NTSTATUS Status;
     
114.         Status = CreateMyDevice(pDriverObject);
     
115. 
     
116.         if (!NT_SUCCESS(Status))
     
117.         {
     
118.                 KdPrint(("Create device failed!"));
     
119.         }
     
120.         else
     
121.         {
     
122.                 KdPrint(("Create device success!"));
     
123.                 KdPrint(("%wZ", pRegistryPath));
     
124.         }
     
125. 
     
126.         pDriverObject->MajorFunction[IRP_MJ_CREATE] = CreateCompleteRoutine;
     
127.         pDriverObject->MajorFunction[IRP_MJ_CLOSE] = CloseCompleteRoutine;
     
128.         pDriverObject->MajorFunction[IRP_MJ_READ] = ReadCompleteRoutine;
     
129.         pDriverObject->MajorFunction[IRP_MJ_WRITE] = WriteCompleteRoutine;
     
130. 
     
131. 
     
132.         pDriverObject->DriverUnload = MyDriverUnload;
     
133.         return STATUS_SUCCESS;
     
134. }

复制代码

应用层：

1. void CCheckMyDriverDlg::OnButton1()
   
2. {
   
3.         // TODO: Add your control notification handler code here
   
4.                                                        
   
5.         HANDLE hDevice = CreateFile(TEXT("\\\\.\\FirstDeviceLinkName"), GENERIC_READ | GENERIC_WRITE, 0, NULL,
   
6.                                                         OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
   
7.        
   
8.         if (hDevice == INVALID_HANDLE_VALUE)
   
9.         {
   
10.                 CString str;
    
11.                 str.Format("%d", GetLastError());
    
12.                 AfxMessageBox("Open device failed! " + str);
    
13.                 return;
    
14.         }
    
15. 
    
16.         CloseHandle(hDevice);
    
17. }

复制代码