论磁盘内存映射

元始天尊

       这个题目有点眼熟吧，文件内存映射用的比较多，用于大文件读写效率较高。其原理是将映射的内存页的来源标记为磁盘驱动器，这样对地址的读写产生中断时，会直接操作磁盘，这样对打文件的读写速度提高很多。然而这种技术微软只用在文件上，对于目录、磁盘和设备均不能使用。而普通缓冲区buffer方式读写，总是要有内存拷贝，所以相对慢一些，在我3年前做数据恢复的时候，为了研究大文件恢复接触到文件内存映射，第一次就很纳闷到底有没有磁盘映射，这样我就不需要总是读来读去。下面给出这2种方式的区别：
Buffered方式：
CreateFile
ReadFile/WriteFile
CloseHandle

mapping方式：
CreateFile
CreateFileMapping
MapViewOfFile
  指针操作
UnmapViewOfFile
CloseHandle
CloseHandle

       在这里我对相关内核API做了逆向和破解，写了一个驱动支持对于磁盘实现内存映射。主要是对IRP_MJ_QUERY_INFORMATION和FastIoQueryStandardInfo的hook，代码如下：
驱动层：

1. 
   
2. #include <ntddk.h>
   
3. #include "DeviceMappingCommon.h"
   
4. 
   
5. typedef struct _DEVICE_MAPPING
   
6. {
   
7.         HANDLE hDevice;
   
8.         PFILE_OBJECT pFileObject;
   
9.         LARGE_INTEGER dwMaximumSize;
   
10. }DEVICE_MAPPING, *PDEVICE_MAPPING;
    
11. 
    
12. PDEVICE_MAPPING gMappingData = NULL;
    
13. PDRIVER_DISPATCH OriginQuery = NULL;
    
14. PFAST_IO_QUERY_STANDARD_INFO OriginFastDispatch = NULL;
    
15. FAST_MUTEX gLock;
    
16. 
    
17. extern "C"
    
18. {
    
19.         BOOLEAN NTAPI ObFindHandleForObject(PEPROCESS, PVOID, PVOID, PVOID, PHANDLE);
    
20.         NTSTATUS NTAPI NtQueryVolumeInformationFile (HANDLE, PIO_STATUS_BLOCK, PVOID, ULONG, FS_INFORMATION_CLASS);
    
21. }
    
22. 
    
23. VOID DeviceMappingDriverUnload(PDRIVER_OBJECT DriverObject)
    
24. {
    
25.         ExAcquireFastMutex(&gLock);
    
26.         if(gMappingData)
    
27.         {
    
28.                 ExFreePool((PVOID)gMappingData);
    
29.                 gMappingData = NULL;
    
30.         }
    
31.         ExReleaseFastMutex(&gLock);
    
32.         if(DriverObject->DeviceObject)
    
33.         {
    
34.                 UNICODE_STRING DriverName;
    
35.                 RtlInitUnicodeString(&DriverName, L"\\Device\\DeviceMapping");
    
36.                 IoDeleteSymbolicLink(&DriverName);
    
37.                 IoDeleteDevice(DriverObject->DeviceObject);
    
38.         }
    
39.         return;
    
40. }
    
41. 
    
42. NTSTATUS DeviceMappingQueryInformation(PDEVICE_OBJECT pDevObj, PIRP pIrp)
    
43. {
    
44.         PIO_STACK_LOCATION stack = IoGetCurrentIrpStackLocation(pIrp);
    
45.         if(stack->Parameters.QueryFile.FileInformationClass == FileStandardInformation && stack->FileObject)
    
46.         {
    
47.                 HANDLE hDevice = NULL;
    
48.                 BOOLEAN IsRef = FALSE;
    
49.                 ExAcquireFastMutex(&gLock);
    
50.                 if(gMappingData && stack->FileObject == gMappingData->pFileObject)
    
51.                 {
    
52.                         PFILE_STANDARD_INFORMATION FileInformation = (PFILE_STANDARD_INFORMATION)pIrp->AssociatedIrp.SystemBuffer;
    
53.                         FileInformation->EndOfFile.QuadPart = gMappingData->dwMaximumSize.QuadPart;
    
54.                         pIrp->IoStatus.Status = STATUS_SUCCESS;
    
55.                         pIrp->IoStatus.Information = sizeof(FILE_STANDARD_INFORMATION);
    
56.                         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
    
57.                         IsRef = TRUE;
    
58.                 }
    
59.                 ExReleaseFastMutex(&gLock);
    
60.                 if(IsRef)
    
61.                         return STATUS_SUCCESS;
    
62.         }
    
63.         OriginQuery(pDevObj, pIrp);
    
64. }
    
65. 
    
66. BOOLEAN DeviceMappingFastIo(PFILE_OBJECT FileObject, BOOLEAN Wait, PFILE_STANDARD_INFORMATION Buffer,PIO_STATUS_BLOCK IoStatus,struct _DEVICE_OBJECT *DeviceObject)
    
67. {
    
68.         BOOLEAN ret = OriginFastDispatch(FileObject, Wait, Buffer, IoStatus, DeviceObject);
    
69.         ExAcquireFastMutex(&gLock);
    
70.         if(FileObject == gMappingData->pFileObject)
    
71.         {
    
72.                 Buffer->Directory = FALSE;
    
73.         }
    
74.         ExReleaseFastMutex(&gLock);
    
75.         return ret;
    
76. }
    
77. 
    
78. NTSTATUS DeviceMappingIOControl(PDEVICE_OBJECT pDevObj, PIRP pIrp)
    
79. {
    
80.         NTSTATUS status = STATUS_SUCCESS;
    
81.         PIO_STACK_LOCATION stack = IoGetCurrentIrpStackLocation(pIrp);
    
82.         ULONG cbin = stack->Parameters.DeviceIoControl.InputBufferLength;
    
83.         ULONG cbout = stack->Parameters.DeviceIoControl.OutputBufferLength;
    
84.         ULONG code = stack->Parameters.DeviceIoControl.IoControlCode;
    
85.         ULONG info = 0;
    
86.         switch(code)
    
87.         {
    
88.         case IOCTL_ENABLE_DEVICEMAPPING:
    
89.                 {
    
90.                         DEVICE_MAPPING tmp;
    
91.                         PFILE_OBJECT fo = NULL;
    
92.                         PDRIVER_DISPATCH* ppdd = NULL;
    
93.                         PFAST_IO_QUERY_STANDARD_INFO* ppfiqsi = NULL;
    
94.                         DEVICEMAPPING_IOCTL_DATA* InputBuffer = (DEVICEMAPPING_IOCTL_DATA*)pIrp->AssociatedIrp.SystemBuffer;
    
95.                         RtlZeroMemory(&tmp, sizeof(tmp));
    
96.                         tmp.hDevice = InputBuffer->hDevice;
    
97.                         status = ObReferenceObjectByHandle(InputBuffer->hDevice, NULL, *IoFileObjectType, KernelMode, (PVOID*)&fo, NULL);
    
98.                         if(NT_SUCCESS(status))
    
99.                         {
    
100.                                 tmp.pFileObject = fo;
     
101.                                 PDEVICE_OBJECT  relatedevice = IoGetRelatedDeviceObject(fo);
     
102.                                 PDRIVER_OBJECT relatedriver = NULL;
     
103.                                 PFAST_IO_DISPATCH relatedfastdispatch = NULL;
     
104.                                 tmp.dwMaximumSize.QuadPart = InputBuffer->size;
     
105.                                 if(relatedriver = relatedevice->DriverObject)
     
106.                                 {
     
107.                                         OriginQuery = relatedriver->MajorFunction[IRP_MJ_QUERY_INFORMATION];
     
108.                                         ppdd = &relatedriver->MajorFunction[IRP_MJ_QUERY_INFORMATION];
     
109.                                         relatedfastdispatch = relatedriver->FastIoDispatch;
     
110.                                         if(relatedfastdispatch && relatedfastdispatch->FastIoQueryStandardInfo)
     
111.                                         {
     
112.                                                 OriginFastDispatch = relatedfastdispatch->FastIoQueryStandardInfo;
     
113.                                                 ppfiqsi = &relatedfastdispatch->FastIoQueryStandardInfo;
     
114.                                         }
     
115.                                 }
     
116.                                 ObDereferenceObject(fo);
     
117.                         }
     
118.                         ExAcquireFastMutex(&gLock);
     
119.                         if(gMappingData)
     
120.                         {
     
121.                                 ExFreePool((PVOID)gMappingData);
     
122.                                 gMappingData = NULL;
     
123.                         }
     
124.                         gMappingData = (DEVICE_MAPPING*)ExAllocatePool(NonPagedPool,sizeof(DEVICE_MAPPING));
     
125.                         if(gMappingData && ppdd)
     
126.                         {
     
127.                                 RtlCopyMemory(gMappingData, &tmp, sizeof(DEVICE_MAPPING));
     
128.                                 *ppdd = DeviceMappingQueryInformation;
     
129.                                 if(ppfiqsi)
     
130.                                 {
     
131.                                         *ppfiqsi = DeviceMappingFastIo;
     
132.                                 }
     
133.                         }
     
134.                         ExReleaseFastMutex(&gLock);
     
135.                         break;
     
136.                 }
     
137.         case IOCTL_DISABLE_DEVICEMAPPING:
     
138.                 {
     
139.                         ExAcquireFastMutex(&gLock);
     
140.                         if(gMappingData)
     
141.                         {
     
142.                                 PFILE_OBJECT fo = NULL;
     
143.                                 PDEVICE_OBJECT relatedevice = NULL;
     
144.                                 PDRIVER_OBJECT relatedriver = NULL;
     
145.                                 status = ObReferenceObjectByHandle(gMappingData->hDevice, NULL, *IoFileObjectType, KernelMode, (PVOID*)&fo, NULL);
     
146.                                 if(NT_SUCCESS(status))
     
147.                                 {
     
148.                                         relatedevice = IoGetRelatedDeviceObject(fo);
     
149.                                         if(relatedevice)
     
150.                                                 relatedriver = relatedevice->DriverObject;
     
151.                                         if(relatedriver && relatedriver->MajorFunction[IRP_MJ_QUERY_INFORMATION] == DeviceMappingQueryInformation)
     
152.                                         {
     
153.                                                 relatedriver->MajorFunction[IRP_MJ_QUERY_INFORMATION] = OriginQuery;
     
154.                                         }
     
155.                                         if(relatedriver && relatedriver->FastIoDispatch && relatedriver->FastIoDispatch->FastIoQueryStandardInfo == DeviceMappingFastIo)
     
156.                                         {
     
157.                                                 relatedriver->FastIoDispatch->FastIoQueryStandardInfo = OriginFastDispatch;
     
158.                                         }
     
159.                                         ObDereferenceObject(fo);
     
160.                                 }
     
161.                                 ExFreePool((PVOID)gMappingData);
     
162.                                 gMappingData = NULL;
     
163.                         }
     
164.                         ExReleaseFastMutex(&gLock);
     
165.                         break;
     
166.                 }
     
167.         default:
     
168.                 status = STATUS_INVALID_VARIANT;
     
169.         }
     
170.         pIrp->IoStatus.Status = status;
     
171.         pIrp->IoStatus.Information = info;
     
172.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
     
173.         return status;
     
174. }
     
175. 
     
176. NTSTATUS DeviceMappingDriverDefaultDispatch(PDEVICE_OBJECT pDevObj, PIRP pIrp)
     
177. {
     
178.         pIrp->IoStatus.Status = STATUS_SUCCESS;
     
179.         pIrp->IoStatus.Information = 0;
     
180.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
     
181.         return STATUS_SUCCESS;
     
182. }
     
183. 
     
184. 
     
185. #pragma code_seg("INIT")
     
186. extern "C" NTSTATUS DriverEntry(PDRIVER_OBJECT pDriverObj, PUNICODE_STRING pRegistryString)
     
187. {
     
188.         UNICODE_STRING DevName;
     
189.         PDEVICE_OBJECT pDevObj = NULL;
     
190.         NTSTATUS status = STATUS_SUCCESS;
     
191.         pDriverObj->DriverUnload = DeviceMappingDriverUnload;
     
192.         pDriverObj->MajorFunction[IRP_MJ_CREATE] = DeviceMappingDriverDefaultDispatch;
     
193.         pDriverObj->MajorFunction[IRP_MJ_CLOSE] = DeviceMappingDriverDefaultDispatch;
     
194.         pDriverObj->MajorFunction[IRP_MJ_READ] = DeviceMappingDriverDefaultDispatch;
     
195.         pDriverObj->MajorFunction[IRP_MJ_WRITE] = DeviceMappingDriverDefaultDispatch;
     
196.         pDriverObj->MajorFunction[IRP_MJ_DEVICE_CONTROL] = DeviceMappingIOControl;
     
197.         RtlInitUnicodeString(&DevName, L"\\Device\\DeviceMapping");
     
198.         ExInitializeFastMutex(&gLock);
     
199.         status = IoCreateDevice(pDriverObj, 0, &DevName, FILE_DEVICE_UNKNOWN, 0, TRUE, &pDevObj);
     
200.         if(NT_SUCCESS(status))
     
201.         {
     
202.                 pDevObj->Flags |= DO_BUFFERED_IO;
     
203.                 UNICODE_STRING SymLinkName;
     
204.                 RtlInitUnicodeString(&SymLinkName, L"\\??\\DeviceMapping");
     
205.                 IoCreateSymbolicLink(&SymLinkName, &DevName);
     
206.         }
     
207.         return status;
     
208. }
     

复制代码

1. 
   
2. #define MAXNAME 256
   
3. 
   
4. #define IOCTL_ENABLE_DEVICEMAPPING CTL_CODE(FILE_DEVICE_UNKNOWN, 0x800, METHOD_BUFFERED, FILE_ANY_ACCESS)
   
5. #define IOCTL_DISABLE_DEVICEMAPPING CTL_CODE(FILE_DEVICE_UNKNOWN, 0x801, METHOD_BUFFERED, FILE_ANY_ACCESS)
   
6. 
   
7. typedef struct _DEVICEMAPPING_IOCTL_DATA//和驱动通信所用data
   
8. {
   
9.         HANDLE hDevice;
   
10.         ULONG size;
    
11. }DEVICEMAPPING_IOCTL_DATA, *PDEVICEMAPPING_IOCTL_DATA;
    

复制代码

应用层：

1. 
   
2. #include <windows.h>
   
3. #include "DeviceMappingCommon.h"
   
4. #include <time.h>
   
5. #include <iostream>
   
6. using namespace std;
   
7. #define BUFSIZE 0x1000
   
8. #define WRITECOUNT 0x10000
   
9. 
   
10. void MappingRead(HANDLE hc)
    
11. {
    
12.         HANDLE hObj = CreateFileA("c:\\1.dat", GENERIC_ALL, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
    
13.         if(hObj == INVALID_HANDLE_VALUE)
    
14.         {
    
15.                 cout<<"Open 1.dat failed"<<endl;
    
16.                 return;
    
17.         }
    
18.         HANDLE hMap = CreateFileMapping(hc, NULL, PAGE_READONLY, 0, BUFSIZE * WRITECOUNT, NULL);
    
19.         if(hMap)
    
20.         {
    
21.                 PBYTE pbDev = (PBYTE)MapViewOfFile(hMap, FILE_MAP_COPY, 0, 0, 0);
    
22.                 DWORD cbwrite;
    
23.                 if(pbDev)
    
24.                 {
    
25.                         for(int i=0;i < WRITECOUNT; i++)
    
26.                         {
    
27.                                 WriteFile(hObj, pbDev + i * BUFSIZE, BUFSIZE, &cbwrite, NULL);
    
28.                         }
    
29.                         UnmapViewOfFile(pbDev);
    
30.                 }
    
31.                 CloseHandle(hMap);
    
32.         }
    
33.         else
    
34.                 cout<<"error:"<<GetLastError();
    
35.         CloseHandle(hObj);
    
36. }
    
37. 
    
38. void BufferedRead(HANDLE hc)
    
39. {
    
40.         BYTE buf[BUFSIZE];
    
41.         int count = 0;
    
42.         DWORD cbwrite;
    
43.         HANDLE hObj = CreateFileA("c:\\2.dat", GENERIC_ALL, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
    
44.         if(hObj == INVALID_HANDLE_VALUE)
    
45.         {
    
46.                 cout<<"Open 2.dat failed"<<endl;
    
47.                 return;
    
48.         }
    
49.         while(ReadFile(hc, buf, BUFSIZE, &cbwrite, NULL) && cbwrite && count++ < WRITECOUNT)
    
50.         {
    
51.                 WriteFile(hObj, buf, BUFSIZE, &cbwrite, NULL);
    
52.         }
    
53.         CloseHandle(hObj);
    
54. }
    
55. 
    
56. void main()
    
57. {
    
58.         HANDLE hFile;
    
59. 
    
60. 
    
61.         hFile = CreateFileA("\\\\.\\c:", GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE,
    
62.                 NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    
63.         if(hFile == INVALID_HANDLE_VALUE)
    
64.         {
    
65.                 cout<<"Open c: failed"<<endl;
    
66.                 return;
    
67.         }
    
68.        
    
69.         time_t begin,end;
    
70.         _asm{int 3};
    
71.         HANDLE hDeviceMapping = CreateFileA("\\\\.\\DeviceMapping",0, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    
72.         int i=GetLastError();
    
73.         if(hDeviceMapping != INVALID_HANDLE_VALUE)
    
74.         {
    
75.                 DEVICEMAPPING_IOCTL_DATA did = {hFile, BUFSIZE * WRITECOUNT};
    
76.                 DWORD retlen;
    
77.                 DeviceIoControl(hDeviceMapping, IOCTL_ENABLE_DEVICEMAPPING, &did, sizeof(did), &did, 0, &retlen, NULL);
    
78.                 begin = time(NULL);
    
79.                 MappingRead(hFile);
    
80.                 end = time(NULL);
    
81.                 DeviceIoControl(hDeviceMapping, IOCTL_DISABLE_DEVICEMAPPING, &did, sizeof(did), &did, 0, &retlen, NULL);
    
82.                 CloseHandle(hDeviceMapping);
    
83.                 cout << "Mapping time:" <<end - begin << endl;
    
84.         }
    
85.         else
    
86.                 cout<< "Error info:"<<GetLastError()<<endl;
    
87. 
    
88.         begin = time(NULL);
    
89.         BufferedRead(hFile);
    
90.         end=time(NULL);
    
91.         cout << "Buffer time:" << end - begin << endl;
    
92. 
    
93.         CloseHandle(hFile);
    
94. }
    
95. 
    

复制代码

分析过程：

1. 
   
2. //#include <ntddk.h>
   
3. #include <ntifs.h>
   
4. 
   
5. typedef struct _HANDLE_TABLE
   
6. {
   
7.         ULONG_PTR TableCode;
   
8.         PEPROCESS QuotaProcess;
   
9.         HANDLE UniqueProcessId;
   
10. #define HANDLE_TABLE_LOCKS 4
    
11.         EX_PUSH_LOCK HandleTableLock[HANDLE_TABLE_LOCKS];
    
12.         LIST_ENTRY HandleTableList;
    
13.         EX_PUSH_LOCK HandleContentionEvent;
    
14.         PVOID DebugInfo;
    
15.         LONG ExtraInfoPages;
    
16.         ULONG FirstFree;
    
17.         ULONG LastFree;
    
18.         ULONG NextHandleNeedingPool;
    
19.         LONG HandleCount;
    
20.         union
    
21.         {
    
22.                 ULONG Flags;
    
23.                 BOOLEAN StrictFIFO : 1;
    
24.         };
    
25. } HANDLE_TABLE, *PHANDLE_TABLE;
    
26. 
    
27. typedef struct _HANDLE_TABLE_ENTRY
    
28. {
    
29.         union
    
30.         {
    
31.                 PVOID Object;
    
32.                 ULONG ObAttributes;
    
33.                 PACCESS_MASK AuditMask;
    
34.                 ULONG_PTR Value;
    
35.         };
    
36.         union
    
37.         {
    
38.                 union
    
39.                 {
    
40.                         ACCESS_MASK GrantedAccess;
    
41.                         struct
    
42.                         {
    
43.                                 USHORT GrantedAccessIndex;
    
44.                                 USHORT CreatorBackTraceIndex;
    
45.                         };
    
46.                 };
    
47.                 LONG NextFreeTableEntry;
    
48.         };
    
49. } HANDLE_TABLE_ENTRY, *PHANDLE_TABLE_ENTRY;
    
50. 
    
51. extern "C"
    
52. {
    
53.         typedef BOOLEAN (*EX_ENUMERATE_HANDLE_ROUTINE)(PHANDLE_TABLE_ENTRY,HANDLE,PVOID );
    
54.         BOOLEAN __stdcall ObFindHandleForObject(PEPROCESS,PVOID,PVOID,PVOID,PHANDLE);
    
55.         PHANDLE_TABLE __stdcall ObReferenceProcessHandleTable (PEPROCESS);
    
56.         BOOLEAN __stdcall ExEnumHandleTable (PHANDLE_TABLE, EX_ENUMERATE_HANDLE_ROUTINE ,PVOID,PHANDLE);
    
57. }
    
58. 
    
59. VOID Unload(PDRIVER_OBJECT pDriverObject)
    
60. {
    
61. 
    
62. }
    
63. PDRIVER_DISPATCH defaultdispatch=NULL;
    
64. 
    
65. NTSTATUS QuerySize (PDEVICE_OBJECT pDeviceObject, PIRP pIrp)
    
66. {
    
67.         PAGED_CODE();
    
68.         NTSTATUS status = STATUS_SUCCESS;
    
69.         PIO_STACK_LOCATION IrpSp = IoGetCurrentIrpStackLocation(pIrp);
    
70.         KdPrint(("MajorFunction=%d MinorFunction=%d\n",IrpSp->MajorFunction,IrpSp->MinorFunction));
    
71.        
    
72.         if(IrpSp->Parameters.QueryFile.FileInformationClass == FileStandardInformation)
    
73.         {
    
74.                 IO_STATUS_BLOCK isb;
    
75.                 FILE_FS_FULL_SIZE_INFORMATION ffsi;
    
76.                 HANDLE handle = NULL;
    
77.                 RtlZeroMemory(&ffsi,sizeof(ffsi));
    
78.                 if(!IrpSp->FileObject){}
    
79.                 else
    
80.                 {
    
81.                         if(ObFindHandleForObject(IoGetCurrentProcess(),IrpSp->FileObject,NULL,NULL,&handle))
    
82.                         {
    
83.                                 status = NtQueryVolumeInformationFile(handle,&isb,&ffsi,sizeof(ffsi),FileFsFullSizeInformation);
    
84.                                 if(NT_SUCCESS(status))
    
85.                                 {
    
86. //                                         KdPrint(("ActualAvailableAllocationUnits=%L\n",&ffsi.ActualAvailableAllocationUnits));
    
87. //                                         KdPrint(("BytesPerSector=%d\n",ffsi.BytesPerSector));
    
88. //                                         KdPrint(("TotalAllocationUnits=%L\n",&ffsi.TotalAllocationUnits));
    
89. //                                         KdPrint(("CallerAvailableAllocationUnits=%L\n",&ffsi.CallerAvailableAllocationUnits));
    
90. //                                         KdPrint(("SectorsPerAllocationUnit=%d\n",ffsi.SectorsPerAllocationUnit));
    
91.                                 }
    
92.                         }
    
93.                         PFILE_STANDARD_INFORMATION FileInformation = (PFILE_STANDARD_INFORMATION)pIrp->AssociatedIrp.SystemBuffer;
    
94.                         FileInformation->EndOfFile.QuadPart = ffsi.TotalAllocationUnits.QuadPart;
    
95.                         FileInformation->EndOfFile.QuadPart *= ffsi.SectorsPerAllocationUnit * ffsi.BytesPerSector;
    
96.                         return STATUS_SUCCESS;
    
97.                 }
    
98.         }
    
99.         return defaultdispatch(pDeviceObject,pIrp);
    
100. }
     
101. 
     
102. typedef struct _DEVICE_EXTENSION
     
103. {
     
104. 
     
105. }DEVICE_EXTENSION,*PDEVICE_EXTENSION;
     
106. 
     
107. extern "C" NTSTATUS DriverEntry(PDRIVER_OBJECT pDriverObj, PUNICODE_STRING pRegistryString)
     
108. {
     
109.         NTSTATUS status = STATUS_SUCCESS;
     
110.         HANDLE hfile;
     
111.         pDriverObj->DriverUnload=Unload;
     
112. 
     
113.         OBJECT_ATTRIBUTES oa;
     
114.         UNICODE_STRING drivename;
     
115.         IO_STATUS_BLOCK isb;
     
116.         PFILE_OBJECT fo;
     
117.         LARGE_INTEGER li={0};
     
118.         RtlInitUnicodeString(&drivename,L"\\??\\c:");
     
119.         InitializeObjectAttributes(&oa,&drivename,OBJ_CASE_INSENSITIVE,NULL,NULL);
     
120.         status = ZwOpenFile(&hfile,FILE_READ_DATA|FILE_READ_ATTRIBUTES|SYNCHRONIZE,&oa,&isb,
     
121.                 FILE_SHARE_READ|FILE_SHARE_WRITE,FILE_OPEN_FOR_FREE_SPACE_QUERY|FILE_SYNCHRONOUS_IO_NONALERT);
     
122.         if(NT_SUCCESS(status))
     
123.         {
     
124.                 KdPrint(("\n\nDriverEntry Entered\nHandle=%08x\n",hfile));
     
125.                 status = ObReferenceObjectByHandle(hfile,0,*IoFileObjectType,KernelMode,(PVOID*)&fo,NULL);
     
126.                 KdPrint(("fo=%p\n",fo));
     
127.                 FILE_FS_FULL_SIZE_INFORMATION ffsi;
     
128.                 status = NtQueryVolumeInformationFile(hfile,&isb,&ffsi,sizeof(ffsi),FileFsFullSizeInformation);
     
129.                 status = FsRtlGetFileSize(fo,&li);
     
130.                 PDRIVER_OBJECT passocdrive = IoGetRelatedDeviceObject(fo)->DriverObject;
     
131.                 defaultdispatch = passocdrive->MajorFunction[IRP_MJ_QUERY_INFORMATION];
     
132.                 passocdrive->MajorFunction[IRP_MJ_QUERY_INFORMATION]=QuerySize;
     
133. 
     
134.         //        PDEVICE_OBJECT DeviceObject = IoGetRelatedDeviceObject(fo);
     
135.                 status = FsRtlGetFileSize(fo,&li);
     
136.                 HANDLE sechandle;
     
137.                 LARGE_INTEGER li={0};
     
138.                 li.LowPart=0x1000;
     
139.                 RtlInitUnicodeString(&drivename,L"\\??\\map1");
     
140.                 status = ZwCreateSection(&sechandle,SECTION_MAP_READ,&oa,&li,PAGE_READONLY,SEC_COMMIT,hfile);
     
141.                 PVOID addr=NULL;
     
142.                 SIZE_T size=0;
     
143.                 if(NT_SUCCESS(status))
     
144.                 {
     
145.                         status = ZwMapViewOfSection(sechandle,ZwCurrentProcess(),&addr,0,0x1000,NULL,&size,ViewShare,
     
146.                                 MEM_TOP_DOWN,PAGE_READONLY);
     
147.                         ULONG data=*(ULONG*)addr;
     
148.                         data=0;
     
149.                 }
     
150. 
     
151.                 status = 0;
     
152.         }
     
153.         return STATUS_SUCCESS;
     
154. }
     
155. 
     

复制代码

测试结果：
分别用缓冲区方式和内存映射方式，分别读取c盘前256MB，分别写入文件中：
Mapping time:23
Buffer time:26
可见还是有一些效果的~~~。

文件内容：
Offset      0  1  2  3  4  5  6  7   8  9  A  B  C  D  E  F

00000000   EB 52 90 4E 54 46 53 20  20 20 20 00 02 08 00 00   隦 NTFS         
00000010   00 00 00 00 00 F8 00 00  3F 00 FF 00 3F 00 00 00        ? ?  ?   
00000020   00 00 00 00 80 00 80 00  B1 8C 7F 02 00 00 00 00       € € 睂      
00000030   00 00 0C 00 00 00 00 00  CB F8 27 00 00 00 00 00           锁'     
00000040   F6 00 00 00 01 00 00 00  CC 44 E9 BC 90 E9 BC 3C   ?      藾榧 榧<
00000050   00 00 00 00 FA 33 C0 8E  D0 BC 00 7C FB B8 C0 07       ?缼屑 |?
00000060   8E D8 E8 16 00 B8 00 0D  8E C0 33 DB C6 06 0E 00   庁? ? 幚3燮   
00000070   10 E8 53 00 68 00 0D 68  6A 02 CB 8A 16 24 00 B4    鑃 h  hj 藠 $ ?
00000080   08 CD 13 73 05 B9 FF FF  8A F1 66 0F B6 C6 40 66    ?s ?婑f 镀@f
00000090   0F B6 D1 80 E2 3F F7 E2  86 CD C0 ED 06 41 66 0F    堆€?麾喭理 Af
000000A0   B7 C9 66 F7 E1 66 A3 20  00 C3 B4 41 BB AA 55 8A   飞f麽f? 么A华U?
000000B0   16 24 00 CD 13 72 0F 81  FB 55 AA 75 09 F6 C1 01    $ ?r  鸘猽 隽
000000C0   74 04 FE 06 14 00 C3 66  60 1E 06 66 A1 10 00 66   t ?  胒`  f? f
000000D0   03 06 1C 00 66 3B 06 20  00 0F 82 3A 00 1E 66 6A       f;    ?  fj
000000E0   00 66 50 06 53 66 68 10  00 01 00 80 3E 14 00 00    fP Sfh    €>   
000000F0   0F 85 0C 00 E8 B3 FF 80  3E 14 00 00 0F 84 61 00    ? 璩€>    刟
00000100   B4 42 8A 16 24 00 16 1F  8B F4 CD 13 66 58 5B 07   碆?$   嬼?fX[
00000110   66 58 66 58 1F EB 2D 66  33 D2 66 0F B7 0E 18 00   fXfX ?f3襢 ?  
00000120   66 F7 F1 FE C2 8A CA 66  8B D0 66 C1 EA 10 F7 36   f黢娛f嬓f陵 ?
00000130   1A 00 86 D6 8A 16 24 00  8A E8 C0 E4 06 0A CC B8     喼?$ 婅冷  谈
00000140   01 02 CD 13 0F 82 19 00  8C C0 05 20 00 8E C0 66     ? ? 尷   幚f
00000150   FF 06 10 00 FF 0E 0E 00  0F 85 6F FF 07 1F 66 61          卭  fa
00000160   C3 A0 F8 01 E8 09 00 A0  FB 01 E8 03 00 FB EB FE   脿?? 狖 ? ?
00000170   B4 01 8B F0 AC 3C 00 74  09 B4 0E BB 07 00 CD 10   ?嬸? t ?? ?
00000180   EB F2 C3 0D 0A 41 20 64  69 73 6B 20 72 65 61 64   腧? A disk read
00000190   20 65 72 72 6F 72 20 6F  63 63 75 72 72 65 64 00    error occurred
000001A0   0D 0A 4E 54 4C 44 52 20  69 73 20 6D 69 73 73 69     NTLDR is missi
000001B0   6E 67 00 0D 0A 4E 54 4C  44 52 20 69 73 20 63 6F   ng   NTLDR is co
000001C0   6D 70 72 65 73 73 65 64  00 0D 0A 50 72 65 73 73   mpressed   Press
000001D0   20 43 74 72 6C 2B 41 6C  74 2B 44 65 6C 20 74 6F    Ctrl+Alt+Del to
000001E0   20 72 65 73 74 61 72 74  0D 0A 00 00 00 00 00 00    restart        
000001F0   00 00 00 00 00 00 00 00  83 A0 B3 C9 00 00 55 AA           儬成  U?
。。。。。。。。。。。。。。。。。。。。。

0x01810

前排学习~

绿色的肥兔

学习了   

FFFFFFFE

666666666666

watermelon

我天，厉害orz