对于new出来的对象,用智能指针无可厚非,避免马虎大意造成内存泄漏。
但是对于malloc出来的内存,就不能用智能指针释放了,比如在使用ffmpeg的时候,申请的内存,这种非常容易忘记释放而导致内存泄漏,我实现了类似于智能指针的类来辅助释放这类内存块。
#include <functional>
template <class T>
class AutoFree
{
public:
AutoFree(T** ptr,std::function<void(void*)> call)
:value(ptr),call_back(call){}//这里为什么传入二级指针,因为方便释放之后把指针置为null,详见下面代码
virtual ~AutoFree()
{
if(*value)
{
call_back((T*)(*value));//调用回调
*value = NULL;
}
}
private:
T **value;
std::function<void(void*)> call_back;
};
代码不多,但是功能很强大。
使用方法:
#include <stdio.h>
#include <string>
#include <stdlib.h>
#include <autofree.h>
int main()
{
char *a = (char*)malloc(100);
{
printf("a start address %p..\n",a);
AutoFree<char> aut(&a,[](void *ptr){free(ptr);}); //通过lambda表达式构造function对象
printf("a address %p.\n",a);
}
printf("a end address %p.\n",a);
}
运行结果:
a start address 0x7ff049403220..
a address 0x7ff049403220.
a end address 0x0.
如果是通过ffmpeg申请AVCodecContext要自动释放就可以这样写
AVCodecContext *ctx = avcodec_alloc_context3();
AutoFree<AVCodecContext> a(&ctx,[](AVCodecContext *ctx){avcodec_close(ctx);avcodec_free_context(&ctx);});
当对象a结束自己的生命周期的时候,ctx就会自动被释放了。