驱动开发:内核物理内存寻址读写

打印 上一主题 下一主题

主题 900|帖子 900|积分 2700

在某些时候我们需要读写的进程可能存在虚拟内存保护机制,在该机制下用户的CR3以及MDL读写将直接失效,从而导致无法读取到正确的数据,本章我们将继续研究如何实现物理级别的寻址读写。
首先,驱动中的物理页读写是指在驱动中直接读写物理内存页(而不是虚拟内存页)。这种方式的优点是它能够更快地访问内存,因为它避免了虚拟内存管理的开销,通过直接读写物理内存,驱动程序可以绕过虚拟内存的保护机制,获得对系统中内存的更高级别的访问权限。
想要实现物理页读写,第一步则是需要找到UserDirectoryTableBase的实际偏移地址,你一定会问这是个什么?别着急,听我来慢慢解释;
在操作系统中,每个进程都有一个KPROCESS结构体,它是进程的内部表示。该结构体中包含了一些重要的信息,包括UserDirectoryTableBase字段,它指向进程的页表目录表(Page Directory Table),也称为DirectoryTable页目录表。
Page Directory Table是一种数据结构,它在虚拟内存管理中起着重要的作用。它被用来存储将虚拟地址映射到物理地址的映射关系,其内部包含了一些指向页表的指针,每个页表中又包含了一些指向物理页面的指针。这些指针一起构成了一个树形结构,它被称为页表树(Page Table Tree)。
  1. kd> dt _KPROCESS
  2. ntdll!_KPROCESS
  3. +0x278 UserTime         : Uint4B
  4. +0x27c ReadyTime        : Uint4B
  5. +0x280 UserDirectoryTableBase : Uint8B
  6. +0x288 AddressPolicy    : UChar
  7. +0x289 Spare2           : [71] UChar
  8. #define GetDirectoryTableOffset 0x280
复制代码
UserDirectoryTableBase字段包含了进程的页表树的根节点的物理地址,通过它可以找到进程的页表树,从而实现虚拟内存的管理。在WinDbg中,通过输入dt _KPROCESS可以查看进程的KPROCESS结构体的定义,从而找到UserDirectoryTableBase字段的偏移量,这样可以获取该字段在内存中的地址,进而获取DirectoryTable的地址。不同操作系统的KPROCESS结构体定义可能会有所不同,因此它们的UserDirectoryTableBase字段的偏移量也会不同。
通过上述原理解释,我们可知要实现物理页读写需要实现一个转换函数,因为在应用层传入的还是一个虚拟地址,通过TransformationCR3函数即可实现将虚拟地址转换到物理地址,函数内部实现了从虚拟地址到物理地址的转换过程,并返回物理地址。
  1. // 从用户层虚拟地址切换到物理页地址的函数
  2. // 将 CR3 寄存器的末尾4个比特清零,这些比特是用于对齐的,不需要考虑
  3. /*
  4.     参数 cr3:物理地址。
  5.     参数 VirtualAddress:虚拟地址。
  6. */
  7. ULONG64 TransformationCR3(ULONG64 cr3, ULONG64 VirtualAddress)
  8. {
  9.   cr3 &= ~0xf;
  10.   // 获取页面偏移量
  11.   ULONG64 PAGE_OFFSET = VirtualAddress & ~(~0ul << 12);
  12.   // 读取虚拟地址所在的三级页表项
  13.   SIZE_T BytesTransferred = 0;
  14.   ULONG64 a = 0, b = 0, c = 0;
  15.   ReadPhysicalAddress((PVOID)(cr3 + 8 * ((VirtualAddress >> 39) & (0x1ffll))), &a, sizeof(a), &BytesTransferred);
  16.   // 如果 P(存在位)为0,表示该页表项没有映射物理内存,返回0
  17.   if (~a & 1)
  18.   {
  19.     return 0;
  20.   }
  21.   // 读取虚拟地址所在的二级页表项
  22.   ReadPhysicalAddress((PVOID)((a & ((~0xfull << 8) & 0xfffffffffull)) + 8 * ((VirtualAddress >> 30) & (0x1ffll))), &b, sizeof(b), &BytesTransferred);
  23.   // 如果 P 为0,表示该页表项没有映射物理内存,返回0
  24.   if (~b & 1)
  25.   {
  26.     return 0;
  27.   }
  28.   // 如果 PS(页面大小)为1,表示该页表项映射的是1GB的物理内存,直接计算出物理地址并返回
  29.   if (b & 0x80)
  30.   {
  31.     return (b & (~0ull << 42 >> 12)) + (VirtualAddress & ~(~0ull << 30));
  32.   }
  33.   // 读取虚拟地址所在的一级页表项
  34.   ReadPhysicalAddress((PVOID)((b & ((~0xfull << 8) & 0xfffffffffull)) + 8 * ((VirtualAddress >> 21) & (0x1ffll))), &c, sizeof(c), &BytesTransferred);
  35.   // 如果 P 为0,表示该页表项没有映射物理内存,返回0
  36.   if (~c & 1)
  37.   {
  38.     return 0;
  39.   }
  40.   // 如果 PS 为1,表示该页表项映射的是2MB的物理内存,直接计算出物理地址并返回
  41.   if (c & 0x80)
  42.   {
  43.     return (c & ((~0xfull << 8) & 0xfffffffffull)) + (VirtualAddress & ~(~0ull << 21));
  44.   }
  45.   // 读取虚拟地址所在的零级页表项,计算出物理地址并返回
  46.   ULONG64 address = 0;
  47.   ReadPhysicalAddress((PVOID)((c & ((~0xfull << 8) & 0xfffffffffull)) + 8 * ((VirtualAddress >> 12) & (0x1ffll))), &address, sizeof(address), &BytesTransferred);
  48.   address &= ((~0xfull << 8) & 0xfffffffffull);
  49.   if (!address)
  50.   {
  51.     return 0;
  52.   }
  53.   return address + PAGE_OFFSET;
  54. }
复制代码
编译并运行上述代码片段,则会读取进程ID为4116的0x401000处的地址数据,并以字节的方式输出前四位,输出效果图如下所示;

写出数据与读取数据基本一致,只是调用方法从ReadPhysicalAddress变为了WritePhysicalAddress其他的照旧,但需要注意的是读者再使用写出时需要自行填充一段堆用于存储需要写出的字节集。
  1. #include <ntifs.h>
  2. #include <windef.h>
  3. #define GetDirectoryTableOffset 0x280
  4. #define bit64 0x28
  5. #define bit32 0x18
  6. // 读取物理内存封装
  7. // 这段代码实现了将物理地址映射到内核空间,然后将物理地址对应的数据读取到指定的缓冲区中。
  8. /*
  9.     address:需要读取的物理地址;
  10.     buffer:读取到的数据需要保存到的缓冲区;
  11.     size:需要读取的数据大小;
  12.     BytesTransferred:实际读取到的数据大小。
  13. */
  14. NTSTATUS ReadPhysicalAddress(PVOID address, PVOID buffer, SIZE_T size, SIZE_T* BytesTransferred)
  15. {
  16.   MM_COPY_ADDRESS Read = { 0 };
  17.   Read.PhysicalAddress.QuadPart = (LONG64)address;
  18.   return MmCopyMemory(buffer, Read, size, MM_COPY_MEMORY_PHYSICAL, BytesTransferred);
  19. }
  20. // 写入物理内存
  21. // 这段代码实现了将数据写入物理地址的功能
  22. /*
  23.     参数 address:要写入的物理地址。
  24.     参数 buffer:要写入的数据缓冲区。
  25.     参数 size:要写入的数据长度。
  26.     参数 BytesTransferred:实际写入的数据长度。
  27. */
  28. NTSTATUS WritePhysicalAddress(PVOID address, PVOID buffer, SIZE_T size, SIZE_T* BytesTransferred)
  29. {
  30.   if (!address)
  31.   {
  32.     return STATUS_UNSUCCESSFUL;
  33.   }
  34.   PHYSICAL_ADDRESS Write = { 0 };
  35.   Write.QuadPart = (LONG64)address;
  36.    
  37.     // 将物理空间映射为虚拟空间
  38.   PVOID map = MmMapIoSpace(Write, size, (MEMORY_CACHING_TYPE)PAGE_READWRITE);
  39.   if (!map)
  40.   {
  41.     return STATUS_UNSUCCESSFUL;
  42.   }
  43.     // 开始拷贝数据
  44.   RtlCopyMemory(map, buffer, size);
  45.   *BytesTransferred = size;
  46.   MmUnmapIoSpace(map, size);
  47.   return STATUS_SUCCESS;
  48. }
  49. // 从用户层虚拟地址切换到物理页地址的函数
  50. // 将 CR3 寄存器的末尾4个比特清零,这些比特是用于对齐的,不需要考虑
  51. /*
  52.     参数 cr3:物理地址。
  53.     参数 VirtualAddress:虚拟地址。
  54. */
  55. ULONG64 TransformationCR3(ULONG64 cr3, ULONG64 VirtualAddress)
  56. {
  57.   cr3 &= ~0xf;
  58.   // 获取页面偏移量
  59.   ULONG64 PAGE_OFFSET = VirtualAddress & ~(~0ul << 12);
  60.   // 读取虚拟地址所在的三级页表项
  61.   SIZE_T BytesTransferred = 0;
  62.   ULONG64 a = 0, b = 0, c = 0;
  63.   ReadPhysicalAddress((PVOID)(cr3 + 8 * ((VirtualAddress >> 39) & (0x1ffll))), &a, sizeof(a), &BytesTransferred);
  64.   // 如果 P(存在位)为0,表示该页表项没有映射物理内存,返回0
  65.   if (~a & 1)
  66.   {
  67.     return 0;
  68.   }
  69.   // 读取虚拟地址所在的二级页表项
  70.   ReadPhysicalAddress((PVOID)((a & ((~0xfull << 8) & 0xfffffffffull)) + 8 * ((VirtualAddress >> 30) & (0x1ffll))), &b, sizeof(b), &BytesTransferred);
  71.   // 如果 P 为0,表示该页表项没有映射物理内存,返回0
  72.   if (~b & 1)
  73.   {
  74.     return 0;
  75.   }
  76.   // 如果 PS(页面大小)为1,表示该页表项映射的是1GB的物理内存,直接计算出物理地址并返回
  77.   if (b & 0x80)
  78.   {
  79.     return (b & (~0ull << 42 >> 12)) + (VirtualAddress & ~(~0ull << 30));
  80.   }
  81.   // 读取虚拟地址所在的一级页表项
  82.   ReadPhysicalAddress((PVOID)((b & ((~0xfull << 8) & 0xfffffffffull)) + 8 * ((VirtualAddress >> 21) & (0x1ffll))), &c, sizeof(c), &BytesTransferred);
  83.   // 如果 P 为0,表示该页表项没有映射物理内存,返回0
  84.   if (~c & 1)
  85.   {
  86.     return 0;
  87.   }
  88.   // 如果 PS 为1,表示该页表项映射的是2MB的物理内存,直接计算出物理地址并返回
  89.   if (c & 0x80)
  90.   {
  91.     return (c & ((~0xfull << 8) & 0xfffffffffull)) + (VirtualAddress & ~(~0ull << 21));
  92.   }
  93.   // 读取虚拟地址所在的零级页表项,计算出物理地址并返回
  94.   ULONG64 address = 0;
  95.   ReadPhysicalAddress((PVOID)((c & ((~0xfull << 8) & 0xfffffffffull)) + 8 * ((VirtualAddress >> 12) & (0x1ffll))), &address, sizeof(address), &BytesTransferred);
  96.   address &= ((~0xfull << 8) & 0xfffffffffull);
  97.   if (!address)
  98.   {
  99.     return 0;
  100.   }
  101.   return address + PAGE_OFFSET;
  102. }
复制代码
如上代码运行后,会向进程ID为4116的0x401000处写出4字节的0x90机器码,读者可通过第三方工具验证内存,输出效果如下所示;


免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

九天猎人

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表