DDK 最純淨的Hello world

只包含DriverEntry,IRP_MJ_DEVICE_CONTROL,DriverUnload函數

是DDK的基本結構

沒有任何功能,所有DDK都要以這個結構來寫

 

1. #include <ntddk.h>
5. #define DEVICE_NAME L"//Device//hello"
6. #define DOS_NAME    L"//??//hello"
8. void Unload(IN PDRIVER_OBJECT DriverObject);
9. NTSTATUS MyIoControl(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp);
10. NTSTATUS MydrvDispatch (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp);
12. NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject , IN PUNICODE_STRING RegistryPath)
13. {
14.     
15.     UNICODE_STRING  nameString, linkString;
16.     NTSTATUS  status=STATUS_SUCCESS;
17.     ULONG     i;
18.     PDEVICE_OBJECT      deviceObject;
19.     RtlInitUnicodeString(&nameString,DEVICE_NAME);
20.     RtlInitUnicodeString(&linkString,DOS_NAME);
21.     status = IoCreateDevice( DriverObject,0,&nameString,FILE_DEVICE_DISK_FILE_SYSTEM,FILE_DEVICE_SECURE_OPEN,TRUE,&deviceObject);   
22.     if (!NT_SUCCESS( status ))
23.         return status;
24.     
25.     deviceObject->Flags|=DO_BUFFERED_IO;
26.     status = IoCreateSymbolicLink (&linkString, &nameString);
27.     if (!NT_SUCCESS( status ))
28.     {
29.         IoDeleteDevice (DriverObject->DeviceObject);
30.         return status;
31.     }  
35.     
38.     for ( i = 0; i < IRP_MJ_MAXIMUM_FUNCTION; i++)
39.     {
40.         DriverObject->MajorFunction[i] = MydrvDispatch;
41.     }
42.     DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = MyIoControl;
43.     DriverObject->DriverUnload=Unload;
44.     
45.     DbgPrint("DriverEntry.../n");
46.     
47.          
49.     return status;
50.     
51. }
53. void Unload(IN PDRIVER_OBJECT DriverObject)
54. {
55.     
56.     PDEVICE_OBJECT deviceObject = DriverObject->DeviceObject;
57.     UNICODE_STRING uniWin32NameString;
60.     RtlInitUnicodeString( &uniWin32NameString, DOS_NAME );  
61.     IoDeleteSymbolicLink( &uniWin32NameString );   
62.     if ( deviceObject != NULL )
63.     {
64.         IoDeleteDevice( deviceObject );
65.     }
67.     return;
68. }
70. NTSTATUS MydrvDispatch (IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
71. {
72.     Irp->IoStatus.Status = STATUS_SUCCESS;
73.     Irp->IoStatus.Information = 0L;
74.     IoCompleteRequest( Irp, 0 );
75.     return Irp->IoStatus.Status;
76. }
78. NTSTATUS MyIoControl(IN PDEVICE_OBJECT DeviceObject, IN PIRP Irp)
79. {
81.     NTSTATUS  status=STATUS_SUCCESS;
83.     
84.     PIO_STACK_LOCATION irpStack; 
85.     PVOID ioBuffer; 
86.     ULONG inputBufferLength; 
87.     ULONG outputBufferLength;       
88.     ULONG ioControlCode;
89.     
91.     irpStack = IoGetCurrentIrpStackLocation(Irp); 
92.     ioControlCode= irpStack->Parameters.DeviceIoControl.IoControlCode; 
93.     ioBuffer = Irp->AssociatedIrp.SystemBuffer; 
94.     inputBufferLength = irpStack->Parameters.DeviceIoControl.InputBufferLength; 
95.     outputBufferLength = irpStack->Parameters.DeviceIoControl.OutputBufferLength;
98.     switch (ioControlCode)
99.     {
100.         
101.     
102.         
106.         
107.     }
108.     Irp->IoStatus.Status = status; 
109.     Irp->IoStatus.Information = outputBufferLength; 
110.     IoCompleteRequest(Irp, IO_NO_INCREMENT); 
111.     return    status;
112.     
114. }