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"
16a8d57407SReid Kleckner #include "llvm/DebugInfo/CodeView/FieldListRecordBuilder.h"
172214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/Line.h"
1870f5bc99SReid Kleckner #include "llvm/DebugInfo/CodeView/SymbolRecord.h"
19fbdbe9e2SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeDumper.h"
20f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeIndex.h"
21f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeRecord.h"
2270f5bc99SReid Kleckner #include "llvm/MC/MCExpr.h"
235d122f87SReid Kleckner #include "llvm/MC/MCSectionCOFF.h"
2470f5bc99SReid Kleckner #include "llvm/MC/MCSymbol.h"
2570f5bc99SReid Kleckner #include "llvm/Support/COFF.h"
26fbdbe9e2SReid Kleckner #include "llvm/Support/ScopedPrinter.h"
27f9c275feSReid Kleckner #include "llvm/Target/TargetSubtargetInfo.h"
28f9c275feSReid Kleckner #include "llvm/Target/TargetRegisterInfo.h"
29f9c275feSReid Kleckner #include "llvm/Target/TargetFrameLowering.h"
3070f5bc99SReid Kleckner 
31f9c275feSReid Kleckner using namespace llvm;
3270f5bc99SReid Kleckner using namespace llvm::codeview;
3370f5bc99SReid Kleckner 
34f9c275feSReid Kleckner CodeViewDebug::CodeViewDebug(AsmPrinter *AP)
35f9c275feSReid Kleckner     : DebugHandlerBase(AP), OS(*Asm->OutStreamer), CurFn(nullptr) {
36f9c275feSReid Kleckner   // If module doesn't have named metadata anchors or COFF debug section
37f9c275feSReid Kleckner   // is not available, skip any debug info related stuff.
38f9c275feSReid Kleckner   if (!MMI->getModule()->getNamedMetadata("llvm.dbg.cu") ||
39f9c275feSReid Kleckner       !AP->getObjFileLowering().getCOFFDebugSymbolsSection()) {
40f9c275feSReid Kleckner     Asm = nullptr;
41f9c275feSReid Kleckner     return;
42f9c275feSReid Kleckner   }
43f9c275feSReid Kleckner 
44f9c275feSReid Kleckner   // Tell MMI that we have debug info.
45f9c275feSReid Kleckner   MMI->setDebugInfoAvailability(true);
46f9c275feSReid Kleckner }
4770f5bc99SReid Kleckner 
489533af4fSReid Kleckner StringRef CodeViewDebug::getFullFilepath(const DIFile *File) {
499533af4fSReid Kleckner   std::string &Filepath = FileToFilepathMap[File];
5070f5bc99SReid Kleckner   if (!Filepath.empty())
5170f5bc99SReid Kleckner     return Filepath;
5270f5bc99SReid Kleckner 
539533af4fSReid Kleckner   StringRef Dir = File->getDirectory(), Filename = File->getFilename();
549533af4fSReid Kleckner 
5570f5bc99SReid Kleckner   // Clang emits directory and relative filename info into the IR, but CodeView
5670f5bc99SReid Kleckner   // operates on full paths.  We could change Clang to emit full paths too, but
5770f5bc99SReid Kleckner   // that would increase the IR size and probably not needed for other users.
5870f5bc99SReid Kleckner   // For now, just concatenate and canonicalize the path here.
5970f5bc99SReid Kleckner   if (Filename.find(':') == 1)
6070f5bc99SReid Kleckner     Filepath = Filename;
6170f5bc99SReid Kleckner   else
6270f5bc99SReid Kleckner     Filepath = (Dir + "\\" + Filename).str();
6370f5bc99SReid Kleckner 
6470f5bc99SReid Kleckner   // Canonicalize the path.  We have to do it textually because we may no longer
6570f5bc99SReid Kleckner   // have access the file in the filesystem.
6670f5bc99SReid Kleckner   // First, replace all slashes with backslashes.
6770f5bc99SReid Kleckner   std::replace(Filepath.begin(), Filepath.end(), '/', '\\');
6870f5bc99SReid Kleckner 
6970f5bc99SReid Kleckner   // Remove all "\.\" with "\".
7070f5bc99SReid Kleckner   size_t Cursor = 0;
7170f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\.\\", Cursor)) != std::string::npos)
7270f5bc99SReid Kleckner     Filepath.erase(Cursor, 2);
7370f5bc99SReid Kleckner 
7470f5bc99SReid Kleckner   // Replace all "\XXX\..\" with "\".  Don't try too hard though as the original
7570f5bc99SReid Kleckner   // path should be well-formatted, e.g. start with a drive letter, etc.
7670f5bc99SReid Kleckner   Cursor = 0;
7770f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\..\\", Cursor)) != std::string::npos) {
7870f5bc99SReid Kleckner     // Something's wrong if the path starts with "\..\", abort.
7970f5bc99SReid Kleckner     if (Cursor == 0)
8070f5bc99SReid Kleckner       break;
8170f5bc99SReid Kleckner 
8270f5bc99SReid Kleckner     size_t PrevSlash = Filepath.rfind('\\', Cursor - 1);
8370f5bc99SReid Kleckner     if (PrevSlash == std::string::npos)
8470f5bc99SReid Kleckner       // Something's wrong, abort.
8570f5bc99SReid Kleckner       break;
8670f5bc99SReid Kleckner 
8770f5bc99SReid Kleckner     Filepath.erase(PrevSlash, Cursor + 3 - PrevSlash);
8870f5bc99SReid Kleckner     // The next ".." might be following the one we've just erased.
8970f5bc99SReid Kleckner     Cursor = PrevSlash;
9070f5bc99SReid Kleckner   }
9170f5bc99SReid Kleckner 
9270f5bc99SReid Kleckner   // Remove all duplicate backslashes.
9370f5bc99SReid Kleckner   Cursor = 0;
9470f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\\\", Cursor)) != std::string::npos)
9570f5bc99SReid Kleckner     Filepath.erase(Cursor, 1);
9670f5bc99SReid Kleckner 
9770f5bc99SReid Kleckner   return Filepath;
9870f5bc99SReid Kleckner }
9970f5bc99SReid Kleckner 
1002214ed89SReid Kleckner unsigned CodeViewDebug::maybeRecordFile(const DIFile *F) {
1012214ed89SReid Kleckner   unsigned NextId = FileIdMap.size() + 1;
1022214ed89SReid Kleckner   auto Insertion = FileIdMap.insert(std::make_pair(F, NextId));
1032214ed89SReid Kleckner   if (Insertion.second) {
1042214ed89SReid Kleckner     // We have to compute the full filepath and emit a .cv_file directive.
1052214ed89SReid Kleckner     StringRef FullPath = getFullFilepath(F);
106dac21b43SReid Kleckner     NextId = OS.EmitCVFileDirective(NextId, FullPath);
1072214ed89SReid Kleckner     assert(NextId == FileIdMap.size() && ".cv_file directive failed");
1082214ed89SReid Kleckner   }
1092214ed89SReid Kleckner   return Insertion.first->second;
1102214ed89SReid Kleckner }
1112214ed89SReid Kleckner 
112876330d5SReid Kleckner CodeViewDebug::InlineSite &
113876330d5SReid Kleckner CodeViewDebug::getInlineSite(const DILocation *InlinedAt,
114876330d5SReid Kleckner                              const DISubprogram *Inlinee) {
115fbd7787dSReid Kleckner   auto SiteInsertion = CurFn->InlineSites.insert({InlinedAt, InlineSite()});
116fbd7787dSReid Kleckner   InlineSite *Site = &SiteInsertion.first->second;
117fbd7787dSReid Kleckner   if (SiteInsertion.second) {
118f9c275feSReid Kleckner     Site->SiteFuncId = NextFuncId++;
119876330d5SReid Kleckner     Site->Inlinee = Inlinee;
1202280f932SReid Kleckner     InlinedSubprograms.insert(Inlinee);
12175c3ebfaSDavid Majnemer     getFuncIdForSubprogram(Inlinee);
122f3b9ba49SReid Kleckner   }
123f9c275feSReid Kleckner   return *Site;
124f3b9ba49SReid Kleckner }
125f3b9ba49SReid Kleckner 
12675c3ebfaSDavid Majnemer TypeIndex CodeViewDebug::getFuncIdForSubprogram(const DISubprogram *SP) {
12775c3ebfaSDavid Majnemer   // It's possible to ask for the FuncId of a function which doesn't have a
12875c3ebfaSDavid Majnemer   // subprogram: inlining a function with debug info into a function with none.
12975c3ebfaSDavid Majnemer   if (!SP)
130b68f32f0SDavid Majnemer     return TypeIndex::None();
1312280f932SReid Kleckner 
13275c3ebfaSDavid Majnemer   // Check if we've already translated this subprogram.
13375c3ebfaSDavid Majnemer   auto I = TypeIndices.find(SP);
13475c3ebfaSDavid Majnemer   if (I != TypeIndices.end())
13575c3ebfaSDavid Majnemer     return I->second;
1362280f932SReid Kleckner 
1372280f932SReid Kleckner   TypeIndex ParentScope = TypeIndex(0);
1382280f932SReid Kleckner   StringRef DisplayName = SP->getDisplayName();
13975c3ebfaSDavid Majnemer   FuncIdRecord FuncId(ParentScope, getTypeIndex(SP->getType()), DisplayName);
1402280f932SReid Kleckner   TypeIndex TI = TypeTable.writeFuncId(FuncId);
14175c3ebfaSDavid Majnemer 
142a8d57407SReid Kleckner   recordTypeIndexForDINode(SP, TI);
14375c3ebfaSDavid Majnemer   return TI;
1442280f932SReid Kleckner }
1452280f932SReid Kleckner 
146a8d57407SReid Kleckner void CodeViewDebug::recordTypeIndexForDINode(const DINode *Node, TypeIndex TI) {
147a8d57407SReid Kleckner   auto InsertResult = TypeIndices.insert({Node, TI});
148a8d57407SReid Kleckner   (void)InsertResult;
149a8d57407SReid Kleckner   assert(InsertResult.second && "DINode was already assigned a type index");
150a8d57407SReid Kleckner }
151a8d57407SReid Kleckner 
152876330d5SReid Kleckner void CodeViewDebug::recordLocalVariable(LocalVariable &&Var,
153876330d5SReid Kleckner                                         const DILocation *InlinedAt) {
154876330d5SReid Kleckner   if (InlinedAt) {
155876330d5SReid Kleckner     // This variable was inlined. Associate it with the InlineSite.
156876330d5SReid Kleckner     const DISubprogram *Inlinee = Var.DIVar->getScope()->getSubprogram();
157876330d5SReid Kleckner     InlineSite &Site = getInlineSite(InlinedAt, Inlinee);
158876330d5SReid Kleckner     Site.InlinedLocals.emplace_back(Var);
159876330d5SReid Kleckner   } else {
160876330d5SReid Kleckner     // This variable goes in the main ProcSym.
161876330d5SReid Kleckner     CurFn->Locals.emplace_back(Var);
162876330d5SReid Kleckner   }
163876330d5SReid Kleckner }
164876330d5SReid Kleckner 
165829365aeSReid Kleckner static void addLocIfNotPresent(SmallVectorImpl<const DILocation *> &Locs,
166829365aeSReid Kleckner                                const DILocation *Loc) {
167829365aeSReid Kleckner   auto B = Locs.begin(), E = Locs.end();
168829365aeSReid Kleckner   if (std::find(B, E, Loc) == E)
169829365aeSReid Kleckner     Locs.push_back(Loc);
170829365aeSReid Kleckner }
171829365aeSReid Kleckner 
172bdc4956bSBenjamin Kramer void CodeViewDebug::maybeRecordLocation(const DebugLoc &DL,
17370f5bc99SReid Kleckner                                         const MachineFunction *MF) {
1749533af4fSReid Kleckner   // Skip this instruction if it has the same location as the previous one.
1759533af4fSReid Kleckner   if (DL == CurFn->LastLoc)
1769533af4fSReid Kleckner     return;
1779533af4fSReid Kleckner 
1789533af4fSReid Kleckner   const DIScope *Scope = DL.get()->getScope();
17970f5bc99SReid Kleckner   if (!Scope)
18070f5bc99SReid Kleckner     return;
1819533af4fSReid Kleckner 
18270f5bc99SReid Kleckner   // Skip this line if it is longer than the maximum we can record.
1832214ed89SReid Kleckner   LineInfo LI(DL.getLine(), DL.getLine(), /*IsStatement=*/true);
1842214ed89SReid Kleckner   if (LI.getStartLine() != DL.getLine() || LI.isAlwaysStepInto() ||
1852214ed89SReid Kleckner       LI.isNeverStepInto())
18670f5bc99SReid Kleckner     return;
18770f5bc99SReid Kleckner 
1882214ed89SReid Kleckner   ColumnInfo CI(DL.getCol(), /*EndColumn=*/0);
1892214ed89SReid Kleckner   if (CI.getStartColumn() != DL.getCol())
1902214ed89SReid Kleckner     return;
19100d9639cSReid Kleckner 
1922214ed89SReid Kleckner   if (!CurFn->HaveLineInfo)
1932214ed89SReid Kleckner     CurFn->HaveLineInfo = true;
1942214ed89SReid Kleckner   unsigned FileId = 0;
1952214ed89SReid Kleckner   if (CurFn->LastLoc.get() && CurFn->LastLoc->getFile() == DL->getFile())
1962214ed89SReid Kleckner     FileId = CurFn->LastFileId;
1972214ed89SReid Kleckner   else
1982214ed89SReid Kleckner     FileId = CurFn->LastFileId = maybeRecordFile(DL->getFile());
1992214ed89SReid Kleckner   CurFn->LastLoc = DL;
200f3b9ba49SReid Kleckner 
201f3b9ba49SReid Kleckner   unsigned FuncId = CurFn->FuncId;
202876330d5SReid Kleckner   if (const DILocation *SiteLoc = DL->getInlinedAt()) {
203829365aeSReid Kleckner     const DILocation *Loc = DL.get();
204829365aeSReid Kleckner 
205f3b9ba49SReid Kleckner     // If this location was actually inlined from somewhere else, give it the ID
206f3b9ba49SReid Kleckner     // of the inline call site.
207876330d5SReid Kleckner     FuncId =
208876330d5SReid Kleckner         getInlineSite(SiteLoc, Loc->getScope()->getSubprogram()).SiteFuncId;
209829365aeSReid Kleckner 
210f3b9ba49SReid Kleckner     // Ensure we have links in the tree of inline call sites.
211829365aeSReid Kleckner     bool FirstLoc = true;
212829365aeSReid Kleckner     while ((SiteLoc = Loc->getInlinedAt())) {
213876330d5SReid Kleckner       InlineSite &Site =
214876330d5SReid Kleckner           getInlineSite(SiteLoc, Loc->getScope()->getSubprogram());
215829365aeSReid Kleckner       if (!FirstLoc)
216829365aeSReid Kleckner         addLocIfNotPresent(Site.ChildSites, Loc);
217829365aeSReid Kleckner       FirstLoc = false;
218829365aeSReid Kleckner       Loc = SiteLoc;
219f3b9ba49SReid Kleckner     }
220829365aeSReid Kleckner     addLocIfNotPresent(CurFn->ChildSites, Loc);
221f3b9ba49SReid Kleckner   }
222f3b9ba49SReid Kleckner 
223dac21b43SReid Kleckner   OS.EmitCVLocDirective(FuncId, FileId, DL.getLine(), DL.getCol(),
224dac21b43SReid Kleckner                         /*PrologueEnd=*/false,
2252214ed89SReid Kleckner                         /*IsStmt=*/false, DL->getFilename());
22670f5bc99SReid Kleckner }
22770f5bc99SReid Kleckner 
2285d122f87SReid Kleckner void CodeViewDebug::emitCodeViewMagicVersion() {
2295d122f87SReid Kleckner   OS.EmitValueToAlignment(4);
2305d122f87SReid Kleckner   OS.AddComment("Debug section magic");
2315d122f87SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
2325d122f87SReid Kleckner }
2335d122f87SReid Kleckner 
23470f5bc99SReid Kleckner void CodeViewDebug::endModule() {
2356f3406dfSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
23670f5bc99SReid Kleckner     return;
23770f5bc99SReid Kleckner 
23870f5bc99SReid Kleckner   assert(Asm != nullptr);
23970f5bc99SReid Kleckner 
24070f5bc99SReid Kleckner   // The COFF .debug$S section consists of several subsections, each starting
24170f5bc99SReid Kleckner   // with a 4-byte control code (e.g. 0xF1, 0xF2, etc) and then a 4-byte length
24270f5bc99SReid Kleckner   // of the payload followed by the payload itself.  The subsections are 4-byte
24370f5bc99SReid Kleckner   // aligned.
24470f5bc99SReid Kleckner 
2456f3406dfSReid Kleckner   // Use the generic .debug$S section, and make a subsection for all the inlined
2466f3406dfSReid Kleckner   // subprograms.
2476f3406dfSReid Kleckner   switchToDebugSectionForSymbol(nullptr);
2485d122f87SReid Kleckner   emitInlineeLinesSubsection();
2491fcd610cSReid Kleckner 
2502214ed89SReid Kleckner   // Emit per-function debug information.
2512214ed89SReid Kleckner   for (auto &P : FnDebugInfo)
252577be0feSDavid Majnemer     if (!P.first->isDeclarationForLinker())
2532214ed89SReid Kleckner       emitDebugInfoForFunction(P.first, P.second);
25470f5bc99SReid Kleckner 
2556f3406dfSReid Kleckner   // Emit global variable debug information.
256*3128b10cSDavid Majnemer   setCurrentSubprogram(nullptr);
2576f3406dfSReid Kleckner   emitDebugInfoForGlobals();
2586f3406dfSReid Kleckner 
2595d122f87SReid Kleckner   // Switch back to the generic .debug$S section after potentially processing
2605d122f87SReid Kleckner   // comdat symbol sections.
2615d122f87SReid Kleckner   switchToDebugSectionForSymbol(nullptr);
2625d122f87SReid Kleckner 
263*3128b10cSDavid Majnemer   // Emit UDT records for any types used by global variables.
264*3128b10cSDavid Majnemer   if (!GlobalUDTs.empty()) {
265*3128b10cSDavid Majnemer     MCSymbol *SymbolsEnd = beginCVSubsection(ModuleSubstreamKind::Symbols);
266*3128b10cSDavid Majnemer     emitDebugInfoForUDTs(GlobalUDTs);
267*3128b10cSDavid Majnemer     endCVSubsection(SymbolsEnd);
268*3128b10cSDavid Majnemer   }
269*3128b10cSDavid Majnemer 
27070f5bc99SReid Kleckner   // This subsection holds a file index to offset in string table table.
271dac21b43SReid Kleckner   OS.AddComment("File index to string table offset subsection");
272dac21b43SReid Kleckner   OS.EmitCVFileChecksumsDirective();
27370f5bc99SReid Kleckner 
27470f5bc99SReid Kleckner   // This subsection holds the string table.
275dac21b43SReid Kleckner   OS.AddComment("String table");
276dac21b43SReid Kleckner   OS.EmitCVStringTableDirective();
27770f5bc99SReid Kleckner 
2785acacbb0SReid Kleckner   // Emit type information last, so that any types we translate while emitting
2795acacbb0SReid Kleckner   // function info are included.
2805acacbb0SReid Kleckner   emitTypeInformation();
2815acacbb0SReid Kleckner 
28270f5bc99SReid Kleckner   clear();
28370f5bc99SReid Kleckner }
28470f5bc99SReid Kleckner 
285b9456a5eSDavid Majnemer static void emitNullTerminatedSymbolName(MCStreamer &OS, StringRef S) {
286b9456a5eSDavid Majnemer   // Microsoft's linker seems to have trouble with symbol names longer than
287b9456a5eSDavid Majnemer   // 0xffd8 bytes.
288b9456a5eSDavid Majnemer   S = S.substr(0, 0xffd8);
289b9456a5eSDavid Majnemer   SmallString<32> NullTerminatedString(S);
290b9456a5eSDavid Majnemer   NullTerminatedString.push_back('\0');
291b9456a5eSDavid Majnemer   OS.EmitBytes(NullTerminatedString);
292b9456a5eSDavid Majnemer }
293b9456a5eSDavid Majnemer 
294f3b9ba49SReid Kleckner void CodeViewDebug::emitTypeInformation() {
2952280f932SReid Kleckner   // Do nothing if we have no debug info or if no non-trivial types were emitted
2962280f932SReid Kleckner   // to TypeTable during codegen.
297fbd7787dSReid Kleckner   NamedMDNode *CU_Nodes =
298fbd7787dSReid Kleckner       MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
299fbd7787dSReid Kleckner   if (!CU_Nodes)
300fbd7787dSReid Kleckner     return;
3012280f932SReid Kleckner   if (TypeTable.empty())
302fbd7787dSReid Kleckner     return;
303fbd7787dSReid Kleckner 
304f3b9ba49SReid Kleckner   // Start the .debug$T section with 0x4.
305dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugTypesSection());
3065d122f87SReid Kleckner   emitCodeViewMagicVersion();
307f3b9ba49SReid Kleckner 
308fbdbe9e2SReid Kleckner   SmallString<8> CommentPrefix;
309fbdbe9e2SReid Kleckner   if (OS.isVerboseAsm()) {
310fbdbe9e2SReid Kleckner     CommentPrefix += '\t';
311fbdbe9e2SReid Kleckner     CommentPrefix += Asm->MAI->getCommentString();
312fbdbe9e2SReid Kleckner     CommentPrefix += ' ';
313fbdbe9e2SReid Kleckner   }
314fbdbe9e2SReid Kleckner 
315fbdbe9e2SReid Kleckner   CVTypeDumper CVTD(nullptr, /*PrintRecordBytes=*/false);
3162280f932SReid Kleckner   TypeTable.ForEachRecord(
317fbdbe9e2SReid Kleckner       [&](TypeIndex Index, StringRef Record) {
318fbdbe9e2SReid Kleckner         if (OS.isVerboseAsm()) {
319fbdbe9e2SReid Kleckner           // Emit a block comment describing the type record for readability.
320fbdbe9e2SReid Kleckner           SmallString<512> CommentBlock;
321fbdbe9e2SReid Kleckner           raw_svector_ostream CommentOS(CommentBlock);
322fbdbe9e2SReid Kleckner           ScopedPrinter SP(CommentOS);
323fbdbe9e2SReid Kleckner           SP.setPrefix(CommentPrefix);
324fbdbe9e2SReid Kleckner           CVTD.setPrinter(&SP);
325fbdbe9e2SReid Kleckner           bool DumpSuccess =
326fbdbe9e2SReid Kleckner               CVTD.dump({Record.bytes_begin(), Record.bytes_end()});
327fbdbe9e2SReid Kleckner           (void)DumpSuccess;
328fbdbe9e2SReid Kleckner           assert(DumpSuccess && "produced malformed type record");
329fbdbe9e2SReid Kleckner           // emitRawComment will insert its own tab and comment string before
330fbdbe9e2SReid Kleckner           // the first line, so strip off our first one. It also prints its own
331fbdbe9e2SReid Kleckner           // newline.
332fbdbe9e2SReid Kleckner           OS.emitRawComment(
333fbdbe9e2SReid Kleckner               CommentOS.str().drop_front(CommentPrefix.size() - 1).rtrim());
334fbdbe9e2SReid Kleckner         }
335fbdbe9e2SReid Kleckner         OS.EmitBinaryData(Record);
3362280f932SReid Kleckner       });
337f3b9ba49SReid Kleckner }
338f3b9ba49SReid Kleckner 
3395d122f87SReid Kleckner void CodeViewDebug::emitInlineeLinesSubsection() {
3401fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
3411fcd610cSReid Kleckner     return;
3421fcd610cSReid Kleckner 
3431fcd610cSReid Kleckner 
3441fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
3456f3406dfSReid Kleckner   MCSymbol *InlineEnd = beginCVSubsection(ModuleSubstreamKind::InlineeLines);
3461fcd610cSReid Kleckner 
3471fcd610cSReid Kleckner   // We don't provide any extra file info.
3481fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
34930579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
3501fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
3511fcd610cSReid Kleckner 
3521fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
3532280f932SReid Kleckner     assert(TypeIndices.count(SP));
3542280f932SReid Kleckner     TypeIndex InlineeIdx = TypeIndices[SP];
3552280f932SReid Kleckner 
35630579ec8SDavid Majnemer     OS.AddBlankLine();
3571fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
3581fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
3591fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
36030579ec8SDavid Majnemer     OS.AddBlankLine();
3611fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
3621fcd610cSReid Kleckner     // 1.
3631fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
36430579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
3652280f932SReid Kleckner     OS.EmitIntValue(InlineeIdx.getIndex(), 4);
36630579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
3671fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
36830579ec8SDavid Majnemer     OS.AddComment("Starting line number");
3691fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
3701fcd610cSReid Kleckner   }
3711fcd610cSReid Kleckner 
3726f3406dfSReid Kleckner   endCVSubsection(InlineEnd);
3731fcd610cSReid Kleckner }
3741fcd610cSReid Kleckner 
3751fcd610cSReid Kleckner void CodeViewDebug::collectInlineSiteChildren(
3761fcd610cSReid Kleckner     SmallVectorImpl<unsigned> &Children, const FunctionInfo &FI,
3771fcd610cSReid Kleckner     const InlineSite &Site) {
3781fcd610cSReid Kleckner   for (const DILocation *ChildSiteLoc : Site.ChildSites) {
3791fcd610cSReid Kleckner     auto I = FI.InlineSites.find(ChildSiteLoc);
3801fcd610cSReid Kleckner     const InlineSite &ChildSite = I->second;
3811fcd610cSReid Kleckner     Children.push_back(ChildSite.SiteFuncId);
3821fcd610cSReid Kleckner     collectInlineSiteChildren(Children, FI, ChildSite);
3831fcd610cSReid Kleckner   }
3841fcd610cSReid Kleckner }
3851fcd610cSReid Kleckner 
386f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
387f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
388f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
389f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
390f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
391f3b9ba49SReid Kleckner 
3922280f932SReid Kleckner   assert(TypeIndices.count(Site.Inlinee));
3932280f932SReid Kleckner   TypeIndex InlineeIdx = TypeIndices[Site.Inlinee];
394f3b9ba49SReid Kleckner 
395f3b9ba49SReid Kleckner   // SymbolRecord
396dac21b43SReid Kleckner   OS.AddComment("Record length");
397eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
398f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
399dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
40063a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE, 2); // RecordKind
401f3b9ba49SReid Kleckner 
402dac21b43SReid Kleckner   OS.AddComment("PtrParent");
403dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
404dac21b43SReid Kleckner   OS.AddComment("PtrEnd");
405dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
406dac21b43SReid Kleckner   OS.AddComment("Inlinee type index");
4072280f932SReid Kleckner   OS.EmitIntValue(InlineeIdx.getIndex(), 4);
408f3b9ba49SReid Kleckner 
4091fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
4101fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
4111fcd610cSReid Kleckner   SmallVector<unsigned, 3> SecondaryFuncIds;
4121fcd610cSReid Kleckner   collectInlineSiteChildren(SecondaryFuncIds, FI, Site);
4131fcd610cSReid Kleckner 
4141fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
415c9911f28SDavid Majnemer                                     FI.Begin, FI.End, SecondaryFuncIds);
416f3b9ba49SReid Kleckner 
417f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
418f3b9ba49SReid Kleckner 
419f9c275feSReid Kleckner   for (const LocalVariable &Var : Site.InlinedLocals)
420f9c275feSReid Kleckner     emitLocalVariable(Var);
421f9c275feSReid Kleckner 
422f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
423f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
424f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
425f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
426f3b9ba49SReid Kleckner            "child site not in function inline site map");
427f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
428f3b9ba49SReid Kleckner   }
429f3b9ba49SReid Kleckner 
430f3b9ba49SReid Kleckner   // Close the scope.
431dac21b43SReid Kleckner   OS.AddComment("Record length");
432dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
433dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
43463a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE_END, 2); // RecordKind
435f3b9ba49SReid Kleckner }
436f3b9ba49SReid Kleckner 
4375d122f87SReid Kleckner void CodeViewDebug::switchToDebugSectionForSymbol(const MCSymbol *GVSym) {
4385d122f87SReid Kleckner   // If we have a symbol, it may be in a section that is COMDAT. If so, find the
4395d122f87SReid Kleckner   // comdat key. A section may be comdat because of -ffunction-sections or
4405d122f87SReid Kleckner   // because it is comdat in the IR.
4415d122f87SReid Kleckner   MCSectionCOFF *GVSec =
4425d122f87SReid Kleckner       GVSym ? dyn_cast<MCSectionCOFF>(&GVSym->getSection()) : nullptr;
4435d122f87SReid Kleckner   const MCSymbol *KeySym = GVSec ? GVSec->getCOMDATSymbol() : nullptr;
4445d122f87SReid Kleckner 
4455d122f87SReid Kleckner   MCSectionCOFF *DebugSec = cast<MCSectionCOFF>(
4465d122f87SReid Kleckner       Asm->getObjFileLowering().getCOFFDebugSymbolsSection());
4475d122f87SReid Kleckner   DebugSec = OS.getContext().getAssociativeCOFFSection(DebugSec, KeySym);
4485d122f87SReid Kleckner 
4495d122f87SReid Kleckner   OS.SwitchSection(DebugSec);
4505d122f87SReid Kleckner 
4515d122f87SReid Kleckner   // Emit the magic version number if this is the first time we've switched to
4525d122f87SReid Kleckner   // this section.
4535d122f87SReid Kleckner   if (ComdatDebugSections.insert(DebugSec).second)
4545d122f87SReid Kleckner     emitCodeViewMagicVersion();
4555d122f87SReid Kleckner }
4565d122f87SReid Kleckner 
4572214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
4582214ed89SReid Kleckner                                              FunctionInfo &FI) {
45970f5bc99SReid Kleckner   // For each function there is a separate subsection
46070f5bc99SReid Kleckner   // which holds the PC to file:line table.
46170f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
46270f5bc99SReid Kleckner   assert(Fn);
46370f5bc99SReid Kleckner 
4645d122f87SReid Kleckner   // Switch to the to a comdat section, if appropriate.
4655d122f87SReid Kleckner   switchToDebugSectionForSymbol(Fn);
4665d122f87SReid Kleckner 
46770f5bc99SReid Kleckner   StringRef FuncName;
468*3128b10cSDavid Majnemer   auto *SP = GV->getSubprogram();
469*3128b10cSDavid Majnemer   setCurrentSubprogram(SP);
470*3128b10cSDavid Majnemer   if (SP != nullptr)
47170f5bc99SReid Kleckner     FuncName = SP->getDisplayName();
47270f5bc99SReid Kleckner 
47370f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
47470f5bc99SReid Kleckner   if (FuncName.empty())
47570f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
47670f5bc99SReid Kleckner 
47770f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
478dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
4796f3406dfSReid Kleckner   MCSymbol *SymbolsEnd = beginCVSubsection(ModuleSubstreamKind::Symbols);
48070f5bc99SReid Kleckner   {
481f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
482f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
483dac21b43SReid Kleckner     OS.AddComment("Record length");
484eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
485dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
48670f5bc99SReid Kleckner 
487dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
48863a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_GPROC32_ID), 2);
48970f5bc99SReid Kleckner 
49030579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
491dac21b43SReid Kleckner     OS.AddComment("PtrParent");
492dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
493dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
494dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
495dac21b43SReid Kleckner     OS.AddComment("PtrNext");
496dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
49770f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
49870f5bc99SReid Kleckner     // code is located and what's its size:
499dac21b43SReid Kleckner     OS.AddComment("Code size");
500eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
501dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
502dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
503dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
504dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
505dac21b43SReid Kleckner     OS.AddComment("Function type index");
50675c3ebfaSDavid Majnemer     OS.EmitIntValue(getFuncIdForSubprogram(GV->getSubprogram()).getIndex(), 4);
507dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
508dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
509dac21b43SReid Kleckner     OS.AddComment("Function section index");
510dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
511dac21b43SReid Kleckner     OS.AddComment("Flags");
512dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
51370f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
514dac21b43SReid Kleckner     OS.AddComment("Function name");
5151256125fSDavid Majnemer     // Truncate the name so we won't overflow the record length field.
516b9456a5eSDavid Majnemer     emitNullTerminatedSymbolName(OS, FuncName);
517dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
51870f5bc99SReid Kleckner 
519f9c275feSReid Kleckner     for (const LocalVariable &Var : FI.Locals)
520f9c275feSReid Kleckner       emitLocalVariable(Var);
521f9c275feSReid Kleckner 
522f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
523f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
524f3b9ba49SReid Kleckner     // of their parent inline site.
525f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
526f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
527f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
528f9c275feSReid Kleckner              "child site not in function inline site map");
529f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
530f3b9ba49SReid Kleckner     }
531f3b9ba49SReid Kleckner 
532*3128b10cSDavid Majnemer     if (SP != nullptr)
533*3128b10cSDavid Majnemer       emitDebugInfoForUDTs(LocalUDTs);
534*3128b10cSDavid Majnemer 
53570f5bc99SReid Kleckner     // We're done with this function.
536dac21b43SReid Kleckner     OS.AddComment("Record length");
537dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
538dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
53963a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_PROC_ID_END), 2);
54070f5bc99SReid Kleckner   }
5416f3406dfSReid Kleckner   endCVSubsection(SymbolsEnd);
54270f5bc99SReid Kleckner 
5432214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
544dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
54570f5bc99SReid Kleckner }
54670f5bc99SReid Kleckner 
547876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
548876330d5SReid Kleckner CodeViewDebug::createDefRangeMem(uint16_t CVRegister, int Offset) {
549876330d5SReid Kleckner   LocalVarDefRange DR;
550c6a2f214SAaron Ballman   DR.InMemory = -1;
551876330d5SReid Kleckner   DR.DataOffset = Offset;
552876330d5SReid Kleckner   assert(DR.DataOffset == Offset && "truncation");
553876330d5SReid Kleckner   DR.StructOffset = 0;
554876330d5SReid Kleckner   DR.CVRegister = CVRegister;
555876330d5SReid Kleckner   return DR;
556876330d5SReid Kleckner }
557876330d5SReid Kleckner 
558876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
559876330d5SReid Kleckner CodeViewDebug::createDefRangeReg(uint16_t CVRegister) {
560876330d5SReid Kleckner   LocalVarDefRange DR;
561876330d5SReid Kleckner   DR.InMemory = 0;
562876330d5SReid Kleckner   DR.DataOffset = 0;
563876330d5SReid Kleckner   DR.StructOffset = 0;
564876330d5SReid Kleckner   DR.CVRegister = CVRegister;
565876330d5SReid Kleckner   return DR;
566876330d5SReid Kleckner }
567876330d5SReid Kleckner 
568876330d5SReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable(
569876330d5SReid Kleckner     DenseSet<InlinedVariable> &Processed) {
570876330d5SReid Kleckner   const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
571876330d5SReid Kleckner   const TargetFrameLowering *TFI = TSI.getFrameLowering();
572876330d5SReid Kleckner   const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
573876330d5SReid Kleckner 
574876330d5SReid Kleckner   for (const MachineModuleInfo::VariableDbgInfo &VI :
575876330d5SReid Kleckner        MMI->getVariableDbgInfo()) {
576f9c275feSReid Kleckner     if (!VI.Var)
577f9c275feSReid Kleckner       continue;
578f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
579f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
580f9c275feSReid Kleckner 
581876330d5SReid Kleckner     Processed.insert(InlinedVariable(VI.Var, VI.Loc->getInlinedAt()));
582f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
583f9c275feSReid Kleckner 
584f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
585f9c275feSReid Kleckner     if (!Scope)
586f9c275feSReid Kleckner       continue;
587f9c275feSReid Kleckner 
588f9c275feSReid Kleckner     // Get the frame register used and the offset.
589f9c275feSReid Kleckner     unsigned FrameReg = 0;
590876330d5SReid Kleckner     int FrameOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
591876330d5SReid Kleckner     uint16_t CVReg = TRI->getCodeViewRegNum(FrameReg);
592f9c275feSReid Kleckner 
593f9c275feSReid Kleckner     // Calculate the label ranges.
594876330d5SReid Kleckner     LocalVarDefRange DefRange = createDefRangeMem(CVReg, FrameOffset);
595f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
596f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
597f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
598876330d5SReid Kleckner       End = End ? End : Asm->getFunctionEnd();
599876330d5SReid Kleckner       DefRange.Ranges.emplace_back(Begin, End);
600f9c275feSReid Kleckner     }
601f9c275feSReid Kleckner 
602876330d5SReid Kleckner     LocalVariable Var;
603876330d5SReid Kleckner     Var.DIVar = VI.Var;
604876330d5SReid Kleckner     Var.DefRanges.emplace_back(std::move(DefRange));
605876330d5SReid Kleckner     recordLocalVariable(std::move(Var), VI.Loc->getInlinedAt());
606f9c275feSReid Kleckner   }
607f9c275feSReid Kleckner }
608876330d5SReid Kleckner 
609876330d5SReid Kleckner void CodeViewDebug::collectVariableInfo(const DISubprogram *SP) {
610876330d5SReid Kleckner   DenseSet<InlinedVariable> Processed;
611876330d5SReid Kleckner   // Grab the variable info that was squirreled away in the MMI side-table.
612876330d5SReid Kleckner   collectVariableInfoFromMMITable(Processed);
613876330d5SReid Kleckner 
614876330d5SReid Kleckner   const TargetRegisterInfo *TRI = Asm->MF->getSubtarget().getRegisterInfo();
615876330d5SReid Kleckner 
616876330d5SReid Kleckner   for (const auto &I : DbgValues) {
617876330d5SReid Kleckner     InlinedVariable IV = I.first;
618876330d5SReid Kleckner     if (Processed.count(IV))
619876330d5SReid Kleckner       continue;
620876330d5SReid Kleckner     const DILocalVariable *DIVar = IV.first;
621876330d5SReid Kleckner     const DILocation *InlinedAt = IV.second;
622876330d5SReid Kleckner 
623876330d5SReid Kleckner     // Instruction ranges, specifying where IV is accessible.
624876330d5SReid Kleckner     const auto &Ranges = I.second;
625876330d5SReid Kleckner 
626876330d5SReid Kleckner     LexicalScope *Scope = nullptr;
627876330d5SReid Kleckner     if (InlinedAt)
628876330d5SReid Kleckner       Scope = LScopes.findInlinedScope(DIVar->getScope(), InlinedAt);
629876330d5SReid Kleckner     else
630876330d5SReid Kleckner       Scope = LScopes.findLexicalScope(DIVar->getScope());
631876330d5SReid Kleckner     // If variable scope is not found then skip this variable.
632876330d5SReid Kleckner     if (!Scope)
633876330d5SReid Kleckner       continue;
634876330d5SReid Kleckner 
635876330d5SReid Kleckner     LocalVariable Var;
636876330d5SReid Kleckner     Var.DIVar = DIVar;
637876330d5SReid Kleckner 
638876330d5SReid Kleckner     // Calculate the definition ranges.
639876330d5SReid Kleckner     for (auto I = Ranges.begin(), E = Ranges.end(); I != E; ++I) {
640876330d5SReid Kleckner       const InsnRange &Range = *I;
641876330d5SReid Kleckner       const MachineInstr *DVInst = Range.first;
642876330d5SReid Kleckner       assert(DVInst->isDebugValue() && "Invalid History entry");
643876330d5SReid Kleckner       const DIExpression *DIExpr = DVInst->getDebugExpression();
644876330d5SReid Kleckner 
645876330d5SReid Kleckner       // Bail if there is a complex DWARF expression for now.
646876330d5SReid Kleckner       if (DIExpr && DIExpr->getNumElements() > 0)
647876330d5SReid Kleckner         continue;
648876330d5SReid Kleckner 
6499a593ee7SReid Kleckner       // Bail if operand 0 is not a valid register. This means the variable is a
6509a593ee7SReid Kleckner       // simple constant, or is described by a complex expression.
6519a593ee7SReid Kleckner       // FIXME: Find a way to represent constant variables, since they are
6529a593ee7SReid Kleckner       // relatively common.
6539a593ee7SReid Kleckner       unsigned Reg =
6549a593ee7SReid Kleckner           DVInst->getOperand(0).isReg() ? DVInst->getOperand(0).getReg() : 0;
6559a593ee7SReid Kleckner       if (Reg == 0)
6566e0d5f57SReid Kleckner         continue;
6576e0d5f57SReid Kleckner 
658876330d5SReid Kleckner       // Handle the two cases we can handle: indirect in memory and in register.
659876330d5SReid Kleckner       bool IsIndirect = DVInst->getOperand(1).isImm();
660876330d5SReid Kleckner       unsigned CVReg = TRI->getCodeViewRegNum(DVInst->getOperand(0).getReg());
661876330d5SReid Kleckner       {
662876330d5SReid Kleckner         LocalVarDefRange DefRange;
663876330d5SReid Kleckner         if (IsIndirect) {
664876330d5SReid Kleckner           int64_t Offset = DVInst->getOperand(1).getImm();
665876330d5SReid Kleckner           DefRange = createDefRangeMem(CVReg, Offset);
666876330d5SReid Kleckner         } else {
667876330d5SReid Kleckner           DefRange = createDefRangeReg(CVReg);
668876330d5SReid Kleckner         }
669876330d5SReid Kleckner         if (Var.DefRanges.empty() ||
670876330d5SReid Kleckner             Var.DefRanges.back().isDifferentLocation(DefRange)) {
671876330d5SReid Kleckner           Var.DefRanges.emplace_back(std::move(DefRange));
672876330d5SReid Kleckner         }
673876330d5SReid Kleckner       }
674876330d5SReid Kleckner 
675876330d5SReid Kleckner       // Compute the label range.
676876330d5SReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
677876330d5SReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
678876330d5SReid Kleckner       if (!End) {
679876330d5SReid Kleckner         if (std::next(I) != E)
680876330d5SReid Kleckner           End = getLabelBeforeInsn(std::next(I)->first);
681876330d5SReid Kleckner         else
682876330d5SReid Kleckner           End = Asm->getFunctionEnd();
683876330d5SReid Kleckner       }
684876330d5SReid Kleckner 
685876330d5SReid Kleckner       // If the last range end is our begin, just extend the last range.
686876330d5SReid Kleckner       // Otherwise make a new range.
687876330d5SReid Kleckner       SmallVectorImpl<std::pair<const MCSymbol *, const MCSymbol *>> &Ranges =
688876330d5SReid Kleckner           Var.DefRanges.back().Ranges;
689876330d5SReid Kleckner       if (!Ranges.empty() && Ranges.back().second == Begin)
690876330d5SReid Kleckner         Ranges.back().second = End;
691876330d5SReid Kleckner       else
692876330d5SReid Kleckner         Ranges.emplace_back(Begin, End);
693876330d5SReid Kleckner 
694876330d5SReid Kleckner       // FIXME: Do more range combining.
695876330d5SReid Kleckner     }
696876330d5SReid Kleckner 
697876330d5SReid Kleckner     recordLocalVariable(std::move(Var), InlinedAt);
698876330d5SReid Kleckner   }
699f9c275feSReid Kleckner }
700f9c275feSReid Kleckner 
70170f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
70270f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
70370f5bc99SReid Kleckner 
704f9c275feSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
70570f5bc99SReid Kleckner     return;
70670f5bc99SReid Kleckner 
707f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
708f9c275feSReid Kleckner 
70970f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
71070f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
71170f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
7122214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
7131fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
71470f5bc99SReid Kleckner 
715f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
716f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
71770f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
71870f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
71970f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
72070f5bc99SReid Kleckner   bool EmptyPrologue = true;
72170f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
72270f5bc99SReid Kleckner     for (const auto &MI : MBB) {
723f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
724f9c275feSReid Kleckner           MI.getDebugLoc()) {
72570f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
72670f5bc99SReid Kleckner         break;
727f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
72870f5bc99SReid Kleckner         EmptyPrologue = false;
72970f5bc99SReid Kleckner       }
73070f5bc99SReid Kleckner     }
731f9c275feSReid Kleckner   }
732f9c275feSReid Kleckner 
73370f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
73470f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
73570f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
73670f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
73770f5bc99SReid Kleckner   }
73870f5bc99SReid Kleckner }
73970f5bc99SReid Kleckner 
7405acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerType(const DIType *Ty) {
7415acacbb0SReid Kleckner   // Generic dispatch for lowering an unknown type.
7425acacbb0SReid Kleckner   switch (Ty->getTag()) {
743f3c3c132SAdrian McCarthy   case dwarf::DW_TAG_array_type:
744f3c3c132SAdrian McCarthy     return lowerTypeArray(cast<DICompositeType>(Ty));
745d065e23dSDavid Majnemer   case dwarf::DW_TAG_typedef:
746d065e23dSDavid Majnemer     return lowerTypeAlias(cast<DIDerivedType>(Ty));
7475acacbb0SReid Kleckner   case dwarf::DW_TAG_base_type:
7485acacbb0SReid Kleckner     return lowerTypeBasic(cast<DIBasicType>(Ty));
7495acacbb0SReid Kleckner   case dwarf::DW_TAG_pointer_type:
7505acacbb0SReid Kleckner   case dwarf::DW_TAG_reference_type:
7515acacbb0SReid Kleckner   case dwarf::DW_TAG_rvalue_reference_type:
7525acacbb0SReid Kleckner     return lowerTypePointer(cast<DIDerivedType>(Ty));
7535acacbb0SReid Kleckner   case dwarf::DW_TAG_ptr_to_member_type:
7545acacbb0SReid Kleckner     return lowerTypeMemberPointer(cast<DIDerivedType>(Ty));
7555acacbb0SReid Kleckner   case dwarf::DW_TAG_const_type:
7565acacbb0SReid Kleckner   case dwarf::DW_TAG_volatile_type:
7575acacbb0SReid Kleckner     return lowerTypeModifier(cast<DIDerivedType>(Ty));
75875c3ebfaSDavid Majnemer   case dwarf::DW_TAG_subroutine_type:
75975c3ebfaSDavid Majnemer     return lowerTypeFunction(cast<DISubroutineType>(Ty));
760a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
761a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
762a8d57407SReid Kleckner     return lowerTypeClass(cast<DICompositeType>(Ty));
763a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
764a8d57407SReid Kleckner     return lowerTypeUnion(cast<DICompositeType>(Ty));
7655acacbb0SReid Kleckner   default:
7665acacbb0SReid Kleckner     // Use the null type index.
7675acacbb0SReid Kleckner     return TypeIndex();
7685acacbb0SReid Kleckner   }
7695acacbb0SReid Kleckner }
7705acacbb0SReid Kleckner 
771*3128b10cSDavid Majnemer static const DISubprogram *getQualifiedNameComponents(
772*3128b10cSDavid Majnemer     const DIScope *Scope, SmallVectorImpl<StringRef> &QualifiedNameComponents) {
773*3128b10cSDavid Majnemer   const DISubprogram *ClosestSubprogram = nullptr;
774*3128b10cSDavid Majnemer   while (Scope != nullptr) {
775*3128b10cSDavid Majnemer     if (ClosestSubprogram == nullptr)
776*3128b10cSDavid Majnemer       ClosestSubprogram = dyn_cast<DISubprogram>(Scope);
777*3128b10cSDavid Majnemer     StringRef ScopeName = Scope->getName();
778*3128b10cSDavid Majnemer     if (!ScopeName.empty())
779*3128b10cSDavid Majnemer       QualifiedNameComponents.push_back(ScopeName);
780*3128b10cSDavid Majnemer     Scope = Scope->getScope().resolve();
781*3128b10cSDavid Majnemer   }
782*3128b10cSDavid Majnemer   return ClosestSubprogram;
783*3128b10cSDavid Majnemer }
784*3128b10cSDavid Majnemer 
785*3128b10cSDavid Majnemer static std::string getQualifiedName(ArrayRef<StringRef> QualifiedNameComponents,
786*3128b10cSDavid Majnemer                                     StringRef TypeName) {
787*3128b10cSDavid Majnemer   std::string FullyQualifiedName;
788*3128b10cSDavid Majnemer   for (StringRef QualifiedNameComponent : reverse(QualifiedNameComponents)) {
789*3128b10cSDavid Majnemer     FullyQualifiedName.append(QualifiedNameComponent);
790*3128b10cSDavid Majnemer     FullyQualifiedName.append("::");
791*3128b10cSDavid Majnemer   }
792*3128b10cSDavid Majnemer   FullyQualifiedName.append(TypeName);
793*3128b10cSDavid Majnemer   return FullyQualifiedName;
794*3128b10cSDavid Majnemer }
795*3128b10cSDavid Majnemer 
796d065e23dSDavid Majnemer TypeIndex CodeViewDebug::lowerTypeAlias(const DIDerivedType *Ty) {
797d065e23dSDavid Majnemer   DITypeRef UnderlyingTypeRef = Ty->getBaseType();
798d065e23dSDavid Majnemer   TypeIndex UnderlyingTypeIndex = getTypeIndex(UnderlyingTypeRef);
799*3128b10cSDavid Majnemer   StringRef TypeName = Ty->getName();
800*3128b10cSDavid Majnemer 
801*3128b10cSDavid Majnemer   SmallVector<StringRef, 5> QualifiedNameComponents;
802*3128b10cSDavid Majnemer   const DISubprogram *ClosestSubprogram = getQualifiedNameComponents(
803*3128b10cSDavid Majnemer       Ty->getScope().resolve(), QualifiedNameComponents);
804*3128b10cSDavid Majnemer 
805*3128b10cSDavid Majnemer   if (ClosestSubprogram == nullptr) {
806*3128b10cSDavid Majnemer     std::string FullyQualifiedName =
807*3128b10cSDavid Majnemer         getQualifiedName(QualifiedNameComponents, TypeName);
808*3128b10cSDavid Majnemer     GlobalUDTs.emplace_back(std::move(FullyQualifiedName), UnderlyingTypeIndex);
809*3128b10cSDavid Majnemer   } else if (ClosestSubprogram == CurrentSubprogram) {
810*3128b10cSDavid Majnemer     std::string FullyQualifiedName =
811*3128b10cSDavid Majnemer         getQualifiedName(QualifiedNameComponents, TypeName);
812*3128b10cSDavid Majnemer     LocalUDTs.emplace_back(std::move(FullyQualifiedName), UnderlyingTypeIndex);
813*3128b10cSDavid Majnemer   }
814*3128b10cSDavid Majnemer   // TODO: What if the ClosestSubprogram is neither null or the current
815*3128b10cSDavid Majnemer   // subprogram?  Currently, the UDT just gets dropped on the floor.
816*3128b10cSDavid Majnemer   //
817*3128b10cSDavid Majnemer   // The current behavior is not desirable.  To get maximal fidelity, we would
818*3128b10cSDavid Majnemer   // need to perform all type translation before beginning emission of .debug$S
819*3128b10cSDavid Majnemer   // and then make LocalUDTs a member of FunctionInfo
820*3128b10cSDavid Majnemer 
821d065e23dSDavid Majnemer   if (UnderlyingTypeIndex == TypeIndex(SimpleTypeKind::Int32Long) &&
822*3128b10cSDavid Majnemer       TypeName == "HRESULT")
823d065e23dSDavid Majnemer     return TypeIndex(SimpleTypeKind::HResult);
8248c46a4ceSDavid Majnemer   if (UnderlyingTypeIndex == TypeIndex(SimpleTypeKind::UInt16Short) &&
825*3128b10cSDavid Majnemer       TypeName == "wchar_t")
8268c46a4ceSDavid Majnemer     return TypeIndex(SimpleTypeKind::WideCharacter);
827d065e23dSDavid Majnemer   return UnderlyingTypeIndex;
828d065e23dSDavid Majnemer }
829d065e23dSDavid Majnemer 
830f3c3c132SAdrian McCarthy TypeIndex CodeViewDebug::lowerTypeArray(const DICompositeType *Ty) {
831f3c3c132SAdrian McCarthy   DITypeRef ElementTypeRef = Ty->getBaseType();
832f3c3c132SAdrian McCarthy   TypeIndex ElementTypeIndex = getTypeIndex(ElementTypeRef);
833f3c3c132SAdrian McCarthy   // IndexType is size_t, which depends on the bitness of the target.
834f3c3c132SAdrian McCarthy   TypeIndex IndexType = Asm->MAI->getPointerSize() == 8
835f3c3c132SAdrian McCarthy                             ? TypeIndex(SimpleTypeKind::UInt64Quad)
836f3c3c132SAdrian McCarthy                             : TypeIndex(SimpleTypeKind::UInt32Long);
837f3c3c132SAdrian McCarthy   uint64_t Size = Ty->getSizeInBits() / 8;
838f3c3c132SAdrian McCarthy   ArrayRecord Record(ElementTypeIndex, IndexType, Size, Ty->getName());
839f3c3c132SAdrian McCarthy   return TypeTable.writeArray(Record);
840f3c3c132SAdrian McCarthy }
841f3c3c132SAdrian McCarthy 
8425acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeBasic(const DIBasicType *Ty) {
8435acacbb0SReid Kleckner   TypeIndex Index;
8445acacbb0SReid Kleckner   dwarf::TypeKind Kind;
8455acacbb0SReid Kleckner   uint32_t ByteSize;
8465acacbb0SReid Kleckner 
8475acacbb0SReid Kleckner   Kind = static_cast<dwarf::TypeKind>(Ty->getEncoding());
848afefa673SDavid Majnemer   ByteSize = Ty->getSizeInBits() / 8;
8495acacbb0SReid Kleckner 
8505acacbb0SReid Kleckner   SimpleTypeKind STK = SimpleTypeKind::None;
8515acacbb0SReid Kleckner   switch (Kind) {
8525acacbb0SReid Kleckner   case dwarf::DW_ATE_address:
8535acacbb0SReid Kleckner     // FIXME: Translate
8545acacbb0SReid Kleckner     break;
8555acacbb0SReid Kleckner   case dwarf::DW_ATE_boolean:
8565acacbb0SReid Kleckner     switch (ByteSize) {
8575acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::Boolean8;   break;
8585acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::Boolean16;  break;
8595acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Boolean32;  break;
8605acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Boolean64;  break;
8611c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::Boolean128; break;
8625acacbb0SReid Kleckner     }
8635acacbb0SReid Kleckner     break;
8645acacbb0SReid Kleckner   case dwarf::DW_ATE_complex_float:
8655acacbb0SReid Kleckner     switch (ByteSize) {
8661c2cb1ddSDavid Majnemer     case 2:  STK = SimpleTypeKind::Complex16;  break;
8675acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Complex32;  break;
8685acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Complex64;  break;
8695acacbb0SReid Kleckner     case 10: STK = SimpleTypeKind::Complex80;  break;
8705acacbb0SReid Kleckner     case 16: STK = SimpleTypeKind::Complex128; break;
8715acacbb0SReid Kleckner     }
8725acacbb0SReid Kleckner     break;
8735acacbb0SReid Kleckner   case dwarf::DW_ATE_float:
8745acacbb0SReid Kleckner     switch (ByteSize) {
8751c2cb1ddSDavid Majnemer     case 2:  STK = SimpleTypeKind::Float16;  break;
8765acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Float32;  break;
8775acacbb0SReid Kleckner     case 6:  STK = SimpleTypeKind::Float48;  break;
8785acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Float64;  break;
8795acacbb0SReid Kleckner     case 10: STK = SimpleTypeKind::Float80;  break;
8805acacbb0SReid Kleckner     case 16: STK = SimpleTypeKind::Float128; break;
8815acacbb0SReid Kleckner     }
8825acacbb0SReid Kleckner     break;
8835acacbb0SReid Kleckner   case dwarf::DW_ATE_signed:
8845acacbb0SReid Kleckner     switch (ByteSize) {
8855acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::SByte;      break;
8865acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::Int16Short; break;
8875acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Int32;      break;
8881c2cb1ddSDavid Majnemer     case 8:  STK = SimpleTypeKind::Int64Quad;  break;
8891c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::Int128Oct;  break;
8905acacbb0SReid Kleckner     }
8915acacbb0SReid Kleckner     break;
8925acacbb0SReid Kleckner   case dwarf::DW_ATE_unsigned:
8935acacbb0SReid Kleckner     switch (ByteSize) {
8945acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::Byte;        break;
8955acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::UInt16Short; break;
8965acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::UInt32;      break;
8971c2cb1ddSDavid Majnemer     case 8:  STK = SimpleTypeKind::UInt64Quad;  break;
8981c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::UInt128Oct;  break;
8995acacbb0SReid Kleckner     }
9005acacbb0SReid Kleckner     break;
9015acacbb0SReid Kleckner   case dwarf::DW_ATE_UTF:
9025acacbb0SReid Kleckner     switch (ByteSize) {
9035acacbb0SReid Kleckner     case 2: STK = SimpleTypeKind::Character16; break;
9045acacbb0SReid Kleckner     case 4: STK = SimpleTypeKind::Character32; break;
9055acacbb0SReid Kleckner     }
9065acacbb0SReid Kleckner     break;
9075acacbb0SReid Kleckner   case dwarf::DW_ATE_signed_char:
9085acacbb0SReid Kleckner     if (ByteSize == 1)
9095acacbb0SReid Kleckner       STK = SimpleTypeKind::SignedCharacter;
9105acacbb0SReid Kleckner     break;
9115acacbb0SReid Kleckner   case dwarf::DW_ATE_unsigned_char:
9125acacbb0SReid Kleckner     if (ByteSize == 1)
9135acacbb0SReid Kleckner       STK = SimpleTypeKind::UnsignedCharacter;
9145acacbb0SReid Kleckner     break;
9155acacbb0SReid Kleckner   default:
9165acacbb0SReid Kleckner     break;
9175acacbb0SReid Kleckner   }
9185acacbb0SReid Kleckner 
9195acacbb0SReid Kleckner   // Apply some fixups based on the source-level type name.
9205acacbb0SReid Kleckner   if (STK == SimpleTypeKind::Int32 && Ty->getName() == "long int")
9215acacbb0SReid Kleckner     STK = SimpleTypeKind::Int32Long;
9225acacbb0SReid Kleckner   if (STK == SimpleTypeKind::UInt32 && Ty->getName() == "long unsigned int")
9235acacbb0SReid Kleckner     STK = SimpleTypeKind::UInt32Long;
9248c46a4ceSDavid Majnemer   if (STK == SimpleTypeKind::UInt16Short &&
9258c46a4ceSDavid Majnemer       (Ty->getName() == "wchar_t" || Ty->getName() == "__wchar_t"))
9265acacbb0SReid Kleckner     STK = SimpleTypeKind::WideCharacter;
9275acacbb0SReid Kleckner   if ((STK == SimpleTypeKind::SignedCharacter ||
9285acacbb0SReid Kleckner        STK == SimpleTypeKind::UnsignedCharacter) &&
9295acacbb0SReid Kleckner       Ty->getName() == "char")
9305acacbb0SReid Kleckner     STK = SimpleTypeKind::NarrowCharacter;
9315acacbb0SReid Kleckner 
9325acacbb0SReid Kleckner   return TypeIndex(STK);
9335acacbb0SReid Kleckner }
9345acacbb0SReid Kleckner 
9355acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypePointer(const DIDerivedType *Ty) {
9365acacbb0SReid Kleckner   TypeIndex PointeeTI = getTypeIndex(Ty->getBaseType());
9375acacbb0SReid Kleckner 
9385acacbb0SReid Kleckner   // Pointers to simple types can use SimpleTypeMode, rather than having a
9395acacbb0SReid Kleckner   // dedicated pointer type record.
9405acacbb0SReid Kleckner   if (PointeeTI.isSimple() &&
9415acacbb0SReid Kleckner       PointeeTI.getSimpleMode() == SimpleTypeMode::Direct &&
9425acacbb0SReid Kleckner       Ty->getTag() == dwarf::DW_TAG_pointer_type) {
9435acacbb0SReid Kleckner     SimpleTypeMode Mode = Ty->getSizeInBits() == 64
9445acacbb0SReid Kleckner                               ? SimpleTypeMode::NearPointer64
9455acacbb0SReid Kleckner                               : SimpleTypeMode::NearPointer32;
9465acacbb0SReid Kleckner     return TypeIndex(PointeeTI.getSimpleKind(), Mode);
9475acacbb0SReid Kleckner   }
9485acacbb0SReid Kleckner 
9495acacbb0SReid Kleckner   PointerKind PK =
9505acacbb0SReid Kleckner       Ty->getSizeInBits() == 64 ? PointerKind::Near64 : PointerKind::Near32;
9515acacbb0SReid Kleckner   PointerMode PM = PointerMode::Pointer;
9525acacbb0SReid Kleckner   switch (Ty->getTag()) {
9535acacbb0SReid Kleckner   default: llvm_unreachable("not a pointer tag type");
9545acacbb0SReid Kleckner   case dwarf::DW_TAG_pointer_type:
9555acacbb0SReid Kleckner     PM = PointerMode::Pointer;
9565acacbb0SReid Kleckner     break;
9575acacbb0SReid Kleckner   case dwarf::DW_TAG_reference_type:
9585acacbb0SReid Kleckner     PM = PointerMode::LValueReference;
9595acacbb0SReid Kleckner     break;
9605acacbb0SReid Kleckner   case dwarf::DW_TAG_rvalue_reference_type:
9615acacbb0SReid Kleckner     PM = PointerMode::RValueReference;
9625acacbb0SReid Kleckner     break;
9635acacbb0SReid Kleckner   }
9645acacbb0SReid Kleckner   // FIXME: MSVC folds qualifiers into PointerOptions in the context of a method
9655acacbb0SReid Kleckner   // 'this' pointer, but not normal contexts. Figure out what we're supposed to
9665acacbb0SReid Kleckner   // do.
9675acacbb0SReid Kleckner   PointerOptions PO = PointerOptions::None;
9685acacbb0SReid Kleckner   PointerRecord PR(PointeeTI, PK, PM, PO, Ty->getSizeInBits() / 8);
9695acacbb0SReid Kleckner   return TypeTable.writePointer(PR);
9705acacbb0SReid Kleckner }
9715acacbb0SReid Kleckner 
9725acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeMemberPointer(const DIDerivedType *Ty) {
9735acacbb0SReid Kleckner   assert(Ty->getTag() == dwarf::DW_TAG_ptr_to_member_type);
9745acacbb0SReid Kleckner   TypeIndex ClassTI = getTypeIndex(Ty->getClassType());
9755acacbb0SReid Kleckner   TypeIndex PointeeTI = getTypeIndex(Ty->getBaseType());
9765acacbb0SReid Kleckner   PointerKind PK = Asm->MAI->getPointerSize() == 8 ? PointerKind::Near64
9775acacbb0SReid Kleckner                                                    : PointerKind::Near32;
9785acacbb0SReid Kleckner   PointerMode PM = isa<DISubroutineType>(Ty->getBaseType())
9795acacbb0SReid Kleckner                        ? PointerMode::PointerToMemberFunction
9805acacbb0SReid Kleckner                        : PointerMode::PointerToDataMember;
9815acacbb0SReid Kleckner   PointerOptions PO = PointerOptions::None; // FIXME
9825acacbb0SReid Kleckner   // FIXME: Thread this ABI info through metadata.
9835acacbb0SReid Kleckner   PointerToMemberRepresentation PMR = PointerToMemberRepresentation::Unknown;
9845acacbb0SReid Kleckner   MemberPointerInfo MPI(ClassTI, PMR);
9855acacbb0SReid Kleckner   PointerRecord PR(PointeeTI, PK, PM, PO, Ty->getSizeInBits() / 8, MPI);
9865acacbb0SReid Kleckner   return TypeTable.writePointer(PR);
9875acacbb0SReid Kleckner }
9885acacbb0SReid Kleckner 
989de3d8b50SReid Kleckner /// Given a DWARF calling convention, get the CodeView equivalent. If we don't
990de3d8b50SReid Kleckner /// have a translation, use the NearC convention.
991de3d8b50SReid Kleckner static CallingConvention dwarfCCToCodeView(unsigned DwarfCC) {
992de3d8b50SReid Kleckner   switch (DwarfCC) {
993de3d8b50SReid Kleckner   case dwarf::DW_CC_normal:             return CallingConvention::NearC;
994de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_msfastcall: return CallingConvention::NearFast;
995de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_thiscall:   return CallingConvention::ThisCall;
996de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_stdcall:    return CallingConvention::NearStdCall;
997de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_pascal:     return CallingConvention::NearPascal;
998de3d8b50SReid Kleckner   case dwarf::DW_CC_LLVM_vectorcall:    return CallingConvention::NearVector;
999de3d8b50SReid Kleckner   }
1000de3d8b50SReid Kleckner   return CallingConvention::NearC;
1001de3d8b50SReid Kleckner }
1002de3d8b50SReid Kleckner 
10035acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeModifier(const DIDerivedType *Ty) {
10045acacbb0SReid Kleckner   ModifierOptions Mods = ModifierOptions::None;
10055acacbb0SReid Kleckner   bool IsModifier = true;
10065acacbb0SReid Kleckner   const DIType *BaseTy = Ty;
1007b9c80fd8SReid Kleckner   while (IsModifier && BaseTy) {
10085acacbb0SReid Kleckner     // FIXME: Need to add DWARF tag for __unaligned.
10095acacbb0SReid Kleckner     switch (BaseTy->getTag()) {
10105acacbb0SReid Kleckner     case dwarf::DW_TAG_const_type:
10115acacbb0SReid Kleckner       Mods |= ModifierOptions::Const;
10125acacbb0SReid Kleckner       break;
10135acacbb0SReid Kleckner     case dwarf::DW_TAG_volatile_type:
10145acacbb0SReid Kleckner       Mods |= ModifierOptions::Volatile;
10155acacbb0SReid Kleckner       break;
10165acacbb0SReid Kleckner     default:
10175acacbb0SReid Kleckner       IsModifier = false;
10185acacbb0SReid Kleckner       break;
10195acacbb0SReid Kleckner     }
10205acacbb0SReid Kleckner     if (IsModifier)
10215acacbb0SReid Kleckner       BaseTy = cast<DIDerivedType>(BaseTy)->getBaseType().resolve();
10225acacbb0SReid Kleckner   }
10235acacbb0SReid Kleckner   TypeIndex ModifiedTI = getTypeIndex(BaseTy);
10245acacbb0SReid Kleckner   ModifierRecord MR(ModifiedTI, Mods);
10255acacbb0SReid Kleckner   return TypeTable.writeModifier(MR);
10265acacbb0SReid Kleckner }
10275acacbb0SReid Kleckner 
102875c3ebfaSDavid Majnemer TypeIndex CodeViewDebug::lowerTypeFunction(const DISubroutineType *Ty) {
102975c3ebfaSDavid Majnemer   SmallVector<TypeIndex, 8> ReturnAndArgTypeIndices;
103075c3ebfaSDavid Majnemer   for (DITypeRef ArgTypeRef : Ty->getTypeArray())
103175c3ebfaSDavid Majnemer     ReturnAndArgTypeIndices.push_back(getTypeIndex(ArgTypeRef));
103275c3ebfaSDavid Majnemer 
103375c3ebfaSDavid Majnemer   TypeIndex ReturnTypeIndex = TypeIndex::Void();
103475c3ebfaSDavid Majnemer   ArrayRef<TypeIndex> ArgTypeIndices = None;
103575c3ebfaSDavid Majnemer   if (!ReturnAndArgTypeIndices.empty()) {
103675c3ebfaSDavid Majnemer     auto ReturnAndArgTypesRef = makeArrayRef(ReturnAndArgTypeIndices);
103775c3ebfaSDavid Majnemer     ReturnTypeIndex = ReturnAndArgTypesRef.front();
103875c3ebfaSDavid Majnemer     ArgTypeIndices = ReturnAndArgTypesRef.drop_front();
103975c3ebfaSDavid Majnemer   }
104075c3ebfaSDavid Majnemer 
104175c3ebfaSDavid Majnemer   ArgListRecord ArgListRec(TypeRecordKind::ArgList, ArgTypeIndices);
104275c3ebfaSDavid Majnemer   TypeIndex ArgListIndex = TypeTable.writeArgList(ArgListRec);
104375c3ebfaSDavid Majnemer 
1044de3d8b50SReid Kleckner   CallingConvention CC = dwarfCCToCodeView(Ty->getCC());
1045de3d8b50SReid Kleckner 
104675c3ebfaSDavid Majnemer   // TODO: Some functions are member functions, we should use a more appropriate
104775c3ebfaSDavid Majnemer   // record for those.
1048de3d8b50SReid Kleckner   ProcedureRecord Procedure(ReturnTypeIndex, CC, FunctionOptions::None,
1049de3d8b50SReid Kleckner                             ArgTypeIndices.size(), ArgListIndex);
105075c3ebfaSDavid Majnemer   return TypeTable.writeProcedure(Procedure);
105175c3ebfaSDavid Majnemer }
105275c3ebfaSDavid Majnemer 
1053a8d57407SReid Kleckner static MemberAccess translateAccessFlags(unsigned RecordTag,
1054a8d57407SReid Kleckner                                          const DIType *Member) {
1055a8d57407SReid Kleckner   switch (Member->getFlags() & DINode::FlagAccessibility) {
1056a8d57407SReid Kleckner   case DINode::FlagPrivate:   return MemberAccess::Private;
1057a8d57407SReid Kleckner   case DINode::FlagPublic:    return MemberAccess::Public;
1058a8d57407SReid Kleckner   case DINode::FlagProtected: return MemberAccess::Protected;
1059a8d57407SReid Kleckner   case 0:
1060a8d57407SReid Kleckner     // If there was no explicit access control, provide the default for the tag.
1061a8d57407SReid Kleckner     return RecordTag == dwarf::DW_TAG_class_type ? MemberAccess::Private
1062a8d57407SReid Kleckner                                                  : MemberAccess::Public;
1063a8d57407SReid Kleckner   }
1064a8d57407SReid Kleckner   llvm_unreachable("access flags are exclusive");
1065a8d57407SReid Kleckner }
1066a8d57407SReid Kleckner 
1067a8d57407SReid Kleckner static TypeRecordKind getRecordKind(const DICompositeType *Ty) {
1068a8d57407SReid Kleckner   switch (Ty->getTag()) {
1069a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:     return TypeRecordKind::Class;
1070a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type: return TypeRecordKind::Struct;
1071a8d57407SReid Kleckner   }
1072a8d57407SReid Kleckner   llvm_unreachable("unexpected tag");
1073a8d57407SReid Kleckner }
1074a8d57407SReid Kleckner 
1075a8d57407SReid Kleckner /// Return the HasUniqueName option if it should be present in ClassOptions, or
1076a8d57407SReid Kleckner /// None otherwise.
1077a8d57407SReid Kleckner static ClassOptions getRecordUniqueNameOption(const DICompositeType *Ty) {
1078a8d57407SReid Kleckner   // MSVC always sets this flag now, even for local types. Clang doesn't always
1079a8d57407SReid Kleckner   // appear to give every type a linkage name, which may be problematic for us.
1080a8d57407SReid Kleckner   // FIXME: Investigate the consequences of not following them here.
1081a8d57407SReid Kleckner   return !Ty->getIdentifier().empty() ? ClassOptions::HasUniqueName
1082a8d57407SReid Kleckner                                       : ClassOptions::None;
1083a8d57407SReid Kleckner }
1084a8d57407SReid Kleckner 
1085a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerTypeClass(const DICompositeType *Ty) {
1086a8d57407SReid Kleckner   // First, construct the forward decl.  Don't look into Ty to compute the
1087a8d57407SReid Kleckner   // forward decl options, since it might not be available in all TUs.
1088a8d57407SReid Kleckner   TypeRecordKind Kind = getRecordKind(Ty);
1089a8d57407SReid Kleckner   ClassOptions CO =
1090a8d57407SReid Kleckner       ClassOptions::ForwardReference | getRecordUniqueNameOption(Ty);
1091a8d57407SReid Kleckner   TypeIndex FwdDeclTI = TypeTable.writeClass(ClassRecord(
1092a8d57407SReid Kleckner       Kind, 0, CO, HfaKind::None, WindowsRTClassKind::None, TypeIndex(),
1093a8d57407SReid Kleckner       TypeIndex(), TypeIndex(), 0, Ty->getName(), Ty->getIdentifier()));
1094a8d57407SReid Kleckner   return FwdDeclTI;
1095a8d57407SReid Kleckner }
1096a8d57407SReid Kleckner 
1097a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerCompleteTypeClass(const DICompositeType *Ty) {
1098a8d57407SReid Kleckner   // Construct the field list and complete type record.
1099a8d57407SReid Kleckner   TypeRecordKind Kind = getRecordKind(Ty);
1100a8d57407SReid Kleckner   // FIXME: Other ClassOptions, like ContainsNestedClass and NestedClass.
1101a8d57407SReid Kleckner   ClassOptions CO = ClassOptions::None | getRecordUniqueNameOption(Ty);
1102a8d57407SReid Kleckner   TypeIndex FTI;
1103a8d57407SReid Kleckner   unsigned FieldCount;
1104a8d57407SReid Kleckner   std::tie(FTI, FieldCount) = lowerRecordFieldList(Ty);
1105a8d57407SReid Kleckner 
1106a8d57407SReid Kleckner   uint64_t SizeInBytes = Ty->getSizeInBits() / 8;
1107a8d57407SReid Kleckner   return TypeTable.writeClass(ClassRecord(Kind, FieldCount, CO, HfaKind::None,
1108a8d57407SReid Kleckner                                           WindowsRTClassKind::None, FTI,
1109a8d57407SReid Kleckner                                           TypeIndex(), TypeIndex(), SizeInBytes,
1110a8d57407SReid Kleckner                                           Ty->getName(), Ty->getIdentifier()));
1111a8d57407SReid Kleckner   // FIXME: Make an LF_UDT_SRC_LINE record.
1112a8d57407SReid Kleckner }
1113a8d57407SReid Kleckner 
1114a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerTypeUnion(const DICompositeType *Ty) {
1115a8d57407SReid Kleckner   ClassOptions CO =
1116a8d57407SReid Kleckner       ClassOptions::ForwardReference | getRecordUniqueNameOption(Ty);
1117a8d57407SReid Kleckner   TypeIndex FwdDeclTI =
1118a8d57407SReid Kleckner       TypeTable.writeUnion(UnionRecord(0, CO, HfaKind::None, TypeIndex(), 0,
1119a8d57407SReid Kleckner                                        Ty->getName(), Ty->getIdentifier()));
1120a8d57407SReid Kleckner   return FwdDeclTI;
1121a8d57407SReid Kleckner }
1122a8d57407SReid Kleckner 
1123a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerCompleteTypeUnion(const DICompositeType *Ty) {
1124a8d57407SReid Kleckner   ClassOptions CO = ClassOptions::None | getRecordUniqueNameOption(Ty);
1125a8d57407SReid Kleckner   TypeIndex FTI;
1126a8d57407SReid Kleckner   unsigned FieldCount;
1127a8d57407SReid Kleckner   std::tie(FTI, FieldCount) = lowerRecordFieldList(Ty);
1128a8d57407SReid Kleckner   uint64_t SizeInBytes = Ty->getSizeInBits() / 8;
1129a8d57407SReid Kleckner   return TypeTable.writeUnion(UnionRecord(FieldCount, CO, HfaKind::None, FTI,
1130a8d57407SReid Kleckner                                           SizeInBytes, Ty->getName(),
1131a8d57407SReid Kleckner                                           Ty->getIdentifier()));
1132a8d57407SReid Kleckner   // FIXME: Make an LF_UDT_SRC_LINE record.
1133a8d57407SReid Kleckner }
1134a8d57407SReid Kleckner 
1135a8d57407SReid Kleckner std::pair<TypeIndex, unsigned>
1136a8d57407SReid Kleckner CodeViewDebug::lowerRecordFieldList(const DICompositeType *Ty) {
1137a8d57407SReid Kleckner   // Manually count members. MSVC appears to count everything that generates a
1138a8d57407SReid Kleckner   // field list record. Each individual overload in a method overload group
1139a8d57407SReid Kleckner   // contributes to this count, even though the overload group is a single field
1140a8d57407SReid Kleckner   // list record.
1141a8d57407SReid Kleckner   unsigned MemberCount = 0;
1142a8d57407SReid Kleckner   FieldListRecordBuilder Fields;
1143a8d57407SReid Kleckner   for (const DINode *Element : Ty->getElements()) {
1144a8d57407SReid Kleckner     // We assume that the frontend provides all members in source declaration
1145a8d57407SReid Kleckner     // order, which is what MSVC does.
1146a8d57407SReid Kleckner     if (!Element)
1147a8d57407SReid Kleckner       continue;
1148a8d57407SReid Kleckner     if (auto *SP = dyn_cast<DISubprogram>(Element)) {
1149a8d57407SReid Kleckner       // C++ method.
1150a8d57407SReid Kleckner       // FIXME: Overloaded methods are grouped together, so we'll need two
1151a8d57407SReid Kleckner       // passes to group them.
1152a8d57407SReid Kleckner       (void)SP;
1153a8d57407SReid Kleckner     } else if (auto *Member = dyn_cast<DIDerivedType>(Element)) {
1154a8d57407SReid Kleckner       if (Member->getTag() == dwarf::DW_TAG_member) {
1155a8d57407SReid Kleckner         if (Member->isStaticMember()) {
1156a8d57407SReid Kleckner           // Static data member.
1157a8d57407SReid Kleckner           Fields.writeStaticDataMember(StaticDataMemberRecord(
1158a8d57407SReid Kleckner               translateAccessFlags(Ty->getTag(), Member),
1159a8d57407SReid Kleckner               getTypeIndex(Member->getBaseType()), Member->getName()));
1160a8d57407SReid Kleckner           MemberCount++;
1161a8d57407SReid Kleckner         } else {
1162a8d57407SReid Kleckner           // Data member.
1163a8d57407SReid Kleckner           // FIXME: Make a BitFieldRecord for bitfields.
1164a8d57407SReid Kleckner           Fields.writeDataMember(DataMemberRecord(
1165a8d57407SReid Kleckner               translateAccessFlags(Ty->getTag(), Member),
1166a8d57407SReid Kleckner               getTypeIndex(Member->getBaseType()),
1167a8d57407SReid Kleckner               Member->getOffsetInBits() / 8, Member->getName()));
1168a8d57407SReid Kleckner           MemberCount++;
1169a8d57407SReid Kleckner         }
1170a8d57407SReid Kleckner       } else if (Member->getTag() == dwarf::DW_TAG_friend) {
1171a8d57407SReid Kleckner         // Ignore friend members. It appears that MSVC emitted info about
1172a8d57407SReid Kleckner         // friends in the past, but modern versions do not.
1173a8d57407SReid Kleckner       }
1174a8d57407SReid Kleckner       // FIXME: Get clang to emit nested types here and do something with
1175a8d57407SReid Kleckner       // them.
1176a8d57407SReid Kleckner     }
1177a8d57407SReid Kleckner     // Skip other unrecognized kinds of elements.
1178a8d57407SReid Kleckner   }
1179a8d57407SReid Kleckner   return {TypeTable.writeFieldList(Fields), MemberCount};
1180a8d57407SReid Kleckner }
1181a8d57407SReid Kleckner 
11825acacbb0SReid Kleckner TypeIndex CodeViewDebug::getTypeIndex(DITypeRef TypeRef) {
11835acacbb0SReid Kleckner   const DIType *Ty = TypeRef.resolve();
11845acacbb0SReid Kleckner 
11855acacbb0SReid Kleckner   // The null DIType is the void type. Don't try to hash it.
11865acacbb0SReid Kleckner   if (!Ty)
11875acacbb0SReid Kleckner     return TypeIndex::Void();
11885acacbb0SReid Kleckner 
1189a8d57407SReid Kleckner   // Check if we've already translated this type. Don't try to do a
1190a8d57407SReid Kleckner   // get-or-create style insertion that caches the hash lookup across the
1191a8d57407SReid Kleckner   // lowerType call. It will update the TypeIndices map.
11925acacbb0SReid Kleckner   auto I = TypeIndices.find(Ty);
11935acacbb0SReid Kleckner   if (I != TypeIndices.end())
11945acacbb0SReid Kleckner     return I->second;
11955acacbb0SReid Kleckner 
11965acacbb0SReid Kleckner   TypeIndex TI = lowerType(Ty);
11975acacbb0SReid Kleckner 
1198a8d57407SReid Kleckner   recordTypeIndexForDINode(Ty, TI);
1199a8d57407SReid Kleckner   return TI;
1200a8d57407SReid Kleckner }
1201a8d57407SReid Kleckner 
1202a8d57407SReid Kleckner TypeIndex CodeViewDebug::getCompleteTypeIndex(DITypeRef TypeRef) {
1203a8d57407SReid Kleckner   const DIType *Ty = TypeRef.resolve();
1204a8d57407SReid Kleckner 
1205a8d57407SReid Kleckner   // The null DIType is the void type. Don't try to hash it.
1206a8d57407SReid Kleckner   if (!Ty)
1207a8d57407SReid Kleckner     return TypeIndex::Void();
1208a8d57407SReid Kleckner 
1209a8d57407SReid Kleckner   // If this is a non-record type, the complete type index is the same as the
1210a8d57407SReid Kleckner   // normal type index. Just call getTypeIndex.
1211a8d57407SReid Kleckner   switch (Ty->getTag()) {
1212a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
1213a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
1214a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
1215a8d57407SReid Kleckner     break;
1216a8d57407SReid Kleckner   default:
1217a8d57407SReid Kleckner     return getTypeIndex(Ty);
1218a8d57407SReid Kleckner   }
1219a8d57407SReid Kleckner 
1220a8d57407SReid Kleckner   // Check if we've already translated the complete record type.  Lowering a
1221a8d57407SReid Kleckner   // complete type should never trigger lowering another complete type, so we
1222a8d57407SReid Kleckner   // can reuse the hash table lookup result.
1223a8d57407SReid Kleckner   const auto *CTy = cast<DICompositeType>(Ty);
1224a8d57407SReid Kleckner   auto InsertResult = CompleteTypeIndices.insert({CTy, TypeIndex()});
1225a8d57407SReid Kleckner   if (!InsertResult.second)
1226a8d57407SReid Kleckner     return InsertResult.first->second;
1227a8d57407SReid Kleckner 
1228a8d57407SReid Kleckner   // Make sure the forward declaration is emitted first. It's unclear if this
1229a8d57407SReid Kleckner   // is necessary, but MSVC does it, and we should follow suit until we can show
1230a8d57407SReid Kleckner   // otherwise.
1231a8d57407SReid Kleckner   TypeIndex FwdDeclTI = getTypeIndex(CTy);
1232a8d57407SReid Kleckner 
1233a8d57407SReid Kleckner   // Just use the forward decl if we don't have complete type info. This might
1234a8d57407SReid Kleckner   // happen if the frontend is using modules and expects the complete definition
1235a8d57407SReid Kleckner   // to be emitted elsewhere.
1236a8d57407SReid Kleckner   if (CTy->isForwardDecl())
1237a8d57407SReid Kleckner     return FwdDeclTI;
1238a8d57407SReid Kleckner 
1239a8d57407SReid Kleckner   TypeIndex TI;
1240a8d57407SReid Kleckner   switch (CTy->getTag()) {
1241a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
1242a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
1243a8d57407SReid Kleckner     TI = lowerCompleteTypeClass(CTy);
1244a8d57407SReid Kleckner     break;
1245a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
1246a8d57407SReid Kleckner     TI = lowerCompleteTypeUnion(CTy);
1247a8d57407SReid Kleckner     break;
1248a8d57407SReid Kleckner   default:
1249a8d57407SReid Kleckner     llvm_unreachable("not a record");
1250a8d57407SReid Kleckner   }
1251a8d57407SReid Kleckner 
1252a8d57407SReid Kleckner   InsertResult.first->second = TI;
12535acacbb0SReid Kleckner   return TI;
12545acacbb0SReid Kleckner }
12555acacbb0SReid Kleckner 
1256f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
1257f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
1258f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
1259f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
1260f9c275feSReid Kleckner   OS.AddComment("Record length");
1261f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
1262f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
1263f9c275feSReid Kleckner 
1264f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
126563a2846eSZachary Turner   OS.EmitIntValue(unsigned(SymbolKind::S_LOCAL), 2);
1266f9c275feSReid Kleckner 
126763a2846eSZachary Turner   LocalSymFlags Flags = LocalSymFlags::None;
1268f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
126963a2846eSZachary Turner     Flags |= LocalSymFlags::IsParameter;
1270876330d5SReid Kleckner   if (Var.DefRanges.empty())
127163a2846eSZachary Turner     Flags |= LocalSymFlags::IsOptimizedOut;
1272f9c275feSReid Kleckner 
1273f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
1274a8d57407SReid Kleckner   TypeIndex TI = getCompleteTypeIndex(Var.DIVar->getType());
12755acacbb0SReid Kleckner   OS.EmitIntValue(TI.getIndex(), 4);
1276f9c275feSReid Kleckner   OS.AddComment("Flags");
127763a2846eSZachary Turner   OS.EmitIntValue(static_cast<uint16_t>(Flags), 2);
12781256125fSDavid Majnemer   // Truncate the name so we won't overflow the record length field.
1279b9456a5eSDavid Majnemer   emitNullTerminatedSymbolName(OS, Var.DIVar->getName());
1280f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
1281f9c275feSReid Kleckner 
1282876330d5SReid Kleckner   // Calculate the on disk prefix of the appropriate def range record. The
1283876330d5SReid Kleckner   // records and on disk formats are described in SymbolRecords.h. BytePrefix
1284876330d5SReid Kleckner   // should be big enough to hold all forms without memory allocation.
1285876330d5SReid Kleckner   SmallString<20> BytePrefix;
1286876330d5SReid Kleckner   for (const LocalVarDefRange &DefRange : Var.DefRanges) {
1287876330d5SReid Kleckner     BytePrefix.clear();
1288876330d5SReid Kleckner     // FIXME: Handle bitpieces.
1289876330d5SReid Kleckner     if (DefRange.StructOffset != 0)
1290876330d5SReid Kleckner       continue;
1291876330d5SReid Kleckner 
1292876330d5SReid Kleckner     if (DefRange.InMemory) {
1293a78ecd1eSZachary Turner       DefRangeRegisterRelSym Sym(DefRange.CVRegister, 0, DefRange.DataOffset, 0,
1294a78ecd1eSZachary Turner                                  0, 0, ArrayRef<LocalVariableAddrGap>());
1295f9c275feSReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
1296876330d5SReid Kleckner       BytePrefix +=
1297876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
1298a78ecd1eSZachary Turner       BytePrefix +=
1299a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
1300a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
1301876330d5SReid Kleckner     } else {
1302876330d5SReid Kleckner       assert(DefRange.DataOffset == 0 && "unexpected offset into register");
1303a78ecd1eSZachary Turner       // Unclear what matters here.
1304a78ecd1eSZachary Turner       DefRangeRegisterSym Sym(DefRange.CVRegister, 0, 0, 0, 0,
1305a78ecd1eSZachary Turner                               ArrayRef<LocalVariableAddrGap>());
1306876330d5SReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER);
1307876330d5SReid Kleckner       BytePrefix +=
1308876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
1309a78ecd1eSZachary Turner       BytePrefix +=
1310a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
1311a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
1312876330d5SReid Kleckner     }
1313876330d5SReid Kleckner     OS.EmitCVDefRangeDirective(DefRange.Ranges, BytePrefix);
1314876330d5SReid Kleckner   }
1315f9c275feSReid Kleckner }
1316f9c275feSReid Kleckner 
131770f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
131870f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
131970f5bc99SReid Kleckner     return;
132070f5bc99SReid Kleckner 
132170f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
132270f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
132370f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
132470f5bc99SReid Kleckner 
1325adebb937SPete Cooper   collectVariableInfo(GV->getSubprogram());
1326876330d5SReid Kleckner 
1327876330d5SReid Kleckner   DebugHandlerBase::endFunction(MF);
1328876330d5SReid Kleckner 
13292214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
13302214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
133170f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
1332f9c275feSReid Kleckner     CurFn = nullptr;
1333f9c275feSReid Kleckner     return;
133470f5bc99SReid Kleckner   }
1335f9c275feSReid Kleckner 
1336f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
1337f9c275feSReid Kleckner 
133870f5bc99SReid Kleckner   CurFn = nullptr;
133970f5bc99SReid Kleckner }
134070f5bc99SReid Kleckner 
134170f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
1342f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
1343f9c275feSReid Kleckner 
134470f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
134570f5bc99SReid Kleckner   if (!Asm || MI->isDebugValue() || MI->getFlag(MachineInstr::FrameSetup))
134670f5bc99SReid Kleckner     return;
134770f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
134870f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
134970f5bc99SReid Kleckner     return;
135070f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
135170f5bc99SReid Kleckner }
13526f3406dfSReid Kleckner 
13536f3406dfSReid Kleckner MCSymbol *CodeViewDebug::beginCVSubsection(ModuleSubstreamKind Kind) {
13546f3406dfSReid Kleckner   MCSymbol *BeginLabel = MMI->getContext().createTempSymbol(),
13556f3406dfSReid Kleckner            *EndLabel = MMI->getContext().createTempSymbol();
13566f3406dfSReid Kleckner   OS.EmitIntValue(unsigned(Kind), 4);
13576f3406dfSReid Kleckner   OS.AddComment("Subsection size");
13586f3406dfSReid Kleckner   OS.emitAbsoluteSymbolDiff(EndLabel, BeginLabel, 4);
13596f3406dfSReid Kleckner   OS.EmitLabel(BeginLabel);
13606f3406dfSReid Kleckner   return EndLabel;
13616f3406dfSReid Kleckner }
13626f3406dfSReid Kleckner 
13636f3406dfSReid Kleckner void CodeViewDebug::endCVSubsection(MCSymbol *EndLabel) {
13646f3406dfSReid Kleckner   OS.EmitLabel(EndLabel);
13656f3406dfSReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
13666f3406dfSReid Kleckner   OS.EmitValueToAlignment(4);
13676f3406dfSReid Kleckner }
13686f3406dfSReid Kleckner 
1369*3128b10cSDavid Majnemer void CodeViewDebug::emitDebugInfoForUDTs(
1370*3128b10cSDavid Majnemer     ArrayRef<std::pair<std::string, TypeIndex>> UDTs) {
1371*3128b10cSDavid Majnemer   for (const std::pair<std::string, codeview::TypeIndex> &UDT : UDTs) {
1372*3128b10cSDavid Majnemer     MCSymbol *UDTRecordBegin = MMI->getContext().createTempSymbol(),
1373*3128b10cSDavid Majnemer              *UDTRecordEnd = MMI->getContext().createTempSymbol();
1374*3128b10cSDavid Majnemer     OS.AddComment("Record length");
1375*3128b10cSDavid Majnemer     OS.emitAbsoluteSymbolDiff(UDTRecordEnd, UDTRecordBegin, 2);
1376*3128b10cSDavid Majnemer     OS.EmitLabel(UDTRecordBegin);
1377*3128b10cSDavid Majnemer 
1378*3128b10cSDavid Majnemer     OS.AddComment("Record kind: S_UDT");
1379*3128b10cSDavid Majnemer     OS.EmitIntValue(unsigned(SymbolKind::S_UDT), 2);
1380*3128b10cSDavid Majnemer 
1381*3128b10cSDavid Majnemer     OS.AddComment("Type");
1382*3128b10cSDavid Majnemer     OS.EmitIntValue(UDT.second.getIndex(), 4);
1383*3128b10cSDavid Majnemer 
1384*3128b10cSDavid Majnemer     emitNullTerminatedSymbolName(OS, UDT.first);
1385*3128b10cSDavid Majnemer     OS.EmitLabel(UDTRecordEnd);
1386*3128b10cSDavid Majnemer   }
1387*3128b10cSDavid Majnemer }
1388*3128b10cSDavid Majnemer 
13896f3406dfSReid Kleckner void CodeViewDebug::emitDebugInfoForGlobals() {
13906f3406dfSReid Kleckner   NamedMDNode *CUs = MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
13916f3406dfSReid Kleckner   for (const MDNode *Node : CUs->operands()) {
13926f3406dfSReid Kleckner     const auto *CU = cast<DICompileUnit>(Node);
13936f3406dfSReid Kleckner 
13946f3406dfSReid Kleckner     // First, emit all globals that are not in a comdat in a single symbol
13956f3406dfSReid Kleckner     // substream. MSVC doesn't like it if the substream is empty, so only open
13966f3406dfSReid Kleckner     // it if we have at least one global to emit.
13976f3406dfSReid Kleckner     switchToDebugSectionForSymbol(nullptr);
13986f3406dfSReid Kleckner     MCSymbol *EndLabel = nullptr;
13996f3406dfSReid Kleckner     for (const DIGlobalVariable *G : CU->getGlobalVariables()) {
14006d1d2754SReid Kleckner       if (const auto *GV = dyn_cast_or_null<GlobalVariable>(G->getVariable())) {
1401577be0feSDavid Majnemer         if (!GV->hasComdat() && !GV->isDeclarationForLinker()) {
14026f3406dfSReid Kleckner           if (!EndLabel) {
14036f3406dfSReid Kleckner             OS.AddComment("Symbol subsection for globals");
14046f3406dfSReid Kleckner             EndLabel = beginCVSubsection(ModuleSubstreamKind::Symbols);
14056f3406dfSReid Kleckner           }
14066f3406dfSReid Kleckner           emitDebugInfoForGlobal(G, Asm->getSymbol(GV));
14076f3406dfSReid Kleckner         }
14086f3406dfSReid Kleckner       }
14096d1d2754SReid Kleckner     }
14106f3406dfSReid Kleckner     if (EndLabel)
14116f3406dfSReid Kleckner       endCVSubsection(EndLabel);
14126f3406dfSReid Kleckner 
14136f3406dfSReid Kleckner     // Second, emit each global that is in a comdat into its own .debug$S
14146f3406dfSReid Kleckner     // section along with its own symbol substream.
14156f3406dfSReid Kleckner     for (const DIGlobalVariable *G : CU->getGlobalVariables()) {
14166d1d2754SReid Kleckner       if (const auto *GV = dyn_cast_or_null<GlobalVariable>(G->getVariable())) {
14176f3406dfSReid Kleckner         if (GV->hasComdat()) {
14186f3406dfSReid Kleckner           MCSymbol *GVSym = Asm->getSymbol(GV);
14196f3406dfSReid Kleckner           OS.AddComment("Symbol subsection for " +
14206f3406dfSReid Kleckner                         Twine(GlobalValue::getRealLinkageName(GV->getName())));
14216f3406dfSReid Kleckner           switchToDebugSectionForSymbol(GVSym);
14226f3406dfSReid Kleckner           EndLabel = beginCVSubsection(ModuleSubstreamKind::Symbols);
14236f3406dfSReid Kleckner           emitDebugInfoForGlobal(G, GVSym);
14246f3406dfSReid Kleckner           endCVSubsection(EndLabel);
14256f3406dfSReid Kleckner         }
14266f3406dfSReid Kleckner       }
14276f3406dfSReid Kleckner     }
14286f3406dfSReid Kleckner   }
14296f3406dfSReid Kleckner }
14306f3406dfSReid Kleckner 
14316f3406dfSReid Kleckner void CodeViewDebug::emitDebugInfoForGlobal(const DIGlobalVariable *DIGV,
14326f3406dfSReid Kleckner                                            MCSymbol *GVSym) {
14336f3406dfSReid Kleckner   // DataSym record, see SymbolRecord.h for more info.
14346f3406dfSReid Kleckner   // FIXME: Thread local data, etc
14356f3406dfSReid Kleckner   MCSymbol *DataBegin = MMI->getContext().createTempSymbol(),
14366f3406dfSReid Kleckner            *DataEnd = MMI->getContext().createTempSymbol();
14376f3406dfSReid Kleckner   OS.AddComment("Record length");
14386f3406dfSReid Kleckner   OS.emitAbsoluteSymbolDiff(DataEnd, DataBegin, 2);
14396f3406dfSReid Kleckner   OS.EmitLabel(DataBegin);
14406f3406dfSReid Kleckner   OS.AddComment("Record kind: S_GDATA32");
14416f3406dfSReid Kleckner   OS.EmitIntValue(unsigned(SymbolKind::S_GDATA32), 2);
14426f3406dfSReid Kleckner   OS.AddComment("Type");
14436f3406dfSReid Kleckner   OS.EmitIntValue(getCompleteTypeIndex(DIGV->getType()).getIndex(), 4);
14446f3406dfSReid Kleckner   OS.AddComment("DataOffset");
14456f3406dfSReid Kleckner   OS.EmitCOFFSecRel32(GVSym);
14466f3406dfSReid Kleckner   OS.AddComment("Segment");
14476f3406dfSReid Kleckner   OS.EmitCOFFSectionIndex(GVSym);
14486f3406dfSReid Kleckner   OS.AddComment("Name");
14496f3406dfSReid Kleckner   emitNullTerminatedSymbolName(OS, DIGV->getName());
14506f3406dfSReid Kleckner   OS.EmitLabel(DataEnd);
14516f3406dfSReid Kleckner }
1452