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"
15156a7239SReid Kleckner #include "llvm/ADT/TinyPtrVector.h"
1670f5bc99SReid Kleckner #include "llvm/DebugInfo/CodeView/CodeView.h"
17a8d57407SReid Kleckner #include "llvm/DebugInfo/CodeView/FieldListRecordBuilder.h"
182214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/Line.h"
1970f5bc99SReid Kleckner #include "llvm/DebugInfo/CodeView/SymbolRecord.h"
20fbdbe9e2SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeDumper.h"
21f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeIndex.h"
22f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeRecord.h"
2370f5bc99SReid Kleckner #include "llvm/MC/MCExpr.h"
245d122f87SReid Kleckner #include "llvm/MC/MCSectionCOFF.h"
2570f5bc99SReid Kleckner #include "llvm/MC/MCSymbol.h"
2670f5bc99SReid Kleckner #include "llvm/Support/COFF.h"
27fbdbe9e2SReid Kleckner #include "llvm/Support/ScopedPrinter.h"
28f9c275feSReid Kleckner #include "llvm/Target/TargetFrameLowering.h"
2976c9eb99SAmjad Aboud #include "llvm/Target/TargetRegisterInfo.h"
3076c9eb99SAmjad Aboud #include "llvm/Target/TargetSubtargetInfo.h"
3170f5bc99SReid Kleckner 
32f9c275feSReid Kleckner using namespace llvm;
3370f5bc99SReid Kleckner using namespace llvm::codeview;
3470f5bc99SReid Kleckner 
35f9c275feSReid Kleckner CodeViewDebug::CodeViewDebug(AsmPrinter *AP)
36f9c275feSReid Kleckner     : DebugHandlerBase(AP), OS(*Asm->OutStreamer), CurFn(nullptr) {
37f9c275feSReid Kleckner   // If module doesn't have named metadata anchors or COFF debug section
38f9c275feSReid Kleckner   // is not available, skip any debug info related stuff.
39f9c275feSReid Kleckner   if (!MMI->getModule()->getNamedMetadata("llvm.dbg.cu") ||
40f9c275feSReid Kleckner       !AP->getObjFileLowering().getCOFFDebugSymbolsSection()) {
41f9c275feSReid Kleckner     Asm = nullptr;
42f9c275feSReid Kleckner     return;
43f9c275feSReid Kleckner   }
44f9c275feSReid Kleckner 
45f9c275feSReid Kleckner   // Tell MMI that we have debug info.
46f9c275feSReid Kleckner   MMI->setDebugInfoAvailability(true);
47f9c275feSReid Kleckner }
4870f5bc99SReid Kleckner 
499533af4fSReid Kleckner StringRef CodeViewDebug::getFullFilepath(const DIFile *File) {
509533af4fSReid Kleckner   std::string &Filepath = FileToFilepathMap[File];
5170f5bc99SReid Kleckner   if (!Filepath.empty())
5270f5bc99SReid Kleckner     return Filepath;
5370f5bc99SReid Kleckner 
549533af4fSReid Kleckner   StringRef Dir = File->getDirectory(), Filename = File->getFilename();
559533af4fSReid Kleckner 
5670f5bc99SReid Kleckner   // Clang emits directory and relative filename info into the IR, but CodeView
5770f5bc99SReid Kleckner   // operates on full paths.  We could change Clang to emit full paths too, but
5870f5bc99SReid Kleckner   // that would increase the IR size and probably not needed for other users.
5970f5bc99SReid Kleckner   // For now, just concatenate and canonicalize the path here.
6070f5bc99SReid Kleckner   if (Filename.find(':') == 1)
6170f5bc99SReid Kleckner     Filepath = Filename;
6270f5bc99SReid Kleckner   else
6370f5bc99SReid Kleckner     Filepath = (Dir + "\\" + Filename).str();
6470f5bc99SReid Kleckner 
6570f5bc99SReid Kleckner   // Canonicalize the path.  We have to do it textually because we may no longer
6670f5bc99SReid Kleckner   // have access the file in the filesystem.
6770f5bc99SReid Kleckner   // First, replace all slashes with backslashes.
6870f5bc99SReid Kleckner   std::replace(Filepath.begin(), Filepath.end(), '/', '\\');
6970f5bc99SReid Kleckner 
7070f5bc99SReid Kleckner   // Remove all "\.\" with "\".
7170f5bc99SReid Kleckner   size_t Cursor = 0;
7270f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\.\\", Cursor)) != std::string::npos)
7370f5bc99SReid Kleckner     Filepath.erase(Cursor, 2);
7470f5bc99SReid Kleckner 
7570f5bc99SReid Kleckner   // Replace all "\XXX\..\" with "\".  Don't try too hard though as the original
7670f5bc99SReid Kleckner   // path should be well-formatted, e.g. start with a drive letter, etc.
7770f5bc99SReid Kleckner   Cursor = 0;
7870f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\..\\", Cursor)) != std::string::npos) {
7970f5bc99SReid Kleckner     // Something's wrong if the path starts with "\..\", abort.
8070f5bc99SReid Kleckner     if (Cursor == 0)
8170f5bc99SReid Kleckner       break;
8270f5bc99SReid Kleckner 
8370f5bc99SReid Kleckner     size_t PrevSlash = Filepath.rfind('\\', Cursor - 1);
8470f5bc99SReid Kleckner     if (PrevSlash == std::string::npos)
8570f5bc99SReid Kleckner       // Something's wrong, abort.
8670f5bc99SReid Kleckner       break;
8770f5bc99SReid Kleckner 
8870f5bc99SReid Kleckner     Filepath.erase(PrevSlash, Cursor + 3 - PrevSlash);
8970f5bc99SReid Kleckner     // The next ".." might be following the one we've just erased.
9070f5bc99SReid Kleckner     Cursor = PrevSlash;
9170f5bc99SReid Kleckner   }
9270f5bc99SReid Kleckner 
9370f5bc99SReid Kleckner   // Remove all duplicate backslashes.
9470f5bc99SReid Kleckner   Cursor = 0;
9570f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\\\", Cursor)) != std::string::npos)
9670f5bc99SReid Kleckner     Filepath.erase(Cursor, 1);
9770f5bc99SReid Kleckner 
9870f5bc99SReid Kleckner   return Filepath;
9970f5bc99SReid Kleckner }
10070f5bc99SReid Kleckner 
1012214ed89SReid Kleckner unsigned CodeViewDebug::maybeRecordFile(const DIFile *F) {
1022214ed89SReid Kleckner   unsigned NextId = FileIdMap.size() + 1;
1032214ed89SReid Kleckner   auto Insertion = FileIdMap.insert(std::make_pair(F, NextId));
1042214ed89SReid Kleckner   if (Insertion.second) {
1052214ed89SReid Kleckner     // We have to compute the full filepath and emit a .cv_file directive.
1062214ed89SReid Kleckner     StringRef FullPath = getFullFilepath(F);
107dac21b43SReid Kleckner     NextId = OS.EmitCVFileDirective(NextId, FullPath);
1082214ed89SReid Kleckner     assert(NextId == FileIdMap.size() && ".cv_file directive failed");
1092214ed89SReid Kleckner   }
1102214ed89SReid Kleckner   return Insertion.first->second;
1112214ed89SReid Kleckner }
1122214ed89SReid Kleckner 
113876330d5SReid Kleckner CodeViewDebug::InlineSite &
114876330d5SReid Kleckner CodeViewDebug::getInlineSite(const DILocation *InlinedAt,
115876330d5SReid Kleckner                              const DISubprogram *Inlinee) {
116fbd7787dSReid Kleckner   auto SiteInsertion = CurFn->InlineSites.insert({InlinedAt, InlineSite()});
117fbd7787dSReid Kleckner   InlineSite *Site = &SiteInsertion.first->second;
118fbd7787dSReid Kleckner   if (SiteInsertion.second) {
119f9c275feSReid Kleckner     Site->SiteFuncId = NextFuncId++;
120876330d5SReid Kleckner     Site->Inlinee = Inlinee;
1212280f932SReid Kleckner     InlinedSubprograms.insert(Inlinee);
12275c3ebfaSDavid Majnemer     getFuncIdForSubprogram(Inlinee);
123f3b9ba49SReid Kleckner   }
124f9c275feSReid Kleckner   return *Site;
125f3b9ba49SReid Kleckner }
126f3b9ba49SReid Kleckner 
1270c5d874bSReid Kleckner static const DISubprogram *getQualifiedNameComponents(
1280c5d874bSReid Kleckner     const DIScope *Scope, SmallVectorImpl<StringRef> &QualifiedNameComponents) {
1290c5d874bSReid Kleckner   const DISubprogram *ClosestSubprogram = nullptr;
1300c5d874bSReid Kleckner   while (Scope != nullptr) {
1310c5d874bSReid Kleckner     if (ClosestSubprogram == nullptr)
1320c5d874bSReid Kleckner       ClosestSubprogram = dyn_cast<DISubprogram>(Scope);
1330c5d874bSReid Kleckner     StringRef ScopeName = Scope->getName();
1340c5d874bSReid Kleckner     if (!ScopeName.empty())
1350c5d874bSReid Kleckner       QualifiedNameComponents.push_back(ScopeName);
1360c5d874bSReid Kleckner     Scope = Scope->getScope().resolve();
1370c5d874bSReid Kleckner   }
1380c5d874bSReid Kleckner   return ClosestSubprogram;
1390c5d874bSReid Kleckner }
1400c5d874bSReid Kleckner 
1410c5d874bSReid Kleckner static std::string getQualifiedName(ArrayRef<StringRef> QualifiedNameComponents,
1420c5d874bSReid Kleckner                                     StringRef TypeName) {
1430c5d874bSReid Kleckner   std::string FullyQualifiedName;
1440c5d874bSReid Kleckner   for (StringRef QualifiedNameComponent : reverse(QualifiedNameComponents)) {
1450c5d874bSReid Kleckner     FullyQualifiedName.append(QualifiedNameComponent);
1460c5d874bSReid Kleckner     FullyQualifiedName.append("::");
1470c5d874bSReid Kleckner   }
1480c5d874bSReid Kleckner   FullyQualifiedName.append(TypeName);
1490c5d874bSReid Kleckner   return FullyQualifiedName;
1500c5d874bSReid Kleckner }
1510c5d874bSReid Kleckner 
1520c5d874bSReid Kleckner static std::string getFullyQualifiedName(const DIScope *Scope, StringRef Name) {
1530c5d874bSReid Kleckner   SmallVector<StringRef, 5> QualifiedNameComponents;
1540c5d874bSReid Kleckner   getQualifiedNameComponents(Scope, QualifiedNameComponents);
1550c5d874bSReid Kleckner   return getQualifiedName(QualifiedNameComponents, Name);
1560c5d874bSReid Kleckner }
1570c5d874bSReid Kleckner 
1580c5d874bSReid Kleckner TypeIndex CodeViewDebug::getScopeIndex(const DIScope *Scope) {
1590c5d874bSReid Kleckner   // No scope means global scope and that uses the zero index.
1600c5d874bSReid Kleckner   if (!Scope || isa<DIFile>(Scope))
1610c5d874bSReid Kleckner     return TypeIndex();
1620c5d874bSReid Kleckner 
1630c5d874bSReid Kleckner   assert(!isa<DIType>(Scope) && "shouldn't make a namespace scope for a type");
1640c5d874bSReid Kleckner 
1650c5d874bSReid Kleckner   // Check if we've already translated this scope.
1660c5d874bSReid Kleckner   auto I = TypeIndices.find({Scope, nullptr});
1670c5d874bSReid Kleckner   if (I != TypeIndices.end())
1680c5d874bSReid Kleckner     return I->second;
1690c5d874bSReid Kleckner 
1700c5d874bSReid Kleckner   // Build the fully qualified name of the scope.
1710c5d874bSReid Kleckner   std::string ScopeName =
1720c5d874bSReid Kleckner       getFullyQualifiedName(Scope->getScope().resolve(), Scope->getName());
1730c5d874bSReid Kleckner   TypeIndex TI =
1740c5d874bSReid Kleckner       TypeTable.writeStringId(StringIdRecord(TypeIndex(), ScopeName));
1750c5d874bSReid Kleckner   return recordTypeIndexForDINode(Scope, TI);
1760c5d874bSReid Kleckner }
1770c5d874bSReid Kleckner 
17875c3ebfaSDavid Majnemer TypeIndex CodeViewDebug::getFuncIdForSubprogram(const DISubprogram *SP) {
17975c3ebfaSDavid Majnemer   // It's possible to ask for the FuncId of a function which doesn't have a
18075c3ebfaSDavid Majnemer   // subprogram: inlining a function with debug info into a function with none.
18175c3ebfaSDavid Majnemer   if (!SP)
182b68f32f0SDavid Majnemer     return TypeIndex::None();
1832280f932SReid Kleckner 
18475c3ebfaSDavid Majnemer   // Check if we've already translated this subprogram.
18576c9eb99SAmjad Aboud   auto I = TypeIndices.find({SP, nullptr});
18675c3ebfaSDavid Majnemer   if (I != TypeIndices.end())
18775c3ebfaSDavid Majnemer     return I->second;
1882280f932SReid Kleckner 
189ac945e27SReid Kleckner   // The display name includes function template arguments. Drop them to match
190ac945e27SReid Kleckner   // MSVC.
191ac945e27SReid Kleckner   StringRef DisplayName = SP->getDisplayName().split('<').first;
19275c3ebfaSDavid Majnemer 
1930c5d874bSReid Kleckner   const DIScope *Scope = SP->getScope().resolve();
1940c5d874bSReid Kleckner   TypeIndex TI;
1950c5d874bSReid Kleckner   if (const auto *Class = dyn_cast_or_null<DICompositeType>(Scope)) {
1960c5d874bSReid Kleckner     // If the scope is a DICompositeType, then this must be a method. Member
1970c5d874bSReid Kleckner     // function types take some special handling, and require access to the
1980c5d874bSReid Kleckner     // subprogram.
1990c5d874bSReid Kleckner     TypeIndex ClassType = getTypeIndex(Class);
2000c5d874bSReid Kleckner     MemberFuncIdRecord MFuncId(ClassType, getMemberFunctionType(SP, Class),
2010c5d874bSReid Kleckner                                DisplayName);
2020c5d874bSReid Kleckner     TI = TypeTable.writeMemberFuncId(MFuncId);
2030c5d874bSReid Kleckner   } else {
2040c5d874bSReid Kleckner     // Otherwise, this must be a free function.
2050c5d874bSReid Kleckner     TypeIndex ParentScope = getScopeIndex(Scope);
2060c5d874bSReid Kleckner     FuncIdRecord FuncId(ParentScope, getTypeIndex(SP->getType()), DisplayName);
2070c5d874bSReid Kleckner     TI = TypeTable.writeFuncId(FuncId);
2082280f932SReid Kleckner   }
2092280f932SReid Kleckner 
2100c5d874bSReid Kleckner   return recordTypeIndexForDINode(SP, TI);
2110c5d874bSReid Kleckner }
2120c5d874bSReid Kleckner 
2130c5d874bSReid Kleckner TypeIndex CodeViewDebug::getMemberFunctionType(const DISubprogram *SP,
2140c5d874bSReid Kleckner                                                const DICompositeType *Class) {
2150c5d874bSReid Kleckner   // Key the MemberFunctionRecord into the map as {SP, Class}. It won't collide
2160c5d874bSReid Kleckner   // with the MemberFuncIdRecord, which is keyed in as {SP, nullptr}.
2170c5d874bSReid Kleckner   auto I = TypeIndices.find({SP, nullptr});
2180c5d874bSReid Kleckner   if (I != TypeIndices.end())
2190c5d874bSReid Kleckner     return I->second;
2200c5d874bSReid Kleckner 
2210c5d874bSReid Kleckner   // FIXME: Get the ThisAdjustment off of SP when it is available.
2220c5d874bSReid Kleckner   TypeIndex TI =
2230c5d874bSReid Kleckner       lowerTypeMemberFunction(SP->getType(), Class, /*ThisAdjustment=*/0);
2240c5d874bSReid Kleckner 
2250c5d874bSReid Kleckner   return recordTypeIndexForDINode(SP, TI, Class);
2260c5d874bSReid Kleckner }
2270c5d874bSReid Kleckner 
2280c5d874bSReid Kleckner TypeIndex CodeViewDebug::recordTypeIndexForDINode(const DINode *Node, TypeIndex TI,
22976c9eb99SAmjad Aboud                                              const DIType *ClassTy) {
23076c9eb99SAmjad Aboud   auto InsertResult = TypeIndices.insert({{Node, ClassTy}, TI});
231a8d57407SReid Kleckner   (void)InsertResult;
232a8d57407SReid Kleckner   assert(InsertResult.second && "DINode was already assigned a type index");
2330c5d874bSReid Kleckner   return TI;
234a8d57407SReid Kleckner }
235a8d57407SReid Kleckner 
23676c9eb99SAmjad Aboud unsigned CodeViewDebug::getPointerSizeInBytes() {
23776c9eb99SAmjad Aboud   return MMI->getModule()->getDataLayout().getPointerSizeInBits() / 8;
23876c9eb99SAmjad Aboud }
23976c9eb99SAmjad Aboud 
240876330d5SReid Kleckner void CodeViewDebug::recordLocalVariable(LocalVariable &&Var,
241876330d5SReid Kleckner                                         const DILocation *InlinedAt) {
242876330d5SReid Kleckner   if (InlinedAt) {
243876330d5SReid Kleckner     // This variable was inlined. Associate it with the InlineSite.
244876330d5SReid Kleckner     const DISubprogram *Inlinee = Var.DIVar->getScope()->getSubprogram();
245876330d5SReid Kleckner     InlineSite &Site = getInlineSite(InlinedAt, Inlinee);
246876330d5SReid Kleckner     Site.InlinedLocals.emplace_back(Var);
247876330d5SReid Kleckner   } else {
248876330d5SReid Kleckner     // This variable goes in the main ProcSym.
249876330d5SReid Kleckner     CurFn->Locals.emplace_back(Var);
250876330d5SReid Kleckner   }
251876330d5SReid Kleckner }
252876330d5SReid Kleckner 
253829365aeSReid Kleckner static void addLocIfNotPresent(SmallVectorImpl<const DILocation *> &Locs,
254829365aeSReid Kleckner                                const DILocation *Loc) {
255829365aeSReid Kleckner   auto B = Locs.begin(), E = Locs.end();
256829365aeSReid Kleckner   if (std::find(B, E, Loc) == E)
257829365aeSReid Kleckner     Locs.push_back(Loc);
258829365aeSReid Kleckner }
259829365aeSReid Kleckner 
260bdc4956bSBenjamin Kramer void CodeViewDebug::maybeRecordLocation(const DebugLoc &DL,
26170f5bc99SReid Kleckner                                         const MachineFunction *MF) {
2629533af4fSReid Kleckner   // Skip this instruction if it has the same location as the previous one.
2639533af4fSReid Kleckner   if (DL == CurFn->LastLoc)
2649533af4fSReid Kleckner     return;
2659533af4fSReid Kleckner 
2669533af4fSReid Kleckner   const DIScope *Scope = DL.get()->getScope();
26770f5bc99SReid Kleckner   if (!Scope)
26870f5bc99SReid Kleckner     return;
2699533af4fSReid Kleckner 
27070f5bc99SReid Kleckner   // Skip this line if it is longer than the maximum we can record.
2712214ed89SReid Kleckner   LineInfo LI(DL.getLine(), DL.getLine(), /*IsStatement=*/true);
2722214ed89SReid Kleckner   if (LI.getStartLine() != DL.getLine() || LI.isAlwaysStepInto() ||
2732214ed89SReid Kleckner       LI.isNeverStepInto())
27470f5bc99SReid Kleckner     return;
27570f5bc99SReid Kleckner 
2762214ed89SReid Kleckner   ColumnInfo CI(DL.getCol(), /*EndColumn=*/0);
2772214ed89SReid Kleckner   if (CI.getStartColumn() != DL.getCol())
2782214ed89SReid Kleckner     return;
27900d9639cSReid Kleckner 
2802214ed89SReid Kleckner   if (!CurFn->HaveLineInfo)
2812214ed89SReid Kleckner     CurFn->HaveLineInfo = true;
2822214ed89SReid Kleckner   unsigned FileId = 0;
2832214ed89SReid Kleckner   if (CurFn->LastLoc.get() && CurFn->LastLoc->getFile() == DL->getFile())
2842214ed89SReid Kleckner     FileId = CurFn->LastFileId;
2852214ed89SReid Kleckner   else
2862214ed89SReid Kleckner     FileId = CurFn->LastFileId = maybeRecordFile(DL->getFile());
2872214ed89SReid Kleckner   CurFn->LastLoc = DL;
288f3b9ba49SReid Kleckner 
289f3b9ba49SReid Kleckner   unsigned FuncId = CurFn->FuncId;
290876330d5SReid Kleckner   if (const DILocation *SiteLoc = DL->getInlinedAt()) {
291829365aeSReid Kleckner     const DILocation *Loc = DL.get();
292829365aeSReid Kleckner 
293f3b9ba49SReid Kleckner     // If this location was actually inlined from somewhere else, give it the ID
294f3b9ba49SReid Kleckner     // of the inline call site.
295876330d5SReid Kleckner     FuncId =
296876330d5SReid Kleckner         getInlineSite(SiteLoc, Loc->getScope()->getSubprogram()).SiteFuncId;
297829365aeSReid Kleckner 
298f3b9ba49SReid Kleckner     // Ensure we have links in the tree of inline call sites.
299829365aeSReid Kleckner     bool FirstLoc = true;
300829365aeSReid Kleckner     while ((SiteLoc = Loc->getInlinedAt())) {
301876330d5SReid Kleckner       InlineSite &Site =
302876330d5SReid Kleckner           getInlineSite(SiteLoc, Loc->getScope()->getSubprogram());
303829365aeSReid Kleckner       if (!FirstLoc)
304829365aeSReid Kleckner         addLocIfNotPresent(Site.ChildSites, Loc);
305829365aeSReid Kleckner       FirstLoc = false;
306829365aeSReid Kleckner       Loc = SiteLoc;
307f3b9ba49SReid Kleckner     }
308829365aeSReid Kleckner     addLocIfNotPresent(CurFn->ChildSites, Loc);
309f3b9ba49SReid Kleckner   }
310f3b9ba49SReid Kleckner 
311dac21b43SReid Kleckner   OS.EmitCVLocDirective(FuncId, FileId, DL.getLine(), DL.getCol(),
312dac21b43SReid Kleckner                         /*PrologueEnd=*/false,
3132214ed89SReid Kleckner                         /*IsStmt=*/false, DL->getFilename());
31470f5bc99SReid Kleckner }
31570f5bc99SReid Kleckner 
3165d122f87SReid Kleckner void CodeViewDebug::emitCodeViewMagicVersion() {
3175d122f87SReid Kleckner   OS.EmitValueToAlignment(4);
3185d122f87SReid Kleckner   OS.AddComment("Debug section magic");
3195d122f87SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
3205d122f87SReid Kleckner }
3215d122f87SReid Kleckner 
32270f5bc99SReid Kleckner void CodeViewDebug::endModule() {
3236f3406dfSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
32470f5bc99SReid Kleckner     return;
32570f5bc99SReid Kleckner 
32670f5bc99SReid Kleckner   assert(Asm != nullptr);
32770f5bc99SReid Kleckner 
32870f5bc99SReid Kleckner   // The COFF .debug$S section consists of several subsections, each starting
32970f5bc99SReid Kleckner   // with a 4-byte control code (e.g. 0xF1, 0xF2, etc) and then a 4-byte length
33070f5bc99SReid Kleckner   // of the payload followed by the payload itself.  The subsections are 4-byte
33170f5bc99SReid Kleckner   // aligned.
33270f5bc99SReid Kleckner 
3336f3406dfSReid Kleckner   // Use the generic .debug$S section, and make a subsection for all the inlined
3346f3406dfSReid Kleckner   // subprograms.
3356f3406dfSReid Kleckner   switchToDebugSectionForSymbol(nullptr);
3365d122f87SReid Kleckner   emitInlineeLinesSubsection();
3371fcd610cSReid Kleckner 
3382214ed89SReid Kleckner   // Emit per-function debug information.
3392214ed89SReid Kleckner   for (auto &P : FnDebugInfo)
340577be0feSDavid Majnemer     if (!P.first->isDeclarationForLinker())
3412214ed89SReid Kleckner       emitDebugInfoForFunction(P.first, P.second);
34270f5bc99SReid Kleckner 
3436f3406dfSReid Kleckner   // Emit global variable debug information.
3443128b10cSDavid Majnemer   setCurrentSubprogram(nullptr);
3456f3406dfSReid Kleckner   emitDebugInfoForGlobals();
3466f3406dfSReid Kleckner 
347b510b458SHans Wennborg   // Emit retained types.
348b510b458SHans Wennborg   emitDebugInfoForRetainedTypes();
349b510b458SHans Wennborg 
3505d122f87SReid Kleckner   // Switch back to the generic .debug$S section after potentially processing
3515d122f87SReid Kleckner   // comdat symbol sections.
3525d122f87SReid Kleckner   switchToDebugSectionForSymbol(nullptr);
3535d122f87SReid Kleckner 
3543128b10cSDavid Majnemer   // Emit UDT records for any types used by global variables.
3553128b10cSDavid Majnemer   if (!GlobalUDTs.empty()) {
3563128b10cSDavid Majnemer     MCSymbol *SymbolsEnd = beginCVSubsection(ModuleSubstreamKind::Symbols);
3573128b10cSDavid Majnemer     emitDebugInfoForUDTs(GlobalUDTs);
3583128b10cSDavid Majnemer     endCVSubsection(SymbolsEnd);
3593128b10cSDavid Majnemer   }
3603128b10cSDavid Majnemer 
36170f5bc99SReid Kleckner   // This subsection holds a file index to offset in string table table.
362dac21b43SReid Kleckner   OS.AddComment("File index to string table offset subsection");
363dac21b43SReid Kleckner   OS.EmitCVFileChecksumsDirective();
36470f5bc99SReid Kleckner 
36570f5bc99SReid Kleckner   // This subsection holds the string table.
366dac21b43SReid Kleckner   OS.AddComment("String table");
367dac21b43SReid Kleckner   OS.EmitCVStringTableDirective();
36870f5bc99SReid Kleckner 
3695acacbb0SReid Kleckner   // Emit type information last, so that any types we translate while emitting
3705acacbb0SReid Kleckner   // function info are included.
3715acacbb0SReid Kleckner   emitTypeInformation();
3725acacbb0SReid Kleckner 
37370f5bc99SReid Kleckner   clear();
37470f5bc99SReid Kleckner }
37570f5bc99SReid Kleckner 
376b9456a5eSDavid Majnemer static void emitNullTerminatedSymbolName(MCStreamer &OS, StringRef S) {
377b9456a5eSDavid Majnemer   // Microsoft's linker seems to have trouble with symbol names longer than
378b9456a5eSDavid Majnemer   // 0xffd8 bytes.
379b9456a5eSDavid Majnemer   S = S.substr(0, 0xffd8);
380b9456a5eSDavid Majnemer   SmallString<32> NullTerminatedString(S);
381b9456a5eSDavid Majnemer   NullTerminatedString.push_back('\0');
382b9456a5eSDavid Majnemer   OS.EmitBytes(NullTerminatedString);
383b9456a5eSDavid Majnemer }
384b9456a5eSDavid Majnemer 
385f3b9ba49SReid Kleckner void CodeViewDebug::emitTypeInformation() {
3862280f932SReid Kleckner   // Do nothing if we have no debug info or if no non-trivial types were emitted
3872280f932SReid Kleckner   // to TypeTable during codegen.
38876c9eb99SAmjad Aboud   NamedMDNode *CU_Nodes = MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
389fbd7787dSReid Kleckner   if (!CU_Nodes)
390fbd7787dSReid Kleckner     return;
3912280f932SReid Kleckner   if (TypeTable.empty())
392fbd7787dSReid Kleckner     return;
393fbd7787dSReid Kleckner 
394f3b9ba49SReid Kleckner   // Start the .debug$T section with 0x4.
395dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugTypesSection());
3965d122f87SReid Kleckner   emitCodeViewMagicVersion();
397f3b9ba49SReid Kleckner 
398fbdbe9e2SReid Kleckner   SmallString<8> CommentPrefix;
399fbdbe9e2SReid Kleckner   if (OS.isVerboseAsm()) {
400fbdbe9e2SReid Kleckner     CommentPrefix += '\t';
401fbdbe9e2SReid Kleckner     CommentPrefix += Asm->MAI->getCommentString();
402fbdbe9e2SReid Kleckner     CommentPrefix += ' ';
403fbdbe9e2SReid Kleckner   }
404fbdbe9e2SReid Kleckner 
405fbdbe9e2SReid Kleckner   CVTypeDumper CVTD(nullptr, /*PrintRecordBytes=*/false);
4062280f932SReid Kleckner   TypeTable.ForEachRecord(
407fbdbe9e2SReid Kleckner       [&](TypeIndex Index, StringRef Record) {
408fbdbe9e2SReid Kleckner         if (OS.isVerboseAsm()) {
409fbdbe9e2SReid Kleckner           // Emit a block comment describing the type record for readability.
410fbdbe9e2SReid Kleckner           SmallString<512> CommentBlock;
411fbdbe9e2SReid Kleckner           raw_svector_ostream CommentOS(CommentBlock);
412fbdbe9e2SReid Kleckner           ScopedPrinter SP(CommentOS);
413fbdbe9e2SReid Kleckner           SP.setPrefix(CommentPrefix);
414fbdbe9e2SReid Kleckner           CVTD.setPrinter(&SP);
41501ee3daeSZachary Turner           Error EC = CVTD.dump({Record.bytes_begin(), Record.bytes_end()});
41601ee3daeSZachary Turner           assert(!EC && "produced malformed type record");
41701ee3daeSZachary Turner           consumeError(std::move(EC));
418fbdbe9e2SReid Kleckner           // emitRawComment will insert its own tab and comment string before
419fbdbe9e2SReid Kleckner           // the first line, so strip off our first one. It also prints its own
420fbdbe9e2SReid Kleckner           // newline.
421fbdbe9e2SReid Kleckner           OS.emitRawComment(
422fbdbe9e2SReid Kleckner               CommentOS.str().drop_front(CommentPrefix.size() - 1).rtrim());
423fbdbe9e2SReid Kleckner         }
424fbdbe9e2SReid Kleckner         OS.EmitBinaryData(Record);
4252280f932SReid Kleckner       });
426f3b9ba49SReid Kleckner }
427f3b9ba49SReid Kleckner 
4285d122f87SReid Kleckner void CodeViewDebug::emitInlineeLinesSubsection() {
4291fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
4301fcd610cSReid Kleckner     return;
4311fcd610cSReid Kleckner 
4321fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
4336f3406dfSReid Kleckner   MCSymbol *InlineEnd = beginCVSubsection(ModuleSubstreamKind::InlineeLines);
4341fcd610cSReid Kleckner 
4351fcd610cSReid Kleckner   // We don't provide any extra file info.
4361fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
43730579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
4381fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
4391fcd610cSReid Kleckner 
4401fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
44176c9eb99SAmjad Aboud     assert(TypeIndices.count({SP, nullptr}));
44276c9eb99SAmjad Aboud     TypeIndex InlineeIdx = TypeIndices[{SP, nullptr}];
4432280f932SReid Kleckner 
44430579ec8SDavid Majnemer     OS.AddBlankLine();
4451fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
4461fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
4471fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
44830579ec8SDavid Majnemer     OS.AddBlankLine();
4491fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
4501fcd610cSReid Kleckner     // 1.
4511fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
45230579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
4532280f932SReid Kleckner     OS.EmitIntValue(InlineeIdx.getIndex(), 4);
45430579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
4551fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
45630579ec8SDavid Majnemer     OS.AddComment("Starting line number");
4571fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
4581fcd610cSReid Kleckner   }
4591fcd610cSReid Kleckner 
4606f3406dfSReid Kleckner   endCVSubsection(InlineEnd);
4611fcd610cSReid Kleckner }
4621fcd610cSReid Kleckner 
4631fcd610cSReid Kleckner void CodeViewDebug::collectInlineSiteChildren(
4641fcd610cSReid Kleckner     SmallVectorImpl<unsigned> &Children, const FunctionInfo &FI,
4651fcd610cSReid Kleckner     const InlineSite &Site) {
4661fcd610cSReid Kleckner   for (const DILocation *ChildSiteLoc : Site.ChildSites) {
4671fcd610cSReid Kleckner     auto I = FI.InlineSites.find(ChildSiteLoc);
4681fcd610cSReid Kleckner     const InlineSite &ChildSite = I->second;
4691fcd610cSReid Kleckner     Children.push_back(ChildSite.SiteFuncId);
4701fcd610cSReid Kleckner     collectInlineSiteChildren(Children, FI, ChildSite);
4711fcd610cSReid Kleckner   }
4721fcd610cSReid Kleckner }
4731fcd610cSReid Kleckner 
474f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
475f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
476f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
477f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
478f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
479f3b9ba49SReid Kleckner 
48076c9eb99SAmjad Aboud   assert(TypeIndices.count({Site.Inlinee, nullptr}));
48176c9eb99SAmjad Aboud   TypeIndex InlineeIdx = TypeIndices[{Site.Inlinee, nullptr}];
482f3b9ba49SReid Kleckner 
483f3b9ba49SReid Kleckner   // SymbolRecord
484dac21b43SReid Kleckner   OS.AddComment("Record length");
485eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
486f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
487dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
48863a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE, 2); // RecordKind
489f3b9ba49SReid Kleckner 
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("Inlinee type index");
4952280f932SReid Kleckner   OS.EmitIntValue(InlineeIdx.getIndex(), 4);
496f3b9ba49SReid Kleckner 
4971fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
4981fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
4991fcd610cSReid Kleckner   SmallVector<unsigned, 3> SecondaryFuncIds;
5001fcd610cSReid Kleckner   collectInlineSiteChildren(SecondaryFuncIds, FI, Site);
5011fcd610cSReid Kleckner 
5021fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
503c9911f28SDavid Majnemer                                     FI.Begin, FI.End, SecondaryFuncIds);
504f3b9ba49SReid Kleckner 
505f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
506f3b9ba49SReid Kleckner 
507f9c275feSReid Kleckner   for (const LocalVariable &Var : Site.InlinedLocals)
508f9c275feSReid Kleckner     emitLocalVariable(Var);
509f9c275feSReid Kleckner 
510f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
511f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
512f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
513f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
514f3b9ba49SReid Kleckner            "child site not in function inline site map");
515f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
516f3b9ba49SReid Kleckner   }
517f3b9ba49SReid Kleckner 
518f3b9ba49SReid Kleckner   // Close the scope.
519dac21b43SReid Kleckner   OS.AddComment("Record length");
520dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
521dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
52263a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE_END, 2); // RecordKind
523f3b9ba49SReid Kleckner }
524f3b9ba49SReid Kleckner 
5255d122f87SReid Kleckner void CodeViewDebug::switchToDebugSectionForSymbol(const MCSymbol *GVSym) {
5265d122f87SReid Kleckner   // If we have a symbol, it may be in a section that is COMDAT. If so, find the
5275d122f87SReid Kleckner   // comdat key. A section may be comdat because of -ffunction-sections or
5285d122f87SReid Kleckner   // because it is comdat in the IR.
5295d122f87SReid Kleckner   MCSectionCOFF *GVSec =
5305d122f87SReid Kleckner       GVSym ? dyn_cast<MCSectionCOFF>(&GVSym->getSection()) : nullptr;
5315d122f87SReid Kleckner   const MCSymbol *KeySym = GVSec ? GVSec->getCOMDATSymbol() : nullptr;
5325d122f87SReid Kleckner 
5335d122f87SReid Kleckner   MCSectionCOFF *DebugSec = cast<MCSectionCOFF>(
5345d122f87SReid Kleckner       Asm->getObjFileLowering().getCOFFDebugSymbolsSection());
5355d122f87SReid Kleckner   DebugSec = OS.getContext().getAssociativeCOFFSection(DebugSec, KeySym);
5365d122f87SReid Kleckner 
5375d122f87SReid Kleckner   OS.SwitchSection(DebugSec);
5385d122f87SReid Kleckner 
5395d122f87SReid Kleckner   // Emit the magic version number if this is the first time we've switched to
5405d122f87SReid Kleckner   // this section.
5415d122f87SReid Kleckner   if (ComdatDebugSections.insert(DebugSec).second)
5425d122f87SReid Kleckner     emitCodeViewMagicVersion();
5435d122f87SReid Kleckner }
5445d122f87SReid Kleckner 
5452214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
5462214ed89SReid Kleckner                                              FunctionInfo &FI) {
54770f5bc99SReid Kleckner   // For each function there is a separate subsection
54870f5bc99SReid Kleckner   // which holds the PC to file:line table.
54970f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
55070f5bc99SReid Kleckner   assert(Fn);
55170f5bc99SReid Kleckner 
5525d122f87SReid Kleckner   // Switch to the to a comdat section, if appropriate.
5535d122f87SReid Kleckner   switchToDebugSectionForSymbol(Fn);
5545d122f87SReid Kleckner 
555ac945e27SReid Kleckner   std::string FuncName;
5563128b10cSDavid Majnemer   auto *SP = GV->getSubprogram();
5573128b10cSDavid Majnemer   setCurrentSubprogram(SP);
558ac945e27SReid Kleckner 
559ac945e27SReid Kleckner   // If we have a display name, build the fully qualified name by walking the
560ac945e27SReid Kleckner   // chain of scopes.
5610c5d874bSReid Kleckner   if (SP != nullptr && !SP->getDisplayName().empty())
5620c5d874bSReid Kleckner     FuncName =
5630c5d874bSReid Kleckner         getFullyQualifiedName(SP->getScope().resolve(), SP->getDisplayName());
56470f5bc99SReid Kleckner 
56570f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
56670f5bc99SReid Kleckner   if (FuncName.empty())
56770f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
56870f5bc99SReid Kleckner 
56970f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
570dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
5716f3406dfSReid Kleckner   MCSymbol *SymbolsEnd = beginCVSubsection(ModuleSubstreamKind::Symbols);
57270f5bc99SReid Kleckner   {
573f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
574f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
575dac21b43SReid Kleckner     OS.AddComment("Record length");
576eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
577dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
57870f5bc99SReid Kleckner 
579dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
58063a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_GPROC32_ID), 2);
58170f5bc99SReid Kleckner 
58230579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
583dac21b43SReid Kleckner     OS.AddComment("PtrParent");
584dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
585dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
586dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
587dac21b43SReid Kleckner     OS.AddComment("PtrNext");
588dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
58970f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
59070f5bc99SReid Kleckner     // code is located and what's its size:
591dac21b43SReid Kleckner     OS.AddComment("Code size");
592eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
593dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
594dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
595dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
596dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
597dac21b43SReid Kleckner     OS.AddComment("Function type index");
59875c3ebfaSDavid Majnemer     OS.EmitIntValue(getFuncIdForSubprogram(GV->getSubprogram()).getIndex(), 4);
599dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
600dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
601dac21b43SReid Kleckner     OS.AddComment("Function section index");
602dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
603dac21b43SReid Kleckner     OS.AddComment("Flags");
604dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
60570f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
606dac21b43SReid Kleckner     OS.AddComment("Function name");
6071256125fSDavid Majnemer     // Truncate the name so we won't overflow the record length field.
608b9456a5eSDavid Majnemer     emitNullTerminatedSymbolName(OS, FuncName);
609dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
61070f5bc99SReid Kleckner 
611f9c275feSReid Kleckner     for (const LocalVariable &Var : FI.Locals)
612f9c275feSReid Kleckner       emitLocalVariable(Var);
613f9c275feSReid Kleckner 
614f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
615f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
616f3b9ba49SReid Kleckner     // of their parent inline site.
617f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
618f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
619f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
620f9c275feSReid Kleckner              "child site not in function inline site map");
621f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
622f3b9ba49SReid Kleckner     }
623f3b9ba49SReid Kleckner 
6243128b10cSDavid Majnemer     if (SP != nullptr)
6253128b10cSDavid Majnemer       emitDebugInfoForUDTs(LocalUDTs);
6263128b10cSDavid Majnemer 
62770f5bc99SReid Kleckner     // We're done with this function.
628dac21b43SReid Kleckner     OS.AddComment("Record length");
629dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
630dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
63163a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_PROC_ID_END), 2);
63270f5bc99SReid Kleckner   }
6336f3406dfSReid Kleckner   endCVSubsection(SymbolsEnd);
63470f5bc99SReid Kleckner 
6352214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
636dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
63770f5bc99SReid Kleckner }
63870f5bc99SReid Kleckner 
639876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
640876330d5SReid Kleckner CodeViewDebug::createDefRangeMem(uint16_t CVRegister, int Offset) {
641876330d5SReid Kleckner   LocalVarDefRange DR;
642c6a2f214SAaron Ballman   DR.InMemory = -1;
643876330d5SReid Kleckner   DR.DataOffset = Offset;
644876330d5SReid Kleckner   assert(DR.DataOffset == Offset && "truncation");
645876330d5SReid Kleckner   DR.StructOffset = 0;
646876330d5SReid Kleckner   DR.CVRegister = CVRegister;
647876330d5SReid Kleckner   return DR;
648876330d5SReid Kleckner }
649876330d5SReid Kleckner 
650876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
651876330d5SReid Kleckner CodeViewDebug::createDefRangeReg(uint16_t CVRegister) {
652876330d5SReid Kleckner   LocalVarDefRange DR;
653876330d5SReid Kleckner   DR.InMemory = 0;
654876330d5SReid Kleckner   DR.DataOffset = 0;
655876330d5SReid Kleckner   DR.StructOffset = 0;
656876330d5SReid Kleckner   DR.CVRegister = CVRegister;
657876330d5SReid Kleckner   return DR;
658876330d5SReid Kleckner }
659876330d5SReid Kleckner 
660876330d5SReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable(
661876330d5SReid Kleckner     DenseSet<InlinedVariable> &Processed) {
662876330d5SReid Kleckner   const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
663876330d5SReid Kleckner   const TargetFrameLowering *TFI = TSI.getFrameLowering();
664876330d5SReid Kleckner   const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
665876330d5SReid Kleckner 
666876330d5SReid Kleckner   for (const MachineModuleInfo::VariableDbgInfo &VI :
667876330d5SReid Kleckner        MMI->getVariableDbgInfo()) {
668f9c275feSReid Kleckner     if (!VI.Var)
669f9c275feSReid Kleckner       continue;
670f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
671f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
672f9c275feSReid Kleckner 
673876330d5SReid Kleckner     Processed.insert(InlinedVariable(VI.Var, VI.Loc->getInlinedAt()));
674f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
675f9c275feSReid Kleckner 
676f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
677f9c275feSReid Kleckner     if (!Scope)
678f9c275feSReid Kleckner       continue;
679f9c275feSReid Kleckner 
680f9c275feSReid Kleckner     // Get the frame register used and the offset.
681f9c275feSReid Kleckner     unsigned FrameReg = 0;
682876330d5SReid Kleckner     int FrameOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
683876330d5SReid Kleckner     uint16_t CVReg = TRI->getCodeViewRegNum(FrameReg);
684f9c275feSReid Kleckner 
685f9c275feSReid Kleckner     // Calculate the label ranges.
686876330d5SReid Kleckner     LocalVarDefRange DefRange = createDefRangeMem(CVReg, FrameOffset);
687f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
688f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
689f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
690876330d5SReid Kleckner       End = End ? End : Asm->getFunctionEnd();
691876330d5SReid Kleckner       DefRange.Ranges.emplace_back(Begin, End);
692f9c275feSReid Kleckner     }
693f9c275feSReid Kleckner 
694876330d5SReid Kleckner     LocalVariable Var;
695876330d5SReid Kleckner     Var.DIVar = VI.Var;
696876330d5SReid Kleckner     Var.DefRanges.emplace_back(std::move(DefRange));
697876330d5SReid Kleckner     recordLocalVariable(std::move(Var), VI.Loc->getInlinedAt());
698f9c275feSReid Kleckner   }
699f9c275feSReid Kleckner }
700876330d5SReid Kleckner 
701876330d5SReid Kleckner void CodeViewDebug::collectVariableInfo(const DISubprogram *SP) {
702876330d5SReid Kleckner   DenseSet<InlinedVariable> Processed;
703876330d5SReid Kleckner   // Grab the variable info that was squirreled away in the MMI side-table.
704876330d5SReid Kleckner   collectVariableInfoFromMMITable(Processed);
705876330d5SReid Kleckner 
706876330d5SReid Kleckner   const TargetRegisterInfo *TRI = Asm->MF->getSubtarget().getRegisterInfo();
707876330d5SReid Kleckner 
708876330d5SReid Kleckner   for (const auto &I : DbgValues) {
709876330d5SReid Kleckner     InlinedVariable IV = I.first;
710876330d5SReid Kleckner     if (Processed.count(IV))
711876330d5SReid Kleckner       continue;
712876330d5SReid Kleckner     const DILocalVariable *DIVar = IV.first;
713876330d5SReid Kleckner     const DILocation *InlinedAt = IV.second;
714876330d5SReid Kleckner 
715876330d5SReid Kleckner     // Instruction ranges, specifying where IV is accessible.
716876330d5SReid Kleckner     const auto &Ranges = I.second;
717876330d5SReid Kleckner 
718876330d5SReid Kleckner     LexicalScope *Scope = nullptr;
719876330d5SReid Kleckner     if (InlinedAt)
720876330d5SReid Kleckner       Scope = LScopes.findInlinedScope(DIVar->getScope(), InlinedAt);
721876330d5SReid Kleckner     else
722876330d5SReid Kleckner       Scope = LScopes.findLexicalScope(DIVar->getScope());
723876330d5SReid Kleckner     // If variable scope is not found then skip this variable.
724876330d5SReid Kleckner     if (!Scope)
725876330d5SReid Kleckner       continue;
726876330d5SReid Kleckner 
727876330d5SReid Kleckner     LocalVariable Var;
728876330d5SReid Kleckner     Var.DIVar = DIVar;
729876330d5SReid Kleckner 
730876330d5SReid Kleckner     // Calculate the definition ranges.
731876330d5SReid Kleckner     for (auto I = Ranges.begin(), E = Ranges.end(); I != E; ++I) {
732876330d5SReid Kleckner       const InsnRange &Range = *I;
733876330d5SReid Kleckner       const MachineInstr *DVInst = Range.first;
734876330d5SReid Kleckner       assert(DVInst->isDebugValue() && "Invalid History entry");
735876330d5SReid Kleckner       const DIExpression *DIExpr = DVInst->getDebugExpression();
736876330d5SReid Kleckner 
737876330d5SReid Kleckner       // Bail if there is a complex DWARF expression for now.
738876330d5SReid Kleckner       if (DIExpr && DIExpr->getNumElements() > 0)
739876330d5SReid Kleckner         continue;
740876330d5SReid Kleckner 
7419a593ee7SReid Kleckner       // Bail if operand 0 is not a valid register. This means the variable is a
7429a593ee7SReid Kleckner       // simple constant, or is described by a complex expression.
7439a593ee7SReid Kleckner       // FIXME: Find a way to represent constant variables, since they are
7449a593ee7SReid Kleckner       // relatively common.
7459a593ee7SReid Kleckner       unsigned Reg =
7469a593ee7SReid Kleckner           DVInst->getOperand(0).isReg() ? DVInst->getOperand(0).getReg() : 0;
7479a593ee7SReid Kleckner       if (Reg == 0)
7486e0d5f57SReid Kleckner         continue;
7496e0d5f57SReid Kleckner 
750876330d5SReid Kleckner       // Handle the two cases we can handle: indirect in memory and in register.
751876330d5SReid Kleckner       bool IsIndirect = DVInst->getOperand(1).isImm();
752876330d5SReid Kleckner       unsigned CVReg = TRI->getCodeViewRegNum(DVInst->getOperand(0).getReg());
753876330d5SReid Kleckner       {
754876330d5SReid Kleckner         LocalVarDefRange DefRange;
755876330d5SReid Kleckner         if (IsIndirect) {
756876330d5SReid Kleckner           int64_t Offset = DVInst->getOperand(1).getImm();
757876330d5SReid Kleckner           DefRange = createDefRangeMem(CVReg, Offset);
758876330d5SReid Kleckner         } else {
759876330d5SReid Kleckner           DefRange = createDefRangeReg(CVReg);
760876330d5SReid Kleckner         }
761876330d5SReid Kleckner         if (Var.DefRanges.empty() ||
762876330d5SReid Kleckner             Var.DefRanges.back().isDifferentLocation(DefRange)) {
763876330d5SReid Kleckner           Var.DefRanges.emplace_back(std::move(DefRange));
764876330d5SReid Kleckner         }
765876330d5SReid Kleckner       }
766876330d5SReid Kleckner 
767876330d5SReid Kleckner       // Compute the label range.
768876330d5SReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
769876330d5SReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
770876330d5SReid Kleckner       if (!End) {
771876330d5SReid Kleckner         if (std::next(I) != E)
772876330d5SReid Kleckner           End = getLabelBeforeInsn(std::next(I)->first);
773876330d5SReid Kleckner         else
774876330d5SReid Kleckner           End = Asm->getFunctionEnd();
775876330d5SReid Kleckner       }
776876330d5SReid Kleckner 
777876330d5SReid Kleckner       // If the last range end is our begin, just extend the last range.
778876330d5SReid Kleckner       // Otherwise make a new range.
779876330d5SReid Kleckner       SmallVectorImpl<std::pair<const MCSymbol *, const MCSymbol *>> &Ranges =
780876330d5SReid Kleckner           Var.DefRanges.back().Ranges;
781876330d5SReid Kleckner       if (!Ranges.empty() && Ranges.back().second == Begin)
782876330d5SReid Kleckner         Ranges.back().second = End;
783876330d5SReid Kleckner       else
784876330d5SReid Kleckner         Ranges.emplace_back(Begin, End);
785876330d5SReid Kleckner 
786876330d5SReid Kleckner       // FIXME: Do more range combining.
787876330d5SReid Kleckner     }
788876330d5SReid Kleckner 
789876330d5SReid Kleckner     recordLocalVariable(std::move(Var), InlinedAt);
790876330d5SReid Kleckner   }
791f9c275feSReid Kleckner }
792f9c275feSReid Kleckner 
79370f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
79470f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
79570f5bc99SReid Kleckner 
796f9c275feSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
79770f5bc99SReid Kleckner     return;
79870f5bc99SReid Kleckner 
799f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
800f9c275feSReid Kleckner 
80170f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
80270f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
80370f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
8042214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
8051fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
80670f5bc99SReid Kleckner 
807f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
808f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
80970f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
81070f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
81170f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
81270f5bc99SReid Kleckner   bool EmptyPrologue = true;
81370f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
81470f5bc99SReid Kleckner     for (const auto &MI : MBB) {
815f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
816f9c275feSReid Kleckner           MI.getDebugLoc()) {
81770f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
81870f5bc99SReid Kleckner         break;
819f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
82070f5bc99SReid Kleckner         EmptyPrologue = false;
82170f5bc99SReid Kleckner       }
82270f5bc99SReid Kleckner     }
823f9c275feSReid Kleckner   }
824f9c275feSReid Kleckner 
82570f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
82670f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
82770f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
82870f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
82970f5bc99SReid Kleckner   }
83070f5bc99SReid Kleckner }
83170f5bc99SReid Kleckner 
832*4b63a98dSHans Wennborg void CodeViewDebug::addToUDTs(const DIType *Ty, TypeIndex TI) {
833*4b63a98dSHans Wennborg   SmallVector<StringRef, 5> QualifiedNameComponents;
834*4b63a98dSHans Wennborg   const DISubprogram *ClosestSubprogram = getQualifiedNameComponents(
835*4b63a98dSHans Wennborg       Ty->getScope().resolve(), QualifiedNameComponents);
836*4b63a98dSHans Wennborg 
837*4b63a98dSHans Wennborg   std::string FullyQualifiedName =
838*4b63a98dSHans Wennborg       getQualifiedName(QualifiedNameComponents, Ty->getName());
839*4b63a98dSHans Wennborg 
840*4b63a98dSHans Wennborg   if (ClosestSubprogram == nullptr)
841*4b63a98dSHans Wennborg     GlobalUDTs.emplace_back(std::move(FullyQualifiedName), TI);
842*4b63a98dSHans Wennborg   else if (ClosestSubprogram == CurrentSubprogram)
843*4b63a98dSHans Wennborg     LocalUDTs.emplace_back(std::move(FullyQualifiedName), TI);
844*4b63a98dSHans Wennborg 
845*4b63a98dSHans Wennborg   // TODO: What if the ClosestSubprogram is neither null or the current
846*4b63a98dSHans Wennborg   // subprogram?  Currently, the UDT just gets dropped on the floor.
847*4b63a98dSHans Wennborg   //
848*4b63a98dSHans Wennborg   // The current behavior is not desirable.  To get maximal fidelity, we would
849*4b63a98dSHans Wennborg   // need to perform all type translation before beginning emission of .debug$S
850*4b63a98dSHans Wennborg   // and then make LocalUDTs a member of FunctionInfo
851*4b63a98dSHans Wennborg }
852*4b63a98dSHans Wennborg 
85376c9eb99SAmjad Aboud TypeIndex CodeViewDebug::lowerType(const DIType *Ty, const DIType *ClassTy) {
8545acacbb0SReid Kleckner   // Generic dispatch for lowering an unknown type.
8555acacbb0SReid Kleckner   switch (Ty->getTag()) {
856f3c3c132SAdrian McCarthy   case dwarf::DW_TAG_array_type:
857f3c3c132SAdrian McCarthy     return lowerTypeArray(cast<DICompositeType>(Ty));
858d065e23dSDavid Majnemer   case dwarf::DW_TAG_typedef:
859d065e23dSDavid Majnemer     return lowerTypeAlias(cast<DIDerivedType>(Ty));
8605acacbb0SReid Kleckner   case dwarf::DW_TAG_base_type:
8615acacbb0SReid Kleckner     return lowerTypeBasic(cast<DIBasicType>(Ty));
8625acacbb0SReid Kleckner   case dwarf::DW_TAG_pointer_type:
8635acacbb0SReid Kleckner   case dwarf::DW_TAG_reference_type:
8645acacbb0SReid Kleckner   case dwarf::DW_TAG_rvalue_reference_type:
8655acacbb0SReid Kleckner     return lowerTypePointer(cast<DIDerivedType>(Ty));
8665acacbb0SReid Kleckner   case dwarf::DW_TAG_ptr_to_member_type:
8675acacbb0SReid Kleckner     return lowerTypeMemberPointer(cast<DIDerivedType>(Ty));
8685acacbb0SReid Kleckner   case dwarf::DW_TAG_const_type:
8695acacbb0SReid Kleckner   case dwarf::DW_TAG_volatile_type:
8705acacbb0SReid Kleckner     return lowerTypeModifier(cast<DIDerivedType>(Ty));
87175c3ebfaSDavid Majnemer   case dwarf::DW_TAG_subroutine_type:
8720c5d874bSReid Kleckner     if (ClassTy) {
8730c5d874bSReid Kleckner       // The member function type of a member function pointer has no
8740c5d874bSReid Kleckner       // ThisAdjustment.
8750c5d874bSReid Kleckner       return lowerTypeMemberFunction(cast<DISubroutineType>(Ty), ClassTy,
8760c5d874bSReid Kleckner                                      /*ThisAdjustment=*/0);
8770c5d874bSReid Kleckner     }
87875c3ebfaSDavid Majnemer     return lowerTypeFunction(cast<DISubroutineType>(Ty));
879979cb888SDavid Majnemer   case dwarf::DW_TAG_enumeration_type:
880979cb888SDavid Majnemer     return lowerTypeEnum(cast<DICompositeType>(Ty));
881a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
882a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
883a8d57407SReid Kleckner     return lowerTypeClass(cast<DICompositeType>(Ty));
884a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
885a8d57407SReid Kleckner     return lowerTypeUnion(cast<DICompositeType>(Ty));
8865acacbb0SReid Kleckner   default:
8875acacbb0SReid Kleckner     // Use the null type index.
8885acacbb0SReid Kleckner     return TypeIndex();
8895acacbb0SReid Kleckner   }
8905acacbb0SReid Kleckner }
8915acacbb0SReid Kleckner 
892d065e23dSDavid Majnemer TypeIndex CodeViewDebug::lowerTypeAlias(const DIDerivedType *Ty) {
893d065e23dSDavid Majnemer   DITypeRef UnderlyingTypeRef = Ty->getBaseType();
894d065e23dSDavid Majnemer   TypeIndex UnderlyingTypeIndex = getTypeIndex(UnderlyingTypeRef);
8953128b10cSDavid Majnemer   StringRef TypeName = Ty->getName();
8963128b10cSDavid Majnemer 
897*4b63a98dSHans Wennborg   addToUDTs(Ty, UnderlyingTypeIndex);
8983128b10cSDavid Majnemer 
899d065e23dSDavid Majnemer   if (UnderlyingTypeIndex == TypeIndex(SimpleTypeKind::Int32Long) &&
9003128b10cSDavid Majnemer       TypeName == "HRESULT")
901d065e23dSDavid Majnemer     return TypeIndex(SimpleTypeKind::HResult);
9028c46a4ceSDavid Majnemer   if (UnderlyingTypeIndex == TypeIndex(SimpleTypeKind::UInt16Short) &&
9033128b10cSDavid Majnemer       TypeName == "wchar_t")
9048c46a4ceSDavid Majnemer     return TypeIndex(SimpleTypeKind::WideCharacter);
905*4b63a98dSHans Wennborg 
906d065e23dSDavid Majnemer   return UnderlyingTypeIndex;
907d065e23dSDavid Majnemer }
908d065e23dSDavid Majnemer 
909f3c3c132SAdrian McCarthy TypeIndex CodeViewDebug::lowerTypeArray(const DICompositeType *Ty) {
910f3c3c132SAdrian McCarthy   DITypeRef ElementTypeRef = Ty->getBaseType();
911f3c3c132SAdrian McCarthy   TypeIndex ElementTypeIndex = getTypeIndex(ElementTypeRef);
912f3c3c132SAdrian McCarthy   // IndexType is size_t, which depends on the bitness of the target.
913f3c3c132SAdrian McCarthy   TypeIndex IndexType = Asm->MAI->getPointerSize() == 8
914f3c3c132SAdrian McCarthy                             ? TypeIndex(SimpleTypeKind::UInt64Quad)
915f3c3c132SAdrian McCarthy                             : TypeIndex(SimpleTypeKind::UInt32Long);
916f3c3c132SAdrian McCarthy   uint64_t Size = Ty->getSizeInBits() / 8;
917f3c3c132SAdrian McCarthy   ArrayRecord Record(ElementTypeIndex, IndexType, Size, Ty->getName());
918f3c3c132SAdrian McCarthy   return TypeTable.writeArray(Record);
919f3c3c132SAdrian McCarthy }
920f3c3c132SAdrian McCarthy 
9215acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeBasic(const DIBasicType *Ty) {
9225acacbb0SReid Kleckner   TypeIndex Index;
9235acacbb0SReid Kleckner   dwarf::TypeKind Kind;
9245acacbb0SReid Kleckner   uint32_t ByteSize;
9255acacbb0SReid Kleckner 
9265acacbb0SReid Kleckner   Kind = static_cast<dwarf::TypeKind>(Ty->getEncoding());
927afefa673SDavid Majnemer   ByteSize = Ty->getSizeInBits() / 8;
9285acacbb0SReid Kleckner 
9295acacbb0SReid Kleckner   SimpleTypeKind STK = SimpleTypeKind::None;
9305acacbb0SReid Kleckner   switch (Kind) {
9315acacbb0SReid Kleckner   case dwarf::DW_ATE_address:
9325acacbb0SReid Kleckner     // FIXME: Translate
9335acacbb0SReid Kleckner     break;
9345acacbb0SReid Kleckner   case dwarf::DW_ATE_boolean:
9355acacbb0SReid Kleckner     switch (ByteSize) {
9365acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::Boolean8;   break;
9375acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::Boolean16;  break;
9385acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Boolean32;  break;
9395acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Boolean64;  break;
9401c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::Boolean128; break;
9415acacbb0SReid Kleckner     }
9425acacbb0SReid Kleckner     break;
9435acacbb0SReid Kleckner   case dwarf::DW_ATE_complex_float:
9445acacbb0SReid Kleckner     switch (ByteSize) {
9451c2cb1ddSDavid Majnemer     case 2:  STK = SimpleTypeKind::Complex16;  break;
9465acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Complex32;  break;
9475acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Complex64;  break;
9485acacbb0SReid Kleckner     case 10: STK = SimpleTypeKind::Complex80;  break;
9495acacbb0SReid Kleckner     case 16: STK = SimpleTypeKind::Complex128; break;
9505acacbb0SReid Kleckner     }
9515acacbb0SReid Kleckner     break;
9525acacbb0SReid Kleckner   case dwarf::DW_ATE_float:
9535acacbb0SReid Kleckner     switch (ByteSize) {
9541c2cb1ddSDavid Majnemer     case 2:  STK = SimpleTypeKind::Float16;  break;
9555acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Float32;  break;
9565acacbb0SReid Kleckner     case 6:  STK = SimpleTypeKind::Float48;  break;
9575acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Float64;  break;
9585acacbb0SReid Kleckner     case 10: STK = SimpleTypeKind::Float80;  break;
9595acacbb0SReid Kleckner     case 16: STK = SimpleTypeKind::Float128; break;
9605acacbb0SReid Kleckner     }
9615acacbb0SReid Kleckner     break;
9625acacbb0SReid Kleckner   case dwarf::DW_ATE_signed:
9635acacbb0SReid Kleckner     switch (ByteSize) {
9645acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::SByte;      break;
9655acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::Int16Short; break;
9665acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Int32;      break;
9671c2cb1ddSDavid Majnemer     case 8:  STK = SimpleTypeKind::Int64Quad;  break;
9681c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::Int128Oct;  break;
9695acacbb0SReid Kleckner     }
9705acacbb0SReid Kleckner     break;
9715acacbb0SReid Kleckner   case dwarf::DW_ATE_unsigned:
9725acacbb0SReid Kleckner     switch (ByteSize) {
9735acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::Byte;        break;
9745acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::UInt16Short; break;
9755acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::UInt32;      break;
9761c2cb1ddSDavid Majnemer     case 8:  STK = SimpleTypeKind::UInt64Quad;  break;
9771c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::UInt128Oct;  break;
9785acacbb0SReid Kleckner     }
9795acacbb0SReid Kleckner     break;
9805acacbb0SReid Kleckner   case dwarf::DW_ATE_UTF:
9815acacbb0SReid Kleckner     switch (ByteSize) {
9825acacbb0SReid Kleckner     case 2: STK = SimpleTypeKind::Character16; break;
9835acacbb0SReid Kleckner     case 4: STK = SimpleTypeKind::Character32; break;
9845acacbb0SReid Kleckner     }
9855acacbb0SReid Kleckner     break;
9865acacbb0SReid Kleckner   case dwarf::DW_ATE_signed_char:
9875acacbb0SReid Kleckner     if (ByteSize == 1)
9885acacbb0SReid Kleckner       STK = SimpleTypeKind::SignedCharacter;
9895acacbb0SReid Kleckner     break;
9905acacbb0SReid Kleckner   case dwarf::DW_ATE_unsigned_char:
9915acacbb0SReid Kleckner     if (ByteSize == 1)
9925acacbb0SReid Kleckner       STK = SimpleTypeKind::UnsignedCharacter;
9935acacbb0SReid Kleckner     break;
9945acacbb0SReid Kleckner   default:
9955acacbb0SReid Kleckner     break;
9965acacbb0SReid Kleckner   }
9975acacbb0SReid Kleckner 
9985acacbb0SReid Kleckner   // Apply some fixups based on the source-level type name.
9995acacbb0SReid Kleckner   if (STK == SimpleTypeKind::Int32 && Ty->getName() == "long int")
10005acacbb0SReid Kleckner     STK = SimpleTypeKind::Int32Long;
10015acacbb0SReid Kleckner   if (STK == SimpleTypeKind::UInt32 && Ty->getName() == "long unsigned int")
10025acacbb0SReid Kleckner     STK = SimpleTypeKind::UInt32Long;
10038c46a4ceSDavid Majnemer   if (STK == SimpleTypeKind::UInt16Short &&
10048c46a4ceSDavid Majnemer       (Ty->getName() == "wchar_t" || Ty->getName() == "__wchar_t"))
10055acacbb0SReid Kleckner     STK = SimpleTypeKind::WideCharacter;
10065acacbb0SReid Kleckner   if ((STK == SimpleTypeKind::SignedCharacter ||
10075acacbb0SReid Kleckner        STK == SimpleTypeKind::UnsignedCharacter) &&
10085acacbb0SReid Kleckner       Ty->getName() == "char")
10095acacbb0SReid Kleckner     STK = SimpleTypeKind::NarrowCharacter;
10105acacbb0SReid Kleckner 
10115acacbb0SReid Kleckner   return TypeIndex(STK);
10125acacbb0SReid Kleckner }
10135acacbb0SReid Kleckner 
10145acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypePointer(const DIDerivedType *Ty) {
10155acacbb0SReid Kleckner   TypeIndex PointeeTI = getTypeIndex(Ty->getBaseType());
10165acacbb0SReid Kleckner 
101776c9eb99SAmjad Aboud   // While processing the type being pointed to it is possible we already
101876c9eb99SAmjad Aboud   // created this pointer type.  If so, we check here and return the existing
101976c9eb99SAmjad Aboud   // pointer type.
102076c9eb99SAmjad Aboud   auto I = TypeIndices.find({Ty, nullptr});
102176c9eb99SAmjad Aboud   if (I != TypeIndices.end())
102276c9eb99SAmjad Aboud     return I->second;
102376c9eb99SAmjad Aboud 
10245acacbb0SReid Kleckner   // Pointers to simple types can use SimpleTypeMode, rather than having a
10255acacbb0SReid Kleckner   // dedicated pointer type record.
10265acacbb0SReid Kleckner   if (PointeeTI.isSimple() &&
10275acacbb0SReid Kleckner       PointeeTI.getSimpleMode() == SimpleTypeMode::Direct &&
10285acacbb0SReid Kleckner       Ty->getTag() == dwarf::DW_TAG_pointer_type) {
10295acacbb0SReid Kleckner     SimpleTypeMode Mode = Ty->getSizeInBits() == 64
10305acacbb0SReid Kleckner                               ? SimpleTypeMode::NearPointer64
10315acacbb0SReid Kleckner                               : SimpleTypeMode::NearPointer32;
10325acacbb0SReid Kleckner     return TypeIndex(PointeeTI.getSimpleKind(), Mode);
10335acacbb0SReid Kleckner   }
10345acacbb0SReid Kleckner 
10355acacbb0SReid Kleckner   PointerKind PK =
10365acacbb0SReid Kleckner       Ty->getSizeInBits() == 64 ? PointerKind::Near64 : PointerKind::Near32;
10375acacbb0SReid Kleckner   PointerMode PM = PointerMode::Pointer;
10385acacbb0SReid Kleckner   switch (Ty->getTag()) {
10395acacbb0SReid Kleckner   default: llvm_unreachable("not a pointer tag type");
10405acacbb0SReid Kleckner   case dwarf::DW_TAG_pointer_type:
10415acacbb0SReid Kleckner     PM = PointerMode::Pointer;
10425acacbb0SReid Kleckner     break;
10435acacbb0SReid Kleckner   case dwarf::DW_TAG_reference_type:
10445acacbb0SReid Kleckner     PM = PointerMode::LValueReference;
10455acacbb0SReid Kleckner     break;
10465acacbb0SReid Kleckner   case dwarf::DW_TAG_rvalue_reference_type:
10475acacbb0SReid Kleckner     PM = PointerMode::RValueReference;
10485acacbb0SReid Kleckner     break;
10495acacbb0SReid Kleckner   }
10505acacbb0SReid Kleckner   // FIXME: MSVC folds qualifiers into PointerOptions in the context of a method
10515acacbb0SReid Kleckner   // 'this' pointer, but not normal contexts. Figure out what we're supposed to
10525acacbb0SReid Kleckner   // do.
10535acacbb0SReid Kleckner   PointerOptions PO = PointerOptions::None;
10545acacbb0SReid Kleckner   PointerRecord PR(PointeeTI, PK, PM, PO, Ty->getSizeInBits() / 8);
10555acacbb0SReid Kleckner   return TypeTable.writePointer(PR);
10565acacbb0SReid Kleckner }
10575acacbb0SReid Kleckner 
10586fa1546aSReid Kleckner static PointerToMemberRepresentation
10596fa1546aSReid Kleckner translatePtrToMemberRep(unsigned SizeInBytes, bool IsPMF, unsigned Flags) {
10606fa1546aSReid Kleckner   // SizeInBytes being zero generally implies that the member pointer type was
10616fa1546aSReid Kleckner   // incomplete, which can happen if it is part of a function prototype. In this
10626fa1546aSReid Kleckner   // case, use the unknown model instead of the general model.
1063604105bbSReid Kleckner   if (IsPMF) {
1064604105bbSReid Kleckner     switch (Flags & DINode::FlagPtrToMemberRep) {
1065604105bbSReid Kleckner     case 0:
10666fa1546aSReid Kleckner       return SizeInBytes == 0 ? PointerToMemberRepresentation::Unknown
10676fa1546aSReid Kleckner                               : PointerToMemberRepresentation::GeneralFunction;
1068604105bbSReid Kleckner     case DINode::FlagSingleInheritance:
1069604105bbSReid Kleckner       return PointerToMemberRepresentation::SingleInheritanceFunction;
1070604105bbSReid Kleckner     case DINode::FlagMultipleInheritance:
1071604105bbSReid Kleckner       return PointerToMemberRepresentation::MultipleInheritanceFunction;
1072604105bbSReid Kleckner     case DINode::FlagVirtualInheritance:
1073604105bbSReid Kleckner       return PointerToMemberRepresentation::VirtualInheritanceFunction;
1074604105bbSReid Kleckner     }
1075604105bbSReid Kleckner   } else {
1076604105bbSReid Kleckner     switch (Flags & DINode::FlagPtrToMemberRep) {
1077604105bbSReid Kleckner     case 0:
10786fa1546aSReid Kleckner       return SizeInBytes == 0 ? PointerToMemberRepresentation::Unknown
10796fa1546aSReid Kleckner                               : PointerToMemberRepresentation::GeneralData;
1080604105bbSReid Kleckner     case DINode::FlagSingleInheritance:
1081604105bbSReid Kleckner       return PointerToMemberRepresentation::SingleInheritanceData;
1082604105bbSReid Kleckner     case DINode::FlagMultipleInheritance:
1083604105bbSReid Kleckner       return PointerToMemberRepresentation::MultipleInheritanceData;
1084604105bbSReid Kleckner     case DINode::FlagVirtualInheritance:
1085604105bbSReid Kleckner       return PointerToMemberRepresentation::VirtualInheritanceData;
1086604105bbSReid Kleckner     }
1087604105bbSReid Kleckner   }
1088604105bbSReid Kleckner   llvm_unreachable("invalid ptr to member representation");
1089604105bbSReid Kleckner }
1090604105bbSReid Kleckner 
10915acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeMemberPointer(const DIDerivedType *Ty) {
10925acacbb0SReid Kleckner   assert(Ty->getTag() == dwarf::DW_TAG_ptr_to_member_type);
10935acacbb0SReid Kleckner   TypeIndex ClassTI = getTypeIndex(Ty->getClassType());
109476c9eb99SAmjad Aboud   TypeIndex PointeeTI = getTypeIndex(Ty->getBaseType(), Ty->getClassType());
10955acacbb0SReid Kleckner   PointerKind PK = Asm->MAI->getPointerSize() == 8 ? PointerKind::Near64
10965acacbb0SReid Kleckner                                                    : PointerKind::Near32;
1097604105bbSReid Kleckner   bool IsPMF = isa<DISubroutineType>(Ty->getBaseType());
1098604105bbSReid Kleckner   PointerMode PM = IsPMF ? PointerMode::PointerToMemberFunction
10995acacbb0SReid Kleckner                          : PointerMode::PointerToDataMember;
11005acacbb0SReid Kleckner   PointerOptions PO = PointerOptions::None; // FIXME
11016fa1546aSReid Kleckner   assert(Ty->getSizeInBits() / 8 <= 0xff && "pointer size too big");
11026fa1546aSReid Kleckner   uint8_t SizeInBytes = Ty->getSizeInBits() / 8;
11036fa1546aSReid Kleckner   MemberPointerInfo MPI(
11046fa1546aSReid Kleckner       ClassTI, translatePtrToMemberRep(SizeInBytes, IsPMF, Ty->getFlags()));
1105604105bbSReid Kleckner   PointerRecord PR(PointeeTI, PK, PM, PO, SizeInBytes, MPI);
11065acacbb0SReid Kleckner   return TypeTable.writePointer(PR);
11075acacbb0SReid Kleckner }
11085acacbb0SReid Kleckner 
1109de3d8b50SReid Kleckner /// Given a DWARF calling convention, get the CodeView equivalent. If we don't
1110de3d8b50SReid Kleckner /// have a translation, use the NearC convention.
1111de3d8b50SReid Kleckner static CallingConvention dwarfCCToCodeView(unsigned DwarfCC) {
1112de3d8b50SReid Kleckner   switch (DwarfCC) {
1113de3d8b50SReid Kleckner   case dwarf::DW_CC_normal:             return CallingConvention::NearC;
1114de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_msfastcall: return CallingConvention::NearFast;
1115de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_thiscall:   return CallingConvention::ThisCall;
1116de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_stdcall:    return CallingConvention::NearStdCall;
1117de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_pascal:     return CallingConvention::NearPascal;
1118de3d8b50SReid Kleckner   case dwarf::DW_CC_LLVM_vectorcall:    return CallingConvention::NearVector;
1119de3d8b50SReid Kleckner   }
1120de3d8b50SReid Kleckner   return CallingConvention::NearC;
1121de3d8b50SReid Kleckner }
1122de3d8b50SReid Kleckner 
11235acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeModifier(const DIDerivedType *Ty) {
11245acacbb0SReid Kleckner   ModifierOptions Mods = ModifierOptions::None;
11255acacbb0SReid Kleckner   bool IsModifier = true;
11265acacbb0SReid Kleckner   const DIType *BaseTy = Ty;
1127b9c80fd8SReid Kleckner   while (IsModifier && BaseTy) {
11285acacbb0SReid Kleckner     // FIXME: Need to add DWARF tag for __unaligned.
11295acacbb0SReid Kleckner     switch (BaseTy->getTag()) {
11305acacbb0SReid Kleckner     case dwarf::DW_TAG_const_type:
11315acacbb0SReid Kleckner       Mods |= ModifierOptions::Const;
11325acacbb0SReid Kleckner       break;
11335acacbb0SReid Kleckner     case dwarf::DW_TAG_volatile_type:
11345acacbb0SReid Kleckner       Mods |= ModifierOptions::Volatile;
11355acacbb0SReid Kleckner       break;
11365acacbb0SReid Kleckner     default:
11375acacbb0SReid Kleckner       IsModifier = false;
11385acacbb0SReid Kleckner       break;
11395acacbb0SReid Kleckner     }
11405acacbb0SReid Kleckner     if (IsModifier)
11415acacbb0SReid Kleckner       BaseTy = cast<DIDerivedType>(BaseTy)->getBaseType().resolve();
11425acacbb0SReid Kleckner   }
11435acacbb0SReid Kleckner   TypeIndex ModifiedTI = getTypeIndex(BaseTy);
114476c9eb99SAmjad Aboud 
114576c9eb99SAmjad Aboud   // While processing the type being pointed to, it is possible we already
114676c9eb99SAmjad Aboud   // created this modifier type.  If so, we check here and return the existing
114776c9eb99SAmjad Aboud   // modifier type.
114876c9eb99SAmjad Aboud   auto I = TypeIndices.find({Ty, nullptr});
114976c9eb99SAmjad Aboud   if (I != TypeIndices.end())
115076c9eb99SAmjad Aboud     return I->second;
115176c9eb99SAmjad Aboud 
11525acacbb0SReid Kleckner   ModifierRecord MR(ModifiedTI, Mods);
11535acacbb0SReid Kleckner   return TypeTable.writeModifier(MR);
11545acacbb0SReid Kleckner }
11555acacbb0SReid Kleckner 
115675c3ebfaSDavid Majnemer TypeIndex CodeViewDebug::lowerTypeFunction(const DISubroutineType *Ty) {
115775c3ebfaSDavid Majnemer   SmallVector<TypeIndex, 8> ReturnAndArgTypeIndices;
115875c3ebfaSDavid Majnemer   for (DITypeRef ArgTypeRef : Ty->getTypeArray())
115975c3ebfaSDavid Majnemer     ReturnAndArgTypeIndices.push_back(getTypeIndex(ArgTypeRef));
116075c3ebfaSDavid Majnemer 
116175c3ebfaSDavid Majnemer   TypeIndex ReturnTypeIndex = TypeIndex::Void();
116275c3ebfaSDavid Majnemer   ArrayRef<TypeIndex> ArgTypeIndices = None;
116375c3ebfaSDavid Majnemer   if (!ReturnAndArgTypeIndices.empty()) {
116475c3ebfaSDavid Majnemer     auto ReturnAndArgTypesRef = makeArrayRef(ReturnAndArgTypeIndices);
116575c3ebfaSDavid Majnemer     ReturnTypeIndex = ReturnAndArgTypesRef.front();
116675c3ebfaSDavid Majnemer     ArgTypeIndices = ReturnAndArgTypesRef.drop_front();
116775c3ebfaSDavid Majnemer   }
116875c3ebfaSDavid Majnemer 
116975c3ebfaSDavid Majnemer   ArgListRecord ArgListRec(TypeRecordKind::ArgList, ArgTypeIndices);
117075c3ebfaSDavid Majnemer   TypeIndex ArgListIndex = TypeTable.writeArgList(ArgListRec);
117175c3ebfaSDavid Majnemer 
1172de3d8b50SReid Kleckner   CallingConvention CC = dwarfCCToCodeView(Ty->getCC());
1173de3d8b50SReid Kleckner 
1174de3d8b50SReid Kleckner   ProcedureRecord Procedure(ReturnTypeIndex, CC, FunctionOptions::None,
1175de3d8b50SReid Kleckner                             ArgTypeIndices.size(), ArgListIndex);
117675c3ebfaSDavid Majnemer   return TypeTable.writeProcedure(Procedure);
117775c3ebfaSDavid Majnemer }
117875c3ebfaSDavid Majnemer 
117976c9eb99SAmjad Aboud TypeIndex CodeViewDebug::lowerTypeMemberFunction(const DISubroutineType *Ty,
11800c5d874bSReid Kleckner                                                  const DIType *ClassTy,
11810c5d874bSReid Kleckner                                                  int ThisAdjustment) {
118276c9eb99SAmjad Aboud   // Lower the containing class type.
118376c9eb99SAmjad Aboud   TypeIndex ClassType = getTypeIndex(ClassTy);
118476c9eb99SAmjad Aboud 
118576c9eb99SAmjad Aboud   SmallVector<TypeIndex, 8> ReturnAndArgTypeIndices;
118676c9eb99SAmjad Aboud   for (DITypeRef ArgTypeRef : Ty->getTypeArray())
118776c9eb99SAmjad Aboud     ReturnAndArgTypeIndices.push_back(getTypeIndex(ArgTypeRef));
118876c9eb99SAmjad Aboud 
118976c9eb99SAmjad Aboud   TypeIndex ReturnTypeIndex = TypeIndex::Void();
119076c9eb99SAmjad Aboud   ArrayRef<TypeIndex> ArgTypeIndices = None;
119176c9eb99SAmjad Aboud   if (!ReturnAndArgTypeIndices.empty()) {
119276c9eb99SAmjad Aboud     auto ReturnAndArgTypesRef = makeArrayRef(ReturnAndArgTypeIndices);
119376c9eb99SAmjad Aboud     ReturnTypeIndex = ReturnAndArgTypesRef.front();
119476c9eb99SAmjad Aboud     ArgTypeIndices = ReturnAndArgTypesRef.drop_front();
119576c9eb99SAmjad Aboud   }
119676c9eb99SAmjad Aboud   TypeIndex ThisTypeIndex = TypeIndex::Void();
119776c9eb99SAmjad Aboud   if (!ArgTypeIndices.empty()) {
119876c9eb99SAmjad Aboud     ThisTypeIndex = ArgTypeIndices.front();
119976c9eb99SAmjad Aboud     ArgTypeIndices = ArgTypeIndices.drop_front();
120076c9eb99SAmjad Aboud   }
120176c9eb99SAmjad Aboud 
120276c9eb99SAmjad Aboud   ArgListRecord ArgListRec(TypeRecordKind::ArgList, ArgTypeIndices);
120376c9eb99SAmjad Aboud   TypeIndex ArgListIndex = TypeTable.writeArgList(ArgListRec);
120476c9eb99SAmjad Aboud 
120576c9eb99SAmjad Aboud   CallingConvention CC = dwarfCCToCodeView(Ty->getCC());
120676c9eb99SAmjad Aboud 
120776c9eb99SAmjad Aboud   // TODO: Need to use the correct values for:
120876c9eb99SAmjad Aboud   //       FunctionOptions
120976c9eb99SAmjad Aboud   //       ThisPointerAdjustment.
121076c9eb99SAmjad Aboud   TypeIndex TI = TypeTable.writeMemberFunction(MemberFunctionRecord(
121176c9eb99SAmjad Aboud       ReturnTypeIndex, ClassType, ThisTypeIndex, CC, FunctionOptions::None,
12120c5d874bSReid Kleckner       ArgTypeIndices.size(), ArgListIndex, ThisAdjustment));
121376c9eb99SAmjad Aboud 
121476c9eb99SAmjad Aboud   return TI;
121576c9eb99SAmjad Aboud }
121676c9eb99SAmjad Aboud 
121776c9eb99SAmjad Aboud static MemberAccess translateAccessFlags(unsigned RecordTag, unsigned Flags) {
121876c9eb99SAmjad Aboud   switch (Flags & DINode::FlagAccessibility) {
1219a8d57407SReid Kleckner   case DINode::FlagPrivate:   return MemberAccess::Private;
1220a8d57407SReid Kleckner   case DINode::FlagPublic:    return MemberAccess::Public;
1221a8d57407SReid Kleckner   case DINode::FlagProtected: return MemberAccess::Protected;
1222a8d57407SReid Kleckner   case 0:
1223a8d57407SReid Kleckner     // If there was no explicit access control, provide the default for the tag.
1224a8d57407SReid Kleckner     return RecordTag == dwarf::DW_TAG_class_type ? MemberAccess::Private
1225a8d57407SReid Kleckner                                                  : MemberAccess::Public;
1226a8d57407SReid Kleckner   }
1227a8d57407SReid Kleckner   llvm_unreachable("access flags are exclusive");
1228a8d57407SReid Kleckner }
1229a8d57407SReid Kleckner 
123076c9eb99SAmjad Aboud static MethodOptions translateMethodOptionFlags(const DISubprogram *SP) {
123176c9eb99SAmjad Aboud   if (SP->isArtificial())
123276c9eb99SAmjad Aboud     return MethodOptions::CompilerGenerated;
123376c9eb99SAmjad Aboud 
123476c9eb99SAmjad Aboud   // FIXME: Handle other MethodOptions.
123576c9eb99SAmjad Aboud 
123676c9eb99SAmjad Aboud   return MethodOptions::None;
123776c9eb99SAmjad Aboud }
123876c9eb99SAmjad Aboud 
123976c9eb99SAmjad Aboud static MethodKind translateMethodKindFlags(const DISubprogram *SP,
124076c9eb99SAmjad Aboud                                            bool Introduced) {
124176c9eb99SAmjad Aboud   switch (SP->getVirtuality()) {
124276c9eb99SAmjad Aboud   case dwarf::DW_VIRTUALITY_none:
124376c9eb99SAmjad Aboud     break;
124476c9eb99SAmjad Aboud   case dwarf::DW_VIRTUALITY_virtual:
124576c9eb99SAmjad Aboud     return Introduced ? MethodKind::IntroducingVirtual : MethodKind::Virtual;
124676c9eb99SAmjad Aboud   case dwarf::DW_VIRTUALITY_pure_virtual:
124776c9eb99SAmjad Aboud     return Introduced ? MethodKind::PureIntroducingVirtual
124876c9eb99SAmjad Aboud                       : MethodKind::PureVirtual;
124976c9eb99SAmjad Aboud   default:
125076c9eb99SAmjad Aboud     llvm_unreachable("unhandled virtuality case");
125176c9eb99SAmjad Aboud   }
125276c9eb99SAmjad Aboud 
125376c9eb99SAmjad Aboud   // FIXME: Get Clang to mark DISubprogram as static and do something with it.
125476c9eb99SAmjad Aboud 
125576c9eb99SAmjad Aboud   return MethodKind::Vanilla;
125676c9eb99SAmjad Aboud }
125776c9eb99SAmjad Aboud 
1258a8d57407SReid Kleckner static TypeRecordKind getRecordKind(const DICompositeType *Ty) {
1259a8d57407SReid Kleckner   switch (Ty->getTag()) {
1260a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:     return TypeRecordKind::Class;
1261a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type: return TypeRecordKind::Struct;
1262a8d57407SReid Kleckner   }
1263a8d57407SReid Kleckner   llvm_unreachable("unexpected tag");
1264a8d57407SReid Kleckner }
1265a8d57407SReid Kleckner 
1266a8d57407SReid Kleckner /// Return the HasUniqueName option if it should be present in ClassOptions, or
1267a8d57407SReid Kleckner /// None otherwise.
1268a8d57407SReid Kleckner static ClassOptions getRecordUniqueNameOption(const DICompositeType *Ty) {
1269a8d57407SReid Kleckner   // MSVC always sets this flag now, even for local types. Clang doesn't always
1270a8d57407SReid Kleckner   // appear to give every type a linkage name, which may be problematic for us.
1271a8d57407SReid Kleckner   // FIXME: Investigate the consequences of not following them here.
1272a8d57407SReid Kleckner   return !Ty->getIdentifier().empty() ? ClassOptions::HasUniqueName
1273a8d57407SReid Kleckner                                       : ClassOptions::None;
1274a8d57407SReid Kleckner }
1275a8d57407SReid Kleckner 
1276979cb888SDavid Majnemer TypeIndex CodeViewDebug::lowerTypeEnum(const DICompositeType *Ty) {
1277979cb888SDavid Majnemer   ClassOptions CO = ClassOptions::None | getRecordUniqueNameOption(Ty);
1278979cb888SDavid Majnemer   TypeIndex FTI;
1279da9548f9SDavid Majnemer   unsigned EnumeratorCount = 0;
1280979cb888SDavid Majnemer 
1281da9548f9SDavid Majnemer   if (Ty->isForwardDecl()) {
1282979cb888SDavid Majnemer     CO |= ClassOptions::ForwardReference;
1283da9548f9SDavid Majnemer   } else {
1284da9548f9SDavid Majnemer     FieldListRecordBuilder Fields;
1285da9548f9SDavid Majnemer     for (const DINode *Element : Ty->getElements()) {
1286da9548f9SDavid Majnemer       // We assume that the frontend provides all members in source declaration
1287da9548f9SDavid Majnemer       // order, which is what MSVC does.
1288da9548f9SDavid Majnemer       if (auto *Enumerator = dyn_cast_or_null<DIEnumerator>(Element)) {
1289da9548f9SDavid Majnemer         Fields.writeEnumerator(EnumeratorRecord(
1290da9548f9SDavid Majnemer             MemberAccess::Public, APSInt::getUnsigned(Enumerator->getValue()),
1291da9548f9SDavid Majnemer             Enumerator->getName()));
1292da9548f9SDavid Majnemer         EnumeratorCount++;
1293da9548f9SDavid Majnemer       }
1294da9548f9SDavid Majnemer     }
1295da9548f9SDavid Majnemer     FTI = TypeTable.writeFieldList(Fields);
1296da9548f9SDavid Majnemer   }
1297979cb888SDavid Majnemer 
12980c5d874bSReid Kleckner   std::string FullName =
12990c5d874bSReid Kleckner       getFullyQualifiedName(Ty->getScope().resolve(), Ty->getName());
13000c5d874bSReid Kleckner 
13010c5d874bSReid Kleckner   return TypeTable.writeEnum(EnumRecord(EnumeratorCount, CO, FTI, FullName,
1302979cb888SDavid Majnemer                                         Ty->getIdentifier(),
1303979cb888SDavid Majnemer                                         getTypeIndex(Ty->getBaseType())));
1304979cb888SDavid Majnemer }
1305979cb888SDavid Majnemer 
130676c9eb99SAmjad Aboud //===----------------------------------------------------------------------===//
130776c9eb99SAmjad Aboud // ClassInfo
130876c9eb99SAmjad Aboud //===----------------------------------------------------------------------===//
130976c9eb99SAmjad Aboud 
131076c9eb99SAmjad Aboud struct llvm::ClassInfo {
131176c9eb99SAmjad Aboud   struct MemberInfo {
131276c9eb99SAmjad Aboud     const DIDerivedType *MemberTypeNode;
131376c9eb99SAmjad Aboud     unsigned BaseOffset;
131476c9eb99SAmjad Aboud   };
131576c9eb99SAmjad Aboud   // [MemberInfo]
131676c9eb99SAmjad Aboud   typedef std::vector<MemberInfo> MemberList;
131776c9eb99SAmjad Aboud 
1318156a7239SReid Kleckner   typedef TinyPtrVector<const DISubprogram *> MethodsList;
131976c9eb99SAmjad Aboud   // MethodName -> MethodsList
132076c9eb99SAmjad Aboud   typedef MapVector<MDString *, MethodsList> MethodsMap;
132176c9eb99SAmjad Aboud 
132276c9eb99SAmjad Aboud   /// Direct members.
132376c9eb99SAmjad Aboud   MemberList Members;
132476c9eb99SAmjad Aboud   // Direct overloaded methods gathered by name.
132576c9eb99SAmjad Aboud   MethodsMap Methods;
132676c9eb99SAmjad Aboud };
132776c9eb99SAmjad Aboud 
132876c9eb99SAmjad Aboud void CodeViewDebug::clear() {
132976c9eb99SAmjad Aboud   assert(CurFn == nullptr);
133076c9eb99SAmjad Aboud   FileIdMap.clear();
133176c9eb99SAmjad Aboud   FnDebugInfo.clear();
133276c9eb99SAmjad Aboud   FileToFilepathMap.clear();
133376c9eb99SAmjad Aboud   LocalUDTs.clear();
133476c9eb99SAmjad Aboud   GlobalUDTs.clear();
133576c9eb99SAmjad Aboud   TypeIndices.clear();
133676c9eb99SAmjad Aboud   CompleteTypeIndices.clear();
133776c9eb99SAmjad Aboud }
133876c9eb99SAmjad Aboud 
133976c9eb99SAmjad Aboud void CodeViewDebug::collectMemberInfo(ClassInfo &Info,
134076c9eb99SAmjad Aboud                                       const DIDerivedType *DDTy) {
134176c9eb99SAmjad Aboud   if (!DDTy->getName().empty()) {
134276c9eb99SAmjad Aboud     Info.Members.push_back({DDTy, 0});
134376c9eb99SAmjad Aboud     return;
134476c9eb99SAmjad Aboud   }
13451ab7eac8SReid Kleckner   // An unnamed member must represent a nested struct or union. Add all the
13461ab7eac8SReid Kleckner   // indirect fields to the current record.
134776c9eb99SAmjad Aboud   assert((DDTy->getOffsetInBits() % 8) == 0 && "Unnamed bitfield member!");
13481ab7eac8SReid Kleckner   unsigned Offset = DDTy->getOffsetInBits() / 8;
134976c9eb99SAmjad Aboud   const DIType *Ty = DDTy->getBaseType().resolve();
13509ff936cfSReid Kleckner   const DICompositeType *DCTy = cast<DICompositeType>(Ty);
13511ab7eac8SReid Kleckner   ClassInfo NestedInfo = collectClassInfo(DCTy);
13521ab7eac8SReid Kleckner   for (const ClassInfo::MemberInfo &IndirectField : NestedInfo.Members)
135376c9eb99SAmjad Aboud     Info.Members.push_back(
13541ab7eac8SReid Kleckner         {IndirectField.MemberTypeNode, IndirectField.BaseOffset + Offset});
135576c9eb99SAmjad Aboud }
135676c9eb99SAmjad Aboud 
13571ab7eac8SReid Kleckner ClassInfo CodeViewDebug::collectClassInfo(const DICompositeType *Ty) {
13581ab7eac8SReid Kleckner   ClassInfo Info;
135976c9eb99SAmjad Aboud   // Add elements to structure type.
136076c9eb99SAmjad Aboud   DINodeArray Elements = Ty->getElements();
136176c9eb99SAmjad Aboud   for (auto *Element : Elements) {
136276c9eb99SAmjad Aboud     // We assume that the frontend provides all members in source declaration
136376c9eb99SAmjad Aboud     // order, which is what MSVC does.
136476c9eb99SAmjad Aboud     if (!Element)
136576c9eb99SAmjad Aboud       continue;
136676c9eb99SAmjad Aboud     if (auto *SP = dyn_cast<DISubprogram>(Element)) {
1367156a7239SReid Kleckner       Info.Methods[SP->getRawName()].push_back(SP);
136876c9eb99SAmjad Aboud     } else if (auto *DDTy = dyn_cast<DIDerivedType>(Element)) {
136976c9eb99SAmjad Aboud       if (DDTy->getTag() == dwarf::DW_TAG_member)
13701ab7eac8SReid Kleckner         collectMemberInfo(Info, DDTy);
137176c9eb99SAmjad Aboud       else if (DDTy->getTag() == dwarf::DW_TAG_inheritance) {
137276c9eb99SAmjad Aboud         // FIXME: collect class info from inheritance.
137376c9eb99SAmjad Aboud       } else if (DDTy->getTag() == dwarf::DW_TAG_friend) {
137476c9eb99SAmjad Aboud         // Ignore friend members. It appears that MSVC emitted info about
137576c9eb99SAmjad Aboud         // friends in the past, but modern versions do not.
137676c9eb99SAmjad Aboud       }
137776c9eb99SAmjad Aboud       // FIXME: Get Clang to emit function virtual table here and handle it.
137876c9eb99SAmjad Aboud       // FIXME: Get clang to emit nested types here and do something with
137976c9eb99SAmjad Aboud       // them.
138076c9eb99SAmjad Aboud     }
138176c9eb99SAmjad Aboud     // Skip other unrecognized kinds of elements.
138276c9eb99SAmjad Aboud   }
13831ab7eac8SReid Kleckner   return Info;
138476c9eb99SAmjad Aboud }
138576c9eb99SAmjad Aboud 
1386a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerTypeClass(const DICompositeType *Ty) {
1387a8d57407SReid Kleckner   // First, construct the forward decl.  Don't look into Ty to compute the
1388a8d57407SReid Kleckner   // forward decl options, since it might not be available in all TUs.
1389a8d57407SReid Kleckner   TypeRecordKind Kind = getRecordKind(Ty);
1390a8d57407SReid Kleckner   ClassOptions CO =
1391a8d57407SReid Kleckner       ClassOptions::ForwardReference | getRecordUniqueNameOption(Ty);
13920c5d874bSReid Kleckner   std::string FullName =
13930c5d874bSReid Kleckner       getFullyQualifiedName(Ty->getScope().resolve(), Ty->getName());
1394a8d57407SReid Kleckner   TypeIndex FwdDeclTI = TypeTable.writeClass(ClassRecord(
1395a8d57407SReid Kleckner       Kind, 0, CO, HfaKind::None, WindowsRTClassKind::None, TypeIndex(),
13960c5d874bSReid Kleckner       TypeIndex(), TypeIndex(), 0, FullName, Ty->getIdentifier()));
1397643dd836SReid Kleckner   if (!Ty->isForwardDecl())
1398643dd836SReid Kleckner     DeferredCompleteTypes.push_back(Ty);
1399a8d57407SReid Kleckner   return FwdDeclTI;
1400a8d57407SReid Kleckner }
1401a8d57407SReid Kleckner 
1402a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerCompleteTypeClass(const DICompositeType *Ty) {
1403a8d57407SReid Kleckner   // Construct the field list and complete type record.
1404a8d57407SReid Kleckner   TypeRecordKind Kind = getRecordKind(Ty);
1405a8d57407SReid Kleckner   // FIXME: Other ClassOptions, like ContainsNestedClass and NestedClass.
1406a8d57407SReid Kleckner   ClassOptions CO = ClassOptions::None | getRecordUniqueNameOption(Ty);
140776c9eb99SAmjad Aboud   TypeIndex FieldTI;
140876c9eb99SAmjad Aboud   TypeIndex VShapeTI;
1409a8d57407SReid Kleckner   unsigned FieldCount;
141076c9eb99SAmjad Aboud   std::tie(FieldTI, VShapeTI, FieldCount) = lowerRecordFieldList(Ty);
1411a8d57407SReid Kleckner 
14120c5d874bSReid Kleckner   std::string FullName =
14130c5d874bSReid Kleckner       getFullyQualifiedName(Ty->getScope().resolve(), Ty->getName());
14140c5d874bSReid Kleckner 
1415a8d57407SReid Kleckner   uint64_t SizeInBytes = Ty->getSizeInBits() / 8;
14169a519a09SHans Wennborg 
14179a519a09SHans Wennborg   TypeIndex ClassTI = TypeTable.writeClass(ClassRecord(
141876c9eb99SAmjad Aboud       Kind, FieldCount, CO, HfaKind::None, WindowsRTClassKind::None, FieldTI,
14190c5d874bSReid Kleckner       TypeIndex(), VShapeTI, SizeInBytes, FullName, Ty->getIdentifier()));
14209a519a09SHans Wennborg 
14219a519a09SHans Wennborg   TypeTable.writeUdtSourceLine(UdtSourceLineRecord(
14229a519a09SHans Wennborg       ClassTI, TypeTable.writeStringId(StringIdRecord(
14239a519a09SHans Wennborg                    TypeIndex(0x0), getFullFilepath(Ty->getFile()))),
14249a519a09SHans Wennborg       Ty->getLine()));
14259a519a09SHans Wennborg 
1426*4b63a98dSHans Wennborg   addToUDTs(Ty, ClassTI);
1427*4b63a98dSHans Wennborg 
14289a519a09SHans Wennborg   return ClassTI;
1429a8d57407SReid Kleckner }
1430a8d57407SReid Kleckner 
1431a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerTypeUnion(const DICompositeType *Ty) {
1432a8d57407SReid Kleckner   ClassOptions CO =
1433a8d57407SReid Kleckner       ClassOptions::ForwardReference | getRecordUniqueNameOption(Ty);
14340c5d874bSReid Kleckner   std::string FullName =
14350c5d874bSReid Kleckner       getFullyQualifiedName(Ty->getScope().resolve(), Ty->getName());
1436a8d57407SReid Kleckner   TypeIndex FwdDeclTI =
1437a8d57407SReid Kleckner       TypeTable.writeUnion(UnionRecord(0, CO, HfaKind::None, TypeIndex(), 0,
14380c5d874bSReid Kleckner                                        FullName, Ty->getIdentifier()));
1439643dd836SReid Kleckner   if (!Ty->isForwardDecl())
1440643dd836SReid Kleckner     DeferredCompleteTypes.push_back(Ty);
1441a8d57407SReid Kleckner   return FwdDeclTI;
1442a8d57407SReid Kleckner }
1443a8d57407SReid Kleckner 
1444a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerCompleteTypeUnion(const DICompositeType *Ty) {
1445a8d57407SReid Kleckner   ClassOptions CO = ClassOptions::None | getRecordUniqueNameOption(Ty);
144676c9eb99SAmjad Aboud   TypeIndex FieldTI;
1447a8d57407SReid Kleckner   unsigned FieldCount;
144876c9eb99SAmjad Aboud   std::tie(FieldTI, std::ignore, FieldCount) = lowerRecordFieldList(Ty);
1449a8d57407SReid Kleckner   uint64_t SizeInBytes = Ty->getSizeInBits() / 8;
14500c5d874bSReid Kleckner   std::string FullName =
14510c5d874bSReid Kleckner       getFullyQualifiedName(Ty->getScope().resolve(), Ty->getName());
14529a519a09SHans Wennborg 
14539a519a09SHans Wennborg   TypeIndex UnionTI = TypeTable.writeUnion(
14549a519a09SHans Wennborg       UnionRecord(FieldCount, CO, HfaKind::None, FieldTI, SizeInBytes, FullName,
1455a8d57407SReid Kleckner                   Ty->getIdentifier()));
14569a519a09SHans Wennborg 
14579a519a09SHans Wennborg   TypeTable.writeUdtSourceLine(UdtSourceLineRecord(
14589a519a09SHans Wennborg       UnionTI, TypeTable.writeStringId(StringIdRecord(
14599a519a09SHans Wennborg                    TypeIndex(0x0), getFullFilepath(Ty->getFile()))),
14609a519a09SHans Wennborg       Ty->getLine()));
14619a519a09SHans Wennborg 
1462*4b63a98dSHans Wennborg   addToUDTs(Ty, UnionTI);
1463*4b63a98dSHans Wennborg 
14649a519a09SHans Wennborg   return UnionTI;
1465a8d57407SReid Kleckner }
1466a8d57407SReid Kleckner 
146776c9eb99SAmjad Aboud std::tuple<TypeIndex, TypeIndex, unsigned>
1468a8d57407SReid Kleckner CodeViewDebug::lowerRecordFieldList(const DICompositeType *Ty) {
1469a8d57407SReid Kleckner   // Manually count members. MSVC appears to count everything that generates a
1470a8d57407SReid Kleckner   // field list record. Each individual overload in a method overload group
1471a8d57407SReid Kleckner   // contributes to this count, even though the overload group is a single field
1472a8d57407SReid Kleckner   // list record.
1473a8d57407SReid Kleckner   unsigned MemberCount = 0;
14741ab7eac8SReid Kleckner   ClassInfo Info = collectClassInfo(Ty);
1475a8d57407SReid Kleckner   FieldListRecordBuilder Fields;
147676c9eb99SAmjad Aboud 
147776c9eb99SAmjad Aboud   // Create members.
147876c9eb99SAmjad Aboud   for (ClassInfo::MemberInfo &MemberInfo : Info.Members) {
147976c9eb99SAmjad Aboud     const DIDerivedType *Member = MemberInfo.MemberTypeNode;
148076c9eb99SAmjad Aboud     TypeIndex MemberBaseType = getTypeIndex(Member->getBaseType());
148176c9eb99SAmjad Aboud 
1482a8d57407SReid Kleckner     if (Member->isStaticMember()) {
1483a8d57407SReid Kleckner       Fields.writeStaticDataMember(StaticDataMemberRecord(
148476c9eb99SAmjad Aboud           translateAccessFlags(Ty->getTag(), Member->getFlags()),
148576c9eb99SAmjad Aboud           MemberBaseType, Member->getName()));
1486a8d57407SReid Kleckner       MemberCount++;
148776c9eb99SAmjad Aboud       continue;
1488a8d57407SReid Kleckner     }
1489a8d57407SReid Kleckner 
149076c9eb99SAmjad Aboud     uint64_t OffsetInBytes = MemberInfo.BaseOffset;
149176c9eb99SAmjad Aboud 
149276c9eb99SAmjad Aboud     // FIXME: Handle bitfield type memeber.
149376c9eb99SAmjad Aboud     OffsetInBytes += Member->getOffsetInBits() / 8;
149476c9eb99SAmjad Aboud 
149576c9eb99SAmjad Aboud     Fields.writeDataMember(
149676c9eb99SAmjad Aboud         DataMemberRecord(translateAccessFlags(Ty->getTag(), Member->getFlags()),
149776c9eb99SAmjad Aboud                          MemberBaseType, OffsetInBytes, Member->getName()));
149876c9eb99SAmjad Aboud     MemberCount++;
149976c9eb99SAmjad Aboud   }
150076c9eb99SAmjad Aboud 
150176c9eb99SAmjad Aboud   // Create methods
150276c9eb99SAmjad Aboud   for (auto &MethodItr : Info.Methods) {
150376c9eb99SAmjad Aboud     StringRef Name = MethodItr.first->getString();
150476c9eb99SAmjad Aboud 
150576c9eb99SAmjad Aboud     std::vector<OneMethodRecord> Methods;
1506156a7239SReid Kleckner     for (const DISubprogram *SP : MethodItr.second) {
1507156a7239SReid Kleckner       TypeIndex MethodType = getMemberFunctionType(SP, Ty);
1508156a7239SReid Kleckner       bool Introduced = SP->getFlags() & DINode::FlagIntroducedVirtual;
150976c9eb99SAmjad Aboud 
151076c9eb99SAmjad Aboud       unsigned VFTableOffset = -1;
151176c9eb99SAmjad Aboud       if (Introduced)
151276c9eb99SAmjad Aboud         VFTableOffset = SP->getVirtualIndex() * getPointerSizeInBytes();
151376c9eb99SAmjad Aboud 
151476c9eb99SAmjad Aboud       Methods.push_back(
151576c9eb99SAmjad Aboud           OneMethodRecord(MethodType, translateMethodKindFlags(SP, Introduced),
151676c9eb99SAmjad Aboud                           translateMethodOptionFlags(SP),
151776c9eb99SAmjad Aboud                           translateAccessFlags(Ty->getTag(), SP->getFlags()),
151876c9eb99SAmjad Aboud                           VFTableOffset, Name));
151976c9eb99SAmjad Aboud       MemberCount++;
152076c9eb99SAmjad Aboud     }
152176c9eb99SAmjad Aboud     assert(Methods.size() > 0 && "Empty methods map entry");
152276c9eb99SAmjad Aboud     if (Methods.size() == 1)
152376c9eb99SAmjad Aboud       Fields.writeOneMethod(Methods[0]);
152476c9eb99SAmjad Aboud     else {
152576c9eb99SAmjad Aboud       TypeIndex MethodList =
152676c9eb99SAmjad Aboud           TypeTable.writeMethodOverloadList(MethodOverloadListRecord(Methods));
152776c9eb99SAmjad Aboud       Fields.writeOverloadedMethod(
152876c9eb99SAmjad Aboud           OverloadedMethodRecord(Methods.size(), MethodList, Name));
152976c9eb99SAmjad Aboud     }
153076c9eb99SAmjad Aboud   }
153176c9eb99SAmjad Aboud   TypeIndex FieldTI = TypeTable.writeFieldList(Fields);
153276c9eb99SAmjad Aboud   return std::make_tuple(FieldTI, TypeIndex(), MemberCount);
153376c9eb99SAmjad Aboud }
153476c9eb99SAmjad Aboud 
1535643dd836SReid Kleckner struct CodeViewDebug::TypeLoweringScope {
1536643dd836SReid Kleckner   TypeLoweringScope(CodeViewDebug &CVD) : CVD(CVD) { ++CVD.TypeEmissionLevel; }
1537643dd836SReid Kleckner   ~TypeLoweringScope() {
1538643dd836SReid Kleckner     // Don't decrement TypeEmissionLevel until after emitting deferred types, so
1539643dd836SReid Kleckner     // inner TypeLoweringScopes don't attempt to emit deferred types.
1540643dd836SReid Kleckner     if (CVD.TypeEmissionLevel == 1)
1541643dd836SReid Kleckner       CVD.emitDeferredCompleteTypes();
1542643dd836SReid Kleckner     --CVD.TypeEmissionLevel;
1543643dd836SReid Kleckner   }
1544643dd836SReid Kleckner   CodeViewDebug &CVD;
1545643dd836SReid Kleckner };
1546643dd836SReid Kleckner 
154776c9eb99SAmjad Aboud TypeIndex CodeViewDebug::getTypeIndex(DITypeRef TypeRef, DITypeRef ClassTyRef) {
15485acacbb0SReid Kleckner   const DIType *Ty = TypeRef.resolve();
154976c9eb99SAmjad Aboud   const DIType *ClassTy = ClassTyRef.resolve();
15505acacbb0SReid Kleckner 
15515acacbb0SReid Kleckner   // The null DIType is the void type. Don't try to hash it.
15525acacbb0SReid Kleckner   if (!Ty)
15535acacbb0SReid Kleckner     return TypeIndex::Void();
15545acacbb0SReid Kleckner 
1555a8d57407SReid Kleckner   // Check if we've already translated this type. Don't try to do a
1556a8d57407SReid Kleckner   // get-or-create style insertion that caches the hash lookup across the
1557a8d57407SReid Kleckner   // lowerType call. It will update the TypeIndices map.
155876c9eb99SAmjad Aboud   auto I = TypeIndices.find({Ty, ClassTy});
15595acacbb0SReid Kleckner   if (I != TypeIndices.end())
15605acacbb0SReid Kleckner     return I->second;
15615acacbb0SReid Kleckner 
1562643dd836SReid Kleckner   TypeIndex TI;
1563643dd836SReid Kleckner   {
1564643dd836SReid Kleckner     TypeLoweringScope S(*this);
1565643dd836SReid Kleckner     TI = lowerType(Ty, ClassTy);
1566643dd836SReid Kleckner     recordTypeIndexForDINode(Ty, TI, ClassTy);
1567643dd836SReid Kleckner   }
15685acacbb0SReid Kleckner 
1569643dd836SReid Kleckner   return TI;
1570a8d57407SReid Kleckner }
1571a8d57407SReid Kleckner 
1572a8d57407SReid Kleckner TypeIndex CodeViewDebug::getCompleteTypeIndex(DITypeRef TypeRef) {
1573a8d57407SReid Kleckner   const DIType *Ty = TypeRef.resolve();
1574a8d57407SReid Kleckner 
1575a8d57407SReid Kleckner   // The null DIType is the void type. Don't try to hash it.
1576a8d57407SReid Kleckner   if (!Ty)
1577a8d57407SReid Kleckner     return TypeIndex::Void();
1578a8d57407SReid Kleckner 
1579a8d57407SReid Kleckner   // If this is a non-record type, the complete type index is the same as the
1580a8d57407SReid Kleckner   // normal type index. Just call getTypeIndex.
1581a8d57407SReid Kleckner   switch (Ty->getTag()) {
1582a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
1583a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
1584a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
1585a8d57407SReid Kleckner     break;
1586a8d57407SReid Kleckner   default:
1587a8d57407SReid Kleckner     return getTypeIndex(Ty);
1588a8d57407SReid Kleckner   }
1589a8d57407SReid Kleckner 
1590a8d57407SReid Kleckner   // Check if we've already translated the complete record type.  Lowering a
1591a8d57407SReid Kleckner   // complete type should never trigger lowering another complete type, so we
1592a8d57407SReid Kleckner   // can reuse the hash table lookup result.
1593a8d57407SReid Kleckner   const auto *CTy = cast<DICompositeType>(Ty);
1594a8d57407SReid Kleckner   auto InsertResult = CompleteTypeIndices.insert({CTy, TypeIndex()});
1595a8d57407SReid Kleckner   if (!InsertResult.second)
1596a8d57407SReid Kleckner     return InsertResult.first->second;
1597a8d57407SReid Kleckner 
1598643dd836SReid Kleckner   TypeLoweringScope S(*this);
1599643dd836SReid Kleckner 
1600a8d57407SReid Kleckner   // Make sure the forward declaration is emitted first. It's unclear if this
1601a8d57407SReid Kleckner   // is necessary, but MSVC does it, and we should follow suit until we can show
1602a8d57407SReid Kleckner   // otherwise.
1603a8d57407SReid Kleckner   TypeIndex FwdDeclTI = getTypeIndex(CTy);
1604a8d57407SReid Kleckner 
1605a8d57407SReid Kleckner   // Just use the forward decl if we don't have complete type info. This might
1606a8d57407SReid Kleckner   // happen if the frontend is using modules and expects the complete definition
1607a8d57407SReid Kleckner   // to be emitted elsewhere.
1608a8d57407SReid Kleckner   if (CTy->isForwardDecl())
1609a8d57407SReid Kleckner     return FwdDeclTI;
1610a8d57407SReid Kleckner 
1611a8d57407SReid Kleckner   TypeIndex TI;
1612a8d57407SReid Kleckner   switch (CTy->getTag()) {
1613a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
1614a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
1615a8d57407SReid Kleckner     TI = lowerCompleteTypeClass(CTy);
1616a8d57407SReid Kleckner     break;
1617a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
1618a8d57407SReid Kleckner     TI = lowerCompleteTypeUnion(CTy);
1619a8d57407SReid Kleckner     break;
1620a8d57407SReid Kleckner   default:
1621a8d57407SReid Kleckner     llvm_unreachable("not a record");
1622a8d57407SReid Kleckner   }
1623a8d57407SReid Kleckner 
1624a8d57407SReid Kleckner   InsertResult.first->second = TI;
16255acacbb0SReid Kleckner   return TI;
16265acacbb0SReid Kleckner }
16275acacbb0SReid Kleckner 
1628643dd836SReid Kleckner /// Emit all the deferred complete record types. Try to do this in FIFO order,
1629643dd836SReid Kleckner /// and do this until fixpoint, as each complete record type typically references
1630643dd836SReid Kleckner /// many other record types.
1631643dd836SReid Kleckner void CodeViewDebug::emitDeferredCompleteTypes() {
1632643dd836SReid Kleckner   SmallVector<const DICompositeType *, 4> TypesToEmit;
1633643dd836SReid Kleckner   while (!DeferredCompleteTypes.empty()) {
1634643dd836SReid Kleckner     std::swap(DeferredCompleteTypes, TypesToEmit);
1635643dd836SReid Kleckner     for (const DICompositeType *RecordTy : TypesToEmit)
1636643dd836SReid Kleckner       getCompleteTypeIndex(RecordTy);
1637643dd836SReid Kleckner     TypesToEmit.clear();
1638643dd836SReid Kleckner   }
1639643dd836SReid Kleckner }
1640643dd836SReid Kleckner 
1641f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
1642f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
1643f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
1644f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
1645f9c275feSReid Kleckner   OS.AddComment("Record length");
1646f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
1647f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
1648f9c275feSReid Kleckner 
1649f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
165063a2846eSZachary Turner   OS.EmitIntValue(unsigned(SymbolKind::S_LOCAL), 2);
1651f9c275feSReid Kleckner 
165263a2846eSZachary Turner   LocalSymFlags Flags = LocalSymFlags::None;
1653f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
165463a2846eSZachary Turner     Flags |= LocalSymFlags::IsParameter;
1655876330d5SReid Kleckner   if (Var.DefRanges.empty())
165663a2846eSZachary Turner     Flags |= LocalSymFlags::IsOptimizedOut;
1657f9c275feSReid Kleckner 
1658f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
1659a8d57407SReid Kleckner   TypeIndex TI = getCompleteTypeIndex(Var.DIVar->getType());
16605acacbb0SReid Kleckner   OS.EmitIntValue(TI.getIndex(), 4);
1661f9c275feSReid Kleckner   OS.AddComment("Flags");
166263a2846eSZachary Turner   OS.EmitIntValue(static_cast<uint16_t>(Flags), 2);
16631256125fSDavid Majnemer   // Truncate the name so we won't overflow the record length field.
1664b9456a5eSDavid Majnemer   emitNullTerminatedSymbolName(OS, Var.DIVar->getName());
1665f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
1666f9c275feSReid Kleckner 
1667876330d5SReid Kleckner   // Calculate the on disk prefix of the appropriate def range record. The
1668876330d5SReid Kleckner   // records and on disk formats are described in SymbolRecords.h. BytePrefix
1669876330d5SReid Kleckner   // should be big enough to hold all forms without memory allocation.
1670876330d5SReid Kleckner   SmallString<20> BytePrefix;
1671876330d5SReid Kleckner   for (const LocalVarDefRange &DefRange : Var.DefRanges) {
1672876330d5SReid Kleckner     BytePrefix.clear();
1673876330d5SReid Kleckner     // FIXME: Handle bitpieces.
1674876330d5SReid Kleckner     if (DefRange.StructOffset != 0)
1675876330d5SReid Kleckner       continue;
1676876330d5SReid Kleckner 
1677876330d5SReid Kleckner     if (DefRange.InMemory) {
1678a78ecd1eSZachary Turner       DefRangeRegisterRelSym Sym(DefRange.CVRegister, 0, DefRange.DataOffset, 0,
1679a78ecd1eSZachary Turner                                  0, 0, ArrayRef<LocalVariableAddrGap>());
1680f9c275feSReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
1681876330d5SReid Kleckner       BytePrefix +=
1682876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
1683a78ecd1eSZachary Turner       BytePrefix +=
1684a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
1685a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
1686876330d5SReid Kleckner     } else {
1687876330d5SReid Kleckner       assert(DefRange.DataOffset == 0 && "unexpected offset into register");
1688a78ecd1eSZachary Turner       // Unclear what matters here.
1689a78ecd1eSZachary Turner       DefRangeRegisterSym Sym(DefRange.CVRegister, 0, 0, 0, 0,
1690a78ecd1eSZachary Turner                               ArrayRef<LocalVariableAddrGap>());
1691876330d5SReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER);
1692876330d5SReid Kleckner       BytePrefix +=
1693876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
1694a78ecd1eSZachary Turner       BytePrefix +=
1695a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
1696a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
1697876330d5SReid Kleckner     }
1698876330d5SReid Kleckner     OS.EmitCVDefRangeDirective(DefRange.Ranges, BytePrefix);
1699876330d5SReid Kleckner   }
1700f9c275feSReid Kleckner }
1701f9c275feSReid Kleckner 
170270f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
170370f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
170470f5bc99SReid Kleckner     return;
170570f5bc99SReid Kleckner 
170670f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
170770f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
170870f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
170970f5bc99SReid Kleckner 
1710adebb937SPete Cooper   collectVariableInfo(GV->getSubprogram());
1711876330d5SReid Kleckner 
1712876330d5SReid Kleckner   DebugHandlerBase::endFunction(MF);
1713876330d5SReid Kleckner 
17142214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
17152214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
171670f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
1717f9c275feSReid Kleckner     CurFn = nullptr;
1718f9c275feSReid Kleckner     return;
171970f5bc99SReid Kleckner   }
1720f9c275feSReid Kleckner 
1721f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
1722f9c275feSReid Kleckner 
172370f5bc99SReid Kleckner   CurFn = nullptr;
172470f5bc99SReid Kleckner }
172570f5bc99SReid Kleckner 
172670f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
1727f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
1728f9c275feSReid Kleckner 
172970f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
173070f5bc99SReid Kleckner   if (!Asm || MI->isDebugValue() || MI->getFlag(MachineInstr::FrameSetup))
173170f5bc99SReid Kleckner     return;
173270f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
173370f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
173470f5bc99SReid Kleckner     return;
173570f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
173670f5bc99SReid Kleckner }
17376f3406dfSReid Kleckner 
17386f3406dfSReid Kleckner MCSymbol *CodeViewDebug::beginCVSubsection(ModuleSubstreamKind Kind) {
17396f3406dfSReid Kleckner   MCSymbol *BeginLabel = MMI->getContext().createTempSymbol(),
17406f3406dfSReid Kleckner            *EndLabel = MMI->getContext().createTempSymbol();
17416f3406dfSReid Kleckner   OS.EmitIntValue(unsigned(Kind), 4);
17426f3406dfSReid Kleckner   OS.AddComment("Subsection size");
17436f3406dfSReid Kleckner   OS.emitAbsoluteSymbolDiff(EndLabel, BeginLabel, 4);
17446f3406dfSReid Kleckner   OS.EmitLabel(BeginLabel);
17456f3406dfSReid Kleckner   return EndLabel;
17466f3406dfSReid Kleckner }
17476f3406dfSReid Kleckner 
17486f3406dfSReid Kleckner void CodeViewDebug::endCVSubsection(MCSymbol *EndLabel) {
17496f3406dfSReid Kleckner   OS.EmitLabel(EndLabel);
17506f3406dfSReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
17516f3406dfSReid Kleckner   OS.EmitValueToAlignment(4);
17526f3406dfSReid Kleckner }
17536f3406dfSReid Kleckner 
17543128b10cSDavid Majnemer void CodeViewDebug::emitDebugInfoForUDTs(
17553128b10cSDavid Majnemer     ArrayRef<std::pair<std::string, TypeIndex>> UDTs) {
17563128b10cSDavid Majnemer   for (const std::pair<std::string, codeview::TypeIndex> &UDT : UDTs) {
17573128b10cSDavid Majnemer     MCSymbol *UDTRecordBegin = MMI->getContext().createTempSymbol(),
17583128b10cSDavid Majnemer              *UDTRecordEnd = MMI->getContext().createTempSymbol();
17593128b10cSDavid Majnemer     OS.AddComment("Record length");
17603128b10cSDavid Majnemer     OS.emitAbsoluteSymbolDiff(UDTRecordEnd, UDTRecordBegin, 2);
17613128b10cSDavid Majnemer     OS.EmitLabel(UDTRecordBegin);
17623128b10cSDavid Majnemer 
17633128b10cSDavid Majnemer     OS.AddComment("Record kind: S_UDT");
17643128b10cSDavid Majnemer     OS.EmitIntValue(unsigned(SymbolKind::S_UDT), 2);
17653128b10cSDavid Majnemer 
17663128b10cSDavid Majnemer     OS.AddComment("Type");
17673128b10cSDavid Majnemer     OS.EmitIntValue(UDT.second.getIndex(), 4);
17683128b10cSDavid Majnemer 
17693128b10cSDavid Majnemer     emitNullTerminatedSymbolName(OS, UDT.first);
17703128b10cSDavid Majnemer     OS.EmitLabel(UDTRecordEnd);
17713128b10cSDavid Majnemer   }
17723128b10cSDavid Majnemer }
17733128b10cSDavid Majnemer 
17746f3406dfSReid Kleckner void CodeViewDebug::emitDebugInfoForGlobals() {
17756f3406dfSReid Kleckner   NamedMDNode *CUs = MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
17766f3406dfSReid Kleckner   for (const MDNode *Node : CUs->operands()) {
17776f3406dfSReid Kleckner     const auto *CU = cast<DICompileUnit>(Node);
17786f3406dfSReid Kleckner 
17796f3406dfSReid Kleckner     // First, emit all globals that are not in a comdat in a single symbol
17806f3406dfSReid Kleckner     // substream. MSVC doesn't like it if the substream is empty, so only open
17816f3406dfSReid Kleckner     // it if we have at least one global to emit.
17826f3406dfSReid Kleckner     switchToDebugSectionForSymbol(nullptr);
17836f3406dfSReid Kleckner     MCSymbol *EndLabel = nullptr;
17846f3406dfSReid Kleckner     for (const DIGlobalVariable *G : CU->getGlobalVariables()) {
17856d1d2754SReid Kleckner       if (const auto *GV = dyn_cast_or_null<GlobalVariable>(G->getVariable())) {
1786577be0feSDavid Majnemer         if (!GV->hasComdat() && !GV->isDeclarationForLinker()) {
17876f3406dfSReid Kleckner           if (!EndLabel) {
17886f3406dfSReid Kleckner             OS.AddComment("Symbol subsection for globals");
17896f3406dfSReid Kleckner             EndLabel = beginCVSubsection(ModuleSubstreamKind::Symbols);
17906f3406dfSReid Kleckner           }
17916f3406dfSReid Kleckner           emitDebugInfoForGlobal(G, Asm->getSymbol(GV));
17926f3406dfSReid Kleckner         }
17936f3406dfSReid Kleckner       }
17946d1d2754SReid Kleckner     }
17956f3406dfSReid Kleckner     if (EndLabel)
17966f3406dfSReid Kleckner       endCVSubsection(EndLabel);
17976f3406dfSReid Kleckner 
17986f3406dfSReid Kleckner     // Second, emit each global that is in a comdat into its own .debug$S
17996f3406dfSReid Kleckner     // section along with its own symbol substream.
18006f3406dfSReid Kleckner     for (const DIGlobalVariable *G : CU->getGlobalVariables()) {
18016d1d2754SReid Kleckner       if (const auto *GV = dyn_cast_or_null<GlobalVariable>(G->getVariable())) {
18026f3406dfSReid Kleckner         if (GV->hasComdat()) {
18036f3406dfSReid Kleckner           MCSymbol *GVSym = Asm->getSymbol(GV);
18046f3406dfSReid Kleckner           OS.AddComment("Symbol subsection for " +
18056f3406dfSReid Kleckner                         Twine(GlobalValue::getRealLinkageName(GV->getName())));
18066f3406dfSReid Kleckner           switchToDebugSectionForSymbol(GVSym);
18076f3406dfSReid Kleckner           EndLabel = beginCVSubsection(ModuleSubstreamKind::Symbols);
18086f3406dfSReid Kleckner           emitDebugInfoForGlobal(G, GVSym);
18096f3406dfSReid Kleckner           endCVSubsection(EndLabel);
18106f3406dfSReid Kleckner         }
18116f3406dfSReid Kleckner       }
18126f3406dfSReid Kleckner     }
18136f3406dfSReid Kleckner   }
18146f3406dfSReid Kleckner }
18156f3406dfSReid Kleckner 
1816b510b458SHans Wennborg void CodeViewDebug::emitDebugInfoForRetainedTypes() {
1817b510b458SHans Wennborg   NamedMDNode *CUs = MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
1818b510b458SHans Wennborg   for (const MDNode *Node : CUs->operands()) {
1819b510b458SHans Wennborg     for (auto *Ty : cast<DICompileUnit>(Node)->getRetainedTypes()) {
1820b510b458SHans Wennborg       if (DIType *RT = dyn_cast<DIType>(Ty)) {
1821b510b458SHans Wennborg         getTypeIndex(RT);
1822b510b458SHans Wennborg         // FIXME: Add to global/local DTU list.
1823b510b458SHans Wennborg       }
1824b510b458SHans Wennborg     }
1825b510b458SHans Wennborg   }
1826b510b458SHans Wennborg }
1827b510b458SHans Wennborg 
18286f3406dfSReid Kleckner void CodeViewDebug::emitDebugInfoForGlobal(const DIGlobalVariable *DIGV,
18296f3406dfSReid Kleckner                                            MCSymbol *GVSym) {
18306f3406dfSReid Kleckner   // DataSym record, see SymbolRecord.h for more info.
18316f3406dfSReid Kleckner   // FIXME: Thread local data, etc
18326f3406dfSReid Kleckner   MCSymbol *DataBegin = MMI->getContext().createTempSymbol(),
18336f3406dfSReid Kleckner            *DataEnd = MMI->getContext().createTempSymbol();
18346f3406dfSReid Kleckner   OS.AddComment("Record length");
18356f3406dfSReid Kleckner   OS.emitAbsoluteSymbolDiff(DataEnd, DataBegin, 2);
18366f3406dfSReid Kleckner   OS.EmitLabel(DataBegin);
18376f3406dfSReid Kleckner   OS.AddComment("Record kind: S_GDATA32");
18386f3406dfSReid Kleckner   OS.EmitIntValue(unsigned(SymbolKind::S_GDATA32), 2);
18396f3406dfSReid Kleckner   OS.AddComment("Type");
18406f3406dfSReid Kleckner   OS.EmitIntValue(getCompleteTypeIndex(DIGV->getType()).getIndex(), 4);
18416f3406dfSReid Kleckner   OS.AddComment("DataOffset");
18426f3406dfSReid Kleckner   OS.EmitCOFFSecRel32(GVSym);
18436f3406dfSReid Kleckner   OS.AddComment("Segment");
18446f3406dfSReid Kleckner   OS.EmitCOFFSectionIndex(GVSym);
18456f3406dfSReid Kleckner   OS.AddComment("Name");
18466f3406dfSReid Kleckner   emitNullTerminatedSymbolName(OS, DIGV->getName());
18476f3406dfSReid Kleckner   OS.EmitLabel(DataEnd);
18486f3406dfSReid Kleckner }
1849