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) {
111fbd7787dSReid Kleckner   auto SiteInsertion = CurFn->InlineSites.insert({InlinedAt, InlineSite()});
112fbd7787dSReid Kleckner   InlineSite *Site = &SiteInsertion.first->second;
113fbd7787dSReid Kleckner   if (SiteInsertion.second) {
114f9c275feSReid Kleckner     Site->SiteFuncId = NextFuncId++;
115876330d5SReid Kleckner     Site->Inlinee = Inlinee;
116fbd7787dSReid Kleckner     auto InlineeInsertion =
117fbd7787dSReid Kleckner         SubprogramIndices.insert({Inlinee, InlinedSubprograms.size()});
118fbd7787dSReid Kleckner     if (InlineeInsertion.second)
119fbd7787dSReid 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.
220fbd7787dSReid 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() {
247fbd7787dSReid Kleckner   // Do nothing if we have no debug info or no inlined subprograms.  The types
248fbd7787dSReid Kleckner   // we currently emit exist only to support inlined call site info.
249fbd7787dSReid Kleckner   NamedMDNode *CU_Nodes =
250fbd7787dSReid Kleckner       MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
251fbd7787dSReid Kleckner   if (!CU_Nodes)
252fbd7787dSReid Kleckner     return;
253fbd7787dSReid Kleckner   if (InlinedSubprograms.empty())
254fbd7787dSReid Kleckner     return;
255fbd7787dSReid 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.
2646cd7c918SDavid Majnemer   unsigned ArgListIndex = getNextTypeIndex();
2656cd7c918SDavid Majnemer   OS.AddComment("Type record length");
2666cd7c918SDavid Majnemer   OS.EmitIntValue(ArgListRecord::getLayoutSize(), 2);
2676cd7c918SDavid Majnemer   OS.AddComment("Leaf type: LF_ARGLIST");
2686cd7c918SDavid Majnemer   OS.EmitIntValue(LF_ARGLIST, 2);
2696cd7c918SDavid Majnemer   OS.AddComment("Number of arguments");
2706cd7c918SDavid Majnemer   OS.EmitIntValue(0, 4);
271f3b9ba49SReid Kleckner 
2726cd7c918SDavid Majnemer   unsigned VoidFnTyIdx = getNextTypeIndex();
2736cd7c918SDavid Majnemer   OS.AddComment("Type record length");
2746cd7c918SDavid Majnemer   OS.EmitIntValue(ProcedureRecord::getLayoutSize(), 2);
2756cd7c918SDavid Majnemer   OS.AddComment("Leaf type: LF_PROCEDURE");
2766cd7c918SDavid Majnemer   OS.EmitIntValue(LF_PROCEDURE, 2);
2776cd7c918SDavid Majnemer   OS.AddComment("Return type index");
2786cd7c918SDavid Majnemer   OS.EmitIntValue(TypeIndex::Void().getIndex(), 4);
2796cd7c918SDavid Majnemer   OS.AddComment("Calling convention");
2806cd7c918SDavid Majnemer   OS.EmitIntValue(char(CallingConvention::NearC), 1);
2816cd7c918SDavid Majnemer   OS.AddComment("Function options");
2826cd7c918SDavid Majnemer   OS.EmitIntValue(char(FunctionOptions::None), 1);
2836cd7c918SDavid Majnemer   OS.AddComment("# of parameters");
2846cd7c918SDavid Majnemer   OS.EmitIntValue(0, 2);
2856cd7c918SDavid Majnemer   OS.AddComment("Argument list type index");
2866cd7c918SDavid Majnemer   OS.EmitIntValue(ArgListIndex, 4);
287f3b9ba49SReid Kleckner 
288fbd7787dSReid Kleckner   // Emit LF_FUNC_ID records for all inlined subprograms to the type stream.
289fbd7787dSReid Kleckner   // Allocate one type index for each func id.
2906cd7c918SDavid Majnemer   unsigned NextIdx = getNextTypeIndex(InlinedSubprograms.size());
2916cd7c918SDavid Majnemer   (void)NextIdx;
2926cd7c918SDavid Majnemer   assert(NextIdx == FuncIdTypeIndexStart && "func id type indices broken");
293fbd7787dSReid Kleckner   for (auto *SP : InlinedSubprograms) {
294f3b9ba49SReid Kleckner     StringRef DisplayName = SP->getDisplayName();
295a972d612SAdrian McCarthy     OS.AddComment("Type record length");
2966cd7c918SDavid Majnemer     MCSymbol *FuncBegin = MMI->getContext().createTempSymbol(),
2976cd7c918SDavid Majnemer              *FuncEnd = MMI->getContext().createTempSymbol();
2986cd7c918SDavid Majnemer     OS.emitAbsoluteSymbolDiff(FuncEnd, FuncBegin, 2);
2996cd7c918SDavid Majnemer     OS.EmitLabel(FuncBegin);
3006cd7c918SDavid Majnemer     OS.AddComment("Leaf type: LF_FUNC_ID");
3016cd7c918SDavid Majnemer     OS.EmitIntValue(LF_FUNC_ID, 2);
3026cd7c918SDavid Majnemer 
3036cd7c918SDavid Majnemer     OS.AddComment("Scope type index");
3046cd7c918SDavid Majnemer     OS.EmitIntValue(0, 4);
3056cd7c918SDavid Majnemer     OS.AddComment("Function type");
3066cd7c918SDavid Majnemer     OS.EmitIntValue(VoidFnTyIdx, 4);
3076cd7c918SDavid Majnemer     {
3086cd7c918SDavid Majnemer       OS.AddComment("Function name");
3096cd7c918SDavid Majnemer       emitNullTerminatedSymbolName(OS, DisplayName);
3106cd7c918SDavid Majnemer     }
3116cd7c918SDavid Majnemer     OS.EmitLabel(FuncEnd);
3126cd7c918SDavid Majnemer   }
313f3b9ba49SReid Kleckner }
314f3b9ba49SReid Kleckner 
315fbd7787dSReid Kleckner void CodeViewDebug::emitInlineeFuncIdsAndLines() {
3161fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
3171fcd610cSReid Kleckner     return;
3181fcd610cSReid Kleckner 
319f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
320f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
3211fcd610cSReid Kleckner 
3221fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
3231fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::InlineeLines), 4);
32430579ec8SDavid Majnemer   OS.AddComment("Subsection size");
3251fcd610cSReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 4);
3261fcd610cSReid Kleckner   OS.EmitLabel(InlineBegin);
3271fcd610cSReid Kleckner 
3281fcd610cSReid Kleckner   // We don't provide any extra file info.
3291fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
33030579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
3311fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
3321fcd610cSReid Kleckner 
333fbd7787dSReid Kleckner   unsigned InlineeIndex = FuncIdTypeIndexStart;
3341fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
33530579ec8SDavid Majnemer     OS.AddBlankLine();
3361fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
3371fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
3381fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
33930579ec8SDavid Majnemer     OS.AddBlankLine();
3401fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
3411fcd610cSReid Kleckner     // 1.
3421fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
34330579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
344fbd7787dSReid Kleckner     OS.EmitIntValue(InlineeIndex, 4);
34530579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
3461fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
34730579ec8SDavid Majnemer     OS.AddComment("Starting line number");
3481fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
349fbd7787dSReid Kleckner 
350fbd7787dSReid Kleckner     // The next inlined subprogram has the next function id.
351fbd7787dSReid Kleckner     InlineeIndex++;
3521fcd610cSReid Kleckner   }
3531fcd610cSReid Kleckner 
3541fcd610cSReid Kleckner   OS.EmitLabel(InlineEnd);
3551fcd610cSReid Kleckner }
3561fcd610cSReid Kleckner 
3571fcd610cSReid Kleckner void CodeViewDebug::collectInlineSiteChildren(
3581fcd610cSReid Kleckner     SmallVectorImpl<unsigned> &Children, const FunctionInfo &FI,
3591fcd610cSReid Kleckner     const InlineSite &Site) {
3601fcd610cSReid Kleckner   for (const DILocation *ChildSiteLoc : Site.ChildSites) {
3611fcd610cSReid Kleckner     auto I = FI.InlineSites.find(ChildSiteLoc);
3621fcd610cSReid Kleckner     const InlineSite &ChildSite = I->second;
3631fcd610cSReid Kleckner     Children.push_back(ChildSite.SiteFuncId);
3641fcd610cSReid Kleckner     collectInlineSiteChildren(Children, FI, ChildSite);
3651fcd610cSReid Kleckner   }
3661fcd610cSReid Kleckner }
3671fcd610cSReid Kleckner 
368f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
369f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
370f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
371f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
372f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
373f3b9ba49SReid Kleckner 
374fbd7787dSReid Kleckner   assert(SubprogramIndices.count(Site.Inlinee));
375fbd7787dSReid Kleckner   unsigned InlineeIdx = FuncIdTypeIndexStart + SubprogramIndices[Site.Inlinee];
376f3b9ba49SReid Kleckner 
377f3b9ba49SReid Kleckner   // SymbolRecord
378dac21b43SReid Kleckner   OS.AddComment("Record length");
379eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
380f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
381dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
38263a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE, 2); // RecordKind
383f3b9ba49SReid Kleckner 
384dac21b43SReid Kleckner   OS.AddComment("PtrParent");
385dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
386dac21b43SReid Kleckner   OS.AddComment("PtrEnd");
387dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
388dac21b43SReid Kleckner   OS.AddComment("Inlinee type index");
389fbd7787dSReid Kleckner   OS.EmitIntValue(InlineeIdx, 4);
390f3b9ba49SReid Kleckner 
3911fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
3921fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
3931fcd610cSReid Kleckner   SmallVector<unsigned, 3> SecondaryFuncIds;
3941fcd610cSReid Kleckner   collectInlineSiteChildren(SecondaryFuncIds, FI, Site);
3951fcd610cSReid Kleckner 
3961fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
397c9911f28SDavid Majnemer                                     FI.Begin, FI.End, SecondaryFuncIds);
398f3b9ba49SReid Kleckner 
399f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
400f3b9ba49SReid Kleckner 
401f9c275feSReid Kleckner   for (const LocalVariable &Var : Site.InlinedLocals)
402f9c275feSReid Kleckner     emitLocalVariable(Var);
403f9c275feSReid Kleckner 
404f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
405f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
406f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
407f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
408f3b9ba49SReid Kleckner            "child site not in function inline site map");
409f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
410f3b9ba49SReid Kleckner   }
411f3b9ba49SReid Kleckner 
412f3b9ba49SReid Kleckner   // Close the scope.
413dac21b43SReid Kleckner   OS.AddComment("Record length");
414dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
415dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
41663a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE_END, 2); // RecordKind
417f3b9ba49SReid Kleckner }
418f3b9ba49SReid Kleckner 
4192214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
4202214ed89SReid Kleckner                                              FunctionInfo &FI) {
42170f5bc99SReid Kleckner   // For each function there is a separate subsection
42270f5bc99SReid Kleckner   // which holds the PC to file:line table.
42370f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
42470f5bc99SReid Kleckner   assert(Fn);
42570f5bc99SReid Kleckner 
42670f5bc99SReid Kleckner   StringRef FuncName;
427adebb937SPete Cooper   if (auto *SP = GV->getSubprogram())
42870f5bc99SReid Kleckner     FuncName = SP->getDisplayName();
42970f5bc99SReid Kleckner 
43070f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
43170f5bc99SReid Kleckner   if (FuncName.empty())
43270f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
43370f5bc99SReid Kleckner 
43470f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
435f9c275feSReid Kleckner   MCSymbol *SymbolsBegin = MMI->getContext().createTempSymbol(),
436f9c275feSReid Kleckner            *SymbolsEnd = MMI->getContext().createTempSymbol();
437dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
438dac21b43SReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::Symbols), 4);
439dac21b43SReid Kleckner   OS.AddComment("Subsection size");
440eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(SymbolsEnd, SymbolsBegin, 4);
441dac21b43SReid Kleckner   OS.EmitLabel(SymbolsBegin);
44270f5bc99SReid Kleckner   {
443f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
444f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
445dac21b43SReid Kleckner     OS.AddComment("Record length");
446eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
447dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
44870f5bc99SReid Kleckner 
449dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
45063a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_GPROC32_ID), 2);
45170f5bc99SReid Kleckner 
45230579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
453dac21b43SReid Kleckner     OS.AddComment("PtrParent");
454dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
455dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
456dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
457dac21b43SReid Kleckner     OS.AddComment("PtrNext");
458dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
45970f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
46070f5bc99SReid Kleckner     // code is located and what's its size:
461dac21b43SReid Kleckner     OS.AddComment("Code size");
462eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
463dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
464dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
465dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
466dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
467dac21b43SReid Kleckner     OS.AddComment("Function type index");
468dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
469dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
470dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
471dac21b43SReid Kleckner     OS.AddComment("Function section index");
472dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
473dac21b43SReid Kleckner     OS.AddComment("Flags");
474dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
47570f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
476dac21b43SReid Kleckner     OS.AddComment("Function name");
4771256125fSDavid Majnemer     // Truncate the name so we won't overflow the record length field.
478b9456a5eSDavid Majnemer     emitNullTerminatedSymbolName(OS, FuncName);
479dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
48070f5bc99SReid Kleckner 
481f9c275feSReid Kleckner     for (const LocalVariable &Var : FI.Locals)
482f9c275feSReid Kleckner       emitLocalVariable(Var);
483f9c275feSReid Kleckner 
484f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
485f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
486f3b9ba49SReid Kleckner     // of their parent inline site.
487f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
488f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
489f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
490f9c275feSReid Kleckner              "child site not in function inline site map");
491f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
492f3b9ba49SReid Kleckner     }
493f3b9ba49SReid Kleckner 
49470f5bc99SReid Kleckner     // We're done with this function.
495dac21b43SReid Kleckner     OS.AddComment("Record length");
496dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
497dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
49863a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_PROC_ID_END), 2);
49970f5bc99SReid Kleckner   }
500dac21b43SReid Kleckner   OS.EmitLabel(SymbolsEnd);
50170f5bc99SReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
502dac21b43SReid Kleckner   OS.EmitValueToAlignment(4);
50370f5bc99SReid Kleckner 
5042214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
505dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
50670f5bc99SReid Kleckner }
50770f5bc99SReid Kleckner 
508876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
509876330d5SReid Kleckner CodeViewDebug::createDefRangeMem(uint16_t CVRegister, int Offset) {
510876330d5SReid Kleckner   LocalVarDefRange DR;
511c6a2f214SAaron Ballman   DR.InMemory = -1;
512876330d5SReid Kleckner   DR.DataOffset = Offset;
513876330d5SReid Kleckner   assert(DR.DataOffset == Offset && "truncation");
514876330d5SReid Kleckner   DR.StructOffset = 0;
515876330d5SReid Kleckner   DR.CVRegister = CVRegister;
516876330d5SReid Kleckner   return DR;
517876330d5SReid Kleckner }
518876330d5SReid Kleckner 
519876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
520876330d5SReid Kleckner CodeViewDebug::createDefRangeReg(uint16_t CVRegister) {
521876330d5SReid Kleckner   LocalVarDefRange DR;
522876330d5SReid Kleckner   DR.InMemory = 0;
523876330d5SReid Kleckner   DR.DataOffset = 0;
524876330d5SReid Kleckner   DR.StructOffset = 0;
525876330d5SReid Kleckner   DR.CVRegister = CVRegister;
526876330d5SReid Kleckner   return DR;
527876330d5SReid Kleckner }
528876330d5SReid Kleckner 
529876330d5SReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable(
530876330d5SReid Kleckner     DenseSet<InlinedVariable> &Processed) {
531876330d5SReid Kleckner   const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
532876330d5SReid Kleckner   const TargetFrameLowering *TFI = TSI.getFrameLowering();
533876330d5SReid Kleckner   const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
534876330d5SReid Kleckner 
535876330d5SReid Kleckner   for (const MachineModuleInfo::VariableDbgInfo &VI :
536876330d5SReid Kleckner        MMI->getVariableDbgInfo()) {
537f9c275feSReid Kleckner     if (!VI.Var)
538f9c275feSReid Kleckner       continue;
539f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
540f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
541f9c275feSReid Kleckner 
542876330d5SReid Kleckner     Processed.insert(InlinedVariable(VI.Var, VI.Loc->getInlinedAt()));
543f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
544f9c275feSReid Kleckner 
545f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
546f9c275feSReid Kleckner     if (!Scope)
547f9c275feSReid Kleckner       continue;
548f9c275feSReid Kleckner 
549f9c275feSReid Kleckner     // Get the frame register used and the offset.
550f9c275feSReid Kleckner     unsigned FrameReg = 0;
551876330d5SReid Kleckner     int FrameOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
552876330d5SReid Kleckner     uint16_t CVReg = TRI->getCodeViewRegNum(FrameReg);
553f9c275feSReid Kleckner 
554f9c275feSReid Kleckner     // Calculate the label ranges.
555876330d5SReid Kleckner     LocalVarDefRange DefRange = createDefRangeMem(CVReg, FrameOffset);
556f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
557f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
558f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
559876330d5SReid Kleckner       End = End ? End : Asm->getFunctionEnd();
560876330d5SReid Kleckner       DefRange.Ranges.emplace_back(Begin, End);
561f9c275feSReid Kleckner     }
562f9c275feSReid Kleckner 
563876330d5SReid Kleckner     LocalVariable Var;
564876330d5SReid Kleckner     Var.DIVar = VI.Var;
565876330d5SReid Kleckner     Var.DefRanges.emplace_back(std::move(DefRange));
566876330d5SReid Kleckner     recordLocalVariable(std::move(Var), VI.Loc->getInlinedAt());
567f9c275feSReid Kleckner   }
568f9c275feSReid Kleckner }
569876330d5SReid Kleckner 
570876330d5SReid Kleckner void CodeViewDebug::collectVariableInfo(const DISubprogram *SP) {
571876330d5SReid Kleckner   DenseSet<InlinedVariable> Processed;
572876330d5SReid Kleckner   // Grab the variable info that was squirreled away in the MMI side-table.
573876330d5SReid Kleckner   collectVariableInfoFromMMITable(Processed);
574876330d5SReid Kleckner 
575876330d5SReid Kleckner   const TargetRegisterInfo *TRI = Asm->MF->getSubtarget().getRegisterInfo();
576876330d5SReid Kleckner 
577876330d5SReid Kleckner   for (const auto &I : DbgValues) {
578876330d5SReid Kleckner     InlinedVariable IV = I.first;
579876330d5SReid Kleckner     if (Processed.count(IV))
580876330d5SReid Kleckner       continue;
581876330d5SReid Kleckner     const DILocalVariable *DIVar = IV.first;
582876330d5SReid Kleckner     const DILocation *InlinedAt = IV.second;
583876330d5SReid Kleckner 
584876330d5SReid Kleckner     // Instruction ranges, specifying where IV is accessible.
585876330d5SReid Kleckner     const auto &Ranges = I.second;
586876330d5SReid Kleckner 
587876330d5SReid Kleckner     LexicalScope *Scope = nullptr;
588876330d5SReid Kleckner     if (InlinedAt)
589876330d5SReid Kleckner       Scope = LScopes.findInlinedScope(DIVar->getScope(), InlinedAt);
590876330d5SReid Kleckner     else
591876330d5SReid Kleckner       Scope = LScopes.findLexicalScope(DIVar->getScope());
592876330d5SReid Kleckner     // If variable scope is not found then skip this variable.
593876330d5SReid Kleckner     if (!Scope)
594876330d5SReid Kleckner       continue;
595876330d5SReid Kleckner 
596876330d5SReid Kleckner     LocalVariable Var;
597876330d5SReid Kleckner     Var.DIVar = DIVar;
598876330d5SReid Kleckner 
599876330d5SReid Kleckner     // Calculate the definition ranges.
600876330d5SReid Kleckner     for (auto I = Ranges.begin(), E = Ranges.end(); I != E; ++I) {
601876330d5SReid Kleckner       const InsnRange &Range = *I;
602876330d5SReid Kleckner       const MachineInstr *DVInst = Range.first;
603876330d5SReid Kleckner       assert(DVInst->isDebugValue() && "Invalid History entry");
604876330d5SReid Kleckner       const DIExpression *DIExpr = DVInst->getDebugExpression();
605876330d5SReid Kleckner 
606876330d5SReid Kleckner       // Bail if there is a complex DWARF expression for now.
607876330d5SReid Kleckner       if (DIExpr && DIExpr->getNumElements() > 0)
608876330d5SReid Kleckner         continue;
609876330d5SReid Kleckner 
6109a593ee7SReid Kleckner       // Bail if operand 0 is not a valid register. This means the variable is a
6119a593ee7SReid Kleckner       // simple constant, or is described by a complex expression.
6129a593ee7SReid Kleckner       // FIXME: Find a way to represent constant variables, since they are
6139a593ee7SReid Kleckner       // relatively common.
6149a593ee7SReid Kleckner       unsigned Reg =
6159a593ee7SReid Kleckner           DVInst->getOperand(0).isReg() ? DVInst->getOperand(0).getReg() : 0;
6169a593ee7SReid Kleckner       if (Reg == 0)
6176e0d5f57SReid Kleckner         continue;
6186e0d5f57SReid Kleckner 
619876330d5SReid Kleckner       // Handle the two cases we can handle: indirect in memory and in register.
620876330d5SReid Kleckner       bool IsIndirect = DVInst->getOperand(1).isImm();
621876330d5SReid Kleckner       unsigned CVReg = TRI->getCodeViewRegNum(DVInst->getOperand(0).getReg());
622876330d5SReid Kleckner       {
623876330d5SReid Kleckner         LocalVarDefRange DefRange;
624876330d5SReid Kleckner         if (IsIndirect) {
625876330d5SReid Kleckner           int64_t Offset = DVInst->getOperand(1).getImm();
626876330d5SReid Kleckner           DefRange = createDefRangeMem(CVReg, Offset);
627876330d5SReid Kleckner         } else {
628876330d5SReid Kleckner           DefRange = createDefRangeReg(CVReg);
629876330d5SReid Kleckner         }
630876330d5SReid Kleckner         if (Var.DefRanges.empty() ||
631876330d5SReid Kleckner             Var.DefRanges.back().isDifferentLocation(DefRange)) {
632876330d5SReid Kleckner           Var.DefRanges.emplace_back(std::move(DefRange));
633876330d5SReid Kleckner         }
634876330d5SReid Kleckner       }
635876330d5SReid Kleckner 
636876330d5SReid Kleckner       // Compute the label range.
637876330d5SReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
638876330d5SReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
639876330d5SReid Kleckner       if (!End) {
640876330d5SReid Kleckner         if (std::next(I) != E)
641876330d5SReid Kleckner           End = getLabelBeforeInsn(std::next(I)->first);
642876330d5SReid Kleckner         else
643876330d5SReid Kleckner           End = Asm->getFunctionEnd();
644876330d5SReid Kleckner       }
645876330d5SReid Kleckner 
646876330d5SReid Kleckner       // If the last range end is our begin, just extend the last range.
647876330d5SReid Kleckner       // Otherwise make a new range.
648876330d5SReid Kleckner       SmallVectorImpl<std::pair<const MCSymbol *, const MCSymbol *>> &Ranges =
649876330d5SReid Kleckner           Var.DefRanges.back().Ranges;
650876330d5SReid Kleckner       if (!Ranges.empty() && Ranges.back().second == Begin)
651876330d5SReid Kleckner         Ranges.back().second = End;
652876330d5SReid Kleckner       else
653876330d5SReid Kleckner         Ranges.emplace_back(Begin, End);
654876330d5SReid Kleckner 
655876330d5SReid Kleckner       // FIXME: Do more range combining.
656876330d5SReid Kleckner     }
657876330d5SReid Kleckner 
658876330d5SReid Kleckner     recordLocalVariable(std::move(Var), InlinedAt);
659876330d5SReid Kleckner   }
660f9c275feSReid Kleckner }
661f9c275feSReid Kleckner 
66270f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
66370f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
66470f5bc99SReid Kleckner 
665f9c275feSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
66670f5bc99SReid Kleckner     return;
66770f5bc99SReid Kleckner 
668f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
669f9c275feSReid Kleckner 
67070f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
67170f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
67270f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
6732214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
6741fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
67570f5bc99SReid Kleckner 
676f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
677f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
67870f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
67970f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
68070f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
68170f5bc99SReid Kleckner   bool EmptyPrologue = true;
68270f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
68370f5bc99SReid Kleckner     for (const auto &MI : MBB) {
684f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
685f9c275feSReid Kleckner           MI.getDebugLoc()) {
68670f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
68770f5bc99SReid Kleckner         break;
688f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
68970f5bc99SReid Kleckner         EmptyPrologue = false;
69070f5bc99SReid Kleckner       }
69170f5bc99SReid Kleckner     }
692f9c275feSReid Kleckner   }
693f9c275feSReid Kleckner 
69470f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
69570f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
69670f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
69770f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
69870f5bc99SReid Kleckner   }
69970f5bc99SReid Kleckner }
70070f5bc99SReid Kleckner 
701f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
702f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
703f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
704f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
705f9c275feSReid Kleckner   OS.AddComment("Record length");
706f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
707f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
708f9c275feSReid Kleckner 
709f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
71063a2846eSZachary Turner   OS.EmitIntValue(unsigned(SymbolKind::S_LOCAL), 2);
711f9c275feSReid Kleckner 
71263a2846eSZachary Turner   LocalSymFlags Flags = LocalSymFlags::None;
713f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
71463a2846eSZachary Turner     Flags |= LocalSymFlags::IsParameter;
715876330d5SReid Kleckner   if (Var.DefRanges.empty())
71663a2846eSZachary Turner     Flags |= LocalSymFlags::IsOptimizedOut;
717f9c275feSReid Kleckner 
718f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
719f9c275feSReid Kleckner   OS.EmitIntValue(TypeIndex::Int32().getIndex(), 4);
720f9c275feSReid Kleckner   OS.AddComment("Flags");
72163a2846eSZachary Turner   OS.EmitIntValue(static_cast<uint16_t>(Flags), 2);
7221256125fSDavid Majnemer   // Truncate the name so we won't overflow the record length field.
723b9456a5eSDavid Majnemer   emitNullTerminatedSymbolName(OS, Var.DIVar->getName());
724f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
725f9c275feSReid Kleckner 
726876330d5SReid Kleckner   // Calculate the on disk prefix of the appropriate def range record. The
727876330d5SReid Kleckner   // records and on disk formats are described in SymbolRecords.h. BytePrefix
728876330d5SReid Kleckner   // should be big enough to hold all forms without memory allocation.
729876330d5SReid Kleckner   SmallString<20> BytePrefix;
730876330d5SReid Kleckner   for (const LocalVarDefRange &DefRange : Var.DefRanges) {
731876330d5SReid Kleckner     BytePrefix.clear();
732876330d5SReid Kleckner     // FIXME: Handle bitpieces.
733876330d5SReid Kleckner     if (DefRange.StructOffset != 0)
734876330d5SReid Kleckner       continue;
735876330d5SReid Kleckner 
736876330d5SReid Kleckner     if (DefRange.InMemory) {
737*a78ecd1eSZachary Turner       DefRangeRegisterRelSym Sym(DefRange.CVRegister, 0, DefRange.DataOffset, 0,
738*a78ecd1eSZachary Turner                                  0, 0, ArrayRef<LocalVariableAddrGap>());
739f9c275feSReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
740876330d5SReid Kleckner       BytePrefix +=
741876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
742*a78ecd1eSZachary Turner       BytePrefix +=
743*a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
744*a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
745876330d5SReid Kleckner     } else {
746876330d5SReid Kleckner       assert(DefRange.DataOffset == 0 && "unexpected offset into register");
747*a78ecd1eSZachary Turner       // Unclear what matters here.
748*a78ecd1eSZachary Turner       DefRangeRegisterSym Sym(DefRange.CVRegister, 0, 0, 0, 0,
749*a78ecd1eSZachary Turner                               ArrayRef<LocalVariableAddrGap>());
750876330d5SReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER);
751876330d5SReid Kleckner       BytePrefix +=
752876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
753*a78ecd1eSZachary Turner       BytePrefix +=
754*a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
755*a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
756876330d5SReid Kleckner     }
757876330d5SReid Kleckner     OS.EmitCVDefRangeDirective(DefRange.Ranges, BytePrefix);
758876330d5SReid Kleckner   }
759f9c275feSReid Kleckner }
760f9c275feSReid Kleckner 
76170f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
76270f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
76370f5bc99SReid Kleckner     return;
76470f5bc99SReid Kleckner 
76570f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
76670f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
76770f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
76870f5bc99SReid Kleckner 
769adebb937SPete Cooper   collectVariableInfo(GV->getSubprogram());
770876330d5SReid Kleckner 
771876330d5SReid Kleckner   DebugHandlerBase::endFunction(MF);
772876330d5SReid Kleckner 
7732214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
7742214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
77570f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
776f9c275feSReid Kleckner     CurFn = nullptr;
777f9c275feSReid Kleckner     return;
77870f5bc99SReid Kleckner   }
779f9c275feSReid Kleckner 
780f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
781f9c275feSReid Kleckner 
78270f5bc99SReid Kleckner   CurFn = nullptr;
78370f5bc99SReid Kleckner }
78470f5bc99SReid Kleckner 
78570f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
786f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
787f9c275feSReid Kleckner 
78870f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
78970f5bc99SReid Kleckner   if (!Asm || MI->isDebugValue() || MI->getFlag(MachineInstr::FrameSetup))
79070f5bc99SReid Kleckner     return;
79170f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
79270f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
79370f5bc99SReid Kleckner     return;
79470f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
79570f5bc99SReid Kleckner }
796