InstCombineCalls.cpp revision a311c34d2af7c750f016ef5e4c41bee77a1dfac7
1753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner//===- InstCombineCalls.cpp -----------------------------------------------===// 2753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 3753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// The LLVM Compiler Infrastructure 4753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 5753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// This file is distributed under the University of Illinois Open Source 6753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// License. See LICENSE.TXT for details. 7753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 8753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner//===----------------------------------------------------------------------===// 9753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 10753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// This file implements the visitCall and visitInvoke functions. 11753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 12753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner//===----------------------------------------------------------------------===// 13753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 14753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner#include "InstCombine.h" 15753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner#include "llvm/IntrinsicInst.h" 16753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner#include "llvm/Support/CallSite.h" 17753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner#include "llvm/Target/TargetData.h" 18753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner#include "llvm/Analysis/MemoryBuiltins.h" 1927ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher#include "llvm/Transforms/Utils/BuildLibCalls.h" 20687140c818ba4b896329a83324714140b6580ef8Chris Lattner#include "llvm/Transforms/Utils/Local.h" 21753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerusing namespace llvm; 22753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 23753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// getPromotedType - Return the specified type promoted as it would be to pass 24753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// though a va_arg area. 25753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerstatic const Type *getPromotedType(const Type *Ty) { 26753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (const IntegerType* ITy = dyn_cast<IntegerType>(Ty)) { 27753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ITy->getBitWidth() < 32) 28753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Type::getInt32Ty(Ty->getContext()); 29753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 30753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Ty; 31753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 32753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 33753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 34753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::SimplifyMemTransfer(MemIntrinsic *MI) { 35687140c818ba4b896329a83324714140b6580ef8Chris Lattner unsigned DstAlign = getKnownAlignment(MI->getArgOperand(0), TD); 36687140c818ba4b896329a83324714140b6580ef8Chris Lattner unsigned SrcAlign = getKnownAlignment(MI->getArgOperand(1), TD); 37753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned MinAlign = std::min(DstAlign, SrcAlign); 38753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned CopyAlign = MI->getAlignment(); 39753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 40753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CopyAlign < MinAlign) { 410c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher MI->setAlignment(ConstantInt::get(MI->getAlignmentType(), 42753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MinAlign, false)); 43753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 44753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 450c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 46753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If MemCpyInst length is 1/2/4/8 bytes then replace memcpy with 47753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // load/store. 48bcda85c74304ffada586a9c58ce9bcff93b69a86Gabor Greif ConstantInt *MemOpLength = dyn_cast<ConstantInt>(MI->getArgOperand(2)); 49753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemOpLength == 0) return 0; 500c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 51753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Source and destination pointer types are always "i8*" for intrinsic. See 52753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // if the size is something we can handle with a single primitive load/store. 53753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // A single load+store correctly handles overlapping memory in the memmove 54753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // case. 55753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Size = MemOpLength->getZExtValue(); 56753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Size == 0) return MI; // Delete this mem transfer. 570c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 58753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Size > 8 || (Size&(Size-1))) 59753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; // If not 1/2/4/8 bytes, exit. 600c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 61753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Use an integer load+store unless we can find something better. 6220adc9dc4650313f017b27d9818eb2176238113dMon P Wang unsigned SrcAddrSp = 63bcda85c74304ffada586a9c58ce9bcff93b69a86Gabor Greif cast<PointerType>(MI->getArgOperand(1)->getType())->getAddressSpace(); 644ec2258ffb495d7ce00177e447740ef1123a27dbGabor Greif unsigned DstAddrSp = 65bcda85c74304ffada586a9c58ce9bcff93b69a86Gabor Greif cast<PointerType>(MI->getArgOperand(0)->getType())->getAddressSpace(); 6620adc9dc4650313f017b27d9818eb2176238113dMon P Wang 6720adc9dc4650313f017b27d9818eb2176238113dMon P Wang const IntegerType* IntType = IntegerType::get(MI->getContext(), Size<<3); 6820adc9dc4650313f017b27d9818eb2176238113dMon P Wang Type *NewSrcPtrTy = PointerType::get(IntType, SrcAddrSp); 6920adc9dc4650313f017b27d9818eb2176238113dMon P Wang Type *NewDstPtrTy = PointerType::get(IntType, DstAddrSp); 700c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 71753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Memcpy forces the use of i8* for the source and destination. That means 72753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // that if you're using memcpy to move one double around, you'll get a cast 73753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // from double* to i8*. We'd much rather use a double load+store rather than 74753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // an i64 load+store, here because this improves the odds that the source or 75753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // dest address will be promotable. See if we can find a better type than the 76753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // integer datatype. 77cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *StrippedDest = MI->getArgOperand(0)->stripPointerCasts(); 78cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (StrippedDest != MI->getArgOperand(0)) { 79753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *SrcETy = cast<PointerType>(StrippedDest->getType()) 80753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ->getElementType(); 81753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (TD && SrcETy->isSized() && TD->getTypeStoreSize(SrcETy) == Size) { 82753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The SrcETy might be something like {{{double}}} or [1 x double]. Rip 83753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // down through these levels if so. 84753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner while (!SrcETy->isSingleValueType()) { 85753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (const StructType *STy = dyn_cast<StructType>(SrcETy)) { 86753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (STy->getNumElements() == 1) 87753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SrcETy = STy->getElementType(0); 88753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner else 89753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 90753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else if (const ArrayType *ATy = dyn_cast<ArrayType>(SrcETy)) { 91753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ATy->getNumElements() == 1) 92753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SrcETy = ATy->getElementType(); 93753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner else 94753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 95753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else 96753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 97753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 980c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 9920adc9dc4650313f017b27d9818eb2176238113dMon P Wang if (SrcETy->isSingleValueType()) { 10020adc9dc4650313f017b27d9818eb2176238113dMon P Wang NewSrcPtrTy = PointerType::get(SrcETy, SrcAddrSp); 10120adc9dc4650313f017b27d9818eb2176238113dMon P Wang NewDstPtrTy = PointerType::get(SrcETy, DstAddrSp); 10220adc9dc4650313f017b27d9818eb2176238113dMon P Wang } 103753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 104753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1050c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 1060c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 107753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the memcpy/memmove provides better alignment info than we can 108753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // infer, use it. 109753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SrcAlign = std::max(SrcAlign, CopyAlign); 110753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner DstAlign = std::max(DstAlign, CopyAlign); 1110c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 1129c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif Value *Src = Builder->CreateBitCast(MI->getArgOperand(1), NewSrcPtrTy); 1139c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif Value *Dest = Builder->CreateBitCast(MI->getArgOperand(0), NewDstPtrTy); 11459f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman LoadInst *L = Builder->CreateLoad(Src, MI->isVolatile()); 11559f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman L->setAlignment(SrcAlign); 11659f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman StoreInst *S = Builder->CreateStore(L, Dest, MI->isVolatile()); 11759f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman S->setAlignment(DstAlign); 118753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 119753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Set the size of the copy to 0, it will be deleted on the next iteration. 120a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif MI->setArgOperand(2, Constant::getNullValue(MemOpLength->getType())); 121753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 122753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 123753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 124753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::SimplifyMemSet(MemSetInst *MI) { 125ae47be1ea023e4b1e6bbbdc4687333eea54c84c8Chris Lattner unsigned Alignment = getKnownAlignment(MI->getDest(), TD); 126753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MI->getAlignment() < Alignment) { 127753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MI->setAlignment(ConstantInt::get(MI->getAlignmentType(), 128753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Alignment, false)); 129753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 130753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1310c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 132753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Extract the length and alignment and fill if they are constant. 133753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt *LenC = dyn_cast<ConstantInt>(MI->getLength()); 134753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt *FillC = dyn_cast<ConstantInt>(MI->getValue()); 135b0bc6c361da9009e8414efde317d9bbff755f6c0Duncan Sands if (!LenC || !FillC || !FillC->getType()->isIntegerTy(8)) 136753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 137753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint64_t Len = LenC->getZExtValue(); 138753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Alignment = MI->getAlignment(); 1390c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 140753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the length is zero, this is a no-op 141753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Len == 0) return MI; // memset(d,c,0,a) -> noop 1420c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 143753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // memset(s,c,n) -> store s, c (for n=1,2,4,8) 144753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Len <= 8 && isPowerOf2_32((uint32_t)Len)) { 145753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *ITy = IntegerType::get(MI->getContext(), Len*8); // n=1 -> i8. 1460c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 147753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Dest = MI->getDest(); 14855fb9b0fcf89501a0abe74808f4e08e0b3a17207Mon P Wang unsigned DstAddrSp = cast<PointerType>(Dest->getType())->getAddressSpace(); 14955fb9b0fcf89501a0abe74808f4e08e0b3a17207Mon P Wang Type *NewDstPtrTy = PointerType::get(ITy, DstAddrSp); 15055fb9b0fcf89501a0abe74808f4e08e0b3a17207Mon P Wang Dest = Builder->CreateBitCast(Dest, NewDstPtrTy); 151753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 152753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Alignment 0 is identity for alignment 1 for memset, but not store. 153753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Alignment == 0) Alignment = 1; 1540c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 155753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Extract the fill value and store. 156753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint64_t Fill = FillC->getZExtValue()*0x0101010101010101ULL; 15759f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman StoreInst *S = Builder->CreateStore(ConstantInt::get(ITy, Fill), Dest, 15859f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman MI->isVolatile()); 15959f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman S->setAlignment(Alignment); 1600c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 161753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Set the size of the copy to 0, it will be deleted on the next iteration. 162753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MI->setLength(Constant::getNullValue(LenC->getType())); 163753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 164753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 165753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 166753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 167753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 168753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1690c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher/// visitCallInst - CallInst simplification. This mostly only handles folding 170753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// of intrinsic instructions. For normal calls, it allows visitCallSite to do 171753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// the heavy lifting. 172753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// 173753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitCallInst(CallInst &CI) { 174753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isFreeCall(&CI)) 175753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitFree(CI); 1761d9b973fd7e60e4149678a03aac762551c846f13Duncan Sands if (isMalloc(&CI)) 1771d9b973fd7e60e4149678a03aac762551c846f13Duncan Sands return visitMalloc(CI); 178753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 179753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the caller function is nounwind, mark the call as nounwind, even if the 180753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // callee isn't. 181753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI.getParent()->getParent()->doesNotThrow() && 182753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !CI.doesNotThrow()) { 183753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CI.setDoesNotThrow(); 184753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return &CI; 185753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1860c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 187753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner IntrinsicInst *II = dyn_cast<IntrinsicInst>(&CI); 188753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!II) return visitCallSite(&CI); 189cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif 190753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Intrinsics cannot occur in an invoke, so handle them here instead of in 191753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // visitCallSite. 192753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemIntrinsic *MI = dyn_cast<MemIntrinsic>(II)) { 193753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool Changed = false; 194753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 195753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // memmove/cpy/set of zero bytes is a noop. 196753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Constant *NumBytes = dyn_cast<Constant>(MI->getLength())) { 1976eff75104e660bae5b28963190a6940eeecb7484Chris Lattner if (NumBytes->isNullValue()) 1986eff75104e660bae5b28963190a6940eeecb7484Chris Lattner return EraseInstFromFunction(CI); 199753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 200753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ConstantInt *CI = dyn_cast<ConstantInt>(NumBytes)) 201753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI->getZExtValue() == 1) { 202753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the instruction with just byte operations. We would 203753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // transform other cases to loads/stores, but we don't know if 204753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // alignment is sufficient. 205753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 206753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 2076eff75104e660bae5b28963190a6940eeecb7484Chris Lattner 2086eff75104e660bae5b28963190a6940eeecb7484Chris Lattner // No other transformations apply to volatile transfers. 2096eff75104e660bae5b28963190a6940eeecb7484Chris Lattner if (MI->isVolatile()) 2106eff75104e660bae5b28963190a6940eeecb7484Chris Lattner return 0; 211753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 212753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we have a memmove and the source operation is a constant global, 213753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // then the source and dest pointers can't alias, so we can change this 214753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // into a call to memcpy. 215753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemMoveInst *MMI = dyn_cast<MemMoveInst>(MI)) { 216753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GlobalVariable *GVSrc = dyn_cast<GlobalVariable>(MMI->getSource())) 217753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GVSrc->isConstant()) { 218551754c4958086cc6910da7c950f2875e212f5cfEric Christopher Module *M = CI.getParent()->getParent()->getParent(); 219753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Intrinsic::ID MemCpyID = Intrinsic::memcpy; 220c310fcc909c21c94d1e5b05a97763911568a3ed8Gabor Greif const Type *Tys[3] = { CI.getArgOperand(0)->getType(), 221c310fcc909c21c94d1e5b05a97763911568a3ed8Gabor Greif CI.getArgOperand(1)->getType(), 222c310fcc909c21c94d1e5b05a97763911568a3ed8Gabor Greif CI.getArgOperand(2)->getType() }; 223c310fcc909c21c94d1e5b05a97763911568a3ed8Gabor Greif CI.setCalledFunction(Intrinsic::getDeclaration(M, MemCpyID, Tys, 3)); 224753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 225753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 226753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 227753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 228753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemTransferInst *MTI = dyn_cast<MemTransferInst>(MI)) { 229753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // memmove(x,x,size) -> noop. 230753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MTI->getSource() == MTI->getDest()) 231753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 232551754c4958086cc6910da7c950f2875e212f5cfEric Christopher } 233753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 234551754c4958086cc6910da7c950f2875e212f5cfEric Christopher // If we can determine a pointer alignment that is bigger than currently 235551754c4958086cc6910da7c950f2875e212f5cfEric Christopher // set, update the alignment. 236551754c4958086cc6910da7c950f2875e212f5cfEric Christopher if (isa<MemTransferInst>(MI)) { 237551754c4958086cc6910da7c950f2875e212f5cfEric Christopher if (Instruction *I = SimplifyMemTransfer(MI)) 238753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return I; 239753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else if (MemSetInst *MSI = dyn_cast<MemSetInst>(MI)) { 240753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Instruction *I = SimplifyMemSet(MSI)) 241753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return I; 242753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 243c310fcc909c21c94d1e5b05a97763911568a3ed8Gabor Greif 244753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Changed) return II; 245753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 246551754c4958086cc6910da7c950f2875e212f5cfEric Christopher 247753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner switch (II->getIntrinsicID()) { 248753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner default: break; 249415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher case Intrinsic::objectsize: { 25026d0e892e36e8dcc4248a8ccc8bedbd2d235eaeeEric Christopher // We need target data for just about everything so depend on it. 251415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher if (!TD) break; 25226d0e892e36e8dcc4248a8ccc8bedbd2d235eaeeEric Christopher 253a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng const Type *ReturnTy = CI.getType(); 254783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer uint64_t DontKnow = II->getArgOperand(1) == Builder->getTrue() ? 0 : -1ULL; 255a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng 25626d0e892e36e8dcc4248a8ccc8bedbd2d235eaeeEric Christopher // Get to the real allocated thing and offset as fast as possible. 257cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *Op1 = II->getArgOperand(0)->stripPointerCasts(); 258783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 259783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer uint64_t Offset = 0; 260783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer uint64_t Size = -1ULL; 261783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 262783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // Try to look through constant GEPs. 263783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (GEPOperator *GEP = dyn_cast<GEPOperator>(Op1)) { 264783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (!GEP->hasAllConstantIndices()) break; 265783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 266783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // Get the current byte offset into the thing. Use the original 267783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // operand in case we're looking through a bitcast. 268783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer SmallVector<Value*, 8> Ops(GEP->idx_begin(), GEP->idx_end()); 269783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Offset = TD->getIndexedOffset(GEP->getPointerOperandType(), 270783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Ops.data(), Ops.size()); 271783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 272783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Op1 = GEP->getPointerOperand()->stripPointerCasts(); 273783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 274783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // Make sure we're not a constant offset from an external 275783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // global. 276783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op1)) 277783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (!GV->hasDefinitiveInitializer()) break; 278783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer } 279783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 28026d0e892e36e8dcc4248a8ccc8bedbd2d235eaeeEric Christopher // If we've stripped down to a single global variable that we 28126d0e892e36e8dcc4248a8ccc8bedbd2d235eaeeEric Christopher // can know the size of then just return that. 282415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op1)) { 283415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher if (GV->hasDefinitiveInitializer()) { 284415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher Constant *C = GV->getInitializer(); 285783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Size = TD->getTypeAllocSize(C->getType()); 286415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher } else { 287a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng // Can't determine size of the GV. 288783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Constant *RetVal = ConstantInt::get(ReturnTy, DontKnow); 289415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher return ReplaceInstUsesWith(CI, RetVal); 290415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher } 291a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng } else if (AllocaInst *AI = dyn_cast<AllocaInst>(Op1)) { 292a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng // Get alloca size. 293a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng if (AI->getAllocatedType()->isSized()) { 294783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Size = TD->getTypeAllocSize(AI->getAllocatedType()); 295a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng if (AI->isArrayAllocation()) { 296a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng const ConstantInt *C = dyn_cast<ConstantInt>(AI->getArraySize()); 297a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng if (!C) break; 298783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Size *= C->getZExtValue(); 299a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng } 300a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng } 301687fed3530366b03a9a6433df0c032247ab17d5bEvan Cheng } else if (CallInst *MI = extractMallocCall(Op1)) { 302240d42d18559a1f75558951ada556baaf289c7f0Benjamin Kramer // Get allocation size. 303783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer const Type* MallocType = getMallocAllocatedType(MI); 304783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (MallocType && MallocType->isSized()) 305783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (Value *NElems = getMallocArraySize(MI, TD, true)) 306687fed3530366b03a9a6433df0c032247ab17d5bEvan Cheng if (ConstantInt *NElements = dyn_cast<ConstantInt>(NElems)) 307783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Size = NElements->getZExtValue() * TD->getTypeAllocSize(MallocType); 308783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer } 309a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng 310a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng // Do not return "I don't know" here. Later optimization passes could 311a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng // make it possible to evaluate objectsize to a constant. 312783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (Size == -1ULL) 313783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer break; 314783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 315783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (Size < Offset) { 316783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // Out of bound reference? Negative index normalized to large 317783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // index? Just return "I don't know". 318783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer return ReplaceInstUsesWith(CI, ConstantInt::get(ReturnTy, DontKnow)); 319783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer } 320783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer return ReplaceInstUsesWith(CI, ConstantInt::get(ReturnTy, Size-Offset)); 321415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher } 322753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::bswap: 323753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // bswap(bswap(x)) -> x 324cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (IntrinsicInst *Operand = dyn_cast<IntrinsicInst>(II->getArgOperand(0))) 325753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Operand->getIntrinsicID() == Intrinsic::bswap) 326cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return ReplaceInstUsesWith(CI, Operand->getArgOperand(0)); 3270c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 328753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // bswap(trunc(bswap(x))) -> trunc(lshr(x, c)) 329cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (TruncInst *TI = dyn_cast<TruncInst>(II->getArgOperand(0))) { 330753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *Operand = dyn_cast<IntrinsicInst>(TI->getOperand(0))) 331753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Operand->getIntrinsicID() == Intrinsic::bswap) { 332753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned C = Operand->getType()->getPrimitiveSizeInBits() - 333753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner TI->getType()->getPrimitiveSizeInBits(); 334753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *CV = ConstantInt::get(Operand->getType(), C); 335cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *V = Builder->CreateLShr(Operand->getArgOperand(0), CV); 336753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return new TruncInst(V, TI->getType()); 337753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 338753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 3390c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 340753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 341753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::powi: 342cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (ConstantInt *Power = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 343753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, 0) -> 1.0 344753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isZero()) 345753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantFP::get(CI.getType(), 1.0)); 346753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, 1) -> x 347753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isOne()) 348cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return ReplaceInstUsesWith(CI, II->getArgOperand(0)); 349753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, -1) -> 1/x 350753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isAllOnesValue()) 351753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return BinaryOperator::CreateFDiv(ConstantFP::get(CI.getType(), 1.0), 352cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif II->getArgOperand(0)); 353753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 354753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 355753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::cttz: { 356753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If all bits below the first known one are known zero, 357753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // this value is constant. 358cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif const IntegerType *IT = cast<IntegerType>(II->getArgOperand(0)->getType()); 359753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 360753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownZero(BitWidth, 0); 361753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownOne(BitWidth, 0); 362cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif ComputeMaskedBits(II->getArgOperand(0), APInt::getAllOnesValue(BitWidth), 363753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner KnownZero, KnownOne); 364753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned TrailingZeros = KnownOne.countTrailingZeros(); 365753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask(APInt::getLowBitsSet(BitWidth, TrailingZeros)); 366753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((Mask & KnownZero) == Mask) 367753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantInt::get(IT, 368753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt(BitWidth, TrailingZeros))); 3690c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 370753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 371753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 372753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ctlz: { 373753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If all bits above the first known one are known zero, 374753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // this value is constant. 375cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif const IntegerType *IT = cast<IntegerType>(II->getArgOperand(0)->getType()); 376753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 377753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownZero(BitWidth, 0); 378753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownOne(BitWidth, 0); 379cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif ComputeMaskedBits(II->getArgOperand(0), APInt::getAllOnesValue(BitWidth), 380753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner KnownZero, KnownOne); 381753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned LeadingZeros = KnownOne.countLeadingZeros(); 382753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask(APInt::getHighBitsSet(BitWidth, LeadingZeros)); 383753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((Mask & KnownZero) == Mask) 384753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantInt::get(IT, 385753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt(BitWidth, LeadingZeros))); 3860c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 387753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 388753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 389753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::uadd_with_overflow: { 390cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *LHS = II->getArgOperand(0), *RHS = II->getArgOperand(1); 391cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif const IntegerType *IT = cast<IntegerType>(II->getArgOperand(0)->getType()); 392753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 393753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask = APInt::getSignBit(BitWidth); 394753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt LHSKnownZero(BitWidth, 0); 395753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt LHSKnownOne(BitWidth, 0); 396753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ComputeMaskedBits(LHS, Mask, LHSKnownZero, LHSKnownOne); 397753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool LHSKnownNegative = LHSKnownOne[BitWidth - 1]; 398753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool LHSKnownPositive = LHSKnownZero[BitWidth - 1]; 399753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 400753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownNegative || LHSKnownPositive) { 401753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt RHSKnownZero(BitWidth, 0); 402753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt RHSKnownOne(BitWidth, 0); 403753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ComputeMaskedBits(RHS, Mask, RHSKnownZero, RHSKnownOne); 404753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool RHSKnownNegative = RHSKnownOne[BitWidth - 1]; 405753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool RHSKnownPositive = RHSKnownZero[BitWidth - 1]; 406753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownNegative && RHSKnownNegative) { 407753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The sign bit is set in both cases: this MUST overflow. 408753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Create a simple add instruction, and insert it into the struct. 40959f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Value *Add = Builder->CreateAdd(LHS, RHS); 41059f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Add->takeName(&CI); 411753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 41259f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman UndefValue::get(LHS->getType()), 41359f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman ConstantInt::getTrue(II->getContext()) 414753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 415753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 416753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, Add, 0); 417753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 41859f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman 419753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownPositive && RHSKnownPositive) { 420753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The sign bit is clear in both cases: this CANNOT overflow. 421753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Create a simple add instruction, and insert it into the struct. 42259f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Value *Add = Builder->CreateNUWAdd(LHS, RHS); 42359f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Add->takeName(&CI); 424753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 425753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(LHS->getType()), 426753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 427753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 428753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 429753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, Add, 0); 430753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 431753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 432753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 433753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // FALL THROUGH uadd into sadd 434753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::sadd_with_overflow: 435753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Canonicalize constants into the RHS. 436a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif if (isa<Constant>(II->getArgOperand(0)) && 437a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif !isa<Constant>(II->getArgOperand(1))) { 438a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif Value *LHS = II->getArgOperand(0); 439a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(0, II->getArgOperand(1)); 440a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(1, LHS); 441753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 442753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 443753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 444753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X + undef -> undef 4459c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (isa<UndefValue>(II->getArgOperand(1))) 446753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 4470c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 4489c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (ConstantInt *RHS = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 449753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X + 0 -> {X, false} 450753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHS->isZero()) { 451753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 4524fffb345ed20bd144a31c1c1724cffb6f519b814Eli Friedman UndefValue::get(II->getArgOperand(0)->getType()), 453753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 454753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 455753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 456cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return InsertValueInst::Create(Struct, II->getArgOperand(0), 0); 457753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 458753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 459753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 460753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::usub_with_overflow: 461753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ssub_with_overflow: 462753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // undef - X -> undef 463753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X - undef -> undef 4649c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (isa<UndefValue>(II->getArgOperand(0)) || 4659c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif isa<UndefValue>(II->getArgOperand(1))) 466753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 4670c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 4689c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (ConstantInt *RHS = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 469753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X - 0 -> {X, false} 470753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHS->isZero()) { 471753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 4729c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif UndefValue::get(II->getArgOperand(0)->getType()), 473753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 474753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 475753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 4769c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif return InsertValueInst::Create(Struct, II->getArgOperand(0), 0); 477753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 478753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 479753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 4806b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer case Intrinsic::umul_with_overflow: { 4816b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Value *LHS = II->getArgOperand(0), *RHS = II->getArgOperand(1); 4826b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer unsigned BitWidth = cast<IntegerType>(LHS->getType())->getBitWidth(); 4836b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt Mask = APInt::getAllOnesValue(BitWidth); 4846b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer 4856b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt LHSKnownZero(BitWidth, 0); 4866b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt LHSKnownOne(BitWidth, 0); 4876b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer ComputeMaskedBits(LHS, Mask, LHSKnownZero, LHSKnownOne); 4886b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt RHSKnownZero(BitWidth, 0); 4896b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt RHSKnownOne(BitWidth, 0); 4906b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer ComputeMaskedBits(RHS, Mask, RHSKnownZero, RHSKnownOne); 4916b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer 492d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer // Get the largest possible values for each operand. 493d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer APInt LHSMax = ~LHSKnownZero; 494d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer APInt RHSMax = ~RHSKnownZero; 4956b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer 4966b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer // If multiplying the maximum values does not overflow then we can turn 4976b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer // this into a plain NUW mul. 498d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer bool Overflow; 499d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer LHSMax.umul_ov(RHSMax, Overflow); 500d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer if (!Overflow) { 5016b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Value *Mul = Builder->CreateNUWMul(LHS, RHS, "umul_with_overflow"); 5026b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Constant *V[] = { 5036b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer UndefValue::get(LHS->getType()), 5046b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Builder->getFalse() 5056b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer }; 5066b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 5076b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer return InsertValueInst::Create(Struct, Mul, 0); 5086b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer } 5096b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer } // FALL THROUGH 510753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::smul_with_overflow: 511753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Canonicalize constants into the RHS. 512a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif if (isa<Constant>(II->getArgOperand(0)) && 513a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif !isa<Constant>(II->getArgOperand(1))) { 514a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif Value *LHS = II->getArgOperand(0); 515a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(0, II->getArgOperand(1)); 516a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(1, LHS); 517753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 518753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 519753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 520753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X * undef -> undef 5219c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (isa<UndefValue>(II->getArgOperand(1))) 522753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 5230c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 5249c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (ConstantInt *RHSI = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 525753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X*0 -> {0, false} 526753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHSI->isZero()) 527753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, Constant::getNullValue(II->getType())); 5280c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 529753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X * 1 -> {X, false} 530753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHSI->equalsInt(1)) { 531753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 532cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif UndefValue::get(II->getArgOperand(0)->getType()), 533753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 534753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 535753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 536cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return InsertValueInst::Create(Struct, II->getArgOperand(0), 0); 537753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 538753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 539753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 540753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_lvx: 541753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_lvxl: 542f93f7b2446bec3febc30b7136e18704664bd98ccBill Wendling // Turn PPC lvx -> load if the pointer is known aligned. 543687140c818ba4b896329a83324714140b6580ef8Chris Lattner if (getOrEnforceKnownAlignment(II->getArgOperand(0), 16, TD) >= 16) { 544cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *Ptr = Builder->CreateBitCast(II->getArgOperand(0), 545753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(II->getType())); 546753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return new LoadInst(Ptr); 547753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 548753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 549753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_stvx: 550753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_stvxl: 551753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn stvx -> store if the pointer is known aligned. 552687140c818ba4b896329a83324714140b6580ef8Chris Lattner if (getOrEnforceKnownAlignment(II->getArgOperand(1), 16, TD) >= 16) { 5530c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher const Type *OpPtrTy = 5542f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif PointerType::getUnqual(II->getArgOperand(0)->getType()); 5552f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif Value *Ptr = Builder->CreateBitCast(II->getArgOperand(1), OpPtrTy); 5562f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif return new StoreInst(II->getArgOperand(0), Ptr); 557753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 558753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 559753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse_storeu_ps: 560753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_storeu_pd: 561753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_storeu_dq: 562753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn X86 storeu -> store if the pointer is known aligned. 563687140c818ba4b896329a83324714140b6580ef8Chris Lattner if (getOrEnforceKnownAlignment(II->getArgOperand(0), 16, TD) >= 16) { 5640c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher const Type *OpPtrTy = 5652f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif PointerType::getUnqual(II->getArgOperand(1)->getType()); 5662f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif Value *Ptr = Builder->CreateBitCast(II->getArgOperand(0), OpPtrTy); 5672f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif return new StoreInst(II->getArgOperand(1), Ptr); 568753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 569753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 5709cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth 5719cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvtss2si: 5729cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvtss2si64: 5739cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvttss2si: 5749cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvttss2si64: 5759cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvtsd2si: 5769cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvtsd2si64: 5779cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvttsd2si: 5789cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvttsd2si64: { 5799cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth // These intrinsics only demand the 0th element of their input vectors. If 580753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // we can simplify the input based on that, do so now. 581753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned VWidth = 5829c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif cast<VectorType>(II->getArgOperand(0)->getType())->getNumElements(); 583753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt DemandedElts(VWidth, 1); 584753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt UndefElts(VWidth, 0); 585a399781289092fcdceb58b21174229f4373c4191Gabor Greif if (Value *V = SimplifyDemandedVectorElts(II->getArgOperand(0), 586a399781289092fcdceb58b21174229f4373c4191Gabor Greif DemandedElts, UndefElts)) { 587a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(0, V); 588753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 589753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 590753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 591753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 5929cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth 593ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings 594ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovsxbw: 595ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovsxwd: 596ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovsxdq: 597ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovzxbw: 598ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovzxwd: 599ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovzxdq: { 600aaa7f499c1a2d46491a98d978ef45b0a06d73d31Evan Cheng // pmov{s|z}x ignores the upper half of their input vectors. 601ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings unsigned VWidth = 602ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings cast<VectorType>(II->getArgOperand(0)->getType())->getNumElements(); 603ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings unsigned LowHalfElts = VWidth / 2; 604d116611e0a884e53ea1524de3a1306c1eb447228Stuart Hastings APInt InputDemandedElts(APInt::getBitsSet(VWidth, 0, LowHalfElts)); 605ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings APInt UndefElts(VWidth, 0); 606ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings if (Value *TmpV = SimplifyDemandedVectorElts(II->getArgOperand(0), 607ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings InputDemandedElts, 608ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings UndefElts)) { 609ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings II->setArgOperand(0, TmpV); 610ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings return II; 611ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings } 612ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings break; 613ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings } 614ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings 615753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_vperm: 616753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn vperm(V1,V2,mask) -> shuffle(V1,V2,mask) if mask is a constant. 617cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (ConstantVector *Mask = dyn_cast<ConstantVector>(II->getArgOperand(2))) { 618753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner assert(Mask->getNumOperands() == 16 && "Bad type for intrinsic!"); 6190c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 620753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Check that all of the elements are integer constants or undefs. 621753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool AllEltsOk = true; 622753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != 16; ++i) { 6230c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher if (!isa<ConstantInt>(Mask->getOperand(i)) && 624753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !isa<UndefValue>(Mask->getOperand(i))) { 625753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AllEltsOk = false; 626753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 627753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 628753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 6290c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 630753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (AllEltsOk) { 631753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Cast the input vectors to byte vectors. 632a399781289092fcdceb58b21174229f4373c4191Gabor Greif Value *Op0 = Builder->CreateBitCast(II->getArgOperand(0), 633a399781289092fcdceb58b21174229f4373c4191Gabor Greif Mask->getType()); 634a399781289092fcdceb58b21174229f4373c4191Gabor Greif Value *Op1 = Builder->CreateBitCast(II->getArgOperand(1), 635a399781289092fcdceb58b21174229f4373c4191Gabor Greif Mask->getType()); 636753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Result = UndefValue::get(Op0->getType()); 6370c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 638753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Only extract each element once. 639753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *ExtractedElts[32]; 640753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner memset(ExtractedElts, 0, sizeof(ExtractedElts)); 6410c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 642753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != 16; ++i) { 643753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<UndefValue>(Mask->getOperand(i))) 644753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner continue; 645753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx=cast<ConstantInt>(Mask->getOperand(i))->getZExtValue(); 646753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Idx &= 31; // Match the hardware behavior. 6470c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 648753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ExtractedElts[Idx] == 0) { 6490c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher ExtractedElts[Idx] = 6500c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher Builder->CreateExtractElement(Idx < 16 ? Op0 : Op1, 651753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::get(Type::getInt32Ty(II->getContext()), 652753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Idx&15, false), "tmp"); 653753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 6540c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 655753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert this value into the result vector. 656753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Result = Builder->CreateInsertElement(Result, ExtractedElts[Idx], 657753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::get(Type::getInt32Ty(II->getContext()), 658753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner i, false), "tmp"); 659753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 660753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return CastInst::Create(Instruction::BitCast, Result, CI.getType()); 661753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 662753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 663753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 664753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 665364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld1: 666364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld2: 667364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld3: 668364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld4: 669364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld2lane: 670364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld3lane: 671364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld4lane: 672364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst1: 673364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst2: 674364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst3: 675364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst4: 676364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst2lane: 677364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst3lane: 678364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst4lane: { 679ae47be1ea023e4b1e6bbbdc4687333eea54c84c8Chris Lattner unsigned MemAlign = getKnownAlignment(II->getArgOperand(0), TD); 680364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson unsigned AlignArg = II->getNumArgOperands() - 1; 681364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson ConstantInt *IntrAlign = dyn_cast<ConstantInt>(II->getArgOperand(AlignArg)); 682364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson if (IntrAlign && IntrAlign->getZExtValue() < MemAlign) { 683364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson II->setArgOperand(AlignArg, 684364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson ConstantInt::get(Type::getInt32Ty(II->getContext()), 685364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson MemAlign, false)); 686364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson return II; 687364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson } 688364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson break; 689364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson } 690364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson 691753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::stackrestore: { 692753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the save is right next to the restore, remove the restore. This can 693753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // happen when variable allocas are DCE'd. 694cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (IntrinsicInst *SS = dyn_cast<IntrinsicInst>(II->getArgOperand(0))) { 695753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (SS->getIntrinsicID() == Intrinsic::stacksave) { 696753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator BI = SS; 697753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (&*++BI == II) 698753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 699753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 700753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 7010c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 702753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Scan down this block to see if there is another stack restore in the 703753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // same block without an intervening call/alloca. 704753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator BI = II; 705753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner TerminatorInst *TI = II->getParent()->getTerminator(); 706753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool CannotRemove = false; 707753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (++BI; &*BI != TI; ++BI) { 708753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<AllocaInst>(BI) || isMalloc(BI)) { 709753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CannotRemove = true; 710753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 711753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 712753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CallInst *BCI = dyn_cast<CallInst>(BI)) { 713753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(BCI)) { 714753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If there is a stackrestore below this one, remove this one. 715753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (II->getIntrinsicID() == Intrinsic::stackrestore) 716753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 717753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Otherwise, ignore the intrinsic. 718753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 719753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we found a non-intrinsic call, we can't remove the stack 720753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // restore. 721753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CannotRemove = true; 722753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 723753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 724753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 725753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 7260c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 727753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the stack restore is in a return/unwind block and if there are no 728753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // allocas or calls between the restore and the return, nuke the restore. 729753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CannotRemove && (isa<ReturnInst>(TI) || isa<UnwindInst>(TI))) 730753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 731753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 732753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 733753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 734753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 735753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitCallSite(II); 736753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 737753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 738753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// InvokeInst simplification 739753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 740753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitInvokeInst(InvokeInst &II) { 741753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitCallSite(&II); 742753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 743753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 7440c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher/// isSafeToEliminateVarargsCast - If this cast does not affect the value 745753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// passed through the varargs area, we can eliminate the use of the cast. 746753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerstatic bool isSafeToEliminateVarargsCast(const CallSite CS, 747753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const CastInst * const CI, 748753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const TargetData * const TD, 749753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const int ix) { 750753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CI->isLosslessCast()) 751753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 752753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 753753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The size of ByVal arguments is derived from the type, so we 754753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // can't change to a type with a different size. If the size were 755753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // passed explicitly we could avoid this check. 756753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CS.paramHasAttr(ix, Attribute::ByVal)) 757753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 758753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 7590c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher const Type* SrcTy = 760753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<PointerType>(CI->getOperand(0)->getType())->getElementType(); 761753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type* DstTy = cast<PointerType>(CI->getType())->getElementType(); 762753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!SrcTy->isSized() || !DstTy->isSized()) 763753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 764753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!TD || TD->getTypeAllocSize(SrcTy) != TD->getTypeAllocSize(DstTy)) 765753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 766753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 767753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 768753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 7690b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramernamespace { 7700b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramerclass InstCombineFortifiedLibCalls : public SimplifyFortifiedLibCalls { 7710b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer InstCombiner *IC; 7720b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramerprotected: 7730b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer void replaceCall(Value *With) { 7740b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer NewInstruction = IC->ReplaceInstUsesWith(*CI, With); 7750b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer } 7760b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer bool isFoldable(unsigned SizeCIOp, unsigned SizeArgOp, bool isString) const { 7778143a84c46b8d2fb58eddb46d84dd5d44d03cb6cBenjamin Kramer if (CI->getArgOperand(SizeCIOp) == CI->getArgOperand(SizeArgOp)) 7788143a84c46b8d2fb58eddb46d84dd5d44d03cb6cBenjamin Kramer return true; 779a399781289092fcdceb58b21174229f4373c4191Gabor Greif if (ConstantInt *SizeCI = 780a399781289092fcdceb58b21174229f4373c4191Gabor Greif dyn_cast<ConstantInt>(CI->getArgOperand(SizeCIOp))) { 7810b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer if (SizeCI->isAllOnesValue()) 7820b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer return true; 783b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher if (isString) { 784b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher uint64_t Len = GetStringLength(CI->getArgOperand(SizeArgOp)); 785b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher // If the length is 0 we don't know how long it is and so we can't 786b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher // remove the check. 787b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher if (Len == 0) return false; 788b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher return SizeCI->getZExtValue() >= Len; 789b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher } 790a399781289092fcdceb58b21174229f4373c4191Gabor Greif if (ConstantInt *Arg = dyn_cast<ConstantInt>( 791a399781289092fcdceb58b21174229f4373c4191Gabor Greif CI->getArgOperand(SizeArgOp))) 7929d8f002d8f5a5aba665bb6f85c19eee77810043cEvan Cheng return SizeCI->getZExtValue() >= Arg->getZExtValue(); 7930b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer } 7940b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer return false; 7950b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer } 7960b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramerpublic: 7970b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer InstCombineFortifiedLibCalls(InstCombiner *IC) : IC(IC), NewInstruction(0) { } 7980b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer Instruction *NewInstruction; 7990b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer}; 8000b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer} // end anonymous namespace 8010b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer 80227ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// Try to fold some different type of calls here. 80327ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// Currently we're only working with the checking functions, memcpy_chk, 80427ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// mempcpy_chk, memmove_chk, memset_chk, strcpy_chk, stpcpy_chk, strncpy_chk, 80527ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// strcat_chk and strncat_chk. 80627ceaa13f820e33c44e29adc2095dd3e1e69d09fEric ChristopherInstruction *InstCombiner::tryOptimizeCall(CallInst *CI, const TargetData *TD) { 80727ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher if (CI->getCalledFunction() == 0) return 0; 80827ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher 8090b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer InstCombineFortifiedLibCalls Simplifier(this); 8100b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer Simplifier.fold(CI, TD); 8110b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer return Simplifier.NewInstruction; 81227ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher} 81327ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher 814753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// visitCallSite - Improvements for call and invoke instructions. 815753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 816753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitCallSite(CallSite CS) { 817753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool Changed = false; 818753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 819ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner // If the callee is a pointer to a function, attempt to move any casts to the 820ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner // arguments of the call/invoke. 821753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Callee = CS.getCalledValue(); 822ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner if (!isa<Function>(Callee) && transformConstExprCastCall(CS)) 823ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner return 0; 824753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 825753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Function *CalleeF = dyn_cast<Function>(Callee)) 826d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // If the call and callee calling conventions don't match, this call must 827d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // be unreachable, as the call is undefined. 828d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner if (CalleeF->getCallingConv() != CS.getCallingConv() && 829d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // Only do this for calls to a function with a body. A prototype may 830d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // not actually end up matching the implementation's calling conv for a 831d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // variety of reasons (e.g. it may be written in assembly). 832d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner !CalleeF->isDeclaration()) { 833753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *OldCall = CS.getInstruction(); 834753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner new StoreInst(ConstantInt::getTrue(Callee->getContext()), 8350c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher UndefValue::get(Type::getInt1PtrTy(Callee->getContext())), 836753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner OldCall); 837753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If OldCall dues not return void then replaceAllUsesWith undef. 838753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This allows ValueHandlers and custom metadata to adjust itself. 839753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!OldCall->getType()->isVoidTy()) 8403e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman ReplaceInstUsesWith(*OldCall, UndefValue::get(OldCall->getType())); 841830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner if (isa<CallInst>(OldCall)) 842753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(*OldCall); 8430c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 844830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner // We cannot remove an invoke, because it would change the CFG, just 845830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner // change the callee to a null pointer. 846654c06f6457f80428e40810ad251c412462731e4Gabor Greif cast<InvokeInst>(OldCall)->setCalledFunction( 847830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner Constant::getNullValue(CalleeF->getType())); 848753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 849753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 850753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 851753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<ConstantPointerNull>(Callee) || isa<UndefValue>(Callee)) { 852753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This instruction is not reachable, just remove it. We insert a store to 853753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // undef so that we know that this code is not reachable, despite the fact 854753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // that we can't modify the CFG here. 855753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner new StoreInst(ConstantInt::getTrue(Callee->getContext()), 856753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(Type::getInt1PtrTy(Callee->getContext())), 857753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.getInstruction()); 858753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 859cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif // If CS does not return void then replaceAllUsesWith undef. 860753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This allows ValueHandlers and custom metadata to adjust itself. 861753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CS.getInstruction()->getType()->isVoidTy()) 8623e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman ReplaceInstUsesWith(*CS.getInstruction(), 8633e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman UndefValue::get(CS.getInstruction()->getType())); 864753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 865753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(CS.getInstruction())) { 866753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Don't break the CFG, insert a dummy cond branch. 867753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BranchInst::Create(II->getNormalDest(), II->getUnwindDest(), 868753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getTrue(Callee->getContext()), II); 869753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 870753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(*CS.getInstruction()); 871753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 872753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 873753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (BitCastInst *BC = dyn_cast<BitCastInst>(Callee)) 874753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *In = dyn_cast<IntrinsicInst>(BC->getOperand(0))) 875753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (In->getIntrinsicID() == Intrinsic::init_trampoline) 876753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return transformCallThroughTrampoline(CS); 877753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 878753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const PointerType *PTy = cast<PointerType>(Callee->getType()); 879753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 880753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FTy->isVarArg()) { 881753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner int ix = FTy->getNumParams() + (isa<InvokeInst>(Callee) ? 3 : 1); 882753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // See if we can optimize any arguments passed through the varargs area of 883753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // the call. 884753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(), 885753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = CS.arg_end(); I != E; ++I, ++ix) { 886753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CastInst *CI = dyn_cast<CastInst>(*I); 887753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI && isSafeToEliminateVarargsCast(CS, CI, TD, ix)) { 888753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner *I = CI->getOperand(0); 889753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 890753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 891753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 892753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 893753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 894753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<InlineAsm>(Callee) && !CS.doesNotThrow()) { 895753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Inline asm calls cannot throw - mark them 'nounwind'. 896753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.setDoesNotThrow(); 897753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 898753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 899753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 90027ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher // Try to optimize the call if possible, we require TargetData for most of 90127ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher // this. None of these calls are seen as possibly dead so go ahead and 90227ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher // delete the instruction now. 90327ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher if (CallInst *CI = dyn_cast<CallInst>(CS.getInstruction())) { 90427ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher Instruction *I = tryOptimizeCall(CI, TD); 9057b323a34fa034389e9e439f6c02eeb73e87ab9dbEric Christopher // If we changed something return the result, etc. Otherwise let 9067b323a34fa034389e9e439f6c02eeb73e87ab9dbEric Christopher // the fallthrough check. 9077b323a34fa034389e9e439f6c02eeb73e87ab9dbEric Christopher if (I) return EraseInstFromFunction(*I); 90827ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher } 90927ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher 910753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Changed ? CS.getInstruction() : 0; 911753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 912753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 913753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// transformConstExprCastCall - If the callee is a constexpr cast of a function, 914753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// attempt to move the cast to the arguments of the call/invoke. 915753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 916753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerbool InstCombiner::transformConstExprCastCall(CallSite CS) { 917ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner Function *Callee = 918ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner dyn_cast<Function>(CS.getCalledValue()->stripPointerCasts()); 919ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner if (Callee == 0) 920753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 921753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Caller = CS.getInstruction(); 922753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &CallerPAL = CS.getAttributes(); 923753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 924753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Okay, this is a cast from a function to a different type. Unless doing so 925753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // would cause a type conversion of one of our arguments, change this call to 926753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // be a direct call with arguments casted to the appropriate types. 927753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // 928753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const FunctionType *FT = Callee->getFunctionType(); 929753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *OldRetTy = Caller->getType(); 930753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *NewRetTy = FT->getReturnType(); 931753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 9321df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands if (NewRetTy->isStructTy()) 933753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // TODO: Handle multiple return values. 934753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 935753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Check to see if we are changing the return type... 936753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (OldRetTy != NewRetTy) { 937753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Callee->isDeclaration() && 938753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Conversion is ok if changing from one pointer type to another or from 939753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // a pointer to an integer of the same size. 9401df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands !((OldRetTy->isPointerTy() || !TD || 941753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner OldRetTy == TD->getIntPtrType(Caller->getContext())) && 9421df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands (NewRetTy->isPointerTy() || !TD || 943753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewRetTy == TD->getIntPtrType(Caller->getContext())))) 944753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this return value. 945753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 946753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty() && 947753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // void -> non-void is handled specially 948753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !NewRetTy->isVoidTy() && !CastInst::isCastable(NewRetTy, OldRetTy)) 949753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this return value. 950753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 951753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CallerPAL.isEmpty() && !Caller->use_empty()) { 952753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes RAttrs = CallerPAL.getRetAttributes(); 953753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RAttrs & Attribute::typeIncompatible(NewRetTy)) 954753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Attribute not compatible with transformed value. 955753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 956753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 957753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the callsite is an invoke instruction, and the return value is used by 958753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // a PHI node in a successor, we cannot change the return type of the call 959753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // because there is no place to put the cast instruction (without breaking 960753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // the critical edge). Bail out in this case. 961753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty()) 962753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) 963753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (Value::use_iterator UI = II->use_begin(), E = II->use_end(); 964753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UI != E; ++UI) 965753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PHINode *PN = dyn_cast<PHINode>(*UI)) 966753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PN->getParent() == II->getNormalDest() || 967753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PN->getParent() == II->getUnwindDest()) 968753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 969753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 970753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 971753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin()); 972753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs); 973753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 974753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallSite::arg_iterator AI = CS.arg_begin(); 975753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) { 976753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *ParamTy = FT->getParamType(i); 977753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *ActTy = (*AI)->getType(); 978753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 979753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CastInst::isCastable(ActTy, ParamTy)) 980753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this parameter value. 981753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 9822b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner unsigned Attrs = CallerPAL.getParamAttributes(i + 1); 9832b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner if (Attrs & Attribute::typeIncompatible(ParamTy)) 984753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Attribute not compatible with transformed value. 9852b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner 9862b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner // If the parameter is passed as a byval argument, then we have to have a 9872b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner // sized type and the sized type has to have the same size as the old type. 9882b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner if (ParamTy != ActTy && (Attrs & Attribute::ByVal)) { 9892b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner const PointerType *ParamPTy = dyn_cast<PointerType>(ParamTy); 9902b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner if (ParamPTy == 0 || !ParamPTy->getElementType()->isSized() || TD == 0) 9912b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner return false; 9922b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner 9932b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner const Type *CurElTy = cast<PointerType>(ActTy)->getElementType(); 9942b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner if (TD->getTypeAllocSize(CurElTy) != 9952b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner TD->getTypeAllocSize(ParamPTy->getElementType())) 9962b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner return false; 9972b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner } 998753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 999753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Converting from one pointer type to another or between a pointer and an 1000753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // integer of the same size is safe even if we do not have a body. 1001753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool isConvertible = ActTy == ParamTy || 10021df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands (TD && ((ParamTy->isPointerTy() || 1003753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ParamTy == TD->getIntPtrType(Caller->getContext())) && 10041df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands (ActTy->isPointerTy() || 1005753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ActTy == TD->getIntPtrType(Caller->getContext())))); 1006753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Callee->isDeclaration() && !isConvertible) return false; 1007753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1008753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1009091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner if (Callee->isDeclaration()) { 1010091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // Do not delete arguments unless we have a function body. 1011091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner if (FT->getNumParams() < NumActualArgs && !FT->isVarArg()) 1012091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner return false; 1013091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner 1014091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // If the callee is just a declaration, don't change the varargsness of the 1015091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // call. We don't want to introduce a varargs call where one doesn't 1016091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // already exist. 1017091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner const PointerType *APTy = cast<PointerType>(CS.getCalledValue()->getType()); 1018091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner if (FT->isVarArg()!=cast<FunctionType>(APTy->getElementType())->isVarArg()) 1019091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner return false; 1020091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner } 1021091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner 1022753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FT->getNumParams() < NumActualArgs && FT->isVarArg() && 1023753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !CallerPAL.isEmpty()) 1024753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // In this case we have more arguments than the new function type, but we 1025753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // won't be dropping them. Check that these extra arguments have attributes 1026753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // that are compatible with being a vararg call argument. 1027753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = CallerPAL.getNumSlots(); i; --i) { 1028753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CallerPAL.getSlot(i - 1).Index <= FT->getNumParams()) 1029753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1030753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes PAttrs = CallerPAL.getSlot(i - 1).Attrs; 1031753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PAttrs & Attribute::VarArgsIncompatible) 1032753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 1033753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1034753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1035091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner 1036753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Okay, we decided that this is a safe thing to do: go ahead and start 1037091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // inserting cast instructions as necessary. 1038753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<Value*> Args; 1039753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.reserve(NumActualArgs); 1040753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SmallVector<AttributeWithIndex, 8> attrVec; 1041753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.reserve(NumCommonArgs); 1042753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1043753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Get any return attributes. 1044753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes RAttrs = CallerPAL.getRetAttributes(); 1045753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1046753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the return value is not being used, the type may not be compatible 1047753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // with the existing attributes. Wipe out any problematic attributes. 1048753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner RAttrs &= ~Attribute::typeIncompatible(NewRetTy); 1049753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1050753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the new return attributes. 1051753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RAttrs) 1052753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(0, RAttrs)); 1053753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1054753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AI = CS.arg_begin(); 1055753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) { 1056753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *ParamTy = FT->getParamType(i); 1057753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((*AI)->getType() == ParamTy) { 1058753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(*AI); 1059753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1060753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction::CastOps opcode = CastInst::getCastOpcode(*AI, 1061753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner false, ParamTy, false); 1062753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Builder->CreateCast(opcode, *AI, ParamTy, "tmp")); 1063753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1064753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1065753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any parameter attributes. 1066753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) 1067753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); 1068753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1069753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1070753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the function takes more arguments than the call was taking, add them 1071753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // now. 1072753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i) 1073753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Constant::getNullValue(FT->getParamType(i))); 1074753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1075753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we are removing arguments to the function, emit an obnoxious warning. 1076753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FT->getNumParams() < NumActualArgs) { 1077753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!FT->isVarArg()) { 1078753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner errs() << "WARNING: While resolving call to function '" 1079753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner << Callee->getName() << "' arguments were dropped!\n"; 1080753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1081753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add all of the arguments in their promoted form to the arg list. 1082753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) { 1083753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *PTy = getPromotedType((*AI)->getType()); 1084753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PTy != (*AI)->getType()) { 1085753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Must promote to pass through va_arg area! 1086753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction::CastOps opcode = 1087753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CastInst::getCastOpcode(*AI, false, PTy, false); 1088753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Builder->CreateCast(opcode, *AI, PTy, "tmp")); 1089753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1090753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(*AI); 1091753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1092753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1093753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any parameter attributes. 1094753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) 1095753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); 1096753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1097753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1098753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1099753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1100753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes FnAttrs = CallerPAL.getFnAttributes()) 1101753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(~0, FnAttrs)); 1102753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1103753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NewRetTy->isVoidTy()) 1104753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->setName(""); // Void type should not have a name. 1105753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1106753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &NewCallerPAL = AttrListPtr::get(attrVec.begin(), 1107753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.end()); 1108753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1109753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *NC; 1110753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 1111ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC = Builder->CreateInvoke(Callee, II->getNormalDest(), 1112ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman II->getUnwindDest(), Args.begin(), Args.end()); 1113ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC->takeName(II); 1114753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NC)->setCallingConv(II->getCallingConv()); 1115753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NC)->setAttributes(NewCallerPAL); 1116753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1117753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallInst *CI = cast<CallInst>(Caller); 1118ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC = Builder->CreateCall(Callee, Args.begin(), Args.end()); 1119ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC->takeName(CI); 1120753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI->isTailCall()) 1121753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setTailCall(); 1122753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setCallingConv(CI->getCallingConv()); 1123753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setAttributes(NewCallerPAL); 1124753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1125753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1126753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert a cast of the return type as necessary. 1127753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *NV = NC; 1128753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (OldRetTy != NV->getType() && !Caller->use_empty()) { 1129753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!NV->getType()->isVoidTy()) { 1130ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner Instruction::CastOps opcode = 1131ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner CastInst::getCastOpcode(NC, false, OldRetTy, false); 1132753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NV = NC = CastInst::Create(opcode, NC, OldRetTy, "tmp"); 1133a311c34d2af7c750f016ef5e4c41bee77a1dfac7Eli Friedman NC->setDebugLoc(Caller->getDebugLoc()); 1134753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1135753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If this is an invoke instruction, we should insert it after the first 1136753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // non-phi, instruction in the normal successor block. 1137753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 1138753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator I = II->getNormalDest()->getFirstNonPHI(); 1139753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(NC, *I); 1140753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1141ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner // Otherwise, it's a call, just insert cast right after the call. 1142753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(NC, *Caller); 1143753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1144753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Worklist.AddUsersToWorkList(*Caller); 1145753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1146753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NV = UndefValue::get(Caller->getType()); 1147753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1148753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1149753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1150753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty()) 11513e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman ReplaceInstUsesWith(*Caller, NV); 11523e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman 1153753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner EraseInstFromFunction(*Caller); 1154753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 1155753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 1156753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1157753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// transformCallThroughTrampoline - Turn a call to a function created by the 1158753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// init_trampoline intrinsic into a direct call to the underlying function. 1159753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 1160753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::transformCallThroughTrampoline(CallSite CS) { 1161753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Callee = CS.getCalledValue(); 1162753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const PointerType *PTy = cast<PointerType>(Callee->getType()); 1163753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 1164753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &Attrs = CS.getAttributes(); 1165753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1166753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the call already has the 'nest' attribute somewhere then give up - 1167753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // otherwise 'nest' would occur twice after splicing in the chain. 1168753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attrs.hasAttrSomewhere(Attribute::Nest)) 1169753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 1170753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1171753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner IntrinsicInst *Tramp = 1172753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<IntrinsicInst>(cast<BitCastInst>(Callee)->getOperand(0)); 1173753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1174a399781289092fcdceb58b21174229f4373c4191Gabor Greif Function *NestF =cast<Function>(Tramp->getArgOperand(1)->stripPointerCasts()); 1175753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const PointerType *NestFPTy = cast<PointerType>(NestF->getType()); 1176753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const FunctionType *NestFTy = cast<FunctionType>(NestFPTy->getElementType()); 1177753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1178753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &NestAttrs = NestF->getAttributes(); 1179753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!NestAttrs.isEmpty()) { 1180753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NestIdx = 1; 1181753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *NestTy = 0; 1182753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes NestAttr = Attribute::None; 1183753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1184753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Look for a parameter marked with the 'nest' attribute. 1185753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (FunctionType::param_iterator I = NestFTy->param_begin(), 1186753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = NestFTy->param_end(); I != E; ++NestIdx, ++I) 1187753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestAttrs.paramHasAttr(NestIdx, Attribute::Nest)) { 1188753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Record the parameter type and any other attributes. 1189753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestTy = *I; 1190753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestAttr = NestAttrs.getParamAttributes(NestIdx); 1191753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1192753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1193753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1194753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestTy) { 1195753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Caller = CS.getInstruction(); 1196753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<Value*> NewArgs; 1197753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.reserve(unsigned(CS.arg_end()-CS.arg_begin())+1); 1198753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1199753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SmallVector<AttributeWithIndex, 8> NewAttrs; 1200753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.reserve(Attrs.getNumSlots() + 1); 1201753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1202753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert the nest argument into the call argument list, which may 1203753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // mean appending it. Likewise for attributes. 1204753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1205753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any result attributes. 1206753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes Attr = Attrs.getRetAttributes()) 1207753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back(AttributeWithIndex::get(0, Attr)); 1208753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1209753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner { 1210753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx = 1; 1211753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); 1212753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner do { 1213753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Idx == NestIdx) { 1214753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the chain argument and attributes. 1215cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *NestVal = Tramp->getArgOperand(2); 1216753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestVal->getType() != NestTy) 1217e6f364b6c44eda14cd4ad54366ea5cc7246b9500Eli Friedman NestVal = Builder->CreateBitCast(NestVal, NestTy, "nest"); 1218753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.push_back(NestVal); 1219753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back(AttributeWithIndex::get(NestIdx, NestAttr)); 1220753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1221753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1222753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (I == E) 1223753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1224753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1225753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the original argument and attributes. 1226753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.push_back(*I); 1227753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes Attr = Attrs.getParamAttributes(Idx)) 1228753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back 1229753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner (AttributeWithIndex::get(Idx + (Idx >= NestIdx), Attr)); 1230753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1231753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ++Idx, ++I; 1232753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } while (1); 1233753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1234753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1235753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any function attributes. 1236753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes Attr = Attrs.getFnAttributes()) 1237753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back(AttributeWithIndex::get(~0, Attr)); 1238753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1239753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The trampoline may have been bitcast to a bogus type (FTy). 1240753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Handle this by synthesizing a new function type, equal to FTy 1241753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // with the chain parameter inserted. 1242753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1243753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<const Type*> NewTypes; 1244753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.reserve(FTy->getNumParams()+1); 1245753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1246753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert the chain's type into the list of parameter types, which may 1247753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // mean appending it. 1248753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner { 1249753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx = 1; 1250753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner FunctionType::param_iterator I = FTy->param_begin(), 1251753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = FTy->param_end(); 1252753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1253753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner do { 1254753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Idx == NestIdx) 1255753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the chain's type. 1256753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.push_back(NestTy); 1257753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1258753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (I == E) 1259753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1260753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1261753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the original type. 1262753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.push_back(*I); 1263753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1264753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ++Idx, ++I; 1265753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } while (1); 1266753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1267753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1268753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the trampoline call with a direct call. Let the generic 1269753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // code sort out any function type mismatches. 12700c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher FunctionType *NewFTy = FunctionType::get(FTy->getReturnType(), NewTypes, 1271753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner FTy->isVarArg()); 1272753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *NewCallee = 1273753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestF->getType() == PointerType::getUnqual(NewFTy) ? 12740c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher NestF : ConstantExpr::getBitCast(NestF, 1275753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(NewFTy)); 1276753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &NewPAL = AttrListPtr::get(NewAttrs.begin(), 1277753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.end()); 1278753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1279753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *NewCaller; 1280753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 1281753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewCaller = InvokeInst::Create(NewCallee, 1282753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->getNormalDest(), II->getUnwindDest(), 128359f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman NewArgs.begin(), NewArgs.end()); 1284753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NewCaller)->setCallingConv(II->getCallingConv()); 1285753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NewCaller)->setAttributes(NewPAL); 1286753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 128759f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman NewCaller = CallInst::Create(NewCallee, NewArgs.begin(), NewArgs.end()); 1288753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (cast<CallInst>(Caller)->isTailCall()) 1289753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)->setTailCall(); 1290753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)-> 1291753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner setCallingConv(cast<CallInst>(Caller)->getCallingConv()); 1292753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)->setAttributes(NewPAL); 1293753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 129459f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman 129559f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman return NewCaller; 1296753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1297753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1298753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1299753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the trampoline call with a direct call. Since there is no 'nest' 1300753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // parameter, there is no need to adjust the argument list. Let the generic 1301753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // code sort out any function type mismatches. 1302753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *NewCallee = 13030c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher NestF->getType() == PTy ? NestF : 1304753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantExpr::getBitCast(NestF, PTy); 1305753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.setCalledFunction(NewCallee); 1306753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return CS.getInstruction(); 1307753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 13080c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 1309