使用Clang libTooling重写嵌套三元表达式的异常问题排查

Clang的Rewriter基于原始源码缓冲区（SourceBuffer）工作，RecursiveASTVisitor默认的深度优先遍历会先处理内层三元表达式，但如果直接在遍历回调中修改Rewriter，外层ConditionalOperator节点的SourceRange仍指向原始源码位置。处理外层节点时若直接获取整个范围的原始文本进行替换，会覆盖内层已经完成的修改，导致内层重写效果丢失。

1. 收集节点后按深度逆序处理 #

不在AST遍历回调中直接修改Rewriter，先遍历AST收集所有ConditionalOperator节点，再按**嵌套深度从大到小（内层先处理）**排序后批量修改。这样处理外层节点时，内层的修改已经生效，不会被覆盖。

示例代码片段：

#include <vector>
#include <algorithm>
#include "clang/AST/Expr.h"
#include "clang/Rewrite/Core/Rewriter.h"

std::vector<clang::ConditionalOperator*> CondOps;

// 遍历AST时收集所有三元表达式节点
bool VisitConditionalOperator(clang::ConditionalOperator *CO) {
  CondOps.push_back(CO);
  return true;
}

// 自定义排序逻辑：按节点嵌套深度降序
bool compareCondOpDepth(const clang::ConditionalOperator* a, const clang::ConditionalOperator* b) {
  // 手动计算节点深度：遍历父节点直到根
  auto getDepth = [](const clang::Stmt* stmt) {
    int depth = 0;
    while (stmt->getParent()) {
      depth++;
      stmt = stmt->getParent();
    }
    return depth;
  };
  return getDepth(a) > getDepth(b);
}

// 遍历完成后统一处理重写
void processRewrites(clang::Rewriter &Rewrite) {
  std::sort(CondOps.begin(), CondOps.end(), compareCondOpDepth);
  for (auto *CO : CondOps) {
    // 获取各分支的已重写文本（而非原始文本）
    std::string condText = Rewrite.getRewrittenText(CO->getCond()->getSourceRange());
    std::string trueText = Rewrite.getRewrittenText(CO->getTrueExpr()->getSourceRange());
    std::string falseText = Rewrite.getRewrittenText(CO->getFalseExpr()->getSourceRange());
    
    // 构造带注释的新表达式（按你的需求调整格式）
    std::string newExpr = "/* Cond: */ " + condText + " ? /* True: */ " + trueText + " : /* False: */ " + falseText;
    
    // 替换当前三元表达式的范围
    Rewrite.ReplaceText(CO->getSourceRange(), newExpr);
  }
}

2. 处理外层节点时拆分获取已重写分支文本 #

若必须在遍历回调中实时处理，处理外层ConditionalOperator时，不能直接获取整个节点的原始文本，要分别提取条件、true分支、false分支的已重写文本（用Rewriter::getRewrittenText替代getSourceText），再拼接成新表达式替换。

关键修正示例：

// 错误：直接获取原始文本，会覆盖内层修改
// std::string originalText = Rewrite.getSourceText(CO->getSourceRange());

// 正确：获取各分支的已重写文本
std::string cond = Rewrite.getRewrittenText(CO->getCond()->getSourceRange());
std::string trueExpr = Rewrite.getRewrittenText(CO->getTrueExpr()->getSourceRange());
std::string falseExpr = Rewrite.getRewrittenText(CO->getFalseExpr()->getSourceRange());

// 按需求构造新文本后替换
std::string newText = /* 你的格式化逻辑 */;
Rewrite.ReplaceText(CO->getSourceRange(), newText);

3. 精确计算SourceRange #

部分场景下，AST节点的默认getSourceRange可能包含多余空白或语法符号，可通过Lexer精确计算表达式起止位置，避免替换时出错：

#include "clang/Lex/Lexer.h"

clang::SourceLocation start = CO->getBeginLoc();
clang::SourceLocation end = clang::Lexer::getLocForEndOfToken(
    CO->getEndLoc(), 0, *Rewrite.getSourceManager(), LangOptions()
);
Rewrite.ReplaceText(clang::SourceRange(start, end), newText);

用clang-query确认节点嵌套关系和范围：

clang-query> match ConditionalOperator{hasParent(ConditionalOperator)} dump()

该命令会匹配所有外层三元表达式，通过dump()查看节点的SourceRange和子节点结构，确保处理顺序与范围计算准确。

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