码迷，mamicode.com
首页 > 其他好文 > 详细

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

时间：2016-05-24 18:48:58      阅读：157      评论：0      收藏：0      [点我收藏+]

标签：

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

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

标签：

原文地址：http://www.cnblogs.com/kuangke/p/5524443.html
踩
(0)
赞
(0)
   
举报
评论 一句话评论（0）
登录后才能评论！
分享档案
更多>
2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)
周排行
mamicode.com排行更多图片
更多
友情链接
兰亭集智   国之画   百度统计   站长统计   阿里云   chrome插件   新版天听网
关于我们 - 联系我们 - 留言反馈
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码！