clang-tools  7.0.0
TwineLocalCheck.cpp
Go to the documentation of this file.
1 //===--- TwineLocalCheck.cpp - clang-tidy ---------------------------------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "TwineLocalCheck.h"
11 #include "clang/AST/ASTContext.h"
12 #include "clang/ASTMatchers/ASTMatchers.h"
13 #include "clang/Lex/Lexer.h"
14 
15 using namespace clang::ast_matchers;
16 
17 namespace clang {
18 namespace tidy {
19 namespace llvm {
20 
21 void TwineLocalCheck::registerMatchers(MatchFinder *Finder) {
22  auto TwineType =
23  qualType(hasDeclaration(recordDecl(hasName("::llvm::Twine"))));
24  Finder->addMatcher(varDecl(hasType(TwineType)).bind("variable"), this);
25 }
26 
27 void TwineLocalCheck::check(const MatchFinder::MatchResult &Result) {
28  const auto *VD = Result.Nodes.getNodeAs<VarDecl>("variable");
29  auto Diag = diag(VD->getLocation(),
30  "twine variables are prone to use-after-free bugs");
31 
32  // If this VarDecl has an initializer try to fix it.
33  if (VD->hasInit()) {
34  // Peel away implicit constructors and casts so we can see the actual type
35  // of the initializer.
36  const Expr *C = VD->getInit()->IgnoreImplicit();
37 
38  while (isa<CXXConstructExpr>(C)) {
39  if (cast<CXXConstructExpr>(C)->getNumArgs() == 0)
40  break;
41  C = cast<CXXConstructExpr>(C)->getArg(0)->IgnoreParenImpCasts();
42  }
43 
44  SourceRange TypeRange =
45  VD->getTypeSourceInfo()->getTypeLoc().getSourceRange();
46 
47  // A real Twine, turn it into a std::string.
48  if (VD->getType()->getCanonicalTypeUnqualified() ==
49  C->getType()->getCanonicalTypeUnqualified()) {
50  SourceLocation EndLoc = Lexer::getLocForEndOfToken(
51  VD->getInit()->getLocEnd(), 0, *Result.SourceManager, getLangOpts());
52  Diag << FixItHint::CreateReplacement(TypeRange, "std::string")
53  << FixItHint::CreateInsertion(VD->getInit()->getLocStart(), "(")
54  << FixItHint::CreateInsertion(EndLoc, ").str()");
55  } else {
56  // Just an implicit conversion. Insert the real type.
57  Diag << FixItHint::CreateReplacement(
58  TypeRange,
59  C->getType().getAsString(Result.Context->getPrintingPolicy()));
60  }
61  }
62 }
63 
64 } // namespace llvm
65 } // namespace tidy
66 } // namespace clang
Some operations such as code completion produce a set of candidates.
===– Representation.cpp - ClangDoc Representation --------—*- C++ -*-===//