12214ed89SReid Kleckner //===- MCCodeView.h - Machine Code CodeView support -------------*- C++ -*-===// 22214ed89SReid Kleckner // 32214ed89SReid Kleckner // The LLVM Compiler Infrastructure 42214ed89SReid Kleckner // 52214ed89SReid Kleckner // This file is distributed under the University of Illinois Open Source 62214ed89SReid Kleckner // License. See LICENSE.TXT for details. 72214ed89SReid Kleckner // 82214ed89SReid Kleckner //===----------------------------------------------------------------------===// 92214ed89SReid Kleckner // 102214ed89SReid Kleckner // Holds state from .cv_file and .cv_loc directives for later emission. 112214ed89SReid Kleckner // 122214ed89SReid Kleckner //===----------------------------------------------------------------------===// 132214ed89SReid Kleckner 142214ed89SReid Kleckner #include "llvm/MC/MCCodeView.h" 152214ed89SReid Kleckner #include "llvm/ADT/STLExtras.h" 16*26fa1bf4SReid Kleckner #include "llvm/ADT/StringExtras.h" 17264b5d9eSZachary Turner #include "llvm/BinaryFormat/COFF.h" 182214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/CodeView.h" 192214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/Line.h" 206fcbd7e9SDavid Majnemer #include "llvm/DebugInfo/CodeView/SymbolRecord.h" 21d427383cSZachary Turner #include "llvm/MC/MCAsmLayout.h" 222214ed89SReid Kleckner #include "llvm/MC/MCContext.h" 232214ed89SReid Kleckner #include "llvm/MC/MCObjectStreamer.h" 24408b5e66SDavid Majnemer #include "llvm/MC/MCValue.h" 25858239d5SReid Kleckner #include "llvm/Support/EndianStream.h" 262214ed89SReid Kleckner 272214ed89SReid Kleckner using namespace llvm; 282214ed89SReid Kleckner using namespace llvm::codeview; 292214ed89SReid Kleckner 302214ed89SReid Kleckner CodeViewContext::CodeViewContext() {} 312214ed89SReid Kleckner 322214ed89SReid Kleckner CodeViewContext::~CodeViewContext() { 332214ed89SReid Kleckner // If someone inserted strings into the string table but never actually 342214ed89SReid Kleckner // emitted them somewhere, clean up the fragment. 352214ed89SReid Kleckner if (!InsertedStrTabFragment) 362214ed89SReid Kleckner delete StrTabFragment; 372214ed89SReid Kleckner } 382214ed89SReid Kleckner 392214ed89SReid Kleckner /// This is a valid number for use with .cv_loc if we've already seen a .cv_file 402214ed89SReid Kleckner /// for it. 412214ed89SReid Kleckner bool CodeViewContext::isValidFileNumber(unsigned FileNumber) const { 422214ed89SReid Kleckner unsigned Idx = FileNumber - 1; 43*26fa1bf4SReid Kleckner if (Idx < Files.size()) 44*26fa1bf4SReid Kleckner return Files[Idx].Assigned; 452214ed89SReid Kleckner return false; 462214ed89SReid Kleckner } 472214ed89SReid Kleckner 48*26fa1bf4SReid Kleckner bool CodeViewContext::addFile(MCStreamer &OS, unsigned FileNumber, 49*26fa1bf4SReid Kleckner StringRef Filename, 50*26fa1bf4SReid Kleckner ArrayRef<uint8_t> ChecksumBytes, 51*26fa1bf4SReid Kleckner uint8_t ChecksumKind) { 522214ed89SReid Kleckner assert(FileNumber > 0); 53*26fa1bf4SReid Kleckner auto FilenameOffset = addToStringTable(Filename); 54*26fa1bf4SReid Kleckner Filename = FilenameOffset.first; 552214ed89SReid Kleckner unsigned Idx = FileNumber - 1; 56*26fa1bf4SReid Kleckner if (Idx >= Files.size()) 57*26fa1bf4SReid Kleckner Files.resize(Idx + 1); 582214ed89SReid Kleckner 592214ed89SReid Kleckner if (Filename.empty()) 602214ed89SReid Kleckner Filename = "<stdin>"; 612214ed89SReid Kleckner 62*26fa1bf4SReid Kleckner if (Files[Idx].Assigned) 632214ed89SReid Kleckner return false; 642214ed89SReid Kleckner 65*26fa1bf4SReid Kleckner FilenameOffset = addToStringTable(Filename); 66*26fa1bf4SReid Kleckner Filename = FilenameOffset.first; 67*26fa1bf4SReid Kleckner unsigned Offset = FilenameOffset.second; 682214ed89SReid Kleckner 69*26fa1bf4SReid Kleckner auto ChecksumOffsetSymbol = 70*26fa1bf4SReid Kleckner OS.getContext().createTempSymbol("checksum_offset", false); 71*26fa1bf4SReid Kleckner Files[Idx].StringTableOffset = Offset; 72*26fa1bf4SReid Kleckner Files[Idx].ChecksumTableOffset = ChecksumOffsetSymbol; 73*26fa1bf4SReid Kleckner Files[Idx].Assigned = true; 74*26fa1bf4SReid Kleckner Files[Idx].Checksum = ChecksumBytes; 75*26fa1bf4SReid Kleckner Files[Idx].ChecksumKind = ChecksumKind; 76*26fa1bf4SReid Kleckner 772214ed89SReid Kleckner return true; 782214ed89SReid Kleckner } 792214ed89SReid Kleckner 80a9f4cc95SReid Kleckner bool CodeViewContext::recordFunctionId(unsigned FuncId) { 81a9f4cc95SReid Kleckner if (FuncId >= Functions.size()) 82a9f4cc95SReid Kleckner Functions.resize(FuncId + 1); 83a9f4cc95SReid Kleckner 84a9f4cc95SReid Kleckner // Return false if this function info was already allocated. 85a9f4cc95SReid Kleckner if (!Functions[FuncId].isUnallocatedFunctionInfo()) 86a9f4cc95SReid Kleckner return false; 87a9f4cc95SReid Kleckner 88a9f4cc95SReid Kleckner // Mark this as an allocated normal function, and leave the rest alone. 89a9f4cc95SReid Kleckner Functions[FuncId].ParentFuncIdPlusOne = MCCVFunctionInfo::FunctionSentinel; 90a9f4cc95SReid Kleckner return true; 91a9f4cc95SReid Kleckner } 92a9f4cc95SReid Kleckner 93a9f4cc95SReid Kleckner bool CodeViewContext::recordInlinedCallSiteId(unsigned FuncId, unsigned IAFunc, 94a9f4cc95SReid Kleckner unsigned IAFile, unsigned IALine, 95a9f4cc95SReid Kleckner unsigned IACol) { 96a9f4cc95SReid Kleckner if (FuncId >= Functions.size()) 97a9f4cc95SReid Kleckner Functions.resize(FuncId + 1); 98a9f4cc95SReid Kleckner 99a9f4cc95SReid Kleckner // Return false if this function info was already allocated. 100a9f4cc95SReid Kleckner if (!Functions[FuncId].isUnallocatedFunctionInfo()) 101a9f4cc95SReid Kleckner return false; 102a9f4cc95SReid Kleckner 103a9f4cc95SReid Kleckner MCCVFunctionInfo::LineInfo InlinedAt; 104a9f4cc95SReid Kleckner InlinedAt.File = IAFile; 105a9f4cc95SReid Kleckner InlinedAt.Line = IALine; 106a9f4cc95SReid Kleckner InlinedAt.Col = IACol; 107a9f4cc95SReid Kleckner 108a9f4cc95SReid Kleckner // Mark this as an inlined call site and record call site line info. 109a9f4cc95SReid Kleckner MCCVFunctionInfo *Info = &Functions[FuncId]; 110a9f4cc95SReid Kleckner Info->ParentFuncIdPlusOne = IAFunc + 1; 111a9f4cc95SReid Kleckner Info->InlinedAt = InlinedAt; 112a9f4cc95SReid Kleckner 113a9f4cc95SReid Kleckner // Walk up the call chain adding this function id to the InlinedAtMap of all 114a9f4cc95SReid Kleckner // transitive callers until we hit a real function. 115a9f4cc95SReid Kleckner while (Info->isInlinedCallSite()) { 116a9f4cc95SReid Kleckner InlinedAt = Info->InlinedAt; 117a9f4cc95SReid Kleckner Info = getCVFunctionInfo(Info->getParentFuncId()); 118a9f4cc95SReid Kleckner Info->InlinedAtMap[FuncId] = InlinedAt; 119a9f4cc95SReid Kleckner } 120a9f4cc95SReid Kleckner 121a9f4cc95SReid Kleckner return true; 122a9f4cc95SReid Kleckner } 123a9f4cc95SReid Kleckner 1242214ed89SReid Kleckner MCDataFragment *CodeViewContext::getStringTableFragment() { 1252214ed89SReid Kleckner if (!StrTabFragment) { 1262214ed89SReid Kleckner StrTabFragment = new MCDataFragment(); 1272214ed89SReid Kleckner // Start a new string table out with a null byte. 1282214ed89SReid Kleckner StrTabFragment->getContents().push_back('\0'); 1292214ed89SReid Kleckner } 1302214ed89SReid Kleckner return StrTabFragment; 1312214ed89SReid Kleckner } 1322214ed89SReid Kleckner 133*26fa1bf4SReid Kleckner std::pair<StringRef, unsigned> CodeViewContext::addToStringTable(StringRef S) { 1342214ed89SReid Kleckner SmallVectorImpl<char> &Contents = getStringTableFragment()->getContents(); 1352214ed89SReid Kleckner auto Insertion = 1362214ed89SReid Kleckner StringTable.insert(std::make_pair(S, unsigned(Contents.size()))); 1372214ed89SReid Kleckner // Return the string from the table, since it is stable. 138*26fa1bf4SReid Kleckner std::pair<StringRef, unsigned> Ret = 139*26fa1bf4SReid Kleckner std::make_pair(Insertion.first->first(), Insertion.first->second); 1402214ed89SReid Kleckner if (Insertion.second) { 1412214ed89SReid Kleckner // The string map key is always null terminated. 142*26fa1bf4SReid Kleckner Contents.append(Ret.first.begin(), Ret.first.end() + 1); 1432214ed89SReid Kleckner } 144*26fa1bf4SReid Kleckner return Ret; 1452214ed89SReid Kleckner } 1462214ed89SReid Kleckner 1472214ed89SReid Kleckner unsigned CodeViewContext::getStringTableOffset(StringRef S) { 1482214ed89SReid Kleckner // A string table offset of zero is always the empty string. 1492214ed89SReid Kleckner if (S.empty()) 1502214ed89SReid Kleckner return 0; 1512214ed89SReid Kleckner auto I = StringTable.find(S); 1522214ed89SReid Kleckner assert(I != StringTable.end()); 1532214ed89SReid Kleckner return I->second; 1542214ed89SReid Kleckner } 1552214ed89SReid Kleckner 1562214ed89SReid Kleckner void CodeViewContext::emitStringTable(MCObjectStreamer &OS) { 1572214ed89SReid Kleckner MCContext &Ctx = OS.getContext(); 158a0b44ef6SDavid Blaikie MCSymbol *StringBegin = Ctx.createTempSymbol("strtab_begin", false), 159a0b44ef6SDavid Blaikie *StringEnd = Ctx.createTempSymbol("strtab_end", false); 1602214ed89SReid Kleckner 1618c099fe0SZachary Turner OS.EmitIntValue(unsigned(DebugSubsectionKind::StringTable), 4); 1622214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(StringEnd, StringBegin, 4); 1632214ed89SReid Kleckner OS.EmitLabel(StringBegin); 1642214ed89SReid Kleckner 1652214ed89SReid Kleckner // Put the string table data fragment here, if we haven't already put it 1662214ed89SReid Kleckner // somewhere else. If somebody wants two string tables in their .s file, one 1672214ed89SReid Kleckner // will just be empty. 1682214ed89SReid Kleckner if (!InsertedStrTabFragment) { 1692214ed89SReid Kleckner OS.insert(getStringTableFragment()); 1702214ed89SReid Kleckner InsertedStrTabFragment = true; 1712214ed89SReid Kleckner } 1722214ed89SReid Kleckner 1732214ed89SReid Kleckner OS.EmitValueToAlignment(4, 0); 1742214ed89SReid Kleckner 1752214ed89SReid Kleckner OS.EmitLabel(StringEnd); 1762214ed89SReid Kleckner } 1772214ed89SReid Kleckner 1782214ed89SReid Kleckner void CodeViewContext::emitFileChecksums(MCObjectStreamer &OS) { 179ee641c20SReid Kleckner // Do nothing if there are no file checksums. Microsoft's linker rejects empty 180ee641c20SReid Kleckner // CodeView substreams. 181*26fa1bf4SReid Kleckner if (Files.empty()) 182ee641c20SReid Kleckner return; 183ee641c20SReid Kleckner 1842214ed89SReid Kleckner MCContext &Ctx = OS.getContext(); 185a0b44ef6SDavid Blaikie MCSymbol *FileBegin = Ctx.createTempSymbol("filechecksums_begin", false), 186a0b44ef6SDavid Blaikie *FileEnd = Ctx.createTempSymbol("filechecksums_end", false); 1872214ed89SReid Kleckner 1888c099fe0SZachary Turner OS.EmitIntValue(unsigned(DebugSubsectionKind::FileChecksums), 4); 1892214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(FileEnd, FileBegin, 4); 1902214ed89SReid Kleckner OS.EmitLabel(FileBegin); 1912214ed89SReid Kleckner 192*26fa1bf4SReid Kleckner unsigned CurrentOffset = 0; 193*26fa1bf4SReid Kleckner 1942214ed89SReid Kleckner // Emit an array of FileChecksum entries. We index into this table using the 195*26fa1bf4SReid Kleckner // user-provided file number. Each entry may be a variable number of bytes 196*26fa1bf4SReid Kleckner // determined by the checksum kind and size. 197*26fa1bf4SReid Kleckner for (auto File : Files) { 198*26fa1bf4SReid Kleckner OS.EmitAssignment(File.ChecksumTableOffset, 199*26fa1bf4SReid Kleckner MCConstantExpr::create(CurrentOffset, Ctx)); 200*26fa1bf4SReid Kleckner CurrentOffset += 4; // String table offset. 201*26fa1bf4SReid Kleckner if (!File.ChecksumKind) { 202*26fa1bf4SReid Kleckner CurrentOffset += 203*26fa1bf4SReid Kleckner 4; // One byte each for checksum size and kind, then align to 4 bytes. 204*26fa1bf4SReid Kleckner } else { 205*26fa1bf4SReid Kleckner CurrentOffset += 2; // One byte each for checksum size and kind. 206*26fa1bf4SReid Kleckner CurrentOffset += File.Checksum.size(); 207*26fa1bf4SReid Kleckner CurrentOffset = alignTo(CurrentOffset, 4); 208*26fa1bf4SReid Kleckner } 209*26fa1bf4SReid Kleckner 210*26fa1bf4SReid Kleckner OS.EmitIntValue(File.StringTableOffset, 4); 211*26fa1bf4SReid Kleckner 212*26fa1bf4SReid Kleckner if (!File.ChecksumKind) { 213*26fa1bf4SReid Kleckner // There is no checksum. Therefore zero the next two fields and align 214*26fa1bf4SReid Kleckner // back to 4 bytes. 2152214ed89SReid Kleckner OS.EmitIntValue(0, 4); 216*26fa1bf4SReid Kleckner continue; 217*26fa1bf4SReid Kleckner } 218*26fa1bf4SReid Kleckner OS.EmitIntValue(static_cast<uint8_t>(File.Checksum.size()), 1); 219*26fa1bf4SReid Kleckner OS.EmitIntValue(File.ChecksumKind, 1); 220*26fa1bf4SReid Kleckner OS.EmitBytes(toStringRef(File.Checksum)); 221*26fa1bf4SReid Kleckner OS.EmitValueToAlignment(4); 2222214ed89SReid Kleckner } 2232214ed89SReid Kleckner 2242214ed89SReid Kleckner OS.EmitLabel(FileEnd); 225*26fa1bf4SReid Kleckner 226*26fa1bf4SReid Kleckner ChecksumOffsetsAssigned = true; 227*26fa1bf4SReid Kleckner } 228*26fa1bf4SReid Kleckner 229*26fa1bf4SReid Kleckner // Output checksum table offset of the given file number. It is possible that 230*26fa1bf4SReid Kleckner // not all files have been registered yet, and so the offset cannot be 231*26fa1bf4SReid Kleckner // calculated. In this case a symbol representing the offset is emitted, and 232*26fa1bf4SReid Kleckner // the value of this symbol will be fixed up at a later time. 233*26fa1bf4SReid Kleckner void CodeViewContext::emitFileChecksumOffset(MCObjectStreamer &OS, 234*26fa1bf4SReid Kleckner unsigned FileNo) { 235*26fa1bf4SReid Kleckner unsigned Idx = FileNo - 1; 236*26fa1bf4SReid Kleckner 237*26fa1bf4SReid Kleckner if (Idx >= Files.size()) 238*26fa1bf4SReid Kleckner Files.resize(Idx + 1); 239*26fa1bf4SReid Kleckner 240*26fa1bf4SReid Kleckner if (ChecksumOffsetsAssigned) { 241*26fa1bf4SReid Kleckner OS.EmitSymbolValue(Files[Idx].ChecksumTableOffset, 4); 242*26fa1bf4SReid Kleckner return; 243*26fa1bf4SReid Kleckner } 244*26fa1bf4SReid Kleckner 245*26fa1bf4SReid Kleckner const MCSymbolRefExpr *SRE = 246*26fa1bf4SReid Kleckner MCSymbolRefExpr::create(Files[Idx].ChecksumTableOffset, OS.getContext()); 247*26fa1bf4SReid Kleckner 248*26fa1bf4SReid Kleckner OS.EmitValueImpl(SRE, 4); 2492214ed89SReid Kleckner } 2502214ed89SReid Kleckner 2512214ed89SReid Kleckner void CodeViewContext::emitLineTableForFunction(MCObjectStreamer &OS, 2522214ed89SReid Kleckner unsigned FuncId, 2532214ed89SReid Kleckner const MCSymbol *FuncBegin, 2542214ed89SReid Kleckner const MCSymbol *FuncEnd) { 2552214ed89SReid Kleckner MCContext &Ctx = OS.getContext(); 256a0b44ef6SDavid Blaikie MCSymbol *LineBegin = Ctx.createTempSymbol("linetable_begin", false), 257a0b44ef6SDavid Blaikie *LineEnd = Ctx.createTempSymbol("linetable_end", false); 2582214ed89SReid Kleckner 2598c099fe0SZachary Turner OS.EmitIntValue(unsigned(DebugSubsectionKind::Lines), 4); 2602214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(LineEnd, LineBegin, 4); 2612214ed89SReid Kleckner OS.EmitLabel(LineBegin); 262f7d84ee6SKeno Fischer OS.EmitCOFFSecRel32(FuncBegin, /*Offset=*/0); 2632214ed89SReid Kleckner OS.EmitCOFFSectionIndex(FuncBegin); 2642214ed89SReid Kleckner 2652214ed89SReid Kleckner // Actual line info. 2666fcbd7e9SDavid Majnemer std::vector<MCCVLineEntry> Locs = getFunctionLineEntries(FuncId); 2672214ed89SReid Kleckner bool HaveColumns = any_of(Locs, [](const MCCVLineEntry &LineEntry) { 2682214ed89SReid Kleckner return LineEntry.getColumn() != 0; 2692214ed89SReid Kleckner }); 2705b6e4e0aSZachary Turner OS.EmitIntValue(HaveColumns ? int(LF_HaveColumns) : 0, 2); 2712214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(FuncEnd, FuncBegin, 4); 2722214ed89SReid Kleckner 2732214ed89SReid Kleckner for (auto I = Locs.begin(), E = Locs.end(); I != E;) { 2742214ed89SReid Kleckner // Emit a file segment for the run of locations that share a file id. 2752214ed89SReid Kleckner unsigned CurFileNum = I->getFileNum(); 2762214ed89SReid Kleckner auto FileSegEnd = 2772214ed89SReid Kleckner std::find_if(I, E, [CurFileNum](const MCCVLineEntry &Loc) { 2782214ed89SReid Kleckner return Loc.getFileNum() != CurFileNum; 2792214ed89SReid Kleckner }); 2802214ed89SReid Kleckner unsigned EntryCount = FileSegEnd - I; 281*26fa1bf4SReid Kleckner OS.AddComment( 282*26fa1bf4SReid Kleckner "Segment for file '" + 283*26fa1bf4SReid Kleckner Twine(getStringTableFragment() 284*26fa1bf4SReid Kleckner ->getContents()[Files[CurFileNum - 1].StringTableOffset]) + 2852214ed89SReid Kleckner "' begins"); 286*26fa1bf4SReid Kleckner OS.EmitCVFileChecksumOffsetDirective(CurFileNum); 2872214ed89SReid Kleckner OS.EmitIntValue(EntryCount, 4); 2882214ed89SReid Kleckner uint32_t SegmentSize = 12; 2892214ed89SReid Kleckner SegmentSize += 8 * EntryCount; 2902214ed89SReid Kleckner if (HaveColumns) 2912214ed89SReid Kleckner SegmentSize += 4 * EntryCount; 2922214ed89SReid Kleckner OS.EmitIntValue(SegmentSize, 4); 2932214ed89SReid Kleckner 2942214ed89SReid Kleckner for (auto J = I; J != FileSegEnd; ++J) { 2952214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(J->getLabel(), FuncBegin, 4); 2962214ed89SReid Kleckner unsigned LineData = J->getLine(); 2972214ed89SReid Kleckner if (J->isStmt()) 2986fcbd7e9SDavid Majnemer LineData |= LineInfo::StatementFlag; 2992214ed89SReid Kleckner OS.EmitIntValue(LineData, 4); 3002214ed89SReid Kleckner } 3012214ed89SReid Kleckner if (HaveColumns) { 3022214ed89SReid Kleckner for (auto J = I; J != FileSegEnd; ++J) { 3032214ed89SReid Kleckner OS.EmitIntValue(J->getColumn(), 2); 3042214ed89SReid Kleckner OS.EmitIntValue(0, 2); 3052214ed89SReid Kleckner } 3062214ed89SReid Kleckner } 3072214ed89SReid Kleckner I = FileSegEnd; 3082214ed89SReid Kleckner } 3092214ed89SReid Kleckner OS.EmitLabel(LineEnd); 3102214ed89SReid Kleckner } 3112214ed89SReid Kleckner 3126fcbd7e9SDavid Majnemer static bool compressAnnotation(uint32_t Data, SmallVectorImpl<char> &Buffer) { 3136fcbd7e9SDavid Majnemer if (isUInt<7>(Data)) { 3146fcbd7e9SDavid Majnemer Buffer.push_back(Data); 3156fcbd7e9SDavid Majnemer return true; 3166fcbd7e9SDavid Majnemer } 3176fcbd7e9SDavid Majnemer 3186fcbd7e9SDavid Majnemer if (isUInt<14>(Data)) { 3196fcbd7e9SDavid Majnemer Buffer.push_back((Data >> 8) | 0x80); 3206fcbd7e9SDavid Majnemer Buffer.push_back(Data & 0xff); 3216fcbd7e9SDavid Majnemer return true; 3226fcbd7e9SDavid Majnemer } 3236fcbd7e9SDavid Majnemer 3246fcbd7e9SDavid Majnemer if (isUInt<29>(Data)) { 3256fcbd7e9SDavid Majnemer Buffer.push_back((Data >> 24) | 0xC0); 3266fcbd7e9SDavid Majnemer Buffer.push_back((Data >> 16) & 0xff); 3276fcbd7e9SDavid Majnemer Buffer.push_back((Data >> 8) & 0xff); 3286fcbd7e9SDavid Majnemer Buffer.push_back(Data & 0xff); 3296fcbd7e9SDavid Majnemer return true; 3306fcbd7e9SDavid Majnemer } 3316fcbd7e9SDavid Majnemer 3326fcbd7e9SDavid Majnemer return false; 3336fcbd7e9SDavid Majnemer } 3346fcbd7e9SDavid Majnemer 33563a2846eSZachary Turner static bool compressAnnotation(BinaryAnnotationsOpCode Annotation, 33663a2846eSZachary Turner SmallVectorImpl<char> &Buffer) { 33763a2846eSZachary Turner return compressAnnotation(static_cast<uint32_t>(Annotation), Buffer); 33863a2846eSZachary Turner } 33963a2846eSZachary Turner 3406fcbd7e9SDavid Majnemer static uint32_t encodeSignedNumber(uint32_t Data) { 3416fcbd7e9SDavid Majnemer if (Data >> 31) 3426fcbd7e9SDavid Majnemer return ((-Data) << 1) | 1; 3436fcbd7e9SDavid Majnemer return Data << 1; 3446fcbd7e9SDavid Majnemer } 3456fcbd7e9SDavid Majnemer 346a9f4cc95SReid Kleckner void CodeViewContext::emitInlineLineTableForFunction(MCObjectStreamer &OS, 347a9f4cc95SReid Kleckner unsigned PrimaryFunctionId, 348a9f4cc95SReid Kleckner unsigned SourceFileId, 349a9f4cc95SReid Kleckner unsigned SourceLineNum, 350a9f4cc95SReid Kleckner const MCSymbol *FnStartSym, 351a9f4cc95SReid Kleckner const MCSymbol *FnEndSym) { 3521fcd610cSReid Kleckner // Create and insert a fragment into the current section that will be encoded 3531fcd610cSReid Kleckner // later. 354a9f4cc95SReid Kleckner new MCCVInlineLineTableFragment(PrimaryFunctionId, SourceFileId, 355a9f4cc95SReid Kleckner SourceLineNum, FnStartSym, FnEndSym, 356a9f4cc95SReid Kleckner OS.getCurrentSectionOnly()); 3576fcbd7e9SDavid Majnemer } 3586fcbd7e9SDavid Majnemer 359408b5e66SDavid Majnemer void CodeViewContext::emitDefRange( 360408b5e66SDavid Majnemer MCObjectStreamer &OS, 361408b5e66SDavid Majnemer ArrayRef<std::pair<const MCSymbol *, const MCSymbol *>> Ranges, 362408b5e66SDavid Majnemer StringRef FixedSizePortion) { 363408b5e66SDavid Majnemer // Create and insert a fragment into the current section that will be encoded 364408b5e66SDavid Majnemer // later. 365408b5e66SDavid Majnemer new MCCVDefRangeFragment(Ranges, FixedSizePortion, 366408b5e66SDavid Majnemer OS.getCurrentSectionOnly()); 367408b5e66SDavid Majnemer } 368408b5e66SDavid Majnemer 369cb91e7d3SReid Kleckner static unsigned computeLabelDiff(MCAsmLayout &Layout, const MCSymbol *Begin, 3701fcd610cSReid Kleckner const MCSymbol *End) { 3711fcd610cSReid Kleckner MCContext &Ctx = Layout.getAssembler().getContext(); 3721fcd610cSReid Kleckner MCSymbolRefExpr::VariantKind Variant = MCSymbolRefExpr::VK_None; 3731fcd610cSReid Kleckner const MCExpr *BeginRef = MCSymbolRefExpr::create(Begin, Variant, Ctx), 3741fcd610cSReid Kleckner *EndRef = MCSymbolRefExpr::create(End, Variant, Ctx); 3751fcd610cSReid Kleckner const MCExpr *AddrDelta = 3761fcd610cSReid Kleckner MCBinaryExpr::create(MCBinaryExpr::Sub, EndRef, BeginRef, Ctx); 3771fcd610cSReid Kleckner int64_t Result; 3781fcd610cSReid Kleckner bool Success = AddrDelta->evaluateKnownAbsolute(Result, Layout); 3791fcd610cSReid Kleckner assert(Success && "failed to evaluate label difference as absolute"); 3801fcd610cSReid Kleckner (void)Success; 3811fcd610cSReid Kleckner assert(Result >= 0 && "negative label difference requested"); 3821fcd610cSReid Kleckner assert(Result < UINT_MAX && "label difference greater than 2GB"); 3831fcd610cSReid Kleckner return unsigned(Result); 3841fcd610cSReid Kleckner } 3856fcbd7e9SDavid Majnemer 3861fcd610cSReid Kleckner void CodeViewContext::encodeInlineLineTable(MCAsmLayout &Layout, 3871fcd610cSReid Kleckner MCCVInlineLineTableFragment &Frag) { 3881fcd610cSReid Kleckner size_t LocBegin; 3891fcd610cSReid Kleckner size_t LocEnd; 3901fcd610cSReid Kleckner std::tie(LocBegin, LocEnd) = getLineExtent(Frag.SiteFuncId); 391a9f4cc95SReid Kleckner 392a9f4cc95SReid Kleckner // Include all child inline call sites in our .cv_loc extent. 393a9f4cc95SReid Kleckner MCCVFunctionInfo *SiteInfo = getCVFunctionInfo(Frag.SiteFuncId); 394a9f4cc95SReid Kleckner for (auto &KV : SiteInfo->InlinedAtMap) { 395a9f4cc95SReid Kleckner unsigned ChildId = KV.first; 396a9f4cc95SReid Kleckner auto Extent = getLineExtent(ChildId); 3971fcd610cSReid Kleckner LocBegin = std::min(LocBegin, Extent.first); 3981fcd610cSReid Kleckner LocEnd = std::max(LocEnd, Extent.second); 3991fcd610cSReid Kleckner } 400a9f4cc95SReid Kleckner 4011fcd610cSReid Kleckner if (LocBegin >= LocEnd) 4021fcd610cSReid Kleckner return; 403c9911f28SDavid Majnemer ArrayRef<MCCVLineEntry> Locs = getLinesForExtent(LocBegin, LocEnd); 4041fcd610cSReid Kleckner if (Locs.empty()) 4051fcd610cSReid Kleckner return; 4061fcd610cSReid Kleckner 4071fcd610cSReid Kleckner // Make an artificial start location using the function start and the inlinee 4081fcd610cSReid Kleckner // lines start location information. All deltas start relative to this 4091fcd610cSReid Kleckner // location. 4101fcd610cSReid Kleckner MCCVLineEntry StartLoc(Frag.getFnStartSym(), MCCVLoc(Locs.front())); 4111fcd610cSReid Kleckner StartLoc.setFileNum(Frag.StartFileId); 4121fcd610cSReid Kleckner StartLoc.setLine(Frag.StartLineNum); 413c29b4f07SReid Kleckner bool HaveOpenRange = false; 4141fcd610cSReid Kleckner 415a9f4cc95SReid Kleckner const MCSymbol *LastLabel = Frag.getFnStartSym(); 416a9f4cc95SReid Kleckner MCCVFunctionInfo::LineInfo LastSourceLoc, CurSourceLoc; 417a9f4cc95SReid Kleckner LastSourceLoc.File = Frag.StartFileId; 418a9f4cc95SReid Kleckner LastSourceLoc.Line = Frag.StartLineNum; 419a9f4cc95SReid Kleckner 4201fcd610cSReid Kleckner SmallVectorImpl<char> &Buffer = Frag.getContents(); 4211fcd610cSReid Kleckner Buffer.clear(); // Clear old contents if we went through relaxation. 4221fcd610cSReid Kleckner for (const MCCVLineEntry &Loc : Locs) { 423bb96df60SReid Kleckner // Exit early if our line table would produce an oversized InlineSiteSym 424bb96df60SReid Kleckner // record. Account for the ChangeCodeLength annotation emitted after the 425bb96df60SReid Kleckner // loop ends. 42646225b19SZachary Turner constexpr uint32_t InlineSiteSize = 12; 42746225b19SZachary Turner constexpr uint32_t AnnotationSize = 8; 42846225b19SZachary Turner size_t MaxBufferSize = MaxRecordLength - InlineSiteSize - AnnotationSize; 429bb96df60SReid Kleckner if (Buffer.size() >= MaxBufferSize) 430bb96df60SReid Kleckner break; 431bb96df60SReid Kleckner 432a9f4cc95SReid Kleckner if (Loc.getFunctionId() == Frag.SiteFuncId) { 433a9f4cc95SReid Kleckner CurSourceLoc.File = Loc.getFileNum(); 434a9f4cc95SReid Kleckner CurSourceLoc.Line = Loc.getLine(); 435a9f4cc95SReid Kleckner } else { 436a9f4cc95SReid Kleckner auto I = SiteInfo->InlinedAtMap.find(Loc.getFunctionId()); 437a9f4cc95SReid Kleckner if (I != SiteInfo->InlinedAtMap.end()) { 438a9f4cc95SReid Kleckner // This .cv_loc is from a child inline call site. Use the source 439a9f4cc95SReid Kleckner // location of the inlined call site instead of the .cv_loc directive 440a9f4cc95SReid Kleckner // source location. 441a9f4cc95SReid Kleckner CurSourceLoc = I->second; 442a9f4cc95SReid Kleckner } else { 4431fcd610cSReid Kleckner // We've hit a cv_loc not attributed to this inline call site. Use this 4441fcd610cSReid Kleckner // label to end the PC range. 445c29b4f07SReid Kleckner if (HaveOpenRange) { 446a9f4cc95SReid Kleckner unsigned Length = computeLabelDiff(Layout, LastLabel, Loc.getLabel()); 44763a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeLength, Buffer); 4481fcd610cSReid Kleckner compressAnnotation(Length, Buffer); 449a9f4cc95SReid Kleckner LastLabel = Loc.getLabel(); 4501fcd610cSReid Kleckner } 451c29b4f07SReid Kleckner HaveOpenRange = false; 4521fcd610cSReid Kleckner continue; 4531fcd610cSReid Kleckner } 454a9f4cc95SReid Kleckner } 455c29b4f07SReid Kleckner 456a9f4cc95SReid Kleckner // Skip this .cv_loc if we have an open range and this isn't a meaningful 457a9f4cc95SReid Kleckner // source location update. The current table format does not support column 458a9f4cc95SReid Kleckner // info, so we can skip updates for those. 459a9f4cc95SReid Kleckner if (HaveOpenRange && CurSourceLoc.File == LastSourceLoc.File && 460a9f4cc95SReid Kleckner CurSourceLoc.Line == LastSourceLoc.Line) 461c29b4f07SReid Kleckner continue; 462a9f4cc95SReid Kleckner 463c29b4f07SReid Kleckner HaveOpenRange = true; 4641fcd610cSReid Kleckner 465a9f4cc95SReid Kleckner if (CurSourceLoc.File != LastSourceLoc.File) { 466*26fa1bf4SReid Kleckner unsigned FileOffset = static_cast<const MCConstantExpr *>( 467*26fa1bf4SReid Kleckner Files[CurSourceLoc.File - 1] 468*26fa1bf4SReid Kleckner .ChecksumTableOffset->getVariableValue()) 469*26fa1bf4SReid Kleckner ->getValue(); 47063a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeFile, Buffer); 471344078f5SReid Kleckner compressAnnotation(FileOffset, Buffer); 4721fcd610cSReid Kleckner } 4731fcd610cSReid Kleckner 474a9f4cc95SReid Kleckner int LineDelta = CurSourceLoc.Line - LastSourceLoc.Line; 4751fcd610cSReid Kleckner unsigned EncodedLineDelta = encodeSignedNumber(LineDelta); 476a9f4cc95SReid Kleckner unsigned CodeDelta = computeLabelDiff(Layout, LastLabel, Loc.getLabel()); 477a9f4cc95SReid Kleckner if (CodeDelta == 0 && LineDelta != 0) { 47863a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeLineOffset, Buffer); 4791fcd610cSReid Kleckner compressAnnotation(EncodedLineDelta, Buffer); 4801fcd610cSReid Kleckner } else if (EncodedLineDelta < 0x8 && CodeDelta <= 0xf) { 4811fcd610cSReid Kleckner // The ChangeCodeOffsetAndLineOffset combination opcode is used when the 4821fcd610cSReid Kleckner // encoded line delta uses 3 or fewer set bits and the code offset fits 4831fcd610cSReid Kleckner // in one nibble. 4841fcd610cSReid Kleckner unsigned Operand = (EncodedLineDelta << 4) | CodeDelta; 48563a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeOffsetAndLineOffset, 48663a2846eSZachary Turner Buffer); 4871fcd610cSReid Kleckner compressAnnotation(Operand, Buffer); 4881fcd610cSReid Kleckner } else { 4891fcd610cSReid Kleckner // Otherwise use the separate line and code deltas. 490a9f4cc95SReid Kleckner if (LineDelta != 0) { 49163a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeLineOffset, Buffer); 4921fcd610cSReid Kleckner compressAnnotation(EncodedLineDelta, Buffer); 493a9f4cc95SReid Kleckner } 49463a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeOffset, Buffer); 4951fcd610cSReid Kleckner compressAnnotation(CodeDelta, Buffer); 4961fcd610cSReid Kleckner } 4976fcbd7e9SDavid Majnemer 498a9f4cc95SReid Kleckner LastLabel = Loc.getLabel(); 499a9f4cc95SReid Kleckner LastSourceLoc = CurSourceLoc; 5006fcbd7e9SDavid Majnemer } 501c9911f28SDavid Majnemer 502c29b4f07SReid Kleckner assert(HaveOpenRange); 503c9911f28SDavid Majnemer 504c9911f28SDavid Majnemer unsigned EndSymLength = 505a9f4cc95SReid Kleckner computeLabelDiff(Layout, LastLabel, Frag.getFnEndSym()); 506c9911f28SDavid Majnemer unsigned LocAfterLength = ~0U; 507c9911f28SDavid Majnemer ArrayRef<MCCVLineEntry> LocAfter = getLinesForExtent(LocEnd, LocEnd + 1); 508cb91e7d3SReid Kleckner if (!LocAfter.empty()) { 509cb91e7d3SReid Kleckner // Only try to compute this difference if we're in the same section. 510cb91e7d3SReid Kleckner const MCCVLineEntry &Loc = LocAfter[0]; 511a9f4cc95SReid Kleckner if (&Loc.getLabel()->getSection(false) == &LastLabel->getSection(false)) 512a9f4cc95SReid Kleckner LocAfterLength = computeLabelDiff(Layout, LastLabel, Loc.getLabel()); 513cb91e7d3SReid Kleckner } 514c9911f28SDavid Majnemer 51563a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeLength, Buffer); 516c9911f28SDavid Majnemer compressAnnotation(std::min(EndSymLength, LocAfterLength), Buffer); 5176fcbd7e9SDavid Majnemer } 5186fcbd7e9SDavid Majnemer 519408b5e66SDavid Majnemer void CodeViewContext::encodeDefRange(MCAsmLayout &Layout, 520408b5e66SDavid Majnemer MCCVDefRangeFragment &Frag) { 521408b5e66SDavid Majnemer MCContext &Ctx = Layout.getAssembler().getContext(); 522408b5e66SDavid Majnemer SmallVectorImpl<char> &Contents = Frag.getContents(); 523408b5e66SDavid Majnemer Contents.clear(); 524408b5e66SDavid Majnemer SmallVectorImpl<MCFixup> &Fixups = Frag.getFixups(); 525408b5e66SDavid Majnemer Fixups.clear(); 526408b5e66SDavid Majnemer raw_svector_ostream OS(Contents); 527408b5e66SDavid Majnemer 528be82d3ecSReid Kleckner // Compute all the sizes up front. 529be82d3ecSReid Kleckner SmallVector<std::pair<unsigned, unsigned>, 4> GapAndRangeSizes; 530be82d3ecSReid Kleckner const MCSymbol *LastLabel = nullptr; 531408b5e66SDavid Majnemer for (std::pair<const MCSymbol *, const MCSymbol *> Range : Frag.getRanges()) { 532be82d3ecSReid Kleckner unsigned GapSize = 533be82d3ecSReid Kleckner LastLabel ? computeLabelDiff(Layout, LastLabel, Range.first) : 0; 534408b5e66SDavid Majnemer unsigned RangeSize = computeLabelDiff(Layout, Range.first, Range.second); 535be82d3ecSReid Kleckner GapAndRangeSizes.push_back({GapSize, RangeSize}); 536be82d3ecSReid Kleckner LastLabel = Range.second; 537be82d3ecSReid Kleckner } 538be82d3ecSReid Kleckner 539be82d3ecSReid Kleckner // Write down each range where the variable is defined. 540be82d3ecSReid Kleckner for (size_t I = 0, E = Frag.getRanges().size(); I != E;) { 541be82d3ecSReid Kleckner // If the range size of multiple consecutive ranges is under the max, 542be82d3ecSReid Kleckner // combine the ranges and emit some gaps. 543be82d3ecSReid Kleckner const MCSymbol *RangeBegin = Frag.getRanges()[I].first; 544be82d3ecSReid Kleckner unsigned RangeSize = GapAndRangeSizes[I].second; 545be82d3ecSReid Kleckner size_t J = I + 1; 546be82d3ecSReid Kleckner for (; J != E; ++J) { 547be82d3ecSReid Kleckner unsigned GapAndRangeSize = GapAndRangeSizes[J].first + GapAndRangeSizes[J].second; 548be82d3ecSReid Kleckner if (RangeSize + GapAndRangeSize > MaxDefRange) 549be82d3ecSReid Kleckner break; 550be82d3ecSReid Kleckner RangeSize += GapAndRangeSize; 551be82d3ecSReid Kleckner } 552be82d3ecSReid Kleckner unsigned NumGaps = J - I - 1; 553be82d3ecSReid Kleckner 554be82d3ecSReid Kleckner support::endian::Writer<support::little> LEWriter(OS); 555be82d3ecSReid Kleckner 556408b5e66SDavid Majnemer unsigned Bias = 0; 557408b5e66SDavid Majnemer // We must split the range into chunks of MaxDefRange, this is a fundamental 558408b5e66SDavid Majnemer // limitation of the file format. 559408b5e66SDavid Majnemer do { 560408b5e66SDavid Majnemer uint16_t Chunk = std::min((uint32_t)MaxDefRange, RangeSize); 561408b5e66SDavid Majnemer 562be82d3ecSReid Kleckner const MCSymbolRefExpr *SRE = MCSymbolRefExpr::create(RangeBegin, Ctx); 563408b5e66SDavid Majnemer const MCBinaryExpr *BE = 564408b5e66SDavid Majnemer MCBinaryExpr::createAdd(SRE, MCConstantExpr::create(Bias, Ctx), Ctx); 565408b5e66SDavid Majnemer MCValue Res; 566408b5e66SDavid Majnemer BE->evaluateAsRelocatable(Res, &Layout, /*Fixup=*/nullptr); 567408b5e66SDavid Majnemer 568408b5e66SDavid Majnemer // Each record begins with a 2-byte number indicating how large the record 569408b5e66SDavid Majnemer // is. 570408b5e66SDavid Majnemer StringRef FixedSizePortion = Frag.getFixedSizePortion(); 571408b5e66SDavid Majnemer // Our record is a fixed sized prefix and a LocalVariableAddrRange that we 572408b5e66SDavid Majnemer // are artificially constructing. 573be82d3ecSReid Kleckner size_t RecordSize = FixedSizePortion.size() + 574be82d3ecSReid Kleckner sizeof(LocalVariableAddrRange) + 4 * NumGaps; 57511cf053bSReid Kleckner // Write out the record size. 57611cf053bSReid Kleckner LEWriter.write<uint16_t>(RecordSize); 577408b5e66SDavid Majnemer // Write out the fixed size prefix. 578408b5e66SDavid Majnemer OS << FixedSizePortion; 579408b5e66SDavid Majnemer // Make space for a fixup that will eventually have a section relative 580408b5e66SDavid Majnemer // relocation pointing at the offset where the variable becomes live. 581408b5e66SDavid Majnemer Fixups.push_back(MCFixup::create(Contents.size(), BE, FK_SecRel_4)); 58211cf053bSReid Kleckner LEWriter.write<uint32_t>(0); // Fixup for code start. 583408b5e66SDavid Majnemer // Make space for a fixup that will record the section index for the code. 584408b5e66SDavid Majnemer Fixups.push_back(MCFixup::create(Contents.size(), BE, FK_SecRel_2)); 58511cf053bSReid Kleckner LEWriter.write<uint16_t>(0); // Fixup for section index. 586408b5e66SDavid Majnemer // Write down the range's extent. 587be82d3ecSReid Kleckner LEWriter.write<uint16_t>(Chunk); 588408b5e66SDavid Majnemer 589408b5e66SDavid Majnemer // Move on to the next range. 590408b5e66SDavid Majnemer Bias += Chunk; 591408b5e66SDavid Majnemer RangeSize -= Chunk; 592408b5e66SDavid Majnemer } while (RangeSize > 0); 593be82d3ecSReid Kleckner 594be82d3ecSReid Kleckner // Emit the gaps afterwards. 59511cf053bSReid Kleckner assert((NumGaps == 0 || Bias <= MaxDefRange) && 596be82d3ecSReid Kleckner "large ranges should not have gaps"); 597be82d3ecSReid Kleckner unsigned GapStartOffset = GapAndRangeSizes[I].second; 598be82d3ecSReid Kleckner for (++I; I != J; ++I) { 599be82d3ecSReid Kleckner unsigned GapSize, RangeSize; 600be82d3ecSReid Kleckner assert(I < GapAndRangeSizes.size()); 601be82d3ecSReid Kleckner std::tie(GapSize, RangeSize) = GapAndRangeSizes[I]; 602be82d3ecSReid Kleckner LEWriter.write<uint16_t>(GapStartOffset); 60311cf053bSReid Kleckner LEWriter.write<uint16_t>(GapSize); 604be82d3ecSReid Kleckner GapStartOffset += GapSize + RangeSize; 605be82d3ecSReid Kleckner } 606408b5e66SDavid Majnemer } 607408b5e66SDavid Majnemer } 608408b5e66SDavid Majnemer 6092214ed89SReid Kleckner // 6102214ed89SReid Kleckner // This is called when an instruction is assembled into the specified section 6112214ed89SReid Kleckner // and if there is information from the last .cv_loc directive that has yet to have 6122214ed89SReid Kleckner // a line entry made for it is made. 6132214ed89SReid Kleckner // 6142214ed89SReid Kleckner void MCCVLineEntry::Make(MCObjectStreamer *MCOS) { 615a5b1eef8SReid Kleckner CodeViewContext &CVC = MCOS->getContext().getCVContext(); 616a5b1eef8SReid Kleckner if (!CVC.getCVLocSeen()) 6172214ed89SReid Kleckner return; 6182214ed89SReid Kleckner 6192214ed89SReid Kleckner // Create a symbol at in the current section for use in the line entry. 6202214ed89SReid Kleckner MCSymbol *LineSym = MCOS->getContext().createTempSymbol(); 6212214ed89SReid Kleckner // Set the value of the symbol to use for the MCCVLineEntry. 6222214ed89SReid Kleckner MCOS->EmitLabel(LineSym); 6232214ed89SReid Kleckner 6242214ed89SReid Kleckner // Get the current .loc info saved in the context. 625a5b1eef8SReid Kleckner const MCCVLoc &CVLoc = CVC.getCurrentCVLoc(); 6262214ed89SReid Kleckner 6272214ed89SReid Kleckner // Create a (local) line entry with the symbol and the current .loc info. 6282214ed89SReid Kleckner MCCVLineEntry LineEntry(LineSym, CVLoc); 6292214ed89SReid Kleckner 6302214ed89SReid Kleckner // clear CVLocSeen saying the current .loc info is now used. 631a5b1eef8SReid Kleckner CVC.clearCVLocSeen(); 6322214ed89SReid Kleckner 6332214ed89SReid Kleckner // Add the line entry to this section's entries. 634a5b1eef8SReid Kleckner CVC.addLineEntry(LineEntry); 6352214ed89SReid Kleckner } 636