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 
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()});
111f9c275feSReid Kleckner   InlineSite *Site = &Insertion.first->second;
112f3b9ba49SReid Kleckner   if (Insertion.second) {
113f9c275feSReid Kleckner     Site->SiteFuncId = NextFuncId++;
114f9c275feSReid Kleckner     Site->Inlinee = Loc->getScope()->getSubprogram();
1151fcd610cSReid Kleckner     InlinedSubprograms.insert(Loc->getScope()->getSubprogram());
116f3b9ba49SReid Kleckner   }
117f9c275feSReid Kleckner   return *Site;
118f3b9ba49SReid Kleckner }
119f3b9ba49SReid Kleckner 
120*829365aeSReid Kleckner static void addLocIfNotPresent(SmallVectorImpl<const DILocation *> &Locs,
121*829365aeSReid Kleckner                                const DILocation *Loc) {
122*829365aeSReid Kleckner   auto B = Locs.begin(), E = Locs.end();
123*829365aeSReid Kleckner   if (std::find(B, E, Loc) == E)
124*829365aeSReid Kleckner     Locs.push_back(Loc);
125*829365aeSReid Kleckner }
126*829365aeSReid Kleckner 
12770f5bc99SReid Kleckner void CodeViewDebug::maybeRecordLocation(DebugLoc DL,
12870f5bc99SReid Kleckner                                         const MachineFunction *MF) {
1299533af4fSReid Kleckner   // Skip this instruction if it has the same location as the previous one.
1309533af4fSReid Kleckner   if (DL == CurFn->LastLoc)
1319533af4fSReid Kleckner     return;
1329533af4fSReid Kleckner 
1339533af4fSReid Kleckner   const DIScope *Scope = DL.get()->getScope();
13470f5bc99SReid Kleckner   if (!Scope)
13570f5bc99SReid Kleckner     return;
1369533af4fSReid Kleckner 
13770f5bc99SReid Kleckner   // Skip this line if it is longer than the maximum we can record.
1382214ed89SReid Kleckner   LineInfo LI(DL.getLine(), DL.getLine(), /*IsStatement=*/true);
1392214ed89SReid Kleckner   if (LI.getStartLine() != DL.getLine() || LI.isAlwaysStepInto() ||
1402214ed89SReid Kleckner       LI.isNeverStepInto())
14170f5bc99SReid Kleckner     return;
14270f5bc99SReid Kleckner 
1432214ed89SReid Kleckner   ColumnInfo CI(DL.getCol(), /*EndColumn=*/0);
1442214ed89SReid Kleckner   if (CI.getStartColumn() != DL.getCol())
1452214ed89SReid Kleckner     return;
14600d9639cSReid Kleckner 
1472214ed89SReid Kleckner   if (!CurFn->HaveLineInfo)
1482214ed89SReid Kleckner     CurFn->HaveLineInfo = true;
1492214ed89SReid Kleckner   unsigned FileId = 0;
1502214ed89SReid Kleckner   if (CurFn->LastLoc.get() && CurFn->LastLoc->getFile() == DL->getFile())
1512214ed89SReid Kleckner     FileId = CurFn->LastFileId;
1522214ed89SReid Kleckner   else
1532214ed89SReid Kleckner     FileId = CurFn->LastFileId = maybeRecordFile(DL->getFile());
1542214ed89SReid Kleckner   CurFn->LastLoc = DL;
155f3b9ba49SReid Kleckner 
156f3b9ba49SReid Kleckner   unsigned FuncId = CurFn->FuncId;
157*829365aeSReid Kleckner   if (DL->getInlinedAt()) {
158*829365aeSReid Kleckner     const DILocation *Loc = DL.get();
159*829365aeSReid Kleckner 
160f3b9ba49SReid Kleckner     // If this location was actually inlined from somewhere else, give it the ID
161f3b9ba49SReid Kleckner     // of the inline call site.
162*829365aeSReid Kleckner     FuncId = getInlineSite(Loc).SiteFuncId;
163*829365aeSReid Kleckner 
164f3b9ba49SReid Kleckner     // Ensure we have links in the tree of inline call sites.
165*829365aeSReid Kleckner     const DILocation *SiteLoc;
166*829365aeSReid Kleckner     bool FirstLoc = true;
167*829365aeSReid Kleckner     while ((SiteLoc = Loc->getInlinedAt())) {
168f3b9ba49SReid Kleckner       InlineSite &Site = getInlineSite(Loc);
169*829365aeSReid Kleckner       if (!FirstLoc)
170*829365aeSReid Kleckner         addLocIfNotPresent(Site.ChildSites, Loc);
171*829365aeSReid Kleckner       FirstLoc = false;
172*829365aeSReid Kleckner       Loc = SiteLoc;
173f3b9ba49SReid Kleckner     }
174*829365aeSReid Kleckner     addLocIfNotPresent(CurFn->ChildSites, Loc);
175f3b9ba49SReid Kleckner   }
176f3b9ba49SReid Kleckner 
177dac21b43SReid Kleckner   OS.EmitCVLocDirective(FuncId, FileId, DL.getLine(), DL.getCol(),
178dac21b43SReid Kleckner                         /*PrologueEnd=*/false,
1792214ed89SReid Kleckner                         /*IsStmt=*/false, DL->getFilename());
18070f5bc99SReid Kleckner }
18170f5bc99SReid Kleckner 
18270f5bc99SReid Kleckner void CodeViewDebug::endModule() {
18370f5bc99SReid Kleckner   if (FnDebugInfo.empty())
18470f5bc99SReid Kleckner     return;
18570f5bc99SReid Kleckner 
186f3b9ba49SReid Kleckner   emitTypeInformation();
187f3b9ba49SReid Kleckner 
18870f5bc99SReid Kleckner   // FIXME: For functions that are comdat, we should emit separate .debug$S
18970f5bc99SReid Kleckner   // sections that are comdat associative with the main function instead of
19070f5bc99SReid Kleckner   // having one big .debug$S section.
19170f5bc99SReid Kleckner   assert(Asm != nullptr);
192dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugSymbolsSection());
193dac21b43SReid Kleckner   OS.AddComment("Debug section magic");
194dac21b43SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
19570f5bc99SReid Kleckner 
19670f5bc99SReid Kleckner   // The COFF .debug$S section consists of several subsections, each starting
19770f5bc99SReid Kleckner   // with a 4-byte control code (e.g. 0xF1, 0xF2, etc) and then a 4-byte length
19870f5bc99SReid Kleckner   // of the payload followed by the payload itself.  The subsections are 4-byte
19970f5bc99SReid Kleckner   // aligned.
20070f5bc99SReid Kleckner 
2011fcd610cSReid Kleckner   // Make a subsection for all the inlined subprograms.
2021fcd610cSReid Kleckner   emitInlineeLinesSubsection();
2031fcd610cSReid Kleckner 
2042214ed89SReid Kleckner   // Emit per-function debug information.
2052214ed89SReid Kleckner   for (auto &P : FnDebugInfo)
2062214ed89SReid Kleckner     emitDebugInfoForFunction(P.first, P.second);
20770f5bc99SReid Kleckner 
20870f5bc99SReid Kleckner   // This subsection holds a file index to offset in string table table.
209dac21b43SReid Kleckner   OS.AddComment("File index to string table offset subsection");
210dac21b43SReid Kleckner   OS.EmitCVFileChecksumsDirective();
21170f5bc99SReid Kleckner 
21270f5bc99SReid Kleckner   // This subsection holds the string table.
213dac21b43SReid Kleckner   OS.AddComment("String table");
214dac21b43SReid Kleckner   OS.EmitCVStringTableDirective();
21570f5bc99SReid Kleckner 
21670f5bc99SReid Kleckner   clear();
21770f5bc99SReid Kleckner }
21870f5bc99SReid Kleckner 
219f3b9ba49SReid Kleckner void CodeViewDebug::emitTypeInformation() {
220f3b9ba49SReid Kleckner   // Start the .debug$T section with 0x4.
221dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugTypesSection());
222dac21b43SReid Kleckner   OS.AddComment("Debug section magic");
223dac21b43SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
224f3b9ba49SReid Kleckner 
225f3b9ba49SReid Kleckner   NamedMDNode *CU_Nodes =
226f9c275feSReid Kleckner       MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
227f3b9ba49SReid Kleckner   if (!CU_Nodes)
228f3b9ba49SReid Kleckner     return;
229f3b9ba49SReid Kleckner 
230f3b9ba49SReid Kleckner   // This type info currently only holds function ids for use with inline call
231f3b9ba49SReid Kleckner   // frame info. All functions are assigned a simple 'void ()' type. Emit that
232f3b9ba49SReid Kleckner   // type here.
233f3b9ba49SReid Kleckner   TypeIndex ArgListIdx = getNextTypeIndex();
234dac21b43SReid Kleckner   OS.AddComment("Type record length");
235dac21b43SReid Kleckner   OS.EmitIntValue(2 + sizeof(ArgList), 2);
236dac21b43SReid Kleckner   OS.AddComment("Leaf type: LF_ARGLIST");
237dac21b43SReid Kleckner   OS.EmitIntValue(LF_ARGLIST, 2);
238dac21b43SReid Kleckner   OS.AddComment("Number of arguments");
239dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
240f3b9ba49SReid Kleckner 
241f3b9ba49SReid Kleckner   TypeIndex VoidProcIdx = getNextTypeIndex();
242dac21b43SReid Kleckner   OS.AddComment("Type record length");
243dac21b43SReid Kleckner   OS.EmitIntValue(2 + sizeof(ProcedureType), 2);
244dac21b43SReid Kleckner   OS.AddComment("Leaf type: LF_PROCEDURE");
245dac21b43SReid Kleckner   OS.EmitIntValue(LF_PROCEDURE, 2);
246dac21b43SReid Kleckner   OS.AddComment("Return type index");
247dac21b43SReid Kleckner   OS.EmitIntValue(TypeIndex::Void().getIndex(), 4);
248dac21b43SReid Kleckner   OS.AddComment("Calling convention");
249dac21b43SReid Kleckner   OS.EmitIntValue(char(CallingConvention::NearC), 1);
250dac21b43SReid Kleckner   OS.AddComment("Function options");
251dac21b43SReid Kleckner   OS.EmitIntValue(char(FunctionOptions::None), 1);
252dac21b43SReid Kleckner   OS.AddComment("# of parameters");
253dac21b43SReid Kleckner   OS.EmitIntValue(0, 2);
254dac21b43SReid Kleckner   OS.AddComment("Argument list type index");
255dac21b43SReid Kleckner   OS.EmitIntValue(ArgListIdx.getIndex(), 4);
256f3b9ba49SReid Kleckner 
257f3b9ba49SReid Kleckner   for (MDNode *N : CU_Nodes->operands()) {
258f3b9ba49SReid Kleckner     auto *CUNode = cast<DICompileUnit>(N);
259f3b9ba49SReid Kleckner     for (auto *SP : CUNode->getSubprograms()) {
260f3b9ba49SReid Kleckner       StringRef DisplayName = SP->getDisplayName();
261dac21b43SReid Kleckner       OS.AddComment("Type record length");
262dac21b43SReid Kleckner       OS.EmitIntValue(2 + sizeof(FuncId) + DisplayName.size() + 1, 2);
263dac21b43SReid Kleckner       OS.AddComment("Leaf type: LF_FUNC_ID");
264dac21b43SReid Kleckner       OS.EmitIntValue(LF_FUNC_ID, 2);
265f3b9ba49SReid Kleckner 
266dac21b43SReid Kleckner       OS.AddComment("Scope type index");
267dac21b43SReid Kleckner       OS.EmitIntValue(TypeIndex().getIndex(), 4);
268dac21b43SReid Kleckner       OS.AddComment("Function type");
269dac21b43SReid Kleckner       OS.EmitIntValue(VoidProcIdx.getIndex(), 4);
27030579ec8SDavid Majnemer       {
27130579ec8SDavid Majnemer         SmallString<32> NullTerminatedString(DisplayName);
27230579ec8SDavid Majnemer         if (NullTerminatedString.empty() || NullTerminatedString.back() != '\0')
27330579ec8SDavid Majnemer           NullTerminatedString.push_back('\0');
274dac21b43SReid Kleckner         OS.AddComment("Function name");
275dac21b43SReid Kleckner         OS.EmitBytes(NullTerminatedString);
27630579ec8SDavid Majnemer       }
277f3b9ba49SReid Kleckner 
278f3b9ba49SReid Kleckner       TypeIndex FuncIdIdx = getNextTypeIndex();
279f3b9ba49SReid Kleckner       SubprogramToFuncId.insert(std::make_pair(SP, FuncIdIdx));
280f3b9ba49SReid Kleckner     }
281f3b9ba49SReid Kleckner   }
282f3b9ba49SReid Kleckner }
283f3b9ba49SReid Kleckner 
2841fcd610cSReid Kleckner void CodeViewDebug::emitInlineeLinesSubsection() {
2851fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
2861fcd610cSReid Kleckner     return;
2871fcd610cSReid Kleckner 
288f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
289f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
2901fcd610cSReid Kleckner 
2911fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
2921fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::InlineeLines), 4);
29330579ec8SDavid Majnemer   OS.AddComment("Subsection size");
2941fcd610cSReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 4);
2951fcd610cSReid Kleckner   OS.EmitLabel(InlineBegin);
2961fcd610cSReid Kleckner 
2971fcd610cSReid Kleckner   // We don't provide any extra file info.
2981fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
29930579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
3001fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
3011fcd610cSReid Kleckner 
3021fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
30330579ec8SDavid Majnemer     OS.AddBlankLine();
3041fcd610cSReid Kleckner     TypeIndex TypeId = SubprogramToFuncId[SP];
3051fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
3061fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
3071fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
30830579ec8SDavid Majnemer     OS.AddBlankLine();
3091fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
3101fcd610cSReid Kleckner     // 1.
3111fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
31230579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
3131fcd610cSReid Kleckner     OS.EmitIntValue(TypeId.getIndex(), 4);
31430579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
3151fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
31630579ec8SDavid Majnemer     OS.AddComment("Starting line number");
3171fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
3181fcd610cSReid Kleckner   }
3191fcd610cSReid Kleckner 
3201fcd610cSReid Kleckner   OS.EmitLabel(InlineEnd);
3211fcd610cSReid Kleckner }
3221fcd610cSReid Kleckner 
3231fcd610cSReid Kleckner void CodeViewDebug::collectInlineSiteChildren(
3241fcd610cSReid Kleckner     SmallVectorImpl<unsigned> &Children, const FunctionInfo &FI,
3251fcd610cSReid Kleckner     const InlineSite &Site) {
3261fcd610cSReid Kleckner   for (const DILocation *ChildSiteLoc : Site.ChildSites) {
3271fcd610cSReid Kleckner     auto I = FI.InlineSites.find(ChildSiteLoc);
3281fcd610cSReid Kleckner     const InlineSite &ChildSite = I->second;
3291fcd610cSReid Kleckner     Children.push_back(ChildSite.SiteFuncId);
3301fcd610cSReid Kleckner     collectInlineSiteChildren(Children, FI, ChildSite);
3311fcd610cSReid Kleckner   }
3321fcd610cSReid Kleckner }
3331fcd610cSReid Kleckner 
334f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
335f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
336f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
337f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
338f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
339f3b9ba49SReid Kleckner 
340f3b9ba49SReid Kleckner   assert(SubprogramToFuncId.count(Site.Inlinee));
341f3b9ba49SReid Kleckner   TypeIndex InlineeIdx = SubprogramToFuncId[Site.Inlinee];
342f3b9ba49SReid Kleckner 
343f3b9ba49SReid Kleckner   // SymbolRecord
344dac21b43SReid Kleckner   OS.AddComment("Record length");
345eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
346f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
347dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
348dac21b43SReid Kleckner   OS.EmitIntValue(SymbolRecordKind::S_INLINESITE, 2); // RecordKind
349f3b9ba49SReid Kleckner 
350dac21b43SReid Kleckner   OS.AddComment("PtrParent");
351dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
352dac21b43SReid Kleckner   OS.AddComment("PtrEnd");
353dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
354dac21b43SReid Kleckner   OS.AddComment("Inlinee type index");
355dac21b43SReid Kleckner   OS.EmitIntValue(InlineeIdx.getIndex(), 4);
356f3b9ba49SReid Kleckner 
3571fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
3581fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
3591fcd610cSReid Kleckner   SmallVector<unsigned, 3> SecondaryFuncIds;
3601fcd610cSReid Kleckner   collectInlineSiteChildren(SecondaryFuncIds, FI, Site);
3611fcd610cSReid Kleckner 
3621fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
363c9911f28SDavid Majnemer                                     FI.Begin, FI.End, SecondaryFuncIds);
364f3b9ba49SReid Kleckner 
365f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
366f3b9ba49SReid Kleckner 
367f9c275feSReid Kleckner   for (const LocalVariable &Var : Site.InlinedLocals)
368f9c275feSReid Kleckner     emitLocalVariable(Var);
369f9c275feSReid Kleckner 
370f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
371f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
372f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
373f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
374f3b9ba49SReid Kleckner            "child site not in function inline site map");
375f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
376f3b9ba49SReid Kleckner   }
377f3b9ba49SReid Kleckner 
378f3b9ba49SReid Kleckner   // Close the scope.
379dac21b43SReid Kleckner   OS.AddComment("Record length");
380dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
381dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
382dac21b43SReid Kleckner   OS.EmitIntValue(SymbolRecordKind::S_INLINESITE_END, 2); // RecordKind
383f3b9ba49SReid Kleckner }
384f3b9ba49SReid Kleckner 
385f9c275feSReid Kleckner static void emitNullTerminatedString(MCStreamer &OS, StringRef S) {
386f9c275feSReid Kleckner   SmallString<32> NullTerminatedString(S);
387f9c275feSReid Kleckner   if (NullTerminatedString.empty() || NullTerminatedString.back() != '\0')
388f9c275feSReid Kleckner     NullTerminatedString.push_back('\0');
389f9c275feSReid Kleckner   OS.EmitBytes(NullTerminatedString);
390f9c275feSReid Kleckner }
391f9c275feSReid Kleckner 
3922214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
3932214ed89SReid Kleckner                                              FunctionInfo &FI) {
39470f5bc99SReid Kleckner   // For each function there is a separate subsection
39570f5bc99SReid Kleckner   // which holds the PC to file:line table.
39670f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
39770f5bc99SReid Kleckner   assert(Fn);
39870f5bc99SReid Kleckner 
39970f5bc99SReid Kleckner   StringRef FuncName;
40070f5bc99SReid Kleckner   if (auto *SP = getDISubprogram(GV))
40170f5bc99SReid Kleckner     FuncName = SP->getDisplayName();
40270f5bc99SReid Kleckner 
40370f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
40470f5bc99SReid Kleckner   if (FuncName.empty())
40570f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
40670f5bc99SReid Kleckner 
40770f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
408f9c275feSReid Kleckner   MCSymbol *SymbolsBegin = MMI->getContext().createTempSymbol(),
409f9c275feSReid Kleckner            *SymbolsEnd = MMI->getContext().createTempSymbol();
410dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
411dac21b43SReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::Symbols), 4);
412dac21b43SReid Kleckner   OS.AddComment("Subsection size");
413eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(SymbolsEnd, SymbolsBegin, 4);
414dac21b43SReid Kleckner   OS.EmitLabel(SymbolsBegin);
41570f5bc99SReid Kleckner   {
416f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
417f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
418dac21b43SReid Kleckner     OS.AddComment("Record length");
419eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
420dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
42170f5bc99SReid Kleckner 
422dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
423dac21b43SReid Kleckner     OS.EmitIntValue(unsigned(SymbolRecordKind::S_GPROC32_ID), 2);
42470f5bc99SReid Kleckner 
42530579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
426dac21b43SReid Kleckner     OS.AddComment("PtrParent");
427dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
428dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
429dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
430dac21b43SReid Kleckner     OS.AddComment("PtrNext");
431dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
43270f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
43370f5bc99SReid Kleckner     // code is located and what's its size:
434dac21b43SReid Kleckner     OS.AddComment("Code size");
435eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
436dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
437dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
438dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
439dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
440dac21b43SReid Kleckner     OS.AddComment("Function type index");
441dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
442dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
443dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
444dac21b43SReid Kleckner     OS.AddComment("Function section index");
445dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
446dac21b43SReid Kleckner     OS.AddComment("Flags");
447dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
44870f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
449dac21b43SReid Kleckner     OS.AddComment("Function name");
450f9c275feSReid Kleckner     emitNullTerminatedString(OS, FuncName);
451dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
45270f5bc99SReid Kleckner 
453f9c275feSReid Kleckner     for (const LocalVariable &Var : FI.Locals)
454f9c275feSReid Kleckner       emitLocalVariable(Var);
455f9c275feSReid Kleckner 
456f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
457f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
458f3b9ba49SReid Kleckner     // of their parent inline site.
459f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
460f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
461f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
462f9c275feSReid Kleckner              "child site not in function inline site map");
463f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
464f3b9ba49SReid Kleckner     }
465f3b9ba49SReid Kleckner 
46670f5bc99SReid Kleckner     // We're done with this function.
467dac21b43SReid Kleckner     OS.AddComment("Record length");
468dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
469dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
470dac21b43SReid Kleckner     OS.EmitIntValue(unsigned(SymbolRecordKind::S_PROC_ID_END), 2);
47170f5bc99SReid Kleckner   }
472dac21b43SReid Kleckner   OS.EmitLabel(SymbolsEnd);
47370f5bc99SReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
474dac21b43SReid Kleckner   OS.EmitValueToAlignment(4);
47570f5bc99SReid Kleckner 
4762214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
477dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
47870f5bc99SReid Kleckner }
47970f5bc99SReid Kleckner 
480f9c275feSReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable() {
481f9c275feSReid Kleckner   for (const auto &VI : MMI->getVariableDbgInfo()) {
482f9c275feSReid Kleckner     if (!VI.Var)
483f9c275feSReid Kleckner       continue;
484f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
485f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
486f9c275feSReid Kleckner 
487f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
488f9c275feSReid Kleckner 
489f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
490f9c275feSReid Kleckner     if (!Scope)
491f9c275feSReid Kleckner       continue;
492f9c275feSReid Kleckner 
493f9c275feSReid Kleckner     LocalVariable Var;
494f9c275feSReid Kleckner     Var.DIVar = VI.Var;
495f9c275feSReid Kleckner 
496f9c275feSReid Kleckner     // Get the frame register used and the offset.
497f9c275feSReid Kleckner     unsigned FrameReg = 0;
498f9c275feSReid Kleckner     const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
499f9c275feSReid Kleckner     const TargetFrameLowering *TFI = TSI.getFrameLowering();
500f9c275feSReid Kleckner     const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
501f9c275feSReid Kleckner     Var.RegisterOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
502f9c275feSReid Kleckner     Var.CVRegister = TRI->getCodeViewRegNum(FrameReg);
503f9c275feSReid Kleckner 
504f9c275feSReid Kleckner     // Calculate the label ranges.
505f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
506f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
507f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
508f9c275feSReid Kleckner       Var.Ranges.push_back({Begin, End});
509f9c275feSReid Kleckner     }
510f9c275feSReid Kleckner 
511f9c275feSReid Kleckner     if (VI.Loc->getInlinedAt()) {
512f9c275feSReid Kleckner       // This variable was inlined. Associate it with the InlineSite.
513f9c275feSReid Kleckner       InlineSite &Site = getInlineSite(VI.Loc);
514f9c275feSReid Kleckner       Site.InlinedLocals.emplace_back(std::move(Var));
515f9c275feSReid Kleckner     } else {
516f9c275feSReid Kleckner       // This variable goes in the main ProcSym.
517f9c275feSReid Kleckner       CurFn->Locals.emplace_back(std::move(Var));
518f9c275feSReid Kleckner     }
519f9c275feSReid Kleckner   }
520f9c275feSReid Kleckner }
521f9c275feSReid Kleckner 
52270f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
52370f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
52470f5bc99SReid Kleckner 
525f9c275feSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
52670f5bc99SReid Kleckner     return;
52770f5bc99SReid Kleckner 
528f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
529f9c275feSReid Kleckner 
53070f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
53170f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
53270f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
5332214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
5341fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
53570f5bc99SReid Kleckner 
536f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
537f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
53870f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
53970f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
54070f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
54170f5bc99SReid Kleckner   bool EmptyPrologue = true;
54270f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
54370f5bc99SReid Kleckner     for (const auto &MI : MBB) {
544f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
545f9c275feSReid Kleckner           MI.getDebugLoc()) {
54670f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
54770f5bc99SReid Kleckner         break;
548f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
54970f5bc99SReid Kleckner         EmptyPrologue = false;
55070f5bc99SReid Kleckner       }
55170f5bc99SReid Kleckner     }
552f9c275feSReid Kleckner   }
553f9c275feSReid Kleckner 
55470f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
55570f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
55670f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
55770f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
55870f5bc99SReid Kleckner   }
55970f5bc99SReid Kleckner }
56070f5bc99SReid Kleckner 
561f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
562f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
563f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
564f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
565f9c275feSReid Kleckner   OS.AddComment("Record length");
566f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
567f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
568f9c275feSReid Kleckner 
569f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
570f9c275feSReid Kleckner   OS.EmitIntValue(unsigned(SymbolRecordKind::S_LOCAL), 2);
571f9c275feSReid Kleckner 
572f9c275feSReid Kleckner   uint16_t Flags = 0;
573f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
574f9c275feSReid Kleckner     Flags |= LocalSym::IsParameter;
575f9c275feSReid Kleckner 
576f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
577f9c275feSReid Kleckner   OS.EmitIntValue(TypeIndex::Int32().getIndex(), 4);
578f9c275feSReid Kleckner   OS.AddComment("Flags");
579f9c275feSReid Kleckner   OS.EmitIntValue(Flags, 2);
580f9c275feSReid Kleckner   emitNullTerminatedString(OS, Var.DIVar->getName());
581f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
582f9c275feSReid Kleckner 
583f9c275feSReid Kleckner   // DefRangeRegisterRelSym record, see SymbolRecord.h for more info.  Omit the
584f9c275feSReid Kleckner   // LocalVariableAddrRange field from the record. The directive will emit that.
585f9c275feSReid Kleckner   DefRangeRegisterRelSym Sym{};
586f9c275feSReid Kleckner   ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
587f9c275feSReid Kleckner   Sym.BaseRegister = Var.CVRegister;
588f9c275feSReid Kleckner   Sym.Flags = 0; // Unclear what matters here.
589f9c275feSReid Kleckner   Sym.BasePointerOffset = Var.RegisterOffset;
590f9c275feSReid Kleckner   SmallString<sizeof(Sym) + sizeof(SymKind) - sizeof(LocalVariableAddrRange)>
591f9c275feSReid Kleckner       BytePrefix;
592f9c275feSReid Kleckner   BytePrefix += StringRef(reinterpret_cast<const char *>(&SymKind),
593f9c275feSReid Kleckner                           sizeof(SymKind));
594f9c275feSReid Kleckner   BytePrefix += StringRef(reinterpret_cast<const char *>(&Sym),
595f9c275feSReid Kleckner                           sizeof(Sym) - sizeof(LocalVariableAddrRange));
596f9c275feSReid Kleckner 
597f9c275feSReid Kleckner   OS.EmitCVDefRangeDirective(Var.Ranges, BytePrefix);
598f9c275feSReid Kleckner }
599f9c275feSReid Kleckner 
60070f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
601f9c275feSReid Kleckner   collectVariableInfoFromMMITable();
602f9c275feSReid Kleckner 
603f9c275feSReid Kleckner   DebugHandlerBase::endFunction(MF);
604f9c275feSReid Kleckner 
60570f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
60670f5bc99SReid Kleckner     return;
60770f5bc99SReid Kleckner 
60870f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
60970f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
61070f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
61170f5bc99SReid Kleckner 
6122214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
6132214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
61470f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
615f9c275feSReid Kleckner     CurFn = nullptr;
616f9c275feSReid Kleckner     return;
61770f5bc99SReid Kleckner   }
618f9c275feSReid Kleckner 
619f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
620f9c275feSReid Kleckner 
62170f5bc99SReid Kleckner   CurFn = nullptr;
62270f5bc99SReid Kleckner }
62370f5bc99SReid Kleckner 
62470f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
625f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
626f9c275feSReid Kleckner 
62770f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
62870f5bc99SReid Kleckner   if (!Asm || MI->isDebugValue() || MI->getFlag(MachineInstr::FrameSetup))
62970f5bc99SReid Kleckner     return;
63070f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
63170f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
63270f5bc99SReid Kleckner     return;
63370f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
63470f5bc99SReid Kleckner }
635