10b57cec5SDimitry Andric //===- Bitcode/Writer/BitcodeWriter.cpp - Bitcode Writer ------------------===// 20b57cec5SDimitry Andric // 30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 60b57cec5SDimitry Andric // 70b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 80b57cec5SDimitry Andric // 90b57cec5SDimitry Andric // Bitcode writer implementation. 100b57cec5SDimitry Andric // 110b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 120b57cec5SDimitry Andric 130b57cec5SDimitry Andric #include "llvm/Bitcode/BitcodeWriter.h" 140b57cec5SDimitry Andric #include "ValueEnumerator.h" 150b57cec5SDimitry Andric #include "llvm/ADT/APFloat.h" 160b57cec5SDimitry Andric #include "llvm/ADT/APInt.h" 170b57cec5SDimitry Andric #include "llvm/ADT/ArrayRef.h" 180b57cec5SDimitry Andric #include "llvm/ADT/DenseMap.h" 190b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h" 2081ad6265SDimitry Andric #include "llvm/ADT/SetVector.h" 2181ad6265SDimitry Andric #include "llvm/ADT/SmallPtrSet.h" 220b57cec5SDimitry Andric #include "llvm/ADT/SmallString.h" 230b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h" 240b57cec5SDimitry Andric #include "llvm/ADT/StringMap.h" 250b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h" 26e8d8bef9SDimitry Andric #include "llvm/Bitcode/BitcodeCommon.h" 27480093f4SDimitry Andric #include "llvm/Bitcode/BitcodeReader.h" 28480093f4SDimitry Andric #include "llvm/Bitcode/LLVMBitCodes.h" 290b57cec5SDimitry Andric #include "llvm/Bitstream/BitCodes.h" 300b57cec5SDimitry Andric #include "llvm/Bitstream/BitstreamWriter.h" 310b57cec5SDimitry Andric #include "llvm/Config/llvm-config.h" 320b57cec5SDimitry Andric #include "llvm/IR/Attributes.h" 330b57cec5SDimitry Andric #include "llvm/IR/BasicBlock.h" 340b57cec5SDimitry Andric #include "llvm/IR/Comdat.h" 350b57cec5SDimitry Andric #include "llvm/IR/Constant.h" 360b57cec5SDimitry Andric #include "llvm/IR/Constants.h" 370b57cec5SDimitry Andric #include "llvm/IR/DebugInfoMetadata.h" 380b57cec5SDimitry Andric #include "llvm/IR/DebugLoc.h" 390b57cec5SDimitry Andric #include "llvm/IR/DerivedTypes.h" 400b57cec5SDimitry Andric #include "llvm/IR/Function.h" 410b57cec5SDimitry Andric #include "llvm/IR/GlobalAlias.h" 420b57cec5SDimitry Andric #include "llvm/IR/GlobalIFunc.h" 430b57cec5SDimitry Andric #include "llvm/IR/GlobalObject.h" 440b57cec5SDimitry Andric #include "llvm/IR/GlobalValue.h" 450b57cec5SDimitry Andric #include "llvm/IR/GlobalVariable.h" 460b57cec5SDimitry Andric #include "llvm/IR/InlineAsm.h" 470b57cec5SDimitry Andric #include "llvm/IR/InstrTypes.h" 480b57cec5SDimitry Andric #include "llvm/IR/Instruction.h" 490b57cec5SDimitry Andric #include "llvm/IR/Instructions.h" 500b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h" 510b57cec5SDimitry Andric #include "llvm/IR/Metadata.h" 520b57cec5SDimitry Andric #include "llvm/IR/Module.h" 530b57cec5SDimitry Andric #include "llvm/IR/ModuleSummaryIndex.h" 540b57cec5SDimitry Andric #include "llvm/IR/Operator.h" 550b57cec5SDimitry Andric #include "llvm/IR/Type.h" 560b57cec5SDimitry Andric #include "llvm/IR/UseListOrder.h" 570b57cec5SDimitry Andric #include "llvm/IR/Value.h" 580b57cec5SDimitry Andric #include "llvm/IR/ValueSymbolTable.h" 590b57cec5SDimitry Andric #include "llvm/MC/StringTableBuilder.h" 60349cc55cSDimitry Andric #include "llvm/MC/TargetRegistry.h" 610b57cec5SDimitry Andric #include "llvm/Object/IRSymtab.h" 620b57cec5SDimitry Andric #include "llvm/Support/AtomicOrdering.h" 630b57cec5SDimitry Andric #include "llvm/Support/Casting.h" 640b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h" 650b57cec5SDimitry Andric #include "llvm/Support/Endian.h" 660b57cec5SDimitry Andric #include "llvm/Support/Error.h" 670b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h" 680b57cec5SDimitry Andric #include "llvm/Support/MathExtras.h" 690b57cec5SDimitry Andric #include "llvm/Support/SHA1.h" 700b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h" 71fe013be4SDimitry Andric #include "llvm/TargetParser/Triple.h" 720b57cec5SDimitry Andric #include <algorithm> 730b57cec5SDimitry Andric #include <cassert> 740b57cec5SDimitry Andric #include <cstddef> 750b57cec5SDimitry Andric #include <cstdint> 760b57cec5SDimitry Andric #include <iterator> 770b57cec5SDimitry Andric #include <map> 780b57cec5SDimitry Andric #include <memory> 79bdd1243dSDimitry Andric #include <optional> 800b57cec5SDimitry Andric #include <string> 810b57cec5SDimitry Andric #include <utility> 820b57cec5SDimitry Andric #include <vector> 830b57cec5SDimitry Andric 840b57cec5SDimitry Andric using namespace llvm; 850b57cec5SDimitry Andric 860b57cec5SDimitry Andric static cl::opt<unsigned> 870b57cec5SDimitry Andric IndexThreshold("bitcode-mdindex-threshold", cl::Hidden, cl::init(25), 880b57cec5SDimitry Andric cl::desc("Number of metadatas above which we emit an index " 890b57cec5SDimitry Andric "to enable lazy-loading")); 90e8d8bef9SDimitry Andric static cl::opt<uint32_t> FlushThreshold( 91e8d8bef9SDimitry Andric "bitcode-flush-threshold", cl::Hidden, cl::init(512), 92e8d8bef9SDimitry Andric cl::desc("The threshold (unit M) for flushing LLVM bitcode.")); 930b57cec5SDimitry Andric 948bcb0991SDimitry Andric static cl::opt<bool> WriteRelBFToSummary( 950b57cec5SDimitry Andric "write-relbf-to-summary", cl::Hidden, cl::init(false), 960b57cec5SDimitry Andric cl::desc("Write relative block frequency to function summary ")); 970b57cec5SDimitry Andric 98bdd1243dSDimitry Andric namespace llvm { 990b57cec5SDimitry Andric extern FunctionSummary::ForceSummaryHotnessType ForceSummaryEdgesCold; 100bdd1243dSDimitry Andric } 1010b57cec5SDimitry Andric 1020b57cec5SDimitry Andric namespace { 1030b57cec5SDimitry Andric 1040b57cec5SDimitry Andric /// These are manifest constants used by the bitcode writer. They do not need to 1050b57cec5SDimitry Andric /// be kept in sync with the reader, but need to be consistent within this file. 1060b57cec5SDimitry Andric enum { 1070b57cec5SDimitry Andric // VALUE_SYMTAB_BLOCK abbrev id's. 1080b57cec5SDimitry Andric VST_ENTRY_8_ABBREV = bitc::FIRST_APPLICATION_ABBREV, 1090b57cec5SDimitry Andric VST_ENTRY_7_ABBREV, 1100b57cec5SDimitry Andric VST_ENTRY_6_ABBREV, 1110b57cec5SDimitry Andric VST_BBENTRY_6_ABBREV, 1120b57cec5SDimitry Andric 1130b57cec5SDimitry Andric // CONSTANTS_BLOCK abbrev id's. 1140b57cec5SDimitry Andric CONSTANTS_SETTYPE_ABBREV = bitc::FIRST_APPLICATION_ABBREV, 1150b57cec5SDimitry Andric CONSTANTS_INTEGER_ABBREV, 1160b57cec5SDimitry Andric CONSTANTS_CE_CAST_Abbrev, 1170b57cec5SDimitry Andric CONSTANTS_NULL_Abbrev, 1180b57cec5SDimitry Andric 1190b57cec5SDimitry Andric // FUNCTION_BLOCK abbrev id's. 1200b57cec5SDimitry Andric FUNCTION_INST_LOAD_ABBREV = bitc::FIRST_APPLICATION_ABBREV, 1210b57cec5SDimitry Andric FUNCTION_INST_UNOP_ABBREV, 1220b57cec5SDimitry Andric FUNCTION_INST_UNOP_FLAGS_ABBREV, 1230b57cec5SDimitry Andric FUNCTION_INST_BINOP_ABBREV, 1240b57cec5SDimitry Andric FUNCTION_INST_BINOP_FLAGS_ABBREV, 1250b57cec5SDimitry Andric FUNCTION_INST_CAST_ABBREV, 126*c9157d92SDimitry Andric FUNCTION_INST_CAST_FLAGS_ABBREV, 1270b57cec5SDimitry Andric FUNCTION_INST_RET_VOID_ABBREV, 1280b57cec5SDimitry Andric FUNCTION_INST_RET_VAL_ABBREV, 1290b57cec5SDimitry Andric FUNCTION_INST_UNREACHABLE_ABBREV, 1300b57cec5SDimitry Andric FUNCTION_INST_GEP_ABBREV, 1310b57cec5SDimitry Andric }; 1320b57cec5SDimitry Andric 1330b57cec5SDimitry Andric /// Abstract class to manage the bitcode writing, subclassed for each bitcode 1340b57cec5SDimitry Andric /// file type. 1350b57cec5SDimitry Andric class BitcodeWriterBase { 1360b57cec5SDimitry Andric protected: 1370b57cec5SDimitry Andric /// The stream created and owned by the client. 1380b57cec5SDimitry Andric BitstreamWriter &Stream; 1390b57cec5SDimitry Andric 1400b57cec5SDimitry Andric StringTableBuilder &StrtabBuilder; 1410b57cec5SDimitry Andric 1420b57cec5SDimitry Andric public: 1430b57cec5SDimitry Andric /// Constructs a BitcodeWriterBase object that writes to the provided 1440b57cec5SDimitry Andric /// \p Stream. 1450b57cec5SDimitry Andric BitcodeWriterBase(BitstreamWriter &Stream, StringTableBuilder &StrtabBuilder) 1460b57cec5SDimitry Andric : Stream(Stream), StrtabBuilder(StrtabBuilder) {} 1470b57cec5SDimitry Andric 1480b57cec5SDimitry Andric protected: 1490b57cec5SDimitry Andric void writeModuleVersion(); 1500b57cec5SDimitry Andric }; 1510b57cec5SDimitry Andric 1520b57cec5SDimitry Andric void BitcodeWriterBase::writeModuleVersion() { 1530b57cec5SDimitry Andric // VERSION: [version#] 1540b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_VERSION, ArrayRef<uint64_t>{2}); 1550b57cec5SDimitry Andric } 1560b57cec5SDimitry Andric 1570b57cec5SDimitry Andric /// Base class to manage the module bitcode writing, currently subclassed for 1580b57cec5SDimitry Andric /// ModuleBitcodeWriter and ThinLinkBitcodeWriter. 1590b57cec5SDimitry Andric class ModuleBitcodeWriterBase : public BitcodeWriterBase { 1600b57cec5SDimitry Andric protected: 1610b57cec5SDimitry Andric /// The Module to write to bitcode. 1620b57cec5SDimitry Andric const Module &M; 1630b57cec5SDimitry Andric 1640b57cec5SDimitry Andric /// Enumerates ids for all values in the module. 1650b57cec5SDimitry Andric ValueEnumerator VE; 1660b57cec5SDimitry Andric 1670b57cec5SDimitry Andric /// Optional per-module index to write for ThinLTO. 1680b57cec5SDimitry Andric const ModuleSummaryIndex *Index; 1690b57cec5SDimitry Andric 1700b57cec5SDimitry Andric /// Map that holds the correspondence between GUIDs in the summary index, 1710b57cec5SDimitry Andric /// that came from indirect call profiles, and a value id generated by this 1720b57cec5SDimitry Andric /// class to use in the VST and summary block records. 1730b57cec5SDimitry Andric std::map<GlobalValue::GUID, unsigned> GUIDToValueIdMap; 1740b57cec5SDimitry Andric 1750b57cec5SDimitry Andric /// Tracks the last value id recorded in the GUIDToValueMap. 1760b57cec5SDimitry Andric unsigned GlobalValueId; 1770b57cec5SDimitry Andric 1780b57cec5SDimitry Andric /// Saves the offset of the VSTOffset record that must eventually be 1790b57cec5SDimitry Andric /// backpatched with the offset of the actual VST. 1800b57cec5SDimitry Andric uint64_t VSTOffsetPlaceholder = 0; 1810b57cec5SDimitry Andric 1820b57cec5SDimitry Andric public: 1830b57cec5SDimitry Andric /// Constructs a ModuleBitcodeWriterBase object for the given Module, 1840b57cec5SDimitry Andric /// writing to the provided \p Buffer. 1850b57cec5SDimitry Andric ModuleBitcodeWriterBase(const Module &M, StringTableBuilder &StrtabBuilder, 1860b57cec5SDimitry Andric BitstreamWriter &Stream, 1870b57cec5SDimitry Andric bool ShouldPreserveUseListOrder, 1880b57cec5SDimitry Andric const ModuleSummaryIndex *Index) 1890b57cec5SDimitry Andric : BitcodeWriterBase(Stream, StrtabBuilder), M(M), 1900b57cec5SDimitry Andric VE(M, ShouldPreserveUseListOrder), Index(Index) { 1910b57cec5SDimitry Andric // Assign ValueIds to any callee values in the index that came from 1920b57cec5SDimitry Andric // indirect call profiles and were recorded as a GUID not a Value* 1930b57cec5SDimitry Andric // (which would have been assigned an ID by the ValueEnumerator). 1940b57cec5SDimitry Andric // The starting ValueId is just after the number of values in the 1950b57cec5SDimitry Andric // ValueEnumerator, so that they can be emitted in the VST. 1960b57cec5SDimitry Andric GlobalValueId = VE.getValues().size(); 1970b57cec5SDimitry Andric if (!Index) 1980b57cec5SDimitry Andric return; 1990b57cec5SDimitry Andric for (const auto &GUIDSummaryLists : *Index) 2000b57cec5SDimitry Andric // Examine all summaries for this GUID. 2010b57cec5SDimitry Andric for (auto &Summary : GUIDSummaryLists.second.SummaryList) 2020b57cec5SDimitry Andric if (auto FS = dyn_cast<FunctionSummary>(Summary.get())) 2030b57cec5SDimitry Andric // For each call in the function summary, see if the call 2040b57cec5SDimitry Andric // is to a GUID (which means it is for an indirect call, 2050b57cec5SDimitry Andric // otherwise we would have a Value for it). If so, synthesize 2060b57cec5SDimitry Andric // a value id. 2070b57cec5SDimitry Andric for (auto &CallEdge : FS->calls()) 2080b57cec5SDimitry Andric if (!CallEdge.first.haveGVs() || !CallEdge.first.getValue()) 2090b57cec5SDimitry Andric assignValueId(CallEdge.first.getGUID()); 2100b57cec5SDimitry Andric } 2110b57cec5SDimitry Andric 2120b57cec5SDimitry Andric protected: 2130b57cec5SDimitry Andric void writePerModuleGlobalValueSummary(); 2140b57cec5SDimitry Andric 2150b57cec5SDimitry Andric private: 216bdd1243dSDimitry Andric void writePerModuleFunctionSummaryRecord( 217bdd1243dSDimitry Andric SmallVector<uint64_t, 64> &NameVals, GlobalValueSummary *Summary, 218bdd1243dSDimitry Andric unsigned ValueID, unsigned FSCallsAbbrev, unsigned FSCallsProfileAbbrev, 219bdd1243dSDimitry Andric unsigned CallsiteAbbrev, unsigned AllocAbbrev, const Function &F); 2200b57cec5SDimitry Andric void writeModuleLevelReferences(const GlobalVariable &V, 2210b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, 2220b57cec5SDimitry Andric unsigned FSModRefsAbbrev, 2230b57cec5SDimitry Andric unsigned FSModVTableRefsAbbrev); 2240b57cec5SDimitry Andric 2250b57cec5SDimitry Andric void assignValueId(GlobalValue::GUID ValGUID) { 2260b57cec5SDimitry Andric GUIDToValueIdMap[ValGUID] = ++GlobalValueId; 2270b57cec5SDimitry Andric } 2280b57cec5SDimitry Andric 2290b57cec5SDimitry Andric unsigned getValueId(GlobalValue::GUID ValGUID) { 2300b57cec5SDimitry Andric const auto &VMI = GUIDToValueIdMap.find(ValGUID); 2310b57cec5SDimitry Andric // Expect that any GUID value had a value Id assigned by an 2320b57cec5SDimitry Andric // earlier call to assignValueId. 2330b57cec5SDimitry Andric assert(VMI != GUIDToValueIdMap.end() && 2340b57cec5SDimitry Andric "GUID does not have assigned value Id"); 2350b57cec5SDimitry Andric return VMI->second; 2360b57cec5SDimitry Andric } 2370b57cec5SDimitry Andric 2380b57cec5SDimitry Andric // Helper to get the valueId for the type of value recorded in VI. 2390b57cec5SDimitry Andric unsigned getValueId(ValueInfo VI) { 2400b57cec5SDimitry Andric if (!VI.haveGVs() || !VI.getValue()) 2410b57cec5SDimitry Andric return getValueId(VI.getGUID()); 2420b57cec5SDimitry Andric return VE.getValueID(VI.getValue()); 2430b57cec5SDimitry Andric } 2440b57cec5SDimitry Andric 2450b57cec5SDimitry Andric std::map<GlobalValue::GUID, unsigned> &valueIds() { return GUIDToValueIdMap; } 2460b57cec5SDimitry Andric }; 2470b57cec5SDimitry Andric 2480b57cec5SDimitry Andric /// Class to manage the bitcode writing for a module. 2490b57cec5SDimitry Andric class ModuleBitcodeWriter : public ModuleBitcodeWriterBase { 2500b57cec5SDimitry Andric /// Pointer to the buffer allocated by caller for bitcode writing. 2510b57cec5SDimitry Andric const SmallVectorImpl<char> &Buffer; 2520b57cec5SDimitry Andric 2530b57cec5SDimitry Andric /// True if a module hash record should be written. 2540b57cec5SDimitry Andric bool GenerateHash; 2550b57cec5SDimitry Andric 2560b57cec5SDimitry Andric /// If non-null, when GenerateHash is true, the resulting hash is written 2570b57cec5SDimitry Andric /// into ModHash. 2580b57cec5SDimitry Andric ModuleHash *ModHash; 2590b57cec5SDimitry Andric 2600b57cec5SDimitry Andric SHA1 Hasher; 2610b57cec5SDimitry Andric 2620b57cec5SDimitry Andric /// The start bit of the identification block. 2630b57cec5SDimitry Andric uint64_t BitcodeStartBit; 2640b57cec5SDimitry Andric 2650b57cec5SDimitry Andric public: 2660b57cec5SDimitry Andric /// Constructs a ModuleBitcodeWriter object for the given Module, 2670b57cec5SDimitry Andric /// writing to the provided \p Buffer. 2680b57cec5SDimitry Andric ModuleBitcodeWriter(const Module &M, SmallVectorImpl<char> &Buffer, 2690b57cec5SDimitry Andric StringTableBuilder &StrtabBuilder, 2700b57cec5SDimitry Andric BitstreamWriter &Stream, bool ShouldPreserveUseListOrder, 2710b57cec5SDimitry Andric const ModuleSummaryIndex *Index, bool GenerateHash, 2720b57cec5SDimitry Andric ModuleHash *ModHash = nullptr) 2730b57cec5SDimitry Andric : ModuleBitcodeWriterBase(M, StrtabBuilder, Stream, 2740b57cec5SDimitry Andric ShouldPreserveUseListOrder, Index), 2750b57cec5SDimitry Andric Buffer(Buffer), GenerateHash(GenerateHash), ModHash(ModHash), 2760b57cec5SDimitry Andric BitcodeStartBit(Stream.GetCurrentBitNo()) {} 2770b57cec5SDimitry Andric 2780b57cec5SDimitry Andric /// Emit the current module to the bitstream. 2790b57cec5SDimitry Andric void write(); 2800b57cec5SDimitry Andric 2810b57cec5SDimitry Andric private: 2820b57cec5SDimitry Andric uint64_t bitcodeStartBit() { return BitcodeStartBit; } 2830b57cec5SDimitry Andric 2840b57cec5SDimitry Andric size_t addToStrtab(StringRef Str); 2850b57cec5SDimitry Andric 2860b57cec5SDimitry Andric void writeAttributeGroupTable(); 2870b57cec5SDimitry Andric void writeAttributeTable(); 2880b57cec5SDimitry Andric void writeTypeTable(); 2890b57cec5SDimitry Andric void writeComdats(); 2900b57cec5SDimitry Andric void writeValueSymbolTableForwardDecl(); 2910b57cec5SDimitry Andric void writeModuleInfo(); 2920b57cec5SDimitry Andric void writeValueAsMetadata(const ValueAsMetadata *MD, 2930b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record); 2940b57cec5SDimitry Andric void writeMDTuple(const MDTuple *N, SmallVectorImpl<uint64_t> &Record, 2950b57cec5SDimitry Andric unsigned Abbrev); 2960b57cec5SDimitry Andric unsigned createDILocationAbbrev(); 2970b57cec5SDimitry Andric void writeDILocation(const DILocation *N, SmallVectorImpl<uint64_t> &Record, 2980b57cec5SDimitry Andric unsigned &Abbrev); 2990b57cec5SDimitry Andric unsigned createGenericDINodeAbbrev(); 3000b57cec5SDimitry Andric void writeGenericDINode(const GenericDINode *N, 3010b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned &Abbrev); 3020b57cec5SDimitry Andric void writeDISubrange(const DISubrange *N, SmallVectorImpl<uint64_t> &Record, 3030b57cec5SDimitry Andric unsigned Abbrev); 304e8d8bef9SDimitry Andric void writeDIGenericSubrange(const DIGenericSubrange *N, 305e8d8bef9SDimitry Andric SmallVectorImpl<uint64_t> &Record, 306e8d8bef9SDimitry Andric unsigned Abbrev); 3070b57cec5SDimitry Andric void writeDIEnumerator(const DIEnumerator *N, 3080b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3090b57cec5SDimitry Andric void writeDIBasicType(const DIBasicType *N, SmallVectorImpl<uint64_t> &Record, 3100b57cec5SDimitry Andric unsigned Abbrev); 311e8d8bef9SDimitry Andric void writeDIStringType(const DIStringType *N, 312e8d8bef9SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3130b57cec5SDimitry Andric void writeDIDerivedType(const DIDerivedType *N, 3140b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3150b57cec5SDimitry Andric void writeDICompositeType(const DICompositeType *N, 3160b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3170b57cec5SDimitry Andric void writeDISubroutineType(const DISubroutineType *N, 3180b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 3190b57cec5SDimitry Andric unsigned Abbrev); 3200b57cec5SDimitry Andric void writeDIFile(const DIFile *N, SmallVectorImpl<uint64_t> &Record, 3210b57cec5SDimitry Andric unsigned Abbrev); 3220b57cec5SDimitry Andric void writeDICompileUnit(const DICompileUnit *N, 3230b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3240b57cec5SDimitry Andric void writeDISubprogram(const DISubprogram *N, 3250b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3260b57cec5SDimitry Andric void writeDILexicalBlock(const DILexicalBlock *N, 3270b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3280b57cec5SDimitry Andric void writeDILexicalBlockFile(const DILexicalBlockFile *N, 3290b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 3300b57cec5SDimitry Andric unsigned Abbrev); 3310b57cec5SDimitry Andric void writeDICommonBlock(const DICommonBlock *N, 3320b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3330b57cec5SDimitry Andric void writeDINamespace(const DINamespace *N, SmallVectorImpl<uint64_t> &Record, 3340b57cec5SDimitry Andric unsigned Abbrev); 3350b57cec5SDimitry Andric void writeDIMacro(const DIMacro *N, SmallVectorImpl<uint64_t> &Record, 3360b57cec5SDimitry Andric unsigned Abbrev); 3370b57cec5SDimitry Andric void writeDIMacroFile(const DIMacroFile *N, SmallVectorImpl<uint64_t> &Record, 3380b57cec5SDimitry Andric unsigned Abbrev); 339*c9157d92SDimitry Andric void writeDIArgList(const DIArgList *N, SmallVectorImpl<uint64_t> &Record); 3400b57cec5SDimitry Andric void writeDIModule(const DIModule *N, SmallVectorImpl<uint64_t> &Record, 3410b57cec5SDimitry Andric unsigned Abbrev); 342bdd1243dSDimitry Andric void writeDIAssignID(const DIAssignID *N, SmallVectorImpl<uint64_t> &Record, 343bdd1243dSDimitry Andric unsigned Abbrev); 3440b57cec5SDimitry Andric void writeDITemplateTypeParameter(const DITemplateTypeParameter *N, 3450b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 3460b57cec5SDimitry Andric unsigned Abbrev); 3470b57cec5SDimitry Andric void writeDITemplateValueParameter(const DITemplateValueParameter *N, 3480b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 3490b57cec5SDimitry Andric unsigned Abbrev); 3500b57cec5SDimitry Andric void writeDIGlobalVariable(const DIGlobalVariable *N, 3510b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 3520b57cec5SDimitry Andric unsigned Abbrev); 3530b57cec5SDimitry Andric void writeDILocalVariable(const DILocalVariable *N, 3540b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3550b57cec5SDimitry Andric void writeDILabel(const DILabel *N, 3560b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3570b57cec5SDimitry Andric void writeDIExpression(const DIExpression *N, 3580b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3590b57cec5SDimitry Andric void writeDIGlobalVariableExpression(const DIGlobalVariableExpression *N, 3600b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 3610b57cec5SDimitry Andric unsigned Abbrev); 3620b57cec5SDimitry Andric void writeDIObjCProperty(const DIObjCProperty *N, 3630b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 3640b57cec5SDimitry Andric void writeDIImportedEntity(const DIImportedEntity *N, 3650b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 3660b57cec5SDimitry Andric unsigned Abbrev); 3670b57cec5SDimitry Andric unsigned createNamedMetadataAbbrev(); 3680b57cec5SDimitry Andric void writeNamedMetadata(SmallVectorImpl<uint64_t> &Record); 3690b57cec5SDimitry Andric unsigned createMetadataStringsAbbrev(); 3700b57cec5SDimitry Andric void writeMetadataStrings(ArrayRef<const Metadata *> Strings, 3710b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record); 3720b57cec5SDimitry Andric void writeMetadataRecords(ArrayRef<const Metadata *> MDs, 3730b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 3740b57cec5SDimitry Andric std::vector<unsigned> *MDAbbrevs = nullptr, 3750b57cec5SDimitry Andric std::vector<uint64_t> *IndexPos = nullptr); 3760b57cec5SDimitry Andric void writeModuleMetadata(); 3770b57cec5SDimitry Andric void writeFunctionMetadata(const Function &F); 3780b57cec5SDimitry Andric void writeFunctionMetadataAttachment(const Function &F); 3790b57cec5SDimitry Andric void pushGlobalMetadataAttachment(SmallVectorImpl<uint64_t> &Record, 3800b57cec5SDimitry Andric const GlobalObject &GO); 3810b57cec5SDimitry Andric void writeModuleMetadataKinds(); 3820b57cec5SDimitry Andric void writeOperandBundleTags(); 3830b57cec5SDimitry Andric void writeSyncScopeNames(); 3840b57cec5SDimitry Andric void writeConstants(unsigned FirstVal, unsigned LastVal, bool isGlobal); 3850b57cec5SDimitry Andric void writeModuleConstants(); 3860b57cec5SDimitry Andric bool pushValueAndType(const Value *V, unsigned InstID, 3870b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals); 3885ffd83dbSDimitry Andric void writeOperandBundles(const CallBase &CB, unsigned InstID); 3890b57cec5SDimitry Andric void pushValue(const Value *V, unsigned InstID, 3900b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals); 3910b57cec5SDimitry Andric void pushValueSigned(const Value *V, unsigned InstID, 3920b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Vals); 3930b57cec5SDimitry Andric void writeInstruction(const Instruction &I, unsigned InstID, 3940b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals); 3950b57cec5SDimitry Andric void writeFunctionLevelValueSymbolTable(const ValueSymbolTable &VST); 3960b57cec5SDimitry Andric void writeGlobalValueSymbolTable( 3970b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> &FunctionToBitcodeIndex); 3980b57cec5SDimitry Andric void writeUseList(UseListOrder &&Order); 3990b57cec5SDimitry Andric void writeUseListBlock(const Function *F); 4000b57cec5SDimitry Andric void 4010b57cec5SDimitry Andric writeFunction(const Function &F, 4020b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> &FunctionToBitcodeIndex); 4030b57cec5SDimitry Andric void writeBlockInfo(); 4040b57cec5SDimitry Andric void writeModuleHash(size_t BlockStartPos); 4050b57cec5SDimitry Andric 4060b57cec5SDimitry Andric unsigned getEncodedSyncScopeID(SyncScope::ID SSID) { 4070b57cec5SDimitry Andric return unsigned(SSID); 4080b57cec5SDimitry Andric } 409e8d8bef9SDimitry Andric 410e8d8bef9SDimitry Andric unsigned getEncodedAlign(MaybeAlign Alignment) { return encode(Alignment); } 4110b57cec5SDimitry Andric }; 4120b57cec5SDimitry Andric 4130b57cec5SDimitry Andric /// Class to manage the bitcode writing for a combined index. 4140b57cec5SDimitry Andric class IndexBitcodeWriter : public BitcodeWriterBase { 4150b57cec5SDimitry Andric /// The combined index to write to bitcode. 4160b57cec5SDimitry Andric const ModuleSummaryIndex &Index; 4170b57cec5SDimitry Andric 4180b57cec5SDimitry Andric /// When writing a subset of the index for distributed backends, client 4190b57cec5SDimitry Andric /// provides a map of modules to the corresponding GUIDs/summaries to write. 4200b57cec5SDimitry Andric const std::map<std::string, GVSummaryMapTy> *ModuleToSummariesForIndex; 4210b57cec5SDimitry Andric 4220b57cec5SDimitry Andric /// Map that holds the correspondence between the GUID used in the combined 4230b57cec5SDimitry Andric /// index and a value id generated by this class to use in references. 4240b57cec5SDimitry Andric std::map<GlobalValue::GUID, unsigned> GUIDToValueIdMap; 4250b57cec5SDimitry Andric 426bdd1243dSDimitry Andric // The sorted stack id indices actually used in the summary entries being 427bdd1243dSDimitry Andric // written, which will be a subset of those in the full index in the case of 428bdd1243dSDimitry Andric // distributed indexes. 429bdd1243dSDimitry Andric std::vector<unsigned> StackIdIndices; 430bdd1243dSDimitry Andric 4310b57cec5SDimitry Andric /// Tracks the last value id recorded in the GUIDToValueMap. 4320b57cec5SDimitry Andric unsigned GlobalValueId = 0; 4330b57cec5SDimitry Andric 434*c9157d92SDimitry Andric /// Tracks the assignment of module paths in the module path string table to 435*c9157d92SDimitry Andric /// an id assigned for use in summary references to the module path. 436*c9157d92SDimitry Andric DenseMap<StringRef, uint64_t> ModuleIdMap; 437*c9157d92SDimitry Andric 4380b57cec5SDimitry Andric public: 4390b57cec5SDimitry Andric /// Constructs a IndexBitcodeWriter object for the given combined index, 4400b57cec5SDimitry Andric /// writing to the provided \p Buffer. When writing a subset of the index 4410b57cec5SDimitry Andric /// for a distributed backend, provide a \p ModuleToSummariesForIndex map. 4420b57cec5SDimitry Andric IndexBitcodeWriter(BitstreamWriter &Stream, StringTableBuilder &StrtabBuilder, 4430b57cec5SDimitry Andric const ModuleSummaryIndex &Index, 4440b57cec5SDimitry Andric const std::map<std::string, GVSummaryMapTy> 4450b57cec5SDimitry Andric *ModuleToSummariesForIndex = nullptr) 4460b57cec5SDimitry Andric : BitcodeWriterBase(Stream, StrtabBuilder), Index(Index), 4470b57cec5SDimitry Andric ModuleToSummariesForIndex(ModuleToSummariesForIndex) { 4480b57cec5SDimitry Andric // Assign unique value ids to all summaries to be written, for use 4490b57cec5SDimitry Andric // in writing out the call graph edges. Save the mapping from GUID 4500b57cec5SDimitry Andric // to the new global value id to use when writing those edges, which 4510b57cec5SDimitry Andric // are currently saved in the index in terms of GUID. 452bdd1243dSDimitry Andric forEachSummary([&](GVInfo I, bool IsAliasee) { 4530b57cec5SDimitry Andric GUIDToValueIdMap[I.first] = ++GlobalValueId; 454bdd1243dSDimitry Andric if (IsAliasee) 455bdd1243dSDimitry Andric return; 456bdd1243dSDimitry Andric auto *FS = dyn_cast<FunctionSummary>(I.second); 457bdd1243dSDimitry Andric if (!FS) 458bdd1243dSDimitry Andric return; 459bdd1243dSDimitry Andric // Record all stack id indices actually used in the summary entries being 460bdd1243dSDimitry Andric // written, so that we can compact them in the case of distributed ThinLTO 461bdd1243dSDimitry Andric // indexes. 462bdd1243dSDimitry Andric for (auto &CI : FS->callsites()) 463bdd1243dSDimitry Andric for (auto Idx : CI.StackIdIndices) 464bdd1243dSDimitry Andric StackIdIndices.push_back(Idx); 465bdd1243dSDimitry Andric for (auto &AI : FS->allocs()) 466bdd1243dSDimitry Andric for (auto &MIB : AI.MIBs) 467bdd1243dSDimitry Andric for (auto Idx : MIB.StackIdIndices) 468bdd1243dSDimitry Andric StackIdIndices.push_back(Idx); 4690b57cec5SDimitry Andric }); 470bdd1243dSDimitry Andric llvm::sort(StackIdIndices); 471bdd1243dSDimitry Andric StackIdIndices.erase( 472bdd1243dSDimitry Andric std::unique(StackIdIndices.begin(), StackIdIndices.end()), 473bdd1243dSDimitry Andric StackIdIndices.end()); 4740b57cec5SDimitry Andric } 4750b57cec5SDimitry Andric 4760b57cec5SDimitry Andric /// The below iterator returns the GUID and associated summary. 4770b57cec5SDimitry Andric using GVInfo = std::pair<GlobalValue::GUID, GlobalValueSummary *>; 4780b57cec5SDimitry Andric 4790b57cec5SDimitry Andric /// Calls the callback for each value GUID and summary to be written to 4800b57cec5SDimitry Andric /// bitcode. This hides the details of whether they are being pulled from the 4810b57cec5SDimitry Andric /// entire index or just those in a provided ModuleToSummariesForIndex map. 4820b57cec5SDimitry Andric template<typename Functor> 4830b57cec5SDimitry Andric void forEachSummary(Functor Callback) { 4840b57cec5SDimitry Andric if (ModuleToSummariesForIndex) { 4850b57cec5SDimitry Andric for (auto &M : *ModuleToSummariesForIndex) 4860b57cec5SDimitry Andric for (auto &Summary : M.second) { 4870b57cec5SDimitry Andric Callback(Summary, false); 4880b57cec5SDimitry Andric // Ensure aliasee is handled, e.g. for assigning a valueId, 4890b57cec5SDimitry Andric // even if we are not importing the aliasee directly (the 4900b57cec5SDimitry Andric // imported alias will contain a copy of aliasee). 4910b57cec5SDimitry Andric if (auto *AS = dyn_cast<AliasSummary>(Summary.getSecond())) 4920b57cec5SDimitry Andric Callback({AS->getAliaseeGUID(), &AS->getAliasee()}, true); 4930b57cec5SDimitry Andric } 4940b57cec5SDimitry Andric } else { 4950b57cec5SDimitry Andric for (auto &Summaries : Index) 4960b57cec5SDimitry Andric for (auto &Summary : Summaries.second.SummaryList) 4970b57cec5SDimitry Andric Callback({Summaries.first, Summary.get()}, false); 4980b57cec5SDimitry Andric } 4990b57cec5SDimitry Andric } 5000b57cec5SDimitry Andric 5010b57cec5SDimitry Andric /// Calls the callback for each entry in the modulePaths StringMap that 5020b57cec5SDimitry Andric /// should be written to the module path string table. This hides the details 5030b57cec5SDimitry Andric /// of whether they are being pulled from the entire index or just those in a 5040b57cec5SDimitry Andric /// provided ModuleToSummariesForIndex map. 5050b57cec5SDimitry Andric template <typename Functor> void forEachModule(Functor Callback) { 5060b57cec5SDimitry Andric if (ModuleToSummariesForIndex) { 5070b57cec5SDimitry Andric for (const auto &M : *ModuleToSummariesForIndex) { 5080b57cec5SDimitry Andric const auto &MPI = Index.modulePaths().find(M.first); 5090b57cec5SDimitry Andric if (MPI == Index.modulePaths().end()) { 5100b57cec5SDimitry Andric // This should only happen if the bitcode file was empty, in which 5110b57cec5SDimitry Andric // case we shouldn't be importing (the ModuleToSummariesForIndex 5120b57cec5SDimitry Andric // would only include the module we are writing and index for). 5130b57cec5SDimitry Andric assert(ModuleToSummariesForIndex->size() == 1); 5140b57cec5SDimitry Andric continue; 5150b57cec5SDimitry Andric } 5160b57cec5SDimitry Andric Callback(*MPI); 5170b57cec5SDimitry Andric } 5180b57cec5SDimitry Andric } else { 519*c9157d92SDimitry Andric // Since StringMap iteration order isn't guaranteed, order by path string 520*c9157d92SDimitry Andric // first. 521*c9157d92SDimitry Andric // FIXME: Make this a vector of StringMapEntry instead to avoid the later 522*c9157d92SDimitry Andric // map lookup. 523*c9157d92SDimitry Andric std::vector<StringRef> ModulePaths; 524*c9157d92SDimitry Andric for (auto &[ModPath, _] : Index.modulePaths()) 525*c9157d92SDimitry Andric ModulePaths.push_back(ModPath); 526*c9157d92SDimitry Andric llvm::sort(ModulePaths.begin(), ModulePaths.end()); 527*c9157d92SDimitry Andric for (auto &ModPath : ModulePaths) 528*c9157d92SDimitry Andric Callback(*Index.modulePaths().find(ModPath)); 5290b57cec5SDimitry Andric } 5300b57cec5SDimitry Andric } 5310b57cec5SDimitry Andric 5320b57cec5SDimitry Andric /// Main entry point for writing a combined index to bitcode. 5330b57cec5SDimitry Andric void write(); 5340b57cec5SDimitry Andric 5350b57cec5SDimitry Andric private: 5360b57cec5SDimitry Andric void writeModStrings(); 5370b57cec5SDimitry Andric void writeCombinedGlobalValueSummary(); 5380b57cec5SDimitry Andric 539bdd1243dSDimitry Andric std::optional<unsigned> getValueId(GlobalValue::GUID ValGUID) { 5400b57cec5SDimitry Andric auto VMI = GUIDToValueIdMap.find(ValGUID); 5410b57cec5SDimitry Andric if (VMI == GUIDToValueIdMap.end()) 542bdd1243dSDimitry Andric return std::nullopt; 5430b57cec5SDimitry Andric return VMI->second; 5440b57cec5SDimitry Andric } 5450b57cec5SDimitry Andric 5460b57cec5SDimitry Andric std::map<GlobalValue::GUID, unsigned> &valueIds() { return GUIDToValueIdMap; } 5470b57cec5SDimitry Andric }; 5480b57cec5SDimitry Andric 5490b57cec5SDimitry Andric } // end anonymous namespace 5500b57cec5SDimitry Andric 5510b57cec5SDimitry Andric static unsigned getEncodedCastOpcode(unsigned Opcode) { 5520b57cec5SDimitry Andric switch (Opcode) { 5530b57cec5SDimitry Andric default: llvm_unreachable("Unknown cast instruction!"); 5540b57cec5SDimitry Andric case Instruction::Trunc : return bitc::CAST_TRUNC; 5550b57cec5SDimitry Andric case Instruction::ZExt : return bitc::CAST_ZEXT; 5560b57cec5SDimitry Andric case Instruction::SExt : return bitc::CAST_SEXT; 5570b57cec5SDimitry Andric case Instruction::FPToUI : return bitc::CAST_FPTOUI; 5580b57cec5SDimitry Andric case Instruction::FPToSI : return bitc::CAST_FPTOSI; 5590b57cec5SDimitry Andric case Instruction::UIToFP : return bitc::CAST_UITOFP; 5600b57cec5SDimitry Andric case Instruction::SIToFP : return bitc::CAST_SITOFP; 5610b57cec5SDimitry Andric case Instruction::FPTrunc : return bitc::CAST_FPTRUNC; 5620b57cec5SDimitry Andric case Instruction::FPExt : return bitc::CAST_FPEXT; 5630b57cec5SDimitry Andric case Instruction::PtrToInt: return bitc::CAST_PTRTOINT; 5640b57cec5SDimitry Andric case Instruction::IntToPtr: return bitc::CAST_INTTOPTR; 5650b57cec5SDimitry Andric case Instruction::BitCast : return bitc::CAST_BITCAST; 5660b57cec5SDimitry Andric case Instruction::AddrSpaceCast: return bitc::CAST_ADDRSPACECAST; 5670b57cec5SDimitry Andric } 5680b57cec5SDimitry Andric } 5690b57cec5SDimitry Andric 5700b57cec5SDimitry Andric static unsigned getEncodedUnaryOpcode(unsigned Opcode) { 5710b57cec5SDimitry Andric switch (Opcode) { 5720b57cec5SDimitry Andric default: llvm_unreachable("Unknown binary instruction!"); 5738bcb0991SDimitry Andric case Instruction::FNeg: return bitc::UNOP_FNEG; 5740b57cec5SDimitry Andric } 5750b57cec5SDimitry Andric } 5760b57cec5SDimitry Andric 5770b57cec5SDimitry Andric static unsigned getEncodedBinaryOpcode(unsigned Opcode) { 5780b57cec5SDimitry Andric switch (Opcode) { 5790b57cec5SDimitry Andric default: llvm_unreachable("Unknown binary instruction!"); 5800b57cec5SDimitry Andric case Instruction::Add: 5810b57cec5SDimitry Andric case Instruction::FAdd: return bitc::BINOP_ADD; 5820b57cec5SDimitry Andric case Instruction::Sub: 5830b57cec5SDimitry Andric case Instruction::FSub: return bitc::BINOP_SUB; 5840b57cec5SDimitry Andric case Instruction::Mul: 5850b57cec5SDimitry Andric case Instruction::FMul: return bitc::BINOP_MUL; 5860b57cec5SDimitry Andric case Instruction::UDiv: return bitc::BINOP_UDIV; 5870b57cec5SDimitry Andric case Instruction::FDiv: 5880b57cec5SDimitry Andric case Instruction::SDiv: return bitc::BINOP_SDIV; 5890b57cec5SDimitry Andric case Instruction::URem: return bitc::BINOP_UREM; 5900b57cec5SDimitry Andric case Instruction::FRem: 5910b57cec5SDimitry Andric case Instruction::SRem: return bitc::BINOP_SREM; 5920b57cec5SDimitry Andric case Instruction::Shl: return bitc::BINOP_SHL; 5930b57cec5SDimitry Andric case Instruction::LShr: return bitc::BINOP_LSHR; 5940b57cec5SDimitry Andric case Instruction::AShr: return bitc::BINOP_ASHR; 5950b57cec5SDimitry Andric case Instruction::And: return bitc::BINOP_AND; 5960b57cec5SDimitry Andric case Instruction::Or: return bitc::BINOP_OR; 5970b57cec5SDimitry Andric case Instruction::Xor: return bitc::BINOP_XOR; 5980b57cec5SDimitry Andric } 5990b57cec5SDimitry Andric } 6000b57cec5SDimitry Andric 6010b57cec5SDimitry Andric static unsigned getEncodedRMWOperation(AtomicRMWInst::BinOp Op) { 6020b57cec5SDimitry Andric switch (Op) { 6030b57cec5SDimitry Andric default: llvm_unreachable("Unknown RMW operation!"); 6040b57cec5SDimitry Andric case AtomicRMWInst::Xchg: return bitc::RMW_XCHG; 6050b57cec5SDimitry Andric case AtomicRMWInst::Add: return bitc::RMW_ADD; 6060b57cec5SDimitry Andric case AtomicRMWInst::Sub: return bitc::RMW_SUB; 6070b57cec5SDimitry Andric case AtomicRMWInst::And: return bitc::RMW_AND; 6080b57cec5SDimitry Andric case AtomicRMWInst::Nand: return bitc::RMW_NAND; 6090b57cec5SDimitry Andric case AtomicRMWInst::Or: return bitc::RMW_OR; 6100b57cec5SDimitry Andric case AtomicRMWInst::Xor: return bitc::RMW_XOR; 6110b57cec5SDimitry Andric case AtomicRMWInst::Max: return bitc::RMW_MAX; 6120b57cec5SDimitry Andric case AtomicRMWInst::Min: return bitc::RMW_MIN; 6130b57cec5SDimitry Andric case AtomicRMWInst::UMax: return bitc::RMW_UMAX; 6140b57cec5SDimitry Andric case AtomicRMWInst::UMin: return bitc::RMW_UMIN; 6150b57cec5SDimitry Andric case AtomicRMWInst::FAdd: return bitc::RMW_FADD; 6160b57cec5SDimitry Andric case AtomicRMWInst::FSub: return bitc::RMW_FSUB; 617753f127fSDimitry Andric case AtomicRMWInst::FMax: return bitc::RMW_FMAX; 618753f127fSDimitry Andric case AtomicRMWInst::FMin: return bitc::RMW_FMIN; 619bdd1243dSDimitry Andric case AtomicRMWInst::UIncWrap: 620bdd1243dSDimitry Andric return bitc::RMW_UINC_WRAP; 621bdd1243dSDimitry Andric case AtomicRMWInst::UDecWrap: 622bdd1243dSDimitry Andric return bitc::RMW_UDEC_WRAP; 6230b57cec5SDimitry Andric } 6240b57cec5SDimitry Andric } 6250b57cec5SDimitry Andric 6260b57cec5SDimitry Andric static unsigned getEncodedOrdering(AtomicOrdering Ordering) { 6270b57cec5SDimitry Andric switch (Ordering) { 6280b57cec5SDimitry Andric case AtomicOrdering::NotAtomic: return bitc::ORDERING_NOTATOMIC; 6290b57cec5SDimitry Andric case AtomicOrdering::Unordered: return bitc::ORDERING_UNORDERED; 6300b57cec5SDimitry Andric case AtomicOrdering::Monotonic: return bitc::ORDERING_MONOTONIC; 6310b57cec5SDimitry Andric case AtomicOrdering::Acquire: return bitc::ORDERING_ACQUIRE; 6320b57cec5SDimitry Andric case AtomicOrdering::Release: return bitc::ORDERING_RELEASE; 6330b57cec5SDimitry Andric case AtomicOrdering::AcquireRelease: return bitc::ORDERING_ACQREL; 6340b57cec5SDimitry Andric case AtomicOrdering::SequentiallyConsistent: return bitc::ORDERING_SEQCST; 6350b57cec5SDimitry Andric } 6360b57cec5SDimitry Andric llvm_unreachable("Invalid ordering"); 6370b57cec5SDimitry Andric } 6380b57cec5SDimitry Andric 6390b57cec5SDimitry Andric static void writeStringRecord(BitstreamWriter &Stream, unsigned Code, 6400b57cec5SDimitry Andric StringRef Str, unsigned AbbrevToUse) { 6410b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 6420b57cec5SDimitry Andric 6430b57cec5SDimitry Andric // Code: [strchar x N] 6444824e7fdSDimitry Andric for (char C : Str) { 6454824e7fdSDimitry Andric if (AbbrevToUse && !BitCodeAbbrevOp::isChar6(C)) 6460b57cec5SDimitry Andric AbbrevToUse = 0; 6474824e7fdSDimitry Andric Vals.push_back(C); 6480b57cec5SDimitry Andric } 6490b57cec5SDimitry Andric 6500b57cec5SDimitry Andric // Emit the finished record. 6510b57cec5SDimitry Andric Stream.EmitRecord(Code, Vals, AbbrevToUse); 6520b57cec5SDimitry Andric } 6530b57cec5SDimitry Andric 6540b57cec5SDimitry Andric static uint64_t getAttrKindEncoding(Attribute::AttrKind Kind) { 6550b57cec5SDimitry Andric switch (Kind) { 6560b57cec5SDimitry Andric case Attribute::Alignment: 6570b57cec5SDimitry Andric return bitc::ATTR_KIND_ALIGNMENT; 65881ad6265SDimitry Andric case Attribute::AllocAlign: 65981ad6265SDimitry Andric return bitc::ATTR_KIND_ALLOC_ALIGN; 6600b57cec5SDimitry Andric case Attribute::AllocSize: 6610b57cec5SDimitry Andric return bitc::ATTR_KIND_ALLOC_SIZE; 6620b57cec5SDimitry Andric case Attribute::AlwaysInline: 6630b57cec5SDimitry Andric return bitc::ATTR_KIND_ALWAYS_INLINE; 6640b57cec5SDimitry Andric case Attribute::Builtin: 6650b57cec5SDimitry Andric return bitc::ATTR_KIND_BUILTIN; 6660b57cec5SDimitry Andric case Attribute::ByVal: 6670b57cec5SDimitry Andric return bitc::ATTR_KIND_BY_VAL; 6680b57cec5SDimitry Andric case Attribute::Convergent: 6690b57cec5SDimitry Andric return bitc::ATTR_KIND_CONVERGENT; 6700b57cec5SDimitry Andric case Attribute::InAlloca: 6710b57cec5SDimitry Andric return bitc::ATTR_KIND_IN_ALLOCA; 6720b57cec5SDimitry Andric case Attribute::Cold: 6730b57cec5SDimitry Andric return bitc::ATTR_KIND_COLD; 674349cc55cSDimitry Andric case Attribute::DisableSanitizerInstrumentation: 675349cc55cSDimitry Andric return bitc::ATTR_KIND_DISABLE_SANITIZER_INSTRUMENTATION; 676753f127fSDimitry Andric case Attribute::FnRetThunkExtern: 677753f127fSDimitry Andric return bitc::ATTR_KIND_FNRETTHUNK_EXTERN; 678e8d8bef9SDimitry Andric case Attribute::Hot: 679e8d8bef9SDimitry Andric return bitc::ATTR_KIND_HOT; 680fe6060f1SDimitry Andric case Attribute::ElementType: 681fe6060f1SDimitry Andric return bitc::ATTR_KIND_ELEMENTTYPE; 6820b57cec5SDimitry Andric case Attribute::InlineHint: 6830b57cec5SDimitry Andric return bitc::ATTR_KIND_INLINE_HINT; 6840b57cec5SDimitry Andric case Attribute::InReg: 6850b57cec5SDimitry Andric return bitc::ATTR_KIND_IN_REG; 6860b57cec5SDimitry Andric case Attribute::JumpTable: 6870b57cec5SDimitry Andric return bitc::ATTR_KIND_JUMP_TABLE; 6880b57cec5SDimitry Andric case Attribute::MinSize: 6890b57cec5SDimitry Andric return bitc::ATTR_KIND_MIN_SIZE; 69081ad6265SDimitry Andric case Attribute::AllocatedPointer: 69181ad6265SDimitry Andric return bitc::ATTR_KIND_ALLOCATED_POINTER; 69281ad6265SDimitry Andric case Attribute::AllocKind: 69381ad6265SDimitry Andric return bitc::ATTR_KIND_ALLOC_KIND; 694bdd1243dSDimitry Andric case Attribute::Memory: 695bdd1243dSDimitry Andric return bitc::ATTR_KIND_MEMORY; 696fe013be4SDimitry Andric case Attribute::NoFPClass: 697fe013be4SDimitry Andric return bitc::ATTR_KIND_NOFPCLASS; 6980b57cec5SDimitry Andric case Attribute::Naked: 6990b57cec5SDimitry Andric return bitc::ATTR_KIND_NAKED; 7000b57cec5SDimitry Andric case Attribute::Nest: 7010b57cec5SDimitry Andric return bitc::ATTR_KIND_NEST; 7020b57cec5SDimitry Andric case Attribute::NoAlias: 7030b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_ALIAS; 7040b57cec5SDimitry Andric case Attribute::NoBuiltin: 7050b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_BUILTIN; 706e8d8bef9SDimitry Andric case Attribute::NoCallback: 707e8d8bef9SDimitry Andric return bitc::ATTR_KIND_NO_CALLBACK; 7080b57cec5SDimitry Andric case Attribute::NoCapture: 7090b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_CAPTURE; 7100b57cec5SDimitry Andric case Attribute::NoDuplicate: 7110b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_DUPLICATE; 7120b57cec5SDimitry Andric case Attribute::NoFree: 7130b57cec5SDimitry Andric return bitc::ATTR_KIND_NOFREE; 7140b57cec5SDimitry Andric case Attribute::NoImplicitFloat: 7150b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_IMPLICIT_FLOAT; 7160b57cec5SDimitry Andric case Attribute::NoInline: 7170b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_INLINE; 7180b57cec5SDimitry Andric case Attribute::NoRecurse: 7190b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_RECURSE; 7205ffd83dbSDimitry Andric case Attribute::NoMerge: 7215ffd83dbSDimitry Andric return bitc::ATTR_KIND_NO_MERGE; 7220b57cec5SDimitry Andric case Attribute::NonLazyBind: 7230b57cec5SDimitry Andric return bitc::ATTR_KIND_NON_LAZY_BIND; 7240b57cec5SDimitry Andric case Attribute::NonNull: 7250b57cec5SDimitry Andric return bitc::ATTR_KIND_NON_NULL; 7260b57cec5SDimitry Andric case Attribute::Dereferenceable: 7270b57cec5SDimitry Andric return bitc::ATTR_KIND_DEREFERENCEABLE; 7280b57cec5SDimitry Andric case Attribute::DereferenceableOrNull: 7290b57cec5SDimitry Andric return bitc::ATTR_KIND_DEREFERENCEABLE_OR_NULL; 7300b57cec5SDimitry Andric case Attribute::NoRedZone: 7310b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_RED_ZONE; 7320b57cec5SDimitry Andric case Attribute::NoReturn: 7330b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_RETURN; 7340b57cec5SDimitry Andric case Attribute::NoSync: 7350b57cec5SDimitry Andric return bitc::ATTR_KIND_NOSYNC; 7360b57cec5SDimitry Andric case Attribute::NoCfCheck: 7370b57cec5SDimitry Andric return bitc::ATTR_KIND_NOCF_CHECK; 738e8d8bef9SDimitry Andric case Attribute::NoProfile: 739e8d8bef9SDimitry Andric return bitc::ATTR_KIND_NO_PROFILE; 740bdd1243dSDimitry Andric case Attribute::SkipProfile: 741bdd1243dSDimitry Andric return bitc::ATTR_KIND_SKIP_PROFILE; 7420b57cec5SDimitry Andric case Attribute::NoUnwind: 7430b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_UNWIND; 74481ad6265SDimitry Andric case Attribute::NoSanitizeBounds: 74581ad6265SDimitry Andric return bitc::ATTR_KIND_NO_SANITIZE_BOUNDS; 746fe6060f1SDimitry Andric case Attribute::NoSanitizeCoverage: 747fe6060f1SDimitry Andric return bitc::ATTR_KIND_NO_SANITIZE_COVERAGE; 7485ffd83dbSDimitry Andric case Attribute::NullPointerIsValid: 7495ffd83dbSDimitry Andric return bitc::ATTR_KIND_NULL_POINTER_IS_VALID; 750*c9157d92SDimitry Andric case Attribute::OptimizeForDebugging: 751*c9157d92SDimitry Andric return bitc::ATTR_KIND_OPTIMIZE_FOR_DEBUGGING; 7520b57cec5SDimitry Andric case Attribute::OptForFuzzing: 7530b57cec5SDimitry Andric return bitc::ATTR_KIND_OPT_FOR_FUZZING; 7540b57cec5SDimitry Andric case Attribute::OptimizeForSize: 7550b57cec5SDimitry Andric return bitc::ATTR_KIND_OPTIMIZE_FOR_SIZE; 7560b57cec5SDimitry Andric case Attribute::OptimizeNone: 7570b57cec5SDimitry Andric return bitc::ATTR_KIND_OPTIMIZE_NONE; 7580b57cec5SDimitry Andric case Attribute::ReadNone: 7590b57cec5SDimitry Andric return bitc::ATTR_KIND_READ_NONE; 7600b57cec5SDimitry Andric case Attribute::ReadOnly: 7610b57cec5SDimitry Andric return bitc::ATTR_KIND_READ_ONLY; 7620b57cec5SDimitry Andric case Attribute::Returned: 7630b57cec5SDimitry Andric return bitc::ATTR_KIND_RETURNED; 7640b57cec5SDimitry Andric case Attribute::ReturnsTwice: 7650b57cec5SDimitry Andric return bitc::ATTR_KIND_RETURNS_TWICE; 7660b57cec5SDimitry Andric case Attribute::SExt: 7670b57cec5SDimitry Andric return bitc::ATTR_KIND_S_EXT; 7680b57cec5SDimitry Andric case Attribute::Speculatable: 7690b57cec5SDimitry Andric return bitc::ATTR_KIND_SPECULATABLE; 7700b57cec5SDimitry Andric case Attribute::StackAlignment: 7710b57cec5SDimitry Andric return bitc::ATTR_KIND_STACK_ALIGNMENT; 7720b57cec5SDimitry Andric case Attribute::StackProtect: 7730b57cec5SDimitry Andric return bitc::ATTR_KIND_STACK_PROTECT; 7740b57cec5SDimitry Andric case Attribute::StackProtectReq: 7750b57cec5SDimitry Andric return bitc::ATTR_KIND_STACK_PROTECT_REQ; 7760b57cec5SDimitry Andric case Attribute::StackProtectStrong: 7770b57cec5SDimitry Andric return bitc::ATTR_KIND_STACK_PROTECT_STRONG; 7780b57cec5SDimitry Andric case Attribute::SafeStack: 7790b57cec5SDimitry Andric return bitc::ATTR_KIND_SAFESTACK; 7800b57cec5SDimitry Andric case Attribute::ShadowCallStack: 7810b57cec5SDimitry Andric return bitc::ATTR_KIND_SHADOWCALLSTACK; 7820b57cec5SDimitry Andric case Attribute::StrictFP: 7830b57cec5SDimitry Andric return bitc::ATTR_KIND_STRICT_FP; 7840b57cec5SDimitry Andric case Attribute::StructRet: 7850b57cec5SDimitry Andric return bitc::ATTR_KIND_STRUCT_RET; 7860b57cec5SDimitry Andric case Attribute::SanitizeAddress: 7870b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_ADDRESS; 7880b57cec5SDimitry Andric case Attribute::SanitizeHWAddress: 7890b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_HWADDRESS; 7900b57cec5SDimitry Andric case Attribute::SanitizeThread: 7910b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_THREAD; 7920b57cec5SDimitry Andric case Attribute::SanitizeMemory: 7930b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_MEMORY; 7940b57cec5SDimitry Andric case Attribute::SpeculativeLoadHardening: 7950b57cec5SDimitry Andric return bitc::ATTR_KIND_SPECULATIVE_LOAD_HARDENING; 7960b57cec5SDimitry Andric case Attribute::SwiftError: 7970b57cec5SDimitry Andric return bitc::ATTR_KIND_SWIFT_ERROR; 7980b57cec5SDimitry Andric case Attribute::SwiftSelf: 7990b57cec5SDimitry Andric return bitc::ATTR_KIND_SWIFT_SELF; 800fe6060f1SDimitry Andric case Attribute::SwiftAsync: 801fe6060f1SDimitry Andric return bitc::ATTR_KIND_SWIFT_ASYNC; 8020b57cec5SDimitry Andric case Attribute::UWTable: 8030b57cec5SDimitry Andric return bitc::ATTR_KIND_UW_TABLE; 804fe6060f1SDimitry Andric case Attribute::VScaleRange: 805fe6060f1SDimitry Andric return bitc::ATTR_KIND_VSCALE_RANGE; 8060b57cec5SDimitry Andric case Attribute::WillReturn: 8070b57cec5SDimitry Andric return bitc::ATTR_KIND_WILLRETURN; 8080b57cec5SDimitry Andric case Attribute::WriteOnly: 8090b57cec5SDimitry Andric return bitc::ATTR_KIND_WRITEONLY; 8100b57cec5SDimitry Andric case Attribute::ZExt: 8110b57cec5SDimitry Andric return bitc::ATTR_KIND_Z_EXT; 8120b57cec5SDimitry Andric case Attribute::ImmArg: 8130b57cec5SDimitry Andric return bitc::ATTR_KIND_IMMARG; 8140b57cec5SDimitry Andric case Attribute::SanitizeMemTag: 8150b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_MEMTAG; 8165ffd83dbSDimitry Andric case Attribute::Preallocated: 8175ffd83dbSDimitry Andric return bitc::ATTR_KIND_PREALLOCATED; 8185ffd83dbSDimitry Andric case Attribute::NoUndef: 8195ffd83dbSDimitry Andric return bitc::ATTR_KIND_NOUNDEF; 820e8d8bef9SDimitry Andric case Attribute::ByRef: 821e8d8bef9SDimitry Andric return bitc::ATTR_KIND_BYREF; 822e8d8bef9SDimitry Andric case Attribute::MustProgress: 823e8d8bef9SDimitry Andric return bitc::ATTR_KIND_MUSTPROGRESS; 82481ad6265SDimitry Andric case Attribute::PresplitCoroutine: 82581ad6265SDimitry Andric return bitc::ATTR_KIND_PRESPLIT_COROUTINE; 826*c9157d92SDimitry Andric case Attribute::Writable: 827*c9157d92SDimitry Andric return bitc::ATTR_KIND_WRITABLE; 828*c9157d92SDimitry Andric case Attribute::CoroDestroyOnlyWhenComplete: 829*c9157d92SDimitry Andric return bitc::ATTR_KIND_CORO_ONLY_DESTROY_WHEN_COMPLETE; 830*c9157d92SDimitry Andric case Attribute::DeadOnUnwind: 831*c9157d92SDimitry Andric return bitc::ATTR_KIND_DEAD_ON_UNWIND; 8320b57cec5SDimitry Andric case Attribute::EndAttrKinds: 8330b57cec5SDimitry Andric llvm_unreachable("Can not encode end-attribute kinds marker."); 8340b57cec5SDimitry Andric case Attribute::None: 8350b57cec5SDimitry Andric llvm_unreachable("Can not encode none-attribute."); 8365ffd83dbSDimitry Andric case Attribute::EmptyKey: 8375ffd83dbSDimitry Andric case Attribute::TombstoneKey: 8385ffd83dbSDimitry Andric llvm_unreachable("Trying to encode EmptyKey/TombstoneKey"); 8390b57cec5SDimitry Andric } 8400b57cec5SDimitry Andric 8410b57cec5SDimitry Andric llvm_unreachable("Trying to encode unknown attribute"); 8420b57cec5SDimitry Andric } 8430b57cec5SDimitry Andric 8440b57cec5SDimitry Andric void ModuleBitcodeWriter::writeAttributeGroupTable() { 8450b57cec5SDimitry Andric const std::vector<ValueEnumerator::IndexAndAttrSet> &AttrGrps = 8460b57cec5SDimitry Andric VE.getAttributeGroups(); 8470b57cec5SDimitry Andric if (AttrGrps.empty()) return; 8480b57cec5SDimitry Andric 8490b57cec5SDimitry Andric Stream.EnterSubblock(bitc::PARAMATTR_GROUP_BLOCK_ID, 3); 8500b57cec5SDimitry Andric 8510b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 8520b57cec5SDimitry Andric for (ValueEnumerator::IndexAndAttrSet Pair : AttrGrps) { 8530b57cec5SDimitry Andric unsigned AttrListIndex = Pair.first; 8540b57cec5SDimitry Andric AttributeSet AS = Pair.second; 8550b57cec5SDimitry Andric Record.push_back(VE.getAttributeGroupID(Pair)); 8560b57cec5SDimitry Andric Record.push_back(AttrListIndex); 8570b57cec5SDimitry Andric 8580b57cec5SDimitry Andric for (Attribute Attr : AS) { 8590b57cec5SDimitry Andric if (Attr.isEnumAttribute()) { 8600b57cec5SDimitry Andric Record.push_back(0); 8610b57cec5SDimitry Andric Record.push_back(getAttrKindEncoding(Attr.getKindAsEnum())); 8620b57cec5SDimitry Andric } else if (Attr.isIntAttribute()) { 8630b57cec5SDimitry Andric Record.push_back(1); 8640b57cec5SDimitry Andric Record.push_back(getAttrKindEncoding(Attr.getKindAsEnum())); 8650b57cec5SDimitry Andric Record.push_back(Attr.getValueAsInt()); 8660b57cec5SDimitry Andric } else if (Attr.isStringAttribute()) { 8670b57cec5SDimitry Andric StringRef Kind = Attr.getKindAsString(); 8680b57cec5SDimitry Andric StringRef Val = Attr.getValueAsString(); 8690b57cec5SDimitry Andric 8700b57cec5SDimitry Andric Record.push_back(Val.empty() ? 3 : 4); 8710b57cec5SDimitry Andric Record.append(Kind.begin(), Kind.end()); 8720b57cec5SDimitry Andric Record.push_back(0); 8730b57cec5SDimitry Andric if (!Val.empty()) { 8740b57cec5SDimitry Andric Record.append(Val.begin(), Val.end()); 8750b57cec5SDimitry Andric Record.push_back(0); 8760b57cec5SDimitry Andric } 8770b57cec5SDimitry Andric } else { 8780b57cec5SDimitry Andric assert(Attr.isTypeAttribute()); 8790b57cec5SDimitry Andric Type *Ty = Attr.getValueAsType(); 8800b57cec5SDimitry Andric Record.push_back(Ty ? 6 : 5); 8810b57cec5SDimitry Andric Record.push_back(getAttrKindEncoding(Attr.getKindAsEnum())); 8820b57cec5SDimitry Andric if (Ty) 8830b57cec5SDimitry Andric Record.push_back(VE.getTypeID(Attr.getValueAsType())); 8840b57cec5SDimitry Andric } 8850b57cec5SDimitry Andric } 8860b57cec5SDimitry Andric 8870b57cec5SDimitry Andric Stream.EmitRecord(bitc::PARAMATTR_GRP_CODE_ENTRY, Record); 8880b57cec5SDimitry Andric Record.clear(); 8890b57cec5SDimitry Andric } 8900b57cec5SDimitry Andric 8910b57cec5SDimitry Andric Stream.ExitBlock(); 8920b57cec5SDimitry Andric } 8930b57cec5SDimitry Andric 8940b57cec5SDimitry Andric void ModuleBitcodeWriter::writeAttributeTable() { 8950b57cec5SDimitry Andric const std::vector<AttributeList> &Attrs = VE.getAttributeLists(); 8960b57cec5SDimitry Andric if (Attrs.empty()) return; 8970b57cec5SDimitry Andric 8980b57cec5SDimitry Andric Stream.EnterSubblock(bitc::PARAMATTR_BLOCK_ID, 3); 8990b57cec5SDimitry Andric 9000b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 9010eae32dcSDimitry Andric for (const AttributeList &AL : Attrs) { 902349cc55cSDimitry Andric for (unsigned i : AL.indexes()) { 9030b57cec5SDimitry Andric AttributeSet AS = AL.getAttributes(i); 9040b57cec5SDimitry Andric if (AS.hasAttributes()) 9050b57cec5SDimitry Andric Record.push_back(VE.getAttributeGroupID({i, AS})); 9060b57cec5SDimitry Andric } 9070b57cec5SDimitry Andric 9080b57cec5SDimitry Andric Stream.EmitRecord(bitc::PARAMATTR_CODE_ENTRY, Record); 9090b57cec5SDimitry Andric Record.clear(); 9100b57cec5SDimitry Andric } 9110b57cec5SDimitry Andric 9120b57cec5SDimitry Andric Stream.ExitBlock(); 9130b57cec5SDimitry Andric } 9140b57cec5SDimitry Andric 9150b57cec5SDimitry Andric /// WriteTypeTable - Write out the type table for a module. 9160b57cec5SDimitry Andric void ModuleBitcodeWriter::writeTypeTable() { 9170b57cec5SDimitry Andric const ValueEnumerator::TypeList &TypeList = VE.getTypes(); 9180b57cec5SDimitry Andric 9190b57cec5SDimitry Andric Stream.EnterSubblock(bitc::TYPE_BLOCK_ID_NEW, 4 /*count from # abbrevs */); 9200b57cec5SDimitry Andric SmallVector<uint64_t, 64> TypeVals; 9210b57cec5SDimitry Andric 9220b57cec5SDimitry Andric uint64_t NumBits = VE.computeBitsRequiredForTypeIndicies(); 9230b57cec5SDimitry Andric 924fe6060f1SDimitry Andric // Abbrev for TYPE_CODE_OPAQUE_POINTER. 925fe013be4SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 926fe6060f1SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_OPAQUE_POINTER)); 927fe6060f1SDimitry Andric Abbv->Add(BitCodeAbbrevOp(0)); // Addrspace = 0 928fe6060f1SDimitry Andric unsigned OpaquePtrAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 929fe6060f1SDimitry Andric 9300b57cec5SDimitry Andric // Abbrev for TYPE_CODE_FUNCTION. 9310b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 9320b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_FUNCTION)); 9330b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // isvararg 9340b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 9350b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 9360b57cec5SDimitry Andric unsigned FunctionAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 9370b57cec5SDimitry Andric 9380b57cec5SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_ANON. 9390b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 9400b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_ANON)); 9410b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // ispacked 9420b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 9430b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 9440b57cec5SDimitry Andric unsigned StructAnonAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 9450b57cec5SDimitry Andric 9460b57cec5SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_NAME. 9470b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 9480b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_NAME)); 9490b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 9500b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 9510b57cec5SDimitry Andric unsigned StructNameAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 9520b57cec5SDimitry Andric 9530b57cec5SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_NAMED. 9540b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 9550b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_NAMED)); 9560b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // ispacked 9570b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 9580b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 9590b57cec5SDimitry Andric unsigned StructNamedAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 9600b57cec5SDimitry Andric 9610b57cec5SDimitry Andric // Abbrev for TYPE_CODE_ARRAY. 9620b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 9630b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_ARRAY)); 9640b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // size 9650b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 9660b57cec5SDimitry Andric unsigned ArrayAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 9670b57cec5SDimitry Andric 9680b57cec5SDimitry Andric // Emit an entry count so the reader can reserve space. 9690b57cec5SDimitry Andric TypeVals.push_back(TypeList.size()); 9700b57cec5SDimitry Andric Stream.EmitRecord(bitc::TYPE_CODE_NUMENTRY, TypeVals); 9710b57cec5SDimitry Andric TypeVals.clear(); 9720b57cec5SDimitry Andric 9730b57cec5SDimitry Andric // Loop over all of the types, emitting each in turn. 9744824e7fdSDimitry Andric for (Type *T : TypeList) { 9750b57cec5SDimitry Andric int AbbrevToUse = 0; 9760b57cec5SDimitry Andric unsigned Code = 0; 9770b57cec5SDimitry Andric 9780b57cec5SDimitry Andric switch (T->getTypeID()) { 9790b57cec5SDimitry Andric case Type::VoidTyID: Code = bitc::TYPE_CODE_VOID; break; 9800b57cec5SDimitry Andric case Type::HalfTyID: Code = bitc::TYPE_CODE_HALF; break; 9815ffd83dbSDimitry Andric case Type::BFloatTyID: Code = bitc::TYPE_CODE_BFLOAT; break; 9820b57cec5SDimitry Andric case Type::FloatTyID: Code = bitc::TYPE_CODE_FLOAT; break; 9830b57cec5SDimitry Andric case Type::DoubleTyID: Code = bitc::TYPE_CODE_DOUBLE; break; 9840b57cec5SDimitry Andric case Type::X86_FP80TyID: Code = bitc::TYPE_CODE_X86_FP80; break; 9850b57cec5SDimitry Andric case Type::FP128TyID: Code = bitc::TYPE_CODE_FP128; break; 9860b57cec5SDimitry Andric case Type::PPC_FP128TyID: Code = bitc::TYPE_CODE_PPC_FP128; break; 9870b57cec5SDimitry Andric case Type::LabelTyID: Code = bitc::TYPE_CODE_LABEL; break; 9880b57cec5SDimitry Andric case Type::MetadataTyID: Code = bitc::TYPE_CODE_METADATA; break; 9890b57cec5SDimitry Andric case Type::X86_MMXTyID: Code = bitc::TYPE_CODE_X86_MMX; break; 990e8d8bef9SDimitry Andric case Type::X86_AMXTyID: Code = bitc::TYPE_CODE_X86_AMX; break; 9910b57cec5SDimitry Andric case Type::TokenTyID: Code = bitc::TYPE_CODE_TOKEN; break; 9920b57cec5SDimitry Andric case Type::IntegerTyID: 9930b57cec5SDimitry Andric // INTEGER: [width] 9940b57cec5SDimitry Andric Code = bitc::TYPE_CODE_INTEGER; 9950b57cec5SDimitry Andric TypeVals.push_back(cast<IntegerType>(T)->getBitWidth()); 9960b57cec5SDimitry Andric break; 9970b57cec5SDimitry Andric case Type::PointerTyID: { 9980b57cec5SDimitry Andric PointerType *PTy = cast<PointerType>(T); 999fe6060f1SDimitry Andric unsigned AddressSpace = PTy->getAddressSpace(); 1000fe6060f1SDimitry Andric // OPAQUE_POINTER: [address space] 1001fe6060f1SDimitry Andric Code = bitc::TYPE_CODE_OPAQUE_POINTER; 1002fe6060f1SDimitry Andric TypeVals.push_back(AddressSpace); 1003fe6060f1SDimitry Andric if (AddressSpace == 0) 1004fe6060f1SDimitry Andric AbbrevToUse = OpaquePtrAbbrev; 10050b57cec5SDimitry Andric break; 10060b57cec5SDimitry Andric } 10070b57cec5SDimitry Andric case Type::FunctionTyID: { 10080b57cec5SDimitry Andric FunctionType *FT = cast<FunctionType>(T); 10090b57cec5SDimitry Andric // FUNCTION: [isvararg, retty, paramty x N] 10100b57cec5SDimitry Andric Code = bitc::TYPE_CODE_FUNCTION; 10110b57cec5SDimitry Andric TypeVals.push_back(FT->isVarArg()); 10120b57cec5SDimitry Andric TypeVals.push_back(VE.getTypeID(FT->getReturnType())); 10130b57cec5SDimitry Andric for (unsigned i = 0, e = FT->getNumParams(); i != e; ++i) 10140b57cec5SDimitry Andric TypeVals.push_back(VE.getTypeID(FT->getParamType(i))); 10150b57cec5SDimitry Andric AbbrevToUse = FunctionAbbrev; 10160b57cec5SDimitry Andric break; 10170b57cec5SDimitry Andric } 10180b57cec5SDimitry Andric case Type::StructTyID: { 10190b57cec5SDimitry Andric StructType *ST = cast<StructType>(T); 10200b57cec5SDimitry Andric // STRUCT: [ispacked, eltty x N] 10210b57cec5SDimitry Andric TypeVals.push_back(ST->isPacked()); 10220b57cec5SDimitry Andric // Output all of the element types. 1023349cc55cSDimitry Andric for (Type *ET : ST->elements()) 1024349cc55cSDimitry Andric TypeVals.push_back(VE.getTypeID(ET)); 10250b57cec5SDimitry Andric 10260b57cec5SDimitry Andric if (ST->isLiteral()) { 10270b57cec5SDimitry Andric Code = bitc::TYPE_CODE_STRUCT_ANON; 10280b57cec5SDimitry Andric AbbrevToUse = StructAnonAbbrev; 10290b57cec5SDimitry Andric } else { 10300b57cec5SDimitry Andric if (ST->isOpaque()) { 10310b57cec5SDimitry Andric Code = bitc::TYPE_CODE_OPAQUE; 10320b57cec5SDimitry Andric } else { 10330b57cec5SDimitry Andric Code = bitc::TYPE_CODE_STRUCT_NAMED; 10340b57cec5SDimitry Andric AbbrevToUse = StructNamedAbbrev; 10350b57cec5SDimitry Andric } 10360b57cec5SDimitry Andric 10370b57cec5SDimitry Andric // Emit the name if it is present. 10380b57cec5SDimitry Andric if (!ST->getName().empty()) 10390b57cec5SDimitry Andric writeStringRecord(Stream, bitc::TYPE_CODE_STRUCT_NAME, ST->getName(), 10400b57cec5SDimitry Andric StructNameAbbrev); 10410b57cec5SDimitry Andric } 10420b57cec5SDimitry Andric break; 10430b57cec5SDimitry Andric } 10440b57cec5SDimitry Andric case Type::ArrayTyID: { 10450b57cec5SDimitry Andric ArrayType *AT = cast<ArrayType>(T); 10460b57cec5SDimitry Andric // ARRAY: [numelts, eltty] 10470b57cec5SDimitry Andric Code = bitc::TYPE_CODE_ARRAY; 10480b57cec5SDimitry Andric TypeVals.push_back(AT->getNumElements()); 10490b57cec5SDimitry Andric TypeVals.push_back(VE.getTypeID(AT->getElementType())); 10500b57cec5SDimitry Andric AbbrevToUse = ArrayAbbrev; 10510b57cec5SDimitry Andric break; 10520b57cec5SDimitry Andric } 10535ffd83dbSDimitry Andric case Type::FixedVectorTyID: 10545ffd83dbSDimitry Andric case Type::ScalableVectorTyID: { 10550b57cec5SDimitry Andric VectorType *VT = cast<VectorType>(T); 10560b57cec5SDimitry Andric // VECTOR [numelts, eltty] or 10570b57cec5SDimitry Andric // [numelts, eltty, scalable] 10580b57cec5SDimitry Andric Code = bitc::TYPE_CODE_VECTOR; 1059e8d8bef9SDimitry Andric TypeVals.push_back(VT->getElementCount().getKnownMinValue()); 10600b57cec5SDimitry Andric TypeVals.push_back(VE.getTypeID(VT->getElementType())); 10615ffd83dbSDimitry Andric if (isa<ScalableVectorType>(VT)) 10625ffd83dbSDimitry Andric TypeVals.push_back(true); 10630b57cec5SDimitry Andric break; 10640b57cec5SDimitry Andric } 1065bdd1243dSDimitry Andric case Type::TargetExtTyID: { 1066bdd1243dSDimitry Andric TargetExtType *TET = cast<TargetExtType>(T); 1067bdd1243dSDimitry Andric Code = bitc::TYPE_CODE_TARGET_TYPE; 1068bdd1243dSDimitry Andric writeStringRecord(Stream, bitc::TYPE_CODE_STRUCT_NAME, TET->getName(), 1069bdd1243dSDimitry Andric StructNameAbbrev); 1070bdd1243dSDimitry Andric TypeVals.push_back(TET->getNumTypeParameters()); 1071bdd1243dSDimitry Andric for (Type *InnerTy : TET->type_params()) 1072bdd1243dSDimitry Andric TypeVals.push_back(VE.getTypeID(InnerTy)); 1073bdd1243dSDimitry Andric for (unsigned IntParam : TET->int_params()) 1074bdd1243dSDimitry Andric TypeVals.push_back(IntParam); 1075bdd1243dSDimitry Andric break; 1076bdd1243dSDimitry Andric } 1077bdd1243dSDimitry Andric case Type::TypedPointerTyID: 1078bdd1243dSDimitry Andric llvm_unreachable("Typed pointers cannot be added to IR modules"); 10790b57cec5SDimitry Andric } 10800b57cec5SDimitry Andric 10810b57cec5SDimitry Andric // Emit the finished record. 10820b57cec5SDimitry Andric Stream.EmitRecord(Code, TypeVals, AbbrevToUse); 10830b57cec5SDimitry Andric TypeVals.clear(); 10840b57cec5SDimitry Andric } 10850b57cec5SDimitry Andric 10860b57cec5SDimitry Andric Stream.ExitBlock(); 10870b57cec5SDimitry Andric } 10880b57cec5SDimitry Andric 10890b57cec5SDimitry Andric static unsigned getEncodedLinkage(const GlobalValue::LinkageTypes Linkage) { 10900b57cec5SDimitry Andric switch (Linkage) { 10910b57cec5SDimitry Andric case GlobalValue::ExternalLinkage: 10920b57cec5SDimitry Andric return 0; 10930b57cec5SDimitry Andric case GlobalValue::WeakAnyLinkage: 10940b57cec5SDimitry Andric return 16; 10950b57cec5SDimitry Andric case GlobalValue::AppendingLinkage: 10960b57cec5SDimitry Andric return 2; 10970b57cec5SDimitry Andric case GlobalValue::InternalLinkage: 10980b57cec5SDimitry Andric return 3; 10990b57cec5SDimitry Andric case GlobalValue::LinkOnceAnyLinkage: 11000b57cec5SDimitry Andric return 18; 11010b57cec5SDimitry Andric case GlobalValue::ExternalWeakLinkage: 11020b57cec5SDimitry Andric return 7; 11030b57cec5SDimitry Andric case GlobalValue::CommonLinkage: 11040b57cec5SDimitry Andric return 8; 11050b57cec5SDimitry Andric case GlobalValue::PrivateLinkage: 11060b57cec5SDimitry Andric return 9; 11070b57cec5SDimitry Andric case GlobalValue::WeakODRLinkage: 11080b57cec5SDimitry Andric return 17; 11090b57cec5SDimitry Andric case GlobalValue::LinkOnceODRLinkage: 11100b57cec5SDimitry Andric return 19; 11110b57cec5SDimitry Andric case GlobalValue::AvailableExternallyLinkage: 11120b57cec5SDimitry Andric return 12; 11130b57cec5SDimitry Andric } 11140b57cec5SDimitry Andric llvm_unreachable("Invalid linkage"); 11150b57cec5SDimitry Andric } 11160b57cec5SDimitry Andric 11170b57cec5SDimitry Andric static unsigned getEncodedLinkage(const GlobalValue &GV) { 11180b57cec5SDimitry Andric return getEncodedLinkage(GV.getLinkage()); 11190b57cec5SDimitry Andric } 11200b57cec5SDimitry Andric 11210b57cec5SDimitry Andric static uint64_t getEncodedFFlags(FunctionSummary::FFlags Flags) { 11220b57cec5SDimitry Andric uint64_t RawFlags = 0; 11230b57cec5SDimitry Andric RawFlags |= Flags.ReadNone; 11240b57cec5SDimitry Andric RawFlags |= (Flags.ReadOnly << 1); 11250b57cec5SDimitry Andric RawFlags |= (Flags.NoRecurse << 2); 11260b57cec5SDimitry Andric RawFlags |= (Flags.ReturnDoesNotAlias << 3); 11270b57cec5SDimitry Andric RawFlags |= (Flags.NoInline << 4); 1128480093f4SDimitry Andric RawFlags |= (Flags.AlwaysInline << 5); 1129349cc55cSDimitry Andric RawFlags |= (Flags.NoUnwind << 6); 1130349cc55cSDimitry Andric RawFlags |= (Flags.MayThrow << 7); 1131349cc55cSDimitry Andric RawFlags |= (Flags.HasUnknownCall << 8); 11320eae32dcSDimitry Andric RawFlags |= (Flags.MustBeUnreachable << 9); 11330b57cec5SDimitry Andric return RawFlags; 11340b57cec5SDimitry Andric } 11350b57cec5SDimitry Andric 1136fe6060f1SDimitry Andric // Decode the flags for GlobalValue in the summary. See getDecodedGVSummaryFlags 1137fe6060f1SDimitry Andric // in BitcodeReader.cpp. 11380b57cec5SDimitry Andric static uint64_t getEncodedGVSummaryFlags(GlobalValueSummary::GVFlags Flags) { 11390b57cec5SDimitry Andric uint64_t RawFlags = 0; 11400b57cec5SDimitry Andric 11410b57cec5SDimitry Andric RawFlags |= Flags.NotEligibleToImport; // bool 11420b57cec5SDimitry Andric RawFlags |= (Flags.Live << 1); 11430b57cec5SDimitry Andric RawFlags |= (Flags.DSOLocal << 2); 11440b57cec5SDimitry Andric RawFlags |= (Flags.CanAutoHide << 3); 11450b57cec5SDimitry Andric 11460b57cec5SDimitry Andric // Linkage don't need to be remapped at that time for the summary. Any future 11470b57cec5SDimitry Andric // change to the getEncodedLinkage() function will need to be taken into 11480b57cec5SDimitry Andric // account here as well. 11490b57cec5SDimitry Andric RawFlags = (RawFlags << 4) | Flags.Linkage; // 4 bits 11500b57cec5SDimitry Andric 1151fe6060f1SDimitry Andric RawFlags |= (Flags.Visibility << 8); // 2 bits 1152fe6060f1SDimitry Andric 11530b57cec5SDimitry Andric return RawFlags; 11540b57cec5SDimitry Andric } 11550b57cec5SDimitry Andric 11560b57cec5SDimitry Andric static uint64_t getEncodedGVarFlags(GlobalVarSummary::GVarFlags Flags) { 11575ffd83dbSDimitry Andric uint64_t RawFlags = Flags.MaybeReadOnly | (Flags.MaybeWriteOnly << 1) | 11585ffd83dbSDimitry Andric (Flags.Constant << 2) | Flags.VCallVisibility << 3; 11590b57cec5SDimitry Andric return RawFlags; 11600b57cec5SDimitry Andric } 11610b57cec5SDimitry Andric 1162*c9157d92SDimitry Andric static uint64_t getEncodedHotnessCallEdgeInfo(const CalleeInfo &CI) { 1163*c9157d92SDimitry Andric uint64_t RawFlags = 0; 1164*c9157d92SDimitry Andric 1165*c9157d92SDimitry Andric RawFlags |= CI.Hotness; // 3 bits 1166*c9157d92SDimitry Andric RawFlags |= (CI.HasTailCall << 3); // 1 bit 1167*c9157d92SDimitry Andric 1168*c9157d92SDimitry Andric return RawFlags; 1169*c9157d92SDimitry Andric } 1170*c9157d92SDimitry Andric 1171*c9157d92SDimitry Andric static uint64_t getEncodedRelBFCallEdgeInfo(const CalleeInfo &CI) { 1172*c9157d92SDimitry Andric uint64_t RawFlags = 0; 1173*c9157d92SDimitry Andric 1174*c9157d92SDimitry Andric RawFlags |= CI.RelBlockFreq; // CalleeInfo::RelBlockFreqBits bits 1175*c9157d92SDimitry Andric RawFlags |= (CI.HasTailCall << CalleeInfo::RelBlockFreqBits); // 1 bit 1176*c9157d92SDimitry Andric 1177*c9157d92SDimitry Andric return RawFlags; 1178*c9157d92SDimitry Andric } 1179*c9157d92SDimitry Andric 11800b57cec5SDimitry Andric static unsigned getEncodedVisibility(const GlobalValue &GV) { 11810b57cec5SDimitry Andric switch (GV.getVisibility()) { 11820b57cec5SDimitry Andric case GlobalValue::DefaultVisibility: return 0; 11830b57cec5SDimitry Andric case GlobalValue::HiddenVisibility: return 1; 11840b57cec5SDimitry Andric case GlobalValue::ProtectedVisibility: return 2; 11850b57cec5SDimitry Andric } 11860b57cec5SDimitry Andric llvm_unreachable("Invalid visibility"); 11870b57cec5SDimitry Andric } 11880b57cec5SDimitry Andric 11890b57cec5SDimitry Andric static unsigned getEncodedDLLStorageClass(const GlobalValue &GV) { 11900b57cec5SDimitry Andric switch (GV.getDLLStorageClass()) { 11910b57cec5SDimitry Andric case GlobalValue::DefaultStorageClass: return 0; 11920b57cec5SDimitry Andric case GlobalValue::DLLImportStorageClass: return 1; 11930b57cec5SDimitry Andric case GlobalValue::DLLExportStorageClass: return 2; 11940b57cec5SDimitry Andric } 11950b57cec5SDimitry Andric llvm_unreachable("Invalid DLL storage class"); 11960b57cec5SDimitry Andric } 11970b57cec5SDimitry Andric 11980b57cec5SDimitry Andric static unsigned getEncodedThreadLocalMode(const GlobalValue &GV) { 11990b57cec5SDimitry Andric switch (GV.getThreadLocalMode()) { 12000b57cec5SDimitry Andric case GlobalVariable::NotThreadLocal: return 0; 12010b57cec5SDimitry Andric case GlobalVariable::GeneralDynamicTLSModel: return 1; 12020b57cec5SDimitry Andric case GlobalVariable::LocalDynamicTLSModel: return 2; 12030b57cec5SDimitry Andric case GlobalVariable::InitialExecTLSModel: return 3; 12040b57cec5SDimitry Andric case GlobalVariable::LocalExecTLSModel: return 4; 12050b57cec5SDimitry Andric } 12060b57cec5SDimitry Andric llvm_unreachable("Invalid TLS model"); 12070b57cec5SDimitry Andric } 12080b57cec5SDimitry Andric 12090b57cec5SDimitry Andric static unsigned getEncodedComdatSelectionKind(const Comdat &C) { 12100b57cec5SDimitry Andric switch (C.getSelectionKind()) { 12110b57cec5SDimitry Andric case Comdat::Any: 12120b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_ANY; 12130b57cec5SDimitry Andric case Comdat::ExactMatch: 12140b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_EXACT_MATCH; 12150b57cec5SDimitry Andric case Comdat::Largest: 12160b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_LARGEST; 1217fe6060f1SDimitry Andric case Comdat::NoDeduplicate: 12180b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_NO_DUPLICATES; 12190b57cec5SDimitry Andric case Comdat::SameSize: 12200b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_SAME_SIZE; 12210b57cec5SDimitry Andric } 12220b57cec5SDimitry Andric llvm_unreachable("Invalid selection kind"); 12230b57cec5SDimitry Andric } 12240b57cec5SDimitry Andric 12250b57cec5SDimitry Andric static unsigned getEncodedUnnamedAddr(const GlobalValue &GV) { 12260b57cec5SDimitry Andric switch (GV.getUnnamedAddr()) { 12270b57cec5SDimitry Andric case GlobalValue::UnnamedAddr::None: return 0; 12280b57cec5SDimitry Andric case GlobalValue::UnnamedAddr::Local: return 2; 12290b57cec5SDimitry Andric case GlobalValue::UnnamedAddr::Global: return 1; 12300b57cec5SDimitry Andric } 12310b57cec5SDimitry Andric llvm_unreachable("Invalid unnamed_addr"); 12320b57cec5SDimitry Andric } 12330b57cec5SDimitry Andric 12340b57cec5SDimitry Andric size_t ModuleBitcodeWriter::addToStrtab(StringRef Str) { 12350b57cec5SDimitry Andric if (GenerateHash) 12360b57cec5SDimitry Andric Hasher.update(Str); 12370b57cec5SDimitry Andric return StrtabBuilder.add(Str); 12380b57cec5SDimitry Andric } 12390b57cec5SDimitry Andric 12400b57cec5SDimitry Andric void ModuleBitcodeWriter::writeComdats() { 12410b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 12420b57cec5SDimitry Andric for (const Comdat *C : VE.getComdats()) { 12430b57cec5SDimitry Andric // COMDAT: [strtab offset, strtab size, selection_kind] 12440b57cec5SDimitry Andric Vals.push_back(addToStrtab(C->getName())); 12450b57cec5SDimitry Andric Vals.push_back(C->getName().size()); 12460b57cec5SDimitry Andric Vals.push_back(getEncodedComdatSelectionKind(*C)); 12470b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_COMDAT, Vals, /*AbbrevToUse=*/0); 12480b57cec5SDimitry Andric Vals.clear(); 12490b57cec5SDimitry Andric } 12500b57cec5SDimitry Andric } 12510b57cec5SDimitry Andric 12520b57cec5SDimitry Andric /// Write a record that will eventually hold the word offset of the 12530b57cec5SDimitry Andric /// module-level VST. For now the offset is 0, which will be backpatched 12540b57cec5SDimitry Andric /// after the real VST is written. Saves the bit offset to backpatch. 12550b57cec5SDimitry Andric void ModuleBitcodeWriter::writeValueSymbolTableForwardDecl() { 12560b57cec5SDimitry Andric // Write a placeholder value in for the offset of the real VST, 12570b57cec5SDimitry Andric // which is written after the function blocks so that it can include 12580b57cec5SDimitry Andric // the offset of each function. The placeholder offset will be 12590b57cec5SDimitry Andric // updated when the real VST is written. 12600b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 12610b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_VSTOFFSET)); 12620b57cec5SDimitry Andric // Blocks are 32-bit aligned, so we can use a 32-bit word offset to 12630b57cec5SDimitry Andric // hold the real VST offset. Must use fixed instead of VBR as we don't 12640b57cec5SDimitry Andric // know how many VBR chunks to reserve ahead of time. 12650b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 12660b57cec5SDimitry Andric unsigned VSTOffsetAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 12670b57cec5SDimitry Andric 12680b57cec5SDimitry Andric // Emit the placeholder 12690b57cec5SDimitry Andric uint64_t Vals[] = {bitc::MODULE_CODE_VSTOFFSET, 0}; 12700b57cec5SDimitry Andric Stream.EmitRecordWithAbbrev(VSTOffsetAbbrev, Vals); 12710b57cec5SDimitry Andric 12720b57cec5SDimitry Andric // Compute and save the bit offset to the placeholder, which will be 12730b57cec5SDimitry Andric // patched when the real VST is written. We can simply subtract the 32-bit 12740b57cec5SDimitry Andric // fixed size from the current bit number to get the location to backpatch. 12750b57cec5SDimitry Andric VSTOffsetPlaceholder = Stream.GetCurrentBitNo() - 32; 12760b57cec5SDimitry Andric } 12770b57cec5SDimitry Andric 12780b57cec5SDimitry Andric enum StringEncoding { SE_Char6, SE_Fixed7, SE_Fixed8 }; 12790b57cec5SDimitry Andric 12800b57cec5SDimitry Andric /// Determine the encoding to use for the given string name and length. 12810b57cec5SDimitry Andric static StringEncoding getStringEncoding(StringRef Str) { 12820b57cec5SDimitry Andric bool isChar6 = true; 12830b57cec5SDimitry Andric for (char C : Str) { 12840b57cec5SDimitry Andric if (isChar6) 12850b57cec5SDimitry Andric isChar6 = BitCodeAbbrevOp::isChar6(C); 12860b57cec5SDimitry Andric if ((unsigned char)C & 128) 12870b57cec5SDimitry Andric // don't bother scanning the rest. 12880b57cec5SDimitry Andric return SE_Fixed8; 12890b57cec5SDimitry Andric } 12900b57cec5SDimitry Andric if (isChar6) 12910b57cec5SDimitry Andric return SE_Char6; 12920b57cec5SDimitry Andric return SE_Fixed7; 12930b57cec5SDimitry Andric } 12940b57cec5SDimitry Andric 129581ad6265SDimitry Andric static_assert(sizeof(GlobalValue::SanitizerMetadata) <= sizeof(unsigned), 129681ad6265SDimitry Andric "Sanitizer Metadata is too large for naive serialization."); 129781ad6265SDimitry Andric static unsigned 129881ad6265SDimitry Andric serializeSanitizerMetadata(const GlobalValue::SanitizerMetadata &Meta) { 129981ad6265SDimitry Andric return Meta.NoAddress | (Meta.NoHWAddress << 1) | 1300753f127fSDimitry Andric (Meta.Memtag << 2) | (Meta.IsDynInit << 3); 130181ad6265SDimitry Andric } 130281ad6265SDimitry Andric 13030b57cec5SDimitry Andric /// Emit top-level description of module, including target triple, inline asm, 13040b57cec5SDimitry Andric /// descriptors for global variables, and function prototype info. 13050b57cec5SDimitry Andric /// Returns the bit offset to backpatch with the location of the real VST. 13060b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleInfo() { 13070b57cec5SDimitry Andric // Emit various pieces of data attached to a module. 13080b57cec5SDimitry Andric if (!M.getTargetTriple().empty()) 13090b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_TRIPLE, M.getTargetTriple(), 13100b57cec5SDimitry Andric 0 /*TODO*/); 13110b57cec5SDimitry Andric const std::string &DL = M.getDataLayoutStr(); 13120b57cec5SDimitry Andric if (!DL.empty()) 13130b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_DATALAYOUT, DL, 0 /*TODO*/); 13140b57cec5SDimitry Andric if (!M.getModuleInlineAsm().empty()) 13150b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_ASM, M.getModuleInlineAsm(), 13160b57cec5SDimitry Andric 0 /*TODO*/); 13170b57cec5SDimitry Andric 13180b57cec5SDimitry Andric // Emit information about sections and GC, computing how many there are. Also 13190b57cec5SDimitry Andric // compute the maximum alignment value. 13200b57cec5SDimitry Andric std::map<std::string, unsigned> SectionMap; 13210b57cec5SDimitry Andric std::map<std::string, unsigned> GCMap; 1322e8d8bef9SDimitry Andric MaybeAlign MaxAlignment; 13230b57cec5SDimitry Andric unsigned MaxGlobalType = 0; 1324e8d8bef9SDimitry Andric const auto UpdateMaxAlignment = [&MaxAlignment](const MaybeAlign A) { 1325e8d8bef9SDimitry Andric if (A) 1326e8d8bef9SDimitry Andric MaxAlignment = !MaxAlignment ? *A : std::max(*MaxAlignment, *A); 1327e8d8bef9SDimitry Andric }; 13285ffd83dbSDimitry Andric for (const GlobalVariable &GV : M.globals()) { 1329e8d8bef9SDimitry Andric UpdateMaxAlignment(GV.getAlign()); 13300b57cec5SDimitry Andric MaxGlobalType = std::max(MaxGlobalType, VE.getTypeID(GV.getValueType())); 13310b57cec5SDimitry Andric if (GV.hasSection()) { 13320b57cec5SDimitry Andric // Give section names unique ID's. 13335ffd83dbSDimitry Andric unsigned &Entry = SectionMap[std::string(GV.getSection())]; 13340b57cec5SDimitry Andric if (!Entry) { 13350b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_SECTIONNAME, GV.getSection(), 13360b57cec5SDimitry Andric 0 /*TODO*/); 13370b57cec5SDimitry Andric Entry = SectionMap.size(); 13380b57cec5SDimitry Andric } 13390b57cec5SDimitry Andric } 13400b57cec5SDimitry Andric } 13410b57cec5SDimitry Andric for (const Function &F : M) { 1342e8d8bef9SDimitry Andric UpdateMaxAlignment(F.getAlign()); 13430b57cec5SDimitry Andric if (F.hasSection()) { 13440b57cec5SDimitry Andric // Give section names unique ID's. 13455ffd83dbSDimitry Andric unsigned &Entry = SectionMap[std::string(F.getSection())]; 13460b57cec5SDimitry Andric if (!Entry) { 13470b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_SECTIONNAME, F.getSection(), 13480b57cec5SDimitry Andric 0 /*TODO*/); 13490b57cec5SDimitry Andric Entry = SectionMap.size(); 13500b57cec5SDimitry Andric } 13510b57cec5SDimitry Andric } 13520b57cec5SDimitry Andric if (F.hasGC()) { 13530b57cec5SDimitry Andric // Same for GC names. 13540b57cec5SDimitry Andric unsigned &Entry = GCMap[F.getGC()]; 13550b57cec5SDimitry Andric if (!Entry) { 13560b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_GCNAME, F.getGC(), 13570b57cec5SDimitry Andric 0 /*TODO*/); 13580b57cec5SDimitry Andric Entry = GCMap.size(); 13590b57cec5SDimitry Andric } 13600b57cec5SDimitry Andric } 13610b57cec5SDimitry Andric } 13620b57cec5SDimitry Andric 13630b57cec5SDimitry Andric // Emit abbrev for globals, now that we know # sections and max alignment. 13640b57cec5SDimitry Andric unsigned SimpleGVarAbbrev = 0; 13650b57cec5SDimitry Andric if (!M.global_empty()) { 13660b57cec5SDimitry Andric // Add an abbrev for common globals with no visibility or thread localness. 13670b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 13680b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_GLOBALVAR)); 13690b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 13700b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 13710b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 13720b57cec5SDimitry Andric Log2_32_Ceil(MaxGlobalType+1))); 13730b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // AddrSpace << 2 13740b57cec5SDimitry Andric //| explicitType << 1 13750b57cec5SDimitry Andric //| constant 13760b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // Initializer. 13770b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 5)); // Linkage. 1378e8d8bef9SDimitry Andric if (!MaxAlignment) // Alignment. 13790b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(0)); 13800b57cec5SDimitry Andric else { 1381e8d8bef9SDimitry Andric unsigned MaxEncAlignment = getEncodedAlign(MaxAlignment); 13820b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 13830b57cec5SDimitry Andric Log2_32_Ceil(MaxEncAlignment+1))); 13840b57cec5SDimitry Andric } 13850b57cec5SDimitry Andric if (SectionMap.empty()) // Section. 13860b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(0)); 13870b57cec5SDimitry Andric else 13880b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 13890b57cec5SDimitry Andric Log2_32_Ceil(SectionMap.size()+1))); 13900b57cec5SDimitry Andric // Don't bother emitting vis + thread local. 13910b57cec5SDimitry Andric SimpleGVarAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 13920b57cec5SDimitry Andric } 13930b57cec5SDimitry Andric 13940b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 13950b57cec5SDimitry Andric // Emit the module's source file name. 13960b57cec5SDimitry Andric { 13970b57cec5SDimitry Andric StringEncoding Bits = getStringEncoding(M.getSourceFileName()); 13980b57cec5SDimitry Andric BitCodeAbbrevOp AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8); 13990b57cec5SDimitry Andric if (Bits == SE_Char6) 14000b57cec5SDimitry Andric AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Char6); 14010b57cec5SDimitry Andric else if (Bits == SE_Fixed7) 14020b57cec5SDimitry Andric AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7); 14030b57cec5SDimitry Andric 14040b57cec5SDimitry Andric // MODULE_CODE_SOURCE_FILENAME: [namechar x N] 14050b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 14060b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_SOURCE_FILENAME)); 14070b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 14080b57cec5SDimitry Andric Abbv->Add(AbbrevOpToUse); 14090b57cec5SDimitry Andric unsigned FilenameAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 14100b57cec5SDimitry Andric 14110b57cec5SDimitry Andric for (const auto P : M.getSourceFileName()) 14120b57cec5SDimitry Andric Vals.push_back((unsigned char)P); 14130b57cec5SDimitry Andric 14140b57cec5SDimitry Andric // Emit the finished record. 14150b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_SOURCE_FILENAME, Vals, FilenameAbbrev); 14160b57cec5SDimitry Andric Vals.clear(); 14170b57cec5SDimitry Andric } 14180b57cec5SDimitry Andric 14190b57cec5SDimitry Andric // Emit the global variable information. 14200b57cec5SDimitry Andric for (const GlobalVariable &GV : M.globals()) { 14210b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 14220b57cec5SDimitry Andric 14230b57cec5SDimitry Andric // GLOBALVAR: [strtab offset, strtab size, type, isconst, initid, 14240b57cec5SDimitry Andric // linkage, alignment, section, visibility, threadlocal, 14250b57cec5SDimitry Andric // unnamed_addr, externally_initialized, dllstorageclass, 1426*c9157d92SDimitry Andric // comdat, attributes, DSO_Local, GlobalSanitizer, code_model] 14270b57cec5SDimitry Andric Vals.push_back(addToStrtab(GV.getName())); 14280b57cec5SDimitry Andric Vals.push_back(GV.getName().size()); 14290b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(GV.getValueType())); 14300b57cec5SDimitry Andric Vals.push_back(GV.getType()->getAddressSpace() << 2 | 2 | GV.isConstant()); 14310b57cec5SDimitry Andric Vals.push_back(GV.isDeclaration() ? 0 : 14320b57cec5SDimitry Andric (VE.getValueID(GV.getInitializer()) + 1)); 14330b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(GV)); 1434e8d8bef9SDimitry Andric Vals.push_back(getEncodedAlign(GV.getAlign())); 14355ffd83dbSDimitry Andric Vals.push_back(GV.hasSection() ? SectionMap[std::string(GV.getSection())] 14365ffd83dbSDimitry Andric : 0); 14370b57cec5SDimitry Andric if (GV.isThreadLocal() || 14380b57cec5SDimitry Andric GV.getVisibility() != GlobalValue::DefaultVisibility || 14390b57cec5SDimitry Andric GV.getUnnamedAddr() != GlobalValue::UnnamedAddr::None || 14400b57cec5SDimitry Andric GV.isExternallyInitialized() || 14410b57cec5SDimitry Andric GV.getDLLStorageClass() != GlobalValue::DefaultStorageClass || 144281ad6265SDimitry Andric GV.hasComdat() || GV.hasAttributes() || GV.isDSOLocal() || 1443*c9157d92SDimitry Andric GV.hasPartition() || GV.hasSanitizerMetadata() || GV.getCodeModel()) { 14440b57cec5SDimitry Andric Vals.push_back(getEncodedVisibility(GV)); 14450b57cec5SDimitry Andric Vals.push_back(getEncodedThreadLocalMode(GV)); 14460b57cec5SDimitry Andric Vals.push_back(getEncodedUnnamedAddr(GV)); 14470b57cec5SDimitry Andric Vals.push_back(GV.isExternallyInitialized()); 14480b57cec5SDimitry Andric Vals.push_back(getEncodedDLLStorageClass(GV)); 14490b57cec5SDimitry Andric Vals.push_back(GV.hasComdat() ? VE.getComdatID(GV.getComdat()) : 0); 14500b57cec5SDimitry Andric 14510b57cec5SDimitry Andric auto AL = GV.getAttributesAsList(AttributeList::FunctionIndex); 14520b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(AL)); 14530b57cec5SDimitry Andric 14540b57cec5SDimitry Andric Vals.push_back(GV.isDSOLocal()); 14550b57cec5SDimitry Andric Vals.push_back(addToStrtab(GV.getPartition())); 14560b57cec5SDimitry Andric Vals.push_back(GV.getPartition().size()); 145781ad6265SDimitry Andric 145881ad6265SDimitry Andric Vals.push_back((GV.hasSanitizerMetadata() ? serializeSanitizerMetadata( 145981ad6265SDimitry Andric GV.getSanitizerMetadata()) 146081ad6265SDimitry Andric : 0)); 1461*c9157d92SDimitry Andric Vals.push_back(GV.getCodeModelRaw()); 14620b57cec5SDimitry Andric } else { 14630b57cec5SDimitry Andric AbbrevToUse = SimpleGVarAbbrev; 14640b57cec5SDimitry Andric } 14650b57cec5SDimitry Andric 14660b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_GLOBALVAR, Vals, AbbrevToUse); 14670b57cec5SDimitry Andric Vals.clear(); 14680b57cec5SDimitry Andric } 14690b57cec5SDimitry Andric 14700b57cec5SDimitry Andric // Emit the function proto information. 14710b57cec5SDimitry Andric for (const Function &F : M) { 14720b57cec5SDimitry Andric // FUNCTION: [strtab offset, strtab size, type, callingconv, isproto, 14730b57cec5SDimitry Andric // linkage, paramattrs, alignment, section, visibility, gc, 14740b57cec5SDimitry Andric // unnamed_addr, prologuedata, dllstorageclass, comdat, 14750b57cec5SDimitry Andric // prefixdata, personalityfn, DSO_Local, addrspace] 14760b57cec5SDimitry Andric Vals.push_back(addToStrtab(F.getName())); 14770b57cec5SDimitry Andric Vals.push_back(F.getName().size()); 14780b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(F.getFunctionType())); 14790b57cec5SDimitry Andric Vals.push_back(F.getCallingConv()); 14800b57cec5SDimitry Andric Vals.push_back(F.isDeclaration()); 14810b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(F)); 14820b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(F.getAttributes())); 1483e8d8bef9SDimitry Andric Vals.push_back(getEncodedAlign(F.getAlign())); 14845ffd83dbSDimitry Andric Vals.push_back(F.hasSection() ? SectionMap[std::string(F.getSection())] 14855ffd83dbSDimitry Andric : 0); 14860b57cec5SDimitry Andric Vals.push_back(getEncodedVisibility(F)); 14870b57cec5SDimitry Andric Vals.push_back(F.hasGC() ? GCMap[F.getGC()] : 0); 14880b57cec5SDimitry Andric Vals.push_back(getEncodedUnnamedAddr(F)); 14890b57cec5SDimitry Andric Vals.push_back(F.hasPrologueData() ? (VE.getValueID(F.getPrologueData()) + 1) 14900b57cec5SDimitry Andric : 0); 14910b57cec5SDimitry Andric Vals.push_back(getEncodedDLLStorageClass(F)); 14920b57cec5SDimitry Andric Vals.push_back(F.hasComdat() ? VE.getComdatID(F.getComdat()) : 0); 14930b57cec5SDimitry Andric Vals.push_back(F.hasPrefixData() ? (VE.getValueID(F.getPrefixData()) + 1) 14940b57cec5SDimitry Andric : 0); 14950b57cec5SDimitry Andric Vals.push_back( 14960b57cec5SDimitry Andric F.hasPersonalityFn() ? (VE.getValueID(F.getPersonalityFn()) + 1) : 0); 14970b57cec5SDimitry Andric 14980b57cec5SDimitry Andric Vals.push_back(F.isDSOLocal()); 14990b57cec5SDimitry Andric Vals.push_back(F.getAddressSpace()); 15000b57cec5SDimitry Andric Vals.push_back(addToStrtab(F.getPartition())); 15010b57cec5SDimitry Andric Vals.push_back(F.getPartition().size()); 15020b57cec5SDimitry Andric 15030b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 15040b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_FUNCTION, Vals, AbbrevToUse); 15050b57cec5SDimitry Andric Vals.clear(); 15060b57cec5SDimitry Andric } 15070b57cec5SDimitry Andric 15080b57cec5SDimitry Andric // Emit the alias information. 15090b57cec5SDimitry Andric for (const GlobalAlias &A : M.aliases()) { 15100b57cec5SDimitry Andric // ALIAS: [strtab offset, strtab size, alias type, aliasee val#, linkage, 15110b57cec5SDimitry Andric // visibility, dllstorageclass, threadlocal, unnamed_addr, 15120b57cec5SDimitry Andric // DSO_Local] 15130b57cec5SDimitry Andric Vals.push_back(addToStrtab(A.getName())); 15140b57cec5SDimitry Andric Vals.push_back(A.getName().size()); 15150b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(A.getValueType())); 15160b57cec5SDimitry Andric Vals.push_back(A.getType()->getAddressSpace()); 15170b57cec5SDimitry Andric Vals.push_back(VE.getValueID(A.getAliasee())); 15180b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(A)); 15190b57cec5SDimitry Andric Vals.push_back(getEncodedVisibility(A)); 15200b57cec5SDimitry Andric Vals.push_back(getEncodedDLLStorageClass(A)); 15210b57cec5SDimitry Andric Vals.push_back(getEncodedThreadLocalMode(A)); 15220b57cec5SDimitry Andric Vals.push_back(getEncodedUnnamedAddr(A)); 15230b57cec5SDimitry Andric Vals.push_back(A.isDSOLocal()); 15240b57cec5SDimitry Andric Vals.push_back(addToStrtab(A.getPartition())); 15250b57cec5SDimitry Andric Vals.push_back(A.getPartition().size()); 15260b57cec5SDimitry Andric 15270b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 15280b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_ALIAS, Vals, AbbrevToUse); 15290b57cec5SDimitry Andric Vals.clear(); 15300b57cec5SDimitry Andric } 15310b57cec5SDimitry Andric 15320b57cec5SDimitry Andric // Emit the ifunc information. 15330b57cec5SDimitry Andric for (const GlobalIFunc &I : M.ifuncs()) { 15340b57cec5SDimitry Andric // IFUNC: [strtab offset, strtab size, ifunc type, address space, resolver 15350b57cec5SDimitry Andric // val#, linkage, visibility, DSO_Local] 15360b57cec5SDimitry Andric Vals.push_back(addToStrtab(I.getName())); 15370b57cec5SDimitry Andric Vals.push_back(I.getName().size()); 15380b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getValueType())); 15390b57cec5SDimitry Andric Vals.push_back(I.getType()->getAddressSpace()); 15400b57cec5SDimitry Andric Vals.push_back(VE.getValueID(I.getResolver())); 15410b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(I)); 15420b57cec5SDimitry Andric Vals.push_back(getEncodedVisibility(I)); 15430b57cec5SDimitry Andric Vals.push_back(I.isDSOLocal()); 15440b57cec5SDimitry Andric Vals.push_back(addToStrtab(I.getPartition())); 15450b57cec5SDimitry Andric Vals.push_back(I.getPartition().size()); 15460b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_IFUNC, Vals); 15470b57cec5SDimitry Andric Vals.clear(); 15480b57cec5SDimitry Andric } 15490b57cec5SDimitry Andric 15500b57cec5SDimitry Andric writeValueSymbolTableForwardDecl(); 15510b57cec5SDimitry Andric } 15520b57cec5SDimitry Andric 15530b57cec5SDimitry Andric static uint64_t getOptimizationFlags(const Value *V) { 15540b57cec5SDimitry Andric uint64_t Flags = 0; 15550b57cec5SDimitry Andric 15560b57cec5SDimitry Andric if (const auto *OBO = dyn_cast<OverflowingBinaryOperator>(V)) { 15570b57cec5SDimitry Andric if (OBO->hasNoSignedWrap()) 15580b57cec5SDimitry Andric Flags |= 1 << bitc::OBO_NO_SIGNED_WRAP; 15590b57cec5SDimitry Andric if (OBO->hasNoUnsignedWrap()) 15600b57cec5SDimitry Andric Flags |= 1 << bitc::OBO_NO_UNSIGNED_WRAP; 15610b57cec5SDimitry Andric } else if (const auto *PEO = dyn_cast<PossiblyExactOperator>(V)) { 15620b57cec5SDimitry Andric if (PEO->isExact()) 15630b57cec5SDimitry Andric Flags |= 1 << bitc::PEO_EXACT; 1564*c9157d92SDimitry Andric } else if (const auto *PDI = dyn_cast<PossiblyDisjointInst>(V)) { 1565*c9157d92SDimitry Andric if (PDI->isDisjoint()) 1566*c9157d92SDimitry Andric Flags |= 1 << bitc::PDI_DISJOINT; 15670b57cec5SDimitry Andric } else if (const auto *FPMO = dyn_cast<FPMathOperator>(V)) { 15680b57cec5SDimitry Andric if (FPMO->hasAllowReassoc()) 15690b57cec5SDimitry Andric Flags |= bitc::AllowReassoc; 15700b57cec5SDimitry Andric if (FPMO->hasNoNaNs()) 15710b57cec5SDimitry Andric Flags |= bitc::NoNaNs; 15720b57cec5SDimitry Andric if (FPMO->hasNoInfs()) 15730b57cec5SDimitry Andric Flags |= bitc::NoInfs; 15740b57cec5SDimitry Andric if (FPMO->hasNoSignedZeros()) 15750b57cec5SDimitry Andric Flags |= bitc::NoSignedZeros; 15760b57cec5SDimitry Andric if (FPMO->hasAllowReciprocal()) 15770b57cec5SDimitry Andric Flags |= bitc::AllowReciprocal; 15780b57cec5SDimitry Andric if (FPMO->hasAllowContract()) 15790b57cec5SDimitry Andric Flags |= bitc::AllowContract; 15800b57cec5SDimitry Andric if (FPMO->hasApproxFunc()) 15810b57cec5SDimitry Andric Flags |= bitc::ApproxFunc; 1582*c9157d92SDimitry Andric } else if (const auto *NNI = dyn_cast<PossiblyNonNegInst>(V)) { 1583*c9157d92SDimitry Andric if (NNI->hasNonNeg()) 1584*c9157d92SDimitry Andric Flags |= 1 << bitc::PNNI_NON_NEG; 15850b57cec5SDimitry Andric } 15860b57cec5SDimitry Andric 15870b57cec5SDimitry Andric return Flags; 15880b57cec5SDimitry Andric } 15890b57cec5SDimitry Andric 15900b57cec5SDimitry Andric void ModuleBitcodeWriter::writeValueAsMetadata( 15910b57cec5SDimitry Andric const ValueAsMetadata *MD, SmallVectorImpl<uint64_t> &Record) { 15920b57cec5SDimitry Andric // Mimic an MDNode with a value as one operand. 15930b57cec5SDimitry Andric Value *V = MD->getValue(); 15940b57cec5SDimitry Andric Record.push_back(VE.getTypeID(V->getType())); 15950b57cec5SDimitry Andric Record.push_back(VE.getValueID(V)); 15960b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_VALUE, Record, 0); 15970b57cec5SDimitry Andric Record.clear(); 15980b57cec5SDimitry Andric } 15990b57cec5SDimitry Andric 16000b57cec5SDimitry Andric void ModuleBitcodeWriter::writeMDTuple(const MDTuple *N, 16010b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 16020b57cec5SDimitry Andric unsigned Abbrev) { 16030b57cec5SDimitry Andric for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) { 16040b57cec5SDimitry Andric Metadata *MD = N->getOperand(i); 16050b57cec5SDimitry Andric assert(!(MD && isa<LocalAsMetadata>(MD)) && 16060b57cec5SDimitry Andric "Unexpected function-local metadata"); 16070b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(MD)); 16080b57cec5SDimitry Andric } 16090b57cec5SDimitry Andric Stream.EmitRecord(N->isDistinct() ? bitc::METADATA_DISTINCT_NODE 16100b57cec5SDimitry Andric : bitc::METADATA_NODE, 16110b57cec5SDimitry Andric Record, Abbrev); 16120b57cec5SDimitry Andric Record.clear(); 16130b57cec5SDimitry Andric } 16140b57cec5SDimitry Andric 16150b57cec5SDimitry Andric unsigned ModuleBitcodeWriter::createDILocationAbbrev() { 16160b57cec5SDimitry Andric // Assume the column is usually under 128, and always output the inlined-at 16170b57cec5SDimitry Andric // location (it's never more expensive than building an array size 1). 16180b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 16190b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_LOCATION)); 16200b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 16210b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 16220b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 16230b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 16240b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 16250b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 16260b57cec5SDimitry Andric return Stream.EmitAbbrev(std::move(Abbv)); 16270b57cec5SDimitry Andric } 16280b57cec5SDimitry Andric 16290b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILocation(const DILocation *N, 16300b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 16310b57cec5SDimitry Andric unsigned &Abbrev) { 16320b57cec5SDimitry Andric if (!Abbrev) 16330b57cec5SDimitry Andric Abbrev = createDILocationAbbrev(); 16340b57cec5SDimitry Andric 16350b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 16360b57cec5SDimitry Andric Record.push_back(N->getLine()); 16370b57cec5SDimitry Andric Record.push_back(N->getColumn()); 16380b57cec5SDimitry Andric Record.push_back(VE.getMetadataID(N->getScope())); 16390b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getInlinedAt())); 16400b57cec5SDimitry Andric Record.push_back(N->isImplicitCode()); 16410b57cec5SDimitry Andric 16420b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LOCATION, Record, Abbrev); 16430b57cec5SDimitry Andric Record.clear(); 16440b57cec5SDimitry Andric } 16450b57cec5SDimitry Andric 16460b57cec5SDimitry Andric unsigned ModuleBitcodeWriter::createGenericDINodeAbbrev() { 16470b57cec5SDimitry Andric // Assume the column is usually under 128, and always output the inlined-at 16480b57cec5SDimitry Andric // location (it's never more expensive than building an array size 1). 16490b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 16500b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_GENERIC_DEBUG)); 16510b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 16520b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 16530b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 16540b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 16550b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 16560b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 16570b57cec5SDimitry Andric return Stream.EmitAbbrev(std::move(Abbv)); 16580b57cec5SDimitry Andric } 16590b57cec5SDimitry Andric 16600b57cec5SDimitry Andric void ModuleBitcodeWriter::writeGenericDINode(const GenericDINode *N, 16610b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 16620b57cec5SDimitry Andric unsigned &Abbrev) { 16630b57cec5SDimitry Andric if (!Abbrev) 16640b57cec5SDimitry Andric Abbrev = createGenericDINodeAbbrev(); 16650b57cec5SDimitry Andric 16660b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 16670b57cec5SDimitry Andric Record.push_back(N->getTag()); 16680b57cec5SDimitry Andric Record.push_back(0); // Per-tag version field; unused for now. 16690b57cec5SDimitry Andric 16700b57cec5SDimitry Andric for (auto &I : N->operands()) 16710b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(I)); 16720b57cec5SDimitry Andric 16730b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_GENERIC_DEBUG, Record, Abbrev); 16740b57cec5SDimitry Andric Record.clear(); 16750b57cec5SDimitry Andric } 16760b57cec5SDimitry Andric 16770b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDISubrange(const DISubrange *N, 16780b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 16790b57cec5SDimitry Andric unsigned Abbrev) { 16805ffd83dbSDimitry Andric const uint64_t Version = 2 << 1; 16810b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct() | Version); 16820b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawCountNode())); 16835ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawLowerBound())); 16845ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawUpperBound())); 16855ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawStride())); 16860b57cec5SDimitry Andric 16870b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_SUBRANGE, Record, Abbrev); 16880b57cec5SDimitry Andric Record.clear(); 16890b57cec5SDimitry Andric } 16900b57cec5SDimitry Andric 1691e8d8bef9SDimitry Andric void ModuleBitcodeWriter::writeDIGenericSubrange( 1692e8d8bef9SDimitry Andric const DIGenericSubrange *N, SmallVectorImpl<uint64_t> &Record, 1693e8d8bef9SDimitry Andric unsigned Abbrev) { 1694e8d8bef9SDimitry Andric Record.push_back((uint64_t)N->isDistinct()); 1695e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawCountNode())); 1696e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawLowerBound())); 1697e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawUpperBound())); 1698e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawStride())); 1699e8d8bef9SDimitry Andric 1700e8d8bef9SDimitry Andric Stream.EmitRecord(bitc::METADATA_GENERIC_SUBRANGE, Record, Abbrev); 1701e8d8bef9SDimitry Andric Record.clear(); 1702e8d8bef9SDimitry Andric } 1703e8d8bef9SDimitry Andric 17045ffd83dbSDimitry Andric static void emitSignedInt64(SmallVectorImpl<uint64_t> &Vals, uint64_t V) { 17055ffd83dbSDimitry Andric if ((int64_t)V >= 0) 17065ffd83dbSDimitry Andric Vals.push_back(V << 1); 17075ffd83dbSDimitry Andric else 17085ffd83dbSDimitry Andric Vals.push_back((-V << 1) | 1); 17095ffd83dbSDimitry Andric } 17105ffd83dbSDimitry Andric 17115ffd83dbSDimitry Andric static void emitWideAPInt(SmallVectorImpl<uint64_t> &Vals, const APInt &A) { 17125ffd83dbSDimitry Andric // We have an arbitrary precision integer value to write whose 17135ffd83dbSDimitry Andric // bit width is > 64. However, in canonical unsigned integer 17145ffd83dbSDimitry Andric // format it is likely that the high bits are going to be zero. 17155ffd83dbSDimitry Andric // So, we only write the number of active words. 17165ffd83dbSDimitry Andric unsigned NumWords = A.getActiveWords(); 17175ffd83dbSDimitry Andric const uint64_t *RawData = A.getRawData(); 17185ffd83dbSDimitry Andric for (unsigned i = 0; i < NumWords; i++) 17195ffd83dbSDimitry Andric emitSignedInt64(Vals, RawData[i]); 17205ffd83dbSDimitry Andric } 17215ffd83dbSDimitry Andric 17220b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIEnumerator(const DIEnumerator *N, 17230b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 17240b57cec5SDimitry Andric unsigned Abbrev) { 17255ffd83dbSDimitry Andric const uint64_t IsBigInt = 1 << 2; 17265ffd83dbSDimitry Andric Record.push_back(IsBigInt | (N->isUnsigned() << 1) | N->isDistinct()); 17275ffd83dbSDimitry Andric Record.push_back(N->getValue().getBitWidth()); 17280b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 17295ffd83dbSDimitry Andric emitWideAPInt(Record, N->getValue()); 17300b57cec5SDimitry Andric 17310b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_ENUMERATOR, Record, Abbrev); 17320b57cec5SDimitry Andric Record.clear(); 17330b57cec5SDimitry Andric } 17340b57cec5SDimitry Andric 17350b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIBasicType(const DIBasicType *N, 17360b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 17370b57cec5SDimitry Andric unsigned Abbrev) { 17380b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 17390b57cec5SDimitry Andric Record.push_back(N->getTag()); 17400b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 17410b57cec5SDimitry Andric Record.push_back(N->getSizeInBits()); 17420b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 17430b57cec5SDimitry Andric Record.push_back(N->getEncoding()); 17440b57cec5SDimitry Andric Record.push_back(N->getFlags()); 17450b57cec5SDimitry Andric 17460b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_BASIC_TYPE, Record, Abbrev); 17470b57cec5SDimitry Andric Record.clear(); 17480b57cec5SDimitry Andric } 17490b57cec5SDimitry Andric 1750e8d8bef9SDimitry Andric void ModuleBitcodeWriter::writeDIStringType(const DIStringType *N, 1751e8d8bef9SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1752e8d8bef9SDimitry Andric unsigned Abbrev) { 1753e8d8bef9SDimitry Andric Record.push_back(N->isDistinct()); 1754e8d8bef9SDimitry Andric Record.push_back(N->getTag()); 1755e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 1756e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getStringLength())); 1757e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getStringLengthExp())); 175804eeddc0SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getStringLocationExp())); 1759e8d8bef9SDimitry Andric Record.push_back(N->getSizeInBits()); 1760e8d8bef9SDimitry Andric Record.push_back(N->getAlignInBits()); 1761e8d8bef9SDimitry Andric Record.push_back(N->getEncoding()); 1762e8d8bef9SDimitry Andric 1763e8d8bef9SDimitry Andric Stream.EmitRecord(bitc::METADATA_STRING_TYPE, Record, Abbrev); 1764e8d8bef9SDimitry Andric Record.clear(); 1765e8d8bef9SDimitry Andric } 1766e8d8bef9SDimitry Andric 17670b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIDerivedType(const DIDerivedType *N, 17680b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 17690b57cec5SDimitry Andric unsigned Abbrev) { 17700b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 17710b57cec5SDimitry Andric Record.push_back(N->getTag()); 17720b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 17730b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 17740b57cec5SDimitry Andric Record.push_back(N->getLine()); 17750b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 17760b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getBaseType())); 17770b57cec5SDimitry Andric Record.push_back(N->getSizeInBits()); 17780b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 17790b57cec5SDimitry Andric Record.push_back(N->getOffsetInBits()); 17800b57cec5SDimitry Andric Record.push_back(N->getFlags()); 17810b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getExtraData())); 17820b57cec5SDimitry Andric 17830b57cec5SDimitry Andric // DWARF address space is encoded as N->getDWARFAddressSpace() + 1. 0 means 17840b57cec5SDimitry Andric // that there is no DWARF address space associated with DIDerivedType. 17850b57cec5SDimitry Andric if (const auto &DWARFAddressSpace = N->getDWARFAddressSpace()) 17860b57cec5SDimitry Andric Record.push_back(*DWARFAddressSpace + 1); 17870b57cec5SDimitry Andric else 17880b57cec5SDimitry Andric Record.push_back(0); 17890b57cec5SDimitry Andric 1790349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 1791349cc55cSDimitry Andric 17920b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_DERIVED_TYPE, Record, Abbrev); 17930b57cec5SDimitry Andric Record.clear(); 17940b57cec5SDimitry Andric } 17950b57cec5SDimitry Andric 17960b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDICompositeType( 17970b57cec5SDimitry Andric const DICompositeType *N, SmallVectorImpl<uint64_t> &Record, 17980b57cec5SDimitry Andric unsigned Abbrev) { 17990b57cec5SDimitry Andric const unsigned IsNotUsedInOldTypeRef = 0x2; 18000b57cec5SDimitry Andric Record.push_back(IsNotUsedInOldTypeRef | (unsigned)N->isDistinct()); 18010b57cec5SDimitry Andric Record.push_back(N->getTag()); 18020b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 18030b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 18040b57cec5SDimitry Andric Record.push_back(N->getLine()); 18050b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 18060b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getBaseType())); 18070b57cec5SDimitry Andric Record.push_back(N->getSizeInBits()); 18080b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 18090b57cec5SDimitry Andric Record.push_back(N->getOffsetInBits()); 18100b57cec5SDimitry Andric Record.push_back(N->getFlags()); 18110b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getElements().get())); 18120b57cec5SDimitry Andric Record.push_back(N->getRuntimeLang()); 18130b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getVTableHolder())); 18140b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getTemplateParams().get())); 18150b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawIdentifier())); 18160b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getDiscriminator())); 18175ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawDataLocation())); 1818e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawAssociated())); 1819e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawAllocated())); 1820e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawRank())); 1821349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 18220b57cec5SDimitry Andric 18230b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_COMPOSITE_TYPE, Record, Abbrev); 18240b57cec5SDimitry Andric Record.clear(); 18250b57cec5SDimitry Andric } 18260b57cec5SDimitry Andric 18270b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDISubroutineType( 18280b57cec5SDimitry Andric const DISubroutineType *N, SmallVectorImpl<uint64_t> &Record, 18290b57cec5SDimitry Andric unsigned Abbrev) { 18300b57cec5SDimitry Andric const unsigned HasNoOldTypeRefs = 0x2; 18310b57cec5SDimitry Andric Record.push_back(HasNoOldTypeRefs | (unsigned)N->isDistinct()); 18320b57cec5SDimitry Andric Record.push_back(N->getFlags()); 18330b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getTypeArray().get())); 18340b57cec5SDimitry Andric Record.push_back(N->getCC()); 18350b57cec5SDimitry Andric 18360b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_SUBROUTINE_TYPE, Record, Abbrev); 18370b57cec5SDimitry Andric Record.clear(); 18380b57cec5SDimitry Andric } 18390b57cec5SDimitry Andric 18400b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIFile(const DIFile *N, 18410b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 18420b57cec5SDimitry Andric unsigned Abbrev) { 18430b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 18440b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawFilename())); 18450b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawDirectory())); 18460b57cec5SDimitry Andric if (N->getRawChecksum()) { 18470b57cec5SDimitry Andric Record.push_back(N->getRawChecksum()->Kind); 18480b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawChecksum()->Value)); 18490b57cec5SDimitry Andric } else { 18500b57cec5SDimitry Andric // Maintain backwards compatibility with the old internal representation of 18510b57cec5SDimitry Andric // CSK_None in ChecksumKind by writing nulls here when Checksum is None. 18520b57cec5SDimitry Andric Record.push_back(0); 18530b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(nullptr)); 18540b57cec5SDimitry Andric } 18550b57cec5SDimitry Andric auto Source = N->getRawSource(); 18560b57cec5SDimitry Andric if (Source) 1857bdd1243dSDimitry Andric Record.push_back(VE.getMetadataOrNullID(Source)); 18580b57cec5SDimitry Andric 18590b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_FILE, Record, Abbrev); 18600b57cec5SDimitry Andric Record.clear(); 18610b57cec5SDimitry Andric } 18620b57cec5SDimitry Andric 18630b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDICompileUnit(const DICompileUnit *N, 18640b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 18650b57cec5SDimitry Andric unsigned Abbrev) { 18660b57cec5SDimitry Andric assert(N->isDistinct() && "Expected distinct compile units"); 18670b57cec5SDimitry Andric Record.push_back(/* IsDistinct */ true); 18680b57cec5SDimitry Andric Record.push_back(N->getSourceLanguage()); 18690b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 18700b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawProducer())); 18710b57cec5SDimitry Andric Record.push_back(N->isOptimized()); 18720b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawFlags())); 18730b57cec5SDimitry Andric Record.push_back(N->getRuntimeVersion()); 18740b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawSplitDebugFilename())); 18750b57cec5SDimitry Andric Record.push_back(N->getEmissionKind()); 18760b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getEnumTypes().get())); 18770b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRetainedTypes().get())); 18780b57cec5SDimitry Andric Record.push_back(/* subprograms */ 0); 18790b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getGlobalVariables().get())); 18800b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getImportedEntities().get())); 18810b57cec5SDimitry Andric Record.push_back(N->getDWOId()); 18820b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getMacros().get())); 18830b57cec5SDimitry Andric Record.push_back(N->getSplitDebugInlining()); 18840b57cec5SDimitry Andric Record.push_back(N->getDebugInfoForProfiling()); 18850b57cec5SDimitry Andric Record.push_back((unsigned)N->getNameTableKind()); 18865ffd83dbSDimitry Andric Record.push_back(N->getRangesBaseAddress()); 18875ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawSysRoot())); 18885ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawSDK())); 18890b57cec5SDimitry Andric 18900b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_COMPILE_UNIT, Record, Abbrev); 18910b57cec5SDimitry Andric Record.clear(); 18920b57cec5SDimitry Andric } 18930b57cec5SDimitry Andric 18940b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDISubprogram(const DISubprogram *N, 18950b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 18960b57cec5SDimitry Andric unsigned Abbrev) { 18970b57cec5SDimitry Andric const uint64_t HasUnitFlag = 1 << 1; 18980b57cec5SDimitry Andric const uint64_t HasSPFlagsFlag = 1 << 2; 18990b57cec5SDimitry Andric Record.push_back(uint64_t(N->isDistinct()) | HasUnitFlag | HasSPFlagsFlag); 19000b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 19010b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 19020b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawLinkageName())); 19030b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 19040b57cec5SDimitry Andric Record.push_back(N->getLine()); 19050b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 19060b57cec5SDimitry Andric Record.push_back(N->getScopeLine()); 19070b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getContainingType())); 19080b57cec5SDimitry Andric Record.push_back(N->getSPFlags()); 19090b57cec5SDimitry Andric Record.push_back(N->getVirtualIndex()); 19100b57cec5SDimitry Andric Record.push_back(N->getFlags()); 19110b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawUnit())); 19120b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getTemplateParams().get())); 19130b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getDeclaration())); 19140b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRetainedNodes().get())); 19150b57cec5SDimitry Andric Record.push_back(N->getThisAdjustment()); 19160b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getThrownTypes().get())); 1917349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 191881ad6265SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawTargetFuncName())); 19190b57cec5SDimitry Andric 19200b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_SUBPROGRAM, Record, Abbrev); 19210b57cec5SDimitry Andric Record.clear(); 19220b57cec5SDimitry Andric } 19230b57cec5SDimitry Andric 19240b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILexicalBlock(const DILexicalBlock *N, 19250b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 19260b57cec5SDimitry Andric unsigned Abbrev) { 19270b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 19280b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 19290b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 19300b57cec5SDimitry Andric Record.push_back(N->getLine()); 19310b57cec5SDimitry Andric Record.push_back(N->getColumn()); 19320b57cec5SDimitry Andric 19330b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LEXICAL_BLOCK, Record, Abbrev); 19340b57cec5SDimitry Andric Record.clear(); 19350b57cec5SDimitry Andric } 19360b57cec5SDimitry Andric 19370b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILexicalBlockFile( 19380b57cec5SDimitry Andric const DILexicalBlockFile *N, SmallVectorImpl<uint64_t> &Record, 19390b57cec5SDimitry Andric unsigned Abbrev) { 19400b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 19410b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 19420b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 19430b57cec5SDimitry Andric Record.push_back(N->getDiscriminator()); 19440b57cec5SDimitry Andric 19450b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LEXICAL_BLOCK_FILE, Record, Abbrev); 19460b57cec5SDimitry Andric Record.clear(); 19470b57cec5SDimitry Andric } 19480b57cec5SDimitry Andric 19490b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDICommonBlock(const DICommonBlock *N, 19500b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 19510b57cec5SDimitry Andric unsigned Abbrev) { 19520b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 19530b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 19540b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getDecl())); 19550b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 19560b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 19570b57cec5SDimitry Andric Record.push_back(N->getLineNo()); 19580b57cec5SDimitry Andric 19590b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_COMMON_BLOCK, Record, Abbrev); 19600b57cec5SDimitry Andric Record.clear(); 19610b57cec5SDimitry Andric } 19620b57cec5SDimitry Andric 19630b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDINamespace(const DINamespace *N, 19640b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 19650b57cec5SDimitry Andric unsigned Abbrev) { 19660b57cec5SDimitry Andric Record.push_back(N->isDistinct() | N->getExportSymbols() << 1); 19670b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 19680b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 19690b57cec5SDimitry Andric 19700b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_NAMESPACE, Record, Abbrev); 19710b57cec5SDimitry Andric Record.clear(); 19720b57cec5SDimitry Andric } 19730b57cec5SDimitry Andric 19740b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIMacro(const DIMacro *N, 19750b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 19760b57cec5SDimitry Andric unsigned Abbrev) { 19770b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 19780b57cec5SDimitry Andric Record.push_back(N->getMacinfoType()); 19790b57cec5SDimitry Andric Record.push_back(N->getLine()); 19800b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 19810b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawValue())); 19820b57cec5SDimitry Andric 19830b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_MACRO, Record, Abbrev); 19840b57cec5SDimitry Andric Record.clear(); 19850b57cec5SDimitry Andric } 19860b57cec5SDimitry Andric 19870b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIMacroFile(const DIMacroFile *N, 19880b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 19890b57cec5SDimitry Andric unsigned Abbrev) { 19900b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 19910b57cec5SDimitry Andric Record.push_back(N->getMacinfoType()); 19920b57cec5SDimitry Andric Record.push_back(N->getLine()); 19930b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 19940b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getElements().get())); 19950b57cec5SDimitry Andric 19960b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_MACRO_FILE, Record, Abbrev); 19970b57cec5SDimitry Andric Record.clear(); 19980b57cec5SDimitry Andric } 19990b57cec5SDimitry Andric 2000fe6060f1SDimitry Andric void ModuleBitcodeWriter::writeDIArgList(const DIArgList *N, 2001*c9157d92SDimitry Andric SmallVectorImpl<uint64_t> &Record) { 2002fe6060f1SDimitry Andric Record.reserve(N->getArgs().size()); 2003fe6060f1SDimitry Andric for (ValueAsMetadata *MD : N->getArgs()) 2004fe6060f1SDimitry Andric Record.push_back(VE.getMetadataID(MD)); 2005fe6060f1SDimitry Andric 2006*c9157d92SDimitry Andric Stream.EmitRecord(bitc::METADATA_ARG_LIST, Record); 2007fe6060f1SDimitry Andric Record.clear(); 2008fe6060f1SDimitry Andric } 2009fe6060f1SDimitry Andric 20100b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIModule(const DIModule *N, 20110b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 20120b57cec5SDimitry Andric unsigned Abbrev) { 20130b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 20140b57cec5SDimitry Andric for (auto &I : N->operands()) 20150b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(I)); 20165ffd83dbSDimitry Andric Record.push_back(N->getLineNo()); 2017e8d8bef9SDimitry Andric Record.push_back(N->getIsDecl()); 20180b57cec5SDimitry Andric 20190b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_MODULE, Record, Abbrev); 20200b57cec5SDimitry Andric Record.clear(); 20210b57cec5SDimitry Andric } 20220b57cec5SDimitry Andric 2023bdd1243dSDimitry Andric void ModuleBitcodeWriter::writeDIAssignID(const DIAssignID *N, 2024bdd1243dSDimitry Andric SmallVectorImpl<uint64_t> &Record, 2025bdd1243dSDimitry Andric unsigned Abbrev) { 2026bdd1243dSDimitry Andric // There are no arguments for this metadata type. 2027bdd1243dSDimitry Andric Record.push_back(N->isDistinct()); 2028bdd1243dSDimitry Andric Stream.EmitRecord(bitc::METADATA_ASSIGN_ID, Record, Abbrev); 2029bdd1243dSDimitry Andric Record.clear(); 2030bdd1243dSDimitry Andric } 2031bdd1243dSDimitry Andric 20320b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDITemplateTypeParameter( 20330b57cec5SDimitry Andric const DITemplateTypeParameter *N, SmallVectorImpl<uint64_t> &Record, 20340b57cec5SDimitry Andric unsigned Abbrev) { 20350b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 20360b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 20370b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 20385ffd83dbSDimitry Andric Record.push_back(N->isDefault()); 20390b57cec5SDimitry Andric 20400b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_TEMPLATE_TYPE, Record, Abbrev); 20410b57cec5SDimitry Andric Record.clear(); 20420b57cec5SDimitry Andric } 20430b57cec5SDimitry Andric 20440b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDITemplateValueParameter( 20450b57cec5SDimitry Andric const DITemplateValueParameter *N, SmallVectorImpl<uint64_t> &Record, 20460b57cec5SDimitry Andric unsigned Abbrev) { 20470b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 20480b57cec5SDimitry Andric Record.push_back(N->getTag()); 20490b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 20500b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 20515ffd83dbSDimitry Andric Record.push_back(N->isDefault()); 20520b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getValue())); 20530b57cec5SDimitry Andric 20540b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_TEMPLATE_VALUE, Record, Abbrev); 20550b57cec5SDimitry Andric Record.clear(); 20560b57cec5SDimitry Andric } 20570b57cec5SDimitry Andric 20580b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIGlobalVariable( 20590b57cec5SDimitry Andric const DIGlobalVariable *N, SmallVectorImpl<uint64_t> &Record, 20600b57cec5SDimitry Andric unsigned Abbrev) { 20610b57cec5SDimitry Andric const uint64_t Version = 2 << 1; 20620b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct() | Version); 20630b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 20640b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 20650b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawLinkageName())); 20660b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 20670b57cec5SDimitry Andric Record.push_back(N->getLine()); 20680b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 20690b57cec5SDimitry Andric Record.push_back(N->isLocalToUnit()); 20700b57cec5SDimitry Andric Record.push_back(N->isDefinition()); 20710b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getStaticDataMemberDeclaration())); 20720b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getTemplateParams())); 20730b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 2074349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 20750b57cec5SDimitry Andric 20760b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_GLOBAL_VAR, Record, Abbrev); 20770b57cec5SDimitry Andric Record.clear(); 20780b57cec5SDimitry Andric } 20790b57cec5SDimitry Andric 20800b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILocalVariable( 20810b57cec5SDimitry Andric const DILocalVariable *N, SmallVectorImpl<uint64_t> &Record, 20820b57cec5SDimitry Andric unsigned Abbrev) { 20830b57cec5SDimitry Andric // In order to support all possible bitcode formats in BitcodeReader we need 20840b57cec5SDimitry Andric // to distinguish the following cases: 20850b57cec5SDimitry Andric // 1) Record has no artificial tag (Record[1]), 20860b57cec5SDimitry Andric // has no obsolete inlinedAt field (Record[9]). 20870b57cec5SDimitry Andric // In this case Record size will be 8, HasAlignment flag is false. 20880b57cec5SDimitry Andric // 2) Record has artificial tag (Record[1]), 20890b57cec5SDimitry Andric // has no obsolete inlignedAt field (Record[9]). 20900b57cec5SDimitry Andric // In this case Record size will be 9, HasAlignment flag is false. 20910b57cec5SDimitry Andric // 3) Record has both artificial tag (Record[1]) and 20920b57cec5SDimitry Andric // obsolete inlignedAt field (Record[9]). 20930b57cec5SDimitry Andric // In this case Record size will be 10, HasAlignment flag is false. 20940b57cec5SDimitry Andric // 4) Record has neither artificial tag, nor inlignedAt field, but 20950b57cec5SDimitry Andric // HasAlignment flag is true and Record[8] contains alignment value. 20960b57cec5SDimitry Andric const uint64_t HasAlignmentFlag = 1 << 1; 20970b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct() | HasAlignmentFlag); 20980b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 20990b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 21000b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 21010b57cec5SDimitry Andric Record.push_back(N->getLine()); 21020b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 21030b57cec5SDimitry Andric Record.push_back(N->getArg()); 21040b57cec5SDimitry Andric Record.push_back(N->getFlags()); 21050b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 2106349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 21070b57cec5SDimitry Andric 21080b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LOCAL_VAR, Record, Abbrev); 21090b57cec5SDimitry Andric Record.clear(); 21100b57cec5SDimitry Andric } 21110b57cec5SDimitry Andric 21120b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILabel( 21130b57cec5SDimitry Andric const DILabel *N, SmallVectorImpl<uint64_t> &Record, 21140b57cec5SDimitry Andric unsigned Abbrev) { 21150b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct()); 21160b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 21170b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 21180b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 21190b57cec5SDimitry Andric Record.push_back(N->getLine()); 21200b57cec5SDimitry Andric 21210b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LABEL, Record, Abbrev); 21220b57cec5SDimitry Andric Record.clear(); 21230b57cec5SDimitry Andric } 21240b57cec5SDimitry Andric 21250b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIExpression(const DIExpression *N, 21260b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 21270b57cec5SDimitry Andric unsigned Abbrev) { 21280b57cec5SDimitry Andric Record.reserve(N->getElements().size() + 1); 21290b57cec5SDimitry Andric const uint64_t Version = 3 << 1; 21300b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct() | Version); 21310b57cec5SDimitry Andric Record.append(N->elements_begin(), N->elements_end()); 21320b57cec5SDimitry Andric 21330b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_EXPRESSION, Record, Abbrev); 21340b57cec5SDimitry Andric Record.clear(); 21350b57cec5SDimitry Andric } 21360b57cec5SDimitry Andric 21370b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIGlobalVariableExpression( 21380b57cec5SDimitry Andric const DIGlobalVariableExpression *N, SmallVectorImpl<uint64_t> &Record, 21390b57cec5SDimitry Andric unsigned Abbrev) { 21400b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 21410b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getVariable())); 21420b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getExpression())); 21430b57cec5SDimitry Andric 21440b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_GLOBAL_VAR_EXPR, Record, Abbrev); 21450b57cec5SDimitry Andric Record.clear(); 21460b57cec5SDimitry Andric } 21470b57cec5SDimitry Andric 21480b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIObjCProperty(const DIObjCProperty *N, 21490b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 21500b57cec5SDimitry Andric unsigned Abbrev) { 21510b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 21520b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 21530b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 21540b57cec5SDimitry Andric Record.push_back(N->getLine()); 21550b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawSetterName())); 21560b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawGetterName())); 21570b57cec5SDimitry Andric Record.push_back(N->getAttributes()); 21580b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 21590b57cec5SDimitry Andric 21600b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_OBJC_PROPERTY, Record, Abbrev); 21610b57cec5SDimitry Andric Record.clear(); 21620b57cec5SDimitry Andric } 21630b57cec5SDimitry Andric 21640b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIImportedEntity( 21650b57cec5SDimitry Andric const DIImportedEntity *N, SmallVectorImpl<uint64_t> &Record, 21660b57cec5SDimitry Andric unsigned Abbrev) { 21670b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 21680b57cec5SDimitry Andric Record.push_back(N->getTag()); 21690b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 21700b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getEntity())); 21710b57cec5SDimitry Andric Record.push_back(N->getLine()); 21720b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 21730b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawFile())); 2174349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getElements().get())); 21750b57cec5SDimitry Andric 21760b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_IMPORTED_ENTITY, Record, Abbrev); 21770b57cec5SDimitry Andric Record.clear(); 21780b57cec5SDimitry Andric } 21790b57cec5SDimitry Andric 21800b57cec5SDimitry Andric unsigned ModuleBitcodeWriter::createNamedMetadataAbbrev() { 21810b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 21820b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_NAME)); 21830b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 21840b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 21850b57cec5SDimitry Andric return Stream.EmitAbbrev(std::move(Abbv)); 21860b57cec5SDimitry Andric } 21870b57cec5SDimitry Andric 21880b57cec5SDimitry Andric void ModuleBitcodeWriter::writeNamedMetadata( 21890b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record) { 21900b57cec5SDimitry Andric if (M.named_metadata_empty()) 21910b57cec5SDimitry Andric return; 21920b57cec5SDimitry Andric 21930b57cec5SDimitry Andric unsigned Abbrev = createNamedMetadataAbbrev(); 21940b57cec5SDimitry Andric for (const NamedMDNode &NMD : M.named_metadata()) { 21950b57cec5SDimitry Andric // Write name. 21960b57cec5SDimitry Andric StringRef Str = NMD.getName(); 21970b57cec5SDimitry Andric Record.append(Str.bytes_begin(), Str.bytes_end()); 21980b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_NAME, Record, Abbrev); 21990b57cec5SDimitry Andric Record.clear(); 22000b57cec5SDimitry Andric 22010b57cec5SDimitry Andric // Write named metadata operands. 22020b57cec5SDimitry Andric for (const MDNode *N : NMD.operands()) 22030b57cec5SDimitry Andric Record.push_back(VE.getMetadataID(N)); 22040b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_NAMED_NODE, Record, 0); 22050b57cec5SDimitry Andric Record.clear(); 22060b57cec5SDimitry Andric } 22070b57cec5SDimitry Andric } 22080b57cec5SDimitry Andric 22090b57cec5SDimitry Andric unsigned ModuleBitcodeWriter::createMetadataStringsAbbrev() { 22100b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 22110b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_STRINGS)); 22120b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // # of strings 22130b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // offset to chars 22140b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); 22150b57cec5SDimitry Andric return Stream.EmitAbbrev(std::move(Abbv)); 22160b57cec5SDimitry Andric } 22170b57cec5SDimitry Andric 22180b57cec5SDimitry Andric /// Write out a record for MDString. 22190b57cec5SDimitry Andric /// 22200b57cec5SDimitry Andric /// All the metadata strings in a metadata block are emitted in a single 22210b57cec5SDimitry Andric /// record. The sizes and strings themselves are shoved into a blob. 22220b57cec5SDimitry Andric void ModuleBitcodeWriter::writeMetadataStrings( 22230b57cec5SDimitry Andric ArrayRef<const Metadata *> Strings, SmallVectorImpl<uint64_t> &Record) { 22240b57cec5SDimitry Andric if (Strings.empty()) 22250b57cec5SDimitry Andric return; 22260b57cec5SDimitry Andric 22270b57cec5SDimitry Andric // Start the record with the number of strings. 22280b57cec5SDimitry Andric Record.push_back(bitc::METADATA_STRINGS); 22290b57cec5SDimitry Andric Record.push_back(Strings.size()); 22300b57cec5SDimitry Andric 22310b57cec5SDimitry Andric // Emit the sizes of the strings in the blob. 22320b57cec5SDimitry Andric SmallString<256> Blob; 22330b57cec5SDimitry Andric { 22340b57cec5SDimitry Andric BitstreamWriter W(Blob); 22350b57cec5SDimitry Andric for (const Metadata *MD : Strings) 22360b57cec5SDimitry Andric W.EmitVBR(cast<MDString>(MD)->getLength(), 6); 22370b57cec5SDimitry Andric W.FlushToWord(); 22380b57cec5SDimitry Andric } 22390b57cec5SDimitry Andric 22400b57cec5SDimitry Andric // Add the offset to the strings to the record. 22410b57cec5SDimitry Andric Record.push_back(Blob.size()); 22420b57cec5SDimitry Andric 22430b57cec5SDimitry Andric // Add the strings to the blob. 22440b57cec5SDimitry Andric for (const Metadata *MD : Strings) 22450b57cec5SDimitry Andric Blob.append(cast<MDString>(MD)->getString()); 22460b57cec5SDimitry Andric 22470b57cec5SDimitry Andric // Emit the final record. 22480b57cec5SDimitry Andric Stream.EmitRecordWithBlob(createMetadataStringsAbbrev(), Record, Blob); 22490b57cec5SDimitry Andric Record.clear(); 22500b57cec5SDimitry Andric } 22510b57cec5SDimitry Andric 22520b57cec5SDimitry Andric // Generates an enum to use as an index in the Abbrev array of Metadata record. 22530b57cec5SDimitry Andric enum MetadataAbbrev : unsigned { 22540b57cec5SDimitry Andric #define HANDLE_MDNODE_LEAF(CLASS) CLASS##AbbrevID, 22550b57cec5SDimitry Andric #include "llvm/IR/Metadata.def" 22560b57cec5SDimitry Andric LastPlusOne 22570b57cec5SDimitry Andric }; 22580b57cec5SDimitry Andric 22590b57cec5SDimitry Andric void ModuleBitcodeWriter::writeMetadataRecords( 22600b57cec5SDimitry Andric ArrayRef<const Metadata *> MDs, SmallVectorImpl<uint64_t> &Record, 22610b57cec5SDimitry Andric std::vector<unsigned> *MDAbbrevs, std::vector<uint64_t> *IndexPos) { 22620b57cec5SDimitry Andric if (MDs.empty()) 22630b57cec5SDimitry Andric return; 22640b57cec5SDimitry Andric 22650b57cec5SDimitry Andric // Initialize MDNode abbreviations. 22660b57cec5SDimitry Andric #define HANDLE_MDNODE_LEAF(CLASS) unsigned CLASS##Abbrev = 0; 22670b57cec5SDimitry Andric #include "llvm/IR/Metadata.def" 22680b57cec5SDimitry Andric 22690b57cec5SDimitry Andric for (const Metadata *MD : MDs) { 22700b57cec5SDimitry Andric if (IndexPos) 22710b57cec5SDimitry Andric IndexPos->push_back(Stream.GetCurrentBitNo()); 22720b57cec5SDimitry Andric if (const MDNode *N = dyn_cast<MDNode>(MD)) { 22730b57cec5SDimitry Andric assert(N->isResolved() && "Expected forward references to be resolved"); 22740b57cec5SDimitry Andric 22750b57cec5SDimitry Andric switch (N->getMetadataID()) { 22760b57cec5SDimitry Andric default: 22770b57cec5SDimitry Andric llvm_unreachable("Invalid MDNode subclass"); 22780b57cec5SDimitry Andric #define HANDLE_MDNODE_LEAF(CLASS) \ 22790b57cec5SDimitry Andric case Metadata::CLASS##Kind: \ 22800b57cec5SDimitry Andric if (MDAbbrevs) \ 22810b57cec5SDimitry Andric write##CLASS(cast<CLASS>(N), Record, \ 22820b57cec5SDimitry Andric (*MDAbbrevs)[MetadataAbbrev::CLASS##AbbrevID]); \ 22830b57cec5SDimitry Andric else \ 22840b57cec5SDimitry Andric write##CLASS(cast<CLASS>(N), Record, CLASS##Abbrev); \ 22850b57cec5SDimitry Andric continue; 22860b57cec5SDimitry Andric #include "llvm/IR/Metadata.def" 22870b57cec5SDimitry Andric } 22880b57cec5SDimitry Andric } 2289*c9157d92SDimitry Andric if (auto *AL = dyn_cast<DIArgList>(MD)) { 2290*c9157d92SDimitry Andric writeDIArgList(AL, Record); 2291*c9157d92SDimitry Andric continue; 2292*c9157d92SDimitry Andric } 22930b57cec5SDimitry Andric writeValueAsMetadata(cast<ValueAsMetadata>(MD), Record); 22940b57cec5SDimitry Andric } 22950b57cec5SDimitry Andric } 22960b57cec5SDimitry Andric 22970b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleMetadata() { 22980b57cec5SDimitry Andric if (!VE.hasMDs() && M.named_metadata_empty()) 22990b57cec5SDimitry Andric return; 23000b57cec5SDimitry Andric 23010b57cec5SDimitry Andric Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 4); 23020b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 23030b57cec5SDimitry Andric 23040b57cec5SDimitry Andric // Emit all abbrevs upfront, so that the reader can jump in the middle of the 23050b57cec5SDimitry Andric // block and load any metadata. 23060b57cec5SDimitry Andric std::vector<unsigned> MDAbbrevs; 23070b57cec5SDimitry Andric 23080b57cec5SDimitry Andric MDAbbrevs.resize(MetadataAbbrev::LastPlusOne); 23090b57cec5SDimitry Andric MDAbbrevs[MetadataAbbrev::DILocationAbbrevID] = createDILocationAbbrev(); 23100b57cec5SDimitry Andric MDAbbrevs[MetadataAbbrev::GenericDINodeAbbrevID] = 23110b57cec5SDimitry Andric createGenericDINodeAbbrev(); 23120b57cec5SDimitry Andric 23130b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 23140b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_INDEX_OFFSET)); 23150b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 23160b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 23170b57cec5SDimitry Andric unsigned OffsetAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 23180b57cec5SDimitry Andric 23190b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 23200b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_INDEX)); 23210b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 23220b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 23230b57cec5SDimitry Andric unsigned IndexAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 23240b57cec5SDimitry Andric 23250b57cec5SDimitry Andric // Emit MDStrings together upfront. 23260b57cec5SDimitry Andric writeMetadataStrings(VE.getMDStrings(), Record); 23270b57cec5SDimitry Andric 23280b57cec5SDimitry Andric // We only emit an index for the metadata record if we have more than a given 23290b57cec5SDimitry Andric // (naive) threshold of metadatas, otherwise it is not worth it. 23300b57cec5SDimitry Andric if (VE.getNonMDStrings().size() > IndexThreshold) { 23310b57cec5SDimitry Andric // Write a placeholder value in for the offset of the metadata index, 23320b57cec5SDimitry Andric // which is written after the records, so that it can include 23330b57cec5SDimitry Andric // the offset of each entry. The placeholder offset will be 23340b57cec5SDimitry Andric // updated after all records are emitted. 23350b57cec5SDimitry Andric uint64_t Vals[] = {0, 0}; 23360b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_INDEX_OFFSET, Vals, OffsetAbbrev); 23370b57cec5SDimitry Andric } 23380b57cec5SDimitry Andric 23390b57cec5SDimitry Andric // Compute and save the bit offset to the current position, which will be 23400b57cec5SDimitry Andric // patched when we emit the index later. We can simply subtract the 64-bit 23410b57cec5SDimitry Andric // fixed size from the current bit number to get the location to backpatch. 23420b57cec5SDimitry Andric uint64_t IndexOffsetRecordBitPos = Stream.GetCurrentBitNo(); 23430b57cec5SDimitry Andric 23440b57cec5SDimitry Andric // This index will contain the bitpos for each individual record. 23450b57cec5SDimitry Andric std::vector<uint64_t> IndexPos; 23460b57cec5SDimitry Andric IndexPos.reserve(VE.getNonMDStrings().size()); 23470b57cec5SDimitry Andric 23480b57cec5SDimitry Andric // Write all the records 23490b57cec5SDimitry Andric writeMetadataRecords(VE.getNonMDStrings(), Record, &MDAbbrevs, &IndexPos); 23500b57cec5SDimitry Andric 23510b57cec5SDimitry Andric if (VE.getNonMDStrings().size() > IndexThreshold) { 23520b57cec5SDimitry Andric // Now that we have emitted all the records we will emit the index. But 23530b57cec5SDimitry Andric // first 23540b57cec5SDimitry Andric // backpatch the forward reference so that the reader can skip the records 23550b57cec5SDimitry Andric // efficiently. 23560b57cec5SDimitry Andric Stream.BackpatchWord64(IndexOffsetRecordBitPos - 64, 23570b57cec5SDimitry Andric Stream.GetCurrentBitNo() - IndexOffsetRecordBitPos); 23580b57cec5SDimitry Andric 23590b57cec5SDimitry Andric // Delta encode the index. 23600b57cec5SDimitry Andric uint64_t PreviousValue = IndexOffsetRecordBitPos; 23610b57cec5SDimitry Andric for (auto &Elt : IndexPos) { 23620b57cec5SDimitry Andric auto EltDelta = Elt - PreviousValue; 23630b57cec5SDimitry Andric PreviousValue = Elt; 23640b57cec5SDimitry Andric Elt = EltDelta; 23650b57cec5SDimitry Andric } 23660b57cec5SDimitry Andric // Emit the index record. 23670b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_INDEX, IndexPos, IndexAbbrev); 23680b57cec5SDimitry Andric IndexPos.clear(); 23690b57cec5SDimitry Andric } 23700b57cec5SDimitry Andric 23710b57cec5SDimitry Andric // Write the named metadata now. 23720b57cec5SDimitry Andric writeNamedMetadata(Record); 23730b57cec5SDimitry Andric 23740b57cec5SDimitry Andric auto AddDeclAttachedMetadata = [&](const GlobalObject &GO) { 23750b57cec5SDimitry Andric SmallVector<uint64_t, 4> Record; 23760b57cec5SDimitry Andric Record.push_back(VE.getValueID(&GO)); 23770b57cec5SDimitry Andric pushGlobalMetadataAttachment(Record, GO); 23780b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_GLOBAL_DECL_ATTACHMENT, Record); 23790b57cec5SDimitry Andric }; 23800b57cec5SDimitry Andric for (const Function &F : M) 23810b57cec5SDimitry Andric if (F.isDeclaration() && F.hasMetadata()) 23820b57cec5SDimitry Andric AddDeclAttachedMetadata(F); 23830b57cec5SDimitry Andric // FIXME: Only store metadata for declarations here, and move data for global 23840b57cec5SDimitry Andric // variable definitions to a separate block (PR28134). 23850b57cec5SDimitry Andric for (const GlobalVariable &GV : M.globals()) 23860b57cec5SDimitry Andric if (GV.hasMetadata()) 23870b57cec5SDimitry Andric AddDeclAttachedMetadata(GV); 23880b57cec5SDimitry Andric 23890b57cec5SDimitry Andric Stream.ExitBlock(); 23900b57cec5SDimitry Andric } 23910b57cec5SDimitry Andric 23920b57cec5SDimitry Andric void ModuleBitcodeWriter::writeFunctionMetadata(const Function &F) { 23930b57cec5SDimitry Andric if (!VE.hasMDs()) 23940b57cec5SDimitry Andric return; 23950b57cec5SDimitry Andric 23960b57cec5SDimitry Andric Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 23970b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 23980b57cec5SDimitry Andric writeMetadataStrings(VE.getMDStrings(), Record); 23990b57cec5SDimitry Andric writeMetadataRecords(VE.getNonMDStrings(), Record); 24000b57cec5SDimitry Andric Stream.ExitBlock(); 24010b57cec5SDimitry Andric } 24020b57cec5SDimitry Andric 24030b57cec5SDimitry Andric void ModuleBitcodeWriter::pushGlobalMetadataAttachment( 24040b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, const GlobalObject &GO) { 24050b57cec5SDimitry Andric // [n x [id, mdnode]] 24060b57cec5SDimitry Andric SmallVector<std::pair<unsigned, MDNode *>, 4> MDs; 24070b57cec5SDimitry Andric GO.getAllMetadata(MDs); 24080b57cec5SDimitry Andric for (const auto &I : MDs) { 24090b57cec5SDimitry Andric Record.push_back(I.first); 24100b57cec5SDimitry Andric Record.push_back(VE.getMetadataID(I.second)); 24110b57cec5SDimitry Andric } 24120b57cec5SDimitry Andric } 24130b57cec5SDimitry Andric 24140b57cec5SDimitry Andric void ModuleBitcodeWriter::writeFunctionMetadataAttachment(const Function &F) { 24150b57cec5SDimitry Andric Stream.EnterSubblock(bitc::METADATA_ATTACHMENT_ID, 3); 24160b57cec5SDimitry Andric 24170b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 24180b57cec5SDimitry Andric 24190b57cec5SDimitry Andric if (F.hasMetadata()) { 24200b57cec5SDimitry Andric pushGlobalMetadataAttachment(Record, F); 24210b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_ATTACHMENT, Record, 0); 24220b57cec5SDimitry Andric Record.clear(); 24230b57cec5SDimitry Andric } 24240b57cec5SDimitry Andric 24250b57cec5SDimitry Andric // Write metadata attachments 24260b57cec5SDimitry Andric // METADATA_ATTACHMENT - [m x [value, [n x [id, mdnode]]] 24270b57cec5SDimitry Andric SmallVector<std::pair<unsigned, MDNode *>, 4> MDs; 24280b57cec5SDimitry Andric for (const BasicBlock &BB : F) 24290b57cec5SDimitry Andric for (const Instruction &I : BB) { 24300b57cec5SDimitry Andric MDs.clear(); 24310b57cec5SDimitry Andric I.getAllMetadataOtherThanDebugLoc(MDs); 24320b57cec5SDimitry Andric 24330b57cec5SDimitry Andric // If no metadata, ignore instruction. 24340b57cec5SDimitry Andric if (MDs.empty()) continue; 24350b57cec5SDimitry Andric 24360b57cec5SDimitry Andric Record.push_back(VE.getInstructionID(&I)); 24370b57cec5SDimitry Andric 24380b57cec5SDimitry Andric for (unsigned i = 0, e = MDs.size(); i != e; ++i) { 24390b57cec5SDimitry Andric Record.push_back(MDs[i].first); 24400b57cec5SDimitry Andric Record.push_back(VE.getMetadataID(MDs[i].second)); 24410b57cec5SDimitry Andric } 24420b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_ATTACHMENT, Record, 0); 24430b57cec5SDimitry Andric Record.clear(); 24440b57cec5SDimitry Andric } 24450b57cec5SDimitry Andric 24460b57cec5SDimitry Andric Stream.ExitBlock(); 24470b57cec5SDimitry Andric } 24480b57cec5SDimitry Andric 24490b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleMetadataKinds() { 24500b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 24510b57cec5SDimitry Andric 24520b57cec5SDimitry Andric // Write metadata kinds 24530b57cec5SDimitry Andric // METADATA_KIND - [n x [id, name]] 24540b57cec5SDimitry Andric SmallVector<StringRef, 8> Names; 24550b57cec5SDimitry Andric M.getMDKindNames(Names); 24560b57cec5SDimitry Andric 24570b57cec5SDimitry Andric if (Names.empty()) return; 24580b57cec5SDimitry Andric 24590b57cec5SDimitry Andric Stream.EnterSubblock(bitc::METADATA_KIND_BLOCK_ID, 3); 24600b57cec5SDimitry Andric 24610b57cec5SDimitry Andric for (unsigned MDKindID = 0, e = Names.size(); MDKindID != e; ++MDKindID) { 24620b57cec5SDimitry Andric Record.push_back(MDKindID); 24630b57cec5SDimitry Andric StringRef KName = Names[MDKindID]; 24640b57cec5SDimitry Andric Record.append(KName.begin(), KName.end()); 24650b57cec5SDimitry Andric 24660b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_KIND, Record, 0); 24670b57cec5SDimitry Andric Record.clear(); 24680b57cec5SDimitry Andric } 24690b57cec5SDimitry Andric 24700b57cec5SDimitry Andric Stream.ExitBlock(); 24710b57cec5SDimitry Andric } 24720b57cec5SDimitry Andric 24730b57cec5SDimitry Andric void ModuleBitcodeWriter::writeOperandBundleTags() { 24740b57cec5SDimitry Andric // Write metadata kinds 24750b57cec5SDimitry Andric // 24760b57cec5SDimitry Andric // OPERAND_BUNDLE_TAGS_BLOCK_ID : N x OPERAND_BUNDLE_TAG 24770b57cec5SDimitry Andric // 24780b57cec5SDimitry Andric // OPERAND_BUNDLE_TAG - [strchr x N] 24790b57cec5SDimitry Andric 24800b57cec5SDimitry Andric SmallVector<StringRef, 8> Tags; 24810b57cec5SDimitry Andric M.getOperandBundleTags(Tags); 24820b57cec5SDimitry Andric 24830b57cec5SDimitry Andric if (Tags.empty()) 24840b57cec5SDimitry Andric return; 24850b57cec5SDimitry Andric 24860b57cec5SDimitry Andric Stream.EnterSubblock(bitc::OPERAND_BUNDLE_TAGS_BLOCK_ID, 3); 24870b57cec5SDimitry Andric 24880b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 24890b57cec5SDimitry Andric 24900b57cec5SDimitry Andric for (auto Tag : Tags) { 24910b57cec5SDimitry Andric Record.append(Tag.begin(), Tag.end()); 24920b57cec5SDimitry Andric 24930b57cec5SDimitry Andric Stream.EmitRecord(bitc::OPERAND_BUNDLE_TAG, Record, 0); 24940b57cec5SDimitry Andric Record.clear(); 24950b57cec5SDimitry Andric } 24960b57cec5SDimitry Andric 24970b57cec5SDimitry Andric Stream.ExitBlock(); 24980b57cec5SDimitry Andric } 24990b57cec5SDimitry Andric 25000b57cec5SDimitry Andric void ModuleBitcodeWriter::writeSyncScopeNames() { 25010b57cec5SDimitry Andric SmallVector<StringRef, 8> SSNs; 25020b57cec5SDimitry Andric M.getContext().getSyncScopeNames(SSNs); 25030b57cec5SDimitry Andric if (SSNs.empty()) 25040b57cec5SDimitry Andric return; 25050b57cec5SDimitry Andric 25060b57cec5SDimitry Andric Stream.EnterSubblock(bitc::SYNC_SCOPE_NAMES_BLOCK_ID, 2); 25070b57cec5SDimitry Andric 25080b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 25090b57cec5SDimitry Andric for (auto SSN : SSNs) { 25100b57cec5SDimitry Andric Record.append(SSN.begin(), SSN.end()); 25110b57cec5SDimitry Andric Stream.EmitRecord(bitc::SYNC_SCOPE_NAME, Record, 0); 25120b57cec5SDimitry Andric Record.clear(); 25130b57cec5SDimitry Andric } 25140b57cec5SDimitry Andric 25150b57cec5SDimitry Andric Stream.ExitBlock(); 25160b57cec5SDimitry Andric } 25170b57cec5SDimitry Andric 25180b57cec5SDimitry Andric void ModuleBitcodeWriter::writeConstants(unsigned FirstVal, unsigned LastVal, 25190b57cec5SDimitry Andric bool isGlobal) { 25200b57cec5SDimitry Andric if (FirstVal == LastVal) return; 25210b57cec5SDimitry Andric 25220b57cec5SDimitry Andric Stream.EnterSubblock(bitc::CONSTANTS_BLOCK_ID, 4); 25230b57cec5SDimitry Andric 25240b57cec5SDimitry Andric unsigned AggregateAbbrev = 0; 25250b57cec5SDimitry Andric unsigned String8Abbrev = 0; 25260b57cec5SDimitry Andric unsigned CString7Abbrev = 0; 25270b57cec5SDimitry Andric unsigned CString6Abbrev = 0; 25280b57cec5SDimitry Andric // If this is a constant pool for the module, emit module-specific abbrevs. 25290b57cec5SDimitry Andric if (isGlobal) { 25300b57cec5SDimitry Andric // Abbrev for CST_CODE_AGGREGATE. 25310b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 25320b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_AGGREGATE)); 25330b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 25340b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, Log2_32_Ceil(LastVal+1))); 25350b57cec5SDimitry Andric AggregateAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 25360b57cec5SDimitry Andric 25370b57cec5SDimitry Andric // Abbrev for CST_CODE_STRING. 25380b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 25390b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_STRING)); 25400b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 25410b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 25420b57cec5SDimitry Andric String8Abbrev = Stream.EmitAbbrev(std::move(Abbv)); 25430b57cec5SDimitry Andric // Abbrev for CST_CODE_CSTRING. 25440b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 25450b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CSTRING)); 25460b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 25470b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 25480b57cec5SDimitry Andric CString7Abbrev = Stream.EmitAbbrev(std::move(Abbv)); 25490b57cec5SDimitry Andric // Abbrev for CST_CODE_CSTRING. 25500b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 25510b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CSTRING)); 25520b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 25530b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 25540b57cec5SDimitry Andric CString6Abbrev = Stream.EmitAbbrev(std::move(Abbv)); 25550b57cec5SDimitry Andric } 25560b57cec5SDimitry Andric 25570b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 25580b57cec5SDimitry Andric 25590b57cec5SDimitry Andric const ValueEnumerator::ValueList &Vals = VE.getValues(); 25600b57cec5SDimitry Andric Type *LastTy = nullptr; 25610b57cec5SDimitry Andric for (unsigned i = FirstVal; i != LastVal; ++i) { 25620b57cec5SDimitry Andric const Value *V = Vals[i].first; 25630b57cec5SDimitry Andric // If we need to switch types, do so now. 25640b57cec5SDimitry Andric if (V->getType() != LastTy) { 25650b57cec5SDimitry Andric LastTy = V->getType(); 25660b57cec5SDimitry Andric Record.push_back(VE.getTypeID(LastTy)); 25670b57cec5SDimitry Andric Stream.EmitRecord(bitc::CST_CODE_SETTYPE, Record, 25680b57cec5SDimitry Andric CONSTANTS_SETTYPE_ABBREV); 25690b57cec5SDimitry Andric Record.clear(); 25700b57cec5SDimitry Andric } 25710b57cec5SDimitry Andric 25720b57cec5SDimitry Andric if (const InlineAsm *IA = dyn_cast<InlineAsm>(V)) { 257304eeddc0SDimitry Andric Record.push_back(VE.getTypeID(IA->getFunctionType())); 2574fe6060f1SDimitry Andric Record.push_back( 2575fe6060f1SDimitry Andric unsigned(IA->hasSideEffects()) | unsigned(IA->isAlignStack()) << 1 | 2576fe6060f1SDimitry Andric unsigned(IA->getDialect() & 1) << 2 | unsigned(IA->canThrow()) << 3); 25770b57cec5SDimitry Andric 25780b57cec5SDimitry Andric // Add the asm string. 25790b57cec5SDimitry Andric const std::string &AsmStr = IA->getAsmString(); 25800b57cec5SDimitry Andric Record.push_back(AsmStr.size()); 25810b57cec5SDimitry Andric Record.append(AsmStr.begin(), AsmStr.end()); 25820b57cec5SDimitry Andric 25830b57cec5SDimitry Andric // Add the constraint string. 25840b57cec5SDimitry Andric const std::string &ConstraintStr = IA->getConstraintString(); 25850b57cec5SDimitry Andric Record.push_back(ConstraintStr.size()); 25860b57cec5SDimitry Andric Record.append(ConstraintStr.begin(), ConstraintStr.end()); 25870b57cec5SDimitry Andric Stream.EmitRecord(bitc::CST_CODE_INLINEASM, Record); 25880b57cec5SDimitry Andric Record.clear(); 25890b57cec5SDimitry Andric continue; 25900b57cec5SDimitry Andric } 25910b57cec5SDimitry Andric const Constant *C = cast<Constant>(V); 25920b57cec5SDimitry Andric unsigned Code = -1U; 25930b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 25940b57cec5SDimitry Andric if (C->isNullValue()) { 25950b57cec5SDimitry Andric Code = bitc::CST_CODE_NULL; 2596e8d8bef9SDimitry Andric } else if (isa<PoisonValue>(C)) { 2597e8d8bef9SDimitry Andric Code = bitc::CST_CODE_POISON; 25980b57cec5SDimitry Andric } else if (isa<UndefValue>(C)) { 25990b57cec5SDimitry Andric Code = bitc::CST_CODE_UNDEF; 26000b57cec5SDimitry Andric } else if (const ConstantInt *IV = dyn_cast<ConstantInt>(C)) { 26010b57cec5SDimitry Andric if (IV->getBitWidth() <= 64) { 26020b57cec5SDimitry Andric uint64_t V = IV->getSExtValue(); 26030b57cec5SDimitry Andric emitSignedInt64(Record, V); 26040b57cec5SDimitry Andric Code = bitc::CST_CODE_INTEGER; 26050b57cec5SDimitry Andric AbbrevToUse = CONSTANTS_INTEGER_ABBREV; 26060b57cec5SDimitry Andric } else { // Wide integers, > 64 bits in size. 26075ffd83dbSDimitry Andric emitWideAPInt(Record, IV->getValue()); 26080b57cec5SDimitry Andric Code = bitc::CST_CODE_WIDE_INTEGER; 26090b57cec5SDimitry Andric } 26100b57cec5SDimitry Andric } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(C)) { 26110b57cec5SDimitry Andric Code = bitc::CST_CODE_FLOAT; 26120b57cec5SDimitry Andric Type *Ty = CFP->getType(); 26135ffd83dbSDimitry Andric if (Ty->isHalfTy() || Ty->isBFloatTy() || Ty->isFloatTy() || 26145ffd83dbSDimitry Andric Ty->isDoubleTy()) { 26150b57cec5SDimitry Andric Record.push_back(CFP->getValueAPF().bitcastToAPInt().getZExtValue()); 26160b57cec5SDimitry Andric } else if (Ty->isX86_FP80Ty()) { 26170b57cec5SDimitry Andric // api needed to prevent premature destruction 26180b57cec5SDimitry Andric // bits are not in the same order as a normal i80 APInt, compensate. 26190b57cec5SDimitry Andric APInt api = CFP->getValueAPF().bitcastToAPInt(); 26200b57cec5SDimitry Andric const uint64_t *p = api.getRawData(); 26210b57cec5SDimitry Andric Record.push_back((p[1] << 48) | (p[0] >> 16)); 26220b57cec5SDimitry Andric Record.push_back(p[0] & 0xffffLL); 26230b57cec5SDimitry Andric } else if (Ty->isFP128Ty() || Ty->isPPC_FP128Ty()) { 26240b57cec5SDimitry Andric APInt api = CFP->getValueAPF().bitcastToAPInt(); 26250b57cec5SDimitry Andric const uint64_t *p = api.getRawData(); 26260b57cec5SDimitry Andric Record.push_back(p[0]); 26270b57cec5SDimitry Andric Record.push_back(p[1]); 26280b57cec5SDimitry Andric } else { 26290b57cec5SDimitry Andric assert(0 && "Unknown FP type!"); 26300b57cec5SDimitry Andric } 26310b57cec5SDimitry Andric } else if (isa<ConstantDataSequential>(C) && 26320b57cec5SDimitry Andric cast<ConstantDataSequential>(C)->isString()) { 26330b57cec5SDimitry Andric const ConstantDataSequential *Str = cast<ConstantDataSequential>(C); 26340b57cec5SDimitry Andric // Emit constant strings specially. 26350b57cec5SDimitry Andric unsigned NumElts = Str->getNumElements(); 26360b57cec5SDimitry Andric // If this is a null-terminated string, use the denser CSTRING encoding. 26370b57cec5SDimitry Andric if (Str->isCString()) { 26380b57cec5SDimitry Andric Code = bitc::CST_CODE_CSTRING; 26390b57cec5SDimitry Andric --NumElts; // Don't encode the null, which isn't allowed by char6. 26400b57cec5SDimitry Andric } else { 26410b57cec5SDimitry Andric Code = bitc::CST_CODE_STRING; 26420b57cec5SDimitry Andric AbbrevToUse = String8Abbrev; 26430b57cec5SDimitry Andric } 26440b57cec5SDimitry Andric bool isCStr7 = Code == bitc::CST_CODE_CSTRING; 26450b57cec5SDimitry Andric bool isCStrChar6 = Code == bitc::CST_CODE_CSTRING; 26460b57cec5SDimitry Andric for (unsigned i = 0; i != NumElts; ++i) { 26470b57cec5SDimitry Andric unsigned char V = Str->getElementAsInteger(i); 26480b57cec5SDimitry Andric Record.push_back(V); 26490b57cec5SDimitry Andric isCStr7 &= (V & 128) == 0; 26500b57cec5SDimitry Andric if (isCStrChar6) 26510b57cec5SDimitry Andric isCStrChar6 = BitCodeAbbrevOp::isChar6(V); 26520b57cec5SDimitry Andric } 26530b57cec5SDimitry Andric 26540b57cec5SDimitry Andric if (isCStrChar6) 26550b57cec5SDimitry Andric AbbrevToUse = CString6Abbrev; 26560b57cec5SDimitry Andric else if (isCStr7) 26570b57cec5SDimitry Andric AbbrevToUse = CString7Abbrev; 26580b57cec5SDimitry Andric } else if (const ConstantDataSequential *CDS = 26590b57cec5SDimitry Andric dyn_cast<ConstantDataSequential>(C)) { 26600b57cec5SDimitry Andric Code = bitc::CST_CODE_DATA; 26615ffd83dbSDimitry Andric Type *EltTy = CDS->getElementType(); 26620b57cec5SDimitry Andric if (isa<IntegerType>(EltTy)) { 26630b57cec5SDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) 26640b57cec5SDimitry Andric Record.push_back(CDS->getElementAsInteger(i)); 26650b57cec5SDimitry Andric } else { 26660b57cec5SDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) 26670b57cec5SDimitry Andric Record.push_back( 26680b57cec5SDimitry Andric CDS->getElementAsAPFloat(i).bitcastToAPInt().getLimitedValue()); 26690b57cec5SDimitry Andric } 26700b57cec5SDimitry Andric } else if (isa<ConstantAggregate>(C)) { 26710b57cec5SDimitry Andric Code = bitc::CST_CODE_AGGREGATE; 26720b57cec5SDimitry Andric for (const Value *Op : C->operands()) 26730b57cec5SDimitry Andric Record.push_back(VE.getValueID(Op)); 26740b57cec5SDimitry Andric AbbrevToUse = AggregateAbbrev; 26750b57cec5SDimitry Andric } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 26760b57cec5SDimitry Andric switch (CE->getOpcode()) { 26770b57cec5SDimitry Andric default: 26780b57cec5SDimitry Andric if (Instruction::isCast(CE->getOpcode())) { 26790b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_CAST; 26800b57cec5SDimitry Andric Record.push_back(getEncodedCastOpcode(CE->getOpcode())); 26810b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 26820b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 26830b57cec5SDimitry Andric AbbrevToUse = CONSTANTS_CE_CAST_Abbrev; 26840b57cec5SDimitry Andric } else { 26850b57cec5SDimitry Andric assert(CE->getNumOperands() == 2 && "Unknown constant expr!"); 26860b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_BINOP; 26870b57cec5SDimitry Andric Record.push_back(getEncodedBinaryOpcode(CE->getOpcode())); 26880b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 26890b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 26900b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(CE); 26910b57cec5SDimitry Andric if (Flags != 0) 26920b57cec5SDimitry Andric Record.push_back(Flags); 26930b57cec5SDimitry Andric } 26940b57cec5SDimitry Andric break; 26950b57cec5SDimitry Andric case Instruction::FNeg: { 26960b57cec5SDimitry Andric assert(CE->getNumOperands() == 1 && "Unknown constant expr!"); 26970b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_UNOP; 26980b57cec5SDimitry Andric Record.push_back(getEncodedUnaryOpcode(CE->getOpcode())); 26990b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 27000b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(CE); 27010b57cec5SDimitry Andric if (Flags != 0) 27020b57cec5SDimitry Andric Record.push_back(Flags); 27030b57cec5SDimitry Andric break; 27040b57cec5SDimitry Andric } 27050b57cec5SDimitry Andric case Instruction::GetElementPtr: { 27060b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_GEP; 27070b57cec5SDimitry Andric const auto *GO = cast<GEPOperator>(C); 27080b57cec5SDimitry Andric Record.push_back(VE.getTypeID(GO->getSourceElementType())); 2709bdd1243dSDimitry Andric if (std::optional<unsigned> Idx = GO->getInRangeIndex()) { 27100b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_GEP_WITH_INRANGE_INDEX; 27110b57cec5SDimitry Andric Record.push_back((*Idx << 1) | GO->isInBounds()); 27120b57cec5SDimitry Andric } else if (GO->isInBounds()) 27130b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_INBOUNDS_GEP; 27140b57cec5SDimitry Andric for (unsigned i = 0, e = CE->getNumOperands(); i != e; ++i) { 27150b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(i)->getType())); 27160b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(i))); 27170b57cec5SDimitry Andric } 27180b57cec5SDimitry Andric break; 27190b57cec5SDimitry Andric } 27200b57cec5SDimitry Andric case Instruction::ExtractElement: 27210b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_EXTRACTELT; 27220b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 27230b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 27240b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(1)->getType())); 27250b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 27260b57cec5SDimitry Andric break; 27270b57cec5SDimitry Andric case Instruction::InsertElement: 27280b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_INSERTELT; 27290b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 27300b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 27310b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(2)->getType())); 27320b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(2))); 27330b57cec5SDimitry Andric break; 27340b57cec5SDimitry Andric case Instruction::ShuffleVector: 27350b57cec5SDimitry Andric // If the return type and argument types are the same, this is a 27360b57cec5SDimitry Andric // standard shufflevector instruction. If the types are different, 27370b57cec5SDimitry Andric // then the shuffle is widening or truncating the input vectors, and 27380b57cec5SDimitry Andric // the argument type must also be encoded. 27390b57cec5SDimitry Andric if (C->getType() == C->getOperand(0)->getType()) { 27400b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_SHUFFLEVEC; 27410b57cec5SDimitry Andric } else { 27420b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_SHUFVEC_EX; 27430b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 27440b57cec5SDimitry Andric } 27450b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 27460b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 27475ffd83dbSDimitry Andric Record.push_back(VE.getValueID(CE->getShuffleMaskForBitcode())); 27480b57cec5SDimitry Andric break; 27490b57cec5SDimitry Andric case Instruction::ICmp: 27500b57cec5SDimitry Andric case Instruction::FCmp: 27510b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_CMP; 27520b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 27530b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 27540b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 27550b57cec5SDimitry Andric Record.push_back(CE->getPredicate()); 27560b57cec5SDimitry Andric break; 27570b57cec5SDimitry Andric } 27580b57cec5SDimitry Andric } else if (const BlockAddress *BA = dyn_cast<BlockAddress>(C)) { 27590b57cec5SDimitry Andric Code = bitc::CST_CODE_BLOCKADDRESS; 27600b57cec5SDimitry Andric Record.push_back(VE.getTypeID(BA->getFunction()->getType())); 27610b57cec5SDimitry Andric Record.push_back(VE.getValueID(BA->getFunction())); 27620b57cec5SDimitry Andric Record.push_back(VE.getGlobalBasicBlockID(BA->getBasicBlock())); 2763fe6060f1SDimitry Andric } else if (const auto *Equiv = dyn_cast<DSOLocalEquivalent>(C)) { 2764fe6060f1SDimitry Andric Code = bitc::CST_CODE_DSO_LOCAL_EQUIVALENT; 2765fe6060f1SDimitry Andric Record.push_back(VE.getTypeID(Equiv->getGlobalValue()->getType())); 2766fe6060f1SDimitry Andric Record.push_back(VE.getValueID(Equiv->getGlobalValue())); 27670eae32dcSDimitry Andric } else if (const auto *NC = dyn_cast<NoCFIValue>(C)) { 27680eae32dcSDimitry Andric Code = bitc::CST_CODE_NO_CFI_VALUE; 27690eae32dcSDimitry Andric Record.push_back(VE.getTypeID(NC->getGlobalValue()->getType())); 27700eae32dcSDimitry Andric Record.push_back(VE.getValueID(NC->getGlobalValue())); 27710b57cec5SDimitry Andric } else { 27720b57cec5SDimitry Andric #ifndef NDEBUG 27730b57cec5SDimitry Andric C->dump(); 27740b57cec5SDimitry Andric #endif 27750b57cec5SDimitry Andric llvm_unreachable("Unknown constant!"); 27760b57cec5SDimitry Andric } 27770b57cec5SDimitry Andric Stream.EmitRecord(Code, Record, AbbrevToUse); 27780b57cec5SDimitry Andric Record.clear(); 27790b57cec5SDimitry Andric } 27800b57cec5SDimitry Andric 27810b57cec5SDimitry Andric Stream.ExitBlock(); 27820b57cec5SDimitry Andric } 27830b57cec5SDimitry Andric 27840b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleConstants() { 27850b57cec5SDimitry Andric const ValueEnumerator::ValueList &Vals = VE.getValues(); 27860b57cec5SDimitry Andric 27870b57cec5SDimitry Andric // Find the first constant to emit, which is the first non-globalvalue value. 27880b57cec5SDimitry Andric // We know globalvalues have been emitted by WriteModuleInfo. 27890b57cec5SDimitry Andric for (unsigned i = 0, e = Vals.size(); i != e; ++i) { 27900b57cec5SDimitry Andric if (!isa<GlobalValue>(Vals[i].first)) { 27910b57cec5SDimitry Andric writeConstants(i, Vals.size(), true); 27920b57cec5SDimitry Andric return; 27930b57cec5SDimitry Andric } 27940b57cec5SDimitry Andric } 27950b57cec5SDimitry Andric } 27960b57cec5SDimitry Andric 27970b57cec5SDimitry Andric /// pushValueAndType - The file has to encode both the value and type id for 27980b57cec5SDimitry Andric /// many values, because we need to know what type to create for forward 27990b57cec5SDimitry Andric /// references. However, most operands are not forward references, so this type 28000b57cec5SDimitry Andric /// field is not needed. 28010b57cec5SDimitry Andric /// 28020b57cec5SDimitry Andric /// This function adds V's value ID to Vals. If the value ID is higher than the 28030b57cec5SDimitry Andric /// instruction ID, then it is a forward reference, and it also includes the 28040b57cec5SDimitry Andric /// type ID. The value ID that is written is encoded relative to the InstID. 28050b57cec5SDimitry Andric bool ModuleBitcodeWriter::pushValueAndType(const Value *V, unsigned InstID, 28060b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals) { 28070b57cec5SDimitry Andric unsigned ValID = VE.getValueID(V); 28080b57cec5SDimitry Andric // Make encoding relative to the InstID. 28090b57cec5SDimitry Andric Vals.push_back(InstID - ValID); 28100b57cec5SDimitry Andric if (ValID >= InstID) { 28110b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(V->getType())); 28120b57cec5SDimitry Andric return true; 28130b57cec5SDimitry Andric } 28140b57cec5SDimitry Andric return false; 28150b57cec5SDimitry Andric } 28160b57cec5SDimitry Andric 28175ffd83dbSDimitry Andric void ModuleBitcodeWriter::writeOperandBundles(const CallBase &CS, 28180b57cec5SDimitry Andric unsigned InstID) { 28190b57cec5SDimitry Andric SmallVector<unsigned, 64> Record; 28205ffd83dbSDimitry Andric LLVMContext &C = CS.getContext(); 28210b57cec5SDimitry Andric 28220b57cec5SDimitry Andric for (unsigned i = 0, e = CS.getNumOperandBundles(); i != e; ++i) { 28230b57cec5SDimitry Andric const auto &Bundle = CS.getOperandBundleAt(i); 28240b57cec5SDimitry Andric Record.push_back(C.getOperandBundleTagID(Bundle.getTagName())); 28250b57cec5SDimitry Andric 28260b57cec5SDimitry Andric for (auto &Input : Bundle.Inputs) 28270b57cec5SDimitry Andric pushValueAndType(Input, InstID, Record); 28280b57cec5SDimitry Andric 28290b57cec5SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_OPERAND_BUNDLE, Record); 28300b57cec5SDimitry Andric Record.clear(); 28310b57cec5SDimitry Andric } 28320b57cec5SDimitry Andric } 28330b57cec5SDimitry Andric 28340b57cec5SDimitry Andric /// pushValue - Like pushValueAndType, but where the type of the value is 28350b57cec5SDimitry Andric /// omitted (perhaps it was already encoded in an earlier operand). 28360b57cec5SDimitry Andric void ModuleBitcodeWriter::pushValue(const Value *V, unsigned InstID, 28370b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals) { 28380b57cec5SDimitry Andric unsigned ValID = VE.getValueID(V); 28390b57cec5SDimitry Andric Vals.push_back(InstID - ValID); 28400b57cec5SDimitry Andric } 28410b57cec5SDimitry Andric 28420b57cec5SDimitry Andric void ModuleBitcodeWriter::pushValueSigned(const Value *V, unsigned InstID, 28430b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Vals) { 28440b57cec5SDimitry Andric unsigned ValID = VE.getValueID(V); 28450b57cec5SDimitry Andric int64_t diff = ((int32_t)InstID - (int32_t)ValID); 28460b57cec5SDimitry Andric emitSignedInt64(Vals, diff); 28470b57cec5SDimitry Andric } 28480b57cec5SDimitry Andric 28490b57cec5SDimitry Andric /// WriteInstruction - Emit an instruction to the specified stream. 28500b57cec5SDimitry Andric void ModuleBitcodeWriter::writeInstruction(const Instruction &I, 28510b57cec5SDimitry Andric unsigned InstID, 28520b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals) { 28530b57cec5SDimitry Andric unsigned Code = 0; 28540b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 28550b57cec5SDimitry Andric VE.setInstructionID(&I); 28560b57cec5SDimitry Andric switch (I.getOpcode()) { 28570b57cec5SDimitry Andric default: 28580b57cec5SDimitry Andric if (Instruction::isCast(I.getOpcode())) { 28590b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CAST; 28600b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) 28610b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_CAST_ABBREV; 28620b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getType())); 28630b57cec5SDimitry Andric Vals.push_back(getEncodedCastOpcode(I.getOpcode())); 2864*c9157d92SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 2865*c9157d92SDimitry Andric if (Flags != 0) { 2866*c9157d92SDimitry Andric if (AbbrevToUse == FUNCTION_INST_CAST_ABBREV) 2867*c9157d92SDimitry Andric AbbrevToUse = FUNCTION_INST_CAST_FLAGS_ABBREV; 2868*c9157d92SDimitry Andric Vals.push_back(Flags); 2869*c9157d92SDimitry Andric } 28700b57cec5SDimitry Andric } else { 28710b57cec5SDimitry Andric assert(isa<BinaryOperator>(I) && "Unknown instruction!"); 28720b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_BINOP; 28730b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) 28740b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_BINOP_ABBREV; 28750b57cec5SDimitry Andric pushValue(I.getOperand(1), InstID, Vals); 28760b57cec5SDimitry Andric Vals.push_back(getEncodedBinaryOpcode(I.getOpcode())); 28770b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 28780b57cec5SDimitry Andric if (Flags != 0) { 28790b57cec5SDimitry Andric if (AbbrevToUse == FUNCTION_INST_BINOP_ABBREV) 28800b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_BINOP_FLAGS_ABBREV; 28810b57cec5SDimitry Andric Vals.push_back(Flags); 28820b57cec5SDimitry Andric } 28830b57cec5SDimitry Andric } 28840b57cec5SDimitry Andric break; 28850b57cec5SDimitry Andric case Instruction::FNeg: { 28860b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_UNOP; 28870b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) 28880b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_UNOP_ABBREV; 28890b57cec5SDimitry Andric Vals.push_back(getEncodedUnaryOpcode(I.getOpcode())); 28900b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 28910b57cec5SDimitry Andric if (Flags != 0) { 28920b57cec5SDimitry Andric if (AbbrevToUse == FUNCTION_INST_UNOP_ABBREV) 28930b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_UNOP_FLAGS_ABBREV; 28940b57cec5SDimitry Andric Vals.push_back(Flags); 28950b57cec5SDimitry Andric } 28960b57cec5SDimitry Andric break; 28970b57cec5SDimitry Andric } 28980b57cec5SDimitry Andric case Instruction::GetElementPtr: { 28990b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_GEP; 29000b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_GEP_ABBREV; 29010b57cec5SDimitry Andric auto &GEPInst = cast<GetElementPtrInst>(I); 29020b57cec5SDimitry Andric Vals.push_back(GEPInst.isInBounds()); 29030b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(GEPInst.getSourceElementType())); 29040b57cec5SDimitry Andric for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i) 29050b57cec5SDimitry Andric pushValueAndType(I.getOperand(i), InstID, Vals); 29060b57cec5SDimitry Andric break; 29070b57cec5SDimitry Andric } 29080b57cec5SDimitry Andric case Instruction::ExtractValue: { 29090b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_EXTRACTVAL; 29100b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 29110b57cec5SDimitry Andric const ExtractValueInst *EVI = cast<ExtractValueInst>(&I); 29120b57cec5SDimitry Andric Vals.append(EVI->idx_begin(), EVI->idx_end()); 29130b57cec5SDimitry Andric break; 29140b57cec5SDimitry Andric } 29150b57cec5SDimitry Andric case Instruction::InsertValue: { 29160b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_INSERTVAL; 29170b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 29180b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); 29190b57cec5SDimitry Andric const InsertValueInst *IVI = cast<InsertValueInst>(&I); 29200b57cec5SDimitry Andric Vals.append(IVI->idx_begin(), IVI->idx_end()); 29210b57cec5SDimitry Andric break; 29220b57cec5SDimitry Andric } 29230b57cec5SDimitry Andric case Instruction::Select: { 29240b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_VSELECT; 29250b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); 29260b57cec5SDimitry Andric pushValue(I.getOperand(2), InstID, Vals); 29270b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 29280b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 29290b57cec5SDimitry Andric if (Flags != 0) 29300b57cec5SDimitry Andric Vals.push_back(Flags); 29310b57cec5SDimitry Andric break; 29320b57cec5SDimitry Andric } 29330b57cec5SDimitry Andric case Instruction::ExtractElement: 29340b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_EXTRACTELT; 29350b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 29360b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); 29370b57cec5SDimitry Andric break; 29380b57cec5SDimitry Andric case Instruction::InsertElement: 29390b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_INSERTELT; 29400b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 29410b57cec5SDimitry Andric pushValue(I.getOperand(1), InstID, Vals); 29420b57cec5SDimitry Andric pushValueAndType(I.getOperand(2), InstID, Vals); 29430b57cec5SDimitry Andric break; 29440b57cec5SDimitry Andric case Instruction::ShuffleVector: 29450b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_SHUFFLEVEC; 29460b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 29470b57cec5SDimitry Andric pushValue(I.getOperand(1), InstID, Vals); 29485ffd83dbSDimitry Andric pushValue(cast<ShuffleVectorInst>(I).getShuffleMaskForBitcode(), InstID, 29495ffd83dbSDimitry Andric Vals); 29500b57cec5SDimitry Andric break; 29510b57cec5SDimitry Andric case Instruction::ICmp: 29520b57cec5SDimitry Andric case Instruction::FCmp: { 29530b57cec5SDimitry Andric // compare returning Int1Ty or vector of Int1Ty 29540b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CMP2; 29550b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 29560b57cec5SDimitry Andric pushValue(I.getOperand(1), InstID, Vals); 29570b57cec5SDimitry Andric Vals.push_back(cast<CmpInst>(I).getPredicate()); 29580b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 29590b57cec5SDimitry Andric if (Flags != 0) 29600b57cec5SDimitry Andric Vals.push_back(Flags); 29610b57cec5SDimitry Andric break; 29620b57cec5SDimitry Andric } 29630b57cec5SDimitry Andric 29640b57cec5SDimitry Andric case Instruction::Ret: 29650b57cec5SDimitry Andric { 29660b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_RET; 29670b57cec5SDimitry Andric unsigned NumOperands = I.getNumOperands(); 29680b57cec5SDimitry Andric if (NumOperands == 0) 29690b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_RET_VOID_ABBREV; 29700b57cec5SDimitry Andric else if (NumOperands == 1) { 29710b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) 29720b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_RET_VAL_ABBREV; 29730b57cec5SDimitry Andric } else { 29740b57cec5SDimitry Andric for (unsigned i = 0, e = NumOperands; i != e; ++i) 29750b57cec5SDimitry Andric pushValueAndType(I.getOperand(i), InstID, Vals); 29760b57cec5SDimitry Andric } 29770b57cec5SDimitry Andric } 29780b57cec5SDimitry Andric break; 29790b57cec5SDimitry Andric case Instruction::Br: 29800b57cec5SDimitry Andric { 29810b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_BR; 29820b57cec5SDimitry Andric const BranchInst &II = cast<BranchInst>(I); 29830b57cec5SDimitry Andric Vals.push_back(VE.getValueID(II.getSuccessor(0))); 29840b57cec5SDimitry Andric if (II.isConditional()) { 29850b57cec5SDimitry Andric Vals.push_back(VE.getValueID(II.getSuccessor(1))); 29860b57cec5SDimitry Andric pushValue(II.getCondition(), InstID, Vals); 29870b57cec5SDimitry Andric } 29880b57cec5SDimitry Andric } 29890b57cec5SDimitry Andric break; 29900b57cec5SDimitry Andric case Instruction::Switch: 29910b57cec5SDimitry Andric { 29920b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_SWITCH; 29930b57cec5SDimitry Andric const SwitchInst &SI = cast<SwitchInst>(I); 29940b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(SI.getCondition()->getType())); 29950b57cec5SDimitry Andric pushValue(SI.getCondition(), InstID, Vals); 29960b57cec5SDimitry Andric Vals.push_back(VE.getValueID(SI.getDefaultDest())); 29970b57cec5SDimitry Andric for (auto Case : SI.cases()) { 29980b57cec5SDimitry Andric Vals.push_back(VE.getValueID(Case.getCaseValue())); 29990b57cec5SDimitry Andric Vals.push_back(VE.getValueID(Case.getCaseSuccessor())); 30000b57cec5SDimitry Andric } 30010b57cec5SDimitry Andric } 30020b57cec5SDimitry Andric break; 30030b57cec5SDimitry Andric case Instruction::IndirectBr: 30040b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_INDIRECTBR; 30050b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); 30060b57cec5SDimitry Andric // Encode the address operand as relative, but not the basic blocks. 30070b57cec5SDimitry Andric pushValue(I.getOperand(0), InstID, Vals); 30080b57cec5SDimitry Andric for (unsigned i = 1, e = I.getNumOperands(); i != e; ++i) 30090b57cec5SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(i))); 30100b57cec5SDimitry Andric break; 30110b57cec5SDimitry Andric 30120b57cec5SDimitry Andric case Instruction::Invoke: { 30130b57cec5SDimitry Andric const InvokeInst *II = cast<InvokeInst>(&I); 30145ffd83dbSDimitry Andric const Value *Callee = II->getCalledOperand(); 30150b57cec5SDimitry Andric FunctionType *FTy = II->getFunctionType(); 30160b57cec5SDimitry Andric 30170b57cec5SDimitry Andric if (II->hasOperandBundles()) 30185ffd83dbSDimitry Andric writeOperandBundles(*II, InstID); 30190b57cec5SDimitry Andric 30200b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_INVOKE; 30210b57cec5SDimitry Andric 30220b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(II->getAttributes())); 30230b57cec5SDimitry Andric Vals.push_back(II->getCallingConv() | 1 << 13); 30240b57cec5SDimitry Andric Vals.push_back(VE.getValueID(II->getNormalDest())); 30250b57cec5SDimitry Andric Vals.push_back(VE.getValueID(II->getUnwindDest())); 30260b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(FTy)); 30270b57cec5SDimitry Andric pushValueAndType(Callee, InstID, Vals); 30280b57cec5SDimitry Andric 30290b57cec5SDimitry Andric // Emit value #'s for the fixed parameters. 30300b57cec5SDimitry Andric for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) 30310b57cec5SDimitry Andric pushValue(I.getOperand(i), InstID, Vals); // fixed param. 30320b57cec5SDimitry Andric 30330b57cec5SDimitry Andric // Emit type/value pairs for varargs params. 30340b57cec5SDimitry Andric if (FTy->isVarArg()) { 3035349cc55cSDimitry Andric for (unsigned i = FTy->getNumParams(), e = II->arg_size(); i != e; ++i) 30360b57cec5SDimitry Andric pushValueAndType(I.getOperand(i), InstID, Vals); // vararg 30370b57cec5SDimitry Andric } 30380b57cec5SDimitry Andric break; 30390b57cec5SDimitry Andric } 30400b57cec5SDimitry Andric case Instruction::Resume: 30410b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_RESUME; 30420b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 30430b57cec5SDimitry Andric break; 30440b57cec5SDimitry Andric case Instruction::CleanupRet: { 30450b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CLEANUPRET; 30460b57cec5SDimitry Andric const auto &CRI = cast<CleanupReturnInst>(I); 30470b57cec5SDimitry Andric pushValue(CRI.getCleanupPad(), InstID, Vals); 30480b57cec5SDimitry Andric if (CRI.hasUnwindDest()) 30490b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CRI.getUnwindDest())); 30500b57cec5SDimitry Andric break; 30510b57cec5SDimitry Andric } 30520b57cec5SDimitry Andric case Instruction::CatchRet: { 30530b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CATCHRET; 30540b57cec5SDimitry Andric const auto &CRI = cast<CatchReturnInst>(I); 30550b57cec5SDimitry Andric pushValue(CRI.getCatchPad(), InstID, Vals); 30560b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CRI.getSuccessor())); 30570b57cec5SDimitry Andric break; 30580b57cec5SDimitry Andric } 30590b57cec5SDimitry Andric case Instruction::CleanupPad: 30600b57cec5SDimitry Andric case Instruction::CatchPad: { 30610b57cec5SDimitry Andric const auto &FuncletPad = cast<FuncletPadInst>(I); 30620b57cec5SDimitry Andric Code = isa<CatchPadInst>(FuncletPad) ? bitc::FUNC_CODE_INST_CATCHPAD 30630b57cec5SDimitry Andric : bitc::FUNC_CODE_INST_CLEANUPPAD; 30640b57cec5SDimitry Andric pushValue(FuncletPad.getParentPad(), InstID, Vals); 30650b57cec5SDimitry Andric 3066bdd1243dSDimitry Andric unsigned NumArgOperands = FuncletPad.arg_size(); 30670b57cec5SDimitry Andric Vals.push_back(NumArgOperands); 30680b57cec5SDimitry Andric for (unsigned Op = 0; Op != NumArgOperands; ++Op) 30690b57cec5SDimitry Andric pushValueAndType(FuncletPad.getArgOperand(Op), InstID, Vals); 30700b57cec5SDimitry Andric break; 30710b57cec5SDimitry Andric } 30720b57cec5SDimitry Andric case Instruction::CatchSwitch: { 30730b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CATCHSWITCH; 30740b57cec5SDimitry Andric const auto &CatchSwitch = cast<CatchSwitchInst>(I); 30750b57cec5SDimitry Andric 30760b57cec5SDimitry Andric pushValue(CatchSwitch.getParentPad(), InstID, Vals); 30770b57cec5SDimitry Andric 30780b57cec5SDimitry Andric unsigned NumHandlers = CatchSwitch.getNumHandlers(); 30790b57cec5SDimitry Andric Vals.push_back(NumHandlers); 30800b57cec5SDimitry Andric for (const BasicBlock *CatchPadBB : CatchSwitch.handlers()) 30810b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CatchPadBB)); 30820b57cec5SDimitry Andric 30830b57cec5SDimitry Andric if (CatchSwitch.hasUnwindDest()) 30840b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CatchSwitch.getUnwindDest())); 30850b57cec5SDimitry Andric break; 30860b57cec5SDimitry Andric } 30870b57cec5SDimitry Andric case Instruction::CallBr: { 30880b57cec5SDimitry Andric const CallBrInst *CBI = cast<CallBrInst>(&I); 30895ffd83dbSDimitry Andric const Value *Callee = CBI->getCalledOperand(); 30900b57cec5SDimitry Andric FunctionType *FTy = CBI->getFunctionType(); 30910b57cec5SDimitry Andric 30920b57cec5SDimitry Andric if (CBI->hasOperandBundles()) 30935ffd83dbSDimitry Andric writeOperandBundles(*CBI, InstID); 30940b57cec5SDimitry Andric 30950b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CALLBR; 30960b57cec5SDimitry Andric 30970b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(CBI->getAttributes())); 30980b57cec5SDimitry Andric 30990b57cec5SDimitry Andric Vals.push_back(CBI->getCallingConv() << bitc::CALL_CCONV | 31000b57cec5SDimitry Andric 1 << bitc::CALL_EXPLICIT_TYPE); 31010b57cec5SDimitry Andric 31020b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CBI->getDefaultDest())); 31030b57cec5SDimitry Andric Vals.push_back(CBI->getNumIndirectDests()); 31040b57cec5SDimitry Andric for (unsigned i = 0, e = CBI->getNumIndirectDests(); i != e; ++i) 31050b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CBI->getIndirectDest(i))); 31060b57cec5SDimitry Andric 31070b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(FTy)); 31080b57cec5SDimitry Andric pushValueAndType(Callee, InstID, Vals); 31090b57cec5SDimitry Andric 31100b57cec5SDimitry Andric // Emit value #'s for the fixed parameters. 31110b57cec5SDimitry Andric for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) 31120b57cec5SDimitry Andric pushValue(I.getOperand(i), InstID, Vals); // fixed param. 31130b57cec5SDimitry Andric 31140b57cec5SDimitry Andric // Emit type/value pairs for varargs params. 31150b57cec5SDimitry Andric if (FTy->isVarArg()) { 3116349cc55cSDimitry Andric for (unsigned i = FTy->getNumParams(), e = CBI->arg_size(); i != e; ++i) 31170b57cec5SDimitry Andric pushValueAndType(I.getOperand(i), InstID, Vals); // vararg 31180b57cec5SDimitry Andric } 31190b57cec5SDimitry Andric break; 31200b57cec5SDimitry Andric } 31210b57cec5SDimitry Andric case Instruction::Unreachable: 31220b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_UNREACHABLE; 31230b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_UNREACHABLE_ABBREV; 31240b57cec5SDimitry Andric break; 31250b57cec5SDimitry Andric 31260b57cec5SDimitry Andric case Instruction::PHI: { 31270b57cec5SDimitry Andric const PHINode &PN = cast<PHINode>(I); 31280b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_PHI; 31290b57cec5SDimitry Andric // With the newer instruction encoding, forward references could give 31300b57cec5SDimitry Andric // negative valued IDs. This is most common for PHIs, so we use 31310b57cec5SDimitry Andric // signed VBRs. 31320b57cec5SDimitry Andric SmallVector<uint64_t, 128> Vals64; 31330b57cec5SDimitry Andric Vals64.push_back(VE.getTypeID(PN.getType())); 31340b57cec5SDimitry Andric for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) { 31350b57cec5SDimitry Andric pushValueSigned(PN.getIncomingValue(i), InstID, Vals64); 31360b57cec5SDimitry Andric Vals64.push_back(VE.getValueID(PN.getIncomingBlock(i))); 31370b57cec5SDimitry Andric } 31388bcb0991SDimitry Andric 31398bcb0991SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 31408bcb0991SDimitry Andric if (Flags != 0) 31418bcb0991SDimitry Andric Vals64.push_back(Flags); 31428bcb0991SDimitry Andric 31430b57cec5SDimitry Andric // Emit a Vals64 vector and exit. 31440b57cec5SDimitry Andric Stream.EmitRecord(Code, Vals64, AbbrevToUse); 31450b57cec5SDimitry Andric Vals64.clear(); 31460b57cec5SDimitry Andric return; 31470b57cec5SDimitry Andric } 31480b57cec5SDimitry Andric 31490b57cec5SDimitry Andric case Instruction::LandingPad: { 31500b57cec5SDimitry Andric const LandingPadInst &LP = cast<LandingPadInst>(I); 31510b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_LANDINGPAD; 31520b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(LP.getType())); 31530b57cec5SDimitry Andric Vals.push_back(LP.isCleanup()); 31540b57cec5SDimitry Andric Vals.push_back(LP.getNumClauses()); 31550b57cec5SDimitry Andric for (unsigned I = 0, E = LP.getNumClauses(); I != E; ++I) { 31560b57cec5SDimitry Andric if (LP.isCatch(I)) 31570b57cec5SDimitry Andric Vals.push_back(LandingPadInst::Catch); 31580b57cec5SDimitry Andric else 31590b57cec5SDimitry Andric Vals.push_back(LandingPadInst::Filter); 31600b57cec5SDimitry Andric pushValueAndType(LP.getClause(I), InstID, Vals); 31610b57cec5SDimitry Andric } 31620b57cec5SDimitry Andric break; 31630b57cec5SDimitry Andric } 31640b57cec5SDimitry Andric 31650b57cec5SDimitry Andric case Instruction::Alloca: { 31660b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_ALLOCA; 31670b57cec5SDimitry Andric const AllocaInst &AI = cast<AllocaInst>(I); 31680b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(AI.getAllocatedType())); 31690b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); 31700b57cec5SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(0))); // size. 3171e8d8bef9SDimitry Andric using APV = AllocaPackedValues; 3172e8d8bef9SDimitry Andric unsigned Record = 0; 3173349cc55cSDimitry Andric unsigned EncodedAlign = getEncodedAlign(AI.getAlign()); 3174349cc55cSDimitry Andric Bitfield::set<APV::AlignLower>( 3175349cc55cSDimitry Andric Record, EncodedAlign & ((1 << APV::AlignLower::Bits) - 1)); 3176349cc55cSDimitry Andric Bitfield::set<APV::AlignUpper>(Record, 3177349cc55cSDimitry Andric EncodedAlign >> APV::AlignLower::Bits); 3178e8d8bef9SDimitry Andric Bitfield::set<APV::UsedWithInAlloca>(Record, AI.isUsedWithInAlloca()); 3179e8d8bef9SDimitry Andric Bitfield::set<APV::ExplicitType>(Record, true); 3180e8d8bef9SDimitry Andric Bitfield::set<APV::SwiftError>(Record, AI.isSwiftError()); 3181e8d8bef9SDimitry Andric Vals.push_back(Record); 318281ad6265SDimitry Andric 318381ad6265SDimitry Andric unsigned AS = AI.getAddressSpace(); 318481ad6265SDimitry Andric if (AS != M.getDataLayout().getAllocaAddrSpace()) 318581ad6265SDimitry Andric Vals.push_back(AS); 31860b57cec5SDimitry Andric break; 31870b57cec5SDimitry Andric } 31880b57cec5SDimitry Andric 31890b57cec5SDimitry Andric case Instruction::Load: 31900b57cec5SDimitry Andric if (cast<LoadInst>(I).isAtomic()) { 31910b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_LOADATOMIC; 31920b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 31930b57cec5SDimitry Andric } else { 31940b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_LOAD; 31950b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) // ptr 31960b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_LOAD_ABBREV; 31970b57cec5SDimitry Andric } 31980b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getType())); 3199e8d8bef9SDimitry Andric Vals.push_back(getEncodedAlign(cast<LoadInst>(I).getAlign())); 32000b57cec5SDimitry Andric Vals.push_back(cast<LoadInst>(I).isVolatile()); 32010b57cec5SDimitry Andric if (cast<LoadInst>(I).isAtomic()) { 32020b57cec5SDimitry Andric Vals.push_back(getEncodedOrdering(cast<LoadInst>(I).getOrdering())); 32030b57cec5SDimitry Andric Vals.push_back(getEncodedSyncScopeID(cast<LoadInst>(I).getSyncScopeID())); 32040b57cec5SDimitry Andric } 32050b57cec5SDimitry Andric break; 32060b57cec5SDimitry Andric case Instruction::Store: 32070b57cec5SDimitry Andric if (cast<StoreInst>(I).isAtomic()) 32080b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_STOREATOMIC; 32090b57cec5SDimitry Andric else 32100b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_STORE; 32110b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); // ptrty + ptr 32120b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); // valty + val 3213e8d8bef9SDimitry Andric Vals.push_back(getEncodedAlign(cast<StoreInst>(I).getAlign())); 32140b57cec5SDimitry Andric Vals.push_back(cast<StoreInst>(I).isVolatile()); 32150b57cec5SDimitry Andric if (cast<StoreInst>(I).isAtomic()) { 32160b57cec5SDimitry Andric Vals.push_back(getEncodedOrdering(cast<StoreInst>(I).getOrdering())); 32170b57cec5SDimitry Andric Vals.push_back( 32180b57cec5SDimitry Andric getEncodedSyncScopeID(cast<StoreInst>(I).getSyncScopeID())); 32190b57cec5SDimitry Andric } 32200b57cec5SDimitry Andric break; 32210b57cec5SDimitry Andric case Instruction::AtomicCmpXchg: 32220b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CMPXCHG; 32230b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); // ptrty + ptr 32240b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); // cmp. 32250b57cec5SDimitry Andric pushValue(I.getOperand(2), InstID, Vals); // newval. 32260b57cec5SDimitry Andric Vals.push_back(cast<AtomicCmpXchgInst>(I).isVolatile()); 32270b57cec5SDimitry Andric Vals.push_back( 32280b57cec5SDimitry Andric getEncodedOrdering(cast<AtomicCmpXchgInst>(I).getSuccessOrdering())); 32290b57cec5SDimitry Andric Vals.push_back( 32300b57cec5SDimitry Andric getEncodedSyncScopeID(cast<AtomicCmpXchgInst>(I).getSyncScopeID())); 32310b57cec5SDimitry Andric Vals.push_back( 32320b57cec5SDimitry Andric getEncodedOrdering(cast<AtomicCmpXchgInst>(I).getFailureOrdering())); 32330b57cec5SDimitry Andric Vals.push_back(cast<AtomicCmpXchgInst>(I).isWeak()); 3234fe6060f1SDimitry Andric Vals.push_back(getEncodedAlign(cast<AtomicCmpXchgInst>(I).getAlign())); 32350b57cec5SDimitry Andric break; 32360b57cec5SDimitry Andric case Instruction::AtomicRMW: 32370b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_ATOMICRMW; 32380b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); // ptrty + ptr 3239fe6060f1SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); // valty + val 32400b57cec5SDimitry Andric Vals.push_back( 32410b57cec5SDimitry Andric getEncodedRMWOperation(cast<AtomicRMWInst>(I).getOperation())); 32420b57cec5SDimitry Andric Vals.push_back(cast<AtomicRMWInst>(I).isVolatile()); 32430b57cec5SDimitry Andric Vals.push_back(getEncodedOrdering(cast<AtomicRMWInst>(I).getOrdering())); 32440b57cec5SDimitry Andric Vals.push_back( 32450b57cec5SDimitry Andric getEncodedSyncScopeID(cast<AtomicRMWInst>(I).getSyncScopeID())); 3246fe6060f1SDimitry Andric Vals.push_back(getEncodedAlign(cast<AtomicRMWInst>(I).getAlign())); 32470b57cec5SDimitry Andric break; 32480b57cec5SDimitry Andric case Instruction::Fence: 32490b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_FENCE; 32500b57cec5SDimitry Andric Vals.push_back(getEncodedOrdering(cast<FenceInst>(I).getOrdering())); 32510b57cec5SDimitry Andric Vals.push_back(getEncodedSyncScopeID(cast<FenceInst>(I).getSyncScopeID())); 32520b57cec5SDimitry Andric break; 32530b57cec5SDimitry Andric case Instruction::Call: { 32540b57cec5SDimitry Andric const CallInst &CI = cast<CallInst>(I); 32550b57cec5SDimitry Andric FunctionType *FTy = CI.getFunctionType(); 32560b57cec5SDimitry Andric 32570b57cec5SDimitry Andric if (CI.hasOperandBundles()) 32585ffd83dbSDimitry Andric writeOperandBundles(CI, InstID); 32590b57cec5SDimitry Andric 32600b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CALL; 32610b57cec5SDimitry Andric 32620b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(CI.getAttributes())); 32630b57cec5SDimitry Andric 32640b57cec5SDimitry Andric unsigned Flags = getOptimizationFlags(&I); 32650b57cec5SDimitry Andric Vals.push_back(CI.getCallingConv() << bitc::CALL_CCONV | 32660b57cec5SDimitry Andric unsigned(CI.isTailCall()) << bitc::CALL_TAIL | 32670b57cec5SDimitry Andric unsigned(CI.isMustTailCall()) << bitc::CALL_MUSTTAIL | 32680b57cec5SDimitry Andric 1 << bitc::CALL_EXPLICIT_TYPE | 32690b57cec5SDimitry Andric unsigned(CI.isNoTailCall()) << bitc::CALL_NOTAIL | 32700b57cec5SDimitry Andric unsigned(Flags != 0) << bitc::CALL_FMF); 32710b57cec5SDimitry Andric if (Flags != 0) 32720b57cec5SDimitry Andric Vals.push_back(Flags); 32730b57cec5SDimitry Andric 32740b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(FTy)); 32755ffd83dbSDimitry Andric pushValueAndType(CI.getCalledOperand(), InstID, Vals); // Callee 32760b57cec5SDimitry Andric 32770b57cec5SDimitry Andric // Emit value #'s for the fixed parameters. 32780b57cec5SDimitry Andric for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) { 32790b57cec5SDimitry Andric // Check for labels (can happen with asm labels). 32800b57cec5SDimitry Andric if (FTy->getParamType(i)->isLabelTy()) 32810b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CI.getArgOperand(i))); 32820b57cec5SDimitry Andric else 32830b57cec5SDimitry Andric pushValue(CI.getArgOperand(i), InstID, Vals); // fixed param. 32840b57cec5SDimitry Andric } 32850b57cec5SDimitry Andric 32860b57cec5SDimitry Andric // Emit type/value pairs for varargs params. 32870b57cec5SDimitry Andric if (FTy->isVarArg()) { 3288349cc55cSDimitry Andric for (unsigned i = FTy->getNumParams(), e = CI.arg_size(); i != e; ++i) 32890b57cec5SDimitry Andric pushValueAndType(CI.getArgOperand(i), InstID, Vals); // varargs 32900b57cec5SDimitry Andric } 32910b57cec5SDimitry Andric break; 32920b57cec5SDimitry Andric } 32930b57cec5SDimitry Andric case Instruction::VAArg: 32940b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_VAARG; 32950b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); // valistty 32960b57cec5SDimitry Andric pushValue(I.getOperand(0), InstID, Vals); // valist. 32970b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getType())); // restype. 32980b57cec5SDimitry Andric break; 3299480093f4SDimitry Andric case Instruction::Freeze: 3300480093f4SDimitry Andric Code = bitc::FUNC_CODE_INST_FREEZE; 3301480093f4SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 3302480093f4SDimitry Andric break; 33030b57cec5SDimitry Andric } 33040b57cec5SDimitry Andric 33050b57cec5SDimitry Andric Stream.EmitRecord(Code, Vals, AbbrevToUse); 33060b57cec5SDimitry Andric Vals.clear(); 33070b57cec5SDimitry Andric } 33080b57cec5SDimitry Andric 33090b57cec5SDimitry Andric /// Write a GlobalValue VST to the module. The purpose of this data structure is 33100b57cec5SDimitry Andric /// to allow clients to efficiently find the function body. 33110b57cec5SDimitry Andric void ModuleBitcodeWriter::writeGlobalValueSymbolTable( 33120b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> &FunctionToBitcodeIndex) { 33130b57cec5SDimitry Andric // Get the offset of the VST we are writing, and backpatch it into 33140b57cec5SDimitry Andric // the VST forward declaration record. 33150b57cec5SDimitry Andric uint64_t VSTOffset = Stream.GetCurrentBitNo(); 33160b57cec5SDimitry Andric // The BitcodeStartBit was the stream offset of the identification block. 33170b57cec5SDimitry Andric VSTOffset -= bitcodeStartBit(); 33180b57cec5SDimitry Andric assert((VSTOffset & 31) == 0 && "VST block not 32-bit aligned"); 33190b57cec5SDimitry Andric // Note that we add 1 here because the offset is relative to one word 33200b57cec5SDimitry Andric // before the start of the identification block, which was historically 33210b57cec5SDimitry Andric // always the start of the regular bitcode header. 33220b57cec5SDimitry Andric Stream.BackpatchWord(VSTOffsetPlaceholder, VSTOffset / 32 + 1); 33230b57cec5SDimitry Andric 33240b57cec5SDimitry Andric Stream.EnterSubblock(bitc::VALUE_SYMTAB_BLOCK_ID, 4); 33250b57cec5SDimitry Andric 33260b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 33270b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_FNENTRY)); 33280b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // value id 33290b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // funcoffset 33300b57cec5SDimitry Andric unsigned FnEntryAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 33310b57cec5SDimitry Andric 33320b57cec5SDimitry Andric for (const Function &F : M) { 33330b57cec5SDimitry Andric uint64_t Record[2]; 33340b57cec5SDimitry Andric 33350b57cec5SDimitry Andric if (F.isDeclaration()) 33360b57cec5SDimitry Andric continue; 33370b57cec5SDimitry Andric 33380b57cec5SDimitry Andric Record[0] = VE.getValueID(&F); 33390b57cec5SDimitry Andric 33400b57cec5SDimitry Andric // Save the word offset of the function (from the start of the 33410b57cec5SDimitry Andric // actual bitcode written to the stream). 33420b57cec5SDimitry Andric uint64_t BitcodeIndex = FunctionToBitcodeIndex[&F] - bitcodeStartBit(); 33430b57cec5SDimitry Andric assert((BitcodeIndex & 31) == 0 && "function block not 32-bit aligned"); 33440b57cec5SDimitry Andric // Note that we add 1 here because the offset is relative to one word 33450b57cec5SDimitry Andric // before the start of the identification block, which was historically 33460b57cec5SDimitry Andric // always the start of the regular bitcode header. 33470b57cec5SDimitry Andric Record[1] = BitcodeIndex / 32 + 1; 33480b57cec5SDimitry Andric 33490b57cec5SDimitry Andric Stream.EmitRecord(bitc::VST_CODE_FNENTRY, Record, FnEntryAbbrev); 33500b57cec5SDimitry Andric } 33510b57cec5SDimitry Andric 33520b57cec5SDimitry Andric Stream.ExitBlock(); 33530b57cec5SDimitry Andric } 33540b57cec5SDimitry Andric 33550b57cec5SDimitry Andric /// Emit names for arguments, instructions and basic blocks in a function. 33560b57cec5SDimitry Andric void ModuleBitcodeWriter::writeFunctionLevelValueSymbolTable( 33570b57cec5SDimitry Andric const ValueSymbolTable &VST) { 33580b57cec5SDimitry Andric if (VST.empty()) 33590b57cec5SDimitry Andric return; 33600b57cec5SDimitry Andric 33610b57cec5SDimitry Andric Stream.EnterSubblock(bitc::VALUE_SYMTAB_BLOCK_ID, 4); 33620b57cec5SDimitry Andric 33630b57cec5SDimitry Andric // FIXME: Set up the abbrev, we know how many values there are! 33640b57cec5SDimitry Andric // FIXME: We know if the type names can use 7-bit ascii. 33650b57cec5SDimitry Andric SmallVector<uint64_t, 64> NameVals; 33660b57cec5SDimitry Andric 33670b57cec5SDimitry Andric for (const ValueName &Name : VST) { 33680b57cec5SDimitry Andric // Figure out the encoding to use for the name. 33690b57cec5SDimitry Andric StringEncoding Bits = getStringEncoding(Name.getKey()); 33700b57cec5SDimitry Andric 33710b57cec5SDimitry Andric unsigned AbbrevToUse = VST_ENTRY_8_ABBREV; 33720b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(Name.getValue())); 33730b57cec5SDimitry Andric 33740b57cec5SDimitry Andric // VST_CODE_ENTRY: [valueid, namechar x N] 33750b57cec5SDimitry Andric // VST_CODE_BBENTRY: [bbid, namechar x N] 33760b57cec5SDimitry Andric unsigned Code; 33770b57cec5SDimitry Andric if (isa<BasicBlock>(Name.getValue())) { 33780b57cec5SDimitry Andric Code = bitc::VST_CODE_BBENTRY; 33790b57cec5SDimitry Andric if (Bits == SE_Char6) 33800b57cec5SDimitry Andric AbbrevToUse = VST_BBENTRY_6_ABBREV; 33810b57cec5SDimitry Andric } else { 33820b57cec5SDimitry Andric Code = bitc::VST_CODE_ENTRY; 33830b57cec5SDimitry Andric if (Bits == SE_Char6) 33840b57cec5SDimitry Andric AbbrevToUse = VST_ENTRY_6_ABBREV; 33850b57cec5SDimitry Andric else if (Bits == SE_Fixed7) 33860b57cec5SDimitry Andric AbbrevToUse = VST_ENTRY_7_ABBREV; 33870b57cec5SDimitry Andric } 33880b57cec5SDimitry Andric 33890b57cec5SDimitry Andric for (const auto P : Name.getKey()) 33900b57cec5SDimitry Andric NameVals.push_back((unsigned char)P); 33910b57cec5SDimitry Andric 33920b57cec5SDimitry Andric // Emit the finished record. 33930b57cec5SDimitry Andric Stream.EmitRecord(Code, NameVals, AbbrevToUse); 33940b57cec5SDimitry Andric NameVals.clear(); 33950b57cec5SDimitry Andric } 33960b57cec5SDimitry Andric 33970b57cec5SDimitry Andric Stream.ExitBlock(); 33980b57cec5SDimitry Andric } 33990b57cec5SDimitry Andric 34000b57cec5SDimitry Andric void ModuleBitcodeWriter::writeUseList(UseListOrder &&Order) { 34010b57cec5SDimitry Andric assert(Order.Shuffle.size() >= 2 && "Shuffle too small"); 34020b57cec5SDimitry Andric unsigned Code; 34030b57cec5SDimitry Andric if (isa<BasicBlock>(Order.V)) 34040b57cec5SDimitry Andric Code = bitc::USELIST_CODE_BB; 34050b57cec5SDimitry Andric else 34060b57cec5SDimitry Andric Code = bitc::USELIST_CODE_DEFAULT; 34070b57cec5SDimitry Andric 34080b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record(Order.Shuffle.begin(), Order.Shuffle.end()); 34090b57cec5SDimitry Andric Record.push_back(VE.getValueID(Order.V)); 34100b57cec5SDimitry Andric Stream.EmitRecord(Code, Record); 34110b57cec5SDimitry Andric } 34120b57cec5SDimitry Andric 34130b57cec5SDimitry Andric void ModuleBitcodeWriter::writeUseListBlock(const Function *F) { 34140b57cec5SDimitry Andric assert(VE.shouldPreserveUseListOrder() && 34150b57cec5SDimitry Andric "Expected to be preserving use-list order"); 34160b57cec5SDimitry Andric 34170b57cec5SDimitry Andric auto hasMore = [&]() { 34180b57cec5SDimitry Andric return !VE.UseListOrders.empty() && VE.UseListOrders.back().F == F; 34190b57cec5SDimitry Andric }; 34200b57cec5SDimitry Andric if (!hasMore()) 34210b57cec5SDimitry Andric // Nothing to do. 34220b57cec5SDimitry Andric return; 34230b57cec5SDimitry Andric 34240b57cec5SDimitry Andric Stream.EnterSubblock(bitc::USELIST_BLOCK_ID, 3); 34250b57cec5SDimitry Andric while (hasMore()) { 34260b57cec5SDimitry Andric writeUseList(std::move(VE.UseListOrders.back())); 34270b57cec5SDimitry Andric VE.UseListOrders.pop_back(); 34280b57cec5SDimitry Andric } 34290b57cec5SDimitry Andric Stream.ExitBlock(); 34300b57cec5SDimitry Andric } 34310b57cec5SDimitry Andric 34320b57cec5SDimitry Andric /// Emit a function body to the module stream. 34330b57cec5SDimitry Andric void ModuleBitcodeWriter::writeFunction( 34340b57cec5SDimitry Andric const Function &F, 34350b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> &FunctionToBitcodeIndex) { 34360b57cec5SDimitry Andric // Save the bitcode index of the start of this function block for recording 34370b57cec5SDimitry Andric // in the VST. 34380b57cec5SDimitry Andric FunctionToBitcodeIndex[&F] = Stream.GetCurrentBitNo(); 34390b57cec5SDimitry Andric 34400b57cec5SDimitry Andric Stream.EnterSubblock(bitc::FUNCTION_BLOCK_ID, 4); 34410b57cec5SDimitry Andric VE.incorporateFunction(F); 34420b57cec5SDimitry Andric 34430b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 34440b57cec5SDimitry Andric 34450b57cec5SDimitry Andric // Emit the number of basic blocks, so the reader can create them ahead of 34460b57cec5SDimitry Andric // time. 34470b57cec5SDimitry Andric Vals.push_back(VE.getBasicBlocks().size()); 34480b57cec5SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_DECLAREBLOCKS, Vals); 34490b57cec5SDimitry Andric Vals.clear(); 34500b57cec5SDimitry Andric 34510b57cec5SDimitry Andric // If there are function-local constants, emit them now. 34520b57cec5SDimitry Andric unsigned CstStart, CstEnd; 34530b57cec5SDimitry Andric VE.getFunctionConstantRange(CstStart, CstEnd); 34540b57cec5SDimitry Andric writeConstants(CstStart, CstEnd, false); 34550b57cec5SDimitry Andric 34560b57cec5SDimitry Andric // If there is function-local metadata, emit it now. 34570b57cec5SDimitry Andric writeFunctionMetadata(F); 34580b57cec5SDimitry Andric 34590b57cec5SDimitry Andric // Keep a running idea of what the instruction ID is. 34600b57cec5SDimitry Andric unsigned InstID = CstEnd; 34610b57cec5SDimitry Andric 34620b57cec5SDimitry Andric bool NeedsMetadataAttachment = F.hasMetadata(); 34630b57cec5SDimitry Andric 34640b57cec5SDimitry Andric DILocation *LastDL = nullptr; 346581ad6265SDimitry Andric SmallSetVector<Function *, 4> BlockAddressUsers; 346681ad6265SDimitry Andric 34670b57cec5SDimitry Andric // Finally, emit all the instructions, in order. 346881ad6265SDimitry Andric for (const BasicBlock &BB : F) { 34694824e7fdSDimitry Andric for (const Instruction &I : BB) { 34704824e7fdSDimitry Andric writeInstruction(I, InstID, Vals); 34710b57cec5SDimitry Andric 34724824e7fdSDimitry Andric if (!I.getType()->isVoidTy()) 34730b57cec5SDimitry Andric ++InstID; 34740b57cec5SDimitry Andric 34750b57cec5SDimitry Andric // If the instruction has metadata, write a metadata attachment later. 34764824e7fdSDimitry Andric NeedsMetadataAttachment |= I.hasMetadataOtherThanDebugLoc(); 34770b57cec5SDimitry Andric 34780b57cec5SDimitry Andric // If the instruction has a debug location, emit it. 34794824e7fdSDimitry Andric DILocation *DL = I.getDebugLoc(); 34800b57cec5SDimitry Andric if (!DL) 34810b57cec5SDimitry Andric continue; 34820b57cec5SDimitry Andric 34830b57cec5SDimitry Andric if (DL == LastDL) { 34840b57cec5SDimitry Andric // Just repeat the same debug loc as last time. 34850b57cec5SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_DEBUG_LOC_AGAIN, Vals); 34860b57cec5SDimitry Andric continue; 34870b57cec5SDimitry Andric } 34880b57cec5SDimitry Andric 34890b57cec5SDimitry Andric Vals.push_back(DL->getLine()); 34900b57cec5SDimitry Andric Vals.push_back(DL->getColumn()); 34910b57cec5SDimitry Andric Vals.push_back(VE.getMetadataOrNullID(DL->getScope())); 34920b57cec5SDimitry Andric Vals.push_back(VE.getMetadataOrNullID(DL->getInlinedAt())); 34930b57cec5SDimitry Andric Vals.push_back(DL->isImplicitCode()); 34940b57cec5SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_DEBUG_LOC, Vals); 34950b57cec5SDimitry Andric Vals.clear(); 34960b57cec5SDimitry Andric 34970b57cec5SDimitry Andric LastDL = DL; 34980b57cec5SDimitry Andric } 34990b57cec5SDimitry Andric 350081ad6265SDimitry Andric if (BlockAddress *BA = BlockAddress::lookup(&BB)) { 350181ad6265SDimitry Andric SmallVector<Value *> Worklist{BA}; 350281ad6265SDimitry Andric SmallPtrSet<Value *, 8> Visited{BA}; 350381ad6265SDimitry Andric while (!Worklist.empty()) { 350481ad6265SDimitry Andric Value *V = Worklist.pop_back_val(); 350581ad6265SDimitry Andric for (User *U : V->users()) { 350681ad6265SDimitry Andric if (auto *I = dyn_cast<Instruction>(U)) { 350781ad6265SDimitry Andric Function *P = I->getFunction(); 350881ad6265SDimitry Andric if (P != &F) 350981ad6265SDimitry Andric BlockAddressUsers.insert(P); 351081ad6265SDimitry Andric } else if (isa<Constant>(U) && !isa<GlobalValue>(U) && 351181ad6265SDimitry Andric Visited.insert(U).second) 351281ad6265SDimitry Andric Worklist.push_back(U); 351381ad6265SDimitry Andric } 351481ad6265SDimitry Andric } 351581ad6265SDimitry Andric } 351681ad6265SDimitry Andric } 351781ad6265SDimitry Andric 351881ad6265SDimitry Andric if (!BlockAddressUsers.empty()) { 351981ad6265SDimitry Andric Vals.resize(BlockAddressUsers.size()); 352081ad6265SDimitry Andric for (auto I : llvm::enumerate(BlockAddressUsers)) 352181ad6265SDimitry Andric Vals[I.index()] = VE.getValueID(I.value()); 352281ad6265SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_BLOCKADDR_USERS, Vals); 352381ad6265SDimitry Andric Vals.clear(); 352481ad6265SDimitry Andric } 352581ad6265SDimitry Andric 35260b57cec5SDimitry Andric // Emit names for all the instructions etc. 35270b57cec5SDimitry Andric if (auto *Symtab = F.getValueSymbolTable()) 35280b57cec5SDimitry Andric writeFunctionLevelValueSymbolTable(*Symtab); 35290b57cec5SDimitry Andric 35300b57cec5SDimitry Andric if (NeedsMetadataAttachment) 35310b57cec5SDimitry Andric writeFunctionMetadataAttachment(F); 35320b57cec5SDimitry Andric if (VE.shouldPreserveUseListOrder()) 35330b57cec5SDimitry Andric writeUseListBlock(&F); 35340b57cec5SDimitry Andric VE.purgeFunction(); 35350b57cec5SDimitry Andric Stream.ExitBlock(); 35360b57cec5SDimitry Andric } 35370b57cec5SDimitry Andric 35380b57cec5SDimitry Andric // Emit blockinfo, which defines the standard abbreviations etc. 35390b57cec5SDimitry Andric void ModuleBitcodeWriter::writeBlockInfo() { 35400b57cec5SDimitry Andric // We only want to emit block info records for blocks that have multiple 35410b57cec5SDimitry Andric // instances: CONSTANTS_BLOCK, FUNCTION_BLOCK and VALUE_SYMTAB_BLOCK. 35420b57cec5SDimitry Andric // Other blocks can define their abbrevs inline. 35430b57cec5SDimitry Andric Stream.EnterBlockInfoBlock(); 35440b57cec5SDimitry Andric 35450b57cec5SDimitry Andric { // 8-bit fixed-width VST_CODE_ENTRY/VST_CODE_BBENTRY strings. 35460b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 35470b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 3)); 35480b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 35490b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 35500b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 35510b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, Abbv) != 35520b57cec5SDimitry Andric VST_ENTRY_8_ABBREV) 35530b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 35540b57cec5SDimitry Andric } 35550b57cec5SDimitry Andric 35560b57cec5SDimitry Andric { // 7-bit fixed width VST_CODE_ENTRY strings. 35570b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 35580b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_ENTRY)); 35590b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 35600b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 35610b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 35620b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, Abbv) != 35630b57cec5SDimitry Andric VST_ENTRY_7_ABBREV) 35640b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 35650b57cec5SDimitry Andric } 35660b57cec5SDimitry Andric { // 6-bit char6 VST_CODE_ENTRY strings. 35670b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 35680b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_ENTRY)); 35690b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 35700b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 35710b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 35720b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, Abbv) != 35730b57cec5SDimitry Andric VST_ENTRY_6_ABBREV) 35740b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 35750b57cec5SDimitry Andric } 35760b57cec5SDimitry Andric { // 6-bit char6 VST_CODE_BBENTRY strings. 35770b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 35780b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_BBENTRY)); 35790b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 35800b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 35810b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 35820b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, Abbv) != 35830b57cec5SDimitry Andric VST_BBENTRY_6_ABBREV) 35840b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 35850b57cec5SDimitry Andric } 35860b57cec5SDimitry Andric 35870b57cec5SDimitry Andric { // SETTYPE abbrev for CONSTANTS_BLOCK. 35880b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 35890b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_SETTYPE)); 35900b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 35910b57cec5SDimitry Andric VE.computeBitsRequiredForTypeIndicies())); 35920b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, Abbv) != 35930b57cec5SDimitry Andric CONSTANTS_SETTYPE_ABBREV) 35940b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 35950b57cec5SDimitry Andric } 35960b57cec5SDimitry Andric 35970b57cec5SDimitry Andric { // INTEGER abbrev for CONSTANTS_BLOCK. 35980b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 35990b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_INTEGER)); 36000b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 36010b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, Abbv) != 36020b57cec5SDimitry Andric CONSTANTS_INTEGER_ABBREV) 36030b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 36040b57cec5SDimitry Andric } 36050b57cec5SDimitry Andric 36060b57cec5SDimitry Andric { // CE_CAST abbrev for CONSTANTS_BLOCK. 36070b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 36080b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CE_CAST)); 36090b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // cast opc 36100b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // typeid 36110b57cec5SDimitry Andric VE.computeBitsRequiredForTypeIndicies())); 36120b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // value id 36130b57cec5SDimitry Andric 36140b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, Abbv) != 36150b57cec5SDimitry Andric CONSTANTS_CE_CAST_Abbrev) 36160b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 36170b57cec5SDimitry Andric } 36180b57cec5SDimitry Andric { // NULL abbrev for CONSTANTS_BLOCK. 36190b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 36200b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_NULL)); 36210b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, Abbv) != 36220b57cec5SDimitry Andric CONSTANTS_NULL_Abbrev) 36230b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 36240b57cec5SDimitry Andric } 36250b57cec5SDimitry Andric 36260b57cec5SDimitry Andric // FIXME: This should only use space for first class types! 36270b57cec5SDimitry Andric 36280b57cec5SDimitry Andric { // INST_LOAD abbrev for FUNCTION_BLOCK. 36290b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 36300b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_LOAD)); 36310b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // Ptr 36320b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // dest ty 36330b57cec5SDimitry Andric VE.computeBitsRequiredForTypeIndicies())); 36340b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // Align 36350b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // volatile 36360b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 36370b57cec5SDimitry Andric FUNCTION_INST_LOAD_ABBREV) 36380b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 36390b57cec5SDimitry Andric } 36400b57cec5SDimitry Andric { // INST_UNOP abbrev for FUNCTION_BLOCK. 36410b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 36420b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_UNOP)); 36430b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 36440b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 36450b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 36460b57cec5SDimitry Andric FUNCTION_INST_UNOP_ABBREV) 36470b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 36480b57cec5SDimitry Andric } 36490b57cec5SDimitry Andric { // INST_UNOP_FLAGS abbrev for FUNCTION_BLOCK. 36500b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 36510b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_UNOP)); 36520b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 36530b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 36540b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); // flags 36550b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 36560b57cec5SDimitry Andric FUNCTION_INST_UNOP_FLAGS_ABBREV) 36570b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 36580b57cec5SDimitry Andric } 36590b57cec5SDimitry Andric { // INST_BINOP abbrev for FUNCTION_BLOCK. 36600b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 36610b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_BINOP)); 36620b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 36630b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // RHS 36640b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 36650b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 36660b57cec5SDimitry Andric FUNCTION_INST_BINOP_ABBREV) 36670b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 36680b57cec5SDimitry Andric } 36690b57cec5SDimitry Andric { // INST_BINOP_FLAGS abbrev for FUNCTION_BLOCK. 36700b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 36710b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_BINOP)); 36720b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 36730b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // RHS 36740b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 36750b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); // flags 36760b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 36770b57cec5SDimitry Andric FUNCTION_INST_BINOP_FLAGS_ABBREV) 36780b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 36790b57cec5SDimitry Andric } 36800b57cec5SDimitry Andric { // INST_CAST abbrev for FUNCTION_BLOCK. 36810b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 36820b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_CAST)); 36830b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // OpVal 36840b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // dest ty 36850b57cec5SDimitry Andric VE.computeBitsRequiredForTypeIndicies())); 36860b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 36870b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 36880b57cec5SDimitry Andric FUNCTION_INST_CAST_ABBREV) 36890b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 36900b57cec5SDimitry Andric } 3691*c9157d92SDimitry Andric { // INST_CAST_FLAGS abbrev for FUNCTION_BLOCK. 3692*c9157d92SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3693*c9157d92SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_CAST)); 3694*c9157d92SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // OpVal 3695*c9157d92SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // dest ty 3696*c9157d92SDimitry Andric VE.computeBitsRequiredForTypeIndicies())); 3697*c9157d92SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 3698*c9157d92SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); // flags 3699*c9157d92SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3700*c9157d92SDimitry Andric FUNCTION_INST_CAST_FLAGS_ABBREV) 3701*c9157d92SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3702*c9157d92SDimitry Andric } 37030b57cec5SDimitry Andric 37040b57cec5SDimitry Andric { // INST_RET abbrev for FUNCTION_BLOCK. 37050b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 37060b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_RET)); 37070b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 37080b57cec5SDimitry Andric FUNCTION_INST_RET_VOID_ABBREV) 37090b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 37100b57cec5SDimitry Andric } 37110b57cec5SDimitry Andric { // INST_RET abbrev for FUNCTION_BLOCK. 37120b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 37130b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_RET)); 37140b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // ValID 37150b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 37160b57cec5SDimitry Andric FUNCTION_INST_RET_VAL_ABBREV) 37170b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 37180b57cec5SDimitry Andric } 37190b57cec5SDimitry Andric { // INST_UNREACHABLE abbrev for FUNCTION_BLOCK. 37200b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 37210b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_UNREACHABLE)); 37220b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 37230b57cec5SDimitry Andric FUNCTION_INST_UNREACHABLE_ABBREV) 37240b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 37250b57cec5SDimitry Andric } 37260b57cec5SDimitry Andric { 37270b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 37280b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_GEP)); 37290b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 37300b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // dest ty 37310b57cec5SDimitry Andric Log2_32_Ceil(VE.getTypes().size() + 1))); 37320b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 37330b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 37340b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 37350b57cec5SDimitry Andric FUNCTION_INST_GEP_ABBREV) 37360b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 37370b57cec5SDimitry Andric } 37380b57cec5SDimitry Andric 37390b57cec5SDimitry Andric Stream.ExitBlock(); 37400b57cec5SDimitry Andric } 37410b57cec5SDimitry Andric 37420b57cec5SDimitry Andric /// Write the module path strings, currently only used when generating 37430b57cec5SDimitry Andric /// a combined index file. 37440b57cec5SDimitry Andric void IndexBitcodeWriter::writeModStrings() { 37450b57cec5SDimitry Andric Stream.EnterSubblock(bitc::MODULE_STRTAB_BLOCK_ID, 3); 37460b57cec5SDimitry Andric 37470b57cec5SDimitry Andric // TODO: See which abbrev sizes we actually need to emit 37480b57cec5SDimitry Andric 37490b57cec5SDimitry Andric // 8-bit fixed-width MST_ENTRY strings. 37500b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 37510b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MST_CODE_ENTRY)); 37520b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 37530b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 37540b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 37550b57cec5SDimitry Andric unsigned Abbrev8Bit = Stream.EmitAbbrev(std::move(Abbv)); 37560b57cec5SDimitry Andric 37570b57cec5SDimitry Andric // 7-bit fixed width MST_ENTRY strings. 37580b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 37590b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MST_CODE_ENTRY)); 37600b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 37610b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 37620b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 37630b57cec5SDimitry Andric unsigned Abbrev7Bit = Stream.EmitAbbrev(std::move(Abbv)); 37640b57cec5SDimitry Andric 37650b57cec5SDimitry Andric // 6-bit char6 MST_ENTRY strings. 37660b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 37670b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MST_CODE_ENTRY)); 37680b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 37690b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 37700b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 37710b57cec5SDimitry Andric unsigned Abbrev6Bit = Stream.EmitAbbrev(std::move(Abbv)); 37720b57cec5SDimitry Andric 37730b57cec5SDimitry Andric // Module Hash, 160 bits SHA1. Optionally, emitted after each MST_CODE_ENTRY. 37740b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 37750b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MST_CODE_HASH)); 37760b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 37770b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 37780b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 37790b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 37800b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 37810b57cec5SDimitry Andric unsigned AbbrevHash = Stream.EmitAbbrev(std::move(Abbv)); 37820b57cec5SDimitry Andric 37830b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 3784*c9157d92SDimitry Andric forEachModule([&](const StringMapEntry<ModuleHash> &MPSE) { 37850b57cec5SDimitry Andric StringRef Key = MPSE.getKey(); 3786*c9157d92SDimitry Andric const auto &Hash = MPSE.getValue(); 37870b57cec5SDimitry Andric StringEncoding Bits = getStringEncoding(Key); 37880b57cec5SDimitry Andric unsigned AbbrevToUse = Abbrev8Bit; 37890b57cec5SDimitry Andric if (Bits == SE_Char6) 37900b57cec5SDimitry Andric AbbrevToUse = Abbrev6Bit; 37910b57cec5SDimitry Andric else if (Bits == SE_Fixed7) 37920b57cec5SDimitry Andric AbbrevToUse = Abbrev7Bit; 37930b57cec5SDimitry Andric 3794*c9157d92SDimitry Andric auto ModuleId = ModuleIdMap.size(); 3795*c9157d92SDimitry Andric ModuleIdMap[Key] = ModuleId; 3796*c9157d92SDimitry Andric Vals.push_back(ModuleId); 37970b57cec5SDimitry Andric Vals.append(Key.begin(), Key.end()); 37980b57cec5SDimitry Andric 37990b57cec5SDimitry Andric // Emit the finished record. 38000b57cec5SDimitry Andric Stream.EmitRecord(bitc::MST_CODE_ENTRY, Vals, AbbrevToUse); 38010b57cec5SDimitry Andric 38020b57cec5SDimitry Andric // Emit an optional hash for the module now 38030b57cec5SDimitry Andric if (llvm::any_of(Hash, [](uint32_t H) { return H; })) { 38040b57cec5SDimitry Andric Vals.assign(Hash.begin(), Hash.end()); 38050b57cec5SDimitry Andric // Emit the hash record. 38060b57cec5SDimitry Andric Stream.EmitRecord(bitc::MST_CODE_HASH, Vals, AbbrevHash); 38070b57cec5SDimitry Andric } 38080b57cec5SDimitry Andric 38090b57cec5SDimitry Andric Vals.clear(); 38100b57cec5SDimitry Andric }); 38110b57cec5SDimitry Andric Stream.ExitBlock(); 38120b57cec5SDimitry Andric } 38130b57cec5SDimitry Andric 38140b57cec5SDimitry Andric /// Write the function type metadata related records that need to appear before 38150b57cec5SDimitry Andric /// a function summary entry (whether per-module or combined). 3816e8d8bef9SDimitry Andric template <typename Fn> 38170b57cec5SDimitry Andric static void writeFunctionTypeMetadataRecords(BitstreamWriter &Stream, 3818e8d8bef9SDimitry Andric FunctionSummary *FS, 3819e8d8bef9SDimitry Andric Fn GetValueID) { 38200b57cec5SDimitry Andric if (!FS->type_tests().empty()) 38210b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_TYPE_TESTS, FS->type_tests()); 38220b57cec5SDimitry Andric 38230b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 38240b57cec5SDimitry Andric 38250b57cec5SDimitry Andric auto WriteVFuncIdVec = [&](uint64_t Ty, 38260b57cec5SDimitry Andric ArrayRef<FunctionSummary::VFuncId> VFs) { 38270b57cec5SDimitry Andric if (VFs.empty()) 38280b57cec5SDimitry Andric return; 38290b57cec5SDimitry Andric Record.clear(); 38300b57cec5SDimitry Andric for (auto &VF : VFs) { 38310b57cec5SDimitry Andric Record.push_back(VF.GUID); 38320b57cec5SDimitry Andric Record.push_back(VF.Offset); 38330b57cec5SDimitry Andric } 38340b57cec5SDimitry Andric Stream.EmitRecord(Ty, Record); 38350b57cec5SDimitry Andric }; 38360b57cec5SDimitry Andric 38370b57cec5SDimitry Andric WriteVFuncIdVec(bitc::FS_TYPE_TEST_ASSUME_VCALLS, 38380b57cec5SDimitry Andric FS->type_test_assume_vcalls()); 38390b57cec5SDimitry Andric WriteVFuncIdVec(bitc::FS_TYPE_CHECKED_LOAD_VCALLS, 38400b57cec5SDimitry Andric FS->type_checked_load_vcalls()); 38410b57cec5SDimitry Andric 38420b57cec5SDimitry Andric auto WriteConstVCallVec = [&](uint64_t Ty, 38430b57cec5SDimitry Andric ArrayRef<FunctionSummary::ConstVCall> VCs) { 38440b57cec5SDimitry Andric for (auto &VC : VCs) { 38450b57cec5SDimitry Andric Record.clear(); 38460b57cec5SDimitry Andric Record.push_back(VC.VFunc.GUID); 38470b57cec5SDimitry Andric Record.push_back(VC.VFunc.Offset); 3848e8d8bef9SDimitry Andric llvm::append_range(Record, VC.Args); 38490b57cec5SDimitry Andric Stream.EmitRecord(Ty, Record); 38500b57cec5SDimitry Andric } 38510b57cec5SDimitry Andric }; 38520b57cec5SDimitry Andric 38530b57cec5SDimitry Andric WriteConstVCallVec(bitc::FS_TYPE_TEST_ASSUME_CONST_VCALL, 38540b57cec5SDimitry Andric FS->type_test_assume_const_vcalls()); 38550b57cec5SDimitry Andric WriteConstVCallVec(bitc::FS_TYPE_CHECKED_LOAD_CONST_VCALL, 38560b57cec5SDimitry Andric FS->type_checked_load_const_vcalls()); 38575ffd83dbSDimitry Andric 38585ffd83dbSDimitry Andric auto WriteRange = [&](ConstantRange Range) { 38595ffd83dbSDimitry Andric Range = Range.sextOrTrunc(FunctionSummary::ParamAccess::RangeWidth); 38605ffd83dbSDimitry Andric assert(Range.getLower().getNumWords() == 1); 38615ffd83dbSDimitry Andric assert(Range.getUpper().getNumWords() == 1); 38625ffd83dbSDimitry Andric emitSignedInt64(Record, *Range.getLower().getRawData()); 38635ffd83dbSDimitry Andric emitSignedInt64(Record, *Range.getUpper().getRawData()); 38645ffd83dbSDimitry Andric }; 38655ffd83dbSDimitry Andric 38665ffd83dbSDimitry Andric if (!FS->paramAccesses().empty()) { 38675ffd83dbSDimitry Andric Record.clear(); 38685ffd83dbSDimitry Andric for (auto &Arg : FS->paramAccesses()) { 3869e8d8bef9SDimitry Andric size_t UndoSize = Record.size(); 38705ffd83dbSDimitry Andric Record.push_back(Arg.ParamNo); 38715ffd83dbSDimitry Andric WriteRange(Arg.Use); 38725ffd83dbSDimitry Andric Record.push_back(Arg.Calls.size()); 38735ffd83dbSDimitry Andric for (auto &Call : Arg.Calls) { 38745ffd83dbSDimitry Andric Record.push_back(Call.ParamNo); 3875bdd1243dSDimitry Andric std::optional<unsigned> ValueID = GetValueID(Call.Callee); 3876e8d8bef9SDimitry Andric if (!ValueID) { 3877e8d8bef9SDimitry Andric // If ValueID is unknown we can't drop just this call, we must drop 3878e8d8bef9SDimitry Andric // entire parameter. 3879e8d8bef9SDimitry Andric Record.resize(UndoSize); 3880e8d8bef9SDimitry Andric break; 3881e8d8bef9SDimitry Andric } 3882e8d8bef9SDimitry Andric Record.push_back(*ValueID); 38835ffd83dbSDimitry Andric WriteRange(Call.Offsets); 38845ffd83dbSDimitry Andric } 38855ffd83dbSDimitry Andric } 3886e8d8bef9SDimitry Andric if (!Record.empty()) 38875ffd83dbSDimitry Andric Stream.EmitRecord(bitc::FS_PARAM_ACCESS, Record); 38885ffd83dbSDimitry Andric } 38890b57cec5SDimitry Andric } 38900b57cec5SDimitry Andric 38910b57cec5SDimitry Andric /// Collect type IDs from type tests used by function. 38920b57cec5SDimitry Andric static void 38930b57cec5SDimitry Andric getReferencedTypeIds(FunctionSummary *FS, 38940b57cec5SDimitry Andric std::set<GlobalValue::GUID> &ReferencedTypeIds) { 38950b57cec5SDimitry Andric if (!FS->type_tests().empty()) 38960b57cec5SDimitry Andric for (auto &TT : FS->type_tests()) 38970b57cec5SDimitry Andric ReferencedTypeIds.insert(TT); 38980b57cec5SDimitry Andric 38990b57cec5SDimitry Andric auto GetReferencedTypesFromVFuncIdVec = 39000b57cec5SDimitry Andric [&](ArrayRef<FunctionSummary::VFuncId> VFs) { 39010b57cec5SDimitry Andric for (auto &VF : VFs) 39020b57cec5SDimitry Andric ReferencedTypeIds.insert(VF.GUID); 39030b57cec5SDimitry Andric }; 39040b57cec5SDimitry Andric 39050b57cec5SDimitry Andric GetReferencedTypesFromVFuncIdVec(FS->type_test_assume_vcalls()); 39060b57cec5SDimitry Andric GetReferencedTypesFromVFuncIdVec(FS->type_checked_load_vcalls()); 39070b57cec5SDimitry Andric 39080b57cec5SDimitry Andric auto GetReferencedTypesFromConstVCallVec = 39090b57cec5SDimitry Andric [&](ArrayRef<FunctionSummary::ConstVCall> VCs) { 39100b57cec5SDimitry Andric for (auto &VC : VCs) 39110b57cec5SDimitry Andric ReferencedTypeIds.insert(VC.VFunc.GUID); 39120b57cec5SDimitry Andric }; 39130b57cec5SDimitry Andric 39140b57cec5SDimitry Andric GetReferencedTypesFromConstVCallVec(FS->type_test_assume_const_vcalls()); 39150b57cec5SDimitry Andric GetReferencedTypesFromConstVCallVec(FS->type_checked_load_const_vcalls()); 39160b57cec5SDimitry Andric } 39170b57cec5SDimitry Andric 39180b57cec5SDimitry Andric static void writeWholeProgramDevirtResolutionByArg( 39190b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, const std::vector<uint64_t> &args, 39200b57cec5SDimitry Andric const WholeProgramDevirtResolution::ByArg &ByArg) { 39210b57cec5SDimitry Andric NameVals.push_back(args.size()); 3922e8d8bef9SDimitry Andric llvm::append_range(NameVals, args); 39230b57cec5SDimitry Andric 39240b57cec5SDimitry Andric NameVals.push_back(ByArg.TheKind); 39250b57cec5SDimitry Andric NameVals.push_back(ByArg.Info); 39260b57cec5SDimitry Andric NameVals.push_back(ByArg.Byte); 39270b57cec5SDimitry Andric NameVals.push_back(ByArg.Bit); 39280b57cec5SDimitry Andric } 39290b57cec5SDimitry Andric 39300b57cec5SDimitry Andric static void writeWholeProgramDevirtResolution( 39310b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, StringTableBuilder &StrtabBuilder, 39320b57cec5SDimitry Andric uint64_t Id, const WholeProgramDevirtResolution &Wpd) { 39330b57cec5SDimitry Andric NameVals.push_back(Id); 39340b57cec5SDimitry Andric 39350b57cec5SDimitry Andric NameVals.push_back(Wpd.TheKind); 39360b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(Wpd.SingleImplName)); 39370b57cec5SDimitry Andric NameVals.push_back(Wpd.SingleImplName.size()); 39380b57cec5SDimitry Andric 39390b57cec5SDimitry Andric NameVals.push_back(Wpd.ResByArg.size()); 39400b57cec5SDimitry Andric for (auto &A : Wpd.ResByArg) 39410b57cec5SDimitry Andric writeWholeProgramDevirtResolutionByArg(NameVals, A.first, A.second); 39420b57cec5SDimitry Andric } 39430b57cec5SDimitry Andric 39440b57cec5SDimitry Andric static void writeTypeIdSummaryRecord(SmallVector<uint64_t, 64> &NameVals, 39450b57cec5SDimitry Andric StringTableBuilder &StrtabBuilder, 39460b57cec5SDimitry Andric const std::string &Id, 39470b57cec5SDimitry Andric const TypeIdSummary &Summary) { 39480b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(Id)); 39490b57cec5SDimitry Andric NameVals.push_back(Id.size()); 39500b57cec5SDimitry Andric 39510b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.TheKind); 39520b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.SizeM1BitWidth); 39530b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.AlignLog2); 39540b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.SizeM1); 39550b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.BitMask); 39560b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.InlineBits); 39570b57cec5SDimitry Andric 39580b57cec5SDimitry Andric for (auto &W : Summary.WPDRes) 39590b57cec5SDimitry Andric writeWholeProgramDevirtResolution(NameVals, StrtabBuilder, W.first, 39600b57cec5SDimitry Andric W.second); 39610b57cec5SDimitry Andric } 39620b57cec5SDimitry Andric 39630b57cec5SDimitry Andric static void writeTypeIdCompatibleVtableSummaryRecord( 39640b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, StringTableBuilder &StrtabBuilder, 39650b57cec5SDimitry Andric const std::string &Id, const TypeIdCompatibleVtableInfo &Summary, 39660b57cec5SDimitry Andric ValueEnumerator &VE) { 39670b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(Id)); 39680b57cec5SDimitry Andric NameVals.push_back(Id.size()); 39690b57cec5SDimitry Andric 39700b57cec5SDimitry Andric for (auto &P : Summary) { 39710b57cec5SDimitry Andric NameVals.push_back(P.AddressPointOffset); 39720b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(P.VTableVI.getValue())); 39730b57cec5SDimitry Andric } 39740b57cec5SDimitry Andric } 39750b57cec5SDimitry Andric 3976bdd1243dSDimitry Andric static void writeFunctionHeapProfileRecords( 3977bdd1243dSDimitry Andric BitstreamWriter &Stream, FunctionSummary *FS, unsigned CallsiteAbbrev, 3978bdd1243dSDimitry Andric unsigned AllocAbbrev, bool PerModule, 3979bdd1243dSDimitry Andric std::function<unsigned(const ValueInfo &VI)> GetValueID, 3980bdd1243dSDimitry Andric std::function<unsigned(unsigned)> GetStackIndex) { 3981bdd1243dSDimitry Andric SmallVector<uint64_t> Record; 3982bdd1243dSDimitry Andric 3983bdd1243dSDimitry Andric for (auto &CI : FS->callsites()) { 3984bdd1243dSDimitry Andric Record.clear(); 3985bdd1243dSDimitry Andric // Per module callsite clones should always have a single entry of 3986bdd1243dSDimitry Andric // value 0. 3987bdd1243dSDimitry Andric assert(!PerModule || (CI.Clones.size() == 1 && CI.Clones[0] == 0)); 3988bdd1243dSDimitry Andric Record.push_back(GetValueID(CI.Callee)); 3989bdd1243dSDimitry Andric if (!PerModule) { 3990bdd1243dSDimitry Andric Record.push_back(CI.StackIdIndices.size()); 3991bdd1243dSDimitry Andric Record.push_back(CI.Clones.size()); 3992bdd1243dSDimitry Andric } 3993bdd1243dSDimitry Andric for (auto Id : CI.StackIdIndices) 3994bdd1243dSDimitry Andric Record.push_back(GetStackIndex(Id)); 3995bdd1243dSDimitry Andric if (!PerModule) { 3996bdd1243dSDimitry Andric for (auto V : CI.Clones) 3997bdd1243dSDimitry Andric Record.push_back(V); 3998bdd1243dSDimitry Andric } 3999bdd1243dSDimitry Andric Stream.EmitRecord(PerModule ? bitc::FS_PERMODULE_CALLSITE_INFO 4000bdd1243dSDimitry Andric : bitc::FS_COMBINED_CALLSITE_INFO, 4001bdd1243dSDimitry Andric Record, CallsiteAbbrev); 4002bdd1243dSDimitry Andric } 4003bdd1243dSDimitry Andric 4004bdd1243dSDimitry Andric for (auto &AI : FS->allocs()) { 4005bdd1243dSDimitry Andric Record.clear(); 4006bdd1243dSDimitry Andric // Per module alloc versions should always have a single entry of 4007bdd1243dSDimitry Andric // value 0. 4008bdd1243dSDimitry Andric assert(!PerModule || (AI.Versions.size() == 1 && AI.Versions[0] == 0)); 4009bdd1243dSDimitry Andric if (!PerModule) { 4010bdd1243dSDimitry Andric Record.push_back(AI.MIBs.size()); 4011bdd1243dSDimitry Andric Record.push_back(AI.Versions.size()); 4012bdd1243dSDimitry Andric } 4013bdd1243dSDimitry Andric for (auto &MIB : AI.MIBs) { 4014bdd1243dSDimitry Andric Record.push_back((uint8_t)MIB.AllocType); 4015bdd1243dSDimitry Andric Record.push_back(MIB.StackIdIndices.size()); 4016bdd1243dSDimitry Andric for (auto Id : MIB.StackIdIndices) 4017bdd1243dSDimitry Andric Record.push_back(GetStackIndex(Id)); 4018bdd1243dSDimitry Andric } 4019bdd1243dSDimitry Andric if (!PerModule) { 4020bdd1243dSDimitry Andric for (auto V : AI.Versions) 4021bdd1243dSDimitry Andric Record.push_back(V); 4022bdd1243dSDimitry Andric } 4023bdd1243dSDimitry Andric Stream.EmitRecord(PerModule ? bitc::FS_PERMODULE_ALLOC_INFO 4024bdd1243dSDimitry Andric : bitc::FS_COMBINED_ALLOC_INFO, 4025bdd1243dSDimitry Andric Record, AllocAbbrev); 4026bdd1243dSDimitry Andric } 4027bdd1243dSDimitry Andric } 4028bdd1243dSDimitry Andric 40290b57cec5SDimitry Andric // Helper to emit a single function summary record. 40300b57cec5SDimitry Andric void ModuleBitcodeWriterBase::writePerModuleFunctionSummaryRecord( 40310b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, GlobalValueSummary *Summary, 4032*c9157d92SDimitry Andric unsigned ValueID, unsigned FSCallsRelBFAbbrev, 4033*c9157d92SDimitry Andric unsigned FSCallsProfileAbbrev, unsigned CallsiteAbbrev, 4034*c9157d92SDimitry Andric unsigned AllocAbbrev, const Function &F) { 40350b57cec5SDimitry Andric NameVals.push_back(ValueID); 40360b57cec5SDimitry Andric 40370b57cec5SDimitry Andric FunctionSummary *FS = cast<FunctionSummary>(Summary); 4038e8d8bef9SDimitry Andric 4039e8d8bef9SDimitry Andric writeFunctionTypeMetadataRecords( 4040bdd1243dSDimitry Andric Stream, FS, [&](const ValueInfo &VI) -> std::optional<unsigned> { 4041e8d8bef9SDimitry Andric return {VE.getValueID(VI.getValue())}; 4042e8d8bef9SDimitry Andric }); 40430b57cec5SDimitry Andric 4044bdd1243dSDimitry Andric writeFunctionHeapProfileRecords( 4045bdd1243dSDimitry Andric Stream, FS, CallsiteAbbrev, AllocAbbrev, 4046bdd1243dSDimitry Andric /*PerModule*/ true, 4047bdd1243dSDimitry Andric /*GetValueId*/ [&](const ValueInfo &VI) { return getValueId(VI); }, 4048bdd1243dSDimitry Andric /*GetStackIndex*/ [&](unsigned I) { return I; }); 4049bdd1243dSDimitry Andric 40500b57cec5SDimitry Andric auto SpecialRefCnts = FS->specialRefCounts(); 40510b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(FS->flags())); 40520b57cec5SDimitry Andric NameVals.push_back(FS->instCount()); 40530b57cec5SDimitry Andric NameVals.push_back(getEncodedFFlags(FS->fflags())); 40540b57cec5SDimitry Andric NameVals.push_back(FS->refs().size()); 40550b57cec5SDimitry Andric NameVals.push_back(SpecialRefCnts.first); // rorefcnt 40560b57cec5SDimitry Andric NameVals.push_back(SpecialRefCnts.second); // worefcnt 40570b57cec5SDimitry Andric 40580b57cec5SDimitry Andric for (auto &RI : FS->refs()) 40590b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(RI.getValue())); 40600b57cec5SDimitry Andric 4061*c9157d92SDimitry Andric const bool UseRelBFRecord = 4062*c9157d92SDimitry Andric WriteRelBFToSummary && !F.hasProfileData() && 4063*c9157d92SDimitry Andric ForceSummaryEdgesCold == FunctionSummary::FSHT_None; 40640b57cec5SDimitry Andric for (auto &ECI : FS->calls()) { 40650b57cec5SDimitry Andric NameVals.push_back(getValueId(ECI.first)); 4066*c9157d92SDimitry Andric if (UseRelBFRecord) 4067*c9157d92SDimitry Andric NameVals.push_back(getEncodedRelBFCallEdgeInfo(ECI.second)); 4068*c9157d92SDimitry Andric else 4069*c9157d92SDimitry Andric NameVals.push_back(getEncodedHotnessCallEdgeInfo(ECI.second)); 40700b57cec5SDimitry Andric } 40710b57cec5SDimitry Andric 4072*c9157d92SDimitry Andric unsigned FSAbbrev = 4073*c9157d92SDimitry Andric (UseRelBFRecord ? FSCallsRelBFAbbrev : FSCallsProfileAbbrev); 40740b57cec5SDimitry Andric unsigned Code = 4075*c9157d92SDimitry Andric (UseRelBFRecord ? bitc::FS_PERMODULE_RELBF : bitc::FS_PERMODULE_PROFILE); 40760b57cec5SDimitry Andric 40770b57cec5SDimitry Andric // Emit the finished record. 40780b57cec5SDimitry Andric Stream.EmitRecord(Code, NameVals, FSAbbrev); 40790b57cec5SDimitry Andric NameVals.clear(); 40800b57cec5SDimitry Andric } 40810b57cec5SDimitry Andric 40820b57cec5SDimitry Andric // Collect the global value references in the given variable's initializer, 40830b57cec5SDimitry Andric // and emit them in a summary record. 40840b57cec5SDimitry Andric void ModuleBitcodeWriterBase::writeModuleLevelReferences( 40850b57cec5SDimitry Andric const GlobalVariable &V, SmallVector<uint64_t, 64> &NameVals, 40860b57cec5SDimitry Andric unsigned FSModRefsAbbrev, unsigned FSModVTableRefsAbbrev) { 40870b57cec5SDimitry Andric auto VI = Index->getValueInfo(V.getGUID()); 40880b57cec5SDimitry Andric if (!VI || VI.getSummaryList().empty()) { 40890b57cec5SDimitry Andric // Only declarations should not have a summary (a declaration might however 40900b57cec5SDimitry Andric // have a summary if the def was in module level asm). 40910b57cec5SDimitry Andric assert(V.isDeclaration()); 40920b57cec5SDimitry Andric return; 40930b57cec5SDimitry Andric } 40940b57cec5SDimitry Andric auto *Summary = VI.getSummaryList()[0].get(); 40950b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(&V)); 40960b57cec5SDimitry Andric GlobalVarSummary *VS = cast<GlobalVarSummary>(Summary); 40970b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(VS->flags())); 40980b57cec5SDimitry Andric NameVals.push_back(getEncodedGVarFlags(VS->varflags())); 40990b57cec5SDimitry Andric 41000b57cec5SDimitry Andric auto VTableFuncs = VS->vTableFuncs(); 41010b57cec5SDimitry Andric if (!VTableFuncs.empty()) 41020b57cec5SDimitry Andric NameVals.push_back(VS->refs().size()); 41030b57cec5SDimitry Andric 41040b57cec5SDimitry Andric unsigned SizeBeforeRefs = NameVals.size(); 41050b57cec5SDimitry Andric for (auto &RI : VS->refs()) 41060b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(RI.getValue())); 41070b57cec5SDimitry Andric // Sort the refs for determinism output, the vector returned by FS->refs() has 41080b57cec5SDimitry Andric // been initialized from a DenseSet. 4109e8d8bef9SDimitry Andric llvm::sort(drop_begin(NameVals, SizeBeforeRefs)); 41100b57cec5SDimitry Andric 41110b57cec5SDimitry Andric if (VTableFuncs.empty()) 41120b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_PERMODULE_GLOBALVAR_INIT_REFS, NameVals, 41130b57cec5SDimitry Andric FSModRefsAbbrev); 41140b57cec5SDimitry Andric else { 41150b57cec5SDimitry Andric // VTableFuncs pairs should already be sorted by offset. 41160b57cec5SDimitry Andric for (auto &P : VTableFuncs) { 41170b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(P.FuncVI.getValue())); 41180b57cec5SDimitry Andric NameVals.push_back(P.VTableOffset); 41190b57cec5SDimitry Andric } 41200b57cec5SDimitry Andric 41210b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_PERMODULE_VTABLE_GLOBALVAR_INIT_REFS, NameVals, 41220b57cec5SDimitry Andric FSModVTableRefsAbbrev); 41230b57cec5SDimitry Andric } 41240b57cec5SDimitry Andric NameVals.clear(); 41250b57cec5SDimitry Andric } 41260b57cec5SDimitry Andric 41270b57cec5SDimitry Andric /// Emit the per-module summary section alongside the rest of 41280b57cec5SDimitry Andric /// the module's bitcode. 41290b57cec5SDimitry Andric void ModuleBitcodeWriterBase::writePerModuleGlobalValueSummary() { 41300b57cec5SDimitry Andric // By default we compile with ThinLTO if the module has a summary, but the 41310b57cec5SDimitry Andric // client can request full LTO with a module flag. 41320b57cec5SDimitry Andric bool IsThinLTO = true; 41330b57cec5SDimitry Andric if (auto *MD = 41340b57cec5SDimitry Andric mdconst::extract_or_null<ConstantInt>(M.getModuleFlag("ThinLTO"))) 41350b57cec5SDimitry Andric IsThinLTO = MD->getZExtValue(); 41360b57cec5SDimitry Andric Stream.EnterSubblock(IsThinLTO ? bitc::GLOBALVAL_SUMMARY_BLOCK_ID 41370b57cec5SDimitry Andric : bitc::FULL_LTO_GLOBALVAL_SUMMARY_BLOCK_ID, 41380b57cec5SDimitry Andric 4); 41390b57cec5SDimitry Andric 4140480093f4SDimitry Andric Stream.EmitRecord( 4141480093f4SDimitry Andric bitc::FS_VERSION, 4142480093f4SDimitry Andric ArrayRef<uint64_t>{ModuleSummaryIndex::BitcodeSummaryVersion}); 41430b57cec5SDimitry Andric 41440b57cec5SDimitry Andric // Write the index flags. 41450b57cec5SDimitry Andric uint64_t Flags = 0; 41460b57cec5SDimitry Andric // Bits 1-3 are set only in the combined index, skip them. 41470b57cec5SDimitry Andric if (Index->enableSplitLTOUnit()) 41480b57cec5SDimitry Andric Flags |= 0x8; 4149fe013be4SDimitry Andric if (Index->hasUnifiedLTO()) 4150fe013be4SDimitry Andric Flags |= 0x200; 4151fe013be4SDimitry Andric 41520b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_FLAGS, ArrayRef<uint64_t>{Flags}); 41530b57cec5SDimitry Andric 41540b57cec5SDimitry Andric if (Index->begin() == Index->end()) { 41550b57cec5SDimitry Andric Stream.ExitBlock(); 41560b57cec5SDimitry Andric return; 41570b57cec5SDimitry Andric } 41580b57cec5SDimitry Andric 41590b57cec5SDimitry Andric for (const auto &GVI : valueIds()) { 41600b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_VALUE_GUID, 41610b57cec5SDimitry Andric ArrayRef<uint64_t>{GVI.second, GVI.first}); 41620b57cec5SDimitry Andric } 41630b57cec5SDimitry Andric 4164bdd1243dSDimitry Andric if (!Index->stackIds().empty()) { 4165bdd1243dSDimitry Andric auto StackIdAbbv = std::make_shared<BitCodeAbbrev>(); 4166bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(bitc::FS_STACK_IDS)); 4167bdd1243dSDimitry Andric // numids x stackid 4168bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4169bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4170bdd1243dSDimitry Andric unsigned StackIdAbbvId = Stream.EmitAbbrev(std::move(StackIdAbbv)); 4171bdd1243dSDimitry Andric Stream.EmitRecord(bitc::FS_STACK_IDS, Index->stackIds(), StackIdAbbvId); 4172bdd1243dSDimitry Andric } 4173bdd1243dSDimitry Andric 41740b57cec5SDimitry Andric // Abbrev for FS_PERMODULE_PROFILE. 41750b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 41760b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_PROFILE)); 41770b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4178fe013be4SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // flags 41790b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // instcount 41800b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // fflags 41810b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numrefs 41820b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // rorefcnt 41830b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // worefcnt 4184*c9157d92SDimitry Andric // numrefs x valueid, n x (valueid, hotness+tailcall flags) 41850b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 41860b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 41870b57cec5SDimitry Andric unsigned FSCallsProfileAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 41880b57cec5SDimitry Andric 4189*c9157d92SDimitry Andric // Abbrev for FS_PERMODULE_RELBF. 41900b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 41910b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_RELBF)); 41920b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 41930b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 41940b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // instcount 41950b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // fflags 41960b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numrefs 41970b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // rorefcnt 41980b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // worefcnt 4199*c9157d92SDimitry Andric // numrefs x valueid, n x (valueid, rel_block_freq+tailcall]) 42000b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 42010b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4202*c9157d92SDimitry Andric unsigned FSCallsRelBFAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 42030b57cec5SDimitry Andric 42040b57cec5SDimitry Andric // Abbrev for FS_PERMODULE_GLOBALVAR_INIT_REFS. 42050b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 42060b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_GLOBALVAR_INIT_REFS)); 42070b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 42080b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 42090b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); // valueids 42100b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 42110b57cec5SDimitry Andric unsigned FSModRefsAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 42120b57cec5SDimitry Andric 42130b57cec5SDimitry Andric // Abbrev for FS_PERMODULE_VTABLE_GLOBALVAR_INIT_REFS. 42140b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 42150b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_VTABLE_GLOBALVAR_INIT_REFS)); 42160b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 42170b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 42180b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numrefs 42190b57cec5SDimitry Andric // numrefs x valueid, n x (valueid , offset) 42200b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 42210b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 42220b57cec5SDimitry Andric unsigned FSModVTableRefsAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 42230b57cec5SDimitry Andric 42240b57cec5SDimitry Andric // Abbrev for FS_ALIAS. 42250b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 42260b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_ALIAS)); 42270b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 42280b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 42290b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 42300b57cec5SDimitry Andric unsigned FSAliasAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 42310b57cec5SDimitry Andric 42320b57cec5SDimitry Andric // Abbrev for FS_TYPE_ID_METADATA 42330b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 42340b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_TYPE_ID_METADATA)); 42350b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // typeid strtab index 42360b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // typeid length 42370b57cec5SDimitry Andric // n x (valueid , offset) 42380b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 42390b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 42400b57cec5SDimitry Andric unsigned TypeIdCompatibleVtableAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 42410b57cec5SDimitry Andric 4242bdd1243dSDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4243bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_CALLSITE_INFO)); 4244bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4245bdd1243dSDimitry Andric // n x stackidindex 4246bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4247bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4248bdd1243dSDimitry Andric unsigned CallsiteAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4249bdd1243dSDimitry Andric 4250bdd1243dSDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4251bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_ALLOC_INFO)); 4252bdd1243dSDimitry Andric // n x (alloc type, numstackids, numstackids x stackidindex) 4253bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4254bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4255bdd1243dSDimitry Andric unsigned AllocAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4256bdd1243dSDimitry Andric 42570b57cec5SDimitry Andric SmallVector<uint64_t, 64> NameVals; 42580b57cec5SDimitry Andric // Iterate over the list of functions instead of the Index to 42590b57cec5SDimitry Andric // ensure the ordering is stable. 42600b57cec5SDimitry Andric for (const Function &F : M) { 42610b57cec5SDimitry Andric // Summary emission does not support anonymous functions, they have to 42620b57cec5SDimitry Andric // renamed using the anonymous function renaming pass. 42630b57cec5SDimitry Andric if (!F.hasName()) 42640b57cec5SDimitry Andric report_fatal_error("Unexpected anonymous function when writing summary"); 42650b57cec5SDimitry Andric 42660b57cec5SDimitry Andric ValueInfo VI = Index->getValueInfo(F.getGUID()); 42670b57cec5SDimitry Andric if (!VI || VI.getSummaryList().empty()) { 42680b57cec5SDimitry Andric // Only declarations should not have a summary (a declaration might 42690b57cec5SDimitry Andric // however have a summary if the def was in module level asm). 42700b57cec5SDimitry Andric assert(F.isDeclaration()); 42710b57cec5SDimitry Andric continue; 42720b57cec5SDimitry Andric } 42730b57cec5SDimitry Andric auto *Summary = VI.getSummaryList()[0].get(); 4274*c9157d92SDimitry Andric writePerModuleFunctionSummaryRecord( 4275*c9157d92SDimitry Andric NameVals, Summary, VE.getValueID(&F), FSCallsRelBFAbbrev, 4276*c9157d92SDimitry Andric FSCallsProfileAbbrev, CallsiteAbbrev, AllocAbbrev, F); 42770b57cec5SDimitry Andric } 42780b57cec5SDimitry Andric 42790b57cec5SDimitry Andric // Capture references from GlobalVariable initializers, which are outside 42800b57cec5SDimitry Andric // of a function scope. 42810b57cec5SDimitry Andric for (const GlobalVariable &G : M.globals()) 42820b57cec5SDimitry Andric writeModuleLevelReferences(G, NameVals, FSModRefsAbbrev, 42830b57cec5SDimitry Andric FSModVTableRefsAbbrev); 42840b57cec5SDimitry Andric 42850b57cec5SDimitry Andric for (const GlobalAlias &A : M.aliases()) { 4286349cc55cSDimitry Andric auto *Aliasee = A.getAliaseeObject(); 4287fcaf7f86SDimitry Andric // Skip ifunc and nameless functions which don't have an entry in the 4288fcaf7f86SDimitry Andric // summary. 4289fcaf7f86SDimitry Andric if (!Aliasee->hasName() || isa<GlobalIFunc>(Aliasee)) 42900b57cec5SDimitry Andric continue; 42910b57cec5SDimitry Andric auto AliasId = VE.getValueID(&A); 42920b57cec5SDimitry Andric auto AliaseeId = VE.getValueID(Aliasee); 42930b57cec5SDimitry Andric NameVals.push_back(AliasId); 42940b57cec5SDimitry Andric auto *Summary = Index->getGlobalValueSummary(A); 42950b57cec5SDimitry Andric AliasSummary *AS = cast<AliasSummary>(Summary); 42960b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(AS->flags())); 42970b57cec5SDimitry Andric NameVals.push_back(AliaseeId); 42980b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_ALIAS, NameVals, FSAliasAbbrev); 42990b57cec5SDimitry Andric NameVals.clear(); 43000b57cec5SDimitry Andric } 43010b57cec5SDimitry Andric 43020b57cec5SDimitry Andric for (auto &S : Index->typeIdCompatibleVtableMap()) { 43030b57cec5SDimitry Andric writeTypeIdCompatibleVtableSummaryRecord(NameVals, StrtabBuilder, S.first, 43040b57cec5SDimitry Andric S.second, VE); 43050b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_TYPE_ID_METADATA, NameVals, 43060b57cec5SDimitry Andric TypeIdCompatibleVtableAbbrev); 43070b57cec5SDimitry Andric NameVals.clear(); 43080b57cec5SDimitry Andric } 43090b57cec5SDimitry Andric 4310fe013be4SDimitry Andric if (Index->getBlockCount()) 43115ffd83dbSDimitry Andric Stream.EmitRecord(bitc::FS_BLOCK_COUNT, 43125ffd83dbSDimitry Andric ArrayRef<uint64_t>{Index->getBlockCount()}); 43135ffd83dbSDimitry Andric 43140b57cec5SDimitry Andric Stream.ExitBlock(); 43150b57cec5SDimitry Andric } 43160b57cec5SDimitry Andric 43170b57cec5SDimitry Andric /// Emit the combined summary section into the combined index file. 43180b57cec5SDimitry Andric void IndexBitcodeWriter::writeCombinedGlobalValueSummary() { 4319bdd1243dSDimitry Andric Stream.EnterSubblock(bitc::GLOBALVAL_SUMMARY_BLOCK_ID, 4); 4320480093f4SDimitry Andric Stream.EmitRecord( 4321480093f4SDimitry Andric bitc::FS_VERSION, 4322480093f4SDimitry Andric ArrayRef<uint64_t>{ModuleSummaryIndex::BitcodeSummaryVersion}); 43230b57cec5SDimitry Andric 43240b57cec5SDimitry Andric // Write the index flags. 43255ffd83dbSDimitry Andric Stream.EmitRecord(bitc::FS_FLAGS, ArrayRef<uint64_t>{Index.getFlags()}); 43260b57cec5SDimitry Andric 43270b57cec5SDimitry Andric for (const auto &GVI : valueIds()) { 43280b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_VALUE_GUID, 43290b57cec5SDimitry Andric ArrayRef<uint64_t>{GVI.second, GVI.first}); 43300b57cec5SDimitry Andric } 43310b57cec5SDimitry Andric 4332bdd1243dSDimitry Andric if (!StackIdIndices.empty()) { 4333bdd1243dSDimitry Andric auto StackIdAbbv = std::make_shared<BitCodeAbbrev>(); 4334bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(bitc::FS_STACK_IDS)); 4335bdd1243dSDimitry Andric // numids x stackid 4336bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4337bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4338bdd1243dSDimitry Andric unsigned StackIdAbbvId = Stream.EmitAbbrev(std::move(StackIdAbbv)); 4339bdd1243dSDimitry Andric // Write the stack ids used by this index, which will be a subset of those in 4340bdd1243dSDimitry Andric // the full index in the case of distributed indexes. 4341bdd1243dSDimitry Andric std::vector<uint64_t> StackIds; 4342bdd1243dSDimitry Andric for (auto &I : StackIdIndices) 4343bdd1243dSDimitry Andric StackIds.push_back(Index.getStackIdAtIndex(I)); 4344bdd1243dSDimitry Andric Stream.EmitRecord(bitc::FS_STACK_IDS, StackIds, StackIdAbbvId); 4345bdd1243dSDimitry Andric } 4346bdd1243dSDimitry Andric 43470b57cec5SDimitry Andric // Abbrev for FS_COMBINED_PROFILE. 4348*c9157d92SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 43490b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_PROFILE)); 43500b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 43510b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // modid 43520b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 43530b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // instcount 43540b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // fflags 43550b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // entrycount 43560b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numrefs 43570b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // rorefcnt 43580b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // worefcnt 4359*c9157d92SDimitry Andric // numrefs x valueid, n x (valueid, hotness+tailcall flags) 43600b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 43610b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 43620b57cec5SDimitry Andric unsigned FSCallsProfileAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 43630b57cec5SDimitry Andric 43640b57cec5SDimitry Andric // Abbrev for FS_COMBINED_GLOBALVAR_INIT_REFS. 43650b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 43660b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_GLOBALVAR_INIT_REFS)); 43670b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 43680b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // modid 43690b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 43700b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); // valueids 43710b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 43720b57cec5SDimitry Andric unsigned FSModRefsAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 43730b57cec5SDimitry Andric 43740b57cec5SDimitry Andric // Abbrev for FS_COMBINED_ALIAS. 43750b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 43760b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_ALIAS)); 43770b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 43780b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // modid 43790b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 43800b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 43810b57cec5SDimitry Andric unsigned FSAliasAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 43820b57cec5SDimitry Andric 4383bdd1243dSDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4384bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_CALLSITE_INFO)); 4385bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4386bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numstackindices 4387bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numver 4388bdd1243dSDimitry Andric // numstackindices x stackidindex, numver x version 4389bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4390bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4391bdd1243dSDimitry Andric unsigned CallsiteAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4392bdd1243dSDimitry Andric 4393bdd1243dSDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4394bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_ALLOC_INFO)); 4395bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // nummib 4396bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numver 4397bdd1243dSDimitry Andric // nummib x (alloc type, numstackids, numstackids x stackidindex), 4398bdd1243dSDimitry Andric // numver x version 4399bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4400bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4401bdd1243dSDimitry Andric unsigned AllocAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4402bdd1243dSDimitry Andric 44030b57cec5SDimitry Andric // The aliases are emitted as a post-pass, and will point to the value 44040b57cec5SDimitry Andric // id of the aliasee. Save them in a vector for post-processing. 44050b57cec5SDimitry Andric SmallVector<AliasSummary *, 64> Aliases; 44060b57cec5SDimitry Andric 44070b57cec5SDimitry Andric // Save the value id for each summary for alias emission. 44080b57cec5SDimitry Andric DenseMap<const GlobalValueSummary *, unsigned> SummaryToValueIdMap; 44090b57cec5SDimitry Andric 44100b57cec5SDimitry Andric SmallVector<uint64_t, 64> NameVals; 44110b57cec5SDimitry Andric 44120b57cec5SDimitry Andric // Set that will be populated during call to writeFunctionTypeMetadataRecords 44130b57cec5SDimitry Andric // with the type ids referenced by this index file. 44140b57cec5SDimitry Andric std::set<GlobalValue::GUID> ReferencedTypeIds; 44150b57cec5SDimitry Andric 44160b57cec5SDimitry Andric // For local linkage, we also emit the original name separately 44170b57cec5SDimitry Andric // immediately after the record. 44180b57cec5SDimitry Andric auto MaybeEmitOriginalName = [&](GlobalValueSummary &S) { 4419349cc55cSDimitry Andric // We don't need to emit the original name if we are writing the index for 4420349cc55cSDimitry Andric // distributed backends (in which case ModuleToSummariesForIndex is 4421349cc55cSDimitry Andric // non-null). The original name is only needed during the thin link, since 4422349cc55cSDimitry Andric // for SamplePGO the indirect call targets for local functions have 4423349cc55cSDimitry Andric // have the original name annotated in profile. 4424349cc55cSDimitry Andric // Continue to emit it when writing out the entire combined index, which is 4425349cc55cSDimitry Andric // used in testing the thin link via llvm-lto. 4426349cc55cSDimitry Andric if (ModuleToSummariesForIndex || !GlobalValue::isLocalLinkage(S.linkage())) 44270b57cec5SDimitry Andric return; 44280b57cec5SDimitry Andric NameVals.push_back(S.getOriginalName()); 44290b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_COMBINED_ORIGINAL_NAME, NameVals); 44300b57cec5SDimitry Andric NameVals.clear(); 44310b57cec5SDimitry Andric }; 44320b57cec5SDimitry Andric 44330b57cec5SDimitry Andric std::set<GlobalValue::GUID> DefOrUseGUIDs; 44340b57cec5SDimitry Andric forEachSummary([&](GVInfo I, bool IsAliasee) { 44350b57cec5SDimitry Andric GlobalValueSummary *S = I.second; 44360b57cec5SDimitry Andric assert(S); 44370b57cec5SDimitry Andric DefOrUseGUIDs.insert(I.first); 44380b57cec5SDimitry Andric for (const ValueInfo &VI : S->refs()) 44390b57cec5SDimitry Andric DefOrUseGUIDs.insert(VI.getGUID()); 44400b57cec5SDimitry Andric 44410b57cec5SDimitry Andric auto ValueId = getValueId(I.first); 44420b57cec5SDimitry Andric assert(ValueId); 44430b57cec5SDimitry Andric SummaryToValueIdMap[S] = *ValueId; 44440b57cec5SDimitry Andric 44450b57cec5SDimitry Andric // If this is invoked for an aliasee, we want to record the above 44460b57cec5SDimitry Andric // mapping, but then not emit a summary entry (if the aliasee is 44470b57cec5SDimitry Andric // to be imported, we will invoke this separately with IsAliasee=false). 44480b57cec5SDimitry Andric if (IsAliasee) 44490b57cec5SDimitry Andric return; 44500b57cec5SDimitry Andric 44510b57cec5SDimitry Andric if (auto *AS = dyn_cast<AliasSummary>(S)) { 44520b57cec5SDimitry Andric // Will process aliases as a post-pass because the reader wants all 44530b57cec5SDimitry Andric // global to be loaded first. 44540b57cec5SDimitry Andric Aliases.push_back(AS); 44550b57cec5SDimitry Andric return; 44560b57cec5SDimitry Andric } 44570b57cec5SDimitry Andric 44580b57cec5SDimitry Andric if (auto *VS = dyn_cast<GlobalVarSummary>(S)) { 44590b57cec5SDimitry Andric NameVals.push_back(*ValueId); 4460*c9157d92SDimitry Andric assert(ModuleIdMap.count(VS->modulePath())); 4461*c9157d92SDimitry Andric NameVals.push_back(ModuleIdMap[VS->modulePath()]); 44620b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(VS->flags())); 44630b57cec5SDimitry Andric NameVals.push_back(getEncodedGVarFlags(VS->varflags())); 44640b57cec5SDimitry Andric for (auto &RI : VS->refs()) { 44650b57cec5SDimitry Andric auto RefValueId = getValueId(RI.getGUID()); 44660b57cec5SDimitry Andric if (!RefValueId) 44670b57cec5SDimitry Andric continue; 44680b57cec5SDimitry Andric NameVals.push_back(*RefValueId); 44690b57cec5SDimitry Andric } 44700b57cec5SDimitry Andric 44710b57cec5SDimitry Andric // Emit the finished record. 44720b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_COMBINED_GLOBALVAR_INIT_REFS, NameVals, 44730b57cec5SDimitry Andric FSModRefsAbbrev); 44740b57cec5SDimitry Andric NameVals.clear(); 44750b57cec5SDimitry Andric MaybeEmitOriginalName(*S); 44760b57cec5SDimitry Andric return; 44770b57cec5SDimitry Andric } 44780b57cec5SDimitry Andric 4479bdd1243dSDimitry Andric auto GetValueId = [&](const ValueInfo &VI) -> std::optional<unsigned> { 4480bdd1243dSDimitry Andric if (!VI) 4481bdd1243dSDimitry Andric return std::nullopt; 4482349cc55cSDimitry Andric return getValueId(VI.getGUID()); 4483e8d8bef9SDimitry Andric }; 4484e8d8bef9SDimitry Andric 44850b57cec5SDimitry Andric auto *FS = cast<FunctionSummary>(S); 4486e8d8bef9SDimitry Andric writeFunctionTypeMetadataRecords(Stream, FS, GetValueId); 44870b57cec5SDimitry Andric getReferencedTypeIds(FS, ReferencedTypeIds); 44880b57cec5SDimitry Andric 4489bdd1243dSDimitry Andric writeFunctionHeapProfileRecords( 4490bdd1243dSDimitry Andric Stream, FS, CallsiteAbbrev, AllocAbbrev, 4491bdd1243dSDimitry Andric /*PerModule*/ false, 4492bdd1243dSDimitry Andric /*GetValueId*/ [&](const ValueInfo &VI) -> unsigned { 4493bdd1243dSDimitry Andric std::optional<unsigned> ValueID = GetValueId(VI); 4494bdd1243dSDimitry Andric // This can happen in shared index files for distributed ThinLTO if 4495bdd1243dSDimitry Andric // the callee function summary is not included. Record 0 which we 4496bdd1243dSDimitry Andric // will have to deal with conservatively when doing any kind of 4497bdd1243dSDimitry Andric // validation in the ThinLTO backends. 4498bdd1243dSDimitry Andric if (!ValueID) 4499bdd1243dSDimitry Andric return 0; 4500bdd1243dSDimitry Andric return *ValueID; 4501bdd1243dSDimitry Andric }, 4502bdd1243dSDimitry Andric /*GetStackIndex*/ [&](unsigned I) { 4503bdd1243dSDimitry Andric // Get the corresponding index into the list of StackIdIndices 4504bdd1243dSDimitry Andric // actually being written for this combined index (which may be a 4505bdd1243dSDimitry Andric // subset in the case of distributed indexes). 4506bdd1243dSDimitry Andric auto Lower = llvm::lower_bound(StackIdIndices, I); 4507bdd1243dSDimitry Andric return std::distance(StackIdIndices.begin(), Lower); 4508bdd1243dSDimitry Andric }); 4509bdd1243dSDimitry Andric 45100b57cec5SDimitry Andric NameVals.push_back(*ValueId); 4511*c9157d92SDimitry Andric assert(ModuleIdMap.count(FS->modulePath())); 4512*c9157d92SDimitry Andric NameVals.push_back(ModuleIdMap[FS->modulePath()]); 45130b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(FS->flags())); 45140b57cec5SDimitry Andric NameVals.push_back(FS->instCount()); 45150b57cec5SDimitry Andric NameVals.push_back(getEncodedFFlags(FS->fflags())); 45160b57cec5SDimitry Andric NameVals.push_back(FS->entryCount()); 45170b57cec5SDimitry Andric 45180b57cec5SDimitry Andric // Fill in below 45190b57cec5SDimitry Andric NameVals.push_back(0); // numrefs 45200b57cec5SDimitry Andric NameVals.push_back(0); // rorefcnt 45210b57cec5SDimitry Andric NameVals.push_back(0); // worefcnt 45220b57cec5SDimitry Andric 45230b57cec5SDimitry Andric unsigned Count = 0, RORefCnt = 0, WORefCnt = 0; 45240b57cec5SDimitry Andric for (auto &RI : FS->refs()) { 45250b57cec5SDimitry Andric auto RefValueId = getValueId(RI.getGUID()); 45260b57cec5SDimitry Andric if (!RefValueId) 45270b57cec5SDimitry Andric continue; 45280b57cec5SDimitry Andric NameVals.push_back(*RefValueId); 45290b57cec5SDimitry Andric if (RI.isReadOnly()) 45300b57cec5SDimitry Andric RORefCnt++; 45310b57cec5SDimitry Andric else if (RI.isWriteOnly()) 45320b57cec5SDimitry Andric WORefCnt++; 45330b57cec5SDimitry Andric Count++; 45340b57cec5SDimitry Andric } 45350b57cec5SDimitry Andric NameVals[6] = Count; 45360b57cec5SDimitry Andric NameVals[7] = RORefCnt; 45370b57cec5SDimitry Andric NameVals[8] = WORefCnt; 45380b57cec5SDimitry Andric 45390b57cec5SDimitry Andric for (auto &EI : FS->calls()) { 45400b57cec5SDimitry Andric // If this GUID doesn't have a value id, it doesn't have a function 45410b57cec5SDimitry Andric // summary and we don't need to record any calls to it. 4542bdd1243dSDimitry Andric std::optional<unsigned> CallValueId = GetValueId(EI.first); 45430b57cec5SDimitry Andric if (!CallValueId) 45440b57cec5SDimitry Andric continue; 45450b57cec5SDimitry Andric NameVals.push_back(*CallValueId); 4546*c9157d92SDimitry Andric NameVals.push_back(getEncodedHotnessCallEdgeInfo(EI.second)); 45470b57cec5SDimitry Andric } 45480b57cec5SDimitry Andric 45490b57cec5SDimitry Andric // Emit the finished record. 4550*c9157d92SDimitry Andric Stream.EmitRecord(bitc::FS_COMBINED_PROFILE, NameVals, 4551*c9157d92SDimitry Andric FSCallsProfileAbbrev); 45520b57cec5SDimitry Andric NameVals.clear(); 45530b57cec5SDimitry Andric MaybeEmitOriginalName(*S); 45540b57cec5SDimitry Andric }); 45550b57cec5SDimitry Andric 45560b57cec5SDimitry Andric for (auto *AS : Aliases) { 45570b57cec5SDimitry Andric auto AliasValueId = SummaryToValueIdMap[AS]; 45580b57cec5SDimitry Andric assert(AliasValueId); 45590b57cec5SDimitry Andric NameVals.push_back(AliasValueId); 4560*c9157d92SDimitry Andric assert(ModuleIdMap.count(AS->modulePath())); 4561*c9157d92SDimitry Andric NameVals.push_back(ModuleIdMap[AS->modulePath()]); 45620b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(AS->flags())); 45630b57cec5SDimitry Andric auto AliaseeValueId = SummaryToValueIdMap[&AS->getAliasee()]; 45640b57cec5SDimitry Andric assert(AliaseeValueId); 45650b57cec5SDimitry Andric NameVals.push_back(AliaseeValueId); 45660b57cec5SDimitry Andric 45670b57cec5SDimitry Andric // Emit the finished record. 45680b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_COMBINED_ALIAS, NameVals, FSAliasAbbrev); 45690b57cec5SDimitry Andric NameVals.clear(); 45700b57cec5SDimitry Andric MaybeEmitOriginalName(*AS); 45710b57cec5SDimitry Andric 45720b57cec5SDimitry Andric if (auto *FS = dyn_cast<FunctionSummary>(&AS->getAliasee())) 45730b57cec5SDimitry Andric getReferencedTypeIds(FS, ReferencedTypeIds); 45740b57cec5SDimitry Andric } 45750b57cec5SDimitry Andric 45760b57cec5SDimitry Andric if (!Index.cfiFunctionDefs().empty()) { 45770b57cec5SDimitry Andric for (auto &S : Index.cfiFunctionDefs()) { 45780b57cec5SDimitry Andric if (DefOrUseGUIDs.count( 45790b57cec5SDimitry Andric GlobalValue::getGUID(GlobalValue::dropLLVMManglingEscape(S)))) { 45800b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(S)); 45810b57cec5SDimitry Andric NameVals.push_back(S.size()); 45820b57cec5SDimitry Andric } 45830b57cec5SDimitry Andric } 45840b57cec5SDimitry Andric if (!NameVals.empty()) { 45850b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_CFI_FUNCTION_DEFS, NameVals); 45860b57cec5SDimitry Andric NameVals.clear(); 45870b57cec5SDimitry Andric } 45880b57cec5SDimitry Andric } 45890b57cec5SDimitry Andric 45900b57cec5SDimitry Andric if (!Index.cfiFunctionDecls().empty()) { 45910b57cec5SDimitry Andric for (auto &S : Index.cfiFunctionDecls()) { 45920b57cec5SDimitry Andric if (DefOrUseGUIDs.count( 45930b57cec5SDimitry Andric GlobalValue::getGUID(GlobalValue::dropLLVMManglingEscape(S)))) { 45940b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(S)); 45950b57cec5SDimitry Andric NameVals.push_back(S.size()); 45960b57cec5SDimitry Andric } 45970b57cec5SDimitry Andric } 45980b57cec5SDimitry Andric if (!NameVals.empty()) { 45990b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_CFI_FUNCTION_DECLS, NameVals); 46000b57cec5SDimitry Andric NameVals.clear(); 46010b57cec5SDimitry Andric } 46020b57cec5SDimitry Andric } 46030b57cec5SDimitry Andric 46040b57cec5SDimitry Andric // Walk the GUIDs that were referenced, and write the 46050b57cec5SDimitry Andric // corresponding type id records. 46060b57cec5SDimitry Andric for (auto &T : ReferencedTypeIds) { 46070b57cec5SDimitry Andric auto TidIter = Index.typeIds().equal_range(T); 46080b57cec5SDimitry Andric for (auto It = TidIter.first; It != TidIter.second; ++It) { 46090b57cec5SDimitry Andric writeTypeIdSummaryRecord(NameVals, StrtabBuilder, It->second.first, 46100b57cec5SDimitry Andric It->second.second); 46110b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_TYPE_ID, NameVals); 46120b57cec5SDimitry Andric NameVals.clear(); 46130b57cec5SDimitry Andric } 46140b57cec5SDimitry Andric } 46150b57cec5SDimitry Andric 4616fe013be4SDimitry Andric if (Index.getBlockCount()) 46175ffd83dbSDimitry Andric Stream.EmitRecord(bitc::FS_BLOCK_COUNT, 46185ffd83dbSDimitry Andric ArrayRef<uint64_t>{Index.getBlockCount()}); 46195ffd83dbSDimitry Andric 46200b57cec5SDimitry Andric Stream.ExitBlock(); 46210b57cec5SDimitry Andric } 46220b57cec5SDimitry Andric 46230b57cec5SDimitry Andric /// Create the "IDENTIFICATION_BLOCK_ID" containing a single string with the 46240b57cec5SDimitry Andric /// current llvm version, and a record for the epoch number. 46250b57cec5SDimitry Andric static void writeIdentificationBlock(BitstreamWriter &Stream) { 46260b57cec5SDimitry Andric Stream.EnterSubblock(bitc::IDENTIFICATION_BLOCK_ID, 5); 46270b57cec5SDimitry Andric 46280b57cec5SDimitry Andric // Write the "user readable" string identifying the bitcode producer 46290b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 46300b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::IDENTIFICATION_CODE_STRING)); 46310b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 46320b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 46330b57cec5SDimitry Andric auto StringAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 46340b57cec5SDimitry Andric writeStringRecord(Stream, bitc::IDENTIFICATION_CODE_STRING, 46350b57cec5SDimitry Andric "LLVM" LLVM_VERSION_STRING, StringAbbrev); 46360b57cec5SDimitry Andric 46370b57cec5SDimitry Andric // Write the epoch version 46380b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 46390b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::IDENTIFICATION_CODE_EPOCH)); 46400b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 46410b57cec5SDimitry Andric auto EpochAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 46425ffd83dbSDimitry Andric constexpr std::array<unsigned, 1> Vals = {{bitc::BITCODE_CURRENT_EPOCH}}; 46430b57cec5SDimitry Andric Stream.EmitRecord(bitc::IDENTIFICATION_CODE_EPOCH, Vals, EpochAbbrev); 46440b57cec5SDimitry Andric Stream.ExitBlock(); 46450b57cec5SDimitry Andric } 46460b57cec5SDimitry Andric 46470b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleHash(size_t BlockStartPos) { 46480b57cec5SDimitry Andric // Emit the module's hash. 46490b57cec5SDimitry Andric // MODULE_CODE_HASH: [5*i32] 46500b57cec5SDimitry Andric if (GenerateHash) { 46510b57cec5SDimitry Andric uint32_t Vals[5]; 46520b57cec5SDimitry Andric Hasher.update(ArrayRef<uint8_t>((const uint8_t *)&(Buffer)[BlockStartPos], 46530b57cec5SDimitry Andric Buffer.size() - BlockStartPos)); 465481ad6265SDimitry Andric std::array<uint8_t, 20> Hash = Hasher.result(); 46550b57cec5SDimitry Andric for (int Pos = 0; Pos < 20; Pos += 4) { 46560b57cec5SDimitry Andric Vals[Pos / 4] = support::endian::read32be(Hash.data() + Pos); 46570b57cec5SDimitry Andric } 46580b57cec5SDimitry Andric 46590b57cec5SDimitry Andric // Emit the finished record. 46600b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_HASH, Vals); 46610b57cec5SDimitry Andric 46620b57cec5SDimitry Andric if (ModHash) 46630b57cec5SDimitry Andric // Save the written hash value. 46640b57cec5SDimitry Andric llvm::copy(Vals, std::begin(*ModHash)); 46650b57cec5SDimitry Andric } 46660b57cec5SDimitry Andric } 46670b57cec5SDimitry Andric 46680b57cec5SDimitry Andric void ModuleBitcodeWriter::write() { 46690b57cec5SDimitry Andric writeIdentificationBlock(Stream); 46700b57cec5SDimitry Andric 46710b57cec5SDimitry Andric Stream.EnterSubblock(bitc::MODULE_BLOCK_ID, 3); 46720b57cec5SDimitry Andric size_t BlockStartPos = Buffer.size(); 46730b57cec5SDimitry Andric 46740b57cec5SDimitry Andric writeModuleVersion(); 46750b57cec5SDimitry Andric 46760b57cec5SDimitry Andric // Emit blockinfo, which defines the standard abbreviations etc. 46770b57cec5SDimitry Andric writeBlockInfo(); 46780b57cec5SDimitry Andric 46790b57cec5SDimitry Andric // Emit information describing all of the types in the module. 46800b57cec5SDimitry Andric writeTypeTable(); 46810b57cec5SDimitry Andric 46820b57cec5SDimitry Andric // Emit information about attribute groups. 46830b57cec5SDimitry Andric writeAttributeGroupTable(); 46840b57cec5SDimitry Andric 46850b57cec5SDimitry Andric // Emit information about parameter attributes. 46860b57cec5SDimitry Andric writeAttributeTable(); 46870b57cec5SDimitry Andric 46880b57cec5SDimitry Andric writeComdats(); 46890b57cec5SDimitry Andric 46900b57cec5SDimitry Andric // Emit top-level description of module, including target triple, inline asm, 46910b57cec5SDimitry Andric // descriptors for global variables, and function prototype info. 46920b57cec5SDimitry Andric writeModuleInfo(); 46930b57cec5SDimitry Andric 46940b57cec5SDimitry Andric // Emit constants. 46950b57cec5SDimitry Andric writeModuleConstants(); 46960b57cec5SDimitry Andric 46970b57cec5SDimitry Andric // Emit metadata kind names. 46980b57cec5SDimitry Andric writeModuleMetadataKinds(); 46990b57cec5SDimitry Andric 47000b57cec5SDimitry Andric // Emit metadata. 47010b57cec5SDimitry Andric writeModuleMetadata(); 47020b57cec5SDimitry Andric 47030b57cec5SDimitry Andric // Emit module-level use-lists. 47040b57cec5SDimitry Andric if (VE.shouldPreserveUseListOrder()) 47050b57cec5SDimitry Andric writeUseListBlock(nullptr); 47060b57cec5SDimitry Andric 47070b57cec5SDimitry Andric writeOperandBundleTags(); 47080b57cec5SDimitry Andric writeSyncScopeNames(); 47090b57cec5SDimitry Andric 47100b57cec5SDimitry Andric // Emit function bodies. 47110b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> FunctionToBitcodeIndex; 47124824e7fdSDimitry Andric for (const Function &F : M) 47134824e7fdSDimitry Andric if (!F.isDeclaration()) 47144824e7fdSDimitry Andric writeFunction(F, FunctionToBitcodeIndex); 47150b57cec5SDimitry Andric 47160b57cec5SDimitry Andric // Need to write after the above call to WriteFunction which populates 47170b57cec5SDimitry Andric // the summary information in the index. 47180b57cec5SDimitry Andric if (Index) 47190b57cec5SDimitry Andric writePerModuleGlobalValueSummary(); 47200b57cec5SDimitry Andric 47210b57cec5SDimitry Andric writeGlobalValueSymbolTable(FunctionToBitcodeIndex); 47220b57cec5SDimitry Andric 47230b57cec5SDimitry Andric writeModuleHash(BlockStartPos); 47240b57cec5SDimitry Andric 47250b57cec5SDimitry Andric Stream.ExitBlock(); 47260b57cec5SDimitry Andric } 47270b57cec5SDimitry Andric 47280b57cec5SDimitry Andric static void writeInt32ToBuffer(uint32_t Value, SmallVectorImpl<char> &Buffer, 47290b57cec5SDimitry Andric uint32_t &Position) { 47300b57cec5SDimitry Andric support::endian::write32le(&Buffer[Position], Value); 47310b57cec5SDimitry Andric Position += 4; 47320b57cec5SDimitry Andric } 47330b57cec5SDimitry Andric 47340b57cec5SDimitry Andric /// If generating a bc file on darwin, we have to emit a 47350b57cec5SDimitry Andric /// header and trailer to make it compatible with the system archiver. To do 47360b57cec5SDimitry Andric /// this we emit the following header, and then emit a trailer that pads the 47370b57cec5SDimitry Andric /// file out to be a multiple of 16 bytes. 47380b57cec5SDimitry Andric /// 47390b57cec5SDimitry Andric /// struct bc_header { 47400b57cec5SDimitry Andric /// uint32_t Magic; // 0x0B17C0DE 47410b57cec5SDimitry Andric /// uint32_t Version; // Version, currently always 0. 47420b57cec5SDimitry Andric /// uint32_t BitcodeOffset; // Offset to traditional bitcode file. 47430b57cec5SDimitry Andric /// uint32_t BitcodeSize; // Size of traditional bitcode file. 47440b57cec5SDimitry Andric /// uint32_t CPUType; // CPU specifier. 47450b57cec5SDimitry Andric /// ... potentially more later ... 47460b57cec5SDimitry Andric /// }; 47470b57cec5SDimitry Andric static void emitDarwinBCHeaderAndTrailer(SmallVectorImpl<char> &Buffer, 47480b57cec5SDimitry Andric const Triple &TT) { 47490b57cec5SDimitry Andric unsigned CPUType = ~0U; 47500b57cec5SDimitry Andric 47510b57cec5SDimitry Andric // Match x86_64-*, i[3-9]86-*, powerpc-*, powerpc64-*, arm-*, thumb-*, 47520b57cec5SDimitry Andric // armv[0-9]-*, thumbv[0-9]-*, armv5te-*, or armv6t2-*. The CPUType is a magic 47530b57cec5SDimitry Andric // number from /usr/include/mach/machine.h. It is ok to reproduce the 47540b57cec5SDimitry Andric // specific constants here because they are implicitly part of the Darwin ABI. 47550b57cec5SDimitry Andric enum { 47560b57cec5SDimitry Andric DARWIN_CPU_ARCH_ABI64 = 0x01000000, 47570b57cec5SDimitry Andric DARWIN_CPU_TYPE_X86 = 7, 47580b57cec5SDimitry Andric DARWIN_CPU_TYPE_ARM = 12, 47590b57cec5SDimitry Andric DARWIN_CPU_TYPE_POWERPC = 18 47600b57cec5SDimitry Andric }; 47610b57cec5SDimitry Andric 47620b57cec5SDimitry Andric Triple::ArchType Arch = TT.getArch(); 47630b57cec5SDimitry Andric if (Arch == Triple::x86_64) 47640b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_X86 | DARWIN_CPU_ARCH_ABI64; 47650b57cec5SDimitry Andric else if (Arch == Triple::x86) 47660b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_X86; 47670b57cec5SDimitry Andric else if (Arch == Triple::ppc) 47680b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_POWERPC; 47690b57cec5SDimitry Andric else if (Arch == Triple::ppc64) 47700b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_POWERPC | DARWIN_CPU_ARCH_ABI64; 47710b57cec5SDimitry Andric else if (Arch == Triple::arm || Arch == Triple::thumb) 47720b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_ARM; 47730b57cec5SDimitry Andric 47740b57cec5SDimitry Andric // Traditional Bitcode starts after header. 47750b57cec5SDimitry Andric assert(Buffer.size() >= BWH_HeaderSize && 47760b57cec5SDimitry Andric "Expected header size to be reserved"); 47770b57cec5SDimitry Andric unsigned BCOffset = BWH_HeaderSize; 47780b57cec5SDimitry Andric unsigned BCSize = Buffer.size() - BWH_HeaderSize; 47790b57cec5SDimitry Andric 47800b57cec5SDimitry Andric // Write the magic and version. 47810b57cec5SDimitry Andric unsigned Position = 0; 47820b57cec5SDimitry Andric writeInt32ToBuffer(0x0B17C0DE, Buffer, Position); 47830b57cec5SDimitry Andric writeInt32ToBuffer(0, Buffer, Position); // Version. 47840b57cec5SDimitry Andric writeInt32ToBuffer(BCOffset, Buffer, Position); 47850b57cec5SDimitry Andric writeInt32ToBuffer(BCSize, Buffer, Position); 47860b57cec5SDimitry Andric writeInt32ToBuffer(CPUType, Buffer, Position); 47870b57cec5SDimitry Andric 47880b57cec5SDimitry Andric // If the file is not a multiple of 16 bytes, insert dummy padding. 47890b57cec5SDimitry Andric while (Buffer.size() & 15) 47900b57cec5SDimitry Andric Buffer.push_back(0); 47910b57cec5SDimitry Andric } 47920b57cec5SDimitry Andric 47930b57cec5SDimitry Andric /// Helper to write the header common to all bitcode files. 47940b57cec5SDimitry Andric static void writeBitcodeHeader(BitstreamWriter &Stream) { 47950b57cec5SDimitry Andric // Emit the file header. 47960b57cec5SDimitry Andric Stream.Emit((unsigned)'B', 8); 47970b57cec5SDimitry Andric Stream.Emit((unsigned)'C', 8); 47980b57cec5SDimitry Andric Stream.Emit(0x0, 4); 47990b57cec5SDimitry Andric Stream.Emit(0xC, 4); 48000b57cec5SDimitry Andric Stream.Emit(0xE, 4); 48010b57cec5SDimitry Andric Stream.Emit(0xD, 4); 48020b57cec5SDimitry Andric } 48030b57cec5SDimitry Andric 4804e8d8bef9SDimitry Andric BitcodeWriter::BitcodeWriter(SmallVectorImpl<char> &Buffer, raw_fd_stream *FS) 4805e8d8bef9SDimitry Andric : Buffer(Buffer), Stream(new BitstreamWriter(Buffer, FS, FlushThreshold)) { 48060b57cec5SDimitry Andric writeBitcodeHeader(*Stream); 48070b57cec5SDimitry Andric } 48080b57cec5SDimitry Andric 48090b57cec5SDimitry Andric BitcodeWriter::~BitcodeWriter() { assert(WroteStrtab); } 48100b57cec5SDimitry Andric 48110b57cec5SDimitry Andric void BitcodeWriter::writeBlob(unsigned Block, unsigned Record, StringRef Blob) { 48120b57cec5SDimitry Andric Stream->EnterSubblock(Block, 3); 48130b57cec5SDimitry Andric 48140b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 48150b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(Record)); 48160b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); 48170b57cec5SDimitry Andric auto AbbrevNo = Stream->EmitAbbrev(std::move(Abbv)); 48180b57cec5SDimitry Andric 48190b57cec5SDimitry Andric Stream->EmitRecordWithBlob(AbbrevNo, ArrayRef<uint64_t>{Record}, Blob); 48200b57cec5SDimitry Andric 48210b57cec5SDimitry Andric Stream->ExitBlock(); 48220b57cec5SDimitry Andric } 48230b57cec5SDimitry Andric 48240b57cec5SDimitry Andric void BitcodeWriter::writeSymtab() { 48250b57cec5SDimitry Andric assert(!WroteStrtab && !WroteSymtab); 48260b57cec5SDimitry Andric 48270b57cec5SDimitry Andric // If any module has module-level inline asm, we will require a registered asm 48280b57cec5SDimitry Andric // parser for the target so that we can create an accurate symbol table for 48290b57cec5SDimitry Andric // the module. 48300b57cec5SDimitry Andric for (Module *M : Mods) { 48310b57cec5SDimitry Andric if (M->getModuleInlineAsm().empty()) 48320b57cec5SDimitry Andric continue; 48330b57cec5SDimitry Andric 48340b57cec5SDimitry Andric std::string Err; 48350b57cec5SDimitry Andric const Triple TT(M->getTargetTriple()); 48360b57cec5SDimitry Andric const Target *T = TargetRegistry::lookupTarget(TT.str(), Err); 48370b57cec5SDimitry Andric if (!T || !T->hasMCAsmParser()) 48380b57cec5SDimitry Andric return; 48390b57cec5SDimitry Andric } 48400b57cec5SDimitry Andric 48410b57cec5SDimitry Andric WroteSymtab = true; 48420b57cec5SDimitry Andric SmallVector<char, 0> Symtab; 48430b57cec5SDimitry Andric // The irsymtab::build function may be unable to create a symbol table if the 48440b57cec5SDimitry Andric // module is malformed (e.g. it contains an invalid alias). Writing a symbol 48450b57cec5SDimitry Andric // table is not required for correctness, but we still want to be able to 48460b57cec5SDimitry Andric // write malformed modules to bitcode files, so swallow the error. 48470b57cec5SDimitry Andric if (Error E = irsymtab::build(Mods, Symtab, StrtabBuilder, Alloc)) { 48480b57cec5SDimitry Andric consumeError(std::move(E)); 48490b57cec5SDimitry Andric return; 48500b57cec5SDimitry Andric } 48510b57cec5SDimitry Andric 48520b57cec5SDimitry Andric writeBlob(bitc::SYMTAB_BLOCK_ID, bitc::SYMTAB_BLOB, 48530b57cec5SDimitry Andric {Symtab.data(), Symtab.size()}); 48540b57cec5SDimitry Andric } 48550b57cec5SDimitry Andric 48560b57cec5SDimitry Andric void BitcodeWriter::writeStrtab() { 48570b57cec5SDimitry Andric assert(!WroteStrtab); 48580b57cec5SDimitry Andric 48590b57cec5SDimitry Andric std::vector<char> Strtab; 48600b57cec5SDimitry Andric StrtabBuilder.finalizeInOrder(); 48610b57cec5SDimitry Andric Strtab.resize(StrtabBuilder.getSize()); 48620b57cec5SDimitry Andric StrtabBuilder.write((uint8_t *)Strtab.data()); 48630b57cec5SDimitry Andric 48640b57cec5SDimitry Andric writeBlob(bitc::STRTAB_BLOCK_ID, bitc::STRTAB_BLOB, 48650b57cec5SDimitry Andric {Strtab.data(), Strtab.size()}); 48660b57cec5SDimitry Andric 48670b57cec5SDimitry Andric WroteStrtab = true; 48680b57cec5SDimitry Andric } 48690b57cec5SDimitry Andric 48700b57cec5SDimitry Andric void BitcodeWriter::copyStrtab(StringRef Strtab) { 48710b57cec5SDimitry Andric writeBlob(bitc::STRTAB_BLOCK_ID, bitc::STRTAB_BLOB, Strtab); 48720b57cec5SDimitry Andric WroteStrtab = true; 48730b57cec5SDimitry Andric } 48740b57cec5SDimitry Andric 48750b57cec5SDimitry Andric void BitcodeWriter::writeModule(const Module &M, 48760b57cec5SDimitry Andric bool ShouldPreserveUseListOrder, 48770b57cec5SDimitry Andric const ModuleSummaryIndex *Index, 48780b57cec5SDimitry Andric bool GenerateHash, ModuleHash *ModHash) { 48790b57cec5SDimitry Andric assert(!WroteStrtab); 48800b57cec5SDimitry Andric 48810b57cec5SDimitry Andric // The Mods vector is used by irsymtab::build, which requires non-const 48820b57cec5SDimitry Andric // Modules in case it needs to materialize metadata. But the bitcode writer 48830b57cec5SDimitry Andric // requires that the module is materialized, so we can cast to non-const here, 48840b57cec5SDimitry Andric // after checking that it is in fact materialized. 48850b57cec5SDimitry Andric assert(M.isMaterialized()); 48860b57cec5SDimitry Andric Mods.push_back(const_cast<Module *>(&M)); 48870b57cec5SDimitry Andric 48880b57cec5SDimitry Andric ModuleBitcodeWriter ModuleWriter(M, Buffer, StrtabBuilder, *Stream, 48890b57cec5SDimitry Andric ShouldPreserveUseListOrder, Index, 48900b57cec5SDimitry Andric GenerateHash, ModHash); 48910b57cec5SDimitry Andric ModuleWriter.write(); 48920b57cec5SDimitry Andric } 48930b57cec5SDimitry Andric 48940b57cec5SDimitry Andric void BitcodeWriter::writeIndex( 48950b57cec5SDimitry Andric const ModuleSummaryIndex *Index, 48960b57cec5SDimitry Andric const std::map<std::string, GVSummaryMapTy> *ModuleToSummariesForIndex) { 48970b57cec5SDimitry Andric IndexBitcodeWriter IndexWriter(*Stream, StrtabBuilder, *Index, 48980b57cec5SDimitry Andric ModuleToSummariesForIndex); 48990b57cec5SDimitry Andric IndexWriter.write(); 49000b57cec5SDimitry Andric } 49010b57cec5SDimitry Andric 49020b57cec5SDimitry Andric /// Write the specified module to the specified output stream. 49030b57cec5SDimitry Andric void llvm::WriteBitcodeToFile(const Module &M, raw_ostream &Out, 49040b57cec5SDimitry Andric bool ShouldPreserveUseListOrder, 49050b57cec5SDimitry Andric const ModuleSummaryIndex *Index, 49060b57cec5SDimitry Andric bool GenerateHash, ModuleHash *ModHash) { 49070b57cec5SDimitry Andric SmallVector<char, 0> Buffer; 49080b57cec5SDimitry Andric Buffer.reserve(256*1024); 49090b57cec5SDimitry Andric 49100b57cec5SDimitry Andric // If this is darwin or another generic macho target, reserve space for the 49110b57cec5SDimitry Andric // header. 49120b57cec5SDimitry Andric Triple TT(M.getTargetTriple()); 49130b57cec5SDimitry Andric if (TT.isOSDarwin() || TT.isOSBinFormatMachO()) 49140b57cec5SDimitry Andric Buffer.insert(Buffer.begin(), BWH_HeaderSize, 0); 49150b57cec5SDimitry Andric 4916e8d8bef9SDimitry Andric BitcodeWriter Writer(Buffer, dyn_cast<raw_fd_stream>(&Out)); 49170b57cec5SDimitry Andric Writer.writeModule(M, ShouldPreserveUseListOrder, Index, GenerateHash, 49180b57cec5SDimitry Andric ModHash); 49190b57cec5SDimitry Andric Writer.writeSymtab(); 49200b57cec5SDimitry Andric Writer.writeStrtab(); 49210b57cec5SDimitry Andric 49220b57cec5SDimitry Andric if (TT.isOSDarwin() || TT.isOSBinFormatMachO()) 49230b57cec5SDimitry Andric emitDarwinBCHeaderAndTrailer(Buffer, TT); 49240b57cec5SDimitry Andric 49250b57cec5SDimitry Andric // Write the generated bitstream to "Out". 4926e8d8bef9SDimitry Andric if (!Buffer.empty()) 49270b57cec5SDimitry Andric Out.write((char *)&Buffer.front(), Buffer.size()); 49280b57cec5SDimitry Andric } 49290b57cec5SDimitry Andric 49300b57cec5SDimitry Andric void IndexBitcodeWriter::write() { 49310b57cec5SDimitry Andric Stream.EnterSubblock(bitc::MODULE_BLOCK_ID, 3); 49320b57cec5SDimitry Andric 49330b57cec5SDimitry Andric writeModuleVersion(); 49340b57cec5SDimitry Andric 49350b57cec5SDimitry Andric // Write the module paths in the combined index. 49360b57cec5SDimitry Andric writeModStrings(); 49370b57cec5SDimitry Andric 49380b57cec5SDimitry Andric // Write the summary combined index records. 49390b57cec5SDimitry Andric writeCombinedGlobalValueSummary(); 49400b57cec5SDimitry Andric 49410b57cec5SDimitry Andric Stream.ExitBlock(); 49420b57cec5SDimitry Andric } 49430b57cec5SDimitry Andric 49440b57cec5SDimitry Andric // Write the specified module summary index to the given raw output stream, 49450b57cec5SDimitry Andric // where it will be written in a new bitcode block. This is used when 49460b57cec5SDimitry Andric // writing the combined index file for ThinLTO. When writing a subset of the 49470b57cec5SDimitry Andric // index for a distributed backend, provide a \p ModuleToSummariesForIndex map. 49481fd87a68SDimitry Andric void llvm::writeIndexToFile( 49490b57cec5SDimitry Andric const ModuleSummaryIndex &Index, raw_ostream &Out, 49500b57cec5SDimitry Andric const std::map<std::string, GVSummaryMapTy> *ModuleToSummariesForIndex) { 49510b57cec5SDimitry Andric SmallVector<char, 0> Buffer; 49520b57cec5SDimitry Andric Buffer.reserve(256 * 1024); 49530b57cec5SDimitry Andric 49540b57cec5SDimitry Andric BitcodeWriter Writer(Buffer); 49550b57cec5SDimitry Andric Writer.writeIndex(&Index, ModuleToSummariesForIndex); 49560b57cec5SDimitry Andric Writer.writeStrtab(); 49570b57cec5SDimitry Andric 49580b57cec5SDimitry Andric Out.write((char *)&Buffer.front(), Buffer.size()); 49590b57cec5SDimitry Andric } 49600b57cec5SDimitry Andric 49610b57cec5SDimitry Andric namespace { 49620b57cec5SDimitry Andric 49630b57cec5SDimitry Andric /// Class to manage the bitcode writing for a thin link bitcode file. 49640b57cec5SDimitry Andric class ThinLinkBitcodeWriter : public ModuleBitcodeWriterBase { 49650b57cec5SDimitry Andric /// ModHash is for use in ThinLTO incremental build, generated while writing 49660b57cec5SDimitry Andric /// the module bitcode file. 49670b57cec5SDimitry Andric const ModuleHash *ModHash; 49680b57cec5SDimitry Andric 49690b57cec5SDimitry Andric public: 49700b57cec5SDimitry Andric ThinLinkBitcodeWriter(const Module &M, StringTableBuilder &StrtabBuilder, 49710b57cec5SDimitry Andric BitstreamWriter &Stream, 49720b57cec5SDimitry Andric const ModuleSummaryIndex &Index, 49730b57cec5SDimitry Andric const ModuleHash &ModHash) 49740b57cec5SDimitry Andric : ModuleBitcodeWriterBase(M, StrtabBuilder, Stream, 49750b57cec5SDimitry Andric /*ShouldPreserveUseListOrder=*/false, &Index), 49760b57cec5SDimitry Andric ModHash(&ModHash) {} 49770b57cec5SDimitry Andric 49780b57cec5SDimitry Andric void write(); 49790b57cec5SDimitry Andric 49800b57cec5SDimitry Andric private: 49810b57cec5SDimitry Andric void writeSimplifiedModuleInfo(); 49820b57cec5SDimitry Andric }; 49830b57cec5SDimitry Andric 49840b57cec5SDimitry Andric } // end anonymous namespace 49850b57cec5SDimitry Andric 49860b57cec5SDimitry Andric // This function writes a simpilified module info for thin link bitcode file. 49870b57cec5SDimitry Andric // It only contains the source file name along with the name(the offset and 49880b57cec5SDimitry Andric // size in strtab) and linkage for global values. For the global value info 49890b57cec5SDimitry Andric // entry, in order to keep linkage at offset 5, there are three zeros used 49900b57cec5SDimitry Andric // as padding. 49910b57cec5SDimitry Andric void ThinLinkBitcodeWriter::writeSimplifiedModuleInfo() { 49920b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 49930b57cec5SDimitry Andric // Emit the module's source file name. 49940b57cec5SDimitry Andric { 49950b57cec5SDimitry Andric StringEncoding Bits = getStringEncoding(M.getSourceFileName()); 49960b57cec5SDimitry Andric BitCodeAbbrevOp AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8); 49970b57cec5SDimitry Andric if (Bits == SE_Char6) 49980b57cec5SDimitry Andric AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Char6); 49990b57cec5SDimitry Andric else if (Bits == SE_Fixed7) 50000b57cec5SDimitry Andric AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7); 50010b57cec5SDimitry Andric 50020b57cec5SDimitry Andric // MODULE_CODE_SOURCE_FILENAME: [namechar x N] 50030b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 50040b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_SOURCE_FILENAME)); 50050b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 50060b57cec5SDimitry Andric Abbv->Add(AbbrevOpToUse); 50070b57cec5SDimitry Andric unsigned FilenameAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 50080b57cec5SDimitry Andric 50090b57cec5SDimitry Andric for (const auto P : M.getSourceFileName()) 50100b57cec5SDimitry Andric Vals.push_back((unsigned char)P); 50110b57cec5SDimitry Andric 50120b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_SOURCE_FILENAME, Vals, FilenameAbbrev); 50130b57cec5SDimitry Andric Vals.clear(); 50140b57cec5SDimitry Andric } 50150b57cec5SDimitry Andric 50160b57cec5SDimitry Andric // Emit the global variable information. 50170b57cec5SDimitry Andric for (const GlobalVariable &GV : M.globals()) { 50180b57cec5SDimitry Andric // GLOBALVAR: [strtab offset, strtab size, 0, 0, 0, linkage] 50190b57cec5SDimitry Andric Vals.push_back(StrtabBuilder.add(GV.getName())); 50200b57cec5SDimitry Andric Vals.push_back(GV.getName().size()); 50210b57cec5SDimitry Andric Vals.push_back(0); 50220b57cec5SDimitry Andric Vals.push_back(0); 50230b57cec5SDimitry Andric Vals.push_back(0); 50240b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(GV)); 50250b57cec5SDimitry Andric 50260b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_GLOBALVAR, Vals); 50270b57cec5SDimitry Andric Vals.clear(); 50280b57cec5SDimitry Andric } 50290b57cec5SDimitry Andric 50300b57cec5SDimitry Andric // Emit the function proto information. 50310b57cec5SDimitry Andric for (const Function &F : M) { 50320b57cec5SDimitry Andric // FUNCTION: [strtab offset, strtab size, 0, 0, 0, linkage] 50330b57cec5SDimitry Andric Vals.push_back(StrtabBuilder.add(F.getName())); 50340b57cec5SDimitry Andric Vals.push_back(F.getName().size()); 50350b57cec5SDimitry Andric Vals.push_back(0); 50360b57cec5SDimitry Andric Vals.push_back(0); 50370b57cec5SDimitry Andric Vals.push_back(0); 50380b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(F)); 50390b57cec5SDimitry Andric 50400b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_FUNCTION, Vals); 50410b57cec5SDimitry Andric Vals.clear(); 50420b57cec5SDimitry Andric } 50430b57cec5SDimitry Andric 50440b57cec5SDimitry Andric // Emit the alias information. 50450b57cec5SDimitry Andric for (const GlobalAlias &A : M.aliases()) { 50460b57cec5SDimitry Andric // ALIAS: [strtab offset, strtab size, 0, 0, 0, linkage] 50470b57cec5SDimitry Andric Vals.push_back(StrtabBuilder.add(A.getName())); 50480b57cec5SDimitry Andric Vals.push_back(A.getName().size()); 50490b57cec5SDimitry Andric Vals.push_back(0); 50500b57cec5SDimitry Andric Vals.push_back(0); 50510b57cec5SDimitry Andric Vals.push_back(0); 50520b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(A)); 50530b57cec5SDimitry Andric 50540b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_ALIAS, Vals); 50550b57cec5SDimitry Andric Vals.clear(); 50560b57cec5SDimitry Andric } 50570b57cec5SDimitry Andric 50580b57cec5SDimitry Andric // Emit the ifunc information. 50590b57cec5SDimitry Andric for (const GlobalIFunc &I : M.ifuncs()) { 50600b57cec5SDimitry Andric // IFUNC: [strtab offset, strtab size, 0, 0, 0, linkage] 50610b57cec5SDimitry Andric Vals.push_back(StrtabBuilder.add(I.getName())); 50620b57cec5SDimitry Andric Vals.push_back(I.getName().size()); 50630b57cec5SDimitry Andric Vals.push_back(0); 50640b57cec5SDimitry Andric Vals.push_back(0); 50650b57cec5SDimitry Andric Vals.push_back(0); 50660b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(I)); 50670b57cec5SDimitry Andric 50680b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_IFUNC, Vals); 50690b57cec5SDimitry Andric Vals.clear(); 50700b57cec5SDimitry Andric } 50710b57cec5SDimitry Andric } 50720b57cec5SDimitry Andric 50730b57cec5SDimitry Andric void ThinLinkBitcodeWriter::write() { 50740b57cec5SDimitry Andric Stream.EnterSubblock(bitc::MODULE_BLOCK_ID, 3); 50750b57cec5SDimitry Andric 50760b57cec5SDimitry Andric writeModuleVersion(); 50770b57cec5SDimitry Andric 50780b57cec5SDimitry Andric writeSimplifiedModuleInfo(); 50790b57cec5SDimitry Andric 50800b57cec5SDimitry Andric writePerModuleGlobalValueSummary(); 50810b57cec5SDimitry Andric 50820b57cec5SDimitry Andric // Write module hash. 50830b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_HASH, ArrayRef<uint32_t>(*ModHash)); 50840b57cec5SDimitry Andric 50850b57cec5SDimitry Andric Stream.ExitBlock(); 50860b57cec5SDimitry Andric } 50870b57cec5SDimitry Andric 50880b57cec5SDimitry Andric void BitcodeWriter::writeThinLinkBitcode(const Module &M, 50890b57cec5SDimitry Andric const ModuleSummaryIndex &Index, 50900b57cec5SDimitry Andric const ModuleHash &ModHash) { 50910b57cec5SDimitry Andric assert(!WroteStrtab); 50920b57cec5SDimitry Andric 50930b57cec5SDimitry Andric // The Mods vector is used by irsymtab::build, which requires non-const 50940b57cec5SDimitry Andric // Modules in case it needs to materialize metadata. But the bitcode writer 50950b57cec5SDimitry Andric // requires that the module is materialized, so we can cast to non-const here, 50960b57cec5SDimitry Andric // after checking that it is in fact materialized. 50970b57cec5SDimitry Andric assert(M.isMaterialized()); 50980b57cec5SDimitry Andric Mods.push_back(const_cast<Module *>(&M)); 50990b57cec5SDimitry Andric 51000b57cec5SDimitry Andric ThinLinkBitcodeWriter ThinLinkWriter(M, StrtabBuilder, *Stream, Index, 51010b57cec5SDimitry Andric ModHash); 51020b57cec5SDimitry Andric ThinLinkWriter.write(); 51030b57cec5SDimitry Andric } 51040b57cec5SDimitry Andric 51050b57cec5SDimitry Andric // Write the specified thin link bitcode file to the given raw output stream, 51060b57cec5SDimitry Andric // where it will be written in a new bitcode block. This is used when 51070b57cec5SDimitry Andric // writing the per-module index file for ThinLTO. 51081fd87a68SDimitry Andric void llvm::writeThinLinkBitcodeToFile(const Module &M, raw_ostream &Out, 51090b57cec5SDimitry Andric const ModuleSummaryIndex &Index, 51100b57cec5SDimitry Andric const ModuleHash &ModHash) { 51110b57cec5SDimitry Andric SmallVector<char, 0> Buffer; 51120b57cec5SDimitry Andric Buffer.reserve(256 * 1024); 51130b57cec5SDimitry Andric 51140b57cec5SDimitry Andric BitcodeWriter Writer(Buffer); 51150b57cec5SDimitry Andric Writer.writeThinLinkBitcode(M, Index, ModHash); 51160b57cec5SDimitry Andric Writer.writeSymtab(); 51170b57cec5SDimitry Andric Writer.writeStrtab(); 51180b57cec5SDimitry Andric 51190b57cec5SDimitry Andric Out.write((char *)&Buffer.front(), Buffer.size()); 51200b57cec5SDimitry Andric } 5121480093f4SDimitry Andric 5122480093f4SDimitry Andric static const char *getSectionNameForBitcode(const Triple &T) { 5123480093f4SDimitry Andric switch (T.getObjectFormat()) { 5124480093f4SDimitry Andric case Triple::MachO: 5125480093f4SDimitry Andric return "__LLVM,__bitcode"; 5126480093f4SDimitry Andric case Triple::COFF: 5127480093f4SDimitry Andric case Triple::ELF: 5128480093f4SDimitry Andric case Triple::Wasm: 5129480093f4SDimitry Andric case Triple::UnknownObjectFormat: 5130480093f4SDimitry Andric return ".llvmbc"; 5131e8d8bef9SDimitry Andric case Triple::GOFF: 5132e8d8bef9SDimitry Andric llvm_unreachable("GOFF is not yet implemented"); 5133e8d8bef9SDimitry Andric break; 513481ad6265SDimitry Andric case Triple::SPIRV: 513581ad6265SDimitry Andric llvm_unreachable("SPIRV is not yet implemented"); 513681ad6265SDimitry Andric break; 5137480093f4SDimitry Andric case Triple::XCOFF: 5138480093f4SDimitry Andric llvm_unreachable("XCOFF is not yet implemented"); 5139480093f4SDimitry Andric break; 514081ad6265SDimitry Andric case Triple::DXContainer: 514181ad6265SDimitry Andric llvm_unreachable("DXContainer is not yet implemented"); 514281ad6265SDimitry Andric break; 5143480093f4SDimitry Andric } 5144480093f4SDimitry Andric llvm_unreachable("Unimplemented ObjectFormatType"); 5145480093f4SDimitry Andric } 5146480093f4SDimitry Andric 5147480093f4SDimitry Andric static const char *getSectionNameForCommandline(const Triple &T) { 5148480093f4SDimitry Andric switch (T.getObjectFormat()) { 5149480093f4SDimitry Andric case Triple::MachO: 5150480093f4SDimitry Andric return "__LLVM,__cmdline"; 5151480093f4SDimitry Andric case Triple::COFF: 5152480093f4SDimitry Andric case Triple::ELF: 5153480093f4SDimitry Andric case Triple::Wasm: 5154480093f4SDimitry Andric case Triple::UnknownObjectFormat: 5155480093f4SDimitry Andric return ".llvmcmd"; 5156e8d8bef9SDimitry Andric case Triple::GOFF: 5157e8d8bef9SDimitry Andric llvm_unreachable("GOFF is not yet implemented"); 5158e8d8bef9SDimitry Andric break; 515981ad6265SDimitry Andric case Triple::SPIRV: 516081ad6265SDimitry Andric llvm_unreachable("SPIRV is not yet implemented"); 516181ad6265SDimitry Andric break; 5162480093f4SDimitry Andric case Triple::XCOFF: 5163480093f4SDimitry Andric llvm_unreachable("XCOFF is not yet implemented"); 5164480093f4SDimitry Andric break; 516581ad6265SDimitry Andric case Triple::DXContainer: 516681ad6265SDimitry Andric llvm_unreachable("DXC is not yet implemented"); 516781ad6265SDimitry Andric break; 5168480093f4SDimitry Andric } 5169480093f4SDimitry Andric llvm_unreachable("Unimplemented ObjectFormatType"); 5170480093f4SDimitry Andric } 5171480093f4SDimitry Andric 51721fd87a68SDimitry Andric void llvm::embedBitcodeInModule(llvm::Module &M, llvm::MemoryBufferRef Buf, 5173e8d8bef9SDimitry Andric bool EmbedBitcode, bool EmbedCmdline, 5174e8d8bef9SDimitry Andric const std::vector<uint8_t> &CmdArgs) { 5175480093f4SDimitry Andric // Save llvm.compiler.used and remove it. 5176480093f4SDimitry Andric SmallVector<Constant *, 2> UsedArray; 5177fe6060f1SDimitry Andric SmallVector<GlobalValue *, 4> UsedGlobals; 5178*c9157d92SDimitry Andric Type *UsedElementType = PointerType::getUnqual(M.getContext()); 5179480093f4SDimitry Andric GlobalVariable *Used = collectUsedGlobalVariables(M, UsedGlobals, true); 5180480093f4SDimitry Andric for (auto *GV : UsedGlobals) { 5181480093f4SDimitry Andric if (GV->getName() != "llvm.embedded.module" && 5182480093f4SDimitry Andric GV->getName() != "llvm.cmdline") 5183480093f4SDimitry Andric UsedArray.push_back( 5184480093f4SDimitry Andric ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV, UsedElementType)); 5185480093f4SDimitry Andric } 5186480093f4SDimitry Andric if (Used) 5187480093f4SDimitry Andric Used->eraseFromParent(); 5188480093f4SDimitry Andric 5189480093f4SDimitry Andric // Embed the bitcode for the llvm module. 5190480093f4SDimitry Andric std::string Data; 5191480093f4SDimitry Andric ArrayRef<uint8_t> ModuleData; 5192480093f4SDimitry Andric Triple T(M.getTargetTriple()); 5193e8d8bef9SDimitry Andric 5194480093f4SDimitry Andric if (EmbedBitcode) { 5195e8d8bef9SDimitry Andric if (Buf.getBufferSize() == 0 || 5196e8d8bef9SDimitry Andric !isBitcode((const unsigned char *)Buf.getBufferStart(), 5197480093f4SDimitry Andric (const unsigned char *)Buf.getBufferEnd())) { 5198480093f4SDimitry Andric // If the input is LLVM Assembly, bitcode is produced by serializing 5199480093f4SDimitry Andric // the module. Use-lists order need to be preserved in this case. 5200480093f4SDimitry Andric llvm::raw_string_ostream OS(Data); 5201480093f4SDimitry Andric llvm::WriteBitcodeToFile(M, OS, /* ShouldPreserveUseListOrder */ true); 5202480093f4SDimitry Andric ModuleData = 5203480093f4SDimitry Andric ArrayRef<uint8_t>((const uint8_t *)OS.str().data(), OS.str().size()); 5204480093f4SDimitry Andric } else 5205480093f4SDimitry Andric // If the input is LLVM bitcode, write the input byte stream directly. 5206480093f4SDimitry Andric ModuleData = ArrayRef<uint8_t>((const uint8_t *)Buf.getBufferStart(), 5207480093f4SDimitry Andric Buf.getBufferSize()); 5208480093f4SDimitry Andric } 5209480093f4SDimitry Andric llvm::Constant *ModuleConstant = 5210480093f4SDimitry Andric llvm::ConstantDataArray::get(M.getContext(), ModuleData); 5211480093f4SDimitry Andric llvm::GlobalVariable *GV = new llvm::GlobalVariable( 5212480093f4SDimitry Andric M, ModuleConstant->getType(), true, llvm::GlobalValue::PrivateLinkage, 5213480093f4SDimitry Andric ModuleConstant); 5214480093f4SDimitry Andric GV->setSection(getSectionNameForBitcode(T)); 5215e8d8bef9SDimitry Andric // Set alignment to 1 to prevent padding between two contributions from input 5216e8d8bef9SDimitry Andric // sections after linking. 5217e8d8bef9SDimitry Andric GV->setAlignment(Align(1)); 5218480093f4SDimitry Andric UsedArray.push_back( 5219480093f4SDimitry Andric ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV, UsedElementType)); 5220480093f4SDimitry Andric if (llvm::GlobalVariable *Old = 5221480093f4SDimitry Andric M.getGlobalVariable("llvm.embedded.module", true)) { 522281ad6265SDimitry Andric assert(Old->hasZeroLiveUses() && 5223480093f4SDimitry Andric "llvm.embedded.module can only be used once in llvm.compiler.used"); 5224480093f4SDimitry Andric GV->takeName(Old); 5225480093f4SDimitry Andric Old->eraseFromParent(); 5226480093f4SDimitry Andric } else { 5227480093f4SDimitry Andric GV->setName("llvm.embedded.module"); 5228480093f4SDimitry Andric } 5229480093f4SDimitry Andric 5230480093f4SDimitry Andric // Skip if only bitcode needs to be embedded. 5231e8d8bef9SDimitry Andric if (EmbedCmdline) { 5232480093f4SDimitry Andric // Embed command-line options. 5233e8d8bef9SDimitry Andric ArrayRef<uint8_t> CmdData(const_cast<uint8_t *>(CmdArgs.data()), 5234e8d8bef9SDimitry Andric CmdArgs.size()); 5235480093f4SDimitry Andric llvm::Constant *CmdConstant = 5236480093f4SDimitry Andric llvm::ConstantDataArray::get(M.getContext(), CmdData); 5237480093f4SDimitry Andric GV = new llvm::GlobalVariable(M, CmdConstant->getType(), true, 5238480093f4SDimitry Andric llvm::GlobalValue::PrivateLinkage, 5239480093f4SDimitry Andric CmdConstant); 5240480093f4SDimitry Andric GV->setSection(getSectionNameForCommandline(T)); 5241e8d8bef9SDimitry Andric GV->setAlignment(Align(1)); 5242480093f4SDimitry Andric UsedArray.push_back( 5243480093f4SDimitry Andric ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV, UsedElementType)); 5244480093f4SDimitry Andric if (llvm::GlobalVariable *Old = M.getGlobalVariable("llvm.cmdline", true)) { 524581ad6265SDimitry Andric assert(Old->hasZeroLiveUses() && 5246480093f4SDimitry Andric "llvm.cmdline can only be used once in llvm.compiler.used"); 5247480093f4SDimitry Andric GV->takeName(Old); 5248480093f4SDimitry Andric Old->eraseFromParent(); 5249480093f4SDimitry Andric } else { 5250480093f4SDimitry Andric GV->setName("llvm.cmdline"); 5251480093f4SDimitry Andric } 5252480093f4SDimitry Andric } 5253480093f4SDimitry Andric 5254480093f4SDimitry Andric if (UsedArray.empty()) 5255480093f4SDimitry Andric return; 5256480093f4SDimitry Andric 5257480093f4SDimitry Andric // Recreate llvm.compiler.used. 5258480093f4SDimitry Andric ArrayType *ATy = ArrayType::get(UsedElementType, UsedArray.size()); 5259480093f4SDimitry Andric auto *NewUsed = new GlobalVariable( 5260480093f4SDimitry Andric M, ATy, false, llvm::GlobalValue::AppendingLinkage, 5261480093f4SDimitry Andric llvm::ConstantArray::get(ATy, UsedArray), "llvm.compiler.used"); 5262480093f4SDimitry Andric NewUsed->setSection("llvm.metadata"); 5263480093f4SDimitry Andric } 5264