blob: 7cf03242990195b188daff8be41c1ba12e551fbc [file] [log] [blame]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001//===- BitcodeReader.cpp - Internal BitcodeReader implementation ----------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner4ee451d2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattnercaee0dc2007-04-22 06:23:29 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This header defines the BitcodeReader class.
11//
12//===----------------------------------------------------------------------===//
13
Chris Lattnerc453f762007-04-29 07:54:31 +000014#include "llvm/Bitcode/ReaderWriter.h"
Chris Lattnercaee0dc2007-04-22 06:23:29 +000015#include "BitcodeReader.h"
Chris Lattnere16504e2007-04-24 03:30:34 +000016#include "llvm/Constants.h"
Chris Lattnercaee0dc2007-04-22 06:23:29 +000017#include "llvm/DerivedTypes.h"
Chris Lattner2bce93a2007-05-06 01:58:20 +000018#include "llvm/InlineAsm.h"
Chris Lattnera7c49aa2007-05-01 07:01:57 +000019#include "llvm/Instructions.h"
Nick Lewyckycb337992009-05-10 20:57:05 +000020#include "llvm/MDNode.h"
Chris Lattnercaee0dc2007-04-22 06:23:29 +000021#include "llvm/Module.h"
Chandler Carruth69940402007-08-04 01:51:18 +000022#include "llvm/AutoUpgrade.h"
Chris Lattner0b2482a2007-04-23 21:26:05 +000023#include "llvm/ADT/SmallString.h"
Devang Patelf4511cd2008-02-26 19:38:17 +000024#include "llvm/ADT/SmallVector.h"
Chris Lattner0eef0802007-04-24 04:04:35 +000025#include "llvm/Support/MathExtras.h"
Chris Lattnerc453f762007-04-29 07:54:31 +000026#include "llvm/Support/MemoryBuffer.h"
Gabor Greifefe65362008-05-10 08:32:32 +000027#include "llvm/OperandTraits.h"
Chris Lattnercaee0dc2007-04-22 06:23:29 +000028using namespace llvm;
29
Chris Lattnerb348bb82007-05-18 04:02:46 +000030void BitcodeReader::FreeState() {
Chris Lattnerc453f762007-04-29 07:54:31 +000031 delete Buffer;
Chris Lattnerb348bb82007-05-18 04:02:46 +000032 Buffer = 0;
33 std::vector<PATypeHolder>().swap(TypeList);
34 ValueList.clear();
Chris Lattner461edd92008-03-12 02:25:52 +000035
Devang Patel19c87462008-09-26 22:53:05 +000036 std::vector<AttrListPtr>().swap(MAttributes);
Chris Lattnerb348bb82007-05-18 04:02:46 +000037 std::vector<BasicBlock*>().swap(FunctionBBs);
38 std::vector<Function*>().swap(FunctionsWithBodies);
39 DeferredFunctionInfo.clear();
Chris Lattnerc453f762007-04-29 07:54:31 +000040}
41
Chris Lattner48c85b82007-05-04 03:30:17 +000042//===----------------------------------------------------------------------===//
43// Helper functions to implement forward reference resolution, etc.
44//===----------------------------------------------------------------------===//
Chris Lattnerc453f762007-04-29 07:54:31 +000045
Chris Lattnercaee0dc2007-04-22 06:23:29 +000046/// ConvertToString - Convert a string from a record into an std::string, return
47/// true on failure.
Chris Lattner0b2482a2007-04-23 21:26:05 +000048template<typename StrTy>
Chris Lattnercaee0dc2007-04-22 06:23:29 +000049static bool ConvertToString(SmallVector<uint64_t, 64> &Record, unsigned Idx,
Chris Lattner0b2482a2007-04-23 21:26:05 +000050 StrTy &Result) {
Chris Lattner15e6d172007-05-04 19:11:41 +000051 if (Idx > Record.size())
Chris Lattnercaee0dc2007-04-22 06:23:29 +000052 return true;
53
Chris Lattner15e6d172007-05-04 19:11:41 +000054 for (unsigned i = Idx, e = Record.size(); i != e; ++i)
55 Result += (char)Record[i];
Chris Lattnercaee0dc2007-04-22 06:23:29 +000056 return false;
57}
58
59static GlobalValue::LinkageTypes GetDecodedLinkage(unsigned Val) {
60 switch (Val) {
61 default: // Map unknown/new linkages to external
62 case 0: return GlobalValue::ExternalLinkage;
Duncan Sands667d4b82009-03-07 15:45:40 +000063 case 1: return GlobalValue::WeakAnyLinkage;
Chris Lattnercaee0dc2007-04-22 06:23:29 +000064 case 2: return GlobalValue::AppendingLinkage;
65 case 3: return GlobalValue::InternalLinkage;
Duncan Sands667d4b82009-03-07 15:45:40 +000066 case 4: return GlobalValue::LinkOnceAnyLinkage;
Chris Lattnercaee0dc2007-04-22 06:23:29 +000067 case 5: return GlobalValue::DLLImportLinkage;
68 case 6: return GlobalValue::DLLExportLinkage;
Duncan Sands5f4ee1f2009-03-11 08:08:06 +000069 case 7: return GlobalValue::ExternalWeakLinkage;
Duncan Sands4dc2b392009-03-11 20:14:15 +000070 case 8: return GlobalValue::CommonLinkage;
Rafael Espindolabb46f522009-01-15 20:18:42 +000071 case 9: return GlobalValue::PrivateLinkage;
Duncan Sands667d4b82009-03-07 15:45:40 +000072 case 10: return GlobalValue::WeakODRLinkage;
73 case 11: return GlobalValue::LinkOnceODRLinkage;
Chris Lattner266c7bb2009-04-13 05:44:34 +000074 case 12: return GlobalValue::AvailableExternallyLinkage;
Chris Lattnercaee0dc2007-04-22 06:23:29 +000075 }
76}
77
78static GlobalValue::VisibilityTypes GetDecodedVisibility(unsigned Val) {
79 switch (Val) {
80 default: // Map unknown visibilities to default.
81 case 0: return GlobalValue::DefaultVisibility;
82 case 1: return GlobalValue::HiddenVisibility;
Anton Korobeynikov9cd3ccf2007-04-29 20:56:48 +000083 case 2: return GlobalValue::ProtectedVisibility;
Chris Lattnercaee0dc2007-04-22 06:23:29 +000084 }
85}
86
Chris Lattnerf581c3b2007-04-24 07:07:11 +000087static int GetDecodedCastOpcode(unsigned Val) {
88 switch (Val) {
89 default: return -1;
90 case bitc::CAST_TRUNC : return Instruction::Trunc;
91 case bitc::CAST_ZEXT : return Instruction::ZExt;
92 case bitc::CAST_SEXT : return Instruction::SExt;
93 case bitc::CAST_FPTOUI : return Instruction::FPToUI;
94 case bitc::CAST_FPTOSI : return Instruction::FPToSI;
95 case bitc::CAST_UITOFP : return Instruction::UIToFP;
96 case bitc::CAST_SITOFP : return Instruction::SIToFP;
97 case bitc::CAST_FPTRUNC : return Instruction::FPTrunc;
98 case bitc::CAST_FPEXT : return Instruction::FPExt;
99 case bitc::CAST_PTRTOINT: return Instruction::PtrToInt;
100 case bitc::CAST_INTTOPTR: return Instruction::IntToPtr;
101 case bitc::CAST_BITCAST : return Instruction::BitCast;
102 }
103}
104static int GetDecodedBinaryOpcode(unsigned Val, const Type *Ty) {
105 switch (Val) {
106 default: return -1;
Dan Gohmanae3a0be2009-06-04 22:49:04 +0000107 case bitc::BINOP_ADD:
108 return Ty->isFPOrFPVector() ? Instruction::FAdd : Instruction::Add;
109 case bitc::BINOP_SUB:
110 return Ty->isFPOrFPVector() ? Instruction::FSub : Instruction::Sub;
111 case bitc::BINOP_MUL:
112 return Ty->isFPOrFPVector() ? Instruction::FMul : Instruction::Mul;
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000113 case bitc::BINOP_UDIV: return Instruction::UDiv;
114 case bitc::BINOP_SDIV:
115 return Ty->isFPOrFPVector() ? Instruction::FDiv : Instruction::SDiv;
116 case bitc::BINOP_UREM: return Instruction::URem;
117 case bitc::BINOP_SREM:
118 return Ty->isFPOrFPVector() ? Instruction::FRem : Instruction::SRem;
119 case bitc::BINOP_SHL: return Instruction::Shl;
120 case bitc::BINOP_LSHR: return Instruction::LShr;
121 case bitc::BINOP_ASHR: return Instruction::AShr;
122 case bitc::BINOP_AND: return Instruction::And;
123 case bitc::BINOP_OR: return Instruction::Or;
124 case bitc::BINOP_XOR: return Instruction::Xor;
125 }
126}
127
Gabor Greifefe65362008-05-10 08:32:32 +0000128namespace llvm {
Chris Lattner522b7b12007-04-24 05:48:56 +0000129namespace {
130 /// @brief A class for maintaining the slot number definition
131 /// as a placeholder for the actual definition for forward constants defs.
132 class ConstantPlaceHolder : public ConstantExpr {
133 ConstantPlaceHolder(); // DO NOT IMPLEMENT
134 void operator=(const ConstantPlaceHolder &); // DO NOT IMPLEMENT
Gabor Greif051a9502008-04-06 20:25:17 +0000135 public:
136 // allocate space for exactly one operand
137 void *operator new(size_t s) {
138 return User::operator new(s, 1);
139 }
Dan Gohmanadf3eab2007-11-19 15:30:20 +0000140 explicit ConstantPlaceHolder(const Type *Ty)
Gabor Greifefe65362008-05-10 08:32:32 +0000141 : ConstantExpr(Ty, Instruction::UserOp1, &Op<0>(), 1) {
142 Op<0>() = UndefValue::get(Type::Int32Ty);
Chris Lattner522b7b12007-04-24 05:48:56 +0000143 }
Chris Lattnerea693df2008-08-21 02:34:16 +0000144
145 /// @brief Methods to support type inquiry through isa, cast, and dyn_cast.
146 static inline bool classof(const ConstantPlaceHolder *) { return true; }
147 static bool classof(const Value *V) {
148 return isa<ConstantExpr>(V) &&
149 cast<ConstantExpr>(V)->getOpcode() == Instruction::UserOp1;
150 }
151
152
Gabor Greifefe65362008-05-10 08:32:32 +0000153 /// Provide fast operand accessors
Chris Lattner46e77402009-03-31 22:55:09 +0000154 //DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
Chris Lattner522b7b12007-04-24 05:48:56 +0000155 };
156}
157
Chris Lattner46e77402009-03-31 22:55:09 +0000158// FIXME: can we inherit this from ConstantExpr?
Gabor Greifefe65362008-05-10 08:32:32 +0000159template <>
160struct OperandTraits<ConstantPlaceHolder> : FixedNumOperandTraits<1> {
161};
Gabor Greifefe65362008-05-10 08:32:32 +0000162}
163
Chris Lattner46e77402009-03-31 22:55:09 +0000164
165void BitcodeReaderValueList::AssignValue(Value *V, unsigned Idx) {
166 if (Idx == size()) {
167 push_back(V);
168 return;
169 }
170
171 if (Idx >= size())
172 resize(Idx+1);
173
174 WeakVH &OldV = ValuePtrs[Idx];
175 if (OldV == 0) {
176 OldV = V;
177 return;
178 }
179
180 // Handle constants and non-constants (e.g. instrs) differently for
181 // efficiency.
182 if (Constant *PHC = dyn_cast<Constant>(&*OldV)) {
183 ResolveConstants.push_back(std::make_pair(PHC, Idx));
184 OldV = V;
185 } else {
186 // If there was a forward reference to this value, replace it.
187 Value *PrevVal = OldV;
188 OldV->replaceAllUsesWith(V);
189 delete PrevVal;
Gabor Greifefe65362008-05-10 08:32:32 +0000190 }
191}
Chris Lattner46e77402009-03-31 22:55:09 +0000192
Gabor Greifefe65362008-05-10 08:32:32 +0000193
Chris Lattner522b7b12007-04-24 05:48:56 +0000194Constant *BitcodeReaderValueList::getConstantFwdRef(unsigned Idx,
195 const Type *Ty) {
Chris Lattner46e77402009-03-31 22:55:09 +0000196 if (Idx >= size())
Gabor Greifefe65362008-05-10 08:32:32 +0000197 resize(Idx + 1);
Chris Lattner522b7b12007-04-24 05:48:56 +0000198
Chris Lattner46e77402009-03-31 22:55:09 +0000199 if (Value *V = ValuePtrs[Idx]) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000200 assert(Ty == V->getType() && "Type mismatch in constant table!");
201 return cast<Constant>(V);
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000202 }
Chris Lattner522b7b12007-04-24 05:48:56 +0000203
204 // Create and return a placeholder, which will later be RAUW'd.
205 Constant *C = new ConstantPlaceHolder(Ty);
Chris Lattner46e77402009-03-31 22:55:09 +0000206 ValuePtrs[Idx] = C;
Chris Lattner522b7b12007-04-24 05:48:56 +0000207 return C;
208}
209
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000210Value *BitcodeReaderValueList::getValueFwdRef(unsigned Idx, const Type *Ty) {
Chris Lattner46e77402009-03-31 22:55:09 +0000211 if (Idx >= size())
Gabor Greifefe65362008-05-10 08:32:32 +0000212 resize(Idx + 1);
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000213
Chris Lattner46e77402009-03-31 22:55:09 +0000214 if (Value *V = ValuePtrs[Idx]) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000215 assert((Ty == 0 || Ty == V->getType()) && "Type mismatch in value table!");
216 return V;
217 }
218
Chris Lattner01ff65f2007-05-02 05:16:49 +0000219 // No type specified, must be invalid reference.
220 if (Ty == 0) return 0;
221
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000222 // Create and return a placeholder, which will later be RAUW'd.
223 Value *V = new Argument(Ty);
Chris Lattner46e77402009-03-31 22:55:09 +0000224 ValuePtrs[Idx] = V;
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000225 return V;
226}
227
Chris Lattnerea693df2008-08-21 02:34:16 +0000228/// ResolveConstantForwardRefs - Once all constants are read, this method bulk
229/// resolves any forward references. The idea behind this is that we sometimes
230/// get constants (such as large arrays) which reference *many* forward ref
231/// constants. Replacing each of these causes a lot of thrashing when
232/// building/reuniquing the constant. Instead of doing this, we look at all the
233/// uses and rewrite all the place holders at once for any constant that uses
234/// a placeholder.
235void BitcodeReaderValueList::ResolveConstantForwardRefs() {
236 // Sort the values by-pointer so that they are efficient to look up with a
237 // binary search.
238 std::sort(ResolveConstants.begin(), ResolveConstants.end());
239
240 SmallVector<Constant*, 64> NewOps;
241
242 while (!ResolveConstants.empty()) {
Chris Lattner46e77402009-03-31 22:55:09 +0000243 Value *RealVal = operator[](ResolveConstants.back().second);
Chris Lattnerea693df2008-08-21 02:34:16 +0000244 Constant *Placeholder = ResolveConstants.back().first;
245 ResolveConstants.pop_back();
246
247 // Loop over all users of the placeholder, updating them to reference the
248 // new value. If they reference more than one placeholder, update them all
249 // at once.
250 while (!Placeholder->use_empty()) {
Chris Lattnerb6135a02008-08-21 17:31:45 +0000251 Value::use_iterator UI = Placeholder->use_begin();
252
Chris Lattnerea693df2008-08-21 02:34:16 +0000253 // If the using object isn't uniqued, just update the operands. This
254 // handles instructions and initializers for global variables.
Chris Lattnerb6135a02008-08-21 17:31:45 +0000255 if (!isa<Constant>(*UI) || isa<GlobalValue>(*UI)) {
256 UI.getUse().set(RealVal);
Chris Lattnerea693df2008-08-21 02:34:16 +0000257 continue;
258 }
259
260 // Otherwise, we have a constant that uses the placeholder. Replace that
261 // constant with a new constant that has *all* placeholder uses updated.
Chris Lattnerb6135a02008-08-21 17:31:45 +0000262 Constant *UserC = cast<Constant>(*UI);
Chris Lattnerea693df2008-08-21 02:34:16 +0000263 for (User::op_iterator I = UserC->op_begin(), E = UserC->op_end();
264 I != E; ++I) {
265 Value *NewOp;
266 if (!isa<ConstantPlaceHolder>(*I)) {
267 // Not a placeholder reference.
268 NewOp = *I;
269 } else if (*I == Placeholder) {
270 // Common case is that it just references this one placeholder.
271 NewOp = RealVal;
272 } else {
273 // Otherwise, look up the placeholder in ResolveConstants.
274 ResolveConstantsTy::iterator It =
275 std::lower_bound(ResolveConstants.begin(), ResolveConstants.end(),
276 std::pair<Constant*, unsigned>(cast<Constant>(*I),
277 0));
278 assert(It != ResolveConstants.end() && It->first == *I);
Chris Lattner46e77402009-03-31 22:55:09 +0000279 NewOp = operator[](It->second);
Chris Lattnerea693df2008-08-21 02:34:16 +0000280 }
281
282 NewOps.push_back(cast<Constant>(NewOp));
283 }
284
285 // Make the new constant.
286 Constant *NewC;
287 if (ConstantArray *UserCA = dyn_cast<ConstantArray>(UserC)) {
288 NewC = ConstantArray::get(UserCA->getType(), &NewOps[0], NewOps.size());
289 } else if (ConstantStruct *UserCS = dyn_cast<ConstantStruct>(UserC)) {
290 NewC = ConstantStruct::get(&NewOps[0], NewOps.size(),
291 UserCS->getType()->isPacked());
292 } else if (isa<ConstantVector>(UserC)) {
293 NewC = ConstantVector::get(&NewOps[0], NewOps.size());
Nick Lewyckycb337992009-05-10 20:57:05 +0000294 } else {
295 assert(isa<ConstantExpr>(UserC) && "Must be a ConstantExpr.");
Chris Lattnerea693df2008-08-21 02:34:16 +0000296 NewC = cast<ConstantExpr>(UserC)->getWithOperands(&NewOps[0],
297 NewOps.size());
298 }
299
300 UserC->replaceAllUsesWith(NewC);
301 UserC->destroyConstant();
302 NewOps.clear();
303 }
304
Nick Lewyckycb337992009-05-10 20:57:05 +0000305 // Update all ValueHandles, they should be the only users at this point.
306 Placeholder->replaceAllUsesWith(RealVal);
Chris Lattnerea693df2008-08-21 02:34:16 +0000307 delete Placeholder;
308 }
309}
310
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000311
312const Type *BitcodeReader::getTypeByID(unsigned ID, bool isTypeTable) {
313 // If the TypeID is in range, return it.
314 if (ID < TypeList.size())
315 return TypeList[ID].get();
316 if (!isTypeTable) return 0;
317
318 // The type table allows forward references. Push as many Opaque types as
319 // needed to get up to ID.
320 while (TypeList.size() <= ID)
321 TypeList.push_back(OpaqueType::get());
322 return TypeList.back().get();
323}
324
Chris Lattner48c85b82007-05-04 03:30:17 +0000325//===----------------------------------------------------------------------===//
326// Functions for parsing blocks from the bitcode file
327//===----------------------------------------------------------------------===//
328
Devang Patel05988662008-09-25 21:00:45 +0000329bool BitcodeReader::ParseAttributeBlock() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000330 if (Stream.EnterSubBlock(bitc::PARAMATTR_BLOCK_ID))
Chris Lattner48c85b82007-05-04 03:30:17 +0000331 return Error("Malformed block record");
332
Devang Patel19c87462008-09-26 22:53:05 +0000333 if (!MAttributes.empty())
Chris Lattner48c85b82007-05-04 03:30:17 +0000334 return Error("Multiple PARAMATTR blocks found!");
335
336 SmallVector<uint64_t, 64> Record;
337
Devang Patel05988662008-09-25 21:00:45 +0000338 SmallVector<AttributeWithIndex, 8> Attrs;
Chris Lattner48c85b82007-05-04 03:30:17 +0000339
340 // Read all the records.
341 while (1) {
342 unsigned Code = Stream.ReadCode();
343 if (Code == bitc::END_BLOCK) {
344 if (Stream.ReadBlockEnd())
345 return Error("Error at end of PARAMATTR block");
346 return false;
347 }
348
349 if (Code == bitc::ENTER_SUBBLOCK) {
350 // No known subblocks, always skip them.
351 Stream.ReadSubBlockID();
352 if (Stream.SkipBlock())
353 return Error("Malformed block record");
354 continue;
355 }
356
357 if (Code == bitc::DEFINE_ABBREV) {
358 Stream.ReadAbbrevRecord();
359 continue;
360 }
361
362 // Read a record.
363 Record.clear();
364 switch (Stream.ReadRecord(Code, Record)) {
365 default: // Default behavior: ignore.
366 break;
367 case bitc::PARAMATTR_CODE_ENTRY: { // ENTRY: [paramidx0, attr0, ...]
368 if (Record.size() & 1)
369 return Error("Invalid ENTRY record");
370
Chris Lattner9a6cb152008-10-05 18:22:09 +0000371 // FIXME : Remove this autoupgrade code in LLVM 3.0.
Devang Patel19c87462008-09-26 22:53:05 +0000372 // If Function attributes are using index 0 then transfer them
Chris Lattner9a6cb152008-10-05 18:22:09 +0000373 // to index ~0. Index 0 is used for return value attributes but used to be
374 // used for function attributes.
Devang Patel19c87462008-09-26 22:53:05 +0000375 Attributes RetAttribute = Attribute::None;
376 Attributes FnAttribute = Attribute::None;
Chris Lattner48c85b82007-05-04 03:30:17 +0000377 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Nick Lewycky73ddd4f2008-12-19 09:38:31 +0000378 // FIXME: remove in LLVM 3.0
379 // The alignment is stored as a 16-bit raw value from bits 31--16.
380 // We shift the bits above 31 down by 11 bits.
381
382 unsigned Alignment = (Record[i+1] & (0xffffull << 16)) >> 16;
383 if (Alignment && !isPowerOf2_32(Alignment))
384 return Error("Alignment is not a power of two.");
385
386 Attributes ReconstitutedAttr = Record[i+1] & 0xffff;
387 if (Alignment)
388 ReconstitutedAttr |= Attribute::constructAlignmentFromInt(Alignment);
389 ReconstitutedAttr |= (Record[i+1] & (0xffffull << 32)) >> 11;
390 Record[i+1] = ReconstitutedAttr;
391
Devang Patel19c87462008-09-26 22:53:05 +0000392 if (Record[i] == 0)
393 RetAttribute = Record[i+1];
394 else if (Record[i] == ~0U)
395 FnAttribute = Record[i+1];
396 }
Chris Lattner9a6cb152008-10-05 18:22:09 +0000397
398 unsigned OldRetAttrs = (Attribute::NoUnwind|Attribute::NoReturn|
399 Attribute::ReadOnly|Attribute::ReadNone);
400
401 if (FnAttribute == Attribute::None && RetAttribute != Attribute::None &&
402 (RetAttribute & OldRetAttrs) != 0) {
403 if (FnAttribute == Attribute::None) { // add a slot so they get added.
404 Record.push_back(~0U);
405 Record.push_back(0);
Devang Patel19c87462008-09-26 22:53:05 +0000406 }
Chris Lattner9a6cb152008-10-05 18:22:09 +0000407
408 FnAttribute |= RetAttribute & OldRetAttrs;
409 RetAttribute &= ~OldRetAttrs;
Chris Lattner48c85b82007-05-04 03:30:17 +0000410 }
Chris Lattner461edd92008-03-12 02:25:52 +0000411
Devang Patel19c87462008-09-26 22:53:05 +0000412 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Chris Lattner9a6cb152008-10-05 18:22:09 +0000413 if (Record[i] == 0) {
414 if (RetAttribute != Attribute::None)
415 Attrs.push_back(AttributeWithIndex::get(0, RetAttribute));
416 } else if (Record[i] == ~0U) {
417 if (FnAttribute != Attribute::None)
418 Attrs.push_back(AttributeWithIndex::get(~0U, FnAttribute));
419 } else if (Record[i+1] != Attribute::None)
Devang Patel19c87462008-09-26 22:53:05 +0000420 Attrs.push_back(AttributeWithIndex::get(Record[i], Record[i+1]));
421 }
Devang Patel19c87462008-09-26 22:53:05 +0000422
423 MAttributes.push_back(AttrListPtr::get(Attrs.begin(), Attrs.end()));
Chris Lattner48c85b82007-05-04 03:30:17 +0000424 Attrs.clear();
425 break;
426 }
Duncan Sands5e41f652007-11-20 14:09:29 +0000427 }
Chris Lattner48c85b82007-05-04 03:30:17 +0000428 }
429}
430
431
Chris Lattner86697142007-05-01 05:01:34 +0000432bool BitcodeReader::ParseTypeTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000433 if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000434 return Error("Malformed block record");
435
436 if (!TypeList.empty())
437 return Error("Multiple TYPE_BLOCKs found!");
438
439 SmallVector<uint64_t, 64> Record;
440 unsigned NumRecords = 0;
441
442 // Read all the records for this type table.
443 while (1) {
444 unsigned Code = Stream.ReadCode();
445 if (Code == bitc::END_BLOCK) {
446 if (NumRecords != TypeList.size())
447 return Error("Invalid type forward reference in TYPE_BLOCK");
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000448 if (Stream.ReadBlockEnd())
449 return Error("Error at end of type table block");
450 return false;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000451 }
452
453 if (Code == bitc::ENTER_SUBBLOCK) {
454 // No known subblocks, always skip them.
455 Stream.ReadSubBlockID();
456 if (Stream.SkipBlock())
457 return Error("Malformed block record");
458 continue;
459 }
460
Chris Lattner36d5e7d2007-04-23 16:04:05 +0000461 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +0000462 Stream.ReadAbbrevRecord();
463 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000464 }
465
466 // Read a record.
467 Record.clear();
468 const Type *ResultTy = 0;
469 switch (Stream.ReadRecord(Code, Record)) {
470 default: // Default behavior: unknown type.
471 ResultTy = 0;
472 break;
473 case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
474 // TYPE_CODE_NUMENTRY contains a count of the number of types in the
475 // type list. This allows us to reserve space.
476 if (Record.size() < 1)
477 return Error("Invalid TYPE_CODE_NUMENTRY record");
478 TypeList.reserve(Record[0]);
479 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000480 case bitc::TYPE_CODE_VOID: // VOID
481 ResultTy = Type::VoidTy;
482 break;
483 case bitc::TYPE_CODE_FLOAT: // FLOAT
484 ResultTy = Type::FloatTy;
485 break;
486 case bitc::TYPE_CODE_DOUBLE: // DOUBLE
487 ResultTy = Type::DoubleTy;
488 break;
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000489 case bitc::TYPE_CODE_X86_FP80: // X86_FP80
490 ResultTy = Type::X86_FP80Ty;
491 break;
492 case bitc::TYPE_CODE_FP128: // FP128
493 ResultTy = Type::FP128Ty;
494 break;
495 case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
496 ResultTy = Type::PPC_FP128Ty;
497 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000498 case bitc::TYPE_CODE_LABEL: // LABEL
499 ResultTy = Type::LabelTy;
500 break;
501 case bitc::TYPE_CODE_OPAQUE: // OPAQUE
502 ResultTy = 0;
503 break;
Nick Lewycky7a0370f2009-05-30 05:06:04 +0000504 case bitc::TYPE_CODE_METADATA: // METADATA
505 ResultTy = Type::MetadataTy;
506 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000507 case bitc::TYPE_CODE_INTEGER: // INTEGER: [width]
508 if (Record.size() < 1)
509 return Error("Invalid Integer type record");
510
511 ResultTy = IntegerType::get(Record[0]);
512 break;
Christopher Lambfe63fb92007-12-11 08:59:05 +0000513 case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
514 // [pointee type, address space]
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000515 if (Record.size() < 1)
516 return Error("Invalid POINTER type record");
Christopher Lambfe63fb92007-12-11 08:59:05 +0000517 unsigned AddressSpace = 0;
518 if (Record.size() == 2)
519 AddressSpace = Record[1];
520 ResultTy = PointerType::get(getTypeByID(Record[0], true), AddressSpace);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000521 break;
Christopher Lambfe63fb92007-12-11 08:59:05 +0000522 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000523 case bitc::TYPE_CODE_FUNCTION: {
Chris Lattnera1afde72007-11-27 17:48:06 +0000524 // FIXME: attrid is dead, remove it in LLVM 3.0
525 // FUNCTION: [vararg, attrid, retty, paramty x N]
526 if (Record.size() < 3)
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000527 return Error("Invalid FUNCTION type record");
528 std::vector<const Type*> ArgTys;
Chris Lattnera1afde72007-11-27 17:48:06 +0000529 for (unsigned i = 3, e = Record.size(); i != e; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000530 ArgTys.push_back(getTypeByID(Record[i], true));
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000531
Chris Lattnera1afde72007-11-27 17:48:06 +0000532 ResultTy = FunctionType::get(getTypeByID(Record[2], true), ArgTys,
Duncan Sandsdc024672007-11-27 13:23:08 +0000533 Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000534 break;
535 }
Chris Lattner15e6d172007-05-04 19:11:41 +0000536 case bitc::TYPE_CODE_STRUCT: { // STRUCT: [ispacked, eltty x N]
Chris Lattner7108dce2007-05-06 08:21:50 +0000537 if (Record.size() < 1)
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000538 return Error("Invalid STRUCT type record");
539 std::vector<const Type*> EltTys;
Chris Lattner15e6d172007-05-04 19:11:41 +0000540 for (unsigned i = 1, e = Record.size(); i != e; ++i)
541 EltTys.push_back(getTypeByID(Record[i], true));
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000542 ResultTy = StructType::get(EltTys, Record[0]);
543 break;
544 }
545 case bitc::TYPE_CODE_ARRAY: // ARRAY: [numelts, eltty]
546 if (Record.size() < 2)
547 return Error("Invalid ARRAY type record");
548 ResultTy = ArrayType::get(getTypeByID(Record[1], true), Record[0]);
549 break;
550 case bitc::TYPE_CODE_VECTOR: // VECTOR: [numelts, eltty]
551 if (Record.size() < 2)
552 return Error("Invalid VECTOR type record");
553 ResultTy = VectorType::get(getTypeByID(Record[1], true), Record[0]);
554 break;
555 }
556
557 if (NumRecords == TypeList.size()) {
558 // If this is a new type slot, just append it.
559 TypeList.push_back(ResultTy ? ResultTy : OpaqueType::get());
560 ++NumRecords;
561 } else if (ResultTy == 0) {
562 // Otherwise, this was forward referenced, so an opaque type was created,
563 // but the result type is actually just an opaque. Leave the one we
564 // created previously.
565 ++NumRecords;
566 } else {
567 // Otherwise, this was forward referenced, so an opaque type was created.
568 // Resolve the opaque type to the real type now.
569 assert(NumRecords < TypeList.size() && "Typelist imbalance");
570 const OpaqueType *OldTy = cast<OpaqueType>(TypeList[NumRecords++].get());
571
572 // Don't directly push the new type on the Tab. Instead we want to replace
573 // the opaque type we previously inserted with the new concrete value. The
574 // refinement from the abstract (opaque) type to the new type causes all
575 // uses of the abstract type to use the concrete type (NewTy). This will
576 // also cause the opaque type to be deleted.
577 const_cast<OpaqueType*>(OldTy)->refineAbstractTypeTo(ResultTy);
578
579 // This should have replaced the old opaque type with the new type in the
Chris Lattner0eef0802007-04-24 04:04:35 +0000580 // value table... or with a preexisting type that was already in the
581 // system. Let's just make sure it did.
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000582 assert(TypeList[NumRecords-1].get() != OldTy &&
583 "refineAbstractType didn't work!");
584 }
585 }
586}
587
588
Chris Lattner86697142007-05-01 05:01:34 +0000589bool BitcodeReader::ParseTypeSymbolTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000590 if (Stream.EnterSubBlock(bitc::TYPE_SYMTAB_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000591 return Error("Malformed block record");
592
593 SmallVector<uint64_t, 64> Record;
594
595 // Read all the records for this type table.
596 std::string TypeName;
597 while (1) {
598 unsigned Code = Stream.ReadCode();
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000599 if (Code == bitc::END_BLOCK) {
600 if (Stream.ReadBlockEnd())
601 return Error("Error at end of type symbol table block");
602 return false;
603 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000604
605 if (Code == bitc::ENTER_SUBBLOCK) {
606 // No known subblocks, always skip them.
607 Stream.ReadSubBlockID();
608 if (Stream.SkipBlock())
609 return Error("Malformed block record");
610 continue;
611 }
612
Chris Lattner36d5e7d2007-04-23 16:04:05 +0000613 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +0000614 Stream.ReadAbbrevRecord();
615 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000616 }
617
618 // Read a record.
619 Record.clear();
620 switch (Stream.ReadRecord(Code, Record)) {
621 default: // Default behavior: unknown type.
622 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000623 case bitc::TST_CODE_ENTRY: // TST_ENTRY: [typeid, namechar x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000624 if (ConvertToString(Record, 1, TypeName))
625 return Error("Invalid TST_ENTRY record");
626 unsigned TypeID = Record[0];
627 if (TypeID >= TypeList.size())
628 return Error("Invalid Type ID in TST_ENTRY record");
629
630 TheModule->addTypeName(TypeName, TypeList[TypeID].get());
631 TypeName.clear();
632 break;
633 }
634 }
635}
636
Chris Lattner86697142007-05-01 05:01:34 +0000637bool BitcodeReader::ParseValueSymbolTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000638 if (Stream.EnterSubBlock(bitc::VALUE_SYMTAB_BLOCK_ID))
Chris Lattner0b2482a2007-04-23 21:26:05 +0000639 return Error("Malformed block record");
640
641 SmallVector<uint64_t, 64> Record;
642
643 // Read all the records for this value table.
644 SmallString<128> ValueName;
645 while (1) {
646 unsigned Code = Stream.ReadCode();
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000647 if (Code == bitc::END_BLOCK) {
648 if (Stream.ReadBlockEnd())
649 return Error("Error at end of value symbol table block");
650 return false;
651 }
Chris Lattner0b2482a2007-04-23 21:26:05 +0000652 if (Code == bitc::ENTER_SUBBLOCK) {
653 // No known subblocks, always skip them.
654 Stream.ReadSubBlockID();
655 if (Stream.SkipBlock())
656 return Error("Malformed block record");
657 continue;
658 }
659
660 if (Code == bitc::DEFINE_ABBREV) {
661 Stream.ReadAbbrevRecord();
662 continue;
663 }
664
665 // Read a record.
666 Record.clear();
667 switch (Stream.ReadRecord(Code, Record)) {
668 default: // Default behavior: unknown type.
669 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000670 case bitc::VST_CODE_ENTRY: { // VST_ENTRY: [valueid, namechar x N]
Chris Lattner0b2482a2007-04-23 21:26:05 +0000671 if (ConvertToString(Record, 1, ValueName))
Nick Lewycky88b72932009-05-31 06:07:28 +0000672 return Error("Invalid VST_ENTRY record");
Chris Lattner0b2482a2007-04-23 21:26:05 +0000673 unsigned ValueID = Record[0];
674 if (ValueID >= ValueList.size())
675 return Error("Invalid Value ID in VST_ENTRY record");
676 Value *V = ValueList[ValueID];
677
678 V->setName(&ValueName[0], ValueName.size());
679 ValueName.clear();
680 break;
Reid Spencerc8f8a242007-05-04 01:43:33 +0000681 }
682 case bitc::VST_CODE_BBENTRY: {
Chris Lattnere825ed52007-05-03 22:18:21 +0000683 if (ConvertToString(Record, 1, ValueName))
684 return Error("Invalid VST_BBENTRY record");
685 BasicBlock *BB = getBasicBlock(Record[0]);
686 if (BB == 0)
687 return Error("Invalid BB ID in VST_BBENTRY record");
688
689 BB->setName(&ValueName[0], ValueName.size());
690 ValueName.clear();
691 break;
Chris Lattner0b2482a2007-04-23 21:26:05 +0000692 }
Reid Spencerc8f8a242007-05-04 01:43:33 +0000693 }
Chris Lattner0b2482a2007-04-23 21:26:05 +0000694 }
695}
696
Chris Lattner0eef0802007-04-24 04:04:35 +0000697/// DecodeSignRotatedValue - Decode a signed value stored with the sign bit in
698/// the LSB for dense VBR encoding.
699static uint64_t DecodeSignRotatedValue(uint64_t V) {
700 if ((V & 1) == 0)
701 return V >> 1;
702 if (V != 1)
703 return -(V >> 1);
704 // There is no such thing as -0 with integers. "-0" really means MININT.
705 return 1ULL << 63;
706}
707
Chris Lattner07d98b42007-04-26 02:46:40 +0000708/// ResolveGlobalAndAliasInits - Resolve all of the initializers for global
709/// values and aliases that we can.
710bool BitcodeReader::ResolveGlobalAndAliasInits() {
711 std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInitWorklist;
712 std::vector<std::pair<GlobalAlias*, unsigned> > AliasInitWorklist;
713
714 GlobalInitWorklist.swap(GlobalInits);
715 AliasInitWorklist.swap(AliasInits);
716
717 while (!GlobalInitWorklist.empty()) {
Chris Lattner198f34a2007-04-26 03:27:58 +0000718 unsigned ValID = GlobalInitWorklist.back().second;
Chris Lattner07d98b42007-04-26 02:46:40 +0000719 if (ValID >= ValueList.size()) {
720 // Not ready to resolve this yet, it requires something later in the file.
Chris Lattner198f34a2007-04-26 03:27:58 +0000721 GlobalInits.push_back(GlobalInitWorklist.back());
Chris Lattner07d98b42007-04-26 02:46:40 +0000722 } else {
723 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
724 GlobalInitWorklist.back().first->setInitializer(C);
725 else
726 return Error("Global variable initializer is not a constant!");
727 }
728 GlobalInitWorklist.pop_back();
729 }
730
731 while (!AliasInitWorklist.empty()) {
732 unsigned ValID = AliasInitWorklist.back().second;
733 if (ValID >= ValueList.size()) {
734 AliasInits.push_back(AliasInitWorklist.back());
735 } else {
736 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
Anton Korobeynikov7dde0ff2007-04-28 14:57:59 +0000737 AliasInitWorklist.back().first->setAliasee(C);
Chris Lattner07d98b42007-04-26 02:46:40 +0000738 else
739 return Error("Alias initializer is not a constant!");
740 }
741 AliasInitWorklist.pop_back();
742 }
743 return false;
744}
745
746
Chris Lattner86697142007-05-01 05:01:34 +0000747bool BitcodeReader::ParseConstants() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000748 if (Stream.EnterSubBlock(bitc::CONSTANTS_BLOCK_ID))
Chris Lattnere16504e2007-04-24 03:30:34 +0000749 return Error("Malformed block record");
750
751 SmallVector<uint64_t, 64> Record;
752
753 // Read all the records for this value table.
754 const Type *CurTy = Type::Int32Ty;
Chris Lattner522b7b12007-04-24 05:48:56 +0000755 unsigned NextCstNo = ValueList.size();
Chris Lattnere16504e2007-04-24 03:30:34 +0000756 while (1) {
757 unsigned Code = Stream.ReadCode();
Chris Lattnerea693df2008-08-21 02:34:16 +0000758 if (Code == bitc::END_BLOCK)
759 break;
Chris Lattnere16504e2007-04-24 03:30:34 +0000760
761 if (Code == bitc::ENTER_SUBBLOCK) {
762 // No known subblocks, always skip them.
763 Stream.ReadSubBlockID();
764 if (Stream.SkipBlock())
765 return Error("Malformed block record");
766 continue;
767 }
768
769 if (Code == bitc::DEFINE_ABBREV) {
770 Stream.ReadAbbrevRecord();
771 continue;
772 }
773
774 // Read a record.
775 Record.clear();
776 Value *V = 0;
777 switch (Stream.ReadRecord(Code, Record)) {
778 default: // Default behavior: unknown constant
779 case bitc::CST_CODE_UNDEF: // UNDEF
780 V = UndefValue::get(CurTy);
781 break;
782 case bitc::CST_CODE_SETTYPE: // SETTYPE: [typeid]
783 if (Record.empty())
784 return Error("Malformed CST_SETTYPE record");
785 if (Record[0] >= TypeList.size())
786 return Error("Invalid Type ID in CST_SETTYPE record");
787 CurTy = TypeList[Record[0]];
Chris Lattner0eef0802007-04-24 04:04:35 +0000788 continue; // Skip the ValueList manipulation.
Chris Lattnere16504e2007-04-24 03:30:34 +0000789 case bitc::CST_CODE_NULL: // NULL
790 V = Constant::getNullValue(CurTy);
791 break;
792 case bitc::CST_CODE_INTEGER: // INTEGER: [intval]
Chris Lattner0eef0802007-04-24 04:04:35 +0000793 if (!isa<IntegerType>(CurTy) || Record.empty())
794 return Error("Invalid CST_INTEGER record");
795 V = ConstantInt::get(CurTy, DecodeSignRotatedValue(Record[0]));
796 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000797 case bitc::CST_CODE_WIDE_INTEGER: {// WIDE_INTEGER: [n x intval]
798 if (!isa<IntegerType>(CurTy) || Record.empty())
Chris Lattner0eef0802007-04-24 04:04:35 +0000799 return Error("Invalid WIDE_INTEGER record");
800
Chris Lattner15e6d172007-05-04 19:11:41 +0000801 unsigned NumWords = Record.size();
Chris Lattner084a8442007-04-24 17:22:05 +0000802 SmallVector<uint64_t, 8> Words;
803 Words.resize(NumWords);
Chris Lattner0eef0802007-04-24 04:04:35 +0000804 for (unsigned i = 0; i != NumWords; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000805 Words[i] = DecodeSignRotatedValue(Record[i]);
Chris Lattner0eef0802007-04-24 04:04:35 +0000806 V = ConstantInt::get(APInt(cast<IntegerType>(CurTy)->getBitWidth(),
Chris Lattner084a8442007-04-24 17:22:05 +0000807 NumWords, &Words[0]));
Chris Lattner0eef0802007-04-24 04:04:35 +0000808 break;
809 }
Dale Johannesen3f6eb742007-09-11 18:32:33 +0000810 case bitc::CST_CODE_FLOAT: { // FLOAT: [fpval]
Chris Lattner0eef0802007-04-24 04:04:35 +0000811 if (Record.empty())
812 return Error("Invalid FLOAT record");
813 if (CurTy == Type::FloatTy)
Chris Lattner02a260a2008-04-20 00:41:09 +0000814 V = ConstantFP::get(APFloat(APInt(32, (uint32_t)Record[0])));
Chris Lattner0eef0802007-04-24 04:04:35 +0000815 else if (CurTy == Type::DoubleTy)
Chris Lattner02a260a2008-04-20 00:41:09 +0000816 V = ConstantFP::get(APFloat(APInt(64, Record[0])));
Dale Johannesen1b25cb22009-03-23 21:16:53 +0000817 else if (CurTy == Type::X86_FP80Ty) {
818 // Bits are not stored the same way as a normal i80 APInt, compensate.
819 uint64_t Rearrange[2];
820 Rearrange[0] = (Record[1] & 0xffffLL) | (Record[0] << 16);
821 Rearrange[1] = Record[0] >> 48;
822 V = ConstantFP::get(APFloat(APInt(80, 2, Rearrange)));
823 } else if (CurTy == Type::FP128Ty)
Chris Lattner02a260a2008-04-20 00:41:09 +0000824 V = ConstantFP::get(APFloat(APInt(128, 2, &Record[0]), true));
Dale Johannesen43421b32007-09-06 18:13:44 +0000825 else if (CurTy == Type::PPC_FP128Ty)
Chris Lattner02a260a2008-04-20 00:41:09 +0000826 V = ConstantFP::get(APFloat(APInt(128, 2, &Record[0])));
Chris Lattnere16504e2007-04-24 03:30:34 +0000827 else
Chris Lattner0eef0802007-04-24 04:04:35 +0000828 V = UndefValue::get(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +0000829 break;
Dale Johannesen3f6eb742007-09-11 18:32:33 +0000830 }
Chris Lattner522b7b12007-04-24 05:48:56 +0000831
Chris Lattner15e6d172007-05-04 19:11:41 +0000832 case bitc::CST_CODE_AGGREGATE: {// AGGREGATE: [n x value number]
833 if (Record.empty())
Chris Lattner522b7b12007-04-24 05:48:56 +0000834 return Error("Invalid CST_AGGREGATE record");
835
Chris Lattner15e6d172007-05-04 19:11:41 +0000836 unsigned Size = Record.size();
Chris Lattner522b7b12007-04-24 05:48:56 +0000837 std::vector<Constant*> Elts;
838
839 if (const StructType *STy = dyn_cast<StructType>(CurTy)) {
840 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000841 Elts.push_back(ValueList.getConstantFwdRef(Record[i],
Chris Lattner522b7b12007-04-24 05:48:56 +0000842 STy->getElementType(i)));
843 V = ConstantStruct::get(STy, Elts);
844 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(CurTy)) {
845 const Type *EltTy = ATy->getElementType();
846 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000847 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Chris Lattner522b7b12007-04-24 05:48:56 +0000848 V = ConstantArray::get(ATy, Elts);
849 } else if (const VectorType *VTy = dyn_cast<VectorType>(CurTy)) {
850 const Type *EltTy = VTy->getElementType();
851 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000852 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Chris Lattner522b7b12007-04-24 05:48:56 +0000853 V = ConstantVector::get(Elts);
854 } else {
855 V = UndefValue::get(CurTy);
856 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000857 break;
858 }
Chris Lattnerff7fc5d2007-05-06 00:35:24 +0000859 case bitc::CST_CODE_STRING: { // STRING: [values]
860 if (Record.empty())
861 return Error("Invalid CST_AGGREGATE record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000862
Chris Lattnerff7fc5d2007-05-06 00:35:24 +0000863 const ArrayType *ATy = cast<ArrayType>(CurTy);
864 const Type *EltTy = ATy->getElementType();
865
866 unsigned Size = Record.size();
867 std::vector<Constant*> Elts;
Chris Lattnerff7fc5d2007-05-06 00:35:24 +0000868 for (unsigned i = 0; i != Size; ++i)
869 Elts.push_back(ConstantInt::get(EltTy, Record[i]));
870 V = ConstantArray::get(ATy, Elts);
871 break;
872 }
Chris Lattnercb3d91b2007-05-06 00:53:07 +0000873 case bitc::CST_CODE_CSTRING: { // CSTRING: [values]
874 if (Record.empty())
875 return Error("Invalid CST_AGGREGATE record");
876
877 const ArrayType *ATy = cast<ArrayType>(CurTy);
878 const Type *EltTy = ATy->getElementType();
879
880 unsigned Size = Record.size();
881 std::vector<Constant*> Elts;
882 for (unsigned i = 0; i != Size; ++i)
883 Elts.push_back(ConstantInt::get(EltTy, Record[i]));
884 Elts.push_back(Constant::getNullValue(EltTy));
885 V = ConstantArray::get(ATy, Elts);
886 break;
887 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000888 case bitc::CST_CODE_CE_BINOP: { // CE_BINOP: [opcode, opval, opval]
889 if (Record.size() < 3) return Error("Invalid CE_BINOP record");
890 int Opc = GetDecodedBinaryOpcode(Record[0], CurTy);
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000891 if (Opc < 0) {
892 V = UndefValue::get(CurTy); // Unknown binop.
893 } else {
894 Constant *LHS = ValueList.getConstantFwdRef(Record[1], CurTy);
895 Constant *RHS = ValueList.getConstantFwdRef(Record[2], CurTy);
896 V = ConstantExpr::get(Opc, LHS, RHS);
897 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000898 break;
899 }
900 case bitc::CST_CODE_CE_CAST: { // CE_CAST: [opcode, opty, opval]
901 if (Record.size() < 3) return Error("Invalid CE_CAST record");
902 int Opc = GetDecodedCastOpcode(Record[0]);
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000903 if (Opc < 0) {
904 V = UndefValue::get(CurTy); // Unknown cast.
905 } else {
906 const Type *OpTy = getTypeByID(Record[1]);
Chris Lattnerbfcc3802007-05-06 07:33:01 +0000907 if (!OpTy) return Error("Invalid CE_CAST record");
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000908 Constant *Op = ValueList.getConstantFwdRef(Record[2], OpTy);
909 V = ConstantExpr::getCast(Opc, Op, CurTy);
910 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000911 break;
912 }
913 case bitc::CST_CODE_CE_GEP: { // CE_GEP: [n x operands]
Chris Lattner15e6d172007-05-04 19:11:41 +0000914 if (Record.size() & 1) return Error("Invalid CE_GEP record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000915 SmallVector<Constant*, 16> Elts;
Chris Lattner15e6d172007-05-04 19:11:41 +0000916 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000917 const Type *ElTy = getTypeByID(Record[i]);
918 if (!ElTy) return Error("Invalid CE_GEP record");
919 Elts.push_back(ValueList.getConstantFwdRef(Record[i+1], ElTy));
920 }
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000921 V = ConstantExpr::getGetElementPtr(Elts[0], &Elts[1], Elts.size()-1);
922 break;
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000923 }
924 case bitc::CST_CODE_CE_SELECT: // CE_SELECT: [opval#, opval#, opval#]
925 if (Record.size() < 3) return Error("Invalid CE_SELECT record");
926 V = ConstantExpr::getSelect(ValueList.getConstantFwdRef(Record[0],
927 Type::Int1Ty),
928 ValueList.getConstantFwdRef(Record[1],CurTy),
929 ValueList.getConstantFwdRef(Record[2],CurTy));
930 break;
931 case bitc::CST_CODE_CE_EXTRACTELT: { // CE_EXTRACTELT: [opty, opval, opval]
932 if (Record.size() < 3) return Error("Invalid CE_EXTRACTELT record");
933 const VectorType *OpTy =
934 dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
935 if (OpTy == 0) return Error("Invalid CE_EXTRACTELT record");
936 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
Chris Lattnerba120aa2009-02-03 02:11:28 +0000937 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], Type::Int32Ty);
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000938 V = ConstantExpr::getExtractElement(Op0, Op1);
939 break;
940 }
941 case bitc::CST_CODE_CE_INSERTELT: { // CE_INSERTELT: [opval, opval, opval]
942 const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
943 if (Record.size() < 3 || OpTy == 0)
944 return Error("Invalid CE_INSERTELT record");
945 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
946 Constant *Op1 = ValueList.getConstantFwdRef(Record[1],
947 OpTy->getElementType());
948 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], Type::Int32Ty);
949 V = ConstantExpr::getInsertElement(Op0, Op1, Op2);
950 break;
951 }
952 case bitc::CST_CODE_CE_SHUFFLEVEC: { // CE_SHUFFLEVEC: [opval, opval, opval]
953 const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
954 if (Record.size() < 3 || OpTy == 0)
Nate Begeman0f123cf2009-02-12 21:28:33 +0000955 return Error("Invalid CE_SHUFFLEVEC record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000956 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
957 Constant *Op1 = ValueList.getConstantFwdRef(Record[1], OpTy);
958 const Type *ShufTy=VectorType::get(Type::Int32Ty, OpTy->getNumElements());
959 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], ShufTy);
960 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
961 break;
962 }
Nate Begeman0f123cf2009-02-12 21:28:33 +0000963 case bitc::CST_CODE_CE_SHUFVEC_EX: { // [opty, opval, opval, opval]
964 const VectorType *RTy = dyn_cast<VectorType>(CurTy);
965 const VectorType *OpTy = dyn_cast<VectorType>(getTypeByID(Record[0]));
966 if (Record.size() < 4 || RTy == 0 || OpTy == 0)
967 return Error("Invalid CE_SHUFVEC_EX record");
968 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
969 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
970 const Type *ShufTy=VectorType::get(Type::Int32Ty, RTy->getNumElements());
971 Constant *Op2 = ValueList.getConstantFwdRef(Record[3], ShufTy);
972 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
973 break;
974 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000975 case bitc::CST_CODE_CE_CMP: { // CE_CMP: [opty, opval, opval, pred]
976 if (Record.size() < 4) return Error("Invalid CE_CMP record");
977 const Type *OpTy = getTypeByID(Record[0]);
978 if (OpTy == 0) return Error("Invalid CE_CMP record");
979 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
980 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
981
982 if (OpTy->isFloatingPoint())
983 V = ConstantExpr::getFCmp(Record[3], Op0, Op1);
Nate Begemanbaa64eb2008-05-12 20:33:52 +0000984 else if (!isa<VectorType>(OpTy))
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000985 V = ConstantExpr::getICmp(Record[3], Op0, Op1);
Nate Begemanac80ade2008-05-12 19:01:56 +0000986 else if (OpTy->isFPOrFPVector())
987 V = ConstantExpr::getVFCmp(Record[3], Op0, Op1);
988 else
989 V = ConstantExpr::getVICmp(Record[3], Op0, Op1);
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000990 break;
Chris Lattner522b7b12007-04-24 05:48:56 +0000991 }
Chris Lattner2bce93a2007-05-06 01:58:20 +0000992 case bitc::CST_CODE_INLINEASM: {
993 if (Record.size() < 2) return Error("Invalid INLINEASM record");
994 std::string AsmStr, ConstrStr;
995 bool HasSideEffects = Record[0];
996 unsigned AsmStrSize = Record[1];
997 if (2+AsmStrSize >= Record.size())
998 return Error("Invalid INLINEASM record");
999 unsigned ConstStrSize = Record[2+AsmStrSize];
1000 if (3+AsmStrSize+ConstStrSize > Record.size())
1001 return Error("Invalid INLINEASM record");
1002
1003 for (unsigned i = 0; i != AsmStrSize; ++i)
1004 AsmStr += (char)Record[2+i];
1005 for (unsigned i = 0; i != ConstStrSize; ++i)
1006 ConstrStr += (char)Record[3+AsmStrSize+i];
1007 const PointerType *PTy = cast<PointerType>(CurTy);
1008 V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
1009 AsmStr, ConstrStr, HasSideEffects);
1010 break;
1011 }
Nick Lewycky21cc4462009-04-04 07:22:01 +00001012 case bitc::CST_CODE_MDSTRING: {
1013 if (Record.size() < 2) return Error("Invalid MDSTRING record");
1014 unsigned MDStringLength = Record.size();
1015 SmallString<8> String;
1016 String.resize(MDStringLength);
1017 for (unsigned i = 0; i != MDStringLength; ++i)
1018 String[i] = Record[i];
1019 V = MDString::get(String.c_str(), String.c_str() + MDStringLength);
1020 break;
1021 }
1022 case bitc::CST_CODE_MDNODE: {
1023 if (Record.empty() || Record.size() % 2 == 1)
1024 return Error("Invalid CST_MDNODE record");
1025
1026 unsigned Size = Record.size();
Nick Lewyckycb337992009-05-10 20:57:05 +00001027 SmallVector<Value*, 8> Elts;
Nick Lewycky21cc4462009-04-04 07:22:01 +00001028 for (unsigned i = 0; i != Size; i += 2) {
1029 const Type *Ty = getTypeByID(Record[i], false);
Nick Lewyckycb337992009-05-10 20:57:05 +00001030 if (Ty != Type::VoidTy)
Nick Lewycky3728a022009-06-01 04:42:10 +00001031 Elts.push_back(ValueList.getValueFwdRef(Record[i+1], Ty));
Nick Lewyckycb337992009-05-10 20:57:05 +00001032 else
1033 Elts.push_back(NULL);
Nick Lewycky21cc4462009-04-04 07:22:01 +00001034 }
1035 V = MDNode::get(&Elts[0], Elts.size());
1036 break;
1037 }
Chris Lattnere16504e2007-04-24 03:30:34 +00001038 }
1039
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001040 ValueList.AssignValue(V, NextCstNo);
Chris Lattner522b7b12007-04-24 05:48:56 +00001041 ++NextCstNo;
Chris Lattnere16504e2007-04-24 03:30:34 +00001042 }
Chris Lattnerea693df2008-08-21 02:34:16 +00001043
1044 if (NextCstNo != ValueList.size())
1045 return Error("Invalid constant reference!");
1046
1047 if (Stream.ReadBlockEnd())
1048 return Error("Error at end of constants block");
1049
1050 // Once all the constants have been read, go through and resolve forward
1051 // references.
1052 ValueList.ResolveConstantForwardRefs();
1053 return false;
Chris Lattnere16504e2007-04-24 03:30:34 +00001054}
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001055
Chris Lattner980e5aa2007-05-01 05:52:21 +00001056/// RememberAndSkipFunctionBody - When we see the block for a function body,
1057/// remember where it is and then skip it. This lets us lazily deserialize the
1058/// functions.
1059bool BitcodeReader::RememberAndSkipFunctionBody() {
Chris Lattner48f84872007-05-01 04:59:48 +00001060 // Get the function we are talking about.
1061 if (FunctionsWithBodies.empty())
1062 return Error("Insufficient function protos");
1063
1064 Function *Fn = FunctionsWithBodies.back();
1065 FunctionsWithBodies.pop_back();
1066
1067 // Save the current stream state.
1068 uint64_t CurBit = Stream.GetCurrentBitNo();
1069 DeferredFunctionInfo[Fn] = std::make_pair(CurBit, Fn->getLinkage());
1070
1071 // Set the functions linkage to GhostLinkage so we know it is lazily
1072 // deserialized.
1073 Fn->setLinkage(GlobalValue::GhostLinkage);
1074
1075 // Skip over the function block for now.
1076 if (Stream.SkipBlock())
1077 return Error("Malformed block record");
1078 return false;
1079}
1080
Chris Lattner86697142007-05-01 05:01:34 +00001081bool BitcodeReader::ParseModule(const std::string &ModuleID) {
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001082 // Reject multiple MODULE_BLOCK's in a single bitstream.
1083 if (TheModule)
1084 return Error("Multiple MODULE_BLOCKs in same stream");
1085
Chris Lattnere17b6582007-05-05 00:17:00 +00001086 if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001087 return Error("Malformed block record");
1088
1089 // Otherwise, create the module.
Owen Anderson8b477ed2009-07-01 16:58:40 +00001090 TheModule = new Module(ModuleID, Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001091
1092 SmallVector<uint64_t, 64> Record;
1093 std::vector<std::string> SectionTable;
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001094 std::vector<std::string> GCTable;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001095
1096 // Read all the records for this module.
1097 while (!Stream.AtEndOfStream()) {
1098 unsigned Code = Stream.ReadCode();
Chris Lattnere84bcb92007-04-24 00:21:45 +00001099 if (Code == bitc::END_BLOCK) {
Chris Lattner980e5aa2007-05-01 05:52:21 +00001100 if (Stream.ReadBlockEnd())
1101 return Error("Error at end of module block");
1102
1103 // Patch the initializers for globals and aliases up.
Chris Lattner07d98b42007-04-26 02:46:40 +00001104 ResolveGlobalAndAliasInits();
1105 if (!GlobalInits.empty() || !AliasInits.empty())
Chris Lattnere84bcb92007-04-24 00:21:45 +00001106 return Error("Malformed global initializer set");
Chris Lattner48f84872007-05-01 04:59:48 +00001107 if (!FunctionsWithBodies.empty())
1108 return Error("Too few function bodies found");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001109
Chandler Carruth69940402007-08-04 01:51:18 +00001110 // Look for intrinsic functions which need to be upgraded at some point
1111 for (Module::iterator FI = TheModule->begin(), FE = TheModule->end();
1112 FI != FE; ++FI) {
Evan Chengf9b83fc2007-12-17 22:33:23 +00001113 Function* NewFn;
1114 if (UpgradeIntrinsicFunction(FI, NewFn))
Chandler Carruth69940402007-08-04 01:51:18 +00001115 UpgradedIntrinsics.push_back(std::make_pair(FI, NewFn));
1116 }
1117
Chris Lattner980e5aa2007-05-01 05:52:21 +00001118 // Force deallocation of memory for these vectors to favor the client that
1119 // want lazy deserialization.
1120 std::vector<std::pair<GlobalVariable*, unsigned> >().swap(GlobalInits);
1121 std::vector<std::pair<GlobalAlias*, unsigned> >().swap(AliasInits);
1122 std::vector<Function*>().swap(FunctionsWithBodies);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001123 return false;
Chris Lattnere84bcb92007-04-24 00:21:45 +00001124 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001125
1126 if (Code == bitc::ENTER_SUBBLOCK) {
1127 switch (Stream.ReadSubBlockID()) {
1128 default: // Skip unknown content.
1129 if (Stream.SkipBlock())
1130 return Error("Malformed block record");
1131 break;
Chris Lattner3f799802007-05-05 18:57:30 +00001132 case bitc::BLOCKINFO_BLOCK_ID:
1133 if (Stream.ReadBlockInfoBlock())
1134 return Error("Malformed BlockInfoBlock");
1135 break;
Chris Lattner48c85b82007-05-04 03:30:17 +00001136 case bitc::PARAMATTR_BLOCK_ID:
Devang Patel05988662008-09-25 21:00:45 +00001137 if (ParseAttributeBlock())
Chris Lattner48c85b82007-05-04 03:30:17 +00001138 return true;
1139 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001140 case bitc::TYPE_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001141 if (ParseTypeTable())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001142 return true;
1143 break;
1144 case bitc::TYPE_SYMTAB_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001145 if (ParseTypeSymbolTable())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001146 return true;
1147 break;
Chris Lattner0b2482a2007-04-23 21:26:05 +00001148 case bitc::VALUE_SYMTAB_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001149 if (ParseValueSymbolTable())
Chris Lattner0b2482a2007-04-23 21:26:05 +00001150 return true;
1151 break;
Chris Lattnere16504e2007-04-24 03:30:34 +00001152 case bitc::CONSTANTS_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001153 if (ParseConstants() || ResolveGlobalAndAliasInits())
Chris Lattnere16504e2007-04-24 03:30:34 +00001154 return true;
1155 break;
Chris Lattner48f84872007-05-01 04:59:48 +00001156 case bitc::FUNCTION_BLOCK_ID:
1157 // If this is the first function body we've seen, reverse the
1158 // FunctionsWithBodies list.
1159 if (!HasReversedFunctionsWithBodies) {
1160 std::reverse(FunctionsWithBodies.begin(), FunctionsWithBodies.end());
1161 HasReversedFunctionsWithBodies = true;
1162 }
1163
Chris Lattner980e5aa2007-05-01 05:52:21 +00001164 if (RememberAndSkipFunctionBody())
Chris Lattner48f84872007-05-01 04:59:48 +00001165 return true;
1166 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001167 }
1168 continue;
1169 }
1170
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001171 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +00001172 Stream.ReadAbbrevRecord();
1173 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001174 }
1175
1176 // Read a record.
1177 switch (Stream.ReadRecord(Code, Record)) {
1178 default: break; // Default behavior, ignore unknown content.
1179 case bitc::MODULE_CODE_VERSION: // VERSION: [version#]
1180 if (Record.size() < 1)
1181 return Error("Malformed MODULE_CODE_VERSION");
1182 // Only version #0 is supported so far.
1183 if (Record[0] != 0)
1184 return Error("Unknown bitstream version!");
1185 break;
Chris Lattner15e6d172007-05-04 19:11:41 +00001186 case bitc::MODULE_CODE_TRIPLE: { // TRIPLE: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001187 std::string S;
1188 if (ConvertToString(Record, 0, S))
1189 return Error("Invalid MODULE_CODE_TRIPLE record");
1190 TheModule->setTargetTriple(S);
1191 break;
1192 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001193 case bitc::MODULE_CODE_DATALAYOUT: { // DATALAYOUT: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001194 std::string S;
1195 if (ConvertToString(Record, 0, S))
1196 return Error("Invalid MODULE_CODE_DATALAYOUT record");
1197 TheModule->setDataLayout(S);
1198 break;
1199 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001200 case bitc::MODULE_CODE_ASM: { // ASM: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001201 std::string S;
1202 if (ConvertToString(Record, 0, S))
1203 return Error("Invalid MODULE_CODE_ASM record");
1204 TheModule->setModuleInlineAsm(S);
1205 break;
1206 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001207 case bitc::MODULE_CODE_DEPLIB: { // DEPLIB: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001208 std::string S;
1209 if (ConvertToString(Record, 0, S))
1210 return Error("Invalid MODULE_CODE_DEPLIB record");
1211 TheModule->addLibrary(S);
1212 break;
1213 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001214 case bitc::MODULE_CODE_SECTIONNAME: { // SECTIONNAME: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001215 std::string S;
1216 if (ConvertToString(Record, 0, S))
1217 return Error("Invalid MODULE_CODE_SECTIONNAME record");
1218 SectionTable.push_back(S);
1219 break;
1220 }
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001221 case bitc::MODULE_CODE_GCNAME: { // SECTIONNAME: [strchr x N]
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001222 std::string S;
1223 if (ConvertToString(Record, 0, S))
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001224 return Error("Invalid MODULE_CODE_GCNAME record");
1225 GCTable.push_back(S);
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001226 break;
1227 }
Christopher Lambfe63fb92007-12-11 08:59:05 +00001228 // GLOBALVAR: [pointer type, isconst, initid,
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001229 // linkage, alignment, section, visibility, threadlocal]
1230 case bitc::MODULE_CODE_GLOBALVAR: {
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001231 if (Record.size() < 6)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001232 return Error("Invalid MODULE_CODE_GLOBALVAR record");
1233 const Type *Ty = getTypeByID(Record[0]);
1234 if (!isa<PointerType>(Ty))
1235 return Error("Global not a pointer type!");
Christopher Lambfe63fb92007-12-11 08:59:05 +00001236 unsigned AddressSpace = cast<PointerType>(Ty)->getAddressSpace();
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001237 Ty = cast<PointerType>(Ty)->getElementType();
1238
1239 bool isConstant = Record[1];
1240 GlobalValue::LinkageTypes Linkage = GetDecodedLinkage(Record[3]);
1241 unsigned Alignment = (1 << Record[4]) >> 1;
1242 std::string Section;
1243 if (Record[5]) {
1244 if (Record[5]-1 >= SectionTable.size())
1245 return Error("Invalid section ID");
1246 Section = SectionTable[Record[5]-1];
1247 }
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001248 GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
Chris Lattner5f32c012007-05-06 19:27:46 +00001249 if (Record.size() > 6)
1250 Visibility = GetDecodedVisibility(Record[6]);
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001251 bool isThreadLocal = false;
Chris Lattner5f32c012007-05-06 19:27:46 +00001252 if (Record.size() > 7)
1253 isThreadLocal = Record[7];
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001254
1255 GlobalVariable *NewGV =
Christopher Lambfe63fb92007-12-11 08:59:05 +00001256 new GlobalVariable(Ty, isConstant, Linkage, 0, "", TheModule,
1257 isThreadLocal, AddressSpace);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001258 NewGV->setAlignment(Alignment);
1259 if (!Section.empty())
1260 NewGV->setSection(Section);
1261 NewGV->setVisibility(Visibility);
1262 NewGV->setThreadLocal(isThreadLocal);
1263
Chris Lattner0b2482a2007-04-23 21:26:05 +00001264 ValueList.push_back(NewGV);
1265
Chris Lattner6dbfd7b2007-04-24 00:18:21 +00001266 // Remember which value to use for the global initializer.
1267 if (unsigned InitID = Record[2])
1268 GlobalInits.push_back(std::make_pair(NewGV, InitID-1));
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001269 break;
1270 }
Chris Lattnera9bb7132007-05-08 05:38:01 +00001271 // FUNCTION: [type, callingconv, isproto, linkage, paramattr,
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001272 // alignment, section, visibility, gc]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001273 case bitc::MODULE_CODE_FUNCTION: {
Chris Lattnera9bb7132007-05-08 05:38:01 +00001274 if (Record.size() < 8)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001275 return Error("Invalid MODULE_CODE_FUNCTION record");
1276 const Type *Ty = getTypeByID(Record[0]);
1277 if (!isa<PointerType>(Ty))
1278 return Error("Function not a pointer type!");
1279 const FunctionType *FTy =
1280 dyn_cast<FunctionType>(cast<PointerType>(Ty)->getElementType());
1281 if (!FTy)
1282 return Error("Function not a pointer to function type!");
1283
Gabor Greif051a9502008-04-06 20:25:17 +00001284 Function *Func = Function::Create(FTy, GlobalValue::ExternalLinkage,
1285 "", TheModule);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001286
1287 Func->setCallingConv(Record[1]);
Chris Lattner48f84872007-05-01 04:59:48 +00001288 bool isProto = Record[2];
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001289 Func->setLinkage(GetDecodedLinkage(Record[3]));
Devang Patel05988662008-09-25 21:00:45 +00001290 Func->setAttributes(getAttributes(Record[4]));
Chris Lattnera9bb7132007-05-08 05:38:01 +00001291
1292 Func->setAlignment((1 << Record[5]) >> 1);
1293 if (Record[6]) {
1294 if (Record[6]-1 >= SectionTable.size())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001295 return Error("Invalid section ID");
Chris Lattnera9bb7132007-05-08 05:38:01 +00001296 Func->setSection(SectionTable[Record[6]-1]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001297 }
Chris Lattnera9bb7132007-05-08 05:38:01 +00001298 Func->setVisibility(GetDecodedVisibility(Record[7]));
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001299 if (Record.size() > 8 && Record[8]) {
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001300 if (Record[8]-1 > GCTable.size())
1301 return Error("Invalid GC ID");
1302 Func->setGC(GCTable[Record[8]-1].c_str());
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001303 }
Chris Lattner0b2482a2007-04-23 21:26:05 +00001304 ValueList.push_back(Func);
Chris Lattner48f84872007-05-01 04:59:48 +00001305
1306 // If this is a function with a body, remember the prototype we are
1307 // creating now, so that we can match up the body with them later.
1308 if (!isProto)
1309 FunctionsWithBodies.push_back(Func);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001310 break;
1311 }
Anton Korobeynikov91342d82008-03-12 00:49:19 +00001312 // ALIAS: [alias type, aliasee val#, linkage]
Anton Korobeynikovf8342b92008-03-11 21:40:17 +00001313 // ALIAS: [alias type, aliasee val#, linkage, visibility]
Chris Lattner198f34a2007-04-26 03:27:58 +00001314 case bitc::MODULE_CODE_ALIAS: {
Chris Lattner07d98b42007-04-26 02:46:40 +00001315 if (Record.size() < 3)
1316 return Error("Invalid MODULE_ALIAS record");
1317 const Type *Ty = getTypeByID(Record[0]);
1318 if (!isa<PointerType>(Ty))
1319 return Error("Function not a pointer type!");
1320
1321 GlobalAlias *NewGA = new GlobalAlias(Ty, GetDecodedLinkage(Record[2]),
1322 "", 0, TheModule);
Anton Korobeynikov91342d82008-03-12 00:49:19 +00001323 // Old bitcode files didn't have visibility field.
1324 if (Record.size() > 3)
1325 NewGA->setVisibility(GetDecodedVisibility(Record[3]));
Chris Lattner07d98b42007-04-26 02:46:40 +00001326 ValueList.push_back(NewGA);
1327 AliasInits.push_back(std::make_pair(NewGA, Record[1]));
1328 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001329 }
Chris Lattner198f34a2007-04-26 03:27:58 +00001330 /// MODULE_CODE_PURGEVALS: [numvals]
1331 case bitc::MODULE_CODE_PURGEVALS:
1332 // Trim down the value list to the specified size.
1333 if (Record.size() < 1 || Record[0] > ValueList.size())
1334 return Error("Invalid MODULE_PURGEVALS record");
1335 ValueList.shrinkTo(Record[0]);
1336 break;
1337 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001338 Record.clear();
1339 }
1340
1341 return Error("Premature end of bitstream");
1342}
1343
Chris Lattnerc453f762007-04-29 07:54:31 +00001344bool BitcodeReader::ParseBitcode() {
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001345 TheModule = 0;
1346
Chris Lattnerc453f762007-04-29 07:54:31 +00001347 if (Buffer->getBufferSize() & 3)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001348 return Error("Bitcode stream should be a multiple of 4 bytes in length");
1349
Chris Lattnerc453f762007-04-29 07:54:31 +00001350 unsigned char *BufPtr = (unsigned char *)Buffer->getBufferStart();
Chris Lattner6fa6a322008-07-09 05:14:23 +00001351 unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
1352
1353 // If we have a wrapper header, parse it and ignore the non-bc file contents.
1354 // The magic number is 0x0B17C0DE stored in little endian.
Chris Lattnere2a466b2009-04-06 20:54:32 +00001355 if (isBitcodeWrapper(BufPtr, BufEnd))
1356 if (SkipBitcodeWrapperHeader(BufPtr, BufEnd))
Chris Lattner6fa6a322008-07-09 05:14:23 +00001357 return Error("Invalid bitcode wrapper header");
1358
Chris Lattner962dde32009-04-26 20:59:02 +00001359 StreamFile.init(BufPtr, BufEnd);
1360 Stream.init(StreamFile);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001361
1362 // Sniff for the signature.
1363 if (Stream.Read(8) != 'B' ||
1364 Stream.Read(8) != 'C' ||
1365 Stream.Read(4) != 0x0 ||
1366 Stream.Read(4) != 0xC ||
1367 Stream.Read(4) != 0xE ||
1368 Stream.Read(4) != 0xD)
1369 return Error("Invalid bitcode signature");
1370
1371 // We expect a number of well-defined blocks, though we don't necessarily
1372 // need to understand them all.
1373 while (!Stream.AtEndOfStream()) {
1374 unsigned Code = Stream.ReadCode();
1375
1376 if (Code != bitc::ENTER_SUBBLOCK)
1377 return Error("Invalid record at top-level");
1378
1379 unsigned BlockID = Stream.ReadSubBlockID();
1380
1381 // We only know the MODULE subblock ID.
Chris Lattnere17b6582007-05-05 00:17:00 +00001382 switch (BlockID) {
1383 case bitc::BLOCKINFO_BLOCK_ID:
1384 if (Stream.ReadBlockInfoBlock())
1385 return Error("Malformed BlockInfoBlock");
1386 break;
1387 case bitc::MODULE_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001388 if (ParseModule(Buffer->getBufferIdentifier()))
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001389 return true;
Chris Lattnere17b6582007-05-05 00:17:00 +00001390 break;
1391 default:
1392 if (Stream.SkipBlock())
1393 return Error("Malformed block record");
1394 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001395 }
1396 }
1397
1398 return false;
1399}
Chris Lattnerc453f762007-04-29 07:54:31 +00001400
Chris Lattner48f84872007-05-01 04:59:48 +00001401
Chris Lattner980e5aa2007-05-01 05:52:21 +00001402/// ParseFunctionBody - Lazily parse the specified function body block.
1403bool BitcodeReader::ParseFunctionBody(Function *F) {
Chris Lattnere17b6582007-05-05 00:17:00 +00001404 if (Stream.EnterSubBlock(bitc::FUNCTION_BLOCK_ID))
Chris Lattner980e5aa2007-05-01 05:52:21 +00001405 return Error("Malformed block record");
1406
1407 unsigned ModuleValueListSize = ValueList.size();
1408
1409 // Add all the function arguments to the value table.
1410 for(Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
1411 ValueList.push_back(I);
1412
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001413 unsigned NextValueNo = ValueList.size();
Chris Lattner231cbcb2007-05-02 04:27:25 +00001414 BasicBlock *CurBB = 0;
1415 unsigned CurBBNo = 0;
1416
Chris Lattner980e5aa2007-05-01 05:52:21 +00001417 // Read all the records.
1418 SmallVector<uint64_t, 64> Record;
1419 while (1) {
1420 unsigned Code = Stream.ReadCode();
1421 if (Code == bitc::END_BLOCK) {
1422 if (Stream.ReadBlockEnd())
1423 return Error("Error at end of function block");
1424 break;
1425 }
1426
1427 if (Code == bitc::ENTER_SUBBLOCK) {
1428 switch (Stream.ReadSubBlockID()) {
1429 default: // Skip unknown content.
1430 if (Stream.SkipBlock())
1431 return Error("Malformed block record");
1432 break;
1433 case bitc::CONSTANTS_BLOCK_ID:
1434 if (ParseConstants()) return true;
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001435 NextValueNo = ValueList.size();
Chris Lattner980e5aa2007-05-01 05:52:21 +00001436 break;
1437 case bitc::VALUE_SYMTAB_BLOCK_ID:
1438 if (ParseValueSymbolTable()) return true;
1439 break;
1440 }
1441 continue;
1442 }
1443
1444 if (Code == bitc::DEFINE_ABBREV) {
1445 Stream.ReadAbbrevRecord();
1446 continue;
1447 }
1448
1449 // Read a record.
1450 Record.clear();
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001451 Instruction *I = 0;
Chris Lattner980e5aa2007-05-01 05:52:21 +00001452 switch (Stream.ReadRecord(Code, Record)) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001453 default: // Default behavior: reject
1454 return Error("Unknown instruction");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001455 case bitc::FUNC_CODE_DECLAREBLOCKS: // DECLAREBLOCKS: [nblocks]
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001456 if (Record.size() < 1 || Record[0] == 0)
1457 return Error("Invalid DECLAREBLOCKS record");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001458 // Create all the basic blocks for the function.
Chris Lattnerf61e6452007-05-03 22:09:51 +00001459 FunctionBBs.resize(Record[0]);
Chris Lattner980e5aa2007-05-01 05:52:21 +00001460 for (unsigned i = 0, e = FunctionBBs.size(); i != e; ++i)
Gabor Greif051a9502008-04-06 20:25:17 +00001461 FunctionBBs[i] = BasicBlock::Create("", F);
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001462 CurBB = FunctionBBs[0];
1463 continue;
1464
Chris Lattnerabfbf852007-05-06 00:21:25 +00001465 case bitc::FUNC_CODE_INST_BINOP: { // BINOP: [opval, ty, opval, opcode]
1466 unsigned OpNum = 0;
1467 Value *LHS, *RHS;
1468 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1469 getValue(Record, OpNum, LHS->getType(), RHS) ||
1470 OpNum+1 != Record.size())
1471 return Error("Invalid BINOP record");
1472
1473 int Opc = GetDecodedBinaryOpcode(Record[OpNum], LHS->getType());
1474 if (Opc == -1) return Error("Invalid BINOP record");
Gabor Greif7cbd8a32008-05-16 19:29:10 +00001475 I = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
Chris Lattner980e5aa2007-05-01 05:52:21 +00001476 break;
1477 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00001478 case bitc::FUNC_CODE_INST_CAST: { // CAST: [opval, opty, destty, castopc]
1479 unsigned OpNum = 0;
1480 Value *Op;
1481 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
1482 OpNum+2 != Record.size())
1483 return Error("Invalid CAST record");
1484
1485 const Type *ResTy = getTypeByID(Record[OpNum]);
1486 int Opc = GetDecodedCastOpcode(Record[OpNum+1]);
1487 if (Opc == -1 || ResTy == 0)
Chris Lattner231cbcb2007-05-02 04:27:25 +00001488 return Error("Invalid CAST record");
Gabor Greif7cbd8a32008-05-16 19:29:10 +00001489 I = CastInst::Create((Instruction::CastOps)Opc, Op, ResTy);
Chris Lattner231cbcb2007-05-02 04:27:25 +00001490 break;
1491 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001492 case bitc::FUNC_CODE_INST_GEP: { // GEP: [n x operands]
Chris Lattner7337ab92007-05-06 00:00:00 +00001493 unsigned OpNum = 0;
1494 Value *BasePtr;
1495 if (getValueTypePair(Record, OpNum, NextValueNo, BasePtr))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001496 return Error("Invalid GEP record");
1497
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001498 SmallVector<Value*, 16> GEPIdx;
Chris Lattner7337ab92007-05-06 00:00:00 +00001499 while (OpNum != Record.size()) {
1500 Value *Op;
1501 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001502 return Error("Invalid GEP record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001503 GEPIdx.push_back(Op);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001504 }
1505
Gabor Greif051a9502008-04-06 20:25:17 +00001506 I = GetElementPtrInst::Create(BasePtr, GEPIdx.begin(), GEPIdx.end());
Chris Lattner01ff65f2007-05-02 05:16:49 +00001507 break;
1508 }
Chris Lattner231cbcb2007-05-02 04:27:25 +00001509
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001510 case bitc::FUNC_CODE_INST_EXTRACTVAL: {
1511 // EXTRACTVAL: [opty, opval, n x indices]
Dan Gohmane4977cf2008-05-23 01:55:30 +00001512 unsigned OpNum = 0;
1513 Value *Agg;
1514 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1515 return Error("Invalid EXTRACTVAL record");
1516
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001517 SmallVector<unsigned, 4> EXTRACTVALIdx;
1518 for (unsigned RecSize = Record.size();
1519 OpNum != RecSize; ++OpNum) {
1520 uint64_t Index = Record[OpNum];
1521 if ((unsigned)Index != Index)
1522 return Error("Invalid EXTRACTVAL index");
1523 EXTRACTVALIdx.push_back((unsigned)Index);
Dan Gohmane4977cf2008-05-23 01:55:30 +00001524 }
1525
1526 I = ExtractValueInst::Create(Agg,
1527 EXTRACTVALIdx.begin(), EXTRACTVALIdx.end());
1528 break;
1529 }
1530
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001531 case bitc::FUNC_CODE_INST_INSERTVAL: {
1532 // INSERTVAL: [opty, opval, opty, opval, n x indices]
Dan Gohmane4977cf2008-05-23 01:55:30 +00001533 unsigned OpNum = 0;
1534 Value *Agg;
1535 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1536 return Error("Invalid INSERTVAL record");
1537 Value *Val;
1538 if (getValueTypePair(Record, OpNum, NextValueNo, Val))
1539 return Error("Invalid INSERTVAL record");
1540
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001541 SmallVector<unsigned, 4> INSERTVALIdx;
1542 for (unsigned RecSize = Record.size();
1543 OpNum != RecSize; ++OpNum) {
1544 uint64_t Index = Record[OpNum];
1545 if ((unsigned)Index != Index)
1546 return Error("Invalid INSERTVAL index");
1547 INSERTVALIdx.push_back((unsigned)Index);
Dan Gohmane4977cf2008-05-23 01:55:30 +00001548 }
1549
1550 I = InsertValueInst::Create(Agg, Val,
1551 INSERTVALIdx.begin(), INSERTVALIdx.end());
1552 break;
1553 }
1554
Chris Lattnerabfbf852007-05-06 00:21:25 +00001555 case bitc::FUNC_CODE_INST_SELECT: { // SELECT: [opval, ty, opval, opval]
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001556 // obsolete form of select
1557 // handles select i1 ... in old bitcode
Chris Lattnerabfbf852007-05-06 00:21:25 +00001558 unsigned OpNum = 0;
1559 Value *TrueVal, *FalseVal, *Cond;
1560 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1561 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
Dan Gohmanbe919402008-09-09 02:08:49 +00001562 getValue(Record, OpNum, Type::Int1Ty, Cond))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001563 return Error("Invalid SELECT record");
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001564
1565 I = SelectInst::Create(Cond, TrueVal, FalseVal);
1566 break;
1567 }
1568
1569 case bitc::FUNC_CODE_INST_VSELECT: {// VSELECT: [ty,opval,opval,predty,pred]
1570 // new form of select
1571 // handles select i1 or select [N x i1]
1572 unsigned OpNum = 0;
1573 Value *TrueVal, *FalseVal, *Cond;
1574 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1575 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
1576 getValueTypePair(Record, OpNum, NextValueNo, Cond))
1577 return Error("Invalid SELECT record");
Dan Gohmanf72fb672008-09-09 01:02:47 +00001578
1579 // select condition can be either i1 or [N x i1]
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001580 if (const VectorType* vector_type =
1581 dyn_cast<const VectorType>(Cond->getType())) {
Dan Gohmanf72fb672008-09-09 01:02:47 +00001582 // expect <n x i1>
1583 if (vector_type->getElementType() != Type::Int1Ty)
1584 return Error("Invalid SELECT condition type");
1585 } else {
1586 // expect i1
1587 if (Cond->getType() != Type::Int1Ty)
1588 return Error("Invalid SELECT condition type");
1589 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00001590
Gabor Greif051a9502008-04-06 20:25:17 +00001591 I = SelectInst::Create(Cond, TrueVal, FalseVal);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001592 break;
1593 }
1594
1595 case bitc::FUNC_CODE_INST_EXTRACTELT: { // EXTRACTELT: [opty, opval, opval]
Chris Lattnerabfbf852007-05-06 00:21:25 +00001596 unsigned OpNum = 0;
1597 Value *Vec, *Idx;
1598 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
1599 getValue(Record, OpNum, Type::Int32Ty, Idx))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001600 return Error("Invalid EXTRACTELT record");
1601 I = new ExtractElementInst(Vec, Idx);
1602 break;
1603 }
1604
1605 case bitc::FUNC_CODE_INST_INSERTELT: { // INSERTELT: [ty, opval,opval,opval]
Chris Lattnerabfbf852007-05-06 00:21:25 +00001606 unsigned OpNum = 0;
1607 Value *Vec, *Elt, *Idx;
1608 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
1609 getValue(Record, OpNum,
1610 cast<VectorType>(Vec->getType())->getElementType(), Elt) ||
1611 getValue(Record, OpNum, Type::Int32Ty, Idx))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001612 return Error("Invalid INSERTELT record");
Gabor Greif051a9502008-04-06 20:25:17 +00001613 I = InsertElementInst::Create(Vec, Elt, Idx);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001614 break;
1615 }
1616
Chris Lattnerabfbf852007-05-06 00:21:25 +00001617 case bitc::FUNC_CODE_INST_SHUFFLEVEC: {// SHUFFLEVEC: [opval,ty,opval,opval]
1618 unsigned OpNum = 0;
1619 Value *Vec1, *Vec2, *Mask;
1620 if (getValueTypePair(Record, OpNum, NextValueNo, Vec1) ||
1621 getValue(Record, OpNum, Vec1->getType(), Vec2))
1622 return Error("Invalid SHUFFLEVEC record");
1623
Mon P Wangaeb06d22008-11-10 04:46:22 +00001624 if (getValueTypePair(Record, OpNum, NextValueNo, Mask))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001625 return Error("Invalid SHUFFLEVEC record");
1626 I = new ShuffleVectorInst(Vec1, Vec2, Mask);
1627 break;
1628 }
Mon P Wangaeb06d22008-11-10 04:46:22 +00001629
Chris Lattner01ff65f2007-05-02 05:16:49 +00001630 case bitc::FUNC_CODE_INST_CMP: { // CMP: [opty, opval, opval, pred]
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001631 // VFCmp/VICmp
1632 // or old form of ICmp/FCmp returning bool
Chris Lattner7337ab92007-05-06 00:00:00 +00001633 unsigned OpNum = 0;
1634 Value *LHS, *RHS;
1635 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1636 getValue(Record, OpNum, LHS->getType(), RHS) ||
1637 OpNum+1 != Record.size())
Chris Lattner01ff65f2007-05-02 05:16:49 +00001638 return Error("Invalid CMP record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001639
Nate Begemanbaa64eb2008-05-12 20:33:52 +00001640 if (LHS->getType()->isFloatingPoint())
Nate Begemanac80ade2008-05-12 19:01:56 +00001641 I = new FCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
Nate Begemanbaa64eb2008-05-12 20:33:52 +00001642 else if (!isa<VectorType>(LHS->getType()))
1643 I = new ICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
Nate Begemanac80ade2008-05-12 19:01:56 +00001644 else if (LHS->getType()->isFPOrFPVector())
1645 I = new VFCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
1646 else
1647 I = new VICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001648 break;
1649 }
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001650 case bitc::FUNC_CODE_INST_CMP2: { // CMP2: [opty, opval, opval, pred]
1651 // Fcmp/ICmp returning bool or vector of bool
Dan Gohmanf72fb672008-09-09 01:02:47 +00001652 unsigned OpNum = 0;
1653 Value *LHS, *RHS;
1654 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1655 getValue(Record, OpNum, LHS->getType(), RHS) ||
1656 OpNum+1 != Record.size())
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001657 return Error("Invalid CMP2 record");
Dan Gohmanf72fb672008-09-09 01:02:47 +00001658
Dan Gohmanf72fb672008-09-09 01:02:47 +00001659 if (LHS->getType()->isFPOrFPVector())
1660 I = new FCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
1661 else
1662 I = new ICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
1663 break;
1664 }
Devang Patel197be3d2008-02-22 02:49:49 +00001665 case bitc::FUNC_CODE_INST_GETRESULT: { // GETRESULT: [ty, val, n]
1666 if (Record.size() != 2)
1667 return Error("Invalid GETRESULT record");
1668 unsigned OpNum = 0;
1669 Value *Op;
1670 getValueTypePair(Record, OpNum, NextValueNo, Op);
1671 unsigned Index = Record[1];
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001672 I = ExtractValueInst::Create(Op, Index);
Devang Patel197be3d2008-02-22 02:49:49 +00001673 break;
1674 }
Chris Lattner01ff65f2007-05-02 05:16:49 +00001675
Chris Lattner231cbcb2007-05-02 04:27:25 +00001676 case bitc::FUNC_CODE_INST_RET: // RET: [opty,opval<optional>]
Devang Pateld9d99ff2008-02-26 01:29:32 +00001677 {
1678 unsigned Size = Record.size();
1679 if (Size == 0) {
Gabor Greif051a9502008-04-06 20:25:17 +00001680 I = ReturnInst::Create();
Devang Pateld9d99ff2008-02-26 01:29:32 +00001681 break;
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001682 }
Devang Pateld9d99ff2008-02-26 01:29:32 +00001683
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001684 unsigned OpNum = 0;
1685 SmallVector<Value *,4> Vs;
1686 do {
1687 Value *Op = NULL;
1688 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1689 return Error("Invalid RET record");
1690 Vs.push_back(Op);
1691 } while(OpNum != Record.size());
1692
1693 const Type *ReturnType = F->getReturnType();
1694 if (Vs.size() > 1 ||
1695 (isa<StructType>(ReturnType) &&
1696 (Vs.empty() || Vs[0]->getType() != ReturnType))) {
1697 Value *RV = UndefValue::get(ReturnType);
1698 for (unsigned i = 0, e = Vs.size(); i != e; ++i) {
1699 I = InsertValueInst::Create(RV, Vs[i], i, "mrv");
1700 CurBB->getInstList().push_back(I);
1701 ValueList.AssignValue(I, NextValueNo++);
1702 RV = I;
1703 }
1704 I = ReturnInst::Create(RV);
Devang Pateld9d99ff2008-02-26 01:29:32 +00001705 break;
1706 }
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001707
1708 I = ReturnInst::Create(Vs[0]);
1709 break;
Chris Lattner231cbcb2007-05-02 04:27:25 +00001710 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001711 case bitc::FUNC_CODE_INST_BR: { // BR: [bb#, bb#, opval] or [bb#]
Chris Lattnerf61e6452007-05-03 22:09:51 +00001712 if (Record.size() != 1 && Record.size() != 3)
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001713 return Error("Invalid BR record");
1714 BasicBlock *TrueDest = getBasicBlock(Record[0]);
1715 if (TrueDest == 0)
1716 return Error("Invalid BR record");
1717
1718 if (Record.size() == 1)
Gabor Greif051a9502008-04-06 20:25:17 +00001719 I = BranchInst::Create(TrueDest);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001720 else {
1721 BasicBlock *FalseDest = getBasicBlock(Record[1]);
1722 Value *Cond = getFnValueByID(Record[2], Type::Int1Ty);
1723 if (FalseDest == 0 || Cond == 0)
1724 return Error("Invalid BR record");
Gabor Greif051a9502008-04-06 20:25:17 +00001725 I = BranchInst::Create(TrueDest, FalseDest, Cond);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001726 }
1727 break;
1728 }
1729 case bitc::FUNC_CODE_INST_SWITCH: { // SWITCH: [opty, opval, n, n x ops]
1730 if (Record.size() < 3 || (Record.size() & 1) == 0)
1731 return Error("Invalid SWITCH record");
1732 const Type *OpTy = getTypeByID(Record[0]);
1733 Value *Cond = getFnValueByID(Record[1], OpTy);
1734 BasicBlock *Default = getBasicBlock(Record[2]);
1735 if (OpTy == 0 || Cond == 0 || Default == 0)
1736 return Error("Invalid SWITCH record");
1737 unsigned NumCases = (Record.size()-3)/2;
Gabor Greif051a9502008-04-06 20:25:17 +00001738 SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001739 for (unsigned i = 0, e = NumCases; i != e; ++i) {
1740 ConstantInt *CaseVal =
1741 dyn_cast_or_null<ConstantInt>(getFnValueByID(Record[3+i*2], OpTy));
1742 BasicBlock *DestBB = getBasicBlock(Record[1+3+i*2]);
1743 if (CaseVal == 0 || DestBB == 0) {
1744 delete SI;
1745 return Error("Invalid SWITCH record!");
1746 }
1747 SI->addCase(CaseVal, DestBB);
1748 }
1749 I = SI;
1750 break;
1751 }
1752
Duncan Sandsdc024672007-11-27 13:23:08 +00001753 case bitc::FUNC_CODE_INST_INVOKE: {
1754 // INVOKE: [attrs, cc, normBB, unwindBB, fnty, op0,op1,op2, ...]
Chris Lattnera9bb7132007-05-08 05:38:01 +00001755 if (Record.size() < 4) return Error("Invalid INVOKE record");
Devang Patel05988662008-09-25 21:00:45 +00001756 AttrListPtr PAL = getAttributes(Record[0]);
Chris Lattnera9bb7132007-05-08 05:38:01 +00001757 unsigned CCInfo = Record[1];
1758 BasicBlock *NormalBB = getBasicBlock(Record[2]);
1759 BasicBlock *UnwindBB = getBasicBlock(Record[3]);
Chris Lattner7337ab92007-05-06 00:00:00 +00001760
Chris Lattnera9bb7132007-05-08 05:38:01 +00001761 unsigned OpNum = 4;
Chris Lattner7337ab92007-05-06 00:00:00 +00001762 Value *Callee;
1763 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001764 return Error("Invalid INVOKE record");
1765
Chris Lattner7337ab92007-05-06 00:00:00 +00001766 const PointerType *CalleeTy = dyn_cast<PointerType>(Callee->getType());
1767 const FunctionType *FTy = !CalleeTy ? 0 :
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001768 dyn_cast<FunctionType>(CalleeTy->getElementType());
1769
1770 // Check that the right number of fixed parameters are here.
Chris Lattner7337ab92007-05-06 00:00:00 +00001771 if (FTy == 0 || NormalBB == 0 || UnwindBB == 0 ||
1772 Record.size() < OpNum+FTy->getNumParams())
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001773 return Error("Invalid INVOKE record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001774
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001775 SmallVector<Value*, 16> Ops;
Chris Lattner7337ab92007-05-06 00:00:00 +00001776 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
1777 Ops.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
1778 if (Ops.back() == 0) return Error("Invalid INVOKE record");
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001779 }
1780
Chris Lattner7337ab92007-05-06 00:00:00 +00001781 if (!FTy->isVarArg()) {
1782 if (Record.size() != OpNum)
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001783 return Error("Invalid INVOKE record");
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001784 } else {
Chris Lattner7337ab92007-05-06 00:00:00 +00001785 // Read type/value pairs for varargs params.
1786 while (OpNum != Record.size()) {
1787 Value *Op;
1788 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1789 return Error("Invalid INVOKE record");
1790 Ops.push_back(Op);
1791 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001792 }
1793
Gabor Greifb1dbcd82008-05-15 10:04:30 +00001794 I = InvokeInst::Create(Callee, NormalBB, UnwindBB,
1795 Ops.begin(), Ops.end());
Chris Lattner76520192007-05-03 22:34:03 +00001796 cast<InvokeInst>(I)->setCallingConv(CCInfo);
Devang Patel05988662008-09-25 21:00:45 +00001797 cast<InvokeInst>(I)->setAttributes(PAL);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001798 break;
1799 }
Chris Lattner231cbcb2007-05-02 04:27:25 +00001800 case bitc::FUNC_CODE_INST_UNWIND: // UNWIND
1801 I = new UnwindInst();
1802 break;
1803 case bitc::FUNC_CODE_INST_UNREACHABLE: // UNREACHABLE
1804 I = new UnreachableInst();
1805 break;
Chris Lattnerabfbf852007-05-06 00:21:25 +00001806 case bitc::FUNC_CODE_INST_PHI: { // PHI: [ty, val0,bb0, ...]
Chris Lattner15e6d172007-05-04 19:11:41 +00001807 if (Record.size() < 1 || ((Record.size()-1)&1))
Chris Lattner2a98cca2007-05-03 18:58:09 +00001808 return Error("Invalid PHI record");
1809 const Type *Ty = getTypeByID(Record[0]);
1810 if (!Ty) return Error("Invalid PHI record");
1811
Gabor Greif051a9502008-04-06 20:25:17 +00001812 PHINode *PN = PHINode::Create(Ty);
Chris Lattner86941612008-04-13 00:14:42 +00001813 PN->reserveOperandSpace((Record.size()-1)/2);
Chris Lattner2a98cca2007-05-03 18:58:09 +00001814
Chris Lattner15e6d172007-05-04 19:11:41 +00001815 for (unsigned i = 0, e = Record.size()-1; i != e; i += 2) {
1816 Value *V = getFnValueByID(Record[1+i], Ty);
1817 BasicBlock *BB = getBasicBlock(Record[2+i]);
Chris Lattner2a98cca2007-05-03 18:58:09 +00001818 if (!V || !BB) return Error("Invalid PHI record");
1819 PN->addIncoming(V, BB);
1820 }
1821 I = PN;
1822 break;
1823 }
1824
1825 case bitc::FUNC_CODE_INST_MALLOC: { // MALLOC: [instty, op, align]
1826 if (Record.size() < 3)
1827 return Error("Invalid MALLOC record");
1828 const PointerType *Ty =
1829 dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
1830 Value *Size = getFnValueByID(Record[1], Type::Int32Ty);
1831 unsigned Align = Record[2];
1832 if (!Ty || !Size) return Error("Invalid MALLOC record");
1833 I = new MallocInst(Ty->getElementType(), Size, (1 << Align) >> 1);
1834 break;
1835 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00001836 case bitc::FUNC_CODE_INST_FREE: { // FREE: [op, opty]
1837 unsigned OpNum = 0;
1838 Value *Op;
1839 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
1840 OpNum != Record.size())
Chris Lattner2a98cca2007-05-03 18:58:09 +00001841 return Error("Invalid FREE record");
1842 I = new FreeInst(Op);
1843 break;
1844 }
1845 case bitc::FUNC_CODE_INST_ALLOCA: { // ALLOCA: [instty, op, align]
1846 if (Record.size() < 3)
1847 return Error("Invalid ALLOCA record");
1848 const PointerType *Ty =
1849 dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
1850 Value *Size = getFnValueByID(Record[1], Type::Int32Ty);
1851 unsigned Align = Record[2];
1852 if (!Ty || !Size) return Error("Invalid ALLOCA record");
1853 I = new AllocaInst(Ty->getElementType(), Size, (1 << Align) >> 1);
1854 break;
1855 }
Chris Lattner0579f7f2007-05-03 22:04:19 +00001856 case bitc::FUNC_CODE_INST_LOAD: { // LOAD: [opty, op, align, vol]
Chris Lattner7337ab92007-05-06 00:00:00 +00001857 unsigned OpNum = 0;
1858 Value *Op;
1859 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
1860 OpNum+2 != Record.size())
Chris Lattnerabfbf852007-05-06 00:21:25 +00001861 return Error("Invalid LOAD record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001862
1863 I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001864 break;
Chris Lattner0579f7f2007-05-03 22:04:19 +00001865 }
Christopher Lambfe63fb92007-12-11 08:59:05 +00001866 case bitc::FUNC_CODE_INST_STORE2: { // STORE2:[ptrty, ptr, val, align, vol]
1867 unsigned OpNum = 0;
1868 Value *Val, *Ptr;
1869 if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
1870 getValue(Record, OpNum,
1871 cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
1872 OpNum+2 != Record.size())
1873 return Error("Invalid STORE record");
1874
1875 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
1876 break;
1877 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00001878 case bitc::FUNC_CODE_INST_STORE: { // STORE:[val, valty, ptr, align, vol]
Christopher Lambfe63fb92007-12-11 08:59:05 +00001879 // FIXME: Legacy form of store instruction. Should be removed in LLVM 3.0.
Chris Lattnerabfbf852007-05-06 00:21:25 +00001880 unsigned OpNum = 0;
1881 Value *Val, *Ptr;
1882 if (getValueTypePair(Record, OpNum, NextValueNo, Val) ||
Christopher Lamb43ad6b32007-12-17 01:12:55 +00001883 getValue(Record, OpNum, PointerType::getUnqual(Val->getType()), Ptr)||
Chris Lattnerabfbf852007-05-06 00:21:25 +00001884 OpNum+2 != Record.size())
Chris Lattner0579f7f2007-05-03 22:04:19 +00001885 return Error("Invalid STORE record");
Chris Lattnerabfbf852007-05-06 00:21:25 +00001886
1887 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Chris Lattner0579f7f2007-05-03 22:04:19 +00001888 break;
1889 }
Duncan Sandsdc024672007-11-27 13:23:08 +00001890 case bitc::FUNC_CODE_INST_CALL: {
1891 // CALL: [paramattrs, cc, fnty, fnid, arg0, arg1...]
1892 if (Record.size() < 3)
Chris Lattner0579f7f2007-05-03 22:04:19 +00001893 return Error("Invalid CALL record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001894
Devang Patel05988662008-09-25 21:00:45 +00001895 AttrListPtr PAL = getAttributes(Record[0]);
Chris Lattnera9bb7132007-05-08 05:38:01 +00001896 unsigned CCInfo = Record[1];
1897
1898 unsigned OpNum = 2;
Chris Lattner7337ab92007-05-06 00:00:00 +00001899 Value *Callee;
1900 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
1901 return Error("Invalid CALL record");
1902
1903 const PointerType *OpTy = dyn_cast<PointerType>(Callee->getType());
Chris Lattner0579f7f2007-05-03 22:04:19 +00001904 const FunctionType *FTy = 0;
1905 if (OpTy) FTy = dyn_cast<FunctionType>(OpTy->getElementType());
Chris Lattner7337ab92007-05-06 00:00:00 +00001906 if (!FTy || Record.size() < FTy->getNumParams()+OpNum)
Chris Lattner0579f7f2007-05-03 22:04:19 +00001907 return Error("Invalid CALL record");
1908
1909 SmallVector<Value*, 16> Args;
1910 // Read the fixed params.
Chris Lattner7337ab92007-05-06 00:00:00 +00001911 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
Dale Johanneseneb57ea72007-11-05 21:20:28 +00001912 if (FTy->getParamType(i)->getTypeID()==Type::LabelTyID)
1913 Args.push_back(getBasicBlock(Record[OpNum]));
1914 else
1915 Args.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
Chris Lattner0579f7f2007-05-03 22:04:19 +00001916 if (Args.back() == 0) return Error("Invalid CALL record");
1917 }
1918
Chris Lattner0579f7f2007-05-03 22:04:19 +00001919 // Read type/value pairs for varargs params.
Chris Lattner0579f7f2007-05-03 22:04:19 +00001920 if (!FTy->isVarArg()) {
Chris Lattner7337ab92007-05-06 00:00:00 +00001921 if (OpNum != Record.size())
Chris Lattner0579f7f2007-05-03 22:04:19 +00001922 return Error("Invalid CALL record");
1923 } else {
Chris Lattner7337ab92007-05-06 00:00:00 +00001924 while (OpNum != Record.size()) {
1925 Value *Op;
1926 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1927 return Error("Invalid CALL record");
1928 Args.push_back(Op);
Chris Lattner0579f7f2007-05-03 22:04:19 +00001929 }
1930 }
1931
Gabor Greif051a9502008-04-06 20:25:17 +00001932 I = CallInst::Create(Callee, Args.begin(), Args.end());
Chris Lattner76520192007-05-03 22:34:03 +00001933 cast<CallInst>(I)->setCallingConv(CCInfo>>1);
1934 cast<CallInst>(I)->setTailCall(CCInfo & 1);
Devang Patel05988662008-09-25 21:00:45 +00001935 cast<CallInst>(I)->setAttributes(PAL);
Chris Lattner0579f7f2007-05-03 22:04:19 +00001936 break;
1937 }
1938 case bitc::FUNC_CODE_INST_VAARG: { // VAARG: [valistty, valist, instty]
1939 if (Record.size() < 3)
1940 return Error("Invalid VAARG record");
1941 const Type *OpTy = getTypeByID(Record[0]);
1942 Value *Op = getFnValueByID(Record[1], OpTy);
1943 const Type *ResTy = getTypeByID(Record[2]);
1944 if (!OpTy || !Op || !ResTy)
1945 return Error("Invalid VAARG record");
1946 I = new VAArgInst(Op, ResTy);
1947 break;
1948 }
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001949 }
1950
1951 // Add instruction to end of current BB. If there is no current BB, reject
1952 // this file.
1953 if (CurBB == 0) {
1954 delete I;
1955 return Error("Invalid instruction with no BB");
1956 }
1957 CurBB->getInstList().push_back(I);
1958
1959 // If this was a terminator instruction, move to the next block.
1960 if (isa<TerminatorInst>(I)) {
1961 ++CurBBNo;
1962 CurBB = CurBBNo < FunctionBBs.size() ? FunctionBBs[CurBBNo] : 0;
1963 }
1964
1965 // Non-void values get registered in the value table for future use.
1966 if (I && I->getType() != Type::VoidTy)
1967 ValueList.AssignValue(I, NextValueNo++);
Chris Lattner980e5aa2007-05-01 05:52:21 +00001968 }
1969
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001970 // Check the function list for unresolved values.
1971 if (Argument *A = dyn_cast<Argument>(ValueList.back())) {
1972 if (A->getParent() == 0) {
1973 // We found at least one unresolved value. Nuke them all to avoid leaks.
1974 for (unsigned i = ModuleValueListSize, e = ValueList.size(); i != e; ++i){
1975 if ((A = dyn_cast<Argument>(ValueList.back())) && A->getParent() == 0) {
1976 A->replaceAllUsesWith(UndefValue::get(A->getType()));
1977 delete A;
1978 }
1979 }
Chris Lattner35a04702007-05-04 03:50:29 +00001980 return Error("Never resolved value found in function!");
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001981 }
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001982 }
Chris Lattner980e5aa2007-05-01 05:52:21 +00001983
1984 // Trim the value list down to the size it was before we parsed this function.
1985 ValueList.shrinkTo(ModuleValueListSize);
1986 std::vector<BasicBlock*>().swap(FunctionBBs);
1987
Chris Lattner48f84872007-05-01 04:59:48 +00001988 return false;
1989}
1990
Chris Lattnerb348bb82007-05-18 04:02:46 +00001991//===----------------------------------------------------------------------===//
1992// ModuleProvider implementation
1993//===----------------------------------------------------------------------===//
1994
1995
1996bool BitcodeReader::materializeFunction(Function *F, std::string *ErrInfo) {
1997 // If it already is material, ignore the request.
Gabor Greifa99be512007-07-05 17:07:56 +00001998 if (!F->hasNotBeenReadFromBitcode()) return false;
Chris Lattnerb348bb82007-05-18 04:02:46 +00001999
2000 DenseMap<Function*, std::pair<uint64_t, unsigned> >::iterator DFII =
2001 DeferredFunctionInfo.find(F);
2002 assert(DFII != DeferredFunctionInfo.end() && "Deferred function not found!");
2003
2004 // Move the bit stream to the saved position of the deferred function body and
2005 // restore the real linkage type for the function.
2006 Stream.JumpToBit(DFII->second.first);
2007 F->setLinkage((GlobalValue::LinkageTypes)DFII->second.second);
2008
2009 if (ParseFunctionBody(F)) {
2010 if (ErrInfo) *ErrInfo = ErrorString;
2011 return true;
2012 }
Chandler Carruth69940402007-08-04 01:51:18 +00002013
2014 // Upgrade any old intrinsic calls in the function.
2015 for (UpgradedIntrinsicMap::iterator I = UpgradedIntrinsics.begin(),
2016 E = UpgradedIntrinsics.end(); I != E; ++I) {
2017 if (I->first != I->second) {
2018 for (Value::use_iterator UI = I->first->use_begin(),
2019 UE = I->first->use_end(); UI != UE; ) {
2020 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2021 UpgradeIntrinsicCall(CI, I->second);
2022 }
2023 }
2024 }
Chris Lattnerb348bb82007-05-18 04:02:46 +00002025
2026 return false;
2027}
2028
2029void BitcodeReader::dematerializeFunction(Function *F) {
2030 // If this function isn't materialized, or if it is a proto, this is a noop.
Gabor Greifa99be512007-07-05 17:07:56 +00002031 if (F->hasNotBeenReadFromBitcode() || F->isDeclaration())
Chris Lattnerb348bb82007-05-18 04:02:46 +00002032 return;
2033
2034 assert(DeferredFunctionInfo.count(F) && "No info to read function later?");
2035
2036 // Just forget the function body, we can remat it later.
2037 F->deleteBody();
2038 F->setLinkage(GlobalValue::GhostLinkage);
2039}
2040
2041
2042Module *BitcodeReader::materializeModule(std::string *ErrInfo) {
Chris Lattner714fa952009-06-16 05:15:21 +00002043 // Iterate over the module, deserializing any functions that are still on
2044 // disk.
2045 for (Module::iterator F = TheModule->begin(), E = TheModule->end();
2046 F != E; ++F)
Gabor Greifa99be512007-07-05 17:07:56 +00002047 if (F->hasNotBeenReadFromBitcode() &&
Chris Lattnerb348bb82007-05-18 04:02:46 +00002048 materializeFunction(F, ErrInfo))
2049 return 0;
Chandler Carruth69940402007-08-04 01:51:18 +00002050
2051 // Upgrade any intrinsic calls that slipped through (should not happen!) and
2052 // delete the old functions to clean up. We can't do this unless the entire
2053 // module is materialized because there could always be another function body
2054 // with calls to the old function.
2055 for (std::vector<std::pair<Function*, Function*> >::iterator I =
2056 UpgradedIntrinsics.begin(), E = UpgradedIntrinsics.end(); I != E; ++I) {
2057 if (I->first != I->second) {
2058 for (Value::use_iterator UI = I->first->use_begin(),
2059 UE = I->first->use_end(); UI != UE; ) {
2060 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2061 UpgradeIntrinsicCall(CI, I->second);
2062 }
Chris Lattner7d9eb582009-04-01 01:43:03 +00002063 if (!I->first->use_empty())
2064 I->first->replaceAllUsesWith(I->second);
Chandler Carruth69940402007-08-04 01:51:18 +00002065 I->first->eraseFromParent();
2066 }
2067 }
2068 std::vector<std::pair<Function*, Function*> >().swap(UpgradedIntrinsics);
2069
Chris Lattnerb348bb82007-05-18 04:02:46 +00002070 return TheModule;
2071}
2072
2073
2074/// This method is provided by the parent ModuleProvde class and overriden
2075/// here. It simply releases the module from its provided and frees up our
2076/// state.
2077/// @brief Release our hold on the generated module
2078Module *BitcodeReader::releaseModule(std::string *ErrInfo) {
2079 // Since we're losing control of this Module, we must hand it back complete
2080 Module *M = ModuleProvider::releaseModule(ErrInfo);
2081 FreeState();
2082 return M;
2083}
2084
Chris Lattner48f84872007-05-01 04:59:48 +00002085
Chris Lattnerc453f762007-04-29 07:54:31 +00002086//===----------------------------------------------------------------------===//
2087// External interface
2088//===----------------------------------------------------------------------===//
2089
2090/// getBitcodeModuleProvider - lazy function-at-a-time loading from a file.
2091///
2092ModuleProvider *llvm::getBitcodeModuleProvider(MemoryBuffer *Buffer,
Owen Anderson8b477ed2009-07-01 16:58:40 +00002093 LLVMContext* Context,
Chris Lattnerc453f762007-04-29 07:54:31 +00002094 std::string *ErrMsg) {
Owen Anderson8b477ed2009-07-01 16:58:40 +00002095 BitcodeReader *R = new BitcodeReader(Buffer, Context);
Chris Lattnerc453f762007-04-29 07:54:31 +00002096 if (R->ParseBitcode()) {
2097 if (ErrMsg)
2098 *ErrMsg = R->getErrorString();
2099
2100 // Don't let the BitcodeReader dtor delete 'Buffer'.
2101 R->releaseMemoryBuffer();
2102 delete R;
2103 return 0;
2104 }
2105 return R;
2106}
2107
2108/// ParseBitcodeFile - Read the specified bitcode file, returning the module.
2109/// If an error occurs, return null and fill in *ErrMsg if non-null.
Owen Anderson8b477ed2009-07-01 16:58:40 +00002110Module *llvm::ParseBitcodeFile(MemoryBuffer *Buffer, LLVMContext* Context,
2111 std::string *ErrMsg){
Chris Lattnerc453f762007-04-29 07:54:31 +00002112 BitcodeReader *R;
Owen Anderson8b477ed2009-07-01 16:58:40 +00002113 R = static_cast<BitcodeReader*>(getBitcodeModuleProvider(Buffer, Context,
2114 ErrMsg));
Chris Lattnerc453f762007-04-29 07:54:31 +00002115 if (!R) return 0;
2116
Chris Lattnerb348bb82007-05-18 04:02:46 +00002117 // Read in the entire module.
2118 Module *M = R->materializeModule(ErrMsg);
2119
2120 // Don't let the BitcodeReader dtor delete 'Buffer', regardless of whether
2121 // there was an error.
Chris Lattnerc453f762007-04-29 07:54:31 +00002122 R->releaseMemoryBuffer();
Chris Lattnerb348bb82007-05-18 04:02:46 +00002123
2124 // If there was no error, tell ModuleProvider not to delete it when its dtor
2125 // is run.
2126 if (M)
2127 M = R->releaseModule(ErrMsg);
Chris Lattner714fa952009-06-16 05:15:21 +00002128
Chris Lattnerc453f762007-04-29 07:54:31 +00002129 delete R;
2130 return M;
2131}