contrib/llvm/tools/clang/lib/Analysis/AnalysisDeclContext.cpp

   1 //===- AnalysisDeclContext.cpp - Analysis context for Path Sens analysis --===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file defines AnalysisDeclContext, a class that manages the analysis
  10 // context data for path sensitive analysis.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "clang/Analysis/AnalysisDeclContext.h"
  15 #include "clang/AST/ASTContext.h"
  16 #include "clang/AST/Decl.h"
  17 #include "clang/AST/DeclBase.h"
  18 #include "clang/AST/DeclCXX.h"
  19 #include "clang/AST/DeclObjC.h"
  20 #include "clang/AST/DeclTemplate.h"
  21 #include "clang/AST/Expr.h"
  22 #include "clang/AST/LambdaCapture.h"
  23 #include "clang/AST/ParentMap.h"
  24 #include "clang/AST/PrettyPrinter.h"
  25 #include "clang/AST/Stmt.h"
  26 #include "clang/AST/StmtCXX.h"
  27 #include "clang/AST/StmtVisitor.h"
  28 #include "clang/Analysis/Analyses/CFGReachabilityAnalysis.h"
  29 #include "clang/Analysis/BodyFarm.h"
  30 #include "clang/Analysis/CFG.h"
  31 #include "clang/Analysis/CFGStmtMap.h"
  32 #include "clang/Analysis/Support/BumpVector.h"
  33 #include "clang/Basic/JsonSupport.h"
  34 #include "clang/Basic/LLVM.h"
  35 #include "clang/Basic/SourceLocation.h"
  36 #include "clang/Basic/SourceManager.h"
  37 #include "llvm/ADT/DenseMap.h"
  38 #include "llvm/ADT/FoldingSet.h"
  39 #include "llvm/ADT/STLExtras.h"
  40 #include "llvm/ADT/SmallPtrSet.h"
  41 #include "llvm/ADT/iterator_range.h"
  42 #include "llvm/Support/Allocator.h"
  43 #include "llvm/Support/Casting.h"
  44 #include "llvm/Support/Compiler.h"
  45 #include "llvm/Support/ErrorHandling.h"
  46 #include "llvm/Support/SaveAndRestore.h"
  47 #include "llvm/Support/raw_ostream.h"
  48 #include <cassert>
  49 #include <memory>
  50
  51 using namespace clang;
  52
  53 using ManagedAnalysisMap = llvm::DenseMap<const void *, ManagedAnalysis *>;
  54
  55 AnalysisDeclContext::AnalysisDeclContext(AnalysisDeclContextManager *Mgr,
  56                                          const Decl *d,
  57                                          const CFG::BuildOptions &buildOptions)
  58     : Manager(Mgr), D(d), cfgBuildOptions(buildOptions) {
  59   cfgBuildOptions.forcedBlkExprs = &forcedBlkExprs;
  60 }
  61
  62 AnalysisDeclContext::AnalysisDeclContext(AnalysisDeclContextManager *Mgr,
  63                                          const Decl *d)
  64     : Manager(Mgr), D(d) {
  65   cfgBuildOptions.forcedBlkExprs = &forcedBlkExprs;
  66 }
  67
  68 AnalysisDeclContextManager::AnalysisDeclContextManager(
  69     ASTContext &ASTCtx, bool useUnoptimizedCFG, bool addImplicitDtors,
  70     bool addInitializers, bool addTemporaryDtors, bool addLifetime,
  71     bool addLoopExit, bool addScopes, bool synthesizeBodies,
  72     bool addStaticInitBranch, bool addCXXNewAllocator,
  73     bool addRichCXXConstructors, bool markElidedCXXConstructors,
  74     bool addVirtualBaseBranches, CodeInjector *injector)
  75     : Injector(injector), FunctionBodyFarm(ASTCtx, injector),
  76       SynthesizeBodies(synthesizeBodies) {
  77   cfgBuildOptions.PruneTriviallyFalseEdges = !useUnoptimizedCFG;
  78   cfgBuildOptions.AddImplicitDtors = addImplicitDtors;
  79   cfgBuildOptions.AddInitializers = addInitializers;
  80   cfgBuildOptions.AddTemporaryDtors = addTemporaryDtors;
  81   cfgBuildOptions.AddLifetime = addLifetime;
  82   cfgBuildOptions.AddLoopExit = addLoopExit;
  83   cfgBuildOptions.AddScopes = addScopes;
  84   cfgBuildOptions.AddStaticInitBranches = addStaticInitBranch;
  85   cfgBuildOptions.AddCXXNewAllocator = addCXXNewAllocator;
  86   cfgBuildOptions.AddRichCXXConstructors = addRichCXXConstructors;
  87   cfgBuildOptions.MarkElidedCXXConstructors = markElidedCXXConstructors;
  88   cfgBuildOptions.AddVirtualBaseBranches = addVirtualBaseBranches;
  89 }
  90
  91 void AnalysisDeclContextManager::clear() { Contexts.clear(); }
  92
  93 Stmt *AnalysisDeclContext::getBody(bool &IsAutosynthesized) const {
  94   IsAutosynthesized = false;
  95   if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
  96     Stmt *Body = FD->getBody();
  97     if (auto *CoroBody = dyn_cast_or_null<CoroutineBodyStmt>(Body))
  98       Body = CoroBody->getBody();
  99     if (Manager && Manager->synthesizeBodies()) {
 100       Stmt *SynthesizedBody = Manager->getBodyFarm().getBody(FD);
 101       if (SynthesizedBody) {
 102         Body = SynthesizedBody;
 103         IsAutosynthesized = true;
 104       }
 105     }
 106     return Body;
 107   }
 108   else if (const auto *MD = dyn_cast<ObjCMethodDecl>(D)) {
 109     Stmt *Body = MD->getBody();
 110     if (Manager && Manager->synthesizeBodies()) {
 111       Stmt *SynthesizedBody = Manager->getBodyFarm().getBody(MD);
 112       if (SynthesizedBody) {
 113         Body = SynthesizedBody;
 114         IsAutosynthesized = true;
 115       }
 116     }
 117     return Body;
 118   } else if (const auto *BD = dyn_cast<BlockDecl>(D))
 119     return BD->getBody();
 120   else if (const auto *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
 121     return FunTmpl->getTemplatedDecl()->getBody();
 122
 123   llvm_unreachable("unknown code decl");
 124 }
 125
 126 Stmt *AnalysisDeclContext::getBody() const {
 127   bool Tmp;
 128   return getBody(Tmp);
 129 }
 130
 131 bool AnalysisDeclContext::isBodyAutosynthesized() const {
 132   bool Tmp;
 133   getBody(Tmp);
 134   return Tmp;
 135 }
 136
 137 bool AnalysisDeclContext::isBodyAutosynthesizedFromModelFile() const {
 138   bool Tmp;
 139   Stmt *Body = getBody(Tmp);
 140   return Tmp && Body->getBeginLoc().isValid();
 141 }
 142
 143 /// Returns true if \param VD is an Objective-C implicit 'self' parameter.
 144 static bool isSelfDecl(const VarDecl *VD) {
 145   return isa<ImplicitParamDecl>(VD) && VD->getName() == "self";
 146 }
 147
 148 const ImplicitParamDecl *AnalysisDeclContext::getSelfDecl() const {
 149   if (const auto *MD = dyn_cast<ObjCMethodDecl>(D))
 150     return MD->getSelfDecl();
 151   if (const auto *BD = dyn_cast<BlockDecl>(D)) {
 152     // See if 'self' was captured by the block.
 153     for (const auto &I : BD->captures()) {
 154       const VarDecl *VD = I.getVariable();
 155       if (isSelfDecl(VD))
 156         return dyn_cast<ImplicitParamDecl>(VD);
 157     }
 158   }
 159
 160   auto *CXXMethod = dyn_cast<CXXMethodDecl>(D);
 161   if (!CXXMethod)
 162     return nullptr;
 163
 164   const CXXRecordDecl *parent = CXXMethod->getParent();
 165   if (!parent->isLambda())
 166     return nullptr;
 167
 168   for (const auto &LC : parent->captures()) {
 169     if (!LC.capturesVariable())
 170       continue;
 171
 172     VarDecl *VD = LC.getCapturedVar();
 173     if (isSelfDecl(VD))
 174       return dyn_cast<ImplicitParamDecl>(VD);
 175   }
 176
 177   return nullptr;
 178 }
 179
 180 void AnalysisDeclContext::registerForcedBlockExpression(const Stmt *stmt) {
 181   if (!forcedBlkExprs)
 182     forcedBlkExprs = new CFG::BuildOptions::ForcedBlkExprs();
 183   // Default construct an entry for 'stmt'.
 184   if (const auto *e = dyn_cast<Expr>(stmt))
 185     stmt = e->IgnoreParens();
 186   (void) (*forcedBlkExprs)[stmt];
 187 }
 188
 189 const CFGBlock *
 190 AnalysisDeclContext::getBlockForRegisteredExpression(const Stmt *stmt) {
 191   assert(forcedBlkExprs);
 192   if (const auto *e = dyn_cast<Expr>(stmt))
 193     stmt = e->IgnoreParens();
 194   CFG::BuildOptions::ForcedBlkExprs::const_iterator itr =
 195     forcedBlkExprs->find(stmt);
 196   assert(itr != forcedBlkExprs->end());
 197   return itr->second;
 198 }
 199
 200 /// Add each synthetic statement in the CFG to the parent map, using the
 201 /// source statement's parent.
 202 static void addParentsForSyntheticStmts(const CFG *TheCFG, ParentMap &PM) {
 203   if (!TheCFG)
 204     return;
 205
 206   for (CFG::synthetic_stmt_iterator I = TheCFG->synthetic_stmt_begin(),
 207                                     E = TheCFG->synthetic_stmt_end();
 208        I != E; ++I) {
 209     PM.setParent(I->first, PM.getParent(I->second));
 210   }
 211 }
 212
 213 CFG *AnalysisDeclContext::getCFG() {
 214   if (!cfgBuildOptions.PruneTriviallyFalseEdges)
 215     return getUnoptimizedCFG();
 216
 217   if (!builtCFG) {
 218     cfg = CFG::buildCFG(D, getBody(), &D->getASTContext(), cfgBuildOptions);
 219     // Even when the cfg is not successfully built, we don't
 220     // want to try building it again.
 221     builtCFG = true;
 222
 223     if (PM)
 224       addParentsForSyntheticStmts(cfg.get(), *PM);
 225
 226     // The Observer should only observe one build of the CFG.
 227     getCFGBuildOptions().Observer = nullptr;
 228   }
 229   return cfg.get();
 230 }
 231
 232 CFG *AnalysisDeclContext::getUnoptimizedCFG() {
 233   if (!builtCompleteCFG) {
 234     SaveAndRestore<bool> NotPrune(cfgBuildOptions.PruneTriviallyFalseEdges,
 235                                   false);
 236     completeCFG =
 237         CFG::buildCFG(D, getBody(), &D->getASTContext(), cfgBuildOptions);
 238     // Even when the cfg is not successfully built, we don't
 239     // want to try building it again.
 240     builtCompleteCFG = true;
 241
 242     if (PM)
 243       addParentsForSyntheticStmts(completeCFG.get(), *PM);
 244
 245     // The Observer should only observe one build of the CFG.
 246     getCFGBuildOptions().Observer = nullptr;
 247   }
 248   return completeCFG.get();
 249 }
 250
 251 CFGStmtMap *AnalysisDeclContext::getCFGStmtMap() {
 252   if (cfgStmtMap)
 253     return cfgStmtMap.get();
 254
 255   if (CFG *c = getCFG()) {
 256     cfgStmtMap.reset(CFGStmtMap::Build(c, &getParentMap()));
 257     return cfgStmtMap.get();
 258   }
 259
 260   return nullptr;
 261 }
 262
 263 CFGReverseBlockReachabilityAnalysis *AnalysisDeclContext::getCFGReachablityAnalysis() {
 264   if (CFA)
 265     return CFA.get();
 266
 267   if (CFG *c = getCFG()) {
 268     CFA.reset(new CFGReverseBlockReachabilityAnalysis(*c));
 269     return CFA.get();
 270   }
 271
 272   return nullptr;
 273 }
 274
 275 void AnalysisDeclContext::dumpCFG(bool ShowColors) {
 276   getCFG()->dump(getASTContext().getLangOpts(), ShowColors);
 277 }
 278
 279 ParentMap &AnalysisDeclContext::getParentMap() {
 280   if (!PM) {
 281     PM.reset(new ParentMap(getBody()));
 282     if (const auto *C = dyn_cast<CXXConstructorDecl>(getDecl())) {
 283       for (const auto *I : C->inits()) {
 284         PM->addStmt(I->getInit());
 285       }
 286     }
 287     if (builtCFG)
 288       addParentsForSyntheticStmts(getCFG(), *PM);
 289     if (builtCompleteCFG)
 290       addParentsForSyntheticStmts(getUnoptimizedCFG(), *PM);
 291   }
 292   return *PM;
 293 }
 294
 295 AnalysisDeclContext *AnalysisDeclContextManager::getContext(const Decl *D) {
 296   if (const auto *FD = dyn_cast<FunctionDecl>(D)) {
 297     // Calling 'hasBody' replaces 'FD' in place with the FunctionDecl
 298     // that has the body.
 299     FD->hasBody(FD);
 300     D = FD;
 301   }
 302
 303   std::unique_ptr<AnalysisDeclContext> &AC = Contexts[D];
 304   if (!AC)
 305     AC = llvm::make_unique<AnalysisDeclContext>(this, D, cfgBuildOptions);
 306   return AC.get();
 307 }
 308
 309 BodyFarm &AnalysisDeclContextManager::getBodyFarm() { return FunctionBodyFarm; }
 310
 311 const StackFrameContext *
 312 AnalysisDeclContext::getStackFrame(LocationContext const *Parent, const Stmt *S,
 313                                const CFGBlock *Blk, unsigned Idx) {
 314   return getLocationContextManager().getStackFrame(this, Parent, S, Blk, Idx);
 315 }
 316
 317 const BlockInvocationContext *
 318 AnalysisDeclContext::getBlockInvocationContext(const LocationContext *parent,
 319                                                const BlockDecl *BD,
 320                                                const void *ContextData) {
 321   return getLocationContextManager().getBlockInvocationContext(this, parent,
 322                                                                BD, ContextData);
 323 }
 324
 325 bool AnalysisDeclContext::isInStdNamespace(const Decl *D) {
 326   const DeclContext *DC = D->getDeclContext()->getEnclosingNamespaceContext();
 327   const auto *ND = dyn_cast<NamespaceDecl>(DC);
 328   if (!ND)
 329     return false;
 330
 331   while (const DeclContext *Parent = ND->getParent()) {
 332     if (!isa<NamespaceDecl>(Parent))
 333       break;
 334     ND = cast<NamespaceDecl>(Parent);
 335   }
 336
 337   return ND->isStdNamespace();
 338 }
 339
 340 LocationContextManager &AnalysisDeclContext::getLocationContextManager() {
 341   assert(Manager &&
 342          "Cannot create LocationContexts without an AnalysisDeclContextManager!");
 343   return Manager->getLocationContextManager();
 344 }
 345
 346 //===----------------------------------------------------------------------===//
 347 // FoldingSet profiling.
 348 //===----------------------------------------------------------------------===//
 349
 350 void LocationContext::ProfileCommon(llvm::FoldingSetNodeID &ID,
 351                                     ContextKind ck,
 352                                     AnalysisDeclContext *ctx,
 353                                     const LocationContext *parent,
 354                                     const void *data) {
 355   ID.AddInteger(ck);
 356   ID.AddPointer(ctx);
 357   ID.AddPointer(parent);
 358   ID.AddPointer(data);
 359 }
 360
 361 void StackFrameContext::Profile(llvm::FoldingSetNodeID &ID) {
 362   Profile(ID, getAnalysisDeclContext(), getParent(), CallSite, Block, Index);
 363 }
 364
 365 void ScopeContext::Profile(llvm::FoldingSetNodeID &ID) {
 366   Profile(ID, getAnalysisDeclContext(), getParent(), Enter);
 367 }
 368
 369 void BlockInvocationContext::Profile(llvm::FoldingSetNodeID &ID) {
 370   Profile(ID, getAnalysisDeclContext(), getParent(), BD, ContextData);
 371 }
 372
 373 //===----------------------------------------------------------------------===//
 374 // LocationContext creation.
 375 //===----------------------------------------------------------------------===//
 376
 377 template <typename LOC, typename DATA>
 378 const LOC*
 379 LocationContextManager::getLocationContext(AnalysisDeclContext *ctx,
 380                                            const LocationContext *parent,
 381                                            const DATA *d) {
 382   llvm::FoldingSetNodeID ID;
 383   LOC::Profile(ID, ctx, parent, d);
 384   void *InsertPos;
 385
 386   LOC *L = cast_or_null<LOC>(Contexts.FindNodeOrInsertPos(ID, InsertPos));
 387
 388   if (!L) {
 389     L = new LOC(ctx, parent, d, ++NewID);
 390     Contexts.InsertNode(L, InsertPos);
 391   }
 392   return L;
 393 }
 394
 395 const StackFrameContext*
 396 LocationContextManager::getStackFrame(AnalysisDeclContext *ctx,
 397                                       const LocationContext *parent,
 398                                       const Stmt *s,
 399                                       const CFGBlock *blk, unsigned idx) {
 400   llvm::FoldingSetNodeID ID;
 401   StackFrameContext::Profile(ID, ctx, parent, s, blk, idx);
 402   void *InsertPos;
 403   auto *L =
 404    cast_or_null<StackFrameContext>(Contexts.FindNodeOrInsertPos(ID, InsertPos));
 405   if (!L) {
 406     L = new StackFrameContext(ctx, parent, s, blk, idx, ++NewID);
 407     Contexts.InsertNode(L, InsertPos);
 408   }
 409   return L;
 410 }
 411
 412 const ScopeContext *
 413 LocationContextManager::getScope(AnalysisDeclContext *ctx,
 414                                  const LocationContext *parent,
 415                                  const Stmt *s) {
 416   return getLocationContext<ScopeContext, Stmt>(ctx, parent, s);
 417 }
 418
 419 const BlockInvocationContext *
 420 LocationContextManager::getBlockInvocationContext(AnalysisDeclContext *ctx,
 421                                                   const LocationContext *parent,
 422                                                   const BlockDecl *BD,
 423                                                   const void *ContextData) {
 424   llvm::FoldingSetNodeID ID;
 425   BlockInvocationContext::Profile(ID, ctx, parent, BD, ContextData);
 426   void *InsertPos;
 427   auto *L =
 428     cast_or_null<BlockInvocationContext>(Contexts.FindNodeOrInsertPos(ID,
 429                                                                     InsertPos));
 430   if (!L) {
 431     L = new BlockInvocationContext(ctx, parent, BD, ContextData, ++NewID);
 432     Contexts.InsertNode(L, InsertPos);
 433   }
 434   return L;
 435 }
 436
 437 //===----------------------------------------------------------------------===//
 438 // LocationContext methods.
 439 //===----------------------------------------------------------------------===//
 440
 441 const StackFrameContext *LocationContext::getStackFrame() const {
 442   const LocationContext *LC = this;
 443   while (LC) {
 444     if (const auto *SFC = dyn_cast<StackFrameContext>(LC))
 445       return SFC;
 446     LC = LC->getParent();
 447   }
 448   return nullptr;
 449 }
 450
 451 bool LocationContext::inTopFrame() const {
 452   return getStackFrame()->inTopFrame();
 453 }
 454
 455 bool LocationContext::isParentOf(const LocationContext *LC) const {
 456   do {
 457     const LocationContext *Parent = LC->getParent();
 458     if (Parent == this)
 459       return true;
 460     else
 461       LC = Parent;
 462   } while (LC);
 463
 464   return false;
 465 }
 466
 467 static void printLocation(raw_ostream &Out, const SourceManager &SM,
 468                           SourceLocation Loc) {
 469   if (Loc.isFileID() && SM.isInMainFile(Loc))
 470     Out << SM.getExpansionLineNumber(Loc);
 471   else
 472     Loc.print(Out, SM);
 473 }
 474
 475 void LocationContext::dumpStack(raw_ostream &Out, const char *NL,
 476                                 std::function<void(const LocationContext *)>
 477                                     printMoreInfoPerContext) const {
 478   ASTContext &Ctx = getAnalysisDeclContext()->getASTContext();
 479   PrintingPolicy PP(Ctx.getLangOpts());
 480   PP.TerseOutput = 1;
 481
 482   const SourceManager &SM =
 483       getAnalysisDeclContext()->getASTContext().getSourceManager();
 484
 485   unsigned Frame = 0;
 486   for (const LocationContext *LCtx = this; LCtx; LCtx = LCtx->getParent()) {
 487     switch (LCtx->getKind()) {
 488     case StackFrame:
 489       Out << "\t#" << Frame << ' ';
 490       ++Frame;
 491       if (const auto *D = dyn_cast<NamedDecl>(LCtx->getDecl()))
 492         Out << "Calling " << D->getQualifiedNameAsString();
 493       else
 494         Out << "Calling anonymous code";
 495       if (const Stmt *S = cast<StackFrameContext>(LCtx)->getCallSite()) {
 496         Out << " at line ";
 497         printLocation(Out, SM, S->getBeginLoc());
 498       }
 499       break;
 500     case Scope:
 501       Out << "Entering scope";
 502       break;
 503     case Block:
 504       Out << "Invoking block";
 505       if (const Decl *D = cast<BlockInvocationContext>(LCtx)->getDecl()) {
 506         Out << " defined at line ";
 507         printLocation(Out, SM, D->getBeginLoc());
 508       }
 509       break;
 510     }
 511     Out << NL;
 512
 513     printMoreInfoPerContext(LCtx);
 514   }
 515 }
 516
 517 void LocationContext::printJson(raw_ostream &Out, const char *NL,
 518                                 unsigned int Space, bool IsDot,
 519                                 std::function<void(const LocationContext *)>
 520                                     printMoreInfoPerContext) const {
 521   ASTContext &Ctx = getAnalysisDeclContext()->getASTContext();
 522   PrintingPolicy PP(Ctx.getLangOpts());
 523   PP.TerseOutput = 1;
 524
 525   const SourceManager &SM =
 526       getAnalysisDeclContext()->getASTContext().getSourceManager();
 527
 528   unsigned Frame = 0;
 529   for (const LocationContext *LCtx = this; LCtx; LCtx = LCtx->getParent()) {
 530     Indent(Out, Space, IsDot)
 531         << "{ \"lctx_id\": " << LCtx->getID() << ", \"location_context\": \"";
 532     switch (LCtx->getKind()) {
 533     case StackFrame:
 534       Out << '#' << Frame << " Call\", \"calling\": \"";
 535       ++Frame;
 536       if (const auto *D = dyn_cast<NamedDecl>(LCtx->getDecl()))
 537         Out << D->getQualifiedNameAsString();
 538       else
 539         Out << "anonymous code";
 540
 541       Out << "\", \"location\": ";
 542       if (const Stmt *S = cast<StackFrameContext>(LCtx)->getCallSite()) {
 543         printSourceLocationAsJson(Out, S->getBeginLoc(), SM);
 544       } else {
 545         Out << "null";
 546       }
 547
 548       Out << ", \"items\": ";
 549       break;
 550     case Scope:
 551       Out << "Entering scope\" ";
 552       break;
 553     case Block:
 554       Out << "Invoking block\" ";
 555       if (const Decl *D = cast<BlockInvocationContext>(LCtx)->getDecl()) {
 556         Out << ", \"location\": ";
 557         printSourceLocationAsJson(Out, D->getBeginLoc(), SM);
 558         Out << ' ';
 559       }
 560       break;
 561     }
 562
 563     printMoreInfoPerContext(LCtx);
 564
 565     Out << '}';
 566     if (LCtx->getParent())
 567       Out << ',';
 568     Out << NL;
 569   }
 570 }
 571
 572 LLVM_DUMP_METHOD void LocationContext::dump() const { printJson(llvm::errs()); }
 573
 574 //===----------------------------------------------------------------------===//
 575 // Lazily generated map to query the external variables referenced by a Block.
 576 //===----------------------------------------------------------------------===//
 577
 578 namespace {
 579
 580 class FindBlockDeclRefExprsVals : public StmtVisitor<FindBlockDeclRefExprsVals>{
 581   BumpVector<const VarDecl *> &BEVals;
 582   BumpVectorContext &BC;
 583   llvm::SmallPtrSet<const VarDecl *, 4> Visited;
 584   llvm::SmallPtrSet<const DeclContext *, 4> IgnoredContexts;
 585
 586 public:
 587   FindBlockDeclRefExprsVals(BumpVector<const VarDecl*> &bevals,
 588                             BumpVectorContext &bc)
 589       : BEVals(bevals), BC(bc) {}
 590
 591   void VisitStmt(Stmt *S) {
 592     for (auto *Child : S->children())
 593       if (Child)
 594         Visit(Child);
 595   }
 596
 597   void VisitDeclRefExpr(DeclRefExpr *DR) {
 598     // Non-local variables are also directly modified.
 599     if (const auto *VD = dyn_cast<VarDecl>(DR->getDecl())) {
 600       if (!VD->hasLocalStorage()) {
 601         if (Visited.insert(VD).second)
 602           BEVals.push_back(VD, BC);
 603       }
 604     }
 605   }
 606
 607   void VisitBlockExpr(BlockExpr *BR) {
 608     // Blocks containing blocks can transitively capture more variables.
 609     IgnoredContexts.insert(BR->getBlockDecl());
 610     Visit(BR->getBlockDecl()->getBody());
 611   }
 612
 613   void VisitPseudoObjectExpr(PseudoObjectExpr *PE) {
 614     for (PseudoObjectExpr::semantics_iterator it = PE->semantics_begin(),
 615          et = PE->semantics_end(); it != et; ++it) {
 616       Expr *Semantic = *it;
 617       if (auto *OVE = dyn_cast<OpaqueValueExpr>(Semantic))
 618         Semantic = OVE->getSourceExpr();
 619       Visit(Semantic);
 620     }
 621   }
 622 };
 623
 624 } // namespace
 625
 626 using DeclVec = BumpVector<const VarDecl *>;
 627
 628 static DeclVec* LazyInitializeReferencedDecls(const BlockDecl *BD,
 629                                               void *&Vec,
 630                                               llvm::BumpPtrAllocator &A) {
 631   if (Vec)
 632     return (DeclVec*) Vec;
 633
 634   BumpVectorContext BC(A);
 635   DeclVec *BV = (DeclVec*) A.Allocate<DeclVec>();
 636   new (BV) DeclVec(BC, 10);
 637
 638   // Go through the capture list.
 639   for (const auto &CI : BD->captures()) {
 640     BV->push_back(CI.getVariable(), BC);
 641   }
 642
 643   // Find the referenced global/static variables.
 644   FindBlockDeclRefExprsVals F(*BV, BC);
 645   F.Visit(BD->getBody());
 646
 647   Vec = BV;
 648   return BV;
 649 }
 650
 651 llvm::iterator_range<AnalysisDeclContext::referenced_decls_iterator>
 652 AnalysisDeclContext::getReferencedBlockVars(const BlockDecl *BD) {
 653   if (!ReferencedBlockVars)
 654     ReferencedBlockVars = new llvm::DenseMap<const BlockDecl*,void*>();
 655
 656   const DeclVec *V =
 657       LazyInitializeReferencedDecls(BD, (*ReferencedBlockVars)[BD], A);
 658   return llvm::make_range(V->begin(), V->end());
 659 }
 660
 661 ManagedAnalysis *&AnalysisDeclContext::getAnalysisImpl(const void *tag) {
 662   if (!ManagedAnalyses)
 663     ManagedAnalyses = new ManagedAnalysisMap();
 664   ManagedAnalysisMap *M = (ManagedAnalysisMap*) ManagedAnalyses;
 665   return (*M)[tag];
 666 }
 667
 668 //===----------------------------------------------------------------------===//
 669 // Cleanup.
 670 //===----------------------------------------------------------------------===//
 671
 672 ManagedAnalysis::~ManagedAnalysis() = default;
 673
 674 AnalysisDeclContext::~AnalysisDeclContext() {
 675   delete forcedBlkExprs;
 676   delete ReferencedBlockVars;
 677   // Release the managed analyses.
 678   if (ManagedAnalyses) {
 679     ManagedAnalysisMap *M = (ManagedAnalysisMap*) ManagedAnalyses;
 680     llvm::DeleteContainerSeconds(*M);
 681     delete M;
 682   }
 683 }
 684
 685 LocationContext::~LocationContext() = default;
 686
 687 LocationContextManager::~LocationContextManager() {
 688   clear();
 689 }
 690
 691 void LocationContextManager::clear() {
 692   for (llvm::FoldingSet<LocationContext>::iterator I = Contexts.begin(),
 693        E = Contexts.end(); I != E; ) {
 694     LocationContext *LC = &*I;
 695     ++I;
 696     delete LC;
 697   }
 698   Contexts.clear();
 699 }