如何定位Rust中共享Arc<Vec<u8>>的其他持有者以释放底层数据

标准库默认提供的Arc没有内置引用溯源能力，出于运行时性能考虑，release版本默认不会保留任何克隆、释放的调用栈记录，你可以通过以下几种方案在调试阶段定位所有未释放的引用：

• 自定义带追踪能力的Arc包装 #

  这是最通用、不依赖操作系统工具的方案：你可以自己封装一层TrackedArc<T>包装标准库Arc<T>，在实例新建、克隆时抓取当前调用栈，将栈信息和实例唯一标识关联存入全局并发安全的映射表中；实例被drop时，从映射表中删除对应记录。
  你可以借助调用栈抓取类的crate实现栈捕获逻辑，调试阶段开启追踪，正式发布时通过feature开关关闭所有追踪逻辑，不会带来额外生产性能开销。
  核心逻辑示例：

  use std::sync::{Arc, Mutex};
  use backtrace::Backtrace;
  use std::collections::HashMap;
  
  // 全局追踪表：按Arc指向的堆内存地址聚合所有存活实例的创建栈
  static ARC_LIVE_REFS: Mutex<HashMap<usize, Vec<Backtrace>>> = Mutex::new(HashMap::new());
  
  #[derive(Debug, Clone)]
  struct TrackedArc<T> {
      inner: Arc<T>,
  }
  
  impl<T> TrackedArc<T> {
      pub fn new(val: T) -> Self {
          let inner = Arc::new(val);
          let heap_addr = Arc::as_ptr(&inner) as usize;
          ARC_LIVE_REFS.lock().unwrap()
              .entry(heap_addr)
              .or_default()
              .push(Backtrace::new());
          Self { inner }
      }
  
      // 按需转发需要调用的标准库Arc方法即可
      pub fn strong_count(&self) -> usize {
          Arc::strong_count(&self.inner)
      }
  }
  
  // 实现Drop时移除当前实例对应的栈记录
  impl<T> Drop for TrackedArc<T> {
      fn drop(&mut self) {
          // 实现时注意给每个实例加唯一ID，避免同地址下多个实例的栈记录误删
      }
  }
  

  当你观测到目标Arc<Vec<u8>>的强引用计数不符合预期时，直接查询全局追踪表中对应堆地址的所有调用栈，就能定位到每一个存活实例的创建位置。

• 基于动态探针无侵入追踪 #

  如果你不想修改业务代码，可以在Linux环境下使用bpftrace、dtrace这类动态追踪工具，给标准库Arc的clone、drop逻辑挂用户态探针：

  1. 编译程序时保留调试符号
  2. 给Arc::clone挂探针，每次克隆触发时记录返回的Arc指针值、当前用户态调用栈
  3. 给Arc的释放逻辑（drop_slow路径）挂探针，每次引用释放时删除指针对应的记录
     当程序运行到强引用计数为2的状态时，导出探针收集的所有存活指针对应的调用栈，就能直接定位到剩余引用的持有位置。
     参考探针示例：

  uprobe:/your/program/path:alloc::sync::Arc<*>::clone
  {
      $arc_ptr = reg("ax"); // x86_64架构下函数返回值存于rax寄存器
      @live_arcs[$arc_ptr] = ustack();
  }
  
  uprobe:/your/program/path:alloc::sync::Arc<*>::drop_slow
  {
      $arc_ptr = arg0;
      delete(@live_arcs[$arc_ptr]);
  }
  

• 注意事项 #

  • 所有Arc追踪逻辑都只适合调试阶段使用，抓取调用栈、维护全局映射表都会带来明显的性能开销，禁止在生产环境开启
  • 多线程场景下Arc::strong_count返回的是瞬时快照值，读取时可能有其他线程正在执行克隆/释放操作，仅可作为调试参考，不要用于业务逻辑判断
  • 如果你不想自行实现追踪包装，也可以使用社区提供的现成带引用追踪功能的Arc替代实现crate，调试完成后替换回标准库Arc即可

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