3.6 内存管理

3.6.1 分配系统空间内存

驱动程序可以在设备扩展中保存指向系统分配空间的指针，这个系统分配的空间可以作为保存设备指定信息的全局存储区域。

PVOID ExAllocatrPoolWithTag {IN POOL_TYPE PoolType, IN SIZE_T NumberOfBytes, IN ULONG Tag};

释放函数：ExFreePool或者ExFreePoolWithTag

3.6.2 运行时库管理函数

1.内存比较函数：SIZE_T RtlCompareMemory(IN CONST VOID *Source1, IN CONST VOID *Source2, IN SIZE_T Length);

2.内存拷贝函数：

RtlCopyMemory：VOID RtlCopyMemory(IN VOID UNALIGNED *Destination,IN CONST VOID UNALIGNED *Source,IN SIZE_T Length);

RtlMoveMemory：VOID RtlMoveMemory(IN VOID UNALIGNED *Destination,IN CONST VOID UNALIGNED *Source,IN SIZE_T Length);

3.内存填充函数：

RtlFillMemory：VOID RtlFillMemory(IN VOID UNALIGNED *Destination,IN SIZE_T Length,IN UCHAR Fill);

RtlZeroMemory：VOID RtlZeroMemory(IN VOID UNALIGNED *Destination,IN SIZE_T Length);

3.6.3 使用内核栈

内核栈的大小约是3个页面，使用IoGetRemainingStackSize函数获取可以内核栈的大小

3.6.4 使用Lookaside快速链表

如果一个驱动程序需要动态地分配固定大小的缓冲区来满足I/O操作，那么驱动程序可以考虑使用Lookaside快速链表

一套管理LOOKASIDE_LIST_EX数据结构的函数集合：

1. ExInitializeLookasideListEx用于初始化一个Lookaside链表：

NTSTATUS ExInitializeLookasideListEx(

__out PLOOKASIDE_LIST_EX lookaside, __in_opt PALLOCATE_FUNCTION_EX Allocate,

__in_opt PFREE_FUNCTION_EX Free, __in POOL_TYPE PoolType, __in ULONG Flags,

__in SIZE_T Size, __in ULONG Tag, __in USHORT Depth);

示例：

首先，创建一个数据结构，通过该数据结构可以监控Lookaside链表的节点分配情况

typedef struct

{ULONG NumberOfAllocations; //分配节点的数量

 ULONG NumberOfFrees;  //释放节点的数量

 LOOKASIDE_LIST_EX LookasideField;

} MY_PRIVATE_DATA;

接着，驱动程序调用ExInitializeLookasideListEx函数初始化一个Lookaside链表

#define ENTRY_SIZE 256

MY_PRIVATE_DATE *MyContext;

NTSTATUS status = STATUS_SUCCESS;

MyContext = ExAllocatePoolWithTag(NonPagedPool,sizeof(MY_PRIVATE_DATA),'tsLL');

if (MyContext)

{

MyContext.NumberOfAllocations = 0;

MyContext.NumberOfFree = 0;

status = ExInitializeLookasideListEx(&MyContext.LookasideField, MyLookasideListAllocate, MyLookasideListFreeEx,

  NonPagedPool,0,ENTRY_SIZE,0);

}

else

{

ststus = STSTUS_INSUFFICIENT_RESOURCES;

}

下面的代码介绍了LookasidedListAllocateEx函数如何通过Lookaside参数访问关联的其他统计数据

PVOID MyLookasideListAllocateEx(__in POOL_TYPE PoolType, __in SIZE_T NumberOfBytes, __in ULONG Tag, __inout PLOOKASIDE_LIST_EX lookaside)

{

MY_PRIVATE_DATA *MyContext;

PVOID NewEntry;

MyContext = CONTAINING_RECORD(Lookaside, MY_PRIVATE_DATA, LookasideField);

NewEntry = ExAllocatePoolWithTag(PoolType,NumberOfBytes,Tag)

if (NewEntry)

{

MyContext->NumberOfAllocations += 1;

}

return NewEntry;

}

2. ExAllocateFromLookasideListEx

3. ExFreeToLookasideEx

4. ExFlushLookasideEx

5. ExDeleteLookasideEx

3.6.5 访问用户空间内存

高级驱动程序在其分发函数中可调用MmProbeAndLookPages函数锁定用户模式的虚拟地址，这样，该虚拟地址对应的物理内存页面就不会被换页了，驱动程序就可以安全地访问该内存

3.6.6 内存区对象和视图

内存区对象代表了一片可以共享的内存，内存区对象有两大作用：

一个进程可以使用内存区对象和别的进程共享其内存地址空间

一个进程可以使用内存区对象映射文件到其内存地址空间

1. 基于文件支持和基于页面文件支持的内存区

2. 内存区管理

驱动通过调用ZwCreateSection函数创建一个内存区对象，函数原型：

NTSTATUS ZwCreateSection (OUT PHANDLE SectionHandle, IN ACCESS_MASK DesiredAccess, IN POBJECT_ATTRIBUTES ObjectAttributes OPTIONAL,

IN PLARGE_INTEGER Maximumsize OPTIONAL, IN ULONG SectionPageProtection,

IN ULONG AllocationAttributes, IN HANDLE FileHandle OPTIONAL);

该函数返回一个指向内存区对象的句柄SectionHandle，最后一个参数为NULL则为由页面文件支持为文件句柄则为由普通文件支持

驱动程序调用ZeOpenSection函数打开该内存区，调用ZwMapViewOfSection函数将内存区对象的一个视图映射到当前进程地址空间中，函数原型：

NTSTATUS ZwMapViewOfSection (IN HANDLE SectionHandle, IN HANDLE ProcessHandle, IN OUT PVOID *BaseAddress, IN ULONG_PTR ZeroBits,

IN SIZE_T CommitSize, IN OUT PLARGE_INTEGER SectionOffset OPTIONAL, IN OUT PSIZE_T ViewSize,

IN SECTION_INHERIT InheritDisposition, IN ULONG AllocationType, IN ULONG Win32Protect);

调用ZwUnmapViewOfSection函数解除映射关系

当驱动程序不再需要使用内存区对象时，可以调用ZwClose函数关闭内存区对象句柄

3.6.7 MDL的使用

一个由一系列连续虚拟内存地址组成的I/O缓冲区所对应的物理页面很有可能是非连续的，操作系统使用了MDL结构描述了这些物理页面和虚拟内存地址间的联系

MDL数据结构是非透明的,可以直接访问其中的Next和MdlFlags域，应该使用系统提供的宏执行相关的访问操作：

MmGetMdlVirtualAddress：返回了I/O缓冲区对于的虚拟地址

MmGetMdlByteCount：返回了I/O缓冲区的大小

MmGetMdlByteOffset：返回了该I/O缓冲区的首地址在物理页面中的偏移

MmGetMdlPfnArray：返回了一个指向一组物理页面编号数组的指针

可以调用IoAllocateMdl函数分配一个Mdl结构，函数原型：

PMDL IoAllocateMdl (__in_opt PVOID VirtualAddress, __in ULONG Length, 

__in BOOLEAN SecondaryBuffer, __in BOOLEAN ChargeQuota, __inout_opt PIRP Irp OPTINAL);

当需要释放该结构时，调用IoFreeMdl函数。另外也可以分配一块非分页内存，然后调用MmInitializeMdl函数格式化该内存为MDL结构

对于从非分页内存池中分配的缓冲区，直接调用MmBuildMdlForNonPagedPool函数即可初始化物理页面编号数组，该函数用于更新MDL结构，

关联MDL的虚拟内存地址和物理页面；对于从分页内存池中分配的缓冲区，调用MmProbeAndLockPages函数锁定分页内存并初始化物理页面编号 数组，

因为对于分页内存而言其虚拟地址和物理页面的关联是临时性的，所以必须锁定分页内存，锁定的内存只有调用MmUnlockPages函数才会解除锁定。