10#include "clang/AST/ASTContext.h"
11#include "clang/ASTMatchers/ASTMatchFinder.h"
12#include "clang/Lex/Lexer.h"
19 MatchFinder *Finder) {
20 const auto StrLenFunc = functionDecl(hasAnyName(
21 "::strlen",
"::std::strlen",
"::strnlen",
"::std::strnlen",
"::strnlen_s",
22 "::std::strnlen_s",
"::wcslen",
"::std::wcslen",
"::wcsnlen",
23 "::std::wcsnlen",
"::wcsnlen_s",
"std::wcsnlen_s"));
26 callExpr(callee(StrLenFunc),
27 hasAnyArgument(ignoringImpCasts(
30 hasRHS(ignoringParenImpCasts(integerLiteral(equals(1)))))
34 const auto BadArg = anyOf(
35 allOf(unless(binaryOperator(
36 hasOperatorName(
"+"), hasLHS(BadUse),
37 hasRHS(ignoringParenImpCasts(integerLiteral(equals(1)))))),
38 hasDescendant(BadUse)),
41 const auto Alloc0Func = functionDecl(
42 hasAnyName(
"::malloc",
"std::malloc",
"::alloca",
"std::alloca"));
43 const auto Alloc1Func = functionDecl(
44 hasAnyName(
"::calloc",
"std::calloc",
"::realloc",
"std::realloc"));
46 const auto Alloc0FuncPtr =
47 varDecl(hasType(isConstQualified()),
48 hasInitializer(ignoringParenImpCasts(
49 declRefExpr(hasDeclaration(Alloc0Func)))));
50 const auto Alloc1FuncPtr =
51 varDecl(hasType(isConstQualified()),
52 hasInitializer(ignoringParenImpCasts(
53 declRefExpr(hasDeclaration(Alloc1Func)))));
56 traverse(TK_AsIs, callExpr(callee(decl(anyOf(Alloc0Func, Alloc0FuncPtr))),
57 hasArgument(0, BadArg))
61 traverse(TK_AsIs, callExpr(callee(decl(anyOf(Alloc1Func, Alloc1FuncPtr))),
62 hasArgument(1, BadArg))
67 cxxNewExpr(isArray(), hasArraySize(BadArg)).bind(
"Alloc")),
72 const MatchFinder::MatchResult &Result) {
73 const Expr *Alloc = Result.Nodes.getNodeAs<CallExpr>(
"Alloc");
75 Alloc = Result.Nodes.getNodeAs<CXXNewExpr>(
"Alloc");
76 assert(Alloc &&
"Matched node bound by 'Alloc' should be either 'CallExpr'"
79 const auto *StrLen = Result.Nodes.getNodeAs<CallExpr>(
"StrLen");
80 const auto *BinOp = Result.Nodes.getNodeAs<BinaryOperator>(
"BinOp");
82 const StringRef StrLenText = Lexer::getSourceText(
83 CharSourceRange::getTokenRange(StrLen->getSourceRange()),
85 const StringRef Arg0Text = Lexer::getSourceText(
86 CharSourceRange::getTokenRange(StrLen->getArg(0)->getSourceRange()),
88 const StringRef StrLenBegin = StrLenText.substr(0, StrLenText.find(Arg0Text));
89 const StringRef StrLenEnd = StrLenText.substr(
90 StrLenText.find(Arg0Text) + Arg0Text.size(), StrLenText.size());
92 const StringRef LHSText = Lexer::getSourceText(
93 CharSourceRange::getTokenRange(BinOp->getLHS()->getSourceRange()),
95 const StringRef RHSText = Lexer::getSourceText(
96 CharSourceRange::getTokenRange(BinOp->getRHS()->getSourceRange()),
99 auto Hint = FixItHint::CreateReplacement(
100 StrLen->getSourceRange(),
101 (StrLenBegin + LHSText + StrLenEnd +
" + " + RHSText).str());
103 diag(Alloc->getBeginLoc(),
104 "addition operator is applied to the argument of %0 instead of its "
106 << StrLen->getDirectCallee()->getName() << Hint;
DiagnosticBuilder diag(SourceLocation Loc, StringRef Description, DiagnosticIDs::Level Level=DiagnosticIDs::Warning)
Add a diagnostic with the check's name.
const LangOptions & getLangOpts() const
Returns the language options from the context.
void registerMatchers(ast_matchers::MatchFinder *Finder) override
Override this to register AST matchers with Finder.
void check(const ast_matchers::MatchFinder::MatchResult &Result) override
ClangTidyChecks that register ASTMatchers should do the actual work in here.