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"
18139f7f9bSDimitry Andric #include "llvm/IR/InlineAsm.h"
19139f7f9bSDimitry Andric #include "llvm/IR/IntrinsicInst.h"
20f785676fSDimitry Andric #include "llvm/IR/LLVMContext.h"
21139f7f9bSDimitry Andric #include "llvm/IR/Module.h"
22139f7f9bSDimitry Andric #include "llvm/IR/OperandTraits.h"
23139f7f9bSDimitry Andric #include "llvm/IR/Operator.h"
24dff0c46cSDimitry Andric #include "llvm/Support/DataStream.h"
25f22ef01cSRoman Divacky #include "llvm/Support/MathExtras.h"
26f22ef01cSRoman Divacky #include "llvm/Support/MemoryBuffer.h"
27f785676fSDimitry Andric #include "llvm/Support/raw_ostream.h"
28f22ef01cSRoman Divacky using namespace llvm;
29f22ef01cSRoman Divacky 
307ae0e2c9SDimitry Andric enum {
317ae0e2c9SDimitry Andric   SWITCH_INST_MAGIC = 0x4B5 // May 2012 => 1205 => Hex
327ae0e2c9SDimitry Andric };
337ae0e2c9SDimitry Andric 
34dff0c46cSDimitry Andric void BitcodeReader::materializeForwardReferencedFunctions() {
35dff0c46cSDimitry Andric   while (!BlockAddrFwdRefs.empty()) {
36dff0c46cSDimitry Andric     Function *F = BlockAddrFwdRefs.begin()->first;
37dff0c46cSDimitry Andric     F->Materialize();
38dff0c46cSDimitry Andric   }
39dff0c46cSDimitry Andric }
40dff0c46cSDimitry Andric 
41f22ef01cSRoman Divacky void BitcodeReader::FreeState() {
4291bc56edSDimitry Andric   Buffer = nullptr;
4317a519f9SDimitry Andric   std::vector<Type*>().swap(TypeList);
44f22ef01cSRoman Divacky   ValueList.clear();
45f22ef01cSRoman Divacky   MDValueList.clear();
4691bc56edSDimitry Andric   std::vector<Comdat *>().swap(ComdatList);
47f22ef01cSRoman Divacky 
48139f7f9bSDimitry Andric   std::vector<AttributeSet>().swap(MAttributes);
49f22ef01cSRoman Divacky   std::vector<BasicBlock*>().swap(FunctionBBs);
50f22ef01cSRoman Divacky   std::vector<Function*>().swap(FunctionsWithBodies);
51f22ef01cSRoman Divacky   DeferredFunctionInfo.clear();
52e580952dSDimitry Andric   MDKindMap.clear();
533861d79fSDimitry Andric 
543861d79fSDimitry Andric   assert(BlockAddrFwdRefs.empty() && "Unresolved blockaddress fwd references");
55f22ef01cSRoman Divacky }
56f22ef01cSRoman Divacky 
57f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
58f22ef01cSRoman Divacky //  Helper functions to implement forward reference resolution, etc.
59f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
60f22ef01cSRoman Divacky 
61f22ef01cSRoman Divacky /// ConvertToString - Convert a string from a record into an std::string, return
62f22ef01cSRoman Divacky /// true on failure.
63f22ef01cSRoman Divacky template<typename StrTy>
647ae0e2c9SDimitry Andric static bool ConvertToString(ArrayRef<uint64_t> Record, unsigned Idx,
65f22ef01cSRoman Divacky                             StrTy &Result) {
66f22ef01cSRoman Divacky   if (Idx > Record.size())
67f22ef01cSRoman Divacky     return true;
68f22ef01cSRoman Divacky 
69f22ef01cSRoman Divacky   for (unsigned i = Idx, e = Record.size(); i != e; ++i)
70f22ef01cSRoman Divacky     Result += (char)Record[i];
71f22ef01cSRoman Divacky   return false;
72f22ef01cSRoman Divacky }
73f22ef01cSRoman Divacky 
74f22ef01cSRoman Divacky static GlobalValue::LinkageTypes GetDecodedLinkage(unsigned Val) {
75f22ef01cSRoman Divacky   switch (Val) {
76f22ef01cSRoman Divacky   default: // Map unknown/new linkages to external
77f22ef01cSRoman Divacky   case 0:  return GlobalValue::ExternalLinkage;
78f22ef01cSRoman Divacky   case 1:  return GlobalValue::WeakAnyLinkage;
79f22ef01cSRoman Divacky   case 2:  return GlobalValue::AppendingLinkage;
80f22ef01cSRoman Divacky   case 3:  return GlobalValue::InternalLinkage;
81f22ef01cSRoman Divacky   case 4:  return GlobalValue::LinkOnceAnyLinkage;
8291bc56edSDimitry Andric   case 5:  return GlobalValue::ExternalLinkage; // Obsolete DLLImportLinkage
8391bc56edSDimitry Andric   case 6:  return GlobalValue::ExternalLinkage; // Obsolete DLLExportLinkage
84f22ef01cSRoman Divacky   case 7:  return GlobalValue::ExternalWeakLinkage;
85f22ef01cSRoman Divacky   case 8:  return GlobalValue::CommonLinkage;
86f22ef01cSRoman Divacky   case 9:  return GlobalValue::PrivateLinkage;
87f22ef01cSRoman Divacky   case 10: return GlobalValue::WeakODRLinkage;
88f22ef01cSRoman Divacky   case 11: return GlobalValue::LinkOnceODRLinkage;
89f22ef01cSRoman Divacky   case 12: return GlobalValue::AvailableExternallyLinkage;
9091bc56edSDimitry Andric   case 13:
9191bc56edSDimitry Andric     return GlobalValue::PrivateLinkage; // Obsolete LinkerPrivateLinkage
9291bc56edSDimitry Andric   case 14:
9391bc56edSDimitry Andric     return GlobalValue::PrivateLinkage; // Obsolete LinkerPrivateWeakLinkage
94f22ef01cSRoman Divacky   }
95f22ef01cSRoman Divacky }
96f22ef01cSRoman Divacky 
97f22ef01cSRoman Divacky static GlobalValue::VisibilityTypes GetDecodedVisibility(unsigned Val) {
98f22ef01cSRoman Divacky   switch (Val) {
99f22ef01cSRoman Divacky   default: // Map unknown visibilities to default.
100f22ef01cSRoman Divacky   case 0: return GlobalValue::DefaultVisibility;
101f22ef01cSRoman Divacky   case 1: return GlobalValue::HiddenVisibility;
102f22ef01cSRoman Divacky   case 2: return GlobalValue::ProtectedVisibility;
103f22ef01cSRoman Divacky   }
104f22ef01cSRoman Divacky }
105f22ef01cSRoman Divacky 
10691bc56edSDimitry Andric static GlobalValue::DLLStorageClassTypes
10791bc56edSDimitry Andric GetDecodedDLLStorageClass(unsigned Val) {
10891bc56edSDimitry Andric   switch (Val) {
10991bc56edSDimitry Andric   default: // Map unknown values to default.
11091bc56edSDimitry Andric   case 0: return GlobalValue::DefaultStorageClass;
11191bc56edSDimitry Andric   case 1: return GlobalValue::DLLImportStorageClass;
11291bc56edSDimitry Andric   case 2: return GlobalValue::DLLExportStorageClass;
11391bc56edSDimitry Andric   }
11491bc56edSDimitry Andric }
11591bc56edSDimitry Andric 
1167ae0e2c9SDimitry Andric static GlobalVariable::ThreadLocalMode GetDecodedThreadLocalMode(unsigned Val) {
1177ae0e2c9SDimitry Andric   switch (Val) {
1187ae0e2c9SDimitry Andric     case 0: return GlobalVariable::NotThreadLocal;
1197ae0e2c9SDimitry Andric     default: // Map unknown non-zero value to general dynamic.
1207ae0e2c9SDimitry Andric     case 1: return GlobalVariable::GeneralDynamicTLSModel;
1217ae0e2c9SDimitry Andric     case 2: return GlobalVariable::LocalDynamicTLSModel;
1227ae0e2c9SDimitry Andric     case 3: return GlobalVariable::InitialExecTLSModel;
1237ae0e2c9SDimitry Andric     case 4: return GlobalVariable::LocalExecTLSModel;
1247ae0e2c9SDimitry Andric   }
1257ae0e2c9SDimitry Andric }
1267ae0e2c9SDimitry Andric 
127f22ef01cSRoman Divacky static int GetDecodedCastOpcode(unsigned Val) {
128f22ef01cSRoman Divacky   switch (Val) {
129f22ef01cSRoman Divacky   default: return -1;
130f22ef01cSRoman Divacky   case bitc::CAST_TRUNC   : return Instruction::Trunc;
131f22ef01cSRoman Divacky   case bitc::CAST_ZEXT    : return Instruction::ZExt;
132f22ef01cSRoman Divacky   case bitc::CAST_SEXT    : return Instruction::SExt;
133f22ef01cSRoman Divacky   case bitc::CAST_FPTOUI  : return Instruction::FPToUI;
134f22ef01cSRoman Divacky   case bitc::CAST_FPTOSI  : return Instruction::FPToSI;
135f22ef01cSRoman Divacky   case bitc::CAST_UITOFP  : return Instruction::UIToFP;
136f22ef01cSRoman Divacky   case bitc::CAST_SITOFP  : return Instruction::SIToFP;
137f22ef01cSRoman Divacky   case bitc::CAST_FPTRUNC : return Instruction::FPTrunc;
138f22ef01cSRoman Divacky   case bitc::CAST_FPEXT   : return Instruction::FPExt;
139f22ef01cSRoman Divacky   case bitc::CAST_PTRTOINT: return Instruction::PtrToInt;
140f22ef01cSRoman Divacky   case bitc::CAST_INTTOPTR: return Instruction::IntToPtr;
141f22ef01cSRoman Divacky   case bitc::CAST_BITCAST : return Instruction::BitCast;
142f785676fSDimitry Andric   case bitc::CAST_ADDRSPACECAST: return Instruction::AddrSpaceCast;
143f22ef01cSRoman Divacky   }
144f22ef01cSRoman Divacky }
1456122f3e6SDimitry Andric static int GetDecodedBinaryOpcode(unsigned Val, Type *Ty) {
146f22ef01cSRoman Divacky   switch (Val) {
147f22ef01cSRoman Divacky   default: return -1;
148f22ef01cSRoman Divacky   case bitc::BINOP_ADD:
149f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FAdd : Instruction::Add;
150f22ef01cSRoman Divacky   case bitc::BINOP_SUB:
151f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FSub : Instruction::Sub;
152f22ef01cSRoman Divacky   case bitc::BINOP_MUL:
153f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FMul : Instruction::Mul;
154f22ef01cSRoman Divacky   case bitc::BINOP_UDIV: return Instruction::UDiv;
155f22ef01cSRoman Divacky   case bitc::BINOP_SDIV:
156f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FDiv : Instruction::SDiv;
157f22ef01cSRoman Divacky   case bitc::BINOP_UREM: return Instruction::URem;
158f22ef01cSRoman Divacky   case bitc::BINOP_SREM:
159f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FRem : Instruction::SRem;
160f22ef01cSRoman Divacky   case bitc::BINOP_SHL:  return Instruction::Shl;
161f22ef01cSRoman Divacky   case bitc::BINOP_LSHR: return Instruction::LShr;
162f22ef01cSRoman Divacky   case bitc::BINOP_ASHR: return Instruction::AShr;
163f22ef01cSRoman Divacky   case bitc::BINOP_AND:  return Instruction::And;
164f22ef01cSRoman Divacky   case bitc::BINOP_OR:   return Instruction::Or;
165f22ef01cSRoman Divacky   case bitc::BINOP_XOR:  return Instruction::Xor;
166f22ef01cSRoman Divacky   }
167f22ef01cSRoman Divacky }
168f22ef01cSRoman Divacky 
1696122f3e6SDimitry Andric static AtomicRMWInst::BinOp GetDecodedRMWOperation(unsigned Val) {
1706122f3e6SDimitry Andric   switch (Val) {
1716122f3e6SDimitry Andric   default: return AtomicRMWInst::BAD_BINOP;
1726122f3e6SDimitry Andric   case bitc::RMW_XCHG: return AtomicRMWInst::Xchg;
1736122f3e6SDimitry Andric   case bitc::RMW_ADD: return AtomicRMWInst::Add;
1746122f3e6SDimitry Andric   case bitc::RMW_SUB: return AtomicRMWInst::Sub;
1756122f3e6SDimitry Andric   case bitc::RMW_AND: return AtomicRMWInst::And;
1766122f3e6SDimitry Andric   case bitc::RMW_NAND: return AtomicRMWInst::Nand;
1776122f3e6SDimitry Andric   case bitc::RMW_OR: return AtomicRMWInst::Or;
1786122f3e6SDimitry Andric   case bitc::RMW_XOR: return AtomicRMWInst::Xor;
1796122f3e6SDimitry Andric   case bitc::RMW_MAX: return AtomicRMWInst::Max;
1806122f3e6SDimitry Andric   case bitc::RMW_MIN: return AtomicRMWInst::Min;
1816122f3e6SDimitry Andric   case bitc::RMW_UMAX: return AtomicRMWInst::UMax;
1826122f3e6SDimitry Andric   case bitc::RMW_UMIN: return AtomicRMWInst::UMin;
1836122f3e6SDimitry Andric   }
1846122f3e6SDimitry Andric }
1856122f3e6SDimitry Andric 
1866122f3e6SDimitry Andric static AtomicOrdering GetDecodedOrdering(unsigned Val) {
1876122f3e6SDimitry Andric   switch (Val) {
1886122f3e6SDimitry Andric   case bitc::ORDERING_NOTATOMIC: return NotAtomic;
1896122f3e6SDimitry Andric   case bitc::ORDERING_UNORDERED: return Unordered;
1906122f3e6SDimitry Andric   case bitc::ORDERING_MONOTONIC: return Monotonic;
1916122f3e6SDimitry Andric   case bitc::ORDERING_ACQUIRE: return Acquire;
1926122f3e6SDimitry Andric   case bitc::ORDERING_RELEASE: return Release;
1936122f3e6SDimitry Andric   case bitc::ORDERING_ACQREL: return AcquireRelease;
1946122f3e6SDimitry Andric   default: // Map unknown orderings to sequentially-consistent.
1956122f3e6SDimitry Andric   case bitc::ORDERING_SEQCST: return SequentiallyConsistent;
1966122f3e6SDimitry Andric   }
1976122f3e6SDimitry Andric }
1986122f3e6SDimitry Andric 
1996122f3e6SDimitry Andric static SynchronizationScope GetDecodedSynchScope(unsigned Val) {
2006122f3e6SDimitry Andric   switch (Val) {
2016122f3e6SDimitry Andric   case bitc::SYNCHSCOPE_SINGLETHREAD: return SingleThread;
2026122f3e6SDimitry Andric   default: // Map unknown scopes to cross-thread.
2036122f3e6SDimitry Andric   case bitc::SYNCHSCOPE_CROSSTHREAD: return CrossThread;
2046122f3e6SDimitry Andric   }
2056122f3e6SDimitry Andric }
2066122f3e6SDimitry Andric 
20791bc56edSDimitry Andric static Comdat::SelectionKind getDecodedComdatSelectionKind(unsigned Val) {
20891bc56edSDimitry Andric   switch (Val) {
20991bc56edSDimitry Andric   default: // Map unknown selection kinds to any.
21091bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_ANY:
21191bc56edSDimitry Andric     return Comdat::Any;
21291bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_EXACT_MATCH:
21391bc56edSDimitry Andric     return Comdat::ExactMatch;
21491bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_LARGEST:
21591bc56edSDimitry Andric     return Comdat::Largest;
21691bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_NO_DUPLICATES:
21791bc56edSDimitry Andric     return Comdat::NoDuplicates;
21891bc56edSDimitry Andric   case bitc::COMDAT_SELECTION_KIND_SAME_SIZE:
21991bc56edSDimitry Andric     return Comdat::SameSize;
22091bc56edSDimitry Andric   }
22191bc56edSDimitry Andric }
22291bc56edSDimitry Andric 
22391bc56edSDimitry Andric static void UpgradeDLLImportExportLinkage(llvm::GlobalValue *GV, unsigned Val) {
22491bc56edSDimitry Andric   switch (Val) {
22591bc56edSDimitry Andric   case 5: GV->setDLLStorageClass(GlobalValue::DLLImportStorageClass); break;
22691bc56edSDimitry Andric   case 6: GV->setDLLStorageClass(GlobalValue::DLLExportStorageClass); break;
22791bc56edSDimitry Andric   }
22891bc56edSDimitry Andric }
22991bc56edSDimitry Andric 
230f22ef01cSRoman Divacky namespace llvm {
231f22ef01cSRoman Divacky namespace {
232f22ef01cSRoman Divacky   /// @brief A class for maintaining the slot number definition
233f22ef01cSRoman Divacky   /// as a placeholder for the actual definition for forward constants defs.
234f22ef01cSRoman Divacky   class ConstantPlaceHolder : public ConstantExpr {
2353861d79fSDimitry Andric     void operator=(const ConstantPlaceHolder &) LLVM_DELETED_FUNCTION;
236f22ef01cSRoman Divacky   public:
237f22ef01cSRoman Divacky     // allocate space for exactly one operand
238f22ef01cSRoman Divacky     void *operator new(size_t s) {
239f22ef01cSRoman Divacky       return User::operator new(s, 1);
240f22ef01cSRoman Divacky     }
2416122f3e6SDimitry Andric     explicit ConstantPlaceHolder(Type *Ty, LLVMContext& Context)
242f22ef01cSRoman Divacky       : ConstantExpr(Ty, Instruction::UserOp1, &Op<0>(), 1) {
243f22ef01cSRoman Divacky       Op<0>() = UndefValue::get(Type::getInt32Ty(Context));
244f22ef01cSRoman Divacky     }
245f22ef01cSRoman Divacky 
246f22ef01cSRoman Divacky     /// @brief Methods to support type inquiry through isa, cast, and dyn_cast.
247f22ef01cSRoman Divacky     static bool classof(const Value *V) {
248f22ef01cSRoman Divacky       return isa<ConstantExpr>(V) &&
249f22ef01cSRoman Divacky              cast<ConstantExpr>(V)->getOpcode() == Instruction::UserOp1;
250f22ef01cSRoman Divacky     }
251f22ef01cSRoman Divacky 
252f22ef01cSRoman Divacky 
253f22ef01cSRoman Divacky     /// Provide fast operand accessors
254f22ef01cSRoman Divacky     //DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
255f22ef01cSRoman Divacky   };
256f22ef01cSRoman Divacky }
257f22ef01cSRoman Divacky 
258f22ef01cSRoman Divacky // FIXME: can we inherit this from ConstantExpr?
259f22ef01cSRoman Divacky template <>
2602754fe60SDimitry Andric struct OperandTraits<ConstantPlaceHolder> :
2612754fe60SDimitry Andric   public FixedNumOperandTraits<ConstantPlaceHolder, 1> {
262f22ef01cSRoman Divacky };
263f22ef01cSRoman Divacky }
264f22ef01cSRoman Divacky 
265f22ef01cSRoman Divacky 
266f22ef01cSRoman Divacky void BitcodeReaderValueList::AssignValue(Value *V, unsigned Idx) {
267f22ef01cSRoman Divacky   if (Idx == size()) {
268f22ef01cSRoman Divacky     push_back(V);
269f22ef01cSRoman Divacky     return;
270f22ef01cSRoman Divacky   }
271f22ef01cSRoman Divacky 
272f22ef01cSRoman Divacky   if (Idx >= size())
273f22ef01cSRoman Divacky     resize(Idx+1);
274f22ef01cSRoman Divacky 
275f22ef01cSRoman Divacky   WeakVH &OldV = ValuePtrs[Idx];
27691bc56edSDimitry Andric   if (!OldV) {
277f22ef01cSRoman Divacky     OldV = V;
278f22ef01cSRoman Divacky     return;
279f22ef01cSRoman Divacky   }
280f22ef01cSRoman Divacky 
281f22ef01cSRoman Divacky   // Handle constants and non-constants (e.g. instrs) differently for
282f22ef01cSRoman Divacky   // efficiency.
283f22ef01cSRoman Divacky   if (Constant *PHC = dyn_cast<Constant>(&*OldV)) {
284f22ef01cSRoman Divacky     ResolveConstants.push_back(std::make_pair(PHC, Idx));
285f22ef01cSRoman Divacky     OldV = V;
286f22ef01cSRoman Divacky   } else {
287f22ef01cSRoman Divacky     // If there was a forward reference to this value, replace it.
288f22ef01cSRoman Divacky     Value *PrevVal = OldV;
289f22ef01cSRoman Divacky     OldV->replaceAllUsesWith(V);
290f22ef01cSRoman Divacky     delete PrevVal;
291f22ef01cSRoman Divacky   }
292f22ef01cSRoman Divacky }
293f22ef01cSRoman Divacky 
294f22ef01cSRoman Divacky 
295f22ef01cSRoman Divacky Constant *BitcodeReaderValueList::getConstantFwdRef(unsigned Idx,
2966122f3e6SDimitry Andric                                                     Type *Ty) {
297f22ef01cSRoman Divacky   if (Idx >= size())
298f22ef01cSRoman Divacky     resize(Idx + 1);
299f22ef01cSRoman Divacky 
300f22ef01cSRoman Divacky   if (Value *V = ValuePtrs[Idx]) {
301f22ef01cSRoman Divacky     assert(Ty == V->getType() && "Type mismatch in constant table!");
302f22ef01cSRoman Divacky     return cast<Constant>(V);
303f22ef01cSRoman Divacky   }
304f22ef01cSRoman Divacky 
305f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
306f22ef01cSRoman Divacky   Constant *C = new ConstantPlaceHolder(Ty, Context);
307f22ef01cSRoman Divacky   ValuePtrs[Idx] = C;
308f22ef01cSRoman Divacky   return C;
309f22ef01cSRoman Divacky }
310f22ef01cSRoman Divacky 
3116122f3e6SDimitry Andric Value *BitcodeReaderValueList::getValueFwdRef(unsigned Idx, Type *Ty) {
312f22ef01cSRoman Divacky   if (Idx >= size())
313f22ef01cSRoman Divacky     resize(Idx + 1);
314f22ef01cSRoman Divacky 
315f22ef01cSRoman Divacky   if (Value *V = ValuePtrs[Idx]) {
31691bc56edSDimitry Andric     assert((!Ty || Ty == V->getType()) && "Type mismatch in value table!");
317f22ef01cSRoman Divacky     return V;
318f22ef01cSRoman Divacky   }
319f22ef01cSRoman Divacky 
320f22ef01cSRoman Divacky   // No type specified, must be invalid reference.
32191bc56edSDimitry Andric   if (!Ty) return nullptr;
322f22ef01cSRoman Divacky 
323f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
324f22ef01cSRoman Divacky   Value *V = new Argument(Ty);
325f22ef01cSRoman Divacky   ValuePtrs[Idx] = V;
326f22ef01cSRoman Divacky   return V;
327f22ef01cSRoman Divacky }
328f22ef01cSRoman Divacky 
329f22ef01cSRoman Divacky /// ResolveConstantForwardRefs - Once all constants are read, this method bulk
330f22ef01cSRoman Divacky /// resolves any forward references.  The idea behind this is that we sometimes
331f22ef01cSRoman Divacky /// get constants (such as large arrays) which reference *many* forward ref
332f22ef01cSRoman Divacky /// constants.  Replacing each of these causes a lot of thrashing when
333f22ef01cSRoman Divacky /// building/reuniquing the constant.  Instead of doing this, we look at all the
334f22ef01cSRoman Divacky /// uses and rewrite all the place holders at once for any constant that uses
335f22ef01cSRoman Divacky /// a placeholder.
336f22ef01cSRoman Divacky void BitcodeReaderValueList::ResolveConstantForwardRefs() {
337f22ef01cSRoman Divacky   // Sort the values by-pointer so that they are efficient to look up with a
338f22ef01cSRoman Divacky   // binary search.
339f22ef01cSRoman Divacky   std::sort(ResolveConstants.begin(), ResolveConstants.end());
340f22ef01cSRoman Divacky 
341f22ef01cSRoman Divacky   SmallVector<Constant*, 64> NewOps;
342f22ef01cSRoman Divacky 
343f22ef01cSRoman Divacky   while (!ResolveConstants.empty()) {
344f22ef01cSRoman Divacky     Value *RealVal = operator[](ResolveConstants.back().second);
345f22ef01cSRoman Divacky     Constant *Placeholder = ResolveConstants.back().first;
346f22ef01cSRoman Divacky     ResolveConstants.pop_back();
347f22ef01cSRoman Divacky 
348f22ef01cSRoman Divacky     // Loop over all users of the placeholder, updating them to reference the
349f22ef01cSRoman Divacky     // new value.  If they reference more than one placeholder, update them all
350f22ef01cSRoman Divacky     // at once.
351f22ef01cSRoman Divacky     while (!Placeholder->use_empty()) {
35291bc56edSDimitry Andric       auto UI = Placeholder->user_begin();
353ffd1746dSEd Schouten       User *U = *UI;
354f22ef01cSRoman Divacky 
355f22ef01cSRoman Divacky       // If the using object isn't uniqued, just update the operands.  This
356f22ef01cSRoman Divacky       // handles instructions and initializers for global variables.
357ffd1746dSEd Schouten       if (!isa<Constant>(U) || isa<GlobalValue>(U)) {
358f22ef01cSRoman Divacky         UI.getUse().set(RealVal);
359f22ef01cSRoman Divacky         continue;
360f22ef01cSRoman Divacky       }
361f22ef01cSRoman Divacky 
362f22ef01cSRoman Divacky       // Otherwise, we have a constant that uses the placeholder.  Replace that
363f22ef01cSRoman Divacky       // constant with a new constant that has *all* placeholder uses updated.
364ffd1746dSEd Schouten       Constant *UserC = cast<Constant>(U);
365f22ef01cSRoman Divacky       for (User::op_iterator I = UserC->op_begin(), E = UserC->op_end();
366f22ef01cSRoman Divacky            I != E; ++I) {
367f22ef01cSRoman Divacky         Value *NewOp;
368f22ef01cSRoman Divacky         if (!isa<ConstantPlaceHolder>(*I)) {
369f22ef01cSRoman Divacky           // Not a placeholder reference.
370f22ef01cSRoman Divacky           NewOp = *I;
371f22ef01cSRoman Divacky         } else if (*I == Placeholder) {
372f22ef01cSRoman Divacky           // Common case is that it just references this one placeholder.
373f22ef01cSRoman Divacky           NewOp = RealVal;
374f22ef01cSRoman Divacky         } else {
375f22ef01cSRoman Divacky           // Otherwise, look up the placeholder in ResolveConstants.
376f22ef01cSRoman Divacky           ResolveConstantsTy::iterator It =
377f22ef01cSRoman Divacky             std::lower_bound(ResolveConstants.begin(), ResolveConstants.end(),
378f22ef01cSRoman Divacky                              std::pair<Constant*, unsigned>(cast<Constant>(*I),
379f22ef01cSRoman Divacky                                                             0));
380f22ef01cSRoman Divacky           assert(It != ResolveConstants.end() && It->first == *I);
381f22ef01cSRoman Divacky           NewOp = operator[](It->second);
382f22ef01cSRoman Divacky         }
383f22ef01cSRoman Divacky 
384f22ef01cSRoman Divacky         NewOps.push_back(cast<Constant>(NewOp));
385f22ef01cSRoman Divacky       }
386f22ef01cSRoman Divacky 
387f22ef01cSRoman Divacky       // Make the new constant.
388f22ef01cSRoman Divacky       Constant *NewC;
389f22ef01cSRoman Divacky       if (ConstantArray *UserCA = dyn_cast<ConstantArray>(UserC)) {
39017a519f9SDimitry Andric         NewC = ConstantArray::get(UserCA->getType(), NewOps);
391f22ef01cSRoman Divacky       } else if (ConstantStruct *UserCS = dyn_cast<ConstantStruct>(UserC)) {
39217a519f9SDimitry Andric         NewC = ConstantStruct::get(UserCS->getType(), NewOps);
393f22ef01cSRoman Divacky       } else if (isa<ConstantVector>(UserC)) {
3942754fe60SDimitry Andric         NewC = ConstantVector::get(NewOps);
395f22ef01cSRoman Divacky       } else {
396f22ef01cSRoman Divacky         assert(isa<ConstantExpr>(UserC) && "Must be a ConstantExpr.");
3973b0f4066SDimitry Andric         NewC = cast<ConstantExpr>(UserC)->getWithOperands(NewOps);
398f22ef01cSRoman Divacky       }
399f22ef01cSRoman Divacky 
400f22ef01cSRoman Divacky       UserC->replaceAllUsesWith(NewC);
401f22ef01cSRoman Divacky       UserC->destroyConstant();
402f22ef01cSRoman Divacky       NewOps.clear();
403f22ef01cSRoman Divacky     }
404f22ef01cSRoman Divacky 
405f22ef01cSRoman Divacky     // Update all ValueHandles, they should be the only users at this point.
406f22ef01cSRoman Divacky     Placeholder->replaceAllUsesWith(RealVal);
407f22ef01cSRoman Divacky     delete Placeholder;
408f22ef01cSRoman Divacky   }
409f22ef01cSRoman Divacky }
410f22ef01cSRoman Divacky 
411f22ef01cSRoman Divacky void BitcodeReaderMDValueList::AssignValue(Value *V, unsigned Idx) {
412f22ef01cSRoman Divacky   if (Idx == size()) {
413f22ef01cSRoman Divacky     push_back(V);
414f22ef01cSRoman Divacky     return;
415f22ef01cSRoman Divacky   }
416f22ef01cSRoman Divacky 
417f22ef01cSRoman Divacky   if (Idx >= size())
418f22ef01cSRoman Divacky     resize(Idx+1);
419f22ef01cSRoman Divacky 
420f22ef01cSRoman Divacky   WeakVH &OldV = MDValuePtrs[Idx];
42191bc56edSDimitry Andric   if (!OldV) {
422f22ef01cSRoman Divacky     OldV = V;
423f22ef01cSRoman Divacky     return;
424f22ef01cSRoman Divacky   }
425f22ef01cSRoman Divacky 
426f22ef01cSRoman Divacky   // If there was a forward reference to this value, replace it.
427e580952dSDimitry Andric   MDNode *PrevVal = cast<MDNode>(OldV);
428f22ef01cSRoman Divacky   OldV->replaceAllUsesWith(V);
429e580952dSDimitry Andric   MDNode::deleteTemporary(PrevVal);
430f22ef01cSRoman Divacky   // Deleting PrevVal sets Idx value in MDValuePtrs to null. Set new
431f22ef01cSRoman Divacky   // value for Idx.
432f22ef01cSRoman Divacky   MDValuePtrs[Idx] = V;
433f22ef01cSRoman Divacky }
434f22ef01cSRoman Divacky 
435f22ef01cSRoman Divacky Value *BitcodeReaderMDValueList::getValueFwdRef(unsigned Idx) {
436f22ef01cSRoman Divacky   if (Idx >= size())
437f22ef01cSRoman Divacky     resize(Idx + 1);
438f22ef01cSRoman Divacky 
439f22ef01cSRoman Divacky   if (Value *V = MDValuePtrs[Idx]) {
440f22ef01cSRoman Divacky     assert(V->getType()->isMetadataTy() && "Type mismatch in value table!");
441f22ef01cSRoman Divacky     return V;
442f22ef01cSRoman Divacky   }
443f22ef01cSRoman Divacky 
444f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
445284c1978SDimitry Andric   Value *V = MDNode::getTemporary(Context, None);
446f22ef01cSRoman Divacky   MDValuePtrs[Idx] = V;
447f22ef01cSRoman Divacky   return V;
448f22ef01cSRoman Divacky }
449f22ef01cSRoman Divacky 
45017a519f9SDimitry Andric Type *BitcodeReader::getTypeByID(unsigned ID) {
45117a519f9SDimitry Andric   // The type table size is always specified correctly.
45217a519f9SDimitry Andric   if (ID >= TypeList.size())
45391bc56edSDimitry Andric     return nullptr;
454f22ef01cSRoman Divacky 
45517a519f9SDimitry Andric   if (Type *Ty = TypeList[ID])
45617a519f9SDimitry Andric     return Ty;
45717a519f9SDimitry Andric 
45817a519f9SDimitry Andric   // If we have a forward reference, the only possible case is when it is to a
45917a519f9SDimitry Andric   // named struct.  Just create a placeholder for now.
4606122f3e6SDimitry Andric   return TypeList[ID] = StructType::create(Context);
461f22ef01cSRoman Divacky }
462f22ef01cSRoman Divacky 
46317a519f9SDimitry Andric 
464f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
465f22ef01cSRoman Divacky //  Functions for parsing blocks from the bitcode file
466f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
467f22ef01cSRoman Divacky 
468139f7f9bSDimitry Andric 
469139f7f9bSDimitry Andric /// \brief This fills an AttrBuilder object with the LLVM attributes that have
470139f7f9bSDimitry Andric /// been decoded from the given integer. This function must stay in sync with
471139f7f9bSDimitry Andric /// 'encodeLLVMAttributesForBitcode'.
472139f7f9bSDimitry Andric static void decodeLLVMAttributesForBitcode(AttrBuilder &B,
473139f7f9bSDimitry Andric                                            uint64_t EncodedAttrs) {
474139f7f9bSDimitry Andric   // FIXME: Remove in 4.0.
475139f7f9bSDimitry Andric 
476139f7f9bSDimitry Andric   // The alignment is stored as a 16-bit raw value from bits 31--16.  We shift
477139f7f9bSDimitry Andric   // the bits above 31 down by 11 bits.
478139f7f9bSDimitry Andric   unsigned Alignment = (EncodedAttrs & (0xffffULL << 16)) >> 16;
479139f7f9bSDimitry Andric   assert((!Alignment || isPowerOf2_32(Alignment)) &&
480139f7f9bSDimitry Andric          "Alignment must be a power of two.");
481139f7f9bSDimitry Andric 
482139f7f9bSDimitry Andric   if (Alignment)
483139f7f9bSDimitry Andric     B.addAlignmentAttr(Alignment);
484139f7f9bSDimitry Andric   B.addRawValue(((EncodedAttrs & (0xfffffULL << 32)) >> 11) |
485139f7f9bSDimitry Andric                 (EncodedAttrs & 0xffff));
486139f7f9bSDimitry Andric }
487139f7f9bSDimitry Andric 
48891bc56edSDimitry Andric std::error_code BitcodeReader::ParseAttributeBlock() {
489f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::PARAMATTR_BLOCK_ID))
490f785676fSDimitry Andric     return Error(InvalidRecord);
491f22ef01cSRoman Divacky 
492f22ef01cSRoman Divacky   if (!MAttributes.empty())
493f785676fSDimitry Andric     return Error(InvalidMultipleBlocks);
494f22ef01cSRoman Divacky 
495f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
496f22ef01cSRoman Divacky 
497139f7f9bSDimitry Andric   SmallVector<AttributeSet, 8> Attrs;
498f22ef01cSRoman Divacky 
499f22ef01cSRoman Divacky   // Read all the records.
500f22ef01cSRoman Divacky   while (1) {
501139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
502139f7f9bSDimitry Andric 
503139f7f9bSDimitry Andric     switch (Entry.Kind) {
504139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
505139f7f9bSDimitry Andric     case BitstreamEntry::Error:
506f785676fSDimitry Andric       return Error(MalformedBlock);
507139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
50891bc56edSDimitry Andric       return std::error_code();
509139f7f9bSDimitry Andric     case BitstreamEntry::Record:
510139f7f9bSDimitry Andric       // The interesting case.
511139f7f9bSDimitry Andric       break;
512f22ef01cSRoman Divacky     }
513f22ef01cSRoman Divacky 
514f22ef01cSRoman Divacky     // Read a record.
515f22ef01cSRoman Divacky     Record.clear();
516139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
517f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
518f22ef01cSRoman Divacky       break;
519139f7f9bSDimitry Andric     case bitc::PARAMATTR_CODE_ENTRY_OLD: { // ENTRY: [paramidx0, attr0, ...]
520139f7f9bSDimitry Andric       // FIXME: Remove in 4.0.
521f22ef01cSRoman Divacky       if (Record.size() & 1)
522f785676fSDimitry Andric         return Error(InvalidRecord);
523f22ef01cSRoman Divacky 
524f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
525139f7f9bSDimitry Andric         AttrBuilder B;
526139f7f9bSDimitry Andric         decodeLLVMAttributesForBitcode(B, Record[i+1]);
527139f7f9bSDimitry Andric         Attrs.push_back(AttributeSet::get(Context, Record[i], B));
528f22ef01cSRoman Divacky       }
529f22ef01cSRoman Divacky 
530139f7f9bSDimitry Andric       MAttributes.push_back(AttributeSet::get(Context, Attrs));
531f22ef01cSRoman Divacky       Attrs.clear();
532f22ef01cSRoman Divacky       break;
533f22ef01cSRoman Divacky     }
534139f7f9bSDimitry Andric     case bitc::PARAMATTR_CODE_ENTRY: { // ENTRY: [attrgrp0, attrgrp1, ...]
535139f7f9bSDimitry Andric       for (unsigned i = 0, e = Record.size(); i != e; ++i)
536139f7f9bSDimitry Andric         Attrs.push_back(MAttributeGroups[Record[i]]);
537139f7f9bSDimitry Andric 
538139f7f9bSDimitry Andric       MAttributes.push_back(AttributeSet::get(Context, Attrs));
539139f7f9bSDimitry Andric       Attrs.clear();
540139f7f9bSDimitry Andric       break;
541139f7f9bSDimitry Andric     }
542139f7f9bSDimitry Andric     }
543139f7f9bSDimitry Andric   }
544139f7f9bSDimitry Andric }
545139f7f9bSDimitry Andric 
546f785676fSDimitry Andric // Returns Attribute::None on unrecognized codes.
547f785676fSDimitry Andric static Attribute::AttrKind GetAttrFromCode(uint64_t Code) {
548f785676fSDimitry Andric   switch (Code) {
549f785676fSDimitry Andric   default:
550f785676fSDimitry Andric     return Attribute::None;
551f785676fSDimitry Andric   case bitc::ATTR_KIND_ALIGNMENT:
552f785676fSDimitry Andric     return Attribute::Alignment;
553f785676fSDimitry Andric   case bitc::ATTR_KIND_ALWAYS_INLINE:
554f785676fSDimitry Andric     return Attribute::AlwaysInline;
555f785676fSDimitry Andric   case bitc::ATTR_KIND_BUILTIN:
556f785676fSDimitry Andric     return Attribute::Builtin;
557f785676fSDimitry Andric   case bitc::ATTR_KIND_BY_VAL:
558f785676fSDimitry Andric     return Attribute::ByVal;
55991bc56edSDimitry Andric   case bitc::ATTR_KIND_IN_ALLOCA:
56091bc56edSDimitry Andric     return Attribute::InAlloca;
561f785676fSDimitry Andric   case bitc::ATTR_KIND_COLD:
562f785676fSDimitry Andric     return Attribute::Cold;
563f785676fSDimitry Andric   case bitc::ATTR_KIND_INLINE_HINT:
564f785676fSDimitry Andric     return Attribute::InlineHint;
565f785676fSDimitry Andric   case bitc::ATTR_KIND_IN_REG:
566f785676fSDimitry Andric     return Attribute::InReg;
56791bc56edSDimitry Andric   case bitc::ATTR_KIND_JUMP_TABLE:
56891bc56edSDimitry Andric     return Attribute::JumpTable;
569f785676fSDimitry Andric   case bitc::ATTR_KIND_MIN_SIZE:
570f785676fSDimitry Andric     return Attribute::MinSize;
571f785676fSDimitry Andric   case bitc::ATTR_KIND_NAKED:
572f785676fSDimitry Andric     return Attribute::Naked;
573f785676fSDimitry Andric   case bitc::ATTR_KIND_NEST:
574f785676fSDimitry Andric     return Attribute::Nest;
575f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_ALIAS:
576f785676fSDimitry Andric     return Attribute::NoAlias;
577f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_BUILTIN:
578f785676fSDimitry Andric     return Attribute::NoBuiltin;
579f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_CAPTURE:
580f785676fSDimitry Andric     return Attribute::NoCapture;
581f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_DUPLICATE:
582f785676fSDimitry Andric     return Attribute::NoDuplicate;
583f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_IMPLICIT_FLOAT:
584f785676fSDimitry Andric     return Attribute::NoImplicitFloat;
585f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_INLINE:
586f785676fSDimitry Andric     return Attribute::NoInline;
587f785676fSDimitry Andric   case bitc::ATTR_KIND_NON_LAZY_BIND:
588f785676fSDimitry Andric     return Attribute::NonLazyBind;
58991bc56edSDimitry Andric   case bitc::ATTR_KIND_NON_NULL:
59091bc56edSDimitry Andric     return Attribute::NonNull;
59191bc56edSDimitry Andric   case bitc::ATTR_KIND_DEREFERENCEABLE:
59291bc56edSDimitry Andric     return Attribute::Dereferenceable;
593f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_RED_ZONE:
594f785676fSDimitry Andric     return Attribute::NoRedZone;
595f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_RETURN:
596f785676fSDimitry Andric     return Attribute::NoReturn;
597f785676fSDimitry Andric   case bitc::ATTR_KIND_NO_UNWIND:
598f785676fSDimitry Andric     return Attribute::NoUnwind;
599f785676fSDimitry Andric   case bitc::ATTR_KIND_OPTIMIZE_FOR_SIZE:
600f785676fSDimitry Andric     return Attribute::OptimizeForSize;
601f785676fSDimitry Andric   case bitc::ATTR_KIND_OPTIMIZE_NONE:
602f785676fSDimitry Andric     return Attribute::OptimizeNone;
603f785676fSDimitry Andric   case bitc::ATTR_KIND_READ_NONE:
604f785676fSDimitry Andric     return Attribute::ReadNone;
605f785676fSDimitry Andric   case bitc::ATTR_KIND_READ_ONLY:
606f785676fSDimitry Andric     return Attribute::ReadOnly;
607f785676fSDimitry Andric   case bitc::ATTR_KIND_RETURNED:
608f785676fSDimitry Andric     return Attribute::Returned;
609f785676fSDimitry Andric   case bitc::ATTR_KIND_RETURNS_TWICE:
610f785676fSDimitry Andric     return Attribute::ReturnsTwice;
611f785676fSDimitry Andric   case bitc::ATTR_KIND_S_EXT:
612f785676fSDimitry Andric     return Attribute::SExt;
613f785676fSDimitry Andric   case bitc::ATTR_KIND_STACK_ALIGNMENT:
614f785676fSDimitry Andric     return Attribute::StackAlignment;
615f785676fSDimitry Andric   case bitc::ATTR_KIND_STACK_PROTECT:
616f785676fSDimitry Andric     return Attribute::StackProtect;
617f785676fSDimitry Andric   case bitc::ATTR_KIND_STACK_PROTECT_REQ:
618f785676fSDimitry Andric     return Attribute::StackProtectReq;
619f785676fSDimitry Andric   case bitc::ATTR_KIND_STACK_PROTECT_STRONG:
620f785676fSDimitry Andric     return Attribute::StackProtectStrong;
621f785676fSDimitry Andric   case bitc::ATTR_KIND_STRUCT_RET:
622f785676fSDimitry Andric     return Attribute::StructRet;
623f785676fSDimitry Andric   case bitc::ATTR_KIND_SANITIZE_ADDRESS:
624f785676fSDimitry Andric     return Attribute::SanitizeAddress;
625f785676fSDimitry Andric   case bitc::ATTR_KIND_SANITIZE_THREAD:
626f785676fSDimitry Andric     return Attribute::SanitizeThread;
627f785676fSDimitry Andric   case bitc::ATTR_KIND_SANITIZE_MEMORY:
628f785676fSDimitry Andric     return Attribute::SanitizeMemory;
629f785676fSDimitry Andric   case bitc::ATTR_KIND_UW_TABLE:
630f785676fSDimitry Andric     return Attribute::UWTable;
631f785676fSDimitry Andric   case bitc::ATTR_KIND_Z_EXT:
632f785676fSDimitry Andric     return Attribute::ZExt;
633f785676fSDimitry Andric   }
634f785676fSDimitry Andric }
635f785676fSDimitry Andric 
63691bc56edSDimitry Andric std::error_code BitcodeReader::ParseAttrKind(uint64_t Code,
637f785676fSDimitry Andric                                              Attribute::AttrKind *Kind) {
638f785676fSDimitry Andric   *Kind = GetAttrFromCode(Code);
639f785676fSDimitry Andric   if (*Kind == Attribute::None)
640f785676fSDimitry Andric     return Error(InvalidValue);
64191bc56edSDimitry Andric   return std::error_code();
642f785676fSDimitry Andric }
643f785676fSDimitry Andric 
64491bc56edSDimitry Andric std::error_code BitcodeReader::ParseAttributeGroupBlock() {
645139f7f9bSDimitry Andric   if (Stream.EnterSubBlock(bitc::PARAMATTR_GROUP_BLOCK_ID))
646f785676fSDimitry Andric     return Error(InvalidRecord);
647139f7f9bSDimitry Andric 
648139f7f9bSDimitry Andric   if (!MAttributeGroups.empty())
649f785676fSDimitry Andric     return Error(InvalidMultipleBlocks);
650139f7f9bSDimitry Andric 
651139f7f9bSDimitry Andric   SmallVector<uint64_t, 64> Record;
652139f7f9bSDimitry Andric 
653139f7f9bSDimitry Andric   // Read all the records.
654139f7f9bSDimitry Andric   while (1) {
655139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
656139f7f9bSDimitry Andric 
657139f7f9bSDimitry Andric     switch (Entry.Kind) {
658139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
659139f7f9bSDimitry Andric     case BitstreamEntry::Error:
660f785676fSDimitry Andric       return Error(MalformedBlock);
661139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
66291bc56edSDimitry Andric       return std::error_code();
663139f7f9bSDimitry Andric     case BitstreamEntry::Record:
664139f7f9bSDimitry Andric       // The interesting case.
665139f7f9bSDimitry Andric       break;
666139f7f9bSDimitry Andric     }
667139f7f9bSDimitry Andric 
668139f7f9bSDimitry Andric     // Read a record.
669139f7f9bSDimitry Andric     Record.clear();
670139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
671139f7f9bSDimitry Andric     default:  // Default behavior: ignore.
672139f7f9bSDimitry Andric       break;
673139f7f9bSDimitry Andric     case bitc::PARAMATTR_GRP_CODE_ENTRY: { // ENTRY: [grpid, idx, a0, a1, ...]
674139f7f9bSDimitry Andric       if (Record.size() < 3)
675f785676fSDimitry Andric         return Error(InvalidRecord);
676139f7f9bSDimitry Andric 
677139f7f9bSDimitry Andric       uint64_t GrpID = Record[0];
678139f7f9bSDimitry Andric       uint64_t Idx = Record[1]; // Index of the object this attribute refers to.
679139f7f9bSDimitry Andric 
680139f7f9bSDimitry Andric       AttrBuilder B;
681139f7f9bSDimitry Andric       for (unsigned i = 2, e = Record.size(); i != e; ++i) {
682139f7f9bSDimitry Andric         if (Record[i] == 0) {        // Enum attribute
683f785676fSDimitry Andric           Attribute::AttrKind Kind;
68491bc56edSDimitry Andric           if (std::error_code EC = ParseAttrKind(Record[++i], &Kind))
685f785676fSDimitry Andric             return EC;
686f785676fSDimitry Andric 
687f785676fSDimitry Andric           B.addAttribute(Kind);
68891bc56edSDimitry Andric         } else if (Record[i] == 1) { // Integer attribute
689f785676fSDimitry Andric           Attribute::AttrKind Kind;
69091bc56edSDimitry Andric           if (std::error_code EC = ParseAttrKind(Record[++i], &Kind))
691f785676fSDimitry Andric             return EC;
692f785676fSDimitry Andric           if (Kind == Attribute::Alignment)
693139f7f9bSDimitry Andric             B.addAlignmentAttr(Record[++i]);
69491bc56edSDimitry Andric           else if (Kind == Attribute::StackAlignment)
695139f7f9bSDimitry Andric             B.addStackAlignmentAttr(Record[++i]);
69691bc56edSDimitry Andric           else if (Kind == Attribute::Dereferenceable)
69791bc56edSDimitry Andric             B.addDereferenceableAttr(Record[++i]);
698139f7f9bSDimitry Andric         } else {                     // String attribute
699139f7f9bSDimitry Andric           assert((Record[i] == 3 || Record[i] == 4) &&
700139f7f9bSDimitry Andric                  "Invalid attribute group entry");
701139f7f9bSDimitry Andric           bool HasValue = (Record[i++] == 4);
702139f7f9bSDimitry Andric           SmallString<64> KindStr;
703139f7f9bSDimitry Andric           SmallString<64> ValStr;
704139f7f9bSDimitry Andric 
705139f7f9bSDimitry Andric           while (Record[i] != 0 && i != e)
706139f7f9bSDimitry Andric             KindStr += Record[i++];
707139f7f9bSDimitry Andric           assert(Record[i] == 0 && "Kind string not null terminated");
708139f7f9bSDimitry Andric 
709139f7f9bSDimitry Andric           if (HasValue) {
710139f7f9bSDimitry Andric             // Has a value associated with it.
711139f7f9bSDimitry Andric             ++i; // Skip the '0' that terminates the "kind" string.
712139f7f9bSDimitry Andric             while (Record[i] != 0 && i != e)
713139f7f9bSDimitry Andric               ValStr += Record[i++];
714139f7f9bSDimitry Andric             assert(Record[i] == 0 && "Value string not null terminated");
715139f7f9bSDimitry Andric           }
716139f7f9bSDimitry Andric 
717139f7f9bSDimitry Andric           B.addAttribute(KindStr.str(), ValStr.str());
718139f7f9bSDimitry Andric         }
719139f7f9bSDimitry Andric       }
720139f7f9bSDimitry Andric 
721139f7f9bSDimitry Andric       MAttributeGroups[GrpID] = AttributeSet::get(Context, Idx, B);
722139f7f9bSDimitry Andric       break;
723139f7f9bSDimitry Andric     }
724f22ef01cSRoman Divacky     }
725f22ef01cSRoman Divacky   }
726f22ef01cSRoman Divacky }
727f22ef01cSRoman Divacky 
72891bc56edSDimitry Andric std::error_code BitcodeReader::ParseTypeTable() {
72917a519f9SDimitry Andric   if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID_NEW))
730f785676fSDimitry Andric     return Error(InvalidRecord);
731f22ef01cSRoman Divacky 
73217a519f9SDimitry Andric   return ParseTypeTableBody();
73317a519f9SDimitry Andric }
73417a519f9SDimitry Andric 
73591bc56edSDimitry Andric std::error_code BitcodeReader::ParseTypeTableBody() {
736f22ef01cSRoman Divacky   if (!TypeList.empty())
737f785676fSDimitry Andric     return Error(InvalidMultipleBlocks);
738f22ef01cSRoman Divacky 
739f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
740f22ef01cSRoman Divacky   unsigned NumRecords = 0;
741f22ef01cSRoman Divacky 
74217a519f9SDimitry Andric   SmallString<64> TypeName;
74317a519f9SDimitry Andric 
744f22ef01cSRoman Divacky   // Read all the records for this type table.
745f22ef01cSRoman Divacky   while (1) {
746139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
747139f7f9bSDimitry Andric 
748139f7f9bSDimitry Andric     switch (Entry.Kind) {
749139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
750139f7f9bSDimitry Andric     case BitstreamEntry::Error:
751f785676fSDimitry Andric       return Error(MalformedBlock);
752139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
753f22ef01cSRoman Divacky       if (NumRecords != TypeList.size())
754f785676fSDimitry Andric         return Error(MalformedBlock);
75591bc56edSDimitry Andric       return std::error_code();
756139f7f9bSDimitry Andric     case BitstreamEntry::Record:
757139f7f9bSDimitry Andric       // The interesting case.
758139f7f9bSDimitry Andric       break;
759f22ef01cSRoman Divacky     }
760f22ef01cSRoman Divacky 
761f22ef01cSRoman Divacky     // Read a record.
762f22ef01cSRoman Divacky     Record.clear();
76391bc56edSDimitry Andric     Type *ResultTy = nullptr;
764139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
765f785676fSDimitry Andric     default:
766f785676fSDimitry Andric       return Error(InvalidValue);
767f22ef01cSRoman Divacky     case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
768f22ef01cSRoman Divacky       // TYPE_CODE_NUMENTRY contains a count of the number of types in the
769f22ef01cSRoman Divacky       // type list.  This allows us to reserve space.
770f22ef01cSRoman Divacky       if (Record.size() < 1)
771f785676fSDimitry Andric         return Error(InvalidRecord);
77217a519f9SDimitry Andric       TypeList.resize(Record[0]);
773f22ef01cSRoman Divacky       continue;
774f22ef01cSRoman Divacky     case bitc::TYPE_CODE_VOID:      // VOID
775f22ef01cSRoman Divacky       ResultTy = Type::getVoidTy(Context);
776f22ef01cSRoman Divacky       break;
777dff0c46cSDimitry Andric     case bitc::TYPE_CODE_HALF:     // HALF
778dff0c46cSDimitry Andric       ResultTy = Type::getHalfTy(Context);
779dff0c46cSDimitry Andric       break;
780f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FLOAT:     // FLOAT
781f22ef01cSRoman Divacky       ResultTy = Type::getFloatTy(Context);
782f22ef01cSRoman Divacky       break;
783f22ef01cSRoman Divacky     case bitc::TYPE_CODE_DOUBLE:    // DOUBLE
784f22ef01cSRoman Divacky       ResultTy = Type::getDoubleTy(Context);
785f22ef01cSRoman Divacky       break;
786f22ef01cSRoman Divacky     case bitc::TYPE_CODE_X86_FP80:  // X86_FP80
787f22ef01cSRoman Divacky       ResultTy = Type::getX86_FP80Ty(Context);
788f22ef01cSRoman Divacky       break;
789f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FP128:     // FP128
790f22ef01cSRoman Divacky       ResultTy = Type::getFP128Ty(Context);
791f22ef01cSRoman Divacky       break;
792f22ef01cSRoman Divacky     case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
793f22ef01cSRoman Divacky       ResultTy = Type::getPPC_FP128Ty(Context);
794f22ef01cSRoman Divacky       break;
795f22ef01cSRoman Divacky     case bitc::TYPE_CODE_LABEL:     // LABEL
796f22ef01cSRoman Divacky       ResultTy = Type::getLabelTy(Context);
797f22ef01cSRoman Divacky       break;
798f22ef01cSRoman Divacky     case bitc::TYPE_CODE_METADATA:  // METADATA
799f22ef01cSRoman Divacky       ResultTy = Type::getMetadataTy(Context);
800f22ef01cSRoman Divacky       break;
8012754fe60SDimitry Andric     case bitc::TYPE_CODE_X86_MMX:   // X86_MMX
8022754fe60SDimitry Andric       ResultTy = Type::getX86_MMXTy(Context);
8032754fe60SDimitry Andric       break;
804f22ef01cSRoman Divacky     case bitc::TYPE_CODE_INTEGER:   // INTEGER: [width]
805f22ef01cSRoman Divacky       if (Record.size() < 1)
806f785676fSDimitry Andric         return Error(InvalidRecord);
807f22ef01cSRoman Divacky 
808f22ef01cSRoman Divacky       ResultTy = IntegerType::get(Context, Record[0]);
809f22ef01cSRoman Divacky       break;
810f22ef01cSRoman Divacky     case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
811f22ef01cSRoman Divacky                                     //          [pointee type, address space]
812f22ef01cSRoman Divacky       if (Record.size() < 1)
813f785676fSDimitry Andric         return Error(InvalidRecord);
814f22ef01cSRoman Divacky       unsigned AddressSpace = 0;
815f22ef01cSRoman Divacky       if (Record.size() == 2)
816f22ef01cSRoman Divacky         AddressSpace = Record[1];
81717a519f9SDimitry Andric       ResultTy = getTypeByID(Record[0]);
81891bc56edSDimitry Andric       if (!ResultTy)
819f785676fSDimitry Andric         return Error(InvalidType);
82017a519f9SDimitry Andric       ResultTy = PointerType::get(ResultTy, AddressSpace);
821f22ef01cSRoman Divacky       break;
822f22ef01cSRoman Divacky     }
823dff0c46cSDimitry Andric     case bitc::TYPE_CODE_FUNCTION_OLD: {
8247ae0e2c9SDimitry Andric       // FIXME: attrid is dead, remove it in LLVM 4.0
825f22ef01cSRoman Divacky       // FUNCTION: [vararg, attrid, retty, paramty x N]
826f22ef01cSRoman Divacky       if (Record.size() < 3)
827f785676fSDimitry Andric         return Error(InvalidRecord);
828dff0c46cSDimitry Andric       SmallVector<Type*, 8> ArgTys;
82917a519f9SDimitry Andric       for (unsigned i = 3, e = Record.size(); i != e; ++i) {
83017a519f9SDimitry Andric         if (Type *T = getTypeByID(Record[i]))
83117a519f9SDimitry Andric           ArgTys.push_back(T);
83217a519f9SDimitry Andric         else
833f22ef01cSRoman Divacky           break;
834f22ef01cSRoman Divacky       }
83517a519f9SDimitry Andric 
83617a519f9SDimitry Andric       ResultTy = getTypeByID(Record[2]);
83791bc56edSDimitry Andric       if (!ResultTy || ArgTys.size() < Record.size()-3)
838f785676fSDimitry Andric         return Error(InvalidType);
83917a519f9SDimitry Andric 
84017a519f9SDimitry Andric       ResultTy = FunctionType::get(ResultTy, ArgTys, Record[0]);
84117a519f9SDimitry Andric       break;
84217a519f9SDimitry Andric     }
843dff0c46cSDimitry Andric     case bitc::TYPE_CODE_FUNCTION: {
844dff0c46cSDimitry Andric       // FUNCTION: [vararg, retty, paramty x N]
845dff0c46cSDimitry Andric       if (Record.size() < 2)
846f785676fSDimitry Andric         return Error(InvalidRecord);
847dff0c46cSDimitry Andric       SmallVector<Type*, 8> ArgTys;
848dff0c46cSDimitry Andric       for (unsigned i = 2, e = Record.size(); i != e; ++i) {
849dff0c46cSDimitry Andric         if (Type *T = getTypeByID(Record[i]))
850dff0c46cSDimitry Andric           ArgTys.push_back(T);
851dff0c46cSDimitry Andric         else
852dff0c46cSDimitry Andric           break;
853dff0c46cSDimitry Andric       }
854dff0c46cSDimitry Andric 
855dff0c46cSDimitry Andric       ResultTy = getTypeByID(Record[1]);
85691bc56edSDimitry Andric       if (!ResultTy || ArgTys.size() < Record.size()-2)
857f785676fSDimitry Andric         return Error(InvalidType);
858dff0c46cSDimitry Andric 
859dff0c46cSDimitry Andric       ResultTy = FunctionType::get(ResultTy, ArgTys, Record[0]);
860dff0c46cSDimitry Andric       break;
861dff0c46cSDimitry Andric     }
86217a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_ANON: {  // STRUCT: [ispacked, eltty x N]
863f22ef01cSRoman Divacky       if (Record.size() < 1)
864f785676fSDimitry Andric         return Error(InvalidRecord);
865dff0c46cSDimitry Andric       SmallVector<Type*, 8> EltTys;
86617a519f9SDimitry Andric       for (unsigned i = 1, e = Record.size(); i != e; ++i) {
86717a519f9SDimitry Andric         if (Type *T = getTypeByID(Record[i]))
86817a519f9SDimitry Andric           EltTys.push_back(T);
86917a519f9SDimitry Andric         else
87017a519f9SDimitry Andric           break;
87117a519f9SDimitry Andric       }
87217a519f9SDimitry Andric       if (EltTys.size() != Record.size()-1)
873f785676fSDimitry Andric         return Error(InvalidType);
874f22ef01cSRoman Divacky       ResultTy = StructType::get(Context, EltTys, Record[0]);
875f22ef01cSRoman Divacky       break;
876f22ef01cSRoman Divacky     }
87717a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_NAME:   // STRUCT_NAME: [strchr x N]
87817a519f9SDimitry Andric       if (ConvertToString(Record, 0, TypeName))
879f785676fSDimitry Andric         return Error(InvalidRecord);
88017a519f9SDimitry Andric       continue;
88117a519f9SDimitry Andric 
88217a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_NAMED: { // STRUCT: [ispacked, eltty x N]
88317a519f9SDimitry Andric       if (Record.size() < 1)
884f785676fSDimitry Andric         return Error(InvalidRecord);
88517a519f9SDimitry Andric 
88617a519f9SDimitry Andric       if (NumRecords >= TypeList.size())
887f785676fSDimitry Andric         return Error(InvalidTYPETable);
88817a519f9SDimitry Andric 
88917a519f9SDimitry Andric       // Check to see if this was forward referenced, if so fill in the temp.
89017a519f9SDimitry Andric       StructType *Res = cast_or_null<StructType>(TypeList[NumRecords]);
89117a519f9SDimitry Andric       if (Res) {
89217a519f9SDimitry Andric         Res->setName(TypeName);
89391bc56edSDimitry Andric         TypeList[NumRecords] = nullptr;
89417a519f9SDimitry Andric       } else  // Otherwise, create a new struct.
8956122f3e6SDimitry Andric         Res = StructType::create(Context, TypeName);
89617a519f9SDimitry Andric       TypeName.clear();
89717a519f9SDimitry Andric 
89817a519f9SDimitry Andric       SmallVector<Type*, 8> EltTys;
89917a519f9SDimitry Andric       for (unsigned i = 1, e = Record.size(); i != e; ++i) {
90017a519f9SDimitry Andric         if (Type *T = getTypeByID(Record[i]))
90117a519f9SDimitry Andric           EltTys.push_back(T);
90217a519f9SDimitry Andric         else
90317a519f9SDimitry Andric           break;
90417a519f9SDimitry Andric       }
90517a519f9SDimitry Andric       if (EltTys.size() != Record.size()-1)
906f785676fSDimitry Andric         return Error(InvalidRecord);
90717a519f9SDimitry Andric       Res->setBody(EltTys, Record[0]);
90817a519f9SDimitry Andric       ResultTy = Res;
90917a519f9SDimitry Andric       break;
91017a519f9SDimitry Andric     }
91117a519f9SDimitry Andric     case bitc::TYPE_CODE_OPAQUE: {       // OPAQUE: []
91217a519f9SDimitry Andric       if (Record.size() != 1)
913f785676fSDimitry Andric         return Error(InvalidRecord);
91417a519f9SDimitry Andric 
91517a519f9SDimitry Andric       if (NumRecords >= TypeList.size())
916f785676fSDimitry Andric         return Error(InvalidTYPETable);
91717a519f9SDimitry Andric 
91817a519f9SDimitry Andric       // Check to see if this was forward referenced, if so fill in the temp.
91917a519f9SDimitry Andric       StructType *Res = cast_or_null<StructType>(TypeList[NumRecords]);
92017a519f9SDimitry Andric       if (Res) {
92117a519f9SDimitry Andric         Res->setName(TypeName);
92291bc56edSDimitry Andric         TypeList[NumRecords] = nullptr;
92317a519f9SDimitry Andric       } else  // Otherwise, create a new struct with no body.
9246122f3e6SDimitry Andric         Res = StructType::create(Context, TypeName);
92517a519f9SDimitry Andric       TypeName.clear();
92617a519f9SDimitry Andric       ResultTy = Res;
92717a519f9SDimitry Andric       break;
92817a519f9SDimitry Andric     }
929f22ef01cSRoman Divacky     case bitc::TYPE_CODE_ARRAY:     // ARRAY: [numelts, eltty]
930f22ef01cSRoman Divacky       if (Record.size() < 2)
931f785676fSDimitry Andric         return Error(InvalidRecord);
93217a519f9SDimitry Andric       if ((ResultTy = getTypeByID(Record[1])))
93317a519f9SDimitry Andric         ResultTy = ArrayType::get(ResultTy, Record[0]);
93417a519f9SDimitry Andric       else
935f785676fSDimitry Andric         return Error(InvalidType);
936f22ef01cSRoman Divacky       break;
937f22ef01cSRoman Divacky     case bitc::TYPE_CODE_VECTOR:    // VECTOR: [numelts, eltty]
938f22ef01cSRoman Divacky       if (Record.size() < 2)
939f785676fSDimitry Andric         return Error(InvalidRecord);
94017a519f9SDimitry Andric       if ((ResultTy = getTypeByID(Record[1])))
94117a519f9SDimitry Andric         ResultTy = VectorType::get(ResultTy, Record[0]);
94217a519f9SDimitry Andric       else
943f785676fSDimitry Andric         return Error(InvalidType);
944f22ef01cSRoman Divacky       break;
945f22ef01cSRoman Divacky     }
946f22ef01cSRoman Divacky 
94717a519f9SDimitry Andric     if (NumRecords >= TypeList.size())
948f785676fSDimitry Andric       return Error(InvalidTYPETable);
94917a519f9SDimitry Andric     assert(ResultTy && "Didn't read a type?");
95091bc56edSDimitry Andric     assert(!TypeList[NumRecords] && "Already read type?");
95117a519f9SDimitry Andric     TypeList[NumRecords++] = ResultTy;
952f22ef01cSRoman Divacky   }
953f22ef01cSRoman Divacky }
95417a519f9SDimitry Andric 
95591bc56edSDimitry Andric std::error_code BitcodeReader::ParseValueSymbolTable() {
956f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::VALUE_SYMTAB_BLOCK_ID))
957f785676fSDimitry Andric     return Error(InvalidRecord);
958f22ef01cSRoman Divacky 
959f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
960f22ef01cSRoman Divacky 
961f22ef01cSRoman Divacky   // Read all the records for this value table.
962f22ef01cSRoman Divacky   SmallString<128> ValueName;
963f22ef01cSRoman Divacky   while (1) {
964139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
965f22ef01cSRoman Divacky 
966139f7f9bSDimitry Andric     switch (Entry.Kind) {
967139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
968139f7f9bSDimitry Andric     case BitstreamEntry::Error:
969f785676fSDimitry Andric       return Error(MalformedBlock);
970139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
97191bc56edSDimitry Andric       return std::error_code();
972139f7f9bSDimitry Andric     case BitstreamEntry::Record:
973139f7f9bSDimitry Andric       // The interesting case.
974139f7f9bSDimitry Andric       break;
975f22ef01cSRoman Divacky     }
976f22ef01cSRoman Divacky 
977f22ef01cSRoman Divacky     // Read a record.
978f22ef01cSRoman Divacky     Record.clear();
979139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
980f22ef01cSRoman Divacky     default:  // Default behavior: unknown type.
981f22ef01cSRoman Divacky       break;
982f22ef01cSRoman Divacky     case bitc::VST_CODE_ENTRY: {  // VST_ENTRY: [valueid, namechar x N]
983f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, ValueName))
984f785676fSDimitry Andric         return Error(InvalidRecord);
985f22ef01cSRoman Divacky       unsigned ValueID = Record[0];
98691bc56edSDimitry Andric       if (ValueID >= ValueList.size() || !ValueList[ValueID])
987f785676fSDimitry Andric         return Error(InvalidRecord);
988f22ef01cSRoman Divacky       Value *V = ValueList[ValueID];
989f22ef01cSRoman Divacky 
990f22ef01cSRoman Divacky       V->setName(StringRef(ValueName.data(), ValueName.size()));
991f22ef01cSRoman Divacky       ValueName.clear();
992f22ef01cSRoman Divacky       break;
993f22ef01cSRoman Divacky     }
994f22ef01cSRoman Divacky     case bitc::VST_CODE_BBENTRY: {
995f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, ValueName))
996f785676fSDimitry Andric         return Error(InvalidRecord);
997f22ef01cSRoman Divacky       BasicBlock *BB = getBasicBlock(Record[0]);
99891bc56edSDimitry Andric       if (!BB)
999f785676fSDimitry Andric         return Error(InvalidRecord);
1000f22ef01cSRoman Divacky 
1001f22ef01cSRoman Divacky       BB->setName(StringRef(ValueName.data(), ValueName.size()));
1002f22ef01cSRoman Divacky       ValueName.clear();
1003f22ef01cSRoman Divacky       break;
1004f22ef01cSRoman Divacky     }
1005f22ef01cSRoman Divacky     }
1006f22ef01cSRoman Divacky   }
1007f22ef01cSRoman Divacky }
1008f22ef01cSRoman Divacky 
100991bc56edSDimitry Andric std::error_code BitcodeReader::ParseMetadata() {
1010f22ef01cSRoman Divacky   unsigned NextMDValueNo = MDValueList.size();
1011f22ef01cSRoman Divacky 
1012f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
1013f785676fSDimitry Andric     return Error(InvalidRecord);
1014f22ef01cSRoman Divacky 
1015f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1016f22ef01cSRoman Divacky 
1017f22ef01cSRoman Divacky   // Read all the records.
1018f22ef01cSRoman Divacky   while (1) {
1019139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
1020139f7f9bSDimitry Andric 
1021139f7f9bSDimitry Andric     switch (Entry.Kind) {
1022139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
1023139f7f9bSDimitry Andric     case BitstreamEntry::Error:
1024f785676fSDimitry Andric       return Error(MalformedBlock);
1025139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
102691bc56edSDimitry Andric       return std::error_code();
1027139f7f9bSDimitry Andric     case BitstreamEntry::Record:
1028139f7f9bSDimitry Andric       // The interesting case.
1029139f7f9bSDimitry Andric       break;
1030f22ef01cSRoman Divacky     }
1031f22ef01cSRoman Divacky 
1032f22ef01cSRoman Divacky     bool IsFunctionLocal = false;
1033f22ef01cSRoman Divacky     // Read a record.
1034f22ef01cSRoman Divacky     Record.clear();
1035139f7f9bSDimitry Andric     unsigned Code = Stream.readRecord(Entry.ID, Record);
1036e580952dSDimitry Andric     switch (Code) {
1037f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
1038f22ef01cSRoman Divacky       break;
1039f22ef01cSRoman Divacky     case bitc::METADATA_NAME: {
1040139f7f9bSDimitry Andric       // Read name of the named metadata.
10417ae0e2c9SDimitry Andric       SmallString<8> Name(Record.begin(), Record.end());
1042f22ef01cSRoman Divacky       Record.clear();
1043f22ef01cSRoman Divacky       Code = Stream.ReadCode();
1044f22ef01cSRoman Divacky 
104517a519f9SDimitry Andric       // METADATA_NAME is always followed by METADATA_NAMED_NODE.
1046139f7f9bSDimitry Andric       unsigned NextBitCode = Stream.readRecord(Code, Record);
104717a519f9SDimitry Andric       assert(NextBitCode == bitc::METADATA_NAMED_NODE); (void)NextBitCode;
1048f22ef01cSRoman Divacky 
1049f22ef01cSRoman Divacky       // Read named metadata elements.
1050f22ef01cSRoman Divacky       unsigned Size = Record.size();
1051e580952dSDimitry Andric       NamedMDNode *NMD = TheModule->getOrInsertNamedMetadata(Name);
1052f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; ++i) {
105391bc56edSDimitry Andric         MDNode *MD = dyn_cast_or_null<MDNode>(MDValueList.getValueFwdRef(Record[i]));
105491bc56edSDimitry Andric         if (!MD)
1055f785676fSDimitry Andric           return Error(InvalidRecord);
1056e580952dSDimitry Andric         NMD->addOperand(MD);
1057f22ef01cSRoman Divacky       }
1058f22ef01cSRoman Divacky       break;
1059f22ef01cSRoman Divacky     }
106017a519f9SDimitry Andric     case bitc::METADATA_FN_NODE:
1061f22ef01cSRoman Divacky       IsFunctionLocal = true;
1062f22ef01cSRoman Divacky       // fall-through
106317a519f9SDimitry Andric     case bitc::METADATA_NODE: {
1064ffd1746dSEd Schouten       if (Record.size() % 2 == 1)
1065f785676fSDimitry Andric         return Error(InvalidRecord);
1066f22ef01cSRoman Divacky 
1067f22ef01cSRoman Divacky       unsigned Size = Record.size();
1068f22ef01cSRoman Divacky       SmallVector<Value*, 8> Elts;
1069f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; i += 2) {
10706122f3e6SDimitry Andric         Type *Ty = getTypeByID(Record[i]);
1071f785676fSDimitry Andric         if (!Ty)
1072f785676fSDimitry Andric           return Error(InvalidRecord);
1073f22ef01cSRoman Divacky         if (Ty->isMetadataTy())
1074f22ef01cSRoman Divacky           Elts.push_back(MDValueList.getValueFwdRef(Record[i+1]));
1075f22ef01cSRoman Divacky         else if (!Ty->isVoidTy())
1076f22ef01cSRoman Divacky           Elts.push_back(ValueList.getValueFwdRef(Record[i+1], Ty));
1077f22ef01cSRoman Divacky         else
107891bc56edSDimitry Andric           Elts.push_back(nullptr);
1079f22ef01cSRoman Divacky       }
10803b0f4066SDimitry Andric       Value *V = MDNode::getWhenValsUnresolved(Context, Elts, IsFunctionLocal);
1081f22ef01cSRoman Divacky       IsFunctionLocal = false;
1082f22ef01cSRoman Divacky       MDValueList.AssignValue(V, NextMDValueNo++);
1083f22ef01cSRoman Divacky       break;
1084f22ef01cSRoman Divacky     }
1085f22ef01cSRoman Divacky     case bitc::METADATA_STRING: {
108691bc56edSDimitry Andric       std::string String(Record.begin(), Record.end());
108791bc56edSDimitry Andric       llvm::UpgradeMDStringConstant(String);
10887ae0e2c9SDimitry Andric       Value *V = MDString::get(Context, String);
1089f22ef01cSRoman Divacky       MDValueList.AssignValue(V, NextMDValueNo++);
1090f22ef01cSRoman Divacky       break;
1091f22ef01cSRoman Divacky     }
1092f22ef01cSRoman Divacky     case bitc::METADATA_KIND: {
10937ae0e2c9SDimitry Andric       if (Record.size() < 2)
1094f785676fSDimitry Andric         return Error(InvalidRecord);
10957ae0e2c9SDimitry Andric 
1096f22ef01cSRoman Divacky       unsigned Kind = Record[0];
10977ae0e2c9SDimitry Andric       SmallString<8> Name(Record.begin()+1, Record.end());
1098f22ef01cSRoman Divacky 
1099f22ef01cSRoman Divacky       unsigned NewKind = TheModule->getMDKindID(Name.str());
1100e580952dSDimitry Andric       if (!MDKindMap.insert(std::make_pair(Kind, NewKind)).second)
1101f785676fSDimitry Andric         return Error(ConflictingMETADATA_KINDRecords);
1102f22ef01cSRoman Divacky       break;
1103f22ef01cSRoman Divacky     }
1104f22ef01cSRoman Divacky     }
1105f22ef01cSRoman Divacky   }
1106f22ef01cSRoman Divacky }
1107f22ef01cSRoman Divacky 
11083861d79fSDimitry Andric /// decodeSignRotatedValue - Decode a signed value stored with the sign bit in
1109f22ef01cSRoman Divacky /// the LSB for dense VBR encoding.
11103861d79fSDimitry Andric uint64_t BitcodeReader::decodeSignRotatedValue(uint64_t V) {
1111f22ef01cSRoman Divacky   if ((V & 1) == 0)
1112f22ef01cSRoman Divacky     return V >> 1;
1113f22ef01cSRoman Divacky   if (V != 1)
1114f22ef01cSRoman Divacky     return -(V >> 1);
1115f22ef01cSRoman Divacky   // There is no such thing as -0 with integers.  "-0" really means MININT.
1116f22ef01cSRoman Divacky   return 1ULL << 63;
1117f22ef01cSRoman Divacky }
1118f22ef01cSRoman Divacky 
1119f22ef01cSRoman Divacky /// ResolveGlobalAndAliasInits - Resolve all of the initializers for global
1120f22ef01cSRoman Divacky /// values and aliases that we can.
112191bc56edSDimitry Andric std::error_code BitcodeReader::ResolveGlobalAndAliasInits() {
1122f22ef01cSRoman Divacky   std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInitWorklist;
1123f22ef01cSRoman Divacky   std::vector<std::pair<GlobalAlias*, unsigned> > AliasInitWorklist;
1124f785676fSDimitry Andric   std::vector<std::pair<Function*, unsigned> > FunctionPrefixWorklist;
1125f22ef01cSRoman Divacky 
1126f22ef01cSRoman Divacky   GlobalInitWorklist.swap(GlobalInits);
1127f22ef01cSRoman Divacky   AliasInitWorklist.swap(AliasInits);
1128f785676fSDimitry Andric   FunctionPrefixWorklist.swap(FunctionPrefixes);
1129f22ef01cSRoman Divacky 
1130f22ef01cSRoman Divacky   while (!GlobalInitWorklist.empty()) {
1131f22ef01cSRoman Divacky     unsigned ValID = GlobalInitWorklist.back().second;
1132f22ef01cSRoman Divacky     if (ValID >= ValueList.size()) {
1133f22ef01cSRoman Divacky       // Not ready to resolve this yet, it requires something later in the file.
1134f22ef01cSRoman Divacky       GlobalInits.push_back(GlobalInitWorklist.back());
1135f22ef01cSRoman Divacky     } else {
113691bc56edSDimitry Andric       if (Constant *C = dyn_cast_or_null<Constant>(ValueList[ValID]))
1137f22ef01cSRoman Divacky         GlobalInitWorklist.back().first->setInitializer(C);
1138f22ef01cSRoman Divacky       else
1139f785676fSDimitry Andric         return Error(ExpectedConstant);
1140f22ef01cSRoman Divacky     }
1141f22ef01cSRoman Divacky     GlobalInitWorklist.pop_back();
1142f22ef01cSRoman Divacky   }
1143f22ef01cSRoman Divacky 
1144f22ef01cSRoman Divacky   while (!AliasInitWorklist.empty()) {
1145f22ef01cSRoman Divacky     unsigned ValID = AliasInitWorklist.back().second;
1146f22ef01cSRoman Divacky     if (ValID >= ValueList.size()) {
1147f22ef01cSRoman Divacky       AliasInits.push_back(AliasInitWorklist.back());
1148f22ef01cSRoman Divacky     } else {
114991bc56edSDimitry Andric       if (Constant *C = dyn_cast_or_null<Constant>(ValueList[ValID]))
1150f22ef01cSRoman Divacky         AliasInitWorklist.back().first->setAliasee(C);
1151f22ef01cSRoman Divacky       else
1152f785676fSDimitry Andric         return Error(ExpectedConstant);
1153f22ef01cSRoman Divacky     }
1154f22ef01cSRoman Divacky     AliasInitWorklist.pop_back();
1155f22ef01cSRoman Divacky   }
1156f785676fSDimitry Andric 
1157f785676fSDimitry Andric   while (!FunctionPrefixWorklist.empty()) {
1158f785676fSDimitry Andric     unsigned ValID = FunctionPrefixWorklist.back().second;
1159f785676fSDimitry Andric     if (ValID >= ValueList.size()) {
1160f785676fSDimitry Andric       FunctionPrefixes.push_back(FunctionPrefixWorklist.back());
1161f785676fSDimitry Andric     } else {
116291bc56edSDimitry Andric       if (Constant *C = dyn_cast_or_null<Constant>(ValueList[ValID]))
1163f785676fSDimitry Andric         FunctionPrefixWorklist.back().first->setPrefixData(C);
1164f785676fSDimitry Andric       else
1165f785676fSDimitry Andric         return Error(ExpectedConstant);
1166f785676fSDimitry Andric     }
1167f785676fSDimitry Andric     FunctionPrefixWorklist.pop_back();
1168f785676fSDimitry Andric   }
1169f785676fSDimitry Andric 
117091bc56edSDimitry Andric   return std::error_code();
1171f22ef01cSRoman Divacky }
1172f22ef01cSRoman Divacky 
11737ae0e2c9SDimitry Andric static APInt ReadWideAPInt(ArrayRef<uint64_t> Vals, unsigned TypeBits) {
11747ae0e2c9SDimitry Andric   SmallVector<uint64_t, 8> Words(Vals.size());
11757ae0e2c9SDimitry Andric   std::transform(Vals.begin(), Vals.end(), Words.begin(),
11763861d79fSDimitry Andric                  BitcodeReader::decodeSignRotatedValue);
11777ae0e2c9SDimitry Andric 
11787ae0e2c9SDimitry Andric   return APInt(TypeBits, Words);
11797ae0e2c9SDimitry Andric }
11807ae0e2c9SDimitry Andric 
118191bc56edSDimitry Andric std::error_code BitcodeReader::ParseConstants() {
1182f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::CONSTANTS_BLOCK_ID))
1183f785676fSDimitry Andric     return Error(InvalidRecord);
1184f22ef01cSRoman Divacky 
1185f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1186f22ef01cSRoman Divacky 
1187f22ef01cSRoman Divacky   // Read all the records for this value table.
11886122f3e6SDimitry Andric   Type *CurTy = Type::getInt32Ty(Context);
1189f22ef01cSRoman Divacky   unsigned NextCstNo = ValueList.size();
1190f22ef01cSRoman Divacky   while (1) {
1191139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
1192139f7f9bSDimitry Andric 
1193139f7f9bSDimitry Andric     switch (Entry.Kind) {
1194139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
1195139f7f9bSDimitry Andric     case BitstreamEntry::Error:
1196f785676fSDimitry Andric       return Error(MalformedBlock);
1197139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
1198139f7f9bSDimitry Andric       if (NextCstNo != ValueList.size())
1199f785676fSDimitry Andric         return Error(InvalidConstantReference);
1200139f7f9bSDimitry Andric 
1201139f7f9bSDimitry Andric       // Once all the constants have been read, go through and resolve forward
1202139f7f9bSDimitry Andric       // references.
1203139f7f9bSDimitry Andric       ValueList.ResolveConstantForwardRefs();
120491bc56edSDimitry Andric       return std::error_code();
1205139f7f9bSDimitry Andric     case BitstreamEntry::Record:
1206139f7f9bSDimitry Andric       // The interesting case.
1207f22ef01cSRoman Divacky       break;
1208f22ef01cSRoman Divacky     }
1209f22ef01cSRoman Divacky 
1210f22ef01cSRoman Divacky     // Read a record.
1211f22ef01cSRoman Divacky     Record.clear();
121291bc56edSDimitry Andric     Value *V = nullptr;
1213139f7f9bSDimitry Andric     unsigned BitCode = Stream.readRecord(Entry.ID, Record);
1214f22ef01cSRoman Divacky     switch (BitCode) {
1215f22ef01cSRoman Divacky     default:  // Default behavior: unknown constant
1216f22ef01cSRoman Divacky     case bitc::CST_CODE_UNDEF:     // UNDEF
1217f22ef01cSRoman Divacky       V = UndefValue::get(CurTy);
1218f22ef01cSRoman Divacky       break;
1219f22ef01cSRoman Divacky     case bitc::CST_CODE_SETTYPE:   // SETTYPE: [typeid]
1220f22ef01cSRoman Divacky       if (Record.empty())
1221f785676fSDimitry Andric         return Error(InvalidRecord);
122291bc56edSDimitry Andric       if (Record[0] >= TypeList.size() || !TypeList[Record[0]])
1223f785676fSDimitry Andric         return Error(InvalidRecord);
1224f22ef01cSRoman Divacky       CurTy = TypeList[Record[0]];
1225f22ef01cSRoman Divacky       continue;  // Skip the ValueList manipulation.
1226f22ef01cSRoman Divacky     case bitc::CST_CODE_NULL:      // NULL
1227f22ef01cSRoman Divacky       V = Constant::getNullValue(CurTy);
1228f22ef01cSRoman Divacky       break;
1229f22ef01cSRoman Divacky     case bitc::CST_CODE_INTEGER:   // INTEGER: [intval]
1230f22ef01cSRoman Divacky       if (!CurTy->isIntegerTy() || Record.empty())
1231f785676fSDimitry Andric         return Error(InvalidRecord);
12323861d79fSDimitry Andric       V = ConstantInt::get(CurTy, decodeSignRotatedValue(Record[0]));
1233f22ef01cSRoman Divacky       break;
1234f22ef01cSRoman Divacky     case bitc::CST_CODE_WIDE_INTEGER: {// WIDE_INTEGER: [n x intval]
1235f22ef01cSRoman Divacky       if (!CurTy->isIntegerTy() || Record.empty())
1236f785676fSDimitry Andric         return Error(InvalidRecord);
1237f22ef01cSRoman Divacky 
12387ae0e2c9SDimitry Andric       APInt VInt = ReadWideAPInt(Record,
12397ae0e2c9SDimitry Andric                                  cast<IntegerType>(CurTy)->getBitWidth());
12407ae0e2c9SDimitry Andric       V = ConstantInt::get(Context, VInt);
12417ae0e2c9SDimitry Andric 
1242f22ef01cSRoman Divacky       break;
1243f22ef01cSRoman Divacky     }
1244f22ef01cSRoman Divacky     case bitc::CST_CODE_FLOAT: {    // FLOAT: [fpval]
1245f22ef01cSRoman Divacky       if (Record.empty())
1246f785676fSDimitry Andric         return Error(InvalidRecord);
1247dff0c46cSDimitry Andric       if (CurTy->isHalfTy())
1248139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::IEEEhalf,
1249139f7f9bSDimitry Andric                                              APInt(16, (uint16_t)Record[0])));
1250dff0c46cSDimitry Andric       else if (CurTy->isFloatTy())
1251139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::IEEEsingle,
1252139f7f9bSDimitry Andric                                              APInt(32, (uint32_t)Record[0])));
1253f22ef01cSRoman Divacky       else if (CurTy->isDoubleTy())
1254139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::IEEEdouble,
1255139f7f9bSDimitry Andric                                              APInt(64, Record[0])));
1256f22ef01cSRoman Divacky       else if (CurTy->isX86_FP80Ty()) {
1257f22ef01cSRoman Divacky         // Bits are not stored the same way as a normal i80 APInt, compensate.
1258f22ef01cSRoman Divacky         uint64_t Rearrange[2];
1259f22ef01cSRoman Divacky         Rearrange[0] = (Record[1] & 0xffffLL) | (Record[0] << 16);
1260f22ef01cSRoman Divacky         Rearrange[1] = Record[0] >> 48;
1261139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::x87DoubleExtended,
1262139f7f9bSDimitry Andric                                              APInt(80, Rearrange)));
1263f22ef01cSRoman Divacky       } else if (CurTy->isFP128Ty())
1264139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::IEEEquad,
1265139f7f9bSDimitry Andric                                              APInt(128, Record)));
1266f22ef01cSRoman Divacky       else if (CurTy->isPPC_FP128Ty())
1267139f7f9bSDimitry Andric         V = ConstantFP::get(Context, APFloat(APFloat::PPCDoubleDouble,
1268139f7f9bSDimitry Andric                                              APInt(128, Record)));
1269f22ef01cSRoman Divacky       else
1270f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);
1271f22ef01cSRoman Divacky       break;
1272f22ef01cSRoman Divacky     }
1273f22ef01cSRoman Divacky 
1274f22ef01cSRoman Divacky     case bitc::CST_CODE_AGGREGATE: {// AGGREGATE: [n x value number]
1275f22ef01cSRoman Divacky       if (Record.empty())
1276f785676fSDimitry Andric         return Error(InvalidRecord);
1277f22ef01cSRoman Divacky 
1278f22ef01cSRoman Divacky       unsigned Size = Record.size();
1279dff0c46cSDimitry Andric       SmallVector<Constant*, 16> Elts;
1280f22ef01cSRoman Divacky 
12816122f3e6SDimitry Andric       if (StructType *STy = dyn_cast<StructType>(CurTy)) {
1282f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1283f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i],
1284f22ef01cSRoman Divacky                                                      STy->getElementType(i)));
1285f22ef01cSRoman Divacky         V = ConstantStruct::get(STy, Elts);
12866122f3e6SDimitry Andric       } else if (ArrayType *ATy = dyn_cast<ArrayType>(CurTy)) {
12876122f3e6SDimitry Andric         Type *EltTy = ATy->getElementType();
1288f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1289f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
1290f22ef01cSRoman Divacky         V = ConstantArray::get(ATy, Elts);
12916122f3e6SDimitry Andric       } else if (VectorType *VTy = dyn_cast<VectorType>(CurTy)) {
12926122f3e6SDimitry Andric         Type *EltTy = VTy->getElementType();
1293f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1294f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
1295f22ef01cSRoman Divacky         V = ConstantVector::get(Elts);
1296f22ef01cSRoman Divacky       } else {
1297f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);
1298f22ef01cSRoman Divacky       }
1299f22ef01cSRoman Divacky       break;
1300f22ef01cSRoman Divacky     }
1301dff0c46cSDimitry Andric     case bitc::CST_CODE_STRING:    // STRING: [values]
1302f22ef01cSRoman Divacky     case bitc::CST_CODE_CSTRING: { // CSTRING: [values]
1303f22ef01cSRoman Divacky       if (Record.empty())
1304f785676fSDimitry Andric         return Error(InvalidRecord);
1305f22ef01cSRoman Divacky 
13067ae0e2c9SDimitry Andric       SmallString<16> Elts(Record.begin(), Record.end());
1307dff0c46cSDimitry Andric       V = ConstantDataArray::getString(Context, Elts,
1308dff0c46cSDimitry Andric                                        BitCode == bitc::CST_CODE_CSTRING);
1309f22ef01cSRoman Divacky       break;
1310f22ef01cSRoman Divacky     }
1311dff0c46cSDimitry Andric     case bitc::CST_CODE_DATA: {// DATA: [n x value]
1312dff0c46cSDimitry Andric       if (Record.empty())
1313f785676fSDimitry Andric         return Error(InvalidRecord);
1314dff0c46cSDimitry Andric 
1315dff0c46cSDimitry Andric       Type *EltTy = cast<SequentialType>(CurTy)->getElementType();
1316dff0c46cSDimitry Andric       unsigned Size = Record.size();
1317dff0c46cSDimitry Andric 
1318dff0c46cSDimitry Andric       if (EltTy->isIntegerTy(8)) {
1319dff0c46cSDimitry Andric         SmallVector<uint8_t, 16> Elts(Record.begin(), Record.end());
1320dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1321dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1322dff0c46cSDimitry Andric         else
1323dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1324dff0c46cSDimitry Andric       } else if (EltTy->isIntegerTy(16)) {
1325dff0c46cSDimitry Andric         SmallVector<uint16_t, 16> Elts(Record.begin(), Record.end());
1326dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1327dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1328dff0c46cSDimitry Andric         else
1329dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1330dff0c46cSDimitry Andric       } else if (EltTy->isIntegerTy(32)) {
1331dff0c46cSDimitry Andric         SmallVector<uint32_t, 16> Elts(Record.begin(), Record.end());
1332dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1333dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1334dff0c46cSDimitry Andric         else
1335dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1336dff0c46cSDimitry Andric       } else if (EltTy->isIntegerTy(64)) {
1337dff0c46cSDimitry Andric         SmallVector<uint64_t, 16> Elts(Record.begin(), Record.end());
1338dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1339dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1340dff0c46cSDimitry Andric         else
1341dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1342dff0c46cSDimitry Andric       } else if (EltTy->isFloatTy()) {
13437ae0e2c9SDimitry Andric         SmallVector<float, 16> Elts(Size);
13447ae0e2c9SDimitry Andric         std::transform(Record.begin(), Record.end(), Elts.begin(), BitsToFloat);
1345dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1346dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1347dff0c46cSDimitry Andric         else
1348dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1349dff0c46cSDimitry Andric       } else if (EltTy->isDoubleTy()) {
13507ae0e2c9SDimitry Andric         SmallVector<double, 16> Elts(Size);
13517ae0e2c9SDimitry Andric         std::transform(Record.begin(), Record.end(), Elts.begin(),
13527ae0e2c9SDimitry Andric                        BitsToDouble);
1353dff0c46cSDimitry Andric         if (isa<VectorType>(CurTy))
1354dff0c46cSDimitry Andric           V = ConstantDataVector::get(Context, Elts);
1355dff0c46cSDimitry Andric         else
1356dff0c46cSDimitry Andric           V = ConstantDataArray::get(Context, Elts);
1357dff0c46cSDimitry Andric       } else {
1358f785676fSDimitry Andric         return Error(InvalidTypeForValue);
1359dff0c46cSDimitry Andric       }
1360dff0c46cSDimitry Andric       break;
1361dff0c46cSDimitry Andric     }
1362dff0c46cSDimitry Andric 
1363f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_BINOP: {  // CE_BINOP: [opcode, opval, opval]
1364f785676fSDimitry Andric       if (Record.size() < 3)
1365f785676fSDimitry Andric         return Error(InvalidRecord);
1366f22ef01cSRoman Divacky       int Opc = GetDecodedBinaryOpcode(Record[0], CurTy);
1367f22ef01cSRoman Divacky       if (Opc < 0) {
1368f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);  // Unknown binop.
1369f22ef01cSRoman Divacky       } else {
1370f22ef01cSRoman Divacky         Constant *LHS = ValueList.getConstantFwdRef(Record[1], CurTy);
1371f22ef01cSRoman Divacky         Constant *RHS = ValueList.getConstantFwdRef(Record[2], CurTy);
1372f22ef01cSRoman Divacky         unsigned Flags = 0;
1373f22ef01cSRoman Divacky         if (Record.size() >= 4) {
1374f22ef01cSRoman Divacky           if (Opc == Instruction::Add ||
1375f22ef01cSRoman Divacky               Opc == Instruction::Sub ||
13762754fe60SDimitry Andric               Opc == Instruction::Mul ||
13772754fe60SDimitry Andric               Opc == Instruction::Shl) {
1378f22ef01cSRoman Divacky             if (Record[3] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1379f22ef01cSRoman Divacky               Flags |= OverflowingBinaryOperator::NoSignedWrap;
1380f22ef01cSRoman Divacky             if (Record[3] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1381f22ef01cSRoman Divacky               Flags |= OverflowingBinaryOperator::NoUnsignedWrap;
13822754fe60SDimitry Andric           } else if (Opc == Instruction::SDiv ||
13832754fe60SDimitry Andric                      Opc == Instruction::UDiv ||
13842754fe60SDimitry Andric                      Opc == Instruction::LShr ||
13852754fe60SDimitry Andric                      Opc == Instruction::AShr) {
13862754fe60SDimitry Andric             if (Record[3] & (1 << bitc::PEO_EXACT))
1387f22ef01cSRoman Divacky               Flags |= SDivOperator::IsExact;
1388f22ef01cSRoman Divacky           }
1389f22ef01cSRoman Divacky         }
1390f22ef01cSRoman Divacky         V = ConstantExpr::get(Opc, LHS, RHS, Flags);
1391f22ef01cSRoman Divacky       }
1392f22ef01cSRoman Divacky       break;
1393f22ef01cSRoman Divacky     }
1394f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_CAST: {  // CE_CAST: [opcode, opty, opval]
1395f785676fSDimitry Andric       if (Record.size() < 3)
1396f785676fSDimitry Andric         return Error(InvalidRecord);
1397f22ef01cSRoman Divacky       int Opc = GetDecodedCastOpcode(Record[0]);
1398f22ef01cSRoman Divacky       if (Opc < 0) {
1399f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);  // Unknown cast.
1400f22ef01cSRoman Divacky       } else {
14016122f3e6SDimitry Andric         Type *OpTy = getTypeByID(Record[1]);
1402f785676fSDimitry Andric         if (!OpTy)
1403f785676fSDimitry Andric           return Error(InvalidRecord);
1404f22ef01cSRoman Divacky         Constant *Op = ValueList.getConstantFwdRef(Record[2], OpTy);
1405f785676fSDimitry Andric         V = UpgradeBitCastExpr(Opc, Op, CurTy);
1406f785676fSDimitry Andric         if (!V) V = ConstantExpr::getCast(Opc, Op, CurTy);
1407f22ef01cSRoman Divacky       }
1408f22ef01cSRoman Divacky       break;
1409f22ef01cSRoman Divacky     }
1410f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_INBOUNDS_GEP:
1411f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_GEP: {  // CE_GEP:        [n x operands]
1412f785676fSDimitry Andric       if (Record.size() & 1)
1413f785676fSDimitry Andric         return Error(InvalidRecord);
1414f22ef01cSRoman Divacky       SmallVector<Constant*, 16> Elts;
1415f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
14166122f3e6SDimitry Andric         Type *ElTy = getTypeByID(Record[i]);
1417f785676fSDimitry Andric         if (!ElTy)
1418f785676fSDimitry Andric           return Error(InvalidRecord);
1419f22ef01cSRoman Divacky         Elts.push_back(ValueList.getConstantFwdRef(Record[i+1], ElTy));
1420f22ef01cSRoman Divacky       }
14216122f3e6SDimitry Andric       ArrayRef<Constant *> Indices(Elts.begin() + 1, Elts.end());
14226122f3e6SDimitry Andric       V = ConstantExpr::getGetElementPtr(Elts[0], Indices,
14236122f3e6SDimitry Andric                                          BitCode ==
14246122f3e6SDimitry Andric                                            bitc::CST_CODE_CE_INBOUNDS_GEP);
1425f22ef01cSRoman Divacky       break;
1426f22ef01cSRoman Divacky     }
1427f785676fSDimitry Andric     case bitc::CST_CODE_CE_SELECT: {  // CE_SELECT: [opval#, opval#, opval#]
1428f785676fSDimitry Andric       if (Record.size() < 3)
1429f785676fSDimitry Andric         return Error(InvalidRecord);
1430f785676fSDimitry Andric 
1431f785676fSDimitry Andric       Type *SelectorTy = Type::getInt1Ty(Context);
1432f785676fSDimitry Andric 
1433f785676fSDimitry Andric       // If CurTy is a vector of length n, then Record[0] must be a <n x i1>
1434f785676fSDimitry Andric       // vector. Otherwise, it must be a single bit.
1435f785676fSDimitry Andric       if (VectorType *VTy = dyn_cast<VectorType>(CurTy))
1436f785676fSDimitry Andric         SelectorTy = VectorType::get(Type::getInt1Ty(Context),
1437f785676fSDimitry Andric                                      VTy->getNumElements());
1438f785676fSDimitry Andric 
1439f785676fSDimitry Andric       V = ConstantExpr::getSelect(ValueList.getConstantFwdRef(Record[0],
1440f785676fSDimitry Andric                                                               SelectorTy),
1441f22ef01cSRoman Divacky                                   ValueList.getConstantFwdRef(Record[1],CurTy),
1442f22ef01cSRoman Divacky                                   ValueList.getConstantFwdRef(Record[2],CurTy));
1443f22ef01cSRoman Divacky       break;
1444f785676fSDimitry Andric     }
144591bc56edSDimitry Andric     case bitc::CST_CODE_CE_EXTRACTELT
144691bc56edSDimitry Andric         : { // CE_EXTRACTELT: [opty, opval, opty, opval]
1447f785676fSDimitry Andric       if (Record.size() < 3)
1448f785676fSDimitry Andric         return Error(InvalidRecord);
14496122f3e6SDimitry Andric       VectorType *OpTy =
1450f22ef01cSRoman Divacky         dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
145191bc56edSDimitry Andric       if (!OpTy)
1452f785676fSDimitry Andric         return Error(InvalidRecord);
1453f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
145491bc56edSDimitry Andric       Constant *Op1 = nullptr;
145591bc56edSDimitry Andric       if (Record.size() == 4) {
145691bc56edSDimitry Andric         Type *IdxTy = getTypeByID(Record[2]);
145791bc56edSDimitry Andric         if (!IdxTy)
145891bc56edSDimitry Andric           return Error(InvalidRecord);
145991bc56edSDimitry Andric         Op1 = ValueList.getConstantFwdRef(Record[3], IdxTy);
146091bc56edSDimitry Andric       } else // TODO: Remove with llvm 4.0
146191bc56edSDimitry Andric         Op1 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
146291bc56edSDimitry Andric       if (!Op1)
146391bc56edSDimitry Andric         return Error(InvalidRecord);
1464f22ef01cSRoman Divacky       V = ConstantExpr::getExtractElement(Op0, Op1);
1465f22ef01cSRoman Divacky       break;
1466f22ef01cSRoman Divacky     }
146791bc56edSDimitry Andric     case bitc::CST_CODE_CE_INSERTELT
146891bc56edSDimitry Andric         : { // CE_INSERTELT: [opval, opval, opty, opval]
14696122f3e6SDimitry Andric       VectorType *OpTy = dyn_cast<VectorType>(CurTy);
147091bc56edSDimitry Andric       if (Record.size() < 3 || !OpTy)
1471f785676fSDimitry Andric         return Error(InvalidRecord);
1472f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1473f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[1],
1474f22ef01cSRoman Divacky                                                   OpTy->getElementType());
147591bc56edSDimitry Andric       Constant *Op2 = nullptr;
147691bc56edSDimitry Andric       if (Record.size() == 4) {
147791bc56edSDimitry Andric         Type *IdxTy = getTypeByID(Record[2]);
147891bc56edSDimitry Andric         if (!IdxTy)
147991bc56edSDimitry Andric           return Error(InvalidRecord);
148091bc56edSDimitry Andric         Op2 = ValueList.getConstantFwdRef(Record[3], IdxTy);
148191bc56edSDimitry Andric       } else // TODO: Remove with llvm 4.0
148291bc56edSDimitry Andric         Op2 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
148391bc56edSDimitry Andric       if (!Op2)
148491bc56edSDimitry Andric         return Error(InvalidRecord);
1485f22ef01cSRoman Divacky       V = ConstantExpr::getInsertElement(Op0, Op1, Op2);
1486f22ef01cSRoman Divacky       break;
1487f22ef01cSRoman Divacky     }
1488f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SHUFFLEVEC: { // CE_SHUFFLEVEC: [opval, opval, opval]
14896122f3e6SDimitry Andric       VectorType *OpTy = dyn_cast<VectorType>(CurTy);
149091bc56edSDimitry Andric       if (Record.size() < 3 || !OpTy)
1491f785676fSDimitry Andric         return Error(InvalidRecord);
1492f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1493f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[1], OpTy);
14946122f3e6SDimitry Andric       Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
1495f22ef01cSRoman Divacky                                                  OpTy->getNumElements());
1496f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[2], ShufTy);
1497f22ef01cSRoman Divacky       V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
1498f22ef01cSRoman Divacky       break;
1499f22ef01cSRoman Divacky     }
1500f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SHUFVEC_EX: { // [opty, opval, opval, opval]
15016122f3e6SDimitry Andric       VectorType *RTy = dyn_cast<VectorType>(CurTy);
15026122f3e6SDimitry Andric       VectorType *OpTy =
15032754fe60SDimitry Andric         dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
150491bc56edSDimitry Andric       if (Record.size() < 4 || !RTy || !OpTy)
1505f785676fSDimitry Andric         return Error(InvalidRecord);
1506f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1507f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
15086122f3e6SDimitry Andric       Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
1509f22ef01cSRoman Divacky                                                  RTy->getNumElements());
1510f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[3], ShufTy);
1511f22ef01cSRoman Divacky       V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
1512f22ef01cSRoman Divacky       break;
1513f22ef01cSRoman Divacky     }
1514f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_CMP: {     // CE_CMP: [opty, opval, opval, pred]
1515f785676fSDimitry Andric       if (Record.size() < 4)
1516f785676fSDimitry Andric         return Error(InvalidRecord);
15176122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[0]);
151891bc56edSDimitry Andric       if (!OpTy)
1519f785676fSDimitry Andric         return Error(InvalidRecord);
1520f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1521f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1522f22ef01cSRoman Divacky 
1523f22ef01cSRoman Divacky       if (OpTy->isFPOrFPVectorTy())
1524f22ef01cSRoman Divacky         V = ConstantExpr::getFCmp(Record[3], Op0, Op1);
1525f22ef01cSRoman Divacky       else
1526f22ef01cSRoman Divacky         V = ConstantExpr::getICmp(Record[3], Op0, Op1);
1527f22ef01cSRoman Divacky       break;
1528f22ef01cSRoman Divacky     }
15293861d79fSDimitry Andric     // This maintains backward compatibility, pre-asm dialect keywords.
15303861d79fSDimitry Andric     // FIXME: Remove with the 4.0 release.
15313861d79fSDimitry Andric     case bitc::CST_CODE_INLINEASM_OLD: {
1532f785676fSDimitry Andric       if (Record.size() < 2)
1533f785676fSDimitry Andric         return Error(InvalidRecord);
1534f22ef01cSRoman Divacky       std::string AsmStr, ConstrStr;
1535f22ef01cSRoman Divacky       bool HasSideEffects = Record[0] & 1;
1536f22ef01cSRoman Divacky       bool IsAlignStack = Record[0] >> 1;
1537f22ef01cSRoman Divacky       unsigned AsmStrSize = Record[1];
1538f22ef01cSRoman Divacky       if (2+AsmStrSize >= Record.size())
1539f785676fSDimitry Andric         return Error(InvalidRecord);
1540f22ef01cSRoman Divacky       unsigned ConstStrSize = Record[2+AsmStrSize];
1541f22ef01cSRoman Divacky       if (3+AsmStrSize+ConstStrSize > Record.size())
1542f785676fSDimitry Andric         return Error(InvalidRecord);
1543f22ef01cSRoman Divacky 
1544f22ef01cSRoman Divacky       for (unsigned i = 0; i != AsmStrSize; ++i)
1545f22ef01cSRoman Divacky         AsmStr += (char)Record[2+i];
1546f22ef01cSRoman Divacky       for (unsigned i = 0; i != ConstStrSize; ++i)
1547f22ef01cSRoman Divacky         ConstrStr += (char)Record[3+AsmStrSize+i];
15486122f3e6SDimitry Andric       PointerType *PTy = cast<PointerType>(CurTy);
1549f22ef01cSRoman Divacky       V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
1550f22ef01cSRoman Divacky                          AsmStr, ConstrStr, HasSideEffects, IsAlignStack);
1551f22ef01cSRoman Divacky       break;
1552f22ef01cSRoman Divacky     }
15533861d79fSDimitry Andric     // This version adds support for the asm dialect keywords (e.g.,
15543861d79fSDimitry Andric     // inteldialect).
15553861d79fSDimitry Andric     case bitc::CST_CODE_INLINEASM: {
1556f785676fSDimitry Andric       if (Record.size() < 2)
1557f785676fSDimitry Andric         return Error(InvalidRecord);
15583861d79fSDimitry Andric       std::string AsmStr, ConstrStr;
15593861d79fSDimitry Andric       bool HasSideEffects = Record[0] & 1;
15603861d79fSDimitry Andric       bool IsAlignStack = (Record[0] >> 1) & 1;
15613861d79fSDimitry Andric       unsigned AsmDialect = Record[0] >> 2;
15623861d79fSDimitry Andric       unsigned AsmStrSize = Record[1];
15633861d79fSDimitry Andric       if (2+AsmStrSize >= Record.size())
1564f785676fSDimitry Andric         return Error(InvalidRecord);
15653861d79fSDimitry Andric       unsigned ConstStrSize = Record[2+AsmStrSize];
15663861d79fSDimitry Andric       if (3+AsmStrSize+ConstStrSize > Record.size())
1567f785676fSDimitry Andric         return Error(InvalidRecord);
15683861d79fSDimitry Andric 
15693861d79fSDimitry Andric       for (unsigned i = 0; i != AsmStrSize; ++i)
15703861d79fSDimitry Andric         AsmStr += (char)Record[2+i];
15713861d79fSDimitry Andric       for (unsigned i = 0; i != ConstStrSize; ++i)
15723861d79fSDimitry Andric         ConstrStr += (char)Record[3+AsmStrSize+i];
15733861d79fSDimitry Andric       PointerType *PTy = cast<PointerType>(CurTy);
15743861d79fSDimitry Andric       V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
15753861d79fSDimitry Andric                          AsmStr, ConstrStr, HasSideEffects, IsAlignStack,
15763861d79fSDimitry Andric                          InlineAsm::AsmDialect(AsmDialect));
15773861d79fSDimitry Andric       break;
15783861d79fSDimitry Andric     }
1579f22ef01cSRoman Divacky     case bitc::CST_CODE_BLOCKADDRESS:{
1580f785676fSDimitry Andric       if (Record.size() < 3)
1581f785676fSDimitry Andric         return Error(InvalidRecord);
15826122f3e6SDimitry Andric       Type *FnTy = getTypeByID(Record[0]);
158391bc56edSDimitry Andric       if (!FnTy)
1584f785676fSDimitry Andric         return Error(InvalidRecord);
1585f22ef01cSRoman Divacky       Function *Fn =
1586f22ef01cSRoman Divacky         dyn_cast_or_null<Function>(ValueList.getConstantFwdRef(Record[1],FnTy));
158791bc56edSDimitry Andric       if (!Fn)
1588f785676fSDimitry Andric         return Error(InvalidRecord);
1589f22ef01cSRoman Divacky 
15903861d79fSDimitry Andric       // If the function is already parsed we can insert the block address right
15913861d79fSDimitry Andric       // away.
15923861d79fSDimitry Andric       if (!Fn->empty()) {
15933861d79fSDimitry Andric         Function::iterator BBI = Fn->begin(), BBE = Fn->end();
15943861d79fSDimitry Andric         for (size_t I = 0, E = Record[2]; I != E; ++I) {
15953861d79fSDimitry Andric           if (BBI == BBE)
1596f785676fSDimitry Andric             return Error(InvalidID);
15973861d79fSDimitry Andric           ++BBI;
15983861d79fSDimitry Andric         }
15993861d79fSDimitry Andric         V = BlockAddress::get(Fn, BBI);
16003861d79fSDimitry Andric       } else {
16013861d79fSDimitry Andric         // Otherwise insert a placeholder and remember it so it can be inserted
16023861d79fSDimitry Andric         // when the function is parsed.
1603f22ef01cSRoman Divacky         GlobalVariable *FwdRef = new GlobalVariable(*Fn->getParent(),
1604f22ef01cSRoman Divacky                                                     Type::getInt8Ty(Context),
1605f22ef01cSRoman Divacky                                             false, GlobalValue::InternalLinkage,
160691bc56edSDimitry Andric                                                     nullptr, "");
1607f22ef01cSRoman Divacky         BlockAddrFwdRefs[Fn].push_back(std::make_pair(Record[2], FwdRef));
1608f22ef01cSRoman Divacky         V = FwdRef;
16093861d79fSDimitry Andric       }
1610f22ef01cSRoman Divacky       break;
1611f22ef01cSRoman Divacky     }
1612f22ef01cSRoman Divacky     }
1613f22ef01cSRoman Divacky 
1614f22ef01cSRoman Divacky     ValueList.AssignValue(V, NextCstNo);
1615f22ef01cSRoman Divacky     ++NextCstNo;
1616f22ef01cSRoman Divacky   }
1617f22ef01cSRoman Divacky }
1618f22ef01cSRoman Divacky 
161991bc56edSDimitry Andric std::error_code BitcodeReader::ParseUseLists() {
1620dff0c46cSDimitry Andric   if (Stream.EnterSubBlock(bitc::USELIST_BLOCK_ID))
1621f785676fSDimitry Andric     return Error(InvalidRecord);
1622dff0c46cSDimitry Andric 
1623dff0c46cSDimitry Andric   SmallVector<uint64_t, 64> Record;
1624dff0c46cSDimitry Andric 
1625dff0c46cSDimitry Andric   // Read all the records.
1626dff0c46cSDimitry Andric   while (1) {
1627139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
1628139f7f9bSDimitry Andric 
1629139f7f9bSDimitry Andric     switch (Entry.Kind) {
1630139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
1631139f7f9bSDimitry Andric     case BitstreamEntry::Error:
1632f785676fSDimitry Andric       return Error(MalformedBlock);
1633139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
163491bc56edSDimitry Andric       return std::error_code();
1635139f7f9bSDimitry Andric     case BitstreamEntry::Record:
1636139f7f9bSDimitry Andric       // The interesting case.
1637139f7f9bSDimitry Andric       break;
1638dff0c46cSDimitry Andric     }
1639dff0c46cSDimitry Andric 
1640dff0c46cSDimitry Andric     // Read a use list record.
1641dff0c46cSDimitry Andric     Record.clear();
1642139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
1643dff0c46cSDimitry Andric     default:  // Default behavior: unknown type.
1644dff0c46cSDimitry Andric       break;
1645dff0c46cSDimitry Andric     case bitc::USELIST_CODE_ENTRY: { // USELIST_CODE_ENTRY: TBD.
1646dff0c46cSDimitry Andric       unsigned RecordLength = Record.size();
1647dff0c46cSDimitry Andric       if (RecordLength < 1)
1648f785676fSDimitry Andric         return Error(InvalidRecord);
1649dff0c46cSDimitry Andric       UseListRecords.push_back(Record);
1650dff0c46cSDimitry Andric       break;
1651dff0c46cSDimitry Andric     }
1652dff0c46cSDimitry Andric     }
1653dff0c46cSDimitry Andric   }
1654dff0c46cSDimitry Andric }
1655dff0c46cSDimitry Andric 
1656f22ef01cSRoman Divacky /// RememberAndSkipFunctionBody - When we see the block for a function body,
1657f22ef01cSRoman Divacky /// remember where it is and then skip it.  This lets us lazily deserialize the
1658f22ef01cSRoman Divacky /// functions.
165991bc56edSDimitry Andric std::error_code BitcodeReader::RememberAndSkipFunctionBody() {
1660f22ef01cSRoman Divacky   // Get the function we are talking about.
1661f22ef01cSRoman Divacky   if (FunctionsWithBodies.empty())
1662f785676fSDimitry Andric     return Error(InsufficientFunctionProtos);
1663f22ef01cSRoman Divacky 
1664f22ef01cSRoman Divacky   Function *Fn = FunctionsWithBodies.back();
1665f22ef01cSRoman Divacky   FunctionsWithBodies.pop_back();
1666f22ef01cSRoman Divacky 
1667f22ef01cSRoman Divacky   // Save the current stream state.
1668f22ef01cSRoman Divacky   uint64_t CurBit = Stream.GetCurrentBitNo();
1669f22ef01cSRoman Divacky   DeferredFunctionInfo[Fn] = CurBit;
1670f22ef01cSRoman Divacky 
1671f22ef01cSRoman Divacky   // Skip over the function block for now.
1672f22ef01cSRoman Divacky   if (Stream.SkipBlock())
1673f785676fSDimitry Andric     return Error(InvalidRecord);
167491bc56edSDimitry Andric   return std::error_code();
1675f22ef01cSRoman Divacky }
1676f22ef01cSRoman Divacky 
167791bc56edSDimitry Andric std::error_code BitcodeReader::GlobalCleanup() {
1678f22ef01cSRoman Divacky   // Patch the initializers for globals and aliases up.
1679f22ef01cSRoman Divacky   ResolveGlobalAndAliasInits();
1680f22ef01cSRoman Divacky   if (!GlobalInits.empty() || !AliasInits.empty())
1681f785676fSDimitry Andric     return Error(MalformedGlobalInitializerSet);
1682f22ef01cSRoman Divacky 
1683f22ef01cSRoman Divacky   // Look for intrinsic functions which need to be upgraded at some point
1684f22ef01cSRoman Divacky   for (Module::iterator FI = TheModule->begin(), FE = TheModule->end();
1685f22ef01cSRoman Divacky        FI != FE; ++FI) {
1686f22ef01cSRoman Divacky     Function *NewFn;
1687f22ef01cSRoman Divacky     if (UpgradeIntrinsicFunction(FI, NewFn))
1688f22ef01cSRoman Divacky       UpgradedIntrinsics.push_back(std::make_pair(FI, NewFn));
1689f22ef01cSRoman Divacky   }
1690f22ef01cSRoman Divacky 
1691e580952dSDimitry Andric   // Look for global variables which need to be renamed.
1692e580952dSDimitry Andric   for (Module::global_iterator
1693e580952dSDimitry Andric          GI = TheModule->global_begin(), GE = TheModule->global_end();
169491bc56edSDimitry Andric        GI != GE;) {
169591bc56edSDimitry Andric     GlobalVariable *GV = GI++;
169691bc56edSDimitry Andric     UpgradeGlobalVariable(GV);
169791bc56edSDimitry Andric   }
169891bc56edSDimitry Andric 
1699f22ef01cSRoman Divacky   // Force deallocation of memory for these vectors to favor the client that
1700f22ef01cSRoman Divacky   // want lazy deserialization.
1701f22ef01cSRoman Divacky   std::vector<std::pair<GlobalVariable*, unsigned> >().swap(GlobalInits);
1702f22ef01cSRoman Divacky   std::vector<std::pair<GlobalAlias*, unsigned> >().swap(AliasInits);
170391bc56edSDimitry Andric   return std::error_code();
1704f22ef01cSRoman Divacky }
1705f22ef01cSRoman Divacky 
170691bc56edSDimitry Andric std::error_code BitcodeReader::ParseModule(bool Resume) {
1707dff0c46cSDimitry Andric   if (Resume)
1708dff0c46cSDimitry Andric     Stream.JumpToBit(NextUnreadBit);
1709dff0c46cSDimitry Andric   else if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
1710f785676fSDimitry Andric     return Error(InvalidRecord);
1711dff0c46cSDimitry Andric 
1712dff0c46cSDimitry Andric   SmallVector<uint64_t, 64> Record;
1713dff0c46cSDimitry Andric   std::vector<std::string> SectionTable;
1714dff0c46cSDimitry Andric   std::vector<std::string> GCTable;
1715dff0c46cSDimitry Andric 
1716dff0c46cSDimitry Andric   // Read all the records for this module.
1717139f7f9bSDimitry Andric   while (1) {
1718139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advance();
1719dff0c46cSDimitry Andric 
1720139f7f9bSDimitry Andric     switch (Entry.Kind) {
1721139f7f9bSDimitry Andric     case BitstreamEntry::Error:
1722f785676fSDimitry Andric       return Error(MalformedBlock);
1723139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
1724dff0c46cSDimitry Andric       return GlobalCleanup();
1725dff0c46cSDimitry Andric 
1726139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock:
1727139f7f9bSDimitry Andric       switch (Entry.ID) {
1728f22ef01cSRoman Divacky       default:  // Skip unknown content.
1729f22ef01cSRoman Divacky         if (Stream.SkipBlock())
1730f785676fSDimitry Andric           return Error(InvalidRecord);
1731f22ef01cSRoman Divacky         break;
1732f22ef01cSRoman Divacky       case bitc::BLOCKINFO_BLOCK_ID:
1733f22ef01cSRoman Divacky         if (Stream.ReadBlockInfoBlock())
1734f785676fSDimitry Andric           return Error(MalformedBlock);
1735f22ef01cSRoman Divacky         break;
1736f22ef01cSRoman Divacky       case bitc::PARAMATTR_BLOCK_ID:
173791bc56edSDimitry Andric         if (std::error_code EC = ParseAttributeBlock())
1738f785676fSDimitry Andric           return EC;
1739f22ef01cSRoman Divacky         break;
1740139f7f9bSDimitry Andric       case bitc::PARAMATTR_GROUP_BLOCK_ID:
174191bc56edSDimitry Andric         if (std::error_code EC = ParseAttributeGroupBlock())
1742f785676fSDimitry Andric           return EC;
1743139f7f9bSDimitry Andric         break;
174417a519f9SDimitry Andric       case bitc::TYPE_BLOCK_ID_NEW:
174591bc56edSDimitry Andric         if (std::error_code EC = ParseTypeTable())
1746f785676fSDimitry Andric           return EC;
1747f22ef01cSRoman Divacky         break;
1748f22ef01cSRoman Divacky       case bitc::VALUE_SYMTAB_BLOCK_ID:
174991bc56edSDimitry Andric         if (std::error_code EC = ParseValueSymbolTable())
1750f785676fSDimitry Andric           return EC;
1751dff0c46cSDimitry Andric         SeenValueSymbolTable = true;
1752f22ef01cSRoman Divacky         break;
1753f22ef01cSRoman Divacky       case bitc::CONSTANTS_BLOCK_ID:
175491bc56edSDimitry Andric         if (std::error_code EC = ParseConstants())
1755f785676fSDimitry Andric           return EC;
175691bc56edSDimitry Andric         if (std::error_code EC = ResolveGlobalAndAliasInits())
1757f785676fSDimitry Andric           return EC;
1758f22ef01cSRoman Divacky         break;
1759f22ef01cSRoman Divacky       case bitc::METADATA_BLOCK_ID:
176091bc56edSDimitry Andric         if (std::error_code EC = ParseMetadata())
1761f785676fSDimitry Andric           return EC;
1762f22ef01cSRoman Divacky         break;
1763f22ef01cSRoman Divacky       case bitc::FUNCTION_BLOCK_ID:
1764f22ef01cSRoman Divacky         // If this is the first function body we've seen, reverse the
1765f22ef01cSRoman Divacky         // FunctionsWithBodies list.
1766dff0c46cSDimitry Andric         if (!SeenFirstFunctionBody) {
1767f22ef01cSRoman Divacky           std::reverse(FunctionsWithBodies.begin(), FunctionsWithBodies.end());
176891bc56edSDimitry Andric           if (std::error_code EC = GlobalCleanup())
1769f785676fSDimitry Andric             return EC;
1770dff0c46cSDimitry Andric           SeenFirstFunctionBody = true;
1771f22ef01cSRoman Divacky         }
1772f22ef01cSRoman Divacky 
177391bc56edSDimitry Andric         if (std::error_code EC = RememberAndSkipFunctionBody())
1774f785676fSDimitry Andric           return EC;
1775dff0c46cSDimitry Andric         // For streaming bitcode, suspend parsing when we reach the function
1776dff0c46cSDimitry Andric         // bodies. Subsequent materialization calls will resume it when
1777dff0c46cSDimitry Andric         // necessary. For streaming, the function bodies must be at the end of
1778dff0c46cSDimitry Andric         // the bitcode. If the bitcode file is old, the symbol table will be
1779dff0c46cSDimitry Andric         // at the end instead and will not have been seen yet. In this case,
1780dff0c46cSDimitry Andric         // just finish the parse now.
1781dff0c46cSDimitry Andric         if (LazyStreamer && SeenValueSymbolTable) {
1782dff0c46cSDimitry Andric           NextUnreadBit = Stream.GetCurrentBitNo();
178391bc56edSDimitry Andric           return std::error_code();
1784dff0c46cSDimitry Andric         }
1785dff0c46cSDimitry Andric         break;
1786dff0c46cSDimitry Andric       case bitc::USELIST_BLOCK_ID:
178791bc56edSDimitry Andric         if (std::error_code EC = ParseUseLists())
1788f785676fSDimitry Andric           return EC;
1789f22ef01cSRoman Divacky         break;
1790f22ef01cSRoman Divacky       }
1791f22ef01cSRoman Divacky       continue;
1792139f7f9bSDimitry Andric 
1793139f7f9bSDimitry Andric     case BitstreamEntry::Record:
1794139f7f9bSDimitry Andric       // The interesting case.
1795139f7f9bSDimitry Andric       break;
1796f22ef01cSRoman Divacky     }
1797f22ef01cSRoman Divacky 
1798f22ef01cSRoman Divacky 
1799f22ef01cSRoman Divacky     // Read a record.
1800139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
1801f22ef01cSRoman Divacky     default: break;  // Default behavior, ignore unknown content.
18023861d79fSDimitry Andric     case bitc::MODULE_CODE_VERSION: {  // VERSION: [version#]
1803f22ef01cSRoman Divacky       if (Record.size() < 1)
1804f785676fSDimitry Andric         return Error(InvalidRecord);
18053861d79fSDimitry Andric       // Only version #0 and #1 are supported so far.
18063861d79fSDimitry Andric       unsigned module_version = Record[0];
18073861d79fSDimitry Andric       switch (module_version) {
1808f785676fSDimitry Andric         default:
1809f785676fSDimitry Andric           return Error(InvalidValue);
18103861d79fSDimitry Andric         case 0:
18113861d79fSDimitry Andric           UseRelativeIDs = false;
1812f22ef01cSRoman Divacky           break;
18133861d79fSDimitry Andric         case 1:
18143861d79fSDimitry Andric           UseRelativeIDs = true;
18153861d79fSDimitry Andric           break;
18163861d79fSDimitry Andric       }
18173861d79fSDimitry Andric       break;
18183861d79fSDimitry Andric     }
1819f22ef01cSRoman Divacky     case bitc::MODULE_CODE_TRIPLE: {  // TRIPLE: [strchr x N]
1820f22ef01cSRoman Divacky       std::string S;
1821f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1822f785676fSDimitry Andric         return Error(InvalidRecord);
1823f22ef01cSRoman Divacky       TheModule->setTargetTriple(S);
1824f22ef01cSRoman Divacky       break;
1825f22ef01cSRoman Divacky     }
1826f22ef01cSRoman Divacky     case bitc::MODULE_CODE_DATALAYOUT: {  // DATALAYOUT: [strchr x N]
1827f22ef01cSRoman Divacky       std::string S;
1828f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1829f785676fSDimitry Andric         return Error(InvalidRecord);
1830f22ef01cSRoman Divacky       TheModule->setDataLayout(S);
1831f22ef01cSRoman Divacky       break;
1832f22ef01cSRoman Divacky     }
1833f22ef01cSRoman Divacky     case bitc::MODULE_CODE_ASM: {  // ASM: [strchr x N]
1834f22ef01cSRoman Divacky       std::string S;
1835f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1836f785676fSDimitry Andric         return Error(InvalidRecord);
1837f22ef01cSRoman Divacky       TheModule->setModuleInlineAsm(S);
1838f22ef01cSRoman Divacky       break;
1839f22ef01cSRoman Divacky     }
1840f22ef01cSRoman Divacky     case bitc::MODULE_CODE_DEPLIB: {  // DEPLIB: [strchr x N]
1841139f7f9bSDimitry Andric       // FIXME: Remove in 4.0.
1842f22ef01cSRoman Divacky       std::string S;
1843f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1844f785676fSDimitry Andric         return Error(InvalidRecord);
1845139f7f9bSDimitry Andric       // Ignore value.
1846f22ef01cSRoman Divacky       break;
1847f22ef01cSRoman Divacky     }
1848f22ef01cSRoman Divacky     case bitc::MODULE_CODE_SECTIONNAME: {  // SECTIONNAME: [strchr x N]
1849f22ef01cSRoman Divacky       std::string S;
1850f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1851f785676fSDimitry Andric         return Error(InvalidRecord);
1852f22ef01cSRoman Divacky       SectionTable.push_back(S);
1853f22ef01cSRoman Divacky       break;
1854f22ef01cSRoman Divacky     }
1855f22ef01cSRoman Divacky     case bitc::MODULE_CODE_GCNAME: {  // SECTIONNAME: [strchr x N]
1856f22ef01cSRoman Divacky       std::string S;
1857f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1858f785676fSDimitry Andric         return Error(InvalidRecord);
1859f22ef01cSRoman Divacky       GCTable.push_back(S);
1860f22ef01cSRoman Divacky       break;
1861f22ef01cSRoman Divacky     }
186291bc56edSDimitry Andric     case bitc::MODULE_CODE_COMDAT: { // COMDAT: [selection_kind, name]
186391bc56edSDimitry Andric       if (Record.size() < 2)
186491bc56edSDimitry Andric         return Error(InvalidRecord);
186591bc56edSDimitry Andric       Comdat::SelectionKind SK = getDecodedComdatSelectionKind(Record[0]);
186691bc56edSDimitry Andric       unsigned ComdatNameSize = Record[1];
186791bc56edSDimitry Andric       std::string ComdatName;
186891bc56edSDimitry Andric       ComdatName.reserve(ComdatNameSize);
186991bc56edSDimitry Andric       for (unsigned i = 0; i != ComdatNameSize; ++i)
187091bc56edSDimitry Andric         ComdatName += (char)Record[2 + i];
187191bc56edSDimitry Andric       Comdat *C = TheModule->getOrInsertComdat(ComdatName);
187291bc56edSDimitry Andric       C->setSelectionKind(SK);
187391bc56edSDimitry Andric       ComdatList.push_back(C);
187491bc56edSDimitry Andric       break;
187591bc56edSDimitry Andric     }
1876f22ef01cSRoman Divacky     // GLOBALVAR: [pointer type, isconst, initid,
18772754fe60SDimitry Andric     //             linkage, alignment, section, visibility, threadlocal,
187891bc56edSDimitry Andric     //             unnamed_addr, dllstorageclass]
1879f22ef01cSRoman Divacky     case bitc::MODULE_CODE_GLOBALVAR: {
1880f22ef01cSRoman Divacky       if (Record.size() < 6)
1881f785676fSDimitry Andric         return Error(InvalidRecord);
18826122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
1883f785676fSDimitry Andric       if (!Ty)
1884f785676fSDimitry Andric         return Error(InvalidRecord);
1885f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
1886f785676fSDimitry Andric         return Error(InvalidTypeForValue);
1887f22ef01cSRoman Divacky       unsigned AddressSpace = cast<PointerType>(Ty)->getAddressSpace();
1888f22ef01cSRoman Divacky       Ty = cast<PointerType>(Ty)->getElementType();
1889f22ef01cSRoman Divacky 
1890f22ef01cSRoman Divacky       bool isConstant = Record[1];
1891f22ef01cSRoman Divacky       GlobalValue::LinkageTypes Linkage = GetDecodedLinkage(Record[3]);
1892f22ef01cSRoman Divacky       unsigned Alignment = (1 << Record[4]) >> 1;
1893f22ef01cSRoman Divacky       std::string Section;
1894f22ef01cSRoman Divacky       if (Record[5]) {
1895f22ef01cSRoman Divacky         if (Record[5]-1 >= SectionTable.size())
1896f785676fSDimitry Andric           return Error(InvalidID);
1897f22ef01cSRoman Divacky         Section = SectionTable[Record[5]-1];
1898f22ef01cSRoman Divacky       }
1899f22ef01cSRoman Divacky       GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
190091bc56edSDimitry Andric       // Local linkage must have default visibility.
190191bc56edSDimitry Andric       if (Record.size() > 6 && !GlobalValue::isLocalLinkage(Linkage))
190291bc56edSDimitry Andric         // FIXME: Change to an error if non-default in 4.0.
1903f22ef01cSRoman Divacky         Visibility = GetDecodedVisibility(Record[6]);
19047ae0e2c9SDimitry Andric 
19057ae0e2c9SDimitry Andric       GlobalVariable::ThreadLocalMode TLM = GlobalVariable::NotThreadLocal;
1906f22ef01cSRoman Divacky       if (Record.size() > 7)
19077ae0e2c9SDimitry Andric         TLM = GetDecodedThreadLocalMode(Record[7]);
1908f22ef01cSRoman Divacky 
19092754fe60SDimitry Andric       bool UnnamedAddr = false;
19102754fe60SDimitry Andric       if (Record.size() > 8)
19112754fe60SDimitry Andric         UnnamedAddr = Record[8];
19122754fe60SDimitry Andric 
1913139f7f9bSDimitry Andric       bool ExternallyInitialized = false;
1914139f7f9bSDimitry Andric       if (Record.size() > 9)
1915139f7f9bSDimitry Andric         ExternallyInitialized = Record[9];
1916139f7f9bSDimitry Andric 
1917f22ef01cSRoman Divacky       GlobalVariable *NewGV =
191891bc56edSDimitry Andric         new GlobalVariable(*TheModule, Ty, isConstant, Linkage, nullptr, "", nullptr,
1919139f7f9bSDimitry Andric                            TLM, AddressSpace, ExternallyInitialized);
1920f22ef01cSRoman Divacky       NewGV->setAlignment(Alignment);
1921f22ef01cSRoman Divacky       if (!Section.empty())
1922f22ef01cSRoman Divacky         NewGV->setSection(Section);
1923f22ef01cSRoman Divacky       NewGV->setVisibility(Visibility);
19242754fe60SDimitry Andric       NewGV->setUnnamedAddr(UnnamedAddr);
1925f22ef01cSRoman Divacky 
192691bc56edSDimitry Andric       if (Record.size() > 10)
192791bc56edSDimitry Andric         NewGV->setDLLStorageClass(GetDecodedDLLStorageClass(Record[10]));
192891bc56edSDimitry Andric       else
192991bc56edSDimitry Andric         UpgradeDLLImportExportLinkage(NewGV, Record[3]);
193091bc56edSDimitry Andric 
1931f22ef01cSRoman Divacky       ValueList.push_back(NewGV);
1932f22ef01cSRoman Divacky 
1933f22ef01cSRoman Divacky       // Remember which value to use for the global initializer.
1934f22ef01cSRoman Divacky       if (unsigned InitID = Record[2])
1935f22ef01cSRoman Divacky         GlobalInits.push_back(std::make_pair(NewGV, InitID-1));
193691bc56edSDimitry Andric 
193791bc56edSDimitry Andric       if (Record.size() > 11)
193891bc56edSDimitry Andric         if (unsigned ComdatID = Record[11]) {
193991bc56edSDimitry Andric           assert(ComdatID <= ComdatList.size());
194091bc56edSDimitry Andric           NewGV->setComdat(ComdatList[ComdatID - 1]);
194191bc56edSDimitry Andric         }
1942f22ef01cSRoman Divacky       break;
1943f22ef01cSRoman Divacky     }
1944f22ef01cSRoman Divacky     // FUNCTION:  [type, callingconv, isproto, linkage, paramattr,
194591bc56edSDimitry Andric     //             alignment, section, visibility, gc, unnamed_addr,
194691bc56edSDimitry Andric     //             dllstorageclass]
1947f22ef01cSRoman Divacky     case bitc::MODULE_CODE_FUNCTION: {
1948f22ef01cSRoman Divacky       if (Record.size() < 8)
1949f785676fSDimitry Andric         return Error(InvalidRecord);
19506122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
1951f785676fSDimitry Andric       if (!Ty)
1952f785676fSDimitry Andric         return Error(InvalidRecord);
1953f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
1954f785676fSDimitry Andric         return Error(InvalidTypeForValue);
19556122f3e6SDimitry Andric       FunctionType *FTy =
1956f22ef01cSRoman Divacky         dyn_cast<FunctionType>(cast<PointerType>(Ty)->getElementType());
1957f22ef01cSRoman Divacky       if (!FTy)
1958f785676fSDimitry Andric         return Error(InvalidTypeForValue);
1959f22ef01cSRoman Divacky 
1960f22ef01cSRoman Divacky       Function *Func = Function::Create(FTy, GlobalValue::ExternalLinkage,
1961f22ef01cSRoman Divacky                                         "", TheModule);
1962f22ef01cSRoman Divacky 
1963f22ef01cSRoman Divacky       Func->setCallingConv(static_cast<CallingConv::ID>(Record[1]));
1964f22ef01cSRoman Divacky       bool isProto = Record[2];
1965f22ef01cSRoman Divacky       Func->setLinkage(GetDecodedLinkage(Record[3]));
1966f22ef01cSRoman Divacky       Func->setAttributes(getAttributes(Record[4]));
1967f22ef01cSRoman Divacky 
1968f22ef01cSRoman Divacky       Func->setAlignment((1 << Record[5]) >> 1);
1969f22ef01cSRoman Divacky       if (Record[6]) {
1970f22ef01cSRoman Divacky         if (Record[6]-1 >= SectionTable.size())
1971f785676fSDimitry Andric           return Error(InvalidID);
1972f22ef01cSRoman Divacky         Func->setSection(SectionTable[Record[6]-1]);
1973f22ef01cSRoman Divacky       }
197491bc56edSDimitry Andric       // Local linkage must have default visibility.
197591bc56edSDimitry Andric       if (!Func->hasLocalLinkage())
197691bc56edSDimitry Andric         // FIXME: Change to an error if non-default in 4.0.
1977f22ef01cSRoman Divacky         Func->setVisibility(GetDecodedVisibility(Record[7]));
1978f22ef01cSRoman Divacky       if (Record.size() > 8 && Record[8]) {
1979f22ef01cSRoman Divacky         if (Record[8]-1 > GCTable.size())
1980f785676fSDimitry Andric           return Error(InvalidID);
1981f22ef01cSRoman Divacky         Func->setGC(GCTable[Record[8]-1].c_str());
1982f22ef01cSRoman Divacky       }
19832754fe60SDimitry Andric       bool UnnamedAddr = false;
19842754fe60SDimitry Andric       if (Record.size() > 9)
19852754fe60SDimitry Andric         UnnamedAddr = Record[9];
19862754fe60SDimitry Andric       Func->setUnnamedAddr(UnnamedAddr);
1987f785676fSDimitry Andric       if (Record.size() > 10 && Record[10] != 0)
1988f785676fSDimitry Andric         FunctionPrefixes.push_back(std::make_pair(Func, Record[10]-1));
198991bc56edSDimitry Andric 
199091bc56edSDimitry Andric       if (Record.size() > 11)
199191bc56edSDimitry Andric         Func->setDLLStorageClass(GetDecodedDLLStorageClass(Record[11]));
199291bc56edSDimitry Andric       else
199391bc56edSDimitry Andric         UpgradeDLLImportExportLinkage(Func, Record[3]);
199491bc56edSDimitry Andric 
199591bc56edSDimitry Andric       if (Record.size() > 12)
199691bc56edSDimitry Andric         if (unsigned ComdatID = Record[12]) {
199791bc56edSDimitry Andric           assert(ComdatID <= ComdatList.size());
199891bc56edSDimitry Andric           Func->setComdat(ComdatList[ComdatID - 1]);
199991bc56edSDimitry Andric         }
200091bc56edSDimitry Andric 
2001f22ef01cSRoman Divacky       ValueList.push_back(Func);
2002f22ef01cSRoman Divacky 
2003f22ef01cSRoman Divacky       // If this is a function with a body, remember the prototype we are
2004f22ef01cSRoman Divacky       // creating now, so that we can match up the body with them later.
2005dff0c46cSDimitry Andric       if (!isProto) {
2006f22ef01cSRoman Divacky         FunctionsWithBodies.push_back(Func);
2007dff0c46cSDimitry Andric         if (LazyStreamer) DeferredFunctionInfo[Func] = 0;
2008dff0c46cSDimitry Andric       }
2009f22ef01cSRoman Divacky       break;
2010f22ef01cSRoman Divacky     }
2011f22ef01cSRoman Divacky     // ALIAS: [alias type, aliasee val#, linkage]
201291bc56edSDimitry Andric     // ALIAS: [alias type, aliasee val#, linkage, visibility, dllstorageclass]
2013f22ef01cSRoman Divacky     case bitc::MODULE_CODE_ALIAS: {
2014f22ef01cSRoman Divacky       if (Record.size() < 3)
2015f785676fSDimitry Andric         return Error(InvalidRecord);
20166122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
2017f785676fSDimitry Andric       if (!Ty)
2018f785676fSDimitry Andric         return Error(InvalidRecord);
201991bc56edSDimitry Andric       auto *PTy = dyn_cast<PointerType>(Ty);
202091bc56edSDimitry Andric       if (!PTy)
2021f785676fSDimitry Andric         return Error(InvalidTypeForValue);
2022f22ef01cSRoman Divacky 
202391bc56edSDimitry Andric       auto *NewGA =
202491bc56edSDimitry Andric           GlobalAlias::create(PTy->getElementType(), PTy->getAddressSpace(),
202591bc56edSDimitry Andric                               GetDecodedLinkage(Record[2]), "", TheModule);
2026f22ef01cSRoman Divacky       // Old bitcode files didn't have visibility field.
202791bc56edSDimitry Andric       // Local linkage must have default visibility.
202891bc56edSDimitry Andric       if (Record.size() > 3 && !NewGA->hasLocalLinkage())
202991bc56edSDimitry Andric         // FIXME: Change to an error if non-default in 4.0.
2030f22ef01cSRoman Divacky         NewGA->setVisibility(GetDecodedVisibility(Record[3]));
203191bc56edSDimitry Andric       if (Record.size() > 4)
203291bc56edSDimitry Andric         NewGA->setDLLStorageClass(GetDecodedDLLStorageClass(Record[4]));
203391bc56edSDimitry Andric       else
203491bc56edSDimitry Andric         UpgradeDLLImportExportLinkage(NewGA, Record[2]);
203591bc56edSDimitry Andric       if (Record.size() > 5)
203691bc56edSDimitry Andric 	NewGA->setThreadLocalMode(GetDecodedThreadLocalMode(Record[5]));
203791bc56edSDimitry Andric       if (Record.size() > 6)
203891bc56edSDimitry Andric 	NewGA->setUnnamedAddr(Record[6]);
2039f22ef01cSRoman Divacky       ValueList.push_back(NewGA);
2040f22ef01cSRoman Divacky       AliasInits.push_back(std::make_pair(NewGA, Record[1]));
2041f22ef01cSRoman Divacky       break;
2042f22ef01cSRoman Divacky     }
2043f22ef01cSRoman Divacky     /// MODULE_CODE_PURGEVALS: [numvals]
2044f22ef01cSRoman Divacky     case bitc::MODULE_CODE_PURGEVALS:
2045f22ef01cSRoman Divacky       // Trim down the value list to the specified size.
2046f22ef01cSRoman Divacky       if (Record.size() < 1 || Record[0] > ValueList.size())
2047f785676fSDimitry Andric         return Error(InvalidRecord);
2048f22ef01cSRoman Divacky       ValueList.shrinkTo(Record[0]);
2049f22ef01cSRoman Divacky       break;
2050f22ef01cSRoman Divacky     }
2051f22ef01cSRoman Divacky     Record.clear();
2052f22ef01cSRoman Divacky   }
2053f22ef01cSRoman Divacky }
2054f22ef01cSRoman Divacky 
205591bc56edSDimitry Andric std::error_code BitcodeReader::ParseBitcodeInto(Module *M) {
205691bc56edSDimitry Andric   TheModule = nullptr;
2057f22ef01cSRoman Divacky 
205891bc56edSDimitry Andric   if (std::error_code EC = InitStream())
2059f785676fSDimitry Andric     return EC;
2060f22ef01cSRoman Divacky 
2061f22ef01cSRoman Divacky   // Sniff for the signature.
2062f22ef01cSRoman Divacky   if (Stream.Read(8) != 'B' ||
2063f22ef01cSRoman Divacky       Stream.Read(8) != 'C' ||
2064f22ef01cSRoman Divacky       Stream.Read(4) != 0x0 ||
2065f22ef01cSRoman Divacky       Stream.Read(4) != 0xC ||
2066f22ef01cSRoman Divacky       Stream.Read(4) != 0xE ||
2067f22ef01cSRoman Divacky       Stream.Read(4) != 0xD)
2068f785676fSDimitry Andric     return Error(InvalidBitcodeSignature);
2069f22ef01cSRoman Divacky 
2070f22ef01cSRoman Divacky   // We expect a number of well-defined blocks, though we don't necessarily
2071f22ef01cSRoman Divacky   // need to understand them all.
2072139f7f9bSDimitry Andric   while (1) {
2073139f7f9bSDimitry Andric     if (Stream.AtEndOfStream())
207491bc56edSDimitry Andric       return std::error_code();
2075bd5abe19SDimitry Andric 
2076139f7f9bSDimitry Andric     BitstreamEntry Entry =
2077139f7f9bSDimitry Andric       Stream.advance(BitstreamCursor::AF_DontAutoprocessAbbrevs);
2078f22ef01cSRoman Divacky 
2079139f7f9bSDimitry Andric     switch (Entry.Kind) {
2080139f7f9bSDimitry Andric     case BitstreamEntry::Error:
2081f785676fSDimitry Andric       return Error(MalformedBlock);
2082139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
208391bc56edSDimitry Andric       return std::error_code();
2084f22ef01cSRoman Divacky 
2085139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock:
2086139f7f9bSDimitry Andric       switch (Entry.ID) {
2087f22ef01cSRoman Divacky       case bitc::BLOCKINFO_BLOCK_ID:
2088f22ef01cSRoman Divacky         if (Stream.ReadBlockInfoBlock())
2089f785676fSDimitry Andric           return Error(MalformedBlock);
2090f22ef01cSRoman Divacky         break;
2091f22ef01cSRoman Divacky       case bitc::MODULE_BLOCK_ID:
2092f22ef01cSRoman Divacky         // Reject multiple MODULE_BLOCK's in a single bitstream.
2093f22ef01cSRoman Divacky         if (TheModule)
2094f785676fSDimitry Andric           return Error(InvalidMultipleBlocks);
2095f22ef01cSRoman Divacky         TheModule = M;
209691bc56edSDimitry Andric         if (std::error_code EC = ParseModule(false))
2097f785676fSDimitry Andric           return EC;
2098f785676fSDimitry Andric         if (LazyStreamer)
209991bc56edSDimitry Andric           return std::error_code();
2100f22ef01cSRoman Divacky         break;
2101f22ef01cSRoman Divacky       default:
2102f22ef01cSRoman Divacky         if (Stream.SkipBlock())
2103f785676fSDimitry Andric           return Error(InvalidRecord);
2104f22ef01cSRoman Divacky         break;
2105f22ef01cSRoman Divacky       }
2106139f7f9bSDimitry Andric       continue;
2107139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2108139f7f9bSDimitry Andric       // There should be no records in the top-level of blocks.
2109f22ef01cSRoman Divacky 
2110139f7f9bSDimitry Andric       // The ranlib in Xcode 4 will align archive members by appending newlines
2111139f7f9bSDimitry Andric       // to the end of them. If this file size is a multiple of 4 but not 8, we
2112139f7f9bSDimitry Andric       // have to read and ignore these final 4 bytes :-(
2113139f7f9bSDimitry Andric       if (Stream.getAbbrevIDWidth() == 2 && Entry.ID == 2 &&
2114139f7f9bSDimitry Andric           Stream.Read(6) == 2 && Stream.Read(24) == 0xa0a0a &&
2115139f7f9bSDimitry Andric           Stream.AtEndOfStream())
211691bc56edSDimitry Andric         return std::error_code();
2117139f7f9bSDimitry Andric 
2118f785676fSDimitry Andric       return Error(InvalidRecord);
2119139f7f9bSDimitry Andric     }
2120139f7f9bSDimitry Andric   }
2121f22ef01cSRoman Divacky }
2122f22ef01cSRoman Divacky 
212391bc56edSDimitry Andric ErrorOr<std::string> BitcodeReader::parseModuleTriple() {
21242754fe60SDimitry Andric   if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
2125f785676fSDimitry Andric     return Error(InvalidRecord);
21262754fe60SDimitry Andric 
21272754fe60SDimitry Andric   SmallVector<uint64_t, 64> Record;
21282754fe60SDimitry Andric 
212991bc56edSDimitry Andric   std::string Triple;
21302754fe60SDimitry Andric   // Read all the records for this module.
2131139f7f9bSDimitry Andric   while (1) {
2132139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
21332754fe60SDimitry Andric 
2134139f7f9bSDimitry Andric     switch (Entry.Kind) {
2135139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
2136139f7f9bSDimitry Andric     case BitstreamEntry::Error:
2137f785676fSDimitry Andric       return Error(MalformedBlock);
2138139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
213991bc56edSDimitry Andric       return Triple;
2140139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2141139f7f9bSDimitry Andric       // The interesting case.
21422754fe60SDimitry Andric       break;
21432754fe60SDimitry Andric     }
21442754fe60SDimitry Andric 
21452754fe60SDimitry Andric     // Read a record.
2146139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
21472754fe60SDimitry Andric     default: break;  // Default behavior, ignore unknown content.
21482754fe60SDimitry Andric     case bitc::MODULE_CODE_TRIPLE: {  // TRIPLE: [strchr x N]
21492754fe60SDimitry Andric       std::string S;
21502754fe60SDimitry Andric       if (ConvertToString(Record, 0, S))
2151f785676fSDimitry Andric         return Error(InvalidRecord);
21522754fe60SDimitry Andric       Triple = S;
21532754fe60SDimitry Andric       break;
21542754fe60SDimitry Andric     }
21552754fe60SDimitry Andric     }
21562754fe60SDimitry Andric     Record.clear();
21572754fe60SDimitry Andric   }
215891bc56edSDimitry Andric   llvm_unreachable("Exit infinite loop");
21592754fe60SDimitry Andric }
21602754fe60SDimitry Andric 
216191bc56edSDimitry Andric ErrorOr<std::string> BitcodeReader::parseTriple() {
216291bc56edSDimitry Andric   if (std::error_code EC = InitStream())
2163f785676fSDimitry Andric     return EC;
21642754fe60SDimitry Andric 
21652754fe60SDimitry Andric   // Sniff for the signature.
21662754fe60SDimitry Andric   if (Stream.Read(8) != 'B' ||
21672754fe60SDimitry Andric       Stream.Read(8) != 'C' ||
21682754fe60SDimitry Andric       Stream.Read(4) != 0x0 ||
21692754fe60SDimitry Andric       Stream.Read(4) != 0xC ||
21702754fe60SDimitry Andric       Stream.Read(4) != 0xE ||
21712754fe60SDimitry Andric       Stream.Read(4) != 0xD)
2172f785676fSDimitry Andric     return Error(InvalidBitcodeSignature);
21732754fe60SDimitry Andric 
21742754fe60SDimitry Andric   // We expect a number of well-defined blocks, though we don't necessarily
21752754fe60SDimitry Andric   // need to understand them all.
2176139f7f9bSDimitry Andric   while (1) {
2177139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advance();
21782754fe60SDimitry Andric 
2179139f7f9bSDimitry Andric     switch (Entry.Kind) {
2180139f7f9bSDimitry Andric     case BitstreamEntry::Error:
2181f785676fSDimitry Andric       return Error(MalformedBlock);
2182139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
218391bc56edSDimitry Andric       return std::error_code();
2184139f7f9bSDimitry Andric 
2185139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock:
2186139f7f9bSDimitry Andric       if (Entry.ID == bitc::MODULE_BLOCK_ID)
218791bc56edSDimitry Andric         return parseModuleTriple();
2188139f7f9bSDimitry Andric 
2189139f7f9bSDimitry Andric       // Ignore other sub-blocks.
2190f785676fSDimitry Andric       if (Stream.SkipBlock())
2191f785676fSDimitry Andric         return Error(MalformedBlock);
2192139f7f9bSDimitry Andric       continue;
2193139f7f9bSDimitry Andric 
2194139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2195139f7f9bSDimitry Andric       Stream.skipRecord(Entry.ID);
2196139f7f9bSDimitry Andric       continue;
2197139f7f9bSDimitry Andric     }
2198139f7f9bSDimitry Andric   }
21992754fe60SDimitry Andric }
22002754fe60SDimitry Andric 
2201f22ef01cSRoman Divacky /// ParseMetadataAttachment - Parse metadata attachments.
220291bc56edSDimitry Andric std::error_code BitcodeReader::ParseMetadataAttachment() {
2203f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::METADATA_ATTACHMENT_ID))
2204f785676fSDimitry Andric     return Error(InvalidRecord);
2205f22ef01cSRoman Divacky 
2206f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
2207f22ef01cSRoman Divacky   while (1) {
2208139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advanceSkippingSubblocks();
2209139f7f9bSDimitry Andric 
2210139f7f9bSDimitry Andric     switch (Entry.Kind) {
2211139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock: // Handled for us already.
2212139f7f9bSDimitry Andric     case BitstreamEntry::Error:
2213f785676fSDimitry Andric       return Error(MalformedBlock);
2214139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
221591bc56edSDimitry Andric       return std::error_code();
2216139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2217139f7f9bSDimitry Andric       // The interesting case.
2218f22ef01cSRoman Divacky       break;
2219f22ef01cSRoman Divacky     }
2220139f7f9bSDimitry Andric 
2221f22ef01cSRoman Divacky     // Read a metadata attachment record.
2222f22ef01cSRoman Divacky     Record.clear();
2223139f7f9bSDimitry Andric     switch (Stream.readRecord(Entry.ID, Record)) {
2224f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
2225f22ef01cSRoman Divacky       break;
222617a519f9SDimitry Andric     case bitc::METADATA_ATTACHMENT: {
2227f22ef01cSRoman Divacky       unsigned RecordLength = Record.size();
2228f22ef01cSRoman Divacky       if (Record.empty() || (RecordLength - 1) % 2 == 1)
2229f785676fSDimitry Andric         return Error(InvalidRecord);
2230f22ef01cSRoman Divacky       Instruction *Inst = InstructionList[Record[0]];
2231f22ef01cSRoman Divacky       for (unsigned i = 1; i != RecordLength; i = i+2) {
2232f22ef01cSRoman Divacky         unsigned Kind = Record[i];
2233e580952dSDimitry Andric         DenseMap<unsigned, unsigned>::iterator I =
2234e580952dSDimitry Andric           MDKindMap.find(Kind);
2235e580952dSDimitry Andric         if (I == MDKindMap.end())
2236f785676fSDimitry Andric           return Error(InvalidID);
2237f22ef01cSRoman Divacky         Value *Node = MDValueList.getValueFwdRef(Record[i+1]);
2238e580952dSDimitry Andric         Inst->setMetadata(I->second, cast<MDNode>(Node));
2239f785676fSDimitry Andric         if (I->second == LLVMContext::MD_tbaa)
2240f785676fSDimitry Andric           InstsWithTBAATag.push_back(Inst);
2241f22ef01cSRoman Divacky       }
2242f22ef01cSRoman Divacky       break;
2243f22ef01cSRoman Divacky     }
2244f22ef01cSRoman Divacky     }
2245f22ef01cSRoman Divacky   }
2246f22ef01cSRoman Divacky }
2247f22ef01cSRoman Divacky 
2248f22ef01cSRoman Divacky /// ParseFunctionBody - Lazily parse the specified function body block.
224991bc56edSDimitry Andric std::error_code BitcodeReader::ParseFunctionBody(Function *F) {
2250f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::FUNCTION_BLOCK_ID))
2251f785676fSDimitry Andric     return Error(InvalidRecord);
2252f22ef01cSRoman Divacky 
2253f22ef01cSRoman Divacky   InstructionList.clear();
2254f22ef01cSRoman Divacky   unsigned ModuleValueListSize = ValueList.size();
2255e580952dSDimitry Andric   unsigned ModuleMDValueListSize = MDValueList.size();
2256f22ef01cSRoman Divacky 
2257f22ef01cSRoman Divacky   // Add all the function arguments to the value table.
2258f22ef01cSRoman Divacky   for(Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
2259f22ef01cSRoman Divacky     ValueList.push_back(I);
2260f22ef01cSRoman Divacky 
2261f22ef01cSRoman Divacky   unsigned NextValueNo = ValueList.size();
226291bc56edSDimitry Andric   BasicBlock *CurBB = nullptr;
2263f22ef01cSRoman Divacky   unsigned CurBBNo = 0;
2264f22ef01cSRoman Divacky 
2265f22ef01cSRoman Divacky   DebugLoc LastLoc;
2266f22ef01cSRoman Divacky 
2267f22ef01cSRoman Divacky   // Read all the records.
2268f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
2269f22ef01cSRoman Divacky   while (1) {
2270139f7f9bSDimitry Andric     BitstreamEntry Entry = Stream.advance();
2271f22ef01cSRoman Divacky 
2272139f7f9bSDimitry Andric     switch (Entry.Kind) {
2273139f7f9bSDimitry Andric     case BitstreamEntry::Error:
2274f785676fSDimitry Andric       return Error(MalformedBlock);
2275139f7f9bSDimitry Andric     case BitstreamEntry::EndBlock:
2276139f7f9bSDimitry Andric       goto OutOfRecordLoop;
2277139f7f9bSDimitry Andric 
2278139f7f9bSDimitry Andric     case BitstreamEntry::SubBlock:
2279139f7f9bSDimitry Andric       switch (Entry.ID) {
2280f22ef01cSRoman Divacky       default:  // Skip unknown content.
2281f22ef01cSRoman Divacky         if (Stream.SkipBlock())
2282f785676fSDimitry Andric           return Error(InvalidRecord);
2283f22ef01cSRoman Divacky         break;
2284f22ef01cSRoman Divacky       case bitc::CONSTANTS_BLOCK_ID:
228591bc56edSDimitry Andric         if (std::error_code EC = ParseConstants())
2286f785676fSDimitry Andric           return EC;
2287f22ef01cSRoman Divacky         NextValueNo = ValueList.size();
2288f22ef01cSRoman Divacky         break;
2289f22ef01cSRoman Divacky       case bitc::VALUE_SYMTAB_BLOCK_ID:
229091bc56edSDimitry Andric         if (std::error_code EC = ParseValueSymbolTable())
2291f785676fSDimitry Andric           return EC;
2292f22ef01cSRoman Divacky         break;
2293f22ef01cSRoman Divacky       case bitc::METADATA_ATTACHMENT_ID:
229491bc56edSDimitry Andric         if (std::error_code EC = ParseMetadataAttachment())
2295f785676fSDimitry Andric           return EC;
2296f22ef01cSRoman Divacky         break;
2297f22ef01cSRoman Divacky       case bitc::METADATA_BLOCK_ID:
229891bc56edSDimitry Andric         if (std::error_code EC = ParseMetadata())
2299f785676fSDimitry Andric           return EC;
2300f22ef01cSRoman Divacky         break;
2301f22ef01cSRoman Divacky       }
2302f22ef01cSRoman Divacky       continue;
2303f22ef01cSRoman Divacky 
2304139f7f9bSDimitry Andric     case BitstreamEntry::Record:
2305139f7f9bSDimitry Andric       // The interesting case.
2306139f7f9bSDimitry Andric       break;
2307f22ef01cSRoman Divacky     }
2308f22ef01cSRoman Divacky 
2309f22ef01cSRoman Divacky     // Read a record.
2310f22ef01cSRoman Divacky     Record.clear();
231191bc56edSDimitry Andric     Instruction *I = nullptr;
2312139f7f9bSDimitry Andric     unsigned BitCode = Stream.readRecord(Entry.ID, Record);
2313f22ef01cSRoman Divacky     switch (BitCode) {
2314f22ef01cSRoman Divacky     default: // Default behavior: reject
2315f785676fSDimitry Andric       return Error(InvalidValue);
2316f22ef01cSRoman Divacky     case bitc::FUNC_CODE_DECLAREBLOCKS:     // DECLAREBLOCKS: [nblocks]
2317f22ef01cSRoman Divacky       if (Record.size() < 1 || Record[0] == 0)
2318f785676fSDimitry Andric         return Error(InvalidRecord);
2319f22ef01cSRoman Divacky       // Create all the basic blocks for the function.
2320f22ef01cSRoman Divacky       FunctionBBs.resize(Record[0]);
2321f22ef01cSRoman Divacky       for (unsigned i = 0, e = FunctionBBs.size(); i != e; ++i)
2322f22ef01cSRoman Divacky         FunctionBBs[i] = BasicBlock::Create(Context, "", F);
2323f22ef01cSRoman Divacky       CurBB = FunctionBBs[0];
2324f22ef01cSRoman Divacky       continue;
2325f22ef01cSRoman Divacky 
2326f22ef01cSRoman Divacky     case bitc::FUNC_CODE_DEBUG_LOC_AGAIN:  // DEBUG_LOC_AGAIN
2327f22ef01cSRoman Divacky       // This record indicates that the last instruction is at the same
2328f22ef01cSRoman Divacky       // location as the previous instruction with a location.
232991bc56edSDimitry Andric       I = nullptr;
2330f22ef01cSRoman Divacky 
2331f22ef01cSRoman Divacky       // Get the last instruction emitted.
2332f22ef01cSRoman Divacky       if (CurBB && !CurBB->empty())
2333f22ef01cSRoman Divacky         I = &CurBB->back();
2334f22ef01cSRoman Divacky       else if (CurBBNo && FunctionBBs[CurBBNo-1] &&
2335f22ef01cSRoman Divacky                !FunctionBBs[CurBBNo-1]->empty())
2336f22ef01cSRoman Divacky         I = &FunctionBBs[CurBBNo-1]->back();
2337f22ef01cSRoman Divacky 
233891bc56edSDimitry Andric       if (!I)
2339f785676fSDimitry Andric         return Error(InvalidRecord);
2340f22ef01cSRoman Divacky       I->setDebugLoc(LastLoc);
234191bc56edSDimitry Andric       I = nullptr;
2342f22ef01cSRoman Divacky       continue;
2343f22ef01cSRoman Divacky 
234417a519f9SDimitry Andric     case bitc::FUNC_CODE_DEBUG_LOC: {      // DEBUG_LOC: [line, col, scope, ia]
234591bc56edSDimitry Andric       I = nullptr;     // Get the last instruction emitted.
2346f22ef01cSRoman Divacky       if (CurBB && !CurBB->empty())
2347f22ef01cSRoman Divacky         I = &CurBB->back();
2348f22ef01cSRoman Divacky       else if (CurBBNo && FunctionBBs[CurBBNo-1] &&
2349f22ef01cSRoman Divacky                !FunctionBBs[CurBBNo-1]->empty())
2350f22ef01cSRoman Divacky         I = &FunctionBBs[CurBBNo-1]->back();
235191bc56edSDimitry Andric       if (!I || Record.size() < 4)
2352f785676fSDimitry Andric         return Error(InvalidRecord);
2353f22ef01cSRoman Divacky 
2354f22ef01cSRoman Divacky       unsigned Line = Record[0], Col = Record[1];
2355f22ef01cSRoman Divacky       unsigned ScopeID = Record[2], IAID = Record[3];
2356f22ef01cSRoman Divacky 
235791bc56edSDimitry Andric       MDNode *Scope = nullptr, *IA = nullptr;
2358f22ef01cSRoman Divacky       if (ScopeID) Scope = cast<MDNode>(MDValueList.getValueFwdRef(ScopeID-1));
2359f22ef01cSRoman Divacky       if (IAID)    IA = cast<MDNode>(MDValueList.getValueFwdRef(IAID-1));
2360f22ef01cSRoman Divacky       LastLoc = DebugLoc::get(Line, Col, Scope, IA);
2361f22ef01cSRoman Divacky       I->setDebugLoc(LastLoc);
236291bc56edSDimitry Andric       I = nullptr;
2363f22ef01cSRoman Divacky       continue;
2364f22ef01cSRoman Divacky     }
2365f22ef01cSRoman Divacky 
2366f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_BINOP: {    // BINOP: [opval, ty, opval, opcode]
2367f22ef01cSRoman Divacky       unsigned OpNum = 0;
2368f22ef01cSRoman Divacky       Value *LHS, *RHS;
2369f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
23703861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, LHS->getType(), RHS) ||
2371f22ef01cSRoman Divacky           OpNum+1 > Record.size())
2372f785676fSDimitry Andric         return Error(InvalidRecord);
2373f22ef01cSRoman Divacky 
2374f22ef01cSRoman Divacky       int Opc = GetDecodedBinaryOpcode(Record[OpNum++], LHS->getType());
2375f785676fSDimitry Andric       if (Opc == -1)
2376f785676fSDimitry Andric         return Error(InvalidRecord);
2377f22ef01cSRoman Divacky       I = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
2378f22ef01cSRoman Divacky       InstructionList.push_back(I);
2379f22ef01cSRoman Divacky       if (OpNum < Record.size()) {
2380f22ef01cSRoman Divacky         if (Opc == Instruction::Add ||
2381f22ef01cSRoman Divacky             Opc == Instruction::Sub ||
23822754fe60SDimitry Andric             Opc == Instruction::Mul ||
23832754fe60SDimitry Andric             Opc == Instruction::Shl) {
2384f22ef01cSRoman Divacky           if (Record[OpNum] & (1 << bitc::OBO_NO_SIGNED_WRAP))
2385f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setHasNoSignedWrap(true);
2386f22ef01cSRoman Divacky           if (Record[OpNum] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
2387f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setHasNoUnsignedWrap(true);
23882754fe60SDimitry Andric         } else if (Opc == Instruction::SDiv ||
23892754fe60SDimitry Andric                    Opc == Instruction::UDiv ||
23902754fe60SDimitry Andric                    Opc == Instruction::LShr ||
23912754fe60SDimitry Andric                    Opc == Instruction::AShr) {
23922754fe60SDimitry Andric           if (Record[OpNum] & (1 << bitc::PEO_EXACT))
2393f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setIsExact(true);
2394139f7f9bSDimitry Andric         } else if (isa<FPMathOperator>(I)) {
2395139f7f9bSDimitry Andric           FastMathFlags FMF;
2396139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::UnsafeAlgebra))
2397139f7f9bSDimitry Andric             FMF.setUnsafeAlgebra();
2398139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::NoNaNs))
2399139f7f9bSDimitry Andric             FMF.setNoNaNs();
2400139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::NoInfs))
2401139f7f9bSDimitry Andric             FMF.setNoInfs();
2402139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::NoSignedZeros))
2403139f7f9bSDimitry Andric             FMF.setNoSignedZeros();
2404139f7f9bSDimitry Andric           if (0 != (Record[OpNum] & FastMathFlags::AllowReciprocal))
2405139f7f9bSDimitry Andric             FMF.setAllowReciprocal();
2406139f7f9bSDimitry Andric           if (FMF.any())
2407139f7f9bSDimitry Andric             I->setFastMathFlags(FMF);
2408f22ef01cSRoman Divacky         }
2409139f7f9bSDimitry Andric 
2410f22ef01cSRoman Divacky       }
2411f22ef01cSRoman Divacky       break;
2412f22ef01cSRoman Divacky     }
2413f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CAST: {    // CAST: [opval, opty, destty, castopc]
2414f22ef01cSRoman Divacky       unsigned OpNum = 0;
2415f22ef01cSRoman Divacky       Value *Op;
2416f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2417f22ef01cSRoman Divacky           OpNum+2 != Record.size())
2418f785676fSDimitry Andric         return Error(InvalidRecord);
2419f22ef01cSRoman Divacky 
24206122f3e6SDimitry Andric       Type *ResTy = getTypeByID(Record[OpNum]);
2421f22ef01cSRoman Divacky       int Opc = GetDecodedCastOpcode(Record[OpNum+1]);
242291bc56edSDimitry Andric       if (Opc == -1 || !ResTy)
2423f785676fSDimitry Andric         return Error(InvalidRecord);
242491bc56edSDimitry Andric       Instruction *Temp = nullptr;
2425f785676fSDimitry Andric       if ((I = UpgradeBitCastInst(Opc, Op, ResTy, Temp))) {
2426f785676fSDimitry Andric         if (Temp) {
2427f785676fSDimitry Andric           InstructionList.push_back(Temp);
2428f785676fSDimitry Andric           CurBB->getInstList().push_back(Temp);
2429f785676fSDimitry Andric         }
2430f785676fSDimitry Andric       } else {
2431f22ef01cSRoman Divacky         I = CastInst::Create((Instruction::CastOps)Opc, Op, ResTy);
2432f785676fSDimitry Andric       }
2433f22ef01cSRoman Divacky       InstructionList.push_back(I);
2434f22ef01cSRoman Divacky       break;
2435f22ef01cSRoman Divacky     }
2436f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INBOUNDS_GEP:
2437f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_GEP: { // GEP: [n x operands]
2438f22ef01cSRoman Divacky       unsigned OpNum = 0;
2439f22ef01cSRoman Divacky       Value *BasePtr;
2440f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, BasePtr))
2441f785676fSDimitry Andric         return Error(InvalidRecord);
2442f22ef01cSRoman Divacky 
2443f22ef01cSRoman Divacky       SmallVector<Value*, 16> GEPIdx;
2444f22ef01cSRoman Divacky       while (OpNum != Record.size()) {
2445f22ef01cSRoman Divacky         Value *Op;
2446f22ef01cSRoman Divacky         if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2447f785676fSDimitry Andric           return Error(InvalidRecord);
2448f22ef01cSRoman Divacky         GEPIdx.push_back(Op);
2449f22ef01cSRoman Divacky       }
2450f22ef01cSRoman Divacky 
24516122f3e6SDimitry Andric       I = GetElementPtrInst::Create(BasePtr, GEPIdx);
2452f22ef01cSRoman Divacky       InstructionList.push_back(I);
2453f22ef01cSRoman Divacky       if (BitCode == bitc::FUNC_CODE_INST_INBOUNDS_GEP)
2454f22ef01cSRoman Divacky         cast<GetElementPtrInst>(I)->setIsInBounds(true);
2455f22ef01cSRoman Divacky       break;
2456f22ef01cSRoman Divacky     }
2457f22ef01cSRoman Divacky 
2458f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_EXTRACTVAL: {
2459f22ef01cSRoman Divacky                                        // EXTRACTVAL: [opty, opval, n x indices]
2460f22ef01cSRoman Divacky       unsigned OpNum = 0;
2461f22ef01cSRoman Divacky       Value *Agg;
2462f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
2463f785676fSDimitry Andric         return Error(InvalidRecord);
2464f22ef01cSRoman Divacky 
2465f22ef01cSRoman Divacky       SmallVector<unsigned, 4> EXTRACTVALIdx;
2466f22ef01cSRoman Divacky       for (unsigned RecSize = Record.size();
2467f22ef01cSRoman Divacky            OpNum != RecSize; ++OpNum) {
2468f22ef01cSRoman Divacky         uint64_t Index = Record[OpNum];
2469f22ef01cSRoman Divacky         if ((unsigned)Index != Index)
2470f785676fSDimitry Andric           return Error(InvalidValue);
2471f22ef01cSRoman Divacky         EXTRACTVALIdx.push_back((unsigned)Index);
2472f22ef01cSRoman Divacky       }
2473f22ef01cSRoman Divacky 
247417a519f9SDimitry Andric       I = ExtractValueInst::Create(Agg, EXTRACTVALIdx);
2475f22ef01cSRoman Divacky       InstructionList.push_back(I);
2476f22ef01cSRoman Divacky       break;
2477f22ef01cSRoman Divacky     }
2478f22ef01cSRoman Divacky 
2479f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INSERTVAL: {
2480f22ef01cSRoman Divacky                            // INSERTVAL: [opty, opval, opty, opval, n x indices]
2481f22ef01cSRoman Divacky       unsigned OpNum = 0;
2482f22ef01cSRoman Divacky       Value *Agg;
2483f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
2484f785676fSDimitry Andric         return Error(InvalidRecord);
2485f22ef01cSRoman Divacky       Value *Val;
2486f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Val))
2487f785676fSDimitry Andric         return Error(InvalidRecord);
2488f22ef01cSRoman Divacky 
2489f22ef01cSRoman Divacky       SmallVector<unsigned, 4> INSERTVALIdx;
2490f22ef01cSRoman Divacky       for (unsigned RecSize = Record.size();
2491f22ef01cSRoman Divacky            OpNum != RecSize; ++OpNum) {
2492f22ef01cSRoman Divacky         uint64_t Index = Record[OpNum];
2493f22ef01cSRoman Divacky         if ((unsigned)Index != Index)
2494f785676fSDimitry Andric           return Error(InvalidValue);
2495f22ef01cSRoman Divacky         INSERTVALIdx.push_back((unsigned)Index);
2496f22ef01cSRoman Divacky       }
2497f22ef01cSRoman Divacky 
249817a519f9SDimitry Andric       I = InsertValueInst::Create(Agg, Val, INSERTVALIdx);
2499f22ef01cSRoman Divacky       InstructionList.push_back(I);
2500f22ef01cSRoman Divacky       break;
2501f22ef01cSRoman Divacky     }
2502f22ef01cSRoman Divacky 
2503f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SELECT: { // SELECT: [opval, ty, opval, opval]
2504f22ef01cSRoman Divacky       // obsolete form of select
2505f22ef01cSRoman Divacky       // handles select i1 ... in old bitcode
2506f22ef01cSRoman Divacky       unsigned OpNum = 0;
2507f22ef01cSRoman Divacky       Value *TrueVal, *FalseVal, *Cond;
2508f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
25093861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, TrueVal->getType(), FalseVal) ||
25103861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, Type::getInt1Ty(Context), Cond))
2511f785676fSDimitry Andric         return Error(InvalidRecord);
2512f22ef01cSRoman Divacky 
2513f22ef01cSRoman Divacky       I = SelectInst::Create(Cond, TrueVal, FalseVal);
2514f22ef01cSRoman Divacky       InstructionList.push_back(I);
2515f22ef01cSRoman Divacky       break;
2516f22ef01cSRoman Divacky     }
2517f22ef01cSRoman Divacky 
2518f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_VSELECT: {// VSELECT: [ty,opval,opval,predty,pred]
2519f22ef01cSRoman Divacky       // new form of select
2520f22ef01cSRoman Divacky       // handles select i1 or select [N x i1]
2521f22ef01cSRoman Divacky       unsigned OpNum = 0;
2522f22ef01cSRoman Divacky       Value *TrueVal, *FalseVal, *Cond;
2523f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
25243861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, TrueVal->getType(), FalseVal) ||
2525f22ef01cSRoman Divacky           getValueTypePair(Record, OpNum, NextValueNo, Cond))
2526f785676fSDimitry Andric         return Error(InvalidRecord);
2527f22ef01cSRoman Divacky 
2528f22ef01cSRoman Divacky       // select condition can be either i1 or [N x i1]
25296122f3e6SDimitry Andric       if (VectorType* vector_type =
25306122f3e6SDimitry Andric           dyn_cast<VectorType>(Cond->getType())) {
2531f22ef01cSRoman Divacky         // expect <n x i1>
2532f22ef01cSRoman Divacky         if (vector_type->getElementType() != Type::getInt1Ty(Context))
2533f785676fSDimitry Andric           return Error(InvalidTypeForValue);
2534f22ef01cSRoman Divacky       } else {
2535f22ef01cSRoman Divacky         // expect i1
2536f22ef01cSRoman Divacky         if (Cond->getType() != Type::getInt1Ty(Context))
2537f785676fSDimitry Andric           return Error(InvalidTypeForValue);
2538f22ef01cSRoman Divacky       }
2539f22ef01cSRoman Divacky 
2540f22ef01cSRoman Divacky       I = SelectInst::Create(Cond, TrueVal, FalseVal);
2541f22ef01cSRoman Divacky       InstructionList.push_back(I);
2542f22ef01cSRoman Divacky       break;
2543f22ef01cSRoman Divacky     }
2544f22ef01cSRoman Divacky 
2545f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_EXTRACTELT: { // EXTRACTELT: [opty, opval, opval]
2546f22ef01cSRoman Divacky       unsigned OpNum = 0;
2547f22ef01cSRoman Divacky       Value *Vec, *Idx;
2548f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
254991bc56edSDimitry Andric           getValueTypePair(Record, OpNum, NextValueNo, Idx))
2550f785676fSDimitry Andric         return Error(InvalidRecord);
2551f22ef01cSRoman Divacky       I = ExtractElementInst::Create(Vec, Idx);
2552f22ef01cSRoman Divacky       InstructionList.push_back(I);
2553f22ef01cSRoman Divacky       break;
2554f22ef01cSRoman Divacky     }
2555f22ef01cSRoman Divacky 
2556f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INSERTELT: { // INSERTELT: [ty, opval,opval,opval]
2557f22ef01cSRoman Divacky       unsigned OpNum = 0;
2558f22ef01cSRoman Divacky       Value *Vec, *Elt, *Idx;
2559f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
25603861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
2561f22ef01cSRoman Divacky                    cast<VectorType>(Vec->getType())->getElementType(), Elt) ||
256291bc56edSDimitry Andric           getValueTypePair(Record, OpNum, NextValueNo, Idx))
2563f785676fSDimitry Andric         return Error(InvalidRecord);
2564f22ef01cSRoman Divacky       I = InsertElementInst::Create(Vec, Elt, Idx);
2565f22ef01cSRoman Divacky       InstructionList.push_back(I);
2566f22ef01cSRoman Divacky       break;
2567f22ef01cSRoman Divacky     }
2568f22ef01cSRoman Divacky 
2569f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SHUFFLEVEC: {// SHUFFLEVEC: [opval,ty,opval,opval]
2570f22ef01cSRoman Divacky       unsigned OpNum = 0;
2571f22ef01cSRoman Divacky       Value *Vec1, *Vec2, *Mask;
2572f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec1) ||
25733861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, Vec1->getType(), Vec2))
2574f785676fSDimitry Andric         return Error(InvalidRecord);
2575f22ef01cSRoman Divacky 
2576f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Mask))
2577f785676fSDimitry Andric         return Error(InvalidRecord);
2578f22ef01cSRoman Divacky       I = new ShuffleVectorInst(Vec1, Vec2, Mask);
2579f22ef01cSRoman Divacky       InstructionList.push_back(I);
2580f22ef01cSRoman Divacky       break;
2581f22ef01cSRoman Divacky     }
2582f22ef01cSRoman Divacky 
2583f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CMP:   // CMP: [opty, opval, opval, pred]
2584f22ef01cSRoman Divacky       // Old form of ICmp/FCmp returning bool
2585f22ef01cSRoman Divacky       // Existed to differentiate between icmp/fcmp and vicmp/vfcmp which were
2586f22ef01cSRoman Divacky       // both legal on vectors but had different behaviour.
2587f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CMP2: { // CMP2: [opty, opval, opval, pred]
2588f22ef01cSRoman Divacky       // FCmp/ICmp returning bool or vector of bool
2589f22ef01cSRoman Divacky 
2590f22ef01cSRoman Divacky       unsigned OpNum = 0;
2591f22ef01cSRoman Divacky       Value *LHS, *RHS;
2592f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
25933861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo, LHS->getType(), RHS) ||
2594f22ef01cSRoman Divacky           OpNum+1 != Record.size())
2595f785676fSDimitry Andric         return Error(InvalidRecord);
2596f22ef01cSRoman Divacky 
2597f22ef01cSRoman Divacky       if (LHS->getType()->isFPOrFPVectorTy())
2598f22ef01cSRoman Divacky         I = new FCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
2599f22ef01cSRoman Divacky       else
2600f22ef01cSRoman Divacky         I = new ICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
2601f22ef01cSRoman Divacky       InstructionList.push_back(I);
2602f22ef01cSRoman Divacky       break;
2603f22ef01cSRoman Divacky     }
2604f22ef01cSRoman Divacky 
2605f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_RET: // RET: [opty,opval<optional>]
2606f22ef01cSRoman Divacky       {
2607f22ef01cSRoman Divacky         unsigned Size = Record.size();
2608f22ef01cSRoman Divacky         if (Size == 0) {
2609f22ef01cSRoman Divacky           I = ReturnInst::Create(Context);
2610f22ef01cSRoman Divacky           InstructionList.push_back(I);
2611f22ef01cSRoman Divacky           break;
2612f22ef01cSRoman Divacky         }
2613f22ef01cSRoman Divacky 
2614f22ef01cSRoman Divacky         unsigned OpNum = 0;
261591bc56edSDimitry Andric         Value *Op = nullptr;
2616f22ef01cSRoman Divacky         if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2617f785676fSDimitry Andric           return Error(InvalidRecord);
261817a519f9SDimitry Andric         if (OpNum != Record.size())
2619f785676fSDimitry Andric           return Error(InvalidRecord);
2620f22ef01cSRoman Divacky 
262117a519f9SDimitry Andric         I = ReturnInst::Create(Context, Op);
2622f22ef01cSRoman Divacky         InstructionList.push_back(I);
2623f22ef01cSRoman Divacky         break;
2624f22ef01cSRoman Divacky       }
2625f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_BR: { // BR: [bb#, bb#, opval] or [bb#]
2626f22ef01cSRoman Divacky       if (Record.size() != 1 && Record.size() != 3)
2627f785676fSDimitry Andric         return Error(InvalidRecord);
2628f22ef01cSRoman Divacky       BasicBlock *TrueDest = getBasicBlock(Record[0]);
262991bc56edSDimitry Andric       if (!TrueDest)
2630f785676fSDimitry Andric         return Error(InvalidRecord);
2631f22ef01cSRoman Divacky 
2632f22ef01cSRoman Divacky       if (Record.size() == 1) {
2633f22ef01cSRoman Divacky         I = BranchInst::Create(TrueDest);
2634f22ef01cSRoman Divacky         InstructionList.push_back(I);
2635f22ef01cSRoman Divacky       }
2636f22ef01cSRoman Divacky       else {
2637f22ef01cSRoman Divacky         BasicBlock *FalseDest = getBasicBlock(Record[1]);
26383861d79fSDimitry Andric         Value *Cond = getValue(Record, 2, NextValueNo,
26393861d79fSDimitry Andric                                Type::getInt1Ty(Context));
264091bc56edSDimitry Andric         if (!FalseDest || !Cond)
2641f785676fSDimitry Andric           return Error(InvalidRecord);
2642f22ef01cSRoman Divacky         I = BranchInst::Create(TrueDest, FalseDest, Cond);
2643f22ef01cSRoman Divacky         InstructionList.push_back(I);
2644f22ef01cSRoman Divacky       }
2645f22ef01cSRoman Divacky       break;
2646f22ef01cSRoman Divacky     }
2647f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SWITCH: { // SWITCH: [opty, op0, op1, ...]
26487ae0e2c9SDimitry Andric       // Check magic
26497ae0e2c9SDimitry Andric       if ((Record[0] >> 16) == SWITCH_INST_MAGIC) {
2650f785676fSDimitry Andric         // "New" SwitchInst format with case ranges. The changes to write this
2651f785676fSDimitry Andric         // format were reverted but we still recognize bitcode that uses it.
2652f785676fSDimitry Andric         // Hopefully someday we will have support for case ranges and can use
2653f785676fSDimitry Andric         // this format again.
26547ae0e2c9SDimitry Andric 
26557ae0e2c9SDimitry Andric         Type *OpTy = getTypeByID(Record[1]);
26567ae0e2c9SDimitry Andric         unsigned ValueBitWidth = cast<IntegerType>(OpTy)->getBitWidth();
26577ae0e2c9SDimitry Andric 
26583861d79fSDimitry Andric         Value *Cond = getValue(Record, 2, NextValueNo, OpTy);
26597ae0e2c9SDimitry Andric         BasicBlock *Default = getBasicBlock(Record[3]);
266091bc56edSDimitry Andric         if (!OpTy || !Cond || !Default)
2661f785676fSDimitry Andric           return Error(InvalidRecord);
26627ae0e2c9SDimitry Andric 
26637ae0e2c9SDimitry Andric         unsigned NumCases = Record[4];
26647ae0e2c9SDimitry Andric 
26657ae0e2c9SDimitry Andric         SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
26667ae0e2c9SDimitry Andric         InstructionList.push_back(SI);
26677ae0e2c9SDimitry Andric 
26687ae0e2c9SDimitry Andric         unsigned CurIdx = 5;
26697ae0e2c9SDimitry Andric         for (unsigned i = 0; i != NumCases; ++i) {
2670f785676fSDimitry Andric           SmallVector<ConstantInt*, 1> CaseVals;
26717ae0e2c9SDimitry Andric           unsigned NumItems = Record[CurIdx++];
26727ae0e2c9SDimitry Andric           for (unsigned ci = 0; ci != NumItems; ++ci) {
26737ae0e2c9SDimitry Andric             bool isSingleNumber = Record[CurIdx++];
26747ae0e2c9SDimitry Andric 
26757ae0e2c9SDimitry Andric             APInt Low;
26767ae0e2c9SDimitry Andric             unsigned ActiveWords = 1;
26777ae0e2c9SDimitry Andric             if (ValueBitWidth > 64)
26787ae0e2c9SDimitry Andric               ActiveWords = Record[CurIdx++];
26797ae0e2c9SDimitry Andric             Low = ReadWideAPInt(makeArrayRef(&Record[CurIdx], ActiveWords),
26807ae0e2c9SDimitry Andric                                 ValueBitWidth);
26817ae0e2c9SDimitry Andric             CurIdx += ActiveWords;
26827ae0e2c9SDimitry Andric 
26837ae0e2c9SDimitry Andric             if (!isSingleNumber) {
26847ae0e2c9SDimitry Andric               ActiveWords = 1;
26857ae0e2c9SDimitry Andric               if (ValueBitWidth > 64)
26867ae0e2c9SDimitry Andric                 ActiveWords = Record[CurIdx++];
26877ae0e2c9SDimitry Andric               APInt High =
26887ae0e2c9SDimitry Andric                   ReadWideAPInt(makeArrayRef(&Record[CurIdx], ActiveWords),
26897ae0e2c9SDimitry Andric                                 ValueBitWidth);
26907ae0e2c9SDimitry Andric               CurIdx += ActiveWords;
2691f785676fSDimitry Andric 
2692f785676fSDimitry Andric               // FIXME: It is not clear whether values in the range should be
2693f785676fSDimitry Andric               // compared as signed or unsigned values. The partially
2694f785676fSDimitry Andric               // implemented changes that used this format in the past used
2695f785676fSDimitry Andric               // unsigned comparisons.
2696f785676fSDimitry Andric               for ( ; Low.ule(High); ++Low)
2697f785676fSDimitry Andric                 CaseVals.push_back(ConstantInt::get(Context, Low));
26987ae0e2c9SDimitry Andric             } else
2699f785676fSDimitry Andric               CaseVals.push_back(ConstantInt::get(Context, Low));
27007ae0e2c9SDimitry Andric           }
27017ae0e2c9SDimitry Andric           BasicBlock *DestBB = getBasicBlock(Record[CurIdx++]);
2702f785676fSDimitry Andric           for (SmallVector<ConstantInt*, 1>::iterator cvi = CaseVals.begin(),
2703f785676fSDimitry Andric                  cve = CaseVals.end(); cvi != cve; ++cvi)
2704f785676fSDimitry Andric             SI->addCase(*cvi, DestBB);
27057ae0e2c9SDimitry Andric         }
27067ae0e2c9SDimitry Andric         I = SI;
27077ae0e2c9SDimitry Andric         break;
27087ae0e2c9SDimitry Andric       }
27097ae0e2c9SDimitry Andric 
27107ae0e2c9SDimitry Andric       // Old SwitchInst format without case ranges.
27117ae0e2c9SDimitry Andric 
2712f22ef01cSRoman Divacky       if (Record.size() < 3 || (Record.size() & 1) == 0)
2713f785676fSDimitry Andric         return Error(InvalidRecord);
27146122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[0]);
27153861d79fSDimitry Andric       Value *Cond = getValue(Record, 1, NextValueNo, OpTy);
2716f22ef01cSRoman Divacky       BasicBlock *Default = getBasicBlock(Record[2]);
271791bc56edSDimitry Andric       if (!OpTy || !Cond || !Default)
2718f785676fSDimitry Andric         return Error(InvalidRecord);
2719f22ef01cSRoman Divacky       unsigned NumCases = (Record.size()-3)/2;
2720f22ef01cSRoman Divacky       SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
2721f22ef01cSRoman Divacky       InstructionList.push_back(SI);
2722f22ef01cSRoman Divacky       for (unsigned i = 0, e = NumCases; i != e; ++i) {
2723f22ef01cSRoman Divacky         ConstantInt *CaseVal =
2724f22ef01cSRoman Divacky           dyn_cast_or_null<ConstantInt>(getFnValueByID(Record[3+i*2], OpTy));
2725f22ef01cSRoman Divacky         BasicBlock *DestBB = getBasicBlock(Record[1+3+i*2]);
272691bc56edSDimitry Andric         if (!CaseVal || !DestBB) {
2727f22ef01cSRoman Divacky           delete SI;
2728f785676fSDimitry Andric           return Error(InvalidRecord);
2729f22ef01cSRoman Divacky         }
2730f22ef01cSRoman Divacky         SI->addCase(CaseVal, DestBB);
2731f22ef01cSRoman Divacky       }
2732f22ef01cSRoman Divacky       I = SI;
2733f22ef01cSRoman Divacky       break;
2734f22ef01cSRoman Divacky     }
2735f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INDIRECTBR: { // INDIRECTBR: [opty, op0, op1, ...]
2736f22ef01cSRoman Divacky       if (Record.size() < 2)
2737f785676fSDimitry Andric         return Error(InvalidRecord);
27386122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[0]);
27393861d79fSDimitry Andric       Value *Address = getValue(Record, 1, NextValueNo, OpTy);
274091bc56edSDimitry Andric       if (!OpTy || !Address)
2741f785676fSDimitry Andric         return Error(InvalidRecord);
2742f22ef01cSRoman Divacky       unsigned NumDests = Record.size()-2;
2743f22ef01cSRoman Divacky       IndirectBrInst *IBI = IndirectBrInst::Create(Address, NumDests);
2744f22ef01cSRoman Divacky       InstructionList.push_back(IBI);
2745f22ef01cSRoman Divacky       for (unsigned i = 0, e = NumDests; i != e; ++i) {
2746f22ef01cSRoman Divacky         if (BasicBlock *DestBB = getBasicBlock(Record[2+i])) {
2747f22ef01cSRoman Divacky           IBI->addDestination(DestBB);
2748f22ef01cSRoman Divacky         } else {
2749f22ef01cSRoman Divacky           delete IBI;
2750f785676fSDimitry Andric           return Error(InvalidRecord);
2751f22ef01cSRoman Divacky         }
2752f22ef01cSRoman Divacky       }
2753f22ef01cSRoman Divacky       I = IBI;
2754f22ef01cSRoman Divacky       break;
2755f22ef01cSRoman Divacky     }
2756f22ef01cSRoman Divacky 
2757f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INVOKE: {
2758f22ef01cSRoman Divacky       // INVOKE: [attrs, cc, normBB, unwindBB, fnty, op0,op1,op2, ...]
2759f785676fSDimitry Andric       if (Record.size() < 4)
2760f785676fSDimitry Andric         return Error(InvalidRecord);
2761139f7f9bSDimitry Andric       AttributeSet PAL = getAttributes(Record[0]);
2762f22ef01cSRoman Divacky       unsigned CCInfo = Record[1];
2763f22ef01cSRoman Divacky       BasicBlock *NormalBB = getBasicBlock(Record[2]);
2764f22ef01cSRoman Divacky       BasicBlock *UnwindBB = getBasicBlock(Record[3]);
2765f22ef01cSRoman Divacky 
2766f22ef01cSRoman Divacky       unsigned OpNum = 4;
2767f22ef01cSRoman Divacky       Value *Callee;
2768f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
2769f785676fSDimitry Andric         return Error(InvalidRecord);
2770f22ef01cSRoman Divacky 
27716122f3e6SDimitry Andric       PointerType *CalleeTy = dyn_cast<PointerType>(Callee->getType());
277291bc56edSDimitry Andric       FunctionType *FTy = !CalleeTy ? nullptr :
2773f22ef01cSRoman Divacky         dyn_cast<FunctionType>(CalleeTy->getElementType());
2774f22ef01cSRoman Divacky 
2775f22ef01cSRoman Divacky       // Check that the right number of fixed parameters are here.
277691bc56edSDimitry Andric       if (!FTy || !NormalBB || !UnwindBB ||
2777f22ef01cSRoman Divacky           Record.size() < OpNum+FTy->getNumParams())
2778f785676fSDimitry Andric         return Error(InvalidRecord);
2779f22ef01cSRoman Divacky 
2780f22ef01cSRoman Divacky       SmallVector<Value*, 16> Ops;
2781f22ef01cSRoman Divacky       for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
27823861d79fSDimitry Andric         Ops.push_back(getValue(Record, OpNum, NextValueNo,
27833861d79fSDimitry Andric                                FTy->getParamType(i)));
278491bc56edSDimitry Andric         if (!Ops.back())
2785f785676fSDimitry Andric           return Error(InvalidRecord);
2786f22ef01cSRoman Divacky       }
2787f22ef01cSRoman Divacky 
2788f22ef01cSRoman Divacky       if (!FTy->isVarArg()) {
2789f22ef01cSRoman Divacky         if (Record.size() != OpNum)
2790f785676fSDimitry Andric           return Error(InvalidRecord);
2791f22ef01cSRoman Divacky       } else {
2792f22ef01cSRoman Divacky         // Read type/value pairs for varargs params.
2793f22ef01cSRoman Divacky         while (OpNum != Record.size()) {
2794f22ef01cSRoman Divacky           Value *Op;
2795f22ef01cSRoman Divacky           if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2796f785676fSDimitry Andric             return Error(InvalidRecord);
2797f22ef01cSRoman Divacky           Ops.push_back(Op);
2798f22ef01cSRoman Divacky         }
2799f22ef01cSRoman Divacky       }
2800f22ef01cSRoman Divacky 
280117a519f9SDimitry Andric       I = InvokeInst::Create(Callee, NormalBB, UnwindBB, Ops);
2802f22ef01cSRoman Divacky       InstructionList.push_back(I);
2803f22ef01cSRoman Divacky       cast<InvokeInst>(I)->setCallingConv(
2804f22ef01cSRoman Divacky         static_cast<CallingConv::ID>(CCInfo));
2805f22ef01cSRoman Divacky       cast<InvokeInst>(I)->setAttributes(PAL);
2806f22ef01cSRoman Divacky       break;
2807f22ef01cSRoman Divacky     }
28086122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_RESUME: { // RESUME: [opval]
28096122f3e6SDimitry Andric       unsigned Idx = 0;
281091bc56edSDimitry Andric       Value *Val = nullptr;
28116122f3e6SDimitry Andric       if (getValueTypePair(Record, Idx, NextValueNo, Val))
2812f785676fSDimitry Andric         return Error(InvalidRecord);
28136122f3e6SDimitry Andric       I = ResumeInst::Create(Val);
28146122f3e6SDimitry Andric       InstructionList.push_back(I);
28156122f3e6SDimitry Andric       break;
28166122f3e6SDimitry Andric     }
2817f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_UNREACHABLE: // UNREACHABLE
2818f22ef01cSRoman Divacky       I = new UnreachableInst(Context);
2819f22ef01cSRoman Divacky       InstructionList.push_back(I);
2820f22ef01cSRoman Divacky       break;
2821f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_PHI: { // PHI: [ty, val0,bb0, ...]
2822f22ef01cSRoman Divacky       if (Record.size() < 1 || ((Record.size()-1)&1))
2823f785676fSDimitry Andric         return Error(InvalidRecord);
28246122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[0]);
2825f785676fSDimitry Andric       if (!Ty)
2826f785676fSDimitry Andric         return Error(InvalidRecord);
2827f22ef01cSRoman Divacky 
28283b0f4066SDimitry Andric       PHINode *PN = PHINode::Create(Ty, (Record.size()-1)/2);
2829f22ef01cSRoman Divacky       InstructionList.push_back(PN);
2830f22ef01cSRoman Divacky 
2831f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size()-1; i != e; i += 2) {
28323861d79fSDimitry Andric         Value *V;
28333861d79fSDimitry Andric         // With the new function encoding, it is possible that operands have
28343861d79fSDimitry Andric         // negative IDs (for forward references).  Use a signed VBR
28353861d79fSDimitry Andric         // representation to keep the encoding small.
28363861d79fSDimitry Andric         if (UseRelativeIDs)
28373861d79fSDimitry Andric           V = getValueSigned(Record, 1+i, NextValueNo, Ty);
28383861d79fSDimitry Andric         else
28393861d79fSDimitry Andric           V = getValue(Record, 1+i, NextValueNo, Ty);
2840f22ef01cSRoman Divacky         BasicBlock *BB = getBasicBlock(Record[2+i]);
2841f785676fSDimitry Andric         if (!V || !BB)
2842f785676fSDimitry Andric           return Error(InvalidRecord);
2843f22ef01cSRoman Divacky         PN->addIncoming(V, BB);
2844f22ef01cSRoman Divacky       }
2845f22ef01cSRoman Divacky       I = PN;
2846f22ef01cSRoman Divacky       break;
2847f22ef01cSRoman Divacky     }
2848f22ef01cSRoman Divacky 
28496122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_LANDINGPAD: {
28506122f3e6SDimitry Andric       // LANDINGPAD: [ty, val, val, num, (id0,val0 ...)?]
28516122f3e6SDimitry Andric       unsigned Idx = 0;
28526122f3e6SDimitry Andric       if (Record.size() < 4)
2853f785676fSDimitry Andric         return Error(InvalidRecord);
28546122f3e6SDimitry Andric       Type *Ty = getTypeByID(Record[Idx++]);
2855f785676fSDimitry Andric       if (!Ty)
2856f785676fSDimitry Andric         return Error(InvalidRecord);
285791bc56edSDimitry Andric       Value *PersFn = nullptr;
28586122f3e6SDimitry Andric       if (getValueTypePair(Record, Idx, NextValueNo, PersFn))
2859f785676fSDimitry Andric         return Error(InvalidRecord);
28606122f3e6SDimitry Andric 
28616122f3e6SDimitry Andric       bool IsCleanup = !!Record[Idx++];
28626122f3e6SDimitry Andric       unsigned NumClauses = Record[Idx++];
28636122f3e6SDimitry Andric       LandingPadInst *LP = LandingPadInst::Create(Ty, PersFn, NumClauses);
28646122f3e6SDimitry Andric       LP->setCleanup(IsCleanup);
28656122f3e6SDimitry Andric       for (unsigned J = 0; J != NumClauses; ++J) {
28666122f3e6SDimitry Andric         LandingPadInst::ClauseType CT =
28676122f3e6SDimitry Andric           LandingPadInst::ClauseType(Record[Idx++]); (void)CT;
28686122f3e6SDimitry Andric         Value *Val;
28696122f3e6SDimitry Andric 
28706122f3e6SDimitry Andric         if (getValueTypePair(Record, Idx, NextValueNo, Val)) {
28716122f3e6SDimitry Andric           delete LP;
2872f785676fSDimitry Andric           return Error(InvalidRecord);
28736122f3e6SDimitry Andric         }
28746122f3e6SDimitry Andric 
28756122f3e6SDimitry Andric         assert((CT != LandingPadInst::Catch ||
28766122f3e6SDimitry Andric                 !isa<ArrayType>(Val->getType())) &&
28776122f3e6SDimitry Andric                "Catch clause has a invalid type!");
28786122f3e6SDimitry Andric         assert((CT != LandingPadInst::Filter ||
28796122f3e6SDimitry Andric                 isa<ArrayType>(Val->getType())) &&
28806122f3e6SDimitry Andric                "Filter clause has invalid type!");
288191bc56edSDimitry Andric         LP->addClause(cast<Constant>(Val));
28826122f3e6SDimitry Andric       }
28836122f3e6SDimitry Andric 
28846122f3e6SDimitry Andric       I = LP;
28856122f3e6SDimitry Andric       InstructionList.push_back(I);
28866122f3e6SDimitry Andric       break;
28876122f3e6SDimitry Andric     }
28886122f3e6SDimitry Andric 
288917a519f9SDimitry Andric     case bitc::FUNC_CODE_INST_ALLOCA: { // ALLOCA: [instty, opty, op, align]
289017a519f9SDimitry Andric       if (Record.size() != 4)
2891f785676fSDimitry Andric         return Error(InvalidRecord);
28926122f3e6SDimitry Andric       PointerType *Ty =
2893f22ef01cSRoman Divacky         dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
28946122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[1]);
289517a519f9SDimitry Andric       Value *Size = getFnValueByID(Record[2], OpTy);
289691bc56edSDimitry Andric       unsigned AlignRecord = Record[3];
289791bc56edSDimitry Andric       bool InAlloca = AlignRecord & (1 << 5);
289891bc56edSDimitry Andric       unsigned Align = AlignRecord & ((1 << 5) - 1);
2899f785676fSDimitry Andric       if (!Ty || !Size)
2900f785676fSDimitry Andric         return Error(InvalidRecord);
290191bc56edSDimitry Andric       AllocaInst *AI = new AllocaInst(Ty->getElementType(), Size, (1 << Align) >> 1);
290291bc56edSDimitry Andric       AI->setUsedWithInAlloca(InAlloca);
290391bc56edSDimitry Andric       I = AI;
2904f22ef01cSRoman Divacky       InstructionList.push_back(I);
2905f22ef01cSRoman Divacky       break;
2906f22ef01cSRoman Divacky     }
2907f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_LOAD: { // LOAD: [opty, op, align, vol]
2908f22ef01cSRoman Divacky       unsigned OpNum = 0;
2909f22ef01cSRoman Divacky       Value *Op;
2910f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2911f22ef01cSRoman Divacky           OpNum+2 != Record.size())
2912f785676fSDimitry Andric         return Error(InvalidRecord);
2913f22ef01cSRoman Divacky 
2914f22ef01cSRoman Divacky       I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1);
2915f22ef01cSRoman Divacky       InstructionList.push_back(I);
2916f22ef01cSRoman Divacky       break;
2917f22ef01cSRoman Divacky     }
29186122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_LOADATOMIC: {
29196122f3e6SDimitry Andric        // LOADATOMIC: [opty, op, align, vol, ordering, synchscope]
29206122f3e6SDimitry Andric       unsigned OpNum = 0;
29216122f3e6SDimitry Andric       Value *Op;
29226122f3e6SDimitry Andric       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
29236122f3e6SDimitry Andric           OpNum+4 != Record.size())
2924f785676fSDimitry Andric         return Error(InvalidRecord);
29256122f3e6SDimitry Andric 
29266122f3e6SDimitry Andric 
29276122f3e6SDimitry Andric       AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+2]);
29286122f3e6SDimitry Andric       if (Ordering == NotAtomic || Ordering == Release ||
29296122f3e6SDimitry Andric           Ordering == AcquireRelease)
2930f785676fSDimitry Andric         return Error(InvalidRecord);
29316122f3e6SDimitry Andric       if (Ordering != NotAtomic && Record[OpNum] == 0)
2932f785676fSDimitry Andric         return Error(InvalidRecord);
29336122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+3]);
29346122f3e6SDimitry Andric 
29356122f3e6SDimitry Andric       I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1,
29366122f3e6SDimitry Andric                        Ordering, SynchScope);
29376122f3e6SDimitry Andric       InstructionList.push_back(I);
29386122f3e6SDimitry Andric       break;
29396122f3e6SDimitry Andric     }
294017a519f9SDimitry Andric     case bitc::FUNC_CODE_INST_STORE: { // STORE2:[ptrty, ptr, val, align, vol]
2941f22ef01cSRoman Divacky       unsigned OpNum = 0;
2942f22ef01cSRoman Divacky       Value *Val, *Ptr;
2943f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
29443861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
2945f22ef01cSRoman Divacky                     cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
2946f22ef01cSRoman Divacky           OpNum+2 != Record.size())
2947f785676fSDimitry Andric         return Error(InvalidRecord);
2948f22ef01cSRoman Divacky 
2949f22ef01cSRoman Divacky       I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
2950f22ef01cSRoman Divacky       InstructionList.push_back(I);
2951f22ef01cSRoman Divacky       break;
2952f22ef01cSRoman Divacky     }
29536122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_STOREATOMIC: {
29546122f3e6SDimitry Andric       // STOREATOMIC: [ptrty, ptr, val, align, vol, ordering, synchscope]
29556122f3e6SDimitry Andric       unsigned OpNum = 0;
29566122f3e6SDimitry Andric       Value *Val, *Ptr;
29576122f3e6SDimitry Andric       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
29583861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
29596122f3e6SDimitry Andric                     cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
29606122f3e6SDimitry Andric           OpNum+4 != Record.size())
2961f785676fSDimitry Andric         return Error(InvalidRecord);
29626122f3e6SDimitry Andric 
29636122f3e6SDimitry Andric       AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+2]);
29646122f3e6SDimitry Andric       if (Ordering == NotAtomic || Ordering == Acquire ||
29656122f3e6SDimitry Andric           Ordering == AcquireRelease)
2966f785676fSDimitry Andric         return Error(InvalidRecord);
29676122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+3]);
29686122f3e6SDimitry Andric       if (Ordering != NotAtomic && Record[OpNum] == 0)
2969f785676fSDimitry Andric         return Error(InvalidRecord);
29706122f3e6SDimitry Andric 
29716122f3e6SDimitry Andric       I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1,
29726122f3e6SDimitry Andric                         Ordering, SynchScope);
29736122f3e6SDimitry Andric       InstructionList.push_back(I);
29746122f3e6SDimitry Andric       break;
29756122f3e6SDimitry Andric     }
29766122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_CMPXCHG: {
297791bc56edSDimitry Andric       // CMPXCHG:[ptrty, ptr, cmp, new, vol, successordering, synchscope,
297891bc56edSDimitry Andric       //          failureordering?, isweak?]
29796122f3e6SDimitry Andric       unsigned OpNum = 0;
29806122f3e6SDimitry Andric       Value *Ptr, *Cmp, *New;
29816122f3e6SDimitry Andric       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
29823861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
29836122f3e6SDimitry Andric                     cast<PointerType>(Ptr->getType())->getElementType(), Cmp) ||
29843861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
29856122f3e6SDimitry Andric                     cast<PointerType>(Ptr->getType())->getElementType(), New) ||
298691bc56edSDimitry Andric           (Record.size() < OpNum + 3 || Record.size() > OpNum + 5))
2987f785676fSDimitry Andric         return Error(InvalidRecord);
298891bc56edSDimitry Andric       AtomicOrdering SuccessOrdering = GetDecodedOrdering(Record[OpNum+1]);
298991bc56edSDimitry Andric       if (SuccessOrdering == NotAtomic || SuccessOrdering == Unordered)
2990f785676fSDimitry Andric         return Error(InvalidRecord);
29916122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+2]);
299291bc56edSDimitry Andric 
299391bc56edSDimitry Andric       AtomicOrdering FailureOrdering;
299491bc56edSDimitry Andric       if (Record.size() < 7)
299591bc56edSDimitry Andric         FailureOrdering =
299691bc56edSDimitry Andric             AtomicCmpXchgInst::getStrongestFailureOrdering(SuccessOrdering);
299791bc56edSDimitry Andric       else
299891bc56edSDimitry Andric         FailureOrdering = GetDecodedOrdering(Record[OpNum+3]);
299991bc56edSDimitry Andric 
300091bc56edSDimitry Andric       I = new AtomicCmpXchgInst(Ptr, Cmp, New, SuccessOrdering, FailureOrdering,
300191bc56edSDimitry Andric                                 SynchScope);
30026122f3e6SDimitry Andric       cast<AtomicCmpXchgInst>(I)->setVolatile(Record[OpNum]);
300391bc56edSDimitry Andric 
300491bc56edSDimitry Andric       if (Record.size() < 8) {
300591bc56edSDimitry Andric         // Before weak cmpxchgs existed, the instruction simply returned the
300691bc56edSDimitry Andric         // value loaded from memory, so bitcode files from that era will be
300791bc56edSDimitry Andric         // expecting the first component of a modern cmpxchg.
300891bc56edSDimitry Andric         CurBB->getInstList().push_back(I);
300991bc56edSDimitry Andric         I = ExtractValueInst::Create(I, 0);
301091bc56edSDimitry Andric       } else {
301191bc56edSDimitry Andric         cast<AtomicCmpXchgInst>(I)->setWeak(Record[OpNum+4]);
301291bc56edSDimitry Andric       }
301391bc56edSDimitry Andric 
30146122f3e6SDimitry Andric       InstructionList.push_back(I);
30156122f3e6SDimitry Andric       break;
30166122f3e6SDimitry Andric     }
30176122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_ATOMICRMW: {
30186122f3e6SDimitry Andric       // ATOMICRMW:[ptrty, ptr, val, op, vol, ordering, synchscope]
30196122f3e6SDimitry Andric       unsigned OpNum = 0;
30206122f3e6SDimitry Andric       Value *Ptr, *Val;
30216122f3e6SDimitry Andric       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
30223861d79fSDimitry Andric           popValue(Record, OpNum, NextValueNo,
30236122f3e6SDimitry Andric                     cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
30246122f3e6SDimitry Andric           OpNum+4 != Record.size())
3025f785676fSDimitry Andric         return Error(InvalidRecord);
30266122f3e6SDimitry Andric       AtomicRMWInst::BinOp Operation = GetDecodedRMWOperation(Record[OpNum]);
30276122f3e6SDimitry Andric       if (Operation < AtomicRMWInst::FIRST_BINOP ||
30286122f3e6SDimitry Andric           Operation > AtomicRMWInst::LAST_BINOP)
3029f785676fSDimitry Andric         return Error(InvalidRecord);
30306122f3e6SDimitry Andric       AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+2]);
30316122f3e6SDimitry Andric       if (Ordering == NotAtomic || Ordering == Unordered)
3032f785676fSDimitry Andric         return Error(InvalidRecord);
30336122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+3]);
30346122f3e6SDimitry Andric       I = new AtomicRMWInst(Operation, Ptr, Val, Ordering, SynchScope);
30356122f3e6SDimitry Andric       cast<AtomicRMWInst>(I)->setVolatile(Record[OpNum+1]);
30366122f3e6SDimitry Andric       InstructionList.push_back(I);
30376122f3e6SDimitry Andric       break;
30386122f3e6SDimitry Andric     }
30396122f3e6SDimitry Andric     case bitc::FUNC_CODE_INST_FENCE: { // FENCE:[ordering, synchscope]
30406122f3e6SDimitry Andric       if (2 != Record.size())
3041f785676fSDimitry Andric         return Error(InvalidRecord);
30426122f3e6SDimitry Andric       AtomicOrdering Ordering = GetDecodedOrdering(Record[0]);
30436122f3e6SDimitry Andric       if (Ordering == NotAtomic || Ordering == Unordered ||
30446122f3e6SDimitry Andric           Ordering == Monotonic)
3045f785676fSDimitry Andric         return Error(InvalidRecord);
30466122f3e6SDimitry Andric       SynchronizationScope SynchScope = GetDecodedSynchScope(Record[1]);
30476122f3e6SDimitry Andric       I = new FenceInst(Context, Ordering, SynchScope);
30486122f3e6SDimitry Andric       InstructionList.push_back(I);
30496122f3e6SDimitry Andric       break;
30506122f3e6SDimitry Andric     }
305117a519f9SDimitry Andric     case bitc::FUNC_CODE_INST_CALL: {
3052f22ef01cSRoman Divacky       // CALL: [paramattrs, cc, fnty, fnid, arg0, arg1...]
3053f22ef01cSRoman Divacky       if (Record.size() < 3)
3054f785676fSDimitry Andric         return Error(InvalidRecord);
3055f22ef01cSRoman Divacky 
3056139f7f9bSDimitry Andric       AttributeSet PAL = getAttributes(Record[0]);
3057f22ef01cSRoman Divacky       unsigned CCInfo = Record[1];
3058f22ef01cSRoman Divacky 
3059f22ef01cSRoman Divacky       unsigned OpNum = 2;
3060f22ef01cSRoman Divacky       Value *Callee;
3061f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
3062f785676fSDimitry Andric         return Error(InvalidRecord);
3063f22ef01cSRoman Divacky 
30646122f3e6SDimitry Andric       PointerType *OpTy = dyn_cast<PointerType>(Callee->getType());
306591bc56edSDimitry Andric       FunctionType *FTy = nullptr;
3066f22ef01cSRoman Divacky       if (OpTy) FTy = dyn_cast<FunctionType>(OpTy->getElementType());
3067f22ef01cSRoman Divacky       if (!FTy || Record.size() < FTy->getNumParams()+OpNum)
3068f785676fSDimitry Andric         return Error(InvalidRecord);
3069f22ef01cSRoman Divacky 
3070f22ef01cSRoman Divacky       SmallVector<Value*, 16> Args;
3071f22ef01cSRoman Divacky       // Read the fixed params.
3072f22ef01cSRoman Divacky       for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
307317a519f9SDimitry Andric         if (FTy->getParamType(i)->isLabelTy())
3074f22ef01cSRoman Divacky           Args.push_back(getBasicBlock(Record[OpNum]));
3075f22ef01cSRoman Divacky         else
30763861d79fSDimitry Andric           Args.push_back(getValue(Record, OpNum, NextValueNo,
30773861d79fSDimitry Andric                                   FTy->getParamType(i)));
307891bc56edSDimitry Andric         if (!Args.back())
3079f785676fSDimitry Andric           return Error(InvalidRecord);
3080f22ef01cSRoman Divacky       }
3081f22ef01cSRoman Divacky 
3082f22ef01cSRoman Divacky       // Read type/value pairs for varargs params.
3083f22ef01cSRoman Divacky       if (!FTy->isVarArg()) {
3084f22ef01cSRoman Divacky         if (OpNum != Record.size())
3085f785676fSDimitry Andric           return Error(InvalidRecord);
3086f22ef01cSRoman Divacky       } else {
3087f22ef01cSRoman Divacky         while (OpNum != Record.size()) {
3088f22ef01cSRoman Divacky           Value *Op;
3089f22ef01cSRoman Divacky           if (getValueTypePair(Record, OpNum, NextValueNo, Op))
3090f785676fSDimitry Andric             return Error(InvalidRecord);
3091f22ef01cSRoman Divacky           Args.push_back(Op);
3092f22ef01cSRoman Divacky         }
3093f22ef01cSRoman Divacky       }
3094f22ef01cSRoman Divacky 
309517a519f9SDimitry Andric       I = CallInst::Create(Callee, Args);
3096f22ef01cSRoman Divacky       InstructionList.push_back(I);
3097f22ef01cSRoman Divacky       cast<CallInst>(I)->setCallingConv(
309891bc56edSDimitry Andric           static_cast<CallingConv::ID>((~(1U << 14) & CCInfo) >> 1));
309991bc56edSDimitry Andric       CallInst::TailCallKind TCK = CallInst::TCK_None;
310091bc56edSDimitry Andric       if (CCInfo & 1)
310191bc56edSDimitry Andric         TCK = CallInst::TCK_Tail;
310291bc56edSDimitry Andric       if (CCInfo & (1 << 14))
310391bc56edSDimitry Andric         TCK = CallInst::TCK_MustTail;
310491bc56edSDimitry Andric       cast<CallInst>(I)->setTailCallKind(TCK);
3105f22ef01cSRoman Divacky       cast<CallInst>(I)->setAttributes(PAL);
3106f22ef01cSRoman Divacky       break;
3107f22ef01cSRoman Divacky     }
3108f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_VAARG: { // VAARG: [valistty, valist, instty]
3109f22ef01cSRoman Divacky       if (Record.size() < 3)
3110f785676fSDimitry Andric         return Error(InvalidRecord);
31116122f3e6SDimitry Andric       Type *OpTy = getTypeByID(Record[0]);
31123861d79fSDimitry Andric       Value *Op = getValue(Record, 1, NextValueNo, OpTy);
31136122f3e6SDimitry Andric       Type *ResTy = getTypeByID(Record[2]);
3114f22ef01cSRoman Divacky       if (!OpTy || !Op || !ResTy)
3115f785676fSDimitry Andric         return Error(InvalidRecord);
3116f22ef01cSRoman Divacky       I = new VAArgInst(Op, ResTy);
3117f22ef01cSRoman Divacky       InstructionList.push_back(I);
3118f22ef01cSRoman Divacky       break;
3119f22ef01cSRoman Divacky     }
3120f22ef01cSRoman Divacky     }
3121f22ef01cSRoman Divacky 
3122f22ef01cSRoman Divacky     // Add instruction to end of current BB.  If there is no current BB, reject
3123f22ef01cSRoman Divacky     // this file.
312491bc56edSDimitry Andric     if (!CurBB) {
3125f22ef01cSRoman Divacky       delete I;
3126f785676fSDimitry Andric       return Error(InvalidInstructionWithNoBB);
3127f22ef01cSRoman Divacky     }
3128f22ef01cSRoman Divacky     CurBB->getInstList().push_back(I);
3129f22ef01cSRoman Divacky 
3130f22ef01cSRoman Divacky     // If this was a terminator instruction, move to the next block.
3131f22ef01cSRoman Divacky     if (isa<TerminatorInst>(I)) {
3132f22ef01cSRoman Divacky       ++CurBBNo;
313391bc56edSDimitry Andric       CurBB = CurBBNo < FunctionBBs.size() ? FunctionBBs[CurBBNo] : nullptr;
3134f22ef01cSRoman Divacky     }
3135f22ef01cSRoman Divacky 
3136f22ef01cSRoman Divacky     // Non-void values get registered in the value table for future use.
3137f22ef01cSRoman Divacky     if (I && !I->getType()->isVoidTy())
3138f22ef01cSRoman Divacky       ValueList.AssignValue(I, NextValueNo++);
3139f22ef01cSRoman Divacky   }
3140f22ef01cSRoman Divacky 
3141139f7f9bSDimitry Andric OutOfRecordLoop:
3142139f7f9bSDimitry Andric 
3143f22ef01cSRoman Divacky   // Check the function list for unresolved values.
3144f22ef01cSRoman Divacky   if (Argument *A = dyn_cast<Argument>(ValueList.back())) {
314591bc56edSDimitry Andric     if (!A->getParent()) {
3146f22ef01cSRoman Divacky       // We found at least one unresolved value.  Nuke them all to avoid leaks.
3147f22ef01cSRoman Divacky       for (unsigned i = ModuleValueListSize, e = ValueList.size(); i != e; ++i){
314891bc56edSDimitry Andric         if ((A = dyn_cast_or_null<Argument>(ValueList[i])) && !A->getParent()) {
3149f22ef01cSRoman Divacky           A->replaceAllUsesWith(UndefValue::get(A->getType()));
3150f22ef01cSRoman Divacky           delete A;
3151f22ef01cSRoman Divacky         }
3152f22ef01cSRoman Divacky       }
3153f785676fSDimitry Andric       return Error(NeverResolvedValueFoundInFunction);
3154f22ef01cSRoman Divacky     }
3155f22ef01cSRoman Divacky   }
3156f22ef01cSRoman Divacky 
3157e580952dSDimitry Andric   // FIXME: Check for unresolved forward-declared metadata references
3158e580952dSDimitry Andric   // and clean up leaks.
3159e580952dSDimitry Andric 
3160f22ef01cSRoman Divacky   // See if anything took the address of blocks in this function.  If so,
3161f22ef01cSRoman Divacky   // resolve them now.
3162f22ef01cSRoman Divacky   DenseMap<Function*, std::vector<BlockAddrRefTy> >::iterator BAFRI =
3163f22ef01cSRoman Divacky     BlockAddrFwdRefs.find(F);
3164f22ef01cSRoman Divacky   if (BAFRI != BlockAddrFwdRefs.end()) {
3165f22ef01cSRoman Divacky     std::vector<BlockAddrRefTy> &RefList = BAFRI->second;
3166f22ef01cSRoman Divacky     for (unsigned i = 0, e = RefList.size(); i != e; ++i) {
3167f22ef01cSRoman Divacky       unsigned BlockIdx = RefList[i].first;
3168f22ef01cSRoman Divacky       if (BlockIdx >= FunctionBBs.size())
3169f785676fSDimitry Andric         return Error(InvalidID);
3170f22ef01cSRoman Divacky 
3171f22ef01cSRoman Divacky       GlobalVariable *FwdRef = RefList[i].second;
3172f22ef01cSRoman Divacky       FwdRef->replaceAllUsesWith(BlockAddress::get(F, FunctionBBs[BlockIdx]));
3173f22ef01cSRoman Divacky       FwdRef->eraseFromParent();
3174f22ef01cSRoman Divacky     }
3175f22ef01cSRoman Divacky 
3176f22ef01cSRoman Divacky     BlockAddrFwdRefs.erase(BAFRI);
3177f22ef01cSRoman Divacky   }
3178f22ef01cSRoman Divacky 
3179f22ef01cSRoman Divacky   // Trim the value list down to the size it was before we parsed this function.
3180f22ef01cSRoman Divacky   ValueList.shrinkTo(ModuleValueListSize);
3181e580952dSDimitry Andric   MDValueList.shrinkTo(ModuleMDValueListSize);
3182f22ef01cSRoman Divacky   std::vector<BasicBlock*>().swap(FunctionBBs);
318391bc56edSDimitry Andric   return std::error_code();
3184f22ef01cSRoman Divacky }
3185f22ef01cSRoman Divacky 
3186f785676fSDimitry Andric /// Find the function body in the bitcode stream
318791bc56edSDimitry Andric std::error_code BitcodeReader::FindFunctionInStream(
318891bc56edSDimitry Andric     Function *F,
3189dff0c46cSDimitry Andric     DenseMap<Function *, uint64_t>::iterator DeferredFunctionInfoIterator) {
3190dff0c46cSDimitry Andric   while (DeferredFunctionInfoIterator->second == 0) {
3191dff0c46cSDimitry Andric     if (Stream.AtEndOfStream())
3192f785676fSDimitry Andric       return Error(CouldNotFindFunctionInStream);
3193dff0c46cSDimitry Andric     // ParseModule will parse the next body in the stream and set its
3194dff0c46cSDimitry Andric     // position in the DeferredFunctionInfo map.
319591bc56edSDimitry Andric     if (std::error_code EC = ParseModule(true))
3196f785676fSDimitry Andric       return EC;
3197dff0c46cSDimitry Andric   }
319891bc56edSDimitry Andric   return std::error_code();
3199dff0c46cSDimitry Andric }
3200dff0c46cSDimitry Andric 
3201f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
3202f22ef01cSRoman Divacky // GVMaterializer implementation
3203f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
3204f22ef01cSRoman Divacky 
320591bc56edSDimitry Andric void BitcodeReader::releaseBuffer() { Buffer.release(); }
3206f22ef01cSRoman Divacky 
3207f22ef01cSRoman Divacky bool BitcodeReader::isMaterializable(const GlobalValue *GV) const {
3208f22ef01cSRoman Divacky   if (const Function *F = dyn_cast<Function>(GV)) {
3209f22ef01cSRoman Divacky     return F->isDeclaration() &&
3210f22ef01cSRoman Divacky       DeferredFunctionInfo.count(const_cast<Function*>(F));
3211f22ef01cSRoman Divacky   }
3212f22ef01cSRoman Divacky   return false;
3213f22ef01cSRoman Divacky }
3214f22ef01cSRoman Divacky 
321591bc56edSDimitry Andric std::error_code BitcodeReader::Materialize(GlobalValue *GV) {
3216f22ef01cSRoman Divacky   Function *F = dyn_cast<Function>(GV);
3217f22ef01cSRoman Divacky   // If it's not a function or is already material, ignore the request.
3218f785676fSDimitry Andric   if (!F || !F->isMaterializable())
321991bc56edSDimitry Andric     return std::error_code();
3220f22ef01cSRoman Divacky 
3221f22ef01cSRoman Divacky   DenseMap<Function*, uint64_t>::iterator DFII = DeferredFunctionInfo.find(F);
3222f22ef01cSRoman Divacky   assert(DFII != DeferredFunctionInfo.end() && "Deferred function not found!");
3223dff0c46cSDimitry Andric   // If its position is recorded as 0, its body is somewhere in the stream
3224dff0c46cSDimitry Andric   // but we haven't seen it yet.
3225f785676fSDimitry Andric   if (DFII->second == 0 && LazyStreamer)
322691bc56edSDimitry Andric     if (std::error_code EC = FindFunctionInStream(F, DFII))
3227f785676fSDimitry Andric       return EC;
3228f22ef01cSRoman Divacky 
3229f22ef01cSRoman Divacky   // Move the bit stream to the saved position of the deferred function body.
3230f22ef01cSRoman Divacky   Stream.JumpToBit(DFII->second);
3231f22ef01cSRoman Divacky 
323291bc56edSDimitry Andric   if (std::error_code EC = ParseFunctionBody(F))
3233f785676fSDimitry Andric     return EC;
3234f22ef01cSRoman Divacky 
3235f22ef01cSRoman Divacky   // Upgrade any old intrinsic calls in the function.
3236f22ef01cSRoman Divacky   for (UpgradedIntrinsicMap::iterator I = UpgradedIntrinsics.begin(),
3237f22ef01cSRoman Divacky        E = UpgradedIntrinsics.end(); I != E; ++I) {
3238f22ef01cSRoman Divacky     if (I->first != I->second) {
323991bc56edSDimitry Andric       for (auto UI = I->first->user_begin(), UE = I->first->user_end();
324091bc56edSDimitry Andric            UI != UE;) {
3241f22ef01cSRoman Divacky         if (CallInst* CI = dyn_cast<CallInst>(*UI++))
3242f22ef01cSRoman Divacky           UpgradeIntrinsicCall(CI, I->second);
3243f22ef01cSRoman Divacky       }
3244f22ef01cSRoman Divacky     }
3245f22ef01cSRoman Divacky   }
3246f22ef01cSRoman Divacky 
324791bc56edSDimitry Andric   return std::error_code();
3248f22ef01cSRoman Divacky }
3249f22ef01cSRoman Divacky 
3250f22ef01cSRoman Divacky bool BitcodeReader::isDematerializable(const GlobalValue *GV) const {
3251f22ef01cSRoman Divacky   const Function *F = dyn_cast<Function>(GV);
3252f22ef01cSRoman Divacky   if (!F || F->isDeclaration())
3253f22ef01cSRoman Divacky     return false;
3254f22ef01cSRoman Divacky   return DeferredFunctionInfo.count(const_cast<Function*>(F));
3255f22ef01cSRoman Divacky }
3256f22ef01cSRoman Divacky 
3257f22ef01cSRoman Divacky void BitcodeReader::Dematerialize(GlobalValue *GV) {
3258f22ef01cSRoman Divacky   Function *F = dyn_cast<Function>(GV);
3259f22ef01cSRoman Divacky   // If this function isn't dematerializable, this is a noop.
3260f22ef01cSRoman Divacky   if (!F || !isDematerializable(F))
3261f22ef01cSRoman Divacky     return;
3262f22ef01cSRoman Divacky 
3263f22ef01cSRoman Divacky   assert(DeferredFunctionInfo.count(F) && "No info to read function later?");
3264f22ef01cSRoman Divacky 
3265f22ef01cSRoman Divacky   // Just forget the function body, we can remat it later.
3266f22ef01cSRoman Divacky   F->deleteBody();
3267f22ef01cSRoman Divacky }
3268f22ef01cSRoman Divacky 
326991bc56edSDimitry Andric std::error_code BitcodeReader::MaterializeModule(Module *M) {
3270f22ef01cSRoman Divacky   assert(M == TheModule &&
3271f22ef01cSRoman Divacky          "Can only Materialize the Module this BitcodeReader is attached to.");
3272f22ef01cSRoman Divacky   // Iterate over the module, deserializing any functions that are still on
3273f22ef01cSRoman Divacky   // disk.
3274f22ef01cSRoman Divacky   for (Module::iterator F = TheModule->begin(), E = TheModule->end();
3275f785676fSDimitry Andric        F != E; ++F) {
3276f785676fSDimitry Andric     if (F->isMaterializable()) {
327791bc56edSDimitry Andric       if (std::error_code EC = Materialize(F))
3278f785676fSDimitry Andric         return EC;
3279f785676fSDimitry Andric     }
3280f785676fSDimitry Andric   }
3281dff0c46cSDimitry Andric   // At this point, if there are any function bodies, the current bit is
3282dff0c46cSDimitry Andric   // pointing to the END_BLOCK record after them. Now make sure the rest
3283dff0c46cSDimitry Andric   // of the bits in the module have been read.
3284dff0c46cSDimitry Andric   if (NextUnreadBit)
3285dff0c46cSDimitry Andric     ParseModule(true);
3286dff0c46cSDimitry Andric 
3287f22ef01cSRoman Divacky   // Upgrade any intrinsic calls that slipped through (should not happen!) and
3288f22ef01cSRoman Divacky   // delete the old functions to clean up. We can't do this unless the entire
3289f22ef01cSRoman Divacky   // module is materialized because there could always be another function body
3290f22ef01cSRoman Divacky   // with calls to the old function.
3291f22ef01cSRoman Divacky   for (std::vector<std::pair<Function*, Function*> >::iterator I =
3292f22ef01cSRoman Divacky        UpgradedIntrinsics.begin(), E = UpgradedIntrinsics.end(); I != E; ++I) {
3293f22ef01cSRoman Divacky     if (I->first != I->second) {
329491bc56edSDimitry Andric       for (auto UI = I->first->user_begin(), UE = I->first->user_end();
329591bc56edSDimitry Andric            UI != UE;) {
3296f22ef01cSRoman Divacky         if (CallInst* CI = dyn_cast<CallInst>(*UI++))
3297f22ef01cSRoman Divacky           UpgradeIntrinsicCall(CI, I->second);
3298f22ef01cSRoman Divacky       }
3299f22ef01cSRoman Divacky       if (!I->first->use_empty())
3300f22ef01cSRoman Divacky         I->first->replaceAllUsesWith(I->second);
3301f22ef01cSRoman Divacky       I->first->eraseFromParent();
3302f22ef01cSRoman Divacky     }
3303f22ef01cSRoman Divacky   }
3304f22ef01cSRoman Divacky   std::vector<std::pair<Function*, Function*> >().swap(UpgradedIntrinsics);
3305f22ef01cSRoman Divacky 
3306f785676fSDimitry Andric   for (unsigned I = 0, E = InstsWithTBAATag.size(); I < E; I++)
3307f785676fSDimitry Andric     UpgradeInstWithTBAATag(InstsWithTBAATag[I]);
3308f785676fSDimitry Andric 
3309f785676fSDimitry Andric   UpgradeDebugInfo(*M);
331091bc56edSDimitry Andric   return std::error_code();
3311dff0c46cSDimitry Andric }
33126122f3e6SDimitry Andric 
331391bc56edSDimitry Andric std::error_code BitcodeReader::InitStream() {
3314f785676fSDimitry Andric   if (LazyStreamer)
3315f785676fSDimitry Andric     return InitLazyStream();
3316dff0c46cSDimitry Andric   return InitStreamFromBuffer();
3317dff0c46cSDimitry Andric }
3318dff0c46cSDimitry Andric 
331991bc56edSDimitry Andric std::error_code BitcodeReader::InitStreamFromBuffer() {
33203861d79fSDimitry Andric   const unsigned char *BufPtr = (const unsigned char*)Buffer->getBufferStart();
3321dff0c46cSDimitry Andric   const unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
3322dff0c46cSDimitry Andric 
3323dff0c46cSDimitry Andric   if (Buffer->getBufferSize() & 3) {
3324dff0c46cSDimitry Andric     if (!isRawBitcode(BufPtr, BufEnd) && !isBitcodeWrapper(BufPtr, BufEnd))
3325f785676fSDimitry Andric       return Error(InvalidBitcodeSignature);
3326dff0c46cSDimitry Andric     else
3327f785676fSDimitry Andric       return Error(BitcodeStreamInvalidSize);
3328dff0c46cSDimitry Andric   }
3329dff0c46cSDimitry Andric 
3330dff0c46cSDimitry Andric   // If we have a wrapper header, parse it and ignore the non-bc file contents.
3331dff0c46cSDimitry Andric   // The magic number is 0x0B17C0DE stored in little endian.
3332dff0c46cSDimitry Andric   if (isBitcodeWrapper(BufPtr, BufEnd))
3333dff0c46cSDimitry Andric     if (SkipBitcodeWrapperHeader(BufPtr, BufEnd, true))
3334f785676fSDimitry Andric       return Error(InvalidBitcodeWrapperHeader);
3335dff0c46cSDimitry Andric 
3336dff0c46cSDimitry Andric   StreamFile.reset(new BitstreamReader(BufPtr, BufEnd));
3337dff0c46cSDimitry Andric   Stream.init(*StreamFile);
3338f22ef01cSRoman Divacky 
333991bc56edSDimitry Andric   return std::error_code();
3340f22ef01cSRoman Divacky }
3341f22ef01cSRoman Divacky 
334291bc56edSDimitry Andric std::error_code BitcodeReader::InitLazyStream() {
3343dff0c46cSDimitry Andric   // Check and strip off the bitcode wrapper; BitstreamReader expects never to
3344dff0c46cSDimitry Andric   // see it.
3345dff0c46cSDimitry Andric   StreamingMemoryObject *Bytes = new StreamingMemoryObject(LazyStreamer);
3346dff0c46cSDimitry Andric   StreamFile.reset(new BitstreamReader(Bytes));
3347dff0c46cSDimitry Andric   Stream.init(*StreamFile);
3348dff0c46cSDimitry Andric 
3349dff0c46cSDimitry Andric   unsigned char buf[16];
3350f785676fSDimitry Andric   if (Bytes->readBytes(0, 16, buf) == -1)
3351f785676fSDimitry Andric     return Error(BitcodeStreamInvalidSize);
3352dff0c46cSDimitry Andric 
3353dff0c46cSDimitry Andric   if (!isBitcode(buf, buf + 16))
3354f785676fSDimitry Andric     return Error(InvalidBitcodeSignature);
3355dff0c46cSDimitry Andric 
3356dff0c46cSDimitry Andric   if (isBitcodeWrapper(buf, buf + 4)) {
3357dff0c46cSDimitry Andric     const unsigned char *bitcodeStart = buf;
3358dff0c46cSDimitry Andric     const unsigned char *bitcodeEnd = buf + 16;
3359dff0c46cSDimitry Andric     SkipBitcodeWrapperHeader(bitcodeStart, bitcodeEnd, false);
3360dff0c46cSDimitry Andric     Bytes->dropLeadingBytes(bitcodeStart - buf);
3361dff0c46cSDimitry Andric     Bytes->setKnownObjectSize(bitcodeEnd - bitcodeStart);
3362dff0c46cSDimitry Andric   }
336391bc56edSDimitry Andric   return std::error_code();
3364f785676fSDimitry Andric }
3365f785676fSDimitry Andric 
3366f785676fSDimitry Andric namespace {
336791bc56edSDimitry Andric class BitcodeErrorCategoryType : public std::error_category {
336891bc56edSDimitry Andric   const char *name() const LLVM_NOEXCEPT override {
3369f785676fSDimitry Andric     return "llvm.bitcode";
3370f785676fSDimitry Andric   }
337191bc56edSDimitry Andric   std::string message(int IE) const override {
3372f785676fSDimitry Andric     BitcodeReader::ErrorType E = static_cast<BitcodeReader::ErrorType>(IE);
3373f785676fSDimitry Andric     switch (E) {
3374f785676fSDimitry Andric     case BitcodeReader::BitcodeStreamInvalidSize:
3375f785676fSDimitry Andric       return "Bitcode stream length should be >= 16 bytes and a multiple of 4";
3376f785676fSDimitry Andric     case BitcodeReader::ConflictingMETADATA_KINDRecords:
3377f785676fSDimitry Andric       return "Conflicting METADATA_KIND records";
3378f785676fSDimitry Andric     case BitcodeReader::CouldNotFindFunctionInStream:
3379f785676fSDimitry Andric       return "Could not find function in stream";
3380f785676fSDimitry Andric     case BitcodeReader::ExpectedConstant:
3381f785676fSDimitry Andric       return "Expected a constant";
3382f785676fSDimitry Andric     case BitcodeReader::InsufficientFunctionProtos:
3383f785676fSDimitry Andric       return "Insufficient function protos";
3384f785676fSDimitry Andric     case BitcodeReader::InvalidBitcodeSignature:
3385f785676fSDimitry Andric       return "Invalid bitcode signature";
3386f785676fSDimitry Andric     case BitcodeReader::InvalidBitcodeWrapperHeader:
3387f785676fSDimitry Andric       return "Invalid bitcode wrapper header";
3388f785676fSDimitry Andric     case BitcodeReader::InvalidConstantReference:
3389f785676fSDimitry Andric       return "Invalid ronstant reference";
3390f785676fSDimitry Andric     case BitcodeReader::InvalidID:
3391f785676fSDimitry Andric       return "Invalid ID";
3392f785676fSDimitry Andric     case BitcodeReader::InvalidInstructionWithNoBB:
3393f785676fSDimitry Andric       return "Invalid instruction with no BB";
3394f785676fSDimitry Andric     case BitcodeReader::InvalidRecord:
3395f785676fSDimitry Andric       return "Invalid record";
3396f785676fSDimitry Andric     case BitcodeReader::InvalidTypeForValue:
3397f785676fSDimitry Andric       return "Invalid type for value";
3398f785676fSDimitry Andric     case BitcodeReader::InvalidTYPETable:
3399f785676fSDimitry Andric       return "Invalid TYPE table";
3400f785676fSDimitry Andric     case BitcodeReader::InvalidType:
3401f785676fSDimitry Andric       return "Invalid type";
3402f785676fSDimitry Andric     case BitcodeReader::MalformedBlock:
3403f785676fSDimitry Andric       return "Malformed block";
3404f785676fSDimitry Andric     case BitcodeReader::MalformedGlobalInitializerSet:
3405f785676fSDimitry Andric       return "Malformed global initializer set";
3406f785676fSDimitry Andric     case BitcodeReader::InvalidMultipleBlocks:
3407f785676fSDimitry Andric       return "Invalid multiple blocks";
3408f785676fSDimitry Andric     case BitcodeReader::NeverResolvedValueFoundInFunction:
3409f785676fSDimitry Andric       return "Never resolved value found in function";
3410f785676fSDimitry Andric     case BitcodeReader::InvalidValue:
3411f785676fSDimitry Andric       return "Invalid value";
3412f785676fSDimitry Andric     }
3413f785676fSDimitry Andric     llvm_unreachable("Unknown error type!");
3414f785676fSDimitry Andric   }
3415f785676fSDimitry Andric };
3416f785676fSDimitry Andric }
3417f785676fSDimitry Andric 
341891bc56edSDimitry Andric const std::error_category &BitcodeReader::BitcodeErrorCategory() {
3419f785676fSDimitry Andric   static BitcodeErrorCategoryType O;
3420f785676fSDimitry Andric   return O;
3421dff0c46cSDimitry Andric }
3422f22ef01cSRoman Divacky 
3423f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
3424f22ef01cSRoman Divacky // External interface
3425f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
3426f22ef01cSRoman Divacky 
3427f22ef01cSRoman Divacky /// getLazyBitcodeModule - lazy function-at-a-time loading from a file.
3428f22ef01cSRoman Divacky ///
342991bc56edSDimitry Andric ErrorOr<Module *> llvm::getLazyBitcodeModule(MemoryBuffer *Buffer,
343091bc56edSDimitry Andric                                              LLVMContext &Context) {
3431f22ef01cSRoman Divacky   Module *M = new Module(Buffer->getBufferIdentifier(), Context);
3432f22ef01cSRoman Divacky   BitcodeReader *R = new BitcodeReader(Buffer, Context);
3433f22ef01cSRoman Divacky   M->setMaterializer(R);
343491bc56edSDimitry Andric   if (std::error_code EC = R->ParseBitcodeInto(M)) {
343591bc56edSDimitry Andric     R->releaseBuffer(); // Never take ownership on error.
3436f22ef01cSRoman Divacky     delete M;  // Also deletes R.
343791bc56edSDimitry Andric     return EC;
3438f22ef01cSRoman Divacky   }
3439dff0c46cSDimitry Andric 
3440dff0c46cSDimitry Andric   R->materializeForwardReferencedFunctions();
3441dff0c46cSDimitry Andric 
3442dff0c46cSDimitry Andric   return M;
3443dff0c46cSDimitry Andric }
3444dff0c46cSDimitry Andric 
3445dff0c46cSDimitry Andric 
3446dff0c46cSDimitry Andric Module *llvm::getStreamedBitcodeModule(const std::string &name,
3447dff0c46cSDimitry Andric                                        DataStreamer *streamer,
3448dff0c46cSDimitry Andric                                        LLVMContext &Context,
3449dff0c46cSDimitry Andric                                        std::string *ErrMsg) {
3450dff0c46cSDimitry Andric   Module *M = new Module(name, Context);
3451dff0c46cSDimitry Andric   BitcodeReader *R = new BitcodeReader(streamer, Context);
3452dff0c46cSDimitry Andric   M->setMaterializer(R);
345391bc56edSDimitry Andric   if (std::error_code EC = R->ParseBitcodeInto(M)) {
3454dff0c46cSDimitry Andric     if (ErrMsg)
3455f785676fSDimitry Andric       *ErrMsg = EC.message();
3456dff0c46cSDimitry Andric     delete M;  // Also deletes R.
345791bc56edSDimitry Andric     return nullptr;
3458dff0c46cSDimitry Andric   }
3459f22ef01cSRoman Divacky   return M;
3460f22ef01cSRoman Divacky }
3461f22ef01cSRoman Divacky 
346291bc56edSDimitry Andric ErrorOr<Module *> llvm::parseBitcodeFile(MemoryBuffer *Buffer,
346391bc56edSDimitry Andric                                          LLVMContext &Context) {
346491bc56edSDimitry Andric   ErrorOr<Module *> ModuleOrErr = getLazyBitcodeModule(Buffer, Context);
346591bc56edSDimitry Andric   if (!ModuleOrErr)
346691bc56edSDimitry Andric     return ModuleOrErr;
346791bc56edSDimitry Andric   Module *M = ModuleOrErr.get();
3468f22ef01cSRoman Divacky   // Read in the entire module, and destroy the BitcodeReader.
346991bc56edSDimitry Andric   if (std::error_code EC = M->materializeAllPermanently(true)) {
3470f22ef01cSRoman Divacky     delete M;
347191bc56edSDimitry Andric     return EC;
3472f22ef01cSRoman Divacky   }
34732754fe60SDimitry Andric 
3474dff0c46cSDimitry Andric   // TODO: Restore the use-lists to the in-memory state when the bitcode was
3475dff0c46cSDimitry Andric   // written.  We must defer until the Module has been fully materialized.
3476dff0c46cSDimitry Andric 
3477f22ef01cSRoman Divacky   return M;
3478f22ef01cSRoman Divacky }
34792754fe60SDimitry Andric 
34802754fe60SDimitry Andric std::string llvm::getBitcodeTargetTriple(MemoryBuffer *Buffer,
348191bc56edSDimitry Andric                                          LLVMContext &Context) {
34822754fe60SDimitry Andric   BitcodeReader *R = new BitcodeReader(Buffer, Context);
348391bc56edSDimitry Andric   ErrorOr<std::string> Triple = R->parseTriple();
348491bc56edSDimitry Andric   R->releaseBuffer();
34852754fe60SDimitry Andric   delete R;
348691bc56edSDimitry Andric   if (Triple.getError())
348791bc56edSDimitry Andric     return "";
348891bc56edSDimitry Andric   return Triple.get();
34892754fe60SDimitry Andric }
3490