InstCombineCalls.cpp revision 89d4411cef736898047aa7e3bc159da39cacf8e6
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. 25db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattnerstatic Type *getPromotedType(Type *Ty) { 26db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (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 67db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner 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)) { 79db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner 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()) { 85db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner if (StructType *STy = dyn_cast<StructType>(SrcETy)) { 86753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (STy->getNumElements() == 1) 87753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SrcETy = STy->getElementType(0); 88753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner else 89753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 90db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner } else if (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)) { 145db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner 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; 2205fdd6c8793462549e3593890ec61573da06e3346Jay Foad Type *Tys[3] = { CI.getArgOperand(0)->getType(), 2215fdd6c8793462549e3593890ec61573da06e3346Jay Foad CI.getArgOperand(1)->getType(), 2225fdd6c8793462549e3593890ec61573da06e3346Jay Foad CI.getArgOperand(2)->getType() }; 223eb9a85f09e18b3fe88499710404b38d3a9128f62Benjamin Kramer CI.setCalledFunction(Intrinsic::getDeclaration(M, MemCpyID, Tys)); 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 253db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner 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()); 2698fbbb3980755d74539a0aed02bc18842ed2bd18dJay Foad Offset = TD->getIndexedOffset(GEP->getPointerOperandType(), Ops); 270783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 271783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Op1 = GEP->getPointerOperand()->stripPointerCasts(); 272783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 273783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // Make sure we're not a constant offset from an external 274783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // global. 275783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op1)) 276783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (!GV->hasDefinitiveInitializer()) break; 277783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer } 278783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 27926d0e892e36e8dcc4248a8ccc8bedbd2d235eaeeEric Christopher // If we've stripped down to a single global variable that we 28026d0e892e36e8dcc4248a8ccc8bedbd2d235eaeeEric Christopher // can know the size of then just return that. 281415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Op1)) { 282415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher if (GV->hasDefinitiveInitializer()) { 283415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher Constant *C = GV->getInitializer(); 284783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Size = TD->getTypeAllocSize(C->getType()); 285415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher } else { 286a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng // Can't determine size of the GV. 287783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Constant *RetVal = ConstantInt::get(ReturnTy, DontKnow); 288415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher return ReplaceInstUsesWith(CI, RetVal); 289415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher } 290a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng } else if (AllocaInst *AI = dyn_cast<AllocaInst>(Op1)) { 291a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng // Get alloca size. 292a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng if (AI->getAllocatedType()->isSized()) { 293783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Size = TD->getTypeAllocSize(AI->getAllocatedType()); 294a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng if (AI->isArrayAllocation()) { 295a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng const ConstantInt *C = dyn_cast<ConstantInt>(AI->getArraySize()); 296a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng if (!C) break; 297783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Size *= C->getZExtValue(); 298a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng } 299a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng } 300687fed3530366b03a9a6433df0c032247ab17d5bEvan Cheng } else if (CallInst *MI = extractMallocCall(Op1)) { 301240d42d18559a1f75558951ada556baaf289c7f0Benjamin Kramer // Get allocation size. 302db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type* MallocType = getMallocAllocatedType(MI); 303783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (MallocType && MallocType->isSized()) 304783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (Value *NElems = getMallocArraySize(MI, TD, true)) 305687fed3530366b03a9a6433df0c032247ab17d5bEvan Cheng if (ConstantInt *NElements = dyn_cast<ConstantInt>(NElems)) 306783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer Size = NElements->getZExtValue() * TD->getTypeAllocSize(MallocType); 307783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer } 308a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng 309a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng // Do not return "I don't know" here. Later optimization passes could 310a862326bc72be147d9f668b84b9187258e690dc4Evan Cheng // make it possible to evaluate objectsize to a constant. 311783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (Size == -1ULL) 312783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer break; 313783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer 314783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer if (Size < Offset) { 315783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // Out of bound reference? Negative index normalized to large 316783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer // index? Just return "I don't know". 317783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer return ReplaceInstUsesWith(CI, ConstantInt::get(ReturnTy, DontKnow)); 318783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer } 319783a5c2b695015a5a2dfc48ee28b9936d0077542Benjamin Kramer return ReplaceInstUsesWith(CI, ConstantInt::get(ReturnTy, Size-Offset)); 320415326b4edcc967dfb03c5ab41923b195e7c3cb1Eric Christopher } 321753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::bswap: 322753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // bswap(bswap(x)) -> x 323cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (IntrinsicInst *Operand = dyn_cast<IntrinsicInst>(II->getArgOperand(0))) 324753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Operand->getIntrinsicID() == Intrinsic::bswap) 325cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return ReplaceInstUsesWith(CI, Operand->getArgOperand(0)); 3260c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 327753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // bswap(trunc(bswap(x))) -> trunc(lshr(x, c)) 328cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (TruncInst *TI = dyn_cast<TruncInst>(II->getArgOperand(0))) { 329753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *Operand = dyn_cast<IntrinsicInst>(TI->getOperand(0))) 330753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Operand->getIntrinsicID() == Intrinsic::bswap) { 331753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned C = Operand->getType()->getPrimitiveSizeInBits() - 332753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner TI->getType()->getPrimitiveSizeInBits(); 333753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *CV = ConstantInt::get(Operand->getType(), C); 334cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *V = Builder->CreateLShr(Operand->getArgOperand(0), CV); 335753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return new TruncInst(V, TI->getType()); 336753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 337753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 3380c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 339753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 340753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::powi: 341cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (ConstantInt *Power = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 342753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, 0) -> 1.0 343753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isZero()) 344753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantFP::get(CI.getType(), 1.0)); 345753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, 1) -> x 346753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isOne()) 347cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return ReplaceInstUsesWith(CI, II->getArgOperand(0)); 348753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, -1) -> 1/x 349753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isAllOnesValue()) 350753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return BinaryOperator::CreateFDiv(ConstantFP::get(CI.getType(), 1.0), 351cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif II->getArgOperand(0)); 352753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 353753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 354753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::cttz: { 355753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If all bits below the first known one are known zero, 356753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // this value is constant. 357db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IT = dyn_cast<IntegerType>(II->getArgOperand(0)->getType()); 358f1ac465b67d5fc11a0d9cd09b98ceb4ffa75dd97Owen Anderson // FIXME: Try to simplify vectors of integers. 359f1ac465b67d5fc11a0d9cd09b98ceb4ffa75dd97Owen Anderson if (!IT) break; 360753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 361753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownZero(BitWidth, 0); 362753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownOne(BitWidth, 0); 363cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif ComputeMaskedBits(II->getArgOperand(0), APInt::getAllOnesValue(BitWidth), 364753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner KnownZero, KnownOne); 365753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned TrailingZeros = KnownOne.countTrailingZeros(); 366753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask(APInt::getLowBitsSet(BitWidth, TrailingZeros)); 367753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((Mask & KnownZero) == Mask) 368753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantInt::get(IT, 369753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt(BitWidth, TrailingZeros))); 3700c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 371753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 372753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 373753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ctlz: { 374753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If all bits above the first known one are known zero, 375753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // this value is constant. 376db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IT = dyn_cast<IntegerType>(II->getArgOperand(0)->getType()); 377f1ac465b67d5fc11a0d9cd09b98ceb4ffa75dd97Owen Anderson // FIXME: Try to simplify vectors of integers. 378f1ac465b67d5fc11a0d9cd09b98ceb4ffa75dd97Owen Anderson if (!IT) break; 379753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 380753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownZero(BitWidth, 0); 381753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownOne(BitWidth, 0); 382cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif ComputeMaskedBits(II->getArgOperand(0), APInt::getAllOnesValue(BitWidth), 383753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner KnownZero, KnownOne); 384753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned LeadingZeros = KnownOne.countLeadingZeros(); 385753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask(APInt::getHighBitsSet(BitWidth, LeadingZeros)); 386753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((Mask & KnownZero) == Mask) 387753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantInt::get(IT, 388753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt(BitWidth, LeadingZeros))); 3890c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 390753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 391753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 392753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::uadd_with_overflow: { 393cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *LHS = II->getArgOperand(0), *RHS = II->getArgOperand(1); 394db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner IntegerType *IT = cast<IntegerType>(II->getArgOperand(0)->getType()); 395753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 396753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask = APInt::getSignBit(BitWidth); 397753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt LHSKnownZero(BitWidth, 0); 398753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt LHSKnownOne(BitWidth, 0); 399753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ComputeMaskedBits(LHS, Mask, LHSKnownZero, LHSKnownOne); 400753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool LHSKnownNegative = LHSKnownOne[BitWidth - 1]; 401753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool LHSKnownPositive = LHSKnownZero[BitWidth - 1]; 402753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 403753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownNegative || LHSKnownPositive) { 404753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt RHSKnownZero(BitWidth, 0); 405753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt RHSKnownOne(BitWidth, 0); 406753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ComputeMaskedBits(RHS, Mask, RHSKnownZero, RHSKnownOne); 407753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool RHSKnownNegative = RHSKnownOne[BitWidth - 1]; 408753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool RHSKnownPositive = RHSKnownZero[BitWidth - 1]; 409753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownNegative && RHSKnownNegative) { 410753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The sign bit is set in both cases: this MUST overflow. 411753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Create a simple add instruction, and insert it into the struct. 41259f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Value *Add = Builder->CreateAdd(LHS, RHS); 41359f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Add->takeName(&CI); 414753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 41559f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman UndefValue::get(LHS->getType()), 41659f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman ConstantInt::getTrue(II->getContext()) 417753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 418db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner StructType *ST = cast<StructType>(II->getType()); 419b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = ConstantStruct::get(ST, V); 420753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, Add, 0); 421753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 42259f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman 423753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownPositive && RHSKnownPositive) { 424753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The sign bit is clear in both cases: this CANNOT overflow. 425753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Create a simple add instruction, and insert it into the struct. 42659f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Value *Add = Builder->CreateNUWAdd(LHS, RHS); 42759f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman Add->takeName(&CI); 428753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 429753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(LHS->getType()), 430753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 431753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 432db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner StructType *ST = cast<StructType>(II->getType()); 433b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = ConstantStruct::get(ST, V); 434753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, Add, 0); 435753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 436753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 437753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 438753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // FALL THROUGH uadd into sadd 439753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::sadd_with_overflow: 440753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Canonicalize constants into the RHS. 441a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif if (isa<Constant>(II->getArgOperand(0)) && 442a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif !isa<Constant>(II->getArgOperand(1))) { 443a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif Value *LHS = II->getArgOperand(0); 444a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(0, II->getArgOperand(1)); 445a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(1, LHS); 446753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 447753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 448753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 449753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X + undef -> undef 4509c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (isa<UndefValue>(II->getArgOperand(1))) 451753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 4520c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 4539c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (ConstantInt *RHS = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 454753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X + 0 -> {X, false} 455753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHS->isZero()) { 456753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 4574fffb345ed20bd144a31c1c1724cffb6f519b814Eli Friedman UndefValue::get(II->getArgOperand(0)->getType()), 458753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 459753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 460b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = 461b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner ConstantStruct::get(cast<StructType>(II->getType()), V); 462cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return InsertValueInst::Create(Struct, II->getArgOperand(0), 0); 463753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 464753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 465753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 466753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::usub_with_overflow: 467753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ssub_with_overflow: 468753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // undef - X -> undef 469753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X - undef -> undef 4709c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (isa<UndefValue>(II->getArgOperand(0)) || 4719c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif isa<UndefValue>(II->getArgOperand(1))) 472753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 4730c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 4749c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (ConstantInt *RHS = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 475753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X - 0 -> {X, false} 476753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHS->isZero()) { 477753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 4789c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif UndefValue::get(II->getArgOperand(0)->getType()), 479753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 480753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 481b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = 482b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner ConstantStruct::get(cast<StructType>(II->getType()), V); 4839c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif return InsertValueInst::Create(Struct, II->getArgOperand(0), 0); 484753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 485753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 486753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 4876b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer case Intrinsic::umul_with_overflow: { 4886b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Value *LHS = II->getArgOperand(0), *RHS = II->getArgOperand(1); 4896b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer unsigned BitWidth = cast<IntegerType>(LHS->getType())->getBitWidth(); 4906b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt Mask = APInt::getAllOnesValue(BitWidth); 4916b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer 4926b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt LHSKnownZero(BitWidth, 0); 4936b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt LHSKnownOne(BitWidth, 0); 4946b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer ComputeMaskedBits(LHS, Mask, LHSKnownZero, LHSKnownOne); 4956b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt RHSKnownZero(BitWidth, 0); 4966b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer APInt RHSKnownOne(BitWidth, 0); 4976b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer ComputeMaskedBits(RHS, Mask, RHSKnownZero, RHSKnownOne); 4986b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer 499d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer // Get the largest possible values for each operand. 500d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer APInt LHSMax = ~LHSKnownZero; 501d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer APInt RHSMax = ~RHSKnownZero; 5026b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer 5036b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer // If multiplying the maximum values does not overflow then we can turn 5046b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer // this into a plain NUW mul. 505d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer bool Overflow; 506d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer LHSMax.umul_ov(RHSMax, Overflow); 507d655e6e9dc0d21c3a2d76494dec255bb027572aeBenjamin Kramer if (!Overflow) { 5086b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Value *Mul = Builder->CreateNUWMul(LHS, RHS, "umul_with_overflow"); 5096b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Constant *V[] = { 5106b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer UndefValue::get(LHS->getType()), 5116b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer Builder->getFalse() 5126b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer }; 513b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = ConstantStruct::get(cast<StructType>(II->getType()),V); 5146b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer return InsertValueInst::Create(Struct, Mul, 0); 5156b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer } 5166b96fe7e146f7eb594e67210c6bef511ad0a2058Benjamin Kramer } // FALL THROUGH 517753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::smul_with_overflow: 518753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Canonicalize constants into the RHS. 519a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif if (isa<Constant>(II->getArgOperand(0)) && 520a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif !isa<Constant>(II->getArgOperand(1))) { 521a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif Value *LHS = II->getArgOperand(0); 522a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(0, II->getArgOperand(1)); 523a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(1, LHS); 524753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 525753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 526753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 527753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X * undef -> undef 5289c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (isa<UndefValue>(II->getArgOperand(1))) 529753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 5300c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 5319c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif if (ConstantInt *RHSI = dyn_cast<ConstantInt>(II->getArgOperand(1))) { 532753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X*0 -> {0, false} 533753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHSI->isZero()) 534753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, Constant::getNullValue(II->getType())); 5350c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 536753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X * 1 -> {X, false} 537753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHSI->equalsInt(1)) { 538753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 539cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif UndefValue::get(II->getArgOperand(0)->getType()), 540753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 541753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 542b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner Constant *Struct = 543b065b06c12dba6001b8140df2744d0c856ef6ea1Chris Lattner ConstantStruct::get(cast<StructType>(II->getType()), V); 544cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif return InsertValueInst::Create(Struct, II->getArgOperand(0), 0); 545753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 546753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 547753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 548753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_lvx: 549753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_lvxl: 550f93f7b2446bec3febc30b7136e18704664bd98ccBill Wendling // Turn PPC lvx -> load if the pointer is known aligned. 551687140c818ba4b896329a83324714140b6580ef8Chris Lattner if (getOrEnforceKnownAlignment(II->getArgOperand(0), 16, TD) >= 16) { 552cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *Ptr = Builder->CreateBitCast(II->getArgOperand(0), 553753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(II->getType())); 554753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return new LoadInst(Ptr); 555753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 556753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 557753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_stvx: 558753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_stvxl: 559753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn stvx -> store if the pointer is known aligned. 560687140c818ba4b896329a83324714140b6580ef8Chris Lattner if (getOrEnforceKnownAlignment(II->getArgOperand(1), 16, TD) >= 16) { 561db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *OpPtrTy = 5622f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif PointerType::getUnqual(II->getArgOperand(0)->getType()); 5632f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif Value *Ptr = Builder->CreateBitCast(II->getArgOperand(1), OpPtrTy); 5642f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif return new StoreInst(II->getArgOperand(0), Ptr); 565753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 566753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 567753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse_storeu_ps: 568753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_storeu_pd: 569753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_storeu_dq: 570753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn X86 storeu -> store if the pointer is known aligned. 571687140c818ba4b896329a83324714140b6580ef8Chris Lattner if (getOrEnforceKnownAlignment(II->getArgOperand(0), 16, TD) >= 16) { 572db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *OpPtrTy = 5732f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif PointerType::getUnqual(II->getArgOperand(1)->getType()); 5742f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif Value *Ptr = Builder->CreateBitCast(II->getArgOperand(0), OpPtrTy); 5752f1ab74e763ba7f10a058f62393d0987f6f7e298Gabor Greif return new StoreInst(II->getArgOperand(1), Ptr); 576753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 577753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 5789cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth 5799cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvtss2si: 5809cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvtss2si64: 5819cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvttss2si: 5829cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse_cvttss2si64: 5839cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvtsd2si: 5849cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvtsd2si64: 5859cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvttsd2si: 5869cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth case Intrinsic::x86_sse2_cvttsd2si64: { 5879cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth // These intrinsics only demand the 0th element of their input vectors. If 588753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // we can simplify the input based on that, do so now. 589753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned VWidth = 5909c68a7bb45ebf71cec5ab4eab7c73aacdfce16dfGabor Greif cast<VectorType>(II->getArgOperand(0)->getType())->getNumElements(); 591753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt DemandedElts(VWidth, 1); 592753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt UndefElts(VWidth, 0); 593a399781289092fcdceb58b21174229f4373c4191Gabor Greif if (Value *V = SimplifyDemandedVectorElts(II->getArgOperand(0), 594a399781289092fcdceb58b21174229f4373c4191Gabor Greif DemandedElts, UndefElts)) { 595a90c5c7605e0d8df96e61d4bf0cc47a7eaea92cfGabor Greif II->setArgOperand(0, V); 596753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 597753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 598753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 599753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 6009cc9f50abc28f08fb6082b5ee4f5cc7e1af4969bChandler Carruth 601ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings 602ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovsxbw: 603ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovsxwd: 604ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovsxdq: 605ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovzxbw: 606ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovzxwd: 607ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings case Intrinsic::x86_sse41_pmovzxdq: { 608aaa7f499c1a2d46491a98d978ef45b0a06d73d31Evan Cheng // pmov{s|z}x ignores the upper half of their input vectors. 609ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings unsigned VWidth = 610ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings cast<VectorType>(II->getArgOperand(0)->getType())->getNumElements(); 611ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings unsigned LowHalfElts = VWidth / 2; 612d116611e0a884e53ea1524de3a1306c1eb447228Stuart Hastings APInt InputDemandedElts(APInt::getBitsSet(VWidth, 0, LowHalfElts)); 613ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings APInt UndefElts(VWidth, 0); 614ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings if (Value *TmpV = SimplifyDemandedVectorElts(II->getArgOperand(0), 615ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings InputDemandedElts, 616ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings UndefElts)) { 617ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings II->setArgOperand(0, TmpV); 618ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings return II; 619ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings } 620ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings break; 621ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings } 622ca1ef485854d668f794bf389154aa371aa2ed535Stuart Hastings 623753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_vperm: 624753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn vperm(V1,V2,mask) -> shuffle(V1,V2,mask) if mask is a constant. 625cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (ConstantVector *Mask = dyn_cast<ConstantVector>(II->getArgOperand(2))) { 626753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner assert(Mask->getNumOperands() == 16 && "Bad type for intrinsic!"); 6270c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 628753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Check that all of the elements are integer constants or undefs. 629753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool AllEltsOk = true; 630753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != 16; ++i) { 6310c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher if (!isa<ConstantInt>(Mask->getOperand(i)) && 632753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !isa<UndefValue>(Mask->getOperand(i))) { 633753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AllEltsOk = false; 634753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 635753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 636753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 6370c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 638753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (AllEltsOk) { 639753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Cast the input vectors to byte vectors. 640a399781289092fcdceb58b21174229f4373c4191Gabor Greif Value *Op0 = Builder->CreateBitCast(II->getArgOperand(0), 641a399781289092fcdceb58b21174229f4373c4191Gabor Greif Mask->getType()); 642a399781289092fcdceb58b21174229f4373c4191Gabor Greif Value *Op1 = Builder->CreateBitCast(II->getArgOperand(1), 643a399781289092fcdceb58b21174229f4373c4191Gabor Greif Mask->getType()); 644753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Result = UndefValue::get(Op0->getType()); 6450c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 646753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Only extract each element once. 647753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *ExtractedElts[32]; 648753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner memset(ExtractedElts, 0, sizeof(ExtractedElts)); 6490c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 650753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != 16; ++i) { 651753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<UndefValue>(Mask->getOperand(i))) 652753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner continue; 653753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx=cast<ConstantInt>(Mask->getOperand(i))->getZExtValue(); 654753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Idx &= 31; // Match the hardware behavior. 6550c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 656753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ExtractedElts[Idx] == 0) { 6570c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher ExtractedElts[Idx] = 6580c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher Builder->CreateExtractElement(Idx < 16 ? Op0 : Op1, 659753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::get(Type::getInt32Ty(II->getContext()), 660753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Idx&15, false), "tmp"); 661753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 6620c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 663753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert this value into the result vector. 664753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Result = Builder->CreateInsertElement(Result, ExtractedElts[Idx], 665753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::get(Type::getInt32Ty(II->getContext()), 666753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner i, false), "tmp"); 667753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 668753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return CastInst::Create(Instruction::BitCast, Result, CI.getType()); 669753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 670753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 671753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 672753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 673364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld1: 674364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld2: 675364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld3: 676364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld4: 677364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld2lane: 678364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld3lane: 679364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vld4lane: 680364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst1: 681364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst2: 682364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst3: 683364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst4: 684364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst2lane: 685364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst3lane: 686364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson case Intrinsic::arm_neon_vst4lane: { 687ae47be1ea023e4b1e6bbbdc4687333eea54c84c8Chris Lattner unsigned MemAlign = getKnownAlignment(II->getArgOperand(0), TD); 688364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson unsigned AlignArg = II->getNumArgOperands() - 1; 689364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson ConstantInt *IntrAlign = dyn_cast<ConstantInt>(II->getArgOperand(AlignArg)); 690364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson if (IntrAlign && IntrAlign->getZExtValue() < MemAlign) { 691364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson II->setArgOperand(AlignArg, 692364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson ConstantInt::get(Type::getInt32Ty(II->getContext()), 693364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson MemAlign, false)); 694364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson return II; 695364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson } 696364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson break; 697364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson } 698364f17c47162a657bb0c3516588cdfc5297664f0Bob Wilson 699753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::stackrestore: { 700753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the save is right next to the restore, remove the restore. This can 701753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // happen when variable allocas are DCE'd. 702cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif if (IntrinsicInst *SS = dyn_cast<IntrinsicInst>(II->getArgOperand(0))) { 703753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (SS->getIntrinsicID() == Intrinsic::stacksave) { 704753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator BI = SS; 705753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (&*++BI == II) 706753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 707753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 708753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 7090c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 710753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Scan down this block to see if there is another stack restore in the 711753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // same block without an intervening call/alloca. 712753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator BI = II; 713753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner TerminatorInst *TI = II->getParent()->getTerminator(); 714753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool CannotRemove = false; 715753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (++BI; &*BI != TI; ++BI) { 716753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<AllocaInst>(BI) || isMalloc(BI)) { 717753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CannotRemove = true; 718753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 719753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 720753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CallInst *BCI = dyn_cast<CallInst>(BI)) { 721753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(BCI)) { 722753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If there is a stackrestore below this one, remove this one. 723753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (II->getIntrinsicID() == Intrinsic::stackrestore) 724753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 725753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Otherwise, ignore the intrinsic. 726753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 727753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we found a non-intrinsic call, we can't remove the stack 728753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // restore. 729753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CannotRemove = true; 730753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 731753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 732753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 733753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 7340c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 735dccc03b2423fe65efb5963ae816b99c24fc53374Bill Wendling // If the stack restore is in a return, resume, or unwind block and if there 736dccc03b2423fe65efb5963ae816b99c24fc53374Bill Wendling // are no allocas or calls between the restore and the return, nuke the 737dccc03b2423fe65efb5963ae816b99c24fc53374Bill Wendling // restore. 738dccc03b2423fe65efb5963ae816b99c24fc53374Bill Wendling if (!CannotRemove && (isa<ReturnInst>(TI) || isa<ResumeInst>(TI) || 739dccc03b2423fe65efb5963ae816b99c24fc53374Bill Wendling isa<UnwindInst>(TI))) 740753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 741753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 742753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 743753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 744753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 745753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitCallSite(II); 746753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 747753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 748753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// InvokeInst simplification 749753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 750753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitInvokeInst(InvokeInst &II) { 751753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitCallSite(&II); 752753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 753753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 7540c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher/// isSafeToEliminateVarargsCast - If this cast does not affect the value 755753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// passed through the varargs area, we can eliminate the use of the cast. 756753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerstatic bool isSafeToEliminateVarargsCast(const CallSite CS, 757753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const CastInst * const CI, 758753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const TargetData * const TD, 759753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const int ix) { 760753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CI->isLosslessCast()) 761753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 762753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 763753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The size of ByVal arguments is derived from the type, so we 764753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // can't change to a type with a different size. If the size were 765753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // passed explicitly we could avoid this check. 766753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CS.paramHasAttr(ix, Attribute::ByVal)) 767753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 768753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 769db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type* SrcTy = 770753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<PointerType>(CI->getOperand(0)->getType())->getElementType(); 771db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type* DstTy = cast<PointerType>(CI->getType())->getElementType(); 772753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!SrcTy->isSized() || !DstTy->isSized()) 773753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 774753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!TD || TD->getTypeAllocSize(SrcTy) != TD->getTypeAllocSize(DstTy)) 775753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 776753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 777753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 778753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 7790b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramernamespace { 7800b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramerclass InstCombineFortifiedLibCalls : public SimplifyFortifiedLibCalls { 7810b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer InstCombiner *IC; 7820b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramerprotected: 7830b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer void replaceCall(Value *With) { 7840b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer NewInstruction = IC->ReplaceInstUsesWith(*CI, With); 7850b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer } 7860b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer bool isFoldable(unsigned SizeCIOp, unsigned SizeArgOp, bool isString) const { 7878143a84c46b8d2fb58eddb46d84dd5d44d03cb6cBenjamin Kramer if (CI->getArgOperand(SizeCIOp) == CI->getArgOperand(SizeArgOp)) 7888143a84c46b8d2fb58eddb46d84dd5d44d03cb6cBenjamin Kramer return true; 789a399781289092fcdceb58b21174229f4373c4191Gabor Greif if (ConstantInt *SizeCI = 790a399781289092fcdceb58b21174229f4373c4191Gabor Greif dyn_cast<ConstantInt>(CI->getArgOperand(SizeCIOp))) { 7910b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer if (SizeCI->isAllOnesValue()) 7920b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer return true; 793b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher if (isString) { 794b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher uint64_t Len = GetStringLength(CI->getArgOperand(SizeArgOp)); 795b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher // If the length is 0 we don't know how long it is and so we can't 796b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher // remove the check. 797b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher if (Len == 0) return false; 798b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher return SizeCI->getZExtValue() >= Len; 799b9b80c326847ccb1b0611e9b6dc4d372ecc158d3Eric Christopher } 800a399781289092fcdceb58b21174229f4373c4191Gabor Greif if (ConstantInt *Arg = dyn_cast<ConstantInt>( 801a399781289092fcdceb58b21174229f4373c4191Gabor Greif CI->getArgOperand(SizeArgOp))) 8029d8f002d8f5a5aba665bb6f85c19eee77810043cEvan Cheng return SizeCI->getZExtValue() >= Arg->getZExtValue(); 8030b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer } 8040b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer return false; 8050b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer } 8060b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramerpublic: 8070b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer InstCombineFortifiedLibCalls(InstCombiner *IC) : IC(IC), NewInstruction(0) { } 8080b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer Instruction *NewInstruction; 8090b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer}; 8100b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer} // end anonymous namespace 8110b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer 81227ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// Try to fold some different type of calls here. 81327ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// Currently we're only working with the checking functions, memcpy_chk, 81427ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// mempcpy_chk, memmove_chk, memset_chk, strcpy_chk, stpcpy_chk, strncpy_chk, 81527ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher// strcat_chk and strncat_chk. 81627ceaa13f820e33c44e29adc2095dd3e1e69d09fEric ChristopherInstruction *InstCombiner::tryOptimizeCall(CallInst *CI, const TargetData *TD) { 81727ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher if (CI->getCalledFunction() == 0) return 0; 81827ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher 8190b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer InstCombineFortifiedLibCalls Simplifier(this); 8200b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer Simplifier.fold(CI, TD); 8210b6cb507385c8bd10b6a51b5e45a9b99d8d94798Benjamin Kramer return Simplifier.NewInstruction; 82227ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher} 82327ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher 824753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// visitCallSite - Improvements for call and invoke instructions. 825753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 826753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitCallSite(CallSite CS) { 827753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool Changed = false; 828753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 829ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner // If the callee is a pointer to a function, attempt to move any casts to the 830ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner // arguments of the call/invoke. 831753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Callee = CS.getCalledValue(); 832ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner if (!isa<Function>(Callee) && transformConstExprCastCall(CS)) 833ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner return 0; 834753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 835753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Function *CalleeF = dyn_cast<Function>(Callee)) 836d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // If the call and callee calling conventions don't match, this call must 837d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // be unreachable, as the call is undefined. 838d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner if (CalleeF->getCallingConv() != CS.getCallingConv() && 839d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // Only do this for calls to a function with a body. A prototype may 840d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // not actually end up matching the implementation's calling conv for a 841d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner // variety of reasons (e.g. it may be written in assembly). 842d569561835b0fa4dbbb0fca1b1f0a8de6c01439aChris Lattner !CalleeF->isDeclaration()) { 843753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *OldCall = CS.getInstruction(); 844753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner new StoreInst(ConstantInt::getTrue(Callee->getContext()), 8450c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher UndefValue::get(Type::getInt1PtrTy(Callee->getContext())), 846753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner OldCall); 847753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If OldCall dues not return void then replaceAllUsesWith undef. 848753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This allows ValueHandlers and custom metadata to adjust itself. 849753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!OldCall->getType()->isVoidTy()) 8503e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman ReplaceInstUsesWith(*OldCall, UndefValue::get(OldCall->getType())); 851830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner if (isa<CallInst>(OldCall)) 852753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(*OldCall); 8530c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 854830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner // We cannot remove an invoke, because it would change the CFG, just 855830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner // change the callee to a null pointer. 856654c06f6457f80428e40810ad251c412462731e4Gabor Greif cast<InvokeInst>(OldCall)->setCalledFunction( 857830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner Constant::getNullValue(CalleeF->getType())); 858753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 859753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 860753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 861753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<ConstantPointerNull>(Callee) || isa<UndefValue>(Callee)) { 862753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This instruction is not reachable, just remove it. We insert a store to 863753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // undef so that we know that this code is not reachable, despite the fact 864753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // that we can't modify the CFG here. 865753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner new StoreInst(ConstantInt::getTrue(Callee->getContext()), 866753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(Type::getInt1PtrTy(Callee->getContext())), 867753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.getInstruction()); 868753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 869cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif // If CS does not return void then replaceAllUsesWith undef. 870753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This allows ValueHandlers and custom metadata to adjust itself. 871753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CS.getInstruction()->getType()->isVoidTy()) 8723e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman ReplaceInstUsesWith(*CS.getInstruction(), 8733e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman UndefValue::get(CS.getInstruction()->getType())); 874753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 875753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(CS.getInstruction())) { 876753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Don't break the CFG, insert a dummy cond branch. 877753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BranchInst::Create(II->getNormalDest(), II->getUnwindDest(), 878753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getTrue(Callee->getContext()), II); 879753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 880753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(*CS.getInstruction()); 881753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 882753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 883753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (BitCastInst *BC = dyn_cast<BitCastInst>(Callee)) 884753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *In = dyn_cast<IntrinsicInst>(BC->getOperand(0))) 885753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (In->getIntrinsicID() == Intrinsic::init_trampoline) 886753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return transformCallThroughTrampoline(CS); 887753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 888db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *PTy = cast<PointerType>(Callee->getType()); 889db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 890753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FTy->isVarArg()) { 891753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner int ix = FTy->getNumParams() + (isa<InvokeInst>(Callee) ? 3 : 1); 892753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // See if we can optimize any arguments passed through the varargs area of 893753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // the call. 894753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(), 895753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = CS.arg_end(); I != E; ++I, ++ix) { 896753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CastInst *CI = dyn_cast<CastInst>(*I); 897753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI && isSafeToEliminateVarargsCast(CS, CI, TD, ix)) { 898753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner *I = CI->getOperand(0); 899753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 900753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 901753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 902753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 903753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 904753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<InlineAsm>(Callee) && !CS.doesNotThrow()) { 905753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Inline asm calls cannot throw - mark them 'nounwind'. 906753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.setDoesNotThrow(); 907753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 908753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 909753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 91027ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher // Try to optimize the call if possible, we require TargetData for most of 91127ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher // this. None of these calls are seen as possibly dead so go ahead and 91227ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher // delete the instruction now. 91327ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher if (CallInst *CI = dyn_cast<CallInst>(CS.getInstruction())) { 91427ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher Instruction *I = tryOptimizeCall(CI, TD); 9157b323a34fa034389e9e439f6c02eeb73e87ab9dbEric Christopher // If we changed something return the result, etc. Otherwise let 9167b323a34fa034389e9e439f6c02eeb73e87ab9dbEric Christopher // the fallthrough check. 9177b323a34fa034389e9e439f6c02eeb73e87ab9dbEric Christopher if (I) return EraseInstFromFunction(*I); 91827ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher } 91927ceaa13f820e33c44e29adc2095dd3e1e69d09fEric Christopher 920753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Changed ? CS.getInstruction() : 0; 921753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 922753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 923753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// transformConstExprCastCall - If the callee is a constexpr cast of a function, 924753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// attempt to move the cast to the arguments of the call/invoke. 925753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 926753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerbool InstCombiner::transformConstExprCastCall(CallSite CS) { 927ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner Function *Callee = 928ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner dyn_cast<Function>(CS.getCalledValue()->stripPointerCasts()); 929ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner if (Callee == 0) 930753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 931753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Caller = CS.getInstruction(); 932753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &CallerPAL = CS.getAttributes(); 933753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 934753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Okay, this is a cast from a function to a different type. Unless doing so 935753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // would cause a type conversion of one of our arguments, change this call to 936753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // be a direct call with arguments casted to the appropriate types. 937753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // 938db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner FunctionType *FT = Callee->getFunctionType(); 939db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *OldRetTy = Caller->getType(); 940db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *NewRetTy = FT->getReturnType(); 941753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 9421df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands if (NewRetTy->isStructTy()) 943753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // TODO: Handle multiple return values. 944753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 945753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Check to see if we are changing the return type... 946753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (OldRetTy != NewRetTy) { 947753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Callee->isDeclaration() && 948753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Conversion is ok if changing from one pointer type to another or from 949753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // a pointer to an integer of the same size. 9501df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands !((OldRetTy->isPointerTy() || !TD || 951753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner OldRetTy == TD->getIntPtrType(Caller->getContext())) && 9521df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands (NewRetTy->isPointerTy() || !TD || 953753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewRetTy == TD->getIntPtrType(Caller->getContext())))) 954753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this return value. 955753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 956753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty() && 957753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // void -> non-void is handled specially 958753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !NewRetTy->isVoidTy() && !CastInst::isCastable(NewRetTy, OldRetTy)) 959753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this return value. 960753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 961753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CallerPAL.isEmpty() && !Caller->use_empty()) { 962753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes RAttrs = CallerPAL.getRetAttributes(); 963753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RAttrs & Attribute::typeIncompatible(NewRetTy)) 964753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Attribute not compatible with transformed value. 965753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 966753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 967753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the callsite is an invoke instruction, and the return value is used by 968753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // a PHI node in a successor, we cannot change the return type of the call 969753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // because there is no place to put the cast instruction (without breaking 970753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // the critical edge). Bail out in this case. 971753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty()) 972753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) 973753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (Value::use_iterator UI = II->use_begin(), E = II->use_end(); 974753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UI != E; ++UI) 975753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PHINode *PN = dyn_cast<PHINode>(*UI)) 976753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PN->getParent() == II->getNormalDest() || 977753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PN->getParent() == II->getUnwindDest()) 978753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 979753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 980753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 981753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin()); 982753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs); 983753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 984753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallSite::arg_iterator AI = CS.arg_begin(); 985753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) { 986db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ParamTy = FT->getParamType(i); 987db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ActTy = (*AI)->getType(); 988753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 989753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CastInst::isCastable(ActTy, ParamTy)) 990753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this parameter value. 991753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 9922b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner unsigned Attrs = CallerPAL.getParamAttributes(i + 1); 9932b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner if (Attrs & Attribute::typeIncompatible(ParamTy)) 994753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Attribute not compatible with transformed value. 9952b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner 9962b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner // If the parameter is passed as a byval argument, then we have to have a 9972b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner // sized type and the sized type has to have the same size as the old type. 9982b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner if (ParamTy != ActTy && (Attrs & Attribute::ByVal)) { 999db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *ParamPTy = dyn_cast<PointerType>(ParamTy); 10002b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner if (ParamPTy == 0 || !ParamPTy->getElementType()->isSized() || TD == 0) 10012b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner return false; 10022b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner 1003db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *CurElTy = cast<PointerType>(ActTy)->getElementType(); 10042b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner if (TD->getTypeAllocSize(CurElTy) != 10052b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner TD->getTypeAllocSize(ParamPTy->getElementType())) 10062b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner return false; 10072b9375e44b0e99ce23d2cef8de78f66347eef88aChris Lattner } 1008753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1009753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Converting from one pointer type to another or between a pointer and an 1010753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // integer of the same size is safe even if we do not have a body. 1011753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool isConvertible = ActTy == ParamTy || 10121df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands (TD && ((ParamTy->isPointerTy() || 1013753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ParamTy == TD->getIntPtrType(Caller->getContext())) && 10141df9859c40492511b8aa4321eb76496005d3b75bDuncan Sands (ActTy->isPointerTy() || 1015753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ActTy == TD->getIntPtrType(Caller->getContext())))); 1016753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Callee->isDeclaration() && !isConvertible) return false; 1017753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1018753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1019091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner if (Callee->isDeclaration()) { 1020091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // Do not delete arguments unless we have a function body. 1021091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner if (FT->getNumParams() < NumActualArgs && !FT->isVarArg()) 1022091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner return false; 1023091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner 1024091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // If the callee is just a declaration, don't change the varargsness of the 1025091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // call. We don't want to introduce a varargs call where one doesn't 1026091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // already exist. 1027db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *APTy = cast<PointerType>(CS.getCalledValue()->getType()); 1028091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner if (FT->isVarArg()!=cast<FunctionType>(APTy->getElementType())->isVarArg()) 1029091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner return false; 1030091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner } 1031091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner 1032753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FT->getNumParams() < NumActualArgs && FT->isVarArg() && 1033753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !CallerPAL.isEmpty()) 1034753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // In this case we have more arguments than the new function type, but we 1035753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // won't be dropping them. Check that these extra arguments have attributes 1036753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // that are compatible with being a vararg call argument. 1037753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = CallerPAL.getNumSlots(); i; --i) { 1038753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CallerPAL.getSlot(i - 1).Index <= FT->getNumParams()) 1039753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1040753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes PAttrs = CallerPAL.getSlot(i - 1).Attrs; 1041753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PAttrs & Attribute::VarArgsIncompatible) 1042753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 1043753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1044753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1045091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner 1046753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Okay, we decided that this is a safe thing to do: go ahead and start 1047091b1e3c74fee17e33aa87478ab07bfbba0a2f78Chris Lattner // inserting cast instructions as necessary. 1048753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<Value*> Args; 1049753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.reserve(NumActualArgs); 1050753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SmallVector<AttributeWithIndex, 8> attrVec; 1051753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.reserve(NumCommonArgs); 1052753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1053753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Get any return attributes. 1054753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes RAttrs = CallerPAL.getRetAttributes(); 1055753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1056753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the return value is not being used, the type may not be compatible 1057753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // with the existing attributes. Wipe out any problematic attributes. 1058753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner RAttrs &= ~Attribute::typeIncompatible(NewRetTy); 1059753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1060753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the new return attributes. 1061753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RAttrs) 1062753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(0, RAttrs)); 1063753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1064753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AI = CS.arg_begin(); 1065753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) { 1066db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *ParamTy = FT->getParamType(i); 1067753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((*AI)->getType() == ParamTy) { 1068753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(*AI); 1069753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1070753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction::CastOps opcode = CastInst::getCastOpcode(*AI, 1071753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner false, ParamTy, false); 1072753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Builder->CreateCast(opcode, *AI, ParamTy, "tmp")); 1073753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1074753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1075753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any parameter attributes. 1076753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) 1077753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); 1078753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1079753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1080753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the function takes more arguments than the call was taking, add them 1081753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // now. 1082753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i) 1083753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Constant::getNullValue(FT->getParamType(i))); 1084753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1085753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we are removing arguments to the function, emit an obnoxious warning. 1086753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FT->getNumParams() < NumActualArgs) { 1087753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!FT->isVarArg()) { 1088753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner errs() << "WARNING: While resolving call to function '" 1089753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner << Callee->getName() << "' arguments were dropped!\n"; 1090753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1091753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add all of the arguments in their promoted form to the arg list. 1092753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) { 1093db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner Type *PTy = getPromotedType((*AI)->getType()); 1094753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PTy != (*AI)->getType()) { 1095753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Must promote to pass through va_arg area! 1096753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction::CastOps opcode = 1097753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CastInst::getCastOpcode(*AI, false, PTy, false); 1098753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Builder->CreateCast(opcode, *AI, PTy, "tmp")); 1099753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1100753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(*AI); 1101753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1102753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1103753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any parameter attributes. 1104753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) 1105753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); 1106753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1107753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1108753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1109753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1110753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes FnAttrs = CallerPAL.getFnAttributes()) 1111753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(~0, FnAttrs)); 1112753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1113753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NewRetTy->isVoidTy()) 1114753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->setName(""); // Void type should not have a name. 1115753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1116753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &NewCallerPAL = AttrListPtr::get(attrVec.begin(), 1117753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.end()); 1118753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1119753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *NC; 1120753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 1121ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC = Builder->CreateInvoke(Callee, II->getNormalDest(), 1122a3efbb15ddd5aa9006564cd79086723640084878Jay Foad II->getUnwindDest(), Args); 1123ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC->takeName(II); 1124753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NC)->setCallingConv(II->getCallingConv()); 1125753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NC)->setAttributes(NewCallerPAL); 1126753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1127753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallInst *CI = cast<CallInst>(Caller); 1128a3efbb15ddd5aa9006564cd79086723640084878Jay Foad NC = Builder->CreateCall(Callee, Args); 1129ef819d0ed8c9df3da633e8c1240e582a7879bb94Eli Friedman NC->takeName(CI); 1130753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI->isTailCall()) 1131753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setTailCall(); 1132753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setCallingConv(CI->getCallingConv()); 1133753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setAttributes(NewCallerPAL); 1134753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1135753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1136753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert a cast of the return type as necessary. 1137753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *NV = NC; 1138753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (OldRetTy != NV->getType() && !Caller->use_empty()) { 1139753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!NV->getType()->isVoidTy()) { 1140ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner Instruction::CastOps opcode = 1141ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner CastInst::getCastOpcode(NC, false, OldRetTy, false); 1142753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NV = NC = CastInst::Create(opcode, NC, OldRetTy, "tmp"); 1143a311c34d2af7c750f016ef5e4c41bee77a1dfac7Eli Friedman NC->setDebugLoc(Caller->getDebugLoc()); 1144753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1145753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If this is an invoke instruction, we should insert it after the first 1146753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // non-phi, instruction in the normal successor block. 1147753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 114889d4411cef736898047aa7e3bc159da39cacf8e6Bill Wendling BasicBlock::iterator I = II->getNormalDest()->getFirstInsertionPt(); 1149753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(NC, *I); 1150753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1151ab215bc371677cdeac6eacb3e890e5af666d1ab4Chris Lattner // Otherwise, it's a call, just insert cast right after the call. 1152753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(NC, *Caller); 1153753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1154753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Worklist.AddUsersToWorkList(*Caller); 1155753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1156753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NV = UndefValue::get(Caller->getType()); 1157753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1158753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1159753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1160753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty()) 11613e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman ReplaceInstUsesWith(*Caller, NV); 11623e22cb9ec30cd9b1be9b0f50e400f512124997e5Eli Friedman 1163753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner EraseInstFromFunction(*Caller); 1164753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 1165753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 1166753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1167753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// transformCallThroughTrampoline - Turn a call to a function created by the 1168753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// init_trampoline intrinsic into a direct call to the underlying function. 1169753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 1170753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::transformCallThroughTrampoline(CallSite CS) { 1171753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Callee = CS.getCalledValue(); 1172db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *PTy = cast<PointerType>(Callee->getType()); 1173db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 1174753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &Attrs = CS.getAttributes(); 1175753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1176753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the call already has the 'nest' attribute somewhere then give up - 1177753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // otherwise 'nest' would occur twice after splicing in the chain. 1178753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attrs.hasAttrSomewhere(Attribute::Nest)) 1179753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 1180753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1181753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner IntrinsicInst *Tramp = 1182753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<IntrinsicInst>(cast<BitCastInst>(Callee)->getOperand(0)); 1183753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1184a399781289092fcdceb58b21174229f4373c4191Gabor Greif Function *NestF =cast<Function>(Tramp->getArgOperand(1)->stripPointerCasts()); 1185db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner PointerType *NestFPTy = cast<PointerType>(NestF->getType()); 1186db125cfaf57cc83e7dd7453de2d509bc8efd0e5eChris Lattner FunctionType *NestFTy = cast<FunctionType>(NestFPTy->getElementType()); 1187753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1188753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &NestAttrs = NestF->getAttributes(); 1189753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!NestAttrs.isEmpty()) { 1190753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NestIdx = 1; 11915fdd6c8793462549e3593890ec61573da06e3346Jay Foad Type *NestTy = 0; 1192753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes NestAttr = Attribute::None; 1193753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1194753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Look for a parameter marked with the 'nest' attribute. 1195753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (FunctionType::param_iterator I = NestFTy->param_begin(), 1196753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = NestFTy->param_end(); I != E; ++NestIdx, ++I) 1197753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestAttrs.paramHasAttr(NestIdx, Attribute::Nest)) { 1198753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Record the parameter type and any other attributes. 1199753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestTy = *I; 1200753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestAttr = NestAttrs.getParamAttributes(NestIdx); 1201753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1202753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1203753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1204753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestTy) { 1205753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Caller = CS.getInstruction(); 1206753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<Value*> NewArgs; 1207753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.reserve(unsigned(CS.arg_end()-CS.arg_begin())+1); 1208753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1209753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SmallVector<AttributeWithIndex, 8> NewAttrs; 1210753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.reserve(Attrs.getNumSlots() + 1); 1211753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1212753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert the nest argument into the call argument list, which may 1213753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // mean appending it. Likewise for attributes. 1214753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1215753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any result attributes. 1216753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes Attr = Attrs.getRetAttributes()) 1217753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back(AttributeWithIndex::get(0, Attr)); 1218753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1219753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner { 1220753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx = 1; 1221753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); 1222753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner do { 1223753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Idx == NestIdx) { 1224753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the chain argument and attributes. 1225cea7ac7c01c6a15eccb2cb7a181df1baf915de0dGabor Greif Value *NestVal = Tramp->getArgOperand(2); 1226753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestVal->getType() != NestTy) 1227e6f364b6c44eda14cd4ad54366ea5cc7246b9500Eli Friedman NestVal = Builder->CreateBitCast(NestVal, NestTy, "nest"); 1228753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.push_back(NestVal); 1229753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back(AttributeWithIndex::get(NestIdx, NestAttr)); 1230753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1231753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1232753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (I == E) 1233753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1234753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1235753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the original argument and attributes. 1236753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.push_back(*I); 1237753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes Attr = Attrs.getParamAttributes(Idx)) 1238753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back 1239753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner (AttributeWithIndex::get(Idx + (Idx >= NestIdx), Attr)); 1240753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1241753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ++Idx, ++I; 1242753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } while (1); 1243753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1244753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1245753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any function attributes. 1246753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes Attr = Attrs.getFnAttributes()) 1247753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back(AttributeWithIndex::get(~0, Attr)); 1248753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1249753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The trampoline may have been bitcast to a bogus type (FTy). 1250753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Handle this by synthesizing a new function type, equal to FTy 1251753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // with the chain parameter inserted. 1252753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 12535fdd6c8793462549e3593890ec61573da06e3346Jay Foad std::vector<Type*> NewTypes; 1254753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.reserve(FTy->getNumParams()+1); 1255753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1256753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert the chain's type into the list of parameter types, which may 1257753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // mean appending it. 1258753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner { 1259753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx = 1; 1260753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner FunctionType::param_iterator I = FTy->param_begin(), 1261753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = FTy->param_end(); 1262753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1263753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner do { 1264753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Idx == NestIdx) 1265753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the chain's type. 1266753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.push_back(NestTy); 1267753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1268753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (I == E) 1269753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1270753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1271753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the original type. 1272753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.push_back(*I); 1273753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1274753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ++Idx, ++I; 1275753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } while (1); 1276753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1277753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1278753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the trampoline call with a direct call. Let the generic 1279753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // code sort out any function type mismatches. 12800c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher FunctionType *NewFTy = FunctionType::get(FTy->getReturnType(), NewTypes, 1281753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner FTy->isVarArg()); 1282753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *NewCallee = 1283753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestF->getType() == PointerType::getUnqual(NewFTy) ? 12840c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher NestF : ConstantExpr::getBitCast(NestF, 1285753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(NewFTy)); 1286753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &NewPAL = AttrListPtr::get(NewAttrs.begin(), 1287753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.end()); 1288753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1289753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *NewCaller; 1290753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 1291753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewCaller = InvokeInst::Create(NewCallee, 1292753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->getNormalDest(), II->getUnwindDest(), 1293a3efbb15ddd5aa9006564cd79086723640084878Jay Foad NewArgs); 1294753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NewCaller)->setCallingConv(II->getCallingConv()); 1295753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NewCaller)->setAttributes(NewPAL); 1296753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1297a3efbb15ddd5aa9006564cd79086723640084878Jay Foad NewCaller = CallInst::Create(NewCallee, NewArgs); 1298753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (cast<CallInst>(Caller)->isTailCall()) 1299753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)->setTailCall(); 1300753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)-> 1301753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner setCallingConv(cast<CallInst>(Caller)->getCallingConv()); 1302753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)->setAttributes(NewPAL); 1303753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 130459f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman 130559f15918c28ad340bb9f0ef78e43c2952a5b5c1cEli Friedman return NewCaller; 1306753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1307753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1308753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1309753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the trampoline call with a direct call. Since there is no 'nest' 1310753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // parameter, there is no need to adjust the argument list. Let the generic 1311753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // code sort out any function type mismatches. 1312753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *NewCallee = 13130c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher NestF->getType() == PTy ? NestF : 1314753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantExpr::getBitCast(NestF, PTy); 1315753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.setCalledFunction(NewCallee); 1316753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return CS.getInstruction(); 1317753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 13180c6a8f9eda4651e0f54a7edf171525d16b1ff3faEric Christopher 1319