blob: 0ff4664b2e4c432f822d9b0ed8e3beafde8ce1f6 [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"
Derek Schuff2ea93872012-02-06 22:30:29 +000025#include "llvm/Support/DataStream.h"
Chris Lattner0eef0802007-04-24 04:04:35 +000026#include "llvm/Support/MathExtras.h"
Chris Lattnerc453f762007-04-29 07:54:31 +000027#include "llvm/Support/MemoryBuffer.h"
Gabor Greifefe65362008-05-10 08:32:32 +000028#include "llvm/OperandTraits.h"
Chris Lattnercaee0dc2007-04-22 06:23:29 +000029using namespace llvm;
30
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +000031enum {
32 SWITCH_INST_MAGIC = 0x4B5 // May 2012 => 1205 => Hex
33};
34
Rafael Espindola47f79bb2012-01-02 07:49:53 +000035void BitcodeReader::materializeForwardReferencedFunctions() {
36 while (!BlockAddrFwdRefs.empty()) {
37 Function *F = BlockAddrFwdRefs.begin()->first;
38 F->Materialize();
39 }
40}
41
Chris Lattnerb348bb82007-05-18 04:02:46 +000042void BitcodeReader::FreeState() {
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +000043 if (BufferOwned)
44 delete Buffer;
Chris Lattnerb348bb82007-05-18 04:02:46 +000045 Buffer = 0;
Chris Lattner1afcace2011-07-09 17:41:24 +000046 std::vector<Type*>().swap(TypeList);
Chris Lattnerb348bb82007-05-18 04:02:46 +000047 ValueList.clear();
Devang Pateld5ac4042009-08-04 06:00:18 +000048 MDValueList.clear();
Daniel Dunbara279bc32009-09-20 02:20:51 +000049
Devang Patel19c87462008-09-26 22:53:05 +000050 std::vector<AttrListPtr>().swap(MAttributes);
Chris Lattnerb348bb82007-05-18 04:02:46 +000051 std::vector<BasicBlock*>().swap(FunctionBBs);
52 std::vector<Function*>().swap(FunctionsWithBodies);
53 DeferredFunctionInfo.clear();
Dan Gohman19538d12010-07-20 21:42:28 +000054 MDKindMap.clear();
Chris Lattnerc453f762007-04-29 07:54:31 +000055}
56
Chris Lattner48c85b82007-05-04 03:30:17 +000057//===----------------------------------------------------------------------===//
58// Helper functions to implement forward reference resolution, etc.
59//===----------------------------------------------------------------------===//
Chris Lattnerc453f762007-04-29 07:54:31 +000060
Chris Lattnercaee0dc2007-04-22 06:23:29 +000061/// ConvertToString - Convert a string from a record into an std::string, return
62/// true on failure.
Chris Lattner0b2482a2007-04-23 21:26:05 +000063template<typename StrTy>
Benjamin Kramerf52aea82012-05-28 14:10:31 +000064static bool ConvertToString(ArrayRef<uint64_t> Record, unsigned Idx,
Chris Lattner0b2482a2007-04-23 21:26:05 +000065 StrTy &Result) {
Chris Lattner15e6d172007-05-04 19:11:41 +000066 if (Idx > Record.size())
Chris Lattnercaee0dc2007-04-22 06:23:29 +000067 return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +000068
Chris Lattner15e6d172007-05-04 19:11:41 +000069 for (unsigned i = Idx, e = Record.size(); i != e; ++i)
70 Result += (char)Record[i];
Chris Lattnercaee0dc2007-04-22 06:23:29 +000071 return false;
72}
73
74static GlobalValue::LinkageTypes GetDecodedLinkage(unsigned Val) {
75 switch (Val) {
76 default: // Map unknown/new linkages to external
Bill Wendling3d10a5a2009-07-20 01:03:30 +000077 case 0: return GlobalValue::ExternalLinkage;
78 case 1: return GlobalValue::WeakAnyLinkage;
79 case 2: return GlobalValue::AppendingLinkage;
80 case 3: return GlobalValue::InternalLinkage;
81 case 4: return GlobalValue::LinkOnceAnyLinkage;
82 case 5: return GlobalValue::DLLImportLinkage;
83 case 6: return GlobalValue::DLLExportLinkage;
84 case 7: return GlobalValue::ExternalWeakLinkage;
85 case 8: return GlobalValue::CommonLinkage;
86 case 9: return GlobalValue::PrivateLinkage;
Duncan Sands667d4b82009-03-07 15:45:40 +000087 case 10: return GlobalValue::WeakODRLinkage;
88 case 11: return GlobalValue::LinkOnceODRLinkage;
Chris Lattner266c7bb2009-04-13 05:44:34 +000089 case 12: return GlobalValue::AvailableExternallyLinkage;
Bill Wendling3d10a5a2009-07-20 01:03:30 +000090 case 13: return GlobalValue::LinkerPrivateLinkage;
Bill Wendling5e721d72010-07-01 21:55:59 +000091 case 14: return GlobalValue::LinkerPrivateWeakLinkage;
Bill Wendling32811be2012-08-17 18:33:14 +000092 case 15: return GlobalValue::LinkOnceODRAutoHideLinkage;
Chris Lattnercaee0dc2007-04-22 06:23:29 +000093 }
94}
95
96static GlobalValue::VisibilityTypes GetDecodedVisibility(unsigned Val) {
97 switch (Val) {
98 default: // Map unknown visibilities to default.
99 case 0: return GlobalValue::DefaultVisibility;
100 case 1: return GlobalValue::HiddenVisibility;
Anton Korobeynikov9cd3ccf2007-04-29 20:56:48 +0000101 case 2: return GlobalValue::ProtectedVisibility;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000102 }
103}
104
Hans Wennborgce718ff2012-06-23 11:37:03 +0000105static GlobalVariable::ThreadLocalMode GetDecodedThreadLocalMode(unsigned Val) {
106 switch (Val) {
107 case 0: return GlobalVariable::NotThreadLocal;
108 default: // Map unknown non-zero value to general dynamic.
109 case 1: return GlobalVariable::GeneralDynamicTLSModel;
110 case 2: return GlobalVariable::LocalDynamicTLSModel;
111 case 3: return GlobalVariable::InitialExecTLSModel;
112 case 4: return GlobalVariable::LocalExecTLSModel;
113 }
114}
115
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000116static int GetDecodedCastOpcode(unsigned Val) {
117 switch (Val) {
118 default: return -1;
119 case bitc::CAST_TRUNC : return Instruction::Trunc;
120 case bitc::CAST_ZEXT : return Instruction::ZExt;
121 case bitc::CAST_SEXT : return Instruction::SExt;
122 case bitc::CAST_FPTOUI : return Instruction::FPToUI;
123 case bitc::CAST_FPTOSI : return Instruction::FPToSI;
124 case bitc::CAST_UITOFP : return Instruction::UIToFP;
125 case bitc::CAST_SITOFP : return Instruction::SIToFP;
126 case bitc::CAST_FPTRUNC : return Instruction::FPTrunc;
127 case bitc::CAST_FPEXT : return Instruction::FPExt;
128 case bitc::CAST_PTRTOINT: return Instruction::PtrToInt;
129 case bitc::CAST_INTTOPTR: return Instruction::IntToPtr;
130 case bitc::CAST_BITCAST : return Instruction::BitCast;
131 }
132}
Chris Lattnerdb125cf2011-07-18 04:54:35 +0000133static int GetDecodedBinaryOpcode(unsigned Val, Type *Ty) {
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000134 switch (Val) {
135 default: return -1;
Dan Gohmanae3a0be2009-06-04 22:49:04 +0000136 case bitc::BINOP_ADD:
Duncan Sandsb0bc6c32010-02-15 16:12:20 +0000137 return Ty->isFPOrFPVectorTy() ? Instruction::FAdd : Instruction::Add;
Dan Gohmanae3a0be2009-06-04 22:49:04 +0000138 case bitc::BINOP_SUB:
Duncan Sandsb0bc6c32010-02-15 16:12:20 +0000139 return Ty->isFPOrFPVectorTy() ? Instruction::FSub : Instruction::Sub;
Dan Gohmanae3a0be2009-06-04 22:49:04 +0000140 case bitc::BINOP_MUL:
Duncan Sandsb0bc6c32010-02-15 16:12:20 +0000141 return Ty->isFPOrFPVectorTy() ? Instruction::FMul : Instruction::Mul;
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000142 case bitc::BINOP_UDIV: return Instruction::UDiv;
143 case bitc::BINOP_SDIV:
Duncan Sandsb0bc6c32010-02-15 16:12:20 +0000144 return Ty->isFPOrFPVectorTy() ? Instruction::FDiv : Instruction::SDiv;
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000145 case bitc::BINOP_UREM: return Instruction::URem;
146 case bitc::BINOP_SREM:
Duncan Sandsb0bc6c32010-02-15 16:12:20 +0000147 return Ty->isFPOrFPVectorTy() ? Instruction::FRem : Instruction::SRem;
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000148 case bitc::BINOP_SHL: return Instruction::Shl;
149 case bitc::BINOP_LSHR: return Instruction::LShr;
150 case bitc::BINOP_ASHR: return Instruction::AShr;
151 case bitc::BINOP_AND: return Instruction::And;
152 case bitc::BINOP_OR: return Instruction::Or;
153 case bitc::BINOP_XOR: return Instruction::Xor;
154 }
155}
156
Eli Friedmanff030482011-07-28 21:48:00 +0000157static AtomicRMWInst::BinOp GetDecodedRMWOperation(unsigned Val) {
158 switch (Val) {
159 default: return AtomicRMWInst::BAD_BINOP;
160 case bitc::RMW_XCHG: return AtomicRMWInst::Xchg;
161 case bitc::RMW_ADD: return AtomicRMWInst::Add;
162 case bitc::RMW_SUB: return AtomicRMWInst::Sub;
163 case bitc::RMW_AND: return AtomicRMWInst::And;
164 case bitc::RMW_NAND: return AtomicRMWInst::Nand;
165 case bitc::RMW_OR: return AtomicRMWInst::Or;
166 case bitc::RMW_XOR: return AtomicRMWInst::Xor;
167 case bitc::RMW_MAX: return AtomicRMWInst::Max;
168 case bitc::RMW_MIN: return AtomicRMWInst::Min;
169 case bitc::RMW_UMAX: return AtomicRMWInst::UMax;
170 case bitc::RMW_UMIN: return AtomicRMWInst::UMin;
171 }
172}
173
Eli Friedman47f35132011-07-25 23:16:38 +0000174static AtomicOrdering GetDecodedOrdering(unsigned Val) {
175 switch (Val) {
176 case bitc::ORDERING_NOTATOMIC: return NotAtomic;
177 case bitc::ORDERING_UNORDERED: return Unordered;
178 case bitc::ORDERING_MONOTONIC: return Monotonic;
179 case bitc::ORDERING_ACQUIRE: return Acquire;
180 case bitc::ORDERING_RELEASE: return Release;
181 case bitc::ORDERING_ACQREL: return AcquireRelease;
182 default: // Map unknown orderings to sequentially-consistent.
183 case bitc::ORDERING_SEQCST: return SequentiallyConsistent;
184 }
185}
186
187static SynchronizationScope GetDecodedSynchScope(unsigned Val) {
188 switch (Val) {
189 case bitc::SYNCHSCOPE_SINGLETHREAD: return SingleThread;
190 default: // Map unknown scopes to cross-thread.
191 case bitc::SYNCHSCOPE_CROSSTHREAD: return CrossThread;
192 }
193}
194
Gabor Greifefe65362008-05-10 08:32:32 +0000195namespace llvm {
Chris Lattner522b7b12007-04-24 05:48:56 +0000196namespace {
197 /// @brief A class for maintaining the slot number definition
198 /// as a placeholder for the actual definition for forward constants defs.
199 class ConstantPlaceHolder : public ConstantExpr {
Craig Topper86a1c322012-09-15 17:09:36 +0000200 void operator=(const ConstantPlaceHolder &) LLVM_DELETED_FUNCTION;
Gabor Greif051a9502008-04-06 20:25:17 +0000201 public:
202 // allocate space for exactly one operand
203 void *operator new(size_t s) {
204 return User::operator new(s, 1);
205 }
Chris Lattnerdb125cf2011-07-18 04:54:35 +0000206 explicit ConstantPlaceHolder(Type *Ty, LLVMContext& Context)
Gabor Greifefe65362008-05-10 08:32:32 +0000207 : ConstantExpr(Ty, Instruction::UserOp1, &Op<0>(), 1) {
Owen Anderson1d0be152009-08-13 21:58:54 +0000208 Op<0>() = UndefValue::get(Type::getInt32Ty(Context));
Chris Lattner522b7b12007-04-24 05:48:56 +0000209 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000210
Chris Lattnerea693df2008-08-21 02:34:16 +0000211 /// @brief Methods to support type inquiry through isa, cast, and dyn_cast.
Chris Lattner17aa6802010-09-04 18:12:00 +0000212 //static inline bool classof(const ConstantPlaceHolder *) { return true; }
Chris Lattnerea693df2008-08-21 02:34:16 +0000213 static bool classof(const Value *V) {
Daniel Dunbara279bc32009-09-20 02:20:51 +0000214 return isa<ConstantExpr>(V) &&
Chris Lattnerea693df2008-08-21 02:34:16 +0000215 cast<ConstantExpr>(V)->getOpcode() == Instruction::UserOp1;
216 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000217
218
Gabor Greifefe65362008-05-10 08:32:32 +0000219 /// Provide fast operand accessors
Chris Lattner46e77402009-03-31 22:55:09 +0000220 //DECLARE_TRANSPARENT_OPERAND_ACCESSORS(Value);
Chris Lattner522b7b12007-04-24 05:48:56 +0000221 };
222}
223
Chris Lattner46e77402009-03-31 22:55:09 +0000224// FIXME: can we inherit this from ConstantExpr?
Gabor Greifefe65362008-05-10 08:32:32 +0000225template <>
Jay Foad67c619b2011-01-11 15:07:38 +0000226struct OperandTraits<ConstantPlaceHolder> :
227 public FixedNumOperandTraits<ConstantPlaceHolder, 1> {
Gabor Greifefe65362008-05-10 08:32:32 +0000228};
Gabor Greifefe65362008-05-10 08:32:32 +0000229}
230
Chris Lattner46e77402009-03-31 22:55:09 +0000231
232void BitcodeReaderValueList::AssignValue(Value *V, unsigned Idx) {
233 if (Idx == size()) {
234 push_back(V);
235 return;
236 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000237
Chris Lattner46e77402009-03-31 22:55:09 +0000238 if (Idx >= size())
239 resize(Idx+1);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000240
Chris Lattner46e77402009-03-31 22:55:09 +0000241 WeakVH &OldV = ValuePtrs[Idx];
242 if (OldV == 0) {
243 OldV = V;
244 return;
245 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000246
Chris Lattner46e77402009-03-31 22:55:09 +0000247 // Handle constants and non-constants (e.g. instrs) differently for
248 // efficiency.
249 if (Constant *PHC = dyn_cast<Constant>(&*OldV)) {
250 ResolveConstants.push_back(std::make_pair(PHC, Idx));
251 OldV = V;
252 } else {
253 // If there was a forward reference to this value, replace it.
254 Value *PrevVal = OldV;
255 OldV->replaceAllUsesWith(V);
256 delete PrevVal;
Gabor Greifefe65362008-05-10 08:32:32 +0000257 }
258}
Daniel Dunbara279bc32009-09-20 02:20:51 +0000259
Gabor Greifefe65362008-05-10 08:32:32 +0000260
Chris Lattner522b7b12007-04-24 05:48:56 +0000261Constant *BitcodeReaderValueList::getConstantFwdRef(unsigned Idx,
Chris Lattnerdb125cf2011-07-18 04:54:35 +0000262 Type *Ty) {
Chris Lattner46e77402009-03-31 22:55:09 +0000263 if (Idx >= size())
Gabor Greifefe65362008-05-10 08:32:32 +0000264 resize(Idx + 1);
Chris Lattner522b7b12007-04-24 05:48:56 +0000265
Chris Lattner46e77402009-03-31 22:55:09 +0000266 if (Value *V = ValuePtrs[Idx]) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000267 assert(Ty == V->getType() && "Type mismatch in constant table!");
268 return cast<Constant>(V);
Chris Lattnerf581c3b2007-04-24 07:07:11 +0000269 }
Chris Lattner522b7b12007-04-24 05:48:56 +0000270
271 // Create and return a placeholder, which will later be RAUW'd.
Owen Anderson74a77812009-07-07 20:18:58 +0000272 Constant *C = new ConstantPlaceHolder(Ty, Context);
Chris Lattner46e77402009-03-31 22:55:09 +0000273 ValuePtrs[Idx] = C;
Chris Lattner522b7b12007-04-24 05:48:56 +0000274 return C;
275}
276
Chris Lattnerdb125cf2011-07-18 04:54:35 +0000277Value *BitcodeReaderValueList::getValueFwdRef(unsigned Idx, Type *Ty) {
Chris Lattner46e77402009-03-31 22:55:09 +0000278 if (Idx >= size())
Gabor Greifefe65362008-05-10 08:32:32 +0000279 resize(Idx + 1);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000280
Chris Lattner46e77402009-03-31 22:55:09 +0000281 if (Value *V = ValuePtrs[Idx]) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000282 assert((Ty == 0 || Ty == V->getType()) && "Type mismatch in value table!");
283 return V;
284 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000285
Chris Lattner01ff65f2007-05-02 05:16:49 +0000286 // No type specified, must be invalid reference.
287 if (Ty == 0) return 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000288
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000289 // Create and return a placeholder, which will later be RAUW'd.
290 Value *V = new Argument(Ty);
Chris Lattner46e77402009-03-31 22:55:09 +0000291 ValuePtrs[Idx] = V;
Chris Lattnera7c49aa2007-05-01 07:01:57 +0000292 return V;
293}
294
Chris Lattnerea693df2008-08-21 02:34:16 +0000295/// ResolveConstantForwardRefs - Once all constants are read, this method bulk
296/// resolves any forward references. The idea behind this is that we sometimes
297/// get constants (such as large arrays) which reference *many* forward ref
298/// constants. Replacing each of these causes a lot of thrashing when
299/// building/reuniquing the constant. Instead of doing this, we look at all the
300/// uses and rewrite all the place holders at once for any constant that uses
301/// a placeholder.
302void BitcodeReaderValueList::ResolveConstantForwardRefs() {
Daniel Dunbara279bc32009-09-20 02:20:51 +0000303 // Sort the values by-pointer so that they are efficient to look up with a
Chris Lattnerea693df2008-08-21 02:34:16 +0000304 // binary search.
305 std::sort(ResolveConstants.begin(), ResolveConstants.end());
Daniel Dunbara279bc32009-09-20 02:20:51 +0000306
Chris Lattnerea693df2008-08-21 02:34:16 +0000307 SmallVector<Constant*, 64> NewOps;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000308
Chris Lattnerea693df2008-08-21 02:34:16 +0000309 while (!ResolveConstants.empty()) {
Chris Lattner46e77402009-03-31 22:55:09 +0000310 Value *RealVal = operator[](ResolveConstants.back().second);
Chris Lattnerea693df2008-08-21 02:34:16 +0000311 Constant *Placeholder = ResolveConstants.back().first;
312 ResolveConstants.pop_back();
Daniel Dunbara279bc32009-09-20 02:20:51 +0000313
Chris Lattnerea693df2008-08-21 02:34:16 +0000314 // Loop over all users of the placeholder, updating them to reference the
315 // new value. If they reference more than one placeholder, update them all
316 // at once.
317 while (!Placeholder->use_empty()) {
Chris Lattnerb6135a02008-08-21 17:31:45 +0000318 Value::use_iterator UI = Placeholder->use_begin();
Gabor Greifc654d1b2010-07-09 16:01:21 +0000319 User *U = *UI;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000320
Chris Lattnerea693df2008-08-21 02:34:16 +0000321 // If the using object isn't uniqued, just update the operands. This
322 // handles instructions and initializers for global variables.
Gabor Greifc654d1b2010-07-09 16:01:21 +0000323 if (!isa<Constant>(U) || isa<GlobalValue>(U)) {
Chris Lattnerb6135a02008-08-21 17:31:45 +0000324 UI.getUse().set(RealVal);
Chris Lattnerea693df2008-08-21 02:34:16 +0000325 continue;
326 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000327
Chris Lattnerea693df2008-08-21 02:34:16 +0000328 // Otherwise, we have a constant that uses the placeholder. Replace that
329 // constant with a new constant that has *all* placeholder uses updated.
Gabor Greifc654d1b2010-07-09 16:01:21 +0000330 Constant *UserC = cast<Constant>(U);
Chris Lattnerea693df2008-08-21 02:34:16 +0000331 for (User::op_iterator I = UserC->op_begin(), E = UserC->op_end();
332 I != E; ++I) {
333 Value *NewOp;
334 if (!isa<ConstantPlaceHolder>(*I)) {
335 // Not a placeholder reference.
336 NewOp = *I;
337 } else if (*I == Placeholder) {
338 // Common case is that it just references this one placeholder.
339 NewOp = RealVal;
340 } else {
341 // Otherwise, look up the placeholder in ResolveConstants.
Daniel Dunbara279bc32009-09-20 02:20:51 +0000342 ResolveConstantsTy::iterator It =
343 std::lower_bound(ResolveConstants.begin(), ResolveConstants.end(),
Chris Lattnerea693df2008-08-21 02:34:16 +0000344 std::pair<Constant*, unsigned>(cast<Constant>(*I),
345 0));
346 assert(It != ResolveConstants.end() && It->first == *I);
Chris Lattner46e77402009-03-31 22:55:09 +0000347 NewOp = operator[](It->second);
Chris Lattnerea693df2008-08-21 02:34:16 +0000348 }
349
350 NewOps.push_back(cast<Constant>(NewOp));
351 }
352
353 // Make the new constant.
354 Constant *NewC;
355 if (ConstantArray *UserCA = dyn_cast<ConstantArray>(UserC)) {
Jay Foad26701082011-06-22 09:24:39 +0000356 NewC = ConstantArray::get(UserCA->getType(), NewOps);
Chris Lattnerea693df2008-08-21 02:34:16 +0000357 } else if (ConstantStruct *UserCS = dyn_cast<ConstantStruct>(UserC)) {
Chris Lattnerb065b062011-06-20 04:01:31 +0000358 NewC = ConstantStruct::get(UserCS->getType(), NewOps);
Chris Lattnerea693df2008-08-21 02:34:16 +0000359 } else if (isa<ConstantVector>(UserC)) {
Chris Lattner2ca5c862011-02-15 00:14:00 +0000360 NewC = ConstantVector::get(NewOps);
Nick Lewyckycb337992009-05-10 20:57:05 +0000361 } else {
362 assert(isa<ConstantExpr>(UserC) && "Must be a ConstantExpr.");
Jay Foadb81e4572011-04-13 13:46:01 +0000363 NewC = cast<ConstantExpr>(UserC)->getWithOperands(NewOps);
Chris Lattnerea693df2008-08-21 02:34:16 +0000364 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000365
Chris Lattnerea693df2008-08-21 02:34:16 +0000366 UserC->replaceAllUsesWith(NewC);
367 UserC->destroyConstant();
368 NewOps.clear();
369 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000370
Nick Lewyckycb337992009-05-10 20:57:05 +0000371 // Update all ValueHandles, they should be the only users at this point.
372 Placeholder->replaceAllUsesWith(RealVal);
Chris Lattnerea693df2008-08-21 02:34:16 +0000373 delete Placeholder;
374 }
375}
376
Devang Pateld5ac4042009-08-04 06:00:18 +0000377void BitcodeReaderMDValueList::AssignValue(Value *V, unsigned Idx) {
378 if (Idx == size()) {
379 push_back(V);
380 return;
381 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000382
Devang Pateld5ac4042009-08-04 06:00:18 +0000383 if (Idx >= size())
384 resize(Idx+1);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000385
Devang Pateld5ac4042009-08-04 06:00:18 +0000386 WeakVH &OldV = MDValuePtrs[Idx];
387 if (OldV == 0) {
388 OldV = V;
389 return;
390 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000391
Devang Pateld5ac4042009-08-04 06:00:18 +0000392 // If there was a forward reference to this value, replace it.
Dan Gohman489b29b2010-08-20 22:02:26 +0000393 MDNode *PrevVal = cast<MDNode>(OldV);
Devang Pateld5ac4042009-08-04 06:00:18 +0000394 OldV->replaceAllUsesWith(V);
Dan Gohman489b29b2010-08-20 22:02:26 +0000395 MDNode::deleteTemporary(PrevVal);
Devang Patelc0ff8c82009-09-03 01:38:02 +0000396 // Deleting PrevVal sets Idx value in MDValuePtrs to null. Set new
397 // value for Idx.
398 MDValuePtrs[Idx] = V;
Devang Pateld5ac4042009-08-04 06:00:18 +0000399}
400
401Value *BitcodeReaderMDValueList::getValueFwdRef(unsigned Idx) {
402 if (Idx >= size())
403 resize(Idx + 1);
Daniel Dunbara279bc32009-09-20 02:20:51 +0000404
Devang Pateld5ac4042009-08-04 06:00:18 +0000405 if (Value *V = MDValuePtrs[Idx]) {
Chris Lattnercf0fe8d2009-10-05 05:54:46 +0000406 assert(V->getType()->isMetadataTy() && "Type mismatch in value table!");
Devang Pateld5ac4042009-08-04 06:00:18 +0000407 return V;
408 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000409
Devang Pateld5ac4042009-08-04 06:00:18 +0000410 // Create and return a placeholder, which will later be RAUW'd.
Jay Foadec9186b2011-04-21 19:59:31 +0000411 Value *V = MDNode::getTemporary(Context, ArrayRef<Value*>());
Devang Pateld5ac4042009-08-04 06:00:18 +0000412 MDValuePtrs[Idx] = V;
413 return V;
414}
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000415
Chris Lattner1afcace2011-07-09 17:41:24 +0000416Type *BitcodeReader::getTypeByID(unsigned ID) {
417 // The type table size is always specified correctly.
418 if (ID >= TypeList.size())
419 return 0;
Derek Schufffccf0622012-02-06 19:03:04 +0000420
Chris Lattner1afcace2011-07-09 17:41:24 +0000421 if (Type *Ty = TypeList[ID])
422 return Ty;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000423
Chris Lattner1afcace2011-07-09 17:41:24 +0000424 // If we have a forward reference, the only possible case is when it is to a
425 // named struct. Just create a placeholder for now.
Chris Lattner3ebb6492011-08-12 18:06:37 +0000426 return TypeList[ID] = StructType::create(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000427}
428
Chris Lattner1afcace2011-07-09 17:41:24 +0000429
Chris Lattner48c85b82007-05-04 03:30:17 +0000430//===----------------------------------------------------------------------===//
431// Functions for parsing blocks from the bitcode file
432//===----------------------------------------------------------------------===//
433
Devang Patel05988662008-09-25 21:00:45 +0000434bool BitcodeReader::ParseAttributeBlock() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000435 if (Stream.EnterSubBlock(bitc::PARAMATTR_BLOCK_ID))
Chris Lattner48c85b82007-05-04 03:30:17 +0000436 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000437
Devang Patel19c87462008-09-26 22:53:05 +0000438 if (!MAttributes.empty())
Chris Lattner48c85b82007-05-04 03:30:17 +0000439 return Error("Multiple PARAMATTR blocks found!");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000440
Chris Lattner48c85b82007-05-04 03:30:17 +0000441 SmallVector<uint64_t, 64> Record;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000442
Devang Patel05988662008-09-25 21:00:45 +0000443 SmallVector<AttributeWithIndex, 8> Attrs;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000444
Chris Lattner48c85b82007-05-04 03:30:17 +0000445 // Read all the records.
446 while (1) {
447 unsigned Code = Stream.ReadCode();
448 if (Code == bitc::END_BLOCK) {
449 if (Stream.ReadBlockEnd())
450 return Error("Error at end of PARAMATTR block");
451 return false;
452 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000453
Chris Lattner48c85b82007-05-04 03:30:17 +0000454 if (Code == bitc::ENTER_SUBBLOCK) {
455 // No known subblocks, always skip them.
456 Stream.ReadSubBlockID();
457 if (Stream.SkipBlock())
458 return Error("Malformed block record");
459 continue;
460 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000461
Chris Lattner48c85b82007-05-04 03:30:17 +0000462 if (Code == bitc::DEFINE_ABBREV) {
463 Stream.ReadAbbrevRecord();
464 continue;
465 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000466
Chris Lattner48c85b82007-05-04 03:30:17 +0000467 // Read a record.
468 Record.clear();
469 switch (Stream.ReadRecord(Code, Record)) {
470 default: // Default behavior: ignore.
471 break;
472 case bitc::PARAMATTR_CODE_ENTRY: { // ENTRY: [paramidx0, attr0, ...]
473 if (Record.size() & 1)
474 return Error("Invalid ENTRY record");
475
Chris Lattner48c85b82007-05-04 03:30:17 +0000476 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Meador Ingee99f8be2012-05-28 15:45:43 +0000477 Attributes ReconstitutedAttr =
478 Attribute::decodeLLVMAttributesForBitcode(Record[i+1]);
Kostya Serebryany164b86b2012-01-20 17:56:17 +0000479 Record[i+1] = ReconstitutedAttr.Raw();
Chris Lattner48c85b82007-05-04 03:30:17 +0000480 }
Chris Lattner461edd92008-03-12 02:25:52 +0000481
Devang Patel19c87462008-09-26 22:53:05 +0000482 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Nuno Lopes006c7b92012-05-08 17:07:35 +0000483 if (Attributes(Record[i+1]) != Attribute::None)
Kostya Serebryany164b86b2012-01-20 17:56:17 +0000484 Attrs.push_back(AttributeWithIndex::get(Record[i],
485 Attributes(Record[i+1])));
Devang Patel19c87462008-09-26 22:53:05 +0000486 }
Devang Patel19c87462008-09-26 22:53:05 +0000487
Chris Lattnerd509d0b2012-05-28 01:47:44 +0000488 MAttributes.push_back(AttrListPtr::get(Attrs));
Chris Lattner48c85b82007-05-04 03:30:17 +0000489 Attrs.clear();
490 break;
491 }
Duncan Sands5e41f652007-11-20 14:09:29 +0000492 }
Chris Lattner48c85b82007-05-04 03:30:17 +0000493 }
494}
495
Chris Lattner86697142007-05-01 05:01:34 +0000496bool BitcodeReader::ParseTypeTable() {
Chris Lattner1afcace2011-07-09 17:41:24 +0000497 if (Stream.EnterSubBlock(bitc::TYPE_BLOCK_ID_NEW))
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000498 return Error("Malformed block record");
Derek Schufffccf0622012-02-06 19:03:04 +0000499
Chris Lattner1afcace2011-07-09 17:41:24 +0000500 return ParseTypeTableBody();
501}
Daniel Dunbara279bc32009-09-20 02:20:51 +0000502
Chris Lattner1afcace2011-07-09 17:41:24 +0000503bool BitcodeReader::ParseTypeTableBody() {
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000504 if (!TypeList.empty())
505 return Error("Multiple TYPE_BLOCKs found!");
506
507 SmallVector<uint64_t, 64> Record;
508 unsigned NumRecords = 0;
509
Chris Lattner1afcace2011-07-09 17:41:24 +0000510 SmallString<64> TypeName;
Derek Schufffccf0622012-02-06 19:03:04 +0000511
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000512 // Read all the records for this type table.
513 while (1) {
514 unsigned Code = Stream.ReadCode();
515 if (Code == bitc::END_BLOCK) {
516 if (NumRecords != TypeList.size())
517 return Error("Invalid type forward reference in TYPE_BLOCK");
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000518 if (Stream.ReadBlockEnd())
519 return Error("Error at end of type table block");
520 return false;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000521 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000522
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000523 if (Code == bitc::ENTER_SUBBLOCK) {
524 // No known subblocks, always skip them.
525 Stream.ReadSubBlockID();
526 if (Stream.SkipBlock())
527 return Error("Malformed block record");
528 continue;
529 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000530
Chris Lattner36d5e7d2007-04-23 16:04:05 +0000531 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +0000532 Stream.ReadAbbrevRecord();
533 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000534 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000535
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000536 // Read a record.
537 Record.clear();
Chris Lattner1afcace2011-07-09 17:41:24 +0000538 Type *ResultTy = 0;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000539 switch (Stream.ReadRecord(Code, Record)) {
Chris Lattner1afcace2011-07-09 17:41:24 +0000540 default: return Error("unknown type in type table");
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000541 case bitc::TYPE_CODE_NUMENTRY: // TYPE_CODE_NUMENTRY: [numentries]
542 // TYPE_CODE_NUMENTRY contains a count of the number of types in the
543 // type list. This allows us to reserve space.
544 if (Record.size() < 1)
545 return Error("Invalid TYPE_CODE_NUMENTRY record");
Chris Lattner1afcace2011-07-09 17:41:24 +0000546 TypeList.resize(Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000547 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000548 case bitc::TYPE_CODE_VOID: // VOID
Owen Anderson1d0be152009-08-13 21:58:54 +0000549 ResultTy = Type::getVoidTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000550 break;
Dan Gohmance163392011-12-17 00:04:22 +0000551 case bitc::TYPE_CODE_HALF: // HALF
552 ResultTy = Type::getHalfTy(Context);
553 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000554 case bitc::TYPE_CODE_FLOAT: // FLOAT
Owen Anderson1d0be152009-08-13 21:58:54 +0000555 ResultTy = Type::getFloatTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000556 break;
557 case bitc::TYPE_CODE_DOUBLE: // DOUBLE
Owen Anderson1d0be152009-08-13 21:58:54 +0000558 ResultTy = Type::getDoubleTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000559 break;
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000560 case bitc::TYPE_CODE_X86_FP80: // X86_FP80
Owen Anderson1d0be152009-08-13 21:58:54 +0000561 ResultTy = Type::getX86_FP80Ty(Context);
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000562 break;
563 case bitc::TYPE_CODE_FP128: // FP128
Owen Anderson1d0be152009-08-13 21:58:54 +0000564 ResultTy = Type::getFP128Ty(Context);
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000565 break;
566 case bitc::TYPE_CODE_PPC_FP128: // PPC_FP128
Owen Anderson1d0be152009-08-13 21:58:54 +0000567 ResultTy = Type::getPPC_FP128Ty(Context);
Dale Johannesen320fc8a2007-08-03 01:03:46 +0000568 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000569 case bitc::TYPE_CODE_LABEL: // LABEL
Owen Anderson1d0be152009-08-13 21:58:54 +0000570 ResultTy = Type::getLabelTy(Context);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000571 break;
Nick Lewycky7a0370f2009-05-30 05:06:04 +0000572 case bitc::TYPE_CODE_METADATA: // METADATA
Owen Anderson1d0be152009-08-13 21:58:54 +0000573 ResultTy = Type::getMetadataTy(Context);
Nick Lewycky7a0370f2009-05-30 05:06:04 +0000574 break;
Dale Johannesenbb811a22010-09-10 20:55:01 +0000575 case bitc::TYPE_CODE_X86_MMX: // X86_MMX
576 ResultTy = Type::getX86_MMXTy(Context);
577 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000578 case bitc::TYPE_CODE_INTEGER: // INTEGER: [width]
579 if (Record.size() < 1)
580 return Error("Invalid Integer type record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000581
Owen Anderson1d0be152009-08-13 21:58:54 +0000582 ResultTy = IntegerType::get(Context, Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000583 break;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000584 case bitc::TYPE_CODE_POINTER: { // POINTER: [pointee type] or
Christopher Lambfe63fb92007-12-11 08:59:05 +0000585 // [pointee type, address space]
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000586 if (Record.size() < 1)
587 return Error("Invalid POINTER type record");
Christopher Lambfe63fb92007-12-11 08:59:05 +0000588 unsigned AddressSpace = 0;
589 if (Record.size() == 2)
590 AddressSpace = Record[1];
Chris Lattner1afcace2011-07-09 17:41:24 +0000591 ResultTy = getTypeByID(Record[0]);
592 if (ResultTy == 0) return Error("invalid element type in pointer type");
593 ResultTy = PointerType::get(ResultTy, AddressSpace);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000594 break;
Christopher Lambfe63fb92007-12-11 08:59:05 +0000595 }
Nuno Lopesee8100d2012-05-23 15:19:39 +0000596 case bitc::TYPE_CODE_FUNCTION_OLD: {
597 // FIXME: attrid is dead, remove it in LLVM 4.0
598 // FUNCTION: [vararg, attrid, retty, paramty x N]
599 if (Record.size() < 3)
600 return Error("Invalid FUNCTION type record");
601 SmallVector<Type*, 8> ArgTys;
602 for (unsigned i = 3, e = Record.size(); i != e; ++i) {
603 if (Type *T = getTypeByID(Record[i]))
604 ArgTys.push_back(T);
605 else
606 break;
607 }
608
609 ResultTy = getTypeByID(Record[2]);
610 if (ResultTy == 0 || ArgTys.size() < Record.size()-3)
611 return Error("invalid type in function type");
612
613 ResultTy = FunctionType::get(ResultTy, ArgTys, Record[0]);
614 break;
615 }
Chad Rosiercde54642011-11-03 00:14:01 +0000616 case bitc::TYPE_CODE_FUNCTION: {
617 // FUNCTION: [vararg, retty, paramty x N]
618 if (Record.size() < 2)
619 return Error("Invalid FUNCTION type record");
Chris Lattnerd629efa2012-01-27 03:15:49 +0000620 SmallVector<Type*, 8> ArgTys;
Chad Rosiercde54642011-11-03 00:14:01 +0000621 for (unsigned i = 2, e = Record.size(); i != e; ++i) {
622 if (Type *T = getTypeByID(Record[i]))
623 ArgTys.push_back(T);
624 else
625 break;
626 }
627
628 ResultTy = getTypeByID(Record[1]);
629 if (ResultTy == 0 || ArgTys.size() < Record.size()-2)
630 return Error("invalid type in function type");
631
632 ResultTy = FunctionType::get(ResultTy, ArgTys, Record[0]);
633 break;
634 }
Chris Lattner1afcace2011-07-09 17:41:24 +0000635 case bitc::TYPE_CODE_STRUCT_ANON: { // STRUCT: [ispacked, eltty x N]
Chris Lattner7108dce2007-05-06 08:21:50 +0000636 if (Record.size() < 1)
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000637 return Error("Invalid STRUCT type record");
Chris Lattnerd629efa2012-01-27 03:15:49 +0000638 SmallVector<Type*, 8> EltTys;
Chris Lattner1afcace2011-07-09 17:41:24 +0000639 for (unsigned i = 1, e = Record.size(); i != e; ++i) {
640 if (Type *T = getTypeByID(Record[i]))
641 EltTys.push_back(T);
642 else
643 break;
644 }
645 if (EltTys.size() != Record.size()-1)
646 return Error("invalid type in struct type");
Owen Andersond7f2a6c2009-08-05 23:16:16 +0000647 ResultTy = StructType::get(Context, EltTys, Record[0]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +0000648 break;
649 }
Chris Lattner1afcace2011-07-09 17:41:24 +0000650 case bitc::TYPE_CODE_STRUCT_NAME: // STRUCT_NAME: [strchr x N]
651 if (ConvertToString(Record, 0, TypeName))
652 return Error("Invalid STRUCT_NAME record");
653 continue;
654
655 case bitc::TYPE_CODE_STRUCT_NAMED: { // STRUCT: [ispacked, eltty x N]
656 if (Record.size() < 1)
657 return Error("Invalid STRUCT type record");
658
659 if (NumRecords >= TypeList.size())
660 return Error("invalid TYPE table");
661
662 // Check to see if this was forward referenced, if so fill in the temp.
663 StructType *Res = cast_or_null<StructType>(TypeList[NumRecords]);
664 if (Res) {
665 Res->setName(TypeName);
666 TypeList[NumRecords] = 0;
667 } else // Otherwise, create a new struct.
Chris Lattner3ebb6492011-08-12 18:06:37 +0000668 Res = StructType::create(Context, TypeName);
Chris Lattner1afcace2011-07-09 17:41:24 +0000669 TypeName.clear();
670
671 SmallVector<Type*, 8> EltTys;
672 for (unsigned i = 1, e = Record.size(); i != e; ++i) {
673 if (Type *T = getTypeByID(Record[i]))
674 EltTys.push_back(T);
675 else
676 break;
677 }
678 if (EltTys.size() != Record.size()-1)
679 return Error("invalid STRUCT type record");
680 Res->setBody(EltTys, Record[0]);
681 ResultTy = Res;
682 break;
683 }
684 case bitc::TYPE_CODE_OPAQUE: { // OPAQUE: []
685 if (Record.size() != 1)
686 return Error("Invalid OPAQUE type record");
687
688 if (NumRecords >= TypeList.size())
689 return Error("invalid TYPE table");
690
691 // Check to see if this was forward referenced, if so fill in the temp.
692 StructType *Res = cast_or_null<StructType>(TypeList[NumRecords]);
693 if (Res) {
694 Res->setName(TypeName);
695 TypeList[NumRecords] = 0;
696 } else // Otherwise, create a new struct with no body.
Chris Lattner3ebb6492011-08-12 18:06:37 +0000697 Res = StructType::create(Context, TypeName);
Chris Lattner1afcace2011-07-09 17:41:24 +0000698 TypeName.clear();
699 ResultTy = Res;
700 break;
701 }
702 case bitc::TYPE_CODE_ARRAY: // ARRAY: [numelts, eltty]
703 if (Record.size() < 2)
704 return Error("Invalid ARRAY type record");
705 if ((ResultTy = getTypeByID(Record[1])))
706 ResultTy = ArrayType::get(ResultTy, Record[0]);
707 else
708 return Error("Invalid ARRAY type element");
709 break;
710 case bitc::TYPE_CODE_VECTOR: // VECTOR: [numelts, eltty]
711 if (Record.size() < 2)
712 return Error("Invalid VECTOR type record");
713 if ((ResultTy = getTypeByID(Record[1])))
714 ResultTy = VectorType::get(ResultTy, Record[0]);
715 else
716 return Error("Invalid ARRAY type element");
717 break;
718 }
719
720 if (NumRecords >= TypeList.size())
721 return Error("invalid TYPE table");
722 assert(ResultTy && "Didn't read a type?");
723 assert(TypeList[NumRecords] == 0 && "Already read type?");
724 TypeList[NumRecords++] = ResultTy;
725 }
726}
727
Chris Lattner86697142007-05-01 05:01:34 +0000728bool BitcodeReader::ParseValueSymbolTable() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000729 if (Stream.EnterSubBlock(bitc::VALUE_SYMTAB_BLOCK_ID))
Chris Lattner0b2482a2007-04-23 21:26:05 +0000730 return Error("Malformed block record");
731
732 SmallVector<uint64_t, 64> Record;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000733
Chris Lattner0b2482a2007-04-23 21:26:05 +0000734 // Read all the records for this value table.
735 SmallString<128> ValueName;
736 while (1) {
737 unsigned Code = Stream.ReadCode();
Chris Lattnerf66d20d2007-04-24 18:15:21 +0000738 if (Code == bitc::END_BLOCK) {
739 if (Stream.ReadBlockEnd())
740 return Error("Error at end of value symbol table block");
741 return false;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000742 }
Chris Lattner0b2482a2007-04-23 21:26:05 +0000743 if (Code == bitc::ENTER_SUBBLOCK) {
744 // No known subblocks, always skip them.
745 Stream.ReadSubBlockID();
746 if (Stream.SkipBlock())
747 return Error("Malformed block record");
748 continue;
749 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000750
Chris Lattner0b2482a2007-04-23 21:26:05 +0000751 if (Code == bitc::DEFINE_ABBREV) {
752 Stream.ReadAbbrevRecord();
753 continue;
754 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000755
Chris Lattner0b2482a2007-04-23 21:26:05 +0000756 // Read a record.
757 Record.clear();
Bill Wendling5d7a5a42011-04-10 23:18:04 +0000758 switch (Stream.ReadRecord(Code, Record)) {
Chris Lattner0b2482a2007-04-23 21:26:05 +0000759 default: // Default behavior: unknown type.
760 break;
Chris Lattner15e6d172007-05-04 19:11:41 +0000761 case bitc::VST_CODE_ENTRY: { // VST_ENTRY: [valueid, namechar x N]
Chris Lattner0b2482a2007-04-23 21:26:05 +0000762 if (ConvertToString(Record, 1, ValueName))
Nick Lewycky88b72932009-05-31 06:07:28 +0000763 return Error("Invalid VST_ENTRY record");
Chris Lattner0b2482a2007-04-23 21:26:05 +0000764 unsigned ValueID = Record[0];
765 if (ValueID >= ValueList.size())
766 return Error("Invalid Value ID in VST_ENTRY record");
767 Value *V = ValueList[ValueID];
Daniel Dunbara279bc32009-09-20 02:20:51 +0000768
Daniel Dunbar3f53fa92009-07-26 00:34:27 +0000769 V->setName(StringRef(ValueName.data(), ValueName.size()));
Chris Lattner0b2482a2007-04-23 21:26:05 +0000770 ValueName.clear();
771 break;
Reid Spencerc8f8a242007-05-04 01:43:33 +0000772 }
Bill Wendling5d7a5a42011-04-10 23:18:04 +0000773 case bitc::VST_CODE_BBENTRY: {
Chris Lattnere825ed52007-05-03 22:18:21 +0000774 if (ConvertToString(Record, 1, ValueName))
775 return Error("Invalid VST_BBENTRY record");
776 BasicBlock *BB = getBasicBlock(Record[0]);
777 if (BB == 0)
778 return Error("Invalid BB ID in VST_BBENTRY record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000779
Daniel Dunbar3f53fa92009-07-26 00:34:27 +0000780 BB->setName(StringRef(ValueName.data(), ValueName.size()));
Chris Lattnere825ed52007-05-03 22:18:21 +0000781 ValueName.clear();
782 break;
Chris Lattner0b2482a2007-04-23 21:26:05 +0000783 }
Reid Spencerc8f8a242007-05-04 01:43:33 +0000784 }
Chris Lattner0b2482a2007-04-23 21:26:05 +0000785 }
786}
787
Devang Patele54abc92009-07-22 17:43:22 +0000788bool BitcodeReader::ParseMetadata() {
Devang Patel23598502010-01-11 18:52:33 +0000789 unsigned NextMDValueNo = MDValueList.size();
Devang Patele54abc92009-07-22 17:43:22 +0000790
791 if (Stream.EnterSubBlock(bitc::METADATA_BLOCK_ID))
792 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000793
Devang Patele54abc92009-07-22 17:43:22 +0000794 SmallVector<uint64_t, 64> Record;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000795
Devang Patele54abc92009-07-22 17:43:22 +0000796 // Read all the records.
797 while (1) {
798 unsigned Code = Stream.ReadCode();
799 if (Code == bitc::END_BLOCK) {
800 if (Stream.ReadBlockEnd())
801 return Error("Error at end of PARAMATTR block");
802 return false;
803 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000804
Devang Patele54abc92009-07-22 17:43:22 +0000805 if (Code == bitc::ENTER_SUBBLOCK) {
806 // No known subblocks, always skip them.
807 Stream.ReadSubBlockID();
808 if (Stream.SkipBlock())
809 return Error("Malformed block record");
810 continue;
811 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000812
Devang Patele54abc92009-07-22 17:43:22 +0000813 if (Code == bitc::DEFINE_ABBREV) {
814 Stream.ReadAbbrevRecord();
815 continue;
816 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000817
Victor Hernandez24e64df2010-01-10 07:14:18 +0000818 bool IsFunctionLocal = false;
Devang Patele54abc92009-07-22 17:43:22 +0000819 // Read a record.
820 Record.clear();
Dan Gohman9b10dfb2010-09-13 18:00:48 +0000821 Code = Stream.ReadRecord(Code, Record);
822 switch (Code) {
Devang Patele54abc92009-07-22 17:43:22 +0000823 default: // Default behavior: ignore.
824 break;
Devang Patelaa993142009-07-29 22:34:41 +0000825 case bitc::METADATA_NAME: {
826 // Read named of the named metadata.
Benjamin Kramerf52aea82012-05-28 14:10:31 +0000827 SmallString<8> Name(Record.begin(), Record.end());
Devang Patelaa993142009-07-29 22:34:41 +0000828 Record.clear();
829 Code = Stream.ReadCode();
830
Chris Lattner9d61dd92011-06-17 17:50:30 +0000831 // METADATA_NAME is always followed by METADATA_NAMED_NODE.
Dan Gohman70c2fc02010-09-09 23:12:39 +0000832 unsigned NextBitCode = Stream.ReadRecord(Code, Record);
Chris Lattner9d61dd92011-06-17 17:50:30 +0000833 assert(NextBitCode == bitc::METADATA_NAMED_NODE); (void)NextBitCode;
Devang Patelaa993142009-07-29 22:34:41 +0000834
835 // Read named metadata elements.
836 unsigned Size = Record.size();
Dan Gohman17aa92c2010-07-21 23:38:33 +0000837 NamedMDNode *NMD = TheModule->getOrInsertNamedMetadata(Name);
Devang Patelaa993142009-07-29 22:34:41 +0000838 for (unsigned i = 0; i != Size; ++i) {
Chris Lattner70644e92010-01-09 02:02:37 +0000839 MDNode *MD = dyn_cast<MDNode>(MDValueList.getValueFwdRef(Record[i]));
840 if (MD == 0)
841 return Error("Malformed metadata record");
Dan Gohman17aa92c2010-07-21 23:38:33 +0000842 NMD->addOperand(MD);
Devang Patelaa993142009-07-29 22:34:41 +0000843 }
Devang Patelaa993142009-07-29 22:34:41 +0000844 break;
845 }
Chris Lattner9d61dd92011-06-17 17:50:30 +0000846 case bitc::METADATA_FN_NODE:
Victor Hernandez24e64df2010-01-10 07:14:18 +0000847 IsFunctionLocal = true;
848 // fall-through
Chris Lattner9d61dd92011-06-17 17:50:30 +0000849 case bitc::METADATA_NODE: {
Dan Gohmanac809752010-07-13 19:33:27 +0000850 if (Record.size() % 2 == 1)
Chris Lattner9d61dd92011-06-17 17:50:30 +0000851 return Error("Invalid METADATA_NODE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +0000852
Devang Patel104cf9e2009-07-23 01:07:34 +0000853 unsigned Size = Record.size();
854 SmallVector<Value*, 8> Elts;
855 for (unsigned i = 0; i != Size; i += 2) {
Chris Lattnerdb125cf2011-07-18 04:54:35 +0000856 Type *Ty = getTypeByID(Record[i]);
Chris Lattner9d61dd92011-06-17 17:50:30 +0000857 if (!Ty) return Error("Invalid METADATA_NODE record");
Chris Lattnercf0fe8d2009-10-05 05:54:46 +0000858 if (Ty->isMetadataTy())
Devang Pateld5ac4042009-08-04 06:00:18 +0000859 Elts.push_back(MDValueList.getValueFwdRef(Record[i+1]));
Benjamin Kramerf0127052010-01-05 13:12:22 +0000860 else if (!Ty->isVoidTy())
Devang Patel104cf9e2009-07-23 01:07:34 +0000861 Elts.push_back(ValueList.getValueFwdRef(Record[i+1], Ty));
862 else
863 Elts.push_back(NULL);
864 }
Jay Foadec9186b2011-04-21 19:59:31 +0000865 Value *V = MDNode::getWhenValsUnresolved(Context, Elts, IsFunctionLocal);
Victor Hernandez24e64df2010-01-10 07:14:18 +0000866 IsFunctionLocal = false;
Devang Patel23598502010-01-11 18:52:33 +0000867 MDValueList.AssignValue(V, NextMDValueNo++);
Devang Patel104cf9e2009-07-23 01:07:34 +0000868 break;
869 }
Devang Patele54abc92009-07-22 17:43:22 +0000870 case bitc::METADATA_STRING: {
Benjamin Kramerf52aea82012-05-28 14:10:31 +0000871 SmallString<8> String(Record.begin(), Record.end());
872 Value *V = MDString::get(Context, String);
Devang Patel23598502010-01-11 18:52:33 +0000873 MDValueList.AssignValue(V, NextMDValueNo++);
Devang Patele54abc92009-07-22 17:43:22 +0000874 break;
875 }
Devang Patele8e02132009-09-18 19:26:43 +0000876 case bitc::METADATA_KIND: {
Benjamin Kramerf52aea82012-05-28 14:10:31 +0000877 if (Record.size() < 2)
Daniel Dunbara279bc32009-09-20 02:20:51 +0000878 return Error("Invalid METADATA_KIND record");
Benjamin Kramerf52aea82012-05-28 14:10:31 +0000879
Devang Patela2148402009-09-28 21:14:55 +0000880 unsigned Kind = Record[0];
Benjamin Kramerf52aea82012-05-28 14:10:31 +0000881 SmallString<8> Name(Record.begin()+1, Record.end());
882
Chris Lattner08113472009-12-29 09:01:33 +0000883 unsigned NewKind = TheModule->getMDKindID(Name.str());
Dan Gohman19538d12010-07-20 21:42:28 +0000884 if (!MDKindMap.insert(std::make_pair(Kind, NewKind)).second)
885 return Error("Conflicting METADATA_KIND records");
Devang Patele8e02132009-09-18 19:26:43 +0000886 break;
887 }
Devang Patele54abc92009-07-22 17:43:22 +0000888 }
889 }
890}
891
Chris Lattner0eef0802007-04-24 04:04:35 +0000892/// DecodeSignRotatedValue - Decode a signed value stored with the sign bit in
893/// the LSB for dense VBR encoding.
894static uint64_t DecodeSignRotatedValue(uint64_t V) {
895 if ((V & 1) == 0)
896 return V >> 1;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000897 if (V != 1)
Chris Lattner0eef0802007-04-24 04:04:35 +0000898 return -(V >> 1);
899 // There is no such thing as -0 with integers. "-0" really means MININT.
900 return 1ULL << 63;
901}
902
Chris Lattner07d98b42007-04-26 02:46:40 +0000903/// ResolveGlobalAndAliasInits - Resolve all of the initializers for global
904/// values and aliases that we can.
905bool BitcodeReader::ResolveGlobalAndAliasInits() {
906 std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInitWorklist;
907 std::vector<std::pair<GlobalAlias*, unsigned> > AliasInitWorklist;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000908
Chris Lattner07d98b42007-04-26 02:46:40 +0000909 GlobalInitWorklist.swap(GlobalInits);
910 AliasInitWorklist.swap(AliasInits);
911
912 while (!GlobalInitWorklist.empty()) {
Chris Lattner198f34a2007-04-26 03:27:58 +0000913 unsigned ValID = GlobalInitWorklist.back().second;
Chris Lattner07d98b42007-04-26 02:46:40 +0000914 if (ValID >= ValueList.size()) {
915 // Not ready to resolve this yet, it requires something later in the file.
Chris Lattner198f34a2007-04-26 03:27:58 +0000916 GlobalInits.push_back(GlobalInitWorklist.back());
Chris Lattner07d98b42007-04-26 02:46:40 +0000917 } else {
918 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
919 GlobalInitWorklist.back().first->setInitializer(C);
920 else
921 return Error("Global variable initializer is not a constant!");
922 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000923 GlobalInitWorklist.pop_back();
Chris Lattner07d98b42007-04-26 02:46:40 +0000924 }
925
926 while (!AliasInitWorklist.empty()) {
927 unsigned ValID = AliasInitWorklist.back().second;
928 if (ValID >= ValueList.size()) {
929 AliasInits.push_back(AliasInitWorklist.back());
930 } else {
931 if (Constant *C = dyn_cast<Constant>(ValueList[ValID]))
Anton Korobeynikov7dde0ff2007-04-28 14:57:59 +0000932 AliasInitWorklist.back().first->setAliasee(C);
Chris Lattner07d98b42007-04-26 02:46:40 +0000933 else
934 return Error("Alias initializer is not a constant!");
935 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000936 AliasInitWorklist.pop_back();
Chris Lattner07d98b42007-04-26 02:46:40 +0000937 }
938 return false;
939}
940
Benjamin Kramerf52aea82012-05-28 14:10:31 +0000941static APInt ReadWideAPInt(ArrayRef<uint64_t> Vals, unsigned TypeBits) {
942 SmallVector<uint64_t, 8> Words(Vals.size());
943 std::transform(Vals.begin(), Vals.end(), Words.begin(),
944 DecodeSignRotatedValue);
945
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +0000946 return APInt(TypeBits, Words);
947}
948
Chris Lattner86697142007-05-01 05:01:34 +0000949bool BitcodeReader::ParseConstants() {
Chris Lattnere17b6582007-05-05 00:17:00 +0000950 if (Stream.EnterSubBlock(bitc::CONSTANTS_BLOCK_ID))
Chris Lattnere16504e2007-04-24 03:30:34 +0000951 return Error("Malformed block record");
952
953 SmallVector<uint64_t, 64> Record;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000954
Chris Lattnere16504e2007-04-24 03:30:34 +0000955 // Read all the records for this value table.
Chris Lattnerdb125cf2011-07-18 04:54:35 +0000956 Type *CurTy = Type::getInt32Ty(Context);
Chris Lattner522b7b12007-04-24 05:48:56 +0000957 unsigned NextCstNo = ValueList.size();
Chris Lattnere16504e2007-04-24 03:30:34 +0000958 while (1) {
959 unsigned Code = Stream.ReadCode();
Chris Lattnerea693df2008-08-21 02:34:16 +0000960 if (Code == bitc::END_BLOCK)
961 break;
Daniel Dunbara279bc32009-09-20 02:20:51 +0000962
Chris Lattnere16504e2007-04-24 03:30:34 +0000963 if (Code == bitc::ENTER_SUBBLOCK) {
964 // No known subblocks, always skip them.
965 Stream.ReadSubBlockID();
966 if (Stream.SkipBlock())
967 return Error("Malformed block record");
968 continue;
969 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000970
Chris Lattnere16504e2007-04-24 03:30:34 +0000971 if (Code == bitc::DEFINE_ABBREV) {
972 Stream.ReadAbbrevRecord();
973 continue;
974 }
Daniel Dunbara279bc32009-09-20 02:20:51 +0000975
Chris Lattnere16504e2007-04-24 03:30:34 +0000976 // Read a record.
977 Record.clear();
978 Value *V = 0;
Dan Gohman1224c382009-07-20 21:19:07 +0000979 unsigned BitCode = Stream.ReadRecord(Code, Record);
980 switch (BitCode) {
Chris Lattnere16504e2007-04-24 03:30:34 +0000981 default: // Default behavior: unknown constant
982 case bitc::CST_CODE_UNDEF: // UNDEF
Owen Anderson9e9a0d52009-07-30 23:03:37 +0000983 V = UndefValue::get(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +0000984 break;
985 case bitc::CST_CODE_SETTYPE: // SETTYPE: [typeid]
986 if (Record.empty())
987 return Error("Malformed CST_SETTYPE record");
988 if (Record[0] >= TypeList.size())
989 return Error("Invalid Type ID in CST_SETTYPE record");
990 CurTy = TypeList[Record[0]];
Chris Lattner0eef0802007-04-24 04:04:35 +0000991 continue; // Skip the ValueList manipulation.
Chris Lattnere16504e2007-04-24 03:30:34 +0000992 case bitc::CST_CODE_NULL: // NULL
Owen Andersona7235ea2009-07-31 20:28:14 +0000993 V = Constant::getNullValue(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +0000994 break;
995 case bitc::CST_CODE_INTEGER: // INTEGER: [intval]
Duncan Sands1df98592010-02-16 11:11:14 +0000996 if (!CurTy->isIntegerTy() || Record.empty())
Chris Lattner0eef0802007-04-24 04:04:35 +0000997 return Error("Invalid CST_INTEGER record");
Owen Andersoneed707b2009-07-24 23:12:02 +0000998 V = ConstantInt::get(CurTy, DecodeSignRotatedValue(Record[0]));
Chris Lattner0eef0802007-04-24 04:04:35 +0000999 break;
Chris Lattner15e6d172007-05-04 19:11:41 +00001000 case bitc::CST_CODE_WIDE_INTEGER: {// WIDE_INTEGER: [n x intval]
Duncan Sands1df98592010-02-16 11:11:14 +00001001 if (!CurTy->isIntegerTy() || Record.empty())
Chris Lattner0eef0802007-04-24 04:04:35 +00001002 return Error("Invalid WIDE_INTEGER record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001003
Benjamin Kramerf52aea82012-05-28 14:10:31 +00001004 APInt VInt = ReadWideAPInt(Record,
1005 cast<IntegerType>(CurTy)->getBitWidth());
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +00001006 V = ConstantInt::get(Context, VInt);
1007
Chris Lattner0eef0802007-04-24 04:04:35 +00001008 break;
1009 }
Dale Johannesen3f6eb742007-09-11 18:32:33 +00001010 case bitc::CST_CODE_FLOAT: { // FLOAT: [fpval]
Chris Lattner0eef0802007-04-24 04:04:35 +00001011 if (Record.empty())
1012 return Error("Invalid FLOAT record");
Dan Gohmance163392011-12-17 00:04:22 +00001013 if (CurTy->isHalfTy())
1014 V = ConstantFP::get(Context, APFloat(APInt(16, (uint16_t)Record[0])));
1015 else if (CurTy->isFloatTy())
Owen Anderson6f83c9c2009-07-27 20:59:43 +00001016 V = ConstantFP::get(Context, APFloat(APInt(32, (uint32_t)Record[0])));
Chris Lattnercf0fe8d2009-10-05 05:54:46 +00001017 else if (CurTy->isDoubleTy())
Owen Anderson6f83c9c2009-07-27 20:59:43 +00001018 V = ConstantFP::get(Context, APFloat(APInt(64, Record[0])));
Chris Lattnercf0fe8d2009-10-05 05:54:46 +00001019 else if (CurTy->isX86_FP80Ty()) {
Dale Johannesen1b25cb22009-03-23 21:16:53 +00001020 // Bits are not stored the same way as a normal i80 APInt, compensate.
1021 uint64_t Rearrange[2];
1022 Rearrange[0] = (Record[1] & 0xffffLL) | (Record[0] << 16);
1023 Rearrange[1] = Record[0] >> 48;
Jeffrey Yasskin3ba292d2011-07-18 21:45:40 +00001024 V = ConstantFP::get(Context, APFloat(APInt(80, Rearrange)));
Chris Lattnercf0fe8d2009-10-05 05:54:46 +00001025 } else if (CurTy->isFP128Ty())
Jeffrey Yasskin3ba292d2011-07-18 21:45:40 +00001026 V = ConstantFP::get(Context, APFloat(APInt(128, Record), true));
Chris Lattnercf0fe8d2009-10-05 05:54:46 +00001027 else if (CurTy->isPPC_FP128Ty())
Jeffrey Yasskin3ba292d2011-07-18 21:45:40 +00001028 V = ConstantFP::get(Context, APFloat(APInt(128, Record)));
Chris Lattnere16504e2007-04-24 03:30:34 +00001029 else
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001030 V = UndefValue::get(CurTy);
Chris Lattnere16504e2007-04-24 03:30:34 +00001031 break;
Dale Johannesen3f6eb742007-09-11 18:32:33 +00001032 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001033
Chris Lattner15e6d172007-05-04 19:11:41 +00001034 case bitc::CST_CODE_AGGREGATE: {// AGGREGATE: [n x value number]
1035 if (Record.empty())
Chris Lattner522b7b12007-04-24 05:48:56 +00001036 return Error("Invalid CST_AGGREGATE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001037
Chris Lattner15e6d172007-05-04 19:11:41 +00001038 unsigned Size = Record.size();
Chris Lattnerd629efa2012-01-27 03:15:49 +00001039 SmallVector<Constant*, 16> Elts;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001040
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001041 if (StructType *STy = dyn_cast<StructType>(CurTy)) {
Chris Lattner522b7b12007-04-24 05:48:56 +00001042 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +00001043 Elts.push_back(ValueList.getConstantFwdRef(Record[i],
Chris Lattner522b7b12007-04-24 05:48:56 +00001044 STy->getElementType(i)));
Owen Anderson8fa33382009-07-27 22:29:26 +00001045 V = ConstantStruct::get(STy, Elts);
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001046 } else if (ArrayType *ATy = dyn_cast<ArrayType>(CurTy)) {
1047 Type *EltTy = ATy->getElementType();
Chris Lattner522b7b12007-04-24 05:48:56 +00001048 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +00001049 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Owen Anderson1fd70962009-07-28 18:32:17 +00001050 V = ConstantArray::get(ATy, Elts);
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001051 } else if (VectorType *VTy = dyn_cast<VectorType>(CurTy)) {
1052 Type *EltTy = VTy->getElementType();
Chris Lattner522b7b12007-04-24 05:48:56 +00001053 for (unsigned i = 0; i != Size; ++i)
Chris Lattner15e6d172007-05-04 19:11:41 +00001054 Elts.push_back(ValueList.getConstantFwdRef(Record[i], EltTy));
Owen Andersonaf7ec972009-07-28 21:19:26 +00001055 V = ConstantVector::get(Elts);
Chris Lattner522b7b12007-04-24 05:48:56 +00001056 } else {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001057 V = UndefValue::get(CurTy);
Chris Lattner522b7b12007-04-24 05:48:56 +00001058 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001059 break;
1060 }
Chris Lattner2237f842012-02-05 02:41:35 +00001061 case bitc::CST_CODE_STRING: // STRING: [values]
Chris Lattnercb3d91b2007-05-06 00:53:07 +00001062 case bitc::CST_CODE_CSTRING: { // CSTRING: [values]
1063 if (Record.empty())
Chris Lattner2237f842012-02-05 02:41:35 +00001064 return Error("Invalid CST_STRING record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001065
Benjamin Kramerf52aea82012-05-28 14:10:31 +00001066 SmallString<16> Elts(Record.begin(), Record.end());
Chris Lattner2237f842012-02-05 02:41:35 +00001067 V = ConstantDataArray::getString(Context, Elts,
1068 BitCode == bitc::CST_CODE_CSTRING);
Chris Lattnercb3d91b2007-05-06 00:53:07 +00001069 break;
1070 }
Chris Lattnerd408f062012-01-30 00:51:16 +00001071 case bitc::CST_CODE_DATA: {// DATA: [n x value]
1072 if (Record.empty())
1073 return Error("Invalid CST_DATA record");
1074
1075 Type *EltTy = cast<SequentialType>(CurTy)->getElementType();
1076 unsigned Size = Record.size();
1077
1078 if (EltTy->isIntegerTy(8)) {
1079 SmallVector<uint8_t, 16> Elts(Record.begin(), Record.end());
1080 if (isa<VectorType>(CurTy))
1081 V = ConstantDataVector::get(Context, Elts);
1082 else
1083 V = ConstantDataArray::get(Context, Elts);
1084 } else if (EltTy->isIntegerTy(16)) {
1085 SmallVector<uint16_t, 16> Elts(Record.begin(), Record.end());
1086 if (isa<VectorType>(CurTy))
1087 V = ConstantDataVector::get(Context, Elts);
1088 else
1089 V = ConstantDataArray::get(Context, Elts);
1090 } else if (EltTy->isIntegerTy(32)) {
1091 SmallVector<uint32_t, 16> Elts(Record.begin(), Record.end());
1092 if (isa<VectorType>(CurTy))
1093 V = ConstantDataVector::get(Context, Elts);
1094 else
1095 V = ConstantDataArray::get(Context, Elts);
1096 } else if (EltTy->isIntegerTy(64)) {
1097 SmallVector<uint64_t, 16> Elts(Record.begin(), Record.end());
1098 if (isa<VectorType>(CurTy))
1099 V = ConstantDataVector::get(Context, Elts);
1100 else
1101 V = ConstantDataArray::get(Context, Elts);
1102 } else if (EltTy->isFloatTy()) {
Benjamin Kramerf52aea82012-05-28 14:10:31 +00001103 SmallVector<float, 16> Elts(Size);
1104 std::transform(Record.begin(), Record.end(), Elts.begin(), BitsToFloat);
Chris Lattnerd408f062012-01-30 00:51:16 +00001105 if (isa<VectorType>(CurTy))
1106 V = ConstantDataVector::get(Context, Elts);
1107 else
1108 V = ConstantDataArray::get(Context, Elts);
1109 } else if (EltTy->isDoubleTy()) {
Benjamin Kramerf52aea82012-05-28 14:10:31 +00001110 SmallVector<double, 16> Elts(Size);
1111 std::transform(Record.begin(), Record.end(), Elts.begin(),
1112 BitsToDouble);
Chris Lattnerd408f062012-01-30 00:51:16 +00001113 if (isa<VectorType>(CurTy))
1114 V = ConstantDataVector::get(Context, Elts);
1115 else
1116 V = ConstantDataArray::get(Context, Elts);
1117 } else {
1118 return Error("Unknown element type in CE_DATA");
1119 }
1120 break;
1121 }
1122
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001123 case bitc::CST_CODE_CE_BINOP: { // CE_BINOP: [opcode, opval, opval]
1124 if (Record.size() < 3) return Error("Invalid CE_BINOP record");
1125 int Opc = GetDecodedBinaryOpcode(Record[0], CurTy);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001126 if (Opc < 0) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001127 V = UndefValue::get(CurTy); // Unknown binop.
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001128 } else {
1129 Constant *LHS = ValueList.getConstantFwdRef(Record[1], CurTy);
1130 Constant *RHS = ValueList.getConstantFwdRef(Record[2], CurTy);
Dan Gohmanf8dbee72009-09-07 23:54:19 +00001131 unsigned Flags = 0;
1132 if (Record.size() >= 4) {
1133 if (Opc == Instruction::Add ||
1134 Opc == Instruction::Sub ||
Chris Lattnerf067d582011-02-07 16:40:21 +00001135 Opc == Instruction::Mul ||
1136 Opc == Instruction::Shl) {
Dan Gohmanf8dbee72009-09-07 23:54:19 +00001137 if (Record[3] & (1 << bitc::OBO_NO_SIGNED_WRAP))
1138 Flags |= OverflowingBinaryOperator::NoSignedWrap;
1139 if (Record[3] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
1140 Flags |= OverflowingBinaryOperator::NoUnsignedWrap;
Chris Lattner35bda892011-02-06 21:44:57 +00001141 } else if (Opc == Instruction::SDiv ||
Chris Lattnerf067d582011-02-07 16:40:21 +00001142 Opc == Instruction::UDiv ||
1143 Opc == Instruction::LShr ||
1144 Opc == Instruction::AShr) {
Chris Lattner35bda892011-02-06 21:44:57 +00001145 if (Record[3] & (1 << bitc::PEO_EXACT))
Dan Gohmanf8dbee72009-09-07 23:54:19 +00001146 Flags |= SDivOperator::IsExact;
1147 }
1148 }
1149 V = ConstantExpr::get(Opc, LHS, RHS, Flags);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001150 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001151 break;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001152 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001153 case bitc::CST_CODE_CE_CAST: { // CE_CAST: [opcode, opty, opval]
1154 if (Record.size() < 3) return Error("Invalid CE_CAST record");
1155 int Opc = GetDecodedCastOpcode(Record[0]);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001156 if (Opc < 0) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00001157 V = UndefValue::get(CurTy); // Unknown cast.
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001158 } else {
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001159 Type *OpTy = getTypeByID(Record[1]);
Chris Lattnerbfcc3802007-05-06 07:33:01 +00001160 if (!OpTy) return Error("Invalid CE_CAST record");
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001161 Constant *Op = ValueList.getConstantFwdRef(Record[2], OpTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001162 V = ConstantExpr::getCast(Opc, Op, CurTy);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001163 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001164 break;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001165 }
Dan Gohmandd8004d2009-07-27 21:53:46 +00001166 case bitc::CST_CODE_CE_INBOUNDS_GEP:
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001167 case bitc::CST_CODE_CE_GEP: { // CE_GEP: [n x operands]
Chris Lattner15e6d172007-05-04 19:11:41 +00001168 if (Record.size() & 1) return Error("Invalid CE_GEP record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001169 SmallVector<Constant*, 16> Elts;
Chris Lattner15e6d172007-05-04 19:11:41 +00001170 for (unsigned i = 0, e = Record.size(); i != e; i += 2) {
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001171 Type *ElTy = getTypeByID(Record[i]);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001172 if (!ElTy) return Error("Invalid CE_GEP record");
1173 Elts.push_back(ValueList.getConstantFwdRef(Record[i+1], ElTy));
1174 }
Jay Foaddab3d292011-07-21 14:31:17 +00001175 ArrayRef<Constant *> Indices(Elts.begin() + 1, Elts.end());
Jay Foad4b5e2072011-07-21 15:15:37 +00001176 V = ConstantExpr::getGetElementPtr(Elts[0], Indices,
1177 BitCode ==
1178 bitc::CST_CODE_CE_INBOUNDS_GEP);
Chris Lattnerf66d20d2007-04-24 18:15:21 +00001179 break;
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001180 }
1181 case bitc::CST_CODE_CE_SELECT: // CE_SELECT: [opval#, opval#, opval#]
1182 if (Record.size() < 3) return Error("Invalid CE_SELECT record");
Owen Andersonbaf3c402009-07-29 18:55:55 +00001183 V = ConstantExpr::getSelect(ValueList.getConstantFwdRef(Record[0],
Owen Anderson1d0be152009-08-13 21:58:54 +00001184 Type::getInt1Ty(Context)),
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001185 ValueList.getConstantFwdRef(Record[1],CurTy),
1186 ValueList.getConstantFwdRef(Record[2],CurTy));
1187 break;
1188 case bitc::CST_CODE_CE_EXTRACTELT: { // CE_EXTRACTELT: [opty, opval, opval]
1189 if (Record.size() < 3) return Error("Invalid CE_EXTRACTELT record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001190 VectorType *OpTy =
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001191 dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
1192 if (OpTy == 0) return Error("Invalid CE_EXTRACTELT record");
1193 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
Owen Anderson1d0be152009-08-13 21:58:54 +00001194 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
Owen Andersonbaf3c402009-07-29 18:55:55 +00001195 V = ConstantExpr::getExtractElement(Op0, Op1);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001196 break;
1197 }
1198 case bitc::CST_CODE_CE_INSERTELT: { // CE_INSERTELT: [opval, opval, opval]
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001199 VectorType *OpTy = dyn_cast<VectorType>(CurTy);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001200 if (Record.size() < 3 || OpTy == 0)
1201 return Error("Invalid CE_INSERTELT record");
1202 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1203 Constant *Op1 = ValueList.getConstantFwdRef(Record[1],
1204 OpTy->getElementType());
Owen Anderson1d0be152009-08-13 21:58:54 +00001205 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], Type::getInt32Ty(Context));
Owen Andersonbaf3c402009-07-29 18:55:55 +00001206 V = ConstantExpr::getInsertElement(Op0, Op1, Op2);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001207 break;
1208 }
1209 case bitc::CST_CODE_CE_SHUFFLEVEC: { // CE_SHUFFLEVEC: [opval, opval, opval]
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001210 VectorType *OpTy = dyn_cast<VectorType>(CurTy);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001211 if (Record.size() < 3 || OpTy == 0)
Nate Begeman0f123cf2009-02-12 21:28:33 +00001212 return Error("Invalid CE_SHUFFLEVEC record");
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001213 Constant *Op0 = ValueList.getConstantFwdRef(Record[0], OpTy);
1214 Constant *Op1 = ValueList.getConstantFwdRef(Record[1], OpTy);
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001215 Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
Owen Anderson74a77812009-07-07 20:18:58 +00001216 OpTy->getNumElements());
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001217 Constant *Op2 = ValueList.getConstantFwdRef(Record[2], ShufTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001218 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001219 break;
1220 }
Nate Begeman0f123cf2009-02-12 21:28:33 +00001221 case bitc::CST_CODE_CE_SHUFVEC_EX: { // [opty, opval, opval, opval]
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001222 VectorType *RTy = dyn_cast<VectorType>(CurTy);
1223 VectorType *OpTy =
Duncan Sandsf22b7462010-10-28 15:47:26 +00001224 dyn_cast_or_null<VectorType>(getTypeByID(Record[0]));
Nate Begeman0f123cf2009-02-12 21:28:33 +00001225 if (Record.size() < 4 || RTy == 0 || OpTy == 0)
1226 return Error("Invalid CE_SHUFVEC_EX record");
1227 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1228 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001229 Type *ShufTy = VectorType::get(Type::getInt32Ty(Context),
Owen Anderson74a77812009-07-07 20:18:58 +00001230 RTy->getNumElements());
Nate Begeman0f123cf2009-02-12 21:28:33 +00001231 Constant *Op2 = ValueList.getConstantFwdRef(Record[3], ShufTy);
Owen Andersonbaf3c402009-07-29 18:55:55 +00001232 V = ConstantExpr::getShuffleVector(Op0, Op1, Op2);
Nate Begeman0f123cf2009-02-12 21:28:33 +00001233 break;
1234 }
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001235 case bitc::CST_CODE_CE_CMP: { // CE_CMP: [opty, opval, opval, pred]
1236 if (Record.size() < 4) return Error("Invalid CE_CMP record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001237 Type *OpTy = getTypeByID(Record[0]);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001238 if (OpTy == 0) return Error("Invalid CE_CMP record");
1239 Constant *Op0 = ValueList.getConstantFwdRef(Record[1], OpTy);
1240 Constant *Op1 = ValueList.getConstantFwdRef(Record[2], OpTy);
1241
Duncan Sandsb0bc6c32010-02-15 16:12:20 +00001242 if (OpTy->isFPOrFPVectorTy())
Owen Andersonbaf3c402009-07-29 18:55:55 +00001243 V = ConstantExpr::getFCmp(Record[3], Op0, Op1);
Nate Begemanac80ade2008-05-12 19:01:56 +00001244 else
Owen Andersonbaf3c402009-07-29 18:55:55 +00001245 V = ConstantExpr::getICmp(Record[3], Op0, Op1);
Chris Lattnerf581c3b2007-04-24 07:07:11 +00001246 break;
Chris Lattner522b7b12007-04-24 05:48:56 +00001247 }
Chad Rosier581600b2012-09-05 19:00:49 +00001248 // This maintains backward compatibility, pre-asm dialect keywords.
Chad Rosier27b25c22012-09-05 06:28:52 +00001249 // FIXME: Remove with the 4.0 release.
Chad Rosierf16ae582012-09-05 00:56:20 +00001250 case bitc::CST_CODE_INLINEASM_OLD: {
Chris Lattner2bce93a2007-05-06 01:58:20 +00001251 if (Record.size() < 2) return Error("Invalid INLINEASM record");
1252 std::string AsmStr, ConstrStr;
Dale Johannesen43602982009-10-13 20:46:56 +00001253 bool HasSideEffects = Record[0] & 1;
Dale Johannesen8ba2d5b2009-10-21 23:28:00 +00001254 bool IsAlignStack = Record[0] >> 1;
Chris Lattner2bce93a2007-05-06 01:58:20 +00001255 unsigned AsmStrSize = Record[1];
1256 if (2+AsmStrSize >= Record.size())
1257 return Error("Invalid INLINEASM record");
1258 unsigned ConstStrSize = Record[2+AsmStrSize];
1259 if (3+AsmStrSize+ConstStrSize > Record.size())
1260 return Error("Invalid INLINEASM record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001261
Chris Lattner2bce93a2007-05-06 01:58:20 +00001262 for (unsigned i = 0; i != AsmStrSize; ++i)
1263 AsmStr += (char)Record[2+i];
1264 for (unsigned i = 0; i != ConstStrSize; ++i)
1265 ConstrStr += (char)Record[3+AsmStrSize+i];
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001266 PointerType *PTy = cast<PointerType>(CurTy);
Chris Lattner2bce93a2007-05-06 01:58:20 +00001267 V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
Dale Johannesen8ba2d5b2009-10-21 23:28:00 +00001268 AsmStr, ConstrStr, HasSideEffects, IsAlignStack);
Chris Lattner2bce93a2007-05-06 01:58:20 +00001269 break;
1270 }
Chad Rosier581600b2012-09-05 19:00:49 +00001271 // This version adds support for the asm dialect keywords (e.g.,
1272 // inteldialect).
Chad Rosierf16ae582012-09-05 00:56:20 +00001273 case bitc::CST_CODE_INLINEASM: {
1274 if (Record.size() < 2) return Error("Invalid INLINEASM record");
1275 std::string AsmStr, ConstrStr;
1276 bool HasSideEffects = Record[0] & 1;
1277 bool IsAlignStack = (Record[0] >> 1) & 1;
1278 unsigned AsmDialect = Record[0] >> 2;
1279 unsigned AsmStrSize = Record[1];
1280 if (2+AsmStrSize >= Record.size())
1281 return Error("Invalid INLINEASM record");
1282 unsigned ConstStrSize = Record[2+AsmStrSize];
1283 if (3+AsmStrSize+ConstStrSize > Record.size())
1284 return Error("Invalid INLINEASM record");
1285
1286 for (unsigned i = 0; i != AsmStrSize; ++i)
1287 AsmStr += (char)Record[2+i];
1288 for (unsigned i = 0; i != ConstStrSize; ++i)
1289 ConstrStr += (char)Record[3+AsmStrSize+i];
1290 PointerType *PTy = cast<PointerType>(CurTy);
1291 V = InlineAsm::get(cast<FunctionType>(PTy->getElementType()),
1292 AsmStr, ConstrStr, HasSideEffects, IsAlignStack,
Chad Rosier581600b2012-09-05 19:00:49 +00001293 InlineAsm::AsmDialect(AsmDialect));
Chad Rosierf16ae582012-09-05 00:56:20 +00001294 break;
1295 }
Chris Lattner50b136d2009-10-28 05:53:48 +00001296 case bitc::CST_CODE_BLOCKADDRESS:{
1297 if (Record.size() < 3) return Error("Invalid CE_BLOCKADDRESS record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001298 Type *FnTy = getTypeByID(Record[0]);
Chris Lattner50b136d2009-10-28 05:53:48 +00001299 if (FnTy == 0) return Error("Invalid CE_BLOCKADDRESS record");
1300 Function *Fn =
1301 dyn_cast_or_null<Function>(ValueList.getConstantFwdRef(Record[1],FnTy));
1302 if (Fn == 0) return Error("Invalid CE_BLOCKADDRESS record");
1303
1304 GlobalVariable *FwdRef = new GlobalVariable(*Fn->getParent(),
1305 Type::getInt8Ty(Context),
1306 false, GlobalValue::InternalLinkage,
1307 0, "");
1308 BlockAddrFwdRefs[Fn].push_back(std::make_pair(Record[2], FwdRef));
1309 V = FwdRef;
1310 break;
1311 }
Chris Lattnere16504e2007-04-24 03:30:34 +00001312 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001313
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001314 ValueList.AssignValue(V, NextCstNo);
Chris Lattner522b7b12007-04-24 05:48:56 +00001315 ++NextCstNo;
Chris Lattnere16504e2007-04-24 03:30:34 +00001316 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001317
Chris Lattnerea693df2008-08-21 02:34:16 +00001318 if (NextCstNo != ValueList.size())
1319 return Error("Invalid constant reference!");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001320
Chris Lattnerea693df2008-08-21 02:34:16 +00001321 if (Stream.ReadBlockEnd())
1322 return Error("Error at end of constants block");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001323
Chris Lattnerea693df2008-08-21 02:34:16 +00001324 // Once all the constants have been read, go through and resolve forward
1325 // references.
1326 ValueList.ResolveConstantForwardRefs();
1327 return false;
Chris Lattnere16504e2007-04-24 03:30:34 +00001328}
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001329
Chad Rosiercbbb0962011-12-07 21:44:12 +00001330bool BitcodeReader::ParseUseLists() {
1331 if (Stream.EnterSubBlock(bitc::USELIST_BLOCK_ID))
1332 return Error("Malformed block record");
1333
1334 SmallVector<uint64_t, 64> Record;
1335
1336 // Read all the records.
1337 while (1) {
1338 unsigned Code = Stream.ReadCode();
1339 if (Code == bitc::END_BLOCK) {
1340 if (Stream.ReadBlockEnd())
1341 return Error("Error at end of use-list table block");
1342 return false;
1343 }
1344
1345 if (Code == bitc::ENTER_SUBBLOCK) {
1346 // No known subblocks, always skip them.
1347 Stream.ReadSubBlockID();
1348 if (Stream.SkipBlock())
1349 return Error("Malformed block record");
1350 continue;
1351 }
1352
1353 if (Code == bitc::DEFINE_ABBREV) {
1354 Stream.ReadAbbrevRecord();
1355 continue;
1356 }
1357
1358 // Read a use list record.
1359 Record.clear();
1360 switch (Stream.ReadRecord(Code, Record)) {
1361 default: // Default behavior: unknown type.
1362 break;
1363 case bitc::USELIST_CODE_ENTRY: { // USELIST_CODE_ENTRY: TBD.
1364 unsigned RecordLength = Record.size();
1365 if (RecordLength < 1)
1366 return Error ("Invalid UseList reader!");
1367 UseListRecords.push_back(Record);
1368 break;
1369 }
1370 }
1371 }
1372}
1373
Chris Lattner980e5aa2007-05-01 05:52:21 +00001374/// RememberAndSkipFunctionBody - When we see the block for a function body,
1375/// remember where it is and then skip it. This lets us lazily deserialize the
1376/// functions.
1377bool BitcodeReader::RememberAndSkipFunctionBody() {
Chris Lattner48f84872007-05-01 04:59:48 +00001378 // Get the function we are talking about.
1379 if (FunctionsWithBodies.empty())
1380 return Error("Insufficient function protos");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001381
Chris Lattner48f84872007-05-01 04:59:48 +00001382 Function *Fn = FunctionsWithBodies.back();
1383 FunctionsWithBodies.pop_back();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001384
Chris Lattner48f84872007-05-01 04:59:48 +00001385 // Save the current stream state.
1386 uint64_t CurBit = Stream.GetCurrentBitNo();
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00001387 DeferredFunctionInfo[Fn] = CurBit;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001388
Chris Lattner48f84872007-05-01 04:59:48 +00001389 // Skip over the function block for now.
1390 if (Stream.SkipBlock())
1391 return Error("Malformed block record");
1392 return false;
1393}
1394
Derek Schuff2ea93872012-02-06 22:30:29 +00001395bool BitcodeReader::GlobalCleanup() {
1396 // Patch the initializers for globals and aliases up.
1397 ResolveGlobalAndAliasInits();
1398 if (!GlobalInits.empty() || !AliasInits.empty())
1399 return Error("Malformed global initializer set");
1400
1401 // Look for intrinsic functions which need to be upgraded at some point
1402 for (Module::iterator FI = TheModule->begin(), FE = TheModule->end();
1403 FI != FE; ++FI) {
1404 Function *NewFn;
1405 if (UpgradeIntrinsicFunction(FI, NewFn))
1406 UpgradedIntrinsics.push_back(std::make_pair(FI, NewFn));
1407 }
1408
1409 // Look for global variables which need to be renamed.
1410 for (Module::global_iterator
1411 GI = TheModule->global_begin(), GE = TheModule->global_end();
1412 GI != GE; ++GI)
1413 UpgradeGlobalVariable(GI);
1414 // Force deallocation of memory for these vectors to favor the client that
1415 // want lazy deserialization.
1416 std::vector<std::pair<GlobalVariable*, unsigned> >().swap(GlobalInits);
1417 std::vector<std::pair<GlobalAlias*, unsigned> >().swap(AliasInits);
1418 return false;
1419}
1420
1421bool BitcodeReader::ParseModule(bool Resume) {
1422 if (Resume)
1423 Stream.JumpToBit(NextUnreadBit);
1424 else if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001425 return Error("Malformed block record");
1426
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001427 SmallVector<uint64_t, 64> Record;
1428 std::vector<std::string> SectionTable;
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001429 std::vector<std::string> GCTable;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001430
1431 // Read all the records for this module.
1432 while (!Stream.AtEndOfStream()) {
1433 unsigned Code = Stream.ReadCode();
Chris Lattnere84bcb92007-04-24 00:21:45 +00001434 if (Code == bitc::END_BLOCK) {
Chris Lattner980e5aa2007-05-01 05:52:21 +00001435 if (Stream.ReadBlockEnd())
1436 return Error("Error at end of module block");
1437
Derek Schuff2ea93872012-02-06 22:30:29 +00001438 return GlobalCleanup();
Chris Lattnere84bcb92007-04-24 00:21:45 +00001439 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001440
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001441 if (Code == bitc::ENTER_SUBBLOCK) {
1442 switch (Stream.ReadSubBlockID()) {
1443 default: // Skip unknown content.
1444 if (Stream.SkipBlock())
1445 return Error("Malformed block record");
1446 break;
Chris Lattner3f799802007-05-05 18:57:30 +00001447 case bitc::BLOCKINFO_BLOCK_ID:
1448 if (Stream.ReadBlockInfoBlock())
1449 return Error("Malformed BlockInfoBlock");
1450 break;
Chris Lattner48c85b82007-05-04 03:30:17 +00001451 case bitc::PARAMATTR_BLOCK_ID:
Devang Patel05988662008-09-25 21:00:45 +00001452 if (ParseAttributeBlock())
Chris Lattner48c85b82007-05-04 03:30:17 +00001453 return true;
1454 break;
Chris Lattner1afcace2011-07-09 17:41:24 +00001455 case bitc::TYPE_BLOCK_ID_NEW:
Chris Lattner86697142007-05-01 05:01:34 +00001456 if (ParseTypeTable())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001457 return true;
1458 break;
Chris Lattner0b2482a2007-04-23 21:26:05 +00001459 case bitc::VALUE_SYMTAB_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001460 if (ParseValueSymbolTable())
Chris Lattner0b2482a2007-04-23 21:26:05 +00001461 return true;
Derek Schuff2ea93872012-02-06 22:30:29 +00001462 SeenValueSymbolTable = true;
Chris Lattner0b2482a2007-04-23 21:26:05 +00001463 break;
Chris Lattnere16504e2007-04-24 03:30:34 +00001464 case bitc::CONSTANTS_BLOCK_ID:
Chris Lattner86697142007-05-01 05:01:34 +00001465 if (ParseConstants() || ResolveGlobalAndAliasInits())
Chris Lattnere16504e2007-04-24 03:30:34 +00001466 return true;
1467 break;
Devang Patele54abc92009-07-22 17:43:22 +00001468 case bitc::METADATA_BLOCK_ID:
1469 if (ParseMetadata())
1470 return true;
1471 break;
Chris Lattner48f84872007-05-01 04:59:48 +00001472 case bitc::FUNCTION_BLOCK_ID:
1473 // If this is the first function body we've seen, reverse the
1474 // FunctionsWithBodies list.
Derek Schuff2ea93872012-02-06 22:30:29 +00001475 if (!SeenFirstFunctionBody) {
Chris Lattner48f84872007-05-01 04:59:48 +00001476 std::reverse(FunctionsWithBodies.begin(), FunctionsWithBodies.end());
Derek Schuff2ea93872012-02-06 22:30:29 +00001477 if (GlobalCleanup())
1478 return true;
1479 SeenFirstFunctionBody = true;
Chris Lattner48f84872007-05-01 04:59:48 +00001480 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001481
Chris Lattner980e5aa2007-05-01 05:52:21 +00001482 if (RememberAndSkipFunctionBody())
Chris Lattner48f84872007-05-01 04:59:48 +00001483 return true;
Derek Schuff2ea93872012-02-06 22:30:29 +00001484 // For streaming bitcode, suspend parsing when we reach the function
1485 // bodies. Subsequent materialization calls will resume it when
1486 // necessary. For streaming, the function bodies must be at the end of
1487 // the bitcode. If the bitcode file is old, the symbol table will be
1488 // at the end instead and will not have been seen yet. In this case,
1489 // just finish the parse now.
1490 if (LazyStreamer && SeenValueSymbolTable) {
1491 NextUnreadBit = Stream.GetCurrentBitNo();
1492 return false;
1493 }
Chris Lattner48f84872007-05-01 04:59:48 +00001494 break;
Chad Rosiercbbb0962011-12-07 21:44:12 +00001495 case bitc::USELIST_BLOCK_ID:
1496 if (ParseUseLists())
1497 return true;
1498 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001499 }
1500 continue;
1501 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001502
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001503 if (Code == bitc::DEFINE_ABBREV) {
Chris Lattnerd127c1b2007-04-23 18:58:34 +00001504 Stream.ReadAbbrevRecord();
1505 continue;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001506 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001507
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001508 // Read a record.
1509 switch (Stream.ReadRecord(Code, Record)) {
1510 default: break; // Default behavior, ignore unknown content.
1511 case bitc::MODULE_CODE_VERSION: // VERSION: [version#]
1512 if (Record.size() < 1)
1513 return Error("Malformed MODULE_CODE_VERSION");
1514 // Only version #0 is supported so far.
1515 if (Record[0] != 0)
1516 return Error("Unknown bitstream version!");
1517 break;
Chris Lattner15e6d172007-05-04 19:11:41 +00001518 case bitc::MODULE_CODE_TRIPLE: { // TRIPLE: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001519 std::string S;
1520 if (ConvertToString(Record, 0, S))
1521 return Error("Invalid MODULE_CODE_TRIPLE record");
1522 TheModule->setTargetTriple(S);
1523 break;
1524 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001525 case bitc::MODULE_CODE_DATALAYOUT: { // DATALAYOUT: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001526 std::string S;
1527 if (ConvertToString(Record, 0, S))
1528 return Error("Invalid MODULE_CODE_DATALAYOUT record");
1529 TheModule->setDataLayout(S);
1530 break;
1531 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001532 case bitc::MODULE_CODE_ASM: { // ASM: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001533 std::string S;
1534 if (ConvertToString(Record, 0, S))
1535 return Error("Invalid MODULE_CODE_ASM record");
1536 TheModule->setModuleInlineAsm(S);
1537 break;
1538 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001539 case bitc::MODULE_CODE_DEPLIB: { // DEPLIB: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001540 std::string S;
1541 if (ConvertToString(Record, 0, S))
1542 return Error("Invalid MODULE_CODE_DEPLIB record");
1543 TheModule->addLibrary(S);
1544 break;
1545 }
Chris Lattner15e6d172007-05-04 19:11:41 +00001546 case bitc::MODULE_CODE_SECTIONNAME: { // SECTIONNAME: [strchr x N]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001547 std::string S;
1548 if (ConvertToString(Record, 0, S))
1549 return Error("Invalid MODULE_CODE_SECTIONNAME record");
1550 SectionTable.push_back(S);
1551 break;
1552 }
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001553 case bitc::MODULE_CODE_GCNAME: { // SECTIONNAME: [strchr x N]
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001554 std::string S;
1555 if (ConvertToString(Record, 0, S))
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001556 return Error("Invalid MODULE_CODE_GCNAME record");
1557 GCTable.push_back(S);
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001558 break;
1559 }
Christopher Lambfe63fb92007-12-11 08:59:05 +00001560 // GLOBALVAR: [pointer type, isconst, initid,
Rafael Espindolabea46262011-01-08 16:42:36 +00001561 // linkage, alignment, section, visibility, threadlocal,
1562 // unnamed_addr]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001563 case bitc::MODULE_CODE_GLOBALVAR: {
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001564 if (Record.size() < 6)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001565 return Error("Invalid MODULE_CODE_GLOBALVAR record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001566 Type *Ty = getTypeByID(Record[0]);
Duncan Sandsf22b7462010-10-28 15:47:26 +00001567 if (!Ty) return Error("Invalid MODULE_CODE_GLOBALVAR record");
Duncan Sands1df98592010-02-16 11:11:14 +00001568 if (!Ty->isPointerTy())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001569 return Error("Global not a pointer type!");
Christopher Lambfe63fb92007-12-11 08:59:05 +00001570 unsigned AddressSpace = cast<PointerType>(Ty)->getAddressSpace();
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001571 Ty = cast<PointerType>(Ty)->getElementType();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001572
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001573 bool isConstant = Record[1];
1574 GlobalValue::LinkageTypes Linkage = GetDecodedLinkage(Record[3]);
1575 unsigned Alignment = (1 << Record[4]) >> 1;
1576 std::string Section;
1577 if (Record[5]) {
1578 if (Record[5]-1 >= SectionTable.size())
1579 return Error("Invalid section ID");
1580 Section = SectionTable[Record[5]-1];
1581 }
Chris Lattner36d5e7d2007-04-23 16:04:05 +00001582 GlobalValue::VisibilityTypes Visibility = GlobalValue::DefaultVisibility;
Chris Lattner5f32c012007-05-06 19:27:46 +00001583 if (Record.size() > 6)
1584 Visibility = GetDecodedVisibility(Record[6]);
Hans Wennborgce718ff2012-06-23 11:37:03 +00001585
1586 GlobalVariable::ThreadLocalMode TLM = GlobalVariable::NotThreadLocal;
Chris Lattner5f32c012007-05-06 19:27:46 +00001587 if (Record.size() > 7)
Hans Wennborgce718ff2012-06-23 11:37:03 +00001588 TLM = GetDecodedThreadLocalMode(Record[7]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001589
Rafael Espindolabea46262011-01-08 16:42:36 +00001590 bool UnnamedAddr = false;
1591 if (Record.size() > 8)
1592 UnnamedAddr = Record[8];
1593
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001594 GlobalVariable *NewGV =
Daniel Dunbara279bc32009-09-20 02:20:51 +00001595 new GlobalVariable(*TheModule, Ty, isConstant, Linkage, 0, "", 0,
Hans Wennborgce718ff2012-06-23 11:37:03 +00001596 TLM, AddressSpace);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001597 NewGV->setAlignment(Alignment);
1598 if (!Section.empty())
1599 NewGV->setSection(Section);
1600 NewGV->setVisibility(Visibility);
Rafael Espindolabea46262011-01-08 16:42:36 +00001601 NewGV->setUnnamedAddr(UnnamedAddr);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001602
Chris Lattner0b2482a2007-04-23 21:26:05 +00001603 ValueList.push_back(NewGV);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001604
Chris Lattner6dbfd7b2007-04-24 00:18:21 +00001605 // Remember which value to use for the global initializer.
1606 if (unsigned InitID = Record[2])
1607 GlobalInits.push_back(std::make_pair(NewGV, InitID-1));
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001608 break;
1609 }
Chris Lattnera9bb7132007-05-08 05:38:01 +00001610 // FUNCTION: [type, callingconv, isproto, linkage, paramattr,
Rafael Espindolabea46262011-01-08 16:42:36 +00001611 // alignment, section, visibility, gc, unnamed_addr]
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001612 case bitc::MODULE_CODE_FUNCTION: {
Chris Lattnera9bb7132007-05-08 05:38:01 +00001613 if (Record.size() < 8)
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001614 return Error("Invalid MODULE_CODE_FUNCTION record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001615 Type *Ty = getTypeByID(Record[0]);
Duncan Sandsf22b7462010-10-28 15:47:26 +00001616 if (!Ty) return Error("Invalid MODULE_CODE_FUNCTION record");
Duncan Sands1df98592010-02-16 11:11:14 +00001617 if (!Ty->isPointerTy())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001618 return Error("Function not a pointer type!");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001619 FunctionType *FTy =
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001620 dyn_cast<FunctionType>(cast<PointerType>(Ty)->getElementType());
1621 if (!FTy)
1622 return Error("Function not a pointer to function type!");
1623
Gabor Greif051a9502008-04-06 20:25:17 +00001624 Function *Func = Function::Create(FTy, GlobalValue::ExternalLinkage,
1625 "", TheModule);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001626
Sandeep Patel65c3c8f2009-09-02 08:44:58 +00001627 Func->setCallingConv(static_cast<CallingConv::ID>(Record[1]));
Chris Lattner48f84872007-05-01 04:59:48 +00001628 bool isProto = Record[2];
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001629 Func->setLinkage(GetDecodedLinkage(Record[3]));
Devang Patel05988662008-09-25 21:00:45 +00001630 Func->setAttributes(getAttributes(Record[4]));
Daniel Dunbara279bc32009-09-20 02:20:51 +00001631
Chris Lattnera9bb7132007-05-08 05:38:01 +00001632 Func->setAlignment((1 << Record[5]) >> 1);
1633 if (Record[6]) {
1634 if (Record[6]-1 >= SectionTable.size())
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001635 return Error("Invalid section ID");
Chris Lattnera9bb7132007-05-08 05:38:01 +00001636 Func->setSection(SectionTable[Record[6]-1]);
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001637 }
Chris Lattnera9bb7132007-05-08 05:38:01 +00001638 Func->setVisibility(GetDecodedVisibility(Record[7]));
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001639 if (Record.size() > 8 && Record[8]) {
Gordon Henriksen5eca0752008-08-17 18:44:35 +00001640 if (Record[8]-1 > GCTable.size())
1641 return Error("Invalid GC ID");
1642 Func->setGC(GCTable[Record[8]-1].c_str());
Gordon Henriksen80a75bf2007-12-10 03:18:06 +00001643 }
Rafael Espindolabea46262011-01-08 16:42:36 +00001644 bool UnnamedAddr = false;
1645 if (Record.size() > 9)
1646 UnnamedAddr = Record[9];
1647 Func->setUnnamedAddr(UnnamedAddr);
Chris Lattner0b2482a2007-04-23 21:26:05 +00001648 ValueList.push_back(Func);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001649
Chris Lattner48f84872007-05-01 04:59:48 +00001650 // If this is a function with a body, remember the prototype we are
1651 // creating now, so that we can match up the body with them later.
Derek Schuff2ea93872012-02-06 22:30:29 +00001652 if (!isProto) {
Chris Lattner48f84872007-05-01 04:59:48 +00001653 FunctionsWithBodies.push_back(Func);
Derek Schuff2ea93872012-02-06 22:30:29 +00001654 if (LazyStreamer) DeferredFunctionInfo[Func] = 0;
1655 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001656 break;
1657 }
Anton Korobeynikov91342d82008-03-12 00:49:19 +00001658 // ALIAS: [alias type, aliasee val#, linkage]
Anton Korobeynikovf8342b92008-03-11 21:40:17 +00001659 // ALIAS: [alias type, aliasee val#, linkage, visibility]
Chris Lattner198f34a2007-04-26 03:27:58 +00001660 case bitc::MODULE_CODE_ALIAS: {
Chris Lattner07d98b42007-04-26 02:46:40 +00001661 if (Record.size() < 3)
1662 return Error("Invalid MODULE_ALIAS record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00001663 Type *Ty = getTypeByID(Record[0]);
Duncan Sandsf22b7462010-10-28 15:47:26 +00001664 if (!Ty) return Error("Invalid MODULE_ALIAS record");
Duncan Sands1df98592010-02-16 11:11:14 +00001665 if (!Ty->isPointerTy())
Chris Lattner07d98b42007-04-26 02:46:40 +00001666 return Error("Function not a pointer type!");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001667
Chris Lattner07d98b42007-04-26 02:46:40 +00001668 GlobalAlias *NewGA = new GlobalAlias(Ty, GetDecodedLinkage(Record[2]),
1669 "", 0, TheModule);
Anton Korobeynikov91342d82008-03-12 00:49:19 +00001670 // Old bitcode files didn't have visibility field.
1671 if (Record.size() > 3)
1672 NewGA->setVisibility(GetDecodedVisibility(Record[3]));
Chris Lattner07d98b42007-04-26 02:46:40 +00001673 ValueList.push_back(NewGA);
1674 AliasInits.push_back(std::make_pair(NewGA, Record[1]));
1675 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001676 }
Chris Lattner198f34a2007-04-26 03:27:58 +00001677 /// MODULE_CODE_PURGEVALS: [numvals]
1678 case bitc::MODULE_CODE_PURGEVALS:
1679 // Trim down the value list to the specified size.
1680 if (Record.size() < 1 || Record[0] > ValueList.size())
1681 return Error("Invalid MODULE_PURGEVALS record");
1682 ValueList.shrinkTo(Record[0]);
1683 break;
1684 }
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001685 Record.clear();
1686 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001687
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001688 return Error("Premature end of bitstream");
1689}
1690
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00001691bool BitcodeReader::ParseBitcodeInto(Module *M) {
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001692 TheModule = 0;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001693
Derek Schuff2ea93872012-02-06 22:30:29 +00001694 if (InitStream()) return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00001695
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001696 // Sniff for the signature.
1697 if (Stream.Read(8) != 'B' ||
1698 Stream.Read(8) != 'C' ||
1699 Stream.Read(4) != 0x0 ||
1700 Stream.Read(4) != 0xC ||
1701 Stream.Read(4) != 0xE ||
1702 Stream.Read(4) != 0xD)
1703 return Error("Invalid bitcode signature");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001704
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001705 // We expect a number of well-defined blocks, though we don't necessarily
1706 // need to understand them all.
1707 while (!Stream.AtEndOfStream()) {
1708 unsigned Code = Stream.ReadCode();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001709
Rafael Espindolac9687b32011-05-26 18:59:54 +00001710 if (Code != bitc::ENTER_SUBBLOCK) {
1711
Chad Rosier6ff9aa22011-08-09 22:23:40 +00001712 // The ranlib in xcode 4 will align archive members by appending newlines
1713 // to the end of them. If this file size is a multiple of 4 but not 8, we
1714 // have to read and ignore these final 4 bytes :-(
Rafael Espindolac9687b32011-05-26 18:59:54 +00001715 if (Stream.GetAbbrevIDWidth() == 2 && Code == 2 &&
1716 Stream.Read(6) == 2 && Stream.Read(24) == 0xa0a0a &&
Bill Wendling2127c9b2012-07-19 00:15:11 +00001717 Stream.AtEndOfStream())
Rafael Espindolac9687b32011-05-26 18:59:54 +00001718 return false;
1719
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001720 return Error("Invalid record at top-level");
Rafael Espindolac9687b32011-05-26 18:59:54 +00001721 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001722
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001723 unsigned BlockID = Stream.ReadSubBlockID();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001724
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001725 // We only know the MODULE subblock ID.
Chris Lattnere17b6582007-05-05 00:17:00 +00001726 switch (BlockID) {
1727 case bitc::BLOCKINFO_BLOCK_ID:
1728 if (Stream.ReadBlockInfoBlock())
1729 return Error("Malformed BlockInfoBlock");
1730 break;
1731 case bitc::MODULE_BLOCK_ID:
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00001732 // Reject multiple MODULE_BLOCK's in a single bitstream.
1733 if (TheModule)
1734 return Error("Multiple MODULE_BLOCKs in same stream");
1735 TheModule = M;
Derek Schuff2ea93872012-02-06 22:30:29 +00001736 if (ParseModule(false))
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001737 return true;
Derek Schuff2ea93872012-02-06 22:30:29 +00001738 if (LazyStreamer) return false;
Chris Lattnere17b6582007-05-05 00:17:00 +00001739 break;
1740 default:
1741 if (Stream.SkipBlock())
1742 return Error("Malformed block record");
1743 break;
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001744 }
1745 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001746
Chris Lattnercaee0dc2007-04-22 06:23:29 +00001747 return false;
1748}
Chris Lattnerc453f762007-04-29 07:54:31 +00001749
Bill Wendling34711742010-10-06 01:22:42 +00001750bool BitcodeReader::ParseModuleTriple(std::string &Triple) {
1751 if (Stream.EnterSubBlock(bitc::MODULE_BLOCK_ID))
1752 return Error("Malformed block record");
1753
1754 SmallVector<uint64_t, 64> Record;
1755
1756 // Read all the records for this module.
1757 while (!Stream.AtEndOfStream()) {
1758 unsigned Code = Stream.ReadCode();
1759 if (Code == bitc::END_BLOCK) {
1760 if (Stream.ReadBlockEnd())
1761 return Error("Error at end of module block");
1762
1763 return false;
1764 }
1765
1766 if (Code == bitc::ENTER_SUBBLOCK) {
1767 switch (Stream.ReadSubBlockID()) {
1768 default: // Skip unknown content.
1769 if (Stream.SkipBlock())
1770 return Error("Malformed block record");
1771 break;
1772 }
1773 continue;
1774 }
1775
1776 if (Code == bitc::DEFINE_ABBREV) {
1777 Stream.ReadAbbrevRecord();
1778 continue;
1779 }
1780
1781 // Read a record.
1782 switch (Stream.ReadRecord(Code, Record)) {
1783 default: break; // Default behavior, ignore unknown content.
1784 case bitc::MODULE_CODE_VERSION: // VERSION: [version#]
1785 if (Record.size() < 1)
1786 return Error("Malformed MODULE_CODE_VERSION");
1787 // Only version #0 is supported so far.
1788 if (Record[0] != 0)
1789 return Error("Unknown bitstream version!");
1790 break;
1791 case bitc::MODULE_CODE_TRIPLE: { // TRIPLE: [strchr x N]
1792 std::string S;
1793 if (ConvertToString(Record, 0, S))
1794 return Error("Invalid MODULE_CODE_TRIPLE record");
1795 Triple = S;
1796 break;
1797 }
1798 }
1799 Record.clear();
1800 }
1801
1802 return Error("Premature end of bitstream");
1803}
1804
1805bool BitcodeReader::ParseTriple(std::string &Triple) {
Derek Schuff2ea93872012-02-06 22:30:29 +00001806 if (InitStream()) return true;
Bill Wendling34711742010-10-06 01:22:42 +00001807
1808 // Sniff for the signature.
1809 if (Stream.Read(8) != 'B' ||
1810 Stream.Read(8) != 'C' ||
1811 Stream.Read(4) != 0x0 ||
1812 Stream.Read(4) != 0xC ||
1813 Stream.Read(4) != 0xE ||
1814 Stream.Read(4) != 0xD)
1815 return Error("Invalid bitcode signature");
1816
1817 // We expect a number of well-defined blocks, though we don't necessarily
1818 // need to understand them all.
1819 while (!Stream.AtEndOfStream()) {
1820 unsigned Code = Stream.ReadCode();
1821
1822 if (Code != bitc::ENTER_SUBBLOCK)
1823 return Error("Invalid record at top-level");
1824
1825 unsigned BlockID = Stream.ReadSubBlockID();
1826
1827 // We only know the MODULE subblock ID.
1828 switch (BlockID) {
1829 case bitc::MODULE_BLOCK_ID:
1830 if (ParseModuleTriple(Triple))
1831 return true;
1832 break;
1833 default:
1834 if (Stream.SkipBlock())
1835 return Error("Malformed block record");
1836 break;
1837 }
1838 }
1839
1840 return false;
1841}
1842
Devang Patele8e02132009-09-18 19:26:43 +00001843/// ParseMetadataAttachment - Parse metadata attachments.
1844bool BitcodeReader::ParseMetadataAttachment() {
1845 if (Stream.EnterSubBlock(bitc::METADATA_ATTACHMENT_ID))
1846 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001847
Devang Patele8e02132009-09-18 19:26:43 +00001848 SmallVector<uint64_t, 64> Record;
1849 while(1) {
1850 unsigned Code = Stream.ReadCode();
1851 if (Code == bitc::END_BLOCK) {
1852 if (Stream.ReadBlockEnd())
Daniel Dunbara279bc32009-09-20 02:20:51 +00001853 return Error("Error at end of PARAMATTR block");
Devang Patele8e02132009-09-18 19:26:43 +00001854 break;
1855 }
1856 if (Code == bitc::DEFINE_ABBREV) {
1857 Stream.ReadAbbrevRecord();
1858 continue;
1859 }
1860 // Read a metadata attachment record.
1861 Record.clear();
1862 switch (Stream.ReadRecord(Code, Record)) {
1863 default: // Default behavior: ignore.
1864 break;
Chris Lattner9d61dd92011-06-17 17:50:30 +00001865 case bitc::METADATA_ATTACHMENT: {
Devang Patele8e02132009-09-18 19:26:43 +00001866 unsigned RecordLength = Record.size();
1867 if (Record.empty() || (RecordLength - 1) % 2 == 1)
Daniel Dunbara279bc32009-09-20 02:20:51 +00001868 return Error ("Invalid METADATA_ATTACHMENT reader!");
Devang Patele8e02132009-09-18 19:26:43 +00001869 Instruction *Inst = InstructionList[Record[0]];
1870 for (unsigned i = 1; i != RecordLength; i = i+2) {
Devang Patela2148402009-09-28 21:14:55 +00001871 unsigned Kind = Record[i];
Dan Gohman19538d12010-07-20 21:42:28 +00001872 DenseMap<unsigned, unsigned>::iterator I =
1873 MDKindMap.find(Kind);
1874 if (I == MDKindMap.end())
1875 return Error("Invalid metadata kind ID");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001876 Value *Node = MDValueList.getValueFwdRef(Record[i+1]);
Dan Gohman19538d12010-07-20 21:42:28 +00001877 Inst->setMetadata(I->second, cast<MDNode>(Node));
Devang Patele8e02132009-09-18 19:26:43 +00001878 }
1879 break;
1880 }
1881 }
1882 }
1883 return false;
1884}
Chris Lattner48f84872007-05-01 04:59:48 +00001885
Chris Lattner980e5aa2007-05-01 05:52:21 +00001886/// ParseFunctionBody - Lazily parse the specified function body block.
1887bool BitcodeReader::ParseFunctionBody(Function *F) {
Chris Lattnere17b6582007-05-05 00:17:00 +00001888 if (Stream.EnterSubBlock(bitc::FUNCTION_BLOCK_ID))
Chris Lattner980e5aa2007-05-01 05:52:21 +00001889 return Error("Malformed block record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00001890
Nick Lewycky9a49f152010-02-25 08:30:17 +00001891 InstructionList.clear();
Chris Lattner980e5aa2007-05-01 05:52:21 +00001892 unsigned ModuleValueListSize = ValueList.size();
Dan Gohman69813832010-08-25 20:22:53 +00001893 unsigned ModuleMDValueListSize = MDValueList.size();
Daniel Dunbara279bc32009-09-20 02:20:51 +00001894
Chris Lattner980e5aa2007-05-01 05:52:21 +00001895 // Add all the function arguments to the value table.
1896 for(Function::arg_iterator I = F->arg_begin(), E = F->arg_end(); I != E; ++I)
1897 ValueList.push_back(I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00001898
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001899 unsigned NextValueNo = ValueList.size();
Chris Lattner231cbcb2007-05-02 04:27:25 +00001900 BasicBlock *CurBB = 0;
1901 unsigned CurBBNo = 0;
1902
Chris Lattnera6245242010-04-03 02:17:50 +00001903 DebugLoc LastLoc;
1904
Chris Lattner980e5aa2007-05-01 05:52:21 +00001905 // Read all the records.
1906 SmallVector<uint64_t, 64> Record;
1907 while (1) {
1908 unsigned Code = Stream.ReadCode();
1909 if (Code == bitc::END_BLOCK) {
1910 if (Stream.ReadBlockEnd())
1911 return Error("Error at end of function block");
1912 break;
1913 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001914
Chris Lattner980e5aa2007-05-01 05:52:21 +00001915 if (Code == bitc::ENTER_SUBBLOCK) {
1916 switch (Stream.ReadSubBlockID()) {
1917 default: // Skip unknown content.
1918 if (Stream.SkipBlock())
1919 return Error("Malformed block record");
1920 break;
1921 case bitc::CONSTANTS_BLOCK_ID:
1922 if (ParseConstants()) return true;
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001923 NextValueNo = ValueList.size();
Chris Lattner980e5aa2007-05-01 05:52:21 +00001924 break;
1925 case bitc::VALUE_SYMTAB_BLOCK_ID:
1926 if (ParseValueSymbolTable()) return true;
1927 break;
Devang Patele8e02132009-09-18 19:26:43 +00001928 case bitc::METADATA_ATTACHMENT_ID:
Daniel Dunbara279bc32009-09-20 02:20:51 +00001929 if (ParseMetadataAttachment()) return true;
1930 break;
Victor Hernandezfab9e99c2010-01-13 19:34:08 +00001931 case bitc::METADATA_BLOCK_ID:
1932 if (ParseMetadata()) return true;
1933 break;
Chris Lattner980e5aa2007-05-01 05:52:21 +00001934 }
1935 continue;
1936 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001937
Chris Lattner980e5aa2007-05-01 05:52:21 +00001938 if (Code == bitc::DEFINE_ABBREV) {
1939 Stream.ReadAbbrevRecord();
1940 continue;
1941 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00001942
Chris Lattner980e5aa2007-05-01 05:52:21 +00001943 // Read a record.
1944 Record.clear();
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001945 Instruction *I = 0;
Dan Gohman1224c382009-07-20 21:19:07 +00001946 unsigned BitCode = Stream.ReadRecord(Code, Record);
1947 switch (BitCode) {
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001948 default: // Default behavior: reject
1949 return Error("Unknown instruction");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001950 case bitc::FUNC_CODE_DECLAREBLOCKS: // DECLAREBLOCKS: [nblocks]
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001951 if (Record.size() < 1 || Record[0] == 0)
1952 return Error("Invalid DECLAREBLOCKS record");
Chris Lattner980e5aa2007-05-01 05:52:21 +00001953 // Create all the basic blocks for the function.
Chris Lattnerf61e6452007-05-03 22:09:51 +00001954 FunctionBBs.resize(Record[0]);
Chris Lattner980e5aa2007-05-01 05:52:21 +00001955 for (unsigned i = 0, e = FunctionBBs.size(); i != e; ++i)
Owen Anderson1d0be152009-08-13 21:58:54 +00001956 FunctionBBs[i] = BasicBlock::Create(Context, "", F);
Chris Lattnera7c49aa2007-05-01 07:01:57 +00001957 CurBB = FunctionBBs[0];
1958 continue;
Chris Lattnera6245242010-04-03 02:17:50 +00001959
1960 case bitc::FUNC_CODE_DEBUG_LOC_AGAIN: // DEBUG_LOC_AGAIN
1961 // This record indicates that the last instruction is at the same
1962 // location as the previous instruction with a location.
1963 I = 0;
1964
1965 // Get the last instruction emitted.
1966 if (CurBB && !CurBB->empty())
1967 I = &CurBB->back();
1968 else if (CurBBNo && FunctionBBs[CurBBNo-1] &&
1969 !FunctionBBs[CurBBNo-1]->empty())
1970 I = &FunctionBBs[CurBBNo-1]->back();
1971
1972 if (I == 0) return Error("Invalid DEBUG_LOC_AGAIN record");
1973 I->setDebugLoc(LastLoc);
1974 I = 0;
1975 continue;
1976
Chris Lattner4f6bab92011-06-17 18:17:37 +00001977 case bitc::FUNC_CODE_DEBUG_LOC: { // DEBUG_LOC: [line, col, scope, ia]
Chris Lattnera6245242010-04-03 02:17:50 +00001978 I = 0; // Get the last instruction emitted.
1979 if (CurBB && !CurBB->empty())
1980 I = &CurBB->back();
1981 else if (CurBBNo && FunctionBBs[CurBBNo-1] &&
1982 !FunctionBBs[CurBBNo-1]->empty())
1983 I = &FunctionBBs[CurBBNo-1]->back();
1984 if (I == 0 || Record.size() < 4)
1985 return Error("Invalid FUNC_CODE_DEBUG_LOC record");
1986
1987 unsigned Line = Record[0], Col = Record[1];
1988 unsigned ScopeID = Record[2], IAID = Record[3];
1989
1990 MDNode *Scope = 0, *IA = 0;
1991 if (ScopeID) Scope = cast<MDNode>(MDValueList.getValueFwdRef(ScopeID-1));
1992 if (IAID) IA = cast<MDNode>(MDValueList.getValueFwdRef(IAID-1));
1993 LastLoc = DebugLoc::get(Line, Col, Scope, IA);
1994 I->setDebugLoc(LastLoc);
1995 I = 0;
1996 continue;
1997 }
1998
Chris Lattnerabfbf852007-05-06 00:21:25 +00001999 case bitc::FUNC_CODE_INST_BINOP: { // BINOP: [opval, ty, opval, opcode]
2000 unsigned OpNum = 0;
2001 Value *LHS, *RHS;
2002 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
2003 getValue(Record, OpNum, LHS->getType(), RHS) ||
Dan Gohman1224c382009-07-20 21:19:07 +00002004 OpNum+1 > Record.size())
Chris Lattnerabfbf852007-05-06 00:21:25 +00002005 return Error("Invalid BINOP record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002006
Dan Gohman1224c382009-07-20 21:19:07 +00002007 int Opc = GetDecodedBinaryOpcode(Record[OpNum++], LHS->getType());
Chris Lattnerabfbf852007-05-06 00:21:25 +00002008 if (Opc == -1) return Error("Invalid BINOP record");
Gabor Greif7cbd8a32008-05-16 19:29:10 +00002009 I = BinaryOperator::Create((Instruction::BinaryOps)Opc, LHS, RHS);
Devang Patele8e02132009-09-18 19:26:43 +00002010 InstructionList.push_back(I);
Dan Gohmanf8dbee72009-09-07 23:54:19 +00002011 if (OpNum < Record.size()) {
2012 if (Opc == Instruction::Add ||
2013 Opc == Instruction::Sub ||
Chris Lattnerf067d582011-02-07 16:40:21 +00002014 Opc == Instruction::Mul ||
2015 Opc == Instruction::Shl) {
Dan Gohman26793ed2010-01-25 21:55:39 +00002016 if (Record[OpNum] & (1 << bitc::OBO_NO_SIGNED_WRAP))
Dan Gohmanf8dbee72009-09-07 23:54:19 +00002017 cast<BinaryOperator>(I)->setHasNoSignedWrap(true);
Dan Gohman26793ed2010-01-25 21:55:39 +00002018 if (Record[OpNum] & (1 << bitc::OBO_NO_UNSIGNED_WRAP))
Dan Gohmanf8dbee72009-09-07 23:54:19 +00002019 cast<BinaryOperator>(I)->setHasNoUnsignedWrap(true);
Chris Lattner35bda892011-02-06 21:44:57 +00002020 } else if (Opc == Instruction::SDiv ||
Chris Lattnerf067d582011-02-07 16:40:21 +00002021 Opc == Instruction::UDiv ||
2022 Opc == Instruction::LShr ||
2023 Opc == Instruction::AShr) {
Chris Lattner35bda892011-02-06 21:44:57 +00002024 if (Record[OpNum] & (1 << bitc::PEO_EXACT))
Dan Gohmanf8dbee72009-09-07 23:54:19 +00002025 cast<BinaryOperator>(I)->setIsExact(true);
2026 }
2027 }
Chris Lattner980e5aa2007-05-01 05:52:21 +00002028 break;
2029 }
Chris Lattnerabfbf852007-05-06 00:21:25 +00002030 case bitc::FUNC_CODE_INST_CAST: { // CAST: [opval, opty, destty, castopc]
2031 unsigned OpNum = 0;
2032 Value *Op;
2033 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2034 OpNum+2 != Record.size())
2035 return Error("Invalid CAST record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002036
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002037 Type *ResTy = getTypeByID(Record[OpNum]);
Chris Lattnerabfbf852007-05-06 00:21:25 +00002038 int Opc = GetDecodedCastOpcode(Record[OpNum+1]);
2039 if (Opc == -1 || ResTy == 0)
Chris Lattner231cbcb2007-05-02 04:27:25 +00002040 return Error("Invalid CAST record");
Gabor Greif7cbd8a32008-05-16 19:29:10 +00002041 I = CastInst::Create((Instruction::CastOps)Opc, Op, ResTy);
Devang Patele8e02132009-09-18 19:26:43 +00002042 InstructionList.push_back(I);
Chris Lattner231cbcb2007-05-02 04:27:25 +00002043 break;
2044 }
Dan Gohmandd8004d2009-07-27 21:53:46 +00002045 case bitc::FUNC_CODE_INST_INBOUNDS_GEP:
Chris Lattner15e6d172007-05-04 19:11:41 +00002046 case bitc::FUNC_CODE_INST_GEP: { // GEP: [n x operands]
Chris Lattner7337ab92007-05-06 00:00:00 +00002047 unsigned OpNum = 0;
2048 Value *BasePtr;
2049 if (getValueTypePair(Record, OpNum, NextValueNo, BasePtr))
Chris Lattner01ff65f2007-05-02 05:16:49 +00002050 return Error("Invalid GEP record");
2051
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002052 SmallVector<Value*, 16> GEPIdx;
Chris Lattner7337ab92007-05-06 00:00:00 +00002053 while (OpNum != Record.size()) {
2054 Value *Op;
2055 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
Chris Lattner01ff65f2007-05-02 05:16:49 +00002056 return Error("Invalid GEP record");
Chris Lattner7337ab92007-05-06 00:00:00 +00002057 GEPIdx.push_back(Op);
Chris Lattner01ff65f2007-05-02 05:16:49 +00002058 }
2059
Jay Foada9203102011-07-25 09:48:08 +00002060 I = GetElementPtrInst::Create(BasePtr, GEPIdx);
Devang Patele8e02132009-09-18 19:26:43 +00002061 InstructionList.push_back(I);
Dan Gohmandd8004d2009-07-27 21:53:46 +00002062 if (BitCode == bitc::FUNC_CODE_INST_INBOUNDS_GEP)
Dan Gohmanf8dbee72009-09-07 23:54:19 +00002063 cast<GetElementPtrInst>(I)->setIsInBounds(true);
Chris Lattner01ff65f2007-05-02 05:16:49 +00002064 break;
2065 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002066
Dan Gohman81a0c0b2008-05-31 00:58:22 +00002067 case bitc::FUNC_CODE_INST_EXTRACTVAL: {
2068 // EXTRACTVAL: [opty, opval, n x indices]
Dan Gohmane4977cf2008-05-23 01:55:30 +00002069 unsigned OpNum = 0;
2070 Value *Agg;
2071 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
2072 return Error("Invalid EXTRACTVAL record");
2073
Dan Gohman81a0c0b2008-05-31 00:58:22 +00002074 SmallVector<unsigned, 4> EXTRACTVALIdx;
2075 for (unsigned RecSize = Record.size();
2076 OpNum != RecSize; ++OpNum) {
2077 uint64_t Index = Record[OpNum];
2078 if ((unsigned)Index != Index)
2079 return Error("Invalid EXTRACTVAL index");
2080 EXTRACTVALIdx.push_back((unsigned)Index);
Dan Gohmane4977cf2008-05-23 01:55:30 +00002081 }
2082
Jay Foadfc6d3a42011-07-13 10:26:04 +00002083 I = ExtractValueInst::Create(Agg, EXTRACTVALIdx);
Devang Patele8e02132009-09-18 19:26:43 +00002084 InstructionList.push_back(I);
Dan Gohmane4977cf2008-05-23 01:55:30 +00002085 break;
2086 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002087
Dan Gohman81a0c0b2008-05-31 00:58:22 +00002088 case bitc::FUNC_CODE_INST_INSERTVAL: {
2089 // INSERTVAL: [opty, opval, opty, opval, n x indices]
Dan Gohmane4977cf2008-05-23 01:55:30 +00002090 unsigned OpNum = 0;
2091 Value *Agg;
2092 if (getValueTypePair(Record, OpNum, NextValueNo, Agg))
2093 return Error("Invalid INSERTVAL record");
2094 Value *Val;
2095 if (getValueTypePair(Record, OpNum, NextValueNo, Val))
2096 return Error("Invalid INSERTVAL record");
2097
Dan Gohman81a0c0b2008-05-31 00:58:22 +00002098 SmallVector<unsigned, 4> INSERTVALIdx;
2099 for (unsigned RecSize = Record.size();
2100 OpNum != RecSize; ++OpNum) {
2101 uint64_t Index = Record[OpNum];
2102 if ((unsigned)Index != Index)
2103 return Error("Invalid INSERTVAL index");
2104 INSERTVALIdx.push_back((unsigned)Index);
Dan Gohmane4977cf2008-05-23 01:55:30 +00002105 }
2106
Jay Foadfc6d3a42011-07-13 10:26:04 +00002107 I = InsertValueInst::Create(Agg, Val, INSERTVALIdx);
Devang Patele8e02132009-09-18 19:26:43 +00002108 InstructionList.push_back(I);
Dan Gohmane4977cf2008-05-23 01:55:30 +00002109 break;
2110 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002111
Chris Lattnerabfbf852007-05-06 00:21:25 +00002112 case bitc::FUNC_CODE_INST_SELECT: { // SELECT: [opval, ty, opval, opval]
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00002113 // obsolete form of select
2114 // handles select i1 ... in old bitcode
Chris Lattnerabfbf852007-05-06 00:21:25 +00002115 unsigned OpNum = 0;
2116 Value *TrueVal, *FalseVal, *Cond;
2117 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
2118 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
Owen Anderson1d0be152009-08-13 21:58:54 +00002119 getValue(Record, OpNum, Type::getInt1Ty(Context), Cond))
Chris Lattner01ff65f2007-05-02 05:16:49 +00002120 return Error("Invalid SELECT record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002121
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00002122 I = SelectInst::Create(Cond, TrueVal, FalseVal);
Devang Patele8e02132009-09-18 19:26:43 +00002123 InstructionList.push_back(I);
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00002124 break;
2125 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002126
Dan Gohmanfb2bbbe2008-09-16 01:01:33 +00002127 case bitc::FUNC_CODE_INST_VSELECT: {// VSELECT: [ty,opval,opval,predty,pred]
2128 // new form of select
2129 // handles select i1 or select [N x i1]
2130 unsigned OpNum = 0;
2131 Value *TrueVal, *FalseVal, *Cond;
2132 if (getValueTypePair(Record, OpNum, NextValueNo, TrueVal) ||
2133 getValue(Record, OpNum, TrueVal->getType(), FalseVal) ||
2134 getValueTypePair(Record, OpNum, NextValueNo, Cond))
2135 return Error("Invalid SELECT record");
Dan Gohmanf72fb672008-09-09 01:02:47 +00002136
2137 // select condition can be either i1 or [N x i1]
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002138 if (VectorType* vector_type =
2139 dyn_cast<VectorType>(Cond->getType())) {
Dan Gohmanf72fb672008-09-09 01:02:47 +00002140 // expect <n x i1>
Daniel Dunbara279bc32009-09-20 02:20:51 +00002141 if (vector_type->getElementType() != Type::getInt1Ty(Context))
Dan Gohmanf72fb672008-09-09 01:02:47 +00002142 return Error("Invalid SELECT condition type");
2143 } else {
2144 // expect i1
Daniel Dunbara279bc32009-09-20 02:20:51 +00002145 if (Cond->getType() != Type::getInt1Ty(Context))
Dan Gohmanf72fb672008-09-09 01:02:47 +00002146 return Error("Invalid SELECT condition type");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002147 }
2148
Gabor Greif051a9502008-04-06 20:25:17 +00002149 I = SelectInst::Create(Cond, TrueVal, FalseVal);
Devang Patele8e02132009-09-18 19:26:43 +00002150 InstructionList.push_back(I);
Chris Lattner01ff65f2007-05-02 05:16:49 +00002151 break;
2152 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002153
Chris Lattner01ff65f2007-05-02 05:16:49 +00002154 case bitc::FUNC_CODE_INST_EXTRACTELT: { // EXTRACTELT: [opty, opval, opval]
Chris Lattnerabfbf852007-05-06 00:21:25 +00002155 unsigned OpNum = 0;
2156 Value *Vec, *Idx;
2157 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
Owen Anderson1d0be152009-08-13 21:58:54 +00002158 getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
Chris Lattner01ff65f2007-05-02 05:16:49 +00002159 return Error("Invalid EXTRACTELT record");
Eric Christophera3500da2009-07-25 02:28:41 +00002160 I = ExtractElementInst::Create(Vec, Idx);
Devang Patele8e02132009-09-18 19:26:43 +00002161 InstructionList.push_back(I);
Chris Lattner01ff65f2007-05-02 05:16:49 +00002162 break;
2163 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002164
Chris Lattner01ff65f2007-05-02 05:16:49 +00002165 case bitc::FUNC_CODE_INST_INSERTELT: { // INSERTELT: [ty, opval,opval,opval]
Chris Lattnerabfbf852007-05-06 00:21:25 +00002166 unsigned OpNum = 0;
2167 Value *Vec, *Elt, *Idx;
2168 if (getValueTypePair(Record, OpNum, NextValueNo, Vec) ||
Daniel Dunbara279bc32009-09-20 02:20:51 +00002169 getValue(Record, OpNum,
Chris Lattnerabfbf852007-05-06 00:21:25 +00002170 cast<VectorType>(Vec->getType())->getElementType(), Elt) ||
Owen Anderson1d0be152009-08-13 21:58:54 +00002171 getValue(Record, OpNum, Type::getInt32Ty(Context), Idx))
Chris Lattner01ff65f2007-05-02 05:16:49 +00002172 return Error("Invalid INSERTELT record");
Gabor Greif051a9502008-04-06 20:25:17 +00002173 I = InsertElementInst::Create(Vec, Elt, Idx);
Devang Patele8e02132009-09-18 19:26:43 +00002174 InstructionList.push_back(I);
Chris Lattner01ff65f2007-05-02 05:16:49 +00002175 break;
2176 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002177
Chris Lattnerabfbf852007-05-06 00:21:25 +00002178 case bitc::FUNC_CODE_INST_SHUFFLEVEC: {// SHUFFLEVEC: [opval,ty,opval,opval]
2179 unsigned OpNum = 0;
2180 Value *Vec1, *Vec2, *Mask;
2181 if (getValueTypePair(Record, OpNum, NextValueNo, Vec1) ||
2182 getValue(Record, OpNum, Vec1->getType(), Vec2))
2183 return Error("Invalid SHUFFLEVEC record");
2184
Mon P Wangaeb06d22008-11-10 04:46:22 +00002185 if (getValueTypePair(Record, OpNum, NextValueNo, Mask))
Chris Lattner01ff65f2007-05-02 05:16:49 +00002186 return Error("Invalid SHUFFLEVEC record");
2187 I = new ShuffleVectorInst(Vec1, Vec2, Mask);
Devang Patele8e02132009-09-18 19:26:43 +00002188 InstructionList.push_back(I);
Chris Lattner01ff65f2007-05-02 05:16:49 +00002189 break;
2190 }
Mon P Wangaeb06d22008-11-10 04:46:22 +00002191
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +00002192 case bitc::FUNC_CODE_INST_CMP: // CMP: [opty, opval, opval, pred]
2193 // Old form of ICmp/FCmp returning bool
2194 // Existed to differentiate between icmp/fcmp and vicmp/vfcmp which were
2195 // both legal on vectors but had different behaviour.
2196 case bitc::FUNC_CODE_INST_CMP2: { // CMP2: [opty, opval, opval, pred]
2197 // FCmp/ICmp returning bool or vector of bool
2198
Chris Lattner7337ab92007-05-06 00:00:00 +00002199 unsigned OpNum = 0;
2200 Value *LHS, *RHS;
2201 if (getValueTypePair(Record, OpNum, NextValueNo, LHS) ||
2202 getValue(Record, OpNum, LHS->getType(), RHS) ||
2203 OpNum+1 != Record.size())
Chris Lattner01ff65f2007-05-02 05:16:49 +00002204 return Error("Invalid CMP record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002205
Duncan Sandsb0bc6c32010-02-15 16:12:20 +00002206 if (LHS->getType()->isFPOrFPVectorTy())
Dan Gohman1c8a23c2009-08-25 23:17:54 +00002207 I = new FCmpInst((FCmpInst::Predicate)Record[OpNum], LHS, RHS);
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +00002208 else
Dan Gohman1c8a23c2009-08-25 23:17:54 +00002209 I = new ICmpInst((ICmpInst::Predicate)Record[OpNum], LHS, RHS);
Devang Patele8e02132009-09-18 19:26:43 +00002210 InstructionList.push_back(I);
Dan Gohmanf72fb672008-09-09 01:02:47 +00002211 break;
2212 }
Nick Lewycky7f6aa2b2009-07-08 03:04:38 +00002213
Chris Lattner231cbcb2007-05-02 04:27:25 +00002214 case bitc::FUNC_CODE_INST_RET: // RET: [opty,opval<optional>]
Devang Pateld9d99ff2008-02-26 01:29:32 +00002215 {
2216 unsigned Size = Record.size();
2217 if (Size == 0) {
Owen Anderson1d0be152009-08-13 21:58:54 +00002218 I = ReturnInst::Create(Context);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002219 InstructionList.push_back(I);
Devang Pateld9d99ff2008-02-26 01:29:32 +00002220 break;
Dan Gohmanfc74abf2008-07-23 00:34:11 +00002221 }
Devang Pateld9d99ff2008-02-26 01:29:32 +00002222
Dan Gohmanfc74abf2008-07-23 00:34:11 +00002223 unsigned OpNum = 0;
Chris Lattner96a74c52011-06-17 18:09:11 +00002224 Value *Op = NULL;
2225 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2226 return Error("Invalid RET record");
2227 if (OpNum != Record.size())
2228 return Error("Invalid RET record");
Dan Gohmanfc74abf2008-07-23 00:34:11 +00002229
Chris Lattner96a74c52011-06-17 18:09:11 +00002230 I = ReturnInst::Create(Context, Op);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002231 InstructionList.push_back(I);
Dan Gohmanfc74abf2008-07-23 00:34:11 +00002232 break;
Chris Lattner231cbcb2007-05-02 04:27:25 +00002233 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002234 case bitc::FUNC_CODE_INST_BR: { // BR: [bb#, bb#, opval] or [bb#]
Chris Lattnerf61e6452007-05-03 22:09:51 +00002235 if (Record.size() != 1 && Record.size() != 3)
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002236 return Error("Invalid BR record");
2237 BasicBlock *TrueDest = getBasicBlock(Record[0]);
2238 if (TrueDest == 0)
2239 return Error("Invalid BR record");
2240
Devang Patele8e02132009-09-18 19:26:43 +00002241 if (Record.size() == 1) {
Gabor Greif051a9502008-04-06 20:25:17 +00002242 I = BranchInst::Create(TrueDest);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002243 InstructionList.push_back(I);
Devang Patele8e02132009-09-18 19:26:43 +00002244 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002245 else {
2246 BasicBlock *FalseDest = getBasicBlock(Record[1]);
Owen Anderson1d0be152009-08-13 21:58:54 +00002247 Value *Cond = getFnValueByID(Record[2], Type::getInt1Ty(Context));
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002248 if (FalseDest == 0 || Cond == 0)
2249 return Error("Invalid BR record");
Gabor Greif051a9502008-04-06 20:25:17 +00002250 I = BranchInst::Create(TrueDest, FalseDest, Cond);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002251 InstructionList.push_back(I);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002252 }
2253 break;
2254 }
Chris Lattnerf9be95f2009-10-27 19:13:16 +00002255 case bitc::FUNC_CODE_INST_SWITCH: { // SWITCH: [opty, op0, op1, ...]
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +00002256 // Check magic
2257 if ((Record[0] >> 16) == SWITCH_INST_MAGIC) {
2258 // New SwitchInst format with case ranges.
2259
2260 Type *OpTy = getTypeByID(Record[1]);
2261 unsigned ValueBitWidth = cast<IntegerType>(OpTy)->getBitWidth();
2262
2263 Value *Cond = getFnValueByID(Record[2], OpTy);
2264 BasicBlock *Default = getBasicBlock(Record[3]);
2265 if (OpTy == 0 || Cond == 0 || Default == 0)
2266 return Error("Invalid SWITCH record");
2267
2268 unsigned NumCases = Record[4];
2269
2270 SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
2271 InstructionList.push_back(SI);
2272
2273 unsigned CurIdx = 5;
2274 for (unsigned i = 0; i != NumCases; ++i) {
Stepan Dyatkovskiy0aa32d52012-05-29 12:26:47 +00002275 IntegersSubsetToBB CaseBuilder;
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +00002276 unsigned NumItems = Record[CurIdx++];
2277 for (unsigned ci = 0; ci != NumItems; ++ci) {
2278 bool isSingleNumber = Record[CurIdx++];
2279
2280 APInt Low;
2281 unsigned ActiveWords = 1;
2282 if (ValueBitWidth > 64)
2283 ActiveWords = Record[CurIdx++];
Benjamin Kramerf52aea82012-05-28 14:10:31 +00002284 Low = ReadWideAPInt(makeArrayRef(&Record[CurIdx], ActiveWords),
2285 ValueBitWidth);
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +00002286 CurIdx += ActiveWords;
Stepan Dyatkovskiy484fc932012-05-28 12:39:09 +00002287
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +00002288 if (!isSingleNumber) {
2289 ActiveWords = 1;
2290 if (ValueBitWidth > 64)
2291 ActiveWords = Record[CurIdx++];
2292 APInt High =
Benjamin Kramerf52aea82012-05-28 14:10:31 +00002293 ReadWideAPInt(makeArrayRef(&Record[CurIdx], ActiveWords),
2294 ValueBitWidth);
Stepan Dyatkovskiy484fc932012-05-28 12:39:09 +00002295
2296 CaseBuilder.add(IntItem::fromType(OpTy, Low),
2297 IntItem::fromType(OpTy, High));
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +00002298 CurIdx += ActiveWords;
2299 } else
Stepan Dyatkovskiy484fc932012-05-28 12:39:09 +00002300 CaseBuilder.add(IntItem::fromType(OpTy, Low));
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +00002301 }
2302 BasicBlock *DestBB = getBasicBlock(Record[CurIdx++]);
Stepan Dyatkovskiy0aa32d52012-05-29 12:26:47 +00002303 IntegersSubset Case = CaseBuilder.getCase();
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +00002304 SI->addCase(Case, DestBB);
2305 }
Stepan Dyatkovskiy734dde82012-05-14 08:26:31 +00002306 uint16_t Hash = SI->hash();
Stepan Dyatkovskiy1cce5bf2012-05-12 10:48:17 +00002307 if (Hash != (Record[0] & 0xFFFF))
2308 return Error("Invalid SWITCH record");
2309 I = SI;
2310 break;
2311 }
2312
2313 // Old SwitchInst format without case ranges.
2314
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002315 if (Record.size() < 3 || (Record.size() & 1) == 0)
2316 return Error("Invalid SWITCH record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002317 Type *OpTy = getTypeByID(Record[0]);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002318 Value *Cond = getFnValueByID(Record[1], OpTy);
2319 BasicBlock *Default = getBasicBlock(Record[2]);
2320 if (OpTy == 0 || Cond == 0 || Default == 0)
2321 return Error("Invalid SWITCH record");
2322 unsigned NumCases = (Record.size()-3)/2;
Gabor Greif051a9502008-04-06 20:25:17 +00002323 SwitchInst *SI = SwitchInst::Create(Cond, Default, NumCases);
Devang Patele8e02132009-09-18 19:26:43 +00002324 InstructionList.push_back(SI);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002325 for (unsigned i = 0, e = NumCases; i != e; ++i) {
Daniel Dunbara279bc32009-09-20 02:20:51 +00002326 ConstantInt *CaseVal =
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002327 dyn_cast_or_null<ConstantInt>(getFnValueByID(Record[3+i*2], OpTy));
2328 BasicBlock *DestBB = getBasicBlock(Record[1+3+i*2]);
2329 if (CaseVal == 0 || DestBB == 0) {
2330 delete SI;
2331 return Error("Invalid SWITCH record!");
2332 }
2333 SI->addCase(CaseVal, DestBB);
2334 }
2335 I = SI;
2336 break;
2337 }
Chris Lattnerab21db72009-10-28 00:19:10 +00002338 case bitc::FUNC_CODE_INST_INDIRECTBR: { // INDIRECTBR: [opty, op0, op1, ...]
Chris Lattnerf9be95f2009-10-27 19:13:16 +00002339 if (Record.size() < 2)
Chris Lattnerab21db72009-10-28 00:19:10 +00002340 return Error("Invalid INDIRECTBR record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002341 Type *OpTy = getTypeByID(Record[0]);
Chris Lattnerf9be95f2009-10-27 19:13:16 +00002342 Value *Address = getFnValueByID(Record[1], OpTy);
2343 if (OpTy == 0 || Address == 0)
Chris Lattnerab21db72009-10-28 00:19:10 +00002344 return Error("Invalid INDIRECTBR record");
Chris Lattnerf9be95f2009-10-27 19:13:16 +00002345 unsigned NumDests = Record.size()-2;
Chris Lattnerab21db72009-10-28 00:19:10 +00002346 IndirectBrInst *IBI = IndirectBrInst::Create(Address, NumDests);
Chris Lattnerf9be95f2009-10-27 19:13:16 +00002347 InstructionList.push_back(IBI);
2348 for (unsigned i = 0, e = NumDests; i != e; ++i) {
2349 if (BasicBlock *DestBB = getBasicBlock(Record[2+i])) {
2350 IBI->addDestination(DestBB);
2351 } else {
2352 delete IBI;
Chris Lattnerab21db72009-10-28 00:19:10 +00002353 return Error("Invalid INDIRECTBR record!");
Chris Lattnerf9be95f2009-10-27 19:13:16 +00002354 }
2355 }
2356 I = IBI;
2357 break;
2358 }
2359
Duncan Sandsdc024672007-11-27 13:23:08 +00002360 case bitc::FUNC_CODE_INST_INVOKE: {
2361 // INVOKE: [attrs, cc, normBB, unwindBB, fnty, op0,op1,op2, ...]
Chris Lattnera9bb7132007-05-08 05:38:01 +00002362 if (Record.size() < 4) return Error("Invalid INVOKE record");
Devang Patel05988662008-09-25 21:00:45 +00002363 AttrListPtr PAL = getAttributes(Record[0]);
Chris Lattnera9bb7132007-05-08 05:38:01 +00002364 unsigned CCInfo = Record[1];
2365 BasicBlock *NormalBB = getBasicBlock(Record[2]);
2366 BasicBlock *UnwindBB = getBasicBlock(Record[3]);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002367
Chris Lattnera9bb7132007-05-08 05:38:01 +00002368 unsigned OpNum = 4;
Chris Lattner7337ab92007-05-06 00:00:00 +00002369 Value *Callee;
2370 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002371 return Error("Invalid INVOKE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002372
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002373 PointerType *CalleeTy = dyn_cast<PointerType>(Callee->getType());
2374 FunctionType *FTy = !CalleeTy ? 0 :
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002375 dyn_cast<FunctionType>(CalleeTy->getElementType());
2376
2377 // Check that the right number of fixed parameters are here.
Chris Lattner7337ab92007-05-06 00:00:00 +00002378 if (FTy == 0 || NormalBB == 0 || UnwindBB == 0 ||
2379 Record.size() < OpNum+FTy->getNumParams())
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002380 return Error("Invalid INVOKE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002381
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002382 SmallVector<Value*, 16> Ops;
Chris Lattner7337ab92007-05-06 00:00:00 +00002383 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
2384 Ops.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
2385 if (Ops.back() == 0) return Error("Invalid INVOKE record");
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002386 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002387
Chris Lattner7337ab92007-05-06 00:00:00 +00002388 if (!FTy->isVarArg()) {
2389 if (Record.size() != OpNum)
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002390 return Error("Invalid INVOKE record");
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002391 } else {
Chris Lattner7337ab92007-05-06 00:00:00 +00002392 // Read type/value pairs for varargs params.
2393 while (OpNum != Record.size()) {
2394 Value *Op;
2395 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2396 return Error("Invalid INVOKE record");
2397 Ops.push_back(Op);
2398 }
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002399 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002400
Jay Foada3efbb12011-07-15 08:37:34 +00002401 I = InvokeInst::Create(Callee, NormalBB, UnwindBB, Ops);
Devang Patele8e02132009-09-18 19:26:43 +00002402 InstructionList.push_back(I);
Sandeep Patel65c3c8f2009-09-02 08:44:58 +00002403 cast<InvokeInst>(I)->setCallingConv(
2404 static_cast<CallingConv::ID>(CCInfo));
Devang Patel05988662008-09-25 21:00:45 +00002405 cast<InvokeInst>(I)->setAttributes(PAL);
Chris Lattnerf4c8e522007-05-02 05:46:45 +00002406 break;
2407 }
Bill Wendlingdccc03b2011-07-31 06:30:59 +00002408 case bitc::FUNC_CODE_INST_RESUME: { // RESUME: [opval]
2409 unsigned Idx = 0;
2410 Value *Val = 0;
2411 if (getValueTypePair(Record, Idx, NextValueNo, Val))
2412 return Error("Invalid RESUME record");
2413 I = ResumeInst::Create(Val);
Bill Wendling35726bf2011-09-01 00:50:20 +00002414 InstructionList.push_back(I);
Bill Wendlingdccc03b2011-07-31 06:30:59 +00002415 break;
2416 }
Chris Lattner231cbcb2007-05-02 04:27:25 +00002417 case bitc::FUNC_CODE_INST_UNREACHABLE: // UNREACHABLE
Owen Anderson1d0be152009-08-13 21:58:54 +00002418 I = new UnreachableInst(Context);
Devang Patele8e02132009-09-18 19:26:43 +00002419 InstructionList.push_back(I);
Chris Lattner231cbcb2007-05-02 04:27:25 +00002420 break;
Chris Lattnerabfbf852007-05-06 00:21:25 +00002421 case bitc::FUNC_CODE_INST_PHI: { // PHI: [ty, val0,bb0, ...]
Chris Lattner15e6d172007-05-04 19:11:41 +00002422 if (Record.size() < 1 || ((Record.size()-1)&1))
Chris Lattner2a98cca2007-05-03 18:58:09 +00002423 return Error("Invalid PHI record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002424 Type *Ty = getTypeByID(Record[0]);
Chris Lattner2a98cca2007-05-03 18:58:09 +00002425 if (!Ty) return Error("Invalid PHI record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002426
Jay Foad3ecfc862011-03-30 11:28:46 +00002427 PHINode *PN = PHINode::Create(Ty, (Record.size()-1)/2);
Devang Patele8e02132009-09-18 19:26:43 +00002428 InstructionList.push_back(PN);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002429
Chris Lattner15e6d172007-05-04 19:11:41 +00002430 for (unsigned i = 0, e = Record.size()-1; i != e; i += 2) {
2431 Value *V = getFnValueByID(Record[1+i], Ty);
2432 BasicBlock *BB = getBasicBlock(Record[2+i]);
Chris Lattner2a98cca2007-05-03 18:58:09 +00002433 if (!V || !BB) return Error("Invalid PHI record");
2434 PN->addIncoming(V, BB);
2435 }
2436 I = PN;
2437 break;
2438 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002439
Bill Wendlinge6e88262011-08-12 20:24:12 +00002440 case bitc::FUNC_CODE_INST_LANDINGPAD: {
2441 // LANDINGPAD: [ty, val, val, num, (id0,val0 ...)?]
2442 unsigned Idx = 0;
2443 if (Record.size() < 4)
2444 return Error("Invalid LANDINGPAD record");
2445 Type *Ty = getTypeByID(Record[Idx++]);
2446 if (!Ty) return Error("Invalid LANDINGPAD record");
2447 Value *PersFn = 0;
2448 if (getValueTypePair(Record, Idx, NextValueNo, PersFn))
2449 return Error("Invalid LANDINGPAD record");
2450
2451 bool IsCleanup = !!Record[Idx++];
2452 unsigned NumClauses = Record[Idx++];
2453 LandingPadInst *LP = LandingPadInst::Create(Ty, PersFn, NumClauses);
2454 LP->setCleanup(IsCleanup);
2455 for (unsigned J = 0; J != NumClauses; ++J) {
2456 LandingPadInst::ClauseType CT =
2457 LandingPadInst::ClauseType(Record[Idx++]); (void)CT;
2458 Value *Val;
2459
2460 if (getValueTypePair(Record, Idx, NextValueNo, Val)) {
2461 delete LP;
2462 return Error("Invalid LANDINGPAD record");
2463 }
2464
2465 assert((CT != LandingPadInst::Catch ||
2466 !isa<ArrayType>(Val->getType())) &&
2467 "Catch clause has a invalid type!");
2468 assert((CT != LandingPadInst::Filter ||
2469 isa<ArrayType>(Val->getType())) &&
2470 "Filter clause has invalid type!");
2471 LP->addClause(Val);
2472 }
2473
2474 I = LP;
Bill Wendling35726bf2011-09-01 00:50:20 +00002475 InstructionList.push_back(I);
Bill Wendlinge6e88262011-08-12 20:24:12 +00002476 break;
2477 }
2478
Chris Lattner96a74c52011-06-17 18:09:11 +00002479 case bitc::FUNC_CODE_INST_ALLOCA: { // ALLOCA: [instty, opty, op, align]
2480 if (Record.size() != 4)
2481 return Error("Invalid ALLOCA record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002482 PointerType *Ty =
Chris Lattner2a98cca2007-05-03 18:58:09 +00002483 dyn_cast_or_null<PointerType>(getTypeByID(Record[0]));
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002484 Type *OpTy = getTypeByID(Record[1]);
Chris Lattner96a74c52011-06-17 18:09:11 +00002485 Value *Size = getFnValueByID(Record[2], OpTy);
2486 unsigned Align = Record[3];
Chris Lattner2a98cca2007-05-03 18:58:09 +00002487 if (!Ty || !Size) return Error("Invalid ALLOCA record");
Owen Anderson50dead02009-07-15 23:53:25 +00002488 I = new AllocaInst(Ty->getElementType(), Size, (1 << Align) >> 1);
Devang Patele8e02132009-09-18 19:26:43 +00002489 InstructionList.push_back(I);
Chris Lattner2a98cca2007-05-03 18:58:09 +00002490 break;
2491 }
Chris Lattner0579f7f2007-05-03 22:04:19 +00002492 case bitc::FUNC_CODE_INST_LOAD: { // LOAD: [opty, op, align, vol]
Chris Lattner7337ab92007-05-06 00:00:00 +00002493 unsigned OpNum = 0;
2494 Value *Op;
2495 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2496 OpNum+2 != Record.size())
Chris Lattnerabfbf852007-05-06 00:21:25 +00002497 return Error("Invalid LOAD record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002498
Chris Lattner7337ab92007-05-06 00:00:00 +00002499 I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Devang Patele8e02132009-09-18 19:26:43 +00002500 InstructionList.push_back(I);
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002501 break;
Chris Lattner0579f7f2007-05-03 22:04:19 +00002502 }
Eli Friedman21006d42011-08-09 23:02:53 +00002503 case bitc::FUNC_CODE_INST_LOADATOMIC: {
2504 // LOADATOMIC: [opty, op, align, vol, ordering, synchscope]
2505 unsigned OpNum = 0;
2506 Value *Op;
2507 if (getValueTypePair(Record, OpNum, NextValueNo, Op) ||
2508 OpNum+4 != Record.size())
2509 return Error("Invalid LOADATOMIC record");
2510
2511
2512 AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+2]);
2513 if (Ordering == NotAtomic || Ordering == Release ||
2514 Ordering == AcquireRelease)
2515 return Error("Invalid LOADATOMIC record");
2516 if (Ordering != NotAtomic && Record[OpNum] == 0)
2517 return Error("Invalid LOADATOMIC record");
2518 SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+3]);
2519
2520 I = new LoadInst(Op, "", Record[OpNum+1], (1 << Record[OpNum]) >> 1,
2521 Ordering, SynchScope);
2522 InstructionList.push_back(I);
2523 break;
2524 }
Chris Lattner4f6bab92011-06-17 18:17:37 +00002525 case bitc::FUNC_CODE_INST_STORE: { // STORE2:[ptrty, ptr, val, align, vol]
Christopher Lambfe63fb92007-12-11 08:59:05 +00002526 unsigned OpNum = 0;
2527 Value *Val, *Ptr;
2528 if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
Daniel Dunbara279bc32009-09-20 02:20:51 +00002529 getValue(Record, OpNum,
Christopher Lambfe63fb92007-12-11 08:59:05 +00002530 cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
2531 OpNum+2 != Record.size())
2532 return Error("Invalid STORE record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002533
Christopher Lambfe63fb92007-12-11 08:59:05 +00002534 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1);
Devang Patele8e02132009-09-18 19:26:43 +00002535 InstructionList.push_back(I);
Christopher Lambfe63fb92007-12-11 08:59:05 +00002536 break;
2537 }
Eli Friedman21006d42011-08-09 23:02:53 +00002538 case bitc::FUNC_CODE_INST_STOREATOMIC: {
2539 // STOREATOMIC: [ptrty, ptr, val, align, vol, ordering, synchscope]
2540 unsigned OpNum = 0;
2541 Value *Val, *Ptr;
2542 if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
2543 getValue(Record, OpNum,
2544 cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
2545 OpNum+4 != Record.size())
2546 return Error("Invalid STOREATOMIC record");
2547
2548 AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+2]);
Eli Friedmanc3d35982011-09-19 19:41:28 +00002549 if (Ordering == NotAtomic || Ordering == Acquire ||
Eli Friedman21006d42011-08-09 23:02:53 +00002550 Ordering == AcquireRelease)
2551 return Error("Invalid STOREATOMIC record");
2552 SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+3]);
2553 if (Ordering != NotAtomic && Record[OpNum] == 0)
2554 return Error("Invalid STOREATOMIC record");
2555
2556 I = new StoreInst(Val, Ptr, Record[OpNum+1], (1 << Record[OpNum]) >> 1,
2557 Ordering, SynchScope);
2558 InstructionList.push_back(I);
2559 break;
2560 }
Eli Friedmanff030482011-07-28 21:48:00 +00002561 case bitc::FUNC_CODE_INST_CMPXCHG: {
2562 // CMPXCHG:[ptrty, ptr, cmp, new, vol, ordering, synchscope]
2563 unsigned OpNum = 0;
2564 Value *Ptr, *Cmp, *New;
2565 if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
2566 getValue(Record, OpNum,
2567 cast<PointerType>(Ptr->getType())->getElementType(), Cmp) ||
2568 getValue(Record, OpNum,
2569 cast<PointerType>(Ptr->getType())->getElementType(), New) ||
2570 OpNum+3 != Record.size())
2571 return Error("Invalid CMPXCHG record");
2572 AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+1]);
Eli Friedman21006d42011-08-09 23:02:53 +00002573 if (Ordering == NotAtomic || Ordering == Unordered)
Eli Friedmanff030482011-07-28 21:48:00 +00002574 return Error("Invalid CMPXCHG record");
2575 SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+2]);
2576 I = new AtomicCmpXchgInst(Ptr, Cmp, New, Ordering, SynchScope);
2577 cast<AtomicCmpXchgInst>(I)->setVolatile(Record[OpNum]);
2578 InstructionList.push_back(I);
2579 break;
2580 }
2581 case bitc::FUNC_CODE_INST_ATOMICRMW: {
2582 // ATOMICRMW:[ptrty, ptr, val, op, vol, ordering, synchscope]
2583 unsigned OpNum = 0;
2584 Value *Ptr, *Val;
2585 if (getValueTypePair(Record, OpNum, NextValueNo, Ptr) ||
2586 getValue(Record, OpNum,
2587 cast<PointerType>(Ptr->getType())->getElementType(), Val) ||
2588 OpNum+4 != Record.size())
2589 return Error("Invalid ATOMICRMW record");
2590 AtomicRMWInst::BinOp Operation = GetDecodedRMWOperation(Record[OpNum]);
2591 if (Operation < AtomicRMWInst::FIRST_BINOP ||
2592 Operation > AtomicRMWInst::LAST_BINOP)
2593 return Error("Invalid ATOMICRMW record");
2594 AtomicOrdering Ordering = GetDecodedOrdering(Record[OpNum+2]);
Eli Friedman21006d42011-08-09 23:02:53 +00002595 if (Ordering == NotAtomic || Ordering == Unordered)
Eli Friedmanff030482011-07-28 21:48:00 +00002596 return Error("Invalid ATOMICRMW record");
2597 SynchronizationScope SynchScope = GetDecodedSynchScope(Record[OpNum+3]);
2598 I = new AtomicRMWInst(Operation, Ptr, Val, Ordering, SynchScope);
2599 cast<AtomicRMWInst>(I)->setVolatile(Record[OpNum+1]);
2600 InstructionList.push_back(I);
2601 break;
2602 }
Eli Friedman47f35132011-07-25 23:16:38 +00002603 case bitc::FUNC_CODE_INST_FENCE: { // FENCE:[ordering, synchscope]
2604 if (2 != Record.size())
2605 return Error("Invalid FENCE record");
2606 AtomicOrdering Ordering = GetDecodedOrdering(Record[0]);
2607 if (Ordering == NotAtomic || Ordering == Unordered ||
2608 Ordering == Monotonic)
2609 return Error("Invalid FENCE record");
2610 SynchronizationScope SynchScope = GetDecodedSynchScope(Record[1]);
2611 I = new FenceInst(Context, Ordering, SynchScope);
2612 InstructionList.push_back(I);
2613 break;
2614 }
Chris Lattner4f6bab92011-06-17 18:17:37 +00002615 case bitc::FUNC_CODE_INST_CALL: {
Duncan Sandsdc024672007-11-27 13:23:08 +00002616 // CALL: [paramattrs, cc, fnty, fnid, arg0, arg1...]
2617 if (Record.size() < 3)
Chris Lattner0579f7f2007-05-03 22:04:19 +00002618 return Error("Invalid CALL record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002619
Devang Patel05988662008-09-25 21:00:45 +00002620 AttrListPtr PAL = getAttributes(Record[0]);
Chris Lattnera9bb7132007-05-08 05:38:01 +00002621 unsigned CCInfo = Record[1];
Daniel Dunbara279bc32009-09-20 02:20:51 +00002622
Chris Lattnera9bb7132007-05-08 05:38:01 +00002623 unsigned OpNum = 2;
Chris Lattner7337ab92007-05-06 00:00:00 +00002624 Value *Callee;
2625 if (getValueTypePair(Record, OpNum, NextValueNo, Callee))
2626 return Error("Invalid CALL record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002627
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002628 PointerType *OpTy = dyn_cast<PointerType>(Callee->getType());
2629 FunctionType *FTy = 0;
Chris Lattner0579f7f2007-05-03 22:04:19 +00002630 if (OpTy) FTy = dyn_cast<FunctionType>(OpTy->getElementType());
Chris Lattner7337ab92007-05-06 00:00:00 +00002631 if (!FTy || Record.size() < FTy->getNumParams()+OpNum)
Chris Lattner0579f7f2007-05-03 22:04:19 +00002632 return Error("Invalid CALL record");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002633
Chris Lattner0579f7f2007-05-03 22:04:19 +00002634 SmallVector<Value*, 16> Args;
2635 // Read the fixed params.
Chris Lattner7337ab92007-05-06 00:00:00 +00002636 for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i, ++OpNum) {
Chris Lattner1afcace2011-07-09 17:41:24 +00002637 if (FTy->getParamType(i)->isLabelTy())
Dale Johanneseneb57ea72007-11-05 21:20:28 +00002638 Args.push_back(getBasicBlock(Record[OpNum]));
Dan Gohman9b10dfb2010-09-13 18:00:48 +00002639 else
Dale Johanneseneb57ea72007-11-05 21:20:28 +00002640 Args.push_back(getFnValueByID(Record[OpNum], FTy->getParamType(i)));
Chris Lattner0579f7f2007-05-03 22:04:19 +00002641 if (Args.back() == 0) return Error("Invalid CALL record");
2642 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002643
Chris Lattner0579f7f2007-05-03 22:04:19 +00002644 // Read type/value pairs for varargs params.
Chris Lattner0579f7f2007-05-03 22:04:19 +00002645 if (!FTy->isVarArg()) {
Chris Lattner7337ab92007-05-06 00:00:00 +00002646 if (OpNum != Record.size())
Chris Lattner0579f7f2007-05-03 22:04:19 +00002647 return Error("Invalid CALL record");
2648 } else {
Chris Lattner7337ab92007-05-06 00:00:00 +00002649 while (OpNum != Record.size()) {
2650 Value *Op;
2651 if (getValueTypePair(Record, OpNum, NextValueNo, Op))
2652 return Error("Invalid CALL record");
2653 Args.push_back(Op);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002654 }
2655 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002656
Jay Foada3efbb12011-07-15 08:37:34 +00002657 I = CallInst::Create(Callee, Args);
Devang Patele8e02132009-09-18 19:26:43 +00002658 InstructionList.push_back(I);
Sandeep Patel65c3c8f2009-09-02 08:44:58 +00002659 cast<CallInst>(I)->setCallingConv(
2660 static_cast<CallingConv::ID>(CCInfo>>1));
Chris Lattner76520192007-05-03 22:34:03 +00002661 cast<CallInst>(I)->setTailCall(CCInfo & 1);
Devang Patel05988662008-09-25 21:00:45 +00002662 cast<CallInst>(I)->setAttributes(PAL);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002663 break;
2664 }
2665 case bitc::FUNC_CODE_INST_VAARG: { // VAARG: [valistty, valist, instty]
2666 if (Record.size() < 3)
2667 return Error("Invalid VAARG record");
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002668 Type *OpTy = getTypeByID(Record[0]);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002669 Value *Op = getFnValueByID(Record[1], OpTy);
Chris Lattnerdb125cf2011-07-18 04:54:35 +00002670 Type *ResTy = getTypeByID(Record[2]);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002671 if (!OpTy || !Op || !ResTy)
2672 return Error("Invalid VAARG record");
2673 I = new VAArgInst(Op, ResTy);
Devang Patele8e02132009-09-18 19:26:43 +00002674 InstructionList.push_back(I);
Chris Lattner0579f7f2007-05-03 22:04:19 +00002675 break;
2676 }
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002677 }
2678
2679 // Add instruction to end of current BB. If there is no current BB, reject
2680 // this file.
2681 if (CurBB == 0) {
2682 delete I;
2683 return Error("Invalid instruction with no BB");
2684 }
2685 CurBB->getInstList().push_back(I);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002686
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002687 // If this was a terminator instruction, move to the next block.
2688 if (isa<TerminatorInst>(I)) {
2689 ++CurBBNo;
2690 CurBB = CurBBNo < FunctionBBs.size() ? FunctionBBs[CurBBNo] : 0;
2691 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002692
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002693 // Non-void values get registered in the value table for future use.
Benjamin Kramerf0127052010-01-05 13:12:22 +00002694 if (I && !I->getType()->isVoidTy())
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002695 ValueList.AssignValue(I, NextValueNo++);
Chris Lattner980e5aa2007-05-01 05:52:21 +00002696 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002697
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002698 // Check the function list for unresolved values.
2699 if (Argument *A = dyn_cast<Argument>(ValueList.back())) {
2700 if (A->getParent() == 0) {
2701 // We found at least one unresolved value. Nuke them all to avoid leaks.
2702 for (unsigned i = ModuleValueListSize, e = ValueList.size(); i != e; ++i){
Dan Gohman56e2a572010-08-25 20:20:21 +00002703 if ((A = dyn_cast<Argument>(ValueList[i])) && A->getParent() == 0) {
Owen Anderson9e9a0d52009-07-30 23:03:37 +00002704 A->replaceAllUsesWith(UndefValue::get(A->getType()));
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002705 delete A;
2706 }
2707 }
Chris Lattner35a04702007-05-04 03:50:29 +00002708 return Error("Never resolved value found in function!");
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002709 }
Chris Lattnera7c49aa2007-05-01 07:01:57 +00002710 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002711
Dan Gohman064ff3e2010-08-25 20:23:38 +00002712 // FIXME: Check for unresolved forward-declared metadata references
2713 // and clean up leaks.
2714
Chris Lattner50b136d2009-10-28 05:53:48 +00002715 // See if anything took the address of blocks in this function. If so,
2716 // resolve them now.
Chris Lattner50b136d2009-10-28 05:53:48 +00002717 DenseMap<Function*, std::vector<BlockAddrRefTy> >::iterator BAFRI =
2718 BlockAddrFwdRefs.find(F);
2719 if (BAFRI != BlockAddrFwdRefs.end()) {
2720 std::vector<BlockAddrRefTy> &RefList = BAFRI->second;
2721 for (unsigned i = 0, e = RefList.size(); i != e; ++i) {
2722 unsigned BlockIdx = RefList[i].first;
Chris Lattnercdfc9402009-11-01 01:27:45 +00002723 if (BlockIdx >= FunctionBBs.size())
Chris Lattner50b136d2009-10-28 05:53:48 +00002724 return Error("Invalid blockaddress block #");
2725
2726 GlobalVariable *FwdRef = RefList[i].second;
Chris Lattnercdfc9402009-11-01 01:27:45 +00002727 FwdRef->replaceAllUsesWith(BlockAddress::get(F, FunctionBBs[BlockIdx]));
Chris Lattner50b136d2009-10-28 05:53:48 +00002728 FwdRef->eraseFromParent();
2729 }
2730
2731 BlockAddrFwdRefs.erase(BAFRI);
2732 }
2733
Chris Lattner980e5aa2007-05-01 05:52:21 +00002734 // Trim the value list down to the size it was before we parsed this function.
2735 ValueList.shrinkTo(ModuleValueListSize);
Dan Gohman69813832010-08-25 20:22:53 +00002736 MDValueList.shrinkTo(ModuleMDValueListSize);
Chris Lattner980e5aa2007-05-01 05:52:21 +00002737 std::vector<BasicBlock*>().swap(FunctionBBs);
Chris Lattner48f84872007-05-01 04:59:48 +00002738 return false;
2739}
2740
Derek Schuff2ea93872012-02-06 22:30:29 +00002741/// FindFunctionInStream - Find the function body in the bitcode stream
2742bool BitcodeReader::FindFunctionInStream(Function *F,
2743 DenseMap<Function*, uint64_t>::iterator DeferredFunctionInfoIterator) {
2744 while (DeferredFunctionInfoIterator->second == 0) {
2745 if (Stream.AtEndOfStream())
2746 return Error("Could not find Function in stream");
2747 // ParseModule will parse the next body in the stream and set its
2748 // position in the DeferredFunctionInfo map.
2749 if (ParseModule(true)) return true;
2750 }
2751 return false;
2752}
2753
Chris Lattnerb348bb82007-05-18 04:02:46 +00002754//===----------------------------------------------------------------------===//
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002755// GVMaterializer implementation
Chris Lattnerb348bb82007-05-18 04:02:46 +00002756//===----------------------------------------------------------------------===//
2757
2758
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002759bool BitcodeReader::isMaterializable(const GlobalValue *GV) const {
2760 if (const Function *F = dyn_cast<Function>(GV)) {
2761 return F->isDeclaration() &&
2762 DeferredFunctionInfo.count(const_cast<Function*>(F));
2763 }
2764 return false;
2765}
Daniel Dunbara279bc32009-09-20 02:20:51 +00002766
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002767bool BitcodeReader::Materialize(GlobalValue *GV, std::string *ErrInfo) {
2768 Function *F = dyn_cast<Function>(GV);
2769 // If it's not a function or is already material, ignore the request.
2770 if (!F || !F->isMaterializable()) return false;
2771
2772 DenseMap<Function*, uint64_t>::iterator DFII = DeferredFunctionInfo.find(F);
Chris Lattnerb348bb82007-05-18 04:02:46 +00002773 assert(DFII != DeferredFunctionInfo.end() && "Deferred function not found!");
Derek Schuff2ea93872012-02-06 22:30:29 +00002774 // If its position is recorded as 0, its body is somewhere in the stream
2775 // but we haven't seen it yet.
2776 if (DFII->second == 0)
2777 if (LazyStreamer && FindFunctionInStream(F, DFII)) return true;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002778
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002779 // Move the bit stream to the saved position of the deferred function body.
2780 Stream.JumpToBit(DFII->second);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002781
Chris Lattnerb348bb82007-05-18 04:02:46 +00002782 if (ParseFunctionBody(F)) {
2783 if (ErrInfo) *ErrInfo = ErrorString;
2784 return true;
2785 }
Chandler Carruth69940402007-08-04 01:51:18 +00002786
2787 // Upgrade any old intrinsic calls in the function.
2788 for (UpgradedIntrinsicMap::iterator I = UpgradedIntrinsics.begin(),
2789 E = UpgradedIntrinsics.end(); I != E; ++I) {
2790 if (I->first != I->second) {
2791 for (Value::use_iterator UI = I->first->use_begin(),
2792 UE = I->first->use_end(); UI != UE; ) {
2793 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2794 UpgradeIntrinsicCall(CI, I->second);
2795 }
2796 }
2797 }
Daniel Dunbara279bc32009-09-20 02:20:51 +00002798
Chris Lattnerb348bb82007-05-18 04:02:46 +00002799 return false;
2800}
2801
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002802bool BitcodeReader::isDematerializable(const GlobalValue *GV) const {
2803 const Function *F = dyn_cast<Function>(GV);
2804 if (!F || F->isDeclaration())
2805 return false;
2806 return DeferredFunctionInfo.count(const_cast<Function*>(F));
2807}
2808
2809void BitcodeReader::Dematerialize(GlobalValue *GV) {
2810 Function *F = dyn_cast<Function>(GV);
2811 // If this function isn't dematerializable, this is a noop.
2812 if (!F || !isDematerializable(F))
Chris Lattnerb348bb82007-05-18 04:02:46 +00002813 return;
Daniel Dunbara279bc32009-09-20 02:20:51 +00002814
Chris Lattnerb348bb82007-05-18 04:02:46 +00002815 assert(DeferredFunctionInfo.count(F) && "No info to read function later?");
Daniel Dunbara279bc32009-09-20 02:20:51 +00002816
Chris Lattnerb348bb82007-05-18 04:02:46 +00002817 // Just forget the function body, we can remat it later.
2818 F->deleteBody();
Chris Lattnerb348bb82007-05-18 04:02:46 +00002819}
2820
2821
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002822bool BitcodeReader::MaterializeModule(Module *M, std::string *ErrInfo) {
2823 assert(M == TheModule &&
2824 "Can only Materialize the Module this BitcodeReader is attached to.");
Chris Lattner714fa952009-06-16 05:15:21 +00002825 // Iterate over the module, deserializing any functions that are still on
2826 // disk.
2827 for (Module::iterator F = TheModule->begin(), E = TheModule->end();
2828 F != E; ++F)
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002829 if (F->isMaterializable() &&
2830 Materialize(F, ErrInfo))
2831 return true;
Chandler Carruth69940402007-08-04 01:51:18 +00002832
Derek Schuff0ffe6982012-02-29 00:07:09 +00002833 // At this point, if there are any function bodies, the current bit is
2834 // pointing to the END_BLOCK record after them. Now make sure the rest
2835 // of the bits in the module have been read.
2836 if (NextUnreadBit)
2837 ParseModule(true);
2838
Daniel Dunbara279bc32009-09-20 02:20:51 +00002839 // Upgrade any intrinsic calls that slipped through (should not happen!) and
2840 // delete the old functions to clean up. We can't do this unless the entire
2841 // module is materialized because there could always be another function body
Chandler Carruth69940402007-08-04 01:51:18 +00002842 // with calls to the old function.
2843 for (std::vector<std::pair<Function*, Function*> >::iterator I =
2844 UpgradedIntrinsics.begin(), E = UpgradedIntrinsics.end(); I != E; ++I) {
2845 if (I->first != I->second) {
2846 for (Value::use_iterator UI = I->first->use_begin(),
2847 UE = I->first->use_end(); UI != UE; ) {
2848 if (CallInst* CI = dyn_cast<CallInst>(*UI++))
2849 UpgradeIntrinsicCall(CI, I->second);
2850 }
Chris Lattner7d9eb582009-04-01 01:43:03 +00002851 if (!I->first->use_empty())
2852 I->first->replaceAllUsesWith(I->second);
Chandler Carruth69940402007-08-04 01:51:18 +00002853 I->first->eraseFromParent();
2854 }
2855 }
2856 std::vector<std::pair<Function*, Function*> >().swap(UpgradedIntrinsics);
Devang Patele4b27562009-08-28 23:24:31 +00002857
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002858 return false;
Chris Lattnerb348bb82007-05-18 04:02:46 +00002859}
2860
Derek Schuff2ea93872012-02-06 22:30:29 +00002861bool BitcodeReader::InitStream() {
2862 if (LazyStreamer) return InitLazyStream();
2863 return InitStreamFromBuffer();
2864}
2865
2866bool BitcodeReader::InitStreamFromBuffer() {
Roman Divacky5177b3a2012-09-06 15:42:13 +00002867 const unsigned char *BufPtr = (const unsigned char*)Buffer->getBufferStart();
Derek Schuff2ea93872012-02-06 22:30:29 +00002868 const unsigned char *BufEnd = BufPtr+Buffer->getBufferSize();
2869
2870 if (Buffer->getBufferSize() & 3) {
2871 if (!isRawBitcode(BufPtr, BufEnd) && !isBitcodeWrapper(BufPtr, BufEnd))
2872 return Error("Invalid bitcode signature");
2873 else
2874 return Error("Bitcode stream should be a multiple of 4 bytes in length");
2875 }
2876
2877 // If we have a wrapper header, parse it and ignore the non-bc file contents.
2878 // The magic number is 0x0B17C0DE stored in little endian.
2879 if (isBitcodeWrapper(BufPtr, BufEnd))
2880 if (SkipBitcodeWrapperHeader(BufPtr, BufEnd, true))
2881 return Error("Invalid bitcode wrapper header");
2882
2883 StreamFile.reset(new BitstreamReader(BufPtr, BufEnd));
2884 Stream.init(*StreamFile);
2885
2886 return false;
2887}
2888
2889bool BitcodeReader::InitLazyStream() {
2890 // Check and strip off the bitcode wrapper; BitstreamReader expects never to
2891 // see it.
2892 StreamingMemoryObject *Bytes = new StreamingMemoryObject(LazyStreamer);
2893 StreamFile.reset(new BitstreamReader(Bytes));
2894 Stream.init(*StreamFile);
2895
2896 unsigned char buf[16];
2897 if (Bytes->readBytes(0, 16, buf, NULL) == -1)
2898 return Error("Bitcode stream must be at least 16 bytes in length");
2899
2900 if (!isBitcode(buf, buf + 16))
2901 return Error("Invalid bitcode signature");
2902
2903 if (isBitcodeWrapper(buf, buf + 4)) {
2904 const unsigned char *bitcodeStart = buf;
2905 const unsigned char *bitcodeEnd = buf + 16;
2906 SkipBitcodeWrapperHeader(bitcodeStart, bitcodeEnd, false);
2907 Bytes->dropLeadingBytes(bitcodeStart - buf);
2908 Bytes->setKnownObjectSize(bitcodeEnd - bitcodeStart);
2909 }
2910 return false;
2911}
Chris Lattner48f84872007-05-01 04:59:48 +00002912
Chris Lattnerc453f762007-04-29 07:54:31 +00002913//===----------------------------------------------------------------------===//
2914// External interface
2915//===----------------------------------------------------------------------===//
2916
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002917/// getLazyBitcodeModule - lazy function-at-a-time loading from a file.
Chris Lattnerc453f762007-04-29 07:54:31 +00002918///
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002919Module *llvm::getLazyBitcodeModule(MemoryBuffer *Buffer,
2920 LLVMContext& Context,
2921 std::string *ErrMsg) {
2922 Module *M = new Module(Buffer->getBufferIdentifier(), Context);
Owen Anderson8b477ed2009-07-01 16:58:40 +00002923 BitcodeReader *R = new BitcodeReader(Buffer, Context);
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002924 M->setMaterializer(R);
2925 if (R->ParseBitcodeInto(M)) {
Chris Lattnerc453f762007-04-29 07:54:31 +00002926 if (ErrMsg)
2927 *ErrMsg = R->getErrorString();
Daniel Dunbara279bc32009-09-20 02:20:51 +00002928
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002929 delete M; // Also deletes R.
Chris Lattnerc453f762007-04-29 07:54:31 +00002930 return 0;
2931 }
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002932 // Have the BitcodeReader dtor delete 'Buffer'.
2933 R->setBufferOwned(true);
Rafael Espindola47f79bb2012-01-02 07:49:53 +00002934
2935 R->materializeForwardReferencedFunctions();
2936
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002937 return M;
Chris Lattnerc453f762007-04-29 07:54:31 +00002938}
2939
Derek Schuff2ea93872012-02-06 22:30:29 +00002940
2941Module *llvm::getStreamedBitcodeModule(const std::string &name,
2942 DataStreamer *streamer,
2943 LLVMContext &Context,
2944 std::string *ErrMsg) {
2945 Module *M = new Module(name, Context);
2946 BitcodeReader *R = new BitcodeReader(streamer, Context);
2947 M->setMaterializer(R);
2948 if (R->ParseBitcodeInto(M)) {
2949 if (ErrMsg)
2950 *ErrMsg = R->getErrorString();
2951 delete M; // Also deletes R.
2952 return 0;
2953 }
2954 R->setBufferOwned(false); // no buffer to delete
2955 return M;
2956}
2957
Chris Lattnerc453f762007-04-29 07:54:31 +00002958/// ParseBitcodeFile - Read the specified bitcode file, returning the module.
2959/// If an error occurs, return null and fill in *ErrMsg if non-null.
Daniel Dunbara279bc32009-09-20 02:20:51 +00002960Module *llvm::ParseBitcodeFile(MemoryBuffer *Buffer, LLVMContext& Context,
Owen Anderson8b477ed2009-07-01 16:58:40 +00002961 std::string *ErrMsg){
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002962 Module *M = getLazyBitcodeModule(Buffer, Context, ErrMsg);
2963 if (!M) return 0;
Chris Lattnerb348bb82007-05-18 04:02:46 +00002964
2965 // Don't let the BitcodeReader dtor delete 'Buffer', regardless of whether
2966 // there was an error.
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002967 static_cast<BitcodeReader*>(M->getMaterializer())->setBufferOwned(false);
Daniel Dunbara279bc32009-09-20 02:20:51 +00002968
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002969 // Read in the entire module, and destroy the BitcodeReader.
2970 if (M->MaterializeAllPermanently(ErrMsg)) {
2971 delete M;
Bill Wendling34711742010-10-06 01:22:42 +00002972 return 0;
Jeffrey Yasskinf0356fe2010-01-27 20:34:15 +00002973 }
Bill Wendling34711742010-10-06 01:22:42 +00002974
Chad Rosiercbbb0962011-12-07 21:44:12 +00002975 // TODO: Restore the use-lists to the in-memory state when the bitcode was
2976 // written. We must defer until the Module has been fully materialized.
2977
Chris Lattnerc453f762007-04-29 07:54:31 +00002978 return M;
2979}
Bill Wendling34711742010-10-06 01:22:42 +00002980
2981std::string llvm::getBitcodeTargetTriple(MemoryBuffer *Buffer,
2982 LLVMContext& Context,
2983 std::string *ErrMsg) {
2984 BitcodeReader *R = new BitcodeReader(Buffer, Context);
2985 // Don't let the BitcodeReader dtor delete 'Buffer'.
2986 R->setBufferOwned(false);
2987
2988 std::string Triple("");
2989 if (R->ParseTriple(Triple))
2990 if (ErrMsg)
2991 *ErrMsg = R->getErrorString();
2992
2993 delete R;
2994 return Triple;
2995}