Method insertMacroNode

lib/astirmap/CalledFunctionMacroMapper.cpp:9–73

using namespace ftg;

void CalledFunctionMacroMapper::insertMacroNode(clang::Stmt &S, ASTUnit &U) {

  if (!isa<Expr>(S))
    return;

  auto &E = *dyn_cast<Expr>(&S);
  if (!util::isCallRelatedExpr(E))
    return;

  auto *FuncDecl = util::getFunctionDecl(E);
  if (!FuncDecl)
    return;

  for (auto &MangledName : util::getMangledNames(*FuncDecl)) {
    if (MangledName.empty())
      continue;

    auto BaseLoc = util::getDebugLoc(E);
    auto Loc = LocIndex(U.getSourceManager(), BaseLoc);
    auto Key =
        DBKey(Loc.getPath(), Loc.getLine(), Loc.getColumn(), MangledName);
    if (DupKey.find(Key) != DupKey.end())
      continue;

    auto CN =
        std::make_unique<ASTNode>(ASTNode::CALL, DynTypedNode::create(E), U);
    assert(CN && "Unexpected Program State");

    auto CMInsert = CallMap.emplace(Key, std::move(CN));
    if (!CMInsert.second) {
      auto &SrcManager = U.getSourceManager();

      // if it is not macro arg expansion, then it is duplicates.
      if (SrcManager.isMacroBodyExpansion(E.getExprLoc())) {
        DupKey.insert(Key);
        continue;
      }

      // NOTE: It is not duplicated key if it has same macro caller location if
      //       it is macro arg expansion
      auto &PrevCN = CMInsert.first->second;
      auto NewCN =
          std::make_unique<ASTNode>(ASTNode::CALL, DynTypedNode::create(E), U);
      assert((PrevCN && NewCN) && "Unexpected Program State");

      auto PrevLoc = util::getTopMacroCallerLoc(
          PrevCN->getNode(), PrevCN->getASTUnit().getSourceManager());
      auto NewLoc = util::getTopMacroCallerLoc(
          NewCN->getNode(), NewCN->getASTUnit().getSourceManager());

      auto PrevLI = LocIndex(PrevCN->getASTUnit().getSourceManager(), PrevLoc);
      auto NewLI = LocIndex(NewCN->getASTUnit().getSourceManager(), NewLoc);
      if (PrevLI != NewLI)
        DupKey.insert(Key);

      continue;
    }