简单的虚拟设备驱动 ramdisk masm32源码

Ayala

其余部分看附件 有bug

1. 
   
2. .code
   
3. 
   
4. 
   
5. DispatchAddDevice proc uses esi edi ebx pDriverObject:PDRIVER_OBJECT,TargetDevice:PDEVICE_OBJECT
   
6.         LOCAL DestinationString:UNICODE_STRING
   
7.         LOCAL SourceString:UNICODE_STRING
   
8.         LOCAL pDrvExt:DWORD
   
9.         LOCAL pDevObj:DWORD
   
10.         LOCAL DevExt:_DEVICE_EXTENSION_u
    
11.        
    
12.        
    
13.         $DbgPrint $CTA0("DispatchAddDevice\n")
    
14.        
    
15.        
    
16.         invoke IoGetDriverObjectExtension,pDriverObject,offset DriverEntry
    
17.        
    
18.         mov pDrvExt,eax
    
19. 
    
20.        
    
21.         .if DWORD ptr (_DRIVER_EXTENSION_u ptr [eax]).DeviceInitialized == TRUE
    
22. 
    
23.                 mov edi,STATUS_DEVICE_ALREADY_ATTACHED        ;0C0000038h
    
24.                 jmp done0
    
25.         .endif
    
26.        
    
27.         lea esi,DevExt
    
28.         assume esi:ptr _DEVICE_EXTENSION_u
    
29.        
    
30.         invoke RtlZeroMemory,esi,SizeOf _DEVICE_EXTENSION_u
    
31.        
    
32. 
    
33.         invoke ExAllocatePoolWithTag,PagedPool,20,"2maR"
    
34.         mov [esi].DiskRegInfo.DriveLetter.Buffer,eax
    
35.         .if !eax
    
36.                 mov edi,STATUS_INSUFFICIENT_RESOURCES        ;0C000009Ah
    
37.                 jmp done1
    
38.         .endif       
    
39.                
    
40.         mov [esi].DiskRegInfo.DriveLetter.MaximumLength,20
    
41. 
    
42.                
    
43.         invoke _QueryDiskRegParameters,pDrvExt,addr [esi].DiskRegInfo       
    
44.        
    
45.        
    
46.         invoke RtlInitUnicodeString,addr DestinationString,$CTW0("\\Device\\HarddiskVolumeRD")
    
47.        
    
48.        
    
49.         mov eax,[esi].DiskRegInfo.MediaType        ;1~4
    
50.        
    
51. 
    
52. 
    
53.         .if eax==2       
    
54.                 mov eax,0
    
55.                 mov ecx,FILE_DEVICE_DISK        ;7       
    
56. ;        .elseif eax==1
    
57. ;                mov eax,0
    
58. ;                mov ecx,FILE_DEVICE_VIRTUAL_DISK ;24h
    
59.         .elseif eax==3
    
60.                 mov eax,1        ;FILE_REMOVABLE_MEDIA
    
61.                 mov ecx,FILE_DEVICE_DISK        ;7
    
62.         .elseif eax==4
    
63.                 mov eax,5        ;FILE_FLOPPY_DISKETTE or FILE_REMOVABLE_MEDIA
    
64.                 mov ecx,FILE_DEVICE_DISK        ;7
    
65.         .else
    
66.                 mov edi,STATUS_INSUFFICIENT_RESOURCES        ;0C000009Ah
    
67.                 jmp done0                               
    
68.         .endif
    
69.        
    
70.         or eax,FILE_DEVICE_SECURE_OPEN        ;100h
    
71.        
    
72.         lea edx,pDevObj
    
73.         invoke IoCreateDevice,pDriverObject,\
    
74.                                                   SizeOf _DEVICE_EXTENSION_u,\                ;DevExt size
    
75.                                                   addr DestinationString,\
    
76.                                                   ecx,\
    
77.                                                   eax,\
    
78.                                                   0,\
    
79.                                                   edx          
    
80.                                                   
    
81.         .if eax < SDWORD ptr 0
    
82.                 mov edi,eax
    
83.                 invoke ExFreePoolWithTag,[esi].DiskRegInfo.DriveLetter.Buffer,0
    
84.                 jmp done0
    
85.         .endif
    
86.                
    
87.                
    
88.         mov ebx,pDevObj
    
89.         assume ebx:ptr DEVICE_OBJECT
    
90.        
    
91.        
    
92.         mov esi,[ebx].DeviceExtension
    
93.        
    
94.         invoke RtlMoveMemory,esi,addr DevExt,sizeof _DEVICE_EXTENSION_u
    
95.        
    
96.        
    
97.         mov eax,TargetDevice
    
98.        
    
99.         mov [esi].TagDev,eax
    
100.         mov [esi].DeviceObject,ebx
     
101.        
     
102.         and dword ptr [esi].DevState,0
     
103.        
     
104.         invoke IoInitializeRemoveLockEx,addr [esi].RemoveLock,\
     
105.                                                                         "lmaR",\
     
106.                                                                         1,\
     
107.                                                                         0Ah,\
     
108.                                                                         SizeOf IO_REMOVE_LOCK
     
109. 
     
110.         or [ebx].Flags,DO_POWER_PAGABLE OR DO_DIRECT_IO; 2010h
     
111.        
     
112.         invoke KeInitializeSpinLock,addr [esi].Spinlock
     
113.        
     
114.        
     
115.         mov eax,[esi].DiskRegInfo.NumDevicePage
     
116.         .if !eax
     
117.                 mov edi,STATUS_INSUFFICIENT_RESOURCES;0C000009Ah
     
118.                 jmp done1
     
119.         .endif       
     
120.        
     
121.         mov [esi].MaxNumMDL,eax
     
122.        
     
123.         mov ecx,eax
     
124.         shl ecx,6               
     
125.         mov [esi].Cylinders,ecx
     
126.        
     
127.         mov ecx,[esi].DiskRegInfo.DisksizeM
     
128.         shl ecx,6        ;1024*1024/512/32
     
129.         mov [esi].Tracks,ecx
     
130.        
     
131.         shl eax,2        ;eax*4
     
132.        
     
133.         $DbgPrint $CTA0("Initialize Disk Space %d\n")
     
134.         invoke ExAllocatePoolWithTag,NonPagedPool,eax,"3maR"
     
135.         mov [esi].MDL,eax
     
136.         .if !eax
     
137.                 invoke DeviceContextCleanup,ebx
     
138.                 mov edi,STATUS_INSUFFICIENT_RESOURCES        ;0C000009Ah
     
139.                 jmp done1
     
140.         .endif
     
141.                        
     
142.         mov ecx,[esi].MaxNumMDL
     
143.         shl ecx,2
     
144.        
     
145.         invoke RtlZeroMemory,eax,ecx
     
146.                        
     
147.         invoke DeviceAllocMdl,esi       
     
148.         .if (eax < [esi].MaxNumMDL)
     
149.                 mov edi,STATUS_INSUFFICIENT_RESOURCES;0C000009Ah
     
150.                 jmp done1
     
151.         .endif       
     
152.        
     
153.        
     
154.         invoke DeviceLockPresentPage,esi,0       
     
155.         .if !eax
     
156.                 mov edi,STATUS_INSUFFICIENT_RESOURCES;0C000009Ah
     
157.                 jmp done1
     
158.         .endif
     
159.                        
     
160.         invoke FormatDisk,ebx
     
161.                
     
162.         invoke ExAllocatePoolWithTag,PagedPool,2Eh,"4maR"
     
163.         mov [esi].SymbolicLinkName.Buffer,eax
     
164.         .if !eax
     
165.                 mov edi,STATUS_INSUFFICIENT_RESOURCES;0C000009Ah
     
166.                 jmp done1
     
167.         .endif
     
168.        
     
169.         .if [esi].DiskRegInfo.DriveLetter._Length
     
170.                 invoke RtlInitUnicodeString,addr SourceString,$CTW0("\\DosDevices\")
     
171.                
     
172.                
     
173.                 lea edi,[esi].SymbolicLinkName
     
174.                 assume edi:PTR UNICODE_STRING
     
175.                
     
176.                 mov ax,SourceString._Length
     
177.                 mov [edi]._Length,ax
     
178.                 mov [edi].MaximumLength,2Eh
     
179.                
     
180.                 assume edi:nothing
     
181.                
     
182.                 invoke RtlCopyUnicodeString,edi,addr SourceString
     
183.                 invoke RtlAppendUnicodeStringToString,edi,addr [esi].DiskRegInfo.DriveLetter
     
184.                 invoke IoCreateSymbolicLink,edi,addr DestinationString
     
185.                
     
186.                 mov edi,eax
     
187.                 .if eax < SDWORD ptr 0
     
188.                         jmp done1
     
189.                 .endif
     
190.                 or dword ptr [esi].flag,1
     
191.         .endif
     
192.        
     
193.         invoke IoAttachDeviceToDeviceStack,pDevObj,TargetDevice
     
194.         mov [esi].LoDeviceObject,eax
     
195.         .if !eax
     
196.                 mov edi,STATUS_NO_SUCH_DEVICE        ;0C000000Eh
     
197.                 jmp done1
     
198.         .endif
     
199.        
     
200.         mov eax,pDrvExt
     
201.        
     
202.         mov (_DRIVER_EXTENSION_u ptr [eax]).DeviceInitialized,TRUE       
     
203.        
     
204.         and [ebx].Flags,7Fh
     
205.        
     
206.         mov edi,STATUS_SUCCESS
     
207.         jmp done0
     
208.        
     
209.         assume esi:nothing
     
210. done1:       
     
211.         invoke DeviceContextCleanup,pDevObj               
     
212. 
     
213. done0:       
     
214.         $DbgPrint $CTA0("DeviceInitialize %08X\n"),edi
     
215.         mov eax,edi
     
216.        
     
217. ;done:       
     
218.         ret
     
219. DispatchAddDevice endp
     

复制代码

Ayala

原来附件没上传成功！