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) {
111*fbd7787dSReid Kleckner   auto SiteInsertion = CurFn->InlineSites.insert({InlinedAt, InlineSite()});
112*fbd7787dSReid Kleckner   InlineSite *Site = &SiteInsertion.first->second;
113*fbd7787dSReid Kleckner   if (SiteInsertion.second) {
114f9c275feSReid Kleckner     Site->SiteFuncId = NextFuncId++;
115876330d5SReid Kleckner     Site->Inlinee = Inlinee;
116*fbd7787dSReid Kleckner     auto InlineeInsertion =
117*fbd7787dSReid Kleckner         SubprogramIndices.insert({Inlinee, InlinedSubprograms.size()});
118*fbd7787dSReid Kleckner     if (InlineeInsertion.second)
119*fbd7787dSReid Kleckner       InlinedSubprograms.push_back(Inlinee);
120f3b9ba49SReid Kleckner   }
121f9c275feSReid Kleckner   return *Site;
122f3b9ba49SReid Kleckner }
123f3b9ba49SReid Kleckner 
124876330d5SReid Kleckner void CodeViewDebug::recordLocalVariable(LocalVariable &&Var,
125876330d5SReid Kleckner                                         const DILocation *InlinedAt) {
126876330d5SReid Kleckner   if (InlinedAt) {
127876330d5SReid Kleckner     // This variable was inlined. Associate it with the InlineSite.
128876330d5SReid Kleckner     const DISubprogram *Inlinee = Var.DIVar->getScope()->getSubprogram();
129876330d5SReid Kleckner     InlineSite &Site = getInlineSite(InlinedAt, Inlinee);
130876330d5SReid Kleckner     Site.InlinedLocals.emplace_back(Var);
131876330d5SReid Kleckner   } else {
132876330d5SReid Kleckner     // This variable goes in the main ProcSym.
133876330d5SReid Kleckner     CurFn->Locals.emplace_back(Var);
134876330d5SReid Kleckner   }
135876330d5SReid Kleckner }
136876330d5SReid Kleckner 
137829365aeSReid Kleckner static void addLocIfNotPresent(SmallVectorImpl<const DILocation *> &Locs,
138829365aeSReid Kleckner                                const DILocation *Loc) {
139829365aeSReid Kleckner   auto B = Locs.begin(), E = Locs.end();
140829365aeSReid Kleckner   if (std::find(B, E, Loc) == E)
141829365aeSReid Kleckner     Locs.push_back(Loc);
142829365aeSReid Kleckner }
143829365aeSReid Kleckner 
14470f5bc99SReid Kleckner void CodeViewDebug::maybeRecordLocation(DebugLoc DL,
14570f5bc99SReid Kleckner                                         const MachineFunction *MF) {
1469533af4fSReid Kleckner   // Skip this instruction if it has the same location as the previous one.
1479533af4fSReid Kleckner   if (DL == CurFn->LastLoc)
1489533af4fSReid Kleckner     return;
1499533af4fSReid Kleckner 
1509533af4fSReid Kleckner   const DIScope *Scope = DL.get()->getScope();
15170f5bc99SReid Kleckner   if (!Scope)
15270f5bc99SReid Kleckner     return;
1539533af4fSReid Kleckner 
15470f5bc99SReid Kleckner   // Skip this line if it is longer than the maximum we can record.
1552214ed89SReid Kleckner   LineInfo LI(DL.getLine(), DL.getLine(), /*IsStatement=*/true);
1562214ed89SReid Kleckner   if (LI.getStartLine() != DL.getLine() || LI.isAlwaysStepInto() ||
1572214ed89SReid Kleckner       LI.isNeverStepInto())
15870f5bc99SReid Kleckner     return;
15970f5bc99SReid Kleckner 
1602214ed89SReid Kleckner   ColumnInfo CI(DL.getCol(), /*EndColumn=*/0);
1612214ed89SReid Kleckner   if (CI.getStartColumn() != DL.getCol())
1622214ed89SReid Kleckner     return;
16300d9639cSReid Kleckner 
1642214ed89SReid Kleckner   if (!CurFn->HaveLineInfo)
1652214ed89SReid Kleckner     CurFn->HaveLineInfo = true;
1662214ed89SReid Kleckner   unsigned FileId = 0;
1672214ed89SReid Kleckner   if (CurFn->LastLoc.get() && CurFn->LastLoc->getFile() == DL->getFile())
1682214ed89SReid Kleckner     FileId = CurFn->LastFileId;
1692214ed89SReid Kleckner   else
1702214ed89SReid Kleckner     FileId = CurFn->LastFileId = maybeRecordFile(DL->getFile());
1712214ed89SReid Kleckner   CurFn->LastLoc = DL;
172f3b9ba49SReid Kleckner 
173f3b9ba49SReid Kleckner   unsigned FuncId = CurFn->FuncId;
174876330d5SReid Kleckner   if (const DILocation *SiteLoc = DL->getInlinedAt()) {
175829365aeSReid Kleckner     const DILocation *Loc = DL.get();
176829365aeSReid Kleckner 
177f3b9ba49SReid Kleckner     // If this location was actually inlined from somewhere else, give it the ID
178f3b9ba49SReid Kleckner     // of the inline call site.
179876330d5SReid Kleckner     FuncId =
180876330d5SReid Kleckner         getInlineSite(SiteLoc, Loc->getScope()->getSubprogram()).SiteFuncId;
181829365aeSReid Kleckner 
182f3b9ba49SReid Kleckner     // Ensure we have links in the tree of inline call sites.
183829365aeSReid Kleckner     bool FirstLoc = true;
184829365aeSReid Kleckner     while ((SiteLoc = Loc->getInlinedAt())) {
185876330d5SReid Kleckner       InlineSite &Site =
186876330d5SReid Kleckner           getInlineSite(SiteLoc, Loc->getScope()->getSubprogram());
187829365aeSReid Kleckner       if (!FirstLoc)
188829365aeSReid Kleckner         addLocIfNotPresent(Site.ChildSites, Loc);
189829365aeSReid Kleckner       FirstLoc = false;
190829365aeSReid Kleckner       Loc = SiteLoc;
191f3b9ba49SReid Kleckner     }
192829365aeSReid Kleckner     addLocIfNotPresent(CurFn->ChildSites, Loc);
193f3b9ba49SReid Kleckner   }
194f3b9ba49SReid Kleckner 
195dac21b43SReid Kleckner   OS.EmitCVLocDirective(FuncId, FileId, DL.getLine(), DL.getCol(),
196dac21b43SReid Kleckner                         /*PrologueEnd=*/false,
1972214ed89SReid Kleckner                         /*IsStmt=*/false, DL->getFilename());
19870f5bc99SReid Kleckner }
19970f5bc99SReid Kleckner 
20070f5bc99SReid Kleckner void CodeViewDebug::endModule() {
20170f5bc99SReid Kleckner   if (FnDebugInfo.empty())
20270f5bc99SReid Kleckner     return;
20370f5bc99SReid Kleckner 
204f3b9ba49SReid Kleckner   emitTypeInformation();
205f3b9ba49SReid Kleckner 
20670f5bc99SReid Kleckner   // FIXME: For functions that are comdat, we should emit separate .debug$S
20770f5bc99SReid Kleckner   // sections that are comdat associative with the main function instead of
20870f5bc99SReid Kleckner   // having one big .debug$S section.
20970f5bc99SReid Kleckner   assert(Asm != nullptr);
210dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugSymbolsSection());
211dac21b43SReid Kleckner   OS.AddComment("Debug section magic");
212dac21b43SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
21370f5bc99SReid Kleckner 
21470f5bc99SReid Kleckner   // The COFF .debug$S section consists of several subsections, each starting
21570f5bc99SReid Kleckner   // with a 4-byte control code (e.g. 0xF1, 0xF2, etc) and then a 4-byte length
21670f5bc99SReid Kleckner   // of the payload followed by the payload itself.  The subsections are 4-byte
21770f5bc99SReid Kleckner   // aligned.
21870f5bc99SReid Kleckner 
2191fcd610cSReid Kleckner   // Make a subsection for all the inlined subprograms.
220*fbd7787dSReid Kleckner   emitInlineeFuncIdsAndLines();
2211fcd610cSReid Kleckner 
2222214ed89SReid Kleckner   // Emit per-function debug information.
2232214ed89SReid Kleckner   for (auto &P : FnDebugInfo)
2242214ed89SReid Kleckner     emitDebugInfoForFunction(P.first, P.second);
22570f5bc99SReid Kleckner 
22670f5bc99SReid Kleckner   // This subsection holds a file index to offset in string table table.
227dac21b43SReid Kleckner   OS.AddComment("File index to string table offset subsection");
228dac21b43SReid Kleckner   OS.EmitCVFileChecksumsDirective();
22970f5bc99SReid Kleckner 
23070f5bc99SReid Kleckner   // This subsection holds the string table.
231dac21b43SReid Kleckner   OS.AddComment("String table");
232dac21b43SReid Kleckner   OS.EmitCVStringTableDirective();
23370f5bc99SReid Kleckner 
23470f5bc99SReid Kleckner   clear();
23570f5bc99SReid Kleckner }
23670f5bc99SReid Kleckner 
237b9456a5eSDavid Majnemer static void emitNullTerminatedSymbolName(MCStreamer &OS, StringRef S) {
238b9456a5eSDavid Majnemer   // Microsoft's linker seems to have trouble with symbol names longer than
239b9456a5eSDavid Majnemer   // 0xffd8 bytes.
240b9456a5eSDavid Majnemer   S = S.substr(0, 0xffd8);
241b9456a5eSDavid Majnemer   SmallString<32> NullTerminatedString(S);
242b9456a5eSDavid Majnemer   NullTerminatedString.push_back('\0');
243b9456a5eSDavid Majnemer   OS.EmitBytes(NullTerminatedString);
244b9456a5eSDavid Majnemer }
245b9456a5eSDavid Majnemer 
246f3b9ba49SReid Kleckner void CodeViewDebug::emitTypeInformation() {
247*fbd7787dSReid Kleckner   // Do nothing if we have no debug info or no inlined subprograms.  The types
248*fbd7787dSReid Kleckner   // we currently emit exist only to support inlined call site info.
249*fbd7787dSReid Kleckner   NamedMDNode *CU_Nodes =
250*fbd7787dSReid Kleckner       MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
251*fbd7787dSReid Kleckner   if (!CU_Nodes)
252*fbd7787dSReid Kleckner     return;
253*fbd7787dSReid Kleckner   if (InlinedSubprograms.empty())
254*fbd7787dSReid Kleckner     return;
255*fbd7787dSReid Kleckner 
256f3b9ba49SReid Kleckner   // Start the .debug$T section with 0x4.
257dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugTypesSection());
258dac21b43SReid Kleckner   OS.AddComment("Debug section magic");
259dac21b43SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
260f3b9ba49SReid Kleckner 
261f3b9ba49SReid Kleckner   // This type info currently only holds function ids for use with inline call
262f3b9ba49SReid Kleckner   // frame info. All functions are assigned a simple 'void ()' type. Emit that
263f3b9ba49SReid Kleckner   // type here.
264*fbd7787dSReid Kleckner   unsigned ArgListIndex = getNextTypeIndex();
265dac21b43SReid Kleckner   OS.AddComment("Type record length");
266dac21b43SReid Kleckner   OS.EmitIntValue(2 + sizeof(ArgList), 2);
267dac21b43SReid Kleckner   OS.AddComment("Leaf type: LF_ARGLIST");
268dac21b43SReid Kleckner   OS.EmitIntValue(LF_ARGLIST, 2);
269dac21b43SReid Kleckner   OS.AddComment("Number of arguments");
270dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
271f3b9ba49SReid Kleckner 
272*fbd7787dSReid Kleckner   unsigned VoidFnTyIdx = getNextTypeIndex();
273dac21b43SReid Kleckner   OS.AddComment("Type record length");
274dac21b43SReid Kleckner   OS.EmitIntValue(2 + sizeof(ProcedureType), 2);
275dac21b43SReid Kleckner   OS.AddComment("Leaf type: LF_PROCEDURE");
276dac21b43SReid Kleckner   OS.EmitIntValue(LF_PROCEDURE, 2);
277dac21b43SReid Kleckner   OS.AddComment("Return type index");
278dac21b43SReid Kleckner   OS.EmitIntValue(TypeIndex::Void().getIndex(), 4);
279dac21b43SReid Kleckner   OS.AddComment("Calling convention");
280dac21b43SReid Kleckner   OS.EmitIntValue(char(CallingConvention::NearC), 1);
281dac21b43SReid Kleckner   OS.AddComment("Function options");
282dac21b43SReid Kleckner   OS.EmitIntValue(char(FunctionOptions::None), 1);
283dac21b43SReid Kleckner   OS.AddComment("# of parameters");
284dac21b43SReid Kleckner   OS.EmitIntValue(0, 2);
285dac21b43SReid Kleckner   OS.AddComment("Argument list type index");
286*fbd7787dSReid Kleckner   OS.EmitIntValue(ArgListIndex, 4);
287f3b9ba49SReid Kleckner 
288*fbd7787dSReid Kleckner   // Emit LF_FUNC_ID records for all inlined subprograms to the type stream.
289*fbd7787dSReid Kleckner   // Allocate one type index for each func id.
290*fbd7787dSReid Kleckner   unsigned NextIdx = getNextTypeIndex(InlinedSubprograms.size());
291*fbd7787dSReid Kleckner   assert(NextIdx == FuncIdTypeIndexStart && "func id type indices broken");
292*fbd7787dSReid Kleckner   for (auto *SP : InlinedSubprograms) {
293f3b9ba49SReid Kleckner     StringRef DisplayName = SP->getDisplayName();
294dac21b43SReid Kleckner     OS.AddComment("Type record length");
295b9456a5eSDavid Majnemer     MCSymbol *FuncBegin = MMI->getContext().createTempSymbol(),
296b9456a5eSDavid Majnemer              *FuncEnd = MMI->getContext().createTempSymbol();
297b9456a5eSDavid Majnemer     OS.emitAbsoluteSymbolDiff(FuncEnd, FuncBegin, 2);
298b9456a5eSDavid Majnemer     OS.EmitLabel(FuncBegin);
299dac21b43SReid Kleckner     OS.AddComment("Leaf type: LF_FUNC_ID");
300dac21b43SReid Kleckner     OS.EmitIntValue(LF_FUNC_ID, 2);
301f3b9ba49SReid Kleckner 
302dac21b43SReid Kleckner     OS.AddComment("Scope type index");
303*fbd7787dSReid Kleckner     OS.EmitIntValue(0, 4);
304dac21b43SReid Kleckner     OS.AddComment("Function type");
305*fbd7787dSReid Kleckner     OS.EmitIntValue(VoidFnTyIdx, 4);
30630579ec8SDavid Majnemer     {
307dac21b43SReid Kleckner       OS.AddComment("Function name");
308b9456a5eSDavid Majnemer       emitNullTerminatedSymbolName(OS, DisplayName);
30930579ec8SDavid Majnemer     }
310b9456a5eSDavid Majnemer     OS.EmitLabel(FuncEnd);
311f3b9ba49SReid Kleckner   }
312f3b9ba49SReid Kleckner }
313f3b9ba49SReid Kleckner 
314*fbd7787dSReid Kleckner void CodeViewDebug::emitInlineeFuncIdsAndLines() {
3151fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
3161fcd610cSReid Kleckner     return;
3171fcd610cSReid Kleckner 
318f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
319f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
3201fcd610cSReid Kleckner 
3211fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
3221fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::InlineeLines), 4);
32330579ec8SDavid Majnemer   OS.AddComment("Subsection size");
3241fcd610cSReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 4);
3251fcd610cSReid Kleckner   OS.EmitLabel(InlineBegin);
3261fcd610cSReid Kleckner 
3271fcd610cSReid Kleckner   // We don't provide any extra file info.
3281fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
32930579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
3301fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
3311fcd610cSReid Kleckner 
332*fbd7787dSReid Kleckner   unsigned InlineeIndex = FuncIdTypeIndexStart;
3331fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
33430579ec8SDavid Majnemer     OS.AddBlankLine();
3351fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
3361fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
3371fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
33830579ec8SDavid Majnemer     OS.AddBlankLine();
3391fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
3401fcd610cSReid Kleckner     // 1.
3411fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
34230579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
343*fbd7787dSReid Kleckner     OS.EmitIntValue(InlineeIndex, 4);
34430579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
3451fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
34630579ec8SDavid Majnemer     OS.AddComment("Starting line number");
3471fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
348*fbd7787dSReid Kleckner 
349*fbd7787dSReid Kleckner     // The next inlined subprogram has the next function id.
350*fbd7787dSReid Kleckner     InlineeIndex++;
3511fcd610cSReid Kleckner   }
3521fcd610cSReid Kleckner 
3531fcd610cSReid Kleckner   OS.EmitLabel(InlineEnd);
3541fcd610cSReid Kleckner }
3551fcd610cSReid Kleckner 
3561fcd610cSReid Kleckner void CodeViewDebug::collectInlineSiteChildren(
3571fcd610cSReid Kleckner     SmallVectorImpl<unsigned> &Children, const FunctionInfo &FI,
3581fcd610cSReid Kleckner     const InlineSite &Site) {
3591fcd610cSReid Kleckner   for (const DILocation *ChildSiteLoc : Site.ChildSites) {
3601fcd610cSReid Kleckner     auto I = FI.InlineSites.find(ChildSiteLoc);
3611fcd610cSReid Kleckner     const InlineSite &ChildSite = I->second;
3621fcd610cSReid Kleckner     Children.push_back(ChildSite.SiteFuncId);
3631fcd610cSReid Kleckner     collectInlineSiteChildren(Children, FI, ChildSite);
3641fcd610cSReid Kleckner   }
3651fcd610cSReid Kleckner }
3661fcd610cSReid Kleckner 
367f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
368f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
369f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
370f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
371f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
372f3b9ba49SReid Kleckner 
373*fbd7787dSReid Kleckner   assert(SubprogramIndices.count(Site.Inlinee));
374*fbd7787dSReid Kleckner   unsigned InlineeIdx = FuncIdTypeIndexStart + SubprogramIndices[Site.Inlinee];
375f3b9ba49SReid Kleckner 
376f3b9ba49SReid Kleckner   // SymbolRecord
377dac21b43SReid Kleckner   OS.AddComment("Record length");
378eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
379f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
380dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
381dac21b43SReid Kleckner   OS.EmitIntValue(SymbolRecordKind::S_INLINESITE, 2); // RecordKind
382f3b9ba49SReid Kleckner 
383dac21b43SReid Kleckner   OS.AddComment("PtrParent");
384dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
385dac21b43SReid Kleckner   OS.AddComment("PtrEnd");
386dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
387dac21b43SReid Kleckner   OS.AddComment("Inlinee type index");
388*fbd7787dSReid Kleckner   OS.EmitIntValue(InlineeIdx, 4);
389f3b9ba49SReid Kleckner 
3901fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
3911fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
3921fcd610cSReid Kleckner   SmallVector<unsigned, 3> SecondaryFuncIds;
3931fcd610cSReid Kleckner   collectInlineSiteChildren(SecondaryFuncIds, FI, Site);
3941fcd610cSReid Kleckner 
3951fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
396c9911f28SDavid Majnemer                                     FI.Begin, FI.End, SecondaryFuncIds);
397f3b9ba49SReid Kleckner 
398f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
399f3b9ba49SReid Kleckner 
400f9c275feSReid Kleckner   for (const LocalVariable &Var : Site.InlinedLocals)
401f9c275feSReid Kleckner     emitLocalVariable(Var);
402f9c275feSReid Kleckner 
403f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
404f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
405f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
406f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
407f3b9ba49SReid Kleckner            "child site not in function inline site map");
408f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
409f3b9ba49SReid Kleckner   }
410f3b9ba49SReid Kleckner 
411f3b9ba49SReid Kleckner   // Close the scope.
412dac21b43SReid Kleckner   OS.AddComment("Record length");
413dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
414dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
415dac21b43SReid Kleckner   OS.EmitIntValue(SymbolRecordKind::S_INLINESITE_END, 2); // RecordKind
416f3b9ba49SReid Kleckner }
417f3b9ba49SReid Kleckner 
4182214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
4192214ed89SReid Kleckner                                              FunctionInfo &FI) {
42070f5bc99SReid Kleckner   // For each function there is a separate subsection
42170f5bc99SReid Kleckner   // which holds the PC to file:line table.
42270f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
42370f5bc99SReid Kleckner   assert(Fn);
42470f5bc99SReid Kleckner 
42570f5bc99SReid Kleckner   StringRef FuncName;
426adebb937SPete Cooper   if (auto *SP = GV->getSubprogram())
42770f5bc99SReid Kleckner     FuncName = SP->getDisplayName();
42870f5bc99SReid Kleckner 
42970f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
43070f5bc99SReid Kleckner   if (FuncName.empty())
43170f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
43270f5bc99SReid Kleckner 
43370f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
434f9c275feSReid Kleckner   MCSymbol *SymbolsBegin = MMI->getContext().createTempSymbol(),
435f9c275feSReid Kleckner            *SymbolsEnd = MMI->getContext().createTempSymbol();
436dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
437dac21b43SReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::Symbols), 4);
438dac21b43SReid Kleckner   OS.AddComment("Subsection size");
439eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(SymbolsEnd, SymbolsBegin, 4);
440dac21b43SReid Kleckner   OS.EmitLabel(SymbolsBegin);
44170f5bc99SReid Kleckner   {
442f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
443f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
444dac21b43SReid Kleckner     OS.AddComment("Record length");
445eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
446dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
44770f5bc99SReid Kleckner 
448dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
449dac21b43SReid Kleckner     OS.EmitIntValue(unsigned(SymbolRecordKind::S_GPROC32_ID), 2);
45070f5bc99SReid Kleckner 
45130579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
452dac21b43SReid Kleckner     OS.AddComment("PtrParent");
453dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
454dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
455dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
456dac21b43SReid Kleckner     OS.AddComment("PtrNext");
457dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
45870f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
45970f5bc99SReid Kleckner     // code is located and what's its size:
460dac21b43SReid Kleckner     OS.AddComment("Code size");
461eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
462dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
463dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
464dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
465dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
466dac21b43SReid Kleckner     OS.AddComment("Function type index");
467dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
468dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
469dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
470dac21b43SReid Kleckner     OS.AddComment("Function section index");
471dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
472dac21b43SReid Kleckner     OS.AddComment("Flags");
473dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
47470f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
475dac21b43SReid Kleckner     OS.AddComment("Function name");
4761256125fSDavid Majnemer     // Truncate the name so we won't overflow the record length field.
477b9456a5eSDavid Majnemer     emitNullTerminatedSymbolName(OS, FuncName);
478dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
47970f5bc99SReid Kleckner 
480f9c275feSReid Kleckner     for (const LocalVariable &Var : FI.Locals)
481f9c275feSReid Kleckner       emitLocalVariable(Var);
482f9c275feSReid Kleckner 
483f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
484f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
485f3b9ba49SReid Kleckner     // of their parent inline site.
486f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
487f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
488f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
489f9c275feSReid Kleckner              "child site not in function inline site map");
490f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
491f3b9ba49SReid Kleckner     }
492f3b9ba49SReid Kleckner 
49370f5bc99SReid Kleckner     // We're done with this function.
494dac21b43SReid Kleckner     OS.AddComment("Record length");
495dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
496dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
497dac21b43SReid Kleckner     OS.EmitIntValue(unsigned(SymbolRecordKind::S_PROC_ID_END), 2);
49870f5bc99SReid Kleckner   }
499dac21b43SReid Kleckner   OS.EmitLabel(SymbolsEnd);
50070f5bc99SReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
501dac21b43SReid Kleckner   OS.EmitValueToAlignment(4);
50270f5bc99SReid Kleckner 
5032214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
504dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
50570f5bc99SReid Kleckner }
50670f5bc99SReid Kleckner 
507876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
508876330d5SReid Kleckner CodeViewDebug::createDefRangeMem(uint16_t CVRegister, int Offset) {
509876330d5SReid Kleckner   LocalVarDefRange DR;
510c6a2f214SAaron Ballman   DR.InMemory = -1;
511876330d5SReid Kleckner   DR.DataOffset = Offset;
512876330d5SReid Kleckner   assert(DR.DataOffset == Offset && "truncation");
513876330d5SReid Kleckner   DR.StructOffset = 0;
514876330d5SReid Kleckner   DR.CVRegister = CVRegister;
515876330d5SReid Kleckner   return DR;
516876330d5SReid Kleckner }
517876330d5SReid Kleckner 
518876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
519876330d5SReid Kleckner CodeViewDebug::createDefRangeReg(uint16_t CVRegister) {
520876330d5SReid Kleckner   LocalVarDefRange DR;
521876330d5SReid Kleckner   DR.InMemory = 0;
522876330d5SReid Kleckner   DR.DataOffset = 0;
523876330d5SReid Kleckner   DR.StructOffset = 0;
524876330d5SReid Kleckner   DR.CVRegister = CVRegister;
525876330d5SReid Kleckner   return DR;
526876330d5SReid Kleckner }
527876330d5SReid Kleckner 
528876330d5SReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable(
529876330d5SReid Kleckner     DenseSet<InlinedVariable> &Processed) {
530876330d5SReid Kleckner   const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
531876330d5SReid Kleckner   const TargetFrameLowering *TFI = TSI.getFrameLowering();
532876330d5SReid Kleckner   const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
533876330d5SReid Kleckner 
534876330d5SReid Kleckner   for (const MachineModuleInfo::VariableDbgInfo &VI :
535876330d5SReid Kleckner        MMI->getVariableDbgInfo()) {
536f9c275feSReid Kleckner     if (!VI.Var)
537f9c275feSReid Kleckner       continue;
538f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
539f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
540f9c275feSReid Kleckner 
541876330d5SReid Kleckner     Processed.insert(InlinedVariable(VI.Var, VI.Loc->getInlinedAt()));
542f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
543f9c275feSReid Kleckner 
544f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
545f9c275feSReid Kleckner     if (!Scope)
546f9c275feSReid Kleckner       continue;
547f9c275feSReid Kleckner 
548f9c275feSReid Kleckner     // Get the frame register used and the offset.
549f9c275feSReid Kleckner     unsigned FrameReg = 0;
550876330d5SReid Kleckner     int FrameOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
551876330d5SReid Kleckner     uint16_t CVReg = TRI->getCodeViewRegNum(FrameReg);
552f9c275feSReid Kleckner 
553f9c275feSReid Kleckner     // Calculate the label ranges.
554876330d5SReid Kleckner     LocalVarDefRange DefRange = createDefRangeMem(CVReg, FrameOffset);
555f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
556f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
557f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
558876330d5SReid Kleckner       End = End ? End : Asm->getFunctionEnd();
559876330d5SReid Kleckner       DefRange.Ranges.emplace_back(Begin, End);
560f9c275feSReid Kleckner     }
561f9c275feSReid Kleckner 
562876330d5SReid Kleckner     LocalVariable Var;
563876330d5SReid Kleckner     Var.DIVar = VI.Var;
564876330d5SReid Kleckner     Var.DefRanges.emplace_back(std::move(DefRange));
565876330d5SReid Kleckner     recordLocalVariable(std::move(Var), VI.Loc->getInlinedAt());
566f9c275feSReid Kleckner   }
567f9c275feSReid Kleckner }
568876330d5SReid Kleckner 
569876330d5SReid Kleckner void CodeViewDebug::collectVariableInfo(const DISubprogram *SP) {
570876330d5SReid Kleckner   DenseSet<InlinedVariable> Processed;
571876330d5SReid Kleckner   // Grab the variable info that was squirreled away in the MMI side-table.
572876330d5SReid Kleckner   collectVariableInfoFromMMITable(Processed);
573876330d5SReid Kleckner 
574876330d5SReid Kleckner   const TargetRegisterInfo *TRI = Asm->MF->getSubtarget().getRegisterInfo();
575876330d5SReid Kleckner 
576876330d5SReid Kleckner   for (const auto &I : DbgValues) {
577876330d5SReid Kleckner     InlinedVariable IV = I.first;
578876330d5SReid Kleckner     if (Processed.count(IV))
579876330d5SReid Kleckner       continue;
580876330d5SReid Kleckner     const DILocalVariable *DIVar = IV.first;
581876330d5SReid Kleckner     const DILocation *InlinedAt = IV.second;
582876330d5SReid Kleckner 
583876330d5SReid Kleckner     // Instruction ranges, specifying where IV is accessible.
584876330d5SReid Kleckner     const auto &Ranges = I.second;
585876330d5SReid Kleckner 
586876330d5SReid Kleckner     LexicalScope *Scope = nullptr;
587876330d5SReid Kleckner     if (InlinedAt)
588876330d5SReid Kleckner       Scope = LScopes.findInlinedScope(DIVar->getScope(), InlinedAt);
589876330d5SReid Kleckner     else
590876330d5SReid Kleckner       Scope = LScopes.findLexicalScope(DIVar->getScope());
591876330d5SReid Kleckner     // If variable scope is not found then skip this variable.
592876330d5SReid Kleckner     if (!Scope)
593876330d5SReid Kleckner       continue;
594876330d5SReid Kleckner 
595876330d5SReid Kleckner     LocalVariable Var;
596876330d5SReid Kleckner     Var.DIVar = DIVar;
597876330d5SReid Kleckner 
598876330d5SReid Kleckner     // Calculate the definition ranges.
599876330d5SReid Kleckner     for (auto I = Ranges.begin(), E = Ranges.end(); I != E; ++I) {
600876330d5SReid Kleckner       const InsnRange &Range = *I;
601876330d5SReid Kleckner       const MachineInstr *DVInst = Range.first;
602876330d5SReid Kleckner       assert(DVInst->isDebugValue() && "Invalid History entry");
603876330d5SReid Kleckner       const DIExpression *DIExpr = DVInst->getDebugExpression();
604876330d5SReid Kleckner 
605876330d5SReid Kleckner       // Bail if there is a complex DWARF expression for now.
606876330d5SReid Kleckner       if (DIExpr && DIExpr->getNumElements() > 0)
607876330d5SReid Kleckner         continue;
608876330d5SReid Kleckner 
6099a593ee7SReid Kleckner       // Bail if operand 0 is not a valid register. This means the variable is a
6109a593ee7SReid Kleckner       // simple constant, or is described by a complex expression.
6119a593ee7SReid Kleckner       // FIXME: Find a way to represent constant variables, since they are
6129a593ee7SReid Kleckner       // relatively common.
6139a593ee7SReid Kleckner       unsigned Reg =
6149a593ee7SReid Kleckner           DVInst->getOperand(0).isReg() ? DVInst->getOperand(0).getReg() : 0;
6159a593ee7SReid Kleckner       if (Reg == 0)
6166e0d5f57SReid Kleckner         continue;
6176e0d5f57SReid Kleckner 
618876330d5SReid Kleckner       // Handle the two cases we can handle: indirect in memory and in register.
619876330d5SReid Kleckner       bool IsIndirect = DVInst->getOperand(1).isImm();
620876330d5SReid Kleckner       unsigned CVReg = TRI->getCodeViewRegNum(DVInst->getOperand(0).getReg());
621876330d5SReid Kleckner       {
622876330d5SReid Kleckner         LocalVarDefRange DefRange;
623876330d5SReid Kleckner         if (IsIndirect) {
624876330d5SReid Kleckner           int64_t Offset = DVInst->getOperand(1).getImm();
625876330d5SReid Kleckner           DefRange = createDefRangeMem(CVReg, Offset);
626876330d5SReid Kleckner         } else {
627876330d5SReid Kleckner           DefRange = createDefRangeReg(CVReg);
628876330d5SReid Kleckner         }
629876330d5SReid Kleckner         if (Var.DefRanges.empty() ||
630876330d5SReid Kleckner             Var.DefRanges.back().isDifferentLocation(DefRange)) {
631876330d5SReid Kleckner           Var.DefRanges.emplace_back(std::move(DefRange));
632876330d5SReid Kleckner         }
633876330d5SReid Kleckner       }
634876330d5SReid Kleckner 
635876330d5SReid Kleckner       // Compute the label range.
636876330d5SReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
637876330d5SReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
638876330d5SReid Kleckner       if (!End) {
639876330d5SReid Kleckner         if (std::next(I) != E)
640876330d5SReid Kleckner           End = getLabelBeforeInsn(std::next(I)->first);
641876330d5SReid Kleckner         else
642876330d5SReid Kleckner           End = Asm->getFunctionEnd();
643876330d5SReid Kleckner       }
644876330d5SReid Kleckner 
645876330d5SReid Kleckner       // If the last range end is our begin, just extend the last range.
646876330d5SReid Kleckner       // Otherwise make a new range.
647876330d5SReid Kleckner       SmallVectorImpl<std::pair<const MCSymbol *, const MCSymbol *>> &Ranges =
648876330d5SReid Kleckner           Var.DefRanges.back().Ranges;
649876330d5SReid Kleckner       if (!Ranges.empty() && Ranges.back().second == Begin)
650876330d5SReid Kleckner         Ranges.back().second = End;
651876330d5SReid Kleckner       else
652876330d5SReid Kleckner         Ranges.emplace_back(Begin, End);
653876330d5SReid Kleckner 
654876330d5SReid Kleckner       // FIXME: Do more range combining.
655876330d5SReid Kleckner     }
656876330d5SReid Kleckner 
657876330d5SReid Kleckner     recordLocalVariable(std::move(Var), InlinedAt);
658876330d5SReid Kleckner   }
659f9c275feSReid Kleckner }
660f9c275feSReid Kleckner 
66170f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
66270f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
66370f5bc99SReid Kleckner 
664f9c275feSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
66570f5bc99SReid Kleckner     return;
66670f5bc99SReid Kleckner 
667f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
668f9c275feSReid Kleckner 
66970f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
67070f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
67170f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
6722214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
6731fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
67470f5bc99SReid Kleckner 
675f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
676f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
67770f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
67870f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
67970f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
68070f5bc99SReid Kleckner   bool EmptyPrologue = true;
68170f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
68270f5bc99SReid Kleckner     for (const auto &MI : MBB) {
683f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
684f9c275feSReid Kleckner           MI.getDebugLoc()) {
68570f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
68670f5bc99SReid Kleckner         break;
687f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
68870f5bc99SReid Kleckner         EmptyPrologue = false;
68970f5bc99SReid Kleckner       }
69070f5bc99SReid Kleckner     }
691f9c275feSReid Kleckner   }
692f9c275feSReid Kleckner 
69370f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
69470f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
69570f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
69670f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
69770f5bc99SReid Kleckner   }
69870f5bc99SReid Kleckner }
69970f5bc99SReid Kleckner 
700f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
701f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
702f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
703f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
704f9c275feSReid Kleckner   OS.AddComment("Record length");
705f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
706f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
707f9c275feSReid Kleckner 
708f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
709f9c275feSReid Kleckner   OS.EmitIntValue(unsigned(SymbolRecordKind::S_LOCAL), 2);
710f9c275feSReid Kleckner 
711f9c275feSReid Kleckner   uint16_t Flags = 0;
712f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
713f9c275feSReid Kleckner     Flags |= LocalSym::IsParameter;
714876330d5SReid Kleckner   if (Var.DefRanges.empty())
715876330d5SReid Kleckner     Flags |= LocalSym::IsOptimizedOut;
716f9c275feSReid Kleckner 
717f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
718f9c275feSReid Kleckner   OS.EmitIntValue(TypeIndex::Int32().getIndex(), 4);
719f9c275feSReid Kleckner   OS.AddComment("Flags");
720f9c275feSReid Kleckner   OS.EmitIntValue(Flags, 2);
7211256125fSDavid Majnemer   // Truncate the name so we won't overflow the record length field.
722b9456a5eSDavid Majnemer   emitNullTerminatedSymbolName(OS, Var.DIVar->getName());
723f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
724f9c275feSReid Kleckner 
725876330d5SReid Kleckner   // Calculate the on disk prefix of the appropriate def range record. The
726876330d5SReid Kleckner   // records and on disk formats are described in SymbolRecords.h. BytePrefix
727876330d5SReid Kleckner   // should be big enough to hold all forms without memory allocation.
728876330d5SReid Kleckner   SmallString<20> BytePrefix;
729876330d5SReid Kleckner   for (const LocalVarDefRange &DefRange : Var.DefRanges) {
730876330d5SReid Kleckner     BytePrefix.clear();
731876330d5SReid Kleckner     // FIXME: Handle bitpieces.
732876330d5SReid Kleckner     if (DefRange.StructOffset != 0)
733876330d5SReid Kleckner       continue;
734876330d5SReid Kleckner 
735876330d5SReid Kleckner     if (DefRange.InMemory) {
736f9c275feSReid Kleckner       DefRangeRegisterRelSym Sym{};
737f9c275feSReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
738876330d5SReid Kleckner       Sym.BaseRegister = DefRange.CVRegister;
739f9c275feSReid Kleckner       Sym.Flags = 0; // Unclear what matters here.
740876330d5SReid Kleckner       Sym.BasePointerOffset = DefRange.DataOffset;
741876330d5SReid Kleckner       BytePrefix +=
742876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
743f9c275feSReid Kleckner       BytePrefix += StringRef(reinterpret_cast<const char *>(&Sym),
744f9c275feSReid Kleckner                               sizeof(Sym) - sizeof(LocalVariableAddrRange));
745876330d5SReid Kleckner     } else {
746876330d5SReid Kleckner       assert(DefRange.DataOffset == 0 && "unexpected offset into register");
747876330d5SReid Kleckner       DefRangeRegisterSym Sym{};
748876330d5SReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER);
749876330d5SReid Kleckner       Sym.Register = DefRange.CVRegister;
750876330d5SReid Kleckner       Sym.MayHaveNoName = 0; // Unclear what matters here.
751876330d5SReid Kleckner       BytePrefix +=
752876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
753876330d5SReid Kleckner       BytePrefix += StringRef(reinterpret_cast<const char *>(&Sym),
754876330d5SReid Kleckner                               sizeof(Sym) - sizeof(LocalVariableAddrRange));
755876330d5SReid Kleckner     }
756876330d5SReid Kleckner     OS.EmitCVDefRangeDirective(DefRange.Ranges, BytePrefix);
757876330d5SReid Kleckner   }
758f9c275feSReid Kleckner }
759f9c275feSReid Kleckner 
76070f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
76170f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
76270f5bc99SReid Kleckner     return;
76370f5bc99SReid Kleckner 
76470f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
76570f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
76670f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
76770f5bc99SReid Kleckner 
768adebb937SPete Cooper   collectVariableInfo(GV->getSubprogram());
769876330d5SReid Kleckner 
770876330d5SReid Kleckner   DebugHandlerBase::endFunction(MF);
771876330d5SReid Kleckner 
7722214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
7732214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
77470f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
775f9c275feSReid Kleckner     CurFn = nullptr;
776f9c275feSReid Kleckner     return;
77770f5bc99SReid Kleckner   }
778f9c275feSReid Kleckner 
779f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
780f9c275feSReid Kleckner 
78170f5bc99SReid Kleckner   CurFn = nullptr;
78270f5bc99SReid Kleckner }
78370f5bc99SReid Kleckner 
78470f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
785f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
786f9c275feSReid Kleckner 
78770f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
78870f5bc99SReid Kleckner   if (!Asm || MI->isDebugValue() || MI->getFlag(MachineInstr::FrameSetup))
78970f5bc99SReid Kleckner     return;
79070f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
79170f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
79270f5bc99SReid Kleckner     return;
79370f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
79470f5bc99SReid Kleckner }
795