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"
16c92e9469SReid Kleckner #include "llvm/DebugInfo/CodeView/CVTypeVisitor.h"
1770f5bc99SReid Kleckner #include "llvm/DebugInfo/CodeView/CodeView.h"
18a8d57407SReid Kleckner #include "llvm/DebugInfo/CodeView/FieldListRecordBuilder.h"
192214ed89SReid Kleckner #include "llvm/DebugInfo/CodeView/Line.h"
2070f5bc99SReid Kleckner #include "llvm/DebugInfo/CodeView/SymbolRecord.h"
21fbdbe9e2SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeDumper.h"
22f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeIndex.h"
23f3b9ba49SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeRecord.h"
24c92e9469SReid Kleckner #include "llvm/DebugInfo/CodeView/TypeVisitorCallbacks.h"
25a3225b04SZachary Turner #include "llvm/DebugInfo/MSF/ByteStream.h"
26a3225b04SZachary Turner #include "llvm/DebugInfo/MSF/StreamReader.h"
279319cbc0SDavid Majnemer #include "llvm/IR/Constants.h"
2846cb48c7SReid Kleckner #include "llvm/MC/MCAsmInfo.h"
2970f5bc99SReid Kleckner #include "llvm/MC/MCExpr.h"
305d122f87SReid Kleckner #include "llvm/MC/MCSectionCOFF.h"
3170f5bc99SReid Kleckner #include "llvm/MC/MCSymbol.h"
3270f5bc99SReid Kleckner #include "llvm/Support/COFF.h"
33fbdbe9e2SReid Kleckner #include "llvm/Support/ScopedPrinter.h"
34f9c275feSReid Kleckner #include "llvm/Target/TargetFrameLowering.h"
3576c9eb99SAmjad Aboud #include "llvm/Target/TargetRegisterInfo.h"
3676c9eb99SAmjad Aboud #include "llvm/Target/TargetSubtargetInfo.h"
3770f5bc99SReid Kleckner 
38f9c275feSReid Kleckner using namespace llvm;
3970f5bc99SReid Kleckner using namespace llvm::codeview;
40bac69d33SZachary Turner using namespace llvm::msf;
4170f5bc99SReid Kleckner 
42f9c275feSReid Kleckner CodeViewDebug::CodeViewDebug(AsmPrinter *AP)
43c6d54da8SZachary Turner     : DebugHandlerBase(AP), OS(*Asm->OutStreamer), Allocator(),
44c6d54da8SZachary Turner       TypeTable(Allocator), CurFn(nullptr) {
45f9c275feSReid Kleckner   // If module doesn't have named metadata anchors or COFF debug section
46f9c275feSReid Kleckner   // is not available, skip any debug info related stuff.
47f9c275feSReid Kleckner   if (!MMI->getModule()->getNamedMetadata("llvm.dbg.cu") ||
48f9c275feSReid Kleckner       !AP->getObjFileLowering().getCOFFDebugSymbolsSection()) {
49f9c275feSReid Kleckner     Asm = nullptr;
50f9c275feSReid Kleckner     return;
51f9c275feSReid Kleckner   }
52f9c275feSReid Kleckner 
53f9c275feSReid Kleckner   // Tell MMI that we have debug info.
54f9c275feSReid Kleckner   MMI->setDebugInfoAvailability(true);
55f9c275feSReid Kleckner }
5670f5bc99SReid Kleckner 
579533af4fSReid Kleckner StringRef CodeViewDebug::getFullFilepath(const DIFile *File) {
589533af4fSReid Kleckner   std::string &Filepath = FileToFilepathMap[File];
5970f5bc99SReid Kleckner   if (!Filepath.empty())
6070f5bc99SReid Kleckner     return Filepath;
6170f5bc99SReid Kleckner 
629533af4fSReid Kleckner   StringRef Dir = File->getDirectory(), Filename = File->getFilename();
639533af4fSReid Kleckner 
6470f5bc99SReid Kleckner   // Clang emits directory and relative filename info into the IR, but CodeView
6570f5bc99SReid Kleckner   // operates on full paths.  We could change Clang to emit full paths too, but
6670f5bc99SReid Kleckner   // that would increase the IR size and probably not needed for other users.
6770f5bc99SReid Kleckner   // For now, just concatenate and canonicalize the path here.
6870f5bc99SReid Kleckner   if (Filename.find(':') == 1)
6970f5bc99SReid Kleckner     Filepath = Filename;
7070f5bc99SReid Kleckner   else
7170f5bc99SReid Kleckner     Filepath = (Dir + "\\" + Filename).str();
7270f5bc99SReid Kleckner 
7370f5bc99SReid Kleckner   // Canonicalize the path.  We have to do it textually because we may no longer
7470f5bc99SReid Kleckner   // have access the file in the filesystem.
7570f5bc99SReid Kleckner   // First, replace all slashes with backslashes.
7670f5bc99SReid Kleckner   std::replace(Filepath.begin(), Filepath.end(), '/', '\\');
7770f5bc99SReid Kleckner 
7870f5bc99SReid Kleckner   // Remove all "\.\" with "\".
7970f5bc99SReid Kleckner   size_t Cursor = 0;
8070f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\.\\", Cursor)) != std::string::npos)
8170f5bc99SReid Kleckner     Filepath.erase(Cursor, 2);
8270f5bc99SReid Kleckner 
8370f5bc99SReid Kleckner   // Replace all "\XXX\..\" with "\".  Don't try too hard though as the original
8470f5bc99SReid Kleckner   // path should be well-formatted, e.g. start with a drive letter, etc.
8570f5bc99SReid Kleckner   Cursor = 0;
8670f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\..\\", Cursor)) != std::string::npos) {
8770f5bc99SReid Kleckner     // Something's wrong if the path starts with "\..\", abort.
8870f5bc99SReid Kleckner     if (Cursor == 0)
8970f5bc99SReid Kleckner       break;
9070f5bc99SReid Kleckner 
9170f5bc99SReid Kleckner     size_t PrevSlash = Filepath.rfind('\\', Cursor - 1);
9270f5bc99SReid Kleckner     if (PrevSlash == std::string::npos)
9370f5bc99SReid Kleckner       // Something's wrong, abort.
9470f5bc99SReid Kleckner       break;
9570f5bc99SReid Kleckner 
9670f5bc99SReid Kleckner     Filepath.erase(PrevSlash, Cursor + 3 - PrevSlash);
9770f5bc99SReid Kleckner     // The next ".." might be following the one we've just erased.
9870f5bc99SReid Kleckner     Cursor = PrevSlash;
9970f5bc99SReid Kleckner   }
10070f5bc99SReid Kleckner 
10170f5bc99SReid Kleckner   // Remove all duplicate backslashes.
10270f5bc99SReid Kleckner   Cursor = 0;
10370f5bc99SReid Kleckner   while ((Cursor = Filepath.find("\\\\", Cursor)) != std::string::npos)
10470f5bc99SReid Kleckner     Filepath.erase(Cursor, 1);
10570f5bc99SReid Kleckner 
10670f5bc99SReid Kleckner   return Filepath;
10770f5bc99SReid Kleckner }
10870f5bc99SReid Kleckner 
1092214ed89SReid Kleckner unsigned CodeViewDebug::maybeRecordFile(const DIFile *F) {
1102214ed89SReid Kleckner   unsigned NextId = FileIdMap.size() + 1;
1112214ed89SReid Kleckner   auto Insertion = FileIdMap.insert(std::make_pair(F, NextId));
1122214ed89SReid Kleckner   if (Insertion.second) {
1132214ed89SReid Kleckner     // We have to compute the full filepath and emit a .cv_file directive.
1142214ed89SReid Kleckner     StringRef FullPath = getFullFilepath(F);
115a5b1eef8SReid Kleckner     bool Success = OS.EmitCVFileDirective(NextId, FullPath);
116a5b1eef8SReid Kleckner     (void)Success;
117a5b1eef8SReid Kleckner     assert(Success && ".cv_file directive failed");
1182214ed89SReid Kleckner   }
1192214ed89SReid Kleckner   return Insertion.first->second;
1202214ed89SReid Kleckner }
1212214ed89SReid Kleckner 
122876330d5SReid Kleckner CodeViewDebug::InlineSite &
123876330d5SReid Kleckner CodeViewDebug::getInlineSite(const DILocation *InlinedAt,
124876330d5SReid Kleckner                              const DISubprogram *Inlinee) {
125fbd7787dSReid Kleckner   auto SiteInsertion = CurFn->InlineSites.insert({InlinedAt, InlineSite()});
126fbd7787dSReid Kleckner   InlineSite *Site = &SiteInsertion.first->second;
127fbd7787dSReid Kleckner   if (SiteInsertion.second) {
128a9f4cc95SReid Kleckner     unsigned ParentFuncId = CurFn->FuncId;
129a9f4cc95SReid Kleckner     if (const DILocation *OuterIA = InlinedAt->getInlinedAt())
130a9f4cc95SReid Kleckner       ParentFuncId =
131a9f4cc95SReid Kleckner           getInlineSite(OuterIA, InlinedAt->getScope()->getSubprogram())
132a9f4cc95SReid Kleckner               .SiteFuncId;
133a9f4cc95SReid Kleckner 
134f9c275feSReid Kleckner     Site->SiteFuncId = NextFuncId++;
135a9f4cc95SReid Kleckner     OS.EmitCVInlineSiteIdDirective(
136a9f4cc95SReid Kleckner         Site->SiteFuncId, ParentFuncId, maybeRecordFile(InlinedAt->getFile()),
137a9f4cc95SReid Kleckner         InlinedAt->getLine(), InlinedAt->getColumn(), SMLoc());
138876330d5SReid Kleckner     Site->Inlinee = Inlinee;
1392280f932SReid Kleckner     InlinedSubprograms.insert(Inlinee);
14075c3ebfaSDavid Majnemer     getFuncIdForSubprogram(Inlinee);
141f3b9ba49SReid Kleckner   }
142f9c275feSReid Kleckner   return *Site;
143f3b9ba49SReid Kleckner }
144f3b9ba49SReid Kleckner 
1456bdc24e7SDavid Majnemer static StringRef getPrettyScopeName(const DIScope *Scope) {
1466bdc24e7SDavid Majnemer   StringRef ScopeName = Scope->getName();
1476bdc24e7SDavid Majnemer   if (!ScopeName.empty())
1486bdc24e7SDavid Majnemer     return ScopeName;
1496bdc24e7SDavid Majnemer 
1506bdc24e7SDavid Majnemer   switch (Scope->getTag()) {
1516bdc24e7SDavid Majnemer   case dwarf::DW_TAG_enumeration_type:
1526bdc24e7SDavid Majnemer   case dwarf::DW_TAG_class_type:
1536bdc24e7SDavid Majnemer   case dwarf::DW_TAG_structure_type:
1546bdc24e7SDavid Majnemer   case dwarf::DW_TAG_union_type:
1556bdc24e7SDavid Majnemer     return "<unnamed-tag>";
1566bdc24e7SDavid Majnemer   case dwarf::DW_TAG_namespace:
1576bdc24e7SDavid Majnemer     return "`anonymous namespace'";
1586bdc24e7SDavid Majnemer   }
1596bdc24e7SDavid Majnemer 
1606bdc24e7SDavid Majnemer   return StringRef();
1616bdc24e7SDavid Majnemer }
1626bdc24e7SDavid Majnemer 
1630c5d874bSReid Kleckner static const DISubprogram *getQualifiedNameComponents(
1640c5d874bSReid Kleckner     const DIScope *Scope, SmallVectorImpl<StringRef> &QualifiedNameComponents) {
1650c5d874bSReid Kleckner   const DISubprogram *ClosestSubprogram = nullptr;
1660c5d874bSReid Kleckner   while (Scope != nullptr) {
1670c5d874bSReid Kleckner     if (ClosestSubprogram == nullptr)
1680c5d874bSReid Kleckner       ClosestSubprogram = dyn_cast<DISubprogram>(Scope);
1696bdc24e7SDavid Majnemer     StringRef ScopeName = getPrettyScopeName(Scope);
1700c5d874bSReid Kleckner     if (!ScopeName.empty())
1710c5d874bSReid Kleckner       QualifiedNameComponents.push_back(ScopeName);
1720c5d874bSReid Kleckner     Scope = Scope->getScope().resolve();
1730c5d874bSReid Kleckner   }
1740c5d874bSReid Kleckner   return ClosestSubprogram;
1750c5d874bSReid Kleckner }
1760c5d874bSReid Kleckner 
1770c5d874bSReid Kleckner static std::string getQualifiedName(ArrayRef<StringRef> QualifiedNameComponents,
1780c5d874bSReid Kleckner                                     StringRef TypeName) {
1790c5d874bSReid Kleckner   std::string FullyQualifiedName;
1800c5d874bSReid Kleckner   for (StringRef QualifiedNameComponent : reverse(QualifiedNameComponents)) {
1810c5d874bSReid Kleckner     FullyQualifiedName.append(QualifiedNameComponent);
1820c5d874bSReid Kleckner     FullyQualifiedName.append("::");
1830c5d874bSReid Kleckner   }
1840c5d874bSReid Kleckner   FullyQualifiedName.append(TypeName);
1850c5d874bSReid Kleckner   return FullyQualifiedName;
1860c5d874bSReid Kleckner }
1870c5d874bSReid Kleckner 
1880c5d874bSReid Kleckner static std::string getFullyQualifiedName(const DIScope *Scope, StringRef Name) {
1890c5d874bSReid Kleckner   SmallVector<StringRef, 5> QualifiedNameComponents;
1900c5d874bSReid Kleckner   getQualifiedNameComponents(Scope, QualifiedNameComponents);
1910c5d874bSReid Kleckner   return getQualifiedName(QualifiedNameComponents, Name);
1920c5d874bSReid Kleckner }
1930c5d874bSReid Kleckner 
194b5af11dfSReid Kleckner struct CodeViewDebug::TypeLoweringScope {
195b5af11dfSReid Kleckner   TypeLoweringScope(CodeViewDebug &CVD) : CVD(CVD) { ++CVD.TypeEmissionLevel; }
196b5af11dfSReid Kleckner   ~TypeLoweringScope() {
197b5af11dfSReid Kleckner     // Don't decrement TypeEmissionLevel until after emitting deferred types, so
198b5af11dfSReid Kleckner     // inner TypeLoweringScopes don't attempt to emit deferred types.
199b5af11dfSReid Kleckner     if (CVD.TypeEmissionLevel == 1)
200b5af11dfSReid Kleckner       CVD.emitDeferredCompleteTypes();
201b5af11dfSReid Kleckner     --CVD.TypeEmissionLevel;
202b5af11dfSReid Kleckner   }
203b5af11dfSReid Kleckner   CodeViewDebug &CVD;
204b5af11dfSReid Kleckner };
205b5af11dfSReid Kleckner 
2066bdc24e7SDavid Majnemer static std::string getFullyQualifiedName(const DIScope *Ty) {
2076bdc24e7SDavid Majnemer   const DIScope *Scope = Ty->getScope().resolve();
2086bdc24e7SDavid Majnemer   return getFullyQualifiedName(Scope, getPrettyScopeName(Ty));
2096bdc24e7SDavid Majnemer }
2106bdc24e7SDavid Majnemer 
2110c5d874bSReid Kleckner TypeIndex CodeViewDebug::getScopeIndex(const DIScope *Scope) {
2120c5d874bSReid Kleckner   // No scope means global scope and that uses the zero index.
2130c5d874bSReid Kleckner   if (!Scope || isa<DIFile>(Scope))
2140c5d874bSReid Kleckner     return TypeIndex();
2150c5d874bSReid Kleckner 
2160c5d874bSReid Kleckner   assert(!isa<DIType>(Scope) && "shouldn't make a namespace scope for a type");
2170c5d874bSReid Kleckner 
2180c5d874bSReid Kleckner   // Check if we've already translated this scope.
2190c5d874bSReid Kleckner   auto I = TypeIndices.find({Scope, nullptr});
2200c5d874bSReid Kleckner   if (I != TypeIndices.end())
2210c5d874bSReid Kleckner     return I->second;
2220c5d874bSReid Kleckner 
2230c5d874bSReid Kleckner   // Build the fully qualified name of the scope.
2246bdc24e7SDavid Majnemer   std::string ScopeName = getFullyQualifiedName(Scope);
2250c5d874bSReid Kleckner   TypeIndex TI =
2265e3e4bb2SZachary Turner       TypeTable.writeKnownType(StringIdRecord(TypeIndex(), ScopeName));
2270c5d874bSReid Kleckner   return recordTypeIndexForDINode(Scope, TI);
2280c5d874bSReid Kleckner }
2290c5d874bSReid Kleckner 
23075c3ebfaSDavid Majnemer TypeIndex CodeViewDebug::getFuncIdForSubprogram(const DISubprogram *SP) {
23167f684e1SDavid Majnemer   assert(SP);
2322280f932SReid Kleckner 
23375c3ebfaSDavid Majnemer   // Check if we've already translated this subprogram.
23476c9eb99SAmjad Aboud   auto I = TypeIndices.find({SP, nullptr});
23575c3ebfaSDavid Majnemer   if (I != TypeIndices.end())
23675c3ebfaSDavid Majnemer     return I->second;
2372280f932SReid Kleckner 
238ac945e27SReid Kleckner   // The display name includes function template arguments. Drop them to match
239ac945e27SReid Kleckner   // MSVC.
240ac945e27SReid Kleckner   StringRef DisplayName = SP->getDisplayName().split('<').first;
24175c3ebfaSDavid Majnemer 
2420c5d874bSReid Kleckner   const DIScope *Scope = SP->getScope().resolve();
2430c5d874bSReid Kleckner   TypeIndex TI;
2440c5d874bSReid Kleckner   if (const auto *Class = dyn_cast_or_null<DICompositeType>(Scope)) {
2450c5d874bSReid Kleckner     // If the scope is a DICompositeType, then this must be a method. Member
2460c5d874bSReid Kleckner     // function types take some special handling, and require access to the
2470c5d874bSReid Kleckner     // subprogram.
2480c5d874bSReid Kleckner     TypeIndex ClassType = getTypeIndex(Class);
2490c5d874bSReid Kleckner     MemberFuncIdRecord MFuncId(ClassType, getMemberFunctionType(SP, Class),
2500c5d874bSReid Kleckner                                DisplayName);
2515e3e4bb2SZachary Turner     TI = TypeTable.writeKnownType(MFuncId);
2520c5d874bSReid Kleckner   } else {
2530c5d874bSReid Kleckner     // Otherwise, this must be a free function.
2540c5d874bSReid Kleckner     TypeIndex ParentScope = getScopeIndex(Scope);
2550c5d874bSReid Kleckner     FuncIdRecord FuncId(ParentScope, getTypeIndex(SP->getType()), DisplayName);
2565e3e4bb2SZachary Turner     TI = TypeTable.writeKnownType(FuncId);
2572280f932SReid Kleckner   }
2582280f932SReid Kleckner 
2590c5d874bSReid Kleckner   return recordTypeIndexForDINode(SP, TI);
2600c5d874bSReid Kleckner }
2610c5d874bSReid Kleckner 
2620c5d874bSReid Kleckner TypeIndex CodeViewDebug::getMemberFunctionType(const DISubprogram *SP,
2630c5d874bSReid Kleckner                                                const DICompositeType *Class) {
264b5af11dfSReid Kleckner   // Always use the method declaration as the key for the function type. The
265b5af11dfSReid Kleckner   // method declaration contains the this adjustment.
266b5af11dfSReid Kleckner   if (SP->getDeclaration())
267b5af11dfSReid Kleckner     SP = SP->getDeclaration();
268b5af11dfSReid Kleckner   assert(!SP->getDeclaration() && "should use declaration as key");
269b5af11dfSReid Kleckner 
2700c5d874bSReid Kleckner   // Key the MemberFunctionRecord into the map as {SP, Class}. It won't collide
2710c5d874bSReid Kleckner   // with the MemberFuncIdRecord, which is keyed in as {SP, nullptr}.
272b5af11dfSReid Kleckner   auto I = TypeIndices.find({SP, Class});
2730c5d874bSReid Kleckner   if (I != TypeIndices.end())
2740c5d874bSReid Kleckner     return I->second;
2750c5d874bSReid Kleckner 
276b5af11dfSReid Kleckner   // Make sure complete type info for the class is emitted *after* the member
277b5af11dfSReid Kleckner   // function type, as the complete class type is likely to reference this
278b5af11dfSReid Kleckner   // member function type.
279b5af11dfSReid Kleckner   TypeLoweringScope S(*this);
2800c5d874bSReid Kleckner   TypeIndex TI =
281b5af11dfSReid Kleckner       lowerTypeMemberFunction(SP->getType(), Class, SP->getThisAdjustment());
2820c5d874bSReid Kleckner   return recordTypeIndexForDINode(SP, TI, Class);
2830c5d874bSReid Kleckner }
2840c5d874bSReid Kleckner 
285acee5685SAmjad Aboud TypeIndex CodeViewDebug::recordTypeIndexForDINode(const DINode *Node,
286acee5685SAmjad Aboud                                                   TypeIndex TI,
28776c9eb99SAmjad Aboud                                                   const DIType *ClassTy) {
28876c9eb99SAmjad Aboud   auto InsertResult = TypeIndices.insert({{Node, ClassTy}, TI});
289a8d57407SReid Kleckner   (void)InsertResult;
290a8d57407SReid Kleckner   assert(InsertResult.second && "DINode was already assigned a type index");
2910c5d874bSReid Kleckner   return TI;
292a8d57407SReid Kleckner }
293a8d57407SReid Kleckner 
29476c9eb99SAmjad Aboud unsigned CodeViewDebug::getPointerSizeInBytes() {
29576c9eb99SAmjad Aboud   return MMI->getModule()->getDataLayout().getPointerSizeInBits() / 8;
29676c9eb99SAmjad Aboud }
29776c9eb99SAmjad Aboud 
298876330d5SReid Kleckner void CodeViewDebug::recordLocalVariable(LocalVariable &&Var,
299876330d5SReid Kleckner                                         const DILocation *InlinedAt) {
300876330d5SReid Kleckner   if (InlinedAt) {
301876330d5SReid Kleckner     // This variable was inlined. Associate it with the InlineSite.
302876330d5SReid Kleckner     const DISubprogram *Inlinee = Var.DIVar->getScope()->getSubprogram();
303876330d5SReid Kleckner     InlineSite &Site = getInlineSite(InlinedAt, Inlinee);
304876330d5SReid Kleckner     Site.InlinedLocals.emplace_back(Var);
305876330d5SReid Kleckner   } else {
306876330d5SReid Kleckner     // This variable goes in the main ProcSym.
307876330d5SReid Kleckner     CurFn->Locals.emplace_back(Var);
308876330d5SReid Kleckner   }
309876330d5SReid Kleckner }
310876330d5SReid Kleckner 
311829365aeSReid Kleckner static void addLocIfNotPresent(SmallVectorImpl<const DILocation *> &Locs,
312829365aeSReid Kleckner                                const DILocation *Loc) {
313829365aeSReid Kleckner   auto B = Locs.begin(), E = Locs.end();
314829365aeSReid Kleckner   if (std::find(B, E, Loc) == E)
315829365aeSReid Kleckner     Locs.push_back(Loc);
316829365aeSReid Kleckner }
317829365aeSReid Kleckner 
318bdc4956bSBenjamin Kramer void CodeViewDebug::maybeRecordLocation(const DebugLoc &DL,
31970f5bc99SReid Kleckner                                         const MachineFunction *MF) {
3209533af4fSReid Kleckner   // Skip this instruction if it has the same location as the previous one.
3219533af4fSReid Kleckner   if (DL == CurFn->LastLoc)
3229533af4fSReid Kleckner     return;
3239533af4fSReid Kleckner 
3249533af4fSReid Kleckner   const DIScope *Scope = DL.get()->getScope();
32570f5bc99SReid Kleckner   if (!Scope)
32670f5bc99SReid Kleckner     return;
3279533af4fSReid Kleckner 
32870f5bc99SReid Kleckner   // Skip this line if it is longer than the maximum we can record.
3292214ed89SReid Kleckner   LineInfo LI(DL.getLine(), DL.getLine(), /*IsStatement=*/true);
3302214ed89SReid Kleckner   if (LI.getStartLine() != DL.getLine() || LI.isAlwaysStepInto() ||
3312214ed89SReid Kleckner       LI.isNeverStepInto())
33270f5bc99SReid Kleckner     return;
33370f5bc99SReid Kleckner 
3342214ed89SReid Kleckner   ColumnInfo CI(DL.getCol(), /*EndColumn=*/0);
3352214ed89SReid Kleckner   if (CI.getStartColumn() != DL.getCol())
3362214ed89SReid Kleckner     return;
33700d9639cSReid Kleckner 
3382214ed89SReid Kleckner   if (!CurFn->HaveLineInfo)
3392214ed89SReid Kleckner     CurFn->HaveLineInfo = true;
3402214ed89SReid Kleckner   unsigned FileId = 0;
3412214ed89SReid Kleckner   if (CurFn->LastLoc.get() && CurFn->LastLoc->getFile() == DL->getFile())
3422214ed89SReid Kleckner     FileId = CurFn->LastFileId;
3432214ed89SReid Kleckner   else
3442214ed89SReid Kleckner     FileId = CurFn->LastFileId = maybeRecordFile(DL->getFile());
3452214ed89SReid Kleckner   CurFn->LastLoc = DL;
346f3b9ba49SReid Kleckner 
347f3b9ba49SReid Kleckner   unsigned FuncId = CurFn->FuncId;
348876330d5SReid Kleckner   if (const DILocation *SiteLoc = DL->getInlinedAt()) {
349829365aeSReid Kleckner     const DILocation *Loc = DL.get();
350829365aeSReid Kleckner 
351f3b9ba49SReid Kleckner     // If this location was actually inlined from somewhere else, give it the ID
352f3b9ba49SReid Kleckner     // of the inline call site.
353876330d5SReid Kleckner     FuncId =
354876330d5SReid Kleckner         getInlineSite(SiteLoc, Loc->getScope()->getSubprogram()).SiteFuncId;
355829365aeSReid Kleckner 
356f3b9ba49SReid Kleckner     // Ensure we have links in the tree of inline call sites.
357829365aeSReid Kleckner     bool FirstLoc = true;
358829365aeSReid Kleckner     while ((SiteLoc = Loc->getInlinedAt())) {
359876330d5SReid Kleckner       InlineSite &Site =
360876330d5SReid Kleckner           getInlineSite(SiteLoc, Loc->getScope()->getSubprogram());
361829365aeSReid Kleckner       if (!FirstLoc)
362829365aeSReid Kleckner         addLocIfNotPresent(Site.ChildSites, Loc);
363829365aeSReid Kleckner       FirstLoc = false;
364829365aeSReid Kleckner       Loc = SiteLoc;
365f3b9ba49SReid Kleckner     }
366829365aeSReid Kleckner     addLocIfNotPresent(CurFn->ChildSites, Loc);
367f3b9ba49SReid Kleckner   }
368f3b9ba49SReid Kleckner 
369dac21b43SReid Kleckner   OS.EmitCVLocDirective(FuncId, FileId, DL.getLine(), DL.getCol(),
370a9f4cc95SReid Kleckner                         /*PrologueEnd=*/false, /*IsStmt=*/false,
371a9f4cc95SReid Kleckner                         DL->getFilename(), SMLoc());
37270f5bc99SReid Kleckner }
37370f5bc99SReid Kleckner 
3745d122f87SReid Kleckner void CodeViewDebug::emitCodeViewMagicVersion() {
3755d122f87SReid Kleckner   OS.EmitValueToAlignment(4);
3765d122f87SReid Kleckner   OS.AddComment("Debug section magic");
3775d122f87SReid Kleckner   OS.EmitIntValue(COFF::DEBUG_SECTION_MAGIC, 4);
3785d122f87SReid Kleckner }
3795d122f87SReid Kleckner 
38070f5bc99SReid Kleckner void CodeViewDebug::endModule() {
3816f3406dfSReid Kleckner   if (!Asm || !MMI->hasDebugInfo())
38270f5bc99SReid Kleckner     return;
38370f5bc99SReid Kleckner 
38470f5bc99SReid Kleckner   assert(Asm != nullptr);
38570f5bc99SReid Kleckner 
38670f5bc99SReid Kleckner   // The COFF .debug$S section consists of several subsections, each starting
38770f5bc99SReid Kleckner   // with a 4-byte control code (e.g. 0xF1, 0xF2, etc) and then a 4-byte length
38870f5bc99SReid Kleckner   // of the payload followed by the payload itself.  The subsections are 4-byte
38970f5bc99SReid Kleckner   // aligned.
39070f5bc99SReid Kleckner 
3916f3406dfSReid Kleckner   // Use the generic .debug$S section, and make a subsection for all the inlined
3926f3406dfSReid Kleckner   // subprograms.
3936f3406dfSReid Kleckner   switchToDebugSectionForSymbol(nullptr);
3945d122f87SReid Kleckner   emitInlineeLinesSubsection();
3951fcd610cSReid Kleckner 
3962214ed89SReid Kleckner   // Emit per-function debug information.
3972214ed89SReid Kleckner   for (auto &P : FnDebugInfo)
398577be0feSDavid Majnemer     if (!P.first->isDeclarationForLinker())
3992214ed89SReid Kleckner       emitDebugInfoForFunction(P.first, P.second);
40070f5bc99SReid Kleckner 
4016f3406dfSReid Kleckner   // Emit global variable debug information.
4023128b10cSDavid Majnemer   setCurrentSubprogram(nullptr);
4036f3406dfSReid Kleckner   emitDebugInfoForGlobals();
4046f3406dfSReid Kleckner 
405b510b458SHans Wennborg   // Emit retained types.
406b510b458SHans Wennborg   emitDebugInfoForRetainedTypes();
407b510b458SHans Wennborg 
4085d122f87SReid Kleckner   // Switch back to the generic .debug$S section after potentially processing
4095d122f87SReid Kleckner   // comdat symbol sections.
4105d122f87SReid Kleckner   switchToDebugSectionForSymbol(nullptr);
4115d122f87SReid Kleckner 
4123128b10cSDavid Majnemer   // Emit UDT records for any types used by global variables.
4133128b10cSDavid Majnemer   if (!GlobalUDTs.empty()) {
4143128b10cSDavid Majnemer     MCSymbol *SymbolsEnd = beginCVSubsection(ModuleSubstreamKind::Symbols);
4153128b10cSDavid Majnemer     emitDebugInfoForUDTs(GlobalUDTs);
4163128b10cSDavid Majnemer     endCVSubsection(SymbolsEnd);
4173128b10cSDavid Majnemer   }
4183128b10cSDavid Majnemer 
41970f5bc99SReid Kleckner   // This subsection holds a file index to offset in string table table.
420dac21b43SReid Kleckner   OS.AddComment("File index to string table offset subsection");
421dac21b43SReid Kleckner   OS.EmitCVFileChecksumsDirective();
42270f5bc99SReid Kleckner 
42370f5bc99SReid Kleckner   // This subsection holds the string table.
424dac21b43SReid Kleckner   OS.AddComment("String table");
425dac21b43SReid Kleckner   OS.EmitCVStringTableDirective();
42670f5bc99SReid Kleckner 
4275acacbb0SReid Kleckner   // Emit type information last, so that any types we translate while emitting
4285acacbb0SReid Kleckner   // function info are included.
4295acacbb0SReid Kleckner   emitTypeInformation();
4305acacbb0SReid Kleckner 
43170f5bc99SReid Kleckner   clear();
43270f5bc99SReid Kleckner }
43370f5bc99SReid Kleckner 
434b9456a5eSDavid Majnemer static void emitNullTerminatedSymbolName(MCStreamer &OS, StringRef S) {
435b9456a5eSDavid Majnemer   // Microsoft's linker seems to have trouble with symbol names longer than
436b9456a5eSDavid Majnemer   // 0xffd8 bytes.
437b9456a5eSDavid Majnemer   S = S.substr(0, 0xffd8);
438b9456a5eSDavid Majnemer   SmallString<32> NullTerminatedString(S);
439b9456a5eSDavid Majnemer   NullTerminatedString.push_back('\0');
440b9456a5eSDavid Majnemer   OS.EmitBytes(NullTerminatedString);
441b9456a5eSDavid Majnemer }
442b9456a5eSDavid Majnemer 
443f3b9ba49SReid Kleckner void CodeViewDebug::emitTypeInformation() {
4442280f932SReid Kleckner   // Do nothing if we have no debug info or if no non-trivial types were emitted
4452280f932SReid Kleckner   // to TypeTable during codegen.
44676c9eb99SAmjad Aboud   NamedMDNode *CU_Nodes = MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
447fbd7787dSReid Kleckner   if (!CU_Nodes)
448fbd7787dSReid Kleckner     return;
4492280f932SReid Kleckner   if (TypeTable.empty())
450fbd7787dSReid Kleckner     return;
451fbd7787dSReid Kleckner 
452f3b9ba49SReid Kleckner   // Start the .debug$T section with 0x4.
453dac21b43SReid Kleckner   OS.SwitchSection(Asm->getObjFileLowering().getCOFFDebugTypesSection());
4545d122f87SReid Kleckner   emitCodeViewMagicVersion();
455f3b9ba49SReid Kleckner 
456fbdbe9e2SReid Kleckner   SmallString<8> CommentPrefix;
457fbdbe9e2SReid Kleckner   if (OS.isVerboseAsm()) {
458fbdbe9e2SReid Kleckner     CommentPrefix += '\t';
459fbdbe9e2SReid Kleckner     CommentPrefix += Asm->MAI->getCommentString();
460fbdbe9e2SReid Kleckner     CommentPrefix += ' ';
461fbdbe9e2SReid Kleckner   }
462fbdbe9e2SReid Kleckner 
463fbdbe9e2SReid Kleckner   CVTypeDumper CVTD(nullptr, /*PrintRecordBytes=*/false);
4642280f932SReid Kleckner   TypeTable.ForEachRecord(
465fbdbe9e2SReid Kleckner       [&](TypeIndex Index, StringRef Record) {
466fbdbe9e2SReid Kleckner         if (OS.isVerboseAsm()) {
467fbdbe9e2SReid Kleckner           // Emit a block comment describing the type record for readability.
468fbdbe9e2SReid Kleckner           SmallString<512> CommentBlock;
469fbdbe9e2SReid Kleckner           raw_svector_ostream CommentOS(CommentBlock);
470fbdbe9e2SReid Kleckner           ScopedPrinter SP(CommentOS);
471fbdbe9e2SReid Kleckner           SP.setPrefix(CommentPrefix);
472fbdbe9e2SReid Kleckner           CVTD.setPrinter(&SP);
473c92e9469SReid Kleckner           Error E = CVTD.dump({Record.bytes_begin(), Record.bytes_end()});
474c92e9469SReid Kleckner           if (E) {
475c92e9469SReid Kleckner             logAllUnhandledErrors(std::move(E), errs(), "error: ");
476c92e9469SReid Kleckner             llvm_unreachable("produced malformed type record");
477c92e9469SReid Kleckner           }
478fbdbe9e2SReid Kleckner           // emitRawComment will insert its own tab and comment string before
479fbdbe9e2SReid Kleckner           // the first line, so strip off our first one. It also prints its own
480fbdbe9e2SReid Kleckner           // newline.
481fbdbe9e2SReid Kleckner           OS.emitRawComment(
482fbdbe9e2SReid Kleckner               CommentOS.str().drop_front(CommentPrefix.size() - 1).rtrim());
483c92e9469SReid Kleckner         } else {
484c92e9469SReid Kleckner #ifndef NDEBUG
485c92e9469SReid Kleckner           // Assert that the type data is valid even if we aren't dumping
486c92e9469SReid Kleckner           // comments. The MSVC linker doesn't do much type record validation,
487c92e9469SReid Kleckner           // so the first link of an invalid type record can succeed while
488c92e9469SReid Kleckner           // subsequent links will fail with LNK1285.
489d66889cbSZachary Turner           ByteStream Stream({Record.bytes_begin(), Record.bytes_end()});
490c92e9469SReid Kleckner           CVTypeArray Types;
491c92e9469SReid Kleckner           StreamReader Reader(Stream);
492c92e9469SReid Kleckner           Error E = Reader.readArray(Types, Reader.getLength());
493c92e9469SReid Kleckner           if (!E) {
494c92e9469SReid Kleckner             TypeVisitorCallbacks C;
495c92e9469SReid Kleckner             E = CVTypeVisitor(C).visitTypeStream(Types);
496c92e9469SReid Kleckner           }
497c92e9469SReid Kleckner           if (E) {
498c92e9469SReid Kleckner             logAllUnhandledErrors(std::move(E), errs(), "error: ");
499c92e9469SReid Kleckner             llvm_unreachable("produced malformed type record");
500c92e9469SReid Kleckner           }
501c92e9469SReid Kleckner #endif
502fbdbe9e2SReid Kleckner         }
503fbdbe9e2SReid Kleckner         OS.EmitBinaryData(Record);
5042280f932SReid Kleckner       });
505f3b9ba49SReid Kleckner }
506f3b9ba49SReid Kleckner 
5075d122f87SReid Kleckner void CodeViewDebug::emitInlineeLinesSubsection() {
5081fcd610cSReid Kleckner   if (InlinedSubprograms.empty())
5091fcd610cSReid Kleckner     return;
5101fcd610cSReid Kleckner 
5111fcd610cSReid Kleckner   OS.AddComment("Inlinee lines subsection");
5126f3406dfSReid Kleckner   MCSymbol *InlineEnd = beginCVSubsection(ModuleSubstreamKind::InlineeLines);
5131fcd610cSReid Kleckner 
5141fcd610cSReid Kleckner   // We don't provide any extra file info.
5151fcd610cSReid Kleckner   // FIXME: Find out if debuggers use this info.
51630579ec8SDavid Majnemer   OS.AddComment("Inlinee lines signature");
5171fcd610cSReid Kleckner   OS.EmitIntValue(unsigned(InlineeLinesSignature::Normal), 4);
5181fcd610cSReid Kleckner 
5191fcd610cSReid Kleckner   for (const DISubprogram *SP : InlinedSubprograms) {
52076c9eb99SAmjad Aboud     assert(TypeIndices.count({SP, nullptr}));
52176c9eb99SAmjad Aboud     TypeIndex InlineeIdx = TypeIndices[{SP, nullptr}];
5222280f932SReid Kleckner 
52330579ec8SDavid Majnemer     OS.AddBlankLine();
5241fcd610cSReid Kleckner     unsigned FileId = maybeRecordFile(SP->getFile());
5251fcd610cSReid Kleckner     OS.AddComment("Inlined function " + SP->getDisplayName() + " starts at " +
5261fcd610cSReid Kleckner                   SP->getFilename() + Twine(':') + Twine(SP->getLine()));
52730579ec8SDavid Majnemer     OS.AddBlankLine();
5281fcd610cSReid Kleckner     // The filechecksum table uses 8 byte entries for now, and file ids start at
5291fcd610cSReid Kleckner     // 1.
5301fcd610cSReid Kleckner     unsigned FileOffset = (FileId - 1) * 8;
53130579ec8SDavid Majnemer     OS.AddComment("Type index of inlined function");
5322280f932SReid Kleckner     OS.EmitIntValue(InlineeIdx.getIndex(), 4);
53330579ec8SDavid Majnemer     OS.AddComment("Offset into filechecksum table");
5341fcd610cSReid Kleckner     OS.EmitIntValue(FileOffset, 4);
53530579ec8SDavid Majnemer     OS.AddComment("Starting line number");
5361fcd610cSReid Kleckner     OS.EmitIntValue(SP->getLine(), 4);
5371fcd610cSReid Kleckner   }
5381fcd610cSReid Kleckner 
5396f3406dfSReid Kleckner   endCVSubsection(InlineEnd);
5401fcd610cSReid Kleckner }
5411fcd610cSReid Kleckner 
542f3b9ba49SReid Kleckner void CodeViewDebug::emitInlinedCallSite(const FunctionInfo &FI,
543f3b9ba49SReid Kleckner                                         const DILocation *InlinedAt,
544f3b9ba49SReid Kleckner                                         const InlineSite &Site) {
545f9c275feSReid Kleckner   MCSymbol *InlineBegin = MMI->getContext().createTempSymbol(),
546f9c275feSReid Kleckner            *InlineEnd = MMI->getContext().createTempSymbol();
547f3b9ba49SReid Kleckner 
54876c9eb99SAmjad Aboud   assert(TypeIndices.count({Site.Inlinee, nullptr}));
54976c9eb99SAmjad Aboud   TypeIndex InlineeIdx = TypeIndices[{Site.Inlinee, nullptr}];
550f3b9ba49SReid Kleckner 
551f3b9ba49SReid Kleckner   // SymbolRecord
552dac21b43SReid Kleckner   OS.AddComment("Record length");
553eb3bcdd2SReid Kleckner   OS.emitAbsoluteSymbolDiff(InlineEnd, InlineBegin, 2);   // RecordLength
554f3b9ba49SReid Kleckner   OS.EmitLabel(InlineBegin);
555dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE");
55663a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE, 2); // RecordKind
557f3b9ba49SReid Kleckner 
558dac21b43SReid Kleckner   OS.AddComment("PtrParent");
559dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
560dac21b43SReid Kleckner   OS.AddComment("PtrEnd");
561dac21b43SReid Kleckner   OS.EmitIntValue(0, 4);
562dac21b43SReid Kleckner   OS.AddComment("Inlinee type index");
5632280f932SReid Kleckner   OS.EmitIntValue(InlineeIdx.getIndex(), 4);
564f3b9ba49SReid Kleckner 
5651fcd610cSReid Kleckner   unsigned FileId = maybeRecordFile(Site.Inlinee->getFile());
5661fcd610cSReid Kleckner   unsigned StartLineNum = Site.Inlinee->getLine();
5671fcd610cSReid Kleckner 
5681fcd610cSReid Kleckner   OS.EmitCVInlineLinetableDirective(Site.SiteFuncId, FileId, StartLineNum,
569a9f4cc95SReid Kleckner                                     FI.Begin, FI.End);
570f3b9ba49SReid Kleckner 
571f3b9ba49SReid Kleckner   OS.EmitLabel(InlineEnd);
572f3b9ba49SReid Kleckner 
57310dd55c5SReid Kleckner   emitLocalVariableList(Site.InlinedLocals);
574f9c275feSReid Kleckner 
575f3b9ba49SReid Kleckner   // Recurse on child inlined call sites before closing the scope.
576f3b9ba49SReid Kleckner   for (const DILocation *ChildSite : Site.ChildSites) {
577f3b9ba49SReid Kleckner     auto I = FI.InlineSites.find(ChildSite);
578f3b9ba49SReid Kleckner     assert(I != FI.InlineSites.end() &&
579f3b9ba49SReid Kleckner            "child site not in function inline site map");
580f3b9ba49SReid Kleckner     emitInlinedCallSite(FI, ChildSite, I->second);
581f3b9ba49SReid Kleckner   }
582f3b9ba49SReid Kleckner 
583f3b9ba49SReid Kleckner   // Close the scope.
584dac21b43SReid Kleckner   OS.AddComment("Record length");
585dac21b43SReid Kleckner   OS.EmitIntValue(2, 2);                                  // RecordLength
586dac21b43SReid Kleckner   OS.AddComment("Record kind: S_INLINESITE_END");
58763a2846eSZachary Turner   OS.EmitIntValue(SymbolKind::S_INLINESITE_END, 2); // RecordKind
588f3b9ba49SReid Kleckner }
589f3b9ba49SReid Kleckner 
5905d122f87SReid Kleckner void CodeViewDebug::switchToDebugSectionForSymbol(const MCSymbol *GVSym) {
5915d122f87SReid Kleckner   // If we have a symbol, it may be in a section that is COMDAT. If so, find the
5925d122f87SReid Kleckner   // comdat key. A section may be comdat because of -ffunction-sections or
5935d122f87SReid Kleckner   // because it is comdat in the IR.
5945d122f87SReid Kleckner   MCSectionCOFF *GVSec =
5955d122f87SReid Kleckner       GVSym ? dyn_cast<MCSectionCOFF>(&GVSym->getSection()) : nullptr;
5965d122f87SReid Kleckner   const MCSymbol *KeySym = GVSec ? GVSec->getCOMDATSymbol() : nullptr;
5975d122f87SReid Kleckner 
5985d122f87SReid Kleckner   MCSectionCOFF *DebugSec = cast<MCSectionCOFF>(
5995d122f87SReid Kleckner       Asm->getObjFileLowering().getCOFFDebugSymbolsSection());
6005d122f87SReid Kleckner   DebugSec = OS.getContext().getAssociativeCOFFSection(DebugSec, KeySym);
6015d122f87SReid Kleckner 
6025d122f87SReid Kleckner   OS.SwitchSection(DebugSec);
6035d122f87SReid Kleckner 
6045d122f87SReid Kleckner   // Emit the magic version number if this is the first time we've switched to
6055d122f87SReid Kleckner   // this section.
6065d122f87SReid Kleckner   if (ComdatDebugSections.insert(DebugSec).second)
6075d122f87SReid Kleckner     emitCodeViewMagicVersion();
6085d122f87SReid Kleckner }
6095d122f87SReid Kleckner 
6102214ed89SReid Kleckner void CodeViewDebug::emitDebugInfoForFunction(const Function *GV,
6112214ed89SReid Kleckner                                              FunctionInfo &FI) {
61270f5bc99SReid Kleckner   // For each function there is a separate subsection
61370f5bc99SReid Kleckner   // which holds the PC to file:line table.
61470f5bc99SReid Kleckner   const MCSymbol *Fn = Asm->getSymbol(GV);
61570f5bc99SReid Kleckner   assert(Fn);
61670f5bc99SReid Kleckner 
6175d122f87SReid Kleckner   // Switch to the to a comdat section, if appropriate.
6185d122f87SReid Kleckner   switchToDebugSectionForSymbol(Fn);
6195d122f87SReid Kleckner 
620ac945e27SReid Kleckner   std::string FuncName;
6213128b10cSDavid Majnemer   auto *SP = GV->getSubprogram();
62267f684e1SDavid Majnemer   assert(SP);
6233128b10cSDavid Majnemer   setCurrentSubprogram(SP);
624ac945e27SReid Kleckner 
625ac945e27SReid Kleckner   // If we have a display name, build the fully qualified name by walking the
626ac945e27SReid Kleckner   // chain of scopes.
62767f684e1SDavid Majnemer   if (!SP->getDisplayName().empty())
6280c5d874bSReid Kleckner     FuncName =
6290c5d874bSReid Kleckner         getFullyQualifiedName(SP->getScope().resolve(), SP->getDisplayName());
63070f5bc99SReid Kleckner 
63170f5bc99SReid Kleckner   // If our DISubprogram name is empty, use the mangled name.
63270f5bc99SReid Kleckner   if (FuncName.empty())
63370f5bc99SReid Kleckner     FuncName = GlobalValue::getRealLinkageName(GV->getName());
63470f5bc99SReid Kleckner 
63570f5bc99SReid Kleckner   // Emit a symbol subsection, required by VS2012+ to find function boundaries.
636dac21b43SReid Kleckner   OS.AddComment("Symbol subsection for " + Twine(FuncName));
6376f3406dfSReid Kleckner   MCSymbol *SymbolsEnd = beginCVSubsection(ModuleSubstreamKind::Symbols);
63870f5bc99SReid Kleckner   {
639f9c275feSReid Kleckner     MCSymbol *ProcRecordBegin = MMI->getContext().createTempSymbol(),
640f9c275feSReid Kleckner              *ProcRecordEnd = MMI->getContext().createTempSymbol();
641dac21b43SReid Kleckner     OS.AddComment("Record length");
642eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(ProcRecordEnd, ProcRecordBegin, 2);
643dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordBegin);
64470f5bc99SReid Kleckner 
6457abd269aSDavid Majnemer   if (GV->hasLocalLinkage()) {
6467abd269aSDavid Majnemer     OS.AddComment("Record kind: S_LPROC32_ID");
6477abd269aSDavid Majnemer     OS.EmitIntValue(unsigned(SymbolKind::S_LPROC32_ID), 2);
6487abd269aSDavid Majnemer   } else {
649dac21b43SReid Kleckner     OS.AddComment("Record kind: S_GPROC32_ID");
65063a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_GPROC32_ID), 2);
6517abd269aSDavid Majnemer   }
65270f5bc99SReid Kleckner 
65330579ec8SDavid Majnemer     // These fields are filled in by tools like CVPACK which run after the fact.
654dac21b43SReid Kleckner     OS.AddComment("PtrParent");
655dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
656dac21b43SReid Kleckner     OS.AddComment("PtrEnd");
657dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
658dac21b43SReid Kleckner     OS.AddComment("PtrNext");
659dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
66070f5bc99SReid Kleckner     // This is the important bit that tells the debugger where the function
66170f5bc99SReid Kleckner     // code is located and what's its size:
662dac21b43SReid Kleckner     OS.AddComment("Code size");
663eb3bcdd2SReid Kleckner     OS.emitAbsoluteSymbolDiff(FI.End, Fn, 4);
664dac21b43SReid Kleckner     OS.AddComment("Offset after prologue");
665dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
666dac21b43SReid Kleckner     OS.AddComment("Offset before epilogue");
667dac21b43SReid Kleckner     OS.EmitIntValue(0, 4);
668dac21b43SReid Kleckner     OS.AddComment("Function type index");
66975c3ebfaSDavid Majnemer     OS.EmitIntValue(getFuncIdForSubprogram(GV->getSubprogram()).getIndex(), 4);
670dac21b43SReid Kleckner     OS.AddComment("Function section relative address");
671dac21b43SReid Kleckner     OS.EmitCOFFSecRel32(Fn);
672dac21b43SReid Kleckner     OS.AddComment("Function section index");
673dac21b43SReid Kleckner     OS.EmitCOFFSectionIndex(Fn);
674dac21b43SReid Kleckner     OS.AddComment("Flags");
675dac21b43SReid Kleckner     OS.EmitIntValue(0, 1);
67670f5bc99SReid Kleckner     // Emit the function display name as a null-terminated string.
677dac21b43SReid Kleckner     OS.AddComment("Function name");
6781256125fSDavid Majnemer     // Truncate the name so we won't overflow the record length field.
679b9456a5eSDavid Majnemer     emitNullTerminatedSymbolName(OS, FuncName);
680dac21b43SReid Kleckner     OS.EmitLabel(ProcRecordEnd);
68170f5bc99SReid Kleckner 
68210dd55c5SReid Kleckner     emitLocalVariableList(FI.Locals);
683f9c275feSReid Kleckner 
684f3b9ba49SReid Kleckner     // Emit inlined call site information. Only emit functions inlined directly
685f3b9ba49SReid Kleckner     // into the parent function. We'll emit the other sites recursively as part
686f3b9ba49SReid Kleckner     // of their parent inline site.
687f9c275feSReid Kleckner     for (const DILocation *InlinedAt : FI.ChildSites) {
688f9c275feSReid Kleckner       auto I = FI.InlineSites.find(InlinedAt);
689f9c275feSReid Kleckner       assert(I != FI.InlineSites.end() &&
690f9c275feSReid Kleckner              "child site not in function inline site map");
691f9c275feSReid Kleckner       emitInlinedCallSite(FI, InlinedAt, I->second);
692f3b9ba49SReid Kleckner     }
693f3b9ba49SReid Kleckner 
6943128b10cSDavid Majnemer     if (SP != nullptr)
6953128b10cSDavid Majnemer       emitDebugInfoForUDTs(LocalUDTs);
6963128b10cSDavid Majnemer 
69770f5bc99SReid Kleckner     // We're done with this function.
698dac21b43SReid Kleckner     OS.AddComment("Record length");
699dac21b43SReid Kleckner     OS.EmitIntValue(0x0002, 2);
700dac21b43SReid Kleckner     OS.AddComment("Record kind: S_PROC_ID_END");
70163a2846eSZachary Turner     OS.EmitIntValue(unsigned(SymbolKind::S_PROC_ID_END), 2);
70270f5bc99SReid Kleckner   }
7036f3406dfSReid Kleckner   endCVSubsection(SymbolsEnd);
70470f5bc99SReid Kleckner 
7052214ed89SReid Kleckner   // We have an assembler directive that takes care of the whole line table.
706dac21b43SReid Kleckner   OS.EmitCVLinetableDirective(FI.FuncId, Fn, FI.End);
70770f5bc99SReid Kleckner }
70870f5bc99SReid Kleckner 
709876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
710876330d5SReid Kleckner CodeViewDebug::createDefRangeMem(uint16_t CVRegister, int Offset) {
711876330d5SReid Kleckner   LocalVarDefRange DR;
712c6a2f214SAaron Ballman   DR.InMemory = -1;
713876330d5SReid Kleckner   DR.DataOffset = Offset;
714876330d5SReid Kleckner   assert(DR.DataOffset == Offset && "truncation");
715876330d5SReid Kleckner   DR.StructOffset = 0;
716876330d5SReid Kleckner   DR.CVRegister = CVRegister;
717876330d5SReid Kleckner   return DR;
718876330d5SReid Kleckner }
719876330d5SReid Kleckner 
720876330d5SReid Kleckner CodeViewDebug::LocalVarDefRange
721876330d5SReid Kleckner CodeViewDebug::createDefRangeReg(uint16_t CVRegister) {
722876330d5SReid Kleckner   LocalVarDefRange DR;
723876330d5SReid Kleckner   DR.InMemory = 0;
724876330d5SReid Kleckner   DR.DataOffset = 0;
725876330d5SReid Kleckner   DR.StructOffset = 0;
726876330d5SReid Kleckner   DR.CVRegister = CVRegister;
727876330d5SReid Kleckner   return DR;
728876330d5SReid Kleckner }
729876330d5SReid Kleckner 
730876330d5SReid Kleckner void CodeViewDebug::collectVariableInfoFromMMITable(
731876330d5SReid Kleckner     DenseSet<InlinedVariable> &Processed) {
732876330d5SReid Kleckner   const TargetSubtargetInfo &TSI = Asm->MF->getSubtarget();
733876330d5SReid Kleckner   const TargetFrameLowering *TFI = TSI.getFrameLowering();
734876330d5SReid Kleckner   const TargetRegisterInfo *TRI = TSI.getRegisterInfo();
735876330d5SReid Kleckner 
736876330d5SReid Kleckner   for (const MachineModuleInfo::VariableDbgInfo &VI :
737876330d5SReid Kleckner        MMI->getVariableDbgInfo()) {
738f9c275feSReid Kleckner     if (!VI.Var)
739f9c275feSReid Kleckner       continue;
740f9c275feSReid Kleckner     assert(VI.Var->isValidLocationForIntrinsic(VI.Loc) &&
741f9c275feSReid Kleckner            "Expected inlined-at fields to agree");
742f9c275feSReid Kleckner 
743876330d5SReid Kleckner     Processed.insert(InlinedVariable(VI.Var, VI.Loc->getInlinedAt()));
744f9c275feSReid Kleckner     LexicalScope *Scope = LScopes.findLexicalScope(VI.Loc);
745f9c275feSReid Kleckner 
746f9c275feSReid Kleckner     // If variable scope is not found then skip this variable.
747f9c275feSReid Kleckner     if (!Scope)
748f9c275feSReid Kleckner       continue;
749f9c275feSReid Kleckner 
750f9c275feSReid Kleckner     // Get the frame register used and the offset.
751f9c275feSReid Kleckner     unsigned FrameReg = 0;
752876330d5SReid Kleckner     int FrameOffset = TFI->getFrameIndexReference(*Asm->MF, VI.Slot, FrameReg);
753876330d5SReid Kleckner     uint16_t CVReg = TRI->getCodeViewRegNum(FrameReg);
754f9c275feSReid Kleckner 
755f9c275feSReid Kleckner     // Calculate the label ranges.
756876330d5SReid Kleckner     LocalVarDefRange DefRange = createDefRangeMem(CVReg, FrameOffset);
757f9c275feSReid Kleckner     for (const InsnRange &Range : Scope->getRanges()) {
758f9c275feSReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
759f9c275feSReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
760876330d5SReid Kleckner       End = End ? End : Asm->getFunctionEnd();
761876330d5SReid Kleckner       DefRange.Ranges.emplace_back(Begin, End);
762f9c275feSReid Kleckner     }
763f9c275feSReid Kleckner 
764876330d5SReid Kleckner     LocalVariable Var;
765876330d5SReid Kleckner     Var.DIVar = VI.Var;
766876330d5SReid Kleckner     Var.DefRanges.emplace_back(std::move(DefRange));
767876330d5SReid Kleckner     recordLocalVariable(std::move(Var), VI.Loc->getInlinedAt());
768f9c275feSReid Kleckner   }
769f9c275feSReid Kleckner }
770876330d5SReid Kleckner 
771876330d5SReid Kleckner void CodeViewDebug::collectVariableInfo(const DISubprogram *SP) {
772876330d5SReid Kleckner   DenseSet<InlinedVariable> Processed;
773876330d5SReid Kleckner   // Grab the variable info that was squirreled away in the MMI side-table.
774876330d5SReid Kleckner   collectVariableInfoFromMMITable(Processed);
775876330d5SReid Kleckner 
776876330d5SReid Kleckner   const TargetRegisterInfo *TRI = Asm->MF->getSubtarget().getRegisterInfo();
777876330d5SReid Kleckner 
778876330d5SReid Kleckner   for (const auto &I : DbgValues) {
779876330d5SReid Kleckner     InlinedVariable IV = I.first;
780876330d5SReid Kleckner     if (Processed.count(IV))
781876330d5SReid Kleckner       continue;
782876330d5SReid Kleckner     const DILocalVariable *DIVar = IV.first;
783876330d5SReid Kleckner     const DILocation *InlinedAt = IV.second;
784876330d5SReid Kleckner 
785876330d5SReid Kleckner     // Instruction ranges, specifying where IV is accessible.
786876330d5SReid Kleckner     const auto &Ranges = I.second;
787876330d5SReid Kleckner 
788876330d5SReid Kleckner     LexicalScope *Scope = nullptr;
789876330d5SReid Kleckner     if (InlinedAt)
790876330d5SReid Kleckner       Scope = LScopes.findInlinedScope(DIVar->getScope(), InlinedAt);
791876330d5SReid Kleckner     else
792876330d5SReid Kleckner       Scope = LScopes.findLexicalScope(DIVar->getScope());
793876330d5SReid Kleckner     // If variable scope is not found then skip this variable.
794876330d5SReid Kleckner     if (!Scope)
795876330d5SReid Kleckner       continue;
796876330d5SReid Kleckner 
797876330d5SReid Kleckner     LocalVariable Var;
798876330d5SReid Kleckner     Var.DIVar = DIVar;
799876330d5SReid Kleckner 
800876330d5SReid Kleckner     // Calculate the definition ranges.
801876330d5SReid Kleckner     for (auto I = Ranges.begin(), E = Ranges.end(); I != E; ++I) {
802876330d5SReid Kleckner       const InsnRange &Range = *I;
803876330d5SReid Kleckner       const MachineInstr *DVInst = Range.first;
804876330d5SReid Kleckner       assert(DVInst->isDebugValue() && "Invalid History entry");
805876330d5SReid Kleckner       const DIExpression *DIExpr = DVInst->getDebugExpression();
806876330d5SReid Kleckner 
807876330d5SReid Kleckner       // Bail if there is a complex DWARF expression for now.
808876330d5SReid Kleckner       if (DIExpr && DIExpr->getNumElements() > 0)
809876330d5SReid Kleckner         continue;
810876330d5SReid Kleckner 
8119a593ee7SReid Kleckner       // Bail if operand 0 is not a valid register. This means the variable is a
8129a593ee7SReid Kleckner       // simple constant, or is described by a complex expression.
8139a593ee7SReid Kleckner       // FIXME: Find a way to represent constant variables, since they are
8149a593ee7SReid Kleckner       // relatively common.
8159a593ee7SReid Kleckner       unsigned Reg =
8169a593ee7SReid Kleckner           DVInst->getOperand(0).isReg() ? DVInst->getOperand(0).getReg() : 0;
8179a593ee7SReid Kleckner       if (Reg == 0)
8186e0d5f57SReid Kleckner         continue;
8196e0d5f57SReid Kleckner 
820876330d5SReid Kleckner       // Handle the two cases we can handle: indirect in memory and in register.
821876330d5SReid Kleckner       bool IsIndirect = DVInst->getOperand(1).isImm();
822876330d5SReid Kleckner       unsigned CVReg = TRI->getCodeViewRegNum(DVInst->getOperand(0).getReg());
823876330d5SReid Kleckner       {
824876330d5SReid Kleckner         LocalVarDefRange DefRange;
825876330d5SReid Kleckner         if (IsIndirect) {
826876330d5SReid Kleckner           int64_t Offset = DVInst->getOperand(1).getImm();
827876330d5SReid Kleckner           DefRange = createDefRangeMem(CVReg, Offset);
828876330d5SReid Kleckner         } else {
829876330d5SReid Kleckner           DefRange = createDefRangeReg(CVReg);
830876330d5SReid Kleckner         }
831876330d5SReid Kleckner         if (Var.DefRanges.empty() ||
832876330d5SReid Kleckner             Var.DefRanges.back().isDifferentLocation(DefRange)) {
833876330d5SReid Kleckner           Var.DefRanges.emplace_back(std::move(DefRange));
834876330d5SReid Kleckner         }
835876330d5SReid Kleckner       }
836876330d5SReid Kleckner 
837876330d5SReid Kleckner       // Compute the label range.
838876330d5SReid Kleckner       const MCSymbol *Begin = getLabelBeforeInsn(Range.first);
839876330d5SReid Kleckner       const MCSymbol *End = getLabelAfterInsn(Range.second);
840876330d5SReid Kleckner       if (!End) {
841876330d5SReid Kleckner         if (std::next(I) != E)
842876330d5SReid Kleckner           End = getLabelBeforeInsn(std::next(I)->first);
843876330d5SReid Kleckner         else
844876330d5SReid Kleckner           End = Asm->getFunctionEnd();
845876330d5SReid Kleckner       }
846876330d5SReid Kleckner 
847876330d5SReid Kleckner       // If the last range end is our begin, just extend the last range.
848876330d5SReid Kleckner       // Otherwise make a new range.
849876330d5SReid Kleckner       SmallVectorImpl<std::pair<const MCSymbol *, const MCSymbol *>> &Ranges =
850876330d5SReid Kleckner           Var.DefRanges.back().Ranges;
851876330d5SReid Kleckner       if (!Ranges.empty() && Ranges.back().second == Begin)
852876330d5SReid Kleckner         Ranges.back().second = End;
853876330d5SReid Kleckner       else
854876330d5SReid Kleckner         Ranges.emplace_back(Begin, End);
855876330d5SReid Kleckner 
856876330d5SReid Kleckner       // FIXME: Do more range combining.
857876330d5SReid Kleckner     }
858876330d5SReid Kleckner 
859876330d5SReid Kleckner     recordLocalVariable(std::move(Var), InlinedAt);
860876330d5SReid Kleckner   }
861f9c275feSReid Kleckner }
862f9c275feSReid Kleckner 
86370f5bc99SReid Kleckner void CodeViewDebug::beginFunction(const MachineFunction *MF) {
86470f5bc99SReid Kleckner   assert(!CurFn && "Can't process two functions at once!");
86570f5bc99SReid Kleckner 
86667f684e1SDavid Majnemer   if (!Asm || !MMI->hasDebugInfo() || !MF->getFunction()->getSubprogram())
86770f5bc99SReid Kleckner     return;
86870f5bc99SReid Kleckner 
869f9c275feSReid Kleckner   DebugHandlerBase::beginFunction(MF);
870f9c275feSReid Kleckner 
87170f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
87270f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV) == false);
87370f5bc99SReid Kleckner   CurFn = &FnDebugInfo[GV];
8742214ed89SReid Kleckner   CurFn->FuncId = NextFuncId++;
8751fcd610cSReid Kleckner   CurFn->Begin = Asm->getFunctionBegin();
87670f5bc99SReid Kleckner 
877a9f4cc95SReid Kleckner   OS.EmitCVFuncIdDirective(CurFn->FuncId);
878a9f4cc95SReid Kleckner 
879f9c275feSReid Kleckner   // Find the end of the function prolog.  First known non-DBG_VALUE and
880f9c275feSReid Kleckner   // non-frame setup location marks the beginning of the function body.
88170f5bc99SReid Kleckner   // FIXME: is there a simpler a way to do this? Can we just search
88270f5bc99SReid Kleckner   // for the first instruction of the function, not the last of the prolog?
88370f5bc99SReid Kleckner   DebugLoc PrologEndLoc;
88470f5bc99SReid Kleckner   bool EmptyPrologue = true;
88570f5bc99SReid Kleckner   for (const auto &MBB : *MF) {
88670f5bc99SReid Kleckner     for (const auto &MI : MBB) {
887f9c275feSReid Kleckner       if (!MI.isDebugValue() && !MI.getFlag(MachineInstr::FrameSetup) &&
888f9c275feSReid Kleckner           MI.getDebugLoc()) {
88970f5bc99SReid Kleckner         PrologEndLoc = MI.getDebugLoc();
89070f5bc99SReid Kleckner         break;
891f9c275feSReid Kleckner       } else if (!MI.isDebugValue()) {
89270f5bc99SReid Kleckner         EmptyPrologue = false;
89370f5bc99SReid Kleckner       }
89470f5bc99SReid Kleckner     }
895f9c275feSReid Kleckner   }
896f9c275feSReid Kleckner 
89770f5bc99SReid Kleckner   // Record beginning of function if we have a non-empty prologue.
89870f5bc99SReid Kleckner   if (PrologEndLoc && !EmptyPrologue) {
89970f5bc99SReid Kleckner     DebugLoc FnStartDL = PrologEndLoc.getFnDebugLoc();
90070f5bc99SReid Kleckner     maybeRecordLocation(FnStartDL, MF);
90170f5bc99SReid Kleckner   }
90270f5bc99SReid Kleckner }
90370f5bc99SReid Kleckner 
9044b63a98dSHans Wennborg void CodeViewDebug::addToUDTs(const DIType *Ty, TypeIndex TI) {
905ad56ea31SReid Kleckner   // Don't record empty UDTs.
906ad56ea31SReid Kleckner   if (Ty->getName().empty())
907ad56ea31SReid Kleckner     return;
908ad56ea31SReid Kleckner 
9094b63a98dSHans Wennborg   SmallVector<StringRef, 5> QualifiedNameComponents;
9104b63a98dSHans Wennborg   const DISubprogram *ClosestSubprogram = getQualifiedNameComponents(
9114b63a98dSHans Wennborg       Ty->getScope().resolve(), QualifiedNameComponents);
9124b63a98dSHans Wennborg 
9134b63a98dSHans Wennborg   std::string FullyQualifiedName =
9146bdc24e7SDavid Majnemer       getQualifiedName(QualifiedNameComponents, getPrettyScopeName(Ty));
9154b63a98dSHans Wennborg 
9164b63a98dSHans Wennborg   if (ClosestSubprogram == nullptr)
9174b63a98dSHans Wennborg     GlobalUDTs.emplace_back(std::move(FullyQualifiedName), TI);
9184b63a98dSHans Wennborg   else if (ClosestSubprogram == CurrentSubprogram)
9194b63a98dSHans Wennborg     LocalUDTs.emplace_back(std::move(FullyQualifiedName), TI);
9204b63a98dSHans Wennborg 
9214b63a98dSHans Wennborg   // TODO: What if the ClosestSubprogram is neither null or the current
9224b63a98dSHans Wennborg   // subprogram?  Currently, the UDT just gets dropped on the floor.
9234b63a98dSHans Wennborg   //
9244b63a98dSHans Wennborg   // The current behavior is not desirable.  To get maximal fidelity, we would
9254b63a98dSHans Wennborg   // need to perform all type translation before beginning emission of .debug$S
9264b63a98dSHans Wennborg   // and then make LocalUDTs a member of FunctionInfo
9274b63a98dSHans Wennborg }
9284b63a98dSHans Wennborg 
92976c9eb99SAmjad Aboud TypeIndex CodeViewDebug::lowerType(const DIType *Ty, const DIType *ClassTy) {
9305acacbb0SReid Kleckner   // Generic dispatch for lowering an unknown type.
9315acacbb0SReid Kleckner   switch (Ty->getTag()) {
932f3c3c132SAdrian McCarthy   case dwarf::DW_TAG_array_type:
933f3c3c132SAdrian McCarthy     return lowerTypeArray(cast<DICompositeType>(Ty));
934d065e23dSDavid Majnemer   case dwarf::DW_TAG_typedef:
935d065e23dSDavid Majnemer     return lowerTypeAlias(cast<DIDerivedType>(Ty));
9365acacbb0SReid Kleckner   case dwarf::DW_TAG_base_type:
9375acacbb0SReid Kleckner     return lowerTypeBasic(cast<DIBasicType>(Ty));
9385acacbb0SReid Kleckner   case dwarf::DW_TAG_pointer_type:
9399dac4731SReid Kleckner     if (cast<DIDerivedType>(Ty)->getName() == "__vtbl_ptr_type")
9409dac4731SReid Kleckner       return lowerTypeVFTableShape(cast<DIDerivedType>(Ty));
9419dac4731SReid Kleckner     LLVM_FALLTHROUGH;
9425acacbb0SReid Kleckner   case dwarf::DW_TAG_reference_type:
9435acacbb0SReid Kleckner   case dwarf::DW_TAG_rvalue_reference_type:
9445acacbb0SReid Kleckner     return lowerTypePointer(cast<DIDerivedType>(Ty));
9455acacbb0SReid Kleckner   case dwarf::DW_TAG_ptr_to_member_type:
9465acacbb0SReid Kleckner     return lowerTypeMemberPointer(cast<DIDerivedType>(Ty));
9475acacbb0SReid Kleckner   case dwarf::DW_TAG_const_type:
9485acacbb0SReid Kleckner   case dwarf::DW_TAG_volatile_type:
9495acacbb0SReid Kleckner     return lowerTypeModifier(cast<DIDerivedType>(Ty));
95075c3ebfaSDavid Majnemer   case dwarf::DW_TAG_subroutine_type:
9510c5d874bSReid Kleckner     if (ClassTy) {
9520c5d874bSReid Kleckner       // The member function type of a member function pointer has no
9530c5d874bSReid Kleckner       // ThisAdjustment.
9540c5d874bSReid Kleckner       return lowerTypeMemberFunction(cast<DISubroutineType>(Ty), ClassTy,
9550c5d874bSReid Kleckner                                      /*ThisAdjustment=*/0);
9560c5d874bSReid Kleckner     }
95775c3ebfaSDavid Majnemer     return lowerTypeFunction(cast<DISubroutineType>(Ty));
958979cb888SDavid Majnemer   case dwarf::DW_TAG_enumeration_type:
959979cb888SDavid Majnemer     return lowerTypeEnum(cast<DICompositeType>(Ty));
960a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
961a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
962a8d57407SReid Kleckner     return lowerTypeClass(cast<DICompositeType>(Ty));
963a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
964a8d57407SReid Kleckner     return lowerTypeUnion(cast<DICompositeType>(Ty));
9655acacbb0SReid Kleckner   default:
9665acacbb0SReid Kleckner     // Use the null type index.
9675acacbb0SReid Kleckner     return TypeIndex();
9685acacbb0SReid Kleckner   }
9695acacbb0SReid Kleckner }
9705acacbb0SReid Kleckner 
971d065e23dSDavid Majnemer TypeIndex CodeViewDebug::lowerTypeAlias(const DIDerivedType *Ty) {
972d065e23dSDavid Majnemer   DITypeRef UnderlyingTypeRef = Ty->getBaseType();
973d065e23dSDavid Majnemer   TypeIndex UnderlyingTypeIndex = getTypeIndex(UnderlyingTypeRef);
9743128b10cSDavid Majnemer   StringRef TypeName = Ty->getName();
9753128b10cSDavid Majnemer 
9764b63a98dSHans Wennborg   addToUDTs(Ty, UnderlyingTypeIndex);
9773128b10cSDavid Majnemer 
978d065e23dSDavid Majnemer   if (UnderlyingTypeIndex == TypeIndex(SimpleTypeKind::Int32Long) &&
9793128b10cSDavid Majnemer       TypeName == "HRESULT")
980d065e23dSDavid Majnemer     return TypeIndex(SimpleTypeKind::HResult);
9818c46a4ceSDavid Majnemer   if (UnderlyingTypeIndex == TypeIndex(SimpleTypeKind::UInt16Short) &&
9823128b10cSDavid Majnemer       TypeName == "wchar_t")
9838c46a4ceSDavid Majnemer     return TypeIndex(SimpleTypeKind::WideCharacter);
9844b63a98dSHans Wennborg 
985d065e23dSDavid Majnemer   return UnderlyingTypeIndex;
986d065e23dSDavid Majnemer }
987d065e23dSDavid Majnemer 
988f3c3c132SAdrian McCarthy TypeIndex CodeViewDebug::lowerTypeArray(const DICompositeType *Ty) {
989f3c3c132SAdrian McCarthy   DITypeRef ElementTypeRef = Ty->getBaseType();
990f3c3c132SAdrian McCarthy   TypeIndex ElementTypeIndex = getTypeIndex(ElementTypeRef);
991f3c3c132SAdrian McCarthy   // IndexType is size_t, which depends on the bitness of the target.
992f3c3c132SAdrian McCarthy   TypeIndex IndexType = Asm->MAI->getPointerSize() == 8
993f3c3c132SAdrian McCarthy                             ? TypeIndex(SimpleTypeKind::UInt64Quad)
994f3c3c132SAdrian McCarthy                             : TypeIndex(SimpleTypeKind::UInt32Long);
995acee5685SAmjad Aboud 
996acee5685SAmjad Aboud   uint64_t ElementSize = getBaseTypeSize(ElementTypeRef) / 8;
997acee5685SAmjad Aboud 
998acee5685SAmjad Aboud 
9991076288eSReid Kleckner   // We want to assert that the element type multiplied by the array lengths
10001076288eSReid Kleckner   // match the size of the overall array. However, if we don't have complete
10011076288eSReid Kleckner   // type information for the base type, we can't make this assertion. This
10021076288eSReid Kleckner   // happens if limited debug info is enabled in this case:
10031076288eSReid Kleckner   //   struct VTableOptzn { VTableOptzn(); virtual ~VTableOptzn(); };
10041076288eSReid Kleckner   //   VTableOptzn array[3];
10051076288eSReid Kleckner   // The DICompositeType of VTableOptzn will have size zero, and the array will
10061076288eSReid Kleckner   // have size 3 * sizeof(void*), and we should avoid asserting.
10071076288eSReid Kleckner   //
10081076288eSReid Kleckner   // There is a related bug in the front-end where an array of a structure,
10091076288eSReid Kleckner   // which was declared as incomplete structure first, ends up not getting a
10101076288eSReid Kleckner   // size assigned to it. (PR28303)
1011acee5685SAmjad Aboud   // Example:
1012acee5685SAmjad Aboud   //   struct A(*p)[3];
1013acee5685SAmjad Aboud   //   struct A { int f; } a[3];
10141076288eSReid Kleckner   bool PartiallyIncomplete = false;
10151076288eSReid Kleckner   if (Ty->getSizeInBits() == 0 || ElementSize == 0) {
10161076288eSReid Kleckner     PartiallyIncomplete = true;
1017acee5685SAmjad Aboud   }
1018acee5685SAmjad Aboud 
1019acee5685SAmjad Aboud   // Add subranges to array type.
1020acee5685SAmjad Aboud   DINodeArray Elements = Ty->getElements();
1021acee5685SAmjad Aboud   for (int i = Elements.size() - 1; i >= 0; --i) {
1022acee5685SAmjad Aboud     const DINode *Element = Elements[i];
1023acee5685SAmjad Aboud     assert(Element->getTag() == dwarf::DW_TAG_subrange_type);
1024acee5685SAmjad Aboud 
1025acee5685SAmjad Aboud     const DISubrange *Subrange = cast<DISubrange>(Element);
1026acee5685SAmjad Aboud     assert(Subrange->getLowerBound() == 0 &&
1027acee5685SAmjad Aboud            "codeview doesn't support subranges with lower bounds");
1028acee5685SAmjad Aboud     int64_t Count = Subrange->getCount();
1029acee5685SAmjad Aboud 
1030acee5685SAmjad Aboud     // Variable Length Array (VLA) has Count equal to '-1'.
1031acee5685SAmjad Aboud     // Replace with Count '1', assume it is the minimum VLA length.
1032acee5685SAmjad Aboud     // FIXME: Make front-end support VLA subrange and emit LF_DIMVARLU.
1033acee5685SAmjad Aboud     if (Count == -1) {
1034acee5685SAmjad Aboud       Count = 1;
10351076288eSReid Kleckner       PartiallyIncomplete = true;
1036acee5685SAmjad Aboud     }
1037acee5685SAmjad Aboud 
1038acee5685SAmjad Aboud     // Update the element size and element type index for subsequent subranges.
1039acee5685SAmjad Aboud     ElementSize *= Count;
10401076288eSReid Kleckner 
10411076288eSReid Kleckner     // If this is the outermost array, use the size from the array. It will be
10421076288eSReid Kleckner     // more accurate if PartiallyIncomplete is true.
10431076288eSReid Kleckner     uint64_t ArraySize =
10441076288eSReid Kleckner         (i == 0 && ElementSize == 0) ? Ty->getSizeInBits() / 8 : ElementSize;
10451076288eSReid Kleckner 
10461076288eSReid Kleckner     StringRef Name = (i == 0) ? Ty->getName() : "";
10475e3e4bb2SZachary Turner     ElementTypeIndex = TypeTable.writeKnownType(
10481076288eSReid Kleckner         ArrayRecord(ElementTypeIndex, IndexType, ArraySize, Name));
1049acee5685SAmjad Aboud   }
1050acee5685SAmjad Aboud 
10511076288eSReid Kleckner   (void)PartiallyIncomplete;
10521076288eSReid Kleckner   assert(PartiallyIncomplete || ElementSize == (Ty->getSizeInBits() / 8));
1053acee5685SAmjad Aboud 
1054acee5685SAmjad Aboud   return ElementTypeIndex;
1055f3c3c132SAdrian McCarthy }
1056f3c3c132SAdrian McCarthy 
10575acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeBasic(const DIBasicType *Ty) {
10585acacbb0SReid Kleckner   TypeIndex Index;
10595acacbb0SReid Kleckner   dwarf::TypeKind Kind;
10605acacbb0SReid Kleckner   uint32_t ByteSize;
10615acacbb0SReid Kleckner 
10625acacbb0SReid Kleckner   Kind = static_cast<dwarf::TypeKind>(Ty->getEncoding());
1063afefa673SDavid Majnemer   ByteSize = Ty->getSizeInBits() / 8;
10645acacbb0SReid Kleckner 
10655acacbb0SReid Kleckner   SimpleTypeKind STK = SimpleTypeKind::None;
10665acacbb0SReid Kleckner   switch (Kind) {
10675acacbb0SReid Kleckner   case dwarf::DW_ATE_address:
10685acacbb0SReid Kleckner     // FIXME: Translate
10695acacbb0SReid Kleckner     break;
10705acacbb0SReid Kleckner   case dwarf::DW_ATE_boolean:
10715acacbb0SReid Kleckner     switch (ByteSize) {
10725acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::Boolean8;   break;
10735acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::Boolean16;  break;
10745acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Boolean32;  break;
10755acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Boolean64;  break;
10761c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::Boolean128; break;
10775acacbb0SReid Kleckner     }
10785acacbb0SReid Kleckner     break;
10795acacbb0SReid Kleckner   case dwarf::DW_ATE_complex_float:
10805acacbb0SReid Kleckner     switch (ByteSize) {
10811c2cb1ddSDavid Majnemer     case 2:  STK = SimpleTypeKind::Complex16;  break;
10825acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Complex32;  break;
10835acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Complex64;  break;
10845acacbb0SReid Kleckner     case 10: STK = SimpleTypeKind::Complex80;  break;
10855acacbb0SReid Kleckner     case 16: STK = SimpleTypeKind::Complex128; break;
10865acacbb0SReid Kleckner     }
10875acacbb0SReid Kleckner     break;
10885acacbb0SReid Kleckner   case dwarf::DW_ATE_float:
10895acacbb0SReid Kleckner     switch (ByteSize) {
10901c2cb1ddSDavid Majnemer     case 2:  STK = SimpleTypeKind::Float16;  break;
10915acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Float32;  break;
10925acacbb0SReid Kleckner     case 6:  STK = SimpleTypeKind::Float48;  break;
10935acacbb0SReid Kleckner     case 8:  STK = SimpleTypeKind::Float64;  break;
10945acacbb0SReid Kleckner     case 10: STK = SimpleTypeKind::Float80;  break;
10955acacbb0SReid Kleckner     case 16: STK = SimpleTypeKind::Float128; break;
10965acacbb0SReid Kleckner     }
10975acacbb0SReid Kleckner     break;
10985acacbb0SReid Kleckner   case dwarf::DW_ATE_signed:
10995acacbb0SReid Kleckner     switch (ByteSize) {
11005acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::SByte;      break;
11015acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::Int16Short; break;
11025acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::Int32;      break;
11031c2cb1ddSDavid Majnemer     case 8:  STK = SimpleTypeKind::Int64Quad;  break;
11041c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::Int128Oct;  break;
11055acacbb0SReid Kleckner     }
11065acacbb0SReid Kleckner     break;
11075acacbb0SReid Kleckner   case dwarf::DW_ATE_unsigned:
11085acacbb0SReid Kleckner     switch (ByteSize) {
11095acacbb0SReid Kleckner     case 1:  STK = SimpleTypeKind::Byte;        break;
11105acacbb0SReid Kleckner     case 2:  STK = SimpleTypeKind::UInt16Short; break;
11115acacbb0SReid Kleckner     case 4:  STK = SimpleTypeKind::UInt32;      break;
11121c2cb1ddSDavid Majnemer     case 8:  STK = SimpleTypeKind::UInt64Quad;  break;
11131c2cb1ddSDavid Majnemer     case 16: STK = SimpleTypeKind::UInt128Oct;  break;
11145acacbb0SReid Kleckner     }
11155acacbb0SReid Kleckner     break;
11165acacbb0SReid Kleckner   case dwarf::DW_ATE_UTF:
11175acacbb0SReid Kleckner     switch (ByteSize) {
11185acacbb0SReid Kleckner     case 2: STK = SimpleTypeKind::Character16; break;
11195acacbb0SReid Kleckner     case 4: STK = SimpleTypeKind::Character32; break;
11205acacbb0SReid Kleckner     }
11215acacbb0SReid Kleckner     break;
11225acacbb0SReid Kleckner   case dwarf::DW_ATE_signed_char:
11235acacbb0SReid Kleckner     if (ByteSize == 1)
11245acacbb0SReid Kleckner       STK = SimpleTypeKind::SignedCharacter;
11255acacbb0SReid Kleckner     break;
11265acacbb0SReid Kleckner   case dwarf::DW_ATE_unsigned_char:
11275acacbb0SReid Kleckner     if (ByteSize == 1)
11285acacbb0SReid Kleckner       STK = SimpleTypeKind::UnsignedCharacter;
11295acacbb0SReid Kleckner     break;
11305acacbb0SReid Kleckner   default:
11315acacbb0SReid Kleckner     break;
11325acacbb0SReid Kleckner   }
11335acacbb0SReid Kleckner 
11345acacbb0SReid Kleckner   // Apply some fixups based on the source-level type name.
11355acacbb0SReid Kleckner   if (STK == SimpleTypeKind::Int32 && Ty->getName() == "long int")
11365acacbb0SReid Kleckner     STK = SimpleTypeKind::Int32Long;
11375acacbb0SReid Kleckner   if (STK == SimpleTypeKind::UInt32 && Ty->getName() == "long unsigned int")
11385acacbb0SReid Kleckner     STK = SimpleTypeKind::UInt32Long;
11398c46a4ceSDavid Majnemer   if (STK == SimpleTypeKind::UInt16Short &&
11408c46a4ceSDavid Majnemer       (Ty->getName() == "wchar_t" || Ty->getName() == "__wchar_t"))
11415acacbb0SReid Kleckner     STK = SimpleTypeKind::WideCharacter;
11425acacbb0SReid Kleckner   if ((STK == SimpleTypeKind::SignedCharacter ||
11435acacbb0SReid Kleckner        STK == SimpleTypeKind::UnsignedCharacter) &&
11445acacbb0SReid Kleckner       Ty->getName() == "char")
11455acacbb0SReid Kleckner     STK = SimpleTypeKind::NarrowCharacter;
11465acacbb0SReid Kleckner 
11475acacbb0SReid Kleckner   return TypeIndex(STK);
11485acacbb0SReid Kleckner }
11495acacbb0SReid Kleckner 
11505acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypePointer(const DIDerivedType *Ty) {
11515acacbb0SReid Kleckner   TypeIndex PointeeTI = getTypeIndex(Ty->getBaseType());
11525acacbb0SReid Kleckner 
11535acacbb0SReid Kleckner   // Pointers to simple types can use SimpleTypeMode, rather than having a
11545acacbb0SReid Kleckner   // dedicated pointer type record.
11555acacbb0SReid Kleckner   if (PointeeTI.isSimple() &&
11565acacbb0SReid Kleckner       PointeeTI.getSimpleMode() == SimpleTypeMode::Direct &&
11575acacbb0SReid Kleckner       Ty->getTag() == dwarf::DW_TAG_pointer_type) {
11585acacbb0SReid Kleckner     SimpleTypeMode Mode = Ty->getSizeInBits() == 64
11595acacbb0SReid Kleckner                               ? SimpleTypeMode::NearPointer64
11605acacbb0SReid Kleckner                               : SimpleTypeMode::NearPointer32;
11615acacbb0SReid Kleckner     return TypeIndex(PointeeTI.getSimpleKind(), Mode);
11625acacbb0SReid Kleckner   }
11635acacbb0SReid Kleckner 
11645acacbb0SReid Kleckner   PointerKind PK =
11655acacbb0SReid Kleckner       Ty->getSizeInBits() == 64 ? PointerKind::Near64 : PointerKind::Near32;
11665acacbb0SReid Kleckner   PointerMode PM = PointerMode::Pointer;
11675acacbb0SReid Kleckner   switch (Ty->getTag()) {
11685acacbb0SReid Kleckner   default: llvm_unreachable("not a pointer tag type");
11695acacbb0SReid Kleckner   case dwarf::DW_TAG_pointer_type:
11705acacbb0SReid Kleckner     PM = PointerMode::Pointer;
11715acacbb0SReid Kleckner     break;
11725acacbb0SReid Kleckner   case dwarf::DW_TAG_reference_type:
11735acacbb0SReid Kleckner     PM = PointerMode::LValueReference;
11745acacbb0SReid Kleckner     break;
11755acacbb0SReid Kleckner   case dwarf::DW_TAG_rvalue_reference_type:
11765acacbb0SReid Kleckner     PM = PointerMode::RValueReference;
11775acacbb0SReid Kleckner     break;
11785acacbb0SReid Kleckner   }
11795acacbb0SReid Kleckner   // FIXME: MSVC folds qualifiers into PointerOptions in the context of a method
11805acacbb0SReid Kleckner   // 'this' pointer, but not normal contexts. Figure out what we're supposed to
11815acacbb0SReid Kleckner   // do.
11825acacbb0SReid Kleckner   PointerOptions PO = PointerOptions::None;
11835acacbb0SReid Kleckner   PointerRecord PR(PointeeTI, PK, PM, PO, Ty->getSizeInBits() / 8);
11845e3e4bb2SZachary Turner   return TypeTable.writeKnownType(PR);
11855acacbb0SReid Kleckner }
11865acacbb0SReid Kleckner 
11876fa1546aSReid Kleckner static PointerToMemberRepresentation
11886fa1546aSReid Kleckner translatePtrToMemberRep(unsigned SizeInBytes, bool IsPMF, unsigned Flags) {
11896fa1546aSReid Kleckner   // SizeInBytes being zero generally implies that the member pointer type was
11906fa1546aSReid Kleckner   // incomplete, which can happen if it is part of a function prototype. In this
11916fa1546aSReid Kleckner   // case, use the unknown model instead of the general model.
1192604105bbSReid Kleckner   if (IsPMF) {
1193604105bbSReid Kleckner     switch (Flags & DINode::FlagPtrToMemberRep) {
1194604105bbSReid Kleckner     case 0:
11956fa1546aSReid Kleckner       return SizeInBytes == 0 ? PointerToMemberRepresentation::Unknown
11966fa1546aSReid Kleckner                               : PointerToMemberRepresentation::GeneralFunction;
1197604105bbSReid Kleckner     case DINode::FlagSingleInheritance:
1198604105bbSReid Kleckner       return PointerToMemberRepresentation::SingleInheritanceFunction;
1199604105bbSReid Kleckner     case DINode::FlagMultipleInheritance:
1200604105bbSReid Kleckner       return PointerToMemberRepresentation::MultipleInheritanceFunction;
1201604105bbSReid Kleckner     case DINode::FlagVirtualInheritance:
1202604105bbSReid Kleckner       return PointerToMemberRepresentation::VirtualInheritanceFunction;
1203604105bbSReid Kleckner     }
1204604105bbSReid Kleckner   } else {
1205604105bbSReid Kleckner     switch (Flags & DINode::FlagPtrToMemberRep) {
1206604105bbSReid Kleckner     case 0:
12076fa1546aSReid Kleckner       return SizeInBytes == 0 ? PointerToMemberRepresentation::Unknown
12086fa1546aSReid Kleckner                               : PointerToMemberRepresentation::GeneralData;
1209604105bbSReid Kleckner     case DINode::FlagSingleInheritance:
1210604105bbSReid Kleckner       return PointerToMemberRepresentation::SingleInheritanceData;
1211604105bbSReid Kleckner     case DINode::FlagMultipleInheritance:
1212604105bbSReid Kleckner       return PointerToMemberRepresentation::MultipleInheritanceData;
1213604105bbSReid Kleckner     case DINode::FlagVirtualInheritance:
1214604105bbSReid Kleckner       return PointerToMemberRepresentation::VirtualInheritanceData;
1215604105bbSReid Kleckner     }
1216604105bbSReid Kleckner   }
1217604105bbSReid Kleckner   llvm_unreachable("invalid ptr to member representation");
1218604105bbSReid Kleckner }
1219604105bbSReid Kleckner 
12205acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeMemberPointer(const DIDerivedType *Ty) {
12215acacbb0SReid Kleckner   assert(Ty->getTag() == dwarf::DW_TAG_ptr_to_member_type);
12225acacbb0SReid Kleckner   TypeIndex ClassTI = getTypeIndex(Ty->getClassType());
122376c9eb99SAmjad Aboud   TypeIndex PointeeTI = getTypeIndex(Ty->getBaseType(), Ty->getClassType());
12245acacbb0SReid Kleckner   PointerKind PK = Asm->MAI->getPointerSize() == 8 ? PointerKind::Near64
12255acacbb0SReid Kleckner                                                    : PointerKind::Near32;
1226604105bbSReid Kleckner   bool IsPMF = isa<DISubroutineType>(Ty->getBaseType());
1227604105bbSReid Kleckner   PointerMode PM = IsPMF ? PointerMode::PointerToMemberFunction
12285acacbb0SReid Kleckner                          : PointerMode::PointerToDataMember;
12295acacbb0SReid Kleckner   PointerOptions PO = PointerOptions::None; // FIXME
12306fa1546aSReid Kleckner   assert(Ty->getSizeInBits() / 8 <= 0xff && "pointer size too big");
12316fa1546aSReid Kleckner   uint8_t SizeInBytes = Ty->getSizeInBits() / 8;
12326fa1546aSReid Kleckner   MemberPointerInfo MPI(
12336fa1546aSReid Kleckner       ClassTI, translatePtrToMemberRep(SizeInBytes, IsPMF, Ty->getFlags()));
1234604105bbSReid Kleckner   PointerRecord PR(PointeeTI, PK, PM, PO, SizeInBytes, MPI);
12355e3e4bb2SZachary Turner   return TypeTable.writeKnownType(PR);
12365acacbb0SReid Kleckner }
12375acacbb0SReid Kleckner 
1238de3d8b50SReid Kleckner /// Given a DWARF calling convention, get the CodeView equivalent. If we don't
1239de3d8b50SReid Kleckner /// have a translation, use the NearC convention.
1240de3d8b50SReid Kleckner static CallingConvention dwarfCCToCodeView(unsigned DwarfCC) {
1241de3d8b50SReid Kleckner   switch (DwarfCC) {
1242de3d8b50SReid Kleckner   case dwarf::DW_CC_normal:             return CallingConvention::NearC;
1243de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_msfastcall: return CallingConvention::NearFast;
1244de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_thiscall:   return CallingConvention::ThisCall;
1245de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_stdcall:    return CallingConvention::NearStdCall;
1246de3d8b50SReid Kleckner   case dwarf::DW_CC_BORLAND_pascal:     return CallingConvention::NearPascal;
1247de3d8b50SReid Kleckner   case dwarf::DW_CC_LLVM_vectorcall:    return CallingConvention::NearVector;
1248de3d8b50SReid Kleckner   }
1249de3d8b50SReid Kleckner   return CallingConvention::NearC;
1250de3d8b50SReid Kleckner }
1251de3d8b50SReid Kleckner 
12525acacbb0SReid Kleckner TypeIndex CodeViewDebug::lowerTypeModifier(const DIDerivedType *Ty) {
12535acacbb0SReid Kleckner   ModifierOptions Mods = ModifierOptions::None;
12545acacbb0SReid Kleckner   bool IsModifier = true;
12555acacbb0SReid Kleckner   const DIType *BaseTy = Ty;
1256b9c80fd8SReid Kleckner   while (IsModifier && BaseTy) {
12575acacbb0SReid Kleckner     // FIXME: Need to add DWARF tag for __unaligned.
12585acacbb0SReid Kleckner     switch (BaseTy->getTag()) {
12595acacbb0SReid Kleckner     case dwarf::DW_TAG_const_type:
12605acacbb0SReid Kleckner       Mods |= ModifierOptions::Const;
12615acacbb0SReid Kleckner       break;
12625acacbb0SReid Kleckner     case dwarf::DW_TAG_volatile_type:
12635acacbb0SReid Kleckner       Mods |= ModifierOptions::Volatile;
12645acacbb0SReid Kleckner       break;
12655acacbb0SReid Kleckner     default:
12665acacbb0SReid Kleckner       IsModifier = false;
12675acacbb0SReid Kleckner       break;
12685acacbb0SReid Kleckner     }
12695acacbb0SReid Kleckner     if (IsModifier)
12705acacbb0SReid Kleckner       BaseTy = cast<DIDerivedType>(BaseTy)->getBaseType().resolve();
12715acacbb0SReid Kleckner   }
12725acacbb0SReid Kleckner   TypeIndex ModifiedTI = getTypeIndex(BaseTy);
1273dbaa61cbSReid Kleckner   return TypeTable.writeKnownType(ModifierRecord(ModifiedTI, Mods));
12745acacbb0SReid Kleckner }
12755acacbb0SReid Kleckner 
127675c3ebfaSDavid Majnemer TypeIndex CodeViewDebug::lowerTypeFunction(const DISubroutineType *Ty) {
127775c3ebfaSDavid Majnemer   SmallVector<TypeIndex, 8> ReturnAndArgTypeIndices;
127875c3ebfaSDavid Majnemer   for (DITypeRef ArgTypeRef : Ty->getTypeArray())
127975c3ebfaSDavid Majnemer     ReturnAndArgTypeIndices.push_back(getTypeIndex(ArgTypeRef));
128075c3ebfaSDavid Majnemer 
128175c3ebfaSDavid Majnemer   TypeIndex ReturnTypeIndex = TypeIndex::Void();
128275c3ebfaSDavid Majnemer   ArrayRef<TypeIndex> ArgTypeIndices = None;
128375c3ebfaSDavid Majnemer   if (!ReturnAndArgTypeIndices.empty()) {
128475c3ebfaSDavid Majnemer     auto ReturnAndArgTypesRef = makeArrayRef(ReturnAndArgTypeIndices);
128575c3ebfaSDavid Majnemer     ReturnTypeIndex = ReturnAndArgTypesRef.front();
128675c3ebfaSDavid Majnemer     ArgTypeIndices = ReturnAndArgTypesRef.drop_front();
128775c3ebfaSDavid Majnemer   }
128875c3ebfaSDavid Majnemer 
128975c3ebfaSDavid Majnemer   ArgListRecord ArgListRec(TypeRecordKind::ArgList, ArgTypeIndices);
12905e3e4bb2SZachary Turner   TypeIndex ArgListIndex = TypeTable.writeKnownType(ArgListRec);
129175c3ebfaSDavid Majnemer 
1292de3d8b50SReid Kleckner   CallingConvention CC = dwarfCCToCodeView(Ty->getCC());
1293de3d8b50SReid Kleckner 
1294de3d8b50SReid Kleckner   ProcedureRecord Procedure(ReturnTypeIndex, CC, FunctionOptions::None,
1295de3d8b50SReid Kleckner                             ArgTypeIndices.size(), ArgListIndex);
12965e3e4bb2SZachary Turner   return TypeTable.writeKnownType(Procedure);
129775c3ebfaSDavid Majnemer }
129875c3ebfaSDavid Majnemer 
129976c9eb99SAmjad Aboud TypeIndex CodeViewDebug::lowerTypeMemberFunction(const DISubroutineType *Ty,
13000c5d874bSReid Kleckner                                                  const DIType *ClassTy,
13010c5d874bSReid Kleckner                                                  int ThisAdjustment) {
130276c9eb99SAmjad Aboud   // Lower the containing class type.
130376c9eb99SAmjad Aboud   TypeIndex ClassType = getTypeIndex(ClassTy);
130476c9eb99SAmjad Aboud 
130576c9eb99SAmjad Aboud   SmallVector<TypeIndex, 8> ReturnAndArgTypeIndices;
130676c9eb99SAmjad Aboud   for (DITypeRef ArgTypeRef : Ty->getTypeArray())
130776c9eb99SAmjad Aboud     ReturnAndArgTypeIndices.push_back(getTypeIndex(ArgTypeRef));
130876c9eb99SAmjad Aboud 
130976c9eb99SAmjad Aboud   TypeIndex ReturnTypeIndex = TypeIndex::Void();
131076c9eb99SAmjad Aboud   ArrayRef<TypeIndex> ArgTypeIndices = None;
131176c9eb99SAmjad Aboud   if (!ReturnAndArgTypeIndices.empty()) {
131276c9eb99SAmjad Aboud     auto ReturnAndArgTypesRef = makeArrayRef(ReturnAndArgTypeIndices);
131376c9eb99SAmjad Aboud     ReturnTypeIndex = ReturnAndArgTypesRef.front();
131476c9eb99SAmjad Aboud     ArgTypeIndices = ReturnAndArgTypesRef.drop_front();
131576c9eb99SAmjad Aboud   }
131676c9eb99SAmjad Aboud   TypeIndex ThisTypeIndex = TypeIndex::Void();
131776c9eb99SAmjad Aboud   if (!ArgTypeIndices.empty()) {
131876c9eb99SAmjad Aboud     ThisTypeIndex = ArgTypeIndices.front();
131976c9eb99SAmjad Aboud     ArgTypeIndices = ArgTypeIndices.drop_front();
132076c9eb99SAmjad Aboud   }
132176c9eb99SAmjad Aboud 
132276c9eb99SAmjad Aboud   ArgListRecord ArgListRec(TypeRecordKind::ArgList, ArgTypeIndices);
13235e3e4bb2SZachary Turner   TypeIndex ArgListIndex = TypeTable.writeKnownType(ArgListRec);
132476c9eb99SAmjad Aboud 
132576c9eb99SAmjad Aboud   CallingConvention CC = dwarfCCToCodeView(Ty->getCC());
132676c9eb99SAmjad Aboud 
132776c9eb99SAmjad Aboud   // TODO: Need to use the correct values for:
132876c9eb99SAmjad Aboud   //       FunctionOptions
132976c9eb99SAmjad Aboud   //       ThisPointerAdjustment.
13305e3e4bb2SZachary Turner   TypeIndex TI = TypeTable.writeKnownType(MemberFunctionRecord(
133176c9eb99SAmjad Aboud       ReturnTypeIndex, ClassType, ThisTypeIndex, CC, FunctionOptions::None,
13320c5d874bSReid Kleckner       ArgTypeIndices.size(), ArgListIndex, ThisAdjustment));
133376c9eb99SAmjad Aboud 
133476c9eb99SAmjad Aboud   return TI;
133576c9eb99SAmjad Aboud }
133676c9eb99SAmjad Aboud 
13379dac4731SReid Kleckner TypeIndex CodeViewDebug::lowerTypeVFTableShape(const DIDerivedType *Ty) {
13389dac4731SReid Kleckner   unsigned VSlotCount = Ty->getSizeInBits() / (8 * Asm->MAI->getPointerSize());
13399dac4731SReid Kleckner   SmallVector<VFTableSlotKind, 4> Slots(VSlotCount, VFTableSlotKind::Near);
13409dac4731SReid Kleckner   return TypeTable.writeKnownType(VFTableShapeRecord(Slots));
13419dac4731SReid Kleckner }
13429dac4731SReid Kleckner 
134376c9eb99SAmjad Aboud static MemberAccess translateAccessFlags(unsigned RecordTag, unsigned Flags) {
134476c9eb99SAmjad Aboud   switch (Flags & DINode::FlagAccessibility) {
1345a8d57407SReid Kleckner   case DINode::FlagPrivate:   return MemberAccess::Private;
1346a8d57407SReid Kleckner   case DINode::FlagPublic:    return MemberAccess::Public;
1347a8d57407SReid Kleckner   case DINode::FlagProtected: return MemberAccess::Protected;
1348a8d57407SReid Kleckner   case 0:
1349a8d57407SReid Kleckner     // If there was no explicit access control, provide the default for the tag.
1350a8d57407SReid Kleckner     return RecordTag == dwarf::DW_TAG_class_type ? MemberAccess::Private
1351a8d57407SReid Kleckner                                                  : MemberAccess::Public;
1352a8d57407SReid Kleckner   }
1353a8d57407SReid Kleckner   llvm_unreachable("access flags are exclusive");
1354a8d57407SReid Kleckner }
1355a8d57407SReid Kleckner 
135676c9eb99SAmjad Aboud static MethodOptions translateMethodOptionFlags(const DISubprogram *SP) {
135776c9eb99SAmjad Aboud   if (SP->isArtificial())
135876c9eb99SAmjad Aboud     return MethodOptions::CompilerGenerated;
135976c9eb99SAmjad Aboud 
136076c9eb99SAmjad Aboud   // FIXME: Handle other MethodOptions.
136176c9eb99SAmjad Aboud 
136276c9eb99SAmjad Aboud   return MethodOptions::None;
136376c9eb99SAmjad Aboud }
136476c9eb99SAmjad Aboud 
136576c9eb99SAmjad Aboud static MethodKind translateMethodKindFlags(const DISubprogram *SP,
136676c9eb99SAmjad Aboud                                            bool Introduced) {
136776c9eb99SAmjad Aboud   switch (SP->getVirtuality()) {
136876c9eb99SAmjad Aboud   case dwarf::DW_VIRTUALITY_none:
136976c9eb99SAmjad Aboud     break;
137076c9eb99SAmjad Aboud   case dwarf::DW_VIRTUALITY_virtual:
137176c9eb99SAmjad Aboud     return Introduced ? MethodKind::IntroducingVirtual : MethodKind::Virtual;
137276c9eb99SAmjad Aboud   case dwarf::DW_VIRTUALITY_pure_virtual:
137376c9eb99SAmjad Aboud     return Introduced ? MethodKind::PureIntroducingVirtual
137476c9eb99SAmjad Aboud                       : MethodKind::PureVirtual;
137576c9eb99SAmjad Aboud   default:
137676c9eb99SAmjad Aboud     llvm_unreachable("unhandled virtuality case");
137776c9eb99SAmjad Aboud   }
137876c9eb99SAmjad Aboud 
137976c9eb99SAmjad Aboud   // FIXME: Get Clang to mark DISubprogram as static and do something with it.
138076c9eb99SAmjad Aboud 
138176c9eb99SAmjad Aboud   return MethodKind::Vanilla;
138276c9eb99SAmjad Aboud }
138376c9eb99SAmjad Aboud 
1384a8d57407SReid Kleckner static TypeRecordKind getRecordKind(const DICompositeType *Ty) {
1385a8d57407SReid Kleckner   switch (Ty->getTag()) {
1386a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:     return TypeRecordKind::Class;
1387a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type: return TypeRecordKind::Struct;
1388a8d57407SReid Kleckner   }
1389a8d57407SReid Kleckner   llvm_unreachable("unexpected tag");
1390a8d57407SReid Kleckner }
1391a8d57407SReid Kleckner 
1392e092dad7SReid Kleckner /// Return ClassOptions that should be present on both the forward declaration
1393e092dad7SReid Kleckner /// and the defintion of a tag type.
1394e092dad7SReid Kleckner static ClassOptions getCommonClassOptions(const DICompositeType *Ty) {
1395e092dad7SReid Kleckner   ClassOptions CO = ClassOptions::None;
1396e092dad7SReid Kleckner 
1397e092dad7SReid Kleckner   // MSVC always sets this flag, even for local types. Clang doesn't always
1398a8d57407SReid Kleckner   // appear to give every type a linkage name, which may be problematic for us.
1399a8d57407SReid Kleckner   // FIXME: Investigate the consequences of not following them here.
1400e092dad7SReid Kleckner   if (!Ty->getIdentifier().empty())
1401e092dad7SReid Kleckner     CO |= ClassOptions::HasUniqueName;
1402e092dad7SReid Kleckner 
1403e092dad7SReid Kleckner   // Put the Nested flag on a type if it appears immediately inside a tag type.
1404e092dad7SReid Kleckner   // Do not walk the scope chain. Do not attempt to compute ContainsNestedClass
1405e092dad7SReid Kleckner   // here. That flag is only set on definitions, and not forward declarations.
1406e092dad7SReid Kleckner   const DIScope *ImmediateScope = Ty->getScope().resolve();
1407e092dad7SReid Kleckner   if (ImmediateScope && isa<DICompositeType>(ImmediateScope))
1408e092dad7SReid Kleckner     CO |= ClassOptions::Nested;
1409e092dad7SReid Kleckner 
1410e092dad7SReid Kleckner   // Put the Scoped flag on function-local types.
1411e092dad7SReid Kleckner   for (const DIScope *Scope = ImmediateScope; Scope != nullptr;
1412e092dad7SReid Kleckner        Scope = Scope->getScope().resolve()) {
1413e092dad7SReid Kleckner     if (isa<DISubprogram>(Scope)) {
1414e092dad7SReid Kleckner       CO |= ClassOptions::Scoped;
1415e092dad7SReid Kleckner       break;
1416e092dad7SReid Kleckner     }
1417e092dad7SReid Kleckner   }
1418e092dad7SReid Kleckner 
1419e092dad7SReid Kleckner   return CO;
1420a8d57407SReid Kleckner }
1421a8d57407SReid Kleckner 
1422979cb888SDavid Majnemer TypeIndex CodeViewDebug::lowerTypeEnum(const DICompositeType *Ty) {
1423e092dad7SReid Kleckner   ClassOptions CO = getCommonClassOptions(Ty);
1424979cb888SDavid Majnemer   TypeIndex FTI;
1425da9548f9SDavid Majnemer   unsigned EnumeratorCount = 0;
1426979cb888SDavid Majnemer 
1427da9548f9SDavid Majnemer   if (Ty->isForwardDecl()) {
1428979cb888SDavid Majnemer     CO |= ClassOptions::ForwardReference;
1429da9548f9SDavid Majnemer   } else {
1430da9548f9SDavid Majnemer     FieldListRecordBuilder Fields;
1431da9548f9SDavid Majnemer     for (const DINode *Element : Ty->getElements()) {
1432da9548f9SDavid Majnemer       // We assume that the frontend provides all members in source declaration
1433da9548f9SDavid Majnemer       // order, which is what MSVC does.
1434da9548f9SDavid Majnemer       if (auto *Enumerator = dyn_cast_or_null<DIEnumerator>(Element)) {
14355e3e4bb2SZachary Turner         Fields.writeMemberType(EnumeratorRecord(
1436da9548f9SDavid Majnemer             MemberAccess::Public, APSInt::getUnsigned(Enumerator->getValue()),
1437da9548f9SDavid Majnemer             Enumerator->getName()));
1438da9548f9SDavid Majnemer         EnumeratorCount++;
1439da9548f9SDavid Majnemer       }
1440da9548f9SDavid Majnemer     }
1441da9548f9SDavid Majnemer     FTI = TypeTable.writeFieldList(Fields);
1442da9548f9SDavid Majnemer   }
1443979cb888SDavid Majnemer 
14446bdc24e7SDavid Majnemer   std::string FullName = getFullyQualifiedName(Ty);
14450c5d874bSReid Kleckner 
14465e3e4bb2SZachary Turner   return TypeTable.writeKnownType(EnumRecord(EnumeratorCount, CO, FTI, FullName,
1447979cb888SDavid Majnemer                                              Ty->getIdentifier(),
1448979cb888SDavid Majnemer                                              getTypeIndex(Ty->getBaseType())));
1449979cb888SDavid Majnemer }
1450979cb888SDavid Majnemer 
145176c9eb99SAmjad Aboud //===----------------------------------------------------------------------===//
145276c9eb99SAmjad Aboud // ClassInfo
145376c9eb99SAmjad Aboud //===----------------------------------------------------------------------===//
145476c9eb99SAmjad Aboud 
145576c9eb99SAmjad Aboud struct llvm::ClassInfo {
145676c9eb99SAmjad Aboud   struct MemberInfo {
145776c9eb99SAmjad Aboud     const DIDerivedType *MemberTypeNode;
145808bd744cSDavid Majnemer     uint64_t BaseOffset;
145976c9eb99SAmjad Aboud   };
146076c9eb99SAmjad Aboud   // [MemberInfo]
146176c9eb99SAmjad Aboud   typedef std::vector<MemberInfo> MemberList;
146276c9eb99SAmjad Aboud 
1463156a7239SReid Kleckner   typedef TinyPtrVector<const DISubprogram *> MethodsList;
146476c9eb99SAmjad Aboud   // MethodName -> MethodsList
146576c9eb99SAmjad Aboud   typedef MapVector<MDString *, MethodsList> MethodsMap;
146676c9eb99SAmjad Aboud 
14679f7f3e1eSReid Kleckner   /// Base classes.
14689f7f3e1eSReid Kleckner   std::vector<const DIDerivedType *> Inheritance;
14699f7f3e1eSReid Kleckner 
147076c9eb99SAmjad Aboud   /// Direct members.
147176c9eb99SAmjad Aboud   MemberList Members;
147276c9eb99SAmjad Aboud   // Direct overloaded methods gathered by name.
147376c9eb99SAmjad Aboud   MethodsMap Methods;
1474820ca540SAdrian McCarthy 
14759dac4731SReid Kleckner   TypeIndex VShapeTI;
14769dac4731SReid Kleckner 
1477820ca540SAdrian McCarthy   std::vector<const DICompositeType *> NestedClasses;
147876c9eb99SAmjad Aboud };
147976c9eb99SAmjad Aboud 
148076c9eb99SAmjad Aboud void CodeViewDebug::clear() {
148176c9eb99SAmjad Aboud   assert(CurFn == nullptr);
148276c9eb99SAmjad Aboud   FileIdMap.clear();
148376c9eb99SAmjad Aboud   FnDebugInfo.clear();
148476c9eb99SAmjad Aboud   FileToFilepathMap.clear();
148576c9eb99SAmjad Aboud   LocalUDTs.clear();
148676c9eb99SAmjad Aboud   GlobalUDTs.clear();
148776c9eb99SAmjad Aboud   TypeIndices.clear();
148876c9eb99SAmjad Aboud   CompleteTypeIndices.clear();
148976c9eb99SAmjad Aboud }
149076c9eb99SAmjad Aboud 
149176c9eb99SAmjad Aboud void CodeViewDebug::collectMemberInfo(ClassInfo &Info,
149276c9eb99SAmjad Aboud                                       const DIDerivedType *DDTy) {
149376c9eb99SAmjad Aboud   if (!DDTy->getName().empty()) {
149476c9eb99SAmjad Aboud     Info.Members.push_back({DDTy, 0});
149576c9eb99SAmjad Aboud     return;
149676c9eb99SAmjad Aboud   }
14971ab7eac8SReid Kleckner   // An unnamed member must represent a nested struct or union. Add all the
14981ab7eac8SReid Kleckner   // indirect fields to the current record.
149976c9eb99SAmjad Aboud   assert((DDTy->getOffsetInBits() % 8) == 0 && "Unnamed bitfield member!");
150008bd744cSDavid Majnemer   uint64_t Offset = DDTy->getOffsetInBits();
150176c9eb99SAmjad Aboud   const DIType *Ty = DDTy->getBaseType().resolve();
15029ff936cfSReid Kleckner   const DICompositeType *DCTy = cast<DICompositeType>(Ty);
15031ab7eac8SReid Kleckner   ClassInfo NestedInfo = collectClassInfo(DCTy);
15041ab7eac8SReid Kleckner   for (const ClassInfo::MemberInfo &IndirectField : NestedInfo.Members)
150576c9eb99SAmjad Aboud     Info.Members.push_back(
15061ab7eac8SReid Kleckner         {IndirectField.MemberTypeNode, IndirectField.BaseOffset + Offset});
150776c9eb99SAmjad Aboud }
150876c9eb99SAmjad Aboud 
15091ab7eac8SReid Kleckner ClassInfo CodeViewDebug::collectClassInfo(const DICompositeType *Ty) {
15101ab7eac8SReid Kleckner   ClassInfo Info;
151176c9eb99SAmjad Aboud   // Add elements to structure type.
151276c9eb99SAmjad Aboud   DINodeArray Elements = Ty->getElements();
151376c9eb99SAmjad Aboud   for (auto *Element : Elements) {
151476c9eb99SAmjad Aboud     // We assume that the frontend provides all members in source declaration
151576c9eb99SAmjad Aboud     // order, which is what MSVC does.
151676c9eb99SAmjad Aboud     if (!Element)
151776c9eb99SAmjad Aboud       continue;
151876c9eb99SAmjad Aboud     if (auto *SP = dyn_cast<DISubprogram>(Element)) {
1519156a7239SReid Kleckner       Info.Methods[SP->getRawName()].push_back(SP);
152076c9eb99SAmjad Aboud     } else if (auto *DDTy = dyn_cast<DIDerivedType>(Element)) {
15219f7f3e1eSReid Kleckner       if (DDTy->getTag() == dwarf::DW_TAG_member) {
15221ab7eac8SReid Kleckner         collectMemberInfo(Info, DDTy);
15239f7f3e1eSReid Kleckner       } else if (DDTy->getTag() == dwarf::DW_TAG_inheritance) {
15249f7f3e1eSReid Kleckner         Info.Inheritance.push_back(DDTy);
15259dac4731SReid Kleckner       } else if (DDTy->getTag() == dwarf::DW_TAG_pointer_type &&
15269dac4731SReid Kleckner                  DDTy->getName() == "__vtbl_ptr_type") {
15279dac4731SReid Kleckner         Info.VShapeTI = getTypeIndex(DDTy);
152876c9eb99SAmjad Aboud       } else if (DDTy->getTag() == dwarf::DW_TAG_friend) {
152976c9eb99SAmjad Aboud         // Ignore friend members. It appears that MSVC emitted info about
153076c9eb99SAmjad Aboud         // friends in the past, but modern versions do not.
153176c9eb99SAmjad Aboud       }
1532820ca540SAdrian McCarthy     } else if (auto *Composite = dyn_cast<DICompositeType>(Element)) {
1533820ca540SAdrian McCarthy       Info.NestedClasses.push_back(Composite);
153476c9eb99SAmjad Aboud     }
153576c9eb99SAmjad Aboud     // Skip other unrecognized kinds of elements.
153676c9eb99SAmjad Aboud   }
15371ab7eac8SReid Kleckner   return Info;
153876c9eb99SAmjad Aboud }
153976c9eb99SAmjad Aboud 
1540a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerTypeClass(const DICompositeType *Ty) {
1541a8d57407SReid Kleckner   // First, construct the forward decl.  Don't look into Ty to compute the
1542a8d57407SReid Kleckner   // forward decl options, since it might not be available in all TUs.
1543a8d57407SReid Kleckner   TypeRecordKind Kind = getRecordKind(Ty);
1544a8d57407SReid Kleckner   ClassOptions CO =
1545e092dad7SReid Kleckner       ClassOptions::ForwardReference | getCommonClassOptions(Ty);
15466bdc24e7SDavid Majnemer   std::string FullName = getFullyQualifiedName(Ty);
15475e3e4bb2SZachary Turner   TypeIndex FwdDeclTI = TypeTable.writeKnownType(ClassRecord(
1548a8d57407SReid Kleckner       Kind, 0, CO, HfaKind::None, WindowsRTClassKind::None, TypeIndex(),
15490c5d874bSReid Kleckner       TypeIndex(), TypeIndex(), 0, FullName, Ty->getIdentifier()));
1550643dd836SReid Kleckner   if (!Ty->isForwardDecl())
1551643dd836SReid Kleckner     DeferredCompleteTypes.push_back(Ty);
1552a8d57407SReid Kleckner   return FwdDeclTI;
1553a8d57407SReid Kleckner }
1554a8d57407SReid Kleckner 
1555a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerCompleteTypeClass(const DICompositeType *Ty) {
1556a8d57407SReid Kleckner   // Construct the field list and complete type record.
1557a8d57407SReid Kleckner   TypeRecordKind Kind = getRecordKind(Ty);
1558e092dad7SReid Kleckner   ClassOptions CO = getCommonClassOptions(Ty);
155976c9eb99SAmjad Aboud   TypeIndex FieldTI;
156076c9eb99SAmjad Aboud   TypeIndex VShapeTI;
1561a8d57407SReid Kleckner   unsigned FieldCount;
1562820ca540SAdrian McCarthy   bool ContainsNestedClass;
1563820ca540SAdrian McCarthy   std::tie(FieldTI, VShapeTI, FieldCount, ContainsNestedClass) =
1564820ca540SAdrian McCarthy       lowerRecordFieldList(Ty);
1565820ca540SAdrian McCarthy 
1566820ca540SAdrian McCarthy   if (ContainsNestedClass)
1567820ca540SAdrian McCarthy     CO |= ClassOptions::ContainsNestedClass;
1568a8d57407SReid Kleckner 
15696bdc24e7SDavid Majnemer   std::string FullName = getFullyQualifiedName(Ty);
15700c5d874bSReid Kleckner 
1571a8d57407SReid Kleckner   uint64_t SizeInBytes = Ty->getSizeInBits() / 8;
15729a519a09SHans Wennborg 
15735e3e4bb2SZachary Turner   TypeIndex ClassTI = TypeTable.writeKnownType(ClassRecord(
157476c9eb99SAmjad Aboud       Kind, FieldCount, CO, HfaKind::None, WindowsRTClassKind::None, FieldTI,
15750c5d874bSReid Kleckner       TypeIndex(), VShapeTI, SizeInBytes, FullName, Ty->getIdentifier()));
15769a519a09SHans Wennborg 
15775e3e4bb2SZachary Turner   TypeTable.writeKnownType(UdtSourceLineRecord(
15785e3e4bb2SZachary Turner       ClassTI, TypeTable.writeKnownType(StringIdRecord(
15799a519a09SHans Wennborg                    TypeIndex(0x0), getFullFilepath(Ty->getFile()))),
15809a519a09SHans Wennborg       Ty->getLine()));
15819a519a09SHans Wennborg 
15824b63a98dSHans Wennborg   addToUDTs(Ty, ClassTI);
15834b63a98dSHans Wennborg 
15849a519a09SHans Wennborg   return ClassTI;
1585a8d57407SReid Kleckner }
1586a8d57407SReid Kleckner 
1587a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerTypeUnion(const DICompositeType *Ty) {
1588a8d57407SReid Kleckner   ClassOptions CO =
1589e092dad7SReid Kleckner       ClassOptions::ForwardReference | getCommonClassOptions(Ty);
15906bdc24e7SDavid Majnemer   std::string FullName = getFullyQualifiedName(Ty);
15915e3e4bb2SZachary Turner   TypeIndex FwdDeclTI = TypeTable.writeKnownType(UnionRecord(
15925e3e4bb2SZachary Turner       0, CO, HfaKind::None, TypeIndex(), 0, FullName, Ty->getIdentifier()));
1593643dd836SReid Kleckner   if (!Ty->isForwardDecl())
1594643dd836SReid Kleckner     DeferredCompleteTypes.push_back(Ty);
1595a8d57407SReid Kleckner   return FwdDeclTI;
1596a8d57407SReid Kleckner }
1597a8d57407SReid Kleckner 
1598a8d57407SReid Kleckner TypeIndex CodeViewDebug::lowerCompleteTypeUnion(const DICompositeType *Ty) {
1599e1e7372eSDavid Majnemer   ClassOptions CO = ClassOptions::Sealed | getCommonClassOptions(Ty);
160076c9eb99SAmjad Aboud   TypeIndex FieldTI;
1601a8d57407SReid Kleckner   unsigned FieldCount;
1602820ca540SAdrian McCarthy   bool ContainsNestedClass;
1603820ca540SAdrian McCarthy   std::tie(FieldTI, std::ignore, FieldCount, ContainsNestedClass) =
1604820ca540SAdrian McCarthy       lowerRecordFieldList(Ty);
1605820ca540SAdrian McCarthy 
1606820ca540SAdrian McCarthy   if (ContainsNestedClass)
1607820ca540SAdrian McCarthy     CO |= ClassOptions::ContainsNestedClass;
1608820ca540SAdrian McCarthy 
1609a8d57407SReid Kleckner   uint64_t SizeInBytes = Ty->getSizeInBits() / 8;
16106bdc24e7SDavid Majnemer   std::string FullName = getFullyQualifiedName(Ty);
16119a519a09SHans Wennborg 
16125e3e4bb2SZachary Turner   TypeIndex UnionTI = TypeTable.writeKnownType(
16139a519a09SHans Wennborg       UnionRecord(FieldCount, CO, HfaKind::None, FieldTI, SizeInBytes, FullName,
1614a8d57407SReid Kleckner                   Ty->getIdentifier()));
16159a519a09SHans Wennborg 
16165e3e4bb2SZachary Turner   TypeTable.writeKnownType(UdtSourceLineRecord(
16175e3e4bb2SZachary Turner       UnionTI, TypeTable.writeKnownType(StringIdRecord(
16189a519a09SHans Wennborg                    TypeIndex(0x0), getFullFilepath(Ty->getFile()))),
16199a519a09SHans Wennborg       Ty->getLine()));
16209a519a09SHans Wennborg 
16214b63a98dSHans Wennborg   addToUDTs(Ty, UnionTI);
16224b63a98dSHans Wennborg 
16239a519a09SHans Wennborg   return UnionTI;
1624a8d57407SReid Kleckner }
1625a8d57407SReid Kleckner 
1626820ca540SAdrian McCarthy std::tuple<TypeIndex, TypeIndex, unsigned, bool>
1627a8d57407SReid Kleckner CodeViewDebug::lowerRecordFieldList(const DICompositeType *Ty) {
1628a8d57407SReid Kleckner   // Manually count members. MSVC appears to count everything that generates a
1629a8d57407SReid Kleckner   // field list record. Each individual overload in a method overload group
1630a8d57407SReid Kleckner   // contributes to this count, even though the overload group is a single field
1631a8d57407SReid Kleckner   // list record.
1632a8d57407SReid Kleckner   unsigned MemberCount = 0;
16331ab7eac8SReid Kleckner   ClassInfo Info = collectClassInfo(Ty);
1634a8d57407SReid Kleckner   FieldListRecordBuilder Fields;
163576c9eb99SAmjad Aboud 
16369f7f3e1eSReid Kleckner   // Create base classes.
16379f7f3e1eSReid Kleckner   for (const DIDerivedType *I : Info.Inheritance) {
16389f7f3e1eSReid Kleckner     if (I->getFlags() & DINode::FlagVirtual) {
16399f7f3e1eSReid Kleckner       // Virtual base.
16409f7f3e1eSReid Kleckner       // FIXME: Emit VBPtrOffset when the frontend provides it.
16419f7f3e1eSReid Kleckner       unsigned VBPtrOffset = 0;
16429f7f3e1eSReid Kleckner       // FIXME: Despite the accessor name, the offset is really in bytes.
16439f7f3e1eSReid Kleckner       unsigned VBTableIndex = I->getOffsetInBits() / 4;
16445e3e4bb2SZachary Turner       Fields.writeMemberType(VirtualBaseClassRecord(
16459f7f3e1eSReid Kleckner           translateAccessFlags(Ty->getTag(), I->getFlags()),
16469f7f3e1eSReid Kleckner           getTypeIndex(I->getBaseType()), getVBPTypeIndex(), VBPtrOffset,
16479f7f3e1eSReid Kleckner           VBTableIndex));
16489f7f3e1eSReid Kleckner     } else {
16499f7f3e1eSReid Kleckner       assert(I->getOffsetInBits() % 8 == 0 &&
16509f7f3e1eSReid Kleckner              "bases must be on byte boundaries");
16515e3e4bb2SZachary Turner       Fields.writeMemberType(BaseClassRecord(
16529f7f3e1eSReid Kleckner           translateAccessFlags(Ty->getTag(), I->getFlags()),
16539f7f3e1eSReid Kleckner           getTypeIndex(I->getBaseType()), I->getOffsetInBits() / 8));
16549f7f3e1eSReid Kleckner     }
16559f7f3e1eSReid Kleckner   }
16569f7f3e1eSReid Kleckner 
165776c9eb99SAmjad Aboud   // Create members.
165876c9eb99SAmjad Aboud   for (ClassInfo::MemberInfo &MemberInfo : Info.Members) {
165976c9eb99SAmjad Aboud     const DIDerivedType *Member = MemberInfo.MemberTypeNode;
166076c9eb99SAmjad Aboud     TypeIndex MemberBaseType = getTypeIndex(Member->getBaseType());
16619319cbc0SDavid Majnemer     StringRef MemberName = Member->getName();
16629319cbc0SDavid Majnemer     MemberAccess Access =
16639319cbc0SDavid Majnemer         translateAccessFlags(Ty->getTag(), Member->getFlags());
166476c9eb99SAmjad Aboud 
1665a8d57407SReid Kleckner     if (Member->isStaticMember()) {
16665e3e4bb2SZachary Turner       Fields.writeMemberType(
16679319cbc0SDavid Majnemer           StaticDataMemberRecord(Access, MemberBaseType, MemberName));
1668a8d57407SReid Kleckner       MemberCount++;
166976c9eb99SAmjad Aboud       continue;
1670a8d57407SReid Kleckner     }
1671a8d57407SReid Kleckner 
16729dac4731SReid Kleckner     // Virtual function pointer member.
16739dac4731SReid Kleckner     if ((Member->getFlags() & DINode::FlagArtificial) &&
16749dac4731SReid Kleckner         Member->getName().startswith("_vptr$")) {
16759dac4731SReid Kleckner       Fields.writeMemberType(VFPtrRecord(getTypeIndex(Member->getBaseType())));
16769dac4731SReid Kleckner       MemberCount++;
16779dac4731SReid Kleckner       continue;
16789dac4731SReid Kleckner     }
16799dac4731SReid Kleckner 
16809319cbc0SDavid Majnemer     // Data member.
168108bd744cSDavid Majnemer     uint64_t MemberOffsetInBits =
168208bd744cSDavid Majnemer         Member->getOffsetInBits() + MemberInfo.BaseOffset;
16839319cbc0SDavid Majnemer     if (Member->isBitField()) {
16849319cbc0SDavid Majnemer       uint64_t StartBitOffset = MemberOffsetInBits;
16859319cbc0SDavid Majnemer       if (const auto *CI =
16869319cbc0SDavid Majnemer               dyn_cast_or_null<ConstantInt>(Member->getStorageOffsetInBits())) {
168708bd744cSDavid Majnemer         MemberOffsetInBits = CI->getZExtValue() + MemberInfo.BaseOffset;
16889319cbc0SDavid Majnemer       }
16899319cbc0SDavid Majnemer       StartBitOffset -= MemberOffsetInBits;
16905e3e4bb2SZachary Turner       MemberBaseType = TypeTable.writeKnownType(BitFieldRecord(
16919319cbc0SDavid Majnemer           MemberBaseType, Member->getSizeInBits(), StartBitOffset));
16929319cbc0SDavid Majnemer     }
16939319cbc0SDavid Majnemer     uint64_t MemberOffsetInBytes = MemberOffsetInBits / 8;
16945e3e4bb2SZachary Turner     Fields.writeMemberType(DataMemberRecord(Access, MemberBaseType,
16959319cbc0SDavid Majnemer                                             MemberOffsetInBytes, MemberName));
169676c9eb99SAmjad Aboud     MemberCount++;
169776c9eb99SAmjad Aboud   }
169876c9eb99SAmjad Aboud 
169976c9eb99SAmjad Aboud   // Create methods
170076c9eb99SAmjad Aboud   for (auto &MethodItr : Info.Methods) {
170176c9eb99SAmjad Aboud     StringRef Name = MethodItr.first->getString();
170276c9eb99SAmjad Aboud 
170376c9eb99SAmjad Aboud     std::vector<OneMethodRecord> Methods;
1704156a7239SReid Kleckner     for (const DISubprogram *SP : MethodItr.second) {
1705156a7239SReid Kleckner       TypeIndex MethodType = getMemberFunctionType(SP, Ty);
1706156a7239SReid Kleckner       bool Introduced = SP->getFlags() & DINode::FlagIntroducedVirtual;
170776c9eb99SAmjad Aboud 
170876c9eb99SAmjad Aboud       unsigned VFTableOffset = -1;
170976c9eb99SAmjad Aboud       if (Introduced)
171076c9eb99SAmjad Aboud         VFTableOffset = SP->getVirtualIndex() * getPointerSizeInBytes();
171176c9eb99SAmjad Aboud 
171276c9eb99SAmjad Aboud       Methods.push_back(
171376c9eb99SAmjad Aboud           OneMethodRecord(MethodType, translateMethodKindFlags(SP, Introduced),
171476c9eb99SAmjad Aboud                           translateMethodOptionFlags(SP),
171576c9eb99SAmjad Aboud                           translateAccessFlags(Ty->getTag(), SP->getFlags()),
171676c9eb99SAmjad Aboud                           VFTableOffset, Name));
171776c9eb99SAmjad Aboud       MemberCount++;
171876c9eb99SAmjad Aboud     }
171976c9eb99SAmjad Aboud     assert(Methods.size() > 0 && "Empty methods map entry");
172076c9eb99SAmjad Aboud     if (Methods.size() == 1)
17215e3e4bb2SZachary Turner       Fields.writeMemberType(Methods[0]);
172276c9eb99SAmjad Aboud     else {
172376c9eb99SAmjad Aboud       TypeIndex MethodList =
17245e3e4bb2SZachary Turner           TypeTable.writeKnownType(MethodOverloadListRecord(Methods));
17255e3e4bb2SZachary Turner       Fields.writeMemberType(
172676c9eb99SAmjad Aboud           OverloadedMethodRecord(Methods.size(), MethodList, Name));
172776c9eb99SAmjad Aboud     }
172876c9eb99SAmjad Aboud   }
1729820ca540SAdrian McCarthy 
1730820ca540SAdrian McCarthy   // Create nested classes.
1731820ca540SAdrian McCarthy   for (const DICompositeType *Nested : Info.NestedClasses) {
1732820ca540SAdrian McCarthy     NestedTypeRecord R(getTypeIndex(DITypeRef(Nested)), Nested->getName());
17335e3e4bb2SZachary Turner     Fields.writeMemberType(R);
1734820ca540SAdrian McCarthy     MemberCount++;
1735820ca540SAdrian McCarthy   }
1736820ca540SAdrian McCarthy 
173776c9eb99SAmjad Aboud   TypeIndex FieldTI = TypeTable.writeFieldList(Fields);
17389dac4731SReid Kleckner   return std::make_tuple(FieldTI, Info.VShapeTI, MemberCount,
1739820ca540SAdrian McCarthy                          !Info.NestedClasses.empty());
174076c9eb99SAmjad Aboud }
174176c9eb99SAmjad Aboud 
17429f7f3e1eSReid Kleckner TypeIndex CodeViewDebug::getVBPTypeIndex() {
17439f7f3e1eSReid Kleckner   if (!VBPType.getIndex()) {
17449f7f3e1eSReid Kleckner     // Make a 'const int *' type.
17459f7f3e1eSReid Kleckner     ModifierRecord MR(TypeIndex::Int32(), ModifierOptions::Const);
17465e3e4bb2SZachary Turner     TypeIndex ModifiedTI = TypeTable.writeKnownType(MR);
17479f7f3e1eSReid Kleckner 
17489f7f3e1eSReid Kleckner     PointerKind PK = getPointerSizeInBytes() == 8 ? PointerKind::Near64
17499f7f3e1eSReid Kleckner                                                   : PointerKind::Near32;
17509f7f3e1eSReid Kleckner     PointerMode PM = PointerMode::Pointer;
17519f7f3e1eSReid Kleckner     PointerOptions PO = PointerOptions::None;
17529f7f3e1eSReid Kleckner     PointerRecord PR(ModifiedTI, PK, PM, PO, getPointerSizeInBytes());
17539f7f3e1eSReid Kleckner 
17545e3e4bb2SZachary Turner     VBPType = TypeTable.writeKnownType(PR);
17559f7f3e1eSReid Kleckner   }
17569f7f3e1eSReid Kleckner 
17579f7f3e1eSReid Kleckner   return VBPType;
17589f7f3e1eSReid Kleckner }
17599f7f3e1eSReid Kleckner 
176076c9eb99SAmjad Aboud TypeIndex CodeViewDebug::getTypeIndex(DITypeRef TypeRef, DITypeRef ClassTyRef) {
17615acacbb0SReid Kleckner   const DIType *Ty = TypeRef.resolve();
176276c9eb99SAmjad Aboud   const DIType *ClassTy = ClassTyRef.resolve();
17635acacbb0SReid Kleckner 
17645acacbb0SReid Kleckner   // The null DIType is the void type. Don't try to hash it.
17655acacbb0SReid Kleckner   if (!Ty)
17665acacbb0SReid Kleckner     return TypeIndex::Void();
17675acacbb0SReid Kleckner 
1768a8d57407SReid Kleckner   // Check if we've already translated this type. Don't try to do a
1769a8d57407SReid Kleckner   // get-or-create style insertion that caches the hash lookup across the
1770a8d57407SReid Kleckner   // lowerType call. It will update the TypeIndices map.
177176c9eb99SAmjad Aboud   auto I = TypeIndices.find({Ty, ClassTy});
17725acacbb0SReid Kleckner   if (I != TypeIndices.end())
17735acacbb0SReid Kleckner     return I->second;
17745acacbb0SReid Kleckner 
1775643dd836SReid Kleckner   TypeLoweringScope S(*this);
1776b5af11dfSReid Kleckner   TypeIndex TI = lowerType(Ty, ClassTy);
1777b5af11dfSReid Kleckner   return recordTypeIndexForDINode(Ty, TI, ClassTy);
1778a8d57407SReid Kleckner }
1779a8d57407SReid Kleckner 
1780a8d57407SReid Kleckner TypeIndex CodeViewDebug::getCompleteTypeIndex(DITypeRef TypeRef) {
1781a8d57407SReid Kleckner   const DIType *Ty = TypeRef.resolve();
1782a8d57407SReid Kleckner 
1783a8d57407SReid Kleckner   // The null DIType is the void type. Don't try to hash it.
1784a8d57407SReid Kleckner   if (!Ty)
1785a8d57407SReid Kleckner     return TypeIndex::Void();
1786a8d57407SReid Kleckner 
1787a8d57407SReid Kleckner   // If this is a non-record type, the complete type index is the same as the
1788a8d57407SReid Kleckner   // normal type index. Just call getTypeIndex.
1789a8d57407SReid Kleckner   switch (Ty->getTag()) {
1790a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
1791a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
1792a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
1793a8d57407SReid Kleckner     break;
1794a8d57407SReid Kleckner   default:
1795a8d57407SReid Kleckner     return getTypeIndex(Ty);
1796a8d57407SReid Kleckner   }
1797a8d57407SReid Kleckner 
1798a8d57407SReid Kleckner   // Check if we've already translated the complete record type.  Lowering a
1799a8d57407SReid Kleckner   // complete type should never trigger lowering another complete type, so we
1800a8d57407SReid Kleckner   // can reuse the hash table lookup result.
1801a8d57407SReid Kleckner   const auto *CTy = cast<DICompositeType>(Ty);
1802a8d57407SReid Kleckner   auto InsertResult = CompleteTypeIndices.insert({CTy, TypeIndex()});
1803a8d57407SReid Kleckner   if (!InsertResult.second)
1804a8d57407SReid Kleckner     return InsertResult.first->second;
1805a8d57407SReid Kleckner 
1806643dd836SReid Kleckner   TypeLoweringScope S(*this);
1807643dd836SReid Kleckner 
1808a8d57407SReid Kleckner   // Make sure the forward declaration is emitted first. It's unclear if this
1809a8d57407SReid Kleckner   // is necessary, but MSVC does it, and we should follow suit until we can show
1810a8d57407SReid Kleckner   // otherwise.
1811a8d57407SReid Kleckner   TypeIndex FwdDeclTI = getTypeIndex(CTy);
1812a8d57407SReid Kleckner 
1813a8d57407SReid Kleckner   // Just use the forward decl if we don't have complete type info. This might
1814a8d57407SReid Kleckner   // happen if the frontend is using modules and expects the complete definition
1815a8d57407SReid Kleckner   // to be emitted elsewhere.
1816a8d57407SReid Kleckner   if (CTy->isForwardDecl())
1817a8d57407SReid Kleckner     return FwdDeclTI;
1818a8d57407SReid Kleckner 
1819a8d57407SReid Kleckner   TypeIndex TI;
1820a8d57407SReid Kleckner   switch (CTy->getTag()) {
1821a8d57407SReid Kleckner   case dwarf::DW_TAG_class_type:
1822a8d57407SReid Kleckner   case dwarf::DW_TAG_structure_type:
1823a8d57407SReid Kleckner     TI = lowerCompleteTypeClass(CTy);
1824a8d57407SReid Kleckner     break;
1825a8d57407SReid Kleckner   case dwarf::DW_TAG_union_type:
1826a8d57407SReid Kleckner     TI = lowerCompleteTypeUnion(CTy);
1827a8d57407SReid Kleckner     break;
1828a8d57407SReid Kleckner   default:
1829a8d57407SReid Kleckner     llvm_unreachable("not a record");
1830a8d57407SReid Kleckner   }
1831a8d57407SReid Kleckner 
1832a8d57407SReid Kleckner   InsertResult.first->second = TI;
18335acacbb0SReid Kleckner   return TI;
18345acacbb0SReid Kleckner }
18355acacbb0SReid Kleckner 
1836643dd836SReid Kleckner /// Emit all the deferred complete record types. Try to do this in FIFO order,
1837acee5685SAmjad Aboud /// and do this until fixpoint, as each complete record type typically
1838acee5685SAmjad Aboud /// references
1839643dd836SReid Kleckner /// many other record types.
1840643dd836SReid Kleckner void CodeViewDebug::emitDeferredCompleteTypes() {
1841643dd836SReid Kleckner   SmallVector<const DICompositeType *, 4> TypesToEmit;
1842643dd836SReid Kleckner   while (!DeferredCompleteTypes.empty()) {
1843643dd836SReid Kleckner     std::swap(DeferredCompleteTypes, TypesToEmit);
1844643dd836SReid Kleckner     for (const DICompositeType *RecordTy : TypesToEmit)
1845643dd836SReid Kleckner       getCompleteTypeIndex(RecordTy);
1846643dd836SReid Kleckner     TypesToEmit.clear();
1847643dd836SReid Kleckner   }
1848643dd836SReid Kleckner }
1849643dd836SReid Kleckner 
185010dd55c5SReid Kleckner void CodeViewDebug::emitLocalVariableList(ArrayRef<LocalVariable> Locals) {
185110dd55c5SReid Kleckner   // Get the sorted list of parameters and emit them first.
185210dd55c5SReid Kleckner   SmallVector<const LocalVariable *, 6> Params;
185310dd55c5SReid Kleckner   for (const LocalVariable &L : Locals)
185410dd55c5SReid Kleckner     if (L.DIVar->isParameter())
185510dd55c5SReid Kleckner       Params.push_back(&L);
185610dd55c5SReid Kleckner   std::sort(Params.begin(), Params.end(),
185710dd55c5SReid Kleckner             [](const LocalVariable *L, const LocalVariable *R) {
185810dd55c5SReid Kleckner               return L->DIVar->getArg() < R->DIVar->getArg();
185910dd55c5SReid Kleckner             });
186010dd55c5SReid Kleckner   for (const LocalVariable *L : Params)
186110dd55c5SReid Kleckner     emitLocalVariable(*L);
186210dd55c5SReid Kleckner 
186310dd55c5SReid Kleckner   // Next emit all non-parameters in the order that we found them.
186410dd55c5SReid Kleckner   for (const LocalVariable &L : Locals)
186510dd55c5SReid Kleckner     if (!L.DIVar->isParameter())
186610dd55c5SReid Kleckner       emitLocalVariable(L);
186710dd55c5SReid Kleckner }
186810dd55c5SReid Kleckner 
1869f9c275feSReid Kleckner void CodeViewDebug::emitLocalVariable(const LocalVariable &Var) {
1870f9c275feSReid Kleckner   // LocalSym record, see SymbolRecord.h for more info.
1871f9c275feSReid Kleckner   MCSymbol *LocalBegin = MMI->getContext().createTempSymbol(),
1872f9c275feSReid Kleckner            *LocalEnd = MMI->getContext().createTempSymbol();
1873f9c275feSReid Kleckner   OS.AddComment("Record length");
1874f9c275feSReid Kleckner   OS.emitAbsoluteSymbolDiff(LocalEnd, LocalBegin, 2);
1875f9c275feSReid Kleckner   OS.EmitLabel(LocalBegin);
1876f9c275feSReid Kleckner 
1877f9c275feSReid Kleckner   OS.AddComment("Record kind: S_LOCAL");
187863a2846eSZachary Turner   OS.EmitIntValue(unsigned(SymbolKind::S_LOCAL), 2);
1879f9c275feSReid Kleckner 
188063a2846eSZachary Turner   LocalSymFlags Flags = LocalSymFlags::None;
1881f9c275feSReid Kleckner   if (Var.DIVar->isParameter())
188263a2846eSZachary Turner     Flags |= LocalSymFlags::IsParameter;
1883876330d5SReid Kleckner   if (Var.DefRanges.empty())
188463a2846eSZachary Turner     Flags |= LocalSymFlags::IsOptimizedOut;
1885f9c275feSReid Kleckner 
1886f9c275feSReid Kleckner   OS.AddComment("TypeIndex");
1887a8d57407SReid Kleckner   TypeIndex TI = getCompleteTypeIndex(Var.DIVar->getType());
18885acacbb0SReid Kleckner   OS.EmitIntValue(TI.getIndex(), 4);
1889f9c275feSReid Kleckner   OS.AddComment("Flags");
189063a2846eSZachary Turner   OS.EmitIntValue(static_cast<uint16_t>(Flags), 2);
18911256125fSDavid Majnemer   // Truncate the name so we won't overflow the record length field.
1892b9456a5eSDavid Majnemer   emitNullTerminatedSymbolName(OS, Var.DIVar->getName());
1893f9c275feSReid Kleckner   OS.EmitLabel(LocalEnd);
1894f9c275feSReid Kleckner 
1895876330d5SReid Kleckner   // Calculate the on disk prefix of the appropriate def range record. The
1896876330d5SReid Kleckner   // records and on disk formats are described in SymbolRecords.h. BytePrefix
1897876330d5SReid Kleckner   // should be big enough to hold all forms without memory allocation.
1898876330d5SReid Kleckner   SmallString<20> BytePrefix;
1899876330d5SReid Kleckner   for (const LocalVarDefRange &DefRange : Var.DefRanges) {
1900876330d5SReid Kleckner     BytePrefix.clear();
1901876330d5SReid Kleckner     // FIXME: Handle bitpieces.
1902876330d5SReid Kleckner     if (DefRange.StructOffset != 0)
1903876330d5SReid Kleckner       continue;
1904876330d5SReid Kleckner 
1905876330d5SReid Kleckner     if (DefRange.InMemory) {
1906a78ecd1eSZachary Turner       DefRangeRegisterRelSym Sym(DefRange.CVRegister, 0, DefRange.DataOffset, 0,
1907a78ecd1eSZachary Turner                                  0, 0, ArrayRef<LocalVariableAddrGap>());
1908f9c275feSReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER_REL);
1909876330d5SReid Kleckner       BytePrefix +=
1910876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
1911a78ecd1eSZachary Turner       BytePrefix +=
1912a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
1913a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
1914876330d5SReid Kleckner     } else {
1915876330d5SReid Kleckner       assert(DefRange.DataOffset == 0 && "unexpected offset into register");
1916a78ecd1eSZachary Turner       // Unclear what matters here.
1917a78ecd1eSZachary Turner       DefRangeRegisterSym Sym(DefRange.CVRegister, 0, 0, 0, 0,
1918a78ecd1eSZachary Turner                               ArrayRef<LocalVariableAddrGap>());
1919876330d5SReid Kleckner       ulittle16_t SymKind = ulittle16_t(S_DEFRANGE_REGISTER);
1920876330d5SReid Kleckner       BytePrefix +=
1921876330d5SReid Kleckner           StringRef(reinterpret_cast<const char *>(&SymKind), sizeof(SymKind));
1922a78ecd1eSZachary Turner       BytePrefix +=
1923a78ecd1eSZachary Turner           StringRef(reinterpret_cast<const char *>(&Sym.Header),
1924a78ecd1eSZachary Turner                     sizeof(Sym.Header) - sizeof(LocalVariableAddrRange));
1925876330d5SReid Kleckner     }
1926876330d5SReid Kleckner     OS.EmitCVDefRangeDirective(DefRange.Ranges, BytePrefix);
1927876330d5SReid Kleckner   }
1928f9c275feSReid Kleckner }
1929f9c275feSReid Kleckner 
193070f5bc99SReid Kleckner void CodeViewDebug::endFunction(const MachineFunction *MF) {
193170f5bc99SReid Kleckner   if (!Asm || !CurFn)  // We haven't created any debug info for this function.
193270f5bc99SReid Kleckner     return;
193370f5bc99SReid Kleckner 
193470f5bc99SReid Kleckner   const Function *GV = MF->getFunction();
193570f5bc99SReid Kleckner   assert(FnDebugInfo.count(GV));
193670f5bc99SReid Kleckner   assert(CurFn == &FnDebugInfo[GV]);
193770f5bc99SReid Kleckner 
1938adebb937SPete Cooper   collectVariableInfo(GV->getSubprogram());
1939876330d5SReid Kleckner 
1940876330d5SReid Kleckner   DebugHandlerBase::endFunction(MF);
1941876330d5SReid Kleckner 
19422214ed89SReid Kleckner   // Don't emit anything if we don't have any line tables.
19432214ed89SReid Kleckner   if (!CurFn->HaveLineInfo) {
194470f5bc99SReid Kleckner     FnDebugInfo.erase(GV);
1945f9c275feSReid Kleckner     CurFn = nullptr;
1946f9c275feSReid Kleckner     return;
194770f5bc99SReid Kleckner   }
1948f9c275feSReid Kleckner 
1949f9c275feSReid Kleckner   CurFn->End = Asm->getFunctionEnd();
1950f9c275feSReid Kleckner 
195170f5bc99SReid Kleckner   CurFn = nullptr;
195270f5bc99SReid Kleckner }
195370f5bc99SReid Kleckner 
195470f5bc99SReid Kleckner void CodeViewDebug::beginInstruction(const MachineInstr *MI) {
1955f9c275feSReid Kleckner   DebugHandlerBase::beginInstruction(MI);
1956f9c275feSReid Kleckner 
195770f5bc99SReid Kleckner   // Ignore DBG_VALUE locations and function prologue.
195867f684e1SDavid Majnemer   if (!Asm || !CurFn || MI->isDebugValue() ||
195967f684e1SDavid Majnemer       MI->getFlag(MachineInstr::FrameSetup))
196070f5bc99SReid Kleckner     return;
196170f5bc99SReid Kleckner   DebugLoc DL = MI->getDebugLoc();
196270f5bc99SReid Kleckner   if (DL == PrevInstLoc || !DL)
196370f5bc99SReid Kleckner     return;
196470f5bc99SReid Kleckner   maybeRecordLocation(DL, Asm->MF);
196570f5bc99SReid Kleckner }
19666f3406dfSReid Kleckner 
19676f3406dfSReid Kleckner MCSymbol *CodeViewDebug::beginCVSubsection(ModuleSubstreamKind Kind) {
19686f3406dfSReid Kleckner   MCSymbol *BeginLabel = MMI->getContext().createTempSymbol(),
19696f3406dfSReid Kleckner            *EndLabel = MMI->getContext().createTempSymbol();
19706f3406dfSReid Kleckner   OS.EmitIntValue(unsigned(Kind), 4);
19716f3406dfSReid Kleckner   OS.AddComment("Subsection size");
19726f3406dfSReid Kleckner   OS.emitAbsoluteSymbolDiff(EndLabel, BeginLabel, 4);
19736f3406dfSReid Kleckner   OS.EmitLabel(BeginLabel);
19746f3406dfSReid Kleckner   return EndLabel;
19756f3406dfSReid Kleckner }
19766f3406dfSReid Kleckner 
19776f3406dfSReid Kleckner void CodeViewDebug::endCVSubsection(MCSymbol *EndLabel) {
19786f3406dfSReid Kleckner   OS.EmitLabel(EndLabel);
19796f3406dfSReid Kleckner   // Every subsection must be aligned to a 4-byte boundary.
19806f3406dfSReid Kleckner   OS.EmitValueToAlignment(4);
19816f3406dfSReid Kleckner }
19826f3406dfSReid Kleckner 
19833128b10cSDavid Majnemer void CodeViewDebug::emitDebugInfoForUDTs(
19843128b10cSDavid Majnemer     ArrayRef<std::pair<std::string, TypeIndex>> UDTs) {
19853128b10cSDavid Majnemer   for (const std::pair<std::string, codeview::TypeIndex> &UDT : UDTs) {
19863128b10cSDavid Majnemer     MCSymbol *UDTRecordBegin = MMI->getContext().createTempSymbol(),
19873128b10cSDavid Majnemer              *UDTRecordEnd = MMI->getContext().createTempSymbol();
19883128b10cSDavid Majnemer     OS.AddComment("Record length");
19893128b10cSDavid Majnemer     OS.emitAbsoluteSymbolDiff(UDTRecordEnd, UDTRecordBegin, 2);
19903128b10cSDavid Majnemer     OS.EmitLabel(UDTRecordBegin);
19913128b10cSDavid Majnemer 
19923128b10cSDavid Majnemer     OS.AddComment("Record kind: S_UDT");
19933128b10cSDavid Majnemer     OS.EmitIntValue(unsigned(SymbolKind::S_UDT), 2);
19943128b10cSDavid Majnemer 
19953128b10cSDavid Majnemer     OS.AddComment("Type");
19963128b10cSDavid Majnemer     OS.EmitIntValue(UDT.second.getIndex(), 4);
19973128b10cSDavid Majnemer 
19983128b10cSDavid Majnemer     emitNullTerminatedSymbolName(OS, UDT.first);
19993128b10cSDavid Majnemer     OS.EmitLabel(UDTRecordEnd);
20003128b10cSDavid Majnemer   }
20013128b10cSDavid Majnemer }
20023128b10cSDavid Majnemer 
20036f3406dfSReid Kleckner void CodeViewDebug::emitDebugInfoForGlobals() {
2004*d4135bbcSPeter Collingbourne   DenseMap<const DIGlobalVariable *, const GlobalVariable *> GlobalMap;
2005*d4135bbcSPeter Collingbourne   for (const GlobalVariable &GV : MMI->getModule()->globals()) {
2006*d4135bbcSPeter Collingbourne     SmallVector<MDNode *, 1> MDs;
2007*d4135bbcSPeter Collingbourne     GV.getMetadata(LLVMContext::MD_dbg, MDs);
2008*d4135bbcSPeter Collingbourne     for (MDNode *MD : MDs)
2009*d4135bbcSPeter Collingbourne       GlobalMap[cast<DIGlobalVariable>(MD)] = &GV;
2010*d4135bbcSPeter Collingbourne   }
2011*d4135bbcSPeter Collingbourne 
20126f3406dfSReid Kleckner   NamedMDNode *CUs = MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
20136f3406dfSReid Kleckner   for (const MDNode *Node : CUs->operands()) {
20146f3406dfSReid Kleckner     const auto *CU = cast<DICompileUnit>(Node);
20156f3406dfSReid Kleckner 
20166f3406dfSReid Kleckner     // First, emit all globals that are not in a comdat in a single symbol
20176f3406dfSReid Kleckner     // substream. MSVC doesn't like it if the substream is empty, so only open
20186f3406dfSReid Kleckner     // it if we have at least one global to emit.
20196f3406dfSReid Kleckner     switchToDebugSectionForSymbol(nullptr);
20206f3406dfSReid Kleckner     MCSymbol *EndLabel = nullptr;
20216f3406dfSReid Kleckner     for (const DIGlobalVariable *G : CU->getGlobalVariables()) {
2022*d4135bbcSPeter Collingbourne       if (const auto *GV = GlobalMap.lookup(G))
2023577be0feSDavid Majnemer         if (!GV->hasComdat() && !GV->isDeclarationForLinker()) {
20246f3406dfSReid Kleckner           if (!EndLabel) {
20256f3406dfSReid Kleckner             OS.AddComment("Symbol subsection for globals");
20266f3406dfSReid Kleckner             EndLabel = beginCVSubsection(ModuleSubstreamKind::Symbols);
20276f3406dfSReid Kleckner           }
2028*d4135bbcSPeter Collingbourne           emitDebugInfoForGlobal(G, GV, Asm->getSymbol(GV));
20296f3406dfSReid Kleckner         }
20306d1d2754SReid Kleckner     }
20316f3406dfSReid Kleckner     if (EndLabel)
20326f3406dfSReid Kleckner       endCVSubsection(EndLabel);
20336f3406dfSReid Kleckner 
20346f3406dfSReid Kleckner     // Second, emit each global that is in a comdat into its own .debug$S
20356f3406dfSReid Kleckner     // section along with its own symbol substream.
20366f3406dfSReid Kleckner     for (const DIGlobalVariable *G : CU->getGlobalVariables()) {
2037*d4135bbcSPeter Collingbourne       if (const auto *GV = GlobalMap.lookup(G)) {
20386f3406dfSReid Kleckner         if (GV->hasComdat()) {
20396f3406dfSReid Kleckner           MCSymbol *GVSym = Asm->getSymbol(GV);
20406f3406dfSReid Kleckner           OS.AddComment("Symbol subsection for " +
20416f3406dfSReid Kleckner                         Twine(GlobalValue::getRealLinkageName(GV->getName())));
20426f3406dfSReid Kleckner           switchToDebugSectionForSymbol(GVSym);
20436f3406dfSReid Kleckner           EndLabel = beginCVSubsection(ModuleSubstreamKind::Symbols);
2044*d4135bbcSPeter Collingbourne           emitDebugInfoForGlobal(G, GV, GVSym);
20456f3406dfSReid Kleckner           endCVSubsection(EndLabel);
20466f3406dfSReid Kleckner         }
20476f3406dfSReid Kleckner       }
20486f3406dfSReid Kleckner     }
20496f3406dfSReid Kleckner   }
20506f3406dfSReid Kleckner }
20516f3406dfSReid Kleckner 
2052b510b458SHans Wennborg void CodeViewDebug::emitDebugInfoForRetainedTypes() {
2053b510b458SHans Wennborg   NamedMDNode *CUs = MMI->getModule()->getNamedMetadata("llvm.dbg.cu");
2054b510b458SHans Wennborg   for (const MDNode *Node : CUs->operands()) {
2055b510b458SHans Wennborg     for (auto *Ty : cast<DICompileUnit>(Node)->getRetainedTypes()) {
2056b510b458SHans Wennborg       if (DIType *RT = dyn_cast<DIType>(Ty)) {
2057b510b458SHans Wennborg         getTypeIndex(RT);
2058b510b458SHans Wennborg         // FIXME: Add to global/local DTU list.
2059b510b458SHans Wennborg       }
2060b510b458SHans Wennborg     }
2061b510b458SHans Wennborg   }
2062b510b458SHans Wennborg }
2063b510b458SHans Wennborg 
20646f3406dfSReid Kleckner void CodeViewDebug::emitDebugInfoForGlobal(const DIGlobalVariable *DIGV,
2065*d4135bbcSPeter Collingbourne                                            const GlobalVariable *GV,
20666f3406dfSReid Kleckner                                            MCSymbol *GVSym) {
20676f3406dfSReid Kleckner   // DataSym record, see SymbolRecord.h for more info.
20686f3406dfSReid Kleckner   // FIXME: Thread local data, etc
20696f3406dfSReid Kleckner   MCSymbol *DataBegin = MMI->getContext().createTempSymbol(),
20706f3406dfSReid Kleckner            *DataEnd = MMI->getContext().createTempSymbol();
20716f3406dfSReid Kleckner   OS.AddComment("Record length");
20726f3406dfSReid Kleckner   OS.emitAbsoluteSymbolDiff(DataEnd, DataBegin, 2);
20736f3406dfSReid Kleckner   OS.EmitLabel(DataBegin);
20747abd269aSDavid Majnemer   if (DIGV->isLocalToUnit()) {
2075a54fe1acSDavid Majnemer     if (GV->isThreadLocal()) {
2076a54fe1acSDavid Majnemer       OS.AddComment("Record kind: S_LTHREAD32");
2077a54fe1acSDavid Majnemer       OS.EmitIntValue(unsigned(SymbolKind::S_LTHREAD32), 2);
2078a54fe1acSDavid Majnemer     } else {
20797abd269aSDavid Majnemer       OS.AddComment("Record kind: S_LDATA32");
20807abd269aSDavid Majnemer       OS.EmitIntValue(unsigned(SymbolKind::S_LDATA32), 2);
2081a54fe1acSDavid Majnemer     }
2082a54fe1acSDavid Majnemer   } else {
2083a54fe1acSDavid Majnemer     if (GV->isThreadLocal()) {
2084a54fe1acSDavid Majnemer       OS.AddComment("Record kind: S_GTHREAD32");
2085a54fe1acSDavid Majnemer       OS.EmitIntValue(unsigned(SymbolKind::S_GTHREAD32), 2);
20867abd269aSDavid Majnemer     } else {
20876f3406dfSReid Kleckner       OS.AddComment("Record kind: S_GDATA32");
20886f3406dfSReid Kleckner       OS.EmitIntValue(unsigned(SymbolKind::S_GDATA32), 2);
20897abd269aSDavid Majnemer     }
2090a54fe1acSDavid Majnemer   }
20916f3406dfSReid Kleckner   OS.AddComment("Type");
20926f3406dfSReid Kleckner   OS.EmitIntValue(getCompleteTypeIndex(DIGV->getType()).getIndex(), 4);
20936f3406dfSReid Kleckner   OS.AddComment("DataOffset");
20946f3406dfSReid Kleckner   OS.EmitCOFFSecRel32(GVSym);
20956f3406dfSReid Kleckner   OS.AddComment("Segment");
20966f3406dfSReid Kleckner   OS.EmitCOFFSectionIndex(GVSym);
20976f3406dfSReid Kleckner   OS.AddComment("Name");
20986f3406dfSReid Kleckner   emitNullTerminatedSymbolName(OS, DIGV->getName());
20996f3406dfSReid Kleckner   OS.EmitLabel(DataEnd);
21006f3406dfSReid Kleckner }
2101