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 // This header defines the BitcodeReader class.
11f22ef01cSRoman Divacky //
12f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
13f22ef01cSRoman Divacky 
14f22ef01cSRoman Divacky #include "llvm/Bitcode/ReaderWriter.h"
15f22ef01cSRoman Divacky #include "BitcodeReader.h"
16f22ef01cSRoman Divacky #include "llvm/Constants.h"
17f22ef01cSRoman Divacky #include "llvm/DerivedTypes.h"
18f22ef01cSRoman Divacky #include "llvm/InlineAsm.h"
19f22ef01cSRoman Divacky #include "llvm/IntrinsicInst.h"
20f22ef01cSRoman Divacky #include "llvm/Module.h"
21f22ef01cSRoman Divacky #include "llvm/Operator.h"
22f22ef01cSRoman Divacky #include "llvm/AutoUpgrade.h"
23f22ef01cSRoman Divacky #include "llvm/ADT/SmallString.h"
24f22ef01cSRoman Divacky #include "llvm/ADT/SmallVector.h"
25f22ef01cSRoman Divacky #include "llvm/Support/MathExtras.h"
26f22ef01cSRoman Divacky #include "llvm/Support/MemoryBuffer.h"
27f22ef01cSRoman Divacky #include "llvm/OperandTraits.h"
28f22ef01cSRoman Divacky using namespace llvm;
29f22ef01cSRoman Divacky 
30f22ef01cSRoman Divacky void BitcodeReader::FreeState() {
31f22ef01cSRoman Divacky   if (BufferOwned)
32f22ef01cSRoman Divacky     delete Buffer;
33f22ef01cSRoman Divacky   Buffer = 0;
3417a519f9SDimitry Andric   std::vector<Type*>().swap(TypeList);
35f22ef01cSRoman Divacky   ValueList.clear();
36f22ef01cSRoman Divacky   MDValueList.clear();
37f22ef01cSRoman Divacky 
38f22ef01cSRoman Divacky   std::vector<AttrListPtr>().swap(MAttributes);
39f22ef01cSRoman Divacky   std::vector<BasicBlock*>().swap(FunctionBBs);
40f22ef01cSRoman Divacky   std::vector<Function*>().swap(FunctionsWithBodies);
41f22ef01cSRoman Divacky   DeferredFunctionInfo.clear();
42e580952dSDimitry Andric   MDKindMap.clear();
43f22ef01cSRoman Divacky }
44f22ef01cSRoman Divacky 
45f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
46f22ef01cSRoman Divacky //  Helper functions to implement forward reference resolution, etc.
47f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
48f22ef01cSRoman Divacky 
49f22ef01cSRoman Divacky /// ConvertToString - Convert a string from a record into an std::string, return
50f22ef01cSRoman Divacky /// true on failure.
51f22ef01cSRoman Divacky template<typename StrTy>
52f22ef01cSRoman Divacky static bool ConvertToString(SmallVector<uint64_t, 64> &Record, unsigned Idx,
53f22ef01cSRoman Divacky                             StrTy &Result) {
54f22ef01cSRoman Divacky   if (Idx > Record.size())
55f22ef01cSRoman Divacky     return true;
56f22ef01cSRoman Divacky 
57f22ef01cSRoman Divacky   for (unsigned i = Idx, e = Record.size(); i != e; ++i)
58f22ef01cSRoman Divacky     Result += (char)Record[i];
59f22ef01cSRoman Divacky   return false;
60f22ef01cSRoman Divacky }
61f22ef01cSRoman Divacky 
62f22ef01cSRoman Divacky static GlobalValue::LinkageTypes GetDecodedLinkage(unsigned Val) {
63f22ef01cSRoman Divacky   switch (Val) {
64f22ef01cSRoman Divacky   default: // Map unknown/new linkages to external
65f22ef01cSRoman Divacky   case 0:  return GlobalValue::ExternalLinkage;
66f22ef01cSRoman Divacky   case 1:  return GlobalValue::WeakAnyLinkage;
67f22ef01cSRoman Divacky   case 2:  return GlobalValue::AppendingLinkage;
68f22ef01cSRoman Divacky   case 3:  return GlobalValue::InternalLinkage;
69f22ef01cSRoman Divacky   case 4:  return GlobalValue::LinkOnceAnyLinkage;
70f22ef01cSRoman Divacky   case 5:  return GlobalValue::DLLImportLinkage;
71f22ef01cSRoman Divacky   case 6:  return GlobalValue::DLLExportLinkage;
72f22ef01cSRoman Divacky   case 7:  return GlobalValue::ExternalWeakLinkage;
73f22ef01cSRoman Divacky   case 8:  return GlobalValue::CommonLinkage;
74f22ef01cSRoman Divacky   case 9:  return GlobalValue::PrivateLinkage;
75f22ef01cSRoman Divacky   case 10: return GlobalValue::WeakODRLinkage;
76f22ef01cSRoman Divacky   case 11: return GlobalValue::LinkOnceODRLinkage;
77f22ef01cSRoman Divacky   case 12: return GlobalValue::AvailableExternallyLinkage;
78f22ef01cSRoman Divacky   case 13: return GlobalValue::LinkerPrivateLinkage;
79ffd1746dSEd Schouten   case 14: return GlobalValue::LinkerPrivateWeakLinkage;
80e580952dSDimitry Andric   case 15: return GlobalValue::LinkerPrivateWeakDefAutoLinkage;
81f22ef01cSRoman Divacky   }
82f22ef01cSRoman Divacky }
83f22ef01cSRoman Divacky 
84f22ef01cSRoman Divacky static GlobalValue::VisibilityTypes GetDecodedVisibility(unsigned Val) {
85f22ef01cSRoman Divacky   switch (Val) {
86f22ef01cSRoman Divacky   default: // Map unknown visibilities to default.
87f22ef01cSRoman Divacky   case 0: return GlobalValue::DefaultVisibility;
88f22ef01cSRoman Divacky   case 1: return GlobalValue::HiddenVisibility;
89f22ef01cSRoman Divacky   case 2: return GlobalValue::ProtectedVisibility;
90f22ef01cSRoman Divacky   }
91f22ef01cSRoman Divacky }
92f22ef01cSRoman Divacky 
93f22ef01cSRoman Divacky static int GetDecodedCastOpcode(unsigned Val) {
94f22ef01cSRoman Divacky   switch (Val) {
95f22ef01cSRoman Divacky   default: return -1;
96f22ef01cSRoman Divacky   case bitc::CAST_TRUNC   : return Instruction::Trunc;
97f22ef01cSRoman Divacky   case bitc::CAST_ZEXT    : return Instruction::ZExt;
98f22ef01cSRoman Divacky   case bitc::CAST_SEXT    : return Instruction::SExt;
99f22ef01cSRoman Divacky   case bitc::CAST_FPTOUI  : return Instruction::FPToUI;
100f22ef01cSRoman Divacky   case bitc::CAST_FPTOSI  : return Instruction::FPToSI;
101f22ef01cSRoman Divacky   case bitc::CAST_UITOFP  : return Instruction::UIToFP;
102f22ef01cSRoman Divacky   case bitc::CAST_SITOFP  : return Instruction::SIToFP;
103f22ef01cSRoman Divacky   case bitc::CAST_FPTRUNC : return Instruction::FPTrunc;
104f22ef01cSRoman Divacky   case bitc::CAST_FPEXT   : return Instruction::FPExt;
105f22ef01cSRoman Divacky   case bitc::CAST_PTRTOINT: return Instruction::PtrToInt;
106f22ef01cSRoman Divacky   case bitc::CAST_INTTOPTR: return Instruction::IntToPtr;
107f22ef01cSRoman Divacky   case bitc::CAST_BITCAST : return Instruction::BitCast;
108f22ef01cSRoman Divacky   }
109f22ef01cSRoman Divacky }
110f22ef01cSRoman Divacky static int GetDecodedBinaryOpcode(unsigned Val, const Type *Ty) {
111f22ef01cSRoman Divacky   switch (Val) {
112f22ef01cSRoman Divacky   default: return -1;
113f22ef01cSRoman Divacky   case bitc::BINOP_ADD:
114f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FAdd : Instruction::Add;
115f22ef01cSRoman Divacky   case bitc::BINOP_SUB:
116f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FSub : Instruction::Sub;
117f22ef01cSRoman Divacky   case bitc::BINOP_MUL:
118f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FMul : Instruction::Mul;
119f22ef01cSRoman Divacky   case bitc::BINOP_UDIV: return Instruction::UDiv;
120f22ef01cSRoman Divacky   case bitc::BINOP_SDIV:
121f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FDiv : Instruction::SDiv;
122f22ef01cSRoman Divacky   case bitc::BINOP_UREM: return Instruction::URem;
123f22ef01cSRoman Divacky   case bitc::BINOP_SREM:
124f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FRem : Instruction::SRem;
125f22ef01cSRoman Divacky   case bitc::BINOP_SHL:  return Instruction::Shl;
126f22ef01cSRoman Divacky   case bitc::BINOP_LSHR: return Instruction::LShr;
127f22ef01cSRoman Divacky   case bitc::BINOP_ASHR: return Instruction::AShr;
128f22ef01cSRoman Divacky   case bitc::BINOP_AND:  return Instruction::And;
129f22ef01cSRoman Divacky   case bitc::BINOP_OR:   return Instruction::Or;
130f22ef01cSRoman Divacky   case bitc::BINOP_XOR:  return Instruction::Xor;
131f22ef01cSRoman Divacky   }
132f22ef01cSRoman Divacky }
133f22ef01cSRoman Divacky 
134f22ef01cSRoman Divacky namespace llvm {
135f22ef01cSRoman Divacky namespace {
136f22ef01cSRoman Divacky   /// @brief A class for maintaining the slot number definition
137f22ef01cSRoman Divacky   /// as a placeholder for the actual definition for forward constants defs.
138f22ef01cSRoman Divacky   class ConstantPlaceHolder : public ConstantExpr {
139f22ef01cSRoman Divacky     void operator=(const ConstantPlaceHolder &); // DO NOT IMPLEMENT
140f22ef01cSRoman Divacky   public:
141f22ef01cSRoman Divacky     // allocate space for exactly one operand
142f22ef01cSRoman Divacky     void *operator new(size_t s) {
143f22ef01cSRoman Divacky       return User::operator new(s, 1);
144f22ef01cSRoman Divacky     }
145f22ef01cSRoman Divacky     explicit ConstantPlaceHolder(const Type *Ty, LLVMContext& Context)
146f22ef01cSRoman Divacky       : ConstantExpr(Ty, Instruction::UserOp1, &Op<0>(), 1) {
147f22ef01cSRoman Divacky       Op<0>() = UndefValue::get(Type::getInt32Ty(Context));
148f22ef01cSRoman Divacky     }
149f22ef01cSRoman Divacky 
150f22ef01cSRoman Divacky     /// @brief Methods to support type inquiry through isa, cast, and dyn_cast.
1512754fe60SDimitry Andric     //static inline bool classof(const ConstantPlaceHolder *) { return true; }
152f22ef01cSRoman Divacky     static bool classof(const Value *V) {
153f22ef01cSRoman Divacky       return isa<ConstantExpr>(V) &&
154f22ef01cSRoman Divacky              cast<ConstantExpr>(V)->getOpcode() == Instruction::UserOp1;
155f22ef01cSRoman Divacky     }
156f22ef01cSRoman Divacky 
157f22ef01cSRoman Divacky 
158f22ef01cSRoman Divacky     /// Provide fast operand accessors
159f22ef01cSRoman Divacky     //DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
160f22ef01cSRoman Divacky   };
161f22ef01cSRoman Divacky }
162f22ef01cSRoman Divacky 
163f22ef01cSRoman Divacky // FIXME: can we inherit this from ConstantExpr?
164f22ef01cSRoman Divacky template <>
1652754fe60SDimitry Andric struct OperandTraits<ConstantPlaceHolder> :
1662754fe60SDimitry Andric   public FixedNumOperandTraits<ConstantPlaceHolder, 1> {
167f22ef01cSRoman Divacky };
168f22ef01cSRoman Divacky }
169f22ef01cSRoman Divacky 
170f22ef01cSRoman Divacky 
171f22ef01cSRoman Divacky void BitcodeReaderValueList::AssignValue(Value *V, unsigned Idx) {
172f22ef01cSRoman Divacky   if (Idx == size()) {
173f22ef01cSRoman Divacky     push_back(V);
174f22ef01cSRoman Divacky     return;
175f22ef01cSRoman Divacky   }
176f22ef01cSRoman Divacky 
177f22ef01cSRoman Divacky   if (Idx >= size())
178f22ef01cSRoman Divacky     resize(Idx+1);
179f22ef01cSRoman Divacky 
180f22ef01cSRoman Divacky   WeakVH &OldV = ValuePtrs[Idx];
181f22ef01cSRoman Divacky   if (OldV == 0) {
182f22ef01cSRoman Divacky     OldV = V;
183f22ef01cSRoman Divacky     return;
184f22ef01cSRoman Divacky   }
185f22ef01cSRoman Divacky 
186f22ef01cSRoman Divacky   // Handle constants and non-constants (e.g. instrs) differently for
187f22ef01cSRoman Divacky   // efficiency.
188f22ef01cSRoman Divacky   if (Constant *PHC = dyn_cast<Constant>(&*OldV)) {
189f22ef01cSRoman Divacky     ResolveConstants.push_back(std::make_pair(PHC, Idx));
190f22ef01cSRoman Divacky     OldV = V;
191f22ef01cSRoman Divacky   } else {
192f22ef01cSRoman Divacky     // If there was a forward reference to this value, replace it.
193f22ef01cSRoman Divacky     Value *PrevVal = OldV;
194f22ef01cSRoman Divacky     OldV->replaceAllUsesWith(V);
195f22ef01cSRoman Divacky     delete PrevVal;
196f22ef01cSRoman Divacky   }
197f22ef01cSRoman Divacky }
198f22ef01cSRoman Divacky 
199f22ef01cSRoman Divacky 
200f22ef01cSRoman Divacky Constant *BitcodeReaderValueList::getConstantFwdRef(unsigned Idx,
201f22ef01cSRoman Divacky                                                     const Type *Ty) {
202f22ef01cSRoman Divacky   if (Idx >= size())
203f22ef01cSRoman Divacky     resize(Idx + 1);
204f22ef01cSRoman Divacky 
205f22ef01cSRoman Divacky   if (Value *V = ValuePtrs[Idx]) {
206f22ef01cSRoman Divacky     assert(Ty == V->getType() && "Type mismatch in constant table!");
207f22ef01cSRoman Divacky     return cast<Constant>(V);
208f22ef01cSRoman Divacky   }
209f22ef01cSRoman Divacky 
210f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
211f22ef01cSRoman Divacky   Constant *C = new ConstantPlaceHolder(Ty, Context);
212f22ef01cSRoman Divacky   ValuePtrs[Idx] = C;
213f22ef01cSRoman Divacky   return C;
214f22ef01cSRoman Divacky }
215f22ef01cSRoman Divacky 
216f22ef01cSRoman Divacky Value *BitcodeReaderValueList::getValueFwdRef(unsigned Idx, const Type *Ty) {
217f22ef01cSRoman Divacky   if (Idx >= size())
218f22ef01cSRoman Divacky     resize(Idx + 1);
219f22ef01cSRoman Divacky 
220f22ef01cSRoman Divacky   if (Value *V = ValuePtrs[Idx]) {
221f22ef01cSRoman Divacky     assert((Ty == 0 || Ty == V->getType()) && "Type mismatch in value table!");
222f22ef01cSRoman Divacky     return V;
223f22ef01cSRoman Divacky   }
224f22ef01cSRoman Divacky 
225f22ef01cSRoman Divacky   // No type specified, must be invalid reference.
226f22ef01cSRoman Divacky   if (Ty == 0) return 0;
227f22ef01cSRoman Divacky 
228f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
229f22ef01cSRoman Divacky   Value *V = new Argument(Ty);
230f22ef01cSRoman Divacky   ValuePtrs[Idx] = V;
231f22ef01cSRoman Divacky   return V;
232f22ef01cSRoman Divacky }
233f22ef01cSRoman Divacky 
234f22ef01cSRoman Divacky /// ResolveConstantForwardRefs - Once all constants are read, this method bulk
235f22ef01cSRoman Divacky /// resolves any forward references.  The idea behind this is that we sometimes
236f22ef01cSRoman Divacky /// get constants (such as large arrays) which reference *many* forward ref
237f22ef01cSRoman Divacky /// constants.  Replacing each of these causes a lot of thrashing when
238f22ef01cSRoman Divacky /// building/reuniquing the constant.  Instead of doing this, we look at all the
239f22ef01cSRoman Divacky /// uses and rewrite all the place holders at once for any constant that uses
240f22ef01cSRoman Divacky /// a placeholder.
241f22ef01cSRoman Divacky void BitcodeReaderValueList::ResolveConstantForwardRefs() {
242f22ef01cSRoman Divacky   // Sort the values by-pointer so that they are efficient to look up with a
243f22ef01cSRoman Divacky   // binary search.
244f22ef01cSRoman Divacky   std::sort(ResolveConstants.begin(), ResolveConstants.end());
245f22ef01cSRoman Divacky 
246f22ef01cSRoman Divacky   SmallVector<Constant*, 64> NewOps;
247f22ef01cSRoman Divacky 
248f22ef01cSRoman Divacky   while (!ResolveConstants.empty()) {
249f22ef01cSRoman Divacky     Value *RealVal = operator[](ResolveConstants.back().second);
250f22ef01cSRoman Divacky     Constant *Placeholder = ResolveConstants.back().first;
251f22ef01cSRoman Divacky     ResolveConstants.pop_back();
252f22ef01cSRoman Divacky 
253f22ef01cSRoman Divacky     // Loop over all users of the placeholder, updating them to reference the
254f22ef01cSRoman Divacky     // new value.  If they reference more than one placeholder, update them all
255f22ef01cSRoman Divacky     // at once.
256f22ef01cSRoman Divacky     while (!Placeholder->use_empty()) {
257f22ef01cSRoman Divacky       Value::use_iterator UI = Placeholder->use_begin();
258ffd1746dSEd Schouten       User *U = *UI;
259f22ef01cSRoman Divacky 
260f22ef01cSRoman Divacky       // If the using object isn't uniqued, just update the operands.  This
261f22ef01cSRoman Divacky       // handles instructions and initializers for global variables.
262ffd1746dSEd Schouten       if (!isa<Constant>(U) || isa<GlobalValue>(U)) {
263f22ef01cSRoman Divacky         UI.getUse().set(RealVal);
264f22ef01cSRoman Divacky         continue;
265f22ef01cSRoman Divacky       }
266f22ef01cSRoman Divacky 
267f22ef01cSRoman Divacky       // Otherwise, we have a constant that uses the placeholder.  Replace that
268f22ef01cSRoman Divacky       // constant with a new constant that has *all* placeholder uses updated.
269ffd1746dSEd Schouten       Constant *UserC = cast<Constant>(U);
270f22ef01cSRoman Divacky       for (User::op_iterator I = UserC->op_begin(), E = UserC->op_end();
271f22ef01cSRoman Divacky            I != E; ++I) {
272f22ef01cSRoman Divacky         Value *NewOp;
273f22ef01cSRoman Divacky         if (!isa<ConstantPlaceHolder>(*I)) {
274f22ef01cSRoman Divacky           // Not a placeholder reference.
275f22ef01cSRoman Divacky           NewOp = *I;
276f22ef01cSRoman Divacky         } else if (*I == Placeholder) {
277f22ef01cSRoman Divacky           // Common case is that it just references this one placeholder.
278f22ef01cSRoman Divacky           NewOp = RealVal;
279f22ef01cSRoman Divacky         } else {
280f22ef01cSRoman Divacky           // Otherwise, look up the placeholder in ResolveConstants.
281f22ef01cSRoman Divacky           ResolveConstantsTy::iterator It =
282f22ef01cSRoman Divacky             std::lower_bound(ResolveConstants.begin(), ResolveConstants.end(),
283f22ef01cSRoman Divacky                              std::pair<Constant*, unsigned>(cast<Constant>(*I),
284f22ef01cSRoman Divacky                                                             0));
285f22ef01cSRoman Divacky           assert(It != ResolveConstants.end() && It->first == *I);
286f22ef01cSRoman Divacky           NewOp = operator[](It->second);
287f22ef01cSRoman Divacky         }
288f22ef01cSRoman Divacky 
289f22ef01cSRoman Divacky         NewOps.push_back(cast<Constant>(NewOp));
290f22ef01cSRoman Divacky       }
291f22ef01cSRoman Divacky 
292f22ef01cSRoman Divacky       // Make the new constant.
293f22ef01cSRoman Divacky       Constant *NewC;
294f22ef01cSRoman Divacky       if (ConstantArray *UserCA = dyn_cast<ConstantArray>(UserC)) {
29517a519f9SDimitry Andric         NewC = ConstantArray::get(UserCA->getType(), NewOps);
296f22ef01cSRoman Divacky       } else if (ConstantStruct *UserCS = dyn_cast<ConstantStruct>(UserC)) {
29717a519f9SDimitry Andric         NewC = ConstantStruct::get(UserCS->getType(), NewOps);
298f22ef01cSRoman Divacky       } else if (isa<ConstantVector>(UserC)) {
2992754fe60SDimitry Andric         NewC = ConstantVector::get(NewOps);
300f22ef01cSRoman Divacky       } else {
301f22ef01cSRoman Divacky         assert(isa<ConstantExpr>(UserC) && "Must be a ConstantExpr.");
3023b0f4066SDimitry Andric         NewC = cast<ConstantExpr>(UserC)->getWithOperands(NewOps);
303f22ef01cSRoman Divacky       }
304f22ef01cSRoman Divacky 
305f22ef01cSRoman Divacky       UserC->replaceAllUsesWith(NewC);
306f22ef01cSRoman Divacky       UserC->destroyConstant();
307f22ef01cSRoman Divacky       NewOps.clear();
308f22ef01cSRoman Divacky     }
309f22ef01cSRoman Divacky 
310f22ef01cSRoman Divacky     // Update all ValueHandles, they should be the only users at this point.
311f22ef01cSRoman Divacky     Placeholder->replaceAllUsesWith(RealVal);
312f22ef01cSRoman Divacky     delete Placeholder;
313f22ef01cSRoman Divacky   }
314f22ef01cSRoman Divacky }
315f22ef01cSRoman Divacky 
316f22ef01cSRoman Divacky void BitcodeReaderMDValueList::AssignValue(Value *V, unsigned Idx) {
317f22ef01cSRoman Divacky   if (Idx == size()) {
318f22ef01cSRoman Divacky     push_back(V);
319f22ef01cSRoman Divacky     return;
320f22ef01cSRoman Divacky   }
321f22ef01cSRoman Divacky 
322f22ef01cSRoman Divacky   if (Idx >= size())
323f22ef01cSRoman Divacky     resize(Idx+1);
324f22ef01cSRoman Divacky 
325f22ef01cSRoman Divacky   WeakVH &OldV = MDValuePtrs[Idx];
326f22ef01cSRoman Divacky   if (OldV == 0) {
327f22ef01cSRoman Divacky     OldV = V;
328f22ef01cSRoman Divacky     return;
329f22ef01cSRoman Divacky   }
330f22ef01cSRoman Divacky 
331f22ef01cSRoman Divacky   // If there was a forward reference to this value, replace it.
332e580952dSDimitry Andric   MDNode *PrevVal = cast<MDNode>(OldV);
333f22ef01cSRoman Divacky   OldV->replaceAllUsesWith(V);
334e580952dSDimitry Andric   MDNode::deleteTemporary(PrevVal);
335f22ef01cSRoman Divacky   // Deleting PrevVal sets Idx value in MDValuePtrs to null. Set new
336f22ef01cSRoman Divacky   // value for Idx.
337f22ef01cSRoman Divacky   MDValuePtrs[Idx] = V;
338f22ef01cSRoman Divacky }
339f22ef01cSRoman Divacky 
340f22ef01cSRoman Divacky Value *BitcodeReaderMDValueList::getValueFwdRef(unsigned Idx) {
341f22ef01cSRoman Divacky   if (Idx >= size())
342f22ef01cSRoman Divacky     resize(Idx + 1);
343f22ef01cSRoman Divacky 
344f22ef01cSRoman Divacky   if (Value *V = MDValuePtrs[Idx]) {
345f22ef01cSRoman Divacky     assert(V->getType()->isMetadataTy() && "Type mismatch in value table!");
346f22ef01cSRoman Divacky     return V;
347f22ef01cSRoman Divacky   }
348f22ef01cSRoman Divacky 
349f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
3503b0f4066SDimitry Andric   Value *V = MDNode::getTemporary(Context, ArrayRef<Value*>());
351f22ef01cSRoman Divacky   MDValuePtrs[Idx] = V;
352f22ef01cSRoman Divacky   return V;
353f22ef01cSRoman Divacky }
354f22ef01cSRoman Divacky 
35517a519f9SDimitry Andric Type *BitcodeReader::getTypeByID(unsigned ID) {
35617a519f9SDimitry Andric   // The type table size is always specified correctly.
35717a519f9SDimitry Andric   if (ID >= TypeList.size())
35817a519f9SDimitry Andric     return 0;
359f22ef01cSRoman Divacky 
36017a519f9SDimitry Andric   if (Type *Ty = TypeList[ID])
36117a519f9SDimitry Andric     return Ty;
36217a519f9SDimitry Andric 
36317a519f9SDimitry Andric   // If we have a forward reference, the only possible case is when it is to a
36417a519f9SDimitry Andric   // named struct.  Just create a placeholder for now.
36517a519f9SDimitry Andric   return TypeList[ID] = StructType::createNamed(Context, "");
366f22ef01cSRoman Divacky }
367f22ef01cSRoman Divacky 
36817a519f9SDimitry Andric /// FIXME: Remove in LLVM 3.1, only used by ParseOldTypeTable.
36917a519f9SDimitry Andric Type *BitcodeReader::getTypeByIDOrNull(unsigned ID) {
37017a519f9SDimitry Andric   if (ID >= TypeList.size())
37117a519f9SDimitry Andric     TypeList.resize(ID+1);
37217a519f9SDimitry Andric 
37317a519f9SDimitry Andric   return TypeList[ID];
37417a519f9SDimitry Andric }
37517a519f9SDimitry Andric 
37617a519f9SDimitry Andric 
377f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
378f22ef01cSRoman Divacky //  Functions for parsing blocks from the bitcode file
379f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
380f22ef01cSRoman Divacky 
381f22ef01cSRoman Divacky bool BitcodeReader::ParseAttributeBlock() {
382f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::PARAMATTR_BLOCK_ID))
383f22ef01cSRoman Divacky     return Error("Malformed block record");
384f22ef01cSRoman Divacky 
385f22ef01cSRoman Divacky   if (!MAttributes.empty())
386f22ef01cSRoman Divacky     return Error("Multiple PARAMATTR blocks found!");
387f22ef01cSRoman Divacky 
388f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
389f22ef01cSRoman Divacky 
390f22ef01cSRoman Divacky   SmallVector<AttributeWithIndex, 8> Attrs;
391f22ef01cSRoman Divacky 
392f22ef01cSRoman Divacky   // Read all the records.
393f22ef01cSRoman Divacky   while (1) {
394f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
395f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
396f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
397f22ef01cSRoman Divacky         return Error("Error at end of PARAMATTR block");
398f22ef01cSRoman Divacky       return false;
399f22ef01cSRoman Divacky     }
400f22ef01cSRoman Divacky 
401f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
402f22ef01cSRoman Divacky       // No known subblocks, always skip them.
403f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
404f22ef01cSRoman Divacky       if (Stream.SkipBlock())
405f22ef01cSRoman Divacky         return Error("Malformed block record");
406f22ef01cSRoman Divacky       continue;
407f22ef01cSRoman Divacky     }
408f22ef01cSRoman Divacky 
409f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
410f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
411f22ef01cSRoman Divacky       continue;
412f22ef01cSRoman Divacky     }
413f22ef01cSRoman Divacky 
414f22ef01cSRoman Divacky     // Read a record.
415f22ef01cSRoman Divacky     Record.clear();
416f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
417f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
418f22ef01cSRoman Divacky       break;
419f22ef01cSRoman Divacky     case bitc::PARAMATTR_CODE_ENTRY: { // ENTRY: [paramidx0, attr0, ...]
420f22ef01cSRoman Divacky       if (Record.size() & 1)
421f22ef01cSRoman Divacky         return Error("Invalid ENTRY record");
422f22ef01cSRoman Divacky 
423f22ef01cSRoman Divacky       // FIXME : Remove this autoupgrade code in LLVM 3.0.
424f22ef01cSRoman Divacky       // If Function attributes are using index 0 then transfer them
425f22ef01cSRoman Divacky       // to index ~0. Index 0 is used for return value attributes but used to be
426f22ef01cSRoman Divacky       // used for function attributes.
427f22ef01cSRoman Divacky       Attributes RetAttribute = Attribute::None;
428f22ef01cSRoman Divacky       Attributes FnAttribute = Attribute::None;
429f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
430f22ef01cSRoman Divacky         // FIXME: remove in LLVM 3.0
431f22ef01cSRoman Divacky         // The alignment is stored as a 16-bit raw value from bits 31--16.
432f22ef01cSRoman Divacky         // We shift the bits above 31 down by 11 bits.
433f22ef01cSRoman Divacky 
434f22ef01cSRoman Divacky         unsigned Alignment = (Record[i+1] & (0xffffull << 16)) >> 16;
435f22ef01cSRoman Divacky         if (Alignment && !isPowerOf2_32(Alignment))
436f22ef01cSRoman Divacky           return Error("Alignment is not a power of two.");
437f22ef01cSRoman Divacky 
438f22ef01cSRoman Divacky         Attributes ReconstitutedAttr = Record[i+1] & 0xffff;
439f22ef01cSRoman Divacky         if (Alignment)
440f22ef01cSRoman Divacky           ReconstitutedAttr |= Attribute::constructAlignmentFromInt(Alignment);
441f22ef01cSRoman Divacky         ReconstitutedAttr |= (Record[i+1] & (0xffffull << 32)) >> 11;
442f22ef01cSRoman Divacky         Record[i+1] = ReconstitutedAttr;
443f22ef01cSRoman Divacky 
444f22ef01cSRoman Divacky         if (Record[i] == 0)
445f22ef01cSRoman Divacky           RetAttribute = Record[i+1];
446f22ef01cSRoman Divacky         else if (Record[i] == ~0U)
447f22ef01cSRoman Divacky           FnAttribute = Record[i+1];
448f22ef01cSRoman Divacky       }
449f22ef01cSRoman Divacky 
450f22ef01cSRoman Divacky       unsigned OldRetAttrs = (Attribute::NoUnwind|Attribute::NoReturn|
451f22ef01cSRoman Divacky                               Attribute::ReadOnly|Attribute::ReadNone);
452f22ef01cSRoman Divacky 
453f22ef01cSRoman Divacky       if (FnAttribute == Attribute::None && RetAttribute != Attribute::None &&
454f22ef01cSRoman Divacky           (RetAttribute & OldRetAttrs) != 0) {
455f22ef01cSRoman Divacky         if (FnAttribute == Attribute::None) { // add a slot so they get added.
456f22ef01cSRoman Divacky           Record.push_back(~0U);
457f22ef01cSRoman Divacky           Record.push_back(0);
458f22ef01cSRoman Divacky         }
459f22ef01cSRoman Divacky 
460f22ef01cSRoman Divacky         FnAttribute  |= RetAttribute & OldRetAttrs;
461f22ef01cSRoman Divacky         RetAttribute &= ~OldRetAttrs;
462f22ef01cSRoman Divacky       }
463f22ef01cSRoman Divacky 
464f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
465f22ef01cSRoman Divacky         if (Record[i] == 0) {
466f22ef01cSRoman Divacky           if (RetAttribute != Attribute::None)
467f22ef01cSRoman Divacky             Attrs.push_back(AttributeWithIndex::get(0, RetAttribute));
468f22ef01cSRoman Divacky         } else if (Record[i] == ~0U) {
469f22ef01cSRoman Divacky           if (FnAttribute != Attribute::None)
470f22ef01cSRoman Divacky             Attrs.push_back(AttributeWithIndex::get(~0U, FnAttribute));
471f22ef01cSRoman Divacky         } else if (Record[i+1] != Attribute::None)
472f22ef01cSRoman Divacky           Attrs.push_back(AttributeWithIndex::get(Record[i], Record[i+1]));
473f22ef01cSRoman Divacky       }
474f22ef01cSRoman Divacky 
475f22ef01cSRoman Divacky       MAttributes.push_back(AttrListPtr::get(Attrs.begin(), Attrs.end()));
476f22ef01cSRoman Divacky       Attrs.clear();
477f22ef01cSRoman Divacky       break;
478f22ef01cSRoman Divacky     }
479f22ef01cSRoman Divacky     }
480f22ef01cSRoman Divacky   }
481f22ef01cSRoman Divacky }
482f22ef01cSRoman Divacky 
483f22ef01cSRoman Divacky bool BitcodeReader::ParseTypeTable() {
48417a519f9SDimitry Andric   if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID_NEW))
485f22ef01cSRoman Divacky     return Error("Malformed block record");
486f22ef01cSRoman Divacky 
48717a519f9SDimitry Andric   return ParseTypeTableBody();
48817a519f9SDimitry Andric }
48917a519f9SDimitry Andric 
49017a519f9SDimitry Andric bool BitcodeReader::ParseTypeTableBody() {
491f22ef01cSRoman Divacky   if (!TypeList.empty())
492f22ef01cSRoman Divacky     return Error("Multiple TYPE_BLOCKs found!");
493f22ef01cSRoman Divacky 
494f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
495f22ef01cSRoman Divacky   unsigned NumRecords = 0;
496f22ef01cSRoman Divacky 
49717a519f9SDimitry Andric   SmallString<64> TypeName;
49817a519f9SDimitry Andric 
499f22ef01cSRoman Divacky   // Read all the records for this type table.
500f22ef01cSRoman Divacky   while (1) {
501f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
502f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
503f22ef01cSRoman Divacky       if (NumRecords != TypeList.size())
504f22ef01cSRoman Divacky         return Error("Invalid type forward reference in TYPE_BLOCK");
505f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
506f22ef01cSRoman Divacky         return Error("Error at end of type table block");
507f22ef01cSRoman Divacky       return false;
508f22ef01cSRoman Divacky     }
509f22ef01cSRoman Divacky 
510f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
511f22ef01cSRoman Divacky       // No known subblocks, always skip them.
512f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
513f22ef01cSRoman Divacky       if (Stream.SkipBlock())
514f22ef01cSRoman Divacky         return Error("Malformed block record");
515f22ef01cSRoman Divacky       continue;
516f22ef01cSRoman Divacky     }
517f22ef01cSRoman Divacky 
518f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
519f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
520f22ef01cSRoman Divacky       continue;
521f22ef01cSRoman Divacky     }
522f22ef01cSRoman Divacky 
523f22ef01cSRoman Divacky     // Read a record.
524f22ef01cSRoman Divacky     Record.clear();
52517a519f9SDimitry Andric     Type *ResultTy = 0;
526f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
52717a519f9SDimitry Andric     default: return Error("unknown type in type table");
528f22ef01cSRoman Divacky     case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
529f22ef01cSRoman Divacky       // TYPE_CODE_NUMENTRY contains a count of the number of types in the
530f22ef01cSRoman Divacky       // type list.  This allows us to reserve space.
531f22ef01cSRoman Divacky       if (Record.size() < 1)
532f22ef01cSRoman Divacky         return Error("Invalid TYPE_CODE_NUMENTRY record");
53317a519f9SDimitry Andric       TypeList.resize(Record[0]);
534f22ef01cSRoman Divacky       continue;
535f22ef01cSRoman Divacky     case bitc::TYPE_CODE_VOID:      // VOID
536f22ef01cSRoman Divacky       ResultTy = Type::getVoidTy(Context);
537f22ef01cSRoman Divacky       break;
538f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FLOAT:     // FLOAT
539f22ef01cSRoman Divacky       ResultTy = Type::getFloatTy(Context);
540f22ef01cSRoman Divacky       break;
541f22ef01cSRoman Divacky     case bitc::TYPE_CODE_DOUBLE:    // DOUBLE
542f22ef01cSRoman Divacky       ResultTy = Type::getDoubleTy(Context);
543f22ef01cSRoman Divacky       break;
544f22ef01cSRoman Divacky     case bitc::TYPE_CODE_X86_FP80:  // X86_FP80
545f22ef01cSRoman Divacky       ResultTy = Type::getX86_FP80Ty(Context);
546f22ef01cSRoman Divacky       break;
547f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FP128:     // FP128
548f22ef01cSRoman Divacky       ResultTy = Type::getFP128Ty(Context);
549f22ef01cSRoman Divacky       break;
550f22ef01cSRoman Divacky     case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
551f22ef01cSRoman Divacky       ResultTy = Type::getPPC_FP128Ty(Context);
552f22ef01cSRoman Divacky       break;
553f22ef01cSRoman Divacky     case bitc::TYPE_CODE_LABEL:     // LABEL
554f22ef01cSRoman Divacky       ResultTy = Type::getLabelTy(Context);
555f22ef01cSRoman Divacky       break;
556f22ef01cSRoman Divacky     case bitc::TYPE_CODE_METADATA:  // METADATA
557f22ef01cSRoman Divacky       ResultTy = Type::getMetadataTy(Context);
558f22ef01cSRoman Divacky       break;
5592754fe60SDimitry Andric     case bitc::TYPE_CODE_X86_MMX:   // X86_MMX
5602754fe60SDimitry Andric       ResultTy = Type::getX86_MMXTy(Context);
5612754fe60SDimitry Andric       break;
562f22ef01cSRoman Divacky     case bitc::TYPE_CODE_INTEGER:   // INTEGER: [width]
563f22ef01cSRoman Divacky       if (Record.size() < 1)
564f22ef01cSRoman Divacky         return Error("Invalid Integer type record");
565f22ef01cSRoman Divacky 
566f22ef01cSRoman Divacky       ResultTy = IntegerType::get(Context, Record[0]);
567f22ef01cSRoman Divacky       break;
568f22ef01cSRoman Divacky     case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
569f22ef01cSRoman Divacky                                     //          [pointee type, address space]
570f22ef01cSRoman Divacky       if (Record.size() < 1)
571f22ef01cSRoman Divacky         return Error("Invalid POINTER type record");
572f22ef01cSRoman Divacky       unsigned AddressSpace = 0;
573f22ef01cSRoman Divacky       if (Record.size() == 2)
574f22ef01cSRoman Divacky         AddressSpace = Record[1];
57517a519f9SDimitry Andric       ResultTy = getTypeByID(Record[0]);
57617a519f9SDimitry Andric       if (ResultTy == 0) return Error("invalid element type in pointer type");
57717a519f9SDimitry Andric       ResultTy = PointerType::get(ResultTy, AddressSpace);
578f22ef01cSRoman Divacky       break;
579f22ef01cSRoman Divacky     }
580f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FUNCTION: {
581f22ef01cSRoman Divacky       // FIXME: attrid is dead, remove it in LLVM 3.0
582f22ef01cSRoman Divacky       // FUNCTION: [vararg, attrid, retty, paramty x N]
583f22ef01cSRoman Divacky       if (Record.size() < 3)
584f22ef01cSRoman Divacky         return Error("Invalid FUNCTION type record");
58517a519f9SDimitry Andric       std::vector<Type*> ArgTys;
58617a519f9SDimitry Andric       for (unsigned i = 3, e = Record.size(); i != e; ++i) {
58717a519f9SDimitry Andric         if (Type *T = getTypeByID(Record[i]))
58817a519f9SDimitry Andric           ArgTys.push_back(T);
58917a519f9SDimitry Andric         else
590f22ef01cSRoman Divacky           break;
591f22ef01cSRoman Divacky       }
59217a519f9SDimitry Andric 
59317a519f9SDimitry Andric       ResultTy = getTypeByID(Record[2]);
59417a519f9SDimitry Andric       if (ResultTy == 0 || ArgTys.size() < Record.size()-3)
59517a519f9SDimitry Andric         return Error("invalid type in function type");
59617a519f9SDimitry Andric 
59717a519f9SDimitry Andric       ResultTy = FunctionType::get(ResultTy, ArgTys, Record[0]);
59817a519f9SDimitry Andric       break;
59917a519f9SDimitry Andric     }
60017a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_ANON: {  // STRUCT: [ispacked, eltty x N]
601f22ef01cSRoman Divacky       if (Record.size() < 1)
602f22ef01cSRoman Divacky         return Error("Invalid STRUCT type record");
60317a519f9SDimitry Andric       std::vector<Type*> EltTys;
60417a519f9SDimitry Andric       for (unsigned i = 1, e = Record.size(); i != e; ++i) {
60517a519f9SDimitry Andric         if (Type *T = getTypeByID(Record[i]))
60617a519f9SDimitry Andric           EltTys.push_back(T);
60717a519f9SDimitry Andric         else
60817a519f9SDimitry Andric           break;
60917a519f9SDimitry Andric       }
61017a519f9SDimitry Andric       if (EltTys.size() != Record.size()-1)
61117a519f9SDimitry Andric         return Error("invalid type in struct type");
612f22ef01cSRoman Divacky       ResultTy = StructType::get(Context, EltTys, Record[0]);
613f22ef01cSRoman Divacky       break;
614f22ef01cSRoman Divacky     }
61517a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_NAME:   // STRUCT_NAME: [strchr x N]
61617a519f9SDimitry Andric       if (ConvertToString(Record, 0, TypeName))
61717a519f9SDimitry Andric         return Error("Invalid STRUCT_NAME record");
61817a519f9SDimitry Andric       continue;
61917a519f9SDimitry Andric 
62017a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_NAMED: { // STRUCT: [ispacked, eltty x N]
62117a519f9SDimitry Andric       if (Record.size() < 1)
62217a519f9SDimitry Andric         return Error("Invalid STRUCT type record");
62317a519f9SDimitry Andric 
62417a519f9SDimitry Andric       if (NumRecords >= TypeList.size())
62517a519f9SDimitry Andric         return Error("invalid TYPE table");
62617a519f9SDimitry Andric 
62717a519f9SDimitry Andric       // Check to see if this was forward referenced, if so fill in the temp.
62817a519f9SDimitry Andric       StructType *Res = cast_or_null<StructType>(TypeList[NumRecords]);
62917a519f9SDimitry Andric       if (Res) {
63017a519f9SDimitry Andric         Res->setName(TypeName);
63117a519f9SDimitry Andric         TypeList[NumRecords] = 0;
63217a519f9SDimitry Andric       } else  // Otherwise, create a new struct.
63317a519f9SDimitry Andric         Res = StructType::createNamed(Context, TypeName);
63417a519f9SDimitry Andric       TypeName.clear();
63517a519f9SDimitry Andric 
63617a519f9SDimitry Andric       SmallVector<Type*, 8> EltTys;
63717a519f9SDimitry Andric       for (unsigned i = 1, e = Record.size(); i != e; ++i) {
63817a519f9SDimitry Andric         if (Type *T = getTypeByID(Record[i]))
63917a519f9SDimitry Andric           EltTys.push_back(T);
64017a519f9SDimitry Andric         else
64117a519f9SDimitry Andric           break;
64217a519f9SDimitry Andric       }
64317a519f9SDimitry Andric       if (EltTys.size() != Record.size()-1)
64417a519f9SDimitry Andric         return Error("invalid STRUCT type record");
64517a519f9SDimitry Andric       Res->setBody(EltTys, Record[0]);
64617a519f9SDimitry Andric       ResultTy = Res;
64717a519f9SDimitry Andric       break;
64817a519f9SDimitry Andric     }
64917a519f9SDimitry Andric     case bitc::TYPE_CODE_OPAQUE: {       // OPAQUE: []
65017a519f9SDimitry Andric       if (Record.size() != 1)
65117a519f9SDimitry Andric         return Error("Invalid OPAQUE type record");
65217a519f9SDimitry Andric 
65317a519f9SDimitry Andric       if (NumRecords >= TypeList.size())
65417a519f9SDimitry Andric         return Error("invalid TYPE table");
65517a519f9SDimitry Andric 
65617a519f9SDimitry Andric       // Check to see if this was forward referenced, if so fill in the temp.
65717a519f9SDimitry Andric       StructType *Res = cast_or_null<StructType>(TypeList[NumRecords]);
65817a519f9SDimitry Andric       if (Res) {
65917a519f9SDimitry Andric         Res->setName(TypeName);
66017a519f9SDimitry Andric         TypeList[NumRecords] = 0;
66117a519f9SDimitry Andric       } else  // Otherwise, create a new struct with no body.
66217a519f9SDimitry Andric         Res = StructType::createNamed(Context, TypeName);
66317a519f9SDimitry Andric       TypeName.clear();
66417a519f9SDimitry Andric       ResultTy = Res;
66517a519f9SDimitry Andric       break;
66617a519f9SDimitry Andric     }
667f22ef01cSRoman Divacky     case bitc::TYPE_CODE_ARRAY:     // ARRAY: [numelts, eltty]
668f22ef01cSRoman Divacky       if (Record.size() < 2)
669f22ef01cSRoman Divacky         return Error("Invalid ARRAY type record");
67017a519f9SDimitry Andric       if ((ResultTy = getTypeByID(Record[1])))
67117a519f9SDimitry Andric         ResultTy = ArrayType::get(ResultTy, Record[0]);
67217a519f9SDimitry Andric       else
67317a519f9SDimitry Andric         return Error("Invalid ARRAY type element");
674f22ef01cSRoman Divacky       break;
675f22ef01cSRoman Divacky     case bitc::TYPE_CODE_VECTOR:    // VECTOR: [numelts, eltty]
676f22ef01cSRoman Divacky       if (Record.size() < 2)
677f22ef01cSRoman Divacky         return Error("Invalid VECTOR type record");
67817a519f9SDimitry Andric       if ((ResultTy = getTypeByID(Record[1])))
67917a519f9SDimitry Andric         ResultTy = VectorType::get(ResultTy, Record[0]);
68017a519f9SDimitry Andric       else
68117a519f9SDimitry Andric         return Error("Invalid ARRAY type element");
682f22ef01cSRoman Divacky       break;
683f22ef01cSRoman Divacky     }
684f22ef01cSRoman Divacky 
68517a519f9SDimitry Andric     if (NumRecords >= TypeList.size())
68617a519f9SDimitry Andric       return Error("invalid TYPE table");
68717a519f9SDimitry Andric     assert(ResultTy && "Didn't read a type?");
68817a519f9SDimitry Andric     assert(TypeList[NumRecords] == 0 && "Already read type?");
68917a519f9SDimitry Andric     TypeList[NumRecords++] = ResultTy;
690f22ef01cSRoman Divacky   }
691f22ef01cSRoman Divacky }
69217a519f9SDimitry Andric 
69317a519f9SDimitry Andric // FIXME: Remove in LLVM 3.1
69417a519f9SDimitry Andric bool BitcodeReader::ParseOldTypeTable() {
69517a519f9SDimitry Andric   if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID_OLD))
69617a519f9SDimitry Andric     return Error("Malformed block record");
69717a519f9SDimitry Andric 
69817a519f9SDimitry Andric   if (!TypeList.empty())
69917a519f9SDimitry Andric     return Error("Multiple TYPE_BLOCKs found!");
70017a519f9SDimitry Andric 
70117a519f9SDimitry Andric 
70217a519f9SDimitry Andric   // While horrible, we have no good ordering of types in the bc file.  Just
70317a519f9SDimitry Andric   // iteratively parse types out of the bc file in multiple passes until we get
70417a519f9SDimitry Andric   // them all.  Do this by saving a cursor for the start of the type block.
70517a519f9SDimitry Andric   BitstreamCursor StartOfTypeBlockCursor(Stream);
70617a519f9SDimitry Andric 
70717a519f9SDimitry Andric   unsigned NumTypesRead = 0;
70817a519f9SDimitry Andric 
70917a519f9SDimitry Andric   SmallVector<uint64_t, 64> Record;
71017a519f9SDimitry Andric RestartScan:
71117a519f9SDimitry Andric   unsigned NextTypeID = 0;
71217a519f9SDimitry Andric   bool ReadAnyTypes = false;
71317a519f9SDimitry Andric 
71417a519f9SDimitry Andric   // Read all the records for this type table.
71517a519f9SDimitry Andric   while (1) {
71617a519f9SDimitry Andric     unsigned Code = Stream.ReadCode();
71717a519f9SDimitry Andric     if (Code == bitc::END_BLOCK) {
71817a519f9SDimitry Andric       if (NextTypeID != TypeList.size())
71917a519f9SDimitry Andric         return Error("Invalid type forward reference in TYPE_BLOCK_ID_OLD");
72017a519f9SDimitry Andric 
72117a519f9SDimitry Andric       // If we haven't read all of the types yet, iterate again.
72217a519f9SDimitry Andric       if (NumTypesRead != TypeList.size()) {
72317a519f9SDimitry Andric         // If we didn't successfully read any types in this pass, then we must
72417a519f9SDimitry Andric         // have an unhandled forward reference.
72517a519f9SDimitry Andric         if (!ReadAnyTypes)
72617a519f9SDimitry Andric           return Error("Obsolete bitcode contains unhandled recursive type");
72717a519f9SDimitry Andric 
72817a519f9SDimitry Andric         Stream = StartOfTypeBlockCursor;
72917a519f9SDimitry Andric         goto RestartScan;
73017a519f9SDimitry Andric       }
73117a519f9SDimitry Andric 
73217a519f9SDimitry Andric       if (Stream.ReadBlockEnd())
73317a519f9SDimitry Andric         return Error("Error at end of type table block");
73417a519f9SDimitry Andric       return false;
73517a519f9SDimitry Andric     }
73617a519f9SDimitry Andric 
73717a519f9SDimitry Andric     if (Code == bitc::ENTER_SUBBLOCK) {
73817a519f9SDimitry Andric       // No known subblocks, always skip them.
73917a519f9SDimitry Andric       Stream.ReadSubBlockID();
74017a519f9SDimitry Andric       if (Stream.SkipBlock())
74117a519f9SDimitry Andric         return Error("Malformed block record");
74217a519f9SDimitry Andric       continue;
74317a519f9SDimitry Andric     }
74417a519f9SDimitry Andric 
74517a519f9SDimitry Andric     if (Code == bitc::DEFINE_ABBREV) {
74617a519f9SDimitry Andric       Stream.ReadAbbrevRecord();
74717a519f9SDimitry Andric       continue;
74817a519f9SDimitry Andric     }
74917a519f9SDimitry Andric 
75017a519f9SDimitry Andric     // Read a record.
75117a519f9SDimitry Andric     Record.clear();
75217a519f9SDimitry Andric     Type *ResultTy = 0;
75317a519f9SDimitry Andric     switch (Stream.ReadRecord(Code, Record)) {
75417a519f9SDimitry Andric     default: return Error("unknown type in type table");
75517a519f9SDimitry Andric     case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
75617a519f9SDimitry Andric       // TYPE_CODE_NUMENTRY contains a count of the number of types in the
75717a519f9SDimitry Andric       // type list.  This allows us to reserve space.
75817a519f9SDimitry Andric       if (Record.size() < 1)
75917a519f9SDimitry Andric         return Error("Invalid TYPE_CODE_NUMENTRY record");
76017a519f9SDimitry Andric       TypeList.resize(Record[0]);
76117a519f9SDimitry Andric       continue;
76217a519f9SDimitry Andric     case bitc::TYPE_CODE_VOID:      // VOID
76317a519f9SDimitry Andric       ResultTy = Type::getVoidTy(Context);
76417a519f9SDimitry Andric       break;
76517a519f9SDimitry Andric     case bitc::TYPE_CODE_FLOAT:     // FLOAT
76617a519f9SDimitry Andric       ResultTy = Type::getFloatTy(Context);
76717a519f9SDimitry Andric       break;
76817a519f9SDimitry Andric     case bitc::TYPE_CODE_DOUBLE:    // DOUBLE
76917a519f9SDimitry Andric       ResultTy = Type::getDoubleTy(Context);
77017a519f9SDimitry Andric       break;
77117a519f9SDimitry Andric     case bitc::TYPE_CODE_X86_FP80:  // X86_FP80
77217a519f9SDimitry Andric       ResultTy = Type::getX86_FP80Ty(Context);
77317a519f9SDimitry Andric       break;
77417a519f9SDimitry Andric     case bitc::TYPE_CODE_FP128:     // FP128
77517a519f9SDimitry Andric       ResultTy = Type::getFP128Ty(Context);
77617a519f9SDimitry Andric       break;
77717a519f9SDimitry Andric     case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
77817a519f9SDimitry Andric       ResultTy = Type::getPPC_FP128Ty(Context);
77917a519f9SDimitry Andric       break;
78017a519f9SDimitry Andric     case bitc::TYPE_CODE_LABEL:     // LABEL
78117a519f9SDimitry Andric       ResultTy = Type::getLabelTy(Context);
78217a519f9SDimitry Andric       break;
78317a519f9SDimitry Andric     case bitc::TYPE_CODE_METADATA:  // METADATA
78417a519f9SDimitry Andric       ResultTy = Type::getMetadataTy(Context);
78517a519f9SDimitry Andric       break;
78617a519f9SDimitry Andric     case bitc::TYPE_CODE_X86_MMX:   // X86_MMX
78717a519f9SDimitry Andric       ResultTy = Type::getX86_MMXTy(Context);
78817a519f9SDimitry Andric       break;
78917a519f9SDimitry Andric     case bitc::TYPE_CODE_INTEGER:   // INTEGER: [width]
79017a519f9SDimitry Andric       if (Record.size() < 1)
79117a519f9SDimitry Andric         return Error("Invalid Integer type record");
79217a519f9SDimitry Andric       ResultTy = IntegerType::get(Context, Record[0]);
79317a519f9SDimitry Andric       break;
79417a519f9SDimitry Andric     case bitc::TYPE_CODE_OPAQUE:    // OPAQUE
79517a519f9SDimitry Andric       if (NextTypeID < TypeList.size() && TypeList[NextTypeID] == 0)
79617a519f9SDimitry Andric         ResultTy = StructType::createNamed(Context, "");
79717a519f9SDimitry Andric       break;
79817a519f9SDimitry Andric     case bitc::TYPE_CODE_STRUCT_OLD: {// STRUCT_OLD
79917a519f9SDimitry Andric       if (NextTypeID >= TypeList.size()) break;
80017a519f9SDimitry Andric       // If we already read it, don't reprocess.
80117a519f9SDimitry Andric       if (TypeList[NextTypeID] &&
80217a519f9SDimitry Andric           !cast<StructType>(TypeList[NextTypeID])->isOpaque())
80317a519f9SDimitry Andric         break;
80417a519f9SDimitry Andric 
80517a519f9SDimitry Andric       // Set a type.
80617a519f9SDimitry Andric       if (TypeList[NextTypeID] == 0)
80717a519f9SDimitry Andric         TypeList[NextTypeID] = StructType::createNamed(Context, "");
80817a519f9SDimitry Andric 
80917a519f9SDimitry Andric       std::vector<Type*> EltTys;
81017a519f9SDimitry Andric       for (unsigned i = 1, e = Record.size(); i != e; ++i) {
81117a519f9SDimitry Andric         if (Type *Elt = getTypeByIDOrNull(Record[i]))
81217a519f9SDimitry Andric           EltTys.push_back(Elt);
81317a519f9SDimitry Andric         else
81417a519f9SDimitry Andric           break;
81517a519f9SDimitry Andric       }
81617a519f9SDimitry Andric 
81717a519f9SDimitry Andric       if (EltTys.size() != Record.size()-1)
81817a519f9SDimitry Andric         break;      // Not all elements are ready.
81917a519f9SDimitry Andric 
82017a519f9SDimitry Andric       cast<StructType>(TypeList[NextTypeID])->setBody(EltTys, Record[0]);
82117a519f9SDimitry Andric       ResultTy = TypeList[NextTypeID];
82217a519f9SDimitry Andric       TypeList[NextTypeID] = 0;
82317a519f9SDimitry Andric       break;
82417a519f9SDimitry Andric     }
82517a519f9SDimitry Andric     case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
82617a519f9SDimitry Andric       //          [pointee type, address space]
82717a519f9SDimitry Andric       if (Record.size() < 1)
82817a519f9SDimitry Andric         return Error("Invalid POINTER type record");
82917a519f9SDimitry Andric       unsigned AddressSpace = 0;
83017a519f9SDimitry Andric       if (Record.size() == 2)
83117a519f9SDimitry Andric         AddressSpace = Record[1];
83217a519f9SDimitry Andric       if ((ResultTy = getTypeByIDOrNull(Record[0])))
83317a519f9SDimitry Andric         ResultTy = PointerType::get(ResultTy, AddressSpace);
83417a519f9SDimitry Andric       break;
83517a519f9SDimitry Andric     }
83617a519f9SDimitry Andric     case bitc::TYPE_CODE_FUNCTION: {
83717a519f9SDimitry Andric       // FIXME: attrid is dead, remove it in LLVM 3.0
83817a519f9SDimitry Andric       // FUNCTION: [vararg, attrid, retty, paramty x N]
83917a519f9SDimitry Andric       if (Record.size() < 3)
84017a519f9SDimitry Andric         return Error("Invalid FUNCTION type record");
84117a519f9SDimitry Andric       std::vector<Type*> ArgTys;
84217a519f9SDimitry Andric       for (unsigned i = 3, e = Record.size(); i != e; ++i) {
84317a519f9SDimitry Andric         if (Type *Elt = getTypeByIDOrNull(Record[i]))
84417a519f9SDimitry Andric           ArgTys.push_back(Elt);
84517a519f9SDimitry Andric         else
84617a519f9SDimitry Andric           break;
84717a519f9SDimitry Andric       }
84817a519f9SDimitry Andric       if (ArgTys.size()+3 != Record.size())
84917a519f9SDimitry Andric         break;  // Something was null.
85017a519f9SDimitry Andric       if ((ResultTy = getTypeByIDOrNull(Record[2])))
85117a519f9SDimitry Andric         ResultTy = FunctionType::get(ResultTy, ArgTys, Record[0]);
85217a519f9SDimitry Andric       break;
85317a519f9SDimitry Andric     }
85417a519f9SDimitry Andric     case bitc::TYPE_CODE_ARRAY:     // ARRAY: [numelts, eltty]
85517a519f9SDimitry Andric       if (Record.size() < 2)
85617a519f9SDimitry Andric         return Error("Invalid ARRAY type record");
85717a519f9SDimitry Andric       if ((ResultTy = getTypeByIDOrNull(Record[1])))
85817a519f9SDimitry Andric         ResultTy = ArrayType::get(ResultTy, Record[0]);
85917a519f9SDimitry Andric       break;
86017a519f9SDimitry Andric     case bitc::TYPE_CODE_VECTOR:    // VECTOR: [numelts, eltty]
86117a519f9SDimitry Andric       if (Record.size() < 2)
86217a519f9SDimitry Andric         return Error("Invalid VECTOR type record");
86317a519f9SDimitry Andric       if ((ResultTy = getTypeByIDOrNull(Record[1])))
86417a519f9SDimitry Andric         ResultTy = VectorType::get(ResultTy, Record[0]);
86517a519f9SDimitry Andric       break;
86617a519f9SDimitry Andric     }
86717a519f9SDimitry Andric 
86817a519f9SDimitry Andric     if (NextTypeID >= TypeList.size())
86917a519f9SDimitry Andric       return Error("invalid TYPE table");
87017a519f9SDimitry Andric 
87117a519f9SDimitry Andric     if (ResultTy && TypeList[NextTypeID] == 0) {
87217a519f9SDimitry Andric       ++NumTypesRead;
87317a519f9SDimitry Andric       ReadAnyTypes = true;
87417a519f9SDimitry Andric 
87517a519f9SDimitry Andric       TypeList[NextTypeID] = ResultTy;
87617a519f9SDimitry Andric     }
87717a519f9SDimitry Andric 
87817a519f9SDimitry Andric     ++NextTypeID;
87917a519f9SDimitry Andric   }
880f22ef01cSRoman Divacky }
881f22ef01cSRoman Divacky 
882f22ef01cSRoman Divacky 
88317a519f9SDimitry Andric bool BitcodeReader::ParseOldTypeSymbolTable() {
88417a519f9SDimitry Andric   if (Stream.EnterSubBlock(bitc::TYPE_SYMTAB_BLOCK_ID_OLD))
885f22ef01cSRoman Divacky     return Error("Malformed block record");
886f22ef01cSRoman Divacky 
887f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
888f22ef01cSRoman Divacky 
889f22ef01cSRoman Divacky   // Read all the records for this type table.
890f22ef01cSRoman Divacky   std::string TypeName;
891f22ef01cSRoman Divacky   while (1) {
892f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
893f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
894f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
895f22ef01cSRoman Divacky         return Error("Error at end of type symbol table block");
896f22ef01cSRoman Divacky       return false;
897f22ef01cSRoman Divacky     }
898f22ef01cSRoman Divacky 
899f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
900f22ef01cSRoman Divacky       // No known subblocks, always skip them.
901f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
902f22ef01cSRoman Divacky       if (Stream.SkipBlock())
903f22ef01cSRoman Divacky         return Error("Malformed block record");
904f22ef01cSRoman Divacky       continue;
905f22ef01cSRoman Divacky     }
906f22ef01cSRoman Divacky 
907f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
908f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
909f22ef01cSRoman Divacky       continue;
910f22ef01cSRoman Divacky     }
911f22ef01cSRoman Divacky 
912f22ef01cSRoman Divacky     // Read a record.
913f22ef01cSRoman Divacky     Record.clear();
914f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
915f22ef01cSRoman Divacky     default:  // Default behavior: unknown type.
916f22ef01cSRoman Divacky       break;
917f22ef01cSRoman Divacky     case bitc::TST_CODE_ENTRY:    // TST_ENTRY: [typeid, namechar x N]
918f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, TypeName))
919f22ef01cSRoman Divacky         return Error("Invalid TST_ENTRY record");
920f22ef01cSRoman Divacky       unsigned TypeID = Record[0];
921f22ef01cSRoman Divacky       if (TypeID >= TypeList.size())
922f22ef01cSRoman Divacky         return Error("Invalid Type ID in TST_ENTRY record");
923f22ef01cSRoman Divacky 
92417a519f9SDimitry Andric       // Only apply the type name to a struct type with no name.
92517a519f9SDimitry Andric       if (StructType *STy = dyn_cast<StructType>(TypeList[TypeID]))
92617a519f9SDimitry Andric         if (!STy->isAnonymous() && !STy->hasName())
92717a519f9SDimitry Andric           STy->setName(TypeName);
928f22ef01cSRoman Divacky       TypeName.clear();
929f22ef01cSRoman Divacky       break;
930f22ef01cSRoman Divacky     }
931f22ef01cSRoman Divacky   }
932f22ef01cSRoman Divacky }
933f22ef01cSRoman Divacky 
934f22ef01cSRoman Divacky bool BitcodeReader::ParseValueSymbolTable() {
935f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::VALUE_SYMTAB_BLOCK_ID))
936f22ef01cSRoman Divacky     return Error("Malformed block record");
937f22ef01cSRoman Divacky 
938f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
939f22ef01cSRoman Divacky 
940f22ef01cSRoman Divacky   // Read all the records for this value table.
941f22ef01cSRoman Divacky   SmallString<128> ValueName;
942f22ef01cSRoman Divacky   while (1) {
943f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
944f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
945f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
946f22ef01cSRoman Divacky         return Error("Error at end of value symbol table block");
947f22ef01cSRoman Divacky       return false;
948f22ef01cSRoman Divacky     }
949f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
950f22ef01cSRoman Divacky       // No known subblocks, always skip them.
951f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
952f22ef01cSRoman Divacky       if (Stream.SkipBlock())
953f22ef01cSRoman Divacky         return Error("Malformed block record");
954f22ef01cSRoman Divacky       continue;
955f22ef01cSRoman Divacky     }
956f22ef01cSRoman Divacky 
957f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
958f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
959f22ef01cSRoman Divacky       continue;
960f22ef01cSRoman Divacky     }
961f22ef01cSRoman Divacky 
962f22ef01cSRoman Divacky     // Read a record.
963f22ef01cSRoman Divacky     Record.clear();
964f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
965f22ef01cSRoman Divacky     default:  // Default behavior: unknown type.
966f22ef01cSRoman Divacky       break;
967f22ef01cSRoman Divacky     case bitc::VST_CODE_ENTRY: {  // VST_ENTRY: [valueid, namechar x N]
968f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, ValueName))
969f22ef01cSRoman Divacky         return Error("Invalid VST_ENTRY record");
970f22ef01cSRoman Divacky       unsigned ValueID = Record[0];
971f22ef01cSRoman Divacky       if (ValueID >= ValueList.size())
972f22ef01cSRoman Divacky         return Error("Invalid Value ID in VST_ENTRY record");
973f22ef01cSRoman Divacky       Value *V = ValueList[ValueID];
974f22ef01cSRoman Divacky 
975f22ef01cSRoman Divacky       V->setName(StringRef(ValueName.data(), ValueName.size()));
976f22ef01cSRoman Divacky       ValueName.clear();
977f22ef01cSRoman Divacky       break;
978f22ef01cSRoman Divacky     }
979f22ef01cSRoman Divacky     case bitc::VST_CODE_BBENTRY: {
980f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, ValueName))
981f22ef01cSRoman Divacky         return Error("Invalid VST_BBENTRY record");
982f22ef01cSRoman Divacky       BasicBlock *BB = getBasicBlock(Record[0]);
983f22ef01cSRoman Divacky       if (BB == 0)
984f22ef01cSRoman Divacky         return Error("Invalid BB ID in VST_BBENTRY record");
985f22ef01cSRoman Divacky 
986f22ef01cSRoman Divacky       BB->setName(StringRef(ValueName.data(), ValueName.size()));
987f22ef01cSRoman Divacky       ValueName.clear();
988f22ef01cSRoman Divacky       break;
989f22ef01cSRoman Divacky     }
990f22ef01cSRoman Divacky     }
991f22ef01cSRoman Divacky   }
992f22ef01cSRoman Divacky }
993f22ef01cSRoman Divacky 
994f22ef01cSRoman Divacky bool BitcodeReader::ParseMetadata() {
995f22ef01cSRoman Divacky   unsigned NextMDValueNo = MDValueList.size();
996f22ef01cSRoman Divacky 
997f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
998f22ef01cSRoman Divacky     return Error("Malformed block record");
999f22ef01cSRoman Divacky 
1000f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1001f22ef01cSRoman Divacky 
1002f22ef01cSRoman Divacky   // Read all the records.
1003f22ef01cSRoman Divacky   while (1) {
1004f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
1005f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
1006f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
1007f22ef01cSRoman Divacky         return Error("Error at end of PARAMATTR block");
1008f22ef01cSRoman Divacky       return false;
1009f22ef01cSRoman Divacky     }
1010f22ef01cSRoman Divacky 
1011f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
1012f22ef01cSRoman Divacky       // No known subblocks, always skip them.
1013f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
1014f22ef01cSRoman Divacky       if (Stream.SkipBlock())
1015f22ef01cSRoman Divacky         return Error("Malformed block record");
1016f22ef01cSRoman Divacky       continue;
1017f22ef01cSRoman Divacky     }
1018f22ef01cSRoman Divacky 
1019f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
1020f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
1021f22ef01cSRoman Divacky       continue;
1022f22ef01cSRoman Divacky     }
1023f22ef01cSRoman Divacky 
1024f22ef01cSRoman Divacky     bool IsFunctionLocal = false;
1025f22ef01cSRoman Divacky     // Read a record.
1026f22ef01cSRoman Divacky     Record.clear();
1027e580952dSDimitry Andric     Code = Stream.ReadRecord(Code, Record);
1028e580952dSDimitry Andric     switch (Code) {
1029f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
1030f22ef01cSRoman Divacky       break;
1031f22ef01cSRoman Divacky     case bitc::METADATA_NAME: {
1032f22ef01cSRoman Divacky       // Read named of the named metadata.
1033f22ef01cSRoman Divacky       unsigned NameLength = Record.size();
1034f22ef01cSRoman Divacky       SmallString<8> Name;
1035f22ef01cSRoman Divacky       Name.resize(NameLength);
1036f22ef01cSRoman Divacky       for (unsigned i = 0; i != NameLength; ++i)
1037f22ef01cSRoman Divacky         Name[i] = Record[i];
1038f22ef01cSRoman Divacky       Record.clear();
1039f22ef01cSRoman Divacky       Code = Stream.ReadCode();
1040f22ef01cSRoman Divacky 
104117a519f9SDimitry Andric       // METADATA_NAME is always followed by METADATA_NAMED_NODE.
1042e580952dSDimitry Andric       unsigned NextBitCode = Stream.ReadRecord(Code, Record);
104317a519f9SDimitry Andric       assert(NextBitCode == bitc::METADATA_NAMED_NODE); (void)NextBitCode;
1044f22ef01cSRoman Divacky 
1045f22ef01cSRoman Divacky       // Read named metadata elements.
1046f22ef01cSRoman Divacky       unsigned Size = Record.size();
1047e580952dSDimitry Andric       NamedMDNode *NMD = TheModule->getOrInsertNamedMetadata(Name);
1048f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; ++i) {
1049f22ef01cSRoman Divacky         MDNode *MD = dyn_cast<MDNode>(MDValueList.getValueFwdRef(Record[i]));
1050f22ef01cSRoman Divacky         if (MD == 0)
1051f22ef01cSRoman Divacky           return Error("Malformed metadata record");
1052e580952dSDimitry Andric         NMD->addOperand(MD);
1053f22ef01cSRoman Divacky       }
1054f22ef01cSRoman Divacky       break;
1055f22ef01cSRoman Divacky     }
105617a519f9SDimitry Andric     case bitc::METADATA_FN_NODE:
1057f22ef01cSRoman Divacky       IsFunctionLocal = true;
1058f22ef01cSRoman Divacky       // fall-through
105917a519f9SDimitry Andric     case bitc::METADATA_NODE: {
1060ffd1746dSEd Schouten       if (Record.size() % 2 == 1)
106117a519f9SDimitry Andric         return Error("Invalid METADATA_NODE record");
1062f22ef01cSRoman Divacky 
1063f22ef01cSRoman Divacky       unsigned Size = Record.size();
1064f22ef01cSRoman Divacky       SmallVector<Value*, 8> Elts;
1065f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; i += 2) {
10662754fe60SDimitry Andric         const Type *Ty = getTypeByID(Record[i]);
106717a519f9SDimitry Andric         if (!Ty) return Error("Invalid METADATA_NODE record");
1068f22ef01cSRoman Divacky         if (Ty->isMetadataTy())
1069f22ef01cSRoman Divacky           Elts.push_back(MDValueList.getValueFwdRef(Record[i+1]));
1070f22ef01cSRoman Divacky         else if (!Ty->isVoidTy())
1071f22ef01cSRoman Divacky           Elts.push_back(ValueList.getValueFwdRef(Record[i+1], Ty));
1072f22ef01cSRoman Divacky         else
1073f22ef01cSRoman Divacky           Elts.push_back(NULL);
1074f22ef01cSRoman Divacky       }
10753b0f4066SDimitry Andric       Value *V = MDNode::getWhenValsUnresolved(Context, Elts, IsFunctionLocal);
1076f22ef01cSRoman Divacky       IsFunctionLocal = false;
1077f22ef01cSRoman Divacky       MDValueList.AssignValue(V, NextMDValueNo++);
1078f22ef01cSRoman Divacky       break;
1079f22ef01cSRoman Divacky     }
1080f22ef01cSRoman Divacky     case bitc::METADATA_STRING: {
1081f22ef01cSRoman Divacky       unsigned MDStringLength = Record.size();
1082f22ef01cSRoman Divacky       SmallString<8> String;
1083f22ef01cSRoman Divacky       String.resize(MDStringLength);
1084f22ef01cSRoman Divacky       for (unsigned i = 0; i != MDStringLength; ++i)
1085f22ef01cSRoman Divacky         String[i] = Record[i];
1086f22ef01cSRoman Divacky       Value *V = MDString::get(Context,
1087f22ef01cSRoman Divacky                                StringRef(String.data(), String.size()));
1088f22ef01cSRoman Divacky       MDValueList.AssignValue(V, NextMDValueNo++);
1089f22ef01cSRoman Divacky       break;
1090f22ef01cSRoman Divacky     }
1091f22ef01cSRoman Divacky     case bitc::METADATA_KIND: {
1092f22ef01cSRoman Divacky       unsigned RecordLength = Record.size();
1093f22ef01cSRoman Divacky       if (Record.empty() || RecordLength < 2)
1094f22ef01cSRoman Divacky         return Error("Invalid METADATA_KIND record");
1095f22ef01cSRoman Divacky       SmallString<8> Name;
1096f22ef01cSRoman Divacky       Name.resize(RecordLength-1);
1097f22ef01cSRoman Divacky       unsigned Kind = Record[0];
1098f22ef01cSRoman Divacky       for (unsigned i = 1; i != RecordLength; ++i)
1099f22ef01cSRoman Divacky         Name[i-1] = Record[i];
1100f22ef01cSRoman Divacky 
1101f22ef01cSRoman Divacky       unsigned NewKind = TheModule->getMDKindID(Name.str());
1102e580952dSDimitry Andric       if (!MDKindMap.insert(std::make_pair(Kind, NewKind)).second)
1103e580952dSDimitry Andric         return Error("Conflicting METADATA_KIND records");
1104f22ef01cSRoman Divacky       break;
1105f22ef01cSRoman Divacky     }
1106f22ef01cSRoman Divacky     }
1107f22ef01cSRoman Divacky   }
1108f22ef01cSRoman Divacky }
1109f22ef01cSRoman Divacky 
1110f22ef01cSRoman Divacky /// DecodeSignRotatedValue - Decode a signed value stored with the sign bit in
1111f22ef01cSRoman Divacky /// the LSB for dense VBR encoding.
1112f22ef01cSRoman Divacky static uint64_t DecodeSignRotatedValue(uint64_t V) {
1113f22ef01cSRoman Divacky   if ((V & 1) == 0)
1114f22ef01cSRoman Divacky     return V >> 1;
1115f22ef01cSRoman Divacky   if (V != 1)
1116f22ef01cSRoman Divacky     return -(V >> 1);
1117f22ef01cSRoman Divacky   // There is no such thing as -0 with integers.  "-0" really means MININT.
1118f22ef01cSRoman Divacky   return 1ULL << 63;
1119f22ef01cSRoman Divacky }
1120f22ef01cSRoman Divacky 
1121f22ef01cSRoman Divacky /// ResolveGlobalAndAliasInits - Resolve all of the initializers for global
1122f22ef01cSRoman Divacky /// values and aliases that we can.
1123f22ef01cSRoman Divacky bool BitcodeReader::ResolveGlobalAndAliasInits() {
1124f22ef01cSRoman Divacky   std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInitWorklist;
1125f22ef01cSRoman Divacky   std::vector<std::pair<GlobalAlias*, unsigned> > AliasInitWorklist;
1126f22ef01cSRoman Divacky 
1127f22ef01cSRoman Divacky   GlobalInitWorklist.swap(GlobalInits);
1128f22ef01cSRoman Divacky   AliasInitWorklist.swap(AliasInits);
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 {
1136f22ef01cSRoman Divacky       if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
1137f22ef01cSRoman Divacky         GlobalInitWorklist.back().first->setInitializer(C);
1138f22ef01cSRoman Divacky       else
1139f22ef01cSRoman Divacky         return Error("Global variable initializer is not a constant!");
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 {
1149f22ef01cSRoman Divacky       if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
1150f22ef01cSRoman Divacky         AliasInitWorklist.back().first->setAliasee(C);
1151f22ef01cSRoman Divacky       else
1152f22ef01cSRoman Divacky         return Error("Alias initializer is not a constant!");
1153f22ef01cSRoman Divacky     }
1154f22ef01cSRoman Divacky     AliasInitWorklist.pop_back();
1155f22ef01cSRoman Divacky   }
1156f22ef01cSRoman Divacky   return false;
1157f22ef01cSRoman Divacky }
1158f22ef01cSRoman Divacky 
1159f22ef01cSRoman Divacky bool BitcodeReader::ParseConstants() {
1160f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::CONSTANTS_BLOCK_ID))
1161f22ef01cSRoman Divacky     return Error("Malformed block record");
1162f22ef01cSRoman Divacky 
1163f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1164f22ef01cSRoman Divacky 
1165f22ef01cSRoman Divacky   // Read all the records for this value table.
1166f22ef01cSRoman Divacky   const Type *CurTy = Type::getInt32Ty(Context);
1167f22ef01cSRoman Divacky   unsigned NextCstNo = ValueList.size();
1168f22ef01cSRoman Divacky   while (1) {
1169f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
1170f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK)
1171f22ef01cSRoman Divacky       break;
1172f22ef01cSRoman Divacky 
1173f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
1174f22ef01cSRoman Divacky       // No known subblocks, always skip them.
1175f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
1176f22ef01cSRoman Divacky       if (Stream.SkipBlock())
1177f22ef01cSRoman Divacky         return Error("Malformed block record");
1178f22ef01cSRoman Divacky       continue;
1179f22ef01cSRoman Divacky     }
1180f22ef01cSRoman Divacky 
1181f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
1182f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
1183f22ef01cSRoman Divacky       continue;
1184f22ef01cSRoman Divacky     }
1185f22ef01cSRoman Divacky 
1186f22ef01cSRoman Divacky     // Read a record.
1187f22ef01cSRoman Divacky     Record.clear();
1188f22ef01cSRoman Divacky     Value *V = 0;
1189f22ef01cSRoman Divacky     unsigned BitCode = Stream.ReadRecord(Code, Record);
1190f22ef01cSRoman Divacky     switch (BitCode) {
1191f22ef01cSRoman Divacky     default:  // Default behavior: unknown constant
1192f22ef01cSRoman Divacky     case bitc::CST_CODE_UNDEF:     // UNDEF
1193f22ef01cSRoman Divacky       V = UndefValue::get(CurTy);
1194f22ef01cSRoman Divacky       break;
1195f22ef01cSRoman Divacky     case bitc::CST_CODE_SETTYPE:   // SETTYPE: [typeid]
1196f22ef01cSRoman Divacky       if (Record.empty())
1197f22ef01cSRoman Divacky         return Error("Malformed CST_SETTYPE record");
1198f22ef01cSRoman Divacky       if (Record[0] >= TypeList.size())
1199f22ef01cSRoman Divacky         return Error("Invalid Type ID in CST_SETTYPE record");
1200f22ef01cSRoman Divacky       CurTy = TypeList[Record[0]];
1201f22ef01cSRoman Divacky       continue;  // Skip the ValueList manipulation.
1202f22ef01cSRoman Divacky     case bitc::CST_CODE_NULL:      // NULL
1203f22ef01cSRoman Divacky       V = Constant::getNullValue(CurTy);
1204f22ef01cSRoman Divacky       break;
1205f22ef01cSRoman Divacky     case bitc::CST_CODE_INTEGER:   // INTEGER: [intval]
1206f22ef01cSRoman Divacky       if (!CurTy->isIntegerTy() || Record.empty())
1207f22ef01cSRoman Divacky         return Error("Invalid CST_INTEGER record");
1208f22ef01cSRoman Divacky       V = ConstantInt::get(CurTy, DecodeSignRotatedValue(Record[0]));
1209f22ef01cSRoman Divacky       break;
1210f22ef01cSRoman Divacky     case bitc::CST_CODE_WIDE_INTEGER: {// WIDE_INTEGER: [n x intval]
1211f22ef01cSRoman Divacky       if (!CurTy->isIntegerTy() || Record.empty())
1212f22ef01cSRoman Divacky         return Error("Invalid WIDE_INTEGER record");
1213f22ef01cSRoman Divacky 
1214f22ef01cSRoman Divacky       unsigned NumWords = Record.size();
1215f22ef01cSRoman Divacky       SmallVector<uint64_t, 8> Words;
1216f22ef01cSRoman Divacky       Words.resize(NumWords);
1217f22ef01cSRoman Divacky       for (unsigned i = 0; i != NumWords; ++i)
1218f22ef01cSRoman Divacky         Words[i] = DecodeSignRotatedValue(Record[i]);
1219f22ef01cSRoman Divacky       V = ConstantInt::get(Context,
1220f22ef01cSRoman Divacky                            APInt(cast<IntegerType>(CurTy)->getBitWidth(),
1221f22ef01cSRoman Divacky                            NumWords, &Words[0]));
1222f22ef01cSRoman Divacky       break;
1223f22ef01cSRoman Divacky     }
1224f22ef01cSRoman Divacky     case bitc::CST_CODE_FLOAT: {    // FLOAT: [fpval]
1225f22ef01cSRoman Divacky       if (Record.empty())
1226f22ef01cSRoman Divacky         return Error("Invalid FLOAT record");
1227f22ef01cSRoman Divacky       if (CurTy->isFloatTy())
1228f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(32, (uint32_t)Record[0])));
1229f22ef01cSRoman Divacky       else if (CurTy->isDoubleTy())
1230f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(64, Record[0])));
1231f22ef01cSRoman Divacky       else if (CurTy->isX86_FP80Ty()) {
1232f22ef01cSRoman Divacky         // Bits are not stored the same way as a normal i80 APInt, compensate.
1233f22ef01cSRoman Divacky         uint64_t Rearrange[2];
1234f22ef01cSRoman Divacky         Rearrange[0] = (Record[1] & 0xffffLL) | (Record[0] << 16);
1235f22ef01cSRoman Divacky         Rearrange[1] = Record[0] >> 48;
1236f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(80, 2, Rearrange)));
1237f22ef01cSRoman Divacky       } else if (CurTy->isFP128Ty())
1238f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0]), true));
1239f22ef01cSRoman Divacky       else if (CurTy->isPPC_FP128Ty())
1240f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0])));
1241f22ef01cSRoman Divacky       else
1242f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);
1243f22ef01cSRoman Divacky       break;
1244f22ef01cSRoman Divacky     }
1245f22ef01cSRoman Divacky 
1246f22ef01cSRoman Divacky     case bitc::CST_CODE_AGGREGATE: {// AGGREGATE: [n x value number]
1247f22ef01cSRoman Divacky       if (Record.empty())
1248f22ef01cSRoman Divacky         return Error("Invalid CST_AGGREGATE record");
1249f22ef01cSRoman Divacky 
1250f22ef01cSRoman Divacky       unsigned Size = Record.size();
1251f22ef01cSRoman Divacky       std::vector<Constant*> Elts;
1252f22ef01cSRoman Divacky 
1253f22ef01cSRoman Divacky       if (const StructType *STy = dyn_cast<StructType>(CurTy)) {
1254f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1255f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i],
1256f22ef01cSRoman Divacky                                                      STy->getElementType(i)));
1257f22ef01cSRoman Divacky         V = ConstantStruct::get(STy, Elts);
1258f22ef01cSRoman Divacky       } else if (const ArrayType *ATy = dyn_cast<ArrayType>(CurTy)) {
1259f22ef01cSRoman Divacky         const Type *EltTy = ATy->getElementType();
1260f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1261f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
1262f22ef01cSRoman Divacky         V = ConstantArray::get(ATy, Elts);
1263f22ef01cSRoman Divacky       } else if (const VectorType *VTy = dyn_cast<VectorType>(CurTy)) {
1264f22ef01cSRoman Divacky         const Type *EltTy = VTy->getElementType();
1265f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1266f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
1267f22ef01cSRoman Divacky         V = ConstantVector::get(Elts);
1268f22ef01cSRoman Divacky       } else {
1269f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);
1270f22ef01cSRoman Divacky       }
1271f22ef01cSRoman Divacky       break;
1272f22ef01cSRoman Divacky     }
1273f22ef01cSRoman Divacky     case bitc::CST_CODE_STRING: { // STRING: [values]
1274f22ef01cSRoman Divacky       if (Record.empty())
1275f22ef01cSRoman Divacky         return Error("Invalid CST_AGGREGATE record");
1276f22ef01cSRoman Divacky 
1277f22ef01cSRoman Divacky       const ArrayType *ATy = cast<ArrayType>(CurTy);
1278f22ef01cSRoman Divacky       const Type *EltTy = ATy->getElementType();
1279f22ef01cSRoman Divacky 
1280f22ef01cSRoman Divacky       unsigned Size = Record.size();
1281f22ef01cSRoman Divacky       std::vector<Constant*> Elts;
1282f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; ++i)
1283f22ef01cSRoman Divacky         Elts.push_back(ConstantInt::get(EltTy, Record[i]));
1284f22ef01cSRoman Divacky       V = ConstantArray::get(ATy, Elts);
1285f22ef01cSRoman Divacky       break;
1286f22ef01cSRoman Divacky     }
1287f22ef01cSRoman Divacky     case bitc::CST_CODE_CSTRING: { // CSTRING: [values]
1288f22ef01cSRoman Divacky       if (Record.empty())
1289f22ef01cSRoman Divacky         return Error("Invalid CST_AGGREGATE record");
1290f22ef01cSRoman Divacky 
1291f22ef01cSRoman Divacky       const ArrayType *ATy = cast<ArrayType>(CurTy);
1292f22ef01cSRoman Divacky       const Type *EltTy = ATy->getElementType();
1293f22ef01cSRoman Divacky 
1294f22ef01cSRoman Divacky       unsigned Size = Record.size();
1295f22ef01cSRoman Divacky       std::vector<Constant*> Elts;
1296f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; ++i)
1297f22ef01cSRoman Divacky         Elts.push_back(ConstantInt::get(EltTy, Record[i]));
1298f22ef01cSRoman Divacky       Elts.push_back(Constant::getNullValue(EltTy));
1299f22ef01cSRoman Divacky       V = ConstantArray::get(ATy, Elts);
1300f22ef01cSRoman Divacky       break;
1301f22ef01cSRoman Divacky     }
1302f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_BINOP: {  // CE_BINOP: [opcode, opval, opval]
1303f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_BINOP record");
1304f22ef01cSRoman Divacky       int Opc = GetDecodedBinaryOpcode(Record[0], CurTy);
1305f22ef01cSRoman Divacky       if (Opc < 0) {
1306f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);  // Unknown binop.
1307f22ef01cSRoman Divacky       } else {
1308f22ef01cSRoman Divacky         Constant *LHS = ValueList.getConstantFwdRef(Record[1], CurTy);
1309f22ef01cSRoman Divacky         Constant *RHS = ValueList.getConstantFwdRef(Record[2], CurTy);
1310f22ef01cSRoman Divacky         unsigned Flags = 0;
1311f22ef01cSRoman Divacky         if (Record.size() >= 4) {
1312f22ef01cSRoman Divacky           if (Opc == Instruction::Add ||
1313f22ef01cSRoman Divacky               Opc == Instruction::Sub ||
13142754fe60SDimitry Andric               Opc == Instruction::Mul ||
13152754fe60SDimitry Andric               Opc == Instruction::Shl) {
1316f22ef01cSRoman Divacky             if (Record[3] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1317f22ef01cSRoman Divacky               Flags |= OverflowingBinaryOperator::NoSignedWrap;
1318f22ef01cSRoman Divacky             if (Record[3] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1319f22ef01cSRoman Divacky               Flags |= OverflowingBinaryOperator::NoUnsignedWrap;
13202754fe60SDimitry Andric           } else if (Opc == Instruction::SDiv ||
13212754fe60SDimitry Andric                      Opc == Instruction::UDiv ||
13222754fe60SDimitry Andric                      Opc == Instruction::LShr ||
13232754fe60SDimitry Andric                      Opc == Instruction::AShr) {
13242754fe60SDimitry Andric             if (Record[3] & (1 << bitc::PEO_EXACT))
1325f22ef01cSRoman Divacky               Flags |= SDivOperator::IsExact;
1326f22ef01cSRoman Divacky           }
1327f22ef01cSRoman Divacky         }
1328f22ef01cSRoman Divacky         V = ConstantExpr::get(Opc, LHS, RHS, Flags);
1329f22ef01cSRoman Divacky       }
1330f22ef01cSRoman Divacky       break;
1331f22ef01cSRoman Divacky     }
1332f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_CAST: {  // CE_CAST: [opcode, opty, opval]
1333f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_CAST record");
1334f22ef01cSRoman Divacky       int Opc = GetDecodedCastOpcode(Record[0]);
1335f22ef01cSRoman Divacky       if (Opc < 0) {
1336f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);  // Unknown cast.
1337f22ef01cSRoman Divacky       } else {
1338f22ef01cSRoman Divacky         const Type *OpTy = getTypeByID(Record[1]);
1339f22ef01cSRoman Divacky         if (!OpTy) return Error("Invalid CE_CAST record");
1340f22ef01cSRoman Divacky         Constant *Op = ValueList.getConstantFwdRef(Record[2], OpTy);
1341f22ef01cSRoman Divacky         V = ConstantExpr::getCast(Opc, Op, CurTy);
1342f22ef01cSRoman Divacky       }
1343f22ef01cSRoman Divacky       break;
1344f22ef01cSRoman Divacky     }
1345f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_INBOUNDS_GEP:
1346f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_GEP: {  // CE_GEP:        [n x operands]
1347f22ef01cSRoman Divacky       if (Record.size() & 1) return Error("Invalid CE_GEP record");
1348f22ef01cSRoman Divacky       SmallVector<Constant*, 16> Elts;
1349f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
1350f22ef01cSRoman Divacky         const Type *ElTy = getTypeByID(Record[i]);
1351f22ef01cSRoman Divacky         if (!ElTy) return Error("Invalid CE_GEP record");
1352f22ef01cSRoman Divacky         Elts.push_back(ValueList.getConstantFwdRef(Record[i+1], ElTy));
1353f22ef01cSRoman Divacky       }
1354f22ef01cSRoman Divacky       if (BitCode == bitc::CST_CODE_CE_INBOUNDS_GEP)
1355f22ef01cSRoman Divacky         V = ConstantExpr::getInBoundsGetElementPtr(Elts[0], &Elts[1],
1356f22ef01cSRoman Divacky                                                    Elts.size()-1);
1357f22ef01cSRoman Divacky       else
1358f22ef01cSRoman Divacky         V = ConstantExpr::getGetElementPtr(Elts[0], &Elts[1],
1359f22ef01cSRoman Divacky                                            Elts.size()-1);
1360f22ef01cSRoman Divacky       break;
1361f22ef01cSRoman Divacky     }
1362f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SELECT:  // CE_SELECT: [opval#, opval#, opval#]
1363f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_SELECT record");
1364f22ef01cSRoman Divacky       V = ConstantExpr::getSelect(ValueList.getConstantFwdRef(Record[0],
1365f22ef01cSRoman Divacky                                                               Type::getInt1Ty(Context)),
1366f22ef01cSRoman Divacky                                   ValueList.getConstantFwdRef(Record[1],CurTy),
1367f22ef01cSRoman Divacky                                   ValueList.getConstantFwdRef(Record[2],CurTy));
1368f22ef01cSRoman Divacky       break;
1369f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_EXTRACTELT: { // CE_EXTRACTELT: [opty, opval, opval]
1370f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_EXTRACTELT record");
1371f22ef01cSRoman Divacky       const VectorType *OpTy =
1372f22ef01cSRoman Divacky         dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
1373f22ef01cSRoman Divacky       if (OpTy == 0) return Error("Invalid CE_EXTRACTELT record");
1374f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1375f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
1376f22ef01cSRoman Divacky       V = ConstantExpr::getExtractElement(Op0, Op1);
1377f22ef01cSRoman Divacky       break;
1378f22ef01cSRoman Divacky     }
1379f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_INSERTELT: { // CE_INSERTELT: [opval, opval, opval]
1380f22ef01cSRoman Divacky       const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1381f22ef01cSRoman Divacky       if (Record.size() < 3 || OpTy == 0)
1382f22ef01cSRoman Divacky         return Error("Invalid CE_INSERTELT record");
1383f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1384f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[1],
1385f22ef01cSRoman Divacky                                                   OpTy->getElementType());
1386f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
1387f22ef01cSRoman Divacky       V = ConstantExpr::getInsertElement(Op0, Op1, Op2);
1388f22ef01cSRoman Divacky       break;
1389f22ef01cSRoman Divacky     }
1390f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SHUFFLEVEC: { // CE_SHUFFLEVEC: [opval, opval, opval]
1391f22ef01cSRoman Divacky       const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1392f22ef01cSRoman Divacky       if (Record.size() < 3 || OpTy == 0)
1393f22ef01cSRoman Divacky         return Error("Invalid CE_SHUFFLEVEC record");
1394f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1395f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[1], OpTy);
1396f22ef01cSRoman Divacky       const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
1397f22ef01cSRoman Divacky                                                  OpTy->getNumElements());
1398f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[2], ShufTy);
1399f22ef01cSRoman Divacky       V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
1400f22ef01cSRoman Divacky       break;
1401f22ef01cSRoman Divacky     }
1402f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SHUFVEC_EX: { // [opty, opval, opval, opval]
1403f22ef01cSRoman Divacky       const VectorType *RTy = dyn_cast<VectorType>(CurTy);
14042754fe60SDimitry Andric       const VectorType *OpTy =
14052754fe60SDimitry Andric         dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
1406f22ef01cSRoman Divacky       if (Record.size() < 4 || RTy == 0 || OpTy == 0)
1407f22ef01cSRoman Divacky         return Error("Invalid CE_SHUFVEC_EX record");
1408f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1409f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1410f22ef01cSRoman Divacky       const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
1411f22ef01cSRoman Divacky                                                  RTy->getNumElements());
1412f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[3], ShufTy);
1413f22ef01cSRoman Divacky       V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
1414f22ef01cSRoman Divacky       break;
1415f22ef01cSRoman Divacky     }
1416f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_CMP: {     // CE_CMP: [opty, opval, opval, pred]
1417f22ef01cSRoman Divacky       if (Record.size() < 4) return Error("Invalid CE_CMP record");
1418f22ef01cSRoman Divacky       const Type *OpTy = getTypeByID(Record[0]);
1419f22ef01cSRoman Divacky       if (OpTy == 0) return Error("Invalid CE_CMP record");
1420f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1421f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1422f22ef01cSRoman Divacky 
1423f22ef01cSRoman Divacky       if (OpTy->isFPOrFPVectorTy())
1424f22ef01cSRoman Divacky         V = ConstantExpr::getFCmp(Record[3], Op0, Op1);
1425f22ef01cSRoman Divacky       else
1426f22ef01cSRoman Divacky         V = ConstantExpr::getICmp(Record[3], Op0, Op1);
1427f22ef01cSRoman Divacky       break;
1428f22ef01cSRoman Divacky     }
1429f22ef01cSRoman Divacky     case bitc::CST_CODE_INLINEASM: {
1430f22ef01cSRoman Divacky       if (Record.size() < 2) return Error("Invalid INLINEASM record");
1431f22ef01cSRoman Divacky       std::string AsmStr, ConstrStr;
1432f22ef01cSRoman Divacky       bool HasSideEffects = Record[0] & 1;
1433f22ef01cSRoman Divacky       bool IsAlignStack = Record[0] >> 1;
1434f22ef01cSRoman Divacky       unsigned AsmStrSize = Record[1];
1435f22ef01cSRoman Divacky       if (2+AsmStrSize >= Record.size())
1436f22ef01cSRoman Divacky         return Error("Invalid INLINEASM record");
1437f22ef01cSRoman Divacky       unsigned ConstStrSize = Record[2+AsmStrSize];
1438f22ef01cSRoman Divacky       if (3+AsmStrSize+ConstStrSize > Record.size())
1439f22ef01cSRoman Divacky         return Error("Invalid INLINEASM record");
1440f22ef01cSRoman Divacky 
1441f22ef01cSRoman Divacky       for (unsigned i = 0; i != AsmStrSize; ++i)
1442f22ef01cSRoman Divacky         AsmStr += (char)Record[2+i];
1443f22ef01cSRoman Divacky       for (unsigned i = 0; i != ConstStrSize; ++i)
1444f22ef01cSRoman Divacky         ConstrStr += (char)Record[3+AsmStrSize+i];
1445f22ef01cSRoman Divacky       const PointerType *PTy = cast<PointerType>(CurTy);
1446f22ef01cSRoman Divacky       V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
1447f22ef01cSRoman Divacky                          AsmStr, ConstrStr, HasSideEffects, IsAlignStack);
1448f22ef01cSRoman Divacky       break;
1449f22ef01cSRoman Divacky     }
1450f22ef01cSRoman Divacky     case bitc::CST_CODE_BLOCKADDRESS:{
1451f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_BLOCKADDRESS record");
1452f22ef01cSRoman Divacky       const Type *FnTy = getTypeByID(Record[0]);
1453f22ef01cSRoman Divacky       if (FnTy == 0) return Error("Invalid CE_BLOCKADDRESS record");
1454f22ef01cSRoman Divacky       Function *Fn =
1455f22ef01cSRoman Divacky         dyn_cast_or_null<Function>(ValueList.getConstantFwdRef(Record[1],FnTy));
1456f22ef01cSRoman Divacky       if (Fn == 0) return Error("Invalid CE_BLOCKADDRESS record");
1457f22ef01cSRoman Divacky 
1458f22ef01cSRoman Divacky       GlobalVariable *FwdRef = new GlobalVariable(*Fn->getParent(),
1459f22ef01cSRoman Divacky                                                   Type::getInt8Ty(Context),
1460f22ef01cSRoman Divacky                                             false, GlobalValue::InternalLinkage,
1461f22ef01cSRoman Divacky                                                   0, "");
1462f22ef01cSRoman Divacky       BlockAddrFwdRefs[Fn].push_back(std::make_pair(Record[2], FwdRef));
1463f22ef01cSRoman Divacky       V = FwdRef;
1464f22ef01cSRoman Divacky       break;
1465f22ef01cSRoman Divacky     }
1466f22ef01cSRoman Divacky     }
1467f22ef01cSRoman Divacky 
1468f22ef01cSRoman Divacky     ValueList.AssignValue(V, NextCstNo);
1469f22ef01cSRoman Divacky     ++NextCstNo;
1470f22ef01cSRoman Divacky   }
1471f22ef01cSRoman Divacky 
1472f22ef01cSRoman Divacky   if (NextCstNo != ValueList.size())
1473f22ef01cSRoman Divacky     return Error("Invalid constant reference!");
1474f22ef01cSRoman Divacky 
1475f22ef01cSRoman Divacky   if (Stream.ReadBlockEnd())
1476f22ef01cSRoman Divacky     return Error("Error at end of constants block");
1477f22ef01cSRoman Divacky 
1478f22ef01cSRoman Divacky   // Once all the constants have been read, go through and resolve forward
1479f22ef01cSRoman Divacky   // references.
1480f22ef01cSRoman Divacky   ValueList.ResolveConstantForwardRefs();
1481f22ef01cSRoman Divacky   return false;
1482f22ef01cSRoman Divacky }
1483f22ef01cSRoman Divacky 
1484f22ef01cSRoman Divacky /// RememberAndSkipFunctionBody - When we see the block for a function body,
1485f22ef01cSRoman Divacky /// remember where it is and then skip it.  This lets us lazily deserialize the
1486f22ef01cSRoman Divacky /// functions.
1487f22ef01cSRoman Divacky bool BitcodeReader::RememberAndSkipFunctionBody() {
1488f22ef01cSRoman Divacky   // Get the function we are talking about.
1489f22ef01cSRoman Divacky   if (FunctionsWithBodies.empty())
1490f22ef01cSRoman Divacky     return Error("Insufficient function protos");
1491f22ef01cSRoman Divacky 
1492f22ef01cSRoman Divacky   Function *Fn = FunctionsWithBodies.back();
1493f22ef01cSRoman Divacky   FunctionsWithBodies.pop_back();
1494f22ef01cSRoman Divacky 
1495f22ef01cSRoman Divacky   // Save the current stream state.
1496f22ef01cSRoman Divacky   uint64_t CurBit = Stream.GetCurrentBitNo();
1497f22ef01cSRoman Divacky   DeferredFunctionInfo[Fn] = CurBit;
1498f22ef01cSRoman Divacky 
1499f22ef01cSRoman Divacky   // Skip over the function block for now.
1500f22ef01cSRoman Divacky   if (Stream.SkipBlock())
1501f22ef01cSRoman Divacky     return Error("Malformed block record");
1502f22ef01cSRoman Divacky   return false;
1503f22ef01cSRoman Divacky }
1504f22ef01cSRoman Divacky 
1505f22ef01cSRoman Divacky bool BitcodeReader::ParseModule() {
1506f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
1507f22ef01cSRoman Divacky     return Error("Malformed block record");
1508f22ef01cSRoman Divacky 
1509f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1510f22ef01cSRoman Divacky   std::vector<std::string> SectionTable;
1511f22ef01cSRoman Divacky   std::vector<std::string> GCTable;
1512f22ef01cSRoman Divacky 
1513f22ef01cSRoman Divacky   // Read all the records for this module.
1514f22ef01cSRoman Divacky   while (!Stream.AtEndOfStream()) {
1515f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
1516f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
1517f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
1518f22ef01cSRoman Divacky         return Error("Error at end of module block");
1519f22ef01cSRoman Divacky 
1520f22ef01cSRoman Divacky       // Patch the initializers for globals and aliases up.
1521f22ef01cSRoman Divacky       ResolveGlobalAndAliasInits();
1522f22ef01cSRoman Divacky       if (!GlobalInits.empty() || !AliasInits.empty())
1523f22ef01cSRoman Divacky         return Error("Malformed global initializer set");
1524f22ef01cSRoman Divacky       if (!FunctionsWithBodies.empty())
1525f22ef01cSRoman Divacky         return Error("Too few function bodies found");
1526f22ef01cSRoman Divacky 
1527f22ef01cSRoman Divacky       // Look for intrinsic functions which need to be upgraded at some point
1528f22ef01cSRoman Divacky       for (Module::iterator FI = TheModule->begin(), FE = TheModule->end();
1529f22ef01cSRoman Divacky            FI != FE; ++FI) {
1530f22ef01cSRoman Divacky         Function* NewFn;
1531f22ef01cSRoman Divacky         if (UpgradeIntrinsicFunction(FI, NewFn))
1532f22ef01cSRoman Divacky           UpgradedIntrinsics.push_back(std::make_pair(FI, NewFn));
1533f22ef01cSRoman Divacky       }
1534f22ef01cSRoman Divacky 
1535e580952dSDimitry Andric       // Look for global variables which need to be renamed.
1536e580952dSDimitry Andric       for (Module::global_iterator
1537e580952dSDimitry Andric              GI = TheModule->global_begin(), GE = TheModule->global_end();
1538e580952dSDimitry Andric            GI != GE; ++GI)
1539e580952dSDimitry Andric         UpgradeGlobalVariable(GI);
1540e580952dSDimitry Andric 
1541f22ef01cSRoman Divacky       // Force deallocation of memory for these vectors to favor the client that
1542f22ef01cSRoman Divacky       // want lazy deserialization.
1543f22ef01cSRoman Divacky       std::vector<std::pair<GlobalVariable*, unsigned> >().swap(GlobalInits);
1544f22ef01cSRoman Divacky       std::vector<std::pair<GlobalAlias*, unsigned> >().swap(AliasInits);
1545f22ef01cSRoman Divacky       std::vector<Function*>().swap(FunctionsWithBodies);
1546f22ef01cSRoman Divacky       return false;
1547f22ef01cSRoman Divacky     }
1548f22ef01cSRoman Divacky 
1549f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
1550f22ef01cSRoman Divacky       switch (Stream.ReadSubBlockID()) {
1551f22ef01cSRoman Divacky       default:  // Skip unknown content.
1552f22ef01cSRoman Divacky         if (Stream.SkipBlock())
1553f22ef01cSRoman Divacky           return Error("Malformed block record");
1554f22ef01cSRoman Divacky         break;
1555f22ef01cSRoman Divacky       case bitc::BLOCKINFO_BLOCK_ID:
1556f22ef01cSRoman Divacky         if (Stream.ReadBlockInfoBlock())
1557f22ef01cSRoman Divacky           return Error("Malformed BlockInfoBlock");
1558f22ef01cSRoman Divacky         break;
1559f22ef01cSRoman Divacky       case bitc::PARAMATTR_BLOCK_ID:
1560f22ef01cSRoman Divacky         if (ParseAttributeBlock())
1561f22ef01cSRoman Divacky           return true;
1562f22ef01cSRoman Divacky         break;
156317a519f9SDimitry Andric       case bitc::TYPE_BLOCK_ID_NEW:
1564f22ef01cSRoman Divacky         if (ParseTypeTable())
1565f22ef01cSRoman Divacky           return true;
1566f22ef01cSRoman Divacky         break;
156717a519f9SDimitry Andric       case bitc::TYPE_BLOCK_ID_OLD:
156817a519f9SDimitry Andric         if (ParseOldTypeTable())
156917a519f9SDimitry Andric           return true;
157017a519f9SDimitry Andric         break;
157117a519f9SDimitry Andric       case bitc::TYPE_SYMTAB_BLOCK_ID_OLD:
157217a519f9SDimitry Andric         if (ParseOldTypeSymbolTable())
1573f22ef01cSRoman Divacky           return true;
1574f22ef01cSRoman Divacky         break;
1575f22ef01cSRoman Divacky       case bitc::VALUE_SYMTAB_BLOCK_ID:
1576f22ef01cSRoman Divacky         if (ParseValueSymbolTable())
1577f22ef01cSRoman Divacky           return true;
1578f22ef01cSRoman Divacky         break;
1579f22ef01cSRoman Divacky       case bitc::CONSTANTS_BLOCK_ID:
1580f22ef01cSRoman Divacky         if (ParseConstants() || ResolveGlobalAndAliasInits())
1581f22ef01cSRoman Divacky           return true;
1582f22ef01cSRoman Divacky         break;
1583f22ef01cSRoman Divacky       case bitc::METADATA_BLOCK_ID:
1584f22ef01cSRoman Divacky         if (ParseMetadata())
1585f22ef01cSRoman Divacky           return true;
1586f22ef01cSRoman Divacky         break;
1587f22ef01cSRoman Divacky       case bitc::FUNCTION_BLOCK_ID:
1588f22ef01cSRoman Divacky         // If this is the first function body we've seen, reverse the
1589f22ef01cSRoman Divacky         // FunctionsWithBodies list.
1590f22ef01cSRoman Divacky         if (!HasReversedFunctionsWithBodies) {
1591f22ef01cSRoman Divacky           std::reverse(FunctionsWithBodies.begin(), FunctionsWithBodies.end());
1592f22ef01cSRoman Divacky           HasReversedFunctionsWithBodies = true;
1593f22ef01cSRoman Divacky         }
1594f22ef01cSRoman Divacky 
1595f22ef01cSRoman Divacky         if (RememberAndSkipFunctionBody())
1596f22ef01cSRoman Divacky           return true;
1597f22ef01cSRoman Divacky         break;
1598f22ef01cSRoman Divacky       }
1599f22ef01cSRoman Divacky       continue;
1600f22ef01cSRoman Divacky     }
1601f22ef01cSRoman Divacky 
1602f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
1603f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
1604f22ef01cSRoman Divacky       continue;
1605f22ef01cSRoman Divacky     }
1606f22ef01cSRoman Divacky 
1607f22ef01cSRoman Divacky     // Read a record.
1608f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
1609f22ef01cSRoman Divacky     default: break;  // Default behavior, ignore unknown content.
1610f22ef01cSRoman Divacky     case bitc::MODULE_CODE_VERSION:  // VERSION: [version#]
1611f22ef01cSRoman Divacky       if (Record.size() < 1)
1612f22ef01cSRoman Divacky         return Error("Malformed MODULE_CODE_VERSION");
1613f22ef01cSRoman Divacky       // Only version #0 is supported so far.
1614f22ef01cSRoman Divacky       if (Record[0] != 0)
1615f22ef01cSRoman Divacky         return Error("Unknown bitstream version!");
1616f22ef01cSRoman Divacky       break;
1617f22ef01cSRoman Divacky     case bitc::MODULE_CODE_TRIPLE: {  // TRIPLE: [strchr x N]
1618f22ef01cSRoman Divacky       std::string S;
1619f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1620f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_TRIPLE record");
1621f22ef01cSRoman Divacky       TheModule->setTargetTriple(S);
1622f22ef01cSRoman Divacky       break;
1623f22ef01cSRoman Divacky     }
1624f22ef01cSRoman Divacky     case bitc::MODULE_CODE_DATALAYOUT: {  // DATALAYOUT: [strchr x N]
1625f22ef01cSRoman Divacky       std::string S;
1626f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1627f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_DATALAYOUT record");
1628f22ef01cSRoman Divacky       TheModule->setDataLayout(S);
1629f22ef01cSRoman Divacky       break;
1630f22ef01cSRoman Divacky     }
1631f22ef01cSRoman Divacky     case bitc::MODULE_CODE_ASM: {  // ASM: [strchr x N]
1632f22ef01cSRoman Divacky       std::string S;
1633f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1634f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_ASM record");
1635f22ef01cSRoman Divacky       TheModule->setModuleInlineAsm(S);
1636f22ef01cSRoman Divacky       break;
1637f22ef01cSRoman Divacky     }
1638f22ef01cSRoman Divacky     case bitc::MODULE_CODE_DEPLIB: {  // DEPLIB: [strchr x N]
1639f22ef01cSRoman Divacky       std::string S;
1640f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1641f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_DEPLIB record");
1642f22ef01cSRoman Divacky       TheModule->addLibrary(S);
1643f22ef01cSRoman Divacky       break;
1644f22ef01cSRoman Divacky     }
1645f22ef01cSRoman Divacky     case bitc::MODULE_CODE_SECTIONNAME: {  // SECTIONNAME: [strchr x N]
1646f22ef01cSRoman Divacky       std::string S;
1647f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1648f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_SECTIONNAME record");
1649f22ef01cSRoman Divacky       SectionTable.push_back(S);
1650f22ef01cSRoman Divacky       break;
1651f22ef01cSRoman Divacky     }
1652f22ef01cSRoman Divacky     case bitc::MODULE_CODE_GCNAME: {  // SECTIONNAME: [strchr x N]
1653f22ef01cSRoman Divacky       std::string S;
1654f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1655f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_GCNAME record");
1656f22ef01cSRoman Divacky       GCTable.push_back(S);
1657f22ef01cSRoman Divacky       break;
1658f22ef01cSRoman Divacky     }
1659f22ef01cSRoman Divacky     // GLOBALVAR: [pointer type, isconst, initid,
16602754fe60SDimitry Andric     //             linkage, alignment, section, visibility, threadlocal,
16612754fe60SDimitry Andric     //             unnamed_addr]
1662f22ef01cSRoman Divacky     case bitc::MODULE_CODE_GLOBALVAR: {
1663f22ef01cSRoman Divacky       if (Record.size() < 6)
1664f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_GLOBALVAR record");
1665f22ef01cSRoman Divacky       const Type *Ty = getTypeByID(Record[0]);
16662754fe60SDimitry Andric       if (!Ty) return Error("Invalid MODULE_CODE_GLOBALVAR record");
1667f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
1668f22ef01cSRoman Divacky         return Error("Global not a pointer type!");
1669f22ef01cSRoman Divacky       unsigned AddressSpace = cast<PointerType>(Ty)->getAddressSpace();
1670f22ef01cSRoman Divacky       Ty = cast<PointerType>(Ty)->getElementType();
1671f22ef01cSRoman Divacky 
1672f22ef01cSRoman Divacky       bool isConstant = Record[1];
1673f22ef01cSRoman Divacky       GlobalValue::LinkageTypes Linkage = GetDecodedLinkage(Record[3]);
1674f22ef01cSRoman Divacky       unsigned Alignment = (1 << Record[4]) >> 1;
1675f22ef01cSRoman Divacky       std::string Section;
1676f22ef01cSRoman Divacky       if (Record[5]) {
1677f22ef01cSRoman Divacky         if (Record[5]-1 >= SectionTable.size())
1678f22ef01cSRoman Divacky           return Error("Invalid section ID");
1679f22ef01cSRoman Divacky         Section = SectionTable[Record[5]-1];
1680f22ef01cSRoman Divacky       }
1681f22ef01cSRoman Divacky       GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
1682f22ef01cSRoman Divacky       if (Record.size() > 6)
1683f22ef01cSRoman Divacky         Visibility = GetDecodedVisibility(Record[6]);
1684f22ef01cSRoman Divacky       bool isThreadLocal = false;
1685f22ef01cSRoman Divacky       if (Record.size() > 7)
1686f22ef01cSRoman Divacky         isThreadLocal = Record[7];
1687f22ef01cSRoman Divacky 
16882754fe60SDimitry Andric       bool UnnamedAddr = false;
16892754fe60SDimitry Andric       if (Record.size() > 8)
16902754fe60SDimitry Andric         UnnamedAddr = Record[8];
16912754fe60SDimitry Andric 
1692f22ef01cSRoman Divacky       GlobalVariable *NewGV =
1693f22ef01cSRoman Divacky         new GlobalVariable(*TheModule, Ty, isConstant, Linkage, 0, "", 0,
1694f22ef01cSRoman Divacky                            isThreadLocal, AddressSpace);
1695f22ef01cSRoman Divacky       NewGV->setAlignment(Alignment);
1696f22ef01cSRoman Divacky       if (!Section.empty())
1697f22ef01cSRoman Divacky         NewGV->setSection(Section);
1698f22ef01cSRoman Divacky       NewGV->setVisibility(Visibility);
1699f22ef01cSRoman Divacky       NewGV->setThreadLocal(isThreadLocal);
17002754fe60SDimitry Andric       NewGV->setUnnamedAddr(UnnamedAddr);
1701f22ef01cSRoman Divacky 
1702f22ef01cSRoman Divacky       ValueList.push_back(NewGV);
1703f22ef01cSRoman Divacky 
1704f22ef01cSRoman Divacky       // Remember which value to use for the global initializer.
1705f22ef01cSRoman Divacky       if (unsigned InitID = Record[2])
1706f22ef01cSRoman Divacky         GlobalInits.push_back(std::make_pair(NewGV, InitID-1));
1707f22ef01cSRoman Divacky       break;
1708f22ef01cSRoman Divacky     }
1709f22ef01cSRoman Divacky     // FUNCTION:  [type, callingconv, isproto, linkage, paramattr,
17102754fe60SDimitry Andric     //             alignment, section, visibility, gc, unnamed_addr]
1711f22ef01cSRoman Divacky     case bitc::MODULE_CODE_FUNCTION: {
1712f22ef01cSRoman Divacky       if (Record.size() < 8)
1713f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_FUNCTION record");
1714f22ef01cSRoman Divacky       const Type *Ty = getTypeByID(Record[0]);
17152754fe60SDimitry Andric       if (!Ty) return Error("Invalid MODULE_CODE_FUNCTION record");
1716f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
1717f22ef01cSRoman Divacky         return Error("Function not a pointer type!");
1718f22ef01cSRoman Divacky       const FunctionType *FTy =
1719f22ef01cSRoman Divacky         dyn_cast<FunctionType>(cast<PointerType>(Ty)->getElementType());
1720f22ef01cSRoman Divacky       if (!FTy)
1721f22ef01cSRoman Divacky         return Error("Function not a pointer to function type!");
1722f22ef01cSRoman Divacky 
1723f22ef01cSRoman Divacky       Function *Func = Function::Create(FTy, GlobalValue::ExternalLinkage,
1724f22ef01cSRoman Divacky                                         "", TheModule);
1725f22ef01cSRoman Divacky 
1726f22ef01cSRoman Divacky       Func->setCallingConv(static_cast<CallingConv::ID>(Record[1]));
1727f22ef01cSRoman Divacky       bool isProto = Record[2];
1728f22ef01cSRoman Divacky       Func->setLinkage(GetDecodedLinkage(Record[3]));
1729f22ef01cSRoman Divacky       Func->setAttributes(getAttributes(Record[4]));
1730f22ef01cSRoman Divacky 
1731f22ef01cSRoman Divacky       Func->setAlignment((1 << Record[5]) >> 1);
1732f22ef01cSRoman Divacky       if (Record[6]) {
1733f22ef01cSRoman Divacky         if (Record[6]-1 >= SectionTable.size())
1734f22ef01cSRoman Divacky           return Error("Invalid section ID");
1735f22ef01cSRoman Divacky         Func->setSection(SectionTable[Record[6]-1]);
1736f22ef01cSRoman Divacky       }
1737f22ef01cSRoman Divacky       Func->setVisibility(GetDecodedVisibility(Record[7]));
1738f22ef01cSRoman Divacky       if (Record.size() > 8 && Record[8]) {
1739f22ef01cSRoman Divacky         if (Record[8]-1 > GCTable.size())
1740f22ef01cSRoman Divacky           return Error("Invalid GC ID");
1741f22ef01cSRoman Divacky         Func->setGC(GCTable[Record[8]-1].c_str());
1742f22ef01cSRoman Divacky       }
17432754fe60SDimitry Andric       bool UnnamedAddr = false;
17442754fe60SDimitry Andric       if (Record.size() > 9)
17452754fe60SDimitry Andric         UnnamedAddr = Record[9];
17462754fe60SDimitry Andric       Func->setUnnamedAddr(UnnamedAddr);
1747f22ef01cSRoman Divacky       ValueList.push_back(Func);
1748f22ef01cSRoman Divacky 
1749f22ef01cSRoman Divacky       // If this is a function with a body, remember the prototype we are
1750f22ef01cSRoman Divacky       // creating now, so that we can match up the body with them later.
1751f22ef01cSRoman Divacky       if (!isProto)
1752f22ef01cSRoman Divacky         FunctionsWithBodies.push_back(Func);
1753f22ef01cSRoman Divacky       break;
1754f22ef01cSRoman Divacky     }
1755f22ef01cSRoman Divacky     // ALIAS: [alias type, aliasee val#, linkage]
1756f22ef01cSRoman Divacky     // ALIAS: [alias type, aliasee val#, linkage, visibility]
1757f22ef01cSRoman Divacky     case bitc::MODULE_CODE_ALIAS: {
1758f22ef01cSRoman Divacky       if (Record.size() < 3)
1759f22ef01cSRoman Divacky         return Error("Invalid MODULE_ALIAS record");
1760f22ef01cSRoman Divacky       const Type *Ty = getTypeByID(Record[0]);
17612754fe60SDimitry Andric       if (!Ty) return Error("Invalid MODULE_ALIAS record");
1762f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
1763f22ef01cSRoman Divacky         return Error("Function not a pointer type!");
1764f22ef01cSRoman Divacky 
1765f22ef01cSRoman Divacky       GlobalAlias *NewGA = new GlobalAlias(Ty, GetDecodedLinkage(Record[2]),
1766f22ef01cSRoman Divacky                                            "", 0, TheModule);
1767f22ef01cSRoman Divacky       // Old bitcode files didn't have visibility field.
1768f22ef01cSRoman Divacky       if (Record.size() > 3)
1769f22ef01cSRoman Divacky         NewGA->setVisibility(GetDecodedVisibility(Record[3]));
1770f22ef01cSRoman Divacky       ValueList.push_back(NewGA);
1771f22ef01cSRoman Divacky       AliasInits.push_back(std::make_pair(NewGA, Record[1]));
1772f22ef01cSRoman Divacky       break;
1773f22ef01cSRoman Divacky     }
1774f22ef01cSRoman Divacky     /// MODULE_CODE_PURGEVALS: [numvals]
1775f22ef01cSRoman Divacky     case bitc::MODULE_CODE_PURGEVALS:
1776f22ef01cSRoman Divacky       // Trim down the value list to the specified size.
1777f22ef01cSRoman Divacky       if (Record.size() < 1 || Record[0] > ValueList.size())
1778f22ef01cSRoman Divacky         return Error("Invalid MODULE_PURGEVALS record");
1779f22ef01cSRoman Divacky       ValueList.shrinkTo(Record[0]);
1780f22ef01cSRoman Divacky       break;
1781f22ef01cSRoman Divacky     }
1782f22ef01cSRoman Divacky     Record.clear();
1783f22ef01cSRoman Divacky   }
1784f22ef01cSRoman Divacky 
1785f22ef01cSRoman Divacky   return Error("Premature end of bitstream");
1786f22ef01cSRoman Divacky }
1787f22ef01cSRoman Divacky 
1788f22ef01cSRoman Divacky bool BitcodeReader::ParseBitcodeInto(Module *M) {
1789f22ef01cSRoman Divacky   TheModule = 0;
1790f22ef01cSRoman Divacky 
1791f22ef01cSRoman Divacky   unsigned char *BufPtr = (unsigned char *)Buffer->getBufferStart();
1792f22ef01cSRoman Divacky   unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
1793f22ef01cSRoman Divacky 
1794f22ef01cSRoman Divacky   if (Buffer->getBufferSize() & 3) {
1795f22ef01cSRoman Divacky     if (!isRawBitcode(BufPtr, BufEnd) && !isBitcodeWrapper(BufPtr, BufEnd))
1796f22ef01cSRoman Divacky       return Error("Invalid bitcode signature");
1797f22ef01cSRoman Divacky     else
1798f22ef01cSRoman Divacky       return Error("Bitcode stream should be a multiple of 4 bytes in length");
1799f22ef01cSRoman Divacky   }
1800f22ef01cSRoman Divacky 
1801f22ef01cSRoman Divacky   // If we have a wrapper header, parse it and ignore the non-bc file contents.
1802f22ef01cSRoman Divacky   // The magic number is 0x0B17C0DE stored in little endian.
1803f22ef01cSRoman Divacky   if (isBitcodeWrapper(BufPtr, BufEnd))
1804f22ef01cSRoman Divacky     if (SkipBitcodeWrapperHeader(BufPtr, BufEnd))
1805f22ef01cSRoman Divacky       return Error("Invalid bitcode wrapper header");
1806f22ef01cSRoman Divacky 
1807f22ef01cSRoman Divacky   StreamFile.init(BufPtr, BufEnd);
1808f22ef01cSRoman Divacky   Stream.init(StreamFile);
1809f22ef01cSRoman Divacky 
1810f22ef01cSRoman Divacky   // Sniff for the signature.
1811f22ef01cSRoman Divacky   if (Stream.Read(8) != 'B' ||
1812f22ef01cSRoman Divacky       Stream.Read(8) != 'C' ||
1813f22ef01cSRoman Divacky       Stream.Read(4) != 0x0 ||
1814f22ef01cSRoman Divacky       Stream.Read(4) != 0xC ||
1815f22ef01cSRoman Divacky       Stream.Read(4) != 0xE ||
1816f22ef01cSRoman Divacky       Stream.Read(4) != 0xD)
1817f22ef01cSRoman Divacky     return Error("Invalid bitcode signature");
1818f22ef01cSRoman Divacky 
1819f22ef01cSRoman Divacky   // We expect a number of well-defined blocks, though we don't necessarily
1820f22ef01cSRoman Divacky   // need to understand them all.
1821f22ef01cSRoman Divacky   while (!Stream.AtEndOfStream()) {
1822f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
1823f22ef01cSRoman Divacky 
1824bd5abe19SDimitry Andric     if (Code != bitc::ENTER_SUBBLOCK) {
1825bd5abe19SDimitry Andric 
1826bd5abe19SDimitry Andric       // The ranlib in xcode 4 will align archive members by appending newlines to the
1827bd5abe19SDimitry Andric       // end of them. If this file size is a multiple of 4 but not 8, we have to read and
1828bd5abe19SDimitry Andric       // ignore these final 4 bytes :-(
1829bd5abe19SDimitry Andric       if (Stream.GetAbbrevIDWidth() == 2 && Code == 2 &&
1830bd5abe19SDimitry Andric           Stream.Read(6) == 2 && Stream.Read(24) == 0xa0a0a &&
1831bd5abe19SDimitry Andric 	  Stream.AtEndOfStream())
1832bd5abe19SDimitry Andric         return false;
1833bd5abe19SDimitry Andric 
1834f22ef01cSRoman Divacky       return Error("Invalid record at top-level");
1835bd5abe19SDimitry Andric     }
1836f22ef01cSRoman Divacky 
1837f22ef01cSRoman Divacky     unsigned BlockID = Stream.ReadSubBlockID();
1838f22ef01cSRoman Divacky 
1839f22ef01cSRoman Divacky     // We only know the MODULE subblock ID.
1840f22ef01cSRoman Divacky     switch (BlockID) {
1841f22ef01cSRoman Divacky     case bitc::BLOCKINFO_BLOCK_ID:
1842f22ef01cSRoman Divacky       if (Stream.ReadBlockInfoBlock())
1843f22ef01cSRoman Divacky         return Error("Malformed BlockInfoBlock");
1844f22ef01cSRoman Divacky       break;
1845f22ef01cSRoman Divacky     case bitc::MODULE_BLOCK_ID:
1846f22ef01cSRoman Divacky       // Reject multiple MODULE_BLOCK's in a single bitstream.
1847f22ef01cSRoman Divacky       if (TheModule)
1848f22ef01cSRoman Divacky         return Error("Multiple MODULE_BLOCKs in same stream");
1849f22ef01cSRoman Divacky       TheModule = M;
1850f22ef01cSRoman Divacky       if (ParseModule())
1851f22ef01cSRoman Divacky         return true;
1852f22ef01cSRoman Divacky       break;
1853f22ef01cSRoman Divacky     default:
1854f22ef01cSRoman Divacky       if (Stream.SkipBlock())
1855f22ef01cSRoman Divacky         return Error("Malformed block record");
1856f22ef01cSRoman Divacky       break;
1857f22ef01cSRoman Divacky     }
1858f22ef01cSRoman Divacky   }
1859f22ef01cSRoman Divacky 
1860f22ef01cSRoman Divacky   return false;
1861f22ef01cSRoman Divacky }
1862f22ef01cSRoman Divacky 
18632754fe60SDimitry Andric bool BitcodeReader::ParseModuleTriple(std::string &Triple) {
18642754fe60SDimitry Andric   if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
18652754fe60SDimitry Andric     return Error("Malformed block record");
18662754fe60SDimitry Andric 
18672754fe60SDimitry Andric   SmallVector<uint64_t, 64> Record;
18682754fe60SDimitry Andric 
18692754fe60SDimitry Andric   // Read all the records for this module.
18702754fe60SDimitry Andric   while (!Stream.AtEndOfStream()) {
18712754fe60SDimitry Andric     unsigned Code = Stream.ReadCode();
18722754fe60SDimitry Andric     if (Code == bitc::END_BLOCK) {
18732754fe60SDimitry Andric       if (Stream.ReadBlockEnd())
18742754fe60SDimitry Andric         return Error("Error at end of module block");
18752754fe60SDimitry Andric 
18762754fe60SDimitry Andric       return false;
18772754fe60SDimitry Andric     }
18782754fe60SDimitry Andric 
18792754fe60SDimitry Andric     if (Code == bitc::ENTER_SUBBLOCK) {
18802754fe60SDimitry Andric       switch (Stream.ReadSubBlockID()) {
18812754fe60SDimitry Andric       default:  // Skip unknown content.
18822754fe60SDimitry Andric         if (Stream.SkipBlock())
18832754fe60SDimitry Andric           return Error("Malformed block record");
18842754fe60SDimitry Andric         break;
18852754fe60SDimitry Andric       }
18862754fe60SDimitry Andric       continue;
18872754fe60SDimitry Andric     }
18882754fe60SDimitry Andric 
18892754fe60SDimitry Andric     if (Code == bitc::DEFINE_ABBREV) {
18902754fe60SDimitry Andric       Stream.ReadAbbrevRecord();
18912754fe60SDimitry Andric       continue;
18922754fe60SDimitry Andric     }
18932754fe60SDimitry Andric 
18942754fe60SDimitry Andric     // Read a record.
18952754fe60SDimitry Andric     switch (Stream.ReadRecord(Code, Record)) {
18962754fe60SDimitry Andric     default: break;  // Default behavior, ignore unknown content.
18972754fe60SDimitry Andric     case bitc::MODULE_CODE_VERSION:  // VERSION: [version#]
18982754fe60SDimitry Andric       if (Record.size() < 1)
18992754fe60SDimitry Andric         return Error("Malformed MODULE_CODE_VERSION");
19002754fe60SDimitry Andric       // Only version #0 is supported so far.
19012754fe60SDimitry Andric       if (Record[0] != 0)
19022754fe60SDimitry Andric         return Error("Unknown bitstream version!");
19032754fe60SDimitry Andric       break;
19042754fe60SDimitry Andric     case bitc::MODULE_CODE_TRIPLE: {  // TRIPLE: [strchr x N]
19052754fe60SDimitry Andric       std::string S;
19062754fe60SDimitry Andric       if (ConvertToString(Record, 0, S))
19072754fe60SDimitry Andric         return Error("Invalid MODULE_CODE_TRIPLE record");
19082754fe60SDimitry Andric       Triple = S;
19092754fe60SDimitry Andric       break;
19102754fe60SDimitry Andric     }
19112754fe60SDimitry Andric     }
19122754fe60SDimitry Andric     Record.clear();
19132754fe60SDimitry Andric   }
19142754fe60SDimitry Andric 
19152754fe60SDimitry Andric   return Error("Premature end of bitstream");
19162754fe60SDimitry Andric }
19172754fe60SDimitry Andric 
19182754fe60SDimitry Andric bool BitcodeReader::ParseTriple(std::string &Triple) {
19192754fe60SDimitry Andric   if (Buffer->getBufferSize() & 3)
19202754fe60SDimitry Andric     return Error("Bitcode stream should be a multiple of 4 bytes in length");
19212754fe60SDimitry Andric 
19222754fe60SDimitry Andric   unsigned char *BufPtr = (unsigned char *)Buffer->getBufferStart();
19232754fe60SDimitry Andric   unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
19242754fe60SDimitry Andric 
19252754fe60SDimitry Andric   // If we have a wrapper header, parse it and ignore the non-bc file contents.
19262754fe60SDimitry Andric   // The magic number is 0x0B17C0DE stored in little endian.
19272754fe60SDimitry Andric   if (isBitcodeWrapper(BufPtr, BufEnd))
19282754fe60SDimitry Andric     if (SkipBitcodeWrapperHeader(BufPtr, BufEnd))
19292754fe60SDimitry Andric       return Error("Invalid bitcode wrapper header");
19302754fe60SDimitry Andric 
19312754fe60SDimitry Andric   StreamFile.init(BufPtr, BufEnd);
19322754fe60SDimitry Andric   Stream.init(StreamFile);
19332754fe60SDimitry Andric 
19342754fe60SDimitry Andric   // Sniff for the signature.
19352754fe60SDimitry Andric   if (Stream.Read(8) != 'B' ||
19362754fe60SDimitry Andric       Stream.Read(8) != 'C' ||
19372754fe60SDimitry Andric       Stream.Read(4) != 0x0 ||
19382754fe60SDimitry Andric       Stream.Read(4) != 0xC ||
19392754fe60SDimitry Andric       Stream.Read(4) != 0xE ||
19402754fe60SDimitry Andric       Stream.Read(4) != 0xD)
19412754fe60SDimitry Andric     return Error("Invalid bitcode signature");
19422754fe60SDimitry Andric 
19432754fe60SDimitry Andric   // We expect a number of well-defined blocks, though we don't necessarily
19442754fe60SDimitry Andric   // need to understand them all.
19452754fe60SDimitry Andric   while (!Stream.AtEndOfStream()) {
19462754fe60SDimitry Andric     unsigned Code = Stream.ReadCode();
19472754fe60SDimitry Andric 
19482754fe60SDimitry Andric     if (Code != bitc::ENTER_SUBBLOCK)
19492754fe60SDimitry Andric       return Error("Invalid record at top-level");
19502754fe60SDimitry Andric 
19512754fe60SDimitry Andric     unsigned BlockID = Stream.ReadSubBlockID();
19522754fe60SDimitry Andric 
19532754fe60SDimitry Andric     // We only know the MODULE subblock ID.
19542754fe60SDimitry Andric     switch (BlockID) {
19552754fe60SDimitry Andric     case bitc::MODULE_BLOCK_ID:
19562754fe60SDimitry Andric       if (ParseModuleTriple(Triple))
19572754fe60SDimitry Andric         return true;
19582754fe60SDimitry Andric       break;
19592754fe60SDimitry Andric     default:
19602754fe60SDimitry Andric       if (Stream.SkipBlock())
19612754fe60SDimitry Andric         return Error("Malformed block record");
19622754fe60SDimitry Andric       break;
19632754fe60SDimitry Andric     }
19642754fe60SDimitry Andric   }
19652754fe60SDimitry Andric 
19662754fe60SDimitry Andric   return false;
19672754fe60SDimitry Andric }
19682754fe60SDimitry Andric 
1969f22ef01cSRoman Divacky /// ParseMetadataAttachment - Parse metadata attachments.
1970f22ef01cSRoman Divacky bool BitcodeReader::ParseMetadataAttachment() {
1971f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::METADATA_ATTACHMENT_ID))
1972f22ef01cSRoman Divacky     return Error("Malformed block record");
1973f22ef01cSRoman Divacky 
1974f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1975f22ef01cSRoman Divacky   while(1) {
1976f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
1977f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
1978f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
1979f22ef01cSRoman Divacky         return Error("Error at end of PARAMATTR block");
1980f22ef01cSRoman Divacky       break;
1981f22ef01cSRoman Divacky     }
1982f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
1983f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
1984f22ef01cSRoman Divacky       continue;
1985f22ef01cSRoman Divacky     }
1986f22ef01cSRoman Divacky     // Read a metadata attachment record.
1987f22ef01cSRoman Divacky     Record.clear();
1988f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
1989f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
1990f22ef01cSRoman Divacky       break;
199117a519f9SDimitry Andric     case bitc::METADATA_ATTACHMENT: {
1992f22ef01cSRoman Divacky       unsigned RecordLength = Record.size();
1993f22ef01cSRoman Divacky       if (Record.empty() || (RecordLength - 1) % 2 == 1)
1994f22ef01cSRoman Divacky         return Error ("Invalid METADATA_ATTACHMENT reader!");
1995f22ef01cSRoman Divacky       Instruction *Inst = InstructionList[Record[0]];
1996f22ef01cSRoman Divacky       for (unsigned i = 1; i != RecordLength; i = i+2) {
1997f22ef01cSRoman Divacky         unsigned Kind = Record[i];
1998e580952dSDimitry Andric         DenseMap<unsigned, unsigned>::iterator I =
1999e580952dSDimitry Andric           MDKindMap.find(Kind);
2000e580952dSDimitry Andric         if (I == MDKindMap.end())
2001e580952dSDimitry Andric           return Error("Invalid metadata kind ID");
2002f22ef01cSRoman Divacky         Value *Node = MDValueList.getValueFwdRef(Record[i+1]);
2003e580952dSDimitry Andric         Inst->setMetadata(I->second, cast<MDNode>(Node));
2004f22ef01cSRoman Divacky       }
2005f22ef01cSRoman Divacky       break;
2006f22ef01cSRoman Divacky     }
2007f22ef01cSRoman Divacky     }
2008f22ef01cSRoman Divacky   }
2009f22ef01cSRoman Divacky   return false;
2010f22ef01cSRoman Divacky }
2011f22ef01cSRoman Divacky 
2012f22ef01cSRoman Divacky /// ParseFunctionBody - Lazily parse the specified function body block.
2013f22ef01cSRoman Divacky bool BitcodeReader::ParseFunctionBody(Function *F) {
2014f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::FUNCTION_BLOCK_ID))
2015f22ef01cSRoman Divacky     return Error("Malformed block record");
2016f22ef01cSRoman Divacky 
2017f22ef01cSRoman Divacky   InstructionList.clear();
2018f22ef01cSRoman Divacky   unsigned ModuleValueListSize = ValueList.size();
2019e580952dSDimitry Andric   unsigned ModuleMDValueListSize = MDValueList.size();
2020f22ef01cSRoman Divacky 
2021f22ef01cSRoman Divacky   // Add all the function arguments to the value table.
2022f22ef01cSRoman Divacky   for(Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
2023f22ef01cSRoman Divacky     ValueList.push_back(I);
2024f22ef01cSRoman Divacky 
2025f22ef01cSRoman Divacky   unsigned NextValueNo = ValueList.size();
2026f22ef01cSRoman Divacky   BasicBlock *CurBB = 0;
2027f22ef01cSRoman Divacky   unsigned CurBBNo = 0;
2028f22ef01cSRoman Divacky 
2029f22ef01cSRoman Divacky   DebugLoc LastLoc;
2030f22ef01cSRoman Divacky 
2031f22ef01cSRoman Divacky   // Read all the records.
2032f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
2033f22ef01cSRoman Divacky   while (1) {
2034f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
2035f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
2036f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
2037f22ef01cSRoman Divacky         return Error("Error at end of function block");
2038f22ef01cSRoman Divacky       break;
2039f22ef01cSRoman Divacky     }
2040f22ef01cSRoman Divacky 
2041f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
2042f22ef01cSRoman Divacky       switch (Stream.ReadSubBlockID()) {
2043f22ef01cSRoman Divacky       default:  // Skip unknown content.
2044f22ef01cSRoman Divacky         if (Stream.SkipBlock())
2045f22ef01cSRoman Divacky           return Error("Malformed block record");
2046f22ef01cSRoman Divacky         break;
2047f22ef01cSRoman Divacky       case bitc::CONSTANTS_BLOCK_ID:
2048f22ef01cSRoman Divacky         if (ParseConstants()) return true;
2049f22ef01cSRoman Divacky         NextValueNo = ValueList.size();
2050f22ef01cSRoman Divacky         break;
2051f22ef01cSRoman Divacky       case bitc::VALUE_SYMTAB_BLOCK_ID:
2052f22ef01cSRoman Divacky         if (ParseValueSymbolTable()) return true;
2053f22ef01cSRoman Divacky         break;
2054f22ef01cSRoman Divacky       case bitc::METADATA_ATTACHMENT_ID:
2055f22ef01cSRoman Divacky         if (ParseMetadataAttachment()) return true;
2056f22ef01cSRoman Divacky         break;
2057f22ef01cSRoman Divacky       case bitc::METADATA_BLOCK_ID:
2058f22ef01cSRoman Divacky         if (ParseMetadata()) return true;
2059f22ef01cSRoman Divacky         break;
2060f22ef01cSRoman Divacky       }
2061f22ef01cSRoman Divacky       continue;
2062f22ef01cSRoman Divacky     }
2063f22ef01cSRoman Divacky 
2064f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
2065f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
2066f22ef01cSRoman Divacky       continue;
2067f22ef01cSRoman Divacky     }
2068f22ef01cSRoman Divacky 
2069f22ef01cSRoman Divacky     // Read a record.
2070f22ef01cSRoman Divacky     Record.clear();
2071f22ef01cSRoman Divacky     Instruction *I = 0;
2072f22ef01cSRoman Divacky     unsigned BitCode = Stream.ReadRecord(Code, Record);
2073f22ef01cSRoman Divacky     switch (BitCode) {
2074f22ef01cSRoman Divacky     default: // Default behavior: reject
2075f22ef01cSRoman Divacky       return Error("Unknown instruction");
2076f22ef01cSRoman Divacky     case bitc::FUNC_CODE_DECLAREBLOCKS:     // DECLAREBLOCKS: [nblocks]
2077f22ef01cSRoman Divacky       if (Record.size() < 1 || Record[0] == 0)
2078f22ef01cSRoman Divacky         return Error("Invalid DECLAREBLOCKS record");
2079f22ef01cSRoman Divacky       // Create all the basic blocks for the function.
2080f22ef01cSRoman Divacky       FunctionBBs.resize(Record[0]);
2081f22ef01cSRoman Divacky       for (unsigned i = 0, e = FunctionBBs.size(); i != e; ++i)
2082f22ef01cSRoman Divacky         FunctionBBs[i] = BasicBlock::Create(Context, "", F);
2083f22ef01cSRoman Divacky       CurBB = FunctionBBs[0];
2084f22ef01cSRoman Divacky       continue;
2085f22ef01cSRoman Divacky 
2086f22ef01cSRoman Divacky     case bitc::FUNC_CODE_DEBUG_LOC_AGAIN:  // DEBUG_LOC_AGAIN
2087f22ef01cSRoman Divacky       // This record indicates that the last instruction is at the same
2088f22ef01cSRoman Divacky       // location as the previous instruction with a location.
2089f22ef01cSRoman Divacky       I = 0;
2090f22ef01cSRoman Divacky 
2091f22ef01cSRoman Divacky       // Get the last instruction emitted.
2092f22ef01cSRoman Divacky       if (CurBB && !CurBB->empty())
2093f22ef01cSRoman Divacky         I = &CurBB->back();
2094f22ef01cSRoman Divacky       else if (CurBBNo && FunctionBBs[CurBBNo-1] &&
2095f22ef01cSRoman Divacky                !FunctionBBs[CurBBNo-1]->empty())
2096f22ef01cSRoman Divacky         I = &FunctionBBs[CurBBNo-1]->back();
2097f22ef01cSRoman Divacky 
2098f22ef01cSRoman Divacky       if (I == 0) return Error("Invalid DEBUG_LOC_AGAIN record");
2099f22ef01cSRoman Divacky       I->setDebugLoc(LastLoc);
2100f22ef01cSRoman Divacky       I = 0;
2101f22ef01cSRoman Divacky       continue;
2102f22ef01cSRoman Divacky 
210317a519f9SDimitry Andric     case bitc::FUNC_CODE_DEBUG_LOC: {      // DEBUG_LOC: [line, col, scope, ia]
2104f22ef01cSRoman Divacky       I = 0;     // Get the last instruction emitted.
2105f22ef01cSRoman Divacky       if (CurBB && !CurBB->empty())
2106f22ef01cSRoman Divacky         I = &CurBB->back();
2107f22ef01cSRoman Divacky       else if (CurBBNo && FunctionBBs[CurBBNo-1] &&
2108f22ef01cSRoman Divacky                !FunctionBBs[CurBBNo-1]->empty())
2109f22ef01cSRoman Divacky         I = &FunctionBBs[CurBBNo-1]->back();
2110f22ef01cSRoman Divacky       if (I == 0 || Record.size() < 4)
2111f22ef01cSRoman Divacky         return Error("Invalid FUNC_CODE_DEBUG_LOC record");
2112f22ef01cSRoman Divacky 
2113f22ef01cSRoman Divacky       unsigned Line = Record[0], Col = Record[1];
2114f22ef01cSRoman Divacky       unsigned ScopeID = Record[2], IAID = Record[3];
2115f22ef01cSRoman Divacky 
2116f22ef01cSRoman Divacky       MDNode *Scope = 0, *IA = 0;
2117f22ef01cSRoman Divacky       if (ScopeID) Scope = cast<MDNode>(MDValueList.getValueFwdRef(ScopeID-1));
2118f22ef01cSRoman Divacky       if (IAID)    IA = cast<MDNode>(MDValueList.getValueFwdRef(IAID-1));
2119f22ef01cSRoman Divacky       LastLoc = DebugLoc::get(Line, Col, Scope, IA);
2120f22ef01cSRoman Divacky       I->setDebugLoc(LastLoc);
2121f22ef01cSRoman Divacky       I = 0;
2122f22ef01cSRoman Divacky       continue;
2123f22ef01cSRoman Divacky     }
2124f22ef01cSRoman Divacky 
2125f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_BINOP: {    // BINOP: [opval, ty, opval, opcode]
2126f22ef01cSRoman Divacky       unsigned OpNum = 0;
2127f22ef01cSRoman Divacky       Value *LHS, *RHS;
2128f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
2129f22ef01cSRoman Divacky           getValue(Record, OpNum, LHS->getType(), RHS) ||
2130f22ef01cSRoman Divacky           OpNum+1 > Record.size())
2131f22ef01cSRoman Divacky         return Error("Invalid BINOP record");
2132f22ef01cSRoman Divacky 
2133f22ef01cSRoman Divacky       int Opc = GetDecodedBinaryOpcode(Record[OpNum++], LHS->getType());
2134f22ef01cSRoman Divacky       if (Opc == -1) return Error("Invalid BINOP record");
2135f22ef01cSRoman Divacky       I = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
2136f22ef01cSRoman Divacky       InstructionList.push_back(I);
2137f22ef01cSRoman Divacky       if (OpNum < Record.size()) {
2138f22ef01cSRoman Divacky         if (Opc == Instruction::Add ||
2139f22ef01cSRoman Divacky             Opc == Instruction::Sub ||
21402754fe60SDimitry Andric             Opc == Instruction::Mul ||
21412754fe60SDimitry Andric             Opc == Instruction::Shl) {
2142f22ef01cSRoman Divacky           if (Record[OpNum] & (1 << bitc::OBO_NO_SIGNED_WRAP))
2143f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setHasNoSignedWrap(true);
2144f22ef01cSRoman Divacky           if (Record[OpNum] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
2145f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setHasNoUnsignedWrap(true);
21462754fe60SDimitry Andric         } else if (Opc == Instruction::SDiv ||
21472754fe60SDimitry Andric                    Opc == Instruction::UDiv ||
21482754fe60SDimitry Andric                    Opc == Instruction::LShr ||
21492754fe60SDimitry Andric                    Opc == Instruction::AShr) {
21502754fe60SDimitry Andric           if (Record[OpNum] & (1 << bitc::PEO_EXACT))
2151f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setIsExact(true);
2152f22ef01cSRoman Divacky         }
2153f22ef01cSRoman Divacky       }
2154f22ef01cSRoman Divacky       break;
2155f22ef01cSRoman Divacky     }
2156f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CAST: {    // CAST: [opval, opty, destty, castopc]
2157f22ef01cSRoman Divacky       unsigned OpNum = 0;
2158f22ef01cSRoman Divacky       Value *Op;
2159f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2160f22ef01cSRoman Divacky           OpNum+2 != Record.size())
2161f22ef01cSRoman Divacky         return Error("Invalid CAST record");
2162f22ef01cSRoman Divacky 
2163f22ef01cSRoman Divacky       const Type *ResTy = getTypeByID(Record[OpNum]);
2164f22ef01cSRoman Divacky       int Opc = GetDecodedCastOpcode(Record[OpNum+1]);
2165f22ef01cSRoman Divacky       if (Opc == -1 || ResTy == 0)
2166f22ef01cSRoman Divacky         return Error("Invalid CAST record");
2167f22ef01cSRoman Divacky       I = CastInst::Create((Instruction::CastOps)Opc, Op, ResTy);
2168f22ef01cSRoman Divacky       InstructionList.push_back(I);
2169f22ef01cSRoman Divacky       break;
2170f22ef01cSRoman Divacky     }
2171f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INBOUNDS_GEP:
2172f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_GEP: { // GEP: [n x operands]
2173f22ef01cSRoman Divacky       unsigned OpNum = 0;
2174f22ef01cSRoman Divacky       Value *BasePtr;
2175f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, BasePtr))
2176f22ef01cSRoman Divacky         return Error("Invalid GEP record");
2177f22ef01cSRoman Divacky 
2178f22ef01cSRoman Divacky       SmallVector<Value*, 16> GEPIdx;
2179f22ef01cSRoman Divacky       while (OpNum != Record.size()) {
2180f22ef01cSRoman Divacky         Value *Op;
2181f22ef01cSRoman Divacky         if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2182f22ef01cSRoman Divacky           return Error("Invalid GEP record");
2183f22ef01cSRoman Divacky         GEPIdx.push_back(Op);
2184f22ef01cSRoman Divacky       }
2185f22ef01cSRoman Divacky 
2186f22ef01cSRoman Divacky       I = GetElementPtrInst::Create(BasePtr, GEPIdx.begin(), GEPIdx.end());
2187f22ef01cSRoman Divacky       InstructionList.push_back(I);
2188f22ef01cSRoman Divacky       if (BitCode == bitc::FUNC_CODE_INST_INBOUNDS_GEP)
2189f22ef01cSRoman Divacky         cast<GetElementPtrInst>(I)->setIsInBounds(true);
2190f22ef01cSRoman Divacky       break;
2191f22ef01cSRoman Divacky     }
2192f22ef01cSRoman Divacky 
2193f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_EXTRACTVAL: {
2194f22ef01cSRoman Divacky                                        // EXTRACTVAL: [opty, opval, n x indices]
2195f22ef01cSRoman Divacky       unsigned OpNum = 0;
2196f22ef01cSRoman Divacky       Value *Agg;
2197f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
2198f22ef01cSRoman Divacky         return Error("Invalid EXTRACTVAL record");
2199f22ef01cSRoman Divacky 
2200f22ef01cSRoman Divacky       SmallVector<unsigned, 4> EXTRACTVALIdx;
2201f22ef01cSRoman Divacky       for (unsigned RecSize = Record.size();
2202f22ef01cSRoman Divacky            OpNum != RecSize; ++OpNum) {
2203f22ef01cSRoman Divacky         uint64_t Index = Record[OpNum];
2204f22ef01cSRoman Divacky         if ((unsigned)Index != Index)
2205f22ef01cSRoman Divacky           return Error("Invalid EXTRACTVAL index");
2206f22ef01cSRoman Divacky         EXTRACTVALIdx.push_back((unsigned)Index);
2207f22ef01cSRoman Divacky       }
2208f22ef01cSRoman Divacky 
220917a519f9SDimitry Andric       I = ExtractValueInst::Create(Agg, EXTRACTVALIdx);
2210f22ef01cSRoman Divacky       InstructionList.push_back(I);
2211f22ef01cSRoman Divacky       break;
2212f22ef01cSRoman Divacky     }
2213f22ef01cSRoman Divacky 
2214f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INSERTVAL: {
2215f22ef01cSRoman Divacky                            // INSERTVAL: [opty, opval, opty, opval, n x indices]
2216f22ef01cSRoman Divacky       unsigned OpNum = 0;
2217f22ef01cSRoman Divacky       Value *Agg;
2218f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
2219f22ef01cSRoman Divacky         return Error("Invalid INSERTVAL record");
2220f22ef01cSRoman Divacky       Value *Val;
2221f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Val))
2222f22ef01cSRoman Divacky         return Error("Invalid INSERTVAL record");
2223f22ef01cSRoman Divacky 
2224f22ef01cSRoman Divacky       SmallVector<unsigned, 4> INSERTVALIdx;
2225f22ef01cSRoman Divacky       for (unsigned RecSize = Record.size();
2226f22ef01cSRoman Divacky            OpNum != RecSize; ++OpNum) {
2227f22ef01cSRoman Divacky         uint64_t Index = Record[OpNum];
2228f22ef01cSRoman Divacky         if ((unsigned)Index != Index)
2229f22ef01cSRoman Divacky           return Error("Invalid INSERTVAL index");
2230f22ef01cSRoman Divacky         INSERTVALIdx.push_back((unsigned)Index);
2231f22ef01cSRoman Divacky       }
2232f22ef01cSRoman Divacky 
223317a519f9SDimitry Andric       I = InsertValueInst::Create(Agg, Val, INSERTVALIdx);
2234f22ef01cSRoman Divacky       InstructionList.push_back(I);
2235f22ef01cSRoman Divacky       break;
2236f22ef01cSRoman Divacky     }
2237f22ef01cSRoman Divacky 
2238f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SELECT: { // SELECT: [opval, ty, opval, opval]
2239f22ef01cSRoman Divacky       // obsolete form of select
2240f22ef01cSRoman Divacky       // handles select i1 ... in old bitcode
2241f22ef01cSRoman Divacky       unsigned OpNum = 0;
2242f22ef01cSRoman Divacky       Value *TrueVal, *FalseVal, *Cond;
2243f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
2244f22ef01cSRoman Divacky           getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
2245f22ef01cSRoman Divacky           getValue(Record, OpNum, Type::getInt1Ty(Context), Cond))
2246f22ef01cSRoman Divacky         return Error("Invalid SELECT record");
2247f22ef01cSRoman Divacky 
2248f22ef01cSRoman Divacky       I = SelectInst::Create(Cond, TrueVal, FalseVal);
2249f22ef01cSRoman Divacky       InstructionList.push_back(I);
2250f22ef01cSRoman Divacky       break;
2251f22ef01cSRoman Divacky     }
2252f22ef01cSRoman Divacky 
2253f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_VSELECT: {// VSELECT: [ty,opval,opval,predty,pred]
2254f22ef01cSRoman Divacky       // new form of select
2255f22ef01cSRoman Divacky       // handles select i1 or select [N x i1]
2256f22ef01cSRoman Divacky       unsigned OpNum = 0;
2257f22ef01cSRoman Divacky       Value *TrueVal, *FalseVal, *Cond;
2258f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
2259f22ef01cSRoman Divacky           getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
2260f22ef01cSRoman Divacky           getValueTypePair(Record, OpNum, NextValueNo, Cond))
2261f22ef01cSRoman Divacky         return Error("Invalid SELECT record");
2262f22ef01cSRoman Divacky 
2263f22ef01cSRoman Divacky       // select condition can be either i1 or [N x i1]
2264f22ef01cSRoman Divacky       if (const VectorType* vector_type =
2265f22ef01cSRoman Divacky           dyn_cast<const VectorType>(Cond->getType())) {
2266f22ef01cSRoman Divacky         // expect <n x i1>
2267f22ef01cSRoman Divacky         if (vector_type->getElementType() != Type::getInt1Ty(Context))
2268f22ef01cSRoman Divacky           return Error("Invalid SELECT condition type");
2269f22ef01cSRoman Divacky       } else {
2270f22ef01cSRoman Divacky         // expect i1
2271f22ef01cSRoman Divacky         if (Cond->getType() != Type::getInt1Ty(Context))
2272f22ef01cSRoman Divacky           return Error("Invalid SELECT condition type");
2273f22ef01cSRoman Divacky       }
2274f22ef01cSRoman Divacky 
2275f22ef01cSRoman Divacky       I = SelectInst::Create(Cond, TrueVal, FalseVal);
2276f22ef01cSRoman Divacky       InstructionList.push_back(I);
2277f22ef01cSRoman Divacky       break;
2278f22ef01cSRoman Divacky     }
2279f22ef01cSRoman Divacky 
2280f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_EXTRACTELT: { // EXTRACTELT: [opty, opval, opval]
2281f22ef01cSRoman Divacky       unsigned OpNum = 0;
2282f22ef01cSRoman Divacky       Value *Vec, *Idx;
2283f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
2284f22ef01cSRoman Divacky           getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
2285f22ef01cSRoman Divacky         return Error("Invalid EXTRACTELT record");
2286f22ef01cSRoman Divacky       I = ExtractElementInst::Create(Vec, Idx);
2287f22ef01cSRoman Divacky       InstructionList.push_back(I);
2288f22ef01cSRoman Divacky       break;
2289f22ef01cSRoman Divacky     }
2290f22ef01cSRoman Divacky 
2291f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INSERTELT: { // INSERTELT: [ty, opval,opval,opval]
2292f22ef01cSRoman Divacky       unsigned OpNum = 0;
2293f22ef01cSRoman Divacky       Value *Vec, *Elt, *Idx;
2294f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
2295f22ef01cSRoman Divacky           getValue(Record, OpNum,
2296f22ef01cSRoman Divacky                    cast<VectorType>(Vec->getType())->getElementType(), Elt) ||
2297f22ef01cSRoman Divacky           getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
2298f22ef01cSRoman Divacky         return Error("Invalid INSERTELT record");
2299f22ef01cSRoman Divacky       I = InsertElementInst::Create(Vec, Elt, Idx);
2300f22ef01cSRoman Divacky       InstructionList.push_back(I);
2301f22ef01cSRoman Divacky       break;
2302f22ef01cSRoman Divacky     }
2303f22ef01cSRoman Divacky 
2304f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SHUFFLEVEC: {// SHUFFLEVEC: [opval,ty,opval,opval]
2305f22ef01cSRoman Divacky       unsigned OpNum = 0;
2306f22ef01cSRoman Divacky       Value *Vec1, *Vec2, *Mask;
2307f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec1) ||
2308f22ef01cSRoman Divacky           getValue(Record, OpNum, Vec1->getType(), Vec2))
2309f22ef01cSRoman Divacky         return Error("Invalid SHUFFLEVEC record");
2310f22ef01cSRoman Divacky 
2311f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Mask))
2312f22ef01cSRoman Divacky         return Error("Invalid SHUFFLEVEC record");
2313f22ef01cSRoman Divacky       I = new ShuffleVectorInst(Vec1, Vec2, Mask);
2314f22ef01cSRoman Divacky       InstructionList.push_back(I);
2315f22ef01cSRoman Divacky       break;
2316f22ef01cSRoman Divacky     }
2317f22ef01cSRoman Divacky 
2318f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CMP:   // CMP: [opty, opval, opval, pred]
2319f22ef01cSRoman Divacky       // Old form of ICmp/FCmp returning bool
2320f22ef01cSRoman Divacky       // Existed to differentiate between icmp/fcmp and vicmp/vfcmp which were
2321f22ef01cSRoman Divacky       // both legal on vectors but had different behaviour.
2322f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CMP2: { // CMP2: [opty, opval, opval, pred]
2323f22ef01cSRoman Divacky       // FCmp/ICmp returning bool or vector of bool
2324f22ef01cSRoman Divacky 
2325f22ef01cSRoman Divacky       unsigned OpNum = 0;
2326f22ef01cSRoman Divacky       Value *LHS, *RHS;
2327f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
2328f22ef01cSRoman Divacky           getValue(Record, OpNum, LHS->getType(), RHS) ||
2329f22ef01cSRoman Divacky           OpNum+1 != Record.size())
2330f22ef01cSRoman Divacky         return Error("Invalid CMP record");
2331f22ef01cSRoman Divacky 
2332f22ef01cSRoman Divacky       if (LHS->getType()->isFPOrFPVectorTy())
2333f22ef01cSRoman Divacky         I = new FCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
2334f22ef01cSRoman Divacky       else
2335f22ef01cSRoman Divacky         I = new ICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
2336f22ef01cSRoman Divacky       InstructionList.push_back(I);
2337f22ef01cSRoman Divacky       break;
2338f22ef01cSRoman Divacky     }
2339f22ef01cSRoman Divacky 
2340f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_RET: // RET: [opty,opval<optional>]
2341f22ef01cSRoman Divacky       {
2342f22ef01cSRoman Divacky         unsigned Size = Record.size();
2343f22ef01cSRoman Divacky         if (Size == 0) {
2344f22ef01cSRoman Divacky           I = ReturnInst::Create(Context);
2345f22ef01cSRoman Divacky           InstructionList.push_back(I);
2346f22ef01cSRoman Divacky           break;
2347f22ef01cSRoman Divacky         }
2348f22ef01cSRoman Divacky 
2349f22ef01cSRoman Divacky         unsigned OpNum = 0;
2350f22ef01cSRoman Divacky         Value *Op = NULL;
2351f22ef01cSRoman Divacky         if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2352f22ef01cSRoman Divacky           return Error("Invalid RET record");
235317a519f9SDimitry Andric         if (OpNum != Record.size())
235417a519f9SDimitry Andric           return Error("Invalid RET record");
2355f22ef01cSRoman Divacky 
235617a519f9SDimitry Andric         I = ReturnInst::Create(Context, Op);
2357f22ef01cSRoman Divacky         InstructionList.push_back(I);
2358f22ef01cSRoman Divacky         break;
2359f22ef01cSRoman Divacky       }
2360f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_BR: { // BR: [bb#, bb#, opval] or [bb#]
2361f22ef01cSRoman Divacky       if (Record.size() != 1 && Record.size() != 3)
2362f22ef01cSRoman Divacky         return Error("Invalid BR record");
2363f22ef01cSRoman Divacky       BasicBlock *TrueDest = getBasicBlock(Record[0]);
2364f22ef01cSRoman Divacky       if (TrueDest == 0)
2365f22ef01cSRoman Divacky         return Error("Invalid BR record");
2366f22ef01cSRoman Divacky 
2367f22ef01cSRoman Divacky       if (Record.size() == 1) {
2368f22ef01cSRoman Divacky         I = BranchInst::Create(TrueDest);
2369f22ef01cSRoman Divacky         InstructionList.push_back(I);
2370f22ef01cSRoman Divacky       }
2371f22ef01cSRoman Divacky       else {
2372f22ef01cSRoman Divacky         BasicBlock *FalseDest = getBasicBlock(Record[1]);
2373f22ef01cSRoman Divacky         Value *Cond = getFnValueByID(Record[2], Type::getInt1Ty(Context));
2374f22ef01cSRoman Divacky         if (FalseDest == 0 || Cond == 0)
2375f22ef01cSRoman Divacky           return Error("Invalid BR record");
2376f22ef01cSRoman Divacky         I = BranchInst::Create(TrueDest, FalseDest, Cond);
2377f22ef01cSRoman Divacky         InstructionList.push_back(I);
2378f22ef01cSRoman Divacky       }
2379f22ef01cSRoman Divacky       break;
2380f22ef01cSRoman Divacky     }
2381f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SWITCH: { // SWITCH: [opty, op0, op1, ...]
2382f22ef01cSRoman Divacky       if (Record.size() < 3 || (Record.size() & 1) == 0)
2383f22ef01cSRoman Divacky         return Error("Invalid SWITCH record");
2384f22ef01cSRoman Divacky       const Type *OpTy = getTypeByID(Record[0]);
2385f22ef01cSRoman Divacky       Value *Cond = getFnValueByID(Record[1], OpTy);
2386f22ef01cSRoman Divacky       BasicBlock *Default = getBasicBlock(Record[2]);
2387f22ef01cSRoman Divacky       if (OpTy == 0 || Cond == 0 || Default == 0)
2388f22ef01cSRoman Divacky         return Error("Invalid SWITCH record");
2389f22ef01cSRoman Divacky       unsigned NumCases = (Record.size()-3)/2;
2390f22ef01cSRoman Divacky       SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
2391f22ef01cSRoman Divacky       InstructionList.push_back(SI);
2392f22ef01cSRoman Divacky       for (unsigned i = 0, e = NumCases; i != e; ++i) {
2393f22ef01cSRoman Divacky         ConstantInt *CaseVal =
2394f22ef01cSRoman Divacky           dyn_cast_or_null<ConstantInt>(getFnValueByID(Record[3+i*2], OpTy));
2395f22ef01cSRoman Divacky         BasicBlock *DestBB = getBasicBlock(Record[1+3+i*2]);
2396f22ef01cSRoman Divacky         if (CaseVal == 0 || DestBB == 0) {
2397f22ef01cSRoman Divacky           delete SI;
2398f22ef01cSRoman Divacky           return Error("Invalid SWITCH record!");
2399f22ef01cSRoman Divacky         }
2400f22ef01cSRoman Divacky         SI->addCase(CaseVal, DestBB);
2401f22ef01cSRoman Divacky       }
2402f22ef01cSRoman Divacky       I = SI;
2403f22ef01cSRoman Divacky       break;
2404f22ef01cSRoman Divacky     }
2405f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INDIRECTBR: { // INDIRECTBR: [opty, op0, op1, ...]
2406f22ef01cSRoman Divacky       if (Record.size() < 2)
2407f22ef01cSRoman Divacky         return Error("Invalid INDIRECTBR record");
2408f22ef01cSRoman Divacky       const Type *OpTy = getTypeByID(Record[0]);
2409f22ef01cSRoman Divacky       Value *Address = getFnValueByID(Record[1], OpTy);
2410f22ef01cSRoman Divacky       if (OpTy == 0 || Address == 0)
2411f22ef01cSRoman Divacky         return Error("Invalid INDIRECTBR record");
2412f22ef01cSRoman Divacky       unsigned NumDests = Record.size()-2;
2413f22ef01cSRoman Divacky       IndirectBrInst *IBI = IndirectBrInst::Create(Address, NumDests);
2414f22ef01cSRoman Divacky       InstructionList.push_back(IBI);
2415f22ef01cSRoman Divacky       for (unsigned i = 0, e = NumDests; i != e; ++i) {
2416f22ef01cSRoman Divacky         if (BasicBlock *DestBB = getBasicBlock(Record[2+i])) {
2417f22ef01cSRoman Divacky           IBI->addDestination(DestBB);
2418f22ef01cSRoman Divacky         } else {
2419f22ef01cSRoman Divacky           delete IBI;
2420f22ef01cSRoman Divacky           return Error("Invalid INDIRECTBR record!");
2421f22ef01cSRoman Divacky         }
2422f22ef01cSRoman Divacky       }
2423f22ef01cSRoman Divacky       I = IBI;
2424f22ef01cSRoman Divacky       break;
2425f22ef01cSRoman Divacky     }
2426f22ef01cSRoman Divacky 
2427f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INVOKE: {
2428f22ef01cSRoman Divacky       // INVOKE: [attrs, cc, normBB, unwindBB, fnty, op0,op1,op2, ...]
2429f22ef01cSRoman Divacky       if (Record.size() < 4) return Error("Invalid INVOKE record");
2430f22ef01cSRoman Divacky       AttrListPtr PAL = getAttributes(Record[0]);
2431f22ef01cSRoman Divacky       unsigned CCInfo = Record[1];
2432f22ef01cSRoman Divacky       BasicBlock *NormalBB = getBasicBlock(Record[2]);
2433f22ef01cSRoman Divacky       BasicBlock *UnwindBB = getBasicBlock(Record[3]);
2434f22ef01cSRoman Divacky 
2435f22ef01cSRoman Divacky       unsigned OpNum = 4;
2436f22ef01cSRoman Divacky       Value *Callee;
2437f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
2438f22ef01cSRoman Divacky         return Error("Invalid INVOKE record");
2439f22ef01cSRoman Divacky 
2440f22ef01cSRoman Divacky       const PointerType *CalleeTy = dyn_cast<PointerType>(Callee->getType());
2441f22ef01cSRoman Divacky       const FunctionType *FTy = !CalleeTy ? 0 :
2442f22ef01cSRoman Divacky         dyn_cast<FunctionType>(CalleeTy->getElementType());
2443f22ef01cSRoman Divacky 
2444f22ef01cSRoman Divacky       // Check that the right number of fixed parameters are here.
2445f22ef01cSRoman Divacky       if (FTy == 0 || NormalBB == 0 || UnwindBB == 0 ||
2446f22ef01cSRoman Divacky           Record.size() < OpNum+FTy->getNumParams())
2447f22ef01cSRoman Divacky         return Error("Invalid INVOKE record");
2448f22ef01cSRoman Divacky 
2449f22ef01cSRoman Divacky       SmallVector<Value*, 16> Ops;
2450f22ef01cSRoman Divacky       for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
2451f22ef01cSRoman Divacky         Ops.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
2452f22ef01cSRoman Divacky         if (Ops.back() == 0) return Error("Invalid INVOKE record");
2453f22ef01cSRoman Divacky       }
2454f22ef01cSRoman Divacky 
2455f22ef01cSRoman Divacky       if (!FTy->isVarArg()) {
2456f22ef01cSRoman Divacky         if (Record.size() != OpNum)
2457f22ef01cSRoman Divacky           return Error("Invalid INVOKE record");
2458f22ef01cSRoman Divacky       } else {
2459f22ef01cSRoman Divacky         // Read type/value pairs for varargs params.
2460f22ef01cSRoman Divacky         while (OpNum != Record.size()) {
2461f22ef01cSRoman Divacky           Value *Op;
2462f22ef01cSRoman Divacky           if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2463f22ef01cSRoman Divacky             return Error("Invalid INVOKE record");
2464f22ef01cSRoman Divacky           Ops.push_back(Op);
2465f22ef01cSRoman Divacky         }
2466f22ef01cSRoman Divacky       }
2467f22ef01cSRoman Divacky 
246817a519f9SDimitry Andric       I = InvokeInst::Create(Callee, NormalBB, UnwindBB, Ops);
2469f22ef01cSRoman Divacky       InstructionList.push_back(I);
2470f22ef01cSRoman Divacky       cast<InvokeInst>(I)->setCallingConv(
2471f22ef01cSRoman Divacky         static_cast<CallingConv::ID>(CCInfo));
2472f22ef01cSRoman Divacky       cast<InvokeInst>(I)->setAttributes(PAL);
2473f22ef01cSRoman Divacky       break;
2474f22ef01cSRoman Divacky     }
2475f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_UNWIND: // UNWIND
2476f22ef01cSRoman Divacky       I = new UnwindInst(Context);
2477f22ef01cSRoman Divacky       InstructionList.push_back(I);
2478f22ef01cSRoman Divacky       break;
2479f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_UNREACHABLE: // UNREACHABLE
2480f22ef01cSRoman Divacky       I = new UnreachableInst(Context);
2481f22ef01cSRoman Divacky       InstructionList.push_back(I);
2482f22ef01cSRoman Divacky       break;
2483f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_PHI: { // PHI: [ty, val0,bb0, ...]
2484f22ef01cSRoman Divacky       if (Record.size() < 1 || ((Record.size()-1)&1))
2485f22ef01cSRoman Divacky         return Error("Invalid PHI record");
2486f22ef01cSRoman Divacky       const Type *Ty = getTypeByID(Record[0]);
2487f22ef01cSRoman Divacky       if (!Ty) return Error("Invalid PHI record");
2488f22ef01cSRoman Divacky 
24893b0f4066SDimitry Andric       PHINode *PN = PHINode::Create(Ty, (Record.size()-1)/2);
2490f22ef01cSRoman Divacky       InstructionList.push_back(PN);
2491f22ef01cSRoman Divacky 
2492f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size()-1; i != e; i += 2) {
2493f22ef01cSRoman Divacky         Value *V = getFnValueByID(Record[1+i], Ty);
2494f22ef01cSRoman Divacky         BasicBlock *BB = getBasicBlock(Record[2+i]);
2495f22ef01cSRoman Divacky         if (!V || !BB) return Error("Invalid PHI record");
2496f22ef01cSRoman Divacky         PN->addIncoming(V, BB);
2497f22ef01cSRoman Divacky       }
2498f22ef01cSRoman Divacky       I = PN;
2499f22ef01cSRoman Divacky       break;
2500f22ef01cSRoman Divacky     }
2501f22ef01cSRoman Divacky 
250217a519f9SDimitry Andric     case bitc::FUNC_CODE_INST_ALLOCA: { // ALLOCA: [instty, opty, op, align]
250317a519f9SDimitry Andric       if (Record.size() != 4)
250417a519f9SDimitry Andric         return Error("Invalid ALLOCA record");
2505f22ef01cSRoman Divacky       const PointerType *Ty =
2506f22ef01cSRoman Divacky         dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
250717a519f9SDimitry Andric       const Type *OpTy = getTypeByID(Record[1]);
250817a519f9SDimitry Andric       Value *Size = getFnValueByID(Record[2], OpTy);
250917a519f9SDimitry Andric       unsigned Align = Record[3];
2510f22ef01cSRoman Divacky       if (!Ty || !Size) return Error("Invalid ALLOCA record");
2511f22ef01cSRoman Divacky       I = new AllocaInst(Ty->getElementType(), Size, (1 << Align) >> 1);
2512f22ef01cSRoman Divacky       InstructionList.push_back(I);
2513f22ef01cSRoman Divacky       break;
2514f22ef01cSRoman Divacky     }
2515f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_LOAD: { // LOAD: [opty, op, align, vol]
2516f22ef01cSRoman Divacky       unsigned OpNum = 0;
2517f22ef01cSRoman Divacky       Value *Op;
2518f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2519f22ef01cSRoman Divacky           OpNum+2 != Record.size())
2520f22ef01cSRoman Divacky         return Error("Invalid LOAD record");
2521f22ef01cSRoman Divacky 
2522f22ef01cSRoman Divacky       I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1);
2523f22ef01cSRoman Divacky       InstructionList.push_back(I);
2524f22ef01cSRoman Divacky       break;
2525f22ef01cSRoman Divacky     }
252617a519f9SDimitry Andric     case bitc::FUNC_CODE_INST_STORE: { // STORE2:[ptrty, ptr, val, align, vol]
2527f22ef01cSRoman Divacky       unsigned OpNum = 0;
2528f22ef01cSRoman Divacky       Value *Val, *Ptr;
2529f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
2530f22ef01cSRoman Divacky           getValue(Record, OpNum,
2531f22ef01cSRoman Divacky                     cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
2532f22ef01cSRoman Divacky           OpNum+2 != Record.size())
2533f22ef01cSRoman Divacky         return Error("Invalid STORE record");
2534f22ef01cSRoman Divacky 
2535f22ef01cSRoman Divacky       I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
2536f22ef01cSRoman Divacky       InstructionList.push_back(I);
2537f22ef01cSRoman Divacky       break;
2538f22ef01cSRoman Divacky     }
253917a519f9SDimitry Andric     case bitc::FUNC_CODE_INST_CALL: {
2540f22ef01cSRoman Divacky       // CALL: [paramattrs, cc, fnty, fnid, arg0, arg1...]
2541f22ef01cSRoman Divacky       if (Record.size() < 3)
2542f22ef01cSRoman Divacky         return Error("Invalid CALL record");
2543f22ef01cSRoman Divacky 
2544f22ef01cSRoman Divacky       AttrListPtr PAL = getAttributes(Record[0]);
2545f22ef01cSRoman Divacky       unsigned CCInfo = Record[1];
2546f22ef01cSRoman Divacky 
2547f22ef01cSRoman Divacky       unsigned OpNum = 2;
2548f22ef01cSRoman Divacky       Value *Callee;
2549f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
2550f22ef01cSRoman Divacky         return Error("Invalid CALL record");
2551f22ef01cSRoman Divacky 
2552f22ef01cSRoman Divacky       const PointerType *OpTy = dyn_cast<PointerType>(Callee->getType());
2553f22ef01cSRoman Divacky       const FunctionType *FTy = 0;
2554f22ef01cSRoman Divacky       if (OpTy) FTy = dyn_cast<FunctionType>(OpTy->getElementType());
2555f22ef01cSRoman Divacky       if (!FTy || Record.size() < FTy->getNumParams()+OpNum)
2556f22ef01cSRoman Divacky         return Error("Invalid CALL record");
2557f22ef01cSRoman Divacky 
2558f22ef01cSRoman Divacky       SmallVector<Value*, 16> Args;
2559f22ef01cSRoman Divacky       // Read the fixed params.
2560f22ef01cSRoman Divacky       for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
256117a519f9SDimitry Andric         if (FTy->getParamType(i)->isLabelTy())
2562f22ef01cSRoman Divacky           Args.push_back(getBasicBlock(Record[OpNum]));
2563f22ef01cSRoman Divacky         else
2564f22ef01cSRoman Divacky           Args.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
2565f22ef01cSRoman Divacky         if (Args.back() == 0) return Error("Invalid CALL record");
2566f22ef01cSRoman Divacky       }
2567f22ef01cSRoman Divacky 
2568f22ef01cSRoman Divacky       // Read type/value pairs for varargs params.
2569f22ef01cSRoman Divacky       if (!FTy->isVarArg()) {
2570f22ef01cSRoman Divacky         if (OpNum != Record.size())
2571f22ef01cSRoman Divacky           return Error("Invalid CALL record");
2572f22ef01cSRoman Divacky       } else {
2573f22ef01cSRoman Divacky         while (OpNum != Record.size()) {
2574f22ef01cSRoman Divacky           Value *Op;
2575f22ef01cSRoman Divacky           if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2576f22ef01cSRoman Divacky             return Error("Invalid CALL record");
2577f22ef01cSRoman Divacky           Args.push_back(Op);
2578f22ef01cSRoman Divacky         }
2579f22ef01cSRoman Divacky       }
2580f22ef01cSRoman Divacky 
258117a519f9SDimitry Andric       I = CallInst::Create(Callee, Args);
2582f22ef01cSRoman Divacky       InstructionList.push_back(I);
2583f22ef01cSRoman Divacky       cast<CallInst>(I)->setCallingConv(
2584f22ef01cSRoman Divacky         static_cast<CallingConv::ID>(CCInfo>>1));
2585f22ef01cSRoman Divacky       cast<CallInst>(I)->setTailCall(CCInfo & 1);
2586f22ef01cSRoman Divacky       cast<CallInst>(I)->setAttributes(PAL);
2587f22ef01cSRoman Divacky       break;
2588f22ef01cSRoman Divacky     }
2589f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_VAARG: { // VAARG: [valistty, valist, instty]
2590f22ef01cSRoman Divacky       if (Record.size() < 3)
2591f22ef01cSRoman Divacky         return Error("Invalid VAARG record");
2592f22ef01cSRoman Divacky       const Type *OpTy = getTypeByID(Record[0]);
2593f22ef01cSRoman Divacky       Value *Op = getFnValueByID(Record[1], OpTy);
2594f22ef01cSRoman Divacky       const Type *ResTy = getTypeByID(Record[2]);
2595f22ef01cSRoman Divacky       if (!OpTy || !Op || !ResTy)
2596f22ef01cSRoman Divacky         return Error("Invalid VAARG record");
2597f22ef01cSRoman Divacky       I = new VAArgInst(Op, ResTy);
2598f22ef01cSRoman Divacky       InstructionList.push_back(I);
2599f22ef01cSRoman Divacky       break;
2600f22ef01cSRoman Divacky     }
2601f22ef01cSRoman Divacky     }
2602f22ef01cSRoman Divacky 
2603f22ef01cSRoman Divacky     // Add instruction to end of current BB.  If there is no current BB, reject
2604f22ef01cSRoman Divacky     // this file.
2605f22ef01cSRoman Divacky     if (CurBB == 0) {
2606f22ef01cSRoman Divacky       delete I;
2607f22ef01cSRoman Divacky       return Error("Invalid instruction with no BB");
2608f22ef01cSRoman Divacky     }
2609f22ef01cSRoman Divacky     CurBB->getInstList().push_back(I);
2610f22ef01cSRoman Divacky 
2611f22ef01cSRoman Divacky     // If this was a terminator instruction, move to the next block.
2612f22ef01cSRoman Divacky     if (isa<TerminatorInst>(I)) {
2613f22ef01cSRoman Divacky       ++CurBBNo;
2614f22ef01cSRoman Divacky       CurBB = CurBBNo < FunctionBBs.size() ? FunctionBBs[CurBBNo] : 0;
2615f22ef01cSRoman Divacky     }
2616f22ef01cSRoman Divacky 
2617f22ef01cSRoman Divacky     // Non-void values get registered in the value table for future use.
2618f22ef01cSRoman Divacky     if (I && !I->getType()->isVoidTy())
2619f22ef01cSRoman Divacky       ValueList.AssignValue(I, NextValueNo++);
2620f22ef01cSRoman Divacky   }
2621f22ef01cSRoman Divacky 
2622f22ef01cSRoman Divacky   // Check the function list for unresolved values.
2623f22ef01cSRoman Divacky   if (Argument *A = dyn_cast<Argument>(ValueList.back())) {
2624f22ef01cSRoman Divacky     if (A->getParent() == 0) {
2625f22ef01cSRoman Divacky       // We found at least one unresolved value.  Nuke them all to avoid leaks.
2626f22ef01cSRoman Divacky       for (unsigned i = ModuleValueListSize, e = ValueList.size(); i != e; ++i){
2627e580952dSDimitry Andric         if ((A = dyn_cast<Argument>(ValueList[i])) && A->getParent() == 0) {
2628f22ef01cSRoman Divacky           A->replaceAllUsesWith(UndefValue::get(A->getType()));
2629f22ef01cSRoman Divacky           delete A;
2630f22ef01cSRoman Divacky         }
2631f22ef01cSRoman Divacky       }
2632f22ef01cSRoman Divacky       return Error("Never resolved value found in function!");
2633f22ef01cSRoman Divacky     }
2634f22ef01cSRoman Divacky   }
2635f22ef01cSRoman Divacky 
2636e580952dSDimitry Andric   // FIXME: Check for unresolved forward-declared metadata references
2637e580952dSDimitry Andric   // and clean up leaks.
2638e580952dSDimitry Andric 
2639f22ef01cSRoman Divacky   // See if anything took the address of blocks in this function.  If so,
2640f22ef01cSRoman Divacky   // resolve them now.
2641f22ef01cSRoman Divacky   DenseMap<Function*, std::vector<BlockAddrRefTy> >::iterator BAFRI =
2642f22ef01cSRoman Divacky     BlockAddrFwdRefs.find(F);
2643f22ef01cSRoman Divacky   if (BAFRI != BlockAddrFwdRefs.end()) {
2644f22ef01cSRoman Divacky     std::vector<BlockAddrRefTy> &RefList = BAFRI->second;
2645f22ef01cSRoman Divacky     for (unsigned i = 0, e = RefList.size(); i != e; ++i) {
2646f22ef01cSRoman Divacky       unsigned BlockIdx = RefList[i].first;
2647f22ef01cSRoman Divacky       if (BlockIdx >= FunctionBBs.size())
2648f22ef01cSRoman Divacky         return Error("Invalid blockaddress block #");
2649f22ef01cSRoman Divacky 
2650f22ef01cSRoman Divacky       GlobalVariable *FwdRef = RefList[i].second;
2651f22ef01cSRoman Divacky       FwdRef->replaceAllUsesWith(BlockAddress::get(F, FunctionBBs[BlockIdx]));
2652f22ef01cSRoman Divacky       FwdRef->eraseFromParent();
2653f22ef01cSRoman Divacky     }
2654f22ef01cSRoman Divacky 
2655f22ef01cSRoman Divacky     BlockAddrFwdRefs.erase(BAFRI);
2656f22ef01cSRoman Divacky   }
2657f22ef01cSRoman Divacky 
2658f22ef01cSRoman Divacky   // Trim the value list down to the size it was before we parsed this function.
2659f22ef01cSRoman Divacky   ValueList.shrinkTo(ModuleValueListSize);
2660e580952dSDimitry Andric   MDValueList.shrinkTo(ModuleMDValueListSize);
2661f22ef01cSRoman Divacky   std::vector<BasicBlock*>().swap(FunctionBBs);
2662f22ef01cSRoman Divacky   return false;
2663f22ef01cSRoman Divacky }
2664f22ef01cSRoman Divacky 
2665f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
2666f22ef01cSRoman Divacky // GVMaterializer implementation
2667f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
2668f22ef01cSRoman Divacky 
2669f22ef01cSRoman Divacky 
2670f22ef01cSRoman Divacky bool BitcodeReader::isMaterializable(const GlobalValue *GV) const {
2671f22ef01cSRoman Divacky   if (const Function *F = dyn_cast<Function>(GV)) {
2672f22ef01cSRoman Divacky     return F->isDeclaration() &&
2673f22ef01cSRoman Divacky       DeferredFunctionInfo.count(const_cast<Function*>(F));
2674f22ef01cSRoman Divacky   }
2675f22ef01cSRoman Divacky   return false;
2676f22ef01cSRoman Divacky }
2677f22ef01cSRoman Divacky 
2678f22ef01cSRoman Divacky bool BitcodeReader::Materialize(GlobalValue *GV, std::string *ErrInfo) {
2679f22ef01cSRoman Divacky   Function *F = dyn_cast<Function>(GV);
2680f22ef01cSRoman Divacky   // If it's not a function or is already material, ignore the request.
2681f22ef01cSRoman Divacky   if (!F || !F->isMaterializable()) return false;
2682f22ef01cSRoman Divacky 
2683f22ef01cSRoman Divacky   DenseMap<Function*, uint64_t>::iterator DFII = DeferredFunctionInfo.find(F);
2684f22ef01cSRoman Divacky   assert(DFII != DeferredFunctionInfo.end() && "Deferred function not found!");
2685f22ef01cSRoman Divacky 
2686f22ef01cSRoman Divacky   // Move the bit stream to the saved position of the deferred function body.
2687f22ef01cSRoman Divacky   Stream.JumpToBit(DFII->second);
2688f22ef01cSRoman Divacky 
2689f22ef01cSRoman Divacky   if (ParseFunctionBody(F)) {
2690f22ef01cSRoman Divacky     if (ErrInfo) *ErrInfo = ErrorString;
2691f22ef01cSRoman Divacky     return true;
2692f22ef01cSRoman Divacky   }
2693f22ef01cSRoman Divacky 
2694f22ef01cSRoman Divacky   // Upgrade any old intrinsic calls in the function.
2695f22ef01cSRoman Divacky   for (UpgradedIntrinsicMap::iterator I = UpgradedIntrinsics.begin(),
2696f22ef01cSRoman Divacky        E = UpgradedIntrinsics.end(); I != E; ++I) {
2697f22ef01cSRoman Divacky     if (I->first != I->second) {
2698f22ef01cSRoman Divacky       for (Value::use_iterator UI = I->first->use_begin(),
2699f22ef01cSRoman Divacky            UE = I->first->use_end(); UI != UE; ) {
2700f22ef01cSRoman Divacky         if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2701f22ef01cSRoman Divacky           UpgradeIntrinsicCall(CI, I->second);
2702f22ef01cSRoman Divacky       }
2703f22ef01cSRoman Divacky     }
2704f22ef01cSRoman Divacky   }
2705f22ef01cSRoman Divacky 
2706f22ef01cSRoman Divacky   return false;
2707f22ef01cSRoman Divacky }
2708f22ef01cSRoman Divacky 
2709f22ef01cSRoman Divacky bool BitcodeReader::isDematerializable(const GlobalValue *GV) const {
2710f22ef01cSRoman Divacky   const Function *F = dyn_cast<Function>(GV);
2711f22ef01cSRoman Divacky   if (!F || F->isDeclaration())
2712f22ef01cSRoman Divacky     return false;
2713f22ef01cSRoman Divacky   return DeferredFunctionInfo.count(const_cast<Function*>(F));
2714f22ef01cSRoman Divacky }
2715f22ef01cSRoman Divacky 
2716f22ef01cSRoman Divacky void BitcodeReader::Dematerialize(GlobalValue *GV) {
2717f22ef01cSRoman Divacky   Function *F = dyn_cast<Function>(GV);
2718f22ef01cSRoman Divacky   // If this function isn't dematerializable, this is a noop.
2719f22ef01cSRoman Divacky   if (!F || !isDematerializable(F))
2720f22ef01cSRoman Divacky     return;
2721f22ef01cSRoman Divacky 
2722f22ef01cSRoman Divacky   assert(DeferredFunctionInfo.count(F) && "No info to read function later?");
2723f22ef01cSRoman Divacky 
2724f22ef01cSRoman Divacky   // Just forget the function body, we can remat it later.
2725f22ef01cSRoman Divacky   F->deleteBody();
2726f22ef01cSRoman Divacky }
2727f22ef01cSRoman Divacky 
2728f22ef01cSRoman Divacky 
2729f22ef01cSRoman Divacky bool BitcodeReader::MaterializeModule(Module *M, std::string *ErrInfo) {
2730f22ef01cSRoman Divacky   assert(M == TheModule &&
2731f22ef01cSRoman Divacky          "Can only Materialize the Module this BitcodeReader is attached to.");
2732f22ef01cSRoman Divacky   // Iterate over the module, deserializing any functions that are still on
2733f22ef01cSRoman Divacky   // disk.
2734f22ef01cSRoman Divacky   for (Module::iterator F = TheModule->begin(), E = TheModule->end();
2735f22ef01cSRoman Divacky        F != E; ++F)
2736f22ef01cSRoman Divacky     if (F->isMaterializable() &&
2737f22ef01cSRoman Divacky         Materialize(F, ErrInfo))
2738f22ef01cSRoman Divacky       return true;
2739f22ef01cSRoman Divacky 
2740f22ef01cSRoman Divacky   // Upgrade any intrinsic calls that slipped through (should not happen!) and
2741f22ef01cSRoman Divacky   // delete the old functions to clean up. We can't do this unless the entire
2742f22ef01cSRoman Divacky   // module is materialized because there could always be another function body
2743f22ef01cSRoman Divacky   // with calls to the old function.
2744f22ef01cSRoman Divacky   for (std::vector<std::pair<Function*, Function*> >::iterator I =
2745f22ef01cSRoman Divacky        UpgradedIntrinsics.begin(), E = UpgradedIntrinsics.end(); I != E; ++I) {
2746f22ef01cSRoman Divacky     if (I->first != I->second) {
2747f22ef01cSRoman Divacky       for (Value::use_iterator UI = I->first->use_begin(),
2748f22ef01cSRoman Divacky            UE = I->first->use_end(); UI != UE; ) {
2749f22ef01cSRoman Divacky         if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2750f22ef01cSRoman Divacky           UpgradeIntrinsicCall(CI, I->second);
2751f22ef01cSRoman Divacky       }
2752f22ef01cSRoman Divacky       if (!I->first->use_empty())
2753f22ef01cSRoman Divacky         I->first->replaceAllUsesWith(I->second);
2754f22ef01cSRoman Divacky       I->first->eraseFromParent();
2755f22ef01cSRoman Divacky     }
2756f22ef01cSRoman Divacky   }
2757f22ef01cSRoman Divacky   std::vector<std::pair<Function*, Function*> >().swap(UpgradedIntrinsics);
2758f22ef01cSRoman Divacky 
2759f22ef01cSRoman Divacky   // Check debug info intrinsics.
2760f22ef01cSRoman Divacky   CheckDebugInfoIntrinsics(TheModule);
2761f22ef01cSRoman Divacky 
2762f22ef01cSRoman Divacky   return false;
2763f22ef01cSRoman Divacky }
2764f22ef01cSRoman Divacky 
2765f22ef01cSRoman Divacky 
2766f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
2767f22ef01cSRoman Divacky // External interface
2768f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
2769f22ef01cSRoman Divacky 
2770f22ef01cSRoman Divacky /// getLazyBitcodeModule - lazy function-at-a-time loading from a file.
2771f22ef01cSRoman Divacky ///
2772f22ef01cSRoman Divacky Module *llvm::getLazyBitcodeModule(MemoryBuffer *Buffer,
2773f22ef01cSRoman Divacky                                    LLVMContext& Context,
2774f22ef01cSRoman Divacky                                    std::string *ErrMsg) {
2775f22ef01cSRoman Divacky   Module *M = new Module(Buffer->getBufferIdentifier(), Context);
2776f22ef01cSRoman Divacky   BitcodeReader *R = new BitcodeReader(Buffer, Context);
2777f22ef01cSRoman Divacky   M->setMaterializer(R);
2778f22ef01cSRoman Divacky   if (R->ParseBitcodeInto(M)) {
2779f22ef01cSRoman Divacky     if (ErrMsg)
2780f22ef01cSRoman Divacky       *ErrMsg = R->getErrorString();
2781f22ef01cSRoman Divacky 
2782f22ef01cSRoman Divacky     delete M;  // Also deletes R.
2783f22ef01cSRoman Divacky     return 0;
2784f22ef01cSRoman Divacky   }
2785f22ef01cSRoman Divacky   // Have the BitcodeReader dtor delete 'Buffer'.
2786f22ef01cSRoman Divacky   R->setBufferOwned(true);
2787f22ef01cSRoman Divacky   return M;
2788f22ef01cSRoman Divacky }
2789f22ef01cSRoman Divacky 
2790f22ef01cSRoman Divacky /// ParseBitcodeFile - Read the specified bitcode file, returning the module.
2791f22ef01cSRoman Divacky /// If an error occurs, return null and fill in *ErrMsg if non-null.
2792f22ef01cSRoman Divacky Module *llvm::ParseBitcodeFile(MemoryBuffer *Buffer, LLVMContext& Context,
2793f22ef01cSRoman Divacky                                std::string *ErrMsg){
2794f22ef01cSRoman Divacky   Module *M = getLazyBitcodeModule(Buffer, Context, ErrMsg);
2795f22ef01cSRoman Divacky   if (!M) return 0;
2796f22ef01cSRoman Divacky 
2797f22ef01cSRoman Divacky   // Don't let the BitcodeReader dtor delete 'Buffer', regardless of whether
2798f22ef01cSRoman Divacky   // there was an error.
2799f22ef01cSRoman Divacky   static_cast<BitcodeReader*>(M->getMaterializer())->setBufferOwned(false);
2800f22ef01cSRoman Divacky 
2801f22ef01cSRoman Divacky   // Read in the entire module, and destroy the BitcodeReader.
2802f22ef01cSRoman Divacky   if (M->MaterializeAllPermanently(ErrMsg)) {
2803f22ef01cSRoman Divacky     delete M;
28042754fe60SDimitry Andric     return 0;
2805f22ef01cSRoman Divacky   }
28062754fe60SDimitry Andric 
2807f22ef01cSRoman Divacky   return M;
2808f22ef01cSRoman Divacky }
28092754fe60SDimitry Andric 
28102754fe60SDimitry Andric std::string llvm::getBitcodeTargetTriple(MemoryBuffer *Buffer,
28112754fe60SDimitry Andric                                          LLVMContext& Context,
28122754fe60SDimitry Andric                                          std::string *ErrMsg) {
28132754fe60SDimitry Andric   BitcodeReader *R = new BitcodeReader(Buffer, Context);
28142754fe60SDimitry Andric   // Don't let the BitcodeReader dtor delete 'Buffer'.
28152754fe60SDimitry Andric   R->setBufferOwned(false);
28162754fe60SDimitry Andric 
28172754fe60SDimitry Andric   std::string Triple("");
28182754fe60SDimitry Andric   if (R->ParseTriple(Triple))
28192754fe60SDimitry Andric     if (ErrMsg)
28202754fe60SDimitry Andric       *ErrMsg = R->getErrorString();
28212754fe60SDimitry Andric 
28222754fe60SDimitry Andric   delete R;
28232754fe60SDimitry Andric   return Triple;
28242754fe60SDimitry Andric }
2825