170f5bc99SReid Kleckner //===-- llvm/lib/CodeGen/AsmPrinter/CodeViewDebug.cpp --*- C++ -*--===//
270f5bc99SReid Kleckner //
370f5bc99SReid Kleckner //                     The LLVM Compiler Infrastructure
470f5bc99SReid Kleckner //
570f5bc99SReid Kleckner // This file is distributed under the University of Illinois Open Source
670f5bc99SReid Kleckner // License. See LICENSE.TXT for details.
770f5bc99SReid Kleckner //
870f5bc99SReid Kleckner //===----------------------------------------------------------------------===//
970f5bc99SReid Kleckner //
1070f5bc99SReid Kleckner // This file contains support for writing Microsoft CodeView debug info.
1170f5bc99SReid Kleckner //
1270f5bc99SReid Kleckner //===----------------------------------------------------------------------===//
1370f5bc99SReid Kleckner 
1470f5bc99SReid Kleckner #include "CodeViewDebug.h"
1570f5bc99SReid Kleckner #include "llvm/DebugInfo/CodeView/CodeView.h"
162214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/Line.h"
1770f5bc99SReid Kleckner #include "llvm/DebugInfo/CodeView/SymbolRecord.h"
18f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeIndex.h"
19f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeRecord.h"
2070f5bc99SReid Kleckner #include "llvm/MC/MCExpr.h"
2170f5bc99SReid Kleckner #include "llvm/MC/MCSymbol.h"
2270f5bc99SReid Kleckner #include "llvm/Support/COFF.h"
23f9c275feSReid Kleckner #include "llvm/Target/TargetSubtargetInfo.h"
24f9c275feSReid Kleckner #include "llvm/Target/TargetRegisterInfo.h"
25f9c275feSReid Kleckner #include "llvm/Target/TargetFrameLowering.h"
2670f5bc99SReid Kleckner 
27f9c275feSReid Kleckner using namespace llvm;
2870f5bc99SReid Kleckner using namespace llvm::codeview;
2970f5bc99SReid Kleckner 
30f9c275feSReid Kleckner CodeViewDebug::CodeViewDebug(AsmPrinter *AP)
31f9c275feSReid Kleckner     : DebugHandlerBase(AP), OS(*Asm->OutStreamer), CurFn(nullptr) {
32f9c275feSReid Kleckner   // If module doesn't have named metadata anchors or COFF debug section
33f9c275feSReid Kleckner   // is not available, skip any debug info related stuff.
34f9c275feSReid Kleckner   if (!MMI->getModule()->getNamedMetadata("llvm.dbg.cu") ||
35f9c275feSReid Kleckner       !AP->getObjFileLowering().getCOFFDebugSymbolsSection()) {
36f9c275feSReid Kleckner     Asm = nullptr;
37f9c275feSReid Kleckner     return;
38f9c275feSReid Kleckner   }
39f9c275feSReid Kleckner 
40f9c275feSReid Kleckner   // Tell MMI that we have debug info.
41f9c275feSReid Kleckner   MMI->setDebugInfoAvailability(true);
42f9c275feSReid Kleckner }
4370f5bc99SReid Kleckner 
449533af4fSReid Kleckner StringRef CodeViewDebug::getFullFilepath(const DIFile *File) {
459533af4fSReid Kleckner   std::string &Filepath = FileToFilepathMap[File];
4670f5bc99SReid Kleckner   if (!Filepath.empty())
4770f5bc99SReid Kleckner     return Filepath;
4870f5bc99SReid Kleckner 
499533af4fSReid Kleckner   StringRef Dir = File->getDirectory(), Filename = File->getFilename();
509533af4fSReid Kleckner 
5170f5bc99SReid Kleckner   // Clang emits directory and relative filename info into the IR, but CodeView
5270f5bc99SReid Kleckner   // operates on full paths.  We could change Clang to emit full paths too, but
5370f5bc99SReid Kleckner   // that would increase the IR size and probably not needed for other users.
5470f5bc99SReid Kleckner   // For now, just concatenate and canonicalize the path here.
5570f5bc99SReid Kleckner   if (Filename.find(':') == 1)
5670f5bc99SReid Kleckner     Filepath = Filename;
5770f5bc99SReid Kleckner   else
5870f5bc99SReid Kleckner     Filepath = (Dir + "\\" + Filename).str();
5970f5bc99SReid Kleckner 
6070f5bc99SReid Kleckner   // Canonicalize the path.  We have to do it textually because we may no longer
6170f5bc99SReid Kleckner   // have access the file in the filesystem.
6270f5bc99SReid Kleckner   // First, replace all slashes with backslashes.
6370f5bc99SReid Kleckner   std::replace(Filepath.begin(), Filepath.end(), '/', '\\');
6470f5bc99SReid Kleckner 
6570f5bc99SReid Kleckner   // Remove all "\.\" with "\".
6670f5bc99SReid Kleckner   size_t Cursor = 0;
6770f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\.\\", Cursor)) != std::string::npos)
6870f5bc99SReid Kleckner     Filepath.erase(Cursor, 2);
6970f5bc99SReid Kleckner 
7070f5bc99SReid Kleckner   // Replace all "\XXX\..\" with "\".  Don't try too hard though as the original
7170f5bc99SReid Kleckner   // path should be well-formatted, e.g. start with a drive letter, etc.
7270f5bc99SReid Kleckner   Cursor = 0;
7370f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\..\\", Cursor)) != std::string::npos) {
7470f5bc99SReid Kleckner     // Something's wrong if the path starts with "\..\", abort.
7570f5bc99SReid Kleckner     if (Cursor == 0)
7670f5bc99SReid Kleckner       break;
7770f5bc99SReid Kleckner 
7870f5bc99SReid Kleckner     size_t PrevSlash = Filepath.rfind('\\', Cursor - 1);
7970f5bc99SReid Kleckner     if (PrevSlash == std::string::npos)
8070f5bc99SReid Kleckner       // Something's wrong, abort.
8170f5bc99SReid Kleckner       break;
8270f5bc99SReid Kleckner 
8370f5bc99SReid Kleckner     Filepath.erase(PrevSlash, Cursor + 3 - PrevSlash);
8470f5bc99SReid Kleckner     // The next ".." might be following the one we've just erased.
8570f5bc99SReid Kleckner     Cursor = PrevSlash;
8670f5bc99SReid Kleckner   }
8770f5bc99SReid Kleckner 
8870f5bc99SReid Kleckner   // Remove all duplicate backslashes.
8970f5bc99SReid Kleckner   Cursor = 0;
9070f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\\\", Cursor)) != std::string::npos)
9170f5bc99SReid Kleckner     Filepath.erase(Cursor, 1);
9270f5bc99SReid Kleckner 
9370f5bc99SReid Kleckner   return Filepath;
9470f5bc99SReid Kleckner }
9570f5bc99SReid Kleckner 
962214ed89SReid Kleckner unsigned CodeViewDebug::maybeRecordFile(const DIFile *F) {
972214ed89SReid Kleckner   unsigned NextId = FileIdMap.size() + 1;
982214ed89SReid Kleckner   auto Insertion = FileIdMap.insert(std::make_pair(F, NextId));
992214ed89SReid Kleckner   if (Insertion.second) {
1002214ed89SReid Kleckner     // We have to compute the full filepath and emit a .cv_file directive.
1012214ed89SReid Kleckner     StringRef FullPath = getFullFilepath(F);
102dac21b43SReid Kleckner     NextId = OS.EmitCVFileDirective(NextId, FullPath);
1032214ed89SReid Kleckner     assert(NextId == FileIdMap.size() && ".cv_file directive failed");
1042214ed89SReid Kleckner   }
1052214ed89SReid Kleckner   return Insertion.first->second;
1062214ed89SReid Kleckner }
1072214ed89SReid Kleckner 
108876330d5SReid Kleckner CodeViewDebug::InlineSite &
109876330d5SReid Kleckner CodeViewDebug::getInlineSite(const DILocation *InlinedAt,
110876330d5SReid Kleckner                              const DISubprogram *Inlinee) {
111f3b9ba49SReid Kleckner   auto Insertion = CurFn->InlineSites.insert({InlinedAt, InlineSite()});
112f9c275feSReid Kleckner   InlineSite *Site = &Insertion.first->second;
113f3b9ba49SReid Kleckner   if (Insertion.second) {
114f9c275feSReid Kleckner     Site->SiteFuncId = NextFuncId++;
115876330d5SReid Kleckner     Site->Inlinee = Inlinee;
116876330d5SReid Kleckner     InlinedSubprograms.insert(Inlinee);
117f3b9ba49SReid Kleckner   }
118f9c275feSReid Kleckner   return *Site;
119f3b9ba49SReid Kleckner }
120f3b9ba49SReid Kleckner 
121876330d5SReid Kleckner void CodeViewDebug::recordLocalVariable(LocalVariable &&Var,
122876330d5SReid Kleckner                                         const DILocation *InlinedAt) {
123876330d5SReid Kleckner   if (InlinedAt) {
124876330d5SReid Kleckner     // This variable was inlined. Associate it with the InlineSite.
125876330d5SReid Kleckner     const DISubprogram *Inlinee = Var.DIVar->getScope()->getSubprogram();
126876330d5SReid Kleckner     InlineSite &Site = getInlineSite(InlinedAt, Inlinee);
127876330d5SReid Kleckner     Site.InlinedLocals.emplace_back(Var);
128876330d5SReid Kleckner   } else {
129876330d5SReid Kleckner     // This variable goes in the main ProcSym.
130876330d5SReid Kleckner     CurFn->Locals.emplace_back(Var);
131876330d5SReid Kleckner   }
132876330d5SReid Kleckner }
133876330d5SReid Kleckner 
134829365aeSReid Kleckner static void addLocIfNotPresent(SmallVectorImpl<const DILocation *> &Locs,
135829365aeSReid Kleckner                                const DILocation *Loc) {
136829365aeSReid Kleckner   auto B = Locs.begin(), E = Locs.end();
137829365aeSReid Kleckner   if (std::find(B, E, Loc) == E)
138829365aeSReid Kleckner     Locs.push_back(Loc);
139829365aeSReid Kleckner }
140829365aeSReid Kleckner 
14170f5bc99SReid Kleckner void CodeViewDebug::maybeRecordLocation(DebugLoc DL,
14270f5bc99SReid Kleckner                                         const MachineFunction *MF) {
1439533af4fSReid Kleckner   // Skip this instruction if it has the same location as the previous one.
1449533af4fSReid Kleckner   if (DL == CurFn->LastLoc)
1459533af4fSReid Kleckner     return;
1469533af4fSReid Kleckner 
1479533af4fSReid Kleckner   const DIScope *Scope = DL.get()->getScope();
14870f5bc99SReid Kleckner   if (!Scope)
14970f5bc99SReid Kleckner     return;
1509533af4fSReid Kleckner 
15170f5bc99SReid Kleckner   // Skip this line if it is longer than the maximum we can record.
1522214ed89SReid Kleckner   LineInfo LI(DL.getLine(), DL.getLine(), /*IsStatement=*/true);
1532214ed89SReid Kleckner   if (LI.getStartLine() != DL.getLine() || LI.isAlwaysStepInto() ||
1542214ed89SReid Kleckner       LI.isNeverStepInto())
15570f5bc99SReid Kleckner     return;
15670f5bc99SReid Kleckner 
1572214ed89SReid Kleckner   ColumnInfo CI(DL.getCol(), /*EndColumn=*/0);
1582214ed89SReid Kleckner   if (CI.getStartColumn() != DL.getCol())
1592214ed89SReid Kleckner     return;
16000d9639cSReid Kleckner 
1612214ed89SReid Kleckner   if (!CurFn->HaveLineInfo)
1622214ed89SReid Kleckner     CurFn->HaveLineInfo = true;
1632214ed89SReid Kleckner   unsigned FileId = 0;
1642214ed89SReid Kleckner   if (CurFn->LastLoc.get() && CurFn->LastLoc->getFile() == DL->getFile())
1652214ed89SReid Kleckner     FileId = CurFn->LastFileId;
1662214ed89SReid Kleckner   else
1672214ed89SReid Kleckner     FileId = CurFn->LastFileId = maybeRecordFile(DL->getFile());
1682214ed89SReid Kleckner   CurFn->LastLoc = DL;
169f3b9ba49SReid Kleckner 
170f3b9ba49SReid Kleckner   unsigned FuncId = CurFn->FuncId;
171876330d5SReid Kleckner   if (const DILocation *SiteLoc = DL->getInlinedAt()) {
172829365aeSReid Kleckner     const DILocation *Loc = DL.get();
173829365aeSReid Kleckner 
174f3b9ba49SReid Kleckner     // If this location was actually inlined from somewhere else, give it the ID
175f3b9ba49SReid Kleckner     // of the inline call site.
176876330d5SReid Kleckner     FuncId =
177876330d5SReid Kleckner         getInlineSite(SiteLoc, Loc->getScope()->getSubprogram()).SiteFuncId;
178829365aeSReid Kleckner 
179f3b9ba49SReid Kleckner     // Ensure we have links in the tree of inline call sites.
180829365aeSReid Kleckner     bool FirstLoc = true;
181829365aeSReid Kleckner     while ((SiteLoc = Loc->getInlinedAt())) {
182876330d5SReid Kleckner       InlineSite &Site =
183876330d5SReid Kleckner           getInlineSite(SiteLoc, Loc->getScope()->getSubprogram());
184829365aeSReid Kleckner       if (!FirstLoc)
185829365aeSReid Kleckner         addLocIfNotPresent(Site.ChildSites, Loc);
186829365aeSReid Kleckner       FirstLoc = false;
187829365aeSReid Kleckner       Loc = SiteLoc;
188f3b9ba49SReid Kleckner     }
189829365aeSReid Kleckner     addLocIfNotPresent(CurFn->ChildSites, Loc);
190f3b9ba49SReid Kleckner   }
191f3b9ba49SReid Kleckner 
192dac21b43SReid Kleckner   OS.EmitCVLocDirective(FuncId, FileId, DL.getLine(), DL.getCol(),
193dac21b43SReid Kleckner                         /*PrologueEnd=*/false,
1942214ed89SReid Kleckner                         /*IsStmt=*/false, DL->getFilename());
19570f5bc99SReid Kleckner }
19670f5bc99SReid Kleckner 
19770f5bc99SReid Kleckner void CodeViewDebug::endModule() {
19870f5bc99SReid Kleckner   if (FnDebugInfo.empty())
19970f5bc99SReid Kleckner     return;
20070f5bc99SReid Kleckner 
201f3b9ba49SReid Kleckner   emitTypeInformation();
202f3b9ba49SReid Kleckner 
20370f5bc99SReid Kleckner   // FIXME: For functions that are comdat, we should emit separate .debug$S
20470f5bc99SReid Kleckner   // sections that are comdat associative with the main function instead of
20570f5bc99SReid Kleckner   // having one big .debug$S section.
20670f5bc99SReid Kleckner   assert(Asm != nullptr);
207dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugSymbolsSection());
208dac21b43SReid Kleckner   OS.AddComment("Debug section magic");
209dac21b43SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
21070f5bc99SReid Kleckner 
21170f5bc99SReid Kleckner   // The COFF .debug$S section consists of several subsections, each starting
21270f5bc99SReid Kleckner   // with a 4-byte control code (e.g. 0xF1, 0xF2, etc) and then a 4-byte length
21370f5bc99SReid Kleckner   // of the payload followed by the payload itself.  The subsections are 4-byte
21470f5bc99SReid Kleckner   // aligned.
21570f5bc99SReid Kleckner 
2161fcd610cSReid Kleckner   // Make a subsection for all the inlined subprograms.
2171fcd610cSReid Kleckner   emitInlineeLinesSubsection();
2181fcd610cSReid Kleckner 
2192214ed89SReid Kleckner   // Emit per-function debug information.
2202214ed89SReid Kleckner   for (auto &P : FnDebugInfo)
2212214ed89SReid Kleckner     emitDebugInfoForFunction(P.first, P.second);
22270f5bc99SReid Kleckner 
22370f5bc99SReid Kleckner   // This subsection holds a file index to offset in string table table.
224dac21b43SReid Kleckner   OS.AddComment("File index to string table offset subsection");
225dac21b43SReid Kleckner   OS.EmitCVFileChecksumsDirective();
22670f5bc99SReid Kleckner 
22770f5bc99SReid Kleckner   // This subsection holds the string table.
228dac21b43SReid Kleckner   OS.AddComment("String table");
229dac21b43SReid Kleckner   OS.EmitCVStringTableDirective();
23070f5bc99SReid Kleckner 
23170f5bc99SReid Kleckner   clear();
23270f5bc99SReid Kleckner }
23370f5bc99SReid Kleckner 
234*b9456a5eSDavid Majnemer static void emitNullTerminatedSymbolName(MCStreamer &OS, StringRef S) {
235*b9456a5eSDavid Majnemer   // Microsoft's linker seems to have trouble with symbol names longer than
236*b9456a5eSDavid Majnemer   // 0xffd8 bytes.
237*b9456a5eSDavid Majnemer   S = S.substr(0, 0xffd8);
238*b9456a5eSDavid Majnemer   SmallString<32> NullTerminatedString(S);
239*b9456a5eSDavid Majnemer   NullTerminatedString.push_back('\0');
240*b9456a5eSDavid Majnemer   OS.EmitBytes(NullTerminatedString);
241*b9456a5eSDavid Majnemer }
242*b9456a5eSDavid Majnemer 
243f3b9ba49SReid Kleckner void CodeViewDebug::emitTypeInformation() {
244f3b9ba49SReid Kleckner   // Start the .debug$T section with 0x4.
245dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugTypesSection());
246dac21b43SReid Kleckner   OS.AddComment("Debug section magic");
247dac21b43SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
248f3b9ba49SReid Kleckner 
249f3b9ba49SReid Kleckner   NamedMDNode *CU_Nodes =
250f9c275feSReid Kleckner       MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
251f3b9ba49SReid Kleckner   if (!CU_Nodes)
252f3b9ba49SReid Kleckner     return;
253f3b9ba49SReid Kleckner 
254f3b9ba49SReid Kleckner   // This type info currently only holds function ids for use with inline call
255f3b9ba49SReid Kleckner   // frame info. All functions are assigned a simple 'void ()' type. Emit that
256f3b9ba49SReid Kleckner   // type here.
257f3b9ba49SReid Kleckner   TypeIndex ArgListIdx = getNextTypeIndex();
258dac21b43SReid Kleckner   OS.AddComment("Type record length");
259dac21b43SReid Kleckner   OS.EmitIntValue(2 + sizeof(ArgList), 2);
260dac21b43SReid Kleckner   OS.AddComment("Leaf type: LF_ARGLIST");
261dac21b43SReid Kleckner   OS.EmitIntValue(LF_ARGLIST, 2);
262dac21b43SReid Kleckner   OS.AddComment("Number of arguments");
263dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
264f3b9ba49SReid Kleckner 
265f3b9ba49SReid Kleckner   TypeIndex VoidProcIdx = getNextTypeIndex();
266dac21b43SReid Kleckner   OS.AddComment("Type record length");
267dac21b43SReid Kleckner   OS.EmitIntValue(2 + sizeof(ProcedureType), 2);
268dac21b43SReid Kleckner   OS.AddComment("Leaf type: LF_PROCEDURE");
269dac21b43SReid Kleckner   OS.EmitIntValue(LF_PROCEDURE, 2);
270dac21b43SReid Kleckner   OS.AddComment("Return type index");
271dac21b43SReid Kleckner   OS.EmitIntValue(TypeIndex::Void().getIndex(), 4);
272dac21b43SReid Kleckner   OS.AddComment("Calling convention");
273dac21b43SReid Kleckner   OS.EmitIntValue(char(CallingConvention::NearC), 1);
274dac21b43SReid Kleckner   OS.AddComment("Function options");
275dac21b43SReid Kleckner   OS.EmitIntValue(char(FunctionOptions::None), 1);
276dac21b43SReid Kleckner   OS.AddComment("# of parameters");
277dac21b43SReid Kleckner   OS.EmitIntValue(0, 2);
278dac21b43SReid Kleckner   OS.AddComment("Argument list type index");
279dac21b43SReid Kleckner   OS.EmitIntValue(ArgListIdx.getIndex(), 4);
280f3b9ba49SReid Kleckner 
281f3b9ba49SReid Kleckner   for (MDNode *N : CU_Nodes->operands()) {
282f3b9ba49SReid Kleckner     auto *CUNode = cast<DICompileUnit>(N);
283f3b9ba49SReid Kleckner     for (auto *SP : CUNode->getSubprograms()) {
284f3b9ba49SReid Kleckner       StringRef DisplayName = SP->getDisplayName();
285dac21b43SReid Kleckner       OS.AddComment("Type record length");
286*b9456a5eSDavid Majnemer       MCSymbol *FuncBegin = MMI->getContext().createTempSymbol(),
287*b9456a5eSDavid Majnemer                *FuncEnd = MMI->getContext().createTempSymbol();
288*b9456a5eSDavid Majnemer       OS.emitAbsoluteSymbolDiff(FuncEnd, FuncBegin, 2);
289*b9456a5eSDavid Majnemer       OS.EmitLabel(FuncBegin);
290dac21b43SReid Kleckner       OS.AddComment("Leaf type: LF_FUNC_ID");
291dac21b43SReid Kleckner       OS.EmitIntValue(LF_FUNC_ID, 2);
292f3b9ba49SReid Kleckner 
293dac21b43SReid Kleckner       OS.AddComment("Scope type index");
294dac21b43SReid Kleckner       OS.EmitIntValue(TypeIndex().getIndex(), 4);
295dac21b43SReid Kleckner       OS.AddComment("Function type");
296dac21b43SReid Kleckner       OS.EmitIntValue(VoidProcIdx.getIndex(), 4);
29730579ec8SDavid Majnemer       {
298dac21b43SReid Kleckner         OS.AddComment("Function name");
299*b9456a5eSDavid Majnemer         emitNullTerminatedSymbolName(OS, DisplayName);
30030579ec8SDavid Majnemer       }
301*b9456a5eSDavid Majnemer       OS.EmitLabel(FuncEnd);
302f3b9ba49SReid Kleckner 
303f3b9ba49SReid Kleckner       TypeIndex FuncIdIdx = getNextTypeIndex();
304f3b9ba49SReid Kleckner       SubprogramToFuncId.insert(std::make_pair(SP, FuncIdIdx));
305f3b9ba49SReid Kleckner     }
306f3b9ba49SReid Kleckner   }
307f3b9ba49SReid Kleckner }
308f3b9ba49SReid Kleckner 
3091fcd610cSReid Kleckner void CodeViewDebug::emitInlineeLinesSubsection() {
3101fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
3111fcd610cSReid Kleckner     return;
3121fcd610cSReid Kleckner 
313f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
314f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
3151fcd610cSReid Kleckner 
3161fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
3171fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::InlineeLines), 4);
31830579ec8SDavid Majnemer   OS.AddComment("Subsection size");
3191fcd610cSReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 4);
3201fcd610cSReid Kleckner   OS.EmitLabel(InlineBegin);
3211fcd610cSReid Kleckner 
3221fcd610cSReid Kleckner   // We don't provide any extra file info.
3231fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
32430579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
3251fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
3261fcd610cSReid Kleckner 
3271fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
32830579ec8SDavid Majnemer     OS.AddBlankLine();
3291fcd610cSReid Kleckner     TypeIndex TypeId = SubprogramToFuncId[SP];
3301fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
3311fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
3321fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
33330579ec8SDavid Majnemer     OS.AddBlankLine();
3341fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
3351fcd610cSReid Kleckner     // 1.
3361fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
33730579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
3381fcd610cSReid Kleckner     OS.EmitIntValue(TypeId.getIndex(), 4);
33930579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
3401fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
34130579ec8SDavid Majnemer     OS.AddComment("Starting line number");
3421fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
3431fcd610cSReid Kleckner   }
3441fcd610cSReid Kleckner 
3451fcd610cSReid Kleckner   OS.EmitLabel(InlineEnd);
3461fcd610cSReid Kleckner }
3471fcd610cSReid Kleckner 
3481fcd610cSReid Kleckner void CodeViewDebug::collectInlineSiteChildren(
3491fcd610cSReid Kleckner     SmallVectorImpl<unsigned> &Children, const FunctionInfo &FI,
3501fcd610cSReid Kleckner     const InlineSite &Site) {
3511fcd610cSReid Kleckner   for (const DILocation *ChildSiteLoc : Site.ChildSites) {
3521fcd610cSReid Kleckner     auto I = FI.InlineSites.find(ChildSiteLoc);
3531fcd610cSReid Kleckner     const InlineSite &ChildSite = I->second;
3541fcd610cSReid Kleckner     Children.push_back(ChildSite.SiteFuncId);
3551fcd610cSReid Kleckner     collectInlineSiteChildren(Children, FI, ChildSite);
3561fcd610cSReid Kleckner   }
3571fcd610cSReid Kleckner }
3581fcd610cSReid Kleckner 
359f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
360f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
361f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
362f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
363f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
364f3b9ba49SReid Kleckner 
365f3b9ba49SReid Kleckner   assert(SubprogramToFuncId.count(Site.Inlinee));
366f3b9ba49SReid Kleckner   TypeIndex InlineeIdx = SubprogramToFuncId[Site.Inlinee];
367f3b9ba49SReid Kleckner 
368f3b9ba49SReid Kleckner   // SymbolRecord
369dac21b43SReid Kleckner   OS.AddComment("Record length");
370eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
371f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
372dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
373dac21b43SReid Kleckner   OS.EmitIntValue(SymbolRecordKind::S_INLINESITE, 2); // RecordKind
374f3b9ba49SReid Kleckner 
375dac21b43SReid Kleckner   OS.AddComment("PtrParent");
376dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
377dac21b43SReid Kleckner   OS.AddComment("PtrEnd");
378dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
379dac21b43SReid Kleckner   OS.AddComment("Inlinee type index");
380dac21b43SReid Kleckner   OS.EmitIntValue(InlineeIdx.getIndex(), 4);
381f3b9ba49SReid Kleckner 
3821fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
3831fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
3841fcd610cSReid Kleckner   SmallVector<unsigned, 3> SecondaryFuncIds;
3851fcd610cSReid Kleckner   collectInlineSiteChildren(SecondaryFuncIds, FI, Site);
3861fcd610cSReid Kleckner 
3871fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
388c9911f28SDavid Majnemer                                     FI.Begin, FI.End, SecondaryFuncIds);
389f3b9ba49SReid Kleckner 
390f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
391f3b9ba49SReid Kleckner 
392f9c275feSReid Kleckner   for (const LocalVariable &Var : Site.InlinedLocals)
393f9c275feSReid Kleckner     emitLocalVariable(Var);
394f9c275feSReid Kleckner 
395f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
396f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
397f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
398f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
399f3b9ba49SReid Kleckner            "child site not in function inline site map");
400f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
401f3b9ba49SReid Kleckner   }
402f3b9ba49SReid Kleckner 
403f3b9ba49SReid Kleckner   // Close the scope.
404dac21b43SReid Kleckner   OS.AddComment("Record length");
405dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
406dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
407dac21b43SReid Kleckner   OS.EmitIntValue(SymbolRecordKind::S_INLINESITE_END, 2); // RecordKind
408f3b9ba49SReid Kleckner }
409f3b9ba49SReid Kleckner 
4102214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
4112214ed89SReid Kleckner                                              FunctionInfo &FI) {
41270f5bc99SReid Kleckner   // For each function there is a separate subsection
41370f5bc99SReid Kleckner   // which holds the PC to file:line table.
41470f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
41570f5bc99SReid Kleckner   assert(Fn);
41670f5bc99SReid Kleckner 
41770f5bc99SReid Kleckner   StringRef FuncName;
418adebb937SPete Cooper   if (auto *SP = GV->getSubprogram())
41970f5bc99SReid Kleckner     FuncName = SP->getDisplayName();
42070f5bc99SReid Kleckner 
42170f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
42270f5bc99SReid Kleckner   if (FuncName.empty())
42370f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
42470f5bc99SReid Kleckner 
42570f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
426f9c275feSReid Kleckner   MCSymbol *SymbolsBegin = MMI->getContext().createTempSymbol(),
427f9c275feSReid Kleckner            *SymbolsEnd = MMI->getContext().createTempSymbol();
428dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
429dac21b43SReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::Symbols), 4);
430dac21b43SReid Kleckner   OS.AddComment("Subsection size");
431eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(SymbolsEnd, SymbolsBegin, 4);
432dac21b43SReid Kleckner   OS.EmitLabel(SymbolsBegin);
43370f5bc99SReid Kleckner   {
434f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
435f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
436dac21b43SReid Kleckner     OS.AddComment("Record length");
437eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
438dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
43970f5bc99SReid Kleckner 
440dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
441dac21b43SReid Kleckner     OS.EmitIntValue(unsigned(SymbolRecordKind::S_GPROC32_ID), 2);
44270f5bc99SReid Kleckner 
44330579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
444dac21b43SReid Kleckner     OS.AddComment("PtrParent");
445dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
446dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
447dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
448dac21b43SReid Kleckner     OS.AddComment("PtrNext");
449dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
45070f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
45170f5bc99SReid Kleckner     // code is located and what's its size:
452dac21b43SReid Kleckner     OS.AddComment("Code size");
453eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
454dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
455dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
456dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
457dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
458dac21b43SReid Kleckner     OS.AddComment("Function type index");
459dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
460dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
461dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
462dac21b43SReid Kleckner     OS.AddComment("Function section index");
463dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
464dac21b43SReid Kleckner     OS.AddComment("Flags");
465dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
46670f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
467dac21b43SReid Kleckner     OS.AddComment("Function name");
4681256125fSDavid Majnemer     // Truncate the name so we won't overflow the record length field.
469*b9456a5eSDavid Majnemer     emitNullTerminatedSymbolName(OS, FuncName);
470dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
47170f5bc99SReid Kleckner 
472f9c275feSReid Kleckner     for (const LocalVariable &Var : FI.Locals)
473f9c275feSReid Kleckner       emitLocalVariable(Var);
474f9c275feSReid Kleckner 
475f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
476f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
477f3b9ba49SReid Kleckner     // of their parent inline site.
478f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
479f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
480f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
481f9c275feSReid Kleckner              "child site not in function inline site map");
482f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
483f3b9ba49SReid Kleckner     }
484f3b9ba49SReid Kleckner 
48570f5bc99SReid Kleckner     // We're done with this function.
486dac21b43SReid Kleckner     OS.AddComment("Record length");
487dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
488dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
489dac21b43SReid Kleckner     OS.EmitIntValue(unsigned(SymbolRecordKind::S_PROC_ID_END), 2);
49070f5bc99SReid Kleckner   }
491dac21b43SReid Kleckner   OS.EmitLabel(SymbolsEnd);
49270f5bc99SReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
493dac21b43SReid Kleckner   OS.EmitValueToAlignment(4);
49470f5bc99SReid Kleckner 
4952214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
496dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
49770f5bc99SReid Kleckner }
49870f5bc99SReid Kleckner 
499876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
500876330d5SReid Kleckner CodeViewDebug::createDefRangeMem(uint16_t CVRegister, int Offset) {
501876330d5SReid Kleckner   LocalVarDefRange DR;
502c6a2f214SAaron Ballman   DR.InMemory = -1;
503876330d5SReid Kleckner   DR.DataOffset = Offset;
504876330d5SReid Kleckner   assert(DR.DataOffset == Offset && "truncation");
505876330d5SReid Kleckner   DR.StructOffset = 0;
506876330d5SReid Kleckner   DR.CVRegister = CVRegister;
507876330d5SReid Kleckner   return DR;
508876330d5SReid Kleckner }
509876330d5SReid Kleckner 
510876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
511876330d5SReid Kleckner CodeViewDebug::createDefRangeReg(uint16_t CVRegister) {
512876330d5SReid Kleckner   LocalVarDefRange DR;
513876330d5SReid Kleckner   DR.InMemory = 0;
514876330d5SReid Kleckner   DR.DataOffset = 0;
515876330d5SReid Kleckner   DR.StructOffset = 0;
516876330d5SReid Kleckner   DR.CVRegister = CVRegister;
517876330d5SReid Kleckner   return DR;
518876330d5SReid Kleckner }
519876330d5SReid Kleckner 
520876330d5SReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable(
521876330d5SReid Kleckner     DenseSet<InlinedVariable> &Processed) {
522876330d5SReid Kleckner   const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
523876330d5SReid Kleckner   const TargetFrameLowering *TFI = TSI.getFrameLowering();
524876330d5SReid Kleckner   const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
525876330d5SReid Kleckner 
526876330d5SReid Kleckner   for (const MachineModuleInfo::VariableDbgInfo &VI :
527876330d5SReid Kleckner        MMI->getVariableDbgInfo()) {
528f9c275feSReid Kleckner     if (!VI.Var)
529f9c275feSReid Kleckner       continue;
530f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
531f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
532f9c275feSReid Kleckner 
533876330d5SReid Kleckner     Processed.insert(InlinedVariable(VI.Var, VI.Loc->getInlinedAt()));
534f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
535f9c275feSReid Kleckner 
536f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
537f9c275feSReid Kleckner     if (!Scope)
538f9c275feSReid Kleckner       continue;
539f9c275feSReid Kleckner 
540f9c275feSReid Kleckner     // Get the frame register used and the offset.
541f9c275feSReid Kleckner     unsigned FrameReg = 0;
542876330d5SReid Kleckner     int FrameOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
543876330d5SReid Kleckner     uint16_t CVReg = TRI->getCodeViewRegNum(FrameReg);
544f9c275feSReid Kleckner 
545f9c275feSReid Kleckner     // Calculate the label ranges.
546876330d5SReid Kleckner     LocalVarDefRange DefRange = createDefRangeMem(CVReg, FrameOffset);
547f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
548f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
549f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
550876330d5SReid Kleckner       End = End ? End : Asm->getFunctionEnd();
551876330d5SReid Kleckner       DefRange.Ranges.emplace_back(Begin, End);
552f9c275feSReid Kleckner     }
553f9c275feSReid Kleckner 
554876330d5SReid Kleckner     LocalVariable Var;
555876330d5SReid Kleckner     Var.DIVar = VI.Var;
556876330d5SReid Kleckner     Var.DefRanges.emplace_back(std::move(DefRange));
557876330d5SReid Kleckner     recordLocalVariable(std::move(Var), VI.Loc->getInlinedAt());
558f9c275feSReid Kleckner   }
559f9c275feSReid Kleckner }
560876330d5SReid Kleckner 
561876330d5SReid Kleckner void CodeViewDebug::collectVariableInfo(const DISubprogram *SP) {
562876330d5SReid Kleckner   DenseSet<InlinedVariable> Processed;
563876330d5SReid Kleckner   // Grab the variable info that was squirreled away in the MMI side-table.
564876330d5SReid Kleckner   collectVariableInfoFromMMITable(Processed);
565876330d5SReid Kleckner 
566876330d5SReid Kleckner   const TargetRegisterInfo *TRI = Asm->MF->getSubtarget().getRegisterInfo();
567876330d5SReid Kleckner 
568876330d5SReid Kleckner   for (const auto &I : DbgValues) {
569876330d5SReid Kleckner     InlinedVariable IV = I.first;
570876330d5SReid Kleckner     if (Processed.count(IV))
571876330d5SReid Kleckner       continue;
572876330d5SReid Kleckner     const DILocalVariable *DIVar = IV.first;
573876330d5SReid Kleckner     const DILocation *InlinedAt = IV.second;
574876330d5SReid Kleckner 
575876330d5SReid Kleckner     // Instruction ranges, specifying where IV is accessible.
576876330d5SReid Kleckner     const auto &Ranges = I.second;
577876330d5SReid Kleckner 
578876330d5SReid Kleckner     LexicalScope *Scope = nullptr;
579876330d5SReid Kleckner     if (InlinedAt)
580876330d5SReid Kleckner       Scope = LScopes.findInlinedScope(DIVar->getScope(), InlinedAt);
581876330d5SReid Kleckner     else
582876330d5SReid Kleckner       Scope = LScopes.findLexicalScope(DIVar->getScope());
583876330d5SReid Kleckner     // If variable scope is not found then skip this variable.
584876330d5SReid Kleckner     if (!Scope)
585876330d5SReid Kleckner       continue;
586876330d5SReid Kleckner 
587876330d5SReid Kleckner     LocalVariable Var;
588876330d5SReid Kleckner     Var.DIVar = DIVar;
589876330d5SReid Kleckner 
590876330d5SReid Kleckner     // Calculate the definition ranges.
591876330d5SReid Kleckner     for (auto I = Ranges.begin(), E = Ranges.end(); I != E; ++I) {
592876330d5SReid Kleckner       const InsnRange &Range = *I;
593876330d5SReid Kleckner       const MachineInstr *DVInst = Range.first;
594876330d5SReid Kleckner       assert(DVInst->isDebugValue() && "Invalid History entry");
595876330d5SReid Kleckner       const DIExpression *DIExpr = DVInst->getDebugExpression();
596876330d5SReid Kleckner 
597876330d5SReid Kleckner       // Bail if there is a complex DWARF expression for now.
598876330d5SReid Kleckner       if (DIExpr && DIExpr->getNumElements() > 0)
599876330d5SReid Kleckner         continue;
600876330d5SReid Kleckner 
6019a593ee7SReid Kleckner       // Bail if operand 0 is not a valid register. This means the variable is a
6029a593ee7SReid Kleckner       // simple constant, or is described by a complex expression.
6039a593ee7SReid Kleckner       // FIXME: Find a way to represent constant variables, since they are
6049a593ee7SReid Kleckner       // relatively common.
6059a593ee7SReid Kleckner       unsigned Reg =
6069a593ee7SReid Kleckner           DVInst->getOperand(0).isReg() ? DVInst->getOperand(0).getReg() : 0;
6079a593ee7SReid Kleckner       if (Reg == 0)
6086e0d5f57SReid Kleckner         continue;
6096e0d5f57SReid Kleckner 
610876330d5SReid Kleckner       // Handle the two cases we can handle: indirect in memory and in register.
611876330d5SReid Kleckner       bool IsIndirect = DVInst->getOperand(1).isImm();
612876330d5SReid Kleckner       unsigned CVReg = TRI->getCodeViewRegNum(DVInst->getOperand(0).getReg());
613876330d5SReid Kleckner       {
614876330d5SReid Kleckner         LocalVarDefRange DefRange;
615876330d5SReid Kleckner         if (IsIndirect) {
616876330d5SReid Kleckner           int64_t Offset = DVInst->getOperand(1).getImm();
617876330d5SReid Kleckner           DefRange = createDefRangeMem(CVReg, Offset);
618876330d5SReid Kleckner         } else {
619876330d5SReid Kleckner           DefRange = createDefRangeReg(CVReg);
620876330d5SReid Kleckner         }
621876330d5SReid Kleckner         if (Var.DefRanges.empty() ||
622876330d5SReid Kleckner             Var.DefRanges.back().isDifferentLocation(DefRange)) {
623876330d5SReid Kleckner           Var.DefRanges.emplace_back(std::move(DefRange));
624876330d5SReid Kleckner         }
625876330d5SReid Kleckner       }
626876330d5SReid Kleckner 
627876330d5SReid Kleckner       // Compute the label range.
628876330d5SReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
629876330d5SReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
630876330d5SReid Kleckner       if (!End) {
631876330d5SReid Kleckner         if (std::next(I) != E)
632876330d5SReid Kleckner           End = getLabelBeforeInsn(std::next(I)->first);
633876330d5SReid Kleckner         else
634876330d5SReid Kleckner           End = Asm->getFunctionEnd();
635876330d5SReid Kleckner       }
636876330d5SReid Kleckner 
637876330d5SReid Kleckner       // If the last range end is our begin, just extend the last range.
638876330d5SReid Kleckner       // Otherwise make a new range.
639876330d5SReid Kleckner       SmallVectorImpl<std::pair<const MCSymbol *, const MCSymbol *>> &Ranges =
640876330d5SReid Kleckner           Var.DefRanges.back().Ranges;
641876330d5SReid Kleckner       if (!Ranges.empty() && Ranges.back().second == Begin)
642876330d5SReid Kleckner         Ranges.back().second = End;
643876330d5SReid Kleckner       else
644876330d5SReid Kleckner         Ranges.emplace_back(Begin, End);
645876330d5SReid Kleckner 
646876330d5SReid Kleckner       // FIXME: Do more range combining.
647876330d5SReid Kleckner     }
648876330d5SReid Kleckner 
649876330d5SReid Kleckner     recordLocalVariable(std::move(Var), InlinedAt);
650876330d5SReid Kleckner   }
651f9c275feSReid Kleckner }
652f9c275feSReid Kleckner 
65370f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
65470f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
65570f5bc99SReid Kleckner 
656f9c275feSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
65770f5bc99SReid Kleckner     return;
65870f5bc99SReid Kleckner 
659f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
660f9c275feSReid Kleckner 
66170f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
66270f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
66370f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
6642214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
6651fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
66670f5bc99SReid Kleckner 
667f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
668f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
66970f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
67070f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
67170f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
67270f5bc99SReid Kleckner   bool EmptyPrologue = true;
67370f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
67470f5bc99SReid Kleckner     for (const auto &MI : MBB) {
675f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
676f9c275feSReid Kleckner           MI.getDebugLoc()) {
67770f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
67870f5bc99SReid Kleckner         break;
679f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
68070f5bc99SReid Kleckner         EmptyPrologue = false;
68170f5bc99SReid Kleckner       }
68270f5bc99SReid Kleckner     }
683f9c275feSReid Kleckner   }
684f9c275feSReid Kleckner 
68570f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
68670f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
68770f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
68870f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
68970f5bc99SReid Kleckner   }
69070f5bc99SReid Kleckner }
69170f5bc99SReid Kleckner 
692f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
693f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
694f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
695f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
696f9c275feSReid Kleckner   OS.AddComment("Record length");
697f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
698f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
699f9c275feSReid Kleckner 
700f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
701f9c275feSReid Kleckner   OS.EmitIntValue(unsigned(SymbolRecordKind::S_LOCAL), 2);
702f9c275feSReid Kleckner 
703f9c275feSReid Kleckner   uint16_t Flags = 0;
704f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
705f9c275feSReid Kleckner     Flags |= LocalSym::IsParameter;
706876330d5SReid Kleckner   if (Var.DefRanges.empty())
707876330d5SReid Kleckner     Flags |= LocalSym::IsOptimizedOut;
708f9c275feSReid Kleckner 
709f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
710f9c275feSReid Kleckner   OS.EmitIntValue(TypeIndex::Int32().getIndex(), 4);
711f9c275feSReid Kleckner   OS.AddComment("Flags");
712f9c275feSReid Kleckner   OS.EmitIntValue(Flags, 2);
7131256125fSDavid Majnemer   // Truncate the name so we won't overflow the record length field.
714*b9456a5eSDavid Majnemer   emitNullTerminatedSymbolName(OS, Var.DIVar->getName());
715f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
716f9c275feSReid Kleckner 
717876330d5SReid Kleckner   // Calculate the on disk prefix of the appropriate def range record. The
718876330d5SReid Kleckner   // records and on disk formats are described in SymbolRecords.h. BytePrefix
719876330d5SReid Kleckner   // should be big enough to hold all forms without memory allocation.
720876330d5SReid Kleckner   SmallString<20> BytePrefix;
721876330d5SReid Kleckner   for (const LocalVarDefRange &DefRange : Var.DefRanges) {
722876330d5SReid Kleckner     BytePrefix.clear();
723876330d5SReid Kleckner     // FIXME: Handle bitpieces.
724876330d5SReid Kleckner     if (DefRange.StructOffset != 0)
725876330d5SReid Kleckner       continue;
726876330d5SReid Kleckner 
727876330d5SReid Kleckner     if (DefRange.InMemory) {
728f9c275feSReid Kleckner       DefRangeRegisterRelSym Sym{};
729f9c275feSReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
730876330d5SReid Kleckner       Sym.BaseRegister = DefRange.CVRegister;
731f9c275feSReid Kleckner       Sym.Flags = 0; // Unclear what matters here.
732876330d5SReid Kleckner       Sym.BasePointerOffset = DefRange.DataOffset;
733876330d5SReid Kleckner       BytePrefix +=
734876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
735f9c275feSReid Kleckner       BytePrefix += StringRef(reinterpret_cast<const char *>(&Sym),
736f9c275feSReid Kleckner                               sizeof(Sym) - sizeof(LocalVariableAddrRange));
737876330d5SReid Kleckner     } else {
738876330d5SReid Kleckner       assert(DefRange.DataOffset == 0 && "unexpected offset into register");
739876330d5SReid Kleckner       DefRangeRegisterSym Sym{};
740876330d5SReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER);
741876330d5SReid Kleckner       Sym.Register = DefRange.CVRegister;
742876330d5SReid Kleckner       Sym.MayHaveNoName = 0; // Unclear what matters here.
743876330d5SReid Kleckner       BytePrefix +=
744876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
745876330d5SReid Kleckner       BytePrefix += StringRef(reinterpret_cast<const char *>(&Sym),
746876330d5SReid Kleckner                               sizeof(Sym) - sizeof(LocalVariableAddrRange));
747876330d5SReid Kleckner     }
748876330d5SReid Kleckner     OS.EmitCVDefRangeDirective(DefRange.Ranges, BytePrefix);
749876330d5SReid Kleckner   }
750f9c275feSReid Kleckner }
751f9c275feSReid Kleckner 
75270f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
75370f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
75470f5bc99SReid Kleckner     return;
75570f5bc99SReid Kleckner 
75670f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
75770f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
75870f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
75970f5bc99SReid Kleckner 
760adebb937SPete Cooper   collectVariableInfo(GV->getSubprogram());
761876330d5SReid Kleckner 
762876330d5SReid Kleckner   DebugHandlerBase::endFunction(MF);
763876330d5SReid Kleckner 
7642214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
7652214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
76670f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
767f9c275feSReid Kleckner     CurFn = nullptr;
768f9c275feSReid Kleckner     return;
76970f5bc99SReid Kleckner   }
770f9c275feSReid Kleckner 
771f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
772f9c275feSReid Kleckner 
77370f5bc99SReid Kleckner   CurFn = nullptr;
77470f5bc99SReid Kleckner }
77570f5bc99SReid Kleckner 
77670f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
777f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
778f9c275feSReid Kleckner 
77970f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
78070f5bc99SReid Kleckner   if (!Asm || MI->isDebugValue() || MI->getFlag(MachineInstr::FrameSetup))
78170f5bc99SReid Kleckner     return;
78270f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
78370f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
78470f5bc99SReid Kleckner     return;
78570f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
78670f5bc99SReid Kleckner }
787