内存泄漏专题（8）hook之C++运算符重载

C++主要是针对new/delete运算符进行重载 ，思路和上述hook malloc函数一样，我们可以定义一个MemCheck类，该类用来存储内存信息，包括内存的地址，大小。如果愿意的话，还可以存储调用的位置，这里为了简化处理，就不演示了。在调用 new运算符时，我们将申请内存的相关信息存入MemCheck中的一个map中，它的key是内存地址，value是内存大小。在调用delete时，将该内存地址从map中删除掉。（此处未判断double free的问题，可以考虑下如何去做？）有兴趣的话也可以考虑一下new[]和delete[]的重载实现。

我们定义一个全局静态的MemCheck类实例，这样该实例对象的析构将会在main函数结束后才会进行，因此，我们只需要在析构函数中判断一下，map中是否还有没有元素存在，就知道还有没有内存泄漏了。将其打印出来即可。

完整代码示例如下：

//new.cpp
#include <iostream>
#include <malloc.h>
#include <memory>
#include <map>

bool new_enabled = true;

class MemCheck
{
private:
    std::map<void*, size_t> MemInfo; //key 为申请内存的地址， value为申请内存的大小
public:
    void Allocation(void *p, size_t size){
        //申请内存时将内存信息放入map
        new_enabled = false;
        MemInfo[p] = size;
        new_enabled = true;
    }

    void Freed(void *p) {
        //释放内存时将内存信息从map中移除
        MemInfo.erase(p);
    }

    ~ MemCheck(){
        std::cout << "Memory Not Freed:\n";
        std::map<void *, size_t>::iterator it = MemInfo.begin();
        while (it != MemInfo.end()) {
            std::cout << "\t[" << it->first << "] " << it->second << " bytes\n";
            it++;
        }
    }
};

static MemCheck mcheck;

void *operator new(size_t size)
{
    if (new_enabled) {
        void *p = malloc(size);
        mcheck.Allocation(p, size);
        return p;
    } else {
        return malloc(size);
    }
}

void operator delete(void *p) {
    mcheck.Freed(p);
    free(p);
}

struct Object {
    int x, y, z;
};

void test(){
    std::string str = "hello";
    Object *obj = new Object;
    std::unique_ptr<Object> obj2(new Object());
    //delete obj;
}

int main(void){
    test();
    int *pInt = new int;
    std::unique_ptr<int> p(new int);
    return 0;
}

以上代码，在test函数中，首先string的内存管理是交由编译器处理的，虽然程序员无需手动管理，但实际上也调用了new和delete，obj对象只有申请，没有释放，obj2是智能指针，其内存的生命周期由编译器进行管理，无需手动释放，在test函数结束时会自动释放掉。

在main函数中，pInt的内存只有申请，没有释放，p是智能指针，在main函数结束时会自动释放。因此，该段代码有两处内存泄漏，分别为指针pInt和obj，我们运行该段代码：

[root@ck08 memleak]# g++ new.cpp -std=c++11 -g
[root@ck08 memleak]# ./a.out 
Memory Not Freed:
        [0x7c5080] 12 bytes
        [0x7c50e0] 4 bytes

可以看到，确实打印出了两处内存泄漏，第一处是0x7c5080，该地址是obj的地址，第二处是0x7c50e0，正是pInt的地址。而指针p即使是在main函数结束时才被释放，但MemCheck仍然能够感知到其内存被释放了。

---

本专栏知识点是通过<零声教育>的系统学习，进行梳理总结写下文章，对C/C++课程感兴趣的读者，可以点击链接，查看详细的服务：C/C++Linux服务器开发/高级架构师