23 #include "llvm/ADT/PostOrderIterator.h" 24 #include "llvm/ADT/STLExtras.h" 25 #include "llvm/ADT/Statistic.h" 26 #include "llvm/Support/Casting.h" 27 #include "llvm/Support/Compiler.h" 28 #include "llvm/Support/DOTGraphTraits.h" 29 #include "llvm/Support/GraphWriter.h" 30 #include "llvm/Support/raw_ostream.h" 35 using namespace clang;
37 #define DEBUG_TYPE "CallGraph" 39 STATISTIC(NumObjCCallEdges,
"Number of Objective-C method call edges");
40 STATISTIC(NumBlockCallEdges,
"Number of block call edges");
53 void VisitStmt(
Stmt *S) { VisitChildren(S); }
61 if (
BlockExpr *Block = dyn_cast<BlockExpr>(CEE)) {
63 return Block->getBlockDecl();
69 void addCalledDecl(
Decl *D) {
77 if (
Decl *D = getDeclFromCall(CE))
90 D = IDecl->lookupPrivateMethod(Sel);
92 D = IDecl->lookupPrivateClassMethod(Sel);
100 void VisitChildren(
Stmt *S) {
103 this->Visit(SubStmt);
110 if (
BlockDecl *BD = dyn_cast<BlockDecl>(D))
111 addNodeForDecl(BD,
true);
113 for (
auto *I : D->
decls())
114 if (
auto *DC = dyn_cast<DeclContext>(I))
115 addNodesForBlocks(DC);
119 Root = getOrInsertNode(
nullptr);
129 if (
const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
132 if (FD->isDependentContext())
136 if (II && II->
getName().startswith(
"__inline"))
143 void CallGraph::addNodeForDecl(
Decl* D,
bool IsGlobal) {
150 CGBuilder builder(
this, Node);
156 FunctionMapTy::const_iterator I = FunctionMap.find(F);
157 if (I == FunctionMap.end())
return nullptr;
158 return I->second.get();
162 if (F && !isa<ObjCMethodDecl>(F))
165 std::unique_ptr<CallGraphNode> &
Node = FunctionMap[F];
169 Node = llvm::make_unique<CallGraphNode>(F);
172 Root->addCallee(Node.get());
177 OS <<
" --- Call graph Dump --- \n";
181 llvm::ReversePostOrderTraversal<const CallGraph *> RPOT(
this);
182 for (llvm::ReversePostOrderTraversal<const CallGraph *>::rpo_iterator
183 I = RPOT.begin(), E = RPOT.end(); I != E; ++I) {
194 CE = N->
end(); CI != CE; ++CI) {
195 assert(*CI != Root &&
"No one can call the root node.");
209 llvm::ViewGraph(
this,
"CallGraph");
213 if (
const NamedDecl *ND = dyn_cast_or_null<NamedDecl>(FD))
214 return ND->printQualifiedName(os);
234 return ND->getNameAsString();
The AST-based call graph.
Represents a function declaration or definition.
Smart pointer class that efficiently represents Objective-C method names.
Selector getSelector() const
virtual Stmt * getBody() const
getBody - If this Decl represents a declaration for a body of code, such as a function or method defi...
Specialize PointerLikeTypeTraits to allow LazyGenerationalUpdatePtr to be placed into a PointerUnion...
Stmt - This represents one statement.
ObjCInterfaceDecl * getReceiverInterface() const
Retrieve the Objective-C interface to which this message is being directed, if known.
const T * get() const
Retrieve the stored node as type T.
Decl - This represents one declaration (or definition), e.g.
DOTGraphTraits(bool isSimple=false)
SmallVectorImpl< CallRecord >::const_iterator const_iterator
decl_range decls() const
decls_begin/decls_end - Iterate over the declarations stored in this context.
CallGraphNode * getOrInsertNode(Decl *)
Lookup the node for the given declaration.
void addCallee(CallGraphNode *N)
One of these records is kept for each identifier that is lexed.
DOTGraphTraits(bool isSimple=false)
Forward-declares and imports various common LLVM datatypes that clang wants to use unqualified...
void addNodesForBlocks(DeclContext *D)
Represents an ObjC class declaration.
virtual Decl * getCanonicalDecl()
Retrieves the "canonical" declaration of the given declaration.
static std::string getNodeLabel(const CallGraphNode *Node, const CallGraph *CG)
STATISTIC(NumObjCCallEdges, "Number of Objective-C method call edges")
Represents a block literal declaration, which is like an unnamed FunctionDecl.
This represents one expression.
BlockExpr - Adaptor class for mixing a BlockDecl with expressions.
FunctionDecl * getDirectCallee()
If the callee is a FunctionDecl, return it. Otherwise return null.
Defines the clang::IdentifierInfo, clang::IdentifierTable, and clang::Selector interfaces.
An expression that sends a message to the given Objective-C object or class.
void print(raw_ostream &os) const
virtual bool hasBody() const
Returns true if this Decl represents a declaration for a body of code, such as a function or method d...
static bool includeInGraph(const Decl *D)
Determine if a declaration should be included in the graph.
StmtVisitor - This class implements a simple visitor for Stmt subclasses.
void print(raw_ostream &os) const
StringRef getName() const
Return the actual identifier string.
ast_type_traits::DynTypedNode Node
Dataflow Directional Tag Classes.
DeclContext - This is used only as base class of specific decl types that can act as declaration cont...
CallGraphNode * getNode(const Decl *) const
Lookup the node for the given declaration.
CallGraphNode * getRoot() const
\ brief Get the virtual root of the graph, all the functions available externally are represented as ...
iterator begin()
Iterators through all the callees/children of the node.
Expr * IgnoreParenImpCasts() LLVM_READONLY
Skip past any parentheses and implicit casts which might surround this expression until reaching a fi...
CallExpr - Represents a function call (C99 6.5.2.2, C++ [expr.call]).
This represents a decl that may have a name.
bool isInstanceMessage() const
Determine whether this is an instance message to either a computed object or to super.