驱动开发：内存管理，防止内存泄漏

最新推荐文章于 2024-08-05 18:50:07 发布

原创最新推荐文章于 2024-08-05 18:50:07 发布 · 443 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#windows驱动开发 #内存管理 #内存泄漏

蓝屏专栏收录该内容

6 篇文章

订阅专栏

本文探讨了Windows驱动开发中防止内存泄漏的关键措施，包括动态内存申请后的指针管理，避免重复释放和蓝屏问题，正确使用Rtl*函数进行字符串处理，以及在存在break或return时确保内存正确释放。同时，对于函数内部申请内存并作为返回值的情况，提出了通过改变参数传递或修改函数设计来避免内存泄漏的建议。

动态申请和释放

ExFree**后，需要对指针置NULL，这样再释放之前可以判断是否为NULL来防止重复释放，否则重复释放会蓝屏
若调用了RtlAnsiStringToUnicodeString，需要使用RtlFreeUnicodeString释放
若为一段指针动态申请了一段内存，不要再直接让它指向常量，而应该使用RtlStringCbCopy类函数进行复制。否则后期对常量指针释放会导致蓝屏。
特别注意在申请和释放之间，若存在break或return需要特别注意是不是会跳过释放语句。若会跳过，则应增加释放语句。如

do{
UNICODE_STRING uni_string;
ExAllocatePoolWithTag(&uni_string,1024,TAG);/*申请内存*/
if(**)
{
//若进入这个语句，则会跳出循环而未内存释放，因此应在这里加入
//ExFreePoolWithTag(&uni_string,TAG);
break;
}
if(**)
{
//若进入这个语句，则会结束函数调用而未内存释放，因此应在这里加入
//ExFreePoolWithTag(&uni_string,TAG);
return;
}
ExFreePoolWithTag(&uni_string,TAG);/*释放内存*/
}while(true);

若在函数内部申请了内存，而对应的变量需要作为返回值从而无法在函数内部释放，则可以将返回值改为参数传入改变。

PUNICODE_STRING example()
{
UNICODE_STRING uni_string;
ExAllocateWithTag(&uni_string,1024,TAG);
...
xx(&uni_string,..);//对uni_string进行操作
...
retun &uni_string;
}
NTSTATUS DriverEntry()
{
PUNICODE_STRING String;
ExAllocatePoolWithTag(String,1024,TAG);
String=example;
ExFreePoolWithTag(String,TAG);
...
}

这个时候就会存在原来的指针指向的地址丢失而内存泄漏。
当然，仅对上面这个例子来说，可以尝试不对外面的指针申请内存，而直接调用example函数，再释放内存。
若怕这种方法容易混乱、遗忘释放内存或者存在没办法不申请内存直接调用的情况。可以考虑修改example函数，改完传参调用。

void example(PUNICODE_STRING uni_string)
{
...
xx(uni_string,..);//无须定义，内存申请，直接对uni_string进行操作
...
retun；
}
NTSTATUS DriverEntry()
{
PUNICODE_STRING String;
ExAllocatePoolWithTag(String,1024,TAG);
example（String）;
ExFreePoolWithTag(String,TAG);
...
}