内存操作相关内核 API 的使用

1、RtlCopyMemory 、RtlCopyBytes、RtlMoveMemory;

2、RtlZeroMemory、RtlFillMemory;

3、RtlEqualMemory;

4、ExAllocatePool、ExFreePool;

5、New(重载)、Delete操作符

一、内存的复制与移动

1、RtlCopyMemory

作用:把一个缓冲区的内容复制到另一一个缓冲区。

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

参数:
Destination

        指针移动的目的地(也就是要粘贴的地方)

Source
指针指向的要复制的内存 .
Length
        指定要复制的字节数。
返回值

     无

但是如果内存有互相覆盖的情况可能出错,因为源和目的重叠的部分有可能还没读取就被写入了,要用 RtlMoveMemory 进行移除。

 

2、RtlCopyBytes

作用:复制给定的字节数从一个位置到另一个地方。

VOID RtlCopyBytes(
    IN PVOID Destination,
    IN CONST VOID *Source,
    IN SIZE_T Length
    );

参数:
Destination

        指向目的地的字节(也就是要粘贴的地方)。

Source         指针指向的复制的内存. Length

        指定要复制的字节数。

返回值

     无

推荐驱动应该使用 RtlCopyMemory 进行常规操作,而不是 RtlCopyBytes 。

 

3、RtlMoveMemory

作用:从指定内存中复制内存至另一内存里.简称:复制内存。RtlCopyMemory非重叠复制,而RtlMoveMemory是重叠复制

对内存进行要么向前或向后移动,或未对齐,对齐在4字节块,紧随其后的是剩余的字节。

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

参数:
Destination

        指针移动的目的地(也就是要移到的地方)。

Source         指针指向的复制的内存. Length

        指定要复制的字节数。

返回值

     无

•内存的“copy”和“move”操作之间的区别在于可否容忍源和目的相重叠。move操作不管源和目的是否重叠。而copy操作在源和目的有任何重叠时不工作。

•“byte” 操作和“memory”操作的区别是操作的间隔尺寸。byte操作保证按字节为单位执行。而memory操作可以在内部使用更大的块,所有这些块的和等于 指定的字节数。这个区别会根据平台的不同而改变,在32位Intel计算机上,byte操作实际上是对应memory操作的宏。但在Alpha平台 上,RtlCopyBytes与RtlCopyMemory是完全不同的函数。

 

二、内存的填充

1、RtlZeroMemory

作用:用0来填充一块内存区域。

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

参数:
Destination

        指向一块准备用0来填充的内存区域的开始地址。(也就是要填充的地方)。

Length
返回值

     无

 

2、RtlFillMemory

作用:调用者向提供的缓冲区填满给定的字符。

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

参数:
Destination

        指针指向的内存(也就是要填充的地方)。

Length          指定的字节数(要填充多少(大)) Fill

        指定填充内存的值。

返回值

     无

 

三、内存的比较

1、RtlEqualMemory

作用:对两个内存块进行比较,以确定指定的字节数是相同的。

LOGICAL RtlEqualMemory(
CONST VOID  *Source1,
CONST VOID  *Source2,
SIZE_T Length
    );

参数:
Source1

        指针指向一个调用者提供的要比较的内存块。

Source2  Fill

        待比较内存的长度,单位为字节。

返回值

     返回两块内存相等的字节数,并且如果一致就返回True,否则返回False。

 

四、内存的分配和释放

1、ExAllocatePool

作用: ExAllocatePool分配池指定类型和返回一个指针指向的内存分配的块。另外:ExAllocatePool程序已经过时了,出口仅为现有的二进制文件。建议使用ExAllocatePoolWithTag代替。 

PVOID ExAllocatePool(
IN POOL_TYPE PoolType,
IN SIZE_T NumberOfBytes
    );

参数:
PoolType

        给要分配的内存池指定类型。

NumberOfBytes
返回值

     如果没有足够的内存空闲池中满足要求,返回NULL。否则,程序返回一个指向分配的内存的指针。

 

2、ExFreePool

作用:释放一块内存池。

VOID ExFreePool(
IN PVOID P
    );

参数:
P

        指定要被收回的内存块地址池。

返回值

       无

 

四、内存的重载操作符

1、New

作用:重载New,也就是用来分配内存。所以这个函数里面要用到 ExAllocatePool API函数,让其达到重载 New 这个函数不用直接调用 ExAllocatePool 这个API函数也能分配内存的目的。

void * __cdecl operator new(size_t size,POOL_TYPE PoolType=PagedPool) 

KdPrint(("global operator new\n")); 

KdPrint(("Allocate size :%d\n",size)); 

return ExAllocatePool(PagedPool,size); 

 

2、delete

作用:重载delete,也就是用来释放内存。所以这个函数里面要用到 ExFreePool API函数,让其达到重载 delete 这个函数不用直接调用 ExFreePool 这个API函数也能释放内存的目的。

void __cdecl operator delete(void* pointer) {
KdPrint(("Global delete operator\n"));

 ExFreePool(pointer);

下面看一段代码

#pragma once 
#ifdef _cplusplus 
extern "C"
{ 
#endif 
#include <NTDDK.h> //这里包含需要用C方式编译的头文件 
//#include <winnt.h>  运行库已包含了,就不需要,如果报错就需要添加此头文件
#ifdef _cplusplus 
} 
#endif  

//重载new和delete操作符
//重载new   
void * _cdecl operator new(size_t size, POOL_TYPE PoolType = PagedPool){  //操作符重载,否则报错
    KdPrint(("global operator new\n")); 
    KdPrint(("Allocate size :%d\n", size));
    return ExAllocatePool(PagedPool, size);  //分配内存
} 


//重载delete  
void __cdecl operator delete(void* pointer) {
    KdPrint(("Global delete operator\n"));
    ExFreePool(pointer);//释放指针 
}  



#pragma INITCODE
VOID MemoryOpe() {    
    VOID UNALIGNED *d;  //有关于WIN64的定义,无符号 
    VOID UNALIGNED *s;
    SIZE_T Length = 8; //无符号类型  
    ULONG ulRet; 
    char *buffer = new (PagedPool)  char[111]; //分配111字节指针
    delete buffer;  
    //为s指针分配大小为字节的内核内存  
    s=ExAllocatePool(PagedPool,Length);
    KdPrint(("s=%x \n",(int*)s));
    //  _asm int 3 db 
    //为d指针分配大小为字节的内核内存
    d=ExAllocatePool(PagedPool,Length);
    //  __asm int 3 db 
    KdPrint(("d=%x \n", (int*)d));
    //用来填充s指针指向的内存填充长度为Length=8
    RtlFillMemory(s,Length,'s');
    //  __asm int 3 db  
    KdPrint(("RtlFillMemory 1 \n"));
    //复制S指针指向的内容到D地址复制长度为Length=8
    RtlCopyMemory(d,s,Length); //memcpy
    //  __asm int 3 
    KdPrint(("RtlCopyMemory s to d \n"));
    //复制S指针指向的内容到D地址复制长度为Length=8
    RtlCopyBytes(d,s,Length);     
    //判断内存是否一致 
    ulRet = RtlCompareMemory(d, s, Length); // 返回为1则相等
    //RtlEqualMemory(d,s,Length); 在VS下此句编译错误所以就用DDK进行编译和此函数时请用build 
    if (ulRet == Length){ //如果返回值    
        KdPrint(("111 D和S 内存块相同.\n"));
    }
    else{
        KdPrint(("111 D和S 内存块不相同\n"));
    }


    //清空S指针指向地址
    RtlZeroBytes(s,Length);  
    ulRet = RtlCompareMemory(d, s, Length);
    if (ulRet == Length){  //如果返回值    
        KdPrint(("222 D和S 内存块相同.\n"));
    }
    else{
        KdPrint(("222 D和S 内存块不相同\n"));
    } 
    ExFreePool(s);
    return;  
}

建议在驱动入口函数 DriverEntry 函数中的设备加载过程执行后,再调用此内存操作过程。

在虚拟机下执行过程, 首先是填充了S和D相同的数据所以相同, 然后由于清空S所以S和D就不相同了。

最后将尽量用DDK的RtlCompareMemory函数可移植性比较好.

转载于:https://www.cnblogs.com/lfls128/p/5020282.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值