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" 1626fa1bf4SReid Kleckner #include "llvm/ADT/StringExtras.h" 172214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/CodeView.h" 182214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/Line.h" 196fcbd7e9SDavid Majnemer #include "llvm/DebugInfo/CodeView/SymbolRecord.h" 20d427383cSZachary Turner #include "llvm/MC/MCAsmLayout.h" 212214ed89SReid Kleckner #include "llvm/MC/MCContext.h" 222214ed89SReid Kleckner #include "llvm/MC/MCObjectStreamer.h" 23408b5e66SDavid Majnemer #include "llvm/MC/MCValue.h" 24858239d5SReid Kleckner #include "llvm/Support/EndianStream.h" 252214ed89SReid Kleckner 262214ed89SReid Kleckner using namespace llvm; 272214ed89SReid Kleckner using namespace llvm::codeview; 282214ed89SReid Kleckner 292214ed89SReid Kleckner CodeViewContext::CodeViewContext() {} 302214ed89SReid Kleckner 312214ed89SReid Kleckner CodeViewContext::~CodeViewContext() { 322214ed89SReid Kleckner // If someone inserted strings into the string table but never actually 332214ed89SReid Kleckner // emitted them somewhere, clean up the fragment. 342214ed89SReid Kleckner if (!InsertedStrTabFragment) 352214ed89SReid Kleckner delete StrTabFragment; 362214ed89SReid Kleckner } 372214ed89SReid Kleckner 382214ed89SReid Kleckner /// This is a valid number for use with .cv_loc if we've already seen a .cv_file 392214ed89SReid Kleckner /// for it. 402214ed89SReid Kleckner bool CodeViewContext::isValidFileNumber(unsigned FileNumber) const { 412214ed89SReid Kleckner unsigned Idx = FileNumber - 1; 4226fa1bf4SReid Kleckner if (Idx < Files.size()) 4326fa1bf4SReid Kleckner return Files[Idx].Assigned; 442214ed89SReid Kleckner return false; 452214ed89SReid Kleckner } 462214ed89SReid Kleckner 4726fa1bf4SReid Kleckner bool CodeViewContext::addFile(MCStreamer &OS, unsigned FileNumber, 4826fa1bf4SReid Kleckner StringRef Filename, 4926fa1bf4SReid Kleckner ArrayRef<uint8_t> ChecksumBytes, 5026fa1bf4SReid Kleckner uint8_t ChecksumKind) { 512214ed89SReid Kleckner assert(FileNumber > 0); 5226fa1bf4SReid Kleckner auto FilenameOffset = addToStringTable(Filename); 5326fa1bf4SReid Kleckner Filename = FilenameOffset.first; 542214ed89SReid Kleckner unsigned Idx = FileNumber - 1; 5526fa1bf4SReid Kleckner if (Idx >= Files.size()) 5626fa1bf4SReid Kleckner Files.resize(Idx + 1); 572214ed89SReid Kleckner 582214ed89SReid Kleckner if (Filename.empty()) 592214ed89SReid Kleckner Filename = "<stdin>"; 602214ed89SReid Kleckner 6126fa1bf4SReid Kleckner if (Files[Idx].Assigned) 622214ed89SReid Kleckner return false; 632214ed89SReid Kleckner 6426fa1bf4SReid Kleckner FilenameOffset = addToStringTable(Filename); 6526fa1bf4SReid Kleckner Filename = FilenameOffset.first; 6626fa1bf4SReid Kleckner unsigned Offset = FilenameOffset.second; 672214ed89SReid Kleckner 6826fa1bf4SReid Kleckner auto ChecksumOffsetSymbol = 6926fa1bf4SReid Kleckner OS.getContext().createTempSymbol("checksum_offset", false); 7026fa1bf4SReid Kleckner Files[Idx].StringTableOffset = Offset; 7126fa1bf4SReid Kleckner Files[Idx].ChecksumTableOffset = ChecksumOffsetSymbol; 7226fa1bf4SReid Kleckner Files[Idx].Assigned = true; 7326fa1bf4SReid Kleckner Files[Idx].Checksum = ChecksumBytes; 7426fa1bf4SReid Kleckner Files[Idx].ChecksumKind = ChecksumKind; 7526fa1bf4SReid Kleckner 762214ed89SReid Kleckner return true; 772214ed89SReid Kleckner } 782214ed89SReid Kleckner 79a9f4cc95SReid Kleckner bool CodeViewContext::recordFunctionId(unsigned FuncId) { 80a9f4cc95SReid Kleckner if (FuncId >= Functions.size()) 81a9f4cc95SReid Kleckner Functions.resize(FuncId + 1); 82a9f4cc95SReid Kleckner 83a9f4cc95SReid Kleckner // Return false if this function info was already allocated. 84a9f4cc95SReid Kleckner if (!Functions[FuncId].isUnallocatedFunctionInfo()) 85a9f4cc95SReid Kleckner return false; 86a9f4cc95SReid Kleckner 87a9f4cc95SReid Kleckner // Mark this as an allocated normal function, and leave the rest alone. 88a9f4cc95SReid Kleckner Functions[FuncId].ParentFuncIdPlusOne = MCCVFunctionInfo::FunctionSentinel; 89a9f4cc95SReid Kleckner return true; 90a9f4cc95SReid Kleckner } 91a9f4cc95SReid Kleckner 92a9f4cc95SReid Kleckner bool CodeViewContext::recordInlinedCallSiteId(unsigned FuncId, unsigned IAFunc, 93a9f4cc95SReid Kleckner unsigned IAFile, unsigned IALine, 94a9f4cc95SReid Kleckner unsigned IACol) { 95a9f4cc95SReid Kleckner if (FuncId >= Functions.size()) 96a9f4cc95SReid Kleckner Functions.resize(FuncId + 1); 97a9f4cc95SReid Kleckner 98a9f4cc95SReid Kleckner // Return false if this function info was already allocated. 99a9f4cc95SReid Kleckner if (!Functions[FuncId].isUnallocatedFunctionInfo()) 100a9f4cc95SReid Kleckner return false; 101a9f4cc95SReid Kleckner 102a9f4cc95SReid Kleckner MCCVFunctionInfo::LineInfo InlinedAt; 103a9f4cc95SReid Kleckner InlinedAt.File = IAFile; 104a9f4cc95SReid Kleckner InlinedAt.Line = IALine; 105a9f4cc95SReid Kleckner InlinedAt.Col = IACol; 106a9f4cc95SReid Kleckner 107a9f4cc95SReid Kleckner // Mark this as an inlined call site and record call site line info. 108a9f4cc95SReid Kleckner MCCVFunctionInfo *Info = &Functions[FuncId]; 109a9f4cc95SReid Kleckner Info->ParentFuncIdPlusOne = IAFunc + 1; 110a9f4cc95SReid Kleckner Info->InlinedAt = InlinedAt; 111a9f4cc95SReid Kleckner 112a9f4cc95SReid Kleckner // Walk up the call chain adding this function id to the InlinedAtMap of all 113a9f4cc95SReid Kleckner // transitive callers until we hit a real function. 114a9f4cc95SReid Kleckner while (Info->isInlinedCallSite()) { 115a9f4cc95SReid Kleckner InlinedAt = Info->InlinedAt; 116a9f4cc95SReid Kleckner Info = getCVFunctionInfo(Info->getParentFuncId()); 117a9f4cc95SReid Kleckner Info->InlinedAtMap[FuncId] = InlinedAt; 118a9f4cc95SReid Kleckner } 119a9f4cc95SReid Kleckner 120a9f4cc95SReid Kleckner return true; 121a9f4cc95SReid Kleckner } 122a9f4cc95SReid Kleckner 1232214ed89SReid Kleckner MCDataFragment *CodeViewContext::getStringTableFragment() { 1242214ed89SReid Kleckner if (!StrTabFragment) { 1252214ed89SReid Kleckner StrTabFragment = new MCDataFragment(); 1262214ed89SReid Kleckner // Start a new string table out with a null byte. 1272214ed89SReid Kleckner StrTabFragment->getContents().push_back('\0'); 1282214ed89SReid Kleckner } 1292214ed89SReid Kleckner return StrTabFragment; 1302214ed89SReid Kleckner } 1312214ed89SReid Kleckner 13226fa1bf4SReid Kleckner std::pair<StringRef, unsigned> CodeViewContext::addToStringTable(StringRef S) { 1332214ed89SReid Kleckner SmallVectorImpl<char> &Contents = getStringTableFragment()->getContents(); 1342214ed89SReid Kleckner auto Insertion = 1352214ed89SReid Kleckner StringTable.insert(std::make_pair(S, unsigned(Contents.size()))); 1362214ed89SReid Kleckner // Return the string from the table, since it is stable. 13726fa1bf4SReid Kleckner std::pair<StringRef, unsigned> Ret = 13826fa1bf4SReid Kleckner std::make_pair(Insertion.first->first(), Insertion.first->second); 1392214ed89SReid Kleckner if (Insertion.second) { 1402214ed89SReid Kleckner // The string map key is always null terminated. 14126fa1bf4SReid Kleckner Contents.append(Ret.first.begin(), Ret.first.end() + 1); 1422214ed89SReid Kleckner } 14326fa1bf4SReid Kleckner return Ret; 1442214ed89SReid Kleckner } 1452214ed89SReid Kleckner 1462214ed89SReid Kleckner unsigned CodeViewContext::getStringTableOffset(StringRef S) { 1472214ed89SReid Kleckner // A string table offset of zero is always the empty string. 1482214ed89SReid Kleckner if (S.empty()) 1492214ed89SReid Kleckner return 0; 1502214ed89SReid Kleckner auto I = StringTable.find(S); 1512214ed89SReid Kleckner assert(I != StringTable.end()); 1522214ed89SReid Kleckner return I->second; 1532214ed89SReid Kleckner } 1542214ed89SReid Kleckner 1552214ed89SReid Kleckner void CodeViewContext::emitStringTable(MCObjectStreamer &OS) { 1562214ed89SReid Kleckner MCContext &Ctx = OS.getContext(); 157a0b44ef6SDavid Blaikie MCSymbol *StringBegin = Ctx.createTempSymbol("strtab_begin", false), 158a0b44ef6SDavid Blaikie *StringEnd = Ctx.createTempSymbol("strtab_end", false); 1592214ed89SReid Kleckner 1608c099fe0SZachary Turner OS.EmitIntValue(unsigned(DebugSubsectionKind::StringTable), 4); 1612214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(StringEnd, StringBegin, 4); 1622214ed89SReid Kleckner OS.EmitLabel(StringBegin); 1632214ed89SReid Kleckner 1642214ed89SReid Kleckner // Put the string table data fragment here, if we haven't already put it 1652214ed89SReid Kleckner // somewhere else. If somebody wants two string tables in their .s file, one 1662214ed89SReid Kleckner // will just be empty. 1672214ed89SReid Kleckner if (!InsertedStrTabFragment) { 1682214ed89SReid Kleckner OS.insert(getStringTableFragment()); 1692214ed89SReid Kleckner InsertedStrTabFragment = true; 1702214ed89SReid Kleckner } 1712214ed89SReid Kleckner 1722214ed89SReid Kleckner OS.EmitValueToAlignment(4, 0); 1732214ed89SReid Kleckner 1742214ed89SReid Kleckner OS.EmitLabel(StringEnd); 1752214ed89SReid Kleckner } 1762214ed89SReid Kleckner 1772214ed89SReid Kleckner void CodeViewContext::emitFileChecksums(MCObjectStreamer &OS) { 178ee641c20SReid Kleckner // Do nothing if there are no file checksums. Microsoft's linker rejects empty 179ee641c20SReid Kleckner // CodeView substreams. 18026fa1bf4SReid Kleckner if (Files.empty()) 181ee641c20SReid Kleckner return; 182ee641c20SReid Kleckner 1832214ed89SReid Kleckner MCContext &Ctx = OS.getContext(); 184a0b44ef6SDavid Blaikie MCSymbol *FileBegin = Ctx.createTempSymbol("filechecksums_begin", false), 185a0b44ef6SDavid Blaikie *FileEnd = Ctx.createTempSymbol("filechecksums_end", false); 1862214ed89SReid Kleckner 1878c099fe0SZachary Turner OS.EmitIntValue(unsigned(DebugSubsectionKind::FileChecksums), 4); 1882214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(FileEnd, FileBegin, 4); 1892214ed89SReid Kleckner OS.EmitLabel(FileBegin); 1902214ed89SReid Kleckner 19126fa1bf4SReid Kleckner unsigned CurrentOffset = 0; 19226fa1bf4SReid Kleckner 1932214ed89SReid Kleckner // Emit an array of FileChecksum entries. We index into this table using the 19426fa1bf4SReid Kleckner // user-provided file number. Each entry may be a variable number of bytes 19526fa1bf4SReid Kleckner // determined by the checksum kind and size. 19626fa1bf4SReid Kleckner for (auto File : Files) { 19726fa1bf4SReid Kleckner OS.EmitAssignment(File.ChecksumTableOffset, 19826fa1bf4SReid Kleckner MCConstantExpr::create(CurrentOffset, Ctx)); 19926fa1bf4SReid Kleckner CurrentOffset += 4; // String table offset. 20026fa1bf4SReid Kleckner if (!File.ChecksumKind) { 20126fa1bf4SReid Kleckner CurrentOffset += 20226fa1bf4SReid Kleckner 4; // One byte each for checksum size and kind, then align to 4 bytes. 20326fa1bf4SReid Kleckner } else { 20426fa1bf4SReid Kleckner CurrentOffset += 2; // One byte each for checksum size and kind. 20526fa1bf4SReid Kleckner CurrentOffset += File.Checksum.size(); 20626fa1bf4SReid Kleckner CurrentOffset = alignTo(CurrentOffset, 4); 20726fa1bf4SReid Kleckner } 20826fa1bf4SReid Kleckner 20926fa1bf4SReid Kleckner OS.EmitIntValue(File.StringTableOffset, 4); 21026fa1bf4SReid Kleckner 21126fa1bf4SReid Kleckner if (!File.ChecksumKind) { 21226fa1bf4SReid Kleckner // There is no checksum. Therefore zero the next two fields and align 21326fa1bf4SReid Kleckner // back to 4 bytes. 2142214ed89SReid Kleckner OS.EmitIntValue(0, 4); 21526fa1bf4SReid Kleckner continue; 21626fa1bf4SReid Kleckner } 21726fa1bf4SReid Kleckner OS.EmitIntValue(static_cast<uint8_t>(File.Checksum.size()), 1); 21826fa1bf4SReid Kleckner OS.EmitIntValue(File.ChecksumKind, 1); 21926fa1bf4SReid Kleckner OS.EmitBytes(toStringRef(File.Checksum)); 22026fa1bf4SReid Kleckner OS.EmitValueToAlignment(4); 2212214ed89SReid Kleckner } 2222214ed89SReid Kleckner 2232214ed89SReid Kleckner OS.EmitLabel(FileEnd); 22426fa1bf4SReid Kleckner 22526fa1bf4SReid Kleckner ChecksumOffsetsAssigned = true; 22626fa1bf4SReid Kleckner } 22726fa1bf4SReid Kleckner 22826fa1bf4SReid Kleckner // Output checksum table offset of the given file number. It is possible that 22926fa1bf4SReid Kleckner // not all files have been registered yet, and so the offset cannot be 23026fa1bf4SReid Kleckner // calculated. In this case a symbol representing the offset is emitted, and 23126fa1bf4SReid Kleckner // the value of this symbol will be fixed up at a later time. 23226fa1bf4SReid Kleckner void CodeViewContext::emitFileChecksumOffset(MCObjectStreamer &OS, 23326fa1bf4SReid Kleckner unsigned FileNo) { 23426fa1bf4SReid Kleckner unsigned Idx = FileNo - 1; 23526fa1bf4SReid Kleckner 23626fa1bf4SReid Kleckner if (Idx >= Files.size()) 23726fa1bf4SReid Kleckner Files.resize(Idx + 1); 23826fa1bf4SReid Kleckner 23926fa1bf4SReid Kleckner if (ChecksumOffsetsAssigned) { 24026fa1bf4SReid Kleckner OS.EmitSymbolValue(Files[Idx].ChecksumTableOffset, 4); 24126fa1bf4SReid Kleckner return; 24226fa1bf4SReid Kleckner } 24326fa1bf4SReid Kleckner 24426fa1bf4SReid Kleckner const MCSymbolRefExpr *SRE = 24526fa1bf4SReid Kleckner MCSymbolRefExpr::create(Files[Idx].ChecksumTableOffset, OS.getContext()); 24626fa1bf4SReid Kleckner 24726fa1bf4SReid Kleckner OS.EmitValueImpl(SRE, 4); 2482214ed89SReid Kleckner } 2492214ed89SReid Kleckner 2502214ed89SReid Kleckner void CodeViewContext::emitLineTableForFunction(MCObjectStreamer &OS, 2512214ed89SReid Kleckner unsigned FuncId, 2522214ed89SReid Kleckner const MCSymbol *FuncBegin, 2532214ed89SReid Kleckner const MCSymbol *FuncEnd) { 2542214ed89SReid Kleckner MCContext &Ctx = OS.getContext(); 255a0b44ef6SDavid Blaikie MCSymbol *LineBegin = Ctx.createTempSymbol("linetable_begin", false), 256a0b44ef6SDavid Blaikie *LineEnd = Ctx.createTempSymbol("linetable_end", false); 2572214ed89SReid Kleckner 2588c099fe0SZachary Turner OS.EmitIntValue(unsigned(DebugSubsectionKind::Lines), 4); 2592214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(LineEnd, LineBegin, 4); 2602214ed89SReid Kleckner OS.EmitLabel(LineBegin); 261f7d84ee6SKeno Fischer OS.EmitCOFFSecRel32(FuncBegin, /*Offset=*/0); 2622214ed89SReid Kleckner OS.EmitCOFFSectionIndex(FuncBegin); 2632214ed89SReid Kleckner 2642214ed89SReid Kleckner // Actual line info. 2656fcbd7e9SDavid Majnemer std::vector<MCCVLineEntry> Locs = getFunctionLineEntries(FuncId); 2662214ed89SReid Kleckner bool HaveColumns = any_of(Locs, [](const MCCVLineEntry &LineEntry) { 2672214ed89SReid Kleckner return LineEntry.getColumn() != 0; 2682214ed89SReid Kleckner }); 2695b6e4e0aSZachary Turner OS.EmitIntValue(HaveColumns ? int(LF_HaveColumns) : 0, 2); 2702214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(FuncEnd, FuncBegin, 4); 2712214ed89SReid Kleckner 2722214ed89SReid Kleckner for (auto I = Locs.begin(), E = Locs.end(); I != E;) { 2732214ed89SReid Kleckner // Emit a file segment for the run of locations that share a file id. 2742214ed89SReid Kleckner unsigned CurFileNum = I->getFileNum(); 2752214ed89SReid Kleckner auto FileSegEnd = 2762214ed89SReid Kleckner std::find_if(I, E, [CurFileNum](const MCCVLineEntry &Loc) { 2772214ed89SReid Kleckner return Loc.getFileNum() != CurFileNum; 2782214ed89SReid Kleckner }); 2792214ed89SReid Kleckner unsigned EntryCount = FileSegEnd - I; 28026fa1bf4SReid Kleckner OS.AddComment( 28126fa1bf4SReid Kleckner "Segment for file '" + 28226fa1bf4SReid Kleckner Twine(getStringTableFragment() 28326fa1bf4SReid Kleckner ->getContents()[Files[CurFileNum - 1].StringTableOffset]) + 2842214ed89SReid Kleckner "' begins"); 28526fa1bf4SReid Kleckner OS.EmitCVFileChecksumOffsetDirective(CurFileNum); 2862214ed89SReid Kleckner OS.EmitIntValue(EntryCount, 4); 2872214ed89SReid Kleckner uint32_t SegmentSize = 12; 2882214ed89SReid Kleckner SegmentSize += 8 * EntryCount; 2892214ed89SReid Kleckner if (HaveColumns) 2902214ed89SReid Kleckner SegmentSize += 4 * EntryCount; 2912214ed89SReid Kleckner OS.EmitIntValue(SegmentSize, 4); 2922214ed89SReid Kleckner 2932214ed89SReid Kleckner for (auto J = I; J != FileSegEnd; ++J) { 2942214ed89SReid Kleckner OS.emitAbsoluteSymbolDiff(J->getLabel(), FuncBegin, 4); 2952214ed89SReid Kleckner unsigned LineData = J->getLine(); 2962214ed89SReid Kleckner if (J->isStmt()) 2976fcbd7e9SDavid Majnemer LineData |= LineInfo::StatementFlag; 2982214ed89SReid Kleckner OS.EmitIntValue(LineData, 4); 2992214ed89SReid Kleckner } 3002214ed89SReid Kleckner if (HaveColumns) { 3012214ed89SReid Kleckner for (auto J = I; J != FileSegEnd; ++J) { 3022214ed89SReid Kleckner OS.EmitIntValue(J->getColumn(), 2); 3032214ed89SReid Kleckner OS.EmitIntValue(0, 2); 3042214ed89SReid Kleckner } 3052214ed89SReid Kleckner } 3062214ed89SReid Kleckner I = FileSegEnd; 3072214ed89SReid Kleckner } 3082214ed89SReid Kleckner OS.EmitLabel(LineEnd); 3092214ed89SReid Kleckner } 3102214ed89SReid Kleckner 3116fcbd7e9SDavid Majnemer static bool compressAnnotation(uint32_t Data, SmallVectorImpl<char> &Buffer) { 3126fcbd7e9SDavid Majnemer if (isUInt<7>(Data)) { 3136fcbd7e9SDavid Majnemer Buffer.push_back(Data); 3146fcbd7e9SDavid Majnemer return true; 3156fcbd7e9SDavid Majnemer } 3166fcbd7e9SDavid Majnemer 3176fcbd7e9SDavid Majnemer if (isUInt<14>(Data)) { 3186fcbd7e9SDavid Majnemer Buffer.push_back((Data >> 8) | 0x80); 3196fcbd7e9SDavid Majnemer Buffer.push_back(Data & 0xff); 3206fcbd7e9SDavid Majnemer return true; 3216fcbd7e9SDavid Majnemer } 3226fcbd7e9SDavid Majnemer 3236fcbd7e9SDavid Majnemer if (isUInt<29>(Data)) { 3246fcbd7e9SDavid Majnemer Buffer.push_back((Data >> 24) | 0xC0); 3256fcbd7e9SDavid Majnemer Buffer.push_back((Data >> 16) & 0xff); 3266fcbd7e9SDavid Majnemer Buffer.push_back((Data >> 8) & 0xff); 3276fcbd7e9SDavid Majnemer Buffer.push_back(Data & 0xff); 3286fcbd7e9SDavid Majnemer return true; 3296fcbd7e9SDavid Majnemer } 3306fcbd7e9SDavid Majnemer 3316fcbd7e9SDavid Majnemer return false; 3326fcbd7e9SDavid Majnemer } 3336fcbd7e9SDavid Majnemer 33463a2846eSZachary Turner static bool compressAnnotation(BinaryAnnotationsOpCode Annotation, 33563a2846eSZachary Turner SmallVectorImpl<char> &Buffer) { 33663a2846eSZachary Turner return compressAnnotation(static_cast<uint32_t>(Annotation), Buffer); 33763a2846eSZachary Turner } 33863a2846eSZachary Turner 3396fcbd7e9SDavid Majnemer static uint32_t encodeSignedNumber(uint32_t Data) { 3406fcbd7e9SDavid Majnemer if (Data >> 31) 3416fcbd7e9SDavid Majnemer return ((-Data) << 1) | 1; 3426fcbd7e9SDavid Majnemer return Data << 1; 3436fcbd7e9SDavid Majnemer } 3446fcbd7e9SDavid Majnemer 345a9f4cc95SReid Kleckner void CodeViewContext::emitInlineLineTableForFunction(MCObjectStreamer &OS, 346a9f4cc95SReid Kleckner unsigned PrimaryFunctionId, 347a9f4cc95SReid Kleckner unsigned SourceFileId, 348a9f4cc95SReid Kleckner unsigned SourceLineNum, 349a9f4cc95SReid Kleckner const MCSymbol *FnStartSym, 350a9f4cc95SReid Kleckner const MCSymbol *FnEndSym) { 3511fcd610cSReid Kleckner // Create and insert a fragment into the current section that will be encoded 3521fcd610cSReid Kleckner // later. 353a9f4cc95SReid Kleckner new MCCVInlineLineTableFragment(PrimaryFunctionId, SourceFileId, 354a9f4cc95SReid Kleckner SourceLineNum, FnStartSym, FnEndSym, 355a9f4cc95SReid Kleckner OS.getCurrentSectionOnly()); 3566fcbd7e9SDavid Majnemer } 3576fcbd7e9SDavid Majnemer 358408b5e66SDavid Majnemer void CodeViewContext::emitDefRange( 359408b5e66SDavid Majnemer MCObjectStreamer &OS, 360408b5e66SDavid Majnemer ArrayRef<std::pair<const MCSymbol *, const MCSymbol *>> Ranges, 361408b5e66SDavid Majnemer StringRef FixedSizePortion) { 362408b5e66SDavid Majnemer // Create and insert a fragment into the current section that will be encoded 363408b5e66SDavid Majnemer // later. 364408b5e66SDavid Majnemer new MCCVDefRangeFragment(Ranges, FixedSizePortion, 365408b5e66SDavid Majnemer OS.getCurrentSectionOnly()); 366408b5e66SDavid Majnemer } 367408b5e66SDavid Majnemer 368cb91e7d3SReid Kleckner static unsigned computeLabelDiff(MCAsmLayout &Layout, const MCSymbol *Begin, 3691fcd610cSReid Kleckner const MCSymbol *End) { 3701fcd610cSReid Kleckner MCContext &Ctx = Layout.getAssembler().getContext(); 3711fcd610cSReid Kleckner MCSymbolRefExpr::VariantKind Variant = MCSymbolRefExpr::VK_None; 3721fcd610cSReid Kleckner const MCExpr *BeginRef = MCSymbolRefExpr::create(Begin, Variant, Ctx), 3731fcd610cSReid Kleckner *EndRef = MCSymbolRefExpr::create(End, Variant, Ctx); 3741fcd610cSReid Kleckner const MCExpr *AddrDelta = 3751fcd610cSReid Kleckner MCBinaryExpr::create(MCBinaryExpr::Sub, EndRef, BeginRef, Ctx); 3761fcd610cSReid Kleckner int64_t Result; 3771fcd610cSReid Kleckner bool Success = AddrDelta->evaluateKnownAbsolute(Result, Layout); 3781fcd610cSReid Kleckner assert(Success && "failed to evaluate label difference as absolute"); 3791fcd610cSReid Kleckner (void)Success; 3801fcd610cSReid Kleckner assert(Result >= 0 && "negative label difference requested"); 3811fcd610cSReid Kleckner assert(Result < UINT_MAX && "label difference greater than 2GB"); 3821fcd610cSReid Kleckner return unsigned(Result); 3831fcd610cSReid Kleckner } 3846fcbd7e9SDavid Majnemer 3851fcd610cSReid Kleckner void CodeViewContext::encodeInlineLineTable(MCAsmLayout &Layout, 3861fcd610cSReid Kleckner MCCVInlineLineTableFragment &Frag) { 3871fcd610cSReid Kleckner size_t LocBegin; 3881fcd610cSReid Kleckner size_t LocEnd; 3891fcd610cSReid Kleckner std::tie(LocBegin, LocEnd) = getLineExtent(Frag.SiteFuncId); 390a9f4cc95SReid Kleckner 391a9f4cc95SReid Kleckner // Include all child inline call sites in our .cv_loc extent. 392a9f4cc95SReid Kleckner MCCVFunctionInfo *SiteInfo = getCVFunctionInfo(Frag.SiteFuncId); 393a9f4cc95SReid Kleckner for (auto &KV : SiteInfo->InlinedAtMap) { 394a9f4cc95SReid Kleckner unsigned ChildId = KV.first; 395a9f4cc95SReid Kleckner auto Extent = getLineExtent(ChildId); 3961fcd610cSReid Kleckner LocBegin = std::min(LocBegin, Extent.first); 3971fcd610cSReid Kleckner LocEnd = std::max(LocEnd, Extent.second); 3981fcd610cSReid Kleckner } 399a9f4cc95SReid Kleckner 4001fcd610cSReid Kleckner if (LocBegin >= LocEnd) 4011fcd610cSReid Kleckner return; 402c9911f28SDavid Majnemer ArrayRef<MCCVLineEntry> Locs = getLinesForExtent(LocBegin, LocEnd); 4031fcd610cSReid Kleckner if (Locs.empty()) 4041fcd610cSReid Kleckner return; 4051fcd610cSReid Kleckner 4061fcd610cSReid Kleckner // Make an artificial start location using the function start and the inlinee 4071fcd610cSReid Kleckner // lines start location information. All deltas start relative to this 4081fcd610cSReid Kleckner // location. 4091fcd610cSReid Kleckner MCCVLineEntry StartLoc(Frag.getFnStartSym(), MCCVLoc(Locs.front())); 4101fcd610cSReid Kleckner StartLoc.setFileNum(Frag.StartFileId); 4111fcd610cSReid Kleckner StartLoc.setLine(Frag.StartLineNum); 412c29b4f07SReid Kleckner bool HaveOpenRange = false; 4131fcd610cSReid Kleckner 414a9f4cc95SReid Kleckner const MCSymbol *LastLabel = Frag.getFnStartSym(); 415a9f4cc95SReid Kleckner MCCVFunctionInfo::LineInfo LastSourceLoc, CurSourceLoc; 416a9f4cc95SReid Kleckner LastSourceLoc.File = Frag.StartFileId; 417a9f4cc95SReid Kleckner LastSourceLoc.Line = Frag.StartLineNum; 418a9f4cc95SReid Kleckner 4191fcd610cSReid Kleckner SmallVectorImpl<char> &Buffer = Frag.getContents(); 4201fcd610cSReid Kleckner Buffer.clear(); // Clear old contents if we went through relaxation. 4211fcd610cSReid Kleckner for (const MCCVLineEntry &Loc : Locs) { 422bb96df60SReid Kleckner // Exit early if our line table would produce an oversized InlineSiteSym 423bb96df60SReid Kleckner // record. Account for the ChangeCodeLength annotation emitted after the 424bb96df60SReid Kleckner // loop ends. 42546225b19SZachary Turner constexpr uint32_t InlineSiteSize = 12; 42646225b19SZachary Turner constexpr uint32_t AnnotationSize = 8; 42746225b19SZachary Turner size_t MaxBufferSize = MaxRecordLength - InlineSiteSize - AnnotationSize; 428bb96df60SReid Kleckner if (Buffer.size() >= MaxBufferSize) 429bb96df60SReid Kleckner break; 430bb96df60SReid Kleckner 431a9f4cc95SReid Kleckner if (Loc.getFunctionId() == Frag.SiteFuncId) { 432a9f4cc95SReid Kleckner CurSourceLoc.File = Loc.getFileNum(); 433a9f4cc95SReid Kleckner CurSourceLoc.Line = Loc.getLine(); 434a9f4cc95SReid Kleckner } else { 435a9f4cc95SReid Kleckner auto I = SiteInfo->InlinedAtMap.find(Loc.getFunctionId()); 436a9f4cc95SReid Kleckner if (I != SiteInfo->InlinedAtMap.end()) { 437a9f4cc95SReid Kleckner // This .cv_loc is from a child inline call site. Use the source 438a9f4cc95SReid Kleckner // location of the inlined call site instead of the .cv_loc directive 439a9f4cc95SReid Kleckner // source location. 440a9f4cc95SReid Kleckner CurSourceLoc = I->second; 441a9f4cc95SReid Kleckner } else { 4421fcd610cSReid Kleckner // We've hit a cv_loc not attributed to this inline call site. Use this 4431fcd610cSReid Kleckner // label to end the PC range. 444c29b4f07SReid Kleckner if (HaveOpenRange) { 445a9f4cc95SReid Kleckner unsigned Length = computeLabelDiff(Layout, LastLabel, Loc.getLabel()); 44663a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeLength, Buffer); 4471fcd610cSReid Kleckner compressAnnotation(Length, Buffer); 448a9f4cc95SReid Kleckner LastLabel = Loc.getLabel(); 4491fcd610cSReid Kleckner } 450c29b4f07SReid Kleckner HaveOpenRange = false; 4511fcd610cSReid Kleckner continue; 4521fcd610cSReid Kleckner } 453a9f4cc95SReid Kleckner } 454c29b4f07SReid Kleckner 455a9f4cc95SReid Kleckner // Skip this .cv_loc if we have an open range and this isn't a meaningful 456a9f4cc95SReid Kleckner // source location update. The current table format does not support column 457a9f4cc95SReid Kleckner // info, so we can skip updates for those. 458a9f4cc95SReid Kleckner if (HaveOpenRange && CurSourceLoc.File == LastSourceLoc.File && 459a9f4cc95SReid Kleckner CurSourceLoc.Line == LastSourceLoc.Line) 460c29b4f07SReid Kleckner continue; 461a9f4cc95SReid Kleckner 462c29b4f07SReid Kleckner HaveOpenRange = true; 4631fcd610cSReid Kleckner 464a9f4cc95SReid Kleckner if (CurSourceLoc.File != LastSourceLoc.File) { 46526fa1bf4SReid Kleckner unsigned FileOffset = static_cast<const MCConstantExpr *>( 46626fa1bf4SReid Kleckner Files[CurSourceLoc.File - 1] 46726fa1bf4SReid Kleckner .ChecksumTableOffset->getVariableValue()) 46826fa1bf4SReid Kleckner ->getValue(); 46963a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeFile, Buffer); 470344078f5SReid Kleckner compressAnnotation(FileOffset, Buffer); 4711fcd610cSReid Kleckner } 4721fcd610cSReid Kleckner 473a9f4cc95SReid Kleckner int LineDelta = CurSourceLoc.Line - LastSourceLoc.Line; 4741fcd610cSReid Kleckner unsigned EncodedLineDelta = encodeSignedNumber(LineDelta); 475a9f4cc95SReid Kleckner unsigned CodeDelta = computeLabelDiff(Layout, LastLabel, Loc.getLabel()); 476a9f4cc95SReid Kleckner if (CodeDelta == 0 && LineDelta != 0) { 47763a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeLineOffset, Buffer); 4781fcd610cSReid Kleckner compressAnnotation(EncodedLineDelta, Buffer); 4791fcd610cSReid Kleckner } else if (EncodedLineDelta < 0x8 && CodeDelta <= 0xf) { 4801fcd610cSReid Kleckner // The ChangeCodeOffsetAndLineOffset combination opcode is used when the 4811fcd610cSReid Kleckner // encoded line delta uses 3 or fewer set bits and the code offset fits 4821fcd610cSReid Kleckner // in one nibble. 4831fcd610cSReid Kleckner unsigned Operand = (EncodedLineDelta << 4) | CodeDelta; 48463a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeOffsetAndLineOffset, 48563a2846eSZachary Turner Buffer); 4861fcd610cSReid Kleckner compressAnnotation(Operand, Buffer); 4871fcd610cSReid Kleckner } else { 4881fcd610cSReid Kleckner // Otherwise use the separate line and code deltas. 489a9f4cc95SReid Kleckner if (LineDelta != 0) { 49063a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeLineOffset, Buffer); 4911fcd610cSReid Kleckner compressAnnotation(EncodedLineDelta, Buffer); 492a9f4cc95SReid Kleckner } 49363a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeOffset, Buffer); 4941fcd610cSReid Kleckner compressAnnotation(CodeDelta, Buffer); 4951fcd610cSReid Kleckner } 4966fcbd7e9SDavid Majnemer 497a9f4cc95SReid Kleckner LastLabel = Loc.getLabel(); 498a9f4cc95SReid Kleckner LastSourceLoc = CurSourceLoc; 4996fcbd7e9SDavid Majnemer } 500c9911f28SDavid Majnemer 501c29b4f07SReid Kleckner assert(HaveOpenRange); 502c9911f28SDavid Majnemer 503c9911f28SDavid Majnemer unsigned EndSymLength = 504a9f4cc95SReid Kleckner computeLabelDiff(Layout, LastLabel, Frag.getFnEndSym()); 505c9911f28SDavid Majnemer unsigned LocAfterLength = ~0U; 506c9911f28SDavid Majnemer ArrayRef<MCCVLineEntry> LocAfter = getLinesForExtent(LocEnd, LocEnd + 1); 507cb91e7d3SReid Kleckner if (!LocAfter.empty()) { 508cb91e7d3SReid Kleckner // Only try to compute this difference if we're in the same section. 509cb91e7d3SReid Kleckner const MCCVLineEntry &Loc = LocAfter[0]; 510*5e102eeeSSam Clegg if (&Loc.getLabel()->getSection() == &LastLabel->getSection()) 511a9f4cc95SReid Kleckner LocAfterLength = computeLabelDiff(Layout, LastLabel, Loc.getLabel()); 512cb91e7d3SReid Kleckner } 513c9911f28SDavid Majnemer 51463a2846eSZachary Turner compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeLength, Buffer); 515c9911f28SDavid Majnemer compressAnnotation(std::min(EndSymLength, LocAfterLength), Buffer); 5166fcbd7e9SDavid Majnemer } 5176fcbd7e9SDavid Majnemer 518408b5e66SDavid Majnemer void CodeViewContext::encodeDefRange(MCAsmLayout &Layout, 519408b5e66SDavid Majnemer MCCVDefRangeFragment &Frag) { 520408b5e66SDavid Majnemer MCContext &Ctx = Layout.getAssembler().getContext(); 521408b5e66SDavid Majnemer SmallVectorImpl<char> &Contents = Frag.getContents(); 522408b5e66SDavid Majnemer Contents.clear(); 523408b5e66SDavid Majnemer SmallVectorImpl<MCFixup> &Fixups = Frag.getFixups(); 524408b5e66SDavid Majnemer Fixups.clear(); 525408b5e66SDavid Majnemer raw_svector_ostream OS(Contents); 526408b5e66SDavid Majnemer 527be82d3ecSReid Kleckner // Compute all the sizes up front. 528be82d3ecSReid Kleckner SmallVector<std::pair<unsigned, unsigned>, 4> GapAndRangeSizes; 529be82d3ecSReid Kleckner const MCSymbol *LastLabel = nullptr; 530408b5e66SDavid Majnemer for (std::pair<const MCSymbol *, const MCSymbol *> Range : Frag.getRanges()) { 531be82d3ecSReid Kleckner unsigned GapSize = 532be82d3ecSReid Kleckner LastLabel ? computeLabelDiff(Layout, LastLabel, Range.first) : 0; 533408b5e66SDavid Majnemer unsigned RangeSize = computeLabelDiff(Layout, Range.first, Range.second); 534be82d3ecSReid Kleckner GapAndRangeSizes.push_back({GapSize, RangeSize}); 535be82d3ecSReid Kleckner LastLabel = Range.second; 536be82d3ecSReid Kleckner } 537be82d3ecSReid Kleckner 538be82d3ecSReid Kleckner // Write down each range where the variable is defined. 539be82d3ecSReid Kleckner for (size_t I = 0, E = Frag.getRanges().size(); I != E;) { 540be82d3ecSReid Kleckner // If the range size of multiple consecutive ranges is under the max, 541be82d3ecSReid Kleckner // combine the ranges and emit some gaps. 542be82d3ecSReid Kleckner const MCSymbol *RangeBegin = Frag.getRanges()[I].first; 543be82d3ecSReid Kleckner unsigned RangeSize = GapAndRangeSizes[I].second; 544be82d3ecSReid Kleckner size_t J = I + 1; 545be82d3ecSReid Kleckner for (; J != E; ++J) { 546be82d3ecSReid Kleckner unsigned GapAndRangeSize = GapAndRangeSizes[J].first + GapAndRangeSizes[J].second; 547be82d3ecSReid Kleckner if (RangeSize + GapAndRangeSize > MaxDefRange) 548be82d3ecSReid Kleckner break; 549be82d3ecSReid Kleckner RangeSize += GapAndRangeSize; 550be82d3ecSReid Kleckner } 551be82d3ecSReid Kleckner unsigned NumGaps = J - I - 1; 552be82d3ecSReid Kleckner 553be82d3ecSReid Kleckner support::endian::Writer<support::little> LEWriter(OS); 554be82d3ecSReid Kleckner 555408b5e66SDavid Majnemer unsigned Bias = 0; 556408b5e66SDavid Majnemer // We must split the range into chunks of MaxDefRange, this is a fundamental 557408b5e66SDavid Majnemer // limitation of the file format. 558408b5e66SDavid Majnemer do { 559408b5e66SDavid Majnemer uint16_t Chunk = std::min((uint32_t)MaxDefRange, RangeSize); 560408b5e66SDavid Majnemer 561be82d3ecSReid Kleckner const MCSymbolRefExpr *SRE = MCSymbolRefExpr::create(RangeBegin, Ctx); 562408b5e66SDavid Majnemer const MCBinaryExpr *BE = 563408b5e66SDavid Majnemer MCBinaryExpr::createAdd(SRE, MCConstantExpr::create(Bias, Ctx), Ctx); 564408b5e66SDavid Majnemer MCValue Res; 565408b5e66SDavid Majnemer BE->evaluateAsRelocatable(Res, &Layout, /*Fixup=*/nullptr); 566408b5e66SDavid Majnemer 567408b5e66SDavid Majnemer // Each record begins with a 2-byte number indicating how large the record 568408b5e66SDavid Majnemer // is. 569408b5e66SDavid Majnemer StringRef FixedSizePortion = Frag.getFixedSizePortion(); 570408b5e66SDavid Majnemer // Our record is a fixed sized prefix and a LocalVariableAddrRange that we 571408b5e66SDavid Majnemer // are artificially constructing. 572be82d3ecSReid Kleckner size_t RecordSize = FixedSizePortion.size() + 573be82d3ecSReid Kleckner sizeof(LocalVariableAddrRange) + 4 * NumGaps; 57411cf053bSReid Kleckner // Write out the record size. 57511cf053bSReid Kleckner LEWriter.write<uint16_t>(RecordSize); 576408b5e66SDavid Majnemer // Write out the fixed size prefix. 577408b5e66SDavid Majnemer OS << FixedSizePortion; 578408b5e66SDavid Majnemer // Make space for a fixup that will eventually have a section relative 579408b5e66SDavid Majnemer // relocation pointing at the offset where the variable becomes live. 580408b5e66SDavid Majnemer Fixups.push_back(MCFixup::create(Contents.size(), BE, FK_SecRel_4)); 58111cf053bSReid Kleckner LEWriter.write<uint32_t>(0); // Fixup for code start. 582408b5e66SDavid Majnemer // Make space for a fixup that will record the section index for the code. 583408b5e66SDavid Majnemer Fixups.push_back(MCFixup::create(Contents.size(), BE, FK_SecRel_2)); 58411cf053bSReid Kleckner LEWriter.write<uint16_t>(0); // Fixup for section index. 585408b5e66SDavid Majnemer // Write down the range's extent. 586be82d3ecSReid Kleckner LEWriter.write<uint16_t>(Chunk); 587408b5e66SDavid Majnemer 588408b5e66SDavid Majnemer // Move on to the next range. 589408b5e66SDavid Majnemer Bias += Chunk; 590408b5e66SDavid Majnemer RangeSize -= Chunk; 591408b5e66SDavid Majnemer } while (RangeSize > 0); 592be82d3ecSReid Kleckner 593be82d3ecSReid Kleckner // Emit the gaps afterwards. 59411cf053bSReid Kleckner assert((NumGaps == 0 || Bias <= MaxDefRange) && 595be82d3ecSReid Kleckner "large ranges should not have gaps"); 596be82d3ecSReid Kleckner unsigned GapStartOffset = GapAndRangeSizes[I].second; 597be82d3ecSReid Kleckner for (++I; I != J; ++I) { 598be82d3ecSReid Kleckner unsigned GapSize, RangeSize; 599be82d3ecSReid Kleckner assert(I < GapAndRangeSizes.size()); 600be82d3ecSReid Kleckner std::tie(GapSize, RangeSize) = GapAndRangeSizes[I]; 601be82d3ecSReid Kleckner LEWriter.write<uint16_t>(GapStartOffset); 60211cf053bSReid Kleckner LEWriter.write<uint16_t>(GapSize); 603be82d3ecSReid Kleckner GapStartOffset += GapSize + RangeSize; 604be82d3ecSReid Kleckner } 605408b5e66SDavid Majnemer } 606408b5e66SDavid Majnemer } 607408b5e66SDavid Majnemer 6082214ed89SReid Kleckner // 6092214ed89SReid Kleckner // This is called when an instruction is assembled into the specified section 6102214ed89SReid Kleckner // and if there is information from the last .cv_loc directive that has yet to have 6112214ed89SReid Kleckner // a line entry made for it is made. 6122214ed89SReid Kleckner // 6132214ed89SReid Kleckner void MCCVLineEntry::Make(MCObjectStreamer *MCOS) { 614a5b1eef8SReid Kleckner CodeViewContext &CVC = MCOS->getContext().getCVContext(); 615a5b1eef8SReid Kleckner if (!CVC.getCVLocSeen()) 6162214ed89SReid Kleckner return; 6172214ed89SReid Kleckner 6182214ed89SReid Kleckner // Create a symbol at in the current section for use in the line entry. 6192214ed89SReid Kleckner MCSymbol *LineSym = MCOS->getContext().createTempSymbol(); 6202214ed89SReid Kleckner // Set the value of the symbol to use for the MCCVLineEntry. 6212214ed89SReid Kleckner MCOS->EmitLabel(LineSym); 6222214ed89SReid Kleckner 6232214ed89SReid Kleckner // Get the current .loc info saved in the context. 624a5b1eef8SReid Kleckner const MCCVLoc &CVLoc = CVC.getCurrentCVLoc(); 6252214ed89SReid Kleckner 6262214ed89SReid Kleckner // Create a (local) line entry with the symbol and the current .loc info. 6272214ed89SReid Kleckner MCCVLineEntry LineEntry(LineSym, CVLoc); 6282214ed89SReid Kleckner 6292214ed89SReid Kleckner // clear CVLocSeen saying the current .loc info is now used. 630a5b1eef8SReid Kleckner CVC.clearCVLocSeen(); 6312214ed89SReid Kleckner 6322214ed89SReid Kleckner // Add the line entry to this section's entries. 633a5b1eef8SReid Kleckner CVC.addLineEntry(LineEntry); 6342214ed89SReid Kleckner } 635