1f22ef01cSRoman Divacky //===- BitcodeReader.cpp - Internal BitcodeReader implementation ----------===//
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 #include "llvm/Bitcode/ReaderWriter.h"
11f22ef01cSRoman Divacky #include "BitcodeReader.h"
12f22ef01cSRoman Divacky #include "llvm/ADT/SmallString.h"
13f22ef01cSRoman Divacky #include "llvm/ADT/SmallVector.h"
14f785676fSDimitry Andric #include "llvm/Bitcode/LLVMBitCodes.h"
1591bc56edSDimitry Andric #include "llvm/IR/AutoUpgrade.h"
16139f7f9bSDimitry Andric #include "llvm/IR/Constants.h"
17139f7f9bSDimitry Andric #include "llvm/IR/DerivedTypes.h"
1839d628a0SDimitry Andric #include "llvm/IR/DiagnosticPrinter.h"
19139f7f9bSDimitry Andric #include "llvm/IR/InlineAsm.h"
20139f7f9bSDimitry Andric #include "llvm/IR/IntrinsicInst.h"
21f785676fSDimitry Andric #include "llvm/IR/LLVMContext.h"
22139f7f9bSDimitry Andric #include "llvm/IR/Module.h"
23139f7f9bSDimitry Andric #include "llvm/IR/OperandTraits.h"
24139f7f9bSDimitry Andric #include "llvm/IR/Operator.h"
25dff0c46cSDimitry Andric #include "llvm/Support/DataStream.h"
2639d628a0SDimitry Andric #include "llvm/Support/ManagedStatic.h"
27f22ef01cSRoman Divacky #include "llvm/Support/MathExtras.h"
28f22ef01cSRoman Divacky #include "llvm/Support/MemoryBuffer.h"
29f785676fSDimitry Andric #include "llvm/Support/raw_ostream.h"
3039d628a0SDimitry Andric 
31f22ef01cSRoman Divacky using namespace llvm;
32f22ef01cSRoman Divacky 
337ae0e2c9SDimitry Andric enum {
347ae0e2c9SDimitry Andric   SWITCH_INST_MAGIC = 0x4B5 // May 2012 => 1205 => Hex
357ae0e2c9SDimitry Andric };
367ae0e2c9SDimitry Andric 
3739d628a0SDimitry Andric BitcodeDiagnosticInfo::BitcodeDiagnosticInfo(std::error_code EC,
3839d628a0SDimitry Andric                                              DiagnosticSeverity Severity,
3939d628a0SDimitry Andric                                              const Twine &Msg)
4039d628a0SDimitry Andric     : DiagnosticInfo(DK_Bitcode, Severity), Msg(Msg), EC(EC) {}
4139d628a0SDimitry Andric 
4239d628a0SDimitry Andric void BitcodeDiagnosticInfo::print(DiagnosticPrinter &DP) const { DP << Msg; }
4339d628a0SDimitry Andric 
4439d628a0SDimitry Andric static std::error_code Error(DiagnosticHandlerFunction DiagnosticHandler,
4539d628a0SDimitry Andric                              std::error_code EC, const Twine &Message) {
4639d628a0SDimitry Andric   BitcodeDiagnosticInfo DI(EC, DS_Error, Message);
4739d628a0SDimitry Andric   DiagnosticHandler(DI);
4839d628a0SDimitry Andric   return EC;
49dff0c46cSDimitry Andric }
5039d628a0SDimitry Andric 
5139d628a0SDimitry Andric static std::error_code Error(DiagnosticHandlerFunction DiagnosticHandler,
5239d628a0SDimitry Andric                              std::error_code EC) {
5339d628a0SDimitry Andric   return Error(DiagnosticHandler, EC, EC.message());
5439d628a0SDimitry Andric }
5539d628a0SDimitry Andric 
5639d628a0SDimitry Andric std::error_code BitcodeReader::Error(BitcodeError E, const Twine &Message) {
5739d628a0SDimitry Andric   return ::Error(DiagnosticHandler, make_error_code(E), Message);
5839d628a0SDimitry Andric }
5939d628a0SDimitry Andric 
6039d628a0SDimitry Andric std::error_code BitcodeReader::Error(const Twine &Message) {
6139d628a0SDimitry Andric   return ::Error(DiagnosticHandler,
6239d628a0SDimitry Andric                  make_error_code(BitcodeError::CorruptedBitcode), Message);
6339d628a0SDimitry Andric }
6439d628a0SDimitry Andric 
6539d628a0SDimitry Andric std::error_code BitcodeReader::Error(BitcodeError E) {
6639d628a0SDimitry Andric   return ::Error(DiagnosticHandler, make_error_code(E));
6739d628a0SDimitry Andric }
6839d628a0SDimitry Andric 
6939d628a0SDimitry Andric static DiagnosticHandlerFunction getDiagHandler(DiagnosticHandlerFunction F,
7039d628a0SDimitry Andric                                                 LLVMContext &C) {
7139d628a0SDimitry Andric   if (F)
7239d628a0SDimitry Andric     return F;
7339d628a0SDimitry Andric   return [&C](const DiagnosticInfo &DI) { C.diagnose(DI); };
7439d628a0SDimitry Andric }
7539d628a0SDimitry Andric 
7639d628a0SDimitry Andric BitcodeReader::BitcodeReader(MemoryBuffer *buffer, LLVMContext &C,
7739d628a0SDimitry Andric                              DiagnosticHandlerFunction DiagnosticHandler)
7839d628a0SDimitry Andric     : Context(C), DiagnosticHandler(getDiagHandler(DiagnosticHandler, C)),
7939d628a0SDimitry Andric       TheModule(nullptr), Buffer(buffer), LazyStreamer(nullptr),
8039d628a0SDimitry Andric       NextUnreadBit(0), SeenValueSymbolTable(false), ValueList(C),
8139d628a0SDimitry Andric       MDValueList(C), SeenFirstFunctionBody(false), UseRelativeIDs(false),
8239d628a0SDimitry Andric       WillMaterializeAllForwardRefs(false) {}
8339d628a0SDimitry Andric 
8439d628a0SDimitry Andric BitcodeReader::BitcodeReader(DataStreamer *streamer, LLVMContext &C,
8539d628a0SDimitry Andric                              DiagnosticHandlerFunction DiagnosticHandler)
8639d628a0SDimitry Andric     : Context(C), DiagnosticHandler(getDiagHandler(DiagnosticHandler, C)),
8739d628a0SDimitry Andric       TheModule(nullptr), Buffer(nullptr), LazyStreamer(streamer),
8839d628a0SDimitry Andric       NextUnreadBit(0), SeenValueSymbolTable(false), ValueList(C),
8939d628a0SDimitry Andric       MDValueList(C), SeenFirstFunctionBody(false), UseRelativeIDs(false),
9039d628a0SDimitry Andric       WillMaterializeAllForwardRefs(false) {}
9139d628a0SDimitry Andric 
9239d628a0SDimitry Andric std::error_code BitcodeReader::materializeForwardReferencedFunctions() {
9339d628a0SDimitry Andric   if (WillMaterializeAllForwardRefs)
9439d628a0SDimitry Andric     return std::error_code();
9539d628a0SDimitry Andric 
9639d628a0SDimitry Andric   // Prevent recursion.
9739d628a0SDimitry Andric   WillMaterializeAllForwardRefs = true;
9839d628a0SDimitry Andric 
9939d628a0SDimitry Andric   while (!BasicBlockFwdRefQueue.empty()) {
10039d628a0SDimitry Andric     Function *F = BasicBlockFwdRefQueue.front();
10139d628a0SDimitry Andric     BasicBlockFwdRefQueue.pop_front();
10239d628a0SDimitry Andric     assert(F && "Expected valid function");
10339d628a0SDimitry Andric     if (!BasicBlockFwdRefs.count(F))
10439d628a0SDimitry Andric       // Already materialized.
10539d628a0SDimitry Andric       continue;
10639d628a0SDimitry Andric 
10739d628a0SDimitry Andric     // Check for a function that isn't materializable to prevent an infinite
10839d628a0SDimitry Andric     // loop.  When parsing a blockaddress stored in a global variable, there
10939d628a0SDimitry Andric     // isn't a trivial way to check if a function will have a body without a
11039d628a0SDimitry Andric     // linear search through FunctionsWithBodies, so just check it here.
11139d628a0SDimitry Andric     if (!F->isMaterializable())
11239d628a0SDimitry Andric       return Error("Never resolved function from blockaddress");
11339d628a0SDimitry Andric 
11439d628a0SDimitry Andric     // Try to materialize F.
11539d628a0SDimitry Andric     if (std::error_code EC = materialize(F))
11639d628a0SDimitry Andric       return EC;
11739d628a0SDimitry Andric   }
11839d628a0SDimitry Andric   assert(BasicBlockFwdRefs.empty() && "Function missing from queue");
11939d628a0SDimitry Andric 
12039d628a0SDimitry Andric   // Reset state.
12139d628a0SDimitry Andric   WillMaterializeAllForwardRefs = false;
12239d628a0SDimitry Andric   return std::error_code();
123dff0c46cSDimitry Andric }
124dff0c46cSDimitry Andric 
125f22ef01cSRoman Divacky void BitcodeReader::FreeState() {
12691bc56edSDimitry Andric   Buffer = nullptr;
12717a519f9SDimitry Andric   std::vector<Type*>().swap(TypeList);
128f22ef01cSRoman Divacky   ValueList.clear();
129f22ef01cSRoman Divacky   MDValueList.clear();
13091bc56edSDimitry Andric   std::vector<Comdat *>().swap(ComdatList);
131f22ef01cSRoman Divacky 
132139f7f9bSDimitry Andric   std::vector<AttributeSet>().swap(MAttributes);
133f22ef01cSRoman Divacky   std::vector<BasicBlock*>().swap(FunctionBBs);
134f22ef01cSRoman Divacky   std::vector<Function*>().swap(FunctionsWithBodies);
135f22ef01cSRoman Divacky   DeferredFunctionInfo.clear();
136e580952dSDimitry Andric   MDKindMap.clear();
1373861d79fSDimitry Andric 
13839d628a0SDimitry Andric   assert(BasicBlockFwdRefs.empty() && "Unresolved blockaddress fwd references");
13939d628a0SDimitry Andric   BasicBlockFwdRefQueue.clear();
140f22ef01cSRoman Divacky }
141f22ef01cSRoman Divacky 
142f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
143f22ef01cSRoman Divacky //  Helper functions to implement forward reference resolution, etc.
144f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
145f22ef01cSRoman Divacky 
146f22ef01cSRoman Divacky /// ConvertToString - Convert a string from a record into an std::string, return
147f22ef01cSRoman Divacky /// true on failure.
148f22ef01cSRoman Divacky template<typename StrTy>
1497ae0e2c9SDimitry Andric static bool ConvertToString(ArrayRef<uint64_t> Record, unsigned Idx,
150f22ef01cSRoman Divacky                             StrTy &Result) {
151f22ef01cSRoman Divacky   if (Idx > Record.size())
152f22ef01cSRoman Divacky     return true;
153f22ef01cSRoman Divacky 
154f22ef01cSRoman Divacky   for (unsigned i = Idx, e = Record.size(); i != e; ++i)
155f22ef01cSRoman Divacky     Result += (char)Record[i];
156f22ef01cSRoman Divacky   return false;
157f22ef01cSRoman Divacky }
158f22ef01cSRoman Divacky 
15939d628a0SDimitry Andric static GlobalValue::LinkageTypes getDecodedLinkage(unsigned Val) {
160f22ef01cSRoman Divacky   switch (Val) {
161f22ef01cSRoman Divacky   default: // Map unknown/new linkages to external
16239d628a0SDimitry Andric   case 0:
16339d628a0SDimitry Andric     return GlobalValue::ExternalLinkage;
16439d628a0SDimitry Andric   case 1:
16539d628a0SDimitry Andric     return GlobalValue::WeakAnyLinkage;
16639d628a0SDimitry Andric   case 2:
16739d628a0SDimitry Andric     return GlobalValue::AppendingLinkage;
16839d628a0SDimitry Andric   case 3:
16939d628a0SDimitry Andric     return GlobalValue::InternalLinkage;
17039d628a0SDimitry Andric   case 4:
17139d628a0SDimitry Andric     return GlobalValue::LinkOnceAnyLinkage;
17239d628a0SDimitry Andric   case 5:
17339d628a0SDimitry Andric     return GlobalValue::ExternalLinkage; // Obsolete DLLImportLinkage
17439d628a0SDimitry Andric   case 6:
17539d628a0SDimitry Andric     return GlobalValue::ExternalLinkage; // Obsolete DLLExportLinkage
17639d628a0SDimitry Andric   case 7:
17739d628a0SDimitry Andric     return GlobalValue::ExternalWeakLinkage;
17839d628a0SDimitry Andric   case 8:
17939d628a0SDimitry Andric     return GlobalValue::CommonLinkage;
18039d628a0SDimitry Andric   case 9:
18139d628a0SDimitry Andric     return GlobalValue::PrivateLinkage;
18239d628a0SDimitry Andric   case 10:
18339d628a0SDimitry Andric     return GlobalValue::WeakODRLinkage;
18439d628a0SDimitry Andric   case 11:
18539d628a0SDimitry Andric     return GlobalValue::LinkOnceODRLinkage;
18639d628a0SDimitry Andric   case 12:
18739d628a0SDimitry Andric     return GlobalValue::AvailableExternallyLinkage;
18891bc56edSDimitry Andric   case 13:
18991bc56edSDimitry Andric     return GlobalValue::PrivateLinkage; // Obsolete LinkerPrivateLinkage
19091bc56edSDimitry Andric   case 14:
19191bc56edSDimitry Andric     return GlobalValue::PrivateLinkage; // Obsolete LinkerPrivateWeakLinkage
19239d628a0SDimitry Andric   case 15:
19339d628a0SDimitry Andric     return GlobalValue::ExternalLinkage; // Obsolete LinkOnceODRAutoHideLinkage
194f22ef01cSRoman Divacky   }
195f22ef01cSRoman Divacky }
196f22ef01cSRoman Divacky 
197f22ef01cSRoman Divacky static GlobalValue::VisibilityTypes GetDecodedVisibility(unsigned Val) {
198f22ef01cSRoman Divacky   switch (Val) {
199f22ef01cSRoman Divacky   default: // Map unknown visibilities to default.
200f22ef01cSRoman Divacky   case 0: return GlobalValue::DefaultVisibility;
201f22ef01cSRoman Divacky   case 1: return GlobalValue::HiddenVisibility;
202f22ef01cSRoman Divacky   case 2: return GlobalValue::ProtectedVisibility;
203f22ef01cSRoman Divacky   }
204f22ef01cSRoman Divacky }
205f22ef01cSRoman Divacky 
20691bc56edSDimitry Andric static GlobalValue::DLLStorageClassTypes
20791bc56edSDimitry Andric GetDecodedDLLStorageClass(unsigned Val) {
20891bc56edSDimitry Andric   switch (Val) {
20991bc56edSDimitry Andric   default: // Map unknown values to default.
21091bc56edSDimitry Andric   case 0: return GlobalValue::DefaultStorageClass;
21191bc56edSDimitry Andric   case 1: return GlobalValue::DLLImportStorageClass;
21291bc56edSDimitry Andric   case 2: return GlobalValue::DLLExportStorageClass;
21391bc56edSDimitry Andric   }
21491bc56edSDimitry Andric }
21591bc56edSDimitry Andric 
2167ae0e2c9SDimitry Andric static GlobalVariable::ThreadLocalMode GetDecodedThreadLocalMode(unsigned Val) {
2177ae0e2c9SDimitry Andric   switch (Val) {
2187ae0e2c9SDimitry Andric     case 0: return GlobalVariable::NotThreadLocal;
2197ae0e2c9SDimitry Andric     default: // Map unknown non-zero value to general dynamic.
2207ae0e2c9SDimitry Andric     case 1: return GlobalVariable::GeneralDynamicTLSModel;
2217ae0e2c9SDimitry Andric     case 2: return GlobalVariable::LocalDynamicTLSModel;
2227ae0e2c9SDimitry Andric     case 3: return GlobalVariable::InitialExecTLSModel;
2237ae0e2c9SDimitry Andric     case 4: return GlobalVariable::LocalExecTLSModel;
2247ae0e2c9SDimitry Andric   }
2257ae0e2c9SDimitry Andric }
2267ae0e2c9SDimitry Andric 
227f22ef01cSRoman Divacky static int GetDecodedCastOpcode(unsigned Val) {
228f22ef01cSRoman Divacky   switch (Val) {
229f22ef01cSRoman Divacky   default: return -1;
230f22ef01cSRoman Divacky   case bitc::CAST_TRUNC   : return Instruction::Trunc;
231f22ef01cSRoman Divacky   case bitc::CAST_ZEXT    : return Instruction::ZExt;
232f22ef01cSRoman Divacky   case bitc::CAST_SEXT    : return Instruction::SExt;
233f22ef01cSRoman Divacky   case bitc::CAST_FPTOUI  : return Instruction::FPToUI;
234f22ef01cSRoman Divacky   case bitc::CAST_FPTOSI  : return Instruction::FPToSI;
235f22ef01cSRoman Divacky   case bitc::CAST_UITOFP  : return Instruction::UIToFP;
236f22ef01cSRoman Divacky   case bitc::CAST_SITOFP  : return Instruction::SIToFP;
237f22ef01cSRoman Divacky   case bitc::CAST_FPTRUNC : return Instruction::FPTrunc;
238f22ef01cSRoman Divacky   case bitc::CAST_FPEXT   : return Instruction::FPExt;
239f22ef01cSRoman Divacky   case bitc::CAST_PTRTOINT: return Instruction::PtrToInt;
240f22ef01cSRoman Divacky   case bitc::CAST_INTTOPTR: return Instruction::IntToPtr;
241f22ef01cSRoman Divacky   case bitc::CAST_BITCAST : return Instruction::BitCast;
242f785676fSDimitry Andric   case bitc::CAST_ADDRSPACECAST: return Instruction::AddrSpaceCast;
243f22ef01cSRoman Divacky   }
244f22ef01cSRoman Divacky }
2456122f3e6SDimitry Andric static int GetDecodedBinaryOpcode(unsigned Val, Type *Ty) {
246f22ef01cSRoman Divacky   switch (Val) {
247f22ef01cSRoman Divacky   default: return -1;
248f22ef01cSRoman Divacky   case bitc::BINOP_ADD:
249f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FAdd : Instruction::Add;
250f22ef01cSRoman Divacky   case bitc::BINOP_SUB:
251f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FSub : Instruction::Sub;
252f22ef01cSRoman Divacky   case bitc::BINOP_MUL:
253f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FMul : Instruction::Mul;
254f22ef01cSRoman Divacky   case bitc::BINOP_UDIV: return Instruction::UDiv;
255f22ef01cSRoman Divacky   case bitc::BINOP_SDIV:
256f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FDiv : Instruction::SDiv;
257f22ef01cSRoman Divacky   case bitc::BINOP_UREM: return Instruction::URem;
258f22ef01cSRoman Divacky   case bitc::BINOP_SREM:
259f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FRem : Instruction::SRem;
260f22ef01cSRoman Divacky   case bitc::BINOP_SHL:  return Instruction::Shl;
261f22ef01cSRoman Divacky   case bitc::BINOP_LSHR: return Instruction::LShr;
262f22ef01cSRoman Divacky   case bitc::BINOP_ASHR: return Instruction::AShr;
263f22ef01cSRoman Divacky   case bitc::BINOP_AND:  return Instruction::And;
264f22ef01cSRoman Divacky   case bitc::BINOP_OR:   return Instruction::Or;
265f22ef01cSRoman Divacky   case bitc::BINOP_XOR:  return Instruction::Xor;
266f22ef01cSRoman Divacky   }
267f22ef01cSRoman Divacky }
268f22ef01cSRoman Divacky 
2696122f3e6SDimitry Andric static AtomicRMWInst::BinOp GetDecodedRMWOperation(unsigned Val) {
2706122f3e6SDimitry Andric   switch (Val) {
2716122f3e6SDimitry Andric   default: return AtomicRMWInst::BAD_BINOP;
2726122f3e6SDimitry Andric   case bitc::RMW_XCHG: return AtomicRMWInst::Xchg;
2736122f3e6SDimitry Andric   case bitc::RMW_ADD: return AtomicRMWInst::Add;
2746122f3e6SDimitry Andric   case bitc::RMW_SUB: return AtomicRMWInst::Sub;
2756122f3e6SDimitry Andric   case bitc::RMW_AND: return AtomicRMWInst::And;
2766122f3e6SDimitry Andric   case bitc::RMW_NAND: return AtomicRMWInst::Nand;
2776122f3e6SDimitry Andric   case bitc::RMW_OR: return AtomicRMWInst::Or;
2786122f3e6SDimitry Andric   case bitc::RMW_XOR: return AtomicRMWInst::Xor;
2796122f3e6SDimitry Andric   case bitc::RMW_MAX: return AtomicRMWInst::Max;
2806122f3e6SDimitry Andric   case bitc::RMW_MIN: return AtomicRMWInst::Min;
2816122f3e6SDimitry Andric   case bitc::RMW_UMAX: return AtomicRMWInst::UMax;
2826122f3e6SDimitry Andric   case bitc::RMW_UMIN: return AtomicRMWInst::UMin;
2836122f3e6SDimitry Andric   }
2846122f3e6SDimitry Andric }
2856122f3e6SDimitry Andric 
2866122f3e6SDimitry Andric static AtomicOrdering GetDecodedOrdering(unsigned Val) {
2876122f3e6SDimitry Andric   switch (Val) {
2886122f3e6SDimitry Andric   case bitc::ORDERING_NOTATOMIC: return NotAtomic;
2896122f3e6SDimitry Andric   case bitc::ORDERING_UNORDERED: return Unordered;
2906122f3e6SDimitry Andric   case bitc::ORDERING_MONOTONIC: return Monotonic;
2916122f3e6SDimitry Andric   case bitc::ORDERING_ACQUIRE: return Acquire;
2926122f3e6SDimitry Andric   case bitc::ORDERING_RELEASE: return Release;
2936122f3e6SDimitry Andric   case bitc::ORDERING_ACQREL: return AcquireRelease;
2946122f3e6SDimitry Andric   default: // Map unknown orderings to sequentially-consistent.
2956122f3e6SDimitry Andric   case bitc::ORDERING_SEQCST: return SequentiallyConsistent;
2966122f3e6SDimitry Andric   }
2976122f3e6SDimitry Andric }
2986122f3e6SDimitry Andric 
2996122f3e6SDimitry Andric static SynchronizationScope GetDecodedSynchScope(unsigned Val) {
3006122f3e6SDimitry Andric   switch (Val) {
3016122f3e6SDimitry Andric   case bitc::SYNCHSCOPE_SINGLETHREAD: return SingleThread;
3026122f3e6SDimitry Andric   default: // Map unknown scopes to cross-thread.
3036122f3e6SDimitry Andric   case bitc::SYNCHSCOPE_CROSSTHREAD: return CrossThread;
3046122f3e6SDimitry Andric   }
3056122f3e6SDimitry Andric }
3066122f3e6SDimitry Andric 
30791bc56edSDimitry Andric static Comdat::SelectionKind getDecodedComdatSelectionKind(unsigned Val) {
30891bc56edSDimitry Andric   switch (Val) {
30991bc56edSDimitry Andric   default: // Map unknown selection kinds to any.
31091bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_ANY:
31191bc56edSDimitry Andric     return Comdat::Any;
31291bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_EXACT_MATCH:
31391bc56edSDimitry Andric     return Comdat::ExactMatch;
31491bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_LARGEST:
31591bc56edSDimitry Andric     return Comdat::Largest;
31691bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_NO_DUPLICATES:
31791bc56edSDimitry Andric     return Comdat::NoDuplicates;
31891bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_SAME_SIZE:
31991bc56edSDimitry Andric     return Comdat::SameSize;
32091bc56edSDimitry Andric   }
32191bc56edSDimitry Andric }
32291bc56edSDimitry Andric 
32391bc56edSDimitry Andric static void UpgradeDLLImportExportLinkage(llvm::GlobalValue *GV, unsigned Val) {
32491bc56edSDimitry Andric   switch (Val) {
32591bc56edSDimitry Andric   case 5: GV->setDLLStorageClass(GlobalValue::DLLImportStorageClass); break;
32691bc56edSDimitry Andric   case 6: GV->setDLLStorageClass(GlobalValue::DLLExportStorageClass); break;
32791bc56edSDimitry Andric   }
32891bc56edSDimitry Andric }
32991bc56edSDimitry Andric 
330f22ef01cSRoman Divacky namespace llvm {
331f22ef01cSRoman Divacky namespace {
332f22ef01cSRoman Divacky   /// @brief A class for maintaining the slot number definition
333f22ef01cSRoman Divacky   /// as a placeholder for the actual definition for forward constants defs.
334f22ef01cSRoman Divacky   class ConstantPlaceHolder : public ConstantExpr {
3353861d79fSDimitry Andric     void operator=(const ConstantPlaceHolder &) LLVM_DELETED_FUNCTION;
336f22ef01cSRoman Divacky   public:
337f22ef01cSRoman Divacky     // allocate space for exactly one operand
338f22ef01cSRoman Divacky     void *operator new(size_t s) {
339f22ef01cSRoman Divacky       return User::operator new(s, 1);
340f22ef01cSRoman Divacky     }
3416122f3e6SDimitry Andric     explicit ConstantPlaceHolder(Type *Ty, LLVMContext& Context)
342f22ef01cSRoman Divacky       : ConstantExpr(Ty, Instruction::UserOp1, &Op<0>(), 1) {
343f22ef01cSRoman Divacky       Op<0>() = UndefValue::get(Type::getInt32Ty(Context));
344f22ef01cSRoman Divacky     }
345f22ef01cSRoman Divacky 
346f22ef01cSRoman Divacky     /// @brief Methods to support type inquiry through isa, cast, and dyn_cast.
347f22ef01cSRoman Divacky     static bool classof(const Value *V) {
348f22ef01cSRoman Divacky       return isa<ConstantExpr>(V) &&
349f22ef01cSRoman Divacky              cast<ConstantExpr>(V)->getOpcode() == Instruction::UserOp1;
350f22ef01cSRoman Divacky     }
351f22ef01cSRoman Divacky 
352f22ef01cSRoman Divacky 
353f22ef01cSRoman Divacky     /// Provide fast operand accessors
35439d628a0SDimitry Andric     DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
355f22ef01cSRoman Divacky   };
356f22ef01cSRoman Divacky }
357f22ef01cSRoman Divacky 
358f22ef01cSRoman Divacky // FIXME: can we inherit this from ConstantExpr?
359f22ef01cSRoman Divacky template <>
3602754fe60SDimitry Andric struct OperandTraits<ConstantPlaceHolder> :
3612754fe60SDimitry Andric   public FixedNumOperandTraits<ConstantPlaceHolder, 1> {
362f22ef01cSRoman Divacky };
36339d628a0SDimitry Andric DEFINE_TRANSPARENT_OPERAND_ACCESSORS(ConstantPlaceHolder, Value)
364f22ef01cSRoman Divacky }
365f22ef01cSRoman Divacky 
366f22ef01cSRoman Divacky 
367f22ef01cSRoman Divacky void BitcodeReaderValueList::AssignValue(Value *V, unsigned Idx) {
368f22ef01cSRoman Divacky   if (Idx == size()) {
369f22ef01cSRoman Divacky     push_back(V);
370f22ef01cSRoman Divacky     return;
371f22ef01cSRoman Divacky   }
372f22ef01cSRoman Divacky 
373f22ef01cSRoman Divacky   if (Idx >= size())
374f22ef01cSRoman Divacky     resize(Idx+1);
375f22ef01cSRoman Divacky 
376f22ef01cSRoman Divacky   WeakVH &OldV = ValuePtrs[Idx];
37791bc56edSDimitry Andric   if (!OldV) {
378f22ef01cSRoman Divacky     OldV = V;
379f22ef01cSRoman Divacky     return;
380f22ef01cSRoman Divacky   }
381f22ef01cSRoman Divacky 
382f22ef01cSRoman Divacky   // Handle constants and non-constants (e.g. instrs) differently for
383f22ef01cSRoman Divacky   // efficiency.
384f22ef01cSRoman Divacky   if (Constant *PHC = dyn_cast<Constant>(&*OldV)) {
385f22ef01cSRoman Divacky     ResolveConstants.push_back(std::make_pair(PHC, Idx));
386f22ef01cSRoman Divacky     OldV = V;
387f22ef01cSRoman Divacky   } else {
388f22ef01cSRoman Divacky     // If there was a forward reference to this value, replace it.
389f22ef01cSRoman Divacky     Value *PrevVal = OldV;
390f22ef01cSRoman Divacky     OldV->replaceAllUsesWith(V);
391f22ef01cSRoman Divacky     delete PrevVal;
392f22ef01cSRoman Divacky   }
393f22ef01cSRoman Divacky }
394f22ef01cSRoman Divacky 
395f22ef01cSRoman Divacky 
396f22ef01cSRoman Divacky Constant *BitcodeReaderValueList::getConstantFwdRef(unsigned Idx,
3976122f3e6SDimitry Andric                                                     Type *Ty) {
398f22ef01cSRoman Divacky   if (Idx >= size())
399f22ef01cSRoman Divacky     resize(Idx + 1);
400f22ef01cSRoman Divacky 
401f22ef01cSRoman Divacky   if (Value *V = ValuePtrs[Idx]) {
402f22ef01cSRoman Divacky     assert(Ty == V->getType() && "Type mismatch in constant table!");
403f22ef01cSRoman Divacky     return cast<Constant>(V);
404f22ef01cSRoman Divacky   }
405f22ef01cSRoman Divacky 
406f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
407f22ef01cSRoman Divacky   Constant *C = new ConstantPlaceHolder(Ty, Context);
408f22ef01cSRoman Divacky   ValuePtrs[Idx] = C;
409f22ef01cSRoman Divacky   return C;
410f22ef01cSRoman Divacky }
411f22ef01cSRoman Divacky 
4126122f3e6SDimitry Andric Value *BitcodeReaderValueList::getValueFwdRef(unsigned Idx, Type *Ty) {
413f22ef01cSRoman Divacky   if (Idx >= size())
414f22ef01cSRoman Divacky     resize(Idx + 1);
415f22ef01cSRoman Divacky 
416f22ef01cSRoman Divacky   if (Value *V = ValuePtrs[Idx]) {
41791bc56edSDimitry Andric     assert((!Ty || Ty == V->getType()) && "Type mismatch in value table!");
418f22ef01cSRoman Divacky     return V;
419f22ef01cSRoman Divacky   }
420f22ef01cSRoman Divacky 
421f22ef01cSRoman Divacky   // No type specified, must be invalid reference.
42291bc56edSDimitry Andric   if (!Ty) return nullptr;
423f22ef01cSRoman Divacky 
424f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
425f22ef01cSRoman Divacky   Value *V = new Argument(Ty);
426f22ef01cSRoman Divacky   ValuePtrs[Idx] = V;
427f22ef01cSRoman Divacky   return V;
428f22ef01cSRoman Divacky }
429f22ef01cSRoman Divacky 
430f22ef01cSRoman Divacky /// ResolveConstantForwardRefs - Once all constants are read, this method bulk
431f22ef01cSRoman Divacky /// resolves any forward references.  The idea behind this is that we sometimes
432f22ef01cSRoman Divacky /// get constants (such as large arrays) which reference *many* forward ref
433f22ef01cSRoman Divacky /// constants.  Replacing each of these causes a lot of thrashing when
434f22ef01cSRoman Divacky /// building/reuniquing the constant.  Instead of doing this, we look at all the
435f22ef01cSRoman Divacky /// uses and rewrite all the place holders at once for any constant that uses
436f22ef01cSRoman Divacky /// a placeholder.
437f22ef01cSRoman Divacky void BitcodeReaderValueList::ResolveConstantForwardRefs() {
438f22ef01cSRoman Divacky   // Sort the values by-pointer so that they are efficient to look up with a
439f22ef01cSRoman Divacky   // binary search.
440f22ef01cSRoman Divacky   std::sort(ResolveConstants.begin(), ResolveConstants.end());
441f22ef01cSRoman Divacky 
442f22ef01cSRoman Divacky   SmallVector<Constant*, 64> NewOps;
443f22ef01cSRoman Divacky 
444f22ef01cSRoman Divacky   while (!ResolveConstants.empty()) {
445f22ef01cSRoman Divacky     Value *RealVal = operator[](ResolveConstants.back().second);
446f22ef01cSRoman Divacky     Constant *Placeholder = ResolveConstants.back().first;
447f22ef01cSRoman Divacky     ResolveConstants.pop_back();
448f22ef01cSRoman Divacky 
449f22ef01cSRoman Divacky     // Loop over all users of the placeholder, updating them to reference the
450f22ef01cSRoman Divacky     // new value.  If they reference more than one placeholder, update them all
451f22ef01cSRoman Divacky     // at once.
452f22ef01cSRoman Divacky     while (!Placeholder->use_empty()) {
45391bc56edSDimitry Andric       auto UI = Placeholder->user_begin();
454ffd1746dSEd Schouten       User *U = *UI;
455f22ef01cSRoman Divacky 
456f22ef01cSRoman Divacky       // If the using object isn't uniqued, just update the operands.  This
457f22ef01cSRoman Divacky       // handles instructions and initializers for global variables.
458ffd1746dSEd Schouten       if (!isa<Constant>(U) || isa<GlobalValue>(U)) {
459f22ef01cSRoman Divacky         UI.getUse().set(RealVal);
460f22ef01cSRoman Divacky         continue;
461f22ef01cSRoman Divacky       }
462f22ef01cSRoman Divacky 
463f22ef01cSRoman Divacky       // Otherwise, we have a constant that uses the placeholder.  Replace that
464f22ef01cSRoman Divacky       // constant with a new constant that has *all* placeholder uses updated.
465ffd1746dSEd Schouten       Constant *UserC = cast<Constant>(U);
466f22ef01cSRoman Divacky       for (User::op_iterator I = UserC->op_begin(), E = UserC->op_end();
467f22ef01cSRoman Divacky            I != E; ++I) {
468f22ef01cSRoman Divacky         Value *NewOp;
469f22ef01cSRoman Divacky         if (!isa<ConstantPlaceHolder>(*I)) {
470f22ef01cSRoman Divacky           // Not a placeholder reference.
471f22ef01cSRoman Divacky           NewOp = *I;
472f22ef01cSRoman Divacky         } else if (*I == Placeholder) {
473f22ef01cSRoman Divacky           // Common case is that it just references this one placeholder.
474f22ef01cSRoman Divacky           NewOp = RealVal;
475f22ef01cSRoman Divacky         } else {
476f22ef01cSRoman Divacky           // Otherwise, look up the placeholder in ResolveConstants.
477f22ef01cSRoman Divacky           ResolveConstantsTy::iterator It =
478f22ef01cSRoman Divacky             std::lower_bound(ResolveConstants.begin(), ResolveConstants.end(),
479f22ef01cSRoman Divacky                              std::pair<Constant*, unsigned>(cast<Constant>(*I),
480f22ef01cSRoman Divacky                                                             0));
481f22ef01cSRoman Divacky           assert(It != ResolveConstants.end() && It->first == *I);
482f22ef01cSRoman Divacky           NewOp = operator[](It->second);
483f22ef01cSRoman Divacky         }
484f22ef01cSRoman Divacky 
485f22ef01cSRoman Divacky         NewOps.push_back(cast<Constant>(NewOp));
486f22ef01cSRoman Divacky       }
487f22ef01cSRoman Divacky 
488f22ef01cSRoman Divacky       // Make the new constant.
489f22ef01cSRoman Divacky       Constant *NewC;
490f22ef01cSRoman Divacky       if (ConstantArray *UserCA = dyn_cast<ConstantArray>(UserC)) {
49117a519f9SDimitry Andric         NewC = ConstantArray::get(UserCA->getType(), NewOps);
492f22ef01cSRoman Divacky       } else if (ConstantStruct *UserCS = dyn_cast<ConstantStruct>(UserC)) {
49317a519f9SDimitry Andric         NewC = ConstantStruct::get(UserCS->getType(), NewOps);
494f22ef01cSRoman Divacky       } else if (isa<ConstantVector>(UserC)) {
4952754fe60SDimitry Andric         NewC = ConstantVector::get(NewOps);
496f22ef01cSRoman Divacky       } else {
497f22ef01cSRoman Divacky         assert(isa<ConstantExpr>(UserC) && "Must be a ConstantExpr.");
4983b0f4066SDimitry Andric         NewC = cast<ConstantExpr>(UserC)->getWithOperands(NewOps);
499f22ef01cSRoman Divacky       }
500f22ef01cSRoman Divacky 
501f22ef01cSRoman Divacky       UserC->replaceAllUsesWith(NewC);
502f22ef01cSRoman Divacky       UserC->destroyConstant();
503f22ef01cSRoman Divacky       NewOps.clear();
504f22ef01cSRoman Divacky     }
505f22ef01cSRoman Divacky 
506f22ef01cSRoman Divacky     // Update all ValueHandles, they should be the only users at this point.
507f22ef01cSRoman Divacky     Placeholder->replaceAllUsesWith(RealVal);
508f22ef01cSRoman Divacky     delete Placeholder;
509f22ef01cSRoman Divacky   }
510f22ef01cSRoman Divacky }
511f22ef01cSRoman Divacky 
51239d628a0SDimitry Andric void BitcodeReaderMDValueList::AssignValue(Metadata *MD, unsigned Idx) {
513f22ef01cSRoman Divacky   if (Idx == size()) {
51439d628a0SDimitry Andric     push_back(MD);
515f22ef01cSRoman Divacky     return;
516f22ef01cSRoman Divacky   }
517f22ef01cSRoman Divacky 
518f22ef01cSRoman Divacky   if (Idx >= size())
519f22ef01cSRoman Divacky     resize(Idx+1);
520f22ef01cSRoman Divacky 
52139d628a0SDimitry Andric   TrackingMDRef &OldMD = MDValuePtrs[Idx];
52239d628a0SDimitry Andric   if (!OldMD) {
52339d628a0SDimitry Andric     OldMD.reset(MD);
524f22ef01cSRoman Divacky     return;
525f22ef01cSRoman Divacky   }
526f22ef01cSRoman Divacky 
527f22ef01cSRoman Divacky   // If there was a forward reference to this value, replace it.
52839d628a0SDimitry Andric   MDNodeFwdDecl *PrevMD = cast<MDNodeFwdDecl>(OldMD.get());
52939d628a0SDimitry Andric   PrevMD->replaceAllUsesWith(MD);
53039d628a0SDimitry Andric   MDNode::deleteTemporary(PrevMD);
53139d628a0SDimitry Andric   --NumFwdRefs;
532f22ef01cSRoman Divacky }
533f22ef01cSRoman Divacky 
53439d628a0SDimitry Andric Metadata *BitcodeReaderMDValueList::getValueFwdRef(unsigned Idx) {
535f22ef01cSRoman Divacky   if (Idx >= size())
536f22ef01cSRoman Divacky     resize(Idx + 1);
537f22ef01cSRoman Divacky 
53839d628a0SDimitry Andric   if (Metadata *MD = MDValuePtrs[Idx])
53939d628a0SDimitry Andric     return MD;
540f22ef01cSRoman Divacky 
541f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
54239d628a0SDimitry Andric   AnyFwdRefs = true;
54339d628a0SDimitry Andric   ++NumFwdRefs;
54439d628a0SDimitry Andric   Metadata *MD = MDNode::getTemporary(Context, None);
54539d628a0SDimitry Andric   MDValuePtrs[Idx].reset(MD);
54639d628a0SDimitry Andric   return MD;
54739d628a0SDimitry Andric }
54839d628a0SDimitry Andric 
54939d628a0SDimitry Andric void BitcodeReaderMDValueList::tryToResolveCycles() {
55039d628a0SDimitry Andric   if (!AnyFwdRefs)
55139d628a0SDimitry Andric     // Nothing to do.
55239d628a0SDimitry Andric     return;
55339d628a0SDimitry Andric 
55439d628a0SDimitry Andric   if (NumFwdRefs)
55539d628a0SDimitry Andric     // Still forward references... can't resolve cycles.
55639d628a0SDimitry Andric     return;
55739d628a0SDimitry Andric 
55839d628a0SDimitry Andric   // Resolve any cycles.
55939d628a0SDimitry Andric   for (auto &MD : MDValuePtrs) {
56039d628a0SDimitry Andric     assert(!(MD && isa<MDNodeFwdDecl>(MD)) && "Unexpected forward reference");
56139d628a0SDimitry Andric     if (auto *N = dyn_cast_or_null<UniquableMDNode>(MD))
56239d628a0SDimitry Andric       N->resolveCycles();
56339d628a0SDimitry Andric   }
564f22ef01cSRoman Divacky }
565f22ef01cSRoman Divacky 
56617a519f9SDimitry Andric Type *BitcodeReader::getTypeByID(unsigned ID) {
56717a519f9SDimitry Andric   // The type table size is always specified correctly.
56817a519f9SDimitry Andric   if (ID >= TypeList.size())
56991bc56edSDimitry Andric     return nullptr;
570f22ef01cSRoman Divacky 
57117a519f9SDimitry Andric   if (Type *Ty = TypeList[ID])
57217a519f9SDimitry Andric     return Ty;
57317a519f9SDimitry Andric 
57417a519f9SDimitry Andric   // If we have a forward reference, the only possible case is when it is to a
57517a519f9SDimitry Andric   // named struct.  Just create a placeholder for now.
57639d628a0SDimitry Andric   return TypeList[ID] = createIdentifiedStructType(Context);
57739d628a0SDimitry Andric }
57839d628a0SDimitry Andric 
57939d628a0SDimitry Andric StructType *BitcodeReader::createIdentifiedStructType(LLVMContext &Context,
58039d628a0SDimitry Andric                                                       StringRef Name) {
58139d628a0SDimitry Andric   auto *Ret = StructType::create(Context, Name);
58239d628a0SDimitry Andric   IdentifiedStructTypes.push_back(Ret);
58339d628a0SDimitry Andric   return Ret;
58439d628a0SDimitry Andric }
58539d628a0SDimitry Andric 
58639d628a0SDimitry Andric StructType *BitcodeReader::createIdentifiedStructType(LLVMContext &Context) {
58739d628a0SDimitry Andric   auto *Ret = StructType::create(Context);
58839d628a0SDimitry Andric   IdentifiedStructTypes.push_back(Ret);
58939d628a0SDimitry Andric   return Ret;
590f22ef01cSRoman Divacky }
591f22ef01cSRoman Divacky 
59217a519f9SDimitry Andric 
593f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
594f22ef01cSRoman Divacky //  Functions for parsing blocks from the bitcode file
595f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
596f22ef01cSRoman Divacky 
597139f7f9bSDimitry Andric 
598139f7f9bSDimitry Andric /// \brief This fills an AttrBuilder object with the LLVM attributes that have
599139f7f9bSDimitry Andric /// been decoded from the given integer. This function must stay in sync with
600139f7f9bSDimitry Andric /// 'encodeLLVMAttributesForBitcode'.
601139f7f9bSDimitry Andric static void decodeLLVMAttributesForBitcode(AttrBuilder &B,
602139f7f9bSDimitry Andric                                            uint64_t EncodedAttrs) {
603139f7f9bSDimitry Andric   // FIXME: Remove in 4.0.
604139f7f9bSDimitry Andric 
605139f7f9bSDimitry Andric   // The alignment is stored as a 16-bit raw value from bits 31--16.  We shift
606139f7f9bSDimitry Andric   // the bits above 31 down by 11 bits.
607139f7f9bSDimitry Andric   unsigned Alignment = (EncodedAttrs & (0xffffULL << 16)) >> 16;
608139f7f9bSDimitry Andric   assert((!Alignment || isPowerOf2_32(Alignment)) &&
609139f7f9bSDimitry Andric          "Alignment must be a power of two.");
610139f7f9bSDimitry Andric 
611139f7f9bSDimitry Andric   if (Alignment)
612139f7f9bSDimitry Andric     B.addAlignmentAttr(Alignment);
613139f7f9bSDimitry Andric   B.addRawValue(((EncodedAttrs & (0xfffffULL << 32)) >> 11) |
614139f7f9bSDimitry Andric                 (EncodedAttrs & 0xffff));
615139f7f9bSDimitry Andric }
616139f7f9bSDimitry Andric 
61791bc56edSDimitry Andric std::error_code BitcodeReader::ParseAttributeBlock() {
618f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::PARAMATTR_BLOCK_ID))
61939d628a0SDimitry Andric     return Error("Invalid record");
620f22ef01cSRoman Divacky 
621f22ef01cSRoman Divacky   if (!MAttributes.empty())
62239d628a0SDimitry Andric     return Error("Invalid multiple blocks");
623f22ef01cSRoman Divacky 
624f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
625f22ef01cSRoman Divacky 
626139f7f9bSDimitry Andric   SmallVector<AttributeSet, 8> Attrs;
627f22ef01cSRoman Divacky 
628f22ef01cSRoman Divacky   // Read all the records.
629f22ef01cSRoman Divacky   while (1) {
630139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
631139f7f9bSDimitry Andric 
632139f7f9bSDimitry Andric     switch (Entry.Kind) {
633139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
634139f7f9bSDimitry Andric     case BitstreamEntry::Error:
63539d628a0SDimitry Andric       return Error("Malformed block");
636139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
63791bc56edSDimitry Andric       return std::error_code();
638139f7f9bSDimitry Andric     case BitstreamEntry::Record:
639139f7f9bSDimitry Andric       // The interesting case.
640139f7f9bSDimitry Andric       break;
641f22ef01cSRoman Divacky     }
642f22ef01cSRoman Divacky 
643f22ef01cSRoman Divacky     // Read a record.
644f22ef01cSRoman Divacky     Record.clear();
645139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
646f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
647f22ef01cSRoman Divacky       break;
648139f7f9bSDimitry Andric     case bitc::PARAMATTR_CODE_ENTRY_OLD: { // ENTRY: [paramidx0, attr0, ...]
649139f7f9bSDimitry Andric       // FIXME: Remove in 4.0.
650f22ef01cSRoman Divacky       if (Record.size() & 1)
65139d628a0SDimitry Andric         return Error("Invalid record");
652f22ef01cSRoman Divacky 
653f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
654139f7f9bSDimitry Andric         AttrBuilder B;
655139f7f9bSDimitry Andric         decodeLLVMAttributesForBitcode(B, Record[i+1]);
656139f7f9bSDimitry Andric         Attrs.push_back(AttributeSet::get(Context, Record[i], B));
657f22ef01cSRoman Divacky       }
658f22ef01cSRoman Divacky 
659139f7f9bSDimitry Andric       MAttributes.push_back(AttributeSet::get(Context, Attrs));
660f22ef01cSRoman Divacky       Attrs.clear();
661f22ef01cSRoman Divacky       break;
662f22ef01cSRoman Divacky     }
663139f7f9bSDimitry Andric     case bitc::PARAMATTR_CODE_ENTRY: { // ENTRY: [attrgrp0, attrgrp1, ...]
664139f7f9bSDimitry Andric       for (unsigned i = 0, e = Record.size(); i != e; ++i)
665139f7f9bSDimitry Andric         Attrs.push_back(MAttributeGroups[Record[i]]);
666139f7f9bSDimitry Andric 
667139f7f9bSDimitry Andric       MAttributes.push_back(AttributeSet::get(Context, Attrs));
668139f7f9bSDimitry Andric       Attrs.clear();
669139f7f9bSDimitry Andric       break;
670139f7f9bSDimitry Andric     }
671139f7f9bSDimitry Andric     }
672139f7f9bSDimitry Andric   }
673139f7f9bSDimitry Andric }
674139f7f9bSDimitry Andric 
675f785676fSDimitry Andric // Returns Attribute::None on unrecognized codes.
676f785676fSDimitry Andric static Attribute::AttrKind GetAttrFromCode(uint64_t Code) {
677f785676fSDimitry Andric   switch (Code) {
678f785676fSDimitry Andric   default:
679f785676fSDimitry Andric     return Attribute::None;
680f785676fSDimitry Andric   case bitc::ATTR_KIND_ALIGNMENT:
681f785676fSDimitry Andric     return Attribute::Alignment;
682f785676fSDimitry Andric   case bitc::ATTR_KIND_ALWAYS_INLINE:
683f785676fSDimitry Andric     return Attribute::AlwaysInline;
684f785676fSDimitry Andric   case bitc::ATTR_KIND_BUILTIN:
685f785676fSDimitry Andric     return Attribute::Builtin;
686f785676fSDimitry Andric   case bitc::ATTR_KIND_BY_VAL:
687f785676fSDimitry Andric     return Attribute::ByVal;
68891bc56edSDimitry Andric   case bitc::ATTR_KIND_IN_ALLOCA:
68991bc56edSDimitry Andric     return Attribute::InAlloca;
690f785676fSDimitry Andric   case bitc::ATTR_KIND_COLD:
691f785676fSDimitry Andric     return Attribute::Cold;
692f785676fSDimitry Andric   case bitc::ATTR_KIND_INLINE_HINT:
693f785676fSDimitry Andric     return Attribute::InlineHint;
694f785676fSDimitry Andric   case bitc::ATTR_KIND_IN_REG:
695f785676fSDimitry Andric     return Attribute::InReg;
69691bc56edSDimitry Andric   case bitc::ATTR_KIND_JUMP_TABLE:
69791bc56edSDimitry Andric     return Attribute::JumpTable;
698f785676fSDimitry Andric   case bitc::ATTR_KIND_MIN_SIZE:
699f785676fSDimitry Andric     return Attribute::MinSize;
700f785676fSDimitry Andric   case bitc::ATTR_KIND_NAKED:
701f785676fSDimitry Andric     return Attribute::Naked;
702f785676fSDimitry Andric   case bitc::ATTR_KIND_NEST:
703f785676fSDimitry Andric     return Attribute::Nest;
704f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_ALIAS:
705f785676fSDimitry Andric     return Attribute::NoAlias;
706f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_BUILTIN:
707f785676fSDimitry Andric     return Attribute::NoBuiltin;
708f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_CAPTURE:
709f785676fSDimitry Andric     return Attribute::NoCapture;
710f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_DUPLICATE:
711f785676fSDimitry Andric     return Attribute::NoDuplicate;
712f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_IMPLICIT_FLOAT:
713f785676fSDimitry Andric     return Attribute::NoImplicitFloat;
714f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_INLINE:
715f785676fSDimitry Andric     return Attribute::NoInline;
716f785676fSDimitry Andric   case bitc::ATTR_KIND_NON_LAZY_BIND:
717f785676fSDimitry Andric     return Attribute::NonLazyBind;
71891bc56edSDimitry Andric   case bitc::ATTR_KIND_NON_NULL:
71991bc56edSDimitry Andric     return Attribute::NonNull;
72091bc56edSDimitry Andric   case bitc::ATTR_KIND_DEREFERENCEABLE:
72191bc56edSDimitry Andric     return Attribute::Dereferenceable;
722f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_RED_ZONE:
723f785676fSDimitry Andric     return Attribute::NoRedZone;
724f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_RETURN:
725f785676fSDimitry Andric     return Attribute::NoReturn;
726f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_UNWIND:
727f785676fSDimitry Andric     return Attribute::NoUnwind;
728f785676fSDimitry Andric   case bitc::ATTR_KIND_OPTIMIZE_FOR_SIZE:
729f785676fSDimitry Andric     return Attribute::OptimizeForSize;
730f785676fSDimitry Andric   case bitc::ATTR_KIND_OPTIMIZE_NONE:
731f785676fSDimitry Andric     return Attribute::OptimizeNone;
732f785676fSDimitry Andric   case bitc::ATTR_KIND_READ_NONE:
733f785676fSDimitry Andric     return Attribute::ReadNone;
734f785676fSDimitry Andric   case bitc::ATTR_KIND_READ_ONLY:
735f785676fSDimitry Andric     return Attribute::ReadOnly;
736f785676fSDimitry Andric   case bitc::ATTR_KIND_RETURNED:
737f785676fSDimitry Andric     return Attribute::Returned;
738f785676fSDimitry Andric   case bitc::ATTR_KIND_RETURNS_TWICE:
739f785676fSDimitry Andric     return Attribute::ReturnsTwice;
740f785676fSDimitry Andric   case bitc::ATTR_KIND_S_EXT:
741f785676fSDimitry Andric     return Attribute::SExt;
742f785676fSDimitry Andric   case bitc::ATTR_KIND_STACK_ALIGNMENT:
743f785676fSDimitry Andric     return Attribute::StackAlignment;
744f785676fSDimitry Andric   case bitc::ATTR_KIND_STACK_PROTECT:
745f785676fSDimitry Andric     return Attribute::StackProtect;
746f785676fSDimitry Andric   case bitc::ATTR_KIND_STACK_PROTECT_REQ:
747f785676fSDimitry Andric     return Attribute::StackProtectReq;
748f785676fSDimitry Andric   case bitc::ATTR_KIND_STACK_PROTECT_STRONG:
749f785676fSDimitry Andric     return Attribute::StackProtectStrong;
750f785676fSDimitry Andric   case bitc::ATTR_KIND_STRUCT_RET:
751f785676fSDimitry Andric     return Attribute::StructRet;
752f785676fSDimitry Andric   case bitc::ATTR_KIND_SANITIZE_ADDRESS:
753f785676fSDimitry Andric     return Attribute::SanitizeAddress;
754f785676fSDimitry Andric   case bitc::ATTR_KIND_SANITIZE_THREAD:
755f785676fSDimitry Andric     return Attribute::SanitizeThread;
756f785676fSDimitry Andric   case bitc::ATTR_KIND_SANITIZE_MEMORY:
757f785676fSDimitry Andric     return Attribute::SanitizeMemory;
758f785676fSDimitry Andric   case bitc::ATTR_KIND_UW_TABLE:
759f785676fSDimitry Andric     return Attribute::UWTable;
760f785676fSDimitry Andric   case bitc::ATTR_KIND_Z_EXT:
761f785676fSDimitry Andric     return Attribute::ZExt;
762f785676fSDimitry Andric   }
763f785676fSDimitry Andric }
764f785676fSDimitry Andric 
76591bc56edSDimitry Andric std::error_code BitcodeReader::ParseAttrKind(uint64_t Code,
766f785676fSDimitry Andric                                              Attribute::AttrKind *Kind) {
767f785676fSDimitry Andric   *Kind = GetAttrFromCode(Code);
768f785676fSDimitry Andric   if (*Kind == Attribute::None)
76939d628a0SDimitry Andric     return Error(BitcodeError::CorruptedBitcode,
77039d628a0SDimitry Andric                  "Unknown attribute kind (" + Twine(Code) + ")");
77191bc56edSDimitry Andric   return std::error_code();
772f785676fSDimitry Andric }
773f785676fSDimitry Andric 
77491bc56edSDimitry Andric std::error_code BitcodeReader::ParseAttributeGroupBlock() {
775139f7f9bSDimitry Andric   if (Stream.EnterSubBlock(bitc::PARAMATTR_GROUP_BLOCK_ID))
77639d628a0SDimitry Andric     return Error("Invalid record");
777139f7f9bSDimitry Andric 
778139f7f9bSDimitry Andric   if (!MAttributeGroups.empty())
77939d628a0SDimitry Andric     return Error("Invalid multiple blocks");
780139f7f9bSDimitry Andric 
781139f7f9bSDimitry Andric   SmallVector<uint64_t, 64> Record;
782139f7f9bSDimitry Andric 
783139f7f9bSDimitry Andric   // Read all the records.
784139f7f9bSDimitry Andric   while (1) {
785139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
786139f7f9bSDimitry Andric 
787139f7f9bSDimitry Andric     switch (Entry.Kind) {
788139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
789139f7f9bSDimitry Andric     case BitstreamEntry::Error:
79039d628a0SDimitry Andric       return Error("Malformed block");
791139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
79291bc56edSDimitry Andric       return std::error_code();
793139f7f9bSDimitry Andric     case BitstreamEntry::Record:
794139f7f9bSDimitry Andric       // The interesting case.
795139f7f9bSDimitry Andric       break;
796139f7f9bSDimitry Andric     }
797139f7f9bSDimitry Andric 
798139f7f9bSDimitry Andric     // Read a record.
799139f7f9bSDimitry Andric     Record.clear();
800139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
801139f7f9bSDimitry Andric     default:  // Default behavior: ignore.
802139f7f9bSDimitry Andric       break;
803139f7f9bSDimitry Andric     case bitc::PARAMATTR_GRP_CODE_ENTRY: { // ENTRY: [grpid, idx, a0, a1, ...]
804139f7f9bSDimitry Andric       if (Record.size() < 3)
80539d628a0SDimitry Andric         return Error("Invalid record");
806139f7f9bSDimitry Andric 
807139f7f9bSDimitry Andric       uint64_t GrpID = Record[0];
808139f7f9bSDimitry Andric       uint64_t Idx = Record[1]; // Index of the object this attribute refers to.
809139f7f9bSDimitry Andric 
810139f7f9bSDimitry Andric       AttrBuilder B;
811139f7f9bSDimitry Andric       for (unsigned i = 2, e = Record.size(); i != e; ++i) {
812139f7f9bSDimitry Andric         if (Record[i] == 0) {        // Enum attribute
813f785676fSDimitry Andric           Attribute::AttrKind Kind;
81491bc56edSDimitry Andric           if (std::error_code EC = ParseAttrKind(Record[++i], &Kind))
815f785676fSDimitry Andric             return EC;
816f785676fSDimitry Andric 
817f785676fSDimitry Andric           B.addAttribute(Kind);
81891bc56edSDimitry Andric         } else if (Record[i] == 1) { // Integer attribute
819f785676fSDimitry Andric           Attribute::AttrKind Kind;
82091bc56edSDimitry Andric           if (std::error_code EC = ParseAttrKind(Record[++i], &Kind))
821f785676fSDimitry Andric             return EC;
822f785676fSDimitry Andric           if (Kind == Attribute::Alignment)
823139f7f9bSDimitry Andric             B.addAlignmentAttr(Record[++i]);
82491bc56edSDimitry Andric           else if (Kind == Attribute::StackAlignment)
825139f7f9bSDimitry Andric             B.addStackAlignmentAttr(Record[++i]);
82691bc56edSDimitry Andric           else if (Kind == Attribute::Dereferenceable)
82791bc56edSDimitry Andric             B.addDereferenceableAttr(Record[++i]);
828139f7f9bSDimitry Andric         } else {                     // String attribute
829139f7f9bSDimitry Andric           assert((Record[i] == 3 || Record[i] == 4) &&
830139f7f9bSDimitry Andric                  "Invalid attribute group entry");
831139f7f9bSDimitry Andric           bool HasValue = (Record[i++] == 4);
832139f7f9bSDimitry Andric           SmallString<64> KindStr;
833139f7f9bSDimitry Andric           SmallString<64> ValStr;
834139f7f9bSDimitry Andric 
835139f7f9bSDimitry Andric           while (Record[i] != 0 && i != e)
836139f7f9bSDimitry Andric             KindStr += Record[i++];
837139f7f9bSDimitry Andric           assert(Record[i] == 0 && "Kind string not null terminated");
838139f7f9bSDimitry Andric 
839139f7f9bSDimitry Andric           if (HasValue) {
840139f7f9bSDimitry Andric             // Has a value associated with it.
841139f7f9bSDimitry Andric             ++i; // Skip the '0' that terminates the "kind" string.
842139f7f9bSDimitry Andric             while (Record[i] != 0 && i != e)
843139f7f9bSDimitry Andric               ValStr += Record[i++];
844139f7f9bSDimitry Andric             assert(Record[i] == 0 && "Value string not null terminated");
845139f7f9bSDimitry Andric           }
846139f7f9bSDimitry Andric 
847139f7f9bSDimitry Andric           B.addAttribute(KindStr.str(), ValStr.str());
848139f7f9bSDimitry Andric         }
849139f7f9bSDimitry Andric       }
850139f7f9bSDimitry Andric 
851139f7f9bSDimitry Andric       MAttributeGroups[GrpID] = AttributeSet::get(Context, Idx, B);
852139f7f9bSDimitry Andric       break;
853139f7f9bSDimitry Andric     }
854f22ef01cSRoman Divacky     }
855f22ef01cSRoman Divacky   }
856f22ef01cSRoman Divacky }
857f22ef01cSRoman Divacky 
85891bc56edSDimitry Andric std::error_code BitcodeReader::ParseTypeTable() {
85917a519f9SDimitry Andric   if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID_NEW))
86039d628a0SDimitry Andric     return Error("Invalid record");
861f22ef01cSRoman Divacky 
86217a519f9SDimitry Andric   return ParseTypeTableBody();
86317a519f9SDimitry Andric }
86417a519f9SDimitry Andric 
86591bc56edSDimitry Andric std::error_code BitcodeReader::ParseTypeTableBody() {
866f22ef01cSRoman Divacky   if (!TypeList.empty())
86739d628a0SDimitry Andric     return Error("Invalid multiple blocks");
868f22ef01cSRoman Divacky 
869f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
870f22ef01cSRoman Divacky   unsigned NumRecords = 0;
871f22ef01cSRoman Divacky 
87217a519f9SDimitry Andric   SmallString<64> TypeName;
87317a519f9SDimitry Andric 
874f22ef01cSRoman Divacky   // Read all the records for this type table.
875f22ef01cSRoman Divacky   while (1) {
876139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
877139f7f9bSDimitry Andric 
878139f7f9bSDimitry Andric     switch (Entry.Kind) {
879139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
880139f7f9bSDimitry Andric     case BitstreamEntry::Error:
88139d628a0SDimitry Andric       return Error("Malformed block");
882139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
883f22ef01cSRoman Divacky       if (NumRecords != TypeList.size())
88439d628a0SDimitry Andric         return Error("Malformed block");
88591bc56edSDimitry Andric       return std::error_code();
886139f7f9bSDimitry Andric     case BitstreamEntry::Record:
887139f7f9bSDimitry Andric       // The interesting case.
888139f7f9bSDimitry Andric       break;
889f22ef01cSRoman Divacky     }
890f22ef01cSRoman Divacky 
891f22ef01cSRoman Divacky     // Read a record.
892f22ef01cSRoman Divacky     Record.clear();
89391bc56edSDimitry Andric     Type *ResultTy = nullptr;
894139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
895f785676fSDimitry Andric     default:
89639d628a0SDimitry Andric       return Error("Invalid value");
897f22ef01cSRoman Divacky     case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
898f22ef01cSRoman Divacky       // TYPE_CODE_NUMENTRY contains a count of the number of types in the
899f22ef01cSRoman Divacky       // type list.  This allows us to reserve space.
900f22ef01cSRoman Divacky       if (Record.size() < 1)
90139d628a0SDimitry Andric         return Error("Invalid record");
90217a519f9SDimitry Andric       TypeList.resize(Record[0]);
903f22ef01cSRoman Divacky       continue;
904f22ef01cSRoman Divacky     case bitc::TYPE_CODE_VOID:      // VOID
905f22ef01cSRoman Divacky       ResultTy = Type::getVoidTy(Context);
906f22ef01cSRoman Divacky       break;
907dff0c46cSDimitry Andric     case bitc::TYPE_CODE_HALF:     // HALF
908dff0c46cSDimitry Andric       ResultTy = Type::getHalfTy(Context);
909dff0c46cSDimitry Andric       break;
910f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FLOAT:     // FLOAT
911f22ef01cSRoman Divacky       ResultTy = Type::getFloatTy(Context);
912f22ef01cSRoman Divacky       break;
913f22ef01cSRoman Divacky     case bitc::TYPE_CODE_DOUBLE:    // DOUBLE
914f22ef01cSRoman Divacky       ResultTy = Type::getDoubleTy(Context);
915f22ef01cSRoman Divacky       break;
916f22ef01cSRoman Divacky     case bitc::TYPE_CODE_X86_FP80:  // X86_FP80
917f22ef01cSRoman Divacky       ResultTy = Type::getX86_FP80Ty(Context);
918f22ef01cSRoman Divacky       break;
919f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FP128:     // FP128
920f22ef01cSRoman Divacky       ResultTy = Type::getFP128Ty(Context);
921f22ef01cSRoman Divacky       break;
922f22ef01cSRoman Divacky     case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
923f22ef01cSRoman Divacky       ResultTy = Type::getPPC_FP128Ty(Context);
924f22ef01cSRoman Divacky       break;
925f22ef01cSRoman Divacky     case bitc::TYPE_CODE_LABEL:     // LABEL
926f22ef01cSRoman Divacky       ResultTy = Type::getLabelTy(Context);
927f22ef01cSRoman Divacky       break;
928f22ef01cSRoman Divacky     case bitc::TYPE_CODE_METADATA:  // METADATA
929f22ef01cSRoman Divacky       ResultTy = Type::getMetadataTy(Context);
930f22ef01cSRoman Divacky       break;
9312754fe60SDimitry Andric     case bitc::TYPE_CODE_X86_MMX:   // X86_MMX
9322754fe60SDimitry Andric       ResultTy = Type::getX86_MMXTy(Context);
9332754fe60SDimitry Andric       break;
934f22ef01cSRoman Divacky     case bitc::TYPE_CODE_INTEGER:   // INTEGER: [width]
935f22ef01cSRoman Divacky       if (Record.size() < 1)
93639d628a0SDimitry Andric         return Error("Invalid record");
937f22ef01cSRoman Divacky 
938f22ef01cSRoman Divacky       ResultTy = IntegerType::get(Context, Record[0]);
939f22ef01cSRoman Divacky       break;
940f22ef01cSRoman Divacky     case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
941f22ef01cSRoman Divacky                                     //          [pointee type, address space]
942f22ef01cSRoman Divacky       if (Record.size() < 1)
94339d628a0SDimitry Andric         return Error("Invalid record");
944f22ef01cSRoman Divacky       unsigned AddressSpace = 0;
945f22ef01cSRoman Divacky       if (Record.size() == 2)
946f22ef01cSRoman Divacky         AddressSpace = Record[1];
94717a519f9SDimitry Andric       ResultTy = getTypeByID(Record[0]);
94891bc56edSDimitry Andric       if (!ResultTy)
94939d628a0SDimitry Andric         return Error("Invalid type");
95017a519f9SDimitry Andric       ResultTy = PointerType::get(ResultTy, AddressSpace);
951f22ef01cSRoman Divacky       break;
952f22ef01cSRoman Divacky     }
953dff0c46cSDimitry Andric     case bitc::TYPE_CODE_FUNCTION_OLD: {
9547ae0e2c9SDimitry Andric       // FIXME: attrid is dead, remove it in LLVM 4.0
955f22ef01cSRoman Divacky       // FUNCTION: [vararg, attrid, retty, paramty x N]
956f22ef01cSRoman Divacky       if (Record.size() < 3)
95739d628a0SDimitry Andric         return Error("Invalid record");
958dff0c46cSDimitry Andric       SmallVector<Type*, 8> ArgTys;
95917a519f9SDimitry Andric       for (unsigned i = 3, e = Record.size(); i != e; ++i) {
96017a519f9SDimitry Andric         if (Type *T = getTypeByID(Record[i]))
96117a519f9SDimitry Andric           ArgTys.push_back(T);
96217a519f9SDimitry Andric         else
963f22ef01cSRoman Divacky           break;
964f22ef01cSRoman Divacky       }
96517a519f9SDimitry Andric 
96617a519f9SDimitry Andric       ResultTy = getTypeByID(Record[2]);
96791bc56edSDimitry Andric       if (!ResultTy || ArgTys.size() < Record.size()-3)
96839d628a0SDimitry Andric         return Error("Invalid type");
96917a519f9SDimitry Andric 
97017a519f9SDimitry Andric       ResultTy = FunctionType::get(ResultTy, ArgTys, Record[0]);
97117a519f9SDimitry Andric       break;
97217a519f9SDimitry Andric     }
973dff0c46cSDimitry Andric     case bitc::TYPE_CODE_FUNCTION: {
974dff0c46cSDimitry Andric       // FUNCTION: [vararg, retty, paramty x N]
975dff0c46cSDimitry Andric       if (Record.size() < 2)
97639d628a0SDimitry Andric         return Error("Invalid record");
977dff0c46cSDimitry Andric       SmallVector<Type*, 8> ArgTys;
978dff0c46cSDimitry Andric       for (unsigned i = 2, e = Record.size(); i != e; ++i) {
979dff0c46cSDimitry Andric         if (Type *T = getTypeByID(Record[i]))
980dff0c46cSDimitry Andric           ArgTys.push_back(T);
981dff0c46cSDimitry Andric         else
982dff0c46cSDimitry Andric           break;
983dff0c46cSDimitry Andric       }
984dff0c46cSDimitry Andric 
985dff0c46cSDimitry Andric       ResultTy = getTypeByID(Record[1]);
98691bc56edSDimitry Andric       if (!ResultTy || ArgTys.size() < Record.size()-2)
98739d628a0SDimitry Andric         return Error("Invalid type");
988dff0c46cSDimitry Andric 
989dff0c46cSDimitry Andric       ResultTy = FunctionType::get(ResultTy, ArgTys, Record[0]);
990dff0c46cSDimitry Andric       break;
991dff0c46cSDimitry Andric     }
99217a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_ANON: {  // STRUCT: [ispacked, eltty x N]
993f22ef01cSRoman Divacky       if (Record.size() < 1)
99439d628a0SDimitry Andric         return Error("Invalid record");
995dff0c46cSDimitry Andric       SmallVector<Type*, 8> EltTys;
99617a519f9SDimitry Andric       for (unsigned i = 1, e = Record.size(); i != e; ++i) {
99717a519f9SDimitry Andric         if (Type *T = getTypeByID(Record[i]))
99817a519f9SDimitry Andric           EltTys.push_back(T);
99917a519f9SDimitry Andric         else
100017a519f9SDimitry Andric           break;
100117a519f9SDimitry Andric       }
100217a519f9SDimitry Andric       if (EltTys.size() != Record.size()-1)
100339d628a0SDimitry Andric         return Error("Invalid type");
1004f22ef01cSRoman Divacky       ResultTy = StructType::get(Context, EltTys, Record[0]);
1005f22ef01cSRoman Divacky       break;
1006f22ef01cSRoman Divacky     }
100717a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_NAME:   // STRUCT_NAME: [strchr x N]
100817a519f9SDimitry Andric       if (ConvertToString(Record, 0, TypeName))
100939d628a0SDimitry Andric         return Error("Invalid record");
101017a519f9SDimitry Andric       continue;
101117a519f9SDimitry Andric 
101217a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_NAMED: { // STRUCT: [ispacked, eltty x N]
101317a519f9SDimitry Andric       if (Record.size() < 1)
101439d628a0SDimitry Andric         return Error("Invalid record");
101517a519f9SDimitry Andric 
101617a519f9SDimitry Andric       if (NumRecords >= TypeList.size())
101739d628a0SDimitry Andric         return Error("Invalid TYPE table");
101817a519f9SDimitry Andric 
101917a519f9SDimitry Andric       // Check to see if this was forward referenced, if so fill in the temp.
102017a519f9SDimitry Andric       StructType *Res = cast_or_null<StructType>(TypeList[NumRecords]);
102117a519f9SDimitry Andric       if (Res) {
102217a519f9SDimitry Andric         Res->setName(TypeName);
102391bc56edSDimitry Andric         TypeList[NumRecords] = nullptr;
102417a519f9SDimitry Andric       } else  // Otherwise, create a new struct.
102539d628a0SDimitry Andric         Res = createIdentifiedStructType(Context, TypeName);
102617a519f9SDimitry Andric       TypeName.clear();
102717a519f9SDimitry Andric 
102817a519f9SDimitry Andric       SmallVector<Type*, 8> EltTys;
102917a519f9SDimitry Andric       for (unsigned i = 1, e = Record.size(); i != e; ++i) {
103017a519f9SDimitry Andric         if (Type *T = getTypeByID(Record[i]))
103117a519f9SDimitry Andric           EltTys.push_back(T);
103217a519f9SDimitry Andric         else
103317a519f9SDimitry Andric           break;
103417a519f9SDimitry Andric       }
103517a519f9SDimitry Andric       if (EltTys.size() != Record.size()-1)
103639d628a0SDimitry Andric         return Error("Invalid record");
103717a519f9SDimitry Andric       Res->setBody(EltTys, Record[0]);
103817a519f9SDimitry Andric       ResultTy = Res;
103917a519f9SDimitry Andric       break;
104017a519f9SDimitry Andric     }
104117a519f9SDimitry Andric     case bitc::TYPE_CODE_OPAQUE: {       // OPAQUE: []
104217a519f9SDimitry Andric       if (Record.size() != 1)
104339d628a0SDimitry Andric         return Error("Invalid record");
104417a519f9SDimitry Andric 
104517a519f9SDimitry Andric       if (NumRecords >= TypeList.size())
104639d628a0SDimitry Andric         return Error("Invalid TYPE table");
104717a519f9SDimitry Andric 
104817a519f9SDimitry Andric       // Check to see if this was forward referenced, if so fill in the temp.
104917a519f9SDimitry Andric       StructType *Res = cast_or_null<StructType>(TypeList[NumRecords]);
105017a519f9SDimitry Andric       if (Res) {
105117a519f9SDimitry Andric         Res->setName(TypeName);
105291bc56edSDimitry Andric         TypeList[NumRecords] = nullptr;
105317a519f9SDimitry Andric       } else  // Otherwise, create a new struct with no body.
105439d628a0SDimitry Andric         Res = createIdentifiedStructType(Context, TypeName);
105517a519f9SDimitry Andric       TypeName.clear();
105617a519f9SDimitry Andric       ResultTy = Res;
105717a519f9SDimitry Andric       break;
105817a519f9SDimitry Andric     }
1059f22ef01cSRoman Divacky     case bitc::TYPE_CODE_ARRAY:     // ARRAY: [numelts, eltty]
1060f22ef01cSRoman Divacky       if (Record.size() < 2)
106139d628a0SDimitry Andric         return Error("Invalid record");
106217a519f9SDimitry Andric       if ((ResultTy = getTypeByID(Record[1])))
106317a519f9SDimitry Andric         ResultTy = ArrayType::get(ResultTy, Record[0]);
106417a519f9SDimitry Andric       else
106539d628a0SDimitry Andric         return Error("Invalid type");
1066f22ef01cSRoman Divacky       break;
1067f22ef01cSRoman Divacky     case bitc::TYPE_CODE_VECTOR:    // VECTOR: [numelts, eltty]
1068f22ef01cSRoman Divacky       if (Record.size() < 2)
106939d628a0SDimitry Andric         return Error("Invalid record");
107017a519f9SDimitry Andric       if ((ResultTy = getTypeByID(Record[1])))
107117a519f9SDimitry Andric         ResultTy = VectorType::get(ResultTy, Record[0]);
107217a519f9SDimitry Andric       else
107339d628a0SDimitry Andric         return Error("Invalid type");
1074f22ef01cSRoman Divacky       break;
1075f22ef01cSRoman Divacky     }
1076f22ef01cSRoman Divacky 
107717a519f9SDimitry Andric     if (NumRecords >= TypeList.size())
107839d628a0SDimitry Andric       return Error("Invalid TYPE table");
107917a519f9SDimitry Andric     assert(ResultTy && "Didn't read a type?");
108091bc56edSDimitry Andric     assert(!TypeList[NumRecords] && "Already read type?");
108117a519f9SDimitry Andric     TypeList[NumRecords++] = ResultTy;
1082f22ef01cSRoman Divacky   }
1083f22ef01cSRoman Divacky }
108417a519f9SDimitry Andric 
108591bc56edSDimitry Andric std::error_code BitcodeReader::ParseValueSymbolTable() {
1086f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::VALUE_SYMTAB_BLOCK_ID))
108739d628a0SDimitry Andric     return Error("Invalid record");
1088f22ef01cSRoman Divacky 
1089f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1090f22ef01cSRoman Divacky 
1091f22ef01cSRoman Divacky   // Read all the records for this value table.
1092f22ef01cSRoman Divacky   SmallString<128> ValueName;
1093f22ef01cSRoman Divacky   while (1) {
1094139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
1095f22ef01cSRoman Divacky 
1096139f7f9bSDimitry Andric     switch (Entry.Kind) {
1097139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
1098139f7f9bSDimitry Andric     case BitstreamEntry::Error:
109939d628a0SDimitry Andric       return Error("Malformed block");
1100139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
110191bc56edSDimitry Andric       return std::error_code();
1102139f7f9bSDimitry Andric     case BitstreamEntry::Record:
1103139f7f9bSDimitry Andric       // The interesting case.
1104139f7f9bSDimitry Andric       break;
1105f22ef01cSRoman Divacky     }
1106f22ef01cSRoman Divacky 
1107f22ef01cSRoman Divacky     // Read a record.
1108f22ef01cSRoman Divacky     Record.clear();
1109139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
1110f22ef01cSRoman Divacky     default:  // Default behavior: unknown type.
1111f22ef01cSRoman Divacky       break;
1112f22ef01cSRoman Divacky     case bitc::VST_CODE_ENTRY: {  // VST_ENTRY: [valueid, namechar x N]
1113f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, ValueName))
111439d628a0SDimitry Andric         return Error("Invalid record");
1115f22ef01cSRoman Divacky       unsigned ValueID = Record[0];
111691bc56edSDimitry Andric       if (ValueID >= ValueList.size() || !ValueList[ValueID])
111739d628a0SDimitry Andric         return Error("Invalid record");
1118f22ef01cSRoman Divacky       Value *V = ValueList[ValueID];
1119f22ef01cSRoman Divacky 
1120f22ef01cSRoman Divacky       V->setName(StringRef(ValueName.data(), ValueName.size()));
1121f22ef01cSRoman Divacky       ValueName.clear();
1122f22ef01cSRoman Divacky       break;
1123f22ef01cSRoman Divacky     }
1124f22ef01cSRoman Divacky     case bitc::VST_CODE_BBENTRY: {
1125f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, ValueName))
112639d628a0SDimitry Andric         return Error("Invalid record");
1127f22ef01cSRoman Divacky       BasicBlock *BB = getBasicBlock(Record[0]);
112891bc56edSDimitry Andric       if (!BB)
112939d628a0SDimitry Andric         return Error("Invalid record");
1130f22ef01cSRoman Divacky 
1131f22ef01cSRoman Divacky       BB->setName(StringRef(ValueName.data(), ValueName.size()));
1132f22ef01cSRoman Divacky       ValueName.clear();
1133f22ef01cSRoman Divacky       break;
1134f22ef01cSRoman Divacky     }
1135f22ef01cSRoman Divacky     }
1136f22ef01cSRoman Divacky   }
1137f22ef01cSRoman Divacky }
1138f22ef01cSRoman Divacky 
113991bc56edSDimitry Andric std::error_code BitcodeReader::ParseMetadata() {
1140f22ef01cSRoman Divacky   unsigned NextMDValueNo = MDValueList.size();
1141f22ef01cSRoman Divacky 
1142f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
114339d628a0SDimitry Andric     return Error("Invalid record");
1144f22ef01cSRoman Divacky 
1145f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1146f22ef01cSRoman Divacky 
1147f22ef01cSRoman Divacky   // Read all the records.
1148f22ef01cSRoman Divacky   while (1) {
1149139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
1150139f7f9bSDimitry Andric 
1151139f7f9bSDimitry Andric     switch (Entry.Kind) {
1152139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
1153139f7f9bSDimitry Andric     case BitstreamEntry::Error:
115439d628a0SDimitry Andric       return Error("Malformed block");
1155139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
115639d628a0SDimitry Andric       MDValueList.tryToResolveCycles();
115791bc56edSDimitry Andric       return std::error_code();
1158139f7f9bSDimitry Andric     case BitstreamEntry::Record:
1159139f7f9bSDimitry Andric       // The interesting case.
1160139f7f9bSDimitry Andric       break;
1161f22ef01cSRoman Divacky     }
1162f22ef01cSRoman Divacky 
1163f22ef01cSRoman Divacky     // Read a record.
1164f22ef01cSRoman Divacky     Record.clear();
1165139f7f9bSDimitry Andric     unsigned Code = Stream.readRecord(Entry.ID, Record);
116639d628a0SDimitry Andric     bool IsDistinct = false;
1167e580952dSDimitry Andric     switch (Code) {
1168f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
1169f22ef01cSRoman Divacky       break;
1170f22ef01cSRoman Divacky     case bitc::METADATA_NAME: {
1171139f7f9bSDimitry Andric       // Read name of the named metadata.
11727ae0e2c9SDimitry Andric       SmallString<8> Name(Record.begin(), Record.end());
1173f22ef01cSRoman Divacky       Record.clear();
1174f22ef01cSRoman Divacky       Code = Stream.ReadCode();
1175f22ef01cSRoman Divacky 
117617a519f9SDimitry Andric       // METADATA_NAME is always followed by METADATA_NAMED_NODE.
1177139f7f9bSDimitry Andric       unsigned NextBitCode = Stream.readRecord(Code, Record);
117817a519f9SDimitry Andric       assert(NextBitCode == bitc::METADATA_NAMED_NODE); (void)NextBitCode;
1179f22ef01cSRoman Divacky 
1180f22ef01cSRoman Divacky       // Read named metadata elements.
1181f22ef01cSRoman Divacky       unsigned Size = Record.size();
1182e580952dSDimitry Andric       NamedMDNode *NMD = TheModule->getOrInsertNamedMetadata(Name);
1183f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; ++i) {
118491bc56edSDimitry Andric         MDNode *MD = dyn_cast_or_null<MDNode>(MDValueList.getValueFwdRef(Record[i]));
118591bc56edSDimitry Andric         if (!MD)
118639d628a0SDimitry Andric           return Error("Invalid record");
1187e580952dSDimitry Andric         NMD->addOperand(MD);
1188f22ef01cSRoman Divacky       }
1189f22ef01cSRoman Divacky       break;
1190f22ef01cSRoman Divacky     }
119139d628a0SDimitry Andric     case bitc::METADATA_OLD_FN_NODE: {
119239d628a0SDimitry Andric       // FIXME: Remove in 4.0.
119339d628a0SDimitry Andric       // This is a LocalAsMetadata record, the only type of function-local
119439d628a0SDimitry Andric       // metadata.
1195ffd1746dSEd Schouten       if (Record.size() % 2 == 1)
119639d628a0SDimitry Andric         return Error("Invalid record");
119739d628a0SDimitry Andric 
119839d628a0SDimitry Andric       // If this isn't a LocalAsMetadata record, we're dropping it.  This used
119939d628a0SDimitry Andric       // to be legal, but there's no upgrade path.
120039d628a0SDimitry Andric       auto dropRecord = [&] {
120139d628a0SDimitry Andric         MDValueList.AssignValue(MDNode::get(Context, None), NextMDValueNo++);
120239d628a0SDimitry Andric       };
120339d628a0SDimitry Andric       if (Record.size() != 2) {
120439d628a0SDimitry Andric         dropRecord();
120539d628a0SDimitry Andric         break;
120639d628a0SDimitry Andric       }
120739d628a0SDimitry Andric 
120839d628a0SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
120939d628a0SDimitry Andric       if (Ty->isMetadataTy() || Ty->isVoidTy()) {
121039d628a0SDimitry Andric         dropRecord();
121139d628a0SDimitry Andric         break;
121239d628a0SDimitry Andric       }
121339d628a0SDimitry Andric 
121439d628a0SDimitry Andric       MDValueList.AssignValue(
121539d628a0SDimitry Andric           LocalAsMetadata::get(ValueList.getValueFwdRef(Record[1], Ty)),
121639d628a0SDimitry Andric           NextMDValueNo++);
121739d628a0SDimitry Andric       break;
121839d628a0SDimitry Andric     }
121939d628a0SDimitry Andric     case bitc::METADATA_OLD_NODE: {
122039d628a0SDimitry Andric       // FIXME: Remove in 4.0.
122139d628a0SDimitry Andric       if (Record.size() % 2 == 1)
122239d628a0SDimitry Andric         return Error("Invalid record");
1223f22ef01cSRoman Divacky 
1224f22ef01cSRoman Divacky       unsigned Size = Record.size();
122539d628a0SDimitry Andric       SmallVector<Metadata *, 8> Elts;
1226f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; i += 2) {
12276122f3e6SDimitry Andric         Type *Ty = getTypeByID(Record[i]);
1228f785676fSDimitry Andric         if (!Ty)
122939d628a0SDimitry Andric           return Error("Invalid record");
1230f22ef01cSRoman Divacky         if (Ty->isMetadataTy())
1231f22ef01cSRoman Divacky           Elts.push_back(MDValueList.getValueFwdRef(Record[i+1]));
123239d628a0SDimitry Andric         else if (!Ty->isVoidTy()) {
123339d628a0SDimitry Andric           auto *MD =
123439d628a0SDimitry Andric               ValueAsMetadata::get(ValueList.getValueFwdRef(Record[i + 1], Ty));
123539d628a0SDimitry Andric           assert(isa<ConstantAsMetadata>(MD) &&
123639d628a0SDimitry Andric                  "Expected non-function-local metadata");
123739d628a0SDimitry Andric           Elts.push_back(MD);
123839d628a0SDimitry Andric         } else
123991bc56edSDimitry Andric           Elts.push_back(nullptr);
1240f22ef01cSRoman Divacky       }
124139d628a0SDimitry Andric       MDValueList.AssignValue(MDNode::get(Context, Elts), NextMDValueNo++);
124239d628a0SDimitry Andric       break;
124339d628a0SDimitry Andric     }
124439d628a0SDimitry Andric     case bitc::METADATA_VALUE: {
124539d628a0SDimitry Andric       if (Record.size() != 2)
124639d628a0SDimitry Andric         return Error("Invalid record");
124739d628a0SDimitry Andric 
124839d628a0SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
124939d628a0SDimitry Andric       if (Ty->isMetadataTy() || Ty->isVoidTy())
125039d628a0SDimitry Andric         return Error("Invalid record");
125139d628a0SDimitry Andric 
125239d628a0SDimitry Andric       MDValueList.AssignValue(
125339d628a0SDimitry Andric           ValueAsMetadata::get(ValueList.getValueFwdRef(Record[1], Ty)),
125439d628a0SDimitry Andric           NextMDValueNo++);
125539d628a0SDimitry Andric       break;
125639d628a0SDimitry Andric     }
125739d628a0SDimitry Andric     case bitc::METADATA_DISTINCT_NODE:
125839d628a0SDimitry Andric       IsDistinct = true;
125939d628a0SDimitry Andric       // fallthrough...
126039d628a0SDimitry Andric     case bitc::METADATA_NODE: {
126139d628a0SDimitry Andric       SmallVector<Metadata *, 8> Elts;
126239d628a0SDimitry Andric       Elts.reserve(Record.size());
126339d628a0SDimitry Andric       for (unsigned ID : Record)
126439d628a0SDimitry Andric         Elts.push_back(ID ? MDValueList.getValueFwdRef(ID - 1) : nullptr);
126539d628a0SDimitry Andric       MDValueList.AssignValue(IsDistinct ? MDNode::getDistinct(Context, Elts)
126639d628a0SDimitry Andric                                          : MDNode::get(Context, Elts),
126739d628a0SDimitry Andric                               NextMDValueNo++);
126839d628a0SDimitry Andric       break;
126939d628a0SDimitry Andric     }
127039d628a0SDimitry Andric     case bitc::METADATA_LOCATION: {
127139d628a0SDimitry Andric       if (Record.size() != 5)
127239d628a0SDimitry Andric         return Error("Invalid record");
127339d628a0SDimitry Andric 
127439d628a0SDimitry Andric       auto get = Record[0] ? MDLocation::getDistinct : MDLocation::get;
127539d628a0SDimitry Andric       unsigned Line = Record[1];
127639d628a0SDimitry Andric       unsigned Column = Record[2];
127739d628a0SDimitry Andric       MDNode *Scope = cast<MDNode>(MDValueList.getValueFwdRef(Record[3]));
127839d628a0SDimitry Andric       Metadata *InlinedAt =
127939d628a0SDimitry Andric           Record[4] ? MDValueList.getValueFwdRef(Record[4] - 1) : nullptr;
128039d628a0SDimitry Andric       MDValueList.AssignValue(get(Context, Line, Column, Scope, InlinedAt),
128139d628a0SDimitry Andric                               NextMDValueNo++);
1282f22ef01cSRoman Divacky       break;
1283f22ef01cSRoman Divacky     }
1284f22ef01cSRoman Divacky     case bitc::METADATA_STRING: {
128591bc56edSDimitry Andric       std::string String(Record.begin(), Record.end());
128691bc56edSDimitry Andric       llvm::UpgradeMDStringConstant(String);
128739d628a0SDimitry Andric       Metadata *MD = MDString::get(Context, String);
128839d628a0SDimitry Andric       MDValueList.AssignValue(MD, NextMDValueNo++);
1289f22ef01cSRoman Divacky       break;
1290f22ef01cSRoman Divacky     }
1291f22ef01cSRoman Divacky     case bitc::METADATA_KIND: {
12927ae0e2c9SDimitry Andric       if (Record.size() < 2)
129339d628a0SDimitry Andric         return Error("Invalid record");
12947ae0e2c9SDimitry Andric 
1295f22ef01cSRoman Divacky       unsigned Kind = Record[0];
12967ae0e2c9SDimitry Andric       SmallString<8> Name(Record.begin()+1, Record.end());
1297f22ef01cSRoman Divacky 
1298f22ef01cSRoman Divacky       unsigned NewKind = TheModule->getMDKindID(Name.str());
1299e580952dSDimitry Andric       if (!MDKindMap.insert(std::make_pair(Kind, NewKind)).second)
130039d628a0SDimitry Andric         return Error("Conflicting METADATA_KIND records");
1301f22ef01cSRoman Divacky       break;
1302f22ef01cSRoman Divacky     }
1303f22ef01cSRoman Divacky     }
1304f22ef01cSRoman Divacky   }
1305f22ef01cSRoman Divacky }
1306f22ef01cSRoman Divacky 
13073861d79fSDimitry Andric /// decodeSignRotatedValue - Decode a signed value stored with the sign bit in
1308f22ef01cSRoman Divacky /// the LSB for dense VBR encoding.
13093861d79fSDimitry Andric uint64_t BitcodeReader::decodeSignRotatedValue(uint64_t V) {
1310f22ef01cSRoman Divacky   if ((V & 1) == 0)
1311f22ef01cSRoman Divacky     return V >> 1;
1312f22ef01cSRoman Divacky   if (V != 1)
1313f22ef01cSRoman Divacky     return -(V >> 1);
1314f22ef01cSRoman Divacky   // There is no such thing as -0 with integers.  "-0" really means MININT.
1315f22ef01cSRoman Divacky   return 1ULL << 63;
1316f22ef01cSRoman Divacky }
1317f22ef01cSRoman Divacky 
1318f22ef01cSRoman Divacky /// ResolveGlobalAndAliasInits - Resolve all of the initializers for global
1319f22ef01cSRoman Divacky /// values and aliases that we can.
132091bc56edSDimitry Andric std::error_code BitcodeReader::ResolveGlobalAndAliasInits() {
1321f22ef01cSRoman Divacky   std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInitWorklist;
1322f22ef01cSRoman Divacky   std::vector<std::pair<GlobalAlias*, unsigned> > AliasInitWorklist;
1323f785676fSDimitry Andric   std::vector<std::pair<Function*, unsigned> > FunctionPrefixWorklist;
132439d628a0SDimitry Andric   std::vector<std::pair<Function*, unsigned> > FunctionPrologueWorklist;
1325f22ef01cSRoman Divacky 
1326f22ef01cSRoman Divacky   GlobalInitWorklist.swap(GlobalInits);
1327f22ef01cSRoman Divacky   AliasInitWorklist.swap(AliasInits);
1328f785676fSDimitry Andric   FunctionPrefixWorklist.swap(FunctionPrefixes);
132939d628a0SDimitry Andric   FunctionPrologueWorklist.swap(FunctionPrologues);
1330f22ef01cSRoman Divacky 
1331f22ef01cSRoman Divacky   while (!GlobalInitWorklist.empty()) {
1332f22ef01cSRoman Divacky     unsigned ValID = GlobalInitWorklist.back().second;
1333f22ef01cSRoman Divacky     if (ValID >= ValueList.size()) {
1334f22ef01cSRoman Divacky       // Not ready to resolve this yet, it requires something later in the file.
1335f22ef01cSRoman Divacky       GlobalInits.push_back(GlobalInitWorklist.back());
1336f22ef01cSRoman Divacky     } else {
133791bc56edSDimitry Andric       if (Constant *C = dyn_cast_or_null<Constant>(ValueList[ValID]))
1338f22ef01cSRoman Divacky         GlobalInitWorklist.back().first->setInitializer(C);
1339f22ef01cSRoman Divacky       else
134039d628a0SDimitry Andric         return Error("Expected a constant");
1341f22ef01cSRoman Divacky     }
1342f22ef01cSRoman Divacky     GlobalInitWorklist.pop_back();
1343f22ef01cSRoman Divacky   }
1344f22ef01cSRoman Divacky 
1345f22ef01cSRoman Divacky   while (!AliasInitWorklist.empty()) {
1346f22ef01cSRoman Divacky     unsigned ValID = AliasInitWorklist.back().second;
1347f22ef01cSRoman Divacky     if (ValID >= ValueList.size()) {
1348f22ef01cSRoman Divacky       AliasInits.push_back(AliasInitWorklist.back());
1349f22ef01cSRoman Divacky     } else {
135091bc56edSDimitry Andric       if (Constant *C = dyn_cast_or_null<Constant>(ValueList[ValID]))
1351f22ef01cSRoman Divacky         AliasInitWorklist.back().first->setAliasee(C);
1352f22ef01cSRoman Divacky       else
135339d628a0SDimitry Andric         return Error("Expected a constant");
1354f22ef01cSRoman Divacky     }
1355f22ef01cSRoman Divacky     AliasInitWorklist.pop_back();
1356f22ef01cSRoman Divacky   }
1357f785676fSDimitry Andric 
1358f785676fSDimitry Andric   while (!FunctionPrefixWorklist.empty()) {
1359f785676fSDimitry Andric     unsigned ValID = FunctionPrefixWorklist.back().second;
1360f785676fSDimitry Andric     if (ValID >= ValueList.size()) {
1361f785676fSDimitry Andric       FunctionPrefixes.push_back(FunctionPrefixWorklist.back());
1362f785676fSDimitry Andric     } else {
136391bc56edSDimitry Andric       if (Constant *C = dyn_cast_or_null<Constant>(ValueList[ValID]))
1364f785676fSDimitry Andric         FunctionPrefixWorklist.back().first->setPrefixData(C);
1365f785676fSDimitry Andric       else
136639d628a0SDimitry Andric         return Error("Expected a constant");
1367f785676fSDimitry Andric     }
1368f785676fSDimitry Andric     FunctionPrefixWorklist.pop_back();
1369f785676fSDimitry Andric   }
1370f785676fSDimitry Andric 
137139d628a0SDimitry Andric   while (!FunctionPrologueWorklist.empty()) {
137239d628a0SDimitry Andric     unsigned ValID = FunctionPrologueWorklist.back().second;
137339d628a0SDimitry Andric     if (ValID >= ValueList.size()) {
137439d628a0SDimitry Andric       FunctionPrologues.push_back(FunctionPrologueWorklist.back());
137539d628a0SDimitry Andric     } else {
137639d628a0SDimitry Andric       if (Constant *C = dyn_cast_or_null<Constant>(ValueList[ValID]))
137739d628a0SDimitry Andric         FunctionPrologueWorklist.back().first->setPrologueData(C);
137839d628a0SDimitry Andric       else
137939d628a0SDimitry Andric         return Error("Expected a constant");
138039d628a0SDimitry Andric     }
138139d628a0SDimitry Andric     FunctionPrologueWorklist.pop_back();
138239d628a0SDimitry Andric   }
138339d628a0SDimitry Andric 
138491bc56edSDimitry Andric   return std::error_code();
1385f22ef01cSRoman Divacky }
1386f22ef01cSRoman Divacky 
13877ae0e2c9SDimitry Andric static APInt ReadWideAPInt(ArrayRef<uint64_t> Vals, unsigned TypeBits) {
13887ae0e2c9SDimitry Andric   SmallVector<uint64_t, 8> Words(Vals.size());
13897ae0e2c9SDimitry Andric   std::transform(Vals.begin(), Vals.end(), Words.begin(),
13903861d79fSDimitry Andric                  BitcodeReader::decodeSignRotatedValue);
13917ae0e2c9SDimitry Andric 
13927ae0e2c9SDimitry Andric   return APInt(TypeBits, Words);
13937ae0e2c9SDimitry Andric }
13947ae0e2c9SDimitry Andric 
139591bc56edSDimitry Andric std::error_code BitcodeReader::ParseConstants() {
1396f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::CONSTANTS_BLOCK_ID))
139739d628a0SDimitry Andric     return Error("Invalid record");
1398f22ef01cSRoman Divacky 
1399f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1400f22ef01cSRoman Divacky 
1401f22ef01cSRoman Divacky   // Read all the records for this value table.
14026122f3e6SDimitry Andric   Type *CurTy = Type::getInt32Ty(Context);
1403f22ef01cSRoman Divacky   unsigned NextCstNo = ValueList.size();
1404f22ef01cSRoman Divacky   while (1) {
1405139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
1406139f7f9bSDimitry Andric 
1407139f7f9bSDimitry Andric     switch (Entry.Kind) {
1408139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
1409139f7f9bSDimitry Andric     case BitstreamEntry::Error:
141039d628a0SDimitry Andric       return Error("Malformed block");
1411139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
1412139f7f9bSDimitry Andric       if (NextCstNo != ValueList.size())
141339d628a0SDimitry Andric         return Error("Invalid ronstant reference");
1414139f7f9bSDimitry Andric 
1415139f7f9bSDimitry Andric       // Once all the constants have been read, go through and resolve forward
1416139f7f9bSDimitry Andric       // references.
1417139f7f9bSDimitry Andric       ValueList.ResolveConstantForwardRefs();
141891bc56edSDimitry Andric       return std::error_code();
1419139f7f9bSDimitry Andric     case BitstreamEntry::Record:
1420139f7f9bSDimitry Andric       // The interesting case.
1421f22ef01cSRoman Divacky       break;
1422f22ef01cSRoman Divacky     }
1423f22ef01cSRoman Divacky 
1424f22ef01cSRoman Divacky     // Read a record.
1425f22ef01cSRoman Divacky     Record.clear();
142691bc56edSDimitry Andric     Value *V = nullptr;
1427139f7f9bSDimitry Andric     unsigned BitCode = Stream.readRecord(Entry.ID, Record);
1428f22ef01cSRoman Divacky     switch (BitCode) {
1429f22ef01cSRoman Divacky     default:  // Default behavior: unknown constant
1430f22ef01cSRoman Divacky     case bitc::CST_CODE_UNDEF:     // UNDEF
1431f22ef01cSRoman Divacky       V = UndefValue::get(CurTy);
1432f22ef01cSRoman Divacky       break;
1433f22ef01cSRoman Divacky     case bitc::CST_CODE_SETTYPE:   // SETTYPE: [typeid]
1434f22ef01cSRoman Divacky       if (Record.empty())
143539d628a0SDimitry Andric         return Error("Invalid record");
143691bc56edSDimitry Andric       if (Record[0] >= TypeList.size() || !TypeList[Record[0]])
143739d628a0SDimitry Andric         return Error("Invalid record");
1438f22ef01cSRoman Divacky       CurTy = TypeList[Record[0]];
1439f22ef01cSRoman Divacky       continue;  // Skip the ValueList manipulation.
1440f22ef01cSRoman Divacky     case bitc::CST_CODE_NULL:      // NULL
1441f22ef01cSRoman Divacky       V = Constant::getNullValue(CurTy);
1442f22ef01cSRoman Divacky       break;
1443f22ef01cSRoman Divacky     case bitc::CST_CODE_INTEGER:   // INTEGER: [intval]
1444f22ef01cSRoman Divacky       if (!CurTy->isIntegerTy() || Record.empty())
144539d628a0SDimitry Andric         return Error("Invalid record");
14463861d79fSDimitry Andric       V = ConstantInt::get(CurTy, decodeSignRotatedValue(Record[0]));
1447f22ef01cSRoman Divacky       break;
1448f22ef01cSRoman Divacky     case bitc::CST_CODE_WIDE_INTEGER: {// WIDE_INTEGER: [n x intval]
1449f22ef01cSRoman Divacky       if (!CurTy->isIntegerTy() || Record.empty())
145039d628a0SDimitry Andric         return Error("Invalid record");
1451f22ef01cSRoman Divacky 
14527ae0e2c9SDimitry Andric       APInt VInt = ReadWideAPInt(Record,
14537ae0e2c9SDimitry Andric                                  cast<IntegerType>(CurTy)->getBitWidth());
14547ae0e2c9SDimitry Andric       V = ConstantInt::get(Context, VInt);
14557ae0e2c9SDimitry Andric 
1456f22ef01cSRoman Divacky       break;
1457f22ef01cSRoman Divacky     }
1458f22ef01cSRoman Divacky     case bitc::CST_CODE_FLOAT: {    // FLOAT: [fpval]
1459f22ef01cSRoman Divacky       if (Record.empty())
146039d628a0SDimitry Andric         return Error("Invalid record");
1461dff0c46cSDimitry Andric       if (CurTy->isHalfTy())
1462139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::IEEEhalf,
1463139f7f9bSDimitry Andric                                              APInt(16, (uint16_t)Record[0])));
1464dff0c46cSDimitry Andric       else if (CurTy->isFloatTy())
1465139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::IEEEsingle,
1466139f7f9bSDimitry Andric                                              APInt(32, (uint32_t)Record[0])));
1467f22ef01cSRoman Divacky       else if (CurTy->isDoubleTy())
1468139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::IEEEdouble,
1469139f7f9bSDimitry Andric                                              APInt(64, Record[0])));
1470f22ef01cSRoman Divacky       else if (CurTy->isX86_FP80Ty()) {
1471f22ef01cSRoman Divacky         // Bits are not stored the same way as a normal i80 APInt, compensate.
1472f22ef01cSRoman Divacky         uint64_t Rearrange[2];
1473f22ef01cSRoman Divacky         Rearrange[0] = (Record[1] & 0xffffLL) | (Record[0] << 16);
1474f22ef01cSRoman Divacky         Rearrange[1] = Record[0] >> 48;
1475139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::x87DoubleExtended,
1476139f7f9bSDimitry Andric                                              APInt(80, Rearrange)));
1477f22ef01cSRoman Divacky       } else if (CurTy->isFP128Ty())
1478139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::IEEEquad,
1479139f7f9bSDimitry Andric                                              APInt(128, Record)));
1480f22ef01cSRoman Divacky       else if (CurTy->isPPC_FP128Ty())
1481139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::PPCDoubleDouble,
1482139f7f9bSDimitry Andric                                              APInt(128, Record)));
1483f22ef01cSRoman Divacky       else
1484f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);
1485f22ef01cSRoman Divacky       break;
1486f22ef01cSRoman Divacky     }
1487f22ef01cSRoman Divacky 
1488f22ef01cSRoman Divacky     case bitc::CST_CODE_AGGREGATE: {// AGGREGATE: [n x value number]
1489f22ef01cSRoman Divacky       if (Record.empty())
149039d628a0SDimitry Andric         return Error("Invalid record");
1491f22ef01cSRoman Divacky 
1492f22ef01cSRoman Divacky       unsigned Size = Record.size();
1493dff0c46cSDimitry Andric       SmallVector<Constant*, 16> Elts;
1494f22ef01cSRoman Divacky 
14956122f3e6SDimitry Andric       if (StructType *STy = dyn_cast<StructType>(CurTy)) {
1496f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1497f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i],
1498f22ef01cSRoman Divacky                                                      STy->getElementType(i)));
1499f22ef01cSRoman Divacky         V = ConstantStruct::get(STy, Elts);
15006122f3e6SDimitry Andric       } else if (ArrayType *ATy = dyn_cast<ArrayType>(CurTy)) {
15016122f3e6SDimitry Andric         Type *EltTy = ATy->getElementType();
1502f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1503f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
1504f22ef01cSRoman Divacky         V = ConstantArray::get(ATy, Elts);
15056122f3e6SDimitry Andric       } else if (VectorType *VTy = dyn_cast<VectorType>(CurTy)) {
15066122f3e6SDimitry Andric         Type *EltTy = VTy->getElementType();
1507f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1508f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
1509f22ef01cSRoman Divacky         V = ConstantVector::get(Elts);
1510f22ef01cSRoman Divacky       } else {
1511f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);
1512f22ef01cSRoman Divacky       }
1513f22ef01cSRoman Divacky       break;
1514f22ef01cSRoman Divacky     }
1515dff0c46cSDimitry Andric     case bitc::CST_CODE_STRING:    // STRING: [values]
1516f22ef01cSRoman Divacky     case bitc::CST_CODE_CSTRING: { // CSTRING: [values]
1517f22ef01cSRoman Divacky       if (Record.empty())
151839d628a0SDimitry Andric         return Error("Invalid record");
1519f22ef01cSRoman Divacky 
15207ae0e2c9SDimitry Andric       SmallString<16> Elts(Record.begin(), Record.end());
1521dff0c46cSDimitry Andric       V = ConstantDataArray::getString(Context, Elts,
1522dff0c46cSDimitry Andric                                        BitCode == bitc::CST_CODE_CSTRING);
1523f22ef01cSRoman Divacky       break;
1524f22ef01cSRoman Divacky     }
1525dff0c46cSDimitry Andric     case bitc::CST_CODE_DATA: {// DATA: [n x value]
1526dff0c46cSDimitry Andric       if (Record.empty())
152739d628a0SDimitry Andric         return Error("Invalid record");
1528dff0c46cSDimitry Andric 
1529dff0c46cSDimitry Andric       Type *EltTy = cast<SequentialType>(CurTy)->getElementType();
1530dff0c46cSDimitry Andric       unsigned Size = Record.size();
1531dff0c46cSDimitry Andric 
1532dff0c46cSDimitry Andric       if (EltTy->isIntegerTy(8)) {
1533dff0c46cSDimitry Andric         SmallVector<uint8_t, 16> Elts(Record.begin(), Record.end());
1534dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1535dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1536dff0c46cSDimitry Andric         else
1537dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1538dff0c46cSDimitry Andric       } else if (EltTy->isIntegerTy(16)) {
1539dff0c46cSDimitry Andric         SmallVector<uint16_t, 16> Elts(Record.begin(), Record.end());
1540dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1541dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1542dff0c46cSDimitry Andric         else
1543dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1544dff0c46cSDimitry Andric       } else if (EltTy->isIntegerTy(32)) {
1545dff0c46cSDimitry Andric         SmallVector<uint32_t, 16> Elts(Record.begin(), Record.end());
1546dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1547dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1548dff0c46cSDimitry Andric         else
1549dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1550dff0c46cSDimitry Andric       } else if (EltTy->isIntegerTy(64)) {
1551dff0c46cSDimitry Andric         SmallVector<uint64_t, 16> Elts(Record.begin(), Record.end());
1552dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1553dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1554dff0c46cSDimitry Andric         else
1555dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1556dff0c46cSDimitry Andric       } else if (EltTy->isFloatTy()) {
15577ae0e2c9SDimitry Andric         SmallVector<float, 16> Elts(Size);
15587ae0e2c9SDimitry Andric         std::transform(Record.begin(), Record.end(), Elts.begin(), BitsToFloat);
1559dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1560dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1561dff0c46cSDimitry Andric         else
1562dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1563dff0c46cSDimitry Andric       } else if (EltTy->isDoubleTy()) {
15647ae0e2c9SDimitry Andric         SmallVector<double, 16> Elts(Size);
15657ae0e2c9SDimitry Andric         std::transform(Record.begin(), Record.end(), Elts.begin(),
15667ae0e2c9SDimitry Andric                        BitsToDouble);
1567dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1568dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1569dff0c46cSDimitry Andric         else
1570dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1571dff0c46cSDimitry Andric       } else {
157239d628a0SDimitry Andric         return Error("Invalid type for value");
1573dff0c46cSDimitry Andric       }
1574dff0c46cSDimitry Andric       break;
1575dff0c46cSDimitry Andric     }
1576dff0c46cSDimitry Andric 
1577f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_BINOP: {  // CE_BINOP: [opcode, opval, opval]
1578f785676fSDimitry Andric       if (Record.size() < 3)
157939d628a0SDimitry Andric         return Error("Invalid record");
1580f22ef01cSRoman Divacky       int Opc = GetDecodedBinaryOpcode(Record[0], CurTy);
1581f22ef01cSRoman Divacky       if (Opc < 0) {
1582f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);  // Unknown binop.
1583f22ef01cSRoman Divacky       } else {
1584f22ef01cSRoman Divacky         Constant *LHS = ValueList.getConstantFwdRef(Record[1], CurTy);
1585f22ef01cSRoman Divacky         Constant *RHS = ValueList.getConstantFwdRef(Record[2], CurTy);
1586f22ef01cSRoman Divacky         unsigned Flags = 0;
1587f22ef01cSRoman Divacky         if (Record.size() >= 4) {
1588f22ef01cSRoman Divacky           if (Opc == Instruction::Add ||
1589f22ef01cSRoman Divacky               Opc == Instruction::Sub ||
15902754fe60SDimitry Andric               Opc == Instruction::Mul ||
15912754fe60SDimitry Andric               Opc == Instruction::Shl) {
1592f22ef01cSRoman Divacky             if (Record[3] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1593f22ef01cSRoman Divacky               Flags |= OverflowingBinaryOperator::NoSignedWrap;
1594f22ef01cSRoman Divacky             if (Record[3] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1595f22ef01cSRoman Divacky               Flags |= OverflowingBinaryOperator::NoUnsignedWrap;
15962754fe60SDimitry Andric           } else if (Opc == Instruction::SDiv ||
15972754fe60SDimitry Andric                      Opc == Instruction::UDiv ||
15982754fe60SDimitry Andric                      Opc == Instruction::LShr ||
15992754fe60SDimitry Andric                      Opc == Instruction::AShr) {
16002754fe60SDimitry Andric             if (Record[3] & (1 << bitc::PEO_EXACT))
1601f22ef01cSRoman Divacky               Flags |= SDivOperator::IsExact;
1602f22ef01cSRoman Divacky           }
1603f22ef01cSRoman Divacky         }
1604f22ef01cSRoman Divacky         V = ConstantExpr::get(Opc, LHS, RHS, Flags);
1605f22ef01cSRoman Divacky       }
1606f22ef01cSRoman Divacky       break;
1607f22ef01cSRoman Divacky     }
1608f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_CAST: {  // CE_CAST: [opcode, opty, opval]
1609f785676fSDimitry Andric       if (Record.size() < 3)
161039d628a0SDimitry Andric         return Error("Invalid record");
1611f22ef01cSRoman Divacky       int Opc = GetDecodedCastOpcode(Record[0]);
1612f22ef01cSRoman Divacky       if (Opc < 0) {
1613f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);  // Unknown cast.
1614f22ef01cSRoman Divacky       } else {
16156122f3e6SDimitry Andric         Type *OpTy = getTypeByID(Record[1]);
1616f785676fSDimitry Andric         if (!OpTy)
161739d628a0SDimitry Andric           return Error("Invalid record");
1618f22ef01cSRoman Divacky         Constant *Op = ValueList.getConstantFwdRef(Record[2], OpTy);
1619f785676fSDimitry Andric         V = UpgradeBitCastExpr(Opc, Op, CurTy);
1620f785676fSDimitry Andric         if (!V) V = ConstantExpr::getCast(Opc, Op, CurTy);
1621f22ef01cSRoman Divacky       }
1622f22ef01cSRoman Divacky       break;
1623f22ef01cSRoman Divacky     }
1624f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_INBOUNDS_GEP:
1625f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_GEP: {  // CE_GEP:        [n x operands]
1626f785676fSDimitry Andric       if (Record.size() & 1)
162739d628a0SDimitry Andric         return Error("Invalid record");
1628f22ef01cSRoman Divacky       SmallVector<Constant*, 16> Elts;
1629f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
16306122f3e6SDimitry Andric         Type *ElTy = getTypeByID(Record[i]);
1631f785676fSDimitry Andric         if (!ElTy)
163239d628a0SDimitry Andric           return Error("Invalid record");
1633f22ef01cSRoman Divacky         Elts.push_back(ValueList.getConstantFwdRef(Record[i+1], ElTy));
1634f22ef01cSRoman Divacky       }
16356122f3e6SDimitry Andric       ArrayRef<Constant *> Indices(Elts.begin() + 1, Elts.end());
16366122f3e6SDimitry Andric       V = ConstantExpr::getGetElementPtr(Elts[0], Indices,
16376122f3e6SDimitry Andric                                          BitCode ==
16386122f3e6SDimitry Andric                                            bitc::CST_CODE_CE_INBOUNDS_GEP);
1639f22ef01cSRoman Divacky       break;
1640f22ef01cSRoman Divacky     }
1641f785676fSDimitry Andric     case bitc::CST_CODE_CE_SELECT: {  // CE_SELECT: [opval#, opval#, opval#]
1642f785676fSDimitry Andric       if (Record.size() < 3)
164339d628a0SDimitry Andric         return Error("Invalid record");
1644f785676fSDimitry Andric 
1645f785676fSDimitry Andric       Type *SelectorTy = Type::getInt1Ty(Context);
1646f785676fSDimitry Andric 
1647f785676fSDimitry Andric       // If CurTy is a vector of length n, then Record[0] must be a <n x i1>
1648f785676fSDimitry Andric       // vector. Otherwise, it must be a single bit.
1649f785676fSDimitry Andric       if (VectorType *VTy = dyn_cast<VectorType>(CurTy))
1650f785676fSDimitry Andric         SelectorTy = VectorType::get(Type::getInt1Ty(Context),
1651f785676fSDimitry Andric                                      VTy->getNumElements());
1652f785676fSDimitry Andric 
1653f785676fSDimitry Andric       V = ConstantExpr::getSelect(ValueList.getConstantFwdRef(Record[0],
1654f785676fSDimitry Andric                                                               SelectorTy),
1655f22ef01cSRoman Divacky                                   ValueList.getConstantFwdRef(Record[1],CurTy),
1656f22ef01cSRoman Divacky                                   ValueList.getConstantFwdRef(Record[2],CurTy));
1657f22ef01cSRoman Divacky       break;
1658f785676fSDimitry Andric     }
165991bc56edSDimitry Andric     case bitc::CST_CODE_CE_EXTRACTELT
166091bc56edSDimitry Andric         : { // CE_EXTRACTELT: [opty, opval, opty, opval]
1661f785676fSDimitry Andric       if (Record.size() < 3)
166239d628a0SDimitry Andric         return Error("Invalid record");
16636122f3e6SDimitry Andric       VectorType *OpTy =
1664f22ef01cSRoman Divacky         dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
166591bc56edSDimitry Andric       if (!OpTy)
166639d628a0SDimitry Andric         return Error("Invalid record");
1667f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
166891bc56edSDimitry Andric       Constant *Op1 = nullptr;
166991bc56edSDimitry Andric       if (Record.size() == 4) {
167091bc56edSDimitry Andric         Type *IdxTy = getTypeByID(Record[2]);
167191bc56edSDimitry Andric         if (!IdxTy)
167239d628a0SDimitry Andric           return Error("Invalid record");
167391bc56edSDimitry Andric         Op1 = ValueList.getConstantFwdRef(Record[3], IdxTy);
167491bc56edSDimitry Andric       } else // TODO: Remove with llvm 4.0
167591bc56edSDimitry Andric         Op1 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
167691bc56edSDimitry Andric       if (!Op1)
167739d628a0SDimitry Andric         return Error("Invalid record");
1678f22ef01cSRoman Divacky       V = ConstantExpr::getExtractElement(Op0, Op1);
1679f22ef01cSRoman Divacky       break;
1680f22ef01cSRoman Divacky     }
168191bc56edSDimitry Andric     case bitc::CST_CODE_CE_INSERTELT
168291bc56edSDimitry Andric         : { // CE_INSERTELT: [opval, opval, opty, opval]
16836122f3e6SDimitry Andric       VectorType *OpTy = dyn_cast<VectorType>(CurTy);
168491bc56edSDimitry Andric       if (Record.size() < 3 || !OpTy)
168539d628a0SDimitry Andric         return Error("Invalid record");
1686f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1687f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[1],
1688f22ef01cSRoman Divacky                                                   OpTy->getElementType());
168991bc56edSDimitry Andric       Constant *Op2 = nullptr;
169091bc56edSDimitry Andric       if (Record.size() == 4) {
169191bc56edSDimitry Andric         Type *IdxTy = getTypeByID(Record[2]);
169291bc56edSDimitry Andric         if (!IdxTy)
169339d628a0SDimitry Andric           return Error("Invalid record");
169491bc56edSDimitry Andric         Op2 = ValueList.getConstantFwdRef(Record[3], IdxTy);
169591bc56edSDimitry Andric       } else // TODO: Remove with llvm 4.0
169691bc56edSDimitry Andric         Op2 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
169791bc56edSDimitry Andric       if (!Op2)
169839d628a0SDimitry Andric         return Error("Invalid record");
1699f22ef01cSRoman Divacky       V = ConstantExpr::getInsertElement(Op0, Op1, Op2);
1700f22ef01cSRoman Divacky       break;
1701f22ef01cSRoman Divacky     }
1702f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SHUFFLEVEC: { // CE_SHUFFLEVEC: [opval, opval, opval]
17036122f3e6SDimitry Andric       VectorType *OpTy = dyn_cast<VectorType>(CurTy);
170491bc56edSDimitry Andric       if (Record.size() < 3 || !OpTy)
170539d628a0SDimitry Andric         return Error("Invalid record");
1706f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1707f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[1], OpTy);
17086122f3e6SDimitry Andric       Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
1709f22ef01cSRoman Divacky                                                  OpTy->getNumElements());
1710f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[2], ShufTy);
1711f22ef01cSRoman Divacky       V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
1712f22ef01cSRoman Divacky       break;
1713f22ef01cSRoman Divacky     }
1714f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SHUFVEC_EX: { // [opty, opval, opval, opval]
17156122f3e6SDimitry Andric       VectorType *RTy = dyn_cast<VectorType>(CurTy);
17166122f3e6SDimitry Andric       VectorType *OpTy =
17172754fe60SDimitry Andric         dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
171891bc56edSDimitry Andric       if (Record.size() < 4 || !RTy || !OpTy)
171939d628a0SDimitry Andric         return Error("Invalid record");
1720f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1721f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
17226122f3e6SDimitry Andric       Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
1723f22ef01cSRoman Divacky                                                  RTy->getNumElements());
1724f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[3], ShufTy);
1725f22ef01cSRoman Divacky       V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
1726f22ef01cSRoman Divacky       break;
1727f22ef01cSRoman Divacky     }
1728f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_CMP: {     // CE_CMP: [opty, opval, opval, pred]
1729f785676fSDimitry Andric       if (Record.size() < 4)
173039d628a0SDimitry Andric         return Error("Invalid record");
17316122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[0]);
173291bc56edSDimitry Andric       if (!OpTy)
173339d628a0SDimitry Andric         return Error("Invalid record");
1734f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1735f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1736f22ef01cSRoman Divacky 
1737f22ef01cSRoman Divacky       if (OpTy->isFPOrFPVectorTy())
1738f22ef01cSRoman Divacky         V = ConstantExpr::getFCmp(Record[3], Op0, Op1);
1739f22ef01cSRoman Divacky       else
1740f22ef01cSRoman Divacky         V = ConstantExpr::getICmp(Record[3], Op0, Op1);
1741f22ef01cSRoman Divacky       break;
1742f22ef01cSRoman Divacky     }
17433861d79fSDimitry Andric     // This maintains backward compatibility, pre-asm dialect keywords.
17443861d79fSDimitry Andric     // FIXME: Remove with the 4.0 release.
17453861d79fSDimitry Andric     case bitc::CST_CODE_INLINEASM_OLD: {
1746f785676fSDimitry Andric       if (Record.size() < 2)
174739d628a0SDimitry Andric         return Error("Invalid record");
1748f22ef01cSRoman Divacky       std::string AsmStr, ConstrStr;
1749f22ef01cSRoman Divacky       bool HasSideEffects = Record[0] & 1;
1750f22ef01cSRoman Divacky       bool IsAlignStack = Record[0] >> 1;
1751f22ef01cSRoman Divacky       unsigned AsmStrSize = Record[1];
1752f22ef01cSRoman Divacky       if (2+AsmStrSize >= Record.size())
175339d628a0SDimitry Andric         return Error("Invalid record");
1754f22ef01cSRoman Divacky       unsigned ConstStrSize = Record[2+AsmStrSize];
1755f22ef01cSRoman Divacky       if (3+AsmStrSize+ConstStrSize > Record.size())
175639d628a0SDimitry Andric         return Error("Invalid record");
1757f22ef01cSRoman Divacky 
1758f22ef01cSRoman Divacky       for (unsigned i = 0; i != AsmStrSize; ++i)
1759f22ef01cSRoman Divacky         AsmStr += (char)Record[2+i];
1760f22ef01cSRoman Divacky       for (unsigned i = 0; i != ConstStrSize; ++i)
1761f22ef01cSRoman Divacky         ConstrStr += (char)Record[3+AsmStrSize+i];
17626122f3e6SDimitry Andric       PointerType *PTy = cast<PointerType>(CurTy);
1763f22ef01cSRoman Divacky       V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
1764f22ef01cSRoman Divacky                          AsmStr, ConstrStr, HasSideEffects, IsAlignStack);
1765f22ef01cSRoman Divacky       break;
1766f22ef01cSRoman Divacky     }
17673861d79fSDimitry Andric     // This version adds support for the asm dialect keywords (e.g.,
17683861d79fSDimitry Andric     // inteldialect).
17693861d79fSDimitry Andric     case bitc::CST_CODE_INLINEASM: {
1770f785676fSDimitry Andric       if (Record.size() < 2)
177139d628a0SDimitry Andric         return Error("Invalid record");
17723861d79fSDimitry Andric       std::string AsmStr, ConstrStr;
17733861d79fSDimitry Andric       bool HasSideEffects = Record[0] & 1;
17743861d79fSDimitry Andric       bool IsAlignStack = (Record[0] >> 1) & 1;
17753861d79fSDimitry Andric       unsigned AsmDialect = Record[0] >> 2;
17763861d79fSDimitry Andric       unsigned AsmStrSize = Record[1];
17773861d79fSDimitry Andric       if (2+AsmStrSize >= Record.size())
177839d628a0SDimitry Andric         return Error("Invalid record");
17793861d79fSDimitry Andric       unsigned ConstStrSize = Record[2+AsmStrSize];
17803861d79fSDimitry Andric       if (3+AsmStrSize+ConstStrSize > Record.size())
178139d628a0SDimitry Andric         return Error("Invalid record");
17823861d79fSDimitry Andric 
17833861d79fSDimitry Andric       for (unsigned i = 0; i != AsmStrSize; ++i)
17843861d79fSDimitry Andric         AsmStr += (char)Record[2+i];
17853861d79fSDimitry Andric       for (unsigned i = 0; i != ConstStrSize; ++i)
17863861d79fSDimitry Andric         ConstrStr += (char)Record[3+AsmStrSize+i];
17873861d79fSDimitry Andric       PointerType *PTy = cast<PointerType>(CurTy);
17883861d79fSDimitry Andric       V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
17893861d79fSDimitry Andric                          AsmStr, ConstrStr, HasSideEffects, IsAlignStack,
17903861d79fSDimitry Andric                          InlineAsm::AsmDialect(AsmDialect));
17913861d79fSDimitry Andric       break;
17923861d79fSDimitry Andric     }
1793f22ef01cSRoman Divacky     case bitc::CST_CODE_BLOCKADDRESS:{
1794f785676fSDimitry Andric       if (Record.size() < 3)
179539d628a0SDimitry Andric         return Error("Invalid record");
17966122f3e6SDimitry Andric       Type *FnTy = getTypeByID(Record[0]);
179791bc56edSDimitry Andric       if (!FnTy)
179839d628a0SDimitry Andric         return Error("Invalid record");
1799f22ef01cSRoman Divacky       Function *Fn =
1800f22ef01cSRoman Divacky         dyn_cast_or_null<Function>(ValueList.getConstantFwdRef(Record[1],FnTy));
180191bc56edSDimitry Andric       if (!Fn)
180239d628a0SDimitry Andric         return Error("Invalid record");
180339d628a0SDimitry Andric 
180439d628a0SDimitry Andric       // Don't let Fn get dematerialized.
180539d628a0SDimitry Andric       BlockAddressesTaken.insert(Fn);
1806f22ef01cSRoman Divacky 
18073861d79fSDimitry Andric       // If the function is already parsed we can insert the block address right
18083861d79fSDimitry Andric       // away.
180939d628a0SDimitry Andric       BasicBlock *BB;
181039d628a0SDimitry Andric       unsigned BBID = Record[2];
181139d628a0SDimitry Andric       if (!BBID)
181239d628a0SDimitry Andric         // Invalid reference to entry block.
181339d628a0SDimitry Andric         return Error("Invalid ID");
18143861d79fSDimitry Andric       if (!Fn->empty()) {
18153861d79fSDimitry Andric         Function::iterator BBI = Fn->begin(), BBE = Fn->end();
181639d628a0SDimitry Andric         for (size_t I = 0, E = BBID; I != E; ++I) {
18173861d79fSDimitry Andric           if (BBI == BBE)
181839d628a0SDimitry Andric             return Error("Invalid ID");
18193861d79fSDimitry Andric           ++BBI;
18203861d79fSDimitry Andric         }
182139d628a0SDimitry Andric         BB = BBI;
18223861d79fSDimitry Andric       } else {
18233861d79fSDimitry Andric         // Otherwise insert a placeholder and remember it so it can be inserted
18243861d79fSDimitry Andric         // when the function is parsed.
182539d628a0SDimitry Andric         auto &FwdBBs = BasicBlockFwdRefs[Fn];
182639d628a0SDimitry Andric         if (FwdBBs.empty())
182739d628a0SDimitry Andric           BasicBlockFwdRefQueue.push_back(Fn);
182839d628a0SDimitry Andric         if (FwdBBs.size() < BBID + 1)
182939d628a0SDimitry Andric           FwdBBs.resize(BBID + 1);
183039d628a0SDimitry Andric         if (!FwdBBs[BBID])
183139d628a0SDimitry Andric           FwdBBs[BBID] = BasicBlock::Create(Context);
183239d628a0SDimitry Andric         BB = FwdBBs[BBID];
18333861d79fSDimitry Andric       }
183439d628a0SDimitry Andric       V = BlockAddress::get(Fn, BB);
1835f22ef01cSRoman Divacky       break;
1836f22ef01cSRoman Divacky     }
1837f22ef01cSRoman Divacky     }
1838f22ef01cSRoman Divacky 
1839f22ef01cSRoman Divacky     ValueList.AssignValue(V, NextCstNo);
1840f22ef01cSRoman Divacky     ++NextCstNo;
1841f22ef01cSRoman Divacky   }
1842f22ef01cSRoman Divacky }
1843f22ef01cSRoman Divacky 
184491bc56edSDimitry Andric std::error_code BitcodeReader::ParseUseLists() {
1845dff0c46cSDimitry Andric   if (Stream.EnterSubBlock(bitc::USELIST_BLOCK_ID))
184639d628a0SDimitry Andric     return Error("Invalid record");
1847dff0c46cSDimitry Andric 
1848dff0c46cSDimitry Andric   // Read all the records.
184939d628a0SDimitry Andric   SmallVector<uint64_t, 64> Record;
1850dff0c46cSDimitry Andric   while (1) {
1851139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
1852139f7f9bSDimitry Andric 
1853139f7f9bSDimitry Andric     switch (Entry.Kind) {
1854139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
1855139f7f9bSDimitry Andric     case BitstreamEntry::Error:
185639d628a0SDimitry Andric       return Error("Malformed block");
1857139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
185891bc56edSDimitry Andric       return std::error_code();
1859139f7f9bSDimitry Andric     case BitstreamEntry::Record:
1860139f7f9bSDimitry Andric       // The interesting case.
1861139f7f9bSDimitry Andric       break;
1862dff0c46cSDimitry Andric     }
1863dff0c46cSDimitry Andric 
1864dff0c46cSDimitry Andric     // Read a use list record.
1865dff0c46cSDimitry Andric     Record.clear();
186639d628a0SDimitry Andric     bool IsBB = false;
1867139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
1868dff0c46cSDimitry Andric     default:  // Default behavior: unknown type.
1869dff0c46cSDimitry Andric       break;
187039d628a0SDimitry Andric     case bitc::USELIST_CODE_BB:
187139d628a0SDimitry Andric       IsBB = true;
187239d628a0SDimitry Andric       // fallthrough
187339d628a0SDimitry Andric     case bitc::USELIST_CODE_DEFAULT: {
1874dff0c46cSDimitry Andric       unsigned RecordLength = Record.size();
187539d628a0SDimitry Andric       if (RecordLength < 3)
187639d628a0SDimitry Andric         // Records should have at least an ID and two indexes.
187739d628a0SDimitry Andric         return Error("Invalid record");
187839d628a0SDimitry Andric       unsigned ID = Record.back();
187939d628a0SDimitry Andric       Record.pop_back();
188039d628a0SDimitry Andric 
188139d628a0SDimitry Andric       Value *V;
188239d628a0SDimitry Andric       if (IsBB) {
188339d628a0SDimitry Andric         assert(ID < FunctionBBs.size() && "Basic block not found");
188439d628a0SDimitry Andric         V = FunctionBBs[ID];
188539d628a0SDimitry Andric       } else
188639d628a0SDimitry Andric         V = ValueList[ID];
188739d628a0SDimitry Andric       unsigned NumUses = 0;
188839d628a0SDimitry Andric       SmallDenseMap<const Use *, unsigned, 16> Order;
188939d628a0SDimitry Andric       for (const Use &U : V->uses()) {
189039d628a0SDimitry Andric         if (++NumUses > Record.size())
189139d628a0SDimitry Andric           break;
189239d628a0SDimitry Andric         Order[&U] = Record[NumUses - 1];
189339d628a0SDimitry Andric       }
189439d628a0SDimitry Andric       if (Order.size() != Record.size() || NumUses > Record.size())
189539d628a0SDimitry Andric         // Mismatches can happen if the functions are being materialized lazily
189639d628a0SDimitry Andric         // (out-of-order), or a value has been upgraded.
189739d628a0SDimitry Andric         break;
189839d628a0SDimitry Andric 
189939d628a0SDimitry Andric       V->sortUseList([&](const Use &L, const Use &R) {
190039d628a0SDimitry Andric         return Order.lookup(&L) < Order.lookup(&R);
190139d628a0SDimitry Andric       });
1902dff0c46cSDimitry Andric       break;
1903dff0c46cSDimitry Andric     }
1904dff0c46cSDimitry Andric     }
1905dff0c46cSDimitry Andric   }
1906dff0c46cSDimitry Andric }
1907dff0c46cSDimitry Andric 
1908f22ef01cSRoman Divacky /// RememberAndSkipFunctionBody - When we see the block for a function body,
1909f22ef01cSRoman Divacky /// remember where it is and then skip it.  This lets us lazily deserialize the
1910f22ef01cSRoman Divacky /// functions.
191191bc56edSDimitry Andric std::error_code BitcodeReader::RememberAndSkipFunctionBody() {
1912f22ef01cSRoman Divacky   // Get the function we are talking about.
1913f22ef01cSRoman Divacky   if (FunctionsWithBodies.empty())
191439d628a0SDimitry Andric     return Error("Insufficient function protos");
1915f22ef01cSRoman Divacky 
1916f22ef01cSRoman Divacky   Function *Fn = FunctionsWithBodies.back();
1917f22ef01cSRoman Divacky   FunctionsWithBodies.pop_back();
1918f22ef01cSRoman Divacky 
1919f22ef01cSRoman Divacky   // Save the current stream state.
1920f22ef01cSRoman Divacky   uint64_t CurBit = Stream.GetCurrentBitNo();
1921f22ef01cSRoman Divacky   DeferredFunctionInfo[Fn] = CurBit;
1922f22ef01cSRoman Divacky 
1923f22ef01cSRoman Divacky   // Skip over the function block for now.
1924f22ef01cSRoman Divacky   if (Stream.SkipBlock())
192539d628a0SDimitry Andric     return Error("Invalid record");
192691bc56edSDimitry Andric   return std::error_code();
1927f22ef01cSRoman Divacky }
1928f22ef01cSRoman Divacky 
192991bc56edSDimitry Andric std::error_code BitcodeReader::GlobalCleanup() {
1930f22ef01cSRoman Divacky   // Patch the initializers for globals and aliases up.
1931f22ef01cSRoman Divacky   ResolveGlobalAndAliasInits();
1932f22ef01cSRoman Divacky   if (!GlobalInits.empty() || !AliasInits.empty())
193339d628a0SDimitry Andric     return Error("Malformed global initializer set");
1934f22ef01cSRoman Divacky 
1935f22ef01cSRoman Divacky   // Look for intrinsic functions which need to be upgraded at some point
1936f22ef01cSRoman Divacky   for (Module::iterator FI = TheModule->begin(), FE = TheModule->end();
1937f22ef01cSRoman Divacky        FI != FE; ++FI) {
1938f22ef01cSRoman Divacky     Function *NewFn;
1939f22ef01cSRoman Divacky     if (UpgradeIntrinsicFunction(FI, NewFn))
1940f22ef01cSRoman Divacky       UpgradedIntrinsics.push_back(std::make_pair(FI, NewFn));
1941f22ef01cSRoman Divacky   }
1942f22ef01cSRoman Divacky 
1943e580952dSDimitry Andric   // Look for global variables which need to be renamed.
1944e580952dSDimitry Andric   for (Module::global_iterator
1945e580952dSDimitry Andric          GI = TheModule->global_begin(), GE = TheModule->global_end();
194691bc56edSDimitry Andric        GI != GE;) {
194791bc56edSDimitry Andric     GlobalVariable *GV = GI++;
194891bc56edSDimitry Andric     UpgradeGlobalVariable(GV);
194991bc56edSDimitry Andric   }
195091bc56edSDimitry Andric 
1951f22ef01cSRoman Divacky   // Force deallocation of memory for these vectors to favor the client that
1952f22ef01cSRoman Divacky   // want lazy deserialization.
1953f22ef01cSRoman Divacky   std::vector<std::pair<GlobalVariable*, unsigned> >().swap(GlobalInits);
1954f22ef01cSRoman Divacky   std::vector<std::pair<GlobalAlias*, unsigned> >().swap(AliasInits);
195591bc56edSDimitry Andric   return std::error_code();
1956f22ef01cSRoman Divacky }
1957f22ef01cSRoman Divacky 
195891bc56edSDimitry Andric std::error_code BitcodeReader::ParseModule(bool Resume) {
1959dff0c46cSDimitry Andric   if (Resume)
1960dff0c46cSDimitry Andric     Stream.JumpToBit(NextUnreadBit);
1961dff0c46cSDimitry Andric   else if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
196239d628a0SDimitry Andric     return Error("Invalid record");
1963dff0c46cSDimitry Andric 
1964dff0c46cSDimitry Andric   SmallVector<uint64_t, 64> Record;
1965dff0c46cSDimitry Andric   std::vector<std::string> SectionTable;
1966dff0c46cSDimitry Andric   std::vector<std::string> GCTable;
1967dff0c46cSDimitry Andric 
1968dff0c46cSDimitry Andric   // Read all the records for this module.
1969139f7f9bSDimitry Andric   while (1) {
1970139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advance();
1971dff0c46cSDimitry Andric 
1972139f7f9bSDimitry Andric     switch (Entry.Kind) {
1973139f7f9bSDimitry Andric     case BitstreamEntry::Error:
197439d628a0SDimitry Andric       return Error("Malformed block");
1975139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
1976dff0c46cSDimitry Andric       return GlobalCleanup();
1977dff0c46cSDimitry Andric 
1978139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock:
1979139f7f9bSDimitry Andric       switch (Entry.ID) {
1980f22ef01cSRoman Divacky       default:  // Skip unknown content.
1981f22ef01cSRoman Divacky         if (Stream.SkipBlock())
198239d628a0SDimitry Andric           return Error("Invalid record");
1983f22ef01cSRoman Divacky         break;
1984f22ef01cSRoman Divacky       case bitc::BLOCKINFO_BLOCK_ID:
1985f22ef01cSRoman Divacky         if (Stream.ReadBlockInfoBlock())
198639d628a0SDimitry Andric           return Error("Malformed block");
1987f22ef01cSRoman Divacky         break;
1988f22ef01cSRoman Divacky       case bitc::PARAMATTR_BLOCK_ID:
198991bc56edSDimitry Andric         if (std::error_code EC = ParseAttributeBlock())
1990f785676fSDimitry Andric           return EC;
1991f22ef01cSRoman Divacky         break;
1992139f7f9bSDimitry Andric       case bitc::PARAMATTR_GROUP_BLOCK_ID:
199391bc56edSDimitry Andric         if (std::error_code EC = ParseAttributeGroupBlock())
1994f785676fSDimitry Andric           return EC;
1995139f7f9bSDimitry Andric         break;
199617a519f9SDimitry Andric       case bitc::TYPE_BLOCK_ID_NEW:
199791bc56edSDimitry Andric         if (std::error_code EC = ParseTypeTable())
1998f785676fSDimitry Andric           return EC;
1999f22ef01cSRoman Divacky         break;
2000f22ef01cSRoman Divacky       case bitc::VALUE_SYMTAB_BLOCK_ID:
200191bc56edSDimitry Andric         if (std::error_code EC = ParseValueSymbolTable())
2002f785676fSDimitry Andric           return EC;
2003dff0c46cSDimitry Andric         SeenValueSymbolTable = true;
2004f22ef01cSRoman Divacky         break;
2005f22ef01cSRoman Divacky       case bitc::CONSTANTS_BLOCK_ID:
200691bc56edSDimitry Andric         if (std::error_code EC = ParseConstants())
2007f785676fSDimitry Andric           return EC;
200891bc56edSDimitry Andric         if (std::error_code EC = ResolveGlobalAndAliasInits())
2009f785676fSDimitry Andric           return EC;
2010f22ef01cSRoman Divacky         break;
2011f22ef01cSRoman Divacky       case bitc::METADATA_BLOCK_ID:
201291bc56edSDimitry Andric         if (std::error_code EC = ParseMetadata())
2013f785676fSDimitry Andric           return EC;
2014f22ef01cSRoman Divacky         break;
2015f22ef01cSRoman Divacky       case bitc::FUNCTION_BLOCK_ID:
2016f22ef01cSRoman Divacky         // If this is the first function body we've seen, reverse the
2017f22ef01cSRoman Divacky         // FunctionsWithBodies list.
2018dff0c46cSDimitry Andric         if (!SeenFirstFunctionBody) {
2019f22ef01cSRoman Divacky           std::reverse(FunctionsWithBodies.begin(), FunctionsWithBodies.end());
202091bc56edSDimitry Andric           if (std::error_code EC = GlobalCleanup())
2021f785676fSDimitry Andric             return EC;
2022dff0c46cSDimitry Andric           SeenFirstFunctionBody = true;
2023f22ef01cSRoman Divacky         }
2024f22ef01cSRoman Divacky 
202591bc56edSDimitry Andric         if (std::error_code EC = RememberAndSkipFunctionBody())
2026f785676fSDimitry Andric           return EC;
2027dff0c46cSDimitry Andric         // For streaming bitcode, suspend parsing when we reach the function
2028dff0c46cSDimitry Andric         // bodies. Subsequent materialization calls will resume it when
2029dff0c46cSDimitry Andric         // necessary. For streaming, the function bodies must be at the end of
2030dff0c46cSDimitry Andric         // the bitcode. If the bitcode file is old, the symbol table will be
2031dff0c46cSDimitry Andric         // at the end instead and will not have been seen yet. In this case,
2032dff0c46cSDimitry Andric         // just finish the parse now.
2033dff0c46cSDimitry Andric         if (LazyStreamer && SeenValueSymbolTable) {
2034dff0c46cSDimitry Andric           NextUnreadBit = Stream.GetCurrentBitNo();
203591bc56edSDimitry Andric           return std::error_code();
2036dff0c46cSDimitry Andric         }
2037dff0c46cSDimitry Andric         break;
2038dff0c46cSDimitry Andric       case bitc::USELIST_BLOCK_ID:
203991bc56edSDimitry Andric         if (std::error_code EC = ParseUseLists())
2040f785676fSDimitry Andric           return EC;
2041f22ef01cSRoman Divacky         break;
2042f22ef01cSRoman Divacky       }
2043f22ef01cSRoman Divacky       continue;
2044139f7f9bSDimitry Andric 
2045139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2046139f7f9bSDimitry Andric       // The interesting case.
2047139f7f9bSDimitry Andric       break;
2048f22ef01cSRoman Divacky     }
2049f22ef01cSRoman Divacky 
2050f22ef01cSRoman Divacky 
2051f22ef01cSRoman Divacky     // Read a record.
2052139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
2053f22ef01cSRoman Divacky     default: break;  // Default behavior, ignore unknown content.
20543861d79fSDimitry Andric     case bitc::MODULE_CODE_VERSION: {  // VERSION: [version#]
2055f22ef01cSRoman Divacky       if (Record.size() < 1)
205639d628a0SDimitry Andric         return Error("Invalid record");
20573861d79fSDimitry Andric       // Only version #0 and #1 are supported so far.
20583861d79fSDimitry Andric       unsigned module_version = Record[0];
20593861d79fSDimitry Andric       switch (module_version) {
2060f785676fSDimitry Andric         default:
206139d628a0SDimitry Andric           return Error("Invalid value");
20623861d79fSDimitry Andric         case 0:
20633861d79fSDimitry Andric           UseRelativeIDs = false;
2064f22ef01cSRoman Divacky           break;
20653861d79fSDimitry Andric         case 1:
20663861d79fSDimitry Andric           UseRelativeIDs = true;
20673861d79fSDimitry Andric           break;
20683861d79fSDimitry Andric       }
20693861d79fSDimitry Andric       break;
20703861d79fSDimitry Andric     }
2071f22ef01cSRoman Divacky     case bitc::MODULE_CODE_TRIPLE: {  // TRIPLE: [strchr x N]
2072f22ef01cSRoman Divacky       std::string S;
2073f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
207439d628a0SDimitry Andric         return Error("Invalid record");
2075f22ef01cSRoman Divacky       TheModule->setTargetTriple(S);
2076f22ef01cSRoman Divacky       break;
2077f22ef01cSRoman Divacky     }
2078f22ef01cSRoman Divacky     case bitc::MODULE_CODE_DATALAYOUT: {  // DATALAYOUT: [strchr x N]
2079f22ef01cSRoman Divacky       std::string S;
2080f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
208139d628a0SDimitry Andric         return Error("Invalid record");
2082f22ef01cSRoman Divacky       TheModule->setDataLayout(S);
2083f22ef01cSRoman Divacky       break;
2084f22ef01cSRoman Divacky     }
2085f22ef01cSRoman Divacky     case bitc::MODULE_CODE_ASM: {  // ASM: [strchr x N]
2086f22ef01cSRoman Divacky       std::string S;
2087f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
208839d628a0SDimitry Andric         return Error("Invalid record");
2089f22ef01cSRoman Divacky       TheModule->setModuleInlineAsm(S);
2090f22ef01cSRoman Divacky       break;
2091f22ef01cSRoman Divacky     }
2092f22ef01cSRoman Divacky     case bitc::MODULE_CODE_DEPLIB: {  // DEPLIB: [strchr x N]
2093139f7f9bSDimitry Andric       // FIXME: Remove in 4.0.
2094f22ef01cSRoman Divacky       std::string S;
2095f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
209639d628a0SDimitry Andric         return Error("Invalid record");
2097139f7f9bSDimitry Andric       // Ignore value.
2098f22ef01cSRoman Divacky       break;
2099f22ef01cSRoman Divacky     }
2100f22ef01cSRoman Divacky     case bitc::MODULE_CODE_SECTIONNAME: {  // SECTIONNAME: [strchr x N]
2101f22ef01cSRoman Divacky       std::string S;
2102f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
210339d628a0SDimitry Andric         return Error("Invalid record");
2104f22ef01cSRoman Divacky       SectionTable.push_back(S);
2105f22ef01cSRoman Divacky       break;
2106f22ef01cSRoman Divacky     }
2107f22ef01cSRoman Divacky     case bitc::MODULE_CODE_GCNAME: {  // SECTIONNAME: [strchr x N]
2108f22ef01cSRoman Divacky       std::string S;
2109f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
211039d628a0SDimitry Andric         return Error("Invalid record");
2111f22ef01cSRoman Divacky       GCTable.push_back(S);
2112f22ef01cSRoman Divacky       break;
2113f22ef01cSRoman Divacky     }
211491bc56edSDimitry Andric     case bitc::MODULE_CODE_COMDAT: { // COMDAT: [selection_kind, name]
211591bc56edSDimitry Andric       if (Record.size() < 2)
211639d628a0SDimitry Andric         return Error("Invalid record");
211791bc56edSDimitry Andric       Comdat::SelectionKind SK = getDecodedComdatSelectionKind(Record[0]);
211891bc56edSDimitry Andric       unsigned ComdatNameSize = Record[1];
211991bc56edSDimitry Andric       std::string ComdatName;
212091bc56edSDimitry Andric       ComdatName.reserve(ComdatNameSize);
212191bc56edSDimitry Andric       for (unsigned i = 0; i != ComdatNameSize; ++i)
212291bc56edSDimitry Andric         ComdatName += (char)Record[2 + i];
212391bc56edSDimitry Andric       Comdat *C = TheModule->getOrInsertComdat(ComdatName);
212491bc56edSDimitry Andric       C->setSelectionKind(SK);
212591bc56edSDimitry Andric       ComdatList.push_back(C);
212691bc56edSDimitry Andric       break;
212791bc56edSDimitry Andric     }
2128f22ef01cSRoman Divacky     // GLOBALVAR: [pointer type, isconst, initid,
21292754fe60SDimitry Andric     //             linkage, alignment, section, visibility, threadlocal,
213091bc56edSDimitry Andric     //             unnamed_addr, dllstorageclass]
2131f22ef01cSRoman Divacky     case bitc::MODULE_CODE_GLOBALVAR: {
2132f22ef01cSRoman Divacky       if (Record.size() < 6)
213339d628a0SDimitry Andric         return Error("Invalid record");
21346122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
2135f785676fSDimitry Andric       if (!Ty)
213639d628a0SDimitry Andric         return Error("Invalid record");
2137f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
213839d628a0SDimitry Andric         return Error("Invalid type for value");
2139f22ef01cSRoman Divacky       unsigned AddressSpace = cast<PointerType>(Ty)->getAddressSpace();
2140f22ef01cSRoman Divacky       Ty = cast<PointerType>(Ty)->getElementType();
2141f22ef01cSRoman Divacky 
2142f22ef01cSRoman Divacky       bool isConstant = Record[1];
214339d628a0SDimitry Andric       GlobalValue::LinkageTypes Linkage = getDecodedLinkage(Record[3]);
2144f22ef01cSRoman Divacky       unsigned Alignment = (1 << Record[4]) >> 1;
2145f22ef01cSRoman Divacky       std::string Section;
2146f22ef01cSRoman Divacky       if (Record[5]) {
2147f22ef01cSRoman Divacky         if (Record[5]-1 >= SectionTable.size())
214839d628a0SDimitry Andric           return Error("Invalid ID");
2149f22ef01cSRoman Divacky         Section = SectionTable[Record[5]-1];
2150f22ef01cSRoman Divacky       }
2151f22ef01cSRoman Divacky       GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
215291bc56edSDimitry Andric       // Local linkage must have default visibility.
215391bc56edSDimitry Andric       if (Record.size() > 6 && !GlobalValue::isLocalLinkage(Linkage))
215491bc56edSDimitry Andric         // FIXME: Change to an error if non-default in 4.0.
2155f22ef01cSRoman Divacky         Visibility = GetDecodedVisibility(Record[6]);
21567ae0e2c9SDimitry Andric 
21577ae0e2c9SDimitry Andric       GlobalVariable::ThreadLocalMode TLM = GlobalVariable::NotThreadLocal;
2158f22ef01cSRoman Divacky       if (Record.size() > 7)
21597ae0e2c9SDimitry Andric         TLM = GetDecodedThreadLocalMode(Record[7]);
2160f22ef01cSRoman Divacky 
21612754fe60SDimitry Andric       bool UnnamedAddr = false;
21622754fe60SDimitry Andric       if (Record.size() > 8)
21632754fe60SDimitry Andric         UnnamedAddr = Record[8];
21642754fe60SDimitry Andric 
2165139f7f9bSDimitry Andric       bool ExternallyInitialized = false;
2166139f7f9bSDimitry Andric       if (Record.size() > 9)
2167139f7f9bSDimitry Andric         ExternallyInitialized = Record[9];
2168139f7f9bSDimitry Andric 
2169f22ef01cSRoman Divacky       GlobalVariable *NewGV =
217091bc56edSDimitry Andric         new GlobalVariable(*TheModule, Ty, isConstant, Linkage, nullptr, "", nullptr,
2171139f7f9bSDimitry Andric                            TLM, AddressSpace, ExternallyInitialized);
2172f22ef01cSRoman Divacky       NewGV->setAlignment(Alignment);
2173f22ef01cSRoman Divacky       if (!Section.empty())
2174f22ef01cSRoman Divacky         NewGV->setSection(Section);
2175f22ef01cSRoman Divacky       NewGV->setVisibility(Visibility);
21762754fe60SDimitry Andric       NewGV->setUnnamedAddr(UnnamedAddr);
2177f22ef01cSRoman Divacky 
217891bc56edSDimitry Andric       if (Record.size() > 10)
217991bc56edSDimitry Andric         NewGV->setDLLStorageClass(GetDecodedDLLStorageClass(Record[10]));
218091bc56edSDimitry Andric       else
218191bc56edSDimitry Andric         UpgradeDLLImportExportLinkage(NewGV, Record[3]);
218291bc56edSDimitry Andric 
2183f22ef01cSRoman Divacky       ValueList.push_back(NewGV);
2184f22ef01cSRoman Divacky 
2185f22ef01cSRoman Divacky       // Remember which value to use for the global initializer.
2186f22ef01cSRoman Divacky       if (unsigned InitID = Record[2])
2187f22ef01cSRoman Divacky         GlobalInits.push_back(std::make_pair(NewGV, InitID-1));
218891bc56edSDimitry Andric 
218991bc56edSDimitry Andric       if (Record.size() > 11)
219091bc56edSDimitry Andric         if (unsigned ComdatID = Record[11]) {
219191bc56edSDimitry Andric           assert(ComdatID <= ComdatList.size());
219291bc56edSDimitry Andric           NewGV->setComdat(ComdatList[ComdatID - 1]);
219391bc56edSDimitry Andric         }
2194f22ef01cSRoman Divacky       break;
2195f22ef01cSRoman Divacky     }
2196f22ef01cSRoman Divacky     // FUNCTION:  [type, callingconv, isproto, linkage, paramattr,
219791bc56edSDimitry Andric     //             alignment, section, visibility, gc, unnamed_addr,
219839d628a0SDimitry Andric     //             prologuedata, dllstorageclass, comdat, prefixdata]
2199f22ef01cSRoman Divacky     case bitc::MODULE_CODE_FUNCTION: {
2200f22ef01cSRoman Divacky       if (Record.size() < 8)
220139d628a0SDimitry Andric         return Error("Invalid record");
22026122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
2203f785676fSDimitry Andric       if (!Ty)
220439d628a0SDimitry Andric         return Error("Invalid record");
2205f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
220639d628a0SDimitry Andric         return Error("Invalid type for value");
22076122f3e6SDimitry Andric       FunctionType *FTy =
2208f22ef01cSRoman Divacky         dyn_cast<FunctionType>(cast<PointerType>(Ty)->getElementType());
2209f22ef01cSRoman Divacky       if (!FTy)
221039d628a0SDimitry Andric         return Error("Invalid type for value");
2211f22ef01cSRoman Divacky 
2212f22ef01cSRoman Divacky       Function *Func = Function::Create(FTy, GlobalValue::ExternalLinkage,
2213f22ef01cSRoman Divacky                                         "", TheModule);
2214f22ef01cSRoman Divacky 
2215f22ef01cSRoman Divacky       Func->setCallingConv(static_cast<CallingConv::ID>(Record[1]));
2216f22ef01cSRoman Divacky       bool isProto = Record[2];
221739d628a0SDimitry Andric       Func->setLinkage(getDecodedLinkage(Record[3]));
2218f22ef01cSRoman Divacky       Func->setAttributes(getAttributes(Record[4]));
2219f22ef01cSRoman Divacky 
2220f22ef01cSRoman Divacky       Func->setAlignment((1 << Record[5]) >> 1);
2221f22ef01cSRoman Divacky       if (Record[6]) {
2222f22ef01cSRoman Divacky         if (Record[6]-1 >= SectionTable.size())
222339d628a0SDimitry Andric           return Error("Invalid ID");
2224f22ef01cSRoman Divacky         Func->setSection(SectionTable[Record[6]-1]);
2225f22ef01cSRoman Divacky       }
222691bc56edSDimitry Andric       // Local linkage must have default visibility.
222791bc56edSDimitry Andric       if (!Func->hasLocalLinkage())
222891bc56edSDimitry Andric         // FIXME: Change to an error if non-default in 4.0.
2229f22ef01cSRoman Divacky         Func->setVisibility(GetDecodedVisibility(Record[7]));
2230f22ef01cSRoman Divacky       if (Record.size() > 8 && Record[8]) {
2231f22ef01cSRoman Divacky         if (Record[8]-1 > GCTable.size())
223239d628a0SDimitry Andric           return Error("Invalid ID");
2233f22ef01cSRoman Divacky         Func->setGC(GCTable[Record[8]-1].c_str());
2234f22ef01cSRoman Divacky       }
22352754fe60SDimitry Andric       bool UnnamedAddr = false;
22362754fe60SDimitry Andric       if (Record.size() > 9)
22372754fe60SDimitry Andric         UnnamedAddr = Record[9];
22382754fe60SDimitry Andric       Func->setUnnamedAddr(UnnamedAddr);
2239f785676fSDimitry Andric       if (Record.size() > 10 && Record[10] != 0)
224039d628a0SDimitry Andric         FunctionPrologues.push_back(std::make_pair(Func, Record[10]-1));
224191bc56edSDimitry Andric 
224291bc56edSDimitry Andric       if (Record.size() > 11)
224391bc56edSDimitry Andric         Func->setDLLStorageClass(GetDecodedDLLStorageClass(Record[11]));
224491bc56edSDimitry Andric       else
224591bc56edSDimitry Andric         UpgradeDLLImportExportLinkage(Func, Record[3]);
224691bc56edSDimitry Andric 
224791bc56edSDimitry Andric       if (Record.size() > 12)
224891bc56edSDimitry Andric         if (unsigned ComdatID = Record[12]) {
224991bc56edSDimitry Andric           assert(ComdatID <= ComdatList.size());
225091bc56edSDimitry Andric           Func->setComdat(ComdatList[ComdatID - 1]);
225191bc56edSDimitry Andric         }
225291bc56edSDimitry Andric 
225339d628a0SDimitry Andric       if (Record.size() > 13 && Record[13] != 0)
225439d628a0SDimitry Andric         FunctionPrefixes.push_back(std::make_pair(Func, Record[13]-1));
225539d628a0SDimitry Andric 
2256f22ef01cSRoman Divacky       ValueList.push_back(Func);
2257f22ef01cSRoman Divacky 
2258f22ef01cSRoman Divacky       // If this is a function with a body, remember the prototype we are
2259f22ef01cSRoman Divacky       // creating now, so that we can match up the body with them later.
2260dff0c46cSDimitry Andric       if (!isProto) {
226139d628a0SDimitry Andric         Func->setIsMaterializable(true);
2262f22ef01cSRoman Divacky         FunctionsWithBodies.push_back(Func);
226339d628a0SDimitry Andric         if (LazyStreamer)
226439d628a0SDimitry Andric           DeferredFunctionInfo[Func] = 0;
2265dff0c46cSDimitry Andric       }
2266f22ef01cSRoman Divacky       break;
2267f22ef01cSRoman Divacky     }
2268f22ef01cSRoman Divacky     // ALIAS: [alias type, aliasee val#, linkage]
226991bc56edSDimitry Andric     // ALIAS: [alias type, aliasee val#, linkage, visibility, dllstorageclass]
2270f22ef01cSRoman Divacky     case bitc::MODULE_CODE_ALIAS: {
2271f22ef01cSRoman Divacky       if (Record.size() < 3)
227239d628a0SDimitry Andric         return Error("Invalid record");
22736122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
2274f785676fSDimitry Andric       if (!Ty)
227539d628a0SDimitry Andric         return Error("Invalid record");
227691bc56edSDimitry Andric       auto *PTy = dyn_cast<PointerType>(Ty);
227791bc56edSDimitry Andric       if (!PTy)
227839d628a0SDimitry Andric         return Error("Invalid type for value");
2279f22ef01cSRoman Divacky 
228091bc56edSDimitry Andric       auto *NewGA =
228191bc56edSDimitry Andric           GlobalAlias::create(PTy->getElementType(), PTy->getAddressSpace(),
228239d628a0SDimitry Andric                               getDecodedLinkage(Record[2]), "", TheModule);
2283f22ef01cSRoman Divacky       // Old bitcode files didn't have visibility field.
228491bc56edSDimitry Andric       // Local linkage must have default visibility.
228591bc56edSDimitry Andric       if (Record.size() > 3 && !NewGA->hasLocalLinkage())
228691bc56edSDimitry Andric         // FIXME: Change to an error if non-default in 4.0.
2287f22ef01cSRoman Divacky         NewGA->setVisibility(GetDecodedVisibility(Record[3]));
228891bc56edSDimitry Andric       if (Record.size() > 4)
228991bc56edSDimitry Andric         NewGA->setDLLStorageClass(GetDecodedDLLStorageClass(Record[4]));
229091bc56edSDimitry Andric       else
229191bc56edSDimitry Andric         UpgradeDLLImportExportLinkage(NewGA, Record[2]);
229291bc56edSDimitry Andric       if (Record.size() > 5)
229391bc56edSDimitry Andric         NewGA->setThreadLocalMode(GetDecodedThreadLocalMode(Record[5]));
229491bc56edSDimitry Andric       if (Record.size() > 6)
229591bc56edSDimitry Andric         NewGA->setUnnamedAddr(Record[6]);
2296f22ef01cSRoman Divacky       ValueList.push_back(NewGA);
2297f22ef01cSRoman Divacky       AliasInits.push_back(std::make_pair(NewGA, Record[1]));
2298f22ef01cSRoman Divacky       break;
2299f22ef01cSRoman Divacky     }
2300f22ef01cSRoman Divacky     /// MODULE_CODE_PURGEVALS: [numvals]
2301f22ef01cSRoman Divacky     case bitc::MODULE_CODE_PURGEVALS:
2302f22ef01cSRoman Divacky       // Trim down the value list to the specified size.
2303f22ef01cSRoman Divacky       if (Record.size() < 1 || Record[0] > ValueList.size())
230439d628a0SDimitry Andric         return Error("Invalid record");
2305f22ef01cSRoman Divacky       ValueList.shrinkTo(Record[0]);
2306f22ef01cSRoman Divacky       break;
2307f22ef01cSRoman Divacky     }
2308f22ef01cSRoman Divacky     Record.clear();
2309f22ef01cSRoman Divacky   }
2310f22ef01cSRoman Divacky }
2311f22ef01cSRoman Divacky 
231291bc56edSDimitry Andric std::error_code BitcodeReader::ParseBitcodeInto(Module *M) {
231391bc56edSDimitry Andric   TheModule = nullptr;
2314f22ef01cSRoman Divacky 
231591bc56edSDimitry Andric   if (std::error_code EC = InitStream())
2316f785676fSDimitry Andric     return EC;
2317f22ef01cSRoman Divacky 
2318f22ef01cSRoman Divacky   // Sniff for the signature.
2319f22ef01cSRoman Divacky   if (Stream.Read(8) != 'B' ||
2320f22ef01cSRoman Divacky       Stream.Read(8) != 'C' ||
2321f22ef01cSRoman Divacky       Stream.Read(4) != 0x0 ||
2322f22ef01cSRoman Divacky       Stream.Read(4) != 0xC ||
2323f22ef01cSRoman Divacky       Stream.Read(4) != 0xE ||
2324f22ef01cSRoman Divacky       Stream.Read(4) != 0xD)
232539d628a0SDimitry Andric     return Error("Invalid bitcode signature");
2326f22ef01cSRoman Divacky 
2327f22ef01cSRoman Divacky   // We expect a number of well-defined blocks, though we don't necessarily
2328f22ef01cSRoman Divacky   // need to understand them all.
2329139f7f9bSDimitry Andric   while (1) {
2330139f7f9bSDimitry Andric     if (Stream.AtEndOfStream())
233191bc56edSDimitry Andric       return std::error_code();
2332bd5abe19SDimitry Andric 
2333139f7f9bSDimitry Andric     BitstreamEntry Entry =
2334139f7f9bSDimitry Andric       Stream.advance(BitstreamCursor::AF_DontAutoprocessAbbrevs);
2335f22ef01cSRoman Divacky 
2336139f7f9bSDimitry Andric     switch (Entry.Kind) {
2337139f7f9bSDimitry Andric     case BitstreamEntry::Error:
233839d628a0SDimitry Andric       return Error("Malformed block");
2339139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
234091bc56edSDimitry Andric       return std::error_code();
2341f22ef01cSRoman Divacky 
2342139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock:
2343139f7f9bSDimitry Andric       switch (Entry.ID) {
2344f22ef01cSRoman Divacky       case bitc::BLOCKINFO_BLOCK_ID:
2345f22ef01cSRoman Divacky         if (Stream.ReadBlockInfoBlock())
234639d628a0SDimitry Andric           return Error("Malformed block");
2347f22ef01cSRoman Divacky         break;
2348f22ef01cSRoman Divacky       case bitc::MODULE_BLOCK_ID:
2349f22ef01cSRoman Divacky         // Reject multiple MODULE_BLOCK's in a single bitstream.
2350f22ef01cSRoman Divacky         if (TheModule)
235139d628a0SDimitry Andric           return Error("Invalid multiple blocks");
2352f22ef01cSRoman Divacky         TheModule = M;
235391bc56edSDimitry Andric         if (std::error_code EC = ParseModule(false))
2354f785676fSDimitry Andric           return EC;
2355f785676fSDimitry Andric         if (LazyStreamer)
235691bc56edSDimitry Andric           return std::error_code();
2357f22ef01cSRoman Divacky         break;
2358f22ef01cSRoman Divacky       default:
2359f22ef01cSRoman Divacky         if (Stream.SkipBlock())
236039d628a0SDimitry Andric           return Error("Invalid record");
2361f22ef01cSRoman Divacky         break;
2362f22ef01cSRoman Divacky       }
2363139f7f9bSDimitry Andric       continue;
2364139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2365139f7f9bSDimitry Andric       // There should be no records in the top-level of blocks.
2366f22ef01cSRoman Divacky 
2367139f7f9bSDimitry Andric       // The ranlib in Xcode 4 will align archive members by appending newlines
2368139f7f9bSDimitry Andric       // to the end of them. If this file size is a multiple of 4 but not 8, we
2369139f7f9bSDimitry Andric       // have to read and ignore these final 4 bytes :-(
2370139f7f9bSDimitry Andric       if (Stream.getAbbrevIDWidth() == 2 && Entry.ID == 2 &&
2371139f7f9bSDimitry Andric           Stream.Read(6) == 2 && Stream.Read(24) == 0xa0a0a &&
2372139f7f9bSDimitry Andric           Stream.AtEndOfStream())
237391bc56edSDimitry Andric         return std::error_code();
2374139f7f9bSDimitry Andric 
237539d628a0SDimitry Andric       return Error("Invalid record");
2376139f7f9bSDimitry Andric     }
2377139f7f9bSDimitry Andric   }
2378f22ef01cSRoman Divacky }
2379f22ef01cSRoman Divacky 
238091bc56edSDimitry Andric ErrorOr<std::string> BitcodeReader::parseModuleTriple() {
23812754fe60SDimitry Andric   if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
238239d628a0SDimitry Andric     return Error("Invalid record");
23832754fe60SDimitry Andric 
23842754fe60SDimitry Andric   SmallVector<uint64_t, 64> Record;
23852754fe60SDimitry Andric 
238691bc56edSDimitry Andric   std::string Triple;
23872754fe60SDimitry Andric   // Read all the records for this module.
2388139f7f9bSDimitry Andric   while (1) {
2389139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
23902754fe60SDimitry Andric 
2391139f7f9bSDimitry Andric     switch (Entry.Kind) {
2392139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
2393139f7f9bSDimitry Andric     case BitstreamEntry::Error:
239439d628a0SDimitry Andric       return Error("Malformed block");
2395139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
239691bc56edSDimitry Andric       return Triple;
2397139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2398139f7f9bSDimitry Andric       // The interesting case.
23992754fe60SDimitry Andric       break;
24002754fe60SDimitry Andric     }
24012754fe60SDimitry Andric 
24022754fe60SDimitry Andric     // Read a record.
2403139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
24042754fe60SDimitry Andric     default: break;  // Default behavior, ignore unknown content.
24052754fe60SDimitry Andric     case bitc::MODULE_CODE_TRIPLE: {  // TRIPLE: [strchr x N]
24062754fe60SDimitry Andric       std::string S;
24072754fe60SDimitry Andric       if (ConvertToString(Record, 0, S))
240839d628a0SDimitry Andric         return Error("Invalid record");
24092754fe60SDimitry Andric       Triple = S;
24102754fe60SDimitry Andric       break;
24112754fe60SDimitry Andric     }
24122754fe60SDimitry Andric     }
24132754fe60SDimitry Andric     Record.clear();
24142754fe60SDimitry Andric   }
241591bc56edSDimitry Andric   llvm_unreachable("Exit infinite loop");
24162754fe60SDimitry Andric }
24172754fe60SDimitry Andric 
241891bc56edSDimitry Andric ErrorOr<std::string> BitcodeReader::parseTriple() {
241991bc56edSDimitry Andric   if (std::error_code EC = InitStream())
2420f785676fSDimitry Andric     return EC;
24212754fe60SDimitry Andric 
24222754fe60SDimitry Andric   // Sniff for the signature.
24232754fe60SDimitry Andric   if (Stream.Read(8) != 'B' ||
24242754fe60SDimitry Andric       Stream.Read(8) != 'C' ||
24252754fe60SDimitry Andric       Stream.Read(4) != 0x0 ||
24262754fe60SDimitry Andric       Stream.Read(4) != 0xC ||
24272754fe60SDimitry Andric       Stream.Read(4) != 0xE ||
24282754fe60SDimitry Andric       Stream.Read(4) != 0xD)
242939d628a0SDimitry Andric     return Error("Invalid bitcode signature");
24302754fe60SDimitry Andric 
24312754fe60SDimitry Andric   // We expect a number of well-defined blocks, though we don't necessarily
24322754fe60SDimitry Andric   // need to understand them all.
2433139f7f9bSDimitry Andric   while (1) {
2434139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advance();
24352754fe60SDimitry Andric 
2436139f7f9bSDimitry Andric     switch (Entry.Kind) {
2437139f7f9bSDimitry Andric     case BitstreamEntry::Error:
243839d628a0SDimitry Andric       return Error("Malformed block");
2439139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
244091bc56edSDimitry Andric       return std::error_code();
2441139f7f9bSDimitry Andric 
2442139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock:
2443139f7f9bSDimitry Andric       if (Entry.ID == bitc::MODULE_BLOCK_ID)
244491bc56edSDimitry Andric         return parseModuleTriple();
2445139f7f9bSDimitry Andric 
2446139f7f9bSDimitry Andric       // Ignore other sub-blocks.
2447f785676fSDimitry Andric       if (Stream.SkipBlock())
244839d628a0SDimitry Andric         return Error("Malformed block");
2449139f7f9bSDimitry Andric       continue;
2450139f7f9bSDimitry Andric 
2451139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2452139f7f9bSDimitry Andric       Stream.skipRecord(Entry.ID);
2453139f7f9bSDimitry Andric       continue;
2454139f7f9bSDimitry Andric     }
2455139f7f9bSDimitry Andric   }
24562754fe60SDimitry Andric }
24572754fe60SDimitry Andric 
2458f22ef01cSRoman Divacky /// ParseMetadataAttachment - Parse metadata attachments.
245991bc56edSDimitry Andric std::error_code BitcodeReader::ParseMetadataAttachment() {
2460f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::METADATA_ATTACHMENT_ID))
246139d628a0SDimitry Andric     return Error("Invalid record");
2462f22ef01cSRoman Divacky 
2463f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
2464f22ef01cSRoman Divacky   while (1) {
2465139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
2466139f7f9bSDimitry Andric 
2467139f7f9bSDimitry Andric     switch (Entry.Kind) {
2468139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
2469139f7f9bSDimitry Andric     case BitstreamEntry::Error:
247039d628a0SDimitry Andric       return Error("Malformed block");
2471139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
247291bc56edSDimitry Andric       return std::error_code();
2473139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2474139f7f9bSDimitry Andric       // The interesting case.
2475f22ef01cSRoman Divacky       break;
2476f22ef01cSRoman Divacky     }
2477139f7f9bSDimitry Andric 
2478f22ef01cSRoman Divacky     // Read a metadata attachment record.
2479f22ef01cSRoman Divacky     Record.clear();
2480139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
2481f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
2482f22ef01cSRoman Divacky       break;
248317a519f9SDimitry Andric     case bitc::METADATA_ATTACHMENT: {
2484f22ef01cSRoman Divacky       unsigned RecordLength = Record.size();
2485f22ef01cSRoman Divacky       if (Record.empty() || (RecordLength - 1) % 2 == 1)
248639d628a0SDimitry Andric         return Error("Invalid record");
2487f22ef01cSRoman Divacky       Instruction *Inst = InstructionList[Record[0]];
2488f22ef01cSRoman Divacky       for (unsigned i = 1; i != RecordLength; i = i+2) {
2489f22ef01cSRoman Divacky         unsigned Kind = Record[i];
2490e580952dSDimitry Andric         DenseMap<unsigned, unsigned>::iterator I =
2491e580952dSDimitry Andric           MDKindMap.find(Kind);
2492e580952dSDimitry Andric         if (I == MDKindMap.end())
249339d628a0SDimitry Andric           return Error("Invalid ID");
249439d628a0SDimitry Andric         Metadata *Node = MDValueList.getValueFwdRef(Record[i + 1]);
249539d628a0SDimitry Andric         if (isa<LocalAsMetadata>(Node))
249639d628a0SDimitry Andric           // Drop the attachment.  This used to be legal, but there's no
249739d628a0SDimitry Andric           // upgrade path.
249839d628a0SDimitry Andric           break;
2499e580952dSDimitry Andric         Inst->setMetadata(I->second, cast<MDNode>(Node));
2500f785676fSDimitry Andric         if (I->second == LLVMContext::MD_tbaa)
2501f785676fSDimitry Andric           InstsWithTBAATag.push_back(Inst);
2502f22ef01cSRoman Divacky       }
2503f22ef01cSRoman Divacky       break;
2504f22ef01cSRoman Divacky     }
2505f22ef01cSRoman Divacky     }
2506f22ef01cSRoman Divacky   }
2507f22ef01cSRoman Divacky }
2508f22ef01cSRoman Divacky 
2509f22ef01cSRoman Divacky /// ParseFunctionBody - Lazily parse the specified function body block.
251091bc56edSDimitry Andric std::error_code BitcodeReader::ParseFunctionBody(Function *F) {
2511f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::FUNCTION_BLOCK_ID))
251239d628a0SDimitry Andric     return Error("Invalid record");
2513f22ef01cSRoman Divacky 
2514f22ef01cSRoman Divacky   InstructionList.clear();
2515f22ef01cSRoman Divacky   unsigned ModuleValueListSize = ValueList.size();
2516e580952dSDimitry Andric   unsigned ModuleMDValueListSize = MDValueList.size();
2517f22ef01cSRoman Divacky 
2518f22ef01cSRoman Divacky   // Add all the function arguments to the value table.
2519f22ef01cSRoman Divacky   for(Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
2520f22ef01cSRoman Divacky     ValueList.push_back(I);
2521f22ef01cSRoman Divacky 
2522f22ef01cSRoman Divacky   unsigned NextValueNo = ValueList.size();
252391bc56edSDimitry Andric   BasicBlock *CurBB = nullptr;
2524f22ef01cSRoman Divacky   unsigned CurBBNo = 0;
2525f22ef01cSRoman Divacky 
2526f22ef01cSRoman Divacky   DebugLoc LastLoc;
252739d628a0SDimitry Andric   auto getLastInstruction = [&]() -> Instruction * {
252839d628a0SDimitry Andric     if (CurBB && !CurBB->empty())
252939d628a0SDimitry Andric       return &CurBB->back();
253039d628a0SDimitry Andric     else if (CurBBNo && FunctionBBs[CurBBNo - 1] &&
253139d628a0SDimitry Andric              !FunctionBBs[CurBBNo - 1]->empty())
253239d628a0SDimitry Andric       return &FunctionBBs[CurBBNo - 1]->back();
253339d628a0SDimitry Andric     return nullptr;
253439d628a0SDimitry Andric   };
2535f22ef01cSRoman Divacky 
2536f22ef01cSRoman Divacky   // Read all the records.
2537f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
2538f22ef01cSRoman Divacky   while (1) {
2539139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advance();
2540f22ef01cSRoman Divacky 
2541139f7f9bSDimitry Andric     switch (Entry.Kind) {
2542139f7f9bSDimitry Andric     case BitstreamEntry::Error:
254339d628a0SDimitry Andric       return Error("Malformed block");
2544139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
2545139f7f9bSDimitry Andric       goto OutOfRecordLoop;
2546139f7f9bSDimitry Andric 
2547139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock:
2548139f7f9bSDimitry Andric       switch (Entry.ID) {
2549f22ef01cSRoman Divacky       default:  // Skip unknown content.
2550f22ef01cSRoman Divacky         if (Stream.SkipBlock())
255139d628a0SDimitry Andric           return Error("Invalid record");
2552f22ef01cSRoman Divacky         break;
2553f22ef01cSRoman Divacky       case bitc::CONSTANTS_BLOCK_ID:
255491bc56edSDimitry Andric         if (std::error_code EC = ParseConstants())
2555f785676fSDimitry Andric           return EC;
2556f22ef01cSRoman Divacky         NextValueNo = ValueList.size();
2557f22ef01cSRoman Divacky         break;
2558f22ef01cSRoman Divacky       case bitc::VALUE_SYMTAB_BLOCK_ID:
255991bc56edSDimitry Andric         if (std::error_code EC = ParseValueSymbolTable())
2560f785676fSDimitry Andric           return EC;
2561f22ef01cSRoman Divacky         break;
2562f22ef01cSRoman Divacky       case bitc::METADATA_ATTACHMENT_ID:
256391bc56edSDimitry Andric         if (std::error_code EC = ParseMetadataAttachment())
2564f785676fSDimitry Andric           return EC;
2565f22ef01cSRoman Divacky         break;
2566f22ef01cSRoman Divacky       case bitc::METADATA_BLOCK_ID:
256791bc56edSDimitry Andric         if (std::error_code EC = ParseMetadata())
2568f785676fSDimitry Andric           return EC;
2569f22ef01cSRoman Divacky         break;
257039d628a0SDimitry Andric       case bitc::USELIST_BLOCK_ID:
257139d628a0SDimitry Andric         if (std::error_code EC = ParseUseLists())
257239d628a0SDimitry Andric           return EC;
257339d628a0SDimitry Andric         break;
2574f22ef01cSRoman Divacky       }
2575f22ef01cSRoman Divacky       continue;
2576f22ef01cSRoman Divacky 
2577139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2578139f7f9bSDimitry Andric       // The interesting case.
2579139f7f9bSDimitry Andric       break;
2580f22ef01cSRoman Divacky     }
2581f22ef01cSRoman Divacky 
2582f22ef01cSRoman Divacky     // Read a record.
2583f22ef01cSRoman Divacky     Record.clear();
258491bc56edSDimitry Andric     Instruction *I = nullptr;
2585139f7f9bSDimitry Andric     unsigned BitCode = Stream.readRecord(Entry.ID, Record);
2586f22ef01cSRoman Divacky     switch (BitCode) {
2587f22ef01cSRoman Divacky     default: // Default behavior: reject
258839d628a0SDimitry Andric       return Error("Invalid value");
258939d628a0SDimitry Andric     case bitc::FUNC_CODE_DECLAREBLOCKS: {   // DECLAREBLOCKS: [nblocks]
2590f22ef01cSRoman Divacky       if (Record.size() < 1 || Record[0] == 0)
259139d628a0SDimitry Andric         return Error("Invalid record");
2592f22ef01cSRoman Divacky       // Create all the basic blocks for the function.
2593f22ef01cSRoman Divacky       FunctionBBs.resize(Record[0]);
259439d628a0SDimitry Andric 
259539d628a0SDimitry Andric       // See if anything took the address of blocks in this function.
259639d628a0SDimitry Andric       auto BBFRI = BasicBlockFwdRefs.find(F);
259739d628a0SDimitry Andric       if (BBFRI == BasicBlockFwdRefs.end()) {
2598f22ef01cSRoman Divacky         for (unsigned i = 0, e = FunctionBBs.size(); i != e; ++i)
2599f22ef01cSRoman Divacky           FunctionBBs[i] = BasicBlock::Create(Context, "", F);
260039d628a0SDimitry Andric       } else {
260139d628a0SDimitry Andric         auto &BBRefs = BBFRI->second;
260239d628a0SDimitry Andric         // Check for invalid basic block references.
260339d628a0SDimitry Andric         if (BBRefs.size() > FunctionBBs.size())
260439d628a0SDimitry Andric           return Error("Invalid ID");
260539d628a0SDimitry Andric         assert(!BBRefs.empty() && "Unexpected empty array");
260639d628a0SDimitry Andric         assert(!BBRefs.front() && "Invalid reference to entry block");
260739d628a0SDimitry Andric         for (unsigned I = 0, E = FunctionBBs.size(), RE = BBRefs.size(); I != E;
260839d628a0SDimitry Andric              ++I)
260939d628a0SDimitry Andric           if (I < RE && BBRefs[I]) {
261039d628a0SDimitry Andric             BBRefs[I]->insertInto(F);
261139d628a0SDimitry Andric             FunctionBBs[I] = BBRefs[I];
261239d628a0SDimitry Andric           } else {
261339d628a0SDimitry Andric             FunctionBBs[I] = BasicBlock::Create(Context, "", F);
261439d628a0SDimitry Andric           }
261539d628a0SDimitry Andric 
261639d628a0SDimitry Andric         // Erase from the table.
261739d628a0SDimitry Andric         BasicBlockFwdRefs.erase(BBFRI);
261839d628a0SDimitry Andric       }
261939d628a0SDimitry Andric 
2620f22ef01cSRoman Divacky       CurBB = FunctionBBs[0];
2621f22ef01cSRoman Divacky       continue;
262239d628a0SDimitry Andric     }
2623f22ef01cSRoman Divacky 
2624f22ef01cSRoman Divacky     case bitc::FUNC_CODE_DEBUG_LOC_AGAIN:  // DEBUG_LOC_AGAIN
2625f22ef01cSRoman Divacky       // This record indicates that the last instruction is at the same
2626f22ef01cSRoman Divacky       // location as the previous instruction with a location.
262739d628a0SDimitry Andric       I = getLastInstruction();
2628f22ef01cSRoman Divacky 
262991bc56edSDimitry Andric       if (!I)
263039d628a0SDimitry Andric         return Error("Invalid record");
2631f22ef01cSRoman Divacky       I->setDebugLoc(LastLoc);
263291bc56edSDimitry Andric       I = nullptr;
2633f22ef01cSRoman Divacky       continue;
2634f22ef01cSRoman Divacky 
263517a519f9SDimitry Andric     case bitc::FUNC_CODE_DEBUG_LOC: {      // DEBUG_LOC: [line, col, scope, ia]
263639d628a0SDimitry Andric       I = getLastInstruction();
263791bc56edSDimitry Andric       if (!I || Record.size() < 4)
263839d628a0SDimitry Andric         return Error("Invalid record");
2639f22ef01cSRoman Divacky 
2640f22ef01cSRoman Divacky       unsigned Line = Record[0], Col = Record[1];
2641f22ef01cSRoman Divacky       unsigned ScopeID = Record[2], IAID = Record[3];
2642f22ef01cSRoman Divacky 
264391bc56edSDimitry Andric       MDNode *Scope = nullptr, *IA = nullptr;
2644f22ef01cSRoman Divacky       if (ScopeID) Scope = cast<MDNode>(MDValueList.getValueFwdRef(ScopeID-1));
2645f22ef01cSRoman Divacky       if (IAID)    IA = cast<MDNode>(MDValueList.getValueFwdRef(IAID-1));
2646f22ef01cSRoman Divacky       LastLoc = DebugLoc::get(Line, Col, Scope, IA);
2647f22ef01cSRoman Divacky       I->setDebugLoc(LastLoc);
264891bc56edSDimitry Andric       I = nullptr;
2649f22ef01cSRoman Divacky       continue;
2650f22ef01cSRoman Divacky     }
2651f22ef01cSRoman Divacky 
2652f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_BINOP: {    // BINOP: [opval, ty, opval, opcode]
2653f22ef01cSRoman Divacky       unsigned OpNum = 0;
2654f22ef01cSRoman Divacky       Value *LHS, *RHS;
2655f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
26563861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, LHS->getType(), RHS) ||
2657f22ef01cSRoman Divacky           OpNum+1 > Record.size())
265839d628a0SDimitry Andric         return Error("Invalid record");
2659f22ef01cSRoman Divacky 
2660f22ef01cSRoman Divacky       int Opc = GetDecodedBinaryOpcode(Record[OpNum++], LHS->getType());
2661f785676fSDimitry Andric       if (Opc == -1)
266239d628a0SDimitry Andric         return Error("Invalid record");
2663f22ef01cSRoman Divacky       I = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
2664f22ef01cSRoman Divacky       InstructionList.push_back(I);
2665f22ef01cSRoman Divacky       if (OpNum < Record.size()) {
2666f22ef01cSRoman Divacky         if (Opc == Instruction::Add ||
2667f22ef01cSRoman Divacky             Opc == Instruction::Sub ||
26682754fe60SDimitry Andric             Opc == Instruction::Mul ||
26692754fe60SDimitry Andric             Opc == Instruction::Shl) {
2670f22ef01cSRoman Divacky           if (Record[OpNum] & (1 << bitc::OBO_NO_SIGNED_WRAP))
2671f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setHasNoSignedWrap(true);
2672f22ef01cSRoman Divacky           if (Record[OpNum] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
2673f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setHasNoUnsignedWrap(true);
26742754fe60SDimitry Andric         } else if (Opc == Instruction::SDiv ||
26752754fe60SDimitry Andric                    Opc == Instruction::UDiv ||
26762754fe60SDimitry Andric                    Opc == Instruction::LShr ||
26772754fe60SDimitry Andric                    Opc == Instruction::AShr) {
26782754fe60SDimitry Andric           if (Record[OpNum] & (1 << bitc::PEO_EXACT))
2679f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setIsExact(true);
2680139f7f9bSDimitry Andric         } else if (isa<FPMathOperator>(I)) {
2681139f7f9bSDimitry Andric           FastMathFlags FMF;
2682139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::UnsafeAlgebra))
2683139f7f9bSDimitry Andric             FMF.setUnsafeAlgebra();
2684139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::NoNaNs))
2685139f7f9bSDimitry Andric             FMF.setNoNaNs();
2686139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::NoInfs))
2687139f7f9bSDimitry Andric             FMF.setNoInfs();
2688139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::NoSignedZeros))
2689139f7f9bSDimitry Andric             FMF.setNoSignedZeros();
2690139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::AllowReciprocal))
2691139f7f9bSDimitry Andric             FMF.setAllowReciprocal();
2692139f7f9bSDimitry Andric           if (FMF.any())
2693139f7f9bSDimitry Andric             I->setFastMathFlags(FMF);
2694f22ef01cSRoman Divacky         }
2695139f7f9bSDimitry Andric 
2696f22ef01cSRoman Divacky       }
2697f22ef01cSRoman Divacky       break;
2698f22ef01cSRoman Divacky     }
2699f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CAST: {    // CAST: [opval, opty, destty, castopc]
2700f22ef01cSRoman Divacky       unsigned OpNum = 0;
2701f22ef01cSRoman Divacky       Value *Op;
2702f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2703f22ef01cSRoman Divacky           OpNum+2 != Record.size())
270439d628a0SDimitry Andric         return Error("Invalid record");
2705f22ef01cSRoman Divacky 
27066122f3e6SDimitry Andric       Type *ResTy = getTypeByID(Record[OpNum]);
2707f22ef01cSRoman Divacky       int Opc = GetDecodedCastOpcode(Record[OpNum+1]);
270891bc56edSDimitry Andric       if (Opc == -1 || !ResTy)
270939d628a0SDimitry Andric         return Error("Invalid record");
271091bc56edSDimitry Andric       Instruction *Temp = nullptr;
2711f785676fSDimitry Andric       if ((I = UpgradeBitCastInst(Opc, Op, ResTy, Temp))) {
2712f785676fSDimitry Andric         if (Temp) {
2713f785676fSDimitry Andric           InstructionList.push_back(Temp);
2714f785676fSDimitry Andric           CurBB->getInstList().push_back(Temp);
2715f785676fSDimitry Andric         }
2716f785676fSDimitry Andric       } else {
2717f22ef01cSRoman Divacky         I = CastInst::Create((Instruction::CastOps)Opc, Op, ResTy);
2718f785676fSDimitry Andric       }
2719f22ef01cSRoman Divacky       InstructionList.push_back(I);
2720f22ef01cSRoman Divacky       break;
2721f22ef01cSRoman Divacky     }
2722f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INBOUNDS_GEP:
2723f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_GEP: { // GEP: [n x operands]
2724f22ef01cSRoman Divacky       unsigned OpNum = 0;
2725f22ef01cSRoman Divacky       Value *BasePtr;
2726f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, BasePtr))
272739d628a0SDimitry Andric         return Error("Invalid record");
2728f22ef01cSRoman Divacky 
2729f22ef01cSRoman Divacky       SmallVector<Value*, 16> GEPIdx;
2730f22ef01cSRoman Divacky       while (OpNum != Record.size()) {
2731f22ef01cSRoman Divacky         Value *Op;
2732f22ef01cSRoman Divacky         if (getValueTypePair(Record, OpNum, NextValueNo, Op))
273339d628a0SDimitry Andric           return Error("Invalid record");
2734f22ef01cSRoman Divacky         GEPIdx.push_back(Op);
2735f22ef01cSRoman Divacky       }
2736f22ef01cSRoman Divacky 
27376122f3e6SDimitry Andric       I = GetElementPtrInst::Create(BasePtr, GEPIdx);
2738f22ef01cSRoman Divacky       InstructionList.push_back(I);
2739f22ef01cSRoman Divacky       if (BitCode == bitc::FUNC_CODE_INST_INBOUNDS_GEP)
2740f22ef01cSRoman Divacky         cast<GetElementPtrInst>(I)->setIsInBounds(true);
2741f22ef01cSRoman Divacky       break;
2742f22ef01cSRoman Divacky     }
2743f22ef01cSRoman Divacky 
2744f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_EXTRACTVAL: {
2745f22ef01cSRoman Divacky                                        // EXTRACTVAL: [opty, opval, n x indices]
2746f22ef01cSRoman Divacky       unsigned OpNum = 0;
2747f22ef01cSRoman Divacky       Value *Agg;
2748f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
274939d628a0SDimitry Andric         return Error("Invalid record");
2750f22ef01cSRoman Divacky 
2751f22ef01cSRoman Divacky       SmallVector<unsigned, 4> EXTRACTVALIdx;
2752f22ef01cSRoman Divacky       for (unsigned RecSize = Record.size();
2753f22ef01cSRoman Divacky            OpNum != RecSize; ++OpNum) {
2754f22ef01cSRoman Divacky         uint64_t Index = Record[OpNum];
2755f22ef01cSRoman Divacky         if ((unsigned)Index != Index)
275639d628a0SDimitry Andric           return Error("Invalid value");
2757f22ef01cSRoman Divacky         EXTRACTVALIdx.push_back((unsigned)Index);
2758f22ef01cSRoman Divacky       }
2759f22ef01cSRoman Divacky 
276017a519f9SDimitry Andric       I = ExtractValueInst::Create(Agg, EXTRACTVALIdx);
2761f22ef01cSRoman Divacky       InstructionList.push_back(I);
2762f22ef01cSRoman Divacky       break;
2763f22ef01cSRoman Divacky     }
2764f22ef01cSRoman Divacky 
2765f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INSERTVAL: {
2766f22ef01cSRoman Divacky                            // INSERTVAL: [opty, opval, opty, opval, n x indices]
2767f22ef01cSRoman Divacky       unsigned OpNum = 0;
2768f22ef01cSRoman Divacky       Value *Agg;
2769f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
277039d628a0SDimitry Andric         return Error("Invalid record");
2771f22ef01cSRoman Divacky       Value *Val;
2772f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Val))
277339d628a0SDimitry Andric         return Error("Invalid record");
2774f22ef01cSRoman Divacky 
2775f22ef01cSRoman Divacky       SmallVector<unsigned, 4> INSERTVALIdx;
2776f22ef01cSRoman Divacky       for (unsigned RecSize = Record.size();
2777f22ef01cSRoman Divacky            OpNum != RecSize; ++OpNum) {
2778f22ef01cSRoman Divacky         uint64_t Index = Record[OpNum];
2779f22ef01cSRoman Divacky         if ((unsigned)Index != Index)
278039d628a0SDimitry Andric           return Error("Invalid value");
2781f22ef01cSRoman Divacky         INSERTVALIdx.push_back((unsigned)Index);
2782f22ef01cSRoman Divacky       }
2783f22ef01cSRoman Divacky 
278417a519f9SDimitry Andric       I = InsertValueInst::Create(Agg, Val, INSERTVALIdx);
2785f22ef01cSRoman Divacky       InstructionList.push_back(I);
2786f22ef01cSRoman Divacky       break;
2787f22ef01cSRoman Divacky     }
2788f22ef01cSRoman Divacky 
2789f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SELECT: { // SELECT: [opval, ty, opval, opval]
2790f22ef01cSRoman Divacky       // obsolete form of select
2791f22ef01cSRoman Divacky       // handles select i1 ... in old bitcode
2792f22ef01cSRoman Divacky       unsigned OpNum = 0;
2793f22ef01cSRoman Divacky       Value *TrueVal, *FalseVal, *Cond;
2794f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
27953861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, TrueVal->getType(), FalseVal) ||
27963861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, Type::getInt1Ty(Context), Cond))
279739d628a0SDimitry Andric         return Error("Invalid record");
2798f22ef01cSRoman Divacky 
2799f22ef01cSRoman Divacky       I = SelectInst::Create(Cond, TrueVal, FalseVal);
2800f22ef01cSRoman Divacky       InstructionList.push_back(I);
2801f22ef01cSRoman Divacky       break;
2802f22ef01cSRoman Divacky     }
2803f22ef01cSRoman Divacky 
2804f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_VSELECT: {// VSELECT: [ty,opval,opval,predty,pred]
2805f22ef01cSRoman Divacky       // new form of select
2806f22ef01cSRoman Divacky       // handles select i1 or select [N x i1]
2807f22ef01cSRoman Divacky       unsigned OpNum = 0;
2808f22ef01cSRoman Divacky       Value *TrueVal, *FalseVal, *Cond;
2809f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
28103861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, TrueVal->getType(), FalseVal) ||
2811f22ef01cSRoman Divacky           getValueTypePair(Record, OpNum, NextValueNo, Cond))
281239d628a0SDimitry Andric         return Error("Invalid record");
2813f22ef01cSRoman Divacky 
2814f22ef01cSRoman Divacky       // select condition can be either i1 or [N x i1]
28156122f3e6SDimitry Andric       if (VectorType* vector_type =
28166122f3e6SDimitry Andric           dyn_cast<VectorType>(Cond->getType())) {
2817f22ef01cSRoman Divacky         // expect <n x i1>
2818f22ef01cSRoman Divacky         if (vector_type->getElementType() != Type::getInt1Ty(Context))
281939d628a0SDimitry Andric           return Error("Invalid type for value");
2820f22ef01cSRoman Divacky       } else {
2821f22ef01cSRoman Divacky         // expect i1
2822f22ef01cSRoman Divacky         if (Cond->getType() != Type::getInt1Ty(Context))
282339d628a0SDimitry Andric           return Error("Invalid type for value");
2824f22ef01cSRoman Divacky       }
2825f22ef01cSRoman Divacky 
2826f22ef01cSRoman Divacky       I = SelectInst::Create(Cond, TrueVal, FalseVal);
2827f22ef01cSRoman Divacky       InstructionList.push_back(I);
2828f22ef01cSRoman Divacky       break;
2829f22ef01cSRoman Divacky     }
2830f22ef01cSRoman Divacky 
2831f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_EXTRACTELT: { // EXTRACTELT: [opty, opval, opval]
2832f22ef01cSRoman Divacky       unsigned OpNum = 0;
2833f22ef01cSRoman Divacky       Value *Vec, *Idx;
2834f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
283591bc56edSDimitry Andric           getValueTypePair(Record, OpNum, NextValueNo, Idx))
283639d628a0SDimitry Andric         return Error("Invalid record");
2837f22ef01cSRoman Divacky       I = ExtractElementInst::Create(Vec, Idx);
2838f22ef01cSRoman Divacky       InstructionList.push_back(I);
2839f22ef01cSRoman Divacky       break;
2840f22ef01cSRoman Divacky     }
2841f22ef01cSRoman Divacky 
2842f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INSERTELT: { // INSERTELT: [ty, opval,opval,opval]
2843f22ef01cSRoman Divacky       unsigned OpNum = 0;
2844f22ef01cSRoman Divacky       Value *Vec, *Elt, *Idx;
2845f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
28463861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
2847f22ef01cSRoman Divacky                    cast<VectorType>(Vec->getType())->getElementType(), Elt) ||
284891bc56edSDimitry Andric           getValueTypePair(Record, OpNum, NextValueNo, Idx))
284939d628a0SDimitry Andric         return Error("Invalid record");
2850f22ef01cSRoman Divacky       I = InsertElementInst::Create(Vec, Elt, Idx);
2851f22ef01cSRoman Divacky       InstructionList.push_back(I);
2852f22ef01cSRoman Divacky       break;
2853f22ef01cSRoman Divacky     }
2854f22ef01cSRoman Divacky 
2855f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SHUFFLEVEC: {// SHUFFLEVEC: [opval,ty,opval,opval]
2856f22ef01cSRoman Divacky       unsigned OpNum = 0;
2857f22ef01cSRoman Divacky       Value *Vec1, *Vec2, *Mask;
2858f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec1) ||
28593861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, Vec1->getType(), Vec2))
286039d628a0SDimitry Andric         return Error("Invalid record");
2861f22ef01cSRoman Divacky 
2862f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Mask))
286339d628a0SDimitry Andric         return Error("Invalid record");
2864f22ef01cSRoman Divacky       I = new ShuffleVectorInst(Vec1, Vec2, Mask);
2865f22ef01cSRoman Divacky       InstructionList.push_back(I);
2866f22ef01cSRoman Divacky       break;
2867f22ef01cSRoman Divacky     }
2868f22ef01cSRoman Divacky 
2869f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CMP:   // CMP: [opty, opval, opval, pred]
2870f22ef01cSRoman Divacky       // Old form of ICmp/FCmp returning bool
2871f22ef01cSRoman Divacky       // Existed to differentiate between icmp/fcmp and vicmp/vfcmp which were
2872f22ef01cSRoman Divacky       // both legal on vectors but had different behaviour.
2873f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CMP2: { // CMP2: [opty, opval, opval, pred]
2874f22ef01cSRoman Divacky       // FCmp/ICmp returning bool or vector of bool
2875f22ef01cSRoman Divacky 
2876f22ef01cSRoman Divacky       unsigned OpNum = 0;
2877f22ef01cSRoman Divacky       Value *LHS, *RHS;
2878f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
28793861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, LHS->getType(), RHS) ||
2880f22ef01cSRoman Divacky           OpNum+1 != Record.size())
288139d628a0SDimitry Andric         return Error("Invalid record");
2882f22ef01cSRoman Divacky 
2883f22ef01cSRoman Divacky       if (LHS->getType()->isFPOrFPVectorTy())
2884f22ef01cSRoman Divacky         I = new FCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
2885f22ef01cSRoman Divacky       else
2886f22ef01cSRoman Divacky         I = new ICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
2887f22ef01cSRoman Divacky       InstructionList.push_back(I);
2888f22ef01cSRoman Divacky       break;
2889f22ef01cSRoman Divacky     }
2890f22ef01cSRoman Divacky 
2891f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_RET: // RET: [opty,opval<optional>]
2892f22ef01cSRoman Divacky       {
2893f22ef01cSRoman Divacky         unsigned Size = Record.size();
2894f22ef01cSRoman Divacky         if (Size == 0) {
2895f22ef01cSRoman Divacky           I = ReturnInst::Create(Context);
2896f22ef01cSRoman Divacky           InstructionList.push_back(I);
2897f22ef01cSRoman Divacky           break;
2898f22ef01cSRoman Divacky         }
2899f22ef01cSRoman Divacky 
2900f22ef01cSRoman Divacky         unsigned OpNum = 0;
290191bc56edSDimitry Andric         Value *Op = nullptr;
2902f22ef01cSRoman Divacky         if (getValueTypePair(Record, OpNum, NextValueNo, Op))
290339d628a0SDimitry Andric           return Error("Invalid record");
290417a519f9SDimitry Andric         if (OpNum != Record.size())
290539d628a0SDimitry Andric           return Error("Invalid record");
2906f22ef01cSRoman Divacky 
290717a519f9SDimitry Andric         I = ReturnInst::Create(Context, Op);
2908f22ef01cSRoman Divacky         InstructionList.push_back(I);
2909f22ef01cSRoman Divacky         break;
2910f22ef01cSRoman Divacky       }
2911f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_BR: { // BR: [bb#, bb#, opval] or [bb#]
2912f22ef01cSRoman Divacky       if (Record.size() != 1 && Record.size() != 3)
291339d628a0SDimitry Andric         return Error("Invalid record");
2914f22ef01cSRoman Divacky       BasicBlock *TrueDest = getBasicBlock(Record[0]);
291591bc56edSDimitry Andric       if (!TrueDest)
291639d628a0SDimitry Andric         return Error("Invalid record");
2917f22ef01cSRoman Divacky 
2918f22ef01cSRoman Divacky       if (Record.size() == 1) {
2919f22ef01cSRoman Divacky         I = BranchInst::Create(TrueDest);
2920f22ef01cSRoman Divacky         InstructionList.push_back(I);
2921f22ef01cSRoman Divacky       }
2922f22ef01cSRoman Divacky       else {
2923f22ef01cSRoman Divacky         BasicBlock *FalseDest = getBasicBlock(Record[1]);
29243861d79fSDimitry Andric         Value *Cond = getValue(Record, 2, NextValueNo,
29253861d79fSDimitry Andric                                Type::getInt1Ty(Context));
292691bc56edSDimitry Andric         if (!FalseDest || !Cond)
292739d628a0SDimitry Andric           return Error("Invalid record");
2928f22ef01cSRoman Divacky         I = BranchInst::Create(TrueDest, FalseDest, Cond);
2929f22ef01cSRoman Divacky         InstructionList.push_back(I);
2930f22ef01cSRoman Divacky       }
2931f22ef01cSRoman Divacky       break;
2932f22ef01cSRoman Divacky     }
2933f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SWITCH: { // SWITCH: [opty, op0, op1, ...]
29347ae0e2c9SDimitry Andric       // Check magic
29357ae0e2c9SDimitry Andric       if ((Record[0] >> 16) == SWITCH_INST_MAGIC) {
2936f785676fSDimitry Andric         // "New" SwitchInst format with case ranges. The changes to write this
2937f785676fSDimitry Andric         // format were reverted but we still recognize bitcode that uses it.
2938f785676fSDimitry Andric         // Hopefully someday we will have support for case ranges and can use
2939f785676fSDimitry Andric         // this format again.
29407ae0e2c9SDimitry Andric 
29417ae0e2c9SDimitry Andric         Type *OpTy = getTypeByID(Record[1]);
29427ae0e2c9SDimitry Andric         unsigned ValueBitWidth = cast<IntegerType>(OpTy)->getBitWidth();
29437ae0e2c9SDimitry Andric 
29443861d79fSDimitry Andric         Value *Cond = getValue(Record, 2, NextValueNo, OpTy);
29457ae0e2c9SDimitry Andric         BasicBlock *Default = getBasicBlock(Record[3]);
294691bc56edSDimitry Andric         if (!OpTy || !Cond || !Default)
294739d628a0SDimitry Andric           return Error("Invalid record");
29487ae0e2c9SDimitry Andric 
29497ae0e2c9SDimitry Andric         unsigned NumCases = Record[4];
29507ae0e2c9SDimitry Andric 
29517ae0e2c9SDimitry Andric         SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
29527ae0e2c9SDimitry Andric         InstructionList.push_back(SI);
29537ae0e2c9SDimitry Andric 
29547ae0e2c9SDimitry Andric         unsigned CurIdx = 5;
29557ae0e2c9SDimitry Andric         for (unsigned i = 0; i != NumCases; ++i) {
2956f785676fSDimitry Andric           SmallVector<ConstantInt*, 1> CaseVals;
29577ae0e2c9SDimitry Andric           unsigned NumItems = Record[CurIdx++];
29587ae0e2c9SDimitry Andric           for (unsigned ci = 0; ci != NumItems; ++ci) {
29597ae0e2c9SDimitry Andric             bool isSingleNumber = Record[CurIdx++];
29607ae0e2c9SDimitry Andric 
29617ae0e2c9SDimitry Andric             APInt Low;
29627ae0e2c9SDimitry Andric             unsigned ActiveWords = 1;
29637ae0e2c9SDimitry Andric             if (ValueBitWidth > 64)
29647ae0e2c9SDimitry Andric               ActiveWords = Record[CurIdx++];
29657ae0e2c9SDimitry Andric             Low = ReadWideAPInt(makeArrayRef(&Record[CurIdx], ActiveWords),
29667ae0e2c9SDimitry Andric                                 ValueBitWidth);
29677ae0e2c9SDimitry Andric             CurIdx += ActiveWords;
29687ae0e2c9SDimitry Andric 
29697ae0e2c9SDimitry Andric             if (!isSingleNumber) {
29707ae0e2c9SDimitry Andric               ActiveWords = 1;
29717ae0e2c9SDimitry Andric               if (ValueBitWidth > 64)
29727ae0e2c9SDimitry Andric                 ActiveWords = Record[CurIdx++];
29737ae0e2c9SDimitry Andric               APInt High =
29747ae0e2c9SDimitry Andric                   ReadWideAPInt(makeArrayRef(&Record[CurIdx], ActiveWords),
29757ae0e2c9SDimitry Andric                                 ValueBitWidth);
29767ae0e2c9SDimitry Andric               CurIdx += ActiveWords;
2977f785676fSDimitry Andric 
2978f785676fSDimitry Andric               // FIXME: It is not clear whether values in the range should be
2979f785676fSDimitry Andric               // compared as signed or unsigned values. The partially
2980f785676fSDimitry Andric               // implemented changes that used this format in the past used
2981f785676fSDimitry Andric               // unsigned comparisons.
2982f785676fSDimitry Andric               for ( ; Low.ule(High); ++Low)
2983f785676fSDimitry Andric                 CaseVals.push_back(ConstantInt::get(Context, Low));
29847ae0e2c9SDimitry Andric             } else
2985f785676fSDimitry Andric               CaseVals.push_back(ConstantInt::get(Context, Low));
29867ae0e2c9SDimitry Andric           }
29877ae0e2c9SDimitry Andric           BasicBlock *DestBB = getBasicBlock(Record[CurIdx++]);
2988f785676fSDimitry Andric           for (SmallVector<ConstantInt*, 1>::iterator cvi = CaseVals.begin(),
2989f785676fSDimitry Andric                  cve = CaseVals.end(); cvi != cve; ++cvi)
2990f785676fSDimitry Andric             SI->addCase(*cvi, DestBB);
29917ae0e2c9SDimitry Andric         }
29927ae0e2c9SDimitry Andric         I = SI;
29937ae0e2c9SDimitry Andric         break;
29947ae0e2c9SDimitry Andric       }
29957ae0e2c9SDimitry Andric 
29967ae0e2c9SDimitry Andric       // Old SwitchInst format without case ranges.
29977ae0e2c9SDimitry Andric 
2998f22ef01cSRoman Divacky       if (Record.size() < 3 || (Record.size() & 1) == 0)
299939d628a0SDimitry Andric         return Error("Invalid record");
30006122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[0]);
30013861d79fSDimitry Andric       Value *Cond = getValue(Record, 1, NextValueNo, OpTy);
3002f22ef01cSRoman Divacky       BasicBlock *Default = getBasicBlock(Record[2]);
300391bc56edSDimitry Andric       if (!OpTy || !Cond || !Default)
300439d628a0SDimitry Andric         return Error("Invalid record");
3005f22ef01cSRoman Divacky       unsigned NumCases = (Record.size()-3)/2;
3006f22ef01cSRoman Divacky       SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
3007f22ef01cSRoman Divacky       InstructionList.push_back(SI);
3008f22ef01cSRoman Divacky       for (unsigned i = 0, e = NumCases; i != e; ++i) {
3009f22ef01cSRoman Divacky         ConstantInt *CaseVal =
3010f22ef01cSRoman Divacky           dyn_cast_or_null<ConstantInt>(getFnValueByID(Record[3+i*2], OpTy));
3011f22ef01cSRoman Divacky         BasicBlock *DestBB = getBasicBlock(Record[1+3+i*2]);
301291bc56edSDimitry Andric         if (!CaseVal || !DestBB) {
3013f22ef01cSRoman Divacky           delete SI;
301439d628a0SDimitry Andric           return Error("Invalid record");
3015f22ef01cSRoman Divacky         }
3016f22ef01cSRoman Divacky         SI->addCase(CaseVal, DestBB);
3017f22ef01cSRoman Divacky       }
3018f22ef01cSRoman Divacky       I = SI;
3019f22ef01cSRoman Divacky       break;
3020f22ef01cSRoman Divacky     }
3021f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INDIRECTBR: { // INDIRECTBR: [opty, op0, op1, ...]
3022f22ef01cSRoman Divacky       if (Record.size() < 2)
302339d628a0SDimitry Andric         return Error("Invalid record");
30246122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[0]);
30253861d79fSDimitry Andric       Value *Address = getValue(Record, 1, NextValueNo, OpTy);
302691bc56edSDimitry Andric       if (!OpTy || !Address)
302739d628a0SDimitry Andric         return Error("Invalid record");
3028f22ef01cSRoman Divacky       unsigned NumDests = Record.size()-2;
3029f22ef01cSRoman Divacky       IndirectBrInst *IBI = IndirectBrInst::Create(Address, NumDests);
3030f22ef01cSRoman Divacky       InstructionList.push_back(IBI);
3031f22ef01cSRoman Divacky       for (unsigned i = 0, e = NumDests; i != e; ++i) {
3032f22ef01cSRoman Divacky         if (BasicBlock *DestBB = getBasicBlock(Record[2+i])) {
3033f22ef01cSRoman Divacky           IBI->addDestination(DestBB);
3034f22ef01cSRoman Divacky         } else {
3035f22ef01cSRoman Divacky           delete IBI;
303639d628a0SDimitry Andric           return Error("Invalid record");
3037f22ef01cSRoman Divacky         }
3038f22ef01cSRoman Divacky       }
3039f22ef01cSRoman Divacky       I = IBI;
3040f22ef01cSRoman Divacky       break;
3041f22ef01cSRoman Divacky     }
3042f22ef01cSRoman Divacky 
3043f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INVOKE: {
3044f22ef01cSRoman Divacky       // INVOKE: [attrs, cc, normBB, unwindBB, fnty, op0,op1,op2, ...]
3045f785676fSDimitry Andric       if (Record.size() < 4)
304639d628a0SDimitry Andric         return Error("Invalid record");
3047139f7f9bSDimitry Andric       AttributeSet PAL = getAttributes(Record[0]);
3048f22ef01cSRoman Divacky       unsigned CCInfo = Record[1];
3049f22ef01cSRoman Divacky       BasicBlock *NormalBB = getBasicBlock(Record[2]);
3050f22ef01cSRoman Divacky       BasicBlock *UnwindBB = getBasicBlock(Record[3]);
3051f22ef01cSRoman Divacky 
3052f22ef01cSRoman Divacky       unsigned OpNum = 4;
3053f22ef01cSRoman Divacky       Value *Callee;
3054f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
305539d628a0SDimitry Andric         return Error("Invalid record");
3056f22ef01cSRoman Divacky 
30576122f3e6SDimitry Andric       PointerType *CalleeTy = dyn_cast<PointerType>(Callee->getType());
305891bc56edSDimitry Andric       FunctionType *FTy = !CalleeTy ? nullptr :
3059f22ef01cSRoman Divacky         dyn_cast<FunctionType>(CalleeTy->getElementType());
3060f22ef01cSRoman Divacky 
3061f22ef01cSRoman Divacky       // Check that the right number of fixed parameters are here.
306291bc56edSDimitry Andric       if (!FTy || !NormalBB || !UnwindBB ||
3063f22ef01cSRoman Divacky           Record.size() < OpNum+FTy->getNumParams())
306439d628a0SDimitry Andric         return Error("Invalid record");
3065f22ef01cSRoman Divacky 
3066f22ef01cSRoman Divacky       SmallVector<Value*, 16> Ops;
3067f22ef01cSRoman Divacky       for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
30683861d79fSDimitry Andric         Ops.push_back(getValue(Record, OpNum, NextValueNo,
30693861d79fSDimitry Andric                                FTy->getParamType(i)));
307091bc56edSDimitry Andric         if (!Ops.back())
307139d628a0SDimitry Andric           return Error("Invalid record");
3072f22ef01cSRoman Divacky       }
3073f22ef01cSRoman Divacky 
3074f22ef01cSRoman Divacky       if (!FTy->isVarArg()) {
3075f22ef01cSRoman Divacky         if (Record.size() != OpNum)
307639d628a0SDimitry Andric           return Error("Invalid record");
3077f22ef01cSRoman Divacky       } else {
3078f22ef01cSRoman Divacky         // Read type/value pairs for varargs params.
3079f22ef01cSRoman Divacky         while (OpNum != Record.size()) {
3080f22ef01cSRoman Divacky           Value *Op;
3081f22ef01cSRoman Divacky           if (getValueTypePair(Record, OpNum, NextValueNo, Op))
308239d628a0SDimitry Andric             return Error("Invalid record");
3083f22ef01cSRoman Divacky           Ops.push_back(Op);
3084f22ef01cSRoman Divacky         }
3085f22ef01cSRoman Divacky       }
3086f22ef01cSRoman Divacky 
308717a519f9SDimitry Andric       I = InvokeInst::Create(Callee, NormalBB, UnwindBB, Ops);
3088f22ef01cSRoman Divacky       InstructionList.push_back(I);
3089f22ef01cSRoman Divacky       cast<InvokeInst>(I)->setCallingConv(
3090f22ef01cSRoman Divacky         static_cast<CallingConv::ID>(CCInfo));
3091f22ef01cSRoman Divacky       cast<InvokeInst>(I)->setAttributes(PAL);
3092f22ef01cSRoman Divacky       break;
3093f22ef01cSRoman Divacky     }
30946122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_RESUME: { // RESUME: [opval]
30956122f3e6SDimitry Andric       unsigned Idx = 0;
309691bc56edSDimitry Andric       Value *Val = nullptr;
30976122f3e6SDimitry Andric       if (getValueTypePair(Record, Idx, NextValueNo, Val))
309839d628a0SDimitry Andric         return Error("Invalid record");
30996122f3e6SDimitry Andric       I = ResumeInst::Create(Val);
31006122f3e6SDimitry Andric       InstructionList.push_back(I);
31016122f3e6SDimitry Andric       break;
31026122f3e6SDimitry Andric     }
3103f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_UNREACHABLE: // UNREACHABLE
3104f22ef01cSRoman Divacky       I = new UnreachableInst(Context);
3105f22ef01cSRoman Divacky       InstructionList.push_back(I);
3106f22ef01cSRoman Divacky       break;
3107f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_PHI: { // PHI: [ty, val0,bb0, ...]
3108f22ef01cSRoman Divacky       if (Record.size() < 1 || ((Record.size()-1)&1))
310939d628a0SDimitry Andric         return Error("Invalid record");
31106122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
3111f785676fSDimitry Andric       if (!Ty)
311239d628a0SDimitry Andric         return Error("Invalid record");
3113f22ef01cSRoman Divacky 
31143b0f4066SDimitry Andric       PHINode *PN = PHINode::Create(Ty, (Record.size()-1)/2);
3115f22ef01cSRoman Divacky       InstructionList.push_back(PN);
3116f22ef01cSRoman Divacky 
3117f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size()-1; i != e; i += 2) {
31183861d79fSDimitry Andric         Value *V;
31193861d79fSDimitry Andric         // With the new function encoding, it is possible that operands have
31203861d79fSDimitry Andric         // negative IDs (for forward references).  Use a signed VBR
31213861d79fSDimitry Andric         // representation to keep the encoding small.
31223861d79fSDimitry Andric         if (UseRelativeIDs)
31233861d79fSDimitry Andric           V = getValueSigned(Record, 1+i, NextValueNo, Ty);
31243861d79fSDimitry Andric         else
31253861d79fSDimitry Andric           V = getValue(Record, 1+i, NextValueNo, Ty);
3126f22ef01cSRoman Divacky         BasicBlock *BB = getBasicBlock(Record[2+i]);
3127f785676fSDimitry Andric         if (!V || !BB)
312839d628a0SDimitry Andric           return Error("Invalid record");
3129f22ef01cSRoman Divacky         PN->addIncoming(V, BB);
3130f22ef01cSRoman Divacky       }
3131f22ef01cSRoman Divacky       I = PN;
3132f22ef01cSRoman Divacky       break;
3133f22ef01cSRoman Divacky     }
3134f22ef01cSRoman Divacky 
31356122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_LANDINGPAD: {
31366122f3e6SDimitry Andric       // LANDINGPAD: [ty, val, val, num, (id0,val0 ...)?]
31376122f3e6SDimitry Andric       unsigned Idx = 0;
31386122f3e6SDimitry Andric       if (Record.size() < 4)
313939d628a0SDimitry Andric         return Error("Invalid record");
31406122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[Idx++]);
3141f785676fSDimitry Andric       if (!Ty)
314239d628a0SDimitry Andric         return Error("Invalid record");
314391bc56edSDimitry Andric       Value *PersFn = nullptr;
31446122f3e6SDimitry Andric       if (getValueTypePair(Record, Idx, NextValueNo, PersFn))
314539d628a0SDimitry Andric         return Error("Invalid record");
31466122f3e6SDimitry Andric 
31476122f3e6SDimitry Andric       bool IsCleanup = !!Record[Idx++];
31486122f3e6SDimitry Andric       unsigned NumClauses = Record[Idx++];
31496122f3e6SDimitry Andric       LandingPadInst *LP = LandingPadInst::Create(Ty, PersFn, NumClauses);
31506122f3e6SDimitry Andric       LP->setCleanup(IsCleanup);
31516122f3e6SDimitry Andric       for (unsigned J = 0; J != NumClauses; ++J) {
31526122f3e6SDimitry Andric         LandingPadInst::ClauseType CT =
31536122f3e6SDimitry Andric           LandingPadInst::ClauseType(Record[Idx++]); (void)CT;
31546122f3e6SDimitry Andric         Value *Val;
31556122f3e6SDimitry Andric 
31566122f3e6SDimitry Andric         if (getValueTypePair(Record, Idx, NextValueNo, Val)) {
31576122f3e6SDimitry Andric           delete LP;
315839d628a0SDimitry Andric           return Error("Invalid record");
31596122f3e6SDimitry Andric         }
31606122f3e6SDimitry Andric 
31616122f3e6SDimitry Andric         assert((CT != LandingPadInst::Catch ||
31626122f3e6SDimitry Andric                 !isa<ArrayType>(Val->getType())) &&
31636122f3e6SDimitry Andric                "Catch clause has a invalid type!");
31646122f3e6SDimitry Andric         assert((CT != LandingPadInst::Filter ||
31656122f3e6SDimitry Andric                 isa<ArrayType>(Val->getType())) &&
31666122f3e6SDimitry Andric                "Filter clause has invalid type!");
316791bc56edSDimitry Andric         LP->addClause(cast<Constant>(Val));
31686122f3e6SDimitry Andric       }
31696122f3e6SDimitry Andric 
31706122f3e6SDimitry Andric       I = LP;
31716122f3e6SDimitry Andric       InstructionList.push_back(I);
31726122f3e6SDimitry Andric       break;
31736122f3e6SDimitry Andric     }
31746122f3e6SDimitry Andric 
317517a519f9SDimitry Andric     case bitc::FUNC_CODE_INST_ALLOCA: { // ALLOCA: [instty, opty, op, align]
317617a519f9SDimitry Andric       if (Record.size() != 4)
317739d628a0SDimitry Andric         return Error("Invalid record");
31786122f3e6SDimitry Andric       PointerType *Ty =
3179f22ef01cSRoman Divacky         dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
31806122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[1]);
318117a519f9SDimitry Andric       Value *Size = getFnValueByID(Record[2], OpTy);
318291bc56edSDimitry Andric       unsigned AlignRecord = Record[3];
318391bc56edSDimitry Andric       bool InAlloca = AlignRecord & (1 << 5);
318491bc56edSDimitry Andric       unsigned Align = AlignRecord & ((1 << 5) - 1);
3185f785676fSDimitry Andric       if (!Ty || !Size)
318639d628a0SDimitry Andric         return Error("Invalid record");
318791bc56edSDimitry Andric       AllocaInst *AI = new AllocaInst(Ty->getElementType(), Size, (1 << Align) >> 1);
318891bc56edSDimitry Andric       AI->setUsedWithInAlloca(InAlloca);
318991bc56edSDimitry Andric       I = AI;
3190f22ef01cSRoman Divacky       InstructionList.push_back(I);
3191f22ef01cSRoman Divacky       break;
3192f22ef01cSRoman Divacky     }
3193f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_LOAD: { // LOAD: [opty, op, align, vol]
3194f22ef01cSRoman Divacky       unsigned OpNum = 0;
3195f22ef01cSRoman Divacky       Value *Op;
3196f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
3197f22ef01cSRoman Divacky           OpNum+2 != Record.size())
319839d628a0SDimitry Andric         return Error("Invalid record");
3199f22ef01cSRoman Divacky 
3200f22ef01cSRoman Divacky       I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1);
3201f22ef01cSRoman Divacky       InstructionList.push_back(I);
3202f22ef01cSRoman Divacky       break;
3203f22ef01cSRoman Divacky     }
32046122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_LOADATOMIC: {
32056122f3e6SDimitry Andric        // LOADATOMIC: [opty, op, align, vol, ordering, synchscope]
32066122f3e6SDimitry Andric       unsigned OpNum = 0;
32076122f3e6SDimitry Andric       Value *Op;
32086122f3e6SDimitry Andric       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
32096122f3e6SDimitry Andric           OpNum+4 != Record.size())
321039d628a0SDimitry Andric         return Error("Invalid record");
32116122f3e6SDimitry Andric 
32126122f3e6SDimitry Andric       AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+2]);
32136122f3e6SDimitry Andric       if (Ordering == NotAtomic || Ordering == Release ||
32146122f3e6SDimitry Andric           Ordering == AcquireRelease)
321539d628a0SDimitry Andric         return Error("Invalid record");
32166122f3e6SDimitry Andric       if (Ordering != NotAtomic && Record[OpNum] == 0)
321739d628a0SDimitry Andric         return Error("Invalid record");
32186122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+3]);
32196122f3e6SDimitry Andric 
32206122f3e6SDimitry Andric       I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1,
32216122f3e6SDimitry Andric                        Ordering, SynchScope);
32226122f3e6SDimitry Andric       InstructionList.push_back(I);
32236122f3e6SDimitry Andric       break;
32246122f3e6SDimitry Andric     }
322517a519f9SDimitry Andric     case bitc::FUNC_CODE_INST_STORE: { // STORE2:[ptrty, ptr, val, align, vol]
3226f22ef01cSRoman Divacky       unsigned OpNum = 0;
3227f22ef01cSRoman Divacky       Value *Val, *Ptr;
3228f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
32293861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
3230f22ef01cSRoman Divacky                     cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
3231f22ef01cSRoman Divacky           OpNum+2 != Record.size())
323239d628a0SDimitry Andric         return Error("Invalid record");
3233f22ef01cSRoman Divacky 
3234f22ef01cSRoman Divacky       I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
3235f22ef01cSRoman Divacky       InstructionList.push_back(I);
3236f22ef01cSRoman Divacky       break;
3237f22ef01cSRoman Divacky     }
32386122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_STOREATOMIC: {
32396122f3e6SDimitry Andric       // STOREATOMIC: [ptrty, ptr, val, align, vol, ordering, synchscope]
32406122f3e6SDimitry Andric       unsigned OpNum = 0;
32416122f3e6SDimitry Andric       Value *Val, *Ptr;
32426122f3e6SDimitry Andric       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
32433861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
32446122f3e6SDimitry Andric                     cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
32456122f3e6SDimitry Andric           OpNum+4 != Record.size())
324639d628a0SDimitry Andric         return Error("Invalid record");
32476122f3e6SDimitry Andric 
32486122f3e6SDimitry Andric       AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+2]);
32496122f3e6SDimitry Andric       if (Ordering == NotAtomic || Ordering == Acquire ||
32506122f3e6SDimitry Andric           Ordering == AcquireRelease)
325139d628a0SDimitry Andric         return Error("Invalid record");
32526122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+3]);
32536122f3e6SDimitry Andric       if (Ordering != NotAtomic && Record[OpNum] == 0)
325439d628a0SDimitry Andric         return Error("Invalid record");
32556122f3e6SDimitry Andric 
32566122f3e6SDimitry Andric       I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1,
32576122f3e6SDimitry Andric                         Ordering, SynchScope);
32586122f3e6SDimitry Andric       InstructionList.push_back(I);
32596122f3e6SDimitry Andric       break;
32606122f3e6SDimitry Andric     }
32616122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_CMPXCHG: {
326291bc56edSDimitry Andric       // CMPXCHG:[ptrty, ptr, cmp, new, vol, successordering, synchscope,
326391bc56edSDimitry Andric       //          failureordering?, isweak?]
32646122f3e6SDimitry Andric       unsigned OpNum = 0;
32656122f3e6SDimitry Andric       Value *Ptr, *Cmp, *New;
32666122f3e6SDimitry Andric       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
32673861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
32686122f3e6SDimitry Andric                     cast<PointerType>(Ptr->getType())->getElementType(), Cmp) ||
32693861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
32706122f3e6SDimitry Andric                     cast<PointerType>(Ptr->getType())->getElementType(), New) ||
327191bc56edSDimitry Andric           (Record.size() < OpNum + 3 || Record.size() > OpNum + 5))
327239d628a0SDimitry Andric         return Error("Invalid record");
327391bc56edSDimitry Andric       AtomicOrdering SuccessOrdering = GetDecodedOrdering(Record[OpNum+1]);
327491bc56edSDimitry Andric       if (SuccessOrdering == NotAtomic || SuccessOrdering == Unordered)
327539d628a0SDimitry Andric         return Error("Invalid record");
32766122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+2]);
327791bc56edSDimitry Andric 
327891bc56edSDimitry Andric       AtomicOrdering FailureOrdering;
327991bc56edSDimitry Andric       if (Record.size() < 7)
328091bc56edSDimitry Andric         FailureOrdering =
328191bc56edSDimitry Andric             AtomicCmpXchgInst::getStrongestFailureOrdering(SuccessOrdering);
328291bc56edSDimitry Andric       else
328391bc56edSDimitry Andric         FailureOrdering = GetDecodedOrdering(Record[OpNum+3]);
328491bc56edSDimitry Andric 
328591bc56edSDimitry Andric       I = new AtomicCmpXchgInst(Ptr, Cmp, New, SuccessOrdering, FailureOrdering,
328691bc56edSDimitry Andric                                 SynchScope);
32876122f3e6SDimitry Andric       cast<AtomicCmpXchgInst>(I)->setVolatile(Record[OpNum]);
328891bc56edSDimitry Andric 
328991bc56edSDimitry Andric       if (Record.size() < 8) {
329091bc56edSDimitry Andric         // Before weak cmpxchgs existed, the instruction simply returned the
329191bc56edSDimitry Andric         // value loaded from memory, so bitcode files from that era will be
329291bc56edSDimitry Andric         // expecting the first component of a modern cmpxchg.
329391bc56edSDimitry Andric         CurBB->getInstList().push_back(I);
329491bc56edSDimitry Andric         I = ExtractValueInst::Create(I, 0);
329591bc56edSDimitry Andric       } else {
329691bc56edSDimitry Andric         cast<AtomicCmpXchgInst>(I)->setWeak(Record[OpNum+4]);
329791bc56edSDimitry Andric       }
329891bc56edSDimitry Andric 
32996122f3e6SDimitry Andric       InstructionList.push_back(I);
33006122f3e6SDimitry Andric       break;
33016122f3e6SDimitry Andric     }
33026122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_ATOMICRMW: {
33036122f3e6SDimitry Andric       // ATOMICRMW:[ptrty, ptr, val, op, vol, ordering, synchscope]
33046122f3e6SDimitry Andric       unsigned OpNum = 0;
33056122f3e6SDimitry Andric       Value *Ptr, *Val;
33066122f3e6SDimitry Andric       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
33073861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
33086122f3e6SDimitry Andric                     cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
33096122f3e6SDimitry Andric           OpNum+4 != Record.size())
331039d628a0SDimitry Andric         return Error("Invalid record");
33116122f3e6SDimitry Andric       AtomicRMWInst::BinOp Operation = GetDecodedRMWOperation(Record[OpNum]);
33126122f3e6SDimitry Andric       if (Operation < AtomicRMWInst::FIRST_BINOP ||
33136122f3e6SDimitry Andric           Operation > AtomicRMWInst::LAST_BINOP)
331439d628a0SDimitry Andric         return Error("Invalid record");
33156122f3e6SDimitry Andric       AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+2]);
33166122f3e6SDimitry Andric       if (Ordering == NotAtomic || Ordering == Unordered)
331739d628a0SDimitry Andric         return Error("Invalid record");
33186122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+3]);
33196122f3e6SDimitry Andric       I = new AtomicRMWInst(Operation, Ptr, Val, Ordering, SynchScope);
33206122f3e6SDimitry Andric       cast<AtomicRMWInst>(I)->setVolatile(Record[OpNum+1]);
33216122f3e6SDimitry Andric       InstructionList.push_back(I);
33226122f3e6SDimitry Andric       break;
33236122f3e6SDimitry Andric     }
33246122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_FENCE: { // FENCE:[ordering, synchscope]
33256122f3e6SDimitry Andric       if (2 != Record.size())
332639d628a0SDimitry Andric         return Error("Invalid record");
33276122f3e6SDimitry Andric       AtomicOrdering Ordering = GetDecodedOrdering(Record[0]);
33286122f3e6SDimitry Andric       if (Ordering == NotAtomic || Ordering == Unordered ||
33296122f3e6SDimitry Andric           Ordering == Monotonic)
333039d628a0SDimitry Andric         return Error("Invalid record");
33316122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[1]);
33326122f3e6SDimitry Andric       I = new FenceInst(Context, Ordering, SynchScope);
33336122f3e6SDimitry Andric       InstructionList.push_back(I);
33346122f3e6SDimitry Andric       break;
33356122f3e6SDimitry Andric     }
333617a519f9SDimitry Andric     case bitc::FUNC_CODE_INST_CALL: {
3337f22ef01cSRoman Divacky       // CALL: [paramattrs, cc, fnty, fnid, arg0, arg1...]
3338f22ef01cSRoman Divacky       if (Record.size() < 3)
333939d628a0SDimitry Andric         return Error("Invalid record");
3340f22ef01cSRoman Divacky 
3341139f7f9bSDimitry Andric       AttributeSet PAL = getAttributes(Record[0]);
3342f22ef01cSRoman Divacky       unsigned CCInfo = Record[1];
3343f22ef01cSRoman Divacky 
3344f22ef01cSRoman Divacky       unsigned OpNum = 2;
3345f22ef01cSRoman Divacky       Value *Callee;
3346f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
334739d628a0SDimitry Andric         return Error("Invalid record");
3348f22ef01cSRoman Divacky 
33496122f3e6SDimitry Andric       PointerType *OpTy = dyn_cast<PointerType>(Callee->getType());
335091bc56edSDimitry Andric       FunctionType *FTy = nullptr;
3351f22ef01cSRoman Divacky       if (OpTy) FTy = dyn_cast<FunctionType>(OpTy->getElementType());
3352f22ef01cSRoman Divacky       if (!FTy || Record.size() < FTy->getNumParams()+OpNum)
335339d628a0SDimitry Andric         return Error("Invalid record");
3354f22ef01cSRoman Divacky 
3355f22ef01cSRoman Divacky       SmallVector<Value*, 16> Args;
3356f22ef01cSRoman Divacky       // Read the fixed params.
3357f22ef01cSRoman Divacky       for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
335817a519f9SDimitry Andric         if (FTy->getParamType(i)->isLabelTy())
3359f22ef01cSRoman Divacky           Args.push_back(getBasicBlock(Record[OpNum]));
3360f22ef01cSRoman Divacky         else
33613861d79fSDimitry Andric           Args.push_back(getValue(Record, OpNum, NextValueNo,
33623861d79fSDimitry Andric                                   FTy->getParamType(i)));
336391bc56edSDimitry Andric         if (!Args.back())
336439d628a0SDimitry Andric           return Error("Invalid record");
3365f22ef01cSRoman Divacky       }
3366f22ef01cSRoman Divacky 
3367f22ef01cSRoman Divacky       // Read type/value pairs for varargs params.
3368f22ef01cSRoman Divacky       if (!FTy->isVarArg()) {
3369f22ef01cSRoman Divacky         if (OpNum != Record.size())
337039d628a0SDimitry Andric           return Error("Invalid record");
3371f22ef01cSRoman Divacky       } else {
3372f22ef01cSRoman Divacky         while (OpNum != Record.size()) {
3373f22ef01cSRoman Divacky           Value *Op;
3374f22ef01cSRoman Divacky           if (getValueTypePair(Record, OpNum, NextValueNo, Op))
337539d628a0SDimitry Andric             return Error("Invalid record");
3376f22ef01cSRoman Divacky           Args.push_back(Op);
3377f22ef01cSRoman Divacky         }
3378f22ef01cSRoman Divacky       }
3379f22ef01cSRoman Divacky 
338017a519f9SDimitry Andric       I = CallInst::Create(Callee, Args);
3381f22ef01cSRoman Divacky       InstructionList.push_back(I);
3382f22ef01cSRoman Divacky       cast<CallInst>(I)->setCallingConv(
338391bc56edSDimitry Andric           static_cast<CallingConv::ID>((~(1U << 14) & CCInfo) >> 1));
338491bc56edSDimitry Andric       CallInst::TailCallKind TCK = CallInst::TCK_None;
338591bc56edSDimitry Andric       if (CCInfo & 1)
338691bc56edSDimitry Andric         TCK = CallInst::TCK_Tail;
338791bc56edSDimitry Andric       if (CCInfo & (1 << 14))
338891bc56edSDimitry Andric         TCK = CallInst::TCK_MustTail;
338991bc56edSDimitry Andric       cast<CallInst>(I)->setTailCallKind(TCK);
3390f22ef01cSRoman Divacky       cast<CallInst>(I)->setAttributes(PAL);
3391f22ef01cSRoman Divacky       break;
3392f22ef01cSRoman Divacky     }
3393f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_VAARG: { // VAARG: [valistty, valist, instty]
3394f22ef01cSRoman Divacky       if (Record.size() < 3)
339539d628a0SDimitry Andric         return Error("Invalid record");
33966122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[0]);
33973861d79fSDimitry Andric       Value *Op = getValue(Record, 1, NextValueNo, OpTy);
33986122f3e6SDimitry Andric       Type *ResTy = getTypeByID(Record[2]);
3399f22ef01cSRoman Divacky       if (!OpTy || !Op || !ResTy)
340039d628a0SDimitry Andric         return Error("Invalid record");
3401f22ef01cSRoman Divacky       I = new VAArgInst(Op, ResTy);
3402f22ef01cSRoman Divacky       InstructionList.push_back(I);
3403f22ef01cSRoman Divacky       break;
3404f22ef01cSRoman Divacky     }
3405f22ef01cSRoman Divacky     }
3406f22ef01cSRoman Divacky 
3407f22ef01cSRoman Divacky     // Add instruction to end of current BB.  If there is no current BB, reject
3408f22ef01cSRoman Divacky     // this file.
340991bc56edSDimitry Andric     if (!CurBB) {
3410f22ef01cSRoman Divacky       delete I;
341139d628a0SDimitry Andric       return Error("Invalid instruction with no BB");
3412f22ef01cSRoman Divacky     }
3413f22ef01cSRoman Divacky     CurBB->getInstList().push_back(I);
3414f22ef01cSRoman Divacky 
3415f22ef01cSRoman Divacky     // If this was a terminator instruction, move to the next block.
3416f22ef01cSRoman Divacky     if (isa<TerminatorInst>(I)) {
3417f22ef01cSRoman Divacky       ++CurBBNo;
341891bc56edSDimitry Andric       CurBB = CurBBNo < FunctionBBs.size() ? FunctionBBs[CurBBNo] : nullptr;
3419f22ef01cSRoman Divacky     }
3420f22ef01cSRoman Divacky 
3421f22ef01cSRoman Divacky     // Non-void values get registered in the value table for future use.
3422f22ef01cSRoman Divacky     if (I && !I->getType()->isVoidTy())
3423f22ef01cSRoman Divacky       ValueList.AssignValue(I, NextValueNo++);
3424f22ef01cSRoman Divacky   }
3425f22ef01cSRoman Divacky 
3426139f7f9bSDimitry Andric OutOfRecordLoop:
3427139f7f9bSDimitry Andric 
3428f22ef01cSRoman Divacky   // Check the function list for unresolved values.
3429f22ef01cSRoman Divacky   if (Argument *A = dyn_cast<Argument>(ValueList.back())) {
343091bc56edSDimitry Andric     if (!A->getParent()) {
3431f22ef01cSRoman Divacky       // We found at least one unresolved value.  Nuke them all to avoid leaks.
3432f22ef01cSRoman Divacky       for (unsigned i = ModuleValueListSize, e = ValueList.size(); i != e; ++i){
343391bc56edSDimitry Andric         if ((A = dyn_cast_or_null<Argument>(ValueList[i])) && !A->getParent()) {
3434f22ef01cSRoman Divacky           A->replaceAllUsesWith(UndefValue::get(A->getType()));
3435f22ef01cSRoman Divacky           delete A;
3436f22ef01cSRoman Divacky         }
3437f22ef01cSRoman Divacky       }
343839d628a0SDimitry Andric       return Error("Never resolved value found in function");
3439f22ef01cSRoman Divacky     }
3440f22ef01cSRoman Divacky   }
3441f22ef01cSRoman Divacky 
3442e580952dSDimitry Andric   // FIXME: Check for unresolved forward-declared metadata references
3443e580952dSDimitry Andric   // and clean up leaks.
3444e580952dSDimitry Andric 
3445f22ef01cSRoman Divacky   // Trim the value list down to the size it was before we parsed this function.
3446f22ef01cSRoman Divacky   ValueList.shrinkTo(ModuleValueListSize);
3447e580952dSDimitry Andric   MDValueList.shrinkTo(ModuleMDValueListSize);
3448f22ef01cSRoman Divacky   std::vector<BasicBlock*>().swap(FunctionBBs);
344991bc56edSDimitry Andric   return std::error_code();
3450f22ef01cSRoman Divacky }
3451f22ef01cSRoman Divacky 
3452f785676fSDimitry Andric /// Find the function body in the bitcode stream
345391bc56edSDimitry Andric std::error_code BitcodeReader::FindFunctionInStream(
345491bc56edSDimitry Andric     Function *F,
3455dff0c46cSDimitry Andric     DenseMap<Function *, uint64_t>::iterator DeferredFunctionInfoIterator) {
3456dff0c46cSDimitry Andric   while (DeferredFunctionInfoIterator->second == 0) {
3457dff0c46cSDimitry Andric     if (Stream.AtEndOfStream())
345839d628a0SDimitry Andric       return Error("Could not find function in stream");
3459dff0c46cSDimitry Andric     // ParseModule will parse the next body in the stream and set its
3460dff0c46cSDimitry Andric     // position in the DeferredFunctionInfo map.
346191bc56edSDimitry Andric     if (std::error_code EC = ParseModule(true))
3462f785676fSDimitry Andric       return EC;
3463dff0c46cSDimitry Andric   }
346491bc56edSDimitry Andric   return std::error_code();
3465dff0c46cSDimitry Andric }
3466dff0c46cSDimitry Andric 
3467f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
3468f22ef01cSRoman Divacky // GVMaterializer implementation
3469f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
3470f22ef01cSRoman Divacky 
347191bc56edSDimitry Andric void BitcodeReader::releaseBuffer() { Buffer.release(); }
3472f22ef01cSRoman Divacky 
347339d628a0SDimitry Andric std::error_code BitcodeReader::materialize(GlobalValue *GV) {
3474f22ef01cSRoman Divacky   Function *F = dyn_cast<Function>(GV);
3475f22ef01cSRoman Divacky   // If it's not a function or is already material, ignore the request.
3476f785676fSDimitry Andric   if (!F || !F->isMaterializable())
347791bc56edSDimitry Andric     return std::error_code();
3478f22ef01cSRoman Divacky 
3479f22ef01cSRoman Divacky   DenseMap<Function*, uint64_t>::iterator DFII = DeferredFunctionInfo.find(F);
3480f22ef01cSRoman Divacky   assert(DFII != DeferredFunctionInfo.end() && "Deferred function not found!");
3481dff0c46cSDimitry Andric   // If its position is recorded as 0, its body is somewhere in the stream
3482dff0c46cSDimitry Andric   // but we haven't seen it yet.
3483f785676fSDimitry Andric   if (DFII->second == 0 && LazyStreamer)
348491bc56edSDimitry Andric     if (std::error_code EC = FindFunctionInStream(F, DFII))
3485f785676fSDimitry Andric       return EC;
3486f22ef01cSRoman Divacky 
3487f22ef01cSRoman Divacky   // Move the bit stream to the saved position of the deferred function body.
3488f22ef01cSRoman Divacky   Stream.JumpToBit(DFII->second);
3489f22ef01cSRoman Divacky 
349091bc56edSDimitry Andric   if (std::error_code EC = ParseFunctionBody(F))
3491f785676fSDimitry Andric     return EC;
349239d628a0SDimitry Andric   F->setIsMaterializable(false);
3493f22ef01cSRoman Divacky 
3494f22ef01cSRoman Divacky   // Upgrade any old intrinsic calls in the function.
3495f22ef01cSRoman Divacky   for (UpgradedIntrinsicMap::iterator I = UpgradedIntrinsics.begin(),
3496f22ef01cSRoman Divacky        E = UpgradedIntrinsics.end(); I != E; ++I) {
3497f22ef01cSRoman Divacky     if (I->first != I->second) {
349891bc56edSDimitry Andric       for (auto UI = I->first->user_begin(), UE = I->first->user_end();
349991bc56edSDimitry Andric            UI != UE;) {
3500f22ef01cSRoman Divacky         if (CallInst* CI = dyn_cast<CallInst>(*UI++))
3501f22ef01cSRoman Divacky           UpgradeIntrinsicCall(CI, I->second);
3502f22ef01cSRoman Divacky       }
3503f22ef01cSRoman Divacky     }
3504f22ef01cSRoman Divacky   }
3505f22ef01cSRoman Divacky 
350639d628a0SDimitry Andric   // Bring in any functions that this function forward-referenced via
350739d628a0SDimitry Andric   // blockaddresses.
350839d628a0SDimitry Andric   return materializeForwardReferencedFunctions();
3509f22ef01cSRoman Divacky }
3510f22ef01cSRoman Divacky 
3511f22ef01cSRoman Divacky bool BitcodeReader::isDematerializable(const GlobalValue *GV) const {
3512f22ef01cSRoman Divacky   const Function *F = dyn_cast<Function>(GV);
3513f22ef01cSRoman Divacky   if (!F || F->isDeclaration())
3514f22ef01cSRoman Divacky     return false;
351539d628a0SDimitry Andric 
351639d628a0SDimitry Andric   // Dematerializing F would leave dangling references that wouldn't be
351739d628a0SDimitry Andric   // reconnected on re-materialization.
351839d628a0SDimitry Andric   if (BlockAddressesTaken.count(F))
351939d628a0SDimitry Andric     return false;
352039d628a0SDimitry Andric 
3521f22ef01cSRoman Divacky   return DeferredFunctionInfo.count(const_cast<Function*>(F));
3522f22ef01cSRoman Divacky }
3523f22ef01cSRoman Divacky 
3524f22ef01cSRoman Divacky void BitcodeReader::Dematerialize(GlobalValue *GV) {
3525f22ef01cSRoman Divacky   Function *F = dyn_cast<Function>(GV);
3526f22ef01cSRoman Divacky   // If this function isn't dematerializable, this is a noop.
3527f22ef01cSRoman Divacky   if (!F || !isDematerializable(F))
3528f22ef01cSRoman Divacky     return;
3529f22ef01cSRoman Divacky 
3530f22ef01cSRoman Divacky   assert(DeferredFunctionInfo.count(F) && "No info to read function later?");
3531f22ef01cSRoman Divacky 
3532f22ef01cSRoman Divacky   // Just forget the function body, we can remat it later.
353339d628a0SDimitry Andric   F->dropAllReferences();
353439d628a0SDimitry Andric   F->setIsMaterializable(true);
3535f22ef01cSRoman Divacky }
3536f22ef01cSRoman Divacky 
353791bc56edSDimitry Andric std::error_code BitcodeReader::MaterializeModule(Module *M) {
3538f22ef01cSRoman Divacky   assert(M == TheModule &&
3539f22ef01cSRoman Divacky          "Can only Materialize the Module this BitcodeReader is attached to.");
354039d628a0SDimitry Andric 
354139d628a0SDimitry Andric   // Promise to materialize all forward references.
354239d628a0SDimitry Andric   WillMaterializeAllForwardRefs = true;
354339d628a0SDimitry Andric 
3544f22ef01cSRoman Divacky   // Iterate over the module, deserializing any functions that are still on
3545f22ef01cSRoman Divacky   // disk.
3546f22ef01cSRoman Divacky   for (Module::iterator F = TheModule->begin(), E = TheModule->end();
3547f785676fSDimitry Andric        F != E; ++F) {
354839d628a0SDimitry Andric     if (std::error_code EC = materialize(F))
3549f785676fSDimitry Andric       return EC;
3550f785676fSDimitry Andric   }
3551dff0c46cSDimitry Andric   // At this point, if there are any function bodies, the current bit is
3552dff0c46cSDimitry Andric   // pointing to the END_BLOCK record after them. Now make sure the rest
3553dff0c46cSDimitry Andric   // of the bits in the module have been read.
3554dff0c46cSDimitry Andric   if (NextUnreadBit)
3555dff0c46cSDimitry Andric     ParseModule(true);
3556dff0c46cSDimitry Andric 
355739d628a0SDimitry Andric   // Check that all block address forward references got resolved (as we
355839d628a0SDimitry Andric   // promised above).
355939d628a0SDimitry Andric   if (!BasicBlockFwdRefs.empty())
356039d628a0SDimitry Andric     return Error("Never resolved function from blockaddress");
356139d628a0SDimitry Andric 
3562f22ef01cSRoman Divacky   // Upgrade any intrinsic calls that slipped through (should not happen!) and
3563f22ef01cSRoman Divacky   // delete the old functions to clean up. We can't do this unless the entire
3564f22ef01cSRoman Divacky   // module is materialized because there could always be another function body
3565f22ef01cSRoman Divacky   // with calls to the old function.
3566f22ef01cSRoman Divacky   for (std::vector<std::pair<Function*, Function*> >::iterator I =
3567f22ef01cSRoman Divacky        UpgradedIntrinsics.begin(), E = UpgradedIntrinsics.end(); I != E; ++I) {
3568f22ef01cSRoman Divacky     if (I->first != I->second) {
356991bc56edSDimitry Andric       for (auto UI = I->first->user_begin(), UE = I->first->user_end();
357091bc56edSDimitry Andric            UI != UE;) {
3571f22ef01cSRoman Divacky         if (CallInst* CI = dyn_cast<CallInst>(*UI++))
3572f22ef01cSRoman Divacky           UpgradeIntrinsicCall(CI, I->second);
3573f22ef01cSRoman Divacky       }
3574f22ef01cSRoman Divacky       if (!I->first->use_empty())
3575f22ef01cSRoman Divacky         I->first->replaceAllUsesWith(I->second);
3576f22ef01cSRoman Divacky       I->first->eraseFromParent();
3577f22ef01cSRoman Divacky     }
3578f22ef01cSRoman Divacky   }
3579f22ef01cSRoman Divacky   std::vector<std::pair<Function*, Function*> >().swap(UpgradedIntrinsics);
3580f22ef01cSRoman Divacky 
3581f785676fSDimitry Andric   for (unsigned I = 0, E = InstsWithTBAATag.size(); I < E; I++)
3582f785676fSDimitry Andric     UpgradeInstWithTBAATag(InstsWithTBAATag[I]);
3583f785676fSDimitry Andric 
3584f785676fSDimitry Andric   UpgradeDebugInfo(*M);
358591bc56edSDimitry Andric   return std::error_code();
3586dff0c46cSDimitry Andric }
35876122f3e6SDimitry Andric 
358839d628a0SDimitry Andric std::vector<StructType *> BitcodeReader::getIdentifiedStructTypes() const {
358939d628a0SDimitry Andric   return IdentifiedStructTypes;
359039d628a0SDimitry Andric }
359139d628a0SDimitry Andric 
359291bc56edSDimitry Andric std::error_code BitcodeReader::InitStream() {
3593f785676fSDimitry Andric   if (LazyStreamer)
3594f785676fSDimitry Andric     return InitLazyStream();
3595dff0c46cSDimitry Andric   return InitStreamFromBuffer();
3596dff0c46cSDimitry Andric }
3597dff0c46cSDimitry Andric 
359891bc56edSDimitry Andric std::error_code BitcodeReader::InitStreamFromBuffer() {
35993861d79fSDimitry Andric   const unsigned char *BufPtr = (const unsigned char*)Buffer->getBufferStart();
3600dff0c46cSDimitry Andric   const unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
3601dff0c46cSDimitry Andric 
360239d628a0SDimitry Andric   if (Buffer->getBufferSize() & 3)
360339d628a0SDimitry Andric     return Error("Invalid bitcode signature");
3604dff0c46cSDimitry Andric 
3605dff0c46cSDimitry Andric   // If we have a wrapper header, parse it and ignore the non-bc file contents.
3606dff0c46cSDimitry Andric   // The magic number is 0x0B17C0DE stored in little endian.
3607dff0c46cSDimitry Andric   if (isBitcodeWrapper(BufPtr, BufEnd))
3608dff0c46cSDimitry Andric     if (SkipBitcodeWrapperHeader(BufPtr, BufEnd, true))
360939d628a0SDimitry Andric       return Error("Invalid bitcode wrapper header");
3610dff0c46cSDimitry Andric 
3611dff0c46cSDimitry Andric   StreamFile.reset(new BitstreamReader(BufPtr, BufEnd));
361239d628a0SDimitry Andric   Stream.init(&*StreamFile);
3613f22ef01cSRoman Divacky 
361491bc56edSDimitry Andric   return std::error_code();
3615f22ef01cSRoman Divacky }
3616f22ef01cSRoman Divacky 
361791bc56edSDimitry Andric std::error_code BitcodeReader::InitLazyStream() {
3618dff0c46cSDimitry Andric   // Check and strip off the bitcode wrapper; BitstreamReader expects never to
3619dff0c46cSDimitry Andric   // see it.
362039d628a0SDimitry Andric   auto OwnedBytes = llvm::make_unique<StreamingMemoryObject>(LazyStreamer);
362139d628a0SDimitry Andric   StreamingMemoryObject &Bytes = *OwnedBytes;
362239d628a0SDimitry Andric   StreamFile = llvm::make_unique<BitstreamReader>(std::move(OwnedBytes));
362339d628a0SDimitry Andric   Stream.init(&*StreamFile);
3624dff0c46cSDimitry Andric 
3625dff0c46cSDimitry Andric   unsigned char buf[16];
362639d628a0SDimitry Andric   if (Bytes.readBytes(buf, 16, 0) != 16)
362739d628a0SDimitry Andric     return Error("Invalid bitcode signature");
3628dff0c46cSDimitry Andric 
3629dff0c46cSDimitry Andric   if (!isBitcode(buf, buf + 16))
363039d628a0SDimitry Andric     return Error("Invalid bitcode signature");
3631dff0c46cSDimitry Andric 
3632dff0c46cSDimitry Andric   if (isBitcodeWrapper(buf, buf + 4)) {
3633dff0c46cSDimitry Andric     const unsigned char *bitcodeStart = buf;
3634dff0c46cSDimitry Andric     const unsigned char *bitcodeEnd = buf + 16;
3635dff0c46cSDimitry Andric     SkipBitcodeWrapperHeader(bitcodeStart, bitcodeEnd, false);
363639d628a0SDimitry Andric     Bytes.dropLeadingBytes(bitcodeStart - buf);
363739d628a0SDimitry Andric     Bytes.setKnownObjectSize(bitcodeEnd - bitcodeStart);
3638dff0c46cSDimitry Andric   }
363991bc56edSDimitry Andric   return std::error_code();
3640f785676fSDimitry Andric }
3641f785676fSDimitry Andric 
3642f785676fSDimitry Andric namespace {
364391bc56edSDimitry Andric class BitcodeErrorCategoryType : public std::error_category {
364491bc56edSDimitry Andric   const char *name() const LLVM_NOEXCEPT override {
3645f785676fSDimitry Andric     return "llvm.bitcode";
3646f785676fSDimitry Andric   }
364791bc56edSDimitry Andric   std::string message(int IE) const override {
364839d628a0SDimitry Andric     BitcodeError E = static_cast<BitcodeError>(IE);
3649f785676fSDimitry Andric     switch (E) {
365039d628a0SDimitry Andric     case BitcodeError::InvalidBitcodeSignature:
3651f785676fSDimitry Andric       return "Invalid bitcode signature";
365239d628a0SDimitry Andric     case BitcodeError::CorruptedBitcode:
365339d628a0SDimitry Andric       return "Corrupted bitcode";
3654f785676fSDimitry Andric     }
3655f785676fSDimitry Andric     llvm_unreachable("Unknown error type!");
3656f785676fSDimitry Andric   }
3657f785676fSDimitry Andric };
3658f785676fSDimitry Andric }
3659f785676fSDimitry Andric 
366039d628a0SDimitry Andric static ManagedStatic<BitcodeErrorCategoryType> ErrorCategory;
366139d628a0SDimitry Andric 
366239d628a0SDimitry Andric const std::error_category &llvm::BitcodeErrorCategory() {
366339d628a0SDimitry Andric   return *ErrorCategory;
3664dff0c46cSDimitry Andric }
3665f22ef01cSRoman Divacky 
3666f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
3667f22ef01cSRoman Divacky // External interface
3668f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
3669f22ef01cSRoman Divacky 
367039d628a0SDimitry Andric /// \brief Get a lazy one-at-time loading module from bitcode.
3671f22ef01cSRoman Divacky ///
367239d628a0SDimitry Andric /// This isn't always used in a lazy context.  In particular, it's also used by
367339d628a0SDimitry Andric /// \a parseBitcodeFile().  If this is truly lazy, then we need to eagerly pull
367439d628a0SDimitry Andric /// in forward-referenced functions from block address references.
367539d628a0SDimitry Andric ///
367639d628a0SDimitry Andric /// \param[in] WillMaterializeAll Set to \c true if the caller promises to
367739d628a0SDimitry Andric /// materialize everything -- in particular, if this isn't truly lazy.
367839d628a0SDimitry Andric static ErrorOr<Module *>
367939d628a0SDimitry Andric getLazyBitcodeModuleImpl(std::unique_ptr<MemoryBuffer> &&Buffer,
368039d628a0SDimitry Andric                          LLVMContext &Context, bool WillMaterializeAll,
368139d628a0SDimitry Andric                          DiagnosticHandlerFunction DiagnosticHandler) {
3682f22ef01cSRoman Divacky   Module *M = new Module(Buffer->getBufferIdentifier(), Context);
368339d628a0SDimitry Andric   BitcodeReader *R =
368439d628a0SDimitry Andric       new BitcodeReader(Buffer.get(), Context, DiagnosticHandler);
3685f22ef01cSRoman Divacky   M->setMaterializer(R);
368639d628a0SDimitry Andric 
368739d628a0SDimitry Andric   auto cleanupOnError = [&](std::error_code EC) {
368891bc56edSDimitry Andric     R->releaseBuffer(); // Never take ownership on error.
3689f22ef01cSRoman Divacky     delete M;  // Also deletes R.
369091bc56edSDimitry Andric     return EC;
369139d628a0SDimitry Andric   };
3692dff0c46cSDimitry Andric 
369339d628a0SDimitry Andric   if (std::error_code EC = R->ParseBitcodeInto(M))
369439d628a0SDimitry Andric     return cleanupOnError(EC);
3695dff0c46cSDimitry Andric 
369639d628a0SDimitry Andric   if (!WillMaterializeAll)
369739d628a0SDimitry Andric     // Resolve forward references from blockaddresses.
369839d628a0SDimitry Andric     if (std::error_code EC = R->materializeForwardReferencedFunctions())
369939d628a0SDimitry Andric       return cleanupOnError(EC);
370039d628a0SDimitry Andric 
370139d628a0SDimitry Andric   Buffer.release(); // The BitcodeReader owns it now.
3702dff0c46cSDimitry Andric   return M;
3703dff0c46cSDimitry Andric }
3704dff0c46cSDimitry Andric 
370539d628a0SDimitry Andric ErrorOr<Module *>
370639d628a0SDimitry Andric llvm::getLazyBitcodeModule(std::unique_ptr<MemoryBuffer> &&Buffer,
3707dff0c46cSDimitry Andric                            LLVMContext &Context,
370839d628a0SDimitry Andric                            DiagnosticHandlerFunction DiagnosticHandler) {
370939d628a0SDimitry Andric   return getLazyBitcodeModuleImpl(std::move(Buffer), Context, false,
371039d628a0SDimitry Andric                                   DiagnosticHandler);
3711f22ef01cSRoman Divacky }
3712f22ef01cSRoman Divacky 
371339d628a0SDimitry Andric ErrorOr<std::unique_ptr<Module>>
371439d628a0SDimitry Andric llvm::getStreamedBitcodeModule(StringRef Name, DataStreamer *Streamer,
371539d628a0SDimitry Andric                                LLVMContext &Context,
371639d628a0SDimitry Andric                                DiagnosticHandlerFunction DiagnosticHandler) {
371739d628a0SDimitry Andric   std::unique_ptr<Module> M = make_unique<Module>(Name, Context);
371839d628a0SDimitry Andric   BitcodeReader *R = new BitcodeReader(Streamer, Context, DiagnosticHandler);
371939d628a0SDimitry Andric   M->setMaterializer(R);
372039d628a0SDimitry Andric   if (std::error_code EC = R->ParseBitcodeInto(M.get()))
372139d628a0SDimitry Andric     return EC;
372239d628a0SDimitry Andric   return std::move(M);
372339d628a0SDimitry Andric }
372439d628a0SDimitry Andric 
372539d628a0SDimitry Andric ErrorOr<Module *>
372639d628a0SDimitry Andric llvm::parseBitcodeFile(MemoryBufferRef Buffer, LLVMContext &Context,
372739d628a0SDimitry Andric                        DiagnosticHandlerFunction DiagnosticHandler) {
372839d628a0SDimitry Andric   std::unique_ptr<MemoryBuffer> Buf = MemoryBuffer::getMemBuffer(Buffer, false);
372939d628a0SDimitry Andric   ErrorOr<Module *> ModuleOrErr = getLazyBitcodeModuleImpl(
373039d628a0SDimitry Andric       std::move(Buf), Context, true, DiagnosticHandler);
373191bc56edSDimitry Andric   if (!ModuleOrErr)
373291bc56edSDimitry Andric     return ModuleOrErr;
373391bc56edSDimitry Andric   Module *M = ModuleOrErr.get();
3734f22ef01cSRoman Divacky   // Read in the entire module, and destroy the BitcodeReader.
373539d628a0SDimitry Andric   if (std::error_code EC = M->materializeAllPermanently()) {
3736f22ef01cSRoman Divacky     delete M;
373791bc56edSDimitry Andric     return EC;
3738f22ef01cSRoman Divacky   }
37392754fe60SDimitry Andric 
3740dff0c46cSDimitry Andric   // TODO: Restore the use-lists to the in-memory state when the bitcode was
3741dff0c46cSDimitry Andric   // written.  We must defer until the Module has been fully materialized.
3742dff0c46cSDimitry Andric 
3743f22ef01cSRoman Divacky   return M;
3744f22ef01cSRoman Divacky }
37452754fe60SDimitry Andric 
374639d628a0SDimitry Andric std::string
374739d628a0SDimitry Andric llvm::getBitcodeTargetTriple(MemoryBufferRef Buffer, LLVMContext &Context,
374839d628a0SDimitry Andric                              DiagnosticHandlerFunction DiagnosticHandler) {
374939d628a0SDimitry Andric   std::unique_ptr<MemoryBuffer> Buf = MemoryBuffer::getMemBuffer(Buffer, false);
375039d628a0SDimitry Andric   auto R = llvm::make_unique<BitcodeReader>(Buf.release(), Context,
375139d628a0SDimitry Andric                                             DiagnosticHandler);
375291bc56edSDimitry Andric   ErrorOr<std::string> Triple = R->parseTriple();
375391bc56edSDimitry Andric   if (Triple.getError())
375491bc56edSDimitry Andric     return "";
375591bc56edSDimitry Andric   return Triple.get();
37562754fe60SDimitry Andric }
3757