如何验证C++20协程的编译器堆分配消除？

一、协程内存分配的核心逻辑 #

协程挂起时必须保存执行状态，这块内存默认在堆上，但编译器会在满足条件时做栈分配优化——只要能确定协程生命周期不会超出调用者的栈帧，就会把状态放到栈上，绕开堆分配。需要保存的状态包括：

• 协程控制块（CCB）：存储协程的运行状态、恢复/销毁函数指针等元数据
• 承诺对象（Promise Object）：和协程返回值绑定的对象，负责管理协程的结果或异常
• 跨挂起点的局部变量：比如co_await前定义、挂起后还要用到的变量

二、通过汇编判断堆分配是否被消除 #

直接盯汇编里的堆操作指令就行：

• 堆分配特征：call __imp_operator new（MSVC）、call malloc或call _Znwm（GCC/Clang）
• 堆释放特征：call __imp_operator delete（MSVC）、call free或call _ZdlPv（GCC/Clang）

拿Lewis Baker的示例来说，开启O2优化后，如果协程是同步完成（比如co_await后立即恢复，没有跨函数的挂起），编译器会直接把协程状态放到调用者栈上，这时汇编里不会出现上述堆操作指令，说明堆分配被消除了。

如果能看到这些堆调用，就说明堆分配没被优化掉。

三、未消除堆分配时，查看内存内容 #

堆分配的内存块里按顺序（具体顺序由编译器决定）存着这些内容：

1. 协程控制块：包含协程的状态标记（是否完成、是否挂起）、指向协程入口的指针、承诺对象的指针，还有resume()/destroy()的函数指针
2. 承诺对象：存储协程的返回值、异常信息（如果有），以及和返回对象（比如task<T>）交互的逻辑数据
3. 局部变量：所有需要在挂起后保留的变量，比如非平凡类型、或者跨越挂起点的基础类型变量

验证方法：

• 调试器直接查：在GDB/VS调试器里，拿到堆分配的内存地址，用内存查看功能（比如GDB的x/32xw），结合打印的局部变量、承诺对象的地址，对比就能定位各部分的位置
• 自定义分配器：重载协程的operator new/operator delete，打印分配的地址和大小，同时在协程里打印局部变量、承诺对象的地址，看这些地址是否落在堆分配的内存区间内：

  #include <cstdio>
  #include <coroutine>
  
  struct task {
      struct promise_type {
          task get_return_object() { return {}; }
          std::suspend_never initial_suspend() { return {}; }
          std::suspend_never final_suspend() noexcept { return {}; }
          void return_void() {}
          void unhandled_exception() {}
      };
  };
  
  void* operator new(std::size_t size) {
      void* ptr = malloc(size);
      printf("Allocated %zu bytes at %p\n", size, ptr);
      return ptr;
  }
  
  void operator delete(void* ptr) noexcept {
      printf("Freed at %p\n", ptr);
      free(ptr);
  }
  
  task my_coroutine() {
      int val = 0x12345678;
      printf("Local val at %p\n", &val);
      co_return;
  }
  
  int main() {
      my_coroutine();
      return 0;
  }
  

  运行后如果&val在分配的堆地址范围内，就说明这个局部变量存在堆上的协程状态里。

四、辅助分析的小技巧 #

• 编译器诊断选项：Clang用-Wcoroutine-missing-unwind，GCC用-fcoroutines-diagnostics，能输出协程状态布局的额外信息
• 强制禁用优化：编译时加-O0，能更清晰看到堆分配的汇编代码，方便对比优化前后的差异

内容的提问来源于stack exchange，提问作者Mark Wallace