AnalysisConsumer.cpp revision 9be6e7ce5788e50c62d40c59b0bbc2ea423683f7
1f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===--- AnalysisConsumer.cpp - ASTConsumer for running Analyses ----------===// 2f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// 3f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// The LLVM Compiler Infrastructure 4f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// 5f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// This file is distributed under the University of Illinois Open Source 6f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// License. See LICENSE.TXT for details. 7f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// 8f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===----------------------------------------------------------------------===// 9f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// 10f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// "Meta" ASTConsumer for running different source analyses. 11f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// 12f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===----------------------------------------------------------------------===// 13f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 14e6348c336fecc8da9288ea367375a1b1cd2358d2Argyrios Kyrtzidis#include "AnalysisConsumer.h" 15f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek#include "clang/AST/ASTConsumer.h" 16f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek#include "clang/AST/Decl.h" 17802be99a6817aba6edb166b93c133da4358aa783Zhongxing Xu#include "clang/AST/DeclCXX.h" 18f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek#include "clang/AST/DeclObjC.h" 19efceabd2380f49306bc0229583458e93b062094bDaniel Dunbar#include "clang/AST/ParentMap.h" 20efceabd2380f49306bc0229583458e93b062094bDaniel Dunbar#include "clang/Analysis/CFG.h" 2127af04bcca46f8a3374586be1301477f9123f5e1Argyrios Kyrtzidis#include "clang/StaticAnalyzer/Frontend/CheckerRegistration.h" 2243dee220252ef0b42c5f8a3bb1eca97f84f2565fArgyrios Kyrtzidis#include "clang/StaticAnalyzer/Core/CheckerManager.h" 2321142581d55918beed544a757e4af3bb865b1812Ted Kremenek#include "clang/StaticAnalyzer/Checkers/LocalCheckers.h" 249b663716449b618ba0390b1dbebc54fa8e971124Ted Kremenek#include "clang/StaticAnalyzer/Core/BugReporter/PathDiagnostic.h" 259b663716449b618ba0390b1dbebc54fa8e971124Ted Kremenek#include "clang/StaticAnalyzer/Core/PathSensitive/AnalysisManager.h" 269b663716449b618ba0390b1dbebc54fa8e971124Ted Kremenek#include "clang/StaticAnalyzer/Core/BugReporter/BugReporter.h" 279b663716449b618ba0390b1dbebc54fa8e971124Ted Kremenek#include "clang/StaticAnalyzer/Core/PathSensitive/ExprEngine.h" 289b663716449b618ba0390b1dbebc54fa8e971124Ted Kremenek#include "clang/StaticAnalyzer/Core/PathSensitive/TransferFuncs.h" 299b663716449b618ba0390b1dbebc54fa8e971124Ted Kremenek#include "clang/StaticAnalyzer/Core/PathDiagnosticClients.h" 30a7af5ea88a6c5bdf87497cca6c20831e8c546751Argyrios Kyrtzidis 31efceabd2380f49306bc0229583458e93b062094bDaniel Dunbar#include "clang/Basic/FileManager.h" 32efceabd2380f49306bc0229583458e93b062094bDaniel Dunbar#include "clang/Basic/SourceManager.h" 339b414d3e2d0cb84512b55a3275a98490b090162aDaniel Dunbar#include "clang/Frontend/AnalyzerOptions.h" 34efceabd2380f49306bc0229583458e93b062094bDaniel Dunbar#include "clang/Lex/Preprocessor.h" 35f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek#include "llvm/Support/raw_ostream.h" 3603013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Path.h" 3703013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Program.h" 386cb7c1a43b0c8f739d1f54b7fdae5ede86033496Benjamin Kramer#include "llvm/ADT/OwningPtr.h" 39db09a4dee28a4515438af60f2d2b4a83e4965c31Ted Kremenek 40f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenekusing namespace clang; 419ef6537a894c33003359b1f9b9676e9178e028b7Ted Kremenekusing namespace ento; 42f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 43c5619d901a68dc27a9e310a6a831f03efebcd950Zhongxing Xustatic ExplodedNode::Auditor* CreateUbiViz(); 44ff944a8c481d6c0f1ad2633e4be9bf8b1dd2a09fZhongxing Xu 45be1fe1eb12a1cb91c8e3a9fcc2db4dfe989def6cTed Kremenek//===----------------------------------------------------------------------===// 46f75560670bcdd59b051149bdece3eac14e313853Ted Kremenek// Special PathDiagnosticClients. 47f75560670bcdd59b051149bdece3eac14e313853Ted Kremenek//===----------------------------------------------------------------------===// 48f75560670bcdd59b051149bdece3eac14e313853Ted Kremenek 49f75560670bcdd59b051149bdece3eac14e313853Ted Kremenekstatic PathDiagnosticClient* 50f928dca56542cf8f981a684f090d2c79bdcb2a10Argyrios KyrtzidiscreatePlistHTMLDiagnosticClient(const std::string& prefix, 51efceabd2380f49306bc0229583458e93b062094bDaniel Dunbar const Preprocessor &PP) { 52d5b08bee799d4f77f1a25fa5977ca77b983ab031Michael J. Spencer PathDiagnosticClient *PD = 53d5b08bee799d4f77f1a25fa5977ca77b983ab031Michael J. Spencer createHTMLDiagnosticClient(llvm::sys::path::parent_path(prefix), PP); 54f928dca56542cf8f981a684f090d2c79bdcb2a10Argyrios Kyrtzidis return createPlistDiagnosticClient(prefix, PP, PD); 55f75560670bcdd59b051149bdece3eac14e313853Ted Kremenek} 56f75560670bcdd59b051149bdece3eac14e313853Ted Kremenek 57f75560670bcdd59b051149bdece3eac14e313853Ted Kremenek//===----------------------------------------------------------------------===// 58f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// AnalysisConsumer declaration. 59f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===----------------------------------------------------------------------===// 60f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 61f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremeneknamespace { 62f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 63ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xuclass AnalysisConsumer : public ASTConsumer { 64ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xupublic: 651d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek ASTContext* Ctx; 661d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek const Preprocessor &PP; 671d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek const std::string OutDir; 681d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek AnalyzerOptions Opts; 69d07a0d0279c09d1017f8450fce575a94dc9703c0Zhongxing Xu 701d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek // PD is owned by AnalysisManager. 711d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek PathDiagnosticClient *PD; 72d07a0d0279c09d1017f8450fce575a94dc9703c0Zhongxing Xu 731d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek StoreManagerCreator CreateStoreMgr; 741d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek ConstraintManagerCreator CreateConstraintMgr; 75f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 7643dee220252ef0b42c5f8a3bb1eca97f84f2565fArgyrios Kyrtzidis llvm::OwningPtr<CheckerManager> checkerMgr; 771d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek llvm::OwningPtr<AnalysisManager> Mgr; 78c471e7b44e63ff1b46b480e723c4130aeaef5a8aZhongxing Xu 791d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek AnalysisConsumer(const Preprocessor& pp, 801d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek const std::string& outdir, 811d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek const AnalyzerOptions& opts) 821d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek : Ctx(0), PP(pp), OutDir(outdir), 83c4a1437c15da43eb8d2601cdce13161ef41a4389Ted Kremenek Opts(opts), PD(0) { 841d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek DigestAnalyzerOptions(); 851d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek } 86fda7832b000ff8927386f093b52c067641679469Zhongxing Xu 871d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek void DigestAnalyzerOptions() { 881d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek // Create the PathDiagnosticClient. 891d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek if (!OutDir.empty()) { 901d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek switch (Opts.AnalysisDiagOpt) { 911d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek default: 92fda7832b000ff8927386f093b52c067641679469Zhongxing Xu#define ANALYSIS_DIAGNOSTICS(NAME, CMDFLAG, DESC, CREATEFN, AUTOCREATE) \ 931d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek case PD_##NAME: PD = CREATEFN(OutDir, PP); break; 94fda7832b000ff8927386f093b52c067641679469Zhongxing Xu#include "clang/Frontend/Analyses.def" 95fda7832b000ff8927386f093b52c067641679469Zhongxing Xu } 96a599ae8826b01c0160a519b0fc5a4871f599bf04Argyrios Kyrtzidis } else if (Opts.AnalysisDiagOpt == PD_TEXT) { 97a599ae8826b01c0160a519b0fc5a4871f599bf04Argyrios Kyrtzidis // Create the text client even without a specified output file since 98a599ae8826b01c0160a519b0fc5a4871f599bf04Argyrios Kyrtzidis // it just uses diagnostic notes. 99a599ae8826b01c0160a519b0fc5a4871f599bf04Argyrios Kyrtzidis PD = createTextPathDiagnosticClient("", PP); 1001d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek } 101fda7832b000ff8927386f093b52c067641679469Zhongxing Xu 1021d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek // Create the analyzer component creators. 1035f83d6f36a7308eef21d87104fd70c421e854448Argyrios Kyrtzidis switch (Opts.AnalysisStoreOpt) { 1045f83d6f36a7308eef21d87104fd70c421e854448Argyrios Kyrtzidis default: 1055f83d6f36a7308eef21d87104fd70c421e854448Argyrios Kyrtzidis assert(0 && "Unknown store manager."); 106fda7832b000ff8927386f093b52c067641679469Zhongxing Xu#define ANALYSIS_STORE(NAME, CMDFLAG, DESC, CREATEFN) \ 1075f83d6f36a7308eef21d87104fd70c421e854448Argyrios Kyrtzidis case NAME##Model: CreateStoreMgr = CREATEFN; break; 108fda7832b000ff8927386f093b52c067641679469Zhongxing Xu#include "clang/Frontend/Analyses.def" 1091d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek } 1101eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1115f83d6f36a7308eef21d87104fd70c421e854448Argyrios Kyrtzidis switch (Opts.AnalysisConstraintsOpt) { 1125f83d6f36a7308eef21d87104fd70c421e854448Argyrios Kyrtzidis default: 1135f83d6f36a7308eef21d87104fd70c421e854448Argyrios Kyrtzidis assert(0 && "Unknown store manager."); 114fda7832b000ff8927386f093b52c067641679469Zhongxing Xu#define ANALYSIS_CONSTRAINTS(NAME, CMDFLAG, DESC, CREATEFN) \ 1155f83d6f36a7308eef21d87104fd70c421e854448Argyrios Kyrtzidis case NAME##Model: CreateConstraintMgr = CREATEFN; break; 116fda7832b000ff8927386f093b52c067641679469Zhongxing Xu#include "clang/Frontend/Analyses.def" 117fda7832b000ff8927386f093b52c067641679469Zhongxing Xu } 1181d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek } 119f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek 1201d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek void DisplayFunction(const Decl *D) { 121c4a1437c15da43eb8d2601cdce13161ef41a4389Ted Kremenek if (!Opts.AnalyzerDisplayProgress) 1221d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek return; 123f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek 124fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek SourceManager &SM = Mgr->getASTContext().getSourceManager(); 125fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek PresumedLoc Loc = SM.getPresumedLoc(D->getLocation()); 126cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor if (Loc.isValid()) { 127cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor llvm::errs() << "ANALYZE: " << Loc.getFilename(); 128fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek 129cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor if (isa<FunctionDecl>(D) || isa<ObjCMethodDecl>(D)) { 130cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor const NamedDecl *ND = cast<NamedDecl>(D); 131cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor llvm::errs() << ' ' << ND << '\n'; 132cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor } 133cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor else if (isa<BlockDecl>(D)) { 134cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor llvm::errs() << ' ' << "block(line:" << Loc.getLine() << ",col:" 135cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor << Loc.getColumn() << '\n'; 136cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor } 137cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor else if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D)) { 138cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor Selector S = MD->getSelector(); 139cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor llvm::errs() << ' ' << S.getAsString(); 140cb7b1e17b63967317ab5cc55682168cf0380519aDouglas Gregor } 14135fa76d0bb6fb8c86159a7506efd094a4fe376d2Ted Kremenek } 1421d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek } 1431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1441d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek virtual void Initialize(ASTContext &Context) { 1451d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek Ctx = &Context; 1462e471a3e476396be1ddca4ab8b9df721bcfc9437Argyrios Kyrtzidis checkerMgr.reset(registerCheckers(Opts, PP.getLangOptions(), 1472e471a3e476396be1ddca4ab8b9df721bcfc9437Argyrios Kyrtzidis PP.getDiagnostics())); 1481d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek Mgr.reset(new AnalysisManager(*Ctx, PP.getDiagnostics(), 1491d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek PP.getLangOptions(), PD, 1501d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek CreateStoreMgr, CreateConstraintMgr, 15143dee220252ef0b42c5f8a3bb1eca97f84f2565fArgyrios Kyrtzidis checkerMgr.get(), 152c6238d2786cfd961b94580b3d3675a1b3ff0721cZhongxing Xu /* Indexer */ 0, 1536362b893731ccf4480a96527db9e55e04b801503Zhongxing Xu Opts.MaxNodes, Opts.MaxLoop, 1541d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek Opts.VisualizeEGDot, Opts.VisualizeEGUbi, 1551d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek Opts.PurgeDead, Opts.EagerlyAssume, 1569b823e8e1ccb8a2cb49923bad22a80ca96f41f92Ted Kremenek Opts.TrimGraph, Opts.InlineCall, 1579121ba232903ebe61e7bbe14ca294cf0f07dfa96Marcin Swiderski Opts.UnoptimizedCFG, Opts.CFGAddImplicitDtors, 158d767d81290288c030f3be0be1d3e62b9c8df51dcTed Kremenek Opts.CFGAddInitializers, 159d767d81290288c030f3be0be1d3e62b9c8df51dcTed Kremenek Opts.EagerlyTrimEGraph)); 1601d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek } 1611eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 1621d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek virtual void HandleTranslationUnit(ASTContext &C); 16314cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek void HandleDeclContext(ASTContext &C, DeclContext *dc); 16414cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek 165c367a876d0abcf32cb443712ce2709a0491be00bArgyrios Kyrtzidis void HandleCode(Decl *D); 1661d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek}; 167f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek} // end anonymous namespace 168f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 169f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===----------------------------------------------------------------------===// 170f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// AnalysisConsumer implementation. 171f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===----------------------------------------------------------------------===// 172f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 17314cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenekvoid AnalysisConsumer::HandleDeclContext(ASTContext &C, DeclContext *dc) { 1749fb9474c5b267400d4abfbff63c8b39f378235d4Argyrios Kyrtzidis BugReporter BR(*Mgr); 17514cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek for (DeclContext::decl_iterator I = dc->decls_begin(), E = dc->decls_end(); 176ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xu I != E; ++I) { 177ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xu Decl *D = *I; 1789fb9474c5b267400d4abfbff63c8b39f378235d4Argyrios Kyrtzidis checkerMgr->runCheckersOnASTDecl(D, *Mgr, BR); 1799fb9474c5b267400d4abfbff63c8b39f378235d4Argyrios Kyrtzidis 180ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xu switch (D->getKind()) { 18114cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek case Decl::Namespace: { 18214cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek HandleDeclContext(C, cast<NamespaceDecl>(D)); 18314cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek break; 184ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xu } 18514cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek case Decl::CXXConstructor: 18614cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek case Decl::CXXDestructor: 18714cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek case Decl::CXXConversion: 18814cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek case Decl::CXXMethod: 18914cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek case Decl::Function: { 19014cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek FunctionDecl* FD = cast<FunctionDecl>(D); 19114cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek // We skip function template definitions, as their semantics is 19214cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek // only determined when they are instantiated. 19314cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek if (FD->isThisDeclarationADefinition() && 19414cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek !FD->isDependentContext()) { 195ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xu if (!Opts.AnalyzeSpecificFunction.empty() && 19614cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek FD->getDeclName().getAsString() != Opts.AnalyzeSpecificFunction) 197ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xu break; 19814cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek DisplayFunction(FD); 199c367a876d0abcf32cb443712ce2709a0491be00bArgyrios Kyrtzidis HandleCode(FD); 200ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xu } 20114cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek break; 202ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xu } 20314cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek 20414cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek case Decl::ObjCImplementation: { 20514cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek ObjCImplementationDecl* ID = cast<ObjCImplementationDecl>(*I); 206c367a876d0abcf32cb443712ce2709a0491be00bArgyrios Kyrtzidis HandleCode(ID); 20714cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek 20814cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek for (ObjCImplementationDecl::method_iterator MI = ID->meth_begin(), 20914cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek ME = ID->meth_end(); MI != ME; ++MI) { 210b742dfde440c8bb0f7e655715e9413fa69f0fefdArgyrios Kyrtzidis checkerMgr->runCheckersOnASTDecl(*MI, *Mgr, BR); 211b742dfde440c8bb0f7e655715e9413fa69f0fefdArgyrios Kyrtzidis 21214cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek if ((*MI)->isThisDeclarationADefinition()) { 21314cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek if (!Opts.AnalyzeSpecificFunction.empty() && 21414cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek Opts.AnalyzeSpecificFunction != (*MI)->getSelector().getAsString()) 21514cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek break; 21614cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek DisplayFunction(*MI); 217c367a876d0abcf32cb443712ce2709a0491be00bArgyrios Kyrtzidis HandleCode(*MI); 21814cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek } 21914cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek } 22014cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek break; 22114cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek } 22214cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek 22314cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek default: 22414cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek break; 225ed8afacb8118b71bcfa8017059e51da325e7691bZhongxing Xu } 22614cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek } 22714cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek} 228f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek 22914cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenekvoid AnalysisConsumer::HandleTranslationUnit(ASTContext &C) { 2309fb9474c5b267400d4abfbff63c8b39f378235d4Argyrios Kyrtzidis BugReporter BR(*Mgr); 23114cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek TranslationUnitDecl *TU = C.getTranslationUnitDecl(); 2329fb9474c5b267400d4abfbff63c8b39f378235d4Argyrios Kyrtzidis checkerMgr->runCheckersOnASTDecl(TU, *Mgr, BR); 23314cc9451de4a9539bf79e4e5d63248c2377426dbTed Kremenek HandleDeclContext(C, TU); 234b317f8f5ca8737a5bbad97a3f7566a2dbd2ed61bZhongxing Xu 2359be6e7ce5788e50c62d40c59b0bbc2ea423683f7Ted Kremenek // After all decls handled, run checkers on the entire TranslationUnit. 2369be6e7ce5788e50c62d40c59b0bbc2ea423683f7Ted Kremenek checkerMgr->runCheckersOnEndOfTranslationUnit(TU, *Mgr, BR); 2379be6e7ce5788e50c62d40c59b0bbc2ea423683f7Ted Kremenek 238690a7f431d6863a101711e67636d51ddd13f35c5Ted Kremenek // Explicitly destroy the PathDiagnosticClient. This will flush its output. 239690a7f431d6863a101711e67636d51ddd13f35c5Ted Kremenek // FIXME: This should be replaced with something that doesn't rely on 240da17fd50ad485fd2a1fc5c2f055caacf532992daZhongxing Xu // side-effects in PathDiagnosticClient's destructor. This is required when 241da17fd50ad485fd2a1fc5c2f055caacf532992daZhongxing Xu // used with option -disable-free. 242d07a0d0279c09d1017f8450fce575a94dc9703c0Zhongxing Xu Mgr.reset(NULL); 243db09a4dee28a4515438af60f2d2b4a83e4965c31Ted Kremenek} 244db09a4dee28a4515438af60f2d2b4a83e4965c31Ted Kremenek 245fc576514d06c46a7cac49500169411d82f38d04bTed Kremenekstatic void FindBlocks(DeclContext *D, llvm::SmallVectorImpl<Decl*> &WL) { 246fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek if (BlockDecl *BD = dyn_cast<BlockDecl>(D)) 247fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek WL.push_back(BD); 248f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek 249fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek for (DeclContext::decl_iterator I = D->decls_begin(), E = D->decls_end(); 250fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek I!=E; ++I) 251fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek if (DeclContext *DC = dyn_cast<DeclContext>(*I)) 252fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek FindBlocks(DC, WL); 253fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek} 254fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek 255d655ab28fdf7c940d3f79f8f287954d7f76e0977Argyrios Kyrtzidisstatic void ActionObjCMemChecker(AnalysisConsumer &C, AnalysisManager& mgr, 256d655ab28fdf7c940d3f79f8f287954d7f76e0977Argyrios Kyrtzidis Decl *D); 257d655ab28fdf7c940d3f79f8f287954d7f76e0977Argyrios Kyrtzidis 258c367a876d0abcf32cb443712ce2709a0491be00bArgyrios Kyrtzidisvoid AnalysisConsumer::HandleCode(Decl *D) { 2591eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 260fc8f0e14ad142ed811e90fbd9a30e419e301c717Chris Lattner // Don't run the actions if an error has occurred with parsing the file. 26199e8192c4c8f1f596f9969e5f2cdafcee64ddaacTed Kremenek Diagnostic &Diags = PP.getDiagnostics(); 26299e8192c4c8f1f596f9969e5f2cdafcee64ddaacTed Kremenek if (Diags.hasErrorOccurred() || Diags.hasFatalErrorOccurred()) 263f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek return; 26481922f01352aceeb923c0c3cc8c96b6527322384Ted Kremenek 265f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek // Don't run the actions on declarations in header files unless 266f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek // otherwise specified. 267fcd783d583d270b7ec1ec3e0fcf83cd93d30e381Ted Kremenek SourceManager &SM = Ctx->getSourceManager(); 268fcd783d583d270b7ec1ec3e0fcf83cd93d30e381Ted Kremenek SourceLocation SL = SM.getInstantiationLoc(D->getLocation()); 269fcd783d583d270b7ec1ec3e0fcf83cd93d30e381Ted Kremenek if (!Opts.AnalyzeAll && !SM.isFromMainFile(SL)) 2701eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump return; 271f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 27258f5ec7d56b1ebf5f90ee11226ebe7663f2821eaTed Kremenek // Clear the AnalysisManager of old AnalysisContexts. 27358f5ec7d56b1ebf5f90ee11226ebe7663f2821eaTed Kremenek Mgr->ClearContexts(); 274f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek 2751eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump // Dispatch on the actions. 276fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek llvm::SmallVector<Decl*, 10> WL; 277fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek WL.push_back(D); 278f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek 27906a54a38be5054c910ffc92db60edab23f9ea105Argyrios Kyrtzidis if (D->hasBody() && Opts.AnalyzeNestedBlocks) 280fc576514d06c46a7cac49500169411d82f38d04bTed Kremenek FindBlocks(cast<DeclContext>(D), WL); 281f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek 2829fb9474c5b267400d4abfbff63c8b39f378235d4Argyrios Kyrtzidis BugReporter BR(*Mgr); 2839fb9474c5b267400d4abfbff63c8b39f378235d4Argyrios Kyrtzidis for (llvm::SmallVectorImpl<Decl*>::iterator WI=WL.begin(), WE=WL.end(); 2849fb9474c5b267400d4abfbff63c8b39f378235d4Argyrios Kyrtzidis WI != WE; ++WI) 285d655ab28fdf7c940d3f79f8f287954d7f76e0977Argyrios Kyrtzidis if ((*WI)->hasBody()) { 2869fb9474c5b267400d4abfbff63c8b39f378235d4Argyrios Kyrtzidis checkerMgr->runCheckersOnASTBody(*WI, *Mgr, BR); 287d655ab28fdf7c940d3f79f8f287954d7f76e0977Argyrios Kyrtzidis if (checkerMgr->hasPathSensitiveCheckers()) 288d655ab28fdf7c940d3f79f8f287954d7f76e0977Argyrios Kyrtzidis ActionObjCMemChecker(*this, *Mgr, *WI); 289d655ab28fdf7c940d3f79f8f287954d7f76e0977Argyrios Kyrtzidis } 290f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek} 291f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 292f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===----------------------------------------------------------------------===// 293d655ab28fdf7c940d3f79f8f287954d7f76e0977Argyrios Kyrtzidis// Path-sensitive checking. 294f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===----------------------------------------------------------------------===// 295f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 296d2592a34a059e7cbb2b11dc53649ac4912422909Argyrios Kyrtzidisstatic void ActionExprEngine(AnalysisConsumer &C, AnalysisManager& mgr, 297f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek Decl *D, 298d2592a34a059e7cbb2b11dc53649ac4912422909Argyrios Kyrtzidis TransferFuncs* tf) { 2991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 300d2592a34a059e7cbb2b11dc53649ac4912422909Argyrios Kyrtzidis llvm::OwningPtr<TransferFuncs> TF(tf); 3017032f460fc9828f386056e75933da5af61e88638Ted Kremenek 30275d03cffe20b5c945ef04eba208efb0437339997Ted Kremenek // Construct the analysis engine. We first query for the LiveVariables 30375d03cffe20b5c945ef04eba208efb0437339997Ted Kremenek // information to see if the CFG is valid. 30475d03cffe20b5c945ef04eba208efb0437339997Ted Kremenek // FIXME: Inter-procedural analysis will need to handle invalid CFGs. 30575d03cffe20b5c945ef04eba208efb0437339997Ted Kremenek if (!mgr.getLiveVariables(D)) 306f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek return; 307d2592a34a059e7cbb2b11dc53649ac4912422909Argyrios Kyrtzidis ExprEngine Eng(mgr, TF.take()); 308f6eafcca7734274d277afa121f2c4fb025a54218Ted Kremenek 309f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek // Set the graph auditor. 310c5619d901a68dc27a9e310a6a831f03efebcd950Zhongxing Xu llvm::OwningPtr<ExplodedNode::Auditor> Auditor; 311f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek if (mgr.shouldVisualizeUbigraph()) { 312f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek Auditor.reset(CreateUbiViz()); 313c5619d901a68dc27a9e310a6a831f03efebcd950Zhongxing Xu ExplodedNode::SetAuditor(Auditor.get()); 314f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek } 3151eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 316b35a74a40c1d2656a25e560e773ed48bdf49f9c0Ted Kremenek // Execute the worklist algorithm. 317c6238d2786cfd961b94580b3d3675a1b3ff0721cZhongxing Xu Eng.ExecuteWorkList(mgr.getStackFrame(D, 0), mgr.getMaxNodes()); 3181eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 319f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek // Release the auditor (if any) so that it doesn't monitor the graph 320f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek // created BugReporter. 321c5619d901a68dc27a9e310a6a831f03efebcd950Zhongxing Xu ExplodedNode::SetAuditor(0); 3223df6421150271266b5a90fd4c6bfa6566c38c036Ted Kremenek 32334d7734b6ed1d9c0f647405e065251eb67f42badTed Kremenek // Visualize the exploded graph. 324f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek if (mgr.shouldVisualizeGraphviz()) 32534d7734b6ed1d9c0f647405e065251eb67f42badTed Kremenek Eng.ViewGraph(mgr.shouldTrimGraph()); 3261eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3273df6421150271266b5a90fd4c6bfa6566c38c036Ted Kremenek // Display warnings. 3283df6421150271266b5a90fd4c6bfa6566c38c036Ted Kremenek Eng.getBugReporter().FlushReports(); 329bc46f345838b1c0d420dbd3655c94f5f360fb5b8Ted Kremenek} 330bc46f345838b1c0d420dbd3655c94f5f360fb5b8Ted Kremenek 331565e465c6d0093f1bf8414b2cabdc842022385a9Ted Kremenekstatic void ActionObjCMemCheckerAux(AnalysisConsumer &C, AnalysisManager& mgr, 3321d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek Decl *D, bool GCEnabled) { 3331eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 334d2592a34a059e7cbb2b11dc53649ac4912422909Argyrios Kyrtzidis TransferFuncs* TF = MakeCFRefCountTF(mgr.getASTContext(), 335bc46f345838b1c0d420dbd3655c94f5f360fb5b8Ted Kremenek GCEnabled, 336bc46f345838b1c0d420dbd3655c94f5f360fb5b8Ted Kremenek mgr.getLangOptions()); 3371eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 338d2592a34a059e7cbb2b11dc53649ac4912422909Argyrios Kyrtzidis ActionExprEngine(C, mgr, D, TF); 339b35a74a40c1d2656a25e560e773ed48bdf49f9c0Ted Kremenek} 340b35a74a40c1d2656a25e560e773ed48bdf49f9c0Ted Kremenek 341565e465c6d0093f1bf8414b2cabdc842022385a9Ted Kremenekstatic void ActionObjCMemChecker(AnalysisConsumer &C, AnalysisManager& mgr, 3421d9cbeb76cf4c36acf5545028e2b2ac207086442Ted Kremenek Decl *D) { 3431eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 344b35a74a40c1d2656a25e560e773ed48bdf49f9c0Ted Kremenek switch (mgr.getLangOptions().getGCMode()) { 345b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump default: 346b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump assert (false && "Invalid GC mode."); 347b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump case LangOptions::NonGC: 348565e465c6d0093f1bf8414b2cabdc842022385a9Ted Kremenek ActionObjCMemCheckerAux(C, mgr, D, false); 349b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump break; 350b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump 351b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump case LangOptions::GCOnly: 352565e465c6d0093f1bf8414b2cabdc842022385a9Ted Kremenek ActionObjCMemCheckerAux(C, mgr, D, true); 353b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump break; 354b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump 355b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump case LangOptions::HybridGC: 356565e465c6d0093f1bf8414b2cabdc842022385a9Ted Kremenek ActionObjCMemCheckerAux(C, mgr, D, false); 357565e465c6d0093f1bf8414b2cabdc842022385a9Ted Kremenek ActionObjCMemCheckerAux(C, mgr, D, true); 358b7166334d897e1e4e6a5b428fe2d0ec752ef187fMike Stump break; 359b35a74a40c1d2656a25e560e773ed48bdf49f9c0Ted Kremenek } 360b35a74a40c1d2656a25e560e773ed48bdf49f9c0Ted Kremenek} 361b35a74a40c1d2656a25e560e773ed48bdf49f9c0Ted Kremenek 362f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===----------------------------------------------------------------------===// 363f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek// AnalysisConsumer creation. 364f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek//===----------------------------------------------------------------------===// 365f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 3669ef6537a894c33003359b1f9b9676e9178e028b7Ted KremenekASTConsumer* ento::CreateAnalysisConsumer(const Preprocessor& pp, 367e71b85feb0a2df9273b1b488f5b40279369a6255Eli Friedman const std::string& OutDir, 368e71b85feb0a2df9273b1b488f5b40279369a6255Eli Friedman const AnalyzerOptions& Opts) { 369efceabd2380f49306bc0229583458e93b062094bDaniel Dunbar llvm::OwningPtr<AnalysisConsumer> C(new AnalysisConsumer(pp, OutDir, Opts)); 370be1fe1eb12a1cb91c8e3a9fcc2db4dfe989def6cTed Kremenek 3712c4036eda90fad6d219d9f3fadbd9288fa197e89Ted Kremenek // Last, disable the effects of '-Werror' when using the AnalysisConsumer. 372efceabd2380f49306bc0229583458e93b062094bDaniel Dunbar pp.getDiagnostics().setWarningsAsErrors(false); 373be1fe1eb12a1cb91c8e3a9fcc2db4dfe989def6cTed Kremenek 374f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek return C.take(); 375f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek} 376f4381fddf152a63e1ac97185293c47ec0ac2f1a6Ted Kremenek 377f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek//===----------------------------------------------------------------------===// 378f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek// Ubigraph Visualization. FIXME: Move to separate file. 379f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek//===----------------------------------------------------------------------===// 380f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 381f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremeneknamespace { 3821eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 383c5619d901a68dc27a9e310a6a831f03efebcd950Zhongxing Xuclass UbigraphViz : public ExplodedNode::Auditor { 384f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek llvm::OwningPtr<llvm::raw_ostream> Out; 385710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek llvm::sys::Path Dir, Filename; 386f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek unsigned Cntr; 387f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 388f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek typedef llvm::DenseMap<void*,unsigned> VMap; 389f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek VMap M; 3901eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 391f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenekpublic: 392710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek UbigraphViz(llvm::raw_ostream* out, llvm::sys::Path& dir, 39356b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek llvm::sys::Path& filename); 3941eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 395710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek ~UbigraphViz(); 3961eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 3971eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump virtual void AddEdge(ExplodedNode* Src, ExplodedNode* Dst); 398f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek}; 3991eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 400f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek} // end anonymous namespace 401f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 402c5619d901a68dc27a9e310a6a831f03efebcd950Zhongxing Xustatic ExplodedNode::Auditor* CreateUbiViz() { 403f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek std::string ErrMsg; 4041eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 405710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek llvm::sys::Path Dir = llvm::sys::Path::GetTemporaryDirectory(&ErrMsg); 406f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek if (!ErrMsg.empty()) 407f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek return 0; 408f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 409710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek llvm::sys::Path Filename = Dir; 410f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek Filename.appendComponent("llvm_ubi"); 411f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek Filename.makeUnique(true,&ErrMsg); 412f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 413f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek if (!ErrMsg.empty()) 414f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek return 0; 415f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 416d57a7ef9252964bc6c8471451d7bd395b0520cb8Chris Lattner llvm::errs() << "Writing '" << Filename.str() << "'.\n"; 4171eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 418f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek llvm::OwningPtr<llvm::raw_fd_ostream> Stream; 419b044c473a155b7e827680635c1699cb2b5b6c0eeDan Gohman Stream.reset(new llvm::raw_fd_ostream(Filename.c_str(), ErrMsg)); 420f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 421f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek if (!ErrMsg.empty()) 422f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek return 0; 4231eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 424710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek return new UbigraphViz(Stream.take(), Dir, Filename); 425f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek} 426f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 427c5619d901a68dc27a9e310a6a831f03efebcd950Zhongxing Xuvoid UbigraphViz::AddEdge(ExplodedNode* Src, ExplodedNode* Dst) { 4281eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 42945479c88c8a44c55d27c3bd855273bb318082f37Ted Kremenek assert (Src != Dst && "Self-edges are not allowed."); 4301eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 431f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek // Lookup the Src. If it is a new node, it's a root. 432f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek VMap::iterator SrcI= M.find(Src); 433f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek unsigned SrcID; 4341eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 435f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek if (SrcI == M.end()) { 436f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek M[Src] = SrcID = Cntr++; 437f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek *Out << "('vertex', " << SrcID << ", ('color','#00ff00'))\n"; 438f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek } 439f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek else 440f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek SrcID = SrcI->second; 4411eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 442f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek // Lookup the Dst. 443f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek VMap::iterator DstI= M.find(Dst); 444f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek unsigned DstID; 445f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 446f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek if (DstI == M.end()) { 447f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek M[Dst] = DstID = Cntr++; 448f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek *Out << "('vertex', " << DstID << ")\n"; 449f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek } 45056b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek else { 45156b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek // We have hit DstID before. Change its style to reflect a cache hit. 452f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek DstID = DstI->second; 45356b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek *Out << "('change_vertex_style', " << DstID << ", 1)\n"; 45456b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek } 455f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 456f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek // Add the edge. 4571eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump *Out << "('edge', " << SrcID << ", " << DstID 458d1289327f7d2126d732e23cc189ecd0aa5d07cbbTed Kremenek << ", ('arrow','true'), ('oriented', 'true'))\n"; 459f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek} 460f8ce6991f41d55b6e8526b7a7919771428e2b181Ted Kremenek 46156b98719b0dbebb33cb228afa888c47156be2381Ted KremenekUbigraphViz::UbigraphViz(llvm::raw_ostream* out, llvm::sys::Path& dir, 46256b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek llvm::sys::Path& filename) 46356b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek : Out(out), Dir(dir), Filename(filename), Cntr(0) { 46456b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek 46556b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek *Out << "('vertex_style_attribute', 0, ('shape', 'icosahedron'))\n"; 46656b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek *Out << "('vertex_style', 1, 0, ('shape', 'sphere'), ('color', '#ffcc66')," 46756b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek " ('size', '1.5'))\n"; 46856b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek} 46956b98719b0dbebb33cb228afa888c47156be2381Ted Kremenek 470710ad9343f32b33b336369b20edad1a21a0b3299Ted KremenekUbigraphViz::~UbigraphViz() { 471710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek Out.reset(0); 4726cb7c1a43b0c8f739d1f54b7fdae5ede86033496Benjamin Kramer llvm::errs() << "Running 'ubiviz' program... "; 473710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek std::string ErrMsg; 474710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek llvm::sys::Path Ubiviz = llvm::sys::Program::FindProgramByName("ubiviz"); 475710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek std::vector<const char*> args; 476710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek args.push_back(Ubiviz.c_str()); 477710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek args.push_back(Filename.c_str()); 478710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek args.push_back(0); 4791eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 480710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek if (llvm::sys::Program::ExecuteAndWait(Ubiviz, &args[0],0,0,0,0,&ErrMsg)) { 4816cb7c1a43b0c8f739d1f54b7fdae5ede86033496Benjamin Kramer llvm::errs() << "Error viewing graph: " << ErrMsg << "\n"; 482710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek } 4831eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump 484710ad9343f32b33b336369b20edad1a21a0b3299Ted Kremenek // Delete the directory. 4851eb4433ac451dc16f4133a88af2d002ac26c58efMike Stump Dir.eraseFromDisk(true); 486932680ecdeab4c216ce7c0736093d3dcd2f60f6dDaniel Dunbar} 487