InstCombineCalls.cpp revision 830f3f205d8a77acbf8838fe78c2c733ca924d9c
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" 19753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerusing namespace llvm; 20753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 21753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// getPromotedType - Return the specified type promoted as it would be to pass 22753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// though a va_arg area. 23753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerstatic const Type *getPromotedType(const Type *Ty) { 24753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (const IntegerType* ITy = dyn_cast<IntegerType>(Ty)) { 25753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ITy->getBitWidth() < 32) 26753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Type::getInt32Ty(Ty->getContext()); 27753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 28753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Ty; 29753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 30753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 31753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// EnforceKnownAlignment - If the specified pointer points to an object that 32753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// we control, modify the object's alignment to PrefAlign. This isn't 33753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// often possible though. If alignment is important, a more reliable approach 34753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// is to simply align all global variables and allocation instructions to 35753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// their preferred alignment from the beginning. 36753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// 37753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerstatic unsigned EnforceKnownAlignment(Value *V, 38753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Align, unsigned PrefAlign) { 39753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 40753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner User *U = dyn_cast<User>(V); 41753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!U) return Align; 42753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 43753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner switch (Operator::getOpcode(U)) { 44753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner default: break; 45753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Instruction::BitCast: 46753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EnforceKnownAlignment(U->getOperand(0), Align, PrefAlign); 47753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Instruction::GetElementPtr: { 48753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If all indexes are zero, it is just the alignment of the base pointer. 49753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool AllZeroOperands = true; 50753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (User::op_iterator i = U->op_begin() + 1, e = U->op_end(); i != e; ++i) 51753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!isa<Constant>(*i) || 52753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !cast<Constant>(*i)->isNullValue()) { 53753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AllZeroOperands = false; 54753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 55753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 56753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 57753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (AllZeroOperands) { 58753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Treat this like a bitcast. 59753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EnforceKnownAlignment(U->getOperand(0), Align, PrefAlign); 60753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 61753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 62753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 63753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 64753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 65753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GlobalValue *GV = dyn_cast<GlobalValue>(V)) { 66753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If there is a large requested alignment and we can, bump up the alignment 67753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // of the global. 68753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!GV->isDeclaration()) { 69753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GV->getAlignment() >= PrefAlign) 70753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Align = GV->getAlignment(); 71753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner else { 72753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner GV->setAlignment(PrefAlign); 73753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Align = PrefAlign; 74753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 75753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 76753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else if (AllocaInst *AI = dyn_cast<AllocaInst>(V)) { 77753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If there is a requested alignment and if this is an alloca, round up. 78753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (AI->getAlignment() >= PrefAlign) 79753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Align = AI->getAlignment(); 80753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner else { 81753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AI->setAlignment(PrefAlign); 82753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Align = PrefAlign; 83753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 84753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 85753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 86753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Align; 87753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 88753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 89753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// GetOrEnforceKnownAlignment - If the specified pointer has an alignment that 90753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// we can determine, return it, otherwise return 0. If PrefAlign is specified, 91753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// and it is more than the alignment of the ultimate object, see if we can 92753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// increase the alignment of the ultimate object, making this check succeed. 93753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerunsigned InstCombiner::GetOrEnforceKnownAlignment(Value *V, 94753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned PrefAlign) { 95753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned BitWidth = TD ? TD->getTypeSizeInBits(V->getType()) : 96753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner sizeof(PrefAlign) * CHAR_BIT; 97753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask = APInt::getAllOnesValue(BitWidth); 98753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownZero(BitWidth, 0), KnownOne(BitWidth, 0); 99753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ComputeMaskedBits(V, Mask, KnownZero, KnownOne); 100753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned TrailZ = KnownZero.countTrailingOnes(); 101753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Align = 1u << std::min(BitWidth - 1, TrailZ); 102753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 103753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PrefAlign > Align) 104753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Align = EnforceKnownAlignment(V, Align, PrefAlign); 105753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 106753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // We don't need to make any adjustment. 107753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Align; 108753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 109753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 110753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::SimplifyMemTransfer(MemIntrinsic *MI) { 111753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned DstAlign = GetOrEnforceKnownAlignment(MI->getOperand(1)); 112753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned SrcAlign = GetOrEnforceKnownAlignment(MI->getOperand(2)); 113753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned MinAlign = std::min(DstAlign, SrcAlign); 114753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned CopyAlign = MI->getAlignment(); 115753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 116753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CopyAlign < MinAlign) { 117753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MI->setAlignment(ConstantInt::get(MI->getAlignmentType(), 118753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MinAlign, false)); 119753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 120753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 121753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 122753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If MemCpyInst length is 1/2/4/8 bytes then replace memcpy with 123753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // load/store. 124753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt *MemOpLength = dyn_cast<ConstantInt>(MI->getOperand(3)); 125753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemOpLength == 0) return 0; 126753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 127753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Source and destination pointer types are always "i8*" for intrinsic. See 128753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // if the size is something we can handle with a single primitive load/store. 129753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // A single load+store correctly handles overlapping memory in the memmove 130753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // case. 131753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Size = MemOpLength->getZExtValue(); 132753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Size == 0) return MI; // Delete this mem transfer. 133753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 134753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Size > 8 || (Size&(Size-1))) 135753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; // If not 1/2/4/8 bytes, exit. 136753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 137753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Use an integer load+store unless we can find something better. 138753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Type *NewPtrTy = 139753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(IntegerType::get(MI->getContext(), Size<<3)); 140753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 141753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Memcpy forces the use of i8* for the source and destination. That means 142753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // that if you're using memcpy to move one double around, you'll get a cast 143753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // from double* to i8*. We'd much rather use a double load+store rather than 144753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // an i64 load+store, here because this improves the odds that the source or 145753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // dest address will be promotable. See if we can find a better type than the 146753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // integer datatype. 147753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *StrippedDest = MI->getOperand(1)->stripPointerCasts(); 148753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (StrippedDest != MI->getOperand(1)) { 149753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *SrcETy = cast<PointerType>(StrippedDest->getType()) 150753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ->getElementType(); 151753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (TD && SrcETy->isSized() && TD->getTypeStoreSize(SrcETy) == Size) { 152753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The SrcETy might be something like {{{double}}} or [1 x double]. Rip 153753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // down through these levels if so. 154753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner while (!SrcETy->isSingleValueType()) { 155753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (const StructType *STy = dyn_cast<StructType>(SrcETy)) { 156753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (STy->getNumElements() == 1) 157753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SrcETy = STy->getElementType(0); 158753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner else 159753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 160753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else if (const ArrayType *ATy = dyn_cast<ArrayType>(SrcETy)) { 161753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ATy->getNumElements() == 1) 162753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SrcETy = ATy->getElementType(); 163753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner else 164753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 165753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else 166753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 167753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 168753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 169753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (SrcETy->isSingleValueType()) 170753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewPtrTy = PointerType::getUnqual(SrcETy); 171753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 172753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 173753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 174753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 175753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the memcpy/memmove provides better alignment info than we can 176753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // infer, use it. 177753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SrcAlign = std::max(SrcAlign, CopyAlign); 178753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner DstAlign = std::max(DstAlign, CopyAlign); 179753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 180753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Src = Builder->CreateBitCast(MI->getOperand(2), NewPtrTy); 181753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Dest = Builder->CreateBitCast(MI->getOperand(1), NewPtrTy); 182753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *L = new LoadInst(Src, "tmp", false, SrcAlign); 183753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(L, *MI); 184753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(new StoreInst(L, Dest, false, DstAlign), *MI); 185753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 186753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Set the size of the copy to 0, it will be deleted on the next iteration. 187753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MI->setOperand(3, Constant::getNullValue(MemOpLength->getType())); 188753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 189753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 190753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 191753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::SimplifyMemSet(MemSetInst *MI) { 192753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Alignment = GetOrEnforceKnownAlignment(MI->getDest()); 193753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MI->getAlignment() < Alignment) { 194753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MI->setAlignment(ConstantInt::get(MI->getAlignmentType(), 195753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Alignment, false)); 196753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 197753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 198753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 199753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Extract the length and alignment and fill if they are constant. 200753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt *LenC = dyn_cast<ConstantInt>(MI->getLength()); 201753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt *FillC = dyn_cast<ConstantInt>(MI->getValue()); 2028c65f6e71c1d46d823b9a884819992a9255edd54Benjamin Kramer if (!LenC || !FillC || !FillC->getType()->isInteger(8)) 203753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 204753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint64_t Len = LenC->getZExtValue(); 205753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Alignment = MI->getAlignment(); 206753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 207753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the length is zero, this is a no-op 208753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Len == 0) return MI; // memset(d,c,0,a) -> noop 209753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 210753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // memset(s,c,n) -> store s, c (for n=1,2,4,8) 211753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Len <= 8 && isPowerOf2_32((uint32_t)Len)) { 212753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *ITy = IntegerType::get(MI->getContext(), Len*8); // n=1 -> i8. 213753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 214753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Dest = MI->getDest(); 215753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Dest = Builder->CreateBitCast(Dest, PointerType::getUnqual(ITy)); 216753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 217753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Alignment 0 is identity for alignment 1 for memset, but not store. 218753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Alignment == 0) Alignment = 1; 219753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 220753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Extract the fill value and store. 221753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint64_t Fill = FillC->getZExtValue()*0x0101010101010101ULL; 222753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(new StoreInst(ConstantInt::get(ITy, Fill), 223753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Dest, false, Alignment), *MI); 224753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 225753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Set the size of the copy to 0, it will be deleted on the next iteration. 226753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner MI->setLength(Constant::getNullValue(LenC->getType())); 227753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return MI; 228753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 229753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 230753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 231753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 232753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 233753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 234753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// visitCallInst - CallInst simplification. This mostly only handles folding 235753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// of intrinsic instructions. For normal calls, it allows visitCallSite to do 236753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// the heavy lifting. 237753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// 238753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitCallInst(CallInst &CI) { 239753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isFreeCall(&CI)) 240753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitFree(CI); 241753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 242753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the caller function is nounwind, mark the call as nounwind, even if the 243753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // callee isn't. 244753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI.getParent()->getParent()->doesNotThrow() && 245753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !CI.doesNotThrow()) { 246753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CI.setDoesNotThrow(); 247753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return &CI; 248753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 249753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 250753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner IntrinsicInst *II = dyn_cast<IntrinsicInst>(&CI); 251753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!II) return visitCallSite(&CI); 252753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 253753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Intrinsics cannot occur in an invoke, so handle them here instead of in 254753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // visitCallSite. 255753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemIntrinsic *MI = dyn_cast<MemIntrinsic>(II)) { 256753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool Changed = false; 257753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 258753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // memmove/cpy/set of zero bytes is a noop. 259753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Constant *NumBytes = dyn_cast<Constant>(MI->getLength())) { 260753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NumBytes->isNullValue()) return EraseInstFromFunction(CI); 261753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 262753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ConstantInt *CI = dyn_cast<ConstantInt>(NumBytes)) 263753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI->getZExtValue() == 1) { 264753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the instruction with just byte operations. We would 265753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // transform other cases to loads/stores, but we don't know if 266753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // alignment is sufficient. 267753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 268753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 269753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 270753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we have a memmove and the source operation is a constant global, 271753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // then the source and dest pointers can't alias, so we can change this 272753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // into a call to memcpy. 273753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemMoveInst *MMI = dyn_cast<MemMoveInst>(MI)) { 274753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GlobalVariable *GVSrc = dyn_cast<GlobalVariable>(MMI->getSource())) 275753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GVSrc->isConstant()) { 276753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Module *M = CI.getParent()->getParent()->getParent(); 277753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Intrinsic::ID MemCpyID = Intrinsic::memcpy; 278753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *Tys[1]; 279753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Tys[0] = CI.getOperand(3)->getType(); 280753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CI.setOperand(0, 281753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Intrinsic::getDeclaration(M, MemCpyID, Tys, 1)); 282753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 283753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 284753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 285753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 286753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MemTransferInst *MTI = dyn_cast<MemTransferInst>(MI)) { 287753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // memmove(x,x,size) -> noop. 288753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (MTI->getSource() == MTI->getDest()) 289753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 290753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 291753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 292753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we can determine a pointer alignment that is bigger than currently 293753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // set, update the alignment. 294753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<MemTransferInst>(MI)) { 295753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Instruction *I = SimplifyMemTransfer(MI)) 296753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return I; 297753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else if (MemSetInst *MSI = dyn_cast<MemSetInst>(MI)) { 298753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Instruction *I = SimplifyMemSet(MSI)) 299753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return I; 300753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 301753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 302753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Changed) return II; 303753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 304753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 305753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner switch (II->getIntrinsicID()) { 306753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner default: break; 307753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::bswap: 308753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // bswap(bswap(x)) -> x 309753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *Operand = dyn_cast<IntrinsicInst>(II->getOperand(1))) 310753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Operand->getIntrinsicID() == Intrinsic::bswap) 311753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, Operand->getOperand(1)); 312753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 313753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // bswap(trunc(bswap(x))) -> trunc(lshr(x, c)) 314753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (TruncInst *TI = dyn_cast<TruncInst>(II->getOperand(1))) { 315753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *Operand = dyn_cast<IntrinsicInst>(TI->getOperand(0))) 316753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Operand->getIntrinsicID() == Intrinsic::bswap) { 317753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned C = Operand->getType()->getPrimitiveSizeInBits() - 318753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner TI->getType()->getPrimitiveSizeInBits(); 319753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *CV = ConstantInt::get(Operand->getType(), C); 320753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *V = Builder->CreateLShr(Operand->getOperand(1), CV); 321753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return new TruncInst(V, TI->getType()); 322753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 323753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 324753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 325753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 326753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::powi: 327753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ConstantInt *Power = dyn_cast<ConstantInt>(II->getOperand(2))) { 328753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, 0) -> 1.0 329753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isZero()) 330753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantFP::get(CI.getType(), 1.0)); 331753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, 1) -> x 332753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isOne()) 333753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, II->getOperand(1)); 334753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // powi(x, -1) -> 1/x 335753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Power->isAllOnesValue()) 336753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return BinaryOperator::CreateFDiv(ConstantFP::get(CI.getType(), 1.0), 337753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->getOperand(1)); 338753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 339753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 340753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::cttz: { 341753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If all bits below the first known one are known zero, 342753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // this value is constant. 343753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const IntegerType *IT = cast<IntegerType>(II->getOperand(1)->getType()); 344753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 345753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownZero(BitWidth, 0); 346753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownOne(BitWidth, 0); 347753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ComputeMaskedBits(II->getOperand(1), APInt::getAllOnesValue(BitWidth), 348753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner KnownZero, KnownOne); 349753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned TrailingZeros = KnownOne.countTrailingZeros(); 350753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask(APInt::getLowBitsSet(BitWidth, TrailingZeros)); 351753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((Mask & KnownZero) == Mask) 352753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantInt::get(IT, 353753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt(BitWidth, TrailingZeros))); 354753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 355753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 356753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 357753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ctlz: { 358753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If all bits above the first known one are known zero, 359753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // this value is constant. 360753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const IntegerType *IT = cast<IntegerType>(II->getOperand(1)->getType()); 361753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 362753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownZero(BitWidth, 0); 363753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt KnownOne(BitWidth, 0); 364753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ComputeMaskedBits(II->getOperand(1), APInt::getAllOnesValue(BitWidth), 365753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner KnownZero, KnownOne); 366753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned LeadingZeros = KnownOne.countLeadingZeros(); 367753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask(APInt::getHighBitsSet(BitWidth, LeadingZeros)); 368753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((Mask & KnownZero) == Mask) 369753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, ConstantInt::get(IT, 370753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt(BitWidth, LeadingZeros))); 371753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 372753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 373753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 374753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::uadd_with_overflow: { 375753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *LHS = II->getOperand(1), *RHS = II->getOperand(2); 376753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const IntegerType *IT = cast<IntegerType>(II->getOperand(1)->getType()); 377753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner uint32_t BitWidth = IT->getBitWidth(); 378753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt Mask = APInt::getSignBit(BitWidth); 379753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt LHSKnownZero(BitWidth, 0); 380753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt LHSKnownOne(BitWidth, 0); 381753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ComputeMaskedBits(LHS, Mask, LHSKnownZero, LHSKnownOne); 382753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool LHSKnownNegative = LHSKnownOne[BitWidth - 1]; 383753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool LHSKnownPositive = LHSKnownZero[BitWidth - 1]; 384753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 385753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownNegative || LHSKnownPositive) { 386753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt RHSKnownZero(BitWidth, 0); 387753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt RHSKnownOne(BitWidth, 0); 388753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ComputeMaskedBits(RHS, Mask, RHSKnownZero, RHSKnownOne); 389753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool RHSKnownNegative = RHSKnownOne[BitWidth - 1]; 390753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool RHSKnownPositive = RHSKnownZero[BitWidth - 1]; 391753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownNegative && RHSKnownNegative) { 392753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The sign bit is set in both cases: this MUST overflow. 393753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Create a simple add instruction, and insert it into the struct. 394753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Add = BinaryOperator::CreateAdd(LHS, RHS, "", &CI); 395753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Worklist.Add(Add); 396753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 397753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(LHS->getType()),ConstantInt::getTrue(II->getContext()) 398753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 399753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 400753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, Add, 0); 401753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 402753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 403753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (LHSKnownPositive && RHSKnownPositive) { 404753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The sign bit is clear in both cases: this CANNOT overflow. 405753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Create a simple add instruction, and insert it into the struct. 406753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Add = BinaryOperator::CreateNUWAdd(LHS, RHS, "", &CI); 407753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Worklist.Add(Add); 408753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 409753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(LHS->getType()), 410753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 411753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 412753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 413753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, Add, 0); 414753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 415753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 416753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 417753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // FALL THROUGH uadd into sadd 418753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::sadd_with_overflow: 419753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Canonicalize constants into the RHS. 420753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<Constant>(II->getOperand(1)) && 421753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !isa<Constant>(II->getOperand(2))) { 422753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *LHS = II->getOperand(1); 423753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->setOperand(1, II->getOperand(2)); 424753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->setOperand(2, LHS); 425753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 426753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 427753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 428753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X + undef -> undef 429753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<UndefValue>(II->getOperand(2))) 430753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 431753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 432753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ConstantInt *RHS = dyn_cast<ConstantInt>(II->getOperand(2))) { 433753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X + 0 -> {X, false} 434753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHS->isZero()) { 435753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 436753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(II->getOperand(0)->getType()), 437753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 438753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 439753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 440753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, II->getOperand(1), 0); 441753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 442753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 443753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 444753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::usub_with_overflow: 445753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ssub_with_overflow: 446753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // undef - X -> undef 447753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X - undef -> undef 448753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<UndefValue>(II->getOperand(1)) || 449753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner isa<UndefValue>(II->getOperand(2))) 450753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 451753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 452753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ConstantInt *RHS = dyn_cast<ConstantInt>(II->getOperand(2))) { 453753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X - 0 -> {X, false} 454753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHS->isZero()) { 455753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 456753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(II->getOperand(1)->getType()), 457753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 458753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 459753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 460753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, II->getOperand(1), 0); 461753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 462753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 463753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 464753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::umul_with_overflow: 465753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::smul_with_overflow: 466753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Canonicalize constants into the RHS. 467753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<Constant>(II->getOperand(1)) && 468753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !isa<Constant>(II->getOperand(2))) { 469753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *LHS = II->getOperand(1); 470753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->setOperand(1, II->getOperand(2)); 471753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->setOperand(2, LHS); 472753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 473753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 474753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 475753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X * undef -> undef 476753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<UndefValue>(II->getOperand(2))) 477753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, UndefValue::get(II->getType())); 478753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 479753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ConstantInt *RHSI = dyn_cast<ConstantInt>(II->getOperand(2))) { 480753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X*0 -> {0, false} 481753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHSI->isZero()) 482753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return ReplaceInstUsesWith(CI, Constant::getNullValue(II->getType())); 483753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 484753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // X * 1 -> {X, false} 485753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RHSI->equalsInt(1)) { 486753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *V[] = { 487753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(II->getOperand(1)->getType()), 488753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getFalse(II->getContext()) 489753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner }; 490753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *Struct = ConstantStruct::get(II->getContext(), V, 2, false); 491753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return InsertValueInst::Create(Struct, II->getOperand(1), 0); 492753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 493753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 494753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 495753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_lvx: 496753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_lvxl: 497753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse_loadu_ps: 498753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_loadu_pd: 499753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_loadu_dq: 500753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn PPC lvx -> load if the pointer is known aligned. 501753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn X86 loadups -> load if the pointer is known aligned. 502753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GetOrEnforceKnownAlignment(II->getOperand(1), 16) >= 16) { 503753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Ptr = Builder->CreateBitCast(II->getOperand(1), 504753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(II->getType())); 505753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return new LoadInst(Ptr); 506753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 507753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 508753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_stvx: 509753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_stvxl: 510753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn stvx -> store if the pointer is known aligned. 511753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GetOrEnforceKnownAlignment(II->getOperand(2), 16) >= 16) { 512753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *OpPtrTy = 513753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(II->getOperand(1)->getType()); 514753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Ptr = Builder->CreateBitCast(II->getOperand(2), OpPtrTy); 515753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return new StoreInst(II->getOperand(1), Ptr); 516753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 517753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 518753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse_storeu_ps: 519753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_storeu_pd: 520753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse2_storeu_dq: 521753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn X86 storeu -> store if the pointer is known aligned. 522753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (GetOrEnforceKnownAlignment(II->getOperand(1), 16) >= 16) { 523753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *OpPtrTy = 524753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(II->getOperand(2)->getType()); 525753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Ptr = Builder->CreateBitCast(II->getOperand(1), OpPtrTy); 526753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return new StoreInst(II->getOperand(2), Ptr); 527753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 528753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 529753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 530753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::x86_sse_cvttss2si: { 531753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // These intrinsics only demands the 0th element of its input vector. If 532753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // we can simplify the input based on that, do so now. 533753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned VWidth = 534753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<VectorType>(II->getOperand(1)->getType())->getNumElements(); 535753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt DemandedElts(VWidth, 1); 536753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner APInt UndefElts(VWidth, 0); 537753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Value *V = SimplifyDemandedVectorElts(II->getOperand(1), DemandedElts, 538753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefElts)) { 539753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->setOperand(1, V); 540753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return II; 541753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 542753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 543753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 544753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 545753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::ppc_altivec_vperm: 546753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Turn vperm(V1,V2,mask) -> shuffle(V1,V2,mask) if mask is a constant. 547753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ConstantVector *Mask = dyn_cast<ConstantVector>(II->getOperand(3))) { 548753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner assert(Mask->getNumOperands() == 16 && "Bad type for intrinsic!"); 549753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 550753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Check that all of the elements are integer constants or undefs. 551753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool AllEltsOk = true; 552753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != 16; ++i) { 553753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!isa<ConstantInt>(Mask->getOperand(i)) && 554753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !isa<UndefValue>(Mask->getOperand(i))) { 555753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AllEltsOk = false; 556753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 557753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 558753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 559753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 560753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (AllEltsOk) { 561753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Cast the input vectors to byte vectors. 562753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Op0 = Builder->CreateBitCast(II->getOperand(1), Mask->getType()); 563753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Op1 = Builder->CreateBitCast(II->getOperand(2), Mask->getType()); 564753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Result = UndefValue::get(Op0->getType()); 565753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 566753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Only extract each element once. 567753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *ExtractedElts[32]; 568753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner memset(ExtractedElts, 0, sizeof(ExtractedElts)); 569753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 570753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != 16; ++i) { 571753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<UndefValue>(Mask->getOperand(i))) 572753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner continue; 573753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx=cast<ConstantInt>(Mask->getOperand(i))->getZExtValue(); 574753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Idx &= 31; // Match the hardware behavior. 575753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 576753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (ExtractedElts[Idx] == 0) { 577753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ExtractedElts[Idx] = 578753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Builder->CreateExtractElement(Idx < 16 ? Op0 : Op1, 579753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::get(Type::getInt32Ty(II->getContext()), 580753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Idx&15, false), "tmp"); 581753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 582753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 583753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert this value into the result vector. 584753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Result = Builder->CreateInsertElement(Result, ExtractedElts[Idx], 585753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::get(Type::getInt32Ty(II->getContext()), 586753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner i, false), "tmp"); 587753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 588753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return CastInst::Create(Instruction::BitCast, Result, CI.getType()); 589753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 590753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 591753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 592753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 593753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner case Intrinsic::stackrestore: { 594753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the save is right next to the restore, remove the restore. This can 595753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // happen when variable allocas are DCE'd. 596753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *SS = dyn_cast<IntrinsicInst>(II->getOperand(1))) { 597753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (SS->getIntrinsicID() == Intrinsic::stacksave) { 598753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator BI = SS; 599753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (&*++BI == II) 600753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 601753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 602753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 603753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 604753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Scan down this block to see if there is another stack restore in the 605753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // same block without an intervening call/alloca. 606753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator BI = II; 607753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner TerminatorInst *TI = II->getParent()->getTerminator(); 608753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool CannotRemove = false; 609753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (++BI; &*BI != TI; ++BI) { 610753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<AllocaInst>(BI) || isMalloc(BI)) { 611753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CannotRemove = true; 612753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 613753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 614753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CallInst *BCI = dyn_cast<CallInst>(BI)) { 615753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(BCI)) { 616753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If there is a stackrestore below this one, remove this one. 617753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (II->getIntrinsicID() == Intrinsic::stackrestore) 618753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 619753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Otherwise, ignore the intrinsic. 620753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 621753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we found a non-intrinsic call, we can't remove the stack 622753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // restore. 623753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CannotRemove = true; 624753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 625753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 626753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 627753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 628753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 629753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the stack restore is in a return/unwind block and if there are no 630753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // allocas or calls between the restore and the return, nuke the restore. 631753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CannotRemove && (isa<ReturnInst>(TI) || isa<UnwindInst>(TI))) 632753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(CI); 633753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 634753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 635130063207de67ecb36e5c80b9118f3c0817fbc6fEric Christopher case Intrinsic::objectsize: { 63636664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher ConstantInt *Const = cast<ConstantInt>(II->getOperand(2)); 63736664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher const Type *Ty = CI.getType(); 63836664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher 63936664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher // 0 is maximum number of bytes left, 1 is minimum number of bytes left. 64036664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher // TODO: actually add these values, the current return values are "don't 64136664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher // know". 64236664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher if (Const->getZExtValue() == 0) 64336664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher return ReplaceInstUsesWith(CI, Constant::getAllOnesValue(Ty)); 64436664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher else 64536664bfc7a540f1b45e80c966e36b1aea9d29aa3Eric Christopher return ReplaceInstUsesWith(CI, ConstantInt::get(Ty, 0)); 646130063207de67ecb36e5c80b9118f3c0817fbc6fEric Christopher } 647753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 648753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 649753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitCallSite(II); 650753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 651753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 652753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// InvokeInst simplification 653753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 654753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitInvokeInst(InvokeInst &II) { 655753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return visitCallSite(&II); 656753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 657753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 658753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// isSafeToEliminateVarargsCast - If this cast does not affect the value 659753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner/// passed through the varargs area, we can eliminate the use of the cast. 660753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerstatic bool isSafeToEliminateVarargsCast(const CallSite CS, 661753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const CastInst * const CI, 662753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const TargetData * const TD, 663753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const int ix) { 664753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CI->isLosslessCast()) 665753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 666753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 667753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The size of ByVal arguments is derived from the type, so we 668753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // can't change to a type with a different size. If the size were 669753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // passed explicitly we could avoid this check. 670753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CS.paramHasAttr(ix, Attribute::ByVal)) 671753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 672753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 673753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type* SrcTy = 674753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<PointerType>(CI->getOperand(0)->getType())->getElementType(); 675753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type* DstTy = cast<PointerType>(CI->getType())->getElementType(); 676753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!SrcTy->isSized() || !DstTy->isSized()) 677753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 678753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!TD || TD->getTypeAllocSize(SrcTy) != TD->getTypeAllocSize(DstTy)) 679753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 680753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 681753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 682753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 683753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// visitCallSite - Improvements for call and invoke instructions. 684753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 685753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::visitCallSite(CallSite CS) { 686753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool Changed = false; 687753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 688753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the callee is a constexpr cast of a function, attempt to move the cast 689753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // to the arguments of the call/invoke. 690753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (transformConstExprCastCall(CS)) return 0; 691753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 692753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Callee = CS.getCalledValue(); 693753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 694753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Function *CalleeF = dyn_cast<Function>(Callee)) 695753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CalleeF->getCallingConv() != CS.getCallingConv()) { 696753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *OldCall = CS.getInstruction(); 697753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the call and callee calling conventions don't match, this call must 698753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // be unreachable, as the call is undefined. 699753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner new StoreInst(ConstantInt::getTrue(Callee->getContext()), 700753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(Type::getInt1PtrTy(Callee->getContext())), 701753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner OldCall); 702753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If OldCall dues not return void then replaceAllUsesWith undef. 703753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This allows ValueHandlers and custom metadata to adjust itself. 704753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!OldCall->getType()->isVoidTy()) 705753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner OldCall->replaceAllUsesWith(UndefValue::get(OldCall->getType())); 706830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner if (isa<CallInst>(OldCall)) 707753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(*OldCall); 708830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner 709830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner // We cannot remove an invoke, because it would change the CFG, just 710830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner // change the callee to a null pointer. 711830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner cast<InvokeInst>(OldCall)->setOperand(0, 712830f3f205d8a77acbf8838fe78c2c733ca924d9cChris Lattner Constant::getNullValue(CalleeF->getType())); 713753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 714753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 715753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 716753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<ConstantPointerNull>(Callee) || isa<UndefValue>(Callee)) { 717753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This instruction is not reachable, just remove it. We insert a store to 718753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // undef so that we know that this code is not reachable, despite the fact 719753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // that we can't modify the CFG here. 720753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner new StoreInst(ConstantInt::getTrue(Callee->getContext()), 721753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UndefValue::get(Type::getInt1PtrTy(Callee->getContext())), 722753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.getInstruction()); 723753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 724753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If CS dues not return void then replaceAllUsesWith undef. 725753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // This allows ValueHandlers and custom metadata to adjust itself. 726753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CS.getInstruction()->getType()->isVoidTy()) 727753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.getInstruction()-> 728753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner replaceAllUsesWith(UndefValue::get(CS.getInstruction()->getType())); 729753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 730753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(CS.getInstruction())) { 731753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Don't break the CFG, insert a dummy cond branch. 732753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BranchInst::Create(II->getNormalDest(), II->getUnwindDest(), 733753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantInt::getTrue(Callee->getContext()), II); 734753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 735753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return EraseInstFromFunction(*CS.getInstruction()); 736753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 737753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 738753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (BitCastInst *BC = dyn_cast<BitCastInst>(Callee)) 739753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (IntrinsicInst *In = dyn_cast<IntrinsicInst>(BC->getOperand(0))) 740753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (In->getIntrinsicID() == Intrinsic::init_trampoline) 741753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return transformCallThroughTrampoline(CS); 742753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 743753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const PointerType *PTy = cast<PointerType>(Callee->getType()); 744753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 745753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FTy->isVarArg()) { 746753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner int ix = FTy->getNumParams() + (isa<InvokeInst>(Callee) ? 3 : 1); 747753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // See if we can optimize any arguments passed through the varargs area of 748753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // the call. 749753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (CallSite::arg_iterator I = CS.arg_begin()+FTy->getNumParams(), 750753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = CS.arg_end(); I != E; ++I, ++ix) { 751753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CastInst *CI = dyn_cast<CastInst>(*I); 752753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI && isSafeToEliminateVarargsCast(CS, CI, TD, ix)) { 753753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner *I = CI->getOperand(0); 754753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 755753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 756753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 757753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 758753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 759753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<InlineAsm>(Callee) && !CS.doesNotThrow()) { 760753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Inline asm calls cannot throw - mark them 'nounwind'. 761753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.setDoesNotThrow(); 762753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Changed = true; 763753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 764753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 765753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return Changed ? CS.getInstruction() : 0; 766753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 767753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 768753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// transformConstExprCastCall - If the callee is a constexpr cast of a function, 769753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// attempt to move the cast to the arguments of the call/invoke. 770753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 771753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattnerbool InstCombiner::transformConstExprCastCall(CallSite CS) { 772753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!isa<ConstantExpr>(CS.getCalledValue())) return false; 773753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantExpr *CE = cast<ConstantExpr>(CS.getCalledValue()); 774753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CE->getOpcode() != Instruction::BitCast || 775753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !isa<Function>(CE->getOperand(0))) 776753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 777753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Function *Callee = cast<Function>(CE->getOperand(0)); 778753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Caller = CS.getInstruction(); 779753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &CallerPAL = CS.getAttributes(); 780753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 781753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Okay, this is a cast from a function to a different type. Unless doing so 782753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // would cause a type conversion of one of our arguments, change this call to 783753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // be a direct call with arguments casted to the appropriate types. 784753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // 785753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const FunctionType *FT = Callee->getFunctionType(); 786753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *OldRetTy = Caller->getType(); 787753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *NewRetTy = FT->getReturnType(); 788753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 789753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (isa<StructType>(NewRetTy)) 790753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // TODO: Handle multiple return values. 791753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 792753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Check to see if we are changing the return type... 793753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (OldRetTy != NewRetTy) { 794753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Callee->isDeclaration() && 795753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Conversion is ok if changing from one pointer type to another or from 796753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // a pointer to an integer of the same size. 797753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !((isa<PointerType>(OldRetTy) || !TD || 798753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner OldRetTy == TD->getIntPtrType(Caller->getContext())) && 799753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner (isa<PointerType>(NewRetTy) || !TD || 800753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewRetTy == TD->getIntPtrType(Caller->getContext())))) 801753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this return value. 802753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 803753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty() && 804753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // void -> non-void is handled specially 805753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !NewRetTy->isVoidTy() && !CastInst::isCastable(NewRetTy, OldRetTy)) 806753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this return value. 807753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 808753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CallerPAL.isEmpty() && !Caller->use_empty()) { 809753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes RAttrs = CallerPAL.getRetAttributes(); 810753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RAttrs & Attribute::typeIncompatible(NewRetTy)) 811753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Attribute not compatible with transformed value. 812753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 813753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 814753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the callsite is an invoke instruction, and the return value is used by 815753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // a PHI node in a successor, we cannot change the return type of the call 816753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // because there is no place to put the cast instruction (without breaking 817753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // the critical edge). Bail out in this case. 818753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty()) 819753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) 820753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (Value::use_iterator UI = II->use_begin(), E = II->use_end(); 821753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner UI != E; ++UI) 822753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PHINode *PN = dyn_cast<PHINode>(*UI)) 823753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PN->getParent() == II->getNormalDest() || 824753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PN->getParent() == II->getUnwindDest()) 825753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 826753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 827753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 828753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NumActualArgs = unsigned(CS.arg_end()-CS.arg_begin()); 829753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NumCommonArgs = std::min(FT->getNumParams(), NumActualArgs); 830753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 831753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallSite::arg_iterator AI = CS.arg_begin(); 832753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0, e = NumCommonArgs; i != e; ++i, ++AI) { 833753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *ParamTy = FT->getParamType(i); 834753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *ActTy = (*AI)->getType(); 835753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 836753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!CastInst::isCastable(ActTy, ParamTy)) 837753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Cannot transform this parameter value. 838753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 839753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CallerPAL.getParamAttributes(i + 1) 840753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner & Attribute::typeIncompatible(ParamTy)) 841753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Attribute not compatible with transformed value. 842753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 843753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Converting from one pointer type to another or between a pointer and an 844753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // integer of the same size is safe even if we do not have a body. 845753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner bool isConvertible = ActTy == ParamTy || 846753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner (TD && ((isa<PointerType>(ParamTy) || 847753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ParamTy == TD->getIntPtrType(Caller->getContext())) && 848753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner (isa<PointerType>(ActTy) || 849753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ActTy == TD->getIntPtrType(Caller->getContext())))); 850753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Callee->isDeclaration() && !isConvertible) return false; 851753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 852753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 853753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FT->getNumParams() < NumActualArgs && !FT->isVarArg() && 854753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Callee->isDeclaration()) 855753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; // Do not delete arguments unless we have a function body. 856753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 857753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FT->getNumParams() < NumActualArgs && FT->isVarArg() && 858753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner !CallerPAL.isEmpty()) 859753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // In this case we have more arguments than the new function type, but we 860753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // won't be dropping them. Check that these extra arguments have attributes 861753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // that are compatible with being a vararg call argument. 862753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = CallerPAL.getNumSlots(); i; --i) { 863753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CallerPAL.getSlot(i - 1).Index <= FT->getNumParams()) 864753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 865753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes PAttrs = CallerPAL.getSlot(i - 1).Attrs; 866753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PAttrs & Attribute::VarArgsIncompatible) 867753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return false; 868753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 869753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 870753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Okay, we decided that this is a safe thing to do: go ahead and start 871753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // inserting cast instructions as necessary... 872753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<Value*> Args; 873753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.reserve(NumActualArgs); 874753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SmallVector<AttributeWithIndex, 8> attrVec; 875753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.reserve(NumCommonArgs); 876753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 877753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Get any return attributes. 878753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes RAttrs = CallerPAL.getRetAttributes(); 879753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 880753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the return value is not being used, the type may not be compatible 881753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // with the existing attributes. Wipe out any problematic attributes. 882753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner RAttrs &= ~Attribute::typeIncompatible(NewRetTy); 883753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 884753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the new return attributes. 885753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (RAttrs) 886753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(0, RAttrs)); 887753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 888753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner AI = CS.arg_begin(); 889753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = 0; i != NumCommonArgs; ++i, ++AI) { 890753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *ParamTy = FT->getParamType(i); 891753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if ((*AI)->getType() == ParamTy) { 892753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(*AI); 893753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 894753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction::CastOps opcode = CastInst::getCastOpcode(*AI, 895753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner false, ParamTy, false); 896753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Builder->CreateCast(opcode, *AI, ParamTy, "tmp")); 897753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 898753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 899753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any parameter attributes. 900753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) 901753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); 902753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 903753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 904753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the function takes more arguments than the call was taking, add them 905753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // now. 906753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = NumCommonArgs; i != FT->getNumParams(); ++i) 907753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Constant::getNullValue(FT->getParamType(i))); 908753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 909753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If we are removing arguments to the function, emit an obnoxious warning. 910753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (FT->getNumParams() < NumActualArgs) { 911753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!FT->isVarArg()) { 912753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner errs() << "WARNING: While resolving call to function '" 913753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner << Callee->getName() << "' arguments were dropped!\n"; 914753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 915753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add all of the arguments in their promoted form to the arg list. 916753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (unsigned i = FT->getNumParams(); i != NumActualArgs; ++i, ++AI) { 917753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *PTy = getPromotedType((*AI)->getType()); 918753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (PTy != (*AI)->getType()) { 919753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Must promote to pass through va_arg area! 920753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction::CastOps opcode = 921753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CastInst::getCastOpcode(*AI, false, PTy, false); 922753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(Builder->CreateCast(opcode, *AI, PTy, "tmp")); 923753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 924753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.push_back(*AI); 925753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 926753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 927753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any parameter attributes. 928753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes PAttrs = CallerPAL.getParamAttributes(i + 1)) 929753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(i + 1, PAttrs)); 930753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 931753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 932753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 933753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 934753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes FnAttrs = CallerPAL.getFnAttributes()) 935753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.push_back(AttributeWithIndex::get(~0, FnAttrs)); 936753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 937753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NewRetTy->isVoidTy()) 938753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->setName(""); // Void type should not have a name. 939753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 940753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &NewCallerPAL = AttrListPtr::get(attrVec.begin(), 941753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner attrVec.end()); 942753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 943753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *NC; 944753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 945753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NC = InvokeInst::Create(Callee, II->getNormalDest(), II->getUnwindDest(), 946753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Args.begin(), Args.end(), 947753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->getName(), Caller); 948753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NC)->setCallingConv(II->getCallingConv()); 949753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NC)->setAttributes(NewCallerPAL); 950753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 951753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NC = CallInst::Create(Callee, Args.begin(), Args.end(), 952753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->getName(), Caller); 953753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallInst *CI = cast<CallInst>(Caller); 954753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (CI->isTailCall()) 955753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setTailCall(); 956753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setCallingConv(CI->getCallingConv()); 957753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NC)->setAttributes(NewCallerPAL); 958753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 959753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 960753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert a cast of the return type as necessary. 961753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *NV = NC; 962753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (OldRetTy != NV->getType() && !Caller->use_empty()) { 963753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!NV->getType()->isVoidTy()) { 964753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction::CastOps opcode = CastInst::getCastOpcode(NC, false, 965753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner OldRetTy, false); 966753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NV = NC = CastInst::Create(opcode, NC, OldRetTy, "tmp"); 967753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 968753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If this is an invoke instruction, we should insert it after the first 969753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // non-phi, instruction in the normal successor block. 970753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 971753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner BasicBlock::iterator I = II->getNormalDest()->getFirstNonPHI(); 972753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(NC, *I); 973753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 974753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Otherwise, it's a call, just insert cast right after the call instr 975753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner InsertNewInstBefore(NC, *Caller); 976753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 977753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Worklist.AddUsersToWorkList(*Caller); 978753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 979753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NV = UndefValue::get(Caller->getType()); 980753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 981753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 982753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 983753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 984753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->use_empty()) 985753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->replaceAllUsesWith(NV); 986753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 987753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner EraseInstFromFunction(*Caller); 988753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return true; 989753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 990753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 991753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// transformCallThroughTrampoline - Turn a call to a function created by the 992753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// init_trampoline intrinsic into a direct call to the underlying function. 993753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner// 994753a2b464d0628231bd3dce645edd05dfbaa3a06Chris LattnerInstruction *InstCombiner::transformCallThroughTrampoline(CallSite CS) { 995753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *Callee = CS.getCalledValue(); 996753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const PointerType *PTy = cast<PointerType>(Callee->getType()); 997753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 998753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &Attrs = CS.getAttributes(); 999753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1000753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // If the call already has the 'nest' attribute somewhere then give up - 1001753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // otherwise 'nest' would occur twice after splicing in the chain. 1002753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attrs.hasAttrSomewhere(Attribute::Nest)) 1003753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 1004753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1005753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner IntrinsicInst *Tramp = 1006753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<IntrinsicInst>(cast<BitCastInst>(Callee)->getOperand(0)); 1007753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1008753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Function *NestF = cast<Function>(Tramp->getOperand(2)->stripPointerCasts()); 1009753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const PointerType *NestFPTy = cast<PointerType>(NestF->getType()); 1010753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const FunctionType *NestFTy = cast<FunctionType>(NestFPTy->getElementType()); 1011753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1012753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &NestAttrs = NestF->getAttributes(); 1013753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!NestAttrs.isEmpty()) { 1014753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned NestIdx = 1; 1015753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const Type *NestTy = 0; 1016753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Attributes NestAttr = Attribute::None; 1017753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1018753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Look for a parameter marked with the 'nest' attribute. 1019753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner for (FunctionType::param_iterator I = NestFTy->param_begin(), 1020753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = NestFTy->param_end(); I != E; ++NestIdx, ++I) 1021753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestAttrs.paramHasAttr(NestIdx, Attribute::Nest)) { 1022753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Record the parameter type and any other attributes. 1023753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestTy = *I; 1024753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestAttr = NestAttrs.getParamAttributes(NestIdx); 1025753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1026753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1027753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1028753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestTy) { 1029753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *Caller = CS.getInstruction(); 1030753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<Value*> NewArgs; 1031753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.reserve(unsigned(CS.arg_end()-CS.arg_begin())+1); 1032753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1033753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner SmallVector<AttributeWithIndex, 8> NewAttrs; 1034753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.reserve(Attrs.getNumSlots() + 1); 1035753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1036753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert the nest argument into the call argument list, which may 1037753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // mean appending it. Likewise for attributes. 1038753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1039753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any result attributes. 1040753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes Attr = Attrs.getRetAttributes()) 1041753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back(AttributeWithIndex::get(0, Attr)); 1042753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1043753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner { 1044753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx = 1; 1045753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CallSite::arg_iterator I = CS.arg_begin(), E = CS.arg_end(); 1046753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner do { 1047753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Idx == NestIdx) { 1048753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the chain argument and attributes. 1049753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Value *NestVal = Tramp->getOperand(3); 1050753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (NestVal->getType() != NestTy) 1051753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestVal = new BitCastInst(NestVal, NestTy, "nest", Caller); 1052753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.push_back(NestVal); 1053753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back(AttributeWithIndex::get(NestIdx, NestAttr)); 1054753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1055753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1056753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (I == E) 1057753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1058753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1059753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the original argument and attributes. 1060753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.push_back(*I); 1061753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes Attr = Attrs.getParamAttributes(Idx)) 1062753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back 1063753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner (AttributeWithIndex::get(Idx + (Idx >= NestIdx), Attr)); 1064753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1065753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ++Idx, ++I; 1066753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } while (1); 1067753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1068753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1069753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add any function attributes. 1070753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Attributes Attr = Attrs.getFnAttributes()) 1071753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.push_back(AttributeWithIndex::get(~0, Attr)); 1072753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1073753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // The trampoline may have been bitcast to a bogus type (FTy). 1074753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Handle this by synthesizing a new function type, equal to FTy 1075753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // with the chain parameter inserted. 1076753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1077753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner std::vector<const Type*> NewTypes; 1078753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.reserve(FTy->getNumParams()+1); 1079753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1080753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Insert the chain's type into the list of parameter types, which may 1081753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // mean appending it. 1082753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner { 1083753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner unsigned Idx = 1; 1084753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner FunctionType::param_iterator I = FTy->param_begin(), 1085753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner E = FTy->param_end(); 1086753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1087753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner do { 1088753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (Idx == NestIdx) 1089753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the chain's type. 1090753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.push_back(NestTy); 1091753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1092753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (I == E) 1093753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner break; 1094753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1095753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Add the original type. 1096753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewTypes.push_back(*I); 1097753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1098753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ++Idx, ++I; 1099753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } while (1); 1100753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1101753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1102753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the trampoline call with a direct call. Let the generic 1103753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // code sort out any function type mismatches. 1104753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner FunctionType *NewFTy = FunctionType::get(FTy->getReturnType(), NewTypes, 1105753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner FTy->isVarArg()); 1106753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *NewCallee = 1107753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestF->getType() == PointerType::getUnqual(NewFTy) ? 1108753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestF : ConstantExpr::getBitCast(NestF, 1109753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner PointerType::getUnqual(NewFTy)); 1110753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner const AttrListPtr &NewPAL = AttrListPtr::get(NewAttrs.begin(), 1111753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewAttrs.end()); 1112753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1113753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Instruction *NewCaller; 1114753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (InvokeInst *II = dyn_cast<InvokeInst>(Caller)) { 1115753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewCaller = InvokeInst::Create(NewCallee, 1116753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner II->getNormalDest(), II->getUnwindDest(), 1117753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewArgs.begin(), NewArgs.end(), 1118753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->getName(), Caller); 1119753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NewCaller)->setCallingConv(II->getCallingConv()); 1120753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<InvokeInst>(NewCaller)->setAttributes(NewPAL); 1121753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } else { 1122753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NewCaller = CallInst::Create(NewCallee, NewArgs.begin(), NewArgs.end(), 1123753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->getName(), Caller); 1124753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (cast<CallInst>(Caller)->isTailCall()) 1125753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)->setTailCall(); 1126753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)-> 1127753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner setCallingConv(cast<CallInst>(Caller)->getCallingConv()); 1128753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner cast<CallInst>(NewCaller)->setAttributes(NewPAL); 1129753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1130753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner if (!Caller->getType()->isVoidTy()) 1131753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->replaceAllUsesWith(NewCaller); 1132753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Caller->eraseFromParent(); 1133753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Worklist.Remove(Caller); 1134753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return 0; 1135753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1136753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner } 1137753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1138753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // Replace the trampoline call with a direct call. Since there is no 'nest' 1139753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // parameter, there is no need to adjust the argument list. Let the generic 1140753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner // code sort out any function type mismatches. 1141753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner Constant *NewCallee = 1142753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner NestF->getType() == PTy ? NestF : 1143753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner ConstantExpr::getBitCast(NestF, PTy); 1144753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner CS.setCalledFunction(NewCallee); 1145753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner return CS.getInstruction(); 1146753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner} 1147753a2b464d0628231bd3dce645edd05dfbaa3a06Chris Lattner 1148