[NTOS:MM]
[reactos.git] / rostests / kmtests / ntos_mm / MmSection.c
1 /*
2 * PROJECT: ReactOS kernel-mode tests
3 * LICENSE: GPLv2+ - See COPYING in the top level directory
4 * PURPOSE: Kernel-Mode Test Suite Section Object test
5 * PROGRAMMER: Thomas Faber <thomas.faber@reactos.org>
6 */
7
8 #include <kmt_test.h>
9
10 #define CheckObject(Handle, Pointers, Handles) do \
11 { \
12 PUBLIC_OBJECT_BASIC_INFORMATION ObjectInfo; \
13 Status = ZwQueryObject(Handle, ObjectBasicInformation, \
14 &ObjectInfo, sizeof ObjectInfo, NULL); \
15 ok_eq_hex(Status, STATUS_SUCCESS); \
16 ok_eq_ulong(ObjectInfo.PointerCount, Pointers); \
17 ok_eq_ulong(ObjectInfo.HandleCount, Handles); \
18 } while (0)
19
20 #define CheckSection(SectionObject, SectionFlag) do \
21 { \
22 SECTION_BASIC_INFORMATION Sbi; \
23 HANDLE SectionHandle = NULL; \
24 NTSTATUS Status; \
25 if (skip(SectionObject != NULL && \
26 SectionObject != (PVOID)0x5555555555555555ULL, \
27 "No section object\n")) \
28 break; \
29 Status = ObOpenObjectByPointer(SectionObject, OBJ_KERNEL_HANDLE, \
30 NULL, 0, MmSectionObjectType, \
31 KernelMode, &SectionHandle); \
32 ok_eq_hex(Status, STATUS_SUCCESS); \
33 ok(SectionHandle != NULL, "Section handle null\n"); \
34 if (!skip(NT_SUCCESS(Status) && SectionHandle, \
35 "No section handle\n")) \
36 { \
37 Status = ZwQuerySection(SectionHandle, SectionBasicInformation, \
38 &Sbi, sizeof Sbi, NULL); \
39 ok_eq_hex(Status, STATUS_SUCCESS); \
40 ok_eq_pointer(Sbi.BaseAddress, NULL); \
41 ok_eq_longlong(Sbi.Size.QuadPart, 1LL); \
42 ok_eq_hex(Sbi.Attributes, SectionFlag | SEC_FILE); \
43 ZwClose(SectionHandle); \
44 } \
45 } while (0)
46
47 #define TestMapView(SectionObject, ExpectAtBase, ExpectM) do \
48 { \
49 NTSTATUS Status; \
50 PVOID BaseAddress = NULL; \
51 SIZE_T ViewSize = 0; \
52 LARGE_INTEGER SectionOffset; \
53 if (skip(SectionObject != NULL && \
54 SectionObject != (PVOID)0x5555555555555555ULL, \
55 "No section object\n")) \
56 break; \
57 \
58 SectionOffset.QuadPart = 0; \
59 Status = MmMapViewOfSection(SectionObject, PsGetCurrentProcess(), \
60 &BaseAddress, 0, 1, &SectionOffset, \
61 &ViewSize, ViewUnmap, 0, PAGE_READONLY); \
62 ok_eq_hex(Status, ExpectAtBase ? STATUS_SUCCESS : STATUS_IMAGE_NOT_AT_BASE);\
63 if (!skip(NT_SUCCESS(Status), "Section not mapped\n")) \
64 { \
65 ok((LONG_PTR)BaseAddress > 0, "BaseAddress = %p\n", BaseAddress); \
66 ok_eq_uint(*(PUCHAR)BaseAddress, ExpectM ? 'M' : 0); \
67 Status = MmUnmapViewOfSection(PsGetCurrentProcess(), BaseAddress); \
68 ok_eq_hex(Status, STATUS_SUCCESS); \
69 } \
70 BaseAddress = NULL; \
71 ViewSize = 0; \
72 Status = MmMapViewOfSection(SectionObject, PsGetCurrentProcess(), \
73 &BaseAddress, 0, 1, &SectionOffset, \
74 &ViewSize, ViewUnmap, 0, \
75 PAGE_READONLY | PAGE_NOCACHE); \
76 ok_eq_hex(Status, ExpectAtBase ? STATUS_SUCCESS : STATUS_IMAGE_NOT_AT_BASE);\
77 if (!skip(NT_SUCCESS(Status), "Section not mapped\n")) \
78 { \
79 ok((LONG_PTR)BaseAddress > 0, "BaseAddress = %p\n", BaseAddress); \
80 ok_eq_uint(*(PUCHAR)BaseAddress, ExpectM ? 'M' : 0); \
81 Status = MmUnmapViewOfSection(PsGetCurrentProcess(), BaseAddress); \
82 ok_eq_hex(Status, STATUS_SUCCESS); \
83 } \
84 } while (0)
85
86 static
87 VOID
88 TestCreateSection(
89 IN HANDLE FileHandle1,
90 IN PFILE_OBJECT FileObject1,
91 IN HANDLE FileHandle2,
92 IN PFILE_OBJECT FileObject2)
93 {
94 NTSTATUS Status = STATUS_SUCCESS;
95 PVOID SectionObject;
96 LARGE_INTEGER MaximumSize;
97 ULONG PointerCount1, PointerCount2;
98
99 KmtStartSeh()
100 Status = MmCreateSection(NULL, 0, NULL, NULL, 0, SEC_RESERVE, NULL, NULL);
101 KmtEndSeh(STATUS_SUCCESS);
102 ok_eq_hex(Status, STATUS_INVALID_PAGE_PROTECTION);
103
104 if (!KmtIsCheckedBuild)
105 {
106 /* PAGE_NOACCESS and missing SEC_RESERVE/SEC_COMMIT/SEC_IMAGE assert */
107 KmtStartSeh()
108 Status = MmCreateSection(NULL, 0, NULL, NULL, PAGE_NOACCESS, SEC_RESERVE, NULL, NULL);
109 KmtEndSeh(STATUS_ACCESS_VIOLATION);
110
111 KmtStartSeh()
112 Status = MmCreateSection(NULL, 0, NULL, NULL, PAGE_NOACCESS, 0, NULL, NULL);
113 KmtEndSeh(STATUS_ACCESS_VIOLATION);
114 }
115
116 SectionObject = KmtInvalidPointer;
117 KmtStartSeh()
118 Status = MmCreateSection(&SectionObject, 0, NULL, NULL, 0, SEC_RESERVE, NULL, NULL);
119 KmtEndSeh(STATUS_SUCCESS);
120 ok_eq_hex(Status, STATUS_INVALID_PAGE_PROTECTION);
121 ok_eq_pointer(SectionObject, KmtInvalidPointer);
122
123 if (SectionObject && SectionObject != KmtInvalidPointer)
124 ObDereferenceObject(SectionObject);
125
126 KmtStartSeh()
127 Status = MmCreateSection(NULL, 0, NULL, NULL, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
128 KmtEndSeh(STATUS_ACCESS_VIOLATION);
129
130 SectionObject = KmtInvalidPointer;
131 KmtStartSeh()
132 Status = MmCreateSection(&SectionObject, 0, NULL, NULL, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
133 KmtEndSeh(STATUS_ACCESS_VIOLATION);
134 ok_eq_pointer(SectionObject, KmtInvalidPointer);
135
136 if (SectionObject && SectionObject != KmtInvalidPointer)
137 ObDereferenceObject(SectionObject);
138
139 SectionObject = KmtInvalidPointer;
140 MaximumSize.QuadPart = 0;
141 KmtStartSeh()
142 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, NULL, NULL);
143 KmtEndSeh(STATUS_SUCCESS);
144 ok_eq_hex(Status, STATUS_INVALID_FILE_FOR_SECTION);
145 ok_eq_longlong(MaximumSize.QuadPart, 0LL);
146 ok_eq_pointer(SectionObject, KmtInvalidPointer);
147
148 if (SectionObject && SectionObject != KmtInvalidPointer)
149 ObDereferenceObject(SectionObject);
150
151 MaximumSize.QuadPart = 0;
152 KmtStartSeh()
153 Status = MmCreateSection(NULL, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
154 KmtEndSeh(STATUS_SUCCESS);
155 ok_eq_hex(Status, STATUS_INVALID_PARAMETER_4);
156 ok_eq_longlong(MaximumSize.QuadPart, 0LL);
157
158 if (SectionObject && SectionObject != KmtInvalidPointer)
159 ObDereferenceObject(SectionObject);
160
161 MaximumSize.QuadPart = 1;
162 KmtStartSeh()
163 Status = MmCreateSection(NULL, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
164 KmtEndSeh(STATUS_ACCESS_VIOLATION);
165 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
166
167 if (SectionObject && SectionObject != KmtInvalidPointer)
168 ObDereferenceObject(SectionObject);
169
170 SectionObject = KmtInvalidPointer;
171 MaximumSize.QuadPart = 0;
172 KmtStartSeh()
173 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
174 KmtEndSeh(STATUS_SUCCESS);
175 ok_eq_hex(Status, STATUS_INVALID_PARAMETER_4);
176 ok_eq_longlong(MaximumSize.QuadPart, 0LL);
177 ok_eq_pointer(SectionObject, KmtInvalidPointer);
178
179 if (SectionObject && SectionObject != KmtInvalidPointer)
180 ObDereferenceObject(SectionObject);
181
182 /* page file section */
183 SectionObject = KmtInvalidPointer;
184 MaximumSize.QuadPart = 1;
185 KmtStartSeh()
186 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, NULL, NULL);
187 KmtEndSeh(STATUS_SUCCESS);
188 ok_eq_hex(Status, STATUS_SUCCESS);
189 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
190 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
191 ok(SectionObject != NULL, "Section object pointer NULL\n");
192
193 if (SectionObject && SectionObject != KmtInvalidPointer)
194 ObDereferenceObject(SectionObject);
195
196 if (!skip(FileHandle1 != NULL && FileObject1 != NULL &&
197 FileHandle2 != NULL && FileObject2 != NULL, "No file handle or object\n"))
198 {
199 PointerCount1 = 3;
200 PointerCount2 = 3;
201 /* image section */
202 CheckObject(FileHandle2, PointerCount2, 1L);
203 SectionObject = KmtInvalidPointer;
204 MaximumSize.QuadPart = 1;
205 KmtStartSeh()
206 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, FileHandle2, NULL);
207 KmtEndSeh(STATUS_SUCCESS);
208 ok_eq_hex(Status, STATUS_SUCCESS);
209 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
210 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
211 ok(SectionObject != NULL, "Section object pointer NULL\n");
212 CheckObject(FileHandle2, PointerCount2, 1L);
213 CheckSection(SectionObject, SEC_IMAGE);
214 TestMapView(SectionObject, FALSE, TRUE);
215
216 if (SectionObject && SectionObject != KmtInvalidPointer)
217 ObDereferenceObject(SectionObject);
218
219 CheckObject(FileHandle2, PointerCount2, 1L);
220 SectionObject = KmtInvalidPointer;
221 MaximumSize.QuadPart = 1;
222 KmtStartSeh()
223 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, NULL, FileObject2);
224 KmtEndSeh(STATUS_SUCCESS);
225 ok_eq_hex(Status, STATUS_SUCCESS);
226 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
227 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
228 ok(SectionObject != NULL, "Section object pointer NULL\n");
229 ++PointerCount2;
230 CheckObject(FileHandle2, PointerCount2, 1L);
231 CheckSection(SectionObject, 0);
232 TestMapView(SectionObject, TRUE, TRUE);
233
234 if (SectionObject && SectionObject != KmtInvalidPointer)
235 ObDereferenceObject(SectionObject);
236 //--PointerCount2; // ????
237
238 CheckObject(FileHandle2, PointerCount2, 1L);
239 SectionObject = KmtInvalidPointer;
240 MaximumSize.QuadPart = 1;
241 KmtStartSeh()
242 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, FileHandle2, FileObject2);
243 KmtEndSeh(STATUS_SUCCESS);
244 ok_eq_hex(Status, STATUS_SUCCESS);
245 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
246 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
247 ok(SectionObject != NULL, "Section object pointer NULL\n");
248 CheckObject(FileHandle2, PointerCount2, 1L);
249 CheckSection(SectionObject, 0);
250 TestMapView(SectionObject, TRUE, TRUE);
251
252 if (SectionObject && SectionObject != KmtInvalidPointer)
253 ObDereferenceObject(SectionObject);
254
255 /* image section with inappropriate file */
256 CheckObject(FileHandle1, PointerCount1, 1L);
257 SectionObject = KmtInvalidPointer;
258 MaximumSize.QuadPart = 1;
259 KmtStartSeh()
260 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, FileHandle1, NULL);
261 KmtEndSeh(STATUS_SUCCESS);
262 ok_eq_hex(Status, STATUS_INVALID_IMAGE_NOT_MZ);
263 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
264 ok_eq_pointer(SectionObject, KmtInvalidPointer);
265 CheckObject(FileHandle1, PointerCount1, 1L);
266
267 if (SectionObject && SectionObject != KmtInvalidPointer)
268 ObDereferenceObject(SectionObject);
269
270 CheckObject(FileHandle1, PointerCount1, 1L);
271 SectionObject = KmtInvalidPointer;
272 MaximumSize.QuadPart = 1;
273 KmtStartSeh()
274 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, NULL, FileObject1);
275 KmtEndSeh(STATUS_SUCCESS);
276 ok_eq_hex(Status, STATUS_SUCCESS);
277 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
278 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
279 ok(SectionObject != NULL, "Section object pointer NULL\n");
280 ++PointerCount1;
281 CheckObject(FileHandle1, PointerCount1, 1L);
282 CheckSection(SectionObject, 0);
283 TestMapView(SectionObject, TRUE, FALSE);
284
285 if (SectionObject && SectionObject != KmtInvalidPointer)
286 ObDereferenceObject(SectionObject);
287 //--PointerCount1; // ????
288
289 CheckObject(FileHandle1, PointerCount1, 1L);
290 SectionObject = KmtInvalidPointer;
291 MaximumSize.QuadPart = 1;
292 KmtStartSeh()
293 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, FileHandle1, FileObject1);
294 KmtEndSeh(STATUS_SUCCESS);
295 ok_eq_hex(Status, STATUS_SUCCESS);
296 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
297 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
298 ok(SectionObject != NULL, "Section object pointer NULL\n");
299 CheckObject(FileHandle1, PointerCount1, 1L);
300 CheckSection(SectionObject, 0);
301 TestMapView(SectionObject, TRUE, FALSE);
302
303 if (SectionObject && SectionObject != KmtInvalidPointer)
304 ObDereferenceObject(SectionObject);
305
306 /* image section with two different files */
307 CheckObject(FileHandle1, PointerCount1, 1L);
308 SectionObject = KmtInvalidPointer;
309 MaximumSize.QuadPart = 1;
310 KmtStartSeh()
311 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, FileHandle1, FileObject2);
312 KmtEndSeh(STATUS_SUCCESS);
313 ok_eq_hex(Status, STATUS_SUCCESS);
314 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
315 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
316 ok(SectionObject != NULL, "Section object pointer NULL\n");
317 CheckObject(FileHandle1, PointerCount1, 1L);
318 CheckObject(FileHandle2, PointerCount2, 1L);
319 CheckSection(SectionObject, 0);
320 TestMapView(SectionObject, TRUE, TRUE);
321
322 if (SectionObject && SectionObject != KmtInvalidPointer)
323 ObDereferenceObject(SectionObject);
324
325 CheckObject(FileHandle1, PointerCount1, 1L);
326 SectionObject = KmtInvalidPointer;
327 MaximumSize.QuadPart = 1;
328 KmtStartSeh()
329 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_IMAGE, FileHandle2, FileObject1);
330 KmtEndSeh(STATUS_SUCCESS);
331 ok_eq_hex(Status, STATUS_SUCCESS);
332 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
333 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
334 ok(SectionObject != NULL, "Section object pointer NULL\n");
335 CheckObject(FileHandle1, PointerCount1, 1L);
336 CheckObject(FileHandle2, PointerCount2, 1L);
337 CheckSection(SectionObject, 0);
338 TestMapView(SectionObject, TRUE, FALSE);
339
340 if (SectionObject && SectionObject != KmtInvalidPointer)
341 ObDereferenceObject(SectionObject);
342
343 /* data file section */
344 CheckObject(FileHandle1, PointerCount1, 1L);
345 SectionObject = KmtInvalidPointer;
346 MaximumSize.QuadPart = 1;
347 KmtStartSeh()
348 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, FileHandle1, NULL);
349 KmtEndSeh(STATUS_SUCCESS);
350 ok_eq_hex(Status, STATUS_SUCCESS);
351 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
352 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
353 ok(SectionObject != NULL, "Section object pointer NULL\n");
354 CheckObject(FileHandle1, PointerCount1, 1L);
355 CheckSection(SectionObject, 0);
356 TestMapView(SectionObject, TRUE, FALSE);
357
358 if (SectionObject && SectionObject != KmtInvalidPointer)
359 ObDereferenceObject(SectionObject);
360
361 CheckObject(FileHandle1, PointerCount1, 1L);
362 SectionObject = KmtInvalidPointer;
363 MaximumSize.QuadPart = 1;
364 KmtStartSeh()
365 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, NULL, FileObject1);
366 KmtEndSeh(STATUS_SUCCESS);
367 ok_eq_hex(Status, STATUS_SUCCESS);
368 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
369 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
370 ok(SectionObject != NULL, "Section object pointer NULL\n");
371 CheckObject(FileHandle1, PointerCount1, 1L);
372 CheckSection(SectionObject, 0);
373 TestMapView(SectionObject, TRUE, FALSE);
374
375 if (SectionObject && SectionObject != KmtInvalidPointer)
376 ObDereferenceObject(SectionObject);
377
378 CheckObject(FileHandle1, PointerCount1, 1L);
379 SectionObject = KmtInvalidPointer;
380 MaximumSize.QuadPart = 1;
381 KmtStartSeh()
382 Status = MmCreateSection(&SectionObject, 0, NULL, &MaximumSize, PAGE_READONLY, SEC_RESERVE, FileHandle1, FileObject1);
383 KmtEndSeh(STATUS_SUCCESS);
384 ok_eq_hex(Status, STATUS_SUCCESS);
385 ok_eq_longlong(MaximumSize.QuadPart, 1LL);
386 ok(SectionObject != KmtInvalidPointer, "Section object pointer untouched\n");
387 ok(SectionObject != NULL, "Section object pointer NULL\n");
388 CheckObject(FileHandle1, PointerCount1, 1L);
389 CheckSection(SectionObject, 0);
390 TestMapView(SectionObject, TRUE, FALSE);
391
392 if (SectionObject && SectionObject != KmtInvalidPointer)
393 ObDereferenceObject(SectionObject);
394
395 CheckObject(FileHandle1, PointerCount1, 1L);
396 }
397 }
398
399 static
400 VOID
401 TestPhysicalMemorySection(VOID)
402 {
403 NTSTATUS Status;
404 UNICODE_STRING SectionName = RTL_CONSTANT_STRING(L"\\Device\\PhysicalMemory");
405 OBJECT_ATTRIBUTES ObjectAttributes;
406 HANDLE SectionHandle;
407 PVOID SectionObject;
408 PUCHAR MyPage;
409 PHYSICAL_ADDRESS MyPagePhysical;
410 PUCHAR ZeroPageContents;
411 PHYSICAL_ADDRESS ZeroPagePhysical;
412 PVOID Mapping;
413 PUCHAR MappingBytes;
414 SIZE_T ViewSize;
415 SIZE_T EqualBytes;
416
417 MyPage = ExAllocatePoolWithTag(NonPagedPool, PAGE_SIZE, 'MPmK');
418 if (skip(MyPage != NULL, "Out of memory\n"))
419 return;
420 MyPagePhysical = MmGetPhysicalAddress(MyPage);
421 RtlFillMemory(MyPage + 0 * PAGE_SIZE / 4, PAGE_SIZE / 4, 0x23);
422 RtlFillMemory(MyPage + 1 * PAGE_SIZE / 4, PAGE_SIZE / 4, 0x67);
423 RtlFillMemory(MyPage + 2 * PAGE_SIZE / 4, PAGE_SIZE / 4, 0xab);
424 RtlFillMemory(MyPage + 3 * PAGE_SIZE / 4, PAGE_SIZE / 4, 0xef);
425
426 ZeroPageContents = ExAllocatePoolWithTag(PagedPool, PAGE_SIZE, 'ZPmK');
427 if (skip(ZeroPageContents != NULL, "Out of memory\n"))
428 {
429 ExFreePoolWithTag(MyPage, 'MPmK');
430 return;
431 }
432 ZeroPagePhysical.QuadPart = 0;
433
434 Mapping = MmMapIoSpace(ZeroPagePhysical, PAGE_SIZE, MmCached);
435 if (skip(Mapping != NULL, "Failed to map zero page\n"))
436 {
437 ExFreePoolWithTag(ZeroPageContents, 'ZPmK');
438 ExFreePoolWithTag(MyPage, 'MPmK');
439 return;
440 }
441
442 RtlCopyMemory(ZeroPageContents, Mapping, PAGE_SIZE);
443 MmUnmapIoSpace(Mapping, PAGE_SIZE);
444
445 InitializeObjectAttributes(&ObjectAttributes,
446 &SectionName,
447 OBJ_KERNEL_HANDLE,
448 NULL,
449 NULL);
450 Status = ZwOpenSection(&SectionHandle, SECTION_ALL_ACCESS, &ObjectAttributes);
451 ok_eq_hex(Status, STATUS_SUCCESS);
452 if (!skip(NT_SUCCESS(Status), "No section\n"))
453 {
454 /* Map zero page and compare */
455 Mapping = NULL;
456 ViewSize = PAGE_SIZE;
457 Status = ZwMapViewOfSection(SectionHandle,
458 ZwCurrentProcess(),
459 &Mapping,
460 0,
461 0,
462 &ZeroPagePhysical,
463 &ViewSize,
464 ViewUnmap,
465 0,
466 PAGE_READWRITE);
467 ok_eq_hex(Status, STATUS_SUCCESS);
468 if (!skip(NT_SUCCESS(Status), "No view\n"))
469 {
470 ok((LONG_PTR)Mapping > 0, "Mapping = %p\n", Mapping);
471 EqualBytes = RtlCompareMemory(Mapping,
472 ZeroPageContents,
473 PAGE_SIZE);
474 ok_eq_size(EqualBytes, PAGE_SIZE);
475 Status = ZwUnmapViewOfSection(ZwCurrentProcess(), Mapping);
476 ok_eq_hex(Status, STATUS_SUCCESS);
477 }
478
479 /* Map the zero page non-cached */
480 Mapping = NULL;
481 ViewSize = PAGE_SIZE;
482 Status = ZwMapViewOfSection(SectionHandle,
483 ZwCurrentProcess(),
484 &Mapping,
485 0,
486 0,
487 &ZeroPagePhysical,
488 &ViewSize,
489 ViewUnmap,
490 0,
491 PAGE_READWRITE | PAGE_NOCACHE);
492 ok_eq_hex(Status, STATUS_SUCCESS);
493 if (!skip(NT_SUCCESS(Status), "No view\n"))
494 {
495 ok((LONG_PTR)Mapping > 0, "Mapping = %p\n", Mapping);
496 EqualBytes = RtlCompareMemory(Mapping,
497 ZeroPageContents,
498 PAGE_SIZE);
499 ok_eq_size(EqualBytes, PAGE_SIZE);
500 Status = ZwUnmapViewOfSection(ZwCurrentProcess(), Mapping);
501 ok_eq_hex(Status, STATUS_SUCCESS);
502 }
503
504 /* Map our NP page, compare, and check that modifications are reflected */
505 Mapping = NULL;
506 ViewSize = PAGE_SIZE;
507 Status = ZwMapViewOfSection(SectionHandle,
508 ZwCurrentProcess(),
509 &Mapping,
510 0,
511 0,
512 &MyPagePhysical,
513 &ViewSize,
514 ViewUnmap,
515 0,
516 PAGE_READWRITE);
517 ok_eq_hex(Status, STATUS_SUCCESS);
518 if (!skip(NT_SUCCESS(Status), "No view\n"))
519 {
520 ok((LONG_PTR)Mapping > 0, "Mapping = %p\n", Mapping);
521 EqualBytes = RtlCompareMemory(Mapping,
522 MyPage,
523 PAGE_SIZE);
524 ok_eq_size(EqualBytes, PAGE_SIZE);
525
526 MappingBytes = Mapping;
527 ok(MappingBytes[5] == 0x23, "Mapping[5] = 0x%x\n", MappingBytes[5]);
528 ok(MyPage[5] == 0x23, "MyPage[5] = 0x%x\n", MyPage[5]);
529
530 MyPage[5] = 0x44;
531 ok(MappingBytes[5] == 0x44, "Mapping[5] = 0x%x\n", MappingBytes[5]);
532 ok(MyPage[5] == 0x44, "MyPage[5] = 0x%x\n", MyPage[5]);
533
534 MappingBytes[5] = 0x88;
535 ok(MappingBytes[5] == 0x88, "Mapping[5] = 0x%x\n", MappingBytes[5]);
536 ok(MyPage[5] == 0x88, "MyPage[5] = 0x%x\n", MyPage[5]);
537
538 Status = ZwUnmapViewOfSection(ZwCurrentProcess(), Mapping);
539 ok_eq_hex(Status, STATUS_SUCCESS);
540 }
541
542 Status = ZwClose(SectionHandle);
543 ok_eq_hex(Status, STATUS_SUCCESS);
544 }
545
546 /* Try flag 0x80000000, which ROS calls SEC_PHYSICALMEMORY */
547 InitializeObjectAttributes(&ObjectAttributes,
548 NULL,
549 OBJ_KERNEL_HANDLE,
550 NULL,
551 NULL);
552 Status = ZwCreateSection(&SectionHandle,
553 SECTION_ALL_ACCESS,
554 &ObjectAttributes,
555 NULL,
556 PAGE_READWRITE,
557 0x80000000,
558 NULL);
559 ok_eq_hex(Status, STATUS_INVALID_PARAMETER_6);
560 if (NT_SUCCESS(Status))
561 ZwClose(SectionHandle);
562
563 /* Assertion failure: AllocationAttributes & SEC_IMAGE | SEC_RESERVE | SEC_COMMIT */
564 if (!KmtIsCheckedBuild)
565 {
566 InitializeObjectAttributes(&ObjectAttributes,
567 NULL,
568 OBJ_KERNEL_HANDLE,
569 NULL,
570 NULL);
571 Status = MmCreateSection(&SectionObject,
572 SECTION_ALL_ACCESS,
573 &ObjectAttributes,
574 NULL,
575 PAGE_READWRITE,
576 0x80000000,
577 NULL,
578 NULL);
579 ok_eq_hex(Status, STATUS_INVALID_PARAMETER_6);
580 if (NT_SUCCESS(Status))
581 ObDereferenceObject(SectionObject);
582 }
583
584 InitializeObjectAttributes(&ObjectAttributes,
585 NULL,
586 OBJ_KERNEL_HANDLE,
587 NULL,
588 NULL);
589 Status = MmCreateSection(&SectionObject,
590 SECTION_ALL_ACCESS,
591 &ObjectAttributes,
592 NULL,
593 PAGE_READWRITE,
594 SEC_RESERVE | 0x80000000,
595 NULL,
596 NULL);
597 ok_eq_hex(Status, STATUS_INVALID_PARAMETER_6);
598 if (NT_SUCCESS(Status))
599 ObDereferenceObject(SectionObject);
600
601 ExFreePoolWithTag(ZeroPageContents, 'ZPmK');
602 ExFreePoolWithTag(MyPage, 'MPmK');
603 }
604
605 START_TEST(MmSection)
606 {
607 NTSTATUS Status;
608 HANDLE FileHandle1 = NULL, FileHandle2 = NULL;
609 PFILE_OBJECT FileObject1 = NULL, FileObject2 = NULL;
610 OBJECT_ATTRIBUTES ObjectAttributes;
611 IO_STATUS_BLOCK IoStatusBlock;
612 UNICODE_STRING FileName1 = RTL_CONSTANT_STRING(L"\\SystemRoot\\kmtest-MmSection.txt");
613 UNICODE_STRING FileName2 = RTL_CONSTANT_STRING(L"\\SystemRoot\\system32\\ntdll.dll");
614 LARGE_INTEGER FileOffset;
615 UCHAR FileData = 0;
616
617 ok(ExGetPreviousMode() == UserMode, "Previous mode is kernel mode\n");
618 /* create a one-byte file that we can use */
619 InitializeObjectAttributes(&ObjectAttributes, &FileName1, OBJ_CASE_INSENSITIVE, NULL, NULL);
620 Status = ZwCreateFile(&FileHandle1, GENERIC_WRITE | SYNCHRONIZE, &ObjectAttributes, &IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL, FILE_SHARE_READ, FILE_SUPERSEDE, FILE_NON_DIRECTORY_FILE, NULL, 0);
621 ok_eq_hex(Status, STATUS_SUCCESS);
622 ok_eq_ulongptr(IoStatusBlock.Information, FILE_CREATED);
623 ok(FileHandle1 != NULL, "FileHandle1 is NULL\n");
624 if (FileHandle1)
625 {
626 FileOffset.QuadPart = 0;
627 Status = ZwWriteFile(FileHandle1, NULL, NULL, NULL, &IoStatusBlock, &FileData, sizeof FileData, &FileOffset, NULL);
628 ok(Status == STATUS_SUCCESS || Status == STATUS_PENDING, "Status = 0x%08lx\n", Status);
629 Status = ZwWaitForSingleObject(FileHandle1, FALSE, NULL);
630 ok_eq_hex(Status, STATUS_SUCCESS);
631 ok_eq_ulongptr(IoStatusBlock.Information, 1);
632 Status = ZwClose(FileHandle1);
633 ok_eq_hex(Status, STATUS_SUCCESS);
634 FileHandle1 = NULL;
635 }
636
637 InitializeObjectAttributes(&ObjectAttributes, &FileName1, OBJ_CASE_INSENSITIVE, NULL, NULL);
638 Status = ZwCreateFile(&FileHandle1, GENERIC_ALL, &ObjectAttributes, &IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL, FILE_SHARE_READ, FILE_OPEN, FILE_NON_DIRECTORY_FILE | FILE_DELETE_ON_CLOSE, NULL, 0);
639 ok_eq_hex(Status, STATUS_SUCCESS);
640 ok_eq_ulongptr(IoStatusBlock.Information, FILE_OPENED);
641 ok(FileHandle1 != NULL, "FileHandle1 is NULL\n");
642 CheckObject(FileHandle1, 2L, 1L);
643
644 InitializeObjectAttributes(&ObjectAttributes, &FileName2, OBJ_CASE_INSENSITIVE, NULL, NULL);
645 Status = ZwCreateFile(&FileHandle2, GENERIC_READ, &ObjectAttributes, &IoStatusBlock, NULL, FILE_ATTRIBUTE_NORMAL, FILE_SHARE_READ, FILE_OPEN, FILE_NON_DIRECTORY_FILE, NULL, 0);
646 ok_eq_hex(Status, STATUS_SUCCESS);
647 ok_eq_ulongptr(IoStatusBlock.Information, FILE_OPENED);
648 ok(FileHandle2 != NULL, "FileHandle2 is NULL\n");
649
650 if (!skip(Status == STATUS_SUCCESS && FileHandle1 != NULL, "Failed to open file 1\n"))
651 {
652 Status = ObReferenceObjectByHandle(FileHandle1, FILE_READ_DATA | FILE_WRITE_DATA, *IoFileObjectType, KernelMode, (PVOID *)&FileObject1, NULL);
653 ok_eq_hex(Status, STATUS_SUCCESS);
654 ok(FileObject1 != NULL, "FileObject1 is NULL\n");
655 CheckObject(FileHandle1, 3L, 1L);
656 }
657
658 if (!skip(Status == STATUS_SUCCESS && FileHandle2 != NULL, "Failed to open file 2\n"))
659 {
660 Status = ObReferenceObjectByHandle(FileHandle2, FILE_READ_DATA | FILE_WRITE_DATA, *IoFileObjectType, KernelMode, (PVOID *)&FileObject2, NULL);
661 ok_eq_hex(Status, STATUS_SUCCESS);
662 ok(FileObject2 != NULL, "FileObject2 is NULL\n");
663 }
664
665 trace("FileHandle1=%p, FileObject1=%p\n", FileHandle1, FileObject1);
666 trace("FileHandle2=%p, FileObject2=%p\n", FileHandle2, FileObject2);
667 TestCreateSection(FileHandle1, FileObject1, FileHandle2, FileObject2);
668
669 if (FileObject2)
670 ObDereferenceObject(FileObject2);
671 if (FileObject1)
672 ObDereferenceObject(FileObject1);
673 if (FileHandle2)
674 ZwClose(FileHandle2);
675 if (FileHandle1)
676 ZwClose(FileHandle1);
677
678 TestPhysicalMemorySection();
679 }