C++ 中的 new operator, operator new,placement new

new operator
 string *sp = new string("hello world");
 delete sp;

这里的new称之为 new operator(new 运算符),它做2件事:

  • 调用的是标准库函数operator new分配一块足够大的,未命名的内存空间。
  • 编译器运行相应的构造函数在刚才分配的内存上构造对象 , 并返回指向该对象的指针。

这里的delete它做2件事:

  • 调用该对象的析构函数
  • 调用标准库函数operator delete释放内存
operator new, operator delete函数
/* 这些版本可能抛出异常 new分配失败,抛出bad_alloc异常 */
void *operator new(size_t);    
void *operator new[](size_t);
void *operator delete(void*);
void *operator delete[](void*);

/* 下面版本承诺不抛出异常 new分配失败,不抛出异常,返回nullptr指针 */
void *operator new(size_t, nothrow_t &)  noexcept;    
void *operator new[](size_t, nothrow_t &)  noexcept;
void *operator delete(void*, othrow_t &)  noexcept;
void *operator delete[](void*, othrow_t &)  noexcept;

我们可以在全局或者类里面自定义上面的函数,当我们将其定义成类的成员函数时,它是隐式静态成员函数。

/* 编写operator new(分配内存) 和 operator delete(释放内存)简单实现方式 */
void* operator new(size_t size) 
{
    if (void *mem = malloc(size))
        return mem;
    else
        throw bad_alloc();
}

void* operator delete() noexcept
{
    free(mem);
}
placement new

上面的 operator new 只是分配内存,而 placement new 不分配内存,只负责在预先分配的内存上构造对象。

new (place_address) type;
new (place_address) type(initializers);
new (place_address) type[size];
new (place_address) type[size]{ braced initializers list };
  • 例子:
string *sp = static_cast<string*>(::operator new(sizeof(string)));  //operator new 分配内存
new (sp) string("hello world");  // placement new 构造对象
    
cout << *sp << endl;
    
sp->~string();    // 主动调用析构函数
::operator delete(sp);  //operator delete 释放内存
set_new_handler

set_new_handler 是一个"接受一个不接受参数也无返回值的函数"的函数指针,而这个函数就被用来处理bad_alloc异常。( void *operator new(size_t) 或者 void *operator new 分配内存失败 )。

  • 例子
#include <iostream>     
#include <cstdlib>      
#include <new>

void no_memory() 
{
    std::cout << "Failed to allocate memory!\n";
    std::exit (1);
}

int main() 
{
    std::set_new_handler(no_memory);
    std::cout << "Attempting to allocate 2 GiB..." << std::endl;
    char* p = new char [2 * 1024 * 1024 * 1024];
    std::cout << "ok\n";
    delete[] p;
    return 0;
}
应用:不用工具,实现检测内存泄漏

整体思路:在申请内存时记录下该内存的地址和在代码中申请内存的位置,在内存销毁时删除该地址对应的记录,程序最后统计下还有哪条记录没有被删除,如果还有没被删除的记录就代表有内存泄漏。

  • 重载operator new 函数

    void *operator new(std::size_t size, const char *file, int line);
    void *operator new[](std::size_t size, const char *file, int line);
    
  • 使用宏定义替换new

    #define new new (__FILE__, __LINE__)        // new operator
    
  • 申请内存多申请一段内存记录元信息

    std::mutex new_output_lock;
    std::mutex new_ptr_lock;
    long long total_mem_alloc = 0;
    new_ptr_list_t new_ptr_list;
    
    void *operator new(std::size_t size, const char *file, int line)
    {
        return alloc_mem(size, file, line, false);
    }
    
    void *operator new[](std::size_t size, const char *file, int line)
    {
        return alloc_mem(size, file, line, true);
    }
    
    static void *alloc_mem(std::size_t size, const char *file, int line, bool is_array)
    {
        assert(line >= 0);
    
        std::size_t s = size + ALIGNED_LIST_ITEM_SIZE;
        new_ptr_list_t *ptr = (new_ptr_list_t *)malloc(s);
        if (ptr == nullptr)
        {
            std::unique_lock<std::mutex> lock(new_output_lock);
            printf("Out of memory when allocating %lu bytes\n", (unsigned long)size);
            abort();
        }
        void *usr_ptr = (char *)ptr + ALIGNED_LIST_ITEM_SIZE;
    
        if (line)
        {
            strncpy(ptr->file, file, DEBUG_NEW_FILENAME_LEN - 1);
            ptr->file[DEBUG_NEW_FILENAME_LEN - 1] = '\0';
        }
        else
        {
            ptr->addr = (void *)file;
        }
    
        ptr->line = line;
        ptr->is_array = is_array;
        ptr->size = size;
        ptr->magic = DEBUG_NEW_MAGIC;
    
        // 链表操作
        // 插入链表中
    
        total_mem_alloc += size;
    
        return usr_ptr;
    }
    
  • 释放内存
    链表中找到要对应节点,删除掉,

    void operator delete(void* ptr) noexcept { 
        free_pointer(ptr, nullptr, false); 
    }
    static void free_pointer(void* usr_ptr, void* addr, bool is_array) {
        if (usr_ptr == nullptr) 
        {
            return;
        }
        new_ptr_list_t* ptr = (new_ptr_list_t*)((char*)usr_ptr - ALIGNED_LIST_ITEM_SIZE);
        {
              // 链表移除结点操作
        }
        free(ptr);
    }
    
  • 如何检测是否有内存泄漏?
    遍历链表即可,每次new时候会把这段内存插入链表,delete时候会把这段内存从链表中移出,如果程序最后链表长度不为0,即为有内存泄漏。

  • 没有被new宏包裹的地方可以检测的到吗?
    没有被new宏包裹的地方是会调用operator new(std::size_t sz)函数来申请内存的。这里operator new函数不只可以重载,还可以重新定义它的实现,因为它是一个weak symbol,有关strong symbol和weak symbol的知识点可以看:强弱符号的问题

    void* operator new(std::size_t size) { 
        return operator new(size, nullptr, 0); 
    }
    

参考资料
1、https://www.zhihu.com/question/29859828/answer/1798470821

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,711评论 5 468
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,932评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,770评论 0 330
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,799评论 1 271
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,697评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,069评论 1 276
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,535评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,200评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,353评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,290评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,331评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,020评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,610评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,694评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,927评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,330评论 2 346
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,904评论 2 341

推荐阅读更多精彩内容