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, 657ae0e2c9SDimitry Andric FUNCTION_INST_UNREACHABLE_ABBREV, 667ae0e2c9SDimitry Andric 677ae0e2c9SDimitry Andric // SwitchInst Magic 687ae0e2c9SDimitry Andric SWITCH_INST_MAGIC = 0x4B5 // May 2012 => 1205 => Hex 69f22ef01cSRoman Divacky }; 70f22ef01cSRoman Divacky 71f22ef01cSRoman Divacky static unsigned GetEncodedCastOpcode(unsigned Opcode) { 72f22ef01cSRoman Divacky switch (Opcode) { 73f22ef01cSRoman Divacky default: llvm_unreachable("Unknown cast instruction!"); 74f22ef01cSRoman Divacky case Instruction::Trunc : return bitc::CAST_TRUNC; 75f22ef01cSRoman Divacky case Instruction::ZExt : return bitc::CAST_ZEXT; 76f22ef01cSRoman Divacky case Instruction::SExt : return bitc::CAST_SEXT; 77f22ef01cSRoman Divacky case Instruction::FPToUI : return bitc::CAST_FPTOUI; 78f22ef01cSRoman Divacky case Instruction::FPToSI : return bitc::CAST_FPTOSI; 79f22ef01cSRoman Divacky case Instruction::UIToFP : return bitc::CAST_UITOFP; 80f22ef01cSRoman Divacky case Instruction::SIToFP : return bitc::CAST_SITOFP; 81f22ef01cSRoman Divacky case Instruction::FPTrunc : return bitc::CAST_FPTRUNC; 82f22ef01cSRoman Divacky case Instruction::FPExt : return bitc::CAST_FPEXT; 83f22ef01cSRoman Divacky case Instruction::PtrToInt: return bitc::CAST_PTRTOINT; 84f22ef01cSRoman Divacky case Instruction::IntToPtr: return bitc::CAST_INTTOPTR; 85f22ef01cSRoman Divacky case Instruction::BitCast : return bitc::CAST_BITCAST; 86f22ef01cSRoman Divacky } 87f22ef01cSRoman Divacky } 88f22ef01cSRoman Divacky 89f22ef01cSRoman Divacky static unsigned GetEncodedBinaryOpcode(unsigned Opcode) { 90f22ef01cSRoman Divacky switch (Opcode) { 91f22ef01cSRoman Divacky default: llvm_unreachable("Unknown binary instruction!"); 92f22ef01cSRoman Divacky case Instruction::Add: 93f22ef01cSRoman Divacky case Instruction::FAdd: return bitc::BINOP_ADD; 94f22ef01cSRoman Divacky case Instruction::Sub: 95f22ef01cSRoman Divacky case Instruction::FSub: return bitc::BINOP_SUB; 96f22ef01cSRoman Divacky case Instruction::Mul: 97f22ef01cSRoman Divacky case Instruction::FMul: return bitc::BINOP_MUL; 98f22ef01cSRoman Divacky case Instruction::UDiv: return bitc::BINOP_UDIV; 99f22ef01cSRoman Divacky case Instruction::FDiv: 100f22ef01cSRoman Divacky case Instruction::SDiv: return bitc::BINOP_SDIV; 101f22ef01cSRoman Divacky case Instruction::URem: return bitc::BINOP_UREM; 102f22ef01cSRoman Divacky case Instruction::FRem: 103f22ef01cSRoman Divacky case Instruction::SRem: return bitc::BINOP_SREM; 104f22ef01cSRoman Divacky case Instruction::Shl: return bitc::BINOP_SHL; 105f22ef01cSRoman Divacky case Instruction::LShr: return bitc::BINOP_LSHR; 106f22ef01cSRoman Divacky case Instruction::AShr: return bitc::BINOP_ASHR; 107f22ef01cSRoman Divacky case Instruction::And: return bitc::BINOP_AND; 108f22ef01cSRoman Divacky case Instruction::Or: return bitc::BINOP_OR; 109f22ef01cSRoman Divacky case Instruction::Xor: return bitc::BINOP_XOR; 110f22ef01cSRoman Divacky } 111f22ef01cSRoman Divacky } 112f22ef01cSRoman Divacky 1136122f3e6SDimitry Andric static unsigned GetEncodedRMWOperation(AtomicRMWInst::BinOp Op) { 1146122f3e6SDimitry Andric switch (Op) { 1156122f3e6SDimitry Andric default: llvm_unreachable("Unknown RMW operation!"); 1166122f3e6SDimitry Andric case AtomicRMWInst::Xchg: return bitc::RMW_XCHG; 1176122f3e6SDimitry Andric case AtomicRMWInst::Add: return bitc::RMW_ADD; 1186122f3e6SDimitry Andric case AtomicRMWInst::Sub: return bitc::RMW_SUB; 1196122f3e6SDimitry Andric case AtomicRMWInst::And: return bitc::RMW_AND; 1206122f3e6SDimitry Andric case AtomicRMWInst::Nand: return bitc::RMW_NAND; 1216122f3e6SDimitry Andric case AtomicRMWInst::Or: return bitc::RMW_OR; 1226122f3e6SDimitry Andric case AtomicRMWInst::Xor: return bitc::RMW_XOR; 1236122f3e6SDimitry Andric case AtomicRMWInst::Max: return bitc::RMW_MAX; 1246122f3e6SDimitry Andric case AtomicRMWInst::Min: return bitc::RMW_MIN; 1256122f3e6SDimitry Andric case AtomicRMWInst::UMax: return bitc::RMW_UMAX; 1266122f3e6SDimitry Andric case AtomicRMWInst::UMin: return bitc::RMW_UMIN; 1276122f3e6SDimitry Andric } 1286122f3e6SDimitry Andric } 1296122f3e6SDimitry Andric 1306122f3e6SDimitry Andric static unsigned GetEncodedOrdering(AtomicOrdering Ordering) { 1316122f3e6SDimitry Andric switch (Ordering) { 1326122f3e6SDimitry Andric case NotAtomic: return bitc::ORDERING_NOTATOMIC; 1336122f3e6SDimitry Andric case Unordered: return bitc::ORDERING_UNORDERED; 1346122f3e6SDimitry Andric case Monotonic: return bitc::ORDERING_MONOTONIC; 1356122f3e6SDimitry Andric case Acquire: return bitc::ORDERING_ACQUIRE; 1366122f3e6SDimitry Andric case Release: return bitc::ORDERING_RELEASE; 1376122f3e6SDimitry Andric case AcquireRelease: return bitc::ORDERING_ACQREL; 1386122f3e6SDimitry Andric case SequentiallyConsistent: return bitc::ORDERING_SEQCST; 1396122f3e6SDimitry Andric } 140dff0c46cSDimitry Andric llvm_unreachable("Invalid ordering"); 1416122f3e6SDimitry Andric } 1426122f3e6SDimitry Andric 1436122f3e6SDimitry Andric static unsigned GetEncodedSynchScope(SynchronizationScope SynchScope) { 1446122f3e6SDimitry Andric switch (SynchScope) { 1456122f3e6SDimitry Andric case SingleThread: return bitc::SYNCHSCOPE_SINGLETHREAD; 1466122f3e6SDimitry Andric case CrossThread: return bitc::SYNCHSCOPE_CROSSTHREAD; 1476122f3e6SDimitry Andric } 148dff0c46cSDimitry Andric llvm_unreachable("Invalid synch scope"); 1496122f3e6SDimitry Andric } 1506122f3e6SDimitry Andric 15117a519f9SDimitry Andric static void WriteStringRecord(unsigned Code, StringRef Str, 152f22ef01cSRoman Divacky unsigned AbbrevToUse, BitstreamWriter &Stream) { 153f22ef01cSRoman Divacky SmallVector<unsigned, 64> Vals; 154f22ef01cSRoman Divacky 155f22ef01cSRoman Divacky // Code: [strchar x N] 15617a519f9SDimitry Andric for (unsigned i = 0, e = Str.size(); i != e; ++i) { 15717a519f9SDimitry Andric if (AbbrevToUse && !BitCodeAbbrevOp::isChar6(Str[i])) 15817a519f9SDimitry Andric AbbrevToUse = 0; 159f22ef01cSRoman Divacky Vals.push_back(Str[i]); 16017a519f9SDimitry Andric } 161f22ef01cSRoman Divacky 162f22ef01cSRoman Divacky // Emit the finished record. 163f22ef01cSRoman Divacky Stream.EmitRecord(Code, Vals, AbbrevToUse); 164f22ef01cSRoman Divacky } 165f22ef01cSRoman Divacky 166f22ef01cSRoman Divacky // Emit information about parameter attributes. 167f22ef01cSRoman Divacky static void WriteAttributeTable(const ValueEnumerator &VE, 168f22ef01cSRoman Divacky BitstreamWriter &Stream) { 169f22ef01cSRoman Divacky const std::vector<AttrListPtr> &Attrs = VE.getAttributes(); 170f22ef01cSRoman Divacky if (Attrs.empty()) return; 171f22ef01cSRoman Divacky 172f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::PARAMATTR_BLOCK_ID, 3); 173f22ef01cSRoman Divacky 174f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 175f22ef01cSRoman Divacky for (unsigned i = 0, e = Attrs.size(); i != e; ++i) { 176f22ef01cSRoman Divacky const AttrListPtr &A = Attrs[i]; 177f22ef01cSRoman Divacky for (unsigned i = 0, e = A.getNumSlots(); i != e; ++i) { 178f22ef01cSRoman Divacky const AttributeWithIndex &PAWI = A.getSlot(i); 179f22ef01cSRoman Divacky Record.push_back(PAWI.Index); 1807ae0e2c9SDimitry Andric Record.push_back(Attribute::encodeLLVMAttributesForBitcode(PAWI.Attrs)); 181f22ef01cSRoman Divacky } 182f22ef01cSRoman Divacky 183f22ef01cSRoman Divacky Stream.EmitRecord(bitc::PARAMATTR_CODE_ENTRY, Record); 184f22ef01cSRoman Divacky Record.clear(); 185f22ef01cSRoman Divacky } 186f22ef01cSRoman Divacky 187f22ef01cSRoman Divacky Stream.ExitBlock(); 188f22ef01cSRoman Divacky } 189f22ef01cSRoman Divacky 190f22ef01cSRoman Divacky /// WriteTypeTable - Write out the type table for a module. 191f22ef01cSRoman Divacky static void WriteTypeTable(const ValueEnumerator &VE, BitstreamWriter &Stream) { 192f22ef01cSRoman Divacky const ValueEnumerator::TypeList &TypeList = VE.getTypes(); 193f22ef01cSRoman Divacky 19417a519f9SDimitry Andric Stream.EnterSubblock(bitc::TYPE_BLOCK_ID_NEW, 4 /*count from # abbrevs */); 195f22ef01cSRoman Divacky SmallVector<uint64_t, 64> TypeVals; 196f22ef01cSRoman Divacky 197dff0c46cSDimitry Andric uint64_t NumBits = Log2_32_Ceil(VE.getTypes().size()+1); 198dff0c46cSDimitry Andric 199f22ef01cSRoman Divacky // Abbrev for TYPE_CODE_POINTER. 200f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 201f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_POINTER)); 202dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 203f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(0)); // Addrspace = 0 204f22ef01cSRoman Divacky unsigned PtrAbbrev = Stream.EmitAbbrev(Abbv); 205f22ef01cSRoman Divacky 206f22ef01cSRoman Divacky // Abbrev for TYPE_CODE_FUNCTION. 207f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 208f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_FUNCTION)); 209f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // isvararg 210f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 211dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 212dff0c46cSDimitry Andric 213f22ef01cSRoman Divacky unsigned FunctionAbbrev = Stream.EmitAbbrev(Abbv); 214f22ef01cSRoman Divacky 21517a519f9SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_ANON. 216f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 21717a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_ANON)); 218f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // ispacked 219f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 220dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 221dff0c46cSDimitry Andric 22217a519f9SDimitry Andric unsigned StructAnonAbbrev = Stream.EmitAbbrev(Abbv); 22317a519f9SDimitry Andric 22417a519f9SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_NAME. 22517a519f9SDimitry Andric Abbv = new BitCodeAbbrev(); 22617a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_NAME)); 22717a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 22817a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 22917a519f9SDimitry Andric unsigned StructNameAbbrev = Stream.EmitAbbrev(Abbv); 23017a519f9SDimitry Andric 23117a519f9SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_NAMED. 23217a519f9SDimitry Andric Abbv = new BitCodeAbbrev(); 23317a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_NAMED)); 23417a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // ispacked 23517a519f9SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 236dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 237dff0c46cSDimitry Andric 23817a519f9SDimitry Andric unsigned StructNamedAbbrev = Stream.EmitAbbrev(Abbv); 23917a519f9SDimitry Andric 240f22ef01cSRoman Divacky // Abbrev for TYPE_CODE_ARRAY. 241f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 242f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_ARRAY)); 243f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // size 244dff0c46cSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 245dff0c46cSDimitry Andric 246f22ef01cSRoman Divacky unsigned ArrayAbbrev = Stream.EmitAbbrev(Abbv); 247f22ef01cSRoman Divacky 248f22ef01cSRoman Divacky // Emit an entry count so the reader can reserve space. 249f22ef01cSRoman Divacky TypeVals.push_back(TypeList.size()); 250f22ef01cSRoman Divacky Stream.EmitRecord(bitc::TYPE_CODE_NUMENTRY, TypeVals); 251f22ef01cSRoman Divacky TypeVals.clear(); 252f22ef01cSRoman Divacky 253f22ef01cSRoman Divacky // Loop over all of the types, emitting each in turn. 254f22ef01cSRoman Divacky for (unsigned i = 0, e = TypeList.size(); i != e; ++i) { 2556122f3e6SDimitry Andric Type *T = TypeList[i]; 256f22ef01cSRoman Divacky int AbbrevToUse = 0; 257f22ef01cSRoman Divacky unsigned Code = 0; 258f22ef01cSRoman Divacky 259f22ef01cSRoman Divacky switch (T->getTypeID()) { 260f22ef01cSRoman Divacky default: llvm_unreachable("Unknown type!"); 261f22ef01cSRoman Divacky case Type::VoidTyID: Code = bitc::TYPE_CODE_VOID; break; 262dff0c46cSDimitry Andric case Type::HalfTyID: Code = bitc::TYPE_CODE_HALF; break; 263f22ef01cSRoman Divacky case Type::FloatTyID: Code = bitc::TYPE_CODE_FLOAT; break; 264f22ef01cSRoman Divacky case Type::DoubleTyID: Code = bitc::TYPE_CODE_DOUBLE; break; 265f22ef01cSRoman Divacky case Type::X86_FP80TyID: Code = bitc::TYPE_CODE_X86_FP80; break; 266f22ef01cSRoman Divacky case Type::FP128TyID: Code = bitc::TYPE_CODE_FP128; break; 267f22ef01cSRoman Divacky case Type::PPC_FP128TyID: Code = bitc::TYPE_CODE_PPC_FP128; break; 268f22ef01cSRoman Divacky case Type::LabelTyID: Code = bitc::TYPE_CODE_LABEL; break; 269f22ef01cSRoman Divacky case Type::MetadataTyID: Code = bitc::TYPE_CODE_METADATA; break; 2702754fe60SDimitry Andric case Type::X86_MMXTyID: Code = bitc::TYPE_CODE_X86_MMX; break; 271f22ef01cSRoman Divacky case Type::IntegerTyID: 272f22ef01cSRoman Divacky // INTEGER: [width] 273f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_INTEGER; 274f22ef01cSRoman Divacky TypeVals.push_back(cast<IntegerType>(T)->getBitWidth()); 275f22ef01cSRoman Divacky break; 276f22ef01cSRoman Divacky case Type::PointerTyID: { 2776122f3e6SDimitry Andric PointerType *PTy = cast<PointerType>(T); 278f22ef01cSRoman Divacky // POINTER: [pointee type, address space] 279f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_POINTER; 280f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(PTy->getElementType())); 281f22ef01cSRoman Divacky unsigned AddressSpace = PTy->getAddressSpace(); 282f22ef01cSRoman Divacky TypeVals.push_back(AddressSpace); 283f22ef01cSRoman Divacky if (AddressSpace == 0) AbbrevToUse = PtrAbbrev; 284f22ef01cSRoman Divacky break; 285f22ef01cSRoman Divacky } 286f22ef01cSRoman Divacky case Type::FunctionTyID: { 2876122f3e6SDimitry Andric FunctionType *FT = cast<FunctionType>(T); 288dff0c46cSDimitry Andric // FUNCTION: [isvararg, retty, paramty x N] 289f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_FUNCTION; 290f22ef01cSRoman Divacky TypeVals.push_back(FT->isVarArg()); 291f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(FT->getReturnType())); 292f22ef01cSRoman Divacky for (unsigned i = 0, e = FT->getNumParams(); i != e; ++i) 293f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(FT->getParamType(i))); 294f22ef01cSRoman Divacky AbbrevToUse = FunctionAbbrev; 295f22ef01cSRoman Divacky break; 296f22ef01cSRoman Divacky } 297f22ef01cSRoman Divacky case Type::StructTyID: { 2986122f3e6SDimitry Andric StructType *ST = cast<StructType>(T); 299f22ef01cSRoman Divacky // STRUCT: [ispacked, eltty x N] 300f22ef01cSRoman Divacky TypeVals.push_back(ST->isPacked()); 301f22ef01cSRoman Divacky // Output all of the element types. 302f22ef01cSRoman Divacky for (StructType::element_iterator I = ST->element_begin(), 303f22ef01cSRoman Divacky E = ST->element_end(); I != E; ++I) 304f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(*I)); 30517a519f9SDimitry Andric 3066122f3e6SDimitry Andric if (ST->isLiteral()) { 30717a519f9SDimitry Andric Code = bitc::TYPE_CODE_STRUCT_ANON; 30817a519f9SDimitry Andric AbbrevToUse = StructAnonAbbrev; 30917a519f9SDimitry Andric } else { 31017a519f9SDimitry Andric if (ST->isOpaque()) { 31117a519f9SDimitry Andric Code = bitc::TYPE_CODE_OPAQUE; 31217a519f9SDimitry Andric } else { 31317a519f9SDimitry Andric Code = bitc::TYPE_CODE_STRUCT_NAMED; 31417a519f9SDimitry Andric AbbrevToUse = StructNamedAbbrev; 31517a519f9SDimitry Andric } 31617a519f9SDimitry Andric 31717a519f9SDimitry Andric // Emit the name if it is present. 31817a519f9SDimitry Andric if (!ST->getName().empty()) 31917a519f9SDimitry Andric WriteStringRecord(bitc::TYPE_CODE_STRUCT_NAME, ST->getName(), 32017a519f9SDimitry Andric StructNameAbbrev, Stream); 32117a519f9SDimitry Andric } 322f22ef01cSRoman Divacky break; 323f22ef01cSRoman Divacky } 324f22ef01cSRoman Divacky case Type::ArrayTyID: { 3256122f3e6SDimitry Andric ArrayType *AT = cast<ArrayType>(T); 326f22ef01cSRoman Divacky // ARRAY: [numelts, eltty] 327f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_ARRAY; 328f22ef01cSRoman Divacky TypeVals.push_back(AT->getNumElements()); 329f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(AT->getElementType())); 330f22ef01cSRoman Divacky AbbrevToUse = ArrayAbbrev; 331f22ef01cSRoman Divacky break; 332f22ef01cSRoman Divacky } 333f22ef01cSRoman Divacky case Type::VectorTyID: { 3346122f3e6SDimitry Andric VectorType *VT = cast<VectorType>(T); 335f22ef01cSRoman Divacky // VECTOR [numelts, eltty] 336f22ef01cSRoman Divacky Code = bitc::TYPE_CODE_VECTOR; 337f22ef01cSRoman Divacky TypeVals.push_back(VT->getNumElements()); 338f22ef01cSRoman Divacky TypeVals.push_back(VE.getTypeID(VT->getElementType())); 339f22ef01cSRoman Divacky break; 340f22ef01cSRoman Divacky } 341f22ef01cSRoman Divacky } 342f22ef01cSRoman Divacky 343f22ef01cSRoman Divacky // Emit the finished record. 344f22ef01cSRoman Divacky Stream.EmitRecord(Code, TypeVals, AbbrevToUse); 345f22ef01cSRoman Divacky TypeVals.clear(); 346f22ef01cSRoman Divacky } 347f22ef01cSRoman Divacky 348f22ef01cSRoman Divacky Stream.ExitBlock(); 349f22ef01cSRoman Divacky } 350f22ef01cSRoman Divacky 351f22ef01cSRoman Divacky static unsigned getEncodedLinkage(const GlobalValue *GV) { 352f22ef01cSRoman Divacky switch (GV->getLinkage()) { 353f22ef01cSRoman Divacky case GlobalValue::ExternalLinkage: return 0; 354f22ef01cSRoman Divacky case GlobalValue::WeakAnyLinkage: return 1; 355f22ef01cSRoman Divacky case GlobalValue::AppendingLinkage: return 2; 356f22ef01cSRoman Divacky case GlobalValue::InternalLinkage: return 3; 357f22ef01cSRoman Divacky case GlobalValue::LinkOnceAnyLinkage: return 4; 358f22ef01cSRoman Divacky case GlobalValue::DLLImportLinkage: return 5; 359f22ef01cSRoman Divacky case GlobalValue::DLLExportLinkage: return 6; 360f22ef01cSRoman Divacky case GlobalValue::ExternalWeakLinkage: return 7; 361f22ef01cSRoman Divacky case GlobalValue::CommonLinkage: return 8; 362f22ef01cSRoman Divacky case GlobalValue::PrivateLinkage: return 9; 363f22ef01cSRoman Divacky case GlobalValue::WeakODRLinkage: return 10; 364f22ef01cSRoman Divacky case GlobalValue::LinkOnceODRLinkage: return 11; 365f22ef01cSRoman Divacky case GlobalValue::AvailableExternallyLinkage: return 12; 366f22ef01cSRoman Divacky case GlobalValue::LinkerPrivateLinkage: return 13; 367ffd1746dSEd Schouten case GlobalValue::LinkerPrivateWeakLinkage: return 14; 368e580952dSDimitry Andric case GlobalValue::LinkerPrivateWeakDefAutoLinkage: return 15; 369f22ef01cSRoman Divacky } 370dff0c46cSDimitry Andric llvm_unreachable("Invalid linkage"); 371f22ef01cSRoman Divacky } 372f22ef01cSRoman Divacky 373f22ef01cSRoman Divacky static unsigned getEncodedVisibility(const GlobalValue *GV) { 374f22ef01cSRoman Divacky switch (GV->getVisibility()) { 375f22ef01cSRoman Divacky case GlobalValue::DefaultVisibility: return 0; 376f22ef01cSRoman Divacky case GlobalValue::HiddenVisibility: return 1; 377f22ef01cSRoman Divacky case GlobalValue::ProtectedVisibility: return 2; 378f22ef01cSRoman Divacky } 379dff0c46cSDimitry Andric llvm_unreachable("Invalid visibility"); 380f22ef01cSRoman Divacky } 381f22ef01cSRoman Divacky 3827ae0e2c9SDimitry Andric static unsigned getEncodedThreadLocalMode(const GlobalVariable *GV) { 3837ae0e2c9SDimitry Andric switch (GV->getThreadLocalMode()) { 3847ae0e2c9SDimitry Andric case GlobalVariable::NotThreadLocal: return 0; 3857ae0e2c9SDimitry Andric case GlobalVariable::GeneralDynamicTLSModel: return 1; 3867ae0e2c9SDimitry Andric case GlobalVariable::LocalDynamicTLSModel: return 2; 3877ae0e2c9SDimitry Andric case GlobalVariable::InitialExecTLSModel: return 3; 3887ae0e2c9SDimitry Andric case GlobalVariable::LocalExecTLSModel: return 4; 3897ae0e2c9SDimitry Andric } 3907ae0e2c9SDimitry Andric llvm_unreachable("Invalid TLS model"); 3917ae0e2c9SDimitry Andric } 3927ae0e2c9SDimitry Andric 393f22ef01cSRoman Divacky // Emit top-level description of module, including target triple, inline asm, 394f22ef01cSRoman Divacky // descriptors for global variables, and function prototype info. 395f22ef01cSRoman Divacky static void WriteModuleInfo(const Module *M, const ValueEnumerator &VE, 396f22ef01cSRoman Divacky BitstreamWriter &Stream) { 397f22ef01cSRoman Divacky // Emit the list of dependent libraries for the Module. 398f22ef01cSRoman Divacky for (Module::lib_iterator I = M->lib_begin(), E = M->lib_end(); I != E; ++I) 399f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_DEPLIB, *I, 0/*TODO*/, Stream); 400f22ef01cSRoman Divacky 401f22ef01cSRoman Divacky // Emit various pieces of data attached to a module. 402f22ef01cSRoman Divacky if (!M->getTargetTriple().empty()) 403f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_TRIPLE, M->getTargetTriple(), 404f22ef01cSRoman Divacky 0/*TODO*/, Stream); 405f22ef01cSRoman Divacky if (!M->getDataLayout().empty()) 406f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_DATALAYOUT, M->getDataLayout(), 407f22ef01cSRoman Divacky 0/*TODO*/, Stream); 408f22ef01cSRoman Divacky if (!M->getModuleInlineAsm().empty()) 409f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_ASM, M->getModuleInlineAsm(), 410f22ef01cSRoman Divacky 0/*TODO*/, Stream); 411f22ef01cSRoman Divacky 412f22ef01cSRoman Divacky // Emit information about sections and GC, computing how many there are. Also 413f22ef01cSRoman Divacky // compute the maximum alignment value. 414f22ef01cSRoman Divacky std::map<std::string, unsigned> SectionMap; 415f22ef01cSRoman Divacky std::map<std::string, unsigned> GCMap; 416f22ef01cSRoman Divacky unsigned MaxAlignment = 0; 417f22ef01cSRoman Divacky unsigned MaxGlobalType = 0; 418f22ef01cSRoman Divacky for (Module::const_global_iterator GV = M->global_begin(),E = M->global_end(); 419f22ef01cSRoman Divacky GV != E; ++GV) { 420f22ef01cSRoman Divacky MaxAlignment = std::max(MaxAlignment, GV->getAlignment()); 421f22ef01cSRoman Divacky MaxGlobalType = std::max(MaxGlobalType, VE.getTypeID(GV->getType())); 4226122f3e6SDimitry Andric if (GV->hasSection()) { 423f22ef01cSRoman Divacky // Give section names unique ID's. 424f22ef01cSRoman Divacky unsigned &Entry = SectionMap[GV->getSection()]; 4256122f3e6SDimitry Andric if (!Entry) { 426f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_SECTIONNAME, GV->getSection(), 427f22ef01cSRoman Divacky 0/*TODO*/, Stream); 428f22ef01cSRoman Divacky Entry = SectionMap.size(); 429f22ef01cSRoman Divacky } 4306122f3e6SDimitry Andric } 4316122f3e6SDimitry Andric } 432f22ef01cSRoman Divacky for (Module::const_iterator F = M->begin(), E = M->end(); F != E; ++F) { 433f22ef01cSRoman Divacky MaxAlignment = std::max(MaxAlignment, F->getAlignment()); 434f22ef01cSRoman Divacky if (F->hasSection()) { 435f22ef01cSRoman Divacky // Give section names unique ID's. 436f22ef01cSRoman Divacky unsigned &Entry = SectionMap[F->getSection()]; 437f22ef01cSRoman Divacky if (!Entry) { 438f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_SECTIONNAME, F->getSection(), 439f22ef01cSRoman Divacky 0/*TODO*/, Stream); 440f22ef01cSRoman Divacky Entry = SectionMap.size(); 441f22ef01cSRoman Divacky } 442f22ef01cSRoman Divacky } 443f22ef01cSRoman Divacky if (F->hasGC()) { 444f22ef01cSRoman Divacky // Same for GC names. 445f22ef01cSRoman Divacky unsigned &Entry = GCMap[F->getGC()]; 446f22ef01cSRoman Divacky if (!Entry) { 447f22ef01cSRoman Divacky WriteStringRecord(bitc::MODULE_CODE_GCNAME, F->getGC(), 448f22ef01cSRoman Divacky 0/*TODO*/, Stream); 449f22ef01cSRoman Divacky Entry = GCMap.size(); 450f22ef01cSRoman Divacky } 451f22ef01cSRoman Divacky } 452f22ef01cSRoman Divacky } 453f22ef01cSRoman Divacky 454f22ef01cSRoman Divacky // Emit abbrev for globals, now that we know # sections and max alignment. 455f22ef01cSRoman Divacky unsigned SimpleGVarAbbrev = 0; 456f22ef01cSRoman Divacky if (!M->global_empty()) { 457f22ef01cSRoman Divacky // Add an abbrev for common globals with no visibility or thread localness. 458f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 459f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_GLOBALVAR)); 460f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 461f22ef01cSRoman Divacky Log2_32_Ceil(MaxGlobalType+1))); 462f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // Constant. 463f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // Initializer. 464f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // Linkage. 465f22ef01cSRoman Divacky if (MaxAlignment == 0) // Alignment. 466f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(0)); 467f22ef01cSRoman Divacky else { 468f22ef01cSRoman Divacky unsigned MaxEncAlignment = Log2_32(MaxAlignment)+1; 469f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 470f22ef01cSRoman Divacky Log2_32_Ceil(MaxEncAlignment+1))); 471f22ef01cSRoman Divacky } 472f22ef01cSRoman Divacky if (SectionMap.empty()) // Section. 473f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(0)); 474f22ef01cSRoman Divacky else 475f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 476f22ef01cSRoman Divacky Log2_32_Ceil(SectionMap.size()+1))); 477f22ef01cSRoman Divacky // Don't bother emitting vis + thread local. 478f22ef01cSRoman Divacky SimpleGVarAbbrev = Stream.EmitAbbrev(Abbv); 479f22ef01cSRoman Divacky } 480f22ef01cSRoman Divacky 481f22ef01cSRoman Divacky // Emit the global variable information. 482f22ef01cSRoman Divacky SmallVector<unsigned, 64> Vals; 483f22ef01cSRoman Divacky for (Module::const_global_iterator GV = M->global_begin(),E = M->global_end(); 484f22ef01cSRoman Divacky GV != E; ++GV) { 485f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 486f22ef01cSRoman Divacky 487f22ef01cSRoman Divacky // GLOBALVAR: [type, isconst, initid, 4882754fe60SDimitry Andric // linkage, alignment, section, visibility, threadlocal, 4892754fe60SDimitry Andric // unnamed_addr] 490f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(GV->getType())); 491f22ef01cSRoman Divacky Vals.push_back(GV->isConstant()); 492f22ef01cSRoman Divacky Vals.push_back(GV->isDeclaration() ? 0 : 493f22ef01cSRoman Divacky (VE.getValueID(GV->getInitializer()) + 1)); 494f22ef01cSRoman Divacky Vals.push_back(getEncodedLinkage(GV)); 495f22ef01cSRoman Divacky Vals.push_back(Log2_32(GV->getAlignment())+1); 496f22ef01cSRoman Divacky Vals.push_back(GV->hasSection() ? SectionMap[GV->getSection()] : 0); 497f22ef01cSRoman Divacky if (GV->isThreadLocal() || 4982754fe60SDimitry Andric GV->getVisibility() != GlobalValue::DefaultVisibility || 4992754fe60SDimitry Andric GV->hasUnnamedAddr()) { 500f22ef01cSRoman Divacky Vals.push_back(getEncodedVisibility(GV)); 5017ae0e2c9SDimitry Andric Vals.push_back(getEncodedThreadLocalMode(GV)); 5022754fe60SDimitry Andric Vals.push_back(GV->hasUnnamedAddr()); 503f22ef01cSRoman Divacky } else { 504f22ef01cSRoman Divacky AbbrevToUse = SimpleGVarAbbrev; 505f22ef01cSRoman Divacky } 506f22ef01cSRoman Divacky 507f22ef01cSRoman Divacky Stream.EmitRecord(bitc::MODULE_CODE_GLOBALVAR, Vals, AbbrevToUse); 508f22ef01cSRoman Divacky Vals.clear(); 509f22ef01cSRoman Divacky } 510f22ef01cSRoman Divacky 511f22ef01cSRoman Divacky // Emit the function proto information. 512f22ef01cSRoman Divacky for (Module::const_iterator F = M->begin(), E = M->end(); F != E; ++F) { 513dff0c46cSDimitry Andric // FUNCTION: [type, callingconv, isproto, linkage, paramattrs, alignment, 514dff0c46cSDimitry Andric // section, visibility, gc, unnamed_addr] 515f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(F->getType())); 516f22ef01cSRoman Divacky Vals.push_back(F->getCallingConv()); 517f22ef01cSRoman Divacky Vals.push_back(F->isDeclaration()); 518f22ef01cSRoman Divacky Vals.push_back(getEncodedLinkage(F)); 519f22ef01cSRoman Divacky Vals.push_back(VE.getAttributeID(F->getAttributes())); 520f22ef01cSRoman Divacky Vals.push_back(Log2_32(F->getAlignment())+1); 521f22ef01cSRoman Divacky Vals.push_back(F->hasSection() ? SectionMap[F->getSection()] : 0); 522f22ef01cSRoman Divacky Vals.push_back(getEncodedVisibility(F)); 523f22ef01cSRoman Divacky Vals.push_back(F->hasGC() ? GCMap[F->getGC()] : 0); 5242754fe60SDimitry Andric Vals.push_back(F->hasUnnamedAddr()); 525f22ef01cSRoman Divacky 526f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 527f22ef01cSRoman Divacky Stream.EmitRecord(bitc::MODULE_CODE_FUNCTION, Vals, AbbrevToUse); 528f22ef01cSRoman Divacky Vals.clear(); 529f22ef01cSRoman Divacky } 530f22ef01cSRoman Divacky 531f22ef01cSRoman Divacky // Emit the alias information. 532f22ef01cSRoman Divacky for (Module::const_alias_iterator AI = M->alias_begin(), E = M->alias_end(); 533f22ef01cSRoman Divacky AI != E; ++AI) { 534dff0c46cSDimitry Andric // ALIAS: [alias type, aliasee val#, linkage, visibility] 535f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(AI->getType())); 536f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(AI->getAliasee())); 537f22ef01cSRoman Divacky Vals.push_back(getEncodedLinkage(AI)); 538f22ef01cSRoman Divacky Vals.push_back(getEncodedVisibility(AI)); 539f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 540f22ef01cSRoman Divacky Stream.EmitRecord(bitc::MODULE_CODE_ALIAS, Vals, AbbrevToUse); 541f22ef01cSRoman Divacky Vals.clear(); 542f22ef01cSRoman Divacky } 543f22ef01cSRoman Divacky } 544f22ef01cSRoman Divacky 545f22ef01cSRoman Divacky static uint64_t GetOptimizationFlags(const Value *V) { 546f22ef01cSRoman Divacky uint64_t Flags = 0; 547f22ef01cSRoman Divacky 548f22ef01cSRoman Divacky if (const OverflowingBinaryOperator *OBO = 549f22ef01cSRoman Divacky dyn_cast<OverflowingBinaryOperator>(V)) { 550f22ef01cSRoman Divacky if (OBO->hasNoSignedWrap()) 551f22ef01cSRoman Divacky Flags |= 1 << bitc::OBO_NO_SIGNED_WRAP; 552f22ef01cSRoman Divacky if (OBO->hasNoUnsignedWrap()) 553f22ef01cSRoman Divacky Flags |= 1 << bitc::OBO_NO_UNSIGNED_WRAP; 5542754fe60SDimitry Andric } else if (const PossiblyExactOperator *PEO = 5552754fe60SDimitry Andric dyn_cast<PossiblyExactOperator>(V)) { 5562754fe60SDimitry Andric if (PEO->isExact()) 5572754fe60SDimitry Andric Flags |= 1 << bitc::PEO_EXACT; 558f22ef01cSRoman Divacky } 559f22ef01cSRoman Divacky 560f22ef01cSRoman Divacky return Flags; 561f22ef01cSRoman Divacky } 562f22ef01cSRoman Divacky 563f22ef01cSRoman Divacky static void WriteMDNode(const MDNode *N, 564f22ef01cSRoman Divacky const ValueEnumerator &VE, 565f22ef01cSRoman Divacky BitstreamWriter &Stream, 566f22ef01cSRoman Divacky SmallVector<uint64_t, 64> &Record) { 567f22ef01cSRoman Divacky for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) { 568f22ef01cSRoman Divacky if (N->getOperand(i)) { 569f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(N->getOperand(i)->getType())); 570f22ef01cSRoman Divacky Record.push_back(VE.getValueID(N->getOperand(i))); 571f22ef01cSRoman Divacky } else { 572f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(Type::getVoidTy(N->getContext()))); 573f22ef01cSRoman Divacky Record.push_back(0); 574f22ef01cSRoman Divacky } 575f22ef01cSRoman Divacky } 57617a519f9SDimitry Andric unsigned MDCode = N->isFunctionLocal() ? bitc::METADATA_FN_NODE : 57717a519f9SDimitry Andric bitc::METADATA_NODE; 578f22ef01cSRoman Divacky Stream.EmitRecord(MDCode, Record, 0); 579f22ef01cSRoman Divacky Record.clear(); 580f22ef01cSRoman Divacky } 581f22ef01cSRoman Divacky 582e580952dSDimitry Andric static void WriteModuleMetadata(const Module *M, 583e580952dSDimitry Andric const ValueEnumerator &VE, 584f22ef01cSRoman Divacky BitstreamWriter &Stream) { 585f22ef01cSRoman Divacky const ValueEnumerator::ValueList &Vals = VE.getMDValues(); 586f22ef01cSRoman Divacky bool StartedMetadataBlock = false; 587f22ef01cSRoman Divacky unsigned MDSAbbrev = 0; 588f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 589f22ef01cSRoman Divacky for (unsigned i = 0, e = Vals.size(); i != e; ++i) { 590f22ef01cSRoman Divacky 591f22ef01cSRoman Divacky if (const MDNode *N = dyn_cast<MDNode>(Vals[i].first)) { 592f22ef01cSRoman Divacky if (!N->isFunctionLocal() || !N->getFunction()) { 593f22ef01cSRoman Divacky if (!StartedMetadataBlock) { 594f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 595f22ef01cSRoman Divacky StartedMetadataBlock = true; 596f22ef01cSRoman Divacky } 597f22ef01cSRoman Divacky WriteMDNode(N, VE, Stream, Record); 598f22ef01cSRoman Divacky } 599f22ef01cSRoman Divacky } else if (const MDString *MDS = dyn_cast<MDString>(Vals[i].first)) { 600f22ef01cSRoman Divacky if (!StartedMetadataBlock) { 601f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 602f22ef01cSRoman Divacky 603f22ef01cSRoman Divacky // Abbrev for METADATA_STRING. 604f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 605f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_STRING)); 606f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 607f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 608f22ef01cSRoman Divacky MDSAbbrev = Stream.EmitAbbrev(Abbv); 609f22ef01cSRoman Divacky StartedMetadataBlock = true; 610f22ef01cSRoman Divacky } 611f22ef01cSRoman Divacky 612f22ef01cSRoman Divacky // Code: [strchar x N] 613f22ef01cSRoman Divacky Record.append(MDS->begin(), MDS->end()); 614f22ef01cSRoman Divacky 615f22ef01cSRoman Divacky // Emit the finished record. 616f22ef01cSRoman Divacky Stream.EmitRecord(bitc::METADATA_STRING, Record, MDSAbbrev); 617f22ef01cSRoman Divacky Record.clear(); 618e580952dSDimitry Andric } 619e580952dSDimitry Andric } 620e580952dSDimitry Andric 621e580952dSDimitry Andric // Write named metadata. 622e580952dSDimitry Andric for (Module::const_named_metadata_iterator I = M->named_metadata_begin(), 623e580952dSDimitry Andric E = M->named_metadata_end(); I != E; ++I) { 624e580952dSDimitry Andric const NamedMDNode *NMD = I; 625f22ef01cSRoman Divacky if (!StartedMetadataBlock) { 626f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 627f22ef01cSRoman Divacky StartedMetadataBlock = true; 628f22ef01cSRoman Divacky } 629f22ef01cSRoman Divacky 630f22ef01cSRoman Divacky // Write name. 631f22ef01cSRoman Divacky StringRef Str = NMD->getName(); 632f22ef01cSRoman Divacky for (unsigned i = 0, e = Str.size(); i != e; ++i) 633f22ef01cSRoman Divacky Record.push_back(Str[i]); 634f22ef01cSRoman Divacky Stream.EmitRecord(bitc::METADATA_NAME, Record, 0/*TODO*/); 635f22ef01cSRoman Divacky Record.clear(); 636f22ef01cSRoman Divacky 637f22ef01cSRoman Divacky // Write named metadata operands. 638e580952dSDimitry Andric for (unsigned i = 0, e = NMD->getNumOperands(); i != e; ++i) 639f22ef01cSRoman Divacky Record.push_back(VE.getValueID(NMD->getOperand(i))); 64017a519f9SDimitry Andric Stream.EmitRecord(bitc::METADATA_NAMED_NODE, Record, 0); 641f22ef01cSRoman Divacky Record.clear(); 642f22ef01cSRoman Divacky } 643f22ef01cSRoman Divacky 644f22ef01cSRoman Divacky if (StartedMetadataBlock) 645f22ef01cSRoman Divacky Stream.ExitBlock(); 646f22ef01cSRoman Divacky } 647f22ef01cSRoman Divacky 648f22ef01cSRoman Divacky static void WriteFunctionLocalMetadata(const Function &F, 649f22ef01cSRoman Divacky const ValueEnumerator &VE, 650f22ef01cSRoman Divacky BitstreamWriter &Stream) { 651f22ef01cSRoman Divacky bool StartedMetadataBlock = false; 652f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 653ffd1746dSEd Schouten const SmallVector<const MDNode *, 8> &Vals = VE.getFunctionLocalMDValues(); 654f22ef01cSRoman Divacky for (unsigned i = 0, e = Vals.size(); i != e; ++i) 655ffd1746dSEd Schouten if (const MDNode *N = Vals[i]) 656f22ef01cSRoman Divacky if (N->isFunctionLocal() && N->getFunction() == &F) { 657f22ef01cSRoman Divacky if (!StartedMetadataBlock) { 658f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 659f22ef01cSRoman Divacky StartedMetadataBlock = true; 660f22ef01cSRoman Divacky } 661f22ef01cSRoman Divacky WriteMDNode(N, VE, Stream, Record); 662f22ef01cSRoman Divacky } 663f22ef01cSRoman Divacky 664f22ef01cSRoman Divacky if (StartedMetadataBlock) 665f22ef01cSRoman Divacky Stream.ExitBlock(); 666f22ef01cSRoman Divacky } 667f22ef01cSRoman Divacky 668f22ef01cSRoman Divacky static void WriteMetadataAttachment(const Function &F, 669f22ef01cSRoman Divacky const ValueEnumerator &VE, 670f22ef01cSRoman Divacky BitstreamWriter &Stream) { 671f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_ATTACHMENT_ID, 3); 672f22ef01cSRoman Divacky 673f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 674f22ef01cSRoman Divacky 675f22ef01cSRoman Divacky // Write metadata attachments 67617a519f9SDimitry Andric // METADATA_ATTACHMENT - [m x [value, [n x [id, mdnode]]] 677f22ef01cSRoman Divacky SmallVector<std::pair<unsigned, MDNode*>, 4> MDs; 678f22ef01cSRoman Divacky 679f22ef01cSRoman Divacky for (Function::const_iterator BB = F.begin(), E = F.end(); BB != E; ++BB) 680f22ef01cSRoman Divacky for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); 681f22ef01cSRoman Divacky I != E; ++I) { 682f22ef01cSRoman Divacky MDs.clear(); 683f22ef01cSRoman Divacky I->getAllMetadataOtherThanDebugLoc(MDs); 684f22ef01cSRoman Divacky 685f22ef01cSRoman Divacky // If no metadata, ignore instruction. 686f22ef01cSRoman Divacky if (MDs.empty()) continue; 687f22ef01cSRoman Divacky 688f22ef01cSRoman Divacky Record.push_back(VE.getInstructionID(I)); 689f22ef01cSRoman Divacky 690f22ef01cSRoman Divacky for (unsigned i = 0, e = MDs.size(); i != e; ++i) { 691f22ef01cSRoman Divacky Record.push_back(MDs[i].first); 692f22ef01cSRoman Divacky Record.push_back(VE.getValueID(MDs[i].second)); 693f22ef01cSRoman Divacky } 69417a519f9SDimitry Andric Stream.EmitRecord(bitc::METADATA_ATTACHMENT, Record, 0); 695f22ef01cSRoman Divacky Record.clear(); 696f22ef01cSRoman Divacky } 697f22ef01cSRoman Divacky 698f22ef01cSRoman Divacky Stream.ExitBlock(); 699f22ef01cSRoman Divacky } 700f22ef01cSRoman Divacky 701f22ef01cSRoman Divacky static void WriteModuleMetadataStore(const Module *M, BitstreamWriter &Stream) { 702f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 703f22ef01cSRoman Divacky 704f22ef01cSRoman Divacky // Write metadata kinds 705f22ef01cSRoman Divacky // METADATA_KIND - [n x [id, name]] 706f22ef01cSRoman Divacky SmallVector<StringRef, 4> Names; 707f22ef01cSRoman Divacky M->getMDKindNames(Names); 708f22ef01cSRoman Divacky 709e580952dSDimitry Andric if (Names.empty()) return; 710f22ef01cSRoman Divacky 711f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 712f22ef01cSRoman Divacky 713e580952dSDimitry Andric for (unsigned MDKindID = 0, e = Names.size(); MDKindID != e; ++MDKindID) { 714f22ef01cSRoman Divacky Record.push_back(MDKindID); 715f22ef01cSRoman Divacky StringRef KName = Names[MDKindID]; 716f22ef01cSRoman Divacky Record.append(KName.begin(), KName.end()); 717f22ef01cSRoman Divacky 718f22ef01cSRoman Divacky Stream.EmitRecord(bitc::METADATA_KIND, Record, 0); 719f22ef01cSRoman Divacky Record.clear(); 720f22ef01cSRoman Divacky } 721f22ef01cSRoman Divacky 722f22ef01cSRoman Divacky Stream.ExitBlock(); 723f22ef01cSRoman Divacky } 724f22ef01cSRoman Divacky 7257ae0e2c9SDimitry Andric static void EmitAPInt(SmallVectorImpl<uint64_t> &Vals, 7267ae0e2c9SDimitry Andric unsigned &Code, unsigned &AbbrevToUse, const APInt &Val, 7277ae0e2c9SDimitry Andric bool EmitSizeForWideNumbers = false 7287ae0e2c9SDimitry Andric ) { 7297ae0e2c9SDimitry Andric if (Val.getBitWidth() <= 64) { 7307ae0e2c9SDimitry Andric uint64_t V = Val.getSExtValue(); 7317ae0e2c9SDimitry Andric if ((int64_t)V >= 0) 7327ae0e2c9SDimitry Andric Vals.push_back(V << 1); 7337ae0e2c9SDimitry Andric else 7347ae0e2c9SDimitry Andric Vals.push_back((-V << 1) | 1); 7357ae0e2c9SDimitry Andric Code = bitc::CST_CODE_INTEGER; 7367ae0e2c9SDimitry Andric AbbrevToUse = CONSTANTS_INTEGER_ABBREV; 7377ae0e2c9SDimitry Andric } else { 7387ae0e2c9SDimitry Andric // Wide integers, > 64 bits in size. 7397ae0e2c9SDimitry Andric // We have an arbitrary precision integer value to write whose 7407ae0e2c9SDimitry Andric // bit width is > 64. However, in canonical unsigned integer 7417ae0e2c9SDimitry Andric // format it is likely that the high bits are going to be zero. 7427ae0e2c9SDimitry Andric // So, we only write the number of active words. 7437ae0e2c9SDimitry Andric unsigned NWords = Val.getActiveWords(); 7447ae0e2c9SDimitry Andric 7457ae0e2c9SDimitry Andric if (EmitSizeForWideNumbers) 7467ae0e2c9SDimitry Andric Vals.push_back(NWords); 7477ae0e2c9SDimitry Andric 7487ae0e2c9SDimitry Andric const uint64_t *RawWords = Val.getRawData(); 7497ae0e2c9SDimitry Andric for (unsigned i = 0; i != NWords; ++i) { 7507ae0e2c9SDimitry Andric int64_t V = RawWords[i]; 7517ae0e2c9SDimitry Andric if (V >= 0) 7527ae0e2c9SDimitry Andric Vals.push_back(V << 1); 7537ae0e2c9SDimitry Andric else 7547ae0e2c9SDimitry Andric Vals.push_back((-V << 1) | 1); 7557ae0e2c9SDimitry Andric } 7567ae0e2c9SDimitry Andric Code = bitc::CST_CODE_WIDE_INTEGER; 7577ae0e2c9SDimitry Andric } 7587ae0e2c9SDimitry Andric } 7597ae0e2c9SDimitry Andric 760f22ef01cSRoman Divacky static void WriteConstants(unsigned FirstVal, unsigned LastVal, 761f22ef01cSRoman Divacky const ValueEnumerator &VE, 762f22ef01cSRoman Divacky BitstreamWriter &Stream, bool isGlobal) { 763f22ef01cSRoman Divacky if (FirstVal == LastVal) return; 764f22ef01cSRoman Divacky 765f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::CONSTANTS_BLOCK_ID, 4); 766f22ef01cSRoman Divacky 767f22ef01cSRoman Divacky unsigned AggregateAbbrev = 0; 768f22ef01cSRoman Divacky unsigned String8Abbrev = 0; 769f22ef01cSRoman Divacky unsigned CString7Abbrev = 0; 770f22ef01cSRoman Divacky unsigned CString6Abbrev = 0; 771f22ef01cSRoman Divacky // If this is a constant pool for the module, emit module-specific abbrevs. 772f22ef01cSRoman Divacky if (isGlobal) { 773f22ef01cSRoman Divacky // Abbrev for CST_CODE_AGGREGATE. 774f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 775f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_AGGREGATE)); 776f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 777f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, Log2_32_Ceil(LastVal+1))); 778f22ef01cSRoman Divacky AggregateAbbrev = Stream.EmitAbbrev(Abbv); 779f22ef01cSRoman Divacky 780f22ef01cSRoman Divacky // Abbrev for CST_CODE_STRING. 781f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 782f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_STRING)); 783f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 784f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 785f22ef01cSRoman Divacky String8Abbrev = Stream.EmitAbbrev(Abbv); 786f22ef01cSRoman Divacky // Abbrev for CST_CODE_CSTRING. 787f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 788f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CSTRING)); 789f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 790f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 791f22ef01cSRoman Divacky CString7Abbrev = Stream.EmitAbbrev(Abbv); 792f22ef01cSRoman Divacky // Abbrev for CST_CODE_CSTRING. 793f22ef01cSRoman Divacky Abbv = new BitCodeAbbrev(); 794f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CSTRING)); 795f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 796f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 797f22ef01cSRoman Divacky CString6Abbrev = Stream.EmitAbbrev(Abbv); 798f22ef01cSRoman Divacky } 799f22ef01cSRoman Divacky 800f22ef01cSRoman Divacky SmallVector<uint64_t, 64> Record; 801f22ef01cSRoman Divacky 802f22ef01cSRoman Divacky const ValueEnumerator::ValueList &Vals = VE.getValues(); 8036122f3e6SDimitry Andric Type *LastTy = 0; 804f22ef01cSRoman Divacky for (unsigned i = FirstVal; i != LastVal; ++i) { 805f22ef01cSRoman Divacky const Value *V = Vals[i].first; 806f22ef01cSRoman Divacky // If we need to switch types, do so now. 807f22ef01cSRoman Divacky if (V->getType() != LastTy) { 808f22ef01cSRoman Divacky LastTy = V->getType(); 809f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(LastTy)); 810f22ef01cSRoman Divacky Stream.EmitRecord(bitc::CST_CODE_SETTYPE, Record, 811f22ef01cSRoman Divacky CONSTANTS_SETTYPE_ABBREV); 812f22ef01cSRoman Divacky Record.clear(); 813f22ef01cSRoman Divacky } 814f22ef01cSRoman Divacky 815f22ef01cSRoman Divacky if (const InlineAsm *IA = dyn_cast<InlineAsm>(V)) { 816f22ef01cSRoman Divacky Record.push_back(unsigned(IA->hasSideEffects()) | 817f22ef01cSRoman Divacky unsigned(IA->isAlignStack()) << 1); 818f22ef01cSRoman Divacky 819f22ef01cSRoman Divacky // Add the asm string. 820f22ef01cSRoman Divacky const std::string &AsmStr = IA->getAsmString(); 821f22ef01cSRoman Divacky Record.push_back(AsmStr.size()); 822f22ef01cSRoman Divacky for (unsigned i = 0, e = AsmStr.size(); i != e; ++i) 823f22ef01cSRoman Divacky Record.push_back(AsmStr[i]); 824f22ef01cSRoman Divacky 825f22ef01cSRoman Divacky // Add the constraint string. 826f22ef01cSRoman Divacky const std::string &ConstraintStr = IA->getConstraintString(); 827f22ef01cSRoman Divacky Record.push_back(ConstraintStr.size()); 828f22ef01cSRoman Divacky for (unsigned i = 0, e = ConstraintStr.size(); i != e; ++i) 829f22ef01cSRoman Divacky Record.push_back(ConstraintStr[i]); 830f22ef01cSRoman Divacky Stream.EmitRecord(bitc::CST_CODE_INLINEASM, Record); 831f22ef01cSRoman Divacky Record.clear(); 832f22ef01cSRoman Divacky continue; 833f22ef01cSRoman Divacky } 834f22ef01cSRoman Divacky const Constant *C = cast<Constant>(V); 835f22ef01cSRoman Divacky unsigned Code = -1U; 836f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 837f22ef01cSRoman Divacky if (C->isNullValue()) { 838f22ef01cSRoman Divacky Code = bitc::CST_CODE_NULL; 839f22ef01cSRoman Divacky } else if (isa<UndefValue>(C)) { 840f22ef01cSRoman Divacky Code = bitc::CST_CODE_UNDEF; 841f22ef01cSRoman Divacky } else if (const ConstantInt *IV = dyn_cast<ConstantInt>(C)) { 8427ae0e2c9SDimitry Andric EmitAPInt(Record, Code, AbbrevToUse, IV->getValue()); 843f22ef01cSRoman Divacky } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(C)) { 844f22ef01cSRoman Divacky Code = bitc::CST_CODE_FLOAT; 8456122f3e6SDimitry Andric Type *Ty = CFP->getType(); 846dff0c46cSDimitry Andric if (Ty->isHalfTy() || Ty->isFloatTy() || Ty->isDoubleTy()) { 847f22ef01cSRoman Divacky Record.push_back(CFP->getValueAPF().bitcastToAPInt().getZExtValue()); 848f22ef01cSRoman Divacky } else if (Ty->isX86_FP80Ty()) { 849f22ef01cSRoman Divacky // api needed to prevent premature destruction 850f22ef01cSRoman Divacky // bits are not in the same order as a normal i80 APInt, compensate. 851f22ef01cSRoman Divacky APInt api = CFP->getValueAPF().bitcastToAPInt(); 852f22ef01cSRoman Divacky const uint64_t *p = api.getRawData(); 853f22ef01cSRoman Divacky Record.push_back((p[1] << 48) | (p[0] >> 16)); 854f22ef01cSRoman Divacky Record.push_back(p[0] & 0xffffLL); 855f22ef01cSRoman Divacky } else if (Ty->isFP128Ty() || Ty->isPPC_FP128Ty()) { 856f22ef01cSRoman Divacky APInt api = CFP->getValueAPF().bitcastToAPInt(); 857f22ef01cSRoman Divacky const uint64_t *p = api.getRawData(); 858f22ef01cSRoman Divacky Record.push_back(p[0]); 859f22ef01cSRoman Divacky Record.push_back(p[1]); 860f22ef01cSRoman Divacky } else { 861f22ef01cSRoman Divacky assert (0 && "Unknown FP type!"); 862f22ef01cSRoman Divacky } 863dff0c46cSDimitry Andric } else if (isa<ConstantDataSequential>(C) && 864dff0c46cSDimitry Andric cast<ConstantDataSequential>(C)->isString()) { 865dff0c46cSDimitry Andric const ConstantDataSequential *Str = cast<ConstantDataSequential>(C); 866f22ef01cSRoman Divacky // Emit constant strings specially. 867dff0c46cSDimitry Andric unsigned NumElts = Str->getNumElements(); 868f22ef01cSRoman Divacky // If this is a null-terminated string, use the denser CSTRING encoding. 869dff0c46cSDimitry Andric if (Str->isCString()) { 870f22ef01cSRoman Divacky Code = bitc::CST_CODE_CSTRING; 871dff0c46cSDimitry Andric --NumElts; // Don't encode the null, which isn't allowed by char6. 872f22ef01cSRoman Divacky } else { 873f22ef01cSRoman Divacky Code = bitc::CST_CODE_STRING; 874f22ef01cSRoman Divacky AbbrevToUse = String8Abbrev; 875f22ef01cSRoman Divacky } 876f22ef01cSRoman Divacky bool isCStr7 = Code == bitc::CST_CODE_CSTRING; 877f22ef01cSRoman Divacky bool isCStrChar6 = Code == bitc::CST_CODE_CSTRING; 878dff0c46cSDimitry Andric for (unsigned i = 0; i != NumElts; ++i) { 879dff0c46cSDimitry Andric unsigned char V = Str->getElementAsInteger(i); 880f22ef01cSRoman Divacky Record.push_back(V); 881f22ef01cSRoman Divacky isCStr7 &= (V & 128) == 0; 882f22ef01cSRoman Divacky if (isCStrChar6) 883f22ef01cSRoman Divacky isCStrChar6 = BitCodeAbbrevOp::isChar6(V); 884f22ef01cSRoman Divacky } 885f22ef01cSRoman Divacky 886f22ef01cSRoman Divacky if (isCStrChar6) 887f22ef01cSRoman Divacky AbbrevToUse = CString6Abbrev; 888f22ef01cSRoman Divacky else if (isCStr7) 889f22ef01cSRoman Divacky AbbrevToUse = CString7Abbrev; 890dff0c46cSDimitry Andric } else if (const ConstantDataSequential *CDS = 891dff0c46cSDimitry Andric dyn_cast<ConstantDataSequential>(C)) { 892dff0c46cSDimitry Andric Code = bitc::CST_CODE_DATA; 893dff0c46cSDimitry Andric Type *EltTy = CDS->getType()->getElementType(); 894dff0c46cSDimitry Andric if (isa<IntegerType>(EltTy)) { 895dff0c46cSDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) 896dff0c46cSDimitry Andric Record.push_back(CDS->getElementAsInteger(i)); 897dff0c46cSDimitry Andric } else if (EltTy->isFloatTy()) { 898dff0c46cSDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) { 899dff0c46cSDimitry Andric union { float F; uint32_t I; }; 900dff0c46cSDimitry Andric F = CDS->getElementAsFloat(i); 901dff0c46cSDimitry Andric Record.push_back(I); 902dff0c46cSDimitry Andric } 903dff0c46cSDimitry Andric } else { 904dff0c46cSDimitry Andric assert(EltTy->isDoubleTy() && "Unknown ConstantData element type"); 905dff0c46cSDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) { 906dff0c46cSDimitry Andric union { double F; uint64_t I; }; 907dff0c46cSDimitry Andric F = CDS->getElementAsDouble(i); 908dff0c46cSDimitry Andric Record.push_back(I); 909dff0c46cSDimitry Andric } 910dff0c46cSDimitry Andric } 911dff0c46cSDimitry Andric } else if (isa<ConstantArray>(C) || isa<ConstantStruct>(C) || 912dff0c46cSDimitry Andric isa<ConstantVector>(C)) { 913f22ef01cSRoman Divacky Code = bitc::CST_CODE_AGGREGATE; 914f22ef01cSRoman Divacky for (unsigned i = 0, e = C->getNumOperands(); i != e; ++i) 915f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(i))); 916f22ef01cSRoman Divacky AbbrevToUse = AggregateAbbrev; 917f22ef01cSRoman Divacky } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 918f22ef01cSRoman Divacky switch (CE->getOpcode()) { 919f22ef01cSRoman Divacky default: 920f22ef01cSRoman Divacky if (Instruction::isCast(CE->getOpcode())) { 921f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_CAST; 922f22ef01cSRoman Divacky Record.push_back(GetEncodedCastOpcode(CE->getOpcode())); 923f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 924f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 925f22ef01cSRoman Divacky AbbrevToUse = CONSTANTS_CE_CAST_Abbrev; 926f22ef01cSRoman Divacky } else { 927f22ef01cSRoman Divacky assert(CE->getNumOperands() == 2 && "Unknown constant expr!"); 928f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_BINOP; 929f22ef01cSRoman Divacky Record.push_back(GetEncodedBinaryOpcode(CE->getOpcode())); 930f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 931f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 932f22ef01cSRoman Divacky uint64_t Flags = GetOptimizationFlags(CE); 933f22ef01cSRoman Divacky if (Flags != 0) 934f22ef01cSRoman Divacky Record.push_back(Flags); 935f22ef01cSRoman Divacky } 936f22ef01cSRoman Divacky break; 937f22ef01cSRoman Divacky case Instruction::GetElementPtr: 938f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_GEP; 939f22ef01cSRoman Divacky if (cast<GEPOperator>(C)->isInBounds()) 940f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_INBOUNDS_GEP; 941f22ef01cSRoman Divacky for (unsigned i = 0, e = CE->getNumOperands(); i != e; ++i) { 942f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(i)->getType())); 943f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(i))); 944f22ef01cSRoman Divacky } 945f22ef01cSRoman Divacky break; 946f22ef01cSRoman Divacky case Instruction::Select: 947f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_SELECT; 948f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 949f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 950f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(2))); 951f22ef01cSRoman Divacky break; 952f22ef01cSRoman Divacky case Instruction::ExtractElement: 953f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_EXTRACTELT; 954f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 955f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 956f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 957f22ef01cSRoman Divacky break; 958f22ef01cSRoman Divacky case Instruction::InsertElement: 959f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_INSERTELT; 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::ShuffleVector: 965f22ef01cSRoman Divacky // If the return type and argument types are the same, this is a 966f22ef01cSRoman Divacky // standard shufflevector instruction. If the types are different, 967f22ef01cSRoman Divacky // then the shuffle is widening or truncating the input vectors, and 968f22ef01cSRoman Divacky // the argument type must also be encoded. 969f22ef01cSRoman Divacky if (C->getType() == C->getOperand(0)->getType()) { 970f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_SHUFFLEVEC; 971f22ef01cSRoman Divacky } else { 972f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_SHUFVEC_EX; 973f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 974f22ef01cSRoman Divacky } 975f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 976f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 977f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(2))); 978f22ef01cSRoman Divacky break; 979f22ef01cSRoman Divacky case Instruction::ICmp: 980f22ef01cSRoman Divacky case Instruction::FCmp: 981f22ef01cSRoman Divacky Code = bitc::CST_CODE_CE_CMP; 982f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 983f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(0))); 984f22ef01cSRoman Divacky Record.push_back(VE.getValueID(C->getOperand(1))); 985f22ef01cSRoman Divacky Record.push_back(CE->getPredicate()); 986f22ef01cSRoman Divacky break; 987f22ef01cSRoman Divacky } 988f22ef01cSRoman Divacky } else if (const BlockAddress *BA = dyn_cast<BlockAddress>(C)) { 989f22ef01cSRoman Divacky Code = bitc::CST_CODE_BLOCKADDRESS; 990f22ef01cSRoman Divacky Record.push_back(VE.getTypeID(BA->getFunction()->getType())); 991f22ef01cSRoman Divacky Record.push_back(VE.getValueID(BA->getFunction())); 992f22ef01cSRoman Divacky Record.push_back(VE.getGlobalBasicBlockID(BA->getBasicBlock())); 993f22ef01cSRoman Divacky } else { 994e580952dSDimitry Andric #ifndef NDEBUG 995e580952dSDimitry Andric C->dump(); 996e580952dSDimitry Andric #endif 997f22ef01cSRoman Divacky llvm_unreachable("Unknown constant!"); 998f22ef01cSRoman Divacky } 999f22ef01cSRoman Divacky Stream.EmitRecord(Code, Record, AbbrevToUse); 1000f22ef01cSRoman Divacky Record.clear(); 1001f22ef01cSRoman Divacky } 1002f22ef01cSRoman Divacky 1003f22ef01cSRoman Divacky Stream.ExitBlock(); 1004f22ef01cSRoman Divacky } 1005f22ef01cSRoman Divacky 1006f22ef01cSRoman Divacky static void WriteModuleConstants(const ValueEnumerator &VE, 1007f22ef01cSRoman Divacky BitstreamWriter &Stream) { 1008f22ef01cSRoman Divacky const ValueEnumerator::ValueList &Vals = VE.getValues(); 1009f22ef01cSRoman Divacky 1010f22ef01cSRoman Divacky // Find the first constant to emit, which is the first non-globalvalue value. 1011f22ef01cSRoman Divacky // We know globalvalues have been emitted by WriteModuleInfo. 1012f22ef01cSRoman Divacky for (unsigned i = 0, e = Vals.size(); i != e; ++i) { 1013f22ef01cSRoman Divacky if (!isa<GlobalValue>(Vals[i].first)) { 1014f22ef01cSRoman Divacky WriteConstants(i, Vals.size(), VE, Stream, true); 1015f22ef01cSRoman Divacky return; 1016f22ef01cSRoman Divacky } 1017f22ef01cSRoman Divacky } 1018f22ef01cSRoman Divacky } 1019f22ef01cSRoman Divacky 1020f22ef01cSRoman Divacky /// PushValueAndType - The file has to encode both the value and type id for 1021f22ef01cSRoman Divacky /// many values, because we need to know what type to create for forward 1022f22ef01cSRoman Divacky /// references. However, most operands are not forward references, so this type 1023f22ef01cSRoman Divacky /// field is not needed. 1024f22ef01cSRoman Divacky /// 1025f22ef01cSRoman Divacky /// This function adds V's value ID to Vals. If the value ID is higher than the 1026f22ef01cSRoman Divacky /// instruction ID, then it is a forward reference, and it also includes the 1027f22ef01cSRoman Divacky /// type ID. 1028f22ef01cSRoman Divacky static bool PushValueAndType(const Value *V, unsigned InstID, 1029f22ef01cSRoman Divacky SmallVector<unsigned, 64> &Vals, 1030f22ef01cSRoman Divacky ValueEnumerator &VE) { 1031f22ef01cSRoman Divacky unsigned ValID = VE.getValueID(V); 1032f22ef01cSRoman Divacky Vals.push_back(ValID); 1033f22ef01cSRoman Divacky if (ValID >= InstID) { 1034f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(V->getType())); 1035f22ef01cSRoman Divacky return true; 1036f22ef01cSRoman Divacky } 1037f22ef01cSRoman Divacky return false; 1038f22ef01cSRoman Divacky } 1039f22ef01cSRoman Divacky 1040f22ef01cSRoman Divacky /// WriteInstruction - Emit an instruction to the specified stream. 1041f22ef01cSRoman Divacky static void WriteInstruction(const Instruction &I, unsigned InstID, 1042f22ef01cSRoman Divacky ValueEnumerator &VE, BitstreamWriter &Stream, 1043f22ef01cSRoman Divacky SmallVector<unsigned, 64> &Vals) { 1044f22ef01cSRoman Divacky unsigned Code = 0; 1045f22ef01cSRoman Divacky unsigned AbbrevToUse = 0; 1046f22ef01cSRoman Divacky VE.setInstructionID(&I); 1047f22ef01cSRoman Divacky switch (I.getOpcode()) { 1048f22ef01cSRoman Divacky default: 1049f22ef01cSRoman Divacky if (Instruction::isCast(I.getOpcode())) { 1050f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_CAST; 1051f22ef01cSRoman Divacky if (!PushValueAndType(I.getOperand(0), InstID, Vals, VE)) 1052f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_CAST_ABBREV; 1053f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getType())); 1054f22ef01cSRoman Divacky Vals.push_back(GetEncodedCastOpcode(I.getOpcode())); 1055f22ef01cSRoman Divacky } else { 1056f22ef01cSRoman Divacky assert(isa<BinaryOperator>(I) && "Unknown instruction!"); 1057f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_BINOP; 1058f22ef01cSRoman Divacky if (!PushValueAndType(I.getOperand(0), InstID, Vals, VE)) 1059f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_BINOP_ABBREV; 1060f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1061f22ef01cSRoman Divacky Vals.push_back(GetEncodedBinaryOpcode(I.getOpcode())); 1062f22ef01cSRoman Divacky uint64_t Flags = GetOptimizationFlags(&I); 1063f22ef01cSRoman Divacky if (Flags != 0) { 1064f22ef01cSRoman Divacky if (AbbrevToUse == FUNCTION_INST_BINOP_ABBREV) 1065f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_BINOP_FLAGS_ABBREV; 1066f22ef01cSRoman Divacky Vals.push_back(Flags); 1067f22ef01cSRoman Divacky } 1068f22ef01cSRoman Divacky } 1069f22ef01cSRoman Divacky break; 1070f22ef01cSRoman Divacky 1071f22ef01cSRoman Divacky case Instruction::GetElementPtr: 1072f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_GEP; 1073f22ef01cSRoman Divacky if (cast<GEPOperator>(&I)->isInBounds()) 1074f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INBOUNDS_GEP; 1075f22ef01cSRoman Divacky for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i) 1076f22ef01cSRoman Divacky PushValueAndType(I.getOperand(i), InstID, Vals, VE); 1077f22ef01cSRoman Divacky break; 1078f22ef01cSRoman Divacky case Instruction::ExtractValue: { 1079f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_EXTRACTVAL; 1080f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1081f22ef01cSRoman Divacky const ExtractValueInst *EVI = cast<ExtractValueInst>(&I); 1082f22ef01cSRoman Divacky for (const unsigned *i = EVI->idx_begin(), *e = EVI->idx_end(); i != e; ++i) 1083f22ef01cSRoman Divacky Vals.push_back(*i); 1084f22ef01cSRoman Divacky break; 1085f22ef01cSRoman Divacky } 1086f22ef01cSRoman Divacky case Instruction::InsertValue: { 1087f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INSERTVAL; 1088f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1089f22ef01cSRoman Divacky PushValueAndType(I.getOperand(1), InstID, Vals, VE); 1090f22ef01cSRoman Divacky const InsertValueInst *IVI = cast<InsertValueInst>(&I); 1091f22ef01cSRoman Divacky for (const unsigned *i = IVI->idx_begin(), *e = IVI->idx_end(); i != e; ++i) 1092f22ef01cSRoman Divacky Vals.push_back(*i); 1093f22ef01cSRoman Divacky break; 1094f22ef01cSRoman Divacky } 1095f22ef01cSRoman Divacky case Instruction::Select: 1096f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_VSELECT; 1097f22ef01cSRoman Divacky PushValueAndType(I.getOperand(1), InstID, Vals, VE); 1098f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(2))); 1099f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1100f22ef01cSRoman Divacky break; 1101f22ef01cSRoman Divacky case Instruction::ExtractElement: 1102f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_EXTRACTELT; 1103f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1104f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1105f22ef01cSRoman Divacky break; 1106f22ef01cSRoman Divacky case Instruction::InsertElement: 1107f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INSERTELT; 1108f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1109f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1110f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(2))); 1111f22ef01cSRoman Divacky break; 1112f22ef01cSRoman Divacky case Instruction::ShuffleVector: 1113f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_SHUFFLEVEC; 1114f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1115f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1116f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(2))); 1117f22ef01cSRoman Divacky break; 1118f22ef01cSRoman Divacky case Instruction::ICmp: 1119f22ef01cSRoman Divacky case Instruction::FCmp: 1120f22ef01cSRoman Divacky // compare returning Int1Ty or vector of Int1Ty 1121f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_CMP2; 1122f22ef01cSRoman Divacky PushValueAndType(I.getOperand(0), InstID, Vals, VE); 1123f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(1))); 1124f22ef01cSRoman Divacky Vals.push_back(cast<CmpInst>(I).getPredicate()); 1125f22ef01cSRoman Divacky break; 1126f22ef01cSRoman Divacky 1127f22ef01cSRoman Divacky case Instruction::Ret: 1128f22ef01cSRoman Divacky { 1129f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_RET; 1130f22ef01cSRoman Divacky unsigned NumOperands = I.getNumOperands(); 1131f22ef01cSRoman Divacky if (NumOperands == 0) 1132f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_RET_VOID_ABBREV; 1133f22ef01cSRoman Divacky else if (NumOperands == 1) { 1134f22ef01cSRoman Divacky if (!PushValueAndType(I.getOperand(0), InstID, Vals, VE)) 1135f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_RET_VAL_ABBREV; 1136f22ef01cSRoman Divacky } else { 1137f22ef01cSRoman Divacky for (unsigned i = 0, e = NumOperands; i != e; ++i) 1138f22ef01cSRoman Divacky PushValueAndType(I.getOperand(i), InstID, Vals, VE); 1139f22ef01cSRoman Divacky } 1140f22ef01cSRoman Divacky } 1141f22ef01cSRoman Divacky break; 1142f22ef01cSRoman Divacky case Instruction::Br: 1143f22ef01cSRoman Divacky { 1144f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_BR; 1145f22ef01cSRoman Divacky BranchInst &II = cast<BranchInst>(I); 1146f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II.getSuccessor(0))); 1147f22ef01cSRoman Divacky if (II.isConditional()) { 1148f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II.getSuccessor(1))); 1149f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II.getCondition())); 1150f22ef01cSRoman Divacky } 1151f22ef01cSRoman Divacky } 1152f22ef01cSRoman Divacky break; 1153f22ef01cSRoman Divacky case Instruction::Switch: 1154dff0c46cSDimitry Andric { 11557ae0e2c9SDimitry Andric // Redefine Vals, since here we need to use 64 bit values 11567ae0e2c9SDimitry Andric // explicitly to store large APInt numbers. 11577ae0e2c9SDimitry Andric SmallVector<uint64_t, 128> Vals64; 11587ae0e2c9SDimitry Andric 1159f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_SWITCH; 1160dff0c46cSDimitry Andric SwitchInst &SI = cast<SwitchInst>(I); 11617ae0e2c9SDimitry Andric 11627ae0e2c9SDimitry Andric uint32_t SwitchRecordHeader = SI.hash() | (SWITCH_INST_MAGIC << 16); 11637ae0e2c9SDimitry Andric Vals64.push_back(SwitchRecordHeader); 11647ae0e2c9SDimitry Andric 11657ae0e2c9SDimitry Andric Vals64.push_back(VE.getTypeID(SI.getCondition()->getType())); 11667ae0e2c9SDimitry Andric Vals64.push_back(VE.getValueID(SI.getCondition())); 11677ae0e2c9SDimitry Andric Vals64.push_back(VE.getValueID(SI.getDefaultDest())); 11687ae0e2c9SDimitry Andric Vals64.push_back(SI.getNumCases()); 1169dff0c46cSDimitry Andric for (SwitchInst::CaseIt i = SI.case_begin(), e = SI.case_end(); 1170dff0c46cSDimitry Andric i != e; ++i) { 11717ae0e2c9SDimitry Andric IntegersSubset& CaseRanges = i.getCaseValueEx(); 11727ae0e2c9SDimitry Andric unsigned Code, Abbrev; // will unused. 11737ae0e2c9SDimitry Andric 11747ae0e2c9SDimitry Andric if (CaseRanges.isSingleNumber()) { 11757ae0e2c9SDimitry Andric Vals64.push_back(1/*NumItems = 1*/); 11767ae0e2c9SDimitry Andric Vals64.push_back(true/*IsSingleNumber = true*/); 11777ae0e2c9SDimitry Andric EmitAPInt(Vals64, Code, Abbrev, CaseRanges.getSingleNumber(0), true); 11787ae0e2c9SDimitry Andric } else { 11797ae0e2c9SDimitry Andric 11807ae0e2c9SDimitry Andric Vals64.push_back(CaseRanges.getNumItems()); 11817ae0e2c9SDimitry Andric 11827ae0e2c9SDimitry Andric if (CaseRanges.isSingleNumbersOnly()) { 11837ae0e2c9SDimitry Andric for (unsigned ri = 0, rn = CaseRanges.getNumItems(); 11847ae0e2c9SDimitry Andric ri != rn; ++ri) { 11857ae0e2c9SDimitry Andric 11867ae0e2c9SDimitry Andric Vals64.push_back(true/*IsSingleNumber = true*/); 11877ae0e2c9SDimitry Andric 11887ae0e2c9SDimitry Andric EmitAPInt(Vals64, Code, Abbrev, 11897ae0e2c9SDimitry Andric CaseRanges.getSingleNumber(ri), true); 1190dff0c46cSDimitry Andric } 11917ae0e2c9SDimitry Andric } else 11927ae0e2c9SDimitry Andric for (unsigned ri = 0, rn = CaseRanges.getNumItems(); 11937ae0e2c9SDimitry Andric ri != rn; ++ri) { 11947ae0e2c9SDimitry Andric IntegersSubset::Range r = CaseRanges.getItem(ri); 11957ae0e2c9SDimitry Andric bool IsSingleNumber = CaseRanges.isSingleNumber(ri); 11967ae0e2c9SDimitry Andric 11977ae0e2c9SDimitry Andric Vals64.push_back(IsSingleNumber); 11987ae0e2c9SDimitry Andric 11997ae0e2c9SDimitry Andric EmitAPInt(Vals64, Code, Abbrev, r.getLow(), true); 12007ae0e2c9SDimitry Andric if (!IsSingleNumber) 12017ae0e2c9SDimitry Andric EmitAPInt(Vals64, Code, Abbrev, r.getHigh(), true); 12027ae0e2c9SDimitry Andric } 12037ae0e2c9SDimitry Andric } 12047ae0e2c9SDimitry Andric Vals64.push_back(VE.getValueID(i.getCaseSuccessor())); 12057ae0e2c9SDimitry Andric } 12067ae0e2c9SDimitry Andric 12077ae0e2c9SDimitry Andric Stream.EmitRecord(Code, Vals64, AbbrevToUse); 12087ae0e2c9SDimitry Andric 12097ae0e2c9SDimitry Andric // Also do expected action - clear external Vals collection: 12107ae0e2c9SDimitry Andric Vals.clear(); 12117ae0e2c9SDimitry Andric return; 1212dff0c46cSDimitry Andric } 1213f22ef01cSRoman Divacky break; 1214f22ef01cSRoman Divacky case Instruction::IndirectBr: 1215f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INDIRECTBR; 1216f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); 1217f22ef01cSRoman Divacky for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i) 1218f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(i))); 1219f22ef01cSRoman Divacky break; 1220f22ef01cSRoman Divacky 1221f22ef01cSRoman Divacky case Instruction::Invoke: { 1222f22ef01cSRoman Divacky const InvokeInst *II = cast<InvokeInst>(&I); 1223f22ef01cSRoman Divacky const Value *Callee(II->getCalledValue()); 12246122f3e6SDimitry Andric PointerType *PTy = cast<PointerType>(Callee->getType()); 12256122f3e6SDimitry Andric FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 1226f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_INVOKE; 1227f22ef01cSRoman Divacky 1228f22ef01cSRoman Divacky Vals.push_back(VE.getAttributeID(II->getAttributes())); 1229f22ef01cSRoman Divacky Vals.push_back(II->getCallingConv()); 1230f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II->getNormalDest())); 1231f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(II->getUnwindDest())); 1232f22ef01cSRoman Divacky PushValueAndType(Callee, InstID, Vals, VE); 1233f22ef01cSRoman Divacky 1234f22ef01cSRoman Divacky // Emit value #'s for the fixed parameters. 1235f22ef01cSRoman Divacky for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) 1236f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(i))); // fixed param. 1237f22ef01cSRoman Divacky 1238f22ef01cSRoman Divacky // Emit type/value pairs for varargs params. 1239f22ef01cSRoman Divacky if (FTy->isVarArg()) { 1240f22ef01cSRoman Divacky for (unsigned i = FTy->getNumParams(), e = I.getNumOperands()-3; 1241f22ef01cSRoman Divacky i != e; ++i) 1242f22ef01cSRoman Divacky PushValueAndType(I.getOperand(i), InstID, Vals, VE); // vararg 1243f22ef01cSRoman Divacky } 1244f22ef01cSRoman Divacky break; 1245f22ef01cSRoman Divacky } 12466122f3e6SDimitry Andric case Instruction::Resume: 12476122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_RESUME; 12486122f3e6SDimitry Andric PushValueAndType(I.getOperand(0), InstID, Vals, VE); 12496122f3e6SDimitry Andric break; 1250f22ef01cSRoman Divacky case Instruction::Unreachable: 1251f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_UNREACHABLE; 1252f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_UNREACHABLE_ABBREV; 1253f22ef01cSRoman Divacky break; 1254f22ef01cSRoman Divacky 125517a519f9SDimitry Andric case Instruction::PHI: { 125617a519f9SDimitry Andric const PHINode &PN = cast<PHINode>(I); 1257f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_PHI; 125817a519f9SDimitry Andric Vals.push_back(VE.getTypeID(PN.getType())); 125917a519f9SDimitry Andric for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) { 126017a519f9SDimitry Andric Vals.push_back(VE.getValueID(PN.getIncomingValue(i))); 126117a519f9SDimitry Andric Vals.push_back(VE.getValueID(PN.getIncomingBlock(i))); 126217a519f9SDimitry Andric } 1263f22ef01cSRoman Divacky break; 126417a519f9SDimitry Andric } 1265f22ef01cSRoman Divacky 12666122f3e6SDimitry Andric case Instruction::LandingPad: { 12676122f3e6SDimitry Andric const LandingPadInst &LP = cast<LandingPadInst>(I); 12686122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_LANDINGPAD; 12696122f3e6SDimitry Andric Vals.push_back(VE.getTypeID(LP.getType())); 12706122f3e6SDimitry Andric PushValueAndType(LP.getPersonalityFn(), InstID, Vals, VE); 12716122f3e6SDimitry Andric Vals.push_back(LP.isCleanup()); 12726122f3e6SDimitry Andric Vals.push_back(LP.getNumClauses()); 12736122f3e6SDimitry Andric for (unsigned I = 0, E = LP.getNumClauses(); I != E; ++I) { 12746122f3e6SDimitry Andric if (LP.isCatch(I)) 12756122f3e6SDimitry Andric Vals.push_back(LandingPadInst::Catch); 12766122f3e6SDimitry Andric else 12776122f3e6SDimitry Andric Vals.push_back(LandingPadInst::Filter); 12786122f3e6SDimitry Andric PushValueAndType(LP.getClause(I), InstID, Vals, VE); 12796122f3e6SDimitry Andric } 12806122f3e6SDimitry Andric break; 12816122f3e6SDimitry Andric } 12826122f3e6SDimitry Andric 1283f22ef01cSRoman Divacky case Instruction::Alloca: 1284f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_ALLOCA; 1285f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getType())); 1286ffd1746dSEd Schouten Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); 1287f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(0))); // size. 1288f22ef01cSRoman Divacky Vals.push_back(Log2_32(cast<AllocaInst>(I).getAlignment())+1); 1289f22ef01cSRoman Divacky break; 1290f22ef01cSRoman Divacky 1291f22ef01cSRoman Divacky case Instruction::Load: 12926122f3e6SDimitry Andric if (cast<LoadInst>(I).isAtomic()) { 12936122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_LOADATOMIC; 12946122f3e6SDimitry Andric PushValueAndType(I.getOperand(0), InstID, Vals, VE); 12956122f3e6SDimitry Andric } else { 1296f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_LOAD; 1297f22ef01cSRoman Divacky if (!PushValueAndType(I.getOperand(0), InstID, Vals, VE)) // ptr 1298f22ef01cSRoman Divacky AbbrevToUse = FUNCTION_INST_LOAD_ABBREV; 12996122f3e6SDimitry Andric } 1300f22ef01cSRoman Divacky Vals.push_back(Log2_32(cast<LoadInst>(I).getAlignment())+1); 1301f22ef01cSRoman Divacky Vals.push_back(cast<LoadInst>(I).isVolatile()); 13026122f3e6SDimitry Andric if (cast<LoadInst>(I).isAtomic()) { 13036122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering(cast<LoadInst>(I).getOrdering())); 13046122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope(cast<LoadInst>(I).getSynchScope())); 13056122f3e6SDimitry Andric } 1306f22ef01cSRoman Divacky break; 1307f22ef01cSRoman Divacky case Instruction::Store: 13086122f3e6SDimitry Andric if (cast<StoreInst>(I).isAtomic()) 13096122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_STOREATOMIC; 13106122f3e6SDimitry Andric else 131117a519f9SDimitry Andric Code = bitc::FUNC_CODE_INST_STORE; 1312f22ef01cSRoman Divacky PushValueAndType(I.getOperand(1), InstID, Vals, VE); // ptrty + ptr 1313f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(0))); // val. 1314f22ef01cSRoman Divacky Vals.push_back(Log2_32(cast<StoreInst>(I).getAlignment())+1); 1315f22ef01cSRoman Divacky Vals.push_back(cast<StoreInst>(I).isVolatile()); 13166122f3e6SDimitry Andric if (cast<StoreInst>(I).isAtomic()) { 13176122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering(cast<StoreInst>(I).getOrdering())); 13186122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope(cast<StoreInst>(I).getSynchScope())); 13196122f3e6SDimitry Andric } 13206122f3e6SDimitry Andric break; 13216122f3e6SDimitry Andric case Instruction::AtomicCmpXchg: 13226122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_CMPXCHG; 13236122f3e6SDimitry Andric PushValueAndType(I.getOperand(0), InstID, Vals, VE); // ptrty + ptr 13246122f3e6SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(1))); // cmp. 13256122f3e6SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(2))); // newval. 13266122f3e6SDimitry Andric Vals.push_back(cast<AtomicCmpXchgInst>(I).isVolatile()); 13276122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering( 13286122f3e6SDimitry Andric cast<AtomicCmpXchgInst>(I).getOrdering())); 13296122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope( 13306122f3e6SDimitry Andric cast<AtomicCmpXchgInst>(I).getSynchScope())); 13316122f3e6SDimitry Andric break; 13326122f3e6SDimitry Andric case Instruction::AtomicRMW: 13336122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_ATOMICRMW; 13346122f3e6SDimitry Andric PushValueAndType(I.getOperand(0), InstID, Vals, VE); // ptrty + ptr 13356122f3e6SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(1))); // val. 13366122f3e6SDimitry Andric Vals.push_back(GetEncodedRMWOperation( 13376122f3e6SDimitry Andric cast<AtomicRMWInst>(I).getOperation())); 13386122f3e6SDimitry Andric Vals.push_back(cast<AtomicRMWInst>(I).isVolatile()); 13396122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering(cast<AtomicRMWInst>(I).getOrdering())); 13406122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope( 13416122f3e6SDimitry Andric cast<AtomicRMWInst>(I).getSynchScope())); 13426122f3e6SDimitry Andric break; 13436122f3e6SDimitry Andric case Instruction::Fence: 13446122f3e6SDimitry Andric Code = bitc::FUNC_CODE_INST_FENCE; 13456122f3e6SDimitry Andric Vals.push_back(GetEncodedOrdering(cast<FenceInst>(I).getOrdering())); 13466122f3e6SDimitry Andric Vals.push_back(GetEncodedSynchScope(cast<FenceInst>(I).getSynchScope())); 1347f22ef01cSRoman Divacky break; 1348f22ef01cSRoman Divacky case Instruction::Call: { 1349ffd1746dSEd Schouten const CallInst &CI = cast<CallInst>(I); 13506122f3e6SDimitry Andric PointerType *PTy = cast<PointerType>(CI.getCalledValue()->getType()); 13516122f3e6SDimitry Andric FunctionType *FTy = cast<FunctionType>(PTy->getElementType()); 1352f22ef01cSRoman Divacky 135317a519f9SDimitry Andric Code = bitc::FUNC_CODE_INST_CALL; 1354f22ef01cSRoman Divacky 1355ffd1746dSEd Schouten Vals.push_back(VE.getAttributeID(CI.getAttributes())); 1356ffd1746dSEd Schouten Vals.push_back((CI.getCallingConv() << 1) | unsigned(CI.isTailCall())); 1357ffd1746dSEd Schouten PushValueAndType(CI.getCalledValue(), InstID, Vals, VE); // Callee 1358f22ef01cSRoman Divacky 1359f22ef01cSRoman Divacky // Emit value #'s for the fixed parameters. 1360f22ef01cSRoman Divacky for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) 1361ffd1746dSEd Schouten Vals.push_back(VE.getValueID(CI.getArgOperand(i))); // fixed param. 1362f22ef01cSRoman Divacky 1363f22ef01cSRoman Divacky // Emit type/value pairs for varargs params. 1364f22ef01cSRoman Divacky if (FTy->isVarArg()) { 1365ffd1746dSEd Schouten for (unsigned i = FTy->getNumParams(), e = CI.getNumArgOperands(); 1366f22ef01cSRoman Divacky i != e; ++i) 1367ffd1746dSEd Schouten PushValueAndType(CI.getArgOperand(i), InstID, Vals, VE); // varargs 1368f22ef01cSRoman Divacky } 1369f22ef01cSRoman Divacky break; 1370f22ef01cSRoman Divacky } 1371f22ef01cSRoman Divacky case Instruction::VAArg: 1372f22ef01cSRoman Divacky Code = bitc::FUNC_CODE_INST_VAARG; 1373f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); // valistty 1374f22ef01cSRoman Divacky Vals.push_back(VE.getValueID(I.getOperand(0))); // valist. 1375f22ef01cSRoman Divacky Vals.push_back(VE.getTypeID(I.getType())); // restype. 1376f22ef01cSRoman Divacky break; 1377f22ef01cSRoman Divacky } 1378f22ef01cSRoman Divacky 1379f22ef01cSRoman Divacky Stream.EmitRecord(Code, Vals, AbbrevToUse); 1380f22ef01cSRoman Divacky Vals.clear(); 1381f22ef01cSRoman Divacky } 1382f22ef01cSRoman Divacky 1383f22ef01cSRoman Divacky // Emit names for globals/functions etc. 1384f22ef01cSRoman Divacky static void WriteValueSymbolTable(const ValueSymbolTable &VST, 1385f22ef01cSRoman Divacky const ValueEnumerator &VE, 1386f22ef01cSRoman Divacky BitstreamWriter &Stream) { 1387f22ef01cSRoman Divacky if (VST.empty()) return; 1388f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::VALUE_SYMTAB_BLOCK_ID, 4); 1389f22ef01cSRoman Divacky 1390f22ef01cSRoman Divacky // FIXME: Set up the abbrev, we know how many values there are! 1391f22ef01cSRoman Divacky // FIXME: We know if the type names can use 7-bit ascii. 1392f22ef01cSRoman Divacky SmallVector<unsigned, 64> NameVals; 1393f22ef01cSRoman Divacky 1394f22ef01cSRoman Divacky for (ValueSymbolTable::const_iterator SI = VST.begin(), SE = VST.end(); 1395f22ef01cSRoman Divacky SI != SE; ++SI) { 1396f22ef01cSRoman Divacky 1397f22ef01cSRoman Divacky const ValueName &Name = *SI; 1398f22ef01cSRoman Divacky 1399f22ef01cSRoman Divacky // Figure out the encoding to use for the name. 1400f22ef01cSRoman Divacky bool is7Bit = true; 1401f22ef01cSRoman Divacky bool isChar6 = true; 1402f22ef01cSRoman Divacky for (const char *C = Name.getKeyData(), *E = C+Name.getKeyLength(); 1403f22ef01cSRoman Divacky C != E; ++C) { 1404f22ef01cSRoman Divacky if (isChar6) 1405f22ef01cSRoman Divacky isChar6 = BitCodeAbbrevOp::isChar6(*C); 1406f22ef01cSRoman Divacky if ((unsigned char)*C & 128) { 1407f22ef01cSRoman Divacky is7Bit = false; 1408f22ef01cSRoman Divacky break; // don't bother scanning the rest. 1409f22ef01cSRoman Divacky } 1410f22ef01cSRoman Divacky } 1411f22ef01cSRoman Divacky 1412f22ef01cSRoman Divacky unsigned AbbrevToUse = VST_ENTRY_8_ABBREV; 1413f22ef01cSRoman Divacky 1414f22ef01cSRoman Divacky // VST_ENTRY: [valueid, namechar x N] 1415f22ef01cSRoman Divacky // VST_BBENTRY: [bbid, namechar x N] 1416f22ef01cSRoman Divacky unsigned Code; 1417f22ef01cSRoman Divacky if (isa<BasicBlock>(SI->getValue())) { 1418f22ef01cSRoman Divacky Code = bitc::VST_CODE_BBENTRY; 1419f22ef01cSRoman Divacky if (isChar6) 1420f22ef01cSRoman Divacky AbbrevToUse = VST_BBENTRY_6_ABBREV; 1421f22ef01cSRoman Divacky } else { 1422f22ef01cSRoman Divacky Code = bitc::VST_CODE_ENTRY; 1423f22ef01cSRoman Divacky if (isChar6) 1424f22ef01cSRoman Divacky AbbrevToUse = VST_ENTRY_6_ABBREV; 1425f22ef01cSRoman Divacky else if (is7Bit) 1426f22ef01cSRoman Divacky AbbrevToUse = VST_ENTRY_7_ABBREV; 1427f22ef01cSRoman Divacky } 1428f22ef01cSRoman Divacky 1429f22ef01cSRoman Divacky NameVals.push_back(VE.getValueID(SI->getValue())); 1430f22ef01cSRoman Divacky for (const char *P = Name.getKeyData(), 1431f22ef01cSRoman Divacky *E = Name.getKeyData()+Name.getKeyLength(); P != E; ++P) 1432f22ef01cSRoman Divacky NameVals.push_back((unsigned char)*P); 1433f22ef01cSRoman Divacky 1434f22ef01cSRoman Divacky // Emit the finished record. 1435f22ef01cSRoman Divacky Stream.EmitRecord(Code, NameVals, AbbrevToUse); 1436f22ef01cSRoman Divacky NameVals.clear(); 1437f22ef01cSRoman Divacky } 1438f22ef01cSRoman Divacky Stream.ExitBlock(); 1439f22ef01cSRoman Divacky } 1440f22ef01cSRoman Divacky 1441f22ef01cSRoman Divacky /// WriteFunction - Emit a function body to the module stream. 1442f22ef01cSRoman Divacky static void WriteFunction(const Function &F, ValueEnumerator &VE, 1443f22ef01cSRoman Divacky BitstreamWriter &Stream) { 1444f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::FUNCTION_BLOCK_ID, 4); 1445f22ef01cSRoman Divacky VE.incorporateFunction(F); 1446f22ef01cSRoman Divacky 1447f22ef01cSRoman Divacky SmallVector<unsigned, 64> Vals; 1448f22ef01cSRoman Divacky 1449f22ef01cSRoman Divacky // Emit the number of basic blocks, so the reader can create them ahead of 1450f22ef01cSRoman Divacky // time. 1451f22ef01cSRoman Divacky Vals.push_back(VE.getBasicBlocks().size()); 1452f22ef01cSRoman Divacky Stream.EmitRecord(bitc::FUNC_CODE_DECLAREBLOCKS, Vals); 1453f22ef01cSRoman Divacky Vals.clear(); 1454f22ef01cSRoman Divacky 1455f22ef01cSRoman Divacky // If there are function-local constants, emit them now. 1456f22ef01cSRoman Divacky unsigned CstStart, CstEnd; 1457f22ef01cSRoman Divacky VE.getFunctionConstantRange(CstStart, CstEnd); 1458f22ef01cSRoman Divacky WriteConstants(CstStart, CstEnd, VE, Stream, false); 1459f22ef01cSRoman Divacky 1460f22ef01cSRoman Divacky // If there is function-local metadata, emit it now. 1461f22ef01cSRoman Divacky WriteFunctionLocalMetadata(F, VE, Stream); 1462f22ef01cSRoman Divacky 1463f22ef01cSRoman Divacky // Keep a running idea of what the instruction ID is. 1464f22ef01cSRoman Divacky unsigned InstID = CstEnd; 1465f22ef01cSRoman Divacky 1466f22ef01cSRoman Divacky bool NeedsMetadataAttachment = false; 1467f22ef01cSRoman Divacky 1468f22ef01cSRoman Divacky DebugLoc LastDL; 1469f22ef01cSRoman Divacky 1470f22ef01cSRoman Divacky // Finally, emit all the instructions, in order. 1471f22ef01cSRoman Divacky for (Function::const_iterator BB = F.begin(), E = F.end(); BB != E; ++BB) 1472f22ef01cSRoman Divacky for (BasicBlock::const_iterator I = BB->begin(), E = BB->end(); 1473f22ef01cSRoman Divacky I != E; ++I) { 1474f22ef01cSRoman Divacky WriteInstruction(*I, InstID, VE, Stream, Vals); 1475f22ef01cSRoman Divacky 1476f22ef01cSRoman Divacky if (!I->getType()->isVoidTy()) 1477f22ef01cSRoman Divacky ++InstID; 1478f22ef01cSRoman Divacky 1479f22ef01cSRoman Divacky // If the instruction has metadata, write a metadata attachment later. 1480f22ef01cSRoman Divacky NeedsMetadataAttachment |= I->hasMetadataOtherThanDebugLoc(); 1481f22ef01cSRoman Divacky 1482f22ef01cSRoman Divacky // If the instruction has a debug location, emit it. 1483f22ef01cSRoman Divacky DebugLoc DL = I->getDebugLoc(); 1484f22ef01cSRoman Divacky if (DL.isUnknown()) { 1485f22ef01cSRoman Divacky // nothing todo. 1486f22ef01cSRoman Divacky } else if (DL == LastDL) { 1487f22ef01cSRoman Divacky // Just repeat the same debug loc as last time. 1488f22ef01cSRoman Divacky Stream.EmitRecord(bitc::FUNC_CODE_DEBUG_LOC_AGAIN, Vals); 1489f22ef01cSRoman Divacky } else { 1490f22ef01cSRoman Divacky MDNode *Scope, *IA; 1491f22ef01cSRoman Divacky DL.getScopeAndInlinedAt(Scope, IA, I->getContext()); 1492f22ef01cSRoman Divacky 1493f22ef01cSRoman Divacky Vals.push_back(DL.getLine()); 1494f22ef01cSRoman Divacky Vals.push_back(DL.getCol()); 1495f22ef01cSRoman Divacky Vals.push_back(Scope ? VE.getValueID(Scope)+1 : 0); 1496f22ef01cSRoman Divacky Vals.push_back(IA ? VE.getValueID(IA)+1 : 0); 149717a519f9SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_DEBUG_LOC, Vals); 1498f22ef01cSRoman Divacky Vals.clear(); 1499f22ef01cSRoman Divacky 1500f22ef01cSRoman Divacky LastDL = DL; 1501f22ef01cSRoman Divacky } 1502f22ef01cSRoman Divacky } 1503f22ef01cSRoman Divacky 1504f22ef01cSRoman Divacky // Emit names for all the instructions etc. 1505f22ef01cSRoman Divacky WriteValueSymbolTable(F.getValueSymbolTable(), VE, Stream); 1506f22ef01cSRoman Divacky 1507f22ef01cSRoman Divacky if (NeedsMetadataAttachment) 1508f22ef01cSRoman Divacky WriteMetadataAttachment(F, VE, Stream); 1509f22ef01cSRoman Divacky VE.purgeFunction(); 1510f22ef01cSRoman Divacky Stream.ExitBlock(); 1511f22ef01cSRoman Divacky } 1512f22ef01cSRoman Divacky 1513f22ef01cSRoman Divacky // Emit blockinfo, which defines the standard abbreviations etc. 1514f22ef01cSRoman Divacky static void WriteBlockInfo(const ValueEnumerator &VE, BitstreamWriter &Stream) { 1515f22ef01cSRoman Divacky // We only want to emit block info records for blocks that have multiple 1516f22ef01cSRoman Divacky // instances: CONSTANTS_BLOCK, FUNCTION_BLOCK and VALUE_SYMTAB_BLOCK. Other 1517f22ef01cSRoman Divacky // blocks can defined their abbrevs inline. 1518f22ef01cSRoman Divacky Stream.EnterBlockInfoBlock(2); 1519f22ef01cSRoman Divacky 1520f22ef01cSRoman Divacky { // 8-bit fixed-width VST_ENTRY/VST_BBENTRY strings. 1521f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1522f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 3)); 1523f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1524f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1525f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 1526f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, 1527f22ef01cSRoman Divacky Abbv) != VST_ENTRY_8_ABBREV) 1528f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1529f22ef01cSRoman Divacky } 1530f22ef01cSRoman Divacky 1531f22ef01cSRoman Divacky { // 7-bit fixed width VST_ENTRY strings. 1532f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1533f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_ENTRY)); 1534f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1535f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1536f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 1537f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, 1538f22ef01cSRoman Divacky Abbv) != VST_ENTRY_7_ABBREV) 1539f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1540f22ef01cSRoman Divacky } 1541f22ef01cSRoman Divacky { // 6-bit char6 VST_ENTRY strings. 1542f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1543f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_ENTRY)); 1544f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1545f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1546f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 1547f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, 1548f22ef01cSRoman Divacky Abbv) != VST_ENTRY_6_ABBREV) 1549f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1550f22ef01cSRoman Divacky } 1551f22ef01cSRoman Divacky { // 6-bit char6 VST_BBENTRY strings. 1552f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1553f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_BBENTRY)); 1554f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1555f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1556f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 1557f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, 1558f22ef01cSRoman Divacky Abbv) != VST_BBENTRY_6_ABBREV) 1559f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1560f22ef01cSRoman Divacky } 1561f22ef01cSRoman Divacky 1562f22ef01cSRoman Divacky 1563f22ef01cSRoman Divacky 1564f22ef01cSRoman Divacky { // SETTYPE abbrev for CONSTANTS_BLOCK. 1565f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1566f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_SETTYPE)); 1567f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1568f22ef01cSRoman Divacky Log2_32_Ceil(VE.getTypes().size()+1))); 1569f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, 1570f22ef01cSRoman Divacky Abbv) != CONSTANTS_SETTYPE_ABBREV) 1571f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1572f22ef01cSRoman Divacky } 1573f22ef01cSRoman Divacky 1574f22ef01cSRoman Divacky { // INTEGER abbrev for CONSTANTS_BLOCK. 1575f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1576f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_INTEGER)); 1577f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1578f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, 1579f22ef01cSRoman Divacky Abbv) != CONSTANTS_INTEGER_ABBREV) 1580f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1581f22ef01cSRoman Divacky } 1582f22ef01cSRoman Divacky 1583f22ef01cSRoman Divacky { // CE_CAST abbrev for CONSTANTS_BLOCK. 1584f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1585f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CE_CAST)); 1586f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // cast opc 1587f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // typeid 1588f22ef01cSRoman Divacky Log2_32_Ceil(VE.getTypes().size()+1))); 1589f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // value id 1590f22ef01cSRoman Divacky 1591f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, 1592f22ef01cSRoman Divacky Abbv) != CONSTANTS_CE_CAST_Abbrev) 1593f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1594f22ef01cSRoman Divacky } 1595f22ef01cSRoman Divacky { // NULL abbrev for CONSTANTS_BLOCK. 1596f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1597f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_NULL)); 1598f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, 1599f22ef01cSRoman Divacky Abbv) != CONSTANTS_NULL_Abbrev) 1600f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1601f22ef01cSRoman Divacky } 1602f22ef01cSRoman Divacky 1603f22ef01cSRoman Divacky // FIXME: This should only use space for first class types! 1604f22ef01cSRoman Divacky 1605f22ef01cSRoman Divacky { // INST_LOAD abbrev for FUNCTION_BLOCK. 1606f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1607f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_LOAD)); 1608f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // Ptr 1609f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // Align 1610f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // volatile 1611f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1612f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_LOAD_ABBREV) 1613f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1614f22ef01cSRoman Divacky } 1615f22ef01cSRoman Divacky { // INST_BINOP abbrev for FUNCTION_BLOCK. 1616f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1617f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_BINOP)); 1618f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 1619f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // RHS 1620f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 1621f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1622f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_BINOP_ABBREV) 1623f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1624f22ef01cSRoman Divacky } 1625f22ef01cSRoman Divacky { // INST_BINOP_FLAGS abbrev for FUNCTION_BLOCK. 1626f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1627f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_BINOP)); 1628f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 1629f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // RHS 1630f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 1631f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); // flags 1632f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1633f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_BINOP_FLAGS_ABBREV) 1634f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1635f22ef01cSRoman Divacky } 1636f22ef01cSRoman Divacky { // INST_CAST abbrev for FUNCTION_BLOCK. 1637f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1638f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_CAST)); 1639f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // OpVal 1640f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // dest ty 1641f22ef01cSRoman Divacky Log2_32_Ceil(VE.getTypes().size()+1))); 1642f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 1643f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1644f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_CAST_ABBREV) 1645f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1646f22ef01cSRoman Divacky } 1647f22ef01cSRoman Divacky 1648f22ef01cSRoman Divacky { // INST_RET abbrev for FUNCTION_BLOCK. 1649f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1650f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_RET)); 1651f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1652f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_RET_VOID_ABBREV) 1653f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1654f22ef01cSRoman Divacky } 1655f22ef01cSRoman Divacky { // INST_RET abbrev for FUNCTION_BLOCK. 1656f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1657f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_RET)); 1658f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // ValID 1659f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1660f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_RET_VAL_ABBREV) 1661f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1662f22ef01cSRoman Divacky } 1663f22ef01cSRoman Divacky { // INST_UNREACHABLE abbrev for FUNCTION_BLOCK. 1664f22ef01cSRoman Divacky BitCodeAbbrev *Abbv = new BitCodeAbbrev(); 1665f22ef01cSRoman Divacky Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_UNREACHABLE)); 1666f22ef01cSRoman Divacky if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, 1667f22ef01cSRoman Divacky Abbv) != FUNCTION_INST_UNREACHABLE_ABBREV) 1668f22ef01cSRoman Divacky llvm_unreachable("Unexpected abbrev ordering!"); 1669f22ef01cSRoman Divacky } 1670f22ef01cSRoman Divacky 1671f22ef01cSRoman Divacky Stream.ExitBlock(); 1672f22ef01cSRoman Divacky } 1673f22ef01cSRoman Divacky 1674dff0c46cSDimitry Andric // Sort the Users based on the order in which the reader parses the bitcode 1675dff0c46cSDimitry Andric // file. 1676dff0c46cSDimitry Andric static bool bitcodereader_order(const User *lhs, const User *rhs) { 1677dff0c46cSDimitry Andric // TODO: Implement. 1678dff0c46cSDimitry Andric return true; 1679dff0c46cSDimitry Andric } 1680dff0c46cSDimitry Andric 1681dff0c46cSDimitry Andric static void WriteUseList(const Value *V, const ValueEnumerator &VE, 1682dff0c46cSDimitry Andric BitstreamWriter &Stream) { 1683dff0c46cSDimitry Andric 1684dff0c46cSDimitry Andric // One or zero uses can't get out of order. 1685dff0c46cSDimitry Andric if (V->use_empty() || V->hasNUses(1)) 1686dff0c46cSDimitry Andric return; 1687dff0c46cSDimitry Andric 1688dff0c46cSDimitry Andric // Make a copy of the in-memory use-list for sorting. 1689dff0c46cSDimitry Andric unsigned UseListSize = std::distance(V->use_begin(), V->use_end()); 1690dff0c46cSDimitry Andric SmallVector<const User*, 8> UseList; 1691dff0c46cSDimitry Andric UseList.reserve(UseListSize); 1692dff0c46cSDimitry Andric for (Value::const_use_iterator I = V->use_begin(), E = V->use_end(); 1693dff0c46cSDimitry Andric I != E; ++I) { 1694dff0c46cSDimitry Andric const User *U = *I; 1695dff0c46cSDimitry Andric UseList.push_back(U); 1696dff0c46cSDimitry Andric } 1697dff0c46cSDimitry Andric 1698dff0c46cSDimitry Andric // Sort the copy based on the order read by the BitcodeReader. 1699dff0c46cSDimitry Andric std::sort(UseList.begin(), UseList.end(), bitcodereader_order); 1700dff0c46cSDimitry Andric 1701dff0c46cSDimitry Andric // TODO: Generate a diff between the BitcodeWriter in-memory use-list and the 1702dff0c46cSDimitry Andric // sorted list (i.e., the expected BitcodeReader in-memory use-list). 1703dff0c46cSDimitry Andric 1704dff0c46cSDimitry Andric // TODO: Emit the USELIST_CODE_ENTRYs. 1705dff0c46cSDimitry Andric } 1706dff0c46cSDimitry Andric 1707dff0c46cSDimitry Andric static void WriteFunctionUseList(const Function *F, ValueEnumerator &VE, 1708dff0c46cSDimitry Andric BitstreamWriter &Stream) { 1709dff0c46cSDimitry Andric VE.incorporateFunction(*F); 1710dff0c46cSDimitry Andric 1711dff0c46cSDimitry Andric for (Function::const_arg_iterator AI = F->arg_begin(), AE = F->arg_end(); 1712dff0c46cSDimitry Andric AI != AE; ++AI) 1713dff0c46cSDimitry Andric WriteUseList(AI, VE, Stream); 1714dff0c46cSDimitry Andric for (Function::const_iterator BB = F->begin(), FE = F->end(); BB != FE; 1715dff0c46cSDimitry Andric ++BB) { 1716dff0c46cSDimitry Andric WriteUseList(BB, VE, Stream); 1717dff0c46cSDimitry Andric for (BasicBlock::const_iterator II = BB->begin(), IE = BB->end(); II != IE; 1718dff0c46cSDimitry Andric ++II) { 1719dff0c46cSDimitry Andric WriteUseList(II, VE, Stream); 1720dff0c46cSDimitry Andric for (User::const_op_iterator OI = II->op_begin(), E = II->op_end(); 1721dff0c46cSDimitry Andric OI != E; ++OI) { 1722dff0c46cSDimitry Andric if ((isa<Constant>(*OI) && !isa<GlobalValue>(*OI)) || 1723dff0c46cSDimitry Andric isa<InlineAsm>(*OI)) 1724dff0c46cSDimitry Andric WriteUseList(*OI, VE, Stream); 1725dff0c46cSDimitry Andric } 1726dff0c46cSDimitry Andric } 1727dff0c46cSDimitry Andric } 1728dff0c46cSDimitry Andric VE.purgeFunction(); 1729dff0c46cSDimitry Andric } 1730dff0c46cSDimitry Andric 1731dff0c46cSDimitry Andric // Emit use-lists. 1732dff0c46cSDimitry Andric static void WriteModuleUseLists(const Module *M, ValueEnumerator &VE, 1733dff0c46cSDimitry Andric BitstreamWriter &Stream) { 1734dff0c46cSDimitry Andric Stream.EnterSubblock(bitc::USELIST_BLOCK_ID, 3); 1735dff0c46cSDimitry Andric 1736dff0c46cSDimitry Andric // XXX: this modifies the module, but in a way that should never change the 1737dff0c46cSDimitry Andric // behavior of any pass or codegen in LLVM. The problem is that GVs may 1738dff0c46cSDimitry Andric // contain entries in the use_list that do not exist in the Module and are 1739dff0c46cSDimitry Andric // not stored in the .bc file. 1740dff0c46cSDimitry Andric for (Module::const_global_iterator I = M->global_begin(), E = M->global_end(); 1741dff0c46cSDimitry Andric I != E; ++I) 1742dff0c46cSDimitry Andric I->removeDeadConstantUsers(); 1743dff0c46cSDimitry Andric 1744dff0c46cSDimitry Andric // Write the global variables. 1745dff0c46cSDimitry Andric for (Module::const_global_iterator GI = M->global_begin(), 1746dff0c46cSDimitry Andric GE = M->global_end(); GI != GE; ++GI) { 1747dff0c46cSDimitry Andric WriteUseList(GI, VE, Stream); 1748dff0c46cSDimitry Andric 1749dff0c46cSDimitry Andric // Write the global variable initializers. 1750dff0c46cSDimitry Andric if (GI->hasInitializer()) 1751dff0c46cSDimitry Andric WriteUseList(GI->getInitializer(), VE, Stream); 1752dff0c46cSDimitry Andric } 1753dff0c46cSDimitry Andric 1754dff0c46cSDimitry Andric // Write the functions. 1755dff0c46cSDimitry Andric for (Module::const_iterator FI = M->begin(), FE = M->end(); FI != FE; ++FI) { 1756dff0c46cSDimitry Andric WriteUseList(FI, VE, Stream); 1757dff0c46cSDimitry Andric if (!FI->isDeclaration()) 1758dff0c46cSDimitry Andric WriteFunctionUseList(FI, VE, Stream); 1759dff0c46cSDimitry Andric } 1760dff0c46cSDimitry Andric 1761dff0c46cSDimitry Andric // Write the aliases. 1762dff0c46cSDimitry Andric for (Module::const_alias_iterator AI = M->alias_begin(), AE = M->alias_end(); 1763dff0c46cSDimitry Andric AI != AE; ++AI) { 1764dff0c46cSDimitry Andric WriteUseList(AI, VE, Stream); 1765dff0c46cSDimitry Andric WriteUseList(AI->getAliasee(), VE, Stream); 1766dff0c46cSDimitry Andric } 1767dff0c46cSDimitry Andric 1768dff0c46cSDimitry Andric Stream.ExitBlock(); 1769dff0c46cSDimitry Andric } 1770f22ef01cSRoman Divacky 1771f22ef01cSRoman Divacky /// WriteModule - Emit the specified module to the bitstream. 1772f22ef01cSRoman Divacky static void WriteModule(const Module *M, BitstreamWriter &Stream) { 1773f22ef01cSRoman Divacky Stream.EnterSubblock(bitc::MODULE_BLOCK_ID, 3); 1774f22ef01cSRoman Divacky 1775f22ef01cSRoman Divacky // Emit the version number if it is non-zero. 1776f22ef01cSRoman Divacky if (CurVersion) { 1777f22ef01cSRoman Divacky SmallVector<unsigned, 1> Vals; 1778f22ef01cSRoman Divacky Vals.push_back(CurVersion); 1779f22ef01cSRoman Divacky Stream.EmitRecord(bitc::MODULE_CODE_VERSION, Vals); 1780f22ef01cSRoman Divacky } 1781f22ef01cSRoman Divacky 1782f22ef01cSRoman Divacky // Analyze the module, enumerating globals, functions, etc. 1783f22ef01cSRoman Divacky ValueEnumerator VE(M); 1784f22ef01cSRoman Divacky 1785f22ef01cSRoman Divacky // Emit blockinfo, which defines the standard abbreviations etc. 1786f22ef01cSRoman Divacky WriteBlockInfo(VE, Stream); 1787f22ef01cSRoman Divacky 1788f22ef01cSRoman Divacky // Emit information about parameter attributes. 1789f22ef01cSRoman Divacky WriteAttributeTable(VE, Stream); 1790f22ef01cSRoman Divacky 1791f22ef01cSRoman Divacky // Emit information describing all of the types in the module. 1792f22ef01cSRoman Divacky WriteTypeTable(VE, Stream); 1793f22ef01cSRoman Divacky 1794f22ef01cSRoman Divacky // Emit top-level description of module, including target triple, inline asm, 1795f22ef01cSRoman Divacky // descriptors for global variables, and function prototype info. 1796f22ef01cSRoman Divacky WriteModuleInfo(M, VE, Stream); 1797f22ef01cSRoman Divacky 1798f22ef01cSRoman Divacky // Emit constants. 1799f22ef01cSRoman Divacky WriteModuleConstants(VE, Stream); 1800f22ef01cSRoman Divacky 1801f22ef01cSRoman Divacky // Emit metadata. 1802e580952dSDimitry Andric WriteModuleMetadata(M, VE, Stream); 1803f22ef01cSRoman Divacky 1804f22ef01cSRoman Divacky // Emit metadata. 1805f22ef01cSRoman Divacky WriteModuleMetadataStore(M, Stream); 1806f22ef01cSRoman Divacky 1807f22ef01cSRoman Divacky // Emit names for globals/functions etc. 1808f22ef01cSRoman Divacky WriteValueSymbolTable(M->getValueSymbolTable(), VE, Stream); 1809f22ef01cSRoman Divacky 1810dff0c46cSDimitry Andric // Emit use-lists. 1811dff0c46cSDimitry Andric if (EnablePreserveUseListOrdering) 1812dff0c46cSDimitry Andric WriteModuleUseLists(M, VE, Stream); 1813dff0c46cSDimitry Andric 1814dff0c46cSDimitry Andric // Emit function bodies. 1815dff0c46cSDimitry Andric for (Module::const_iterator F = M->begin(), E = M->end(); F != E; ++F) 1816dff0c46cSDimitry Andric if (!F->isDeclaration()) 1817dff0c46cSDimitry Andric WriteFunction(*F, VE, Stream); 1818dff0c46cSDimitry Andric 1819f22ef01cSRoman Divacky Stream.ExitBlock(); 1820f22ef01cSRoman Divacky } 1821f22ef01cSRoman Divacky 1822f22ef01cSRoman Divacky /// EmitDarwinBCHeader - If generating a bc file on darwin, we have to emit a 1823f22ef01cSRoman Divacky /// header and trailer to make it compatible with the system archiver. To do 1824f22ef01cSRoman Divacky /// this we emit the following header, and then emit a trailer that pads the 1825f22ef01cSRoman Divacky /// file out to be a multiple of 16 bytes. 1826f22ef01cSRoman Divacky /// 1827f22ef01cSRoman Divacky /// struct bc_header { 1828f22ef01cSRoman Divacky /// uint32_t Magic; // 0x0B17C0DE 1829f22ef01cSRoman Divacky /// uint32_t Version; // Version, currently always 0. 1830f22ef01cSRoman Divacky /// uint32_t BitcodeOffset; // Offset to traditional bitcode file. 1831f22ef01cSRoman Divacky /// uint32_t BitcodeSize; // Size of traditional bitcode file. 1832f22ef01cSRoman Divacky /// uint32_t CPUType; // CPU specifier. 1833f22ef01cSRoman Divacky /// ... potentially more later ... 1834f22ef01cSRoman Divacky /// }; 1835f22ef01cSRoman Divacky enum { 1836f22ef01cSRoman Divacky DarwinBCSizeFieldOffset = 3*4, // Offset to bitcode_size. 1837f22ef01cSRoman Divacky DarwinBCHeaderSize = 5*4 1838f22ef01cSRoman Divacky }; 1839f22ef01cSRoman Divacky 1840dff0c46cSDimitry Andric static void WriteInt32ToBuffer(uint32_t Value, SmallVectorImpl<char> &Buffer, 1841dff0c46cSDimitry Andric uint32_t &Position) { 1842dff0c46cSDimitry Andric Buffer[Position + 0] = (unsigned char) (Value >> 0); 1843dff0c46cSDimitry Andric Buffer[Position + 1] = (unsigned char) (Value >> 8); 1844dff0c46cSDimitry Andric Buffer[Position + 2] = (unsigned char) (Value >> 16); 1845dff0c46cSDimitry Andric Buffer[Position + 3] = (unsigned char) (Value >> 24); 1846dff0c46cSDimitry Andric Position += 4; 1847dff0c46cSDimitry Andric } 1848dff0c46cSDimitry Andric 1849dff0c46cSDimitry Andric static void EmitDarwinBCHeaderAndTrailer(SmallVectorImpl<char> &Buffer, 1850dff0c46cSDimitry Andric const Triple &TT) { 1851f22ef01cSRoman Divacky unsigned CPUType = ~0U; 1852f22ef01cSRoman Divacky 1853f22ef01cSRoman Divacky // Match x86_64-*, i[3-9]86-*, powerpc-*, powerpc64-*, arm-*, thumb-*, 1854f22ef01cSRoman Divacky // armv[0-9]-*, thumbv[0-9]-*, armv5te-*, or armv6t2-*. The CPUType is a magic 1855f22ef01cSRoman Divacky // number from /usr/include/mach/machine.h. It is ok to reproduce the 1856f22ef01cSRoman Divacky // specific constants here because they are implicitly part of the Darwin ABI. 1857f22ef01cSRoman Divacky enum { 1858f22ef01cSRoman Divacky DARWIN_CPU_ARCH_ABI64 = 0x01000000, 1859f22ef01cSRoman Divacky DARWIN_CPU_TYPE_X86 = 7, 1860f22ef01cSRoman Divacky DARWIN_CPU_TYPE_ARM = 12, 1861f22ef01cSRoman Divacky DARWIN_CPU_TYPE_POWERPC = 18 1862f22ef01cSRoman Divacky }; 1863f22ef01cSRoman Divacky 186417a519f9SDimitry Andric Triple::ArchType Arch = TT.getArch(); 186517a519f9SDimitry Andric if (Arch == Triple::x86_64) 1866f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_X86 | DARWIN_CPU_ARCH_ABI64; 186717a519f9SDimitry Andric else if (Arch == Triple::x86) 1868f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_X86; 186917a519f9SDimitry Andric else if (Arch == Triple::ppc) 1870f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_POWERPC; 187117a519f9SDimitry Andric else if (Arch == Triple::ppc64) 1872f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_POWERPC | DARWIN_CPU_ARCH_ABI64; 187317a519f9SDimitry Andric else if (Arch == Triple::arm || Arch == Triple::thumb) 1874f22ef01cSRoman Divacky CPUType = DARWIN_CPU_TYPE_ARM; 1875f22ef01cSRoman Divacky 1876f22ef01cSRoman Divacky // Traditional Bitcode starts after header. 1877dff0c46cSDimitry Andric assert(Buffer.size() >= DarwinBCHeaderSize && 1878dff0c46cSDimitry Andric "Expected header size to be reserved"); 1879f22ef01cSRoman Divacky unsigned BCOffset = DarwinBCHeaderSize; 1880dff0c46cSDimitry Andric unsigned BCSize = Buffer.size()-DarwinBCHeaderSize; 1881f22ef01cSRoman Divacky 1882dff0c46cSDimitry Andric // Write the magic and version. 1883dff0c46cSDimitry Andric unsigned Position = 0; 1884dff0c46cSDimitry Andric WriteInt32ToBuffer(0x0B17C0DE , Buffer, Position); 1885dff0c46cSDimitry Andric WriteInt32ToBuffer(0 , Buffer, Position); // Version. 1886dff0c46cSDimitry Andric WriteInt32ToBuffer(BCOffset , Buffer, Position); 1887dff0c46cSDimitry Andric WriteInt32ToBuffer(BCSize , Buffer, Position); 1888dff0c46cSDimitry Andric WriteInt32ToBuffer(CPUType , Buffer, Position); 1889f22ef01cSRoman Divacky 1890f22ef01cSRoman Divacky // If the file is not a multiple of 16 bytes, insert dummy padding. 1891dff0c46cSDimitry Andric while (Buffer.size() & 15) 1892dff0c46cSDimitry Andric Buffer.push_back(0); 1893f22ef01cSRoman Divacky } 1894f22ef01cSRoman Divacky 1895f22ef01cSRoman Divacky /// WriteBitcodeToFile - Write the specified module to the specified output 1896f22ef01cSRoman Divacky /// stream. 1897f22ef01cSRoman Divacky void llvm::WriteBitcodeToFile(const Module *M, raw_ostream &Out) { 1898dff0c46cSDimitry Andric SmallVector<char, 1024> Buffer; 1899f22ef01cSRoman Divacky Buffer.reserve(256*1024); 1900f22ef01cSRoman Divacky 1901dff0c46cSDimitry Andric // If this is darwin or another generic macho target, reserve space for the 1902dff0c46cSDimitry Andric // header. 190317a519f9SDimitry Andric Triple TT(M->getTargetTriple()); 190417a519f9SDimitry Andric if (TT.isOSDarwin()) 1905dff0c46cSDimitry Andric Buffer.insert(Buffer.begin(), DarwinBCHeaderSize, 0); 1906dff0c46cSDimitry Andric 1907dff0c46cSDimitry Andric // Emit the module into the buffer. 1908dff0c46cSDimitry Andric { 1909dff0c46cSDimitry Andric BitstreamWriter Stream(Buffer); 1910f22ef01cSRoman Divacky 1911f22ef01cSRoman Divacky // Emit the file header. 1912f22ef01cSRoman Divacky Stream.Emit((unsigned)'B', 8); 1913f22ef01cSRoman Divacky Stream.Emit((unsigned)'C', 8); 1914f22ef01cSRoman Divacky Stream.Emit(0x0, 4); 1915f22ef01cSRoman Divacky Stream.Emit(0xC, 4); 1916f22ef01cSRoman Divacky Stream.Emit(0xE, 4); 1917f22ef01cSRoman Divacky Stream.Emit(0xD, 4); 1918f22ef01cSRoman Divacky 1919f22ef01cSRoman Divacky // Emit the module. 1920f22ef01cSRoman Divacky WriteModule(M, Stream); 1921dff0c46cSDimitry Andric } 1922f22ef01cSRoman Divacky 192317a519f9SDimitry Andric if (TT.isOSDarwin()) 1924dff0c46cSDimitry Andric EmitDarwinBCHeaderAndTrailer(Buffer, TT); 1925dff0c46cSDimitry Andric 1926dff0c46cSDimitry Andric // Write the generated bitstream to "Out". 1927dff0c46cSDimitry Andric Out.write((char*)&Buffer.front(), Buffer.size()); 1928f22ef01cSRoman Divacky } 1929