================ @@ -0,0 +1,357 @@ +//===----------------------------------------------------------------------===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#include "MacroConditionCheck.h" +#include "clang/Lex/Lexer.h" +#include "clang/Lex/MacroInfo.h" +#include "clang/Lex/PPCallbacks.h" +#include "clang/Lex/Preprocessor.h" +#include "llvm/ADT/DenseSet.h" +#include "llvm/ADT/STLExtras.h" +#include "llvm/ADT/SmallVector.h" +#include "llvm/ADT/StringSet.h" +#include <memory> +#include <string> +#include <utility> + +namespace clang::tidy::bugprone { + +namespace { +class MacroConditionCallbacks : public PPCallbacks { +public: + MacroConditionCallbacks(MacroConditionCheck *Check, const SourceManager &SM, + Preprocessor &PP) + : Check(Check), SM(SM), PP(PP) {} + + void If(SourceLocation Loc, SourceRange ConditionRange, + ConditionValueKind ConditionValue) override; + void Ifdef(SourceLocation Loc, const Token &MacroNameTok, + const MacroDefinition &MD) override; + void Ifndef(SourceLocation Loc, const Token &MacroNameTok, + const MacroDefinition &MD) override; + void Elif(SourceLocation Loc, SourceRange ConditionRange, + ConditionValueKind ConditionValue, SourceLocation IfLoc) override; + void Elifdef(SourceLocation Loc, const Token &MacroNameTok, + const MacroDefinition &MD) override; + void Elifdef(SourceLocation Loc, SourceRange ConditionRange, + SourceLocation IfLoc) override; + void Elifndef(SourceLocation Loc, const Token &MacroNameTok, + const MacroDefinition &MD) override; + void Elifndef(SourceLocation Loc, SourceRange ConditionRange, + SourceLocation IfLoc) override; + void Else(SourceLocation Loc, SourceLocation IfLoc) override; + void Endif(SourceLocation Loc, SourceLocation IfLoc) override; + +private: + struct MacroReference { + std::string Name; + SourceLocation Loc; + }; + + struct ConditionReferences { + SmallVector<MacroReference, 2> Definition; + SmallVector<MacroReference, 2> Value; + }; + + struct DefinitionCheck { + std::string Name; + SourceLocation DefinitionLoc; + SourceLocation CheckLoc; + bool ValueTested = false; + }; + + struct ConditionalBranch { + SmallVector<DefinitionCheck, 2> Checks; + llvm::StringSet<> DefinitionTests; + }; + + ConditionReferences referencesInCondition(SourceRange ConditionRange) const; + MacroReference referenceFromRange(SourceRange Range, + SourceLocation Loc) const; + bool isIgnoredIdentifier(StringRef Name) const; + void startCondition(const ConditionReferences &References); + void startDefinitionCondition(StringRef Name, SourceLocation Loc); + void nextBranch(const ConditionReferences &References = {}); + void processReferences(const ConditionReferences &References); + void finishBranch(ConditionalBranch &Branch); + void checkDefinitionReference(const MacroReference &Reference, + ConditionalBranch &Branch); + void checkValueReference(const MacroReference &Reference); + bool isDefinitionTestActive(StringRef Name) const; + + SmallVector<ConditionalBranch, 8> Conditions; + llvm::DenseSet<unsigned> DiagnosedDefinitions; ---------------- zwuis wrote:
Can we use `DenseSet<SourceLocation>`? https://github.com/llvm/llvm-project/pull/210768 _______________________________________________ cfe-commits mailing list [email protected] https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
