UseAnyOfAllOfCheck.cpp
3.62 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
//===--- UseAnyOfAllOfCheck.cpp - clang-tidy-------------------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#include "UseAnyOfAllOfCheck.h"
#include "clang/AST/ASTContext.h"
#include "clang/ASTMatchers/ASTMatchFinder.h"
#include "clang/Analysis/Analyses/ExprMutationAnalyzer.h"
#include "clang/Frontend/CompilerInstance.h"
using namespace clang::ast_matchers;
namespace clang {
namespace {
/// Matches a Stmt whose parent is a CompoundStmt, and which is directly
/// followed by a Stmt matching the inner matcher.
AST_MATCHER_P(Stmt, nextStmt, ast_matchers::internal::Matcher<Stmt>,
InnerMatcher) {
DynTypedNodeList Parents = Finder->getASTContext().getParents(Node);
if (Parents.size() != 1)
return false;
auto *C = Parents[0].get<CompoundStmt>();
if (!C)
return false;
const auto *I = llvm::find(C->body(), &Node);
assert(I != C->body_end() && "C is parent of Node");
if (++I == C->body_end())
return false; // Node is last statement.
return InnerMatcher.matches(**I, Finder, Builder);
}
} // namespace
namespace tidy {
namespace readability {
void UseAnyOfAllOfCheck::registerMatchers(MatchFinder *Finder) {
auto returns = [](bool V) {
return returnStmt(hasReturnValue(cxxBoolLiteral(equals(V))));
};
auto returnsButNotTrue =
returnStmt(hasReturnValue(unless(cxxBoolLiteral(equals(true)))));
auto returnsButNotFalse =
returnStmt(hasReturnValue(unless(cxxBoolLiteral(equals(false)))));
Finder->addMatcher(
cxxForRangeStmt(
nextStmt(returns(false).bind("final_return")),
hasBody(allOf(hasDescendant(returns(true)),
unless(anyOf(hasDescendant(breakStmt()),
hasDescendant(gotoStmt()),
hasDescendant(returnsButNotTrue))))))
.bind("any_of_loop"),
this);
Finder->addMatcher(
cxxForRangeStmt(
nextStmt(returns(true).bind("final_return")),
hasBody(allOf(hasDescendant(returns(false)),
unless(anyOf(hasDescendant(breakStmt()),
hasDescendant(gotoStmt()),
hasDescendant(returnsButNotFalse))))))
.bind("all_of_loop"),
this);
}
static bool isViableLoop(const CXXForRangeStmt &S, ASTContext &Context) {
ExprMutationAnalyzer Mutations(*S.getBody(), Context);
if (Mutations.isMutated(S.getLoopVariable()))
return false;
const auto Matches =
match(findAll(declRefExpr().bind("decl_ref")), *S.getBody(), Context);
return llvm::none_of(Matches, [&Mutations](auto &DeclRef) {
// TODO: allow modifications of loop-local variables
return Mutations.isMutated(
DeclRef.template getNodeAs<DeclRefExpr>("decl_ref")->getDecl());
});
}
void UseAnyOfAllOfCheck::check(const MatchFinder::MatchResult &Result) {
StringRef Ranges = getLangOpts().CPlusPlus20 ? "::ranges" : "";
if (const auto *S = Result.Nodes.getNodeAs<CXXForRangeStmt>("any_of_loop")) {
if (!isViableLoop(*S, *Result.Context))
return;
diag(S->getForLoc(), "replace loop by 'std%0::any_of()'") << Ranges;
} else if (const auto *S =
Result.Nodes.getNodeAs<CXXForRangeStmt>("all_of_loop")) {
if (!isViableLoop(*S, *Result.Context))
return;
diag(S->getForLoc(), "replace loop by 'std%0::all_of()'") << Ranges;
}
}
} // namespace readability
} // namespace tidy
} // namespace clang