blob: 33da5e4d611a09742f8f6cec481ce8e2fc1fc655 [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"
Devang Patele4b27562009-08-28 23:24:31 +000019#include "llvm/IntrinsicInst.h"
Chris Lattnercaee0dc2007-04-22 06:23:29 +000020#include "llvm/Module.h"
Dan Gohman1224c382009-07-20 21:19:07 +000021#include "llvm/Operator.h"
Chandler Carruth69940402007-08-04 01:51:18 +000022#include "llvm/AutoUpgrade.h"
Chris Lattner0b2482a2007-04-23 21:26:05 +000023#include "llvm/ADT/SmallString.h"
Devang Patelf4511cd2008-02-26 19:38:17 +000024#include "llvm/ADT/SmallVector.h"
Chris Lattner0eef0802007-04-24 04:04:35 +000025#include "llvm/Support/MathExtras.h"
Chris Lattnerc453f762007-04-29 07:54:31 +000026#include "llvm/Support/MemoryBuffer.h"
Gabor Greifefe65362008-05-10 08:32:32 +000027#include "llvm/OperandTraits.h"
Chris Lattnercaee0dc2007-04-22 06:23:29 +000028using namespace llvm;
29
Chris Lattnerb348bb82007-05-18 04:02:46 +000030void BitcodeReader::FreeState() {
Chris Lattnerc453f762007-04-29 07:54:31 +000031 delete Buffer;
Chris Lattnerb348bb82007-05-18 04:02:46 +000032 Buffer = 0;
33 std::vector<PATypeHolder>().swap(TypeList);
34 ValueList.clear();
Devang Pateld5ac4042009-08-04 06:00:18 +000035 MDValueList.clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +000036
Devang Patel19c87462008-09-26 22:53:05 +000037 std::vector<AttrListPtr>().swap(MAttributes);
Chris Lattnerb348bb82007-05-18 04:02:46 +000038 std::vector<BasicBlock*>().swap(FunctionBBs);
39 std::vector<Function*>().swap(FunctionsWithBodies);
40 DeferredFunctionInfo.clear();
Chris Lattnerc453f762007-04-29 07:54:31 +000041}
42
Chris Lattner48c85b82007-05-04 03:30:17 +000043//===----------------------------------------------------------------------===//
44// Helper functions to implement forward reference resolution, etc.
45//===----------------------------------------------------------------------===//
Chris Lattnerc453f762007-04-29 07:54:31 +000046
Chris Lattnercaee0dc2007-04-22 06:23:29 +000047/// ConvertToString - Convert a string from a record into an std::string, return
48/// true on failure.
Chris Lattner0b2482a2007-04-23 21:26:05 +000049template<typename StrTy>
Chris Lattnercaee0dc2007-04-22 06:23:29 +000050static bool ConvertToString(SmallVector<uint64_t, 64> &Record, unsigned Idx,
Chris Lattner0b2482a2007-04-23 21:26:05 +000051 StrTy &Result) {
Chris Lattner15e6d172007-05-04 19:11:41 +000052 if (Idx > Record.size())
Chris Lattnercaee0dc2007-04-22 06:23:29 +000053 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +000054
Chris Lattner15e6d172007-05-04 19:11:41 +000055 for (unsigned i = Idx, e = Record.size(); i != e; ++i)
56 Result += (char)Record[i];
Chris Lattnercaee0dc2007-04-22 06:23:29 +000057 return false;
58}
59
60static GlobalValue::LinkageTypes GetDecodedLinkage(unsigned Val) {
61 switch (Val) {
62 default: // Map unknown/new linkages to external
Bill Wendling3d10a5a2009-07-20 01:03:30 +000063 case 0: return GlobalValue::ExternalLinkage;
64 case 1: return GlobalValue::WeakAnyLinkage;
65 case 2: return GlobalValue::AppendingLinkage;
66 case 3: return GlobalValue::InternalLinkage;
67 case 4: return GlobalValue::LinkOnceAnyLinkage;
68 case 5: return GlobalValue::DLLImportLinkage;
69 case 6: return GlobalValue::DLLExportLinkage;
70 case 7: return GlobalValue::ExternalWeakLinkage;
71 case 8: return GlobalValue::CommonLinkage;
72 case 9: return GlobalValue::PrivateLinkage;
Duncan Sands667d4b82009-03-07 15:45:40 +000073 case 10: return GlobalValue::WeakODRLinkage;
74 case 11: return GlobalValue::LinkOnceODRLinkage;
Chris Lattner266c7bb2009-04-13 05:44:34 +000075 case 12: return GlobalValue::AvailableExternallyLinkage;
Bill Wendling3d10a5a2009-07-20 01:03:30 +000076 case 13: return GlobalValue::LinkerPrivateLinkage;
Chris Lattnercaee0dc2007-04-22 06:23:29 +000077 }
78}
79
80static GlobalValue::VisibilityTypes GetDecodedVisibility(unsigned Val) {
81 switch (Val) {
82 default: // Map unknown visibilities to default.
83 case 0: return GlobalValue::DefaultVisibility;
84 case 1: return GlobalValue::HiddenVisibility;
Anton Korobeynikov9cd3ccf2007-04-29 20:56:48 +000085 case 2: return GlobalValue::ProtectedVisibility;
Chris Lattnercaee0dc2007-04-22 06:23:29 +000086 }
87}
88
Chris Lattnerf581c3b2007-04-24 07:07:11 +000089static int GetDecodedCastOpcode(unsigned Val) {
90 switch (Val) {
91 default: return -1;
92 case bitc::CAST_TRUNC : return Instruction::Trunc;
93 case bitc::CAST_ZEXT : return Instruction::ZExt;
94 case bitc::CAST_SEXT : return Instruction::SExt;
95 case bitc::CAST_FPTOUI : return Instruction::FPToUI;
96 case bitc::CAST_FPTOSI : return Instruction::FPToSI;
97 case bitc::CAST_UITOFP : return Instruction::UIToFP;
98 case bitc::CAST_SITOFP : return Instruction::SIToFP;
99 case bitc::CAST_FPTRUNC : return Instruction::FPTrunc;
100 case bitc::CAST_FPEXT : return Instruction::FPExt;
101 case bitc::CAST_PTRTOINT: return Instruction::PtrToInt;
102 case bitc::CAST_INTTOPTR: return Instruction::IntToPtr;
103 case bitc::CAST_BITCAST : return Instruction::BitCast;
104 }
105}
106static int GetDecodedBinaryOpcode(unsigned Val, const Type *Ty) {
107 switch (Val) {
108 default: return -1;
Dan Gohmanae3a0be2009-06-04 22:49:04 +0000109 case bitc::BINOP_ADD:
110 return Ty->isFPOrFPVector() ? Instruction::FAdd : Instruction::Add;
111 case bitc::BINOP_SUB:
112 return Ty->isFPOrFPVector() ? Instruction::FSub : Instruction::Sub;
113 case bitc::BINOP_MUL:
114 return Ty->isFPOrFPVector() ? Instruction::FMul : Instruction::Mul;
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000115 case bitc::BINOP_UDIV: return Instruction::UDiv;
116 case bitc::BINOP_SDIV:
117 return Ty->isFPOrFPVector() ? Instruction::FDiv : Instruction::SDiv;
118 case bitc::BINOP_UREM: return Instruction::URem;
119 case bitc::BINOP_SREM:
120 return Ty->isFPOrFPVector() ? Instruction::FRem : Instruction::SRem;
121 case bitc::BINOP_SHL: return Instruction::Shl;
122 case bitc::BINOP_LSHR: return Instruction::LShr;
123 case bitc::BINOP_ASHR: return Instruction::AShr;
124 case bitc::BINOP_AND: return Instruction::And;
125 case bitc::BINOP_OR: return Instruction::Or;
126 case bitc::BINOP_XOR: return Instruction::Xor;
127 }
128}
129
Gabor Greifefe65362008-05-10 08:32:32 +0000130namespace llvm {
Chris Lattner522b7b12007-04-24 05:48:56 +0000131namespace {
132 /// @brief A class for maintaining the slot number definition
133 /// as a placeholder for the actual definition for forward constants defs.
134 class ConstantPlaceHolder : public ConstantExpr {
135 ConstantPlaceHolder(); // DO NOT IMPLEMENT
136 void operator=(const ConstantPlaceHolder &); // DO NOT IMPLEMENT
Gabor Greif051a9502008-04-06 20:25:17 +0000137 public:
138 // allocate space for exactly one operand
139 void *operator new(size_t s) {
140 return User::operator new(s, 1);
141 }
Owen Anderson74a77812009-07-07 20:18:58 +0000142 explicit ConstantPlaceHolder(const Type *Ty, LLVMContext& Context)
Gabor Greifefe65362008-05-10 08:32:32 +0000143 : ConstantExpr(Ty, Instruction::UserOp1, &Op<0>(), 1) {
Owen Anderson1d0be152009-08-13 21:58:54 +0000144 Op<0>() = UndefValue::get(Type::getInt32Ty(Context));
Chris Lattner522b7b12007-04-24 05:48:56 +0000145 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000146
Chris Lattnerea693df2008-08-21 02:34:16 +0000147 /// @brief Methods to support type inquiry through isa, cast, and dyn_cast.
148 static inline bool classof(const ConstantPlaceHolder *) { return true; }
149 static bool classof(const Value *V) {
Daniel Dunbara279bc32009-09-20 02:20:51 +0000150 return isa<ConstantExpr>(V) &&
Chris Lattnerea693df2008-08-21 02:34:16 +0000151 cast<ConstantExpr>(V)->getOpcode() == Instruction::UserOp1;
152 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000153
154
Gabor Greifefe65362008-05-10 08:32:32 +0000155 /// Provide fast operand accessors
Chris Lattner46e77402009-03-31 22:55:09 +0000156 //DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
Chris Lattner522b7b12007-04-24 05:48:56 +0000157 };
158}
159
Chris Lattner46e77402009-03-31 22:55:09 +0000160// FIXME: can we inherit this from ConstantExpr?
Gabor Greifefe65362008-05-10 08:32:32 +0000161template <>
Duncan Sands59bf4fc2009-09-06 08:55:57 +0000162struct OperandTraits<ConstantPlaceHolder> : public FixedNumOperandTraits<1> {
Gabor Greifefe65362008-05-10 08:32:32 +0000163};
Gabor Greifefe65362008-05-10 08:32:32 +0000164}
165
Chris Lattner46e77402009-03-31 22:55:09 +0000166
167void BitcodeReaderValueList::AssignValue(Value *V, unsigned Idx) {
168 if (Idx == size()) {
169 push_back(V);
170 return;
171 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000172
Chris Lattner46e77402009-03-31 22:55:09 +0000173 if (Idx >= size())
174 resize(Idx+1);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000175
Chris Lattner46e77402009-03-31 22:55:09 +0000176 WeakVH &OldV = ValuePtrs[Idx];
177 if (OldV == 0) {
178 OldV = V;
179 return;
180 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000181
Chris Lattner46e77402009-03-31 22:55:09 +0000182 // Handle constants and non-constants (e.g. instrs) differently for
183 // efficiency.
184 if (Constant *PHC = dyn_cast<Constant>(&*OldV)) {
185 ResolveConstants.push_back(std::make_pair(PHC, Idx));
186 OldV = V;
187 } else {
188 // If there was a forward reference to this value, replace it.
189 Value *PrevVal = OldV;
190 OldV->replaceAllUsesWith(V);
191 delete PrevVal;
Gabor Greifefe65362008-05-10 08:32:32 +0000192 }
193}
Daniel Dunbara279bc32009-09-20 02:20:51 +0000194
Gabor Greifefe65362008-05-10 08:32:32 +0000195
Chris Lattner522b7b12007-04-24 05:48:56 +0000196Constant *BitcodeReaderValueList::getConstantFwdRef(unsigned Idx,
197 const Type *Ty) {
Chris Lattner46e77402009-03-31 22:55:09 +0000198 if (Idx >= size())
Gabor Greifefe65362008-05-10 08:32:32 +0000199 resize(Idx + 1);
Chris Lattner522b7b12007-04-24 05:48:56 +0000200
Chris Lattner46e77402009-03-31 22:55:09 +0000201 if (Value *V = ValuePtrs[Idx]) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000202 assert(Ty == V->getType() && "Type mismatch in constant table!");
203 return cast<Constant>(V);
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000204 }
Chris Lattner522b7b12007-04-24 05:48:56 +0000205
206 // Create and return a placeholder, which will later be RAUW'd.
Owen Anderson74a77812009-07-07 20:18:58 +0000207 Constant *C = new ConstantPlaceHolder(Ty, Context);
Chris Lattner46e77402009-03-31 22:55:09 +0000208 ValuePtrs[Idx] = C;
Chris Lattner522b7b12007-04-24 05:48:56 +0000209 return C;
210}
211
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000212Value *BitcodeReaderValueList::getValueFwdRef(unsigned Idx, const Type *Ty) {
Chris Lattner46e77402009-03-31 22:55:09 +0000213 if (Idx >= size())
Gabor Greifefe65362008-05-10 08:32:32 +0000214 resize(Idx + 1);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000215
Chris Lattner46e77402009-03-31 22:55:09 +0000216 if (Value *V = ValuePtrs[Idx]) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000217 assert((Ty == 0 || Ty == V->getType()) && "Type mismatch in value table!");
218 return V;
219 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000220
Chris Lattner01ff65f2007-05-02 05:16:49 +0000221 // No type specified, must be invalid reference.
222 if (Ty == 0) return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000223
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000224 // Create and return a placeholder, which will later be RAUW'd.
225 Value *V = new Argument(Ty);
Chris Lattner46e77402009-03-31 22:55:09 +0000226 ValuePtrs[Idx] = V;
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000227 return V;
228}
229
Chris Lattnerea693df2008-08-21 02:34:16 +0000230/// ResolveConstantForwardRefs - Once all constants are read, this method bulk
231/// resolves any forward references. The idea behind this is that we sometimes
232/// get constants (such as large arrays) which reference *many* forward ref
233/// constants. Replacing each of these causes a lot of thrashing when
234/// building/reuniquing the constant. Instead of doing this, we look at all the
235/// uses and rewrite all the place holders at once for any constant that uses
236/// a placeholder.
237void BitcodeReaderValueList::ResolveConstantForwardRefs() {
Daniel Dunbara279bc32009-09-20 02:20:51 +0000238 // Sort the values by-pointer so that they are efficient to look up with a
Chris Lattnerea693df2008-08-21 02:34:16 +0000239 // binary search.
240 std::sort(ResolveConstants.begin(), ResolveConstants.end());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000241
Chris Lattnerea693df2008-08-21 02:34:16 +0000242 SmallVector<Constant*, 64> NewOps;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000243
Chris Lattnerea693df2008-08-21 02:34:16 +0000244 while (!ResolveConstants.empty()) {
Chris Lattner46e77402009-03-31 22:55:09 +0000245 Value *RealVal = operator[](ResolveConstants.back().second);
Chris Lattnerea693df2008-08-21 02:34:16 +0000246 Constant *Placeholder = ResolveConstants.back().first;
247 ResolveConstants.pop_back();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000248
Chris Lattnerea693df2008-08-21 02:34:16 +0000249 // Loop over all users of the placeholder, updating them to reference the
250 // new value. If they reference more than one placeholder, update them all
251 // at once.
252 while (!Placeholder->use_empty()) {
Chris Lattnerb6135a02008-08-21 17:31:45 +0000253 Value::use_iterator UI = Placeholder->use_begin();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000254
Chris Lattnerea693df2008-08-21 02:34:16 +0000255 // If the using object isn't uniqued, just update the operands. This
256 // handles instructions and initializers for global variables.
Chris Lattnerb6135a02008-08-21 17:31:45 +0000257 if (!isa<Constant>(*UI) || isa<GlobalValue>(*UI)) {
258 UI.getUse().set(RealVal);
Chris Lattnerea693df2008-08-21 02:34:16 +0000259 continue;
260 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000261
Chris Lattnerea693df2008-08-21 02:34:16 +0000262 // Otherwise, we have a constant that uses the placeholder. Replace that
263 // constant with a new constant that has *all* placeholder uses updated.
Chris Lattnerb6135a02008-08-21 17:31:45 +0000264 Constant *UserC = cast<Constant>(*UI);
Chris Lattnerea693df2008-08-21 02:34:16 +0000265 for (User::op_iterator I = UserC->op_begin(), E = UserC->op_end();
266 I != E; ++I) {
267 Value *NewOp;
268 if (!isa<ConstantPlaceHolder>(*I)) {
269 // Not a placeholder reference.
270 NewOp = *I;
271 } else if (*I == Placeholder) {
272 // Common case is that it just references this one placeholder.
273 NewOp = RealVal;
274 } else {
275 // Otherwise, look up the placeholder in ResolveConstants.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000276 ResolveConstantsTy::iterator It =
277 std::lower_bound(ResolveConstants.begin(), ResolveConstants.end(),
Chris Lattnerea693df2008-08-21 02:34:16 +0000278 std::pair<Constant*, unsigned>(cast<Constant>(*I),
279 0));
280 assert(It != ResolveConstants.end() && It->first == *I);
Chris Lattner46e77402009-03-31 22:55:09 +0000281 NewOp = operator[](It->second);
Chris Lattnerea693df2008-08-21 02:34:16 +0000282 }
283
284 NewOps.push_back(cast<Constant>(NewOp));
285 }
286
287 // Make the new constant.
288 Constant *NewC;
289 if (ConstantArray *UserCA = dyn_cast<ConstantArray>(UserC)) {
Owen Anderson1fd70962009-07-28 18:32:17 +0000290 NewC = ConstantArray::get(UserCA->getType(), &NewOps[0],
Owen Anderson74a77812009-07-07 20:18:58 +0000291 NewOps.size());
Chris Lattnerea693df2008-08-21 02:34:16 +0000292 } else if (ConstantStruct *UserCS = dyn_cast<ConstantStruct>(UserC)) {
Owen Andersond7f2a6c2009-08-05 23:16:16 +0000293 NewC = ConstantStruct::get(Context, &NewOps[0], NewOps.size(),
Owen Anderson74a77812009-07-07 20:18:58 +0000294 UserCS->getType()->isPacked());
Chris Lattnerea693df2008-08-21 02:34:16 +0000295 } else if (isa<ConstantVector>(UserC)) {
Owen Andersonaf7ec972009-07-28 21:19:26 +0000296 NewC = ConstantVector::get(&NewOps[0], NewOps.size());
Nick Lewyckycb337992009-05-10 20:57:05 +0000297 } else {
298 assert(isa<ConstantExpr>(UserC) && "Must be a ConstantExpr.");
Chris Lattnerea693df2008-08-21 02:34:16 +0000299 NewC = cast<ConstantExpr>(UserC)->getWithOperands(&NewOps[0],
300 NewOps.size());
301 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000302
Chris Lattnerea693df2008-08-21 02:34:16 +0000303 UserC->replaceAllUsesWith(NewC);
304 UserC->destroyConstant();
305 NewOps.clear();
306 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000307
Nick Lewyckycb337992009-05-10 20:57:05 +0000308 // Update all ValueHandles, they should be the only users at this point.
309 Placeholder->replaceAllUsesWith(RealVal);
Chris Lattnerea693df2008-08-21 02:34:16 +0000310 delete Placeholder;
311 }
312}
313
Devang Pateld5ac4042009-08-04 06:00:18 +0000314void BitcodeReaderMDValueList::AssignValue(Value *V, unsigned Idx) {
315 if (Idx == size()) {
316 push_back(V);
317 return;
318 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000319
Devang Pateld5ac4042009-08-04 06:00:18 +0000320 if (Idx >= size())
321 resize(Idx+1);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000322
Devang Pateld5ac4042009-08-04 06:00:18 +0000323 WeakVH &OldV = MDValuePtrs[Idx];
324 if (OldV == 0) {
325 OldV = V;
326 return;
327 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000328
Devang Pateld5ac4042009-08-04 06:00:18 +0000329 // If there was a forward reference to this value, replace it.
330 Value *PrevVal = OldV;
331 OldV->replaceAllUsesWith(V);
332 delete PrevVal;
Devang Patelc0ff8c82009-09-03 01:38:02 +0000333 // Deleting PrevVal sets Idx value in MDValuePtrs to null. Set new
334 // value for Idx.
335 MDValuePtrs[Idx] = V;
Devang Pateld5ac4042009-08-04 06:00:18 +0000336}
337
338Value *BitcodeReaderMDValueList::getValueFwdRef(unsigned Idx) {
339 if (Idx >= size())
340 resize(Idx + 1);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000341
Devang Pateld5ac4042009-08-04 06:00:18 +0000342 if (Value *V = MDValuePtrs[Idx]) {
Chris Lattnercf0fe8d2009-10-05 05:54:46 +0000343 assert(V->getType()->isMetadataTy() && "Type mismatch in value table!");
Devang Pateld5ac4042009-08-04 06:00:18 +0000344 return V;
345 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000346
Devang Pateld5ac4042009-08-04 06:00:18 +0000347 // Create and return a placeholder, which will later be RAUW'd.
Owen Anderson1d0be152009-08-13 21:58:54 +0000348 Value *V = new Argument(Type::getMetadataTy(Context));
Devang Pateld5ac4042009-08-04 06:00:18 +0000349 MDValuePtrs[Idx] = V;
350 return V;
351}
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000352
353const Type *BitcodeReader::getTypeByID(unsigned ID, bool isTypeTable) {
354 // If the TypeID is in range, return it.
355 if (ID < TypeList.size())
356 return TypeList[ID].get();
357 if (!isTypeTable) return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000358
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000359 // The type table allows forward references. Push as many Opaque types as
360 // needed to get up to ID.
361 while (TypeList.size() <= ID)
Owen Anderson0e275dc2009-08-13 23:27:32 +0000362 TypeList.push_back(OpaqueType::get(Context));
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000363 return TypeList.back().get();
364}
365
Chris Lattner48c85b82007-05-04 03:30:17 +0000366//===----------------------------------------------------------------------===//
367// Functions for parsing blocks from the bitcode file
368//===----------------------------------------------------------------------===//
369
Devang Patel05988662008-09-25 21:00:45 +0000370bool BitcodeReader::ParseAttributeBlock() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000371 if (Stream.EnterSubBlock(bitc::PARAMATTR_BLOCK_ID))
Chris Lattner48c85b82007-05-04 03:30:17 +0000372 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000373
Devang Patel19c87462008-09-26 22:53:05 +0000374 if (!MAttributes.empty())
Chris Lattner48c85b82007-05-04 03:30:17 +0000375 return Error("Multiple PARAMATTR blocks found!");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000376
Chris Lattner48c85b82007-05-04 03:30:17 +0000377 SmallVector<uint64_t, 64> Record;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000378
Devang Patel05988662008-09-25 21:00:45 +0000379 SmallVector<AttributeWithIndex, 8> Attrs;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000380
Chris Lattner48c85b82007-05-04 03:30:17 +0000381 // Read all the records.
382 while (1) {
383 unsigned Code = Stream.ReadCode();
384 if (Code == bitc::END_BLOCK) {
385 if (Stream.ReadBlockEnd())
386 return Error("Error at end of PARAMATTR block");
387 return false;
388 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000389
Chris Lattner48c85b82007-05-04 03:30:17 +0000390 if (Code == bitc::ENTER_SUBBLOCK) {
391 // No known subblocks, always skip them.
392 Stream.ReadSubBlockID();
393 if (Stream.SkipBlock())
394 return Error("Malformed block record");
395 continue;
396 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000397
Chris Lattner48c85b82007-05-04 03:30:17 +0000398 if (Code == bitc::DEFINE_ABBREV) {
399 Stream.ReadAbbrevRecord();
400 continue;
401 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000402
Chris Lattner48c85b82007-05-04 03:30:17 +0000403 // Read a record.
404 Record.clear();
405 switch (Stream.ReadRecord(Code, Record)) {
406 default: // Default behavior: ignore.
407 break;
408 case bitc::PARAMATTR_CODE_ENTRY: { // ENTRY: [paramidx0, attr0, ...]
409 if (Record.size() & 1)
410 return Error("Invalid ENTRY record");
411
Chris Lattner9a6cb152008-10-05 18:22:09 +0000412 // FIXME : Remove this autoupgrade code in LLVM 3.0.
Devang Patel19c87462008-09-26 22:53:05 +0000413 // If Function attributes are using index 0 then transfer them
Chris Lattner9a6cb152008-10-05 18:22:09 +0000414 // to index ~0. Index 0 is used for return value attributes but used to be
415 // used for function attributes.
Devang Patel19c87462008-09-26 22:53:05 +0000416 Attributes RetAttribute = Attribute::None;
417 Attributes FnAttribute = Attribute::None;
Chris Lattner48c85b82007-05-04 03:30:17 +0000418 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Nick Lewycky73ddd4f2008-12-19 09:38:31 +0000419 // FIXME: remove in LLVM 3.0
420 // The alignment is stored as a 16-bit raw value from bits 31--16.
421 // We shift the bits above 31 down by 11 bits.
422
423 unsigned Alignment = (Record[i+1] & (0xffffull << 16)) >> 16;
424 if (Alignment && !isPowerOf2_32(Alignment))
425 return Error("Alignment is not a power of two.");
426
427 Attributes ReconstitutedAttr = Record[i+1] & 0xffff;
428 if (Alignment)
429 ReconstitutedAttr |= Attribute::constructAlignmentFromInt(Alignment);
430 ReconstitutedAttr |= (Record[i+1] & (0xffffull << 32)) >> 11;
431 Record[i+1] = ReconstitutedAttr;
432
Devang Patel19c87462008-09-26 22:53:05 +0000433 if (Record[i] == 0)
434 RetAttribute = Record[i+1];
435 else if (Record[i] == ~0U)
436 FnAttribute = Record[i+1];
437 }
Chris Lattner9a6cb152008-10-05 18:22:09 +0000438
439 unsigned OldRetAttrs = (Attribute::NoUnwind|Attribute::NoReturn|
440 Attribute::ReadOnly|Attribute::ReadNone);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000441
Chris Lattner9a6cb152008-10-05 18:22:09 +0000442 if (FnAttribute == Attribute::None && RetAttribute != Attribute::None &&
443 (RetAttribute & OldRetAttrs) != 0) {
444 if (FnAttribute == Attribute::None) { // add a slot so they get added.
445 Record.push_back(~0U);
446 Record.push_back(0);
Devang Patel19c87462008-09-26 22:53:05 +0000447 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000448
Chris Lattner9a6cb152008-10-05 18:22:09 +0000449 FnAttribute |= RetAttribute & OldRetAttrs;
450 RetAttribute &= ~OldRetAttrs;
Chris Lattner48c85b82007-05-04 03:30:17 +0000451 }
Chris Lattner461edd92008-03-12 02:25:52 +0000452
Devang Patel19c87462008-09-26 22:53:05 +0000453 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Chris Lattner9a6cb152008-10-05 18:22:09 +0000454 if (Record[i] == 0) {
455 if (RetAttribute != Attribute::None)
456 Attrs.push_back(AttributeWithIndex::get(0, RetAttribute));
457 } else if (Record[i] == ~0U) {
458 if (FnAttribute != Attribute::None)
459 Attrs.push_back(AttributeWithIndex::get(~0U, FnAttribute));
460 } else if (Record[i+1] != Attribute::None)
Devang Patel19c87462008-09-26 22:53:05 +0000461 Attrs.push_back(AttributeWithIndex::get(Record[i], Record[i+1]));
462 }
Devang Patel19c87462008-09-26 22:53:05 +0000463
464 MAttributes.push_back(AttrListPtr::get(Attrs.begin(), Attrs.end()));
Chris Lattner48c85b82007-05-04 03:30:17 +0000465 Attrs.clear();
466 break;
467 }
Duncan Sands5e41f652007-11-20 14:09:29 +0000468 }
Chris Lattner48c85b82007-05-04 03:30:17 +0000469 }
470}
471
472
Chris Lattner86697142007-05-01 05:01:34 +0000473bool BitcodeReader::ParseTypeTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000474 if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000475 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000476
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000477 if (!TypeList.empty())
478 return Error("Multiple TYPE_BLOCKs found!");
479
480 SmallVector<uint64_t, 64> Record;
481 unsigned NumRecords = 0;
482
483 // Read all the records for this type table.
484 while (1) {
485 unsigned Code = Stream.ReadCode();
486 if (Code == bitc::END_BLOCK) {
487 if (NumRecords != TypeList.size())
488 return Error("Invalid type forward reference in TYPE_BLOCK");
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000489 if (Stream.ReadBlockEnd())
490 return Error("Error at end of type table block");
491 return false;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000492 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000493
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000494 if (Code == bitc::ENTER_SUBBLOCK) {
495 // No known subblocks, always skip them.
496 Stream.ReadSubBlockID();
497 if (Stream.SkipBlock())
498 return Error("Malformed block record");
499 continue;
500 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000501
Chris Lattner36d5e7d2007-04-23 16:04:05 +0000502 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +0000503 Stream.ReadAbbrevRecord();
504 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000505 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000506
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000507 // Read a record.
508 Record.clear();
509 const Type *ResultTy = 0;
510 switch (Stream.ReadRecord(Code, Record)) {
511 default: // Default behavior: unknown type.
512 ResultTy = 0;
513 break;
514 case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
515 // TYPE_CODE_NUMENTRY contains a count of the number of types in the
516 // type list. This allows us to reserve space.
517 if (Record.size() < 1)
518 return Error("Invalid TYPE_CODE_NUMENTRY record");
519 TypeList.reserve(Record[0]);
520 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000521 case bitc::TYPE_CODE_VOID: // VOID
Owen Anderson1d0be152009-08-13 21:58:54 +0000522 ResultTy = Type::getVoidTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000523 break;
524 case bitc::TYPE_CODE_FLOAT: // FLOAT
Owen Anderson1d0be152009-08-13 21:58:54 +0000525 ResultTy = Type::getFloatTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000526 break;
527 case bitc::TYPE_CODE_DOUBLE: // DOUBLE
Owen Anderson1d0be152009-08-13 21:58:54 +0000528 ResultTy = Type::getDoubleTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000529 break;
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000530 case bitc::TYPE_CODE_X86_FP80: // X86_FP80
Owen Anderson1d0be152009-08-13 21:58:54 +0000531 ResultTy = Type::getX86_FP80Ty(Context);
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000532 break;
533 case bitc::TYPE_CODE_FP128: // FP128
Owen Anderson1d0be152009-08-13 21:58:54 +0000534 ResultTy = Type::getFP128Ty(Context);
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000535 break;
536 case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
Owen Anderson1d0be152009-08-13 21:58:54 +0000537 ResultTy = Type::getPPC_FP128Ty(Context);
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000538 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000539 case bitc::TYPE_CODE_LABEL: // LABEL
Owen Anderson1d0be152009-08-13 21:58:54 +0000540 ResultTy = Type::getLabelTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000541 break;
542 case bitc::TYPE_CODE_OPAQUE: // OPAQUE
543 ResultTy = 0;
544 break;
Nick Lewycky7a0370f2009-05-30 05:06:04 +0000545 case bitc::TYPE_CODE_METADATA: // METADATA
Owen Anderson1d0be152009-08-13 21:58:54 +0000546 ResultTy = Type::getMetadataTy(Context);
Nick Lewycky7a0370f2009-05-30 05:06:04 +0000547 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000548 case bitc::TYPE_CODE_INTEGER: // INTEGER: [width]
549 if (Record.size() < 1)
550 return Error("Invalid Integer type record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000551
Owen Anderson1d0be152009-08-13 21:58:54 +0000552 ResultTy = IntegerType::get(Context, Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000553 break;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000554 case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
Christopher Lambfe63fb92007-12-11 08:59:05 +0000555 // [pointee type, address space]
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000556 if (Record.size() < 1)
557 return Error("Invalid POINTER type record");
Christopher Lambfe63fb92007-12-11 08:59:05 +0000558 unsigned AddressSpace = 0;
559 if (Record.size() == 2)
560 AddressSpace = Record[1];
Owen Andersondebcb012009-07-29 22:17:13 +0000561 ResultTy = PointerType::get(getTypeByID(Record[0], true),
Owen Anderson74a77812009-07-07 20:18:58 +0000562 AddressSpace);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000563 break;
Christopher Lambfe63fb92007-12-11 08:59:05 +0000564 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000565 case bitc::TYPE_CODE_FUNCTION: {
Chris Lattnera1afde72007-11-27 17:48:06 +0000566 // FIXME: attrid is dead, remove it in LLVM 3.0
567 // FUNCTION: [vararg, attrid, retty, paramty x N]
568 if (Record.size() < 3)
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000569 return Error("Invalid FUNCTION type record");
570 std::vector<const Type*> ArgTys;
Chris Lattnera1afde72007-11-27 17:48:06 +0000571 for (unsigned i = 3, e = Record.size(); i != e; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000572 ArgTys.push_back(getTypeByID(Record[i], true));
Daniel Dunbara279bc32009-09-20 02:20:51 +0000573
Owen Andersondebcb012009-07-29 22:17:13 +0000574 ResultTy = FunctionType::get(getTypeByID(Record[2], true), ArgTys,
Duncan Sandsdc024672007-11-27 13:23:08 +0000575 Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000576 break;
577 }
Chris Lattner15e6d172007-05-04 19:11:41 +0000578 case bitc::TYPE_CODE_STRUCT: { // STRUCT: [ispacked, eltty x N]
Chris Lattner7108dce2007-05-06 08:21:50 +0000579 if (Record.size() < 1)
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000580 return Error("Invalid STRUCT type record");
581 std::vector<const Type*> EltTys;
Chris Lattner15e6d172007-05-04 19:11:41 +0000582 for (unsigned i = 1, e = Record.size(); i != e; ++i)
583 EltTys.push_back(getTypeByID(Record[i], true));
Owen Andersond7f2a6c2009-08-05 23:16:16 +0000584 ResultTy = StructType::get(Context, EltTys, Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000585 break;
586 }
587 case bitc::TYPE_CODE_ARRAY: // ARRAY: [numelts, eltty]
588 if (Record.size() < 2)
589 return Error("Invalid ARRAY type record");
Owen Andersondebcb012009-07-29 22:17:13 +0000590 ResultTy = ArrayType::get(getTypeByID(Record[1], true), Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000591 break;
592 case bitc::TYPE_CODE_VECTOR: // VECTOR: [numelts, eltty]
593 if (Record.size() < 2)
594 return Error("Invalid VECTOR type record");
Owen Andersondebcb012009-07-29 22:17:13 +0000595 ResultTy = VectorType::get(getTypeByID(Record[1], true), Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000596 break;
597 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000598
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000599 if (NumRecords == TypeList.size()) {
600 // If this is a new type slot, just append it.
Owen Anderson0e275dc2009-08-13 23:27:32 +0000601 TypeList.push_back(ResultTy ? ResultTy : OpaqueType::get(Context));
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000602 ++NumRecords;
603 } else if (ResultTy == 0) {
604 // Otherwise, this was forward referenced, so an opaque type was created,
605 // but the result type is actually just an opaque. Leave the one we
606 // created previously.
607 ++NumRecords;
608 } else {
609 // Otherwise, this was forward referenced, so an opaque type was created.
610 // Resolve the opaque type to the real type now.
611 assert(NumRecords < TypeList.size() && "Typelist imbalance");
612 const OpaqueType *OldTy = cast<OpaqueType>(TypeList[NumRecords++].get());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000613
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000614 // Don't directly push the new type on the Tab. Instead we want to replace
615 // the opaque type we previously inserted with the new concrete value. The
616 // refinement from the abstract (opaque) type to the new type causes all
617 // uses of the abstract type to use the concrete type (NewTy). This will
618 // also cause the opaque type to be deleted.
619 const_cast<OpaqueType*>(OldTy)->refineAbstractTypeTo(ResultTy);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000620
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000621 // This should have replaced the old opaque type with the new type in the
Chris Lattner0eef0802007-04-24 04:04:35 +0000622 // value table... or with a preexisting type that was already in the
623 // system. Let's just make sure it did.
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000624 assert(TypeList[NumRecords-1].get() != OldTy &&
625 "refineAbstractType didn't work!");
626 }
627 }
628}
629
630
Chris Lattner86697142007-05-01 05:01:34 +0000631bool BitcodeReader::ParseTypeSymbolTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000632 if (Stream.EnterSubBlock(bitc::TYPE_SYMTAB_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000633 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000634
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000635 SmallVector<uint64_t, 64> Record;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000636
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000637 // Read all the records for this type table.
638 std::string TypeName;
639 while (1) {
640 unsigned Code = Stream.ReadCode();
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000641 if (Code == bitc::END_BLOCK) {
642 if (Stream.ReadBlockEnd())
643 return Error("Error at end of type symbol table block");
644 return false;
645 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000646
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000647 if (Code == bitc::ENTER_SUBBLOCK) {
648 // No known subblocks, always skip them.
649 Stream.ReadSubBlockID();
650 if (Stream.SkipBlock())
651 return Error("Malformed block record");
652 continue;
653 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000654
Chris Lattner36d5e7d2007-04-23 16:04:05 +0000655 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +0000656 Stream.ReadAbbrevRecord();
657 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000658 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000659
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000660 // Read a record.
661 Record.clear();
662 switch (Stream.ReadRecord(Code, Record)) {
663 default: // Default behavior: unknown type.
664 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000665 case bitc::TST_CODE_ENTRY: // TST_ENTRY: [typeid, namechar x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000666 if (ConvertToString(Record, 1, TypeName))
667 return Error("Invalid TST_ENTRY record");
668 unsigned TypeID = Record[0];
669 if (TypeID >= TypeList.size())
670 return Error("Invalid Type ID in TST_ENTRY record");
671
672 TheModule->addTypeName(TypeName, TypeList[TypeID].get());
673 TypeName.clear();
674 break;
675 }
676 }
677}
678
Chris Lattner86697142007-05-01 05:01:34 +0000679bool BitcodeReader::ParseValueSymbolTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000680 if (Stream.EnterSubBlock(bitc::VALUE_SYMTAB_BLOCK_ID))
Chris Lattner0b2482a2007-04-23 21:26:05 +0000681 return Error("Malformed block record");
682
683 SmallVector<uint64_t, 64> Record;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000684
Chris Lattner0b2482a2007-04-23 21:26:05 +0000685 // Read all the records for this value table.
686 SmallString<128> ValueName;
687 while (1) {
688 unsigned Code = Stream.ReadCode();
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000689 if (Code == bitc::END_BLOCK) {
690 if (Stream.ReadBlockEnd())
691 return Error("Error at end of value symbol table block");
692 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000693 }
Chris Lattner0b2482a2007-04-23 21:26:05 +0000694 if (Code == bitc::ENTER_SUBBLOCK) {
695 // No known subblocks, always skip them.
696 Stream.ReadSubBlockID();
697 if (Stream.SkipBlock())
698 return Error("Malformed block record");
699 continue;
700 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000701
Chris Lattner0b2482a2007-04-23 21:26:05 +0000702 if (Code == bitc::DEFINE_ABBREV) {
703 Stream.ReadAbbrevRecord();
704 continue;
705 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000706
Chris Lattner0b2482a2007-04-23 21:26:05 +0000707 // Read a record.
708 Record.clear();
709 switch (Stream.ReadRecord(Code, Record)) {
710 default: // Default behavior: unknown type.
711 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000712 case bitc::VST_CODE_ENTRY: { // VST_ENTRY: [valueid, namechar x N]
Chris Lattner0b2482a2007-04-23 21:26:05 +0000713 if (ConvertToString(Record, 1, ValueName))
Nick Lewycky88b72932009-05-31 06:07:28 +0000714 return Error("Invalid VST_ENTRY record");
Chris Lattner0b2482a2007-04-23 21:26:05 +0000715 unsigned ValueID = Record[0];
716 if (ValueID >= ValueList.size())
717 return Error("Invalid Value ID in VST_ENTRY record");
718 Value *V = ValueList[ValueID];
Daniel Dunbara279bc32009-09-20 02:20:51 +0000719
Daniel Dunbar3f53fa92009-07-26 00:34:27 +0000720 V->setName(StringRef(ValueName.data(), ValueName.size()));
Chris Lattner0b2482a2007-04-23 21:26:05 +0000721 ValueName.clear();
722 break;
Reid Spencerc8f8a242007-05-04 01:43:33 +0000723 }
724 case bitc::VST_CODE_BBENTRY: {
Chris Lattnere825ed52007-05-03 22:18:21 +0000725 if (ConvertToString(Record, 1, ValueName))
726 return Error("Invalid VST_BBENTRY record");
727 BasicBlock *BB = getBasicBlock(Record[0]);
728 if (BB == 0)
729 return Error("Invalid BB ID in VST_BBENTRY record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000730
Daniel Dunbar3f53fa92009-07-26 00:34:27 +0000731 BB->setName(StringRef(ValueName.data(), ValueName.size()));
Chris Lattnere825ed52007-05-03 22:18:21 +0000732 ValueName.clear();
733 break;
Chris Lattner0b2482a2007-04-23 21:26:05 +0000734 }
Reid Spencerc8f8a242007-05-04 01:43:33 +0000735 }
Chris Lattner0b2482a2007-04-23 21:26:05 +0000736 }
737}
738
Devang Patele54abc92009-07-22 17:43:22 +0000739bool BitcodeReader::ParseMetadata() {
Devang Pateld5ac4042009-08-04 06:00:18 +0000740 unsigned NextValueNo = MDValueList.size();
Devang Patele54abc92009-07-22 17:43:22 +0000741
742 if (Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
743 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000744
Devang Patele54abc92009-07-22 17:43:22 +0000745 SmallVector<uint64_t, 64> Record;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000746
Devang Patele54abc92009-07-22 17:43:22 +0000747 // Read all the records.
748 while (1) {
749 unsigned Code = Stream.ReadCode();
750 if (Code == bitc::END_BLOCK) {
751 if (Stream.ReadBlockEnd())
752 return Error("Error at end of PARAMATTR block");
753 return false;
754 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000755
Devang Patele54abc92009-07-22 17:43:22 +0000756 if (Code == bitc::ENTER_SUBBLOCK) {
757 // No known subblocks, always skip them.
758 Stream.ReadSubBlockID();
759 if (Stream.SkipBlock())
760 return Error("Malformed block record");
761 continue;
762 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000763
Devang Patele54abc92009-07-22 17:43:22 +0000764 if (Code == bitc::DEFINE_ABBREV) {
765 Stream.ReadAbbrevRecord();
766 continue;
767 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000768
Devang Patele54abc92009-07-22 17:43:22 +0000769 // Read a record.
770 Record.clear();
771 switch (Stream.ReadRecord(Code, Record)) {
772 default: // Default behavior: ignore.
773 break;
Devang Patelaa993142009-07-29 22:34:41 +0000774 case bitc::METADATA_NAME: {
775 // Read named of the named metadata.
776 unsigned NameLength = Record.size();
777 SmallString<8> Name;
778 Name.resize(NameLength);
779 for (unsigned i = 0; i != NameLength; ++i)
780 Name[i] = Record[i];
781 Record.clear();
782 Code = Stream.ReadCode();
783
784 // METADATA_NAME is always followed by METADATA_NAMED_NODE.
785 if (Stream.ReadRecord(Code, Record) != bitc::METADATA_NAMED_NODE)
786 assert ( 0 && "Inavlid Named Metadata record");
787
788 // Read named metadata elements.
789 unsigned Size = Record.size();
Devang Patel3e30c2a2010-01-05 20:41:31 +0000790 SmallVector<MDNode *, 8> Elts;
Devang Patelaa993142009-07-29 22:34:41 +0000791 for (unsigned i = 0; i != Size; ++i) {
Devang Patel69d02e02010-01-05 21:47:32 +0000792 if (Record[i] == ~0U)
793 Elts.push_back(NULL);
794 else {
795 Value *MD = MDValueList.getValueFwdRef(Record[i]);
796 if (MDNode *B = dyn_cast_or_null<MDNode>(MD))
797 Elts.push_back(B);
798 }
Devang Patelaa993142009-07-29 22:34:41 +0000799 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000800 Value *V = NamedMDNode::Create(Context, Name.str(), Elts.data(),
Owen Anderson1d0be152009-08-13 21:58:54 +0000801 Elts.size(), TheModule);
Devang Pateld5ac4042009-08-04 06:00:18 +0000802 MDValueList.AssignValue(V, NextValueNo++);
Devang Patelaa993142009-07-29 22:34:41 +0000803 break;
804 }
Devang Patel104cf9e2009-07-23 01:07:34 +0000805 case bitc::METADATA_NODE: {
806 if (Record.empty() || Record.size() % 2 == 1)
807 return Error("Invalid METADATA_NODE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000808
Devang Patel104cf9e2009-07-23 01:07:34 +0000809 unsigned Size = Record.size();
810 SmallVector<Value*, 8> Elts;
811 for (unsigned i = 0; i != Size; i += 2) {
812 const Type *Ty = getTypeByID(Record[i], false);
Chris Lattnercf0fe8d2009-10-05 05:54:46 +0000813 if (Ty->isMetadataTy())
Devang Pateld5ac4042009-08-04 06:00:18 +0000814 Elts.push_back(MDValueList.getValueFwdRef(Record[i+1]));
Benjamin Kramerf0127052010-01-05 13:12:22 +0000815 else if (!Ty->isVoidTy())
Devang Patel104cf9e2009-07-23 01:07:34 +0000816 Elts.push_back(ValueList.getValueFwdRef(Record[i+1], Ty));
817 else
818 Elts.push_back(NULL);
819 }
Owen Anderson647e3012009-07-31 21:35:40 +0000820 Value *V = MDNode::get(Context, &Elts[0], Elts.size());
Devang Pateld5ac4042009-08-04 06:00:18 +0000821 MDValueList.AssignValue(V, NextValueNo++);
Devang Patel104cf9e2009-07-23 01:07:34 +0000822 break;
823 }
Devang Patele54abc92009-07-22 17:43:22 +0000824 case bitc::METADATA_STRING: {
825 unsigned MDStringLength = Record.size();
826 SmallString<8> String;
827 String.resize(MDStringLength);
828 for (unsigned i = 0; i != MDStringLength; ++i)
829 String[i] = Record[i];
Daniel Dunbara279bc32009-09-20 02:20:51 +0000830 Value *V = MDString::get(Context,
Owen Anderson647e3012009-07-31 21:35:40 +0000831 StringRef(String.data(), String.size()));
Devang Pateld5ac4042009-08-04 06:00:18 +0000832 MDValueList.AssignValue(V, NextValueNo++);
Devang Patele54abc92009-07-22 17:43:22 +0000833 break;
834 }
Devang Patele8e02132009-09-18 19:26:43 +0000835 case bitc::METADATA_KIND: {
836 unsigned RecordLength = Record.size();
837 if (Record.empty() || RecordLength < 2)
Daniel Dunbara279bc32009-09-20 02:20:51 +0000838 return Error("Invalid METADATA_KIND record");
Devang Patele8e02132009-09-18 19:26:43 +0000839 SmallString<8> Name;
840 Name.resize(RecordLength-1);
Devang Patela2148402009-09-28 21:14:55 +0000841 unsigned Kind = Record[0];
Daniel Dunbar22bbd9b2009-10-25 23:11:06 +0000842 (void) Kind;
Devang Patele8e02132009-09-18 19:26:43 +0000843 for (unsigned i = 1; i != RecordLength; ++i)
Daniel Dunbara279bc32009-09-20 02:20:51 +0000844 Name[i-1] = Record[i];
Chris Lattner0eb41982009-12-28 20:45:51 +0000845
Chris Lattner08113472009-12-29 09:01:33 +0000846 unsigned NewKind = TheModule->getMDKindID(Name.str());
Chris Lattner0eb41982009-12-28 20:45:51 +0000847 assert(Kind == NewKind &&
848 "FIXME: Unable to handle custom metadata mismatch!");(void)NewKind;
Devang Patele8e02132009-09-18 19:26:43 +0000849 break;
850 }
Devang Patele54abc92009-07-22 17:43:22 +0000851 }
852 }
853}
854
Chris Lattner0eef0802007-04-24 04:04:35 +0000855/// DecodeSignRotatedValue - Decode a signed value stored with the sign bit in
856/// the LSB for dense VBR encoding.
857static uint64_t DecodeSignRotatedValue(uint64_t V) {
858 if ((V & 1) == 0)
859 return V >> 1;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000860 if (V != 1)
Chris Lattner0eef0802007-04-24 04:04:35 +0000861 return -(V >> 1);
862 // There is no such thing as -0 with integers. "-0" really means MININT.
863 return 1ULL << 63;
864}
865
Chris Lattner07d98b42007-04-26 02:46:40 +0000866/// ResolveGlobalAndAliasInits - Resolve all of the initializers for global
867/// values and aliases that we can.
868bool BitcodeReader::ResolveGlobalAndAliasInits() {
869 std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInitWorklist;
870 std::vector<std::pair<GlobalAlias*, unsigned> > AliasInitWorklist;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000871
Chris Lattner07d98b42007-04-26 02:46:40 +0000872 GlobalInitWorklist.swap(GlobalInits);
873 AliasInitWorklist.swap(AliasInits);
874
875 while (!GlobalInitWorklist.empty()) {
Chris Lattner198f34a2007-04-26 03:27:58 +0000876 unsigned ValID = GlobalInitWorklist.back().second;
Chris Lattner07d98b42007-04-26 02:46:40 +0000877 if (ValID >= ValueList.size()) {
878 // Not ready to resolve this yet, it requires something later in the file.
Chris Lattner198f34a2007-04-26 03:27:58 +0000879 GlobalInits.push_back(GlobalInitWorklist.back());
Chris Lattner07d98b42007-04-26 02:46:40 +0000880 } else {
881 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
882 GlobalInitWorklist.back().first->setInitializer(C);
883 else
884 return Error("Global variable initializer is not a constant!");
885 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000886 GlobalInitWorklist.pop_back();
Chris Lattner07d98b42007-04-26 02:46:40 +0000887 }
888
889 while (!AliasInitWorklist.empty()) {
890 unsigned ValID = AliasInitWorklist.back().second;
891 if (ValID >= ValueList.size()) {
892 AliasInits.push_back(AliasInitWorklist.back());
893 } else {
894 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
Anton Korobeynikov7dde0ff2007-04-28 14:57:59 +0000895 AliasInitWorklist.back().first->setAliasee(C);
Chris Lattner07d98b42007-04-26 02:46:40 +0000896 else
897 return Error("Alias initializer is not a constant!");
898 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000899 AliasInitWorklist.pop_back();
Chris Lattner07d98b42007-04-26 02:46:40 +0000900 }
901 return false;
902}
903
Chris Lattner86697142007-05-01 05:01:34 +0000904bool BitcodeReader::ParseConstants() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000905 if (Stream.EnterSubBlock(bitc::CONSTANTS_BLOCK_ID))
Chris Lattnere16504e2007-04-24 03:30:34 +0000906 return Error("Malformed block record");
907
908 SmallVector<uint64_t, 64> Record;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000909
Chris Lattnere16504e2007-04-24 03:30:34 +0000910 // Read all the records for this value table.
Owen Anderson1d0be152009-08-13 21:58:54 +0000911 const Type *CurTy = Type::getInt32Ty(Context);
Chris Lattner522b7b12007-04-24 05:48:56 +0000912 unsigned NextCstNo = ValueList.size();
Chris Lattnere16504e2007-04-24 03:30:34 +0000913 while (1) {
914 unsigned Code = Stream.ReadCode();
Chris Lattnerea693df2008-08-21 02:34:16 +0000915 if (Code == bitc::END_BLOCK)
916 break;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000917
Chris Lattnere16504e2007-04-24 03:30:34 +0000918 if (Code == bitc::ENTER_SUBBLOCK) {
919 // No known subblocks, always skip them.
920 Stream.ReadSubBlockID();
921 if (Stream.SkipBlock())
922 return Error("Malformed block record");
923 continue;
924 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000925
Chris Lattnere16504e2007-04-24 03:30:34 +0000926 if (Code == bitc::DEFINE_ABBREV) {
927 Stream.ReadAbbrevRecord();
928 continue;
929 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000930
Chris Lattnere16504e2007-04-24 03:30:34 +0000931 // Read a record.
932 Record.clear();
933 Value *V = 0;
Dan Gohman1224c382009-07-20 21:19:07 +0000934 unsigned BitCode = Stream.ReadRecord(Code, Record);
935 switch (BitCode) {
Chris Lattnere16504e2007-04-24 03:30:34 +0000936 default: // Default behavior: unknown constant
937 case bitc::CST_CODE_UNDEF: // UNDEF
Owen Anderson9e9a0d52009-07-30 23:03:37 +0000938 V = UndefValue::get(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +0000939 break;
940 case bitc::CST_CODE_SETTYPE: // SETTYPE: [typeid]
941 if (Record.empty())
942 return Error("Malformed CST_SETTYPE record");
943 if (Record[0] >= TypeList.size())
944 return Error("Invalid Type ID in CST_SETTYPE record");
945 CurTy = TypeList[Record[0]];
Chris Lattner0eef0802007-04-24 04:04:35 +0000946 continue; // Skip the ValueList manipulation.
Chris Lattnere16504e2007-04-24 03:30:34 +0000947 case bitc::CST_CODE_NULL: // NULL
Owen Andersona7235ea2009-07-31 20:28:14 +0000948 V = Constant::getNullValue(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +0000949 break;
950 case bitc::CST_CODE_INTEGER: // INTEGER: [intval]
Chris Lattner0eef0802007-04-24 04:04:35 +0000951 if (!isa<IntegerType>(CurTy) || Record.empty())
952 return Error("Invalid CST_INTEGER record");
Owen Andersoneed707b2009-07-24 23:12:02 +0000953 V = ConstantInt::get(CurTy, DecodeSignRotatedValue(Record[0]));
Chris Lattner0eef0802007-04-24 04:04:35 +0000954 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000955 case bitc::CST_CODE_WIDE_INTEGER: {// WIDE_INTEGER: [n x intval]
956 if (!isa<IntegerType>(CurTy) || Record.empty())
Chris Lattner0eef0802007-04-24 04:04:35 +0000957 return Error("Invalid WIDE_INTEGER record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000958
Chris Lattner15e6d172007-05-04 19:11:41 +0000959 unsigned NumWords = Record.size();
Chris Lattner084a8442007-04-24 17:22:05 +0000960 SmallVector<uint64_t, 8> Words;
961 Words.resize(NumWords);
Chris Lattner0eef0802007-04-24 04:04:35 +0000962 for (unsigned i = 0; i != NumWords; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +0000963 Words[i] = DecodeSignRotatedValue(Record[i]);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000964 V = ConstantInt::get(Context,
Owen Andersoneed707b2009-07-24 23:12:02 +0000965 APInt(cast<IntegerType>(CurTy)->getBitWidth(),
966 NumWords, &Words[0]));
Chris Lattner0eef0802007-04-24 04:04:35 +0000967 break;
968 }
Dale Johannesen3f6eb742007-09-11 18:32:33 +0000969 case bitc::CST_CODE_FLOAT: { // FLOAT: [fpval]
Chris Lattner0eef0802007-04-24 04:04:35 +0000970 if (Record.empty())
971 return Error("Invalid FLOAT record");
Chris Lattnercf0fe8d2009-10-05 05:54:46 +0000972 if (CurTy->isFloatTy())
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000973 V = ConstantFP::get(Context, APFloat(APInt(32, (uint32_t)Record[0])));
Chris Lattnercf0fe8d2009-10-05 05:54:46 +0000974 else if (CurTy->isDoubleTy())
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000975 V = ConstantFP::get(Context, APFloat(APInt(64, Record[0])));
Chris Lattnercf0fe8d2009-10-05 05:54:46 +0000976 else if (CurTy->isX86_FP80Ty()) {
Dale Johannesen1b25cb22009-03-23 21:16:53 +0000977 // Bits are not stored the same way as a normal i80 APInt, compensate.
978 uint64_t Rearrange[2];
979 Rearrange[0] = (Record[1] & 0xffffLL) | (Record[0] << 16);
980 Rearrange[1] = Record[0] >> 48;
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000981 V = ConstantFP::get(Context, APFloat(APInt(80, 2, Rearrange)));
Chris Lattnercf0fe8d2009-10-05 05:54:46 +0000982 } else if (CurTy->isFP128Ty())
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000983 V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0]), true));
Chris Lattnercf0fe8d2009-10-05 05:54:46 +0000984 else if (CurTy->isPPC_FP128Ty())
Owen Anderson6f83c9c2009-07-27 20:59:43 +0000985 V = ConstantFP::get(Context, APFloat(APInt(128, 2, &Record[0])));
Chris Lattnere16504e2007-04-24 03:30:34 +0000986 else
Owen Anderson9e9a0d52009-07-30 23:03:37 +0000987 V = UndefValue::get(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +0000988 break;
Dale Johannesen3f6eb742007-09-11 18:32:33 +0000989 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000990
Chris Lattner15e6d172007-05-04 19:11:41 +0000991 case bitc::CST_CODE_AGGREGATE: {// AGGREGATE: [n x value number]
992 if (Record.empty())
Chris Lattner522b7b12007-04-24 05:48:56 +0000993 return Error("Invalid CST_AGGREGATE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000994
Chris Lattner15e6d172007-05-04 19:11:41 +0000995 unsigned Size = Record.size();
Chris Lattner522b7b12007-04-24 05:48:56 +0000996 std::vector<Constant*> Elts;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000997
Chris Lattner522b7b12007-04-24 05:48:56 +0000998 if (const StructType *STy = dyn_cast<StructType>(CurTy)) {
999 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +00001000 Elts.push_back(ValueList.getConstantFwdRef(Record[i],
Chris Lattner522b7b12007-04-24 05:48:56 +00001001 STy->getElementType(i)));
Owen Anderson8fa33382009-07-27 22:29:26 +00001002 V = ConstantStruct::get(STy, Elts);
Chris Lattner522b7b12007-04-24 05:48:56 +00001003 } else if (const ArrayType *ATy = dyn_cast<ArrayType>(CurTy)) {
1004 const Type *EltTy = ATy->getElementType();
1005 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +00001006 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Owen Anderson1fd70962009-07-28 18:32:17 +00001007 V = ConstantArray::get(ATy, Elts);
Chris Lattner522b7b12007-04-24 05:48:56 +00001008 } else if (const VectorType *VTy = dyn_cast<VectorType>(CurTy)) {
1009 const Type *EltTy = VTy->getElementType();
1010 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +00001011 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Owen Andersonaf7ec972009-07-28 21:19:26 +00001012 V = ConstantVector::get(Elts);
Chris Lattner522b7b12007-04-24 05:48:56 +00001013 } else {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001014 V = UndefValue::get(CurTy);
Chris Lattner522b7b12007-04-24 05:48:56 +00001015 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001016 break;
1017 }
Chris Lattnerff7fc5d2007-05-06 00:35:24 +00001018 case bitc::CST_CODE_STRING: { // STRING: [values]
1019 if (Record.empty())
1020 return Error("Invalid CST_AGGREGATE record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001021
Chris Lattnerff7fc5d2007-05-06 00:35:24 +00001022 const ArrayType *ATy = cast<ArrayType>(CurTy);
1023 const Type *EltTy = ATy->getElementType();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001024
Chris Lattnerff7fc5d2007-05-06 00:35:24 +00001025 unsigned Size = Record.size();
1026 std::vector<Constant*> Elts;
Chris Lattnerff7fc5d2007-05-06 00:35:24 +00001027 for (unsigned i = 0; i != Size; ++i)
Owen Andersoneed707b2009-07-24 23:12:02 +00001028 Elts.push_back(ConstantInt::get(EltTy, Record[i]));
Owen Anderson1fd70962009-07-28 18:32:17 +00001029 V = ConstantArray::get(ATy, Elts);
Chris Lattnerff7fc5d2007-05-06 00:35:24 +00001030 break;
1031 }
Chris Lattnercb3d91b2007-05-06 00:53:07 +00001032 case bitc::CST_CODE_CSTRING: { // CSTRING: [values]
1033 if (Record.empty())
1034 return Error("Invalid CST_AGGREGATE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001035
Chris Lattnercb3d91b2007-05-06 00:53:07 +00001036 const ArrayType *ATy = cast<ArrayType>(CurTy);
1037 const Type *EltTy = ATy->getElementType();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001038
Chris Lattnercb3d91b2007-05-06 00:53:07 +00001039 unsigned Size = Record.size();
1040 std::vector<Constant*> Elts;
1041 for (unsigned i = 0; i != Size; ++i)
Owen Andersoneed707b2009-07-24 23:12:02 +00001042 Elts.push_back(ConstantInt::get(EltTy, Record[i]));
Owen Andersona7235ea2009-07-31 20:28:14 +00001043 Elts.push_back(Constant::getNullValue(EltTy));
Owen Anderson1fd70962009-07-28 18:32:17 +00001044 V = ConstantArray::get(ATy, Elts);
Chris Lattnercb3d91b2007-05-06 00:53:07 +00001045 break;
1046 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001047 case bitc::CST_CODE_CE_BINOP: { // CE_BINOP: [opcode, opval, opval]
1048 if (Record.size() < 3) return Error("Invalid CE_BINOP record");
1049 int Opc = GetDecodedBinaryOpcode(Record[0], CurTy);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001050 if (Opc < 0) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001051 V = UndefValue::get(CurTy); // Unknown binop.
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001052 } else {
1053 Constant *LHS = ValueList.getConstantFwdRef(Record[1], CurTy);
1054 Constant *RHS = ValueList.getConstantFwdRef(Record[2], CurTy);
Dan Gohmanf8dbee72009-09-07 23:54:19 +00001055 unsigned Flags = 0;
1056 if (Record.size() >= 4) {
1057 if (Opc == Instruction::Add ||
1058 Opc == Instruction::Sub ||
1059 Opc == Instruction::Mul) {
1060 if (Record[3] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1061 Flags |= OverflowingBinaryOperator::NoSignedWrap;
1062 if (Record[3] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1063 Flags |= OverflowingBinaryOperator::NoUnsignedWrap;
1064 } else if (Opc == Instruction::SDiv) {
1065 if (Record[3] & (1 << bitc::SDIV_EXACT))
1066 Flags |= SDivOperator::IsExact;
1067 }
1068 }
1069 V = ConstantExpr::get(Opc, LHS, RHS, Flags);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001070 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001071 break;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001072 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001073 case bitc::CST_CODE_CE_CAST: { // CE_CAST: [opcode, opty, opval]
1074 if (Record.size() < 3) return Error("Invalid CE_CAST record");
1075 int Opc = GetDecodedCastOpcode(Record[0]);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001076 if (Opc < 0) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001077 V = UndefValue::get(CurTy); // Unknown cast.
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001078 } else {
1079 const Type *OpTy = getTypeByID(Record[1]);
Chris Lattnerbfcc3802007-05-06 07:33:01 +00001080 if (!OpTy) return Error("Invalid CE_CAST record");
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001081 Constant *Op = ValueList.getConstantFwdRef(Record[2], OpTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001082 V = ConstantExpr::getCast(Opc, Op, CurTy);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001083 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001084 break;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001085 }
Dan Gohmandd8004d2009-07-27 21:53:46 +00001086 case bitc::CST_CODE_CE_INBOUNDS_GEP:
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001087 case bitc::CST_CODE_CE_GEP: { // CE_GEP: [n x operands]
Chris Lattner15e6d172007-05-04 19:11:41 +00001088 if (Record.size() & 1) return Error("Invalid CE_GEP record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001089 SmallVector<Constant*, 16> Elts;
Chris Lattner15e6d172007-05-04 19:11:41 +00001090 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001091 const Type *ElTy = getTypeByID(Record[i]);
1092 if (!ElTy) return Error("Invalid CE_GEP record");
1093 Elts.push_back(ValueList.getConstantFwdRef(Record[i+1], ElTy));
1094 }
Dan Gohmandd8004d2009-07-27 21:53:46 +00001095 if (BitCode == bitc::CST_CODE_CE_INBOUNDS_GEP)
Dan Gohmanf8dbee72009-09-07 23:54:19 +00001096 V = ConstantExpr::getInBoundsGetElementPtr(Elts[0], &Elts[1],
1097 Elts.size()-1);
1098 else
1099 V = ConstantExpr::getGetElementPtr(Elts[0], &Elts[1],
1100 Elts.size()-1);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001101 break;
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001102 }
1103 case bitc::CST_CODE_CE_SELECT: // CE_SELECT: [opval#, opval#, opval#]
1104 if (Record.size() < 3) return Error("Invalid CE_SELECT record");
Owen Andersonbaf3c402009-07-29 18:55:55 +00001105 V = ConstantExpr::getSelect(ValueList.getConstantFwdRef(Record[0],
Owen Anderson1d0be152009-08-13 21:58:54 +00001106 Type::getInt1Ty(Context)),
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001107 ValueList.getConstantFwdRef(Record[1],CurTy),
1108 ValueList.getConstantFwdRef(Record[2],CurTy));
1109 break;
1110 case bitc::CST_CODE_CE_EXTRACTELT: { // CE_EXTRACTELT: [opty, opval, opval]
1111 if (Record.size() < 3) return Error("Invalid CE_EXTRACTELT record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001112 const VectorType *OpTy =
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001113 dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
1114 if (OpTy == 0) return Error("Invalid CE_EXTRACTELT record");
1115 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
Owen Anderson1d0be152009-08-13 21:58:54 +00001116 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
Owen Andersonbaf3c402009-07-29 18:55:55 +00001117 V = ConstantExpr::getExtractElement(Op0, Op1);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001118 break;
1119 }
1120 case bitc::CST_CODE_CE_INSERTELT: { // CE_INSERTELT: [opval, opval, opval]
1121 const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1122 if (Record.size() < 3 || OpTy == 0)
1123 return Error("Invalid CE_INSERTELT record");
1124 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1125 Constant *Op1 = ValueList.getConstantFwdRef(Record[1],
1126 OpTy->getElementType());
Owen Anderson1d0be152009-08-13 21:58:54 +00001127 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
Owen Andersonbaf3c402009-07-29 18:55:55 +00001128 V = ConstantExpr::getInsertElement(Op0, Op1, Op2);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001129 break;
1130 }
1131 case bitc::CST_CODE_CE_SHUFFLEVEC: { // CE_SHUFFLEVEC: [opval, opval, opval]
1132 const VectorType *OpTy = dyn_cast<VectorType>(CurTy);
1133 if (Record.size() < 3 || OpTy == 0)
Nate Begeman0f123cf2009-02-12 21:28:33 +00001134 return Error("Invalid CE_SHUFFLEVEC record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001135 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1136 Constant *Op1 = ValueList.getConstantFwdRef(Record[1], OpTy);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001137 const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
Owen Anderson74a77812009-07-07 20:18:58 +00001138 OpTy->getNumElements());
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001139 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], ShufTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001140 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001141 break;
1142 }
Nate Begeman0f123cf2009-02-12 21:28:33 +00001143 case bitc::CST_CODE_CE_SHUFVEC_EX: { // [opty, opval, opval, opval]
1144 const VectorType *RTy = dyn_cast<VectorType>(CurTy);
1145 const VectorType *OpTy = dyn_cast<VectorType>(getTypeByID(Record[0]));
1146 if (Record.size() < 4 || RTy == 0 || OpTy == 0)
1147 return Error("Invalid CE_SHUFVEC_EX record");
1148 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1149 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001150 const Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
Owen Anderson74a77812009-07-07 20:18:58 +00001151 RTy->getNumElements());
Nate Begeman0f123cf2009-02-12 21:28:33 +00001152 Constant *Op2 = ValueList.getConstantFwdRef(Record[3], ShufTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001153 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
Nate Begeman0f123cf2009-02-12 21:28:33 +00001154 break;
1155 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001156 case bitc::CST_CODE_CE_CMP: { // CE_CMP: [opty, opval, opval, pred]
1157 if (Record.size() < 4) return Error("Invalid CE_CMP record");
1158 const Type *OpTy = getTypeByID(Record[0]);
1159 if (OpTy == 0) return Error("Invalid CE_CMP record");
1160 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1161 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1162
1163 if (OpTy->isFloatingPoint())
Owen Andersonbaf3c402009-07-29 18:55:55 +00001164 V = ConstantExpr::getFCmp(Record[3], Op0, Op1);
Nate Begemanac80ade2008-05-12 19:01:56 +00001165 else
Owen Andersonbaf3c402009-07-29 18:55:55 +00001166 V = ConstantExpr::getICmp(Record[3], Op0, Op1);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001167 break;
Chris Lattner522b7b12007-04-24 05:48:56 +00001168 }
Chris Lattner2bce93a2007-05-06 01:58:20 +00001169 case bitc::CST_CODE_INLINEASM: {
1170 if (Record.size() < 2) return Error("Invalid INLINEASM record");
1171 std::string AsmStr, ConstrStr;
Dale Johannesen43602982009-10-13 20:46:56 +00001172 bool HasSideEffects = Record[0] & 1;
Dale Johannesen8ba2d5b2009-10-21 23:28:00 +00001173 bool IsAlignStack = Record[0] >> 1;
Chris Lattner2bce93a2007-05-06 01:58:20 +00001174 unsigned AsmStrSize = Record[1];
1175 if (2+AsmStrSize >= Record.size())
1176 return Error("Invalid INLINEASM record");
1177 unsigned ConstStrSize = Record[2+AsmStrSize];
1178 if (3+AsmStrSize+ConstStrSize > Record.size())
1179 return Error("Invalid INLINEASM record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001180
Chris Lattner2bce93a2007-05-06 01:58:20 +00001181 for (unsigned i = 0; i != AsmStrSize; ++i)
1182 AsmStr += (char)Record[2+i];
1183 for (unsigned i = 0; i != ConstStrSize; ++i)
1184 ConstrStr += (char)Record[3+AsmStrSize+i];
1185 const PointerType *PTy = cast<PointerType>(CurTy);
1186 V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
Dale Johannesen8ba2d5b2009-10-21 23:28:00 +00001187 AsmStr, ConstrStr, HasSideEffects, IsAlignStack);
Chris Lattner2bce93a2007-05-06 01:58:20 +00001188 break;
1189 }
Chris Lattner50b136d2009-10-28 05:53:48 +00001190 case bitc::CST_CODE_BLOCKADDRESS:{
1191 if (Record.size() < 3) return Error("Invalid CE_BLOCKADDRESS record");
1192 const Type *FnTy = getTypeByID(Record[0]);
1193 if (FnTy == 0) return Error("Invalid CE_BLOCKADDRESS record");
1194 Function *Fn =
1195 dyn_cast_or_null<Function>(ValueList.getConstantFwdRef(Record[1],FnTy));
1196 if (Fn == 0) return Error("Invalid CE_BLOCKADDRESS record");
1197
1198 GlobalVariable *FwdRef = new GlobalVariable(*Fn->getParent(),
1199 Type::getInt8Ty(Context),
1200 false, GlobalValue::InternalLinkage,
1201 0, "");
1202 BlockAddrFwdRefs[Fn].push_back(std::make_pair(Record[2], FwdRef));
1203 V = FwdRef;
1204 break;
1205 }
Chris Lattnere16504e2007-04-24 03:30:34 +00001206 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001207
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001208 ValueList.AssignValue(V, NextCstNo);
Chris Lattner522b7b12007-04-24 05:48:56 +00001209 ++NextCstNo;
Chris Lattnere16504e2007-04-24 03:30:34 +00001210 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001211
Chris Lattnerea693df2008-08-21 02:34:16 +00001212 if (NextCstNo != ValueList.size())
1213 return Error("Invalid constant reference!");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001214
Chris Lattnerea693df2008-08-21 02:34:16 +00001215 if (Stream.ReadBlockEnd())
1216 return Error("Error at end of constants block");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001217
Chris Lattnerea693df2008-08-21 02:34:16 +00001218 // Once all the constants have been read, go through and resolve forward
1219 // references.
1220 ValueList.ResolveConstantForwardRefs();
1221 return false;
Chris Lattnere16504e2007-04-24 03:30:34 +00001222}
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001223
Chris Lattner980e5aa2007-05-01 05:52:21 +00001224/// RememberAndSkipFunctionBody - When we see the block for a function body,
1225/// remember where it is and then skip it. This lets us lazily deserialize the
1226/// functions.
1227bool BitcodeReader::RememberAndSkipFunctionBody() {
Chris Lattner48f84872007-05-01 04:59:48 +00001228 // Get the function we are talking about.
1229 if (FunctionsWithBodies.empty())
1230 return Error("Insufficient function protos");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001231
Chris Lattner48f84872007-05-01 04:59:48 +00001232 Function *Fn = FunctionsWithBodies.back();
1233 FunctionsWithBodies.pop_back();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001234
Chris Lattner48f84872007-05-01 04:59:48 +00001235 // Save the current stream state.
1236 uint64_t CurBit = Stream.GetCurrentBitNo();
1237 DeferredFunctionInfo[Fn] = std::make_pair(CurBit, Fn->getLinkage());
Daniel Dunbara279bc32009-09-20 02:20:51 +00001238
Chris Lattner48f84872007-05-01 04:59:48 +00001239 // Set the functions linkage to GhostLinkage so we know it is lazily
1240 // deserialized.
1241 Fn->setLinkage(GlobalValue::GhostLinkage);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001242
Chris Lattner48f84872007-05-01 04:59:48 +00001243 // Skip over the function block for now.
1244 if (Stream.SkipBlock())
1245 return Error("Malformed block record");
1246 return false;
1247}
1248
Chris Lattner86697142007-05-01 05:01:34 +00001249bool BitcodeReader::ParseModule(const std::string &ModuleID) {
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001250 // Reject multiple MODULE_BLOCK's in a single bitstream.
1251 if (TheModule)
1252 return Error("Multiple MODULE_BLOCKs in same stream");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001253
Chris Lattnere17b6582007-05-05 00:17:00 +00001254 if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001255 return Error("Malformed block record");
1256
1257 // Otherwise, create the module.
Owen Anderson8b477ed2009-07-01 16:58:40 +00001258 TheModule = new Module(ModuleID, Context);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001259
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001260 SmallVector<uint64_t, 64> Record;
1261 std::vector<std::string> SectionTable;
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001262 std::vector<std::string> GCTable;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001263
1264 // Read all the records for this module.
1265 while (!Stream.AtEndOfStream()) {
1266 unsigned Code = Stream.ReadCode();
Chris Lattnere84bcb92007-04-24 00:21:45 +00001267 if (Code == bitc::END_BLOCK) {
Chris Lattner980e5aa2007-05-01 05:52:21 +00001268 if (Stream.ReadBlockEnd())
1269 return Error("Error at end of module block");
1270
1271 // Patch the initializers for globals and aliases up.
Chris Lattner07d98b42007-04-26 02:46:40 +00001272 ResolveGlobalAndAliasInits();
1273 if (!GlobalInits.empty() || !AliasInits.empty())
Chris Lattnere84bcb92007-04-24 00:21:45 +00001274 return Error("Malformed global initializer set");
Chris Lattner48f84872007-05-01 04:59:48 +00001275 if (!FunctionsWithBodies.empty())
1276 return Error("Too few function bodies found");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001277
Chandler Carruth69940402007-08-04 01:51:18 +00001278 // Look for intrinsic functions which need to be upgraded at some point
1279 for (Module::iterator FI = TheModule->begin(), FE = TheModule->end();
1280 FI != FE; ++FI) {
Evan Chengf9b83fc2007-12-17 22:33:23 +00001281 Function* NewFn;
1282 if (UpgradeIntrinsicFunction(FI, NewFn))
Chandler Carruth69940402007-08-04 01:51:18 +00001283 UpgradedIntrinsics.push_back(std::make_pair(FI, NewFn));
1284 }
1285
Chris Lattner980e5aa2007-05-01 05:52:21 +00001286 // Force deallocation of memory for these vectors to favor the client that
1287 // want lazy deserialization.
1288 std::vector<std::pair<GlobalVariable*, unsigned> >().swap(GlobalInits);
1289 std::vector<std::pair<GlobalAlias*, unsigned> >().swap(AliasInits);
1290 std::vector<Function*>().swap(FunctionsWithBodies);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001291 return false;
Chris Lattnere84bcb92007-04-24 00:21:45 +00001292 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001293
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001294 if (Code == bitc::ENTER_SUBBLOCK) {
1295 switch (Stream.ReadSubBlockID()) {
1296 default: // Skip unknown content.
1297 if (Stream.SkipBlock())
1298 return Error("Malformed block record");
1299 break;
Chris Lattner3f799802007-05-05 18:57:30 +00001300 case bitc::BLOCKINFO_BLOCK_ID:
1301 if (Stream.ReadBlockInfoBlock())
1302 return Error("Malformed BlockInfoBlock");
1303 break;
Chris Lattner48c85b82007-05-04 03:30:17 +00001304 case bitc::PARAMATTR_BLOCK_ID:
Devang Patel05988662008-09-25 21:00:45 +00001305 if (ParseAttributeBlock())
Chris Lattner48c85b82007-05-04 03:30:17 +00001306 return true;
1307 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001308 case bitc::TYPE_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001309 if (ParseTypeTable())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001310 return true;
1311 break;
1312 case bitc::TYPE_SYMTAB_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001313 if (ParseTypeSymbolTable())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001314 return true;
1315 break;
Chris Lattner0b2482a2007-04-23 21:26:05 +00001316 case bitc::VALUE_SYMTAB_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001317 if (ParseValueSymbolTable())
Chris Lattner0b2482a2007-04-23 21:26:05 +00001318 return true;
1319 break;
Chris Lattnere16504e2007-04-24 03:30:34 +00001320 case bitc::CONSTANTS_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001321 if (ParseConstants() || ResolveGlobalAndAliasInits())
Chris Lattnere16504e2007-04-24 03:30:34 +00001322 return true;
1323 break;
Devang Patele54abc92009-07-22 17:43:22 +00001324 case bitc::METADATA_BLOCK_ID:
1325 if (ParseMetadata())
1326 return true;
1327 break;
Chris Lattner48f84872007-05-01 04:59:48 +00001328 case bitc::FUNCTION_BLOCK_ID:
1329 // If this is the first function body we've seen, reverse the
1330 // FunctionsWithBodies list.
1331 if (!HasReversedFunctionsWithBodies) {
1332 std::reverse(FunctionsWithBodies.begin(), FunctionsWithBodies.end());
1333 HasReversedFunctionsWithBodies = true;
1334 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001335
Chris Lattner980e5aa2007-05-01 05:52:21 +00001336 if (RememberAndSkipFunctionBody())
Chris Lattner48f84872007-05-01 04:59:48 +00001337 return true;
1338 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001339 }
1340 continue;
1341 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001342
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001343 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +00001344 Stream.ReadAbbrevRecord();
1345 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001346 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001347
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001348 // Read a record.
1349 switch (Stream.ReadRecord(Code, Record)) {
1350 default: break; // Default behavior, ignore unknown content.
1351 case bitc::MODULE_CODE_VERSION: // VERSION: [version#]
1352 if (Record.size() < 1)
1353 return Error("Malformed MODULE_CODE_VERSION");
1354 // Only version #0 is supported so far.
1355 if (Record[0] != 0)
1356 return Error("Unknown bitstream version!");
1357 break;
Chris Lattner15e6d172007-05-04 19:11:41 +00001358 case bitc::MODULE_CODE_TRIPLE: { // TRIPLE: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001359 std::string S;
1360 if (ConvertToString(Record, 0, S))
1361 return Error("Invalid MODULE_CODE_TRIPLE record");
1362 TheModule->setTargetTriple(S);
1363 break;
1364 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001365 case bitc::MODULE_CODE_DATALAYOUT: { // DATALAYOUT: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001366 std::string S;
1367 if (ConvertToString(Record, 0, S))
1368 return Error("Invalid MODULE_CODE_DATALAYOUT record");
1369 TheModule->setDataLayout(S);
1370 break;
1371 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001372 case bitc::MODULE_CODE_ASM: { // ASM: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001373 std::string S;
1374 if (ConvertToString(Record, 0, S))
1375 return Error("Invalid MODULE_CODE_ASM record");
1376 TheModule->setModuleInlineAsm(S);
1377 break;
1378 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001379 case bitc::MODULE_CODE_DEPLIB: { // DEPLIB: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001380 std::string S;
1381 if (ConvertToString(Record, 0, S))
1382 return Error("Invalid MODULE_CODE_DEPLIB record");
1383 TheModule->addLibrary(S);
1384 break;
1385 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001386 case bitc::MODULE_CODE_SECTIONNAME: { // SECTIONNAME: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001387 std::string S;
1388 if (ConvertToString(Record, 0, S))
1389 return Error("Invalid MODULE_CODE_SECTIONNAME record");
1390 SectionTable.push_back(S);
1391 break;
1392 }
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001393 case bitc::MODULE_CODE_GCNAME: { // SECTIONNAME: [strchr x N]
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001394 std::string S;
1395 if (ConvertToString(Record, 0, S))
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001396 return Error("Invalid MODULE_CODE_GCNAME record");
1397 GCTable.push_back(S);
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001398 break;
1399 }
Christopher Lambfe63fb92007-12-11 08:59:05 +00001400 // GLOBALVAR: [pointer type, isconst, initid,
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001401 // linkage, alignment, section, visibility, threadlocal]
1402 case bitc::MODULE_CODE_GLOBALVAR: {
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001403 if (Record.size() < 6)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001404 return Error("Invalid MODULE_CODE_GLOBALVAR record");
1405 const Type *Ty = getTypeByID(Record[0]);
1406 if (!isa<PointerType>(Ty))
1407 return Error("Global not a pointer type!");
Christopher Lambfe63fb92007-12-11 08:59:05 +00001408 unsigned AddressSpace = cast<PointerType>(Ty)->getAddressSpace();
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001409 Ty = cast<PointerType>(Ty)->getElementType();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001410
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001411 bool isConstant = Record[1];
1412 GlobalValue::LinkageTypes Linkage = GetDecodedLinkage(Record[3]);
1413 unsigned Alignment = (1 << Record[4]) >> 1;
1414 std::string Section;
1415 if (Record[5]) {
1416 if (Record[5]-1 >= SectionTable.size())
1417 return Error("Invalid section ID");
1418 Section = SectionTable[Record[5]-1];
1419 }
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001420 GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
Chris Lattner5f32c012007-05-06 19:27:46 +00001421 if (Record.size() > 6)
1422 Visibility = GetDecodedVisibility(Record[6]);
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001423 bool isThreadLocal = false;
Chris Lattner5f32c012007-05-06 19:27:46 +00001424 if (Record.size() > 7)
1425 isThreadLocal = Record[7];
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001426
1427 GlobalVariable *NewGV =
Daniel Dunbara279bc32009-09-20 02:20:51 +00001428 new GlobalVariable(*TheModule, Ty, isConstant, Linkage, 0, "", 0,
Christopher Lambfe63fb92007-12-11 08:59:05 +00001429 isThreadLocal, AddressSpace);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001430 NewGV->setAlignment(Alignment);
1431 if (!Section.empty())
1432 NewGV->setSection(Section);
1433 NewGV->setVisibility(Visibility);
1434 NewGV->setThreadLocal(isThreadLocal);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001435
Chris Lattner0b2482a2007-04-23 21:26:05 +00001436 ValueList.push_back(NewGV);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001437
Chris Lattner6dbfd7b2007-04-24 00:18:21 +00001438 // Remember which value to use for the global initializer.
1439 if (unsigned InitID = Record[2])
1440 GlobalInits.push_back(std::make_pair(NewGV, InitID-1));
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001441 break;
1442 }
Chris Lattnera9bb7132007-05-08 05:38:01 +00001443 // FUNCTION: [type, callingconv, isproto, linkage, paramattr,
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001444 // alignment, section, visibility, gc]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001445 case bitc::MODULE_CODE_FUNCTION: {
Chris Lattnera9bb7132007-05-08 05:38:01 +00001446 if (Record.size() < 8)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001447 return Error("Invalid MODULE_CODE_FUNCTION record");
1448 const Type *Ty = getTypeByID(Record[0]);
1449 if (!isa<PointerType>(Ty))
1450 return Error("Function not a pointer type!");
1451 const FunctionType *FTy =
1452 dyn_cast<FunctionType>(cast<PointerType>(Ty)->getElementType());
1453 if (!FTy)
1454 return Error("Function not a pointer to function type!");
1455
Gabor Greif051a9502008-04-06 20:25:17 +00001456 Function *Func = Function::Create(FTy, GlobalValue::ExternalLinkage,
1457 "", TheModule);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001458
Sandeep Patel65c3c8f2009-09-02 08:44:58 +00001459 Func->setCallingConv(static_cast<CallingConv::ID>(Record[1]));
Chris Lattner48f84872007-05-01 04:59:48 +00001460 bool isProto = Record[2];
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001461 Func->setLinkage(GetDecodedLinkage(Record[3]));
Devang Patel05988662008-09-25 21:00:45 +00001462 Func->setAttributes(getAttributes(Record[4]));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001463
Chris Lattnera9bb7132007-05-08 05:38:01 +00001464 Func->setAlignment((1 << Record[5]) >> 1);
1465 if (Record[6]) {
1466 if (Record[6]-1 >= SectionTable.size())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001467 return Error("Invalid section ID");
Chris Lattnera9bb7132007-05-08 05:38:01 +00001468 Func->setSection(SectionTable[Record[6]-1]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001469 }
Chris Lattnera9bb7132007-05-08 05:38:01 +00001470 Func->setVisibility(GetDecodedVisibility(Record[7]));
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001471 if (Record.size() > 8 && Record[8]) {
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001472 if (Record[8]-1 > GCTable.size())
1473 return Error("Invalid GC ID");
1474 Func->setGC(GCTable[Record[8]-1].c_str());
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001475 }
Chris Lattner0b2482a2007-04-23 21:26:05 +00001476 ValueList.push_back(Func);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001477
Chris Lattner48f84872007-05-01 04:59:48 +00001478 // If this is a function with a body, remember the prototype we are
1479 // creating now, so that we can match up the body with them later.
1480 if (!isProto)
1481 FunctionsWithBodies.push_back(Func);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001482 break;
1483 }
Anton Korobeynikov91342d82008-03-12 00:49:19 +00001484 // ALIAS: [alias type, aliasee val#, linkage]
Anton Korobeynikovf8342b92008-03-11 21:40:17 +00001485 // ALIAS: [alias type, aliasee val#, linkage, visibility]
Chris Lattner198f34a2007-04-26 03:27:58 +00001486 case bitc::MODULE_CODE_ALIAS: {
Chris Lattner07d98b42007-04-26 02:46:40 +00001487 if (Record.size() < 3)
1488 return Error("Invalid MODULE_ALIAS record");
1489 const Type *Ty = getTypeByID(Record[0]);
1490 if (!isa<PointerType>(Ty))
1491 return Error("Function not a pointer type!");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001492
Chris Lattner07d98b42007-04-26 02:46:40 +00001493 GlobalAlias *NewGA = new GlobalAlias(Ty, GetDecodedLinkage(Record[2]),
1494 "", 0, TheModule);
Anton Korobeynikov91342d82008-03-12 00:49:19 +00001495 // Old bitcode files didn't have visibility field.
1496 if (Record.size() > 3)
1497 NewGA->setVisibility(GetDecodedVisibility(Record[3]));
Chris Lattner07d98b42007-04-26 02:46:40 +00001498 ValueList.push_back(NewGA);
1499 AliasInits.push_back(std::make_pair(NewGA, Record[1]));
1500 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001501 }
Chris Lattner198f34a2007-04-26 03:27:58 +00001502 /// MODULE_CODE_PURGEVALS: [numvals]
1503 case bitc::MODULE_CODE_PURGEVALS:
1504 // Trim down the value list to the specified size.
1505 if (Record.size() < 1 || Record[0] > ValueList.size())
1506 return Error("Invalid MODULE_PURGEVALS record");
1507 ValueList.shrinkTo(Record[0]);
1508 break;
1509 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001510 Record.clear();
1511 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001512
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001513 return Error("Premature end of bitstream");
1514}
1515
Chris Lattnerc453f762007-04-29 07:54:31 +00001516bool BitcodeReader::ParseBitcode() {
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001517 TheModule = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001518
Chris Lattnerc453f762007-04-29 07:54:31 +00001519 if (Buffer->getBufferSize() & 3)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001520 return Error("Bitcode stream should be a multiple of 4 bytes in length");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001521
Chris Lattnerc453f762007-04-29 07:54:31 +00001522 unsigned char *BufPtr = (unsigned char *)Buffer->getBufferStart();
Chris Lattner6fa6a322008-07-09 05:14:23 +00001523 unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001524
Chris Lattner6fa6a322008-07-09 05:14:23 +00001525 // If we have a wrapper header, parse it and ignore the non-bc file contents.
1526 // The magic number is 0x0B17C0DE stored in little endian.
Chris Lattnere2a466b2009-04-06 20:54:32 +00001527 if (isBitcodeWrapper(BufPtr, BufEnd))
1528 if (SkipBitcodeWrapperHeader(BufPtr, BufEnd))
Chris Lattner6fa6a322008-07-09 05:14:23 +00001529 return Error("Invalid bitcode wrapper header");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001530
Chris Lattner962dde32009-04-26 20:59:02 +00001531 StreamFile.init(BufPtr, BufEnd);
1532 Stream.init(StreamFile);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001533
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001534 // Sniff for the signature.
1535 if (Stream.Read(8) != 'B' ||
1536 Stream.Read(8) != 'C' ||
1537 Stream.Read(4) != 0x0 ||
1538 Stream.Read(4) != 0xC ||
1539 Stream.Read(4) != 0xE ||
1540 Stream.Read(4) != 0xD)
1541 return Error("Invalid bitcode signature");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001542
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001543 // We expect a number of well-defined blocks, though we don't necessarily
1544 // need to understand them all.
1545 while (!Stream.AtEndOfStream()) {
1546 unsigned Code = Stream.ReadCode();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001547
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001548 if (Code != bitc::ENTER_SUBBLOCK)
1549 return Error("Invalid record at top-level");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001550
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001551 unsigned BlockID = Stream.ReadSubBlockID();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001552
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001553 // We only know the MODULE subblock ID.
Chris Lattnere17b6582007-05-05 00:17:00 +00001554 switch (BlockID) {
1555 case bitc::BLOCKINFO_BLOCK_ID:
1556 if (Stream.ReadBlockInfoBlock())
1557 return Error("Malformed BlockInfoBlock");
1558 break;
1559 case bitc::MODULE_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001560 if (ParseModule(Buffer->getBufferIdentifier()))
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001561 return true;
Chris Lattnere17b6582007-05-05 00:17:00 +00001562 break;
1563 default:
1564 if (Stream.SkipBlock())
1565 return Error("Malformed block record");
1566 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001567 }
1568 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001569
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001570 return false;
1571}
Chris Lattnerc453f762007-04-29 07:54:31 +00001572
Devang Patele8e02132009-09-18 19:26:43 +00001573/// ParseMetadataAttachment - Parse metadata attachments.
1574bool BitcodeReader::ParseMetadataAttachment() {
1575 if (Stream.EnterSubBlock(bitc::METADATA_ATTACHMENT_ID))
1576 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001577
Devang Patele8e02132009-09-18 19:26:43 +00001578 SmallVector<uint64_t, 64> Record;
1579 while(1) {
1580 unsigned Code = Stream.ReadCode();
1581 if (Code == bitc::END_BLOCK) {
1582 if (Stream.ReadBlockEnd())
Daniel Dunbara279bc32009-09-20 02:20:51 +00001583 return Error("Error at end of PARAMATTR block");
Devang Patele8e02132009-09-18 19:26:43 +00001584 break;
1585 }
1586 if (Code == bitc::DEFINE_ABBREV) {
1587 Stream.ReadAbbrevRecord();
1588 continue;
1589 }
1590 // Read a metadata attachment record.
1591 Record.clear();
1592 switch (Stream.ReadRecord(Code, Record)) {
1593 default: // Default behavior: ignore.
1594 break;
1595 case bitc::METADATA_ATTACHMENT: {
1596 unsigned RecordLength = Record.size();
1597 if (Record.empty() || (RecordLength - 1) % 2 == 1)
Daniel Dunbara279bc32009-09-20 02:20:51 +00001598 return Error ("Invalid METADATA_ATTACHMENT reader!");
Devang Patele8e02132009-09-18 19:26:43 +00001599 Instruction *Inst = InstructionList[Record[0]];
1600 for (unsigned i = 1; i != RecordLength; i = i+2) {
Devang Patela2148402009-09-28 21:14:55 +00001601 unsigned Kind = Record[i];
Daniel Dunbara279bc32009-09-20 02:20:51 +00001602 Value *Node = MDValueList.getValueFwdRef(Record[i+1]);
Chris Lattner3990b122009-12-28 23:41:32 +00001603 Inst->setMetadata(Kind, cast<MDNode>(Node));
Devang Patele8e02132009-09-18 19:26:43 +00001604 }
1605 break;
1606 }
1607 }
1608 }
1609 return false;
1610}
Chris Lattner48f84872007-05-01 04:59:48 +00001611
Chris Lattner980e5aa2007-05-01 05:52:21 +00001612/// ParseFunctionBody - Lazily parse the specified function body block.
1613bool BitcodeReader::ParseFunctionBody(Function *F) {
Chris Lattnere17b6582007-05-05 00:17:00 +00001614 if (Stream.EnterSubBlock(bitc::FUNCTION_BLOCK_ID))
Chris Lattner980e5aa2007-05-01 05:52:21 +00001615 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001616
Chris Lattner980e5aa2007-05-01 05:52:21 +00001617 unsigned ModuleValueListSize = ValueList.size();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001618
Chris Lattner980e5aa2007-05-01 05:52:21 +00001619 // Add all the function arguments to the value table.
1620 for(Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
1621 ValueList.push_back(I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001622
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001623 unsigned NextValueNo = ValueList.size();
Chris Lattner231cbcb2007-05-02 04:27:25 +00001624 BasicBlock *CurBB = 0;
1625 unsigned CurBBNo = 0;
1626
Chris Lattner980e5aa2007-05-01 05:52:21 +00001627 // Read all the records.
1628 SmallVector<uint64_t, 64> Record;
1629 while (1) {
1630 unsigned Code = Stream.ReadCode();
1631 if (Code == bitc::END_BLOCK) {
1632 if (Stream.ReadBlockEnd())
1633 return Error("Error at end of function block");
1634 break;
1635 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001636
Chris Lattner980e5aa2007-05-01 05:52:21 +00001637 if (Code == bitc::ENTER_SUBBLOCK) {
1638 switch (Stream.ReadSubBlockID()) {
1639 default: // Skip unknown content.
1640 if (Stream.SkipBlock())
1641 return Error("Malformed block record");
1642 break;
1643 case bitc::CONSTANTS_BLOCK_ID:
1644 if (ParseConstants()) return true;
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001645 NextValueNo = ValueList.size();
Chris Lattner980e5aa2007-05-01 05:52:21 +00001646 break;
1647 case bitc::VALUE_SYMTAB_BLOCK_ID:
1648 if (ParseValueSymbolTable()) return true;
1649 break;
Devang Patele8e02132009-09-18 19:26:43 +00001650 case bitc::METADATA_ATTACHMENT_ID:
Daniel Dunbara279bc32009-09-20 02:20:51 +00001651 if (ParseMetadataAttachment()) return true;
1652 break;
Chris Lattner980e5aa2007-05-01 05:52:21 +00001653 }
1654 continue;
1655 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001656
Chris Lattner980e5aa2007-05-01 05:52:21 +00001657 if (Code == bitc::DEFINE_ABBREV) {
1658 Stream.ReadAbbrevRecord();
1659 continue;
1660 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001661
Chris Lattner980e5aa2007-05-01 05:52:21 +00001662 // Read a record.
1663 Record.clear();
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001664 Instruction *I = 0;
Dan Gohman1224c382009-07-20 21:19:07 +00001665 unsigned BitCode = Stream.ReadRecord(Code, Record);
1666 switch (BitCode) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001667 default: // Default behavior: reject
1668 return Error("Unknown instruction");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001669 case bitc::FUNC_CODE_DECLAREBLOCKS: // DECLAREBLOCKS: [nblocks]
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001670 if (Record.size() < 1 || Record[0] == 0)
1671 return Error("Invalid DECLAREBLOCKS record");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001672 // Create all the basic blocks for the function.
Chris Lattnerf61e6452007-05-03 22:09:51 +00001673 FunctionBBs.resize(Record[0]);
Chris Lattner980e5aa2007-05-01 05:52:21 +00001674 for (unsigned i = 0, e = FunctionBBs.size(); i != e; ++i)
Owen Anderson1d0be152009-08-13 21:58:54 +00001675 FunctionBBs[i] = BasicBlock::Create(Context, "", F);
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001676 CurBB = FunctionBBs[0];
1677 continue;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001678
Chris Lattnerabfbf852007-05-06 00:21:25 +00001679 case bitc::FUNC_CODE_INST_BINOP: { // BINOP: [opval, ty, opval, opcode]
1680 unsigned OpNum = 0;
1681 Value *LHS, *RHS;
1682 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1683 getValue(Record, OpNum, LHS->getType(), RHS) ||
Dan Gohman1224c382009-07-20 21:19:07 +00001684 OpNum+1 > Record.size())
Chris Lattnerabfbf852007-05-06 00:21:25 +00001685 return Error("Invalid BINOP record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001686
Dan Gohman1224c382009-07-20 21:19:07 +00001687 int Opc = GetDecodedBinaryOpcode(Record[OpNum++], LHS->getType());
Chris Lattnerabfbf852007-05-06 00:21:25 +00001688 if (Opc == -1) return Error("Invalid BINOP record");
Gabor Greif7cbd8a32008-05-16 19:29:10 +00001689 I = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
Devang Patele8e02132009-09-18 19:26:43 +00001690 InstructionList.push_back(I);
Dan Gohmanf8dbee72009-09-07 23:54:19 +00001691 if (OpNum < Record.size()) {
1692 if (Opc == Instruction::Add ||
1693 Opc == Instruction::Sub ||
1694 Opc == Instruction::Mul) {
1695 if (Record[3] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1696 cast<BinaryOperator>(I)->setHasNoSignedWrap(true);
1697 if (Record[3] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1698 cast<BinaryOperator>(I)->setHasNoUnsignedWrap(true);
1699 } else if (Opc == Instruction::SDiv) {
1700 if (Record[3] & (1 << bitc::SDIV_EXACT))
1701 cast<BinaryOperator>(I)->setIsExact(true);
1702 }
1703 }
Chris Lattner980e5aa2007-05-01 05:52:21 +00001704 break;
1705 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00001706 case bitc::FUNC_CODE_INST_CAST: { // CAST: [opval, opty, destty, castopc]
1707 unsigned OpNum = 0;
1708 Value *Op;
1709 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
1710 OpNum+2 != Record.size())
1711 return Error("Invalid CAST record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001712
Chris Lattnerabfbf852007-05-06 00:21:25 +00001713 const Type *ResTy = getTypeByID(Record[OpNum]);
1714 int Opc = GetDecodedCastOpcode(Record[OpNum+1]);
1715 if (Opc == -1 || ResTy == 0)
Chris Lattner231cbcb2007-05-02 04:27:25 +00001716 return Error("Invalid CAST record");
Gabor Greif7cbd8a32008-05-16 19:29:10 +00001717 I = CastInst::Create((Instruction::CastOps)Opc, Op, ResTy);
Devang Patele8e02132009-09-18 19:26:43 +00001718 InstructionList.push_back(I);
Chris Lattner231cbcb2007-05-02 04:27:25 +00001719 break;
1720 }
Dan Gohmandd8004d2009-07-27 21:53:46 +00001721 case bitc::FUNC_CODE_INST_INBOUNDS_GEP:
Chris Lattner15e6d172007-05-04 19:11:41 +00001722 case bitc::FUNC_CODE_INST_GEP: { // GEP: [n x operands]
Chris Lattner7337ab92007-05-06 00:00:00 +00001723 unsigned OpNum = 0;
1724 Value *BasePtr;
1725 if (getValueTypePair(Record, OpNum, NextValueNo, BasePtr))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001726 return Error("Invalid GEP record");
1727
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001728 SmallVector<Value*, 16> GEPIdx;
Chris Lattner7337ab92007-05-06 00:00:00 +00001729 while (OpNum != Record.size()) {
1730 Value *Op;
1731 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001732 return Error("Invalid GEP record");
Chris Lattner7337ab92007-05-06 00:00:00 +00001733 GEPIdx.push_back(Op);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001734 }
1735
Gabor Greif051a9502008-04-06 20:25:17 +00001736 I = GetElementPtrInst::Create(BasePtr, GEPIdx.begin(), GEPIdx.end());
Devang Patele8e02132009-09-18 19:26:43 +00001737 InstructionList.push_back(I);
Dan Gohmandd8004d2009-07-27 21:53:46 +00001738 if (BitCode == bitc::FUNC_CODE_INST_INBOUNDS_GEP)
Dan Gohmanf8dbee72009-09-07 23:54:19 +00001739 cast<GetElementPtrInst>(I)->setIsInBounds(true);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001740 break;
1741 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001742
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001743 case bitc::FUNC_CODE_INST_EXTRACTVAL: {
1744 // EXTRACTVAL: [opty, opval, n x indices]
Dan Gohmane4977cf2008-05-23 01:55:30 +00001745 unsigned OpNum = 0;
1746 Value *Agg;
1747 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1748 return Error("Invalid EXTRACTVAL record");
1749
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001750 SmallVector<unsigned, 4> EXTRACTVALIdx;
1751 for (unsigned RecSize = Record.size();
1752 OpNum != RecSize; ++OpNum) {
1753 uint64_t Index = Record[OpNum];
1754 if ((unsigned)Index != Index)
1755 return Error("Invalid EXTRACTVAL index");
1756 EXTRACTVALIdx.push_back((unsigned)Index);
Dan Gohmane4977cf2008-05-23 01:55:30 +00001757 }
1758
1759 I = ExtractValueInst::Create(Agg,
1760 EXTRACTVALIdx.begin(), EXTRACTVALIdx.end());
Devang Patele8e02132009-09-18 19:26:43 +00001761 InstructionList.push_back(I);
Dan Gohmane4977cf2008-05-23 01:55:30 +00001762 break;
1763 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001764
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001765 case bitc::FUNC_CODE_INST_INSERTVAL: {
1766 // INSERTVAL: [opty, opval, opty, opval, n x indices]
Dan Gohmane4977cf2008-05-23 01:55:30 +00001767 unsigned OpNum = 0;
1768 Value *Agg;
1769 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
1770 return Error("Invalid INSERTVAL record");
1771 Value *Val;
1772 if (getValueTypePair(Record, OpNum, NextValueNo, Val))
1773 return Error("Invalid INSERTVAL record");
1774
Dan Gohman81a0c0b2008-05-31 00:58:22 +00001775 SmallVector<unsigned, 4> INSERTVALIdx;
1776 for (unsigned RecSize = Record.size();
1777 OpNum != RecSize; ++OpNum) {
1778 uint64_t Index = Record[OpNum];
1779 if ((unsigned)Index != Index)
1780 return Error("Invalid INSERTVAL index");
1781 INSERTVALIdx.push_back((unsigned)Index);
Dan Gohmane4977cf2008-05-23 01:55:30 +00001782 }
1783
1784 I = InsertValueInst::Create(Agg, Val,
1785 INSERTVALIdx.begin(), INSERTVALIdx.end());
Devang Patele8e02132009-09-18 19:26:43 +00001786 InstructionList.push_back(I);
Dan Gohmane4977cf2008-05-23 01:55:30 +00001787 break;
1788 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001789
Chris Lattnerabfbf852007-05-06 00:21:25 +00001790 case bitc::FUNC_CODE_INST_SELECT: { // SELECT: [opval, ty, opval, opval]
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001791 // obsolete form of select
1792 // handles select i1 ... in old bitcode
Chris Lattnerabfbf852007-05-06 00:21:25 +00001793 unsigned OpNum = 0;
1794 Value *TrueVal, *FalseVal, *Cond;
1795 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1796 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
Owen Anderson1d0be152009-08-13 21:58:54 +00001797 getValue(Record, OpNum, Type::getInt1Ty(Context), Cond))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001798 return Error("Invalid SELECT record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001799
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001800 I = SelectInst::Create(Cond, TrueVal, FalseVal);
Devang Patele8e02132009-09-18 19:26:43 +00001801 InstructionList.push_back(I);
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001802 break;
1803 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001804
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001805 case bitc::FUNC_CODE_INST_VSELECT: {// VSELECT: [ty,opval,opval,predty,pred]
1806 // new form of select
1807 // handles select i1 or select [N x i1]
1808 unsigned OpNum = 0;
1809 Value *TrueVal, *FalseVal, *Cond;
1810 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
1811 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
1812 getValueTypePair(Record, OpNum, NextValueNo, Cond))
1813 return Error("Invalid SELECT record");
Dan Gohmanf72fb672008-09-09 01:02:47 +00001814
1815 // select condition can be either i1 or [N x i1]
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00001816 if (const VectorType* vector_type =
1817 dyn_cast<const VectorType>(Cond->getType())) {
Dan Gohmanf72fb672008-09-09 01:02:47 +00001818 // expect <n x i1>
Daniel Dunbara279bc32009-09-20 02:20:51 +00001819 if (vector_type->getElementType() != Type::getInt1Ty(Context))
Dan Gohmanf72fb672008-09-09 01:02:47 +00001820 return Error("Invalid SELECT condition type");
1821 } else {
1822 // expect i1
Daniel Dunbara279bc32009-09-20 02:20:51 +00001823 if (Cond->getType() != Type::getInt1Ty(Context))
Dan Gohmanf72fb672008-09-09 01:02:47 +00001824 return Error("Invalid SELECT condition type");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001825 }
1826
Gabor Greif051a9502008-04-06 20:25:17 +00001827 I = SelectInst::Create(Cond, TrueVal, FalseVal);
Devang Patele8e02132009-09-18 19:26:43 +00001828 InstructionList.push_back(I);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001829 break;
1830 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001831
Chris Lattner01ff65f2007-05-02 05:16:49 +00001832 case bitc::FUNC_CODE_INST_EXTRACTELT: { // EXTRACTELT: [opty, opval, opval]
Chris Lattnerabfbf852007-05-06 00:21:25 +00001833 unsigned OpNum = 0;
1834 Value *Vec, *Idx;
1835 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
Owen Anderson1d0be152009-08-13 21:58:54 +00001836 getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001837 return Error("Invalid EXTRACTELT record");
Eric Christophera3500da2009-07-25 02:28:41 +00001838 I = ExtractElementInst::Create(Vec, Idx);
Devang Patele8e02132009-09-18 19:26:43 +00001839 InstructionList.push_back(I);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001840 break;
1841 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001842
Chris Lattner01ff65f2007-05-02 05:16:49 +00001843 case bitc::FUNC_CODE_INST_INSERTELT: { // INSERTELT: [ty, opval,opval,opval]
Chris Lattnerabfbf852007-05-06 00:21:25 +00001844 unsigned OpNum = 0;
1845 Value *Vec, *Elt, *Idx;
1846 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
Daniel Dunbara279bc32009-09-20 02:20:51 +00001847 getValue(Record, OpNum,
Chris Lattnerabfbf852007-05-06 00:21:25 +00001848 cast<VectorType>(Vec->getType())->getElementType(), Elt) ||
Owen Anderson1d0be152009-08-13 21:58:54 +00001849 getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001850 return Error("Invalid INSERTELT record");
Gabor Greif051a9502008-04-06 20:25:17 +00001851 I = InsertElementInst::Create(Vec, Elt, Idx);
Devang Patele8e02132009-09-18 19:26:43 +00001852 InstructionList.push_back(I);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001853 break;
1854 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001855
Chris Lattnerabfbf852007-05-06 00:21:25 +00001856 case bitc::FUNC_CODE_INST_SHUFFLEVEC: {// SHUFFLEVEC: [opval,ty,opval,opval]
1857 unsigned OpNum = 0;
1858 Value *Vec1, *Vec2, *Mask;
1859 if (getValueTypePair(Record, OpNum, NextValueNo, Vec1) ||
1860 getValue(Record, OpNum, Vec1->getType(), Vec2))
1861 return Error("Invalid SHUFFLEVEC record");
1862
Mon P Wangaeb06d22008-11-10 04:46:22 +00001863 if (getValueTypePair(Record, OpNum, NextValueNo, Mask))
Chris Lattner01ff65f2007-05-02 05:16:49 +00001864 return Error("Invalid SHUFFLEVEC record");
1865 I = new ShuffleVectorInst(Vec1, Vec2, Mask);
Devang Patele8e02132009-09-18 19:26:43 +00001866 InstructionList.push_back(I);
Chris Lattner01ff65f2007-05-02 05:16:49 +00001867 break;
1868 }
Mon P Wangaeb06d22008-11-10 04:46:22 +00001869
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +00001870 case bitc::FUNC_CODE_INST_CMP: // CMP: [opty, opval, opval, pred]
1871 // Old form of ICmp/FCmp returning bool
1872 // Existed to differentiate between icmp/fcmp and vicmp/vfcmp which were
1873 // both legal on vectors but had different behaviour.
1874 case bitc::FUNC_CODE_INST_CMP2: { // CMP2: [opty, opval, opval, pred]
1875 // FCmp/ICmp returning bool or vector of bool
1876
Chris Lattner7337ab92007-05-06 00:00:00 +00001877 unsigned OpNum = 0;
1878 Value *LHS, *RHS;
1879 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
1880 getValue(Record, OpNum, LHS->getType(), RHS) ||
1881 OpNum+1 != Record.size())
Chris Lattner01ff65f2007-05-02 05:16:49 +00001882 return Error("Invalid CMP record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001883
Dan Gohmanf72fb672008-09-09 01:02:47 +00001884 if (LHS->getType()->isFPOrFPVector())
Dan Gohman1c8a23c2009-08-25 23:17:54 +00001885 I = new FCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +00001886 else
Dan Gohman1c8a23c2009-08-25 23:17:54 +00001887 I = new ICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
Devang Patele8e02132009-09-18 19:26:43 +00001888 InstructionList.push_back(I);
Dan Gohmanf72fb672008-09-09 01:02:47 +00001889 break;
1890 }
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +00001891
Devang Patel197be3d2008-02-22 02:49:49 +00001892 case bitc::FUNC_CODE_INST_GETRESULT: { // GETRESULT: [ty, val, n]
1893 if (Record.size() != 2)
1894 return Error("Invalid GETRESULT record");
1895 unsigned OpNum = 0;
1896 Value *Op;
1897 getValueTypePair(Record, OpNum, NextValueNo, Op);
1898 unsigned Index = Record[1];
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001899 I = ExtractValueInst::Create(Op, Index);
Devang Patele8e02132009-09-18 19:26:43 +00001900 InstructionList.push_back(I);
Devang Patel197be3d2008-02-22 02:49:49 +00001901 break;
1902 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001903
Chris Lattner231cbcb2007-05-02 04:27:25 +00001904 case bitc::FUNC_CODE_INST_RET: // RET: [opty,opval<optional>]
Devang Pateld9d99ff2008-02-26 01:29:32 +00001905 {
1906 unsigned Size = Record.size();
1907 if (Size == 0) {
Owen Anderson1d0be152009-08-13 21:58:54 +00001908 I = ReturnInst::Create(Context);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001909 InstructionList.push_back(I);
Devang Pateld9d99ff2008-02-26 01:29:32 +00001910 break;
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001911 }
Devang Pateld9d99ff2008-02-26 01:29:32 +00001912
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001913 unsigned OpNum = 0;
1914 SmallVector<Value *,4> Vs;
1915 do {
1916 Value *Op = NULL;
1917 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
1918 return Error("Invalid RET record");
1919 Vs.push_back(Op);
1920 } while(OpNum != Record.size());
1921
1922 const Type *ReturnType = F->getReturnType();
1923 if (Vs.size() > 1 ||
1924 (isa<StructType>(ReturnType) &&
1925 (Vs.empty() || Vs[0]->getType() != ReturnType))) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001926 Value *RV = UndefValue::get(ReturnType);
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001927 for (unsigned i = 0, e = Vs.size(); i != e; ++i) {
1928 I = InsertValueInst::Create(RV, Vs[i], i, "mrv");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001929 InstructionList.push_back(I);
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001930 CurBB->getInstList().push_back(I);
1931 ValueList.AssignValue(I, NextValueNo++);
1932 RV = I;
1933 }
Owen Anderson1d0be152009-08-13 21:58:54 +00001934 I = ReturnInst::Create(Context, RV);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001935 InstructionList.push_back(I);
Devang Pateld9d99ff2008-02-26 01:29:32 +00001936 break;
1937 }
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001938
Owen Anderson1d0be152009-08-13 21:58:54 +00001939 I = ReturnInst::Create(Context, Vs[0]);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001940 InstructionList.push_back(I);
Dan Gohmanfc74abf2008-07-23 00:34:11 +00001941 break;
Chris Lattner231cbcb2007-05-02 04:27:25 +00001942 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001943 case bitc::FUNC_CODE_INST_BR: { // BR: [bb#, bb#, opval] or [bb#]
Chris Lattnerf61e6452007-05-03 22:09:51 +00001944 if (Record.size() != 1 && Record.size() != 3)
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001945 return Error("Invalid BR record");
1946 BasicBlock *TrueDest = getBasicBlock(Record[0]);
1947 if (TrueDest == 0)
1948 return Error("Invalid BR record");
1949
Devang Patele8e02132009-09-18 19:26:43 +00001950 if (Record.size() == 1) {
Gabor Greif051a9502008-04-06 20:25:17 +00001951 I = BranchInst::Create(TrueDest);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001952 InstructionList.push_back(I);
Devang Patele8e02132009-09-18 19:26:43 +00001953 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001954 else {
1955 BasicBlock *FalseDest = getBasicBlock(Record[1]);
Owen Anderson1d0be152009-08-13 21:58:54 +00001956 Value *Cond = getFnValueByID(Record[2], Type::getInt1Ty(Context));
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001957 if (FalseDest == 0 || Cond == 0)
1958 return Error("Invalid BR record");
Gabor Greif051a9502008-04-06 20:25:17 +00001959 I = BranchInst::Create(TrueDest, FalseDest, Cond);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001960 InstructionList.push_back(I);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001961 }
1962 break;
1963 }
Chris Lattnerf9be95f2009-10-27 19:13:16 +00001964 case bitc::FUNC_CODE_INST_SWITCH: { // SWITCH: [opty, op0, op1, ...]
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001965 if (Record.size() < 3 || (Record.size() & 1) == 0)
1966 return Error("Invalid SWITCH record");
1967 const Type *OpTy = getTypeByID(Record[0]);
1968 Value *Cond = getFnValueByID(Record[1], OpTy);
1969 BasicBlock *Default = getBasicBlock(Record[2]);
1970 if (OpTy == 0 || Cond == 0 || Default == 0)
1971 return Error("Invalid SWITCH record");
1972 unsigned NumCases = (Record.size()-3)/2;
Gabor Greif051a9502008-04-06 20:25:17 +00001973 SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
Devang Patele8e02132009-09-18 19:26:43 +00001974 InstructionList.push_back(SI);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001975 for (unsigned i = 0, e = NumCases; i != e; ++i) {
Daniel Dunbara279bc32009-09-20 02:20:51 +00001976 ConstantInt *CaseVal =
Chris Lattnerf4c8e522007-05-02 05:46:45 +00001977 dyn_cast_or_null<ConstantInt>(getFnValueByID(Record[3+i*2], OpTy));
1978 BasicBlock *DestBB = getBasicBlock(Record[1+3+i*2]);
1979 if (CaseVal == 0 || DestBB == 0) {
1980 delete SI;
1981 return Error("Invalid SWITCH record!");
1982 }
1983 SI->addCase(CaseVal, DestBB);
1984 }
1985 I = SI;
1986 break;
1987 }
Chris Lattnerab21db72009-10-28 00:19:10 +00001988 case bitc::FUNC_CODE_INST_INDIRECTBR: { // INDIRECTBR: [opty, op0, op1, ...]
Chris Lattnerf9be95f2009-10-27 19:13:16 +00001989 if (Record.size() < 2)
Chris Lattnerab21db72009-10-28 00:19:10 +00001990 return Error("Invalid INDIRECTBR record");
Chris Lattnerf9be95f2009-10-27 19:13:16 +00001991 const Type *OpTy = getTypeByID(Record[0]);
1992 Value *Address = getFnValueByID(Record[1], OpTy);
1993 if (OpTy == 0 || Address == 0)
Chris Lattnerab21db72009-10-28 00:19:10 +00001994 return Error("Invalid INDIRECTBR record");
Chris Lattnerf9be95f2009-10-27 19:13:16 +00001995 unsigned NumDests = Record.size()-2;
Chris Lattnerab21db72009-10-28 00:19:10 +00001996 IndirectBrInst *IBI = IndirectBrInst::Create(Address, NumDests);
Chris Lattnerf9be95f2009-10-27 19:13:16 +00001997 InstructionList.push_back(IBI);
1998 for (unsigned i = 0, e = NumDests; i != e; ++i) {
1999 if (BasicBlock *DestBB = getBasicBlock(Record[2+i])) {
2000 IBI->addDestination(DestBB);
2001 } else {
2002 delete IBI;
Chris Lattnerab21db72009-10-28 00:19:10 +00002003 return Error("Invalid INDIRECTBR record!");
Chris Lattnerf9be95f2009-10-27 19:13:16 +00002004 }
2005 }
2006 I = IBI;
2007 break;
2008 }
2009
Duncan Sandsdc024672007-11-27 13:23:08 +00002010 case bitc::FUNC_CODE_INST_INVOKE: {
2011 // INVOKE: [attrs, cc, normBB, unwindBB, fnty, op0,op1,op2, ...]
Chris Lattnera9bb7132007-05-08 05:38:01 +00002012 if (Record.size() < 4) return Error("Invalid INVOKE record");
Devang Patel05988662008-09-25 21:00:45 +00002013 AttrListPtr PAL = getAttributes(Record[0]);
Chris Lattnera9bb7132007-05-08 05:38:01 +00002014 unsigned CCInfo = Record[1];
2015 BasicBlock *NormalBB = getBasicBlock(Record[2]);
2016 BasicBlock *UnwindBB = getBasicBlock(Record[3]);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002017
Chris Lattnera9bb7132007-05-08 05:38:01 +00002018 unsigned OpNum = 4;
Chris Lattner7337ab92007-05-06 00:00:00 +00002019 Value *Callee;
2020 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002021 return Error("Invalid INVOKE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002022
Chris Lattner7337ab92007-05-06 00:00:00 +00002023 const PointerType *CalleeTy = dyn_cast<PointerType>(Callee->getType());
2024 const FunctionType *FTy = !CalleeTy ? 0 :
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002025 dyn_cast<FunctionType>(CalleeTy->getElementType());
2026
2027 // Check that the right number of fixed parameters are here.
Chris Lattner7337ab92007-05-06 00:00:00 +00002028 if (FTy == 0 || NormalBB == 0 || UnwindBB == 0 ||
2029 Record.size() < OpNum+FTy->getNumParams())
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002030 return Error("Invalid INVOKE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002031
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002032 SmallVector<Value*, 16> Ops;
Chris Lattner7337ab92007-05-06 00:00:00 +00002033 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
2034 Ops.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
2035 if (Ops.back() == 0) return Error("Invalid INVOKE record");
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002036 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002037
Chris Lattner7337ab92007-05-06 00:00:00 +00002038 if (!FTy->isVarArg()) {
2039 if (Record.size() != OpNum)
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002040 return Error("Invalid INVOKE record");
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002041 } else {
Chris Lattner7337ab92007-05-06 00:00:00 +00002042 // Read type/value pairs for varargs params.
2043 while (OpNum != Record.size()) {
2044 Value *Op;
2045 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2046 return Error("Invalid INVOKE record");
2047 Ops.push_back(Op);
2048 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002049 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002050
Gabor Greifb1dbcd82008-05-15 10:04:30 +00002051 I = InvokeInst::Create(Callee, NormalBB, UnwindBB,
2052 Ops.begin(), Ops.end());
Devang Patele8e02132009-09-18 19:26:43 +00002053 InstructionList.push_back(I);
Sandeep Patel65c3c8f2009-09-02 08:44:58 +00002054 cast<InvokeInst>(I)->setCallingConv(
2055 static_cast<CallingConv::ID>(CCInfo));
Devang Patel05988662008-09-25 21:00:45 +00002056 cast<InvokeInst>(I)->setAttributes(PAL);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002057 break;
2058 }
Chris Lattner231cbcb2007-05-02 04:27:25 +00002059 case bitc::FUNC_CODE_INST_UNWIND: // UNWIND
Owen Anderson1d0be152009-08-13 21:58:54 +00002060 I = new UnwindInst(Context);
Devang Patele8e02132009-09-18 19:26:43 +00002061 InstructionList.push_back(I);
Chris Lattner231cbcb2007-05-02 04:27:25 +00002062 break;
2063 case bitc::FUNC_CODE_INST_UNREACHABLE: // UNREACHABLE
Owen Anderson1d0be152009-08-13 21:58:54 +00002064 I = new UnreachableInst(Context);
Devang Patele8e02132009-09-18 19:26:43 +00002065 InstructionList.push_back(I);
Chris Lattner231cbcb2007-05-02 04:27:25 +00002066 break;
Chris Lattnerabfbf852007-05-06 00:21:25 +00002067 case bitc::FUNC_CODE_INST_PHI: { // PHI: [ty, val0,bb0, ...]
Chris Lattner15e6d172007-05-04 19:11:41 +00002068 if (Record.size() < 1 || ((Record.size()-1)&1))
Chris Lattner2a98cca2007-05-03 18:58:09 +00002069 return Error("Invalid PHI record");
2070 const Type *Ty = getTypeByID(Record[0]);
2071 if (!Ty) return Error("Invalid PHI record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002072
Gabor Greif051a9502008-04-06 20:25:17 +00002073 PHINode *PN = PHINode::Create(Ty);
Devang Patele8e02132009-09-18 19:26:43 +00002074 InstructionList.push_back(PN);
Chris Lattner86941612008-04-13 00:14:42 +00002075 PN->reserveOperandSpace((Record.size()-1)/2);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002076
Chris Lattner15e6d172007-05-04 19:11:41 +00002077 for (unsigned i = 0, e = Record.size()-1; i != e; i += 2) {
2078 Value *V = getFnValueByID(Record[1+i], Ty);
2079 BasicBlock *BB = getBasicBlock(Record[2+i]);
Chris Lattner2a98cca2007-05-03 18:58:09 +00002080 if (!V || !BB) return Error("Invalid PHI record");
2081 PN->addIncoming(V, BB);
2082 }
2083 I = PN;
2084 break;
2085 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002086
Chris Lattner2a98cca2007-05-03 18:58:09 +00002087 case bitc::FUNC_CODE_INST_MALLOC: { // MALLOC: [instty, op, align]
Victor Hernandez13ad5aa2009-10-17 00:00:19 +00002088 // Autoupgrade malloc instruction to malloc call.
Victor Hernandez68afa542009-10-21 19:11:40 +00002089 // FIXME: Remove in LLVM 3.0.
Chris Lattner2a98cca2007-05-03 18:58:09 +00002090 if (Record.size() < 3)
2091 return Error("Invalid MALLOC record");
2092 const PointerType *Ty =
2093 dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
Owen Anderson1d0be152009-08-13 21:58:54 +00002094 Value *Size = getFnValueByID(Record[1], Type::getInt32Ty(Context));
Chris Lattner2a98cca2007-05-03 18:58:09 +00002095 if (!Ty || !Size) return Error("Invalid MALLOC record");
Victor Hernandez13ad5aa2009-10-17 00:00:19 +00002096 if (!CurBB) return Error("Invalid malloc instruction with no BB");
Victor Hernandez68afa542009-10-21 19:11:40 +00002097 const Type *Int32Ty = IntegerType::getInt32Ty(CurBB->getContext());
Victor Hernandez9d0b7042009-11-07 00:16:28 +00002098 Constant *AllocSize = ConstantExpr::getSizeOf(Ty->getElementType());
2099 AllocSize = ConstantExpr::getTruncOrBitCast(AllocSize, Int32Ty);
Victor Hernandez68afa542009-10-21 19:11:40 +00002100 I = CallInst::CreateMalloc(CurBB, Int32Ty, Ty->getElementType(),
Victor Hernandez9d0b7042009-11-07 00:16:28 +00002101 AllocSize, Size, NULL);
Devang Patele8e02132009-09-18 19:26:43 +00002102 InstructionList.push_back(I);
Chris Lattner2a98cca2007-05-03 18:58:09 +00002103 break;
2104 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00002105 case bitc::FUNC_CODE_INST_FREE: { // FREE: [op, opty]
2106 unsigned OpNum = 0;
2107 Value *Op;
2108 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2109 OpNum != Record.size())
Chris Lattner2a98cca2007-05-03 18:58:09 +00002110 return Error("Invalid FREE record");
Victor Hernandez66284e02009-10-24 04:23:03 +00002111 if (!CurBB) return Error("Invalid free instruction with no BB");
2112 I = CallInst::CreateFree(Op, CurBB);
Devang Patele8e02132009-09-18 19:26:43 +00002113 InstructionList.push_back(I);
Chris Lattner2a98cca2007-05-03 18:58:09 +00002114 break;
2115 }
2116 case bitc::FUNC_CODE_INST_ALLOCA: { // ALLOCA: [instty, op, align]
2117 if (Record.size() < 3)
2118 return Error("Invalid ALLOCA record");
2119 const PointerType *Ty =
2120 dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
Owen Anderson1d0be152009-08-13 21:58:54 +00002121 Value *Size = getFnValueByID(Record[1], Type::getInt32Ty(Context));
Chris Lattner2a98cca2007-05-03 18:58:09 +00002122 unsigned Align = Record[2];
2123 if (!Ty || !Size) return Error("Invalid ALLOCA record");
Owen Anderson50dead02009-07-15 23:53:25 +00002124 I = new AllocaInst(Ty->getElementType(), Size, (1 << Align) >> 1);
Devang Patele8e02132009-09-18 19:26:43 +00002125 InstructionList.push_back(I);
Chris Lattner2a98cca2007-05-03 18:58:09 +00002126 break;
2127 }
Chris Lattner0579f7f2007-05-03 22:04:19 +00002128 case bitc::FUNC_CODE_INST_LOAD: { // LOAD: [opty, op, align, vol]
Chris Lattner7337ab92007-05-06 00:00:00 +00002129 unsigned OpNum = 0;
2130 Value *Op;
2131 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2132 OpNum+2 != Record.size())
Chris Lattnerabfbf852007-05-06 00:21:25 +00002133 return Error("Invalid LOAD record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002134
Chris Lattner7337ab92007-05-06 00:00:00 +00002135 I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Devang Patele8e02132009-09-18 19:26:43 +00002136 InstructionList.push_back(I);
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002137 break;
Chris Lattner0579f7f2007-05-03 22:04:19 +00002138 }
Christopher Lambfe63fb92007-12-11 08:59:05 +00002139 case bitc::FUNC_CODE_INST_STORE2: { // STORE2:[ptrty, ptr, val, align, vol]
2140 unsigned OpNum = 0;
2141 Value *Val, *Ptr;
2142 if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
Daniel Dunbara279bc32009-09-20 02:20:51 +00002143 getValue(Record, OpNum,
Christopher Lambfe63fb92007-12-11 08:59:05 +00002144 cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
2145 OpNum+2 != Record.size())
2146 return Error("Invalid STORE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002147
Christopher Lambfe63fb92007-12-11 08:59:05 +00002148 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Devang Patele8e02132009-09-18 19:26:43 +00002149 InstructionList.push_back(I);
Christopher Lambfe63fb92007-12-11 08:59:05 +00002150 break;
2151 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00002152 case bitc::FUNC_CODE_INST_STORE: { // STORE:[val, valty, ptr, align, vol]
Christopher Lambfe63fb92007-12-11 08:59:05 +00002153 // FIXME: Legacy form of store instruction. Should be removed in LLVM 3.0.
Chris Lattnerabfbf852007-05-06 00:21:25 +00002154 unsigned OpNum = 0;
2155 Value *Val, *Ptr;
2156 if (getValueTypePair(Record, OpNum, NextValueNo, Val) ||
Daniel Dunbara279bc32009-09-20 02:20:51 +00002157 getValue(Record, OpNum,
Owen Andersondebcb012009-07-29 22:17:13 +00002158 PointerType::getUnqual(Val->getType()), Ptr)||
Chris Lattnerabfbf852007-05-06 00:21:25 +00002159 OpNum+2 != Record.size())
Chris Lattner0579f7f2007-05-03 22:04:19 +00002160 return Error("Invalid STORE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002161
Chris Lattnerabfbf852007-05-06 00:21:25 +00002162 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Devang Patele8e02132009-09-18 19:26:43 +00002163 InstructionList.push_back(I);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002164 break;
2165 }
Duncan Sandsdc024672007-11-27 13:23:08 +00002166 case bitc::FUNC_CODE_INST_CALL: {
2167 // CALL: [paramattrs, cc, fnty, fnid, arg0, arg1...]
2168 if (Record.size() < 3)
Chris Lattner0579f7f2007-05-03 22:04:19 +00002169 return Error("Invalid CALL record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002170
Devang Patel05988662008-09-25 21:00:45 +00002171 AttrListPtr PAL = getAttributes(Record[0]);
Chris Lattnera9bb7132007-05-08 05:38:01 +00002172 unsigned CCInfo = Record[1];
Daniel Dunbara279bc32009-09-20 02:20:51 +00002173
Chris Lattnera9bb7132007-05-08 05:38:01 +00002174 unsigned OpNum = 2;
Chris Lattner7337ab92007-05-06 00:00:00 +00002175 Value *Callee;
2176 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
2177 return Error("Invalid CALL record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002178
Chris Lattner7337ab92007-05-06 00:00:00 +00002179 const PointerType *OpTy = dyn_cast<PointerType>(Callee->getType());
Chris Lattner0579f7f2007-05-03 22:04:19 +00002180 const FunctionType *FTy = 0;
2181 if (OpTy) FTy = dyn_cast<FunctionType>(OpTy->getElementType());
Chris Lattner7337ab92007-05-06 00:00:00 +00002182 if (!FTy || Record.size() < FTy->getNumParams()+OpNum)
Chris Lattner0579f7f2007-05-03 22:04:19 +00002183 return Error("Invalid CALL record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002184
Chris Lattner0579f7f2007-05-03 22:04:19 +00002185 SmallVector<Value*, 16> Args;
2186 // Read the fixed params.
Chris Lattner7337ab92007-05-06 00:00:00 +00002187 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
Dale Johanneseneb57ea72007-11-05 21:20:28 +00002188 if (FTy->getParamType(i)->getTypeID()==Type::LabelTyID)
2189 Args.push_back(getBasicBlock(Record[OpNum]));
2190 else
2191 Args.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
Chris Lattner0579f7f2007-05-03 22:04:19 +00002192 if (Args.back() == 0) return Error("Invalid CALL record");
2193 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002194
Chris Lattner0579f7f2007-05-03 22:04:19 +00002195 // Read type/value pairs for varargs params.
Chris Lattner0579f7f2007-05-03 22:04:19 +00002196 if (!FTy->isVarArg()) {
Chris Lattner7337ab92007-05-06 00:00:00 +00002197 if (OpNum != Record.size())
Chris Lattner0579f7f2007-05-03 22:04:19 +00002198 return Error("Invalid CALL record");
2199 } else {
Chris Lattner7337ab92007-05-06 00:00:00 +00002200 while (OpNum != Record.size()) {
2201 Value *Op;
2202 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2203 return Error("Invalid CALL record");
2204 Args.push_back(Op);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002205 }
2206 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002207
Gabor Greif051a9502008-04-06 20:25:17 +00002208 I = CallInst::Create(Callee, Args.begin(), Args.end());
Devang Patele8e02132009-09-18 19:26:43 +00002209 InstructionList.push_back(I);
Sandeep Patel65c3c8f2009-09-02 08:44:58 +00002210 cast<CallInst>(I)->setCallingConv(
2211 static_cast<CallingConv::ID>(CCInfo>>1));
Chris Lattner76520192007-05-03 22:34:03 +00002212 cast<CallInst>(I)->setTailCall(CCInfo & 1);
Devang Patel05988662008-09-25 21:00:45 +00002213 cast<CallInst>(I)->setAttributes(PAL);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002214 break;
2215 }
2216 case bitc::FUNC_CODE_INST_VAARG: { // VAARG: [valistty, valist, instty]
2217 if (Record.size() < 3)
2218 return Error("Invalid VAARG record");
2219 const Type *OpTy = getTypeByID(Record[0]);
2220 Value *Op = getFnValueByID(Record[1], OpTy);
2221 const Type *ResTy = getTypeByID(Record[2]);
2222 if (!OpTy || !Op || !ResTy)
2223 return Error("Invalid VAARG record");
2224 I = new VAArgInst(Op, ResTy);
Devang Patele8e02132009-09-18 19:26:43 +00002225 InstructionList.push_back(I);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002226 break;
2227 }
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002228 }
2229
2230 // Add instruction to end of current BB. If there is no current BB, reject
2231 // this file.
2232 if (CurBB == 0) {
2233 delete I;
2234 return Error("Invalid instruction with no BB");
2235 }
2236 CurBB->getInstList().push_back(I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002237
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002238 // If this was a terminator instruction, move to the next block.
2239 if (isa<TerminatorInst>(I)) {
2240 ++CurBBNo;
2241 CurBB = CurBBNo < FunctionBBs.size() ? FunctionBBs[CurBBNo] : 0;
2242 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002243
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002244 // Non-void values get registered in the value table for future use.
Benjamin Kramerf0127052010-01-05 13:12:22 +00002245 if (I && !I->getType()->isVoidTy())
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002246 ValueList.AssignValue(I, NextValueNo++);
Chris Lattner980e5aa2007-05-01 05:52:21 +00002247 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002248
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002249 // Check the function list for unresolved values.
2250 if (Argument *A = dyn_cast<Argument>(ValueList.back())) {
2251 if (A->getParent() == 0) {
2252 // We found at least one unresolved value. Nuke them all to avoid leaks.
2253 for (unsigned i = ModuleValueListSize, e = ValueList.size(); i != e; ++i){
2254 if ((A = dyn_cast<Argument>(ValueList.back())) && A->getParent() == 0) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00002255 A->replaceAllUsesWith(UndefValue::get(A->getType()));
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002256 delete A;
2257 }
2258 }
Chris Lattner35a04702007-05-04 03:50:29 +00002259 return Error("Never resolved value found in function!");
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002260 }
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002261 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002262
Chris Lattner50b136d2009-10-28 05:53:48 +00002263 // See if anything took the address of blocks in this function. If so,
2264 // resolve them now.
2265 /// BlockAddrFwdRefs - These are blockaddr references to basic blocks. These
2266 /// are resolved lazily when functions are loaded.
2267 DenseMap<Function*, std::vector<BlockAddrRefTy> >::iterator BAFRI =
2268 BlockAddrFwdRefs.find(F);
2269 if (BAFRI != BlockAddrFwdRefs.end()) {
2270 std::vector<BlockAddrRefTy> &RefList = BAFRI->second;
2271 for (unsigned i = 0, e = RefList.size(); i != e; ++i) {
2272 unsigned BlockIdx = RefList[i].first;
Chris Lattnercdfc9402009-11-01 01:27:45 +00002273 if (BlockIdx >= FunctionBBs.size())
Chris Lattner50b136d2009-10-28 05:53:48 +00002274 return Error("Invalid blockaddress block #");
2275
2276 GlobalVariable *FwdRef = RefList[i].second;
Chris Lattnercdfc9402009-11-01 01:27:45 +00002277 FwdRef->replaceAllUsesWith(BlockAddress::get(F, FunctionBBs[BlockIdx]));
Chris Lattner50b136d2009-10-28 05:53:48 +00002278 FwdRef->eraseFromParent();
2279 }
2280
2281 BlockAddrFwdRefs.erase(BAFRI);
2282 }
2283
Chris Lattner980e5aa2007-05-01 05:52:21 +00002284 // Trim the value list down to the size it was before we parsed this function.
2285 ValueList.shrinkTo(ModuleValueListSize);
2286 std::vector<BasicBlock*>().swap(FunctionBBs);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002287
Chris Lattner48f84872007-05-01 04:59:48 +00002288 return false;
2289}
2290
Chris Lattnerb348bb82007-05-18 04:02:46 +00002291//===----------------------------------------------------------------------===//
2292// ModuleProvider implementation
2293//===----------------------------------------------------------------------===//
2294
2295
2296bool BitcodeReader::materializeFunction(Function *F, std::string *ErrInfo) {
2297 // If it already is material, ignore the request.
Gabor Greifa99be512007-07-05 17:07:56 +00002298 if (!F->hasNotBeenReadFromBitcode()) return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002299
2300 DenseMap<Function*, std::pair<uint64_t, unsigned> >::iterator DFII =
Chris Lattnerb348bb82007-05-18 04:02:46 +00002301 DeferredFunctionInfo.find(F);
2302 assert(DFII != DeferredFunctionInfo.end() && "Deferred function not found!");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002303
Chris Lattnerb348bb82007-05-18 04:02:46 +00002304 // Move the bit stream to the saved position of the deferred function body and
2305 // restore the real linkage type for the function.
2306 Stream.JumpToBit(DFII->second.first);
2307 F->setLinkage((GlobalValue::LinkageTypes)DFII->second.second);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002308
Chris Lattnerb348bb82007-05-18 04:02:46 +00002309 if (ParseFunctionBody(F)) {
2310 if (ErrInfo) *ErrInfo = ErrorString;
2311 return true;
2312 }
Chandler Carruth69940402007-08-04 01:51:18 +00002313
2314 // Upgrade any old intrinsic calls in the function.
2315 for (UpgradedIntrinsicMap::iterator I = UpgradedIntrinsics.begin(),
2316 E = UpgradedIntrinsics.end(); I != E; ++I) {
2317 if (I->first != I->second) {
2318 for (Value::use_iterator UI = I->first->use_begin(),
2319 UE = I->first->use_end(); UI != UE; ) {
2320 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2321 UpgradeIntrinsicCall(CI, I->second);
2322 }
2323 }
2324 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002325
Chris Lattnerb348bb82007-05-18 04:02:46 +00002326 return false;
2327}
2328
2329void BitcodeReader::dematerializeFunction(Function *F) {
2330 // If this function isn't materialized, or if it is a proto, this is a noop.
Gabor Greifa99be512007-07-05 17:07:56 +00002331 if (F->hasNotBeenReadFromBitcode() || F->isDeclaration())
Chris Lattnerb348bb82007-05-18 04:02:46 +00002332 return;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002333
Chris Lattnerb348bb82007-05-18 04:02:46 +00002334 assert(DeferredFunctionInfo.count(F) && "No info to read function later?");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002335
Chris Lattnerb348bb82007-05-18 04:02:46 +00002336 // Just forget the function body, we can remat it later.
2337 F->deleteBody();
2338 F->setLinkage(GlobalValue::GhostLinkage);
2339}
2340
2341
2342Module *BitcodeReader::materializeModule(std::string *ErrInfo) {
Chris Lattner714fa952009-06-16 05:15:21 +00002343 // Iterate over the module, deserializing any functions that are still on
2344 // disk.
2345 for (Module::iterator F = TheModule->begin(), E = TheModule->end();
2346 F != E; ++F)
Gabor Greifa99be512007-07-05 17:07:56 +00002347 if (F->hasNotBeenReadFromBitcode() &&
Chris Lattnerb348bb82007-05-18 04:02:46 +00002348 materializeFunction(F, ErrInfo))
2349 return 0;
Chandler Carruth69940402007-08-04 01:51:18 +00002350
Daniel Dunbara279bc32009-09-20 02:20:51 +00002351 // Upgrade any intrinsic calls that slipped through (should not happen!) and
2352 // delete the old functions to clean up. We can't do this unless the entire
2353 // module is materialized because there could always be another function body
Chandler Carruth69940402007-08-04 01:51:18 +00002354 // with calls to the old function.
2355 for (std::vector<std::pair<Function*, Function*> >::iterator I =
2356 UpgradedIntrinsics.begin(), E = UpgradedIntrinsics.end(); I != E; ++I) {
2357 if (I->first != I->second) {
2358 for (Value::use_iterator UI = I->first->use_begin(),
2359 UE = I->first->use_end(); UI != UE; ) {
2360 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2361 UpgradeIntrinsicCall(CI, I->second);
2362 }
Chris Lattner7d9eb582009-04-01 01:43:03 +00002363 if (!I->first->use_empty())
2364 I->first->replaceAllUsesWith(I->second);
Chandler Carruth69940402007-08-04 01:51:18 +00002365 I->first->eraseFromParent();
2366 }
2367 }
2368 std::vector<std::pair<Function*, Function*> >().swap(UpgradedIntrinsics);
Devang Patele4b27562009-08-28 23:24:31 +00002369
2370 // Check debug info intrinsics.
2371 CheckDebugInfoIntrinsics(TheModule);
2372
Chris Lattnerb348bb82007-05-18 04:02:46 +00002373 return TheModule;
2374}
2375
2376
2377/// This method is provided by the parent ModuleProvde class and overriden
2378/// here. It simply releases the module from its provided and frees up our
2379/// state.
2380/// @brief Release our hold on the generated module
2381Module *BitcodeReader::releaseModule(std::string *ErrInfo) {
2382 // Since we're losing control of this Module, we must hand it back complete
2383 Module *M = ModuleProvider::releaseModule(ErrInfo);
2384 FreeState();
2385 return M;
2386}
2387
Chris Lattner48f84872007-05-01 04:59:48 +00002388
Chris Lattnerc453f762007-04-29 07:54:31 +00002389//===----------------------------------------------------------------------===//
2390// External interface
2391//===----------------------------------------------------------------------===//
2392
2393/// getBitcodeModuleProvider - lazy function-at-a-time loading from a file.
2394///
2395ModuleProvider *llvm::getBitcodeModuleProvider(MemoryBuffer *Buffer,
Owen Anderson4434ed42009-07-01 23:13:44 +00002396 LLVMContext& Context,
Chris Lattnerc453f762007-04-29 07:54:31 +00002397 std::string *ErrMsg) {
Owen Anderson8b477ed2009-07-01 16:58:40 +00002398 BitcodeReader *R = new BitcodeReader(Buffer, Context);
Chris Lattnerc453f762007-04-29 07:54:31 +00002399 if (R->ParseBitcode()) {
2400 if (ErrMsg)
2401 *ErrMsg = R->getErrorString();
Daniel Dunbara279bc32009-09-20 02:20:51 +00002402
Chris Lattnerc453f762007-04-29 07:54:31 +00002403 // Don't let the BitcodeReader dtor delete 'Buffer'.
2404 R->releaseMemoryBuffer();
2405 delete R;
2406 return 0;
2407 }
2408 return R;
2409}
2410
2411/// ParseBitcodeFile - Read the specified bitcode file, returning the module.
2412/// If an error occurs, return null and fill in *ErrMsg if non-null.
Daniel Dunbara279bc32009-09-20 02:20:51 +00002413Module *llvm::ParseBitcodeFile(MemoryBuffer *Buffer, LLVMContext& Context,
Owen Anderson8b477ed2009-07-01 16:58:40 +00002414 std::string *ErrMsg){
Chris Lattnerc453f762007-04-29 07:54:31 +00002415 BitcodeReader *R;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002416 R = static_cast<BitcodeReader*>(getBitcodeModuleProvider(Buffer, Context,
Owen Anderson8b477ed2009-07-01 16:58:40 +00002417 ErrMsg));
Chris Lattnerc453f762007-04-29 07:54:31 +00002418 if (!R) return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002419
Chris Lattnerb348bb82007-05-18 04:02:46 +00002420 // Read in the entire module.
2421 Module *M = R->materializeModule(ErrMsg);
2422
2423 // Don't let the BitcodeReader dtor delete 'Buffer', regardless of whether
2424 // there was an error.
Chris Lattnerc453f762007-04-29 07:54:31 +00002425 R->releaseMemoryBuffer();
Daniel Dunbara279bc32009-09-20 02:20:51 +00002426
Chris Lattnerb348bb82007-05-18 04:02:46 +00002427 // If there was no error, tell ModuleProvider not to delete it when its dtor
2428 // is run.
2429 if (M)
2430 M = R->releaseModule(ErrMsg);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002431
Chris Lattnerc453f762007-04-29 07:54:31 +00002432 delete R;
2433 return M;
2434}