需求

在不严肃的场景下，你想测试、跟踪自己的代码的内存分配，粗略评估有无错误的、意外的行为，怎么做？

代码

很简单，直接重写 new 操作符。

#include <iostream>static int malloc_count = 0;void* operator new(size_t size)
{std::cout << "new malloc size:" << size << std::endl;malloc_count++;return malloc(size);
}

我们用两个测试例子试用一下。

void Print(const char* const &c_str)
{std::cout << c_str << std::endl;
}void testNetOpStringMain()
{
#define USE_STRING 0
#if USE_STRING == 1std::string s = "Test String."; Print(s.c_str());
#elseconst char* c = "Test Const Char.";Print(c);
#endifstd::cout << "Total malloc times:" << malloc_count << std::endl;
}

使用 std::string 的输出如下：

new malloc size:16
Test String.
Total malloc times:1

使用 const char* 的输出如下：

Test Const Char.
Total malloc times:0

what do we learn?

• new 只是一个操作符，你可以重载掉它的行为。参考上面代码来重载 operator new（内存分配操作符）部分。另一个初始化部分 placement new（初始化操作符）有机会后面介绍。
• std::string 会在堆上分配字符缓冲区，会带来字符串操作的代价。如非必要，尽量使用字符串常量，这些常量由编译器存放在统一的区域，常常能被复用。
• 有时候用这个技巧来测试代码的分配行为，非常好用。可以拓展设置一些阈值性的条件，配合 stack 就能做一个很简易的分配行为跟踪。