From 47c6fd73172b07211ffaaad7189847ccd2416a73 Mon Sep 17 00:00:00 2001
From: Matthijs Kooijman <matthijs@stdin.nl>
Date: Thu, 5 Jun 2008 08:57:20 +0000
Subject: Let StructRetPromotion check if all if its users are really calls or
 invokesn, not other instructions. This fixes a crash with the added testcase.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@51992 91177308-0d34-0410-b5e6-96231b3b80d8
---
 lib/Transforms/IPO/StructRetPromotion.cpp | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'lib/Transforms/IPO')

diff --git a/lib/Transforms/IPO/StructRetPromotion.cpp b/lib/Transforms/IPO/StructRetPromotion.cpp
index 97a532b..94bf4c6 100644
--- a/lib/Transforms/IPO/StructRetPromotion.cpp
+++ b/lib/Transforms/IPO/StructRetPromotion.cpp
@@ -165,6 +165,10 @@ bool SRETPromotion::isSafeToUpdateAllCallers(Function *F) {
 
     CallSite CS = CallSite::get(*FnUseI);
     Instruction *Call = CS.getInstruction();
+    // The function is used by something else than a call or invoke instruction,
+    // we can't change it!
+    if (!Call)
+      return false;
     CallSite::arg_iterator AI = CS.arg_begin();
     Value *FirstArg = *AI;
 
-- 
cgit v1.1