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 
79b5258722SReid Kleckner MCCVFunctionInfo *CodeViewContext::getCVFunctionInfo(unsigned FuncId) {
80b5258722SReid Kleckner   if (FuncId >= Functions.size())
81b5258722SReid Kleckner     return nullptr;
82b5258722SReid Kleckner   if (Functions[FuncId].isUnallocatedFunctionInfo())
83b5258722SReid Kleckner     return nullptr;
84b5258722SReid Kleckner   return &Functions[FuncId];
85b5258722SReid Kleckner }
86b5258722SReid Kleckner 
87a9f4cc95SReid Kleckner bool CodeViewContext::recordFunctionId(unsigned FuncId) {
88a9f4cc95SReid Kleckner   if (FuncId >= Functions.size())
89a9f4cc95SReid Kleckner     Functions.resize(FuncId + 1);
90a9f4cc95SReid Kleckner 
91a9f4cc95SReid Kleckner   // Return false if this function info was already allocated.
92a9f4cc95SReid Kleckner   if (!Functions[FuncId].isUnallocatedFunctionInfo())
93a9f4cc95SReid Kleckner     return false;
94a9f4cc95SReid Kleckner 
95a9f4cc95SReid Kleckner   // Mark this as an allocated normal function, and leave the rest alone.
96a9f4cc95SReid Kleckner   Functions[FuncId].ParentFuncIdPlusOne = MCCVFunctionInfo::FunctionSentinel;
97a9f4cc95SReid Kleckner   return true;
98a9f4cc95SReid Kleckner }
99a9f4cc95SReid Kleckner 
100a9f4cc95SReid Kleckner bool CodeViewContext::recordInlinedCallSiteId(unsigned FuncId, unsigned IAFunc,
101a9f4cc95SReid Kleckner                                               unsigned IAFile, unsigned IALine,
102a9f4cc95SReid Kleckner                                               unsigned IACol) {
103a9f4cc95SReid Kleckner   if (FuncId >= Functions.size())
104a9f4cc95SReid Kleckner     Functions.resize(FuncId + 1);
105a9f4cc95SReid Kleckner 
106a9f4cc95SReid Kleckner   // Return false if this function info was already allocated.
107a9f4cc95SReid Kleckner   if (!Functions[FuncId].isUnallocatedFunctionInfo())
108a9f4cc95SReid Kleckner     return false;
109a9f4cc95SReid Kleckner 
110a9f4cc95SReid Kleckner   MCCVFunctionInfo::LineInfo InlinedAt;
111a9f4cc95SReid Kleckner   InlinedAt.File = IAFile;
112a9f4cc95SReid Kleckner   InlinedAt.Line = IALine;
113a9f4cc95SReid Kleckner   InlinedAt.Col = IACol;
114a9f4cc95SReid Kleckner 
115a9f4cc95SReid Kleckner   // Mark this as an inlined call site and record call site line info.
116a9f4cc95SReid Kleckner   MCCVFunctionInfo *Info = &Functions[FuncId];
117a9f4cc95SReid Kleckner   Info->ParentFuncIdPlusOne = IAFunc + 1;
118a9f4cc95SReid Kleckner   Info->InlinedAt = InlinedAt;
119a9f4cc95SReid Kleckner 
120a9f4cc95SReid Kleckner   // Walk up the call chain adding this function id to the InlinedAtMap of all
121a9f4cc95SReid Kleckner   // transitive callers until we hit a real function.
122a9f4cc95SReid Kleckner   while (Info->isInlinedCallSite()) {
123a9f4cc95SReid Kleckner     InlinedAt = Info->InlinedAt;
124a9f4cc95SReid Kleckner     Info = getCVFunctionInfo(Info->getParentFuncId());
125a9f4cc95SReid Kleckner     Info->InlinedAtMap[FuncId] = InlinedAt;
126a9f4cc95SReid Kleckner   }
127a9f4cc95SReid Kleckner 
128a9f4cc95SReid Kleckner   return true;
129a9f4cc95SReid Kleckner }
130a9f4cc95SReid Kleckner 
1312214ed89SReid Kleckner MCDataFragment *CodeViewContext::getStringTableFragment() {
1322214ed89SReid Kleckner   if (!StrTabFragment) {
1332214ed89SReid Kleckner     StrTabFragment = new MCDataFragment();
1342214ed89SReid Kleckner     // Start a new string table out with a null byte.
1352214ed89SReid Kleckner     StrTabFragment->getContents().push_back('\0');
1362214ed89SReid Kleckner   }
1372214ed89SReid Kleckner   return StrTabFragment;
1382214ed89SReid Kleckner }
1392214ed89SReid Kleckner 
14026fa1bf4SReid Kleckner std::pair<StringRef, unsigned> CodeViewContext::addToStringTable(StringRef S) {
1412214ed89SReid Kleckner   SmallVectorImpl<char> &Contents = getStringTableFragment()->getContents();
1422214ed89SReid Kleckner   auto Insertion =
1432214ed89SReid Kleckner       StringTable.insert(std::make_pair(S, unsigned(Contents.size())));
1442214ed89SReid Kleckner   // Return the string from the table, since it is stable.
14526fa1bf4SReid Kleckner   std::pair<StringRef, unsigned> Ret =
14626fa1bf4SReid Kleckner       std::make_pair(Insertion.first->first(), Insertion.first->second);
1472214ed89SReid Kleckner   if (Insertion.second) {
1482214ed89SReid Kleckner     // The string map key is always null terminated.
14926fa1bf4SReid Kleckner     Contents.append(Ret.first.begin(), Ret.first.end() + 1);
1502214ed89SReid Kleckner   }
15126fa1bf4SReid Kleckner   return Ret;
1522214ed89SReid Kleckner }
1532214ed89SReid Kleckner 
1542214ed89SReid Kleckner unsigned CodeViewContext::getStringTableOffset(StringRef S) {
1552214ed89SReid Kleckner   // A string table offset of zero is always the empty string.
1562214ed89SReid Kleckner   if (S.empty())
1572214ed89SReid Kleckner     return 0;
1582214ed89SReid Kleckner   auto I = StringTable.find(S);
1592214ed89SReid Kleckner   assert(I != StringTable.end());
1602214ed89SReid Kleckner   return I->second;
1612214ed89SReid Kleckner }
1622214ed89SReid Kleckner 
1632214ed89SReid Kleckner void CodeViewContext::emitStringTable(MCObjectStreamer &OS) {
1642214ed89SReid Kleckner   MCContext &Ctx = OS.getContext();
165a0b44ef6SDavid Blaikie   MCSymbol *StringBegin = Ctx.createTempSymbol("strtab_begin", false),
166a0b44ef6SDavid Blaikie            *StringEnd = Ctx.createTempSymbol("strtab_end", false);
1672214ed89SReid Kleckner 
1688c099fe0SZachary Turner   OS.EmitIntValue(unsigned(DebugSubsectionKind::StringTable), 4);
1692214ed89SReid Kleckner   OS.emitAbsoluteSymbolDiff(StringEnd, StringBegin, 4);
1702214ed89SReid Kleckner   OS.EmitLabel(StringBegin);
1712214ed89SReid Kleckner 
1722214ed89SReid Kleckner   // Put the string table data fragment here, if we haven't already put it
1732214ed89SReid Kleckner   // somewhere else. If somebody wants two string tables in their .s file, one
1742214ed89SReid Kleckner   // will just be empty.
1752214ed89SReid Kleckner   if (!InsertedStrTabFragment) {
1762214ed89SReid Kleckner     OS.insert(getStringTableFragment());
1772214ed89SReid Kleckner     InsertedStrTabFragment = true;
1782214ed89SReid Kleckner   }
1792214ed89SReid Kleckner 
1802214ed89SReid Kleckner   OS.EmitValueToAlignment(4, 0);
1812214ed89SReid Kleckner 
1822214ed89SReid Kleckner   OS.EmitLabel(StringEnd);
1832214ed89SReid Kleckner }
1842214ed89SReid Kleckner 
1852214ed89SReid Kleckner void CodeViewContext::emitFileChecksums(MCObjectStreamer &OS) {
186ee641c20SReid Kleckner   // Do nothing if there are no file checksums. Microsoft's linker rejects empty
187ee641c20SReid Kleckner   // CodeView substreams.
18826fa1bf4SReid Kleckner   if (Files.empty())
189ee641c20SReid Kleckner     return;
190ee641c20SReid Kleckner 
1912214ed89SReid Kleckner   MCContext &Ctx = OS.getContext();
192a0b44ef6SDavid Blaikie   MCSymbol *FileBegin = Ctx.createTempSymbol("filechecksums_begin", false),
193a0b44ef6SDavid Blaikie            *FileEnd = Ctx.createTempSymbol("filechecksums_end", false);
1942214ed89SReid Kleckner 
1958c099fe0SZachary Turner   OS.EmitIntValue(unsigned(DebugSubsectionKind::FileChecksums), 4);
1962214ed89SReid Kleckner   OS.emitAbsoluteSymbolDiff(FileEnd, FileBegin, 4);
1972214ed89SReid Kleckner   OS.EmitLabel(FileBegin);
1982214ed89SReid Kleckner 
19926fa1bf4SReid Kleckner   unsigned CurrentOffset = 0;
20026fa1bf4SReid Kleckner 
2012214ed89SReid Kleckner   // Emit an array of FileChecksum entries. We index into this table using the
20226fa1bf4SReid Kleckner   // user-provided file number.  Each entry may be a variable number of bytes
20326fa1bf4SReid Kleckner   // determined by the checksum kind and size.
20426fa1bf4SReid Kleckner   for (auto File : Files) {
20526fa1bf4SReid Kleckner     OS.EmitAssignment(File.ChecksumTableOffset,
20626fa1bf4SReid Kleckner                       MCConstantExpr::create(CurrentOffset, Ctx));
20726fa1bf4SReid Kleckner     CurrentOffset += 4; // String table offset.
20826fa1bf4SReid Kleckner     if (!File.ChecksumKind) {
20926fa1bf4SReid Kleckner       CurrentOffset +=
21026fa1bf4SReid Kleckner           4; // One byte each for checksum size and kind, then align to 4 bytes.
21126fa1bf4SReid Kleckner     } else {
21226fa1bf4SReid Kleckner       CurrentOffset += 2; // One byte each for checksum size and kind.
21326fa1bf4SReid Kleckner       CurrentOffset += File.Checksum.size();
21426fa1bf4SReid Kleckner       CurrentOffset = alignTo(CurrentOffset, 4);
21526fa1bf4SReid Kleckner     }
21626fa1bf4SReid Kleckner 
21726fa1bf4SReid Kleckner     OS.EmitIntValue(File.StringTableOffset, 4);
21826fa1bf4SReid Kleckner 
21926fa1bf4SReid Kleckner     if (!File.ChecksumKind) {
22026fa1bf4SReid Kleckner       // There is no checksum.  Therefore zero the next two fields and align
22126fa1bf4SReid Kleckner       // back to 4 bytes.
2222214ed89SReid Kleckner       OS.EmitIntValue(0, 4);
22326fa1bf4SReid Kleckner       continue;
22426fa1bf4SReid Kleckner     }
22526fa1bf4SReid Kleckner     OS.EmitIntValue(static_cast<uint8_t>(File.Checksum.size()), 1);
22626fa1bf4SReid Kleckner     OS.EmitIntValue(File.ChecksumKind, 1);
22726fa1bf4SReid Kleckner     OS.EmitBytes(toStringRef(File.Checksum));
22826fa1bf4SReid Kleckner     OS.EmitValueToAlignment(4);
2292214ed89SReid Kleckner   }
2302214ed89SReid Kleckner 
2312214ed89SReid Kleckner   OS.EmitLabel(FileEnd);
23226fa1bf4SReid Kleckner 
23326fa1bf4SReid Kleckner   ChecksumOffsetsAssigned = true;
23426fa1bf4SReid Kleckner }
23526fa1bf4SReid Kleckner 
23626fa1bf4SReid Kleckner // Output checksum table offset of the given file number.  It is possible that
23726fa1bf4SReid Kleckner // not all files have been registered yet, and so the offset cannot be
23826fa1bf4SReid Kleckner // calculated.  In this case a symbol representing the offset is emitted, and
23926fa1bf4SReid Kleckner // the value of this symbol will be fixed up at a later time.
24026fa1bf4SReid Kleckner void CodeViewContext::emitFileChecksumOffset(MCObjectStreamer &OS,
24126fa1bf4SReid Kleckner                                              unsigned FileNo) {
24226fa1bf4SReid Kleckner   unsigned Idx = FileNo - 1;
24326fa1bf4SReid Kleckner 
24426fa1bf4SReid Kleckner   if (Idx >= Files.size())
24526fa1bf4SReid Kleckner     Files.resize(Idx + 1);
24626fa1bf4SReid Kleckner 
24726fa1bf4SReid Kleckner   if (ChecksumOffsetsAssigned) {
24826fa1bf4SReid Kleckner     OS.EmitSymbolValue(Files[Idx].ChecksumTableOffset, 4);
24926fa1bf4SReid Kleckner     return;
25026fa1bf4SReid Kleckner   }
25126fa1bf4SReid Kleckner 
25226fa1bf4SReid Kleckner   const MCSymbolRefExpr *SRE =
25326fa1bf4SReid Kleckner       MCSymbolRefExpr::create(Files[Idx].ChecksumTableOffset, OS.getContext());
25426fa1bf4SReid Kleckner 
25526fa1bf4SReid Kleckner   OS.EmitValueImpl(SRE, 4);
2562214ed89SReid Kleckner }
2572214ed89SReid Kleckner 
258b5258722SReid Kleckner void CodeViewContext::addLineEntry(const MCCVLineEntry &LineEntry) {
259b5258722SReid Kleckner   size_t Offset = MCCVLines.size();
260b5258722SReid Kleckner   auto I = MCCVLineStartStop.insert(
261b5258722SReid Kleckner       {LineEntry.getFunctionId(), {Offset, Offset + 1}});
262b5258722SReid Kleckner   if (!I.second)
263b5258722SReid Kleckner     I.first->second.second = Offset + 1;
264b5258722SReid Kleckner   MCCVLines.push_back(LineEntry);
265b5258722SReid Kleckner }
266b5258722SReid Kleckner 
267b5258722SReid Kleckner std::vector<MCCVLineEntry>
268b5258722SReid Kleckner CodeViewContext::getFunctionLineEntries(unsigned FuncId) {
269b5258722SReid Kleckner   std::vector<MCCVLineEntry> FilteredLines;
270b5258722SReid Kleckner   auto I = MCCVLineStartStop.find(FuncId);
2717897a789SReid Kleckner   if (I != MCCVLineStartStop.end()) {
2727897a789SReid Kleckner     MCCVFunctionInfo *SiteInfo = getCVFunctionInfo(FuncId);
273b5258722SReid Kleckner     for (size_t Idx = I->second.first, End = I->second.second; Idx != End;
2747897a789SReid Kleckner          ++Idx) {
2757897a789SReid Kleckner       unsigned LocationFuncId = MCCVLines[Idx].getFunctionId();
2767897a789SReid Kleckner       if (LocationFuncId == FuncId) {
2777897a789SReid Kleckner         // This was a .cv_loc directly for FuncId, so record it.
278b5258722SReid Kleckner         FilteredLines.push_back(MCCVLines[Idx]);
2797897a789SReid Kleckner       } else {
2807897a789SReid Kleckner         // Check if the current location is inlined in this function. If it is,
2817897a789SReid Kleckner         // synthesize a statement .cv_loc at the original inlined call site.
2827897a789SReid Kleckner         auto I = SiteInfo->InlinedAtMap.find(LocationFuncId);
2837897a789SReid Kleckner         if (I != SiteInfo->InlinedAtMap.end()) {
2847897a789SReid Kleckner           MCCVFunctionInfo::LineInfo &IA = I->second;
2857897a789SReid Kleckner           // Only add the location if it differs from the previous location.
2867897a789SReid Kleckner           // Large inlined calls will have many .cv_loc entries and we only need
2877897a789SReid Kleckner           // one line table entry in the parent function.
2887897a789SReid Kleckner           if (FilteredLines.empty() ||
2897897a789SReid Kleckner               FilteredLines.back().getFileNum() != IA.File ||
2907897a789SReid Kleckner               FilteredLines.back().getLine() != IA.Line ||
2917897a789SReid Kleckner               FilteredLines.back().getColumn() != IA.Col) {
2927897a789SReid Kleckner             FilteredLines.push_back(MCCVLineEntry(
2937897a789SReid Kleckner                 MCCVLines[Idx].getLabel(),
2947897a789SReid Kleckner                 MCCVLoc(FuncId, IA.File, IA.Line, IA.Col, false, false)));
2957897a789SReid Kleckner           }
2967897a789SReid Kleckner         }
2977897a789SReid Kleckner       }
2987897a789SReid Kleckner     }
2997897a789SReid Kleckner   }
300b5258722SReid Kleckner   return FilteredLines;
301b5258722SReid Kleckner }
302b5258722SReid Kleckner 
303b5258722SReid Kleckner std::pair<size_t, size_t> CodeViewContext::getLineExtent(unsigned FuncId) {
304b5258722SReid Kleckner   auto I = MCCVLineStartStop.find(FuncId);
305b5258722SReid Kleckner   // Return an empty extent if there are no cv_locs for this function id.
306b5258722SReid Kleckner   if (I == MCCVLineStartStop.end())
307b5258722SReid Kleckner     return {~0ULL, 0};
308b5258722SReid Kleckner   return I->second;
309b5258722SReid Kleckner }
310b5258722SReid Kleckner 
311b5258722SReid Kleckner ArrayRef<MCCVLineEntry> CodeViewContext::getLinesForExtent(size_t L, size_t R) {
312b5258722SReid Kleckner   if (R <= L)
313b5258722SReid Kleckner     return None;
314b5258722SReid Kleckner   if (L >= MCCVLines.size())
315b5258722SReid Kleckner     return None;
316b5258722SReid Kleckner   return makeArrayRef(&MCCVLines[L], R - L);
317b5258722SReid Kleckner }
318b5258722SReid Kleckner 
3192214ed89SReid Kleckner void CodeViewContext::emitLineTableForFunction(MCObjectStreamer &OS,
3202214ed89SReid Kleckner                                                unsigned FuncId,
3212214ed89SReid Kleckner                                                const MCSymbol *FuncBegin,
3222214ed89SReid Kleckner                                                const MCSymbol *FuncEnd) {
3232214ed89SReid Kleckner   MCContext &Ctx = OS.getContext();
324a0b44ef6SDavid Blaikie   MCSymbol *LineBegin = Ctx.createTempSymbol("linetable_begin", false),
325a0b44ef6SDavid Blaikie            *LineEnd = Ctx.createTempSymbol("linetable_end", false);
3262214ed89SReid Kleckner 
3278c099fe0SZachary Turner   OS.EmitIntValue(unsigned(DebugSubsectionKind::Lines), 4);
3282214ed89SReid Kleckner   OS.emitAbsoluteSymbolDiff(LineEnd, LineBegin, 4);
3292214ed89SReid Kleckner   OS.EmitLabel(LineBegin);
330f7d84ee6SKeno Fischer   OS.EmitCOFFSecRel32(FuncBegin, /*Offset=*/0);
3312214ed89SReid Kleckner   OS.EmitCOFFSectionIndex(FuncBegin);
3322214ed89SReid Kleckner 
3332214ed89SReid Kleckner   // Actual line info.
3346fcbd7e9SDavid Majnemer   std::vector<MCCVLineEntry> Locs = getFunctionLineEntries(FuncId);
3352214ed89SReid Kleckner   bool HaveColumns = any_of(Locs, [](const MCCVLineEntry &LineEntry) {
3362214ed89SReid Kleckner     return LineEntry.getColumn() != 0;
3372214ed89SReid Kleckner   });
3385b6e4e0aSZachary Turner   OS.EmitIntValue(HaveColumns ? int(LF_HaveColumns) : 0, 2);
3392214ed89SReid Kleckner   OS.emitAbsoluteSymbolDiff(FuncEnd, FuncBegin, 4);
3402214ed89SReid Kleckner 
3412214ed89SReid Kleckner   for (auto I = Locs.begin(), E = Locs.end(); I != E;) {
3422214ed89SReid Kleckner     // Emit a file segment for the run of locations that share a file id.
3432214ed89SReid Kleckner     unsigned CurFileNum = I->getFileNum();
3442214ed89SReid Kleckner     auto FileSegEnd =
3452214ed89SReid Kleckner         std::find_if(I, E, [CurFileNum](const MCCVLineEntry &Loc) {
3462214ed89SReid Kleckner           return Loc.getFileNum() != CurFileNum;
3472214ed89SReid Kleckner         });
3482214ed89SReid Kleckner     unsigned EntryCount = FileSegEnd - I;
34926fa1bf4SReid Kleckner     OS.AddComment(
35026fa1bf4SReid Kleckner         "Segment for file '" +
35126fa1bf4SReid Kleckner         Twine(getStringTableFragment()
35226fa1bf4SReid Kleckner                   ->getContents()[Files[CurFileNum - 1].StringTableOffset]) +
3532214ed89SReid Kleckner         "' begins");
35426fa1bf4SReid Kleckner     OS.EmitCVFileChecksumOffsetDirective(CurFileNum);
3552214ed89SReid Kleckner     OS.EmitIntValue(EntryCount, 4);
3562214ed89SReid Kleckner     uint32_t SegmentSize = 12;
3572214ed89SReid Kleckner     SegmentSize += 8 * EntryCount;
3582214ed89SReid Kleckner     if (HaveColumns)
3592214ed89SReid Kleckner       SegmentSize += 4 * EntryCount;
3602214ed89SReid Kleckner     OS.EmitIntValue(SegmentSize, 4);
3612214ed89SReid Kleckner 
3622214ed89SReid Kleckner     for (auto J = I; J != FileSegEnd; ++J) {
3632214ed89SReid Kleckner       OS.emitAbsoluteSymbolDiff(J->getLabel(), FuncBegin, 4);
3642214ed89SReid Kleckner       unsigned LineData = J->getLine();
3652214ed89SReid Kleckner       if (J->isStmt())
3666fcbd7e9SDavid Majnemer         LineData |= LineInfo::StatementFlag;
3672214ed89SReid Kleckner       OS.EmitIntValue(LineData, 4);
3682214ed89SReid Kleckner     }
3692214ed89SReid Kleckner     if (HaveColumns) {
3702214ed89SReid Kleckner       for (auto J = I; J != FileSegEnd; ++J) {
3712214ed89SReid Kleckner         OS.EmitIntValue(J->getColumn(), 2);
3722214ed89SReid Kleckner         OS.EmitIntValue(0, 2);
3732214ed89SReid Kleckner       }
3742214ed89SReid Kleckner     }
3752214ed89SReid Kleckner     I = FileSegEnd;
3762214ed89SReid Kleckner   }
3772214ed89SReid Kleckner   OS.EmitLabel(LineEnd);
3782214ed89SReid Kleckner }
3792214ed89SReid Kleckner 
3806fcbd7e9SDavid Majnemer static bool compressAnnotation(uint32_t Data, SmallVectorImpl<char> &Buffer) {
3816fcbd7e9SDavid Majnemer   if (isUInt<7>(Data)) {
3826fcbd7e9SDavid Majnemer     Buffer.push_back(Data);
3836fcbd7e9SDavid Majnemer     return true;
3846fcbd7e9SDavid Majnemer   }
3856fcbd7e9SDavid Majnemer 
3866fcbd7e9SDavid Majnemer   if (isUInt<14>(Data)) {
3876fcbd7e9SDavid Majnemer     Buffer.push_back((Data >> 8) | 0x80);
3886fcbd7e9SDavid Majnemer     Buffer.push_back(Data & 0xff);
3896fcbd7e9SDavid Majnemer     return true;
3906fcbd7e9SDavid Majnemer   }
3916fcbd7e9SDavid Majnemer 
3926fcbd7e9SDavid Majnemer   if (isUInt<29>(Data)) {
3936fcbd7e9SDavid Majnemer     Buffer.push_back((Data >> 24) | 0xC0);
3946fcbd7e9SDavid Majnemer     Buffer.push_back((Data >> 16) & 0xff);
3956fcbd7e9SDavid Majnemer     Buffer.push_back((Data >> 8) & 0xff);
3966fcbd7e9SDavid Majnemer     Buffer.push_back(Data & 0xff);
3976fcbd7e9SDavid Majnemer     return true;
3986fcbd7e9SDavid Majnemer   }
3996fcbd7e9SDavid Majnemer 
4006fcbd7e9SDavid Majnemer   return false;
4016fcbd7e9SDavid Majnemer }
4026fcbd7e9SDavid Majnemer 
40363a2846eSZachary Turner static bool compressAnnotation(BinaryAnnotationsOpCode Annotation,
40463a2846eSZachary Turner                                SmallVectorImpl<char> &Buffer) {
40563a2846eSZachary Turner   return compressAnnotation(static_cast<uint32_t>(Annotation), Buffer);
40663a2846eSZachary Turner }
40763a2846eSZachary Turner 
4086fcbd7e9SDavid Majnemer static uint32_t encodeSignedNumber(uint32_t Data) {
4096fcbd7e9SDavid Majnemer   if (Data >> 31)
4106fcbd7e9SDavid Majnemer     return ((-Data) << 1) | 1;
4116fcbd7e9SDavid Majnemer   return Data << 1;
4126fcbd7e9SDavid Majnemer }
4136fcbd7e9SDavid Majnemer 
414a9f4cc95SReid Kleckner void CodeViewContext::emitInlineLineTableForFunction(MCObjectStreamer &OS,
415a9f4cc95SReid Kleckner                                                      unsigned PrimaryFunctionId,
416a9f4cc95SReid Kleckner                                                      unsigned SourceFileId,
417a9f4cc95SReid Kleckner                                                      unsigned SourceLineNum,
418a9f4cc95SReid Kleckner                                                      const MCSymbol *FnStartSym,
419a9f4cc95SReid Kleckner                                                      const MCSymbol *FnEndSym) {
4201fcd610cSReid Kleckner   // Create and insert a fragment into the current section that will be encoded
4211fcd610cSReid Kleckner   // later.
422a9f4cc95SReid Kleckner   new MCCVInlineLineTableFragment(PrimaryFunctionId, SourceFileId,
423a9f4cc95SReid Kleckner                                   SourceLineNum, FnStartSym, FnEndSym,
424a9f4cc95SReid Kleckner                                   OS.getCurrentSectionOnly());
4256fcbd7e9SDavid Majnemer }
4266fcbd7e9SDavid Majnemer 
427408b5e66SDavid Majnemer void CodeViewContext::emitDefRange(
428408b5e66SDavid Majnemer     MCObjectStreamer &OS,
429408b5e66SDavid Majnemer     ArrayRef<std::pair<const MCSymbol *, const MCSymbol *>> Ranges,
430408b5e66SDavid Majnemer     StringRef FixedSizePortion) {
431408b5e66SDavid Majnemer   // Create and insert a fragment into the current section that will be encoded
432408b5e66SDavid Majnemer   // later.
433408b5e66SDavid Majnemer   new MCCVDefRangeFragment(Ranges, FixedSizePortion,
434408b5e66SDavid Majnemer                            OS.getCurrentSectionOnly());
435408b5e66SDavid Majnemer }
436408b5e66SDavid Majnemer 
437cb91e7d3SReid Kleckner static unsigned computeLabelDiff(MCAsmLayout &Layout, const MCSymbol *Begin,
4381fcd610cSReid Kleckner                                  const MCSymbol *End) {
4391fcd610cSReid Kleckner   MCContext &Ctx = Layout.getAssembler().getContext();
4401fcd610cSReid Kleckner   MCSymbolRefExpr::VariantKind Variant = MCSymbolRefExpr::VK_None;
4411fcd610cSReid Kleckner   const MCExpr *BeginRef = MCSymbolRefExpr::create(Begin, Variant, Ctx),
4421fcd610cSReid Kleckner                *EndRef = MCSymbolRefExpr::create(End, Variant, Ctx);
4431fcd610cSReid Kleckner   const MCExpr *AddrDelta =
4441fcd610cSReid Kleckner       MCBinaryExpr::create(MCBinaryExpr::Sub, EndRef, BeginRef, Ctx);
4451fcd610cSReid Kleckner   int64_t Result;
4461fcd610cSReid Kleckner   bool Success = AddrDelta->evaluateKnownAbsolute(Result, Layout);
4471fcd610cSReid Kleckner   assert(Success && "failed to evaluate label difference as absolute");
4481fcd610cSReid Kleckner   (void)Success;
4491fcd610cSReid Kleckner   assert(Result >= 0 && "negative label difference requested");
4501fcd610cSReid Kleckner   assert(Result < UINT_MAX && "label difference greater than 2GB");
4511fcd610cSReid Kleckner   return unsigned(Result);
4521fcd610cSReid Kleckner }
4536fcbd7e9SDavid Majnemer 
4541fcd610cSReid Kleckner void CodeViewContext::encodeInlineLineTable(MCAsmLayout &Layout,
4551fcd610cSReid Kleckner                                             MCCVInlineLineTableFragment &Frag) {
4561fcd610cSReid Kleckner   size_t LocBegin;
4571fcd610cSReid Kleckner   size_t LocEnd;
4581fcd610cSReid Kleckner   std::tie(LocBegin, LocEnd) = getLineExtent(Frag.SiteFuncId);
459a9f4cc95SReid Kleckner 
460a9f4cc95SReid Kleckner   // Include all child inline call sites in our .cv_loc extent.
461a9f4cc95SReid Kleckner   MCCVFunctionInfo *SiteInfo = getCVFunctionInfo(Frag.SiteFuncId);
462a9f4cc95SReid Kleckner   for (auto &KV : SiteInfo->InlinedAtMap) {
463a9f4cc95SReid Kleckner     unsigned ChildId = KV.first;
464a9f4cc95SReid Kleckner     auto Extent = getLineExtent(ChildId);
4651fcd610cSReid Kleckner     LocBegin = std::min(LocBegin, Extent.first);
4661fcd610cSReid Kleckner     LocEnd = std::max(LocEnd, Extent.second);
4671fcd610cSReid Kleckner   }
468a9f4cc95SReid Kleckner 
4691fcd610cSReid Kleckner   if (LocBegin >= LocEnd)
4701fcd610cSReid Kleckner     return;
471c9911f28SDavid Majnemer   ArrayRef<MCCVLineEntry> Locs = getLinesForExtent(LocBegin, LocEnd);
4721fcd610cSReid Kleckner   if (Locs.empty())
4731fcd610cSReid Kleckner     return;
4741fcd610cSReid Kleckner 
4752c6430feSReid Kleckner   // Check that the locations are all in the same section.
4762c6430feSReid Kleckner #ifndef NDEBUG
4772c6430feSReid Kleckner   const MCSection *FirstSec = &Locs.front().getLabel()->getSection();
4782c6430feSReid Kleckner   for (const MCCVLineEntry &Loc : Locs) {
4792c6430feSReid Kleckner     if (&Loc.getLabel()->getSection() != FirstSec) {
4802c6430feSReid Kleckner       errs() << ".cv_loc " << Loc.getFunctionId() << ' ' << Loc.getFileNum()
4812c6430feSReid Kleckner              << ' ' << Loc.getLine() << ' ' << Loc.getColumn()
4822c6430feSReid Kleckner              << " is in the wrong section\n";
4832c6430feSReid Kleckner       llvm_unreachable(".cv_loc crosses sections");
4842c6430feSReid Kleckner     }
4852c6430feSReid Kleckner   }
4862c6430feSReid Kleckner #endif
4872c6430feSReid Kleckner 
4881fcd610cSReid Kleckner   // Make an artificial start location using the function start and the inlinee
4891fcd610cSReid Kleckner   // lines start location information. All deltas start relative to this
4901fcd610cSReid Kleckner   // location.
4911fcd610cSReid Kleckner   MCCVLineEntry StartLoc(Frag.getFnStartSym(), MCCVLoc(Locs.front()));
4921fcd610cSReid Kleckner   StartLoc.setFileNum(Frag.StartFileId);
4931fcd610cSReid Kleckner   StartLoc.setLine(Frag.StartLineNum);
494c29b4f07SReid Kleckner   bool HaveOpenRange = false;
4951fcd610cSReid Kleckner 
496a9f4cc95SReid Kleckner   const MCSymbol *LastLabel = Frag.getFnStartSym();
497a9f4cc95SReid Kleckner   MCCVFunctionInfo::LineInfo LastSourceLoc, CurSourceLoc;
498a9f4cc95SReid Kleckner   LastSourceLoc.File = Frag.StartFileId;
499a9f4cc95SReid Kleckner   LastSourceLoc.Line = Frag.StartLineNum;
500a9f4cc95SReid Kleckner 
5011fcd610cSReid Kleckner   SmallVectorImpl<char> &Buffer = Frag.getContents();
5021fcd610cSReid Kleckner   Buffer.clear(); // Clear old contents if we went through relaxation.
5031fcd610cSReid Kleckner   for (const MCCVLineEntry &Loc : Locs) {
504bb96df60SReid Kleckner     // Exit early if our line table would produce an oversized InlineSiteSym
505bb96df60SReid Kleckner     // record. Account for the ChangeCodeLength annotation emitted after the
506bb96df60SReid Kleckner     // loop ends.
50746225b19SZachary Turner     constexpr uint32_t InlineSiteSize = 12;
50846225b19SZachary Turner     constexpr uint32_t AnnotationSize = 8;
50946225b19SZachary Turner     size_t MaxBufferSize = MaxRecordLength - InlineSiteSize - AnnotationSize;
510bb96df60SReid Kleckner     if (Buffer.size() >= MaxBufferSize)
511bb96df60SReid Kleckner       break;
512bb96df60SReid Kleckner 
513a9f4cc95SReid Kleckner     if (Loc.getFunctionId() == Frag.SiteFuncId) {
514a9f4cc95SReid Kleckner       CurSourceLoc.File = Loc.getFileNum();
515a9f4cc95SReid Kleckner       CurSourceLoc.Line = Loc.getLine();
516a9f4cc95SReid Kleckner     } else {
517a9f4cc95SReid Kleckner       auto I = SiteInfo->InlinedAtMap.find(Loc.getFunctionId());
518a9f4cc95SReid Kleckner       if (I != SiteInfo->InlinedAtMap.end()) {
519a9f4cc95SReid Kleckner         // This .cv_loc is from a child inline call site. Use the source
520a9f4cc95SReid Kleckner         // location of the inlined call site instead of the .cv_loc directive
521a9f4cc95SReid Kleckner         // source location.
522a9f4cc95SReid Kleckner         CurSourceLoc = I->second;
523a9f4cc95SReid Kleckner       } else {
5241fcd610cSReid Kleckner         // We've hit a cv_loc not attributed to this inline call site. Use this
5251fcd610cSReid Kleckner         // label to end the PC range.
526c29b4f07SReid Kleckner         if (HaveOpenRange) {
527a9f4cc95SReid Kleckner           unsigned Length = computeLabelDiff(Layout, LastLabel, Loc.getLabel());
52863a2846eSZachary Turner           compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeLength, Buffer);
5291fcd610cSReid Kleckner           compressAnnotation(Length, Buffer);
530a9f4cc95SReid Kleckner           LastLabel = Loc.getLabel();
5311fcd610cSReid Kleckner         }
532c29b4f07SReid Kleckner         HaveOpenRange = false;
5331fcd610cSReid Kleckner         continue;
5341fcd610cSReid Kleckner       }
535a9f4cc95SReid Kleckner     }
536c29b4f07SReid Kleckner 
537a9f4cc95SReid Kleckner     // Skip this .cv_loc if we have an open range and this isn't a meaningful
538a9f4cc95SReid Kleckner     // source location update. The current table format does not support column
539a9f4cc95SReid Kleckner     // info, so we can skip updates for those.
540a9f4cc95SReid Kleckner     if (HaveOpenRange && CurSourceLoc.File == LastSourceLoc.File &&
541a9f4cc95SReid Kleckner         CurSourceLoc.Line == LastSourceLoc.Line)
542c29b4f07SReid Kleckner       continue;
543a9f4cc95SReid Kleckner 
544c29b4f07SReid Kleckner     HaveOpenRange = true;
5451fcd610cSReid Kleckner 
546a9f4cc95SReid Kleckner     if (CurSourceLoc.File != LastSourceLoc.File) {
54726fa1bf4SReid Kleckner       unsigned FileOffset = static_cast<const MCConstantExpr *>(
54826fa1bf4SReid Kleckner                                 Files[CurSourceLoc.File - 1]
54926fa1bf4SReid Kleckner                                     .ChecksumTableOffset->getVariableValue())
55026fa1bf4SReid Kleckner                                 ->getValue();
55163a2846eSZachary Turner       compressAnnotation(BinaryAnnotationsOpCode::ChangeFile, Buffer);
552344078f5SReid Kleckner       compressAnnotation(FileOffset, Buffer);
5531fcd610cSReid Kleckner     }
5541fcd610cSReid Kleckner 
555a9f4cc95SReid Kleckner     int LineDelta = CurSourceLoc.Line - LastSourceLoc.Line;
5561fcd610cSReid Kleckner     unsigned EncodedLineDelta = encodeSignedNumber(LineDelta);
557a9f4cc95SReid Kleckner     unsigned CodeDelta = computeLabelDiff(Layout, LastLabel, Loc.getLabel());
558a9f4cc95SReid Kleckner     if (CodeDelta == 0 && LineDelta != 0) {
55963a2846eSZachary Turner       compressAnnotation(BinaryAnnotationsOpCode::ChangeLineOffset, Buffer);
5601fcd610cSReid Kleckner       compressAnnotation(EncodedLineDelta, Buffer);
5611fcd610cSReid Kleckner     } else if (EncodedLineDelta < 0x8 && CodeDelta <= 0xf) {
5621fcd610cSReid Kleckner       // The ChangeCodeOffsetAndLineOffset combination opcode is used when the
5631fcd610cSReid Kleckner       // encoded line delta uses 3 or fewer set bits and the code offset fits
5641fcd610cSReid Kleckner       // in one nibble.
5651fcd610cSReid Kleckner       unsigned Operand = (EncodedLineDelta << 4) | CodeDelta;
56663a2846eSZachary Turner       compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeOffsetAndLineOffset,
56763a2846eSZachary Turner                          Buffer);
5681fcd610cSReid Kleckner       compressAnnotation(Operand, Buffer);
5691fcd610cSReid Kleckner     } else {
5701fcd610cSReid Kleckner       // Otherwise use the separate line and code deltas.
571a9f4cc95SReid Kleckner       if (LineDelta != 0) {
57263a2846eSZachary Turner         compressAnnotation(BinaryAnnotationsOpCode::ChangeLineOffset, Buffer);
5731fcd610cSReid Kleckner         compressAnnotation(EncodedLineDelta, Buffer);
574a9f4cc95SReid Kleckner       }
57563a2846eSZachary Turner       compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeOffset, Buffer);
5761fcd610cSReid Kleckner       compressAnnotation(CodeDelta, Buffer);
5771fcd610cSReid Kleckner     }
5786fcbd7e9SDavid Majnemer 
579a9f4cc95SReid Kleckner     LastLabel = Loc.getLabel();
580a9f4cc95SReid Kleckner     LastSourceLoc = CurSourceLoc;
5816fcbd7e9SDavid Majnemer   }
582c9911f28SDavid Majnemer 
583c29b4f07SReid Kleckner   assert(HaveOpenRange);
584c9911f28SDavid Majnemer 
585c9911f28SDavid Majnemer   unsigned EndSymLength =
586a9f4cc95SReid Kleckner       computeLabelDiff(Layout, LastLabel, Frag.getFnEndSym());
587c9911f28SDavid Majnemer   unsigned LocAfterLength = ~0U;
588c9911f28SDavid Majnemer   ArrayRef<MCCVLineEntry> LocAfter = getLinesForExtent(LocEnd, LocEnd + 1);
589cb91e7d3SReid Kleckner   if (!LocAfter.empty()) {
590cb91e7d3SReid Kleckner     // Only try to compute this difference if we're in the same section.
591cb91e7d3SReid Kleckner     const MCCVLineEntry &Loc = LocAfter[0];
5925e102eeeSSam Clegg     if (&Loc.getLabel()->getSection() == &LastLabel->getSection())
593a9f4cc95SReid Kleckner       LocAfterLength = computeLabelDiff(Layout, LastLabel, Loc.getLabel());
594cb91e7d3SReid Kleckner   }
595c9911f28SDavid Majnemer 
59663a2846eSZachary Turner   compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeLength, Buffer);
597c9911f28SDavid Majnemer   compressAnnotation(std::min(EndSymLength, LocAfterLength), Buffer);
5986fcbd7e9SDavid Majnemer }
5996fcbd7e9SDavid Majnemer 
600408b5e66SDavid Majnemer void CodeViewContext::encodeDefRange(MCAsmLayout &Layout,
601408b5e66SDavid Majnemer                                      MCCVDefRangeFragment &Frag) {
602408b5e66SDavid Majnemer   MCContext &Ctx = Layout.getAssembler().getContext();
603408b5e66SDavid Majnemer   SmallVectorImpl<char> &Contents = Frag.getContents();
604408b5e66SDavid Majnemer   Contents.clear();
605408b5e66SDavid Majnemer   SmallVectorImpl<MCFixup> &Fixups = Frag.getFixups();
606408b5e66SDavid Majnemer   Fixups.clear();
607408b5e66SDavid Majnemer   raw_svector_ostream OS(Contents);
608408b5e66SDavid Majnemer 
609be82d3ecSReid Kleckner   // Compute all the sizes up front.
610be82d3ecSReid Kleckner   SmallVector<std::pair<unsigned, unsigned>, 4> GapAndRangeSizes;
611be82d3ecSReid Kleckner   const MCSymbol *LastLabel = nullptr;
612408b5e66SDavid Majnemer   for (std::pair<const MCSymbol *, const MCSymbol *> Range : Frag.getRanges()) {
613be82d3ecSReid Kleckner     unsigned GapSize =
614be82d3ecSReid Kleckner         LastLabel ? computeLabelDiff(Layout, LastLabel, Range.first) : 0;
615408b5e66SDavid Majnemer     unsigned RangeSize = computeLabelDiff(Layout, Range.first, Range.second);
616be82d3ecSReid Kleckner     GapAndRangeSizes.push_back({GapSize, RangeSize});
617be82d3ecSReid Kleckner     LastLabel = Range.second;
618be82d3ecSReid Kleckner   }
619be82d3ecSReid Kleckner 
620be82d3ecSReid Kleckner   // Write down each range where the variable is defined.
621be82d3ecSReid Kleckner   for (size_t I = 0, E = Frag.getRanges().size(); I != E;) {
622be82d3ecSReid Kleckner     // If the range size of multiple consecutive ranges is under the max,
623be82d3ecSReid Kleckner     // combine the ranges and emit some gaps.
624be82d3ecSReid Kleckner     const MCSymbol *RangeBegin = Frag.getRanges()[I].first;
625be82d3ecSReid Kleckner     unsigned RangeSize = GapAndRangeSizes[I].second;
626be82d3ecSReid Kleckner     size_t J = I + 1;
627be82d3ecSReid Kleckner     for (; J != E; ++J) {
628be82d3ecSReid Kleckner       unsigned GapAndRangeSize = GapAndRangeSizes[J].first + GapAndRangeSizes[J].second;
629be82d3ecSReid Kleckner       if (RangeSize + GapAndRangeSize > MaxDefRange)
630be82d3ecSReid Kleckner         break;
631be82d3ecSReid Kleckner       RangeSize += GapAndRangeSize;
632be82d3ecSReid Kleckner     }
633be82d3ecSReid Kleckner     unsigned NumGaps = J - I - 1;
634be82d3ecSReid Kleckner 
635*e3f65297SPeter Collingbourne     support::endian::Writer LEWriter(OS, support::little);
636be82d3ecSReid Kleckner 
637408b5e66SDavid Majnemer     unsigned Bias = 0;
638408b5e66SDavid Majnemer     // We must split the range into chunks of MaxDefRange, this is a fundamental
639408b5e66SDavid Majnemer     // limitation of the file format.
640408b5e66SDavid Majnemer     do {
641408b5e66SDavid Majnemer       uint16_t Chunk = std::min((uint32_t)MaxDefRange, RangeSize);
642408b5e66SDavid Majnemer 
643be82d3ecSReid Kleckner       const MCSymbolRefExpr *SRE = MCSymbolRefExpr::create(RangeBegin, Ctx);
644408b5e66SDavid Majnemer       const MCBinaryExpr *BE =
645408b5e66SDavid Majnemer           MCBinaryExpr::createAdd(SRE, MCConstantExpr::create(Bias, Ctx), Ctx);
646408b5e66SDavid Majnemer       MCValue Res;
647408b5e66SDavid Majnemer       BE->evaluateAsRelocatable(Res, &Layout, /*Fixup=*/nullptr);
648408b5e66SDavid Majnemer 
649408b5e66SDavid Majnemer       // Each record begins with a 2-byte number indicating how large the record
650408b5e66SDavid Majnemer       // is.
651408b5e66SDavid Majnemer       StringRef FixedSizePortion = Frag.getFixedSizePortion();
652408b5e66SDavid Majnemer       // Our record is a fixed sized prefix and a LocalVariableAddrRange that we
653408b5e66SDavid Majnemer       // are artificially constructing.
654be82d3ecSReid Kleckner       size_t RecordSize = FixedSizePortion.size() +
655be82d3ecSReid Kleckner                           sizeof(LocalVariableAddrRange) + 4 * NumGaps;
65611cf053bSReid Kleckner       // Write out the record size.
65711cf053bSReid Kleckner       LEWriter.write<uint16_t>(RecordSize);
658408b5e66SDavid Majnemer       // Write out the fixed size prefix.
659408b5e66SDavid Majnemer       OS << FixedSizePortion;
660408b5e66SDavid Majnemer       // Make space for a fixup that will eventually have a section relative
661408b5e66SDavid Majnemer       // relocation pointing at the offset where the variable becomes live.
662408b5e66SDavid Majnemer       Fixups.push_back(MCFixup::create(Contents.size(), BE, FK_SecRel_4));
66311cf053bSReid Kleckner       LEWriter.write<uint32_t>(0); // Fixup for code start.
664408b5e66SDavid Majnemer       // Make space for a fixup that will record the section index for the code.
665408b5e66SDavid Majnemer       Fixups.push_back(MCFixup::create(Contents.size(), BE, FK_SecRel_2));
66611cf053bSReid Kleckner       LEWriter.write<uint16_t>(0); // Fixup for section index.
667408b5e66SDavid Majnemer       // Write down the range's extent.
668be82d3ecSReid Kleckner       LEWriter.write<uint16_t>(Chunk);
669408b5e66SDavid Majnemer 
670408b5e66SDavid Majnemer       // Move on to the next range.
671408b5e66SDavid Majnemer       Bias += Chunk;
672408b5e66SDavid Majnemer       RangeSize -= Chunk;
673408b5e66SDavid Majnemer     } while (RangeSize > 0);
674be82d3ecSReid Kleckner 
675be82d3ecSReid Kleckner     // Emit the gaps afterwards.
67611cf053bSReid Kleckner     assert((NumGaps == 0 || Bias <= MaxDefRange) &&
677be82d3ecSReid Kleckner            "large ranges should not have gaps");
678be82d3ecSReid Kleckner     unsigned GapStartOffset = GapAndRangeSizes[I].second;
679be82d3ecSReid Kleckner     for (++I; I != J; ++I) {
680be82d3ecSReid Kleckner       unsigned GapSize, RangeSize;
681be82d3ecSReid Kleckner       assert(I < GapAndRangeSizes.size());
682be82d3ecSReid Kleckner       std::tie(GapSize, RangeSize) = GapAndRangeSizes[I];
683be82d3ecSReid Kleckner       LEWriter.write<uint16_t>(GapStartOffset);
68411cf053bSReid Kleckner       LEWriter.write<uint16_t>(GapSize);
685be82d3ecSReid Kleckner       GapStartOffset += GapSize + RangeSize;
686be82d3ecSReid Kleckner     }
687408b5e66SDavid Majnemer   }
688408b5e66SDavid Majnemer }
689408b5e66SDavid Majnemer 
6902214ed89SReid Kleckner //
6912214ed89SReid Kleckner // This is called when an instruction is assembled into the specified section
6922214ed89SReid Kleckner // and if there is information from the last .cv_loc directive that has yet to have
6932214ed89SReid Kleckner // a line entry made for it is made.
6942214ed89SReid Kleckner //
6952214ed89SReid Kleckner void MCCVLineEntry::Make(MCObjectStreamer *MCOS) {
696a5b1eef8SReid Kleckner   CodeViewContext &CVC = MCOS->getContext().getCVContext();
697a5b1eef8SReid Kleckner   if (!CVC.getCVLocSeen())
6982214ed89SReid Kleckner     return;
6992214ed89SReid Kleckner 
7002214ed89SReid Kleckner   // Create a symbol at in the current section for use in the line entry.
7012214ed89SReid Kleckner   MCSymbol *LineSym = MCOS->getContext().createTempSymbol();
7022214ed89SReid Kleckner   // Set the value of the symbol to use for the MCCVLineEntry.
7032214ed89SReid Kleckner   MCOS->EmitLabel(LineSym);
7042214ed89SReid Kleckner 
7052214ed89SReid Kleckner   // Get the current .loc info saved in the context.
706a5b1eef8SReid Kleckner   const MCCVLoc &CVLoc = CVC.getCurrentCVLoc();
7072214ed89SReid Kleckner 
7082214ed89SReid Kleckner   // Create a (local) line entry with the symbol and the current .loc info.
7092214ed89SReid Kleckner   MCCVLineEntry LineEntry(LineSym, CVLoc);
7102214ed89SReid Kleckner 
7112214ed89SReid Kleckner   // clear CVLocSeen saying the current .loc info is now used.
712a5b1eef8SReid Kleckner   CVC.clearCVLocSeen();
7132214ed89SReid Kleckner 
7142214ed89SReid Kleckner   // Add the line entry to this section's entries.
715a5b1eef8SReid Kleckner   CVC.addLineEntry(LineEntry);
7162214ed89SReid Kleckner }
717