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);
138ac945e27SReid Kleckner   // The display name includes function template arguments. Drop them to match
139ac945e27SReid Kleckner   // MSVC.
140ac945e27SReid Kleckner   StringRef DisplayName = SP->getDisplayName().split('<').first;
14175c3ebfaSDavid Majnemer   FuncIdRecord FuncId(ParentScope, getTypeIndex(SP->getType()), DisplayName);
1422280f932SReid Kleckner   TypeIndex TI = TypeTable.writeFuncId(FuncId);
14375c3ebfaSDavid Majnemer 
144a8d57407SReid Kleckner   recordTypeIndexForDINode(SP, TI);
14575c3ebfaSDavid Majnemer   return TI;
1462280f932SReid Kleckner }
1472280f932SReid Kleckner 
148a8d57407SReid Kleckner void CodeViewDebug::recordTypeIndexForDINode(const DINode *Node, TypeIndex TI) {
149a8d57407SReid Kleckner   auto InsertResult = TypeIndices.insert({Node, TI});
150a8d57407SReid Kleckner   (void)InsertResult;
151a8d57407SReid Kleckner   assert(InsertResult.second && "DINode was already assigned a type index");
152a8d57407SReid Kleckner }
153a8d57407SReid Kleckner 
154876330d5SReid Kleckner void CodeViewDebug::recordLocalVariable(LocalVariable &&Var,
155876330d5SReid Kleckner                                         const DILocation *InlinedAt) {
156876330d5SReid Kleckner   if (InlinedAt) {
157876330d5SReid Kleckner     // This variable was inlined. Associate it with the InlineSite.
158876330d5SReid Kleckner     const DISubprogram *Inlinee = Var.DIVar->getScope()->getSubprogram();
159876330d5SReid Kleckner     InlineSite &Site = getInlineSite(InlinedAt, Inlinee);
160876330d5SReid Kleckner     Site.InlinedLocals.emplace_back(Var);
161876330d5SReid Kleckner   } else {
162876330d5SReid Kleckner     // This variable goes in the main ProcSym.
163876330d5SReid Kleckner     CurFn->Locals.emplace_back(Var);
164876330d5SReid Kleckner   }
165876330d5SReid Kleckner }
166876330d5SReid Kleckner 
167829365aeSReid Kleckner static void addLocIfNotPresent(SmallVectorImpl<const DILocation *> &Locs,
168829365aeSReid Kleckner                                const DILocation *Loc) {
169829365aeSReid Kleckner   auto B = Locs.begin(), E = Locs.end();
170829365aeSReid Kleckner   if (std::find(B, E, Loc) == E)
171829365aeSReid Kleckner     Locs.push_back(Loc);
172829365aeSReid Kleckner }
173829365aeSReid Kleckner 
174bdc4956bSBenjamin Kramer void CodeViewDebug::maybeRecordLocation(const DebugLoc &DL,
17570f5bc99SReid Kleckner                                         const MachineFunction *MF) {
1769533af4fSReid Kleckner   // Skip this instruction if it has the same location as the previous one.
1779533af4fSReid Kleckner   if (DL == CurFn->LastLoc)
1789533af4fSReid Kleckner     return;
1799533af4fSReid Kleckner 
1809533af4fSReid Kleckner   const DIScope *Scope = DL.get()->getScope();
18170f5bc99SReid Kleckner   if (!Scope)
18270f5bc99SReid Kleckner     return;
1839533af4fSReid Kleckner 
18470f5bc99SReid Kleckner   // Skip this line if it is longer than the maximum we can record.
1852214ed89SReid Kleckner   LineInfo LI(DL.getLine(), DL.getLine(), /*IsStatement=*/true);
1862214ed89SReid Kleckner   if (LI.getStartLine() != DL.getLine() || LI.isAlwaysStepInto() ||
1872214ed89SReid Kleckner       LI.isNeverStepInto())
18870f5bc99SReid Kleckner     return;
18970f5bc99SReid Kleckner 
1902214ed89SReid Kleckner   ColumnInfo CI(DL.getCol(), /*EndColumn=*/0);
1912214ed89SReid Kleckner   if (CI.getStartColumn() != DL.getCol())
1922214ed89SReid Kleckner     return;
19300d9639cSReid Kleckner 
1942214ed89SReid Kleckner   if (!CurFn->HaveLineInfo)
1952214ed89SReid Kleckner     CurFn->HaveLineInfo = true;
1962214ed89SReid Kleckner   unsigned FileId = 0;
1972214ed89SReid Kleckner   if (CurFn->LastLoc.get() && CurFn->LastLoc->getFile() == DL->getFile())
1982214ed89SReid Kleckner     FileId = CurFn->LastFileId;
1992214ed89SReid Kleckner   else
2002214ed89SReid Kleckner     FileId = CurFn->LastFileId = maybeRecordFile(DL->getFile());
2012214ed89SReid Kleckner   CurFn->LastLoc = DL;
202f3b9ba49SReid Kleckner 
203f3b9ba49SReid Kleckner   unsigned FuncId = CurFn->FuncId;
204876330d5SReid Kleckner   if (const DILocation *SiteLoc = DL->getInlinedAt()) {
205829365aeSReid Kleckner     const DILocation *Loc = DL.get();
206829365aeSReid Kleckner 
207f3b9ba49SReid Kleckner     // If this location was actually inlined from somewhere else, give it the ID
208f3b9ba49SReid Kleckner     // of the inline call site.
209876330d5SReid Kleckner     FuncId =
210876330d5SReid Kleckner         getInlineSite(SiteLoc, Loc->getScope()->getSubprogram()).SiteFuncId;
211829365aeSReid Kleckner 
212f3b9ba49SReid Kleckner     // Ensure we have links in the tree of inline call sites.
213829365aeSReid Kleckner     bool FirstLoc = true;
214829365aeSReid Kleckner     while ((SiteLoc = Loc->getInlinedAt())) {
215876330d5SReid Kleckner       InlineSite &Site =
216876330d5SReid Kleckner           getInlineSite(SiteLoc, Loc->getScope()->getSubprogram());
217829365aeSReid Kleckner       if (!FirstLoc)
218829365aeSReid Kleckner         addLocIfNotPresent(Site.ChildSites, Loc);
219829365aeSReid Kleckner       FirstLoc = false;
220829365aeSReid Kleckner       Loc = SiteLoc;
221f3b9ba49SReid Kleckner     }
222829365aeSReid Kleckner     addLocIfNotPresent(CurFn->ChildSites, Loc);
223f3b9ba49SReid Kleckner   }
224f3b9ba49SReid Kleckner 
225dac21b43SReid Kleckner   OS.EmitCVLocDirective(FuncId, FileId, DL.getLine(), DL.getCol(),
226dac21b43SReid Kleckner                         /*PrologueEnd=*/false,
2272214ed89SReid Kleckner                         /*IsStmt=*/false, DL->getFilename());
22870f5bc99SReid Kleckner }
22970f5bc99SReid Kleckner 
2305d122f87SReid Kleckner void CodeViewDebug::emitCodeViewMagicVersion() {
2315d122f87SReid Kleckner   OS.EmitValueToAlignment(4);
2325d122f87SReid Kleckner   OS.AddComment("Debug section magic");
2335d122f87SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
2345d122f87SReid Kleckner }
2355d122f87SReid Kleckner 
23670f5bc99SReid Kleckner void CodeViewDebug::endModule() {
2376f3406dfSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
23870f5bc99SReid Kleckner     return;
23970f5bc99SReid Kleckner 
24070f5bc99SReid Kleckner   assert(Asm != nullptr);
24170f5bc99SReid Kleckner 
24270f5bc99SReid Kleckner   // The COFF .debug$S section consists of several subsections, each starting
24370f5bc99SReid Kleckner   // with a 4-byte control code (e.g. 0xF1, 0xF2, etc) and then a 4-byte length
24470f5bc99SReid Kleckner   // of the payload followed by the payload itself.  The subsections are 4-byte
24570f5bc99SReid Kleckner   // aligned.
24670f5bc99SReid Kleckner 
2476f3406dfSReid Kleckner   // Use the generic .debug$S section, and make a subsection for all the inlined
2486f3406dfSReid Kleckner   // subprograms.
2496f3406dfSReid Kleckner   switchToDebugSectionForSymbol(nullptr);
2505d122f87SReid Kleckner   emitInlineeLinesSubsection();
2511fcd610cSReid Kleckner 
2522214ed89SReid Kleckner   // Emit per-function debug information.
2532214ed89SReid Kleckner   for (auto &P : FnDebugInfo)
254577be0feSDavid Majnemer     if (!P.first->isDeclarationForLinker())
2552214ed89SReid Kleckner       emitDebugInfoForFunction(P.first, P.second);
25670f5bc99SReid Kleckner 
2576f3406dfSReid Kleckner   // Emit global variable debug information.
2583128b10cSDavid Majnemer   setCurrentSubprogram(nullptr);
2596f3406dfSReid Kleckner   emitDebugInfoForGlobals();
2606f3406dfSReid Kleckner 
2615d122f87SReid Kleckner   // Switch back to the generic .debug$S section after potentially processing
2625d122f87SReid Kleckner   // comdat symbol sections.
2635d122f87SReid Kleckner   switchToDebugSectionForSymbol(nullptr);
2645d122f87SReid Kleckner 
2653128b10cSDavid Majnemer   // Emit UDT records for any types used by global variables.
2663128b10cSDavid Majnemer   if (!GlobalUDTs.empty()) {
2673128b10cSDavid Majnemer     MCSymbol *SymbolsEnd = beginCVSubsection(ModuleSubstreamKind::Symbols);
2683128b10cSDavid Majnemer     emitDebugInfoForUDTs(GlobalUDTs);
2693128b10cSDavid Majnemer     endCVSubsection(SymbolsEnd);
2703128b10cSDavid Majnemer   }
2713128b10cSDavid Majnemer 
27270f5bc99SReid Kleckner   // This subsection holds a file index to offset in string table table.
273dac21b43SReid Kleckner   OS.AddComment("File index to string table offset subsection");
274dac21b43SReid Kleckner   OS.EmitCVFileChecksumsDirective();
27570f5bc99SReid Kleckner 
27670f5bc99SReid Kleckner   // This subsection holds the string table.
277dac21b43SReid Kleckner   OS.AddComment("String table");
278dac21b43SReid Kleckner   OS.EmitCVStringTableDirective();
27970f5bc99SReid Kleckner 
2805acacbb0SReid Kleckner   // Emit type information last, so that any types we translate while emitting
2815acacbb0SReid Kleckner   // function info are included.
2825acacbb0SReid Kleckner   emitTypeInformation();
2835acacbb0SReid Kleckner 
28470f5bc99SReid Kleckner   clear();
28570f5bc99SReid Kleckner }
28670f5bc99SReid Kleckner 
287b9456a5eSDavid Majnemer static void emitNullTerminatedSymbolName(MCStreamer &OS, StringRef S) {
288b9456a5eSDavid Majnemer   // Microsoft's linker seems to have trouble with symbol names longer than
289b9456a5eSDavid Majnemer   // 0xffd8 bytes.
290b9456a5eSDavid Majnemer   S = S.substr(0, 0xffd8);
291b9456a5eSDavid Majnemer   SmallString<32> NullTerminatedString(S);
292b9456a5eSDavid Majnemer   NullTerminatedString.push_back('\0');
293b9456a5eSDavid Majnemer   OS.EmitBytes(NullTerminatedString);
294b9456a5eSDavid Majnemer }
295b9456a5eSDavid Majnemer 
296f3b9ba49SReid Kleckner void CodeViewDebug::emitTypeInformation() {
2972280f932SReid Kleckner   // Do nothing if we have no debug info or if no non-trivial types were emitted
2982280f932SReid Kleckner   // to TypeTable during codegen.
299fbd7787dSReid Kleckner   NamedMDNode *CU_Nodes =
300fbd7787dSReid Kleckner       MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
301fbd7787dSReid Kleckner   if (!CU_Nodes)
302fbd7787dSReid Kleckner     return;
3032280f932SReid Kleckner   if (TypeTable.empty())
304fbd7787dSReid Kleckner     return;
305fbd7787dSReid Kleckner 
306f3b9ba49SReid Kleckner   // Start the .debug$T section with 0x4.
307dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugTypesSection());
3085d122f87SReid Kleckner   emitCodeViewMagicVersion();
309f3b9ba49SReid Kleckner 
310fbdbe9e2SReid Kleckner   SmallString<8> CommentPrefix;
311fbdbe9e2SReid Kleckner   if (OS.isVerboseAsm()) {
312fbdbe9e2SReid Kleckner     CommentPrefix += '\t';
313fbdbe9e2SReid Kleckner     CommentPrefix += Asm->MAI->getCommentString();
314fbdbe9e2SReid Kleckner     CommentPrefix += ' ';
315fbdbe9e2SReid Kleckner   }
316fbdbe9e2SReid Kleckner 
317fbdbe9e2SReid Kleckner   CVTypeDumper CVTD(nullptr, /*PrintRecordBytes=*/false);
3182280f932SReid Kleckner   TypeTable.ForEachRecord(
319fbdbe9e2SReid Kleckner       [&](TypeIndex Index, StringRef Record) {
320fbdbe9e2SReid Kleckner         if (OS.isVerboseAsm()) {
321fbdbe9e2SReid Kleckner           // Emit a block comment describing the type record for readability.
322fbdbe9e2SReid Kleckner           SmallString<512> CommentBlock;
323fbdbe9e2SReid Kleckner           raw_svector_ostream CommentOS(CommentBlock);
324fbdbe9e2SReid Kleckner           ScopedPrinter SP(CommentOS);
325fbdbe9e2SReid Kleckner           SP.setPrefix(CommentPrefix);
326fbdbe9e2SReid Kleckner           CVTD.setPrinter(&SP);
32701ee3daeSZachary Turner           Error EC = CVTD.dump({Record.bytes_begin(), Record.bytes_end()});
32801ee3daeSZachary Turner           assert(!EC && "produced malformed type record");
32901ee3daeSZachary Turner           consumeError(std::move(EC));
330fbdbe9e2SReid Kleckner           // emitRawComment will insert its own tab and comment string before
331fbdbe9e2SReid Kleckner           // the first line, so strip off our first one. It also prints its own
332fbdbe9e2SReid Kleckner           // newline.
333fbdbe9e2SReid Kleckner           OS.emitRawComment(
334fbdbe9e2SReid Kleckner               CommentOS.str().drop_front(CommentPrefix.size() - 1).rtrim());
335fbdbe9e2SReid Kleckner         }
336fbdbe9e2SReid Kleckner         OS.EmitBinaryData(Record);
3372280f932SReid Kleckner       });
338f3b9ba49SReid Kleckner }
339f3b9ba49SReid Kleckner 
3405d122f87SReid Kleckner void CodeViewDebug::emitInlineeLinesSubsection() {
3411fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
3421fcd610cSReid Kleckner     return;
3431fcd610cSReid Kleckner 
3441fcd610cSReid Kleckner 
3451fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
3466f3406dfSReid Kleckner   MCSymbol *InlineEnd = beginCVSubsection(ModuleSubstreamKind::InlineeLines);
3471fcd610cSReid Kleckner 
3481fcd610cSReid Kleckner   // We don't provide any extra file info.
3491fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
35030579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
3511fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
3521fcd610cSReid Kleckner 
3531fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
3542280f932SReid Kleckner     assert(TypeIndices.count(SP));
3552280f932SReid Kleckner     TypeIndex InlineeIdx = TypeIndices[SP];
3562280f932SReid Kleckner 
35730579ec8SDavid Majnemer     OS.AddBlankLine();
3581fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
3591fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
3601fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
36130579ec8SDavid Majnemer     OS.AddBlankLine();
3621fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
3631fcd610cSReid Kleckner     // 1.
3641fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
36530579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
3662280f932SReid Kleckner     OS.EmitIntValue(InlineeIdx.getIndex(), 4);
36730579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
3681fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
36930579ec8SDavid Majnemer     OS.AddComment("Starting line number");
3701fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
3711fcd610cSReid Kleckner   }
3721fcd610cSReid Kleckner 
3736f3406dfSReid Kleckner   endCVSubsection(InlineEnd);
3741fcd610cSReid Kleckner }
3751fcd610cSReid Kleckner 
3761fcd610cSReid Kleckner void CodeViewDebug::collectInlineSiteChildren(
3771fcd610cSReid Kleckner     SmallVectorImpl<unsigned> &Children, const FunctionInfo &FI,
3781fcd610cSReid Kleckner     const InlineSite &Site) {
3791fcd610cSReid Kleckner   for (const DILocation *ChildSiteLoc : Site.ChildSites) {
3801fcd610cSReid Kleckner     auto I = FI.InlineSites.find(ChildSiteLoc);
3811fcd610cSReid Kleckner     const InlineSite &ChildSite = I->second;
3821fcd610cSReid Kleckner     Children.push_back(ChildSite.SiteFuncId);
3831fcd610cSReid Kleckner     collectInlineSiteChildren(Children, FI, ChildSite);
3841fcd610cSReid Kleckner   }
3851fcd610cSReid Kleckner }
3861fcd610cSReid Kleckner 
387f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
388f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
389f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
390f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
391f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
392f3b9ba49SReid Kleckner 
3932280f932SReid Kleckner   assert(TypeIndices.count(Site.Inlinee));
3942280f932SReid Kleckner   TypeIndex InlineeIdx = TypeIndices[Site.Inlinee];
395f3b9ba49SReid Kleckner 
396f3b9ba49SReid Kleckner   // SymbolRecord
397dac21b43SReid Kleckner   OS.AddComment("Record length");
398eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
399f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
400dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
40163a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE, 2); // RecordKind
402f3b9ba49SReid Kleckner 
403dac21b43SReid Kleckner   OS.AddComment("PtrParent");
404dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
405dac21b43SReid Kleckner   OS.AddComment("PtrEnd");
406dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
407dac21b43SReid Kleckner   OS.AddComment("Inlinee type index");
4082280f932SReid Kleckner   OS.EmitIntValue(InlineeIdx.getIndex(), 4);
409f3b9ba49SReid Kleckner 
4101fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
4111fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
4121fcd610cSReid Kleckner   SmallVector<unsigned, 3> SecondaryFuncIds;
4131fcd610cSReid Kleckner   collectInlineSiteChildren(SecondaryFuncIds, FI, Site);
4141fcd610cSReid Kleckner 
4151fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
416c9911f28SDavid Majnemer                                     FI.Begin, FI.End, SecondaryFuncIds);
417f3b9ba49SReid Kleckner 
418f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
419f3b9ba49SReid Kleckner 
420f9c275feSReid Kleckner   for (const LocalVariable &Var : Site.InlinedLocals)
421f9c275feSReid Kleckner     emitLocalVariable(Var);
422f9c275feSReid Kleckner 
423f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
424f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
425f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
426f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
427f3b9ba49SReid Kleckner            "child site not in function inline site map");
428f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
429f3b9ba49SReid Kleckner   }
430f3b9ba49SReid Kleckner 
431f3b9ba49SReid Kleckner   // Close the scope.
432dac21b43SReid Kleckner   OS.AddComment("Record length");
433dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
434dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
43563a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE_END, 2); // RecordKind
436f3b9ba49SReid Kleckner }
437f3b9ba49SReid Kleckner 
4385d122f87SReid Kleckner void CodeViewDebug::switchToDebugSectionForSymbol(const MCSymbol *GVSym) {
4395d122f87SReid Kleckner   // If we have a symbol, it may be in a section that is COMDAT. If so, find the
4405d122f87SReid Kleckner   // comdat key. A section may be comdat because of -ffunction-sections or
4415d122f87SReid Kleckner   // because it is comdat in the IR.
4425d122f87SReid Kleckner   MCSectionCOFF *GVSec =
4435d122f87SReid Kleckner       GVSym ? dyn_cast<MCSectionCOFF>(&GVSym->getSection()) : nullptr;
4445d122f87SReid Kleckner   const MCSymbol *KeySym = GVSec ? GVSec->getCOMDATSymbol() : nullptr;
4455d122f87SReid Kleckner 
4465d122f87SReid Kleckner   MCSectionCOFF *DebugSec = cast<MCSectionCOFF>(
4475d122f87SReid Kleckner       Asm->getObjFileLowering().getCOFFDebugSymbolsSection());
4485d122f87SReid Kleckner   DebugSec = OS.getContext().getAssociativeCOFFSection(DebugSec, KeySym);
4495d122f87SReid Kleckner 
4505d122f87SReid Kleckner   OS.SwitchSection(DebugSec);
4515d122f87SReid Kleckner 
4525d122f87SReid Kleckner   // Emit the magic version number if this is the first time we've switched to
4535d122f87SReid Kleckner   // this section.
4545d122f87SReid Kleckner   if (ComdatDebugSections.insert(DebugSec).second)
4555d122f87SReid Kleckner     emitCodeViewMagicVersion();
4565d122f87SReid Kleckner }
4575d122f87SReid Kleckner 
458ac945e27SReid Kleckner static const DISubprogram *getQualifiedNameComponents(
459ac945e27SReid Kleckner     const DIScope *Scope, SmallVectorImpl<StringRef> &QualifiedNameComponents) {
460ac945e27SReid Kleckner   const DISubprogram *ClosestSubprogram = nullptr;
461ac945e27SReid Kleckner   while (Scope != nullptr) {
462ac945e27SReid Kleckner     if (ClosestSubprogram == nullptr)
463ac945e27SReid Kleckner       ClosestSubprogram = dyn_cast<DISubprogram>(Scope);
464ac945e27SReid Kleckner     StringRef ScopeName = Scope->getName();
465ac945e27SReid Kleckner     if (!ScopeName.empty())
466ac945e27SReid Kleckner       QualifiedNameComponents.push_back(ScopeName);
467ac945e27SReid Kleckner     Scope = Scope->getScope().resolve();
468ac945e27SReid Kleckner   }
469ac945e27SReid Kleckner   return ClosestSubprogram;
470ac945e27SReid Kleckner }
471ac945e27SReid Kleckner 
472ac945e27SReid Kleckner static std::string getQualifiedName(ArrayRef<StringRef> QualifiedNameComponents,
473ac945e27SReid Kleckner                                     StringRef TypeName) {
474ac945e27SReid Kleckner   std::string FullyQualifiedName;
475ac945e27SReid Kleckner   for (StringRef QualifiedNameComponent : reverse(QualifiedNameComponents)) {
476ac945e27SReid Kleckner     FullyQualifiedName.append(QualifiedNameComponent);
477ac945e27SReid Kleckner     FullyQualifiedName.append("::");
478ac945e27SReid Kleckner   }
479ac945e27SReid Kleckner   FullyQualifiedName.append(TypeName);
480ac945e27SReid Kleckner   return FullyQualifiedName;
481ac945e27SReid Kleckner }
482ac945e27SReid Kleckner 
4832214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
4842214ed89SReid Kleckner                                              FunctionInfo &FI) {
48570f5bc99SReid Kleckner   // For each function there is a separate subsection
48670f5bc99SReid Kleckner   // which holds the PC to file:line table.
48770f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
48870f5bc99SReid Kleckner   assert(Fn);
48970f5bc99SReid Kleckner 
4905d122f87SReid Kleckner   // Switch to the to a comdat section, if appropriate.
4915d122f87SReid Kleckner   switchToDebugSectionForSymbol(Fn);
4925d122f87SReid Kleckner 
493ac945e27SReid Kleckner   std::string FuncName;
4943128b10cSDavid Majnemer   auto *SP = GV->getSubprogram();
4953128b10cSDavid Majnemer   setCurrentSubprogram(SP);
496ac945e27SReid Kleckner 
497ac945e27SReid Kleckner   // If we have a display name, build the fully qualified name by walking the
498ac945e27SReid Kleckner   // chain of scopes.
499ac945e27SReid Kleckner   if (SP != nullptr && !SP->getDisplayName().empty()) {
500ac945e27SReid Kleckner     SmallVector<StringRef, 5> QualifiedNameComponents;
501ac945e27SReid Kleckner     getQualifiedNameComponents(SP->getScope().resolve(),
502ac945e27SReid Kleckner                                QualifiedNameComponents);
503ac945e27SReid Kleckner     FuncName = getQualifiedName(QualifiedNameComponents, SP->getDisplayName());
504ac945e27SReid Kleckner   }
50570f5bc99SReid Kleckner 
50670f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
50770f5bc99SReid Kleckner   if (FuncName.empty())
50870f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
50970f5bc99SReid Kleckner 
51070f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
511dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
5126f3406dfSReid Kleckner   MCSymbol *SymbolsEnd = beginCVSubsection(ModuleSubstreamKind::Symbols);
51370f5bc99SReid Kleckner   {
514f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
515f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
516dac21b43SReid Kleckner     OS.AddComment("Record length");
517eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
518dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
51970f5bc99SReid Kleckner 
520dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
52163a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_GPROC32_ID), 2);
52270f5bc99SReid Kleckner 
52330579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
524dac21b43SReid Kleckner     OS.AddComment("PtrParent");
525dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
526dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
527dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
528dac21b43SReid Kleckner     OS.AddComment("PtrNext");
529dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
53070f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
53170f5bc99SReid Kleckner     // code is located and what's its size:
532dac21b43SReid Kleckner     OS.AddComment("Code size");
533eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
534dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
535dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
536dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
537dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
538dac21b43SReid Kleckner     OS.AddComment("Function type index");
53975c3ebfaSDavid Majnemer     OS.EmitIntValue(getFuncIdForSubprogram(GV->getSubprogram()).getIndex(), 4);
540dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
541dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
542dac21b43SReid Kleckner     OS.AddComment("Function section index");
543dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
544dac21b43SReid Kleckner     OS.AddComment("Flags");
545dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
54670f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
547dac21b43SReid Kleckner     OS.AddComment("Function name");
5481256125fSDavid Majnemer     // Truncate the name so we won't overflow the record length field.
549b9456a5eSDavid Majnemer     emitNullTerminatedSymbolName(OS, FuncName);
550dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
55170f5bc99SReid Kleckner 
552f9c275feSReid Kleckner     for (const LocalVariable &Var : FI.Locals)
553f9c275feSReid Kleckner       emitLocalVariable(Var);
554f9c275feSReid Kleckner 
555f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
556f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
557f3b9ba49SReid Kleckner     // of their parent inline site.
558f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
559f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
560f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
561f9c275feSReid Kleckner              "child site not in function inline site map");
562f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
563f3b9ba49SReid Kleckner     }
564f3b9ba49SReid Kleckner 
5653128b10cSDavid Majnemer     if (SP != nullptr)
5663128b10cSDavid Majnemer       emitDebugInfoForUDTs(LocalUDTs);
5673128b10cSDavid Majnemer 
56870f5bc99SReid Kleckner     // We're done with this function.
569dac21b43SReid Kleckner     OS.AddComment("Record length");
570dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
571dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
57263a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_PROC_ID_END), 2);
57370f5bc99SReid Kleckner   }
5746f3406dfSReid Kleckner   endCVSubsection(SymbolsEnd);
57570f5bc99SReid Kleckner 
5762214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
577dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
57870f5bc99SReid Kleckner }
57970f5bc99SReid Kleckner 
580876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
581876330d5SReid Kleckner CodeViewDebug::createDefRangeMem(uint16_t CVRegister, int Offset) {
582876330d5SReid Kleckner   LocalVarDefRange DR;
583c6a2f214SAaron Ballman   DR.InMemory = -1;
584876330d5SReid Kleckner   DR.DataOffset = Offset;
585876330d5SReid Kleckner   assert(DR.DataOffset == Offset && "truncation");
586876330d5SReid Kleckner   DR.StructOffset = 0;
587876330d5SReid Kleckner   DR.CVRegister = CVRegister;
588876330d5SReid Kleckner   return DR;
589876330d5SReid Kleckner }
590876330d5SReid Kleckner 
591876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
592876330d5SReid Kleckner CodeViewDebug::createDefRangeReg(uint16_t CVRegister) {
593876330d5SReid Kleckner   LocalVarDefRange DR;
594876330d5SReid Kleckner   DR.InMemory = 0;
595876330d5SReid Kleckner   DR.DataOffset = 0;
596876330d5SReid Kleckner   DR.StructOffset = 0;
597876330d5SReid Kleckner   DR.CVRegister = CVRegister;
598876330d5SReid Kleckner   return DR;
599876330d5SReid Kleckner }
600876330d5SReid Kleckner 
601876330d5SReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable(
602876330d5SReid Kleckner     DenseSet<InlinedVariable> &Processed) {
603876330d5SReid Kleckner   const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
604876330d5SReid Kleckner   const TargetFrameLowering *TFI = TSI.getFrameLowering();
605876330d5SReid Kleckner   const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
606876330d5SReid Kleckner 
607876330d5SReid Kleckner   for (const MachineModuleInfo::VariableDbgInfo &VI :
608876330d5SReid Kleckner        MMI->getVariableDbgInfo()) {
609f9c275feSReid Kleckner     if (!VI.Var)
610f9c275feSReid Kleckner       continue;
611f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
612f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
613f9c275feSReid Kleckner 
614876330d5SReid Kleckner     Processed.insert(InlinedVariable(VI.Var, VI.Loc->getInlinedAt()));
615f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
616f9c275feSReid Kleckner 
617f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
618f9c275feSReid Kleckner     if (!Scope)
619f9c275feSReid Kleckner       continue;
620f9c275feSReid Kleckner 
621f9c275feSReid Kleckner     // Get the frame register used and the offset.
622f9c275feSReid Kleckner     unsigned FrameReg = 0;
623876330d5SReid Kleckner     int FrameOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
624876330d5SReid Kleckner     uint16_t CVReg = TRI->getCodeViewRegNum(FrameReg);
625f9c275feSReid Kleckner 
626f9c275feSReid Kleckner     // Calculate the label ranges.
627876330d5SReid Kleckner     LocalVarDefRange DefRange = createDefRangeMem(CVReg, FrameOffset);
628f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
629f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
630f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
631876330d5SReid Kleckner       End = End ? End : Asm->getFunctionEnd();
632876330d5SReid Kleckner       DefRange.Ranges.emplace_back(Begin, End);
633f9c275feSReid Kleckner     }
634f9c275feSReid Kleckner 
635876330d5SReid Kleckner     LocalVariable Var;
636876330d5SReid Kleckner     Var.DIVar = VI.Var;
637876330d5SReid Kleckner     Var.DefRanges.emplace_back(std::move(DefRange));
638876330d5SReid Kleckner     recordLocalVariable(std::move(Var), VI.Loc->getInlinedAt());
639f9c275feSReid Kleckner   }
640f9c275feSReid Kleckner }
641876330d5SReid Kleckner 
642876330d5SReid Kleckner void CodeViewDebug::collectVariableInfo(const DISubprogram *SP) {
643876330d5SReid Kleckner   DenseSet<InlinedVariable> Processed;
644876330d5SReid Kleckner   // Grab the variable info that was squirreled away in the MMI side-table.
645876330d5SReid Kleckner   collectVariableInfoFromMMITable(Processed);
646876330d5SReid Kleckner 
647876330d5SReid Kleckner   const TargetRegisterInfo *TRI = Asm->MF->getSubtarget().getRegisterInfo();
648876330d5SReid Kleckner 
649876330d5SReid Kleckner   for (const auto &I : DbgValues) {
650876330d5SReid Kleckner     InlinedVariable IV = I.first;
651876330d5SReid Kleckner     if (Processed.count(IV))
652876330d5SReid Kleckner       continue;
653876330d5SReid Kleckner     const DILocalVariable *DIVar = IV.first;
654876330d5SReid Kleckner     const DILocation *InlinedAt = IV.second;
655876330d5SReid Kleckner 
656876330d5SReid Kleckner     // Instruction ranges, specifying where IV is accessible.
657876330d5SReid Kleckner     const auto &Ranges = I.second;
658876330d5SReid Kleckner 
659876330d5SReid Kleckner     LexicalScope *Scope = nullptr;
660876330d5SReid Kleckner     if (InlinedAt)
661876330d5SReid Kleckner       Scope = LScopes.findInlinedScope(DIVar->getScope(), InlinedAt);
662876330d5SReid Kleckner     else
663876330d5SReid Kleckner       Scope = LScopes.findLexicalScope(DIVar->getScope());
664876330d5SReid Kleckner     // If variable scope is not found then skip this variable.
665876330d5SReid Kleckner     if (!Scope)
666876330d5SReid Kleckner       continue;
667876330d5SReid Kleckner 
668876330d5SReid Kleckner     LocalVariable Var;
669876330d5SReid Kleckner     Var.DIVar = DIVar;
670876330d5SReid Kleckner 
671876330d5SReid Kleckner     // Calculate the definition ranges.
672876330d5SReid Kleckner     for (auto I = Ranges.begin(), E = Ranges.end(); I != E; ++I) {
673876330d5SReid Kleckner       const InsnRange &Range = *I;
674876330d5SReid Kleckner       const MachineInstr *DVInst = Range.first;
675876330d5SReid Kleckner       assert(DVInst->isDebugValue() && "Invalid History entry");
676876330d5SReid Kleckner       const DIExpression *DIExpr = DVInst->getDebugExpression();
677876330d5SReid Kleckner 
678876330d5SReid Kleckner       // Bail if there is a complex DWARF expression for now.
679876330d5SReid Kleckner       if (DIExpr && DIExpr->getNumElements() > 0)
680876330d5SReid Kleckner         continue;
681876330d5SReid Kleckner 
6829a593ee7SReid Kleckner       // Bail if operand 0 is not a valid register. This means the variable is a
6839a593ee7SReid Kleckner       // simple constant, or is described by a complex expression.
6849a593ee7SReid Kleckner       // FIXME: Find a way to represent constant variables, since they are
6859a593ee7SReid Kleckner       // relatively common.
6869a593ee7SReid Kleckner       unsigned Reg =
6879a593ee7SReid Kleckner           DVInst->getOperand(0).isReg() ? DVInst->getOperand(0).getReg() : 0;
6889a593ee7SReid Kleckner       if (Reg == 0)
6896e0d5f57SReid Kleckner         continue;
6906e0d5f57SReid Kleckner 
691876330d5SReid Kleckner       // Handle the two cases we can handle: indirect in memory and in register.
692876330d5SReid Kleckner       bool IsIndirect = DVInst->getOperand(1).isImm();
693876330d5SReid Kleckner       unsigned CVReg = TRI->getCodeViewRegNum(DVInst->getOperand(0).getReg());
694876330d5SReid Kleckner       {
695876330d5SReid Kleckner         LocalVarDefRange DefRange;
696876330d5SReid Kleckner         if (IsIndirect) {
697876330d5SReid Kleckner           int64_t Offset = DVInst->getOperand(1).getImm();
698876330d5SReid Kleckner           DefRange = createDefRangeMem(CVReg, Offset);
699876330d5SReid Kleckner         } else {
700876330d5SReid Kleckner           DefRange = createDefRangeReg(CVReg);
701876330d5SReid Kleckner         }
702876330d5SReid Kleckner         if (Var.DefRanges.empty() ||
703876330d5SReid Kleckner             Var.DefRanges.back().isDifferentLocation(DefRange)) {
704876330d5SReid Kleckner           Var.DefRanges.emplace_back(std::move(DefRange));
705876330d5SReid Kleckner         }
706876330d5SReid Kleckner       }
707876330d5SReid Kleckner 
708876330d5SReid Kleckner       // Compute the label range.
709876330d5SReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
710876330d5SReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
711876330d5SReid Kleckner       if (!End) {
712876330d5SReid Kleckner         if (std::next(I) != E)
713876330d5SReid Kleckner           End = getLabelBeforeInsn(std::next(I)->first);
714876330d5SReid Kleckner         else
715876330d5SReid Kleckner           End = Asm->getFunctionEnd();
716876330d5SReid Kleckner       }
717876330d5SReid Kleckner 
718876330d5SReid Kleckner       // If the last range end is our begin, just extend the last range.
719876330d5SReid Kleckner       // Otherwise make a new range.
720876330d5SReid Kleckner       SmallVectorImpl<std::pair<const MCSymbol *, const MCSymbol *>> &Ranges =
721876330d5SReid Kleckner           Var.DefRanges.back().Ranges;
722876330d5SReid Kleckner       if (!Ranges.empty() && Ranges.back().second == Begin)
723876330d5SReid Kleckner         Ranges.back().second = End;
724876330d5SReid Kleckner       else
725876330d5SReid Kleckner         Ranges.emplace_back(Begin, End);
726876330d5SReid Kleckner 
727876330d5SReid Kleckner       // FIXME: Do more range combining.
728876330d5SReid Kleckner     }
729876330d5SReid Kleckner 
730876330d5SReid Kleckner     recordLocalVariable(std::move(Var), InlinedAt);
731876330d5SReid Kleckner   }
732f9c275feSReid Kleckner }
733f9c275feSReid Kleckner 
73470f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
73570f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
73670f5bc99SReid Kleckner 
737f9c275feSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
73870f5bc99SReid Kleckner     return;
73970f5bc99SReid Kleckner 
740f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
741f9c275feSReid Kleckner 
74270f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
74370f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
74470f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
7452214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
7461fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
74770f5bc99SReid Kleckner 
748f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
749f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
75070f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
75170f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
75270f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
75370f5bc99SReid Kleckner   bool EmptyPrologue = true;
75470f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
75570f5bc99SReid Kleckner     for (const auto &MI : MBB) {
756f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
757f9c275feSReid Kleckner           MI.getDebugLoc()) {
75870f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
75970f5bc99SReid Kleckner         break;
760f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
76170f5bc99SReid Kleckner         EmptyPrologue = false;
76270f5bc99SReid Kleckner       }
76370f5bc99SReid Kleckner     }
764f9c275feSReid Kleckner   }
765f9c275feSReid Kleckner 
76670f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
76770f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
76870f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
76970f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
77070f5bc99SReid Kleckner   }
77170f5bc99SReid Kleckner }
77270f5bc99SReid Kleckner 
7735acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerType(const DIType *Ty) {
7745acacbb0SReid Kleckner   // Generic dispatch for lowering an unknown type.
7755acacbb0SReid Kleckner   switch (Ty->getTag()) {
776f3c3c132SAdrian McCarthy   case dwarf::DW_TAG_array_type:
777f3c3c132SAdrian McCarthy     return lowerTypeArray(cast<DICompositeType>(Ty));
778d065e23dSDavid Majnemer   case dwarf::DW_TAG_typedef:
779d065e23dSDavid Majnemer     return lowerTypeAlias(cast<DIDerivedType>(Ty));
7805acacbb0SReid Kleckner   case dwarf::DW_TAG_base_type:
7815acacbb0SReid Kleckner     return lowerTypeBasic(cast<DIBasicType>(Ty));
7825acacbb0SReid Kleckner   case dwarf::DW_TAG_pointer_type:
7835acacbb0SReid Kleckner   case dwarf::DW_TAG_reference_type:
7845acacbb0SReid Kleckner   case dwarf::DW_TAG_rvalue_reference_type:
7855acacbb0SReid Kleckner     return lowerTypePointer(cast<DIDerivedType>(Ty));
7865acacbb0SReid Kleckner   case dwarf::DW_TAG_ptr_to_member_type:
7875acacbb0SReid Kleckner     return lowerTypeMemberPointer(cast<DIDerivedType>(Ty));
7885acacbb0SReid Kleckner   case dwarf::DW_TAG_const_type:
7895acacbb0SReid Kleckner   case dwarf::DW_TAG_volatile_type:
7905acacbb0SReid Kleckner     return lowerTypeModifier(cast<DIDerivedType>(Ty));
79175c3ebfaSDavid Majnemer   case dwarf::DW_TAG_subroutine_type:
79275c3ebfaSDavid Majnemer     return lowerTypeFunction(cast<DISubroutineType>(Ty));
793979cb888SDavid Majnemer   case dwarf::DW_TAG_enumeration_type:
794979cb888SDavid Majnemer     return lowerTypeEnum(cast<DICompositeType>(Ty));
795a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
796a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
797a8d57407SReid Kleckner     return lowerTypeClass(cast<DICompositeType>(Ty));
798a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
799a8d57407SReid Kleckner     return lowerTypeUnion(cast<DICompositeType>(Ty));
8005acacbb0SReid Kleckner   default:
8015acacbb0SReid Kleckner     // Use the null type index.
8025acacbb0SReid Kleckner     return TypeIndex();
8035acacbb0SReid Kleckner   }
8045acacbb0SReid Kleckner }
8055acacbb0SReid Kleckner 
806d065e23dSDavid Majnemer TypeIndex CodeViewDebug::lowerTypeAlias(const DIDerivedType *Ty) {
807d065e23dSDavid Majnemer   DITypeRef UnderlyingTypeRef = Ty->getBaseType();
808d065e23dSDavid Majnemer   TypeIndex UnderlyingTypeIndex = getTypeIndex(UnderlyingTypeRef);
8093128b10cSDavid Majnemer   StringRef TypeName = Ty->getName();
8103128b10cSDavid Majnemer 
8113128b10cSDavid Majnemer   SmallVector<StringRef, 5> QualifiedNameComponents;
8123128b10cSDavid Majnemer   const DISubprogram *ClosestSubprogram = getQualifiedNameComponents(
8133128b10cSDavid Majnemer       Ty->getScope().resolve(), QualifiedNameComponents);
8143128b10cSDavid Majnemer 
8153128b10cSDavid Majnemer   if (ClosestSubprogram == nullptr) {
8163128b10cSDavid Majnemer     std::string FullyQualifiedName =
8173128b10cSDavid Majnemer         getQualifiedName(QualifiedNameComponents, TypeName);
8183128b10cSDavid Majnemer     GlobalUDTs.emplace_back(std::move(FullyQualifiedName), UnderlyingTypeIndex);
8193128b10cSDavid Majnemer   } else if (ClosestSubprogram == CurrentSubprogram) {
8203128b10cSDavid Majnemer     std::string FullyQualifiedName =
8213128b10cSDavid Majnemer         getQualifiedName(QualifiedNameComponents, TypeName);
8223128b10cSDavid Majnemer     LocalUDTs.emplace_back(std::move(FullyQualifiedName), UnderlyingTypeIndex);
8233128b10cSDavid Majnemer   }
8243128b10cSDavid Majnemer   // TODO: What if the ClosestSubprogram is neither null or the current
8253128b10cSDavid Majnemer   // subprogram?  Currently, the UDT just gets dropped on the floor.
8263128b10cSDavid Majnemer   //
8273128b10cSDavid Majnemer   // The current behavior is not desirable.  To get maximal fidelity, we would
8283128b10cSDavid Majnemer   // need to perform all type translation before beginning emission of .debug$S
8293128b10cSDavid Majnemer   // and then make LocalUDTs a member of FunctionInfo
8303128b10cSDavid Majnemer 
831d065e23dSDavid Majnemer   if (UnderlyingTypeIndex == TypeIndex(SimpleTypeKind::Int32Long) &&
8323128b10cSDavid Majnemer       TypeName == "HRESULT")
833d065e23dSDavid Majnemer     return TypeIndex(SimpleTypeKind::HResult);
8348c46a4ceSDavid Majnemer   if (UnderlyingTypeIndex == TypeIndex(SimpleTypeKind::UInt16Short) &&
8353128b10cSDavid Majnemer       TypeName == "wchar_t")
8368c46a4ceSDavid Majnemer     return TypeIndex(SimpleTypeKind::WideCharacter);
837d065e23dSDavid Majnemer   return UnderlyingTypeIndex;
838d065e23dSDavid Majnemer }
839d065e23dSDavid Majnemer 
840f3c3c132SAdrian McCarthy TypeIndex CodeViewDebug::lowerTypeArray(const DICompositeType *Ty) {
841f3c3c132SAdrian McCarthy   DITypeRef ElementTypeRef = Ty->getBaseType();
842f3c3c132SAdrian McCarthy   TypeIndex ElementTypeIndex = getTypeIndex(ElementTypeRef);
843f3c3c132SAdrian McCarthy   // IndexType is size_t, which depends on the bitness of the target.
844f3c3c132SAdrian McCarthy   TypeIndex IndexType = Asm->MAI->getPointerSize() == 8
845f3c3c132SAdrian McCarthy                             ? TypeIndex(SimpleTypeKind::UInt64Quad)
846f3c3c132SAdrian McCarthy                             : TypeIndex(SimpleTypeKind::UInt32Long);
847f3c3c132SAdrian McCarthy   uint64_t Size = Ty->getSizeInBits() / 8;
848f3c3c132SAdrian McCarthy   ArrayRecord Record(ElementTypeIndex, IndexType, Size, Ty->getName());
849f3c3c132SAdrian McCarthy   return TypeTable.writeArray(Record);
850f3c3c132SAdrian McCarthy }
851f3c3c132SAdrian McCarthy 
8525acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeBasic(const DIBasicType *Ty) {
8535acacbb0SReid Kleckner   TypeIndex Index;
8545acacbb0SReid Kleckner   dwarf::TypeKind Kind;
8555acacbb0SReid Kleckner   uint32_t ByteSize;
8565acacbb0SReid Kleckner 
8575acacbb0SReid Kleckner   Kind = static_cast<dwarf::TypeKind>(Ty->getEncoding());
858afefa673SDavid Majnemer   ByteSize = Ty->getSizeInBits() / 8;
8595acacbb0SReid Kleckner 
8605acacbb0SReid Kleckner   SimpleTypeKind STK = SimpleTypeKind::None;
8615acacbb0SReid Kleckner   switch (Kind) {
8625acacbb0SReid Kleckner   case dwarf::DW_ATE_address:
8635acacbb0SReid Kleckner     // FIXME: Translate
8645acacbb0SReid Kleckner     break;
8655acacbb0SReid Kleckner   case dwarf::DW_ATE_boolean:
8665acacbb0SReid Kleckner     switch (ByteSize) {
8675acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::Boolean8;   break;
8685acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::Boolean16;  break;
8695acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Boolean32;  break;
8705acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Boolean64;  break;
8711c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::Boolean128; break;
8725acacbb0SReid Kleckner     }
8735acacbb0SReid Kleckner     break;
8745acacbb0SReid Kleckner   case dwarf::DW_ATE_complex_float:
8755acacbb0SReid Kleckner     switch (ByteSize) {
8761c2cb1ddSDavid Majnemer     case 2:  STK = SimpleTypeKind::Complex16;  break;
8775acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Complex32;  break;
8785acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Complex64;  break;
8795acacbb0SReid Kleckner     case 10: STK = SimpleTypeKind::Complex80;  break;
8805acacbb0SReid Kleckner     case 16: STK = SimpleTypeKind::Complex128; break;
8815acacbb0SReid Kleckner     }
8825acacbb0SReid Kleckner     break;
8835acacbb0SReid Kleckner   case dwarf::DW_ATE_float:
8845acacbb0SReid Kleckner     switch (ByteSize) {
8851c2cb1ddSDavid Majnemer     case 2:  STK = SimpleTypeKind::Float16;  break;
8865acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Float32;  break;
8875acacbb0SReid Kleckner     case 6:  STK = SimpleTypeKind::Float48;  break;
8885acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Float64;  break;
8895acacbb0SReid Kleckner     case 10: STK = SimpleTypeKind::Float80;  break;
8905acacbb0SReid Kleckner     case 16: STK = SimpleTypeKind::Float128; break;
8915acacbb0SReid Kleckner     }
8925acacbb0SReid Kleckner     break;
8935acacbb0SReid Kleckner   case dwarf::DW_ATE_signed:
8945acacbb0SReid Kleckner     switch (ByteSize) {
8955acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::SByte;      break;
8965acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::Int16Short; break;
8975acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Int32;      break;
8981c2cb1ddSDavid Majnemer     case 8:  STK = SimpleTypeKind::Int64Quad;  break;
8991c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::Int128Oct;  break;
9005acacbb0SReid Kleckner     }
9015acacbb0SReid Kleckner     break;
9025acacbb0SReid Kleckner   case dwarf::DW_ATE_unsigned:
9035acacbb0SReid Kleckner     switch (ByteSize) {
9045acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::Byte;        break;
9055acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::UInt16Short; break;
9065acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::UInt32;      break;
9071c2cb1ddSDavid Majnemer     case 8:  STK = SimpleTypeKind::UInt64Quad;  break;
9081c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::UInt128Oct;  break;
9095acacbb0SReid Kleckner     }
9105acacbb0SReid Kleckner     break;
9115acacbb0SReid Kleckner   case dwarf::DW_ATE_UTF:
9125acacbb0SReid Kleckner     switch (ByteSize) {
9135acacbb0SReid Kleckner     case 2: STK = SimpleTypeKind::Character16; break;
9145acacbb0SReid Kleckner     case 4: STK = SimpleTypeKind::Character32; break;
9155acacbb0SReid Kleckner     }
9165acacbb0SReid Kleckner     break;
9175acacbb0SReid Kleckner   case dwarf::DW_ATE_signed_char:
9185acacbb0SReid Kleckner     if (ByteSize == 1)
9195acacbb0SReid Kleckner       STK = SimpleTypeKind::SignedCharacter;
9205acacbb0SReid Kleckner     break;
9215acacbb0SReid Kleckner   case dwarf::DW_ATE_unsigned_char:
9225acacbb0SReid Kleckner     if (ByteSize == 1)
9235acacbb0SReid Kleckner       STK = SimpleTypeKind::UnsignedCharacter;
9245acacbb0SReid Kleckner     break;
9255acacbb0SReid Kleckner   default:
9265acacbb0SReid Kleckner     break;
9275acacbb0SReid Kleckner   }
9285acacbb0SReid Kleckner 
9295acacbb0SReid Kleckner   // Apply some fixups based on the source-level type name.
9305acacbb0SReid Kleckner   if (STK == SimpleTypeKind::Int32 && Ty->getName() == "long int")
9315acacbb0SReid Kleckner     STK = SimpleTypeKind::Int32Long;
9325acacbb0SReid Kleckner   if (STK == SimpleTypeKind::UInt32 && Ty->getName() == "long unsigned int")
9335acacbb0SReid Kleckner     STK = SimpleTypeKind::UInt32Long;
9348c46a4ceSDavid Majnemer   if (STK == SimpleTypeKind::UInt16Short &&
9358c46a4ceSDavid Majnemer       (Ty->getName() == "wchar_t" || Ty->getName() == "__wchar_t"))
9365acacbb0SReid Kleckner     STK = SimpleTypeKind::WideCharacter;
9375acacbb0SReid Kleckner   if ((STK == SimpleTypeKind::SignedCharacter ||
9385acacbb0SReid Kleckner        STK == SimpleTypeKind::UnsignedCharacter) &&
9395acacbb0SReid Kleckner       Ty->getName() == "char")
9405acacbb0SReid Kleckner     STK = SimpleTypeKind::NarrowCharacter;
9415acacbb0SReid Kleckner 
9425acacbb0SReid Kleckner   return TypeIndex(STK);
9435acacbb0SReid Kleckner }
9445acacbb0SReid Kleckner 
9455acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypePointer(const DIDerivedType *Ty) {
9465acacbb0SReid Kleckner   TypeIndex PointeeTI = getTypeIndex(Ty->getBaseType());
9475acacbb0SReid Kleckner 
9485acacbb0SReid Kleckner   // Pointers to simple types can use SimpleTypeMode, rather than having a
9495acacbb0SReid Kleckner   // dedicated pointer type record.
9505acacbb0SReid Kleckner   if (PointeeTI.isSimple() &&
9515acacbb0SReid Kleckner       PointeeTI.getSimpleMode() == SimpleTypeMode::Direct &&
9525acacbb0SReid Kleckner       Ty->getTag() == dwarf::DW_TAG_pointer_type) {
9535acacbb0SReid Kleckner     SimpleTypeMode Mode = Ty->getSizeInBits() == 64
9545acacbb0SReid Kleckner                               ? SimpleTypeMode::NearPointer64
9555acacbb0SReid Kleckner                               : SimpleTypeMode::NearPointer32;
9565acacbb0SReid Kleckner     return TypeIndex(PointeeTI.getSimpleKind(), Mode);
9575acacbb0SReid Kleckner   }
9585acacbb0SReid Kleckner 
9595acacbb0SReid Kleckner   PointerKind PK =
9605acacbb0SReid Kleckner       Ty->getSizeInBits() == 64 ? PointerKind::Near64 : PointerKind::Near32;
9615acacbb0SReid Kleckner   PointerMode PM = PointerMode::Pointer;
9625acacbb0SReid Kleckner   switch (Ty->getTag()) {
9635acacbb0SReid Kleckner   default: llvm_unreachable("not a pointer tag type");
9645acacbb0SReid Kleckner   case dwarf::DW_TAG_pointer_type:
9655acacbb0SReid Kleckner     PM = PointerMode::Pointer;
9665acacbb0SReid Kleckner     break;
9675acacbb0SReid Kleckner   case dwarf::DW_TAG_reference_type:
9685acacbb0SReid Kleckner     PM = PointerMode::LValueReference;
9695acacbb0SReid Kleckner     break;
9705acacbb0SReid Kleckner   case dwarf::DW_TAG_rvalue_reference_type:
9715acacbb0SReid Kleckner     PM = PointerMode::RValueReference;
9725acacbb0SReid Kleckner     break;
9735acacbb0SReid Kleckner   }
9745acacbb0SReid Kleckner   // FIXME: MSVC folds qualifiers into PointerOptions in the context of a method
9755acacbb0SReid Kleckner   // 'this' pointer, but not normal contexts. Figure out what we're supposed to
9765acacbb0SReid Kleckner   // do.
9775acacbb0SReid Kleckner   PointerOptions PO = PointerOptions::None;
9785acacbb0SReid Kleckner   PointerRecord PR(PointeeTI, PK, PM, PO, Ty->getSizeInBits() / 8);
9795acacbb0SReid Kleckner   return TypeTable.writePointer(PR);
9805acacbb0SReid Kleckner }
9815acacbb0SReid Kleckner 
982*604105bbSReid Kleckner static PointerToMemberRepresentation translatePtrToMemberRep(bool IsPMF,
983*604105bbSReid Kleckner                                                              unsigned Flags) {
984*604105bbSReid Kleckner   if (IsPMF) {
985*604105bbSReid Kleckner     switch (Flags & DINode::FlagPtrToMemberRep) {
986*604105bbSReid Kleckner     case 0:
987*604105bbSReid Kleckner       return PointerToMemberRepresentation::GeneralFunction;
988*604105bbSReid Kleckner     case DINode::FlagSingleInheritance:
989*604105bbSReid Kleckner       return PointerToMemberRepresentation::SingleInheritanceFunction;
990*604105bbSReid Kleckner     case DINode::FlagMultipleInheritance:
991*604105bbSReid Kleckner       return PointerToMemberRepresentation::MultipleInheritanceFunction;
992*604105bbSReid Kleckner     case DINode::FlagVirtualInheritance:
993*604105bbSReid Kleckner       return PointerToMemberRepresentation::VirtualInheritanceFunction;
994*604105bbSReid Kleckner     }
995*604105bbSReid Kleckner   } else {
996*604105bbSReid Kleckner     switch (Flags & DINode::FlagPtrToMemberRep) {
997*604105bbSReid Kleckner     case 0:
998*604105bbSReid Kleckner       return PointerToMemberRepresentation::GeneralData;
999*604105bbSReid Kleckner     case DINode::FlagSingleInheritance:
1000*604105bbSReid Kleckner       return PointerToMemberRepresentation::SingleInheritanceData;
1001*604105bbSReid Kleckner     case DINode::FlagMultipleInheritance:
1002*604105bbSReid Kleckner       return PointerToMemberRepresentation::MultipleInheritanceData;
1003*604105bbSReid Kleckner     case DINode::FlagVirtualInheritance:
1004*604105bbSReid Kleckner       return PointerToMemberRepresentation::VirtualInheritanceData;
1005*604105bbSReid Kleckner     }
1006*604105bbSReid Kleckner   }
1007*604105bbSReid Kleckner   llvm_unreachable("invalid ptr to member representation");
1008*604105bbSReid Kleckner }
1009*604105bbSReid Kleckner 
10105acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeMemberPointer(const DIDerivedType *Ty) {
10115acacbb0SReid Kleckner   assert(Ty->getTag() == dwarf::DW_TAG_ptr_to_member_type);
10125acacbb0SReid Kleckner   TypeIndex ClassTI = getTypeIndex(Ty->getClassType());
10135acacbb0SReid Kleckner   TypeIndex PointeeTI = getTypeIndex(Ty->getBaseType());
10145acacbb0SReid Kleckner   PointerKind PK = Asm->MAI->getPointerSize() == 8 ? PointerKind::Near64
10155acacbb0SReid Kleckner                                                    : PointerKind::Near32;
1016*604105bbSReid Kleckner   bool IsPMF = isa<DISubroutineType>(Ty->getBaseType());
1017*604105bbSReid Kleckner   PointerMode PM = IsPMF ? PointerMode::PointerToMemberFunction
10185acacbb0SReid Kleckner                          : PointerMode::PointerToDataMember;
10195acacbb0SReid Kleckner   PointerOptions PO = PointerOptions::None; // FIXME
1020*604105bbSReid Kleckner   MemberPointerInfo MPI(ClassTI,
1021*604105bbSReid Kleckner                         translatePtrToMemberRep(IsPMF, Ty->getFlags()));
1022*604105bbSReid Kleckner   uint64_t SizeInBytes = Ty->getSizeInBits() / 8;
1023*604105bbSReid Kleckner   PointerRecord PR(PointeeTI, PK, PM, PO, SizeInBytes, MPI);
10245acacbb0SReid Kleckner   return TypeTable.writePointer(PR);
10255acacbb0SReid Kleckner }
10265acacbb0SReid Kleckner 
1027de3d8b50SReid Kleckner /// Given a DWARF calling convention, get the CodeView equivalent. If we don't
1028de3d8b50SReid Kleckner /// have a translation, use the NearC convention.
1029de3d8b50SReid Kleckner static CallingConvention dwarfCCToCodeView(unsigned DwarfCC) {
1030de3d8b50SReid Kleckner   switch (DwarfCC) {
1031de3d8b50SReid Kleckner   case dwarf::DW_CC_normal:             return CallingConvention::NearC;
1032de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_msfastcall: return CallingConvention::NearFast;
1033de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_thiscall:   return CallingConvention::ThisCall;
1034de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_stdcall:    return CallingConvention::NearStdCall;
1035de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_pascal:     return CallingConvention::NearPascal;
1036de3d8b50SReid Kleckner   case dwarf::DW_CC_LLVM_vectorcall:    return CallingConvention::NearVector;
1037de3d8b50SReid Kleckner   }
1038de3d8b50SReid Kleckner   return CallingConvention::NearC;
1039de3d8b50SReid Kleckner }
1040de3d8b50SReid Kleckner 
10415acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeModifier(const DIDerivedType *Ty) {
10425acacbb0SReid Kleckner   ModifierOptions Mods = ModifierOptions::None;
10435acacbb0SReid Kleckner   bool IsModifier = true;
10445acacbb0SReid Kleckner   const DIType *BaseTy = Ty;
1045b9c80fd8SReid Kleckner   while (IsModifier && BaseTy) {
10465acacbb0SReid Kleckner     // FIXME: Need to add DWARF tag for __unaligned.
10475acacbb0SReid Kleckner     switch (BaseTy->getTag()) {
10485acacbb0SReid Kleckner     case dwarf::DW_TAG_const_type:
10495acacbb0SReid Kleckner       Mods |= ModifierOptions::Const;
10505acacbb0SReid Kleckner       break;
10515acacbb0SReid Kleckner     case dwarf::DW_TAG_volatile_type:
10525acacbb0SReid Kleckner       Mods |= ModifierOptions::Volatile;
10535acacbb0SReid Kleckner       break;
10545acacbb0SReid Kleckner     default:
10555acacbb0SReid Kleckner       IsModifier = false;
10565acacbb0SReid Kleckner       break;
10575acacbb0SReid Kleckner     }
10585acacbb0SReid Kleckner     if (IsModifier)
10595acacbb0SReid Kleckner       BaseTy = cast<DIDerivedType>(BaseTy)->getBaseType().resolve();
10605acacbb0SReid Kleckner   }
10615acacbb0SReid Kleckner   TypeIndex ModifiedTI = getTypeIndex(BaseTy);
10625acacbb0SReid Kleckner   ModifierRecord MR(ModifiedTI, Mods);
10635acacbb0SReid Kleckner   return TypeTable.writeModifier(MR);
10645acacbb0SReid Kleckner }
10655acacbb0SReid Kleckner 
106675c3ebfaSDavid Majnemer TypeIndex CodeViewDebug::lowerTypeFunction(const DISubroutineType *Ty) {
106775c3ebfaSDavid Majnemer   SmallVector<TypeIndex, 8> ReturnAndArgTypeIndices;
106875c3ebfaSDavid Majnemer   for (DITypeRef ArgTypeRef : Ty->getTypeArray())
106975c3ebfaSDavid Majnemer     ReturnAndArgTypeIndices.push_back(getTypeIndex(ArgTypeRef));
107075c3ebfaSDavid Majnemer 
107175c3ebfaSDavid Majnemer   TypeIndex ReturnTypeIndex = TypeIndex::Void();
107275c3ebfaSDavid Majnemer   ArrayRef<TypeIndex> ArgTypeIndices = None;
107375c3ebfaSDavid Majnemer   if (!ReturnAndArgTypeIndices.empty()) {
107475c3ebfaSDavid Majnemer     auto ReturnAndArgTypesRef = makeArrayRef(ReturnAndArgTypeIndices);
107575c3ebfaSDavid Majnemer     ReturnTypeIndex = ReturnAndArgTypesRef.front();
107675c3ebfaSDavid Majnemer     ArgTypeIndices = ReturnAndArgTypesRef.drop_front();
107775c3ebfaSDavid Majnemer   }
107875c3ebfaSDavid Majnemer 
107975c3ebfaSDavid Majnemer   ArgListRecord ArgListRec(TypeRecordKind::ArgList, ArgTypeIndices);
108075c3ebfaSDavid Majnemer   TypeIndex ArgListIndex = TypeTable.writeArgList(ArgListRec);
108175c3ebfaSDavid Majnemer 
1082de3d8b50SReid Kleckner   CallingConvention CC = dwarfCCToCodeView(Ty->getCC());
1083de3d8b50SReid Kleckner 
108475c3ebfaSDavid Majnemer   // TODO: Some functions are member functions, we should use a more appropriate
108575c3ebfaSDavid Majnemer   // record for those.
1086de3d8b50SReid Kleckner   ProcedureRecord Procedure(ReturnTypeIndex, CC, FunctionOptions::None,
1087de3d8b50SReid Kleckner                             ArgTypeIndices.size(), ArgListIndex);
108875c3ebfaSDavid Majnemer   return TypeTable.writeProcedure(Procedure);
108975c3ebfaSDavid Majnemer }
109075c3ebfaSDavid Majnemer 
1091a8d57407SReid Kleckner static MemberAccess translateAccessFlags(unsigned RecordTag,
1092a8d57407SReid Kleckner                                          const DIType *Member) {
1093a8d57407SReid Kleckner   switch (Member->getFlags() & DINode::FlagAccessibility) {
1094a8d57407SReid Kleckner   case DINode::FlagPrivate:   return MemberAccess::Private;
1095a8d57407SReid Kleckner   case DINode::FlagPublic:    return MemberAccess::Public;
1096a8d57407SReid Kleckner   case DINode::FlagProtected: return MemberAccess::Protected;
1097a8d57407SReid Kleckner   case 0:
1098a8d57407SReid Kleckner     // If there was no explicit access control, provide the default for the tag.
1099a8d57407SReid Kleckner     return RecordTag == dwarf::DW_TAG_class_type ? MemberAccess::Private
1100a8d57407SReid Kleckner                                                  : MemberAccess::Public;
1101a8d57407SReid Kleckner   }
1102a8d57407SReid Kleckner   llvm_unreachable("access flags are exclusive");
1103a8d57407SReid Kleckner }
1104a8d57407SReid Kleckner 
1105a8d57407SReid Kleckner static TypeRecordKind getRecordKind(const DICompositeType *Ty) {
1106a8d57407SReid Kleckner   switch (Ty->getTag()) {
1107a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:     return TypeRecordKind::Class;
1108a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type: return TypeRecordKind::Struct;
1109a8d57407SReid Kleckner   }
1110a8d57407SReid Kleckner   llvm_unreachable("unexpected tag");
1111a8d57407SReid Kleckner }
1112a8d57407SReid Kleckner 
1113a8d57407SReid Kleckner /// Return the HasUniqueName option if it should be present in ClassOptions, or
1114a8d57407SReid Kleckner /// None otherwise.
1115a8d57407SReid Kleckner static ClassOptions getRecordUniqueNameOption(const DICompositeType *Ty) {
1116a8d57407SReid Kleckner   // MSVC always sets this flag now, even for local types. Clang doesn't always
1117a8d57407SReid Kleckner   // appear to give every type a linkage name, which may be problematic for us.
1118a8d57407SReid Kleckner   // FIXME: Investigate the consequences of not following them here.
1119a8d57407SReid Kleckner   return !Ty->getIdentifier().empty() ? ClassOptions::HasUniqueName
1120a8d57407SReid Kleckner                                       : ClassOptions::None;
1121a8d57407SReid Kleckner }
1122a8d57407SReid Kleckner 
1123979cb888SDavid Majnemer TypeIndex CodeViewDebug::lowerTypeEnum(const DICompositeType *Ty) {
1124979cb888SDavid Majnemer   ClassOptions CO = ClassOptions::None | getRecordUniqueNameOption(Ty);
1125979cb888SDavid Majnemer   TypeIndex FTI;
1126da9548f9SDavid Majnemer   unsigned EnumeratorCount = 0;
1127979cb888SDavid Majnemer 
1128da9548f9SDavid Majnemer   if (Ty->isForwardDecl()) {
1129979cb888SDavid Majnemer     CO |= ClassOptions::ForwardReference;
1130da9548f9SDavid Majnemer   } else {
1131da9548f9SDavid Majnemer     FieldListRecordBuilder Fields;
1132da9548f9SDavid Majnemer     for (const DINode *Element : Ty->getElements()) {
1133da9548f9SDavid Majnemer       // We assume that the frontend provides all members in source declaration
1134da9548f9SDavid Majnemer       // order, which is what MSVC does.
1135da9548f9SDavid Majnemer       if (auto *Enumerator = dyn_cast_or_null<DIEnumerator>(Element)) {
1136da9548f9SDavid Majnemer         Fields.writeEnumerator(EnumeratorRecord(
1137da9548f9SDavid Majnemer             MemberAccess::Public, APSInt::getUnsigned(Enumerator->getValue()),
1138da9548f9SDavid Majnemer             Enumerator->getName()));
1139da9548f9SDavid Majnemer         EnumeratorCount++;
1140da9548f9SDavid Majnemer       }
1141da9548f9SDavid Majnemer     }
1142da9548f9SDavid Majnemer     FTI = TypeTable.writeFieldList(Fields);
1143da9548f9SDavid Majnemer   }
1144979cb888SDavid Majnemer 
1145da9548f9SDavid Majnemer   return TypeTable.writeEnum(EnumRecord(EnumeratorCount, CO, FTI, Ty->getName(),
1146979cb888SDavid Majnemer                                         Ty->getIdentifier(),
1147979cb888SDavid Majnemer                                         getTypeIndex(Ty->getBaseType())));
1148979cb888SDavid Majnemer }
1149979cb888SDavid Majnemer 
1150a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerTypeClass(const DICompositeType *Ty) {
1151a8d57407SReid Kleckner   // First, construct the forward decl.  Don't look into Ty to compute the
1152a8d57407SReid Kleckner   // forward decl options, since it might not be available in all TUs.
1153a8d57407SReid Kleckner   TypeRecordKind Kind = getRecordKind(Ty);
1154a8d57407SReid Kleckner   ClassOptions CO =
1155a8d57407SReid Kleckner       ClassOptions::ForwardReference | getRecordUniqueNameOption(Ty);
1156a8d57407SReid Kleckner   TypeIndex FwdDeclTI = TypeTable.writeClass(ClassRecord(
1157a8d57407SReid Kleckner       Kind, 0, CO, HfaKind::None, WindowsRTClassKind::None, TypeIndex(),
1158a8d57407SReid Kleckner       TypeIndex(), TypeIndex(), 0, Ty->getName(), Ty->getIdentifier()));
1159a8d57407SReid Kleckner   return FwdDeclTI;
1160a8d57407SReid Kleckner }
1161a8d57407SReid Kleckner 
1162a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerCompleteTypeClass(const DICompositeType *Ty) {
1163a8d57407SReid Kleckner   // Construct the field list and complete type record.
1164a8d57407SReid Kleckner   TypeRecordKind Kind = getRecordKind(Ty);
1165a8d57407SReid Kleckner   // FIXME: Other ClassOptions, like ContainsNestedClass and NestedClass.
1166a8d57407SReid Kleckner   ClassOptions CO = ClassOptions::None | getRecordUniqueNameOption(Ty);
1167a8d57407SReid Kleckner   TypeIndex FTI;
1168a8d57407SReid Kleckner   unsigned FieldCount;
1169a8d57407SReid Kleckner   std::tie(FTI, FieldCount) = lowerRecordFieldList(Ty);
1170a8d57407SReid Kleckner 
1171a8d57407SReid Kleckner   uint64_t SizeInBytes = Ty->getSizeInBits() / 8;
1172a8d57407SReid Kleckner   return TypeTable.writeClass(ClassRecord(Kind, FieldCount, CO, HfaKind::None,
1173a8d57407SReid Kleckner                                           WindowsRTClassKind::None, FTI,
1174a8d57407SReid Kleckner                                           TypeIndex(), TypeIndex(), SizeInBytes,
1175a8d57407SReid Kleckner                                           Ty->getName(), Ty->getIdentifier()));
1176a8d57407SReid Kleckner   // FIXME: Make an LF_UDT_SRC_LINE record.
1177a8d57407SReid Kleckner }
1178a8d57407SReid Kleckner 
1179a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerTypeUnion(const DICompositeType *Ty) {
1180a8d57407SReid Kleckner   ClassOptions CO =
1181a8d57407SReid Kleckner       ClassOptions::ForwardReference | getRecordUniqueNameOption(Ty);
1182a8d57407SReid Kleckner   TypeIndex FwdDeclTI =
1183a8d57407SReid Kleckner       TypeTable.writeUnion(UnionRecord(0, CO, HfaKind::None, TypeIndex(), 0,
1184a8d57407SReid Kleckner                                        Ty->getName(), Ty->getIdentifier()));
1185a8d57407SReid Kleckner   return FwdDeclTI;
1186a8d57407SReid Kleckner }
1187a8d57407SReid Kleckner 
1188a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerCompleteTypeUnion(const DICompositeType *Ty) {
1189a8d57407SReid Kleckner   ClassOptions CO = ClassOptions::None | getRecordUniqueNameOption(Ty);
1190a8d57407SReid Kleckner   TypeIndex FTI;
1191a8d57407SReid Kleckner   unsigned FieldCount;
1192a8d57407SReid Kleckner   std::tie(FTI, FieldCount) = lowerRecordFieldList(Ty);
1193a8d57407SReid Kleckner   uint64_t SizeInBytes = Ty->getSizeInBits() / 8;
1194a8d57407SReid Kleckner   return TypeTable.writeUnion(UnionRecord(FieldCount, CO, HfaKind::None, FTI,
1195a8d57407SReid Kleckner                                           SizeInBytes, Ty->getName(),
1196a8d57407SReid Kleckner                                           Ty->getIdentifier()));
1197a8d57407SReid Kleckner   // FIXME: Make an LF_UDT_SRC_LINE record.
1198a8d57407SReid Kleckner }
1199a8d57407SReid Kleckner 
1200a8d57407SReid Kleckner std::pair<TypeIndex, unsigned>
1201a8d57407SReid Kleckner CodeViewDebug::lowerRecordFieldList(const DICompositeType *Ty) {
1202a8d57407SReid Kleckner   // Manually count members. MSVC appears to count everything that generates a
1203a8d57407SReid Kleckner   // field list record. Each individual overload in a method overload group
1204a8d57407SReid Kleckner   // contributes to this count, even though the overload group is a single field
1205a8d57407SReid Kleckner   // list record.
1206a8d57407SReid Kleckner   unsigned MemberCount = 0;
1207a8d57407SReid Kleckner   FieldListRecordBuilder Fields;
1208a8d57407SReid Kleckner   for (const DINode *Element : Ty->getElements()) {
1209a8d57407SReid Kleckner     // We assume that the frontend provides all members in source declaration
1210a8d57407SReid Kleckner     // order, which is what MSVC does.
1211a8d57407SReid Kleckner     if (!Element)
1212a8d57407SReid Kleckner       continue;
1213a8d57407SReid Kleckner     if (auto *SP = dyn_cast<DISubprogram>(Element)) {
1214a8d57407SReid Kleckner       // C++ method.
1215a8d57407SReid Kleckner       // FIXME: Overloaded methods are grouped together, so we'll need two
1216a8d57407SReid Kleckner       // passes to group them.
1217a8d57407SReid Kleckner       (void)SP;
1218a8d57407SReid Kleckner     } else if (auto *Member = dyn_cast<DIDerivedType>(Element)) {
1219a8d57407SReid Kleckner       if (Member->getTag() == dwarf::DW_TAG_member) {
1220a8d57407SReid Kleckner         if (Member->isStaticMember()) {
1221a8d57407SReid Kleckner           // Static data member.
1222a8d57407SReid Kleckner           Fields.writeStaticDataMember(StaticDataMemberRecord(
1223a8d57407SReid Kleckner               translateAccessFlags(Ty->getTag(), Member),
1224a8d57407SReid Kleckner               getTypeIndex(Member->getBaseType()), Member->getName()));
1225a8d57407SReid Kleckner           MemberCount++;
1226a8d57407SReid Kleckner         } else {
1227a8d57407SReid Kleckner           // Data member.
1228a8d57407SReid Kleckner           // FIXME: Make a BitFieldRecord for bitfields.
1229a8d57407SReid Kleckner           Fields.writeDataMember(DataMemberRecord(
1230a8d57407SReid Kleckner               translateAccessFlags(Ty->getTag(), Member),
1231a8d57407SReid Kleckner               getTypeIndex(Member->getBaseType()),
1232a8d57407SReid Kleckner               Member->getOffsetInBits() / 8, Member->getName()));
1233a8d57407SReid Kleckner           MemberCount++;
1234a8d57407SReid Kleckner         }
1235a8d57407SReid Kleckner       } else if (Member->getTag() == dwarf::DW_TAG_friend) {
1236a8d57407SReid Kleckner         // Ignore friend members. It appears that MSVC emitted info about
1237a8d57407SReid Kleckner         // friends in the past, but modern versions do not.
1238a8d57407SReid Kleckner       }
1239a8d57407SReid Kleckner       // FIXME: Get clang to emit nested types here and do something with
1240a8d57407SReid Kleckner       // them.
1241a8d57407SReid Kleckner     }
1242a8d57407SReid Kleckner     // Skip other unrecognized kinds of elements.
1243a8d57407SReid Kleckner   }
1244a8d57407SReid Kleckner   return {TypeTable.writeFieldList(Fields), MemberCount};
1245a8d57407SReid Kleckner }
1246a8d57407SReid Kleckner 
12475acacbb0SReid Kleckner TypeIndex CodeViewDebug::getTypeIndex(DITypeRef TypeRef) {
12485acacbb0SReid Kleckner   const DIType *Ty = TypeRef.resolve();
12495acacbb0SReid Kleckner 
12505acacbb0SReid Kleckner   // The null DIType is the void type. Don't try to hash it.
12515acacbb0SReid Kleckner   if (!Ty)
12525acacbb0SReid Kleckner     return TypeIndex::Void();
12535acacbb0SReid Kleckner 
1254a8d57407SReid Kleckner   // Check if we've already translated this type. Don't try to do a
1255a8d57407SReid Kleckner   // get-or-create style insertion that caches the hash lookup across the
1256a8d57407SReid Kleckner   // lowerType call. It will update the TypeIndices map.
12575acacbb0SReid Kleckner   auto I = TypeIndices.find(Ty);
12585acacbb0SReid Kleckner   if (I != TypeIndices.end())
12595acacbb0SReid Kleckner     return I->second;
12605acacbb0SReid Kleckner 
12615acacbb0SReid Kleckner   TypeIndex TI = lowerType(Ty);
12625acacbb0SReid Kleckner 
1263a8d57407SReid Kleckner   recordTypeIndexForDINode(Ty, TI);
1264a8d57407SReid Kleckner   return TI;
1265a8d57407SReid Kleckner }
1266a8d57407SReid Kleckner 
1267a8d57407SReid Kleckner TypeIndex CodeViewDebug::getCompleteTypeIndex(DITypeRef TypeRef) {
1268a8d57407SReid Kleckner   const DIType *Ty = TypeRef.resolve();
1269a8d57407SReid Kleckner 
1270a8d57407SReid Kleckner   // The null DIType is the void type. Don't try to hash it.
1271a8d57407SReid Kleckner   if (!Ty)
1272a8d57407SReid Kleckner     return TypeIndex::Void();
1273a8d57407SReid Kleckner 
1274a8d57407SReid Kleckner   // If this is a non-record type, the complete type index is the same as the
1275a8d57407SReid Kleckner   // normal type index. Just call getTypeIndex.
1276a8d57407SReid Kleckner   switch (Ty->getTag()) {
1277a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
1278a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
1279a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
1280a8d57407SReid Kleckner     break;
1281a8d57407SReid Kleckner   default:
1282a8d57407SReid Kleckner     return getTypeIndex(Ty);
1283a8d57407SReid Kleckner   }
1284a8d57407SReid Kleckner 
1285a8d57407SReid Kleckner   // Check if we've already translated the complete record type.  Lowering a
1286a8d57407SReid Kleckner   // complete type should never trigger lowering another complete type, so we
1287a8d57407SReid Kleckner   // can reuse the hash table lookup result.
1288a8d57407SReid Kleckner   const auto *CTy = cast<DICompositeType>(Ty);
1289a8d57407SReid Kleckner   auto InsertResult = CompleteTypeIndices.insert({CTy, TypeIndex()});
1290a8d57407SReid Kleckner   if (!InsertResult.second)
1291a8d57407SReid Kleckner     return InsertResult.first->second;
1292a8d57407SReid Kleckner 
1293a8d57407SReid Kleckner   // Make sure the forward declaration is emitted first. It's unclear if this
1294a8d57407SReid Kleckner   // is necessary, but MSVC does it, and we should follow suit until we can show
1295a8d57407SReid Kleckner   // otherwise.
1296a8d57407SReid Kleckner   TypeIndex FwdDeclTI = getTypeIndex(CTy);
1297a8d57407SReid Kleckner 
1298a8d57407SReid Kleckner   // Just use the forward decl if we don't have complete type info. This might
1299a8d57407SReid Kleckner   // happen if the frontend is using modules and expects the complete definition
1300a8d57407SReid Kleckner   // to be emitted elsewhere.
1301a8d57407SReid Kleckner   if (CTy->isForwardDecl())
1302a8d57407SReid Kleckner     return FwdDeclTI;
1303a8d57407SReid Kleckner 
1304a8d57407SReid Kleckner   TypeIndex TI;
1305a8d57407SReid Kleckner   switch (CTy->getTag()) {
1306a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
1307a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
1308a8d57407SReid Kleckner     TI = lowerCompleteTypeClass(CTy);
1309a8d57407SReid Kleckner     break;
1310a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
1311a8d57407SReid Kleckner     TI = lowerCompleteTypeUnion(CTy);
1312a8d57407SReid Kleckner     break;
1313a8d57407SReid Kleckner   default:
1314a8d57407SReid Kleckner     llvm_unreachable("not a record");
1315a8d57407SReid Kleckner   }
1316a8d57407SReid Kleckner 
1317a8d57407SReid Kleckner   InsertResult.first->second = TI;
13185acacbb0SReid Kleckner   return TI;
13195acacbb0SReid Kleckner }
13205acacbb0SReid Kleckner 
1321f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
1322f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
1323f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
1324f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
1325f9c275feSReid Kleckner   OS.AddComment("Record length");
1326f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
1327f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
1328f9c275feSReid Kleckner 
1329f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
133063a2846eSZachary Turner   OS.EmitIntValue(unsigned(SymbolKind::S_LOCAL), 2);
1331f9c275feSReid Kleckner 
133263a2846eSZachary Turner   LocalSymFlags Flags = LocalSymFlags::None;
1333f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
133463a2846eSZachary Turner     Flags |= LocalSymFlags::IsParameter;
1335876330d5SReid Kleckner   if (Var.DefRanges.empty())
133663a2846eSZachary Turner     Flags |= LocalSymFlags::IsOptimizedOut;
1337f9c275feSReid Kleckner 
1338f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
1339a8d57407SReid Kleckner   TypeIndex TI = getCompleteTypeIndex(Var.DIVar->getType());
13405acacbb0SReid Kleckner   OS.EmitIntValue(TI.getIndex(), 4);
1341f9c275feSReid Kleckner   OS.AddComment("Flags");
134263a2846eSZachary Turner   OS.EmitIntValue(static_cast<uint16_t>(Flags), 2);
13431256125fSDavid Majnemer   // Truncate the name so we won't overflow the record length field.
1344b9456a5eSDavid Majnemer   emitNullTerminatedSymbolName(OS, Var.DIVar->getName());
1345f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
1346f9c275feSReid Kleckner 
1347876330d5SReid Kleckner   // Calculate the on disk prefix of the appropriate def range record. The
1348876330d5SReid Kleckner   // records and on disk formats are described in SymbolRecords.h. BytePrefix
1349876330d5SReid Kleckner   // should be big enough to hold all forms without memory allocation.
1350876330d5SReid Kleckner   SmallString<20> BytePrefix;
1351876330d5SReid Kleckner   for (const LocalVarDefRange &DefRange : Var.DefRanges) {
1352876330d5SReid Kleckner     BytePrefix.clear();
1353876330d5SReid Kleckner     // FIXME: Handle bitpieces.
1354876330d5SReid Kleckner     if (DefRange.StructOffset != 0)
1355876330d5SReid Kleckner       continue;
1356876330d5SReid Kleckner 
1357876330d5SReid Kleckner     if (DefRange.InMemory) {
1358a78ecd1eSZachary Turner       DefRangeRegisterRelSym Sym(DefRange.CVRegister, 0, DefRange.DataOffset, 0,
1359a78ecd1eSZachary Turner                                  0, 0, ArrayRef<LocalVariableAddrGap>());
1360f9c275feSReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
1361876330d5SReid Kleckner       BytePrefix +=
1362876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
1363a78ecd1eSZachary Turner       BytePrefix +=
1364a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
1365a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
1366876330d5SReid Kleckner     } else {
1367876330d5SReid Kleckner       assert(DefRange.DataOffset == 0 && "unexpected offset into register");
1368a78ecd1eSZachary Turner       // Unclear what matters here.
1369a78ecd1eSZachary Turner       DefRangeRegisterSym Sym(DefRange.CVRegister, 0, 0, 0, 0,
1370a78ecd1eSZachary Turner                               ArrayRef<LocalVariableAddrGap>());
1371876330d5SReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER);
1372876330d5SReid Kleckner       BytePrefix +=
1373876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
1374a78ecd1eSZachary Turner       BytePrefix +=
1375a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
1376a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
1377876330d5SReid Kleckner     }
1378876330d5SReid Kleckner     OS.EmitCVDefRangeDirective(DefRange.Ranges, BytePrefix);
1379876330d5SReid Kleckner   }
1380f9c275feSReid Kleckner }
1381f9c275feSReid Kleckner 
138270f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
138370f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
138470f5bc99SReid Kleckner     return;
138570f5bc99SReid Kleckner 
138670f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
138770f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
138870f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
138970f5bc99SReid Kleckner 
1390adebb937SPete Cooper   collectVariableInfo(GV->getSubprogram());
1391876330d5SReid Kleckner 
1392876330d5SReid Kleckner   DebugHandlerBase::endFunction(MF);
1393876330d5SReid Kleckner 
13942214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
13952214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
139670f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
1397f9c275feSReid Kleckner     CurFn = nullptr;
1398f9c275feSReid Kleckner     return;
139970f5bc99SReid Kleckner   }
1400f9c275feSReid Kleckner 
1401f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
1402f9c275feSReid Kleckner 
140370f5bc99SReid Kleckner   CurFn = nullptr;
140470f5bc99SReid Kleckner }
140570f5bc99SReid Kleckner 
140670f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
1407f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
1408f9c275feSReid Kleckner 
140970f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
141070f5bc99SReid Kleckner   if (!Asm || MI->isDebugValue() || MI->getFlag(MachineInstr::FrameSetup))
141170f5bc99SReid Kleckner     return;
141270f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
141370f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
141470f5bc99SReid Kleckner     return;
141570f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
141670f5bc99SReid Kleckner }
14176f3406dfSReid Kleckner 
14186f3406dfSReid Kleckner MCSymbol *CodeViewDebug::beginCVSubsection(ModuleSubstreamKind Kind) {
14196f3406dfSReid Kleckner   MCSymbol *BeginLabel = MMI->getContext().createTempSymbol(),
14206f3406dfSReid Kleckner            *EndLabel = MMI->getContext().createTempSymbol();
14216f3406dfSReid Kleckner   OS.EmitIntValue(unsigned(Kind), 4);
14226f3406dfSReid Kleckner   OS.AddComment("Subsection size");
14236f3406dfSReid Kleckner   OS.emitAbsoluteSymbolDiff(EndLabel, BeginLabel, 4);
14246f3406dfSReid Kleckner   OS.EmitLabel(BeginLabel);
14256f3406dfSReid Kleckner   return EndLabel;
14266f3406dfSReid Kleckner }
14276f3406dfSReid Kleckner 
14286f3406dfSReid Kleckner void CodeViewDebug::endCVSubsection(MCSymbol *EndLabel) {
14296f3406dfSReid Kleckner   OS.EmitLabel(EndLabel);
14306f3406dfSReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
14316f3406dfSReid Kleckner   OS.EmitValueToAlignment(4);
14326f3406dfSReid Kleckner }
14336f3406dfSReid Kleckner 
14343128b10cSDavid Majnemer void CodeViewDebug::emitDebugInfoForUDTs(
14353128b10cSDavid Majnemer     ArrayRef<std::pair<std::string, TypeIndex>> UDTs) {
14363128b10cSDavid Majnemer   for (const std::pair<std::string, codeview::TypeIndex> &UDT : UDTs) {
14373128b10cSDavid Majnemer     MCSymbol *UDTRecordBegin = MMI->getContext().createTempSymbol(),
14383128b10cSDavid Majnemer              *UDTRecordEnd = MMI->getContext().createTempSymbol();
14393128b10cSDavid Majnemer     OS.AddComment("Record length");
14403128b10cSDavid Majnemer     OS.emitAbsoluteSymbolDiff(UDTRecordEnd, UDTRecordBegin, 2);
14413128b10cSDavid Majnemer     OS.EmitLabel(UDTRecordBegin);
14423128b10cSDavid Majnemer 
14433128b10cSDavid Majnemer     OS.AddComment("Record kind: S_UDT");
14443128b10cSDavid Majnemer     OS.EmitIntValue(unsigned(SymbolKind::S_UDT), 2);
14453128b10cSDavid Majnemer 
14463128b10cSDavid Majnemer     OS.AddComment("Type");
14473128b10cSDavid Majnemer     OS.EmitIntValue(UDT.second.getIndex(), 4);
14483128b10cSDavid Majnemer 
14493128b10cSDavid Majnemer     emitNullTerminatedSymbolName(OS, UDT.first);
14503128b10cSDavid Majnemer     OS.EmitLabel(UDTRecordEnd);
14513128b10cSDavid Majnemer   }
14523128b10cSDavid Majnemer }
14533128b10cSDavid Majnemer 
14546f3406dfSReid Kleckner void CodeViewDebug::emitDebugInfoForGlobals() {
14556f3406dfSReid Kleckner   NamedMDNode *CUs = MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
14566f3406dfSReid Kleckner   for (const MDNode *Node : CUs->operands()) {
14576f3406dfSReid Kleckner     const auto *CU = cast<DICompileUnit>(Node);
14586f3406dfSReid Kleckner 
14596f3406dfSReid Kleckner     // First, emit all globals that are not in a comdat in a single symbol
14606f3406dfSReid Kleckner     // substream. MSVC doesn't like it if the substream is empty, so only open
14616f3406dfSReid Kleckner     // it if we have at least one global to emit.
14626f3406dfSReid Kleckner     switchToDebugSectionForSymbol(nullptr);
14636f3406dfSReid Kleckner     MCSymbol *EndLabel = nullptr;
14646f3406dfSReid Kleckner     for (const DIGlobalVariable *G : CU->getGlobalVariables()) {
14656d1d2754SReid Kleckner       if (const auto *GV = dyn_cast_or_null<GlobalVariable>(G->getVariable())) {
1466577be0feSDavid Majnemer         if (!GV->hasComdat() && !GV->isDeclarationForLinker()) {
14676f3406dfSReid Kleckner           if (!EndLabel) {
14686f3406dfSReid Kleckner             OS.AddComment("Symbol subsection for globals");
14696f3406dfSReid Kleckner             EndLabel = beginCVSubsection(ModuleSubstreamKind::Symbols);
14706f3406dfSReid Kleckner           }
14716f3406dfSReid Kleckner           emitDebugInfoForGlobal(G, Asm->getSymbol(GV));
14726f3406dfSReid Kleckner         }
14736f3406dfSReid Kleckner       }
14746d1d2754SReid Kleckner     }
14756f3406dfSReid Kleckner     if (EndLabel)
14766f3406dfSReid Kleckner       endCVSubsection(EndLabel);
14776f3406dfSReid Kleckner 
14786f3406dfSReid Kleckner     // Second, emit each global that is in a comdat into its own .debug$S
14796f3406dfSReid Kleckner     // section along with its own symbol substream.
14806f3406dfSReid Kleckner     for (const DIGlobalVariable *G : CU->getGlobalVariables()) {
14816d1d2754SReid Kleckner       if (const auto *GV = dyn_cast_or_null<GlobalVariable>(G->getVariable())) {
14826f3406dfSReid Kleckner         if (GV->hasComdat()) {
14836f3406dfSReid Kleckner           MCSymbol *GVSym = Asm->getSymbol(GV);
14846f3406dfSReid Kleckner           OS.AddComment("Symbol subsection for " +
14856f3406dfSReid Kleckner                         Twine(GlobalValue::getRealLinkageName(GV->getName())));
14866f3406dfSReid Kleckner           switchToDebugSectionForSymbol(GVSym);
14876f3406dfSReid Kleckner           EndLabel = beginCVSubsection(ModuleSubstreamKind::Symbols);
14886f3406dfSReid Kleckner           emitDebugInfoForGlobal(G, GVSym);
14896f3406dfSReid Kleckner           endCVSubsection(EndLabel);
14906f3406dfSReid Kleckner         }
14916f3406dfSReid Kleckner       }
14926f3406dfSReid Kleckner     }
14936f3406dfSReid Kleckner   }
14946f3406dfSReid Kleckner }
14956f3406dfSReid Kleckner 
14966f3406dfSReid Kleckner void CodeViewDebug::emitDebugInfoForGlobal(const DIGlobalVariable *DIGV,
14976f3406dfSReid Kleckner                                            MCSymbol *GVSym) {
14986f3406dfSReid Kleckner   // DataSym record, see SymbolRecord.h for more info.
14996f3406dfSReid Kleckner   // FIXME: Thread local data, etc
15006f3406dfSReid Kleckner   MCSymbol *DataBegin = MMI->getContext().createTempSymbol(),
15016f3406dfSReid Kleckner            *DataEnd = MMI->getContext().createTempSymbol();
15026f3406dfSReid Kleckner   OS.AddComment("Record length");
15036f3406dfSReid Kleckner   OS.emitAbsoluteSymbolDiff(DataEnd, DataBegin, 2);
15046f3406dfSReid Kleckner   OS.EmitLabel(DataBegin);
15056f3406dfSReid Kleckner   OS.AddComment("Record kind: S_GDATA32");
15066f3406dfSReid Kleckner   OS.EmitIntValue(unsigned(SymbolKind::S_GDATA32), 2);
15076f3406dfSReid Kleckner   OS.AddComment("Type");
15086f3406dfSReid Kleckner   OS.EmitIntValue(getCompleteTypeIndex(DIGV->getType()).getIndex(), 4);
15096f3406dfSReid Kleckner   OS.AddComment("DataOffset");
15106f3406dfSReid Kleckner   OS.EmitCOFFSecRel32(GVSym);
15116f3406dfSReid Kleckner   OS.AddComment("Segment");
15126f3406dfSReid Kleckner   OS.EmitCOFFSectionIndex(GVSym);
15136f3406dfSReid Kleckner   OS.AddComment("Name");
15146f3406dfSReid Kleckner   emitNullTerminatedSymbolName(OS, DIGV->getName());
15156f3406dfSReid Kleckner   OS.EmitLabel(DataEnd);
15166f3406dfSReid Kleckner }
1517