Clang LibTooling：如何让PPCallbacks仅处理主源文件的宏定义

可以通过Clang的SourceManager判断宏定义所在文件是否为主源文件，从而在MacroDefined回调中跳过头文件内的宏定义，具体实现如下：

• 在自定义的PPCallbacks子类中，持有SourceManager的引用（也可在回调中通过Preprocessor实例获取）
• 在MacroDefined回调内，获取宏定义的SourceLocation，通过SourceManager判断该位置是否属于主源文件：
  1. 调用SourceManager::isInMainFile(Location)，若返回false，直接跳过当前宏的处理
  2. 也可通过SourceManager::getFileID(Location)获取文件ID，与getSourceManager().getMainFileID()对比，不相等则跳过

示例代码片段：

class MyPPCallbacks : public PPCallbacks {
private:
  SourceManager &SM;

public:
  MyPPCallbacks(SourceManager &SM) : SM(SM) {}

  void MacroDefined(const Token &MacroNameTok, const MacroDirective *MD) override {
    SourceLocation Loc = MacroNameTok.getLocation();
    
    // 跳过非主源文件中的宏定义
    if (!Loc.isValid() || !SM.isInMainFile(Loc)) {
      return;
    }

    // 处理主源文件内的宏，比如提取宏名称
    StringRef MacroName = MacroNameTok.getIdentifierInfo()->getName();
    // 后续业务逻辑...
  }
};

注意事项：

• 先通过Loc.isValid()检查位置有效性，避免无效位置导致的异常
• 若编译多个主源文件，需同步更新主文件ID的判断逻辑
• 该方式会过滤所有#include引入的文件（系统头文件、自定义头文件均包含在内），符合排除头文件宏的需求

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