blob: f3ab806271a3068dbf2015e1651613fafbe5a25e [file] [log] [blame]
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001//===- BitcodeReader.cpp - Internal BitcodeReader implementation ----------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner081ce942007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Dan Gohmanf17a25c2007-07-18 16:29:46 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This header defines the BitcodeReader class.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/Bitcode/ReaderWriter.h"
15#include "BitcodeReader.h"
16#include "llvm/Constants.h"
17#include "llvm/DerivedTypes.h"
18#include "llvm/InlineAsm.h"
Devang Patel15e723d2009-08-28 23:24:31 +000019#include "llvm/IntrinsicInst.h"
Owen Anderson4956cde2009-07-07 20:18:58 +000020#include "llvm/LLVMContext.h"
Devang Patel7c368852009-07-28 21:49:47 +000021#include "llvm/Metadata.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000022#include "llvm/Module.h"
Dan Gohman799a8c82009-07-20 21:19:07 +000023#include "llvm/Operator.h"
Chandler Carrutha228e392007-08-04 01:51:18 +000024#include "llvm/AutoUpgrade.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000025#include "llvm/ADT/SmallString.h"
Devang Patel57652392008-02-26 19:38:17 +000026#include "llvm/ADT/SmallVector.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000027#include "llvm/Support/MathExtras.h"
28#include "llvm/Support/MemoryBuffer.h"
Gabor Greif25bc3222008-05-10 08:32:32 +000029#include "llvm/OperandTraits.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000030using namespace llvm;
31
32void BitcodeReader::FreeState() {
33 delete Buffer;
34 Buffer = 0;
35 std::vector<PATypeHolder>().swap(TypeList);
36 ValueList.clear();
Devang Patelea27c232009-08-04 06:00:18 +000037 MDValueList.clear();
Daniel Dunbar3be44e62009-09-20 02:20:51 +000038
Devang Patelf2a4a922008-09-26 22:53:05 +000039 std::vector<AttrListPtr>().swap(MAttributes);
Dan Gohmanf17a25c2007-07-18 16:29:46 +000040 std::vector<BasicBlock*>().swap(FunctionBBs);
41 std::vector<Function*>().swap(FunctionsWithBodies);
42 DeferredFunctionInfo.clear();
43}
44
45//===----------------------------------------------------------------------===//
46// Helper functions to implement forward reference resolution, etc.
47//===----------------------------------------------------------------------===//
48
49/// ConvertToString - Convert a string from a record into an std::string, return
50/// true on failure.
51template<typename StrTy>
52static bool ConvertToString(SmallVector<uint64_t, 64> &Record, unsigned Idx,
53 StrTy &Result) {
54 if (Idx > Record.size())
55 return true;
Daniel Dunbar3be44e62009-09-20 02:20:51 +000056
Dan Gohmanf17a25c2007-07-18 16:29:46 +000057 for (unsigned i = Idx, e = Record.size(); i != e; ++i)
58 Result += (char)Record[i];
59 return false;
60}
61
62static GlobalValue::LinkageTypes GetDecodedLinkage(unsigned Val) {
63 switch (Val) {
64 default: // Map unknown/new linkages to external
Bill Wendling41a07852009-07-20 01:03:30 +000065 case 0: return GlobalValue::ExternalLinkage;
66 case 1: return GlobalValue::WeakAnyLinkage;
67 case 2: return GlobalValue::AppendingLinkage;
68 case 3: return GlobalValue::InternalLinkage;
69 case 4: return GlobalValue::LinkOnceAnyLinkage;
70 case 5: return GlobalValue::DLLImportLinkage;
71 case 6: return GlobalValue::DLLExportLinkage;
72 case 7: return GlobalValue::ExternalWeakLinkage;
73 case 8: return GlobalValue::CommonLinkage;
74 case 9: return GlobalValue::PrivateLinkage;
Duncan Sands19d161f2009-03-07 15:45:40 +000075 case 10: return GlobalValue::WeakODRLinkage;
76 case 11: return GlobalValue::LinkOnceODRLinkage;
Chris Lattner68433442009-04-13 05:44:34 +000077 case 12: return GlobalValue::AvailableExternallyLinkage;
Bill Wendling41a07852009-07-20 01:03:30 +000078 case 13: return GlobalValue::LinkerPrivateLinkage;
Dan Gohmanf17a25c2007-07-18 16:29:46 +000079 }
80}
81
82static GlobalValue::VisibilityTypes GetDecodedVisibility(unsigned Val) {
83 switch (Val) {
84 default: // Map unknown visibilities to default.
85 case 0: return GlobalValue::DefaultVisibility;
86 case 1: return GlobalValue::HiddenVisibility;
87 case 2: return GlobalValue::ProtectedVisibility;
88 }
89}
90
91static int GetDecodedCastOpcode(unsigned Val) {
92 switch (Val) {
93 default: return -1;
94 case bitc::CAST_TRUNC : return Instruction::Trunc;
95 case bitc::CAST_ZEXT : return Instruction::ZExt;
96 case bitc::CAST_SEXT : return Instruction::SExt;
97 case bitc::CAST_FPTOUI : return Instruction::FPToUI;
98 case bitc::CAST_FPTOSI : return Instruction::FPToSI;
99 case bitc::CAST_UITOFP : return Instruction::UIToFP;
100 case bitc::CAST_SITOFP : return Instruction::SIToFP;
101 case bitc::CAST_FPTRUNC : return Instruction::FPTrunc;
102 case bitc::CAST_FPEXT : return Instruction::FPExt;
103 case bitc::CAST_PTRTOINT: return Instruction::PtrToInt;
104 case bitc::CAST_INTTOPTR: return Instruction::IntToPtr;
105 case bitc::CAST_BITCAST : return Instruction::BitCast;
106 }
107}
108static int GetDecodedBinaryOpcode(unsigned Val, const Type *Ty) {
109 switch (Val) {
110 default: return -1;
Dan Gohman7ce405e2009-06-04 22:49:04 +0000111 case bitc::BINOP_ADD:
112 return Ty->isFPOrFPVector() ? Instruction::FAdd : Instruction::Add;
113 case bitc::BINOP_SUB:
114 return Ty->isFPOrFPVector() ? Instruction::FSub : Instruction::Sub;
115 case bitc::BINOP_MUL:
116 return Ty->isFPOrFPVector() ? Instruction::FMul : Instruction::Mul;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000117 case bitc::BINOP_UDIV: return Instruction::UDiv;
118 case bitc::BINOP_SDIV:
119 return Ty->isFPOrFPVector() ? Instruction::FDiv : Instruction::SDiv;
120 case bitc::BINOP_UREM: return Instruction::URem;
121 case bitc::BINOP_SREM:
122 return Ty->isFPOrFPVector() ? Instruction::FRem : Instruction::SRem;
123 case bitc::BINOP_SHL: return Instruction::Shl;
124 case bitc::BINOP_LSHR: return Instruction::LShr;
125 case bitc::BINOP_ASHR: return Instruction::AShr;
126 case bitc::BINOP_AND: return Instruction::And;
127 case bitc::BINOP_OR: return Instruction::Or;
128 case bitc::BINOP_XOR: return Instruction::Xor;
129 }
130}
131
Gabor Greif25bc3222008-05-10 08:32:32 +0000132namespace llvm {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000133namespace {
134 /// @brief A class for maintaining the slot number definition
135 /// as a placeholder for the actual definition for forward constants defs.
136 class ConstantPlaceHolder : public ConstantExpr {
137 ConstantPlaceHolder(); // DO NOT IMPLEMENT
138 void operator=(const ConstantPlaceHolder &); // DO NOT IMPLEMENT
Gabor Greifd6da1d02008-04-06 20:25:17 +0000139 public:
140 // allocate space for exactly one operand
141 void *operator new(size_t s) {
142 return User::operator new(s, 1);
143 }
Owen Anderson4956cde2009-07-07 20:18:58 +0000144 explicit ConstantPlaceHolder(const Type *Ty, LLVMContext& Context)
Gabor Greif25bc3222008-05-10 08:32:32 +0000145 : ConstantExpr(Ty, Instruction::UserOp1, &Op<0>(), 1) {
Owen Anderson35b47072009-08-13 21:58:54 +0000146 Op<0>() = UndefValue::get(Type::getInt32Ty(Context));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000147 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000148
Chris Lattner345bbb72008-08-21 02:34:16 +0000149 /// @brief Methods to support type inquiry through isa, cast, and dyn_cast.
150 static inline bool classof(const ConstantPlaceHolder *) { return true; }
151 static bool classof(const Value *V) {
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000152 return isa<ConstantExpr>(V) &&
Chris Lattner345bbb72008-08-21 02:34:16 +0000153 cast<ConstantExpr>(V)->getOpcode() == Instruction::UserOp1;
154 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000155
156
Gabor Greif25bc3222008-05-10 08:32:32 +0000157 /// Provide fast operand accessors
Chris Lattnered490d12009-03-31 22:55:09 +0000158 //DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000159 };
160}
161
Chris Lattnered490d12009-03-31 22:55:09 +0000162// FIXME: can we inherit this from ConstantExpr?
Gabor Greif25bc3222008-05-10 08:32:32 +0000163template <>
Duncan Sandse0223c22009-09-06 08:55:57 +0000164struct OperandTraits<ConstantPlaceHolder> : public FixedNumOperandTraits<1> {
Gabor Greif25bc3222008-05-10 08:32:32 +0000165};
Gabor Greif25bc3222008-05-10 08:32:32 +0000166}
167
Chris Lattnered490d12009-03-31 22:55:09 +0000168
169void BitcodeReaderValueList::AssignValue(Value *V, unsigned Idx) {
170 if (Idx == size()) {
171 push_back(V);
172 return;
173 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000174
Chris Lattnered490d12009-03-31 22:55:09 +0000175 if (Idx >= size())
176 resize(Idx+1);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000177
Chris Lattnered490d12009-03-31 22:55:09 +0000178 WeakVH &OldV = ValuePtrs[Idx];
179 if (OldV == 0) {
180 OldV = V;
181 return;
182 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000183
Chris Lattnered490d12009-03-31 22:55:09 +0000184 // Handle constants and non-constants (e.g. instrs) differently for
185 // efficiency.
186 if (Constant *PHC = dyn_cast<Constant>(&*OldV)) {
187 ResolveConstants.push_back(std::make_pair(PHC, Idx));
188 OldV = V;
189 } else {
190 // If there was a forward reference to this value, replace it.
191 Value *PrevVal = OldV;
192 OldV->replaceAllUsesWith(V);
193 delete PrevVal;
Gabor Greif25bc3222008-05-10 08:32:32 +0000194 }
195}
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000196
Gabor Greif25bc3222008-05-10 08:32:32 +0000197
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000198Constant *BitcodeReaderValueList::getConstantFwdRef(unsigned Idx,
199 const Type *Ty) {
Chris Lattnered490d12009-03-31 22:55:09 +0000200 if (Idx >= size())
Gabor Greif25bc3222008-05-10 08:32:32 +0000201 resize(Idx + 1);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000202
Chris Lattnered490d12009-03-31 22:55:09 +0000203 if (Value *V = ValuePtrs[Idx]) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000204 assert(Ty == V->getType() && "Type mismatch in constant table!");
205 return cast<Constant>(V);
206 }
207
208 // Create and return a placeholder, which will later be RAUW'd.
Owen Anderson4956cde2009-07-07 20:18:58 +0000209 Constant *C = new ConstantPlaceHolder(Ty, Context);
Chris Lattnered490d12009-03-31 22:55:09 +0000210 ValuePtrs[Idx] = C;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000211 return C;
212}
213
214Value *BitcodeReaderValueList::getValueFwdRef(unsigned Idx, const Type *Ty) {
Chris Lattnered490d12009-03-31 22:55:09 +0000215 if (Idx >= size())
Gabor Greif25bc3222008-05-10 08:32:32 +0000216 resize(Idx + 1);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000217
Chris Lattnered490d12009-03-31 22:55:09 +0000218 if (Value *V = ValuePtrs[Idx]) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000219 assert((Ty == 0 || Ty == V->getType()) && "Type mismatch in value table!");
220 return V;
221 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000222
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000223 // No type specified, must be invalid reference.
224 if (Ty == 0) return 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000225
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000226 // Create and return a placeholder, which will later be RAUW'd.
227 Value *V = new Argument(Ty);
Chris Lattnered490d12009-03-31 22:55:09 +0000228 ValuePtrs[Idx] = V;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000229 return V;
230}
231
Chris Lattner345bbb72008-08-21 02:34:16 +0000232/// ResolveConstantForwardRefs - Once all constants are read, this method bulk
233/// resolves any forward references. The idea behind this is that we sometimes
234/// get constants (such as large arrays) which reference *many* forward ref
235/// constants. Replacing each of these causes a lot of thrashing when
236/// building/reuniquing the constant. Instead of doing this, we look at all the
237/// uses and rewrite all the place holders at once for any constant that uses
238/// a placeholder.
239void BitcodeReaderValueList::ResolveConstantForwardRefs() {
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000240 // Sort the values by-pointer so that they are efficient to look up with a
Chris Lattner345bbb72008-08-21 02:34:16 +0000241 // binary search.
242 std::sort(ResolveConstants.begin(), ResolveConstants.end());
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000243
Chris Lattner345bbb72008-08-21 02:34:16 +0000244 SmallVector<Constant*, 64> NewOps;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000245
Chris Lattner345bbb72008-08-21 02:34:16 +0000246 while (!ResolveConstants.empty()) {
Chris Lattnered490d12009-03-31 22:55:09 +0000247 Value *RealVal = operator[](ResolveConstants.back().second);
Chris Lattner345bbb72008-08-21 02:34:16 +0000248 Constant *Placeholder = ResolveConstants.back().first;
249 ResolveConstants.pop_back();
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000250
Chris Lattner345bbb72008-08-21 02:34:16 +0000251 // Loop over all users of the placeholder, updating them to reference the
252 // new value. If they reference more than one placeholder, update them all
253 // at once.
254 while (!Placeholder->use_empty()) {
Chris Lattner8ae2e252008-08-21 17:31:45 +0000255 Value::use_iterator UI = Placeholder->use_begin();
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000256
Chris Lattner345bbb72008-08-21 02:34:16 +0000257 // If the using object isn't uniqued, just update the operands. This
258 // handles instructions and initializers for global variables.
Chris Lattner8ae2e252008-08-21 17:31:45 +0000259 if (!isa<Constant>(*UI) || isa<GlobalValue>(*UI)) {
260 UI.getUse().set(RealVal);
Chris Lattner345bbb72008-08-21 02:34:16 +0000261 continue;
262 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000263
Chris Lattner345bbb72008-08-21 02:34:16 +0000264 // Otherwise, we have a constant that uses the placeholder. Replace that
265 // constant with a new constant that has *all* placeholder uses updated.
Chris Lattner8ae2e252008-08-21 17:31:45 +0000266 Constant *UserC = cast<Constant>(*UI);
Chris Lattner345bbb72008-08-21 02:34:16 +0000267 for (User::op_iterator I = UserC->op_begin(), E = UserC->op_end();
268 I != E; ++I) {
269 Value *NewOp;
270 if (!isa<ConstantPlaceHolder>(*I)) {
271 // Not a placeholder reference.
272 NewOp = *I;
273 } else if (*I == Placeholder) {
274 // Common case is that it just references this one placeholder.
275 NewOp = RealVal;
276 } else {
277 // Otherwise, look up the placeholder in ResolveConstants.
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000278 ResolveConstantsTy::iterator It =
279 std::lower_bound(ResolveConstants.begin(), ResolveConstants.end(),
Chris Lattner345bbb72008-08-21 02:34:16 +0000280 std::pair<Constant*, unsigned>(cast<Constant>(*I),
281 0));
282 assert(It != ResolveConstants.end() && It->first == *I);
Chris Lattnered490d12009-03-31 22:55:09 +0000283 NewOp = operator[](It->second);
Chris Lattner345bbb72008-08-21 02:34:16 +0000284 }
285
286 NewOps.push_back(cast<Constant>(NewOp));
287 }
288
289 // Make the new constant.
290 Constant *NewC;
291 if (ConstantArray *UserCA = dyn_cast<ConstantArray>(UserC)) {
Owen Anderson7b4f9f82009-07-28 18:32:17 +0000292 NewC = ConstantArray::get(UserCA->getType(), &NewOps[0],
Owen Anderson4956cde2009-07-07 20:18:58 +0000293 NewOps.size());
Chris Lattner345bbb72008-08-21 02:34:16 +0000294 } else if (ConstantStruct *UserCS = dyn_cast<ConstantStruct>(UserC)) {
Owen Andersond2ed7452009-08-05 23:16:16 +0000295 NewC = ConstantStruct::get(Context, &NewOps[0], NewOps.size(),
Owen Anderson4956cde2009-07-07 20:18:58 +0000296 UserCS->getType()->isPacked());
Chris Lattner345bbb72008-08-21 02:34:16 +0000297 } else if (isa<ConstantVector>(UserC)) {
Owen Anderson2f422e02009-07-28 21:19:26 +0000298 NewC = ConstantVector::get(&NewOps[0], NewOps.size());
Nick Lewycky117f4382009-05-10 20:57:05 +0000299 } else {
300 assert(isa<ConstantExpr>(UserC) && "Must be a ConstantExpr.");
Chris Lattner345bbb72008-08-21 02:34:16 +0000301 NewC = cast<ConstantExpr>(UserC)->getWithOperands(&NewOps[0],
302 NewOps.size());
303 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000304
Chris Lattner345bbb72008-08-21 02:34:16 +0000305 UserC->replaceAllUsesWith(NewC);
306 UserC->destroyConstant();
307 NewOps.clear();
308 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000309
Nick Lewycky117f4382009-05-10 20:57:05 +0000310 // Update all ValueHandles, they should be the only users at this point.
311 Placeholder->replaceAllUsesWith(RealVal);
Chris Lattner345bbb72008-08-21 02:34:16 +0000312 delete Placeholder;
313 }
314}
315
Devang Patelea27c232009-08-04 06:00:18 +0000316void BitcodeReaderMDValueList::AssignValue(Value *V, unsigned Idx) {
317 if (Idx == size()) {
318 push_back(V);
319 return;
320 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000321
Devang Patelea27c232009-08-04 06:00:18 +0000322 if (Idx >= size())
323 resize(Idx+1);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000324
Devang Patelea27c232009-08-04 06:00:18 +0000325 WeakVH &OldV = MDValuePtrs[Idx];
326 if (OldV == 0) {
327 OldV = V;
328 return;
329 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000330
Devang Patelea27c232009-08-04 06:00:18 +0000331 // If there was a forward reference to this value, replace it.
332 Value *PrevVal = OldV;
333 OldV->replaceAllUsesWith(V);
334 delete PrevVal;
Devang Pateldeda3952009-09-03 01:38:02 +0000335 // Deleting PrevVal sets Idx value in MDValuePtrs to null. Set new
336 // value for Idx.
337 MDValuePtrs[Idx] = V;
Devang Patelea27c232009-08-04 06:00:18 +0000338}
339
340Value *BitcodeReaderMDValueList::getValueFwdRef(unsigned Idx) {
341 if (Idx >= size())
342 resize(Idx + 1);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000343
Devang Patelea27c232009-08-04 06:00:18 +0000344 if (Value *V = MDValuePtrs[Idx]) {
Owen Anderson35b47072009-08-13 21:58:54 +0000345 assert(V->getType() == Type::getMetadataTy(Context) && "Type mismatch in value table!");
Devang Patelea27c232009-08-04 06:00:18 +0000346 return V;
347 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000348
Devang Patelea27c232009-08-04 06:00:18 +0000349 // Create and return a placeholder, which will later be RAUW'd.
Owen Anderson35b47072009-08-13 21:58:54 +0000350 Value *V = new Argument(Type::getMetadataTy(Context));
Devang Patelea27c232009-08-04 06:00:18 +0000351 MDValuePtrs[Idx] = V;
352 return V;
353}
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000354
355const Type *BitcodeReader::getTypeByID(unsigned ID, bool isTypeTable) {
356 // If the TypeID is in range, return it.
357 if (ID < TypeList.size())
358 return TypeList[ID].get();
359 if (!isTypeTable) return 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000360
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000361 // The type table allows forward references. Push as many Opaque types as
362 // needed to get up to ID.
363 while (TypeList.size() <= ID)
Owen Anderson02f7bd52009-08-13 23:27:32 +0000364 TypeList.push_back(OpaqueType::get(Context));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000365 return TypeList.back().get();
366}
367
368//===----------------------------------------------------------------------===//
369// Functions for parsing blocks from the bitcode file
370//===----------------------------------------------------------------------===//
371
Devang Pateld222f862008-09-25 21:00:45 +0000372bool BitcodeReader::ParseAttributeBlock() {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000373 if (Stream.EnterSubBlock(bitc::PARAMATTR_BLOCK_ID))
374 return Error("Malformed block record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000375
Devang Patelf2a4a922008-09-26 22:53:05 +0000376 if (!MAttributes.empty())
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000377 return Error("Multiple PARAMATTR blocks found!");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000378
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000379 SmallVector<uint64_t, 64> Record;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000380
Devang Pateld222f862008-09-25 21:00:45 +0000381 SmallVector<AttributeWithIndex, 8> Attrs;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000382
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000383 // Read all the records.
384 while (1) {
385 unsigned Code = Stream.ReadCode();
386 if (Code == bitc::END_BLOCK) {
387 if (Stream.ReadBlockEnd())
388 return Error("Error at end of PARAMATTR block");
389 return false;
390 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000391
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000392 if (Code == bitc::ENTER_SUBBLOCK) {
393 // No known subblocks, always skip them.
394 Stream.ReadSubBlockID();
395 if (Stream.SkipBlock())
396 return Error("Malformed block record");
397 continue;
398 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000399
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000400 if (Code == bitc::DEFINE_ABBREV) {
401 Stream.ReadAbbrevRecord();
402 continue;
403 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000404
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000405 // Read a record.
406 Record.clear();
407 switch (Stream.ReadRecord(Code, Record)) {
408 default: // Default behavior: ignore.
409 break;
410 case bitc::PARAMATTR_CODE_ENTRY: { // ENTRY: [paramidx0, attr0, ...]
411 if (Record.size() & 1)
412 return Error("Invalid ENTRY record");
413
Chris Lattner58601542008-10-05 18:22:09 +0000414 // FIXME : Remove this autoupgrade code in LLVM 3.0.
Devang Patelf2a4a922008-09-26 22:53:05 +0000415 // If Function attributes are using index 0 then transfer them
Chris Lattner58601542008-10-05 18:22:09 +0000416 // to index ~0. Index 0 is used for return value attributes but used to be
417 // used for function attributes.
Devang Patelf2a4a922008-09-26 22:53:05 +0000418 Attributes RetAttribute = Attribute::None;
419 Attributes FnAttribute = Attribute::None;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000420 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Nick Lewycky4029a942008-12-19 09:38:31 +0000421 // FIXME: remove in LLVM 3.0
422 // The alignment is stored as a 16-bit raw value from bits 31--16.
423 // We shift the bits above 31 down by 11 bits.
424
425 unsigned Alignment = (Record[i+1] & (0xffffull << 16)) >> 16;
426 if (Alignment && !isPowerOf2_32(Alignment))
427 return Error("Alignment is not a power of two.");
428
429 Attributes ReconstitutedAttr = Record[i+1] & 0xffff;
430 if (Alignment)
431 ReconstitutedAttr |= Attribute::constructAlignmentFromInt(Alignment);
432 ReconstitutedAttr |= (Record[i+1] & (0xffffull << 32)) >> 11;
433 Record[i+1] = ReconstitutedAttr;
434
Devang Patelf2a4a922008-09-26 22:53:05 +0000435 if (Record[i] == 0)
436 RetAttribute = Record[i+1];
437 else if (Record[i] == ~0U)
438 FnAttribute = Record[i+1];
439 }
Chris Lattner58601542008-10-05 18:22:09 +0000440
441 unsigned OldRetAttrs = (Attribute::NoUnwind|Attribute::NoReturn|
442 Attribute::ReadOnly|Attribute::ReadNone);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000443
Chris Lattner58601542008-10-05 18:22:09 +0000444 if (FnAttribute == Attribute::None && RetAttribute != Attribute::None &&
445 (RetAttribute & OldRetAttrs) != 0) {
446 if (FnAttribute == Attribute::None) { // add a slot so they get added.
447 Record.push_back(~0U);
448 Record.push_back(0);
Devang Patelf2a4a922008-09-26 22:53:05 +0000449 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000450
Chris Lattner58601542008-10-05 18:22:09 +0000451 FnAttribute |= RetAttribute & OldRetAttrs;
452 RetAttribute &= ~OldRetAttrs;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000453 }
Chris Lattner2ee85532008-03-12 02:25:52 +0000454
Devang Patelf2a4a922008-09-26 22:53:05 +0000455 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Chris Lattner58601542008-10-05 18:22:09 +0000456 if (Record[i] == 0) {
457 if (RetAttribute != Attribute::None)
458 Attrs.push_back(AttributeWithIndex::get(0, RetAttribute));
459 } else if (Record[i] == ~0U) {
460 if (FnAttribute != Attribute::None)
461 Attrs.push_back(AttributeWithIndex::get(~0U, FnAttribute));
462 } else if (Record[i+1] != Attribute::None)
Devang Patelf2a4a922008-09-26 22:53:05 +0000463 Attrs.push_back(AttributeWithIndex::get(Record[i], Record[i+1]));
464 }
Devang Patelf2a4a922008-09-26 22:53:05 +0000465
466 MAttributes.push_back(AttrListPtr::get(Attrs.begin(), Attrs.end()));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000467 Attrs.clear();
468 break;
469 }
Duncan Sandse2898f02007-11-20 14:09:29 +0000470 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000471 }
472}
473
474
475bool BitcodeReader::ParseTypeTable() {
476 if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID))
477 return Error("Malformed block record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000478
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000479 if (!TypeList.empty())
480 return Error("Multiple TYPE_BLOCKs found!");
481
482 SmallVector<uint64_t, 64> Record;
483 unsigned NumRecords = 0;
484
485 // Read all the records for this type table.
486 while (1) {
487 unsigned Code = Stream.ReadCode();
488 if (Code == bitc::END_BLOCK) {
489 if (NumRecords != TypeList.size())
490 return Error("Invalid type forward reference in TYPE_BLOCK");
491 if (Stream.ReadBlockEnd())
492 return Error("Error at end of type table block");
493 return false;
494 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000495
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000496 if (Code == bitc::ENTER_SUBBLOCK) {
497 // No known subblocks, always skip them.
498 Stream.ReadSubBlockID();
499 if (Stream.SkipBlock())
500 return Error("Malformed block record");
501 continue;
502 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000503
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000504 if (Code == bitc::DEFINE_ABBREV) {
505 Stream.ReadAbbrevRecord();
506 continue;
507 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000508
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000509 // Read a record.
510 Record.clear();
511 const Type *ResultTy = 0;
512 switch (Stream.ReadRecord(Code, Record)) {
513 default: // Default behavior: unknown type.
514 ResultTy = 0;
515 break;
516 case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
517 // TYPE_CODE_NUMENTRY contains a count of the number of types in the
518 // type list. This allows us to reserve space.
519 if (Record.size() < 1)
520 return Error("Invalid TYPE_CODE_NUMENTRY record");
521 TypeList.reserve(Record[0]);
522 continue;
523 case bitc::TYPE_CODE_VOID: // VOID
Owen Anderson35b47072009-08-13 21:58:54 +0000524 ResultTy = Type::getVoidTy(Context);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000525 break;
526 case bitc::TYPE_CODE_FLOAT: // FLOAT
Owen Anderson35b47072009-08-13 21:58:54 +0000527 ResultTy = Type::getFloatTy(Context);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000528 break;
529 case bitc::TYPE_CODE_DOUBLE: // DOUBLE
Owen Anderson35b47072009-08-13 21:58:54 +0000530 ResultTy = Type::getDoubleTy(Context);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000531 break;
Dale Johannesenf325d9f2007-08-03 01:03:46 +0000532 case bitc::TYPE_CODE_X86_FP80: // X86_FP80
Owen Anderson35b47072009-08-13 21:58:54 +0000533 ResultTy = Type::getX86_FP80Ty(Context);
Dale Johannesenf325d9f2007-08-03 01:03:46 +0000534 break;
535 case bitc::TYPE_CODE_FP128: // FP128
Owen Anderson35b47072009-08-13 21:58:54 +0000536 ResultTy = Type::getFP128Ty(Context);
Dale Johannesenf325d9f2007-08-03 01:03:46 +0000537 break;
538 case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
Owen Anderson35b47072009-08-13 21:58:54 +0000539 ResultTy = Type::getPPC_FP128Ty(Context);
Dale Johannesenf325d9f2007-08-03 01:03:46 +0000540 break;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000541 case bitc::TYPE_CODE_LABEL: // LABEL
Owen Anderson35b47072009-08-13 21:58:54 +0000542 ResultTy = Type::getLabelTy(Context);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000543 break;
544 case bitc::TYPE_CODE_OPAQUE: // OPAQUE
545 ResultTy = 0;
546 break;
Nick Lewycky29aaef82009-05-30 05:06:04 +0000547 case bitc::TYPE_CODE_METADATA: // METADATA
Owen Anderson35b47072009-08-13 21:58:54 +0000548 ResultTy = Type::getMetadataTy(Context);
Nick Lewycky29aaef82009-05-30 05:06:04 +0000549 break;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000550 case bitc::TYPE_CODE_INTEGER: // INTEGER: [width]
551 if (Record.size() < 1)
552 return Error("Invalid Integer type record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000553
Owen Anderson35b47072009-08-13 21:58:54 +0000554 ResultTy = IntegerType::get(Context, Record[0]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000555 break;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000556 case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
Christopher Lamb44d62f62007-12-11 08:59:05 +0000557 // [pointee type, address space]
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000558 if (Record.size() < 1)
559 return Error("Invalid POINTER type record");
Christopher Lamb44d62f62007-12-11 08:59:05 +0000560 unsigned AddressSpace = 0;
561 if (Record.size() == 2)
562 AddressSpace = Record[1];
Owen Anderson6b6e2d92009-07-29 22:17:13 +0000563 ResultTy = PointerType::get(getTypeByID(Record[0], true),
Owen Anderson4956cde2009-07-07 20:18:58 +0000564 AddressSpace);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000565 break;
Christopher Lamb44d62f62007-12-11 08:59:05 +0000566 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000567 case bitc::TYPE_CODE_FUNCTION: {
Chris Lattner70b644d2007-11-27 17:48:06 +0000568 // FIXME: attrid is dead, remove it in LLVM 3.0
569 // FUNCTION: [vararg, attrid, retty, paramty x N]
570 if (Record.size() < 3)
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000571 return Error("Invalid FUNCTION type record");
572 std::vector<const Type*> ArgTys;
Chris Lattner70b644d2007-11-27 17:48:06 +0000573 for (unsigned i = 3, e = Record.size(); i != e; ++i)
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000574 ArgTys.push_back(getTypeByID(Record[i], true));
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000575
Owen Anderson6b6e2d92009-07-29 22:17:13 +0000576 ResultTy = FunctionType::get(getTypeByID(Record[2], true), ArgTys,
Duncan Sandsf5588dc2007-11-27 13:23:08 +0000577 Record[0]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000578 break;
579 }
580 case bitc::TYPE_CODE_STRUCT: { // STRUCT: [ispacked, eltty x N]
581 if (Record.size() < 1)
582 return Error("Invalid STRUCT type record");
583 std::vector<const Type*> EltTys;
584 for (unsigned i = 1, e = Record.size(); i != e; ++i)
585 EltTys.push_back(getTypeByID(Record[i], true));
Owen Andersond2ed7452009-08-05 23:16:16 +0000586 ResultTy = StructType::get(Context, EltTys, Record[0]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000587 break;
588 }
589 case bitc::TYPE_CODE_ARRAY: // ARRAY: [numelts, eltty]
590 if (Record.size() < 2)
591 return Error("Invalid ARRAY type record");
Owen Anderson6b6e2d92009-07-29 22:17:13 +0000592 ResultTy = ArrayType::get(getTypeByID(Record[1], true), Record[0]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000593 break;
594 case bitc::TYPE_CODE_VECTOR: // VECTOR: [numelts, eltty]
595 if (Record.size() < 2)
596 return Error("Invalid VECTOR type record");
Owen Anderson6b6e2d92009-07-29 22:17:13 +0000597 ResultTy = VectorType::get(getTypeByID(Record[1], true), Record[0]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000598 break;
599 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000600
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000601 if (NumRecords == TypeList.size()) {
602 // If this is a new type slot, just append it.
Owen Anderson02f7bd52009-08-13 23:27:32 +0000603 TypeList.push_back(ResultTy ? ResultTy : OpaqueType::get(Context));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000604 ++NumRecords;
605 } else if (ResultTy == 0) {
606 // Otherwise, this was forward referenced, so an opaque type was created,
607 // but the result type is actually just an opaque. Leave the one we
608 // created previously.
609 ++NumRecords;
610 } else {
611 // Otherwise, this was forward referenced, so an opaque type was created.
612 // Resolve the opaque type to the real type now.
613 assert(NumRecords < TypeList.size() && "Typelist imbalance");
614 const OpaqueType *OldTy = cast<OpaqueType>(TypeList[NumRecords++].get());
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000615
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000616 // Don't directly push the new type on the Tab. Instead we want to replace
617 // the opaque type we previously inserted with the new concrete value. The
618 // refinement from the abstract (opaque) type to the new type causes all
619 // uses of the abstract type to use the concrete type (NewTy). This will
620 // also cause the opaque type to be deleted.
621 const_cast<OpaqueType*>(OldTy)->refineAbstractTypeTo(ResultTy);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000622
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000623 // This should have replaced the old opaque type with the new type in the
624 // value table... or with a preexisting type that was already in the
625 // system. Let's just make sure it did.
626 assert(TypeList[NumRecords-1].get() != OldTy &&
627 "refineAbstractType didn't work!");
628 }
629 }
630}
631
632
633bool BitcodeReader::ParseTypeSymbolTable() {
634 if (Stream.EnterSubBlock(bitc::TYPE_SYMTAB_BLOCK_ID))
635 return Error("Malformed block record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000636
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000637 SmallVector<uint64_t, 64> Record;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000638
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000639 // Read all the records for this type table.
640 std::string TypeName;
641 while (1) {
642 unsigned Code = Stream.ReadCode();
643 if (Code == bitc::END_BLOCK) {
644 if (Stream.ReadBlockEnd())
645 return Error("Error at end of type symbol table block");
646 return false;
647 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000648
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000649 if (Code == bitc::ENTER_SUBBLOCK) {
650 // No known subblocks, always skip them.
651 Stream.ReadSubBlockID();
652 if (Stream.SkipBlock())
653 return Error("Malformed block record");
654 continue;
655 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000656
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000657 if (Code == bitc::DEFINE_ABBREV) {
658 Stream.ReadAbbrevRecord();
659 continue;
660 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000661
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000662 // Read a record.
663 Record.clear();
664 switch (Stream.ReadRecord(Code, Record)) {
665 default: // Default behavior: unknown type.
666 break;
667 case bitc::TST_CODE_ENTRY: // TST_ENTRY: [typeid, namechar x N]
668 if (ConvertToString(Record, 1, TypeName))
669 return Error("Invalid TST_ENTRY record");
670 unsigned TypeID = Record[0];
671 if (TypeID >= TypeList.size())
672 return Error("Invalid Type ID in TST_ENTRY record");
673
674 TheModule->addTypeName(TypeName, TypeList[TypeID].get());
675 TypeName.clear();
676 break;
677 }
678 }
679}
680
681bool BitcodeReader::ParseValueSymbolTable() {
682 if (Stream.EnterSubBlock(bitc::VALUE_SYMTAB_BLOCK_ID))
683 return Error("Malformed block record");
684
685 SmallVector<uint64_t, 64> Record;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000686
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000687 // Read all the records for this value table.
688 SmallString<128> ValueName;
689 while (1) {
690 unsigned Code = Stream.ReadCode();
691 if (Code == bitc::END_BLOCK) {
692 if (Stream.ReadBlockEnd())
693 return Error("Error at end of value symbol table block");
694 return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000695 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000696 if (Code == bitc::ENTER_SUBBLOCK) {
697 // No known subblocks, always skip them.
698 Stream.ReadSubBlockID();
699 if (Stream.SkipBlock())
700 return Error("Malformed block record");
701 continue;
702 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000703
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000704 if (Code == bitc::DEFINE_ABBREV) {
705 Stream.ReadAbbrevRecord();
706 continue;
707 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000708
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000709 // Read a record.
710 Record.clear();
711 switch (Stream.ReadRecord(Code, Record)) {
712 default: // Default behavior: unknown type.
713 break;
714 case bitc::VST_CODE_ENTRY: { // VST_ENTRY: [valueid, namechar x N]
715 if (ConvertToString(Record, 1, ValueName))
Nick Lewyckydc9b1b52009-05-31 06:07:28 +0000716 return Error("Invalid VST_ENTRY record");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000717 unsigned ValueID = Record[0];
718 if (ValueID >= ValueList.size())
719 return Error("Invalid Value ID in VST_ENTRY record");
720 Value *V = ValueList[ValueID];
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000721
Daniel Dunbar1be13862009-07-26 00:34:27 +0000722 V->setName(StringRef(ValueName.data(), ValueName.size()));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000723 ValueName.clear();
724 break;
725 }
726 case bitc::VST_CODE_BBENTRY: {
727 if (ConvertToString(Record, 1, ValueName))
728 return Error("Invalid VST_BBENTRY record");
729 BasicBlock *BB = getBasicBlock(Record[0]);
730 if (BB == 0)
731 return Error("Invalid BB ID in VST_BBENTRY record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000732
Daniel Dunbar1be13862009-07-26 00:34:27 +0000733 BB->setName(StringRef(ValueName.data(), ValueName.size()));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000734 ValueName.clear();
735 break;
736 }
737 }
738 }
739}
740
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000741bool BitcodeReader::ParseMetadata() {
Devang Patelea27c232009-08-04 06:00:18 +0000742 unsigned NextValueNo = MDValueList.size();
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000743
744 if (Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
745 return Error("Malformed block record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000746
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000747 SmallVector<uint64_t, 64> Record;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000748
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000749 // Read all the records.
750 while (1) {
751 unsigned Code = Stream.ReadCode();
752 if (Code == bitc::END_BLOCK) {
753 if (Stream.ReadBlockEnd())
754 return Error("Error at end of PARAMATTR block");
755 return false;
756 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000757
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000758 if (Code == bitc::ENTER_SUBBLOCK) {
759 // No known subblocks, always skip them.
760 Stream.ReadSubBlockID();
761 if (Stream.SkipBlock())
762 return Error("Malformed block record");
763 continue;
764 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000765
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000766 if (Code == bitc::DEFINE_ABBREV) {
767 Stream.ReadAbbrevRecord();
768 continue;
769 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000770
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000771 // Read a record.
772 Record.clear();
773 switch (Stream.ReadRecord(Code, Record)) {
774 default: // Default behavior: ignore.
775 break;
Devang Patel3e1ef932009-07-29 22:34:41 +0000776 case bitc::METADATA_NAME: {
777 // Read named of the named metadata.
778 unsigned NameLength = Record.size();
779 SmallString<8> Name;
780 Name.resize(NameLength);
781 for (unsigned i = 0; i != NameLength; ++i)
782 Name[i] = Record[i];
783 Record.clear();
784 Code = Stream.ReadCode();
785
786 // METADATA_NAME is always followed by METADATA_NAMED_NODE.
787 if (Stream.ReadRecord(Code, Record) != bitc::METADATA_NAMED_NODE)
788 assert ( 0 && "Inavlid Named Metadata record");
789
790 // Read named metadata elements.
791 unsigned Size = Record.size();
792 SmallVector<MetadataBase*, 8> Elts;
793 for (unsigned i = 0; i != Size; ++i) {
Devang Patelea27c232009-08-04 06:00:18 +0000794 Value *MD = MDValueList.getValueFwdRef(Record[i]);
Devang Patel3e1ef932009-07-29 22:34:41 +0000795 if (MetadataBase *B = dyn_cast<MetadataBase>(MD))
796 Elts.push_back(B);
797 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000798 Value *V = NamedMDNode::Create(Context, Name.str(), Elts.data(),
Owen Anderson35b47072009-08-13 21:58:54 +0000799 Elts.size(), TheModule);
Devang Patelea27c232009-08-04 06:00:18 +0000800 MDValueList.AssignValue(V, NextValueNo++);
Devang Patel3e1ef932009-07-29 22:34:41 +0000801 break;
802 }
Devang Patel54199ef2009-07-23 01:07:34 +0000803 case bitc::METADATA_NODE: {
804 if (Record.empty() || Record.size() % 2 == 1)
805 return Error("Invalid METADATA_NODE record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000806
Devang Patel54199ef2009-07-23 01:07:34 +0000807 unsigned Size = Record.size();
808 SmallVector<Value*, 8> Elts;
809 for (unsigned i = 0; i != Size; i += 2) {
810 const Type *Ty = getTypeByID(Record[i], false);
Owen Anderson35b47072009-08-13 21:58:54 +0000811 if (Ty == Type::getMetadataTy(Context))
Devang Patelea27c232009-08-04 06:00:18 +0000812 Elts.push_back(MDValueList.getValueFwdRef(Record[i+1]));
Owen Anderson35b47072009-08-13 21:58:54 +0000813 else if (Ty != Type::getVoidTy(Context))
Devang Patel54199ef2009-07-23 01:07:34 +0000814 Elts.push_back(ValueList.getValueFwdRef(Record[i+1], Ty));
815 else
816 Elts.push_back(NULL);
817 }
Owen Anderson20be0582009-07-31 21:35:40 +0000818 Value *V = MDNode::get(Context, &Elts[0], Elts.size());
Devang Patelea27c232009-08-04 06:00:18 +0000819 MDValueList.AssignValue(V, NextValueNo++);
Devang Patel54199ef2009-07-23 01:07:34 +0000820 break;
821 }
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000822 case bitc::METADATA_STRING: {
823 unsigned MDStringLength = Record.size();
824 SmallString<8> String;
825 String.resize(MDStringLength);
826 for (unsigned i = 0; i != MDStringLength; ++i)
827 String[i] = Record[i];
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000828 Value *V = MDString::get(Context,
Owen Anderson20be0582009-07-31 21:35:40 +0000829 StringRef(String.data(), String.size()));
Devang Patelea27c232009-08-04 06:00:18 +0000830 MDValueList.AssignValue(V, NextValueNo++);
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000831 break;
832 }
Devang Pateladc37612009-09-18 19:26:43 +0000833 case bitc::METADATA_KIND: {
834 unsigned RecordLength = Record.size();
835 if (Record.empty() || RecordLength < 2)
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000836 return Error("Invalid METADATA_KIND record");
Devang Pateladc37612009-09-18 19:26:43 +0000837 SmallString<8> Name;
838 Name.resize(RecordLength-1);
839 MDKindID Kind = Record[0];
840 for (unsigned i = 1; i != RecordLength; ++i)
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000841 Name[i-1] = Record[i];
Devang Pateladc37612009-09-18 19:26:43 +0000842 Metadata &TheMetadata = Context.getMetadata();
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000843 assert(TheMetadata.MDHandlerNames.find(Name.str())
844 == TheMetadata.MDHandlerNames.end() &&
845 "Already registered MDKind!");
Devang Pateladc37612009-09-18 19:26:43 +0000846 TheMetadata.MDHandlerNames[Name.str()] = Kind;
847 break;
848 }
Devang Patel0c0a6ca2009-07-22 17:43:22 +0000849 }
850 }
851}
852
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000853/// DecodeSignRotatedValue - Decode a signed value stored with the sign bit in
854/// the LSB for dense VBR encoding.
855static uint64_t DecodeSignRotatedValue(uint64_t V) {
856 if ((V & 1) == 0)
857 return V >> 1;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000858 if (V != 1)
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000859 return -(V >> 1);
860 // There is no such thing as -0 with integers. "-0" really means MININT.
861 return 1ULL << 63;
862}
863
864/// ResolveGlobalAndAliasInits - Resolve all of the initializers for global
865/// values and aliases that we can.
866bool BitcodeReader::ResolveGlobalAndAliasInits() {
867 std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInitWorklist;
868 std::vector<std::pair<GlobalAlias*, unsigned> > AliasInitWorklist;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000869
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000870 GlobalInitWorklist.swap(GlobalInits);
871 AliasInitWorklist.swap(AliasInits);
872
873 while (!GlobalInitWorklist.empty()) {
874 unsigned ValID = GlobalInitWorklist.back().second;
875 if (ValID >= ValueList.size()) {
876 // Not ready to resolve this yet, it requires something later in the file.
877 GlobalInits.push_back(GlobalInitWorklist.back());
878 } else {
879 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
880 GlobalInitWorklist.back().first->setInitializer(C);
881 else
882 return Error("Global variable initializer is not a constant!");
883 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000884 GlobalInitWorklist.pop_back();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000885 }
886
887 while (!AliasInitWorklist.empty()) {
888 unsigned ValID = AliasInitWorklist.back().second;
889 if (ValID >= ValueList.size()) {
890 AliasInits.push_back(AliasInitWorklist.back());
891 } else {
892 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
893 AliasInitWorklist.back().first->setAliasee(C);
894 else
895 return Error("Alias initializer is not a constant!");
896 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000897 AliasInitWorklist.pop_back();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000898 }
899 return false;
900}
901
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000902bool BitcodeReader::ParseConstants() {
903 if (Stream.EnterSubBlock(bitc::CONSTANTS_BLOCK_ID))
904 return Error("Malformed block record");
905
906 SmallVector<uint64_t, 64> Record;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000907
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000908 // Read all the records for this value table.
Owen Anderson35b47072009-08-13 21:58:54 +0000909 const Type *CurTy = Type::getInt32Ty(Context);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000910 unsigned NextCstNo = ValueList.size();
911 while (1) {
912 unsigned Code = Stream.ReadCode();
Chris Lattner345bbb72008-08-21 02:34:16 +0000913 if (Code == bitc::END_BLOCK)
914 break;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000915
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000916 if (Code == bitc::ENTER_SUBBLOCK) {
917 // No known subblocks, always skip them.
918 Stream.ReadSubBlockID();
919 if (Stream.SkipBlock())
920 return Error("Malformed block record");
921 continue;
922 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000923
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000924 if (Code == bitc::DEFINE_ABBREV) {
925 Stream.ReadAbbrevRecord();
926 continue;
927 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000928
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000929 // Read a record.
930 Record.clear();
931 Value *V = 0;
Dan Gohman799a8c82009-07-20 21:19:07 +0000932 unsigned BitCode = Stream.ReadRecord(Code, Record);
933 switch (BitCode) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000934 default: // Default behavior: unknown constant
935 case bitc::CST_CODE_UNDEF: // UNDEF
Owen Andersonb99ecca2009-07-30 23:03:37 +0000936 V = UndefValue::get(CurTy);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000937 break;
938 case bitc::CST_CODE_SETTYPE: // SETTYPE: [typeid]
939 if (Record.empty())
940 return Error("Malformed CST_SETTYPE record");
941 if (Record[0] >= TypeList.size())
942 return Error("Invalid Type ID in CST_SETTYPE record");
943 CurTy = TypeList[Record[0]];
944 continue; // Skip the ValueList manipulation.
945 case bitc::CST_CODE_NULL: // NULL
Owen Andersonaac28372009-07-31 20:28:14 +0000946 V = Constant::getNullValue(CurTy);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000947 break;
948 case bitc::CST_CODE_INTEGER: // INTEGER: [intval]
949 if (!isa<IntegerType>(CurTy) || Record.empty())
950 return Error("Invalid CST_INTEGER record");
Owen Andersoneacb44d2009-07-24 23:12:02 +0000951 V = ConstantInt::get(CurTy, DecodeSignRotatedValue(Record[0]));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000952 break;
953 case bitc::CST_CODE_WIDE_INTEGER: {// WIDE_INTEGER: [n x intval]
954 if (!isa<IntegerType>(CurTy) || Record.empty())
955 return Error("Invalid WIDE_INTEGER record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000956
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000957 unsigned NumWords = Record.size();
958 SmallVector<uint64_t, 8> Words;
959 Words.resize(NumWords);
960 for (unsigned i = 0; i != NumWords; ++i)
961 Words[i] = DecodeSignRotatedValue(Record[i]);
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000962 V = ConstantInt::get(Context,
Owen Andersoneacb44d2009-07-24 23:12:02 +0000963 APInt(cast<IntegerType>(CurTy)->getBitWidth(),
964 NumWords, &Words[0]));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000965 break;
966 }
Dale Johannesen1616e902007-09-11 18:32:33 +0000967 case bitc::CST_CODE_FLOAT: { // FLOAT: [fpval]
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000968 if (Record.empty())
969 return Error("Invalid FLOAT record");
Owen Anderson35b47072009-08-13 21:58:54 +0000970 if (CurTy == Type::getFloatTy(Context))
Owen Andersond363a0e2009-07-27 20:59:43 +0000971 V = ConstantFP::get(Context, APFloat(APInt(32, (uint32_t)Record[0])));
Owen Anderson35b47072009-08-13 21:58:54 +0000972 else if (CurTy == Type::getDoubleTy(Context))
Owen Andersond363a0e2009-07-27 20:59:43 +0000973 V = ConstantFP::get(Context, APFloat(APInt(64, Record[0])));
Owen Anderson35b47072009-08-13 21:58:54 +0000974 else if (CurTy == Type::getX86_FP80Ty(Context)) {
Dale Johannesen0a92eac2009-03-23 21:16:53 +0000975 // Bits are not stored the same way as a normal i80 APInt, compensate.
976 uint64_t Rearrange[2];
977 Rearrange[0] = (Record[1] & 0xffffLL) | (Record[0] << 16);
978 Rearrange[1] = Record[0] >> 48;
Owen Andersond363a0e2009-07-27 20:59:43 +0000979 V = ConstantFP::get(Context, APFloat(APInt(80, 2, Rearrange)));
Owen Anderson35b47072009-08-13 21:58:54 +0000980 } else if (CurTy == Type::getFP128Ty(Context))
Owen Andersond363a0e2009-07-27 20:59:43 +0000981 V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0]), true));
Owen Anderson35b47072009-08-13 21:58:54 +0000982 else if (CurTy == Type::getPPC_FP128Ty(Context))
Owen Andersond363a0e2009-07-27 20:59:43 +0000983 V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0])));
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000984 else
Owen Andersonb99ecca2009-07-30 23:03:37 +0000985 V = UndefValue::get(CurTy);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000986 break;
Dale Johannesen1616e902007-09-11 18:32:33 +0000987 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000988
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000989 case bitc::CST_CODE_AGGREGATE: {// AGGREGATE: [n x value number]
990 if (Record.empty())
991 return Error("Invalid CST_AGGREGATE record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000992
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000993 unsigned Size = Record.size();
994 std::vector<Constant*> Elts;
Daniel Dunbar3be44e62009-09-20 02:20:51 +0000995
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000996 if (const StructType *STy = dyn_cast<StructType>(CurTy)) {
997 for (unsigned i = 0; i != Size; ++i)
998 Elts.push_back(ValueList.getConstantFwdRef(Record[i],
999 STy->getElementType(i)));
Owen Anderson9c9f10e2009-07-27 22:29:26 +00001000 V = ConstantStruct::get(STy, Elts);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001001 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(CurTy)) {
1002 const Type *EltTy = ATy->getElementType();
1003 for (unsigned i = 0; i != Size; ++i)
1004 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Owen Anderson7b4f9f82009-07-28 18:32:17 +00001005 V = ConstantArray::get(ATy, Elts);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001006 } else if (const VectorType *VTy = dyn_cast<VectorType>(CurTy)) {
1007 const Type *EltTy = VTy->getElementType();
1008 for (unsigned i = 0; i != Size; ++i)
1009 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Owen Anderson2f422e02009-07-28 21:19:26 +00001010 V = ConstantVector::get(Elts);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001011 } else {
Owen Andersonb99ecca2009-07-30 23:03:37 +00001012 V = UndefValue::get(CurTy);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001013 }
1014 break;
1015 }
1016 case bitc::CST_CODE_STRING: { // STRING: [values]
1017 if (Record.empty())
1018 return Error("Invalid CST_AGGREGATE record");
1019
1020 const ArrayType *ATy = cast<ArrayType>(CurTy);
1021 const Type *EltTy = ATy->getElementType();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001022
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001023 unsigned Size = Record.size();
1024 std::vector<Constant*> Elts;
1025 for (unsigned i = 0; i != Size; ++i)
Owen Andersoneacb44d2009-07-24 23:12:02 +00001026 Elts.push_back(ConstantInt::get(EltTy, Record[i]));
Owen Anderson7b4f9f82009-07-28 18:32:17 +00001027 V = ConstantArray::get(ATy, Elts);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001028 break;
1029 }
1030 case bitc::CST_CODE_CSTRING: { // CSTRING: [values]
1031 if (Record.empty())
1032 return Error("Invalid CST_AGGREGATE record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001033
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001034 const ArrayType *ATy = cast<ArrayType>(CurTy);
1035 const Type *EltTy = ATy->getElementType();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001036
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001037 unsigned Size = Record.size();
1038 std::vector<Constant*> Elts;
1039 for (unsigned i = 0; i != Size; ++i)
Owen Andersoneacb44d2009-07-24 23:12:02 +00001040 Elts.push_back(ConstantInt::get(EltTy, Record[i]));
Owen Andersonaac28372009-07-31 20:28:14 +00001041 Elts.push_back(Constant::getNullValue(EltTy));
Owen Anderson7b4f9f82009-07-28 18:32:17 +00001042 V = ConstantArray::get(ATy, Elts);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001043 break;
1044 }
1045 case bitc::CST_CODE_CE_BINOP: { // CE_BINOP: [opcode, opval, opval]
1046 if (Record.size() < 3) return Error("Invalid CE_BINOP record");
1047 int Opc = GetDecodedBinaryOpcode(Record[0], CurTy);
1048 if (Opc < 0) {
Owen Andersonb99ecca2009-07-30 23:03:37 +00001049 V = UndefValue::get(CurTy); // Unknown binop.
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001050 } else {
1051 Constant *LHS = ValueList.getConstantFwdRef(Record[1], CurTy);
1052 Constant *RHS = ValueList.getConstantFwdRef(Record[2], CurTy);
Dan Gohmanf3a08b82009-09-07 23:54:19 +00001053 unsigned Flags = 0;
1054 if (Record.size() >= 4) {
1055 if (Opc == Instruction::Add ||
1056 Opc == Instruction::Sub ||
1057 Opc == Instruction::Mul) {
1058 if (Record[3] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1059 Flags |= OverflowingBinaryOperator::NoSignedWrap;
1060 if (Record[3] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1061 Flags |= OverflowingBinaryOperator::NoUnsignedWrap;
1062 } else if (Opc == Instruction::SDiv) {
1063 if (Record[3] & (1 << bitc::SDIV_EXACT))
1064 Flags |= SDivOperator::IsExact;
1065 }
1066 }
1067 V = ConstantExpr::get(Opc, LHS, RHS, Flags);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001068 }
1069 break;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001070 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001071 case bitc::CST_CODE_CE_CAST: { // CE_CAST: [opcode, opty, opval]
1072 if (Record.size() < 3) return Error("Invalid CE_CAST record");
1073 int Opc = GetDecodedCastOpcode(Record[0]);
1074 if (Opc < 0) {
Owen Andersonb99ecca2009-07-30 23:03:37 +00001075 V = UndefValue::get(CurTy); // Unknown cast.
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001076 } else {
1077 const Type *OpTy = getTypeByID(Record[1]);
1078 if (!OpTy) return Error("Invalid CE_CAST record");
1079 Constant *Op = ValueList.getConstantFwdRef(Record[2], OpTy);
Owen Anderson02b48c32009-07-29 18:55:55 +00001080 V = ConstantExpr::getCast(Opc, Op, CurTy);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001081 }
1082 break;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001083 }
Dan Gohman106b2ae2009-07-27 21:53:46 +00001084 case bitc::CST_CODE_CE_INBOUNDS_GEP:
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001085 case bitc::CST_CODE_CE_GEP: { // CE_GEP: [n x operands]
1086 if (Record.size() & 1) return Error("Invalid CE_GEP record");
1087 SmallVector<Constant*, 16> Elts;
1088 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
1089 const Type *ElTy = getTypeByID(Record[i]);
1090 if (!ElTy) return Error("Invalid CE_GEP record");
1091 Elts.push_back(ValueList.getConstantFwdRef(Record[i+1], ElTy));
1092 }
Dan Gohman106b2ae2009-07-27 21:53:46 +00001093 if (BitCode == bitc::CST_CODE_CE_INBOUNDS_GEP)
Dan Gohmanf3a08b82009-09-07 23:54:19 +00001094 V = ConstantExpr::getInBoundsGetElementPtr(Elts[0], &Elts[1],
1095 Elts.size()-1);
1096 else
1097 V = ConstantExpr::getGetElementPtr(Elts[0], &Elts[1],
1098 Elts.size()-1);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001099 break;
1100 }
1101 case bitc::CST_CODE_CE_SELECT: // CE_SELECT: [opval#, opval#, opval#]
1102 if (Record.size() < 3) return Error("Invalid CE_SELECT record");
Owen Anderson02b48c32009-07-29 18:55:55 +00001103 V = ConstantExpr::getSelect(ValueList.getConstantFwdRef(Record[0],
Owen Anderson35b47072009-08-13 21:58:54 +00001104 Type::getInt1Ty(Context)),
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001105 ValueList.getConstantFwdRef(Record[1],CurTy),
1106 ValueList.getConstantFwdRef(Record[2],CurTy));
1107 break;
1108 case bitc::CST_CODE_CE_EXTRACTELT: { // CE_EXTRACTELT: [opty, opval, opval]
1109 if (Record.size() < 3) return Error("Invalid CE_EXTRACTELT record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001110 const VectorType *OpTy =
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001111 dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
1112 if (OpTy == 0) return Error("Invalid CE_EXTRACTELT record");
1113 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
Owen Anderson35b47072009-08-13 21:58:54 +00001114 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
Owen Anderson02b48c32009-07-29 18:55:55 +00001115 V = ConstantExpr::getExtractElement(Op0, Op1);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001116 break;
1117 }
1118 case bitc::CST_CODE_CE_INSERTELT: { // CE_INSERTELT: [opval, opval, opval]
1119 const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1120 if (Record.size() < 3 || OpTy == 0)
1121 return Error("Invalid CE_INSERTELT record");
1122 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1123 Constant *Op1 = ValueList.getConstantFwdRef(Record[1],
1124 OpTy->getElementType());
Owen Anderson35b47072009-08-13 21:58:54 +00001125 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
Owen Anderson02b48c32009-07-29 18:55:55 +00001126 V = ConstantExpr::getInsertElement(Op0, Op1, Op2);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001127 break;
1128 }
1129 case bitc::CST_CODE_CE_SHUFFLEVEC: { // CE_SHUFFLEVEC: [opval, opval, opval]
1130 const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1131 if (Record.size() < 3 || OpTy == 0)
Nate Begemand6d715b2009-02-12 21:28:33 +00001132 return Error("Invalid CE_SHUFFLEVEC record");
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001133 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1134 Constant *Op1 = ValueList.getConstantFwdRef(Record[1], OpTy);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001135 const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
Owen Anderson4956cde2009-07-07 20:18:58 +00001136 OpTy->getNumElements());
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001137 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], ShufTy);
Owen Anderson02b48c32009-07-29 18:55:55 +00001138 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001139 break;
1140 }
Nate Begemand6d715b2009-02-12 21:28:33 +00001141 case bitc::CST_CODE_CE_SHUFVEC_EX: { // [opty, opval, opval, opval]
1142 const VectorType *RTy = dyn_cast<VectorType>(CurTy);
1143 const VectorType *OpTy = dyn_cast<VectorType>(getTypeByID(Record[0]));
1144 if (Record.size() < 4 || RTy == 0 || OpTy == 0)
1145 return Error("Invalid CE_SHUFVEC_EX record");
1146 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1147 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001148 const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
Owen Anderson4956cde2009-07-07 20:18:58 +00001149 RTy->getNumElements());
Nate Begemand6d715b2009-02-12 21:28:33 +00001150 Constant *Op2 = ValueList.getConstantFwdRef(Record[3], ShufTy);
Owen Anderson02b48c32009-07-29 18:55:55 +00001151 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
Nate Begemand6d715b2009-02-12 21:28:33 +00001152 break;
1153 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001154 case bitc::CST_CODE_CE_CMP: { // CE_CMP: [opty, opval, opval, pred]
1155 if (Record.size() < 4) return Error("Invalid CE_CMP record");
1156 const Type *OpTy = getTypeByID(Record[0]);
1157 if (OpTy == 0) return Error("Invalid CE_CMP record");
1158 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1159 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1160
1161 if (OpTy->isFloatingPoint())
Owen Anderson02b48c32009-07-29 18:55:55 +00001162 V = ConstantExpr::getFCmp(Record[3], Op0, Op1);
Nate Begeman646fa482008-05-12 19:01:56 +00001163 else
Owen Anderson02b48c32009-07-29 18:55:55 +00001164 V = ConstantExpr::getICmp(Record[3], Op0, Op1);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001165 break;
1166 }
1167 case bitc::CST_CODE_INLINEASM: {
1168 if (Record.size() < 2) return Error("Invalid INLINEASM record");
1169 std::string AsmStr, ConstrStr;
1170 bool HasSideEffects = Record[0];
1171 unsigned AsmStrSize = Record[1];
1172 if (2+AsmStrSize >= Record.size())
1173 return Error("Invalid INLINEASM record");
1174 unsigned ConstStrSize = Record[2+AsmStrSize];
1175 if (3+AsmStrSize+ConstStrSize > Record.size())
1176 return Error("Invalid INLINEASM record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001177
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001178 for (unsigned i = 0; i != AsmStrSize; ++i)
1179 AsmStr += (char)Record[2+i];
1180 for (unsigned i = 0; i != ConstStrSize; ++i)
1181 ConstrStr += (char)Record[3+AsmStrSize+i];
1182 const PointerType *PTy = cast<PointerType>(CurTy);
1183 V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
1184 AsmStr, ConstrStr, HasSideEffects);
1185 break;
1186 }
1187 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001188
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001189 ValueList.AssignValue(V, NextCstNo);
1190 ++NextCstNo;
1191 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001192
Chris Lattner345bbb72008-08-21 02:34:16 +00001193 if (NextCstNo != ValueList.size())
1194 return Error("Invalid constant reference!");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001195
Chris Lattner345bbb72008-08-21 02:34:16 +00001196 if (Stream.ReadBlockEnd())
1197 return Error("Error at end of constants block");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001198
Chris Lattner345bbb72008-08-21 02:34:16 +00001199 // Once all the constants have been read, go through and resolve forward
1200 // references.
1201 ValueList.ResolveConstantForwardRefs();
1202 return false;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001203}
1204
1205/// RememberAndSkipFunctionBody - When we see the block for a function body,
1206/// remember where it is and then skip it. This lets us lazily deserialize the
1207/// functions.
1208bool BitcodeReader::RememberAndSkipFunctionBody() {
1209 // Get the function we are talking about.
1210 if (FunctionsWithBodies.empty())
1211 return Error("Insufficient function protos");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001212
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001213 Function *Fn = FunctionsWithBodies.back();
1214 FunctionsWithBodies.pop_back();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001215
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001216 // Save the current stream state.
1217 uint64_t CurBit = Stream.GetCurrentBitNo();
1218 DeferredFunctionInfo[Fn] = std::make_pair(CurBit, Fn->getLinkage());
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001219
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001220 // Set the functions linkage to GhostLinkage so we know it is lazily
1221 // deserialized.
1222 Fn->setLinkage(GlobalValue::GhostLinkage);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001223
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001224 // Skip over the function block for now.
1225 if (Stream.SkipBlock())
1226 return Error("Malformed block record");
1227 return false;
1228}
1229
1230bool BitcodeReader::ParseModule(const std::string &ModuleID) {
1231 // Reject multiple MODULE_BLOCK's in a single bitstream.
1232 if (TheModule)
1233 return Error("Multiple MODULE_BLOCKs in same stream");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001234
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001235 if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
1236 return Error("Malformed block record");
1237
1238 // Otherwise, create the module.
Owen Anderson25209b42009-07-01 16:58:40 +00001239 TheModule = new Module(ModuleID, Context);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001240
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001241 SmallVector<uint64_t, 64> Record;
1242 std::vector<std::string> SectionTable;
Gordon Henriksen1aed5992008-08-17 18:44:35 +00001243 std::vector<std::string> GCTable;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001244
1245 // Read all the records for this module.
1246 while (!Stream.AtEndOfStream()) {
1247 unsigned Code = Stream.ReadCode();
1248 if (Code == bitc::END_BLOCK) {
1249 if (Stream.ReadBlockEnd())
1250 return Error("Error at end of module block");
1251
1252 // Patch the initializers for globals and aliases up.
1253 ResolveGlobalAndAliasInits();
1254 if (!GlobalInits.empty() || !AliasInits.empty())
1255 return Error("Malformed global initializer set");
1256 if (!FunctionsWithBodies.empty())
1257 return Error("Too few function bodies found");
1258
Chandler Carrutha228e392007-08-04 01:51:18 +00001259 // Look for intrinsic functions which need to be upgraded at some point
1260 for (Module::iterator FI = TheModule->begin(), FE = TheModule->end();
1261 FI != FE; ++FI) {
Evan Cheng7f05d852007-12-17 22:33:23 +00001262 Function* NewFn;
1263 if (UpgradeIntrinsicFunction(FI, NewFn))
Chandler Carrutha228e392007-08-04 01:51:18 +00001264 UpgradedIntrinsics.push_back(std::make_pair(FI, NewFn));
1265 }
1266
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001267 // Force deallocation of memory for these vectors to favor the client that
1268 // want lazy deserialization.
1269 std::vector<std::pair<GlobalVariable*, unsigned> >().swap(GlobalInits);
1270 std::vector<std::pair<GlobalAlias*, unsigned> >().swap(AliasInits);
1271 std::vector<Function*>().swap(FunctionsWithBodies);
1272 return false;
1273 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001274
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001275 if (Code == bitc::ENTER_SUBBLOCK) {
1276 switch (Stream.ReadSubBlockID()) {
1277 default: // Skip unknown content.
1278 if (Stream.SkipBlock())
1279 return Error("Malformed block record");
1280 break;
1281 case bitc::BLOCKINFO_BLOCK_ID:
1282 if (Stream.ReadBlockInfoBlock())
1283 return Error("Malformed BlockInfoBlock");
1284 break;
1285 case bitc::PARAMATTR_BLOCK_ID:
Devang Pateld222f862008-09-25 21:00:45 +00001286 if (ParseAttributeBlock())
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001287 return true;
1288 break;
1289 case bitc::TYPE_BLOCK_ID:
1290 if (ParseTypeTable())
1291 return true;
1292 break;
1293 case bitc::TYPE_SYMTAB_BLOCK_ID:
1294 if (ParseTypeSymbolTable())
1295 return true;
1296 break;
1297 case bitc::VALUE_SYMTAB_BLOCK_ID:
1298 if (ParseValueSymbolTable())
1299 return true;
1300 break;
1301 case bitc::CONSTANTS_BLOCK_ID:
1302 if (ParseConstants() || ResolveGlobalAndAliasInits())
1303 return true;
1304 break;
Devang Patel0c0a6ca2009-07-22 17:43:22 +00001305 case bitc::METADATA_BLOCK_ID:
1306 if (ParseMetadata())
1307 return true;
1308 break;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001309 case bitc::FUNCTION_BLOCK_ID:
1310 // If this is the first function body we've seen, reverse the
1311 // FunctionsWithBodies list.
1312 if (!HasReversedFunctionsWithBodies) {
1313 std::reverse(FunctionsWithBodies.begin(), FunctionsWithBodies.end());
1314 HasReversedFunctionsWithBodies = true;
1315 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001316
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001317 if (RememberAndSkipFunctionBody())
1318 return true;
1319 break;
1320 }
1321 continue;
1322 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001323
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001324 if (Code == bitc::DEFINE_ABBREV) {
1325 Stream.ReadAbbrevRecord();
1326 continue;
1327 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001328
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001329 // Read a record.
1330 switch (Stream.ReadRecord(Code, Record)) {
1331 default: break; // Default behavior, ignore unknown content.
1332 case bitc::MODULE_CODE_VERSION: // VERSION: [version#]
1333 if (Record.size() < 1)
1334 return Error("Malformed MODULE_CODE_VERSION");
1335 // Only version #0 is supported so far.
1336 if (Record[0] != 0)
1337 return Error("Unknown bitstream version!");
1338 break;
1339 case bitc::MODULE_CODE_TRIPLE: { // TRIPLE: [strchr x N]
1340 std::string S;
1341 if (ConvertToString(Record, 0, S))
1342 return Error("Invalid MODULE_CODE_TRIPLE record");
1343 TheModule->setTargetTriple(S);
1344 break;
1345 }
1346 case bitc::MODULE_CODE_DATALAYOUT: { // DATALAYOUT: [strchr x N]
1347 std::string S;
1348 if (ConvertToString(Record, 0, S))
1349 return Error("Invalid MODULE_CODE_DATALAYOUT record");
1350 TheModule->setDataLayout(S);
1351 break;
1352 }
1353 case bitc::MODULE_CODE_ASM: { // ASM: [strchr x N]
1354 std::string S;
1355 if (ConvertToString(Record, 0, S))
1356 return Error("Invalid MODULE_CODE_ASM record");
1357 TheModule->setModuleInlineAsm(S);
1358 break;
1359 }
1360 case bitc::MODULE_CODE_DEPLIB: { // DEPLIB: [strchr x N]
1361 std::string S;
1362 if (ConvertToString(Record, 0, S))
1363 return Error("Invalid MODULE_CODE_DEPLIB record");
1364 TheModule->addLibrary(S);
1365 break;
1366 }
1367 case bitc::MODULE_CODE_SECTIONNAME: { // SECTIONNAME: [strchr x N]
1368 std::string S;
1369 if (ConvertToString(Record, 0, S))
1370 return Error("Invalid MODULE_CODE_SECTIONNAME record");
1371 SectionTable.push_back(S);
1372 break;
1373 }
Gordon Henriksen1aed5992008-08-17 18:44:35 +00001374 case bitc::MODULE_CODE_GCNAME: { // SECTIONNAME: [strchr x N]
Gordon Henriksen13fe5e32007-12-10 03:18:06 +00001375 std::string S;
1376 if (ConvertToString(Record, 0, S))
Gordon Henriksen1aed5992008-08-17 18:44:35 +00001377 return Error("Invalid MODULE_CODE_GCNAME record");
1378 GCTable.push_back(S);
Gordon Henriksen13fe5e32007-12-10 03:18:06 +00001379 break;
1380 }
Christopher Lamb44d62f62007-12-11 08:59:05 +00001381 // GLOBALVAR: [pointer type, isconst, initid,
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001382 // linkage, alignment, section, visibility, threadlocal]
1383 case bitc::MODULE_CODE_GLOBALVAR: {
1384 if (Record.size() < 6)
1385 return Error("Invalid MODULE_CODE_GLOBALVAR record");
1386 const Type *Ty = getTypeByID(Record[0]);
1387 if (!isa<PointerType>(Ty))
1388 return Error("Global not a pointer type!");
Christopher Lamb44d62f62007-12-11 08:59:05 +00001389 unsigned AddressSpace = cast<PointerType>(Ty)->getAddressSpace();
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001390 Ty = cast<PointerType>(Ty)->getElementType();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001391
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001392 bool isConstant = Record[1];
1393 GlobalValue::LinkageTypes Linkage = GetDecodedLinkage(Record[3]);
1394 unsigned Alignment = (1 << Record[4]) >> 1;
1395 std::string Section;
1396 if (Record[5]) {
1397 if (Record[5]-1 >= SectionTable.size())
1398 return Error("Invalid section ID");
1399 Section = SectionTable[Record[5]-1];
1400 }
1401 GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
1402 if (Record.size() > 6)
1403 Visibility = GetDecodedVisibility(Record[6]);
1404 bool isThreadLocal = false;
1405 if (Record.size() > 7)
1406 isThreadLocal = Record[7];
1407
1408 GlobalVariable *NewGV =
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001409 new GlobalVariable(*TheModule, Ty, isConstant, Linkage, 0, "", 0,
Christopher Lamb44d62f62007-12-11 08:59:05 +00001410 isThreadLocal, AddressSpace);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001411 NewGV->setAlignment(Alignment);
1412 if (!Section.empty())
1413 NewGV->setSection(Section);
1414 NewGV->setVisibility(Visibility);
1415 NewGV->setThreadLocal(isThreadLocal);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001416
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001417 ValueList.push_back(NewGV);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001418
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001419 // Remember which value to use for the global initializer.
1420 if (unsigned InitID = Record[2])
1421 GlobalInits.push_back(std::make_pair(NewGV, InitID-1));
1422 break;
1423 }
1424 // FUNCTION: [type, callingconv, isproto, linkage, paramattr,
Gordon Henriksen1aed5992008-08-17 18:44:35 +00001425 // alignment, section, visibility, gc]
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001426 case bitc::MODULE_CODE_FUNCTION: {
1427 if (Record.size() < 8)
1428 return Error("Invalid MODULE_CODE_FUNCTION record");
1429 const Type *Ty = getTypeByID(Record[0]);
1430 if (!isa<PointerType>(Ty))
1431 return Error("Function not a pointer type!");
1432 const FunctionType *FTy =
1433 dyn_cast<FunctionType>(cast<PointerType>(Ty)->getElementType());
1434 if (!FTy)
1435 return Error("Function not a pointer to function type!");
1436
Gabor Greifd6da1d02008-04-06 20:25:17 +00001437 Function *Func = Function::Create(FTy, GlobalValue::ExternalLinkage,
1438 "", TheModule);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001439
Sandeep Patel5838baa2009-09-02 08:44:58 +00001440 Func->setCallingConv(static_cast<CallingConv::ID>(Record[1]));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001441 bool isProto = Record[2];
1442 Func->setLinkage(GetDecodedLinkage(Record[3]));
Devang Pateld222f862008-09-25 21:00:45 +00001443 Func->setAttributes(getAttributes(Record[4]));
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001444
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001445 Func->setAlignment((1 << Record[5]) >> 1);
1446 if (Record[6]) {
1447 if (Record[6]-1 >= SectionTable.size())
1448 return Error("Invalid section ID");
1449 Func->setSection(SectionTable[Record[6]-1]);
1450 }
1451 Func->setVisibility(GetDecodedVisibility(Record[7]));
Gordon Henriksen13fe5e32007-12-10 03:18:06 +00001452 if (Record.size() > 8 && Record[8]) {
Gordon Henriksen1aed5992008-08-17 18:44:35 +00001453 if (Record[8]-1 > GCTable.size())
1454 return Error("Invalid GC ID");
1455 Func->setGC(GCTable[Record[8]-1].c_str());
Gordon Henriksen13fe5e32007-12-10 03:18:06 +00001456 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001457 ValueList.push_back(Func);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001458
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001459 // If this is a function with a body, remember the prototype we are
1460 // creating now, so that we can match up the body with them later.
1461 if (!isProto)
1462 FunctionsWithBodies.push_back(Func);
1463 break;
1464 }
Anton Korobeynikov5789f8d2008-03-12 00:49:19 +00001465 // ALIAS: [alias type, aliasee val#, linkage]
Anton Korobeynikov902a0112008-03-11 21:40:17 +00001466 // ALIAS: [alias type, aliasee val#, linkage, visibility]
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001467 case bitc::MODULE_CODE_ALIAS: {
1468 if (Record.size() < 3)
1469 return Error("Invalid MODULE_ALIAS record");
1470 const Type *Ty = getTypeByID(Record[0]);
1471 if (!isa<PointerType>(Ty))
1472 return Error("Function not a pointer type!");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001473
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001474 GlobalAlias *NewGA = new GlobalAlias(Ty, GetDecodedLinkage(Record[2]),
1475 "", 0, TheModule);
Anton Korobeynikov5789f8d2008-03-12 00:49:19 +00001476 // Old bitcode files didn't have visibility field.
1477 if (Record.size() > 3)
1478 NewGA->setVisibility(GetDecodedVisibility(Record[3]));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001479 ValueList.push_back(NewGA);
1480 AliasInits.push_back(std::make_pair(NewGA, Record[1]));
1481 break;
1482 }
1483 /// MODULE_CODE_PURGEVALS: [numvals]
1484 case bitc::MODULE_CODE_PURGEVALS:
1485 // Trim down the value list to the specified size.
1486 if (Record.size() < 1 || Record[0] > ValueList.size())
1487 return Error("Invalid MODULE_PURGEVALS record");
1488 ValueList.shrinkTo(Record[0]);
1489 break;
1490 }
1491 Record.clear();
1492 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001493
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001494 return Error("Premature end of bitstream");
1495}
1496
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001497bool BitcodeReader::ParseBitcode() {
1498 TheModule = 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001499
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001500 if (Buffer->getBufferSize() & 3)
1501 return Error("Bitcode stream should be a multiple of 4 bytes in length");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001502
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001503 unsigned char *BufPtr = (unsigned char *)Buffer->getBufferStart();
Chris Lattner65b13ff2008-07-09 05:14:23 +00001504 unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001505
Chris Lattner65b13ff2008-07-09 05:14:23 +00001506 // If we have a wrapper header, parse it and ignore the non-bc file contents.
1507 // The magic number is 0x0B17C0DE stored in little endian.
Chris Lattnerabd0e442009-04-06 20:54:32 +00001508 if (isBitcodeWrapper(BufPtr, BufEnd))
1509 if (SkipBitcodeWrapperHeader(BufPtr, BufEnd))
Chris Lattner65b13ff2008-07-09 05:14:23 +00001510 return Error("Invalid bitcode wrapper header");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001511
Chris Lattner25a6abf2009-04-26 20:59:02 +00001512 StreamFile.init(BufPtr, BufEnd);
1513 Stream.init(StreamFile);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001514
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001515 // Sniff for the signature.
1516 if (Stream.Read(8) != 'B' ||
1517 Stream.Read(8) != 'C' ||
1518 Stream.Read(4) != 0x0 ||
1519 Stream.Read(4) != 0xC ||
1520 Stream.Read(4) != 0xE ||
1521 Stream.Read(4) != 0xD)
1522 return Error("Invalid bitcode signature");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001523
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001524 // We expect a number of well-defined blocks, though we don't necessarily
1525 // need to understand them all.
1526 while (!Stream.AtEndOfStream()) {
1527 unsigned Code = Stream.ReadCode();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001528
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001529 if (Code != bitc::ENTER_SUBBLOCK)
1530 return Error("Invalid record at top-level");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001531
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001532 unsigned BlockID = Stream.ReadSubBlockID();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001533
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001534 // We only know the MODULE subblock ID.
1535 switch (BlockID) {
1536 case bitc::BLOCKINFO_BLOCK_ID:
1537 if (Stream.ReadBlockInfoBlock())
1538 return Error("Malformed BlockInfoBlock");
1539 break;
1540 case bitc::MODULE_BLOCK_ID:
1541 if (ParseModule(Buffer->getBufferIdentifier()))
1542 return true;
1543 break;
1544 default:
1545 if (Stream.SkipBlock())
1546 return Error("Malformed block record");
1547 break;
1548 }
1549 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001550
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001551 return false;
1552}
1553
Devang Pateladc37612009-09-18 19:26:43 +00001554/// ParseMetadataAttachment - Parse metadata attachments.
1555bool BitcodeReader::ParseMetadataAttachment() {
1556 if (Stream.EnterSubBlock(bitc::METADATA_ATTACHMENT_ID))
1557 return Error("Malformed block record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001558
Devang Pateladc37612009-09-18 19:26:43 +00001559 Metadata &TheMetadata = Context.getMetadata();
1560 SmallVector<uint64_t, 64> Record;
1561 while(1) {
1562 unsigned Code = Stream.ReadCode();
1563 if (Code == bitc::END_BLOCK) {
1564 if (Stream.ReadBlockEnd())
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001565 return Error("Error at end of PARAMATTR block");
Devang Pateladc37612009-09-18 19:26:43 +00001566 break;
1567 }
1568 if (Code == bitc::DEFINE_ABBREV) {
1569 Stream.ReadAbbrevRecord();
1570 continue;
1571 }
1572 // Read a metadata attachment record.
1573 Record.clear();
1574 switch (Stream.ReadRecord(Code, Record)) {
1575 default: // Default behavior: ignore.
1576 break;
1577 case bitc::METADATA_ATTACHMENT: {
1578 unsigned RecordLength = Record.size();
1579 if (Record.empty() || (RecordLength - 1) % 2 == 1)
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001580 return Error ("Invalid METADATA_ATTACHMENT reader!");
Devang Pateladc37612009-09-18 19:26:43 +00001581 Instruction *Inst = InstructionList[Record[0]];
1582 for (unsigned i = 1; i != RecordLength; i = i+2) {
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001583 MDKindID Kind = Record[i];
1584 Value *Node = MDValueList.getValueFwdRef(Record[i+1]);
1585 TheMetadata.setMD(Kind, cast<MDNode>(Node), Inst);
Devang Pateladc37612009-09-18 19:26:43 +00001586 }
1587 break;
1588 }
1589 }
1590 }
1591 return false;
1592}
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001593
1594/// ParseFunctionBody - Lazily parse the specified function body block.
1595bool BitcodeReader::ParseFunctionBody(Function *F) {
1596 if (Stream.EnterSubBlock(bitc::FUNCTION_BLOCK_ID))
1597 return Error("Malformed block record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001598
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001599 unsigned ModuleValueListSize = ValueList.size();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001600
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001601 // Add all the function arguments to the value table.
1602 for(Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
1603 ValueList.push_back(I);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001604
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001605 unsigned NextValueNo = ValueList.size();
1606 BasicBlock *CurBB = 0;
1607 unsigned CurBBNo = 0;
1608
1609 // Read all the records.
1610 SmallVector<uint64_t, 64> Record;
1611 while (1) {
1612 unsigned Code = Stream.ReadCode();
1613 if (Code == bitc::END_BLOCK) {
1614 if (Stream.ReadBlockEnd())
1615 return Error("Error at end of function block");
1616 break;
1617 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001618
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001619 if (Code == bitc::ENTER_SUBBLOCK) {
1620 switch (Stream.ReadSubBlockID()) {
1621 default: // Skip unknown content.
1622 if (Stream.SkipBlock())
1623 return Error("Malformed block record");
1624 break;
1625 case bitc::CONSTANTS_BLOCK_ID:
1626 if (ParseConstants()) return true;
1627 NextValueNo = ValueList.size();
1628 break;
1629 case bitc::VALUE_SYMTAB_BLOCK_ID:
1630 if (ParseValueSymbolTable()) return true;
1631 break;
Devang Pateladc37612009-09-18 19:26:43 +00001632 case bitc::METADATA_ATTACHMENT_ID:
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001633 if (ParseMetadataAttachment()) return true;
1634 break;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001635 }
1636 continue;
1637 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001638
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001639 if (Code == bitc::DEFINE_ABBREV) {
1640 Stream.ReadAbbrevRecord();
1641 continue;
1642 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001643
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001644 // Read a record.
1645 Record.clear();
1646 Instruction *I = 0;
Dan Gohman799a8c82009-07-20 21:19:07 +00001647 unsigned BitCode = Stream.ReadRecord(Code, Record);
1648 switch (BitCode) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001649 default: // Default behavior: reject
1650 return Error("Unknown instruction");
1651 case bitc::FUNC_CODE_DECLAREBLOCKS: // DECLAREBLOCKS: [nblocks]
1652 if (Record.size() < 1 || Record[0] == 0)
1653 return Error("Invalid DECLAREBLOCKS record");
1654 // Create all the basic blocks for the function.
1655 FunctionBBs.resize(Record[0]);
1656 for (unsigned i = 0, e = FunctionBBs.size(); i != e; ++i)
Owen Anderson35b47072009-08-13 21:58:54 +00001657 FunctionBBs[i] = BasicBlock::Create(Context, "", F);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001658 CurBB = FunctionBBs[0];
1659 continue;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001660
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001661 case bitc::FUNC_CODE_INST_BINOP: { // BINOP: [opval, ty, opval, opcode]
1662 unsigned OpNum = 0;
1663 Value *LHS, *RHS;
1664 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1665 getValue(Record, OpNum, LHS->getType(), RHS) ||
Dan Gohman799a8c82009-07-20 21:19:07 +00001666 OpNum+1 > Record.size())
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001667 return Error("Invalid BINOP record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001668
Dan Gohman799a8c82009-07-20 21:19:07 +00001669 int Opc = GetDecodedBinaryOpcode(Record[OpNum++], LHS->getType());
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001670 if (Opc == -1) return Error("Invalid BINOP record");
Gabor Greifa645dd32008-05-16 19:29:10 +00001671 I = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
Devang Pateladc37612009-09-18 19:26:43 +00001672 InstructionList.push_back(I);
Dan Gohmanf3a08b82009-09-07 23:54:19 +00001673 if (OpNum < Record.size()) {
1674 if (Opc == Instruction::Add ||
1675 Opc == Instruction::Sub ||
1676 Opc == Instruction::Mul) {
1677 if (Record[3] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1678 cast<BinaryOperator>(I)->setHasNoSignedWrap(true);
1679 if (Record[3] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1680 cast<BinaryOperator>(I)->setHasNoUnsignedWrap(true);
1681 } else if (Opc == Instruction::SDiv) {
1682 if (Record[3] & (1 << bitc::SDIV_EXACT))
1683 cast<BinaryOperator>(I)->setIsExact(true);
1684 }
1685 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001686 break;
1687 }
1688 case bitc::FUNC_CODE_INST_CAST: { // CAST: [opval, opty, destty, castopc]
1689 unsigned OpNum = 0;
1690 Value *Op;
1691 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
1692 OpNum+2 != Record.size())
1693 return Error("Invalid CAST record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001694
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001695 const Type *ResTy = getTypeByID(Record[OpNum]);
1696 int Opc = GetDecodedCastOpcode(Record[OpNum+1]);
1697 if (Opc == -1 || ResTy == 0)
1698 return Error("Invalid CAST record");
Gabor Greifa645dd32008-05-16 19:29:10 +00001699 I = CastInst::Create((Instruction::CastOps)Opc, Op, ResTy);
Devang Pateladc37612009-09-18 19:26:43 +00001700 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001701 break;
1702 }
Dan Gohman106b2ae2009-07-27 21:53:46 +00001703 case bitc::FUNC_CODE_INST_INBOUNDS_GEP:
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001704 case bitc::FUNC_CODE_INST_GEP: { // GEP: [n x operands]
1705 unsigned OpNum = 0;
1706 Value *BasePtr;
1707 if (getValueTypePair(Record, OpNum, NextValueNo, BasePtr))
1708 return Error("Invalid GEP record");
1709
1710 SmallVector<Value*, 16> GEPIdx;
1711 while (OpNum != Record.size()) {
1712 Value *Op;
1713 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1714 return Error("Invalid GEP record");
1715 GEPIdx.push_back(Op);
1716 }
1717
Gabor Greifd6da1d02008-04-06 20:25:17 +00001718 I = GetElementPtrInst::Create(BasePtr, GEPIdx.begin(), GEPIdx.end());
Devang Pateladc37612009-09-18 19:26:43 +00001719 InstructionList.push_back(I);
Dan Gohman106b2ae2009-07-27 21:53:46 +00001720 if (BitCode == bitc::FUNC_CODE_INST_INBOUNDS_GEP)
Dan Gohmanf3a08b82009-09-07 23:54:19 +00001721 cast<GetElementPtrInst>(I)->setIsInBounds(true);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001722 break;
1723 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001724
Dan Gohmane5febe42008-05-31 00:58:22 +00001725 case bitc::FUNC_CODE_INST_EXTRACTVAL: {
1726 // EXTRACTVAL: [opty, opval, n x indices]
Dan Gohmane6b1ee62008-05-23 01:55:30 +00001727 unsigned OpNum = 0;
1728 Value *Agg;
1729 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1730 return Error("Invalid EXTRACTVAL record");
1731
Dan Gohmane5febe42008-05-31 00:58:22 +00001732 SmallVector<unsigned, 4> EXTRACTVALIdx;
1733 for (unsigned RecSize = Record.size();
1734 OpNum != RecSize; ++OpNum) {
1735 uint64_t Index = Record[OpNum];
1736 if ((unsigned)Index != Index)
1737 return Error("Invalid EXTRACTVAL index");
1738 EXTRACTVALIdx.push_back((unsigned)Index);
Dan Gohmane6b1ee62008-05-23 01:55:30 +00001739 }
1740
1741 I = ExtractValueInst::Create(Agg,
1742 EXTRACTVALIdx.begin(), EXTRACTVALIdx.end());
Devang Pateladc37612009-09-18 19:26:43 +00001743 InstructionList.push_back(I);
Dan Gohmane6b1ee62008-05-23 01:55:30 +00001744 break;
1745 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001746
Dan Gohmane5febe42008-05-31 00:58:22 +00001747 case bitc::FUNC_CODE_INST_INSERTVAL: {
1748 // INSERTVAL: [opty, opval, opty, opval, n x indices]
Dan Gohmane6b1ee62008-05-23 01:55:30 +00001749 unsigned OpNum = 0;
1750 Value *Agg;
1751 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1752 return Error("Invalid INSERTVAL record");
1753 Value *Val;
1754 if (getValueTypePair(Record, OpNum, NextValueNo, Val))
1755 return Error("Invalid INSERTVAL record");
1756
Dan Gohmane5febe42008-05-31 00:58:22 +00001757 SmallVector<unsigned, 4> INSERTVALIdx;
1758 for (unsigned RecSize = Record.size();
1759 OpNum != RecSize; ++OpNum) {
1760 uint64_t Index = Record[OpNum];
1761 if ((unsigned)Index != Index)
1762 return Error("Invalid INSERTVAL index");
1763 INSERTVALIdx.push_back((unsigned)Index);
Dan Gohmane6b1ee62008-05-23 01:55:30 +00001764 }
1765
1766 I = InsertValueInst::Create(Agg, Val,
1767 INSERTVALIdx.begin(), INSERTVALIdx.end());
Devang Pateladc37612009-09-18 19:26:43 +00001768 InstructionList.push_back(I);
Dan Gohmane6b1ee62008-05-23 01:55:30 +00001769 break;
1770 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001771
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001772 case bitc::FUNC_CODE_INST_SELECT: { // SELECT: [opval, ty, opval, opval]
Dan Gohman6fa5bce2008-09-16 01:01:33 +00001773 // obsolete form of select
1774 // handles select i1 ... in old bitcode
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001775 unsigned OpNum = 0;
1776 Value *TrueVal, *FalseVal, *Cond;
1777 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1778 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
Owen Anderson35b47072009-08-13 21:58:54 +00001779 getValue(Record, OpNum, Type::getInt1Ty(Context), Cond))
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001780 return Error("Invalid SELECT record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001781
Dan Gohman6fa5bce2008-09-16 01:01:33 +00001782 I = SelectInst::Create(Cond, TrueVal, FalseVal);
Devang Pateladc37612009-09-18 19:26:43 +00001783 InstructionList.push_back(I);
Dan Gohman6fa5bce2008-09-16 01:01:33 +00001784 break;
1785 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001786
Dan Gohman6fa5bce2008-09-16 01:01:33 +00001787 case bitc::FUNC_CODE_INST_VSELECT: {// VSELECT: [ty,opval,opval,predty,pred]
1788 // new form of select
1789 // handles select i1 or select [N x i1]
1790 unsigned OpNum = 0;
1791 Value *TrueVal, *FalseVal, *Cond;
1792 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1793 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
1794 getValueTypePair(Record, OpNum, NextValueNo, Cond))
1795 return Error("Invalid SELECT record");
Dan Gohmanb60ca3c2008-09-09 01:02:47 +00001796
1797 // select condition can be either i1 or [N x i1]
Dan Gohman6fa5bce2008-09-16 01:01:33 +00001798 if (const VectorType* vector_type =
1799 dyn_cast<const VectorType>(Cond->getType())) {
Dan Gohmanb60ca3c2008-09-09 01:02:47 +00001800 // expect <n x i1>
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001801 if (vector_type->getElementType() != Type::getInt1Ty(Context))
Dan Gohmanb60ca3c2008-09-09 01:02:47 +00001802 return Error("Invalid SELECT condition type");
1803 } else {
1804 // expect i1
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001805 if (Cond->getType() != Type::getInt1Ty(Context))
Dan Gohmanb60ca3c2008-09-09 01:02:47 +00001806 return Error("Invalid SELECT condition type");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001807 }
1808
Gabor Greifd6da1d02008-04-06 20:25:17 +00001809 I = SelectInst::Create(Cond, TrueVal, FalseVal);
Devang Pateladc37612009-09-18 19:26:43 +00001810 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001811 break;
1812 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001813
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001814 case bitc::FUNC_CODE_INST_EXTRACTELT: { // EXTRACTELT: [opty, opval, opval]
1815 unsigned OpNum = 0;
1816 Value *Vec, *Idx;
1817 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
Owen Anderson35b47072009-08-13 21:58:54 +00001818 getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001819 return Error("Invalid EXTRACTELT record");
Eric Christopher1ba36872009-07-25 02:28:41 +00001820 I = ExtractElementInst::Create(Vec, Idx);
Devang Pateladc37612009-09-18 19:26:43 +00001821 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001822 break;
1823 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001824
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001825 case bitc::FUNC_CODE_INST_INSERTELT: { // INSERTELT: [ty, opval,opval,opval]
1826 unsigned OpNum = 0;
1827 Value *Vec, *Elt, *Idx;
1828 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001829 getValue(Record, OpNum,
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001830 cast<VectorType>(Vec->getType())->getElementType(), Elt) ||
Owen Anderson35b47072009-08-13 21:58:54 +00001831 getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001832 return Error("Invalid INSERTELT record");
Gabor Greifd6da1d02008-04-06 20:25:17 +00001833 I = InsertElementInst::Create(Vec, Elt, Idx);
Devang Pateladc37612009-09-18 19:26:43 +00001834 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001835 break;
1836 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001837
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001838 case bitc::FUNC_CODE_INST_SHUFFLEVEC: {// SHUFFLEVEC: [opval,ty,opval,opval]
1839 unsigned OpNum = 0;
1840 Value *Vec1, *Vec2, *Mask;
1841 if (getValueTypePair(Record, OpNum, NextValueNo, Vec1) ||
1842 getValue(Record, OpNum, Vec1->getType(), Vec2))
1843 return Error("Invalid SHUFFLEVEC record");
1844
Mon P Wangbff5d9c2008-11-10 04:46:22 +00001845 if (getValueTypePair(Record, OpNum, NextValueNo, Mask))
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001846 return Error("Invalid SHUFFLEVEC record");
1847 I = new ShuffleVectorInst(Vec1, Vec2, Mask);
Devang Pateladc37612009-09-18 19:26:43 +00001848 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001849 break;
1850 }
Mon P Wangbff5d9c2008-11-10 04:46:22 +00001851
Nick Lewycky8f5253b2009-07-08 03:04:38 +00001852 case bitc::FUNC_CODE_INST_CMP: // CMP: [opty, opval, opval, pred]
1853 // Old form of ICmp/FCmp returning bool
1854 // Existed to differentiate between icmp/fcmp and vicmp/vfcmp which were
1855 // both legal on vectors but had different behaviour.
1856 case bitc::FUNC_CODE_INST_CMP2: { // CMP2: [opty, opval, opval, pred]
1857 // FCmp/ICmp returning bool or vector of bool
1858
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001859 unsigned OpNum = 0;
1860 Value *LHS, *RHS;
1861 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1862 getValue(Record, OpNum, LHS->getType(), RHS) ||
1863 OpNum+1 != Record.size())
1864 return Error("Invalid CMP record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001865
Dan Gohmanb60ca3c2008-09-09 01:02:47 +00001866 if (LHS->getType()->isFPOrFPVector())
Dan Gohmane6803b82009-08-25 23:17:54 +00001867 I = new FCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
Nick Lewycky8f5253b2009-07-08 03:04:38 +00001868 else
Dan Gohmane6803b82009-08-25 23:17:54 +00001869 I = new ICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
Devang Pateladc37612009-09-18 19:26:43 +00001870 InstructionList.push_back(I);
Dan Gohmanb60ca3c2008-09-09 01:02:47 +00001871 break;
1872 }
Nick Lewycky8f5253b2009-07-08 03:04:38 +00001873
Devang Patel774ee9e2008-02-22 02:49:49 +00001874 case bitc::FUNC_CODE_INST_GETRESULT: { // GETRESULT: [ty, val, n]
1875 if (Record.size() != 2)
1876 return Error("Invalid GETRESULT record");
1877 unsigned OpNum = 0;
1878 Value *Op;
1879 getValueTypePair(Record, OpNum, NextValueNo, Op);
1880 unsigned Index = Record[1];
Dan Gohman29474e92008-07-23 00:34:11 +00001881 I = ExtractValueInst::Create(Op, Index);
Devang Pateladc37612009-09-18 19:26:43 +00001882 InstructionList.push_back(I);
Devang Patel774ee9e2008-02-22 02:49:49 +00001883 break;
1884 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001885
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001886 case bitc::FUNC_CODE_INST_RET: // RET: [opty,opval<optional>]
Devang Patelb7fbc8b2008-02-26 01:29:32 +00001887 {
1888 unsigned Size = Record.size();
1889 if (Size == 0) {
Owen Anderson35b47072009-08-13 21:58:54 +00001890 I = ReturnInst::Create(Context);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001891 InstructionList.push_back(I);
Devang Patelb7fbc8b2008-02-26 01:29:32 +00001892 break;
Dan Gohman29474e92008-07-23 00:34:11 +00001893 }
Devang Patelb7fbc8b2008-02-26 01:29:32 +00001894
Dan Gohman29474e92008-07-23 00:34:11 +00001895 unsigned OpNum = 0;
1896 SmallVector<Value *,4> Vs;
1897 do {
1898 Value *Op = NULL;
1899 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1900 return Error("Invalid RET record");
1901 Vs.push_back(Op);
1902 } while(OpNum != Record.size());
1903
1904 const Type *ReturnType = F->getReturnType();
1905 if (Vs.size() > 1 ||
1906 (isa<StructType>(ReturnType) &&
1907 (Vs.empty() || Vs[0]->getType() != ReturnType))) {
Owen Andersonb99ecca2009-07-30 23:03:37 +00001908 Value *RV = UndefValue::get(ReturnType);
Dan Gohman29474e92008-07-23 00:34:11 +00001909 for (unsigned i = 0, e = Vs.size(); i != e; ++i) {
1910 I = InsertValueInst::Create(RV, Vs[i], i, "mrv");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001911 InstructionList.push_back(I);
Dan Gohman29474e92008-07-23 00:34:11 +00001912 CurBB->getInstList().push_back(I);
1913 ValueList.AssignValue(I, NextValueNo++);
1914 RV = I;
1915 }
Owen Anderson35b47072009-08-13 21:58:54 +00001916 I = ReturnInst::Create(Context, RV);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001917 InstructionList.push_back(I);
Devang Patelb7fbc8b2008-02-26 01:29:32 +00001918 break;
1919 }
Dan Gohman29474e92008-07-23 00:34:11 +00001920
Owen Anderson35b47072009-08-13 21:58:54 +00001921 I = ReturnInst::Create(Context, Vs[0]);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001922 InstructionList.push_back(I);
Dan Gohman29474e92008-07-23 00:34:11 +00001923 break;
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001924 }
1925 case bitc::FUNC_CODE_INST_BR: { // BR: [bb#, bb#, opval] or [bb#]
1926 if (Record.size() != 1 && Record.size() != 3)
1927 return Error("Invalid BR record");
1928 BasicBlock *TrueDest = getBasicBlock(Record[0]);
1929 if (TrueDest == 0)
1930 return Error("Invalid BR record");
1931
Devang Pateladc37612009-09-18 19:26:43 +00001932 if (Record.size() == 1) {
Gabor Greifd6da1d02008-04-06 20:25:17 +00001933 I = BranchInst::Create(TrueDest);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001934 InstructionList.push_back(I);
Devang Pateladc37612009-09-18 19:26:43 +00001935 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001936 else {
1937 BasicBlock *FalseDest = getBasicBlock(Record[1]);
Owen Anderson35b47072009-08-13 21:58:54 +00001938 Value *Cond = getFnValueByID(Record[2], Type::getInt1Ty(Context));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001939 if (FalseDest == 0 || Cond == 0)
1940 return Error("Invalid BR record");
Gabor Greifd6da1d02008-04-06 20:25:17 +00001941 I = BranchInst::Create(TrueDest, FalseDest, Cond);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001942 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001943 }
1944 break;
1945 }
1946 case bitc::FUNC_CODE_INST_SWITCH: { // SWITCH: [opty, opval, n, n x ops]
1947 if (Record.size() < 3 || (Record.size() & 1) == 0)
1948 return Error("Invalid SWITCH record");
1949 const Type *OpTy = getTypeByID(Record[0]);
1950 Value *Cond = getFnValueByID(Record[1], OpTy);
1951 BasicBlock *Default = getBasicBlock(Record[2]);
1952 if (OpTy == 0 || Cond == 0 || Default == 0)
1953 return Error("Invalid SWITCH record");
1954 unsigned NumCases = (Record.size()-3)/2;
Gabor Greifd6da1d02008-04-06 20:25:17 +00001955 SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
Devang Pateladc37612009-09-18 19:26:43 +00001956 InstructionList.push_back(SI);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001957 for (unsigned i = 0, e = NumCases; i != e; ++i) {
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001958 ConstantInt *CaseVal =
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001959 dyn_cast_or_null<ConstantInt>(getFnValueByID(Record[3+i*2], OpTy));
1960 BasicBlock *DestBB = getBasicBlock(Record[1+3+i*2]);
1961 if (CaseVal == 0 || DestBB == 0) {
1962 delete SI;
1963 return Error("Invalid SWITCH record!");
1964 }
1965 SI->addCase(CaseVal, DestBB);
1966 }
1967 I = SI;
1968 break;
1969 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001970
Duncan Sandsf5588dc2007-11-27 13:23:08 +00001971 case bitc::FUNC_CODE_INST_INVOKE: {
1972 // INVOKE: [attrs, cc, normBB, unwindBB, fnty, op0,op1,op2, ...]
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001973 if (Record.size() < 4) return Error("Invalid INVOKE record");
Devang Pateld222f862008-09-25 21:00:45 +00001974 AttrListPtr PAL = getAttributes(Record[0]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001975 unsigned CCInfo = Record[1];
1976 BasicBlock *NormalBB = getBasicBlock(Record[2]);
1977 BasicBlock *UnwindBB = getBasicBlock(Record[3]);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001978
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001979 unsigned OpNum = 4;
1980 Value *Callee;
1981 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
1982 return Error("Invalid INVOKE record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001983
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001984 const PointerType *CalleeTy = dyn_cast<PointerType>(Callee->getType());
1985 const FunctionType *FTy = !CalleeTy ? 0 :
1986 dyn_cast<FunctionType>(CalleeTy->getElementType());
1987
1988 // Check that the right number of fixed parameters are here.
1989 if (FTy == 0 || NormalBB == 0 || UnwindBB == 0 ||
1990 Record.size() < OpNum+FTy->getNumParams())
1991 return Error("Invalid INVOKE record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001992
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001993 SmallVector<Value*, 16> Ops;
1994 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
1995 Ops.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
1996 if (Ops.back() == 0) return Error("Invalid INVOKE record");
1997 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00001998
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001999 if (!FTy->isVarArg()) {
2000 if (Record.size() != OpNum)
2001 return Error("Invalid INVOKE record");
2002 } else {
2003 // Read type/value pairs for varargs params.
2004 while (OpNum != Record.size()) {
2005 Value *Op;
2006 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2007 return Error("Invalid INVOKE record");
2008 Ops.push_back(Op);
2009 }
2010 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002011
Gabor Greifb91ea9d2008-05-15 10:04:30 +00002012 I = InvokeInst::Create(Callee, NormalBB, UnwindBB,
2013 Ops.begin(), Ops.end());
Devang Pateladc37612009-09-18 19:26:43 +00002014 InstructionList.push_back(I);
Sandeep Patel5838baa2009-09-02 08:44:58 +00002015 cast<InvokeInst>(I)->setCallingConv(
2016 static_cast<CallingConv::ID>(CCInfo));
Devang Pateld222f862008-09-25 21:00:45 +00002017 cast<InvokeInst>(I)->setAttributes(PAL);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002018 break;
2019 }
2020 case bitc::FUNC_CODE_INST_UNWIND: // UNWIND
Owen Anderson35b47072009-08-13 21:58:54 +00002021 I = new UnwindInst(Context);
Devang Pateladc37612009-09-18 19:26:43 +00002022 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002023 break;
2024 case bitc::FUNC_CODE_INST_UNREACHABLE: // UNREACHABLE
Owen Anderson35b47072009-08-13 21:58:54 +00002025 I = new UnreachableInst(Context);
Devang Pateladc37612009-09-18 19:26:43 +00002026 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002027 break;
2028 case bitc::FUNC_CODE_INST_PHI: { // PHI: [ty, val0,bb0, ...]
2029 if (Record.size() < 1 || ((Record.size()-1)&1))
2030 return Error("Invalid PHI record");
2031 const Type *Ty = getTypeByID(Record[0]);
2032 if (!Ty) return Error("Invalid PHI record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002033
Gabor Greifd6da1d02008-04-06 20:25:17 +00002034 PHINode *PN = PHINode::Create(Ty);
Devang Pateladc37612009-09-18 19:26:43 +00002035 InstructionList.push_back(PN);
Chris Lattner20e7f1a2008-04-13 00:14:42 +00002036 PN->reserveOperandSpace((Record.size()-1)/2);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002037
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002038 for (unsigned i = 0, e = Record.size()-1; i != e; i += 2) {
2039 Value *V = getFnValueByID(Record[1+i], Ty);
2040 BasicBlock *BB = getBasicBlock(Record[2+i]);
2041 if (!V || !BB) return Error("Invalid PHI record");
2042 PN->addIncoming(V, BB);
2043 }
2044 I = PN;
2045 break;
2046 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002047
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002048 case bitc::FUNC_CODE_INST_MALLOC: { // MALLOC: [instty, op, align]
2049 if (Record.size() < 3)
2050 return Error("Invalid MALLOC record");
2051 const PointerType *Ty =
2052 dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
Owen Anderson35b47072009-08-13 21:58:54 +00002053 Value *Size = getFnValueByID(Record[1], Type::getInt32Ty(Context));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002054 unsigned Align = Record[2];
2055 if (!Ty || !Size) return Error("Invalid MALLOC record");
Owen Anderson140166d2009-07-15 23:53:25 +00002056 I = new MallocInst(Ty->getElementType(), Size, (1 << Align) >> 1);
Devang Pateladc37612009-09-18 19:26:43 +00002057 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002058 break;
2059 }
2060 case bitc::FUNC_CODE_INST_FREE: { // FREE: [op, opty]
2061 unsigned OpNum = 0;
2062 Value *Op;
2063 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2064 OpNum != Record.size())
2065 return Error("Invalid FREE record");
2066 I = new FreeInst(Op);
Devang Pateladc37612009-09-18 19:26:43 +00002067 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002068 break;
2069 }
2070 case bitc::FUNC_CODE_INST_ALLOCA: { // ALLOCA: [instty, op, align]
2071 if (Record.size() < 3)
2072 return Error("Invalid ALLOCA record");
2073 const PointerType *Ty =
2074 dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
Owen Anderson35b47072009-08-13 21:58:54 +00002075 Value *Size = getFnValueByID(Record[1], Type::getInt32Ty(Context));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002076 unsigned Align = Record[2];
2077 if (!Ty || !Size) return Error("Invalid ALLOCA record");
Owen Anderson140166d2009-07-15 23:53:25 +00002078 I = new AllocaInst(Ty->getElementType(), Size, (1 << Align) >> 1);
Devang Pateladc37612009-09-18 19:26:43 +00002079 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002080 break;
2081 }
2082 case bitc::FUNC_CODE_INST_LOAD: { // LOAD: [opty, op, align, vol]
2083 unsigned OpNum = 0;
2084 Value *Op;
2085 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2086 OpNum+2 != Record.size())
2087 return Error("Invalid LOAD record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002088
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002089 I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Devang Pateladc37612009-09-18 19:26:43 +00002090 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002091 break;
2092 }
Christopher Lamb44d62f62007-12-11 08:59:05 +00002093 case bitc::FUNC_CODE_INST_STORE2: { // STORE2:[ptrty, ptr, val, align, vol]
2094 unsigned OpNum = 0;
2095 Value *Val, *Ptr;
2096 if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002097 getValue(Record, OpNum,
Christopher Lamb44d62f62007-12-11 08:59:05 +00002098 cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
2099 OpNum+2 != Record.size())
2100 return Error("Invalid STORE record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002101
Christopher Lamb44d62f62007-12-11 08:59:05 +00002102 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Devang Pateladc37612009-09-18 19:26:43 +00002103 InstructionList.push_back(I);
Christopher Lamb44d62f62007-12-11 08:59:05 +00002104 break;
2105 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002106 case bitc::FUNC_CODE_INST_STORE: { // STORE:[val, valty, ptr, align, vol]
Christopher Lamb44d62f62007-12-11 08:59:05 +00002107 // FIXME: Legacy form of store instruction. Should be removed in LLVM 3.0.
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002108 unsigned OpNum = 0;
2109 Value *Val, *Ptr;
2110 if (getValueTypePair(Record, OpNum, NextValueNo, Val) ||
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002111 getValue(Record, OpNum,
Owen Anderson6b6e2d92009-07-29 22:17:13 +00002112 PointerType::getUnqual(Val->getType()), Ptr)||
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002113 OpNum+2 != Record.size())
2114 return Error("Invalid STORE record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002115
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002116 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Devang Pateladc37612009-09-18 19:26:43 +00002117 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002118 break;
2119 }
Duncan Sandsf5588dc2007-11-27 13:23:08 +00002120 case bitc::FUNC_CODE_INST_CALL: {
2121 // CALL: [paramattrs, cc, fnty, fnid, arg0, arg1...]
2122 if (Record.size() < 3)
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002123 return Error("Invalid CALL record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002124
Devang Pateld222f862008-09-25 21:00:45 +00002125 AttrListPtr PAL = getAttributes(Record[0]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002126 unsigned CCInfo = Record[1];
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002127
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002128 unsigned OpNum = 2;
2129 Value *Callee;
2130 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
2131 return Error("Invalid CALL record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002132
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002133 const PointerType *OpTy = dyn_cast<PointerType>(Callee->getType());
2134 const FunctionType *FTy = 0;
2135 if (OpTy) FTy = dyn_cast<FunctionType>(OpTy->getElementType());
2136 if (!FTy || Record.size() < FTy->getNumParams()+OpNum)
2137 return Error("Invalid CALL record");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002138
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002139 SmallVector<Value*, 16> Args;
2140 // Read the fixed params.
2141 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
Dale Johannesencfb19e62007-11-05 21:20:28 +00002142 if (FTy->getParamType(i)->getTypeID()==Type::LabelTyID)
2143 Args.push_back(getBasicBlock(Record[OpNum]));
2144 else
2145 Args.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002146 if (Args.back() == 0) return Error("Invalid CALL record");
2147 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002148
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002149 // Read type/value pairs for varargs params.
2150 if (!FTy->isVarArg()) {
2151 if (OpNum != Record.size())
2152 return Error("Invalid CALL record");
2153 } else {
2154 while (OpNum != Record.size()) {
2155 Value *Op;
2156 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2157 return Error("Invalid CALL record");
2158 Args.push_back(Op);
2159 }
2160 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002161
Gabor Greifd6da1d02008-04-06 20:25:17 +00002162 I = CallInst::Create(Callee, Args.begin(), Args.end());
Devang Pateladc37612009-09-18 19:26:43 +00002163 InstructionList.push_back(I);
Sandeep Patel5838baa2009-09-02 08:44:58 +00002164 cast<CallInst>(I)->setCallingConv(
2165 static_cast<CallingConv::ID>(CCInfo>>1));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002166 cast<CallInst>(I)->setTailCall(CCInfo & 1);
Devang Pateld222f862008-09-25 21:00:45 +00002167 cast<CallInst>(I)->setAttributes(PAL);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002168 break;
2169 }
2170 case bitc::FUNC_CODE_INST_VAARG: { // VAARG: [valistty, valist, instty]
2171 if (Record.size() < 3)
2172 return Error("Invalid VAARG record");
2173 const Type *OpTy = getTypeByID(Record[0]);
2174 Value *Op = getFnValueByID(Record[1], OpTy);
2175 const Type *ResTy = getTypeByID(Record[2]);
2176 if (!OpTy || !Op || !ResTy)
2177 return Error("Invalid VAARG record");
2178 I = new VAArgInst(Op, ResTy);
Devang Pateladc37612009-09-18 19:26:43 +00002179 InstructionList.push_back(I);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002180 break;
2181 }
2182 }
2183
2184 // Add instruction to end of current BB. If there is no current BB, reject
2185 // this file.
2186 if (CurBB == 0) {
2187 delete I;
2188 return Error("Invalid instruction with no BB");
2189 }
2190 CurBB->getInstList().push_back(I);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002191
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002192 // If this was a terminator instruction, move to the next block.
2193 if (isa<TerminatorInst>(I)) {
2194 ++CurBBNo;
2195 CurBB = CurBBNo < FunctionBBs.size() ? FunctionBBs[CurBBNo] : 0;
2196 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002197
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002198 // Non-void values get registered in the value table for future use.
Owen Anderson35b47072009-08-13 21:58:54 +00002199 if (I && I->getType() != Type::getVoidTy(Context))
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002200 ValueList.AssignValue(I, NextValueNo++);
2201 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002202
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002203 // Check the function list for unresolved values.
2204 if (Argument *A = dyn_cast<Argument>(ValueList.back())) {
2205 if (A->getParent() == 0) {
2206 // We found at least one unresolved value. Nuke them all to avoid leaks.
2207 for (unsigned i = ModuleValueListSize, e = ValueList.size(); i != e; ++i){
2208 if ((A = dyn_cast<Argument>(ValueList.back())) && A->getParent() == 0) {
Owen Andersonb99ecca2009-07-30 23:03:37 +00002209 A->replaceAllUsesWith(UndefValue::get(A->getType()));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002210 delete A;
2211 }
2212 }
2213 return Error("Never resolved value found in function!");
2214 }
2215 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002216
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002217 // Trim the value list down to the size it was before we parsed this function.
2218 ValueList.shrinkTo(ModuleValueListSize);
2219 std::vector<BasicBlock*>().swap(FunctionBBs);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002220
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002221 return false;
2222}
2223
2224//===----------------------------------------------------------------------===//
2225// ModuleProvider implementation
2226//===----------------------------------------------------------------------===//
2227
2228
2229bool BitcodeReader::materializeFunction(Function *F, std::string *ErrInfo) {
2230 // If it already is material, ignore the request.
2231 if (!F->hasNotBeenReadFromBitcode()) return false;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002232
2233 DenseMap<Function*, std::pair<uint64_t, unsigned> >::iterator DFII =
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002234 DeferredFunctionInfo.find(F);
2235 assert(DFII != DeferredFunctionInfo.end() && "Deferred function not found!");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002236
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002237 // Move the bit stream to the saved position of the deferred function body and
2238 // restore the real linkage type for the function.
2239 Stream.JumpToBit(DFII->second.first);
2240 F->setLinkage((GlobalValue::LinkageTypes)DFII->second.second);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002241
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002242 if (ParseFunctionBody(F)) {
2243 if (ErrInfo) *ErrInfo = ErrorString;
2244 return true;
2245 }
Chandler Carrutha228e392007-08-04 01:51:18 +00002246
2247 // Upgrade any old intrinsic calls in the function.
2248 for (UpgradedIntrinsicMap::iterator I = UpgradedIntrinsics.begin(),
2249 E = UpgradedIntrinsics.end(); I != E; ++I) {
2250 if (I->first != I->second) {
2251 for (Value::use_iterator UI = I->first->use_begin(),
2252 UE = I->first->use_end(); UI != UE; ) {
2253 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2254 UpgradeIntrinsicCall(CI, I->second);
2255 }
2256 }
2257 }
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002258
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002259 return false;
2260}
2261
2262void BitcodeReader::dematerializeFunction(Function *F) {
2263 // If this function isn't materialized, or if it is a proto, this is a noop.
2264 if (F->hasNotBeenReadFromBitcode() || F->isDeclaration())
2265 return;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002266
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002267 assert(DeferredFunctionInfo.count(F) && "No info to read function later?");
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002268
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002269 // Just forget the function body, we can remat it later.
2270 F->deleteBody();
2271 F->setLinkage(GlobalValue::GhostLinkage);
2272}
2273
2274
2275Module *BitcodeReader::materializeModule(std::string *ErrInfo) {
Chris Lattner6b69fff2009-06-16 05:15:21 +00002276 // Iterate over the module, deserializing any functions that are still on
2277 // disk.
2278 for (Module::iterator F = TheModule->begin(), E = TheModule->end();
2279 F != E; ++F)
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002280 if (F->hasNotBeenReadFromBitcode() &&
2281 materializeFunction(F, ErrInfo))
2282 return 0;
Chandler Carrutha228e392007-08-04 01:51:18 +00002283
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002284 // Upgrade any intrinsic calls that slipped through (should not happen!) and
2285 // delete the old functions to clean up. We can't do this unless the entire
2286 // module is materialized because there could always be another function body
Chandler Carrutha228e392007-08-04 01:51:18 +00002287 // with calls to the old function.
2288 for (std::vector<std::pair<Function*, Function*> >::iterator I =
2289 UpgradedIntrinsics.begin(), E = UpgradedIntrinsics.end(); I != E; ++I) {
2290 if (I->first != I->second) {
2291 for (Value::use_iterator UI = I->first->use_begin(),
2292 UE = I->first->use_end(); UI != UE; ) {
2293 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2294 UpgradeIntrinsicCall(CI, I->second);
2295 }
Chris Lattnerda486ce2009-04-01 01:43:03 +00002296 if (!I->first->use_empty())
2297 I->first->replaceAllUsesWith(I->second);
Chandler Carrutha228e392007-08-04 01:51:18 +00002298 I->first->eraseFromParent();
2299 }
2300 }
2301 std::vector<std::pair<Function*, Function*> >().swap(UpgradedIntrinsics);
Devang Patel15e723d2009-08-28 23:24:31 +00002302
2303 // Check debug info intrinsics.
2304 CheckDebugInfoIntrinsics(TheModule);
2305
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002306 return TheModule;
2307}
2308
2309
2310/// This method is provided by the parent ModuleProvde class and overriden
2311/// here. It simply releases the module from its provided and frees up our
2312/// state.
2313/// @brief Release our hold on the generated module
2314Module *BitcodeReader::releaseModule(std::string *ErrInfo) {
2315 // Since we're losing control of this Module, we must hand it back complete
2316 Module *M = ModuleProvider::releaseModule(ErrInfo);
2317 FreeState();
2318 return M;
2319}
2320
2321
2322//===----------------------------------------------------------------------===//
2323// External interface
2324//===----------------------------------------------------------------------===//
2325
2326/// getBitcodeModuleProvider - lazy function-at-a-time loading from a file.
2327///
2328ModuleProvider *llvm::getBitcodeModuleProvider(MemoryBuffer *Buffer,
Owen Anderson92adb182009-07-01 23:13:44 +00002329 LLVMContext& Context,
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002330 std::string *ErrMsg) {
Owen Anderson25209b42009-07-01 16:58:40 +00002331 BitcodeReader *R = new BitcodeReader(Buffer, Context);
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002332 if (R->ParseBitcode()) {
2333 if (ErrMsg)
2334 *ErrMsg = R->getErrorString();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002335
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002336 // Don't let the BitcodeReader dtor delete 'Buffer'.
2337 R->releaseMemoryBuffer();
2338 delete R;
2339 return 0;
2340 }
2341 return R;
2342}
2343
2344/// ParseBitcodeFile - Read the specified bitcode file, returning the module.
2345/// If an error occurs, return null and fill in *ErrMsg if non-null.
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002346Module *llvm::ParseBitcodeFile(MemoryBuffer *Buffer, LLVMContext& Context,
Owen Anderson25209b42009-07-01 16:58:40 +00002347 std::string *ErrMsg){
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002348 BitcodeReader *R;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002349 R = static_cast<BitcodeReader*>(getBitcodeModuleProvider(Buffer, Context,
Owen Anderson25209b42009-07-01 16:58:40 +00002350 ErrMsg));
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002351 if (!R) return 0;
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002352
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002353 // Read in the entire module.
2354 Module *M = R->materializeModule(ErrMsg);
2355
2356 // Don't let the BitcodeReader dtor delete 'Buffer', regardless of whether
2357 // there was an error.
2358 R->releaseMemoryBuffer();
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002359
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002360 // If there was no error, tell ModuleProvider not to delete it when its dtor
2361 // is run.
2362 if (M)
2363 M = R->releaseModule(ErrMsg);
Daniel Dunbar3be44e62009-09-20 02:20:51 +00002364
Dan Gohmanf17a25c2007-07-18 16:29:46 +00002365 delete R;
2366 return M;
2367}