C++奇淫巧计：如何快速观察内存分配

natsusao

已于 2024-02-06 14:36:04 修改

阅读量312

点赞数 13

分类专栏： C++ tricks 文章标签： c++ 开发语言

于 2024-02-06 12:17:04 首次发布

本文链接：https://blog.csdn.net/weixin_42445065/article/details/136054651

版权

C++ tricks 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

文章介绍了如何通过重写`operatornew`来测试和跟踪代码的内存分配，比较了std::string和constchar*的内存使用情况，提倡在非必要时使用字符串常量以减少内存开销。作者还提到可以设置阈值进行简易的分配行为跟踪。

摘要由CSDN通过智能技术生成

需求

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

代码

很简单，直接重写 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 == 1
	std::string s = "Test String."; 
	Print(s.c_str());
#else
	const char* c = "Test Const Char.";
	Print(c);
#endif
	std::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 就能做一个很简易的分配行为跟踪。