标准MDL方法修改Page、NonPage内存的属性

1. typedef struct _REPROTECT_CONTEXT
2. {
3. PMDL   Mdl;
4. PUCHAR LockedVa;
5. } REPROTECT_CONTEXT, * PREPROTECT_CONTEXT;
6. NTSTATUS
7. MmLockVaForWrite(
8. __in PVOID Va,
9. __in ULONG Length,
10. __out PREPROTECT_CONTEXT ReprotectContext
11. )
12. {
13. NTSTATUS Status;
14. Status = STATUS_SUCCESS;
15. ReprotectContext->Mdl      = 0;
16. ReprotectContext->LockedVa = 0;
17. ReprotectContext->Mdl = IoAllocateMdl(
18. Va,
19. Length,
20. FALSE,
21. FALSE,
22. 0
23. );
24. if (!ReprotectContext->Mdl)
25. {
26. return STATUS_INSUFFICIENT_RESOURCES;
27. }
28. //
29. // Retrieve a locked VA mapping.
30. //
31. __try
32. {
33. MmProbeAndLockPages(
34. ReprotectContext->Mdl,
35. KernelMode,
36. IoModifyAccess
37. );
38. }
39. __except (EXCEPTION_EXECUTE_HANDLER)
40. {
41. return GetExceptionCode();
42. }
43. ReprotectContext->LockedVa = (PUCHAR)MmMapLockedPagesSpecifyCache(
44. ReprotectContext->Mdl,
45. KernelMode,
46. MmCached,
47. 0,
48. FALSE,
49. NormalPagePriority
50. );
51. if (!ReprotectContext->LockedVa)
52. {
53. IoFreeMdl(
54. ReprotectContext->Mdl
55. );
56. ReprotectContext->Mdl = 0;
57. return STATUS_ACCESS_VIOLATION;
58. }
59. //
60. // Reprotect.
61. //
62. Status = MmProtectMdlSystemAddress(
63. ReprotectContext->Mdl,
64. PAGE_EXECUTE_READWRITE
65. );
66. if (!NT_SUCCESS(Status))
67. {
68. MmUnmapLockedPages(
69. ReprotectContext->LockedVa,
70. ReprotectContext->Mdl
71. );
72. MmUnlockPages(
73. ReprotectContext->Mdl
74. );
75. IoFreeMdl(
76. ReprotectContext->Mdl
77. );
78. ReprotectContext->LockedVa = 0;
79. ReprotectContext->Mdl      = 0;
80. }
81. return Status;
82. }
83. NTSTATUS
84. MmUnlockVaForWrite(
85. __in PREPROTECT_CONTEXT ReprotectContext
86. )
87. {
88. if (ReprotectContext->LockedVa)
89. {
90. MmUnmapLockedPages(
91. ReprotectContext->LockedVa,
92. ReprotectContext->Mdl
93. );
94. MmUnlockPages(
95. ReprotectContext->Mdl
96. );
97. IoFreeMdl(
98. ReprotectContext->Mdl
99. );
100. ReprotectContext->LockedVa = 0;
101. ReprotectContext->Mdl      = 0;
102. }
103. return STATUS_SUCCESS;
104. }