12214ed89SReid Kleckner //===- MCCodeView.h - Machine Code CodeView support -------------*- C++ -*-===//
22214ed89SReid Kleckner //
32946cd70SChandler Carruth // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
42946cd70SChandler Carruth // See https://llvm.org/LICENSE.txt for license information.
52946cd70SChandler Carruth // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
62214ed89SReid Kleckner //
72214ed89SReid Kleckner //===----------------------------------------------------------------------===//
82214ed89SReid Kleckner //
92214ed89SReid Kleckner // Holds state from .cv_file and .cv_loc directives for later emission.
102214ed89SReid Kleckner //
112214ed89SReid Kleckner //===----------------------------------------------------------------------===//
122214ed89SReid Kleckner 
132214ed89SReid Kleckner #include "llvm/MC/MCCodeView.h"
142214ed89SReid Kleckner #include "llvm/ADT/STLExtras.h"
1526fa1bf4SReid Kleckner #include "llvm/ADT/StringExtras.h"
162214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/CodeView.h"
172214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/Line.h"
186fcbd7e9SDavid Majnemer #include "llvm/DebugInfo/CodeView/SymbolRecord.h"
19d427383cSZachary Turner #include "llvm/MC/MCAsmLayout.h"
20*ef736a1cSserge-sans-paille #include "llvm/MC/MCAssembler.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 
293a3cb929SKazu Hirata CodeViewContext::CodeViewContext() = default;
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 
131689f7733SReid Kleckner void CodeViewContext::recordCVLoc(MCContext &Ctx, const MCSymbol *Label,
132689f7733SReid Kleckner                                   unsigned FunctionId, unsigned FileNo,
133689f7733SReid Kleckner                                   unsigned Line, unsigned Column,
134689f7733SReid Kleckner                                   bool PrologueEnd, bool IsStmt) {
135689f7733SReid Kleckner   addLineEntry(MCCVLoc{
136689f7733SReid Kleckner       Label, FunctionId, FileNo, Line, Column, PrologueEnd, IsStmt});
137689f7733SReid Kleckner }
138689f7733SReid Kleckner 
1392214ed89SReid Kleckner MCDataFragment *CodeViewContext::getStringTableFragment() {
1402214ed89SReid Kleckner   if (!StrTabFragment) {
1412214ed89SReid Kleckner     StrTabFragment = new MCDataFragment();
1422214ed89SReid Kleckner     // Start a new string table out with a null byte.
1432214ed89SReid Kleckner     StrTabFragment->getContents().push_back('\0');
1442214ed89SReid Kleckner   }
1452214ed89SReid Kleckner   return StrTabFragment;
1462214ed89SReid Kleckner }
1472214ed89SReid Kleckner 
14826fa1bf4SReid Kleckner std::pair<StringRef, unsigned> CodeViewContext::addToStringTable(StringRef S) {
1492214ed89SReid Kleckner   SmallVectorImpl<char> &Contents = getStringTableFragment()->getContents();
1502214ed89SReid Kleckner   auto Insertion =
1512214ed89SReid Kleckner       StringTable.insert(std::make_pair(S, unsigned(Contents.size())));
1522214ed89SReid Kleckner   // Return the string from the table, since it is stable.
15326fa1bf4SReid Kleckner   std::pair<StringRef, unsigned> Ret =
15426fa1bf4SReid Kleckner       std::make_pair(Insertion.first->first(), Insertion.first->second);
1552214ed89SReid Kleckner   if (Insertion.second) {
1562214ed89SReid Kleckner     // The string map key is always null terminated.
15726fa1bf4SReid Kleckner     Contents.append(Ret.first.begin(), Ret.first.end() + 1);
1582214ed89SReid Kleckner   }
15926fa1bf4SReid Kleckner   return Ret;
1602214ed89SReid Kleckner }
1612214ed89SReid Kleckner 
1622214ed89SReid Kleckner unsigned CodeViewContext::getStringTableOffset(StringRef S) {
1632214ed89SReid Kleckner   // A string table offset of zero is always the empty string.
1642214ed89SReid Kleckner   if (S.empty())
1652214ed89SReid Kleckner     return 0;
1662214ed89SReid Kleckner   auto I = StringTable.find(S);
1672214ed89SReid Kleckner   assert(I != StringTable.end());
1682214ed89SReid Kleckner   return I->second;
1692214ed89SReid Kleckner }
1702214ed89SReid Kleckner 
1712214ed89SReid Kleckner void CodeViewContext::emitStringTable(MCObjectStreamer &OS) {
1722214ed89SReid Kleckner   MCContext &Ctx = OS.getContext();
173a0b44ef6SDavid Blaikie   MCSymbol *StringBegin = Ctx.createTempSymbol("strtab_begin", false),
174a0b44ef6SDavid Blaikie            *StringEnd = Ctx.createTempSymbol("strtab_end", false);
1752214ed89SReid Kleckner 
176692e0c96SFangrui Song   OS.emitInt32(uint32_t(DebugSubsectionKind::StringTable));
1772214ed89SReid Kleckner   OS.emitAbsoluteSymbolDiff(StringEnd, StringBegin, 4);
1786d2d589bSFangrui Song   OS.emitLabel(StringBegin);
1792214ed89SReid Kleckner 
1802214ed89SReid Kleckner   // Put the string table data fragment here, if we haven't already put it
1812214ed89SReid Kleckner   // somewhere else. If somebody wants two string tables in their .s file, one
1822214ed89SReid Kleckner   // will just be empty.
1832214ed89SReid Kleckner   if (!InsertedStrTabFragment) {
1842214ed89SReid Kleckner     OS.insert(getStringTableFragment());
1852214ed89SReid Kleckner     InsertedStrTabFragment = true;
1862214ed89SReid Kleckner   }
1872214ed89SReid Kleckner 
1886d2d589bSFangrui Song   OS.emitValueToAlignment(4, 0);
1892214ed89SReid Kleckner 
1906d2d589bSFangrui Song   OS.emitLabel(StringEnd);
1912214ed89SReid Kleckner }
1922214ed89SReid Kleckner 
1932214ed89SReid Kleckner void CodeViewContext::emitFileChecksums(MCObjectStreamer &OS) {
194ee641c20SReid Kleckner   // Do nothing if there are no file checksums. Microsoft's linker rejects empty
195ee641c20SReid Kleckner   // CodeView substreams.
19626fa1bf4SReid Kleckner   if (Files.empty())
197ee641c20SReid Kleckner     return;
198ee641c20SReid Kleckner 
1992214ed89SReid Kleckner   MCContext &Ctx = OS.getContext();
200a0b44ef6SDavid Blaikie   MCSymbol *FileBegin = Ctx.createTempSymbol("filechecksums_begin", false),
201a0b44ef6SDavid Blaikie            *FileEnd = Ctx.createTempSymbol("filechecksums_end", false);
2022214ed89SReid Kleckner 
203692e0c96SFangrui Song   OS.emitInt32(uint32_t(DebugSubsectionKind::FileChecksums));
2042214ed89SReid Kleckner   OS.emitAbsoluteSymbolDiff(FileEnd, FileBegin, 4);
2056d2d589bSFangrui Song   OS.emitLabel(FileBegin);
2062214ed89SReid Kleckner 
20726fa1bf4SReid Kleckner   unsigned CurrentOffset = 0;
20826fa1bf4SReid Kleckner 
2092214ed89SReid Kleckner   // Emit an array of FileChecksum entries. We index into this table using the
21026fa1bf4SReid Kleckner   // user-provided file number.  Each entry may be a variable number of bytes
21126fa1bf4SReid Kleckner   // determined by the checksum kind and size.
21226fa1bf4SReid Kleckner   for (auto File : Files) {
213a55daa14SFangrui Song     OS.emitAssignment(File.ChecksumTableOffset,
21426fa1bf4SReid Kleckner                       MCConstantExpr::create(CurrentOffset, Ctx));
21526fa1bf4SReid Kleckner     CurrentOffset += 4; // String table offset.
21626fa1bf4SReid Kleckner     if (!File.ChecksumKind) {
21726fa1bf4SReid Kleckner       CurrentOffset +=
21826fa1bf4SReid Kleckner           4; // One byte each for checksum size and kind, then align to 4 bytes.
21926fa1bf4SReid Kleckner     } else {
22026fa1bf4SReid Kleckner       CurrentOffset += 2; // One byte each for checksum size and kind.
22126fa1bf4SReid Kleckner       CurrentOffset += File.Checksum.size();
22226fa1bf4SReid Kleckner       CurrentOffset = alignTo(CurrentOffset, 4);
22326fa1bf4SReid Kleckner     }
22426fa1bf4SReid Kleckner 
225692e0c96SFangrui Song     OS.emitInt32(File.StringTableOffset);
22626fa1bf4SReid Kleckner 
22726fa1bf4SReid Kleckner     if (!File.ChecksumKind) {
22826fa1bf4SReid Kleckner       // There is no checksum.  Therefore zero the next two fields and align
22926fa1bf4SReid Kleckner       // back to 4 bytes.
230692e0c96SFangrui Song       OS.emitInt32(0);
23126fa1bf4SReid Kleckner       continue;
23226fa1bf4SReid Kleckner     }
233692e0c96SFangrui Song     OS.emitInt8(static_cast<uint8_t>(File.Checksum.size()));
234692e0c96SFangrui Song     OS.emitInt8(File.ChecksumKind);
235a55daa14SFangrui Song     OS.emitBytes(toStringRef(File.Checksum));
2366d2d589bSFangrui Song     OS.emitValueToAlignment(4);
2372214ed89SReid Kleckner   }
2382214ed89SReid Kleckner 
2396d2d589bSFangrui Song   OS.emitLabel(FileEnd);
24026fa1bf4SReid Kleckner 
24126fa1bf4SReid Kleckner   ChecksumOffsetsAssigned = true;
24226fa1bf4SReid Kleckner }
24326fa1bf4SReid Kleckner 
24426fa1bf4SReid Kleckner // Output checksum table offset of the given file number.  It is possible that
24526fa1bf4SReid Kleckner // not all files have been registered yet, and so the offset cannot be
24626fa1bf4SReid Kleckner // calculated.  In this case a symbol representing the offset is emitted, and
24726fa1bf4SReid Kleckner // the value of this symbol will be fixed up at a later time.
24826fa1bf4SReid Kleckner void CodeViewContext::emitFileChecksumOffset(MCObjectStreamer &OS,
24926fa1bf4SReid Kleckner                                              unsigned FileNo) {
25026fa1bf4SReid Kleckner   unsigned Idx = FileNo - 1;
25126fa1bf4SReid Kleckner 
25226fa1bf4SReid Kleckner   if (Idx >= Files.size())
25326fa1bf4SReid Kleckner     Files.resize(Idx + 1);
25426fa1bf4SReid Kleckner 
25526fa1bf4SReid Kleckner   if (ChecksumOffsetsAssigned) {
2566d2d589bSFangrui Song     OS.emitSymbolValue(Files[Idx].ChecksumTableOffset, 4);
25726fa1bf4SReid Kleckner     return;
25826fa1bf4SReid Kleckner   }
25926fa1bf4SReid Kleckner 
26026fa1bf4SReid Kleckner   const MCSymbolRefExpr *SRE =
26126fa1bf4SReid Kleckner       MCSymbolRefExpr::create(Files[Idx].ChecksumTableOffset, OS.getContext());
26226fa1bf4SReid Kleckner 
2636d2d589bSFangrui Song   OS.emitValueImpl(SRE, 4);
2642214ed89SReid Kleckner }
2652214ed89SReid Kleckner 
266689f7733SReid Kleckner void CodeViewContext::addLineEntry(const MCCVLoc &LineEntry) {
267b5258722SReid Kleckner   size_t Offset = MCCVLines.size();
268b5258722SReid Kleckner   auto I = MCCVLineStartStop.insert(
269b5258722SReid Kleckner       {LineEntry.getFunctionId(), {Offset, Offset + 1}});
270b5258722SReid Kleckner   if (!I.second)
271b5258722SReid Kleckner     I.first->second.second = Offset + 1;
272b5258722SReid Kleckner   MCCVLines.push_back(LineEntry);
273b5258722SReid Kleckner }
274b5258722SReid Kleckner 
275689f7733SReid Kleckner std::vector<MCCVLoc>
276b5258722SReid Kleckner CodeViewContext::getFunctionLineEntries(unsigned FuncId) {
277689f7733SReid Kleckner   std::vector<MCCVLoc> FilteredLines;
278b5258722SReid Kleckner   auto I = MCCVLineStartStop.find(FuncId);
2797897a789SReid Kleckner   if (I != MCCVLineStartStop.end()) {
2807897a789SReid Kleckner     MCCVFunctionInfo *SiteInfo = getCVFunctionInfo(FuncId);
281b5258722SReid Kleckner     for (size_t Idx = I->second.first, End = I->second.second; Idx != End;
2827897a789SReid Kleckner          ++Idx) {
2837897a789SReid Kleckner       unsigned LocationFuncId = MCCVLines[Idx].getFunctionId();
2847897a789SReid Kleckner       if (LocationFuncId == FuncId) {
2857897a789SReid Kleckner         // This was a .cv_loc directly for FuncId, so record it.
286b5258722SReid Kleckner         FilteredLines.push_back(MCCVLines[Idx]);
2877897a789SReid Kleckner       } else {
2887897a789SReid Kleckner         // Check if the current location is inlined in this function. If it is,
2897897a789SReid Kleckner         // synthesize a statement .cv_loc at the original inlined call site.
2907897a789SReid Kleckner         auto I = SiteInfo->InlinedAtMap.find(LocationFuncId);
2917897a789SReid Kleckner         if (I != SiteInfo->InlinedAtMap.end()) {
2927897a789SReid Kleckner           MCCVFunctionInfo::LineInfo &IA = I->second;
2937897a789SReid Kleckner           // Only add the location if it differs from the previous location.
2947897a789SReid Kleckner           // Large inlined calls will have many .cv_loc entries and we only need
2957897a789SReid Kleckner           // one line table entry in the parent function.
2967897a789SReid Kleckner           if (FilteredLines.empty() ||
2977897a789SReid Kleckner               FilteredLines.back().getFileNum() != IA.File ||
2987897a789SReid Kleckner               FilteredLines.back().getLine() != IA.Line ||
2997897a789SReid Kleckner               FilteredLines.back().getColumn() != IA.Col) {
300689f7733SReid Kleckner             FilteredLines.push_back(MCCVLoc(
3017897a789SReid Kleckner                 MCCVLines[Idx].getLabel(),
302689f7733SReid Kleckner                 FuncId, IA.File, IA.Line, IA.Col, false, false));
3037897a789SReid Kleckner           }
3047897a789SReid Kleckner         }
3057897a789SReid Kleckner       }
3067897a789SReid Kleckner     }
3077897a789SReid Kleckner   }
308b5258722SReid Kleckner   return FilteredLines;
309b5258722SReid Kleckner }
310b5258722SReid Kleckner 
311b5258722SReid Kleckner std::pair<size_t, size_t> CodeViewContext::getLineExtent(unsigned FuncId) {
312b5258722SReid Kleckner   auto I = MCCVLineStartStop.find(FuncId);
313b5258722SReid Kleckner   // Return an empty extent if there are no cv_locs for this function id.
314b5258722SReid Kleckner   if (I == MCCVLineStartStop.end())
315b5258722SReid Kleckner     return {~0ULL, 0};
316b5258722SReid Kleckner   return I->second;
317b5258722SReid Kleckner }
318b5258722SReid Kleckner 
319689f7733SReid Kleckner ArrayRef<MCCVLoc> CodeViewContext::getLinesForExtent(size_t L, size_t R) {
320b5258722SReid Kleckner   if (R <= L)
321b5258722SReid Kleckner     return None;
322b5258722SReid Kleckner   if (L >= MCCVLines.size())
323b5258722SReid Kleckner     return None;
324b5258722SReid Kleckner   return makeArrayRef(&MCCVLines[L], R - L);
325b5258722SReid Kleckner }
326b5258722SReid Kleckner 
3272214ed89SReid Kleckner void CodeViewContext::emitLineTableForFunction(MCObjectStreamer &OS,
3282214ed89SReid Kleckner                                                unsigned FuncId,
3292214ed89SReid Kleckner                                                const MCSymbol *FuncBegin,
3302214ed89SReid Kleckner                                                const MCSymbol *FuncEnd) {
3312214ed89SReid Kleckner   MCContext &Ctx = OS.getContext();
332a0b44ef6SDavid Blaikie   MCSymbol *LineBegin = Ctx.createTempSymbol("linetable_begin", false),
333a0b44ef6SDavid Blaikie            *LineEnd = Ctx.createTempSymbol("linetable_end", false);
3342214ed89SReid Kleckner 
335692e0c96SFangrui Song   OS.emitInt32(uint32_t(DebugSubsectionKind::Lines));
3362214ed89SReid Kleckner   OS.emitAbsoluteSymbolDiff(LineEnd, LineBegin, 4);
3376d2d589bSFangrui Song   OS.emitLabel(LineBegin);
338f7d84ee6SKeno Fischer   OS.EmitCOFFSecRel32(FuncBegin, /*Offset=*/0);
3392214ed89SReid Kleckner   OS.EmitCOFFSectionIndex(FuncBegin);
3402214ed89SReid Kleckner 
3412214ed89SReid Kleckner   // Actual line info.
342689f7733SReid Kleckner   std::vector<MCCVLoc> Locs = getFunctionLineEntries(FuncId);
343689f7733SReid Kleckner   bool HaveColumns = any_of(Locs, [](const MCCVLoc &LineEntry) {
3442214ed89SReid Kleckner     return LineEntry.getColumn() != 0;
3452214ed89SReid Kleckner   });
346692e0c96SFangrui Song   OS.emitInt16(HaveColumns ? int(LF_HaveColumns) : 0);
3472214ed89SReid Kleckner   OS.emitAbsoluteSymbolDiff(FuncEnd, FuncBegin, 4);
3482214ed89SReid Kleckner 
3492214ed89SReid Kleckner   for (auto I = Locs.begin(), E = Locs.end(); I != E;) {
3502214ed89SReid Kleckner     // Emit a file segment for the run of locations that share a file id.
3512214ed89SReid Kleckner     unsigned CurFileNum = I->getFileNum();
3522214ed89SReid Kleckner     auto FileSegEnd =
353689f7733SReid Kleckner         std::find_if(I, E, [CurFileNum](const MCCVLoc &Loc) {
3542214ed89SReid Kleckner           return Loc.getFileNum() != CurFileNum;
3552214ed89SReid Kleckner         });
3562214ed89SReid Kleckner     unsigned EntryCount = FileSegEnd - I;
35726fa1bf4SReid Kleckner     OS.AddComment(
35826fa1bf4SReid Kleckner         "Segment for file '" +
35926fa1bf4SReid Kleckner         Twine(getStringTableFragment()
36026fa1bf4SReid Kleckner                   ->getContents()[Files[CurFileNum - 1].StringTableOffset]) +
3612214ed89SReid Kleckner         "' begins");
362c031378cSShengchen Kan     OS.emitCVFileChecksumOffsetDirective(CurFileNum);
363692e0c96SFangrui Song     OS.emitInt32(EntryCount);
3642214ed89SReid Kleckner     uint32_t SegmentSize = 12;
3652214ed89SReid Kleckner     SegmentSize += 8 * EntryCount;
3662214ed89SReid Kleckner     if (HaveColumns)
3672214ed89SReid Kleckner       SegmentSize += 4 * EntryCount;
368692e0c96SFangrui Song     OS.emitInt32(SegmentSize);
3692214ed89SReid Kleckner 
3702214ed89SReid Kleckner     for (auto J = I; J != FileSegEnd; ++J) {
3712214ed89SReid Kleckner       OS.emitAbsoluteSymbolDiff(J->getLabel(), FuncBegin, 4);
3722214ed89SReid Kleckner       unsigned LineData = J->getLine();
3732214ed89SReid Kleckner       if (J->isStmt())
3746fcbd7e9SDavid Majnemer         LineData |= LineInfo::StatementFlag;
375692e0c96SFangrui Song       OS.emitInt32(LineData);
3762214ed89SReid Kleckner     }
3772214ed89SReid Kleckner     if (HaveColumns) {
3782214ed89SReid Kleckner       for (auto J = I; J != FileSegEnd; ++J) {
379692e0c96SFangrui Song         OS.emitInt16(J->getColumn());
380692e0c96SFangrui Song         OS.emitInt16(0);
3812214ed89SReid Kleckner       }
3822214ed89SReid Kleckner     }
3832214ed89SReid Kleckner     I = FileSegEnd;
3842214ed89SReid Kleckner   }
3856d2d589bSFangrui Song   OS.emitLabel(LineEnd);
3862214ed89SReid Kleckner }
3872214ed89SReid Kleckner 
3886fcbd7e9SDavid Majnemer static bool compressAnnotation(uint32_t Data, SmallVectorImpl<char> &Buffer) {
3896fcbd7e9SDavid Majnemer   if (isUInt<7>(Data)) {
3906fcbd7e9SDavid Majnemer     Buffer.push_back(Data);
3916fcbd7e9SDavid Majnemer     return true;
3926fcbd7e9SDavid Majnemer   }
3936fcbd7e9SDavid Majnemer 
3946fcbd7e9SDavid Majnemer   if (isUInt<14>(Data)) {
3956fcbd7e9SDavid Majnemer     Buffer.push_back((Data >> 8) | 0x80);
3966fcbd7e9SDavid Majnemer     Buffer.push_back(Data & 0xff);
3976fcbd7e9SDavid Majnemer     return true;
3986fcbd7e9SDavid Majnemer   }
3996fcbd7e9SDavid Majnemer 
4006fcbd7e9SDavid Majnemer   if (isUInt<29>(Data)) {
4016fcbd7e9SDavid Majnemer     Buffer.push_back((Data >> 24) | 0xC0);
4026fcbd7e9SDavid Majnemer     Buffer.push_back((Data >> 16) & 0xff);
4036fcbd7e9SDavid Majnemer     Buffer.push_back((Data >> 8) & 0xff);
4046fcbd7e9SDavid Majnemer     Buffer.push_back(Data & 0xff);
4056fcbd7e9SDavid Majnemer     return true;
4066fcbd7e9SDavid Majnemer   }
4076fcbd7e9SDavid Majnemer 
4086fcbd7e9SDavid Majnemer   return false;
4096fcbd7e9SDavid Majnemer }
4106fcbd7e9SDavid Majnemer 
41163a2846eSZachary Turner static bool compressAnnotation(BinaryAnnotationsOpCode Annotation,
41263a2846eSZachary Turner                                SmallVectorImpl<char> &Buffer) {
41363a2846eSZachary Turner   return compressAnnotation(static_cast<uint32_t>(Annotation), Buffer);
41463a2846eSZachary Turner }
41563a2846eSZachary Turner 
4166fcbd7e9SDavid Majnemer static uint32_t encodeSignedNumber(uint32_t Data) {
4176fcbd7e9SDavid Majnemer   if (Data >> 31)
4186fcbd7e9SDavid Majnemer     return ((-Data) << 1) | 1;
4196fcbd7e9SDavid Majnemer   return Data << 1;
4206fcbd7e9SDavid Majnemer }
4216fcbd7e9SDavid Majnemer 
422a9f4cc95SReid Kleckner void CodeViewContext::emitInlineLineTableForFunction(MCObjectStreamer &OS,
423a9f4cc95SReid Kleckner                                                      unsigned PrimaryFunctionId,
424a9f4cc95SReid Kleckner                                                      unsigned SourceFileId,
425a9f4cc95SReid Kleckner                                                      unsigned SourceLineNum,
426a9f4cc95SReid Kleckner                                                      const MCSymbol *FnStartSym,
427a9f4cc95SReid Kleckner                                                      const MCSymbol *FnEndSym) {
4281fcd610cSReid Kleckner   // Create and insert a fragment into the current section that will be encoded
4291fcd610cSReid Kleckner   // later.
430a9f4cc95SReid Kleckner   new MCCVInlineLineTableFragment(PrimaryFunctionId, SourceFileId,
431a9f4cc95SReid Kleckner                                   SourceLineNum, FnStartSym, FnEndSym,
432a9f4cc95SReid Kleckner                                   OS.getCurrentSectionOnly());
4336fcbd7e9SDavid Majnemer }
4346fcbd7e9SDavid Majnemer 
43594ee0728SReid Kleckner MCFragment *CodeViewContext::emitDefRange(
436408b5e66SDavid Majnemer     MCObjectStreamer &OS,
437408b5e66SDavid Majnemer     ArrayRef<std::pair<const MCSymbol *, const MCSymbol *>> Ranges,
438408b5e66SDavid Majnemer     StringRef FixedSizePortion) {
439408b5e66SDavid Majnemer   // Create and insert a fragment into the current section that will be encoded
440408b5e66SDavid Majnemer   // later.
44194ee0728SReid Kleckner   return new MCCVDefRangeFragment(Ranges, FixedSizePortion,
442408b5e66SDavid Majnemer                            OS.getCurrentSectionOnly());
443408b5e66SDavid Majnemer }
444408b5e66SDavid Majnemer 
445cb91e7d3SReid Kleckner static unsigned computeLabelDiff(MCAsmLayout &Layout, const MCSymbol *Begin,
4461fcd610cSReid Kleckner                                  const MCSymbol *End) {
4471fcd610cSReid Kleckner   MCContext &Ctx = Layout.getAssembler().getContext();
4481fcd610cSReid Kleckner   MCSymbolRefExpr::VariantKind Variant = MCSymbolRefExpr::VK_None;
4491fcd610cSReid Kleckner   const MCExpr *BeginRef = MCSymbolRefExpr::create(Begin, Variant, Ctx),
4501fcd610cSReid Kleckner                *EndRef = MCSymbolRefExpr::create(End, Variant, Ctx);
4511fcd610cSReid Kleckner   const MCExpr *AddrDelta =
4521fcd610cSReid Kleckner       MCBinaryExpr::create(MCBinaryExpr::Sub, EndRef, BeginRef, Ctx);
4531fcd610cSReid Kleckner   int64_t Result;
4541fcd610cSReid Kleckner   bool Success = AddrDelta->evaluateKnownAbsolute(Result, Layout);
4551fcd610cSReid Kleckner   assert(Success && "failed to evaluate label difference as absolute");
4561fcd610cSReid Kleckner   (void)Success;
4571fcd610cSReid Kleckner   assert(Result >= 0 && "negative label difference requested");
4581fcd610cSReid Kleckner   assert(Result < UINT_MAX && "label difference greater than 2GB");
4591fcd610cSReid Kleckner   return unsigned(Result);
4601fcd610cSReid Kleckner }
4616fcbd7e9SDavid Majnemer 
4621fcd610cSReid Kleckner void CodeViewContext::encodeInlineLineTable(MCAsmLayout &Layout,
4631fcd610cSReid Kleckner                                             MCCVInlineLineTableFragment &Frag) {
4641fcd610cSReid Kleckner   size_t LocBegin;
4651fcd610cSReid Kleckner   size_t LocEnd;
4661fcd610cSReid Kleckner   std::tie(LocBegin, LocEnd) = getLineExtent(Frag.SiteFuncId);
467a9f4cc95SReid Kleckner 
468a9f4cc95SReid Kleckner   // Include all child inline call sites in our .cv_loc extent.
469a9f4cc95SReid Kleckner   MCCVFunctionInfo *SiteInfo = getCVFunctionInfo(Frag.SiteFuncId);
470a9f4cc95SReid Kleckner   for (auto &KV : SiteInfo->InlinedAtMap) {
471a9f4cc95SReid Kleckner     unsigned ChildId = KV.first;
472a9f4cc95SReid Kleckner     auto Extent = getLineExtent(ChildId);
4731fcd610cSReid Kleckner     LocBegin = std::min(LocBegin, Extent.first);
4741fcd610cSReid Kleckner     LocEnd = std::max(LocEnd, Extent.second);
4751fcd610cSReid Kleckner   }
476a9f4cc95SReid Kleckner 
4771fcd610cSReid Kleckner   if (LocBegin >= LocEnd)
4781fcd610cSReid Kleckner     return;
479689f7733SReid Kleckner   ArrayRef<MCCVLoc> Locs = getLinesForExtent(LocBegin, LocEnd);
4801fcd610cSReid Kleckner   if (Locs.empty())
4811fcd610cSReid Kleckner     return;
4821fcd610cSReid Kleckner 
4832c6430feSReid Kleckner   // Check that the locations are all in the same section.
4842c6430feSReid Kleckner #ifndef NDEBUG
4852c6430feSReid Kleckner   const MCSection *FirstSec = &Locs.front().getLabel()->getSection();
486689f7733SReid Kleckner   for (const MCCVLoc &Loc : Locs) {
4872c6430feSReid Kleckner     if (&Loc.getLabel()->getSection() != FirstSec) {
4882c6430feSReid Kleckner       errs() << ".cv_loc " << Loc.getFunctionId() << ' ' << Loc.getFileNum()
4892c6430feSReid Kleckner              << ' ' << Loc.getLine() << ' ' << Loc.getColumn()
4902c6430feSReid Kleckner              << " is in the wrong section\n";
4912c6430feSReid Kleckner       llvm_unreachable(".cv_loc crosses sections");
4922c6430feSReid Kleckner     }
4932c6430feSReid Kleckner   }
4942c6430feSReid Kleckner #endif
4952c6430feSReid Kleckner 
4961fcd610cSReid Kleckner   // Make an artificial start location using the function start and the inlinee
4971fcd610cSReid Kleckner   // lines start location information. All deltas start relative to this
4981fcd610cSReid Kleckner   // location.
499689f7733SReid Kleckner   MCCVLoc StartLoc = Locs.front();
500689f7733SReid Kleckner   StartLoc.setLabel(Frag.getFnStartSym());
5011fcd610cSReid Kleckner   StartLoc.setFileNum(Frag.StartFileId);
5021fcd610cSReid Kleckner   StartLoc.setLine(Frag.StartLineNum);
503c29b4f07SReid Kleckner   bool HaveOpenRange = false;
5041fcd610cSReid Kleckner 
505a9f4cc95SReid Kleckner   const MCSymbol *LastLabel = Frag.getFnStartSym();
506a9f4cc95SReid Kleckner   MCCVFunctionInfo::LineInfo LastSourceLoc, CurSourceLoc;
507a9f4cc95SReid Kleckner   LastSourceLoc.File = Frag.StartFileId;
508a9f4cc95SReid Kleckner   LastSourceLoc.Line = Frag.StartLineNum;
509a9f4cc95SReid Kleckner 
5101fcd610cSReid Kleckner   SmallVectorImpl<char> &Buffer = Frag.getContents();
5111fcd610cSReid Kleckner   Buffer.clear(); // Clear old contents if we went through relaxation.
512689f7733SReid Kleckner   for (const MCCVLoc &Loc : Locs) {
513bb96df60SReid Kleckner     // Exit early if our line table would produce an oversized InlineSiteSym
514bb96df60SReid Kleckner     // record. Account for the ChangeCodeLength annotation emitted after the
515bb96df60SReid Kleckner     // loop ends.
51646225b19SZachary Turner     constexpr uint32_t InlineSiteSize = 12;
51746225b19SZachary Turner     constexpr uint32_t AnnotationSize = 8;
51846225b19SZachary Turner     size_t MaxBufferSize = MaxRecordLength - InlineSiteSize - AnnotationSize;
519bb96df60SReid Kleckner     if (Buffer.size() >= MaxBufferSize)
520bb96df60SReid Kleckner       break;
521bb96df60SReid Kleckner 
522a9f4cc95SReid Kleckner     if (Loc.getFunctionId() == Frag.SiteFuncId) {
523a9f4cc95SReid Kleckner       CurSourceLoc.File = Loc.getFileNum();
524a9f4cc95SReid Kleckner       CurSourceLoc.Line = Loc.getLine();
525a9f4cc95SReid Kleckner     } else {
526a9f4cc95SReid Kleckner       auto I = SiteInfo->InlinedAtMap.find(Loc.getFunctionId());
527a9f4cc95SReid Kleckner       if (I != SiteInfo->InlinedAtMap.end()) {
528a9f4cc95SReid Kleckner         // This .cv_loc is from a child inline call site. Use the source
529a9f4cc95SReid Kleckner         // location of the inlined call site instead of the .cv_loc directive
530a9f4cc95SReid Kleckner         // source location.
531a9f4cc95SReid Kleckner         CurSourceLoc = I->second;
532a9f4cc95SReid Kleckner       } else {
5331fcd610cSReid Kleckner         // We've hit a cv_loc not attributed to this inline call site. Use this
5341fcd610cSReid Kleckner         // label to end the PC range.
535c29b4f07SReid Kleckner         if (HaveOpenRange) {
536a9f4cc95SReid Kleckner           unsigned Length = computeLabelDiff(Layout, LastLabel, Loc.getLabel());
53763a2846eSZachary Turner           compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeLength, Buffer);
5381fcd610cSReid Kleckner           compressAnnotation(Length, Buffer);
539eddd6c25SReid Kleckner           LastLabel = Loc.getLabel();
5401fcd610cSReid Kleckner         }
541c29b4f07SReid Kleckner         HaveOpenRange = false;
5421fcd610cSReid Kleckner         continue;
5431fcd610cSReid Kleckner       }
544a9f4cc95SReid Kleckner     }
545c29b4f07SReid Kleckner 
546a9f4cc95SReid Kleckner     // Skip this .cv_loc if we have an open range and this isn't a meaningful
547a9f4cc95SReid Kleckner     // source location update. The current table format does not support column
548a9f4cc95SReid Kleckner     // info, so we can skip updates for those.
549a9f4cc95SReid Kleckner     if (HaveOpenRange && CurSourceLoc.File == LastSourceLoc.File &&
550a9f4cc95SReid Kleckner         CurSourceLoc.Line == LastSourceLoc.Line)
551c29b4f07SReid Kleckner       continue;
552a9f4cc95SReid Kleckner 
553c29b4f07SReid Kleckner     HaveOpenRange = true;
5541fcd610cSReid Kleckner 
555a9f4cc95SReid Kleckner     if (CurSourceLoc.File != LastSourceLoc.File) {
55626fa1bf4SReid Kleckner       unsigned FileOffset = static_cast<const MCConstantExpr *>(
55726fa1bf4SReid Kleckner                                 Files[CurSourceLoc.File - 1]
55826fa1bf4SReid Kleckner                                     .ChecksumTableOffset->getVariableValue())
55926fa1bf4SReid Kleckner                                 ->getValue();
56063a2846eSZachary Turner       compressAnnotation(BinaryAnnotationsOpCode::ChangeFile, Buffer);
561344078f5SReid Kleckner       compressAnnotation(FileOffset, Buffer);
5621fcd610cSReid Kleckner     }
5631fcd610cSReid Kleckner 
564a9f4cc95SReid Kleckner     int LineDelta = CurSourceLoc.Line - LastSourceLoc.Line;
5651fcd610cSReid Kleckner     unsigned EncodedLineDelta = encodeSignedNumber(LineDelta);
566a9f4cc95SReid Kleckner     unsigned CodeDelta = computeLabelDiff(Layout, LastLabel, Loc.getLabel());
567399bc48eSAmy Huang     if (EncodedLineDelta < 0x8 && CodeDelta <= 0xf) {
5681fcd610cSReid Kleckner       // The ChangeCodeOffsetAndLineOffset combination opcode is used when the
5691fcd610cSReid Kleckner       // encoded line delta uses 3 or fewer set bits and the code offset fits
5701fcd610cSReid Kleckner       // in one nibble.
5711fcd610cSReid Kleckner       unsigned Operand = (EncodedLineDelta << 4) | CodeDelta;
57263a2846eSZachary Turner       compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeOffsetAndLineOffset,
57363a2846eSZachary Turner                          Buffer);
5741fcd610cSReid Kleckner       compressAnnotation(Operand, Buffer);
5751fcd610cSReid Kleckner     } else {
5761fcd610cSReid Kleckner       // Otherwise use the separate line and code deltas.
577a9f4cc95SReid Kleckner       if (LineDelta != 0) {
57863a2846eSZachary Turner         compressAnnotation(BinaryAnnotationsOpCode::ChangeLineOffset, Buffer);
5791fcd610cSReid Kleckner         compressAnnotation(EncodedLineDelta, Buffer);
580a9f4cc95SReid Kleckner       }
58163a2846eSZachary Turner       compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeOffset, Buffer);
5821fcd610cSReid Kleckner       compressAnnotation(CodeDelta, Buffer);
5831fcd610cSReid Kleckner     }
5846fcbd7e9SDavid Majnemer 
585a9f4cc95SReid Kleckner     LastLabel = Loc.getLabel();
586a9f4cc95SReid Kleckner     LastSourceLoc = CurSourceLoc;
5876fcbd7e9SDavid Majnemer   }
588c9911f28SDavid Majnemer 
589c29b4f07SReid Kleckner   assert(HaveOpenRange);
590c9911f28SDavid Majnemer 
591c9911f28SDavid Majnemer   unsigned EndSymLength =
592a9f4cc95SReid Kleckner       computeLabelDiff(Layout, LastLabel, Frag.getFnEndSym());
593c9911f28SDavid Majnemer   unsigned LocAfterLength = ~0U;
594689f7733SReid Kleckner   ArrayRef<MCCVLoc> LocAfter = getLinesForExtent(LocEnd, LocEnd + 1);
595cb91e7d3SReid Kleckner   if (!LocAfter.empty()) {
596cb91e7d3SReid Kleckner     // Only try to compute this difference if we're in the same section.
597689f7733SReid Kleckner     const MCCVLoc &Loc = LocAfter[0];
5985e102eeeSSam Clegg     if (&Loc.getLabel()->getSection() == &LastLabel->getSection())
599a9f4cc95SReid Kleckner       LocAfterLength = computeLabelDiff(Layout, LastLabel, Loc.getLabel());
600cb91e7d3SReid Kleckner   }
601c9911f28SDavid Majnemer 
60263a2846eSZachary Turner   compressAnnotation(BinaryAnnotationsOpCode::ChangeCodeLength, Buffer);
603c9911f28SDavid Majnemer   compressAnnotation(std::min(EndSymLength, LocAfterLength), Buffer);
6046fcbd7e9SDavid Majnemer }
6056fcbd7e9SDavid Majnemer 
606408b5e66SDavid Majnemer void CodeViewContext::encodeDefRange(MCAsmLayout &Layout,
607408b5e66SDavid Majnemer                                      MCCVDefRangeFragment &Frag) {
608408b5e66SDavid Majnemer   MCContext &Ctx = Layout.getAssembler().getContext();
609408b5e66SDavid Majnemer   SmallVectorImpl<char> &Contents = Frag.getContents();
610408b5e66SDavid Majnemer   Contents.clear();
611408b5e66SDavid Majnemer   SmallVectorImpl<MCFixup> &Fixups = Frag.getFixups();
612408b5e66SDavid Majnemer   Fixups.clear();
613408b5e66SDavid Majnemer   raw_svector_ostream OS(Contents);
614408b5e66SDavid Majnemer 
615be82d3ecSReid Kleckner   // Compute all the sizes up front.
616be82d3ecSReid Kleckner   SmallVector<std::pair<unsigned, unsigned>, 4> GapAndRangeSizes;
617be82d3ecSReid Kleckner   const MCSymbol *LastLabel = nullptr;
618408b5e66SDavid Majnemer   for (std::pair<const MCSymbol *, const MCSymbol *> Range : Frag.getRanges()) {
619be82d3ecSReid Kleckner     unsigned GapSize =
620be82d3ecSReid Kleckner         LastLabel ? computeLabelDiff(Layout, LastLabel, Range.first) : 0;
621408b5e66SDavid Majnemer     unsigned RangeSize = computeLabelDiff(Layout, Range.first, Range.second);
622be82d3ecSReid Kleckner     GapAndRangeSizes.push_back({GapSize, RangeSize});
623be82d3ecSReid Kleckner     LastLabel = Range.second;
624be82d3ecSReid Kleckner   }
625be82d3ecSReid Kleckner 
626be82d3ecSReid Kleckner   // Write down each range where the variable is defined.
627be82d3ecSReid Kleckner   for (size_t I = 0, E = Frag.getRanges().size(); I != E;) {
628be82d3ecSReid Kleckner     // If the range size of multiple consecutive ranges is under the max,
629be82d3ecSReid Kleckner     // combine the ranges and emit some gaps.
630be82d3ecSReid Kleckner     const MCSymbol *RangeBegin = Frag.getRanges()[I].first;
631be82d3ecSReid Kleckner     unsigned RangeSize = GapAndRangeSizes[I].second;
632be82d3ecSReid Kleckner     size_t J = I + 1;
633be82d3ecSReid Kleckner     for (; J != E; ++J) {
634be82d3ecSReid Kleckner       unsigned GapAndRangeSize = GapAndRangeSizes[J].first + GapAndRangeSizes[J].second;
635be82d3ecSReid Kleckner       if (RangeSize + GapAndRangeSize > MaxDefRange)
636be82d3ecSReid Kleckner         break;
637be82d3ecSReid Kleckner       RangeSize += GapAndRangeSize;
638be82d3ecSReid Kleckner     }
639be82d3ecSReid Kleckner     unsigned NumGaps = J - I - 1;
640be82d3ecSReid Kleckner 
641e3f65297SPeter Collingbourne     support::endian::Writer LEWriter(OS, support::little);
642be82d3ecSReid Kleckner 
643408b5e66SDavid Majnemer     unsigned Bias = 0;
644408b5e66SDavid Majnemer     // We must split the range into chunks of MaxDefRange, this is a fundamental
645408b5e66SDavid Majnemer     // limitation of the file format.
646408b5e66SDavid Majnemer     do {
647408b5e66SDavid Majnemer       uint16_t Chunk = std::min((uint32_t)MaxDefRange, RangeSize);
648408b5e66SDavid Majnemer 
649be82d3ecSReid Kleckner       const MCSymbolRefExpr *SRE = MCSymbolRefExpr::create(RangeBegin, Ctx);
650408b5e66SDavid Majnemer       const MCBinaryExpr *BE =
651408b5e66SDavid Majnemer           MCBinaryExpr::createAdd(SRE, MCConstantExpr::create(Bias, Ctx), Ctx);
652408b5e66SDavid Majnemer       MCValue Res;
653408b5e66SDavid Majnemer       BE->evaluateAsRelocatable(Res, &Layout, /*Fixup=*/nullptr);
654408b5e66SDavid Majnemer 
655408b5e66SDavid Majnemer       // Each record begins with a 2-byte number indicating how large the record
656408b5e66SDavid Majnemer       // is.
657408b5e66SDavid Majnemer       StringRef FixedSizePortion = Frag.getFixedSizePortion();
658408b5e66SDavid Majnemer       // Our record is a fixed sized prefix and a LocalVariableAddrRange that we
659408b5e66SDavid Majnemer       // are artificially constructing.
660be82d3ecSReid Kleckner       size_t RecordSize = FixedSizePortion.size() +
661be82d3ecSReid Kleckner                           sizeof(LocalVariableAddrRange) + 4 * NumGaps;
66211cf053bSReid Kleckner       // Write out the record size.
66311cf053bSReid Kleckner       LEWriter.write<uint16_t>(RecordSize);
664408b5e66SDavid Majnemer       // Write out the fixed size prefix.
665408b5e66SDavid Majnemer       OS << FixedSizePortion;
666408b5e66SDavid Majnemer       // Make space for a fixup that will eventually have a section relative
667408b5e66SDavid Majnemer       // relocation pointing at the offset where the variable becomes live.
668408b5e66SDavid Majnemer       Fixups.push_back(MCFixup::create(Contents.size(), BE, FK_SecRel_4));
66911cf053bSReid Kleckner       LEWriter.write<uint32_t>(0); // Fixup for code start.
670408b5e66SDavid Majnemer       // Make space for a fixup that will record the section index for the code.
671408b5e66SDavid Majnemer       Fixups.push_back(MCFixup::create(Contents.size(), BE, FK_SecRel_2));
67211cf053bSReid Kleckner       LEWriter.write<uint16_t>(0); // Fixup for section index.
673408b5e66SDavid Majnemer       // Write down the range's extent.
674be82d3ecSReid Kleckner       LEWriter.write<uint16_t>(Chunk);
675408b5e66SDavid Majnemer 
676408b5e66SDavid Majnemer       // Move on to the next range.
677408b5e66SDavid Majnemer       Bias += Chunk;
678408b5e66SDavid Majnemer       RangeSize -= Chunk;
679408b5e66SDavid Majnemer     } while (RangeSize > 0);
680be82d3ecSReid Kleckner 
681be82d3ecSReid Kleckner     // Emit the gaps afterwards.
68211cf053bSReid Kleckner     assert((NumGaps == 0 || Bias <= MaxDefRange) &&
683be82d3ecSReid Kleckner            "large ranges should not have gaps");
684be82d3ecSReid Kleckner     unsigned GapStartOffset = GapAndRangeSizes[I].second;
685be82d3ecSReid Kleckner     for (++I; I != J; ++I) {
686be82d3ecSReid Kleckner       unsigned GapSize, RangeSize;
687be82d3ecSReid Kleckner       assert(I < GapAndRangeSizes.size());
688be82d3ecSReid Kleckner       std::tie(GapSize, RangeSize) = GapAndRangeSizes[I];
689be82d3ecSReid Kleckner       LEWriter.write<uint16_t>(GapStartOffset);
69011cf053bSReid Kleckner       LEWriter.write<uint16_t>(GapSize);
691be82d3ecSReid Kleckner       GapStartOffset += GapSize + RangeSize;
692be82d3ecSReid Kleckner     }
693408b5e66SDavid Majnemer   }
694408b5e66SDavid Majnemer }
695