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"
18*fbdbe9e2SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeDumper.h"
19f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeIndex.h"
20f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeRecord.h"
2170f5bc99SReid Kleckner #include "llvm/MC/MCExpr.h"
225d122f87SReid Kleckner #include "llvm/MC/MCSectionCOFF.h"
2370f5bc99SReid Kleckner #include "llvm/MC/MCSymbol.h"
2470f5bc99SReid Kleckner #include "llvm/Support/COFF.h"
25*fbdbe9e2SReid Kleckner #include "llvm/Support/ScopedPrinter.h"
26f9c275feSReid Kleckner #include "llvm/Target/TargetSubtargetInfo.h"
27f9c275feSReid Kleckner #include "llvm/Target/TargetRegisterInfo.h"
28f9c275feSReid Kleckner #include "llvm/Target/TargetFrameLowering.h"
2970f5bc99SReid Kleckner 
30f9c275feSReid Kleckner using namespace llvm;
3170f5bc99SReid Kleckner using namespace llvm::codeview;
3270f5bc99SReid Kleckner 
33f9c275feSReid Kleckner CodeViewDebug::CodeViewDebug(AsmPrinter *AP)
34f9c275feSReid Kleckner     : DebugHandlerBase(AP), OS(*Asm->OutStreamer), CurFn(nullptr) {
35f9c275feSReid Kleckner   // If module doesn't have named metadata anchors or COFF debug section
36f9c275feSReid Kleckner   // is not available, skip any debug info related stuff.
37f9c275feSReid Kleckner   if (!MMI->getModule()->getNamedMetadata("llvm.dbg.cu") ||
38f9c275feSReid Kleckner       !AP->getObjFileLowering().getCOFFDebugSymbolsSection()) {
39f9c275feSReid Kleckner     Asm = nullptr;
40f9c275feSReid Kleckner     return;
41f9c275feSReid Kleckner   }
42f9c275feSReid Kleckner 
43f9c275feSReid Kleckner   // Tell MMI that we have debug info.
44f9c275feSReid Kleckner   MMI->setDebugInfoAvailability(true);
45f9c275feSReid Kleckner }
4670f5bc99SReid Kleckner 
479533af4fSReid Kleckner StringRef CodeViewDebug::getFullFilepath(const DIFile *File) {
489533af4fSReid Kleckner   std::string &Filepath = FileToFilepathMap[File];
4970f5bc99SReid Kleckner   if (!Filepath.empty())
5070f5bc99SReid Kleckner     return Filepath;
5170f5bc99SReid Kleckner 
529533af4fSReid Kleckner   StringRef Dir = File->getDirectory(), Filename = File->getFilename();
539533af4fSReid Kleckner 
5470f5bc99SReid Kleckner   // Clang emits directory and relative filename info into the IR, but CodeView
5570f5bc99SReid Kleckner   // operates on full paths.  We could change Clang to emit full paths too, but
5670f5bc99SReid Kleckner   // that would increase the IR size and probably not needed for other users.
5770f5bc99SReid Kleckner   // For now, just concatenate and canonicalize the path here.
5870f5bc99SReid Kleckner   if (Filename.find(':') == 1)
5970f5bc99SReid Kleckner     Filepath = Filename;
6070f5bc99SReid Kleckner   else
6170f5bc99SReid Kleckner     Filepath = (Dir + "\\" + Filename).str();
6270f5bc99SReid Kleckner 
6370f5bc99SReid Kleckner   // Canonicalize the path.  We have to do it textually because we may no longer
6470f5bc99SReid Kleckner   // have access the file in the filesystem.
6570f5bc99SReid Kleckner   // First, replace all slashes with backslashes.
6670f5bc99SReid Kleckner   std::replace(Filepath.begin(), Filepath.end(), '/', '\\');
6770f5bc99SReid Kleckner 
6870f5bc99SReid Kleckner   // Remove all "\.\" with "\".
6970f5bc99SReid Kleckner   size_t Cursor = 0;
7070f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\.\\", Cursor)) != std::string::npos)
7170f5bc99SReid Kleckner     Filepath.erase(Cursor, 2);
7270f5bc99SReid Kleckner 
7370f5bc99SReid Kleckner   // Replace all "\XXX\..\" with "\".  Don't try too hard though as the original
7470f5bc99SReid Kleckner   // path should be well-formatted, e.g. start with a drive letter, etc.
7570f5bc99SReid Kleckner   Cursor = 0;
7670f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\..\\", Cursor)) != std::string::npos) {
7770f5bc99SReid Kleckner     // Something's wrong if the path starts with "\..\", abort.
7870f5bc99SReid Kleckner     if (Cursor == 0)
7970f5bc99SReid Kleckner       break;
8070f5bc99SReid Kleckner 
8170f5bc99SReid Kleckner     size_t PrevSlash = Filepath.rfind('\\', Cursor - 1);
8270f5bc99SReid Kleckner     if (PrevSlash == std::string::npos)
8370f5bc99SReid Kleckner       // Something's wrong, abort.
8470f5bc99SReid Kleckner       break;
8570f5bc99SReid Kleckner 
8670f5bc99SReid Kleckner     Filepath.erase(PrevSlash, Cursor + 3 - PrevSlash);
8770f5bc99SReid Kleckner     // The next ".." might be following the one we've just erased.
8870f5bc99SReid Kleckner     Cursor = PrevSlash;
8970f5bc99SReid Kleckner   }
9070f5bc99SReid Kleckner 
9170f5bc99SReid Kleckner   // Remove all duplicate backslashes.
9270f5bc99SReid Kleckner   Cursor = 0;
9370f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\\\", Cursor)) != std::string::npos)
9470f5bc99SReid Kleckner     Filepath.erase(Cursor, 1);
9570f5bc99SReid Kleckner 
9670f5bc99SReid Kleckner   return Filepath;
9770f5bc99SReid Kleckner }
9870f5bc99SReid Kleckner 
992214ed89SReid Kleckner unsigned CodeViewDebug::maybeRecordFile(const DIFile *F) {
1002214ed89SReid Kleckner   unsigned NextId = FileIdMap.size() + 1;
1012214ed89SReid Kleckner   auto Insertion = FileIdMap.insert(std::make_pair(F, NextId));
1022214ed89SReid Kleckner   if (Insertion.second) {
1032214ed89SReid Kleckner     // We have to compute the full filepath and emit a .cv_file directive.
1042214ed89SReid Kleckner     StringRef FullPath = getFullFilepath(F);
105dac21b43SReid Kleckner     NextId = OS.EmitCVFileDirective(NextId, FullPath);
1062214ed89SReid Kleckner     assert(NextId == FileIdMap.size() && ".cv_file directive failed");
1072214ed89SReid Kleckner   }
1082214ed89SReid Kleckner   return Insertion.first->second;
1092214ed89SReid Kleckner }
1102214ed89SReid Kleckner 
111876330d5SReid Kleckner CodeViewDebug::InlineSite &
112876330d5SReid Kleckner CodeViewDebug::getInlineSite(const DILocation *InlinedAt,
113876330d5SReid Kleckner                              const DISubprogram *Inlinee) {
114fbd7787dSReid Kleckner   auto SiteInsertion = CurFn->InlineSites.insert({InlinedAt, InlineSite()});
115fbd7787dSReid Kleckner   InlineSite *Site = &SiteInsertion.first->second;
116fbd7787dSReid Kleckner   if (SiteInsertion.second) {
117f9c275feSReid Kleckner     Site->SiteFuncId = NextFuncId++;
118876330d5SReid Kleckner     Site->Inlinee = Inlinee;
1192280f932SReid Kleckner     InlinedSubprograms.insert(Inlinee);
1202280f932SReid Kleckner     recordFuncIdForSubprogram(Inlinee);
121f3b9ba49SReid Kleckner   }
122f9c275feSReid Kleckner   return *Site;
123f3b9ba49SReid Kleckner }
124f3b9ba49SReid Kleckner 
1252280f932SReid Kleckner TypeIndex CodeViewDebug::getGenericFunctionTypeIndex() {
1262280f932SReid Kleckner   if (VoidFnTyIdx.getIndex() != 0)
1272280f932SReid Kleckner     return VoidFnTyIdx;
1282280f932SReid Kleckner 
1292280f932SReid Kleckner   ArrayRef<TypeIndex> NoArgs;
1302280f932SReid Kleckner   ArgListRecord ArgListRec(TypeRecordKind::ArgList, NoArgs);
1312280f932SReid Kleckner   TypeIndex ArgListIndex = TypeTable.writeArgList(ArgListRec);
1322280f932SReid Kleckner 
1332280f932SReid Kleckner   ProcedureRecord Procedure(TypeIndex::Void(), CallingConvention::NearC,
1342280f932SReid Kleckner                             FunctionOptions::None, 0, ArgListIndex);
1352280f932SReid Kleckner   VoidFnTyIdx = TypeTable.writeProcedure(Procedure);
1362280f932SReid Kleckner   return VoidFnTyIdx;
1372280f932SReid Kleckner }
1382280f932SReid Kleckner 
1392280f932SReid Kleckner void CodeViewDebug::recordFuncIdForSubprogram(const DISubprogram *SP) {
1402280f932SReid Kleckner   TypeIndex ParentScope = TypeIndex(0);
1412280f932SReid Kleckner   StringRef DisplayName = SP->getDisplayName();
1422280f932SReid Kleckner   FuncIdRecord FuncId(ParentScope, getGenericFunctionTypeIndex(), DisplayName);
1432280f932SReid Kleckner   TypeIndex TI = TypeTable.writeFuncId(FuncId);
1442280f932SReid Kleckner   TypeIndices[SP] = TI;
1452280f932SReid Kleckner }
1462280f932SReid Kleckner 
147876330d5SReid Kleckner void CodeViewDebug::recordLocalVariable(LocalVariable &&Var,
148876330d5SReid Kleckner                                         const DILocation *InlinedAt) {
149876330d5SReid Kleckner   if (InlinedAt) {
150876330d5SReid Kleckner     // This variable was inlined. Associate it with the InlineSite.
151876330d5SReid Kleckner     const DISubprogram *Inlinee = Var.DIVar->getScope()->getSubprogram();
152876330d5SReid Kleckner     InlineSite &Site = getInlineSite(InlinedAt, Inlinee);
153876330d5SReid Kleckner     Site.InlinedLocals.emplace_back(Var);
154876330d5SReid Kleckner   } else {
155876330d5SReid Kleckner     // This variable goes in the main ProcSym.
156876330d5SReid Kleckner     CurFn->Locals.emplace_back(Var);
157876330d5SReid Kleckner   }
158876330d5SReid Kleckner }
159876330d5SReid Kleckner 
160829365aeSReid Kleckner static void addLocIfNotPresent(SmallVectorImpl<const DILocation *> &Locs,
161829365aeSReid Kleckner                                const DILocation *Loc) {
162829365aeSReid Kleckner   auto B = Locs.begin(), E = Locs.end();
163829365aeSReid Kleckner   if (std::find(B, E, Loc) == E)
164829365aeSReid Kleckner     Locs.push_back(Loc);
165829365aeSReid Kleckner }
166829365aeSReid Kleckner 
16770f5bc99SReid Kleckner void CodeViewDebug::maybeRecordLocation(DebugLoc DL,
16870f5bc99SReid Kleckner                                         const MachineFunction *MF) {
1699533af4fSReid Kleckner   // Skip this instruction if it has the same location as the previous one.
1709533af4fSReid Kleckner   if (DL == CurFn->LastLoc)
1719533af4fSReid Kleckner     return;
1729533af4fSReid Kleckner 
1739533af4fSReid Kleckner   const DIScope *Scope = DL.get()->getScope();
17470f5bc99SReid Kleckner   if (!Scope)
17570f5bc99SReid Kleckner     return;
1769533af4fSReid Kleckner 
17770f5bc99SReid Kleckner   // Skip this line if it is longer than the maximum we can record.
1782214ed89SReid Kleckner   LineInfo LI(DL.getLine(), DL.getLine(), /*IsStatement=*/true);
1792214ed89SReid Kleckner   if (LI.getStartLine() != DL.getLine() || LI.isAlwaysStepInto() ||
1802214ed89SReid Kleckner       LI.isNeverStepInto())
18170f5bc99SReid Kleckner     return;
18270f5bc99SReid Kleckner 
1832214ed89SReid Kleckner   ColumnInfo CI(DL.getCol(), /*EndColumn=*/0);
1842214ed89SReid Kleckner   if (CI.getStartColumn() != DL.getCol())
1852214ed89SReid Kleckner     return;
18600d9639cSReid Kleckner 
1872214ed89SReid Kleckner   if (!CurFn->HaveLineInfo)
1882214ed89SReid Kleckner     CurFn->HaveLineInfo = true;
1892214ed89SReid Kleckner   unsigned FileId = 0;
1902214ed89SReid Kleckner   if (CurFn->LastLoc.get() && CurFn->LastLoc->getFile() == DL->getFile())
1912214ed89SReid Kleckner     FileId = CurFn->LastFileId;
1922214ed89SReid Kleckner   else
1932214ed89SReid Kleckner     FileId = CurFn->LastFileId = maybeRecordFile(DL->getFile());
1942214ed89SReid Kleckner   CurFn->LastLoc = DL;
195f3b9ba49SReid Kleckner 
196f3b9ba49SReid Kleckner   unsigned FuncId = CurFn->FuncId;
197876330d5SReid Kleckner   if (const DILocation *SiteLoc = DL->getInlinedAt()) {
198829365aeSReid Kleckner     const DILocation *Loc = DL.get();
199829365aeSReid Kleckner 
200f3b9ba49SReid Kleckner     // If this location was actually inlined from somewhere else, give it the ID
201f3b9ba49SReid Kleckner     // of the inline call site.
202876330d5SReid Kleckner     FuncId =
203876330d5SReid Kleckner         getInlineSite(SiteLoc, Loc->getScope()->getSubprogram()).SiteFuncId;
204829365aeSReid Kleckner 
205f3b9ba49SReid Kleckner     // Ensure we have links in the tree of inline call sites.
206829365aeSReid Kleckner     bool FirstLoc = true;
207829365aeSReid Kleckner     while ((SiteLoc = Loc->getInlinedAt())) {
208876330d5SReid Kleckner       InlineSite &Site =
209876330d5SReid Kleckner           getInlineSite(SiteLoc, Loc->getScope()->getSubprogram());
210829365aeSReid Kleckner       if (!FirstLoc)
211829365aeSReid Kleckner         addLocIfNotPresent(Site.ChildSites, Loc);
212829365aeSReid Kleckner       FirstLoc = false;
213829365aeSReid Kleckner       Loc = SiteLoc;
214f3b9ba49SReid Kleckner     }
215829365aeSReid Kleckner     addLocIfNotPresent(CurFn->ChildSites, Loc);
216f3b9ba49SReid Kleckner   }
217f3b9ba49SReid Kleckner 
218dac21b43SReid Kleckner   OS.EmitCVLocDirective(FuncId, FileId, DL.getLine(), DL.getCol(),
219dac21b43SReid Kleckner                         /*PrologueEnd=*/false,
2202214ed89SReid Kleckner                         /*IsStmt=*/false, DL->getFilename());
22170f5bc99SReid Kleckner }
22270f5bc99SReid Kleckner 
2235d122f87SReid Kleckner void CodeViewDebug::emitCodeViewMagicVersion() {
2245d122f87SReid Kleckner   OS.EmitValueToAlignment(4);
2255d122f87SReid Kleckner   OS.AddComment("Debug section magic");
2265d122f87SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
2275d122f87SReid Kleckner }
2285d122f87SReid Kleckner 
22970f5bc99SReid Kleckner void CodeViewDebug::endModule() {
23070f5bc99SReid Kleckner   if (FnDebugInfo.empty())
23170f5bc99SReid Kleckner     return;
23270f5bc99SReid Kleckner 
233f3b9ba49SReid Kleckner   emitTypeInformation();
234f3b9ba49SReid Kleckner 
23570f5bc99SReid Kleckner   assert(Asm != nullptr);
23670f5bc99SReid Kleckner 
23770f5bc99SReid Kleckner   // The COFF .debug$S section consists of several subsections, each starting
23870f5bc99SReid Kleckner   // with a 4-byte control code (e.g. 0xF1, 0xF2, etc) and then a 4-byte length
23970f5bc99SReid Kleckner   // of the payload followed by the payload itself.  The subsections are 4-byte
24070f5bc99SReid Kleckner   // aligned.
24170f5bc99SReid Kleckner 
2421fcd610cSReid Kleckner   // Make a subsection for all the inlined subprograms.
2435d122f87SReid Kleckner   emitInlineeLinesSubsection();
2441fcd610cSReid Kleckner 
2452214ed89SReid Kleckner   // Emit per-function debug information.
2462214ed89SReid Kleckner   for (auto &P : FnDebugInfo)
2472214ed89SReid Kleckner     emitDebugInfoForFunction(P.first, P.second);
24870f5bc99SReid Kleckner 
2495d122f87SReid Kleckner   // Switch back to the generic .debug$S section after potentially processing
2505d122f87SReid Kleckner   // comdat symbol sections.
2515d122f87SReid Kleckner   switchToDebugSectionForSymbol(nullptr);
2525d122f87SReid Kleckner 
25370f5bc99SReid Kleckner   // This subsection holds a file index to offset in string table table.
254dac21b43SReid Kleckner   OS.AddComment("File index to string table offset subsection");
255dac21b43SReid Kleckner   OS.EmitCVFileChecksumsDirective();
25670f5bc99SReid Kleckner 
25770f5bc99SReid Kleckner   // This subsection holds the string table.
258dac21b43SReid Kleckner   OS.AddComment("String table");
259dac21b43SReid Kleckner   OS.EmitCVStringTableDirective();
26070f5bc99SReid Kleckner 
26170f5bc99SReid Kleckner   clear();
26270f5bc99SReid Kleckner }
26370f5bc99SReid Kleckner 
264b9456a5eSDavid Majnemer static void emitNullTerminatedSymbolName(MCStreamer &OS, StringRef S) {
265b9456a5eSDavid Majnemer   // Microsoft's linker seems to have trouble with symbol names longer than
266b9456a5eSDavid Majnemer   // 0xffd8 bytes.
267b9456a5eSDavid Majnemer   S = S.substr(0, 0xffd8);
268b9456a5eSDavid Majnemer   SmallString<32> NullTerminatedString(S);
269b9456a5eSDavid Majnemer   NullTerminatedString.push_back('\0');
270b9456a5eSDavid Majnemer   OS.EmitBytes(NullTerminatedString);
271b9456a5eSDavid Majnemer }
272b9456a5eSDavid Majnemer 
273f3b9ba49SReid Kleckner void CodeViewDebug::emitTypeInformation() {
2742280f932SReid Kleckner   // Do nothing if we have no debug info or if no non-trivial types were emitted
2752280f932SReid Kleckner   // to TypeTable during codegen.
276fbd7787dSReid Kleckner   NamedMDNode *CU_Nodes =
277fbd7787dSReid Kleckner       MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
278fbd7787dSReid Kleckner   if (!CU_Nodes)
279fbd7787dSReid Kleckner     return;
2802280f932SReid Kleckner   if (TypeTable.empty())
281fbd7787dSReid Kleckner     return;
282fbd7787dSReid Kleckner 
283f3b9ba49SReid Kleckner   // Start the .debug$T section with 0x4.
284dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugTypesSection());
2855d122f87SReid Kleckner   emitCodeViewMagicVersion();
286f3b9ba49SReid Kleckner 
287*fbdbe9e2SReid Kleckner   SmallString<8> CommentPrefix;
288*fbdbe9e2SReid Kleckner   if (OS.isVerboseAsm()) {
289*fbdbe9e2SReid Kleckner     CommentPrefix += '\t';
290*fbdbe9e2SReid Kleckner     CommentPrefix += Asm->MAI->getCommentString();
291*fbdbe9e2SReid Kleckner     CommentPrefix += ' ';
292*fbdbe9e2SReid Kleckner   }
293*fbdbe9e2SReid Kleckner 
294*fbdbe9e2SReid Kleckner   CVTypeDumper CVTD(nullptr, /*PrintRecordBytes=*/false);
2952280f932SReid Kleckner   TypeTable.ForEachRecord(
296*fbdbe9e2SReid Kleckner       [&](TypeIndex Index, StringRef Record) {
297*fbdbe9e2SReid Kleckner         if (OS.isVerboseAsm()) {
298*fbdbe9e2SReid Kleckner           // Emit a block comment describing the type record for readability.
299*fbdbe9e2SReid Kleckner           SmallString<512> CommentBlock;
300*fbdbe9e2SReid Kleckner           raw_svector_ostream CommentOS(CommentBlock);
301*fbdbe9e2SReid Kleckner           ScopedPrinter SP(CommentOS);
302*fbdbe9e2SReid Kleckner           SP.setPrefix(CommentPrefix);
303*fbdbe9e2SReid Kleckner           CVTD.setPrinter(&SP);
304*fbdbe9e2SReid Kleckner           bool DumpSuccess =
305*fbdbe9e2SReid Kleckner               CVTD.dump({Record.bytes_begin(), Record.bytes_end()});
306*fbdbe9e2SReid Kleckner           (void)DumpSuccess;
307*fbdbe9e2SReid Kleckner           assert(DumpSuccess && "produced malformed type record");
308*fbdbe9e2SReid Kleckner           // emitRawComment will insert its own tab and comment string before
309*fbdbe9e2SReid Kleckner           // the first line, so strip off our first one. It also prints its own
310*fbdbe9e2SReid Kleckner           // newline.
311*fbdbe9e2SReid Kleckner           OS.emitRawComment(
312*fbdbe9e2SReid Kleckner               CommentOS.str().drop_front(CommentPrefix.size() - 1).rtrim());
313*fbdbe9e2SReid Kleckner         }
314*fbdbe9e2SReid Kleckner         OS.EmitBinaryData(Record);
3152280f932SReid Kleckner       });
316f3b9ba49SReid Kleckner }
317f3b9ba49SReid Kleckner 
3185d122f87SReid Kleckner void CodeViewDebug::emitInlineeLinesSubsection() {
3191fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
3201fcd610cSReid Kleckner     return;
3211fcd610cSReid Kleckner 
3225d122f87SReid Kleckner   // Use the generic .debug$S section.
3235d122f87SReid Kleckner   switchToDebugSectionForSymbol(nullptr);
3245d122f87SReid Kleckner 
325f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
326f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
3271fcd610cSReid Kleckner 
3281fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
3291fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::InlineeLines), 4);
33030579ec8SDavid Majnemer   OS.AddComment("Subsection size");
3311fcd610cSReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 4);
3321fcd610cSReid Kleckner   OS.EmitLabel(InlineBegin);
3331fcd610cSReid Kleckner 
3341fcd610cSReid Kleckner   // We don't provide any extra file info.
3351fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
33630579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
3371fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
3381fcd610cSReid Kleckner 
3391fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
3402280f932SReid Kleckner     assert(TypeIndices.count(SP));
3412280f932SReid Kleckner     TypeIndex InlineeIdx = TypeIndices[SP];
3422280f932SReid Kleckner 
34330579ec8SDavid Majnemer     OS.AddBlankLine();
3441fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
3451fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
3461fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
34730579ec8SDavid Majnemer     OS.AddBlankLine();
3481fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
3491fcd610cSReid Kleckner     // 1.
3501fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
35130579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
3522280f932SReid Kleckner     OS.EmitIntValue(InlineeIdx.getIndex(), 4);
35330579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
3541fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
35530579ec8SDavid Majnemer     OS.AddComment("Starting line number");
3561fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
3571fcd610cSReid Kleckner   }
3581fcd610cSReid Kleckner 
3591fcd610cSReid Kleckner   OS.EmitLabel(InlineEnd);
3601fcd610cSReid Kleckner }
3611fcd610cSReid Kleckner 
3621fcd610cSReid Kleckner void CodeViewDebug::collectInlineSiteChildren(
3631fcd610cSReid Kleckner     SmallVectorImpl<unsigned> &Children, const FunctionInfo &FI,
3641fcd610cSReid Kleckner     const InlineSite &Site) {
3651fcd610cSReid Kleckner   for (const DILocation *ChildSiteLoc : Site.ChildSites) {
3661fcd610cSReid Kleckner     auto I = FI.InlineSites.find(ChildSiteLoc);
3671fcd610cSReid Kleckner     const InlineSite &ChildSite = I->second;
3681fcd610cSReid Kleckner     Children.push_back(ChildSite.SiteFuncId);
3691fcd610cSReid Kleckner     collectInlineSiteChildren(Children, FI, ChildSite);
3701fcd610cSReid Kleckner   }
3711fcd610cSReid Kleckner }
3721fcd610cSReid Kleckner 
373f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
374f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
375f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
376f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
377f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
378f3b9ba49SReid Kleckner 
3792280f932SReid Kleckner   assert(TypeIndices.count(Site.Inlinee));
3802280f932SReid Kleckner   TypeIndex InlineeIdx = TypeIndices[Site.Inlinee];
381f3b9ba49SReid Kleckner 
382f3b9ba49SReid Kleckner   // SymbolRecord
383dac21b43SReid Kleckner   OS.AddComment("Record length");
384eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
385f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
386dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
38763a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE, 2); // RecordKind
388f3b9ba49SReid Kleckner 
389dac21b43SReid Kleckner   OS.AddComment("PtrParent");
390dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
391dac21b43SReid Kleckner   OS.AddComment("PtrEnd");
392dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
393dac21b43SReid Kleckner   OS.AddComment("Inlinee type index");
3942280f932SReid Kleckner   OS.EmitIntValue(InlineeIdx.getIndex(), 4);
395f3b9ba49SReid Kleckner 
3961fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
3971fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
3981fcd610cSReid Kleckner   SmallVector<unsigned, 3> SecondaryFuncIds;
3991fcd610cSReid Kleckner   collectInlineSiteChildren(SecondaryFuncIds, FI, Site);
4001fcd610cSReid Kleckner 
4011fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
402c9911f28SDavid Majnemer                                     FI.Begin, FI.End, SecondaryFuncIds);
403f3b9ba49SReid Kleckner 
404f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
405f3b9ba49SReid Kleckner 
406f9c275feSReid Kleckner   for (const LocalVariable &Var : Site.InlinedLocals)
407f9c275feSReid Kleckner     emitLocalVariable(Var);
408f9c275feSReid Kleckner 
409f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
410f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
411f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
412f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
413f3b9ba49SReid Kleckner            "child site not in function inline site map");
414f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
415f3b9ba49SReid Kleckner   }
416f3b9ba49SReid Kleckner 
417f3b9ba49SReid Kleckner   // Close the scope.
418dac21b43SReid Kleckner   OS.AddComment("Record length");
419dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
420dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
42163a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE_END, 2); // RecordKind
422f3b9ba49SReid Kleckner }
423f3b9ba49SReid Kleckner 
4245d122f87SReid Kleckner void CodeViewDebug::switchToDebugSectionForSymbol(const MCSymbol *GVSym) {
4255d122f87SReid Kleckner   // If we have a symbol, it may be in a section that is COMDAT. If so, find the
4265d122f87SReid Kleckner   // comdat key. A section may be comdat because of -ffunction-sections or
4275d122f87SReid Kleckner   // because it is comdat in the IR.
4285d122f87SReid Kleckner   MCSectionCOFF *GVSec =
4295d122f87SReid Kleckner       GVSym ? dyn_cast<MCSectionCOFF>(&GVSym->getSection()) : nullptr;
4305d122f87SReid Kleckner   const MCSymbol *KeySym = GVSec ? GVSec->getCOMDATSymbol() : nullptr;
4315d122f87SReid Kleckner 
4325d122f87SReid Kleckner   MCSectionCOFF *DebugSec = cast<MCSectionCOFF>(
4335d122f87SReid Kleckner       Asm->getObjFileLowering().getCOFFDebugSymbolsSection());
4345d122f87SReid Kleckner   DebugSec = OS.getContext().getAssociativeCOFFSection(DebugSec, KeySym);
4355d122f87SReid Kleckner 
4365d122f87SReid Kleckner   OS.SwitchSection(DebugSec);
4375d122f87SReid Kleckner 
4385d122f87SReid Kleckner   // Emit the magic version number if this is the first time we've switched to
4395d122f87SReid Kleckner   // this section.
4405d122f87SReid Kleckner   if (ComdatDebugSections.insert(DebugSec).second)
4415d122f87SReid Kleckner     emitCodeViewMagicVersion();
4425d122f87SReid Kleckner }
4435d122f87SReid Kleckner 
4442214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
4452214ed89SReid Kleckner                                              FunctionInfo &FI) {
44670f5bc99SReid Kleckner   // For each function there is a separate subsection
44770f5bc99SReid Kleckner   // which holds the PC to file:line table.
44870f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
44970f5bc99SReid Kleckner   assert(Fn);
45070f5bc99SReid Kleckner 
4515d122f87SReid Kleckner   // Switch to the to a comdat section, if appropriate.
4525d122f87SReid Kleckner   switchToDebugSectionForSymbol(Fn);
4535d122f87SReid Kleckner 
45470f5bc99SReid Kleckner   StringRef FuncName;
455adebb937SPete Cooper   if (auto *SP = GV->getSubprogram())
45670f5bc99SReid Kleckner     FuncName = SP->getDisplayName();
45770f5bc99SReid Kleckner 
45870f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
45970f5bc99SReid Kleckner   if (FuncName.empty())
46070f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
46170f5bc99SReid Kleckner 
46270f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
463f9c275feSReid Kleckner   MCSymbol *SymbolsBegin = MMI->getContext().createTempSymbol(),
464f9c275feSReid Kleckner            *SymbolsEnd = MMI->getContext().createTempSymbol();
465dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
466dac21b43SReid Kleckner   OS.EmitIntValue(unsigned(ModuleSubstreamKind::Symbols), 4);
467dac21b43SReid Kleckner   OS.AddComment("Subsection size");
468eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(SymbolsEnd, SymbolsBegin, 4);
469dac21b43SReid Kleckner   OS.EmitLabel(SymbolsBegin);
47070f5bc99SReid Kleckner   {
471f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
472f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
473dac21b43SReid Kleckner     OS.AddComment("Record length");
474eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
475dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
47670f5bc99SReid Kleckner 
477dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
47863a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_GPROC32_ID), 2);
47970f5bc99SReid Kleckner 
48030579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
481dac21b43SReid Kleckner     OS.AddComment("PtrParent");
482dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
483dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
484dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
485dac21b43SReid Kleckner     OS.AddComment("PtrNext");
486dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
48770f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
48870f5bc99SReid Kleckner     // code is located and what's its size:
489dac21b43SReid Kleckner     OS.AddComment("Code size");
490eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
491dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
492dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
493dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
494dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
495dac21b43SReid Kleckner     OS.AddComment("Function type index");
496dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
497dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
498dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
499dac21b43SReid Kleckner     OS.AddComment("Function section index");
500dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
501dac21b43SReid Kleckner     OS.AddComment("Flags");
502dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
50370f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
504dac21b43SReid Kleckner     OS.AddComment("Function name");
5051256125fSDavid Majnemer     // Truncate the name so we won't overflow the record length field.
506b9456a5eSDavid Majnemer     emitNullTerminatedSymbolName(OS, FuncName);
507dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
50870f5bc99SReid Kleckner 
509f9c275feSReid Kleckner     for (const LocalVariable &Var : FI.Locals)
510f9c275feSReid Kleckner       emitLocalVariable(Var);
511f9c275feSReid Kleckner 
512f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
513f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
514f3b9ba49SReid Kleckner     // of their parent inline site.
515f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
516f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
517f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
518f9c275feSReid Kleckner              "child site not in function inline site map");
519f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
520f3b9ba49SReid Kleckner     }
521f3b9ba49SReid Kleckner 
52270f5bc99SReid Kleckner     // We're done with this function.
523dac21b43SReid Kleckner     OS.AddComment("Record length");
524dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
525dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
52663a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_PROC_ID_END), 2);
52770f5bc99SReid Kleckner   }
528dac21b43SReid Kleckner   OS.EmitLabel(SymbolsEnd);
52970f5bc99SReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
530dac21b43SReid Kleckner   OS.EmitValueToAlignment(4);
53170f5bc99SReid Kleckner 
5322214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
533dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
53470f5bc99SReid Kleckner }
53570f5bc99SReid Kleckner 
536876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
537876330d5SReid Kleckner CodeViewDebug::createDefRangeMem(uint16_t CVRegister, int Offset) {
538876330d5SReid Kleckner   LocalVarDefRange DR;
539c6a2f214SAaron Ballman   DR.InMemory = -1;
540876330d5SReid Kleckner   DR.DataOffset = Offset;
541876330d5SReid Kleckner   assert(DR.DataOffset == Offset && "truncation");
542876330d5SReid Kleckner   DR.StructOffset = 0;
543876330d5SReid Kleckner   DR.CVRegister = CVRegister;
544876330d5SReid Kleckner   return DR;
545876330d5SReid Kleckner }
546876330d5SReid Kleckner 
547876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
548876330d5SReid Kleckner CodeViewDebug::createDefRangeReg(uint16_t CVRegister) {
549876330d5SReid Kleckner   LocalVarDefRange DR;
550876330d5SReid Kleckner   DR.InMemory = 0;
551876330d5SReid Kleckner   DR.DataOffset = 0;
552876330d5SReid Kleckner   DR.StructOffset = 0;
553876330d5SReid Kleckner   DR.CVRegister = CVRegister;
554876330d5SReid Kleckner   return DR;
555876330d5SReid Kleckner }
556876330d5SReid Kleckner 
557876330d5SReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable(
558876330d5SReid Kleckner     DenseSet<InlinedVariable> &Processed) {
559876330d5SReid Kleckner   const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
560876330d5SReid Kleckner   const TargetFrameLowering *TFI = TSI.getFrameLowering();
561876330d5SReid Kleckner   const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
562876330d5SReid Kleckner 
563876330d5SReid Kleckner   for (const MachineModuleInfo::VariableDbgInfo &VI :
564876330d5SReid Kleckner        MMI->getVariableDbgInfo()) {
565f9c275feSReid Kleckner     if (!VI.Var)
566f9c275feSReid Kleckner       continue;
567f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
568f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
569f9c275feSReid Kleckner 
570876330d5SReid Kleckner     Processed.insert(InlinedVariable(VI.Var, VI.Loc->getInlinedAt()));
571f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
572f9c275feSReid Kleckner 
573f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
574f9c275feSReid Kleckner     if (!Scope)
575f9c275feSReid Kleckner       continue;
576f9c275feSReid Kleckner 
577f9c275feSReid Kleckner     // Get the frame register used and the offset.
578f9c275feSReid Kleckner     unsigned FrameReg = 0;
579876330d5SReid Kleckner     int FrameOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
580876330d5SReid Kleckner     uint16_t CVReg = TRI->getCodeViewRegNum(FrameReg);
581f9c275feSReid Kleckner 
582f9c275feSReid Kleckner     // Calculate the label ranges.
583876330d5SReid Kleckner     LocalVarDefRange DefRange = createDefRangeMem(CVReg, FrameOffset);
584f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
585f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
586f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
587876330d5SReid Kleckner       End = End ? End : Asm->getFunctionEnd();
588876330d5SReid Kleckner       DefRange.Ranges.emplace_back(Begin, End);
589f9c275feSReid Kleckner     }
590f9c275feSReid Kleckner 
591876330d5SReid Kleckner     LocalVariable Var;
592876330d5SReid Kleckner     Var.DIVar = VI.Var;
593876330d5SReid Kleckner     Var.DefRanges.emplace_back(std::move(DefRange));
594876330d5SReid Kleckner     recordLocalVariable(std::move(Var), VI.Loc->getInlinedAt());
595f9c275feSReid Kleckner   }
596f9c275feSReid Kleckner }
597876330d5SReid Kleckner 
598876330d5SReid Kleckner void CodeViewDebug::collectVariableInfo(const DISubprogram *SP) {
599876330d5SReid Kleckner   DenseSet<InlinedVariable> Processed;
600876330d5SReid Kleckner   // Grab the variable info that was squirreled away in the MMI side-table.
601876330d5SReid Kleckner   collectVariableInfoFromMMITable(Processed);
602876330d5SReid Kleckner 
603876330d5SReid Kleckner   const TargetRegisterInfo *TRI = Asm->MF->getSubtarget().getRegisterInfo();
604876330d5SReid Kleckner 
605876330d5SReid Kleckner   for (const auto &I : DbgValues) {
606876330d5SReid Kleckner     InlinedVariable IV = I.first;
607876330d5SReid Kleckner     if (Processed.count(IV))
608876330d5SReid Kleckner       continue;
609876330d5SReid Kleckner     const DILocalVariable *DIVar = IV.first;
610876330d5SReid Kleckner     const DILocation *InlinedAt = IV.second;
611876330d5SReid Kleckner 
612876330d5SReid Kleckner     // Instruction ranges, specifying where IV is accessible.
613876330d5SReid Kleckner     const auto &Ranges = I.second;
614876330d5SReid Kleckner 
615876330d5SReid Kleckner     LexicalScope *Scope = nullptr;
616876330d5SReid Kleckner     if (InlinedAt)
617876330d5SReid Kleckner       Scope = LScopes.findInlinedScope(DIVar->getScope(), InlinedAt);
618876330d5SReid Kleckner     else
619876330d5SReid Kleckner       Scope = LScopes.findLexicalScope(DIVar->getScope());
620876330d5SReid Kleckner     // If variable scope is not found then skip this variable.
621876330d5SReid Kleckner     if (!Scope)
622876330d5SReid Kleckner       continue;
623876330d5SReid Kleckner 
624876330d5SReid Kleckner     LocalVariable Var;
625876330d5SReid Kleckner     Var.DIVar = DIVar;
626876330d5SReid Kleckner 
627876330d5SReid Kleckner     // Calculate the definition ranges.
628876330d5SReid Kleckner     for (auto I = Ranges.begin(), E = Ranges.end(); I != E; ++I) {
629876330d5SReid Kleckner       const InsnRange &Range = *I;
630876330d5SReid Kleckner       const MachineInstr *DVInst = Range.first;
631876330d5SReid Kleckner       assert(DVInst->isDebugValue() && "Invalid History entry");
632876330d5SReid Kleckner       const DIExpression *DIExpr = DVInst->getDebugExpression();
633876330d5SReid Kleckner 
634876330d5SReid Kleckner       // Bail if there is a complex DWARF expression for now.
635876330d5SReid Kleckner       if (DIExpr && DIExpr->getNumElements() > 0)
636876330d5SReid Kleckner         continue;
637876330d5SReid Kleckner 
6389a593ee7SReid Kleckner       // Bail if operand 0 is not a valid register. This means the variable is a
6399a593ee7SReid Kleckner       // simple constant, or is described by a complex expression.
6409a593ee7SReid Kleckner       // FIXME: Find a way to represent constant variables, since they are
6419a593ee7SReid Kleckner       // relatively common.
6429a593ee7SReid Kleckner       unsigned Reg =
6439a593ee7SReid Kleckner           DVInst->getOperand(0).isReg() ? DVInst->getOperand(0).getReg() : 0;
6449a593ee7SReid Kleckner       if (Reg == 0)
6456e0d5f57SReid Kleckner         continue;
6466e0d5f57SReid Kleckner 
647876330d5SReid Kleckner       // Handle the two cases we can handle: indirect in memory and in register.
648876330d5SReid Kleckner       bool IsIndirect = DVInst->getOperand(1).isImm();
649876330d5SReid Kleckner       unsigned CVReg = TRI->getCodeViewRegNum(DVInst->getOperand(0).getReg());
650876330d5SReid Kleckner       {
651876330d5SReid Kleckner         LocalVarDefRange DefRange;
652876330d5SReid Kleckner         if (IsIndirect) {
653876330d5SReid Kleckner           int64_t Offset = DVInst->getOperand(1).getImm();
654876330d5SReid Kleckner           DefRange = createDefRangeMem(CVReg, Offset);
655876330d5SReid Kleckner         } else {
656876330d5SReid Kleckner           DefRange = createDefRangeReg(CVReg);
657876330d5SReid Kleckner         }
658876330d5SReid Kleckner         if (Var.DefRanges.empty() ||
659876330d5SReid Kleckner             Var.DefRanges.back().isDifferentLocation(DefRange)) {
660876330d5SReid Kleckner           Var.DefRanges.emplace_back(std::move(DefRange));
661876330d5SReid Kleckner         }
662876330d5SReid Kleckner       }
663876330d5SReid Kleckner 
664876330d5SReid Kleckner       // Compute the label range.
665876330d5SReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
666876330d5SReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
667876330d5SReid Kleckner       if (!End) {
668876330d5SReid Kleckner         if (std::next(I) != E)
669876330d5SReid Kleckner           End = getLabelBeforeInsn(std::next(I)->first);
670876330d5SReid Kleckner         else
671876330d5SReid Kleckner           End = Asm->getFunctionEnd();
672876330d5SReid Kleckner       }
673876330d5SReid Kleckner 
674876330d5SReid Kleckner       // If the last range end is our begin, just extend the last range.
675876330d5SReid Kleckner       // Otherwise make a new range.
676876330d5SReid Kleckner       SmallVectorImpl<std::pair<const MCSymbol *, const MCSymbol *>> &Ranges =
677876330d5SReid Kleckner           Var.DefRanges.back().Ranges;
678876330d5SReid Kleckner       if (!Ranges.empty() && Ranges.back().second == Begin)
679876330d5SReid Kleckner         Ranges.back().second = End;
680876330d5SReid Kleckner       else
681876330d5SReid Kleckner         Ranges.emplace_back(Begin, End);
682876330d5SReid Kleckner 
683876330d5SReid Kleckner       // FIXME: Do more range combining.
684876330d5SReid Kleckner     }
685876330d5SReid Kleckner 
686876330d5SReid Kleckner     recordLocalVariable(std::move(Var), InlinedAt);
687876330d5SReid Kleckner   }
688f9c275feSReid Kleckner }
689f9c275feSReid Kleckner 
69070f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
69170f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
69270f5bc99SReid Kleckner 
693f9c275feSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
69470f5bc99SReid Kleckner     return;
69570f5bc99SReid Kleckner 
696f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
697f9c275feSReid Kleckner 
69870f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
69970f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
70070f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
7012214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
7021fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
70370f5bc99SReid Kleckner 
704f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
705f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
70670f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
70770f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
70870f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
70970f5bc99SReid Kleckner   bool EmptyPrologue = true;
71070f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
71170f5bc99SReid Kleckner     for (const auto &MI : MBB) {
712f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
713f9c275feSReid Kleckner           MI.getDebugLoc()) {
71470f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
71570f5bc99SReid Kleckner         break;
716f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
71770f5bc99SReid Kleckner         EmptyPrologue = false;
71870f5bc99SReid Kleckner       }
71970f5bc99SReid Kleckner     }
720f9c275feSReid Kleckner   }
721f9c275feSReid Kleckner 
72270f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
72370f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
72470f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
72570f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
72670f5bc99SReid Kleckner   }
72770f5bc99SReid Kleckner }
72870f5bc99SReid Kleckner 
729f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
730f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
731f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
732f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
733f9c275feSReid Kleckner   OS.AddComment("Record length");
734f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
735f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
736f9c275feSReid Kleckner 
737f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
73863a2846eSZachary Turner   OS.EmitIntValue(unsigned(SymbolKind::S_LOCAL), 2);
739f9c275feSReid Kleckner 
74063a2846eSZachary Turner   LocalSymFlags Flags = LocalSymFlags::None;
741f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
74263a2846eSZachary Turner     Flags |= LocalSymFlags::IsParameter;
743876330d5SReid Kleckner   if (Var.DefRanges.empty())
74463a2846eSZachary Turner     Flags |= LocalSymFlags::IsOptimizedOut;
745f9c275feSReid Kleckner 
746f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
747f9c275feSReid Kleckner   OS.EmitIntValue(TypeIndex::Int32().getIndex(), 4);
748f9c275feSReid Kleckner   OS.AddComment("Flags");
74963a2846eSZachary Turner   OS.EmitIntValue(static_cast<uint16_t>(Flags), 2);
7501256125fSDavid Majnemer   // Truncate the name so we won't overflow the record length field.
751b9456a5eSDavid Majnemer   emitNullTerminatedSymbolName(OS, Var.DIVar->getName());
752f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
753f9c275feSReid Kleckner 
754876330d5SReid Kleckner   // Calculate the on disk prefix of the appropriate def range record. The
755876330d5SReid Kleckner   // records and on disk formats are described in SymbolRecords.h. BytePrefix
756876330d5SReid Kleckner   // should be big enough to hold all forms without memory allocation.
757876330d5SReid Kleckner   SmallString<20> BytePrefix;
758876330d5SReid Kleckner   for (const LocalVarDefRange &DefRange : Var.DefRanges) {
759876330d5SReid Kleckner     BytePrefix.clear();
760876330d5SReid Kleckner     // FIXME: Handle bitpieces.
761876330d5SReid Kleckner     if (DefRange.StructOffset != 0)
762876330d5SReid Kleckner       continue;
763876330d5SReid Kleckner 
764876330d5SReid Kleckner     if (DefRange.InMemory) {
765a78ecd1eSZachary Turner       DefRangeRegisterRelSym Sym(DefRange.CVRegister, 0, DefRange.DataOffset, 0,
766a78ecd1eSZachary Turner                                  0, 0, ArrayRef<LocalVariableAddrGap>());
767f9c275feSReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
768876330d5SReid Kleckner       BytePrefix +=
769876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
770a78ecd1eSZachary Turner       BytePrefix +=
771a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
772a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
773876330d5SReid Kleckner     } else {
774876330d5SReid Kleckner       assert(DefRange.DataOffset == 0 && "unexpected offset into register");
775a78ecd1eSZachary Turner       // Unclear what matters here.
776a78ecd1eSZachary Turner       DefRangeRegisterSym Sym(DefRange.CVRegister, 0, 0, 0, 0,
777a78ecd1eSZachary Turner                               ArrayRef<LocalVariableAddrGap>());
778876330d5SReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER);
779876330d5SReid Kleckner       BytePrefix +=
780876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
781a78ecd1eSZachary Turner       BytePrefix +=
782a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
783a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
784876330d5SReid Kleckner     }
785876330d5SReid Kleckner     OS.EmitCVDefRangeDirective(DefRange.Ranges, BytePrefix);
786876330d5SReid Kleckner   }
787f9c275feSReid Kleckner }
788f9c275feSReid Kleckner 
78970f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
79070f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
79170f5bc99SReid Kleckner     return;
79270f5bc99SReid Kleckner 
79370f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
79470f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
79570f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
79670f5bc99SReid Kleckner 
797adebb937SPete Cooper   collectVariableInfo(GV->getSubprogram());
798876330d5SReid Kleckner 
799876330d5SReid Kleckner   DebugHandlerBase::endFunction(MF);
800876330d5SReid Kleckner 
8012214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
8022214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
80370f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
804f9c275feSReid Kleckner     CurFn = nullptr;
805f9c275feSReid Kleckner     return;
80670f5bc99SReid Kleckner   }
807f9c275feSReid Kleckner 
808f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
809f9c275feSReid Kleckner 
81070f5bc99SReid Kleckner   CurFn = nullptr;
81170f5bc99SReid Kleckner }
81270f5bc99SReid Kleckner 
81370f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
814f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
815f9c275feSReid Kleckner 
81670f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
81770f5bc99SReid Kleckner   if (!Asm || MI->isDebugValue() || MI->getFlag(MachineInstr::FrameSetup))
81870f5bc99SReid Kleckner     return;
81970f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
82070f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
82170f5bc99SReid Kleckner     return;
82270f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
82370f5bc99SReid Kleckner }
824