LLVM-tutor 学习笔记4: StaticCallCounter

跳跃的芋头人

已于 2024-02-22 20:48:27 修改

阅读量376

点赞数 6

CC 4.0 BY-SA版权

文章标签：学习笔记

于 2024-02-20 10:35:33 首次发布

本文链接：https://blog.csdn.net/qq_43688320/article/details/136184234

这篇文章详细介绍了如何创建一个基于LLVM的analysispass，StaticCallCounter，用于统计模块中函数调用次数。它通过遍历IR模块，利用动态分派技术检测call指令，存储并累加每个函数的调用计数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Basic

这是一个analysis pass, 在程序的编译器静态分析每个函数的调用次数。

与之前一样，为了输出 analysis pass 的结果，我们需要一个专门输出结果的 printing pass。

由于 printing pass 和 pass manger 注册和之前类似，本文不再描述。

struct StaticCallCounter : public llvm::AnalysisInfoMixin<StaticCallCounter>

因为要统计各个函数之间的相互调用，这次选用了 moudle 粒度的 pass

runOnModule

主要逻辑是一步步地从 Module 遍历到每一条指令，再检查每条指令是否是函数调用，统计函数调用的信息。

在判断一个指令是否是 call 指令时，使用了动态分派的技巧。

如果分派后的结果为 nullptr 则说明该指令不是 call 类指令。


StaticCallCounter::Result StaticCallCounter::runOnModule(llvm::Module &M) {
  llvm::MapVector<const llvm::Function *, unsigned> Res;
  for (auto &Func : M) {
    for (auto &BB : Func) {
      for (auto &Ins : BB) {
        auto *CB = dyn_cast<CallBase>(&Ins);
        if (nullptr == CB) {
          continue;
        }

        auto DirectInvoc = CB->getCalledFunction();
        if (nullptr == DirectInvoc) {
          continue;
        }

        auto CallCount = Res.find(DirectInvoc);
        if (Res.end() == CallCount) {
          CallCount = Res.insert(std::make_pair(DirectInvoc, 0)).first;
        }
        ++CallCount->second;
      }
    }
  }
  return Res;
}