diff --git a/clang/lib/ExtractAPI/ExtractAPIConsumer.cpp b/clang/lib/ExtractAPI/ExtractAPIConsumer.cpp --- a/clang/lib/ExtractAPI/ExtractAPIConsumer.cpp +++ b/clang/lib/ExtractAPI/ExtractAPIConsumer.cpp @@ -649,8 +649,9 @@ class MacroCallback : public PPCallbacks { public: - MacroCallback(const SourceManager &SM, LocationFileChecker &LCF, APISet &API) - : SM(SM), LCF(LCF), API(API) {} + MacroCallback(const SourceManager &SM, LocationFileChecker &LCF, APISet &API, + Preprocessor &PP) + : SM(SM), LCF(LCF), API(API), PP(PP) {} void MacroDefined(const Token &MacroNameToken, const MacroDirective *MD) override { @@ -677,9 +678,9 @@ if (!Undef) return; - llvm::erase_if(PendingMacros, [&MD](const PendingMacro &PM) { - return MD.getMacroInfo()->getDefinitionLoc() == - PM.MD->getMacroInfo()->getDefinitionLoc(); + llvm::erase_if(PendingMacros, [&MD, this](const PendingMacro &PM) { + return MD.getMacroInfo()->isIdenticalTo(*PM.MD->getMacroInfo(), PP, + /*Syntactically*/ false); }); } @@ -719,6 +720,7 @@ const SourceManager &SM; LocationFileChecker &LCF; APISet &API; + Preprocessor &PP; llvm::SmallVector PendingMacros; }; @@ -741,9 +743,8 @@ auto LCF = std::make_unique(CI.getSourceManager(), KnownInputFiles); - // Register preprocessor callbacks that will add macro definitions to API. - CI.getPreprocessor().addPPCallbacks( - std::make_unique(CI.getSourceManager(), *LCF, *API)); + CI.getPreprocessor().addPPCallbacks(std::make_unique( + CI.getSourceManager(), *LCF, *API, CI.getPreprocessor())); return std::make_unique(CI.getASTContext(), std::move(LCF), *API);