From: Gabor Greif <ggreif@gmail.com>
Date: Thu, 1 Jul 2010 11:26:05 +0000 (+0000)
Subject: reformulate CallSiteBase::getCallee to adapt to CallInst::ArgOffset, and make it... 
X-Git-Url: http://demsky.eecs.uci.edu/git/?a=commitdiff_plain;h=cf32c6e0049405b521f0baf720d1fe76ec510b26;p=oota-llvm.git

reformulate CallSiteBase::getCallee to adapt to CallInst::ArgOffset, and make it work even if CallInst::op_* are private

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@107392 91177308-0d34-0410-b5e6-96231b3b80d8
---

diff --git a/include/llvm/Support/CallSite.h b/include/llvm/Support/CallSite.h
index 0650b61fbcf..c4d442a5f9c 100644
--- a/include/llvm/Support/CallSite.h
+++ b/include/llvm/Support/CallSite.h
@@ -273,7 +273,9 @@ private:
       // of the op_*() functions here. See CallSite::getCallee.
       //
     if (isCall())
-      return getInstruction()->op_begin(); // Unchanged (ATM)
+      return CallInst::ArgOffset
+             ? getInstruction()->op_begin() // Unchanged
+             : getInstruction()->op_end() - 1; // Skip Function
     else
       return getInstruction()->op_end() - 3; // Skip BB, BB, Function
   }