================
@@ -442,21 +442,94 @@ class LifetimeSafetySemaHelperImpl : public
LifetimeSafetySemaHelper {
}
private:
+ struct LifetimeBoundMacroCache {
+ bool IsBuilt = false;
+ SmallVector<const IdentifierInfo *> Candidates;
+ };
+
+ void buildLifetimeBoundMacroCache(LifetimeBoundMacroCache &Cache,
+ ArrayRef<TokenValue> Tokens) {
+ if (Cache.IsBuilt)
+ return;
+
+ const Preprocessor &PP = S.getPreprocessor();
+ // Collect macro names that were ever defined as a lifetimebound attribute.
+ for (const auto &M : PP.macros()) {
+ const IdentifierInfo *II = M.first;
+ const MacroDirective *MD = PP.getLocalMacroDirectiveHistory(II);
+ if (!MD)
+ continue;
+
+ // Include earlier matching definitions to handle redefinitions.
+ for (MacroDirective::DefInfo Def = MD->getDefinition(); Def;
+ Def = Def.getPreviousDefinition()) {
+ const MacroInfo *MI = Def.getMacroInfo();
+ if (MI->isObjectLike() && Tokens.size() == MI->getNumTokens() &&
+ std::equal(Tokens.begin(), Tokens.end(), MI->tokens_begin())) {
+ Cache.Candidates.push_back(II);
+ break;
+ }
+ }
+ }
+ Cache.IsBuilt = true;
+ }
+
+ StringRef getLastCachedMacroWithSpelling(SourceLocation Loc,
+ llvm::ArrayRef<TokenValue> Tokens,
+ LifetimeBoundMacroCache &Cache) {
+ if (Loc.isInvalid())
+ return {};
+
+ buildLifetimeBoundMacroCache(Cache, Tokens);
+
+ const Preprocessor &PP = S.getPreprocessor();
+ const SourceManager &SM = S.getSourceManager();
+ SourceLocation BestLocation;
+ StringRef BestSpelling;
+ for (const IdentifierInfo *II : Cache.Candidates) {
+ const MacroDirective *MD = PP.getLocalMacroDirectiveHistory(II);
+ if (!MD)
+ continue;
----------------
NeKon69 wrote:
It's not that big of a deal really.
https://github.com/llvm/llvm-project/pull/205250
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
