From 9f9e2befd0d6366f1a2ebd082a1a6d703da21103 Mon Sep 17 00:00:00 2001 From: Chris Lattner Date: Sat, 13 Oct 2001 06:33:19 +0000 Subject: [PATCH] * Add support for Invoke instructions * Add support for indirect calls git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@752 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/Analysis/IPA/CallGraph.cpp | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/lib/Analysis/IPA/CallGraph.cpp b/lib/Analysis/IPA/CallGraph.cpp index 87dbf2b4255..3f997a9dd40 100644 --- a/lib/Analysis/IPA/CallGraph.cpp +++ b/lib/Analysis/IPA/CallGraph.cpp @@ -4,6 +4,10 @@ // eventually implement call graph serialization and deserialization for // annotation support. // +// This call graph represents a dynamic method invocation as a null method node. +// A call graph may only have up to one null method node that represents all of +// the dynamic method invocations. +// //===----------------------------------------------------------------------===// #include "llvm/Analysis/CallGraph.h" @@ -12,6 +16,7 @@ #include "llvm/Module.h" #include "llvm/Method.h" #include "llvm/iOther.h" +#include "llvm/iTerminators.h" #include using namespace cfg; @@ -36,10 +41,13 @@ CallGraphNode *CallGraph::getNodeFor(Method *M) { void CallGraph::addToCallGraph(Method *M) { CallGraphNode *Node = getNodeFor(M); - for (Method::inst_iterator II = M->inst_begin(), IE = M->inst_end(); - II != IE; ++II) { - if (CallInst *CI = dyn_cast(*II)) + for (Method::inst_iterator I = M->inst_begin(), E = M->inst_end(); + I != E; ++I) { + // Dynamic calls will cause Null nodes to be created + if (CallInst *CI = dyn_cast(*I)) Node->addCalledMethod(getNodeFor(CI->getCalledMethod())); + else if (InvokeInst *II = dyn_cast(*I)) + Node->addCalledMethod(getNodeFor(II->getCalledMethod())); } }