BitcodeReader.h revision 4434ed44c45c87a72b7a0bf2f91211f895022b91
1caee0dccffb77a003681345ab3281bcf8684526cChris Lattner//===- BitcodeReader.h - Internal BitcodeReader impl ------------*- C++ -*-===// 2caee0dccffb77a003681345ab3281bcf8684526cChris Lattner// 3caee0dccffb77a003681345ab3281bcf8684526cChris Lattner// The LLVM Compiler Infrastructure 4caee0dccffb77a003681345ab3281bcf8684526cChris Lattner// 54ee451de366474b9c228b4e5fa573795a715216dChris Lattner// This file is distributed under the University of Illinois Open Source 64ee451de366474b9c228b4e5fa573795a715216dChris Lattner// License. See LICENSE.TXT for details. 7caee0dccffb77a003681345ab3281bcf8684526cChris Lattner// 8caee0dccffb77a003681345ab3281bcf8684526cChris Lattner//===----------------------------------------------------------------------===// 9caee0dccffb77a003681345ab3281bcf8684526cChris Lattner// 10caee0dccffb77a003681345ab3281bcf8684526cChris Lattner// This header defines the BitcodeReader class. 11caee0dccffb77a003681345ab3281bcf8684526cChris Lattner// 12caee0dccffb77a003681345ab3281bcf8684526cChris Lattner//===----------------------------------------------------------------------===// 13caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 14caee0dccffb77a003681345ab3281bcf8684526cChris Lattner#ifndef BITCODE_READER_H 15caee0dccffb77a003681345ab3281bcf8684526cChris Lattner#define BITCODE_READER_H 16caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 17caee0dccffb77a003681345ab3281bcf8684526cChris Lattner#include "llvm/ModuleProvider.h" 18eaf42abab6d465c38891345d999255871cf03943Devang Patel#include "llvm/Attributes.h" 19522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner#include "llvm/Type.h" 20efe65369a74871c3140a540a6c95ce5d1f080954Gabor Greif#include "llvm/OperandTraits.h" 2148f848716e80d01619b239111db48bfac77baad1Chris Lattner#include "llvm/Bitcode/BitstreamReader.h" 2247f96bf24687b5068aec7166cb8b3ac33ae964aeChris Lattner#include "llvm/Bitcode/LLVMBitCodes.h" 2346e7740a4433383e6e5561f089a091c22125bd07Chris Lattner#include "llvm/Support/ValueHandle.h" 2448f848716e80d01619b239111db48bfac77baad1Chris Lattner#include "llvm/ADT/DenseMap.h" 25caee0dccffb77a003681345ab3281bcf8684526cChris Lattner#include <vector> 26caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 27caee0dccffb77a003681345ab3281bcf8684526cChris Lattnernamespace llvm { 28c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner class MemoryBuffer; 298b477ed579794ba6d76915d56b3f448a7dd20120Owen Anderson class LLVMContext; 30522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner 31efe65369a74871c3140a540a6c95ce5d1f080954Gabor Greif//===----------------------------------------------------------------------===// 32efe65369a74871c3140a540a6c95ce5d1f080954Gabor Greif// BitcodeReaderValueList Class 33efe65369a74871c3140a540a6c95ce5d1f080954Gabor Greif//===----------------------------------------------------------------------===// 34efe65369a74871c3140a540a6c95ce5d1f080954Gabor Greif 3546e7740a4433383e6e5561f089a091c22125bd07Chris Lattnerclass BitcodeReaderValueList { 3646e7740a4433383e6e5561f089a091c22125bd07Chris Lattner std::vector<WeakVH> ValuePtrs; 37ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner 38ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner /// ResolveConstants - As we resolve forward-referenced constants, we add 39ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner /// information about them to this vector. This allows us to resolve them in 40ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner /// bulk instead of resolving each reference at a time. See the code in 41ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner /// ResolveConstantForwardRefs for more information about this. 42ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner /// 43ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner /// The key of this vector is the placeholder constant, the value is the slot 44ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner /// number that holds the resolved value. 45ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner typedef std::vector<std::pair<Constant*, unsigned> > ResolveConstantsTy; 46ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner ResolveConstantsTy ResolveConstants; 47522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattnerpublic: 4846e7740a4433383e6e5561f089a091c22125bd07Chris Lattner BitcodeReaderValueList() {} 49ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner ~BitcodeReaderValueList() { 50ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner assert(ResolveConstants.empty() && "Constants not resolved?"); 51ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner } 52efe65369a74871c3140a540a6c95ce5d1f080954Gabor Greif 53522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner // vector compatibility methods 5446e7740a4433383e6e5561f089a091c22125bd07Chris Lattner unsigned size() const { return ValuePtrs.size(); } 5546e7740a4433383e6e5561f089a091c22125bd07Chris Lattner void resize(unsigned N) { ValuePtrs.resize(N); } 56522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner void push_back(Value *V) { 5746e7740a4433383e6e5561f089a091c22125bd07Chris Lattner ValuePtrs.push_back(V); 58522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner } 59522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner 60b348bb81253a1105d23ab1a1771f8d2a6546aa1bChris Lattner void clear() { 61ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner assert(ResolveConstants.empty() && "Constants not resolved?"); 6246e7740a4433383e6e5561f089a091c22125bd07Chris Lattner ValuePtrs.clear(); 63b348bb81253a1105d23ab1a1771f8d2a6546aa1bChris Lattner } 64b348bb81253a1105d23ab1a1771f8d2a6546aa1bChris Lattner 6546e7740a4433383e6e5561f089a091c22125bd07Chris Lattner Value *operator[](unsigned i) const { 6646e7740a4433383e6e5561f089a091c22125bd07Chris Lattner assert(i < ValuePtrs.size()); 6746e7740a4433383e6e5561f089a091c22125bd07Chris Lattner return ValuePtrs[i]; 6846e7740a4433383e6e5561f089a091c22125bd07Chris Lattner } 69522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner 7046e7740a4433383e6e5561f089a091c22125bd07Chris Lattner Value *back() const { return ValuePtrs.back(); } 7146e7740a4433383e6e5561f089a091c22125bd07Chris Lattner void pop_back() { ValuePtrs.pop_back(); } 7246e7740a4433383e6e5561f089a091c22125bd07Chris Lattner bool empty() const { return ValuePtrs.empty(); } 73198f34ac359c48018c6e1f784cf3770ead63b253Chris Lattner void shrinkTo(unsigned N) { 7446e7740a4433383e6e5561f089a091c22125bd07Chris Lattner assert(N <= size() && "Invalid shrinkTo request!"); 7546e7740a4433383e6e5561f089a091c22125bd07Chris Lattner ValuePtrs.resize(N); 76198f34ac359c48018c6e1f784cf3770ead63b253Chris Lattner } 77522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner 78522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner Constant *getConstantFwdRef(unsigned Idx, const Type *Ty); 79a7c49aac984fafa5dfdfcc2762d4d51b26788e38Chris Lattner Value *getValueFwdRef(unsigned Idx, const Type *Ty); 80a7c49aac984fafa5dfdfcc2762d4d51b26788e38Chris Lattner 8146e7740a4433383e6e5561f089a091c22125bd07Chris Lattner void AssignValue(Value *V, unsigned Idx); 82a7c49aac984fafa5dfdfcc2762d4d51b26788e38Chris Lattner 83ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner /// ResolveConstantForwardRefs - Once all constants are read, this method bulk 84ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner /// resolves any forward references. 85ea693dfab48ee6fc07cc21abc20f487df5057a6bChris Lattner void ResolveConstantForwardRefs(); 86efe65369a74871c3140a540a6c95ce5d1f080954Gabor Greif}; 87efe65369a74871c3140a540a6c95ce5d1f080954Gabor Greif 88caee0dccffb77a003681345ab3281bcf8684526cChris Lattnerclass BitcodeReader : public ModuleProvider { 894434ed44c45c87a72b7a0bf2f91211f895022b91Owen Anderson LLVMContext& Context; 90c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner MemoryBuffer *Buffer; 91962dde3cef3184f1683d5070c298c9a29509d62eChris Lattner BitstreamReader StreamFile; 92962dde3cef3184f1683d5070c298c9a29509d62eChris Lattner BitstreamCursor Stream; 9348f848716e80d01619b239111db48bfac77baad1Chris Lattner 94caee0dccffb77a003681345ab3281bcf8684526cChris Lattner const char *ErrorString; 95caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 96caee0dccffb77a003681345ab3281bcf8684526cChris Lattner std::vector<PATypeHolder> TypeList; 97522b7b104c864da81c19d8b16c43b7a1f6a2fc40Chris Lattner BitcodeReaderValueList ValueList; 98e16504eb4ef8f09611cdf6e9a0be9eb886b4ed89Chris Lattner std::vector<std::pair<GlobalVariable*, unsigned> > GlobalInits; 9907d98b4afbdcbb4eed048400d9116de1ec83e866Chris Lattner std::vector<std::pair<GlobalAlias*, unsigned> > AliasInits; 10048f848716e80d01619b239111db48bfac77baad1Chris Lattner 10119c874638d9478a5d5028854817a5ee72293bb2bDevang Patel /// MAttributes - The set of attributes by index. Index zero in the 10248c85b84c1b66fb6a1b0d2afddf33da5bd82960dChris Lattner /// file is for null, and is thus not represented here. As such all indices 10348c85b84c1b66fb6a1b0d2afddf33da5bd82960dChris Lattner /// are off by one. 10419c874638d9478a5d5028854817a5ee72293bb2bDevang Patel std::vector<AttrListPtr> MAttributes; 10548c85b84c1b66fb6a1b0d2afddf33da5bd82960dChris Lattner 106980e5aad4cfaa32e13b297f4201eb1088ca96cc4Chris Lattner /// FunctionBBs - While parsing a function body, this is a list of the basic 107980e5aad4cfaa32e13b297f4201eb1088ca96cc4Chris Lattner /// blocks for the function. 108980e5aad4cfaa32e13b297f4201eb1088ca96cc4Chris Lattner std::vector<BasicBlock*> FunctionBBs; 109980e5aad4cfaa32e13b297f4201eb1088ca96cc4Chris Lattner 11048f848716e80d01619b239111db48bfac77baad1Chris Lattner // When reading the module header, this list is populated with functions that 11148f848716e80d01619b239111db48bfac77baad1Chris Lattner // have bodies later in the file. 11248f848716e80d01619b239111db48bfac77baad1Chris Lattner std::vector<Function*> FunctionsWithBodies; 1136994040a952e5fb27605eb3cf29ed86c4e59cf62Chandler Carruth 1146994040a952e5fb27605eb3cf29ed86c4e59cf62Chandler Carruth // When intrinsic functions are encountered which require upgrading they are 1156994040a952e5fb27605eb3cf29ed86c4e59cf62Chandler Carruth // stored here with their replacement function. 1166994040a952e5fb27605eb3cf29ed86c4e59cf62Chandler Carruth typedef std::vector<std::pair<Function*, Function*> > UpgradedIntrinsicMap; 1176994040a952e5fb27605eb3cf29ed86c4e59cf62Chandler Carruth UpgradedIntrinsicMap UpgradedIntrinsics; 11848f848716e80d01619b239111db48bfac77baad1Chris Lattner 11948f848716e80d01619b239111db48bfac77baad1Chris Lattner // After the module header has been read, the FunctionsWithBodies list is 12048f848716e80d01619b239111db48bfac77baad1Chris Lattner // reversed. This keeps track of whether we've done this yet. 12148f848716e80d01619b239111db48bfac77baad1Chris Lattner bool HasReversedFunctionsWithBodies; 12248f848716e80d01619b239111db48bfac77baad1Chris Lattner 12348f848716e80d01619b239111db48bfac77baad1Chris Lattner /// DeferredFunctionInfo - When function bodies are initially scanned, this 12448f848716e80d01619b239111db48bfac77baad1Chris Lattner /// map contains info about where to find deferred function body (in the 12548f848716e80d01619b239111db48bfac77baad1Chris Lattner /// stream) and what linkage the original function had. 12648f848716e80d01619b239111db48bfac77baad1Chris Lattner DenseMap<Function*, std::pair<uint64_t, unsigned> > DeferredFunctionInfo; 127caee0dccffb77a003681345ab3281bcf8684526cChris Lattnerpublic: 1284434ed44c45c87a72b7a0bf2f91211f895022b91Owen Anderson explicit BitcodeReader(MemoryBuffer *buffer, LLVMContext& C) 1298b477ed579794ba6d76915d56b3f448a7dd20120Owen Anderson : Context(C), Buffer(buffer), ErrorString(0) { 13048f848716e80d01619b239111db48bfac77baad1Chris Lattner HasReversedFunctionsWithBodies = false; 13148f848716e80d01619b239111db48bfac77baad1Chris Lattner } 132b348bb81253a1105d23ab1a1771f8d2a6546aa1bChris Lattner ~BitcodeReader() { 133b348bb81253a1105d23ab1a1771f8d2a6546aa1bChris Lattner FreeState(); 134b348bb81253a1105d23ab1a1771f8d2a6546aa1bChris Lattner } 135caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 136b348bb81253a1105d23ab1a1771f8d2a6546aa1bChris Lattner void FreeState(); 137c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner 138c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner /// releaseMemoryBuffer - This causes the reader to completely forget about 139c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner /// the memory buffer it contains, which prevents the buffer from being 140c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner /// destroyed when it is deleted. 141c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner void releaseMemoryBuffer() { 142c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner Buffer = 0; 143c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner } 144caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 14548f848716e80d01619b239111db48bfac77baad1Chris Lattner virtual bool materializeFunction(Function *F, std::string *ErrInfo = 0); 146980e5aad4cfaa32e13b297f4201eb1088ca96cc4Chris Lattner virtual Module *materializeModule(std::string *ErrInfo = 0); 147d67c632d968157e228cf42b588f8759059730ec0Chris Lattner virtual void dematerializeFunction(Function *F); 148b348bb81253a1105d23ab1a1771f8d2a6546aa1bChris Lattner virtual Module *releaseModule(std::string *ErrInfo = 0); 149d67c632d968157e228cf42b588f8759059730ec0Chris Lattner 150caee0dccffb77a003681345ab3281bcf8684526cChris Lattner bool Error(const char *Str) { 151caee0dccffb77a003681345ab3281bcf8684526cChris Lattner ErrorString = Str; 152caee0dccffb77a003681345ab3281bcf8684526cChris Lattner return true; 153caee0dccffb77a003681345ab3281bcf8684526cChris Lattner } 154caee0dccffb77a003681345ab3281bcf8684526cChris Lattner const char *getErrorString() const { return ErrorString; } 155caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 156caee0dccffb77a003681345ab3281bcf8684526cChris Lattner /// @brief Main interface to parsing a bitcode buffer. 157caee0dccffb77a003681345ab3281bcf8684526cChris Lattner /// @returns true if an error occurred. 158c453f76e2b4d7fd1e042b5b6d4c20556779186dfChris Lattner bool ParseBitcode(); 159caee0dccffb77a003681345ab3281bcf8684526cChris Lattnerprivate: 160caee0dccffb77a003681345ab3281bcf8684526cChris Lattner const Type *getTypeByID(unsigned ID, bool isTypeTable = false); 161a7c49aac984fafa5dfdfcc2762d4d51b26788e38Chris Lattner Value *getFnValueByID(unsigned ID, const Type *Ty) { 162a7c49aac984fafa5dfdfcc2762d4d51b26788e38Chris Lattner return ValueList.getValueFwdRef(ID, Ty); 163a7c49aac984fafa5dfdfcc2762d4d51b26788e38Chris Lattner } 164f4c8e5243376af58e52c4a0930d838509bbbea2fChris Lattner BasicBlock *getBasicBlock(unsigned ID) const { 165f4c8e5243376af58e52c4a0930d838509bbbea2fChris Lattner if (ID >= FunctionBBs.size()) return 0; // Invalid ID 166f4c8e5243376af58e52c4a0930d838509bbbea2fChris Lattner return FunctionBBs[ID]; 167f4c8e5243376af58e52c4a0930d838509bbbea2fChris Lattner } 1680598866c052147c31b808391f58434ce3dbfb838Devang Patel AttrListPtr getAttributes(unsigned i) const { 16919c874638d9478a5d5028854817a5ee72293bb2bDevang Patel if (i-1 < MAttributes.size()) 17019c874638d9478a5d5028854817a5ee72293bb2bDevang Patel return MAttributes[i-1]; 1710598866c052147c31b808391f58434ce3dbfb838Devang Patel return AttrListPtr(); 17248c85b84c1b66fb6a1b0d2afddf33da5bd82960dChris Lattner } 1737337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner 1747337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner /// getValueTypePair - Read a value/type pair out of the specified record from 1757337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner /// slot 'Slot'. Increment Slot past the number of slots used in the record. 1767337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner /// Return true on failure. 1777337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner bool getValueTypePair(SmallVector<uint64_t, 64> &Record, unsigned &Slot, 1787337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner unsigned InstNum, Value *&ResVal) { 1797337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner if (Slot == Record.size()) return true; 180650c938c41adc95e0935c75d7ed4349c46d280d1Jeff Cohen unsigned ValNo = (unsigned)Record[Slot++]; 1817337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner if (ValNo < InstNum) { 1827337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner // If this is not a forward reference, just return the value we already 1837337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner // have. 1847337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner ResVal = getFnValueByID(ValNo, 0); 1857337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner return ResVal == 0; 1867337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner } else if (Slot == Record.size()) { 1877337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner return true; 1887337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner } 1897337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner 190650c938c41adc95e0935c75d7ed4349c46d280d1Jeff Cohen unsigned TypeNo = (unsigned)Record[Slot++]; 1917337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner ResVal = getFnValueByID(ValNo, getTypeByID(TypeNo)); 1927337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner return ResVal == 0; 1937337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner } 1947337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner bool getValue(SmallVector<uint64_t, 64> &Record, unsigned &Slot, 1957337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner const Type *Ty, Value *&ResVal) { 1967337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner if (Slot == Record.size()) return true; 197650c938c41adc95e0935c75d7ed4349c46d280d1Jeff Cohen unsigned ValNo = (unsigned)Record[Slot++]; 1987337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner ResVal = getFnValueByID(ValNo, Ty); 1997337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner return ResVal == 0; 2007337ab9e929a4a787f90d37ee0ed8f4cc27170abChris Lattner } 20148c85b84c1b66fb6a1b0d2afddf33da5bd82960dChris Lattner 202caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 203866971474ced63f725bd3dbf10fdb9e3badb7d9eChris Lattner bool ParseModule(const std::string &ModuleID); 2040598866c052147c31b808391f58434ce3dbfb838Devang Patel bool ParseAttributeBlock(); 205866971474ced63f725bd3dbf10fdb9e3badb7d9eChris Lattner bool ParseTypeTable(); 206866971474ced63f725bd3dbf10fdb9e3badb7d9eChris Lattner bool ParseTypeSymbolTable(); 207866971474ced63f725bd3dbf10fdb9e3badb7d9eChris Lattner bool ParseValueSymbolTable(); 208866971474ced63f725bd3dbf10fdb9e3badb7d9eChris Lattner bool ParseConstants(); 209980e5aad4cfaa32e13b297f4201eb1088ca96cc4Chris Lattner bool RememberAndSkipFunctionBody(); 210980e5aad4cfaa32e13b297f4201eb1088ca96cc4Chris Lattner bool ParseFunctionBody(Function *F); 21107d98b4afbdcbb4eed048400d9116de1ec83e866Chris Lattner bool ResolveGlobalAndAliasInits(); 212caee0dccffb77a003681345ab3281bcf8684526cChris Lattner}; 213caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 214caee0dccffb77a003681345ab3281bcf8684526cChris Lattner} // End llvm namespace 215caee0dccffb77a003681345ab3281bcf8684526cChris Lattner 216caee0dccffb77a003681345ab3281bcf8684526cChris Lattner#endif 217