MmProbeAndLockPages

来源:互联网 发布:人工智能会毁了人类吗 编辑:程序博客网 时间:2024/06/06 03:25

MmProbeAndLockPages 到底锁定什么?

驱动程序有时候需要锁定内存页以使得它们在特定的操作期间驻留在内存中,例如在 DPC 例程中将数据从设备复制到数据缓冲区或者对缓冲区进行 DMA。MmProbeAndLockPages 例程使得一个指定的内存范围驻留(如果尚未驻留),确认内存页允许以指定的访问模式执行指定的操作,并锁定内存中的页以使得它们不能被页交换出去。

MmProbeAndLockPages 通过在页帧号 (page frame number) 数据库(描述物理内存中每个页的状态)中的页帧号(PFN)条目上增加引用计数实现这个功能。只要 PFN 引用计数非零,物理页就不会被重用。

但是,不要混淆物理页及其虚拟地址,这一点非常重要。MmProbeAndLockPages 在内存中锁定物理页,但是如果相关的工作集被修改(或者如果进程释放或取消映射地址范围),那么它们的虚拟地址会变得无效。即使可以保证后备物理页保持驻留,虚拟地址映射也没有保证。驱动程序中这种情况的一个症状是 MmIsAddressValid 将指示虚拟地址页故障,认为内存已经被页交换出去。如果驱动程序试图使用虚拟地址(假设地址范围未被删除),那么它会遇到与试图使用可分页内存时相同的问题:如果驱动程序以低于 DISPATCH_LEVEL 的级别运行,那么系统可以指示页故障;如果驱动程序在以 DISPATCH_LEVEL 或高于 DISPATCH_LEVEL 的级别运行,那么系统将进行错误检查。(关于 Windows 如何管理虚拟内存和物理内存的深入讨论,请参见 Inside Windows 2000(第三版)的第 7 章。)

如果在 DISPATCH_LEVEL 或高于 DISPATCH_LEVEL 的级别上需要虚拟地址,或者在另一个进程上下文中需要虚拟地址,那么驱动程序必须使用 MmGetSystemAddressForMdlSafe 映射 MDL 来获取 MDL 的未经修改的系统虚拟地址。这个虚拟地址从未被插入到工作集中,因此它不能被删除。不管线程或 IRQL 的进程上下文(在其中引用地址)如何,它都能保证正常工作。

您应该做什么?

  • 调用 MmProbeAndLockPages 在内存中锁定 MDL 描述的页。始终应该在 try/except 块中执行此操作,因为如果 MmProbeAndLockPages 失败,将抛出一个异常。

  • 要将锁定的物理页映射到系统空间,请使用 MDL 的指针来调用 MmGetSystemAddressForMdlSafe。(为必须在 Windows 98 上运行的驱动程序使用 MmGetSystemAddressForMdl。)为了避免浪费系统资源,只有在确实需要使用虚拟地址来访问内存页时才进行这种操作。

  • 使用 MmGetSystemAddressForMdlSafe 返回的系统地址来通过虚拟地址访问锁定的页。

  • 当您不再需要 MDL 描述的页时,请调用 MmUnlockPages 将它们解除锁定,然后调用 IoFreeMdl 来释放它们。

下列来自 Windows DDK(%winddk%/src/general/ioctl/sys/sioctl.c)中的系统 IOCTL 例子代码片段展示了如何进行这种操作:

  1. mdl = IoAllocateMdl(inBuf, inBufLength,  FALSE, TRUE, NULL);
  2.         if(!mdl)
  3.         {
  4.             ntStatus = STATUS_INSUFFICIENT_RESOURCES;
  5.             break;
  6.         }
  7.             
  8.         try 
  9.         {            
  10.            
  11.             //
  12.             // Probe and lock the pages of this buffer in physical memory.
  13.             // You can specify IoReadAccess, IoWriteAccess or IoModifyAccess
  14.             // Always perform this operation in a try except block. 
  15.             //  MmProbeAndLockPages will raise an exception if it fails.
  16.             //
  17.             MmProbeAndLockPages(mdl, UserMode, IoReadAccess);
  18.         }
  19.         except(EXCEPTION_EXECUTE_HANDLER) 
  20.         {
  21.  
  22.             ntStatus = GetExceptionCode();
  23.             SIOCTL_KDPRINT((
  24.                 "Exception while locking inBuf 0X%08X in METHOD_NEITHER/n",
  25.                     ntStatus));
  26.             IoFreeMdl(mdl);
  27.             break;
  28.         }
  29.  
  30.         //         
  31.         // Map the physical pages described by the MDL into system space. 
  32.         // Note: double mapping the buffer this way causes lot of 
  33.         // system overhead for large size buffers. 
  34.         // 
  35.  
  36.         buffer = MmGetSystemAddressForMdlSafe(mdl, NormalPagePriority );
  37.  
  38.         if(!buffer) {
  39.                 ntStatus = STATUS_INSUFFICIENT_RESOURCES;
  40.                 MmUnlockPages(mdl);
  41.                 IoFreeMdl(mdl);
  42.                 break;           
  43.         }
  44.         
  45.         // 
  46.         // Now you can safely read the data from the buffer.
  47.         //
  48.         SIOCTL_KDPRINT(("/tData from User (SystemAddress) : "));
  49.         PrintChars(buffer, inBufLength);
  50.      
  51.         //
  52.         // Once the read is over unmap and unlock the pages.
  53.         //
  54.  
  55.         MmUnlockPages(mdl);
  56.         IoFreeMdl(mdl);
原创粉丝点击