blob: 168f32951d56bba276ad31bc1fe509dd7ef48d89 [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"
Owen Anderson74a77812009-07-07 20:18:58 +000020#include "llvm/LLVMContext.h"
Devang Patel0a9f7b92009-07-28 21:49:47 +000021#include "llvm/Metadata.h"
Chris Lattnercaee0dc2007-04-22 06:23:29 +000022#include "llvm/Module.h"
Dan Gohman1224c382009-07-20 21:19:07 +000023#include "llvm/Operator.h"
Chandler Carruth69940402007-08-04 01:51:18 +000024#include "llvm/AutoUpgrade.h"
Chris Lattner0b2482a2007-04-23 21:26:05 +000025#include "llvm/ADT/SmallString.h"
Devang Patelf4511cd2008-02-26 19:38:17 +000026#include "llvm/ADT/SmallVector.h"
Chris Lattner0eef0802007-04-24 04:04:35 +000027#include "llvm/Support/MathExtras.h"
Chris Lattnerc453f762007-04-29 07:54:31 +000028#include "llvm/Support/MemoryBuffer.h"
Gabor Greifefe65362008-05-10 08:32:32 +000029#include "llvm/OperandTraits.h"
Chris Lattnercaee0dc2007-04-22 06:23:29 +000030using namespace llvm;
31
Chris Lattnerb348bb82007-05-18 04:02:46 +000032void BitcodeReader::FreeState() {
Chris Lattnerc453f762007-04-29 07:54:31 +000033 delete Buffer;
Chris Lattnerb348bb82007-05-18 04:02:46 +000034 Buffer = 0;
35 std::vector<PATypeHolder>().swap(TypeList);
36 ValueList.clear();
Devang Pateld5ac4042009-08-04 06:00:18 +000037 MDValueList.clear();
Chris Lattner461edd92008-03-12 02:25:52 +000038
Devang Patel19c87462008-09-26 22:53:05 +000039 std::vector<AttrListPtr>().swap(MAttributes);
Chris Lattnerb348bb82007-05-18 04:02:46 +000040 std::vector<BasicBlock*>().swap(FunctionBBs);
41 std::vector<Function*>().swap(FunctionsWithBodies);
42 DeferredFunctionInfo.clear();
Chris Lattnerc453f762007-04-29 07:54:31 +000043}
44
Chris Lattner48c85b82007-05-04 03:30:17 +000045//===----------------------------------------------------------------------===//
46// Helper functions to implement forward reference resolution, etc.
47//===----------------------------------------------------------------------===//
Chris Lattnerc453f762007-04-29 07:54:31 +000048
Chris Lattnercaee0dc2007-04-22 06:23:29 +000049/// ConvertToString - Convert a string from a record into an std::string, return
50/// true on failure.
Chris Lattner0b2482a2007-04-23 21:26:05 +000051template<typename StrTy>
Chris Lattnercaee0dc2007-04-22 06:23:29 +000052static bool ConvertToString(SmallVector<uint64_t, 64> &Record, unsigned Idx,
Chris Lattner0b2482a2007-04-23 21:26:05 +000053 StrTy &Result) {
Chris Lattner15e6d172007-05-04 19:11:41 +000054 if (Idx > Record.size())
Chris Lattnercaee0dc2007-04-22 06:23:29 +000055 return true;
56
Chris Lattner15e6d172007-05-04 19:11:41 +000057 for (unsigned i = Idx, e = Record.size(); i != e; ++i)
58 Result += (char)Record[i];
Chris Lattnercaee0dc2007-04-22 06:23:29 +000059 return false;
60}
61
62static GlobalValue::LinkageTypes GetDecodedLinkage(unsigned Val) {
63 switch (Val) {
64 default: // Map unknown/new linkages to external
Bill Wendling3d10a5a2009-07-20 01:03:30 +000065 case 0: return GlobalValue::ExternalLinkage;
66 case 1: return GlobalValue::WeakAnyLinkage;
67 case 2: return GlobalValue::AppendingLinkage;
68 case 3: return GlobalValue::InternalLinkage;
69 case 4: return GlobalValue::LinkOnceAnyLinkage;
70 case 5: return GlobalValue::DLLImportLinkage;
71 case 6: return GlobalValue::DLLExportLinkage;
72 case 7: return GlobalValue::ExternalWeakLinkage;
73 case 8: return GlobalValue::CommonLinkage;
74 case 9: return GlobalValue::PrivateLinkage;
Duncan Sands667d4b82009-03-07 15:45:40 +000075 case 10: return GlobalValue::WeakODRLinkage;
76 case 11: return GlobalValue::LinkOnceODRLinkage;
Chris Lattner266c7bb2009-04-13 05:44:34 +000077 case 12: return GlobalValue::AvailableExternallyLinkage;
Bill Wendling3d10a5a2009-07-20 01:03:30 +000078 case 13: return GlobalValue::LinkerPrivateLinkage;
Chris Lattnercaee0dc2007-04-22 06:23:29 +000079 }
80}
81
82static GlobalValue::VisibilityTypes GetDecodedVisibility(unsigned Val) {
83 switch (Val) {
84 default: // Map unknown visibilities to default.
85 case 0: return GlobalValue::DefaultVisibility;
86 case 1: return GlobalValue::HiddenVisibility;
Anton Korobeynikov9cd3ccf2007-04-29 20:56:48 +000087 case 2: return GlobalValue::ProtectedVisibility;
Chris Lattnercaee0dc2007-04-22 06:23:29 +000088 }
89}
90
Chris Lattnerf581c3b2007-04-24 07:07:11 +000091static int GetDecodedCastOpcode(unsigned Val) {
92 switch (Val) {
93 default: return -1;
94 case bitc::CAST_TRUNC : return Instruction::Trunc;
95 case bitc::CAST_ZEXT : return Instruction::ZExt;
96 case bitc::CAST_SEXT : return Instruction::SExt;
97 case bitc::CAST_FPTOUI : return Instruction::FPToUI;
98 case bitc::CAST_FPTOSI : return Instruction::FPToSI;
99 case bitc::CAST_UITOFP : return Instruction::UIToFP;
100 case bitc::CAST_SITOFP : return Instruction::SIToFP;
101 case bitc::CAST_FPTRUNC : return Instruction::FPTrunc;
102 case bitc::CAST_FPEXT : return Instruction::FPExt;
103 case bitc::CAST_PTRTOINT: return Instruction::PtrToInt;
104 case bitc::CAST_INTTOPTR: return Instruction::IntToPtr;
105 case bitc::CAST_BITCAST : return Instruction::BitCast;
106 }
107}
108static int GetDecodedBinaryOpcode(unsigned Val, const Type *Ty) {
109 switch (Val) {
110 default: return -1;
Dan Gohmanae3a0be2009-06-04 22:49:04 +0000111 case bitc::BINOP_ADD:
112 return Ty->isFPOrFPVector() ? Instruction::FAdd : Instruction::Add;
113 case bitc::BINOP_SUB:
114 return Ty->isFPOrFPVector() ? Instruction::FSub : Instruction::Sub;
115 case bitc::BINOP_MUL:
116 return Ty->isFPOrFPVector() ? Instruction::FMul : Instruction::Mul;
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000117 case bitc::BINOP_UDIV: return Instruction::UDiv;
118 case bitc::BINOP_SDIV:
119 return Ty->isFPOrFPVector() ? Instruction::FDiv : Instruction::SDiv;
120 case bitc::BINOP_UREM: return Instruction::URem;
121 case bitc::BINOP_SREM:
122 return Ty->isFPOrFPVector() ? Instruction::FRem : Instruction::SRem;
123 case bitc::BINOP_SHL: return Instruction::Shl;
124 case bitc::BINOP_LSHR: return Instruction::LShr;
125 case bitc::BINOP_ASHR: return Instruction::AShr;
126 case bitc::BINOP_AND: return Instruction::And;
127 case bitc::BINOP_OR: return Instruction::Or;
128 case bitc::BINOP_XOR: return Instruction::Xor;
129 }
130}
131
Gabor Greifefe65362008-05-10 08:32:32 +0000132namespace llvm {
Chris Lattner522b7b12007-04-24 05:48:56 +0000133namespace {
134 /// @brief A class for maintaining the slot number definition
135 /// as a placeholder for the actual definition for forward constants defs.
136 class ConstantPlaceHolder : public ConstantExpr {
137 ConstantPlaceHolder(); // DO NOT IMPLEMENT
138 void operator=(const ConstantPlaceHolder &); // DO NOT IMPLEMENT
Gabor Greif051a9502008-04-06 20:25:17 +0000139 public:
140 // allocate space for exactly one operand
141 void *operator new(size_t s) {
142 return User::operator new(s, 1);
143 }
Owen Anderson74a77812009-07-07 20:18:58 +0000144 explicit ConstantPlaceHolder(const Type *Ty, LLVMContext& Context)
Gabor Greifefe65362008-05-10 08:32:32 +0000145 : ConstantExpr(Ty, Instruction::UserOp1, &Op<0>(), 1) {
Owen Anderson1d0be152009-08-13 21:58:54 +0000146 Op<0>() = UndefValue::get(Type::getInt32Ty(Context));
Chris Lattner522b7b12007-04-24 05:48:56 +0000147 }
Chris Lattnerea693df2008-08-21 02:34:16 +0000148
149 /// @brief Methods to support type inquiry through isa, cast, and dyn_cast.
150 static inline bool classof(const ConstantPlaceHolder *) { return true; }
151 static bool classof(const Value *V) {
152 return isa<ConstantExpr>(V) &&
153 cast<ConstantExpr>(V)->getOpcode() == Instruction::UserOp1;
154 }
155
156
Gabor Greifefe65362008-05-10 08:32:32 +0000157 /// Provide fast operand accessors
Chris Lattner46e77402009-03-31 22:55:09 +0000158 //DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
Chris Lattner522b7b12007-04-24 05:48:56 +0000159 };
160}
161
Chris Lattner46e77402009-03-31 22:55:09 +0000162// FIXME: can we inherit this from ConstantExpr?
Gabor Greifefe65362008-05-10 08:32:32 +0000163template <>
164struct OperandTraits<ConstantPlaceHolder> : FixedNumOperandTraits<1> {
165};
Gabor Greifefe65362008-05-10 08:32:32 +0000166}
167
Chris Lattner46e77402009-03-31 22:55:09 +0000168
169void BitcodeReaderValueList::AssignValue(Value *V, unsigned Idx) {
170 if (Idx == size()) {
171 push_back(V);
172 return;
173 }
174
175 if (Idx >= size())
176 resize(Idx+1);
177
178 WeakVH &OldV = ValuePtrs[Idx];
179 if (OldV == 0) {
180 OldV = V;
181 return;
182 }
183
184 // Handle constants and non-constants (e.g. instrs) differently for
185 // efficiency.
186 if (Constant *PHC = dyn_cast<Constant>(&*OldV)) {
187 ResolveConstants.push_back(std::make_pair(PHC, Idx));
188 OldV = V;
189 } else {
190 // If there was a forward reference to this value, replace it.
191 Value *PrevVal = OldV;
192 OldV->replaceAllUsesWith(V);
193 delete PrevVal;
Gabor Greifefe65362008-05-10 08:32:32 +0000194 }
195}
Chris Lattner46e77402009-03-31 22:55:09 +0000196
Gabor Greifefe65362008-05-10 08:32:32 +0000197
Chris Lattner522b7b12007-04-24 05:48:56 +0000198Constant *BitcodeReaderValueList::getConstantFwdRef(unsigned Idx,
199 const Type *Ty) {
Chris Lattner46e77402009-03-31 22:55:09 +0000200 if (Idx >= size())
Gabor Greifefe65362008-05-10 08:32:32 +0000201 resize(Idx + 1);
Chris Lattner522b7b12007-04-24 05:48:56 +0000202
Chris Lattner46e77402009-03-31 22:55:09 +0000203 if (Value *V = ValuePtrs[Idx]) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000204 assert(Ty == V->getType() && "Type mismatch in constant table!");
205 return cast<Constant>(V);
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000206 }
Chris Lattner522b7b12007-04-24 05:48:56 +0000207
208 // Create and return a placeholder, which will later be RAUW'd.
Owen Anderson74a77812009-07-07 20:18:58 +0000209 Constant *C = new ConstantPlaceHolder(Ty, Context);
Chris Lattner46e77402009-03-31 22:55:09 +0000210 ValuePtrs[Idx] = C;
Chris Lattner522b7b12007-04-24 05:48:56 +0000211 return C;
212}
213
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000214Value *BitcodeReaderValueList::getValueFwdRef(unsigned Idx, const Type *Ty) {
Chris Lattner46e77402009-03-31 22:55:09 +0000215 if (Idx >= size())
Gabor Greifefe65362008-05-10 08:32:32 +0000216 resize(Idx + 1);
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000217
Chris Lattner46e77402009-03-31 22:55:09 +0000218 if (Value *V = ValuePtrs[Idx]) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000219 assert((Ty == 0 || Ty == V->getType()) && "Type mismatch in value table!");
220 return V;
221 }
222
Chris Lattner01ff65f2007-05-02 05:16:49 +0000223 // No type specified, must be invalid reference.
224 if (Ty == 0) return 0;
225
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000226 // Create and return a placeholder, which will later be RAUW'd.
227 Value *V = new Argument(Ty);
Chris Lattner46e77402009-03-31 22:55:09 +0000228 ValuePtrs[Idx] = V;
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000229 return V;
230}
231
Chris Lattnerea693df2008-08-21 02:34:16 +0000232/// ResolveConstantForwardRefs - Once all constants are read, this method bulk
233/// resolves any forward references. The idea behind this is that we sometimes
234/// get constants (such as large arrays) which reference *many* forward ref
235/// constants. Replacing each of these causes a lot of thrashing when
236/// building/reuniquing the constant. Instead of doing this, we look at all the
237/// uses and rewrite all the place holders at once for any constant that uses
238/// a placeholder.
239void BitcodeReaderValueList::ResolveConstantForwardRefs() {
240 // Sort the values by-pointer so that they are efficient to look up with a
241 // binary search.
242 std::sort(ResolveConstants.begin(), ResolveConstants.end());
243
244 SmallVector<Constant*, 64> NewOps;
245
246 while (!ResolveConstants.empty()) {
Chris Lattner46e77402009-03-31 22:55:09 +0000247 Value *RealVal = operator[](ResolveConstants.back().second);
Chris Lattnerea693df2008-08-21 02:34:16 +0000248 Constant *Placeholder = ResolveConstants.back().first;
249 ResolveConstants.pop_back();
250
251 // Loop over all users of the placeholder, updating them to reference the
252 // new value. If they reference more than one placeholder, update them all
253 // at once.
254 while (!Placeholder->use_empty()) {
Chris Lattnerb6135a02008-08-21 17:31:45 +0000255 Value::use_iterator UI = Placeholder->use_begin();
256
Chris Lattnerea693df2008-08-21 02:34:16 +0000257 // If the using object isn't uniqued, just update the operands. This
258 // handles instructions and initializers for global variables.
Chris Lattnerb6135a02008-08-21 17:31:45 +0000259 if (!isa<Constant>(*UI) || isa<GlobalValue>(*UI)) {
260 UI.getUse().set(RealVal);
Chris Lattnerea693df2008-08-21 02:34:16 +0000261 continue;
262 }
263
264 // Otherwise, we have a constant that uses the placeholder. Replace that
265 // constant with a new constant that has *all* placeholder uses updated.
Chris Lattnerb6135a02008-08-21 17:31:45 +0000266 Constant *UserC = cast<Constant>(*UI);
Chris Lattnerea693df2008-08-21 02:34:16 +0000267 for (User::op_iterator I = UserC->op_begin(), E = UserC->op_end();
268 I != E; ++I) {
269 Value *NewOp;
270 if (!isa<ConstantPlaceHolder>(*I)) {
271 // Not a placeholder reference.
272 NewOp = *I;
273 } else if (*I == Placeholder) {
274 // Common case is that it just references this one placeholder.
275 NewOp = RealVal;
276 } else {
277 // Otherwise, look up the placeholder in ResolveConstants.
278 ResolveConstantsTy::iterator It =
279 std::lower_bound(ResolveConstants.begin(), ResolveConstants.end(),
280 std::pair<Constant*, unsigned>(cast<Constant>(*I),
281 0));
282 assert(It != ResolveConstants.end() && It->first == *I);
Chris Lattner46e77402009-03-31 22:55:09 +0000283 NewOp = operator[](It->second);
Chris Lattnerea693df2008-08-21 02:34:16 +0000284 }
285
286 NewOps.push_back(cast<Constant>(NewOp));
287 }
288
289 // Make the new constant.
290 Constant *NewC;
291 if (ConstantArray *UserCA = dyn_cast<ConstantArray>(UserC)) {
Owen Anderson1fd70962009-07-28 18:32:17 +0000292 NewC = ConstantArray::get(UserCA->getType(), &NewOps[0],
Owen Anderson74a77812009-07-07 20:18:58 +0000293 NewOps.size());
Chris Lattnerea693df2008-08-21 02:34:16 +0000294 } else if (ConstantStruct *UserCS = dyn_cast<ConstantStruct>(UserC)) {
Owen Andersond7f2a6c2009-08-05 23:16:16 +0000295 NewC = ConstantStruct::get(Context, &NewOps[0], NewOps.size(),
Owen Anderson74a77812009-07-07 20:18:58 +0000296 UserCS->getType()->isPacked());
Chris Lattnerea693df2008-08-21 02:34:16 +0000297 } else if (isa<ConstantVector>(UserC)) {
Owen Andersonaf7ec972009-07-28 21:19:26 +0000298 NewC = ConstantVector::get(&NewOps[0], NewOps.size());
Nick Lewyckycb337992009-05-10 20:57:05 +0000299 } else {
300 assert(isa<ConstantExpr>(UserC) && "Must be a ConstantExpr.");
Chris Lattnerea693df2008-08-21 02:34:16 +0000301 NewC = cast<ConstantExpr>(UserC)->getWithOperands(&NewOps[0],
302 NewOps.size());
303 }
304
305 UserC->replaceAllUsesWith(NewC);
306 UserC->destroyConstant();
307 NewOps.clear();
308 }
309
Nick Lewyckycb337992009-05-10 20:57:05 +0000310 // Update all ValueHandles, they should be the only users at this point.
311 Placeholder->replaceAllUsesWith(RealVal);
Chris Lattnerea693df2008-08-21 02:34:16 +0000312 delete Placeholder;
313 }
314}
315
Devang Pateld5ac4042009-08-04 06:00:18 +0000316void BitcodeReaderMDValueList::AssignValue(Value *V, unsigned Idx) {
317 if (Idx == size()) {
318 push_back(V);
319 return;
320 }
321
322 if (Idx >= size())
323 resize(Idx+1);
324
325 WeakVH &OldV = MDValuePtrs[Idx];
326 if (OldV == 0) {
327 OldV = V;
328 return;
329 }
330
331 // If there was a forward reference to this value, replace it.
332 Value *PrevVal = OldV;
333 OldV->replaceAllUsesWith(V);
334 delete PrevVal;
335}
336
337Value *BitcodeReaderMDValueList::getValueFwdRef(unsigned Idx) {
338 if (Idx >= size())
339 resize(Idx + 1);
340
341 if (Value *V = MDValuePtrs[Idx]) {
Owen Anderson1d0be152009-08-13 21:58:54 +0000342 assert(V->getType() == Type::getMetadataTy(Context) && "Type mismatch in value table!");
Devang Pateld5ac4042009-08-04 06:00:18 +0000343 return V;
344 }
345
346 // Create and return a placeholder, which will later be RAUW'd.
Owen Anderson1d0be152009-08-13 21:58:54 +0000347 Value *V = new Argument(Type::getMetadataTy(Context));
Devang Pateld5ac4042009-08-04 06:00:18 +0000348 MDValuePtrs[Idx] = V;
349 return V;
350}
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000351
352const Type *BitcodeReader::getTypeByID(unsigned ID, bool isTypeTable) {
353 // If the TypeID is in range, return it.
354 if (ID < TypeList.size())
355 return TypeList[ID].get();
356 if (!isTypeTable) return 0;
357
358 // The type table allows forward references. Push as many Opaque types as
359 // needed to get up to ID.
360 while (TypeList.size() <= ID)
Owen Andersondebcb012009-07-29 22:17:13 +0000361 TypeList.push_back(OpaqueType::get());
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000362 return TypeList.back().get();
363}
364
Chris Lattner48c85b82007-05-04 03:30:17 +0000365//===----------------------------------------------------------------------===//
366// Functions for parsing blocks from the bitcode file
367//===----------------------------------------------------------------------===//
368
Devang Patel05988662008-09-25 21:00:45 +0000369bool BitcodeReader::ParseAttributeBlock() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000370 if (Stream.EnterSubBlock(bitc::PARAMATTR_BLOCK_ID))
Chris Lattner48c85b82007-05-04 03:30:17 +0000371 return Error("Malformed block record");
372
Devang Patel19c87462008-09-26 22:53:05 +0000373 if (!MAttributes.empty())
Chris Lattner48c85b82007-05-04 03:30:17 +0000374 return Error("Multiple PARAMATTR blocks found!");
375
376 SmallVector<uint64_t, 64> Record;
377
Devang Patel05988662008-09-25 21:00:45 +0000378 SmallVector<AttributeWithIndex, 8> Attrs;
Chris Lattner48c85b82007-05-04 03:30:17 +0000379
380 // Read all the records.
381 while (1) {
382 unsigned Code = Stream.ReadCode();
383 if (Code == bitc::END_BLOCK) {
384 if (Stream.ReadBlockEnd())
385 return Error("Error at end of PARAMATTR block");
386 return false;
387 }
388
389 if (Code == bitc::ENTER_SUBBLOCK) {
390 // No known subblocks, always skip them.
391 Stream.ReadSubBlockID();
392 if (Stream.SkipBlock())
393 return Error("Malformed block record");
394 continue;
395 }
396
397 if (Code == bitc::DEFINE_ABBREV) {
398 Stream.ReadAbbrevRecord();
399 continue;
400 }
401
402 // Read a record.
403 Record.clear();
404 switch (Stream.ReadRecord(Code, Record)) {
405 default: // Default behavior: ignore.
406 break;
407 case bitc::PARAMATTR_CODE_ENTRY: { // ENTRY: [paramidx0, attr0, ...]
408 if (Record.size() & 1)
409 return Error("Invalid ENTRY record");
410
Chris Lattner9a6cb152008-10-05 18:22:09 +0000411 // FIXME : Remove this autoupgrade code in LLVM 3.0.
Devang Patel19c87462008-09-26 22:53:05 +0000412 // If Function attributes are using index 0 then transfer them
Chris Lattner9a6cb152008-10-05 18:22:09 +0000413 // to index ~0. Index 0 is used for return value attributes but used to be
414 // used for function attributes.
Devang Patel19c87462008-09-26 22:53:05 +0000415 Attributes RetAttribute = Attribute::None;
416 Attributes FnAttribute = Attribute::None;
Chris Lattner48c85b82007-05-04 03:30:17 +0000417 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Nick Lewycky73ddd4f2008-12-19 09:38:31 +0000418 // FIXME: remove in LLVM 3.0
419 // The alignment is stored as a 16-bit raw value from bits 31--16.
420 // We shift the bits above 31 down by 11 bits.
421
422 unsigned Alignment = (Record[i+1] & (0xffffull << 16)) >> 16;
423 if (Alignment && !isPowerOf2_32(Alignment))
424 return Error("Alignment is not a power of two.");
425
426 Attributes ReconstitutedAttr = Record[i+1] & 0xffff;
427 if (Alignment)
428 ReconstitutedAttr |= Attribute::constructAlignmentFromInt(Alignment);
429 ReconstitutedAttr |= (Record[i+1] & (0xffffull << 32)) >> 11;
430 Record[i+1] = ReconstitutedAttr;
431
Devang Patel19c87462008-09-26 22:53:05 +0000432 if (Record[i] == 0)
433 RetAttribute = Record[i+1];
434 else if (Record[i] == ~0U)
435 FnAttribute = Record[i+1];
436 }
Chris Lattner9a6cb152008-10-05 18:22:09 +0000437
438 unsigned OldRetAttrs = (Attribute::NoUnwind|Attribute::NoReturn|
439 Attribute::ReadOnly|Attribute::ReadNone);
440
441 if (FnAttribute == Attribute::None && RetAttribute != Attribute::None &&
442 (RetAttribute & OldRetAttrs) != 0) {
443 if (FnAttribute == Attribute::None) { // add a slot so they get added.
444 Record.push_back(~0U);
445 Record.push_back(0);
Devang Patel19c87462008-09-26 22:53:05 +0000446 }
Chris Lattner9a6cb152008-10-05 18:22:09 +0000447
448 FnAttribute |= RetAttribute & OldRetAttrs;
449 RetAttribute &= ~OldRetAttrs;
Chris Lattner48c85b82007-05-04 03:30:17 +0000450 }
Chris Lattner461edd92008-03-12 02:25:52 +0000451
Devang Patel19c87462008-09-26 22:53:05 +0000452 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Chris Lattner9a6cb152008-10-05 18:22:09 +0000453 if (Record[i] == 0) {
454 if (RetAttribute != Attribute::None)
455 Attrs.push_back(AttributeWithIndex::get(0, RetAttribute));
456 } else if (Record[i] == ~0U) {
457 if (FnAttribute != Attribute::None)
458 Attrs.push_back(AttributeWithIndex::get(~0U, FnAttribute));
459 } else if (Record[i+1] != Attribute::None)
Devang Patel19c87462008-09-26 22:53:05 +0000460 Attrs.push_back(AttributeWithIndex::get(Record[i], Record[i+1]));
461 }
Devang Patel19c87462008-09-26 22:53:05 +0000462
463 MAttributes.push_back(AttrListPtr::get(Attrs.begin(), Attrs.end()));
Chris Lattner48c85b82007-05-04 03:30:17 +0000464 Attrs.clear();
465 break;
466 }
Duncan Sands5e41f652007-11-20 14:09:29 +0000467 }
Chris Lattner48c85b82007-05-04 03:30:17 +0000468 }
469}
470
471
Chris Lattner86697142007-05-01 05:01:34 +0000472bool BitcodeReader::ParseTypeTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000473 if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000474 return Error("Malformed block record");
475
476 if (!TypeList.empty())
477 return Error("Multiple TYPE_BLOCKs found!");
478
479 SmallVector<uint64_t, 64> Record;
480 unsigned NumRecords = 0;
481
482 // Read all the records for this type table.
483 while (1) {
484 unsigned Code = Stream.ReadCode();
485 if (Code == bitc::END_BLOCK) {
486 if (NumRecords != TypeList.size())
487 return Error("Invalid type forward reference in TYPE_BLOCK");
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000488 if (Stream.ReadBlockEnd())
489 return Error("Error at end of type table block");
490 return false;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000491 }
492
493 if (Code == bitc::ENTER_SUBBLOCK) {
494 // No known subblocks, always skip them.
495 Stream.ReadSubBlockID();
496 if (Stream.SkipBlock())
497 return Error("Malformed block record");
498 continue;
499 }
500
Chris Lattner36d5e7d2007-04-23 16:04:05 +0000501 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +0000502 Stream.ReadAbbrevRecord();
503 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000504 }
505
506 // Read a record.
507 Record.clear();
508 const Type *ResultTy = 0;
509 switch (Stream.ReadRecord(Code, Record)) {
510 default: // Default behavior: unknown type.
511 ResultTy = 0;
512 break;
513 case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
514 // TYPE_CODE_NUMENTRY contains a count of the number of types in the
515 // type list. This allows us to reserve space.
516 if (Record.size() < 1)
517 return Error("Invalid TYPE_CODE_NUMENTRY record");
518 TypeList.reserve(Record[0]);
519 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000520 case bitc::TYPE_CODE_VOID: // VOID
Owen Anderson1d0be152009-08-13 21:58:54 +0000521 ResultTy = Type::getVoidTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000522 break;
523 case bitc::TYPE_CODE_FLOAT: // FLOAT
Owen Anderson1d0be152009-08-13 21:58:54 +0000524 ResultTy = Type::getFloatTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000525 break;
526 case bitc::TYPE_CODE_DOUBLE: // DOUBLE
Owen Anderson1d0be152009-08-13 21:58:54 +0000527 ResultTy = Type::getDoubleTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000528 break;
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000529 case bitc::TYPE_CODE_X86_FP80: // X86_FP80
Owen Anderson1d0be152009-08-13 21:58:54 +0000530 ResultTy = Type::getX86_FP80Ty(Context);
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000531 break;
532 case bitc::TYPE_CODE_FP128: // FP128
Owen Anderson1d0be152009-08-13 21:58:54 +0000533 ResultTy = Type::getFP128Ty(Context);
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000534 break;
535 case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
Owen Anderson1d0be152009-08-13 21:58:54 +0000536 ResultTy = Type::getPPC_FP128Ty(Context);
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000537 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000538 case bitc::TYPE_CODE_LABEL: // LABEL
Owen Anderson1d0be152009-08-13 21:58:54 +0000539 ResultTy = Type::getLabelTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000540 break;
541 case bitc::TYPE_CODE_OPAQUE: // OPAQUE
542 ResultTy = 0;
543 break;
Nick Lewycky7a0370f2009-05-30 05:06:04 +0000544 case bitc::TYPE_CODE_METADATA: // METADATA
Owen Anderson1d0be152009-08-13 21:58:54 +0000545 ResultTy = Type::getMetadataTy(Context);
Nick Lewycky7a0370f2009-05-30 05:06:04 +0000546 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000547 case bitc::TYPE_CODE_INTEGER: // INTEGER: [width]
548 if (Record.size() < 1)
549 return Error("Invalid Integer type record");
550
Owen Anderson1d0be152009-08-13 21:58:54 +0000551 ResultTy = IntegerType::get(Context, Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000552 break;
Christopher Lambfe63fb92007-12-11 08:59:05 +0000553 case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
554 // [pointee type, address space]
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000555 if (Record.size() < 1)
556 return Error("Invalid POINTER type record");
Christopher Lambfe63fb92007-12-11 08:59:05 +0000557 unsigned AddressSpace = 0;
558 if (Record.size() == 2)
559 AddressSpace = Record[1];
Owen Andersondebcb012009-07-29 22:17:13 +0000560 ResultTy = PointerType::get(getTypeByID(Record[0], true),
Owen Anderson74a77812009-07-07 20:18:58 +0000561 AddressSpace);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000562 break;
Christopher Lambfe63fb92007-12-11 08:59:05 +0000563 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000564 case bitc::TYPE_CODE_FUNCTION: {
Chris Lattnera1afde72007-11-27 17:48:06 +0000565 // FIXME: attrid is dead, remove it in LLVM 3.0
566 // FUNCTION: [vararg, attrid, retty, paramty x N]
567 if (Record.size() < 3)
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000568 return Error("Invalid FUNCTION type record");
569 std::vector<const Type*> ArgTys;
Chris Lattnera1afde72007-11-27 17:48:06 +0000570 for (unsigned i = 3, e = Record.size(); i != e; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000571 ArgTys.push_back(getTypeByID(Record[i], true));
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000572
Owen Andersondebcb012009-07-29 22:17:13 +0000573 ResultTy = FunctionType::get(getTypeByID(Record[2], true), ArgTys,
Duncan Sandsdc024672007-11-27 13:23:08 +0000574 Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000575 break;
576 }
Chris Lattner15e6d172007-05-04 19:11:41 +0000577 case bitc::TYPE_CODE_STRUCT: { // STRUCT: [ispacked, eltty x N]
Chris Lattner7108dce2007-05-06 08:21:50 +0000578 if (Record.size() < 1)
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000579 return Error("Invalid STRUCT type record");
580 std::vector<const Type*> EltTys;
Chris Lattner15e6d172007-05-04 19:11:41 +0000581 for (unsigned i = 1, e = Record.size(); i != e; ++i)
582 EltTys.push_back(getTypeByID(Record[i], true));
Owen Andersond7f2a6c2009-08-05 23:16:16 +0000583 ResultTy = StructType::get(Context, EltTys, Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000584 break;
585 }
586 case bitc::TYPE_CODE_ARRAY: // ARRAY: [numelts, eltty]
587 if (Record.size() < 2)
588 return Error("Invalid ARRAY type record");
Owen Andersondebcb012009-07-29 22:17:13 +0000589 ResultTy = ArrayType::get(getTypeByID(Record[1], true), Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000590 break;
591 case bitc::TYPE_CODE_VECTOR: // VECTOR: [numelts, eltty]
592 if (Record.size() < 2)
593 return Error("Invalid VECTOR type record");
Owen Andersondebcb012009-07-29 22:17:13 +0000594 ResultTy = VectorType::get(getTypeByID(Record[1], true), Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000595 break;
596 }
597
598 if (NumRecords == TypeList.size()) {
599 // If this is a new type slot, just append it.
Owen Andersondebcb012009-07-29 22:17:13 +0000600 TypeList.push_back(ResultTy ? ResultTy : OpaqueType::get());
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000601 ++NumRecords;
602 } else if (ResultTy == 0) {
603 // Otherwise, this was forward referenced, so an opaque type was created,
604 // but the result type is actually just an opaque. Leave the one we
605 // created previously.
606 ++NumRecords;
607 } else {
608 // Otherwise, this was forward referenced, so an opaque type was created.
609 // Resolve the opaque type to the real type now.
610 assert(NumRecords < TypeList.size() && "Typelist imbalance");
611 const OpaqueType *OldTy = cast<OpaqueType>(TypeList[NumRecords++].get());
612
613 // Don't directly push the new type on the Tab. Instead we want to replace
614 // the opaque type we previously inserted with the new concrete value. The
615 // refinement from the abstract (opaque) type to the new type causes all
616 // uses of the abstract type to use the concrete type (NewTy). This will
617 // also cause the opaque type to be deleted.
618 const_cast<OpaqueType*>(OldTy)->refineAbstractTypeTo(ResultTy);
619
620 // This should have replaced the old opaque type with the new type in the
Chris Lattner0eef0802007-04-24 04:04:35 +0000621 // value table... or with a preexisting type that was already in the
622 // system. Let's just make sure it did.
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000623 assert(TypeList[NumRecords-1].get() != OldTy &&
624 "refineAbstractType didn't work!");
625 }
626 }
627}
628
629
Chris Lattner86697142007-05-01 05:01:34 +0000630bool BitcodeReader::ParseTypeSymbolTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000631 if (Stream.EnterSubBlock(bitc::TYPE_SYMTAB_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000632 return Error("Malformed block record");
633
634 SmallVector<uint64_t, 64> Record;
635
636 // Read all the records for this type table.
637 std::string TypeName;
638 while (1) {
639 unsigned Code = Stream.ReadCode();
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000640 if (Code == bitc::END_BLOCK) {
641 if (Stream.ReadBlockEnd())
642 return Error("Error at end of type symbol table block");
643 return false;
644 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000645
646 if (Code == bitc::ENTER_SUBBLOCK) {
647 // No known subblocks, always skip them.
648 Stream.ReadSubBlockID();
649 if (Stream.SkipBlock())
650 return Error("Malformed block record");
651 continue;
652 }
653
Chris Lattner36d5e7d2007-04-23 16:04:05 +0000654 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +0000655 Stream.ReadAbbrevRecord();
656 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000657 }
658
659 // Read a record.
660 Record.clear();
661 switch (Stream.ReadRecord(Code, Record)) {
662 default: // Default behavior: unknown type.
663 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000664 case bitc::TST_CODE_ENTRY: // TST_ENTRY: [typeid, namechar x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000665 if (ConvertToString(Record, 1, TypeName))
666 return Error("Invalid TST_ENTRY record");
667 unsigned TypeID = Record[0];
668 if (TypeID >= TypeList.size())
669 return Error("Invalid Type ID in TST_ENTRY record");
670
671 TheModule->addTypeName(TypeName, TypeList[TypeID].get());
672 TypeName.clear();
673 break;
674 }
675 }
676}
677
Chris Lattner86697142007-05-01 05:01:34 +0000678bool BitcodeReader::ParseValueSymbolTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000679 if (Stream.EnterSubBlock(bitc::VALUE_SYMTAB_BLOCK_ID))
Chris Lattner0b2482a2007-04-23 21:26:05 +0000680 return Error("Malformed block record");
681
682 SmallVector<uint64_t, 64> Record;
683
684 // Read all the records for this value table.
685 SmallString<128> ValueName;
686 while (1) {
687 unsigned Code = Stream.ReadCode();
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000688 if (Code == bitc::END_BLOCK) {
689 if (Stream.ReadBlockEnd())
690 return Error("Error at end of value symbol table block");
691 return false;
692 }
Chris Lattner0b2482a2007-04-23 21:26:05 +0000693 if (Code == bitc::ENTER_SUBBLOCK) {
694 // No known subblocks, always skip them.
695 Stream.ReadSubBlockID();
696 if (Stream.SkipBlock())
697 return Error("Malformed block record");
698 continue;
699 }
700
701 if (Code == bitc::DEFINE_ABBREV) {
702 Stream.ReadAbbrevRecord();
703 continue;
704 }
705
706 // Read a record.
707 Record.clear();
708 switch (Stream.ReadRecord(Code, Record)) {
709 default: // Default behavior: unknown type.
710 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000711 case bitc::VST_CODE_ENTRY: { // VST_ENTRY: [valueid, namechar x N]
Chris Lattner0b2482a2007-04-23 21:26:05 +0000712 if (ConvertToString(Record, 1, ValueName))
Nick Lewycky88b72932009-05-31 06:07:28 +0000713 return Error("Invalid VST_ENTRY record");
Chris Lattner0b2482a2007-04-23 21:26:05 +0000714 unsigned ValueID = Record[0];
715 if (ValueID >= ValueList.size())
716 return Error("Invalid Value ID in VST_ENTRY record");
717 Value *V = ValueList[ValueID];
718
Daniel Dunbar3f53fa92009-07-26 00:34:27 +0000719 V->setName(StringRef(ValueName.data(), ValueName.size()));
Chris Lattner0b2482a2007-04-23 21:26:05 +0000720 ValueName.clear();
721 break;
Reid Spencerc8f8a242007-05-04 01:43:33 +0000722 }
723 case bitc::VST_CODE_BBENTRY: {
Chris Lattnere825ed52007-05-03 22:18:21 +0000724 if (ConvertToString(Record, 1, ValueName))
725 return Error("Invalid VST_BBENTRY record");
726 BasicBlock *BB = getBasicBlock(Record[0]);
727 if (BB == 0)
728 return Error("Invalid BB ID in VST_BBENTRY record");
729
Daniel Dunbar3f53fa92009-07-26 00:34:27 +0000730 BB->setName(StringRef(ValueName.data(), ValueName.size()));
Chris Lattnere825ed52007-05-03 22:18:21 +0000731 ValueName.clear();
732 break;
Chris Lattner0b2482a2007-04-23 21:26:05 +0000733 }
Reid Spencerc8f8a242007-05-04 01:43:33 +0000734 }
Chris Lattner0b2482a2007-04-23 21:26:05 +0000735 }
736}
737
Devang Patele54abc92009-07-22 17:43:22 +0000738bool BitcodeReader::ParseMetadata() {
Devang Pateld5ac4042009-08-04 06:00:18 +0000739 unsigned NextValueNo = MDValueList.size();
Devang Patele54abc92009-07-22 17:43:22 +0000740
741 if (Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
742 return Error("Malformed block record");
743
744 SmallVector<uint64_t, 64> Record;
745
746 // Read all the records.
747 while (1) {
748 unsigned Code = Stream.ReadCode();
749 if (Code == bitc::END_BLOCK) {
750 if (Stream.ReadBlockEnd())
751 return Error("Error at end of PARAMATTR block");
752 return false;
753 }
754
755 if (Code == bitc::ENTER_SUBBLOCK) {
756 // No known subblocks, always skip them.
757 Stream.ReadSubBlockID();
758 if (Stream.SkipBlock())
759 return Error("Malformed block record");
760 continue;
761 }
762
763 if (Code == bitc::DEFINE_ABBREV) {
764 Stream.ReadAbbrevRecord();
765 continue;
766 }
767
768 // Read a record.
769 Record.clear();
770 switch (Stream.ReadRecord(Code, Record)) {
771 default: // Default behavior: ignore.
772 break;
Devang Patelaa993142009-07-29 22:34:41 +0000773 case bitc::METADATA_NAME: {
774 // Read named of the named metadata.
775 unsigned NameLength = Record.size();
776 SmallString<8> Name;
777 Name.resize(NameLength);
778 for (unsigned i = 0; i != NameLength; ++i)
779 Name[i] = Record[i];
780 Record.clear();
781 Code = Stream.ReadCode();
782
783 // METADATA_NAME is always followed by METADATA_NAMED_NODE.
784 if (Stream.ReadRecord(Code, Record) != bitc::METADATA_NAMED_NODE)
785 assert ( 0 && "Inavlid Named Metadata record");
786
787 // Read named metadata elements.
788 unsigned Size = Record.size();
789 SmallVector<MetadataBase*, 8> Elts;
790 for (unsigned i = 0; i != Size; ++i) {
Devang Pateld5ac4042009-08-04 06:00:18 +0000791 Value *MD = MDValueList.getValueFwdRef(Record[i]);
Devang Patelaa993142009-07-29 22:34:41 +0000792 if (MetadataBase *B = dyn_cast<MetadataBase>(MD))
793 Elts.push_back(B);
794 }
Owen Anderson1d0be152009-08-13 21:58:54 +0000795 Value *V = NamedMDNode::Create(Context, Name.c_str(), Elts.data(),
796 Elts.size(), TheModule);
Devang Pateld5ac4042009-08-04 06:00:18 +0000797 MDValueList.AssignValue(V, NextValueNo++);
Devang Patelaa993142009-07-29 22:34:41 +0000798 break;
799 }
Devang Patel104cf9e2009-07-23 01:07:34 +0000800 case bitc::METADATA_NODE: {
801 if (Record.empty() || Record.size() % 2 == 1)
802 return Error("Invalid METADATA_NODE record");
803
804 unsigned Size = Record.size();
805 SmallVector<Value*, 8> Elts;
806 for (unsigned i = 0; i != Size; i += 2) {
807 const Type *Ty = getTypeByID(Record[i], false);
Owen Anderson1d0be152009-08-13 21:58:54 +0000808 if (Ty == Type::getMetadataTy(Context))
Devang Pateld5ac4042009-08-04 06:00:18 +0000809 Elts.push_back(MDValueList.getValueFwdRef(Record[i+1]));
Owen Anderson1d0be152009-08-13 21:58:54 +0000810 else if (Ty != Type::getVoidTy(Context))
Devang Patel104cf9e2009-07-23 01:07:34 +0000811 Elts.push_back(ValueList.getValueFwdRef(Record[i+1], Ty));
812 else
813 Elts.push_back(NULL);
814 }
Owen Anderson647e3012009-07-31 21:35:40 +0000815 Value *V = MDNode::get(Context, &Elts[0], Elts.size());
Devang Pateld5ac4042009-08-04 06:00:18 +0000816 MDValueList.AssignValue(V, NextValueNo++);
Devang Patel104cf9e2009-07-23 01:07:34 +0000817 break;
818 }
Devang Patele54abc92009-07-22 17:43:22 +0000819 case bitc::METADATA_STRING: {
820 unsigned MDStringLength = Record.size();
821 SmallString<8> String;
822 String.resize(MDStringLength);
823 for (unsigned i = 0; i != MDStringLength; ++i)
824 String[i] = Record[i];
Owen Anderson647e3012009-07-31 21:35:40 +0000825 Value *V = MDString::get(Context,
826 StringRef(String.data(), String.size()));
Devang Pateld5ac4042009-08-04 06:00:18 +0000827 MDValueList.AssignValue(V, NextValueNo++);
Devang Patele54abc92009-07-22 17:43:22 +0000828 break;
829 }
830 }
831 }
832}
833
Chris Lattner0eef0802007-04-24 04:04:35 +0000834/// DecodeSignRotatedValue - Decode a signed value stored with the sign bit in
835/// the LSB for dense VBR encoding.
836static uint64_t DecodeSignRotatedValue(uint64_t V) {
837 if ((V & 1) == 0)
838 return V >> 1;
839 if (V != 1)
840 return -(V >> 1);
841 // There is no such thing as -0 with integers. "-0" really means MININT.
842 return 1ULL << 63;
843}
844
Chris Lattner07d98b42007-04-26 02:46:40 +0000845/// ResolveGlobalAndAliasInits - Resolve all of the initializers for global
846/// values and aliases that we can.
847bool BitcodeReader::ResolveGlobalAndAliasInits() {
848 std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInitWorklist;
849 std::vector<std::pair<GlobalAlias*, unsigned> > AliasInitWorklist;
850
851 GlobalInitWorklist.swap(GlobalInits);
852 AliasInitWorklist.swap(AliasInits);
853
854 while (!GlobalInitWorklist.empty()) {
Chris Lattner198f34a2007-04-26 03:27:58 +0000855 unsigned ValID = GlobalInitWorklist.back().second;
Chris Lattner07d98b42007-04-26 02:46:40 +0000856 if (ValID >= ValueList.size()) {
857 // Not ready to resolve this yet, it requires something later in the file.
Chris Lattner198f34a2007-04-26 03:27:58 +0000858 GlobalInits.push_back(GlobalInitWorklist.back());
Chris Lattner07d98b42007-04-26 02:46:40 +0000859 } else {
860 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
861 GlobalInitWorklist.back().first->setInitializer(C);
862 else
863 return Error("Global variable initializer is not a constant!");
864 }
865 GlobalInitWorklist.pop_back();
866 }
867
868 while (!AliasInitWorklist.empty()) {
869 unsigned ValID = AliasInitWorklist.back().second;
870 if (ValID >= ValueList.size()) {
871 AliasInits.push_back(AliasInitWorklist.back());
872 } else {
873 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
Anton Korobeynikov7dde0ff2007-04-28 14:57:59 +0000874 AliasInitWorklist.back().first->setAliasee(C);
Chris Lattner07d98b42007-04-26 02:46:40 +0000875 else
876 return Error("Alias initializer is not a constant!");
877 }
878 AliasInitWorklist.pop_back();
879 }
880 return false;
881}
882
Dan Gohman1224c382009-07-20 21:19:07 +0000883static void SetOptimizationFlags(Value *V, uint64_t Flags) {
884 if (OverflowingBinaryOperator *OBO =
885 dyn_cast<OverflowingBinaryOperator>(V)) {
886 if (Flags & (1 << bitc::OBO_NO_SIGNED_OVERFLOW))
887 OBO->setHasNoSignedOverflow(true);
888 if (Flags & (1 << bitc::OBO_NO_UNSIGNED_OVERFLOW))
889 OBO->setHasNoUnsignedOverflow(true);
890 } else if (SDivOperator *Div = dyn_cast<SDivOperator>(V)) {
891 if (Flags & (1 << bitc::SDIV_EXACT))
892 Div->setIsExact(true);
893 }
894}
Chris Lattner07d98b42007-04-26 02:46:40 +0000895
Chris Lattner86697142007-05-01 05:01:34 +0000896bool BitcodeReader::ParseConstants() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000897 if (Stream.EnterSubBlock(bitc::CONSTANTS_BLOCK_ID))
Chris Lattnere16504e2007-04-24 03:30:34 +0000898 return Error("Malformed block record");
899
900 SmallVector<uint64_t, 64> Record;
901
902 // Read all the records for this value table.
Owen Anderson1d0be152009-08-13 21:58:54 +0000903 const Type *CurTy = Type::getInt32Ty(Context);
Chris Lattner522b7b12007-04-24 05:48:56 +0000904 unsigned NextCstNo = ValueList.size();
Chris Lattnere16504e2007-04-24 03:30:34 +0000905 while (1) {
906 unsigned Code = Stream.ReadCode();
Chris Lattnerea693df2008-08-21 02:34:16 +0000907 if (Code == bitc::END_BLOCK)
908 break;
Chris Lattnere16504e2007-04-24 03:30:34 +0000909
910 if (Code == bitc::ENTER_SUBBLOCK) {
911 // No known subblocks, always skip them.
912 Stream.ReadSubBlockID();
913 if (Stream.SkipBlock())
914 return Error("Malformed block record");
915 continue;
916 }
917
918 if (Code == bitc::DEFINE_ABBREV) {
919 Stream.ReadAbbrevRecord();
920 continue;
921 }
922
923 // Read a record.
924 Record.clear();
925 Value *V = 0;
Dan Gohman1224c382009-07-20 21:19:07 +0000926 unsigned BitCode = Stream.ReadRecord(Code, Record);
927 switch (BitCode) {
Chris Lattnere16504e2007-04-24 03:30:34 +0000928 default: // Default behavior: unknown constant
929 case bitc::CST_CODE_UNDEF: // UNDEF
Owen Anderson9e9a0d52009-07-30 23:03:37 +0000930 V = UndefValue::get(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +0000931 break;
932 case bitc::CST_CODE_SETTYPE: // SETTYPE: [typeid]
933 if (Record.empty())
934 return Error("Malformed CST_SETTYPE record");
935 if (Record[0] >= TypeList.size())
936 return Error("Invalid Type ID in CST_SETTYPE record");
937 CurTy = TypeList[Record[0]];
Chris Lattner0eef0802007-04-24 04:04:35 +0000938 continue; // Skip the ValueList manipulation.
Chris Lattnere16504e2007-04-24 03:30:34 +0000939 case bitc::CST_CODE_NULL: // NULL
Owen Andersona7235ea2009-07-31 20:28:14 +0000940 V = Constant::getNullValue(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +0000941 break;
942 case bitc::CST_CODE_INTEGER: // INTEGER: [intval]
Chris Lattner0eef0802007-04-24 04:04:35 +0000943 if (!isa<IntegerType>(CurTy) || Record.empty())
944 return Error("Invalid CST_INTEGER record");
Owen Andersoneed707b2009-07-24 23:12:02 +0000945 V = ConstantInt::get(CurTy, DecodeSignRotatedValue(Record[0]));
Chris Lattner0eef0802007-04-24 04:04:35 +0000946 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000947 case bitc::CST_CODE_WIDE_INTEGER: {// WIDE_INTEGER: [n x intval]
948 if (!isa<IntegerType>(CurTy) || Record.empty())
Chris Lattner0eef0802007-04-24 04:04:35 +0000949 return Error("Invalid WIDE_INTEGER record");
950
Chris Lattner15e6d172007-05-04 19:11:41 +0000951 unsigned NumWords = Record.size();
Chris Lattner084a8442007-04-24 17:22:05 +0000952 SmallVector<uint64_t, 8> Words;
953 Words.resize(NumWords);
Chris Lattner0eef0802007-04-24 04:04:35 +0000954 for (unsigned i = 0; i != NumWords; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000955 Words[i] = DecodeSignRotatedValue(Record[i]);
Owen Andersoneed707b2009-07-24 23:12:02 +0000956 V = ConstantInt::get(Context,
957 APInt(cast<IntegerType>(CurTy)->getBitWidth(),
958 NumWords, &Words[0]));
Chris Lattner0eef0802007-04-24 04:04:35 +0000959 break;
960 }
Dale Johannesen3f6eb742007-09-11 18:32:33 +0000961 case bitc::CST_CODE_FLOAT: { // FLOAT: [fpval]
Chris Lattner0eef0802007-04-24 04:04:35 +0000962 if (Record.empty())
963 return Error("Invalid FLOAT record");
Owen Anderson1d0be152009-08-13 21:58:54 +0000964 if (CurTy == Type::getFloatTy(Context))
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000965 V = ConstantFP::get(Context, APFloat(APInt(32, (uint32_t)Record[0])));
Owen Anderson1d0be152009-08-13 21:58:54 +0000966 else if (CurTy == Type::getDoubleTy(Context))
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000967 V = ConstantFP::get(Context, APFloat(APInt(64, Record[0])));
Owen Anderson1d0be152009-08-13 21:58:54 +0000968 else if (CurTy == Type::getX86_FP80Ty(Context)) {
Dale Johannesen1b25cb22009-03-23 21:16:53 +0000969 // Bits are not stored the same way as a normal i80 APInt, compensate.
970 uint64_t Rearrange[2];
971 Rearrange[0] = (Record[1] & 0xffffLL) | (Record[0] << 16);
972 Rearrange[1] = Record[0] >> 48;
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000973 V = ConstantFP::get(Context, APFloat(APInt(80, 2, Rearrange)));
Owen Anderson1d0be152009-08-13 21:58:54 +0000974 } else if (CurTy == Type::getFP128Ty(Context))
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000975 V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0]), true));
Owen Anderson1d0be152009-08-13 21:58:54 +0000976 else if (CurTy == Type::getPPC_FP128Ty(Context))
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000977 V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0])));
Chris Lattnere16504e2007-04-24 03:30:34 +0000978 else
Owen Anderson9e9a0d52009-07-30 23:03:37 +0000979 V = UndefValue::get(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +0000980 break;
Dale Johannesen3f6eb742007-09-11 18:32:33 +0000981 }
Chris Lattner522b7b12007-04-24 05:48:56 +0000982
Chris Lattner15e6d172007-05-04 19:11:41 +0000983 case bitc::CST_CODE_AGGREGATE: {// AGGREGATE: [n x value number]
984 if (Record.empty())
Chris Lattner522b7b12007-04-24 05:48:56 +0000985 return Error("Invalid CST_AGGREGATE record");
986
Chris Lattner15e6d172007-05-04 19:11:41 +0000987 unsigned Size = Record.size();
Chris Lattner522b7b12007-04-24 05:48:56 +0000988 std::vector<Constant*> Elts;
989
990 if (const StructType *STy = dyn_cast<StructType>(CurTy)) {
991 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000992 Elts.push_back(ValueList.getConstantFwdRef(Record[i],
Chris Lattner522b7b12007-04-24 05:48:56 +0000993 STy->getElementType(i)));
Owen Anderson8fa33382009-07-27 22:29:26 +0000994 V = ConstantStruct::get(STy, Elts);
Chris Lattner522b7b12007-04-24 05:48:56 +0000995 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(CurTy)) {
996 const Type *EltTy = ATy->getElementType();
997 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000998 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Owen Anderson1fd70962009-07-28 18:32:17 +0000999 V = ConstantArray::get(ATy, Elts);
Chris Lattner522b7b12007-04-24 05:48:56 +00001000 } else if (const VectorType *VTy = dyn_cast<VectorType>(CurTy)) {
1001 const Type *EltTy = VTy->getElementType();
1002 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +00001003 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Owen Andersonaf7ec972009-07-28 21:19:26 +00001004 V = ConstantVector::get(Elts);
Chris Lattner522b7b12007-04-24 05:48:56 +00001005 } else {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001006 V = UndefValue::get(CurTy);
Chris Lattner522b7b12007-04-24 05:48:56 +00001007 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001008 break;
1009 }
Chris Lattnerff7fc5d2007-05-06 00:35:24 +00001010 case bitc::CST_CODE_STRING: { // STRING: [values]
1011 if (Record.empty())
1012 return Error("Invalid CST_AGGREGATE record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001013
Chris Lattnerff7fc5d2007-05-06 00:35:24 +00001014 const ArrayType *ATy = cast<ArrayType>(CurTy);
1015 const Type *EltTy = ATy->getElementType();
1016
1017 unsigned Size = Record.size();
1018 std::vector<Constant*> Elts;
Chris Lattnerff7fc5d2007-05-06 00:35:24 +00001019 for (unsigned i = 0; i != Size; ++i)
Owen Andersoneed707b2009-07-24 23:12:02 +00001020 Elts.push_back(ConstantInt::get(EltTy, Record[i]));
Owen Anderson1fd70962009-07-28 18:32:17 +00001021 V = ConstantArray::get(ATy, Elts);
Chris Lattnerff7fc5d2007-05-06 00:35:24 +00001022 break;
1023 }
Chris Lattnercb3d91b2007-05-06 00:53:07 +00001024 case bitc::CST_CODE_CSTRING: { // CSTRING: [values]
1025 if (Record.empty())
1026 return Error("Invalid CST_AGGREGATE record");
1027
1028 const ArrayType *ATy = cast<ArrayType>(CurTy);
1029 const Type *EltTy = ATy->getElementType();
1030
1031 unsigned Size = Record.size();
1032 std::vector<Constant*> Elts;
1033 for (unsigned i = 0; i != Size; ++i)
Owen Andersoneed707b2009-07-24 23:12:02 +00001034 Elts.push_back(ConstantInt::get(EltTy, Record[i]));
Owen Andersona7235ea2009-07-31 20:28:14 +00001035 Elts.push_back(Constant::getNullValue(EltTy));
Owen Anderson1fd70962009-07-28 18:32:17 +00001036 V = ConstantArray::get(ATy, Elts);
Chris Lattnercb3d91b2007-05-06 00:53:07 +00001037 break;
1038 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001039 case bitc::CST_CODE_CE_BINOP: { // CE_BINOP: [opcode, opval, opval]
1040 if (Record.size() < 3) return Error("Invalid CE_BINOP record");
1041 int Opc = GetDecodedBinaryOpcode(Record[0], CurTy);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001042 if (Opc < 0) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001043 V = UndefValue::get(CurTy); // Unknown binop.
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001044 } else {
1045 Constant *LHS = ValueList.getConstantFwdRef(Record[1], CurTy);
1046 Constant *RHS = ValueList.getConstantFwdRef(Record[2], CurTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001047 V = ConstantExpr::get(Opc, LHS, RHS);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001048 }
Dan Gohman1224c382009-07-20 21:19:07 +00001049 if (Record.size() >= 4)
1050 SetOptimizationFlags(V, Record[3]);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001051 break;
1052 }
1053 case bitc::CST_CODE_CE_CAST: { // CE_CAST: [opcode, opty, opval]
1054 if (Record.size() < 3) return Error("Invalid CE_CAST record");
1055 int Opc = GetDecodedCastOpcode(Record[0]);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001056 if (Opc < 0) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001057 V = UndefValue::get(CurTy); // Unknown cast.
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001058 } else {
1059 const Type *OpTy = getTypeByID(Record[1]);
Chris Lattnerbfcc3802007-05-06 07:33:01 +00001060 if (!OpTy) return Error("Invalid CE_CAST record");
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001061 Constant *Op = ValueList.getConstantFwdRef(Record[2], OpTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001062 V = ConstantExpr::getCast(Opc, Op, CurTy);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001063 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001064 break;
1065 }
Dan Gohmandd8004d2009-07-27 21:53:46 +00001066 case bitc::CST_CODE_CE_INBOUNDS_GEP:
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001067 case bitc::CST_CODE_CE_GEP: { // CE_GEP: [n x operands]
Chris Lattner15e6d172007-05-04 19:11:41 +00001068 if (Record.size() & 1) return Error("Invalid CE_GEP record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001069 SmallVector<Constant*, 16> Elts;
Chris Lattner15e6d172007-05-04 19:11:41 +00001070 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001071 const Type *ElTy = getTypeByID(Record[i]);
1072 if (!ElTy) return Error("Invalid CE_GEP record");
1073 Elts.push_back(ValueList.getConstantFwdRef(Record[i+1], ElTy));
1074 }
Owen Andersonbaf3c402009-07-29 18:55:55 +00001075 V = ConstantExpr::getGetElementPtr(Elts[0], &Elts[1],
Owen Anderson74a77812009-07-07 20:18:58 +00001076 Elts.size()-1);
Dan Gohmandd8004d2009-07-27 21:53:46 +00001077 if (BitCode == bitc::CST_CODE_CE_INBOUNDS_GEP)
1078 cast<GEPOperator>(V)->setIsInBounds(true);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001079 break;
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001080 }
1081 case bitc::CST_CODE_CE_SELECT: // CE_SELECT: [opval#, opval#, opval#]
1082 if (Record.size() < 3) return Error("Invalid CE_SELECT record");
Owen Andersonbaf3c402009-07-29 18:55:55 +00001083 V = ConstantExpr::getSelect(ValueList.getConstantFwdRef(Record[0],
Owen Anderson1d0be152009-08-13 21:58:54 +00001084 Type::getInt1Ty(Context)),
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001085 ValueList.getConstantFwdRef(Record[1],CurTy),
1086 ValueList.getConstantFwdRef(Record[2],CurTy));
1087 break;
1088 case bitc::CST_CODE_CE_EXTRACTELT: { // CE_EXTRACTELT: [opty, opval, opval]
1089 if (Record.size() < 3) return Error("Invalid CE_EXTRACTELT record");
1090 const VectorType *OpTy =
1091 dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
1092 if (OpTy == 0) return Error("Invalid CE_EXTRACTELT record");
1093 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
Owen Anderson1d0be152009-08-13 21:58:54 +00001094 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
Owen Andersonbaf3c402009-07-29 18:55:55 +00001095 V = ConstantExpr::getExtractElement(Op0, Op1);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001096 break;
1097 }
1098 case bitc::CST_CODE_CE_INSERTELT: { // CE_INSERTELT: [opval, opval, opval]
1099 const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1100 if (Record.size() < 3 || OpTy == 0)
1101 return Error("Invalid CE_INSERTELT record");
1102 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1103 Constant *Op1 = ValueList.getConstantFwdRef(Record[1],
1104 OpTy->getElementType());
Owen Anderson1d0be152009-08-13 21:58:54 +00001105 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
Owen Andersonbaf3c402009-07-29 18:55:55 +00001106 V = ConstantExpr::getInsertElement(Op0, Op1, Op2);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001107 break;
1108 }
1109 case bitc::CST_CODE_CE_SHUFFLEVEC: { // CE_SHUFFLEVEC: [opval, opval, opval]
1110 const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1111 if (Record.size() < 3 || OpTy == 0)
Nate Begeman0f123cf2009-02-12 21:28:33 +00001112 return Error("Invalid CE_SHUFFLEVEC record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001113 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1114 Constant *Op1 = ValueList.getConstantFwdRef(Record[1], OpTy);
Owen Anderson1d0be152009-08-13 21:58:54 +00001115 const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
Owen Anderson74a77812009-07-07 20:18:58 +00001116 OpTy->getNumElements());
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001117 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], ShufTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001118 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001119 break;
1120 }
Nate Begeman0f123cf2009-02-12 21:28:33 +00001121 case bitc::CST_CODE_CE_SHUFVEC_EX: { // [opty, opval, opval, opval]
1122 const VectorType *RTy = dyn_cast<VectorType>(CurTy);
1123 const VectorType *OpTy = dyn_cast<VectorType>(getTypeByID(Record[0]));
1124 if (Record.size() < 4 || RTy == 0 || OpTy == 0)
1125 return Error("Invalid CE_SHUFVEC_EX record");
1126 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1127 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
Owen Anderson1d0be152009-08-13 21:58:54 +00001128 const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
Owen Anderson74a77812009-07-07 20:18:58 +00001129 RTy->getNumElements());
Nate Begeman0f123cf2009-02-12 21:28:33 +00001130 Constant *Op2 = ValueList.getConstantFwdRef(Record[3], ShufTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001131 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
Nate Begeman0f123cf2009-02-12 21:28:33 +00001132 break;
1133 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001134 case bitc::CST_CODE_CE_CMP: { // CE_CMP: [opty, opval, opval, pred]
1135 if (Record.size() < 4) return Error("Invalid CE_CMP record");
1136 const Type *OpTy = getTypeByID(Record[0]);
1137 if (OpTy == 0) return Error("Invalid CE_CMP record");
1138 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1139 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1140
1141 if (OpTy->isFloatingPoint())
Owen Andersonbaf3c402009-07-29 18:55:55 +00001142 V = ConstantExpr::getFCmp(Record[3], Op0, Op1);
Nate Begemanac80ade2008-05-12 19:01:56 +00001143 else
Owen Andersonbaf3c402009-07-29 18:55:55 +00001144 V = ConstantExpr::getICmp(Record[3], Op0, Op1);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001145 break;
Chris Lattner522b7b12007-04-24 05:48:56 +00001146 }
Chris Lattner2bce93a2007-05-06 01:58:20 +00001147 case bitc::CST_CODE_INLINEASM: {
1148 if (Record.size() < 2) return Error("Invalid INLINEASM record");
1149 std::string AsmStr, ConstrStr;
1150 bool HasSideEffects = Record[0];
1151 unsigned AsmStrSize = Record[1];
1152 if (2+AsmStrSize >= Record.size())
1153 return Error("Invalid INLINEASM record");
1154 unsigned ConstStrSize = Record[2+AsmStrSize];
1155 if (3+AsmStrSize+ConstStrSize > Record.size())
1156 return Error("Invalid INLINEASM record");
1157
1158 for (unsigned i = 0; i != AsmStrSize; ++i)
1159 AsmStr += (char)Record[2+i];
1160 for (unsigned i = 0; i != ConstStrSize; ++i)
1161 ConstrStr += (char)Record[3+AsmStrSize+i];
1162 const PointerType *PTy = cast<PointerType>(CurTy);
1163 V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
1164 AsmStr, ConstrStr, HasSideEffects);
1165 break;
1166 }
Chris Lattnere16504e2007-04-24 03:30:34 +00001167 }
1168
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001169 ValueList.AssignValue(V, NextCstNo);
Chris Lattner522b7b12007-04-24 05:48:56 +00001170 ++NextCstNo;
Chris Lattnere16504e2007-04-24 03:30:34 +00001171 }
Chris Lattnerea693df2008-08-21 02:34:16 +00001172
1173 if (NextCstNo != ValueList.size())
1174 return Error("Invalid constant reference!");
1175
1176 if (Stream.ReadBlockEnd())
1177 return Error("Error at end of constants block");
1178
1179 // Once all the constants have been read, go through and resolve forward
1180 // references.
1181 ValueList.ResolveConstantForwardRefs();
1182 return false;
Chris Lattnere16504e2007-04-24 03:30:34 +00001183}
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001184
Chris Lattner980e5aa2007-05-01 05:52:21 +00001185/// RememberAndSkipFunctionBody - When we see the block for a function body,
1186/// remember where it is and then skip it. This lets us lazily deserialize the
1187/// functions.
1188bool BitcodeReader::RememberAndSkipFunctionBody() {
Chris Lattner48f84872007-05-01 04:59:48 +00001189 // Get the function we are talking about.
1190 if (FunctionsWithBodies.empty())
1191 return Error("Insufficient function protos");
1192
1193 Function *Fn = FunctionsWithBodies.back();
1194 FunctionsWithBodies.pop_back();
1195
1196 // Save the current stream state.
1197 uint64_t CurBit = Stream.GetCurrentBitNo();
1198 DeferredFunctionInfo[Fn] = std::make_pair(CurBit, Fn->getLinkage());
1199
1200 // Set the functions linkage to GhostLinkage so we know it is lazily
1201 // deserialized.
1202 Fn->setLinkage(GlobalValue::GhostLinkage);
1203
1204 // Skip over the function block for now.
1205 if (Stream.SkipBlock())
1206 return Error("Malformed block record");
1207 return false;
1208}
1209
Chris Lattner86697142007-05-01 05:01:34 +00001210bool BitcodeReader::ParseModule(const std::string &ModuleID) {
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001211 // Reject multiple MODULE_BLOCK's in a single bitstream.
1212 if (TheModule)
1213 return Error("Multiple MODULE_BLOCKs in same stream");
1214
Chris Lattnere17b6582007-05-05 00:17:00 +00001215 if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001216 return Error("Malformed block record");
1217
1218 // Otherwise, create the module.
Owen Anderson8b477ed2009-07-01 16:58:40 +00001219 TheModule = new Module(ModuleID, Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001220
1221 SmallVector<uint64_t, 64> Record;
1222 std::vector<std::string> SectionTable;
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001223 std::vector<std::string> GCTable;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001224
1225 // Read all the records for this module.
1226 while (!Stream.AtEndOfStream()) {
1227 unsigned Code = Stream.ReadCode();
Chris Lattnere84bcb92007-04-24 00:21:45 +00001228 if (Code == bitc::END_BLOCK) {
Chris Lattner980e5aa2007-05-01 05:52:21 +00001229 if (Stream.ReadBlockEnd())
1230 return Error("Error at end of module block");
1231
1232 // Patch the initializers for globals and aliases up.
Chris Lattner07d98b42007-04-26 02:46:40 +00001233 ResolveGlobalAndAliasInits();
1234 if (!GlobalInits.empty() || !AliasInits.empty())
Chris Lattnere84bcb92007-04-24 00:21:45 +00001235 return Error("Malformed global initializer set");
Chris Lattner48f84872007-05-01 04:59:48 +00001236 if (!FunctionsWithBodies.empty())
1237 return Error("Too few function bodies found");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001238
Chandler Carruth69940402007-08-04 01:51:18 +00001239 // Look for intrinsic functions which need to be upgraded at some point
1240 for (Module::iterator FI = TheModule->begin(), FE = TheModule->end();
1241 FI != FE; ++FI) {
Evan Chengf9b83fc2007-12-17 22:33:23 +00001242 Function* NewFn;
1243 if (UpgradeIntrinsicFunction(FI, NewFn))
Chandler Carruth69940402007-08-04 01:51:18 +00001244 UpgradedIntrinsics.push_back(std::make_pair(FI, NewFn));
1245 }
1246
Chris Lattner980e5aa2007-05-01 05:52:21 +00001247 // Force deallocation of memory for these vectors to favor the client that
1248 // want lazy deserialization.
1249 std::vector<std::pair<GlobalVariable*, unsigned> >().swap(GlobalInits);
1250 std::vector<std::pair<GlobalAlias*, unsigned> >().swap(AliasInits);
1251 std::vector<Function*>().swap(FunctionsWithBodies);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001252 return false;
Chris Lattnere84bcb92007-04-24 00:21:45 +00001253 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001254
1255 if (Code == bitc::ENTER_SUBBLOCK) {
1256 switch (Stream.ReadSubBlockID()) {
1257 default: // Skip unknown content.
1258 if (Stream.SkipBlock())
1259 return Error("Malformed block record");
1260 break;
Chris Lattner3f799802007-05-05 18:57:30 +00001261 case bitc::BLOCKINFO_BLOCK_ID:
1262 if (Stream.ReadBlockInfoBlock())
1263 return Error("Malformed BlockInfoBlock");
1264 break;
Chris Lattner48c85b82007-05-04 03:30:17 +00001265 case bitc::PARAMATTR_BLOCK_ID:
Devang Patel05988662008-09-25 21:00:45 +00001266 if (ParseAttributeBlock())
Chris Lattner48c85b82007-05-04 03:30:17 +00001267 return true;
1268 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001269 case bitc::TYPE_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001270 if (ParseTypeTable())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001271 return true;
1272 break;
1273 case bitc::TYPE_SYMTAB_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001274 if (ParseTypeSymbolTable())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001275 return true;
1276 break;
Chris Lattner0b2482a2007-04-23 21:26:05 +00001277 case bitc::VALUE_SYMTAB_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001278 if (ParseValueSymbolTable())
Chris Lattner0b2482a2007-04-23 21:26:05 +00001279 return true;
1280 break;
Chris Lattnere16504e2007-04-24 03:30:34 +00001281 case bitc::CONSTANTS_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001282 if (ParseConstants() || ResolveGlobalAndAliasInits())
Chris Lattnere16504e2007-04-24 03:30:34 +00001283 return true;
1284 break;
Devang Patele54abc92009-07-22 17:43:22 +00001285 case bitc::METADATA_BLOCK_ID:
1286 if (ParseMetadata())
1287 return true;
1288 break;
Chris Lattner48f84872007-05-01 04:59:48 +00001289 case bitc::FUNCTION_BLOCK_ID:
1290 // If this is the first function body we've seen, reverse the
1291 // FunctionsWithBodies list.
1292 if (!HasReversedFunctionsWithBodies) {
1293 std::reverse(FunctionsWithBodies.begin(), FunctionsWithBodies.end());
1294 HasReversedFunctionsWithBodies = true;
1295 }
1296
Chris Lattner980e5aa2007-05-01 05:52:21 +00001297 if (RememberAndSkipFunctionBody())
Chris Lattner48f84872007-05-01 04:59:48 +00001298 return true;
1299 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001300 }
1301 continue;
1302 }
1303
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001304 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +00001305 Stream.ReadAbbrevRecord();
1306 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001307 }
1308
1309 // Read a record.
1310 switch (Stream.ReadRecord(Code, Record)) {
1311 default: break; // Default behavior, ignore unknown content.
1312 case bitc::MODULE_CODE_VERSION: // VERSION: [version#]
1313 if (Record.size() < 1)
1314 return Error("Malformed MODULE_CODE_VERSION");
1315 // Only version #0 is supported so far.
1316 if (Record[0] != 0)
1317 return Error("Unknown bitstream version!");
1318 break;
Chris Lattner15e6d172007-05-04 19:11:41 +00001319 case bitc::MODULE_CODE_TRIPLE: { // TRIPLE: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001320 std::string S;
1321 if (ConvertToString(Record, 0, S))
1322 return Error("Invalid MODULE_CODE_TRIPLE record");
1323 TheModule->setTargetTriple(S);
1324 break;
1325 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001326 case bitc::MODULE_CODE_DATALAYOUT: { // DATALAYOUT: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001327 std::string S;
1328 if (ConvertToString(Record, 0, S))
1329 return Error("Invalid MODULE_CODE_DATALAYOUT record");
1330 TheModule->setDataLayout(S);
1331 break;
1332 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001333 case bitc::MODULE_CODE_ASM: { // ASM: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001334 std::string S;
1335 if (ConvertToString(Record, 0, S))
1336 return Error("Invalid MODULE_CODE_ASM record");
1337 TheModule->setModuleInlineAsm(S);
1338 break;
1339 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001340 case bitc::MODULE_CODE_DEPLIB: { // DEPLIB: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001341 std::string S;
1342 if (ConvertToString(Record, 0, S))
1343 return Error("Invalid MODULE_CODE_DEPLIB record");
1344 TheModule->addLibrary(S);
1345 break;
1346 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001347 case bitc::MODULE_CODE_SECTIONNAME: { // SECTIONNAME: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001348 std::string S;
1349 if (ConvertToString(Record, 0, S))
1350 return Error("Invalid MODULE_CODE_SECTIONNAME record");
1351 SectionTable.push_back(S);
1352 break;
1353 }
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001354 case bitc::MODULE_CODE_GCNAME: { // SECTIONNAME: [strchr x N]
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001355 std::string S;
1356 if (ConvertToString(Record, 0, S))
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001357 return Error("Invalid MODULE_CODE_GCNAME record");
1358 GCTable.push_back(S);
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001359 break;
1360 }
Christopher Lambfe63fb92007-12-11 08:59:05 +00001361 // GLOBALVAR: [pointer type, isconst, initid,
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001362 // linkage, alignment, section, visibility, threadlocal]
1363 case bitc::MODULE_CODE_GLOBALVAR: {
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001364 if (Record.size() < 6)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001365 return Error("Invalid MODULE_CODE_GLOBALVAR record");
1366 const Type *Ty = getTypeByID(Record[0]);
1367 if (!isa<PointerType>(Ty))
1368 return Error("Global not a pointer type!");
Christopher Lambfe63fb92007-12-11 08:59:05 +00001369 unsigned AddressSpace = cast<PointerType>(Ty)->getAddressSpace();
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001370 Ty = cast<PointerType>(Ty)->getElementType();
1371
1372 bool isConstant = Record[1];
1373 GlobalValue::LinkageTypes Linkage = GetDecodedLinkage(Record[3]);
1374 unsigned Alignment = (1 << Record[4]) >> 1;
1375 std::string Section;
1376 if (Record[5]) {
1377 if (Record[5]-1 >= SectionTable.size())
1378 return Error("Invalid section ID");
1379 Section = SectionTable[Record[5]-1];
1380 }
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001381 GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
Chris Lattner5f32c012007-05-06 19:27:46 +00001382 if (Record.size() > 6)
1383 Visibility = GetDecodedVisibility(Record[6]);
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001384 bool isThreadLocal = false;
Chris Lattner5f32c012007-05-06 19:27:46 +00001385 if (Record.size() > 7)
1386 isThreadLocal = Record[7];
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001387
1388 GlobalVariable *NewGV =
Owen Andersone9b11b42009-07-08 19:03:57 +00001389 new GlobalVariable(*TheModule, Ty, isConstant, Linkage, 0, "", 0,
Christopher Lambfe63fb92007-12-11 08:59:05 +00001390 isThreadLocal, AddressSpace);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001391 NewGV->setAlignment(Alignment);
1392 if (!Section.empty())
1393 NewGV->setSection(Section);
1394 NewGV->setVisibility(Visibility);
1395 NewGV->setThreadLocal(isThreadLocal);
1396
Chris Lattner0b2482a2007-04-23 21:26:05 +00001397 ValueList.push_back(NewGV);
1398
Chris Lattner6dbfd7b2007-04-24 00:18:21 +00001399 // Remember which value to use for the global initializer.
1400 if (unsigned InitID = Record[2])
1401 GlobalInits.push_back(std::make_pair(NewGV, InitID-1));
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001402 break;
1403 }
Chris Lattnera9bb7132007-05-08 05:38:01 +00001404 // FUNCTION: [type, callingconv, isproto, linkage, paramattr,
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001405 // alignment, section, visibility, gc]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001406 case bitc::MODULE_CODE_FUNCTION: {
Chris Lattnera9bb7132007-05-08 05:38:01 +00001407 if (Record.size() < 8)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001408 return Error("Invalid MODULE_CODE_FUNCTION record");
1409 const Type *Ty = getTypeByID(Record[0]);
1410 if (!isa<PointerType>(Ty))
1411 return Error("Function not a pointer type!");
1412 const FunctionType *FTy =
1413 dyn_cast<FunctionType>(cast<PointerType>(Ty)->getElementType());
1414 if (!FTy)
1415 return Error("Function not a pointer to function type!");
1416
Gabor Greif051a9502008-04-06 20:25:17 +00001417 Function *Func = Function::Create(FTy, GlobalValue::ExternalLinkage,
1418 "", TheModule);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001419
1420 Func->setCallingConv(Record[1]);
Chris Lattner48f84872007-05-01 04:59:48 +00001421 bool isProto = Record[2];
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001422 Func->setLinkage(GetDecodedLinkage(Record[3]));
Devang Patel05988662008-09-25 21:00:45 +00001423 Func->setAttributes(getAttributes(Record[4]));
Chris Lattnera9bb7132007-05-08 05:38:01 +00001424
1425 Func->setAlignment((1 << Record[5]) >> 1);
1426 if (Record[6]) {
1427 if (Record[6]-1 >= SectionTable.size())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001428 return Error("Invalid section ID");
Chris Lattnera9bb7132007-05-08 05:38:01 +00001429 Func->setSection(SectionTable[Record[6]-1]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001430 }
Chris Lattnera9bb7132007-05-08 05:38:01 +00001431 Func->setVisibility(GetDecodedVisibility(Record[7]));
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001432 if (Record.size() > 8 && Record[8]) {
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001433 if (Record[8]-1 > GCTable.size())
1434 return Error("Invalid GC ID");
1435 Func->setGC(GCTable[Record[8]-1].c_str());
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001436 }
Chris Lattner0b2482a2007-04-23 21:26:05 +00001437 ValueList.push_back(Func);
Chris Lattner48f84872007-05-01 04:59:48 +00001438
1439 // If this is a function with a body, remember the prototype we are
1440 // creating now, so that we can match up the body with them later.
1441 if (!isProto)
1442 FunctionsWithBodies.push_back(Func);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001443 break;
1444 }
Anton Korobeynikov91342d82008-03-12 00:49:19 +00001445 // ALIAS: [alias type, aliasee val#, linkage]
Anton Korobeynikovf8342b92008-03-11 21:40:17 +00001446 // ALIAS: [alias type, aliasee val#, linkage, visibility]
Chris Lattner198f34a2007-04-26 03:27:58 +00001447 case bitc::MODULE_CODE_ALIAS: {
Chris Lattner07d98b42007-04-26 02:46:40 +00001448 if (Record.size() < 3)
1449 return Error("Invalid MODULE_ALIAS record");
1450 const Type *Ty = getTypeByID(Record[0]);
1451 if (!isa<PointerType>(Ty))
1452 return Error("Function not a pointer type!");
1453
1454 GlobalAlias *NewGA = new GlobalAlias(Ty, GetDecodedLinkage(Record[2]),
1455 "", 0, TheModule);
Anton Korobeynikov91342d82008-03-12 00:49:19 +00001456 // Old bitcode files didn't have visibility field.
1457 if (Record.size() > 3)
1458 NewGA->setVisibility(GetDecodedVisibility(Record[3]));
Chris Lattner07d98b42007-04-26 02:46:40 +00001459 ValueList.push_back(NewGA);
1460 AliasInits.push_back(std::make_pair(NewGA, Record[1]));
1461 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001462 }
Chris Lattner198f34a2007-04-26 03:27:58 +00001463 /// MODULE_CODE_PURGEVALS: [numvals]
1464 case bitc::MODULE_CODE_PURGEVALS:
1465 // Trim down the value list to the specified size.
1466 if (Record.size() < 1 || Record[0] > ValueList.size())
1467 return Error("Invalid MODULE_PURGEVALS record");
1468 ValueList.shrinkTo(Record[0]);
1469 break;
1470 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001471 Record.clear();
1472 }
1473
1474 return Error("Premature end of bitstream");
1475}
1476
Chris Lattnerc453f762007-04-29 07:54:31 +00001477bool BitcodeReader::ParseBitcode() {
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001478 TheModule = 0;
1479
Chris Lattnerc453f762007-04-29 07:54:31 +00001480 if (Buffer->getBufferSize() & 3)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001481 return Error("Bitcode stream should be a multiple of 4 bytes in length");
1482
Chris Lattnerc453f762007-04-29 07:54:31 +00001483 unsigned char *BufPtr = (unsigned char *)Buffer->getBufferStart();
Chris Lattner6fa6a322008-07-09 05:14:23 +00001484 unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
1485
1486 // If we have a wrapper header, parse it and ignore the non-bc file contents.
1487 // The magic number is 0x0B17C0DE stored in little endian.
Chris Lattnere2a466b2009-04-06 20:54:32 +00001488 if (isBitcodeWrapper(BufPtr, BufEnd))
1489 if (SkipBitcodeWrapperHeader(BufPtr, BufEnd))
Chris Lattner6fa6a322008-07-09 05:14:23 +00001490 return Error("Invalid bitcode wrapper header");
1491
Chris Lattner962dde32009-04-26 20:59:02 +00001492 StreamFile.init(BufPtr, BufEnd);
1493 Stream.init(StreamFile);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001494
1495 // Sniff for the signature.
1496 if (Stream.Read(8) != 'B' ||
1497 Stream.Read(8) != 'C' ||
1498 Stream.Read(4) != 0x0 ||
1499 Stream.Read(4) != 0xC ||
1500 Stream.Read(4) != 0xE ||
1501 Stream.Read(4) != 0xD)
1502 return Error("Invalid bitcode signature");
1503
1504 // We expect a number of well-defined blocks, though we don't necessarily
1505 // need to understand them all.
1506 while (!Stream.AtEndOfStream()) {
1507 unsigned Code = Stream.ReadCode();
1508
1509 if (Code != bitc::ENTER_SUBBLOCK)
1510 return Error("Invalid record at top-level");
1511
1512 unsigned BlockID = Stream.ReadSubBlockID();
1513
1514 // We only know the MODULE subblock ID.
Chris Lattnere17b6582007-05-05 00:17:00 +00001515 switch (BlockID) {
1516 case bitc::BLOCKINFO_BLOCK_ID:
1517 if (Stream.ReadBlockInfoBlock())
1518 return Error("Malformed BlockInfoBlock");
1519 break;
1520 case bitc::MODULE_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001521 if (ParseModule(Buffer->getBufferIdentifier()))
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001522 return true;
Chris Lattnere17b6582007-05-05 00:17:00 +00001523 break;
1524 default:
1525 if (Stream.SkipBlock())
1526 return Error("Malformed block record");
1527 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001528 }
1529 }
1530
1531 return false;
1532}
Chris Lattnerc453f762007-04-29 07:54:31 +00001533
Chris Lattner48f84872007-05-01 04:59:48 +00001534
Chris Lattner980e5aa2007-05-01 05:52:21 +00001535/// ParseFunctionBody - Lazily parse the specified function body block.
1536bool BitcodeReader::ParseFunctionBody(Function *F) {
Chris Lattnere17b6582007-05-05 00:17:00 +00001537 if (Stream.EnterSubBlock(bitc::FUNCTION_BLOCK_ID))
Chris Lattner980e5aa2007-05-01 05:52:21 +00001538 return Error("Malformed block record");
1539
1540 unsigned ModuleValueListSize = ValueList.size();
1541
1542 // Add all the function arguments to the value table.
1543 for(Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
1544 ValueList.push_back(I);
1545
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001546 unsigned NextValueNo = ValueList.size();
Chris Lattner231cbcb2007-05-02 04:27:25 +00001547 BasicBlock *CurBB = 0;
1548 unsigned CurBBNo = 0;
1549
Chris Lattner980e5aa2007-05-01 05:52:21 +00001550 // Read all the records.
1551 SmallVector<uint64_t, 64> Record;
1552 while (1) {
1553 unsigned Code = Stream.ReadCode();
1554 if (Code == bitc::END_BLOCK) {
1555 if (Stream.ReadBlockEnd())
1556 return Error("Error at end of function block");
1557 break;
1558 }
1559
1560 if (Code == bitc::ENTER_SUBBLOCK) {
1561 switch (Stream.ReadSubBlockID()) {
1562 default: // Skip unknown content.
1563 if (Stream.SkipBlock())
1564 return Error("Malformed block record");
1565 break;
1566 case bitc::CONSTANTS_BLOCK_ID:
1567 if (ParseConstants()) return true;
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001568 NextValueNo = ValueList.size();
Chris Lattner980e5aa2007-05-01 05:52:21 +00001569 break;
1570 case bitc::VALUE_SYMTAB_BLOCK_ID:
1571 if (ParseValueSymbolTable()) return true;
1572 break;
1573 }
1574 continue;
1575 }
1576
1577 if (Code == bitc::DEFINE_ABBREV) {
1578 Stream.ReadAbbrevRecord();
1579 continue;
1580 }
1581
1582 // Read a record.
1583 Record.clear();
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001584 Instruction *I = 0;
Dan Gohman1224c382009-07-20 21:19:07 +00001585 unsigned BitCode = Stream.ReadRecord(Code, Record);
1586 switch (BitCode) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001587 default: // Default behavior: reject
1588 return Error("Unknown instruction");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001589 case bitc::FUNC_CODE_DECLAREBLOCKS: // DECLAREBLOCKS: [nblocks]
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001590 if (Record.size() < 1 || Record[0] == 0)
1591 return Error("Invalid DECLAREBLOCKS record");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001592 // Create all the basic blocks for the function.
Chris Lattnerf61e6452007-05-03 22:09:51 +00001593 FunctionBBs.resize(Record[0]);
Chris Lattner980e5aa2007-05-01 05:52:21 +00001594 for (unsigned i = 0, e = FunctionBBs.size(); i != e; ++i)
Owen Anderson1d0be152009-08-13 21:58:54 +00001595 FunctionBBs[i] = BasicBlock::Create(Context, "", F);
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001596 CurBB = FunctionBBs[0];
1597 continue;
1598
Chris Lattnerabfbf852007-05-06 00:21:25 +00001599 case bitc::FUNC_CODE_INST_BINOP: { // BINOP: [opval, ty, opval, opcode]
1600 unsigned OpNum = 0;
1601 Value *LHS, *RHS;
1602 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1603 getValue(Record, OpNum, LHS->getType(), RHS) ||
Dan Gohman1224c382009-07-20 21:19:07 +00001604 OpNum+1 > Record.size())
Chris Lattnerabfbf852007-05-06 00:21:25 +00001605 return Error("Invalid BINOP record");
1606
Dan Gohman1224c382009-07-20 21:19:07 +00001607 int Opc = GetDecodedBinaryOpcode(Record[OpNum++], LHS->getType());
Chris Lattnerabfbf852007-05-06 00:21:25 +00001608 if (Opc == -1) return Error("Invalid BINOP record");
Gabor Greif7cbd8a32008-05-16 19:29:10 +00001609 I = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
Dan Gohman1224c382009-07-20 21:19:07 +00001610 if (OpNum < Record.size())
1611 SetOptimizationFlags(I, Record[3]);
Chris Lattner980e5aa2007-05-01 05:52:21 +00001612 break;
1613 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00001614 case bitc::FUNC_CODE_INST_CAST: { // CAST: [opval, opty, destty, castopc]
1615 unsigned OpNum = 0;
1616 Value *Op;
1617 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
1618 OpNum+2 != Record.size())
1619 return Error("Invalid CAST record");
1620
1621 const Type *ResTy = getTypeByID(Record[OpNum]);
1622 int Opc = GetDecodedCastOpcode(Record[OpNum+1]);
1623 if (Opc == -1 || ResTy == 0)
Chris Lattner231cbcb2007-05-02 04:27:25 +00001624 return Error("Invalid CAST record");
Gabor Greif7cbd8a32008-05-16 19:29:10 +00001625 I = CastInst::Create((Instruction::CastOps)Opc, Op, ResTy);
Chris Lattner231cbcb2007-05-02 04:27:25 +00001626 break;
1627 }
Dan Gohmandd8004d2009-07-27 21:53:46 +00001628 case bitc::FUNC_CODE_INST_INBOUNDS_GEP:
Chris Lattner15e6d172007-05-04 19:11:41 +00001629 case bitc::FUNC_CODE_INST_GEP: { // GEP: [n x operands]
Chris Lattner7337ab92007-05-06 00:00:00 +00001630 unsigned OpNum = 0;
1631 Value *BasePtr;
1632 if (getValueTypePair(Record, OpNum, NextValueNo, BasePtr))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001633 return Error("Invalid GEP record");
1634
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001635 SmallVector<Value*, 16> GEPIdx;
Chris Lattner7337ab92007-05-06 00:00:00 +00001636 while (OpNum != Record.size()) {
1637 Value *Op;
1638 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001639 return Error("Invalid GEP record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001640 GEPIdx.push_back(Op);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001641 }
1642
Gabor Greif051a9502008-04-06 20:25:17 +00001643 I = GetElementPtrInst::Create(BasePtr, GEPIdx.begin(), GEPIdx.end());
Dan Gohmandd8004d2009-07-27 21:53:46 +00001644 if (BitCode == bitc::FUNC_CODE_INST_INBOUNDS_GEP)
1645 cast<GEPOperator>(I)->setIsInBounds(true);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001646 break;
1647 }
Chris Lattner231cbcb2007-05-02 04:27:25 +00001648
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001649 case bitc::FUNC_CODE_INST_EXTRACTVAL: {
1650 // EXTRACTVAL: [opty, opval, n x indices]
Dan Gohmane4977cf2008-05-23 01:55:30 +00001651 unsigned OpNum = 0;
1652 Value *Agg;
1653 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1654 return Error("Invalid EXTRACTVAL record");
1655
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001656 SmallVector<unsigned, 4> EXTRACTVALIdx;
1657 for (unsigned RecSize = Record.size();
1658 OpNum != RecSize; ++OpNum) {
1659 uint64_t Index = Record[OpNum];
1660 if ((unsigned)Index != Index)
1661 return Error("Invalid EXTRACTVAL index");
1662 EXTRACTVALIdx.push_back((unsigned)Index);
Dan Gohmane4977cf2008-05-23 01:55:30 +00001663 }
1664
1665 I = ExtractValueInst::Create(Agg,
1666 EXTRACTVALIdx.begin(), EXTRACTVALIdx.end());
1667 break;
1668 }
1669
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001670 case bitc::FUNC_CODE_INST_INSERTVAL: {
1671 // INSERTVAL: [opty, opval, opty, opval, n x indices]
Dan Gohmane4977cf2008-05-23 01:55:30 +00001672 unsigned OpNum = 0;
1673 Value *Agg;
1674 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1675 return Error("Invalid INSERTVAL record");
1676 Value *Val;
1677 if (getValueTypePair(Record, OpNum, NextValueNo, Val))
1678 return Error("Invalid INSERTVAL record");
1679
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001680 SmallVector<unsigned, 4> INSERTVALIdx;
1681 for (unsigned RecSize = Record.size();
1682 OpNum != RecSize; ++OpNum) {
1683 uint64_t Index = Record[OpNum];
1684 if ((unsigned)Index != Index)
1685 return Error("Invalid INSERTVAL index");
1686 INSERTVALIdx.push_back((unsigned)Index);
Dan Gohmane4977cf2008-05-23 01:55:30 +00001687 }
1688
1689 I = InsertValueInst::Create(Agg, Val,
1690 INSERTVALIdx.begin(), INSERTVALIdx.end());
1691 break;
1692 }
1693
Chris Lattnerabfbf852007-05-06 00:21:25 +00001694 case bitc::FUNC_CODE_INST_SELECT: { // SELECT: [opval, ty, opval, opval]
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001695 // obsolete form of select
1696 // handles select i1 ... in old bitcode
Chris Lattnerabfbf852007-05-06 00:21:25 +00001697 unsigned OpNum = 0;
1698 Value *TrueVal, *FalseVal, *Cond;
1699 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1700 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
Owen Anderson1d0be152009-08-13 21:58:54 +00001701 getValue(Record, OpNum, Type::getInt1Ty(Context), Cond))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001702 return Error("Invalid SELECT record");
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001703
1704 I = SelectInst::Create(Cond, TrueVal, FalseVal);
1705 break;
1706 }
1707
1708 case bitc::FUNC_CODE_INST_VSELECT: {// VSELECT: [ty,opval,opval,predty,pred]
1709 // new form of select
1710 // handles select i1 or select [N x i1]
1711 unsigned OpNum = 0;
1712 Value *TrueVal, *FalseVal, *Cond;
1713 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1714 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
1715 getValueTypePair(Record, OpNum, NextValueNo, Cond))
1716 return Error("Invalid SELECT record");
Dan Gohmanf72fb672008-09-09 01:02:47 +00001717
1718 // select condition can be either i1 or [N x i1]
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001719 if (const VectorType* vector_type =
1720 dyn_cast<const VectorType>(Cond->getType())) {
Dan Gohmanf72fb672008-09-09 01:02:47 +00001721 // expect <n x i1>
Owen Anderson1d0be152009-08-13 21:58:54 +00001722 if (vector_type->getElementType() != Type::getInt1Ty(Context))
Dan Gohmanf72fb672008-09-09 01:02:47 +00001723 return Error("Invalid SELECT condition type");
1724 } else {
1725 // expect i1
Owen Anderson1d0be152009-08-13 21:58:54 +00001726 if (Cond->getType() != Type::getInt1Ty(Context))
Dan Gohmanf72fb672008-09-09 01:02:47 +00001727 return Error("Invalid SELECT condition type");
1728 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00001729
Gabor Greif051a9502008-04-06 20:25:17 +00001730 I = SelectInst::Create(Cond, TrueVal, FalseVal);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001731 break;
1732 }
1733
1734 case bitc::FUNC_CODE_INST_EXTRACTELT: { // EXTRACTELT: [opty, opval, opval]
Chris Lattnerabfbf852007-05-06 00:21:25 +00001735 unsigned OpNum = 0;
1736 Value *Vec, *Idx;
1737 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
Owen Anderson1d0be152009-08-13 21:58:54 +00001738 getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001739 return Error("Invalid EXTRACTELT record");
Eric Christophera3500da2009-07-25 02:28:41 +00001740 I = ExtractElementInst::Create(Vec, Idx);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001741 break;
1742 }
1743
1744 case bitc::FUNC_CODE_INST_INSERTELT: { // INSERTELT: [ty, opval,opval,opval]
Chris Lattnerabfbf852007-05-06 00:21:25 +00001745 unsigned OpNum = 0;
1746 Value *Vec, *Elt, *Idx;
1747 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
1748 getValue(Record, OpNum,
1749 cast<VectorType>(Vec->getType())->getElementType(), Elt) ||
Owen Anderson1d0be152009-08-13 21:58:54 +00001750 getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001751 return Error("Invalid INSERTELT record");
Gabor Greif051a9502008-04-06 20:25:17 +00001752 I = InsertElementInst::Create(Vec, Elt, Idx);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001753 break;
1754 }
1755
Chris Lattnerabfbf852007-05-06 00:21:25 +00001756 case bitc::FUNC_CODE_INST_SHUFFLEVEC: {// SHUFFLEVEC: [opval,ty,opval,opval]
1757 unsigned OpNum = 0;
1758 Value *Vec1, *Vec2, *Mask;
1759 if (getValueTypePair(Record, OpNum, NextValueNo, Vec1) ||
1760 getValue(Record, OpNum, Vec1->getType(), Vec2))
1761 return Error("Invalid SHUFFLEVEC record");
1762
Mon P Wangaeb06d22008-11-10 04:46:22 +00001763 if (getValueTypePair(Record, OpNum, NextValueNo, Mask))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001764 return Error("Invalid SHUFFLEVEC record");
1765 I = new ShuffleVectorInst(Vec1, Vec2, Mask);
1766 break;
1767 }
Mon P Wangaeb06d22008-11-10 04:46:22 +00001768
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +00001769 case bitc::FUNC_CODE_INST_CMP: // CMP: [opty, opval, opval, pred]
1770 // Old form of ICmp/FCmp returning bool
1771 // Existed to differentiate between icmp/fcmp and vicmp/vfcmp which were
1772 // both legal on vectors but had different behaviour.
1773 case bitc::FUNC_CODE_INST_CMP2: { // CMP2: [opty, opval, opval, pred]
1774 // FCmp/ICmp returning bool or vector of bool
1775
Chris Lattner7337ab92007-05-06 00:00:00 +00001776 unsigned OpNum = 0;
1777 Value *LHS, *RHS;
1778 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1779 getValue(Record, OpNum, LHS->getType(), RHS) ||
1780 OpNum+1 != Record.size())
Chris Lattner01ff65f2007-05-02 05:16:49 +00001781 return Error("Invalid CMP record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001782
Dan Gohmanf72fb672008-09-09 01:02:47 +00001783 if (LHS->getType()->isFPOrFPVector())
Owen Anderson333c4002009-07-09 23:48:35 +00001784 I = new FCmpInst(Context, (FCmpInst::Predicate)Record[OpNum], LHS, RHS);
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +00001785 else
Owen Anderson333c4002009-07-09 23:48:35 +00001786 I = new ICmpInst(Context, (ICmpInst::Predicate)Record[OpNum], LHS, RHS);
Dan Gohmanf72fb672008-09-09 01:02:47 +00001787 break;
1788 }
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +00001789
Devang Patel197be3d2008-02-22 02:49:49 +00001790 case bitc::FUNC_CODE_INST_GETRESULT: { // GETRESULT: [ty, val, n]
1791 if (Record.size() != 2)
1792 return Error("Invalid GETRESULT record");
1793 unsigned OpNum = 0;
1794 Value *Op;
1795 getValueTypePair(Record, OpNum, NextValueNo, Op);
1796 unsigned Index = Record[1];
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001797 I = ExtractValueInst::Create(Op, Index);
Devang Patel197be3d2008-02-22 02:49:49 +00001798 break;
1799 }
Chris Lattner01ff65f2007-05-02 05:16:49 +00001800
Chris Lattner231cbcb2007-05-02 04:27:25 +00001801 case bitc::FUNC_CODE_INST_RET: // RET: [opty,opval<optional>]
Devang Pateld9d99ff2008-02-26 01:29:32 +00001802 {
1803 unsigned Size = Record.size();
1804 if (Size == 0) {
Owen Anderson1d0be152009-08-13 21:58:54 +00001805 I = ReturnInst::Create(Context);
Devang Pateld9d99ff2008-02-26 01:29:32 +00001806 break;
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001807 }
Devang Pateld9d99ff2008-02-26 01:29:32 +00001808
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001809 unsigned OpNum = 0;
1810 SmallVector<Value *,4> Vs;
1811 do {
1812 Value *Op = NULL;
1813 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1814 return Error("Invalid RET record");
1815 Vs.push_back(Op);
1816 } while(OpNum != Record.size());
1817
1818 const Type *ReturnType = F->getReturnType();
1819 if (Vs.size() > 1 ||
1820 (isa<StructType>(ReturnType) &&
1821 (Vs.empty() || Vs[0]->getType() != ReturnType))) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001822 Value *RV = UndefValue::get(ReturnType);
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001823 for (unsigned i = 0, e = Vs.size(); i != e; ++i) {
1824 I = InsertValueInst::Create(RV, Vs[i], i, "mrv");
1825 CurBB->getInstList().push_back(I);
1826 ValueList.AssignValue(I, NextValueNo++);
1827 RV = I;
1828 }
Owen Anderson1d0be152009-08-13 21:58:54 +00001829 I = ReturnInst::Create(Context, RV);
Devang Pateld9d99ff2008-02-26 01:29:32 +00001830 break;
1831 }
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001832
Owen Anderson1d0be152009-08-13 21:58:54 +00001833 I = ReturnInst::Create(Context, Vs[0]);
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001834 break;
Chris Lattner231cbcb2007-05-02 04:27:25 +00001835 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001836 case bitc::FUNC_CODE_INST_BR: { // BR: [bb#, bb#, opval] or [bb#]
Chris Lattnerf61e6452007-05-03 22:09:51 +00001837 if (Record.size() != 1 && Record.size() != 3)
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001838 return Error("Invalid BR record");
1839 BasicBlock *TrueDest = getBasicBlock(Record[0]);
1840 if (TrueDest == 0)
1841 return Error("Invalid BR record");
1842
1843 if (Record.size() == 1)
Gabor Greif051a9502008-04-06 20:25:17 +00001844 I = BranchInst::Create(TrueDest);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001845 else {
1846 BasicBlock *FalseDest = getBasicBlock(Record[1]);
Owen Anderson1d0be152009-08-13 21:58:54 +00001847 Value *Cond = getFnValueByID(Record[2], Type::getInt1Ty(Context));
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001848 if (FalseDest == 0 || Cond == 0)
1849 return Error("Invalid BR record");
Gabor Greif051a9502008-04-06 20:25:17 +00001850 I = BranchInst::Create(TrueDest, FalseDest, Cond);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001851 }
1852 break;
1853 }
1854 case bitc::FUNC_CODE_INST_SWITCH: { // SWITCH: [opty, opval, n, n x ops]
1855 if (Record.size() < 3 || (Record.size() & 1) == 0)
1856 return Error("Invalid SWITCH record");
1857 const Type *OpTy = getTypeByID(Record[0]);
1858 Value *Cond = getFnValueByID(Record[1], OpTy);
1859 BasicBlock *Default = getBasicBlock(Record[2]);
1860 if (OpTy == 0 || Cond == 0 || Default == 0)
1861 return Error("Invalid SWITCH record");
1862 unsigned NumCases = (Record.size()-3)/2;
Gabor Greif051a9502008-04-06 20:25:17 +00001863 SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001864 for (unsigned i = 0, e = NumCases; i != e; ++i) {
1865 ConstantInt *CaseVal =
1866 dyn_cast_or_null<ConstantInt>(getFnValueByID(Record[3+i*2], OpTy));
1867 BasicBlock *DestBB = getBasicBlock(Record[1+3+i*2]);
1868 if (CaseVal == 0 || DestBB == 0) {
1869 delete SI;
1870 return Error("Invalid SWITCH record!");
1871 }
1872 SI->addCase(CaseVal, DestBB);
1873 }
1874 I = SI;
1875 break;
1876 }
1877
Duncan Sandsdc024672007-11-27 13:23:08 +00001878 case bitc::FUNC_CODE_INST_INVOKE: {
1879 // INVOKE: [attrs, cc, normBB, unwindBB, fnty, op0,op1,op2, ...]
Chris Lattnera9bb7132007-05-08 05:38:01 +00001880 if (Record.size() < 4) return Error("Invalid INVOKE record");
Devang Patel05988662008-09-25 21:00:45 +00001881 AttrListPtr PAL = getAttributes(Record[0]);
Chris Lattnera9bb7132007-05-08 05:38:01 +00001882 unsigned CCInfo = Record[1];
1883 BasicBlock *NormalBB = getBasicBlock(Record[2]);
1884 BasicBlock *UnwindBB = getBasicBlock(Record[3]);
Chris Lattner7337ab92007-05-06 00:00:00 +00001885
Chris Lattnera9bb7132007-05-08 05:38:01 +00001886 unsigned OpNum = 4;
Chris Lattner7337ab92007-05-06 00:00:00 +00001887 Value *Callee;
1888 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001889 return Error("Invalid INVOKE record");
1890
Chris Lattner7337ab92007-05-06 00:00:00 +00001891 const PointerType *CalleeTy = dyn_cast<PointerType>(Callee->getType());
1892 const FunctionType *FTy = !CalleeTy ? 0 :
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001893 dyn_cast<FunctionType>(CalleeTy->getElementType());
1894
1895 // Check that the right number of fixed parameters are here.
Chris Lattner7337ab92007-05-06 00:00:00 +00001896 if (FTy == 0 || NormalBB == 0 || UnwindBB == 0 ||
1897 Record.size() < OpNum+FTy->getNumParams())
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001898 return Error("Invalid INVOKE record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001899
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001900 SmallVector<Value*, 16> Ops;
Chris Lattner7337ab92007-05-06 00:00:00 +00001901 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
1902 Ops.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
1903 if (Ops.back() == 0) return Error("Invalid INVOKE record");
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001904 }
1905
Chris Lattner7337ab92007-05-06 00:00:00 +00001906 if (!FTy->isVarArg()) {
1907 if (Record.size() != OpNum)
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001908 return Error("Invalid INVOKE record");
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001909 } else {
Chris Lattner7337ab92007-05-06 00:00:00 +00001910 // Read type/value pairs for varargs params.
1911 while (OpNum != Record.size()) {
1912 Value *Op;
1913 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1914 return Error("Invalid INVOKE record");
1915 Ops.push_back(Op);
1916 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001917 }
1918
Gabor Greifb1dbcd82008-05-15 10:04:30 +00001919 I = InvokeInst::Create(Callee, NormalBB, UnwindBB,
1920 Ops.begin(), Ops.end());
Chris Lattner76520192007-05-03 22:34:03 +00001921 cast<InvokeInst>(I)->setCallingConv(CCInfo);
Devang Patel05988662008-09-25 21:00:45 +00001922 cast<InvokeInst>(I)->setAttributes(PAL);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001923 break;
1924 }
Chris Lattner231cbcb2007-05-02 04:27:25 +00001925 case bitc::FUNC_CODE_INST_UNWIND: // UNWIND
Owen Anderson1d0be152009-08-13 21:58:54 +00001926 I = new UnwindInst(Context);
Chris Lattner231cbcb2007-05-02 04:27:25 +00001927 break;
1928 case bitc::FUNC_CODE_INST_UNREACHABLE: // UNREACHABLE
Owen Anderson1d0be152009-08-13 21:58:54 +00001929 I = new UnreachableInst(Context);
Chris Lattner231cbcb2007-05-02 04:27:25 +00001930 break;
Chris Lattnerabfbf852007-05-06 00:21:25 +00001931 case bitc::FUNC_CODE_INST_PHI: { // PHI: [ty, val0,bb0, ...]
Chris Lattner15e6d172007-05-04 19:11:41 +00001932 if (Record.size() < 1 || ((Record.size()-1)&1))
Chris Lattner2a98cca2007-05-03 18:58:09 +00001933 return Error("Invalid PHI record");
1934 const Type *Ty = getTypeByID(Record[0]);
1935 if (!Ty) return Error("Invalid PHI record");
1936
Gabor Greif051a9502008-04-06 20:25:17 +00001937 PHINode *PN = PHINode::Create(Ty);
Chris Lattner86941612008-04-13 00:14:42 +00001938 PN->reserveOperandSpace((Record.size()-1)/2);
Chris Lattner2a98cca2007-05-03 18:58:09 +00001939
Chris Lattner15e6d172007-05-04 19:11:41 +00001940 for (unsigned i = 0, e = Record.size()-1; i != e; i += 2) {
1941 Value *V = getFnValueByID(Record[1+i], Ty);
1942 BasicBlock *BB = getBasicBlock(Record[2+i]);
Chris Lattner2a98cca2007-05-03 18:58:09 +00001943 if (!V || !BB) return Error("Invalid PHI record");
1944 PN->addIncoming(V, BB);
1945 }
1946 I = PN;
1947 break;
1948 }
1949
1950 case bitc::FUNC_CODE_INST_MALLOC: { // MALLOC: [instty, op, align]
1951 if (Record.size() < 3)
1952 return Error("Invalid MALLOC record");
1953 const PointerType *Ty =
1954 dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
Owen Anderson1d0be152009-08-13 21:58:54 +00001955 Value *Size = getFnValueByID(Record[1], Type::getInt32Ty(Context));
Chris Lattner2a98cca2007-05-03 18:58:09 +00001956 unsigned Align = Record[2];
1957 if (!Ty || !Size) return Error("Invalid MALLOC record");
Owen Anderson50dead02009-07-15 23:53:25 +00001958 I = new MallocInst(Ty->getElementType(), Size, (1 << Align) >> 1);
Chris Lattner2a98cca2007-05-03 18:58:09 +00001959 break;
1960 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00001961 case bitc::FUNC_CODE_INST_FREE: { // FREE: [op, opty]
1962 unsigned OpNum = 0;
1963 Value *Op;
1964 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
1965 OpNum != Record.size())
Chris Lattner2a98cca2007-05-03 18:58:09 +00001966 return Error("Invalid FREE record");
1967 I = new FreeInst(Op);
1968 break;
1969 }
1970 case bitc::FUNC_CODE_INST_ALLOCA: { // ALLOCA: [instty, op, align]
1971 if (Record.size() < 3)
1972 return Error("Invalid ALLOCA record");
1973 const PointerType *Ty =
1974 dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
Owen Anderson1d0be152009-08-13 21:58:54 +00001975 Value *Size = getFnValueByID(Record[1], Type::getInt32Ty(Context));
Chris Lattner2a98cca2007-05-03 18:58:09 +00001976 unsigned Align = Record[2];
1977 if (!Ty || !Size) return Error("Invalid ALLOCA record");
Owen Anderson50dead02009-07-15 23:53:25 +00001978 I = new AllocaInst(Ty->getElementType(), Size, (1 << Align) >> 1);
Chris Lattner2a98cca2007-05-03 18:58:09 +00001979 break;
1980 }
Chris Lattner0579f7f2007-05-03 22:04:19 +00001981 case bitc::FUNC_CODE_INST_LOAD: { // LOAD: [opty, op, align, vol]
Chris Lattner7337ab92007-05-06 00:00:00 +00001982 unsigned OpNum = 0;
1983 Value *Op;
1984 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
1985 OpNum+2 != Record.size())
Chris Lattnerabfbf852007-05-06 00:21:25 +00001986 return Error("Invalid LOAD record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001987
1988 I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001989 break;
Chris Lattner0579f7f2007-05-03 22:04:19 +00001990 }
Christopher Lambfe63fb92007-12-11 08:59:05 +00001991 case bitc::FUNC_CODE_INST_STORE2: { // STORE2:[ptrty, ptr, val, align, vol]
1992 unsigned OpNum = 0;
1993 Value *Val, *Ptr;
1994 if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
1995 getValue(Record, OpNum,
1996 cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
1997 OpNum+2 != Record.size())
1998 return Error("Invalid STORE record");
1999
2000 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
2001 break;
2002 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00002003 case bitc::FUNC_CODE_INST_STORE: { // STORE:[val, valty, ptr, align, vol]
Christopher Lambfe63fb92007-12-11 08:59:05 +00002004 // FIXME: Legacy form of store instruction. Should be removed in LLVM 3.0.
Chris Lattnerabfbf852007-05-06 00:21:25 +00002005 unsigned OpNum = 0;
2006 Value *Val, *Ptr;
2007 if (getValueTypePair(Record, OpNum, NextValueNo, Val) ||
Owen Anderson74a77812009-07-07 20:18:58 +00002008 getValue(Record, OpNum,
Owen Andersondebcb012009-07-29 22:17:13 +00002009 PointerType::getUnqual(Val->getType()), Ptr)||
Chris Lattnerabfbf852007-05-06 00:21:25 +00002010 OpNum+2 != Record.size())
Chris Lattner0579f7f2007-05-03 22:04:19 +00002011 return Error("Invalid STORE record");
Chris Lattnerabfbf852007-05-06 00:21:25 +00002012
2013 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002014 break;
2015 }
Duncan Sandsdc024672007-11-27 13:23:08 +00002016 case bitc::FUNC_CODE_INST_CALL: {
2017 // CALL: [paramattrs, cc, fnty, fnid, arg0, arg1...]
2018 if (Record.size() < 3)
Chris Lattner0579f7f2007-05-03 22:04:19 +00002019 return Error("Invalid CALL record");
Chris Lattner7337ab92007-05-06 00:00:00 +00002020
Devang Patel05988662008-09-25 21:00:45 +00002021 AttrListPtr PAL = getAttributes(Record[0]);
Chris Lattnera9bb7132007-05-08 05:38:01 +00002022 unsigned CCInfo = Record[1];
2023
2024 unsigned OpNum = 2;
Chris Lattner7337ab92007-05-06 00:00:00 +00002025 Value *Callee;
2026 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
2027 return Error("Invalid CALL record");
2028
2029 const PointerType *OpTy = dyn_cast<PointerType>(Callee->getType());
Chris Lattner0579f7f2007-05-03 22:04:19 +00002030 const FunctionType *FTy = 0;
2031 if (OpTy) FTy = dyn_cast<FunctionType>(OpTy->getElementType());
Chris Lattner7337ab92007-05-06 00:00:00 +00002032 if (!FTy || Record.size() < FTy->getNumParams()+OpNum)
Chris Lattner0579f7f2007-05-03 22:04:19 +00002033 return Error("Invalid CALL record");
2034
2035 SmallVector<Value*, 16> Args;
2036 // Read the fixed params.
Chris Lattner7337ab92007-05-06 00:00:00 +00002037 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
Dale Johanneseneb57ea72007-11-05 21:20:28 +00002038 if (FTy->getParamType(i)->getTypeID()==Type::LabelTyID)
2039 Args.push_back(getBasicBlock(Record[OpNum]));
2040 else
2041 Args.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
Chris Lattner0579f7f2007-05-03 22:04:19 +00002042 if (Args.back() == 0) return Error("Invalid CALL record");
2043 }
2044
Chris Lattner0579f7f2007-05-03 22:04:19 +00002045 // Read type/value pairs for varargs params.
Chris Lattner0579f7f2007-05-03 22:04:19 +00002046 if (!FTy->isVarArg()) {
Chris Lattner7337ab92007-05-06 00:00:00 +00002047 if (OpNum != Record.size())
Chris Lattner0579f7f2007-05-03 22:04:19 +00002048 return Error("Invalid CALL record");
2049 } else {
Chris Lattner7337ab92007-05-06 00:00:00 +00002050 while (OpNum != Record.size()) {
2051 Value *Op;
2052 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2053 return Error("Invalid CALL record");
2054 Args.push_back(Op);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002055 }
2056 }
2057
Gabor Greif051a9502008-04-06 20:25:17 +00002058 I = CallInst::Create(Callee, Args.begin(), Args.end());
Chris Lattner76520192007-05-03 22:34:03 +00002059 cast<CallInst>(I)->setCallingConv(CCInfo>>1);
2060 cast<CallInst>(I)->setTailCall(CCInfo & 1);
Devang Patel05988662008-09-25 21:00:45 +00002061 cast<CallInst>(I)->setAttributes(PAL);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002062 break;
2063 }
2064 case bitc::FUNC_CODE_INST_VAARG: { // VAARG: [valistty, valist, instty]
2065 if (Record.size() < 3)
2066 return Error("Invalid VAARG record");
2067 const Type *OpTy = getTypeByID(Record[0]);
2068 Value *Op = getFnValueByID(Record[1], OpTy);
2069 const Type *ResTy = getTypeByID(Record[2]);
2070 if (!OpTy || !Op || !ResTy)
2071 return Error("Invalid VAARG record");
2072 I = new VAArgInst(Op, ResTy);
2073 break;
2074 }
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002075 }
2076
2077 // Add instruction to end of current BB. If there is no current BB, reject
2078 // this file.
2079 if (CurBB == 0) {
2080 delete I;
2081 return Error("Invalid instruction with no BB");
2082 }
2083 CurBB->getInstList().push_back(I);
2084
2085 // If this was a terminator instruction, move to the next block.
2086 if (isa<TerminatorInst>(I)) {
2087 ++CurBBNo;
2088 CurBB = CurBBNo < FunctionBBs.size() ? FunctionBBs[CurBBNo] : 0;
2089 }
2090
2091 // Non-void values get registered in the value table for future use.
Owen Anderson1d0be152009-08-13 21:58:54 +00002092 if (I && I->getType() != Type::getVoidTy(Context))
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002093 ValueList.AssignValue(I, NextValueNo++);
Chris Lattner980e5aa2007-05-01 05:52:21 +00002094 }
2095
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002096 // Check the function list for unresolved values.
2097 if (Argument *A = dyn_cast<Argument>(ValueList.back())) {
2098 if (A->getParent() == 0) {
2099 // We found at least one unresolved value. Nuke them all to avoid leaks.
2100 for (unsigned i = ModuleValueListSize, e = ValueList.size(); i != e; ++i){
2101 if ((A = dyn_cast<Argument>(ValueList.back())) && A->getParent() == 0) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00002102 A->replaceAllUsesWith(UndefValue::get(A->getType()));
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002103 delete A;
2104 }
2105 }
Chris Lattner35a04702007-05-04 03:50:29 +00002106 return Error("Never resolved value found in function!");
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002107 }
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002108 }
Chris Lattner980e5aa2007-05-01 05:52:21 +00002109
2110 // Trim the value list down to the size it was before we parsed this function.
2111 ValueList.shrinkTo(ModuleValueListSize);
2112 std::vector<BasicBlock*>().swap(FunctionBBs);
2113
Chris Lattner48f84872007-05-01 04:59:48 +00002114 return false;
2115}
2116
Chris Lattnerb348bb82007-05-18 04:02:46 +00002117//===----------------------------------------------------------------------===//
2118// ModuleProvider implementation
2119//===----------------------------------------------------------------------===//
2120
2121
2122bool BitcodeReader::materializeFunction(Function *F, std::string *ErrInfo) {
2123 // If it already is material, ignore the request.
Gabor Greifa99be512007-07-05 17:07:56 +00002124 if (!F->hasNotBeenReadFromBitcode()) return false;
Chris Lattnerb348bb82007-05-18 04:02:46 +00002125
2126 DenseMap<Function*, std::pair<uint64_t, unsigned> >::iterator DFII =
2127 DeferredFunctionInfo.find(F);
2128 assert(DFII != DeferredFunctionInfo.end() && "Deferred function not found!");
2129
2130 // Move the bit stream to the saved position of the deferred function body and
2131 // restore the real linkage type for the function.
2132 Stream.JumpToBit(DFII->second.first);
2133 F->setLinkage((GlobalValue::LinkageTypes)DFII->second.second);
2134
2135 if (ParseFunctionBody(F)) {
2136 if (ErrInfo) *ErrInfo = ErrorString;
2137 return true;
2138 }
Chandler Carruth69940402007-08-04 01:51:18 +00002139
2140 // Upgrade any old intrinsic calls in the function.
2141 for (UpgradedIntrinsicMap::iterator I = UpgradedIntrinsics.begin(),
2142 E = UpgradedIntrinsics.end(); I != E; ++I) {
2143 if (I->first != I->second) {
2144 for (Value::use_iterator UI = I->first->use_begin(),
2145 UE = I->first->use_end(); UI != UE; ) {
2146 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2147 UpgradeIntrinsicCall(CI, I->second);
2148 }
2149 }
2150 }
Chris Lattnerb348bb82007-05-18 04:02:46 +00002151
2152 return false;
2153}
2154
2155void BitcodeReader::dematerializeFunction(Function *F) {
2156 // If this function isn't materialized, or if it is a proto, this is a noop.
Gabor Greifa99be512007-07-05 17:07:56 +00002157 if (F->hasNotBeenReadFromBitcode() || F->isDeclaration())
Chris Lattnerb348bb82007-05-18 04:02:46 +00002158 return;
2159
2160 assert(DeferredFunctionInfo.count(F) && "No info to read function later?");
2161
2162 // Just forget the function body, we can remat it later.
2163 F->deleteBody();
2164 F->setLinkage(GlobalValue::GhostLinkage);
2165}
2166
2167
2168Module *BitcodeReader::materializeModule(std::string *ErrInfo) {
Chris Lattner714fa952009-06-16 05:15:21 +00002169 // Iterate over the module, deserializing any functions that are still on
2170 // disk.
2171 for (Module::iterator F = TheModule->begin(), E = TheModule->end();
2172 F != E; ++F)
Gabor Greifa99be512007-07-05 17:07:56 +00002173 if (F->hasNotBeenReadFromBitcode() &&
Chris Lattnerb348bb82007-05-18 04:02:46 +00002174 materializeFunction(F, ErrInfo))
2175 return 0;
Chandler Carruth69940402007-08-04 01:51:18 +00002176
2177 // Upgrade any intrinsic calls that slipped through (should not happen!) and
2178 // delete the old functions to clean up. We can't do this unless the entire
2179 // module is materialized because there could always be another function body
2180 // with calls to the old function.
2181 for (std::vector<std::pair<Function*, Function*> >::iterator I =
2182 UpgradedIntrinsics.begin(), E = UpgradedIntrinsics.end(); I != E; ++I) {
2183 if (I->first != I->second) {
2184 for (Value::use_iterator UI = I->first->use_begin(),
2185 UE = I->first->use_end(); UI != UE; ) {
2186 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2187 UpgradeIntrinsicCall(CI, I->second);
2188 }
Chris Lattner7d9eb582009-04-01 01:43:03 +00002189 if (!I->first->use_empty())
2190 I->first->replaceAllUsesWith(I->second);
Chandler Carruth69940402007-08-04 01:51:18 +00002191 I->first->eraseFromParent();
2192 }
2193 }
2194 std::vector<std::pair<Function*, Function*> >().swap(UpgradedIntrinsics);
2195
Chris Lattnerb348bb82007-05-18 04:02:46 +00002196 return TheModule;
2197}
2198
2199
2200/// This method is provided by the parent ModuleProvde class and overriden
2201/// here. It simply releases the module from its provided and frees up our
2202/// state.
2203/// @brief Release our hold on the generated module
2204Module *BitcodeReader::releaseModule(std::string *ErrInfo) {
2205 // Since we're losing control of this Module, we must hand it back complete
2206 Module *M = ModuleProvider::releaseModule(ErrInfo);
2207 FreeState();
2208 return M;
2209}
2210
Chris Lattner48f84872007-05-01 04:59:48 +00002211
Chris Lattnerc453f762007-04-29 07:54:31 +00002212//===----------------------------------------------------------------------===//
2213// External interface
2214//===----------------------------------------------------------------------===//
2215
2216/// getBitcodeModuleProvider - lazy function-at-a-time loading from a file.
2217///
2218ModuleProvider *llvm::getBitcodeModuleProvider(MemoryBuffer *Buffer,
Owen Anderson4434ed42009-07-01 23:13:44 +00002219 LLVMContext& Context,
Chris Lattnerc453f762007-04-29 07:54:31 +00002220 std::string *ErrMsg) {
Owen Anderson8b477ed2009-07-01 16:58:40 +00002221 BitcodeReader *R = new BitcodeReader(Buffer, Context);
Chris Lattnerc453f762007-04-29 07:54:31 +00002222 if (R->ParseBitcode()) {
2223 if (ErrMsg)
2224 *ErrMsg = R->getErrorString();
2225
2226 // Don't let the BitcodeReader dtor delete 'Buffer'.
2227 R->releaseMemoryBuffer();
2228 delete R;
2229 return 0;
2230 }
2231 return R;
2232}
2233
2234/// ParseBitcodeFile - Read the specified bitcode file, returning the module.
2235/// If an error occurs, return null and fill in *ErrMsg if non-null.
Owen Anderson4434ed42009-07-01 23:13:44 +00002236Module *llvm::ParseBitcodeFile(MemoryBuffer *Buffer, LLVMContext& Context,
Owen Anderson8b477ed2009-07-01 16:58:40 +00002237 std::string *ErrMsg){
Chris Lattnerc453f762007-04-29 07:54:31 +00002238 BitcodeReader *R;
Owen Anderson8b477ed2009-07-01 16:58:40 +00002239 R = static_cast<BitcodeReader*>(getBitcodeModuleProvider(Buffer, Context,
2240 ErrMsg));
Chris Lattnerc453f762007-04-29 07:54:31 +00002241 if (!R) return 0;
2242
Chris Lattnerb348bb82007-05-18 04:02:46 +00002243 // Read in the entire module.
2244 Module *M = R->materializeModule(ErrMsg);
2245
2246 // Don't let the BitcodeReader dtor delete 'Buffer', regardless of whether
2247 // there was an error.
Chris Lattnerc453f762007-04-29 07:54:31 +00002248 R->releaseMemoryBuffer();
Chris Lattnerb348bb82007-05-18 04:02:46 +00002249
2250 // If there was no error, tell ModuleProvider not to delete it when its dtor
2251 // is run.
2252 if (M)
2253 M = R->releaseModule(ErrMsg);
Chris Lattner714fa952009-06-16 05:15:21 +00002254
Chris Lattnerc453f762007-04-29 07:54:31 +00002255 delete R;
2256 return M;
2257}