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"
23*f9c275feSReid Kleckner #include "llvm/Target/TargetSubtargetInfo.h"
24*f9c275feSReid Kleckner #include "llvm/Target/TargetRegisterInfo.h"
25*f9c275feSReid Kleckner #include "llvm/Target/TargetFrameLowering.h"
2670f5bc99SReid Kleckner 
27*f9c275feSReid Kleckner using namespace llvm;
2870f5bc99SReid Kleckner using namespace llvm::codeview;
2970f5bc99SReid Kleckner 
30*f9c275feSReid Kleckner CodeViewDebug::CodeViewDebug(AsmPrinter *AP)
31*f9c275feSReid Kleckner     : DebugHandlerBase(AP), OS(*Asm->OutStreamer), CurFn(nullptr) {
32*f9c275feSReid Kleckner   // If module doesn't have named metadata anchors or COFF debug section
33*f9c275feSReid Kleckner   // is not available, skip any debug info related stuff.
34*f9c275feSReid Kleckner   if (!MMI->getModule()->getNamedMetadata("llvm.dbg.cu") ||
35*f9c275feSReid Kleckner       !AP->getObjFileLowering().getCOFFDebugSymbolsSection()) {
36*f9c275feSReid Kleckner     Asm = nullptr;
37*f9c275feSReid Kleckner     return;
38*f9c275feSReid Kleckner   }
39*f9c275feSReid Kleckner 
40*f9c275feSReid Kleckner   // Tell MMI that we have debug info.
41*f9c275feSReid Kleckner   MMI->setDebugInfoAvailability(true);
42*f9c275feSReid 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 
108f3b9ba49SReid Kleckner CodeViewDebug::InlineSite &CodeViewDebug::getInlineSite(const DILocation *Loc) {
109f3b9ba49SReid Kleckner   const DILocation *InlinedAt = Loc->getInlinedAt();
110f3b9ba49SReid Kleckner   auto Insertion = CurFn->InlineSites.insert({InlinedAt, InlineSite()});
111*f9c275feSReid Kleckner   InlineSite *Site = &Insertion.first->second;
112f3b9ba49SReid Kleckner   if (Insertion.second) {
113*f9c275feSReid Kleckner     Site->SiteFuncId = NextFuncId++;
114*f9c275feSReid Kleckner     Site->Inlinee = Loc->getScope()->getSubprogram();
1151fcd610cSReid Kleckner     InlinedSubprograms.insert(Loc->getScope()->getSubprogram());
116f3b9ba49SReid Kleckner   }
117*f9c275feSReid Kleckner   return *Site;
118f3b9ba49SReid Kleckner }
119f3b9ba49SReid Kleckner 
12070f5bc99SReid Kleckner void CodeViewDebug::maybeRecordLocation(DebugLoc DL,
12170f5bc99SReid Kleckner                                         const MachineFunction *MF) {
1229533af4fSReid Kleckner   // Skip this instruction if it has the same location as the previous one.
1239533af4fSReid Kleckner   if (DL == CurFn->LastLoc)
1249533af4fSReid Kleckner     return;
1259533af4fSReid Kleckner 
1269533af4fSReid Kleckner   const DIScope *Scope = DL.get()->getScope();
12770f5bc99SReid Kleckner   if (!Scope)
12870f5bc99SReid Kleckner     return;
1299533af4fSReid Kleckner 
13070f5bc99SReid Kleckner   // Skip this line if it is longer than the maximum we can record.
1312214ed89SReid Kleckner   LineInfo LI(DL.getLine(), DL.getLine(), /*IsStatement=*/true);
1322214ed89SReid Kleckner   if (LI.getStartLine() != DL.getLine() || LI.isAlwaysStepInto() ||
1332214ed89SReid Kleckner       LI.isNeverStepInto())
13470f5bc99SReid Kleckner     return;
13570f5bc99SReid Kleckner 
1362214ed89SReid Kleckner   ColumnInfo CI(DL.getCol(), /*EndColumn=*/0);
1372214ed89SReid Kleckner   if (CI.getStartColumn() != DL.getCol())
1382214ed89SReid Kleckner     return;
13900d9639cSReid Kleckner 
1402214ed89SReid Kleckner   if (!CurFn->HaveLineInfo)
1412214ed89SReid Kleckner     CurFn->HaveLineInfo = true;
1422214ed89SReid Kleckner   unsigned FileId = 0;
1432214ed89SReid Kleckner   if (CurFn->LastLoc.get() && CurFn->LastLoc->getFile() == DL->getFile())
1442214ed89SReid Kleckner     FileId = CurFn->LastFileId;
1452214ed89SReid Kleckner   else
1462214ed89SReid Kleckner     FileId = CurFn->LastFileId = maybeRecordFile(DL->getFile());
1472214ed89SReid Kleckner   CurFn->LastLoc = DL;
148f3b9ba49SReid Kleckner 
149f3b9ba49SReid Kleckner   unsigned FuncId = CurFn->FuncId;
150f3b9ba49SReid Kleckner   if (const DILocation *Loc = DL->getInlinedAt()) {
151f3b9ba49SReid Kleckner     // If this location was actually inlined from somewhere else, give it the ID
152f3b9ba49SReid Kleckner     // of the inline call site.
153f3b9ba49SReid Kleckner     FuncId = getInlineSite(DL.get()).SiteFuncId;
154*f9c275feSReid Kleckner     CurFn->ChildSites.push_back(Loc);
155f3b9ba49SReid Kleckner     // Ensure we have links in the tree of inline call sites.
156f3b9ba49SReid Kleckner     const DILocation *ChildLoc = nullptr;
157f3b9ba49SReid Kleckner     while (Loc->getInlinedAt()) {
158f3b9ba49SReid Kleckner       InlineSite &Site = getInlineSite(Loc);
159f3b9ba49SReid Kleckner       if (ChildLoc) {
160f3b9ba49SReid Kleckner         // Record the child inline site if not already present.
161f3b9ba49SReid Kleckner         auto B = Site.ChildSites.begin(), E = Site.ChildSites.end();
162f3b9ba49SReid Kleckner         if (std::find(B, E, Loc) != E)
163f3b9ba49SReid Kleckner           break;
164f3b9ba49SReid Kleckner         Site.ChildSites.push_back(Loc);
165f3b9ba49SReid Kleckner       }
166f3b9ba49SReid Kleckner       ChildLoc = Loc;
167f3b9ba49SReid Kleckner     }
168f3b9ba49SReid Kleckner   }
169f3b9ba49SReid Kleckner 
170dac21b43SReid Kleckner   OS.EmitCVLocDirective(FuncId, FileId, DL.getLine(), DL.getCol(),
171dac21b43SReid Kleckner                         /*PrologueEnd=*/false,
1722214ed89SReid Kleckner                         /*IsStmt=*/false, DL->getFilename());
17370f5bc99SReid Kleckner }
17470f5bc99SReid Kleckner 
17570f5bc99SReid Kleckner void CodeViewDebug::endModule() {
17670f5bc99SReid Kleckner   if (FnDebugInfo.empty())
17770f5bc99SReid Kleckner     return;
17870f5bc99SReid Kleckner 
179f3b9ba49SReid Kleckner   emitTypeInformation();
180f3b9ba49SReid Kleckner 
18170f5bc99SReid Kleckner   // FIXME: For functions that are comdat, we should emit separate .debug$S
18270f5bc99SReid Kleckner   // sections that are comdat associative with the main function instead of
18370f5bc99SReid Kleckner   // having one big .debug$S section.
18470f5bc99SReid Kleckner   assert(Asm != nullptr);
185dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugSymbolsSection());
186dac21b43SReid Kleckner   OS.AddComment("Debug section magic");
187dac21b43SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
18870f5bc99SReid Kleckner 
18970f5bc99SReid Kleckner   // The COFF .debug$S section consists of several subsections, each starting
19070f5bc99SReid Kleckner   // with a 4-byte control code (e.g. 0xF1, 0xF2, etc) and then a 4-byte length
19170f5bc99SReid Kleckner   // of the payload followed by the payload itself.  The subsections are 4-byte
19270f5bc99SReid Kleckner   // aligned.
19370f5bc99SReid Kleckner 
1941fcd610cSReid Kleckner   // Make a subsection for all the inlined subprograms.
1951fcd610cSReid Kleckner   emitInlineeLinesSubsection();
1961fcd610cSReid Kleckner 
1972214ed89SReid Kleckner   // Emit per-function debug information.
1982214ed89SReid Kleckner   for (auto &P : FnDebugInfo)
1992214ed89SReid Kleckner     emitDebugInfoForFunction(P.first, P.second);
20070f5bc99SReid Kleckner 
20170f5bc99SReid Kleckner   // This subsection holds a file index to offset in string table table.
202dac21b43SReid Kleckner   OS.AddComment("File index to string table offset subsection");
203dac21b43SReid Kleckner   OS.EmitCVFileChecksumsDirective();
20470f5bc99SReid Kleckner 
20570f5bc99SReid Kleckner   // This subsection holds the string table.
206dac21b43SReid Kleckner   OS.AddComment("String table");
207dac21b43SReid Kleckner   OS.EmitCVStringTableDirective();
20870f5bc99SReid Kleckner 
20970f5bc99SReid Kleckner   clear();
21070f5bc99SReid Kleckner }
21170f5bc99SReid Kleckner 
212f3b9ba49SReid Kleckner void CodeViewDebug::emitTypeInformation() {
213f3b9ba49SReid Kleckner   // Start the .debug$T section with 0x4.
214dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugTypesSection());
215dac21b43SReid Kleckner   OS.AddComment("Debug section magic");
216dac21b43SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
217f3b9ba49SReid Kleckner 
218f3b9ba49SReid Kleckner   NamedMDNode *CU_Nodes =
219*f9c275feSReid Kleckner       MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
220f3b9ba49SReid Kleckner   if (!CU_Nodes)
221f3b9ba49SReid Kleckner     return;
222f3b9ba49SReid Kleckner 
223f3b9ba49SReid Kleckner   // This type info currently only holds function ids for use with inline call
224f3b9ba49SReid Kleckner   // frame info. All functions are assigned a simple 'void ()' type. Emit that
225f3b9ba49SReid Kleckner   // type here.
226f3b9ba49SReid Kleckner   TypeIndex ArgListIdx = getNextTypeIndex();
227dac21b43SReid Kleckner   OS.AddComment("Type record length");
228dac21b43SReid Kleckner   OS.EmitIntValue(2 + sizeof(ArgList), 2);
229dac21b43SReid Kleckner   OS.AddComment("Leaf type: LF_ARGLIST");
230dac21b43SReid Kleckner   OS.EmitIntValue(LF_ARGLIST, 2);
231dac21b43SReid Kleckner   OS.AddComment("Number of arguments");
232dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
233f3b9ba49SReid Kleckner 
234f3b9ba49SReid Kleckner   TypeIndex VoidProcIdx = getNextTypeIndex();
235dac21b43SReid Kleckner   OS.AddComment("Type record length");
236dac21b43SReid Kleckner   OS.EmitIntValue(2 + sizeof(ProcedureType), 2);
237dac21b43SReid Kleckner   OS.AddComment("Leaf type: LF_PROCEDURE");
238dac21b43SReid Kleckner   OS.EmitIntValue(LF_PROCEDURE, 2);
239dac21b43SReid Kleckner   OS.AddComment("Return type index");
240dac21b43SReid Kleckner   OS.EmitIntValue(TypeIndex::Void().getIndex(), 4);
241dac21b43SReid Kleckner   OS.AddComment("Calling convention");
242dac21b43SReid Kleckner   OS.EmitIntValue(char(CallingConvention::NearC), 1);
243dac21b43SReid Kleckner   OS.AddComment("Function options");
244dac21b43SReid Kleckner   OS.EmitIntValue(char(FunctionOptions::None), 1);
245dac21b43SReid Kleckner   OS.AddComment("# of parameters");
246dac21b43SReid Kleckner   OS.EmitIntValue(0, 2);
247dac21b43SReid Kleckner   OS.AddComment("Argument list type index");
248dac21b43SReid Kleckner   OS.EmitIntValue(ArgListIdx.getIndex(), 4);
249f3b9ba49SReid Kleckner 
250f3b9ba49SReid Kleckner   for (MDNode *N : CU_Nodes->operands()) {
251f3b9ba49SReid Kleckner     auto *CUNode = cast<DICompileUnit>(N);
252f3b9ba49SReid Kleckner     for (auto *SP : CUNode->getSubprograms()) {
253f3b9ba49SReid Kleckner       StringRef DisplayName = SP->getDisplayName();
254dac21b43SReid Kleckner       OS.AddComment("Type record length");
255dac21b43SReid Kleckner       OS.EmitIntValue(2 + sizeof(FuncId) + DisplayName.size() + 1, 2);
256dac21b43SReid Kleckner       OS.AddComment("Leaf type: LF_FUNC_ID");
257dac21b43SReid Kleckner       OS.EmitIntValue(LF_FUNC_ID, 2);
258f3b9ba49SReid Kleckner 
259dac21b43SReid Kleckner       OS.AddComment("Scope type index");
260dac21b43SReid Kleckner       OS.EmitIntValue(TypeIndex().getIndex(), 4);
261dac21b43SReid Kleckner       OS.AddComment("Function type");
262dac21b43SReid Kleckner       OS.EmitIntValue(VoidProcIdx.getIndex(), 4);
26330579ec8SDavid Majnemer       {
26430579ec8SDavid Majnemer         SmallString<32> NullTerminatedString(DisplayName);
26530579ec8SDavid Majnemer         if (NullTerminatedString.empty() || NullTerminatedString.back() != '\0')
26630579ec8SDavid Majnemer           NullTerminatedString.push_back('\0');
267dac21b43SReid Kleckner         OS.AddComment("Function name");
268dac21b43SReid Kleckner         OS.EmitBytes(NullTerminatedString);
26930579ec8SDavid Majnemer       }
270f3b9ba49SReid Kleckner 
271f3b9ba49SReid Kleckner       TypeIndex FuncIdIdx = getNextTypeIndex();
272f3b9ba49SReid Kleckner       SubprogramToFuncId.insert(std::make_pair(SP, FuncIdIdx));
273f3b9ba49SReid Kleckner     }
274f3b9ba49SReid Kleckner   }
275f3b9ba49SReid Kleckner }
276f3b9ba49SReid Kleckner 
2771fcd610cSReid Kleckner void CodeViewDebug::emitInlineeLinesSubsection() {
2781fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
2791fcd610cSReid Kleckner     return;
2801fcd610cSReid Kleckner 
281*f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
282*f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
2831fcd610cSReid Kleckner 
2841fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
2851fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::InlineeLines), 4);
28630579ec8SDavid Majnemer   OS.AddComment("Subsection size");
2871fcd610cSReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 4);
2881fcd610cSReid Kleckner   OS.EmitLabel(InlineBegin);
2891fcd610cSReid Kleckner 
2901fcd610cSReid Kleckner   // We don't provide any extra file info.
2911fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
29230579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
2931fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
2941fcd610cSReid Kleckner 
2951fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
29630579ec8SDavid Majnemer     OS.AddBlankLine();
2971fcd610cSReid Kleckner     TypeIndex TypeId = SubprogramToFuncId[SP];
2981fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
2991fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
3001fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
30130579ec8SDavid Majnemer     OS.AddBlankLine();
3021fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
3031fcd610cSReid Kleckner     // 1.
3041fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
30530579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
3061fcd610cSReid Kleckner     OS.EmitIntValue(TypeId.getIndex(), 4);
30730579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
3081fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
30930579ec8SDavid Majnemer     OS.AddComment("Starting line number");
3101fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
3111fcd610cSReid Kleckner   }
3121fcd610cSReid Kleckner 
3131fcd610cSReid Kleckner   OS.EmitLabel(InlineEnd);
3141fcd610cSReid Kleckner }
3151fcd610cSReid Kleckner 
3161fcd610cSReid Kleckner void CodeViewDebug::collectInlineSiteChildren(
3171fcd610cSReid Kleckner     SmallVectorImpl<unsigned> &Children, const FunctionInfo &FI,
3181fcd610cSReid Kleckner     const InlineSite &Site) {
3191fcd610cSReid Kleckner   for (const DILocation *ChildSiteLoc : Site.ChildSites) {
3201fcd610cSReid Kleckner     auto I = FI.InlineSites.find(ChildSiteLoc);
3211fcd610cSReid Kleckner     const InlineSite &ChildSite = I->second;
3221fcd610cSReid Kleckner     Children.push_back(ChildSite.SiteFuncId);
3231fcd610cSReid Kleckner     collectInlineSiteChildren(Children, FI, ChildSite);
3241fcd610cSReid Kleckner   }
3251fcd610cSReid Kleckner }
3261fcd610cSReid Kleckner 
327f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
328f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
329f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
330*f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
331*f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
332f3b9ba49SReid Kleckner 
333f3b9ba49SReid Kleckner   assert(SubprogramToFuncId.count(Site.Inlinee));
334f3b9ba49SReid Kleckner   TypeIndex InlineeIdx = SubprogramToFuncId[Site.Inlinee];
335f3b9ba49SReid Kleckner 
336f3b9ba49SReid Kleckner   // SymbolRecord
337dac21b43SReid Kleckner   OS.AddComment("Record length");
338eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
339f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
340dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
341dac21b43SReid Kleckner   OS.EmitIntValue(SymbolRecordKind::S_INLINESITE, 2); // RecordKind
342f3b9ba49SReid Kleckner 
343dac21b43SReid Kleckner   OS.AddComment("PtrParent");
344dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
345dac21b43SReid Kleckner   OS.AddComment("PtrEnd");
346dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
347dac21b43SReid Kleckner   OS.AddComment("Inlinee type index");
348dac21b43SReid Kleckner   OS.EmitIntValue(InlineeIdx.getIndex(), 4);
349f3b9ba49SReid Kleckner 
3501fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
3511fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
3521fcd610cSReid Kleckner   SmallVector<unsigned, 3> SecondaryFuncIds;
3531fcd610cSReid Kleckner   collectInlineSiteChildren(SecondaryFuncIds, FI, Site);
3541fcd610cSReid Kleckner 
3551fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
356c9911f28SDavid Majnemer                                     FI.Begin, FI.End, SecondaryFuncIds);
357f3b9ba49SReid Kleckner 
358f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
359f3b9ba49SReid Kleckner 
360*f9c275feSReid Kleckner   for (const LocalVariable &Var : Site.InlinedLocals)
361*f9c275feSReid Kleckner     emitLocalVariable(Var);
362*f9c275feSReid Kleckner 
363f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
364f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
365f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
366f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
367f3b9ba49SReid Kleckner            "child site not in function inline site map");
368f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
369f3b9ba49SReid Kleckner   }
370f3b9ba49SReid Kleckner 
371f3b9ba49SReid Kleckner   // Close the scope.
372dac21b43SReid Kleckner   OS.AddComment("Record length");
373dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
374dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
375dac21b43SReid Kleckner   OS.EmitIntValue(SymbolRecordKind::S_INLINESITE_END, 2); // RecordKind
376f3b9ba49SReid Kleckner }
377f3b9ba49SReid Kleckner 
378*f9c275feSReid Kleckner static void emitNullTerminatedString(MCStreamer &OS, StringRef S) {
379*f9c275feSReid Kleckner   SmallString<32> NullTerminatedString(S);
380*f9c275feSReid Kleckner   if (NullTerminatedString.empty() || NullTerminatedString.back() != '\0')
381*f9c275feSReid Kleckner     NullTerminatedString.push_back('\0');
382*f9c275feSReid Kleckner   OS.EmitBytes(NullTerminatedString);
383*f9c275feSReid Kleckner }
384*f9c275feSReid Kleckner 
3852214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
3862214ed89SReid Kleckner                                              FunctionInfo &FI) {
38770f5bc99SReid Kleckner   // For each function there is a separate subsection
38870f5bc99SReid Kleckner   // which holds the PC to file:line table.
38970f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
39070f5bc99SReid Kleckner   assert(Fn);
39170f5bc99SReid Kleckner 
39270f5bc99SReid Kleckner   StringRef FuncName;
39370f5bc99SReid Kleckner   if (auto *SP = getDISubprogram(GV))
39470f5bc99SReid Kleckner     FuncName = SP->getDisplayName();
39570f5bc99SReid Kleckner 
39670f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
39770f5bc99SReid Kleckner   if (FuncName.empty())
39870f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
39970f5bc99SReid Kleckner 
40070f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
401*f9c275feSReid Kleckner   MCSymbol *SymbolsBegin = MMI->getContext().createTempSymbol(),
402*f9c275feSReid Kleckner            *SymbolsEnd = MMI->getContext().createTempSymbol();
403dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
404dac21b43SReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::Symbols), 4);
405dac21b43SReid Kleckner   OS.AddComment("Subsection size");
406eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(SymbolsEnd, SymbolsBegin, 4);
407dac21b43SReid Kleckner   OS.EmitLabel(SymbolsBegin);
40870f5bc99SReid Kleckner   {
409*f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
410*f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
411dac21b43SReid Kleckner     OS.AddComment("Record length");
412eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
413dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
41470f5bc99SReid Kleckner 
415dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
416dac21b43SReid Kleckner     OS.EmitIntValue(unsigned(SymbolRecordKind::S_GPROC32_ID), 2);
41770f5bc99SReid Kleckner 
41830579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
419dac21b43SReid Kleckner     OS.AddComment("PtrParent");
420dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
421dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
422dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
423dac21b43SReid Kleckner     OS.AddComment("PtrNext");
424dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
42570f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
42670f5bc99SReid Kleckner     // code is located and what's its size:
427dac21b43SReid Kleckner     OS.AddComment("Code size");
428eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
429dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
430dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
431dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
432dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
433dac21b43SReid Kleckner     OS.AddComment("Function type index");
434dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
435dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
436dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
437dac21b43SReid Kleckner     OS.AddComment("Function section index");
438dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
439dac21b43SReid Kleckner     OS.AddComment("Flags");
440dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
44170f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
442dac21b43SReid Kleckner     OS.AddComment("Function name");
443*f9c275feSReid Kleckner     emitNullTerminatedString(OS, FuncName);
444dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
44570f5bc99SReid Kleckner 
446*f9c275feSReid Kleckner     for (const LocalVariable &Var : FI.Locals)
447*f9c275feSReid Kleckner       emitLocalVariable(Var);
448*f9c275feSReid Kleckner 
449f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
450f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
451f3b9ba49SReid Kleckner     // of their parent inline site.
452*f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
453*f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
454*f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
455*f9c275feSReid Kleckner              "child site not in function inline site map");
456*f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
457f3b9ba49SReid Kleckner     }
458f3b9ba49SReid Kleckner 
45970f5bc99SReid Kleckner     // We're done with this function.
460dac21b43SReid Kleckner     OS.AddComment("Record length");
461dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
462dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
463dac21b43SReid Kleckner     OS.EmitIntValue(unsigned(SymbolRecordKind::S_PROC_ID_END), 2);
46470f5bc99SReid Kleckner   }
465dac21b43SReid Kleckner   OS.EmitLabel(SymbolsEnd);
46670f5bc99SReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
467dac21b43SReid Kleckner   OS.EmitValueToAlignment(4);
46870f5bc99SReid Kleckner 
4692214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
470dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
47170f5bc99SReid Kleckner }
47270f5bc99SReid Kleckner 
473*f9c275feSReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable() {
474*f9c275feSReid Kleckner   for (const auto &VI : MMI->getVariableDbgInfo()) {
475*f9c275feSReid Kleckner     if (!VI.Var)
476*f9c275feSReid Kleckner       continue;
477*f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
478*f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
479*f9c275feSReid Kleckner 
480*f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
481*f9c275feSReid Kleckner 
482*f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
483*f9c275feSReid Kleckner     if (!Scope)
484*f9c275feSReid Kleckner       continue;
485*f9c275feSReid Kleckner 
486*f9c275feSReid Kleckner     LocalVariable Var;
487*f9c275feSReid Kleckner     Var.DIVar = VI.Var;
488*f9c275feSReid Kleckner 
489*f9c275feSReid Kleckner     // Get the frame register used and the offset.
490*f9c275feSReid Kleckner     unsigned FrameReg = 0;
491*f9c275feSReid Kleckner     const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
492*f9c275feSReid Kleckner     const TargetFrameLowering *TFI = TSI.getFrameLowering();
493*f9c275feSReid Kleckner     const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
494*f9c275feSReid Kleckner     Var.RegisterOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
495*f9c275feSReid Kleckner     Var.CVRegister = TRI->getCodeViewRegNum(FrameReg);
496*f9c275feSReid Kleckner 
497*f9c275feSReid Kleckner     // Calculate the label ranges.
498*f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
499*f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
500*f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
501*f9c275feSReid Kleckner       Var.Ranges.push_back({Begin, End});
502*f9c275feSReid Kleckner     }
503*f9c275feSReid Kleckner 
504*f9c275feSReid Kleckner     if (VI.Loc->getInlinedAt()) {
505*f9c275feSReid Kleckner       // This variable was inlined. Associate it with the InlineSite.
506*f9c275feSReid Kleckner       InlineSite &Site = getInlineSite(VI.Loc);
507*f9c275feSReid Kleckner       Site.InlinedLocals.emplace_back(std::move(Var));
508*f9c275feSReid Kleckner     } else {
509*f9c275feSReid Kleckner       // This variable goes in the main ProcSym.
510*f9c275feSReid Kleckner       CurFn->Locals.emplace_back(std::move(Var));
511*f9c275feSReid Kleckner     }
512*f9c275feSReid Kleckner   }
513*f9c275feSReid Kleckner }
514*f9c275feSReid Kleckner 
51570f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
51670f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
51770f5bc99SReid Kleckner 
518*f9c275feSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
51970f5bc99SReid Kleckner     return;
52070f5bc99SReid Kleckner 
521*f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
522*f9c275feSReid Kleckner 
52370f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
52470f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
52570f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
5262214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
5271fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
52870f5bc99SReid Kleckner 
529*f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
530*f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
53170f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
53270f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
53370f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
53470f5bc99SReid Kleckner   bool EmptyPrologue = true;
53570f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
53670f5bc99SReid Kleckner     for (const auto &MI : MBB) {
537*f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
538*f9c275feSReid Kleckner           MI.getDebugLoc()) {
53970f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
54070f5bc99SReid Kleckner         break;
541*f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
54270f5bc99SReid Kleckner         EmptyPrologue = false;
54370f5bc99SReid Kleckner       }
54470f5bc99SReid Kleckner     }
545*f9c275feSReid Kleckner   }
546*f9c275feSReid Kleckner 
54770f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
54870f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
54970f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
55070f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
55170f5bc99SReid Kleckner   }
55270f5bc99SReid Kleckner }
55370f5bc99SReid Kleckner 
554*f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
555*f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
556*f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
557*f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
558*f9c275feSReid Kleckner   OS.AddComment("Record length");
559*f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
560*f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
561*f9c275feSReid Kleckner 
562*f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
563*f9c275feSReid Kleckner   OS.EmitIntValue(unsigned(SymbolRecordKind::S_LOCAL), 2);
564*f9c275feSReid Kleckner 
565*f9c275feSReid Kleckner   uint16_t Flags = 0;
566*f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
567*f9c275feSReid Kleckner     Flags |= LocalSym::IsParameter;
568*f9c275feSReid Kleckner 
569*f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
570*f9c275feSReid Kleckner   OS.EmitIntValue(TypeIndex::Int32().getIndex(), 4);
571*f9c275feSReid Kleckner   OS.AddComment("Flags");
572*f9c275feSReid Kleckner   OS.EmitIntValue(Flags, 2);
573*f9c275feSReid Kleckner   emitNullTerminatedString(OS, Var.DIVar->getName());
574*f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
575*f9c275feSReid Kleckner 
576*f9c275feSReid Kleckner   // DefRangeRegisterRelSym record, see SymbolRecord.h for more info.  Omit the
577*f9c275feSReid Kleckner   // LocalVariableAddrRange field from the record. The directive will emit that.
578*f9c275feSReid Kleckner   DefRangeRegisterRelSym Sym{};
579*f9c275feSReid Kleckner   ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
580*f9c275feSReid Kleckner   Sym.BaseRegister = Var.CVRegister;
581*f9c275feSReid Kleckner   Sym.Flags = 0; // Unclear what matters here.
582*f9c275feSReid Kleckner   Sym.BasePointerOffset = Var.RegisterOffset;
583*f9c275feSReid Kleckner   SmallString<sizeof(Sym) + sizeof(SymKind) - sizeof(LocalVariableAddrRange)>
584*f9c275feSReid Kleckner       BytePrefix;
585*f9c275feSReid Kleckner   BytePrefix += StringRef(reinterpret_cast<const char *>(&SymKind),
586*f9c275feSReid Kleckner                           sizeof(SymKind));
587*f9c275feSReid Kleckner   BytePrefix += StringRef(reinterpret_cast<const char *>(&Sym),
588*f9c275feSReid Kleckner                           sizeof(Sym) - sizeof(LocalVariableAddrRange));
589*f9c275feSReid Kleckner 
590*f9c275feSReid Kleckner   OS.EmitCVDefRangeDirective(Var.Ranges, BytePrefix);
591*f9c275feSReid Kleckner }
592*f9c275feSReid Kleckner 
59370f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
594*f9c275feSReid Kleckner   collectVariableInfoFromMMITable();
595*f9c275feSReid Kleckner 
596*f9c275feSReid Kleckner   DebugHandlerBase::endFunction(MF);
597*f9c275feSReid Kleckner 
59870f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
59970f5bc99SReid Kleckner     return;
60070f5bc99SReid Kleckner 
60170f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
60270f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
60370f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
60470f5bc99SReid Kleckner 
6052214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
6062214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
60770f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
608*f9c275feSReid Kleckner     CurFn = nullptr;
609*f9c275feSReid Kleckner     return;
61070f5bc99SReid Kleckner   }
611*f9c275feSReid Kleckner 
612*f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
613*f9c275feSReid Kleckner 
61470f5bc99SReid Kleckner   CurFn = nullptr;
61570f5bc99SReid Kleckner }
61670f5bc99SReid Kleckner 
61770f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
618*f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
619*f9c275feSReid Kleckner 
62070f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
62170f5bc99SReid Kleckner   if (!Asm || MI->isDebugValue() || MI->getFlag(MachineInstr::FrameSetup))
62270f5bc99SReid Kleckner     return;
62370f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
62470f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
62570f5bc99SReid Kleckner     return;
62670f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
62770f5bc99SReid Kleckner }
628