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