C++ 内存分配处理函数set_new_handler的使用

当C++程序在运行时发现内存分配失败时，会抛出一个std::bad_alloc异常。为了避免程序崩溃，我们可以使用set_new_handler函数来注册一个新的处理函数，当内存分配失败时，程序会调用该函数来处理内存分配失败的情况。

set_new_handler函数的语法

set_new_handler函数是一个全局函数，它的原型如下：

std::new_handler set_new_handler(std::new_handler handler) noexcept;

其中，handler是一个回调函数，当程序在分配内存时发现无法分配时，会自动调用该函数。如果handler为空，则表示取消注册处理函数。

这个函数返回的是之前注册的处理函数的指针。如果返回nullptr，则表示之前没有注册过处理函数。

需要注意的是，set_new_handler函数在多线程程序中可能不可靠。因此，当我们使用多线程时，需要使用更加复杂的并发内存分配方案。

set_new_handler函数的使用示例

下面是set_new_handler函数的两个使用示例。

示例1：使用自定义的处理函数来避免程序崩溃

在这个例子中，假设程序发现内存分配失败时，它将调用一个名为my_new_handler的函数来处理该情况。如果该函数成功分配了内存，则返回一个非空指针。否则，该函数将打印一个错误消息，并返回nullptr以指示分配失败。

#include <iostream>
#include <new>

void* my_new_handler(std::size_t size) {
    std::cout << "Failed to allocate memory of size " << size << std::endl;
    return nullptr;
}

int main() {
    std::set_new_handler(my_new_handler);
    int* my_array = new int[1000000000000000000]; // 大于系统的可分配内存
    return 0;
}

当程序运行时，由于无法分配这么大的内存，程序将自动调用my_new_handler函数来处理内存分配失败的情况。在本例中，由于该函数返回nullptr，程序将打印一个消息并终止。

示例2：使用空的处理函数来取消注册处理函数

在这个例子中，我们将使用set_new_handler函数将处理函数取消注册，这样程序将恢复默认的行为，并抛出std::bad_alloc异常。

#include <iostream>
#include <new>

int main() {
    std::set_new_handler(nullptr);
    int* my_array = new int[1000000000000000000]; // 大于系统的可分配内存
    return 0;
}

当程序运行时，我们将不再使用自定义的处理函数，而是使用默认的行为。如果无法分配所需的内存，则程序将抛出std::bad_alloc异常。