1 //===--- MicrosoftMangle.cpp - Microsoft Visual C++ Name Mangling ---------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This provides C++ name mangling targeting the Microsoft Visual C++ ABI.
11 //
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/AST/Mangle.h"
15 #include "clang/AST/ASTContext.h"
16 #include "clang/AST/Attr.h"
17 #include "clang/AST/CXXInheritance.h"
18 #include "clang/AST/CharUnits.h"
19 #include "clang/AST/Decl.h"
20 #include "clang/AST/DeclCXX.h"
21 #include "clang/AST/DeclObjC.h"
22 #include "clang/AST/DeclOpenMP.h"
23 #include "clang/AST/DeclTemplate.h"
24 #include "clang/AST/Expr.h"
25 #include "clang/AST/ExprCXX.h"
26 #include "clang/AST/VTableBuilder.h"
27 #include "clang/Basic/ABI.h"
28 #include "clang/Basic/DiagnosticOptions.h"
29 #include "clang/Basic/TargetInfo.h"
30 #include "llvm/ADT/StringExtras.h"
31 #include "llvm/Support/JamCRC.h"
32 #include "llvm/Support/MD5.h"
33 #include "llvm/Support/MathExtras.h"
34 
35 using namespace clang;
36 
37 namespace {
38 
39 struct msvc_hashing_ostream : public llvm::raw_svector_ostream {
40   raw_ostream &OS;
41   llvm::SmallString<64> Buffer;
42 
43   msvc_hashing_ostream(raw_ostream &OS)
44       : llvm::raw_svector_ostream(Buffer), OS(OS) {}
45   ~msvc_hashing_ostream() override {
46     StringRef MangledName = str();
47     bool StartsWithEscape = MangledName.startswith("\01");
48     if (StartsWithEscape)
49       MangledName = MangledName.drop_front(1);
50     if (MangledName.size() <= 4096) {
51       OS << str();
52       return;
53     }
54 
55     llvm::MD5 Hasher;
56     llvm::MD5::MD5Result Hash;
57     Hasher.update(MangledName);
58     Hasher.final(Hash);
59 
60     SmallString<32> HexString;
61     llvm::MD5::stringifyResult(Hash, HexString);
62 
63     if (StartsWithEscape)
64       OS << '\01';
65     OS << "??@" << HexString << '@';
66   }
67 };
68 
69 static const DeclContext *
70 getLambdaDefaultArgumentDeclContext(const Decl *D) {
71   if (const auto *RD = dyn_cast<CXXRecordDecl>(D))
72     if (RD->isLambda())
73       if (const auto *Parm =
74               dyn_cast_or_null<ParmVarDecl>(RD->getLambdaContextDecl()))
75         return Parm->getDeclContext();
76   return nullptr;
77 }
78 
79 /// Retrieve the declaration context that should be used when mangling
80 /// the given declaration.
81 static const DeclContext *getEffectiveDeclContext(const Decl *D) {
82   // The ABI assumes that lambda closure types that occur within
83   // default arguments live in the context of the function. However, due to
84   // the way in which Clang parses and creates function declarations, this is
85   // not the case: the lambda closure type ends up living in the context
86   // where the function itself resides, because the function declaration itself
87   // had not yet been created. Fix the context here.
88   if (const auto *LDADC = getLambdaDefaultArgumentDeclContext(D))
89     return LDADC;
90 
91   // Perform the same check for block literals.
92   if (const BlockDecl *BD = dyn_cast<BlockDecl>(D)) {
93     if (ParmVarDecl *ContextParam =
94             dyn_cast_or_null<ParmVarDecl>(BD->getBlockManglingContextDecl()))
95       return ContextParam->getDeclContext();
96   }
97 
98   const DeclContext *DC = D->getDeclContext();
99   if (isa<CapturedDecl>(DC) || isa<OMPDeclareReductionDecl>(DC)) {
100     return getEffectiveDeclContext(cast<Decl>(DC));
101   }
102 
103   return DC->getRedeclContext();
104 }
105 
106 static const DeclContext *getEffectiveParentContext(const DeclContext *DC) {
107   return getEffectiveDeclContext(cast<Decl>(DC));
108 }
109 
110 static const FunctionDecl *getStructor(const NamedDecl *ND) {
111   if (const auto *FTD = dyn_cast<FunctionTemplateDecl>(ND))
112     return FTD->getTemplatedDecl()->getCanonicalDecl();
113 
114   const auto *FD = cast<FunctionDecl>(ND);
115   if (const auto *FTD = FD->getPrimaryTemplate())
116     return FTD->getTemplatedDecl()->getCanonicalDecl();
117 
118   return FD->getCanonicalDecl();
119 }
120 
121 /// MicrosoftMangleContextImpl - Overrides the default MangleContext for the
122 /// Microsoft Visual C++ ABI.
123 class MicrosoftMangleContextImpl : public MicrosoftMangleContext {
124   typedef std::pair<const DeclContext *, IdentifierInfo *> DiscriminatorKeyTy;
125   llvm::DenseMap<DiscriminatorKeyTy, unsigned> Discriminator;
126   llvm::DenseMap<const NamedDecl *, unsigned> Uniquifier;
127   llvm::DenseMap<const CXXRecordDecl *, unsigned> LambdaIds;
128   llvm::DenseMap<const NamedDecl *, unsigned> SEHFilterIds;
129   llvm::DenseMap<const NamedDecl *, unsigned> SEHFinallyIds;
130 
131 public:
132   MicrosoftMangleContextImpl(ASTContext &Context, DiagnosticsEngine &Diags)
133       : MicrosoftMangleContext(Context, Diags) {}
134   bool shouldMangleCXXName(const NamedDecl *D) override;
135   bool shouldMangleStringLiteral(const StringLiteral *SL) override;
136   void mangleCXXName(const NamedDecl *D, raw_ostream &Out) override;
137   void mangleVirtualMemPtrThunk(const CXXMethodDecl *MD,
138                                 const MethodVFTableLocation &ML,
139                                 raw_ostream &Out) override;
140   void mangleThunk(const CXXMethodDecl *MD, const ThunkInfo &Thunk,
141                    raw_ostream &) override;
142   void mangleCXXDtorThunk(const CXXDestructorDecl *DD, CXXDtorType Type,
143                           const ThisAdjustment &ThisAdjustment,
144                           raw_ostream &) override;
145   void mangleCXXVFTable(const CXXRecordDecl *Derived,
146                         ArrayRef<const CXXRecordDecl *> BasePath,
147                         raw_ostream &Out) override;
148   void mangleCXXVBTable(const CXXRecordDecl *Derived,
149                         ArrayRef<const CXXRecordDecl *> BasePath,
150                         raw_ostream &Out) override;
151   void mangleCXXVirtualDisplacementMap(const CXXRecordDecl *SrcRD,
152                                        const CXXRecordDecl *DstRD,
153                                        raw_ostream &Out) override;
154   void mangleCXXThrowInfo(QualType T, bool IsConst, bool IsVolatile,
155                           bool IsUnaligned, uint32_t NumEntries,
156                           raw_ostream &Out) override;
157   void mangleCXXCatchableTypeArray(QualType T, uint32_t NumEntries,
158                                    raw_ostream &Out) override;
159   void mangleCXXCatchableType(QualType T, const CXXConstructorDecl *CD,
160                               CXXCtorType CT, uint32_t Size, uint32_t NVOffset,
161                               int32_t VBPtrOffset, uint32_t VBIndex,
162                               raw_ostream &Out) override;
163   void mangleCXXRTTI(QualType T, raw_ostream &Out) override;
164   void mangleCXXRTTIName(QualType T, raw_ostream &Out) override;
165   void mangleCXXRTTIBaseClassDescriptor(const CXXRecordDecl *Derived,
166                                         uint32_t NVOffset, int32_t VBPtrOffset,
167                                         uint32_t VBTableOffset, uint32_t Flags,
168                                         raw_ostream &Out) override;
169   void mangleCXXRTTIBaseClassArray(const CXXRecordDecl *Derived,
170                                    raw_ostream &Out) override;
171   void mangleCXXRTTIClassHierarchyDescriptor(const CXXRecordDecl *Derived,
172                                              raw_ostream &Out) override;
173   void
174   mangleCXXRTTICompleteObjectLocator(const CXXRecordDecl *Derived,
175                                      ArrayRef<const CXXRecordDecl *> BasePath,
176                                      raw_ostream &Out) override;
177   void mangleTypeName(QualType T, raw_ostream &) override;
178   void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type,
179                      raw_ostream &) override;
180   void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type,
181                      raw_ostream &) override;
182   void mangleReferenceTemporary(const VarDecl *, unsigned ManglingNumber,
183                                 raw_ostream &) override;
184   void mangleStaticGuardVariable(const VarDecl *D, raw_ostream &Out) override;
185   void mangleThreadSafeStaticGuardVariable(const VarDecl *D, unsigned GuardNum,
186                                            raw_ostream &Out) override;
187   void mangleDynamicInitializer(const VarDecl *D, raw_ostream &Out) override;
188   void mangleDynamicAtExitDestructor(const VarDecl *D,
189                                      raw_ostream &Out) override;
190   void mangleSEHFilterExpression(const NamedDecl *EnclosingDecl,
191                                  raw_ostream &Out) override;
192   void mangleSEHFinallyBlock(const NamedDecl *EnclosingDecl,
193                              raw_ostream &Out) override;
194   void mangleStringLiteral(const StringLiteral *SL, raw_ostream &Out) override;
195   bool getNextDiscriminator(const NamedDecl *ND, unsigned &disc) {
196     const DeclContext *DC = getEffectiveDeclContext(ND);
197     if (!DC->isFunctionOrMethod())
198       return false;
199 
200     // Lambda closure types are already numbered, give out a phony number so
201     // that they demangle nicely.
202     if (const auto *RD = dyn_cast<CXXRecordDecl>(ND)) {
203       if (RD->isLambda()) {
204         disc = 1;
205         return true;
206       }
207     }
208 
209     // Use the canonical number for externally visible decls.
210     if (ND->isExternallyVisible()) {
211       disc = getASTContext().getManglingNumber(ND);
212       return true;
213     }
214 
215     // Anonymous tags are already numbered.
216     if (const TagDecl *Tag = dyn_cast<TagDecl>(ND)) {
217       if (!Tag->hasNameForLinkage() &&
218           !getASTContext().getDeclaratorForUnnamedTagDecl(Tag) &&
219           !getASTContext().getTypedefNameForUnnamedTagDecl(Tag))
220         return false;
221     }
222 
223     // Make up a reasonable number for internal decls.
224     unsigned &discriminator = Uniquifier[ND];
225     if (!discriminator)
226       discriminator = ++Discriminator[std::make_pair(DC, ND->getIdentifier())];
227     disc = discriminator + 1;
228     return true;
229   }
230 
231   unsigned getLambdaId(const CXXRecordDecl *RD) {
232     assert(RD->isLambda() && "RD must be a lambda!");
233     assert(!RD->isExternallyVisible() && "RD must not be visible!");
234     assert(RD->getLambdaManglingNumber() == 0 &&
235            "RD must not have a mangling number!");
236     std::pair<llvm::DenseMap<const CXXRecordDecl *, unsigned>::iterator, bool>
237         Result = LambdaIds.insert(std::make_pair(RD, LambdaIds.size()));
238     return Result.first->second;
239   }
240 
241 private:
242   void mangleInitFiniStub(const VarDecl *D, char CharCode, raw_ostream &Out);
243 };
244 
245 /// MicrosoftCXXNameMangler - Manage the mangling of a single name for the
246 /// Microsoft Visual C++ ABI.
247 class MicrosoftCXXNameMangler {
248   MicrosoftMangleContextImpl &Context;
249   raw_ostream &Out;
250 
251   /// The "structor" is the top-level declaration being mangled, if
252   /// that's not a template specialization; otherwise it's the pattern
253   /// for that specialization.
254   const NamedDecl *Structor;
255   unsigned StructorType;
256 
257   typedef llvm::SmallVector<std::string, 10> BackRefVec;
258   BackRefVec NameBackReferences;
259 
260   typedef llvm::DenseMap<const void *, unsigned> ArgBackRefMap;
261   ArgBackRefMap TypeBackReferences;
262 
263   typedef std::set<int> PassObjectSizeArgsSet;
264   PassObjectSizeArgsSet PassObjectSizeArgs;
265 
266   ASTContext &getASTContext() const { return Context.getASTContext(); }
267 
268   // FIXME: If we add support for __ptr32/64 qualifiers, then we should push
269   // this check into mangleQualifiers().
270   const bool PointersAre64Bit;
271 
272 public:
273   enum QualifierMangleMode { QMM_Drop, QMM_Mangle, QMM_Escape, QMM_Result };
274 
275   MicrosoftCXXNameMangler(MicrosoftMangleContextImpl &C, raw_ostream &Out_)
276       : Context(C), Out(Out_), Structor(nullptr), StructorType(-1),
277         PointersAre64Bit(C.getASTContext().getTargetInfo().getPointerWidth(0) ==
278                          64) {}
279 
280   MicrosoftCXXNameMangler(MicrosoftMangleContextImpl &C, raw_ostream &Out_,
281                           const CXXConstructorDecl *D, CXXCtorType Type)
282       : Context(C), Out(Out_), Structor(getStructor(D)), StructorType(Type),
283         PointersAre64Bit(C.getASTContext().getTargetInfo().getPointerWidth(0) ==
284                          64) {}
285 
286   MicrosoftCXXNameMangler(MicrosoftMangleContextImpl &C, raw_ostream &Out_,
287                           const CXXDestructorDecl *D, CXXDtorType Type)
288       : Context(C), Out(Out_), Structor(getStructor(D)), StructorType(Type),
289         PointersAre64Bit(C.getASTContext().getTargetInfo().getPointerWidth(0) ==
290                          64) {}
291 
292   raw_ostream &getStream() const { return Out; }
293 
294   void mangle(const NamedDecl *D, StringRef Prefix = "?");
295   void mangleName(const NamedDecl *ND);
296   void mangleFunctionEncoding(const FunctionDecl *FD, bool ShouldMangle);
297   void mangleVariableEncoding(const VarDecl *VD);
298   void mangleMemberDataPointer(const CXXRecordDecl *RD, const ValueDecl *VD);
299   void mangleMemberFunctionPointer(const CXXRecordDecl *RD,
300                                    const CXXMethodDecl *MD);
301   void mangleVirtualMemPtrThunk(const CXXMethodDecl *MD,
302                                 const MethodVFTableLocation &ML);
303   void mangleNumber(int64_t Number);
304   void mangleTagTypeKind(TagTypeKind TK);
305   void mangleArtificalTagType(TagTypeKind TK, StringRef UnqualifiedName,
306                               ArrayRef<StringRef> NestedNames = None);
307   void mangleType(QualType T, SourceRange Range,
308                   QualifierMangleMode QMM = QMM_Mangle);
309   void mangleFunctionType(const FunctionType *T,
310                           const FunctionDecl *D = nullptr,
311                           bool ForceThisQuals = false);
312   void mangleNestedName(const NamedDecl *ND);
313 
314 private:
315   bool isStructorDecl(const NamedDecl *ND) const {
316     return ND == Structor || getStructor(ND) == Structor;
317   }
318 
319   void mangleUnqualifiedName(const NamedDecl *ND) {
320     mangleUnqualifiedName(ND, ND->getDeclName());
321   }
322   void mangleUnqualifiedName(const NamedDecl *ND, DeclarationName Name);
323   void mangleSourceName(StringRef Name);
324   void mangleOperatorName(OverloadedOperatorKind OO, SourceLocation Loc);
325   void mangleCXXDtorType(CXXDtorType T);
326   void mangleQualifiers(Qualifiers Quals, bool IsMember);
327   void mangleRefQualifier(RefQualifierKind RefQualifier);
328   void manglePointerCVQualifiers(Qualifiers Quals);
329   void manglePointerExtQualifiers(Qualifiers Quals, QualType PointeeType);
330 
331   void mangleUnscopedTemplateName(const TemplateDecl *ND);
332   void
333   mangleTemplateInstantiationName(const TemplateDecl *TD,
334                                   const TemplateArgumentList &TemplateArgs);
335   void mangleObjCMethodName(const ObjCMethodDecl *MD);
336 
337   void mangleArgumentType(QualType T, SourceRange Range);
338   void manglePassObjectSizeArg(const PassObjectSizeAttr *POSA);
339 
340   bool isArtificialTagType(QualType T) const;
341 
342   // Declare manglers for every type class.
343 #define ABSTRACT_TYPE(CLASS, PARENT)
344 #define NON_CANONICAL_TYPE(CLASS, PARENT)
345 #define TYPE(CLASS, PARENT) void mangleType(const CLASS##Type *T, \
346                                             Qualifiers Quals, \
347                                             SourceRange Range);
348 #include "clang/AST/TypeNodes.def"
349 #undef ABSTRACT_TYPE
350 #undef NON_CANONICAL_TYPE
351 #undef TYPE
352 
353   void mangleType(const TagDecl *TD);
354   void mangleDecayedArrayType(const ArrayType *T);
355   void mangleArrayType(const ArrayType *T);
356   void mangleFunctionClass(const FunctionDecl *FD);
357   void mangleCallingConvention(CallingConv CC);
358   void mangleCallingConvention(const FunctionType *T);
359   void mangleIntegerLiteral(const llvm::APSInt &Number, bool IsBoolean);
360   void mangleExpression(const Expr *E);
361   void mangleThrowSpecification(const FunctionProtoType *T);
362 
363   void mangleTemplateArgs(const TemplateDecl *TD,
364                           const TemplateArgumentList &TemplateArgs);
365   void mangleTemplateArg(const TemplateDecl *TD, const TemplateArgument &TA,
366                          const NamedDecl *Parm);
367 
368   void mangleObjCProtocol(const ObjCProtocolDecl *PD);
369   void mangleObjCLifetime(const QualType T, Qualifiers Quals,
370                           SourceRange Range);
371 };
372 }
373 
374 bool MicrosoftMangleContextImpl::shouldMangleCXXName(const NamedDecl *D) {
375   if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
376     LanguageLinkage L = FD->getLanguageLinkage();
377     // Overloadable functions need mangling.
378     if (FD->hasAttr<OverloadableAttr>())
379       return true;
380 
381     // The ABI expects that we would never mangle "typical" user-defined entry
382     // points regardless of visibility or freestanding-ness.
383     //
384     // N.B. This is distinct from asking about "main".  "main" has a lot of
385     // special rules associated with it in the standard while these
386     // user-defined entry points are outside of the purview of the standard.
387     // For example, there can be only one definition for "main" in a standards
388     // compliant program; however nothing forbids the existence of wmain and
389     // WinMain in the same translation unit.
390     if (FD->isMSVCRTEntryPoint())
391       return false;
392 
393     // C++ functions and those whose names are not a simple identifier need
394     // mangling.
395     if (!FD->getDeclName().isIdentifier() || L == CXXLanguageLinkage)
396       return true;
397 
398     // C functions are not mangled.
399     if (L == CLanguageLinkage)
400       return false;
401   }
402 
403   // Otherwise, no mangling is done outside C++ mode.
404   if (!getASTContext().getLangOpts().CPlusPlus)
405     return false;
406 
407   const VarDecl *VD = dyn_cast<VarDecl>(D);
408   if (VD && !isa<DecompositionDecl>(D)) {
409     // C variables are not mangled.
410     if (VD->isExternC())
411       return false;
412 
413     // Variables at global scope with non-internal linkage are not mangled.
414     const DeclContext *DC = getEffectiveDeclContext(D);
415     // Check for extern variable declared locally.
416     if (DC->isFunctionOrMethod() && D->hasLinkage())
417       while (!DC->isNamespace() && !DC->isTranslationUnit())
418         DC = getEffectiveParentContext(DC);
419 
420     if (DC->isTranslationUnit() && D->getFormalLinkage() == InternalLinkage &&
421         !isa<VarTemplateSpecializationDecl>(D) &&
422         D->getIdentifier() != nullptr)
423       return false;
424   }
425 
426   return true;
427 }
428 
429 bool
430 MicrosoftMangleContextImpl::shouldMangleStringLiteral(const StringLiteral *SL) {
431   return true;
432 }
433 
434 void MicrosoftCXXNameMangler::mangle(const NamedDecl *D, StringRef Prefix) {
435   // MSVC doesn't mangle C++ names the same way it mangles extern "C" names.
436   // Therefore it's really important that we don't decorate the
437   // name with leading underscores or leading/trailing at signs. So, by
438   // default, we emit an asm marker at the start so we get the name right.
439   // Callers can override this with a custom prefix.
440 
441   // <mangled-name> ::= ? <name> <type-encoding>
442   Out << Prefix;
443   mangleName(D);
444   if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
445     mangleFunctionEncoding(FD, Context.shouldMangleDeclName(FD));
446   else if (const VarDecl *VD = dyn_cast<VarDecl>(D))
447     mangleVariableEncoding(VD);
448   else if (!isa<ObjCInterfaceDecl>(D))
449     llvm_unreachable("Tried to mangle unexpected NamedDecl!");
450 }
451 
452 void MicrosoftCXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD,
453                                                      bool ShouldMangle) {
454   // <type-encoding> ::= <function-class> <function-type>
455 
456   // Since MSVC operates on the type as written and not the canonical type, it
457   // actually matters which decl we have here.  MSVC appears to choose the
458   // first, since it is most likely to be the declaration in a header file.
459   FD = FD->getFirstDecl();
460 
461   // We should never ever see a FunctionNoProtoType at this point.
462   // We don't even know how to mangle their types anyway :).
463   const FunctionProtoType *FT = FD->getType()->castAs<FunctionProtoType>();
464 
465   // extern "C" functions can hold entities that must be mangled.
466   // As it stands, these functions still need to get expressed in the full
467   // external name.  They have their class and type omitted, replaced with '9'.
468   if (ShouldMangle) {
469     // We would like to mangle all extern "C" functions using this additional
470     // component but this would break compatibility with MSVC's behavior.
471     // Instead, do this when we know that compatibility isn't important (in
472     // other words, when it is an overloaded extern "C" function).
473     if (FD->isExternC() && FD->hasAttr<OverloadableAttr>())
474       Out << "$$J0";
475 
476     mangleFunctionClass(FD);
477 
478     mangleFunctionType(FT, FD);
479   } else {
480     Out << '9';
481   }
482 }
483 
484 void MicrosoftCXXNameMangler::mangleVariableEncoding(const VarDecl *VD) {
485   // <type-encoding> ::= <storage-class> <variable-type>
486   // <storage-class> ::= 0  # private static member
487   //                 ::= 1  # protected static member
488   //                 ::= 2  # public static member
489   //                 ::= 3  # global
490   //                 ::= 4  # static local
491 
492   // The first character in the encoding (after the name) is the storage class.
493   if (VD->isStaticDataMember()) {
494     // If it's a static member, it also encodes the access level.
495     switch (VD->getAccess()) {
496       default:
497       case AS_private: Out << '0'; break;
498       case AS_protected: Out << '1'; break;
499       case AS_public: Out << '2'; break;
500     }
501   }
502   else if (!VD->isStaticLocal())
503     Out << '3';
504   else
505     Out << '4';
506   // Now mangle the type.
507   // <variable-type> ::= <type> <cvr-qualifiers>
508   //                 ::= <type> <pointee-cvr-qualifiers> # pointers, references
509   // Pointers and references are odd. The type of 'int * const foo;' gets
510   // mangled as 'QAHA' instead of 'PAHB', for example.
511   SourceRange SR = VD->getSourceRange();
512   QualType Ty = VD->getType();
513   if (Ty->isPointerType() || Ty->isReferenceType() ||
514       Ty->isMemberPointerType()) {
515     mangleType(Ty, SR, QMM_Drop);
516     manglePointerExtQualifiers(
517         Ty.getDesugaredType(getASTContext()).getLocalQualifiers(), QualType());
518     if (const MemberPointerType *MPT = Ty->getAs<MemberPointerType>()) {
519       mangleQualifiers(MPT->getPointeeType().getQualifiers(), true);
520       // Member pointers are suffixed with a back reference to the member
521       // pointer's class name.
522       mangleName(MPT->getClass()->getAsCXXRecordDecl());
523     } else
524       mangleQualifiers(Ty->getPointeeType().getQualifiers(), false);
525   } else if (const ArrayType *AT = getASTContext().getAsArrayType(Ty)) {
526     // Global arrays are funny, too.
527     mangleDecayedArrayType(AT);
528     if (AT->getElementType()->isArrayType())
529       Out << 'A';
530     else
531       mangleQualifiers(Ty.getQualifiers(), false);
532   } else {
533     mangleType(Ty, SR, QMM_Drop);
534     mangleQualifiers(Ty.getQualifiers(), false);
535   }
536 }
537 
538 void MicrosoftCXXNameMangler::mangleMemberDataPointer(const CXXRecordDecl *RD,
539                                                       const ValueDecl *VD) {
540   // <member-data-pointer> ::= <integer-literal>
541   //                       ::= $F <number> <number>
542   //                       ::= $G <number> <number> <number>
543 
544   int64_t FieldOffset;
545   int64_t VBTableOffset;
546   MSInheritanceAttr::Spelling IM = RD->getMSInheritanceModel();
547   if (VD) {
548     FieldOffset = getASTContext().getFieldOffset(VD);
549     assert(FieldOffset % getASTContext().getCharWidth() == 0 &&
550            "cannot take address of bitfield");
551     FieldOffset /= getASTContext().getCharWidth();
552 
553     VBTableOffset = 0;
554 
555     if (IM == MSInheritanceAttr::Keyword_virtual_inheritance)
556       FieldOffset -= getASTContext().getOffsetOfBaseWithVBPtr(RD).getQuantity();
557   } else {
558     FieldOffset = RD->nullFieldOffsetIsZero() ? 0 : -1;
559 
560     VBTableOffset = -1;
561   }
562 
563   char Code = '\0';
564   switch (IM) {
565   case MSInheritanceAttr::Keyword_single_inheritance:      Code = '0'; break;
566   case MSInheritanceAttr::Keyword_multiple_inheritance:    Code = '0'; break;
567   case MSInheritanceAttr::Keyword_virtual_inheritance:     Code = 'F'; break;
568   case MSInheritanceAttr::Keyword_unspecified_inheritance: Code = 'G'; break;
569   }
570 
571   Out << '$' << Code;
572 
573   mangleNumber(FieldOffset);
574 
575   // The C++ standard doesn't allow base-to-derived member pointer conversions
576   // in template parameter contexts, so the vbptr offset of data member pointers
577   // is always zero.
578   if (MSInheritanceAttr::hasVBPtrOffsetField(IM))
579     mangleNumber(0);
580   if (MSInheritanceAttr::hasVBTableOffsetField(IM))
581     mangleNumber(VBTableOffset);
582 }
583 
584 void
585 MicrosoftCXXNameMangler::mangleMemberFunctionPointer(const CXXRecordDecl *RD,
586                                                      const CXXMethodDecl *MD) {
587   // <member-function-pointer> ::= $1? <name>
588   //                           ::= $H? <name> <number>
589   //                           ::= $I? <name> <number> <number>
590   //                           ::= $J? <name> <number> <number> <number>
591 
592   MSInheritanceAttr::Spelling IM = RD->getMSInheritanceModel();
593 
594   char Code = '\0';
595   switch (IM) {
596   case MSInheritanceAttr::Keyword_single_inheritance:      Code = '1'; break;
597   case MSInheritanceAttr::Keyword_multiple_inheritance:    Code = 'H'; break;
598   case MSInheritanceAttr::Keyword_virtual_inheritance:     Code = 'I'; break;
599   case MSInheritanceAttr::Keyword_unspecified_inheritance: Code = 'J'; break;
600   }
601 
602   // If non-virtual, mangle the name.  If virtual, mangle as a virtual memptr
603   // thunk.
604   uint64_t NVOffset = 0;
605   uint64_t VBTableOffset = 0;
606   uint64_t VBPtrOffset = 0;
607   if (MD) {
608     Out << '$' << Code << '?';
609     if (MD->isVirtual()) {
610       MicrosoftVTableContext *VTContext =
611           cast<MicrosoftVTableContext>(getASTContext().getVTableContext());
612       MethodVFTableLocation ML =
613           VTContext->getMethodVFTableLocation(GlobalDecl(MD));
614       mangleVirtualMemPtrThunk(MD, ML);
615       NVOffset = ML.VFPtrOffset.getQuantity();
616       VBTableOffset = ML.VBTableIndex * 4;
617       if (ML.VBase) {
618         const ASTRecordLayout &Layout = getASTContext().getASTRecordLayout(RD);
619         VBPtrOffset = Layout.getVBPtrOffset().getQuantity();
620       }
621     } else {
622       mangleName(MD);
623       mangleFunctionEncoding(MD, /*ShouldMangle=*/true);
624     }
625 
626     if (VBTableOffset == 0 &&
627         IM == MSInheritanceAttr::Keyword_virtual_inheritance)
628       NVOffset -= getASTContext().getOffsetOfBaseWithVBPtr(RD).getQuantity();
629   } else {
630     // Null single inheritance member functions are encoded as a simple nullptr.
631     if (IM == MSInheritanceAttr::Keyword_single_inheritance) {
632       Out << "$0A@";
633       return;
634     }
635     if (IM == MSInheritanceAttr::Keyword_unspecified_inheritance)
636       VBTableOffset = -1;
637     Out << '$' << Code;
638   }
639 
640   if (MSInheritanceAttr::hasNVOffsetField(/*IsMemberFunction=*/true, IM))
641     mangleNumber(static_cast<uint32_t>(NVOffset));
642   if (MSInheritanceAttr::hasVBPtrOffsetField(IM))
643     mangleNumber(VBPtrOffset);
644   if (MSInheritanceAttr::hasVBTableOffsetField(IM))
645     mangleNumber(VBTableOffset);
646 }
647 
648 void MicrosoftCXXNameMangler::mangleVirtualMemPtrThunk(
649     const CXXMethodDecl *MD, const MethodVFTableLocation &ML) {
650   // Get the vftable offset.
651   CharUnits PointerWidth = getASTContext().toCharUnitsFromBits(
652       getASTContext().getTargetInfo().getPointerWidth(0));
653   uint64_t OffsetInVFTable = ML.Index * PointerWidth.getQuantity();
654 
655   Out << "?_9";
656   mangleName(MD->getParent());
657   Out << "$B";
658   mangleNumber(OffsetInVFTable);
659   Out << 'A';
660   mangleCallingConvention(MD->getType()->getAs<FunctionProtoType>());
661 }
662 
663 void MicrosoftCXXNameMangler::mangleName(const NamedDecl *ND) {
664   // <name> ::= <unscoped-name> {[<named-scope>]+ | [<nested-name>]}? @
665 
666   // Always start with the unqualified name.
667   mangleUnqualifiedName(ND);
668 
669   mangleNestedName(ND);
670 
671   // Terminate the whole name with an '@'.
672   Out << '@';
673 }
674 
675 void MicrosoftCXXNameMangler::mangleNumber(int64_t Number) {
676   // <non-negative integer> ::= A@              # when Number == 0
677   //                        ::= <decimal digit> # when 1 <= Number <= 10
678   //                        ::= <hex digit>+ @  # when Number >= 10
679   //
680   // <number>               ::= [?] <non-negative integer>
681 
682   uint64_t Value = static_cast<uint64_t>(Number);
683   if (Number < 0) {
684     Value = -Value;
685     Out << '?';
686   }
687 
688   if (Value == 0)
689     Out << "A@";
690   else if (Value >= 1 && Value <= 10)
691     Out << (Value - 1);
692   else {
693     // Numbers that are not encoded as decimal digits are represented as nibbles
694     // in the range of ASCII characters 'A' to 'P'.
695     // The number 0x123450 would be encoded as 'BCDEFA'
696     char EncodedNumberBuffer[sizeof(uint64_t) * 2];
697     MutableArrayRef<char> BufferRef(EncodedNumberBuffer);
698     MutableArrayRef<char>::reverse_iterator I = BufferRef.rbegin();
699     for (; Value != 0; Value >>= 4)
700       *I++ = 'A' + (Value & 0xf);
701     Out.write(I.base(), I - BufferRef.rbegin());
702     Out << '@';
703   }
704 }
705 
706 static const TemplateDecl *
707 isTemplate(const NamedDecl *ND, const TemplateArgumentList *&TemplateArgs) {
708   // Check if we have a function template.
709   if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) {
710     if (const TemplateDecl *TD = FD->getPrimaryTemplate()) {
711       TemplateArgs = FD->getTemplateSpecializationArgs();
712       return TD;
713     }
714   }
715 
716   // Check if we have a class template.
717   if (const ClassTemplateSpecializationDecl *Spec =
718           dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
719     TemplateArgs = &Spec->getTemplateArgs();
720     return Spec->getSpecializedTemplate();
721   }
722 
723   // Check if we have a variable template.
724   if (const VarTemplateSpecializationDecl *Spec =
725           dyn_cast<VarTemplateSpecializationDecl>(ND)) {
726     TemplateArgs = &Spec->getTemplateArgs();
727     return Spec->getSpecializedTemplate();
728   }
729 
730   return nullptr;
731 }
732 
733 void MicrosoftCXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND,
734                                                     DeclarationName Name) {
735   //  <unqualified-name> ::= <operator-name>
736   //                     ::= <ctor-dtor-name>
737   //                     ::= <source-name>
738   //                     ::= <template-name>
739 
740   // Check if we have a template.
741   const TemplateArgumentList *TemplateArgs = nullptr;
742   if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
743     // Function templates aren't considered for name back referencing.  This
744     // makes sense since function templates aren't likely to occur multiple
745     // times in a symbol.
746     if (isa<FunctionTemplateDecl>(TD)) {
747       mangleTemplateInstantiationName(TD, *TemplateArgs);
748       Out << '@';
749       return;
750     }
751 
752     // Here comes the tricky thing: if we need to mangle something like
753     //   void foo(A::X<Y>, B::X<Y>),
754     // the X<Y> part is aliased. However, if you need to mangle
755     //   void foo(A::X<A::Y>, A::X<B::Y>),
756     // the A::X<> part is not aliased.
757     // That said, from the mangler's perspective we have a structure like this:
758     //   namespace[s] -> type[ -> template-parameters]
759     // but from the Clang perspective we have
760     //   type [ -> template-parameters]
761     //      \-> namespace[s]
762     // What we do is we create a new mangler, mangle the same type (without
763     // a namespace suffix) to a string using the extra mangler and then use
764     // the mangled type name as a key to check the mangling of different types
765     // for aliasing.
766 
767     llvm::SmallString<64> TemplateMangling;
768     llvm::raw_svector_ostream Stream(TemplateMangling);
769     MicrosoftCXXNameMangler Extra(Context, Stream);
770     Extra.mangleTemplateInstantiationName(TD, *TemplateArgs);
771 
772     mangleSourceName(TemplateMangling);
773     return;
774   }
775 
776   switch (Name.getNameKind()) {
777     case DeclarationName::Identifier: {
778       if (const IdentifierInfo *II = Name.getAsIdentifierInfo()) {
779         mangleSourceName(II->getName());
780         break;
781       }
782 
783       // Otherwise, an anonymous entity.  We must have a declaration.
784       assert(ND && "mangling empty name without declaration");
785 
786       if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
787         if (NS->isAnonymousNamespace()) {
788           Out << "?A@";
789           break;
790         }
791       }
792 
793       if (const DecompositionDecl *DD = dyn_cast<DecompositionDecl>(ND)) {
794         // FIXME: Invented mangling for decomposition declarations:
795         //   [X,Y,Z]
796         // where X,Y,Z are the names of the bindings.
797         llvm::SmallString<128> Name("[");
798         for (auto *BD : DD->bindings()) {
799           if (Name.size() > 1)
800             Name += ',';
801           Name += BD->getDeclName().getAsIdentifierInfo()->getName();
802         }
803         Name += ']';
804         mangleSourceName(Name);
805         break;
806       }
807 
808       if (const VarDecl *VD = dyn_cast<VarDecl>(ND)) {
809         // We must have an anonymous union or struct declaration.
810         const CXXRecordDecl *RD = VD->getType()->getAsCXXRecordDecl();
811         assert(RD && "expected variable decl to have a record type");
812         // Anonymous types with no tag or typedef get the name of their
813         // declarator mangled in.  If they have no declarator, number them with
814         // a $S prefix.
815         llvm::SmallString<64> Name("$S");
816         // Get a unique id for the anonymous struct.
817         Name += llvm::utostr(Context.getAnonymousStructId(RD) + 1);
818         mangleSourceName(Name.str());
819         break;
820       }
821 
822       // We must have an anonymous struct.
823       const TagDecl *TD = cast<TagDecl>(ND);
824       if (const TypedefNameDecl *D = TD->getTypedefNameForAnonDecl()) {
825         assert(TD->getDeclContext() == D->getDeclContext() &&
826                "Typedef should not be in another decl context!");
827         assert(D->getDeclName().getAsIdentifierInfo() &&
828                "Typedef was not named!");
829         mangleSourceName(D->getDeclName().getAsIdentifierInfo()->getName());
830         break;
831       }
832 
833       if (const CXXRecordDecl *Record = dyn_cast<CXXRecordDecl>(TD)) {
834         if (Record->isLambda()) {
835           llvm::SmallString<10> Name("<lambda_");
836 
837           Decl *LambdaContextDecl = Record->getLambdaContextDecl();
838           unsigned LambdaManglingNumber = Record->getLambdaManglingNumber();
839           unsigned LambdaId;
840           const ParmVarDecl *Parm =
841               dyn_cast_or_null<ParmVarDecl>(LambdaContextDecl);
842           const FunctionDecl *Func =
843               Parm ? dyn_cast<FunctionDecl>(Parm->getDeclContext()) : nullptr;
844 
845           if (Func) {
846             unsigned DefaultArgNo =
847                 Func->getNumParams() - Parm->getFunctionScopeIndex();
848             Name += llvm::utostr(DefaultArgNo);
849             Name += "_";
850           }
851 
852           if (LambdaManglingNumber)
853             LambdaId = LambdaManglingNumber;
854           else
855             LambdaId = Context.getLambdaId(Record);
856 
857           Name += llvm::utostr(LambdaId);
858           Name += ">";
859 
860           mangleSourceName(Name);
861 
862           // If the context of a closure type is an initializer for a class
863           // member (static or nonstatic), it is encoded in a qualified name.
864           if (LambdaManglingNumber && LambdaContextDecl) {
865             if ((isa<VarDecl>(LambdaContextDecl) ||
866                  isa<FieldDecl>(LambdaContextDecl)) &&
867                 LambdaContextDecl->getDeclContext()->isRecord()) {
868               mangleUnqualifiedName(cast<NamedDecl>(LambdaContextDecl));
869             }
870           }
871           break;
872         }
873       }
874 
875       llvm::SmallString<64> Name;
876       if (DeclaratorDecl *DD =
877               Context.getASTContext().getDeclaratorForUnnamedTagDecl(TD)) {
878         // Anonymous types without a name for linkage purposes have their
879         // declarator mangled in if they have one.
880         Name += "<unnamed-type-";
881         Name += DD->getName();
882       } else if (TypedefNameDecl *TND =
883                      Context.getASTContext().getTypedefNameForUnnamedTagDecl(
884                          TD)) {
885         // Anonymous types without a name for linkage purposes have their
886         // associate typedef mangled in if they have one.
887         Name += "<unnamed-type-";
888         Name += TND->getName();
889       } else if (isa<EnumDecl>(TD) &&
890                  cast<EnumDecl>(TD)->enumerator_begin() !=
891                      cast<EnumDecl>(TD)->enumerator_end()) {
892         // Anonymous non-empty enums mangle in the first enumerator.
893         auto *ED = cast<EnumDecl>(TD);
894         Name += "<unnamed-enum-";
895         Name += ED->enumerator_begin()->getName();
896       } else {
897         // Otherwise, number the types using a $S prefix.
898         Name += "<unnamed-type-$S";
899         Name += llvm::utostr(Context.getAnonymousStructId(TD) + 1);
900       }
901       Name += ">";
902       mangleSourceName(Name.str());
903       break;
904     }
905 
906     case DeclarationName::ObjCZeroArgSelector:
907     case DeclarationName::ObjCOneArgSelector:
908     case DeclarationName::ObjCMultiArgSelector: {
909       // This is reachable only when constructing an outlined SEH finally
910       // block.  Nothing depends on this mangling and it's used only with
911       // functinos with internal linkage.
912       llvm::SmallString<64> Name;
913       mangleSourceName(Name.str());
914       break;
915     }
916 
917     case DeclarationName::CXXConstructorName:
918       if (isStructorDecl(ND)) {
919         if (StructorType == Ctor_CopyingClosure) {
920           Out << "?_O";
921           return;
922         }
923         if (StructorType == Ctor_DefaultClosure) {
924           Out << "?_F";
925           return;
926         }
927       }
928       Out << "?0";
929       return;
930 
931     case DeclarationName::CXXDestructorName:
932       if (isStructorDecl(ND))
933         // If the named decl is the C++ destructor we're mangling,
934         // use the type we were given.
935         mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
936       else
937         // Otherwise, use the base destructor name. This is relevant if a
938         // class with a destructor is declared within a destructor.
939         mangleCXXDtorType(Dtor_Base);
940       break;
941 
942     case DeclarationName::CXXConversionFunctionName:
943       // <operator-name> ::= ?B # (cast)
944       // The target type is encoded as the return type.
945       Out << "?B";
946       break;
947 
948     case DeclarationName::CXXOperatorName:
949       mangleOperatorName(Name.getCXXOverloadedOperator(), ND->getLocation());
950       break;
951 
952     case DeclarationName::CXXLiteralOperatorName: {
953       Out << "?__K";
954       mangleSourceName(Name.getCXXLiteralIdentifier()->getName());
955       break;
956     }
957 
958     case DeclarationName::CXXDeductionGuideName:
959       llvm_unreachable("Can't mangle a deduction guide name!");
960 
961     case DeclarationName::CXXUsingDirective:
962       llvm_unreachable("Can't mangle a using directive name!");
963   }
964 }
965 
966 // <postfix> ::= <unqualified-name> [<postfix>]
967 //           ::= <substitution> [<postfix>]
968 void MicrosoftCXXNameMangler::mangleNestedName(const NamedDecl *ND) {
969   const DeclContext *DC = getEffectiveDeclContext(ND);
970   while (!DC->isTranslationUnit()) {
971     if (isa<TagDecl>(ND) || isa<VarDecl>(ND)) {
972       unsigned Disc;
973       if (Context.getNextDiscriminator(ND, Disc)) {
974         Out << '?';
975         mangleNumber(Disc);
976         Out << '?';
977       }
978     }
979 
980     if (const BlockDecl *BD = dyn_cast<BlockDecl>(DC)) {
981       auto Discriminate =
982           [](StringRef Name, const unsigned Discriminator,
983              const unsigned ParameterDiscriminator) -> std::string {
984         std::string Buffer;
985         llvm::raw_string_ostream Stream(Buffer);
986         Stream << Name;
987         if (Discriminator)
988           Stream << '_' << Discriminator;
989         if (ParameterDiscriminator)
990           Stream << '_' << ParameterDiscriminator;
991         return Stream.str();
992       };
993 
994       unsigned Discriminator = BD->getBlockManglingNumber();
995       if (!Discriminator)
996         Discriminator = Context.getBlockId(BD, /*Local=*/false);
997 
998       // Mangle the parameter position as a discriminator to deal with unnamed
999       // parameters.  Rather than mangling the unqualified parameter name,
1000       // always use the position to give a uniform mangling.
1001       unsigned ParameterDiscriminator = 0;
1002       if (const auto *MC = BD->getBlockManglingContextDecl())
1003         if (const auto *P = dyn_cast<ParmVarDecl>(MC))
1004           if (const auto *F = dyn_cast<FunctionDecl>(P->getDeclContext()))
1005             ParameterDiscriminator =
1006                 F->getNumParams() - P->getFunctionScopeIndex();
1007 
1008       DC = getEffectiveDeclContext(BD);
1009 
1010       Out << '?';
1011       mangleSourceName(Discriminate("_block_invoke", Discriminator,
1012                                     ParameterDiscriminator));
1013       // If we have a block mangling context, encode that now.  This allows us
1014       // to discriminate between named static data initializers in the same
1015       // scope.  This is handled differently from parameters, which use
1016       // positions to discriminate between multiple instances.
1017       if (const auto *MC = BD->getBlockManglingContextDecl())
1018         if (!isa<ParmVarDecl>(MC))
1019           if (const auto *ND = dyn_cast<NamedDecl>(MC))
1020             mangleUnqualifiedName(ND);
1021       // MS ABI and Itanium manglings are in inverted scopes.  In the case of a
1022       // RecordDecl, mangle the entire scope hierarchy at this point rather than
1023       // just the unqualified name to get the ordering correct.
1024       if (const auto *RD = dyn_cast<RecordDecl>(DC))
1025         mangleName(RD);
1026       else
1027         Out << '@';
1028       // void __cdecl
1029       Out << "YAX";
1030       // struct __block_literal *
1031       Out << 'P';
1032       // __ptr64
1033       if (PointersAre64Bit)
1034         Out << 'E';
1035       Out << 'A';
1036       mangleArtificalTagType(TTK_Struct,
1037                              Discriminate("__block_literal", Discriminator,
1038                                           ParameterDiscriminator));
1039       Out << "@Z";
1040 
1041       // If the effective context was a Record, we have fully mangled the
1042       // qualified name and do not need to continue.
1043       if (isa<RecordDecl>(DC))
1044         break;
1045       continue;
1046     } else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(DC)) {
1047       mangleObjCMethodName(Method);
1048     } else if (isa<NamedDecl>(DC)) {
1049       ND = cast<NamedDecl>(DC);
1050       if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) {
1051         mangle(FD, "?");
1052         break;
1053       } else {
1054         mangleUnqualifiedName(ND);
1055         // Lambdas in default arguments conceptually belong to the function the
1056         // parameter corresponds to.
1057         if (const auto *LDADC = getLambdaDefaultArgumentDeclContext(ND)) {
1058           DC = LDADC;
1059           continue;
1060         }
1061       }
1062     }
1063     DC = DC->getParent();
1064   }
1065 }
1066 
1067 void MicrosoftCXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
1068   // Microsoft uses the names on the case labels for these dtor variants.  Clang
1069   // uses the Itanium terminology internally.  Everything in this ABI delegates
1070   // towards the base dtor.
1071   switch (T) {
1072   // <operator-name> ::= ?1  # destructor
1073   case Dtor_Base: Out << "?1"; return;
1074   // <operator-name> ::= ?_D # vbase destructor
1075   case Dtor_Complete: Out << "?_D"; return;
1076   // <operator-name> ::= ?_G # scalar deleting destructor
1077   case Dtor_Deleting: Out << "?_G"; return;
1078   // <operator-name> ::= ?_E # vector deleting destructor
1079   // FIXME: Add a vector deleting dtor type.  It goes in the vtable, so we need
1080   // it.
1081   case Dtor_Comdat:
1082     llvm_unreachable("not expecting a COMDAT");
1083   }
1084   llvm_unreachable("Unsupported dtor type?");
1085 }
1086 
1087 void MicrosoftCXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO,
1088                                                  SourceLocation Loc) {
1089   switch (OO) {
1090   //                     ?0 # constructor
1091   //                     ?1 # destructor
1092   // <operator-name> ::= ?2 # new
1093   case OO_New: Out << "?2"; break;
1094   // <operator-name> ::= ?3 # delete
1095   case OO_Delete: Out << "?3"; break;
1096   // <operator-name> ::= ?4 # =
1097   case OO_Equal: Out << "?4"; break;
1098   // <operator-name> ::= ?5 # >>
1099   case OO_GreaterGreater: Out << "?5"; break;
1100   // <operator-name> ::= ?6 # <<
1101   case OO_LessLess: Out << "?6"; break;
1102   // <operator-name> ::= ?7 # !
1103   case OO_Exclaim: Out << "?7"; break;
1104   // <operator-name> ::= ?8 # ==
1105   case OO_EqualEqual: Out << "?8"; break;
1106   // <operator-name> ::= ?9 # !=
1107   case OO_ExclaimEqual: Out << "?9"; break;
1108   // <operator-name> ::= ?A # []
1109   case OO_Subscript: Out << "?A"; break;
1110   //                     ?B # conversion
1111   // <operator-name> ::= ?C # ->
1112   case OO_Arrow: Out << "?C"; break;
1113   // <operator-name> ::= ?D # *
1114   case OO_Star: Out << "?D"; break;
1115   // <operator-name> ::= ?E # ++
1116   case OO_PlusPlus: Out << "?E"; break;
1117   // <operator-name> ::= ?F # --
1118   case OO_MinusMinus: Out << "?F"; break;
1119   // <operator-name> ::= ?G # -
1120   case OO_Minus: Out << "?G"; break;
1121   // <operator-name> ::= ?H # +
1122   case OO_Plus: Out << "?H"; break;
1123   // <operator-name> ::= ?I # &
1124   case OO_Amp: Out << "?I"; break;
1125   // <operator-name> ::= ?J # ->*
1126   case OO_ArrowStar: Out << "?J"; break;
1127   // <operator-name> ::= ?K # /
1128   case OO_Slash: Out << "?K"; break;
1129   // <operator-name> ::= ?L # %
1130   case OO_Percent: Out << "?L"; break;
1131   // <operator-name> ::= ?M # <
1132   case OO_Less: Out << "?M"; break;
1133   // <operator-name> ::= ?N # <=
1134   case OO_LessEqual: Out << "?N"; break;
1135   // <operator-name> ::= ?O # >
1136   case OO_Greater: Out << "?O"; break;
1137   // <operator-name> ::= ?P # >=
1138   case OO_GreaterEqual: Out << "?P"; break;
1139   // <operator-name> ::= ?Q # ,
1140   case OO_Comma: Out << "?Q"; break;
1141   // <operator-name> ::= ?R # ()
1142   case OO_Call: Out << "?R"; break;
1143   // <operator-name> ::= ?S # ~
1144   case OO_Tilde: Out << "?S"; break;
1145   // <operator-name> ::= ?T # ^
1146   case OO_Caret: Out << "?T"; break;
1147   // <operator-name> ::= ?U # |
1148   case OO_Pipe: Out << "?U"; break;
1149   // <operator-name> ::= ?V # &&
1150   case OO_AmpAmp: Out << "?V"; break;
1151   // <operator-name> ::= ?W # ||
1152   case OO_PipePipe: Out << "?W"; break;
1153   // <operator-name> ::= ?X # *=
1154   case OO_StarEqual: Out << "?X"; break;
1155   // <operator-name> ::= ?Y # +=
1156   case OO_PlusEqual: Out << "?Y"; break;
1157   // <operator-name> ::= ?Z # -=
1158   case OO_MinusEqual: Out << "?Z"; break;
1159   // <operator-name> ::= ?_0 # /=
1160   case OO_SlashEqual: Out << "?_0"; break;
1161   // <operator-name> ::= ?_1 # %=
1162   case OO_PercentEqual: Out << "?_1"; break;
1163   // <operator-name> ::= ?_2 # >>=
1164   case OO_GreaterGreaterEqual: Out << "?_2"; break;
1165   // <operator-name> ::= ?_3 # <<=
1166   case OO_LessLessEqual: Out << "?_3"; break;
1167   // <operator-name> ::= ?_4 # &=
1168   case OO_AmpEqual: Out << "?_4"; break;
1169   // <operator-name> ::= ?_5 # |=
1170   case OO_PipeEqual: Out << "?_5"; break;
1171   // <operator-name> ::= ?_6 # ^=
1172   case OO_CaretEqual: Out << "?_6"; break;
1173   //                     ?_7 # vftable
1174   //                     ?_8 # vbtable
1175   //                     ?_9 # vcall
1176   //                     ?_A # typeof
1177   //                     ?_B # local static guard
1178   //                     ?_C # string
1179   //                     ?_D # vbase destructor
1180   //                     ?_E # vector deleting destructor
1181   //                     ?_F # default constructor closure
1182   //                     ?_G # scalar deleting destructor
1183   //                     ?_H # vector constructor iterator
1184   //                     ?_I # vector destructor iterator
1185   //                     ?_J # vector vbase constructor iterator
1186   //                     ?_K # virtual displacement map
1187   //                     ?_L # eh vector constructor iterator
1188   //                     ?_M # eh vector destructor iterator
1189   //                     ?_N # eh vector vbase constructor iterator
1190   //                     ?_O # copy constructor closure
1191   //                     ?_P<name> # udt returning <name>
1192   //                     ?_Q # <unknown>
1193   //                     ?_R0 # RTTI Type Descriptor
1194   //                     ?_R1 # RTTI Base Class Descriptor at (a,b,c,d)
1195   //                     ?_R2 # RTTI Base Class Array
1196   //                     ?_R3 # RTTI Class Hierarchy Descriptor
1197   //                     ?_R4 # RTTI Complete Object Locator
1198   //                     ?_S # local vftable
1199   //                     ?_T # local vftable constructor closure
1200   // <operator-name> ::= ?_U # new[]
1201   case OO_Array_New: Out << "?_U"; break;
1202   // <operator-name> ::= ?_V # delete[]
1203   case OO_Array_Delete: Out << "?_V"; break;
1204   // <operator-name> ::= ?__L # co_await
1205   case OO_Coawait: Out << "?__L"; break;
1206 
1207   case OO_Spaceship: {
1208     // FIXME: Once MS picks a mangling, use it.
1209     DiagnosticsEngine &Diags = Context.getDiags();
1210     unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
1211       "cannot mangle this three-way comparison operator yet");
1212     Diags.Report(Loc, DiagID);
1213     break;
1214   }
1215 
1216   case OO_Conditional: {
1217     DiagnosticsEngine &Diags = Context.getDiags();
1218     unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
1219       "cannot mangle this conditional operator yet");
1220     Diags.Report(Loc, DiagID);
1221     break;
1222   }
1223 
1224   case OO_None:
1225   case NUM_OVERLOADED_OPERATORS:
1226     llvm_unreachable("Not an overloaded operator");
1227   }
1228 }
1229 
1230 void MicrosoftCXXNameMangler::mangleSourceName(StringRef Name) {
1231   // <source name> ::= <identifier> @
1232   BackRefVec::iterator Found =
1233       std::find(NameBackReferences.begin(), NameBackReferences.end(), Name);
1234   if (Found == NameBackReferences.end()) {
1235     if (NameBackReferences.size() < 10)
1236       NameBackReferences.push_back(Name);
1237     Out << Name << '@';
1238   } else {
1239     Out << (Found - NameBackReferences.begin());
1240   }
1241 }
1242 
1243 void MicrosoftCXXNameMangler::mangleObjCMethodName(const ObjCMethodDecl *MD) {
1244   Context.mangleObjCMethodName(MD, Out);
1245 }
1246 
1247 void MicrosoftCXXNameMangler::mangleTemplateInstantiationName(
1248     const TemplateDecl *TD, const TemplateArgumentList &TemplateArgs) {
1249   // <template-name> ::= <unscoped-template-name> <template-args>
1250   //                 ::= <substitution>
1251   // Always start with the unqualified name.
1252 
1253   // Templates have their own context for back references.
1254   ArgBackRefMap OuterArgsContext;
1255   BackRefVec OuterTemplateContext;
1256   PassObjectSizeArgsSet OuterPassObjectSizeArgs;
1257   NameBackReferences.swap(OuterTemplateContext);
1258   TypeBackReferences.swap(OuterArgsContext);
1259   PassObjectSizeArgs.swap(OuterPassObjectSizeArgs);
1260 
1261   mangleUnscopedTemplateName(TD);
1262   mangleTemplateArgs(TD, TemplateArgs);
1263 
1264   // Restore the previous back reference contexts.
1265   NameBackReferences.swap(OuterTemplateContext);
1266   TypeBackReferences.swap(OuterArgsContext);
1267   PassObjectSizeArgs.swap(OuterPassObjectSizeArgs);
1268 }
1269 
1270 void
1271 MicrosoftCXXNameMangler::mangleUnscopedTemplateName(const TemplateDecl *TD) {
1272   // <unscoped-template-name> ::= ?$ <unqualified-name>
1273   Out << "?$";
1274   mangleUnqualifiedName(TD);
1275 }
1276 
1277 void MicrosoftCXXNameMangler::mangleIntegerLiteral(const llvm::APSInt &Value,
1278                                                    bool IsBoolean) {
1279   // <integer-literal> ::= $0 <number>
1280   Out << "$0";
1281   // Make sure booleans are encoded as 0/1.
1282   if (IsBoolean && Value.getBoolValue())
1283     mangleNumber(1);
1284   else if (Value.isSigned())
1285     mangleNumber(Value.getSExtValue());
1286   else
1287     mangleNumber(Value.getZExtValue());
1288 }
1289 
1290 void MicrosoftCXXNameMangler::mangleExpression(const Expr *E) {
1291   // See if this is a constant expression.
1292   llvm::APSInt Value;
1293   if (E->isIntegerConstantExpr(Value, Context.getASTContext())) {
1294     mangleIntegerLiteral(Value, E->getType()->isBooleanType());
1295     return;
1296   }
1297 
1298   // Look through no-op casts like template parameter substitutions.
1299   E = E->IgnoreParenNoopCasts(Context.getASTContext());
1300 
1301   const CXXUuidofExpr *UE = nullptr;
1302   if (const UnaryOperator *UO = dyn_cast<UnaryOperator>(E)) {
1303     if (UO->getOpcode() == UO_AddrOf)
1304       UE = dyn_cast<CXXUuidofExpr>(UO->getSubExpr());
1305   } else
1306     UE = dyn_cast<CXXUuidofExpr>(E);
1307 
1308   if (UE) {
1309     // If we had to peek through an address-of operator, treat this like we are
1310     // dealing with a pointer type.  Otherwise, treat it like a const reference.
1311     //
1312     // N.B. This matches up with the handling of TemplateArgument::Declaration
1313     // in mangleTemplateArg
1314     if (UE == E)
1315       Out << "$E?";
1316     else
1317       Out << "$1?";
1318 
1319     // This CXXUuidofExpr is mangled as-if it were actually a VarDecl from
1320     // const __s_GUID _GUID_{lower case UUID with underscores}
1321     StringRef Uuid = UE->getUuidStr();
1322     std::string Name = "_GUID_" + Uuid.lower();
1323     std::replace(Name.begin(), Name.end(), '-', '_');
1324 
1325     mangleSourceName(Name);
1326     // Terminate the whole name with an '@'.
1327     Out << '@';
1328     // It's a global variable.
1329     Out << '3';
1330     // It's a struct called __s_GUID.
1331     mangleArtificalTagType(TTK_Struct, "__s_GUID");
1332     // It's const.
1333     Out << 'B';
1334     return;
1335   }
1336 
1337   // As bad as this diagnostic is, it's better than crashing.
1338   DiagnosticsEngine &Diags = Context.getDiags();
1339   unsigned DiagID = Diags.getCustomDiagID(
1340       DiagnosticsEngine::Error, "cannot yet mangle expression type %0");
1341   Diags.Report(E->getExprLoc(), DiagID) << E->getStmtClassName()
1342                                         << E->getSourceRange();
1343 }
1344 
1345 void MicrosoftCXXNameMangler::mangleTemplateArgs(
1346     const TemplateDecl *TD, const TemplateArgumentList &TemplateArgs) {
1347   // <template-args> ::= <template-arg>+
1348   const TemplateParameterList *TPL = TD->getTemplateParameters();
1349   assert(TPL->size() == TemplateArgs.size() &&
1350          "size mismatch between args and parms!");
1351 
1352   unsigned Idx = 0;
1353   for (const TemplateArgument &TA : TemplateArgs.asArray())
1354     mangleTemplateArg(TD, TA, TPL->getParam(Idx++));
1355 }
1356 
1357 void MicrosoftCXXNameMangler::mangleTemplateArg(const TemplateDecl *TD,
1358                                                 const TemplateArgument &TA,
1359                                                 const NamedDecl *Parm) {
1360   // <template-arg> ::= <type>
1361   //                ::= <integer-literal>
1362   //                ::= <member-data-pointer>
1363   //                ::= <member-function-pointer>
1364   //                ::= $E? <name> <type-encoding>
1365   //                ::= $1? <name> <type-encoding>
1366   //                ::= $0A@
1367   //                ::= <template-args>
1368 
1369   switch (TA.getKind()) {
1370   case TemplateArgument::Null:
1371     llvm_unreachable("Can't mangle null template arguments!");
1372   case TemplateArgument::TemplateExpansion:
1373     llvm_unreachable("Can't mangle template expansion arguments!");
1374   case TemplateArgument::Type: {
1375     QualType T = TA.getAsType();
1376     mangleType(T, SourceRange(), QMM_Escape);
1377     break;
1378   }
1379   case TemplateArgument::Declaration: {
1380     const NamedDecl *ND = TA.getAsDecl();
1381     if (isa<FieldDecl>(ND) || isa<IndirectFieldDecl>(ND)) {
1382       mangleMemberDataPointer(cast<CXXRecordDecl>(ND->getDeclContext())
1383                                   ->getMostRecentNonInjectedDecl(),
1384                               cast<ValueDecl>(ND));
1385     } else if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) {
1386       const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD);
1387       if (MD && MD->isInstance()) {
1388         mangleMemberFunctionPointer(
1389             MD->getParent()->getMostRecentNonInjectedDecl(), MD);
1390       } else {
1391         Out << "$1?";
1392         mangleName(FD);
1393         mangleFunctionEncoding(FD, /*ShouldMangle=*/true);
1394       }
1395     } else {
1396       mangle(ND, TA.getParamTypeForDecl()->isReferenceType() ? "$E?" : "$1?");
1397     }
1398     break;
1399   }
1400   case TemplateArgument::Integral:
1401     mangleIntegerLiteral(TA.getAsIntegral(),
1402                          TA.getIntegralType()->isBooleanType());
1403     break;
1404   case TemplateArgument::NullPtr: {
1405     QualType T = TA.getNullPtrType();
1406     if (const MemberPointerType *MPT = T->getAs<MemberPointerType>()) {
1407       const CXXRecordDecl *RD = MPT->getMostRecentCXXRecordDecl();
1408       if (MPT->isMemberFunctionPointerType() &&
1409           !isa<FunctionTemplateDecl>(TD)) {
1410         mangleMemberFunctionPointer(RD, nullptr);
1411         return;
1412       }
1413       if (MPT->isMemberDataPointer()) {
1414         if (!isa<FunctionTemplateDecl>(TD)) {
1415           mangleMemberDataPointer(RD, nullptr);
1416           return;
1417         }
1418         // nullptr data pointers are always represented with a single field
1419         // which is initialized with either 0 or -1.  Why -1?  Well, we need to
1420         // distinguish the case where the data member is at offset zero in the
1421         // record.
1422         // However, we are free to use 0 *if* we would use multiple fields for
1423         // non-nullptr member pointers.
1424         if (!RD->nullFieldOffsetIsZero()) {
1425           mangleIntegerLiteral(llvm::APSInt::get(-1), /*IsBoolean=*/false);
1426           return;
1427         }
1428       }
1429     }
1430     mangleIntegerLiteral(llvm::APSInt::getUnsigned(0), /*IsBoolean=*/false);
1431     break;
1432   }
1433   case TemplateArgument::Expression:
1434     mangleExpression(TA.getAsExpr());
1435     break;
1436   case TemplateArgument::Pack: {
1437     ArrayRef<TemplateArgument> TemplateArgs = TA.getPackAsArray();
1438     if (TemplateArgs.empty()) {
1439       if (isa<TemplateTypeParmDecl>(Parm) ||
1440           isa<TemplateTemplateParmDecl>(Parm))
1441         // MSVC 2015 changed the mangling for empty expanded template packs,
1442         // use the old mangling for link compatibility for old versions.
1443         Out << (Context.getASTContext().getLangOpts().isCompatibleWithMSVC(
1444                     LangOptions::MSVC2015)
1445                     ? "$$V"
1446                     : "$$$V");
1447       else if (isa<NonTypeTemplateParmDecl>(Parm))
1448         Out << "$S";
1449       else
1450         llvm_unreachable("unexpected template parameter decl!");
1451     } else {
1452       for (const TemplateArgument &PA : TemplateArgs)
1453         mangleTemplateArg(TD, PA, Parm);
1454     }
1455     break;
1456   }
1457   case TemplateArgument::Template: {
1458     const NamedDecl *ND =
1459         TA.getAsTemplate().getAsTemplateDecl()->getTemplatedDecl();
1460     if (const auto *TD = dyn_cast<TagDecl>(ND)) {
1461       mangleType(TD);
1462     } else if (isa<TypeAliasDecl>(ND)) {
1463       Out << "$$Y";
1464       mangleName(ND);
1465     } else {
1466       llvm_unreachable("unexpected template template NamedDecl!");
1467     }
1468     break;
1469   }
1470   }
1471 }
1472 
1473 void MicrosoftCXXNameMangler::mangleObjCProtocol(const ObjCProtocolDecl *PD) {
1474   llvm::SmallString<64> TemplateMangling;
1475   llvm::raw_svector_ostream Stream(TemplateMangling);
1476   MicrosoftCXXNameMangler Extra(Context, Stream);
1477 
1478   Stream << "?$";
1479   Extra.mangleSourceName("Protocol");
1480   Extra.mangleArtificalTagType(TTK_Struct, PD->getName());
1481 
1482   mangleArtificalTagType(TTK_Struct, TemplateMangling, {"__ObjC"});
1483 }
1484 
1485 void MicrosoftCXXNameMangler::mangleObjCLifetime(const QualType Type,
1486                                                  Qualifiers Quals,
1487                                                  SourceRange Range) {
1488   llvm::SmallString<64> TemplateMangling;
1489   llvm::raw_svector_ostream Stream(TemplateMangling);
1490   MicrosoftCXXNameMangler Extra(Context, Stream);
1491 
1492   Stream << "?$";
1493   switch (Quals.getObjCLifetime()) {
1494   case Qualifiers::OCL_None:
1495   case Qualifiers::OCL_ExplicitNone:
1496     break;
1497   case Qualifiers::OCL_Autoreleasing:
1498     Extra.mangleSourceName("Autoreleasing");
1499     break;
1500   case Qualifiers::OCL_Strong:
1501     Extra.mangleSourceName("Strong");
1502     break;
1503   case Qualifiers::OCL_Weak:
1504     Extra.mangleSourceName("Weak");
1505     break;
1506   }
1507   Extra.manglePointerCVQualifiers(Quals);
1508   Extra.manglePointerExtQualifiers(Quals, Type);
1509   Extra.mangleType(Type, Range);
1510 
1511   mangleArtificalTagType(TTK_Struct, TemplateMangling, {"__ObjC"});
1512 }
1513 
1514 void MicrosoftCXXNameMangler::mangleQualifiers(Qualifiers Quals,
1515                                                bool IsMember) {
1516   // <cvr-qualifiers> ::= [E] [F] [I] <base-cvr-qualifiers>
1517   // 'E' means __ptr64 (32-bit only); 'F' means __unaligned (32/64-bit only);
1518   // 'I' means __restrict (32/64-bit).
1519   // Note that the MSVC __restrict keyword isn't the same as the C99 restrict
1520   // keyword!
1521   // <base-cvr-qualifiers> ::= A  # near
1522   //                       ::= B  # near const
1523   //                       ::= C  # near volatile
1524   //                       ::= D  # near const volatile
1525   //                       ::= E  # far (16-bit)
1526   //                       ::= F  # far const (16-bit)
1527   //                       ::= G  # far volatile (16-bit)
1528   //                       ::= H  # far const volatile (16-bit)
1529   //                       ::= I  # huge (16-bit)
1530   //                       ::= J  # huge const (16-bit)
1531   //                       ::= K  # huge volatile (16-bit)
1532   //                       ::= L  # huge const volatile (16-bit)
1533   //                       ::= M <basis> # based
1534   //                       ::= N <basis> # based const
1535   //                       ::= O <basis> # based volatile
1536   //                       ::= P <basis> # based const volatile
1537   //                       ::= Q  # near member
1538   //                       ::= R  # near const member
1539   //                       ::= S  # near volatile member
1540   //                       ::= T  # near const volatile member
1541   //                       ::= U  # far member (16-bit)
1542   //                       ::= V  # far const member (16-bit)
1543   //                       ::= W  # far volatile member (16-bit)
1544   //                       ::= X  # far const volatile member (16-bit)
1545   //                       ::= Y  # huge member (16-bit)
1546   //                       ::= Z  # huge const member (16-bit)
1547   //                       ::= 0  # huge volatile member (16-bit)
1548   //                       ::= 1  # huge const volatile member (16-bit)
1549   //                       ::= 2 <basis> # based member
1550   //                       ::= 3 <basis> # based const member
1551   //                       ::= 4 <basis> # based volatile member
1552   //                       ::= 5 <basis> # based const volatile member
1553   //                       ::= 6  # near function (pointers only)
1554   //                       ::= 7  # far function (pointers only)
1555   //                       ::= 8  # near method (pointers only)
1556   //                       ::= 9  # far method (pointers only)
1557   //                       ::= _A <basis> # based function (pointers only)
1558   //                       ::= _B <basis> # based function (far?) (pointers only)
1559   //                       ::= _C <basis> # based method (pointers only)
1560   //                       ::= _D <basis> # based method (far?) (pointers only)
1561   //                       ::= _E # block (Clang)
1562   // <basis> ::= 0 # __based(void)
1563   //         ::= 1 # __based(segment)?
1564   //         ::= 2 <name> # __based(name)
1565   //         ::= 3 # ?
1566   //         ::= 4 # ?
1567   //         ::= 5 # not really based
1568   bool HasConst = Quals.hasConst(),
1569        HasVolatile = Quals.hasVolatile();
1570 
1571   if (!IsMember) {
1572     if (HasConst && HasVolatile) {
1573       Out << 'D';
1574     } else if (HasVolatile) {
1575       Out << 'C';
1576     } else if (HasConst) {
1577       Out << 'B';
1578     } else {
1579       Out << 'A';
1580     }
1581   } else {
1582     if (HasConst && HasVolatile) {
1583       Out << 'T';
1584     } else if (HasVolatile) {
1585       Out << 'S';
1586     } else if (HasConst) {
1587       Out << 'R';
1588     } else {
1589       Out << 'Q';
1590     }
1591   }
1592 
1593   // FIXME: For now, just drop all extension qualifiers on the floor.
1594 }
1595 
1596 void
1597 MicrosoftCXXNameMangler::mangleRefQualifier(RefQualifierKind RefQualifier) {
1598   // <ref-qualifier> ::= G                # lvalue reference
1599   //                 ::= H                # rvalue-reference
1600   switch (RefQualifier) {
1601   case RQ_None:
1602     break;
1603 
1604   case RQ_LValue:
1605     Out << 'G';
1606     break;
1607 
1608   case RQ_RValue:
1609     Out << 'H';
1610     break;
1611   }
1612 }
1613 
1614 void MicrosoftCXXNameMangler::manglePointerExtQualifiers(Qualifiers Quals,
1615                                                          QualType PointeeType) {
1616   if (PointersAre64Bit &&
1617       (PointeeType.isNull() || !PointeeType->isFunctionType()))
1618     Out << 'E';
1619 
1620   if (Quals.hasRestrict())
1621     Out << 'I';
1622 
1623   if (Quals.hasUnaligned() ||
1624       (!PointeeType.isNull() && PointeeType.getLocalQualifiers().hasUnaligned()))
1625     Out << 'F';
1626 }
1627 
1628 void MicrosoftCXXNameMangler::manglePointerCVQualifiers(Qualifiers Quals) {
1629   // <pointer-cv-qualifiers> ::= P  # no qualifiers
1630   //                         ::= Q  # const
1631   //                         ::= R  # volatile
1632   //                         ::= S  # const volatile
1633   bool HasConst = Quals.hasConst(),
1634        HasVolatile = Quals.hasVolatile();
1635 
1636   if (HasConst && HasVolatile) {
1637     Out << 'S';
1638   } else if (HasVolatile) {
1639     Out << 'R';
1640   } else if (HasConst) {
1641     Out << 'Q';
1642   } else {
1643     Out << 'P';
1644   }
1645 }
1646 
1647 void MicrosoftCXXNameMangler::mangleArgumentType(QualType T,
1648                                                  SourceRange Range) {
1649   // MSVC will backreference two canonically equivalent types that have slightly
1650   // different manglings when mangled alone.
1651 
1652   // Decayed types do not match up with non-decayed versions of the same type.
1653   //
1654   // e.g.
1655   // void (*x)(void) will not form a backreference with void x(void)
1656   void *TypePtr;
1657   if (const auto *DT = T->getAs<DecayedType>()) {
1658     QualType OriginalType = DT->getOriginalType();
1659     // All decayed ArrayTypes should be treated identically; as-if they were
1660     // a decayed IncompleteArrayType.
1661     if (const auto *AT = getASTContext().getAsArrayType(OriginalType))
1662       OriginalType = getASTContext().getIncompleteArrayType(
1663           AT->getElementType(), AT->getSizeModifier(),
1664           AT->getIndexTypeCVRQualifiers());
1665 
1666     TypePtr = OriginalType.getCanonicalType().getAsOpaquePtr();
1667     // If the original parameter was textually written as an array,
1668     // instead treat the decayed parameter like it's const.
1669     //
1670     // e.g.
1671     // int [] -> int * const
1672     if (OriginalType->isArrayType())
1673       T = T.withConst();
1674   } else {
1675     TypePtr = T.getCanonicalType().getAsOpaquePtr();
1676   }
1677 
1678   ArgBackRefMap::iterator Found = TypeBackReferences.find(TypePtr);
1679 
1680   if (Found == TypeBackReferences.end()) {
1681     size_t OutSizeBefore = Out.tell();
1682 
1683     mangleType(T, Range, QMM_Drop);
1684 
1685     // See if it's worth creating a back reference.
1686     // Only types longer than 1 character are considered
1687     // and only 10 back references slots are available:
1688     bool LongerThanOneChar = (Out.tell() - OutSizeBefore > 1);
1689     if (LongerThanOneChar && TypeBackReferences.size() < 10) {
1690       size_t Size = TypeBackReferences.size();
1691       TypeBackReferences[TypePtr] = Size;
1692     }
1693   } else {
1694     Out << Found->second;
1695   }
1696 }
1697 
1698 void MicrosoftCXXNameMangler::manglePassObjectSizeArg(
1699     const PassObjectSizeAttr *POSA) {
1700   int Type = POSA->getType();
1701 
1702   auto Iter = PassObjectSizeArgs.insert(Type).first;
1703   auto *TypePtr = (const void *)&*Iter;
1704   ArgBackRefMap::iterator Found = TypeBackReferences.find(TypePtr);
1705 
1706   if (Found == TypeBackReferences.end()) {
1707     mangleArtificalTagType(TTK_Enum, "__pass_object_size" + llvm::utostr(Type),
1708                            {"__clang"});
1709 
1710     if (TypeBackReferences.size() < 10) {
1711       size_t Size = TypeBackReferences.size();
1712       TypeBackReferences[TypePtr] = Size;
1713     }
1714   } else {
1715     Out << Found->second;
1716   }
1717 }
1718 
1719 void MicrosoftCXXNameMangler::mangleType(QualType T, SourceRange Range,
1720                                          QualifierMangleMode QMM) {
1721   // Don't use the canonical types.  MSVC includes things like 'const' on
1722   // pointer arguments to function pointers that canonicalization strips away.
1723   T = T.getDesugaredType(getASTContext());
1724   Qualifiers Quals = T.getLocalQualifiers();
1725   if (const ArrayType *AT = getASTContext().getAsArrayType(T)) {
1726     // If there were any Quals, getAsArrayType() pushed them onto the array
1727     // element type.
1728     if (QMM == QMM_Mangle)
1729       Out << 'A';
1730     else if (QMM == QMM_Escape || QMM == QMM_Result)
1731       Out << "$$B";
1732     mangleArrayType(AT);
1733     return;
1734   }
1735 
1736   bool IsPointer = T->isAnyPointerType() || T->isMemberPointerType() ||
1737                    T->isReferenceType() || T->isBlockPointerType();
1738 
1739   switch (QMM) {
1740   case QMM_Drop:
1741     if (Quals.hasObjCLifetime())
1742       Quals = Quals.withoutObjCLifetime();
1743     break;
1744   case QMM_Mangle:
1745     if (const FunctionType *FT = dyn_cast<FunctionType>(T)) {
1746       Out << '6';
1747       mangleFunctionType(FT);
1748       return;
1749     }
1750     mangleQualifiers(Quals, false);
1751     break;
1752   case QMM_Escape:
1753     if (!IsPointer && Quals) {
1754       Out << "$$C";
1755       mangleQualifiers(Quals, false);
1756     }
1757     break;
1758   case QMM_Result:
1759     // Presence of __unaligned qualifier shouldn't affect mangling here.
1760     Quals.removeUnaligned();
1761     if (Quals.hasObjCLifetime())
1762       Quals = Quals.withoutObjCLifetime();
1763     if ((!IsPointer && Quals) || isa<TagType>(T) || isArtificialTagType(T)) {
1764       Out << '?';
1765       mangleQualifiers(Quals, false);
1766     }
1767     break;
1768   }
1769 
1770   const Type *ty = T.getTypePtr();
1771 
1772   switch (ty->getTypeClass()) {
1773 #define ABSTRACT_TYPE(CLASS, PARENT)
1774 #define NON_CANONICAL_TYPE(CLASS, PARENT) \
1775   case Type::CLASS: \
1776     llvm_unreachable("can't mangle non-canonical type " #CLASS "Type"); \
1777     return;
1778 #define TYPE(CLASS, PARENT) \
1779   case Type::CLASS: \
1780     mangleType(cast<CLASS##Type>(ty), Quals, Range); \
1781     break;
1782 #include "clang/AST/TypeNodes.def"
1783 #undef ABSTRACT_TYPE
1784 #undef NON_CANONICAL_TYPE
1785 #undef TYPE
1786   }
1787 }
1788 
1789 void MicrosoftCXXNameMangler::mangleType(const BuiltinType *T, Qualifiers,
1790                                          SourceRange Range) {
1791   //  <type>         ::= <builtin-type>
1792   //  <builtin-type> ::= X  # void
1793   //                 ::= C  # signed char
1794   //                 ::= D  # char
1795   //                 ::= E  # unsigned char
1796   //                 ::= F  # short
1797   //                 ::= G  # unsigned short (or wchar_t if it's not a builtin)
1798   //                 ::= H  # int
1799   //                 ::= I  # unsigned int
1800   //                 ::= J  # long
1801   //                 ::= K  # unsigned long
1802   //                     L  # <none>
1803   //                 ::= M  # float
1804   //                 ::= N  # double
1805   //                 ::= O  # long double (__float80 is mangled differently)
1806   //                 ::= _J # long long, __int64
1807   //                 ::= _K # unsigned long long, __int64
1808   //                 ::= _L # __int128
1809   //                 ::= _M # unsigned __int128
1810   //                 ::= _N # bool
1811   //                     _O # <array in parameter>
1812   //                 ::= _T # __float80 (Intel)
1813   //                 ::= _S # char16_t
1814   //                 ::= _U # char32_t
1815   //                 ::= _W # wchar_t
1816   //                 ::= _Z # __float80 (Digital Mars)
1817   switch (T->getKind()) {
1818   case BuiltinType::Void:
1819     Out << 'X';
1820     break;
1821   case BuiltinType::SChar:
1822     Out << 'C';
1823     break;
1824   case BuiltinType::Char_U:
1825   case BuiltinType::Char_S:
1826     Out << 'D';
1827     break;
1828   case BuiltinType::UChar:
1829     Out << 'E';
1830     break;
1831   case BuiltinType::Short:
1832     Out << 'F';
1833     break;
1834   case BuiltinType::UShort:
1835     Out << 'G';
1836     break;
1837   case BuiltinType::Int:
1838     Out << 'H';
1839     break;
1840   case BuiltinType::UInt:
1841     Out << 'I';
1842     break;
1843   case BuiltinType::Long:
1844     Out << 'J';
1845     break;
1846   case BuiltinType::ULong:
1847     Out << 'K';
1848     break;
1849   case BuiltinType::Float:
1850     Out << 'M';
1851     break;
1852   case BuiltinType::Double:
1853     Out << 'N';
1854     break;
1855   // TODO: Determine size and mangle accordingly
1856   case BuiltinType::LongDouble:
1857     Out << 'O';
1858     break;
1859   case BuiltinType::LongLong:
1860     Out << "_J";
1861     break;
1862   case BuiltinType::ULongLong:
1863     Out << "_K";
1864     break;
1865   case BuiltinType::Int128:
1866     Out << "_L";
1867     break;
1868   case BuiltinType::UInt128:
1869     Out << "_M";
1870     break;
1871   case BuiltinType::Bool:
1872     Out << "_N";
1873     break;
1874   case BuiltinType::Char16:
1875     Out << "_S";
1876     break;
1877   case BuiltinType::Char32:
1878     Out << "_U";
1879     break;
1880   case BuiltinType::WChar_S:
1881   case BuiltinType::WChar_U:
1882     Out << "_W";
1883     break;
1884 
1885 #define BUILTIN_TYPE(Id, SingletonId)
1886 #define PLACEHOLDER_TYPE(Id, SingletonId) \
1887   case BuiltinType::Id:
1888 #include "clang/AST/BuiltinTypes.def"
1889   case BuiltinType::Dependent:
1890     llvm_unreachable("placeholder types shouldn't get to name mangling");
1891 
1892   case BuiltinType::ObjCId:
1893     mangleArtificalTagType(TTK_Struct, ".objc_object");
1894     break;
1895   case BuiltinType::ObjCClass:
1896     mangleArtificalTagType(TTK_Struct, ".objc_class");
1897     break;
1898   case BuiltinType::ObjCSel:
1899     mangleArtificalTagType(TTK_Struct, ".objc_selector");
1900     break;
1901 
1902 #define IMAGE_TYPE(ImgType, Id, SingletonId, Access, Suffix) \
1903   case BuiltinType::Id: \
1904     Out << "PAUocl_" #ImgType "_" #Suffix "@@"; \
1905     break;
1906 #include "clang/Basic/OpenCLImageTypes.def"
1907   case BuiltinType::OCLSampler:
1908     Out << "PA";
1909     mangleArtificalTagType(TTK_Struct, "ocl_sampler");
1910     break;
1911   case BuiltinType::OCLEvent:
1912     Out << "PA";
1913     mangleArtificalTagType(TTK_Struct, "ocl_event");
1914     break;
1915   case BuiltinType::OCLClkEvent:
1916     Out << "PA";
1917     mangleArtificalTagType(TTK_Struct, "ocl_clkevent");
1918     break;
1919   case BuiltinType::OCLQueue:
1920     Out << "PA";
1921     mangleArtificalTagType(TTK_Struct, "ocl_queue");
1922     break;
1923   case BuiltinType::OCLReserveID:
1924     Out << "PA";
1925     mangleArtificalTagType(TTK_Struct, "ocl_reserveid");
1926     break;
1927 
1928   case BuiltinType::NullPtr:
1929     Out << "$$T";
1930     break;
1931 
1932   case BuiltinType::Float16:
1933     mangleArtificalTagType(TTK_Struct, "_Float16", {"__clang"});
1934     break;
1935 
1936   case BuiltinType::Half:
1937     mangleArtificalTagType(TTK_Struct, "_Half", {"__clang"});
1938     break;
1939 
1940   case BuiltinType::ShortAccum:
1941   case BuiltinType::Accum:
1942   case BuiltinType::LongAccum:
1943   case BuiltinType::UShortAccum:
1944   case BuiltinType::UAccum:
1945   case BuiltinType::ULongAccum:
1946   case BuiltinType::ShortFract:
1947   case BuiltinType::Fract:
1948   case BuiltinType::LongFract:
1949   case BuiltinType::UShortFract:
1950   case BuiltinType::UFract:
1951   case BuiltinType::ULongFract:
1952   case BuiltinType::SatShortAccum:
1953   case BuiltinType::SatAccum:
1954   case BuiltinType::SatLongAccum:
1955   case BuiltinType::SatUShortAccum:
1956   case BuiltinType::SatUAccum:
1957   case BuiltinType::SatULongAccum:
1958   case BuiltinType::SatShortFract:
1959   case BuiltinType::SatFract:
1960   case BuiltinType::SatLongFract:
1961   case BuiltinType::SatUShortFract:
1962   case BuiltinType::SatUFract:
1963   case BuiltinType::SatULongFract:
1964   case BuiltinType::Char8:
1965   case BuiltinType::Float128: {
1966     DiagnosticsEngine &Diags = Context.getDiags();
1967     unsigned DiagID = Diags.getCustomDiagID(
1968         DiagnosticsEngine::Error, "cannot mangle this built-in %0 type yet");
1969     Diags.Report(Range.getBegin(), DiagID)
1970         << T->getName(Context.getASTContext().getPrintingPolicy()) << Range;
1971     break;
1972   }
1973   }
1974 }
1975 
1976 // <type>          ::= <function-type>
1977 void MicrosoftCXXNameMangler::mangleType(const FunctionProtoType *T, Qualifiers,
1978                                          SourceRange) {
1979   // Structors only appear in decls, so at this point we know it's not a
1980   // structor type.
1981   // FIXME: This may not be lambda-friendly.
1982   if (T->getTypeQuals() || T->getRefQualifier() != RQ_None) {
1983     Out << "$$A8@@";
1984     mangleFunctionType(T, /*D=*/nullptr, /*ForceThisQuals=*/true);
1985   } else {
1986     Out << "$$A6";
1987     mangleFunctionType(T);
1988   }
1989 }
1990 void MicrosoftCXXNameMangler::mangleType(const FunctionNoProtoType *T,
1991                                          Qualifiers, SourceRange) {
1992   Out << "$$A6";
1993   mangleFunctionType(T);
1994 }
1995 
1996 void MicrosoftCXXNameMangler::mangleFunctionType(const FunctionType *T,
1997                                                  const FunctionDecl *D,
1998                                                  bool ForceThisQuals) {
1999   // <function-type> ::= <this-cvr-qualifiers> <calling-convention>
2000   //                     <return-type> <argument-list> <throw-spec>
2001   const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(T);
2002 
2003   SourceRange Range;
2004   if (D) Range = D->getSourceRange();
2005 
2006   bool IsInLambda = false;
2007   bool IsStructor = false, HasThisQuals = ForceThisQuals, IsCtorClosure = false;
2008   CallingConv CC = T->getCallConv();
2009   if (const CXXMethodDecl *MD = dyn_cast_or_null<CXXMethodDecl>(D)) {
2010     if (MD->getParent()->isLambda())
2011       IsInLambda = true;
2012     if (MD->isInstance())
2013       HasThisQuals = true;
2014     if (isa<CXXDestructorDecl>(MD)) {
2015       IsStructor = true;
2016     } else if (isa<CXXConstructorDecl>(MD)) {
2017       IsStructor = true;
2018       IsCtorClosure = (StructorType == Ctor_CopyingClosure ||
2019                        StructorType == Ctor_DefaultClosure) &&
2020                       isStructorDecl(MD);
2021       if (IsCtorClosure)
2022         CC = getASTContext().getDefaultCallingConvention(
2023             /*IsVariadic=*/false, /*IsCXXMethod=*/true);
2024     }
2025   }
2026 
2027   // If this is a C++ instance method, mangle the CVR qualifiers for the
2028   // this pointer.
2029   if (HasThisQuals) {
2030     Qualifiers Quals = Qualifiers::fromCVRUMask(Proto->getTypeQuals());
2031     manglePointerExtQualifiers(Quals, /*PointeeType=*/QualType());
2032     mangleRefQualifier(Proto->getRefQualifier());
2033     mangleQualifiers(Quals, /*IsMember=*/false);
2034   }
2035 
2036   mangleCallingConvention(CC);
2037 
2038   // <return-type> ::= <type>
2039   //               ::= @ # structors (they have no declared return type)
2040   if (IsStructor) {
2041     if (isa<CXXDestructorDecl>(D) && isStructorDecl(D)) {
2042       // The scalar deleting destructor takes an extra int argument which is not
2043       // reflected in the AST.
2044       if (StructorType == Dtor_Deleting) {
2045         Out << (PointersAre64Bit ? "PEAXI@Z" : "PAXI@Z");
2046         return;
2047       }
2048       // The vbase destructor returns void which is not reflected in the AST.
2049       if (StructorType == Dtor_Complete) {
2050         Out << "XXZ";
2051         return;
2052       }
2053     }
2054     if (IsCtorClosure) {
2055       // Default constructor closure and copy constructor closure both return
2056       // void.
2057       Out << 'X';
2058 
2059       if (StructorType == Ctor_DefaultClosure) {
2060         // Default constructor closure always has no arguments.
2061         Out << 'X';
2062       } else if (StructorType == Ctor_CopyingClosure) {
2063         // Copy constructor closure always takes an unqualified reference.
2064         mangleArgumentType(getASTContext().getLValueReferenceType(
2065                                Proto->getParamType(0)
2066                                    ->getAs<LValueReferenceType>()
2067                                    ->getPointeeType(),
2068                                /*SpelledAsLValue=*/true),
2069                            Range);
2070         Out << '@';
2071       } else {
2072         llvm_unreachable("unexpected constructor closure!");
2073       }
2074       Out << 'Z';
2075       return;
2076     }
2077     Out << '@';
2078   } else {
2079     QualType ResultType = T->getReturnType();
2080     if (const auto *AT =
2081             dyn_cast_or_null<AutoType>(ResultType->getContainedAutoType())) {
2082       Out << '?';
2083       mangleQualifiers(ResultType.getLocalQualifiers(), /*IsMember=*/false);
2084       Out << '?';
2085       assert(AT->getKeyword() != AutoTypeKeyword::GNUAutoType &&
2086              "shouldn't need to mangle __auto_type!");
2087       mangleSourceName(AT->isDecltypeAuto() ? "<decltype-auto>" : "<auto>");
2088       Out << '@';
2089     } else if (IsInLambda) {
2090       Out << '@';
2091     } else {
2092       if (ResultType->isVoidType())
2093         ResultType = ResultType.getUnqualifiedType();
2094       mangleType(ResultType, Range, QMM_Result);
2095     }
2096   }
2097 
2098   // <argument-list> ::= X # void
2099   //                 ::= <type>+ @
2100   //                 ::= <type>* Z # varargs
2101   if (!Proto) {
2102     // Function types without prototypes can arise when mangling a function type
2103     // within an overloadable function in C. We mangle these as the absence of
2104     // any parameter types (not even an empty parameter list).
2105     Out << '@';
2106   } else if (Proto->getNumParams() == 0 && !Proto->isVariadic()) {
2107     Out << 'X';
2108   } else {
2109     // Happens for function pointer type arguments for example.
2110     for (unsigned I = 0, E = Proto->getNumParams(); I != E; ++I) {
2111       mangleArgumentType(Proto->getParamType(I), Range);
2112       // Mangle each pass_object_size parameter as if it's a parameter of enum
2113       // type passed directly after the parameter with the pass_object_size
2114       // attribute. The aforementioned enum's name is __pass_object_size, and we
2115       // pretend it resides in a top-level namespace called __clang.
2116       //
2117       // FIXME: Is there a defined extension notation for the MS ABI, or is it
2118       // necessary to just cross our fingers and hope this type+namespace
2119       // combination doesn't conflict with anything?
2120       if (D)
2121         if (const auto *P = D->getParamDecl(I)->getAttr<PassObjectSizeAttr>())
2122           manglePassObjectSizeArg(P);
2123     }
2124     // <builtin-type>      ::= Z  # ellipsis
2125     if (Proto->isVariadic())
2126       Out << 'Z';
2127     else
2128       Out << '@';
2129   }
2130 
2131   mangleThrowSpecification(Proto);
2132 }
2133 
2134 void MicrosoftCXXNameMangler::mangleFunctionClass(const FunctionDecl *FD) {
2135   // <function-class>  ::= <member-function> E? # E designates a 64-bit 'this'
2136   //                                            # pointer. in 64-bit mode *all*
2137   //                                            # 'this' pointers are 64-bit.
2138   //                   ::= <global-function>
2139   // <member-function> ::= A # private: near
2140   //                   ::= B # private: far
2141   //                   ::= C # private: static near
2142   //                   ::= D # private: static far
2143   //                   ::= E # private: virtual near
2144   //                   ::= F # private: virtual far
2145   //                   ::= I # protected: near
2146   //                   ::= J # protected: far
2147   //                   ::= K # protected: static near
2148   //                   ::= L # protected: static far
2149   //                   ::= M # protected: virtual near
2150   //                   ::= N # protected: virtual far
2151   //                   ::= Q # public: near
2152   //                   ::= R # public: far
2153   //                   ::= S # public: static near
2154   //                   ::= T # public: static far
2155   //                   ::= U # public: virtual near
2156   //                   ::= V # public: virtual far
2157   // <global-function> ::= Y # global near
2158   //                   ::= Z # global far
2159   if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD)) {
2160     bool IsVirtual = MD->isVirtual();
2161     // When mangling vbase destructor variants, ignore whether or not the
2162     // underlying destructor was defined to be virtual.
2163     if (isa<CXXDestructorDecl>(MD) && isStructorDecl(MD) &&
2164         StructorType == Dtor_Complete) {
2165       IsVirtual = false;
2166     }
2167     switch (MD->getAccess()) {
2168       case AS_none:
2169         llvm_unreachable("Unsupported access specifier");
2170       case AS_private:
2171         if (MD->isStatic())
2172           Out << 'C';
2173         else if (IsVirtual)
2174           Out << 'E';
2175         else
2176           Out << 'A';
2177         break;
2178       case AS_protected:
2179         if (MD->isStatic())
2180           Out << 'K';
2181         else if (IsVirtual)
2182           Out << 'M';
2183         else
2184           Out << 'I';
2185         break;
2186       case AS_public:
2187         if (MD->isStatic())
2188           Out << 'S';
2189         else if (IsVirtual)
2190           Out << 'U';
2191         else
2192           Out << 'Q';
2193     }
2194   } else {
2195     Out << 'Y';
2196   }
2197 }
2198 void MicrosoftCXXNameMangler::mangleCallingConvention(CallingConv CC) {
2199   // <calling-convention> ::= A # __cdecl
2200   //                      ::= B # __export __cdecl
2201   //                      ::= C # __pascal
2202   //                      ::= D # __export __pascal
2203   //                      ::= E # __thiscall
2204   //                      ::= F # __export __thiscall
2205   //                      ::= G # __stdcall
2206   //                      ::= H # __export __stdcall
2207   //                      ::= I # __fastcall
2208   //                      ::= J # __export __fastcall
2209   //                      ::= Q # __vectorcall
2210   //                      ::= w # __regcall
2211   // The 'export' calling conventions are from a bygone era
2212   // (*cough*Win16*cough*) when functions were declared for export with
2213   // that keyword. (It didn't actually export them, it just made them so
2214   // that they could be in a DLL and somebody from another module could call
2215   // them.)
2216 
2217   switch (CC) {
2218     default:
2219       llvm_unreachable("Unsupported CC for mangling");
2220     case CC_Win64:
2221     case CC_X86_64SysV:
2222     case CC_C: Out << 'A'; break;
2223     case CC_X86Pascal: Out << 'C'; break;
2224     case CC_X86ThisCall: Out << 'E'; break;
2225     case CC_X86StdCall: Out << 'G'; break;
2226     case CC_X86FastCall: Out << 'I'; break;
2227     case CC_X86VectorCall: Out << 'Q'; break;
2228     case CC_Swift: Out << 'S'; break;
2229     case CC_PreserveMost: Out << 'U'; break;
2230     case CC_X86RegCall: Out << 'w'; break;
2231   }
2232 }
2233 void MicrosoftCXXNameMangler::mangleCallingConvention(const FunctionType *T) {
2234   mangleCallingConvention(T->getCallConv());
2235 }
2236 void MicrosoftCXXNameMangler::mangleThrowSpecification(
2237                                                 const FunctionProtoType *FT) {
2238   // <throw-spec> ::= Z # throw(...) (default)
2239   //              ::= @ # throw() or __declspec/__attribute__((nothrow))
2240   //              ::= <type>+
2241   // NOTE: Since the Microsoft compiler ignores throw specifications, they are
2242   // all actually mangled as 'Z'. (They're ignored because their associated
2243   // functionality isn't implemented, and probably never will be.)
2244   Out << 'Z';
2245 }
2246 
2247 void MicrosoftCXXNameMangler::mangleType(const UnresolvedUsingType *T,
2248                                          Qualifiers, SourceRange Range) {
2249   // Probably should be mangled as a template instantiation; need to see what
2250   // VC does first.
2251   DiagnosticsEngine &Diags = Context.getDiags();
2252   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2253     "cannot mangle this unresolved dependent type yet");
2254   Diags.Report(Range.getBegin(), DiagID)
2255     << Range;
2256 }
2257 
2258 // <type>        ::= <union-type> | <struct-type> | <class-type> | <enum-type>
2259 // <union-type>  ::= T <name>
2260 // <struct-type> ::= U <name>
2261 // <class-type>  ::= V <name>
2262 // <enum-type>   ::= W4 <name>
2263 void MicrosoftCXXNameMangler::mangleTagTypeKind(TagTypeKind TTK) {
2264   switch (TTK) {
2265     case TTK_Union:
2266       Out << 'T';
2267       break;
2268     case TTK_Struct:
2269     case TTK_Interface:
2270       Out << 'U';
2271       break;
2272     case TTK_Class:
2273       Out << 'V';
2274       break;
2275     case TTK_Enum:
2276       Out << "W4";
2277       break;
2278   }
2279 }
2280 void MicrosoftCXXNameMangler::mangleType(const EnumType *T, Qualifiers,
2281                                          SourceRange) {
2282   mangleType(cast<TagType>(T)->getDecl());
2283 }
2284 void MicrosoftCXXNameMangler::mangleType(const RecordType *T, Qualifiers,
2285                                          SourceRange) {
2286   mangleType(cast<TagType>(T)->getDecl());
2287 }
2288 void MicrosoftCXXNameMangler::mangleType(const TagDecl *TD) {
2289   mangleTagTypeKind(TD->getTagKind());
2290   mangleName(TD);
2291 }
2292 
2293 // If you add a call to this, consider updating isArtificialTagType() too.
2294 void MicrosoftCXXNameMangler::mangleArtificalTagType(
2295     TagTypeKind TK, StringRef UnqualifiedName,
2296     ArrayRef<StringRef> NestedNames) {
2297   // <name> ::= <unscoped-name> {[<named-scope>]+ | [<nested-name>]}? @
2298   mangleTagTypeKind(TK);
2299 
2300   // Always start with the unqualified name.
2301   mangleSourceName(UnqualifiedName);
2302 
2303   for (auto I = NestedNames.rbegin(), E = NestedNames.rend(); I != E; ++I)
2304     mangleSourceName(*I);
2305 
2306   // Terminate the whole name with an '@'.
2307   Out << '@';
2308 }
2309 
2310 // <type>       ::= <array-type>
2311 // <array-type> ::= <pointer-cvr-qualifiers> <cvr-qualifiers>
2312 //                  [Y <dimension-count> <dimension>+]
2313 //                  <element-type> # as global, E is never required
2314 // It's supposed to be the other way around, but for some strange reason, it
2315 // isn't. Today this behavior is retained for the sole purpose of backwards
2316 // compatibility.
2317 void MicrosoftCXXNameMangler::mangleDecayedArrayType(const ArrayType *T) {
2318   // This isn't a recursive mangling, so now we have to do it all in this
2319   // one call.
2320   manglePointerCVQualifiers(T->getElementType().getQualifiers());
2321   mangleType(T->getElementType(), SourceRange());
2322 }
2323 void MicrosoftCXXNameMangler::mangleType(const ConstantArrayType *T, Qualifiers,
2324                                          SourceRange) {
2325   llvm_unreachable("Should have been special cased");
2326 }
2327 void MicrosoftCXXNameMangler::mangleType(const VariableArrayType *T, Qualifiers,
2328                                          SourceRange) {
2329   llvm_unreachable("Should have been special cased");
2330 }
2331 void MicrosoftCXXNameMangler::mangleType(const DependentSizedArrayType *T,
2332                                          Qualifiers, SourceRange) {
2333   llvm_unreachable("Should have been special cased");
2334 }
2335 void MicrosoftCXXNameMangler::mangleType(const IncompleteArrayType *T,
2336                                          Qualifiers, SourceRange) {
2337   llvm_unreachable("Should have been special cased");
2338 }
2339 void MicrosoftCXXNameMangler::mangleArrayType(const ArrayType *T) {
2340   QualType ElementTy(T, 0);
2341   SmallVector<llvm::APInt, 3> Dimensions;
2342   for (;;) {
2343     if (ElementTy->isConstantArrayType()) {
2344       const ConstantArrayType *CAT =
2345           getASTContext().getAsConstantArrayType(ElementTy);
2346       Dimensions.push_back(CAT->getSize());
2347       ElementTy = CAT->getElementType();
2348     } else if (ElementTy->isIncompleteArrayType()) {
2349       const IncompleteArrayType *IAT =
2350           getASTContext().getAsIncompleteArrayType(ElementTy);
2351       Dimensions.push_back(llvm::APInt(32, 0));
2352       ElementTy = IAT->getElementType();
2353     } else if (ElementTy->isVariableArrayType()) {
2354       const VariableArrayType *VAT =
2355         getASTContext().getAsVariableArrayType(ElementTy);
2356       Dimensions.push_back(llvm::APInt(32, 0));
2357       ElementTy = VAT->getElementType();
2358     } else if (ElementTy->isDependentSizedArrayType()) {
2359       // The dependent expression has to be folded into a constant (TODO).
2360       const DependentSizedArrayType *DSAT =
2361         getASTContext().getAsDependentSizedArrayType(ElementTy);
2362       DiagnosticsEngine &Diags = Context.getDiags();
2363       unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2364         "cannot mangle this dependent-length array yet");
2365       Diags.Report(DSAT->getSizeExpr()->getExprLoc(), DiagID)
2366         << DSAT->getBracketsRange();
2367       return;
2368     } else {
2369       break;
2370     }
2371   }
2372   Out << 'Y';
2373   // <dimension-count> ::= <number> # number of extra dimensions
2374   mangleNumber(Dimensions.size());
2375   for (const llvm::APInt &Dimension : Dimensions)
2376     mangleNumber(Dimension.getLimitedValue());
2377   mangleType(ElementTy, SourceRange(), QMM_Escape);
2378 }
2379 
2380 // <type>                   ::= <pointer-to-member-type>
2381 // <pointer-to-member-type> ::= <pointer-cvr-qualifiers> <cvr-qualifiers>
2382 //                                                          <class name> <type>
2383 void MicrosoftCXXNameMangler::mangleType(const MemberPointerType *T,
2384                                          Qualifiers Quals, SourceRange Range) {
2385   QualType PointeeType = T->getPointeeType();
2386   manglePointerCVQualifiers(Quals);
2387   manglePointerExtQualifiers(Quals, PointeeType);
2388   if (const FunctionProtoType *FPT = PointeeType->getAs<FunctionProtoType>()) {
2389     Out << '8';
2390     mangleName(T->getClass()->castAs<RecordType>()->getDecl());
2391     mangleFunctionType(FPT, nullptr, true);
2392   } else {
2393     mangleQualifiers(PointeeType.getQualifiers(), true);
2394     mangleName(T->getClass()->castAs<RecordType>()->getDecl());
2395     mangleType(PointeeType, Range, QMM_Drop);
2396   }
2397 }
2398 
2399 void MicrosoftCXXNameMangler::mangleType(const TemplateTypeParmType *T,
2400                                          Qualifiers, SourceRange Range) {
2401   DiagnosticsEngine &Diags = Context.getDiags();
2402   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2403     "cannot mangle this template type parameter type yet");
2404   Diags.Report(Range.getBegin(), DiagID)
2405     << Range;
2406 }
2407 
2408 void MicrosoftCXXNameMangler::mangleType(const SubstTemplateTypeParmPackType *T,
2409                                          Qualifiers, SourceRange Range) {
2410   DiagnosticsEngine &Diags = Context.getDiags();
2411   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2412     "cannot mangle this substituted parameter pack yet");
2413   Diags.Report(Range.getBegin(), DiagID)
2414     << Range;
2415 }
2416 
2417 // <type> ::= <pointer-type>
2418 // <pointer-type> ::= E? <pointer-cvr-qualifiers> <cvr-qualifiers> <type>
2419 //                       # the E is required for 64-bit non-static pointers
2420 void MicrosoftCXXNameMangler::mangleType(const PointerType *T, Qualifiers Quals,
2421                                          SourceRange Range) {
2422   QualType PointeeType = T->getPointeeType();
2423   manglePointerCVQualifiers(Quals);
2424   manglePointerExtQualifiers(Quals, PointeeType);
2425   mangleType(PointeeType, Range);
2426 }
2427 
2428 void MicrosoftCXXNameMangler::mangleType(const ObjCObjectPointerType *T,
2429                                          Qualifiers Quals, SourceRange Range) {
2430   QualType PointeeType = T->getPointeeType();
2431   switch (Quals.getObjCLifetime()) {
2432   case Qualifiers::OCL_None:
2433   case Qualifiers::OCL_ExplicitNone:
2434     break;
2435   case Qualifiers::OCL_Autoreleasing:
2436   case Qualifiers::OCL_Strong:
2437   case Qualifiers::OCL_Weak:
2438     return mangleObjCLifetime(PointeeType, Quals, Range);
2439   }
2440   manglePointerCVQualifiers(Quals);
2441   manglePointerExtQualifiers(Quals, PointeeType);
2442   mangleType(PointeeType, Range);
2443 }
2444 
2445 // <type> ::= <reference-type>
2446 // <reference-type> ::= A E? <cvr-qualifiers> <type>
2447 //                 # the E is required for 64-bit non-static lvalue references
2448 void MicrosoftCXXNameMangler::mangleType(const LValueReferenceType *T,
2449                                          Qualifiers Quals, SourceRange Range) {
2450   QualType PointeeType = T->getPointeeType();
2451   assert(!Quals.hasConst() && !Quals.hasVolatile() && "unexpected qualifier!");
2452   Out << 'A';
2453   manglePointerExtQualifiers(Quals, PointeeType);
2454   mangleType(PointeeType, Range);
2455 }
2456 
2457 // <type> ::= <r-value-reference-type>
2458 // <r-value-reference-type> ::= $$Q E? <cvr-qualifiers> <type>
2459 //                 # the E is required for 64-bit non-static rvalue references
2460 void MicrosoftCXXNameMangler::mangleType(const RValueReferenceType *T,
2461                                          Qualifiers Quals, SourceRange Range) {
2462   QualType PointeeType = T->getPointeeType();
2463   assert(!Quals.hasConst() && !Quals.hasVolatile() && "unexpected qualifier!");
2464   Out << "$$Q";
2465   manglePointerExtQualifiers(Quals, PointeeType);
2466   mangleType(PointeeType, Range);
2467 }
2468 
2469 void MicrosoftCXXNameMangler::mangleType(const ComplexType *T, Qualifiers,
2470                                          SourceRange Range) {
2471   QualType ElementType = T->getElementType();
2472 
2473   llvm::SmallString<64> TemplateMangling;
2474   llvm::raw_svector_ostream Stream(TemplateMangling);
2475   MicrosoftCXXNameMangler Extra(Context, Stream);
2476   Stream << "?$";
2477   Extra.mangleSourceName("_Complex");
2478   Extra.mangleType(ElementType, Range, QMM_Escape);
2479 
2480   mangleArtificalTagType(TTK_Struct, TemplateMangling, {"__clang"});
2481 }
2482 
2483 // Returns true for types that mangleArtificalTagType() gets called for with
2484 // TTK_Union, TTK_Struct, TTK_Class and where compatibility with MSVC's
2485 // mangling matters.
2486 // (It doesn't matter for Objective-C types and the like that cl.exe doesn't
2487 // support.)
2488 bool MicrosoftCXXNameMangler::isArtificialTagType(QualType T) const {
2489   const Type *ty = T.getTypePtr();
2490   switch (ty->getTypeClass()) {
2491   default:
2492     return false;
2493 
2494   case Type::Vector: {
2495     // For ABI compatibility only __m64, __m128(id), and __m256(id) matter,
2496     // but since mangleType(VectorType*) always calls mangleArtificalTagType()
2497     // just always return true (the other vector types are clang-only).
2498     return true;
2499   }
2500   }
2501 }
2502 
2503 void MicrosoftCXXNameMangler::mangleType(const VectorType *T, Qualifiers Quals,
2504                                          SourceRange Range) {
2505   const BuiltinType *ET = T->getElementType()->getAs<BuiltinType>();
2506   assert(ET && "vectors with non-builtin elements are unsupported");
2507   uint64_t Width = getASTContext().getTypeSize(T);
2508   // Pattern match exactly the typedefs in our intrinsic headers.  Anything that
2509   // doesn't match the Intel types uses a custom mangling below.
2510   size_t OutSizeBefore = Out.tell();
2511   llvm::Triple::ArchType AT =
2512       getASTContext().getTargetInfo().getTriple().getArch();
2513   if (AT == llvm::Triple::x86 || AT == llvm::Triple::x86_64) {
2514     if (Width == 64 && ET->getKind() == BuiltinType::LongLong) {
2515       mangleArtificalTagType(TTK_Union, "__m64");
2516     } else if (Width >= 128) {
2517       if (ET->getKind() == BuiltinType::Float)
2518         mangleArtificalTagType(TTK_Union, "__m" + llvm::utostr(Width));
2519       else if (ET->getKind() == BuiltinType::LongLong)
2520         mangleArtificalTagType(TTK_Union, "__m" + llvm::utostr(Width) + 'i');
2521       else if (ET->getKind() == BuiltinType::Double)
2522         mangleArtificalTagType(TTK_Struct, "__m" + llvm::utostr(Width) + 'd');
2523     }
2524   }
2525 
2526   bool IsBuiltin = Out.tell() != OutSizeBefore;
2527   if (!IsBuiltin) {
2528     // The MS ABI doesn't have a special mangling for vector types, so we define
2529     // our own mangling to handle uses of __vector_size__ on user-specified
2530     // types, and for extensions like __v4sf.
2531 
2532     llvm::SmallString<64> TemplateMangling;
2533     llvm::raw_svector_ostream Stream(TemplateMangling);
2534     MicrosoftCXXNameMangler Extra(Context, Stream);
2535     Stream << "?$";
2536     Extra.mangleSourceName("__vector");
2537     Extra.mangleType(QualType(ET, 0), Range, QMM_Escape);
2538     Extra.mangleIntegerLiteral(llvm::APSInt::getUnsigned(T->getNumElements()),
2539                                /*IsBoolean=*/false);
2540 
2541     mangleArtificalTagType(TTK_Union, TemplateMangling, {"__clang"});
2542   }
2543 }
2544 
2545 void MicrosoftCXXNameMangler::mangleType(const ExtVectorType *T,
2546                                          Qualifiers Quals, SourceRange Range) {
2547   mangleType(static_cast<const VectorType *>(T), Quals, Range);
2548 }
2549 
2550 void MicrosoftCXXNameMangler::mangleType(const DependentVectorType *T,
2551                                          Qualifiers, SourceRange Range) {
2552   DiagnosticsEngine &Diags = Context.getDiags();
2553   unsigned DiagID = Diags.getCustomDiagID(
2554       DiagnosticsEngine::Error,
2555       "cannot mangle this dependent-sized vector type yet");
2556   Diags.Report(Range.getBegin(), DiagID) << Range;
2557 }
2558 
2559 void MicrosoftCXXNameMangler::mangleType(const DependentSizedExtVectorType *T,
2560                                          Qualifiers, SourceRange Range) {
2561   DiagnosticsEngine &Diags = Context.getDiags();
2562   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2563     "cannot mangle this dependent-sized extended vector type yet");
2564   Diags.Report(Range.getBegin(), DiagID)
2565     << Range;
2566 }
2567 
2568 void MicrosoftCXXNameMangler::mangleType(const DependentAddressSpaceType *T,
2569                                          Qualifiers, SourceRange Range) {
2570   DiagnosticsEngine &Diags = Context.getDiags();
2571   unsigned DiagID = Diags.getCustomDiagID(
2572       DiagnosticsEngine::Error,
2573       "cannot mangle this dependent address space type yet");
2574   Diags.Report(Range.getBegin(), DiagID) << Range;
2575 }
2576 
2577 void MicrosoftCXXNameMangler::mangleType(const ObjCInterfaceType *T, Qualifiers,
2578                                          SourceRange) {
2579   // ObjC interfaces are mangled as if they were structs with a name that is
2580   // not a valid C/C++ identifier
2581   mangleTagTypeKind(TTK_Struct);
2582   mangle(T->getDecl(), ".objc_cls_");
2583 }
2584 
2585 void MicrosoftCXXNameMangler::mangleType(const ObjCObjectType *T, Qualifiers,
2586                                          SourceRange Range) {
2587   if (T->qual_empty())
2588     return mangleType(T->getBaseType(), Range, QMM_Drop);
2589 
2590   ArgBackRefMap OuterArgsContext;
2591   BackRefVec OuterTemplateContext;
2592 
2593   TypeBackReferences.swap(OuterArgsContext);
2594   NameBackReferences.swap(OuterTemplateContext);
2595 
2596   mangleTagTypeKind(TTK_Struct);
2597 
2598   Out << "?$";
2599   if (T->isObjCId())
2600     mangleSourceName(".objc_object");
2601   else if (T->isObjCClass())
2602     mangleSourceName(".objc_class");
2603   else
2604     mangleSourceName((".objc_cls_" + T->getInterface()->getName()).str());
2605 
2606   for (const auto &Q : T->quals())
2607     mangleObjCProtocol(Q);
2608   Out << '@';
2609 
2610   Out << '@';
2611 
2612   TypeBackReferences.swap(OuterArgsContext);
2613   NameBackReferences.swap(OuterTemplateContext);
2614 }
2615 
2616 void MicrosoftCXXNameMangler::mangleType(const BlockPointerType *T,
2617                                          Qualifiers Quals, SourceRange Range) {
2618   QualType PointeeType = T->getPointeeType();
2619   manglePointerCVQualifiers(Quals);
2620   manglePointerExtQualifiers(Quals, PointeeType);
2621 
2622   Out << "_E";
2623 
2624   mangleFunctionType(PointeeType->castAs<FunctionProtoType>());
2625 }
2626 
2627 void MicrosoftCXXNameMangler::mangleType(const InjectedClassNameType *,
2628                                          Qualifiers, SourceRange) {
2629   llvm_unreachable("Cannot mangle injected class name type.");
2630 }
2631 
2632 void MicrosoftCXXNameMangler::mangleType(const TemplateSpecializationType *T,
2633                                          Qualifiers, SourceRange Range) {
2634   DiagnosticsEngine &Diags = Context.getDiags();
2635   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2636     "cannot mangle this template specialization type yet");
2637   Diags.Report(Range.getBegin(), DiagID)
2638     << Range;
2639 }
2640 
2641 void MicrosoftCXXNameMangler::mangleType(const DependentNameType *T, Qualifiers,
2642                                          SourceRange Range) {
2643   DiagnosticsEngine &Diags = Context.getDiags();
2644   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2645     "cannot mangle this dependent name type yet");
2646   Diags.Report(Range.getBegin(), DiagID)
2647     << Range;
2648 }
2649 
2650 void MicrosoftCXXNameMangler::mangleType(
2651     const DependentTemplateSpecializationType *T, Qualifiers,
2652     SourceRange Range) {
2653   DiagnosticsEngine &Diags = Context.getDiags();
2654   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2655     "cannot mangle this dependent template specialization type yet");
2656   Diags.Report(Range.getBegin(), DiagID)
2657     << Range;
2658 }
2659 
2660 void MicrosoftCXXNameMangler::mangleType(const PackExpansionType *T, Qualifiers,
2661                                          SourceRange Range) {
2662   DiagnosticsEngine &Diags = Context.getDiags();
2663   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2664     "cannot mangle this pack expansion yet");
2665   Diags.Report(Range.getBegin(), DiagID)
2666     << Range;
2667 }
2668 
2669 void MicrosoftCXXNameMangler::mangleType(const TypeOfType *T, Qualifiers,
2670                                          SourceRange Range) {
2671   DiagnosticsEngine &Diags = Context.getDiags();
2672   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2673     "cannot mangle this typeof(type) yet");
2674   Diags.Report(Range.getBegin(), DiagID)
2675     << Range;
2676 }
2677 
2678 void MicrosoftCXXNameMangler::mangleType(const TypeOfExprType *T, Qualifiers,
2679                                          SourceRange Range) {
2680   DiagnosticsEngine &Diags = Context.getDiags();
2681   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2682     "cannot mangle this typeof(expression) yet");
2683   Diags.Report(Range.getBegin(), DiagID)
2684     << Range;
2685 }
2686 
2687 void MicrosoftCXXNameMangler::mangleType(const DecltypeType *T, Qualifiers,
2688                                          SourceRange Range) {
2689   DiagnosticsEngine &Diags = Context.getDiags();
2690   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2691     "cannot mangle this decltype() yet");
2692   Diags.Report(Range.getBegin(), DiagID)
2693     << Range;
2694 }
2695 
2696 void MicrosoftCXXNameMangler::mangleType(const UnaryTransformType *T,
2697                                          Qualifiers, SourceRange Range) {
2698   DiagnosticsEngine &Diags = Context.getDiags();
2699   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2700     "cannot mangle this unary transform type yet");
2701   Diags.Report(Range.getBegin(), DiagID)
2702     << Range;
2703 }
2704 
2705 void MicrosoftCXXNameMangler::mangleType(const AutoType *T, Qualifiers,
2706                                          SourceRange Range) {
2707   assert(T->getDeducedType().isNull() && "expecting a dependent type!");
2708 
2709   DiagnosticsEngine &Diags = Context.getDiags();
2710   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2711     "cannot mangle this 'auto' type yet");
2712   Diags.Report(Range.getBegin(), DiagID)
2713     << Range;
2714 }
2715 
2716 void MicrosoftCXXNameMangler::mangleType(
2717     const DeducedTemplateSpecializationType *T, Qualifiers, SourceRange Range) {
2718   assert(T->getDeducedType().isNull() && "expecting a dependent type!");
2719 
2720   DiagnosticsEngine &Diags = Context.getDiags();
2721   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2722     "cannot mangle this deduced class template specialization type yet");
2723   Diags.Report(Range.getBegin(), DiagID)
2724     << Range;
2725 }
2726 
2727 void MicrosoftCXXNameMangler::mangleType(const AtomicType *T, Qualifiers,
2728                                          SourceRange Range) {
2729   QualType ValueType = T->getValueType();
2730 
2731   llvm::SmallString<64> TemplateMangling;
2732   llvm::raw_svector_ostream Stream(TemplateMangling);
2733   MicrosoftCXXNameMangler Extra(Context, Stream);
2734   Stream << "?$";
2735   Extra.mangleSourceName("_Atomic");
2736   Extra.mangleType(ValueType, Range, QMM_Escape);
2737 
2738   mangleArtificalTagType(TTK_Struct, TemplateMangling, {"__clang"});
2739 }
2740 
2741 void MicrosoftCXXNameMangler::mangleType(const PipeType *T, Qualifiers,
2742                                          SourceRange Range) {
2743   DiagnosticsEngine &Diags = Context.getDiags();
2744   unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2745     "cannot mangle this OpenCL pipe type yet");
2746   Diags.Report(Range.getBegin(), DiagID)
2747     << Range;
2748 }
2749 
2750 void MicrosoftMangleContextImpl::mangleCXXName(const NamedDecl *D,
2751                                                raw_ostream &Out) {
2752   assert((isa<FunctionDecl>(D) || isa<VarDecl>(D)) &&
2753          "Invalid mangleName() call, argument is not a variable or function!");
2754   assert(!isa<CXXConstructorDecl>(D) && !isa<CXXDestructorDecl>(D) &&
2755          "Invalid mangleName() call on 'structor decl!");
2756 
2757   PrettyStackTraceDecl CrashInfo(D, SourceLocation(),
2758                                  getASTContext().getSourceManager(),
2759                                  "Mangling declaration");
2760 
2761   msvc_hashing_ostream MHO(Out);
2762   MicrosoftCXXNameMangler Mangler(*this, MHO);
2763   return Mangler.mangle(D);
2764 }
2765 
2766 // <this-adjustment> ::= <no-adjustment> | <static-adjustment> |
2767 //                       <virtual-adjustment>
2768 // <no-adjustment>      ::= A # private near
2769 //                      ::= B # private far
2770 //                      ::= I # protected near
2771 //                      ::= J # protected far
2772 //                      ::= Q # public near
2773 //                      ::= R # public far
2774 // <static-adjustment>  ::= G <static-offset> # private near
2775 //                      ::= H <static-offset> # private far
2776 //                      ::= O <static-offset> # protected near
2777 //                      ::= P <static-offset> # protected far
2778 //                      ::= W <static-offset> # public near
2779 //                      ::= X <static-offset> # public far
2780 // <virtual-adjustment> ::= $0 <virtual-shift> <static-offset> # private near
2781 //                      ::= $1 <virtual-shift> <static-offset> # private far
2782 //                      ::= $2 <virtual-shift> <static-offset> # protected near
2783 //                      ::= $3 <virtual-shift> <static-offset> # protected far
2784 //                      ::= $4 <virtual-shift> <static-offset> # public near
2785 //                      ::= $5 <virtual-shift> <static-offset> # public far
2786 // <virtual-shift>      ::= <vtordisp-shift> | <vtordispex-shift>
2787 // <vtordisp-shift>     ::= <offset-to-vtordisp>
2788 // <vtordispex-shift>   ::= <offset-to-vbptr> <vbase-offset-offset>
2789 //                          <offset-to-vtordisp>
2790 static void mangleThunkThisAdjustment(const CXXMethodDecl *MD,
2791                                       const ThisAdjustment &Adjustment,
2792                                       MicrosoftCXXNameMangler &Mangler,
2793                                       raw_ostream &Out) {
2794   if (!Adjustment.Virtual.isEmpty()) {
2795     Out << '$';
2796     char AccessSpec;
2797     switch (MD->getAccess()) {
2798     case AS_none:
2799       llvm_unreachable("Unsupported access specifier");
2800     case AS_private:
2801       AccessSpec = '0';
2802       break;
2803     case AS_protected:
2804       AccessSpec = '2';
2805       break;
2806     case AS_public:
2807       AccessSpec = '4';
2808     }
2809     if (Adjustment.Virtual.Microsoft.VBPtrOffset) {
2810       Out << 'R' << AccessSpec;
2811       Mangler.mangleNumber(
2812           static_cast<uint32_t>(Adjustment.Virtual.Microsoft.VBPtrOffset));
2813       Mangler.mangleNumber(
2814           static_cast<uint32_t>(Adjustment.Virtual.Microsoft.VBOffsetOffset));
2815       Mangler.mangleNumber(
2816           static_cast<uint32_t>(Adjustment.Virtual.Microsoft.VtordispOffset));
2817       Mangler.mangleNumber(static_cast<uint32_t>(Adjustment.NonVirtual));
2818     } else {
2819       Out << AccessSpec;
2820       Mangler.mangleNumber(
2821           static_cast<uint32_t>(Adjustment.Virtual.Microsoft.VtordispOffset));
2822       Mangler.mangleNumber(-static_cast<uint32_t>(Adjustment.NonVirtual));
2823     }
2824   } else if (Adjustment.NonVirtual != 0) {
2825     switch (MD->getAccess()) {
2826     case AS_none:
2827       llvm_unreachable("Unsupported access specifier");
2828     case AS_private:
2829       Out << 'G';
2830       break;
2831     case AS_protected:
2832       Out << 'O';
2833       break;
2834     case AS_public:
2835       Out << 'W';
2836     }
2837     Mangler.mangleNumber(-static_cast<uint32_t>(Adjustment.NonVirtual));
2838   } else {
2839     switch (MD->getAccess()) {
2840     case AS_none:
2841       llvm_unreachable("Unsupported access specifier");
2842     case AS_private:
2843       Out << 'A';
2844       break;
2845     case AS_protected:
2846       Out << 'I';
2847       break;
2848     case AS_public:
2849       Out << 'Q';
2850     }
2851   }
2852 }
2853 
2854 void MicrosoftMangleContextImpl::mangleVirtualMemPtrThunk(
2855     const CXXMethodDecl *MD, const MethodVFTableLocation &ML,
2856     raw_ostream &Out) {
2857   msvc_hashing_ostream MHO(Out);
2858   MicrosoftCXXNameMangler Mangler(*this, MHO);
2859   Mangler.getStream() << '?';
2860   Mangler.mangleVirtualMemPtrThunk(MD, ML);
2861 }
2862 
2863 void MicrosoftMangleContextImpl::mangleThunk(const CXXMethodDecl *MD,
2864                                              const ThunkInfo &Thunk,
2865                                              raw_ostream &Out) {
2866   msvc_hashing_ostream MHO(Out);
2867   MicrosoftCXXNameMangler Mangler(*this, MHO);
2868   Mangler.getStream() << '?';
2869   Mangler.mangleName(MD);
2870   mangleThunkThisAdjustment(MD, Thunk.This, Mangler, MHO);
2871   if (!Thunk.Return.isEmpty())
2872     assert(Thunk.Method != nullptr &&
2873            "Thunk info should hold the overridee decl");
2874 
2875   const CXXMethodDecl *DeclForFPT = Thunk.Method ? Thunk.Method : MD;
2876   Mangler.mangleFunctionType(
2877       DeclForFPT->getType()->castAs<FunctionProtoType>(), MD);
2878 }
2879 
2880 void MicrosoftMangleContextImpl::mangleCXXDtorThunk(
2881     const CXXDestructorDecl *DD, CXXDtorType Type,
2882     const ThisAdjustment &Adjustment, raw_ostream &Out) {
2883   // FIXME: Actually, the dtor thunk should be emitted for vector deleting
2884   // dtors rather than scalar deleting dtors. Just use the vector deleting dtor
2885   // mangling manually until we support both deleting dtor types.
2886   assert(Type == Dtor_Deleting);
2887   msvc_hashing_ostream MHO(Out);
2888   MicrosoftCXXNameMangler Mangler(*this, MHO, DD, Type);
2889   Mangler.getStream() << "??_E";
2890   Mangler.mangleName(DD->getParent());
2891   mangleThunkThisAdjustment(DD, Adjustment, Mangler, MHO);
2892   Mangler.mangleFunctionType(DD->getType()->castAs<FunctionProtoType>(), DD);
2893 }
2894 
2895 void MicrosoftMangleContextImpl::mangleCXXVFTable(
2896     const CXXRecordDecl *Derived, ArrayRef<const CXXRecordDecl *> BasePath,
2897     raw_ostream &Out) {
2898   // <mangled-name> ::= ?_7 <class-name> <storage-class>
2899   //                    <cvr-qualifiers> [<name>] @
2900   // NOTE: <cvr-qualifiers> here is always 'B' (const). <storage-class>
2901   // is always '6' for vftables.
2902   msvc_hashing_ostream MHO(Out);
2903   MicrosoftCXXNameMangler Mangler(*this, MHO);
2904   if (Derived->hasAttr<DLLImportAttr>())
2905     Mangler.getStream() << "??_S";
2906   else
2907     Mangler.getStream() << "??_7";
2908   Mangler.mangleName(Derived);
2909   Mangler.getStream() << "6B"; // '6' for vftable, 'B' for const.
2910   for (const CXXRecordDecl *RD : BasePath)
2911     Mangler.mangleName(RD);
2912   Mangler.getStream() << '@';
2913 }
2914 
2915 void MicrosoftMangleContextImpl::mangleCXXVBTable(
2916     const CXXRecordDecl *Derived, ArrayRef<const CXXRecordDecl *> BasePath,
2917     raw_ostream &Out) {
2918   // <mangled-name> ::= ?_8 <class-name> <storage-class>
2919   //                    <cvr-qualifiers> [<name>] @
2920   // NOTE: <cvr-qualifiers> here is always 'B' (const). <storage-class>
2921   // is always '7' for vbtables.
2922   msvc_hashing_ostream MHO(Out);
2923   MicrosoftCXXNameMangler Mangler(*this, MHO);
2924   Mangler.getStream() << "??_8";
2925   Mangler.mangleName(Derived);
2926   Mangler.getStream() << "7B";  // '7' for vbtable, 'B' for const.
2927   for (const CXXRecordDecl *RD : BasePath)
2928     Mangler.mangleName(RD);
2929   Mangler.getStream() << '@';
2930 }
2931 
2932 void MicrosoftMangleContextImpl::mangleCXXRTTI(QualType T, raw_ostream &Out) {
2933   msvc_hashing_ostream MHO(Out);
2934   MicrosoftCXXNameMangler Mangler(*this, MHO);
2935   Mangler.getStream() << "??_R0";
2936   Mangler.mangleType(T, SourceRange(), MicrosoftCXXNameMangler::QMM_Result);
2937   Mangler.getStream() << "@8";
2938 }
2939 
2940 void MicrosoftMangleContextImpl::mangleCXXRTTIName(QualType T,
2941                                                    raw_ostream &Out) {
2942   MicrosoftCXXNameMangler Mangler(*this, Out);
2943   Mangler.getStream() << '.';
2944   Mangler.mangleType(T, SourceRange(), MicrosoftCXXNameMangler::QMM_Result);
2945 }
2946 
2947 void MicrosoftMangleContextImpl::mangleCXXVirtualDisplacementMap(
2948     const CXXRecordDecl *SrcRD, const CXXRecordDecl *DstRD, raw_ostream &Out) {
2949   msvc_hashing_ostream MHO(Out);
2950   MicrosoftCXXNameMangler Mangler(*this, MHO);
2951   Mangler.getStream() << "??_K";
2952   Mangler.mangleName(SrcRD);
2953   Mangler.getStream() << "$C";
2954   Mangler.mangleName(DstRD);
2955 }
2956 
2957 void MicrosoftMangleContextImpl::mangleCXXThrowInfo(QualType T, bool IsConst,
2958                                                     bool IsVolatile,
2959                                                     bool IsUnaligned,
2960                                                     uint32_t NumEntries,
2961                                                     raw_ostream &Out) {
2962   msvc_hashing_ostream MHO(Out);
2963   MicrosoftCXXNameMangler Mangler(*this, MHO);
2964   Mangler.getStream() << "_TI";
2965   if (IsConst)
2966     Mangler.getStream() << 'C';
2967   if (IsVolatile)
2968     Mangler.getStream() << 'V';
2969   if (IsUnaligned)
2970     Mangler.getStream() << 'U';
2971   Mangler.getStream() << NumEntries;
2972   Mangler.mangleType(T, SourceRange(), MicrosoftCXXNameMangler::QMM_Result);
2973 }
2974 
2975 void MicrosoftMangleContextImpl::mangleCXXCatchableTypeArray(
2976     QualType T, uint32_t NumEntries, raw_ostream &Out) {
2977   msvc_hashing_ostream MHO(Out);
2978   MicrosoftCXXNameMangler Mangler(*this, MHO);
2979   Mangler.getStream() << "_CTA";
2980   Mangler.getStream() << NumEntries;
2981   Mangler.mangleType(T, SourceRange(), MicrosoftCXXNameMangler::QMM_Result);
2982 }
2983 
2984 void MicrosoftMangleContextImpl::mangleCXXCatchableType(
2985     QualType T, const CXXConstructorDecl *CD, CXXCtorType CT, uint32_t Size,
2986     uint32_t NVOffset, int32_t VBPtrOffset, uint32_t VBIndex,
2987     raw_ostream &Out) {
2988   MicrosoftCXXNameMangler Mangler(*this, Out);
2989   Mangler.getStream() << "_CT";
2990 
2991   llvm::SmallString<64> RTTIMangling;
2992   {
2993     llvm::raw_svector_ostream Stream(RTTIMangling);
2994     msvc_hashing_ostream MHO(Stream);
2995     mangleCXXRTTI(T, MHO);
2996   }
2997   Mangler.getStream() << RTTIMangling;
2998 
2999   // VS2015 CTP6 omits the copy-constructor in the mangled name.  This name is,
3000   // in fact, superfluous but I'm not sure the change was made consciously.
3001   llvm::SmallString<64> CopyCtorMangling;
3002   if (!getASTContext().getLangOpts().isCompatibleWithMSVC(
3003           LangOptions::MSVC2015) &&
3004       CD) {
3005     llvm::raw_svector_ostream Stream(CopyCtorMangling);
3006     msvc_hashing_ostream MHO(Stream);
3007     mangleCXXCtor(CD, CT, MHO);
3008   }
3009   Mangler.getStream() << CopyCtorMangling;
3010 
3011   Mangler.getStream() << Size;
3012   if (VBPtrOffset == -1) {
3013     if (NVOffset) {
3014       Mangler.getStream() << NVOffset;
3015     }
3016   } else {
3017     Mangler.getStream() << NVOffset;
3018     Mangler.getStream() << VBPtrOffset;
3019     Mangler.getStream() << VBIndex;
3020   }
3021 }
3022 
3023 void MicrosoftMangleContextImpl::mangleCXXRTTIBaseClassDescriptor(
3024     const CXXRecordDecl *Derived, uint32_t NVOffset, int32_t VBPtrOffset,
3025     uint32_t VBTableOffset, uint32_t Flags, raw_ostream &Out) {
3026   msvc_hashing_ostream MHO(Out);
3027   MicrosoftCXXNameMangler Mangler(*this, MHO);
3028   Mangler.getStream() << "??_R1";
3029   Mangler.mangleNumber(NVOffset);
3030   Mangler.mangleNumber(VBPtrOffset);
3031   Mangler.mangleNumber(VBTableOffset);
3032   Mangler.mangleNumber(Flags);
3033   Mangler.mangleName(Derived);
3034   Mangler.getStream() << "8";
3035 }
3036 
3037 void MicrosoftMangleContextImpl::mangleCXXRTTIBaseClassArray(
3038     const CXXRecordDecl *Derived, raw_ostream &Out) {
3039   msvc_hashing_ostream MHO(Out);
3040   MicrosoftCXXNameMangler Mangler(*this, MHO);
3041   Mangler.getStream() << "??_R2";
3042   Mangler.mangleName(Derived);
3043   Mangler.getStream() << "8";
3044 }
3045 
3046 void MicrosoftMangleContextImpl::mangleCXXRTTIClassHierarchyDescriptor(
3047     const CXXRecordDecl *Derived, raw_ostream &Out) {
3048   msvc_hashing_ostream MHO(Out);
3049   MicrosoftCXXNameMangler Mangler(*this, MHO);
3050   Mangler.getStream() << "??_R3";
3051   Mangler.mangleName(Derived);
3052   Mangler.getStream() << "8";
3053 }
3054 
3055 void MicrosoftMangleContextImpl::mangleCXXRTTICompleteObjectLocator(
3056     const CXXRecordDecl *Derived, ArrayRef<const CXXRecordDecl *> BasePath,
3057     raw_ostream &Out) {
3058   // <mangled-name> ::= ?_R4 <class-name> <storage-class>
3059   //                    <cvr-qualifiers> [<name>] @
3060   // NOTE: <cvr-qualifiers> here is always 'B' (const). <storage-class>
3061   // is always '6' for vftables.
3062   llvm::SmallString<64> VFTableMangling;
3063   llvm::raw_svector_ostream Stream(VFTableMangling);
3064   mangleCXXVFTable(Derived, BasePath, Stream);
3065 
3066   if (VFTableMangling.startswith("??@")) {
3067     assert(VFTableMangling.endswith("@"));
3068     Out << VFTableMangling << "??_R4@";
3069     return;
3070   }
3071 
3072   assert(VFTableMangling.startswith("??_7") ||
3073          VFTableMangling.startswith("??_S"));
3074 
3075   Out << "??_R4" << StringRef(VFTableMangling).drop_front(4);
3076 }
3077 
3078 void MicrosoftMangleContextImpl::mangleSEHFilterExpression(
3079     const NamedDecl *EnclosingDecl, raw_ostream &Out) {
3080   msvc_hashing_ostream MHO(Out);
3081   MicrosoftCXXNameMangler Mangler(*this, MHO);
3082   // The function body is in the same comdat as the function with the handler,
3083   // so the numbering here doesn't have to be the same across TUs.
3084   //
3085   // <mangled-name> ::= ?filt$ <filter-number> @0
3086   Mangler.getStream() << "?filt$" << SEHFilterIds[EnclosingDecl]++ << "@0@";
3087   Mangler.mangleName(EnclosingDecl);
3088 }
3089 
3090 void MicrosoftMangleContextImpl::mangleSEHFinallyBlock(
3091     const NamedDecl *EnclosingDecl, raw_ostream &Out) {
3092   msvc_hashing_ostream MHO(Out);
3093   MicrosoftCXXNameMangler Mangler(*this, MHO);
3094   // The function body is in the same comdat as the function with the handler,
3095   // so the numbering here doesn't have to be the same across TUs.
3096   //
3097   // <mangled-name> ::= ?fin$ <filter-number> @0
3098   Mangler.getStream() << "?fin$" << SEHFinallyIds[EnclosingDecl]++ << "@0@";
3099   Mangler.mangleName(EnclosingDecl);
3100 }
3101 
3102 void MicrosoftMangleContextImpl::mangleTypeName(QualType T, raw_ostream &Out) {
3103   // This is just a made up unique string for the purposes of tbaa.  undname
3104   // does *not* know how to demangle it.
3105   MicrosoftCXXNameMangler Mangler(*this, Out);
3106   Mangler.getStream() << '?';
3107   Mangler.mangleType(T, SourceRange());
3108 }
3109 
3110 void MicrosoftMangleContextImpl::mangleCXXCtor(const CXXConstructorDecl *D,
3111                                                CXXCtorType Type,
3112                                                raw_ostream &Out) {
3113   msvc_hashing_ostream MHO(Out);
3114   MicrosoftCXXNameMangler mangler(*this, MHO, D, Type);
3115   mangler.mangle(D);
3116 }
3117 
3118 void MicrosoftMangleContextImpl::mangleCXXDtor(const CXXDestructorDecl *D,
3119                                                CXXDtorType Type,
3120                                                raw_ostream &Out) {
3121   msvc_hashing_ostream MHO(Out);
3122   MicrosoftCXXNameMangler mangler(*this, MHO, D, Type);
3123   mangler.mangle(D);
3124 }
3125 
3126 void MicrosoftMangleContextImpl::mangleReferenceTemporary(
3127     const VarDecl *VD, unsigned ManglingNumber, raw_ostream &Out) {
3128   msvc_hashing_ostream MHO(Out);
3129   MicrosoftCXXNameMangler Mangler(*this, MHO);
3130 
3131   Mangler.getStream() << "?$RT" << ManglingNumber << '@';
3132   Mangler.mangle(VD, "");
3133 }
3134 
3135 void MicrosoftMangleContextImpl::mangleThreadSafeStaticGuardVariable(
3136     const VarDecl *VD, unsigned GuardNum, raw_ostream &Out) {
3137   msvc_hashing_ostream MHO(Out);
3138   MicrosoftCXXNameMangler Mangler(*this, MHO);
3139 
3140   Mangler.getStream() << "?$TSS" << GuardNum << '@';
3141   Mangler.mangleNestedName(VD);
3142   Mangler.getStream() << "@4HA";
3143 }
3144 
3145 void MicrosoftMangleContextImpl::mangleStaticGuardVariable(const VarDecl *VD,
3146                                                            raw_ostream &Out) {
3147   // <guard-name> ::= ?_B <postfix> @5 <scope-depth>
3148   //              ::= ?__J <postfix> @5 <scope-depth>
3149   //              ::= ?$S <guard-num> @ <postfix> @4IA
3150 
3151   // The first mangling is what MSVC uses to guard static locals in inline
3152   // functions.  It uses a different mangling in external functions to support
3153   // guarding more than 32 variables.  MSVC rejects inline functions with more
3154   // than 32 static locals.  We don't fully implement the second mangling
3155   // because those guards are not externally visible, and instead use LLVM's
3156   // default renaming when creating a new guard variable.
3157   msvc_hashing_ostream MHO(Out);
3158   MicrosoftCXXNameMangler Mangler(*this, MHO);
3159 
3160   bool Visible = VD->isExternallyVisible();
3161   if (Visible) {
3162     Mangler.getStream() << (VD->getTLSKind() ? "??__J" : "??_B");
3163   } else {
3164     Mangler.getStream() << "?$S1@";
3165   }
3166   unsigned ScopeDepth = 0;
3167   if (Visible && !getNextDiscriminator(VD, ScopeDepth))
3168     // If we do not have a discriminator and are emitting a guard variable for
3169     // use at global scope, then mangling the nested name will not be enough to
3170     // remove ambiguities.
3171     Mangler.mangle(VD, "");
3172   else
3173     Mangler.mangleNestedName(VD);
3174   Mangler.getStream() << (Visible ? "@5" : "@4IA");
3175   if (ScopeDepth)
3176     Mangler.mangleNumber(ScopeDepth);
3177 }
3178 
3179 void MicrosoftMangleContextImpl::mangleInitFiniStub(const VarDecl *D,
3180                                                     char CharCode,
3181                                                     raw_ostream &Out) {
3182   msvc_hashing_ostream MHO(Out);
3183   MicrosoftCXXNameMangler Mangler(*this, MHO);
3184   Mangler.getStream() << "??__" << CharCode;
3185   Mangler.mangleName(D);
3186   if (D->isStaticDataMember()) {
3187     Mangler.mangleVariableEncoding(D);
3188     Mangler.getStream() << '@';
3189   }
3190   // This is the function class mangling.  These stubs are global, non-variadic,
3191   // cdecl functions that return void and take no args.
3192   Mangler.getStream() << "YAXXZ";
3193 }
3194 
3195 void MicrosoftMangleContextImpl::mangleDynamicInitializer(const VarDecl *D,
3196                                                           raw_ostream &Out) {
3197   // <initializer-name> ::= ?__E <name> YAXXZ
3198   mangleInitFiniStub(D, 'E', Out);
3199 }
3200 
3201 void
3202 MicrosoftMangleContextImpl::mangleDynamicAtExitDestructor(const VarDecl *D,
3203                                                           raw_ostream &Out) {
3204   // <destructor-name> ::= ?__F <name> YAXXZ
3205   mangleInitFiniStub(D, 'F', Out);
3206 }
3207 
3208 void MicrosoftMangleContextImpl::mangleStringLiteral(const StringLiteral *SL,
3209                                                      raw_ostream &Out) {
3210   // <char-type> ::= 0   # char, char16_t, char32_t
3211   //                     # (little endian char data in mangling)
3212   //             ::= 1   # wchar_t (big endian char data in mangling)
3213   //
3214   // <literal-length> ::= <non-negative integer>  # the length of the literal
3215   //
3216   // <encoded-crc>    ::= <hex digit>+ @          # crc of the literal including
3217   //                                              # trailing null bytes
3218   //
3219   // <encoded-string> ::= <simple character>           # uninteresting character
3220   //                  ::= '?$' <hex digit> <hex digit> # these two nibbles
3221   //                                                   # encode the byte for the
3222   //                                                   # character
3223   //                  ::= '?' [a-z]                    # \xe1 - \xfa
3224   //                  ::= '?' [A-Z]                    # \xc1 - \xda
3225   //                  ::= '?' [0-9]                    # [,/\:. \n\t'-]
3226   //
3227   // <literal> ::= '??_C@_' <char-type> <literal-length> <encoded-crc>
3228   //               <encoded-string> '@'
3229   MicrosoftCXXNameMangler Mangler(*this, Out);
3230   Mangler.getStream() << "??_C@_";
3231 
3232   // The actual string length might be different from that of the string literal
3233   // in cases like:
3234   // char foo[3] = "foobar";
3235   // char bar[42] = "foobar";
3236   // Where it is truncated or zero-padded to fit the array. This is the length
3237   // used for mangling, and any trailing null-bytes also need to be mangled.
3238   unsigned StringLength = getASTContext()
3239                               .getAsConstantArrayType(SL->getType())
3240                               ->getSize()
3241                               .getZExtValue();
3242   unsigned StringByteLength = StringLength * SL->getCharByteWidth();
3243 
3244   // <char-type>: The "kind" of string literal is encoded into the mangled name.
3245   if (SL->isWide())
3246     Mangler.getStream() << '1';
3247   else
3248     Mangler.getStream() << '0';
3249 
3250   // <literal-length>: The next part of the mangled name consists of the length
3251   // of the string in bytes.
3252   Mangler.mangleNumber(StringByteLength);
3253 
3254   auto GetLittleEndianByte = [&SL](unsigned Index) {
3255     unsigned CharByteWidth = SL->getCharByteWidth();
3256     if (Index / CharByteWidth >= SL->getLength())
3257       return static_cast<char>(0);
3258     uint32_t CodeUnit = SL->getCodeUnit(Index / CharByteWidth);
3259     unsigned OffsetInCodeUnit = Index % CharByteWidth;
3260     return static_cast<char>((CodeUnit >> (8 * OffsetInCodeUnit)) & 0xff);
3261   };
3262 
3263   auto GetBigEndianByte = [&SL](unsigned Index) {
3264     unsigned CharByteWidth = SL->getCharByteWidth();
3265     if (Index / CharByteWidth >= SL->getLength())
3266       return static_cast<char>(0);
3267     uint32_t CodeUnit = SL->getCodeUnit(Index / CharByteWidth);
3268     unsigned OffsetInCodeUnit = (CharByteWidth - 1) - (Index % CharByteWidth);
3269     return static_cast<char>((CodeUnit >> (8 * OffsetInCodeUnit)) & 0xff);
3270   };
3271 
3272   // CRC all the bytes of the StringLiteral.
3273   llvm::JamCRC JC;
3274   for (unsigned I = 0, E = StringByteLength; I != E; ++I)
3275     JC.update(GetLittleEndianByte(I));
3276 
3277   // <encoded-crc>: The CRC is encoded utilizing the standard number mangling
3278   // scheme.
3279   Mangler.mangleNumber(JC.getCRC());
3280 
3281   // <encoded-string>: The mangled name also contains the first 32 bytes
3282   // (including null-terminator bytes) of the encoded StringLiteral.
3283   // Each character is encoded by splitting them into bytes and then encoding
3284   // the constituent bytes.
3285   auto MangleByte = [&Mangler](char Byte) {
3286     // There are five different manglings for characters:
3287     // - [a-zA-Z0-9_$]: A one-to-one mapping.
3288     // - ?[a-z]: The range from \xe1 to \xfa.
3289     // - ?[A-Z]: The range from \xc1 to \xda.
3290     // - ?[0-9]: The set of [,/\:. \n\t'-].
3291     // - ?$XX: A fallback which maps nibbles.
3292     if (isIdentifierBody(Byte, /*AllowDollar=*/true)) {
3293       Mangler.getStream() << Byte;
3294     } else if (isLetter(Byte & 0x7f)) {
3295       Mangler.getStream() << '?' << static_cast<char>(Byte & 0x7f);
3296     } else {
3297       const char SpecialChars[] = {',', '/',  '\\', ':',  '.',
3298                                    ' ', '\n', '\t', '\'', '-'};
3299       const char *Pos =
3300           std::find(std::begin(SpecialChars), std::end(SpecialChars), Byte);
3301       if (Pos != std::end(SpecialChars)) {
3302         Mangler.getStream() << '?' << (Pos - std::begin(SpecialChars));
3303       } else {
3304         Mangler.getStream() << "?$";
3305         Mangler.getStream() << static_cast<char>('A' + ((Byte >> 4) & 0xf));
3306         Mangler.getStream() << static_cast<char>('A' + (Byte & 0xf));
3307       }
3308     }
3309   };
3310 
3311   // Enforce our 32 bytes max, except wchar_t which gets 32 chars instead.
3312   unsigned MaxBytesToMangle = SL->isWide() ? 64U : 32U;
3313   unsigned NumBytesToMangle = std::min(MaxBytesToMangle, StringByteLength);
3314   for (unsigned I = 0; I != NumBytesToMangle; ++I) {
3315     if (SL->isWide())
3316       MangleByte(GetBigEndianByte(I));
3317     else
3318       MangleByte(GetLittleEndianByte(I));
3319   }
3320 
3321   Mangler.getStream() << '@';
3322 }
3323 
3324 MicrosoftMangleContext *
3325 MicrosoftMangleContext::create(ASTContext &Context, DiagnosticsEngine &Diags) {
3326   return new MicrosoftMangleContextImpl(Context, Diags);
3327 }
3328