1f22ef01cSRoman Divacky //===--- Bitcode/Writer/BitcodeWriter.cpp - Bitcode Writer ----------------===// 2f22ef01cSRoman Divacky // 3f22ef01cSRoman Divacky // The LLVM Compiler Infrastructure 4f22ef01cSRoman Divacky // 5f22ef01cSRoman Divacky // This file is distributed under the University of Illinois Open Source 6f22ef01cSRoman Divacky // License. See LICENSE.TXT for details. 7f22ef01cSRoman Divacky // 8f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 9f22ef01cSRoman Divacky // 10f22ef01cSRoman Divacky // Bitcode writer implementation. 11f22ef01cSRoman Divacky // 12f22ef01cSRoman Divacky //===----------------------------------------------------------------------===// 13f22ef01cSRoman Divacky 14f22ef01cSRoman Divacky #include "llvm/Bitcode/ReaderWriter.h" 15f22ef01cSRoman Divacky #include "llvm/Bitcode/BitstreamWriter.h" 16f22ef01cSRoman Divacky #include "llvm/Bitcode/LLVMBitCodes.h" 17f22ef01cSRoman Divacky #include "ValueEnumerator.h" 18f22ef01cSRoman Divacky #include "llvm/Constants.h" 19f22ef01cSRoman Divacky #include "llvm/DerivedTypes.h" 20f22ef01cSRoman Divacky #include "llvm/InlineAsm.h" 21f22ef01cSRoman Divacky #include "llvm/Instructions.h" 22f22ef01cSRoman Divacky #include "llvm/Module.h" 23f22ef01cSRoman Divacky #include "llvm/Operator.h" 24f22ef01cSRoman Divacky #include "llvm/ValueSymbolTable.h" 2517a519f9SDimitry Andric #include "llvm/ADT/Triple.h" 26dff0c46cSDimitry Andric #include "llvm/Support/CommandLine.h" 27f22ef01cSRoman Divacky #include "llvm/Support/ErrorHandling.h" 28f22ef01cSRoman Divacky #include "llvm/Support/MathExtras.h" 29f22ef01cSRoman Divacky #include "llvm/Support/raw_ostream.h" 302754fe60SDimitry Andric #include "llvm/Support/Program.h" 312754fe60SDimitry Andric #include <cctype> 3217a519f9SDimitry Andric #include <map> 33f22ef01cSRoman Divacky using namespace llvm; 34f22ef01cSRoman Divacky 35dff0c46cSDimitry Andric static cl::opt<bool> 36dff0c46cSDimitry Andric EnablePreserveUseListOrdering("enable-bc-uselist-preserve", 37dff0c46cSDimitry Andric cl::desc("Turn on experimental support for " 38dff0c46cSDimitry Andric "use-list order preservation."), 39dff0c46cSDimitry Andric cl::init(false), cl::Hidden); 40dff0c46cSDimitry Andric 41f22ef01cSRoman Divacky /// These are manifest constants used by the bitcode writer. They do not need to 42f22ef01cSRoman Divacky /// be kept in sync with the reader, but need to be consistent within this file. 43f22ef01cSRoman Divacky enum { 44f22ef01cSRoman Divacky CurVersion = 0, 45f22ef01cSRoman Divacky 46f22ef01cSRoman Divacky // VALUE_SYMTAB_BLOCK abbrev id's. 47f22ef01cSRoman Divacky VST_ENTRY_8_ABBREV = bitc::FIRST_APPLICATION_ABBREV, 48f22ef01cSRoman Divacky VST_ENTRY_7_ABBREV, 49f22ef01cSRoman Divacky VST_ENTRY_6_ABBREV, 50f22ef01cSRoman Divacky VST_BBENTRY_6_ABBREV, 51f22ef01cSRoman Divacky 52f22ef01cSRoman Divacky // CONSTANTS_BLOCK abbrev id's. 53f22ef01cSRoman Divacky CONSTANTS_SETTYPE_ABBREV = bitc::FIRST_APPLICATION_ABBREV, 54f22ef01cSRoman Divacky CONSTANTS_INTEGER_ABBREV, 55f22ef01cSRoman Divacky CONSTANTS_CE_CAST_Abbrev, 56f22ef01cSRoman Divacky CONSTANTS_NULL_Abbrev, 57f22ef01cSRoman Divacky 58f22ef01cSRoman Divacky // FUNCTION_BLOCK abbrev id's. 59f22ef01cSRoman Divacky FUNCTION_INST_LOAD_ABBREV = bitc::FIRST_APPLICATION_ABBREV, 60f22ef01cSRoman Divacky FUNCTION_INST_BINOP_ABBREV, 61f22ef01cSRoman Divacky FUNCTION_INST_BINOP_FLAGS_ABBREV, 62f22ef01cSRoman Divacky FUNCTION_INST_CAST_ABBREV, 63f22ef01cSRoman Divacky FUNCTION_INST_RET_VOID_ABBREV, 64f22ef01cSRoman Divacky FUNCTION_INST_RET_VAL_ABBREV, 65f22ef01cSRoman Divacky FUNCTION_INST_UNREACHABLE_ABBREV 66f22ef01cSRoman Divacky }; 67f22ef01cSRoman Divacky 68f22ef01cSRoman Divacky static unsigned GetEncodedCastOpcode(unsigned Opcode) { 69f22ef01cSRoman Divacky switch (Opcode) { 70f22ef01cSRoman Divacky default: llvm_unreachable("Unknown cast instruction!"); 71f22ef01cSRoman Divacky case Instruction::Trunc : return bitc::CAST_TRUNC; 72f22ef01cSRoman Divacky case Instruction::ZExt : return bitc::CAST_ZEXT; 73f22ef01cSRoman Divacky case Instruction::SExt : return bitc::CAST_SEXT; 74f22ef01cSRoman Divacky case Instruction::FPToUI : return bitc::CAST_FPTOUI; 75f22ef01cSRoman Divacky case Instruction::FPToSI : return bitc::CAST_FPTOSI; 76f22ef01cSRoman Divacky case Instruction::UIToFP : return bitc::CAST_UITOFP; 77f22ef01cSRoman Divacky case Instruction::SIToFP : return bitc::CAST_SITOFP; 78f22ef01cSRoman Divacky case Instruction::FPTrunc : return bitc::CAST_FPTRUNC; 79f22ef01cSRoman Divacky case Instruction::FPExt : return bitc::CAST_FPEXT; 80f22ef01cSRoman Divacky case Instruction::PtrToInt: return bitc::CAST_PTRTOINT; 81f22ef01cSRoman Divacky case Instruction::IntToPtr: return bitc::CAST_INTTOPTR; 82f22ef01cSRoman Divacky case Instruction::BitCast : return bitc::CAST_BITCAST; 83f22ef01cSRoman Divacky } 84f22ef01cSRoman Divacky } 85f22ef01cSRoman Divacky 86f22ef01cSRoman Divacky static unsigned GetEncodedBinaryOpcode(unsigned Opcode) { 87f22ef01cSRoman Divacky switch (Opcode) { 88f22ef01cSRoman Divacky default: llvm_unreachable("Unknown binary instruction!"); 89f22ef01cSRoman Divacky case Instruction::Add: 90f22ef01cSRoman Divacky case Instruction::FAdd: return bitc::BINOP_ADD; 91f22ef01cSRoman Divacky case Instruction::Sub: 92f22ef01cSRoman Divacky case Instruction::FSub: return bitc::BINOP_SUB; 93f22ef01cSRoman Divacky case Instruction::Mul: 94f22ef01cSRoman Divacky case Instruction::FMul: return bitc::BINOP_MUL; 95f22ef01cSRoman Divacky case Instruction::UDiv: return bitc::BINOP_UDIV; 96f22ef01cSRoman Divacky case Instruction::FDiv: 97f22ef01cSRoman Divacky case Instruction::SDiv: return bitc::BINOP_SDIV; 98f22ef01cSRoman Divacky case Instruction::URem: return bitc::BINOP_UREM; 99f22ef01cSRoman Divacky case Instruction::FRem: 100f22ef01cSRoman Divacky case Instruction::SRem: return bitc::BINOP_SREM; 101f22ef01cSRoman Divacky case Instruction::Shl: return bitc::BINOP_SHL; 102f22ef01cSRoman Divacky case Instruction::LShr: return bitc::BINOP_LSHR; 103f22ef01cSRoman Divacky case Instruction::AShr: return bitc::BINOP_ASHR; 104f22ef01cSRoman Divacky case Instruction::And: return bitc::BINOP_AND; 105f22ef01cSRoman Divacky case Instruction::Or: return bitc::BINOP_OR; 106f22ef01cSRoman Divacky case Instruction::Xor: return bitc::BINOP_XOR; 107f22ef01cSRoman Divacky } 108f22ef01cSRoman Divacky } 109f22ef01cSRoman Divacky 1106122f3e6SDimitry Andric static unsigned GetEncodedRMWOperation(AtomicRMWInst::BinOp Op) { 1116122f3e6SDimitry Andric switch (Op) { 1126122f3e6SDimitry Andric default: llvm_unreachable("Unknown RMW operation!"); 1136122f3e6SDimitry Andric case AtomicRMWInst::Xchg: return bitc::RMW_XCHG; 1146122f3e6SDimitry Andric case AtomicRMWInst::Add: return bitc::RMW_ADD; 1156122f3e6SDimitry Andric case AtomicRMWInst::Sub: return bitc::RMW_SUB; 1166122f3e6SDimitry Andric case AtomicRMWInst::And: return bitc::RMW_AND; 1176122f3e6SDimitry Andric case AtomicRMWInst::Nand: return bitc::RMW_NAND; 1186122f3e6SDimitry Andric case AtomicRMWInst::Or: return bitc::RMW_OR; 1196122f3e6SDimitry Andric case AtomicRMWInst::Xor: return bitc::RMW_XOR; 1206122f3e6SDimitry Andric case AtomicRMWInst::Max: return bitc::RMW_MAX; 1216122f3e6SDimitry Andric case AtomicRMWInst::Min: return bitc::RMW_MIN; 1226122f3e6SDimitry Andric case AtomicRMWInst::UMax: return bitc::RMW_UMAX; 1236122f3e6SDimitry Andric case AtomicRMWInst::UMin: return bitc::RMW_UMIN; 1246122f3e6SDimitry Andric } 1256122f3e6SDimitry Andric } 1266122f3e6SDimitry Andric 1276122f3e6SDimitry Andric static unsigned GetEncodedOrdering(AtomicOrdering Ordering) { 1286122f3e6SDimitry Andric switch (Ordering) { 1296122f3e6SDimitry Andric case NotAtomic: return bitc::ORDERING_NOTATOMIC; 1306122f3e6SDimitry Andric case Unordered: return bitc::ORDERING_UNORDERED; 1316122f3e6SDimitry Andric case Monotonic: return bitc::ORDERING_MONOTONIC; 1326122f3e6SDimitry Andric case Acquire: return bitc::ORDERING_ACQUIRE; 1336122f3e6SDimitry Andric case Release: return bitc::ORDERING_RELEASE; 1346122f3e6SDimitry Andric case AcquireRelease: return bitc::ORDERING_ACQREL; 1356122f3e6SDimitry Andric case SequentiallyConsistent: return bitc::ORDERING_SEQCST; 1366122f3e6SDimitry Andric } 137dff0c46cSDimitry Andric llvm_unreachable("Invalid ordering"); 1386122f3e6SDimitry Andric } 1396122f3e6SDimitry Andric 1406122f3e6SDimitry Andric static unsigned GetEncodedSynchScope(SynchronizationScope SynchScope) { 1416122f3e6SDimitry Andric switch (SynchScope) { 1426122f3e6SDimitry Andric case SingleThread: return bitc::SYNCHSCOPE_SINGLETHREAD; 1436122f3e6SDimitry Andric case CrossThread: return bitc::SYNCHSCOPE_CROSSTHREAD; 1446122f3e6SDimitry Andric } 145dff0c46cSDimitry Andric llvm_unreachable("Invalid synch scope"); 1466122f3e6SDimitry Andric } 1476122f3e6SDimitry Andric 14817a519f9SDimitry Andric static void WriteStringRecord(unsigned Code, StringRef Str, 149f22ef01cSRoman Divacky unsigned AbbrevToUse, BitstreamWriter &Stream) { 150f22ef01cSRoman Divacky SmallVector<unsigned, 64> Vals; 151f22ef01cSRoman Divacky 152f22ef01cSRoman Divacky // Code: [strchar x N] 15317a519f9SDimitry Andric for (unsigned i = 0, e = Str.size(); i != e; ++i) { 15417a519f9SDimitry Andric if (AbbrevToUse && !BitCodeAbbrevOp::isChar6(Str[i])) 15517a519f9SDimitry Andric AbbrevToUse = 0; 156f22ef01cSRoman Divacky Vals.push_back(Str[i]); 15717a519f9SDimitry Andric } 158f22ef01cSRoman Divacky 159f22ef01cSRoman Divacky // Emit the finished record. 160f22ef01cSRoman Divacky Stream.EmitRecord(Code, Vals, AbbrevToUse); 161f22ef01cSRoman Divacky } 162f22ef01cSRoman Divacky 163f22ef01cSRoman Divacky // Emit information about parameter attributes. 164f22ef01cSRoman Divacky static void WriteAttributeTable(const ValueEnumerator &VE, 165f22ef01cSRoman Divacky BitstreamWriter &Stream) { 166f22ef01cSRoman Divacky const std::vector<AttrListPtr> &Attrs = VE.getAttributes(); 167f22ef01cSRoman Divacky if (Attrs.empty()) return; 168f22ef01cSRoman Divacky 169f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::PARAMATTR_BLOCK_ID, 3); 170f22ef01cSRoman Divacky 171f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 172f22ef01cSRoman Divacky for (unsigned i = 0, e = Attrs.size(); i != e; ++i) { 173f22ef01cSRoman Divacky const AttrListPtr &A = Attrs[i]; 174f22ef01cSRoman Divacky for (unsigned i = 0, e = A.getNumSlots(); i != e; ++i) { 175f22ef01cSRoman Divacky const AttributeWithIndex &PAWI = A.getSlot(i); 176f22ef01cSRoman Divacky Record.push_back(PAWI.Index); 177f22ef01cSRoman Divacky 178f22ef01cSRoman Divacky // FIXME: remove in LLVM 3.0 179f22ef01cSRoman Divacky // Store the alignment in the bitcode as a 16-bit raw value instead of a 180f22ef01cSRoman Divacky // 5-bit log2 encoded value. Shift the bits above the alignment up by 181f22ef01cSRoman Divacky // 11 bits. 182dff0c46cSDimitry Andric uint64_t FauxAttr = PAWI.Attrs.Raw() & 0xffff; 183f22ef01cSRoman Divacky if (PAWI.Attrs & Attribute::Alignment) 184dff0c46cSDimitry Andric FauxAttr |= (1ull<<16)<< 185dff0c46cSDimitry Andric (((PAWI.Attrs & Attribute::Alignment).Raw()-1) >> 16); 186dff0c46cSDimitry Andric FauxAttr |= (PAWI.Attrs.Raw() & (0x3FFull << 21)) << 11; 187f22ef01cSRoman Divacky 188f22ef01cSRoman Divacky Record.push_back(FauxAttr); 189f22ef01cSRoman Divacky } 190f22ef01cSRoman Divacky 191f22ef01cSRoman Divacky Stream.EmitRecord(bitc::PARAMATTR_CODE_ENTRY, Record); 192f22ef01cSRoman Divacky Record.clear(); 193f22ef01cSRoman Divacky } 194f22ef01cSRoman Divacky 195f22ef01cSRoman Divacky Stream.ExitBlock(); 196f22ef01cSRoman Divacky } 197f22ef01cSRoman Divacky 198f22ef01cSRoman Divacky /// WriteTypeTable - Write out the type table for a module. 199f22ef01cSRoman Divacky static void WriteTypeTable(const ValueEnumerator &VE, BitstreamWriter &Stream) { 200f22ef01cSRoman Divacky const ValueEnumerator::TypeList &TypeList = VE.getTypes(); 201f22ef01cSRoman Divacky 20217a519f9SDimitry Andric Stream.EnterSubblock(bitc::TYPE_BLOCK_ID_NEW, 4 /*count from # abbrevs */); 203f22ef01cSRoman Divacky SmallVector<uint64_t, 64> TypeVals; 204f22ef01cSRoman Divacky 205dff0c46cSDimitry Andric uint64_t NumBits = Log2_32_Ceil(VE.getTypes().size()+1); 206dff0c46cSDimitry Andric 207f22ef01cSRoman Divacky // Abbrev for TYPE_CODE_POINTER. 208f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 209f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_POINTER)); 210dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 211f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(0)); // Addrspace = 0 212f22ef01cSRoman Divacky unsigned PtrAbbrev = Stream.EmitAbbrev(Abbv); 213f22ef01cSRoman Divacky 214f22ef01cSRoman Divacky // Abbrev for TYPE_CODE_FUNCTION. 215f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 216f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_FUNCTION)); 217f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // isvararg 218f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 219dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 220dff0c46cSDimitry Andric 221f22ef01cSRoman Divacky unsigned FunctionAbbrev = Stream.EmitAbbrev(Abbv); 222f22ef01cSRoman Divacky 22317a519f9SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_ANON. 224f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 22517a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_ANON)); 226f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // ispacked 227f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 228dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 229dff0c46cSDimitry Andric 23017a519f9SDimitry Andric unsigned StructAnonAbbrev = Stream.EmitAbbrev(Abbv); 23117a519f9SDimitry Andric 23217a519f9SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_NAME. 23317a519f9SDimitry Andric Abbv = new BitCodeAbbrev(); 23417a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_NAME)); 23517a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 23617a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 23717a519f9SDimitry Andric unsigned StructNameAbbrev = Stream.EmitAbbrev(Abbv); 23817a519f9SDimitry Andric 23917a519f9SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_NAMED. 24017a519f9SDimitry Andric Abbv = new BitCodeAbbrev(); 24117a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_NAMED)); 24217a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // ispacked 24317a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 244dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 245dff0c46cSDimitry Andric 24617a519f9SDimitry Andric unsigned StructNamedAbbrev = Stream.EmitAbbrev(Abbv); 24717a519f9SDimitry Andric 248f22ef01cSRoman Divacky // Abbrev for TYPE_CODE_ARRAY. 249f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 250f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_ARRAY)); 251f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // size 252dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 253dff0c46cSDimitry Andric 254f22ef01cSRoman Divacky unsigned ArrayAbbrev = Stream.EmitAbbrev(Abbv); 255f22ef01cSRoman Divacky 256f22ef01cSRoman Divacky // Emit an entry count so the reader can reserve space. 257f22ef01cSRoman Divacky TypeVals.push_back(TypeList.size()); 258f22ef01cSRoman Divacky Stream.EmitRecord(bitc::TYPE_CODE_NUMENTRY, TypeVals); 259f22ef01cSRoman Divacky TypeVals.clear(); 260f22ef01cSRoman Divacky 261f22ef01cSRoman Divacky // Loop over all of the types, emitting each in turn. 262f22ef01cSRoman Divacky for (unsigned i = 0, e = TypeList.size(); i != e; ++i) { 2636122f3e6SDimitry Andric Type *T = TypeList[i]; 264f22ef01cSRoman Divacky int AbbrevToUse = 0; 265f22ef01cSRoman Divacky unsigned Code = 0; 266f22ef01cSRoman Divacky 267f22ef01cSRoman Divacky switch (T->getTypeID()) { 268f22ef01cSRoman Divacky default: llvm_unreachable("Unknown type!"); 269f22ef01cSRoman Divacky case Type::VoidTyID: Code = bitc::TYPE_CODE_VOID; break; 270dff0c46cSDimitry Andric case Type::HalfTyID: Code = bitc::TYPE_CODE_HALF; break; 271f22ef01cSRoman Divacky case Type::FloatTyID: Code = bitc::TYPE_CODE_FLOAT; break; 272f22ef01cSRoman Divacky case Type::DoubleTyID: Code = bitc::TYPE_CODE_DOUBLE; break; 273f22ef01cSRoman Divacky case Type::X86_FP80TyID: Code = bitc::TYPE_CODE_X86_FP80; break; 274f22ef01cSRoman Divacky case Type::FP128TyID: Code = bitc::TYPE_CODE_FP128; break; 275f22ef01cSRoman Divacky case Type::PPC_FP128TyID: Code = bitc::TYPE_CODE_PPC_FP128; break; 276f22ef01cSRoman Divacky case Type::LabelTyID: Code = bitc::TYPE_CODE_LABEL; break; 277f22ef01cSRoman Divacky case Type::MetadataTyID: Code = bitc::TYPE_CODE_METADATA; break; 2782754fe60SDimitry Andric case Type::X86_MMXTyID: Code = bitc::TYPE_CODE_X86_MMX; break; 279f22ef01cSRoman Divacky case Type::IntegerTyID: 280f22ef01cSRoman Divacky // INTEGER: [width] 281f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_INTEGER; 282f22ef01cSRoman Divacky TypeVals.push_back(cast<IntegerType>(T)->getBitWidth()); 283f22ef01cSRoman Divacky break; 284f22ef01cSRoman Divacky case Type::PointerTyID: { 2856122f3e6SDimitry Andric PointerType *PTy = cast<PointerType>(T); 286f22ef01cSRoman Divacky // POINTER: [pointee type, address space] 287f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_POINTER; 288f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(PTy->getElementType())); 289f22ef01cSRoman Divacky unsigned AddressSpace = PTy->getAddressSpace(); 290f22ef01cSRoman Divacky TypeVals.push_back(AddressSpace); 291f22ef01cSRoman Divacky if (AddressSpace == 0) AbbrevToUse = PtrAbbrev; 292f22ef01cSRoman Divacky break; 293f22ef01cSRoman Divacky } 294f22ef01cSRoman Divacky case Type::FunctionTyID: { 2956122f3e6SDimitry Andric FunctionType *FT = cast<FunctionType>(T); 296dff0c46cSDimitry Andric // FUNCTION: [isvararg, retty, paramty x N] 297f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_FUNCTION; 298f22ef01cSRoman Divacky TypeVals.push_back(FT->isVarArg()); 299f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(FT->getReturnType())); 300f22ef01cSRoman Divacky for (unsigned i = 0, e = FT->getNumParams(); i != e; ++i) 301f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(FT->getParamType(i))); 302f22ef01cSRoman Divacky AbbrevToUse = FunctionAbbrev; 303f22ef01cSRoman Divacky break; 304f22ef01cSRoman Divacky } 305f22ef01cSRoman Divacky case Type::StructTyID: { 3066122f3e6SDimitry Andric StructType *ST = cast<StructType>(T); 307f22ef01cSRoman Divacky // STRUCT: [ispacked, eltty x N] 308f22ef01cSRoman Divacky TypeVals.push_back(ST->isPacked()); 309f22ef01cSRoman Divacky // Output all of the element types. 310f22ef01cSRoman Divacky for (StructType::element_iterator I = ST->element_begin(), 311f22ef01cSRoman Divacky E = ST->element_end(); I != E; ++I) 312f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(*I)); 31317a519f9SDimitry Andric 3146122f3e6SDimitry Andric if (ST->isLiteral()) { 31517a519f9SDimitry Andric Code = bitc::TYPE_CODE_STRUCT_ANON; 31617a519f9SDimitry Andric AbbrevToUse = StructAnonAbbrev; 31717a519f9SDimitry Andric } else { 31817a519f9SDimitry Andric if (ST->isOpaque()) { 31917a519f9SDimitry Andric Code = bitc::TYPE_CODE_OPAQUE; 32017a519f9SDimitry Andric } else { 32117a519f9SDimitry Andric Code = bitc::TYPE_CODE_STRUCT_NAMED; 32217a519f9SDimitry Andric AbbrevToUse = StructNamedAbbrev; 32317a519f9SDimitry Andric } 32417a519f9SDimitry Andric 32517a519f9SDimitry Andric // Emit the name if it is present. 32617a519f9SDimitry Andric if (!ST->getName().empty()) 32717a519f9SDimitry Andric WriteStringRecord(bitc::TYPE_CODE_STRUCT_NAME, ST->getName(), 32817a519f9SDimitry Andric StructNameAbbrev, Stream); 32917a519f9SDimitry Andric } 330f22ef01cSRoman Divacky break; 331f22ef01cSRoman Divacky } 332f22ef01cSRoman Divacky case Type::ArrayTyID: { 3336122f3e6SDimitry Andric ArrayType *AT = cast<ArrayType>(T); 334f22ef01cSRoman Divacky // ARRAY: [numelts, eltty] 335f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_ARRAY; 336f22ef01cSRoman Divacky TypeVals.push_back(AT->getNumElements()); 337f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(AT->getElementType())); 338f22ef01cSRoman Divacky AbbrevToUse = ArrayAbbrev; 339f22ef01cSRoman Divacky break; 340f22ef01cSRoman Divacky } 341f22ef01cSRoman Divacky case Type::VectorTyID: { 3426122f3e6SDimitry Andric VectorType *VT = cast<VectorType>(T); 343f22ef01cSRoman Divacky // VECTOR [numelts, eltty] 344f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_VECTOR; 345f22ef01cSRoman Divacky TypeVals.push_back(VT->getNumElements()); 346f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(VT->getElementType())); 347f22ef01cSRoman Divacky break; 348f22ef01cSRoman Divacky } 349f22ef01cSRoman Divacky } 350f22ef01cSRoman Divacky 351f22ef01cSRoman Divacky // Emit the finished record. 352f22ef01cSRoman Divacky Stream.EmitRecord(Code, TypeVals, AbbrevToUse); 353f22ef01cSRoman Divacky TypeVals.clear(); 354f22ef01cSRoman Divacky } 355f22ef01cSRoman Divacky 356f22ef01cSRoman Divacky Stream.ExitBlock(); 357f22ef01cSRoman Divacky } 358f22ef01cSRoman Divacky 359f22ef01cSRoman Divacky static unsigned getEncodedLinkage(const GlobalValue *GV) { 360f22ef01cSRoman Divacky switch (GV->getLinkage()) { 361f22ef01cSRoman Divacky case GlobalValue::ExternalLinkage: return 0; 362f22ef01cSRoman Divacky case GlobalValue::WeakAnyLinkage: return 1; 363f22ef01cSRoman Divacky case GlobalValue::AppendingLinkage: return 2; 364f22ef01cSRoman Divacky case GlobalValue::InternalLinkage: return 3; 365f22ef01cSRoman Divacky case GlobalValue::LinkOnceAnyLinkage: return 4; 366f22ef01cSRoman Divacky case GlobalValue::DLLImportLinkage: return 5; 367f22ef01cSRoman Divacky case GlobalValue::DLLExportLinkage: return 6; 368f22ef01cSRoman Divacky case GlobalValue::ExternalWeakLinkage: return 7; 369f22ef01cSRoman Divacky case GlobalValue::CommonLinkage: return 8; 370f22ef01cSRoman Divacky case GlobalValue::PrivateLinkage: return 9; 371f22ef01cSRoman Divacky case GlobalValue::WeakODRLinkage: return 10; 372f22ef01cSRoman Divacky case GlobalValue::LinkOnceODRLinkage: return 11; 373f22ef01cSRoman Divacky case GlobalValue::AvailableExternallyLinkage: return 12; 374f22ef01cSRoman Divacky case GlobalValue::LinkerPrivateLinkage: return 13; 375ffd1746dSEd Schouten case GlobalValue::LinkerPrivateWeakLinkage: return 14; 376e580952dSDimitry Andric case GlobalValue::LinkerPrivateWeakDefAutoLinkage: return 15; 377f22ef01cSRoman Divacky } 378dff0c46cSDimitry Andric llvm_unreachable("Invalid linkage"); 379f22ef01cSRoman Divacky } 380f22ef01cSRoman Divacky 381f22ef01cSRoman Divacky static unsigned getEncodedVisibility(const GlobalValue *GV) { 382f22ef01cSRoman Divacky switch (GV->getVisibility()) { 383f22ef01cSRoman Divacky case GlobalValue::DefaultVisibility: return 0; 384f22ef01cSRoman Divacky case GlobalValue::HiddenVisibility: return 1; 385f22ef01cSRoman Divacky case GlobalValue::ProtectedVisibility: return 2; 386f22ef01cSRoman Divacky } 387dff0c46cSDimitry Andric llvm_unreachable("Invalid visibility"); 388f22ef01cSRoman Divacky } 389f22ef01cSRoman Divacky 390f22ef01cSRoman Divacky // Emit top-level description of module, including target triple, inline asm, 391f22ef01cSRoman Divacky // descriptors for global variables, and function prototype info. 392f22ef01cSRoman Divacky static void WriteModuleInfo(const Module *M, const ValueEnumerator &VE, 393f22ef01cSRoman Divacky BitstreamWriter &Stream) { 394f22ef01cSRoman Divacky // Emit the list of dependent libraries for the Module. 395f22ef01cSRoman Divacky for (Module::lib_iterator I = M->lib_begin(), E = M->lib_end(); I != E; ++I) 396f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_DEPLIB, *I, 0/*TODO*/, Stream); 397f22ef01cSRoman Divacky 398f22ef01cSRoman Divacky // Emit various pieces of data attached to a module. 399f22ef01cSRoman Divacky if (!M->getTargetTriple().empty()) 400f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_TRIPLE, M->getTargetTriple(), 401f22ef01cSRoman Divacky 0/*TODO*/, Stream); 402f22ef01cSRoman Divacky if (!M->getDataLayout().empty()) 403f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_DATALAYOUT, M->getDataLayout(), 404f22ef01cSRoman Divacky 0/*TODO*/, Stream); 405f22ef01cSRoman Divacky if (!M->getModuleInlineAsm().empty()) 406f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_ASM, M->getModuleInlineAsm(), 407f22ef01cSRoman Divacky 0/*TODO*/, Stream); 408f22ef01cSRoman Divacky 409f22ef01cSRoman Divacky // Emit information about sections and GC, computing how many there are. Also 410f22ef01cSRoman Divacky // compute the maximum alignment value. 411f22ef01cSRoman Divacky std::map<std::string, unsigned> SectionMap; 412f22ef01cSRoman Divacky std::map<std::string, unsigned> GCMap; 413f22ef01cSRoman Divacky unsigned MaxAlignment = 0; 414f22ef01cSRoman Divacky unsigned MaxGlobalType = 0; 415f22ef01cSRoman Divacky for (Module::const_global_iterator GV = M->global_begin(),E = M->global_end(); 416f22ef01cSRoman Divacky GV != E; ++GV) { 417f22ef01cSRoman Divacky MaxAlignment = std::max(MaxAlignment, GV->getAlignment()); 418f22ef01cSRoman Divacky MaxGlobalType = std::max(MaxGlobalType, VE.getTypeID(GV->getType())); 4196122f3e6SDimitry Andric if (GV->hasSection()) { 420f22ef01cSRoman Divacky // Give section names unique ID's. 421f22ef01cSRoman Divacky unsigned &Entry = SectionMap[GV->getSection()]; 4226122f3e6SDimitry Andric if (!Entry) { 423f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_SECTIONNAME, GV->getSection(), 424f22ef01cSRoman Divacky 0/*TODO*/, Stream); 425f22ef01cSRoman Divacky Entry = SectionMap.size(); 426f22ef01cSRoman Divacky } 4276122f3e6SDimitry Andric } 4286122f3e6SDimitry Andric } 429f22ef01cSRoman Divacky for (Module::const_iterator F = M->begin(), E = M->end(); F != E; ++F) { 430f22ef01cSRoman Divacky MaxAlignment = std::max(MaxAlignment, F->getAlignment()); 431f22ef01cSRoman Divacky if (F->hasSection()) { 432f22ef01cSRoman Divacky // Give section names unique ID's. 433f22ef01cSRoman Divacky unsigned &Entry = SectionMap[F->getSection()]; 434f22ef01cSRoman Divacky if (!Entry) { 435f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_SECTIONNAME, F->getSection(), 436f22ef01cSRoman Divacky 0/*TODO*/, Stream); 437f22ef01cSRoman Divacky Entry = SectionMap.size(); 438f22ef01cSRoman Divacky } 439f22ef01cSRoman Divacky } 440f22ef01cSRoman Divacky if (F->hasGC()) { 441f22ef01cSRoman Divacky // Same for GC names. 442f22ef01cSRoman Divacky unsigned &Entry = GCMap[F->getGC()]; 443f22ef01cSRoman Divacky if (!Entry) { 444f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_GCNAME, F->getGC(), 445f22ef01cSRoman Divacky 0/*TODO*/, Stream); 446f22ef01cSRoman Divacky Entry = GCMap.size(); 447f22ef01cSRoman Divacky } 448f22ef01cSRoman Divacky } 449f22ef01cSRoman Divacky } 450f22ef01cSRoman Divacky 451f22ef01cSRoman Divacky // Emit abbrev for globals, now that we know # sections and max alignment. 452f22ef01cSRoman Divacky unsigned SimpleGVarAbbrev = 0; 453f22ef01cSRoman Divacky if (!M->global_empty()) { 454f22ef01cSRoman Divacky // Add an abbrev for common globals with no visibility or thread localness. 455f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 456f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_GLOBALVAR)); 457f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 458f22ef01cSRoman Divacky Log2_32_Ceil(MaxGlobalType+1))); 459f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // Constant. 460f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // Initializer. 461f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // Linkage. 462f22ef01cSRoman Divacky if (MaxAlignment == 0) // Alignment. 463f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(0)); 464f22ef01cSRoman Divacky else { 465f22ef01cSRoman Divacky unsigned MaxEncAlignment = Log2_32(MaxAlignment)+1; 466f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 467f22ef01cSRoman Divacky Log2_32_Ceil(MaxEncAlignment+1))); 468f22ef01cSRoman Divacky } 469f22ef01cSRoman Divacky if (SectionMap.empty()) // Section. 470f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(0)); 471f22ef01cSRoman Divacky else 472f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 473f22ef01cSRoman Divacky Log2_32_Ceil(SectionMap.size()+1))); 474f22ef01cSRoman Divacky // Don't bother emitting vis + thread local. 475f22ef01cSRoman Divacky SimpleGVarAbbrev = Stream.EmitAbbrev(Abbv); 476f22ef01cSRoman Divacky } 477f22ef01cSRoman Divacky 478f22ef01cSRoman Divacky // Emit the global variable information. 479f22ef01cSRoman Divacky SmallVector<unsigned, 64> Vals; 480f22ef01cSRoman Divacky for (Module::const_global_iterator GV = M->global_begin(),E = M->global_end(); 481f22ef01cSRoman Divacky GV != E; ++GV) { 482f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 483f22ef01cSRoman Divacky 484f22ef01cSRoman Divacky // GLOBALVAR: [type, isconst, initid, 4852754fe60SDimitry Andric // linkage, alignment, section, visibility, threadlocal, 4862754fe60SDimitry Andric // unnamed_addr] 487f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(GV->getType())); 488f22ef01cSRoman Divacky Vals.push_back(GV->isConstant()); 489f22ef01cSRoman Divacky Vals.push_back(GV->isDeclaration() ? 0 : 490f22ef01cSRoman Divacky (VE.getValueID(GV->getInitializer()) + 1)); 491f22ef01cSRoman Divacky Vals.push_back(getEncodedLinkage(GV)); 492f22ef01cSRoman Divacky Vals.push_back(Log2_32(GV->getAlignment())+1); 493f22ef01cSRoman Divacky Vals.push_back(GV->hasSection() ? SectionMap[GV->getSection()] : 0); 494f22ef01cSRoman Divacky if (GV->isThreadLocal() || 4952754fe60SDimitry Andric GV->getVisibility() != GlobalValue::DefaultVisibility || 4962754fe60SDimitry Andric GV->hasUnnamedAddr()) { 497f22ef01cSRoman Divacky Vals.push_back(getEncodedVisibility(GV)); 498f22ef01cSRoman Divacky Vals.push_back(GV->isThreadLocal()); 4992754fe60SDimitry Andric Vals.push_back(GV->hasUnnamedAddr()); 500f22ef01cSRoman Divacky } else { 501f22ef01cSRoman Divacky AbbrevToUse = SimpleGVarAbbrev; 502f22ef01cSRoman Divacky } 503f22ef01cSRoman Divacky 504f22ef01cSRoman Divacky Stream.EmitRecord(bitc::MODULE_CODE_GLOBALVAR, Vals, AbbrevToUse); 505f22ef01cSRoman Divacky Vals.clear(); 506f22ef01cSRoman Divacky } 507f22ef01cSRoman Divacky 508f22ef01cSRoman Divacky // Emit the function proto information. 509f22ef01cSRoman Divacky for (Module::const_iterator F = M->begin(), E = M->end(); F != E; ++F) { 510dff0c46cSDimitry Andric // FUNCTION: [type, callingconv, isproto, linkage, paramattrs, alignment, 511dff0c46cSDimitry Andric // section, visibility, gc, unnamed_addr] 512f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(F->getType())); 513f22ef01cSRoman Divacky Vals.push_back(F->getCallingConv()); 514f22ef01cSRoman Divacky Vals.push_back(F->isDeclaration()); 515f22ef01cSRoman Divacky Vals.push_back(getEncodedLinkage(F)); 516f22ef01cSRoman Divacky Vals.push_back(VE.getAttributeID(F->getAttributes())); 517f22ef01cSRoman Divacky Vals.push_back(Log2_32(F->getAlignment())+1); 518f22ef01cSRoman Divacky Vals.push_back(F->hasSection() ? SectionMap[F->getSection()] : 0); 519f22ef01cSRoman Divacky Vals.push_back(getEncodedVisibility(F)); 520f22ef01cSRoman Divacky Vals.push_back(F->hasGC() ? GCMap[F->getGC()] : 0); 5212754fe60SDimitry Andric Vals.push_back(F->hasUnnamedAddr()); 522f22ef01cSRoman Divacky 523f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 524f22ef01cSRoman Divacky Stream.EmitRecord(bitc::MODULE_CODE_FUNCTION, Vals, AbbrevToUse); 525f22ef01cSRoman Divacky Vals.clear(); 526f22ef01cSRoman Divacky } 527f22ef01cSRoman Divacky 528f22ef01cSRoman Divacky // Emit the alias information. 529f22ef01cSRoman Divacky for (Module::const_alias_iterator AI = M->alias_begin(), E = M->alias_end(); 530f22ef01cSRoman Divacky AI != E; ++AI) { 531dff0c46cSDimitry Andric // ALIAS: [alias type, aliasee val#, linkage, visibility] 532f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(AI->getType())); 533f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(AI->getAliasee())); 534f22ef01cSRoman Divacky Vals.push_back(getEncodedLinkage(AI)); 535f22ef01cSRoman Divacky Vals.push_back(getEncodedVisibility(AI)); 536f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 537f22ef01cSRoman Divacky Stream.EmitRecord(bitc::MODULE_CODE_ALIAS, Vals, AbbrevToUse); 538f22ef01cSRoman Divacky Vals.clear(); 539f22ef01cSRoman Divacky } 540f22ef01cSRoman Divacky } 541f22ef01cSRoman Divacky 542f22ef01cSRoman Divacky static uint64_t GetOptimizationFlags(const Value *V) { 543f22ef01cSRoman Divacky uint64_t Flags = 0; 544f22ef01cSRoman Divacky 545f22ef01cSRoman Divacky if (const OverflowingBinaryOperator *OBO = 546f22ef01cSRoman Divacky dyn_cast<OverflowingBinaryOperator>(V)) { 547f22ef01cSRoman Divacky if (OBO->hasNoSignedWrap()) 548f22ef01cSRoman Divacky Flags |= 1 << bitc::OBO_NO_SIGNED_WRAP; 549f22ef01cSRoman Divacky if (OBO->hasNoUnsignedWrap()) 550f22ef01cSRoman Divacky Flags |= 1 << bitc::OBO_NO_UNSIGNED_WRAP; 5512754fe60SDimitry Andric } else if (const PossiblyExactOperator *PEO = 5522754fe60SDimitry Andric dyn_cast<PossiblyExactOperator>(V)) { 5532754fe60SDimitry Andric if (PEO->isExact()) 5542754fe60SDimitry Andric Flags |= 1 << bitc::PEO_EXACT; 555f22ef01cSRoman Divacky } 556f22ef01cSRoman Divacky 557f22ef01cSRoman Divacky return Flags; 558f22ef01cSRoman Divacky } 559f22ef01cSRoman Divacky 560f22ef01cSRoman Divacky static void WriteMDNode(const MDNode *N, 561f22ef01cSRoman Divacky const ValueEnumerator &VE, 562f22ef01cSRoman Divacky BitstreamWriter &Stream, 563f22ef01cSRoman Divacky SmallVector<uint64_t, 64> &Record) { 564f22ef01cSRoman Divacky for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) { 565f22ef01cSRoman Divacky if (N->getOperand(i)) { 566f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(N->getOperand(i)->getType())); 567f22ef01cSRoman Divacky Record.push_back(VE.getValueID(N->getOperand(i))); 568f22ef01cSRoman Divacky } else { 569f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(Type::getVoidTy(N->getContext()))); 570f22ef01cSRoman Divacky Record.push_back(0); 571f22ef01cSRoman Divacky } 572f22ef01cSRoman Divacky } 57317a519f9SDimitry Andric unsigned MDCode = N->isFunctionLocal() ? bitc::METADATA_FN_NODE : 57417a519f9SDimitry Andric bitc::METADATA_NODE; 575f22ef01cSRoman Divacky Stream.EmitRecord(MDCode, Record, 0); 576f22ef01cSRoman Divacky Record.clear(); 577f22ef01cSRoman Divacky } 578f22ef01cSRoman Divacky 579e580952dSDimitry Andric static void WriteModuleMetadata(const Module *M, 580e580952dSDimitry Andric const ValueEnumerator &VE, 581f22ef01cSRoman Divacky BitstreamWriter &Stream) { 582f22ef01cSRoman Divacky const ValueEnumerator::ValueList &Vals = VE.getMDValues(); 583f22ef01cSRoman Divacky bool StartedMetadataBlock = false; 584f22ef01cSRoman Divacky unsigned MDSAbbrev = 0; 585f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 586f22ef01cSRoman Divacky for (unsigned i = 0, e = Vals.size(); i != e; ++i) { 587f22ef01cSRoman Divacky 588f22ef01cSRoman Divacky if (const MDNode *N = dyn_cast<MDNode>(Vals[i].first)) { 589f22ef01cSRoman Divacky if (!N->isFunctionLocal() || !N->getFunction()) { 590f22ef01cSRoman Divacky if (!StartedMetadataBlock) { 591f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 592f22ef01cSRoman Divacky StartedMetadataBlock = true; 593f22ef01cSRoman Divacky } 594f22ef01cSRoman Divacky WriteMDNode(N, VE, Stream, Record); 595f22ef01cSRoman Divacky } 596f22ef01cSRoman Divacky } else if (const MDString *MDS = dyn_cast<MDString>(Vals[i].first)) { 597f22ef01cSRoman Divacky if (!StartedMetadataBlock) { 598f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 599f22ef01cSRoman Divacky 600f22ef01cSRoman Divacky // Abbrev for METADATA_STRING. 601f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 602f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_STRING)); 603f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 604f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 605f22ef01cSRoman Divacky MDSAbbrev = Stream.EmitAbbrev(Abbv); 606f22ef01cSRoman Divacky StartedMetadataBlock = true; 607f22ef01cSRoman Divacky } 608f22ef01cSRoman Divacky 609f22ef01cSRoman Divacky // Code: [strchar x N] 610f22ef01cSRoman Divacky Record.append(MDS->begin(), MDS->end()); 611f22ef01cSRoman Divacky 612f22ef01cSRoman Divacky // Emit the finished record. 613f22ef01cSRoman Divacky Stream.EmitRecord(bitc::METADATA_STRING, Record, MDSAbbrev); 614f22ef01cSRoman Divacky Record.clear(); 615e580952dSDimitry Andric } 616e580952dSDimitry Andric } 617e580952dSDimitry Andric 618e580952dSDimitry Andric // Write named metadata. 619e580952dSDimitry Andric for (Module::const_named_metadata_iterator I = M->named_metadata_begin(), 620e580952dSDimitry Andric E = M->named_metadata_end(); I != E; ++I) { 621e580952dSDimitry Andric const NamedMDNode *NMD = I; 622f22ef01cSRoman Divacky if (!StartedMetadataBlock) { 623f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 624f22ef01cSRoman Divacky StartedMetadataBlock = true; 625f22ef01cSRoman Divacky } 626f22ef01cSRoman Divacky 627f22ef01cSRoman Divacky // Write name. 628f22ef01cSRoman Divacky StringRef Str = NMD->getName(); 629f22ef01cSRoman Divacky for (unsigned i = 0, e = Str.size(); i != e; ++i) 630f22ef01cSRoman Divacky Record.push_back(Str[i]); 631f22ef01cSRoman Divacky Stream.EmitRecord(bitc::METADATA_NAME, Record, 0/*TODO*/); 632f22ef01cSRoman Divacky Record.clear(); 633f22ef01cSRoman Divacky 634f22ef01cSRoman Divacky // Write named metadata operands. 635e580952dSDimitry Andric for (unsigned i = 0, e = NMD->getNumOperands(); i != e; ++i) 636f22ef01cSRoman Divacky Record.push_back(VE.getValueID(NMD->getOperand(i))); 63717a519f9SDimitry Andric Stream.EmitRecord(bitc::METADATA_NAMED_NODE, Record, 0); 638f22ef01cSRoman Divacky Record.clear(); 639f22ef01cSRoman Divacky } 640f22ef01cSRoman Divacky 641f22ef01cSRoman Divacky if (StartedMetadataBlock) 642f22ef01cSRoman Divacky Stream.ExitBlock(); 643f22ef01cSRoman Divacky } 644f22ef01cSRoman Divacky 645f22ef01cSRoman Divacky static void WriteFunctionLocalMetadata(const Function &F, 646f22ef01cSRoman Divacky const ValueEnumerator &VE, 647f22ef01cSRoman Divacky BitstreamWriter &Stream) { 648f22ef01cSRoman Divacky bool StartedMetadataBlock = false; 649f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 650ffd1746dSEd Schouten const SmallVector<const MDNode *, 8> &Vals = VE.getFunctionLocalMDValues(); 651f22ef01cSRoman Divacky for (unsigned i = 0, e = Vals.size(); i != e; ++i) 652ffd1746dSEd Schouten if (const MDNode *N = Vals[i]) 653f22ef01cSRoman Divacky if (N->isFunctionLocal() && N->getFunction() == &F) { 654f22ef01cSRoman Divacky if (!StartedMetadataBlock) { 655f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 656f22ef01cSRoman Divacky StartedMetadataBlock = true; 657f22ef01cSRoman Divacky } 658f22ef01cSRoman Divacky WriteMDNode(N, VE, Stream, Record); 659f22ef01cSRoman Divacky } 660f22ef01cSRoman Divacky 661f22ef01cSRoman Divacky if (StartedMetadataBlock) 662f22ef01cSRoman Divacky Stream.ExitBlock(); 663f22ef01cSRoman Divacky } 664f22ef01cSRoman Divacky 665f22ef01cSRoman Divacky static void WriteMetadataAttachment(const Function &F, 666f22ef01cSRoman Divacky const ValueEnumerator &VE, 667f22ef01cSRoman Divacky BitstreamWriter &Stream) { 668f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_ATTACHMENT_ID, 3); 669f22ef01cSRoman Divacky 670f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 671f22ef01cSRoman Divacky 672f22ef01cSRoman Divacky // Write metadata attachments 67317a519f9SDimitry Andric // METADATA_ATTACHMENT - [m x [value, [n x [id, mdnode]]] 674f22ef01cSRoman Divacky SmallVector<std::pair<unsigned, MDNode*>, 4> MDs; 675f22ef01cSRoman Divacky 676f22ef01cSRoman Divacky for (Function::const_iterator BB = F.begin(), E = F.end(); BB != E; ++BB) 677f22ef01cSRoman Divacky for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); 678f22ef01cSRoman Divacky I != E; ++I) { 679f22ef01cSRoman Divacky MDs.clear(); 680f22ef01cSRoman Divacky I->getAllMetadataOtherThanDebugLoc(MDs); 681f22ef01cSRoman Divacky 682f22ef01cSRoman Divacky // If no metadata, ignore instruction. 683f22ef01cSRoman Divacky if (MDs.empty()) continue; 684f22ef01cSRoman Divacky 685f22ef01cSRoman Divacky Record.push_back(VE.getInstructionID(I)); 686f22ef01cSRoman Divacky 687f22ef01cSRoman Divacky for (unsigned i = 0, e = MDs.size(); i != e; ++i) { 688f22ef01cSRoman Divacky Record.push_back(MDs[i].first); 689f22ef01cSRoman Divacky Record.push_back(VE.getValueID(MDs[i].second)); 690f22ef01cSRoman Divacky } 69117a519f9SDimitry Andric Stream.EmitRecord(bitc::METADATA_ATTACHMENT, Record, 0); 692f22ef01cSRoman Divacky Record.clear(); 693f22ef01cSRoman Divacky } 694f22ef01cSRoman Divacky 695f22ef01cSRoman Divacky Stream.ExitBlock(); 696f22ef01cSRoman Divacky } 697f22ef01cSRoman Divacky 698f22ef01cSRoman Divacky static void WriteModuleMetadataStore(const Module *M, BitstreamWriter &Stream) { 699f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 700f22ef01cSRoman Divacky 701f22ef01cSRoman Divacky // Write metadata kinds 702f22ef01cSRoman Divacky // METADATA_KIND - [n x [id, name]] 703f22ef01cSRoman Divacky SmallVector<StringRef, 4> Names; 704f22ef01cSRoman Divacky M->getMDKindNames(Names); 705f22ef01cSRoman Divacky 706e580952dSDimitry Andric if (Names.empty()) return; 707f22ef01cSRoman Divacky 708f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 709f22ef01cSRoman Divacky 710e580952dSDimitry Andric for (unsigned MDKindID = 0, e = Names.size(); MDKindID != e; ++MDKindID) { 711f22ef01cSRoman Divacky Record.push_back(MDKindID); 712f22ef01cSRoman Divacky StringRef KName = Names[MDKindID]; 713f22ef01cSRoman Divacky Record.append(KName.begin(), KName.end()); 714f22ef01cSRoman Divacky 715f22ef01cSRoman Divacky Stream.EmitRecord(bitc::METADATA_KIND, Record, 0); 716f22ef01cSRoman Divacky Record.clear(); 717f22ef01cSRoman Divacky } 718f22ef01cSRoman Divacky 719f22ef01cSRoman Divacky Stream.ExitBlock(); 720f22ef01cSRoman Divacky } 721f22ef01cSRoman Divacky 722f22ef01cSRoman Divacky static void WriteConstants(unsigned FirstVal, unsigned LastVal, 723f22ef01cSRoman Divacky const ValueEnumerator &VE, 724f22ef01cSRoman Divacky BitstreamWriter &Stream, bool isGlobal) { 725f22ef01cSRoman Divacky if (FirstVal == LastVal) return; 726f22ef01cSRoman Divacky 727f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::CONSTANTS_BLOCK_ID, 4); 728f22ef01cSRoman Divacky 729f22ef01cSRoman Divacky unsigned AggregateAbbrev = 0; 730f22ef01cSRoman Divacky unsigned String8Abbrev = 0; 731f22ef01cSRoman Divacky unsigned CString7Abbrev = 0; 732f22ef01cSRoman Divacky unsigned CString6Abbrev = 0; 733f22ef01cSRoman Divacky // If this is a constant pool for the module, emit module-specific abbrevs. 734f22ef01cSRoman Divacky if (isGlobal) { 735f22ef01cSRoman Divacky // Abbrev for CST_CODE_AGGREGATE. 736f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 737f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_AGGREGATE)); 738f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 739f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, Log2_32_Ceil(LastVal+1))); 740f22ef01cSRoman Divacky AggregateAbbrev = Stream.EmitAbbrev(Abbv); 741f22ef01cSRoman Divacky 742f22ef01cSRoman Divacky // Abbrev for CST_CODE_STRING. 743f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 744f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_STRING)); 745f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 746f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 747f22ef01cSRoman Divacky String8Abbrev = Stream.EmitAbbrev(Abbv); 748f22ef01cSRoman Divacky // Abbrev for CST_CODE_CSTRING. 749f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 750f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CSTRING)); 751f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 752f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 753f22ef01cSRoman Divacky CString7Abbrev = Stream.EmitAbbrev(Abbv); 754f22ef01cSRoman Divacky // Abbrev for CST_CODE_CSTRING. 755f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 756f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CSTRING)); 757f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 758f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 759f22ef01cSRoman Divacky CString6Abbrev = Stream.EmitAbbrev(Abbv); 760f22ef01cSRoman Divacky } 761f22ef01cSRoman Divacky 762f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 763f22ef01cSRoman Divacky 764f22ef01cSRoman Divacky const ValueEnumerator::ValueList &Vals = VE.getValues(); 7656122f3e6SDimitry Andric Type *LastTy = 0; 766f22ef01cSRoman Divacky for (unsigned i = FirstVal; i != LastVal; ++i) { 767f22ef01cSRoman Divacky const Value *V = Vals[i].first; 768f22ef01cSRoman Divacky // If we need to switch types, do so now. 769f22ef01cSRoman Divacky if (V->getType() != LastTy) { 770f22ef01cSRoman Divacky LastTy = V->getType(); 771f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(LastTy)); 772f22ef01cSRoman Divacky Stream.EmitRecord(bitc::CST_CODE_SETTYPE, Record, 773f22ef01cSRoman Divacky CONSTANTS_SETTYPE_ABBREV); 774f22ef01cSRoman Divacky Record.clear(); 775f22ef01cSRoman Divacky } 776f22ef01cSRoman Divacky 777f22ef01cSRoman Divacky if (const InlineAsm *IA = dyn_cast<InlineAsm>(V)) { 778f22ef01cSRoman Divacky Record.push_back(unsigned(IA->hasSideEffects()) | 779f22ef01cSRoman Divacky unsigned(IA->isAlignStack()) << 1); 780f22ef01cSRoman Divacky 781f22ef01cSRoman Divacky // Add the asm string. 782f22ef01cSRoman Divacky const std::string &AsmStr = IA->getAsmString(); 783f22ef01cSRoman Divacky Record.push_back(AsmStr.size()); 784f22ef01cSRoman Divacky for (unsigned i = 0, e = AsmStr.size(); i != e; ++i) 785f22ef01cSRoman Divacky Record.push_back(AsmStr[i]); 786f22ef01cSRoman Divacky 787f22ef01cSRoman Divacky // Add the constraint string. 788f22ef01cSRoman Divacky const std::string &ConstraintStr = IA->getConstraintString(); 789f22ef01cSRoman Divacky Record.push_back(ConstraintStr.size()); 790f22ef01cSRoman Divacky for (unsigned i = 0, e = ConstraintStr.size(); i != e; ++i) 791f22ef01cSRoman Divacky Record.push_back(ConstraintStr[i]); 792f22ef01cSRoman Divacky Stream.EmitRecord(bitc::CST_CODE_INLINEASM, Record); 793f22ef01cSRoman Divacky Record.clear(); 794f22ef01cSRoman Divacky continue; 795f22ef01cSRoman Divacky } 796f22ef01cSRoman Divacky const Constant *C = cast<Constant>(V); 797f22ef01cSRoman Divacky unsigned Code = -1U; 798f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 799f22ef01cSRoman Divacky if (C->isNullValue()) { 800f22ef01cSRoman Divacky Code = bitc::CST_CODE_NULL; 801f22ef01cSRoman Divacky } else if (isa<UndefValue>(C)) { 802f22ef01cSRoman Divacky Code = bitc::CST_CODE_UNDEF; 803f22ef01cSRoman Divacky } else if (const ConstantInt *IV = dyn_cast<ConstantInt>(C)) { 804f22ef01cSRoman Divacky if (IV->getBitWidth() <= 64) { 805e580952dSDimitry Andric uint64_t V = IV->getSExtValue(); 806e580952dSDimitry Andric if ((int64_t)V >= 0) 807f22ef01cSRoman Divacky Record.push_back(V << 1); 808f22ef01cSRoman Divacky else 809f22ef01cSRoman Divacky Record.push_back((-V << 1) | 1); 810f22ef01cSRoman Divacky Code = bitc::CST_CODE_INTEGER; 811f22ef01cSRoman Divacky AbbrevToUse = CONSTANTS_INTEGER_ABBREV; 812f22ef01cSRoman Divacky } else { // Wide integers, > 64 bits in size. 813f22ef01cSRoman Divacky // We have an arbitrary precision integer value to write whose 814f22ef01cSRoman Divacky // bit width is > 64. However, in canonical unsigned integer 815f22ef01cSRoman Divacky // format it is likely that the high bits are going to be zero. 816f22ef01cSRoman Divacky // So, we only write the number of active words. 817f22ef01cSRoman Divacky unsigned NWords = IV->getValue().getActiveWords(); 818f22ef01cSRoman Divacky const uint64_t *RawWords = IV->getValue().getRawData(); 819f22ef01cSRoman Divacky for (unsigned i = 0; i != NWords; ++i) { 820f22ef01cSRoman Divacky int64_t V = RawWords[i]; 821f22ef01cSRoman Divacky if (V >= 0) 822f22ef01cSRoman Divacky Record.push_back(V << 1); 823f22ef01cSRoman Divacky else 824f22ef01cSRoman Divacky Record.push_back((-V << 1) | 1); 825f22ef01cSRoman Divacky } 826f22ef01cSRoman Divacky Code = bitc::CST_CODE_WIDE_INTEGER; 827f22ef01cSRoman Divacky } 828f22ef01cSRoman Divacky } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(C)) { 829f22ef01cSRoman Divacky Code = bitc::CST_CODE_FLOAT; 8306122f3e6SDimitry Andric Type *Ty = CFP->getType(); 831dff0c46cSDimitry Andric if (Ty->isHalfTy() || Ty->isFloatTy() || Ty->isDoubleTy()) { 832f22ef01cSRoman Divacky Record.push_back(CFP->getValueAPF().bitcastToAPInt().getZExtValue()); 833f22ef01cSRoman Divacky } else if (Ty->isX86_FP80Ty()) { 834f22ef01cSRoman Divacky // api needed to prevent premature destruction 835f22ef01cSRoman Divacky // bits are not in the same order as a normal i80 APInt, compensate. 836f22ef01cSRoman Divacky APInt api = CFP->getValueAPF().bitcastToAPInt(); 837f22ef01cSRoman Divacky const uint64_t *p = api.getRawData(); 838f22ef01cSRoman Divacky Record.push_back((p[1] << 48) | (p[0] >> 16)); 839f22ef01cSRoman Divacky Record.push_back(p[0] & 0xffffLL); 840f22ef01cSRoman Divacky } else if (Ty->isFP128Ty() || Ty->isPPC_FP128Ty()) { 841f22ef01cSRoman Divacky APInt api = CFP->getValueAPF().bitcastToAPInt(); 842f22ef01cSRoman Divacky const uint64_t *p = api.getRawData(); 843f22ef01cSRoman Divacky Record.push_back(p[0]); 844f22ef01cSRoman Divacky Record.push_back(p[1]); 845f22ef01cSRoman Divacky } else { 846f22ef01cSRoman Divacky assert (0 && "Unknown FP type!"); 847f22ef01cSRoman Divacky } 848dff0c46cSDimitry Andric } else if (isa<ConstantDataSequential>(C) && 849dff0c46cSDimitry Andric cast<ConstantDataSequential>(C)->isString()) { 850dff0c46cSDimitry Andric const ConstantDataSequential *Str = cast<ConstantDataSequential>(C); 851f22ef01cSRoman Divacky // Emit constant strings specially. 852dff0c46cSDimitry Andric unsigned NumElts = Str->getNumElements(); 853f22ef01cSRoman Divacky // If this is a null-terminated string, use the denser CSTRING encoding. 854dff0c46cSDimitry Andric if (Str->isCString()) { 855f22ef01cSRoman Divacky Code = bitc::CST_CODE_CSTRING; 856dff0c46cSDimitry Andric --NumElts; // Don't encode the null, which isn't allowed by char6. 857f22ef01cSRoman Divacky } else { 858f22ef01cSRoman Divacky Code = bitc::CST_CODE_STRING; 859f22ef01cSRoman Divacky AbbrevToUse = String8Abbrev; 860f22ef01cSRoman Divacky } 861f22ef01cSRoman Divacky bool isCStr7 = Code == bitc::CST_CODE_CSTRING; 862f22ef01cSRoman Divacky bool isCStrChar6 = Code == bitc::CST_CODE_CSTRING; 863dff0c46cSDimitry Andric for (unsigned i = 0; i != NumElts; ++i) { 864dff0c46cSDimitry Andric unsigned char V = Str->getElementAsInteger(i); 865f22ef01cSRoman Divacky Record.push_back(V); 866f22ef01cSRoman Divacky isCStr7 &= (V & 128) == 0; 867f22ef01cSRoman Divacky if (isCStrChar6) 868f22ef01cSRoman Divacky isCStrChar6 = BitCodeAbbrevOp::isChar6(V); 869f22ef01cSRoman Divacky } 870f22ef01cSRoman Divacky 871f22ef01cSRoman Divacky if (isCStrChar6) 872f22ef01cSRoman Divacky AbbrevToUse = CString6Abbrev; 873f22ef01cSRoman Divacky else if (isCStr7) 874f22ef01cSRoman Divacky AbbrevToUse = CString7Abbrev; 875dff0c46cSDimitry Andric } else if (const ConstantDataSequential *CDS = 876dff0c46cSDimitry Andric dyn_cast<ConstantDataSequential>(C)) { 877dff0c46cSDimitry Andric Code = bitc::CST_CODE_DATA; 878dff0c46cSDimitry Andric Type *EltTy = CDS->getType()->getElementType(); 879dff0c46cSDimitry Andric if (isa<IntegerType>(EltTy)) { 880dff0c46cSDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) 881dff0c46cSDimitry Andric Record.push_back(CDS->getElementAsInteger(i)); 882dff0c46cSDimitry Andric } else if (EltTy->isFloatTy()) { 883dff0c46cSDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) { 884dff0c46cSDimitry Andric union { float F; uint32_t I; }; 885dff0c46cSDimitry Andric F = CDS->getElementAsFloat(i); 886dff0c46cSDimitry Andric Record.push_back(I); 887dff0c46cSDimitry Andric } 888dff0c46cSDimitry Andric } else { 889dff0c46cSDimitry Andric assert(EltTy->isDoubleTy() && "Unknown ConstantData element type"); 890dff0c46cSDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) { 891dff0c46cSDimitry Andric union { double F; uint64_t I; }; 892dff0c46cSDimitry Andric F = CDS->getElementAsDouble(i); 893dff0c46cSDimitry Andric Record.push_back(I); 894dff0c46cSDimitry Andric } 895dff0c46cSDimitry Andric } 896dff0c46cSDimitry Andric } else if (isa<ConstantArray>(C) || isa<ConstantStruct>(C) || 897dff0c46cSDimitry Andric isa<ConstantVector>(C)) { 898f22ef01cSRoman Divacky Code = bitc::CST_CODE_AGGREGATE; 899f22ef01cSRoman Divacky for (unsigned i = 0, e = C->getNumOperands(); i != e; ++i) 900f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(i))); 901f22ef01cSRoman Divacky AbbrevToUse = AggregateAbbrev; 902f22ef01cSRoman Divacky } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 903f22ef01cSRoman Divacky switch (CE->getOpcode()) { 904f22ef01cSRoman Divacky default: 905f22ef01cSRoman Divacky if (Instruction::isCast(CE->getOpcode())) { 906f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_CAST; 907f22ef01cSRoman Divacky Record.push_back(GetEncodedCastOpcode(CE->getOpcode())); 908f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 909f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 910f22ef01cSRoman Divacky AbbrevToUse = CONSTANTS_CE_CAST_Abbrev; 911f22ef01cSRoman Divacky } else { 912f22ef01cSRoman Divacky assert(CE->getNumOperands() == 2 && "Unknown constant expr!"); 913f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_BINOP; 914f22ef01cSRoman Divacky Record.push_back(GetEncodedBinaryOpcode(CE->getOpcode())); 915f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 916f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 917f22ef01cSRoman Divacky uint64_t Flags = GetOptimizationFlags(CE); 918f22ef01cSRoman Divacky if (Flags != 0) 919f22ef01cSRoman Divacky Record.push_back(Flags); 920f22ef01cSRoman Divacky } 921f22ef01cSRoman Divacky break; 922f22ef01cSRoman Divacky case Instruction::GetElementPtr: 923f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_GEP; 924f22ef01cSRoman Divacky if (cast<GEPOperator>(C)->isInBounds()) 925f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_INBOUNDS_GEP; 926f22ef01cSRoman Divacky for (unsigned i = 0, e = CE->getNumOperands(); i != e; ++i) { 927f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(i)->getType())); 928f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(i))); 929f22ef01cSRoman Divacky } 930f22ef01cSRoman Divacky break; 931f22ef01cSRoman Divacky case Instruction::Select: 932f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_SELECT; 933f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 934f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 935f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(2))); 936f22ef01cSRoman Divacky break; 937f22ef01cSRoman Divacky case Instruction::ExtractElement: 938f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_EXTRACTELT; 939f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 940f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 941f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 942f22ef01cSRoman Divacky break; 943f22ef01cSRoman Divacky case Instruction::InsertElement: 944f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_INSERTELT; 945f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 946f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 947f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(2))); 948f22ef01cSRoman Divacky break; 949f22ef01cSRoman Divacky case Instruction::ShuffleVector: 950f22ef01cSRoman Divacky // If the return type and argument types are the same, this is a 951f22ef01cSRoman Divacky // standard shufflevector instruction. If the types are different, 952f22ef01cSRoman Divacky // then the shuffle is widening or truncating the input vectors, and 953f22ef01cSRoman Divacky // the argument type must also be encoded. 954f22ef01cSRoman Divacky if (C->getType() == C->getOperand(0)->getType()) { 955f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_SHUFFLEVEC; 956f22ef01cSRoman Divacky } else { 957f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_SHUFVEC_EX; 958f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 959f22ef01cSRoman Divacky } 960f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 961f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 962f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(2))); 963f22ef01cSRoman Divacky break; 964f22ef01cSRoman Divacky case Instruction::ICmp: 965f22ef01cSRoman Divacky case Instruction::FCmp: 966f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_CMP; 967f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 968f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 969f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 970f22ef01cSRoman Divacky Record.push_back(CE->getPredicate()); 971f22ef01cSRoman Divacky break; 972f22ef01cSRoman Divacky } 973f22ef01cSRoman Divacky } else if (const BlockAddress *BA = dyn_cast<BlockAddress>(C)) { 974f22ef01cSRoman Divacky Code = bitc::CST_CODE_BLOCKADDRESS; 975f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(BA->getFunction()->getType())); 976f22ef01cSRoman Divacky Record.push_back(VE.getValueID(BA->getFunction())); 977f22ef01cSRoman Divacky Record.push_back(VE.getGlobalBasicBlockID(BA->getBasicBlock())); 978f22ef01cSRoman Divacky } else { 979e580952dSDimitry Andric #ifndef NDEBUG 980e580952dSDimitry Andric C->dump(); 981e580952dSDimitry Andric #endif 982f22ef01cSRoman Divacky llvm_unreachable("Unknown constant!"); 983f22ef01cSRoman Divacky } 984f22ef01cSRoman Divacky Stream.EmitRecord(Code, Record, AbbrevToUse); 985f22ef01cSRoman Divacky Record.clear(); 986f22ef01cSRoman Divacky } 987f22ef01cSRoman Divacky 988f22ef01cSRoman Divacky Stream.ExitBlock(); 989f22ef01cSRoman Divacky } 990f22ef01cSRoman Divacky 991f22ef01cSRoman Divacky static void WriteModuleConstants(const ValueEnumerator &VE, 992f22ef01cSRoman Divacky BitstreamWriter &Stream) { 993f22ef01cSRoman Divacky const ValueEnumerator::ValueList &Vals = VE.getValues(); 994f22ef01cSRoman Divacky 995f22ef01cSRoman Divacky // Find the first constant to emit, which is the first non-globalvalue value. 996f22ef01cSRoman Divacky // We know globalvalues have been emitted by WriteModuleInfo. 997f22ef01cSRoman Divacky for (unsigned i = 0, e = Vals.size(); i != e; ++i) { 998f22ef01cSRoman Divacky if (!isa<GlobalValue>(Vals[i].first)) { 999f22ef01cSRoman Divacky WriteConstants(i, Vals.size(), VE, Stream, true); 1000f22ef01cSRoman Divacky return; 1001f22ef01cSRoman Divacky } 1002f22ef01cSRoman Divacky } 1003f22ef01cSRoman Divacky } 1004f22ef01cSRoman Divacky 1005f22ef01cSRoman Divacky /// PushValueAndType - The file has to encode both the value and type id for 1006f22ef01cSRoman Divacky /// many values, because we need to know what type to create for forward 1007f22ef01cSRoman Divacky /// references. However, most operands are not forward references, so this type 1008f22ef01cSRoman Divacky /// field is not needed. 1009f22ef01cSRoman Divacky /// 1010f22ef01cSRoman Divacky /// This function adds V's value ID to Vals. If the value ID is higher than the 1011f22ef01cSRoman Divacky /// instruction ID, then it is a forward reference, and it also includes the 1012f22ef01cSRoman Divacky /// type ID. 1013f22ef01cSRoman Divacky static bool PushValueAndType(const Value *V, unsigned InstID, 1014f22ef01cSRoman Divacky SmallVector<unsigned, 64> &Vals, 1015f22ef01cSRoman Divacky ValueEnumerator &VE) { 1016f22ef01cSRoman Divacky unsigned ValID = VE.getValueID(V); 1017f22ef01cSRoman Divacky Vals.push_back(ValID); 1018f22ef01cSRoman Divacky if (ValID >= InstID) { 1019f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(V->getType())); 1020f22ef01cSRoman Divacky return true; 1021f22ef01cSRoman Divacky } 1022f22ef01cSRoman Divacky return false; 1023f22ef01cSRoman Divacky } 1024f22ef01cSRoman Divacky 1025f22ef01cSRoman Divacky /// WriteInstruction - Emit an instruction to the specified stream. 1026f22ef01cSRoman Divacky static void WriteInstruction(const Instruction &I, unsigned InstID, 1027f22ef01cSRoman Divacky ValueEnumerator &VE, BitstreamWriter &Stream, 1028f22ef01cSRoman Divacky SmallVector<unsigned, 64> &Vals) { 1029f22ef01cSRoman Divacky unsigned Code = 0; 1030f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 1031f22ef01cSRoman Divacky VE.setInstructionID(&I); 1032f22ef01cSRoman Divacky switch (I.getOpcode()) { 1033f22ef01cSRoman Divacky default: 1034f22ef01cSRoman Divacky if (Instruction::isCast(I.getOpcode())) { 1035f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_CAST; 1036f22ef01cSRoman Divacky if (!PushValueAndType(I.getOperand(0), InstID, Vals, VE)) 1037f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_CAST_ABBREV; 1038f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getType())); 1039f22ef01cSRoman Divacky Vals.push_back(GetEncodedCastOpcode(I.getOpcode())); 1040f22ef01cSRoman Divacky } else { 1041f22ef01cSRoman Divacky assert(isa<BinaryOperator>(I) && "Unknown instruction!"); 1042f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_BINOP; 1043f22ef01cSRoman Divacky if (!PushValueAndType(I.getOperand(0), InstID, Vals, VE)) 1044f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_BINOP_ABBREV; 1045f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1046f22ef01cSRoman Divacky Vals.push_back(GetEncodedBinaryOpcode(I.getOpcode())); 1047f22ef01cSRoman Divacky uint64_t Flags = GetOptimizationFlags(&I); 1048f22ef01cSRoman Divacky if (Flags != 0) { 1049f22ef01cSRoman Divacky if (AbbrevToUse == FUNCTION_INST_BINOP_ABBREV) 1050f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_BINOP_FLAGS_ABBREV; 1051f22ef01cSRoman Divacky Vals.push_back(Flags); 1052f22ef01cSRoman Divacky } 1053f22ef01cSRoman Divacky } 1054f22ef01cSRoman Divacky break; 1055f22ef01cSRoman Divacky 1056f22ef01cSRoman Divacky case Instruction::GetElementPtr: 1057f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_GEP; 1058f22ef01cSRoman Divacky if (cast<GEPOperator>(&I)->isInBounds()) 1059f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INBOUNDS_GEP; 1060f22ef01cSRoman Divacky for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i) 1061f22ef01cSRoman Divacky PushValueAndType(I.getOperand(i), InstID, Vals, VE); 1062f22ef01cSRoman Divacky break; 1063f22ef01cSRoman Divacky case Instruction::ExtractValue: { 1064f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_EXTRACTVAL; 1065f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1066f22ef01cSRoman Divacky const ExtractValueInst *EVI = cast<ExtractValueInst>(&I); 1067f22ef01cSRoman Divacky for (const unsigned *i = EVI->idx_begin(), *e = EVI->idx_end(); i != e; ++i) 1068f22ef01cSRoman Divacky Vals.push_back(*i); 1069f22ef01cSRoman Divacky break; 1070f22ef01cSRoman Divacky } 1071f22ef01cSRoman Divacky case Instruction::InsertValue: { 1072f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INSERTVAL; 1073f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1074f22ef01cSRoman Divacky PushValueAndType(I.getOperand(1), InstID, Vals, VE); 1075f22ef01cSRoman Divacky const InsertValueInst *IVI = cast<InsertValueInst>(&I); 1076f22ef01cSRoman Divacky for (const unsigned *i = IVI->idx_begin(), *e = IVI->idx_end(); i != e; ++i) 1077f22ef01cSRoman Divacky Vals.push_back(*i); 1078f22ef01cSRoman Divacky break; 1079f22ef01cSRoman Divacky } 1080f22ef01cSRoman Divacky case Instruction::Select: 1081f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_VSELECT; 1082f22ef01cSRoman Divacky PushValueAndType(I.getOperand(1), InstID, Vals, VE); 1083f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(2))); 1084f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1085f22ef01cSRoman Divacky break; 1086f22ef01cSRoman Divacky case Instruction::ExtractElement: 1087f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_EXTRACTELT; 1088f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1089f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1090f22ef01cSRoman Divacky break; 1091f22ef01cSRoman Divacky case Instruction::InsertElement: 1092f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INSERTELT; 1093f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1094f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1095f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(2))); 1096f22ef01cSRoman Divacky break; 1097f22ef01cSRoman Divacky case Instruction::ShuffleVector: 1098f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_SHUFFLEVEC; 1099f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1100f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1101f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(2))); 1102f22ef01cSRoman Divacky break; 1103f22ef01cSRoman Divacky case Instruction::ICmp: 1104f22ef01cSRoman Divacky case Instruction::FCmp: 1105f22ef01cSRoman Divacky // compare returning Int1Ty or vector of Int1Ty 1106f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_CMP2; 1107f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1108f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1109f22ef01cSRoman Divacky Vals.push_back(cast<CmpInst>(I).getPredicate()); 1110f22ef01cSRoman Divacky break; 1111f22ef01cSRoman Divacky 1112f22ef01cSRoman Divacky case Instruction::Ret: 1113f22ef01cSRoman Divacky { 1114f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_RET; 1115f22ef01cSRoman Divacky unsigned NumOperands = I.getNumOperands(); 1116f22ef01cSRoman Divacky if (NumOperands == 0) 1117f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_RET_VOID_ABBREV; 1118f22ef01cSRoman Divacky else if (NumOperands == 1) { 1119f22ef01cSRoman Divacky if (!PushValueAndType(I.getOperand(0), InstID, Vals, VE)) 1120f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_RET_VAL_ABBREV; 1121f22ef01cSRoman Divacky } else { 1122f22ef01cSRoman Divacky for (unsigned i = 0, e = NumOperands; i != e; ++i) 1123f22ef01cSRoman Divacky PushValueAndType(I.getOperand(i), InstID, Vals, VE); 1124f22ef01cSRoman Divacky } 1125f22ef01cSRoman Divacky } 1126f22ef01cSRoman Divacky break; 1127f22ef01cSRoman Divacky case Instruction::Br: 1128f22ef01cSRoman Divacky { 1129f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_BR; 1130f22ef01cSRoman Divacky BranchInst &II = cast<BranchInst>(I); 1131f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II.getSuccessor(0))); 1132f22ef01cSRoman Divacky if (II.isConditional()) { 1133f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II.getSuccessor(1))); 1134f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II.getCondition())); 1135f22ef01cSRoman Divacky } 1136f22ef01cSRoman Divacky } 1137f22ef01cSRoman Divacky break; 1138f22ef01cSRoman Divacky case Instruction::Switch: 1139dff0c46cSDimitry Andric { 1140f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_SWITCH; 1141dff0c46cSDimitry Andric SwitchInst &SI = cast<SwitchInst>(I); 1142dff0c46cSDimitry Andric Vals.push_back(VE.getTypeID(SI.getCondition()->getType())); 1143dff0c46cSDimitry Andric Vals.push_back(VE.getValueID(SI.getCondition())); 1144dff0c46cSDimitry Andric Vals.push_back(VE.getValueID(SI.getDefaultDest())); 1145dff0c46cSDimitry Andric for (SwitchInst::CaseIt i = SI.case_begin(), e = SI.case_end(); 1146dff0c46cSDimitry Andric i != e; ++i) { 1147dff0c46cSDimitry Andric Vals.push_back(VE.getValueID(i.getCaseValue())); 1148dff0c46cSDimitry Andric Vals.push_back(VE.getValueID(i.getCaseSuccessor())); 1149dff0c46cSDimitry Andric } 1150dff0c46cSDimitry Andric } 1151f22ef01cSRoman Divacky break; 1152f22ef01cSRoman Divacky case Instruction::IndirectBr: 1153f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INDIRECTBR; 1154f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); 1155f22ef01cSRoman Divacky for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i) 1156f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(i))); 1157f22ef01cSRoman Divacky break; 1158f22ef01cSRoman Divacky 1159f22ef01cSRoman Divacky case Instruction::Invoke: { 1160f22ef01cSRoman Divacky const InvokeInst *II = cast<InvokeInst>(&I); 1161f22ef01cSRoman Divacky const Value *Callee(II->getCalledValue()); 11626122f3e6SDimitry Andric PointerType *PTy = cast<PointerType>(Callee->getType()); 11636122f3e6SDimitry Andric FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 1164f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INVOKE; 1165f22ef01cSRoman Divacky 1166f22ef01cSRoman Divacky Vals.push_back(VE.getAttributeID(II->getAttributes())); 1167f22ef01cSRoman Divacky Vals.push_back(II->getCallingConv()); 1168f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II->getNormalDest())); 1169f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II->getUnwindDest())); 1170f22ef01cSRoman Divacky PushValueAndType(Callee, InstID, Vals, VE); 1171f22ef01cSRoman Divacky 1172f22ef01cSRoman Divacky // Emit value #'s for the fixed parameters. 1173f22ef01cSRoman Divacky for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) 1174f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(i))); // fixed param. 1175f22ef01cSRoman Divacky 1176f22ef01cSRoman Divacky // Emit type/value pairs for varargs params. 1177f22ef01cSRoman Divacky if (FTy->isVarArg()) { 1178f22ef01cSRoman Divacky for (unsigned i = FTy->getNumParams(), e = I.getNumOperands()-3; 1179f22ef01cSRoman Divacky i != e; ++i) 1180f22ef01cSRoman Divacky PushValueAndType(I.getOperand(i), InstID, Vals, VE); // vararg 1181f22ef01cSRoman Divacky } 1182f22ef01cSRoman Divacky break; 1183f22ef01cSRoman Divacky } 11846122f3e6SDimitry Andric case Instruction::Resume: 11856122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_RESUME; 11866122f3e6SDimitry Andric PushValueAndType(I.getOperand(0), InstID, Vals, VE); 11876122f3e6SDimitry Andric break; 1188f22ef01cSRoman Divacky case Instruction::Unreachable: 1189f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_UNREACHABLE; 1190f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_UNREACHABLE_ABBREV; 1191f22ef01cSRoman Divacky break; 1192f22ef01cSRoman Divacky 119317a519f9SDimitry Andric case Instruction::PHI: { 119417a519f9SDimitry Andric const PHINode &PN = cast<PHINode>(I); 1195f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_PHI; 119617a519f9SDimitry Andric Vals.push_back(VE.getTypeID(PN.getType())); 119717a519f9SDimitry Andric for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) { 119817a519f9SDimitry Andric Vals.push_back(VE.getValueID(PN.getIncomingValue(i))); 119917a519f9SDimitry Andric Vals.push_back(VE.getValueID(PN.getIncomingBlock(i))); 120017a519f9SDimitry Andric } 1201f22ef01cSRoman Divacky break; 120217a519f9SDimitry Andric } 1203f22ef01cSRoman Divacky 12046122f3e6SDimitry Andric case Instruction::LandingPad: { 12056122f3e6SDimitry Andric const LandingPadInst &LP = cast<LandingPadInst>(I); 12066122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_LANDINGPAD; 12076122f3e6SDimitry Andric Vals.push_back(VE.getTypeID(LP.getType())); 12086122f3e6SDimitry Andric PushValueAndType(LP.getPersonalityFn(), InstID, Vals, VE); 12096122f3e6SDimitry Andric Vals.push_back(LP.isCleanup()); 12106122f3e6SDimitry Andric Vals.push_back(LP.getNumClauses()); 12116122f3e6SDimitry Andric for (unsigned I = 0, E = LP.getNumClauses(); I != E; ++I) { 12126122f3e6SDimitry Andric if (LP.isCatch(I)) 12136122f3e6SDimitry Andric Vals.push_back(LandingPadInst::Catch); 12146122f3e6SDimitry Andric else 12156122f3e6SDimitry Andric Vals.push_back(LandingPadInst::Filter); 12166122f3e6SDimitry Andric PushValueAndType(LP.getClause(I), InstID, Vals, VE); 12176122f3e6SDimitry Andric } 12186122f3e6SDimitry Andric break; 12196122f3e6SDimitry Andric } 12206122f3e6SDimitry Andric 1221f22ef01cSRoman Divacky case Instruction::Alloca: 1222f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_ALLOCA; 1223f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getType())); 1224ffd1746dSEd Schouten Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); 1225f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(0))); // size. 1226f22ef01cSRoman Divacky Vals.push_back(Log2_32(cast<AllocaInst>(I).getAlignment())+1); 1227f22ef01cSRoman Divacky break; 1228f22ef01cSRoman Divacky 1229f22ef01cSRoman Divacky case Instruction::Load: 12306122f3e6SDimitry Andric if (cast<LoadInst>(I).isAtomic()) { 12316122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_LOADATOMIC; 12326122f3e6SDimitry Andric PushValueAndType(I.getOperand(0), InstID, Vals, VE); 12336122f3e6SDimitry Andric } else { 1234f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_LOAD; 1235f22ef01cSRoman Divacky if (!PushValueAndType(I.getOperand(0), InstID, Vals, VE)) // ptr 1236f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_LOAD_ABBREV; 12376122f3e6SDimitry Andric } 1238f22ef01cSRoman Divacky Vals.push_back(Log2_32(cast<LoadInst>(I).getAlignment())+1); 1239f22ef01cSRoman Divacky Vals.push_back(cast<LoadInst>(I).isVolatile()); 12406122f3e6SDimitry Andric if (cast<LoadInst>(I).isAtomic()) { 12416122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering(cast<LoadInst>(I).getOrdering())); 12426122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope(cast<LoadInst>(I).getSynchScope())); 12436122f3e6SDimitry Andric } 1244f22ef01cSRoman Divacky break; 1245f22ef01cSRoman Divacky case Instruction::Store: 12466122f3e6SDimitry Andric if (cast<StoreInst>(I).isAtomic()) 12476122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_STOREATOMIC; 12486122f3e6SDimitry Andric else 124917a519f9SDimitry Andric Code = bitc::FUNC_CODE_INST_STORE; 1250f22ef01cSRoman Divacky PushValueAndType(I.getOperand(1), InstID, Vals, VE); // ptrty + ptr 1251f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(0))); // val. 1252f22ef01cSRoman Divacky Vals.push_back(Log2_32(cast<StoreInst>(I).getAlignment())+1); 1253f22ef01cSRoman Divacky Vals.push_back(cast<StoreInst>(I).isVolatile()); 12546122f3e6SDimitry Andric if (cast<StoreInst>(I).isAtomic()) { 12556122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering(cast<StoreInst>(I).getOrdering())); 12566122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope(cast<StoreInst>(I).getSynchScope())); 12576122f3e6SDimitry Andric } 12586122f3e6SDimitry Andric break; 12596122f3e6SDimitry Andric case Instruction::AtomicCmpXchg: 12606122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_CMPXCHG; 12616122f3e6SDimitry Andric PushValueAndType(I.getOperand(0), InstID, Vals, VE); // ptrty + ptr 12626122f3e6SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(1))); // cmp. 12636122f3e6SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(2))); // newval. 12646122f3e6SDimitry Andric Vals.push_back(cast<AtomicCmpXchgInst>(I).isVolatile()); 12656122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering( 12666122f3e6SDimitry Andric cast<AtomicCmpXchgInst>(I).getOrdering())); 12676122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope( 12686122f3e6SDimitry Andric cast<AtomicCmpXchgInst>(I).getSynchScope())); 12696122f3e6SDimitry Andric break; 12706122f3e6SDimitry Andric case Instruction::AtomicRMW: 12716122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_ATOMICRMW; 12726122f3e6SDimitry Andric PushValueAndType(I.getOperand(0), InstID, Vals, VE); // ptrty + ptr 12736122f3e6SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(1))); // val. 12746122f3e6SDimitry Andric Vals.push_back(GetEncodedRMWOperation( 12756122f3e6SDimitry Andric cast<AtomicRMWInst>(I).getOperation())); 12766122f3e6SDimitry Andric Vals.push_back(cast<AtomicRMWInst>(I).isVolatile()); 12776122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering(cast<AtomicRMWInst>(I).getOrdering())); 12786122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope( 12796122f3e6SDimitry Andric cast<AtomicRMWInst>(I).getSynchScope())); 12806122f3e6SDimitry Andric break; 12816122f3e6SDimitry Andric case Instruction::Fence: 12826122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_FENCE; 12836122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering(cast<FenceInst>(I).getOrdering())); 12846122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope(cast<FenceInst>(I).getSynchScope())); 1285f22ef01cSRoman Divacky break; 1286f22ef01cSRoman Divacky case Instruction::Call: { 1287ffd1746dSEd Schouten const CallInst &CI = cast<CallInst>(I); 12886122f3e6SDimitry Andric PointerType *PTy = cast<PointerType>(CI.getCalledValue()->getType()); 12896122f3e6SDimitry Andric FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 1290f22ef01cSRoman Divacky 129117a519f9SDimitry Andric Code = bitc::FUNC_CODE_INST_CALL; 1292f22ef01cSRoman Divacky 1293ffd1746dSEd Schouten Vals.push_back(VE.getAttributeID(CI.getAttributes())); 1294ffd1746dSEd Schouten Vals.push_back((CI.getCallingConv() << 1) | unsigned(CI.isTailCall())); 1295ffd1746dSEd Schouten PushValueAndType(CI.getCalledValue(), InstID, Vals, VE); // Callee 1296f22ef01cSRoman Divacky 1297f22ef01cSRoman Divacky // Emit value #'s for the fixed parameters. 1298f22ef01cSRoman Divacky for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) 1299ffd1746dSEd Schouten Vals.push_back(VE.getValueID(CI.getArgOperand(i))); // fixed param. 1300f22ef01cSRoman Divacky 1301f22ef01cSRoman Divacky // Emit type/value pairs for varargs params. 1302f22ef01cSRoman Divacky if (FTy->isVarArg()) { 1303ffd1746dSEd Schouten for (unsigned i = FTy->getNumParams(), e = CI.getNumArgOperands(); 1304f22ef01cSRoman Divacky i != e; ++i) 1305ffd1746dSEd Schouten PushValueAndType(CI.getArgOperand(i), InstID, Vals, VE); // varargs 1306f22ef01cSRoman Divacky } 1307f22ef01cSRoman Divacky break; 1308f22ef01cSRoman Divacky } 1309f22ef01cSRoman Divacky case Instruction::VAArg: 1310f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_VAARG; 1311f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); // valistty 1312f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(0))); // valist. 1313f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getType())); // restype. 1314f22ef01cSRoman Divacky break; 1315f22ef01cSRoman Divacky } 1316f22ef01cSRoman Divacky 1317f22ef01cSRoman Divacky Stream.EmitRecord(Code, Vals, AbbrevToUse); 1318f22ef01cSRoman Divacky Vals.clear(); 1319f22ef01cSRoman Divacky } 1320f22ef01cSRoman Divacky 1321f22ef01cSRoman Divacky // Emit names for globals/functions etc. 1322f22ef01cSRoman Divacky static void WriteValueSymbolTable(const ValueSymbolTable &VST, 1323f22ef01cSRoman Divacky const ValueEnumerator &VE, 1324f22ef01cSRoman Divacky BitstreamWriter &Stream) { 1325f22ef01cSRoman Divacky if (VST.empty()) return; 1326f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::VALUE_SYMTAB_BLOCK_ID, 4); 1327f22ef01cSRoman Divacky 1328f22ef01cSRoman Divacky // FIXME: Set up the abbrev, we know how many values there are! 1329f22ef01cSRoman Divacky // FIXME: We know if the type names can use 7-bit ascii. 1330f22ef01cSRoman Divacky SmallVector<unsigned, 64> NameVals; 1331f22ef01cSRoman Divacky 1332f22ef01cSRoman Divacky for (ValueSymbolTable::const_iterator SI = VST.begin(), SE = VST.end(); 1333f22ef01cSRoman Divacky SI != SE; ++SI) { 1334f22ef01cSRoman Divacky 1335f22ef01cSRoman Divacky const ValueName &Name = *SI; 1336f22ef01cSRoman Divacky 1337f22ef01cSRoman Divacky // Figure out the encoding to use for the name. 1338f22ef01cSRoman Divacky bool is7Bit = true; 1339f22ef01cSRoman Divacky bool isChar6 = true; 1340f22ef01cSRoman Divacky for (const char *C = Name.getKeyData(), *E = C+Name.getKeyLength(); 1341f22ef01cSRoman Divacky C != E; ++C) { 1342f22ef01cSRoman Divacky if (isChar6) 1343f22ef01cSRoman Divacky isChar6 = BitCodeAbbrevOp::isChar6(*C); 1344f22ef01cSRoman Divacky if ((unsigned char)*C & 128) { 1345f22ef01cSRoman Divacky is7Bit = false; 1346f22ef01cSRoman Divacky break; // don't bother scanning the rest. 1347f22ef01cSRoman Divacky } 1348f22ef01cSRoman Divacky } 1349f22ef01cSRoman Divacky 1350f22ef01cSRoman Divacky unsigned AbbrevToUse = VST_ENTRY_8_ABBREV; 1351f22ef01cSRoman Divacky 1352f22ef01cSRoman Divacky // VST_ENTRY: [valueid, namechar x N] 1353f22ef01cSRoman Divacky // VST_BBENTRY: [bbid, namechar x N] 1354f22ef01cSRoman Divacky unsigned Code; 1355f22ef01cSRoman Divacky if (isa<BasicBlock>(SI->getValue())) { 1356f22ef01cSRoman Divacky Code = bitc::VST_CODE_BBENTRY; 1357f22ef01cSRoman Divacky if (isChar6) 1358f22ef01cSRoman Divacky AbbrevToUse = VST_BBENTRY_6_ABBREV; 1359f22ef01cSRoman Divacky } else { 1360f22ef01cSRoman Divacky Code = bitc::VST_CODE_ENTRY; 1361f22ef01cSRoman Divacky if (isChar6) 1362f22ef01cSRoman Divacky AbbrevToUse = VST_ENTRY_6_ABBREV; 1363f22ef01cSRoman Divacky else if (is7Bit) 1364f22ef01cSRoman Divacky AbbrevToUse = VST_ENTRY_7_ABBREV; 1365f22ef01cSRoman Divacky } 1366f22ef01cSRoman Divacky 1367f22ef01cSRoman Divacky NameVals.push_back(VE.getValueID(SI->getValue())); 1368f22ef01cSRoman Divacky for (const char *P = Name.getKeyData(), 1369f22ef01cSRoman Divacky *E = Name.getKeyData()+Name.getKeyLength(); P != E; ++P) 1370f22ef01cSRoman Divacky NameVals.push_back((unsigned char)*P); 1371f22ef01cSRoman Divacky 1372f22ef01cSRoman Divacky // Emit the finished record. 1373f22ef01cSRoman Divacky Stream.EmitRecord(Code, NameVals, AbbrevToUse); 1374f22ef01cSRoman Divacky NameVals.clear(); 1375f22ef01cSRoman Divacky } 1376f22ef01cSRoman Divacky Stream.ExitBlock(); 1377f22ef01cSRoman Divacky } 1378f22ef01cSRoman Divacky 1379f22ef01cSRoman Divacky /// WriteFunction - Emit a function body to the module stream. 1380f22ef01cSRoman Divacky static void WriteFunction(const Function &F, ValueEnumerator &VE, 1381f22ef01cSRoman Divacky BitstreamWriter &Stream) { 1382f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::FUNCTION_BLOCK_ID, 4); 1383f22ef01cSRoman Divacky VE.incorporateFunction(F); 1384f22ef01cSRoman Divacky 1385f22ef01cSRoman Divacky SmallVector<unsigned, 64> Vals; 1386f22ef01cSRoman Divacky 1387f22ef01cSRoman Divacky // Emit the number of basic blocks, so the reader can create them ahead of 1388f22ef01cSRoman Divacky // time. 1389f22ef01cSRoman Divacky Vals.push_back(VE.getBasicBlocks().size()); 1390f22ef01cSRoman Divacky Stream.EmitRecord(bitc::FUNC_CODE_DECLAREBLOCKS, Vals); 1391f22ef01cSRoman Divacky Vals.clear(); 1392f22ef01cSRoman Divacky 1393f22ef01cSRoman Divacky // If there are function-local constants, emit them now. 1394f22ef01cSRoman Divacky unsigned CstStart, CstEnd; 1395f22ef01cSRoman Divacky VE.getFunctionConstantRange(CstStart, CstEnd); 1396f22ef01cSRoman Divacky WriteConstants(CstStart, CstEnd, VE, Stream, false); 1397f22ef01cSRoman Divacky 1398f22ef01cSRoman Divacky // If there is function-local metadata, emit it now. 1399f22ef01cSRoman Divacky WriteFunctionLocalMetadata(F, VE, Stream); 1400f22ef01cSRoman Divacky 1401f22ef01cSRoman Divacky // Keep a running idea of what the instruction ID is. 1402f22ef01cSRoman Divacky unsigned InstID = CstEnd; 1403f22ef01cSRoman Divacky 1404f22ef01cSRoman Divacky bool NeedsMetadataAttachment = false; 1405f22ef01cSRoman Divacky 1406f22ef01cSRoman Divacky DebugLoc LastDL; 1407f22ef01cSRoman Divacky 1408f22ef01cSRoman Divacky // Finally, emit all the instructions, in order. 1409f22ef01cSRoman Divacky for (Function::const_iterator BB = F.begin(), E = F.end(); BB != E; ++BB) 1410f22ef01cSRoman Divacky for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); 1411f22ef01cSRoman Divacky I != E; ++I) { 1412f22ef01cSRoman Divacky WriteInstruction(*I, InstID, VE, Stream, Vals); 1413f22ef01cSRoman Divacky 1414f22ef01cSRoman Divacky if (!I->getType()->isVoidTy()) 1415f22ef01cSRoman Divacky ++InstID; 1416f22ef01cSRoman Divacky 1417f22ef01cSRoman Divacky // If the instruction has metadata, write a metadata attachment later. 1418f22ef01cSRoman Divacky NeedsMetadataAttachment |= I->hasMetadataOtherThanDebugLoc(); 1419f22ef01cSRoman Divacky 1420f22ef01cSRoman Divacky // If the instruction has a debug location, emit it. 1421f22ef01cSRoman Divacky DebugLoc DL = I->getDebugLoc(); 1422f22ef01cSRoman Divacky if (DL.isUnknown()) { 1423f22ef01cSRoman Divacky // nothing todo. 1424f22ef01cSRoman Divacky } else if (DL == LastDL) { 1425f22ef01cSRoman Divacky // Just repeat the same debug loc as last time. 1426f22ef01cSRoman Divacky Stream.EmitRecord(bitc::FUNC_CODE_DEBUG_LOC_AGAIN, Vals); 1427f22ef01cSRoman Divacky } else { 1428f22ef01cSRoman Divacky MDNode *Scope, *IA; 1429f22ef01cSRoman Divacky DL.getScopeAndInlinedAt(Scope, IA, I->getContext()); 1430f22ef01cSRoman Divacky 1431f22ef01cSRoman Divacky Vals.push_back(DL.getLine()); 1432f22ef01cSRoman Divacky Vals.push_back(DL.getCol()); 1433f22ef01cSRoman Divacky Vals.push_back(Scope ? VE.getValueID(Scope)+1 : 0); 1434f22ef01cSRoman Divacky Vals.push_back(IA ? VE.getValueID(IA)+1 : 0); 143517a519f9SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_DEBUG_LOC, Vals); 1436f22ef01cSRoman Divacky Vals.clear(); 1437f22ef01cSRoman Divacky 1438f22ef01cSRoman Divacky LastDL = DL; 1439f22ef01cSRoman Divacky } 1440f22ef01cSRoman Divacky } 1441f22ef01cSRoman Divacky 1442f22ef01cSRoman Divacky // Emit names for all the instructions etc. 1443f22ef01cSRoman Divacky WriteValueSymbolTable(F.getValueSymbolTable(), VE, Stream); 1444f22ef01cSRoman Divacky 1445f22ef01cSRoman Divacky if (NeedsMetadataAttachment) 1446f22ef01cSRoman Divacky WriteMetadataAttachment(F, VE, Stream); 1447f22ef01cSRoman Divacky VE.purgeFunction(); 1448f22ef01cSRoman Divacky Stream.ExitBlock(); 1449f22ef01cSRoman Divacky } 1450f22ef01cSRoman Divacky 1451f22ef01cSRoman Divacky // Emit blockinfo, which defines the standard abbreviations etc. 1452f22ef01cSRoman Divacky static void WriteBlockInfo(const ValueEnumerator &VE, BitstreamWriter &Stream) { 1453f22ef01cSRoman Divacky // We only want to emit block info records for blocks that have multiple 1454f22ef01cSRoman Divacky // instances: CONSTANTS_BLOCK, FUNCTION_BLOCK and VALUE_SYMTAB_BLOCK. Other 1455f22ef01cSRoman Divacky // blocks can defined their abbrevs inline. 1456f22ef01cSRoman Divacky Stream.EnterBlockInfoBlock(2); 1457f22ef01cSRoman Divacky 1458f22ef01cSRoman Divacky { // 8-bit fixed-width VST_ENTRY/VST_BBENTRY strings. 1459f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1460f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 3)); 1461f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1462f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1463f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 1464f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, 1465f22ef01cSRoman Divacky Abbv) != VST_ENTRY_8_ABBREV) 1466f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1467f22ef01cSRoman Divacky } 1468f22ef01cSRoman Divacky 1469f22ef01cSRoman Divacky { // 7-bit fixed width VST_ENTRY strings. 1470f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1471f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_ENTRY)); 1472f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1473f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1474f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 1475f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, 1476f22ef01cSRoman Divacky Abbv) != VST_ENTRY_7_ABBREV) 1477f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1478f22ef01cSRoman Divacky } 1479f22ef01cSRoman Divacky { // 6-bit char6 VST_ENTRY strings. 1480f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1481f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_ENTRY)); 1482f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1483f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1484f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 1485f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, 1486f22ef01cSRoman Divacky Abbv) != VST_ENTRY_6_ABBREV) 1487f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1488f22ef01cSRoman Divacky } 1489f22ef01cSRoman Divacky { // 6-bit char6 VST_BBENTRY strings. 1490f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1491f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_BBENTRY)); 1492f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1493f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1494f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 1495f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, 1496f22ef01cSRoman Divacky Abbv) != VST_BBENTRY_6_ABBREV) 1497f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1498f22ef01cSRoman Divacky } 1499f22ef01cSRoman Divacky 1500f22ef01cSRoman Divacky 1501f22ef01cSRoman Divacky 1502f22ef01cSRoman Divacky { // SETTYPE abbrev for CONSTANTS_BLOCK. 1503f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1504f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_SETTYPE)); 1505f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1506f22ef01cSRoman Divacky Log2_32_Ceil(VE.getTypes().size()+1))); 1507f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, 1508f22ef01cSRoman Divacky Abbv) != CONSTANTS_SETTYPE_ABBREV) 1509f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1510f22ef01cSRoman Divacky } 1511f22ef01cSRoman Divacky 1512f22ef01cSRoman Divacky { // INTEGER abbrev for CONSTANTS_BLOCK. 1513f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1514f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_INTEGER)); 1515f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1516f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, 1517f22ef01cSRoman Divacky Abbv) != CONSTANTS_INTEGER_ABBREV) 1518f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1519f22ef01cSRoman Divacky } 1520f22ef01cSRoman Divacky 1521f22ef01cSRoman Divacky { // CE_CAST abbrev for CONSTANTS_BLOCK. 1522f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1523f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CE_CAST)); 1524f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // cast opc 1525f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // typeid 1526f22ef01cSRoman Divacky Log2_32_Ceil(VE.getTypes().size()+1))); 1527f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // value id 1528f22ef01cSRoman Divacky 1529f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, 1530f22ef01cSRoman Divacky Abbv) != CONSTANTS_CE_CAST_Abbrev) 1531f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1532f22ef01cSRoman Divacky } 1533f22ef01cSRoman Divacky { // NULL abbrev for CONSTANTS_BLOCK. 1534f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1535f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_NULL)); 1536f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, 1537f22ef01cSRoman Divacky Abbv) != CONSTANTS_NULL_Abbrev) 1538f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1539f22ef01cSRoman Divacky } 1540f22ef01cSRoman Divacky 1541f22ef01cSRoman Divacky // FIXME: This should only use space for first class types! 1542f22ef01cSRoman Divacky 1543f22ef01cSRoman Divacky { // INST_LOAD abbrev for FUNCTION_BLOCK. 1544f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1545f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_LOAD)); 1546f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // Ptr 1547f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // Align 1548f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // volatile 1549f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1550f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_LOAD_ABBREV) 1551f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1552f22ef01cSRoman Divacky } 1553f22ef01cSRoman Divacky { // INST_BINOP abbrev for FUNCTION_BLOCK. 1554f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1555f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_BINOP)); 1556f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 1557f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // RHS 1558f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 1559f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1560f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_BINOP_ABBREV) 1561f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1562f22ef01cSRoman Divacky } 1563f22ef01cSRoman Divacky { // INST_BINOP_FLAGS abbrev for FUNCTION_BLOCK. 1564f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1565f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_BINOP)); 1566f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 1567f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // RHS 1568f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 1569f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); // flags 1570f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1571f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_BINOP_FLAGS_ABBREV) 1572f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1573f22ef01cSRoman Divacky } 1574f22ef01cSRoman Divacky { // INST_CAST abbrev for FUNCTION_BLOCK. 1575f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1576f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_CAST)); 1577f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // OpVal 1578f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // dest ty 1579f22ef01cSRoman Divacky Log2_32_Ceil(VE.getTypes().size()+1))); 1580f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 1581f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1582f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_CAST_ABBREV) 1583f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1584f22ef01cSRoman Divacky } 1585f22ef01cSRoman Divacky 1586f22ef01cSRoman Divacky { // INST_RET abbrev for FUNCTION_BLOCK. 1587f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1588f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_RET)); 1589f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1590f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_RET_VOID_ABBREV) 1591f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1592f22ef01cSRoman Divacky } 1593f22ef01cSRoman Divacky { // INST_RET abbrev for FUNCTION_BLOCK. 1594f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1595f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_RET)); 1596f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // ValID 1597f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1598f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_RET_VAL_ABBREV) 1599f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1600f22ef01cSRoman Divacky } 1601f22ef01cSRoman Divacky { // INST_UNREACHABLE abbrev for FUNCTION_BLOCK. 1602f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1603f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_UNREACHABLE)); 1604f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1605f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_UNREACHABLE_ABBREV) 1606f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1607f22ef01cSRoman Divacky } 1608f22ef01cSRoman Divacky 1609f22ef01cSRoman Divacky Stream.ExitBlock(); 1610f22ef01cSRoman Divacky } 1611f22ef01cSRoman Divacky 1612dff0c46cSDimitry Andric // Sort the Users based on the order in which the reader parses the bitcode 1613dff0c46cSDimitry Andric // file. 1614dff0c46cSDimitry Andric static bool bitcodereader_order(const User *lhs, const User *rhs) { 1615dff0c46cSDimitry Andric // TODO: Implement. 1616dff0c46cSDimitry Andric return true; 1617dff0c46cSDimitry Andric } 1618dff0c46cSDimitry Andric 1619dff0c46cSDimitry Andric static void WriteUseList(const Value *V, const ValueEnumerator &VE, 1620dff0c46cSDimitry Andric BitstreamWriter &Stream) { 1621dff0c46cSDimitry Andric 1622dff0c46cSDimitry Andric // One or zero uses can't get out of order. 1623dff0c46cSDimitry Andric if (V->use_empty() || V->hasNUses(1)) 1624dff0c46cSDimitry Andric return; 1625dff0c46cSDimitry Andric 1626dff0c46cSDimitry Andric // Make a copy of the in-memory use-list for sorting. 1627dff0c46cSDimitry Andric unsigned UseListSize = std::distance(V->use_begin(), V->use_end()); 1628dff0c46cSDimitry Andric SmallVector<const User*, 8> UseList; 1629dff0c46cSDimitry Andric UseList.reserve(UseListSize); 1630dff0c46cSDimitry Andric for (Value::const_use_iterator I = V->use_begin(), E = V->use_end(); 1631dff0c46cSDimitry Andric I != E; ++I) { 1632dff0c46cSDimitry Andric const User *U = *I; 1633dff0c46cSDimitry Andric UseList.push_back(U); 1634dff0c46cSDimitry Andric } 1635dff0c46cSDimitry Andric 1636dff0c46cSDimitry Andric // Sort the copy based on the order read by the BitcodeReader. 1637dff0c46cSDimitry Andric std::sort(UseList.begin(), UseList.end(), bitcodereader_order); 1638dff0c46cSDimitry Andric 1639dff0c46cSDimitry Andric // TODO: Generate a diff between the BitcodeWriter in-memory use-list and the 1640dff0c46cSDimitry Andric // sorted list (i.e., the expected BitcodeReader in-memory use-list). 1641dff0c46cSDimitry Andric 1642dff0c46cSDimitry Andric // TODO: Emit the USELIST_CODE_ENTRYs. 1643dff0c46cSDimitry Andric } 1644dff0c46cSDimitry Andric 1645dff0c46cSDimitry Andric static void WriteFunctionUseList(const Function *F, ValueEnumerator &VE, 1646dff0c46cSDimitry Andric BitstreamWriter &Stream) { 1647dff0c46cSDimitry Andric VE.incorporateFunction(*F); 1648dff0c46cSDimitry Andric 1649dff0c46cSDimitry Andric for (Function::const_arg_iterator AI = F->arg_begin(), AE = F->arg_end(); 1650dff0c46cSDimitry Andric AI != AE; ++AI) 1651dff0c46cSDimitry Andric WriteUseList(AI, VE, Stream); 1652dff0c46cSDimitry Andric for (Function::const_iterator BB = F->begin(), FE = F->end(); BB != FE; 1653dff0c46cSDimitry Andric ++BB) { 1654dff0c46cSDimitry Andric WriteUseList(BB, VE, Stream); 1655dff0c46cSDimitry Andric for (BasicBlock::const_iterator II = BB->begin(), IE = BB->end(); II != IE; 1656dff0c46cSDimitry Andric ++II) { 1657dff0c46cSDimitry Andric WriteUseList(II, VE, Stream); 1658dff0c46cSDimitry Andric for (User::const_op_iterator OI = II->op_begin(), E = II->op_end(); 1659dff0c46cSDimitry Andric OI != E; ++OI) { 1660dff0c46cSDimitry Andric if ((isa<Constant>(*OI) && !isa<GlobalValue>(*OI)) || 1661dff0c46cSDimitry Andric isa<InlineAsm>(*OI)) 1662dff0c46cSDimitry Andric WriteUseList(*OI, VE, Stream); 1663dff0c46cSDimitry Andric } 1664dff0c46cSDimitry Andric } 1665dff0c46cSDimitry Andric } 1666dff0c46cSDimitry Andric VE.purgeFunction(); 1667dff0c46cSDimitry Andric } 1668dff0c46cSDimitry Andric 1669dff0c46cSDimitry Andric // Emit use-lists. 1670dff0c46cSDimitry Andric static void WriteModuleUseLists(const Module *M, ValueEnumerator &VE, 1671dff0c46cSDimitry Andric BitstreamWriter &Stream) { 1672dff0c46cSDimitry Andric Stream.EnterSubblock(bitc::USELIST_BLOCK_ID, 3); 1673dff0c46cSDimitry Andric 1674dff0c46cSDimitry Andric // XXX: this modifies the module, but in a way that should never change the 1675dff0c46cSDimitry Andric // behavior of any pass or codegen in LLVM. The problem is that GVs may 1676dff0c46cSDimitry Andric // contain entries in the use_list that do not exist in the Module and are 1677dff0c46cSDimitry Andric // not stored in the .bc file. 1678dff0c46cSDimitry Andric for (Module::const_global_iterator I = M->global_begin(), E = M->global_end(); 1679dff0c46cSDimitry Andric I != E; ++I) 1680dff0c46cSDimitry Andric I->removeDeadConstantUsers(); 1681dff0c46cSDimitry Andric 1682dff0c46cSDimitry Andric // Write the global variables. 1683dff0c46cSDimitry Andric for (Module::const_global_iterator GI = M->global_begin(), 1684dff0c46cSDimitry Andric GE = M->global_end(); GI != GE; ++GI) { 1685dff0c46cSDimitry Andric WriteUseList(GI, VE, Stream); 1686dff0c46cSDimitry Andric 1687dff0c46cSDimitry Andric // Write the global variable initializers. 1688dff0c46cSDimitry Andric if (GI->hasInitializer()) 1689dff0c46cSDimitry Andric WriteUseList(GI->getInitializer(), VE, Stream); 1690dff0c46cSDimitry Andric } 1691dff0c46cSDimitry Andric 1692dff0c46cSDimitry Andric // Write the functions. 1693dff0c46cSDimitry Andric for (Module::const_iterator FI = M->begin(), FE = M->end(); FI != FE; ++FI) { 1694dff0c46cSDimitry Andric WriteUseList(FI, VE, Stream); 1695dff0c46cSDimitry Andric if (!FI->isDeclaration()) 1696dff0c46cSDimitry Andric WriteFunctionUseList(FI, VE, Stream); 1697dff0c46cSDimitry Andric } 1698dff0c46cSDimitry Andric 1699dff0c46cSDimitry Andric // Write the aliases. 1700dff0c46cSDimitry Andric for (Module::const_alias_iterator AI = M->alias_begin(), AE = M->alias_end(); 1701dff0c46cSDimitry Andric AI != AE; ++AI) { 1702dff0c46cSDimitry Andric WriteUseList(AI, VE, Stream); 1703dff0c46cSDimitry Andric WriteUseList(AI->getAliasee(), VE, Stream); 1704dff0c46cSDimitry Andric } 1705dff0c46cSDimitry Andric 1706dff0c46cSDimitry Andric Stream.ExitBlock(); 1707dff0c46cSDimitry Andric } 1708f22ef01cSRoman Divacky 1709f22ef01cSRoman Divacky /// WriteModule - Emit the specified module to the bitstream. 1710f22ef01cSRoman Divacky static void WriteModule(const Module *M, BitstreamWriter &Stream) { 1711f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::MODULE_BLOCK_ID, 3); 1712f22ef01cSRoman Divacky 1713f22ef01cSRoman Divacky // Emit the version number if it is non-zero. 1714f22ef01cSRoman Divacky if (CurVersion) { 1715f22ef01cSRoman Divacky SmallVector<unsigned, 1> Vals; 1716f22ef01cSRoman Divacky Vals.push_back(CurVersion); 1717f22ef01cSRoman Divacky Stream.EmitRecord(bitc::MODULE_CODE_VERSION, Vals); 1718f22ef01cSRoman Divacky } 1719f22ef01cSRoman Divacky 1720f22ef01cSRoman Divacky // Analyze the module, enumerating globals, functions, etc. 1721f22ef01cSRoman Divacky ValueEnumerator VE(M); 1722f22ef01cSRoman Divacky 1723f22ef01cSRoman Divacky // Emit blockinfo, which defines the standard abbreviations etc. 1724f22ef01cSRoman Divacky WriteBlockInfo(VE, Stream); 1725f22ef01cSRoman Divacky 1726f22ef01cSRoman Divacky // Emit information about parameter attributes. 1727f22ef01cSRoman Divacky WriteAttributeTable(VE, Stream); 1728f22ef01cSRoman Divacky 1729f22ef01cSRoman Divacky // Emit information describing all of the types in the module. 1730f22ef01cSRoman Divacky WriteTypeTable(VE, Stream); 1731f22ef01cSRoman Divacky 1732f22ef01cSRoman Divacky // Emit top-level description of module, including target triple, inline asm, 1733f22ef01cSRoman Divacky // descriptors for global variables, and function prototype info. 1734f22ef01cSRoman Divacky WriteModuleInfo(M, VE, Stream); 1735f22ef01cSRoman Divacky 1736f22ef01cSRoman Divacky // Emit constants. 1737f22ef01cSRoman Divacky WriteModuleConstants(VE, Stream); 1738f22ef01cSRoman Divacky 1739f22ef01cSRoman Divacky // Emit metadata. 1740e580952dSDimitry Andric WriteModuleMetadata(M, VE, Stream); 1741f22ef01cSRoman Divacky 1742f22ef01cSRoman Divacky // Emit metadata. 1743f22ef01cSRoman Divacky WriteModuleMetadataStore(M, Stream); 1744f22ef01cSRoman Divacky 1745f22ef01cSRoman Divacky // Emit names for globals/functions etc. 1746f22ef01cSRoman Divacky WriteValueSymbolTable(M->getValueSymbolTable(), VE, Stream); 1747f22ef01cSRoman Divacky 1748dff0c46cSDimitry Andric // Emit use-lists. 1749dff0c46cSDimitry Andric if (EnablePreserveUseListOrdering) 1750dff0c46cSDimitry Andric WriteModuleUseLists(M, VE, Stream); 1751dff0c46cSDimitry Andric 1752dff0c46cSDimitry Andric // Emit function bodies. 1753dff0c46cSDimitry Andric for (Module::const_iterator F = M->begin(), E = M->end(); F != E; ++F) 1754dff0c46cSDimitry Andric if (!F->isDeclaration()) 1755dff0c46cSDimitry Andric WriteFunction(*F, VE, Stream); 1756dff0c46cSDimitry Andric 1757f22ef01cSRoman Divacky Stream.ExitBlock(); 1758f22ef01cSRoman Divacky } 1759f22ef01cSRoman Divacky 1760f22ef01cSRoman Divacky /// EmitDarwinBCHeader - If generating a bc file on darwin, we have to emit a 1761f22ef01cSRoman Divacky /// header and trailer to make it compatible with the system archiver. To do 1762f22ef01cSRoman Divacky /// this we emit the following header, and then emit a trailer that pads the 1763f22ef01cSRoman Divacky /// file out to be a multiple of 16 bytes. 1764f22ef01cSRoman Divacky /// 1765f22ef01cSRoman Divacky /// struct bc_header { 1766f22ef01cSRoman Divacky /// uint32_t Magic; // 0x0B17C0DE 1767f22ef01cSRoman Divacky /// uint32_t Version; // Version, currently always 0. 1768f22ef01cSRoman Divacky /// uint32_t BitcodeOffset; // Offset to traditional bitcode file. 1769f22ef01cSRoman Divacky /// uint32_t BitcodeSize; // Size of traditional bitcode file. 1770f22ef01cSRoman Divacky /// uint32_t CPUType; // CPU specifier. 1771f22ef01cSRoman Divacky /// ... potentially more later ... 1772f22ef01cSRoman Divacky /// }; 1773f22ef01cSRoman Divacky enum { 1774f22ef01cSRoman Divacky DarwinBCSizeFieldOffset = 3*4, // Offset to bitcode_size. 1775f22ef01cSRoman Divacky DarwinBCHeaderSize = 5*4 1776f22ef01cSRoman Divacky }; 1777f22ef01cSRoman Divacky 1778dff0c46cSDimitry Andric static void WriteInt32ToBuffer(uint32_t Value, SmallVectorImpl<char> &Buffer, 1779dff0c46cSDimitry Andric uint32_t &Position) { 1780dff0c46cSDimitry Andric Buffer[Position + 0] = (unsigned char) (Value >> 0); 1781dff0c46cSDimitry Andric Buffer[Position + 1] = (unsigned char) (Value >> 8); 1782dff0c46cSDimitry Andric Buffer[Position + 2] = (unsigned char) (Value >> 16); 1783dff0c46cSDimitry Andric Buffer[Position + 3] = (unsigned char) (Value >> 24); 1784dff0c46cSDimitry Andric Position += 4; 1785dff0c46cSDimitry Andric } 1786dff0c46cSDimitry Andric 1787dff0c46cSDimitry Andric static void EmitDarwinBCHeaderAndTrailer(SmallVectorImpl<char> &Buffer, 1788dff0c46cSDimitry Andric const Triple &TT) { 1789f22ef01cSRoman Divacky unsigned CPUType = ~0U; 1790f22ef01cSRoman Divacky 1791f22ef01cSRoman Divacky // Match x86_64-*, i[3-9]86-*, powerpc-*, powerpc64-*, arm-*, thumb-*, 1792f22ef01cSRoman Divacky // armv[0-9]-*, thumbv[0-9]-*, armv5te-*, or armv6t2-*. The CPUType is a magic 1793f22ef01cSRoman Divacky // number from /usr/include/mach/machine.h. It is ok to reproduce the 1794f22ef01cSRoman Divacky // specific constants here because they are implicitly part of the Darwin ABI. 1795f22ef01cSRoman Divacky enum { 1796f22ef01cSRoman Divacky DARWIN_CPU_ARCH_ABI64 = 0x01000000, 1797f22ef01cSRoman Divacky DARWIN_CPU_TYPE_X86 = 7, 1798f22ef01cSRoman Divacky DARWIN_CPU_TYPE_ARM = 12, 1799f22ef01cSRoman Divacky DARWIN_CPU_TYPE_POWERPC = 18 1800f22ef01cSRoman Divacky }; 1801f22ef01cSRoman Divacky 180217a519f9SDimitry Andric Triple::ArchType Arch = TT.getArch(); 180317a519f9SDimitry Andric if (Arch == Triple::x86_64) 1804f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_X86 | DARWIN_CPU_ARCH_ABI64; 180517a519f9SDimitry Andric else if (Arch == Triple::x86) 1806f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_X86; 180717a519f9SDimitry Andric else if (Arch == Triple::ppc) 1808f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_POWERPC; 180917a519f9SDimitry Andric else if (Arch == Triple::ppc64) 1810f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_POWERPC | DARWIN_CPU_ARCH_ABI64; 181117a519f9SDimitry Andric else if (Arch == Triple::arm || Arch == Triple::thumb) 1812f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_ARM; 1813f22ef01cSRoman Divacky 1814f22ef01cSRoman Divacky // Traditional Bitcode starts after header. 1815dff0c46cSDimitry Andric assert(Buffer.size() >= DarwinBCHeaderSize && 1816dff0c46cSDimitry Andric "Expected header size to be reserved"); 1817f22ef01cSRoman Divacky unsigned BCOffset = DarwinBCHeaderSize; 1818dff0c46cSDimitry Andric unsigned BCSize = Buffer.size()-DarwinBCHeaderSize; 1819f22ef01cSRoman Divacky 1820dff0c46cSDimitry Andric // Write the magic and version. 1821dff0c46cSDimitry Andric unsigned Position = 0; 1822dff0c46cSDimitry Andric WriteInt32ToBuffer(0x0B17C0DE , Buffer, Position); 1823dff0c46cSDimitry Andric WriteInt32ToBuffer(0 , Buffer, Position); // Version. 1824dff0c46cSDimitry Andric WriteInt32ToBuffer(BCOffset , Buffer, Position); 1825dff0c46cSDimitry Andric WriteInt32ToBuffer(BCSize , Buffer, Position); 1826dff0c46cSDimitry Andric WriteInt32ToBuffer(CPUType , Buffer, Position); 1827f22ef01cSRoman Divacky 1828f22ef01cSRoman Divacky // If the file is not a multiple of 16 bytes, insert dummy padding. 1829dff0c46cSDimitry Andric while (Buffer.size() & 15) 1830dff0c46cSDimitry Andric Buffer.push_back(0); 1831f22ef01cSRoman Divacky } 1832f22ef01cSRoman Divacky 1833f22ef01cSRoman Divacky /// WriteBitcodeToFile - Write the specified module to the specified output 1834f22ef01cSRoman Divacky /// stream. 1835f22ef01cSRoman Divacky void llvm::WriteBitcodeToFile(const Module *M, raw_ostream &Out) { 1836dff0c46cSDimitry Andric SmallVector<char, 1024> Buffer; 1837f22ef01cSRoman Divacky Buffer.reserve(256*1024); 1838f22ef01cSRoman Divacky 1839dff0c46cSDimitry Andric // If this is darwin or another generic macho target, reserve space for the 1840dff0c46cSDimitry Andric // header. 184117a519f9SDimitry Andric Triple TT(M->getTargetTriple()); 184217a519f9SDimitry Andric if (TT.isOSDarwin()) 1843dff0c46cSDimitry Andric Buffer.insert(Buffer.begin(), DarwinBCHeaderSize, 0); 1844dff0c46cSDimitry Andric 1845dff0c46cSDimitry Andric // Emit the module into the buffer. 1846dff0c46cSDimitry Andric { 1847dff0c46cSDimitry Andric BitstreamWriter Stream(Buffer); 1848f22ef01cSRoman Divacky 1849f22ef01cSRoman Divacky // Emit the file header. 1850f22ef01cSRoman Divacky Stream.Emit((unsigned)'B', 8); 1851f22ef01cSRoman Divacky Stream.Emit((unsigned)'C', 8); 1852f22ef01cSRoman Divacky Stream.Emit(0x0, 4); 1853f22ef01cSRoman Divacky Stream.Emit(0xC, 4); 1854f22ef01cSRoman Divacky Stream.Emit(0xE, 4); 1855f22ef01cSRoman Divacky Stream.Emit(0xD, 4); 1856f22ef01cSRoman Divacky 1857f22ef01cSRoman Divacky // Emit the module. 1858f22ef01cSRoman Divacky WriteModule(M, Stream); 1859dff0c46cSDimitry Andric } 1860f22ef01cSRoman Divacky 186117a519f9SDimitry Andric if (TT.isOSDarwin()) 1862dff0c46cSDimitry Andric EmitDarwinBCHeaderAndTrailer(Buffer, TT); 1863dff0c46cSDimitry Andric 1864dff0c46cSDimitry Andric // Write the generated bitstream to "Out". 1865dff0c46cSDimitry Andric Out.write((char*)&Buffer.front(), Buffer.size()); 1866f22ef01cSRoman Divacky } 1867