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;
34f22ef01cSRoman Divacky   std::vector<PATypeHolder>().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();
42f22ef01cSRoman Divacky }
43f22ef01cSRoman Divacky 
44f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
45f22ef01cSRoman Divacky //  Helper functions to implement forward reference resolution, etc.
46f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
47f22ef01cSRoman Divacky 
48f22ef01cSRoman Divacky /// ConvertToString - Convert a string from a record into an std::string, return
49f22ef01cSRoman Divacky /// true on failure.
50f22ef01cSRoman Divacky template<typename StrTy>
51f22ef01cSRoman Divacky static bool ConvertToString(SmallVector<uint64_t, 64> &Record, unsigned Idx,
52f22ef01cSRoman Divacky                             StrTy &Result) {
53f22ef01cSRoman Divacky   if (Idx > Record.size())
54f22ef01cSRoman Divacky     return true;
55f22ef01cSRoman Divacky 
56f22ef01cSRoman Divacky   for (unsigned i = Idx, e = Record.size(); i != e; ++i)
57f22ef01cSRoman Divacky     Result += (char)Record[i];
58f22ef01cSRoman Divacky   return false;
59f22ef01cSRoman Divacky }
60f22ef01cSRoman Divacky 
61f22ef01cSRoman Divacky static GlobalValue::LinkageTypes GetDecodedLinkage(unsigned Val) {
62f22ef01cSRoman Divacky   switch (Val) {
63f22ef01cSRoman Divacky   default: // Map unknown/new linkages to external
64f22ef01cSRoman Divacky   case 0:  return GlobalValue::ExternalLinkage;
65f22ef01cSRoman Divacky   case 1:  return GlobalValue::WeakAnyLinkage;
66f22ef01cSRoman Divacky   case 2:  return GlobalValue::AppendingLinkage;
67f22ef01cSRoman Divacky   case 3:  return GlobalValue::InternalLinkage;
68f22ef01cSRoman Divacky   case 4:  return GlobalValue::LinkOnceAnyLinkage;
69f22ef01cSRoman Divacky   case 5:  return GlobalValue::DLLImportLinkage;
70f22ef01cSRoman Divacky   case 6:  return GlobalValue::DLLExportLinkage;
71f22ef01cSRoman Divacky   case 7:  return GlobalValue::ExternalWeakLinkage;
72f22ef01cSRoman Divacky   case 8:  return GlobalValue::CommonLinkage;
73f22ef01cSRoman Divacky   case 9:  return GlobalValue::PrivateLinkage;
74f22ef01cSRoman Divacky   case 10: return GlobalValue::WeakODRLinkage;
75f22ef01cSRoman Divacky   case 11: return GlobalValue::LinkOnceODRLinkage;
76f22ef01cSRoman Divacky   case 12: return GlobalValue::AvailableExternallyLinkage;
77f22ef01cSRoman Divacky   case 13: return GlobalValue::LinkerPrivateLinkage;
78f22ef01cSRoman Divacky   }
79f22ef01cSRoman Divacky }
80f22ef01cSRoman Divacky 
81f22ef01cSRoman Divacky static GlobalValue::VisibilityTypes GetDecodedVisibility(unsigned Val) {
82f22ef01cSRoman Divacky   switch (Val) {
83f22ef01cSRoman Divacky   default: // Map unknown visibilities to default.
84f22ef01cSRoman Divacky   case 0: return GlobalValue::DefaultVisibility;
85f22ef01cSRoman Divacky   case 1: return GlobalValue::HiddenVisibility;
86f22ef01cSRoman Divacky   case 2: return GlobalValue::ProtectedVisibility;
87f22ef01cSRoman Divacky   }
88f22ef01cSRoman Divacky }
89f22ef01cSRoman Divacky 
90f22ef01cSRoman Divacky static int GetDecodedCastOpcode(unsigned Val) {
91f22ef01cSRoman Divacky   switch (Val) {
92f22ef01cSRoman Divacky   default: return -1;
93f22ef01cSRoman Divacky   case bitc::CAST_TRUNC   : return Instruction::Trunc;
94f22ef01cSRoman Divacky   case bitc::CAST_ZEXT    : return Instruction::ZExt;
95f22ef01cSRoman Divacky   case bitc::CAST_SEXT    : return Instruction::SExt;
96f22ef01cSRoman Divacky   case bitc::CAST_FPTOUI  : return Instruction::FPToUI;
97f22ef01cSRoman Divacky   case bitc::CAST_FPTOSI  : return Instruction::FPToSI;
98f22ef01cSRoman Divacky   case bitc::CAST_UITOFP  : return Instruction::UIToFP;
99f22ef01cSRoman Divacky   case bitc::CAST_SITOFP  : return Instruction::SIToFP;
100f22ef01cSRoman Divacky   case bitc::CAST_FPTRUNC : return Instruction::FPTrunc;
101f22ef01cSRoman Divacky   case bitc::CAST_FPEXT   : return Instruction::FPExt;
102f22ef01cSRoman Divacky   case bitc::CAST_PTRTOINT: return Instruction::PtrToInt;
103f22ef01cSRoman Divacky   case bitc::CAST_INTTOPTR: return Instruction::IntToPtr;
104f22ef01cSRoman Divacky   case bitc::CAST_BITCAST : return Instruction::BitCast;
105f22ef01cSRoman Divacky   }
106f22ef01cSRoman Divacky }
107f22ef01cSRoman Divacky static int GetDecodedBinaryOpcode(unsigned Val, const Type *Ty) {
108f22ef01cSRoman Divacky   switch (Val) {
109f22ef01cSRoman Divacky   default: return -1;
110f22ef01cSRoman Divacky   case bitc::BINOP_ADD:
111f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FAdd : Instruction::Add;
112f22ef01cSRoman Divacky   case bitc::BINOP_SUB:
113f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FSub : Instruction::Sub;
114f22ef01cSRoman Divacky   case bitc::BINOP_MUL:
115f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FMul : Instruction::Mul;
116f22ef01cSRoman Divacky   case bitc::BINOP_UDIV: return Instruction::UDiv;
117f22ef01cSRoman Divacky   case bitc::BINOP_SDIV:
118f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FDiv : Instruction::SDiv;
119f22ef01cSRoman Divacky   case bitc::BINOP_UREM: return Instruction::URem;
120f22ef01cSRoman Divacky   case bitc::BINOP_SREM:
121f22ef01cSRoman Divacky     return Ty->isFPOrFPVectorTy() ? Instruction::FRem : Instruction::SRem;
122f22ef01cSRoman Divacky   case bitc::BINOP_SHL:  return Instruction::Shl;
123f22ef01cSRoman Divacky   case bitc::BINOP_LSHR: return Instruction::LShr;
124f22ef01cSRoman Divacky   case bitc::BINOP_ASHR: return Instruction::AShr;
125f22ef01cSRoman Divacky   case bitc::BINOP_AND:  return Instruction::And;
126f22ef01cSRoman Divacky   case bitc::BINOP_OR:   return Instruction::Or;
127f22ef01cSRoman Divacky   case bitc::BINOP_XOR:  return Instruction::Xor;
128f22ef01cSRoman Divacky   }
129f22ef01cSRoman Divacky }
130f22ef01cSRoman Divacky 
131f22ef01cSRoman Divacky namespace llvm {
132f22ef01cSRoman Divacky namespace {
133f22ef01cSRoman Divacky   /// @brief A class for maintaining the slot number definition
134f22ef01cSRoman Divacky   /// as a placeholder for the actual definition for forward constants defs.
135f22ef01cSRoman Divacky   class ConstantPlaceHolder : public ConstantExpr {
136f22ef01cSRoman Divacky     ConstantPlaceHolder();                       // DO NOT IMPLEMENT
137f22ef01cSRoman Divacky     void operator=(const ConstantPlaceHolder &); // DO NOT IMPLEMENT
138f22ef01cSRoman Divacky   public:
139f22ef01cSRoman Divacky     // allocate space for exactly one operand
140f22ef01cSRoman Divacky     void *operator new(size_t s) {
141f22ef01cSRoman Divacky       return User::operator new(s, 1);
142f22ef01cSRoman Divacky     }
143f22ef01cSRoman Divacky     explicit ConstantPlaceHolder(const Type *Ty, LLVMContext& Context)
144f22ef01cSRoman Divacky       : ConstantExpr(Ty, Instruction::UserOp1, &Op<0>(), 1) {
145f22ef01cSRoman Divacky       Op<0>() = UndefValue::get(Type::getInt32Ty(Context));
146f22ef01cSRoman Divacky     }
147f22ef01cSRoman Divacky 
148f22ef01cSRoman Divacky     /// @brief Methods to support type inquiry through isa, cast, and dyn_cast.
149f22ef01cSRoman Divacky     static inline bool classof(const ConstantPlaceHolder *) { return true; }
150f22ef01cSRoman Divacky     static bool classof(const Value *V) {
151f22ef01cSRoman Divacky       return isa<ConstantExpr>(V) &&
152f22ef01cSRoman Divacky              cast<ConstantExpr>(V)->getOpcode() == Instruction::UserOp1;
153f22ef01cSRoman Divacky     }
154f22ef01cSRoman Divacky 
155f22ef01cSRoman Divacky 
156f22ef01cSRoman Divacky     /// Provide fast operand accessors
157f22ef01cSRoman Divacky     //DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
158f22ef01cSRoman Divacky   };
159f22ef01cSRoman Divacky }
160f22ef01cSRoman Divacky 
161f22ef01cSRoman Divacky // FIXME: can we inherit this from ConstantExpr?
162f22ef01cSRoman Divacky template <>
163f22ef01cSRoman Divacky struct OperandTraits<ConstantPlaceHolder> : public FixedNumOperandTraits<1> {
164f22ef01cSRoman Divacky };
165f22ef01cSRoman Divacky }
166f22ef01cSRoman Divacky 
167f22ef01cSRoman Divacky 
168f22ef01cSRoman Divacky void BitcodeReaderValueList::AssignValue(Value *V, unsigned Idx) {
169f22ef01cSRoman Divacky   if (Idx == size()) {
170f22ef01cSRoman Divacky     push_back(V);
171f22ef01cSRoman Divacky     return;
172f22ef01cSRoman Divacky   }
173f22ef01cSRoman Divacky 
174f22ef01cSRoman Divacky   if (Idx >= size())
175f22ef01cSRoman Divacky     resize(Idx+1);
176f22ef01cSRoman Divacky 
177f22ef01cSRoman Divacky   WeakVH &OldV = ValuePtrs[Idx];
178f22ef01cSRoman Divacky   if (OldV == 0) {
179f22ef01cSRoman Divacky     OldV = V;
180f22ef01cSRoman Divacky     return;
181f22ef01cSRoman Divacky   }
182f22ef01cSRoman Divacky 
183f22ef01cSRoman Divacky   // Handle constants and non-constants (e.g. instrs) differently for
184f22ef01cSRoman Divacky   // efficiency.
185f22ef01cSRoman Divacky   if (Constant *PHC = dyn_cast<Constant>(&*OldV)) {
186f22ef01cSRoman Divacky     ResolveConstants.push_back(std::make_pair(PHC, Idx));
187f22ef01cSRoman Divacky     OldV = V;
188f22ef01cSRoman Divacky   } else {
189f22ef01cSRoman Divacky     // If there was a forward reference to this value, replace it.
190f22ef01cSRoman Divacky     Value *PrevVal = OldV;
191f22ef01cSRoman Divacky     OldV->replaceAllUsesWith(V);
192f22ef01cSRoman Divacky     delete PrevVal;
193f22ef01cSRoman Divacky   }
194f22ef01cSRoman Divacky }
195f22ef01cSRoman Divacky 
196f22ef01cSRoman Divacky 
197f22ef01cSRoman Divacky Constant *BitcodeReaderValueList::getConstantFwdRef(unsigned Idx,
198f22ef01cSRoman Divacky                                                     const Type *Ty) {
199f22ef01cSRoman Divacky   if (Idx >= size())
200f22ef01cSRoman Divacky     resize(Idx + 1);
201f22ef01cSRoman Divacky 
202f22ef01cSRoman Divacky   if (Value *V = ValuePtrs[Idx]) {
203f22ef01cSRoman Divacky     assert(Ty == V->getType() && "Type mismatch in constant table!");
204f22ef01cSRoman Divacky     return cast<Constant>(V);
205f22ef01cSRoman Divacky   }
206f22ef01cSRoman Divacky 
207f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
208f22ef01cSRoman Divacky   Constant *C = new ConstantPlaceHolder(Ty, Context);
209f22ef01cSRoman Divacky   ValuePtrs[Idx] = C;
210f22ef01cSRoman Divacky   return C;
211f22ef01cSRoman Divacky }
212f22ef01cSRoman Divacky 
213f22ef01cSRoman Divacky Value *BitcodeReaderValueList::getValueFwdRef(unsigned Idx, const Type *Ty) {
214f22ef01cSRoman Divacky   if (Idx >= size())
215f22ef01cSRoman Divacky     resize(Idx + 1);
216f22ef01cSRoman Divacky 
217f22ef01cSRoman Divacky   if (Value *V = ValuePtrs[Idx]) {
218f22ef01cSRoman Divacky     assert((Ty == 0 || Ty == V->getType()) && "Type mismatch in value table!");
219f22ef01cSRoman Divacky     return V;
220f22ef01cSRoman Divacky   }
221f22ef01cSRoman Divacky 
222f22ef01cSRoman Divacky   // No type specified, must be invalid reference.
223f22ef01cSRoman Divacky   if (Ty == 0) return 0;
224f22ef01cSRoman Divacky 
225f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
226f22ef01cSRoman Divacky   Value *V = new Argument(Ty);
227f22ef01cSRoman Divacky   ValuePtrs[Idx] = V;
228f22ef01cSRoman Divacky   return V;
229f22ef01cSRoman Divacky }
230f22ef01cSRoman Divacky 
231f22ef01cSRoman Divacky /// ResolveConstantForwardRefs - Once all constants are read, this method bulk
232f22ef01cSRoman Divacky /// resolves any forward references.  The idea behind this is that we sometimes
233f22ef01cSRoman Divacky /// get constants (such as large arrays) which reference *many* forward ref
234f22ef01cSRoman Divacky /// constants.  Replacing each of these causes a lot of thrashing when
235f22ef01cSRoman Divacky /// building/reuniquing the constant.  Instead of doing this, we look at all the
236f22ef01cSRoman Divacky /// uses and rewrite all the place holders at once for any constant that uses
237f22ef01cSRoman Divacky /// a placeholder.
238f22ef01cSRoman Divacky void BitcodeReaderValueList::ResolveConstantForwardRefs() {
239f22ef01cSRoman Divacky   // Sort the values by-pointer so that they are efficient to look up with a
240f22ef01cSRoman Divacky   // binary search.
241f22ef01cSRoman Divacky   std::sort(ResolveConstants.begin(), ResolveConstants.end());
242f22ef01cSRoman Divacky 
243f22ef01cSRoman Divacky   SmallVector<Constant*, 64> NewOps;
244f22ef01cSRoman Divacky 
245f22ef01cSRoman Divacky   while (!ResolveConstants.empty()) {
246f22ef01cSRoman Divacky     Value *RealVal = operator[](ResolveConstants.back().second);
247f22ef01cSRoman Divacky     Constant *Placeholder = ResolveConstants.back().first;
248f22ef01cSRoman Divacky     ResolveConstants.pop_back();
249f22ef01cSRoman Divacky 
250f22ef01cSRoman Divacky     // Loop over all users of the placeholder, updating them to reference the
251f22ef01cSRoman Divacky     // new value.  If they reference more than one placeholder, update them all
252f22ef01cSRoman Divacky     // at once.
253f22ef01cSRoman Divacky     while (!Placeholder->use_empty()) {
254f22ef01cSRoman Divacky       Value::use_iterator UI = Placeholder->use_begin();
255f22ef01cSRoman Divacky 
256f22ef01cSRoman Divacky       // If the using object isn't uniqued, just update the operands.  This
257f22ef01cSRoman Divacky       // handles instructions and initializers for global variables.
258f22ef01cSRoman Divacky       if (!isa<Constant>(*UI) || isa<GlobalValue>(*UI)) {
259f22ef01cSRoman Divacky         UI.getUse().set(RealVal);
260f22ef01cSRoman Divacky         continue;
261f22ef01cSRoman Divacky       }
262f22ef01cSRoman Divacky 
263f22ef01cSRoman Divacky       // Otherwise, we have a constant that uses the placeholder.  Replace that
264f22ef01cSRoman Divacky       // constant with a new constant that has *all* placeholder uses updated.
265f22ef01cSRoman Divacky       Constant *UserC = cast<Constant>(*UI);
266f22ef01cSRoman Divacky       for (User::op_iterator I = UserC->op_begin(), E = UserC->op_end();
267f22ef01cSRoman Divacky            I != E; ++I) {
268f22ef01cSRoman Divacky         Value *NewOp;
269f22ef01cSRoman Divacky         if (!isa<ConstantPlaceHolder>(*I)) {
270f22ef01cSRoman Divacky           // Not a placeholder reference.
271f22ef01cSRoman Divacky           NewOp = *I;
272f22ef01cSRoman Divacky         } else if (*I == Placeholder) {
273f22ef01cSRoman Divacky           // Common case is that it just references this one placeholder.
274f22ef01cSRoman Divacky           NewOp = RealVal;
275f22ef01cSRoman Divacky         } else {
276f22ef01cSRoman Divacky           // Otherwise, look up the placeholder in ResolveConstants.
277f22ef01cSRoman Divacky           ResolveConstantsTy::iterator It =
278f22ef01cSRoman Divacky             std::lower_bound(ResolveConstants.begin(), ResolveConstants.end(),
279f22ef01cSRoman Divacky                              std::pair<Constant*, unsigned>(cast<Constant>(*I),
280f22ef01cSRoman Divacky                                                             0));
281f22ef01cSRoman Divacky           assert(It != ResolveConstants.end() && It->first == *I);
282f22ef01cSRoman Divacky           NewOp = operator[](It->second);
283f22ef01cSRoman Divacky         }
284f22ef01cSRoman Divacky 
285f22ef01cSRoman Divacky         NewOps.push_back(cast<Constant>(NewOp));
286f22ef01cSRoman Divacky       }
287f22ef01cSRoman Divacky 
288f22ef01cSRoman Divacky       // Make the new constant.
289f22ef01cSRoman Divacky       Constant *NewC;
290f22ef01cSRoman Divacky       if (ConstantArray *UserCA = dyn_cast<ConstantArray>(UserC)) {
291f22ef01cSRoman Divacky         NewC = ConstantArray::get(UserCA->getType(), &NewOps[0],
292f22ef01cSRoman Divacky                                         NewOps.size());
293f22ef01cSRoman Divacky       } else if (ConstantStruct *UserCS = dyn_cast<ConstantStruct>(UserC)) {
294f22ef01cSRoman Divacky         NewC = ConstantStruct::get(Context, &NewOps[0], NewOps.size(),
295f22ef01cSRoman Divacky                                          UserCS->getType()->isPacked());
296f22ef01cSRoman Divacky       } else if (ConstantUnion *UserCU = dyn_cast<ConstantUnion>(UserC)) {
297f22ef01cSRoman Divacky         NewC = ConstantUnion::get(UserCU->getType(), NewOps[0]);
298f22ef01cSRoman Divacky       } else if (isa<ConstantVector>(UserC)) {
299f22ef01cSRoman Divacky         NewC = ConstantVector::get(&NewOps[0], NewOps.size());
300f22ef01cSRoman Divacky       } else {
301f22ef01cSRoman Divacky         assert(isa<ConstantExpr>(UserC) && "Must be a ConstantExpr.");
302f22ef01cSRoman Divacky         NewC = cast<ConstantExpr>(UserC)->getWithOperands(&NewOps[0],
303f22ef01cSRoman Divacky                                                           NewOps.size());
304f22ef01cSRoman Divacky       }
305f22ef01cSRoman Divacky 
306f22ef01cSRoman Divacky       UserC->replaceAllUsesWith(NewC);
307f22ef01cSRoman Divacky       UserC->destroyConstant();
308f22ef01cSRoman Divacky       NewOps.clear();
309f22ef01cSRoman Divacky     }
310f22ef01cSRoman Divacky 
311f22ef01cSRoman Divacky     // Update all ValueHandles, they should be the only users at this point.
312f22ef01cSRoman Divacky     Placeholder->replaceAllUsesWith(RealVal);
313f22ef01cSRoman Divacky     delete Placeholder;
314f22ef01cSRoman Divacky   }
315f22ef01cSRoman Divacky }
316f22ef01cSRoman Divacky 
317f22ef01cSRoman Divacky void BitcodeReaderMDValueList::AssignValue(Value *V, unsigned Idx) {
318f22ef01cSRoman Divacky   if (Idx == size()) {
319f22ef01cSRoman Divacky     push_back(V);
320f22ef01cSRoman Divacky     return;
321f22ef01cSRoman Divacky   }
322f22ef01cSRoman Divacky 
323f22ef01cSRoman Divacky   if (Idx >= size())
324f22ef01cSRoman Divacky     resize(Idx+1);
325f22ef01cSRoman Divacky 
326f22ef01cSRoman Divacky   WeakVH &OldV = MDValuePtrs[Idx];
327f22ef01cSRoman Divacky   if (OldV == 0) {
328f22ef01cSRoman Divacky     OldV = V;
329f22ef01cSRoman Divacky     return;
330f22ef01cSRoman Divacky   }
331f22ef01cSRoman Divacky 
332f22ef01cSRoman Divacky   // If there was a forward reference to this value, replace it.
333f22ef01cSRoman Divacky   Value *PrevVal = OldV;
334f22ef01cSRoman Divacky   OldV->replaceAllUsesWith(V);
335f22ef01cSRoman Divacky   delete PrevVal;
336f22ef01cSRoman Divacky   // Deleting PrevVal sets Idx value in MDValuePtrs to null. Set new
337f22ef01cSRoman Divacky   // value for Idx.
338f22ef01cSRoman Divacky   MDValuePtrs[Idx] = V;
339f22ef01cSRoman Divacky }
340f22ef01cSRoman Divacky 
341f22ef01cSRoman Divacky Value *BitcodeReaderMDValueList::getValueFwdRef(unsigned Idx) {
342f22ef01cSRoman Divacky   if (Idx >= size())
343f22ef01cSRoman Divacky     resize(Idx + 1);
344f22ef01cSRoman Divacky 
345f22ef01cSRoman Divacky   if (Value *V = MDValuePtrs[Idx]) {
346f22ef01cSRoman Divacky     assert(V->getType()->isMetadataTy() && "Type mismatch in value table!");
347f22ef01cSRoman Divacky     return V;
348f22ef01cSRoman Divacky   }
349f22ef01cSRoman Divacky 
350f22ef01cSRoman Divacky   // Create and return a placeholder, which will later be RAUW'd.
351f22ef01cSRoman Divacky   Value *V = new Argument(Type::getMetadataTy(Context));
352f22ef01cSRoman Divacky   MDValuePtrs[Idx] = V;
353f22ef01cSRoman Divacky   return V;
354f22ef01cSRoman Divacky }
355f22ef01cSRoman Divacky 
356f22ef01cSRoman Divacky const Type *BitcodeReader::getTypeByID(unsigned ID, bool isTypeTable) {
357f22ef01cSRoman Divacky   // If the TypeID is in range, return it.
358f22ef01cSRoman Divacky   if (ID < TypeList.size())
359f22ef01cSRoman Divacky     return TypeList[ID].get();
360f22ef01cSRoman Divacky   if (!isTypeTable) return 0;
361f22ef01cSRoman Divacky 
362f22ef01cSRoman Divacky   // The type table allows forward references.  Push as many Opaque types as
363f22ef01cSRoman Divacky   // needed to get up to ID.
364f22ef01cSRoman Divacky   while (TypeList.size() <= ID)
365f22ef01cSRoman Divacky     TypeList.push_back(OpaqueType::get(Context));
366f22ef01cSRoman Divacky   return TypeList.back().get();
367f22ef01cSRoman Divacky }
368f22ef01cSRoman Divacky 
369f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
370f22ef01cSRoman Divacky //  Functions for parsing blocks from the bitcode file
371f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
372f22ef01cSRoman Divacky 
373f22ef01cSRoman Divacky bool BitcodeReader::ParseAttributeBlock() {
374f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::PARAMATTR_BLOCK_ID))
375f22ef01cSRoman Divacky     return Error("Malformed block record");
376f22ef01cSRoman Divacky 
377f22ef01cSRoman Divacky   if (!MAttributes.empty())
378f22ef01cSRoman Divacky     return Error("Multiple PARAMATTR blocks found!");
379f22ef01cSRoman Divacky 
380f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
381f22ef01cSRoman Divacky 
382f22ef01cSRoman Divacky   SmallVector<AttributeWithIndex, 8> Attrs;
383f22ef01cSRoman Divacky 
384f22ef01cSRoman Divacky   // Read all the records.
385f22ef01cSRoman Divacky   while (1) {
386f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
387f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
388f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
389f22ef01cSRoman Divacky         return Error("Error at end of PARAMATTR block");
390f22ef01cSRoman Divacky       return false;
391f22ef01cSRoman Divacky     }
392f22ef01cSRoman Divacky 
393f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
394f22ef01cSRoman Divacky       // No known subblocks, always skip them.
395f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
396f22ef01cSRoman Divacky       if (Stream.SkipBlock())
397f22ef01cSRoman Divacky         return Error("Malformed block record");
398f22ef01cSRoman Divacky       continue;
399f22ef01cSRoman Divacky     }
400f22ef01cSRoman Divacky 
401f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
402f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
403f22ef01cSRoman Divacky       continue;
404f22ef01cSRoman Divacky     }
405f22ef01cSRoman Divacky 
406f22ef01cSRoman Divacky     // Read a record.
407f22ef01cSRoman Divacky     Record.clear();
408f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
409f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
410f22ef01cSRoman Divacky       break;
411f22ef01cSRoman Divacky     case bitc::PARAMATTR_CODE_ENTRY: { // ENTRY: [paramidx0, attr0, ...]
412f22ef01cSRoman Divacky       if (Record.size() & 1)
413f22ef01cSRoman Divacky         return Error("Invalid ENTRY record");
414f22ef01cSRoman Divacky 
415f22ef01cSRoman Divacky       // FIXME : Remove this autoupgrade code in LLVM 3.0.
416f22ef01cSRoman Divacky       // If Function attributes are using index 0 then transfer them
417f22ef01cSRoman Divacky       // to index ~0. Index 0 is used for return value attributes but used to be
418f22ef01cSRoman Divacky       // used for function attributes.
419f22ef01cSRoman Divacky       Attributes RetAttribute = Attribute::None;
420f22ef01cSRoman Divacky       Attributes FnAttribute = Attribute::None;
421f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
422f22ef01cSRoman Divacky         // FIXME: remove in LLVM 3.0
423f22ef01cSRoman Divacky         // The alignment is stored as a 16-bit raw value from bits 31--16.
424f22ef01cSRoman Divacky         // We shift the bits above 31 down by 11 bits.
425f22ef01cSRoman Divacky 
426f22ef01cSRoman Divacky         unsigned Alignment = (Record[i+1] & (0xffffull << 16)) >> 16;
427f22ef01cSRoman Divacky         if (Alignment && !isPowerOf2_32(Alignment))
428f22ef01cSRoman Divacky           return Error("Alignment is not a power of two.");
429f22ef01cSRoman Divacky 
430f22ef01cSRoman Divacky         Attributes ReconstitutedAttr = Record[i+1] & 0xffff;
431f22ef01cSRoman Divacky         if (Alignment)
432f22ef01cSRoman Divacky           ReconstitutedAttr |= Attribute::constructAlignmentFromInt(Alignment);
433f22ef01cSRoman Divacky         ReconstitutedAttr |= (Record[i+1] & (0xffffull << 32)) >> 11;
434f22ef01cSRoman Divacky         Record[i+1] = ReconstitutedAttr;
435f22ef01cSRoman Divacky 
436f22ef01cSRoman Divacky         if (Record[i] == 0)
437f22ef01cSRoman Divacky           RetAttribute = Record[i+1];
438f22ef01cSRoman Divacky         else if (Record[i] == ~0U)
439f22ef01cSRoman Divacky           FnAttribute = Record[i+1];
440f22ef01cSRoman Divacky       }
441f22ef01cSRoman Divacky 
442f22ef01cSRoman Divacky       unsigned OldRetAttrs = (Attribute::NoUnwind|Attribute::NoReturn|
443f22ef01cSRoman Divacky                               Attribute::ReadOnly|Attribute::ReadNone);
444f22ef01cSRoman Divacky 
445f22ef01cSRoman Divacky       if (FnAttribute == Attribute::None && RetAttribute != Attribute::None &&
446f22ef01cSRoman Divacky           (RetAttribute & OldRetAttrs) != 0) {
447f22ef01cSRoman Divacky         if (FnAttribute == Attribute::None) { // add a slot so they get added.
448f22ef01cSRoman Divacky           Record.push_back(~0U);
449f22ef01cSRoman Divacky           Record.push_back(0);
450f22ef01cSRoman Divacky         }
451f22ef01cSRoman Divacky 
452f22ef01cSRoman Divacky         FnAttribute  |= RetAttribute & OldRetAttrs;
453f22ef01cSRoman Divacky         RetAttribute &= ~OldRetAttrs;
454f22ef01cSRoman Divacky       }
455f22ef01cSRoman Divacky 
456f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
457f22ef01cSRoman Divacky         if (Record[i] == 0) {
458f22ef01cSRoman Divacky           if (RetAttribute != Attribute::None)
459f22ef01cSRoman Divacky             Attrs.push_back(AttributeWithIndex::get(0, RetAttribute));
460f22ef01cSRoman Divacky         } else if (Record[i] == ~0U) {
461f22ef01cSRoman Divacky           if (FnAttribute != Attribute::None)
462f22ef01cSRoman Divacky             Attrs.push_back(AttributeWithIndex::get(~0U, FnAttribute));
463f22ef01cSRoman Divacky         } else if (Record[i+1] != Attribute::None)
464f22ef01cSRoman Divacky           Attrs.push_back(AttributeWithIndex::get(Record[i], Record[i+1]));
465f22ef01cSRoman Divacky       }
466f22ef01cSRoman Divacky 
467f22ef01cSRoman Divacky       MAttributes.push_back(AttrListPtr::get(Attrs.begin(), Attrs.end()));
468f22ef01cSRoman Divacky       Attrs.clear();
469f22ef01cSRoman Divacky       break;
470f22ef01cSRoman Divacky     }
471f22ef01cSRoman Divacky     }
472f22ef01cSRoman Divacky   }
473f22ef01cSRoman Divacky }
474f22ef01cSRoman Divacky 
475f22ef01cSRoman Divacky 
476f22ef01cSRoman Divacky bool BitcodeReader::ParseTypeTable() {
477f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID))
478f22ef01cSRoman Divacky     return Error("Malformed block record");
479f22ef01cSRoman Divacky 
480f22ef01cSRoman Divacky   if (!TypeList.empty())
481f22ef01cSRoman Divacky     return Error("Multiple TYPE_BLOCKs found!");
482f22ef01cSRoman Divacky 
483f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
484f22ef01cSRoman Divacky   unsigned NumRecords = 0;
485f22ef01cSRoman Divacky 
486f22ef01cSRoman Divacky   // Read all the records for this type table.
487f22ef01cSRoman Divacky   while (1) {
488f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
489f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
490f22ef01cSRoman Divacky       if (NumRecords != TypeList.size())
491f22ef01cSRoman Divacky         return Error("Invalid type forward reference in TYPE_BLOCK");
492f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
493f22ef01cSRoman Divacky         return Error("Error at end of type table block");
494f22ef01cSRoman Divacky       return false;
495f22ef01cSRoman Divacky     }
496f22ef01cSRoman Divacky 
497f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
498f22ef01cSRoman Divacky       // No known subblocks, always skip them.
499f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
500f22ef01cSRoman Divacky       if (Stream.SkipBlock())
501f22ef01cSRoman Divacky         return Error("Malformed block record");
502f22ef01cSRoman Divacky       continue;
503f22ef01cSRoman Divacky     }
504f22ef01cSRoman Divacky 
505f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
506f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
507f22ef01cSRoman Divacky       continue;
508f22ef01cSRoman Divacky     }
509f22ef01cSRoman Divacky 
510f22ef01cSRoman Divacky     // Read a record.
511f22ef01cSRoman Divacky     Record.clear();
512f22ef01cSRoman Divacky     const Type *ResultTy = 0;
513f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
514f22ef01cSRoman Divacky     default:  // Default behavior: unknown type.
515f22ef01cSRoman Divacky       ResultTy = 0;
516f22ef01cSRoman Divacky       break;
517f22ef01cSRoman Divacky     case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
518f22ef01cSRoman Divacky       // TYPE_CODE_NUMENTRY contains a count of the number of types in the
519f22ef01cSRoman Divacky       // type list.  This allows us to reserve space.
520f22ef01cSRoman Divacky       if (Record.size() < 1)
521f22ef01cSRoman Divacky         return Error("Invalid TYPE_CODE_NUMENTRY record");
522f22ef01cSRoman Divacky       TypeList.reserve(Record[0]);
523f22ef01cSRoman Divacky       continue;
524f22ef01cSRoman Divacky     case bitc::TYPE_CODE_VOID:      // VOID
525f22ef01cSRoman Divacky       ResultTy = Type::getVoidTy(Context);
526f22ef01cSRoman Divacky       break;
527f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FLOAT:     // FLOAT
528f22ef01cSRoman Divacky       ResultTy = Type::getFloatTy(Context);
529f22ef01cSRoman Divacky       break;
530f22ef01cSRoman Divacky     case bitc::TYPE_CODE_DOUBLE:    // DOUBLE
531f22ef01cSRoman Divacky       ResultTy = Type::getDoubleTy(Context);
532f22ef01cSRoman Divacky       break;
533f22ef01cSRoman Divacky     case bitc::TYPE_CODE_X86_FP80:  // X86_FP80
534f22ef01cSRoman Divacky       ResultTy = Type::getX86_FP80Ty(Context);
535f22ef01cSRoman Divacky       break;
536f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FP128:     // FP128
537f22ef01cSRoman Divacky       ResultTy = Type::getFP128Ty(Context);
538f22ef01cSRoman Divacky       break;
539f22ef01cSRoman Divacky     case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
540f22ef01cSRoman Divacky       ResultTy = Type::getPPC_FP128Ty(Context);
541f22ef01cSRoman Divacky       break;
542f22ef01cSRoman Divacky     case bitc::TYPE_CODE_LABEL:     // LABEL
543f22ef01cSRoman Divacky       ResultTy = Type::getLabelTy(Context);
544f22ef01cSRoman Divacky       break;
545f22ef01cSRoman Divacky     case bitc::TYPE_CODE_OPAQUE:    // OPAQUE
546f22ef01cSRoman Divacky       ResultTy = 0;
547f22ef01cSRoman Divacky       break;
548f22ef01cSRoman Divacky     case bitc::TYPE_CODE_METADATA:  // METADATA
549f22ef01cSRoman Divacky       ResultTy = Type::getMetadataTy(Context);
550f22ef01cSRoman Divacky       break;
551f22ef01cSRoman Divacky     case bitc::TYPE_CODE_INTEGER:   // INTEGER: [width]
552f22ef01cSRoman Divacky       if (Record.size() < 1)
553f22ef01cSRoman Divacky         return Error("Invalid Integer type record");
554f22ef01cSRoman Divacky 
555f22ef01cSRoman Divacky       ResultTy = IntegerType::get(Context, Record[0]);
556f22ef01cSRoman Divacky       break;
557f22ef01cSRoman Divacky     case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
558f22ef01cSRoman Divacky                                     //          [pointee type, address space]
559f22ef01cSRoman Divacky       if (Record.size() < 1)
560f22ef01cSRoman Divacky         return Error("Invalid POINTER type record");
561f22ef01cSRoman Divacky       unsigned AddressSpace = 0;
562f22ef01cSRoman Divacky       if (Record.size() == 2)
563f22ef01cSRoman Divacky         AddressSpace = Record[1];
564f22ef01cSRoman Divacky       ResultTy = PointerType::get(getTypeByID(Record[0], true),
565f22ef01cSRoman Divacky                                         AddressSpace);
566f22ef01cSRoman Divacky       break;
567f22ef01cSRoman Divacky     }
568f22ef01cSRoman Divacky     case bitc::TYPE_CODE_FUNCTION: {
569f22ef01cSRoman Divacky       // FIXME: attrid is dead, remove it in LLVM 3.0
570f22ef01cSRoman Divacky       // FUNCTION: [vararg, attrid, retty, paramty x N]
571f22ef01cSRoman Divacky       if (Record.size() < 3)
572f22ef01cSRoman Divacky         return Error("Invalid FUNCTION type record");
573f22ef01cSRoman Divacky       std::vector<const Type*> ArgTys;
574f22ef01cSRoman Divacky       for (unsigned i = 3, e = Record.size(); i != e; ++i)
575f22ef01cSRoman Divacky         ArgTys.push_back(getTypeByID(Record[i], true));
576f22ef01cSRoman Divacky 
577f22ef01cSRoman Divacky       ResultTy = FunctionType::get(getTypeByID(Record[2], true), ArgTys,
578f22ef01cSRoman Divacky                                    Record[0]);
579f22ef01cSRoman Divacky       break;
580f22ef01cSRoman Divacky     }
581f22ef01cSRoman Divacky     case bitc::TYPE_CODE_STRUCT: {  // STRUCT: [ispacked, eltty x N]
582f22ef01cSRoman Divacky       if (Record.size() < 1)
583f22ef01cSRoman Divacky         return Error("Invalid STRUCT type record");
584f22ef01cSRoman Divacky       std::vector<const Type*> EltTys;
585f22ef01cSRoman Divacky       for (unsigned i = 1, e = Record.size(); i != e; ++i)
586f22ef01cSRoman Divacky         EltTys.push_back(getTypeByID(Record[i], true));
587f22ef01cSRoman Divacky       ResultTy = StructType::get(Context, EltTys, Record[0]);
588f22ef01cSRoman Divacky       break;
589f22ef01cSRoman Divacky     }
590f22ef01cSRoman Divacky     case bitc::TYPE_CODE_UNION: {  // UNION: [eltty x N]
591f22ef01cSRoman Divacky       SmallVector<const Type*, 8> EltTys;
592f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; ++i)
593f22ef01cSRoman Divacky         EltTys.push_back(getTypeByID(Record[i], true));
594f22ef01cSRoman Divacky       ResultTy = UnionType::get(&EltTys[0], EltTys.size());
595f22ef01cSRoman Divacky       break;
596f22ef01cSRoman Divacky     }
597f22ef01cSRoman Divacky     case bitc::TYPE_CODE_ARRAY:     // ARRAY: [numelts, eltty]
598f22ef01cSRoman Divacky       if (Record.size() < 2)
599f22ef01cSRoman Divacky         return Error("Invalid ARRAY type record");
600f22ef01cSRoman Divacky       ResultTy = ArrayType::get(getTypeByID(Record[1], true), Record[0]);
601f22ef01cSRoman Divacky       break;
602f22ef01cSRoman Divacky     case bitc::TYPE_CODE_VECTOR:    // VECTOR: [numelts, eltty]
603f22ef01cSRoman Divacky       if (Record.size() < 2)
604f22ef01cSRoman Divacky         return Error("Invalid VECTOR type record");
605f22ef01cSRoman Divacky       ResultTy = VectorType::get(getTypeByID(Record[1], true), Record[0]);
606f22ef01cSRoman Divacky       break;
607f22ef01cSRoman Divacky     }
608f22ef01cSRoman Divacky 
609f22ef01cSRoman Divacky     if (NumRecords == TypeList.size()) {
610f22ef01cSRoman Divacky       // If this is a new type slot, just append it.
611f22ef01cSRoman Divacky       TypeList.push_back(ResultTy ? ResultTy : OpaqueType::get(Context));
612f22ef01cSRoman Divacky       ++NumRecords;
613f22ef01cSRoman Divacky     } else if (ResultTy == 0) {
614f22ef01cSRoman Divacky       // Otherwise, this was forward referenced, so an opaque type was created,
615f22ef01cSRoman Divacky       // but the result type is actually just an opaque.  Leave the one we
616f22ef01cSRoman Divacky       // created previously.
617f22ef01cSRoman Divacky       ++NumRecords;
618f22ef01cSRoman Divacky     } else {
619f22ef01cSRoman Divacky       // Otherwise, this was forward referenced, so an opaque type was created.
620f22ef01cSRoman Divacky       // Resolve the opaque type to the real type now.
621f22ef01cSRoman Divacky       assert(NumRecords < TypeList.size() && "Typelist imbalance");
622f22ef01cSRoman Divacky       const OpaqueType *OldTy = cast<OpaqueType>(TypeList[NumRecords++].get());
623f22ef01cSRoman Divacky 
624f22ef01cSRoman Divacky       // Don't directly push the new type on the Tab. Instead we want to replace
625f22ef01cSRoman Divacky       // the opaque type we previously inserted with the new concrete value. The
626f22ef01cSRoman Divacky       // refinement from the abstract (opaque) type to the new type causes all
627f22ef01cSRoman Divacky       // uses of the abstract type to use the concrete type (NewTy). This will
628f22ef01cSRoman Divacky       // also cause the opaque type to be deleted.
629f22ef01cSRoman Divacky       const_cast<OpaqueType*>(OldTy)->refineAbstractTypeTo(ResultTy);
630f22ef01cSRoman Divacky 
631f22ef01cSRoman Divacky       // This should have replaced the old opaque type with the new type in the
632f22ef01cSRoman Divacky       // value table... or with a preexisting type that was already in the
633f22ef01cSRoman Divacky       // system.  Let's just make sure it did.
634f22ef01cSRoman Divacky       assert(TypeList[NumRecords-1].get() != OldTy &&
635f22ef01cSRoman Divacky              "refineAbstractType didn't work!");
636f22ef01cSRoman Divacky     }
637f22ef01cSRoman Divacky   }
638f22ef01cSRoman Divacky }
639f22ef01cSRoman Divacky 
640f22ef01cSRoman Divacky 
641f22ef01cSRoman Divacky bool BitcodeReader::ParseTypeSymbolTable() {
642f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::TYPE_SYMTAB_BLOCK_ID))
643f22ef01cSRoman Divacky     return Error("Malformed block record");
644f22ef01cSRoman Divacky 
645f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
646f22ef01cSRoman Divacky 
647f22ef01cSRoman Divacky   // Read all the records for this type table.
648f22ef01cSRoman Divacky   std::string TypeName;
649f22ef01cSRoman Divacky   while (1) {
650f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
651f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
652f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
653f22ef01cSRoman Divacky         return Error("Error at end of type symbol table block");
654f22ef01cSRoman Divacky       return false;
655f22ef01cSRoman Divacky     }
656f22ef01cSRoman Divacky 
657f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
658f22ef01cSRoman Divacky       // No known subblocks, always skip them.
659f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
660f22ef01cSRoman Divacky       if (Stream.SkipBlock())
661f22ef01cSRoman Divacky         return Error("Malformed block record");
662f22ef01cSRoman Divacky       continue;
663f22ef01cSRoman Divacky     }
664f22ef01cSRoman Divacky 
665f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
666f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
667f22ef01cSRoman Divacky       continue;
668f22ef01cSRoman Divacky     }
669f22ef01cSRoman Divacky 
670f22ef01cSRoman Divacky     // Read a record.
671f22ef01cSRoman Divacky     Record.clear();
672f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
673f22ef01cSRoman Divacky     default:  // Default behavior: unknown type.
674f22ef01cSRoman Divacky       break;
675f22ef01cSRoman Divacky     case bitc::TST_CODE_ENTRY:    // TST_ENTRY: [typeid, namechar x N]
676f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, TypeName))
677f22ef01cSRoman Divacky         return Error("Invalid TST_ENTRY record");
678f22ef01cSRoman Divacky       unsigned TypeID = Record[0];
679f22ef01cSRoman Divacky       if (TypeID >= TypeList.size())
680f22ef01cSRoman Divacky         return Error("Invalid Type ID in TST_ENTRY record");
681f22ef01cSRoman Divacky 
682f22ef01cSRoman Divacky       TheModule->addTypeName(TypeName, TypeList[TypeID].get());
683f22ef01cSRoman Divacky       TypeName.clear();
684f22ef01cSRoman Divacky       break;
685f22ef01cSRoman Divacky     }
686f22ef01cSRoman Divacky   }
687f22ef01cSRoman Divacky }
688f22ef01cSRoman Divacky 
689f22ef01cSRoman Divacky bool BitcodeReader::ParseValueSymbolTable() {
690f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::VALUE_SYMTAB_BLOCK_ID))
691f22ef01cSRoman Divacky     return Error("Malformed block record");
692f22ef01cSRoman Divacky 
693f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
694f22ef01cSRoman Divacky 
695f22ef01cSRoman Divacky   // Read all the records for this value table.
696f22ef01cSRoman Divacky   SmallString<128> ValueName;
697f22ef01cSRoman Divacky   while (1) {
698f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
699f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
700f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
701f22ef01cSRoman Divacky         return Error("Error at end of value symbol table block");
702f22ef01cSRoman Divacky       return false;
703f22ef01cSRoman Divacky     }
704f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
705f22ef01cSRoman Divacky       // No known subblocks, always skip them.
706f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
707f22ef01cSRoman Divacky       if (Stream.SkipBlock())
708f22ef01cSRoman Divacky         return Error("Malformed block record");
709f22ef01cSRoman Divacky       continue;
710f22ef01cSRoman Divacky     }
711f22ef01cSRoman Divacky 
712f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
713f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
714f22ef01cSRoman Divacky       continue;
715f22ef01cSRoman Divacky     }
716f22ef01cSRoman Divacky 
717f22ef01cSRoman Divacky     // Read a record.
718f22ef01cSRoman Divacky     Record.clear();
719f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
720f22ef01cSRoman Divacky     default:  // Default behavior: unknown type.
721f22ef01cSRoman Divacky       break;
722f22ef01cSRoman Divacky     case bitc::VST_CODE_ENTRY: {  // VST_ENTRY: [valueid, namechar x N]
723f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, ValueName))
724f22ef01cSRoman Divacky         return Error("Invalid VST_ENTRY record");
725f22ef01cSRoman Divacky       unsigned ValueID = Record[0];
726f22ef01cSRoman Divacky       if (ValueID >= ValueList.size())
727f22ef01cSRoman Divacky         return Error("Invalid Value ID in VST_ENTRY record");
728f22ef01cSRoman Divacky       Value *V = ValueList[ValueID];
729f22ef01cSRoman Divacky 
730f22ef01cSRoman Divacky       V->setName(StringRef(ValueName.data(), ValueName.size()));
731f22ef01cSRoman Divacky       ValueName.clear();
732f22ef01cSRoman Divacky       break;
733f22ef01cSRoman Divacky     }
734f22ef01cSRoman Divacky     case bitc::VST_CODE_BBENTRY: {
735f22ef01cSRoman Divacky       if (ConvertToString(Record, 1, ValueName))
736f22ef01cSRoman Divacky         return Error("Invalid VST_BBENTRY record");
737f22ef01cSRoman Divacky       BasicBlock *BB = getBasicBlock(Record[0]);
738f22ef01cSRoman Divacky       if (BB == 0)
739f22ef01cSRoman Divacky         return Error("Invalid BB ID in VST_BBENTRY record");
740f22ef01cSRoman Divacky 
741f22ef01cSRoman Divacky       BB->setName(StringRef(ValueName.data(), ValueName.size()));
742f22ef01cSRoman Divacky       ValueName.clear();
743f22ef01cSRoman Divacky       break;
744f22ef01cSRoman Divacky     }
745f22ef01cSRoman Divacky     }
746f22ef01cSRoman Divacky   }
747f22ef01cSRoman Divacky }
748f22ef01cSRoman Divacky 
749f22ef01cSRoman Divacky bool BitcodeReader::ParseMetadata() {
750f22ef01cSRoman Divacky   unsigned NextMDValueNo = MDValueList.size();
751f22ef01cSRoman Divacky 
752f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
753f22ef01cSRoman Divacky     return Error("Malformed block record");
754f22ef01cSRoman Divacky 
755f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
756f22ef01cSRoman Divacky 
757f22ef01cSRoman Divacky   // Read all the records.
758f22ef01cSRoman Divacky   while (1) {
759f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
760f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
761f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
762f22ef01cSRoman Divacky         return Error("Error at end of PARAMATTR block");
763f22ef01cSRoman Divacky       return false;
764f22ef01cSRoman Divacky     }
765f22ef01cSRoman Divacky 
766f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
767f22ef01cSRoman Divacky       // No known subblocks, always skip them.
768f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
769f22ef01cSRoman Divacky       if (Stream.SkipBlock())
770f22ef01cSRoman Divacky         return Error("Malformed block record");
771f22ef01cSRoman Divacky       continue;
772f22ef01cSRoman Divacky     }
773f22ef01cSRoman Divacky 
774f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
775f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
776f22ef01cSRoman Divacky       continue;
777f22ef01cSRoman Divacky     }
778f22ef01cSRoman Divacky 
779f22ef01cSRoman Divacky     bool IsFunctionLocal = false;
780f22ef01cSRoman Divacky     // Read a record.
781f22ef01cSRoman Divacky     Record.clear();
782f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
783f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
784f22ef01cSRoman Divacky       break;
785f22ef01cSRoman Divacky     case bitc::METADATA_NAME: {
786f22ef01cSRoman Divacky       // Read named of the named metadata.
787f22ef01cSRoman Divacky       unsigned NameLength = Record.size();
788f22ef01cSRoman Divacky       SmallString<8> Name;
789f22ef01cSRoman Divacky       Name.resize(NameLength);
790f22ef01cSRoman Divacky       for (unsigned i = 0; i != NameLength; ++i)
791f22ef01cSRoman Divacky         Name[i] = Record[i];
792f22ef01cSRoman Divacky       Record.clear();
793f22ef01cSRoman Divacky       Code = Stream.ReadCode();
794f22ef01cSRoman Divacky 
795f22ef01cSRoman Divacky       // METADATA_NAME is always followed by METADATA_NAMED_NODE.
796f22ef01cSRoman Divacky       if (Stream.ReadRecord(Code, Record) != bitc::METADATA_NAMED_NODE)
797f22ef01cSRoman Divacky         assert ( 0 && "Inavlid Named Metadata record");
798f22ef01cSRoman Divacky 
799f22ef01cSRoman Divacky       // Read named metadata elements.
800f22ef01cSRoman Divacky       unsigned Size = Record.size();
801f22ef01cSRoman Divacky       SmallVector<MDNode *, 8> Elts;
802f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; ++i) {
803f22ef01cSRoman Divacky         if (Record[i] == ~0U) {
804f22ef01cSRoman Divacky           Elts.push_back(NULL);
805f22ef01cSRoman Divacky           continue;
806f22ef01cSRoman Divacky         }
807f22ef01cSRoman Divacky         MDNode *MD = dyn_cast<MDNode>(MDValueList.getValueFwdRef(Record[i]));
808f22ef01cSRoman Divacky         if (MD == 0)
809f22ef01cSRoman Divacky           return Error("Malformed metadata record");
810f22ef01cSRoman Divacky         Elts.push_back(MD);
811f22ef01cSRoman Divacky       }
812f22ef01cSRoman Divacky       Value *V = NamedMDNode::Create(Context, Name.str(), Elts.data(),
813f22ef01cSRoman Divacky                                      Elts.size(), TheModule);
814f22ef01cSRoman Divacky       MDValueList.AssignValue(V, NextMDValueNo++);
815f22ef01cSRoman Divacky       break;
816f22ef01cSRoman Divacky     }
817f22ef01cSRoman Divacky     case bitc::METADATA_FN_NODE:
818f22ef01cSRoman Divacky       IsFunctionLocal = true;
819f22ef01cSRoman Divacky       // fall-through
820f22ef01cSRoman Divacky     case bitc::METADATA_NODE: {
821f22ef01cSRoman Divacky       if (Record.empty() || Record.size() % 2 == 1)
822f22ef01cSRoman Divacky         return Error("Invalid METADATA_NODE record");
823f22ef01cSRoman Divacky 
824f22ef01cSRoman Divacky       unsigned Size = Record.size();
825f22ef01cSRoman Divacky       SmallVector<Value*, 8> Elts;
826f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; i += 2) {
827f22ef01cSRoman Divacky         const Type *Ty = getTypeByID(Record[i], false);
828f22ef01cSRoman Divacky         if (Ty->isMetadataTy())
829f22ef01cSRoman Divacky           Elts.push_back(MDValueList.getValueFwdRef(Record[i+1]));
830f22ef01cSRoman Divacky         else if (!Ty->isVoidTy())
831f22ef01cSRoman Divacky           Elts.push_back(ValueList.getValueFwdRef(Record[i+1], Ty));
832f22ef01cSRoman Divacky         else
833f22ef01cSRoman Divacky           Elts.push_back(NULL);
834f22ef01cSRoman Divacky       }
835f22ef01cSRoman Divacky       Value *V = MDNode::getWhenValsUnresolved(Context, &Elts[0], Elts.size(),
836f22ef01cSRoman Divacky                                                IsFunctionLocal);
837f22ef01cSRoman Divacky       IsFunctionLocal = false;
838f22ef01cSRoman Divacky       MDValueList.AssignValue(V, NextMDValueNo++);
839f22ef01cSRoman Divacky       break;
840f22ef01cSRoman Divacky     }
841f22ef01cSRoman Divacky     case bitc::METADATA_STRING: {
842f22ef01cSRoman Divacky       unsigned MDStringLength = Record.size();
843f22ef01cSRoman Divacky       SmallString<8> String;
844f22ef01cSRoman Divacky       String.resize(MDStringLength);
845f22ef01cSRoman Divacky       for (unsigned i = 0; i != MDStringLength; ++i)
846f22ef01cSRoman Divacky         String[i] = Record[i];
847f22ef01cSRoman Divacky       Value *V = MDString::get(Context,
848f22ef01cSRoman Divacky                                StringRef(String.data(), String.size()));
849f22ef01cSRoman Divacky       MDValueList.AssignValue(V, NextMDValueNo++);
850f22ef01cSRoman Divacky       break;
851f22ef01cSRoman Divacky     }
852f22ef01cSRoman Divacky     case bitc::METADATA_KIND: {
853f22ef01cSRoman Divacky       unsigned RecordLength = Record.size();
854f22ef01cSRoman Divacky       if (Record.empty() || RecordLength < 2)
855f22ef01cSRoman Divacky         return Error("Invalid METADATA_KIND record");
856f22ef01cSRoman Divacky       SmallString<8> Name;
857f22ef01cSRoman Divacky       Name.resize(RecordLength-1);
858f22ef01cSRoman Divacky       unsigned Kind = Record[0];
859f22ef01cSRoman Divacky       (void) Kind;
860f22ef01cSRoman Divacky       for (unsigned i = 1; i != RecordLength; ++i)
861f22ef01cSRoman Divacky         Name[i-1] = Record[i];
862f22ef01cSRoman Divacky 
863f22ef01cSRoman Divacky       unsigned NewKind = TheModule->getMDKindID(Name.str());
864f22ef01cSRoman Divacky       assert(Kind == NewKind &&
865f22ef01cSRoman Divacky              "FIXME: Unable to handle custom metadata mismatch!");(void)NewKind;
866f22ef01cSRoman Divacky       break;
867f22ef01cSRoman Divacky     }
868f22ef01cSRoman Divacky     }
869f22ef01cSRoman Divacky   }
870f22ef01cSRoman Divacky }
871f22ef01cSRoman Divacky 
872f22ef01cSRoman Divacky /// DecodeSignRotatedValue - Decode a signed value stored with the sign bit in
873f22ef01cSRoman Divacky /// the LSB for dense VBR encoding.
874f22ef01cSRoman Divacky static uint64_t DecodeSignRotatedValue(uint64_t V) {
875f22ef01cSRoman Divacky   if ((V & 1) == 0)
876f22ef01cSRoman Divacky     return V >> 1;
877f22ef01cSRoman Divacky   if (V != 1)
878f22ef01cSRoman Divacky     return -(V >> 1);
879f22ef01cSRoman Divacky   // There is no such thing as -0 with integers.  "-0" really means MININT.
880f22ef01cSRoman Divacky   return 1ULL << 63;
881f22ef01cSRoman Divacky }
882f22ef01cSRoman Divacky 
883f22ef01cSRoman Divacky /// ResolveGlobalAndAliasInits - Resolve all of the initializers for global
884f22ef01cSRoman Divacky /// values and aliases that we can.
885f22ef01cSRoman Divacky bool BitcodeReader::ResolveGlobalAndAliasInits() {
886f22ef01cSRoman Divacky   std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInitWorklist;
887f22ef01cSRoman Divacky   std::vector<std::pair<GlobalAlias*, unsigned> > AliasInitWorklist;
888f22ef01cSRoman Divacky 
889f22ef01cSRoman Divacky   GlobalInitWorklist.swap(GlobalInits);
890f22ef01cSRoman Divacky   AliasInitWorklist.swap(AliasInits);
891f22ef01cSRoman Divacky 
892f22ef01cSRoman Divacky   while (!GlobalInitWorklist.empty()) {
893f22ef01cSRoman Divacky     unsigned ValID = GlobalInitWorklist.back().second;
894f22ef01cSRoman Divacky     if (ValID >= ValueList.size()) {
895f22ef01cSRoman Divacky       // Not ready to resolve this yet, it requires something later in the file.
896f22ef01cSRoman Divacky       GlobalInits.push_back(GlobalInitWorklist.back());
897f22ef01cSRoman Divacky     } else {
898f22ef01cSRoman Divacky       if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
899f22ef01cSRoman Divacky         GlobalInitWorklist.back().first->setInitializer(C);
900f22ef01cSRoman Divacky       else
901f22ef01cSRoman Divacky         return Error("Global variable initializer is not a constant!");
902f22ef01cSRoman Divacky     }
903f22ef01cSRoman Divacky     GlobalInitWorklist.pop_back();
904f22ef01cSRoman Divacky   }
905f22ef01cSRoman Divacky 
906f22ef01cSRoman Divacky   while (!AliasInitWorklist.empty()) {
907f22ef01cSRoman Divacky     unsigned ValID = AliasInitWorklist.back().second;
908f22ef01cSRoman Divacky     if (ValID >= ValueList.size()) {
909f22ef01cSRoman Divacky       AliasInits.push_back(AliasInitWorklist.back());
910f22ef01cSRoman Divacky     } else {
911f22ef01cSRoman Divacky       if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
912f22ef01cSRoman Divacky         AliasInitWorklist.back().first->setAliasee(C);
913f22ef01cSRoman Divacky       else
914f22ef01cSRoman Divacky         return Error("Alias initializer is not a constant!");
915f22ef01cSRoman Divacky     }
916f22ef01cSRoman Divacky     AliasInitWorklist.pop_back();
917f22ef01cSRoman Divacky   }
918f22ef01cSRoman Divacky   return false;
919f22ef01cSRoman Divacky }
920f22ef01cSRoman Divacky 
921f22ef01cSRoman Divacky bool BitcodeReader::ParseConstants() {
922f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::CONSTANTS_BLOCK_ID))
923f22ef01cSRoman Divacky     return Error("Malformed block record");
924f22ef01cSRoman Divacky 
925f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
926f22ef01cSRoman Divacky 
927f22ef01cSRoman Divacky   // Read all the records for this value table.
928f22ef01cSRoman Divacky   const Type *CurTy = Type::getInt32Ty(Context);
929f22ef01cSRoman Divacky   unsigned NextCstNo = ValueList.size();
930f22ef01cSRoman Divacky   while (1) {
931f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
932f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK)
933f22ef01cSRoman Divacky       break;
934f22ef01cSRoman Divacky 
935f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
936f22ef01cSRoman Divacky       // No known subblocks, always skip them.
937f22ef01cSRoman Divacky       Stream.ReadSubBlockID();
938f22ef01cSRoman Divacky       if (Stream.SkipBlock())
939f22ef01cSRoman Divacky         return Error("Malformed block record");
940f22ef01cSRoman Divacky       continue;
941f22ef01cSRoman Divacky     }
942f22ef01cSRoman Divacky 
943f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
944f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
945f22ef01cSRoman Divacky       continue;
946f22ef01cSRoman Divacky     }
947f22ef01cSRoman Divacky 
948f22ef01cSRoman Divacky     // Read a record.
949f22ef01cSRoman Divacky     Record.clear();
950f22ef01cSRoman Divacky     Value *V = 0;
951f22ef01cSRoman Divacky     unsigned BitCode = Stream.ReadRecord(Code, Record);
952f22ef01cSRoman Divacky     switch (BitCode) {
953f22ef01cSRoman Divacky     default:  // Default behavior: unknown constant
954f22ef01cSRoman Divacky     case bitc::CST_CODE_UNDEF:     // UNDEF
955f22ef01cSRoman Divacky       V = UndefValue::get(CurTy);
956f22ef01cSRoman Divacky       break;
957f22ef01cSRoman Divacky     case bitc::CST_CODE_SETTYPE:   // SETTYPE: [typeid]
958f22ef01cSRoman Divacky       if (Record.empty())
959f22ef01cSRoman Divacky         return Error("Malformed CST_SETTYPE record");
960f22ef01cSRoman Divacky       if (Record[0] >= TypeList.size())
961f22ef01cSRoman Divacky         return Error("Invalid Type ID in CST_SETTYPE record");
962f22ef01cSRoman Divacky       CurTy = TypeList[Record[0]];
963f22ef01cSRoman Divacky       continue;  // Skip the ValueList manipulation.
964f22ef01cSRoman Divacky     case bitc::CST_CODE_NULL:      // NULL
965f22ef01cSRoman Divacky       V = Constant::getNullValue(CurTy);
966f22ef01cSRoman Divacky       break;
967f22ef01cSRoman Divacky     case bitc::CST_CODE_INTEGER:   // INTEGER: [intval]
968f22ef01cSRoman Divacky       if (!CurTy->isIntegerTy() || Record.empty())
969f22ef01cSRoman Divacky         return Error("Invalid CST_INTEGER record");
970f22ef01cSRoman Divacky       V = ConstantInt::get(CurTy, DecodeSignRotatedValue(Record[0]));
971f22ef01cSRoman Divacky       break;
972f22ef01cSRoman Divacky     case bitc::CST_CODE_WIDE_INTEGER: {// WIDE_INTEGER: [n x intval]
973f22ef01cSRoman Divacky       if (!CurTy->isIntegerTy() || Record.empty())
974f22ef01cSRoman Divacky         return Error("Invalid WIDE_INTEGER record");
975f22ef01cSRoman Divacky 
976f22ef01cSRoman Divacky       unsigned NumWords = Record.size();
977f22ef01cSRoman Divacky       SmallVector<uint64_t, 8> Words;
978f22ef01cSRoman Divacky       Words.resize(NumWords);
979f22ef01cSRoman Divacky       for (unsigned i = 0; i != NumWords; ++i)
980f22ef01cSRoman Divacky         Words[i] = DecodeSignRotatedValue(Record[i]);
981f22ef01cSRoman Divacky       V = ConstantInt::get(Context,
982f22ef01cSRoman Divacky                            APInt(cast<IntegerType>(CurTy)->getBitWidth(),
983f22ef01cSRoman Divacky                            NumWords, &Words[0]));
984f22ef01cSRoman Divacky       break;
985f22ef01cSRoman Divacky     }
986f22ef01cSRoman Divacky     case bitc::CST_CODE_FLOAT: {    // FLOAT: [fpval]
987f22ef01cSRoman Divacky       if (Record.empty())
988f22ef01cSRoman Divacky         return Error("Invalid FLOAT record");
989f22ef01cSRoman Divacky       if (CurTy->isFloatTy())
990f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(32, (uint32_t)Record[0])));
991f22ef01cSRoman Divacky       else if (CurTy->isDoubleTy())
992f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(64, Record[0])));
993f22ef01cSRoman Divacky       else if (CurTy->isX86_FP80Ty()) {
994f22ef01cSRoman Divacky         // Bits are not stored the same way as a normal i80 APInt, compensate.
995f22ef01cSRoman Divacky         uint64_t Rearrange[2];
996f22ef01cSRoman Divacky         Rearrange[0] = (Record[1] & 0xffffLL) | (Record[0] << 16);
997f22ef01cSRoman Divacky         Rearrange[1] = Record[0] >> 48;
998f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(80, 2, Rearrange)));
999f22ef01cSRoman Divacky       } else if (CurTy->isFP128Ty())
1000f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0]), true));
1001f22ef01cSRoman Divacky       else if (CurTy->isPPC_FP128Ty())
1002f22ef01cSRoman Divacky         V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0])));
1003f22ef01cSRoman Divacky       else
1004f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);
1005f22ef01cSRoman Divacky       break;
1006f22ef01cSRoman Divacky     }
1007f22ef01cSRoman Divacky 
1008f22ef01cSRoman Divacky     case bitc::CST_CODE_AGGREGATE: {// AGGREGATE: [n x value number]
1009f22ef01cSRoman Divacky       if (Record.empty())
1010f22ef01cSRoman Divacky         return Error("Invalid CST_AGGREGATE record");
1011f22ef01cSRoman Divacky 
1012f22ef01cSRoman Divacky       unsigned Size = Record.size();
1013f22ef01cSRoman Divacky       std::vector<Constant*> Elts;
1014f22ef01cSRoman Divacky 
1015f22ef01cSRoman Divacky       if (const StructType *STy = dyn_cast<StructType>(CurTy)) {
1016f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1017f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i],
1018f22ef01cSRoman Divacky                                                      STy->getElementType(i)));
1019f22ef01cSRoman Divacky         V = ConstantStruct::get(STy, Elts);
1020f22ef01cSRoman Divacky       } else if (const UnionType *UnTy = dyn_cast<UnionType>(CurTy)) {
1021f22ef01cSRoman Divacky         uint64_t Index = Record[0];
1022f22ef01cSRoman Divacky         Constant *Val = ValueList.getConstantFwdRef(Record[1],
1023f22ef01cSRoman Divacky                                         UnTy->getElementType(Index));
1024f22ef01cSRoman Divacky         V = ConstantUnion::get(UnTy, Val);
1025f22ef01cSRoman Divacky       } else if (const ArrayType *ATy = dyn_cast<ArrayType>(CurTy)) {
1026f22ef01cSRoman Divacky         const Type *EltTy = ATy->getElementType();
1027f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1028f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
1029f22ef01cSRoman Divacky         V = ConstantArray::get(ATy, Elts);
1030f22ef01cSRoman Divacky       } else if (const VectorType *VTy = dyn_cast<VectorType>(CurTy)) {
1031f22ef01cSRoman Divacky         const Type *EltTy = VTy->getElementType();
1032f22ef01cSRoman Divacky         for (unsigned i = 0; i != Size; ++i)
1033f22ef01cSRoman Divacky           Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
1034f22ef01cSRoman Divacky         V = ConstantVector::get(Elts);
1035f22ef01cSRoman Divacky       } else {
1036f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);
1037f22ef01cSRoman Divacky       }
1038f22ef01cSRoman Divacky       break;
1039f22ef01cSRoman Divacky     }
1040f22ef01cSRoman Divacky     case bitc::CST_CODE_STRING: { // STRING: [values]
1041f22ef01cSRoman Divacky       if (Record.empty())
1042f22ef01cSRoman Divacky         return Error("Invalid CST_AGGREGATE record");
1043f22ef01cSRoman Divacky 
1044f22ef01cSRoman Divacky       const ArrayType *ATy = cast<ArrayType>(CurTy);
1045f22ef01cSRoman Divacky       const Type *EltTy = ATy->getElementType();
1046f22ef01cSRoman Divacky 
1047f22ef01cSRoman Divacky       unsigned Size = Record.size();
1048f22ef01cSRoman Divacky       std::vector<Constant*> Elts;
1049f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; ++i)
1050f22ef01cSRoman Divacky         Elts.push_back(ConstantInt::get(EltTy, Record[i]));
1051f22ef01cSRoman Divacky       V = ConstantArray::get(ATy, Elts);
1052f22ef01cSRoman Divacky       break;
1053f22ef01cSRoman Divacky     }
1054f22ef01cSRoman Divacky     case bitc::CST_CODE_CSTRING: { // CSTRING: [values]
1055f22ef01cSRoman Divacky       if (Record.empty())
1056f22ef01cSRoman Divacky         return Error("Invalid CST_AGGREGATE record");
1057f22ef01cSRoman Divacky 
1058f22ef01cSRoman Divacky       const ArrayType *ATy = cast<ArrayType>(CurTy);
1059f22ef01cSRoman Divacky       const Type *EltTy = ATy->getElementType();
1060f22ef01cSRoman Divacky 
1061f22ef01cSRoman Divacky       unsigned Size = Record.size();
1062f22ef01cSRoman Divacky       std::vector<Constant*> Elts;
1063f22ef01cSRoman Divacky       for (unsigned i = 0; i != Size; ++i)
1064f22ef01cSRoman Divacky         Elts.push_back(ConstantInt::get(EltTy, Record[i]));
1065f22ef01cSRoman Divacky       Elts.push_back(Constant::getNullValue(EltTy));
1066f22ef01cSRoman Divacky       V = ConstantArray::get(ATy, Elts);
1067f22ef01cSRoman Divacky       break;
1068f22ef01cSRoman Divacky     }
1069f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_BINOP: {  // CE_BINOP: [opcode, opval, opval]
1070f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_BINOP record");
1071f22ef01cSRoman Divacky       int Opc = GetDecodedBinaryOpcode(Record[0], CurTy);
1072f22ef01cSRoman Divacky       if (Opc < 0) {
1073f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);  // Unknown binop.
1074f22ef01cSRoman Divacky       } else {
1075f22ef01cSRoman Divacky         Constant *LHS = ValueList.getConstantFwdRef(Record[1], CurTy);
1076f22ef01cSRoman Divacky         Constant *RHS = ValueList.getConstantFwdRef(Record[2], CurTy);
1077f22ef01cSRoman Divacky         unsigned Flags = 0;
1078f22ef01cSRoman Divacky         if (Record.size() >= 4) {
1079f22ef01cSRoman Divacky           if (Opc == Instruction::Add ||
1080f22ef01cSRoman Divacky               Opc == Instruction::Sub ||
1081f22ef01cSRoman Divacky               Opc == Instruction::Mul) {
1082f22ef01cSRoman Divacky             if (Record[3] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1083f22ef01cSRoman Divacky               Flags |= OverflowingBinaryOperator::NoSignedWrap;
1084f22ef01cSRoman Divacky             if (Record[3] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1085f22ef01cSRoman Divacky               Flags |= OverflowingBinaryOperator::NoUnsignedWrap;
1086f22ef01cSRoman Divacky           } else if (Opc == Instruction::SDiv) {
1087f22ef01cSRoman Divacky             if (Record[3] & (1 << bitc::SDIV_EXACT))
1088f22ef01cSRoman Divacky               Flags |= SDivOperator::IsExact;
1089f22ef01cSRoman Divacky           }
1090f22ef01cSRoman Divacky         }
1091f22ef01cSRoman Divacky         V = ConstantExpr::get(Opc, LHS, RHS, Flags);
1092f22ef01cSRoman Divacky       }
1093f22ef01cSRoman Divacky       break;
1094f22ef01cSRoman Divacky     }
1095f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_CAST: {  // CE_CAST: [opcode, opty, opval]
1096f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_CAST record");
1097f22ef01cSRoman Divacky       int Opc = GetDecodedCastOpcode(Record[0]);
1098f22ef01cSRoman Divacky       if (Opc < 0) {
1099f22ef01cSRoman Divacky         V = UndefValue::get(CurTy);  // Unknown cast.
1100f22ef01cSRoman Divacky       } else {
1101f22ef01cSRoman Divacky         const Type *OpTy = getTypeByID(Record[1]);
1102f22ef01cSRoman Divacky         if (!OpTy) return Error("Invalid CE_CAST record");
1103f22ef01cSRoman Divacky         Constant *Op = ValueList.getConstantFwdRef(Record[2], OpTy);
1104f22ef01cSRoman Divacky         V = ConstantExpr::getCast(Opc, Op, CurTy);
1105f22ef01cSRoman Divacky       }
1106f22ef01cSRoman Divacky       break;
1107f22ef01cSRoman Divacky     }
1108f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_INBOUNDS_GEP:
1109f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_GEP: {  // CE_GEP:        [n x operands]
1110f22ef01cSRoman Divacky       if (Record.size() & 1) return Error("Invalid CE_GEP record");
1111f22ef01cSRoman Divacky       SmallVector<Constant*, 16> Elts;
1112f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
1113f22ef01cSRoman Divacky         const Type *ElTy = getTypeByID(Record[i]);
1114f22ef01cSRoman Divacky         if (!ElTy) return Error("Invalid CE_GEP record");
1115f22ef01cSRoman Divacky         Elts.push_back(ValueList.getConstantFwdRef(Record[i+1], ElTy));
1116f22ef01cSRoman Divacky       }
1117f22ef01cSRoman Divacky       if (BitCode == bitc::CST_CODE_CE_INBOUNDS_GEP)
1118f22ef01cSRoman Divacky         V = ConstantExpr::getInBoundsGetElementPtr(Elts[0], &Elts[1],
1119f22ef01cSRoman Divacky                                                    Elts.size()-1);
1120f22ef01cSRoman Divacky       else
1121f22ef01cSRoman Divacky         V = ConstantExpr::getGetElementPtr(Elts[0], &Elts[1],
1122f22ef01cSRoman Divacky                                            Elts.size()-1);
1123f22ef01cSRoman Divacky       break;
1124f22ef01cSRoman Divacky     }
1125f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SELECT:  // CE_SELECT: [opval#, opval#, opval#]
1126f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_SELECT record");
1127f22ef01cSRoman Divacky       V = ConstantExpr::getSelect(ValueList.getConstantFwdRef(Record[0],
1128f22ef01cSRoman Divacky                                                               Type::getInt1Ty(Context)),
1129f22ef01cSRoman Divacky                                   ValueList.getConstantFwdRef(Record[1],CurTy),
1130f22ef01cSRoman Divacky                                   ValueList.getConstantFwdRef(Record[2],CurTy));
1131f22ef01cSRoman Divacky       break;
1132f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_EXTRACTELT: { // CE_EXTRACTELT: [opty, opval, opval]
1133f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_EXTRACTELT record");
1134f22ef01cSRoman Divacky       const VectorType *OpTy =
1135f22ef01cSRoman Divacky         dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
1136f22ef01cSRoman Divacky       if (OpTy == 0) return Error("Invalid CE_EXTRACTELT record");
1137f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1138f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
1139f22ef01cSRoman Divacky       V = ConstantExpr::getExtractElement(Op0, Op1);
1140f22ef01cSRoman Divacky       break;
1141f22ef01cSRoman Divacky     }
1142f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_INSERTELT: { // CE_INSERTELT: [opval, opval, opval]
1143f22ef01cSRoman Divacky       const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1144f22ef01cSRoman Divacky       if (Record.size() < 3 || OpTy == 0)
1145f22ef01cSRoman Divacky         return Error("Invalid CE_INSERTELT record");
1146f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1147f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[1],
1148f22ef01cSRoman Divacky                                                   OpTy->getElementType());
1149f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
1150f22ef01cSRoman Divacky       V = ConstantExpr::getInsertElement(Op0, Op1, Op2);
1151f22ef01cSRoman Divacky       break;
1152f22ef01cSRoman Divacky     }
1153f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SHUFFLEVEC: { // CE_SHUFFLEVEC: [opval, opval, opval]
1154f22ef01cSRoman Divacky       const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1155f22ef01cSRoman Divacky       if (Record.size() < 3 || OpTy == 0)
1156f22ef01cSRoman Divacky         return Error("Invalid CE_SHUFFLEVEC record");
1157f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1158f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[1], OpTy);
1159f22ef01cSRoman Divacky       const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
1160f22ef01cSRoman Divacky                                                  OpTy->getNumElements());
1161f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[2], ShufTy);
1162f22ef01cSRoman Divacky       V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
1163f22ef01cSRoman Divacky       break;
1164f22ef01cSRoman Divacky     }
1165f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_SHUFVEC_EX: { // [opty, opval, opval, opval]
1166f22ef01cSRoman Divacky       const VectorType *RTy = dyn_cast<VectorType>(CurTy);
1167f22ef01cSRoman Divacky       const VectorType *OpTy = dyn_cast<VectorType>(getTypeByID(Record[0]));
1168f22ef01cSRoman Divacky       if (Record.size() < 4 || RTy == 0 || OpTy == 0)
1169f22ef01cSRoman Divacky         return Error("Invalid CE_SHUFVEC_EX record");
1170f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1171f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1172f22ef01cSRoman Divacky       const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
1173f22ef01cSRoman Divacky                                                  RTy->getNumElements());
1174f22ef01cSRoman Divacky       Constant *Op2 = ValueList.getConstantFwdRef(Record[3], ShufTy);
1175f22ef01cSRoman Divacky       V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
1176f22ef01cSRoman Divacky       break;
1177f22ef01cSRoman Divacky     }
1178f22ef01cSRoman Divacky     case bitc::CST_CODE_CE_CMP: {     // CE_CMP: [opty, opval, opval, pred]
1179f22ef01cSRoman Divacky       if (Record.size() < 4) return Error("Invalid CE_CMP record");
1180f22ef01cSRoman Divacky       const Type *OpTy = getTypeByID(Record[0]);
1181f22ef01cSRoman Divacky       if (OpTy == 0) return Error("Invalid CE_CMP record");
1182f22ef01cSRoman Divacky       Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1183f22ef01cSRoman Divacky       Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1184f22ef01cSRoman Divacky 
1185f22ef01cSRoman Divacky       if (OpTy->isFPOrFPVectorTy())
1186f22ef01cSRoman Divacky         V = ConstantExpr::getFCmp(Record[3], Op0, Op1);
1187f22ef01cSRoman Divacky       else
1188f22ef01cSRoman Divacky         V = ConstantExpr::getICmp(Record[3], Op0, Op1);
1189f22ef01cSRoman Divacky       break;
1190f22ef01cSRoman Divacky     }
1191f22ef01cSRoman Divacky     case bitc::CST_CODE_INLINEASM: {
1192f22ef01cSRoman Divacky       if (Record.size() < 2) return Error("Invalid INLINEASM record");
1193f22ef01cSRoman Divacky       std::string AsmStr, ConstrStr;
1194f22ef01cSRoman Divacky       bool HasSideEffects = Record[0] & 1;
1195f22ef01cSRoman Divacky       bool IsAlignStack = Record[0] >> 1;
1196f22ef01cSRoman Divacky       unsigned AsmStrSize = Record[1];
1197f22ef01cSRoman Divacky       if (2+AsmStrSize >= Record.size())
1198f22ef01cSRoman Divacky         return Error("Invalid INLINEASM record");
1199f22ef01cSRoman Divacky       unsigned ConstStrSize = Record[2+AsmStrSize];
1200f22ef01cSRoman Divacky       if (3+AsmStrSize+ConstStrSize > Record.size())
1201f22ef01cSRoman Divacky         return Error("Invalid INLINEASM record");
1202f22ef01cSRoman Divacky 
1203f22ef01cSRoman Divacky       for (unsigned i = 0; i != AsmStrSize; ++i)
1204f22ef01cSRoman Divacky         AsmStr += (char)Record[2+i];
1205f22ef01cSRoman Divacky       for (unsigned i = 0; i != ConstStrSize; ++i)
1206f22ef01cSRoman Divacky         ConstrStr += (char)Record[3+AsmStrSize+i];
1207f22ef01cSRoman Divacky       const PointerType *PTy = cast<PointerType>(CurTy);
1208f22ef01cSRoman Divacky       V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
1209f22ef01cSRoman Divacky                          AsmStr, ConstrStr, HasSideEffects, IsAlignStack);
1210f22ef01cSRoman Divacky       break;
1211f22ef01cSRoman Divacky     }
1212f22ef01cSRoman Divacky     case bitc::CST_CODE_BLOCKADDRESS:{
1213f22ef01cSRoman Divacky       if (Record.size() < 3) return Error("Invalid CE_BLOCKADDRESS record");
1214f22ef01cSRoman Divacky       const Type *FnTy = getTypeByID(Record[0]);
1215f22ef01cSRoman Divacky       if (FnTy == 0) return Error("Invalid CE_BLOCKADDRESS record");
1216f22ef01cSRoman Divacky       Function *Fn =
1217f22ef01cSRoman Divacky         dyn_cast_or_null<Function>(ValueList.getConstantFwdRef(Record[1],FnTy));
1218f22ef01cSRoman Divacky       if (Fn == 0) return Error("Invalid CE_BLOCKADDRESS record");
1219f22ef01cSRoman Divacky 
1220f22ef01cSRoman Divacky       GlobalVariable *FwdRef = new GlobalVariable(*Fn->getParent(),
1221f22ef01cSRoman Divacky                                                   Type::getInt8Ty(Context),
1222f22ef01cSRoman Divacky                                             false, GlobalValue::InternalLinkage,
1223f22ef01cSRoman Divacky                                                   0, "");
1224f22ef01cSRoman Divacky       BlockAddrFwdRefs[Fn].push_back(std::make_pair(Record[2], FwdRef));
1225f22ef01cSRoman Divacky       V = FwdRef;
1226f22ef01cSRoman Divacky       break;
1227f22ef01cSRoman Divacky     }
1228f22ef01cSRoman Divacky     }
1229f22ef01cSRoman Divacky 
1230f22ef01cSRoman Divacky     ValueList.AssignValue(V, NextCstNo);
1231f22ef01cSRoman Divacky     ++NextCstNo;
1232f22ef01cSRoman Divacky   }
1233f22ef01cSRoman Divacky 
1234f22ef01cSRoman Divacky   if (NextCstNo != ValueList.size())
1235f22ef01cSRoman Divacky     return Error("Invalid constant reference!");
1236f22ef01cSRoman Divacky 
1237f22ef01cSRoman Divacky   if (Stream.ReadBlockEnd())
1238f22ef01cSRoman Divacky     return Error("Error at end of constants block");
1239f22ef01cSRoman Divacky 
1240f22ef01cSRoman Divacky   // Once all the constants have been read, go through and resolve forward
1241f22ef01cSRoman Divacky   // references.
1242f22ef01cSRoman Divacky   ValueList.ResolveConstantForwardRefs();
1243f22ef01cSRoman Divacky   return false;
1244f22ef01cSRoman Divacky }
1245f22ef01cSRoman Divacky 
1246f22ef01cSRoman Divacky /// RememberAndSkipFunctionBody - When we see the block for a function body,
1247f22ef01cSRoman Divacky /// remember where it is and then skip it.  This lets us lazily deserialize the
1248f22ef01cSRoman Divacky /// functions.
1249f22ef01cSRoman Divacky bool BitcodeReader::RememberAndSkipFunctionBody() {
1250f22ef01cSRoman Divacky   // Get the function we are talking about.
1251f22ef01cSRoman Divacky   if (FunctionsWithBodies.empty())
1252f22ef01cSRoman Divacky     return Error("Insufficient function protos");
1253f22ef01cSRoman Divacky 
1254f22ef01cSRoman Divacky   Function *Fn = FunctionsWithBodies.back();
1255f22ef01cSRoman Divacky   FunctionsWithBodies.pop_back();
1256f22ef01cSRoman Divacky 
1257f22ef01cSRoman Divacky   // Save the current stream state.
1258f22ef01cSRoman Divacky   uint64_t CurBit = Stream.GetCurrentBitNo();
1259f22ef01cSRoman Divacky   DeferredFunctionInfo[Fn] = CurBit;
1260f22ef01cSRoman Divacky 
1261f22ef01cSRoman Divacky   // Skip over the function block for now.
1262f22ef01cSRoman Divacky   if (Stream.SkipBlock())
1263f22ef01cSRoman Divacky     return Error("Malformed block record");
1264f22ef01cSRoman Divacky   return false;
1265f22ef01cSRoman Divacky }
1266f22ef01cSRoman Divacky 
1267f22ef01cSRoman Divacky bool BitcodeReader::ParseModule() {
1268f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
1269f22ef01cSRoman Divacky     return Error("Malformed block record");
1270f22ef01cSRoman Divacky 
1271f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1272f22ef01cSRoman Divacky   std::vector<std::string> SectionTable;
1273f22ef01cSRoman Divacky   std::vector<std::string> GCTable;
1274f22ef01cSRoman Divacky 
1275f22ef01cSRoman Divacky   // Read all the records for this module.
1276f22ef01cSRoman Divacky   while (!Stream.AtEndOfStream()) {
1277f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
1278f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
1279f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
1280f22ef01cSRoman Divacky         return Error("Error at end of module block");
1281f22ef01cSRoman Divacky 
1282f22ef01cSRoman Divacky       // Patch the initializers for globals and aliases up.
1283f22ef01cSRoman Divacky       ResolveGlobalAndAliasInits();
1284f22ef01cSRoman Divacky       if (!GlobalInits.empty() || !AliasInits.empty())
1285f22ef01cSRoman Divacky         return Error("Malformed global initializer set");
1286f22ef01cSRoman Divacky       if (!FunctionsWithBodies.empty())
1287f22ef01cSRoman Divacky         return Error("Too few function bodies found");
1288f22ef01cSRoman Divacky 
1289f22ef01cSRoman Divacky       // Look for intrinsic functions which need to be upgraded at some point
1290f22ef01cSRoman Divacky       for (Module::iterator FI = TheModule->begin(), FE = TheModule->end();
1291f22ef01cSRoman Divacky            FI != FE; ++FI) {
1292f22ef01cSRoman Divacky         Function* NewFn;
1293f22ef01cSRoman Divacky         if (UpgradeIntrinsicFunction(FI, NewFn))
1294f22ef01cSRoman Divacky           UpgradedIntrinsics.push_back(std::make_pair(FI, NewFn));
1295f22ef01cSRoman Divacky       }
1296f22ef01cSRoman Divacky 
1297f22ef01cSRoman Divacky       // Force deallocation of memory for these vectors to favor the client that
1298f22ef01cSRoman Divacky       // want lazy deserialization.
1299f22ef01cSRoman Divacky       std::vector<std::pair<GlobalVariable*, unsigned> >().swap(GlobalInits);
1300f22ef01cSRoman Divacky       std::vector<std::pair<GlobalAlias*, unsigned> >().swap(AliasInits);
1301f22ef01cSRoman Divacky       std::vector<Function*>().swap(FunctionsWithBodies);
1302f22ef01cSRoman Divacky       return false;
1303f22ef01cSRoman Divacky     }
1304f22ef01cSRoman Divacky 
1305f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
1306f22ef01cSRoman Divacky       switch (Stream.ReadSubBlockID()) {
1307f22ef01cSRoman Divacky       default:  // Skip unknown content.
1308f22ef01cSRoman Divacky         if (Stream.SkipBlock())
1309f22ef01cSRoman Divacky           return Error("Malformed block record");
1310f22ef01cSRoman Divacky         break;
1311f22ef01cSRoman Divacky       case bitc::BLOCKINFO_BLOCK_ID:
1312f22ef01cSRoman Divacky         if (Stream.ReadBlockInfoBlock())
1313f22ef01cSRoman Divacky           return Error("Malformed BlockInfoBlock");
1314f22ef01cSRoman Divacky         break;
1315f22ef01cSRoman Divacky       case bitc::PARAMATTR_BLOCK_ID:
1316f22ef01cSRoman Divacky         if (ParseAttributeBlock())
1317f22ef01cSRoman Divacky           return true;
1318f22ef01cSRoman Divacky         break;
1319f22ef01cSRoman Divacky       case bitc::TYPE_BLOCK_ID:
1320f22ef01cSRoman Divacky         if (ParseTypeTable())
1321f22ef01cSRoman Divacky           return true;
1322f22ef01cSRoman Divacky         break;
1323f22ef01cSRoman Divacky       case bitc::TYPE_SYMTAB_BLOCK_ID:
1324f22ef01cSRoman Divacky         if (ParseTypeSymbolTable())
1325f22ef01cSRoman Divacky           return true;
1326f22ef01cSRoman Divacky         break;
1327f22ef01cSRoman Divacky       case bitc::VALUE_SYMTAB_BLOCK_ID:
1328f22ef01cSRoman Divacky         if (ParseValueSymbolTable())
1329f22ef01cSRoman Divacky           return true;
1330f22ef01cSRoman Divacky         break;
1331f22ef01cSRoman Divacky       case bitc::CONSTANTS_BLOCK_ID:
1332f22ef01cSRoman Divacky         if (ParseConstants() || ResolveGlobalAndAliasInits())
1333f22ef01cSRoman Divacky           return true;
1334f22ef01cSRoman Divacky         break;
1335f22ef01cSRoman Divacky       case bitc::METADATA_BLOCK_ID:
1336f22ef01cSRoman Divacky         if (ParseMetadata())
1337f22ef01cSRoman Divacky           return true;
1338f22ef01cSRoman Divacky         break;
1339f22ef01cSRoman Divacky       case bitc::FUNCTION_BLOCK_ID:
1340f22ef01cSRoman Divacky         // If this is the first function body we've seen, reverse the
1341f22ef01cSRoman Divacky         // FunctionsWithBodies list.
1342f22ef01cSRoman Divacky         if (!HasReversedFunctionsWithBodies) {
1343f22ef01cSRoman Divacky           std::reverse(FunctionsWithBodies.begin(), FunctionsWithBodies.end());
1344f22ef01cSRoman Divacky           HasReversedFunctionsWithBodies = true;
1345f22ef01cSRoman Divacky         }
1346f22ef01cSRoman Divacky 
1347f22ef01cSRoman Divacky         if (RememberAndSkipFunctionBody())
1348f22ef01cSRoman Divacky           return true;
1349f22ef01cSRoman Divacky         break;
1350f22ef01cSRoman Divacky       }
1351f22ef01cSRoman Divacky       continue;
1352f22ef01cSRoman Divacky     }
1353f22ef01cSRoman Divacky 
1354f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
1355f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
1356f22ef01cSRoman Divacky       continue;
1357f22ef01cSRoman Divacky     }
1358f22ef01cSRoman Divacky 
1359f22ef01cSRoman Divacky     // Read a record.
1360f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
1361f22ef01cSRoman Divacky     default: break;  // Default behavior, ignore unknown content.
1362f22ef01cSRoman Divacky     case bitc::MODULE_CODE_VERSION:  // VERSION: [version#]
1363f22ef01cSRoman Divacky       if (Record.size() < 1)
1364f22ef01cSRoman Divacky         return Error("Malformed MODULE_CODE_VERSION");
1365f22ef01cSRoman Divacky       // Only version #0 is supported so far.
1366f22ef01cSRoman Divacky       if (Record[0] != 0)
1367f22ef01cSRoman Divacky         return Error("Unknown bitstream version!");
1368f22ef01cSRoman Divacky       break;
1369f22ef01cSRoman Divacky     case bitc::MODULE_CODE_TRIPLE: {  // TRIPLE: [strchr x N]
1370f22ef01cSRoman Divacky       std::string S;
1371f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1372f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_TRIPLE record");
1373f22ef01cSRoman Divacky       TheModule->setTargetTriple(S);
1374f22ef01cSRoman Divacky       break;
1375f22ef01cSRoman Divacky     }
1376f22ef01cSRoman Divacky     case bitc::MODULE_CODE_DATALAYOUT: {  // DATALAYOUT: [strchr x N]
1377f22ef01cSRoman Divacky       std::string S;
1378f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1379f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_DATALAYOUT record");
1380f22ef01cSRoman Divacky       TheModule->setDataLayout(S);
1381f22ef01cSRoman Divacky       break;
1382f22ef01cSRoman Divacky     }
1383f22ef01cSRoman Divacky     case bitc::MODULE_CODE_ASM: {  // ASM: [strchr x N]
1384f22ef01cSRoman Divacky       std::string S;
1385f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1386f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_ASM record");
1387f22ef01cSRoman Divacky       TheModule->setModuleInlineAsm(S);
1388f22ef01cSRoman Divacky       break;
1389f22ef01cSRoman Divacky     }
1390f22ef01cSRoman Divacky     case bitc::MODULE_CODE_DEPLIB: {  // DEPLIB: [strchr x N]
1391f22ef01cSRoman Divacky       std::string S;
1392f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1393f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_DEPLIB record");
1394f22ef01cSRoman Divacky       TheModule->addLibrary(S);
1395f22ef01cSRoman Divacky       break;
1396f22ef01cSRoman Divacky     }
1397f22ef01cSRoman Divacky     case bitc::MODULE_CODE_SECTIONNAME: {  // SECTIONNAME: [strchr x N]
1398f22ef01cSRoman Divacky       std::string S;
1399f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1400f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_SECTIONNAME record");
1401f22ef01cSRoman Divacky       SectionTable.push_back(S);
1402f22ef01cSRoman Divacky       break;
1403f22ef01cSRoman Divacky     }
1404f22ef01cSRoman Divacky     case bitc::MODULE_CODE_GCNAME: {  // SECTIONNAME: [strchr x N]
1405f22ef01cSRoman Divacky       std::string S;
1406f22ef01cSRoman Divacky       if (ConvertToString(Record, 0, S))
1407f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_GCNAME record");
1408f22ef01cSRoman Divacky       GCTable.push_back(S);
1409f22ef01cSRoman Divacky       break;
1410f22ef01cSRoman Divacky     }
1411f22ef01cSRoman Divacky     // GLOBALVAR: [pointer type, isconst, initid,
1412f22ef01cSRoman Divacky     //             linkage, alignment, section, visibility, threadlocal]
1413f22ef01cSRoman Divacky     case bitc::MODULE_CODE_GLOBALVAR: {
1414f22ef01cSRoman Divacky       if (Record.size() < 6)
1415f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_GLOBALVAR record");
1416f22ef01cSRoman Divacky       const Type *Ty = getTypeByID(Record[0]);
1417f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
1418f22ef01cSRoman Divacky         return Error("Global not a pointer type!");
1419f22ef01cSRoman Divacky       unsigned AddressSpace = cast<PointerType>(Ty)->getAddressSpace();
1420f22ef01cSRoman Divacky       Ty = cast<PointerType>(Ty)->getElementType();
1421f22ef01cSRoman Divacky 
1422f22ef01cSRoman Divacky       bool isConstant = Record[1];
1423f22ef01cSRoman Divacky       GlobalValue::LinkageTypes Linkage = GetDecodedLinkage(Record[3]);
1424f22ef01cSRoman Divacky       unsigned Alignment = (1 << Record[4]) >> 1;
1425f22ef01cSRoman Divacky       std::string Section;
1426f22ef01cSRoman Divacky       if (Record[5]) {
1427f22ef01cSRoman Divacky         if (Record[5]-1 >= SectionTable.size())
1428f22ef01cSRoman Divacky           return Error("Invalid section ID");
1429f22ef01cSRoman Divacky         Section = SectionTable[Record[5]-1];
1430f22ef01cSRoman Divacky       }
1431f22ef01cSRoman Divacky       GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
1432f22ef01cSRoman Divacky       if (Record.size() > 6)
1433f22ef01cSRoman Divacky         Visibility = GetDecodedVisibility(Record[6]);
1434f22ef01cSRoman Divacky       bool isThreadLocal = false;
1435f22ef01cSRoman Divacky       if (Record.size() > 7)
1436f22ef01cSRoman Divacky         isThreadLocal = Record[7];
1437f22ef01cSRoman Divacky 
1438f22ef01cSRoman Divacky       GlobalVariable *NewGV =
1439f22ef01cSRoman Divacky         new GlobalVariable(*TheModule, Ty, isConstant, Linkage, 0, "", 0,
1440f22ef01cSRoman Divacky                            isThreadLocal, AddressSpace);
1441f22ef01cSRoman Divacky       NewGV->setAlignment(Alignment);
1442f22ef01cSRoman Divacky       if (!Section.empty())
1443f22ef01cSRoman Divacky         NewGV->setSection(Section);
1444f22ef01cSRoman Divacky       NewGV->setVisibility(Visibility);
1445f22ef01cSRoman Divacky       NewGV->setThreadLocal(isThreadLocal);
1446f22ef01cSRoman Divacky 
1447f22ef01cSRoman Divacky       ValueList.push_back(NewGV);
1448f22ef01cSRoman Divacky 
1449f22ef01cSRoman Divacky       // Remember which value to use for the global initializer.
1450f22ef01cSRoman Divacky       if (unsigned InitID = Record[2])
1451f22ef01cSRoman Divacky         GlobalInits.push_back(std::make_pair(NewGV, InitID-1));
1452f22ef01cSRoman Divacky       break;
1453f22ef01cSRoman Divacky     }
1454f22ef01cSRoman Divacky     // FUNCTION:  [type, callingconv, isproto, linkage, paramattr,
1455f22ef01cSRoman Divacky     //             alignment, section, visibility, gc]
1456f22ef01cSRoman Divacky     case bitc::MODULE_CODE_FUNCTION: {
1457f22ef01cSRoman Divacky       if (Record.size() < 8)
1458f22ef01cSRoman Divacky         return Error("Invalid MODULE_CODE_FUNCTION record");
1459f22ef01cSRoman Divacky       const Type *Ty = getTypeByID(Record[0]);
1460f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
1461f22ef01cSRoman Divacky         return Error("Function not a pointer type!");
1462f22ef01cSRoman Divacky       const FunctionType *FTy =
1463f22ef01cSRoman Divacky         dyn_cast<FunctionType>(cast<PointerType>(Ty)->getElementType());
1464f22ef01cSRoman Divacky       if (!FTy)
1465f22ef01cSRoman Divacky         return Error("Function not a pointer to function type!");
1466f22ef01cSRoman Divacky 
1467f22ef01cSRoman Divacky       Function *Func = Function::Create(FTy, GlobalValue::ExternalLinkage,
1468f22ef01cSRoman Divacky                                         "", TheModule);
1469f22ef01cSRoman Divacky 
1470f22ef01cSRoman Divacky       Func->setCallingConv(static_cast<CallingConv::ID>(Record[1]));
1471f22ef01cSRoman Divacky       bool isProto = Record[2];
1472f22ef01cSRoman Divacky       Func->setLinkage(GetDecodedLinkage(Record[3]));
1473f22ef01cSRoman Divacky       Func->setAttributes(getAttributes(Record[4]));
1474f22ef01cSRoman Divacky 
1475f22ef01cSRoman Divacky       Func->setAlignment((1 << Record[5]) >> 1);
1476f22ef01cSRoman Divacky       if (Record[6]) {
1477f22ef01cSRoman Divacky         if (Record[6]-1 >= SectionTable.size())
1478f22ef01cSRoman Divacky           return Error("Invalid section ID");
1479f22ef01cSRoman Divacky         Func->setSection(SectionTable[Record[6]-1]);
1480f22ef01cSRoman Divacky       }
1481f22ef01cSRoman Divacky       Func->setVisibility(GetDecodedVisibility(Record[7]));
1482f22ef01cSRoman Divacky       if (Record.size() > 8 && Record[8]) {
1483f22ef01cSRoman Divacky         if (Record[8]-1 > GCTable.size())
1484f22ef01cSRoman Divacky           return Error("Invalid GC ID");
1485f22ef01cSRoman Divacky         Func->setGC(GCTable[Record[8]-1].c_str());
1486f22ef01cSRoman Divacky       }
1487f22ef01cSRoman Divacky       ValueList.push_back(Func);
1488f22ef01cSRoman Divacky 
1489f22ef01cSRoman Divacky       // If this is a function with a body, remember the prototype we are
1490f22ef01cSRoman Divacky       // creating now, so that we can match up the body with them later.
1491f22ef01cSRoman Divacky       if (!isProto)
1492f22ef01cSRoman Divacky         FunctionsWithBodies.push_back(Func);
1493f22ef01cSRoman Divacky       break;
1494f22ef01cSRoman Divacky     }
1495f22ef01cSRoman Divacky     // ALIAS: [alias type, aliasee val#, linkage]
1496f22ef01cSRoman Divacky     // ALIAS: [alias type, aliasee val#, linkage, visibility]
1497f22ef01cSRoman Divacky     case bitc::MODULE_CODE_ALIAS: {
1498f22ef01cSRoman Divacky       if (Record.size() < 3)
1499f22ef01cSRoman Divacky         return Error("Invalid MODULE_ALIAS record");
1500f22ef01cSRoman Divacky       const Type *Ty = getTypeByID(Record[0]);
1501f22ef01cSRoman Divacky       if (!Ty->isPointerTy())
1502f22ef01cSRoman Divacky         return Error("Function not a pointer type!");
1503f22ef01cSRoman Divacky 
1504f22ef01cSRoman Divacky       GlobalAlias *NewGA = new GlobalAlias(Ty, GetDecodedLinkage(Record[2]),
1505f22ef01cSRoman Divacky                                            "", 0, TheModule);
1506f22ef01cSRoman Divacky       // Old bitcode files didn't have visibility field.
1507f22ef01cSRoman Divacky       if (Record.size() > 3)
1508f22ef01cSRoman Divacky         NewGA->setVisibility(GetDecodedVisibility(Record[3]));
1509f22ef01cSRoman Divacky       ValueList.push_back(NewGA);
1510f22ef01cSRoman Divacky       AliasInits.push_back(std::make_pair(NewGA, Record[1]));
1511f22ef01cSRoman Divacky       break;
1512f22ef01cSRoman Divacky     }
1513f22ef01cSRoman Divacky     /// MODULE_CODE_PURGEVALS: [numvals]
1514f22ef01cSRoman Divacky     case bitc::MODULE_CODE_PURGEVALS:
1515f22ef01cSRoman Divacky       // Trim down the value list to the specified size.
1516f22ef01cSRoman Divacky       if (Record.size() < 1 || Record[0] > ValueList.size())
1517f22ef01cSRoman Divacky         return Error("Invalid MODULE_PURGEVALS record");
1518f22ef01cSRoman Divacky       ValueList.shrinkTo(Record[0]);
1519f22ef01cSRoman Divacky       break;
1520f22ef01cSRoman Divacky     }
1521f22ef01cSRoman Divacky     Record.clear();
1522f22ef01cSRoman Divacky   }
1523f22ef01cSRoman Divacky 
1524f22ef01cSRoman Divacky   return Error("Premature end of bitstream");
1525f22ef01cSRoman Divacky }
1526f22ef01cSRoman Divacky 
1527f22ef01cSRoman Divacky bool BitcodeReader::ParseBitcodeInto(Module *M) {
1528f22ef01cSRoman Divacky   TheModule = 0;
1529f22ef01cSRoman Divacky 
1530f22ef01cSRoman Divacky   unsigned char *BufPtr = (unsigned char *)Buffer->getBufferStart();
1531f22ef01cSRoman Divacky   unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
1532f22ef01cSRoman Divacky 
1533f22ef01cSRoman Divacky   if (Buffer->getBufferSize() & 3) {
1534f22ef01cSRoman Divacky     if (!isRawBitcode(BufPtr, BufEnd) && !isBitcodeWrapper(BufPtr, BufEnd))
1535f22ef01cSRoman Divacky       return Error("Invalid bitcode signature");
1536f22ef01cSRoman Divacky     else
1537f22ef01cSRoman Divacky       return Error("Bitcode stream should be a multiple of 4 bytes in length");
1538f22ef01cSRoman Divacky   }
1539f22ef01cSRoman Divacky 
1540f22ef01cSRoman Divacky   // If we have a wrapper header, parse it and ignore the non-bc file contents.
1541f22ef01cSRoman Divacky   // The magic number is 0x0B17C0DE stored in little endian.
1542f22ef01cSRoman Divacky   if (isBitcodeWrapper(BufPtr, BufEnd))
1543f22ef01cSRoman Divacky     if (SkipBitcodeWrapperHeader(BufPtr, BufEnd))
1544f22ef01cSRoman Divacky       return Error("Invalid bitcode wrapper header");
1545f22ef01cSRoman Divacky 
1546f22ef01cSRoman Divacky   StreamFile.init(BufPtr, BufEnd);
1547f22ef01cSRoman Divacky   Stream.init(StreamFile);
1548f22ef01cSRoman Divacky 
1549f22ef01cSRoman Divacky   // Sniff for the signature.
1550f22ef01cSRoman Divacky   if (Stream.Read(8) != 'B' ||
1551f22ef01cSRoman Divacky       Stream.Read(8) != 'C' ||
1552f22ef01cSRoman Divacky       Stream.Read(4) != 0x0 ||
1553f22ef01cSRoman Divacky       Stream.Read(4) != 0xC ||
1554f22ef01cSRoman Divacky       Stream.Read(4) != 0xE ||
1555f22ef01cSRoman Divacky       Stream.Read(4) != 0xD)
1556f22ef01cSRoman Divacky     return Error("Invalid bitcode signature");
1557f22ef01cSRoman Divacky 
1558f22ef01cSRoman Divacky   // We expect a number of well-defined blocks, though we don't necessarily
1559f22ef01cSRoman Divacky   // need to understand them all.
1560f22ef01cSRoman Divacky   while (!Stream.AtEndOfStream()) {
1561f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
1562f22ef01cSRoman Divacky 
1563f22ef01cSRoman Divacky     if (Code != bitc::ENTER_SUBBLOCK)
1564f22ef01cSRoman Divacky       return Error("Invalid record at top-level");
1565f22ef01cSRoman Divacky 
1566f22ef01cSRoman Divacky     unsigned BlockID = Stream.ReadSubBlockID();
1567f22ef01cSRoman Divacky 
1568f22ef01cSRoman Divacky     // We only know the MODULE subblock ID.
1569f22ef01cSRoman Divacky     switch (BlockID) {
1570f22ef01cSRoman Divacky     case bitc::BLOCKINFO_BLOCK_ID:
1571f22ef01cSRoman Divacky       if (Stream.ReadBlockInfoBlock())
1572f22ef01cSRoman Divacky         return Error("Malformed BlockInfoBlock");
1573f22ef01cSRoman Divacky       break;
1574f22ef01cSRoman Divacky     case bitc::MODULE_BLOCK_ID:
1575f22ef01cSRoman Divacky       // Reject multiple MODULE_BLOCK's in a single bitstream.
1576f22ef01cSRoman Divacky       if (TheModule)
1577f22ef01cSRoman Divacky         return Error("Multiple MODULE_BLOCKs in same stream");
1578f22ef01cSRoman Divacky       TheModule = M;
1579f22ef01cSRoman Divacky       if (ParseModule())
1580f22ef01cSRoman Divacky         return true;
1581f22ef01cSRoman Divacky       break;
1582f22ef01cSRoman Divacky     default:
1583f22ef01cSRoman Divacky       if (Stream.SkipBlock())
1584f22ef01cSRoman Divacky         return Error("Malformed block record");
1585f22ef01cSRoman Divacky       break;
1586f22ef01cSRoman Divacky     }
1587f22ef01cSRoman Divacky   }
1588f22ef01cSRoman Divacky 
1589f22ef01cSRoman Divacky   return false;
1590f22ef01cSRoman Divacky }
1591f22ef01cSRoman Divacky 
1592f22ef01cSRoman Divacky /// ParseMetadataAttachment - Parse metadata attachments.
1593f22ef01cSRoman Divacky bool BitcodeReader::ParseMetadataAttachment() {
1594f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::METADATA_ATTACHMENT_ID))
1595f22ef01cSRoman Divacky     return Error("Malformed block record");
1596f22ef01cSRoman Divacky 
1597f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1598f22ef01cSRoman Divacky   while(1) {
1599f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
1600f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
1601f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
1602f22ef01cSRoman Divacky         return Error("Error at end of PARAMATTR block");
1603f22ef01cSRoman Divacky       break;
1604f22ef01cSRoman Divacky     }
1605f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
1606f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
1607f22ef01cSRoman Divacky       continue;
1608f22ef01cSRoman Divacky     }
1609f22ef01cSRoman Divacky     // Read a metadata attachment record.
1610f22ef01cSRoman Divacky     Record.clear();
1611f22ef01cSRoman Divacky     switch (Stream.ReadRecord(Code, Record)) {
1612f22ef01cSRoman Divacky     default:  // Default behavior: ignore.
1613f22ef01cSRoman Divacky       break;
1614f22ef01cSRoman Divacky     case bitc::METADATA_ATTACHMENT: {
1615f22ef01cSRoman Divacky       unsigned RecordLength = Record.size();
1616f22ef01cSRoman Divacky       if (Record.empty() || (RecordLength - 1) % 2 == 1)
1617f22ef01cSRoman Divacky         return Error ("Invalid METADATA_ATTACHMENT reader!");
1618f22ef01cSRoman Divacky       Instruction *Inst = InstructionList[Record[0]];
1619f22ef01cSRoman Divacky       for (unsigned i = 1; i != RecordLength; i = i+2) {
1620f22ef01cSRoman Divacky         unsigned Kind = Record[i];
1621f22ef01cSRoman Divacky         Value *Node = MDValueList.getValueFwdRef(Record[i+1]);
1622f22ef01cSRoman Divacky         Inst->setMetadata(Kind, cast<MDNode>(Node));
1623f22ef01cSRoman Divacky       }
1624f22ef01cSRoman Divacky       break;
1625f22ef01cSRoman Divacky     }
1626f22ef01cSRoman Divacky     }
1627f22ef01cSRoman Divacky   }
1628f22ef01cSRoman Divacky   return false;
1629f22ef01cSRoman Divacky }
1630f22ef01cSRoman Divacky 
1631f22ef01cSRoman Divacky /// ParseFunctionBody - Lazily parse the specified function body block.
1632f22ef01cSRoman Divacky bool BitcodeReader::ParseFunctionBody(Function *F) {
1633f22ef01cSRoman Divacky   if (Stream.EnterSubBlock(bitc::FUNCTION_BLOCK_ID))
1634f22ef01cSRoman Divacky     return Error("Malformed block record");
1635f22ef01cSRoman Divacky 
1636f22ef01cSRoman Divacky   InstructionList.clear();
1637f22ef01cSRoman Divacky   unsigned ModuleValueListSize = ValueList.size();
1638f22ef01cSRoman Divacky 
1639f22ef01cSRoman Divacky   // Add all the function arguments to the value table.
1640f22ef01cSRoman Divacky   for(Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
1641f22ef01cSRoman Divacky     ValueList.push_back(I);
1642f22ef01cSRoman Divacky 
1643f22ef01cSRoman Divacky   unsigned NextValueNo = ValueList.size();
1644f22ef01cSRoman Divacky   BasicBlock *CurBB = 0;
1645f22ef01cSRoman Divacky   unsigned CurBBNo = 0;
1646f22ef01cSRoman Divacky 
1647f22ef01cSRoman Divacky   DebugLoc LastLoc;
1648f22ef01cSRoman Divacky 
1649f22ef01cSRoman Divacky   // Read all the records.
1650f22ef01cSRoman Divacky   SmallVector<uint64_t, 64> Record;
1651f22ef01cSRoman Divacky   while (1) {
1652f22ef01cSRoman Divacky     unsigned Code = Stream.ReadCode();
1653f22ef01cSRoman Divacky     if (Code == bitc::END_BLOCK) {
1654f22ef01cSRoman Divacky       if (Stream.ReadBlockEnd())
1655f22ef01cSRoman Divacky         return Error("Error at end of function block");
1656f22ef01cSRoman Divacky       break;
1657f22ef01cSRoman Divacky     }
1658f22ef01cSRoman Divacky 
1659f22ef01cSRoman Divacky     if (Code == bitc::ENTER_SUBBLOCK) {
1660f22ef01cSRoman Divacky       switch (Stream.ReadSubBlockID()) {
1661f22ef01cSRoman Divacky       default:  // Skip unknown content.
1662f22ef01cSRoman Divacky         if (Stream.SkipBlock())
1663f22ef01cSRoman Divacky           return Error("Malformed block record");
1664f22ef01cSRoman Divacky         break;
1665f22ef01cSRoman Divacky       case bitc::CONSTANTS_BLOCK_ID:
1666f22ef01cSRoman Divacky         if (ParseConstants()) return true;
1667f22ef01cSRoman Divacky         NextValueNo = ValueList.size();
1668f22ef01cSRoman Divacky         break;
1669f22ef01cSRoman Divacky       case bitc::VALUE_SYMTAB_BLOCK_ID:
1670f22ef01cSRoman Divacky         if (ParseValueSymbolTable()) return true;
1671f22ef01cSRoman Divacky         break;
1672f22ef01cSRoman Divacky       case bitc::METADATA_ATTACHMENT_ID:
1673f22ef01cSRoman Divacky         if (ParseMetadataAttachment()) return true;
1674f22ef01cSRoman Divacky         break;
1675f22ef01cSRoman Divacky       case bitc::METADATA_BLOCK_ID:
1676f22ef01cSRoman Divacky         if (ParseMetadata()) return true;
1677f22ef01cSRoman Divacky         break;
1678f22ef01cSRoman Divacky       }
1679f22ef01cSRoman Divacky       continue;
1680f22ef01cSRoman Divacky     }
1681f22ef01cSRoman Divacky 
1682f22ef01cSRoman Divacky     if (Code == bitc::DEFINE_ABBREV) {
1683f22ef01cSRoman Divacky       Stream.ReadAbbrevRecord();
1684f22ef01cSRoman Divacky       continue;
1685f22ef01cSRoman Divacky     }
1686f22ef01cSRoman Divacky 
1687f22ef01cSRoman Divacky     // Read a record.
1688f22ef01cSRoman Divacky     Record.clear();
1689f22ef01cSRoman Divacky     Instruction *I = 0;
1690f22ef01cSRoman Divacky     unsigned BitCode = Stream.ReadRecord(Code, Record);
1691f22ef01cSRoman Divacky     switch (BitCode) {
1692f22ef01cSRoman Divacky     default: // Default behavior: reject
1693f22ef01cSRoman Divacky       return Error("Unknown instruction");
1694f22ef01cSRoman Divacky     case bitc::FUNC_CODE_DECLAREBLOCKS:     // DECLAREBLOCKS: [nblocks]
1695f22ef01cSRoman Divacky       if (Record.size() < 1 || Record[0] == 0)
1696f22ef01cSRoman Divacky         return Error("Invalid DECLAREBLOCKS record");
1697f22ef01cSRoman Divacky       // Create all the basic blocks for the function.
1698f22ef01cSRoman Divacky       FunctionBBs.resize(Record[0]);
1699f22ef01cSRoman Divacky       for (unsigned i = 0, e = FunctionBBs.size(); i != e; ++i)
1700f22ef01cSRoman Divacky         FunctionBBs[i] = BasicBlock::Create(Context, "", F);
1701f22ef01cSRoman Divacky       CurBB = FunctionBBs[0];
1702f22ef01cSRoman Divacky       continue;
1703f22ef01cSRoman Divacky 
1704f22ef01cSRoman Divacky 
1705f22ef01cSRoman Divacky     case bitc::FUNC_CODE_DEBUG_LOC_AGAIN:  // DEBUG_LOC_AGAIN
1706f22ef01cSRoman Divacky       // This record indicates that the last instruction is at the same
1707f22ef01cSRoman Divacky       // location as the previous instruction with a location.
1708f22ef01cSRoman Divacky       I = 0;
1709f22ef01cSRoman Divacky 
1710f22ef01cSRoman Divacky       // Get the last instruction emitted.
1711f22ef01cSRoman Divacky       if (CurBB && !CurBB->empty())
1712f22ef01cSRoman Divacky         I = &CurBB->back();
1713f22ef01cSRoman Divacky       else if (CurBBNo && FunctionBBs[CurBBNo-1] &&
1714f22ef01cSRoman Divacky                !FunctionBBs[CurBBNo-1]->empty())
1715f22ef01cSRoman Divacky         I = &FunctionBBs[CurBBNo-1]->back();
1716f22ef01cSRoman Divacky 
1717f22ef01cSRoman Divacky       if (I == 0) return Error("Invalid DEBUG_LOC_AGAIN record");
1718f22ef01cSRoman Divacky       I->setDebugLoc(LastLoc);
1719f22ef01cSRoman Divacky       I = 0;
1720f22ef01cSRoman Divacky       continue;
1721f22ef01cSRoman Divacky 
1722f22ef01cSRoman Divacky     case bitc::FUNC_CODE_DEBUG_LOC: {      // DEBUG_LOC: [line, col, scope, ia]
1723f22ef01cSRoman Divacky       I = 0;     // Get the last instruction emitted.
1724f22ef01cSRoman Divacky       if (CurBB && !CurBB->empty())
1725f22ef01cSRoman Divacky         I = &CurBB->back();
1726f22ef01cSRoman Divacky       else if (CurBBNo && FunctionBBs[CurBBNo-1] &&
1727f22ef01cSRoman Divacky                !FunctionBBs[CurBBNo-1]->empty())
1728f22ef01cSRoman Divacky         I = &FunctionBBs[CurBBNo-1]->back();
1729f22ef01cSRoman Divacky       if (I == 0 || Record.size() < 4)
1730f22ef01cSRoman Divacky         return Error("Invalid FUNC_CODE_DEBUG_LOC record");
1731f22ef01cSRoman Divacky 
1732f22ef01cSRoman Divacky       unsigned Line = Record[0], Col = Record[1];
1733f22ef01cSRoman Divacky       unsigned ScopeID = Record[2], IAID = Record[3];
1734f22ef01cSRoman Divacky 
1735f22ef01cSRoman Divacky       MDNode *Scope = 0, *IA = 0;
1736f22ef01cSRoman Divacky       if (ScopeID) Scope = cast<MDNode>(MDValueList.getValueFwdRef(ScopeID-1));
1737f22ef01cSRoman Divacky       if (IAID)    IA = cast<MDNode>(MDValueList.getValueFwdRef(IAID-1));
1738f22ef01cSRoman Divacky       LastLoc = DebugLoc::get(Line, Col, Scope, IA);
1739f22ef01cSRoman Divacky       I->setDebugLoc(LastLoc);
1740f22ef01cSRoman Divacky       I = 0;
1741f22ef01cSRoman Divacky       continue;
1742f22ef01cSRoman Divacky     }
1743f22ef01cSRoman Divacky 
1744f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_BINOP: {    // BINOP: [opval, ty, opval, opcode]
1745f22ef01cSRoman Divacky       unsigned OpNum = 0;
1746f22ef01cSRoman Divacky       Value *LHS, *RHS;
1747f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1748f22ef01cSRoman Divacky           getValue(Record, OpNum, LHS->getType(), RHS) ||
1749f22ef01cSRoman Divacky           OpNum+1 > Record.size())
1750f22ef01cSRoman Divacky         return Error("Invalid BINOP record");
1751f22ef01cSRoman Divacky 
1752f22ef01cSRoman Divacky       int Opc = GetDecodedBinaryOpcode(Record[OpNum++], LHS->getType());
1753f22ef01cSRoman Divacky       if (Opc == -1) return Error("Invalid BINOP record");
1754f22ef01cSRoman Divacky       I = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
1755f22ef01cSRoman Divacky       InstructionList.push_back(I);
1756f22ef01cSRoman Divacky       if (OpNum < Record.size()) {
1757f22ef01cSRoman Divacky         if (Opc == Instruction::Add ||
1758f22ef01cSRoman Divacky             Opc == Instruction::Sub ||
1759f22ef01cSRoman Divacky             Opc == Instruction::Mul) {
1760f22ef01cSRoman Divacky           if (Record[OpNum] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1761f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setHasNoSignedWrap(true);
1762f22ef01cSRoman Divacky           if (Record[OpNum] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1763f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setHasNoUnsignedWrap(true);
1764f22ef01cSRoman Divacky         } else if (Opc == Instruction::SDiv) {
1765f22ef01cSRoman Divacky           if (Record[OpNum] & (1 << bitc::SDIV_EXACT))
1766f22ef01cSRoman Divacky             cast<BinaryOperator>(I)->setIsExact(true);
1767f22ef01cSRoman Divacky         }
1768f22ef01cSRoman Divacky       }
1769f22ef01cSRoman Divacky       break;
1770f22ef01cSRoman Divacky     }
1771f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CAST: {    // CAST: [opval, opty, destty, castopc]
1772f22ef01cSRoman Divacky       unsigned OpNum = 0;
1773f22ef01cSRoman Divacky       Value *Op;
1774f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
1775f22ef01cSRoman Divacky           OpNum+2 != Record.size())
1776f22ef01cSRoman Divacky         return Error("Invalid CAST record");
1777f22ef01cSRoman Divacky 
1778f22ef01cSRoman Divacky       const Type *ResTy = getTypeByID(Record[OpNum]);
1779f22ef01cSRoman Divacky       int Opc = GetDecodedCastOpcode(Record[OpNum+1]);
1780f22ef01cSRoman Divacky       if (Opc == -1 || ResTy == 0)
1781f22ef01cSRoman Divacky         return Error("Invalid CAST record");
1782f22ef01cSRoman Divacky       I = CastInst::Create((Instruction::CastOps)Opc, Op, ResTy);
1783f22ef01cSRoman Divacky       InstructionList.push_back(I);
1784f22ef01cSRoman Divacky       break;
1785f22ef01cSRoman Divacky     }
1786f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INBOUNDS_GEP:
1787f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_GEP: { // GEP: [n x operands]
1788f22ef01cSRoman Divacky       unsigned OpNum = 0;
1789f22ef01cSRoman Divacky       Value *BasePtr;
1790f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, BasePtr))
1791f22ef01cSRoman Divacky         return Error("Invalid GEP record");
1792f22ef01cSRoman Divacky 
1793f22ef01cSRoman Divacky       SmallVector<Value*, 16> GEPIdx;
1794f22ef01cSRoman Divacky       while (OpNum != Record.size()) {
1795f22ef01cSRoman Divacky         Value *Op;
1796f22ef01cSRoman Divacky         if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1797f22ef01cSRoman Divacky           return Error("Invalid GEP record");
1798f22ef01cSRoman Divacky         GEPIdx.push_back(Op);
1799f22ef01cSRoman Divacky       }
1800f22ef01cSRoman Divacky 
1801f22ef01cSRoman Divacky       I = GetElementPtrInst::Create(BasePtr, GEPIdx.begin(), GEPIdx.end());
1802f22ef01cSRoman Divacky       InstructionList.push_back(I);
1803f22ef01cSRoman Divacky       if (BitCode == bitc::FUNC_CODE_INST_INBOUNDS_GEP)
1804f22ef01cSRoman Divacky         cast<GetElementPtrInst>(I)->setIsInBounds(true);
1805f22ef01cSRoman Divacky       break;
1806f22ef01cSRoman Divacky     }
1807f22ef01cSRoman Divacky 
1808f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_EXTRACTVAL: {
1809f22ef01cSRoman Divacky                                        // EXTRACTVAL: [opty, opval, n x indices]
1810f22ef01cSRoman Divacky       unsigned OpNum = 0;
1811f22ef01cSRoman Divacky       Value *Agg;
1812f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1813f22ef01cSRoman Divacky         return Error("Invalid EXTRACTVAL record");
1814f22ef01cSRoman Divacky 
1815f22ef01cSRoman Divacky       SmallVector<unsigned, 4> EXTRACTVALIdx;
1816f22ef01cSRoman Divacky       for (unsigned RecSize = Record.size();
1817f22ef01cSRoman Divacky            OpNum != RecSize; ++OpNum) {
1818f22ef01cSRoman Divacky         uint64_t Index = Record[OpNum];
1819f22ef01cSRoman Divacky         if ((unsigned)Index != Index)
1820f22ef01cSRoman Divacky           return Error("Invalid EXTRACTVAL index");
1821f22ef01cSRoman Divacky         EXTRACTVALIdx.push_back((unsigned)Index);
1822f22ef01cSRoman Divacky       }
1823f22ef01cSRoman Divacky 
1824f22ef01cSRoman Divacky       I = ExtractValueInst::Create(Agg,
1825f22ef01cSRoman Divacky                                    EXTRACTVALIdx.begin(), EXTRACTVALIdx.end());
1826f22ef01cSRoman Divacky       InstructionList.push_back(I);
1827f22ef01cSRoman Divacky       break;
1828f22ef01cSRoman Divacky     }
1829f22ef01cSRoman Divacky 
1830f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INSERTVAL: {
1831f22ef01cSRoman Divacky                            // INSERTVAL: [opty, opval, opty, opval, n x indices]
1832f22ef01cSRoman Divacky       unsigned OpNum = 0;
1833f22ef01cSRoman Divacky       Value *Agg;
1834f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1835f22ef01cSRoman Divacky         return Error("Invalid INSERTVAL record");
1836f22ef01cSRoman Divacky       Value *Val;
1837f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Val))
1838f22ef01cSRoman Divacky         return Error("Invalid INSERTVAL record");
1839f22ef01cSRoman Divacky 
1840f22ef01cSRoman Divacky       SmallVector<unsigned, 4> INSERTVALIdx;
1841f22ef01cSRoman Divacky       for (unsigned RecSize = Record.size();
1842f22ef01cSRoman Divacky            OpNum != RecSize; ++OpNum) {
1843f22ef01cSRoman Divacky         uint64_t Index = Record[OpNum];
1844f22ef01cSRoman Divacky         if ((unsigned)Index != Index)
1845f22ef01cSRoman Divacky           return Error("Invalid INSERTVAL index");
1846f22ef01cSRoman Divacky         INSERTVALIdx.push_back((unsigned)Index);
1847f22ef01cSRoman Divacky       }
1848f22ef01cSRoman Divacky 
1849f22ef01cSRoman Divacky       I = InsertValueInst::Create(Agg, Val,
1850f22ef01cSRoman Divacky                                   INSERTVALIdx.begin(), INSERTVALIdx.end());
1851f22ef01cSRoman Divacky       InstructionList.push_back(I);
1852f22ef01cSRoman Divacky       break;
1853f22ef01cSRoman Divacky     }
1854f22ef01cSRoman Divacky 
1855f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SELECT: { // SELECT: [opval, ty, opval, opval]
1856f22ef01cSRoman Divacky       // obsolete form of select
1857f22ef01cSRoman Divacky       // handles select i1 ... in old bitcode
1858f22ef01cSRoman Divacky       unsigned OpNum = 0;
1859f22ef01cSRoman Divacky       Value *TrueVal, *FalseVal, *Cond;
1860f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1861f22ef01cSRoman Divacky           getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
1862f22ef01cSRoman Divacky           getValue(Record, OpNum, Type::getInt1Ty(Context), Cond))
1863f22ef01cSRoman Divacky         return Error("Invalid SELECT record");
1864f22ef01cSRoman Divacky 
1865f22ef01cSRoman Divacky       I = SelectInst::Create(Cond, TrueVal, FalseVal);
1866f22ef01cSRoman Divacky       InstructionList.push_back(I);
1867f22ef01cSRoman Divacky       break;
1868f22ef01cSRoman Divacky     }
1869f22ef01cSRoman Divacky 
1870f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_VSELECT: {// VSELECT: [ty,opval,opval,predty,pred]
1871f22ef01cSRoman Divacky       // new form of select
1872f22ef01cSRoman Divacky       // handles select i1 or select [N x i1]
1873f22ef01cSRoman Divacky       unsigned OpNum = 0;
1874f22ef01cSRoman Divacky       Value *TrueVal, *FalseVal, *Cond;
1875f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1876f22ef01cSRoman Divacky           getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
1877f22ef01cSRoman Divacky           getValueTypePair(Record, OpNum, NextValueNo, Cond))
1878f22ef01cSRoman Divacky         return Error("Invalid SELECT record");
1879f22ef01cSRoman Divacky 
1880f22ef01cSRoman Divacky       // select condition can be either i1 or [N x i1]
1881f22ef01cSRoman Divacky       if (const VectorType* vector_type =
1882f22ef01cSRoman Divacky           dyn_cast<const VectorType>(Cond->getType())) {
1883f22ef01cSRoman Divacky         // expect <n x i1>
1884f22ef01cSRoman Divacky         if (vector_type->getElementType() != Type::getInt1Ty(Context))
1885f22ef01cSRoman Divacky           return Error("Invalid SELECT condition type");
1886f22ef01cSRoman Divacky       } else {
1887f22ef01cSRoman Divacky         // expect i1
1888f22ef01cSRoman Divacky         if (Cond->getType() != Type::getInt1Ty(Context))
1889f22ef01cSRoman Divacky           return Error("Invalid SELECT condition type");
1890f22ef01cSRoman Divacky       }
1891f22ef01cSRoman Divacky 
1892f22ef01cSRoman Divacky       I = SelectInst::Create(Cond, TrueVal, FalseVal);
1893f22ef01cSRoman Divacky       InstructionList.push_back(I);
1894f22ef01cSRoman Divacky       break;
1895f22ef01cSRoman Divacky     }
1896f22ef01cSRoman Divacky 
1897f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_EXTRACTELT: { // EXTRACTELT: [opty, opval, opval]
1898f22ef01cSRoman Divacky       unsigned OpNum = 0;
1899f22ef01cSRoman Divacky       Value *Vec, *Idx;
1900f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
1901f22ef01cSRoman Divacky           getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
1902f22ef01cSRoman Divacky         return Error("Invalid EXTRACTELT record");
1903f22ef01cSRoman Divacky       I = ExtractElementInst::Create(Vec, Idx);
1904f22ef01cSRoman Divacky       InstructionList.push_back(I);
1905f22ef01cSRoman Divacky       break;
1906f22ef01cSRoman Divacky     }
1907f22ef01cSRoman Divacky 
1908f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INSERTELT: { // INSERTELT: [ty, opval,opval,opval]
1909f22ef01cSRoman Divacky       unsigned OpNum = 0;
1910f22ef01cSRoman Divacky       Value *Vec, *Elt, *Idx;
1911f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
1912f22ef01cSRoman Divacky           getValue(Record, OpNum,
1913f22ef01cSRoman Divacky                    cast<VectorType>(Vec->getType())->getElementType(), Elt) ||
1914f22ef01cSRoman Divacky           getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
1915f22ef01cSRoman Divacky         return Error("Invalid INSERTELT record");
1916f22ef01cSRoman Divacky       I = InsertElementInst::Create(Vec, Elt, Idx);
1917f22ef01cSRoman Divacky       InstructionList.push_back(I);
1918f22ef01cSRoman Divacky       break;
1919f22ef01cSRoman Divacky     }
1920f22ef01cSRoman Divacky 
1921f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SHUFFLEVEC: {// SHUFFLEVEC: [opval,ty,opval,opval]
1922f22ef01cSRoman Divacky       unsigned OpNum = 0;
1923f22ef01cSRoman Divacky       Value *Vec1, *Vec2, *Mask;
1924f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Vec1) ||
1925f22ef01cSRoman Divacky           getValue(Record, OpNum, Vec1->getType(), Vec2))
1926f22ef01cSRoman Divacky         return Error("Invalid SHUFFLEVEC record");
1927f22ef01cSRoman Divacky 
1928f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Mask))
1929f22ef01cSRoman Divacky         return Error("Invalid SHUFFLEVEC record");
1930f22ef01cSRoman Divacky       I = new ShuffleVectorInst(Vec1, Vec2, Mask);
1931f22ef01cSRoman Divacky       InstructionList.push_back(I);
1932f22ef01cSRoman Divacky       break;
1933f22ef01cSRoman Divacky     }
1934f22ef01cSRoman Divacky 
1935f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CMP:   // CMP: [opty, opval, opval, pred]
1936f22ef01cSRoman Divacky       // Old form of ICmp/FCmp returning bool
1937f22ef01cSRoman Divacky       // Existed to differentiate between icmp/fcmp and vicmp/vfcmp which were
1938f22ef01cSRoman Divacky       // both legal on vectors but had different behaviour.
1939f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CMP2: { // CMP2: [opty, opval, opval, pred]
1940f22ef01cSRoman Divacky       // FCmp/ICmp returning bool or vector of bool
1941f22ef01cSRoman Divacky 
1942f22ef01cSRoman Divacky       unsigned OpNum = 0;
1943f22ef01cSRoman Divacky       Value *LHS, *RHS;
1944f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1945f22ef01cSRoman Divacky           getValue(Record, OpNum, LHS->getType(), RHS) ||
1946f22ef01cSRoman Divacky           OpNum+1 != Record.size())
1947f22ef01cSRoman Divacky         return Error("Invalid CMP record");
1948f22ef01cSRoman Divacky 
1949f22ef01cSRoman Divacky       if (LHS->getType()->isFPOrFPVectorTy())
1950f22ef01cSRoman Divacky         I = new FCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
1951f22ef01cSRoman Divacky       else
1952f22ef01cSRoman Divacky         I = new ICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
1953f22ef01cSRoman Divacky       InstructionList.push_back(I);
1954f22ef01cSRoman Divacky       break;
1955f22ef01cSRoman Divacky     }
1956f22ef01cSRoman Divacky 
1957f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_GETRESULT: { // GETRESULT: [ty, val, n]
1958f22ef01cSRoman Divacky       if (Record.size() != 2)
1959f22ef01cSRoman Divacky         return Error("Invalid GETRESULT record");
1960f22ef01cSRoman Divacky       unsigned OpNum = 0;
1961f22ef01cSRoman Divacky       Value *Op;
1962f22ef01cSRoman Divacky       getValueTypePair(Record, OpNum, NextValueNo, Op);
1963f22ef01cSRoman Divacky       unsigned Index = Record[1];
1964f22ef01cSRoman Divacky       I = ExtractValueInst::Create(Op, Index);
1965f22ef01cSRoman Divacky       InstructionList.push_back(I);
1966f22ef01cSRoman Divacky       break;
1967f22ef01cSRoman Divacky     }
1968f22ef01cSRoman Divacky 
1969f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_RET: // RET: [opty,opval<optional>]
1970f22ef01cSRoman Divacky       {
1971f22ef01cSRoman Divacky         unsigned Size = Record.size();
1972f22ef01cSRoman Divacky         if (Size == 0) {
1973f22ef01cSRoman Divacky           I = ReturnInst::Create(Context);
1974f22ef01cSRoman Divacky           InstructionList.push_back(I);
1975f22ef01cSRoman Divacky           break;
1976f22ef01cSRoman Divacky         }
1977f22ef01cSRoman Divacky 
1978f22ef01cSRoman Divacky         unsigned OpNum = 0;
1979f22ef01cSRoman Divacky         SmallVector<Value *,4> Vs;
1980f22ef01cSRoman Divacky         do {
1981f22ef01cSRoman Divacky           Value *Op = NULL;
1982f22ef01cSRoman Divacky           if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1983f22ef01cSRoman Divacky             return Error("Invalid RET record");
1984f22ef01cSRoman Divacky           Vs.push_back(Op);
1985f22ef01cSRoman Divacky         } while(OpNum != Record.size());
1986f22ef01cSRoman Divacky 
1987f22ef01cSRoman Divacky         const Type *ReturnType = F->getReturnType();
1988f22ef01cSRoman Divacky         if (Vs.size() > 1 ||
1989f22ef01cSRoman Divacky             (ReturnType->isStructTy() &&
1990f22ef01cSRoman Divacky              (Vs.empty() || Vs[0]->getType() != ReturnType))) {
1991f22ef01cSRoman Divacky           Value *RV = UndefValue::get(ReturnType);
1992f22ef01cSRoman Divacky           for (unsigned i = 0, e = Vs.size(); i != e; ++i) {
1993f22ef01cSRoman Divacky             I = InsertValueInst::Create(RV, Vs[i], i, "mrv");
1994f22ef01cSRoman Divacky             InstructionList.push_back(I);
1995f22ef01cSRoman Divacky             CurBB->getInstList().push_back(I);
1996f22ef01cSRoman Divacky             ValueList.AssignValue(I, NextValueNo++);
1997f22ef01cSRoman Divacky             RV = I;
1998f22ef01cSRoman Divacky           }
1999f22ef01cSRoman Divacky           I = ReturnInst::Create(Context, RV);
2000f22ef01cSRoman Divacky           InstructionList.push_back(I);
2001f22ef01cSRoman Divacky           break;
2002f22ef01cSRoman Divacky         }
2003f22ef01cSRoman Divacky 
2004f22ef01cSRoman Divacky         I = ReturnInst::Create(Context, Vs[0]);
2005f22ef01cSRoman Divacky         InstructionList.push_back(I);
2006f22ef01cSRoman Divacky         break;
2007f22ef01cSRoman Divacky       }
2008f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_BR: { // BR: [bb#, bb#, opval] or [bb#]
2009f22ef01cSRoman Divacky       if (Record.size() != 1 && Record.size() != 3)
2010f22ef01cSRoman Divacky         return Error("Invalid BR record");
2011f22ef01cSRoman Divacky       BasicBlock *TrueDest = getBasicBlock(Record[0]);
2012f22ef01cSRoman Divacky       if (TrueDest == 0)
2013f22ef01cSRoman Divacky         return Error("Invalid BR record");
2014f22ef01cSRoman Divacky 
2015f22ef01cSRoman Divacky       if (Record.size() == 1) {
2016f22ef01cSRoman Divacky         I = BranchInst::Create(TrueDest);
2017f22ef01cSRoman Divacky         InstructionList.push_back(I);
2018f22ef01cSRoman Divacky       }
2019f22ef01cSRoman Divacky       else {
2020f22ef01cSRoman Divacky         BasicBlock *FalseDest = getBasicBlock(Record[1]);
2021f22ef01cSRoman Divacky         Value *Cond = getFnValueByID(Record[2], Type::getInt1Ty(Context));
2022f22ef01cSRoman Divacky         if (FalseDest == 0 || Cond == 0)
2023f22ef01cSRoman Divacky           return Error("Invalid BR record");
2024f22ef01cSRoman Divacky         I = BranchInst::Create(TrueDest, FalseDest, Cond);
2025f22ef01cSRoman Divacky         InstructionList.push_back(I);
2026f22ef01cSRoman Divacky       }
2027f22ef01cSRoman Divacky       break;
2028f22ef01cSRoman Divacky     }
2029f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_SWITCH: { // SWITCH: [opty, op0, op1, ...]
2030f22ef01cSRoman Divacky       if (Record.size() < 3 || (Record.size() & 1) == 0)
2031f22ef01cSRoman Divacky         return Error("Invalid SWITCH record");
2032f22ef01cSRoman Divacky       const Type *OpTy = getTypeByID(Record[0]);
2033f22ef01cSRoman Divacky       Value *Cond = getFnValueByID(Record[1], OpTy);
2034f22ef01cSRoman Divacky       BasicBlock *Default = getBasicBlock(Record[2]);
2035f22ef01cSRoman Divacky       if (OpTy == 0 || Cond == 0 || Default == 0)
2036f22ef01cSRoman Divacky         return Error("Invalid SWITCH record");
2037f22ef01cSRoman Divacky       unsigned NumCases = (Record.size()-3)/2;
2038f22ef01cSRoman Divacky       SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
2039f22ef01cSRoman Divacky       InstructionList.push_back(SI);
2040f22ef01cSRoman Divacky       for (unsigned i = 0, e = NumCases; i != e; ++i) {
2041f22ef01cSRoman Divacky         ConstantInt *CaseVal =
2042f22ef01cSRoman Divacky           dyn_cast_or_null<ConstantInt>(getFnValueByID(Record[3+i*2], OpTy));
2043f22ef01cSRoman Divacky         BasicBlock *DestBB = getBasicBlock(Record[1+3+i*2]);
2044f22ef01cSRoman Divacky         if (CaseVal == 0 || DestBB == 0) {
2045f22ef01cSRoman Divacky           delete SI;
2046f22ef01cSRoman Divacky           return Error("Invalid SWITCH record!");
2047f22ef01cSRoman Divacky         }
2048f22ef01cSRoman Divacky         SI->addCase(CaseVal, DestBB);
2049f22ef01cSRoman Divacky       }
2050f22ef01cSRoman Divacky       I = SI;
2051f22ef01cSRoman Divacky       break;
2052f22ef01cSRoman Divacky     }
2053f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INDIRECTBR: { // INDIRECTBR: [opty, op0, op1, ...]
2054f22ef01cSRoman Divacky       if (Record.size() < 2)
2055f22ef01cSRoman Divacky         return Error("Invalid INDIRECTBR record");
2056f22ef01cSRoman Divacky       const Type *OpTy = getTypeByID(Record[0]);
2057f22ef01cSRoman Divacky       Value *Address = getFnValueByID(Record[1], OpTy);
2058f22ef01cSRoman Divacky       if (OpTy == 0 || Address == 0)
2059f22ef01cSRoman Divacky         return Error("Invalid INDIRECTBR record");
2060f22ef01cSRoman Divacky       unsigned NumDests = Record.size()-2;
2061f22ef01cSRoman Divacky       IndirectBrInst *IBI = IndirectBrInst::Create(Address, NumDests);
2062f22ef01cSRoman Divacky       InstructionList.push_back(IBI);
2063f22ef01cSRoman Divacky       for (unsigned i = 0, e = NumDests; i != e; ++i) {
2064f22ef01cSRoman Divacky         if (BasicBlock *DestBB = getBasicBlock(Record[2+i])) {
2065f22ef01cSRoman Divacky           IBI->addDestination(DestBB);
2066f22ef01cSRoman Divacky         } else {
2067f22ef01cSRoman Divacky           delete IBI;
2068f22ef01cSRoman Divacky           return Error("Invalid INDIRECTBR record!");
2069f22ef01cSRoman Divacky         }
2070f22ef01cSRoman Divacky       }
2071f22ef01cSRoman Divacky       I = IBI;
2072f22ef01cSRoman Divacky       break;
2073f22ef01cSRoman Divacky     }
2074f22ef01cSRoman Divacky 
2075f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_INVOKE: {
2076f22ef01cSRoman Divacky       // INVOKE: [attrs, cc, normBB, unwindBB, fnty, op0,op1,op2, ...]
2077f22ef01cSRoman Divacky       if (Record.size() < 4) return Error("Invalid INVOKE record");
2078f22ef01cSRoman Divacky       AttrListPtr PAL = getAttributes(Record[0]);
2079f22ef01cSRoman Divacky       unsigned CCInfo = Record[1];
2080f22ef01cSRoman Divacky       BasicBlock *NormalBB = getBasicBlock(Record[2]);
2081f22ef01cSRoman Divacky       BasicBlock *UnwindBB = getBasicBlock(Record[3]);
2082f22ef01cSRoman Divacky 
2083f22ef01cSRoman Divacky       unsigned OpNum = 4;
2084f22ef01cSRoman Divacky       Value *Callee;
2085f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
2086f22ef01cSRoman Divacky         return Error("Invalid INVOKE record");
2087f22ef01cSRoman Divacky 
2088f22ef01cSRoman Divacky       const PointerType *CalleeTy = dyn_cast<PointerType>(Callee->getType());
2089f22ef01cSRoman Divacky       const FunctionType *FTy = !CalleeTy ? 0 :
2090f22ef01cSRoman Divacky         dyn_cast<FunctionType>(CalleeTy->getElementType());
2091f22ef01cSRoman Divacky 
2092f22ef01cSRoman Divacky       // Check that the right number of fixed parameters are here.
2093f22ef01cSRoman Divacky       if (FTy == 0 || NormalBB == 0 || UnwindBB == 0 ||
2094f22ef01cSRoman Divacky           Record.size() < OpNum+FTy->getNumParams())
2095f22ef01cSRoman Divacky         return Error("Invalid INVOKE record");
2096f22ef01cSRoman Divacky 
2097f22ef01cSRoman Divacky       SmallVector<Value*, 16> Ops;
2098f22ef01cSRoman Divacky       for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
2099f22ef01cSRoman Divacky         Ops.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
2100f22ef01cSRoman Divacky         if (Ops.back() == 0) return Error("Invalid INVOKE record");
2101f22ef01cSRoman Divacky       }
2102f22ef01cSRoman Divacky 
2103f22ef01cSRoman Divacky       if (!FTy->isVarArg()) {
2104f22ef01cSRoman Divacky         if (Record.size() != OpNum)
2105f22ef01cSRoman Divacky           return Error("Invalid INVOKE record");
2106f22ef01cSRoman Divacky       } else {
2107f22ef01cSRoman Divacky         // Read type/value pairs for varargs params.
2108f22ef01cSRoman Divacky         while (OpNum != Record.size()) {
2109f22ef01cSRoman Divacky           Value *Op;
2110f22ef01cSRoman Divacky           if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2111f22ef01cSRoman Divacky             return Error("Invalid INVOKE record");
2112f22ef01cSRoman Divacky           Ops.push_back(Op);
2113f22ef01cSRoman Divacky         }
2114f22ef01cSRoman Divacky       }
2115f22ef01cSRoman Divacky 
2116f22ef01cSRoman Divacky       I = InvokeInst::Create(Callee, NormalBB, UnwindBB,
2117f22ef01cSRoman Divacky                              Ops.begin(), Ops.end());
2118f22ef01cSRoman Divacky       InstructionList.push_back(I);
2119f22ef01cSRoman Divacky       cast<InvokeInst>(I)->setCallingConv(
2120f22ef01cSRoman Divacky         static_cast<CallingConv::ID>(CCInfo));
2121f22ef01cSRoman Divacky       cast<InvokeInst>(I)->setAttributes(PAL);
2122f22ef01cSRoman Divacky       break;
2123f22ef01cSRoman Divacky     }
2124f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_UNWIND: // UNWIND
2125f22ef01cSRoman Divacky       I = new UnwindInst(Context);
2126f22ef01cSRoman Divacky       InstructionList.push_back(I);
2127f22ef01cSRoman Divacky       break;
2128f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_UNREACHABLE: // UNREACHABLE
2129f22ef01cSRoman Divacky       I = new UnreachableInst(Context);
2130f22ef01cSRoman Divacky       InstructionList.push_back(I);
2131f22ef01cSRoman Divacky       break;
2132f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_PHI: { // PHI: [ty, val0,bb0, ...]
2133f22ef01cSRoman Divacky       if (Record.size() < 1 || ((Record.size()-1)&1))
2134f22ef01cSRoman Divacky         return Error("Invalid PHI record");
2135f22ef01cSRoman Divacky       const Type *Ty = getTypeByID(Record[0]);
2136f22ef01cSRoman Divacky       if (!Ty) return Error("Invalid PHI record");
2137f22ef01cSRoman Divacky 
2138f22ef01cSRoman Divacky       PHINode *PN = PHINode::Create(Ty);
2139f22ef01cSRoman Divacky       InstructionList.push_back(PN);
2140f22ef01cSRoman Divacky       PN->reserveOperandSpace((Record.size()-1)/2);
2141f22ef01cSRoman Divacky 
2142f22ef01cSRoman Divacky       for (unsigned i = 0, e = Record.size()-1; i != e; i += 2) {
2143f22ef01cSRoman Divacky         Value *V = getFnValueByID(Record[1+i], Ty);
2144f22ef01cSRoman Divacky         BasicBlock *BB = getBasicBlock(Record[2+i]);
2145f22ef01cSRoman Divacky         if (!V || !BB) return Error("Invalid PHI record");
2146f22ef01cSRoman Divacky         PN->addIncoming(V, BB);
2147f22ef01cSRoman Divacky       }
2148f22ef01cSRoman Divacky       I = PN;
2149f22ef01cSRoman Divacky       break;
2150f22ef01cSRoman Divacky     }
2151f22ef01cSRoman Divacky 
2152f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_MALLOC: { // MALLOC: [instty, op, align]
2153f22ef01cSRoman Divacky       // Autoupgrade malloc instruction to malloc call.
2154f22ef01cSRoman Divacky       // FIXME: Remove in LLVM 3.0.
2155f22ef01cSRoman Divacky       if (Record.size() < 3)
2156f22ef01cSRoman Divacky         return Error("Invalid MALLOC record");
2157f22ef01cSRoman Divacky       const PointerType *Ty =
2158f22ef01cSRoman Divacky         dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
2159f22ef01cSRoman Divacky       Value *Size = getFnValueByID(Record[1], Type::getInt32Ty(Context));
2160f22ef01cSRoman Divacky       if (!Ty || !Size) return Error("Invalid MALLOC record");
2161f22ef01cSRoman Divacky       if (!CurBB) return Error("Invalid malloc instruction with no BB");
2162f22ef01cSRoman Divacky       const Type *Int32Ty = IntegerType::getInt32Ty(CurBB->getContext());
2163f22ef01cSRoman Divacky       Constant *AllocSize = ConstantExpr::getSizeOf(Ty->getElementType());
2164f22ef01cSRoman Divacky       AllocSize = ConstantExpr::getTruncOrBitCast(AllocSize, Int32Ty);
2165f22ef01cSRoman Divacky       I = CallInst::CreateMalloc(CurBB, Int32Ty, Ty->getElementType(),
2166f22ef01cSRoman Divacky                                  AllocSize, Size, NULL);
2167f22ef01cSRoman Divacky       InstructionList.push_back(I);
2168f22ef01cSRoman Divacky       break;
2169f22ef01cSRoman Divacky     }
2170f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_FREE: { // FREE: [op, opty]
2171f22ef01cSRoman Divacky       unsigned OpNum = 0;
2172f22ef01cSRoman Divacky       Value *Op;
2173f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2174f22ef01cSRoman Divacky           OpNum != Record.size())
2175f22ef01cSRoman Divacky         return Error("Invalid FREE record");
2176f22ef01cSRoman Divacky       if (!CurBB) return Error("Invalid free instruction with no BB");
2177f22ef01cSRoman Divacky       I = CallInst::CreateFree(Op, CurBB);
2178f22ef01cSRoman Divacky       InstructionList.push_back(I);
2179f22ef01cSRoman Divacky       break;
2180f22ef01cSRoman Divacky     }
2181f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_ALLOCA: { // ALLOCA: [instty, op, align]
2182f22ef01cSRoman Divacky       if (Record.size() < 3)
2183f22ef01cSRoman Divacky         return Error("Invalid ALLOCA record");
2184f22ef01cSRoman Divacky       const PointerType *Ty =
2185f22ef01cSRoman Divacky         dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
2186f22ef01cSRoman Divacky       Value *Size = getFnValueByID(Record[1], Type::getInt32Ty(Context));
2187f22ef01cSRoman Divacky       unsigned Align = Record[2];
2188f22ef01cSRoman Divacky       if (!Ty || !Size) return Error("Invalid ALLOCA record");
2189f22ef01cSRoman Divacky       I = new AllocaInst(Ty->getElementType(), Size, (1 << Align) >> 1);
2190f22ef01cSRoman Divacky       InstructionList.push_back(I);
2191f22ef01cSRoman Divacky       break;
2192f22ef01cSRoman Divacky     }
2193f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_LOAD: { // LOAD: [opty, op, align, vol]
2194f22ef01cSRoman Divacky       unsigned OpNum = 0;
2195f22ef01cSRoman Divacky       Value *Op;
2196f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2197f22ef01cSRoman Divacky           OpNum+2 != Record.size())
2198f22ef01cSRoman Divacky         return Error("Invalid LOAD record");
2199f22ef01cSRoman Divacky 
2200f22ef01cSRoman Divacky       I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1);
2201f22ef01cSRoman Divacky       InstructionList.push_back(I);
2202f22ef01cSRoman Divacky       break;
2203f22ef01cSRoman Divacky     }
2204f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_STORE2: { // STORE2:[ptrty, ptr, val, align, vol]
2205f22ef01cSRoman Divacky       unsigned OpNum = 0;
2206f22ef01cSRoman Divacky       Value *Val, *Ptr;
2207f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
2208f22ef01cSRoman Divacky           getValue(Record, OpNum,
2209f22ef01cSRoman Divacky                     cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
2210f22ef01cSRoman Divacky           OpNum+2 != Record.size())
2211f22ef01cSRoman Divacky         return Error("Invalid STORE record");
2212f22ef01cSRoman Divacky 
2213f22ef01cSRoman Divacky       I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
2214f22ef01cSRoman Divacky       InstructionList.push_back(I);
2215f22ef01cSRoman Divacky       break;
2216f22ef01cSRoman Divacky     }
2217f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_STORE: { // STORE:[val, valty, ptr, align, vol]
2218f22ef01cSRoman Divacky       // FIXME: Legacy form of store instruction. Should be removed in LLVM 3.0.
2219f22ef01cSRoman Divacky       unsigned OpNum = 0;
2220f22ef01cSRoman Divacky       Value *Val, *Ptr;
2221f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Val) ||
2222f22ef01cSRoman Divacky           getValue(Record, OpNum,
2223f22ef01cSRoman Divacky                    PointerType::getUnqual(Val->getType()), Ptr)||
2224f22ef01cSRoman Divacky           OpNum+2 != Record.size())
2225f22ef01cSRoman Divacky         return Error("Invalid STORE record");
2226f22ef01cSRoman Divacky 
2227f22ef01cSRoman Divacky       I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
2228f22ef01cSRoman Divacky       InstructionList.push_back(I);
2229f22ef01cSRoman Divacky       break;
2230f22ef01cSRoman Divacky     }
2231f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_CALL: {
2232f22ef01cSRoman Divacky       // CALL: [paramattrs, cc, fnty, fnid, arg0, arg1...]
2233f22ef01cSRoman Divacky       if (Record.size() < 3)
2234f22ef01cSRoman Divacky         return Error("Invalid CALL record");
2235f22ef01cSRoman Divacky 
2236f22ef01cSRoman Divacky       AttrListPtr PAL = getAttributes(Record[0]);
2237f22ef01cSRoman Divacky       unsigned CCInfo = Record[1];
2238f22ef01cSRoman Divacky 
2239f22ef01cSRoman Divacky       unsigned OpNum = 2;
2240f22ef01cSRoman Divacky       Value *Callee;
2241f22ef01cSRoman Divacky       if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
2242f22ef01cSRoman Divacky         return Error("Invalid CALL record");
2243f22ef01cSRoman Divacky 
2244f22ef01cSRoman Divacky       const PointerType *OpTy = dyn_cast<PointerType>(Callee->getType());
2245f22ef01cSRoman Divacky       const FunctionType *FTy = 0;
2246f22ef01cSRoman Divacky       if (OpTy) FTy = dyn_cast<FunctionType>(OpTy->getElementType());
2247f22ef01cSRoman Divacky       if (!FTy || Record.size() < FTy->getNumParams()+OpNum)
2248f22ef01cSRoman Divacky         return Error("Invalid CALL record");
2249f22ef01cSRoman Divacky 
2250f22ef01cSRoman Divacky       SmallVector<Value*, 16> Args;
2251f22ef01cSRoman Divacky       // Read the fixed params.
2252f22ef01cSRoman Divacky       for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
2253f22ef01cSRoman Divacky         if (FTy->getParamType(i)->getTypeID()==Type::LabelTyID)
2254f22ef01cSRoman Divacky           Args.push_back(getBasicBlock(Record[OpNum]));
2255f22ef01cSRoman Divacky         else
2256f22ef01cSRoman Divacky           Args.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
2257f22ef01cSRoman Divacky         if (Args.back() == 0) return Error("Invalid CALL record");
2258f22ef01cSRoman Divacky       }
2259f22ef01cSRoman Divacky 
2260f22ef01cSRoman Divacky       // Read type/value pairs for varargs params.
2261f22ef01cSRoman Divacky       if (!FTy->isVarArg()) {
2262f22ef01cSRoman Divacky         if (OpNum != Record.size())
2263f22ef01cSRoman Divacky           return Error("Invalid CALL record");
2264f22ef01cSRoman Divacky       } else {
2265f22ef01cSRoman Divacky         while (OpNum != Record.size()) {
2266f22ef01cSRoman Divacky           Value *Op;
2267f22ef01cSRoman Divacky           if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2268f22ef01cSRoman Divacky             return Error("Invalid CALL record");
2269f22ef01cSRoman Divacky           Args.push_back(Op);
2270f22ef01cSRoman Divacky         }
2271f22ef01cSRoman Divacky       }
2272f22ef01cSRoman Divacky 
2273f22ef01cSRoman Divacky       I = CallInst::Create(Callee, Args.begin(), Args.end());
2274f22ef01cSRoman Divacky       InstructionList.push_back(I);
2275f22ef01cSRoman Divacky       cast<CallInst>(I)->setCallingConv(
2276f22ef01cSRoman Divacky         static_cast<CallingConv::ID>(CCInfo>>1));
2277f22ef01cSRoman Divacky       cast<CallInst>(I)->setTailCall(CCInfo & 1);
2278f22ef01cSRoman Divacky       cast<CallInst>(I)->setAttributes(PAL);
2279f22ef01cSRoman Divacky       break;
2280f22ef01cSRoman Divacky     }
2281f22ef01cSRoman Divacky     case bitc::FUNC_CODE_INST_VAARG: { // VAARG: [valistty, valist, instty]
2282f22ef01cSRoman Divacky       if (Record.size() < 3)
2283f22ef01cSRoman Divacky         return Error("Invalid VAARG record");
2284f22ef01cSRoman Divacky       const Type *OpTy = getTypeByID(Record[0]);
2285f22ef01cSRoman Divacky       Value *Op = getFnValueByID(Record[1], OpTy);
2286f22ef01cSRoman Divacky       const Type *ResTy = getTypeByID(Record[2]);
2287f22ef01cSRoman Divacky       if (!OpTy || !Op || !ResTy)
2288f22ef01cSRoman Divacky         return Error("Invalid VAARG record");
2289f22ef01cSRoman Divacky       I = new VAArgInst(Op, ResTy);
2290f22ef01cSRoman Divacky       InstructionList.push_back(I);
2291f22ef01cSRoman Divacky       break;
2292f22ef01cSRoman Divacky     }
2293f22ef01cSRoman Divacky     }
2294f22ef01cSRoman Divacky 
2295f22ef01cSRoman Divacky     // Add instruction to end of current BB.  If there is no current BB, reject
2296f22ef01cSRoman Divacky     // this file.
2297f22ef01cSRoman Divacky     if (CurBB == 0) {
2298f22ef01cSRoman Divacky       delete I;
2299f22ef01cSRoman Divacky       return Error("Invalid instruction with no BB");
2300f22ef01cSRoman Divacky     }
2301f22ef01cSRoman Divacky     CurBB->getInstList().push_back(I);
2302f22ef01cSRoman Divacky 
2303f22ef01cSRoman Divacky     // If this was a terminator instruction, move to the next block.
2304f22ef01cSRoman Divacky     if (isa<TerminatorInst>(I)) {
2305f22ef01cSRoman Divacky       ++CurBBNo;
2306f22ef01cSRoman Divacky       CurBB = CurBBNo < FunctionBBs.size() ? FunctionBBs[CurBBNo] : 0;
2307f22ef01cSRoman Divacky     }
2308f22ef01cSRoman Divacky 
2309f22ef01cSRoman Divacky     // Non-void values get registered in the value table for future use.
2310f22ef01cSRoman Divacky     if (I && !I->getType()->isVoidTy())
2311f22ef01cSRoman Divacky       ValueList.AssignValue(I, NextValueNo++);
2312f22ef01cSRoman Divacky   }
2313f22ef01cSRoman Divacky 
2314f22ef01cSRoman Divacky   // Check the function list for unresolved values.
2315f22ef01cSRoman Divacky   if (Argument *A = dyn_cast<Argument>(ValueList.back())) {
2316f22ef01cSRoman Divacky     if (A->getParent() == 0) {
2317f22ef01cSRoman Divacky       // We found at least one unresolved value.  Nuke them all to avoid leaks.
2318f22ef01cSRoman Divacky       for (unsigned i = ModuleValueListSize, e = ValueList.size(); i != e; ++i){
2319f22ef01cSRoman Divacky         if ((A = dyn_cast<Argument>(ValueList.back())) && A->getParent() == 0) {
2320f22ef01cSRoman Divacky           A->replaceAllUsesWith(UndefValue::get(A->getType()));
2321f22ef01cSRoman Divacky           delete A;
2322f22ef01cSRoman Divacky         }
2323f22ef01cSRoman Divacky       }
2324f22ef01cSRoman Divacky       return Error("Never resolved value found in function!");
2325f22ef01cSRoman Divacky     }
2326f22ef01cSRoman Divacky   }
2327f22ef01cSRoman Divacky 
2328f22ef01cSRoman Divacky   // See if anything took the address of blocks in this function.  If so,
2329f22ef01cSRoman Divacky   // resolve them now.
2330f22ef01cSRoman Divacky   DenseMap<Function*, std::vector<BlockAddrRefTy> >::iterator BAFRI =
2331f22ef01cSRoman Divacky     BlockAddrFwdRefs.find(F);
2332f22ef01cSRoman Divacky   if (BAFRI != BlockAddrFwdRefs.end()) {
2333f22ef01cSRoman Divacky     std::vector<BlockAddrRefTy> &RefList = BAFRI->second;
2334f22ef01cSRoman Divacky     for (unsigned i = 0, e = RefList.size(); i != e; ++i) {
2335f22ef01cSRoman Divacky       unsigned BlockIdx = RefList[i].first;
2336f22ef01cSRoman Divacky       if (BlockIdx >= FunctionBBs.size())
2337f22ef01cSRoman Divacky         return Error("Invalid blockaddress block #");
2338f22ef01cSRoman Divacky 
2339f22ef01cSRoman Divacky       GlobalVariable *FwdRef = RefList[i].second;
2340f22ef01cSRoman Divacky       FwdRef->replaceAllUsesWith(BlockAddress::get(F, FunctionBBs[BlockIdx]));
2341f22ef01cSRoman Divacky       FwdRef->eraseFromParent();
2342f22ef01cSRoman Divacky     }
2343f22ef01cSRoman Divacky 
2344f22ef01cSRoman Divacky     BlockAddrFwdRefs.erase(BAFRI);
2345f22ef01cSRoman Divacky   }
2346f22ef01cSRoman Divacky 
2347f22ef01cSRoman Divacky   // Trim the value list down to the size it was before we parsed this function.
2348f22ef01cSRoman Divacky   ValueList.shrinkTo(ModuleValueListSize);
2349f22ef01cSRoman Divacky   std::vector<BasicBlock*>().swap(FunctionBBs);
2350f22ef01cSRoman Divacky 
2351f22ef01cSRoman Divacky   return false;
2352f22ef01cSRoman Divacky }
2353f22ef01cSRoman Divacky 
2354f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
2355f22ef01cSRoman Divacky // GVMaterializer implementation
2356f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
2357f22ef01cSRoman Divacky 
2358f22ef01cSRoman Divacky 
2359f22ef01cSRoman Divacky bool BitcodeReader::isMaterializable(const GlobalValue *GV) const {
2360f22ef01cSRoman Divacky   if (const Function *F = dyn_cast<Function>(GV)) {
2361f22ef01cSRoman Divacky     return F->isDeclaration() &&
2362f22ef01cSRoman Divacky       DeferredFunctionInfo.count(const_cast<Function*>(F));
2363f22ef01cSRoman Divacky   }
2364f22ef01cSRoman Divacky   return false;
2365f22ef01cSRoman Divacky }
2366f22ef01cSRoman Divacky 
2367f22ef01cSRoman Divacky bool BitcodeReader::Materialize(GlobalValue *GV, std::string *ErrInfo) {
2368f22ef01cSRoman Divacky   Function *F = dyn_cast<Function>(GV);
2369f22ef01cSRoman Divacky   // If it's not a function or is already material, ignore the request.
2370f22ef01cSRoman Divacky   if (!F || !F->isMaterializable()) return false;
2371f22ef01cSRoman Divacky 
2372f22ef01cSRoman Divacky   DenseMap<Function*, uint64_t>::iterator DFII = DeferredFunctionInfo.find(F);
2373f22ef01cSRoman Divacky   assert(DFII != DeferredFunctionInfo.end() && "Deferred function not found!");
2374f22ef01cSRoman Divacky 
2375f22ef01cSRoman Divacky   // Move the bit stream to the saved position of the deferred function body.
2376f22ef01cSRoman Divacky   Stream.JumpToBit(DFII->second);
2377f22ef01cSRoman Divacky 
2378f22ef01cSRoman Divacky   if (ParseFunctionBody(F)) {
2379f22ef01cSRoman Divacky     if (ErrInfo) *ErrInfo = ErrorString;
2380f22ef01cSRoman Divacky     return true;
2381f22ef01cSRoman Divacky   }
2382f22ef01cSRoman Divacky 
2383f22ef01cSRoman Divacky   // Upgrade any old intrinsic calls in the function.
2384f22ef01cSRoman Divacky   for (UpgradedIntrinsicMap::iterator I = UpgradedIntrinsics.begin(),
2385f22ef01cSRoman Divacky        E = UpgradedIntrinsics.end(); I != E; ++I) {
2386f22ef01cSRoman Divacky     if (I->first != I->second) {
2387f22ef01cSRoman Divacky       for (Value::use_iterator UI = I->first->use_begin(),
2388f22ef01cSRoman Divacky            UE = I->first->use_end(); UI != UE; ) {
2389f22ef01cSRoman Divacky         if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2390f22ef01cSRoman Divacky           UpgradeIntrinsicCall(CI, I->second);
2391f22ef01cSRoman Divacky       }
2392f22ef01cSRoman Divacky     }
2393f22ef01cSRoman Divacky   }
2394f22ef01cSRoman Divacky 
2395f22ef01cSRoman Divacky   return false;
2396f22ef01cSRoman Divacky }
2397f22ef01cSRoman Divacky 
2398f22ef01cSRoman Divacky bool BitcodeReader::isDematerializable(const GlobalValue *GV) const {
2399f22ef01cSRoman Divacky   const Function *F = dyn_cast<Function>(GV);
2400f22ef01cSRoman Divacky   if (!F || F->isDeclaration())
2401f22ef01cSRoman Divacky     return false;
2402f22ef01cSRoman Divacky   return DeferredFunctionInfo.count(const_cast<Function*>(F));
2403f22ef01cSRoman Divacky }
2404f22ef01cSRoman Divacky 
2405f22ef01cSRoman Divacky void BitcodeReader::Dematerialize(GlobalValue *GV) {
2406f22ef01cSRoman Divacky   Function *F = dyn_cast<Function>(GV);
2407f22ef01cSRoman Divacky   // If this function isn't dematerializable, this is a noop.
2408f22ef01cSRoman Divacky   if (!F || !isDematerializable(F))
2409f22ef01cSRoman Divacky     return;
2410f22ef01cSRoman Divacky 
2411f22ef01cSRoman Divacky   assert(DeferredFunctionInfo.count(F) && "No info to read function later?");
2412f22ef01cSRoman Divacky 
2413f22ef01cSRoman Divacky   // Just forget the function body, we can remat it later.
2414f22ef01cSRoman Divacky   F->deleteBody();
2415f22ef01cSRoman Divacky }
2416f22ef01cSRoman Divacky 
2417f22ef01cSRoman Divacky 
2418f22ef01cSRoman Divacky bool BitcodeReader::MaterializeModule(Module *M, std::string *ErrInfo) {
2419f22ef01cSRoman Divacky   assert(M == TheModule &&
2420f22ef01cSRoman Divacky          "Can only Materialize the Module this BitcodeReader is attached to.");
2421f22ef01cSRoman Divacky   // Iterate over the module, deserializing any functions that are still on
2422f22ef01cSRoman Divacky   // disk.
2423f22ef01cSRoman Divacky   for (Module::iterator F = TheModule->begin(), E = TheModule->end();
2424f22ef01cSRoman Divacky        F != E; ++F)
2425f22ef01cSRoman Divacky     if (F->isMaterializable() &&
2426f22ef01cSRoman Divacky         Materialize(F, ErrInfo))
2427f22ef01cSRoman Divacky       return true;
2428f22ef01cSRoman Divacky 
2429f22ef01cSRoman Divacky   // Upgrade any intrinsic calls that slipped through (should not happen!) and
2430f22ef01cSRoman Divacky   // delete the old functions to clean up. We can't do this unless the entire
2431f22ef01cSRoman Divacky   // module is materialized because there could always be another function body
2432f22ef01cSRoman Divacky   // with calls to the old function.
2433f22ef01cSRoman Divacky   for (std::vector<std::pair<Function*, Function*> >::iterator I =
2434f22ef01cSRoman Divacky        UpgradedIntrinsics.begin(), E = UpgradedIntrinsics.end(); I != E; ++I) {
2435f22ef01cSRoman Divacky     if (I->first != I->second) {
2436f22ef01cSRoman Divacky       for (Value::use_iterator UI = I->first->use_begin(),
2437f22ef01cSRoman Divacky            UE = I->first->use_end(); UI != UE; ) {
2438f22ef01cSRoman Divacky         if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2439f22ef01cSRoman Divacky           UpgradeIntrinsicCall(CI, I->second);
2440f22ef01cSRoman Divacky       }
2441f22ef01cSRoman Divacky       if (!I->first->use_empty())
2442f22ef01cSRoman Divacky         I->first->replaceAllUsesWith(I->second);
2443f22ef01cSRoman Divacky       I->first->eraseFromParent();
2444f22ef01cSRoman Divacky     }
2445f22ef01cSRoman Divacky   }
2446f22ef01cSRoman Divacky   std::vector<std::pair<Function*, Function*> >().swap(UpgradedIntrinsics);
2447f22ef01cSRoman Divacky 
2448f22ef01cSRoman Divacky   // Check debug info intrinsics.
2449f22ef01cSRoman Divacky   CheckDebugInfoIntrinsics(TheModule);
2450f22ef01cSRoman Divacky 
2451f22ef01cSRoman Divacky   return false;
2452f22ef01cSRoman Divacky }
2453f22ef01cSRoman Divacky 
2454f22ef01cSRoman Divacky 
2455f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
2456f22ef01cSRoman Divacky // External interface
2457f22ef01cSRoman Divacky //===----------------------------------------------------------------------===//
2458f22ef01cSRoman Divacky 
2459f22ef01cSRoman Divacky /// getLazyBitcodeModule - lazy function-at-a-time loading from a file.
2460f22ef01cSRoman Divacky ///
2461f22ef01cSRoman Divacky Module *llvm::getLazyBitcodeModule(MemoryBuffer *Buffer,
2462f22ef01cSRoman Divacky                                    LLVMContext& Context,
2463f22ef01cSRoman Divacky                                    std::string *ErrMsg) {
2464f22ef01cSRoman Divacky   Module *M = new Module(Buffer->getBufferIdentifier(), Context);
2465f22ef01cSRoman Divacky   BitcodeReader *R = new BitcodeReader(Buffer, Context);
2466f22ef01cSRoman Divacky   M->setMaterializer(R);
2467f22ef01cSRoman Divacky   if (R->ParseBitcodeInto(M)) {
2468f22ef01cSRoman Divacky     if (ErrMsg)
2469f22ef01cSRoman Divacky       *ErrMsg = R->getErrorString();
2470f22ef01cSRoman Divacky 
2471f22ef01cSRoman Divacky     delete M;  // Also deletes R.
2472f22ef01cSRoman Divacky     return 0;
2473f22ef01cSRoman Divacky   }
2474f22ef01cSRoman Divacky   // Have the BitcodeReader dtor delete 'Buffer'.
2475f22ef01cSRoman Divacky   R->setBufferOwned(true);
2476f22ef01cSRoman Divacky   return M;
2477f22ef01cSRoman Divacky }
2478f22ef01cSRoman Divacky 
2479f22ef01cSRoman Divacky /// ParseBitcodeFile - Read the specified bitcode file, returning the module.
2480f22ef01cSRoman Divacky /// If an error occurs, return null and fill in *ErrMsg if non-null.
2481f22ef01cSRoman Divacky Module *llvm::ParseBitcodeFile(MemoryBuffer *Buffer, LLVMContext& Context,
2482f22ef01cSRoman Divacky                                std::string *ErrMsg){
2483f22ef01cSRoman Divacky   Module *M = getLazyBitcodeModule(Buffer, Context, ErrMsg);
2484f22ef01cSRoman Divacky   if (!M) return 0;
2485f22ef01cSRoman Divacky 
2486f22ef01cSRoman Divacky   // Don't let the BitcodeReader dtor delete 'Buffer', regardless of whether
2487f22ef01cSRoman Divacky   // there was an error.
2488f22ef01cSRoman Divacky   static_cast<BitcodeReader*>(M->getMaterializer())->setBufferOwned(false);
2489f22ef01cSRoman Divacky 
2490f22ef01cSRoman Divacky   // Read in the entire module, and destroy the BitcodeReader.
2491f22ef01cSRoman Divacky   if (M->MaterializeAllPermanently(ErrMsg)) {
2492f22ef01cSRoman Divacky     delete M;
2493f22ef01cSRoman Divacky     return NULL;
2494f22ef01cSRoman Divacky   }
2495f22ef01cSRoman Divacky   return M;
2496f22ef01cSRoman Divacky }
2497