1*0b57cec5SDimitry Andric //===- Bitcode/Writer/BitcodeWriter.cpp - Bitcode Writer ------------------===// 2*0b57cec5SDimitry Andric // 3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*0b57cec5SDimitry Andric // 7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 8*0b57cec5SDimitry Andric // 9*0b57cec5SDimitry Andric // Bitcode writer implementation. 10*0b57cec5SDimitry Andric // 11*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 12*0b57cec5SDimitry Andric 13*0b57cec5SDimitry Andric #include "llvm/Bitcode/BitcodeWriter.h" 14*0b57cec5SDimitry Andric #include "ValueEnumerator.h" 15*0b57cec5SDimitry Andric #include "llvm/ADT/APFloat.h" 16*0b57cec5SDimitry Andric #include "llvm/ADT/APInt.h" 17*0b57cec5SDimitry Andric #include "llvm/ADT/ArrayRef.h" 18*0b57cec5SDimitry Andric #include "llvm/ADT/DenseMap.h" 19*0b57cec5SDimitry Andric #include "llvm/ADT/STLExtras.h" 2081ad6265SDimitry Andric #include "llvm/ADT/SetVector.h" 2181ad6265SDimitry Andric #include "llvm/ADT/SmallPtrSet.h" 22*0b57cec5SDimitry Andric #include "llvm/ADT/SmallString.h" 23*0b57cec5SDimitry Andric #include "llvm/ADT/SmallVector.h" 24*0b57cec5SDimitry Andric #include "llvm/ADT/StringMap.h" 25*0b57cec5SDimitry Andric #include "llvm/ADT/StringRef.h" 26*0b57cec5SDimitry Andric #include "llvm/ADT/Triple.h" 27e8d8bef9SDimitry Andric #include "llvm/Bitcode/BitcodeCommon.h" 28480093f4SDimitry Andric #include "llvm/Bitcode/BitcodeReader.h" 29480093f4SDimitry Andric #include "llvm/Bitcode/LLVMBitCodes.h" 30*0b57cec5SDimitry Andric #include "llvm/Bitstream/BitCodes.h" 31*0b57cec5SDimitry Andric #include "llvm/Bitstream/BitstreamWriter.h" 32*0b57cec5SDimitry Andric #include "llvm/Config/llvm-config.h" 33*0b57cec5SDimitry Andric #include "llvm/IR/Attributes.h" 34*0b57cec5SDimitry Andric #include "llvm/IR/BasicBlock.h" 35*0b57cec5SDimitry Andric #include "llvm/IR/Comdat.h" 36*0b57cec5SDimitry Andric #include "llvm/IR/Constant.h" 37*0b57cec5SDimitry Andric #include "llvm/IR/Constants.h" 38*0b57cec5SDimitry Andric #include "llvm/IR/DebugInfoMetadata.h" 39*0b57cec5SDimitry Andric #include "llvm/IR/DebugLoc.h" 40*0b57cec5SDimitry Andric #include "llvm/IR/DerivedTypes.h" 41*0b57cec5SDimitry Andric #include "llvm/IR/Function.h" 42*0b57cec5SDimitry Andric #include "llvm/IR/GlobalAlias.h" 43*0b57cec5SDimitry Andric #include "llvm/IR/GlobalIFunc.h" 44*0b57cec5SDimitry Andric #include "llvm/IR/GlobalObject.h" 45*0b57cec5SDimitry Andric #include "llvm/IR/GlobalValue.h" 46*0b57cec5SDimitry Andric #include "llvm/IR/GlobalVariable.h" 47*0b57cec5SDimitry Andric #include "llvm/IR/InlineAsm.h" 48*0b57cec5SDimitry Andric #include "llvm/IR/InstrTypes.h" 49*0b57cec5SDimitry Andric #include "llvm/IR/Instruction.h" 50*0b57cec5SDimitry Andric #include "llvm/IR/Instructions.h" 51*0b57cec5SDimitry Andric #include "llvm/IR/LLVMContext.h" 52*0b57cec5SDimitry Andric #include "llvm/IR/Metadata.h" 53*0b57cec5SDimitry Andric #include "llvm/IR/Module.h" 54*0b57cec5SDimitry Andric #include "llvm/IR/ModuleSummaryIndex.h" 55*0b57cec5SDimitry Andric #include "llvm/IR/Operator.h" 56*0b57cec5SDimitry Andric #include "llvm/IR/Type.h" 57*0b57cec5SDimitry Andric #include "llvm/IR/UseListOrder.h" 58*0b57cec5SDimitry Andric #include "llvm/IR/Value.h" 59*0b57cec5SDimitry Andric #include "llvm/IR/ValueSymbolTable.h" 60*0b57cec5SDimitry Andric #include "llvm/MC/StringTableBuilder.h" 61349cc55cSDimitry Andric #include "llvm/MC/TargetRegistry.h" 62*0b57cec5SDimitry Andric #include "llvm/Object/IRSymtab.h" 63*0b57cec5SDimitry Andric #include "llvm/Support/AtomicOrdering.h" 64*0b57cec5SDimitry Andric #include "llvm/Support/Casting.h" 65*0b57cec5SDimitry Andric #include "llvm/Support/CommandLine.h" 66*0b57cec5SDimitry Andric #include "llvm/Support/Endian.h" 67*0b57cec5SDimitry Andric #include "llvm/Support/Error.h" 68*0b57cec5SDimitry Andric #include "llvm/Support/ErrorHandling.h" 69*0b57cec5SDimitry Andric #include "llvm/Support/MathExtras.h" 70*0b57cec5SDimitry Andric #include "llvm/Support/SHA1.h" 71*0b57cec5SDimitry Andric #include "llvm/Support/raw_ostream.h" 72*0b57cec5SDimitry Andric #include <algorithm> 73*0b57cec5SDimitry Andric #include <cassert> 74*0b57cec5SDimitry Andric #include <cstddef> 75*0b57cec5SDimitry Andric #include <cstdint> 76*0b57cec5SDimitry Andric #include <iterator> 77*0b57cec5SDimitry Andric #include <map> 78*0b57cec5SDimitry Andric #include <memory> 79bdd1243dSDimitry Andric #include <optional> 80*0b57cec5SDimitry Andric #include <string> 81*0b57cec5SDimitry Andric #include <utility> 82*0b57cec5SDimitry Andric #include <vector> 83*0b57cec5SDimitry Andric 84*0b57cec5SDimitry Andric using namespace llvm; 85*0b57cec5SDimitry Andric 86*0b57cec5SDimitry Andric static cl::opt<unsigned> 87*0b57cec5SDimitry Andric IndexThreshold("bitcode-mdindex-threshold", cl::Hidden, cl::init(25), 88*0b57cec5SDimitry Andric cl::desc("Number of metadatas above which we emit an index " 89*0b57cec5SDimitry 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.")); 93*0b57cec5SDimitry Andric 948bcb0991SDimitry Andric static cl::opt<bool> WriteRelBFToSummary( 95*0b57cec5SDimitry Andric "write-relbf-to-summary", cl::Hidden, cl::init(false), 96*0b57cec5SDimitry Andric cl::desc("Write relative block frequency to function summary ")); 97*0b57cec5SDimitry Andric 98bdd1243dSDimitry Andric namespace llvm { 99*0b57cec5SDimitry Andric extern FunctionSummary::ForceSummaryHotnessType ForceSummaryEdgesCold; 100bdd1243dSDimitry Andric } 101*0b57cec5SDimitry Andric 102*0b57cec5SDimitry Andric namespace { 103*0b57cec5SDimitry Andric 104*0b57cec5SDimitry Andric /// These are manifest constants used by the bitcode writer. They do not need to 105*0b57cec5SDimitry Andric /// be kept in sync with the reader, but need to be consistent within this file. 106*0b57cec5SDimitry Andric enum { 107*0b57cec5SDimitry Andric // VALUE_SYMTAB_BLOCK abbrev id's. 108*0b57cec5SDimitry Andric VST_ENTRY_8_ABBREV = bitc::FIRST_APPLICATION_ABBREV, 109*0b57cec5SDimitry Andric VST_ENTRY_7_ABBREV, 110*0b57cec5SDimitry Andric VST_ENTRY_6_ABBREV, 111*0b57cec5SDimitry Andric VST_BBENTRY_6_ABBREV, 112*0b57cec5SDimitry Andric 113*0b57cec5SDimitry Andric // CONSTANTS_BLOCK abbrev id's. 114*0b57cec5SDimitry Andric CONSTANTS_SETTYPE_ABBREV = bitc::FIRST_APPLICATION_ABBREV, 115*0b57cec5SDimitry Andric CONSTANTS_INTEGER_ABBREV, 116*0b57cec5SDimitry Andric CONSTANTS_CE_CAST_Abbrev, 117*0b57cec5SDimitry Andric CONSTANTS_NULL_Abbrev, 118*0b57cec5SDimitry Andric 119*0b57cec5SDimitry Andric // FUNCTION_BLOCK abbrev id's. 120*0b57cec5SDimitry Andric FUNCTION_INST_LOAD_ABBREV = bitc::FIRST_APPLICATION_ABBREV, 121*0b57cec5SDimitry Andric FUNCTION_INST_UNOP_ABBREV, 122*0b57cec5SDimitry Andric FUNCTION_INST_UNOP_FLAGS_ABBREV, 123*0b57cec5SDimitry Andric FUNCTION_INST_BINOP_ABBREV, 124*0b57cec5SDimitry Andric FUNCTION_INST_BINOP_FLAGS_ABBREV, 125*0b57cec5SDimitry Andric FUNCTION_INST_CAST_ABBREV, 126*0b57cec5SDimitry Andric FUNCTION_INST_RET_VOID_ABBREV, 127*0b57cec5SDimitry Andric FUNCTION_INST_RET_VAL_ABBREV, 128*0b57cec5SDimitry Andric FUNCTION_INST_UNREACHABLE_ABBREV, 129*0b57cec5SDimitry Andric FUNCTION_INST_GEP_ABBREV, 130*0b57cec5SDimitry Andric }; 131*0b57cec5SDimitry Andric 132*0b57cec5SDimitry Andric /// Abstract class to manage the bitcode writing, subclassed for each bitcode 133*0b57cec5SDimitry Andric /// file type. 134*0b57cec5SDimitry Andric class BitcodeWriterBase { 135*0b57cec5SDimitry Andric protected: 136*0b57cec5SDimitry Andric /// The stream created and owned by the client. 137*0b57cec5SDimitry Andric BitstreamWriter &Stream; 138*0b57cec5SDimitry Andric 139*0b57cec5SDimitry Andric StringTableBuilder &StrtabBuilder; 140*0b57cec5SDimitry Andric 141*0b57cec5SDimitry Andric public: 142*0b57cec5SDimitry Andric /// Constructs a BitcodeWriterBase object that writes to the provided 143*0b57cec5SDimitry Andric /// \p Stream. 144*0b57cec5SDimitry Andric BitcodeWriterBase(BitstreamWriter &Stream, StringTableBuilder &StrtabBuilder) 145*0b57cec5SDimitry Andric : Stream(Stream), StrtabBuilder(StrtabBuilder) {} 146*0b57cec5SDimitry Andric 147*0b57cec5SDimitry Andric protected: 148*0b57cec5SDimitry Andric void writeModuleVersion(); 149*0b57cec5SDimitry Andric }; 150*0b57cec5SDimitry Andric 151*0b57cec5SDimitry Andric void BitcodeWriterBase::writeModuleVersion() { 152*0b57cec5SDimitry Andric // VERSION: [version#] 153*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_VERSION, ArrayRef<uint64_t>{2}); 154*0b57cec5SDimitry Andric } 155*0b57cec5SDimitry Andric 156*0b57cec5SDimitry Andric /// Base class to manage the module bitcode writing, currently subclassed for 157*0b57cec5SDimitry Andric /// ModuleBitcodeWriter and ThinLinkBitcodeWriter. 158*0b57cec5SDimitry Andric class ModuleBitcodeWriterBase : public BitcodeWriterBase { 159*0b57cec5SDimitry Andric protected: 160*0b57cec5SDimitry Andric /// The Module to write to bitcode. 161*0b57cec5SDimitry Andric const Module &M; 162*0b57cec5SDimitry Andric 163*0b57cec5SDimitry Andric /// Enumerates ids for all values in the module. 164*0b57cec5SDimitry Andric ValueEnumerator VE; 165*0b57cec5SDimitry Andric 166*0b57cec5SDimitry Andric /// Optional per-module index to write for ThinLTO. 167*0b57cec5SDimitry Andric const ModuleSummaryIndex *Index; 168*0b57cec5SDimitry Andric 169*0b57cec5SDimitry Andric /// Map that holds the correspondence between GUIDs in the summary index, 170*0b57cec5SDimitry Andric /// that came from indirect call profiles, and a value id generated by this 171*0b57cec5SDimitry Andric /// class to use in the VST and summary block records. 172*0b57cec5SDimitry Andric std::map<GlobalValue::GUID, unsigned> GUIDToValueIdMap; 173*0b57cec5SDimitry Andric 174*0b57cec5SDimitry Andric /// Tracks the last value id recorded in the GUIDToValueMap. 175*0b57cec5SDimitry Andric unsigned GlobalValueId; 176*0b57cec5SDimitry Andric 177*0b57cec5SDimitry Andric /// Saves the offset of the VSTOffset record that must eventually be 178*0b57cec5SDimitry Andric /// backpatched with the offset of the actual VST. 179*0b57cec5SDimitry Andric uint64_t VSTOffsetPlaceholder = 0; 180*0b57cec5SDimitry Andric 181*0b57cec5SDimitry Andric public: 182*0b57cec5SDimitry Andric /// Constructs a ModuleBitcodeWriterBase object for the given Module, 183*0b57cec5SDimitry Andric /// writing to the provided \p Buffer. 184*0b57cec5SDimitry Andric ModuleBitcodeWriterBase(const Module &M, StringTableBuilder &StrtabBuilder, 185*0b57cec5SDimitry Andric BitstreamWriter &Stream, 186*0b57cec5SDimitry Andric bool ShouldPreserveUseListOrder, 187*0b57cec5SDimitry Andric const ModuleSummaryIndex *Index) 188*0b57cec5SDimitry Andric : BitcodeWriterBase(Stream, StrtabBuilder), M(M), 189*0b57cec5SDimitry Andric VE(M, ShouldPreserveUseListOrder), Index(Index) { 190*0b57cec5SDimitry Andric // Assign ValueIds to any callee values in the index that came from 191*0b57cec5SDimitry Andric // indirect call profiles and were recorded as a GUID not a Value* 192*0b57cec5SDimitry Andric // (which would have been assigned an ID by the ValueEnumerator). 193*0b57cec5SDimitry Andric // The starting ValueId is just after the number of values in the 194*0b57cec5SDimitry Andric // ValueEnumerator, so that they can be emitted in the VST. 195*0b57cec5SDimitry Andric GlobalValueId = VE.getValues().size(); 196*0b57cec5SDimitry Andric if (!Index) 197*0b57cec5SDimitry Andric return; 198*0b57cec5SDimitry Andric for (const auto &GUIDSummaryLists : *Index) 199*0b57cec5SDimitry Andric // Examine all summaries for this GUID. 200*0b57cec5SDimitry Andric for (auto &Summary : GUIDSummaryLists.second.SummaryList) 201*0b57cec5SDimitry Andric if (auto FS = dyn_cast<FunctionSummary>(Summary.get())) 202*0b57cec5SDimitry Andric // For each call in the function summary, see if the call 203*0b57cec5SDimitry Andric // is to a GUID (which means it is for an indirect call, 204*0b57cec5SDimitry Andric // otherwise we would have a Value for it). If so, synthesize 205*0b57cec5SDimitry Andric // a value id. 206*0b57cec5SDimitry Andric for (auto &CallEdge : FS->calls()) 207*0b57cec5SDimitry Andric if (!CallEdge.first.haveGVs() || !CallEdge.first.getValue()) 208*0b57cec5SDimitry Andric assignValueId(CallEdge.first.getGUID()); 209*0b57cec5SDimitry Andric } 210*0b57cec5SDimitry Andric 211*0b57cec5SDimitry Andric protected: 212*0b57cec5SDimitry Andric void writePerModuleGlobalValueSummary(); 213*0b57cec5SDimitry Andric 214*0b57cec5SDimitry Andric private: 215bdd1243dSDimitry Andric void writePerModuleFunctionSummaryRecord( 216bdd1243dSDimitry Andric SmallVector<uint64_t, 64> &NameVals, GlobalValueSummary *Summary, 217bdd1243dSDimitry Andric unsigned ValueID, unsigned FSCallsAbbrev, unsigned FSCallsProfileAbbrev, 218bdd1243dSDimitry Andric unsigned CallsiteAbbrev, unsigned AllocAbbrev, const Function &F); 219*0b57cec5SDimitry Andric void writeModuleLevelReferences(const GlobalVariable &V, 220*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, 221*0b57cec5SDimitry Andric unsigned FSModRefsAbbrev, 222*0b57cec5SDimitry Andric unsigned FSModVTableRefsAbbrev); 223*0b57cec5SDimitry Andric 224*0b57cec5SDimitry Andric void assignValueId(GlobalValue::GUID ValGUID) { 225*0b57cec5SDimitry Andric GUIDToValueIdMap[ValGUID] = ++GlobalValueId; 226*0b57cec5SDimitry Andric } 227*0b57cec5SDimitry Andric 228*0b57cec5SDimitry Andric unsigned getValueId(GlobalValue::GUID ValGUID) { 229*0b57cec5SDimitry Andric const auto &VMI = GUIDToValueIdMap.find(ValGUID); 230*0b57cec5SDimitry Andric // Expect that any GUID value had a value Id assigned by an 231*0b57cec5SDimitry Andric // earlier call to assignValueId. 232*0b57cec5SDimitry Andric assert(VMI != GUIDToValueIdMap.end() && 233*0b57cec5SDimitry Andric "GUID does not have assigned value Id"); 234*0b57cec5SDimitry Andric return VMI->second; 235*0b57cec5SDimitry Andric } 236*0b57cec5SDimitry Andric 237*0b57cec5SDimitry Andric // Helper to get the valueId for the type of value recorded in VI. 238*0b57cec5SDimitry Andric unsigned getValueId(ValueInfo VI) { 239*0b57cec5SDimitry Andric if (!VI.haveGVs() || !VI.getValue()) 240*0b57cec5SDimitry Andric return getValueId(VI.getGUID()); 241*0b57cec5SDimitry Andric return VE.getValueID(VI.getValue()); 242*0b57cec5SDimitry Andric } 243*0b57cec5SDimitry Andric 244*0b57cec5SDimitry Andric std::map<GlobalValue::GUID, unsigned> &valueIds() { return GUIDToValueIdMap; } 245*0b57cec5SDimitry Andric }; 246*0b57cec5SDimitry Andric 247*0b57cec5SDimitry Andric /// Class to manage the bitcode writing for a module. 248*0b57cec5SDimitry Andric class ModuleBitcodeWriter : public ModuleBitcodeWriterBase { 249*0b57cec5SDimitry Andric /// Pointer to the buffer allocated by caller for bitcode writing. 250*0b57cec5SDimitry Andric const SmallVectorImpl<char> &Buffer; 251*0b57cec5SDimitry Andric 252*0b57cec5SDimitry Andric /// True if a module hash record should be written. 253*0b57cec5SDimitry Andric bool GenerateHash; 254*0b57cec5SDimitry Andric 255*0b57cec5SDimitry Andric /// If non-null, when GenerateHash is true, the resulting hash is written 256*0b57cec5SDimitry Andric /// into ModHash. 257*0b57cec5SDimitry Andric ModuleHash *ModHash; 258*0b57cec5SDimitry Andric 259*0b57cec5SDimitry Andric SHA1 Hasher; 260*0b57cec5SDimitry Andric 261*0b57cec5SDimitry Andric /// The start bit of the identification block. 262*0b57cec5SDimitry Andric uint64_t BitcodeStartBit; 263*0b57cec5SDimitry Andric 264*0b57cec5SDimitry Andric public: 265*0b57cec5SDimitry Andric /// Constructs a ModuleBitcodeWriter object for the given Module, 266*0b57cec5SDimitry Andric /// writing to the provided \p Buffer. 267*0b57cec5SDimitry Andric ModuleBitcodeWriter(const Module &M, SmallVectorImpl<char> &Buffer, 268*0b57cec5SDimitry Andric StringTableBuilder &StrtabBuilder, 269*0b57cec5SDimitry Andric BitstreamWriter &Stream, bool ShouldPreserveUseListOrder, 270*0b57cec5SDimitry Andric const ModuleSummaryIndex *Index, bool GenerateHash, 271*0b57cec5SDimitry Andric ModuleHash *ModHash = nullptr) 272*0b57cec5SDimitry Andric : ModuleBitcodeWriterBase(M, StrtabBuilder, Stream, 273*0b57cec5SDimitry Andric ShouldPreserveUseListOrder, Index), 274*0b57cec5SDimitry Andric Buffer(Buffer), GenerateHash(GenerateHash), ModHash(ModHash), 275*0b57cec5SDimitry Andric BitcodeStartBit(Stream.GetCurrentBitNo()) {} 276*0b57cec5SDimitry Andric 277*0b57cec5SDimitry Andric /// Emit the current module to the bitstream. 278*0b57cec5SDimitry Andric void write(); 279*0b57cec5SDimitry Andric 280*0b57cec5SDimitry Andric private: 281*0b57cec5SDimitry Andric uint64_t bitcodeStartBit() { return BitcodeStartBit; } 282*0b57cec5SDimitry Andric 283*0b57cec5SDimitry Andric size_t addToStrtab(StringRef Str); 284*0b57cec5SDimitry Andric 285*0b57cec5SDimitry Andric void writeAttributeGroupTable(); 286*0b57cec5SDimitry Andric void writeAttributeTable(); 287*0b57cec5SDimitry Andric void writeTypeTable(); 288*0b57cec5SDimitry Andric void writeComdats(); 289*0b57cec5SDimitry Andric void writeValueSymbolTableForwardDecl(); 290*0b57cec5SDimitry Andric void writeModuleInfo(); 291*0b57cec5SDimitry Andric void writeValueAsMetadata(const ValueAsMetadata *MD, 292*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record); 293*0b57cec5SDimitry Andric void writeMDTuple(const MDTuple *N, SmallVectorImpl<uint64_t> &Record, 294*0b57cec5SDimitry Andric unsigned Abbrev); 295*0b57cec5SDimitry Andric unsigned createDILocationAbbrev(); 296*0b57cec5SDimitry Andric void writeDILocation(const DILocation *N, SmallVectorImpl<uint64_t> &Record, 297*0b57cec5SDimitry Andric unsigned &Abbrev); 298*0b57cec5SDimitry Andric unsigned createGenericDINodeAbbrev(); 299*0b57cec5SDimitry Andric void writeGenericDINode(const GenericDINode *N, 300*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned &Abbrev); 301*0b57cec5SDimitry Andric void writeDISubrange(const DISubrange *N, SmallVectorImpl<uint64_t> &Record, 302*0b57cec5SDimitry Andric unsigned Abbrev); 303e8d8bef9SDimitry Andric void writeDIGenericSubrange(const DIGenericSubrange *N, 304e8d8bef9SDimitry Andric SmallVectorImpl<uint64_t> &Record, 305e8d8bef9SDimitry Andric unsigned Abbrev); 306*0b57cec5SDimitry Andric void writeDIEnumerator(const DIEnumerator *N, 307*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 308*0b57cec5SDimitry Andric void writeDIBasicType(const DIBasicType *N, SmallVectorImpl<uint64_t> &Record, 309*0b57cec5SDimitry Andric unsigned Abbrev); 310e8d8bef9SDimitry Andric void writeDIStringType(const DIStringType *N, 311e8d8bef9SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 312*0b57cec5SDimitry Andric void writeDIDerivedType(const DIDerivedType *N, 313*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 314*0b57cec5SDimitry Andric void writeDICompositeType(const DICompositeType *N, 315*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 316*0b57cec5SDimitry Andric void writeDISubroutineType(const DISubroutineType *N, 317*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 318*0b57cec5SDimitry Andric unsigned Abbrev); 319*0b57cec5SDimitry Andric void writeDIFile(const DIFile *N, SmallVectorImpl<uint64_t> &Record, 320*0b57cec5SDimitry Andric unsigned Abbrev); 321*0b57cec5SDimitry Andric void writeDICompileUnit(const DICompileUnit *N, 322*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 323*0b57cec5SDimitry Andric void writeDISubprogram(const DISubprogram *N, 324*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 325*0b57cec5SDimitry Andric void writeDILexicalBlock(const DILexicalBlock *N, 326*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 327*0b57cec5SDimitry Andric void writeDILexicalBlockFile(const DILexicalBlockFile *N, 328*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 329*0b57cec5SDimitry Andric unsigned Abbrev); 330*0b57cec5SDimitry Andric void writeDICommonBlock(const DICommonBlock *N, 331*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 332*0b57cec5SDimitry Andric void writeDINamespace(const DINamespace *N, SmallVectorImpl<uint64_t> &Record, 333*0b57cec5SDimitry Andric unsigned Abbrev); 334*0b57cec5SDimitry Andric void writeDIMacro(const DIMacro *N, SmallVectorImpl<uint64_t> &Record, 335*0b57cec5SDimitry Andric unsigned Abbrev); 336*0b57cec5SDimitry Andric void writeDIMacroFile(const DIMacroFile *N, SmallVectorImpl<uint64_t> &Record, 337*0b57cec5SDimitry Andric unsigned Abbrev); 338fe6060f1SDimitry Andric void writeDIArgList(const DIArgList *N, SmallVectorImpl<uint64_t> &Record, 339fe6060f1SDimitry Andric unsigned Abbrev); 340*0b57cec5SDimitry Andric void writeDIModule(const DIModule *N, SmallVectorImpl<uint64_t> &Record, 341*0b57cec5SDimitry Andric unsigned Abbrev); 342bdd1243dSDimitry Andric void writeDIAssignID(const DIAssignID *N, SmallVectorImpl<uint64_t> &Record, 343bdd1243dSDimitry Andric unsigned Abbrev); 344*0b57cec5SDimitry Andric void writeDITemplateTypeParameter(const DITemplateTypeParameter *N, 345*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 346*0b57cec5SDimitry Andric unsigned Abbrev); 347*0b57cec5SDimitry Andric void writeDITemplateValueParameter(const DITemplateValueParameter *N, 348*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 349*0b57cec5SDimitry Andric unsigned Abbrev); 350*0b57cec5SDimitry Andric void writeDIGlobalVariable(const DIGlobalVariable *N, 351*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 352*0b57cec5SDimitry Andric unsigned Abbrev); 353*0b57cec5SDimitry Andric void writeDILocalVariable(const DILocalVariable *N, 354*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 355*0b57cec5SDimitry Andric void writeDILabel(const DILabel *N, 356*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 357*0b57cec5SDimitry Andric void writeDIExpression(const DIExpression *N, 358*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 359*0b57cec5SDimitry Andric void writeDIGlobalVariableExpression(const DIGlobalVariableExpression *N, 360*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 361*0b57cec5SDimitry Andric unsigned Abbrev); 362*0b57cec5SDimitry Andric void writeDIObjCProperty(const DIObjCProperty *N, 363*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, unsigned Abbrev); 364*0b57cec5SDimitry Andric void writeDIImportedEntity(const DIImportedEntity *N, 365*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 366*0b57cec5SDimitry Andric unsigned Abbrev); 367*0b57cec5SDimitry Andric unsigned createNamedMetadataAbbrev(); 368*0b57cec5SDimitry Andric void writeNamedMetadata(SmallVectorImpl<uint64_t> &Record); 369*0b57cec5SDimitry Andric unsigned createMetadataStringsAbbrev(); 370*0b57cec5SDimitry Andric void writeMetadataStrings(ArrayRef<const Metadata *> Strings, 371*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record); 372*0b57cec5SDimitry Andric void writeMetadataRecords(ArrayRef<const Metadata *> MDs, 373*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 374*0b57cec5SDimitry Andric std::vector<unsigned> *MDAbbrevs = nullptr, 375*0b57cec5SDimitry Andric std::vector<uint64_t> *IndexPos = nullptr); 376*0b57cec5SDimitry Andric void writeModuleMetadata(); 377*0b57cec5SDimitry Andric void writeFunctionMetadata(const Function &F); 378*0b57cec5SDimitry Andric void writeFunctionMetadataAttachment(const Function &F); 379*0b57cec5SDimitry Andric void pushGlobalMetadataAttachment(SmallVectorImpl<uint64_t> &Record, 380*0b57cec5SDimitry Andric const GlobalObject &GO); 381*0b57cec5SDimitry Andric void writeModuleMetadataKinds(); 382*0b57cec5SDimitry Andric void writeOperandBundleTags(); 383*0b57cec5SDimitry Andric void writeSyncScopeNames(); 384*0b57cec5SDimitry Andric void writeConstants(unsigned FirstVal, unsigned LastVal, bool isGlobal); 385*0b57cec5SDimitry Andric void writeModuleConstants(); 386*0b57cec5SDimitry Andric bool pushValueAndType(const Value *V, unsigned InstID, 387*0b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals); 3885ffd83dbSDimitry Andric void writeOperandBundles(const CallBase &CB, unsigned InstID); 389*0b57cec5SDimitry Andric void pushValue(const Value *V, unsigned InstID, 390*0b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals); 391*0b57cec5SDimitry Andric void pushValueSigned(const Value *V, unsigned InstID, 392*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Vals); 393*0b57cec5SDimitry Andric void writeInstruction(const Instruction &I, unsigned InstID, 394*0b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals); 395*0b57cec5SDimitry Andric void writeFunctionLevelValueSymbolTable(const ValueSymbolTable &VST); 396*0b57cec5SDimitry Andric void writeGlobalValueSymbolTable( 397*0b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> &FunctionToBitcodeIndex); 398*0b57cec5SDimitry Andric void writeUseList(UseListOrder &&Order); 399*0b57cec5SDimitry Andric void writeUseListBlock(const Function *F); 400*0b57cec5SDimitry Andric void 401*0b57cec5SDimitry Andric writeFunction(const Function &F, 402*0b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> &FunctionToBitcodeIndex); 403*0b57cec5SDimitry Andric void writeBlockInfo(); 404*0b57cec5SDimitry Andric void writeModuleHash(size_t BlockStartPos); 405*0b57cec5SDimitry Andric 406*0b57cec5SDimitry Andric unsigned getEncodedSyncScopeID(SyncScope::ID SSID) { 407*0b57cec5SDimitry Andric return unsigned(SSID); 408*0b57cec5SDimitry Andric } 409e8d8bef9SDimitry Andric 410e8d8bef9SDimitry Andric unsigned getEncodedAlign(MaybeAlign Alignment) { return encode(Alignment); } 411*0b57cec5SDimitry Andric }; 412*0b57cec5SDimitry Andric 413*0b57cec5SDimitry Andric /// Class to manage the bitcode writing for a combined index. 414*0b57cec5SDimitry Andric class IndexBitcodeWriter : public BitcodeWriterBase { 415*0b57cec5SDimitry Andric /// The combined index to write to bitcode. 416*0b57cec5SDimitry Andric const ModuleSummaryIndex &Index; 417*0b57cec5SDimitry Andric 418*0b57cec5SDimitry Andric /// When writing a subset of the index for distributed backends, client 419*0b57cec5SDimitry Andric /// provides a map of modules to the corresponding GUIDs/summaries to write. 420*0b57cec5SDimitry Andric const std::map<std::string, GVSummaryMapTy> *ModuleToSummariesForIndex; 421*0b57cec5SDimitry Andric 422*0b57cec5SDimitry Andric /// Map that holds the correspondence between the GUID used in the combined 423*0b57cec5SDimitry Andric /// index and a value id generated by this class to use in references. 424*0b57cec5SDimitry Andric std::map<GlobalValue::GUID, unsigned> GUIDToValueIdMap; 425*0b57cec5SDimitry 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 431*0b57cec5SDimitry Andric /// Tracks the last value id recorded in the GUIDToValueMap. 432*0b57cec5SDimitry Andric unsigned GlobalValueId = 0; 433*0b57cec5SDimitry Andric 434*0b57cec5SDimitry Andric public: 435*0b57cec5SDimitry Andric /// Constructs a IndexBitcodeWriter object for the given combined index, 436*0b57cec5SDimitry Andric /// writing to the provided \p Buffer. When writing a subset of the index 437*0b57cec5SDimitry Andric /// for a distributed backend, provide a \p ModuleToSummariesForIndex map. 438*0b57cec5SDimitry Andric IndexBitcodeWriter(BitstreamWriter &Stream, StringTableBuilder &StrtabBuilder, 439*0b57cec5SDimitry Andric const ModuleSummaryIndex &Index, 440*0b57cec5SDimitry Andric const std::map<std::string, GVSummaryMapTy> 441*0b57cec5SDimitry Andric *ModuleToSummariesForIndex = nullptr) 442*0b57cec5SDimitry Andric : BitcodeWriterBase(Stream, StrtabBuilder), Index(Index), 443*0b57cec5SDimitry Andric ModuleToSummariesForIndex(ModuleToSummariesForIndex) { 444*0b57cec5SDimitry Andric // Assign unique value ids to all summaries to be written, for use 445*0b57cec5SDimitry Andric // in writing out the call graph edges. Save the mapping from GUID 446*0b57cec5SDimitry Andric // to the new global value id to use when writing those edges, which 447*0b57cec5SDimitry Andric // are currently saved in the index in terms of GUID. 448bdd1243dSDimitry Andric forEachSummary([&](GVInfo I, bool IsAliasee) { 449*0b57cec5SDimitry Andric GUIDToValueIdMap[I.first] = ++GlobalValueId; 450bdd1243dSDimitry Andric if (IsAliasee) 451bdd1243dSDimitry Andric return; 452bdd1243dSDimitry Andric auto *FS = dyn_cast<FunctionSummary>(I.second); 453bdd1243dSDimitry Andric if (!FS) 454bdd1243dSDimitry Andric return; 455bdd1243dSDimitry Andric // Record all stack id indices actually used in the summary entries being 456bdd1243dSDimitry Andric // written, so that we can compact them in the case of distributed ThinLTO 457bdd1243dSDimitry Andric // indexes. 458bdd1243dSDimitry Andric for (auto &CI : FS->callsites()) 459bdd1243dSDimitry Andric for (auto Idx : CI.StackIdIndices) 460bdd1243dSDimitry Andric StackIdIndices.push_back(Idx); 461bdd1243dSDimitry Andric for (auto &AI : FS->allocs()) 462bdd1243dSDimitry Andric for (auto &MIB : AI.MIBs) 463bdd1243dSDimitry Andric for (auto Idx : MIB.StackIdIndices) 464bdd1243dSDimitry Andric StackIdIndices.push_back(Idx); 465*0b57cec5SDimitry Andric }); 466bdd1243dSDimitry Andric llvm::sort(StackIdIndices); 467bdd1243dSDimitry Andric StackIdIndices.erase( 468bdd1243dSDimitry Andric std::unique(StackIdIndices.begin(), StackIdIndices.end()), 469bdd1243dSDimitry Andric StackIdIndices.end()); 470*0b57cec5SDimitry Andric } 471*0b57cec5SDimitry Andric 472*0b57cec5SDimitry Andric /// The below iterator returns the GUID and associated summary. 473*0b57cec5SDimitry Andric using GVInfo = std::pair<GlobalValue::GUID, GlobalValueSummary *>; 474*0b57cec5SDimitry Andric 475*0b57cec5SDimitry Andric /// Calls the callback for each value GUID and summary to be written to 476*0b57cec5SDimitry Andric /// bitcode. This hides the details of whether they are being pulled from the 477*0b57cec5SDimitry Andric /// entire index or just those in a provided ModuleToSummariesForIndex map. 478*0b57cec5SDimitry Andric template<typename Functor> 479*0b57cec5SDimitry Andric void forEachSummary(Functor Callback) { 480*0b57cec5SDimitry Andric if (ModuleToSummariesForIndex) { 481*0b57cec5SDimitry Andric for (auto &M : *ModuleToSummariesForIndex) 482*0b57cec5SDimitry Andric for (auto &Summary : M.second) { 483*0b57cec5SDimitry Andric Callback(Summary, false); 484*0b57cec5SDimitry Andric // Ensure aliasee is handled, e.g. for assigning a valueId, 485*0b57cec5SDimitry Andric // even if we are not importing the aliasee directly (the 486*0b57cec5SDimitry Andric // imported alias will contain a copy of aliasee). 487*0b57cec5SDimitry Andric if (auto *AS = dyn_cast<AliasSummary>(Summary.getSecond())) 488*0b57cec5SDimitry Andric Callback({AS->getAliaseeGUID(), &AS->getAliasee()}, true); 489*0b57cec5SDimitry Andric } 490*0b57cec5SDimitry Andric } else { 491*0b57cec5SDimitry Andric for (auto &Summaries : Index) 492*0b57cec5SDimitry Andric for (auto &Summary : Summaries.second.SummaryList) 493*0b57cec5SDimitry Andric Callback({Summaries.first, Summary.get()}, false); 494*0b57cec5SDimitry Andric } 495*0b57cec5SDimitry Andric } 496*0b57cec5SDimitry Andric 497*0b57cec5SDimitry Andric /// Calls the callback for each entry in the modulePaths StringMap that 498*0b57cec5SDimitry Andric /// should be written to the module path string table. This hides the details 499*0b57cec5SDimitry Andric /// of whether they are being pulled from the entire index or just those in a 500*0b57cec5SDimitry Andric /// provided ModuleToSummariesForIndex map. 501*0b57cec5SDimitry Andric template <typename Functor> void forEachModule(Functor Callback) { 502*0b57cec5SDimitry Andric if (ModuleToSummariesForIndex) { 503*0b57cec5SDimitry Andric for (const auto &M : *ModuleToSummariesForIndex) { 504*0b57cec5SDimitry Andric const auto &MPI = Index.modulePaths().find(M.first); 505*0b57cec5SDimitry Andric if (MPI == Index.modulePaths().end()) { 506*0b57cec5SDimitry Andric // This should only happen if the bitcode file was empty, in which 507*0b57cec5SDimitry Andric // case we shouldn't be importing (the ModuleToSummariesForIndex 508*0b57cec5SDimitry Andric // would only include the module we are writing and index for). 509*0b57cec5SDimitry Andric assert(ModuleToSummariesForIndex->size() == 1); 510*0b57cec5SDimitry Andric continue; 511*0b57cec5SDimitry Andric } 512*0b57cec5SDimitry Andric Callback(*MPI); 513*0b57cec5SDimitry Andric } 514*0b57cec5SDimitry Andric } else { 515*0b57cec5SDimitry Andric for (const auto &MPSE : Index.modulePaths()) 516*0b57cec5SDimitry Andric Callback(MPSE); 517*0b57cec5SDimitry Andric } 518*0b57cec5SDimitry Andric } 519*0b57cec5SDimitry Andric 520*0b57cec5SDimitry Andric /// Main entry point for writing a combined index to bitcode. 521*0b57cec5SDimitry Andric void write(); 522*0b57cec5SDimitry Andric 523*0b57cec5SDimitry Andric private: 524*0b57cec5SDimitry Andric void writeModStrings(); 525*0b57cec5SDimitry Andric void writeCombinedGlobalValueSummary(); 526*0b57cec5SDimitry Andric 527bdd1243dSDimitry Andric std::optional<unsigned> getValueId(GlobalValue::GUID ValGUID) { 528*0b57cec5SDimitry Andric auto VMI = GUIDToValueIdMap.find(ValGUID); 529*0b57cec5SDimitry Andric if (VMI == GUIDToValueIdMap.end()) 530bdd1243dSDimitry Andric return std::nullopt; 531*0b57cec5SDimitry Andric return VMI->second; 532*0b57cec5SDimitry Andric } 533*0b57cec5SDimitry Andric 534*0b57cec5SDimitry Andric std::map<GlobalValue::GUID, unsigned> &valueIds() { return GUIDToValueIdMap; } 535*0b57cec5SDimitry Andric }; 536*0b57cec5SDimitry Andric 537*0b57cec5SDimitry Andric } // end anonymous namespace 538*0b57cec5SDimitry Andric 539*0b57cec5SDimitry Andric static unsigned getEncodedCastOpcode(unsigned Opcode) { 540*0b57cec5SDimitry Andric switch (Opcode) { 541*0b57cec5SDimitry Andric default: llvm_unreachable("Unknown cast instruction!"); 542*0b57cec5SDimitry Andric case Instruction::Trunc : return bitc::CAST_TRUNC; 543*0b57cec5SDimitry Andric case Instruction::ZExt : return bitc::CAST_ZEXT; 544*0b57cec5SDimitry Andric case Instruction::SExt : return bitc::CAST_SEXT; 545*0b57cec5SDimitry Andric case Instruction::FPToUI : return bitc::CAST_FPTOUI; 546*0b57cec5SDimitry Andric case Instruction::FPToSI : return bitc::CAST_FPTOSI; 547*0b57cec5SDimitry Andric case Instruction::UIToFP : return bitc::CAST_UITOFP; 548*0b57cec5SDimitry Andric case Instruction::SIToFP : return bitc::CAST_SITOFP; 549*0b57cec5SDimitry Andric case Instruction::FPTrunc : return bitc::CAST_FPTRUNC; 550*0b57cec5SDimitry Andric case Instruction::FPExt : return bitc::CAST_FPEXT; 551*0b57cec5SDimitry Andric case Instruction::PtrToInt: return bitc::CAST_PTRTOINT; 552*0b57cec5SDimitry Andric case Instruction::IntToPtr: return bitc::CAST_INTTOPTR; 553*0b57cec5SDimitry Andric case Instruction::BitCast : return bitc::CAST_BITCAST; 554*0b57cec5SDimitry Andric case Instruction::AddrSpaceCast: return bitc::CAST_ADDRSPACECAST; 555*0b57cec5SDimitry Andric } 556*0b57cec5SDimitry Andric } 557*0b57cec5SDimitry Andric 558*0b57cec5SDimitry Andric static unsigned getEncodedUnaryOpcode(unsigned Opcode) { 559*0b57cec5SDimitry Andric switch (Opcode) { 560*0b57cec5SDimitry Andric default: llvm_unreachable("Unknown binary instruction!"); 5618bcb0991SDimitry Andric case Instruction::FNeg: return bitc::UNOP_FNEG; 562*0b57cec5SDimitry Andric } 563*0b57cec5SDimitry Andric } 564*0b57cec5SDimitry Andric 565*0b57cec5SDimitry Andric static unsigned getEncodedBinaryOpcode(unsigned Opcode) { 566*0b57cec5SDimitry Andric switch (Opcode) { 567*0b57cec5SDimitry Andric default: llvm_unreachable("Unknown binary instruction!"); 568*0b57cec5SDimitry Andric case Instruction::Add: 569*0b57cec5SDimitry Andric case Instruction::FAdd: return bitc::BINOP_ADD; 570*0b57cec5SDimitry Andric case Instruction::Sub: 571*0b57cec5SDimitry Andric case Instruction::FSub: return bitc::BINOP_SUB; 572*0b57cec5SDimitry Andric case Instruction::Mul: 573*0b57cec5SDimitry Andric case Instruction::FMul: return bitc::BINOP_MUL; 574*0b57cec5SDimitry Andric case Instruction::UDiv: return bitc::BINOP_UDIV; 575*0b57cec5SDimitry Andric case Instruction::FDiv: 576*0b57cec5SDimitry Andric case Instruction::SDiv: return bitc::BINOP_SDIV; 577*0b57cec5SDimitry Andric case Instruction::URem: return bitc::BINOP_UREM; 578*0b57cec5SDimitry Andric case Instruction::FRem: 579*0b57cec5SDimitry Andric case Instruction::SRem: return bitc::BINOP_SREM; 580*0b57cec5SDimitry Andric case Instruction::Shl: return bitc::BINOP_SHL; 581*0b57cec5SDimitry Andric case Instruction::LShr: return bitc::BINOP_LSHR; 582*0b57cec5SDimitry Andric case Instruction::AShr: return bitc::BINOP_ASHR; 583*0b57cec5SDimitry Andric case Instruction::And: return bitc::BINOP_AND; 584*0b57cec5SDimitry Andric case Instruction::Or: return bitc::BINOP_OR; 585*0b57cec5SDimitry Andric case Instruction::Xor: return bitc::BINOP_XOR; 586*0b57cec5SDimitry Andric } 587*0b57cec5SDimitry Andric } 588*0b57cec5SDimitry Andric 589*0b57cec5SDimitry Andric static unsigned getEncodedRMWOperation(AtomicRMWInst::BinOp Op) { 590*0b57cec5SDimitry Andric switch (Op) { 591*0b57cec5SDimitry Andric default: llvm_unreachable("Unknown RMW operation!"); 592*0b57cec5SDimitry Andric case AtomicRMWInst::Xchg: return bitc::RMW_XCHG; 593*0b57cec5SDimitry Andric case AtomicRMWInst::Add: return bitc::RMW_ADD; 594*0b57cec5SDimitry Andric case AtomicRMWInst::Sub: return bitc::RMW_SUB; 595*0b57cec5SDimitry Andric case AtomicRMWInst::And: return bitc::RMW_AND; 596*0b57cec5SDimitry Andric case AtomicRMWInst::Nand: return bitc::RMW_NAND; 597*0b57cec5SDimitry Andric case AtomicRMWInst::Or: return bitc::RMW_OR; 598*0b57cec5SDimitry Andric case AtomicRMWInst::Xor: return bitc::RMW_XOR; 599*0b57cec5SDimitry Andric case AtomicRMWInst::Max: return bitc::RMW_MAX; 600*0b57cec5SDimitry Andric case AtomicRMWInst::Min: return bitc::RMW_MIN; 601*0b57cec5SDimitry Andric case AtomicRMWInst::UMax: return bitc::RMW_UMAX; 602*0b57cec5SDimitry Andric case AtomicRMWInst::UMin: return bitc::RMW_UMIN; 603*0b57cec5SDimitry Andric case AtomicRMWInst::FAdd: return bitc::RMW_FADD; 604*0b57cec5SDimitry Andric case AtomicRMWInst::FSub: return bitc::RMW_FSUB; 605753f127fSDimitry Andric case AtomicRMWInst::FMax: return bitc::RMW_FMAX; 606753f127fSDimitry Andric case AtomicRMWInst::FMin: return bitc::RMW_FMIN; 607bdd1243dSDimitry Andric case AtomicRMWInst::UIncWrap: 608bdd1243dSDimitry Andric return bitc::RMW_UINC_WRAP; 609bdd1243dSDimitry Andric case AtomicRMWInst::UDecWrap: 610bdd1243dSDimitry Andric return bitc::RMW_UDEC_WRAP; 611*0b57cec5SDimitry Andric } 612*0b57cec5SDimitry Andric } 613*0b57cec5SDimitry Andric 614*0b57cec5SDimitry Andric static unsigned getEncodedOrdering(AtomicOrdering Ordering) { 615*0b57cec5SDimitry Andric switch (Ordering) { 616*0b57cec5SDimitry Andric case AtomicOrdering::NotAtomic: return bitc::ORDERING_NOTATOMIC; 617*0b57cec5SDimitry Andric case AtomicOrdering::Unordered: return bitc::ORDERING_UNORDERED; 618*0b57cec5SDimitry Andric case AtomicOrdering::Monotonic: return bitc::ORDERING_MONOTONIC; 619*0b57cec5SDimitry Andric case AtomicOrdering::Acquire: return bitc::ORDERING_ACQUIRE; 620*0b57cec5SDimitry Andric case AtomicOrdering::Release: return bitc::ORDERING_RELEASE; 621*0b57cec5SDimitry Andric case AtomicOrdering::AcquireRelease: return bitc::ORDERING_ACQREL; 622*0b57cec5SDimitry Andric case AtomicOrdering::SequentiallyConsistent: return bitc::ORDERING_SEQCST; 623*0b57cec5SDimitry Andric } 624*0b57cec5SDimitry Andric llvm_unreachable("Invalid ordering"); 625*0b57cec5SDimitry Andric } 626*0b57cec5SDimitry Andric 627*0b57cec5SDimitry Andric static void writeStringRecord(BitstreamWriter &Stream, unsigned Code, 628*0b57cec5SDimitry Andric StringRef Str, unsigned AbbrevToUse) { 629*0b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 630*0b57cec5SDimitry Andric 631*0b57cec5SDimitry Andric // Code: [strchar x N] 6324824e7fdSDimitry Andric for (char C : Str) { 6334824e7fdSDimitry Andric if (AbbrevToUse && !BitCodeAbbrevOp::isChar6(C)) 634*0b57cec5SDimitry Andric AbbrevToUse = 0; 6354824e7fdSDimitry Andric Vals.push_back(C); 636*0b57cec5SDimitry Andric } 637*0b57cec5SDimitry Andric 638*0b57cec5SDimitry Andric // Emit the finished record. 639*0b57cec5SDimitry Andric Stream.EmitRecord(Code, Vals, AbbrevToUse); 640*0b57cec5SDimitry Andric } 641*0b57cec5SDimitry Andric 642*0b57cec5SDimitry Andric static uint64_t getAttrKindEncoding(Attribute::AttrKind Kind) { 643*0b57cec5SDimitry Andric switch (Kind) { 644*0b57cec5SDimitry Andric case Attribute::Alignment: 645*0b57cec5SDimitry Andric return bitc::ATTR_KIND_ALIGNMENT; 64681ad6265SDimitry Andric case Attribute::AllocAlign: 64781ad6265SDimitry Andric return bitc::ATTR_KIND_ALLOC_ALIGN; 648*0b57cec5SDimitry Andric case Attribute::AllocSize: 649*0b57cec5SDimitry Andric return bitc::ATTR_KIND_ALLOC_SIZE; 650*0b57cec5SDimitry Andric case Attribute::AlwaysInline: 651*0b57cec5SDimitry Andric return bitc::ATTR_KIND_ALWAYS_INLINE; 652*0b57cec5SDimitry Andric case Attribute::Builtin: 653*0b57cec5SDimitry Andric return bitc::ATTR_KIND_BUILTIN; 654*0b57cec5SDimitry Andric case Attribute::ByVal: 655*0b57cec5SDimitry Andric return bitc::ATTR_KIND_BY_VAL; 656*0b57cec5SDimitry Andric case Attribute::Convergent: 657*0b57cec5SDimitry Andric return bitc::ATTR_KIND_CONVERGENT; 658*0b57cec5SDimitry Andric case Attribute::InAlloca: 659*0b57cec5SDimitry Andric return bitc::ATTR_KIND_IN_ALLOCA; 660*0b57cec5SDimitry Andric case Attribute::Cold: 661*0b57cec5SDimitry Andric return bitc::ATTR_KIND_COLD; 662349cc55cSDimitry Andric case Attribute::DisableSanitizerInstrumentation: 663349cc55cSDimitry Andric return bitc::ATTR_KIND_DISABLE_SANITIZER_INSTRUMENTATION; 664753f127fSDimitry Andric case Attribute::FnRetThunkExtern: 665753f127fSDimitry Andric return bitc::ATTR_KIND_FNRETTHUNK_EXTERN; 666e8d8bef9SDimitry Andric case Attribute::Hot: 667e8d8bef9SDimitry Andric return bitc::ATTR_KIND_HOT; 668fe6060f1SDimitry Andric case Attribute::ElementType: 669fe6060f1SDimitry Andric return bitc::ATTR_KIND_ELEMENTTYPE; 670*0b57cec5SDimitry Andric case Attribute::InlineHint: 671*0b57cec5SDimitry Andric return bitc::ATTR_KIND_INLINE_HINT; 672*0b57cec5SDimitry Andric case Attribute::InReg: 673*0b57cec5SDimitry Andric return bitc::ATTR_KIND_IN_REG; 674*0b57cec5SDimitry Andric case Attribute::JumpTable: 675*0b57cec5SDimitry Andric return bitc::ATTR_KIND_JUMP_TABLE; 676*0b57cec5SDimitry Andric case Attribute::MinSize: 677*0b57cec5SDimitry Andric return bitc::ATTR_KIND_MIN_SIZE; 67881ad6265SDimitry Andric case Attribute::AllocatedPointer: 67981ad6265SDimitry Andric return bitc::ATTR_KIND_ALLOCATED_POINTER; 68081ad6265SDimitry Andric case Attribute::AllocKind: 68181ad6265SDimitry Andric return bitc::ATTR_KIND_ALLOC_KIND; 682bdd1243dSDimitry Andric case Attribute::Memory: 683bdd1243dSDimitry Andric return bitc::ATTR_KIND_MEMORY; 684*0b57cec5SDimitry Andric case Attribute::Naked: 685*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NAKED; 686*0b57cec5SDimitry Andric case Attribute::Nest: 687*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NEST; 688*0b57cec5SDimitry Andric case Attribute::NoAlias: 689*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_ALIAS; 690*0b57cec5SDimitry Andric case Attribute::NoBuiltin: 691*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_BUILTIN; 692e8d8bef9SDimitry Andric case Attribute::NoCallback: 693e8d8bef9SDimitry Andric return bitc::ATTR_KIND_NO_CALLBACK; 694*0b57cec5SDimitry Andric case Attribute::NoCapture: 695*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_CAPTURE; 696*0b57cec5SDimitry Andric case Attribute::NoDuplicate: 697*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_DUPLICATE; 698*0b57cec5SDimitry Andric case Attribute::NoFree: 699*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NOFREE; 700*0b57cec5SDimitry Andric case Attribute::NoImplicitFloat: 701*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_IMPLICIT_FLOAT; 702*0b57cec5SDimitry Andric case Attribute::NoInline: 703*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_INLINE; 704*0b57cec5SDimitry Andric case Attribute::NoRecurse: 705*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_RECURSE; 7065ffd83dbSDimitry Andric case Attribute::NoMerge: 7075ffd83dbSDimitry Andric return bitc::ATTR_KIND_NO_MERGE; 708*0b57cec5SDimitry Andric case Attribute::NonLazyBind: 709*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NON_LAZY_BIND; 710*0b57cec5SDimitry Andric case Attribute::NonNull: 711*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NON_NULL; 712*0b57cec5SDimitry Andric case Attribute::Dereferenceable: 713*0b57cec5SDimitry Andric return bitc::ATTR_KIND_DEREFERENCEABLE; 714*0b57cec5SDimitry Andric case Attribute::DereferenceableOrNull: 715*0b57cec5SDimitry Andric return bitc::ATTR_KIND_DEREFERENCEABLE_OR_NULL; 716*0b57cec5SDimitry Andric case Attribute::NoRedZone: 717*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_RED_ZONE; 718*0b57cec5SDimitry Andric case Attribute::NoReturn: 719*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_RETURN; 720*0b57cec5SDimitry Andric case Attribute::NoSync: 721*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NOSYNC; 722*0b57cec5SDimitry Andric case Attribute::NoCfCheck: 723*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NOCF_CHECK; 724e8d8bef9SDimitry Andric case Attribute::NoProfile: 725e8d8bef9SDimitry Andric return bitc::ATTR_KIND_NO_PROFILE; 726bdd1243dSDimitry Andric case Attribute::SkipProfile: 727bdd1243dSDimitry Andric return bitc::ATTR_KIND_SKIP_PROFILE; 728*0b57cec5SDimitry Andric case Attribute::NoUnwind: 729*0b57cec5SDimitry Andric return bitc::ATTR_KIND_NO_UNWIND; 73081ad6265SDimitry Andric case Attribute::NoSanitizeBounds: 73181ad6265SDimitry Andric return bitc::ATTR_KIND_NO_SANITIZE_BOUNDS; 732fe6060f1SDimitry Andric case Attribute::NoSanitizeCoverage: 733fe6060f1SDimitry Andric return bitc::ATTR_KIND_NO_SANITIZE_COVERAGE; 7345ffd83dbSDimitry Andric case Attribute::NullPointerIsValid: 7355ffd83dbSDimitry Andric return bitc::ATTR_KIND_NULL_POINTER_IS_VALID; 736*0b57cec5SDimitry Andric case Attribute::OptForFuzzing: 737*0b57cec5SDimitry Andric return bitc::ATTR_KIND_OPT_FOR_FUZZING; 738*0b57cec5SDimitry Andric case Attribute::OptimizeForSize: 739*0b57cec5SDimitry Andric return bitc::ATTR_KIND_OPTIMIZE_FOR_SIZE; 740*0b57cec5SDimitry Andric case Attribute::OptimizeNone: 741*0b57cec5SDimitry Andric return bitc::ATTR_KIND_OPTIMIZE_NONE; 742*0b57cec5SDimitry Andric case Attribute::ReadNone: 743*0b57cec5SDimitry Andric return bitc::ATTR_KIND_READ_NONE; 744*0b57cec5SDimitry Andric case Attribute::ReadOnly: 745*0b57cec5SDimitry Andric return bitc::ATTR_KIND_READ_ONLY; 746*0b57cec5SDimitry Andric case Attribute::Returned: 747*0b57cec5SDimitry Andric return bitc::ATTR_KIND_RETURNED; 748*0b57cec5SDimitry Andric case Attribute::ReturnsTwice: 749*0b57cec5SDimitry Andric return bitc::ATTR_KIND_RETURNS_TWICE; 750*0b57cec5SDimitry Andric case Attribute::SExt: 751*0b57cec5SDimitry Andric return bitc::ATTR_KIND_S_EXT; 752*0b57cec5SDimitry Andric case Attribute::Speculatable: 753*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SPECULATABLE; 754*0b57cec5SDimitry Andric case Attribute::StackAlignment: 755*0b57cec5SDimitry Andric return bitc::ATTR_KIND_STACK_ALIGNMENT; 756*0b57cec5SDimitry Andric case Attribute::StackProtect: 757*0b57cec5SDimitry Andric return bitc::ATTR_KIND_STACK_PROTECT; 758*0b57cec5SDimitry Andric case Attribute::StackProtectReq: 759*0b57cec5SDimitry Andric return bitc::ATTR_KIND_STACK_PROTECT_REQ; 760*0b57cec5SDimitry Andric case Attribute::StackProtectStrong: 761*0b57cec5SDimitry Andric return bitc::ATTR_KIND_STACK_PROTECT_STRONG; 762*0b57cec5SDimitry Andric case Attribute::SafeStack: 763*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SAFESTACK; 764*0b57cec5SDimitry Andric case Attribute::ShadowCallStack: 765*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SHADOWCALLSTACK; 766*0b57cec5SDimitry Andric case Attribute::StrictFP: 767*0b57cec5SDimitry Andric return bitc::ATTR_KIND_STRICT_FP; 768*0b57cec5SDimitry Andric case Attribute::StructRet: 769*0b57cec5SDimitry Andric return bitc::ATTR_KIND_STRUCT_RET; 770*0b57cec5SDimitry Andric case Attribute::SanitizeAddress: 771*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_ADDRESS; 772*0b57cec5SDimitry Andric case Attribute::SanitizeHWAddress: 773*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_HWADDRESS; 774*0b57cec5SDimitry Andric case Attribute::SanitizeThread: 775*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_THREAD; 776*0b57cec5SDimitry Andric case Attribute::SanitizeMemory: 777*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_MEMORY; 778*0b57cec5SDimitry Andric case Attribute::SpeculativeLoadHardening: 779*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SPECULATIVE_LOAD_HARDENING; 780*0b57cec5SDimitry Andric case Attribute::SwiftError: 781*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SWIFT_ERROR; 782*0b57cec5SDimitry Andric case Attribute::SwiftSelf: 783*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SWIFT_SELF; 784fe6060f1SDimitry Andric case Attribute::SwiftAsync: 785fe6060f1SDimitry Andric return bitc::ATTR_KIND_SWIFT_ASYNC; 786*0b57cec5SDimitry Andric case Attribute::UWTable: 787*0b57cec5SDimitry Andric return bitc::ATTR_KIND_UW_TABLE; 788fe6060f1SDimitry Andric case Attribute::VScaleRange: 789fe6060f1SDimitry Andric return bitc::ATTR_KIND_VSCALE_RANGE; 790*0b57cec5SDimitry Andric case Attribute::WillReturn: 791*0b57cec5SDimitry Andric return bitc::ATTR_KIND_WILLRETURN; 792*0b57cec5SDimitry Andric case Attribute::WriteOnly: 793*0b57cec5SDimitry Andric return bitc::ATTR_KIND_WRITEONLY; 794*0b57cec5SDimitry Andric case Attribute::ZExt: 795*0b57cec5SDimitry Andric return bitc::ATTR_KIND_Z_EXT; 796*0b57cec5SDimitry Andric case Attribute::ImmArg: 797*0b57cec5SDimitry Andric return bitc::ATTR_KIND_IMMARG; 798*0b57cec5SDimitry Andric case Attribute::SanitizeMemTag: 799*0b57cec5SDimitry Andric return bitc::ATTR_KIND_SANITIZE_MEMTAG; 8005ffd83dbSDimitry Andric case Attribute::Preallocated: 8015ffd83dbSDimitry Andric return bitc::ATTR_KIND_PREALLOCATED; 8025ffd83dbSDimitry Andric case Attribute::NoUndef: 8035ffd83dbSDimitry Andric return bitc::ATTR_KIND_NOUNDEF; 804e8d8bef9SDimitry Andric case Attribute::ByRef: 805e8d8bef9SDimitry Andric return bitc::ATTR_KIND_BYREF; 806e8d8bef9SDimitry Andric case Attribute::MustProgress: 807e8d8bef9SDimitry Andric return bitc::ATTR_KIND_MUSTPROGRESS; 80881ad6265SDimitry Andric case Attribute::PresplitCoroutine: 80981ad6265SDimitry Andric return bitc::ATTR_KIND_PRESPLIT_COROUTINE; 810*0b57cec5SDimitry Andric case Attribute::EndAttrKinds: 811*0b57cec5SDimitry Andric llvm_unreachable("Can not encode end-attribute kinds marker."); 812*0b57cec5SDimitry Andric case Attribute::None: 813*0b57cec5SDimitry Andric llvm_unreachable("Can not encode none-attribute."); 8145ffd83dbSDimitry Andric case Attribute::EmptyKey: 8155ffd83dbSDimitry Andric case Attribute::TombstoneKey: 8165ffd83dbSDimitry Andric llvm_unreachable("Trying to encode EmptyKey/TombstoneKey"); 817*0b57cec5SDimitry Andric } 818*0b57cec5SDimitry Andric 819*0b57cec5SDimitry Andric llvm_unreachable("Trying to encode unknown attribute"); 820*0b57cec5SDimitry Andric } 821*0b57cec5SDimitry Andric 822*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeAttributeGroupTable() { 823*0b57cec5SDimitry Andric const std::vector<ValueEnumerator::IndexAndAttrSet> &AttrGrps = 824*0b57cec5SDimitry Andric VE.getAttributeGroups(); 825*0b57cec5SDimitry Andric if (AttrGrps.empty()) return; 826*0b57cec5SDimitry Andric 827*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::PARAMATTR_GROUP_BLOCK_ID, 3); 828*0b57cec5SDimitry Andric 829*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 830*0b57cec5SDimitry Andric for (ValueEnumerator::IndexAndAttrSet Pair : AttrGrps) { 831*0b57cec5SDimitry Andric unsigned AttrListIndex = Pair.first; 832*0b57cec5SDimitry Andric AttributeSet AS = Pair.second; 833*0b57cec5SDimitry Andric Record.push_back(VE.getAttributeGroupID(Pair)); 834*0b57cec5SDimitry Andric Record.push_back(AttrListIndex); 835*0b57cec5SDimitry Andric 836*0b57cec5SDimitry Andric for (Attribute Attr : AS) { 837*0b57cec5SDimitry Andric if (Attr.isEnumAttribute()) { 838*0b57cec5SDimitry Andric Record.push_back(0); 839*0b57cec5SDimitry Andric Record.push_back(getAttrKindEncoding(Attr.getKindAsEnum())); 840*0b57cec5SDimitry Andric } else if (Attr.isIntAttribute()) { 841*0b57cec5SDimitry Andric Record.push_back(1); 842*0b57cec5SDimitry Andric Record.push_back(getAttrKindEncoding(Attr.getKindAsEnum())); 843*0b57cec5SDimitry Andric Record.push_back(Attr.getValueAsInt()); 844*0b57cec5SDimitry Andric } else if (Attr.isStringAttribute()) { 845*0b57cec5SDimitry Andric StringRef Kind = Attr.getKindAsString(); 846*0b57cec5SDimitry Andric StringRef Val = Attr.getValueAsString(); 847*0b57cec5SDimitry Andric 848*0b57cec5SDimitry Andric Record.push_back(Val.empty() ? 3 : 4); 849*0b57cec5SDimitry Andric Record.append(Kind.begin(), Kind.end()); 850*0b57cec5SDimitry Andric Record.push_back(0); 851*0b57cec5SDimitry Andric if (!Val.empty()) { 852*0b57cec5SDimitry Andric Record.append(Val.begin(), Val.end()); 853*0b57cec5SDimitry Andric Record.push_back(0); 854*0b57cec5SDimitry Andric } 855*0b57cec5SDimitry Andric } else { 856*0b57cec5SDimitry Andric assert(Attr.isTypeAttribute()); 857*0b57cec5SDimitry Andric Type *Ty = Attr.getValueAsType(); 858*0b57cec5SDimitry Andric Record.push_back(Ty ? 6 : 5); 859*0b57cec5SDimitry Andric Record.push_back(getAttrKindEncoding(Attr.getKindAsEnum())); 860*0b57cec5SDimitry Andric if (Ty) 861*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(Attr.getValueAsType())); 862*0b57cec5SDimitry Andric } 863*0b57cec5SDimitry Andric } 864*0b57cec5SDimitry Andric 865*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::PARAMATTR_GRP_CODE_ENTRY, Record); 866*0b57cec5SDimitry Andric Record.clear(); 867*0b57cec5SDimitry Andric } 868*0b57cec5SDimitry Andric 869*0b57cec5SDimitry Andric Stream.ExitBlock(); 870*0b57cec5SDimitry Andric } 871*0b57cec5SDimitry Andric 872*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeAttributeTable() { 873*0b57cec5SDimitry Andric const std::vector<AttributeList> &Attrs = VE.getAttributeLists(); 874*0b57cec5SDimitry Andric if (Attrs.empty()) return; 875*0b57cec5SDimitry Andric 876*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::PARAMATTR_BLOCK_ID, 3); 877*0b57cec5SDimitry Andric 878*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 8790eae32dcSDimitry Andric for (const AttributeList &AL : Attrs) { 880349cc55cSDimitry Andric for (unsigned i : AL.indexes()) { 881*0b57cec5SDimitry Andric AttributeSet AS = AL.getAttributes(i); 882*0b57cec5SDimitry Andric if (AS.hasAttributes()) 883*0b57cec5SDimitry Andric Record.push_back(VE.getAttributeGroupID({i, AS})); 884*0b57cec5SDimitry Andric } 885*0b57cec5SDimitry Andric 886*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::PARAMATTR_CODE_ENTRY, Record); 887*0b57cec5SDimitry Andric Record.clear(); 888*0b57cec5SDimitry Andric } 889*0b57cec5SDimitry Andric 890*0b57cec5SDimitry Andric Stream.ExitBlock(); 891*0b57cec5SDimitry Andric } 892*0b57cec5SDimitry Andric 893*0b57cec5SDimitry Andric /// WriteTypeTable - Write out the type table for a module. 894*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeTypeTable() { 895*0b57cec5SDimitry Andric const ValueEnumerator::TypeList &TypeList = VE.getTypes(); 896*0b57cec5SDimitry Andric 897*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::TYPE_BLOCK_ID_NEW, 4 /*count from # abbrevs */); 898*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> TypeVals; 899*0b57cec5SDimitry Andric 900*0b57cec5SDimitry Andric uint64_t NumBits = VE.computeBitsRequiredForTypeIndicies(); 901*0b57cec5SDimitry Andric 902*0b57cec5SDimitry Andric // Abbrev for TYPE_CODE_POINTER. 903*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 904*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_POINTER)); 905*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 906*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(0)); // Addrspace = 0 907*0b57cec5SDimitry Andric unsigned PtrAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 908*0b57cec5SDimitry Andric 909fe6060f1SDimitry Andric // Abbrev for TYPE_CODE_OPAQUE_POINTER. 910fe6060f1SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 911fe6060f1SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_OPAQUE_POINTER)); 912fe6060f1SDimitry Andric Abbv->Add(BitCodeAbbrevOp(0)); // Addrspace = 0 913fe6060f1SDimitry Andric unsigned OpaquePtrAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 914fe6060f1SDimitry Andric 915*0b57cec5SDimitry Andric // Abbrev for TYPE_CODE_FUNCTION. 916*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 917*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_FUNCTION)); 918*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // isvararg 919*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 920*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 921*0b57cec5SDimitry Andric unsigned FunctionAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 922*0b57cec5SDimitry Andric 923*0b57cec5SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_ANON. 924*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 925*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_ANON)); 926*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // ispacked 927*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 928*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 929*0b57cec5SDimitry Andric unsigned StructAnonAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 930*0b57cec5SDimitry Andric 931*0b57cec5SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_NAME. 932*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 933*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_NAME)); 934*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 935*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 936*0b57cec5SDimitry Andric unsigned StructNameAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 937*0b57cec5SDimitry Andric 938*0b57cec5SDimitry Andric // Abbrev for TYPE_CODE_STRUCT_NAMED. 939*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 940*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_STRUCT_NAMED)); 941*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // ispacked 942*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 943*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 944*0b57cec5SDimitry Andric unsigned StructNamedAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 945*0b57cec5SDimitry Andric 946*0b57cec5SDimitry Andric // Abbrev for TYPE_CODE_ARRAY. 947*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 948*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::TYPE_CODE_ARRAY)); 949*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // size 950*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, NumBits)); 951*0b57cec5SDimitry Andric unsigned ArrayAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 952*0b57cec5SDimitry Andric 953*0b57cec5SDimitry Andric // Emit an entry count so the reader can reserve space. 954*0b57cec5SDimitry Andric TypeVals.push_back(TypeList.size()); 955*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::TYPE_CODE_NUMENTRY, TypeVals); 956*0b57cec5SDimitry Andric TypeVals.clear(); 957*0b57cec5SDimitry Andric 958*0b57cec5SDimitry Andric // Loop over all of the types, emitting each in turn. 9594824e7fdSDimitry Andric for (Type *T : TypeList) { 960*0b57cec5SDimitry Andric int AbbrevToUse = 0; 961*0b57cec5SDimitry Andric unsigned Code = 0; 962*0b57cec5SDimitry Andric 963*0b57cec5SDimitry Andric switch (T->getTypeID()) { 964*0b57cec5SDimitry Andric case Type::VoidTyID: Code = bitc::TYPE_CODE_VOID; break; 965*0b57cec5SDimitry Andric case Type::HalfTyID: Code = bitc::TYPE_CODE_HALF; break; 9665ffd83dbSDimitry Andric case Type::BFloatTyID: Code = bitc::TYPE_CODE_BFLOAT; break; 967*0b57cec5SDimitry Andric case Type::FloatTyID: Code = bitc::TYPE_CODE_FLOAT; break; 968*0b57cec5SDimitry Andric case Type::DoubleTyID: Code = bitc::TYPE_CODE_DOUBLE; break; 969*0b57cec5SDimitry Andric case Type::X86_FP80TyID: Code = bitc::TYPE_CODE_X86_FP80; break; 970*0b57cec5SDimitry Andric case Type::FP128TyID: Code = bitc::TYPE_CODE_FP128; break; 971*0b57cec5SDimitry Andric case Type::PPC_FP128TyID: Code = bitc::TYPE_CODE_PPC_FP128; break; 972*0b57cec5SDimitry Andric case Type::LabelTyID: Code = bitc::TYPE_CODE_LABEL; break; 973*0b57cec5SDimitry Andric case Type::MetadataTyID: Code = bitc::TYPE_CODE_METADATA; break; 974*0b57cec5SDimitry Andric case Type::X86_MMXTyID: Code = bitc::TYPE_CODE_X86_MMX; break; 975e8d8bef9SDimitry Andric case Type::X86_AMXTyID: Code = bitc::TYPE_CODE_X86_AMX; break; 976*0b57cec5SDimitry Andric case Type::TokenTyID: Code = bitc::TYPE_CODE_TOKEN; break; 977*0b57cec5SDimitry Andric case Type::IntegerTyID: 978*0b57cec5SDimitry Andric // INTEGER: [width] 979*0b57cec5SDimitry Andric Code = bitc::TYPE_CODE_INTEGER; 980*0b57cec5SDimitry Andric TypeVals.push_back(cast<IntegerType>(T)->getBitWidth()); 981*0b57cec5SDimitry Andric break; 982*0b57cec5SDimitry Andric case Type::PointerTyID: { 983*0b57cec5SDimitry Andric PointerType *PTy = cast<PointerType>(T); 984fe6060f1SDimitry Andric unsigned AddressSpace = PTy->getAddressSpace(); 985fe6060f1SDimitry Andric if (PTy->isOpaque()) { 986fe6060f1SDimitry Andric // OPAQUE_POINTER: [address space] 987fe6060f1SDimitry Andric Code = bitc::TYPE_CODE_OPAQUE_POINTER; 988fe6060f1SDimitry Andric TypeVals.push_back(AddressSpace); 989fe6060f1SDimitry Andric if (AddressSpace == 0) 990fe6060f1SDimitry Andric AbbrevToUse = OpaquePtrAbbrev; 991fe6060f1SDimitry Andric } else { 992*0b57cec5SDimitry Andric // POINTER: [pointee type, address space] 993*0b57cec5SDimitry Andric Code = bitc::TYPE_CODE_POINTER; 99404eeddc0SDimitry Andric TypeVals.push_back(VE.getTypeID(PTy->getNonOpaquePointerElementType())); 995*0b57cec5SDimitry Andric TypeVals.push_back(AddressSpace); 996fe6060f1SDimitry Andric if (AddressSpace == 0) 997fe6060f1SDimitry Andric AbbrevToUse = PtrAbbrev; 998fe6060f1SDimitry Andric } 999*0b57cec5SDimitry Andric break; 1000*0b57cec5SDimitry Andric } 1001*0b57cec5SDimitry Andric case Type::FunctionTyID: { 1002*0b57cec5SDimitry Andric FunctionType *FT = cast<FunctionType>(T); 1003*0b57cec5SDimitry Andric // FUNCTION: [isvararg, retty, paramty x N] 1004*0b57cec5SDimitry Andric Code = bitc::TYPE_CODE_FUNCTION; 1005*0b57cec5SDimitry Andric TypeVals.push_back(FT->isVarArg()); 1006*0b57cec5SDimitry Andric TypeVals.push_back(VE.getTypeID(FT->getReturnType())); 1007*0b57cec5SDimitry Andric for (unsigned i = 0, e = FT->getNumParams(); i != e; ++i) 1008*0b57cec5SDimitry Andric TypeVals.push_back(VE.getTypeID(FT->getParamType(i))); 1009*0b57cec5SDimitry Andric AbbrevToUse = FunctionAbbrev; 1010*0b57cec5SDimitry Andric break; 1011*0b57cec5SDimitry Andric } 1012*0b57cec5SDimitry Andric case Type::StructTyID: { 1013*0b57cec5SDimitry Andric StructType *ST = cast<StructType>(T); 1014*0b57cec5SDimitry Andric // STRUCT: [ispacked, eltty x N] 1015*0b57cec5SDimitry Andric TypeVals.push_back(ST->isPacked()); 1016*0b57cec5SDimitry Andric // Output all of the element types. 1017349cc55cSDimitry Andric for (Type *ET : ST->elements()) 1018349cc55cSDimitry Andric TypeVals.push_back(VE.getTypeID(ET)); 1019*0b57cec5SDimitry Andric 1020*0b57cec5SDimitry Andric if (ST->isLiteral()) { 1021*0b57cec5SDimitry Andric Code = bitc::TYPE_CODE_STRUCT_ANON; 1022*0b57cec5SDimitry Andric AbbrevToUse = StructAnonAbbrev; 1023*0b57cec5SDimitry Andric } else { 1024*0b57cec5SDimitry Andric if (ST->isOpaque()) { 1025*0b57cec5SDimitry Andric Code = bitc::TYPE_CODE_OPAQUE; 1026*0b57cec5SDimitry Andric } else { 1027*0b57cec5SDimitry Andric Code = bitc::TYPE_CODE_STRUCT_NAMED; 1028*0b57cec5SDimitry Andric AbbrevToUse = StructNamedAbbrev; 1029*0b57cec5SDimitry Andric } 1030*0b57cec5SDimitry Andric 1031*0b57cec5SDimitry Andric // Emit the name if it is present. 1032*0b57cec5SDimitry Andric if (!ST->getName().empty()) 1033*0b57cec5SDimitry Andric writeStringRecord(Stream, bitc::TYPE_CODE_STRUCT_NAME, ST->getName(), 1034*0b57cec5SDimitry Andric StructNameAbbrev); 1035*0b57cec5SDimitry Andric } 1036*0b57cec5SDimitry Andric break; 1037*0b57cec5SDimitry Andric } 1038*0b57cec5SDimitry Andric case Type::ArrayTyID: { 1039*0b57cec5SDimitry Andric ArrayType *AT = cast<ArrayType>(T); 1040*0b57cec5SDimitry Andric // ARRAY: [numelts, eltty] 1041*0b57cec5SDimitry Andric Code = bitc::TYPE_CODE_ARRAY; 1042*0b57cec5SDimitry Andric TypeVals.push_back(AT->getNumElements()); 1043*0b57cec5SDimitry Andric TypeVals.push_back(VE.getTypeID(AT->getElementType())); 1044*0b57cec5SDimitry Andric AbbrevToUse = ArrayAbbrev; 1045*0b57cec5SDimitry Andric break; 1046*0b57cec5SDimitry Andric } 10475ffd83dbSDimitry Andric case Type::FixedVectorTyID: 10485ffd83dbSDimitry Andric case Type::ScalableVectorTyID: { 1049*0b57cec5SDimitry Andric VectorType *VT = cast<VectorType>(T); 1050*0b57cec5SDimitry Andric // VECTOR [numelts, eltty] or 1051*0b57cec5SDimitry Andric // [numelts, eltty, scalable] 1052*0b57cec5SDimitry Andric Code = bitc::TYPE_CODE_VECTOR; 1053e8d8bef9SDimitry Andric TypeVals.push_back(VT->getElementCount().getKnownMinValue()); 1054*0b57cec5SDimitry Andric TypeVals.push_back(VE.getTypeID(VT->getElementType())); 10555ffd83dbSDimitry Andric if (isa<ScalableVectorType>(VT)) 10565ffd83dbSDimitry Andric TypeVals.push_back(true); 1057*0b57cec5SDimitry Andric break; 1058*0b57cec5SDimitry Andric } 1059bdd1243dSDimitry Andric case Type::TargetExtTyID: { 1060bdd1243dSDimitry Andric TargetExtType *TET = cast<TargetExtType>(T); 1061bdd1243dSDimitry Andric Code = bitc::TYPE_CODE_TARGET_TYPE; 1062bdd1243dSDimitry Andric writeStringRecord(Stream, bitc::TYPE_CODE_STRUCT_NAME, TET->getName(), 1063bdd1243dSDimitry Andric StructNameAbbrev); 1064bdd1243dSDimitry Andric TypeVals.push_back(TET->getNumTypeParameters()); 1065bdd1243dSDimitry Andric for (Type *InnerTy : TET->type_params()) 1066bdd1243dSDimitry Andric TypeVals.push_back(VE.getTypeID(InnerTy)); 1067bdd1243dSDimitry Andric for (unsigned IntParam : TET->int_params()) 1068bdd1243dSDimitry Andric TypeVals.push_back(IntParam); 1069bdd1243dSDimitry Andric break; 1070bdd1243dSDimitry Andric } 1071bdd1243dSDimitry Andric case Type::TypedPointerTyID: 1072bdd1243dSDimitry Andric llvm_unreachable("Typed pointers cannot be added to IR modules"); 1073*0b57cec5SDimitry Andric } 1074*0b57cec5SDimitry Andric 1075*0b57cec5SDimitry Andric // Emit the finished record. 1076*0b57cec5SDimitry Andric Stream.EmitRecord(Code, TypeVals, AbbrevToUse); 1077*0b57cec5SDimitry Andric TypeVals.clear(); 1078*0b57cec5SDimitry Andric } 1079*0b57cec5SDimitry Andric 1080*0b57cec5SDimitry Andric Stream.ExitBlock(); 1081*0b57cec5SDimitry Andric } 1082*0b57cec5SDimitry Andric 1083*0b57cec5SDimitry Andric static unsigned getEncodedLinkage(const GlobalValue::LinkageTypes Linkage) { 1084*0b57cec5SDimitry Andric switch (Linkage) { 1085*0b57cec5SDimitry Andric case GlobalValue::ExternalLinkage: 1086*0b57cec5SDimitry Andric return 0; 1087*0b57cec5SDimitry Andric case GlobalValue::WeakAnyLinkage: 1088*0b57cec5SDimitry Andric return 16; 1089*0b57cec5SDimitry Andric case GlobalValue::AppendingLinkage: 1090*0b57cec5SDimitry Andric return 2; 1091*0b57cec5SDimitry Andric case GlobalValue::InternalLinkage: 1092*0b57cec5SDimitry Andric return 3; 1093*0b57cec5SDimitry Andric case GlobalValue::LinkOnceAnyLinkage: 1094*0b57cec5SDimitry Andric return 18; 1095*0b57cec5SDimitry Andric case GlobalValue::ExternalWeakLinkage: 1096*0b57cec5SDimitry Andric return 7; 1097*0b57cec5SDimitry Andric case GlobalValue::CommonLinkage: 1098*0b57cec5SDimitry Andric return 8; 1099*0b57cec5SDimitry Andric case GlobalValue::PrivateLinkage: 1100*0b57cec5SDimitry Andric return 9; 1101*0b57cec5SDimitry Andric case GlobalValue::WeakODRLinkage: 1102*0b57cec5SDimitry Andric return 17; 1103*0b57cec5SDimitry Andric case GlobalValue::LinkOnceODRLinkage: 1104*0b57cec5SDimitry Andric return 19; 1105*0b57cec5SDimitry Andric case GlobalValue::AvailableExternallyLinkage: 1106*0b57cec5SDimitry Andric return 12; 1107*0b57cec5SDimitry Andric } 1108*0b57cec5SDimitry Andric llvm_unreachable("Invalid linkage"); 1109*0b57cec5SDimitry Andric } 1110*0b57cec5SDimitry Andric 1111*0b57cec5SDimitry Andric static unsigned getEncodedLinkage(const GlobalValue &GV) { 1112*0b57cec5SDimitry Andric return getEncodedLinkage(GV.getLinkage()); 1113*0b57cec5SDimitry Andric } 1114*0b57cec5SDimitry Andric 1115*0b57cec5SDimitry Andric static uint64_t getEncodedFFlags(FunctionSummary::FFlags Flags) { 1116*0b57cec5SDimitry Andric uint64_t RawFlags = 0; 1117*0b57cec5SDimitry Andric RawFlags |= Flags.ReadNone; 1118*0b57cec5SDimitry Andric RawFlags |= (Flags.ReadOnly << 1); 1119*0b57cec5SDimitry Andric RawFlags |= (Flags.NoRecurse << 2); 1120*0b57cec5SDimitry Andric RawFlags |= (Flags.ReturnDoesNotAlias << 3); 1121*0b57cec5SDimitry Andric RawFlags |= (Flags.NoInline << 4); 1122480093f4SDimitry Andric RawFlags |= (Flags.AlwaysInline << 5); 1123349cc55cSDimitry Andric RawFlags |= (Flags.NoUnwind << 6); 1124349cc55cSDimitry Andric RawFlags |= (Flags.MayThrow << 7); 1125349cc55cSDimitry Andric RawFlags |= (Flags.HasUnknownCall << 8); 11260eae32dcSDimitry Andric RawFlags |= (Flags.MustBeUnreachable << 9); 1127*0b57cec5SDimitry Andric return RawFlags; 1128*0b57cec5SDimitry Andric } 1129*0b57cec5SDimitry Andric 1130fe6060f1SDimitry Andric // Decode the flags for GlobalValue in the summary. See getDecodedGVSummaryFlags 1131fe6060f1SDimitry Andric // in BitcodeReader.cpp. 1132*0b57cec5SDimitry Andric static uint64_t getEncodedGVSummaryFlags(GlobalValueSummary::GVFlags Flags) { 1133*0b57cec5SDimitry Andric uint64_t RawFlags = 0; 1134*0b57cec5SDimitry Andric 1135*0b57cec5SDimitry Andric RawFlags |= Flags.NotEligibleToImport; // bool 1136*0b57cec5SDimitry Andric RawFlags |= (Flags.Live << 1); 1137*0b57cec5SDimitry Andric RawFlags |= (Flags.DSOLocal << 2); 1138*0b57cec5SDimitry Andric RawFlags |= (Flags.CanAutoHide << 3); 1139*0b57cec5SDimitry Andric 1140*0b57cec5SDimitry Andric // Linkage don't need to be remapped at that time for the summary. Any future 1141*0b57cec5SDimitry Andric // change to the getEncodedLinkage() function will need to be taken into 1142*0b57cec5SDimitry Andric // account here as well. 1143*0b57cec5SDimitry Andric RawFlags = (RawFlags << 4) | Flags.Linkage; // 4 bits 1144*0b57cec5SDimitry Andric 1145fe6060f1SDimitry Andric RawFlags |= (Flags.Visibility << 8); // 2 bits 1146fe6060f1SDimitry Andric 1147*0b57cec5SDimitry Andric return RawFlags; 1148*0b57cec5SDimitry Andric } 1149*0b57cec5SDimitry Andric 1150*0b57cec5SDimitry Andric static uint64_t getEncodedGVarFlags(GlobalVarSummary::GVarFlags Flags) { 11515ffd83dbSDimitry Andric uint64_t RawFlags = Flags.MaybeReadOnly | (Flags.MaybeWriteOnly << 1) | 11525ffd83dbSDimitry Andric (Flags.Constant << 2) | Flags.VCallVisibility << 3; 1153*0b57cec5SDimitry Andric return RawFlags; 1154*0b57cec5SDimitry Andric } 1155*0b57cec5SDimitry Andric 1156*0b57cec5SDimitry Andric static unsigned getEncodedVisibility(const GlobalValue &GV) { 1157*0b57cec5SDimitry Andric switch (GV.getVisibility()) { 1158*0b57cec5SDimitry Andric case GlobalValue::DefaultVisibility: return 0; 1159*0b57cec5SDimitry Andric case GlobalValue::HiddenVisibility: return 1; 1160*0b57cec5SDimitry Andric case GlobalValue::ProtectedVisibility: return 2; 1161*0b57cec5SDimitry Andric } 1162*0b57cec5SDimitry Andric llvm_unreachable("Invalid visibility"); 1163*0b57cec5SDimitry Andric } 1164*0b57cec5SDimitry Andric 1165*0b57cec5SDimitry Andric static unsigned getEncodedDLLStorageClass(const GlobalValue &GV) { 1166*0b57cec5SDimitry Andric switch (GV.getDLLStorageClass()) { 1167*0b57cec5SDimitry Andric case GlobalValue::DefaultStorageClass: return 0; 1168*0b57cec5SDimitry Andric case GlobalValue::DLLImportStorageClass: return 1; 1169*0b57cec5SDimitry Andric case GlobalValue::DLLExportStorageClass: return 2; 1170*0b57cec5SDimitry Andric } 1171*0b57cec5SDimitry Andric llvm_unreachable("Invalid DLL storage class"); 1172*0b57cec5SDimitry Andric } 1173*0b57cec5SDimitry Andric 1174*0b57cec5SDimitry Andric static unsigned getEncodedThreadLocalMode(const GlobalValue &GV) { 1175*0b57cec5SDimitry Andric switch (GV.getThreadLocalMode()) { 1176*0b57cec5SDimitry Andric case GlobalVariable::NotThreadLocal: return 0; 1177*0b57cec5SDimitry Andric case GlobalVariable::GeneralDynamicTLSModel: return 1; 1178*0b57cec5SDimitry Andric case GlobalVariable::LocalDynamicTLSModel: return 2; 1179*0b57cec5SDimitry Andric case GlobalVariable::InitialExecTLSModel: return 3; 1180*0b57cec5SDimitry Andric case GlobalVariable::LocalExecTLSModel: return 4; 1181*0b57cec5SDimitry Andric } 1182*0b57cec5SDimitry Andric llvm_unreachable("Invalid TLS model"); 1183*0b57cec5SDimitry Andric } 1184*0b57cec5SDimitry Andric 1185*0b57cec5SDimitry Andric static unsigned getEncodedComdatSelectionKind(const Comdat &C) { 1186*0b57cec5SDimitry Andric switch (C.getSelectionKind()) { 1187*0b57cec5SDimitry Andric case Comdat::Any: 1188*0b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_ANY; 1189*0b57cec5SDimitry Andric case Comdat::ExactMatch: 1190*0b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_EXACT_MATCH; 1191*0b57cec5SDimitry Andric case Comdat::Largest: 1192*0b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_LARGEST; 1193fe6060f1SDimitry Andric case Comdat::NoDeduplicate: 1194*0b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_NO_DUPLICATES; 1195*0b57cec5SDimitry Andric case Comdat::SameSize: 1196*0b57cec5SDimitry Andric return bitc::COMDAT_SELECTION_KIND_SAME_SIZE; 1197*0b57cec5SDimitry Andric } 1198*0b57cec5SDimitry Andric llvm_unreachable("Invalid selection kind"); 1199*0b57cec5SDimitry Andric } 1200*0b57cec5SDimitry Andric 1201*0b57cec5SDimitry Andric static unsigned getEncodedUnnamedAddr(const GlobalValue &GV) { 1202*0b57cec5SDimitry Andric switch (GV.getUnnamedAddr()) { 1203*0b57cec5SDimitry Andric case GlobalValue::UnnamedAddr::None: return 0; 1204*0b57cec5SDimitry Andric case GlobalValue::UnnamedAddr::Local: return 2; 1205*0b57cec5SDimitry Andric case GlobalValue::UnnamedAddr::Global: return 1; 1206*0b57cec5SDimitry Andric } 1207*0b57cec5SDimitry Andric llvm_unreachable("Invalid unnamed_addr"); 1208*0b57cec5SDimitry Andric } 1209*0b57cec5SDimitry Andric 1210*0b57cec5SDimitry Andric size_t ModuleBitcodeWriter::addToStrtab(StringRef Str) { 1211*0b57cec5SDimitry Andric if (GenerateHash) 1212*0b57cec5SDimitry Andric Hasher.update(Str); 1213*0b57cec5SDimitry Andric return StrtabBuilder.add(Str); 1214*0b57cec5SDimitry Andric } 1215*0b57cec5SDimitry Andric 1216*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeComdats() { 1217*0b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 1218*0b57cec5SDimitry Andric for (const Comdat *C : VE.getComdats()) { 1219*0b57cec5SDimitry Andric // COMDAT: [strtab offset, strtab size, selection_kind] 1220*0b57cec5SDimitry Andric Vals.push_back(addToStrtab(C->getName())); 1221*0b57cec5SDimitry Andric Vals.push_back(C->getName().size()); 1222*0b57cec5SDimitry Andric Vals.push_back(getEncodedComdatSelectionKind(*C)); 1223*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_COMDAT, Vals, /*AbbrevToUse=*/0); 1224*0b57cec5SDimitry Andric Vals.clear(); 1225*0b57cec5SDimitry Andric } 1226*0b57cec5SDimitry Andric } 1227*0b57cec5SDimitry Andric 1228*0b57cec5SDimitry Andric /// Write a record that will eventually hold the word offset of the 1229*0b57cec5SDimitry Andric /// module-level VST. For now the offset is 0, which will be backpatched 1230*0b57cec5SDimitry Andric /// after the real VST is written. Saves the bit offset to backpatch. 1231*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeValueSymbolTableForwardDecl() { 1232*0b57cec5SDimitry Andric // Write a placeholder value in for the offset of the real VST, 1233*0b57cec5SDimitry Andric // which is written after the function blocks so that it can include 1234*0b57cec5SDimitry Andric // the offset of each function. The placeholder offset will be 1235*0b57cec5SDimitry Andric // updated when the real VST is written. 1236*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 1237*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_VSTOFFSET)); 1238*0b57cec5SDimitry Andric // Blocks are 32-bit aligned, so we can use a 32-bit word offset to 1239*0b57cec5SDimitry Andric // hold the real VST offset. Must use fixed instead of VBR as we don't 1240*0b57cec5SDimitry Andric // know how many VBR chunks to reserve ahead of time. 1241*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 1242*0b57cec5SDimitry Andric unsigned VSTOffsetAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 1243*0b57cec5SDimitry Andric 1244*0b57cec5SDimitry Andric // Emit the placeholder 1245*0b57cec5SDimitry Andric uint64_t Vals[] = {bitc::MODULE_CODE_VSTOFFSET, 0}; 1246*0b57cec5SDimitry Andric Stream.EmitRecordWithAbbrev(VSTOffsetAbbrev, Vals); 1247*0b57cec5SDimitry Andric 1248*0b57cec5SDimitry Andric // Compute and save the bit offset to the placeholder, which will be 1249*0b57cec5SDimitry Andric // patched when the real VST is written. We can simply subtract the 32-bit 1250*0b57cec5SDimitry Andric // fixed size from the current bit number to get the location to backpatch. 1251*0b57cec5SDimitry Andric VSTOffsetPlaceholder = Stream.GetCurrentBitNo() - 32; 1252*0b57cec5SDimitry Andric } 1253*0b57cec5SDimitry Andric 1254*0b57cec5SDimitry Andric enum StringEncoding { SE_Char6, SE_Fixed7, SE_Fixed8 }; 1255*0b57cec5SDimitry Andric 1256*0b57cec5SDimitry Andric /// Determine the encoding to use for the given string name and length. 1257*0b57cec5SDimitry Andric static StringEncoding getStringEncoding(StringRef Str) { 1258*0b57cec5SDimitry Andric bool isChar6 = true; 1259*0b57cec5SDimitry Andric for (char C : Str) { 1260*0b57cec5SDimitry Andric if (isChar6) 1261*0b57cec5SDimitry Andric isChar6 = BitCodeAbbrevOp::isChar6(C); 1262*0b57cec5SDimitry Andric if ((unsigned char)C & 128) 1263*0b57cec5SDimitry Andric // don't bother scanning the rest. 1264*0b57cec5SDimitry Andric return SE_Fixed8; 1265*0b57cec5SDimitry Andric } 1266*0b57cec5SDimitry Andric if (isChar6) 1267*0b57cec5SDimitry Andric return SE_Char6; 1268*0b57cec5SDimitry Andric return SE_Fixed7; 1269*0b57cec5SDimitry Andric } 1270*0b57cec5SDimitry Andric 127181ad6265SDimitry Andric static_assert(sizeof(GlobalValue::SanitizerMetadata) <= sizeof(unsigned), 127281ad6265SDimitry Andric "Sanitizer Metadata is too large for naive serialization."); 127381ad6265SDimitry Andric static unsigned 127481ad6265SDimitry Andric serializeSanitizerMetadata(const GlobalValue::SanitizerMetadata &Meta) { 127581ad6265SDimitry Andric return Meta.NoAddress | (Meta.NoHWAddress << 1) | 1276753f127fSDimitry Andric (Meta.Memtag << 2) | (Meta.IsDynInit << 3); 127781ad6265SDimitry Andric } 127881ad6265SDimitry Andric 1279*0b57cec5SDimitry Andric /// Emit top-level description of module, including target triple, inline asm, 1280*0b57cec5SDimitry Andric /// descriptors for global variables, and function prototype info. 1281*0b57cec5SDimitry Andric /// Returns the bit offset to backpatch with the location of the real VST. 1282*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleInfo() { 1283*0b57cec5SDimitry Andric // Emit various pieces of data attached to a module. 1284*0b57cec5SDimitry Andric if (!M.getTargetTriple().empty()) 1285*0b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_TRIPLE, M.getTargetTriple(), 1286*0b57cec5SDimitry Andric 0 /*TODO*/); 1287*0b57cec5SDimitry Andric const std::string &DL = M.getDataLayoutStr(); 1288*0b57cec5SDimitry Andric if (!DL.empty()) 1289*0b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_DATALAYOUT, DL, 0 /*TODO*/); 1290*0b57cec5SDimitry Andric if (!M.getModuleInlineAsm().empty()) 1291*0b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_ASM, M.getModuleInlineAsm(), 1292*0b57cec5SDimitry Andric 0 /*TODO*/); 1293*0b57cec5SDimitry Andric 1294*0b57cec5SDimitry Andric // Emit information about sections and GC, computing how many there are. Also 1295*0b57cec5SDimitry Andric // compute the maximum alignment value. 1296*0b57cec5SDimitry Andric std::map<std::string, unsigned> SectionMap; 1297*0b57cec5SDimitry Andric std::map<std::string, unsigned> GCMap; 1298e8d8bef9SDimitry Andric MaybeAlign MaxAlignment; 1299*0b57cec5SDimitry Andric unsigned MaxGlobalType = 0; 1300e8d8bef9SDimitry Andric const auto UpdateMaxAlignment = [&MaxAlignment](const MaybeAlign A) { 1301e8d8bef9SDimitry Andric if (A) 1302e8d8bef9SDimitry Andric MaxAlignment = !MaxAlignment ? *A : std::max(*MaxAlignment, *A); 1303e8d8bef9SDimitry Andric }; 13045ffd83dbSDimitry Andric for (const GlobalVariable &GV : M.globals()) { 1305e8d8bef9SDimitry Andric UpdateMaxAlignment(GV.getAlign()); 1306*0b57cec5SDimitry Andric MaxGlobalType = std::max(MaxGlobalType, VE.getTypeID(GV.getValueType())); 1307*0b57cec5SDimitry Andric if (GV.hasSection()) { 1308*0b57cec5SDimitry Andric // Give section names unique ID's. 13095ffd83dbSDimitry Andric unsigned &Entry = SectionMap[std::string(GV.getSection())]; 1310*0b57cec5SDimitry Andric if (!Entry) { 1311*0b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_SECTIONNAME, GV.getSection(), 1312*0b57cec5SDimitry Andric 0 /*TODO*/); 1313*0b57cec5SDimitry Andric Entry = SectionMap.size(); 1314*0b57cec5SDimitry Andric } 1315*0b57cec5SDimitry Andric } 1316*0b57cec5SDimitry Andric } 1317*0b57cec5SDimitry Andric for (const Function &F : M) { 1318e8d8bef9SDimitry Andric UpdateMaxAlignment(F.getAlign()); 1319*0b57cec5SDimitry Andric if (F.hasSection()) { 1320*0b57cec5SDimitry Andric // Give section names unique ID's. 13215ffd83dbSDimitry Andric unsigned &Entry = SectionMap[std::string(F.getSection())]; 1322*0b57cec5SDimitry Andric if (!Entry) { 1323*0b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_SECTIONNAME, F.getSection(), 1324*0b57cec5SDimitry Andric 0 /*TODO*/); 1325*0b57cec5SDimitry Andric Entry = SectionMap.size(); 1326*0b57cec5SDimitry Andric } 1327*0b57cec5SDimitry Andric } 1328*0b57cec5SDimitry Andric if (F.hasGC()) { 1329*0b57cec5SDimitry Andric // Same for GC names. 1330*0b57cec5SDimitry Andric unsigned &Entry = GCMap[F.getGC()]; 1331*0b57cec5SDimitry Andric if (!Entry) { 1332*0b57cec5SDimitry Andric writeStringRecord(Stream, bitc::MODULE_CODE_GCNAME, F.getGC(), 1333*0b57cec5SDimitry Andric 0 /*TODO*/); 1334*0b57cec5SDimitry Andric Entry = GCMap.size(); 1335*0b57cec5SDimitry Andric } 1336*0b57cec5SDimitry Andric } 1337*0b57cec5SDimitry Andric } 1338*0b57cec5SDimitry Andric 1339*0b57cec5SDimitry Andric // Emit abbrev for globals, now that we know # sections and max alignment. 1340*0b57cec5SDimitry Andric unsigned SimpleGVarAbbrev = 0; 1341*0b57cec5SDimitry Andric if (!M.global_empty()) { 1342*0b57cec5SDimitry Andric // Add an abbrev for common globals with no visibility or thread localness. 1343*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 1344*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_GLOBALVAR)); 1345*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1346*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1347*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1348*0b57cec5SDimitry Andric Log2_32_Ceil(MaxGlobalType+1))); 1349*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // AddrSpace << 2 1350*0b57cec5SDimitry Andric //| explicitType << 1 1351*0b57cec5SDimitry Andric //| constant 1352*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // Initializer. 1353*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 5)); // Linkage. 1354e8d8bef9SDimitry Andric if (!MaxAlignment) // Alignment. 1355*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(0)); 1356*0b57cec5SDimitry Andric else { 1357e8d8bef9SDimitry Andric unsigned MaxEncAlignment = getEncodedAlign(MaxAlignment); 1358*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1359*0b57cec5SDimitry Andric Log2_32_Ceil(MaxEncAlignment+1))); 1360*0b57cec5SDimitry Andric } 1361*0b57cec5SDimitry Andric if (SectionMap.empty()) // Section. 1362*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(0)); 1363*0b57cec5SDimitry Andric else 1364*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1365*0b57cec5SDimitry Andric Log2_32_Ceil(SectionMap.size()+1))); 1366*0b57cec5SDimitry Andric // Don't bother emitting vis + thread local. 1367*0b57cec5SDimitry Andric SimpleGVarAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 1368*0b57cec5SDimitry Andric } 1369*0b57cec5SDimitry Andric 1370*0b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 1371*0b57cec5SDimitry Andric // Emit the module's source file name. 1372*0b57cec5SDimitry Andric { 1373*0b57cec5SDimitry Andric StringEncoding Bits = getStringEncoding(M.getSourceFileName()); 1374*0b57cec5SDimitry Andric BitCodeAbbrevOp AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8); 1375*0b57cec5SDimitry Andric if (Bits == SE_Char6) 1376*0b57cec5SDimitry Andric AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Char6); 1377*0b57cec5SDimitry Andric else if (Bits == SE_Fixed7) 1378*0b57cec5SDimitry Andric AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7); 1379*0b57cec5SDimitry Andric 1380*0b57cec5SDimitry Andric // MODULE_CODE_SOURCE_FILENAME: [namechar x N] 1381*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 1382*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_SOURCE_FILENAME)); 1383*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1384*0b57cec5SDimitry Andric Abbv->Add(AbbrevOpToUse); 1385*0b57cec5SDimitry Andric unsigned FilenameAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 1386*0b57cec5SDimitry Andric 1387*0b57cec5SDimitry Andric for (const auto P : M.getSourceFileName()) 1388*0b57cec5SDimitry Andric Vals.push_back((unsigned char)P); 1389*0b57cec5SDimitry Andric 1390*0b57cec5SDimitry Andric // Emit the finished record. 1391*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_SOURCE_FILENAME, Vals, FilenameAbbrev); 1392*0b57cec5SDimitry Andric Vals.clear(); 1393*0b57cec5SDimitry Andric } 1394*0b57cec5SDimitry Andric 1395*0b57cec5SDimitry Andric // Emit the global variable information. 1396*0b57cec5SDimitry Andric for (const GlobalVariable &GV : M.globals()) { 1397*0b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 1398*0b57cec5SDimitry Andric 1399*0b57cec5SDimitry Andric // GLOBALVAR: [strtab offset, strtab size, type, isconst, initid, 1400*0b57cec5SDimitry Andric // linkage, alignment, section, visibility, threadlocal, 1401*0b57cec5SDimitry Andric // unnamed_addr, externally_initialized, dllstorageclass, 140281ad6265SDimitry Andric // comdat, attributes, DSO_Local, GlobalSanitizer] 1403*0b57cec5SDimitry Andric Vals.push_back(addToStrtab(GV.getName())); 1404*0b57cec5SDimitry Andric Vals.push_back(GV.getName().size()); 1405*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(GV.getValueType())); 1406*0b57cec5SDimitry Andric Vals.push_back(GV.getType()->getAddressSpace() << 2 | 2 | GV.isConstant()); 1407*0b57cec5SDimitry Andric Vals.push_back(GV.isDeclaration() ? 0 : 1408*0b57cec5SDimitry Andric (VE.getValueID(GV.getInitializer()) + 1)); 1409*0b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(GV)); 1410e8d8bef9SDimitry Andric Vals.push_back(getEncodedAlign(GV.getAlign())); 14115ffd83dbSDimitry Andric Vals.push_back(GV.hasSection() ? SectionMap[std::string(GV.getSection())] 14125ffd83dbSDimitry Andric : 0); 1413*0b57cec5SDimitry Andric if (GV.isThreadLocal() || 1414*0b57cec5SDimitry Andric GV.getVisibility() != GlobalValue::DefaultVisibility || 1415*0b57cec5SDimitry Andric GV.getUnnamedAddr() != GlobalValue::UnnamedAddr::None || 1416*0b57cec5SDimitry Andric GV.isExternallyInitialized() || 1417*0b57cec5SDimitry Andric GV.getDLLStorageClass() != GlobalValue::DefaultStorageClass || 141881ad6265SDimitry Andric GV.hasComdat() || GV.hasAttributes() || GV.isDSOLocal() || 141981ad6265SDimitry Andric GV.hasPartition() || GV.hasSanitizerMetadata()) { 1420*0b57cec5SDimitry Andric Vals.push_back(getEncodedVisibility(GV)); 1421*0b57cec5SDimitry Andric Vals.push_back(getEncodedThreadLocalMode(GV)); 1422*0b57cec5SDimitry Andric Vals.push_back(getEncodedUnnamedAddr(GV)); 1423*0b57cec5SDimitry Andric Vals.push_back(GV.isExternallyInitialized()); 1424*0b57cec5SDimitry Andric Vals.push_back(getEncodedDLLStorageClass(GV)); 1425*0b57cec5SDimitry Andric Vals.push_back(GV.hasComdat() ? VE.getComdatID(GV.getComdat()) : 0); 1426*0b57cec5SDimitry Andric 1427*0b57cec5SDimitry Andric auto AL = GV.getAttributesAsList(AttributeList::FunctionIndex); 1428*0b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(AL)); 1429*0b57cec5SDimitry Andric 1430*0b57cec5SDimitry Andric Vals.push_back(GV.isDSOLocal()); 1431*0b57cec5SDimitry Andric Vals.push_back(addToStrtab(GV.getPartition())); 1432*0b57cec5SDimitry Andric Vals.push_back(GV.getPartition().size()); 143381ad6265SDimitry Andric 143481ad6265SDimitry Andric Vals.push_back((GV.hasSanitizerMetadata() ? serializeSanitizerMetadata( 143581ad6265SDimitry Andric GV.getSanitizerMetadata()) 143681ad6265SDimitry Andric : 0)); 1437*0b57cec5SDimitry Andric } else { 1438*0b57cec5SDimitry Andric AbbrevToUse = SimpleGVarAbbrev; 1439*0b57cec5SDimitry Andric } 1440*0b57cec5SDimitry Andric 1441*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_GLOBALVAR, Vals, AbbrevToUse); 1442*0b57cec5SDimitry Andric Vals.clear(); 1443*0b57cec5SDimitry Andric } 1444*0b57cec5SDimitry Andric 1445*0b57cec5SDimitry Andric // Emit the function proto information. 1446*0b57cec5SDimitry Andric for (const Function &F : M) { 1447*0b57cec5SDimitry Andric // FUNCTION: [strtab offset, strtab size, type, callingconv, isproto, 1448*0b57cec5SDimitry Andric // linkage, paramattrs, alignment, section, visibility, gc, 1449*0b57cec5SDimitry Andric // unnamed_addr, prologuedata, dllstorageclass, comdat, 1450*0b57cec5SDimitry Andric // prefixdata, personalityfn, DSO_Local, addrspace] 1451*0b57cec5SDimitry Andric Vals.push_back(addToStrtab(F.getName())); 1452*0b57cec5SDimitry Andric Vals.push_back(F.getName().size()); 1453*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(F.getFunctionType())); 1454*0b57cec5SDimitry Andric Vals.push_back(F.getCallingConv()); 1455*0b57cec5SDimitry Andric Vals.push_back(F.isDeclaration()); 1456*0b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(F)); 1457*0b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(F.getAttributes())); 1458e8d8bef9SDimitry Andric Vals.push_back(getEncodedAlign(F.getAlign())); 14595ffd83dbSDimitry Andric Vals.push_back(F.hasSection() ? SectionMap[std::string(F.getSection())] 14605ffd83dbSDimitry Andric : 0); 1461*0b57cec5SDimitry Andric Vals.push_back(getEncodedVisibility(F)); 1462*0b57cec5SDimitry Andric Vals.push_back(F.hasGC() ? GCMap[F.getGC()] : 0); 1463*0b57cec5SDimitry Andric Vals.push_back(getEncodedUnnamedAddr(F)); 1464*0b57cec5SDimitry Andric Vals.push_back(F.hasPrologueData() ? (VE.getValueID(F.getPrologueData()) + 1) 1465*0b57cec5SDimitry Andric : 0); 1466*0b57cec5SDimitry Andric Vals.push_back(getEncodedDLLStorageClass(F)); 1467*0b57cec5SDimitry Andric Vals.push_back(F.hasComdat() ? VE.getComdatID(F.getComdat()) : 0); 1468*0b57cec5SDimitry Andric Vals.push_back(F.hasPrefixData() ? (VE.getValueID(F.getPrefixData()) + 1) 1469*0b57cec5SDimitry Andric : 0); 1470*0b57cec5SDimitry Andric Vals.push_back( 1471*0b57cec5SDimitry Andric F.hasPersonalityFn() ? (VE.getValueID(F.getPersonalityFn()) + 1) : 0); 1472*0b57cec5SDimitry Andric 1473*0b57cec5SDimitry Andric Vals.push_back(F.isDSOLocal()); 1474*0b57cec5SDimitry Andric Vals.push_back(F.getAddressSpace()); 1475*0b57cec5SDimitry Andric Vals.push_back(addToStrtab(F.getPartition())); 1476*0b57cec5SDimitry Andric Vals.push_back(F.getPartition().size()); 1477*0b57cec5SDimitry Andric 1478*0b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 1479*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_FUNCTION, Vals, AbbrevToUse); 1480*0b57cec5SDimitry Andric Vals.clear(); 1481*0b57cec5SDimitry Andric } 1482*0b57cec5SDimitry Andric 1483*0b57cec5SDimitry Andric // Emit the alias information. 1484*0b57cec5SDimitry Andric for (const GlobalAlias &A : M.aliases()) { 1485*0b57cec5SDimitry Andric // ALIAS: [strtab offset, strtab size, alias type, aliasee val#, linkage, 1486*0b57cec5SDimitry Andric // visibility, dllstorageclass, threadlocal, unnamed_addr, 1487*0b57cec5SDimitry Andric // DSO_Local] 1488*0b57cec5SDimitry Andric Vals.push_back(addToStrtab(A.getName())); 1489*0b57cec5SDimitry Andric Vals.push_back(A.getName().size()); 1490*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(A.getValueType())); 1491*0b57cec5SDimitry Andric Vals.push_back(A.getType()->getAddressSpace()); 1492*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(A.getAliasee())); 1493*0b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(A)); 1494*0b57cec5SDimitry Andric Vals.push_back(getEncodedVisibility(A)); 1495*0b57cec5SDimitry Andric Vals.push_back(getEncodedDLLStorageClass(A)); 1496*0b57cec5SDimitry Andric Vals.push_back(getEncodedThreadLocalMode(A)); 1497*0b57cec5SDimitry Andric Vals.push_back(getEncodedUnnamedAddr(A)); 1498*0b57cec5SDimitry Andric Vals.push_back(A.isDSOLocal()); 1499*0b57cec5SDimitry Andric Vals.push_back(addToStrtab(A.getPartition())); 1500*0b57cec5SDimitry Andric Vals.push_back(A.getPartition().size()); 1501*0b57cec5SDimitry Andric 1502*0b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 1503*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_ALIAS, Vals, AbbrevToUse); 1504*0b57cec5SDimitry Andric Vals.clear(); 1505*0b57cec5SDimitry Andric } 1506*0b57cec5SDimitry Andric 1507*0b57cec5SDimitry Andric // Emit the ifunc information. 1508*0b57cec5SDimitry Andric for (const GlobalIFunc &I : M.ifuncs()) { 1509*0b57cec5SDimitry Andric // IFUNC: [strtab offset, strtab size, ifunc type, address space, resolver 1510*0b57cec5SDimitry Andric // val#, linkage, visibility, DSO_Local] 1511*0b57cec5SDimitry Andric Vals.push_back(addToStrtab(I.getName())); 1512*0b57cec5SDimitry Andric Vals.push_back(I.getName().size()); 1513*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getValueType())); 1514*0b57cec5SDimitry Andric Vals.push_back(I.getType()->getAddressSpace()); 1515*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(I.getResolver())); 1516*0b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(I)); 1517*0b57cec5SDimitry Andric Vals.push_back(getEncodedVisibility(I)); 1518*0b57cec5SDimitry Andric Vals.push_back(I.isDSOLocal()); 1519*0b57cec5SDimitry Andric Vals.push_back(addToStrtab(I.getPartition())); 1520*0b57cec5SDimitry Andric Vals.push_back(I.getPartition().size()); 1521*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_IFUNC, Vals); 1522*0b57cec5SDimitry Andric Vals.clear(); 1523*0b57cec5SDimitry Andric } 1524*0b57cec5SDimitry Andric 1525*0b57cec5SDimitry Andric writeValueSymbolTableForwardDecl(); 1526*0b57cec5SDimitry Andric } 1527*0b57cec5SDimitry Andric 1528*0b57cec5SDimitry Andric static uint64_t getOptimizationFlags(const Value *V) { 1529*0b57cec5SDimitry Andric uint64_t Flags = 0; 1530*0b57cec5SDimitry Andric 1531*0b57cec5SDimitry Andric if (const auto *OBO = dyn_cast<OverflowingBinaryOperator>(V)) { 1532*0b57cec5SDimitry Andric if (OBO->hasNoSignedWrap()) 1533*0b57cec5SDimitry Andric Flags |= 1 << bitc::OBO_NO_SIGNED_WRAP; 1534*0b57cec5SDimitry Andric if (OBO->hasNoUnsignedWrap()) 1535*0b57cec5SDimitry Andric Flags |= 1 << bitc::OBO_NO_UNSIGNED_WRAP; 1536*0b57cec5SDimitry Andric } else if (const auto *PEO = dyn_cast<PossiblyExactOperator>(V)) { 1537*0b57cec5SDimitry Andric if (PEO->isExact()) 1538*0b57cec5SDimitry Andric Flags |= 1 << bitc::PEO_EXACT; 1539*0b57cec5SDimitry Andric } else if (const auto *FPMO = dyn_cast<FPMathOperator>(V)) { 1540*0b57cec5SDimitry Andric if (FPMO->hasAllowReassoc()) 1541*0b57cec5SDimitry Andric Flags |= bitc::AllowReassoc; 1542*0b57cec5SDimitry Andric if (FPMO->hasNoNaNs()) 1543*0b57cec5SDimitry Andric Flags |= bitc::NoNaNs; 1544*0b57cec5SDimitry Andric if (FPMO->hasNoInfs()) 1545*0b57cec5SDimitry Andric Flags |= bitc::NoInfs; 1546*0b57cec5SDimitry Andric if (FPMO->hasNoSignedZeros()) 1547*0b57cec5SDimitry Andric Flags |= bitc::NoSignedZeros; 1548*0b57cec5SDimitry Andric if (FPMO->hasAllowReciprocal()) 1549*0b57cec5SDimitry Andric Flags |= bitc::AllowReciprocal; 1550*0b57cec5SDimitry Andric if (FPMO->hasAllowContract()) 1551*0b57cec5SDimitry Andric Flags |= bitc::AllowContract; 1552*0b57cec5SDimitry Andric if (FPMO->hasApproxFunc()) 1553*0b57cec5SDimitry Andric Flags |= bitc::ApproxFunc; 1554*0b57cec5SDimitry Andric } 1555*0b57cec5SDimitry Andric 1556*0b57cec5SDimitry Andric return Flags; 1557*0b57cec5SDimitry Andric } 1558*0b57cec5SDimitry Andric 1559*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeValueAsMetadata( 1560*0b57cec5SDimitry Andric const ValueAsMetadata *MD, SmallVectorImpl<uint64_t> &Record) { 1561*0b57cec5SDimitry Andric // Mimic an MDNode with a value as one operand. 1562*0b57cec5SDimitry Andric Value *V = MD->getValue(); 1563*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(V->getType())); 1564*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(V)); 1565*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_VALUE, Record, 0); 1566*0b57cec5SDimitry Andric Record.clear(); 1567*0b57cec5SDimitry Andric } 1568*0b57cec5SDimitry Andric 1569*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeMDTuple(const MDTuple *N, 1570*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1571*0b57cec5SDimitry Andric unsigned Abbrev) { 1572*0b57cec5SDimitry Andric for (unsigned i = 0, e = N->getNumOperands(); i != e; ++i) { 1573*0b57cec5SDimitry Andric Metadata *MD = N->getOperand(i); 1574*0b57cec5SDimitry Andric assert(!(MD && isa<LocalAsMetadata>(MD)) && 1575*0b57cec5SDimitry Andric "Unexpected function-local metadata"); 1576*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(MD)); 1577*0b57cec5SDimitry Andric } 1578*0b57cec5SDimitry Andric Stream.EmitRecord(N->isDistinct() ? bitc::METADATA_DISTINCT_NODE 1579*0b57cec5SDimitry Andric : bitc::METADATA_NODE, 1580*0b57cec5SDimitry Andric Record, Abbrev); 1581*0b57cec5SDimitry Andric Record.clear(); 1582*0b57cec5SDimitry Andric } 1583*0b57cec5SDimitry Andric 1584*0b57cec5SDimitry Andric unsigned ModuleBitcodeWriter::createDILocationAbbrev() { 1585*0b57cec5SDimitry Andric // Assume the column is usually under 128, and always output the inlined-at 1586*0b57cec5SDimitry Andric // location (it's never more expensive than building an array size 1). 1587*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 1588*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_LOCATION)); 1589*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 1590*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 1591*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 1592*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 1593*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 1594*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 1595*0b57cec5SDimitry Andric return Stream.EmitAbbrev(std::move(Abbv)); 1596*0b57cec5SDimitry Andric } 1597*0b57cec5SDimitry Andric 1598*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILocation(const DILocation *N, 1599*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1600*0b57cec5SDimitry Andric unsigned &Abbrev) { 1601*0b57cec5SDimitry Andric if (!Abbrev) 1602*0b57cec5SDimitry Andric Abbrev = createDILocationAbbrev(); 1603*0b57cec5SDimitry Andric 1604*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1605*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 1606*0b57cec5SDimitry Andric Record.push_back(N->getColumn()); 1607*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataID(N->getScope())); 1608*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getInlinedAt())); 1609*0b57cec5SDimitry Andric Record.push_back(N->isImplicitCode()); 1610*0b57cec5SDimitry Andric 1611*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LOCATION, Record, Abbrev); 1612*0b57cec5SDimitry Andric Record.clear(); 1613*0b57cec5SDimitry Andric } 1614*0b57cec5SDimitry Andric 1615*0b57cec5SDimitry Andric unsigned ModuleBitcodeWriter::createGenericDINodeAbbrev() { 1616*0b57cec5SDimitry Andric // Assume the column is usually under 128, and always output the inlined-at 1617*0b57cec5SDimitry Andric // location (it's never more expensive than building an array size 1). 1618*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 1619*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_GENERIC_DEBUG)); 1620*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 1621*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 1622*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 1623*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 1624*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 1625*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 1626*0b57cec5SDimitry Andric return Stream.EmitAbbrev(std::move(Abbv)); 1627*0b57cec5SDimitry Andric } 1628*0b57cec5SDimitry Andric 1629*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeGenericDINode(const GenericDINode *N, 1630*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1631*0b57cec5SDimitry Andric unsigned &Abbrev) { 1632*0b57cec5SDimitry Andric if (!Abbrev) 1633*0b57cec5SDimitry Andric Abbrev = createGenericDINodeAbbrev(); 1634*0b57cec5SDimitry Andric 1635*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1636*0b57cec5SDimitry Andric Record.push_back(N->getTag()); 1637*0b57cec5SDimitry Andric Record.push_back(0); // Per-tag version field; unused for now. 1638*0b57cec5SDimitry Andric 1639*0b57cec5SDimitry Andric for (auto &I : N->operands()) 1640*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(I)); 1641*0b57cec5SDimitry Andric 1642*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_GENERIC_DEBUG, Record, Abbrev); 1643*0b57cec5SDimitry Andric Record.clear(); 1644*0b57cec5SDimitry Andric } 1645*0b57cec5SDimitry Andric 1646*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDISubrange(const DISubrange *N, 1647*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1648*0b57cec5SDimitry Andric unsigned Abbrev) { 16495ffd83dbSDimitry Andric const uint64_t Version = 2 << 1; 1650*0b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct() | Version); 1651*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawCountNode())); 16525ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawLowerBound())); 16535ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawUpperBound())); 16545ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawStride())); 1655*0b57cec5SDimitry Andric 1656*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_SUBRANGE, Record, Abbrev); 1657*0b57cec5SDimitry Andric Record.clear(); 1658*0b57cec5SDimitry Andric } 1659*0b57cec5SDimitry Andric 1660e8d8bef9SDimitry Andric void ModuleBitcodeWriter::writeDIGenericSubrange( 1661e8d8bef9SDimitry Andric const DIGenericSubrange *N, SmallVectorImpl<uint64_t> &Record, 1662e8d8bef9SDimitry Andric unsigned Abbrev) { 1663e8d8bef9SDimitry Andric Record.push_back((uint64_t)N->isDistinct()); 1664e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawCountNode())); 1665e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawLowerBound())); 1666e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawUpperBound())); 1667e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawStride())); 1668e8d8bef9SDimitry Andric 1669e8d8bef9SDimitry Andric Stream.EmitRecord(bitc::METADATA_GENERIC_SUBRANGE, Record, Abbrev); 1670e8d8bef9SDimitry Andric Record.clear(); 1671e8d8bef9SDimitry Andric } 1672e8d8bef9SDimitry Andric 16735ffd83dbSDimitry Andric static void emitSignedInt64(SmallVectorImpl<uint64_t> &Vals, uint64_t V) { 16745ffd83dbSDimitry Andric if ((int64_t)V >= 0) 16755ffd83dbSDimitry Andric Vals.push_back(V << 1); 16765ffd83dbSDimitry Andric else 16775ffd83dbSDimitry Andric Vals.push_back((-V << 1) | 1); 16785ffd83dbSDimitry Andric } 16795ffd83dbSDimitry Andric 16805ffd83dbSDimitry Andric static void emitWideAPInt(SmallVectorImpl<uint64_t> &Vals, const APInt &A) { 16815ffd83dbSDimitry Andric // We have an arbitrary precision integer value to write whose 16825ffd83dbSDimitry Andric // bit width is > 64. However, in canonical unsigned integer 16835ffd83dbSDimitry Andric // format it is likely that the high bits are going to be zero. 16845ffd83dbSDimitry Andric // So, we only write the number of active words. 16855ffd83dbSDimitry Andric unsigned NumWords = A.getActiveWords(); 16865ffd83dbSDimitry Andric const uint64_t *RawData = A.getRawData(); 16875ffd83dbSDimitry Andric for (unsigned i = 0; i < NumWords; i++) 16885ffd83dbSDimitry Andric emitSignedInt64(Vals, RawData[i]); 16895ffd83dbSDimitry Andric } 16905ffd83dbSDimitry Andric 1691*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIEnumerator(const DIEnumerator *N, 1692*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1693*0b57cec5SDimitry Andric unsigned Abbrev) { 16945ffd83dbSDimitry Andric const uint64_t IsBigInt = 1 << 2; 16955ffd83dbSDimitry Andric Record.push_back(IsBigInt | (N->isUnsigned() << 1) | N->isDistinct()); 16965ffd83dbSDimitry Andric Record.push_back(N->getValue().getBitWidth()); 1697*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 16985ffd83dbSDimitry Andric emitWideAPInt(Record, N->getValue()); 1699*0b57cec5SDimitry Andric 1700*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_ENUMERATOR, Record, Abbrev); 1701*0b57cec5SDimitry Andric Record.clear(); 1702*0b57cec5SDimitry Andric } 1703*0b57cec5SDimitry Andric 1704*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIBasicType(const DIBasicType *N, 1705*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1706*0b57cec5SDimitry Andric unsigned Abbrev) { 1707*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1708*0b57cec5SDimitry Andric Record.push_back(N->getTag()); 1709*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 1710*0b57cec5SDimitry Andric Record.push_back(N->getSizeInBits()); 1711*0b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 1712*0b57cec5SDimitry Andric Record.push_back(N->getEncoding()); 1713*0b57cec5SDimitry Andric Record.push_back(N->getFlags()); 1714*0b57cec5SDimitry Andric 1715*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_BASIC_TYPE, Record, Abbrev); 1716*0b57cec5SDimitry Andric Record.clear(); 1717*0b57cec5SDimitry Andric } 1718*0b57cec5SDimitry Andric 1719e8d8bef9SDimitry Andric void ModuleBitcodeWriter::writeDIStringType(const DIStringType *N, 1720e8d8bef9SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1721e8d8bef9SDimitry Andric unsigned Abbrev) { 1722e8d8bef9SDimitry Andric Record.push_back(N->isDistinct()); 1723e8d8bef9SDimitry Andric Record.push_back(N->getTag()); 1724e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 1725e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getStringLength())); 1726e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getStringLengthExp())); 172704eeddc0SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getStringLocationExp())); 1728e8d8bef9SDimitry Andric Record.push_back(N->getSizeInBits()); 1729e8d8bef9SDimitry Andric Record.push_back(N->getAlignInBits()); 1730e8d8bef9SDimitry Andric Record.push_back(N->getEncoding()); 1731e8d8bef9SDimitry Andric 1732e8d8bef9SDimitry Andric Stream.EmitRecord(bitc::METADATA_STRING_TYPE, Record, Abbrev); 1733e8d8bef9SDimitry Andric Record.clear(); 1734e8d8bef9SDimitry Andric } 1735e8d8bef9SDimitry Andric 1736*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIDerivedType(const DIDerivedType *N, 1737*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1738*0b57cec5SDimitry Andric unsigned Abbrev) { 1739*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1740*0b57cec5SDimitry Andric Record.push_back(N->getTag()); 1741*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 1742*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 1743*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 1744*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 1745*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getBaseType())); 1746*0b57cec5SDimitry Andric Record.push_back(N->getSizeInBits()); 1747*0b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 1748*0b57cec5SDimitry Andric Record.push_back(N->getOffsetInBits()); 1749*0b57cec5SDimitry Andric Record.push_back(N->getFlags()); 1750*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getExtraData())); 1751*0b57cec5SDimitry Andric 1752*0b57cec5SDimitry Andric // DWARF address space is encoded as N->getDWARFAddressSpace() + 1. 0 means 1753*0b57cec5SDimitry Andric // that there is no DWARF address space associated with DIDerivedType. 1754*0b57cec5SDimitry Andric if (const auto &DWARFAddressSpace = N->getDWARFAddressSpace()) 1755*0b57cec5SDimitry Andric Record.push_back(*DWARFAddressSpace + 1); 1756*0b57cec5SDimitry Andric else 1757*0b57cec5SDimitry Andric Record.push_back(0); 1758*0b57cec5SDimitry Andric 1759349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 1760349cc55cSDimitry Andric 1761*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_DERIVED_TYPE, Record, Abbrev); 1762*0b57cec5SDimitry Andric Record.clear(); 1763*0b57cec5SDimitry Andric } 1764*0b57cec5SDimitry Andric 1765*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDICompositeType( 1766*0b57cec5SDimitry Andric const DICompositeType *N, SmallVectorImpl<uint64_t> &Record, 1767*0b57cec5SDimitry Andric unsigned Abbrev) { 1768*0b57cec5SDimitry Andric const unsigned IsNotUsedInOldTypeRef = 0x2; 1769*0b57cec5SDimitry Andric Record.push_back(IsNotUsedInOldTypeRef | (unsigned)N->isDistinct()); 1770*0b57cec5SDimitry Andric Record.push_back(N->getTag()); 1771*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 1772*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 1773*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 1774*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 1775*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getBaseType())); 1776*0b57cec5SDimitry Andric Record.push_back(N->getSizeInBits()); 1777*0b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 1778*0b57cec5SDimitry Andric Record.push_back(N->getOffsetInBits()); 1779*0b57cec5SDimitry Andric Record.push_back(N->getFlags()); 1780*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getElements().get())); 1781*0b57cec5SDimitry Andric Record.push_back(N->getRuntimeLang()); 1782*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getVTableHolder())); 1783*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getTemplateParams().get())); 1784*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawIdentifier())); 1785*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getDiscriminator())); 17865ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawDataLocation())); 1787e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawAssociated())); 1788e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawAllocated())); 1789e8d8bef9SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawRank())); 1790349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 1791*0b57cec5SDimitry Andric 1792*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_COMPOSITE_TYPE, Record, Abbrev); 1793*0b57cec5SDimitry Andric Record.clear(); 1794*0b57cec5SDimitry Andric } 1795*0b57cec5SDimitry Andric 1796*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDISubroutineType( 1797*0b57cec5SDimitry Andric const DISubroutineType *N, SmallVectorImpl<uint64_t> &Record, 1798*0b57cec5SDimitry Andric unsigned Abbrev) { 1799*0b57cec5SDimitry Andric const unsigned HasNoOldTypeRefs = 0x2; 1800*0b57cec5SDimitry Andric Record.push_back(HasNoOldTypeRefs | (unsigned)N->isDistinct()); 1801*0b57cec5SDimitry Andric Record.push_back(N->getFlags()); 1802*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getTypeArray().get())); 1803*0b57cec5SDimitry Andric Record.push_back(N->getCC()); 1804*0b57cec5SDimitry Andric 1805*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_SUBROUTINE_TYPE, Record, Abbrev); 1806*0b57cec5SDimitry Andric Record.clear(); 1807*0b57cec5SDimitry Andric } 1808*0b57cec5SDimitry Andric 1809*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIFile(const DIFile *N, 1810*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1811*0b57cec5SDimitry Andric unsigned Abbrev) { 1812*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1813*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawFilename())); 1814*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawDirectory())); 1815*0b57cec5SDimitry Andric if (N->getRawChecksum()) { 1816*0b57cec5SDimitry Andric Record.push_back(N->getRawChecksum()->Kind); 1817*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawChecksum()->Value)); 1818*0b57cec5SDimitry Andric } else { 1819*0b57cec5SDimitry Andric // Maintain backwards compatibility with the old internal representation of 1820*0b57cec5SDimitry Andric // CSK_None in ChecksumKind by writing nulls here when Checksum is None. 1821*0b57cec5SDimitry Andric Record.push_back(0); 1822*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(nullptr)); 1823*0b57cec5SDimitry Andric } 1824*0b57cec5SDimitry Andric auto Source = N->getRawSource(); 1825*0b57cec5SDimitry Andric if (Source) 1826bdd1243dSDimitry Andric Record.push_back(VE.getMetadataOrNullID(Source)); 1827*0b57cec5SDimitry Andric 1828*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_FILE, Record, Abbrev); 1829*0b57cec5SDimitry Andric Record.clear(); 1830*0b57cec5SDimitry Andric } 1831*0b57cec5SDimitry Andric 1832*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDICompileUnit(const DICompileUnit *N, 1833*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1834*0b57cec5SDimitry Andric unsigned Abbrev) { 1835*0b57cec5SDimitry Andric assert(N->isDistinct() && "Expected distinct compile units"); 1836*0b57cec5SDimitry Andric Record.push_back(/* IsDistinct */ true); 1837*0b57cec5SDimitry Andric Record.push_back(N->getSourceLanguage()); 1838*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 1839*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawProducer())); 1840*0b57cec5SDimitry Andric Record.push_back(N->isOptimized()); 1841*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawFlags())); 1842*0b57cec5SDimitry Andric Record.push_back(N->getRuntimeVersion()); 1843*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawSplitDebugFilename())); 1844*0b57cec5SDimitry Andric Record.push_back(N->getEmissionKind()); 1845*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getEnumTypes().get())); 1846*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRetainedTypes().get())); 1847*0b57cec5SDimitry Andric Record.push_back(/* subprograms */ 0); 1848*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getGlobalVariables().get())); 1849*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getImportedEntities().get())); 1850*0b57cec5SDimitry Andric Record.push_back(N->getDWOId()); 1851*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getMacros().get())); 1852*0b57cec5SDimitry Andric Record.push_back(N->getSplitDebugInlining()); 1853*0b57cec5SDimitry Andric Record.push_back(N->getDebugInfoForProfiling()); 1854*0b57cec5SDimitry Andric Record.push_back((unsigned)N->getNameTableKind()); 18555ffd83dbSDimitry Andric Record.push_back(N->getRangesBaseAddress()); 18565ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawSysRoot())); 18575ffd83dbSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawSDK())); 1858*0b57cec5SDimitry Andric 1859*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_COMPILE_UNIT, Record, Abbrev); 1860*0b57cec5SDimitry Andric Record.clear(); 1861*0b57cec5SDimitry Andric } 1862*0b57cec5SDimitry Andric 1863*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDISubprogram(const DISubprogram *N, 1864*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1865*0b57cec5SDimitry Andric unsigned Abbrev) { 1866*0b57cec5SDimitry Andric const uint64_t HasUnitFlag = 1 << 1; 1867*0b57cec5SDimitry Andric const uint64_t HasSPFlagsFlag = 1 << 2; 1868*0b57cec5SDimitry Andric Record.push_back(uint64_t(N->isDistinct()) | HasUnitFlag | HasSPFlagsFlag); 1869*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 1870*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 1871*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawLinkageName())); 1872*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 1873*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 1874*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 1875*0b57cec5SDimitry Andric Record.push_back(N->getScopeLine()); 1876*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getContainingType())); 1877*0b57cec5SDimitry Andric Record.push_back(N->getSPFlags()); 1878*0b57cec5SDimitry Andric Record.push_back(N->getVirtualIndex()); 1879*0b57cec5SDimitry Andric Record.push_back(N->getFlags()); 1880*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawUnit())); 1881*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getTemplateParams().get())); 1882*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getDeclaration())); 1883*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRetainedNodes().get())); 1884*0b57cec5SDimitry Andric Record.push_back(N->getThisAdjustment()); 1885*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getThrownTypes().get())); 1886349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 188781ad6265SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawTargetFuncName())); 1888*0b57cec5SDimitry Andric 1889*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_SUBPROGRAM, Record, Abbrev); 1890*0b57cec5SDimitry Andric Record.clear(); 1891*0b57cec5SDimitry Andric } 1892*0b57cec5SDimitry Andric 1893*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILexicalBlock(const DILexicalBlock *N, 1894*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1895*0b57cec5SDimitry Andric unsigned Abbrev) { 1896*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1897*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 1898*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 1899*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 1900*0b57cec5SDimitry Andric Record.push_back(N->getColumn()); 1901*0b57cec5SDimitry Andric 1902*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LEXICAL_BLOCK, Record, Abbrev); 1903*0b57cec5SDimitry Andric Record.clear(); 1904*0b57cec5SDimitry Andric } 1905*0b57cec5SDimitry Andric 1906*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILexicalBlockFile( 1907*0b57cec5SDimitry Andric const DILexicalBlockFile *N, SmallVectorImpl<uint64_t> &Record, 1908*0b57cec5SDimitry Andric unsigned Abbrev) { 1909*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1910*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 1911*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 1912*0b57cec5SDimitry Andric Record.push_back(N->getDiscriminator()); 1913*0b57cec5SDimitry Andric 1914*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LEXICAL_BLOCK_FILE, Record, Abbrev); 1915*0b57cec5SDimitry Andric Record.clear(); 1916*0b57cec5SDimitry Andric } 1917*0b57cec5SDimitry Andric 1918*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDICommonBlock(const DICommonBlock *N, 1919*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1920*0b57cec5SDimitry Andric unsigned Abbrev) { 1921*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1922*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 1923*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getDecl())); 1924*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 1925*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 1926*0b57cec5SDimitry Andric Record.push_back(N->getLineNo()); 1927*0b57cec5SDimitry Andric 1928*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_COMMON_BLOCK, Record, Abbrev); 1929*0b57cec5SDimitry Andric Record.clear(); 1930*0b57cec5SDimitry Andric } 1931*0b57cec5SDimitry Andric 1932*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDINamespace(const DINamespace *N, 1933*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1934*0b57cec5SDimitry Andric unsigned Abbrev) { 1935*0b57cec5SDimitry Andric Record.push_back(N->isDistinct() | N->getExportSymbols() << 1); 1936*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 1937*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 1938*0b57cec5SDimitry Andric 1939*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_NAMESPACE, Record, Abbrev); 1940*0b57cec5SDimitry Andric Record.clear(); 1941*0b57cec5SDimitry Andric } 1942*0b57cec5SDimitry Andric 1943*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIMacro(const DIMacro *N, 1944*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1945*0b57cec5SDimitry Andric unsigned Abbrev) { 1946*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1947*0b57cec5SDimitry Andric Record.push_back(N->getMacinfoType()); 1948*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 1949*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 1950*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawValue())); 1951*0b57cec5SDimitry Andric 1952*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_MACRO, Record, Abbrev); 1953*0b57cec5SDimitry Andric Record.clear(); 1954*0b57cec5SDimitry Andric } 1955*0b57cec5SDimitry Andric 1956*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIMacroFile(const DIMacroFile *N, 1957*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1958*0b57cec5SDimitry Andric unsigned Abbrev) { 1959*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1960*0b57cec5SDimitry Andric Record.push_back(N->getMacinfoType()); 1961*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 1962*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 1963*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getElements().get())); 1964*0b57cec5SDimitry Andric 1965*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_MACRO_FILE, Record, Abbrev); 1966*0b57cec5SDimitry Andric Record.clear(); 1967*0b57cec5SDimitry Andric } 1968*0b57cec5SDimitry Andric 1969fe6060f1SDimitry Andric void ModuleBitcodeWriter::writeDIArgList(const DIArgList *N, 1970fe6060f1SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1971fe6060f1SDimitry Andric unsigned Abbrev) { 1972fe6060f1SDimitry Andric Record.reserve(N->getArgs().size()); 1973fe6060f1SDimitry Andric for (ValueAsMetadata *MD : N->getArgs()) 1974fe6060f1SDimitry Andric Record.push_back(VE.getMetadataID(MD)); 1975fe6060f1SDimitry Andric 1976fe6060f1SDimitry Andric Stream.EmitRecord(bitc::METADATA_ARG_LIST, Record, Abbrev); 1977fe6060f1SDimitry Andric Record.clear(); 1978fe6060f1SDimitry Andric } 1979fe6060f1SDimitry Andric 1980*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIModule(const DIModule *N, 1981*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 1982*0b57cec5SDimitry Andric unsigned Abbrev) { 1983*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 1984*0b57cec5SDimitry Andric for (auto &I : N->operands()) 1985*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(I)); 19865ffd83dbSDimitry Andric Record.push_back(N->getLineNo()); 1987e8d8bef9SDimitry Andric Record.push_back(N->getIsDecl()); 1988*0b57cec5SDimitry Andric 1989*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_MODULE, Record, Abbrev); 1990*0b57cec5SDimitry Andric Record.clear(); 1991*0b57cec5SDimitry Andric } 1992*0b57cec5SDimitry Andric 1993bdd1243dSDimitry Andric void ModuleBitcodeWriter::writeDIAssignID(const DIAssignID *N, 1994bdd1243dSDimitry Andric SmallVectorImpl<uint64_t> &Record, 1995bdd1243dSDimitry Andric unsigned Abbrev) { 1996bdd1243dSDimitry Andric // There are no arguments for this metadata type. 1997bdd1243dSDimitry Andric Record.push_back(N->isDistinct()); 1998bdd1243dSDimitry Andric Stream.EmitRecord(bitc::METADATA_ASSIGN_ID, Record, Abbrev); 1999bdd1243dSDimitry Andric Record.clear(); 2000bdd1243dSDimitry Andric } 2001bdd1243dSDimitry Andric 2002*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDITemplateTypeParameter( 2003*0b57cec5SDimitry Andric const DITemplateTypeParameter *N, SmallVectorImpl<uint64_t> &Record, 2004*0b57cec5SDimitry Andric unsigned Abbrev) { 2005*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 2006*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 2007*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 20085ffd83dbSDimitry Andric Record.push_back(N->isDefault()); 2009*0b57cec5SDimitry Andric 2010*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_TEMPLATE_TYPE, Record, Abbrev); 2011*0b57cec5SDimitry Andric Record.clear(); 2012*0b57cec5SDimitry Andric } 2013*0b57cec5SDimitry Andric 2014*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDITemplateValueParameter( 2015*0b57cec5SDimitry Andric const DITemplateValueParameter *N, SmallVectorImpl<uint64_t> &Record, 2016*0b57cec5SDimitry Andric unsigned Abbrev) { 2017*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 2018*0b57cec5SDimitry Andric Record.push_back(N->getTag()); 2019*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 2020*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 20215ffd83dbSDimitry Andric Record.push_back(N->isDefault()); 2022*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getValue())); 2023*0b57cec5SDimitry Andric 2024*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_TEMPLATE_VALUE, Record, Abbrev); 2025*0b57cec5SDimitry Andric Record.clear(); 2026*0b57cec5SDimitry Andric } 2027*0b57cec5SDimitry Andric 2028*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIGlobalVariable( 2029*0b57cec5SDimitry Andric const DIGlobalVariable *N, SmallVectorImpl<uint64_t> &Record, 2030*0b57cec5SDimitry Andric unsigned Abbrev) { 2031*0b57cec5SDimitry Andric const uint64_t Version = 2 << 1; 2032*0b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct() | Version); 2033*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 2034*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 2035*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawLinkageName())); 2036*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 2037*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 2038*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 2039*0b57cec5SDimitry Andric Record.push_back(N->isLocalToUnit()); 2040*0b57cec5SDimitry Andric Record.push_back(N->isDefinition()); 2041*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getStaticDataMemberDeclaration())); 2042*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getTemplateParams())); 2043*0b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 2044349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 2045*0b57cec5SDimitry Andric 2046*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_GLOBAL_VAR, Record, Abbrev); 2047*0b57cec5SDimitry Andric Record.clear(); 2048*0b57cec5SDimitry Andric } 2049*0b57cec5SDimitry Andric 2050*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILocalVariable( 2051*0b57cec5SDimitry Andric const DILocalVariable *N, SmallVectorImpl<uint64_t> &Record, 2052*0b57cec5SDimitry Andric unsigned Abbrev) { 2053*0b57cec5SDimitry Andric // In order to support all possible bitcode formats in BitcodeReader we need 2054*0b57cec5SDimitry Andric // to distinguish the following cases: 2055*0b57cec5SDimitry Andric // 1) Record has no artificial tag (Record[1]), 2056*0b57cec5SDimitry Andric // has no obsolete inlinedAt field (Record[9]). 2057*0b57cec5SDimitry Andric // In this case Record size will be 8, HasAlignment flag is false. 2058*0b57cec5SDimitry Andric // 2) Record has artificial tag (Record[1]), 2059*0b57cec5SDimitry Andric // has no obsolete inlignedAt field (Record[9]). 2060*0b57cec5SDimitry Andric // In this case Record size will be 9, HasAlignment flag is false. 2061*0b57cec5SDimitry Andric // 3) Record has both artificial tag (Record[1]) and 2062*0b57cec5SDimitry Andric // obsolete inlignedAt field (Record[9]). 2063*0b57cec5SDimitry Andric // In this case Record size will be 10, HasAlignment flag is false. 2064*0b57cec5SDimitry Andric // 4) Record has neither artificial tag, nor inlignedAt field, but 2065*0b57cec5SDimitry Andric // HasAlignment flag is true and Record[8] contains alignment value. 2066*0b57cec5SDimitry Andric const uint64_t HasAlignmentFlag = 1 << 1; 2067*0b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct() | HasAlignmentFlag); 2068*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 2069*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 2070*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 2071*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 2072*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 2073*0b57cec5SDimitry Andric Record.push_back(N->getArg()); 2074*0b57cec5SDimitry Andric Record.push_back(N->getFlags()); 2075*0b57cec5SDimitry Andric Record.push_back(N->getAlignInBits()); 2076349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getAnnotations().get())); 2077*0b57cec5SDimitry Andric 2078*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LOCAL_VAR, Record, Abbrev); 2079*0b57cec5SDimitry Andric Record.clear(); 2080*0b57cec5SDimitry Andric } 2081*0b57cec5SDimitry Andric 2082*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDILabel( 2083*0b57cec5SDimitry Andric const DILabel *N, SmallVectorImpl<uint64_t> &Record, 2084*0b57cec5SDimitry Andric unsigned Abbrev) { 2085*0b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct()); 2086*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 2087*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 2088*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 2089*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 2090*0b57cec5SDimitry Andric 2091*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_LABEL, Record, Abbrev); 2092*0b57cec5SDimitry Andric Record.clear(); 2093*0b57cec5SDimitry Andric } 2094*0b57cec5SDimitry Andric 2095*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIExpression(const DIExpression *N, 2096*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 2097*0b57cec5SDimitry Andric unsigned Abbrev) { 2098*0b57cec5SDimitry Andric Record.reserve(N->getElements().size() + 1); 2099*0b57cec5SDimitry Andric const uint64_t Version = 3 << 1; 2100*0b57cec5SDimitry Andric Record.push_back((uint64_t)N->isDistinct() | Version); 2101*0b57cec5SDimitry Andric Record.append(N->elements_begin(), N->elements_end()); 2102*0b57cec5SDimitry Andric 2103*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_EXPRESSION, Record, Abbrev); 2104*0b57cec5SDimitry Andric Record.clear(); 2105*0b57cec5SDimitry Andric } 2106*0b57cec5SDimitry Andric 2107*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIGlobalVariableExpression( 2108*0b57cec5SDimitry Andric const DIGlobalVariableExpression *N, SmallVectorImpl<uint64_t> &Record, 2109*0b57cec5SDimitry Andric unsigned Abbrev) { 2110*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 2111*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getVariable())); 2112*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getExpression())); 2113*0b57cec5SDimitry Andric 2114*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_GLOBAL_VAR_EXPR, Record, Abbrev); 2115*0b57cec5SDimitry Andric Record.clear(); 2116*0b57cec5SDimitry Andric } 2117*0b57cec5SDimitry Andric 2118*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIObjCProperty(const DIObjCProperty *N, 2119*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, 2120*0b57cec5SDimitry Andric unsigned Abbrev) { 2121*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 2122*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 2123*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getFile())); 2124*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 2125*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawSetterName())); 2126*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawGetterName())); 2127*0b57cec5SDimitry Andric Record.push_back(N->getAttributes()); 2128*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getType())); 2129*0b57cec5SDimitry Andric 2130*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_OBJC_PROPERTY, Record, Abbrev); 2131*0b57cec5SDimitry Andric Record.clear(); 2132*0b57cec5SDimitry Andric } 2133*0b57cec5SDimitry Andric 2134*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeDIImportedEntity( 2135*0b57cec5SDimitry Andric const DIImportedEntity *N, SmallVectorImpl<uint64_t> &Record, 2136*0b57cec5SDimitry Andric unsigned Abbrev) { 2137*0b57cec5SDimitry Andric Record.push_back(N->isDistinct()); 2138*0b57cec5SDimitry Andric Record.push_back(N->getTag()); 2139*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getScope())); 2140*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getEntity())); 2141*0b57cec5SDimitry Andric Record.push_back(N->getLine()); 2142*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawName())); 2143*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getRawFile())); 2144349cc55cSDimitry Andric Record.push_back(VE.getMetadataOrNullID(N->getElements().get())); 2145*0b57cec5SDimitry Andric 2146*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_IMPORTED_ENTITY, Record, Abbrev); 2147*0b57cec5SDimitry Andric Record.clear(); 2148*0b57cec5SDimitry Andric } 2149*0b57cec5SDimitry Andric 2150*0b57cec5SDimitry Andric unsigned ModuleBitcodeWriter::createNamedMetadataAbbrev() { 2151*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 2152*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_NAME)); 2153*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 2154*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 2155*0b57cec5SDimitry Andric return Stream.EmitAbbrev(std::move(Abbv)); 2156*0b57cec5SDimitry Andric } 2157*0b57cec5SDimitry Andric 2158*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeNamedMetadata( 2159*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record) { 2160*0b57cec5SDimitry Andric if (M.named_metadata_empty()) 2161*0b57cec5SDimitry Andric return; 2162*0b57cec5SDimitry Andric 2163*0b57cec5SDimitry Andric unsigned Abbrev = createNamedMetadataAbbrev(); 2164*0b57cec5SDimitry Andric for (const NamedMDNode &NMD : M.named_metadata()) { 2165*0b57cec5SDimitry Andric // Write name. 2166*0b57cec5SDimitry Andric StringRef Str = NMD.getName(); 2167*0b57cec5SDimitry Andric Record.append(Str.bytes_begin(), Str.bytes_end()); 2168*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_NAME, Record, Abbrev); 2169*0b57cec5SDimitry Andric Record.clear(); 2170*0b57cec5SDimitry Andric 2171*0b57cec5SDimitry Andric // Write named metadata operands. 2172*0b57cec5SDimitry Andric for (const MDNode *N : NMD.operands()) 2173*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataID(N)); 2174*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_NAMED_NODE, Record, 0); 2175*0b57cec5SDimitry Andric Record.clear(); 2176*0b57cec5SDimitry Andric } 2177*0b57cec5SDimitry Andric } 2178*0b57cec5SDimitry Andric 2179*0b57cec5SDimitry Andric unsigned ModuleBitcodeWriter::createMetadataStringsAbbrev() { 2180*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 2181*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_STRINGS)); 2182*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // # of strings 2183*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // offset to chars 2184*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); 2185*0b57cec5SDimitry Andric return Stream.EmitAbbrev(std::move(Abbv)); 2186*0b57cec5SDimitry Andric } 2187*0b57cec5SDimitry Andric 2188*0b57cec5SDimitry Andric /// Write out a record for MDString. 2189*0b57cec5SDimitry Andric /// 2190*0b57cec5SDimitry Andric /// All the metadata strings in a metadata block are emitted in a single 2191*0b57cec5SDimitry Andric /// record. The sizes and strings themselves are shoved into a blob. 2192*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeMetadataStrings( 2193*0b57cec5SDimitry Andric ArrayRef<const Metadata *> Strings, SmallVectorImpl<uint64_t> &Record) { 2194*0b57cec5SDimitry Andric if (Strings.empty()) 2195*0b57cec5SDimitry Andric return; 2196*0b57cec5SDimitry Andric 2197*0b57cec5SDimitry Andric // Start the record with the number of strings. 2198*0b57cec5SDimitry Andric Record.push_back(bitc::METADATA_STRINGS); 2199*0b57cec5SDimitry Andric Record.push_back(Strings.size()); 2200*0b57cec5SDimitry Andric 2201*0b57cec5SDimitry Andric // Emit the sizes of the strings in the blob. 2202*0b57cec5SDimitry Andric SmallString<256> Blob; 2203*0b57cec5SDimitry Andric { 2204*0b57cec5SDimitry Andric BitstreamWriter W(Blob); 2205*0b57cec5SDimitry Andric for (const Metadata *MD : Strings) 2206*0b57cec5SDimitry Andric W.EmitVBR(cast<MDString>(MD)->getLength(), 6); 2207*0b57cec5SDimitry Andric W.FlushToWord(); 2208*0b57cec5SDimitry Andric } 2209*0b57cec5SDimitry Andric 2210*0b57cec5SDimitry Andric // Add the offset to the strings to the record. 2211*0b57cec5SDimitry Andric Record.push_back(Blob.size()); 2212*0b57cec5SDimitry Andric 2213*0b57cec5SDimitry Andric // Add the strings to the blob. 2214*0b57cec5SDimitry Andric for (const Metadata *MD : Strings) 2215*0b57cec5SDimitry Andric Blob.append(cast<MDString>(MD)->getString()); 2216*0b57cec5SDimitry Andric 2217*0b57cec5SDimitry Andric // Emit the final record. 2218*0b57cec5SDimitry Andric Stream.EmitRecordWithBlob(createMetadataStringsAbbrev(), Record, Blob); 2219*0b57cec5SDimitry Andric Record.clear(); 2220*0b57cec5SDimitry Andric } 2221*0b57cec5SDimitry Andric 2222*0b57cec5SDimitry Andric // Generates an enum to use as an index in the Abbrev array of Metadata record. 2223*0b57cec5SDimitry Andric enum MetadataAbbrev : unsigned { 2224*0b57cec5SDimitry Andric #define HANDLE_MDNODE_LEAF(CLASS) CLASS##AbbrevID, 2225*0b57cec5SDimitry Andric #include "llvm/IR/Metadata.def" 2226*0b57cec5SDimitry Andric LastPlusOne 2227*0b57cec5SDimitry Andric }; 2228*0b57cec5SDimitry Andric 2229*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeMetadataRecords( 2230*0b57cec5SDimitry Andric ArrayRef<const Metadata *> MDs, SmallVectorImpl<uint64_t> &Record, 2231*0b57cec5SDimitry Andric std::vector<unsigned> *MDAbbrevs, std::vector<uint64_t> *IndexPos) { 2232*0b57cec5SDimitry Andric if (MDs.empty()) 2233*0b57cec5SDimitry Andric return; 2234*0b57cec5SDimitry Andric 2235*0b57cec5SDimitry Andric // Initialize MDNode abbreviations. 2236*0b57cec5SDimitry Andric #define HANDLE_MDNODE_LEAF(CLASS) unsigned CLASS##Abbrev = 0; 2237*0b57cec5SDimitry Andric #include "llvm/IR/Metadata.def" 2238*0b57cec5SDimitry Andric 2239*0b57cec5SDimitry Andric for (const Metadata *MD : MDs) { 2240*0b57cec5SDimitry Andric if (IndexPos) 2241*0b57cec5SDimitry Andric IndexPos->push_back(Stream.GetCurrentBitNo()); 2242*0b57cec5SDimitry Andric if (const MDNode *N = dyn_cast<MDNode>(MD)) { 2243*0b57cec5SDimitry Andric assert(N->isResolved() && "Expected forward references to be resolved"); 2244*0b57cec5SDimitry Andric 2245*0b57cec5SDimitry Andric switch (N->getMetadataID()) { 2246*0b57cec5SDimitry Andric default: 2247*0b57cec5SDimitry Andric llvm_unreachable("Invalid MDNode subclass"); 2248*0b57cec5SDimitry Andric #define HANDLE_MDNODE_LEAF(CLASS) \ 2249*0b57cec5SDimitry Andric case Metadata::CLASS##Kind: \ 2250*0b57cec5SDimitry Andric if (MDAbbrevs) \ 2251*0b57cec5SDimitry Andric write##CLASS(cast<CLASS>(N), Record, \ 2252*0b57cec5SDimitry Andric (*MDAbbrevs)[MetadataAbbrev::CLASS##AbbrevID]); \ 2253*0b57cec5SDimitry Andric else \ 2254*0b57cec5SDimitry Andric write##CLASS(cast<CLASS>(N), Record, CLASS##Abbrev); \ 2255*0b57cec5SDimitry Andric continue; 2256*0b57cec5SDimitry Andric #include "llvm/IR/Metadata.def" 2257*0b57cec5SDimitry Andric } 2258*0b57cec5SDimitry Andric } 2259*0b57cec5SDimitry Andric writeValueAsMetadata(cast<ValueAsMetadata>(MD), Record); 2260*0b57cec5SDimitry Andric } 2261*0b57cec5SDimitry Andric } 2262*0b57cec5SDimitry Andric 2263*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleMetadata() { 2264*0b57cec5SDimitry Andric if (!VE.hasMDs() && M.named_metadata_empty()) 2265*0b57cec5SDimitry Andric return; 2266*0b57cec5SDimitry Andric 2267*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 4); 2268*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 2269*0b57cec5SDimitry Andric 2270*0b57cec5SDimitry Andric // Emit all abbrevs upfront, so that the reader can jump in the middle of the 2271*0b57cec5SDimitry Andric // block and load any metadata. 2272*0b57cec5SDimitry Andric std::vector<unsigned> MDAbbrevs; 2273*0b57cec5SDimitry Andric 2274*0b57cec5SDimitry Andric MDAbbrevs.resize(MetadataAbbrev::LastPlusOne); 2275*0b57cec5SDimitry Andric MDAbbrevs[MetadataAbbrev::DILocationAbbrevID] = createDILocationAbbrev(); 2276*0b57cec5SDimitry Andric MDAbbrevs[MetadataAbbrev::GenericDINodeAbbrevID] = 2277*0b57cec5SDimitry Andric createGenericDINodeAbbrev(); 2278*0b57cec5SDimitry Andric 2279*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 2280*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_INDEX_OFFSET)); 2281*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 2282*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 2283*0b57cec5SDimitry Andric unsigned OffsetAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 2284*0b57cec5SDimitry Andric 2285*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 2286*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::METADATA_INDEX)); 2287*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 2288*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 2289*0b57cec5SDimitry Andric unsigned IndexAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 2290*0b57cec5SDimitry Andric 2291*0b57cec5SDimitry Andric // Emit MDStrings together upfront. 2292*0b57cec5SDimitry Andric writeMetadataStrings(VE.getMDStrings(), Record); 2293*0b57cec5SDimitry Andric 2294*0b57cec5SDimitry Andric // We only emit an index for the metadata record if we have more than a given 2295*0b57cec5SDimitry Andric // (naive) threshold of metadatas, otherwise it is not worth it. 2296*0b57cec5SDimitry Andric if (VE.getNonMDStrings().size() > IndexThreshold) { 2297*0b57cec5SDimitry Andric // Write a placeholder value in for the offset of the metadata index, 2298*0b57cec5SDimitry Andric // which is written after the records, so that it can include 2299*0b57cec5SDimitry Andric // the offset of each entry. The placeholder offset will be 2300*0b57cec5SDimitry Andric // updated after all records are emitted. 2301*0b57cec5SDimitry Andric uint64_t Vals[] = {0, 0}; 2302*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_INDEX_OFFSET, Vals, OffsetAbbrev); 2303*0b57cec5SDimitry Andric } 2304*0b57cec5SDimitry Andric 2305*0b57cec5SDimitry Andric // Compute and save the bit offset to the current position, which will be 2306*0b57cec5SDimitry Andric // patched when we emit the index later. We can simply subtract the 64-bit 2307*0b57cec5SDimitry Andric // fixed size from the current bit number to get the location to backpatch. 2308*0b57cec5SDimitry Andric uint64_t IndexOffsetRecordBitPos = Stream.GetCurrentBitNo(); 2309*0b57cec5SDimitry Andric 2310*0b57cec5SDimitry Andric // This index will contain the bitpos for each individual record. 2311*0b57cec5SDimitry Andric std::vector<uint64_t> IndexPos; 2312*0b57cec5SDimitry Andric IndexPos.reserve(VE.getNonMDStrings().size()); 2313*0b57cec5SDimitry Andric 2314*0b57cec5SDimitry Andric // Write all the records 2315*0b57cec5SDimitry Andric writeMetadataRecords(VE.getNonMDStrings(), Record, &MDAbbrevs, &IndexPos); 2316*0b57cec5SDimitry Andric 2317*0b57cec5SDimitry Andric if (VE.getNonMDStrings().size() > IndexThreshold) { 2318*0b57cec5SDimitry Andric // Now that we have emitted all the records we will emit the index. But 2319*0b57cec5SDimitry Andric // first 2320*0b57cec5SDimitry Andric // backpatch the forward reference so that the reader can skip the records 2321*0b57cec5SDimitry Andric // efficiently. 2322*0b57cec5SDimitry Andric Stream.BackpatchWord64(IndexOffsetRecordBitPos - 64, 2323*0b57cec5SDimitry Andric Stream.GetCurrentBitNo() - IndexOffsetRecordBitPos); 2324*0b57cec5SDimitry Andric 2325*0b57cec5SDimitry Andric // Delta encode the index. 2326*0b57cec5SDimitry Andric uint64_t PreviousValue = IndexOffsetRecordBitPos; 2327*0b57cec5SDimitry Andric for (auto &Elt : IndexPos) { 2328*0b57cec5SDimitry Andric auto EltDelta = Elt - PreviousValue; 2329*0b57cec5SDimitry Andric PreviousValue = Elt; 2330*0b57cec5SDimitry Andric Elt = EltDelta; 2331*0b57cec5SDimitry Andric } 2332*0b57cec5SDimitry Andric // Emit the index record. 2333*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_INDEX, IndexPos, IndexAbbrev); 2334*0b57cec5SDimitry Andric IndexPos.clear(); 2335*0b57cec5SDimitry Andric } 2336*0b57cec5SDimitry Andric 2337*0b57cec5SDimitry Andric // Write the named metadata now. 2338*0b57cec5SDimitry Andric writeNamedMetadata(Record); 2339*0b57cec5SDimitry Andric 2340*0b57cec5SDimitry Andric auto AddDeclAttachedMetadata = [&](const GlobalObject &GO) { 2341*0b57cec5SDimitry Andric SmallVector<uint64_t, 4> Record; 2342*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(&GO)); 2343*0b57cec5SDimitry Andric pushGlobalMetadataAttachment(Record, GO); 2344*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_GLOBAL_DECL_ATTACHMENT, Record); 2345*0b57cec5SDimitry Andric }; 2346*0b57cec5SDimitry Andric for (const Function &F : M) 2347*0b57cec5SDimitry Andric if (F.isDeclaration() && F.hasMetadata()) 2348*0b57cec5SDimitry Andric AddDeclAttachedMetadata(F); 2349*0b57cec5SDimitry Andric // FIXME: Only store metadata for declarations here, and move data for global 2350*0b57cec5SDimitry Andric // variable definitions to a separate block (PR28134). 2351*0b57cec5SDimitry Andric for (const GlobalVariable &GV : M.globals()) 2352*0b57cec5SDimitry Andric if (GV.hasMetadata()) 2353*0b57cec5SDimitry Andric AddDeclAttachedMetadata(GV); 2354*0b57cec5SDimitry Andric 2355*0b57cec5SDimitry Andric Stream.ExitBlock(); 2356*0b57cec5SDimitry Andric } 2357*0b57cec5SDimitry Andric 2358*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeFunctionMetadata(const Function &F) { 2359*0b57cec5SDimitry Andric if (!VE.hasMDs()) 2360*0b57cec5SDimitry Andric return; 2361*0b57cec5SDimitry Andric 2362*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::METADATA_BLOCK_ID, 3); 2363*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 2364*0b57cec5SDimitry Andric writeMetadataStrings(VE.getMDStrings(), Record); 2365*0b57cec5SDimitry Andric writeMetadataRecords(VE.getNonMDStrings(), Record); 2366*0b57cec5SDimitry Andric Stream.ExitBlock(); 2367*0b57cec5SDimitry Andric } 2368*0b57cec5SDimitry Andric 2369*0b57cec5SDimitry Andric void ModuleBitcodeWriter::pushGlobalMetadataAttachment( 2370*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Record, const GlobalObject &GO) { 2371*0b57cec5SDimitry Andric // [n x [id, mdnode]] 2372*0b57cec5SDimitry Andric SmallVector<std::pair<unsigned, MDNode *>, 4> MDs; 2373*0b57cec5SDimitry Andric GO.getAllMetadata(MDs); 2374*0b57cec5SDimitry Andric for (const auto &I : MDs) { 2375*0b57cec5SDimitry Andric Record.push_back(I.first); 2376*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataID(I.second)); 2377*0b57cec5SDimitry Andric } 2378*0b57cec5SDimitry Andric } 2379*0b57cec5SDimitry Andric 2380*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeFunctionMetadataAttachment(const Function &F) { 2381*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::METADATA_ATTACHMENT_ID, 3); 2382*0b57cec5SDimitry Andric 2383*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 2384*0b57cec5SDimitry Andric 2385*0b57cec5SDimitry Andric if (F.hasMetadata()) { 2386*0b57cec5SDimitry Andric pushGlobalMetadataAttachment(Record, F); 2387*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_ATTACHMENT, Record, 0); 2388*0b57cec5SDimitry Andric Record.clear(); 2389*0b57cec5SDimitry Andric } 2390*0b57cec5SDimitry Andric 2391*0b57cec5SDimitry Andric // Write metadata attachments 2392*0b57cec5SDimitry Andric // METADATA_ATTACHMENT - [m x [value, [n x [id, mdnode]]] 2393*0b57cec5SDimitry Andric SmallVector<std::pair<unsigned, MDNode *>, 4> MDs; 2394*0b57cec5SDimitry Andric for (const BasicBlock &BB : F) 2395*0b57cec5SDimitry Andric for (const Instruction &I : BB) { 2396*0b57cec5SDimitry Andric MDs.clear(); 2397*0b57cec5SDimitry Andric I.getAllMetadataOtherThanDebugLoc(MDs); 2398*0b57cec5SDimitry Andric 2399*0b57cec5SDimitry Andric // If no metadata, ignore instruction. 2400*0b57cec5SDimitry Andric if (MDs.empty()) continue; 2401*0b57cec5SDimitry Andric 2402*0b57cec5SDimitry Andric Record.push_back(VE.getInstructionID(&I)); 2403*0b57cec5SDimitry Andric 2404*0b57cec5SDimitry Andric for (unsigned i = 0, e = MDs.size(); i != e; ++i) { 2405*0b57cec5SDimitry Andric Record.push_back(MDs[i].first); 2406*0b57cec5SDimitry Andric Record.push_back(VE.getMetadataID(MDs[i].second)); 2407*0b57cec5SDimitry Andric } 2408*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_ATTACHMENT, Record, 0); 2409*0b57cec5SDimitry Andric Record.clear(); 2410*0b57cec5SDimitry Andric } 2411*0b57cec5SDimitry Andric 2412*0b57cec5SDimitry Andric Stream.ExitBlock(); 2413*0b57cec5SDimitry Andric } 2414*0b57cec5SDimitry Andric 2415*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleMetadataKinds() { 2416*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 2417*0b57cec5SDimitry Andric 2418*0b57cec5SDimitry Andric // Write metadata kinds 2419*0b57cec5SDimitry Andric // METADATA_KIND - [n x [id, name]] 2420*0b57cec5SDimitry Andric SmallVector<StringRef, 8> Names; 2421*0b57cec5SDimitry Andric M.getMDKindNames(Names); 2422*0b57cec5SDimitry Andric 2423*0b57cec5SDimitry Andric if (Names.empty()) return; 2424*0b57cec5SDimitry Andric 2425*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::METADATA_KIND_BLOCK_ID, 3); 2426*0b57cec5SDimitry Andric 2427*0b57cec5SDimitry Andric for (unsigned MDKindID = 0, e = Names.size(); MDKindID != e; ++MDKindID) { 2428*0b57cec5SDimitry Andric Record.push_back(MDKindID); 2429*0b57cec5SDimitry Andric StringRef KName = Names[MDKindID]; 2430*0b57cec5SDimitry Andric Record.append(KName.begin(), KName.end()); 2431*0b57cec5SDimitry Andric 2432*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::METADATA_KIND, Record, 0); 2433*0b57cec5SDimitry Andric Record.clear(); 2434*0b57cec5SDimitry Andric } 2435*0b57cec5SDimitry Andric 2436*0b57cec5SDimitry Andric Stream.ExitBlock(); 2437*0b57cec5SDimitry Andric } 2438*0b57cec5SDimitry Andric 2439*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeOperandBundleTags() { 2440*0b57cec5SDimitry Andric // Write metadata kinds 2441*0b57cec5SDimitry Andric // 2442*0b57cec5SDimitry Andric // OPERAND_BUNDLE_TAGS_BLOCK_ID : N x OPERAND_BUNDLE_TAG 2443*0b57cec5SDimitry Andric // 2444*0b57cec5SDimitry Andric // OPERAND_BUNDLE_TAG - [strchr x N] 2445*0b57cec5SDimitry Andric 2446*0b57cec5SDimitry Andric SmallVector<StringRef, 8> Tags; 2447*0b57cec5SDimitry Andric M.getOperandBundleTags(Tags); 2448*0b57cec5SDimitry Andric 2449*0b57cec5SDimitry Andric if (Tags.empty()) 2450*0b57cec5SDimitry Andric return; 2451*0b57cec5SDimitry Andric 2452*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::OPERAND_BUNDLE_TAGS_BLOCK_ID, 3); 2453*0b57cec5SDimitry Andric 2454*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 2455*0b57cec5SDimitry Andric 2456*0b57cec5SDimitry Andric for (auto Tag : Tags) { 2457*0b57cec5SDimitry Andric Record.append(Tag.begin(), Tag.end()); 2458*0b57cec5SDimitry Andric 2459*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::OPERAND_BUNDLE_TAG, Record, 0); 2460*0b57cec5SDimitry Andric Record.clear(); 2461*0b57cec5SDimitry Andric } 2462*0b57cec5SDimitry Andric 2463*0b57cec5SDimitry Andric Stream.ExitBlock(); 2464*0b57cec5SDimitry Andric } 2465*0b57cec5SDimitry Andric 2466*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeSyncScopeNames() { 2467*0b57cec5SDimitry Andric SmallVector<StringRef, 8> SSNs; 2468*0b57cec5SDimitry Andric M.getContext().getSyncScopeNames(SSNs); 2469*0b57cec5SDimitry Andric if (SSNs.empty()) 2470*0b57cec5SDimitry Andric return; 2471*0b57cec5SDimitry Andric 2472*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::SYNC_SCOPE_NAMES_BLOCK_ID, 2); 2473*0b57cec5SDimitry Andric 2474*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 2475*0b57cec5SDimitry Andric for (auto SSN : SSNs) { 2476*0b57cec5SDimitry Andric Record.append(SSN.begin(), SSN.end()); 2477*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::SYNC_SCOPE_NAME, Record, 0); 2478*0b57cec5SDimitry Andric Record.clear(); 2479*0b57cec5SDimitry Andric } 2480*0b57cec5SDimitry Andric 2481*0b57cec5SDimitry Andric Stream.ExitBlock(); 2482*0b57cec5SDimitry Andric } 2483*0b57cec5SDimitry Andric 2484*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeConstants(unsigned FirstVal, unsigned LastVal, 2485*0b57cec5SDimitry Andric bool isGlobal) { 2486*0b57cec5SDimitry Andric if (FirstVal == LastVal) return; 2487*0b57cec5SDimitry Andric 2488*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::CONSTANTS_BLOCK_ID, 4); 2489*0b57cec5SDimitry Andric 2490*0b57cec5SDimitry Andric unsigned AggregateAbbrev = 0; 2491*0b57cec5SDimitry Andric unsigned String8Abbrev = 0; 2492*0b57cec5SDimitry Andric unsigned CString7Abbrev = 0; 2493*0b57cec5SDimitry Andric unsigned CString6Abbrev = 0; 2494*0b57cec5SDimitry Andric // If this is a constant pool for the module, emit module-specific abbrevs. 2495*0b57cec5SDimitry Andric if (isGlobal) { 2496*0b57cec5SDimitry Andric // Abbrev for CST_CODE_AGGREGATE. 2497*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 2498*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_AGGREGATE)); 2499*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 2500*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, Log2_32_Ceil(LastVal+1))); 2501*0b57cec5SDimitry Andric AggregateAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 2502*0b57cec5SDimitry Andric 2503*0b57cec5SDimitry Andric // Abbrev for CST_CODE_STRING. 2504*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 2505*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_STRING)); 2506*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 2507*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 2508*0b57cec5SDimitry Andric String8Abbrev = Stream.EmitAbbrev(std::move(Abbv)); 2509*0b57cec5SDimitry Andric // Abbrev for CST_CODE_CSTRING. 2510*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 2511*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CSTRING)); 2512*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 2513*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 2514*0b57cec5SDimitry Andric CString7Abbrev = Stream.EmitAbbrev(std::move(Abbv)); 2515*0b57cec5SDimitry Andric // Abbrev for CST_CODE_CSTRING. 2516*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 2517*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CSTRING)); 2518*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 2519*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 2520*0b57cec5SDimitry Andric CString6Abbrev = Stream.EmitAbbrev(std::move(Abbv)); 2521*0b57cec5SDimitry Andric } 2522*0b57cec5SDimitry Andric 2523*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 2524*0b57cec5SDimitry Andric 2525*0b57cec5SDimitry Andric const ValueEnumerator::ValueList &Vals = VE.getValues(); 2526*0b57cec5SDimitry Andric Type *LastTy = nullptr; 2527*0b57cec5SDimitry Andric for (unsigned i = FirstVal; i != LastVal; ++i) { 2528*0b57cec5SDimitry Andric const Value *V = Vals[i].first; 2529*0b57cec5SDimitry Andric // If we need to switch types, do so now. 2530*0b57cec5SDimitry Andric if (V->getType() != LastTy) { 2531*0b57cec5SDimitry Andric LastTy = V->getType(); 2532*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(LastTy)); 2533*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::CST_CODE_SETTYPE, Record, 2534*0b57cec5SDimitry Andric CONSTANTS_SETTYPE_ABBREV); 2535*0b57cec5SDimitry Andric Record.clear(); 2536*0b57cec5SDimitry Andric } 2537*0b57cec5SDimitry Andric 2538*0b57cec5SDimitry Andric if (const InlineAsm *IA = dyn_cast<InlineAsm>(V)) { 253904eeddc0SDimitry Andric Record.push_back(VE.getTypeID(IA->getFunctionType())); 2540fe6060f1SDimitry Andric Record.push_back( 2541fe6060f1SDimitry Andric unsigned(IA->hasSideEffects()) | unsigned(IA->isAlignStack()) << 1 | 2542fe6060f1SDimitry Andric unsigned(IA->getDialect() & 1) << 2 | unsigned(IA->canThrow()) << 3); 2543*0b57cec5SDimitry Andric 2544*0b57cec5SDimitry Andric // Add the asm string. 2545*0b57cec5SDimitry Andric const std::string &AsmStr = IA->getAsmString(); 2546*0b57cec5SDimitry Andric Record.push_back(AsmStr.size()); 2547*0b57cec5SDimitry Andric Record.append(AsmStr.begin(), AsmStr.end()); 2548*0b57cec5SDimitry Andric 2549*0b57cec5SDimitry Andric // Add the constraint string. 2550*0b57cec5SDimitry Andric const std::string &ConstraintStr = IA->getConstraintString(); 2551*0b57cec5SDimitry Andric Record.push_back(ConstraintStr.size()); 2552*0b57cec5SDimitry Andric Record.append(ConstraintStr.begin(), ConstraintStr.end()); 2553*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::CST_CODE_INLINEASM, Record); 2554*0b57cec5SDimitry Andric Record.clear(); 2555*0b57cec5SDimitry Andric continue; 2556*0b57cec5SDimitry Andric } 2557*0b57cec5SDimitry Andric const Constant *C = cast<Constant>(V); 2558*0b57cec5SDimitry Andric unsigned Code = -1U; 2559*0b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 2560*0b57cec5SDimitry Andric if (C->isNullValue()) { 2561*0b57cec5SDimitry Andric Code = bitc::CST_CODE_NULL; 2562e8d8bef9SDimitry Andric } else if (isa<PoisonValue>(C)) { 2563e8d8bef9SDimitry Andric Code = bitc::CST_CODE_POISON; 2564*0b57cec5SDimitry Andric } else if (isa<UndefValue>(C)) { 2565*0b57cec5SDimitry Andric Code = bitc::CST_CODE_UNDEF; 2566*0b57cec5SDimitry Andric } else if (const ConstantInt *IV = dyn_cast<ConstantInt>(C)) { 2567*0b57cec5SDimitry Andric if (IV->getBitWidth() <= 64) { 2568*0b57cec5SDimitry Andric uint64_t V = IV->getSExtValue(); 2569*0b57cec5SDimitry Andric emitSignedInt64(Record, V); 2570*0b57cec5SDimitry Andric Code = bitc::CST_CODE_INTEGER; 2571*0b57cec5SDimitry Andric AbbrevToUse = CONSTANTS_INTEGER_ABBREV; 2572*0b57cec5SDimitry Andric } else { // Wide integers, > 64 bits in size. 25735ffd83dbSDimitry Andric emitWideAPInt(Record, IV->getValue()); 2574*0b57cec5SDimitry Andric Code = bitc::CST_CODE_WIDE_INTEGER; 2575*0b57cec5SDimitry Andric } 2576*0b57cec5SDimitry Andric } else if (const ConstantFP *CFP = dyn_cast<ConstantFP>(C)) { 2577*0b57cec5SDimitry Andric Code = bitc::CST_CODE_FLOAT; 2578*0b57cec5SDimitry Andric Type *Ty = CFP->getType(); 25795ffd83dbSDimitry Andric if (Ty->isHalfTy() || Ty->isBFloatTy() || Ty->isFloatTy() || 25805ffd83dbSDimitry Andric Ty->isDoubleTy()) { 2581*0b57cec5SDimitry Andric Record.push_back(CFP->getValueAPF().bitcastToAPInt().getZExtValue()); 2582*0b57cec5SDimitry Andric } else if (Ty->isX86_FP80Ty()) { 2583*0b57cec5SDimitry Andric // api needed to prevent premature destruction 2584*0b57cec5SDimitry Andric // bits are not in the same order as a normal i80 APInt, compensate. 2585*0b57cec5SDimitry Andric APInt api = CFP->getValueAPF().bitcastToAPInt(); 2586*0b57cec5SDimitry Andric const uint64_t *p = api.getRawData(); 2587*0b57cec5SDimitry Andric Record.push_back((p[1] << 48) | (p[0] >> 16)); 2588*0b57cec5SDimitry Andric Record.push_back(p[0] & 0xffffLL); 2589*0b57cec5SDimitry Andric } else if (Ty->isFP128Ty() || Ty->isPPC_FP128Ty()) { 2590*0b57cec5SDimitry Andric APInt api = CFP->getValueAPF().bitcastToAPInt(); 2591*0b57cec5SDimitry Andric const uint64_t *p = api.getRawData(); 2592*0b57cec5SDimitry Andric Record.push_back(p[0]); 2593*0b57cec5SDimitry Andric Record.push_back(p[1]); 2594*0b57cec5SDimitry Andric } else { 2595*0b57cec5SDimitry Andric assert(0 && "Unknown FP type!"); 2596*0b57cec5SDimitry Andric } 2597*0b57cec5SDimitry Andric } else if (isa<ConstantDataSequential>(C) && 2598*0b57cec5SDimitry Andric cast<ConstantDataSequential>(C)->isString()) { 2599*0b57cec5SDimitry Andric const ConstantDataSequential *Str = cast<ConstantDataSequential>(C); 2600*0b57cec5SDimitry Andric // Emit constant strings specially. 2601*0b57cec5SDimitry Andric unsigned NumElts = Str->getNumElements(); 2602*0b57cec5SDimitry Andric // If this is a null-terminated string, use the denser CSTRING encoding. 2603*0b57cec5SDimitry Andric if (Str->isCString()) { 2604*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CSTRING; 2605*0b57cec5SDimitry Andric --NumElts; // Don't encode the null, which isn't allowed by char6. 2606*0b57cec5SDimitry Andric } else { 2607*0b57cec5SDimitry Andric Code = bitc::CST_CODE_STRING; 2608*0b57cec5SDimitry Andric AbbrevToUse = String8Abbrev; 2609*0b57cec5SDimitry Andric } 2610*0b57cec5SDimitry Andric bool isCStr7 = Code == bitc::CST_CODE_CSTRING; 2611*0b57cec5SDimitry Andric bool isCStrChar6 = Code == bitc::CST_CODE_CSTRING; 2612*0b57cec5SDimitry Andric for (unsigned i = 0; i != NumElts; ++i) { 2613*0b57cec5SDimitry Andric unsigned char V = Str->getElementAsInteger(i); 2614*0b57cec5SDimitry Andric Record.push_back(V); 2615*0b57cec5SDimitry Andric isCStr7 &= (V & 128) == 0; 2616*0b57cec5SDimitry Andric if (isCStrChar6) 2617*0b57cec5SDimitry Andric isCStrChar6 = BitCodeAbbrevOp::isChar6(V); 2618*0b57cec5SDimitry Andric } 2619*0b57cec5SDimitry Andric 2620*0b57cec5SDimitry Andric if (isCStrChar6) 2621*0b57cec5SDimitry Andric AbbrevToUse = CString6Abbrev; 2622*0b57cec5SDimitry Andric else if (isCStr7) 2623*0b57cec5SDimitry Andric AbbrevToUse = CString7Abbrev; 2624*0b57cec5SDimitry Andric } else if (const ConstantDataSequential *CDS = 2625*0b57cec5SDimitry Andric dyn_cast<ConstantDataSequential>(C)) { 2626*0b57cec5SDimitry Andric Code = bitc::CST_CODE_DATA; 26275ffd83dbSDimitry Andric Type *EltTy = CDS->getElementType(); 2628*0b57cec5SDimitry Andric if (isa<IntegerType>(EltTy)) { 2629*0b57cec5SDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) 2630*0b57cec5SDimitry Andric Record.push_back(CDS->getElementAsInteger(i)); 2631*0b57cec5SDimitry Andric } else { 2632*0b57cec5SDimitry Andric for (unsigned i = 0, e = CDS->getNumElements(); i != e; ++i) 2633*0b57cec5SDimitry Andric Record.push_back( 2634*0b57cec5SDimitry Andric CDS->getElementAsAPFloat(i).bitcastToAPInt().getLimitedValue()); 2635*0b57cec5SDimitry Andric } 2636*0b57cec5SDimitry Andric } else if (isa<ConstantAggregate>(C)) { 2637*0b57cec5SDimitry Andric Code = bitc::CST_CODE_AGGREGATE; 2638*0b57cec5SDimitry Andric for (const Value *Op : C->operands()) 2639*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(Op)); 2640*0b57cec5SDimitry Andric AbbrevToUse = AggregateAbbrev; 2641*0b57cec5SDimitry Andric } else if (const ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) { 2642*0b57cec5SDimitry Andric switch (CE->getOpcode()) { 2643*0b57cec5SDimitry Andric default: 2644*0b57cec5SDimitry Andric if (Instruction::isCast(CE->getOpcode())) { 2645*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_CAST; 2646*0b57cec5SDimitry Andric Record.push_back(getEncodedCastOpcode(CE->getOpcode())); 2647*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 2648*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 2649*0b57cec5SDimitry Andric AbbrevToUse = CONSTANTS_CE_CAST_Abbrev; 2650*0b57cec5SDimitry Andric } else { 2651*0b57cec5SDimitry Andric assert(CE->getNumOperands() == 2 && "Unknown constant expr!"); 2652*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_BINOP; 2653*0b57cec5SDimitry Andric Record.push_back(getEncodedBinaryOpcode(CE->getOpcode())); 2654*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 2655*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 2656*0b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(CE); 2657*0b57cec5SDimitry Andric if (Flags != 0) 2658*0b57cec5SDimitry Andric Record.push_back(Flags); 2659*0b57cec5SDimitry Andric } 2660*0b57cec5SDimitry Andric break; 2661*0b57cec5SDimitry Andric case Instruction::FNeg: { 2662*0b57cec5SDimitry Andric assert(CE->getNumOperands() == 1 && "Unknown constant expr!"); 2663*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_UNOP; 2664*0b57cec5SDimitry Andric Record.push_back(getEncodedUnaryOpcode(CE->getOpcode())); 2665*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 2666*0b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(CE); 2667*0b57cec5SDimitry Andric if (Flags != 0) 2668*0b57cec5SDimitry Andric Record.push_back(Flags); 2669*0b57cec5SDimitry Andric break; 2670*0b57cec5SDimitry Andric } 2671*0b57cec5SDimitry Andric case Instruction::GetElementPtr: { 2672*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_GEP; 2673*0b57cec5SDimitry Andric const auto *GO = cast<GEPOperator>(C); 2674*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(GO->getSourceElementType())); 2675bdd1243dSDimitry Andric if (std::optional<unsigned> Idx = GO->getInRangeIndex()) { 2676*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_GEP_WITH_INRANGE_INDEX; 2677*0b57cec5SDimitry Andric Record.push_back((*Idx << 1) | GO->isInBounds()); 2678*0b57cec5SDimitry Andric } else if (GO->isInBounds()) 2679*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_INBOUNDS_GEP; 2680*0b57cec5SDimitry Andric for (unsigned i = 0, e = CE->getNumOperands(); i != e; ++i) { 2681*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(i)->getType())); 2682*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(i))); 2683*0b57cec5SDimitry Andric } 2684*0b57cec5SDimitry Andric break; 2685*0b57cec5SDimitry Andric } 2686*0b57cec5SDimitry Andric case Instruction::Select: 2687*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_SELECT; 2688*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 2689*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 2690*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(2))); 2691*0b57cec5SDimitry Andric break; 2692*0b57cec5SDimitry Andric case Instruction::ExtractElement: 2693*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_EXTRACTELT; 2694*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 2695*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 2696*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(1)->getType())); 2697*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 2698*0b57cec5SDimitry Andric break; 2699*0b57cec5SDimitry Andric case Instruction::InsertElement: 2700*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_INSERTELT; 2701*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 2702*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 2703*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(2)->getType())); 2704*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(2))); 2705*0b57cec5SDimitry Andric break; 2706*0b57cec5SDimitry Andric case Instruction::ShuffleVector: 2707*0b57cec5SDimitry Andric // If the return type and argument types are the same, this is a 2708*0b57cec5SDimitry Andric // standard shufflevector instruction. If the types are different, 2709*0b57cec5SDimitry Andric // then the shuffle is widening or truncating the input vectors, and 2710*0b57cec5SDimitry Andric // the argument type must also be encoded. 2711*0b57cec5SDimitry Andric if (C->getType() == C->getOperand(0)->getType()) { 2712*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_SHUFFLEVEC; 2713*0b57cec5SDimitry Andric } else { 2714*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_SHUFVEC_EX; 2715*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 2716*0b57cec5SDimitry Andric } 2717*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 2718*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 27195ffd83dbSDimitry Andric Record.push_back(VE.getValueID(CE->getShuffleMaskForBitcode())); 2720*0b57cec5SDimitry Andric break; 2721*0b57cec5SDimitry Andric case Instruction::ICmp: 2722*0b57cec5SDimitry Andric case Instruction::FCmp: 2723*0b57cec5SDimitry Andric Code = bitc::CST_CODE_CE_CMP; 2724*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(C->getOperand(0)->getType())); 2725*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(0))); 2726*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(C->getOperand(1))); 2727*0b57cec5SDimitry Andric Record.push_back(CE->getPredicate()); 2728*0b57cec5SDimitry Andric break; 2729*0b57cec5SDimitry Andric } 2730*0b57cec5SDimitry Andric } else if (const BlockAddress *BA = dyn_cast<BlockAddress>(C)) { 2731*0b57cec5SDimitry Andric Code = bitc::CST_CODE_BLOCKADDRESS; 2732*0b57cec5SDimitry Andric Record.push_back(VE.getTypeID(BA->getFunction()->getType())); 2733*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(BA->getFunction())); 2734*0b57cec5SDimitry Andric Record.push_back(VE.getGlobalBasicBlockID(BA->getBasicBlock())); 2735fe6060f1SDimitry Andric } else if (const auto *Equiv = dyn_cast<DSOLocalEquivalent>(C)) { 2736fe6060f1SDimitry Andric Code = bitc::CST_CODE_DSO_LOCAL_EQUIVALENT; 2737fe6060f1SDimitry Andric Record.push_back(VE.getTypeID(Equiv->getGlobalValue()->getType())); 2738fe6060f1SDimitry Andric Record.push_back(VE.getValueID(Equiv->getGlobalValue())); 27390eae32dcSDimitry Andric } else if (const auto *NC = dyn_cast<NoCFIValue>(C)) { 27400eae32dcSDimitry Andric Code = bitc::CST_CODE_NO_CFI_VALUE; 27410eae32dcSDimitry Andric Record.push_back(VE.getTypeID(NC->getGlobalValue()->getType())); 27420eae32dcSDimitry Andric Record.push_back(VE.getValueID(NC->getGlobalValue())); 2743*0b57cec5SDimitry Andric } else { 2744*0b57cec5SDimitry Andric #ifndef NDEBUG 2745*0b57cec5SDimitry Andric C->dump(); 2746*0b57cec5SDimitry Andric #endif 2747*0b57cec5SDimitry Andric llvm_unreachable("Unknown constant!"); 2748*0b57cec5SDimitry Andric } 2749*0b57cec5SDimitry Andric Stream.EmitRecord(Code, Record, AbbrevToUse); 2750*0b57cec5SDimitry Andric Record.clear(); 2751*0b57cec5SDimitry Andric } 2752*0b57cec5SDimitry Andric 2753*0b57cec5SDimitry Andric Stream.ExitBlock(); 2754*0b57cec5SDimitry Andric } 2755*0b57cec5SDimitry Andric 2756*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleConstants() { 2757*0b57cec5SDimitry Andric const ValueEnumerator::ValueList &Vals = VE.getValues(); 2758*0b57cec5SDimitry Andric 2759*0b57cec5SDimitry Andric // Find the first constant to emit, which is the first non-globalvalue value. 2760*0b57cec5SDimitry Andric // We know globalvalues have been emitted by WriteModuleInfo. 2761*0b57cec5SDimitry Andric for (unsigned i = 0, e = Vals.size(); i != e; ++i) { 2762*0b57cec5SDimitry Andric if (!isa<GlobalValue>(Vals[i].first)) { 2763*0b57cec5SDimitry Andric writeConstants(i, Vals.size(), true); 2764*0b57cec5SDimitry Andric return; 2765*0b57cec5SDimitry Andric } 2766*0b57cec5SDimitry Andric } 2767*0b57cec5SDimitry Andric } 2768*0b57cec5SDimitry Andric 2769*0b57cec5SDimitry Andric /// pushValueAndType - The file has to encode both the value and type id for 2770*0b57cec5SDimitry Andric /// many values, because we need to know what type to create for forward 2771*0b57cec5SDimitry Andric /// references. However, most operands are not forward references, so this type 2772*0b57cec5SDimitry Andric /// field is not needed. 2773*0b57cec5SDimitry Andric /// 2774*0b57cec5SDimitry Andric /// This function adds V's value ID to Vals. If the value ID is higher than the 2775*0b57cec5SDimitry Andric /// instruction ID, then it is a forward reference, and it also includes the 2776*0b57cec5SDimitry Andric /// type ID. The value ID that is written is encoded relative to the InstID. 2777*0b57cec5SDimitry Andric bool ModuleBitcodeWriter::pushValueAndType(const Value *V, unsigned InstID, 2778*0b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals) { 2779*0b57cec5SDimitry Andric unsigned ValID = VE.getValueID(V); 2780*0b57cec5SDimitry Andric // Make encoding relative to the InstID. 2781*0b57cec5SDimitry Andric Vals.push_back(InstID - ValID); 2782*0b57cec5SDimitry Andric if (ValID >= InstID) { 2783*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(V->getType())); 2784*0b57cec5SDimitry Andric return true; 2785*0b57cec5SDimitry Andric } 2786*0b57cec5SDimitry Andric return false; 2787*0b57cec5SDimitry Andric } 2788*0b57cec5SDimitry Andric 27895ffd83dbSDimitry Andric void ModuleBitcodeWriter::writeOperandBundles(const CallBase &CS, 2790*0b57cec5SDimitry Andric unsigned InstID) { 2791*0b57cec5SDimitry Andric SmallVector<unsigned, 64> Record; 27925ffd83dbSDimitry Andric LLVMContext &C = CS.getContext(); 2793*0b57cec5SDimitry Andric 2794*0b57cec5SDimitry Andric for (unsigned i = 0, e = CS.getNumOperandBundles(); i != e; ++i) { 2795*0b57cec5SDimitry Andric const auto &Bundle = CS.getOperandBundleAt(i); 2796*0b57cec5SDimitry Andric Record.push_back(C.getOperandBundleTagID(Bundle.getTagName())); 2797*0b57cec5SDimitry Andric 2798*0b57cec5SDimitry Andric for (auto &Input : Bundle.Inputs) 2799*0b57cec5SDimitry Andric pushValueAndType(Input, InstID, Record); 2800*0b57cec5SDimitry Andric 2801*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_OPERAND_BUNDLE, Record); 2802*0b57cec5SDimitry Andric Record.clear(); 2803*0b57cec5SDimitry Andric } 2804*0b57cec5SDimitry Andric } 2805*0b57cec5SDimitry Andric 2806*0b57cec5SDimitry Andric /// pushValue - Like pushValueAndType, but where the type of the value is 2807*0b57cec5SDimitry Andric /// omitted (perhaps it was already encoded in an earlier operand). 2808*0b57cec5SDimitry Andric void ModuleBitcodeWriter::pushValue(const Value *V, unsigned InstID, 2809*0b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals) { 2810*0b57cec5SDimitry Andric unsigned ValID = VE.getValueID(V); 2811*0b57cec5SDimitry Andric Vals.push_back(InstID - ValID); 2812*0b57cec5SDimitry Andric } 2813*0b57cec5SDimitry Andric 2814*0b57cec5SDimitry Andric void ModuleBitcodeWriter::pushValueSigned(const Value *V, unsigned InstID, 2815*0b57cec5SDimitry Andric SmallVectorImpl<uint64_t> &Vals) { 2816*0b57cec5SDimitry Andric unsigned ValID = VE.getValueID(V); 2817*0b57cec5SDimitry Andric int64_t diff = ((int32_t)InstID - (int32_t)ValID); 2818*0b57cec5SDimitry Andric emitSignedInt64(Vals, diff); 2819*0b57cec5SDimitry Andric } 2820*0b57cec5SDimitry Andric 2821*0b57cec5SDimitry Andric /// WriteInstruction - Emit an instruction to the specified stream. 2822*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeInstruction(const Instruction &I, 2823*0b57cec5SDimitry Andric unsigned InstID, 2824*0b57cec5SDimitry Andric SmallVectorImpl<unsigned> &Vals) { 2825*0b57cec5SDimitry Andric unsigned Code = 0; 2826*0b57cec5SDimitry Andric unsigned AbbrevToUse = 0; 2827*0b57cec5SDimitry Andric VE.setInstructionID(&I); 2828*0b57cec5SDimitry Andric switch (I.getOpcode()) { 2829*0b57cec5SDimitry Andric default: 2830*0b57cec5SDimitry Andric if (Instruction::isCast(I.getOpcode())) { 2831*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CAST; 2832*0b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) 2833*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_CAST_ABBREV; 2834*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getType())); 2835*0b57cec5SDimitry Andric Vals.push_back(getEncodedCastOpcode(I.getOpcode())); 2836*0b57cec5SDimitry Andric } else { 2837*0b57cec5SDimitry Andric assert(isa<BinaryOperator>(I) && "Unknown instruction!"); 2838*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_BINOP; 2839*0b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) 2840*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_BINOP_ABBREV; 2841*0b57cec5SDimitry Andric pushValue(I.getOperand(1), InstID, Vals); 2842*0b57cec5SDimitry Andric Vals.push_back(getEncodedBinaryOpcode(I.getOpcode())); 2843*0b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 2844*0b57cec5SDimitry Andric if (Flags != 0) { 2845*0b57cec5SDimitry Andric if (AbbrevToUse == FUNCTION_INST_BINOP_ABBREV) 2846*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_BINOP_FLAGS_ABBREV; 2847*0b57cec5SDimitry Andric Vals.push_back(Flags); 2848*0b57cec5SDimitry Andric } 2849*0b57cec5SDimitry Andric } 2850*0b57cec5SDimitry Andric break; 2851*0b57cec5SDimitry Andric case Instruction::FNeg: { 2852*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_UNOP; 2853*0b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) 2854*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_UNOP_ABBREV; 2855*0b57cec5SDimitry Andric Vals.push_back(getEncodedUnaryOpcode(I.getOpcode())); 2856*0b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 2857*0b57cec5SDimitry Andric if (Flags != 0) { 2858*0b57cec5SDimitry Andric if (AbbrevToUse == FUNCTION_INST_UNOP_ABBREV) 2859*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_UNOP_FLAGS_ABBREV; 2860*0b57cec5SDimitry Andric Vals.push_back(Flags); 2861*0b57cec5SDimitry Andric } 2862*0b57cec5SDimitry Andric break; 2863*0b57cec5SDimitry Andric } 2864*0b57cec5SDimitry Andric case Instruction::GetElementPtr: { 2865*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_GEP; 2866*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_GEP_ABBREV; 2867*0b57cec5SDimitry Andric auto &GEPInst = cast<GetElementPtrInst>(I); 2868*0b57cec5SDimitry Andric Vals.push_back(GEPInst.isInBounds()); 2869*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(GEPInst.getSourceElementType())); 2870*0b57cec5SDimitry Andric for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i) 2871*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(i), InstID, Vals); 2872*0b57cec5SDimitry Andric break; 2873*0b57cec5SDimitry Andric } 2874*0b57cec5SDimitry Andric case Instruction::ExtractValue: { 2875*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_EXTRACTVAL; 2876*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 2877*0b57cec5SDimitry Andric const ExtractValueInst *EVI = cast<ExtractValueInst>(&I); 2878*0b57cec5SDimitry Andric Vals.append(EVI->idx_begin(), EVI->idx_end()); 2879*0b57cec5SDimitry Andric break; 2880*0b57cec5SDimitry Andric } 2881*0b57cec5SDimitry Andric case Instruction::InsertValue: { 2882*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_INSERTVAL; 2883*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 2884*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); 2885*0b57cec5SDimitry Andric const InsertValueInst *IVI = cast<InsertValueInst>(&I); 2886*0b57cec5SDimitry Andric Vals.append(IVI->idx_begin(), IVI->idx_end()); 2887*0b57cec5SDimitry Andric break; 2888*0b57cec5SDimitry Andric } 2889*0b57cec5SDimitry Andric case Instruction::Select: { 2890*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_VSELECT; 2891*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); 2892*0b57cec5SDimitry Andric pushValue(I.getOperand(2), InstID, Vals); 2893*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 2894*0b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 2895*0b57cec5SDimitry Andric if (Flags != 0) 2896*0b57cec5SDimitry Andric Vals.push_back(Flags); 2897*0b57cec5SDimitry Andric break; 2898*0b57cec5SDimitry Andric } 2899*0b57cec5SDimitry Andric case Instruction::ExtractElement: 2900*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_EXTRACTELT; 2901*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 2902*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); 2903*0b57cec5SDimitry Andric break; 2904*0b57cec5SDimitry Andric case Instruction::InsertElement: 2905*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_INSERTELT; 2906*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 2907*0b57cec5SDimitry Andric pushValue(I.getOperand(1), InstID, Vals); 2908*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(2), InstID, Vals); 2909*0b57cec5SDimitry Andric break; 2910*0b57cec5SDimitry Andric case Instruction::ShuffleVector: 2911*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_SHUFFLEVEC; 2912*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 2913*0b57cec5SDimitry Andric pushValue(I.getOperand(1), InstID, Vals); 29145ffd83dbSDimitry Andric pushValue(cast<ShuffleVectorInst>(I).getShuffleMaskForBitcode(), InstID, 29155ffd83dbSDimitry Andric Vals); 2916*0b57cec5SDimitry Andric break; 2917*0b57cec5SDimitry Andric case Instruction::ICmp: 2918*0b57cec5SDimitry Andric case Instruction::FCmp: { 2919*0b57cec5SDimitry Andric // compare returning Int1Ty or vector of Int1Ty 2920*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CMP2; 2921*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 2922*0b57cec5SDimitry Andric pushValue(I.getOperand(1), InstID, Vals); 2923*0b57cec5SDimitry Andric Vals.push_back(cast<CmpInst>(I).getPredicate()); 2924*0b57cec5SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 2925*0b57cec5SDimitry Andric if (Flags != 0) 2926*0b57cec5SDimitry Andric Vals.push_back(Flags); 2927*0b57cec5SDimitry Andric break; 2928*0b57cec5SDimitry Andric } 2929*0b57cec5SDimitry Andric 2930*0b57cec5SDimitry Andric case Instruction::Ret: 2931*0b57cec5SDimitry Andric { 2932*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_RET; 2933*0b57cec5SDimitry Andric unsigned NumOperands = I.getNumOperands(); 2934*0b57cec5SDimitry Andric if (NumOperands == 0) 2935*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_RET_VOID_ABBREV; 2936*0b57cec5SDimitry Andric else if (NumOperands == 1) { 2937*0b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) 2938*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_RET_VAL_ABBREV; 2939*0b57cec5SDimitry Andric } else { 2940*0b57cec5SDimitry Andric for (unsigned i = 0, e = NumOperands; i != e; ++i) 2941*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(i), InstID, Vals); 2942*0b57cec5SDimitry Andric } 2943*0b57cec5SDimitry Andric } 2944*0b57cec5SDimitry Andric break; 2945*0b57cec5SDimitry Andric case Instruction::Br: 2946*0b57cec5SDimitry Andric { 2947*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_BR; 2948*0b57cec5SDimitry Andric const BranchInst &II = cast<BranchInst>(I); 2949*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(II.getSuccessor(0))); 2950*0b57cec5SDimitry Andric if (II.isConditional()) { 2951*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(II.getSuccessor(1))); 2952*0b57cec5SDimitry Andric pushValue(II.getCondition(), InstID, Vals); 2953*0b57cec5SDimitry Andric } 2954*0b57cec5SDimitry Andric } 2955*0b57cec5SDimitry Andric break; 2956*0b57cec5SDimitry Andric case Instruction::Switch: 2957*0b57cec5SDimitry Andric { 2958*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_SWITCH; 2959*0b57cec5SDimitry Andric const SwitchInst &SI = cast<SwitchInst>(I); 2960*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(SI.getCondition()->getType())); 2961*0b57cec5SDimitry Andric pushValue(SI.getCondition(), InstID, Vals); 2962*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(SI.getDefaultDest())); 2963*0b57cec5SDimitry Andric for (auto Case : SI.cases()) { 2964*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(Case.getCaseValue())); 2965*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(Case.getCaseSuccessor())); 2966*0b57cec5SDimitry Andric } 2967*0b57cec5SDimitry Andric } 2968*0b57cec5SDimitry Andric break; 2969*0b57cec5SDimitry Andric case Instruction::IndirectBr: 2970*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_INDIRECTBR; 2971*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); 2972*0b57cec5SDimitry Andric // Encode the address operand as relative, but not the basic blocks. 2973*0b57cec5SDimitry Andric pushValue(I.getOperand(0), InstID, Vals); 2974*0b57cec5SDimitry Andric for (unsigned i = 1, e = I.getNumOperands(); i != e; ++i) 2975*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(i))); 2976*0b57cec5SDimitry Andric break; 2977*0b57cec5SDimitry Andric 2978*0b57cec5SDimitry Andric case Instruction::Invoke: { 2979*0b57cec5SDimitry Andric const InvokeInst *II = cast<InvokeInst>(&I); 29805ffd83dbSDimitry Andric const Value *Callee = II->getCalledOperand(); 2981*0b57cec5SDimitry Andric FunctionType *FTy = II->getFunctionType(); 2982*0b57cec5SDimitry Andric 2983*0b57cec5SDimitry Andric if (II->hasOperandBundles()) 29845ffd83dbSDimitry Andric writeOperandBundles(*II, InstID); 2985*0b57cec5SDimitry Andric 2986*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_INVOKE; 2987*0b57cec5SDimitry Andric 2988*0b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(II->getAttributes())); 2989*0b57cec5SDimitry Andric Vals.push_back(II->getCallingConv() | 1 << 13); 2990*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(II->getNormalDest())); 2991*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(II->getUnwindDest())); 2992*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(FTy)); 2993*0b57cec5SDimitry Andric pushValueAndType(Callee, InstID, Vals); 2994*0b57cec5SDimitry Andric 2995*0b57cec5SDimitry Andric // Emit value #'s for the fixed parameters. 2996*0b57cec5SDimitry Andric for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) 2997*0b57cec5SDimitry Andric pushValue(I.getOperand(i), InstID, Vals); // fixed param. 2998*0b57cec5SDimitry Andric 2999*0b57cec5SDimitry Andric // Emit type/value pairs for varargs params. 3000*0b57cec5SDimitry Andric if (FTy->isVarArg()) { 3001349cc55cSDimitry Andric for (unsigned i = FTy->getNumParams(), e = II->arg_size(); i != e; ++i) 3002*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(i), InstID, Vals); // vararg 3003*0b57cec5SDimitry Andric } 3004*0b57cec5SDimitry Andric break; 3005*0b57cec5SDimitry Andric } 3006*0b57cec5SDimitry Andric case Instruction::Resume: 3007*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_RESUME; 3008*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 3009*0b57cec5SDimitry Andric break; 3010*0b57cec5SDimitry Andric case Instruction::CleanupRet: { 3011*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CLEANUPRET; 3012*0b57cec5SDimitry Andric const auto &CRI = cast<CleanupReturnInst>(I); 3013*0b57cec5SDimitry Andric pushValue(CRI.getCleanupPad(), InstID, Vals); 3014*0b57cec5SDimitry Andric if (CRI.hasUnwindDest()) 3015*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CRI.getUnwindDest())); 3016*0b57cec5SDimitry Andric break; 3017*0b57cec5SDimitry Andric } 3018*0b57cec5SDimitry Andric case Instruction::CatchRet: { 3019*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CATCHRET; 3020*0b57cec5SDimitry Andric const auto &CRI = cast<CatchReturnInst>(I); 3021*0b57cec5SDimitry Andric pushValue(CRI.getCatchPad(), InstID, Vals); 3022*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CRI.getSuccessor())); 3023*0b57cec5SDimitry Andric break; 3024*0b57cec5SDimitry Andric } 3025*0b57cec5SDimitry Andric case Instruction::CleanupPad: 3026*0b57cec5SDimitry Andric case Instruction::CatchPad: { 3027*0b57cec5SDimitry Andric const auto &FuncletPad = cast<FuncletPadInst>(I); 3028*0b57cec5SDimitry Andric Code = isa<CatchPadInst>(FuncletPad) ? bitc::FUNC_CODE_INST_CATCHPAD 3029*0b57cec5SDimitry Andric : bitc::FUNC_CODE_INST_CLEANUPPAD; 3030*0b57cec5SDimitry Andric pushValue(FuncletPad.getParentPad(), InstID, Vals); 3031*0b57cec5SDimitry Andric 3032bdd1243dSDimitry Andric unsigned NumArgOperands = FuncletPad.arg_size(); 3033*0b57cec5SDimitry Andric Vals.push_back(NumArgOperands); 3034*0b57cec5SDimitry Andric for (unsigned Op = 0; Op != NumArgOperands; ++Op) 3035*0b57cec5SDimitry Andric pushValueAndType(FuncletPad.getArgOperand(Op), InstID, Vals); 3036*0b57cec5SDimitry Andric break; 3037*0b57cec5SDimitry Andric } 3038*0b57cec5SDimitry Andric case Instruction::CatchSwitch: { 3039*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CATCHSWITCH; 3040*0b57cec5SDimitry Andric const auto &CatchSwitch = cast<CatchSwitchInst>(I); 3041*0b57cec5SDimitry Andric 3042*0b57cec5SDimitry Andric pushValue(CatchSwitch.getParentPad(), InstID, Vals); 3043*0b57cec5SDimitry Andric 3044*0b57cec5SDimitry Andric unsigned NumHandlers = CatchSwitch.getNumHandlers(); 3045*0b57cec5SDimitry Andric Vals.push_back(NumHandlers); 3046*0b57cec5SDimitry Andric for (const BasicBlock *CatchPadBB : CatchSwitch.handlers()) 3047*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CatchPadBB)); 3048*0b57cec5SDimitry Andric 3049*0b57cec5SDimitry Andric if (CatchSwitch.hasUnwindDest()) 3050*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CatchSwitch.getUnwindDest())); 3051*0b57cec5SDimitry Andric break; 3052*0b57cec5SDimitry Andric } 3053*0b57cec5SDimitry Andric case Instruction::CallBr: { 3054*0b57cec5SDimitry Andric const CallBrInst *CBI = cast<CallBrInst>(&I); 30555ffd83dbSDimitry Andric const Value *Callee = CBI->getCalledOperand(); 3056*0b57cec5SDimitry Andric FunctionType *FTy = CBI->getFunctionType(); 3057*0b57cec5SDimitry Andric 3058*0b57cec5SDimitry Andric if (CBI->hasOperandBundles()) 30595ffd83dbSDimitry Andric writeOperandBundles(*CBI, InstID); 3060*0b57cec5SDimitry Andric 3061*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CALLBR; 3062*0b57cec5SDimitry Andric 3063*0b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(CBI->getAttributes())); 3064*0b57cec5SDimitry Andric 3065*0b57cec5SDimitry Andric Vals.push_back(CBI->getCallingConv() << bitc::CALL_CCONV | 3066*0b57cec5SDimitry Andric 1 << bitc::CALL_EXPLICIT_TYPE); 3067*0b57cec5SDimitry Andric 3068*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CBI->getDefaultDest())); 3069*0b57cec5SDimitry Andric Vals.push_back(CBI->getNumIndirectDests()); 3070*0b57cec5SDimitry Andric for (unsigned i = 0, e = CBI->getNumIndirectDests(); i != e; ++i) 3071*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CBI->getIndirectDest(i))); 3072*0b57cec5SDimitry Andric 3073*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(FTy)); 3074*0b57cec5SDimitry Andric pushValueAndType(Callee, InstID, Vals); 3075*0b57cec5SDimitry Andric 3076*0b57cec5SDimitry Andric // Emit value #'s for the fixed parameters. 3077*0b57cec5SDimitry Andric for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) 3078*0b57cec5SDimitry Andric pushValue(I.getOperand(i), InstID, Vals); // fixed param. 3079*0b57cec5SDimitry Andric 3080*0b57cec5SDimitry Andric // Emit type/value pairs for varargs params. 3081*0b57cec5SDimitry Andric if (FTy->isVarArg()) { 3082349cc55cSDimitry Andric for (unsigned i = FTy->getNumParams(), e = CBI->arg_size(); i != e; ++i) 3083*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(i), InstID, Vals); // vararg 3084*0b57cec5SDimitry Andric } 3085*0b57cec5SDimitry Andric break; 3086*0b57cec5SDimitry Andric } 3087*0b57cec5SDimitry Andric case Instruction::Unreachable: 3088*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_UNREACHABLE; 3089*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_UNREACHABLE_ABBREV; 3090*0b57cec5SDimitry Andric break; 3091*0b57cec5SDimitry Andric 3092*0b57cec5SDimitry Andric case Instruction::PHI: { 3093*0b57cec5SDimitry Andric const PHINode &PN = cast<PHINode>(I); 3094*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_PHI; 3095*0b57cec5SDimitry Andric // With the newer instruction encoding, forward references could give 3096*0b57cec5SDimitry Andric // negative valued IDs. This is most common for PHIs, so we use 3097*0b57cec5SDimitry Andric // signed VBRs. 3098*0b57cec5SDimitry Andric SmallVector<uint64_t, 128> Vals64; 3099*0b57cec5SDimitry Andric Vals64.push_back(VE.getTypeID(PN.getType())); 3100*0b57cec5SDimitry Andric for (unsigned i = 0, e = PN.getNumIncomingValues(); i != e; ++i) { 3101*0b57cec5SDimitry Andric pushValueSigned(PN.getIncomingValue(i), InstID, Vals64); 3102*0b57cec5SDimitry Andric Vals64.push_back(VE.getValueID(PN.getIncomingBlock(i))); 3103*0b57cec5SDimitry Andric } 31048bcb0991SDimitry Andric 31058bcb0991SDimitry Andric uint64_t Flags = getOptimizationFlags(&I); 31068bcb0991SDimitry Andric if (Flags != 0) 31078bcb0991SDimitry Andric Vals64.push_back(Flags); 31088bcb0991SDimitry Andric 3109*0b57cec5SDimitry Andric // Emit a Vals64 vector and exit. 3110*0b57cec5SDimitry Andric Stream.EmitRecord(Code, Vals64, AbbrevToUse); 3111*0b57cec5SDimitry Andric Vals64.clear(); 3112*0b57cec5SDimitry Andric return; 3113*0b57cec5SDimitry Andric } 3114*0b57cec5SDimitry Andric 3115*0b57cec5SDimitry Andric case Instruction::LandingPad: { 3116*0b57cec5SDimitry Andric const LandingPadInst &LP = cast<LandingPadInst>(I); 3117*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_LANDINGPAD; 3118*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(LP.getType())); 3119*0b57cec5SDimitry Andric Vals.push_back(LP.isCleanup()); 3120*0b57cec5SDimitry Andric Vals.push_back(LP.getNumClauses()); 3121*0b57cec5SDimitry Andric for (unsigned I = 0, E = LP.getNumClauses(); I != E; ++I) { 3122*0b57cec5SDimitry Andric if (LP.isCatch(I)) 3123*0b57cec5SDimitry Andric Vals.push_back(LandingPadInst::Catch); 3124*0b57cec5SDimitry Andric else 3125*0b57cec5SDimitry Andric Vals.push_back(LandingPadInst::Filter); 3126*0b57cec5SDimitry Andric pushValueAndType(LP.getClause(I), InstID, Vals); 3127*0b57cec5SDimitry Andric } 3128*0b57cec5SDimitry Andric break; 3129*0b57cec5SDimitry Andric } 3130*0b57cec5SDimitry Andric 3131*0b57cec5SDimitry Andric case Instruction::Alloca: { 3132*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_ALLOCA; 3133*0b57cec5SDimitry Andric const AllocaInst &AI = cast<AllocaInst>(I); 3134*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(AI.getAllocatedType())); 3135*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); 3136*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(I.getOperand(0))); // size. 3137e8d8bef9SDimitry Andric using APV = AllocaPackedValues; 3138e8d8bef9SDimitry Andric unsigned Record = 0; 3139349cc55cSDimitry Andric unsigned EncodedAlign = getEncodedAlign(AI.getAlign()); 3140349cc55cSDimitry Andric Bitfield::set<APV::AlignLower>( 3141349cc55cSDimitry Andric Record, EncodedAlign & ((1 << APV::AlignLower::Bits) - 1)); 3142349cc55cSDimitry Andric Bitfield::set<APV::AlignUpper>(Record, 3143349cc55cSDimitry Andric EncodedAlign >> APV::AlignLower::Bits); 3144e8d8bef9SDimitry Andric Bitfield::set<APV::UsedWithInAlloca>(Record, AI.isUsedWithInAlloca()); 3145e8d8bef9SDimitry Andric Bitfield::set<APV::ExplicitType>(Record, true); 3146e8d8bef9SDimitry Andric Bitfield::set<APV::SwiftError>(Record, AI.isSwiftError()); 3147e8d8bef9SDimitry Andric Vals.push_back(Record); 314881ad6265SDimitry Andric 314981ad6265SDimitry Andric unsigned AS = AI.getAddressSpace(); 315081ad6265SDimitry Andric if (AS != M.getDataLayout().getAllocaAddrSpace()) 315181ad6265SDimitry Andric Vals.push_back(AS); 3152*0b57cec5SDimitry Andric break; 3153*0b57cec5SDimitry Andric } 3154*0b57cec5SDimitry Andric 3155*0b57cec5SDimitry Andric case Instruction::Load: 3156*0b57cec5SDimitry Andric if (cast<LoadInst>(I).isAtomic()) { 3157*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_LOADATOMIC; 3158*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 3159*0b57cec5SDimitry Andric } else { 3160*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_LOAD; 3161*0b57cec5SDimitry Andric if (!pushValueAndType(I.getOperand(0), InstID, Vals)) // ptr 3162*0b57cec5SDimitry Andric AbbrevToUse = FUNCTION_INST_LOAD_ABBREV; 3163*0b57cec5SDimitry Andric } 3164*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getType())); 3165e8d8bef9SDimitry Andric Vals.push_back(getEncodedAlign(cast<LoadInst>(I).getAlign())); 3166*0b57cec5SDimitry Andric Vals.push_back(cast<LoadInst>(I).isVolatile()); 3167*0b57cec5SDimitry Andric if (cast<LoadInst>(I).isAtomic()) { 3168*0b57cec5SDimitry Andric Vals.push_back(getEncodedOrdering(cast<LoadInst>(I).getOrdering())); 3169*0b57cec5SDimitry Andric Vals.push_back(getEncodedSyncScopeID(cast<LoadInst>(I).getSyncScopeID())); 3170*0b57cec5SDimitry Andric } 3171*0b57cec5SDimitry Andric break; 3172*0b57cec5SDimitry Andric case Instruction::Store: 3173*0b57cec5SDimitry Andric if (cast<StoreInst>(I).isAtomic()) 3174*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_STOREATOMIC; 3175*0b57cec5SDimitry Andric else 3176*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_STORE; 3177*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); // ptrty + ptr 3178*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); // valty + val 3179e8d8bef9SDimitry Andric Vals.push_back(getEncodedAlign(cast<StoreInst>(I).getAlign())); 3180*0b57cec5SDimitry Andric Vals.push_back(cast<StoreInst>(I).isVolatile()); 3181*0b57cec5SDimitry Andric if (cast<StoreInst>(I).isAtomic()) { 3182*0b57cec5SDimitry Andric Vals.push_back(getEncodedOrdering(cast<StoreInst>(I).getOrdering())); 3183*0b57cec5SDimitry Andric Vals.push_back( 3184*0b57cec5SDimitry Andric getEncodedSyncScopeID(cast<StoreInst>(I).getSyncScopeID())); 3185*0b57cec5SDimitry Andric } 3186*0b57cec5SDimitry Andric break; 3187*0b57cec5SDimitry Andric case Instruction::AtomicCmpXchg: 3188*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CMPXCHG; 3189*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); // ptrty + ptr 3190*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); // cmp. 3191*0b57cec5SDimitry Andric pushValue(I.getOperand(2), InstID, Vals); // newval. 3192*0b57cec5SDimitry Andric Vals.push_back(cast<AtomicCmpXchgInst>(I).isVolatile()); 3193*0b57cec5SDimitry Andric Vals.push_back( 3194*0b57cec5SDimitry Andric getEncodedOrdering(cast<AtomicCmpXchgInst>(I).getSuccessOrdering())); 3195*0b57cec5SDimitry Andric Vals.push_back( 3196*0b57cec5SDimitry Andric getEncodedSyncScopeID(cast<AtomicCmpXchgInst>(I).getSyncScopeID())); 3197*0b57cec5SDimitry Andric Vals.push_back( 3198*0b57cec5SDimitry Andric getEncodedOrdering(cast<AtomicCmpXchgInst>(I).getFailureOrdering())); 3199*0b57cec5SDimitry Andric Vals.push_back(cast<AtomicCmpXchgInst>(I).isWeak()); 3200fe6060f1SDimitry Andric Vals.push_back(getEncodedAlign(cast<AtomicCmpXchgInst>(I).getAlign())); 3201*0b57cec5SDimitry Andric break; 3202*0b57cec5SDimitry Andric case Instruction::AtomicRMW: 3203*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_ATOMICRMW; 3204*0b57cec5SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); // ptrty + ptr 3205fe6060f1SDimitry Andric pushValueAndType(I.getOperand(1), InstID, Vals); // valty + val 3206*0b57cec5SDimitry Andric Vals.push_back( 3207*0b57cec5SDimitry Andric getEncodedRMWOperation(cast<AtomicRMWInst>(I).getOperation())); 3208*0b57cec5SDimitry Andric Vals.push_back(cast<AtomicRMWInst>(I).isVolatile()); 3209*0b57cec5SDimitry Andric Vals.push_back(getEncodedOrdering(cast<AtomicRMWInst>(I).getOrdering())); 3210*0b57cec5SDimitry Andric Vals.push_back( 3211*0b57cec5SDimitry Andric getEncodedSyncScopeID(cast<AtomicRMWInst>(I).getSyncScopeID())); 3212fe6060f1SDimitry Andric Vals.push_back(getEncodedAlign(cast<AtomicRMWInst>(I).getAlign())); 3213*0b57cec5SDimitry Andric break; 3214*0b57cec5SDimitry Andric case Instruction::Fence: 3215*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_FENCE; 3216*0b57cec5SDimitry Andric Vals.push_back(getEncodedOrdering(cast<FenceInst>(I).getOrdering())); 3217*0b57cec5SDimitry Andric Vals.push_back(getEncodedSyncScopeID(cast<FenceInst>(I).getSyncScopeID())); 3218*0b57cec5SDimitry Andric break; 3219*0b57cec5SDimitry Andric case Instruction::Call: { 3220*0b57cec5SDimitry Andric const CallInst &CI = cast<CallInst>(I); 3221*0b57cec5SDimitry Andric FunctionType *FTy = CI.getFunctionType(); 3222*0b57cec5SDimitry Andric 3223*0b57cec5SDimitry Andric if (CI.hasOperandBundles()) 32245ffd83dbSDimitry Andric writeOperandBundles(CI, InstID); 3225*0b57cec5SDimitry Andric 3226*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_CALL; 3227*0b57cec5SDimitry Andric 3228*0b57cec5SDimitry Andric Vals.push_back(VE.getAttributeListID(CI.getAttributes())); 3229*0b57cec5SDimitry Andric 3230*0b57cec5SDimitry Andric unsigned Flags = getOptimizationFlags(&I); 3231*0b57cec5SDimitry Andric Vals.push_back(CI.getCallingConv() << bitc::CALL_CCONV | 3232*0b57cec5SDimitry Andric unsigned(CI.isTailCall()) << bitc::CALL_TAIL | 3233*0b57cec5SDimitry Andric unsigned(CI.isMustTailCall()) << bitc::CALL_MUSTTAIL | 3234*0b57cec5SDimitry Andric 1 << bitc::CALL_EXPLICIT_TYPE | 3235*0b57cec5SDimitry Andric unsigned(CI.isNoTailCall()) << bitc::CALL_NOTAIL | 3236*0b57cec5SDimitry Andric unsigned(Flags != 0) << bitc::CALL_FMF); 3237*0b57cec5SDimitry Andric if (Flags != 0) 3238*0b57cec5SDimitry Andric Vals.push_back(Flags); 3239*0b57cec5SDimitry Andric 3240*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(FTy)); 32415ffd83dbSDimitry Andric pushValueAndType(CI.getCalledOperand(), InstID, Vals); // Callee 3242*0b57cec5SDimitry Andric 3243*0b57cec5SDimitry Andric // Emit value #'s for the fixed parameters. 3244*0b57cec5SDimitry Andric for (unsigned i = 0, e = FTy->getNumParams(); i != e; ++i) { 3245*0b57cec5SDimitry Andric // Check for labels (can happen with asm labels). 3246*0b57cec5SDimitry Andric if (FTy->getParamType(i)->isLabelTy()) 3247*0b57cec5SDimitry Andric Vals.push_back(VE.getValueID(CI.getArgOperand(i))); 3248*0b57cec5SDimitry Andric else 3249*0b57cec5SDimitry Andric pushValue(CI.getArgOperand(i), InstID, Vals); // fixed param. 3250*0b57cec5SDimitry Andric } 3251*0b57cec5SDimitry Andric 3252*0b57cec5SDimitry Andric // Emit type/value pairs for varargs params. 3253*0b57cec5SDimitry Andric if (FTy->isVarArg()) { 3254349cc55cSDimitry Andric for (unsigned i = FTy->getNumParams(), e = CI.arg_size(); i != e; ++i) 3255*0b57cec5SDimitry Andric pushValueAndType(CI.getArgOperand(i), InstID, Vals); // varargs 3256*0b57cec5SDimitry Andric } 3257*0b57cec5SDimitry Andric break; 3258*0b57cec5SDimitry Andric } 3259*0b57cec5SDimitry Andric case Instruction::VAArg: 3260*0b57cec5SDimitry Andric Code = bitc::FUNC_CODE_INST_VAARG; 3261*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); // valistty 3262*0b57cec5SDimitry Andric pushValue(I.getOperand(0), InstID, Vals); // valist. 3263*0b57cec5SDimitry Andric Vals.push_back(VE.getTypeID(I.getType())); // restype. 3264*0b57cec5SDimitry Andric break; 3265480093f4SDimitry Andric case Instruction::Freeze: 3266480093f4SDimitry Andric Code = bitc::FUNC_CODE_INST_FREEZE; 3267480093f4SDimitry Andric pushValueAndType(I.getOperand(0), InstID, Vals); 3268480093f4SDimitry Andric break; 3269*0b57cec5SDimitry Andric } 3270*0b57cec5SDimitry Andric 3271*0b57cec5SDimitry Andric Stream.EmitRecord(Code, Vals, AbbrevToUse); 3272*0b57cec5SDimitry Andric Vals.clear(); 3273*0b57cec5SDimitry Andric } 3274*0b57cec5SDimitry Andric 3275*0b57cec5SDimitry Andric /// Write a GlobalValue VST to the module. The purpose of this data structure is 3276*0b57cec5SDimitry Andric /// to allow clients to efficiently find the function body. 3277*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeGlobalValueSymbolTable( 3278*0b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> &FunctionToBitcodeIndex) { 3279*0b57cec5SDimitry Andric // Get the offset of the VST we are writing, and backpatch it into 3280*0b57cec5SDimitry Andric // the VST forward declaration record. 3281*0b57cec5SDimitry Andric uint64_t VSTOffset = Stream.GetCurrentBitNo(); 3282*0b57cec5SDimitry Andric // The BitcodeStartBit was the stream offset of the identification block. 3283*0b57cec5SDimitry Andric VSTOffset -= bitcodeStartBit(); 3284*0b57cec5SDimitry Andric assert((VSTOffset & 31) == 0 && "VST block not 32-bit aligned"); 3285*0b57cec5SDimitry Andric // Note that we add 1 here because the offset is relative to one word 3286*0b57cec5SDimitry Andric // before the start of the identification block, which was historically 3287*0b57cec5SDimitry Andric // always the start of the regular bitcode header. 3288*0b57cec5SDimitry Andric Stream.BackpatchWord(VSTOffsetPlaceholder, VSTOffset / 32 + 1); 3289*0b57cec5SDimitry Andric 3290*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::VALUE_SYMTAB_BLOCK_ID, 4); 3291*0b57cec5SDimitry Andric 3292*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3293*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_FNENTRY)); 3294*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // value id 3295*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // funcoffset 3296*0b57cec5SDimitry Andric unsigned FnEntryAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 3297*0b57cec5SDimitry Andric 3298*0b57cec5SDimitry Andric for (const Function &F : M) { 3299*0b57cec5SDimitry Andric uint64_t Record[2]; 3300*0b57cec5SDimitry Andric 3301*0b57cec5SDimitry Andric if (F.isDeclaration()) 3302*0b57cec5SDimitry Andric continue; 3303*0b57cec5SDimitry Andric 3304*0b57cec5SDimitry Andric Record[0] = VE.getValueID(&F); 3305*0b57cec5SDimitry Andric 3306*0b57cec5SDimitry Andric // Save the word offset of the function (from the start of the 3307*0b57cec5SDimitry Andric // actual bitcode written to the stream). 3308*0b57cec5SDimitry Andric uint64_t BitcodeIndex = FunctionToBitcodeIndex[&F] - bitcodeStartBit(); 3309*0b57cec5SDimitry Andric assert((BitcodeIndex & 31) == 0 && "function block not 32-bit aligned"); 3310*0b57cec5SDimitry Andric // Note that we add 1 here because the offset is relative to one word 3311*0b57cec5SDimitry Andric // before the start of the identification block, which was historically 3312*0b57cec5SDimitry Andric // always the start of the regular bitcode header. 3313*0b57cec5SDimitry Andric Record[1] = BitcodeIndex / 32 + 1; 3314*0b57cec5SDimitry Andric 3315*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::VST_CODE_FNENTRY, Record, FnEntryAbbrev); 3316*0b57cec5SDimitry Andric } 3317*0b57cec5SDimitry Andric 3318*0b57cec5SDimitry Andric Stream.ExitBlock(); 3319*0b57cec5SDimitry Andric } 3320*0b57cec5SDimitry Andric 3321*0b57cec5SDimitry Andric /// Emit names for arguments, instructions and basic blocks in a function. 3322*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeFunctionLevelValueSymbolTable( 3323*0b57cec5SDimitry Andric const ValueSymbolTable &VST) { 3324*0b57cec5SDimitry Andric if (VST.empty()) 3325*0b57cec5SDimitry Andric return; 3326*0b57cec5SDimitry Andric 3327*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::VALUE_SYMTAB_BLOCK_ID, 4); 3328*0b57cec5SDimitry Andric 3329*0b57cec5SDimitry Andric // FIXME: Set up the abbrev, we know how many values there are! 3330*0b57cec5SDimitry Andric // FIXME: We know if the type names can use 7-bit ascii. 3331*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> NameVals; 3332*0b57cec5SDimitry Andric 3333*0b57cec5SDimitry Andric for (const ValueName &Name : VST) { 3334*0b57cec5SDimitry Andric // Figure out the encoding to use for the name. 3335*0b57cec5SDimitry Andric StringEncoding Bits = getStringEncoding(Name.getKey()); 3336*0b57cec5SDimitry Andric 3337*0b57cec5SDimitry Andric unsigned AbbrevToUse = VST_ENTRY_8_ABBREV; 3338*0b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(Name.getValue())); 3339*0b57cec5SDimitry Andric 3340*0b57cec5SDimitry Andric // VST_CODE_ENTRY: [valueid, namechar x N] 3341*0b57cec5SDimitry Andric // VST_CODE_BBENTRY: [bbid, namechar x N] 3342*0b57cec5SDimitry Andric unsigned Code; 3343*0b57cec5SDimitry Andric if (isa<BasicBlock>(Name.getValue())) { 3344*0b57cec5SDimitry Andric Code = bitc::VST_CODE_BBENTRY; 3345*0b57cec5SDimitry Andric if (Bits == SE_Char6) 3346*0b57cec5SDimitry Andric AbbrevToUse = VST_BBENTRY_6_ABBREV; 3347*0b57cec5SDimitry Andric } else { 3348*0b57cec5SDimitry Andric Code = bitc::VST_CODE_ENTRY; 3349*0b57cec5SDimitry Andric if (Bits == SE_Char6) 3350*0b57cec5SDimitry Andric AbbrevToUse = VST_ENTRY_6_ABBREV; 3351*0b57cec5SDimitry Andric else if (Bits == SE_Fixed7) 3352*0b57cec5SDimitry Andric AbbrevToUse = VST_ENTRY_7_ABBREV; 3353*0b57cec5SDimitry Andric } 3354*0b57cec5SDimitry Andric 3355*0b57cec5SDimitry Andric for (const auto P : Name.getKey()) 3356*0b57cec5SDimitry Andric NameVals.push_back((unsigned char)P); 3357*0b57cec5SDimitry Andric 3358*0b57cec5SDimitry Andric // Emit the finished record. 3359*0b57cec5SDimitry Andric Stream.EmitRecord(Code, NameVals, AbbrevToUse); 3360*0b57cec5SDimitry Andric NameVals.clear(); 3361*0b57cec5SDimitry Andric } 3362*0b57cec5SDimitry Andric 3363*0b57cec5SDimitry Andric Stream.ExitBlock(); 3364*0b57cec5SDimitry Andric } 3365*0b57cec5SDimitry Andric 3366*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeUseList(UseListOrder &&Order) { 3367*0b57cec5SDimitry Andric assert(Order.Shuffle.size() >= 2 && "Shuffle too small"); 3368*0b57cec5SDimitry Andric unsigned Code; 3369*0b57cec5SDimitry Andric if (isa<BasicBlock>(Order.V)) 3370*0b57cec5SDimitry Andric Code = bitc::USELIST_CODE_BB; 3371*0b57cec5SDimitry Andric else 3372*0b57cec5SDimitry Andric Code = bitc::USELIST_CODE_DEFAULT; 3373*0b57cec5SDimitry Andric 3374*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record(Order.Shuffle.begin(), Order.Shuffle.end()); 3375*0b57cec5SDimitry Andric Record.push_back(VE.getValueID(Order.V)); 3376*0b57cec5SDimitry Andric Stream.EmitRecord(Code, Record); 3377*0b57cec5SDimitry Andric } 3378*0b57cec5SDimitry Andric 3379*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeUseListBlock(const Function *F) { 3380*0b57cec5SDimitry Andric assert(VE.shouldPreserveUseListOrder() && 3381*0b57cec5SDimitry Andric "Expected to be preserving use-list order"); 3382*0b57cec5SDimitry Andric 3383*0b57cec5SDimitry Andric auto hasMore = [&]() { 3384*0b57cec5SDimitry Andric return !VE.UseListOrders.empty() && VE.UseListOrders.back().F == F; 3385*0b57cec5SDimitry Andric }; 3386*0b57cec5SDimitry Andric if (!hasMore()) 3387*0b57cec5SDimitry Andric // Nothing to do. 3388*0b57cec5SDimitry Andric return; 3389*0b57cec5SDimitry Andric 3390*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::USELIST_BLOCK_ID, 3); 3391*0b57cec5SDimitry Andric while (hasMore()) { 3392*0b57cec5SDimitry Andric writeUseList(std::move(VE.UseListOrders.back())); 3393*0b57cec5SDimitry Andric VE.UseListOrders.pop_back(); 3394*0b57cec5SDimitry Andric } 3395*0b57cec5SDimitry Andric Stream.ExitBlock(); 3396*0b57cec5SDimitry Andric } 3397*0b57cec5SDimitry Andric 3398*0b57cec5SDimitry Andric /// Emit a function body to the module stream. 3399*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeFunction( 3400*0b57cec5SDimitry Andric const Function &F, 3401*0b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> &FunctionToBitcodeIndex) { 3402*0b57cec5SDimitry Andric // Save the bitcode index of the start of this function block for recording 3403*0b57cec5SDimitry Andric // in the VST. 3404*0b57cec5SDimitry Andric FunctionToBitcodeIndex[&F] = Stream.GetCurrentBitNo(); 3405*0b57cec5SDimitry Andric 3406*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::FUNCTION_BLOCK_ID, 4); 3407*0b57cec5SDimitry Andric VE.incorporateFunction(F); 3408*0b57cec5SDimitry Andric 3409*0b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 3410*0b57cec5SDimitry Andric 3411*0b57cec5SDimitry Andric // Emit the number of basic blocks, so the reader can create them ahead of 3412*0b57cec5SDimitry Andric // time. 3413*0b57cec5SDimitry Andric Vals.push_back(VE.getBasicBlocks().size()); 3414*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_DECLAREBLOCKS, Vals); 3415*0b57cec5SDimitry Andric Vals.clear(); 3416*0b57cec5SDimitry Andric 3417*0b57cec5SDimitry Andric // If there are function-local constants, emit them now. 3418*0b57cec5SDimitry Andric unsigned CstStart, CstEnd; 3419*0b57cec5SDimitry Andric VE.getFunctionConstantRange(CstStart, CstEnd); 3420*0b57cec5SDimitry Andric writeConstants(CstStart, CstEnd, false); 3421*0b57cec5SDimitry Andric 3422*0b57cec5SDimitry Andric // If there is function-local metadata, emit it now. 3423*0b57cec5SDimitry Andric writeFunctionMetadata(F); 3424*0b57cec5SDimitry Andric 3425*0b57cec5SDimitry Andric // Keep a running idea of what the instruction ID is. 3426*0b57cec5SDimitry Andric unsigned InstID = CstEnd; 3427*0b57cec5SDimitry Andric 3428*0b57cec5SDimitry Andric bool NeedsMetadataAttachment = F.hasMetadata(); 3429*0b57cec5SDimitry Andric 3430*0b57cec5SDimitry Andric DILocation *LastDL = nullptr; 343181ad6265SDimitry Andric SmallSetVector<Function *, 4> BlockAddressUsers; 343281ad6265SDimitry Andric 3433*0b57cec5SDimitry Andric // Finally, emit all the instructions, in order. 343481ad6265SDimitry Andric for (const BasicBlock &BB : F) { 34354824e7fdSDimitry Andric for (const Instruction &I : BB) { 34364824e7fdSDimitry Andric writeInstruction(I, InstID, Vals); 3437*0b57cec5SDimitry Andric 34384824e7fdSDimitry Andric if (!I.getType()->isVoidTy()) 3439*0b57cec5SDimitry Andric ++InstID; 3440*0b57cec5SDimitry Andric 3441*0b57cec5SDimitry Andric // If the instruction has metadata, write a metadata attachment later. 34424824e7fdSDimitry Andric NeedsMetadataAttachment |= I.hasMetadataOtherThanDebugLoc(); 3443*0b57cec5SDimitry Andric 3444*0b57cec5SDimitry Andric // If the instruction has a debug location, emit it. 34454824e7fdSDimitry Andric DILocation *DL = I.getDebugLoc(); 3446*0b57cec5SDimitry Andric if (!DL) 3447*0b57cec5SDimitry Andric continue; 3448*0b57cec5SDimitry Andric 3449*0b57cec5SDimitry Andric if (DL == LastDL) { 3450*0b57cec5SDimitry Andric // Just repeat the same debug loc as last time. 3451*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_DEBUG_LOC_AGAIN, Vals); 3452*0b57cec5SDimitry Andric continue; 3453*0b57cec5SDimitry Andric } 3454*0b57cec5SDimitry Andric 3455*0b57cec5SDimitry Andric Vals.push_back(DL->getLine()); 3456*0b57cec5SDimitry Andric Vals.push_back(DL->getColumn()); 3457*0b57cec5SDimitry Andric Vals.push_back(VE.getMetadataOrNullID(DL->getScope())); 3458*0b57cec5SDimitry Andric Vals.push_back(VE.getMetadataOrNullID(DL->getInlinedAt())); 3459*0b57cec5SDimitry Andric Vals.push_back(DL->isImplicitCode()); 3460*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_DEBUG_LOC, Vals); 3461*0b57cec5SDimitry Andric Vals.clear(); 3462*0b57cec5SDimitry Andric 3463*0b57cec5SDimitry Andric LastDL = DL; 3464*0b57cec5SDimitry Andric } 3465*0b57cec5SDimitry Andric 346681ad6265SDimitry Andric if (BlockAddress *BA = BlockAddress::lookup(&BB)) { 346781ad6265SDimitry Andric SmallVector<Value *> Worklist{BA}; 346881ad6265SDimitry Andric SmallPtrSet<Value *, 8> Visited{BA}; 346981ad6265SDimitry Andric while (!Worklist.empty()) { 347081ad6265SDimitry Andric Value *V = Worklist.pop_back_val(); 347181ad6265SDimitry Andric for (User *U : V->users()) { 347281ad6265SDimitry Andric if (auto *I = dyn_cast<Instruction>(U)) { 347381ad6265SDimitry Andric Function *P = I->getFunction(); 347481ad6265SDimitry Andric if (P != &F) 347581ad6265SDimitry Andric BlockAddressUsers.insert(P); 347681ad6265SDimitry Andric } else if (isa<Constant>(U) && !isa<GlobalValue>(U) && 347781ad6265SDimitry Andric Visited.insert(U).second) 347881ad6265SDimitry Andric Worklist.push_back(U); 347981ad6265SDimitry Andric } 348081ad6265SDimitry Andric } 348181ad6265SDimitry Andric } 348281ad6265SDimitry Andric } 348381ad6265SDimitry Andric 348481ad6265SDimitry Andric if (!BlockAddressUsers.empty()) { 348581ad6265SDimitry Andric Vals.resize(BlockAddressUsers.size()); 348681ad6265SDimitry Andric for (auto I : llvm::enumerate(BlockAddressUsers)) 348781ad6265SDimitry Andric Vals[I.index()] = VE.getValueID(I.value()); 348881ad6265SDimitry Andric Stream.EmitRecord(bitc::FUNC_CODE_BLOCKADDR_USERS, Vals); 348981ad6265SDimitry Andric Vals.clear(); 349081ad6265SDimitry Andric } 349181ad6265SDimitry Andric 3492*0b57cec5SDimitry Andric // Emit names for all the instructions etc. 3493*0b57cec5SDimitry Andric if (auto *Symtab = F.getValueSymbolTable()) 3494*0b57cec5SDimitry Andric writeFunctionLevelValueSymbolTable(*Symtab); 3495*0b57cec5SDimitry Andric 3496*0b57cec5SDimitry Andric if (NeedsMetadataAttachment) 3497*0b57cec5SDimitry Andric writeFunctionMetadataAttachment(F); 3498*0b57cec5SDimitry Andric if (VE.shouldPreserveUseListOrder()) 3499*0b57cec5SDimitry Andric writeUseListBlock(&F); 3500*0b57cec5SDimitry Andric VE.purgeFunction(); 3501*0b57cec5SDimitry Andric Stream.ExitBlock(); 3502*0b57cec5SDimitry Andric } 3503*0b57cec5SDimitry Andric 3504*0b57cec5SDimitry Andric // Emit blockinfo, which defines the standard abbreviations etc. 3505*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeBlockInfo() { 3506*0b57cec5SDimitry Andric // We only want to emit block info records for blocks that have multiple 3507*0b57cec5SDimitry Andric // instances: CONSTANTS_BLOCK, FUNCTION_BLOCK and VALUE_SYMTAB_BLOCK. 3508*0b57cec5SDimitry Andric // Other blocks can define their abbrevs inline. 3509*0b57cec5SDimitry Andric Stream.EnterBlockInfoBlock(); 3510*0b57cec5SDimitry Andric 3511*0b57cec5SDimitry Andric { // 8-bit fixed-width VST_CODE_ENTRY/VST_CODE_BBENTRY strings. 3512*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3513*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 3)); 3514*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 3515*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 3516*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 3517*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, Abbv) != 3518*0b57cec5SDimitry Andric VST_ENTRY_8_ABBREV) 3519*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3520*0b57cec5SDimitry Andric } 3521*0b57cec5SDimitry Andric 3522*0b57cec5SDimitry Andric { // 7-bit fixed width VST_CODE_ENTRY strings. 3523*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3524*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_ENTRY)); 3525*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 3526*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 3527*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 3528*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, Abbv) != 3529*0b57cec5SDimitry Andric VST_ENTRY_7_ABBREV) 3530*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3531*0b57cec5SDimitry Andric } 3532*0b57cec5SDimitry Andric { // 6-bit char6 VST_CODE_ENTRY strings. 3533*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3534*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_ENTRY)); 3535*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 3536*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 3537*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 3538*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, Abbv) != 3539*0b57cec5SDimitry Andric VST_ENTRY_6_ABBREV) 3540*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3541*0b57cec5SDimitry Andric } 3542*0b57cec5SDimitry Andric { // 6-bit char6 VST_CODE_BBENTRY strings. 3543*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3544*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::VST_CODE_BBENTRY)); 3545*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 3546*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 3547*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 3548*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::VALUE_SYMTAB_BLOCK_ID, Abbv) != 3549*0b57cec5SDimitry Andric VST_BBENTRY_6_ABBREV) 3550*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3551*0b57cec5SDimitry Andric } 3552*0b57cec5SDimitry Andric 3553*0b57cec5SDimitry Andric { // SETTYPE abbrev for CONSTANTS_BLOCK. 3554*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3555*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_SETTYPE)); 3556*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 3557*0b57cec5SDimitry Andric VE.computeBitsRequiredForTypeIndicies())); 3558*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, Abbv) != 3559*0b57cec5SDimitry Andric CONSTANTS_SETTYPE_ABBREV) 3560*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3561*0b57cec5SDimitry Andric } 3562*0b57cec5SDimitry Andric 3563*0b57cec5SDimitry Andric { // INTEGER abbrev for CONSTANTS_BLOCK. 3564*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3565*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_INTEGER)); 3566*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 3567*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, Abbv) != 3568*0b57cec5SDimitry Andric CONSTANTS_INTEGER_ABBREV) 3569*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3570*0b57cec5SDimitry Andric } 3571*0b57cec5SDimitry Andric 3572*0b57cec5SDimitry Andric { // CE_CAST abbrev for CONSTANTS_BLOCK. 3573*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3574*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_CE_CAST)); 3575*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // cast opc 3576*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // typeid 3577*0b57cec5SDimitry Andric VE.computeBitsRequiredForTypeIndicies())); 3578*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // value id 3579*0b57cec5SDimitry Andric 3580*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, Abbv) != 3581*0b57cec5SDimitry Andric CONSTANTS_CE_CAST_Abbrev) 3582*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3583*0b57cec5SDimitry Andric } 3584*0b57cec5SDimitry Andric { // NULL abbrev for CONSTANTS_BLOCK. 3585*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3586*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::CST_CODE_NULL)); 3587*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::CONSTANTS_BLOCK_ID, Abbv) != 3588*0b57cec5SDimitry Andric CONSTANTS_NULL_Abbrev) 3589*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3590*0b57cec5SDimitry Andric } 3591*0b57cec5SDimitry Andric 3592*0b57cec5SDimitry Andric // FIXME: This should only use space for first class types! 3593*0b57cec5SDimitry Andric 3594*0b57cec5SDimitry Andric { // INST_LOAD abbrev for FUNCTION_BLOCK. 3595*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3596*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_LOAD)); 3597*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // Ptr 3598*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // dest ty 3599*0b57cec5SDimitry Andric VE.computeBitsRequiredForTypeIndicies())); 3600*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // Align 3601*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); // volatile 3602*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3603*0b57cec5SDimitry Andric FUNCTION_INST_LOAD_ABBREV) 3604*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3605*0b57cec5SDimitry Andric } 3606*0b57cec5SDimitry Andric { // INST_UNOP abbrev for FUNCTION_BLOCK. 3607*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3608*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_UNOP)); 3609*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 3610*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 3611*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3612*0b57cec5SDimitry Andric FUNCTION_INST_UNOP_ABBREV) 3613*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3614*0b57cec5SDimitry Andric } 3615*0b57cec5SDimitry Andric { // INST_UNOP_FLAGS abbrev for FUNCTION_BLOCK. 3616*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3617*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_UNOP)); 3618*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 3619*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 3620*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); // flags 3621*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3622*0b57cec5SDimitry Andric FUNCTION_INST_UNOP_FLAGS_ABBREV) 3623*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3624*0b57cec5SDimitry Andric } 3625*0b57cec5SDimitry Andric { // INST_BINOP abbrev for FUNCTION_BLOCK. 3626*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3627*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_BINOP)); 3628*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 3629*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // RHS 3630*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 3631*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3632*0b57cec5SDimitry Andric FUNCTION_INST_BINOP_ABBREV) 3633*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3634*0b57cec5SDimitry Andric } 3635*0b57cec5SDimitry Andric { // INST_BINOP_FLAGS abbrev for FUNCTION_BLOCK. 3636*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3637*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_BINOP)); 3638*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // LHS 3639*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // RHS 3640*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 3641*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); // flags 3642*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3643*0b57cec5SDimitry Andric FUNCTION_INST_BINOP_FLAGS_ABBREV) 3644*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3645*0b57cec5SDimitry Andric } 3646*0b57cec5SDimitry Andric { // INST_CAST abbrev for FUNCTION_BLOCK. 3647*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3648*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_CAST)); 3649*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // OpVal 3650*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // dest ty 3651*0b57cec5SDimitry Andric VE.computeBitsRequiredForTypeIndicies())); 3652*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 4)); // opc 3653*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3654*0b57cec5SDimitry Andric FUNCTION_INST_CAST_ABBREV) 3655*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3656*0b57cec5SDimitry Andric } 3657*0b57cec5SDimitry Andric 3658*0b57cec5SDimitry Andric { // INST_RET abbrev for FUNCTION_BLOCK. 3659*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3660*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_RET)); 3661*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3662*0b57cec5SDimitry Andric FUNCTION_INST_RET_VOID_ABBREV) 3663*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3664*0b57cec5SDimitry Andric } 3665*0b57cec5SDimitry Andric { // INST_RET abbrev for FUNCTION_BLOCK. 3666*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3667*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_RET)); 3668*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // ValID 3669*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3670*0b57cec5SDimitry Andric FUNCTION_INST_RET_VAL_ABBREV) 3671*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3672*0b57cec5SDimitry Andric } 3673*0b57cec5SDimitry Andric { // INST_UNREACHABLE abbrev for FUNCTION_BLOCK. 3674*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3675*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_UNREACHABLE)); 3676*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3677*0b57cec5SDimitry Andric FUNCTION_INST_UNREACHABLE_ABBREV) 3678*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3679*0b57cec5SDimitry Andric } 3680*0b57cec5SDimitry Andric { 3681*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3682*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FUNC_CODE_INST_GEP)); 3683*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 1)); 3684*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, // dest ty 3685*0b57cec5SDimitry Andric Log2_32_Ceil(VE.getTypes().size() + 1))); 3686*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 3687*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 3688*0b57cec5SDimitry Andric if (Stream.EmitBlockInfoAbbrev(bitc::FUNCTION_BLOCK_ID, Abbv) != 3689*0b57cec5SDimitry Andric FUNCTION_INST_GEP_ABBREV) 3690*0b57cec5SDimitry Andric llvm_unreachable("Unexpected abbrev ordering!"); 3691*0b57cec5SDimitry Andric } 3692*0b57cec5SDimitry Andric 3693*0b57cec5SDimitry Andric Stream.ExitBlock(); 3694*0b57cec5SDimitry Andric } 3695*0b57cec5SDimitry Andric 3696*0b57cec5SDimitry Andric /// Write the module path strings, currently only used when generating 3697*0b57cec5SDimitry Andric /// a combined index file. 3698*0b57cec5SDimitry Andric void IndexBitcodeWriter::writeModStrings() { 3699*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::MODULE_STRTAB_BLOCK_ID, 3); 3700*0b57cec5SDimitry Andric 3701*0b57cec5SDimitry Andric // TODO: See which abbrev sizes we actually need to emit 3702*0b57cec5SDimitry Andric 3703*0b57cec5SDimitry Andric // 8-bit fixed-width MST_ENTRY strings. 3704*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 3705*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MST_CODE_ENTRY)); 3706*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 3707*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 3708*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8)); 3709*0b57cec5SDimitry Andric unsigned Abbrev8Bit = Stream.EmitAbbrev(std::move(Abbv)); 3710*0b57cec5SDimitry Andric 3711*0b57cec5SDimitry Andric // 7-bit fixed width MST_ENTRY strings. 3712*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 3713*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MST_CODE_ENTRY)); 3714*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 3715*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 3716*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7)); 3717*0b57cec5SDimitry Andric unsigned Abbrev7Bit = Stream.EmitAbbrev(std::move(Abbv)); 3718*0b57cec5SDimitry Andric 3719*0b57cec5SDimitry Andric // 6-bit char6 MST_ENTRY strings. 3720*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 3721*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MST_CODE_ENTRY)); 3722*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 3723*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 3724*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 3725*0b57cec5SDimitry Andric unsigned Abbrev6Bit = Stream.EmitAbbrev(std::move(Abbv)); 3726*0b57cec5SDimitry Andric 3727*0b57cec5SDimitry Andric // Module Hash, 160 bits SHA1. Optionally, emitted after each MST_CODE_ENTRY. 3728*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 3729*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MST_CODE_HASH)); 3730*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 3731*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 3732*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 3733*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 3734*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32)); 3735*0b57cec5SDimitry Andric unsigned AbbrevHash = Stream.EmitAbbrev(std::move(Abbv)); 3736*0b57cec5SDimitry Andric 3737*0b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 3738*0b57cec5SDimitry Andric forEachModule( 3739*0b57cec5SDimitry Andric [&](const StringMapEntry<std::pair<uint64_t, ModuleHash>> &MPSE) { 3740*0b57cec5SDimitry Andric StringRef Key = MPSE.getKey(); 3741*0b57cec5SDimitry Andric const auto &Value = MPSE.getValue(); 3742*0b57cec5SDimitry Andric StringEncoding Bits = getStringEncoding(Key); 3743*0b57cec5SDimitry Andric unsigned AbbrevToUse = Abbrev8Bit; 3744*0b57cec5SDimitry Andric if (Bits == SE_Char6) 3745*0b57cec5SDimitry Andric AbbrevToUse = Abbrev6Bit; 3746*0b57cec5SDimitry Andric else if (Bits == SE_Fixed7) 3747*0b57cec5SDimitry Andric AbbrevToUse = Abbrev7Bit; 3748*0b57cec5SDimitry Andric 3749*0b57cec5SDimitry Andric Vals.push_back(Value.first); 3750*0b57cec5SDimitry Andric Vals.append(Key.begin(), Key.end()); 3751*0b57cec5SDimitry Andric 3752*0b57cec5SDimitry Andric // Emit the finished record. 3753*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MST_CODE_ENTRY, Vals, AbbrevToUse); 3754*0b57cec5SDimitry Andric 3755*0b57cec5SDimitry Andric // Emit an optional hash for the module now 3756*0b57cec5SDimitry Andric const auto &Hash = Value.second; 3757*0b57cec5SDimitry Andric if (llvm::any_of(Hash, [](uint32_t H) { return H; })) { 3758*0b57cec5SDimitry Andric Vals.assign(Hash.begin(), Hash.end()); 3759*0b57cec5SDimitry Andric // Emit the hash record. 3760*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MST_CODE_HASH, Vals, AbbrevHash); 3761*0b57cec5SDimitry Andric } 3762*0b57cec5SDimitry Andric 3763*0b57cec5SDimitry Andric Vals.clear(); 3764*0b57cec5SDimitry Andric }); 3765*0b57cec5SDimitry Andric Stream.ExitBlock(); 3766*0b57cec5SDimitry Andric } 3767*0b57cec5SDimitry Andric 3768*0b57cec5SDimitry Andric /// Write the function type metadata related records that need to appear before 3769*0b57cec5SDimitry Andric /// a function summary entry (whether per-module or combined). 3770e8d8bef9SDimitry Andric template <typename Fn> 3771*0b57cec5SDimitry Andric static void writeFunctionTypeMetadataRecords(BitstreamWriter &Stream, 3772e8d8bef9SDimitry Andric FunctionSummary *FS, 3773e8d8bef9SDimitry Andric Fn GetValueID) { 3774*0b57cec5SDimitry Andric if (!FS->type_tests().empty()) 3775*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_TYPE_TESTS, FS->type_tests()); 3776*0b57cec5SDimitry Andric 3777*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> Record; 3778*0b57cec5SDimitry Andric 3779*0b57cec5SDimitry Andric auto WriteVFuncIdVec = [&](uint64_t Ty, 3780*0b57cec5SDimitry Andric ArrayRef<FunctionSummary::VFuncId> VFs) { 3781*0b57cec5SDimitry Andric if (VFs.empty()) 3782*0b57cec5SDimitry Andric return; 3783*0b57cec5SDimitry Andric Record.clear(); 3784*0b57cec5SDimitry Andric for (auto &VF : VFs) { 3785*0b57cec5SDimitry Andric Record.push_back(VF.GUID); 3786*0b57cec5SDimitry Andric Record.push_back(VF.Offset); 3787*0b57cec5SDimitry Andric } 3788*0b57cec5SDimitry Andric Stream.EmitRecord(Ty, Record); 3789*0b57cec5SDimitry Andric }; 3790*0b57cec5SDimitry Andric 3791*0b57cec5SDimitry Andric WriteVFuncIdVec(bitc::FS_TYPE_TEST_ASSUME_VCALLS, 3792*0b57cec5SDimitry Andric FS->type_test_assume_vcalls()); 3793*0b57cec5SDimitry Andric WriteVFuncIdVec(bitc::FS_TYPE_CHECKED_LOAD_VCALLS, 3794*0b57cec5SDimitry Andric FS->type_checked_load_vcalls()); 3795*0b57cec5SDimitry Andric 3796*0b57cec5SDimitry Andric auto WriteConstVCallVec = [&](uint64_t Ty, 3797*0b57cec5SDimitry Andric ArrayRef<FunctionSummary::ConstVCall> VCs) { 3798*0b57cec5SDimitry Andric for (auto &VC : VCs) { 3799*0b57cec5SDimitry Andric Record.clear(); 3800*0b57cec5SDimitry Andric Record.push_back(VC.VFunc.GUID); 3801*0b57cec5SDimitry Andric Record.push_back(VC.VFunc.Offset); 3802e8d8bef9SDimitry Andric llvm::append_range(Record, VC.Args); 3803*0b57cec5SDimitry Andric Stream.EmitRecord(Ty, Record); 3804*0b57cec5SDimitry Andric } 3805*0b57cec5SDimitry Andric }; 3806*0b57cec5SDimitry Andric 3807*0b57cec5SDimitry Andric WriteConstVCallVec(bitc::FS_TYPE_TEST_ASSUME_CONST_VCALL, 3808*0b57cec5SDimitry Andric FS->type_test_assume_const_vcalls()); 3809*0b57cec5SDimitry Andric WriteConstVCallVec(bitc::FS_TYPE_CHECKED_LOAD_CONST_VCALL, 3810*0b57cec5SDimitry Andric FS->type_checked_load_const_vcalls()); 38115ffd83dbSDimitry Andric 38125ffd83dbSDimitry Andric auto WriteRange = [&](ConstantRange Range) { 38135ffd83dbSDimitry Andric Range = Range.sextOrTrunc(FunctionSummary::ParamAccess::RangeWidth); 38145ffd83dbSDimitry Andric assert(Range.getLower().getNumWords() == 1); 38155ffd83dbSDimitry Andric assert(Range.getUpper().getNumWords() == 1); 38165ffd83dbSDimitry Andric emitSignedInt64(Record, *Range.getLower().getRawData()); 38175ffd83dbSDimitry Andric emitSignedInt64(Record, *Range.getUpper().getRawData()); 38185ffd83dbSDimitry Andric }; 38195ffd83dbSDimitry Andric 38205ffd83dbSDimitry Andric if (!FS->paramAccesses().empty()) { 38215ffd83dbSDimitry Andric Record.clear(); 38225ffd83dbSDimitry Andric for (auto &Arg : FS->paramAccesses()) { 3823e8d8bef9SDimitry Andric size_t UndoSize = Record.size(); 38245ffd83dbSDimitry Andric Record.push_back(Arg.ParamNo); 38255ffd83dbSDimitry Andric WriteRange(Arg.Use); 38265ffd83dbSDimitry Andric Record.push_back(Arg.Calls.size()); 38275ffd83dbSDimitry Andric for (auto &Call : Arg.Calls) { 38285ffd83dbSDimitry Andric Record.push_back(Call.ParamNo); 3829bdd1243dSDimitry Andric std::optional<unsigned> ValueID = GetValueID(Call.Callee); 3830e8d8bef9SDimitry Andric if (!ValueID) { 3831e8d8bef9SDimitry Andric // If ValueID is unknown we can't drop just this call, we must drop 3832e8d8bef9SDimitry Andric // entire parameter. 3833e8d8bef9SDimitry Andric Record.resize(UndoSize); 3834e8d8bef9SDimitry Andric break; 3835e8d8bef9SDimitry Andric } 3836e8d8bef9SDimitry Andric Record.push_back(*ValueID); 38375ffd83dbSDimitry Andric WriteRange(Call.Offsets); 38385ffd83dbSDimitry Andric } 38395ffd83dbSDimitry Andric } 3840e8d8bef9SDimitry Andric if (!Record.empty()) 38415ffd83dbSDimitry Andric Stream.EmitRecord(bitc::FS_PARAM_ACCESS, Record); 38425ffd83dbSDimitry Andric } 3843*0b57cec5SDimitry Andric } 3844*0b57cec5SDimitry Andric 3845*0b57cec5SDimitry Andric /// Collect type IDs from type tests used by function. 3846*0b57cec5SDimitry Andric static void 3847*0b57cec5SDimitry Andric getReferencedTypeIds(FunctionSummary *FS, 3848*0b57cec5SDimitry Andric std::set<GlobalValue::GUID> &ReferencedTypeIds) { 3849*0b57cec5SDimitry Andric if (!FS->type_tests().empty()) 3850*0b57cec5SDimitry Andric for (auto &TT : FS->type_tests()) 3851*0b57cec5SDimitry Andric ReferencedTypeIds.insert(TT); 3852*0b57cec5SDimitry Andric 3853*0b57cec5SDimitry Andric auto GetReferencedTypesFromVFuncIdVec = 3854*0b57cec5SDimitry Andric [&](ArrayRef<FunctionSummary::VFuncId> VFs) { 3855*0b57cec5SDimitry Andric for (auto &VF : VFs) 3856*0b57cec5SDimitry Andric ReferencedTypeIds.insert(VF.GUID); 3857*0b57cec5SDimitry Andric }; 3858*0b57cec5SDimitry Andric 3859*0b57cec5SDimitry Andric GetReferencedTypesFromVFuncIdVec(FS->type_test_assume_vcalls()); 3860*0b57cec5SDimitry Andric GetReferencedTypesFromVFuncIdVec(FS->type_checked_load_vcalls()); 3861*0b57cec5SDimitry Andric 3862*0b57cec5SDimitry Andric auto GetReferencedTypesFromConstVCallVec = 3863*0b57cec5SDimitry Andric [&](ArrayRef<FunctionSummary::ConstVCall> VCs) { 3864*0b57cec5SDimitry Andric for (auto &VC : VCs) 3865*0b57cec5SDimitry Andric ReferencedTypeIds.insert(VC.VFunc.GUID); 3866*0b57cec5SDimitry Andric }; 3867*0b57cec5SDimitry Andric 3868*0b57cec5SDimitry Andric GetReferencedTypesFromConstVCallVec(FS->type_test_assume_const_vcalls()); 3869*0b57cec5SDimitry Andric GetReferencedTypesFromConstVCallVec(FS->type_checked_load_const_vcalls()); 3870*0b57cec5SDimitry Andric } 3871*0b57cec5SDimitry Andric 3872*0b57cec5SDimitry Andric static void writeWholeProgramDevirtResolutionByArg( 3873*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, const std::vector<uint64_t> &args, 3874*0b57cec5SDimitry Andric const WholeProgramDevirtResolution::ByArg &ByArg) { 3875*0b57cec5SDimitry Andric NameVals.push_back(args.size()); 3876e8d8bef9SDimitry Andric llvm::append_range(NameVals, args); 3877*0b57cec5SDimitry Andric 3878*0b57cec5SDimitry Andric NameVals.push_back(ByArg.TheKind); 3879*0b57cec5SDimitry Andric NameVals.push_back(ByArg.Info); 3880*0b57cec5SDimitry Andric NameVals.push_back(ByArg.Byte); 3881*0b57cec5SDimitry Andric NameVals.push_back(ByArg.Bit); 3882*0b57cec5SDimitry Andric } 3883*0b57cec5SDimitry Andric 3884*0b57cec5SDimitry Andric static void writeWholeProgramDevirtResolution( 3885*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, StringTableBuilder &StrtabBuilder, 3886*0b57cec5SDimitry Andric uint64_t Id, const WholeProgramDevirtResolution &Wpd) { 3887*0b57cec5SDimitry Andric NameVals.push_back(Id); 3888*0b57cec5SDimitry Andric 3889*0b57cec5SDimitry Andric NameVals.push_back(Wpd.TheKind); 3890*0b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(Wpd.SingleImplName)); 3891*0b57cec5SDimitry Andric NameVals.push_back(Wpd.SingleImplName.size()); 3892*0b57cec5SDimitry Andric 3893*0b57cec5SDimitry Andric NameVals.push_back(Wpd.ResByArg.size()); 3894*0b57cec5SDimitry Andric for (auto &A : Wpd.ResByArg) 3895*0b57cec5SDimitry Andric writeWholeProgramDevirtResolutionByArg(NameVals, A.first, A.second); 3896*0b57cec5SDimitry Andric } 3897*0b57cec5SDimitry Andric 3898*0b57cec5SDimitry Andric static void writeTypeIdSummaryRecord(SmallVector<uint64_t, 64> &NameVals, 3899*0b57cec5SDimitry Andric StringTableBuilder &StrtabBuilder, 3900*0b57cec5SDimitry Andric const std::string &Id, 3901*0b57cec5SDimitry Andric const TypeIdSummary &Summary) { 3902*0b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(Id)); 3903*0b57cec5SDimitry Andric NameVals.push_back(Id.size()); 3904*0b57cec5SDimitry Andric 3905*0b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.TheKind); 3906*0b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.SizeM1BitWidth); 3907*0b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.AlignLog2); 3908*0b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.SizeM1); 3909*0b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.BitMask); 3910*0b57cec5SDimitry Andric NameVals.push_back(Summary.TTRes.InlineBits); 3911*0b57cec5SDimitry Andric 3912*0b57cec5SDimitry Andric for (auto &W : Summary.WPDRes) 3913*0b57cec5SDimitry Andric writeWholeProgramDevirtResolution(NameVals, StrtabBuilder, W.first, 3914*0b57cec5SDimitry Andric W.second); 3915*0b57cec5SDimitry Andric } 3916*0b57cec5SDimitry Andric 3917*0b57cec5SDimitry Andric static void writeTypeIdCompatibleVtableSummaryRecord( 3918*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, StringTableBuilder &StrtabBuilder, 3919*0b57cec5SDimitry Andric const std::string &Id, const TypeIdCompatibleVtableInfo &Summary, 3920*0b57cec5SDimitry Andric ValueEnumerator &VE) { 3921*0b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(Id)); 3922*0b57cec5SDimitry Andric NameVals.push_back(Id.size()); 3923*0b57cec5SDimitry Andric 3924*0b57cec5SDimitry Andric for (auto &P : Summary) { 3925*0b57cec5SDimitry Andric NameVals.push_back(P.AddressPointOffset); 3926*0b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(P.VTableVI.getValue())); 3927*0b57cec5SDimitry Andric } 3928*0b57cec5SDimitry Andric } 3929*0b57cec5SDimitry Andric 3930bdd1243dSDimitry Andric static void writeFunctionHeapProfileRecords( 3931bdd1243dSDimitry Andric BitstreamWriter &Stream, FunctionSummary *FS, unsigned CallsiteAbbrev, 3932bdd1243dSDimitry Andric unsigned AllocAbbrev, bool PerModule, 3933bdd1243dSDimitry Andric std::function<unsigned(const ValueInfo &VI)> GetValueID, 3934bdd1243dSDimitry Andric std::function<unsigned(unsigned)> GetStackIndex) { 3935bdd1243dSDimitry Andric SmallVector<uint64_t> Record; 3936bdd1243dSDimitry Andric 3937bdd1243dSDimitry Andric for (auto &CI : FS->callsites()) { 3938bdd1243dSDimitry Andric Record.clear(); 3939bdd1243dSDimitry Andric // Per module callsite clones should always have a single entry of 3940bdd1243dSDimitry Andric // value 0. 3941bdd1243dSDimitry Andric assert(!PerModule || (CI.Clones.size() == 1 && CI.Clones[0] == 0)); 3942bdd1243dSDimitry Andric Record.push_back(GetValueID(CI.Callee)); 3943bdd1243dSDimitry Andric if (!PerModule) { 3944bdd1243dSDimitry Andric Record.push_back(CI.StackIdIndices.size()); 3945bdd1243dSDimitry Andric Record.push_back(CI.Clones.size()); 3946bdd1243dSDimitry Andric } 3947bdd1243dSDimitry Andric for (auto Id : CI.StackIdIndices) 3948bdd1243dSDimitry Andric Record.push_back(GetStackIndex(Id)); 3949bdd1243dSDimitry Andric if (!PerModule) { 3950bdd1243dSDimitry Andric for (auto V : CI.Clones) 3951bdd1243dSDimitry Andric Record.push_back(V); 3952bdd1243dSDimitry Andric } 3953bdd1243dSDimitry Andric Stream.EmitRecord(PerModule ? bitc::FS_PERMODULE_CALLSITE_INFO 3954bdd1243dSDimitry Andric : bitc::FS_COMBINED_CALLSITE_INFO, 3955bdd1243dSDimitry Andric Record, CallsiteAbbrev); 3956bdd1243dSDimitry Andric } 3957bdd1243dSDimitry Andric 3958bdd1243dSDimitry Andric for (auto &AI : FS->allocs()) { 3959bdd1243dSDimitry Andric Record.clear(); 3960bdd1243dSDimitry Andric // Per module alloc versions should always have a single entry of 3961bdd1243dSDimitry Andric // value 0. 3962bdd1243dSDimitry Andric assert(!PerModule || (AI.Versions.size() == 1 && AI.Versions[0] == 0)); 3963bdd1243dSDimitry Andric if (!PerModule) { 3964bdd1243dSDimitry Andric Record.push_back(AI.MIBs.size()); 3965bdd1243dSDimitry Andric Record.push_back(AI.Versions.size()); 3966bdd1243dSDimitry Andric } 3967bdd1243dSDimitry Andric for (auto &MIB : AI.MIBs) { 3968bdd1243dSDimitry Andric Record.push_back((uint8_t)MIB.AllocType); 3969bdd1243dSDimitry Andric Record.push_back(MIB.StackIdIndices.size()); 3970bdd1243dSDimitry Andric for (auto Id : MIB.StackIdIndices) 3971bdd1243dSDimitry Andric Record.push_back(GetStackIndex(Id)); 3972bdd1243dSDimitry Andric } 3973bdd1243dSDimitry Andric if (!PerModule) { 3974bdd1243dSDimitry Andric for (auto V : AI.Versions) 3975bdd1243dSDimitry Andric Record.push_back(V); 3976bdd1243dSDimitry Andric } 3977bdd1243dSDimitry Andric Stream.EmitRecord(PerModule ? bitc::FS_PERMODULE_ALLOC_INFO 3978bdd1243dSDimitry Andric : bitc::FS_COMBINED_ALLOC_INFO, 3979bdd1243dSDimitry Andric Record, AllocAbbrev); 3980bdd1243dSDimitry Andric } 3981bdd1243dSDimitry Andric } 3982bdd1243dSDimitry Andric 3983*0b57cec5SDimitry Andric // Helper to emit a single function summary record. 3984*0b57cec5SDimitry Andric void ModuleBitcodeWriterBase::writePerModuleFunctionSummaryRecord( 3985*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> &NameVals, GlobalValueSummary *Summary, 3986*0b57cec5SDimitry Andric unsigned ValueID, unsigned FSCallsAbbrev, unsigned FSCallsProfileAbbrev, 3987bdd1243dSDimitry Andric unsigned CallsiteAbbrev, unsigned AllocAbbrev, const Function &F) { 3988*0b57cec5SDimitry Andric NameVals.push_back(ValueID); 3989*0b57cec5SDimitry Andric 3990*0b57cec5SDimitry Andric FunctionSummary *FS = cast<FunctionSummary>(Summary); 3991e8d8bef9SDimitry Andric 3992e8d8bef9SDimitry Andric writeFunctionTypeMetadataRecords( 3993bdd1243dSDimitry Andric Stream, FS, [&](const ValueInfo &VI) -> std::optional<unsigned> { 3994e8d8bef9SDimitry Andric return {VE.getValueID(VI.getValue())}; 3995e8d8bef9SDimitry Andric }); 3996*0b57cec5SDimitry Andric 3997bdd1243dSDimitry Andric writeFunctionHeapProfileRecords( 3998bdd1243dSDimitry Andric Stream, FS, CallsiteAbbrev, AllocAbbrev, 3999bdd1243dSDimitry Andric /*PerModule*/ true, 4000bdd1243dSDimitry Andric /*GetValueId*/ [&](const ValueInfo &VI) { return getValueId(VI); }, 4001bdd1243dSDimitry Andric /*GetStackIndex*/ [&](unsigned I) { return I; }); 4002bdd1243dSDimitry Andric 4003*0b57cec5SDimitry Andric auto SpecialRefCnts = FS->specialRefCounts(); 4004*0b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(FS->flags())); 4005*0b57cec5SDimitry Andric NameVals.push_back(FS->instCount()); 4006*0b57cec5SDimitry Andric NameVals.push_back(getEncodedFFlags(FS->fflags())); 4007*0b57cec5SDimitry Andric NameVals.push_back(FS->refs().size()); 4008*0b57cec5SDimitry Andric NameVals.push_back(SpecialRefCnts.first); // rorefcnt 4009*0b57cec5SDimitry Andric NameVals.push_back(SpecialRefCnts.second); // worefcnt 4010*0b57cec5SDimitry Andric 4011*0b57cec5SDimitry Andric for (auto &RI : FS->refs()) 4012*0b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(RI.getValue())); 4013*0b57cec5SDimitry Andric 4014*0b57cec5SDimitry Andric bool HasProfileData = 4015*0b57cec5SDimitry Andric F.hasProfileData() || ForceSummaryEdgesCold != FunctionSummary::FSHT_None; 4016*0b57cec5SDimitry Andric for (auto &ECI : FS->calls()) { 4017*0b57cec5SDimitry Andric NameVals.push_back(getValueId(ECI.first)); 4018*0b57cec5SDimitry Andric if (HasProfileData) 4019*0b57cec5SDimitry Andric NameVals.push_back(static_cast<uint8_t>(ECI.second.Hotness)); 4020*0b57cec5SDimitry Andric else if (WriteRelBFToSummary) 4021*0b57cec5SDimitry Andric NameVals.push_back(ECI.second.RelBlockFreq); 4022*0b57cec5SDimitry Andric } 4023*0b57cec5SDimitry Andric 4024*0b57cec5SDimitry Andric unsigned FSAbbrev = (HasProfileData ? FSCallsProfileAbbrev : FSCallsAbbrev); 4025*0b57cec5SDimitry Andric unsigned Code = 4026*0b57cec5SDimitry Andric (HasProfileData ? bitc::FS_PERMODULE_PROFILE 4027*0b57cec5SDimitry Andric : (WriteRelBFToSummary ? bitc::FS_PERMODULE_RELBF 4028*0b57cec5SDimitry Andric : bitc::FS_PERMODULE)); 4029*0b57cec5SDimitry Andric 4030*0b57cec5SDimitry Andric // Emit the finished record. 4031*0b57cec5SDimitry Andric Stream.EmitRecord(Code, NameVals, FSAbbrev); 4032*0b57cec5SDimitry Andric NameVals.clear(); 4033*0b57cec5SDimitry Andric } 4034*0b57cec5SDimitry Andric 4035*0b57cec5SDimitry Andric // Collect the global value references in the given variable's initializer, 4036*0b57cec5SDimitry Andric // and emit them in a summary record. 4037*0b57cec5SDimitry Andric void ModuleBitcodeWriterBase::writeModuleLevelReferences( 4038*0b57cec5SDimitry Andric const GlobalVariable &V, SmallVector<uint64_t, 64> &NameVals, 4039*0b57cec5SDimitry Andric unsigned FSModRefsAbbrev, unsigned FSModVTableRefsAbbrev) { 4040*0b57cec5SDimitry Andric auto VI = Index->getValueInfo(V.getGUID()); 4041*0b57cec5SDimitry Andric if (!VI || VI.getSummaryList().empty()) { 4042*0b57cec5SDimitry Andric // Only declarations should not have a summary (a declaration might however 4043*0b57cec5SDimitry Andric // have a summary if the def was in module level asm). 4044*0b57cec5SDimitry Andric assert(V.isDeclaration()); 4045*0b57cec5SDimitry Andric return; 4046*0b57cec5SDimitry Andric } 4047*0b57cec5SDimitry Andric auto *Summary = VI.getSummaryList()[0].get(); 4048*0b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(&V)); 4049*0b57cec5SDimitry Andric GlobalVarSummary *VS = cast<GlobalVarSummary>(Summary); 4050*0b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(VS->flags())); 4051*0b57cec5SDimitry Andric NameVals.push_back(getEncodedGVarFlags(VS->varflags())); 4052*0b57cec5SDimitry Andric 4053*0b57cec5SDimitry Andric auto VTableFuncs = VS->vTableFuncs(); 4054*0b57cec5SDimitry Andric if (!VTableFuncs.empty()) 4055*0b57cec5SDimitry Andric NameVals.push_back(VS->refs().size()); 4056*0b57cec5SDimitry Andric 4057*0b57cec5SDimitry Andric unsigned SizeBeforeRefs = NameVals.size(); 4058*0b57cec5SDimitry Andric for (auto &RI : VS->refs()) 4059*0b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(RI.getValue())); 4060*0b57cec5SDimitry Andric // Sort the refs for determinism output, the vector returned by FS->refs() has 4061*0b57cec5SDimitry Andric // been initialized from a DenseSet. 4062e8d8bef9SDimitry Andric llvm::sort(drop_begin(NameVals, SizeBeforeRefs)); 4063*0b57cec5SDimitry Andric 4064*0b57cec5SDimitry Andric if (VTableFuncs.empty()) 4065*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_PERMODULE_GLOBALVAR_INIT_REFS, NameVals, 4066*0b57cec5SDimitry Andric FSModRefsAbbrev); 4067*0b57cec5SDimitry Andric else { 4068*0b57cec5SDimitry Andric // VTableFuncs pairs should already be sorted by offset. 4069*0b57cec5SDimitry Andric for (auto &P : VTableFuncs) { 4070*0b57cec5SDimitry Andric NameVals.push_back(VE.getValueID(P.FuncVI.getValue())); 4071*0b57cec5SDimitry Andric NameVals.push_back(P.VTableOffset); 4072*0b57cec5SDimitry Andric } 4073*0b57cec5SDimitry Andric 4074*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_PERMODULE_VTABLE_GLOBALVAR_INIT_REFS, NameVals, 4075*0b57cec5SDimitry Andric FSModVTableRefsAbbrev); 4076*0b57cec5SDimitry Andric } 4077*0b57cec5SDimitry Andric NameVals.clear(); 4078*0b57cec5SDimitry Andric } 4079*0b57cec5SDimitry Andric 4080*0b57cec5SDimitry Andric /// Emit the per-module summary section alongside the rest of 4081*0b57cec5SDimitry Andric /// the module's bitcode. 4082*0b57cec5SDimitry Andric void ModuleBitcodeWriterBase::writePerModuleGlobalValueSummary() { 4083*0b57cec5SDimitry Andric // By default we compile with ThinLTO if the module has a summary, but the 4084*0b57cec5SDimitry Andric // client can request full LTO with a module flag. 4085*0b57cec5SDimitry Andric bool IsThinLTO = true; 4086*0b57cec5SDimitry Andric if (auto *MD = 4087*0b57cec5SDimitry Andric mdconst::extract_or_null<ConstantInt>(M.getModuleFlag("ThinLTO"))) 4088*0b57cec5SDimitry Andric IsThinLTO = MD->getZExtValue(); 4089*0b57cec5SDimitry Andric Stream.EnterSubblock(IsThinLTO ? bitc::GLOBALVAL_SUMMARY_BLOCK_ID 4090*0b57cec5SDimitry Andric : bitc::FULL_LTO_GLOBALVAL_SUMMARY_BLOCK_ID, 4091*0b57cec5SDimitry Andric 4); 4092*0b57cec5SDimitry Andric 4093480093f4SDimitry Andric Stream.EmitRecord( 4094480093f4SDimitry Andric bitc::FS_VERSION, 4095480093f4SDimitry Andric ArrayRef<uint64_t>{ModuleSummaryIndex::BitcodeSummaryVersion}); 4096*0b57cec5SDimitry Andric 4097*0b57cec5SDimitry Andric // Write the index flags. 4098*0b57cec5SDimitry Andric uint64_t Flags = 0; 4099*0b57cec5SDimitry Andric // Bits 1-3 are set only in the combined index, skip them. 4100*0b57cec5SDimitry Andric if (Index->enableSplitLTOUnit()) 4101*0b57cec5SDimitry Andric Flags |= 0x8; 4102*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_FLAGS, ArrayRef<uint64_t>{Flags}); 4103*0b57cec5SDimitry Andric 4104*0b57cec5SDimitry Andric if (Index->begin() == Index->end()) { 4105*0b57cec5SDimitry Andric Stream.ExitBlock(); 4106*0b57cec5SDimitry Andric return; 4107*0b57cec5SDimitry Andric } 4108*0b57cec5SDimitry Andric 4109*0b57cec5SDimitry Andric for (const auto &GVI : valueIds()) { 4110*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_VALUE_GUID, 4111*0b57cec5SDimitry Andric ArrayRef<uint64_t>{GVI.second, GVI.first}); 4112*0b57cec5SDimitry Andric } 4113*0b57cec5SDimitry Andric 4114bdd1243dSDimitry Andric if (!Index->stackIds().empty()) { 4115bdd1243dSDimitry Andric auto StackIdAbbv = std::make_shared<BitCodeAbbrev>(); 4116bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(bitc::FS_STACK_IDS)); 4117bdd1243dSDimitry Andric // numids x stackid 4118bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4119bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4120bdd1243dSDimitry Andric unsigned StackIdAbbvId = Stream.EmitAbbrev(std::move(StackIdAbbv)); 4121bdd1243dSDimitry Andric Stream.EmitRecord(bitc::FS_STACK_IDS, Index->stackIds(), StackIdAbbvId); 4122bdd1243dSDimitry Andric } 4123bdd1243dSDimitry Andric 4124*0b57cec5SDimitry Andric // Abbrev for FS_PERMODULE_PROFILE. 4125*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 4126*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_PROFILE)); 4127*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4128*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 4129*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // instcount 4130*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // fflags 4131*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numrefs 4132*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // rorefcnt 4133*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // worefcnt 4134*0b57cec5SDimitry Andric // numrefs x valueid, n x (valueid, hotness) 4135*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4136*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4137*0b57cec5SDimitry Andric unsigned FSCallsProfileAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4138*0b57cec5SDimitry Andric 4139*0b57cec5SDimitry Andric // Abbrev for FS_PERMODULE or FS_PERMODULE_RELBF. 4140*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4141*0b57cec5SDimitry Andric if (WriteRelBFToSummary) 4142*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_RELBF)); 4143*0b57cec5SDimitry Andric else 4144*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE)); 4145*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4146*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 4147*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // instcount 4148*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // fflags 4149*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numrefs 4150*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // rorefcnt 4151*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // worefcnt 4152*0b57cec5SDimitry Andric // numrefs x valueid, n x (valueid [, rel_block_freq]) 4153*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4154*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4155*0b57cec5SDimitry Andric unsigned FSCallsAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4156*0b57cec5SDimitry Andric 4157*0b57cec5SDimitry Andric // Abbrev for FS_PERMODULE_GLOBALVAR_INIT_REFS. 4158*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4159*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_GLOBALVAR_INIT_REFS)); 4160*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4161*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 4162*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); // valueids 4163*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4164*0b57cec5SDimitry Andric unsigned FSModRefsAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4165*0b57cec5SDimitry Andric 4166*0b57cec5SDimitry Andric // Abbrev for FS_PERMODULE_VTABLE_GLOBALVAR_INIT_REFS. 4167*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4168*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_VTABLE_GLOBALVAR_INIT_REFS)); 4169*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4170*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 4171*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numrefs 4172*0b57cec5SDimitry Andric // numrefs x valueid, n x (valueid , offset) 4173*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4174*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4175*0b57cec5SDimitry Andric unsigned FSModVTableRefsAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4176*0b57cec5SDimitry Andric 4177*0b57cec5SDimitry Andric // Abbrev for FS_ALIAS. 4178*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4179*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_ALIAS)); 4180*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4181*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 4182*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4183*0b57cec5SDimitry Andric unsigned FSAliasAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4184*0b57cec5SDimitry Andric 4185*0b57cec5SDimitry Andric // Abbrev for FS_TYPE_ID_METADATA 4186*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4187*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_TYPE_ID_METADATA)); 4188*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // typeid strtab index 4189*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // typeid length 4190*0b57cec5SDimitry Andric // n x (valueid , offset) 4191*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4192*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4193*0b57cec5SDimitry Andric unsigned TypeIdCompatibleVtableAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4194*0b57cec5SDimitry Andric 4195bdd1243dSDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4196bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_CALLSITE_INFO)); 4197bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4198bdd1243dSDimitry Andric // n x stackidindex 4199bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4200bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4201bdd1243dSDimitry Andric unsigned CallsiteAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4202bdd1243dSDimitry Andric 4203bdd1243dSDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4204bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_PERMODULE_ALLOC_INFO)); 4205bdd1243dSDimitry Andric // n x (alloc type, numstackids, numstackids x stackidindex) 4206bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4207bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4208bdd1243dSDimitry Andric unsigned AllocAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4209bdd1243dSDimitry Andric 4210*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> NameVals; 4211*0b57cec5SDimitry Andric // Iterate over the list of functions instead of the Index to 4212*0b57cec5SDimitry Andric // ensure the ordering is stable. 4213*0b57cec5SDimitry Andric for (const Function &F : M) { 4214*0b57cec5SDimitry Andric // Summary emission does not support anonymous functions, they have to 4215*0b57cec5SDimitry Andric // renamed using the anonymous function renaming pass. 4216*0b57cec5SDimitry Andric if (!F.hasName()) 4217*0b57cec5SDimitry Andric report_fatal_error("Unexpected anonymous function when writing summary"); 4218*0b57cec5SDimitry Andric 4219*0b57cec5SDimitry Andric ValueInfo VI = Index->getValueInfo(F.getGUID()); 4220*0b57cec5SDimitry Andric if (!VI || VI.getSummaryList().empty()) { 4221*0b57cec5SDimitry Andric // Only declarations should not have a summary (a declaration might 4222*0b57cec5SDimitry Andric // however have a summary if the def was in module level asm). 4223*0b57cec5SDimitry Andric assert(F.isDeclaration()); 4224*0b57cec5SDimitry Andric continue; 4225*0b57cec5SDimitry Andric } 4226*0b57cec5SDimitry Andric auto *Summary = VI.getSummaryList()[0].get(); 4227*0b57cec5SDimitry Andric writePerModuleFunctionSummaryRecord(NameVals, Summary, VE.getValueID(&F), 4228bdd1243dSDimitry Andric FSCallsAbbrev, FSCallsProfileAbbrev, 4229bdd1243dSDimitry Andric CallsiteAbbrev, AllocAbbrev, F); 4230*0b57cec5SDimitry Andric } 4231*0b57cec5SDimitry Andric 4232*0b57cec5SDimitry Andric // Capture references from GlobalVariable initializers, which are outside 4233*0b57cec5SDimitry Andric // of a function scope. 4234*0b57cec5SDimitry Andric for (const GlobalVariable &G : M.globals()) 4235*0b57cec5SDimitry Andric writeModuleLevelReferences(G, NameVals, FSModRefsAbbrev, 4236*0b57cec5SDimitry Andric FSModVTableRefsAbbrev); 4237*0b57cec5SDimitry Andric 4238*0b57cec5SDimitry Andric for (const GlobalAlias &A : M.aliases()) { 4239349cc55cSDimitry Andric auto *Aliasee = A.getAliaseeObject(); 4240fcaf7f86SDimitry Andric // Skip ifunc and nameless functions which don't have an entry in the 4241fcaf7f86SDimitry Andric // summary. 4242fcaf7f86SDimitry Andric if (!Aliasee->hasName() || isa<GlobalIFunc>(Aliasee)) 4243*0b57cec5SDimitry Andric continue; 4244*0b57cec5SDimitry Andric auto AliasId = VE.getValueID(&A); 4245*0b57cec5SDimitry Andric auto AliaseeId = VE.getValueID(Aliasee); 4246*0b57cec5SDimitry Andric NameVals.push_back(AliasId); 4247*0b57cec5SDimitry Andric auto *Summary = Index->getGlobalValueSummary(A); 4248*0b57cec5SDimitry Andric AliasSummary *AS = cast<AliasSummary>(Summary); 4249*0b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(AS->flags())); 4250*0b57cec5SDimitry Andric NameVals.push_back(AliaseeId); 4251*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_ALIAS, NameVals, FSAliasAbbrev); 4252*0b57cec5SDimitry Andric NameVals.clear(); 4253*0b57cec5SDimitry Andric } 4254*0b57cec5SDimitry Andric 4255*0b57cec5SDimitry Andric for (auto &S : Index->typeIdCompatibleVtableMap()) { 4256*0b57cec5SDimitry Andric writeTypeIdCompatibleVtableSummaryRecord(NameVals, StrtabBuilder, S.first, 4257*0b57cec5SDimitry Andric S.second, VE); 4258*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_TYPE_ID_METADATA, NameVals, 4259*0b57cec5SDimitry Andric TypeIdCompatibleVtableAbbrev); 4260*0b57cec5SDimitry Andric NameVals.clear(); 4261*0b57cec5SDimitry Andric } 4262*0b57cec5SDimitry Andric 42635ffd83dbSDimitry Andric Stream.EmitRecord(bitc::FS_BLOCK_COUNT, 42645ffd83dbSDimitry Andric ArrayRef<uint64_t>{Index->getBlockCount()}); 42655ffd83dbSDimitry Andric 4266*0b57cec5SDimitry Andric Stream.ExitBlock(); 4267*0b57cec5SDimitry Andric } 4268*0b57cec5SDimitry Andric 4269*0b57cec5SDimitry Andric /// Emit the combined summary section into the combined index file. 4270*0b57cec5SDimitry Andric void IndexBitcodeWriter::writeCombinedGlobalValueSummary() { 4271bdd1243dSDimitry Andric Stream.EnterSubblock(bitc::GLOBALVAL_SUMMARY_BLOCK_ID, 4); 4272480093f4SDimitry Andric Stream.EmitRecord( 4273480093f4SDimitry Andric bitc::FS_VERSION, 4274480093f4SDimitry Andric ArrayRef<uint64_t>{ModuleSummaryIndex::BitcodeSummaryVersion}); 4275*0b57cec5SDimitry Andric 4276*0b57cec5SDimitry Andric // Write the index flags. 42775ffd83dbSDimitry Andric Stream.EmitRecord(bitc::FS_FLAGS, ArrayRef<uint64_t>{Index.getFlags()}); 4278*0b57cec5SDimitry Andric 4279*0b57cec5SDimitry Andric for (const auto &GVI : valueIds()) { 4280*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_VALUE_GUID, 4281*0b57cec5SDimitry Andric ArrayRef<uint64_t>{GVI.second, GVI.first}); 4282*0b57cec5SDimitry Andric } 4283*0b57cec5SDimitry Andric 4284bdd1243dSDimitry Andric if (!StackIdIndices.empty()) { 4285bdd1243dSDimitry Andric auto StackIdAbbv = std::make_shared<BitCodeAbbrev>(); 4286bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(bitc::FS_STACK_IDS)); 4287bdd1243dSDimitry Andric // numids x stackid 4288bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4289bdd1243dSDimitry Andric StackIdAbbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4290bdd1243dSDimitry Andric unsigned StackIdAbbvId = Stream.EmitAbbrev(std::move(StackIdAbbv)); 4291bdd1243dSDimitry Andric // Write the stack ids used by this index, which will be a subset of those in 4292bdd1243dSDimitry Andric // the full index in the case of distributed indexes. 4293bdd1243dSDimitry Andric std::vector<uint64_t> StackIds; 4294bdd1243dSDimitry Andric for (auto &I : StackIdIndices) 4295bdd1243dSDimitry Andric StackIds.push_back(Index.getStackIdAtIndex(I)); 4296bdd1243dSDimitry Andric Stream.EmitRecord(bitc::FS_STACK_IDS, StackIds, StackIdAbbvId); 4297bdd1243dSDimitry Andric } 4298bdd1243dSDimitry Andric 4299*0b57cec5SDimitry Andric // Abbrev for FS_COMBINED. 4300*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 4301*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED)); 4302*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4303*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // modid 4304*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 4305*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // instcount 4306*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // fflags 4307*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // entrycount 4308*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numrefs 4309*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // rorefcnt 4310*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // worefcnt 4311*0b57cec5SDimitry Andric // numrefs x valueid, n x (valueid) 4312*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4313*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4314*0b57cec5SDimitry Andric unsigned FSCallsAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4315*0b57cec5SDimitry Andric 4316*0b57cec5SDimitry Andric // Abbrev for FS_COMBINED_PROFILE. 4317*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4318*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_PROFILE)); 4319*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4320*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // modid 4321*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 4322*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // instcount 4323*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // fflags 4324*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // entrycount 4325*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numrefs 4326*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // rorefcnt 4327*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // worefcnt 4328*0b57cec5SDimitry Andric // numrefs x valueid, n x (valueid, hotness) 4329*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4330*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4331*0b57cec5SDimitry Andric unsigned FSCallsProfileAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4332*0b57cec5SDimitry Andric 4333*0b57cec5SDimitry Andric // Abbrev for FS_COMBINED_GLOBALVAR_INIT_REFS. 4334*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4335*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_GLOBALVAR_INIT_REFS)); 4336*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4337*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // modid 4338*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 4339*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); // valueids 4340*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4341*0b57cec5SDimitry Andric unsigned FSModRefsAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4342*0b57cec5SDimitry Andric 4343*0b57cec5SDimitry Andric // Abbrev for FS_COMBINED_ALIAS. 4344*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4345*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_ALIAS)); 4346*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4347*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // modid 4348*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); // flags 4349*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4350*0b57cec5SDimitry Andric unsigned FSAliasAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4351*0b57cec5SDimitry Andric 4352bdd1243dSDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4353bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_CALLSITE_INFO)); 4354bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); // valueid 4355bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numstackindices 4356bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numver 4357bdd1243dSDimitry Andric // numstackindices x stackidindex, numver x version 4358bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4359bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4360bdd1243dSDimitry Andric unsigned CallsiteAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4361bdd1243dSDimitry Andric 4362bdd1243dSDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4363bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::FS_COMBINED_ALLOC_INFO)); 4364bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // nummib 4365bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 4)); // numver 4366bdd1243dSDimitry Andric // nummib x (alloc type, numstackids, numstackids x stackidindex), 4367bdd1243dSDimitry Andric // numver x version 4368bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4369bdd1243dSDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 8)); 4370bdd1243dSDimitry Andric unsigned AllocAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4371bdd1243dSDimitry Andric 4372*0b57cec5SDimitry Andric // The aliases are emitted as a post-pass, and will point to the value 4373*0b57cec5SDimitry Andric // id of the aliasee. Save them in a vector for post-processing. 4374*0b57cec5SDimitry Andric SmallVector<AliasSummary *, 64> Aliases; 4375*0b57cec5SDimitry Andric 4376*0b57cec5SDimitry Andric // Save the value id for each summary for alias emission. 4377*0b57cec5SDimitry Andric DenseMap<const GlobalValueSummary *, unsigned> SummaryToValueIdMap; 4378*0b57cec5SDimitry Andric 4379*0b57cec5SDimitry Andric SmallVector<uint64_t, 64> NameVals; 4380*0b57cec5SDimitry Andric 4381*0b57cec5SDimitry Andric // Set that will be populated during call to writeFunctionTypeMetadataRecords 4382*0b57cec5SDimitry Andric // with the type ids referenced by this index file. 4383*0b57cec5SDimitry Andric std::set<GlobalValue::GUID> ReferencedTypeIds; 4384*0b57cec5SDimitry Andric 4385*0b57cec5SDimitry Andric // For local linkage, we also emit the original name separately 4386*0b57cec5SDimitry Andric // immediately after the record. 4387*0b57cec5SDimitry Andric auto MaybeEmitOriginalName = [&](GlobalValueSummary &S) { 4388349cc55cSDimitry Andric // We don't need to emit the original name if we are writing the index for 4389349cc55cSDimitry Andric // distributed backends (in which case ModuleToSummariesForIndex is 4390349cc55cSDimitry Andric // non-null). The original name is only needed during the thin link, since 4391349cc55cSDimitry Andric // for SamplePGO the indirect call targets for local functions have 4392349cc55cSDimitry Andric // have the original name annotated in profile. 4393349cc55cSDimitry Andric // Continue to emit it when writing out the entire combined index, which is 4394349cc55cSDimitry Andric // used in testing the thin link via llvm-lto. 4395349cc55cSDimitry Andric if (ModuleToSummariesForIndex || !GlobalValue::isLocalLinkage(S.linkage())) 4396*0b57cec5SDimitry Andric return; 4397*0b57cec5SDimitry Andric NameVals.push_back(S.getOriginalName()); 4398*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_COMBINED_ORIGINAL_NAME, NameVals); 4399*0b57cec5SDimitry Andric NameVals.clear(); 4400*0b57cec5SDimitry Andric }; 4401*0b57cec5SDimitry Andric 4402*0b57cec5SDimitry Andric std::set<GlobalValue::GUID> DefOrUseGUIDs; 4403*0b57cec5SDimitry Andric forEachSummary([&](GVInfo I, bool IsAliasee) { 4404*0b57cec5SDimitry Andric GlobalValueSummary *S = I.second; 4405*0b57cec5SDimitry Andric assert(S); 4406*0b57cec5SDimitry Andric DefOrUseGUIDs.insert(I.first); 4407*0b57cec5SDimitry Andric for (const ValueInfo &VI : S->refs()) 4408*0b57cec5SDimitry Andric DefOrUseGUIDs.insert(VI.getGUID()); 4409*0b57cec5SDimitry Andric 4410*0b57cec5SDimitry Andric auto ValueId = getValueId(I.first); 4411*0b57cec5SDimitry Andric assert(ValueId); 4412*0b57cec5SDimitry Andric SummaryToValueIdMap[S] = *ValueId; 4413*0b57cec5SDimitry Andric 4414*0b57cec5SDimitry Andric // If this is invoked for an aliasee, we want to record the above 4415*0b57cec5SDimitry Andric // mapping, but then not emit a summary entry (if the aliasee is 4416*0b57cec5SDimitry Andric // to be imported, we will invoke this separately with IsAliasee=false). 4417*0b57cec5SDimitry Andric if (IsAliasee) 4418*0b57cec5SDimitry Andric return; 4419*0b57cec5SDimitry Andric 4420*0b57cec5SDimitry Andric if (auto *AS = dyn_cast<AliasSummary>(S)) { 4421*0b57cec5SDimitry Andric // Will process aliases as a post-pass because the reader wants all 4422*0b57cec5SDimitry Andric // global to be loaded first. 4423*0b57cec5SDimitry Andric Aliases.push_back(AS); 4424*0b57cec5SDimitry Andric return; 4425*0b57cec5SDimitry Andric } 4426*0b57cec5SDimitry Andric 4427*0b57cec5SDimitry Andric if (auto *VS = dyn_cast<GlobalVarSummary>(S)) { 4428*0b57cec5SDimitry Andric NameVals.push_back(*ValueId); 4429*0b57cec5SDimitry Andric NameVals.push_back(Index.getModuleId(VS->modulePath())); 4430*0b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(VS->flags())); 4431*0b57cec5SDimitry Andric NameVals.push_back(getEncodedGVarFlags(VS->varflags())); 4432*0b57cec5SDimitry Andric for (auto &RI : VS->refs()) { 4433*0b57cec5SDimitry Andric auto RefValueId = getValueId(RI.getGUID()); 4434*0b57cec5SDimitry Andric if (!RefValueId) 4435*0b57cec5SDimitry Andric continue; 4436*0b57cec5SDimitry Andric NameVals.push_back(*RefValueId); 4437*0b57cec5SDimitry Andric } 4438*0b57cec5SDimitry Andric 4439*0b57cec5SDimitry Andric // Emit the finished record. 4440*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_COMBINED_GLOBALVAR_INIT_REFS, NameVals, 4441*0b57cec5SDimitry Andric FSModRefsAbbrev); 4442*0b57cec5SDimitry Andric NameVals.clear(); 4443*0b57cec5SDimitry Andric MaybeEmitOriginalName(*S); 4444*0b57cec5SDimitry Andric return; 4445*0b57cec5SDimitry Andric } 4446*0b57cec5SDimitry Andric 4447bdd1243dSDimitry Andric auto GetValueId = [&](const ValueInfo &VI) -> std::optional<unsigned> { 4448bdd1243dSDimitry Andric if (!VI) 4449bdd1243dSDimitry Andric return std::nullopt; 4450349cc55cSDimitry Andric return getValueId(VI.getGUID()); 4451e8d8bef9SDimitry Andric }; 4452e8d8bef9SDimitry Andric 4453*0b57cec5SDimitry Andric auto *FS = cast<FunctionSummary>(S); 4454e8d8bef9SDimitry Andric writeFunctionTypeMetadataRecords(Stream, FS, GetValueId); 4455*0b57cec5SDimitry Andric getReferencedTypeIds(FS, ReferencedTypeIds); 4456*0b57cec5SDimitry Andric 4457bdd1243dSDimitry Andric writeFunctionHeapProfileRecords( 4458bdd1243dSDimitry Andric Stream, FS, CallsiteAbbrev, AllocAbbrev, 4459bdd1243dSDimitry Andric /*PerModule*/ false, 4460bdd1243dSDimitry Andric /*GetValueId*/ [&](const ValueInfo &VI) -> unsigned { 4461bdd1243dSDimitry Andric std::optional<unsigned> ValueID = GetValueId(VI); 4462bdd1243dSDimitry Andric // This can happen in shared index files for distributed ThinLTO if 4463bdd1243dSDimitry Andric // the callee function summary is not included. Record 0 which we 4464bdd1243dSDimitry Andric // will have to deal with conservatively when doing any kind of 4465bdd1243dSDimitry Andric // validation in the ThinLTO backends. 4466bdd1243dSDimitry Andric if (!ValueID) 4467bdd1243dSDimitry Andric return 0; 4468bdd1243dSDimitry Andric return *ValueID; 4469bdd1243dSDimitry Andric }, 4470bdd1243dSDimitry Andric /*GetStackIndex*/ [&](unsigned I) { 4471bdd1243dSDimitry Andric // Get the corresponding index into the list of StackIdIndices 4472bdd1243dSDimitry Andric // actually being written for this combined index (which may be a 4473bdd1243dSDimitry Andric // subset in the case of distributed indexes). 4474bdd1243dSDimitry Andric auto Lower = llvm::lower_bound(StackIdIndices, I); 4475bdd1243dSDimitry Andric return std::distance(StackIdIndices.begin(), Lower); 4476bdd1243dSDimitry Andric }); 4477bdd1243dSDimitry Andric 4478*0b57cec5SDimitry Andric NameVals.push_back(*ValueId); 4479*0b57cec5SDimitry Andric NameVals.push_back(Index.getModuleId(FS->modulePath())); 4480*0b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(FS->flags())); 4481*0b57cec5SDimitry Andric NameVals.push_back(FS->instCount()); 4482*0b57cec5SDimitry Andric NameVals.push_back(getEncodedFFlags(FS->fflags())); 4483*0b57cec5SDimitry Andric NameVals.push_back(FS->entryCount()); 4484*0b57cec5SDimitry Andric 4485*0b57cec5SDimitry Andric // Fill in below 4486*0b57cec5SDimitry Andric NameVals.push_back(0); // numrefs 4487*0b57cec5SDimitry Andric NameVals.push_back(0); // rorefcnt 4488*0b57cec5SDimitry Andric NameVals.push_back(0); // worefcnt 4489*0b57cec5SDimitry Andric 4490*0b57cec5SDimitry Andric unsigned Count = 0, RORefCnt = 0, WORefCnt = 0; 4491*0b57cec5SDimitry Andric for (auto &RI : FS->refs()) { 4492*0b57cec5SDimitry Andric auto RefValueId = getValueId(RI.getGUID()); 4493*0b57cec5SDimitry Andric if (!RefValueId) 4494*0b57cec5SDimitry Andric continue; 4495*0b57cec5SDimitry Andric NameVals.push_back(*RefValueId); 4496*0b57cec5SDimitry Andric if (RI.isReadOnly()) 4497*0b57cec5SDimitry Andric RORefCnt++; 4498*0b57cec5SDimitry Andric else if (RI.isWriteOnly()) 4499*0b57cec5SDimitry Andric WORefCnt++; 4500*0b57cec5SDimitry Andric Count++; 4501*0b57cec5SDimitry Andric } 4502*0b57cec5SDimitry Andric NameVals[6] = Count; 4503*0b57cec5SDimitry Andric NameVals[7] = RORefCnt; 4504*0b57cec5SDimitry Andric NameVals[8] = WORefCnt; 4505*0b57cec5SDimitry Andric 4506*0b57cec5SDimitry Andric bool HasProfileData = false; 4507*0b57cec5SDimitry Andric for (auto &EI : FS->calls()) { 4508*0b57cec5SDimitry Andric HasProfileData |= 4509*0b57cec5SDimitry Andric EI.second.getHotness() != CalleeInfo::HotnessType::Unknown; 4510*0b57cec5SDimitry Andric if (HasProfileData) 4511*0b57cec5SDimitry Andric break; 4512*0b57cec5SDimitry Andric } 4513*0b57cec5SDimitry Andric 4514*0b57cec5SDimitry Andric for (auto &EI : FS->calls()) { 4515*0b57cec5SDimitry Andric // If this GUID doesn't have a value id, it doesn't have a function 4516*0b57cec5SDimitry Andric // summary and we don't need to record any calls to it. 4517bdd1243dSDimitry Andric std::optional<unsigned> CallValueId = GetValueId(EI.first); 4518*0b57cec5SDimitry Andric if (!CallValueId) 4519*0b57cec5SDimitry Andric continue; 4520*0b57cec5SDimitry Andric NameVals.push_back(*CallValueId); 4521*0b57cec5SDimitry Andric if (HasProfileData) 4522*0b57cec5SDimitry Andric NameVals.push_back(static_cast<uint8_t>(EI.second.Hotness)); 4523*0b57cec5SDimitry Andric } 4524*0b57cec5SDimitry Andric 4525*0b57cec5SDimitry Andric unsigned FSAbbrev = (HasProfileData ? FSCallsProfileAbbrev : FSCallsAbbrev); 4526*0b57cec5SDimitry Andric unsigned Code = 4527*0b57cec5SDimitry Andric (HasProfileData ? bitc::FS_COMBINED_PROFILE : bitc::FS_COMBINED); 4528*0b57cec5SDimitry Andric 4529*0b57cec5SDimitry Andric // Emit the finished record. 4530*0b57cec5SDimitry Andric Stream.EmitRecord(Code, NameVals, FSAbbrev); 4531*0b57cec5SDimitry Andric NameVals.clear(); 4532*0b57cec5SDimitry Andric MaybeEmitOriginalName(*S); 4533*0b57cec5SDimitry Andric }); 4534*0b57cec5SDimitry Andric 4535*0b57cec5SDimitry Andric for (auto *AS : Aliases) { 4536*0b57cec5SDimitry Andric auto AliasValueId = SummaryToValueIdMap[AS]; 4537*0b57cec5SDimitry Andric assert(AliasValueId); 4538*0b57cec5SDimitry Andric NameVals.push_back(AliasValueId); 4539*0b57cec5SDimitry Andric NameVals.push_back(Index.getModuleId(AS->modulePath())); 4540*0b57cec5SDimitry Andric NameVals.push_back(getEncodedGVSummaryFlags(AS->flags())); 4541*0b57cec5SDimitry Andric auto AliaseeValueId = SummaryToValueIdMap[&AS->getAliasee()]; 4542*0b57cec5SDimitry Andric assert(AliaseeValueId); 4543*0b57cec5SDimitry Andric NameVals.push_back(AliaseeValueId); 4544*0b57cec5SDimitry Andric 4545*0b57cec5SDimitry Andric // Emit the finished record. 4546*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_COMBINED_ALIAS, NameVals, FSAliasAbbrev); 4547*0b57cec5SDimitry Andric NameVals.clear(); 4548*0b57cec5SDimitry Andric MaybeEmitOriginalName(*AS); 4549*0b57cec5SDimitry Andric 4550*0b57cec5SDimitry Andric if (auto *FS = dyn_cast<FunctionSummary>(&AS->getAliasee())) 4551*0b57cec5SDimitry Andric getReferencedTypeIds(FS, ReferencedTypeIds); 4552*0b57cec5SDimitry Andric } 4553*0b57cec5SDimitry Andric 4554*0b57cec5SDimitry Andric if (!Index.cfiFunctionDefs().empty()) { 4555*0b57cec5SDimitry Andric for (auto &S : Index.cfiFunctionDefs()) { 4556*0b57cec5SDimitry Andric if (DefOrUseGUIDs.count( 4557*0b57cec5SDimitry Andric GlobalValue::getGUID(GlobalValue::dropLLVMManglingEscape(S)))) { 4558*0b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(S)); 4559*0b57cec5SDimitry Andric NameVals.push_back(S.size()); 4560*0b57cec5SDimitry Andric } 4561*0b57cec5SDimitry Andric } 4562*0b57cec5SDimitry Andric if (!NameVals.empty()) { 4563*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_CFI_FUNCTION_DEFS, NameVals); 4564*0b57cec5SDimitry Andric NameVals.clear(); 4565*0b57cec5SDimitry Andric } 4566*0b57cec5SDimitry Andric } 4567*0b57cec5SDimitry Andric 4568*0b57cec5SDimitry Andric if (!Index.cfiFunctionDecls().empty()) { 4569*0b57cec5SDimitry Andric for (auto &S : Index.cfiFunctionDecls()) { 4570*0b57cec5SDimitry Andric if (DefOrUseGUIDs.count( 4571*0b57cec5SDimitry Andric GlobalValue::getGUID(GlobalValue::dropLLVMManglingEscape(S)))) { 4572*0b57cec5SDimitry Andric NameVals.push_back(StrtabBuilder.add(S)); 4573*0b57cec5SDimitry Andric NameVals.push_back(S.size()); 4574*0b57cec5SDimitry Andric } 4575*0b57cec5SDimitry Andric } 4576*0b57cec5SDimitry Andric if (!NameVals.empty()) { 4577*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_CFI_FUNCTION_DECLS, NameVals); 4578*0b57cec5SDimitry Andric NameVals.clear(); 4579*0b57cec5SDimitry Andric } 4580*0b57cec5SDimitry Andric } 4581*0b57cec5SDimitry Andric 4582*0b57cec5SDimitry Andric // Walk the GUIDs that were referenced, and write the 4583*0b57cec5SDimitry Andric // corresponding type id records. 4584*0b57cec5SDimitry Andric for (auto &T : ReferencedTypeIds) { 4585*0b57cec5SDimitry Andric auto TidIter = Index.typeIds().equal_range(T); 4586*0b57cec5SDimitry Andric for (auto It = TidIter.first; It != TidIter.second; ++It) { 4587*0b57cec5SDimitry Andric writeTypeIdSummaryRecord(NameVals, StrtabBuilder, It->second.first, 4588*0b57cec5SDimitry Andric It->second.second); 4589*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::FS_TYPE_ID, NameVals); 4590*0b57cec5SDimitry Andric NameVals.clear(); 4591*0b57cec5SDimitry Andric } 4592*0b57cec5SDimitry Andric } 4593*0b57cec5SDimitry Andric 45945ffd83dbSDimitry Andric Stream.EmitRecord(bitc::FS_BLOCK_COUNT, 45955ffd83dbSDimitry Andric ArrayRef<uint64_t>{Index.getBlockCount()}); 45965ffd83dbSDimitry Andric 4597*0b57cec5SDimitry Andric Stream.ExitBlock(); 4598*0b57cec5SDimitry Andric } 4599*0b57cec5SDimitry Andric 4600*0b57cec5SDimitry Andric /// Create the "IDENTIFICATION_BLOCK_ID" containing a single string with the 4601*0b57cec5SDimitry Andric /// current llvm version, and a record for the epoch number. 4602*0b57cec5SDimitry Andric static void writeIdentificationBlock(BitstreamWriter &Stream) { 4603*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::IDENTIFICATION_BLOCK_ID, 5); 4604*0b57cec5SDimitry Andric 4605*0b57cec5SDimitry Andric // Write the "user readable" string identifying the bitcode producer 4606*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 4607*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::IDENTIFICATION_CODE_STRING)); 4608*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4609*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Char6)); 4610*0b57cec5SDimitry Andric auto StringAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4611*0b57cec5SDimitry Andric writeStringRecord(Stream, bitc::IDENTIFICATION_CODE_STRING, 4612*0b57cec5SDimitry Andric "LLVM" LLVM_VERSION_STRING, StringAbbrev); 4613*0b57cec5SDimitry Andric 4614*0b57cec5SDimitry Andric // Write the epoch version 4615*0b57cec5SDimitry Andric Abbv = std::make_shared<BitCodeAbbrev>(); 4616*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::IDENTIFICATION_CODE_EPOCH)); 4617*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::VBR, 6)); 4618*0b57cec5SDimitry Andric auto EpochAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 46195ffd83dbSDimitry Andric constexpr std::array<unsigned, 1> Vals = {{bitc::BITCODE_CURRENT_EPOCH}}; 4620*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::IDENTIFICATION_CODE_EPOCH, Vals, EpochAbbrev); 4621*0b57cec5SDimitry Andric Stream.ExitBlock(); 4622*0b57cec5SDimitry Andric } 4623*0b57cec5SDimitry Andric 4624*0b57cec5SDimitry Andric void ModuleBitcodeWriter::writeModuleHash(size_t BlockStartPos) { 4625*0b57cec5SDimitry Andric // Emit the module's hash. 4626*0b57cec5SDimitry Andric // MODULE_CODE_HASH: [5*i32] 4627*0b57cec5SDimitry Andric if (GenerateHash) { 4628*0b57cec5SDimitry Andric uint32_t Vals[5]; 4629*0b57cec5SDimitry Andric Hasher.update(ArrayRef<uint8_t>((const uint8_t *)&(Buffer)[BlockStartPos], 4630*0b57cec5SDimitry Andric Buffer.size() - BlockStartPos)); 463181ad6265SDimitry Andric std::array<uint8_t, 20> Hash = Hasher.result(); 4632*0b57cec5SDimitry Andric for (int Pos = 0; Pos < 20; Pos += 4) { 4633*0b57cec5SDimitry Andric Vals[Pos / 4] = support::endian::read32be(Hash.data() + Pos); 4634*0b57cec5SDimitry Andric } 4635*0b57cec5SDimitry Andric 4636*0b57cec5SDimitry Andric // Emit the finished record. 4637*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_HASH, Vals); 4638*0b57cec5SDimitry Andric 4639*0b57cec5SDimitry Andric if (ModHash) 4640*0b57cec5SDimitry Andric // Save the written hash value. 4641*0b57cec5SDimitry Andric llvm::copy(Vals, std::begin(*ModHash)); 4642*0b57cec5SDimitry Andric } 4643*0b57cec5SDimitry Andric } 4644*0b57cec5SDimitry Andric 4645*0b57cec5SDimitry Andric void ModuleBitcodeWriter::write() { 4646*0b57cec5SDimitry Andric writeIdentificationBlock(Stream); 4647*0b57cec5SDimitry Andric 4648*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::MODULE_BLOCK_ID, 3); 4649*0b57cec5SDimitry Andric size_t BlockStartPos = Buffer.size(); 4650*0b57cec5SDimitry Andric 4651*0b57cec5SDimitry Andric writeModuleVersion(); 4652*0b57cec5SDimitry Andric 4653*0b57cec5SDimitry Andric // Emit blockinfo, which defines the standard abbreviations etc. 4654*0b57cec5SDimitry Andric writeBlockInfo(); 4655*0b57cec5SDimitry Andric 4656*0b57cec5SDimitry Andric // Emit information describing all of the types in the module. 4657*0b57cec5SDimitry Andric writeTypeTable(); 4658*0b57cec5SDimitry Andric 4659*0b57cec5SDimitry Andric // Emit information about attribute groups. 4660*0b57cec5SDimitry Andric writeAttributeGroupTable(); 4661*0b57cec5SDimitry Andric 4662*0b57cec5SDimitry Andric // Emit information about parameter attributes. 4663*0b57cec5SDimitry Andric writeAttributeTable(); 4664*0b57cec5SDimitry Andric 4665*0b57cec5SDimitry Andric writeComdats(); 4666*0b57cec5SDimitry Andric 4667*0b57cec5SDimitry Andric // Emit top-level description of module, including target triple, inline asm, 4668*0b57cec5SDimitry Andric // descriptors for global variables, and function prototype info. 4669*0b57cec5SDimitry Andric writeModuleInfo(); 4670*0b57cec5SDimitry Andric 4671*0b57cec5SDimitry Andric // Emit constants. 4672*0b57cec5SDimitry Andric writeModuleConstants(); 4673*0b57cec5SDimitry Andric 4674*0b57cec5SDimitry Andric // Emit metadata kind names. 4675*0b57cec5SDimitry Andric writeModuleMetadataKinds(); 4676*0b57cec5SDimitry Andric 4677*0b57cec5SDimitry Andric // Emit metadata. 4678*0b57cec5SDimitry Andric writeModuleMetadata(); 4679*0b57cec5SDimitry Andric 4680*0b57cec5SDimitry Andric // Emit module-level use-lists. 4681*0b57cec5SDimitry Andric if (VE.shouldPreserveUseListOrder()) 4682*0b57cec5SDimitry Andric writeUseListBlock(nullptr); 4683*0b57cec5SDimitry Andric 4684*0b57cec5SDimitry Andric writeOperandBundleTags(); 4685*0b57cec5SDimitry Andric writeSyncScopeNames(); 4686*0b57cec5SDimitry Andric 4687*0b57cec5SDimitry Andric // Emit function bodies. 4688*0b57cec5SDimitry Andric DenseMap<const Function *, uint64_t> FunctionToBitcodeIndex; 46894824e7fdSDimitry Andric for (const Function &F : M) 46904824e7fdSDimitry Andric if (!F.isDeclaration()) 46914824e7fdSDimitry Andric writeFunction(F, FunctionToBitcodeIndex); 4692*0b57cec5SDimitry Andric 4693*0b57cec5SDimitry Andric // Need to write after the above call to WriteFunction which populates 4694*0b57cec5SDimitry Andric // the summary information in the index. 4695*0b57cec5SDimitry Andric if (Index) 4696*0b57cec5SDimitry Andric writePerModuleGlobalValueSummary(); 4697*0b57cec5SDimitry Andric 4698*0b57cec5SDimitry Andric writeGlobalValueSymbolTable(FunctionToBitcodeIndex); 4699*0b57cec5SDimitry Andric 4700*0b57cec5SDimitry Andric writeModuleHash(BlockStartPos); 4701*0b57cec5SDimitry Andric 4702*0b57cec5SDimitry Andric Stream.ExitBlock(); 4703*0b57cec5SDimitry Andric } 4704*0b57cec5SDimitry Andric 4705*0b57cec5SDimitry Andric static void writeInt32ToBuffer(uint32_t Value, SmallVectorImpl<char> &Buffer, 4706*0b57cec5SDimitry Andric uint32_t &Position) { 4707*0b57cec5SDimitry Andric support::endian::write32le(&Buffer[Position], Value); 4708*0b57cec5SDimitry Andric Position += 4; 4709*0b57cec5SDimitry Andric } 4710*0b57cec5SDimitry Andric 4711*0b57cec5SDimitry Andric /// If generating a bc file on darwin, we have to emit a 4712*0b57cec5SDimitry Andric /// header and trailer to make it compatible with the system archiver. To do 4713*0b57cec5SDimitry Andric /// this we emit the following header, and then emit a trailer that pads the 4714*0b57cec5SDimitry Andric /// file out to be a multiple of 16 bytes. 4715*0b57cec5SDimitry Andric /// 4716*0b57cec5SDimitry Andric /// struct bc_header { 4717*0b57cec5SDimitry Andric /// uint32_t Magic; // 0x0B17C0DE 4718*0b57cec5SDimitry Andric /// uint32_t Version; // Version, currently always 0. 4719*0b57cec5SDimitry Andric /// uint32_t BitcodeOffset; // Offset to traditional bitcode file. 4720*0b57cec5SDimitry Andric /// uint32_t BitcodeSize; // Size of traditional bitcode file. 4721*0b57cec5SDimitry Andric /// uint32_t CPUType; // CPU specifier. 4722*0b57cec5SDimitry Andric /// ... potentially more later ... 4723*0b57cec5SDimitry Andric /// }; 4724*0b57cec5SDimitry Andric static void emitDarwinBCHeaderAndTrailer(SmallVectorImpl<char> &Buffer, 4725*0b57cec5SDimitry Andric const Triple &TT) { 4726*0b57cec5SDimitry Andric unsigned CPUType = ~0U; 4727*0b57cec5SDimitry Andric 4728*0b57cec5SDimitry Andric // Match x86_64-*, i[3-9]86-*, powerpc-*, powerpc64-*, arm-*, thumb-*, 4729*0b57cec5SDimitry Andric // armv[0-9]-*, thumbv[0-9]-*, armv5te-*, or armv6t2-*. The CPUType is a magic 4730*0b57cec5SDimitry Andric // number from /usr/include/mach/machine.h. It is ok to reproduce the 4731*0b57cec5SDimitry Andric // specific constants here because they are implicitly part of the Darwin ABI. 4732*0b57cec5SDimitry Andric enum { 4733*0b57cec5SDimitry Andric DARWIN_CPU_ARCH_ABI64 = 0x01000000, 4734*0b57cec5SDimitry Andric DARWIN_CPU_TYPE_X86 = 7, 4735*0b57cec5SDimitry Andric DARWIN_CPU_TYPE_ARM = 12, 4736*0b57cec5SDimitry Andric DARWIN_CPU_TYPE_POWERPC = 18 4737*0b57cec5SDimitry Andric }; 4738*0b57cec5SDimitry Andric 4739*0b57cec5SDimitry Andric Triple::ArchType Arch = TT.getArch(); 4740*0b57cec5SDimitry Andric if (Arch == Triple::x86_64) 4741*0b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_X86 | DARWIN_CPU_ARCH_ABI64; 4742*0b57cec5SDimitry Andric else if (Arch == Triple::x86) 4743*0b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_X86; 4744*0b57cec5SDimitry Andric else if (Arch == Triple::ppc) 4745*0b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_POWERPC; 4746*0b57cec5SDimitry Andric else if (Arch == Triple::ppc64) 4747*0b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_POWERPC | DARWIN_CPU_ARCH_ABI64; 4748*0b57cec5SDimitry Andric else if (Arch == Triple::arm || Arch == Triple::thumb) 4749*0b57cec5SDimitry Andric CPUType = DARWIN_CPU_TYPE_ARM; 4750*0b57cec5SDimitry Andric 4751*0b57cec5SDimitry Andric // Traditional Bitcode starts after header. 4752*0b57cec5SDimitry Andric assert(Buffer.size() >= BWH_HeaderSize && 4753*0b57cec5SDimitry Andric "Expected header size to be reserved"); 4754*0b57cec5SDimitry Andric unsigned BCOffset = BWH_HeaderSize; 4755*0b57cec5SDimitry Andric unsigned BCSize = Buffer.size() - BWH_HeaderSize; 4756*0b57cec5SDimitry Andric 4757*0b57cec5SDimitry Andric // Write the magic and version. 4758*0b57cec5SDimitry Andric unsigned Position = 0; 4759*0b57cec5SDimitry Andric writeInt32ToBuffer(0x0B17C0DE, Buffer, Position); 4760*0b57cec5SDimitry Andric writeInt32ToBuffer(0, Buffer, Position); // Version. 4761*0b57cec5SDimitry Andric writeInt32ToBuffer(BCOffset, Buffer, Position); 4762*0b57cec5SDimitry Andric writeInt32ToBuffer(BCSize, Buffer, Position); 4763*0b57cec5SDimitry Andric writeInt32ToBuffer(CPUType, Buffer, Position); 4764*0b57cec5SDimitry Andric 4765*0b57cec5SDimitry Andric // If the file is not a multiple of 16 bytes, insert dummy padding. 4766*0b57cec5SDimitry Andric while (Buffer.size() & 15) 4767*0b57cec5SDimitry Andric Buffer.push_back(0); 4768*0b57cec5SDimitry Andric } 4769*0b57cec5SDimitry Andric 4770*0b57cec5SDimitry Andric /// Helper to write the header common to all bitcode files. 4771*0b57cec5SDimitry Andric static void writeBitcodeHeader(BitstreamWriter &Stream) { 4772*0b57cec5SDimitry Andric // Emit the file header. 4773*0b57cec5SDimitry Andric Stream.Emit((unsigned)'B', 8); 4774*0b57cec5SDimitry Andric Stream.Emit((unsigned)'C', 8); 4775*0b57cec5SDimitry Andric Stream.Emit(0x0, 4); 4776*0b57cec5SDimitry Andric Stream.Emit(0xC, 4); 4777*0b57cec5SDimitry Andric Stream.Emit(0xE, 4); 4778*0b57cec5SDimitry Andric Stream.Emit(0xD, 4); 4779*0b57cec5SDimitry Andric } 4780*0b57cec5SDimitry Andric 4781e8d8bef9SDimitry Andric BitcodeWriter::BitcodeWriter(SmallVectorImpl<char> &Buffer, raw_fd_stream *FS) 4782e8d8bef9SDimitry Andric : Buffer(Buffer), Stream(new BitstreamWriter(Buffer, FS, FlushThreshold)) { 4783*0b57cec5SDimitry Andric writeBitcodeHeader(*Stream); 4784*0b57cec5SDimitry Andric } 4785*0b57cec5SDimitry Andric 4786*0b57cec5SDimitry Andric BitcodeWriter::~BitcodeWriter() { assert(WroteStrtab); } 4787*0b57cec5SDimitry Andric 4788*0b57cec5SDimitry Andric void BitcodeWriter::writeBlob(unsigned Block, unsigned Record, StringRef Blob) { 4789*0b57cec5SDimitry Andric Stream->EnterSubblock(Block, 3); 4790*0b57cec5SDimitry Andric 4791*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 4792*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(Record)); 4793*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob)); 4794*0b57cec5SDimitry Andric auto AbbrevNo = Stream->EmitAbbrev(std::move(Abbv)); 4795*0b57cec5SDimitry Andric 4796*0b57cec5SDimitry Andric Stream->EmitRecordWithBlob(AbbrevNo, ArrayRef<uint64_t>{Record}, Blob); 4797*0b57cec5SDimitry Andric 4798*0b57cec5SDimitry Andric Stream->ExitBlock(); 4799*0b57cec5SDimitry Andric } 4800*0b57cec5SDimitry Andric 4801*0b57cec5SDimitry Andric void BitcodeWriter::writeSymtab() { 4802*0b57cec5SDimitry Andric assert(!WroteStrtab && !WroteSymtab); 4803*0b57cec5SDimitry Andric 4804*0b57cec5SDimitry Andric // If any module has module-level inline asm, we will require a registered asm 4805*0b57cec5SDimitry Andric // parser for the target so that we can create an accurate symbol table for 4806*0b57cec5SDimitry Andric // the module. 4807*0b57cec5SDimitry Andric for (Module *M : Mods) { 4808*0b57cec5SDimitry Andric if (M->getModuleInlineAsm().empty()) 4809*0b57cec5SDimitry Andric continue; 4810*0b57cec5SDimitry Andric 4811*0b57cec5SDimitry Andric std::string Err; 4812*0b57cec5SDimitry Andric const Triple TT(M->getTargetTriple()); 4813*0b57cec5SDimitry Andric const Target *T = TargetRegistry::lookupTarget(TT.str(), Err); 4814*0b57cec5SDimitry Andric if (!T || !T->hasMCAsmParser()) 4815*0b57cec5SDimitry Andric return; 4816*0b57cec5SDimitry Andric } 4817*0b57cec5SDimitry Andric 4818*0b57cec5SDimitry Andric WroteSymtab = true; 4819*0b57cec5SDimitry Andric SmallVector<char, 0> Symtab; 4820*0b57cec5SDimitry Andric // The irsymtab::build function may be unable to create a symbol table if the 4821*0b57cec5SDimitry Andric // module is malformed (e.g. it contains an invalid alias). Writing a symbol 4822*0b57cec5SDimitry Andric // table is not required for correctness, but we still want to be able to 4823*0b57cec5SDimitry Andric // write malformed modules to bitcode files, so swallow the error. 4824*0b57cec5SDimitry Andric if (Error E = irsymtab::build(Mods, Symtab, StrtabBuilder, Alloc)) { 4825*0b57cec5SDimitry Andric consumeError(std::move(E)); 4826*0b57cec5SDimitry Andric return; 4827*0b57cec5SDimitry Andric } 4828*0b57cec5SDimitry Andric 4829*0b57cec5SDimitry Andric writeBlob(bitc::SYMTAB_BLOCK_ID, bitc::SYMTAB_BLOB, 4830*0b57cec5SDimitry Andric {Symtab.data(), Symtab.size()}); 4831*0b57cec5SDimitry Andric } 4832*0b57cec5SDimitry Andric 4833*0b57cec5SDimitry Andric void BitcodeWriter::writeStrtab() { 4834*0b57cec5SDimitry Andric assert(!WroteStrtab); 4835*0b57cec5SDimitry Andric 4836*0b57cec5SDimitry Andric std::vector<char> Strtab; 4837*0b57cec5SDimitry Andric StrtabBuilder.finalizeInOrder(); 4838*0b57cec5SDimitry Andric Strtab.resize(StrtabBuilder.getSize()); 4839*0b57cec5SDimitry Andric StrtabBuilder.write((uint8_t *)Strtab.data()); 4840*0b57cec5SDimitry Andric 4841*0b57cec5SDimitry Andric writeBlob(bitc::STRTAB_BLOCK_ID, bitc::STRTAB_BLOB, 4842*0b57cec5SDimitry Andric {Strtab.data(), Strtab.size()}); 4843*0b57cec5SDimitry Andric 4844*0b57cec5SDimitry Andric WroteStrtab = true; 4845*0b57cec5SDimitry Andric } 4846*0b57cec5SDimitry Andric 4847*0b57cec5SDimitry Andric void BitcodeWriter::copyStrtab(StringRef Strtab) { 4848*0b57cec5SDimitry Andric writeBlob(bitc::STRTAB_BLOCK_ID, bitc::STRTAB_BLOB, Strtab); 4849*0b57cec5SDimitry Andric WroteStrtab = true; 4850*0b57cec5SDimitry Andric } 4851*0b57cec5SDimitry Andric 4852*0b57cec5SDimitry Andric void BitcodeWriter::writeModule(const Module &M, 4853*0b57cec5SDimitry Andric bool ShouldPreserveUseListOrder, 4854*0b57cec5SDimitry Andric const ModuleSummaryIndex *Index, 4855*0b57cec5SDimitry Andric bool GenerateHash, ModuleHash *ModHash) { 4856*0b57cec5SDimitry Andric assert(!WroteStrtab); 4857*0b57cec5SDimitry Andric 4858*0b57cec5SDimitry Andric // The Mods vector is used by irsymtab::build, which requires non-const 4859*0b57cec5SDimitry Andric // Modules in case it needs to materialize metadata. But the bitcode writer 4860*0b57cec5SDimitry Andric // requires that the module is materialized, so we can cast to non-const here, 4861*0b57cec5SDimitry Andric // after checking that it is in fact materialized. 4862*0b57cec5SDimitry Andric assert(M.isMaterialized()); 4863*0b57cec5SDimitry Andric Mods.push_back(const_cast<Module *>(&M)); 4864*0b57cec5SDimitry Andric 4865*0b57cec5SDimitry Andric ModuleBitcodeWriter ModuleWriter(M, Buffer, StrtabBuilder, *Stream, 4866*0b57cec5SDimitry Andric ShouldPreserveUseListOrder, Index, 4867*0b57cec5SDimitry Andric GenerateHash, ModHash); 4868*0b57cec5SDimitry Andric ModuleWriter.write(); 4869*0b57cec5SDimitry Andric } 4870*0b57cec5SDimitry Andric 4871*0b57cec5SDimitry Andric void BitcodeWriter::writeIndex( 4872*0b57cec5SDimitry Andric const ModuleSummaryIndex *Index, 4873*0b57cec5SDimitry Andric const std::map<std::string, GVSummaryMapTy> *ModuleToSummariesForIndex) { 4874*0b57cec5SDimitry Andric IndexBitcodeWriter IndexWriter(*Stream, StrtabBuilder, *Index, 4875*0b57cec5SDimitry Andric ModuleToSummariesForIndex); 4876*0b57cec5SDimitry Andric IndexWriter.write(); 4877*0b57cec5SDimitry Andric } 4878*0b57cec5SDimitry Andric 4879*0b57cec5SDimitry Andric /// Write the specified module to the specified output stream. 4880*0b57cec5SDimitry Andric void llvm::WriteBitcodeToFile(const Module &M, raw_ostream &Out, 4881*0b57cec5SDimitry Andric bool ShouldPreserveUseListOrder, 4882*0b57cec5SDimitry Andric const ModuleSummaryIndex *Index, 4883*0b57cec5SDimitry Andric bool GenerateHash, ModuleHash *ModHash) { 4884*0b57cec5SDimitry Andric SmallVector<char, 0> Buffer; 4885*0b57cec5SDimitry Andric Buffer.reserve(256*1024); 4886*0b57cec5SDimitry Andric 4887*0b57cec5SDimitry Andric // If this is darwin or another generic macho target, reserve space for the 4888*0b57cec5SDimitry Andric // header. 4889*0b57cec5SDimitry Andric Triple TT(M.getTargetTriple()); 4890*0b57cec5SDimitry Andric if (TT.isOSDarwin() || TT.isOSBinFormatMachO()) 4891*0b57cec5SDimitry Andric Buffer.insert(Buffer.begin(), BWH_HeaderSize, 0); 4892*0b57cec5SDimitry Andric 4893e8d8bef9SDimitry Andric BitcodeWriter Writer(Buffer, dyn_cast<raw_fd_stream>(&Out)); 4894*0b57cec5SDimitry Andric Writer.writeModule(M, ShouldPreserveUseListOrder, Index, GenerateHash, 4895*0b57cec5SDimitry Andric ModHash); 4896*0b57cec5SDimitry Andric Writer.writeSymtab(); 4897*0b57cec5SDimitry Andric Writer.writeStrtab(); 4898*0b57cec5SDimitry Andric 4899*0b57cec5SDimitry Andric if (TT.isOSDarwin() || TT.isOSBinFormatMachO()) 4900*0b57cec5SDimitry Andric emitDarwinBCHeaderAndTrailer(Buffer, TT); 4901*0b57cec5SDimitry Andric 4902*0b57cec5SDimitry Andric // Write the generated bitstream to "Out". 4903e8d8bef9SDimitry Andric if (!Buffer.empty()) 4904*0b57cec5SDimitry Andric Out.write((char *)&Buffer.front(), Buffer.size()); 4905*0b57cec5SDimitry Andric } 4906*0b57cec5SDimitry Andric 4907*0b57cec5SDimitry Andric void IndexBitcodeWriter::write() { 4908*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::MODULE_BLOCK_ID, 3); 4909*0b57cec5SDimitry Andric 4910*0b57cec5SDimitry Andric writeModuleVersion(); 4911*0b57cec5SDimitry Andric 4912*0b57cec5SDimitry Andric // Write the module paths in the combined index. 4913*0b57cec5SDimitry Andric writeModStrings(); 4914*0b57cec5SDimitry Andric 4915*0b57cec5SDimitry Andric // Write the summary combined index records. 4916*0b57cec5SDimitry Andric writeCombinedGlobalValueSummary(); 4917*0b57cec5SDimitry Andric 4918*0b57cec5SDimitry Andric Stream.ExitBlock(); 4919*0b57cec5SDimitry Andric } 4920*0b57cec5SDimitry Andric 4921*0b57cec5SDimitry Andric // Write the specified module summary index to the given raw output stream, 4922*0b57cec5SDimitry Andric // where it will be written in a new bitcode block. This is used when 4923*0b57cec5SDimitry Andric // writing the combined index file for ThinLTO. When writing a subset of the 4924*0b57cec5SDimitry Andric // index for a distributed backend, provide a \p ModuleToSummariesForIndex map. 49251fd87a68SDimitry Andric void llvm::writeIndexToFile( 4926*0b57cec5SDimitry Andric const ModuleSummaryIndex &Index, raw_ostream &Out, 4927*0b57cec5SDimitry Andric const std::map<std::string, GVSummaryMapTy> *ModuleToSummariesForIndex) { 4928*0b57cec5SDimitry Andric SmallVector<char, 0> Buffer; 4929*0b57cec5SDimitry Andric Buffer.reserve(256 * 1024); 4930*0b57cec5SDimitry Andric 4931*0b57cec5SDimitry Andric BitcodeWriter Writer(Buffer); 4932*0b57cec5SDimitry Andric Writer.writeIndex(&Index, ModuleToSummariesForIndex); 4933*0b57cec5SDimitry Andric Writer.writeStrtab(); 4934*0b57cec5SDimitry Andric 4935*0b57cec5SDimitry Andric Out.write((char *)&Buffer.front(), Buffer.size()); 4936*0b57cec5SDimitry Andric } 4937*0b57cec5SDimitry Andric 4938*0b57cec5SDimitry Andric namespace { 4939*0b57cec5SDimitry Andric 4940*0b57cec5SDimitry Andric /// Class to manage the bitcode writing for a thin link bitcode file. 4941*0b57cec5SDimitry Andric class ThinLinkBitcodeWriter : public ModuleBitcodeWriterBase { 4942*0b57cec5SDimitry Andric /// ModHash is for use in ThinLTO incremental build, generated while writing 4943*0b57cec5SDimitry Andric /// the module bitcode file. 4944*0b57cec5SDimitry Andric const ModuleHash *ModHash; 4945*0b57cec5SDimitry Andric 4946*0b57cec5SDimitry Andric public: 4947*0b57cec5SDimitry Andric ThinLinkBitcodeWriter(const Module &M, StringTableBuilder &StrtabBuilder, 4948*0b57cec5SDimitry Andric BitstreamWriter &Stream, 4949*0b57cec5SDimitry Andric const ModuleSummaryIndex &Index, 4950*0b57cec5SDimitry Andric const ModuleHash &ModHash) 4951*0b57cec5SDimitry Andric : ModuleBitcodeWriterBase(M, StrtabBuilder, Stream, 4952*0b57cec5SDimitry Andric /*ShouldPreserveUseListOrder=*/false, &Index), 4953*0b57cec5SDimitry Andric ModHash(&ModHash) {} 4954*0b57cec5SDimitry Andric 4955*0b57cec5SDimitry Andric void write(); 4956*0b57cec5SDimitry Andric 4957*0b57cec5SDimitry Andric private: 4958*0b57cec5SDimitry Andric void writeSimplifiedModuleInfo(); 4959*0b57cec5SDimitry Andric }; 4960*0b57cec5SDimitry Andric 4961*0b57cec5SDimitry Andric } // end anonymous namespace 4962*0b57cec5SDimitry Andric 4963*0b57cec5SDimitry Andric // This function writes a simpilified module info for thin link bitcode file. 4964*0b57cec5SDimitry Andric // It only contains the source file name along with the name(the offset and 4965*0b57cec5SDimitry Andric // size in strtab) and linkage for global values. For the global value info 4966*0b57cec5SDimitry Andric // entry, in order to keep linkage at offset 5, there are three zeros used 4967*0b57cec5SDimitry Andric // as padding. 4968*0b57cec5SDimitry Andric void ThinLinkBitcodeWriter::writeSimplifiedModuleInfo() { 4969*0b57cec5SDimitry Andric SmallVector<unsigned, 64> Vals; 4970*0b57cec5SDimitry Andric // Emit the module's source file name. 4971*0b57cec5SDimitry Andric { 4972*0b57cec5SDimitry Andric StringEncoding Bits = getStringEncoding(M.getSourceFileName()); 4973*0b57cec5SDimitry Andric BitCodeAbbrevOp AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 8); 4974*0b57cec5SDimitry Andric if (Bits == SE_Char6) 4975*0b57cec5SDimitry Andric AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Char6); 4976*0b57cec5SDimitry Andric else if (Bits == SE_Fixed7) 4977*0b57cec5SDimitry Andric AbbrevOpToUse = BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 7); 4978*0b57cec5SDimitry Andric 4979*0b57cec5SDimitry Andric // MODULE_CODE_SOURCE_FILENAME: [namechar x N] 4980*0b57cec5SDimitry Andric auto Abbv = std::make_shared<BitCodeAbbrev>(); 4981*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(bitc::MODULE_CODE_SOURCE_FILENAME)); 4982*0b57cec5SDimitry Andric Abbv->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Array)); 4983*0b57cec5SDimitry Andric Abbv->Add(AbbrevOpToUse); 4984*0b57cec5SDimitry Andric unsigned FilenameAbbrev = Stream.EmitAbbrev(std::move(Abbv)); 4985*0b57cec5SDimitry Andric 4986*0b57cec5SDimitry Andric for (const auto P : M.getSourceFileName()) 4987*0b57cec5SDimitry Andric Vals.push_back((unsigned char)P); 4988*0b57cec5SDimitry Andric 4989*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_SOURCE_FILENAME, Vals, FilenameAbbrev); 4990*0b57cec5SDimitry Andric Vals.clear(); 4991*0b57cec5SDimitry Andric } 4992*0b57cec5SDimitry Andric 4993*0b57cec5SDimitry Andric // Emit the global variable information. 4994*0b57cec5SDimitry Andric for (const GlobalVariable &GV : M.globals()) { 4995*0b57cec5SDimitry Andric // GLOBALVAR: [strtab offset, strtab size, 0, 0, 0, linkage] 4996*0b57cec5SDimitry Andric Vals.push_back(StrtabBuilder.add(GV.getName())); 4997*0b57cec5SDimitry Andric Vals.push_back(GV.getName().size()); 4998*0b57cec5SDimitry Andric Vals.push_back(0); 4999*0b57cec5SDimitry Andric Vals.push_back(0); 5000*0b57cec5SDimitry Andric Vals.push_back(0); 5001*0b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(GV)); 5002*0b57cec5SDimitry Andric 5003*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_GLOBALVAR, Vals); 5004*0b57cec5SDimitry Andric Vals.clear(); 5005*0b57cec5SDimitry Andric } 5006*0b57cec5SDimitry Andric 5007*0b57cec5SDimitry Andric // Emit the function proto information. 5008*0b57cec5SDimitry Andric for (const Function &F : M) { 5009*0b57cec5SDimitry Andric // FUNCTION: [strtab offset, strtab size, 0, 0, 0, linkage] 5010*0b57cec5SDimitry Andric Vals.push_back(StrtabBuilder.add(F.getName())); 5011*0b57cec5SDimitry Andric Vals.push_back(F.getName().size()); 5012*0b57cec5SDimitry Andric Vals.push_back(0); 5013*0b57cec5SDimitry Andric Vals.push_back(0); 5014*0b57cec5SDimitry Andric Vals.push_back(0); 5015*0b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(F)); 5016*0b57cec5SDimitry Andric 5017*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_FUNCTION, Vals); 5018*0b57cec5SDimitry Andric Vals.clear(); 5019*0b57cec5SDimitry Andric } 5020*0b57cec5SDimitry Andric 5021*0b57cec5SDimitry Andric // Emit the alias information. 5022*0b57cec5SDimitry Andric for (const GlobalAlias &A : M.aliases()) { 5023*0b57cec5SDimitry Andric // ALIAS: [strtab offset, strtab size, 0, 0, 0, linkage] 5024*0b57cec5SDimitry Andric Vals.push_back(StrtabBuilder.add(A.getName())); 5025*0b57cec5SDimitry Andric Vals.push_back(A.getName().size()); 5026*0b57cec5SDimitry Andric Vals.push_back(0); 5027*0b57cec5SDimitry Andric Vals.push_back(0); 5028*0b57cec5SDimitry Andric Vals.push_back(0); 5029*0b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(A)); 5030*0b57cec5SDimitry Andric 5031*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_ALIAS, Vals); 5032*0b57cec5SDimitry Andric Vals.clear(); 5033*0b57cec5SDimitry Andric } 5034*0b57cec5SDimitry Andric 5035*0b57cec5SDimitry Andric // Emit the ifunc information. 5036*0b57cec5SDimitry Andric for (const GlobalIFunc &I : M.ifuncs()) { 5037*0b57cec5SDimitry Andric // IFUNC: [strtab offset, strtab size, 0, 0, 0, linkage] 5038*0b57cec5SDimitry Andric Vals.push_back(StrtabBuilder.add(I.getName())); 5039*0b57cec5SDimitry Andric Vals.push_back(I.getName().size()); 5040*0b57cec5SDimitry Andric Vals.push_back(0); 5041*0b57cec5SDimitry Andric Vals.push_back(0); 5042*0b57cec5SDimitry Andric Vals.push_back(0); 5043*0b57cec5SDimitry Andric Vals.push_back(getEncodedLinkage(I)); 5044*0b57cec5SDimitry Andric 5045*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_IFUNC, Vals); 5046*0b57cec5SDimitry Andric Vals.clear(); 5047*0b57cec5SDimitry Andric } 5048*0b57cec5SDimitry Andric } 5049*0b57cec5SDimitry Andric 5050*0b57cec5SDimitry Andric void ThinLinkBitcodeWriter::write() { 5051*0b57cec5SDimitry Andric Stream.EnterSubblock(bitc::MODULE_BLOCK_ID, 3); 5052*0b57cec5SDimitry Andric 5053*0b57cec5SDimitry Andric writeModuleVersion(); 5054*0b57cec5SDimitry Andric 5055*0b57cec5SDimitry Andric writeSimplifiedModuleInfo(); 5056*0b57cec5SDimitry Andric 5057*0b57cec5SDimitry Andric writePerModuleGlobalValueSummary(); 5058*0b57cec5SDimitry Andric 5059*0b57cec5SDimitry Andric // Write module hash. 5060*0b57cec5SDimitry Andric Stream.EmitRecord(bitc::MODULE_CODE_HASH, ArrayRef<uint32_t>(*ModHash)); 5061*0b57cec5SDimitry Andric 5062*0b57cec5SDimitry Andric Stream.ExitBlock(); 5063*0b57cec5SDimitry Andric } 5064*0b57cec5SDimitry Andric 5065*0b57cec5SDimitry Andric void BitcodeWriter::writeThinLinkBitcode(const Module &M, 5066*0b57cec5SDimitry Andric const ModuleSummaryIndex &Index, 5067*0b57cec5SDimitry Andric const ModuleHash &ModHash) { 5068*0b57cec5SDimitry Andric assert(!WroteStrtab); 5069*0b57cec5SDimitry Andric 5070*0b57cec5SDimitry Andric // The Mods vector is used by irsymtab::build, which requires non-const 5071*0b57cec5SDimitry Andric // Modules in case it needs to materialize metadata. But the bitcode writer 5072*0b57cec5SDimitry Andric // requires that the module is materialized, so we can cast to non-const here, 5073*0b57cec5SDimitry Andric // after checking that it is in fact materialized. 5074*0b57cec5SDimitry Andric assert(M.isMaterialized()); 5075*0b57cec5SDimitry Andric Mods.push_back(const_cast<Module *>(&M)); 5076*0b57cec5SDimitry Andric 5077*0b57cec5SDimitry Andric ThinLinkBitcodeWriter ThinLinkWriter(M, StrtabBuilder, *Stream, Index, 5078*0b57cec5SDimitry Andric ModHash); 5079*0b57cec5SDimitry Andric ThinLinkWriter.write(); 5080*0b57cec5SDimitry Andric } 5081*0b57cec5SDimitry Andric 5082*0b57cec5SDimitry Andric // Write the specified thin link bitcode file to the given raw output stream, 5083*0b57cec5SDimitry Andric // where it will be written in a new bitcode block. This is used when 5084*0b57cec5SDimitry Andric // writing the per-module index file for ThinLTO. 50851fd87a68SDimitry Andric void llvm::writeThinLinkBitcodeToFile(const Module &M, raw_ostream &Out, 5086*0b57cec5SDimitry Andric const ModuleSummaryIndex &Index, 5087*0b57cec5SDimitry Andric const ModuleHash &ModHash) { 5088*0b57cec5SDimitry Andric SmallVector<char, 0> Buffer; 5089*0b57cec5SDimitry Andric Buffer.reserve(256 * 1024); 5090*0b57cec5SDimitry Andric 5091*0b57cec5SDimitry Andric BitcodeWriter Writer(Buffer); 5092*0b57cec5SDimitry Andric Writer.writeThinLinkBitcode(M, Index, ModHash); 5093*0b57cec5SDimitry Andric Writer.writeSymtab(); 5094*0b57cec5SDimitry Andric Writer.writeStrtab(); 5095*0b57cec5SDimitry Andric 5096*0b57cec5SDimitry Andric Out.write((char *)&Buffer.front(), Buffer.size()); 5097*0b57cec5SDimitry Andric } 5098480093f4SDimitry Andric 5099480093f4SDimitry Andric static const char *getSectionNameForBitcode(const Triple &T) { 5100480093f4SDimitry Andric switch (T.getObjectFormat()) { 5101480093f4SDimitry Andric case Triple::MachO: 5102480093f4SDimitry Andric return "__LLVM,__bitcode"; 5103480093f4SDimitry Andric case Triple::COFF: 5104480093f4SDimitry Andric case Triple::ELF: 5105480093f4SDimitry Andric case Triple::Wasm: 5106480093f4SDimitry Andric case Triple::UnknownObjectFormat: 5107480093f4SDimitry Andric return ".llvmbc"; 5108e8d8bef9SDimitry Andric case Triple::GOFF: 5109e8d8bef9SDimitry Andric llvm_unreachable("GOFF is not yet implemented"); 5110e8d8bef9SDimitry Andric break; 511181ad6265SDimitry Andric case Triple::SPIRV: 511281ad6265SDimitry Andric llvm_unreachable("SPIRV is not yet implemented"); 511381ad6265SDimitry Andric break; 5114480093f4SDimitry Andric case Triple::XCOFF: 5115480093f4SDimitry Andric llvm_unreachable("XCOFF is not yet implemented"); 5116480093f4SDimitry Andric break; 511781ad6265SDimitry Andric case Triple::DXContainer: 511881ad6265SDimitry Andric llvm_unreachable("DXContainer is not yet implemented"); 511981ad6265SDimitry Andric break; 5120480093f4SDimitry Andric } 5121480093f4SDimitry Andric llvm_unreachable("Unimplemented ObjectFormatType"); 5122480093f4SDimitry Andric } 5123480093f4SDimitry Andric 5124480093f4SDimitry Andric static const char *getSectionNameForCommandline(const Triple &T) { 5125480093f4SDimitry Andric switch (T.getObjectFormat()) { 5126480093f4SDimitry Andric case Triple::MachO: 5127480093f4SDimitry Andric return "__LLVM,__cmdline"; 5128480093f4SDimitry Andric case Triple::COFF: 5129480093f4SDimitry Andric case Triple::ELF: 5130480093f4SDimitry Andric case Triple::Wasm: 5131480093f4SDimitry Andric case Triple::UnknownObjectFormat: 5132480093f4SDimitry Andric return ".llvmcmd"; 5133e8d8bef9SDimitry Andric case Triple::GOFF: 5134e8d8bef9SDimitry Andric llvm_unreachable("GOFF is not yet implemented"); 5135e8d8bef9SDimitry Andric break; 513681ad6265SDimitry Andric case Triple::SPIRV: 513781ad6265SDimitry Andric llvm_unreachable("SPIRV is not yet implemented"); 513881ad6265SDimitry Andric break; 5139480093f4SDimitry Andric case Triple::XCOFF: 5140480093f4SDimitry Andric llvm_unreachable("XCOFF is not yet implemented"); 5141480093f4SDimitry Andric break; 514281ad6265SDimitry Andric case Triple::DXContainer: 514381ad6265SDimitry Andric llvm_unreachable("DXC is not yet implemented"); 514481ad6265SDimitry Andric break; 5145480093f4SDimitry Andric } 5146480093f4SDimitry Andric llvm_unreachable("Unimplemented ObjectFormatType"); 5147480093f4SDimitry Andric } 5148480093f4SDimitry Andric 51491fd87a68SDimitry Andric void llvm::embedBitcodeInModule(llvm::Module &M, llvm::MemoryBufferRef Buf, 5150e8d8bef9SDimitry Andric bool EmbedBitcode, bool EmbedCmdline, 5151e8d8bef9SDimitry Andric const std::vector<uint8_t> &CmdArgs) { 5152480093f4SDimitry Andric // Save llvm.compiler.used and remove it. 5153480093f4SDimitry Andric SmallVector<Constant *, 2> UsedArray; 5154fe6060f1SDimitry Andric SmallVector<GlobalValue *, 4> UsedGlobals; 5155480093f4SDimitry Andric Type *UsedElementType = Type::getInt8Ty(M.getContext())->getPointerTo(0); 5156480093f4SDimitry Andric GlobalVariable *Used = collectUsedGlobalVariables(M, UsedGlobals, true); 5157480093f4SDimitry Andric for (auto *GV : UsedGlobals) { 5158480093f4SDimitry Andric if (GV->getName() != "llvm.embedded.module" && 5159480093f4SDimitry Andric GV->getName() != "llvm.cmdline") 5160480093f4SDimitry Andric UsedArray.push_back( 5161480093f4SDimitry Andric ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV, UsedElementType)); 5162480093f4SDimitry Andric } 5163480093f4SDimitry Andric if (Used) 5164480093f4SDimitry Andric Used->eraseFromParent(); 5165480093f4SDimitry Andric 5166480093f4SDimitry Andric // Embed the bitcode for the llvm module. 5167480093f4SDimitry Andric std::string Data; 5168480093f4SDimitry Andric ArrayRef<uint8_t> ModuleData; 5169480093f4SDimitry Andric Triple T(M.getTargetTriple()); 5170e8d8bef9SDimitry Andric 5171480093f4SDimitry Andric if (EmbedBitcode) { 5172e8d8bef9SDimitry Andric if (Buf.getBufferSize() == 0 || 5173e8d8bef9SDimitry Andric !isBitcode((const unsigned char *)Buf.getBufferStart(), 5174480093f4SDimitry Andric (const unsigned char *)Buf.getBufferEnd())) { 5175480093f4SDimitry Andric // If the input is LLVM Assembly, bitcode is produced by serializing 5176480093f4SDimitry Andric // the module. Use-lists order need to be preserved in this case. 5177480093f4SDimitry Andric llvm::raw_string_ostream OS(Data); 5178480093f4SDimitry Andric llvm::WriteBitcodeToFile(M, OS, /* ShouldPreserveUseListOrder */ true); 5179480093f4SDimitry Andric ModuleData = 5180480093f4SDimitry Andric ArrayRef<uint8_t>((const uint8_t *)OS.str().data(), OS.str().size()); 5181480093f4SDimitry Andric } else 5182480093f4SDimitry Andric // If the input is LLVM bitcode, write the input byte stream directly. 5183480093f4SDimitry Andric ModuleData = ArrayRef<uint8_t>((const uint8_t *)Buf.getBufferStart(), 5184480093f4SDimitry Andric Buf.getBufferSize()); 5185480093f4SDimitry Andric } 5186480093f4SDimitry Andric llvm::Constant *ModuleConstant = 5187480093f4SDimitry Andric llvm::ConstantDataArray::get(M.getContext(), ModuleData); 5188480093f4SDimitry Andric llvm::GlobalVariable *GV = new llvm::GlobalVariable( 5189480093f4SDimitry Andric M, ModuleConstant->getType(), true, llvm::GlobalValue::PrivateLinkage, 5190480093f4SDimitry Andric ModuleConstant); 5191480093f4SDimitry Andric GV->setSection(getSectionNameForBitcode(T)); 5192e8d8bef9SDimitry Andric // Set alignment to 1 to prevent padding between two contributions from input 5193e8d8bef9SDimitry Andric // sections after linking. 5194e8d8bef9SDimitry Andric GV->setAlignment(Align(1)); 5195480093f4SDimitry Andric UsedArray.push_back( 5196480093f4SDimitry Andric ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV, UsedElementType)); 5197480093f4SDimitry Andric if (llvm::GlobalVariable *Old = 5198480093f4SDimitry Andric M.getGlobalVariable("llvm.embedded.module", true)) { 519981ad6265SDimitry Andric assert(Old->hasZeroLiveUses() && 5200480093f4SDimitry Andric "llvm.embedded.module can only be used once in llvm.compiler.used"); 5201480093f4SDimitry Andric GV->takeName(Old); 5202480093f4SDimitry Andric Old->eraseFromParent(); 5203480093f4SDimitry Andric } else { 5204480093f4SDimitry Andric GV->setName("llvm.embedded.module"); 5205480093f4SDimitry Andric } 5206480093f4SDimitry Andric 5207480093f4SDimitry Andric // Skip if only bitcode needs to be embedded. 5208e8d8bef9SDimitry Andric if (EmbedCmdline) { 5209480093f4SDimitry Andric // Embed command-line options. 5210e8d8bef9SDimitry Andric ArrayRef<uint8_t> CmdData(const_cast<uint8_t *>(CmdArgs.data()), 5211e8d8bef9SDimitry Andric CmdArgs.size()); 5212480093f4SDimitry Andric llvm::Constant *CmdConstant = 5213480093f4SDimitry Andric llvm::ConstantDataArray::get(M.getContext(), CmdData); 5214480093f4SDimitry Andric GV = new llvm::GlobalVariable(M, CmdConstant->getType(), true, 5215480093f4SDimitry Andric llvm::GlobalValue::PrivateLinkage, 5216480093f4SDimitry Andric CmdConstant); 5217480093f4SDimitry Andric GV->setSection(getSectionNameForCommandline(T)); 5218e8d8bef9SDimitry Andric GV->setAlignment(Align(1)); 5219480093f4SDimitry Andric UsedArray.push_back( 5220480093f4SDimitry Andric ConstantExpr::getPointerBitCastOrAddrSpaceCast(GV, UsedElementType)); 5221480093f4SDimitry Andric if (llvm::GlobalVariable *Old = M.getGlobalVariable("llvm.cmdline", true)) { 522281ad6265SDimitry Andric assert(Old->hasZeroLiveUses() && 5223480093f4SDimitry Andric "llvm.cmdline 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.cmdline"); 5228480093f4SDimitry Andric } 5229480093f4SDimitry Andric } 5230480093f4SDimitry Andric 5231480093f4SDimitry Andric if (UsedArray.empty()) 5232480093f4SDimitry Andric return; 5233480093f4SDimitry Andric 5234480093f4SDimitry Andric // Recreate llvm.compiler.used. 5235480093f4SDimitry Andric ArrayType *ATy = ArrayType::get(UsedElementType, UsedArray.size()); 5236480093f4SDimitry Andric auto *NewUsed = new GlobalVariable( 5237480093f4SDimitry Andric M, ATy, false, llvm::GlobalValue::AppendingLinkage, 5238480093f4SDimitry Andric llvm::ConstantArray::get(ATy, UsedArray), "llvm.compiler.used"); 5239480093f4SDimitry Andric NewUsed->setSection("llvm.metadata"); 5240480093f4SDimitry Andric } 5241