blob: f7a456e06ad960f25f13cffa11b211c4ba1e91a1 [file] [log] [blame]
Guy Benyei11169dd2012-12-18 14:30:41 +00001//===--- MicrosoftMangle.cpp - Microsoft Visual C++ Name Mangling ---------===//
2//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
Guy Benyei11169dd2012-12-18 14:30:41 +00006//
7//===----------------------------------------------------------------------===//
8//
9// This provides C++ name mangling targeting the Microsoft Visual C++ ABI.
10//
11//===----------------------------------------------------------------------===//
12
13#include "clang/AST/Mangle.h"
14#include "clang/AST/ASTContext.h"
15#include "clang/AST/Attr.h"
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +000016#include "clang/AST/CXXInheritance.h"
Chandler Carruth5553d0d2014-01-07 11:51:46 +000017#include "clang/AST/CharUnits.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000018#include "clang/AST/Decl.h"
19#include "clang/AST/DeclCXX.h"
20#include "clang/AST/DeclObjC.h"
Alexey Bataev94a4f0c2016-03-03 05:21:39 +000021#include "clang/AST/DeclOpenMP.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000022#include "clang/AST/DeclTemplate.h"
David Majnemer58e5bee2014-03-24 21:43:36 +000023#include "clang/AST/Expr.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000024#include "clang/AST/ExprCXX.h"
Reid Kleckner96f8f932014-02-05 17:27:08 +000025#include "clang/AST/VTableBuilder.h"
Guy Benyei11169dd2012-12-18 14:30:41 +000026#include "clang/Basic/ABI.h"
27#include "clang/Basic/DiagnosticOptions.h"
Reid Kleckner369f3162013-05-14 20:30:42 +000028#include "clang/Basic/TargetInfo.h"
David Majnemer2206bf52014-03-05 08:57:59 +000029#include "llvm/ADT/StringExtras.h"
David Majnemerb8ed3642015-09-01 21:24:07 +000030#include "llvm/Support/JamCRC.h"
Reid Klecknerdf8dd612018-08-17 20:59:27 +000031#include "llvm/Support/xxhash.h"
David Majnemerfebb2722016-03-07 08:51:17 +000032#include "llvm/Support/MD5.h"
33#include "llvm/Support/MathExtras.h"
Nico Weber48707762019-06-04 23:27:40 +000034#include "llvm/Support/StringSaver.h"
David Majnemer3843a052014-03-23 17:47:16 +000035
Guy Benyei11169dd2012-12-18 14:30:41 +000036using namespace clang;
37
38namespace {
39
David Majnemerfebb2722016-03-07 08:51:17 +000040struct msvc_hashing_ostream : public llvm::raw_svector_ostream {
41 raw_ostream &OS;
42 llvm::SmallString<64> Buffer;
43
44 msvc_hashing_ostream(raw_ostream &OS)
45 : llvm::raw_svector_ostream(Buffer), OS(OS) {}
46 ~msvc_hashing_ostream() override {
47 StringRef MangledName = str();
48 bool StartsWithEscape = MangledName.startswith("\01");
49 if (StartsWithEscape)
50 MangledName = MangledName.drop_front(1);
51 if (MangledName.size() <= 4096) {
52 OS << str();
53 return;
54 }
55
56 llvm::MD5 Hasher;
57 llvm::MD5::MD5Result Hash;
58 Hasher.update(MangledName);
59 Hasher.final(Hash);
60
61 SmallString<32> HexString;
62 llvm::MD5::stringifyResult(Hash, HexString);
63
64 if (StartsWithEscape)
65 OS << '\01';
66 OS << "??@" << HexString << '@';
67 }
68};
69
David Majnemer984498f2016-12-06 17:01:02 +000070static const DeclContext *
71getLambdaDefaultArgumentDeclContext(const Decl *D) {
72 if (const auto *RD = dyn_cast<CXXRecordDecl>(D))
73 if (RD->isLambda())
74 if (const auto *Parm =
75 dyn_cast_or_null<ParmVarDecl>(RD->getLambdaContextDecl()))
76 return Parm->getDeclContext();
77 return nullptr;
78}
79
Adrian Prantl9fc8faf2018-05-09 01:00:01 +000080/// Retrieve the declaration context that should be used when mangling
David Majnemerd5a42b82013-09-13 09:03:14 +000081/// the given declaration.
82static const DeclContext *getEffectiveDeclContext(const Decl *D) {
83 // The ABI assumes that lambda closure types that occur within
84 // default arguments live in the context of the function. However, due to
85 // the way in which Clang parses and creates function declarations, this is
86 // not the case: the lambda closure type ends up living in the context
87 // where the function itself resides, because the function declaration itself
88 // had not yet been created. Fix the context here.
David Majnemer984498f2016-12-06 17:01:02 +000089 if (const auto *LDADC = getLambdaDefaultArgumentDeclContext(D))
90 return LDADC;
David Majnemerd5a42b82013-09-13 09:03:14 +000091
92 // Perform the same check for block literals.
93 if (const BlockDecl *BD = dyn_cast<BlockDecl>(D)) {
94 if (ParmVarDecl *ContextParam =
95 dyn_cast_or_null<ParmVarDecl>(BD->getBlockManglingContextDecl()))
96 return ContextParam->getDeclContext();
97 }
98
99 const DeclContext *DC = D->getDeclContext();
Michael Kruse251e1482019-02-01 20:25:04 +0000100 if (isa<CapturedDecl>(DC) || isa<OMPDeclareReductionDecl>(DC) ||
101 isa<OMPDeclareMapperDecl>(DC)) {
Alexey Bataev94a4f0c2016-03-03 05:21:39 +0000102 return getEffectiveDeclContext(cast<Decl>(DC));
103 }
David Majnemerd5a42b82013-09-13 09:03:14 +0000104
David Majnemerb0f1dbd2016-05-19 18:15:53 +0000105 return DC->getRedeclContext();
David Majnemerd5a42b82013-09-13 09:03:14 +0000106}
107
108static const DeclContext *getEffectiveParentContext(const DeclContext *DC) {
109 return getEffectiveDeclContext(cast<Decl>(DC));
110}
111
David Majnemerdfa6d202015-03-11 18:36:39 +0000112static const FunctionDecl *getStructor(const NamedDecl *ND) {
113 if (const auto *FTD = dyn_cast<FunctionTemplateDecl>(ND))
Reid Kleckner290feb92017-01-09 17:04:37 +0000114 return FTD->getTemplatedDecl()->getCanonicalDecl();
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +0000115
David Majnemerdfa6d202015-03-11 18:36:39 +0000116 const auto *FD = cast<FunctionDecl>(ND);
117 if (const auto *FTD = FD->getPrimaryTemplate())
Reid Kleckner290feb92017-01-09 17:04:37 +0000118 return FTD->getTemplatedDecl()->getCanonicalDecl();
David Majnemerdfa6d202015-03-11 18:36:39 +0000119
Reid Kleckner290feb92017-01-09 17:04:37 +0000120 return FD->getCanonicalDecl();
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +0000121}
122
Timur Iskhodzhanov67455222013-10-03 06:26:13 +0000123/// MicrosoftMangleContextImpl - Overrides the default MangleContext for the
Guy Benyei11169dd2012-12-18 14:30:41 +0000124/// Microsoft Visual C++ ABI.
Timur Iskhodzhanov67455222013-10-03 06:26:13 +0000125class MicrosoftMangleContextImpl : public MicrosoftMangleContext {
David Majnemer2206bf52014-03-05 08:57:59 +0000126 typedef std::pair<const DeclContext *, IdentifierInfo *> DiscriminatorKeyTy;
127 llvm::DenseMap<DiscriminatorKeyTy, unsigned> Discriminator;
David Majnemer02e9af42014-04-23 05:16:51 +0000128 llvm::DenseMap<const NamedDecl *, unsigned> Uniquifier;
David Majnemerf017ec32014-03-05 10:35:06 +0000129 llvm::DenseMap<const CXXRecordDecl *, unsigned> LambdaIds;
Reid Kleckner1d59f992015-01-22 01:36:17 +0000130 llvm::DenseMap<const NamedDecl *, unsigned> SEHFilterIds;
Reid Klecknerebaf28d2015-04-14 20:59:00 +0000131 llvm::DenseMap<const NamedDecl *, unsigned> SEHFinallyIds;
Reid Klecknerdf8dd612018-08-17 20:59:27 +0000132 SmallString<16> AnonymousNamespaceHash;
David Majnemer2206bf52014-03-05 08:57:59 +0000133
Guy Benyei11169dd2012-12-18 14:30:41 +0000134public:
Reid Klecknerdf8dd612018-08-17 20:59:27 +0000135 MicrosoftMangleContextImpl(ASTContext &Context, DiagnosticsEngine &Diags);
Craig Toppercbce6e92014-03-11 06:22:39 +0000136 bool shouldMangleCXXName(const NamedDecl *D) override;
David Majnemer58e5bee2014-03-24 21:43:36 +0000137 bool shouldMangleStringLiteral(const StringLiteral *SL) override;
Craig Toppercbce6e92014-03-11 06:22:39 +0000138 void mangleCXXName(const NamedDecl *D, raw_ostream &Out) override;
David Majnemer02e9af42014-04-23 05:16:51 +0000139 void mangleVirtualMemPtrThunk(const CXXMethodDecl *MD,
Reid Klecknercbec0262018-04-02 20:00:39 +0000140 const MethodVFTableLocation &ML,
141 raw_ostream &Out) override;
Craig Toppercbce6e92014-03-11 06:22:39 +0000142 void mangleThunk(const CXXMethodDecl *MD, const ThunkInfo &Thunk,
143 raw_ostream &) override;
144 void mangleCXXDtorThunk(const CXXDestructorDecl *DD, CXXDtorType Type,
145 const ThisAdjustment &ThisAdjustment,
146 raw_ostream &) override;
147 void mangleCXXVFTable(const CXXRecordDecl *Derived,
148 ArrayRef<const CXXRecordDecl *> BasePath,
149 raw_ostream &Out) override;
150 void mangleCXXVBTable(const CXXRecordDecl *Derived,
151 ArrayRef<const CXXRecordDecl *> BasePath,
152 raw_ostream &Out) override;
David Majnemerc1709d32015-06-23 07:31:11 +0000153 void mangleCXXVirtualDisplacementMap(const CXXRecordDecl *SrcRD,
154 const CXXRecordDecl *DstRD,
155 raw_ostream &Out) override;
David Majnemer7c237072015-03-05 00:46:22 +0000156 void mangleCXXThrowInfo(QualType T, bool IsConst, bool IsVolatile,
David Majnemer526793d2016-07-12 04:42:50 +0000157 bool IsUnaligned, uint32_t NumEntries,
158 raw_ostream &Out) override;
David Majnemer7c237072015-03-05 00:46:22 +0000159 void mangleCXXCatchableTypeArray(QualType T, uint32_t NumEntries,
160 raw_ostream &Out) override;
David Majnemere7a818f2015-03-06 18:53:55 +0000161 void mangleCXXCatchableType(QualType T, const CXXConstructorDecl *CD,
David Majnemerdfa6d202015-03-11 18:36:39 +0000162 CXXCtorType CT, uint32_t Size, uint32_t NVOffset,
David Majnemer999cbf92015-03-10 19:01:51 +0000163 int32_t VBPtrOffset, uint32_t VBIndex,
164 raw_ostream &Out) override;
David Majnemera96b7ee2014-05-13 00:44:44 +0000165 void mangleCXXRTTI(QualType T, raw_ostream &Out) override;
166 void mangleCXXRTTIName(QualType T, raw_ostream &Out) override;
David Majnemeraeb55c92014-05-13 06:57:43 +0000167 void mangleCXXRTTIBaseClassDescriptor(const CXXRecordDecl *Derived,
Warren Hunt5c2b4ea2014-05-23 16:07:43 +0000168 uint32_t NVOffset, int32_t VBPtrOffset,
David Majnemeraeb55c92014-05-13 06:57:43 +0000169 uint32_t VBTableOffset, uint32_t Flags,
170 raw_ostream &Out) override;
David Majnemera96b7ee2014-05-13 00:44:44 +0000171 void mangleCXXRTTIBaseClassArray(const CXXRecordDecl *Derived,
172 raw_ostream &Out) override;
173 void mangleCXXRTTIClassHierarchyDescriptor(const CXXRecordDecl *Derived,
174 raw_ostream &Out) override;
David Majnemeraeb55c92014-05-13 06:57:43 +0000175 void
176 mangleCXXRTTICompleteObjectLocator(const CXXRecordDecl *Derived,
177 ArrayRef<const CXXRecordDecl *> BasePath,
178 raw_ostream &Out) override;
Craig Toppercbce6e92014-03-11 06:22:39 +0000179 void mangleTypeName(QualType T, raw_ostream &) override;
180 void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type,
181 raw_ostream &) override;
182 void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type,
183 raw_ostream &) override;
David Majnemerdaff3702014-05-01 17:50:17 +0000184 void mangleReferenceTemporary(const VarDecl *, unsigned ManglingNumber,
185 raw_ostream &) override;
Craig Toppercbce6e92014-03-11 06:22:39 +0000186 void mangleStaticGuardVariable(const VarDecl *D, raw_ostream &Out) override;
David Majnemer8354eee2015-05-07 06:15:46 +0000187 void mangleThreadSafeStaticGuardVariable(const VarDecl *D, unsigned GuardNum,
188 raw_ostream &Out) override;
Craig Toppercbce6e92014-03-11 06:22:39 +0000189 void mangleDynamicInitializer(const VarDecl *D, raw_ostream &Out) override;
190 void mangleDynamicAtExitDestructor(const VarDecl *D,
191 raw_ostream &Out) override;
Reid Kleckner1d59f992015-01-22 01:36:17 +0000192 void mangleSEHFilterExpression(const NamedDecl *EnclosingDecl,
193 raw_ostream &Out) override;
Reid Klecknerebaf28d2015-04-14 20:59:00 +0000194 void mangleSEHFinallyBlock(const NamedDecl *EnclosingDecl,
195 raw_ostream &Out) override;
David Majnemer58e5bee2014-03-24 21:43:36 +0000196 void mangleStringLiteral(const StringLiteral *SL, raw_ostream &Out) override;
David Majnemer2206bf52014-03-05 08:57:59 +0000197 bool getNextDiscriminator(const NamedDecl *ND, unsigned &disc) {
David Majnemer2206bf52014-03-05 08:57:59 +0000198 const DeclContext *DC = getEffectiveDeclContext(ND);
199 if (!DC->isFunctionOrMethod())
200 return false;
201
David Majnemer41311192016-01-17 07:09:24 +0000202 // Lambda closure types are already numbered, give out a phony number so
203 // that they demangle nicely.
David Majnemer984498f2016-12-06 17:01:02 +0000204 if (const auto *RD = dyn_cast<CXXRecordDecl>(ND)) {
205 if (RD->isLambda()) {
206 disc = 1;
207 return true;
208 }
David Majnemer41311192016-01-17 07:09:24 +0000209 }
210
David Majnemer2206bf52014-03-05 08:57:59 +0000211 // Use the canonical number for externally visible decls.
212 if (ND->isExternallyVisible()) {
213 disc = getASTContext().getManglingNumber(ND);
214 return true;
215 }
216
217 // Anonymous tags are already numbered.
218 if (const TagDecl *Tag = dyn_cast<TagDecl>(ND)) {
David Majnemerb33cd902015-09-02 15:50:38 +0000219 if (!Tag->hasNameForLinkage() &&
220 !getASTContext().getDeclaratorForUnnamedTagDecl(Tag) &&
221 !getASTContext().getTypedefNameForUnnamedTagDecl(Tag))
David Majnemer2206bf52014-03-05 08:57:59 +0000222 return false;
223 }
224
225 // Make up a reasonable number for internal decls.
226 unsigned &discriminator = Uniquifier[ND];
227 if (!discriminator)
228 discriminator = ++Discriminator[std::make_pair(DC, ND->getIdentifier())];
David Majnemer040fa342014-10-05 06:44:53 +0000229 disc = discriminator + 1;
David Majnemer2206bf52014-03-05 08:57:59 +0000230 return true;
231 }
Reid Kleckner1ece9fc2013-09-10 20:43:12 +0000232
David Majnemerf017ec32014-03-05 10:35:06 +0000233 unsigned getLambdaId(const CXXRecordDecl *RD) {
234 assert(RD->isLambda() && "RD must be a lambda!");
235 assert(!RD->isExternallyVisible() && "RD must not be visible!");
236 assert(RD->getLambdaManglingNumber() == 0 &&
237 "RD must not have a mangling number!");
238 std::pair<llvm::DenseMap<const CXXRecordDecl *, unsigned>::iterator, bool>
239 Result = LambdaIds.insert(std::make_pair(RD, LambdaIds.size()));
240 return Result.first->second;
241 }
242
Reid Klecknerdf8dd612018-08-17 20:59:27 +0000243 /// Return a character sequence that is (somewhat) unique to the TU suitable
244 /// for mangling anonymous namespaces.
245 StringRef getAnonymousNamespaceHash() const {
246 return AnonymousNamespaceHash;
247 }
248
Reid Kleckner1ece9fc2013-09-10 20:43:12 +0000249private:
David Majnemerfebb2722016-03-07 08:51:17 +0000250 void mangleInitFiniStub(const VarDecl *D, char CharCode, raw_ostream &Out);
Guy Benyei11169dd2012-12-18 14:30:41 +0000251};
252
David Majnemer2206bf52014-03-05 08:57:59 +0000253/// MicrosoftCXXNameMangler - Manage the mangling of a single name for the
254/// Microsoft Visual C++ ABI.
255class MicrosoftCXXNameMangler {
256 MicrosoftMangleContextImpl &Context;
257 raw_ostream &Out;
258
259 /// The "structor" is the top-level declaration being mangled, if
260 /// that's not a template specialization; otherwise it's the pattern
261 /// for that specialization.
262 const NamedDecl *Structor;
263 unsigned StructorType;
264
David Majnemer0d0560c2014-09-25 19:43:56 +0000265 typedef llvm::SmallVector<std::string, 10> BackRefVec;
266 BackRefVec NameBackReferences;
David Majnemer2206bf52014-03-05 08:57:59 +0000267
David Majnemer622abd12015-12-31 05:36:52 +0000268 typedef llvm::DenseMap<const void *, unsigned> ArgBackRefMap;
Nico Weber73f05842019-06-01 10:12:07 +0000269 ArgBackRefMap FunArgBackReferences;
270 ArgBackRefMap TemplateArgBackReferences;
David Majnemer2206bf52014-03-05 08:57:59 +0000271
Nico Weber48707762019-06-04 23:27:40 +0000272 typedef llvm::DenseMap<const void *, StringRef> TemplateArgStringMap;
273 TemplateArgStringMap TemplateArgStrings;
274 llvm::StringSaver TemplateArgStringStorage;
275 llvm::BumpPtrAllocator TemplateArgStringStorageAlloc;
276
Erik Pilkington02d5fb12019-03-19 20:44:18 +0000277 typedef std::set<std::pair<int, bool>> PassObjectSizeArgsSet;
David Majnemer5f177622015-12-30 05:13:03 +0000278 PassObjectSizeArgsSet PassObjectSizeArgs;
279
David Majnemer2206bf52014-03-05 08:57:59 +0000280 ASTContext &getASTContext() const { return Context.getASTContext(); }
281
282 // FIXME: If we add support for __ptr32/64 qualifiers, then we should push
283 // this check into mangleQualifiers().
284 const bool PointersAre64Bit;
285
286public:
287 enum QualifierMangleMode { QMM_Drop, QMM_Mangle, QMM_Escape, QMM_Result };
288
289 MicrosoftCXXNameMangler(MicrosoftMangleContextImpl &C, raw_ostream &Out_)
Craig Topper36250ad2014-05-12 05:36:57 +0000290 : Context(C), Out(Out_), Structor(nullptr), StructorType(-1),
Nico Weber48707762019-06-04 23:27:40 +0000291 TemplateArgStringStorage(TemplateArgStringStorageAlloc),
David Majnemer02e9af42014-04-23 05:16:51 +0000292 PointersAre64Bit(C.getASTContext().getTargetInfo().getPointerWidth(0) ==
293 64) {}
David Majnemer2206bf52014-03-05 08:57:59 +0000294
295 MicrosoftCXXNameMangler(MicrosoftMangleContextImpl &C, raw_ostream &Out_,
David Majnemerdfa6d202015-03-11 18:36:39 +0000296 const CXXConstructorDecl *D, CXXCtorType Type)
297 : Context(C), Out(Out_), Structor(getStructor(D)), StructorType(Type),
Nico Weber48707762019-06-04 23:27:40 +0000298 TemplateArgStringStorage(TemplateArgStringStorageAlloc),
David Majnemerdfa6d202015-03-11 18:36:39 +0000299 PointersAre64Bit(C.getASTContext().getTargetInfo().getPointerWidth(0) ==
300 64) {}
301
302 MicrosoftCXXNameMangler(MicrosoftMangleContextImpl &C, raw_ostream &Out_,
David Majnemer2206bf52014-03-05 08:57:59 +0000303 const CXXDestructorDecl *D, CXXDtorType Type)
David Majnemer02e9af42014-04-23 05:16:51 +0000304 : Context(C), Out(Out_), Structor(getStructor(D)), StructorType(Type),
Nico Weber48707762019-06-04 23:27:40 +0000305 TemplateArgStringStorage(TemplateArgStringStorageAlloc),
David Majnemer02e9af42014-04-23 05:16:51 +0000306 PointersAre64Bit(C.getASTContext().getTargetInfo().getPointerWidth(0) ==
307 64) {}
David Majnemer2206bf52014-03-05 08:57:59 +0000308
309 raw_ostream &getStream() const { return Out; }
310
Reid Klecknerfb931542018-03-16 20:36:49 +0000311 void mangle(const NamedDecl *D, StringRef Prefix = "?");
David Majnemer2206bf52014-03-05 08:57:59 +0000312 void mangleName(const NamedDecl *ND);
David Majnemerab4b4a12015-05-18 00:05:25 +0000313 void mangleFunctionEncoding(const FunctionDecl *FD, bool ShouldMangle);
David Majnemer2206bf52014-03-05 08:57:59 +0000314 void mangleVariableEncoding(const VarDecl *VD);
315 void mangleMemberDataPointer(const CXXRecordDecl *RD, const ValueDecl *VD);
316 void mangleMemberFunctionPointer(const CXXRecordDecl *RD,
317 const CXXMethodDecl *MD);
Reid Klecknercbec0262018-04-02 20:00:39 +0000318 void mangleVirtualMemPtrThunk(const CXXMethodDecl *MD,
319 const MethodVFTableLocation &ML);
David Majnemer2206bf52014-03-05 08:57:59 +0000320 void mangleNumber(int64_t Number);
David Majnemer3add70b2015-12-29 11:45:53 +0000321 void mangleTagTypeKind(TagTypeKind TK);
Shoaib Meenaid84cf132018-11-14 19:16:22 +0000322 void mangleArtificialTagType(TagTypeKind TK, StringRef UnqualifiedName,
David Majnemer3add70b2015-12-29 11:45:53 +0000323 ArrayRef<StringRef> NestedNames = None);
Erich Keane24a0f042018-12-14 23:17:34 +0000324 void mangleAddressSpaceType(QualType T, Qualifiers Quals, SourceRange Range);
David Majnemer2206bf52014-03-05 08:57:59 +0000325 void mangleType(QualType T, SourceRange Range,
326 QualifierMangleMode QMM = QMM_Mangle);
Craig Topper36250ad2014-05-12 05:36:57 +0000327 void mangleFunctionType(const FunctionType *T,
328 const FunctionDecl *D = nullptr,
Reid Klecknerd2f98772018-12-17 23:10:43 +0000329 bool ForceThisQuals = false,
330 bool MangleExceptionSpec = true);
David Majnemer2206bf52014-03-05 08:57:59 +0000331 void mangleNestedName(const NamedDecl *ND);
332
333private:
Reid Kleckner290feb92017-01-09 17:04:37 +0000334 bool isStructorDecl(const NamedDecl *ND) const {
335 return ND == Structor || getStructor(ND) == Structor;
336 }
337
David Majnemer2206bf52014-03-05 08:57:59 +0000338 void mangleUnqualifiedName(const NamedDecl *ND) {
339 mangleUnqualifiedName(ND, ND->getDeclName());
340 }
341 void mangleUnqualifiedName(const NamedDecl *ND, DeclarationName Name);
342 void mangleSourceName(StringRef Name);
343 void mangleOperatorName(OverloadedOperatorKind OO, SourceLocation Loc);
344 void mangleCXXDtorType(CXXDtorType T);
345 void mangleQualifiers(Qualifiers Quals, bool IsMember);
David Majnemere3785bb2014-04-23 05:16:56 +0000346 void mangleRefQualifier(RefQualifierKind RefQualifier);
David Majnemer2206bf52014-03-05 08:57:59 +0000347 void manglePointerCVQualifiers(Qualifiers Quals);
David Majnemer7a3ce0c2015-05-26 01:30:45 +0000348 void manglePointerExtQualifiers(Qualifiers Quals, QualType PointeeType);
David Majnemer2206bf52014-03-05 08:57:59 +0000349
350 void mangleUnscopedTemplateName(const TemplateDecl *ND);
David Majnemer02e9af42014-04-23 05:16:51 +0000351 void
352 mangleTemplateInstantiationName(const TemplateDecl *TD,
353 const TemplateArgumentList &TemplateArgs);
David Majnemer2206bf52014-03-05 08:57:59 +0000354 void mangleObjCMethodName(const ObjCMethodDecl *MD);
355
Nico Weber73f05842019-06-01 10:12:07 +0000356 void mangleFunctionArgumentType(QualType T, SourceRange Range);
David Majnemer5f177622015-12-30 05:13:03 +0000357 void manglePassObjectSizeArg(const PassObjectSizeAttr *POSA);
David Majnemer2206bf52014-03-05 08:57:59 +0000358
Nico Weber3239f132018-07-23 20:04:00 +0000359 bool isArtificialTagType(QualType T) const;
360
David Majnemer2206bf52014-03-05 08:57:59 +0000361 // Declare manglers for every type class.
362#define ABSTRACT_TYPE(CLASS, PARENT)
363#define NON_CANONICAL_TYPE(CLASS, PARENT)
364#define TYPE(CLASS, PARENT) void mangleType(const CLASS##Type *T, \
David Majnemer7a3ce0c2015-05-26 01:30:45 +0000365 Qualifiers Quals, \
David Majnemer2206bf52014-03-05 08:57:59 +0000366 SourceRange Range);
367#include "clang/AST/TypeNodes.def"
368#undef ABSTRACT_TYPE
369#undef NON_CANONICAL_TYPE
370#undef TYPE
371
372 void mangleType(const TagDecl *TD);
373 void mangleDecayedArrayType(const ArrayType *T);
374 void mangleArrayType(const ArrayType *T);
375 void mangleFunctionClass(const FunctionDecl *FD);
David Majnemerdfa6d202015-03-11 18:36:39 +0000376 void mangleCallingConvention(CallingConv CC);
David Majnemer2206bf52014-03-05 08:57:59 +0000377 void mangleCallingConvention(const FunctionType *T);
378 void mangleIntegerLiteral(const llvm::APSInt &Number, bool IsBoolean);
379 void mangleExpression(const Expr *E);
380 void mangleThrowSpecification(const FunctionProtoType *T);
381
382 void mangleTemplateArgs(const TemplateDecl *TD,
383 const TemplateArgumentList &TemplateArgs);
David Majnemer57fbc0c2014-08-05 22:43:45 +0000384 void mangleTemplateArg(const TemplateDecl *TD, const TemplateArgument &TA,
385 const NamedDecl *Parm);
Saleem Abdulrasoolbbce34a2018-02-09 03:23:54 +0000386
387 void mangleObjCProtocol(const ObjCProtocolDecl *PD);
388 void mangleObjCLifetime(const QualType T, Qualifiers Quals,
389 SourceRange Range);
Saleem Abdulrasoolf3a981e2018-09-28 16:47:53 +0000390 void mangleObjCKindOfType(const ObjCObjectType *T, Qualifiers Quals,
391 SourceRange Range);
David Majnemer2206bf52014-03-05 08:57:59 +0000392};
Alexander Kornienkoab9db512015-06-22 23:07:51 +0000393}
Guy Benyei11169dd2012-12-18 14:30:41 +0000394
Reid Klecknerdf8dd612018-08-17 20:59:27 +0000395MicrosoftMangleContextImpl::MicrosoftMangleContextImpl(ASTContext &Context,
396 DiagnosticsEngine &Diags)
397 : MicrosoftMangleContext(Context, Diags) {
398 // To mangle anonymous namespaces, hash the path to the main source file. The
399 // path should be whatever (probably relative) path was passed on the command
400 // line. The goal is for the compiler to produce the same output regardless of
401 // working directory, so use the uncanonicalized relative path.
402 //
403 // It's important to make the mangled names unique because, when CodeView
404 // debug info is in use, the debugger uses mangled type names to distinguish
405 // between otherwise identically named types in anonymous namespaces.
406 //
407 // These symbols are always internal, so there is no need for the hash to
408 // match what MSVC produces. For the same reason, clang is free to change the
409 // hash at any time without breaking compatibility with old versions of clang.
410 // The generated names are intended to look similar to what MSVC generates,
411 // which are something like "?A0x01234567@".
412 SourceManager &SM = Context.getSourceManager();
413 if (const FileEntry *FE = SM.getFileEntryForID(SM.getMainFileID())) {
414 // Truncate the hash so we get 8 characters of hexadecimal.
415 uint32_t TruncatedHash = uint32_t(xxHash64(FE->getName()));
416 AnonymousNamespaceHash = llvm::utohexstr(TruncatedHash);
417 } else {
418 // If we don't have a path to the main file, we'll just use 0.
419 AnonymousNamespaceHash = "0";
420 }
421}
422
Rafael Espindola002667c2013-10-16 01:40:34 +0000423bool MicrosoftMangleContextImpl::shouldMangleCXXName(const NamedDecl *D) {
David Majnemerd5a42b82013-09-13 09:03:14 +0000424 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
425 LanguageLinkage L = FD->getLanguageLinkage();
426 // Overloadable functions need mangling.
427 if (FD->hasAttr<OverloadableAttr>())
428 return true;
429
David Majnemerc729b0b2013-09-16 22:44:20 +0000430 // The ABI expects that we would never mangle "typical" user-defined entry
431 // points regardless of visibility or freestanding-ness.
432 //
433 // N.B. This is distinct from asking about "main". "main" has a lot of
434 // special rules associated with it in the standard while these
435 // user-defined entry points are outside of the purview of the standard.
436 // For example, there can be only one definition for "main" in a standards
437 // compliant program; however nothing forbids the existence of wmain and
438 // WinMain in the same translation unit.
439 if (FD->isMSVCRTEntryPoint())
David Majnemerd5a42b82013-09-13 09:03:14 +0000440 return false;
441
442 // C++ functions and those whose names are not a simple identifier need
443 // mangling.
444 if (!FD->getDeclName().isIdentifier() || L == CXXLanguageLinkage)
445 return true;
446
447 // C functions are not mangled.
448 if (L == CLanguageLinkage)
449 return false;
450 }
Guy Benyei11169dd2012-12-18 14:30:41 +0000451
452 // Otherwise, no mangling is done outside C++ mode.
453 if (!getASTContext().getLangOpts().CPlusPlus)
454 return false;
455
Richard Smithda383632016-08-15 01:33:41 +0000456 const VarDecl *VD = dyn_cast<VarDecl>(D);
457 if (VD && !isa<DecompositionDecl>(D)) {
David Majnemerd5a42b82013-09-13 09:03:14 +0000458 // C variables are not mangled.
459 if (VD->isExternC())
460 return false;
461
462 // Variables at global scope with non-internal linkage are not mangled.
463 const DeclContext *DC = getEffectiveDeclContext(D);
464 // Check for extern variable declared locally.
465 if (DC->isFunctionOrMethod() && D->hasLinkage())
466 while (!DC->isNamespace() && !DC->isTranslationUnit())
467 DC = getEffectiveParentContext(DC);
468
469 if (DC->isTranslationUnit() && D->getFormalLinkage() == InternalLinkage &&
David Majnemeree6c4092015-10-22 07:15:56 +0000470 !isa<VarTemplateSpecializationDecl>(D) &&
471 D->getIdentifier() != nullptr)
Guy Benyei11169dd2012-12-18 14:30:41 +0000472 return false;
473 }
474
Guy Benyei11169dd2012-12-18 14:30:41 +0000475 return true;
476}
477
David Majnemer58e5bee2014-03-24 21:43:36 +0000478bool
479MicrosoftMangleContextImpl::shouldMangleStringLiteral(const StringLiteral *SL) {
David Majnemer6fc2d812014-11-22 06:20:38 +0000480 return true;
David Majnemer58e5bee2014-03-24 21:43:36 +0000481}
482
David Majnemer02e9af42014-04-23 05:16:51 +0000483void MicrosoftCXXNameMangler::mangle(const NamedDecl *D, StringRef Prefix) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000484 // MSVC doesn't mangle C++ names the same way it mangles extern "C" names.
485 // Therefore it's really important that we don't decorate the
486 // name with leading underscores or leading/trailing at signs. So, by
487 // default, we emit an asm marker at the start so we get the name right.
488 // Callers can override this with a custom prefix.
489
Guy Benyei11169dd2012-12-18 14:30:41 +0000490 // <mangled-name> ::= ? <name> <type-encoding>
491 Out << Prefix;
492 mangleName(D);
493 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
David Majnemerab4b4a12015-05-18 00:05:25 +0000494 mangleFunctionEncoding(FD, Context.shouldMangleDeclName(FD));
Guy Benyei11169dd2012-12-18 14:30:41 +0000495 else if (const VarDecl *VD = dyn_cast<VarDecl>(D))
496 mangleVariableEncoding(VD);
Shoaib Meenaic8714d22018-10-04 19:50:14 +0000497 else
David Majnemer53b60a12015-12-29 22:02:10 +0000498 llvm_unreachable("Tried to mangle unexpected NamedDecl!");
Guy Benyei11169dd2012-12-18 14:30:41 +0000499}
500
David Majnemerab4b4a12015-05-18 00:05:25 +0000501void MicrosoftCXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD,
502 bool ShouldMangle) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000503 // <type-encoding> ::= <function-class> <function-type>
504
Reid Kleckner18da98e2013-06-24 19:21:52 +0000505 // Since MSVC operates on the type as written and not the canonical type, it
506 // actually matters which decl we have here. MSVC appears to choose the
507 // first, since it is most likely to be the declaration in a header file.
Rafael Espindola8db352d2013-10-17 15:37:26 +0000508 FD = FD->getFirstDecl();
Reid Kleckner18da98e2013-06-24 19:21:52 +0000509
Guy Benyei11169dd2012-12-18 14:30:41 +0000510 // We should never ever see a FunctionNoProtoType at this point.
511 // We don't even know how to mangle their types anyway :).
Reid Kleckner9a7f3e62013-10-08 00:58:57 +0000512 const FunctionProtoType *FT = FD->getType()->castAs<FunctionProtoType>();
Guy Benyei11169dd2012-12-18 14:30:41 +0000513
David Majnemerd5a42b82013-09-13 09:03:14 +0000514 // extern "C" functions can hold entities that must be mangled.
515 // As it stands, these functions still need to get expressed in the full
516 // external name. They have their class and type omitted, replaced with '9'.
David Majnemerab4b4a12015-05-18 00:05:25 +0000517 if (ShouldMangle) {
David Majnemer5a7cfea2015-05-18 00:05:29 +0000518 // We would like to mangle all extern "C" functions using this additional
519 // component but this would break compatibility with MSVC's behavior.
520 // Instead, do this when we know that compatibility isn't important (in
Eric Christopherbe528662015-12-16 23:10:46 +0000521 // other words, when it is an overloaded extern "C" function).
David Majnemer5a7cfea2015-05-18 00:05:29 +0000522 if (FD->isExternC() && FD->hasAttr<OverloadableAttr>())
523 Out << "$$J0";
524
David Majnemerd5a42b82013-09-13 09:03:14 +0000525 mangleFunctionClass(FD);
Guy Benyei11169dd2012-12-18 14:30:41 +0000526
Reid Klecknerd2f98772018-12-17 23:10:43 +0000527 mangleFunctionType(FT, FD, false, false);
David Majnemerab4b4a12015-05-18 00:05:25 +0000528 } else {
David Majnemerd5a42b82013-09-13 09:03:14 +0000529 Out << '9';
David Majnemerab4b4a12015-05-18 00:05:25 +0000530 }
Guy Benyei11169dd2012-12-18 14:30:41 +0000531}
532
533void MicrosoftCXXNameMangler::mangleVariableEncoding(const VarDecl *VD) {
534 // <type-encoding> ::= <storage-class> <variable-type>
535 // <storage-class> ::= 0 # private static member
536 // ::= 1 # protected static member
537 // ::= 2 # public static member
538 // ::= 3 # global
539 // ::= 4 # static local
David Majnemerb4119f72013-12-13 01:06:04 +0000540
Guy Benyei11169dd2012-12-18 14:30:41 +0000541 // The first character in the encoding (after the name) is the storage class.
542 if (VD->isStaticDataMember()) {
543 // If it's a static member, it also encodes the access level.
544 switch (VD->getAccess()) {
545 default:
546 case AS_private: Out << '0'; break;
547 case AS_protected: Out << '1'; break;
548 case AS_public: Out << '2'; break;
549 }
550 }
551 else if (!VD->isStaticLocal())
552 Out << '3';
553 else
554 Out << '4';
555 // Now mangle the type.
556 // <variable-type> ::= <type> <cvr-qualifiers>
557 // ::= <type> <pointee-cvr-qualifiers> # pointers, references
558 // Pointers and references are odd. The type of 'int * const foo;' gets
559 // mangled as 'QAHA' instead of 'PAHB', for example.
Zachary Turner87422d92014-06-25 05:37:25 +0000560 SourceRange SR = VD->getSourceRange();
David Majnemerb9a5f2d2014-01-21 20:33:36 +0000561 QualType Ty = VD->getType();
David Majnemer6dda7bb2013-08-15 08:13:23 +0000562 if (Ty->isPointerType() || Ty->isReferenceType() ||
563 Ty->isMemberPointerType()) {
Zachary Turner87422d92014-06-25 05:37:25 +0000564 mangleType(Ty, SR, QMM_Drop);
David Majnemer8eec58f2014-02-18 14:20:10 +0000565 manglePointerExtQualifiers(
David Majnemer7a3ce0c2015-05-26 01:30:45 +0000566 Ty.getDesugaredType(getASTContext()).getLocalQualifiers(), QualType());
David Majnemer6dda7bb2013-08-15 08:13:23 +0000567 if (const MemberPointerType *MPT = Ty->getAs<MemberPointerType>()) {
568 mangleQualifiers(MPT->getPointeeType().getQualifiers(), true);
569 // Member pointers are suffixed with a back reference to the member
570 // pointer's class name.
571 mangleName(MPT->getClass()->getAsCXXRecordDecl());
572 } else
573 mangleQualifiers(Ty->getPointeeType().getQualifiers(), false);
David Majnemera2724ae2013-08-09 05:56:24 +0000574 } else if (const ArrayType *AT = getASTContext().getAsArrayType(Ty)) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000575 // Global arrays are funny, too.
David Majnemer5a1b2042013-09-11 04:44:30 +0000576 mangleDecayedArrayType(AT);
Peter Collingbourne2816c022013-04-25 04:25:40 +0000577 if (AT->getElementType()->isArrayType())
578 Out << 'A';
579 else
580 mangleQualifiers(Ty.getQualifiers(), false);
Guy Benyei11169dd2012-12-18 14:30:41 +0000581 } else {
Zachary Turner87422d92014-06-25 05:37:25 +0000582 mangleType(Ty, SR, QMM_Drop);
Will Wilson5f383672014-11-05 13:54:21 +0000583 mangleQualifiers(Ty.getQualifiers(), false);
Guy Benyei11169dd2012-12-18 14:30:41 +0000584 }
585}
586
Reid Kleckner96f8f932014-02-05 17:27:08 +0000587void MicrosoftCXXNameMangler::mangleMemberDataPointer(const CXXRecordDecl *RD,
David Majnemer1e378e42014-02-06 12:46:52 +0000588 const ValueDecl *VD) {
Reid Kleckner96f8f932014-02-05 17:27:08 +0000589 // <member-data-pointer> ::= <integer-literal>
590 // ::= $F <number> <number>
591 // ::= $G <number> <number> <number>
592
David Majnemer763584d2014-02-06 10:59:19 +0000593 int64_t FieldOffset;
594 int64_t VBTableOffset;
Reid Kleckner96f8f932014-02-05 17:27:08 +0000595 MSInheritanceAttr::Spelling IM = RD->getMSInheritanceModel();
David Majnemer1e378e42014-02-06 12:46:52 +0000596 if (VD) {
597 FieldOffset = getASTContext().getFieldOffset(VD);
David Majnemer763584d2014-02-06 10:59:19 +0000598 assert(FieldOffset % getASTContext().getCharWidth() == 0 &&
Reid Kleckner96f8f932014-02-05 17:27:08 +0000599 "cannot take address of bitfield");
David Majnemer763584d2014-02-06 10:59:19 +0000600 FieldOffset /= getASTContext().getCharWidth();
601
602 VBTableOffset = 0;
David Majnemer08ef2ba2015-06-23 20:34:18 +0000603
604 if (IM == MSInheritanceAttr::Keyword_virtual_inheritance)
605 FieldOffset -= getASTContext().getOffsetOfBaseWithVBPtr(RD).getQuantity();
David Majnemer763584d2014-02-06 10:59:19 +0000606 } else {
607 FieldOffset = RD->nullFieldOffsetIsZero() ? 0 : -1;
608
609 VBTableOffset = -1;
Reid Kleckner96f8f932014-02-05 17:27:08 +0000610 }
611
David Majnemer763584d2014-02-06 10:59:19 +0000612 char Code = '\0';
Reid Kleckner96f8f932014-02-05 17:27:08 +0000613 switch (IM) {
David Majnemer763584d2014-02-06 10:59:19 +0000614 case MSInheritanceAttr::Keyword_single_inheritance: Code = '0'; break;
615 case MSInheritanceAttr::Keyword_multiple_inheritance: Code = '0'; break;
616 case MSInheritanceAttr::Keyword_virtual_inheritance: Code = 'F'; break;
617 case MSInheritanceAttr::Keyword_unspecified_inheritance: Code = 'G'; break;
Reid Kleckner96f8f932014-02-05 17:27:08 +0000618 }
619
David Majnemer763584d2014-02-06 10:59:19 +0000620 Out << '$' << Code;
Reid Kleckner96f8f932014-02-05 17:27:08 +0000621
David Majnemer763584d2014-02-06 10:59:19 +0000622 mangleNumber(FieldOffset);
623
Reid Klecknerbf94e6e2014-04-07 18:07:03 +0000624 // The C++ standard doesn't allow base-to-derived member pointer conversions
625 // in template parameter contexts, so the vbptr offset of data member pointers
626 // is always zero.
David Majnemer763584d2014-02-06 10:59:19 +0000627 if (MSInheritanceAttr::hasVBPtrOffsetField(IM))
Reid Kleckner96f8f932014-02-05 17:27:08 +0000628 mangleNumber(0);
David Majnemer763584d2014-02-06 10:59:19 +0000629 if (MSInheritanceAttr::hasVBTableOffsetField(IM))
630 mangleNumber(VBTableOffset);
Reid Kleckner96f8f932014-02-05 17:27:08 +0000631}
632
633void
634MicrosoftCXXNameMangler::mangleMemberFunctionPointer(const CXXRecordDecl *RD,
635 const CXXMethodDecl *MD) {
636 // <member-function-pointer> ::= $1? <name>
637 // ::= $H? <name> <number>
638 // ::= $I? <name> <number> <number>
639 // ::= $J? <name> <number> <number> <number>
Reid Kleckner96f8f932014-02-05 17:27:08 +0000640
641 MSInheritanceAttr::Spelling IM = RD->getMSInheritanceModel();
642
Reid Kleckner96f8f932014-02-05 17:27:08 +0000643 char Code = '\0';
644 switch (IM) {
645 case MSInheritanceAttr::Keyword_single_inheritance: Code = '1'; break;
646 case MSInheritanceAttr::Keyword_multiple_inheritance: Code = 'H'; break;
647 case MSInheritanceAttr::Keyword_virtual_inheritance: Code = 'I'; break;
648 case MSInheritanceAttr::Keyword_unspecified_inheritance: Code = 'J'; break;
649 }
650
Reid Kleckner96f8f932014-02-05 17:27:08 +0000651 // If non-virtual, mangle the name. If virtual, mangle as a virtual memptr
652 // thunk.
653 uint64_t NVOffset = 0;
654 uint64_t VBTableOffset = 0;
Reid Klecknerbf94e6e2014-04-07 18:07:03 +0000655 uint64_t VBPtrOffset = 0;
David Majnemer6a729c62014-06-11 04:55:08 +0000656 if (MD) {
657 Out << '$' << Code << '?';
658 if (MD->isVirtual()) {
659 MicrosoftVTableContext *VTContext =
660 cast<MicrosoftVTableContext>(getASTContext().getVTableContext());
Reid Klecknercbec0262018-04-02 20:00:39 +0000661 MethodVFTableLocation ML =
David Majnemer6a729c62014-06-11 04:55:08 +0000662 VTContext->getMethodVFTableLocation(GlobalDecl(MD));
663 mangleVirtualMemPtrThunk(MD, ML);
664 NVOffset = ML.VFPtrOffset.getQuantity();
665 VBTableOffset = ML.VBTableIndex * 4;
666 if (ML.VBase) {
667 const ASTRecordLayout &Layout = getASTContext().getASTRecordLayout(RD);
668 VBPtrOffset = Layout.getVBPtrOffset().getQuantity();
669 }
670 } else {
671 mangleName(MD);
David Majnemerab4b4a12015-05-18 00:05:25 +0000672 mangleFunctionEncoding(MD, /*ShouldMangle=*/true);
Reid Kleckner96f8f932014-02-05 17:27:08 +0000673 }
David Majnemer08ef2ba2015-06-23 20:34:18 +0000674
675 if (VBTableOffset == 0 &&
676 IM == MSInheritanceAttr::Keyword_virtual_inheritance)
677 NVOffset -= getASTContext().getOffsetOfBaseWithVBPtr(RD).getQuantity();
Reid Kleckner96f8f932014-02-05 17:27:08 +0000678 } else {
David Majnemer6a729c62014-06-11 04:55:08 +0000679 // Null single inheritance member functions are encoded as a simple nullptr.
680 if (IM == MSInheritanceAttr::Keyword_single_inheritance) {
681 Out << "$0A@";
682 return;
683 }
684 if (IM == MSInheritanceAttr::Keyword_unspecified_inheritance)
685 VBTableOffset = -1;
686 Out << '$' << Code;
Reid Kleckner96f8f932014-02-05 17:27:08 +0000687 }
688
689 if (MSInheritanceAttr::hasNVOffsetField(/*IsMemberFunction=*/true, IM))
David Majnemer08ef2ba2015-06-23 20:34:18 +0000690 mangleNumber(static_cast<uint32_t>(NVOffset));
Reid Kleckner96f8f932014-02-05 17:27:08 +0000691 if (MSInheritanceAttr::hasVBPtrOffsetField(IM))
Reid Klecknerbf94e6e2014-04-07 18:07:03 +0000692 mangleNumber(VBPtrOffset);
Reid Kleckner96f8f932014-02-05 17:27:08 +0000693 if (MSInheritanceAttr::hasVBTableOffsetField(IM))
694 mangleNumber(VBTableOffset);
695}
696
697void MicrosoftCXXNameMangler::mangleVirtualMemPtrThunk(
Reid Klecknercbec0262018-04-02 20:00:39 +0000698 const CXXMethodDecl *MD, const MethodVFTableLocation &ML) {
Reid Kleckner96f8f932014-02-05 17:27:08 +0000699 // Get the vftable offset.
700 CharUnits PointerWidth = getASTContext().toCharUnitsFromBits(
701 getASTContext().getTargetInfo().getPointerWidth(0));
702 uint64_t OffsetInVFTable = ML.Index * PointerWidth.getQuantity();
703
704 Out << "?_9";
705 mangleName(MD->getParent());
706 Out << "$B";
707 mangleNumber(OffsetInVFTable);
708 Out << 'A';
David Majnemerbc02d322015-03-14 06:34:41 +0000709 mangleCallingConvention(MD->getType()->getAs<FunctionProtoType>());
Reid Kleckner96f8f932014-02-05 17:27:08 +0000710}
711
Guy Benyei11169dd2012-12-18 14:30:41 +0000712void MicrosoftCXXNameMangler::mangleName(const NamedDecl *ND) {
713 // <name> ::= <unscoped-name> {[<named-scope>]+ | [<nested-name>]}? @
Guy Benyei11169dd2012-12-18 14:30:41 +0000714
715 // Always start with the unqualified name.
David Majnemerb4119f72013-12-13 01:06:04 +0000716 mangleUnqualifiedName(ND);
Guy Benyei11169dd2012-12-18 14:30:41 +0000717
David Majnemer2206bf52014-03-05 08:57:59 +0000718 mangleNestedName(ND);
Guy Benyei11169dd2012-12-18 14:30:41 +0000719
720 // Terminate the whole name with an '@'.
721 Out << '@';
722}
723
David Majnemer2a816452013-12-09 10:44:32 +0000724void MicrosoftCXXNameMangler::mangleNumber(int64_t Number) {
725 // <non-negative integer> ::= A@ # when Number == 0
726 // ::= <decimal digit> # when 1 <= Number <= 10
727 // ::= <hex digit>+ @ # when Number >= 10
728 //
729 // <number> ::= [?] <non-negative integer>
Guy Benyei11169dd2012-12-18 14:30:41 +0000730
David Majnemer2a816452013-12-09 10:44:32 +0000731 uint64_t Value = static_cast<uint64_t>(Number);
732 if (Number < 0) {
733 Value = -Value;
Guy Benyei11169dd2012-12-18 14:30:41 +0000734 Out << '?';
Guy Benyei11169dd2012-12-18 14:30:41 +0000735 }
David Majnemer2a816452013-12-09 10:44:32 +0000736
737 if (Value == 0)
738 Out << "A@";
739 else if (Value >= 1 && Value <= 10)
740 Out << (Value - 1);
741 else {
742 // Numbers that are not encoded as decimal digits are represented as nibbles
743 // in the range of ASCII characters 'A' to 'P'.
744 // The number 0x123450 would be encoded as 'BCDEFA'
745 char EncodedNumberBuffer[sizeof(uint64_t) * 2];
Craig Toppere3d2ecbe2014-06-28 23:22:33 +0000746 MutableArrayRef<char> BufferRef(EncodedNumberBuffer);
747 MutableArrayRef<char>::reverse_iterator I = BufferRef.rbegin();
David Majnemer2a816452013-12-09 10:44:32 +0000748 for (; Value != 0; Value >>= 4)
749 *I++ = 'A' + (Value & 0xf);
750 Out.write(I.base(), I - BufferRef.rbegin());
Guy Benyei11169dd2012-12-18 14:30:41 +0000751 Out << '@';
752 }
753}
754
755static const TemplateDecl *
Reid Kleckner52518862013-03-20 01:40:23 +0000756isTemplate(const NamedDecl *ND, const TemplateArgumentList *&TemplateArgs) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000757 // Check if we have a function template.
David Majnemer02e9af42014-04-23 05:16:51 +0000758 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000759 if (const TemplateDecl *TD = FD->getPrimaryTemplate()) {
Reid Kleckner52518862013-03-20 01:40:23 +0000760 TemplateArgs = FD->getTemplateSpecializationArgs();
Guy Benyei11169dd2012-12-18 14:30:41 +0000761 return TD;
762 }
763 }
764
765 // Check if we have a class template.
766 if (const ClassTemplateSpecializationDecl *Spec =
David Majnemer02e9af42014-04-23 05:16:51 +0000767 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
Reid Kleckner52518862013-03-20 01:40:23 +0000768 TemplateArgs = &Spec->getTemplateArgs();
Guy Benyei11169dd2012-12-18 14:30:41 +0000769 return Spec->getSpecializedTemplate();
770 }
771
David Majnemer8f774532014-03-04 05:38:05 +0000772 // Check if we have a variable template.
773 if (const VarTemplateSpecializationDecl *Spec =
774 dyn_cast<VarTemplateSpecializationDecl>(ND)) {
775 TemplateArgs = &Spec->getTemplateArgs();
776 return Spec->getSpecializedTemplate();
777 }
778
Craig Topper36250ad2014-05-12 05:36:57 +0000779 return nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +0000780}
781
David Majnemer02e9af42014-04-23 05:16:51 +0000782void MicrosoftCXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND,
783 DeclarationName Name) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000784 // <unqualified-name> ::= <operator-name>
785 // ::= <ctor-dtor-name>
786 // ::= <source-name>
787 // ::= <template-name>
Reid Kleckner52518862013-03-20 01:40:23 +0000788
Guy Benyei11169dd2012-12-18 14:30:41 +0000789 // Check if we have a template.
Craig Topper36250ad2014-05-12 05:36:57 +0000790 const TemplateArgumentList *TemplateArgs = nullptr;
Guy Benyei11169dd2012-12-18 14:30:41 +0000791 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Reid Klecknerc16c4472013-07-13 00:43:39 +0000792 // Function templates aren't considered for name back referencing. This
793 // makes sense since function templates aren't likely to occur multiple
794 // times in a symbol.
David Majnemer56e2adb2015-12-30 05:40:32 +0000795 if (isa<FunctionTemplateDecl>(TD)) {
Reid Klecknerc16c4472013-07-13 00:43:39 +0000796 mangleTemplateInstantiationName(TD, *TemplateArgs);
Reid Kleckner11c6f612014-06-26 22:42:18 +0000797 Out << '@';
Reid Klecknerc16c4472013-07-13 00:43:39 +0000798 return;
799 }
800
Guy Benyei11169dd2012-12-18 14:30:41 +0000801 // Here comes the tricky thing: if we need to mangle something like
802 // void foo(A::X<Y>, B::X<Y>),
803 // the X<Y> part is aliased. However, if you need to mangle
804 // void foo(A::X<A::Y>, A::X<B::Y>),
805 // the A::X<> part is not aliased.
Nico Weber73f05842019-06-01 10:12:07 +0000806 // That is, from the mangler's perspective we have a structure like this:
Guy Benyei11169dd2012-12-18 14:30:41 +0000807 // namespace[s] -> type[ -> template-parameters]
808 // but from the Clang perspective we have
809 // type [ -> template-parameters]
810 // \-> namespace[s]
811 // What we do is we create a new mangler, mangle the same type (without
Fangrui Song6907ce22018-07-30 19:24:48 +0000812 // a namespace suffix) to a string using the extra mangler and then use
David Majnemerc986fcc2014-06-08 04:51:13 +0000813 // the mangled type name as a key to check the mangling of different types
814 // for aliasing.
Guy Benyei11169dd2012-12-18 14:30:41 +0000815
Nico Weber73f05842019-06-01 10:12:07 +0000816 // It's important to key cache reads off ND, not TD -- the same TD can
817 // be used with different TemplateArgs, but ND uniquely identifies
818 // TD / TemplateArg pairs.
819 ArgBackRefMap::iterator Found = TemplateArgBackReferences.find(ND);
820 if (Found == TemplateArgBackReferences.end()) {
Guy Benyei11169dd2012-12-18 14:30:41 +0000821
Nico Weber48707762019-06-04 23:27:40 +0000822 TemplateArgStringMap::iterator Found = TemplateArgStrings.find(ND);
823 if (Found == TemplateArgStrings.end()) {
824 // Mangle full template name into temporary buffer.
825 llvm::SmallString<64> TemplateMangling;
826 llvm::raw_svector_ostream Stream(TemplateMangling);
827 MicrosoftCXXNameMangler Extra(Context, Stream);
828 Extra.mangleTemplateInstantiationName(TD, *TemplateArgs);
Nico Weber73f05842019-06-01 10:12:07 +0000829
Nico Weber48707762019-06-04 23:27:40 +0000830 // Use the string backref vector to possibly get a back reference.
831 mangleSourceName(TemplateMangling);
832
833 // Memoize back reference for this type if one exist, else memoize
834 // the mangling itself.
835 BackRefVec::iterator StringFound =
836 llvm::find(NameBackReferences, TemplateMangling);
837 if (StringFound != NameBackReferences.end()) {
838 TemplateArgBackReferences[ND] =
839 StringFound - NameBackReferences.begin();
840 } else {
841 TemplateArgStrings[ND] =
842 TemplateArgStringStorage.save(TemplateMangling.str());
843 }
844 } else {
845 Out << Found->second; // Outputs a StringRef.
Nico Weber73f05842019-06-01 10:12:07 +0000846 }
847 } else {
Nico Weber48707762019-06-04 23:27:40 +0000848 Out << Found->second; // Outputs a back reference (an int).
Nico Weber73f05842019-06-01 10:12:07 +0000849 }
Guy Benyei11169dd2012-12-18 14:30:41 +0000850 return;
851 }
852
853 switch (Name.getNameKind()) {
854 case DeclarationName::Identifier: {
855 if (const IdentifierInfo *II = Name.getAsIdentifierInfo()) {
David Majnemer956bc112013-11-25 17:50:19 +0000856 mangleSourceName(II->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +0000857 break;
858 }
David Majnemerb4119f72013-12-13 01:06:04 +0000859
Guy Benyei11169dd2012-12-18 14:30:41 +0000860 // Otherwise, an anonymous entity. We must have a declaration.
861 assert(ND && "mangling empty name without declaration");
David Majnemerb4119f72013-12-13 01:06:04 +0000862
Guy Benyei11169dd2012-12-18 14:30:41 +0000863 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
864 if (NS->isAnonymousNamespace()) {
Reid Klecknerdf8dd612018-08-17 20:59:27 +0000865 Out << "?A0x" << Context.getAnonymousNamespaceHash() << '@';
Guy Benyei11169dd2012-12-18 14:30:41 +0000866 break;
867 }
868 }
David Majnemerb4119f72013-12-13 01:06:04 +0000869
Richard Smithda383632016-08-15 01:33:41 +0000870 if (const DecompositionDecl *DD = dyn_cast<DecompositionDecl>(ND)) {
871 // FIXME: Invented mangling for decomposition declarations:
872 // [X,Y,Z]
873 // where X,Y,Z are the names of the bindings.
874 llvm::SmallString<128> Name("[");
875 for (auto *BD : DD->bindings()) {
876 if (Name.size() > 1)
877 Name += ',';
878 Name += BD->getDeclName().getAsIdentifierInfo()->getName();
879 }
880 Name += ']';
881 mangleSourceName(Name);
882 break;
883 }
884
David Majnemer2206bf52014-03-05 08:57:59 +0000885 if (const VarDecl *VD = dyn_cast<VarDecl>(ND)) {
886 // We must have an anonymous union or struct declaration.
887 const CXXRecordDecl *RD = VD->getType()->getAsCXXRecordDecl();
888 assert(RD && "expected variable decl to have a record type");
889 // Anonymous types with no tag or typedef get the name of their
890 // declarator mangled in. If they have no declarator, number them with
891 // a $S prefix.
892 llvm::SmallString<64> Name("$S");
893 // Get a unique id for the anonymous struct.
894 Name += llvm::utostr(Context.getAnonymousStructId(RD) + 1);
895 mangleSourceName(Name.str());
896 break;
897 }
898
Guy Benyei11169dd2012-12-18 14:30:41 +0000899 // We must have an anonymous struct.
900 const TagDecl *TD = cast<TagDecl>(ND);
901 if (const TypedefNameDecl *D = TD->getTypedefNameForAnonDecl()) {
902 assert(TD->getDeclContext() == D->getDeclContext() &&
903 "Typedef should not be in another decl context!");
904 assert(D->getDeclName().getAsIdentifierInfo() &&
905 "Typedef was not named!");
David Majnemer956bc112013-11-25 17:50:19 +0000906 mangleSourceName(D->getDeclName().getAsIdentifierInfo()->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +0000907 break;
908 }
909
David Majnemerf017ec32014-03-05 10:35:06 +0000910 if (const CXXRecordDecl *Record = dyn_cast<CXXRecordDecl>(TD)) {
911 if (Record->isLambda()) {
912 llvm::SmallString<10> Name("<lambda_");
David Majnemer984498f2016-12-06 17:01:02 +0000913
914 Decl *LambdaContextDecl = Record->getLambdaContextDecl();
915 unsigned LambdaManglingNumber = Record->getLambdaManglingNumber();
David Majnemerf017ec32014-03-05 10:35:06 +0000916 unsigned LambdaId;
David Majnemer984498f2016-12-06 17:01:02 +0000917 const ParmVarDecl *Parm =
918 dyn_cast_or_null<ParmVarDecl>(LambdaContextDecl);
919 const FunctionDecl *Func =
920 Parm ? dyn_cast<FunctionDecl>(Parm->getDeclContext()) : nullptr;
921
922 if (Func) {
923 unsigned DefaultArgNo =
924 Func->getNumParams() - Parm->getFunctionScopeIndex();
925 Name += llvm::utostr(DefaultArgNo);
926 Name += "_";
927 }
928
929 if (LambdaManglingNumber)
930 LambdaId = LambdaManglingNumber;
David Majnemerf017ec32014-03-05 10:35:06 +0000931 else
932 LambdaId = Context.getLambdaId(Record);
933
934 Name += llvm::utostr(LambdaId);
935 Name += ">";
936
937 mangleSourceName(Name);
David Majnemer984498f2016-12-06 17:01:02 +0000938
939 // If the context of a closure type is an initializer for a class
940 // member (static or nonstatic), it is encoded in a qualified name.
941 if (LambdaManglingNumber && LambdaContextDecl) {
942 if ((isa<VarDecl>(LambdaContextDecl) ||
943 isa<FieldDecl>(LambdaContextDecl)) &&
944 LambdaContextDecl->getDeclContext()->isRecord()) {
945 mangleUnqualifiedName(cast<NamedDecl>(LambdaContextDecl));
946 }
947 }
David Majnemerf017ec32014-03-05 10:35:06 +0000948 break;
949 }
950 }
951
David Majnemera5cfddc2016-12-25 05:26:02 +0000952 llvm::SmallString<64> Name;
David Majnemer00350522015-08-31 18:48:39 +0000953 if (DeclaratorDecl *DD =
954 Context.getASTContext().getDeclaratorForUnnamedTagDecl(TD)) {
955 // Anonymous types without a name for linkage purposes have their
David Majnemer2206bf52014-03-05 08:57:59 +0000956 // declarator mangled in if they have one.
David Majnemera5cfddc2016-12-25 05:26:02 +0000957 Name += "<unnamed-type-";
David Majnemer00350522015-08-31 18:48:39 +0000958 Name += DD->getName();
959 } else if (TypedefNameDecl *TND =
960 Context.getASTContext().getTypedefNameForUnnamedTagDecl(
961 TD)) {
962 // Anonymous types without a name for linkage purposes have their
963 // associate typedef mangled in if they have one.
David Majnemera5cfddc2016-12-25 05:26:02 +0000964 Name += "<unnamed-type-";
David Majnemer00350522015-08-31 18:48:39 +0000965 Name += TND->getName();
Hans Wennborg4cae35f2018-06-11 06:54:23 +0000966 } else if (isa<EnumDecl>(TD) &&
967 cast<EnumDecl>(TD)->enumerator_begin() !=
968 cast<EnumDecl>(TD)->enumerator_end()) {
969 // Anonymous non-empty enums mangle in the first enumerator.
970 auto *ED = cast<EnumDecl>(TD);
David Majnemera5cfddc2016-12-25 05:26:02 +0000971 Name += "<unnamed-enum-";
Hans Wennborg4cae35f2018-06-11 06:54:23 +0000972 Name += ED->enumerator_begin()->getName();
David Majnemer956bc112013-11-25 17:50:19 +0000973 } else {
David Majnemer2206bf52014-03-05 08:57:59 +0000974 // Otherwise, number the types using a $S prefix.
David Majnemera5cfddc2016-12-25 05:26:02 +0000975 Name += "<unnamed-type-$S";
David Majnemeree6c4092015-10-22 07:15:56 +0000976 Name += llvm::utostr(Context.getAnonymousStructId(TD) + 1);
David Majnemer956bc112013-11-25 17:50:19 +0000977 }
David Majnemer2206bf52014-03-05 08:57:59 +0000978 Name += ">";
979 mangleSourceName(Name.str());
Guy Benyei11169dd2012-12-18 14:30:41 +0000980 break;
981 }
David Majnemerb4119f72013-12-13 01:06:04 +0000982
Guy Benyei11169dd2012-12-18 14:30:41 +0000983 case DeclarationName::ObjCZeroArgSelector:
984 case DeclarationName::ObjCOneArgSelector:
David Chisnall060a7892018-08-14 10:04:36 +0000985 case DeclarationName::ObjCMultiArgSelector: {
986 // This is reachable only when constructing an outlined SEH finally
987 // block. Nothing depends on this mangling and it's used only with
988 // functinos with internal linkage.
989 llvm::SmallString<64> Name;
990 mangleSourceName(Name.str());
991 break;
992 }
David Majnemerb4119f72013-12-13 01:06:04 +0000993
Guy Benyei11169dd2012-12-18 14:30:41 +0000994 case DeclarationName::CXXConstructorName:
Reid Klecknere4bb54a2017-01-09 17:09:59 +0000995 if (isStructorDecl(ND)) {
David Majnemerdfa6d202015-03-11 18:36:39 +0000996 if (StructorType == Ctor_CopyingClosure) {
997 Out << "?_O";
998 return;
999 }
1000 if (StructorType == Ctor_DefaultClosure) {
1001 Out << "?_F";
1002 return;
1003 }
Timur Iskhodzhanov57cbe5c2013-02-27 13:46:31 +00001004 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001005 Out << "?0";
David Majnemerdfa6d202015-03-11 18:36:39 +00001006 return;
David Majnemerb4119f72013-12-13 01:06:04 +00001007
Guy Benyei11169dd2012-12-18 14:30:41 +00001008 case DeclarationName::CXXDestructorName:
Reid Kleckner290feb92017-01-09 17:04:37 +00001009 if (isStructorDecl(ND))
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +00001010 // If the named decl is the C++ destructor we're mangling,
1011 // use the type we were given.
1012 mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
1013 else
Reid Klecknere7de47e2013-07-22 13:51:44 +00001014 // Otherwise, use the base destructor name. This is relevant if a
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +00001015 // class with a destructor is declared within a destructor.
Reid Klecknere7de47e2013-07-22 13:51:44 +00001016 mangleCXXDtorType(Dtor_Base);
Guy Benyei11169dd2012-12-18 14:30:41 +00001017 break;
David Majnemerb4119f72013-12-13 01:06:04 +00001018
Guy Benyei11169dd2012-12-18 14:30:41 +00001019 case DeclarationName::CXXConversionFunctionName:
1020 // <operator-name> ::= ?B # (cast)
1021 // The target type is encoded as the return type.
1022 Out << "?B";
1023 break;
David Majnemerb4119f72013-12-13 01:06:04 +00001024
Guy Benyei11169dd2012-12-18 14:30:41 +00001025 case DeclarationName::CXXOperatorName:
1026 mangleOperatorName(Name.getCXXOverloadedOperator(), ND->getLocation());
1027 break;
David Majnemerb4119f72013-12-13 01:06:04 +00001028
Guy Benyei11169dd2012-12-18 14:30:41 +00001029 case DeclarationName::CXXLiteralOperatorName: {
David Majnemere31a3ed2014-06-04 16:46:26 +00001030 Out << "?__K";
1031 mangleSourceName(Name.getCXXLiteralIdentifier()->getName());
Guy Benyei11169dd2012-12-18 14:30:41 +00001032 break;
1033 }
David Majnemerb4119f72013-12-13 01:06:04 +00001034
Richard Smith35845152017-02-07 01:37:30 +00001035 case DeclarationName::CXXDeductionGuideName:
1036 llvm_unreachable("Can't mangle a deduction guide name!");
1037
Guy Benyei11169dd2012-12-18 14:30:41 +00001038 case DeclarationName::CXXUsingDirective:
1039 llvm_unreachable("Can't mangle a using directive name!");
1040 }
1041}
1042
Saleem Abdulrasool58340522018-02-07 01:55:08 +00001043// <postfix> ::= <unqualified-name> [<postfix>]
1044// ::= <substitution> [<postfix>]
David Majnemer2206bf52014-03-05 08:57:59 +00001045void MicrosoftCXXNameMangler::mangleNestedName(const NamedDecl *ND) {
David Majnemerff9075f2014-08-06 03:12:47 +00001046 const DeclContext *DC = getEffectiveDeclContext(ND);
David Majnemer2206bf52014-03-05 08:57:59 +00001047 while (!DC->isTranslationUnit()) {
1048 if (isa<TagDecl>(ND) || isa<VarDecl>(ND)) {
1049 unsigned Disc;
1050 if (Context.getNextDiscriminator(ND, Disc)) {
1051 Out << '?';
1052 mangleNumber(Disc);
1053 Out << '?';
1054 }
1055 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001056
David Majnemer2206bf52014-03-05 08:57:59 +00001057 if (const BlockDecl *BD = dyn_cast<BlockDecl>(DC)) {
Saleem Abdulrasool3cc6d0b2017-06-26 23:28:42 +00001058 auto Discriminate =
1059 [](StringRef Name, const unsigned Discriminator,
1060 const unsigned ParameterDiscriminator) -> std::string {
1061 std::string Buffer;
1062 llvm::raw_string_ostream Stream(Buffer);
1063 Stream << Name;
1064 if (Discriminator)
1065 Stream << '_' << Discriminator;
1066 if (ParameterDiscriminator)
1067 Stream << '_' << ParameterDiscriminator;
1068 return Stream.str();
1069 };
David Majnemer2206bf52014-03-05 08:57:59 +00001070
Saleem Abdulrasool3cc6d0b2017-06-26 23:28:42 +00001071 unsigned Discriminator = BD->getBlockManglingNumber();
1072 if (!Discriminator)
1073 Discriminator = Context.getBlockId(BD, /*Local=*/false);
1074
1075 // Mangle the parameter position as a discriminator to deal with unnamed
1076 // parameters. Rather than mangling the unqualified parameter name,
1077 // always use the position to give a uniform mangling.
1078 unsigned ParameterDiscriminator = 0;
1079 if (const auto *MC = BD->getBlockManglingContextDecl())
1080 if (const auto *P = dyn_cast<ParmVarDecl>(MC))
1081 if (const auto *F = dyn_cast<FunctionDecl>(P->getDeclContext()))
1082 ParameterDiscriminator =
1083 F->getNumParams() - P->getFunctionScopeIndex();
1084
1085 DC = getEffectiveDeclContext(BD);
1086
1087 Out << '?';
1088 mangleSourceName(Discriminate("_block_invoke", Discriminator,
1089 ParameterDiscriminator));
1090 // If we have a block mangling context, encode that now. This allows us
1091 // to discriminate between named static data initializers in the same
1092 // scope. This is handled differently from parameters, which use
1093 // positions to discriminate between multiple instances.
1094 if (const auto *MC = BD->getBlockManglingContextDecl())
1095 if (!isa<ParmVarDecl>(MC))
1096 if (const auto *ND = dyn_cast<NamedDecl>(MC))
1097 mangleUnqualifiedName(ND);
1098 // MS ABI and Itanium manglings are in inverted scopes. In the case of a
Alexander Kornienko2a8c18d2018-04-06 15:14:32 +00001099 // RecordDecl, mangle the entire scope hierarchy at this point rather than
Saleem Abdulrasool3cc6d0b2017-06-26 23:28:42 +00001100 // just the unqualified name to get the ordering correct.
1101 if (const auto *RD = dyn_cast<RecordDecl>(DC))
1102 mangleName(RD);
1103 else
1104 Out << '@';
1105 // void __cdecl
1106 Out << "YAX";
1107 // struct __block_literal *
1108 Out << 'P';
1109 // __ptr64
1110 if (PointersAre64Bit)
1111 Out << 'E';
1112 Out << 'A';
Shoaib Meenaid84cf132018-11-14 19:16:22 +00001113 mangleArtificialTagType(TTK_Struct,
Saleem Abdulrasool3cc6d0b2017-06-26 23:28:42 +00001114 Discriminate("__block_literal", Discriminator,
1115 ParameterDiscriminator));
1116 Out << "@Z";
1117
1118 // If the effective context was a Record, we have fully mangled the
1119 // qualified name and do not need to continue.
1120 if (isa<RecordDecl>(DC))
1121 break;
1122 continue;
David Majnemer2206bf52014-03-05 08:57:59 +00001123 } else if (const ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(DC)) {
1124 mangleObjCMethodName(Method);
1125 } else if (isa<NamedDecl>(DC)) {
1126 ND = cast<NamedDecl>(DC);
1127 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) {
1128 mangle(FD, "?");
1129 break;
David Majnemer984498f2016-12-06 17:01:02 +00001130 } else {
David Majnemer2206bf52014-03-05 08:57:59 +00001131 mangleUnqualifiedName(ND);
David Majnemer984498f2016-12-06 17:01:02 +00001132 // Lambdas in default arguments conceptually belong to the function the
1133 // parameter corresponds to.
1134 if (const auto *LDADC = getLambdaDefaultArgumentDeclContext(ND)) {
1135 DC = LDADC;
1136 continue;
1137 }
1138 }
David Majnemer2206bf52014-03-05 08:57:59 +00001139 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001140 DC = DC->getParent();
Guy Benyei11169dd2012-12-18 14:30:41 +00001141 }
1142}
1143
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +00001144void MicrosoftCXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
Reid Klecknere7de47e2013-07-22 13:51:44 +00001145 // Microsoft uses the names on the case labels for these dtor variants. Clang
1146 // uses the Itanium terminology internally. Everything in this ABI delegates
1147 // towards the base dtor.
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +00001148 switch (T) {
Reid Klecknere7de47e2013-07-22 13:51:44 +00001149 // <operator-name> ::= ?1 # destructor
1150 case Dtor_Base: Out << "?1"; return;
1151 // <operator-name> ::= ?_D # vbase destructor
1152 case Dtor_Complete: Out << "?_D"; return;
1153 // <operator-name> ::= ?_G # scalar deleting destructor
1154 case Dtor_Deleting: Out << "?_G"; return;
1155 // <operator-name> ::= ?_E # vector deleting destructor
1156 // FIXME: Add a vector deleting dtor type. It goes in the vtable, so we need
1157 // it.
Rafael Espindola1e4df922014-09-16 15:18:21 +00001158 case Dtor_Comdat:
1159 llvm_unreachable("not expecting a COMDAT");
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +00001160 }
1161 llvm_unreachable("Unsupported dtor type?");
1162}
1163
Guy Benyei11169dd2012-12-18 14:30:41 +00001164void MicrosoftCXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO,
1165 SourceLocation Loc) {
1166 switch (OO) {
1167 // ?0 # constructor
1168 // ?1 # destructor
1169 // <operator-name> ::= ?2 # new
1170 case OO_New: Out << "?2"; break;
1171 // <operator-name> ::= ?3 # delete
1172 case OO_Delete: Out << "?3"; break;
1173 // <operator-name> ::= ?4 # =
1174 case OO_Equal: Out << "?4"; break;
1175 // <operator-name> ::= ?5 # >>
1176 case OO_GreaterGreater: Out << "?5"; break;
1177 // <operator-name> ::= ?6 # <<
1178 case OO_LessLess: Out << "?6"; break;
1179 // <operator-name> ::= ?7 # !
1180 case OO_Exclaim: Out << "?7"; break;
1181 // <operator-name> ::= ?8 # ==
1182 case OO_EqualEqual: Out << "?8"; break;
1183 // <operator-name> ::= ?9 # !=
1184 case OO_ExclaimEqual: Out << "?9"; break;
1185 // <operator-name> ::= ?A # []
1186 case OO_Subscript: Out << "?A"; break;
1187 // ?B # conversion
1188 // <operator-name> ::= ?C # ->
1189 case OO_Arrow: Out << "?C"; break;
1190 // <operator-name> ::= ?D # *
1191 case OO_Star: Out << "?D"; break;
1192 // <operator-name> ::= ?E # ++
1193 case OO_PlusPlus: Out << "?E"; break;
1194 // <operator-name> ::= ?F # --
1195 case OO_MinusMinus: Out << "?F"; break;
1196 // <operator-name> ::= ?G # -
1197 case OO_Minus: Out << "?G"; break;
1198 // <operator-name> ::= ?H # +
1199 case OO_Plus: Out << "?H"; break;
1200 // <operator-name> ::= ?I # &
1201 case OO_Amp: Out << "?I"; break;
1202 // <operator-name> ::= ?J # ->*
1203 case OO_ArrowStar: Out << "?J"; break;
1204 // <operator-name> ::= ?K # /
1205 case OO_Slash: Out << "?K"; break;
1206 // <operator-name> ::= ?L # %
1207 case OO_Percent: Out << "?L"; break;
1208 // <operator-name> ::= ?M # <
1209 case OO_Less: Out << "?M"; break;
1210 // <operator-name> ::= ?N # <=
1211 case OO_LessEqual: Out << "?N"; break;
1212 // <operator-name> ::= ?O # >
1213 case OO_Greater: Out << "?O"; break;
1214 // <operator-name> ::= ?P # >=
1215 case OO_GreaterEqual: Out << "?P"; break;
1216 // <operator-name> ::= ?Q # ,
1217 case OO_Comma: Out << "?Q"; break;
1218 // <operator-name> ::= ?R # ()
1219 case OO_Call: Out << "?R"; break;
1220 // <operator-name> ::= ?S # ~
1221 case OO_Tilde: Out << "?S"; break;
1222 // <operator-name> ::= ?T # ^
1223 case OO_Caret: Out << "?T"; break;
1224 // <operator-name> ::= ?U # |
1225 case OO_Pipe: Out << "?U"; break;
1226 // <operator-name> ::= ?V # &&
1227 case OO_AmpAmp: Out << "?V"; break;
1228 // <operator-name> ::= ?W # ||
1229 case OO_PipePipe: Out << "?W"; break;
1230 // <operator-name> ::= ?X # *=
1231 case OO_StarEqual: Out << "?X"; break;
1232 // <operator-name> ::= ?Y # +=
1233 case OO_PlusEqual: Out << "?Y"; break;
1234 // <operator-name> ::= ?Z # -=
1235 case OO_MinusEqual: Out << "?Z"; break;
1236 // <operator-name> ::= ?_0 # /=
1237 case OO_SlashEqual: Out << "?_0"; break;
1238 // <operator-name> ::= ?_1 # %=
1239 case OO_PercentEqual: Out << "?_1"; break;
1240 // <operator-name> ::= ?_2 # >>=
1241 case OO_GreaterGreaterEqual: Out << "?_2"; break;
1242 // <operator-name> ::= ?_3 # <<=
1243 case OO_LessLessEqual: Out << "?_3"; break;
1244 // <operator-name> ::= ?_4 # &=
1245 case OO_AmpEqual: Out << "?_4"; break;
1246 // <operator-name> ::= ?_5 # |=
1247 case OO_PipeEqual: Out << "?_5"; break;
1248 // <operator-name> ::= ?_6 # ^=
1249 case OO_CaretEqual: Out << "?_6"; break;
1250 // ?_7 # vftable
1251 // ?_8 # vbtable
1252 // ?_9 # vcall
1253 // ?_A # typeof
1254 // ?_B # local static guard
1255 // ?_C # string
1256 // ?_D # vbase destructor
1257 // ?_E # vector deleting destructor
1258 // ?_F # default constructor closure
1259 // ?_G # scalar deleting destructor
1260 // ?_H # vector constructor iterator
1261 // ?_I # vector destructor iterator
1262 // ?_J # vector vbase constructor iterator
1263 // ?_K # virtual displacement map
1264 // ?_L # eh vector constructor iterator
1265 // ?_M # eh vector destructor iterator
1266 // ?_N # eh vector vbase constructor iterator
1267 // ?_O # copy constructor closure
1268 // ?_P<name> # udt returning <name>
1269 // ?_Q # <unknown>
1270 // ?_R0 # RTTI Type Descriptor
1271 // ?_R1 # RTTI Base Class Descriptor at (a,b,c,d)
1272 // ?_R2 # RTTI Base Class Array
1273 // ?_R3 # RTTI Class Hierarchy Descriptor
1274 // ?_R4 # RTTI Complete Object Locator
1275 // ?_S # local vftable
1276 // ?_T # local vftable constructor closure
1277 // <operator-name> ::= ?_U # new[]
1278 case OO_Array_New: Out << "?_U"; break;
1279 // <operator-name> ::= ?_V # delete[]
1280 case OO_Array_Delete: Out << "?_V"; break;
Gor Nishanove2f51be2016-09-28 22:37:17 +00001281 // <operator-name> ::= ?__L # co_await
1282 case OO_Coawait: Out << "?__L"; break;
Nico Weber27df4092019-04-23 16:37:42 +00001283 // <operator-name> ::= ?__M # <=>
1284 case OO_Spaceship: Out << "?__M"; break;
Richard Smithd30b23d2017-12-01 02:13:10 +00001285
Guy Benyei11169dd2012-12-18 14:30:41 +00001286 case OO_Conditional: {
1287 DiagnosticsEngine &Diags = Context.getDiags();
1288 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
1289 "cannot mangle this conditional operator yet");
1290 Diags.Report(Loc, DiagID);
1291 break;
1292 }
David Majnemerb4119f72013-12-13 01:06:04 +00001293
Guy Benyei11169dd2012-12-18 14:30:41 +00001294 case OO_None:
1295 case NUM_OVERLOADED_OPERATORS:
1296 llvm_unreachable("Not an overloaded operator");
1297 }
1298}
1299
David Majnemer956bc112013-11-25 17:50:19 +00001300void MicrosoftCXXNameMangler::mangleSourceName(StringRef Name) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001301 // <source name> ::= <identifier> @
Fangrui Song75e74e02019-03-31 08:48:19 +00001302 BackRefVec::iterator Found = llvm::find(NameBackReferences, Name);
David Majnemerc986fcc2014-06-08 04:51:13 +00001303 if (Found == NameBackReferences.end()) {
David Majnemer0d0560c2014-09-25 19:43:56 +00001304 if (NameBackReferences.size() < 10)
1305 NameBackReferences.push_back(Name);
David Majnemer956bc112013-11-25 17:50:19 +00001306 Out << Name << '@';
Guy Benyei11169dd2012-12-18 14:30:41 +00001307 } else {
David Majnemer0d0560c2014-09-25 19:43:56 +00001308 Out << (Found - NameBackReferences.begin());
Guy Benyei11169dd2012-12-18 14:30:41 +00001309 }
1310}
1311
1312void MicrosoftCXXNameMangler::mangleObjCMethodName(const ObjCMethodDecl *MD) {
1313 Context.mangleObjCMethodName(MD, Out);
1314}
1315
Guy Benyei11169dd2012-12-18 14:30:41 +00001316void MicrosoftCXXNameMangler::mangleTemplateInstantiationName(
David Majnemer02e9af42014-04-23 05:16:51 +00001317 const TemplateDecl *TD, const TemplateArgumentList &TemplateArgs) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001318 // <template-name> ::= <unscoped-template-name> <template-args>
1319 // ::= <substitution>
1320 // Always start with the unqualified name.
1321
1322 // Templates have their own context for back references.
Nico Weber73f05842019-06-01 10:12:07 +00001323 ArgBackRefMap OuterFunArgsContext;
1324 ArgBackRefMap OuterTemplateArgsContext;
David Majnemer0d0560c2014-09-25 19:43:56 +00001325 BackRefVec OuterTemplateContext;
David Majnemer5f177622015-12-30 05:13:03 +00001326 PassObjectSizeArgsSet OuterPassObjectSizeArgs;
Guy Benyei11169dd2012-12-18 14:30:41 +00001327 NameBackReferences.swap(OuterTemplateContext);
Nico Weber73f05842019-06-01 10:12:07 +00001328 FunArgBackReferences.swap(OuterFunArgsContext);
1329 TemplateArgBackReferences.swap(OuterTemplateArgsContext);
David Majnemer5f177622015-12-30 05:13:03 +00001330 PassObjectSizeArgs.swap(OuterPassObjectSizeArgs);
Guy Benyei11169dd2012-12-18 14:30:41 +00001331
1332 mangleUnscopedTemplateName(TD);
Reid Kleckner52518862013-03-20 01:40:23 +00001333 mangleTemplateArgs(TD, TemplateArgs);
Guy Benyei11169dd2012-12-18 14:30:41 +00001334
1335 // Restore the previous back reference contexts.
1336 NameBackReferences.swap(OuterTemplateContext);
Nico Weber73f05842019-06-01 10:12:07 +00001337 FunArgBackReferences.swap(OuterFunArgsContext);
1338 TemplateArgBackReferences.swap(OuterTemplateArgsContext);
David Majnemer5f177622015-12-30 05:13:03 +00001339 PassObjectSizeArgs.swap(OuterPassObjectSizeArgs);
Guy Benyei11169dd2012-12-18 14:30:41 +00001340}
1341
1342void
1343MicrosoftCXXNameMangler::mangleUnscopedTemplateName(const TemplateDecl *TD) {
1344 // <unscoped-template-name> ::= ?$ <unqualified-name>
1345 Out << "?$";
1346 mangleUnqualifiedName(TD);
1347}
1348
David Majnemer02e9af42014-04-23 05:16:51 +00001349void MicrosoftCXXNameMangler::mangleIntegerLiteral(const llvm::APSInt &Value,
1350 bool IsBoolean) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001351 // <integer-literal> ::= $0 <number>
1352 Out << "$0";
1353 // Make sure booleans are encoded as 0/1.
1354 if (IsBoolean && Value.getBoolValue())
1355 mangleNumber(1);
Will Wilsone74c2812014-12-11 05:47:10 +00001356 else if (Value.isSigned())
David Majnemer2a816452013-12-09 10:44:32 +00001357 mangleNumber(Value.getSExtValue());
Will Wilsone74c2812014-12-11 05:47:10 +00001358 else
1359 mangleNumber(Value.getZExtValue());
Guy Benyei11169dd2012-12-18 14:30:41 +00001360}
1361
David Majnemer02e9af42014-04-23 05:16:51 +00001362void MicrosoftCXXNameMangler::mangleExpression(const Expr *E) {
Guy Benyei11169dd2012-12-18 14:30:41 +00001363 // See if this is a constant expression.
1364 llvm::APSInt Value;
1365 if (E->isIntegerConstantExpr(Value, Context.getASTContext())) {
1366 mangleIntegerLiteral(Value, E->getType()->isBooleanType());
1367 return;
1368 }
1369
Reid Klecknerb4848e72014-06-10 20:06:25 +00001370 // Look through no-op casts like template parameter substitutions.
1371 E = E->IgnoreParenNoopCasts(Context.getASTContext());
1372
Craig Topper36250ad2014-05-12 05:36:57 +00001373 const CXXUuidofExpr *UE = nullptr;
David Majnemer8eaab6f2013-08-13 06:32:20 +00001374 if (const UnaryOperator *UO = dyn_cast<UnaryOperator>(E)) {
1375 if (UO->getOpcode() == UO_AddrOf)
1376 UE = dyn_cast<CXXUuidofExpr>(UO->getSubExpr());
1377 } else
1378 UE = dyn_cast<CXXUuidofExpr>(E);
1379
1380 if (UE) {
David Majnemere9cab2f2013-08-13 09:17:25 +00001381 // If we had to peek through an address-of operator, treat this like we are
David Majnemer8eaab6f2013-08-13 06:32:20 +00001382 // dealing with a pointer type. Otherwise, treat it like a const reference.
1383 //
1384 // N.B. This matches up with the handling of TemplateArgument::Declaration
1385 // in mangleTemplateArg
1386 if (UE == E)
1387 Out << "$E?";
1388 else
1389 Out << "$1?";
David Majnemer3add70b2015-12-29 11:45:53 +00001390
1391 // This CXXUuidofExpr is mangled as-if it were actually a VarDecl from
1392 // const __s_GUID _GUID_{lower case UUID with underscores}
David Majnemer2041b462016-03-28 03:19:50 +00001393 StringRef Uuid = UE->getUuidStr();
David Majnemer3add70b2015-12-29 11:45:53 +00001394 std::string Name = "_GUID_" + Uuid.lower();
1395 std::replace(Name.begin(), Name.end(), '-', '_');
1396
1397 mangleSourceName(Name);
1398 // Terminate the whole name with an '@'.
1399 Out << '@';
1400 // It's a global variable.
1401 Out << '3';
1402 // It's a struct called __s_GUID.
Shoaib Meenaid84cf132018-11-14 19:16:22 +00001403 mangleArtificialTagType(TTK_Struct, "__s_GUID");
David Majnemer3add70b2015-12-29 11:45:53 +00001404 // It's const.
1405 Out << 'B';
David Majnemer8eaab6f2013-08-13 06:32:20 +00001406 return;
1407 }
1408
Guy Benyei11169dd2012-12-18 14:30:41 +00001409 // As bad as this diagnostic is, it's better than crashing.
1410 DiagnosticsEngine &Diags = Context.getDiags();
David Majnemer02e9af42014-04-23 05:16:51 +00001411 unsigned DiagID = Diags.getCustomDiagID(
1412 DiagnosticsEngine::Error, "cannot yet mangle expression type %0");
1413 Diags.Report(E->getExprLoc(), DiagID) << E->getStmtClassName()
1414 << E->getSourceRange();
Guy Benyei11169dd2012-12-18 14:30:41 +00001415}
1416
David Majnemer8265dec2014-03-30 16:30:54 +00001417void MicrosoftCXXNameMangler::mangleTemplateArgs(
1418 const TemplateDecl *TD, const TemplateArgumentList &TemplateArgs) {
Reid Kleckner11c6f612014-06-26 22:42:18 +00001419 // <template-args> ::= <template-arg>+
David Majnemer57fbc0c2014-08-05 22:43:45 +00001420 const TemplateParameterList *TPL = TD->getTemplateParameters();
1421 assert(TPL->size() == TemplateArgs.size() &&
1422 "size mismatch between args and parms!");
1423
Nico Webere2dd1572018-09-08 20:58:39 +00001424 for (size_t i = 0; i < TemplateArgs.size(); ++i) {
1425 const TemplateArgument &TA = TemplateArgs[i];
1426
1427 // Separate consecutive packs by $$Z.
1428 if (i > 0 && TA.getKind() == TemplateArgument::Pack &&
1429 TemplateArgs[i - 1].getKind() == TemplateArgument::Pack)
1430 Out << "$$Z";
1431
1432 mangleTemplateArg(TD, TA, TPL->getParam(i));
1433 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001434}
1435
Reid Klecknerf0ae35b2013-07-02 18:10:07 +00001436void MicrosoftCXXNameMangler::mangleTemplateArg(const TemplateDecl *TD,
David Majnemer57fbc0c2014-08-05 22:43:45 +00001437 const TemplateArgument &TA,
1438 const NamedDecl *Parm) {
Reid Kleckner96f8f932014-02-05 17:27:08 +00001439 // <template-arg> ::= <type>
1440 // ::= <integer-literal>
1441 // ::= <member-data-pointer>
1442 // ::= <member-function-pointer>
1443 // ::= $E? <name> <type-encoding>
1444 // ::= $1? <name> <type-encoding>
1445 // ::= $0A@
1446 // ::= <template-args>
1447
Reid Klecknerf0ae35b2013-07-02 18:10:07 +00001448 switch (TA.getKind()) {
1449 case TemplateArgument::Null:
1450 llvm_unreachable("Can't mangle null template arguments!");
David Majnemer08177c52013-08-27 08:21:25 +00001451 case TemplateArgument::TemplateExpansion:
1452 llvm_unreachable("Can't mangle template expansion arguments!");
Reid Klecknerf0ae35b2013-07-02 18:10:07 +00001453 case TemplateArgument::Type: {
1454 QualType T = TA.getAsType();
1455 mangleType(T, SourceRange(), QMM_Escape);
1456 break;
1457 }
David Majnemere8fdc062013-08-13 01:25:35 +00001458 case TemplateArgument::Declaration: {
George Burgess IV00f70bd2018-03-01 05:43:23 +00001459 const NamedDecl *ND = TA.getAsDecl();
David Majnemer1e378e42014-02-06 12:46:52 +00001460 if (isa<FieldDecl>(ND) || isa<IndirectFieldDecl>(ND)) {
Nico Weber2b5f3fd2018-08-01 11:56:20 +00001461 mangleMemberDataPointer(cast<CXXRecordDecl>(ND->getDeclContext())
1462 ->getMostRecentNonInjectedDecl(),
1463 cast<ValueDecl>(ND));
Reid Kleckner09b47d12014-02-05 18:59:38 +00001464 } else if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)) {
Nick Lewycky1f529662014-02-05 23:53:29 +00001465 const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD);
David Majnemerab4b4a12015-05-18 00:05:25 +00001466 if (MD && MD->isInstance()) {
Nico Weber2b5f3fd2018-08-01 11:56:20 +00001467 mangleMemberFunctionPointer(
1468 MD->getParent()->getMostRecentNonInjectedDecl(), MD);
David Majnemerab4b4a12015-05-18 00:05:25 +00001469 } else {
1470 Out << "$1?";
1471 mangleName(FD);
1472 mangleFunctionEncoding(FD, /*ShouldMangle=*/true);
1473 }
Reid Kleckner09b47d12014-02-05 18:59:38 +00001474 } else {
David Blaikie952a9b12014-10-17 18:00:12 +00001475 mangle(ND, TA.getParamTypeForDecl()->isReferenceType() ? "$E?" : "$1?");
Reid Kleckner09b47d12014-02-05 18:59:38 +00001476 }
Reid Klecknerf0ae35b2013-07-02 18:10:07 +00001477 break;
David Majnemere8fdc062013-08-13 01:25:35 +00001478 }
Reid Klecknerf0ae35b2013-07-02 18:10:07 +00001479 case TemplateArgument::Integral:
1480 mangleIntegerLiteral(TA.getAsIntegral(),
1481 TA.getIntegralType()->isBooleanType());
1482 break;
Reid Kleckner96f8f932014-02-05 17:27:08 +00001483 case TemplateArgument::NullPtr: {
1484 QualType T = TA.getNullPtrType();
1485 if (const MemberPointerType *MPT = T->getAs<MemberPointerType>()) {
David Majnemer763584d2014-02-06 10:59:19 +00001486 const CXXRecordDecl *RD = MPT->getMostRecentCXXRecordDecl();
David Majnemerf3027172015-12-01 19:13:51 +00001487 if (MPT->isMemberFunctionPointerType() &&
1488 !isa<FunctionTemplateDecl>(TD)) {
Craig Topper36250ad2014-05-12 05:36:57 +00001489 mangleMemberFunctionPointer(RD, nullptr);
David Majnemer19187872014-06-11 07:08:37 +00001490 return;
1491 }
1492 if (MPT->isMemberDataPointer()) {
David Majnemerf3027172015-12-01 19:13:51 +00001493 if (!isa<FunctionTemplateDecl>(TD)) {
David Majnemer10236682015-07-02 09:43:11 +00001494 mangleMemberDataPointer(RD, nullptr);
1495 return;
1496 }
1497 // nullptr data pointers are always represented with a single field
1498 // which is initialized with either 0 or -1. Why -1? Well, we need to
1499 // distinguish the case where the data member is at offset zero in the
1500 // record.
1501 // However, we are free to use 0 *if* we would use multiple fields for
1502 // non-nullptr member pointers.
1503 if (!RD->nullFieldOffsetIsZero()) {
1504 mangleIntegerLiteral(llvm::APSInt::get(-1), /*IsBoolean=*/false);
1505 return;
1506 }
David Majnemer19187872014-06-11 07:08:37 +00001507 }
Reid Kleckner96f8f932014-02-05 17:27:08 +00001508 }
David Majnemer10236682015-07-02 09:43:11 +00001509 mangleIntegerLiteral(llvm::APSInt::getUnsigned(0), /*IsBoolean=*/false);
David Majnemerae465ef2013-08-05 21:33:59 +00001510 break;
Reid Kleckner96f8f932014-02-05 17:27:08 +00001511 }
Reid Klecknerf0ae35b2013-07-02 18:10:07 +00001512 case TemplateArgument::Expression:
1513 mangleExpression(TA.getAsExpr());
1514 break;
David Majnemer06fa05a2014-06-04 16:46:32 +00001515 case TemplateArgument::Pack: {
Craig Topper00bbdcf2014-06-28 23:22:23 +00001516 ArrayRef<TemplateArgument> TemplateArgs = TA.getPackAsArray();
David Majnemer06fa05a2014-06-04 16:46:32 +00001517 if (TemplateArgs.empty()) {
David Majnemer57fbc0c2014-08-05 22:43:45 +00001518 if (isa<TemplateTypeParmDecl>(Parm) ||
1519 isa<TemplateTemplateParmDecl>(Parm))
David Majnemer4deaac72015-03-18 03:56:27 +00001520 // MSVC 2015 changed the mangling for empty expanded template packs,
1521 // use the old mangling for link compatibility for old versions.
David Majnemerb710a932015-05-11 03:57:49 +00001522 Out << (Context.getASTContext().getLangOpts().isCompatibleWithMSVC(
1523 LangOptions::MSVC2015)
David Majnemer4deaac72015-03-18 03:56:27 +00001524 ? "$$V"
1525 : "$$$V");
David Majnemer57fbc0c2014-08-05 22:43:45 +00001526 else if (isa<NonTypeTemplateParmDecl>(Parm))
1527 Out << "$S";
1528 else
1529 llvm_unreachable("unexpected template parameter decl!");
David Majnemer06fa05a2014-06-04 16:46:32 +00001530 } else {
1531 for (const TemplateArgument &PA : TemplateArgs)
David Majnemer57fbc0c2014-08-05 22:43:45 +00001532 mangleTemplateArg(TD, PA, Parm);
David Majnemer06fa05a2014-06-04 16:46:32 +00001533 }
Reid Klecknerf0ae35b2013-07-02 18:10:07 +00001534 break;
David Majnemer06fa05a2014-06-04 16:46:32 +00001535 }
David Majnemer8d5f9ab2014-07-30 08:20:03 +00001536 case TemplateArgument::Template: {
1537 const NamedDecl *ND =
1538 TA.getAsTemplate().getAsTemplateDecl()->getTemplatedDecl();
1539 if (const auto *TD = dyn_cast<TagDecl>(ND)) {
1540 mangleType(TD);
1541 } else if (isa<TypeAliasDecl>(ND)) {
David Majnemer9a3bc592014-08-19 07:29:03 +00001542 Out << "$$Y";
1543 mangleName(ND);
David Majnemer8d5f9ab2014-07-30 08:20:03 +00001544 } else {
1545 llvm_unreachable("unexpected template template NamedDecl!");
1546 }
David Majnemer0db0ca42013-08-05 22:26:46 +00001547 break;
Reid Klecknerf0ae35b2013-07-02 18:10:07 +00001548 }
David Majnemer8d5f9ab2014-07-30 08:20:03 +00001549 }
Reid Klecknerf0ae35b2013-07-02 18:10:07 +00001550}
1551
Saleem Abdulrasoolbbce34a2018-02-09 03:23:54 +00001552void MicrosoftCXXNameMangler::mangleObjCProtocol(const ObjCProtocolDecl *PD) {
1553 llvm::SmallString<64> TemplateMangling;
1554 llvm::raw_svector_ostream Stream(TemplateMangling);
1555 MicrosoftCXXNameMangler Extra(Context, Stream);
1556
1557 Stream << "?$";
1558 Extra.mangleSourceName("Protocol");
Shoaib Meenaid84cf132018-11-14 19:16:22 +00001559 Extra.mangleArtificialTagType(TTK_Struct, PD->getName());
Saleem Abdulrasoolbbce34a2018-02-09 03:23:54 +00001560
Shoaib Meenaid84cf132018-11-14 19:16:22 +00001561 mangleArtificialTagType(TTK_Struct, TemplateMangling, {"__ObjC"});
Saleem Abdulrasoolbbce34a2018-02-09 03:23:54 +00001562}
1563
1564void MicrosoftCXXNameMangler::mangleObjCLifetime(const QualType Type,
1565 Qualifiers Quals,
1566 SourceRange Range) {
1567 llvm::SmallString<64> TemplateMangling;
1568 llvm::raw_svector_ostream Stream(TemplateMangling);
1569 MicrosoftCXXNameMangler Extra(Context, Stream);
1570
1571 Stream << "?$";
1572 switch (Quals.getObjCLifetime()) {
1573 case Qualifiers::OCL_None:
1574 case Qualifiers::OCL_ExplicitNone:
1575 break;
1576 case Qualifiers::OCL_Autoreleasing:
1577 Extra.mangleSourceName("Autoreleasing");
1578 break;
1579 case Qualifiers::OCL_Strong:
1580 Extra.mangleSourceName("Strong");
1581 break;
1582 case Qualifiers::OCL_Weak:
1583 Extra.mangleSourceName("Weak");
1584 break;
1585 }
1586 Extra.manglePointerCVQualifiers(Quals);
1587 Extra.manglePointerExtQualifiers(Quals, Type);
1588 Extra.mangleType(Type, Range);
1589
Shoaib Meenaid84cf132018-11-14 19:16:22 +00001590 mangleArtificialTagType(TTK_Struct, TemplateMangling, {"__ObjC"});
Saleem Abdulrasoolbbce34a2018-02-09 03:23:54 +00001591}
1592
Saleem Abdulrasoolf3a981e2018-09-28 16:47:53 +00001593void MicrosoftCXXNameMangler::mangleObjCKindOfType(const ObjCObjectType *T,
1594 Qualifiers Quals,
1595 SourceRange Range) {
1596 llvm::SmallString<64> TemplateMangling;
1597 llvm::raw_svector_ostream Stream(TemplateMangling);
1598 MicrosoftCXXNameMangler Extra(Context, Stream);
1599
1600 Stream << "?$";
1601 Extra.mangleSourceName("KindOf");
1602 Extra.mangleType(QualType(T, 0)
1603 .stripObjCKindOfType(getASTContext())
1604 ->getAs<ObjCObjectType>(),
1605 Quals, Range);
1606
Shoaib Meenaid84cf132018-11-14 19:16:22 +00001607 mangleArtificialTagType(TTK_Struct, TemplateMangling, {"__ObjC"});
Saleem Abdulrasoolf3a981e2018-09-28 16:47:53 +00001608}
1609
Guy Benyei11169dd2012-12-18 14:30:41 +00001610void MicrosoftCXXNameMangler::mangleQualifiers(Qualifiers Quals,
1611 bool IsMember) {
1612 // <cvr-qualifiers> ::= [E] [F] [I] <base-cvr-qualifiers>
1613 // 'E' means __ptr64 (32-bit only); 'F' means __unaligned (32/64-bit only);
1614 // 'I' means __restrict (32/64-bit).
1615 // Note that the MSVC __restrict keyword isn't the same as the C99 restrict
1616 // keyword!
1617 // <base-cvr-qualifiers> ::= A # near
1618 // ::= B # near const
1619 // ::= C # near volatile
1620 // ::= D # near const volatile
1621 // ::= E # far (16-bit)
1622 // ::= F # far const (16-bit)
1623 // ::= G # far volatile (16-bit)
1624 // ::= H # far const volatile (16-bit)
1625 // ::= I # huge (16-bit)
1626 // ::= J # huge const (16-bit)
1627 // ::= K # huge volatile (16-bit)
1628 // ::= L # huge const volatile (16-bit)
1629 // ::= M <basis> # based
1630 // ::= N <basis> # based const
1631 // ::= O <basis> # based volatile
1632 // ::= P <basis> # based const volatile
1633 // ::= Q # near member
1634 // ::= R # near const member
1635 // ::= S # near volatile member
1636 // ::= T # near const volatile member
1637 // ::= U # far member (16-bit)
1638 // ::= V # far const member (16-bit)
1639 // ::= W # far volatile member (16-bit)
1640 // ::= X # far const volatile member (16-bit)
1641 // ::= Y # huge member (16-bit)
1642 // ::= Z # huge const member (16-bit)
1643 // ::= 0 # huge volatile member (16-bit)
1644 // ::= 1 # huge const volatile member (16-bit)
1645 // ::= 2 <basis> # based member
1646 // ::= 3 <basis> # based const member
1647 // ::= 4 <basis> # based volatile member
1648 // ::= 5 <basis> # based const volatile member
1649 // ::= 6 # near function (pointers only)
1650 // ::= 7 # far function (pointers only)
1651 // ::= 8 # near method (pointers only)
1652 // ::= 9 # far method (pointers only)
1653 // ::= _A <basis> # based function (pointers only)
1654 // ::= _B <basis> # based function (far?) (pointers only)
1655 // ::= _C <basis> # based method (pointers only)
1656 // ::= _D <basis> # based method (far?) (pointers only)
1657 // ::= _E # block (Clang)
1658 // <basis> ::= 0 # __based(void)
1659 // ::= 1 # __based(segment)?
1660 // ::= 2 <name> # __based(name)
1661 // ::= 3 # ?
1662 // ::= 4 # ?
1663 // ::= 5 # not really based
1664 bool HasConst = Quals.hasConst(),
1665 HasVolatile = Quals.hasVolatile();
David Majnemer89594f32013-08-05 22:43:06 +00001666
Guy Benyei11169dd2012-12-18 14:30:41 +00001667 if (!IsMember) {
1668 if (HasConst && HasVolatile) {
1669 Out << 'D';
1670 } else if (HasVolatile) {
1671 Out << 'C';
1672 } else if (HasConst) {
1673 Out << 'B';
1674 } else {
1675 Out << 'A';
1676 }
1677 } else {
1678 if (HasConst && HasVolatile) {
1679 Out << 'T';
1680 } else if (HasVolatile) {
1681 Out << 'S';
1682 } else if (HasConst) {
1683 Out << 'R';
1684 } else {
1685 Out << 'Q';
1686 }
1687 }
1688
1689 // FIXME: For now, just drop all extension qualifiers on the floor.
1690}
1691
David Majnemer8eec58f2014-02-18 14:20:10 +00001692void
David Majnemere3785bb2014-04-23 05:16:56 +00001693MicrosoftCXXNameMangler::mangleRefQualifier(RefQualifierKind RefQualifier) {
1694 // <ref-qualifier> ::= G # lvalue reference
1695 // ::= H # rvalue-reference
1696 switch (RefQualifier) {
1697 case RQ_None:
1698 break;
1699
1700 case RQ_LValue:
1701 Out << 'G';
1702 break;
1703
1704 case RQ_RValue:
1705 Out << 'H';
1706 break;
1707 }
1708}
1709
David Majnemer7a3ce0c2015-05-26 01:30:45 +00001710void MicrosoftCXXNameMangler::manglePointerExtQualifiers(Qualifiers Quals,
1711 QualType PointeeType) {
David Majnemer7a3ce0c2015-05-26 01:30:45 +00001712 if (PointersAre64Bit &&
1713 (PointeeType.isNull() || !PointeeType->isFunctionType()))
David Majnemer8eec58f2014-02-18 14:20:10 +00001714 Out << 'E';
1715
Saleem Abdulrasool26d9bdb2018-01-25 19:54:31 +00001716 if (Quals.hasRestrict())
David Majnemer8eec58f2014-02-18 14:20:10 +00001717 Out << 'I';
Andrey Bokhanko45d41322016-05-11 18:38:21 +00001718
Andrey Bokhanko67a41862016-05-26 10:06:01 +00001719 if (Quals.hasUnaligned() ||
1720 (!PointeeType.isNull() && PointeeType.getLocalQualifiers().hasUnaligned()))
Andrey Bokhanko45d41322016-05-11 18:38:21 +00001721 Out << 'F';
David Majnemer8eec58f2014-02-18 14:20:10 +00001722}
1723
1724void MicrosoftCXXNameMangler::manglePointerCVQualifiers(Qualifiers Quals) {
1725 // <pointer-cv-qualifiers> ::= P # no qualifiers
1726 // ::= Q # const
1727 // ::= R # volatile
1728 // ::= S # const volatile
Guy Benyei11169dd2012-12-18 14:30:41 +00001729 bool HasConst = Quals.hasConst(),
David Majnemer8eec58f2014-02-18 14:20:10 +00001730 HasVolatile = Quals.hasVolatile();
David Majnemer0b6bf8a2014-02-18 12:58:35 +00001731
Guy Benyei11169dd2012-12-18 14:30:41 +00001732 if (HasConst && HasVolatile) {
1733 Out << 'S';
1734 } else if (HasVolatile) {
1735 Out << 'R';
1736 } else if (HasConst) {
1737 Out << 'Q';
1738 } else {
1739 Out << 'P';
1740 }
1741}
1742
Nico Weber73f05842019-06-01 10:12:07 +00001743void MicrosoftCXXNameMangler::mangleFunctionArgumentType(QualType T,
1744 SourceRange Range) {
Reid Kleckner18da98e2013-06-24 19:21:52 +00001745 // MSVC will backreference two canonically equivalent types that have slightly
1746 // different manglings when mangled alone.
David Majnemer5a1b2042013-09-11 04:44:30 +00001747
1748 // Decayed types do not match up with non-decayed versions of the same type.
1749 //
1750 // e.g.
1751 // void (*x)(void) will not form a backreference with void x(void)
1752 void *TypePtr;
David Majnemer96183f62015-04-23 05:21:20 +00001753 if (const auto *DT = T->getAs<DecayedType>()) {
1754 QualType OriginalType = DT->getOriginalType();
David Majnemerbdc1fed2015-04-27 03:07:47 +00001755 // All decayed ArrayTypes should be treated identically; as-if they were
1756 // a decayed IncompleteArrayType.
1757 if (const auto *AT = getASTContext().getAsArrayType(OriginalType))
David Majnemer96183f62015-04-23 05:21:20 +00001758 OriginalType = getASTContext().getIncompleteArrayType(
David Majnemerbdc1fed2015-04-27 03:07:47 +00001759 AT->getElementType(), AT->getSizeModifier(),
1760 AT->getIndexTypeCVRQualifiers());
David Majnemer96183f62015-04-23 05:21:20 +00001761
1762 TypePtr = OriginalType.getCanonicalType().getAsOpaquePtr();
David Majnemer5a1b2042013-09-11 04:44:30 +00001763 // If the original parameter was textually written as an array,
1764 // instead treat the decayed parameter like it's const.
1765 //
1766 // e.g.
1767 // int [] -> int * const
David Majnemer96183f62015-04-23 05:21:20 +00001768 if (OriginalType->isArrayType())
David Majnemer5a1b2042013-09-11 04:44:30 +00001769 T = T.withConst();
David Majnemer96183f62015-04-23 05:21:20 +00001770 } else {
David Majnemer5a1b2042013-09-11 04:44:30 +00001771 TypePtr = T.getCanonicalType().getAsOpaquePtr();
David Majnemer96183f62015-04-23 05:21:20 +00001772 }
David Majnemer5a1b2042013-09-11 04:44:30 +00001773
Nico Weber73f05842019-06-01 10:12:07 +00001774 ArgBackRefMap::iterator Found = FunArgBackReferences.find(TypePtr);
Guy Benyei11169dd2012-12-18 14:30:41 +00001775
Nico Weber73f05842019-06-01 10:12:07 +00001776 if (Found == FunArgBackReferences.end()) {
Yaron Keren2155f802015-05-02 15:31:08 +00001777 size_t OutSizeBefore = Out.tell();
Guy Benyei11169dd2012-12-18 14:30:41 +00001778
David Majnemer5a1b2042013-09-11 04:44:30 +00001779 mangleType(T, Range, QMM_Drop);
Guy Benyei11169dd2012-12-18 14:30:41 +00001780
1781 // See if it's worth creating a back reference.
1782 // Only types longer than 1 character are considered
1783 // and only 10 back references slots are available:
Yaron Keren2155f802015-05-02 15:31:08 +00001784 bool LongerThanOneChar = (Out.tell() - OutSizeBefore > 1);
Nico Weber73f05842019-06-01 10:12:07 +00001785 if (LongerThanOneChar && FunArgBackReferences.size() < 10) {
1786 size_t Size = FunArgBackReferences.size();
1787 FunArgBackReferences[TypePtr] = Size;
Guy Benyei11169dd2012-12-18 14:30:41 +00001788 }
1789 } else {
1790 Out << Found->second;
1791 }
1792}
1793
David Majnemer5f177622015-12-30 05:13:03 +00001794void MicrosoftCXXNameMangler::manglePassObjectSizeArg(
1795 const PassObjectSizeAttr *POSA) {
1796 int Type = POSA->getType();
Erik Pilkington02d5fb12019-03-19 20:44:18 +00001797 bool Dynamic = POSA->isDynamic();
David Majnemer5f177622015-12-30 05:13:03 +00001798
Erik Pilkington02d5fb12019-03-19 20:44:18 +00001799 auto Iter = PassObjectSizeArgs.insert({Type, Dynamic}).first;
David Majnemer622abd12015-12-31 05:36:52 +00001800 auto *TypePtr = (const void *)&*Iter;
Nico Weber73f05842019-06-01 10:12:07 +00001801 ArgBackRefMap::iterator Found = FunArgBackReferences.find(TypePtr);
David Majnemer5f177622015-12-30 05:13:03 +00001802
Nico Weber73f05842019-06-01 10:12:07 +00001803 if (Found == FunArgBackReferences.end()) {
Erik Pilkington02d5fb12019-03-19 20:44:18 +00001804 std::string Name =
1805 Dynamic ? "__pass_dynamic_object_size" : "__pass_object_size";
1806 mangleArtificialTagType(TTK_Enum, Name + llvm::utostr(Type), {"__clang"});
David Majnemer5f177622015-12-30 05:13:03 +00001807
Nico Weber73f05842019-06-01 10:12:07 +00001808 if (FunArgBackReferences.size() < 10) {
1809 size_t Size = FunArgBackReferences.size();
1810 FunArgBackReferences[TypePtr] = Size;
David Majnemer5f177622015-12-30 05:13:03 +00001811 }
1812 } else {
1813 Out << Found->second;
1814 }
1815}
1816
Erich Keane24a0f042018-12-14 23:17:34 +00001817void MicrosoftCXXNameMangler::mangleAddressSpaceType(QualType T,
1818 Qualifiers Quals,
1819 SourceRange Range) {
1820 // Address space is mangled as an unqualified templated type in the __clang
1821 // namespace. The demangled version of this is:
1822 // In the case of a language specific address space:
1823 // __clang::struct _AS[language_addr_space]<Type>
1824 // where:
1825 // <language_addr_space> ::= <OpenCL-addrspace> | <CUDA-addrspace>
1826 // <OpenCL-addrspace> ::= "CL" [ "global" | "local" | "constant" |
1827 // "private"| "generic" ]
1828 // <CUDA-addrspace> ::= "CU" [ "device" | "constant" | "shared" ]
1829 // Note that the above were chosen to match the Itanium mangling for this.
1830 //
1831 // In the case of a non-language specific address space:
1832 // __clang::struct _AS<TargetAS, Type>
1833 assert(Quals.hasAddressSpace() && "Not valid without address space");
1834 llvm::SmallString<32> ASMangling;
1835 llvm::raw_svector_ostream Stream(ASMangling);
1836 MicrosoftCXXNameMangler Extra(Context, Stream);
1837 Stream << "?$";
1838
1839 LangAS AS = Quals.getAddressSpace();
1840 if (Context.getASTContext().addressSpaceMapManglingFor(AS)) {
1841 unsigned TargetAS = Context.getASTContext().getTargetAddressSpace(AS);
1842 Extra.mangleSourceName("_AS");
1843 Extra.mangleIntegerLiteral(llvm::APSInt::getUnsigned(TargetAS),
1844 /*IsBoolean*/ false);
1845 } else {
1846 switch (AS) {
1847 default:
1848 llvm_unreachable("Not a language specific address space");
1849 case LangAS::opencl_global:
1850 Extra.mangleSourceName("_ASCLglobal");
1851 break;
1852 case LangAS::opencl_local:
1853 Extra.mangleSourceName("_ASCLlocal");
1854 break;
1855 case LangAS::opencl_constant:
1856 Extra.mangleSourceName("_ASCLconstant");
1857 break;
1858 case LangAS::opencl_private:
1859 Extra.mangleSourceName("_ASCLprivate");
1860 break;
1861 case LangAS::opencl_generic:
1862 Extra.mangleSourceName("_ASCLgeneric");
1863 break;
1864 case LangAS::cuda_device:
1865 Extra.mangleSourceName("_ASCUdevice");
1866 break;
1867 case LangAS::cuda_constant:
1868 Extra.mangleSourceName("_ASCUconstant");
1869 break;
1870 case LangAS::cuda_shared:
1871 Extra.mangleSourceName("_ASCUshared");
1872 break;
1873 }
1874 }
1875
1876 Extra.mangleType(T, Range, QMM_Escape);
1877 mangleQualifiers(Qualifiers(), false);
1878 mangleArtificialTagType(TTK_Struct, ASMangling, {"__clang"});
1879}
1880
Guy Benyei11169dd2012-12-18 14:30:41 +00001881void MicrosoftCXXNameMangler::mangleType(QualType T, SourceRange Range,
Peter Collingbourne2816c022013-04-25 04:25:40 +00001882 QualifierMangleMode QMM) {
Reid Kleckner18da98e2013-06-24 19:21:52 +00001883 // Don't use the canonical types. MSVC includes things like 'const' on
1884 // pointer arguments to function pointers that canonicalization strips away.
1885 T = T.getDesugaredType(getASTContext());
Guy Benyei11169dd2012-12-18 14:30:41 +00001886 Qualifiers Quals = T.getLocalQualifiers();
Erich Keane24a0f042018-12-14 23:17:34 +00001887
Reid Kleckner18da98e2013-06-24 19:21:52 +00001888 if (const ArrayType *AT = getASTContext().getAsArrayType(T)) {
1889 // If there were any Quals, getAsArrayType() pushed them onto the array
1890 // element type.
Peter Collingbourne2816c022013-04-25 04:25:40 +00001891 if (QMM == QMM_Mangle)
1892 Out << 'A';
1893 else if (QMM == QMM_Escape || QMM == QMM_Result)
1894 Out << "$$B";
Reid Kleckner18da98e2013-06-24 19:21:52 +00001895 mangleArrayType(AT);
Peter Collingbourne2816c022013-04-25 04:25:40 +00001896 return;
1897 }
1898
1899 bool IsPointer = T->isAnyPointerType() || T->isMemberPointerType() ||
David Majnemer7a3ce0c2015-05-26 01:30:45 +00001900 T->isReferenceType() || T->isBlockPointerType();
Peter Collingbourne2816c022013-04-25 04:25:40 +00001901
1902 switch (QMM) {
1903 case QMM_Drop:
Saleem Abdulrasoolbbce34a2018-02-09 03:23:54 +00001904 if (Quals.hasObjCLifetime())
1905 Quals = Quals.withoutObjCLifetime();
Peter Collingbourne2816c022013-04-25 04:25:40 +00001906 break;
1907 case QMM_Mangle:
1908 if (const FunctionType *FT = dyn_cast<FunctionType>(T)) {
1909 Out << '6';
Timur Iskhodzhanov555a7722013-10-04 11:25:05 +00001910 mangleFunctionType(FT);
Peter Collingbourne2816c022013-04-25 04:25:40 +00001911 return;
1912 }
Guy Benyei11169dd2012-12-18 14:30:41 +00001913 mangleQualifiers(Quals, false);
Peter Collingbourne2816c022013-04-25 04:25:40 +00001914 break;
1915 case QMM_Escape:
1916 if (!IsPointer && Quals) {
1917 Out << "$$C";
1918 mangleQualifiers(Quals, false);
1919 }
1920 break;
1921 case QMM_Result:
Andrey Bokhanko45d41322016-05-11 18:38:21 +00001922 // Presence of __unaligned qualifier shouldn't affect mangling here.
1923 Quals.removeUnaligned();
Saleem Abdulrasoolbbce34a2018-02-09 03:23:54 +00001924 if (Quals.hasObjCLifetime())
1925 Quals = Quals.withoutObjCLifetime();
Nico Weber3239f132018-07-23 20:04:00 +00001926 if ((!IsPointer && Quals) || isa<TagType>(T) || isArtificialTagType(T)) {
Peter Collingbourne2816c022013-04-25 04:25:40 +00001927 Out << '?';
1928 mangleQualifiers(Quals, false);
1929 }
1930 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00001931 }
1932
Peter Collingbourne2816c022013-04-25 04:25:40 +00001933 const Type *ty = T.getTypePtr();
Guy Benyei11169dd2012-12-18 14:30:41 +00001934
1935 switch (ty->getTypeClass()) {
1936#define ABSTRACT_TYPE(CLASS, PARENT)
1937#define NON_CANONICAL_TYPE(CLASS, PARENT) \
1938 case Type::CLASS: \
1939 llvm_unreachable("can't mangle non-canonical type " #CLASS "Type"); \
1940 return;
1941#define TYPE(CLASS, PARENT) \
1942 case Type::CLASS: \
David Majnemer7a3ce0c2015-05-26 01:30:45 +00001943 mangleType(cast<CLASS##Type>(ty), Quals, Range); \
Guy Benyei11169dd2012-12-18 14:30:41 +00001944 break;
1945#include "clang/AST/TypeNodes.def"
1946#undef ABSTRACT_TYPE
1947#undef NON_CANONICAL_TYPE
1948#undef TYPE
1949 }
1950}
1951
David Majnemer7a3ce0c2015-05-26 01:30:45 +00001952void MicrosoftCXXNameMangler::mangleType(const BuiltinType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00001953 SourceRange Range) {
1954 // <type> ::= <builtin-type>
1955 // <builtin-type> ::= X # void
1956 // ::= C # signed char
1957 // ::= D # char
1958 // ::= E # unsigned char
1959 // ::= F # short
1960 // ::= G # unsigned short (or wchar_t if it's not a builtin)
1961 // ::= H # int
1962 // ::= I # unsigned int
1963 // ::= J # long
1964 // ::= K # unsigned long
1965 // L # <none>
1966 // ::= M # float
1967 // ::= N # double
1968 // ::= O # long double (__float80 is mangled differently)
1969 // ::= _J # long long, __int64
1970 // ::= _K # unsigned long long, __int64
1971 // ::= _L # __int128
1972 // ::= _M # unsigned __int128
1973 // ::= _N # bool
1974 // _O # <array in parameter>
Richard Smith5178c6b2019-02-21 23:04:35 +00001975 // ::= _Q # char8_t
Reid Kleckner9e2584a2017-05-26 17:11:10 +00001976 // ::= _S # char16_t
Richard Smith5178c6b2019-02-21 23:04:35 +00001977 // ::= _T # __float80 (Intel)
Reid Kleckner9e2584a2017-05-26 17:11:10 +00001978 // ::= _U # char32_t
Guy Benyei11169dd2012-12-18 14:30:41 +00001979 // ::= _W # wchar_t
1980 // ::= _Z # __float80 (Digital Mars)
1981 switch (T->getKind()) {
Alexey Baderbdf7c842015-09-15 12:18:29 +00001982 case BuiltinType::Void:
1983 Out << 'X';
1984 break;
1985 case BuiltinType::SChar:
1986 Out << 'C';
1987 break;
1988 case BuiltinType::Char_U:
1989 case BuiltinType::Char_S:
1990 Out << 'D';
1991 break;
1992 case BuiltinType::UChar:
1993 Out << 'E';
1994 break;
1995 case BuiltinType::Short:
1996 Out << 'F';
1997 break;
1998 case BuiltinType::UShort:
1999 Out << 'G';
2000 break;
2001 case BuiltinType::Int:
2002 Out << 'H';
2003 break;
2004 case BuiltinType::UInt:
2005 Out << 'I';
2006 break;
2007 case BuiltinType::Long:
2008 Out << 'J';
2009 break;
2010 case BuiltinType::ULong:
2011 Out << 'K';
2012 break;
2013 case BuiltinType::Float:
2014 Out << 'M';
2015 break;
2016 case BuiltinType::Double:
2017 Out << 'N';
2018 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00002019 // TODO: Determine size and mangle accordingly
Alexey Baderbdf7c842015-09-15 12:18:29 +00002020 case BuiltinType::LongDouble:
2021 Out << 'O';
2022 break;
2023 case BuiltinType::LongLong:
2024 Out << "_J";
2025 break;
2026 case BuiltinType::ULongLong:
2027 Out << "_K";
2028 break;
2029 case BuiltinType::Int128:
2030 Out << "_L";
2031 break;
2032 case BuiltinType::UInt128:
2033 Out << "_M";
2034 break;
2035 case BuiltinType::Bool:
2036 Out << "_N";
2037 break;
Richard Smith5178c6b2019-02-21 23:04:35 +00002038 case BuiltinType::Char8:
2039 Out << "_Q";
2040 break;
Alexey Baderbdf7c842015-09-15 12:18:29 +00002041 case BuiltinType::Char16:
2042 Out << "_S";
2043 break;
2044 case BuiltinType::Char32:
2045 Out << "_U";
2046 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00002047 case BuiltinType::WChar_S:
Alexey Baderbdf7c842015-09-15 12:18:29 +00002048 case BuiltinType::WChar_U:
2049 Out << "_W";
2050 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00002051
2052#define BUILTIN_TYPE(Id, SingletonId)
2053#define PLACEHOLDER_TYPE(Id, SingletonId) \
2054 case BuiltinType::Id:
2055#include "clang/AST/BuiltinTypes.def"
2056 case BuiltinType::Dependent:
2057 llvm_unreachable("placeholder types shouldn't get to name mangling");
2058
Alexey Baderbdf7c842015-09-15 12:18:29 +00002059 case BuiltinType::ObjCId:
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002060 mangleArtificialTagType(TTK_Struct, "objc_object");
Alexey Baderbdf7c842015-09-15 12:18:29 +00002061 break;
2062 case BuiltinType::ObjCClass:
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002063 mangleArtificialTagType(TTK_Struct, "objc_class");
Alexey Baderbdf7c842015-09-15 12:18:29 +00002064 break;
2065 case BuiltinType::ObjCSel:
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002066 mangleArtificialTagType(TTK_Struct, "objc_selector");
Alexey Baderbdf7c842015-09-15 12:18:29 +00002067 break;
Guy Benyeid8a08ea2012-12-18 14:38:23 +00002068
Alexey Bader954ba212016-04-08 13:40:33 +00002069#define IMAGE_TYPE(ImgType, Id, SingletonId, Access, Suffix) \
2070 case BuiltinType::Id: \
2071 Out << "PAUocl_" #ImgType "_" #Suffix "@@"; \
Alexey Baderbdf7c842015-09-15 12:18:29 +00002072 break;
Alexey Baderb62f1442016-04-13 08:33:41 +00002073#include "clang/Basic/OpenCLImageTypes.def"
Alexey Baderbdf7c842015-09-15 12:18:29 +00002074 case BuiltinType::OCLSampler:
David Majnemer3add70b2015-12-29 11:45:53 +00002075 Out << "PA";
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002076 mangleArtificialTagType(TTK_Struct, "ocl_sampler");
Alexey Baderbdf7c842015-09-15 12:18:29 +00002077 break;
2078 case BuiltinType::OCLEvent:
David Majnemer3add70b2015-12-29 11:45:53 +00002079 Out << "PA";
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002080 mangleArtificialTagType(TTK_Struct, "ocl_event");
Alexey Baderbdf7c842015-09-15 12:18:29 +00002081 break;
Alexey Bader9c8453f2015-09-15 11:18:52 +00002082 case BuiltinType::OCLClkEvent:
David Majnemer3add70b2015-12-29 11:45:53 +00002083 Out << "PA";
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002084 mangleArtificialTagType(TTK_Struct, "ocl_clkevent");
Alexey Bader9c8453f2015-09-15 11:18:52 +00002085 break;
2086 case BuiltinType::OCLQueue:
David Majnemer3add70b2015-12-29 11:45:53 +00002087 Out << "PA";
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002088 mangleArtificialTagType(TTK_Struct, "ocl_queue");
Alexey Bader9c8453f2015-09-15 11:18:52 +00002089 break;
Alexey Bader9c8453f2015-09-15 11:18:52 +00002090 case BuiltinType::OCLReserveID:
David Majnemer3add70b2015-12-29 11:45:53 +00002091 Out << "PA";
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002092 mangleArtificialTagType(TTK_Struct, "ocl_reserveid");
Alexey Bader9c8453f2015-09-15 11:18:52 +00002093 break;
Andrew Savonichev3fee3512018-11-08 11:25:41 +00002094#define EXT_OPAQUE_TYPE(ExtType, Id, Ext) \
2095 case BuiltinType::Id: \
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002096 mangleArtificialTagType(TTK_Struct, "ocl_" #ExtType); \
Andrew Savonichev3fee3512018-11-08 11:25:41 +00002097 break;
2098#include "clang/Basic/OpenCLExtensionTypes.def"
David Majnemerb4119f72013-12-13 01:06:04 +00002099
Alexey Baderbdf7c842015-09-15 12:18:29 +00002100 case BuiltinType::NullPtr:
2101 Out << "$$T";
2102 break;
Guy Benyei11169dd2012-12-18 14:30:41 +00002103
Sjoerd Meijercc623ad2017-09-08 15:15:00 +00002104 case BuiltinType::Float16:
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002105 mangleArtificialTagType(TTK_Struct, "_Float16", {"__clang"});
Erich Keane5da1e692018-04-17 22:00:54 +00002106 break;
2107
Erich Keane14c10852018-05-01 14:16:15 +00002108 case BuiltinType::Half:
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002109 mangleArtificialTagType(TTK_Struct, "_Half", {"__clang"});
Erich Keane14c10852018-05-01 14:16:15 +00002110 break;
2111
Richard Sandifordeb485fb2019-08-09 08:52:54 +00002112#define SVE_TYPE(Name, Id, SingletonId) \
2113 case BuiltinType::Id:
2114#include "clang/Basic/AArch64SVEACLETypes.def"
Leonard Chanf921d852018-06-04 16:07:52 +00002115 case BuiltinType::ShortAccum:
2116 case BuiltinType::Accum:
2117 case BuiltinType::LongAccum:
2118 case BuiltinType::UShortAccum:
2119 case BuiltinType::UAccum:
2120 case BuiltinType::ULongAccum:
Leonard Chanab80f3c2018-06-14 14:53:51 +00002121 case BuiltinType::ShortFract:
2122 case BuiltinType::Fract:
2123 case BuiltinType::LongFract:
2124 case BuiltinType::UShortFract:
2125 case BuiltinType::UFract:
2126 case BuiltinType::ULongFract:
2127 case BuiltinType::SatShortAccum:
2128 case BuiltinType::SatAccum:
2129 case BuiltinType::SatLongAccum:
2130 case BuiltinType::SatUShortAccum:
2131 case BuiltinType::SatUAccum:
2132 case BuiltinType::SatULongAccum:
2133 case BuiltinType::SatShortFract:
2134 case BuiltinType::SatFract:
2135 case BuiltinType::SatLongFract:
2136 case BuiltinType::SatUShortFract:
2137 case BuiltinType::SatUFract:
2138 case BuiltinType::SatULongFract:
Erich Keane14c10852018-05-01 14:16:15 +00002139 case BuiltinType::Float128: {
Guy Benyei11169dd2012-12-18 14:30:41 +00002140 DiagnosticsEngine &Diags = Context.getDiags();
Alexey Baderbdf7c842015-09-15 12:18:29 +00002141 unsigned DiagID = Diags.getCustomDiagID(
2142 DiagnosticsEngine::Error, "cannot mangle this built-in %0 type yet");
Guy Benyei11169dd2012-12-18 14:30:41 +00002143 Diags.Report(Range.getBegin(), DiagID)
Alexey Baderbdf7c842015-09-15 12:18:29 +00002144 << T->getName(Context.getASTContext().getPrintingPolicy()) << Range;
Guy Benyei11169dd2012-12-18 14:30:41 +00002145 break;
2146 }
2147 }
2148}
2149
2150// <type> ::= <function-type>
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002151void MicrosoftCXXNameMangler::mangleType(const FunctionProtoType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002152 SourceRange) {
2153 // Structors only appear in decls, so at this point we know it's not a
2154 // structor type.
2155 // FIXME: This may not be lambda-friendly.
Anastasia Stulovac61eaa52019-01-28 11:37:49 +00002156 if (T->getMethodQuals() || T->getRefQualifier() != RQ_None) {
David Majnemer2f6a96e2014-08-12 17:53:10 +00002157 Out << "$$A8@@";
2158 mangleFunctionType(T, /*D=*/nullptr, /*ForceThisQuals=*/true);
2159 } else {
2160 Out << "$$A6";
2161 mangleFunctionType(T);
2162 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002163}
2164void MicrosoftCXXNameMangler::mangleType(const FunctionNoProtoType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002165 Qualifiers, SourceRange) {
Peter Collingbourneeeebc412015-08-07 23:25:47 +00002166 Out << "$$A6";
2167 mangleFunctionType(T);
Guy Benyei11169dd2012-12-18 14:30:41 +00002168}
2169
Peter Collingbourne2816c022013-04-25 04:25:40 +00002170void MicrosoftCXXNameMangler::mangleFunctionType(const FunctionType *T,
2171 const FunctionDecl *D,
Reid Klecknerd2f98772018-12-17 23:10:43 +00002172 bool ForceThisQuals,
2173 bool MangleExceptionSpec) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002174 // <function-type> ::= <this-cvr-qualifiers> <calling-convention>
2175 // <return-type> <argument-list> <throw-spec>
Peter Collingbourneeeebc412015-08-07 23:25:47 +00002176 const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(T);
Guy Benyei11169dd2012-12-18 14:30:41 +00002177
Reid Kleckner18da98e2013-06-24 19:21:52 +00002178 SourceRange Range;
2179 if (D) Range = D->getSourceRange();
2180
David Majnemer41311192016-01-17 07:09:24 +00002181 bool IsInLambda = false;
David Majnemerdfa6d202015-03-11 18:36:39 +00002182 bool IsStructor = false, HasThisQuals = ForceThisQuals, IsCtorClosure = false;
2183 CallingConv CC = T->getCallConv();
Timur Iskhodzhanov555a7722013-10-04 11:25:05 +00002184 if (const CXXMethodDecl *MD = dyn_cast_or_null<CXXMethodDecl>(D)) {
David Majnemer41311192016-01-17 07:09:24 +00002185 if (MD->getParent()->isLambda())
2186 IsInLambda = true;
Timur Iskhodzhanov555a7722013-10-04 11:25:05 +00002187 if (MD->isInstance())
David Majnemer2f6a96e2014-08-12 17:53:10 +00002188 HasThisQuals = true;
David Majnemerdfa6d202015-03-11 18:36:39 +00002189 if (isa<CXXDestructorDecl>(MD)) {
Timur Iskhodzhanov555a7722013-10-04 11:25:05 +00002190 IsStructor = true;
David Majnemerdfa6d202015-03-11 18:36:39 +00002191 } else if (isa<CXXConstructorDecl>(MD)) {
2192 IsStructor = true;
2193 IsCtorClosure = (StructorType == Ctor_CopyingClosure ||
2194 StructorType == Ctor_DefaultClosure) &&
Reid Kleckner290feb92017-01-09 17:04:37 +00002195 isStructorDecl(MD);
David Majnemerdfa6d202015-03-11 18:36:39 +00002196 if (IsCtorClosure)
2197 CC = getASTContext().getDefaultCallingConvention(
2198 /*IsVariadic=*/false, /*IsCXXMethod=*/true);
2199 }
Timur Iskhodzhanov555a7722013-10-04 11:25:05 +00002200 }
2201
Guy Benyei11169dd2012-12-18 14:30:41 +00002202 // If this is a C++ instance method, mangle the CVR qualifiers for the
2203 // this pointer.
David Majnemer2f6a96e2014-08-12 17:53:10 +00002204 if (HasThisQuals) {
Anastasia Stulovac61eaa52019-01-28 11:37:49 +00002205 Qualifiers Quals = Proto->getMethodQuals();
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002206 manglePointerExtQualifiers(Quals, /*PointeeType=*/QualType());
David Majnemere3785bb2014-04-23 05:16:56 +00002207 mangleRefQualifier(Proto->getRefQualifier());
David Majnemer2f6a96e2014-08-12 17:53:10 +00002208 mangleQualifiers(Quals, /*IsMember=*/false);
David Majnemer6dda7bb2013-08-15 08:13:23 +00002209 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002210
David Majnemerdfa6d202015-03-11 18:36:39 +00002211 mangleCallingConvention(CC);
Guy Benyei11169dd2012-12-18 14:30:41 +00002212
2213 // <return-type> ::= <type>
2214 // ::= @ # structors (they have no declared return type)
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +00002215 if (IsStructor) {
David Majnemerdc169752017-02-14 00:54:11 +00002216 if (isa<CXXDestructorDecl>(D) && isStructorDecl(D)) {
2217 // The scalar deleting destructor takes an extra int argument which is not
2218 // reflected in the AST.
2219 if (StructorType == Dtor_Deleting) {
2220 Out << (PointersAre64Bit ? "PEAXI@Z" : "PAXI@Z");
2221 return;
2222 }
2223 // The vbase destructor returns void which is not reflected in the AST.
2224 if (StructorType == Dtor_Complete) {
2225 Out << "XXZ";
2226 return;
2227 }
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +00002228 }
David Majnemerdfa6d202015-03-11 18:36:39 +00002229 if (IsCtorClosure) {
2230 // Default constructor closure and copy constructor closure both return
2231 // void.
2232 Out << 'X';
2233
2234 if (StructorType == Ctor_DefaultClosure) {
2235 // Default constructor closure always has no arguments.
2236 Out << 'X';
2237 } else if (StructorType == Ctor_CopyingClosure) {
2238 // Copy constructor closure always takes an unqualified reference.
Nico Weber73f05842019-06-01 10:12:07 +00002239 mangleFunctionArgumentType(getASTContext().getLValueReferenceType(
2240 Proto->getParamType(0)
2241 ->getAs<LValueReferenceType>()
2242 ->getPointeeType(),
2243 /*SpelledAsLValue=*/true),
2244 Range);
David Majnemer37fd66e2015-03-13 22:36:55 +00002245 Out << '@';
David Majnemerdfa6d202015-03-11 18:36:39 +00002246 } else {
2247 llvm_unreachable("unexpected constructor closure!");
2248 }
David Majnemer37fd66e2015-03-13 22:36:55 +00002249 Out << 'Z';
David Majnemerdfa6d202015-03-11 18:36:39 +00002250 return;
2251 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002252 Out << '@';
Timur Iskhodzhanovee6bc532013-02-13 08:37:51 +00002253 } else {
Peter Collingbourneeeebc412015-08-07 23:25:47 +00002254 QualType ResultType = T->getReturnType();
David Majnemer2e1e04912014-04-01 05:29:46 +00002255 if (const auto *AT =
2256 dyn_cast_or_null<AutoType>(ResultType->getContainedAutoType())) {
2257 Out << '?';
2258 mangleQualifiers(ResultType.getLocalQualifiers(), /*IsMember=*/false);
2259 Out << '?';
Richard Smithe301ba22015-11-11 02:02:15 +00002260 assert(AT->getKeyword() != AutoTypeKeyword::GNUAutoType &&
2261 "shouldn't need to mangle __auto_type!");
David Majnemer2e1e04912014-04-01 05:29:46 +00002262 mangleSourceName(AT->isDecltypeAuto() ? "<decltype-auto>" : "<auto>");
2263 Out << '@';
David Majnemer41311192016-01-17 07:09:24 +00002264 } else if (IsInLambda) {
2265 Out << '@';
David Majnemer2e1e04912014-04-01 05:29:46 +00002266 } else {
2267 if (ResultType->isVoidType())
2268 ResultType = ResultType.getUnqualifiedType();
2269 mangleType(ResultType, Range, QMM_Result);
2270 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002271 }
2272
2273 // <argument-list> ::= X # void
2274 // ::= <type>+ @
2275 // ::= <type>* Z # varargs
Peter Collingbourneeeebc412015-08-07 23:25:47 +00002276 if (!Proto) {
2277 // Function types without prototypes can arise when mangling a function type
2278 // within an overloadable function in C. We mangle these as the absence of
2279 // any parameter types (not even an empty parameter list).
2280 Out << '@';
2281 } else if (Proto->getNumParams() == 0 && !Proto->isVariadic()) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002282 Out << 'X';
2283 } else {
Reid Kleckner18da98e2013-06-24 19:21:52 +00002284 // Happens for function pointer type arguments for example.
George Burgess IV3e3bb95b2015-12-02 21:58:08 +00002285 for (unsigned I = 0, E = Proto->getNumParams(); I != E; ++I) {
Nico Weber73f05842019-06-01 10:12:07 +00002286 mangleFunctionArgumentType(Proto->getParamType(I), Range);
Simon Pilgrim27cc0542017-02-15 15:12:06 +00002287 // Mangle each pass_object_size parameter as if it's a parameter of enum
George Burgess IV3e3bb95b2015-12-02 21:58:08 +00002288 // type passed directly after the parameter with the pass_object_size
2289 // attribute. The aforementioned enum's name is __pass_object_size, and we
2290 // pretend it resides in a top-level namespace called __clang.
2291 //
2292 // FIXME: Is there a defined extension notation for the MS ABI, or is it
2293 // necessary to just cross our fingers and hope this type+namespace
2294 // combination doesn't conflict with anything?
2295 if (D)
David Majnemer5f177622015-12-30 05:13:03 +00002296 if (const auto *P = D->getParamDecl(I)->getAttr<PassObjectSizeAttr>())
2297 manglePassObjectSizeArg(P);
George Burgess IV3e3bb95b2015-12-02 21:58:08 +00002298 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002299 // <builtin-type> ::= Z # ellipsis
2300 if (Proto->isVariadic())
2301 Out << 'Z';
2302 else
2303 Out << '@';
2304 }
2305
Reid Klecknerd2f98772018-12-17 23:10:43 +00002306 if (MangleExceptionSpec && getASTContext().getLangOpts().CPlusPlus17 &&
2307 getASTContext().getLangOpts().isCompatibleWithMSVC(
2308 LangOptions::MSVC2017_5))
2309 mangleThrowSpecification(Proto);
2310 else
2311 Out << 'Z';
Guy Benyei11169dd2012-12-18 14:30:41 +00002312}
2313
2314void MicrosoftCXXNameMangler::mangleFunctionClass(const FunctionDecl *FD) {
Reid Kleckner369f3162013-05-14 20:30:42 +00002315 // <function-class> ::= <member-function> E? # E designates a 64-bit 'this'
2316 // # pointer. in 64-bit mode *all*
2317 // # 'this' pointers are 64-bit.
2318 // ::= <global-function>
2319 // <member-function> ::= A # private: near
2320 // ::= B # private: far
2321 // ::= C # private: static near
2322 // ::= D # private: static far
2323 // ::= E # private: virtual near
2324 // ::= F # private: virtual far
Reid Kleckner369f3162013-05-14 20:30:42 +00002325 // ::= I # protected: near
2326 // ::= J # protected: far
2327 // ::= K # protected: static near
2328 // ::= L # protected: static far
2329 // ::= M # protected: virtual near
2330 // ::= N # protected: virtual far
Reid Kleckner369f3162013-05-14 20:30:42 +00002331 // ::= Q # public: near
2332 // ::= R # public: far
2333 // ::= S # public: static near
2334 // ::= T # public: static far
2335 // ::= U # public: virtual near
2336 // ::= V # public: virtual far
Reid Kleckner369f3162013-05-14 20:30:42 +00002337 // <global-function> ::= Y # global near
2338 // ::= Z # global far
Guy Benyei11169dd2012-12-18 14:30:41 +00002339 if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD)) {
David Majnemerdc169752017-02-14 00:54:11 +00002340 bool IsVirtual = MD->isVirtual();
2341 // When mangling vbase destructor variants, ignore whether or not the
2342 // underlying destructor was defined to be virtual.
2343 if (isa<CXXDestructorDecl>(MD) && isStructorDecl(MD) &&
2344 StructorType == Dtor_Complete) {
2345 IsVirtual = false;
2346 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002347 switch (MD->getAccess()) {
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00002348 case AS_none:
2349 llvm_unreachable("Unsupported access specifier");
Guy Benyei11169dd2012-12-18 14:30:41 +00002350 case AS_private:
2351 if (MD->isStatic())
2352 Out << 'C';
David Majnemerdc169752017-02-14 00:54:11 +00002353 else if (IsVirtual)
Guy Benyei11169dd2012-12-18 14:30:41 +00002354 Out << 'E';
2355 else
2356 Out << 'A';
2357 break;
2358 case AS_protected:
2359 if (MD->isStatic())
2360 Out << 'K';
David Majnemerdc169752017-02-14 00:54:11 +00002361 else if (IsVirtual)
Guy Benyei11169dd2012-12-18 14:30:41 +00002362 Out << 'M';
2363 else
2364 Out << 'I';
2365 break;
2366 case AS_public:
2367 if (MD->isStatic())
2368 Out << 'S';
David Majnemerdc169752017-02-14 00:54:11 +00002369 else if (IsVirtual)
Guy Benyei11169dd2012-12-18 14:30:41 +00002370 Out << 'U';
2371 else
2372 Out << 'Q';
2373 }
David Majnemer5a7cfea2015-05-18 00:05:29 +00002374 } else {
Guy Benyei11169dd2012-12-18 14:30:41 +00002375 Out << 'Y';
David Majnemer5a7cfea2015-05-18 00:05:29 +00002376 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002377}
David Majnemerdfa6d202015-03-11 18:36:39 +00002378void MicrosoftCXXNameMangler::mangleCallingConvention(CallingConv CC) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002379 // <calling-convention> ::= A # __cdecl
2380 // ::= B # __export __cdecl
2381 // ::= C # __pascal
2382 // ::= D # __export __pascal
2383 // ::= E # __thiscall
2384 // ::= F # __export __thiscall
2385 // ::= G # __stdcall
2386 // ::= H # __export __stdcall
2387 // ::= I # __fastcall
2388 // ::= J # __export __fastcall
Reid Klecknerd7857f02014-10-24 17:42:17 +00002389 // ::= Q # __vectorcall
Erich Keane757d3172016-11-02 18:29:35 +00002390 // ::= w # __regcall
Guy Benyei11169dd2012-12-18 14:30:41 +00002391 // The 'export' calling conventions are from a bygone era
2392 // (*cough*Win16*cough*) when functions were declared for export with
2393 // that keyword. (It didn't actually export them, it just made them so
2394 // that they could be in a DLL and somebody from another module could call
2395 // them.)
David Majnemerdfa6d202015-03-11 18:36:39 +00002396
Guy Benyei11169dd2012-12-18 14:30:41 +00002397 switch (CC) {
2398 default:
2399 llvm_unreachable("Unsupported CC for mangling");
Martin Storsjo022e7822017-07-17 20:49:45 +00002400 case CC_Win64:
Charles Davisb5a214e2013-08-30 04:39:01 +00002401 case CC_X86_64SysV:
Guy Benyei11169dd2012-12-18 14:30:41 +00002402 case CC_C: Out << 'A'; break;
2403 case CC_X86Pascal: Out << 'C'; break;
2404 case CC_X86ThisCall: Out << 'E'; break;
2405 case CC_X86StdCall: Out << 'G'; break;
2406 case CC_X86FastCall: Out << 'I'; break;
Reid Klecknerd7857f02014-10-24 17:42:17 +00002407 case CC_X86VectorCall: Out << 'Q'; break;
Saleem Abdulrasool58340522018-02-07 01:55:08 +00002408 case CC_Swift: Out << 'S'; break;
Saleem Abdulrasool13aeee02018-04-02 22:25:50 +00002409 case CC_PreserveMost: Out << 'U'; break;
Erich Keane757d3172016-11-02 18:29:35 +00002410 case CC_X86RegCall: Out << 'w'; break;
Guy Benyei11169dd2012-12-18 14:30:41 +00002411 }
2412}
David Majnemerdfa6d202015-03-11 18:36:39 +00002413void MicrosoftCXXNameMangler::mangleCallingConvention(const FunctionType *T) {
2414 mangleCallingConvention(T->getCallConv());
2415}
Reid Klecknerd2f98772018-12-17 23:10:43 +00002416
Guy Benyei11169dd2012-12-18 14:30:41 +00002417void MicrosoftCXXNameMangler::mangleThrowSpecification(
2418 const FunctionProtoType *FT) {
Reid Klecknerd2f98772018-12-17 23:10:43 +00002419 // <throw-spec> ::= Z # (default)
2420 // ::= _E # noexcept
2421 if (FT->canThrow())
2422 Out << 'Z';
2423 else
2424 Out << "_E";
Guy Benyei11169dd2012-12-18 14:30:41 +00002425}
2426
2427void MicrosoftCXXNameMangler::mangleType(const UnresolvedUsingType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002428 Qualifiers, SourceRange Range) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002429 // Probably should be mangled as a template instantiation; need to see what
2430 // VC does first.
2431 DiagnosticsEngine &Diags = Context.getDiags();
2432 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2433 "cannot mangle this unresolved dependent type yet");
2434 Diags.Report(Range.getBegin(), DiagID)
2435 << Range;
2436}
2437
2438// <type> ::= <union-type> | <struct-type> | <class-type> | <enum-type>
2439// <union-type> ::= T <name>
2440// <struct-type> ::= U <name>
2441// <class-type> ::= V <name>
David Majnemer048f90c2013-12-09 04:28:34 +00002442// <enum-type> ::= W4 <name>
David Majnemer3add70b2015-12-29 11:45:53 +00002443void MicrosoftCXXNameMangler::mangleTagTypeKind(TagTypeKind TTK) {
2444 switch (TTK) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002445 case TTK_Union:
2446 Out << 'T';
2447 break;
2448 case TTK_Struct:
2449 case TTK_Interface:
2450 Out << 'U';
2451 break;
2452 case TTK_Class:
2453 Out << 'V';
2454 break;
2455 case TTK_Enum:
David Majnemer048f90c2013-12-09 04:28:34 +00002456 Out << "W4";
Guy Benyei11169dd2012-12-18 14:30:41 +00002457 break;
2458 }
David Majnemer3add70b2015-12-29 11:45:53 +00002459}
2460void MicrosoftCXXNameMangler::mangleType(const EnumType *T, Qualifiers,
2461 SourceRange) {
2462 mangleType(cast<TagType>(T)->getDecl());
2463}
2464void MicrosoftCXXNameMangler::mangleType(const RecordType *T, Qualifiers,
2465 SourceRange) {
2466 mangleType(cast<TagType>(T)->getDecl());
2467}
2468void MicrosoftCXXNameMangler::mangleType(const TagDecl *TD) {
2469 mangleTagTypeKind(TD->getTagKind());
David Majnemer0db0ca42013-08-05 22:26:46 +00002470 mangleName(TD);
Guy Benyei11169dd2012-12-18 14:30:41 +00002471}
Nico Weber3239f132018-07-23 20:04:00 +00002472
2473// If you add a call to this, consider updating isArtificialTagType() too.
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002474void MicrosoftCXXNameMangler::mangleArtificialTagType(
Nico Weber2b5f3fd2018-08-01 11:56:20 +00002475 TagTypeKind TK, StringRef UnqualifiedName,
2476 ArrayRef<StringRef> NestedNames) {
David Majnemer3add70b2015-12-29 11:45:53 +00002477 // <name> ::= <unscoped-name> {[<named-scope>]+ | [<nested-name>]}? @
2478 mangleTagTypeKind(TK);
2479
2480 // Always start with the unqualified name.
2481 mangleSourceName(UnqualifiedName);
2482
2483 for (auto I = NestedNames.rbegin(), E = NestedNames.rend(); I != E; ++I)
2484 mangleSourceName(*I);
2485
2486 // Terminate the whole name with an '@'.
2487 Out << '@';
2488}
Guy Benyei11169dd2012-12-18 14:30:41 +00002489
2490// <type> ::= <array-type>
2491// <array-type> ::= <pointer-cvr-qualifiers> <cvr-qualifiers>
2492// [Y <dimension-count> <dimension>+]
Reid Kleckner369f3162013-05-14 20:30:42 +00002493// <element-type> # as global, E is never required
Guy Benyei11169dd2012-12-18 14:30:41 +00002494// It's supposed to be the other way around, but for some strange reason, it
2495// isn't. Today this behavior is retained for the sole purpose of backwards
2496// compatibility.
David Majnemer5a1b2042013-09-11 04:44:30 +00002497void MicrosoftCXXNameMangler::mangleDecayedArrayType(const ArrayType *T) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002498 // This isn't a recursive mangling, so now we have to do it all in this
2499 // one call.
David Majnemer8eec58f2014-02-18 14:20:10 +00002500 manglePointerCVQualifiers(T->getElementType().getQualifiers());
Peter Collingbourne2816c022013-04-25 04:25:40 +00002501 mangleType(T->getElementType(), SourceRange());
Guy Benyei11169dd2012-12-18 14:30:41 +00002502}
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002503void MicrosoftCXXNameMangler::mangleType(const ConstantArrayType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002504 SourceRange) {
Peter Collingbourne2816c022013-04-25 04:25:40 +00002505 llvm_unreachable("Should have been special cased");
Guy Benyei11169dd2012-12-18 14:30:41 +00002506}
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002507void MicrosoftCXXNameMangler::mangleType(const VariableArrayType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002508 SourceRange) {
Peter Collingbourne2816c022013-04-25 04:25:40 +00002509 llvm_unreachable("Should have been special cased");
Guy Benyei11169dd2012-12-18 14:30:41 +00002510}
2511void MicrosoftCXXNameMangler::mangleType(const DependentSizedArrayType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002512 Qualifiers, SourceRange) {
Peter Collingbourne2816c022013-04-25 04:25:40 +00002513 llvm_unreachable("Should have been special cased");
Guy Benyei11169dd2012-12-18 14:30:41 +00002514}
2515void MicrosoftCXXNameMangler::mangleType(const IncompleteArrayType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002516 Qualifiers, SourceRange) {
Peter Collingbourne2816c022013-04-25 04:25:40 +00002517 llvm_unreachable("Should have been special cased");
Guy Benyei11169dd2012-12-18 14:30:41 +00002518}
Reid Kleckner18da98e2013-06-24 19:21:52 +00002519void MicrosoftCXXNameMangler::mangleArrayType(const ArrayType *T) {
Peter Collingbourne2816c022013-04-25 04:25:40 +00002520 QualType ElementTy(T, 0);
Guy Benyei11169dd2012-12-18 14:30:41 +00002521 SmallVector<llvm::APInt, 3> Dimensions;
2522 for (;;) {
David Majnemer9595b7d2015-04-23 07:42:08 +00002523 if (ElementTy->isConstantArrayType()) {
2524 const ConstantArrayType *CAT =
2525 getASTContext().getAsConstantArrayType(ElementTy);
Guy Benyei11169dd2012-12-18 14:30:41 +00002526 Dimensions.push_back(CAT->getSize());
2527 ElementTy = CAT->getElementType();
David Majnemer9595b7d2015-04-23 07:42:08 +00002528 } else if (ElementTy->isIncompleteArrayType()) {
2529 const IncompleteArrayType *IAT =
2530 getASTContext().getAsIncompleteArrayType(ElementTy);
2531 Dimensions.push_back(llvm::APInt(32, 0));
2532 ElementTy = IAT->getElementType();
Guy Benyei11169dd2012-12-18 14:30:41 +00002533 } else if (ElementTy->isVariableArrayType()) {
2534 const VariableArrayType *VAT =
2535 getASTContext().getAsVariableArrayType(ElementTy);
David Majnemer9595b7d2015-04-23 07:42:08 +00002536 Dimensions.push_back(llvm::APInt(32, 0));
2537 ElementTy = VAT->getElementType();
Guy Benyei11169dd2012-12-18 14:30:41 +00002538 } else if (ElementTy->isDependentSizedArrayType()) {
2539 // The dependent expression has to be folded into a constant (TODO).
2540 const DependentSizedArrayType *DSAT =
2541 getASTContext().getAsDependentSizedArrayType(ElementTy);
2542 DiagnosticsEngine &Diags = Context.getDiags();
2543 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2544 "cannot mangle this dependent-length array yet");
2545 Diags.Report(DSAT->getSizeExpr()->getExprLoc(), DiagID)
2546 << DSAT->getBracketsRange();
2547 return;
David Majnemer9595b7d2015-04-23 07:42:08 +00002548 } else {
2549 break;
Peter Collingbourne2816c022013-04-25 04:25:40 +00002550 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002551 }
Peter Collingbourne2816c022013-04-25 04:25:40 +00002552 Out << 'Y';
2553 // <dimension-count> ::= <number> # number of extra dimensions
2554 mangleNumber(Dimensions.size());
David Majnemerb926dbc2014-04-23 05:16:53 +00002555 for (const llvm::APInt &Dimension : Dimensions)
2556 mangleNumber(Dimension.getLimitedValue());
Reid Kleckner18da98e2013-06-24 19:21:52 +00002557 mangleType(ElementTy, SourceRange(), QMM_Escape);
Guy Benyei11169dd2012-12-18 14:30:41 +00002558}
2559
2560// <type> ::= <pointer-to-member-type>
2561// <pointer-to-member-type> ::= <pointer-cvr-qualifiers> <cvr-qualifiers>
2562// <class name> <type>
Nico Weber2b5f3fd2018-08-01 11:56:20 +00002563void MicrosoftCXXNameMangler::mangleType(const MemberPointerType *T,
2564 Qualifiers Quals, SourceRange Range) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002565 QualType PointeeType = T->getPointeeType();
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002566 manglePointerCVQualifiers(Quals);
2567 manglePointerExtQualifiers(Quals, PointeeType);
Guy Benyei11169dd2012-12-18 14:30:41 +00002568 if (const FunctionProtoType *FPT = PointeeType->getAs<FunctionProtoType>()) {
2569 Out << '8';
2570 mangleName(T->getClass()->castAs<RecordType>()->getDecl());
Craig Topper36250ad2014-05-12 05:36:57 +00002571 mangleFunctionType(FPT, nullptr, true);
Guy Benyei11169dd2012-12-18 14:30:41 +00002572 } else {
2573 mangleQualifiers(PointeeType.getQualifiers(), true);
2574 mangleName(T->getClass()->castAs<RecordType>()->getDecl());
Peter Collingbourne2816c022013-04-25 04:25:40 +00002575 mangleType(PointeeType, Range, QMM_Drop);
Guy Benyei11169dd2012-12-18 14:30:41 +00002576 }
2577}
2578
2579void MicrosoftCXXNameMangler::mangleType(const TemplateTypeParmType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002580 Qualifiers, SourceRange Range) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002581 DiagnosticsEngine &Diags = Context.getDiags();
2582 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2583 "cannot mangle this template type parameter type yet");
2584 Diags.Report(Range.getBegin(), DiagID)
2585 << Range;
2586}
2587
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002588void MicrosoftCXXNameMangler::mangleType(const SubstTemplateTypeParmPackType *T,
2589 Qualifiers, SourceRange Range) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002590 DiagnosticsEngine &Diags = Context.getDiags();
2591 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2592 "cannot mangle this substituted parameter pack yet");
2593 Diags.Report(Range.getBegin(), DiagID)
2594 << Range;
2595}
2596
2597// <type> ::= <pointer-type>
Reid Kleckner369f3162013-05-14 20:30:42 +00002598// <pointer-type> ::= E? <pointer-cvr-qualifiers> <cvr-qualifiers> <type>
Alp Tokerd4733632013-12-05 04:47:09 +00002599// # the E is required for 64-bit non-static pointers
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002600void MicrosoftCXXNameMangler::mangleType(const PointerType *T, Qualifiers Quals,
Guy Benyei11169dd2012-12-18 14:30:41 +00002601 SourceRange Range) {
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002602 QualType PointeeType = T->getPointeeType();
2603 manglePointerCVQualifiers(Quals);
2604 manglePointerExtQualifiers(Quals, PointeeType);
Erich Keane24a0f042018-12-14 23:17:34 +00002605
2606 if (PointeeType.getQualifiers().hasAddressSpace())
2607 mangleAddressSpaceType(PointeeType, PointeeType.getQualifiers(), Range);
2608 else
2609 mangleType(PointeeType, Range);
Guy Benyei11169dd2012-12-18 14:30:41 +00002610}
Saleem Abdulrasool5a3e50a2017-08-23 22:38:58 +00002611
Guy Benyei11169dd2012-12-18 14:30:41 +00002612void MicrosoftCXXNameMangler::mangleType(const ObjCObjectPointerType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002613 Qualifiers Quals, SourceRange Range) {
2614 QualType PointeeType = T->getPointeeType();
Saleem Abdulrasoolbbce34a2018-02-09 03:23:54 +00002615 switch (Quals.getObjCLifetime()) {
2616 case Qualifiers::OCL_None:
2617 case Qualifiers::OCL_ExplicitNone:
2618 break;
2619 case Qualifiers::OCL_Autoreleasing:
2620 case Qualifiers::OCL_Strong:
2621 case Qualifiers::OCL_Weak:
2622 return mangleObjCLifetime(PointeeType, Quals, Range);
2623 }
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002624 manglePointerCVQualifiers(Quals);
2625 manglePointerExtQualifiers(Quals, PointeeType);
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002626 mangleType(PointeeType, Range);
Guy Benyei11169dd2012-12-18 14:30:41 +00002627}
2628
2629// <type> ::= <reference-type>
Reid Kleckner369f3162013-05-14 20:30:42 +00002630// <reference-type> ::= A E? <cvr-qualifiers> <type>
Alp Tokerd4733632013-12-05 04:47:09 +00002631// # the E is required for 64-bit non-static lvalue references
Guy Benyei11169dd2012-12-18 14:30:41 +00002632void MicrosoftCXXNameMangler::mangleType(const LValueReferenceType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002633 Qualifiers Quals, SourceRange Range) {
2634 QualType PointeeType = T->getPointeeType();
David Majnemere3d71912015-12-31 05:36:50 +00002635 assert(!Quals.hasConst() && !Quals.hasVolatile() && "unexpected qualifier!");
2636 Out << 'A';
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002637 manglePointerExtQualifiers(Quals, PointeeType);
2638 mangleType(PointeeType, Range);
Guy Benyei11169dd2012-12-18 14:30:41 +00002639}
2640
2641// <type> ::= <r-value-reference-type>
Reid Kleckner369f3162013-05-14 20:30:42 +00002642// <r-value-reference-type> ::= $$Q E? <cvr-qualifiers> <type>
Alp Tokerd4733632013-12-05 04:47:09 +00002643// # the E is required for 64-bit non-static rvalue references
Guy Benyei11169dd2012-12-18 14:30:41 +00002644void MicrosoftCXXNameMangler::mangleType(const RValueReferenceType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002645 Qualifiers Quals, SourceRange Range) {
2646 QualType PointeeType = T->getPointeeType();
David Majnemere3d71912015-12-31 05:36:50 +00002647 assert(!Quals.hasConst() && !Quals.hasVolatile() && "unexpected qualifier!");
2648 Out << "$$Q";
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002649 manglePointerExtQualifiers(Quals, PointeeType);
2650 mangleType(PointeeType, Range);
Guy Benyei11169dd2012-12-18 14:30:41 +00002651}
2652
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002653void MicrosoftCXXNameMangler::mangleType(const ComplexType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002654 SourceRange Range) {
David Majnemer0b996dd2015-12-29 22:02:15 +00002655 QualType ElementType = T->getElementType();
2656
2657 llvm::SmallString<64> TemplateMangling;
2658 llvm::raw_svector_ostream Stream(TemplateMangling);
2659 MicrosoftCXXNameMangler Extra(Context, Stream);
2660 Stream << "?$";
2661 Extra.mangleSourceName("_Complex");
2662 Extra.mangleType(ElementType, Range, QMM_Escape);
2663
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002664 mangleArtificialTagType(TTK_Struct, TemplateMangling, {"__clang"});
Guy Benyei11169dd2012-12-18 14:30:41 +00002665}
2666
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002667// Returns true for types that mangleArtificialTagType() gets called for with
Nico Weber3239f132018-07-23 20:04:00 +00002668// TTK_Union, TTK_Struct, TTK_Class and where compatibility with MSVC's
2669// mangling matters.
2670// (It doesn't matter for Objective-C types and the like that cl.exe doesn't
2671// support.)
2672bool MicrosoftCXXNameMangler::isArtificialTagType(QualType T) const {
2673 const Type *ty = T.getTypePtr();
2674 switch (ty->getTypeClass()) {
2675 default:
2676 return false;
2677
2678 case Type::Vector: {
2679 // For ABI compatibility only __m64, __m128(id), and __m256(id) matter,
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002680 // but since mangleType(VectorType*) always calls mangleArtificialTagType()
Nico Weber3239f132018-07-23 20:04:00 +00002681 // just always return true (the other vector types are clang-only).
2682 return true;
2683 }
2684 }
2685}
2686
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002687void MicrosoftCXXNameMangler::mangleType(const VectorType *T, Qualifiers Quals,
Guy Benyei11169dd2012-12-18 14:30:41 +00002688 SourceRange Range) {
Reid Klecknere7e64d82013-03-26 16:56:59 +00002689 const BuiltinType *ET = T->getElementType()->getAs<BuiltinType>();
2690 assert(ET && "vectors with non-builtin elements are unsupported");
2691 uint64_t Width = getASTContext().getTypeSize(T);
2692 // Pattern match exactly the typedefs in our intrinsic headers. Anything that
2693 // doesn't match the Intel types uses a custom mangling below.
David Majnemer0a29fed2015-12-29 11:45:58 +00002694 size_t OutSizeBefore = Out.tell();
Saleem Abdulrasool21568182018-10-28 18:05:20 +00002695 if (!isa<ExtVectorType>(T)) {
2696 llvm::Triple::ArchType AT =
2697 getASTContext().getTargetInfo().getTriple().getArch();
2698 if (AT == llvm::Triple::x86 || AT == llvm::Triple::x86_64) {
2699 if (Width == 64 && ET->getKind() == BuiltinType::LongLong) {
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002700 mangleArtificialTagType(TTK_Union, "__m64");
Saleem Abdulrasool21568182018-10-28 18:05:20 +00002701 } else if (Width >= 128) {
2702 if (ET->getKind() == BuiltinType::Float)
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002703 mangleArtificialTagType(TTK_Union, "__m" + llvm::utostr(Width));
Saleem Abdulrasool21568182018-10-28 18:05:20 +00002704 else if (ET->getKind() == BuiltinType::LongLong)
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002705 mangleArtificialTagType(TTK_Union, "__m" + llvm::utostr(Width) + 'i');
Saleem Abdulrasool21568182018-10-28 18:05:20 +00002706 else if (ET->getKind() == BuiltinType::Double)
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002707 mangleArtificialTagType(TTK_Struct, "__m" + llvm::utostr(Width) + 'd');
Saleem Abdulrasool21568182018-10-28 18:05:20 +00002708 }
David Majnemerf04e3862015-06-01 07:41:37 +00002709 }
Reid Klecknere7e64d82013-03-26 16:56:59 +00002710 }
2711
David Majnemer0a29fed2015-12-29 11:45:58 +00002712 bool IsBuiltin = Out.tell() != OutSizeBefore;
David Majnemerf04e3862015-06-01 07:41:37 +00002713 if (!IsBuiltin) {
Reid Klecknere7e64d82013-03-26 16:56:59 +00002714 // The MS ABI doesn't have a special mangling for vector types, so we define
2715 // our own mangling to handle uses of __vector_size__ on user-specified
2716 // types, and for extensions like __v4sf.
David Majnemer0a29fed2015-12-29 11:45:58 +00002717
2718 llvm::SmallString<64> TemplateMangling;
2719 llvm::raw_svector_ostream Stream(TemplateMangling);
2720 MicrosoftCXXNameMangler Extra(Context, Stream);
2721 Stream << "?$";
2722 Extra.mangleSourceName("__vector");
2723 Extra.mangleType(QualType(ET, 0), Range, QMM_Escape);
2724 Extra.mangleIntegerLiteral(llvm::APSInt::getUnsigned(T->getNumElements()),
2725 /*IsBoolean=*/false);
2726
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002727 mangleArtificialTagType(TTK_Union, TemplateMangling, {"__clang"});
Reid Klecknere7e64d82013-03-26 16:56:59 +00002728 }
Guy Benyei11169dd2012-12-18 14:30:41 +00002729}
Reid Klecknere7e64d82013-03-26 16:56:59 +00002730
David Majnemere5c87872015-08-01 05:51:55 +00002731void MicrosoftCXXNameMangler::mangleType(const ExtVectorType *T,
2732 Qualifiers Quals, SourceRange Range) {
2733 mangleType(static_cast<const VectorType *>(T), Quals, Range);
Guy Benyei11169dd2012-12-18 14:30:41 +00002734}
Erich Keanef702b022018-07-13 19:46:04 +00002735
2736void MicrosoftCXXNameMangler::mangleType(const DependentVectorType *T,
2737 Qualifiers, SourceRange Range) {
2738 DiagnosticsEngine &Diags = Context.getDiags();
2739 unsigned DiagID = Diags.getCustomDiagID(
2740 DiagnosticsEngine::Error,
2741 "cannot mangle this dependent-sized vector type yet");
2742 Diags.Report(Range.getBegin(), DiagID) << Range;
2743}
2744
Guy Benyei11169dd2012-12-18 14:30:41 +00002745void MicrosoftCXXNameMangler::mangleType(const DependentSizedExtVectorType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002746 Qualifiers, SourceRange Range) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002747 DiagnosticsEngine &Diags = Context.getDiags();
2748 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2749 "cannot mangle this dependent-sized extended vector type yet");
2750 Diags.Report(Range.getBegin(), DiagID)
2751 << Range;
2752}
2753
Andrew Gozillon572bbb02017-10-02 06:25:51 +00002754void MicrosoftCXXNameMangler::mangleType(const DependentAddressSpaceType *T,
2755 Qualifiers, SourceRange Range) {
2756 DiagnosticsEngine &Diags = Context.getDiags();
2757 unsigned DiagID = Diags.getCustomDiagID(
2758 DiagnosticsEngine::Error,
2759 "cannot mangle this dependent address space type yet");
2760 Diags.Report(Range.getBegin(), DiagID) << Range;
2761}
2762
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002763void MicrosoftCXXNameMangler::mangleType(const ObjCInterfaceType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002764 SourceRange) {
Shoaib Meenaic8714d22018-10-04 19:50:14 +00002765 // ObjC interfaces have structs underlying them.
David Majnemer3add70b2015-12-29 11:45:53 +00002766 mangleTagTypeKind(TTK_Struct);
Shoaib Meenaic8714d22018-10-04 19:50:14 +00002767 mangleName(T->getDecl());
Guy Benyei11169dd2012-12-18 14:30:41 +00002768}
2769
Saleem Abdulrasoolf3a981e2018-09-28 16:47:53 +00002770void MicrosoftCXXNameMangler::mangleType(const ObjCObjectType *T,
2771 Qualifiers Quals, SourceRange Range) {
2772 if (T->isKindOfType())
2773 return mangleObjCKindOfType(T, Quals, Range);
2774
2775 if (T->qual_empty() && !T->isSpecialized())
Saleem Abdulrasoolf591e362018-01-26 19:08:26 +00002776 return mangleType(T->getBaseType(), Range, QMM_Drop);
2777
Nico Weber73f05842019-06-01 10:12:07 +00002778 ArgBackRefMap OuterFunArgsContext;
2779 ArgBackRefMap OuterTemplateArgsContext;
Saleem Abdulrasoolf591e362018-01-26 19:08:26 +00002780 BackRefVec OuterTemplateContext;
2781
Nico Weber73f05842019-06-01 10:12:07 +00002782 FunArgBackReferences.swap(OuterFunArgsContext);
2783 TemplateArgBackReferences.swap(OuterTemplateArgsContext);
Saleem Abdulrasoolf591e362018-01-26 19:08:26 +00002784 NameBackReferences.swap(OuterTemplateContext);
2785
2786 mangleTagTypeKind(TTK_Struct);
2787
2788 Out << "?$";
2789 if (T->isObjCId())
Shoaib Meenaic8714d22018-10-04 19:50:14 +00002790 mangleSourceName("objc_object");
Saleem Abdulrasoolf591e362018-01-26 19:08:26 +00002791 else if (T->isObjCClass())
Shoaib Meenaic8714d22018-10-04 19:50:14 +00002792 mangleSourceName("objc_class");
Saleem Abdulrasoolf591e362018-01-26 19:08:26 +00002793 else
Shoaib Meenaic8714d22018-10-04 19:50:14 +00002794 mangleSourceName(T->getInterface()->getName());
Saleem Abdulrasoolf591e362018-01-26 19:08:26 +00002795
Saleem Abdulrasoolbbce34a2018-02-09 03:23:54 +00002796 for (const auto &Q : T->quals())
2797 mangleObjCProtocol(Q);
Saleem Abdulrasoolf3a981e2018-09-28 16:47:53 +00002798
2799 if (T->isSpecialized())
2800 for (const auto &TA : T->getTypeArgs())
2801 mangleType(TA, Range, QMM_Drop);
2802
Saleem Abdulrasoolf591e362018-01-26 19:08:26 +00002803 Out << '@';
2804
2805 Out << '@';
2806
Nico Weber73f05842019-06-01 10:12:07 +00002807 FunArgBackReferences.swap(OuterFunArgsContext);
2808 TemplateArgBackReferences.swap(OuterTemplateArgsContext);
Saleem Abdulrasoolf591e362018-01-26 19:08:26 +00002809 NameBackReferences.swap(OuterTemplateContext);
Guy Benyei11169dd2012-12-18 14:30:41 +00002810}
2811
2812void MicrosoftCXXNameMangler::mangleType(const BlockPointerType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002813 Qualifiers Quals, SourceRange Range) {
2814 QualType PointeeType = T->getPointeeType();
2815 manglePointerCVQualifiers(Quals);
2816 manglePointerExtQualifiers(Quals, PointeeType);
2817
Guy Benyei11169dd2012-12-18 14:30:41 +00002818 Out << "_E";
2819
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002820 mangleFunctionType(PointeeType->castAs<FunctionProtoType>());
Guy Benyei11169dd2012-12-18 14:30:41 +00002821}
2822
David Majnemerf0a84f22013-08-16 08:29:13 +00002823void MicrosoftCXXNameMangler::mangleType(const InjectedClassNameType *,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002824 Qualifiers, SourceRange) {
David Majnemerf0a84f22013-08-16 08:29:13 +00002825 llvm_unreachable("Cannot mangle injected class name type.");
Guy Benyei11169dd2012-12-18 14:30:41 +00002826}
2827
2828void MicrosoftCXXNameMangler::mangleType(const TemplateSpecializationType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002829 Qualifiers, SourceRange Range) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002830 DiagnosticsEngine &Diags = Context.getDiags();
2831 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2832 "cannot mangle this template specialization type yet");
2833 Diags.Report(Range.getBegin(), DiagID)
2834 << Range;
2835}
2836
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002837void MicrosoftCXXNameMangler::mangleType(const DependentNameType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002838 SourceRange Range) {
2839 DiagnosticsEngine &Diags = Context.getDiags();
2840 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2841 "cannot mangle this dependent name type yet");
2842 Diags.Report(Range.getBegin(), DiagID)
2843 << Range;
2844}
2845
2846void MicrosoftCXXNameMangler::mangleType(
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002847 const DependentTemplateSpecializationType *T, Qualifiers,
2848 SourceRange Range) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002849 DiagnosticsEngine &Diags = Context.getDiags();
2850 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2851 "cannot mangle this dependent template specialization type yet");
2852 Diags.Report(Range.getBegin(), DiagID)
2853 << Range;
2854}
2855
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002856void MicrosoftCXXNameMangler::mangleType(const PackExpansionType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002857 SourceRange Range) {
2858 DiagnosticsEngine &Diags = Context.getDiags();
2859 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2860 "cannot mangle this pack expansion yet");
2861 Diags.Report(Range.getBegin(), DiagID)
2862 << Range;
2863}
2864
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002865void MicrosoftCXXNameMangler::mangleType(const TypeOfType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002866 SourceRange Range) {
2867 DiagnosticsEngine &Diags = Context.getDiags();
2868 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2869 "cannot mangle this typeof(type) yet");
2870 Diags.Report(Range.getBegin(), DiagID)
2871 << Range;
2872}
2873
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002874void MicrosoftCXXNameMangler::mangleType(const TypeOfExprType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002875 SourceRange Range) {
2876 DiagnosticsEngine &Diags = Context.getDiags();
2877 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2878 "cannot mangle this typeof(expression) yet");
2879 Diags.Report(Range.getBegin(), DiagID)
2880 << Range;
2881}
2882
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002883void MicrosoftCXXNameMangler::mangleType(const DecltypeType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002884 SourceRange Range) {
2885 DiagnosticsEngine &Diags = Context.getDiags();
2886 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2887 "cannot mangle this decltype() yet");
2888 Diags.Report(Range.getBegin(), DiagID)
2889 << Range;
2890}
2891
2892void MicrosoftCXXNameMangler::mangleType(const UnaryTransformType *T,
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002893 Qualifiers, SourceRange Range) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002894 DiagnosticsEngine &Diags = Context.getDiags();
2895 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2896 "cannot mangle this unary transform type yet");
2897 Diags.Report(Range.getBegin(), DiagID)
2898 << Range;
2899}
2900
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002901void MicrosoftCXXNameMangler::mangleType(const AutoType *T, Qualifiers,
2902 SourceRange Range) {
David Majnemerb9a5f2d2014-01-21 20:33:36 +00002903 assert(T->getDeducedType().isNull() && "expecting a dependent type!");
2904
Guy Benyei11169dd2012-12-18 14:30:41 +00002905 DiagnosticsEngine &Diags = Context.getDiags();
2906 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2907 "cannot mangle this 'auto' type yet");
2908 Diags.Report(Range.getBegin(), DiagID)
2909 << Range;
2910}
2911
Richard Smith600b5262017-01-26 20:40:47 +00002912void MicrosoftCXXNameMangler::mangleType(
2913 const DeducedTemplateSpecializationType *T, Qualifiers, SourceRange Range) {
2914 assert(T->getDeducedType().isNull() && "expecting a dependent type!");
2915
2916 DiagnosticsEngine &Diags = Context.getDiags();
2917 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2918 "cannot mangle this deduced class template specialization type yet");
2919 Diags.Report(Range.getBegin(), DiagID)
2920 << Range;
2921}
2922
David Majnemer7a3ce0c2015-05-26 01:30:45 +00002923void MicrosoftCXXNameMangler::mangleType(const AtomicType *T, Qualifiers,
Guy Benyei11169dd2012-12-18 14:30:41 +00002924 SourceRange Range) {
David Majnemer3421fb62015-12-29 11:46:00 +00002925 QualType ValueType = T->getValueType();
2926
2927 llvm::SmallString<64> TemplateMangling;
2928 llvm::raw_svector_ostream Stream(TemplateMangling);
2929 MicrosoftCXXNameMangler Extra(Context, Stream);
2930 Stream << "?$";
2931 Extra.mangleSourceName("_Atomic");
2932 Extra.mangleType(ValueType, Range, QMM_Escape);
2933
Shoaib Meenaid84cf132018-11-14 19:16:22 +00002934 mangleArtificialTagType(TTK_Struct, TemplateMangling, {"__clang"});
Guy Benyei11169dd2012-12-18 14:30:41 +00002935}
2936
Xiuli Pan9c14e282016-01-09 12:53:17 +00002937void MicrosoftCXXNameMangler::mangleType(const PipeType *T, Qualifiers,
2938 SourceRange Range) {
2939 DiagnosticsEngine &Diags = Context.getDiags();
2940 unsigned DiagID = Diags.getCustomDiagID(DiagnosticsEngine::Error,
2941 "cannot mangle this OpenCL pipe type yet");
2942 Diags.Report(Range.getBegin(), DiagID)
2943 << Range;
2944}
2945
Rafael Espindola002667c2013-10-16 01:40:34 +00002946void MicrosoftMangleContextImpl::mangleCXXName(const NamedDecl *D,
2947 raw_ostream &Out) {
Guy Benyei11169dd2012-12-18 14:30:41 +00002948 assert((isa<FunctionDecl>(D) || isa<VarDecl>(D)) &&
2949 "Invalid mangleName() call, argument is not a variable or function!");
2950 assert(!isa<CXXConstructorDecl>(D) && !isa<CXXDestructorDecl>(D) &&
2951 "Invalid mangleName() call on 'structor decl!");
2952
2953 PrettyStackTraceDecl CrashInfo(D, SourceLocation(),
2954 getASTContext().getSourceManager(),
2955 "Mangling declaration");
2956
David Majnemerfebb2722016-03-07 08:51:17 +00002957 msvc_hashing_ostream MHO(Out);
2958 MicrosoftCXXNameMangler Mangler(*this, MHO);
Guy Benyei11169dd2012-12-18 14:30:41 +00002959 return Mangler.mangle(D);
2960}
Timur Iskhodzhanovdf7e7fb2013-07-30 09:46:19 +00002961
Timur Iskhodzhanov053142a2013-11-06 06:24:31 +00002962// <this-adjustment> ::= <no-adjustment> | <static-adjustment> |
2963// <virtual-adjustment>
2964// <no-adjustment> ::= A # private near
2965// ::= B # private far
2966// ::= I # protected near
2967// ::= J # protected far
2968// ::= Q # public near
2969// ::= R # public far
2970// <static-adjustment> ::= G <static-offset> # private near
2971// ::= H <static-offset> # private far
2972// ::= O <static-offset> # protected near
2973// ::= P <static-offset> # protected far
2974// ::= W <static-offset> # public near
2975// ::= X <static-offset> # public far
2976// <virtual-adjustment> ::= $0 <virtual-shift> <static-offset> # private near
2977// ::= $1 <virtual-shift> <static-offset> # private far
2978// ::= $2 <virtual-shift> <static-offset> # protected near
2979// ::= $3 <virtual-shift> <static-offset> # protected far
2980// ::= $4 <virtual-shift> <static-offset> # public near
2981// ::= $5 <virtual-shift> <static-offset> # public far
2982// <virtual-shift> ::= <vtordisp-shift> | <vtordispex-shift>
2983// <vtordisp-shift> ::= <offset-to-vtordisp>
2984// <vtordispex-shift> ::= <offset-to-vbptr> <vbase-offset-offset>
2985// <offset-to-vtordisp>
Reid Kleckner3ab5a9c2018-12-26 20:07:52 +00002986static void mangleThunkThisAdjustment(AccessSpecifier AS,
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00002987 const ThisAdjustment &Adjustment,
2988 MicrosoftCXXNameMangler &Mangler,
2989 raw_ostream &Out) {
Timur Iskhodzhanov053142a2013-11-06 06:24:31 +00002990 if (!Adjustment.Virtual.isEmpty()) {
2991 Out << '$';
2992 char AccessSpec;
Reid Kleckner3ab5a9c2018-12-26 20:07:52 +00002993 switch (AS) {
Timur Iskhodzhanov053142a2013-11-06 06:24:31 +00002994 case AS_none:
2995 llvm_unreachable("Unsupported access specifier");
2996 case AS_private:
2997 AccessSpec = '0';
2998 break;
2999 case AS_protected:
3000 AccessSpec = '2';
3001 break;
3002 case AS_public:
3003 AccessSpec = '4';
3004 }
3005 if (Adjustment.Virtual.Microsoft.VBPtrOffset) {
3006 Out << 'R' << AccessSpec;
David Majnemer2a816452013-12-09 10:44:32 +00003007 Mangler.mangleNumber(
3008 static_cast<uint32_t>(Adjustment.Virtual.Microsoft.VBPtrOffset));
3009 Mangler.mangleNumber(
3010 static_cast<uint32_t>(Adjustment.Virtual.Microsoft.VBOffsetOffset));
3011 Mangler.mangleNumber(
3012 static_cast<uint32_t>(Adjustment.Virtual.Microsoft.VtordispOffset));
3013 Mangler.mangleNumber(static_cast<uint32_t>(Adjustment.NonVirtual));
Timur Iskhodzhanov053142a2013-11-06 06:24:31 +00003014 } else {
3015 Out << AccessSpec;
David Majnemer2a816452013-12-09 10:44:32 +00003016 Mangler.mangleNumber(
3017 static_cast<uint32_t>(Adjustment.Virtual.Microsoft.VtordispOffset));
3018 Mangler.mangleNumber(-static_cast<uint32_t>(Adjustment.NonVirtual));
Timur Iskhodzhanov053142a2013-11-06 06:24:31 +00003019 }
3020 } else if (Adjustment.NonVirtual != 0) {
Reid Kleckner3ab5a9c2018-12-26 20:07:52 +00003021 switch (AS) {
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00003022 case AS_none:
3023 llvm_unreachable("Unsupported access specifier");
3024 case AS_private:
3025 Out << 'G';
3026 break;
3027 case AS_protected:
3028 Out << 'O';
3029 break;
3030 case AS_public:
3031 Out << 'W';
3032 }
David Majnemer2a816452013-12-09 10:44:32 +00003033 Mangler.mangleNumber(-static_cast<uint32_t>(Adjustment.NonVirtual));
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00003034 } else {
Reid Kleckner3ab5a9c2018-12-26 20:07:52 +00003035 switch (AS) {
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00003036 case AS_none:
3037 llvm_unreachable("Unsupported access specifier");
3038 case AS_private:
3039 Out << 'A';
3040 break;
3041 case AS_protected:
3042 Out << 'I';
3043 break;
3044 case AS_public:
3045 Out << 'Q';
3046 }
3047 }
3048}
3049
Reid Klecknercbec0262018-04-02 20:00:39 +00003050void MicrosoftMangleContextImpl::mangleVirtualMemPtrThunk(
3051 const CXXMethodDecl *MD, const MethodVFTableLocation &ML,
3052 raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003053 msvc_hashing_ostream MHO(Out);
3054 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerfb931542018-03-16 20:36:49 +00003055 Mangler.getStream() << '?';
Reid Kleckner96f8f932014-02-05 17:27:08 +00003056 Mangler.mangleVirtualMemPtrThunk(MD, ML);
Hans Wennborg88497d62013-11-15 17:24:45 +00003057}
3058
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003059void MicrosoftMangleContextImpl::mangleThunk(const CXXMethodDecl *MD,
3060 const ThunkInfo &Thunk,
3061 raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003062 msvc_hashing_ostream MHO(Out);
3063 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerfb931542018-03-16 20:36:49 +00003064 Mangler.getStream() << '?';
Timur Iskhodzhanovdf7e7fb2013-07-30 09:46:19 +00003065 Mangler.mangleName(MD);
Reid Kleckner3ab5a9c2018-12-26 20:07:52 +00003066
3067 // Usually the thunk uses the access specifier of the new method, but if this
3068 // is a covariant return thunk, then MSVC always uses the public access
3069 // specifier, and we do the same.
3070 AccessSpecifier AS = Thunk.Return.isEmpty() ? MD->getAccess() : AS_public;
3071 mangleThunkThisAdjustment(AS, Thunk.This, Mangler, MHO);
3072
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00003073 if (!Thunk.Return.isEmpty())
Craig Topper36250ad2014-05-12 05:36:57 +00003074 assert(Thunk.Method != nullptr &&
3075 "Thunk info should hold the overridee decl");
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00003076
3077 const CXXMethodDecl *DeclForFPT = Thunk.Method ? Thunk.Method : MD;
3078 Mangler.mangleFunctionType(
3079 DeclForFPT->getType()->castAs<FunctionProtoType>(), MD);
Guy Benyei11169dd2012-12-18 14:30:41 +00003080}
Timur Iskhodzhanovdf7e7fb2013-07-30 09:46:19 +00003081
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00003082void MicrosoftMangleContextImpl::mangleCXXDtorThunk(
3083 const CXXDestructorDecl *DD, CXXDtorType Type,
3084 const ThisAdjustment &Adjustment, raw_ostream &Out) {
3085 // FIXME: Actually, the dtor thunk should be emitted for vector deleting
3086 // dtors rather than scalar deleting dtors. Just use the vector deleting dtor
3087 // mangling manually until we support both deleting dtor types.
3088 assert(Type == Dtor_Deleting);
David Majnemerfebb2722016-03-07 08:51:17 +00003089 msvc_hashing_ostream MHO(Out);
3090 MicrosoftCXXNameMangler Mangler(*this, MHO, DD, Type);
Reid Klecknerfb931542018-03-16 20:36:49 +00003091 Mangler.getStream() << "??_E";
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00003092 Mangler.mangleName(DD->getParent());
Reid Kleckner3ab5a9c2018-12-26 20:07:52 +00003093 mangleThunkThisAdjustment(DD->getAccess(), Adjustment, Mangler, MHO);
Timur Iskhodzhanovad9d3b82013-10-09 09:23:58 +00003094 Mangler.mangleFunctionType(DD->getType()->castAs<FunctionProtoType>(), DD);
Guy Benyei11169dd2012-12-18 14:30:41 +00003095}
Reid Kleckner7810af02013-06-19 15:20:38 +00003096
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003097void MicrosoftMangleContextImpl::mangleCXXVFTable(
Timur Iskhodzhanov8b5987e2013-09-27 14:48:01 +00003098 const CXXRecordDecl *Derived, ArrayRef<const CXXRecordDecl *> BasePath,
3099 raw_ostream &Out) {
Reid Kleckner7810af02013-06-19 15:20:38 +00003100 // <mangled-name> ::= ?_7 <class-name> <storage-class>
3101 // <cvr-qualifiers> [<name>] @
Guy Benyei11169dd2012-12-18 14:30:41 +00003102 // NOTE: <cvr-qualifiers> here is always 'B' (const). <storage-class>
Reid Kleckner7810af02013-06-19 15:20:38 +00003103 // is always '6' for vftables.
David Majnemerfebb2722016-03-07 08:51:17 +00003104 msvc_hashing_ostream MHO(Out);
3105 MicrosoftCXXNameMangler Mangler(*this, MHO);
David Majnemer2d8b2002016-02-11 17:49:28 +00003106 if (Derived->hasAttr<DLLImportAttr>())
Reid Klecknerfb931542018-03-16 20:36:49 +00003107 Mangler.getStream() << "??_S";
David Majnemer2d8b2002016-02-11 17:49:28 +00003108 else
Reid Klecknerfb931542018-03-16 20:36:49 +00003109 Mangler.getStream() << "??_7";
Timur Iskhodzhanov8b5987e2013-09-27 14:48:01 +00003110 Mangler.mangleName(Derived);
3111 Mangler.getStream() << "6B"; // '6' for vftable, 'B' for const.
David Majnemerb926dbc2014-04-23 05:16:53 +00003112 for (const CXXRecordDecl *RD : BasePath)
3113 Mangler.mangleName(RD);
Guy Benyei11169dd2012-12-18 14:30:41 +00003114 Mangler.getStream() << '@';
3115}
Reid Kleckner7810af02013-06-19 15:20:38 +00003116
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003117void MicrosoftMangleContextImpl::mangleCXXVBTable(
Reid Kleckner7810af02013-06-19 15:20:38 +00003118 const CXXRecordDecl *Derived, ArrayRef<const CXXRecordDecl *> BasePath,
3119 raw_ostream &Out) {
3120 // <mangled-name> ::= ?_8 <class-name> <storage-class>
3121 // <cvr-qualifiers> [<name>] @
3122 // NOTE: <cvr-qualifiers> here is always 'B' (const). <storage-class>
3123 // is always '7' for vbtables.
David Majnemerfebb2722016-03-07 08:51:17 +00003124 msvc_hashing_ostream MHO(Out);
3125 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerfb931542018-03-16 20:36:49 +00003126 Mangler.getStream() << "??_8";
Reid Kleckner7810af02013-06-19 15:20:38 +00003127 Mangler.mangleName(Derived);
3128 Mangler.getStream() << "7B"; // '7' for vbtable, 'B' for const.
David Majnemerb926dbc2014-04-23 05:16:53 +00003129 for (const CXXRecordDecl *RD : BasePath)
3130 Mangler.mangleName(RD);
Reid Kleckner7810af02013-06-19 15:20:38 +00003131 Mangler.getStream() << '@';
3132}
3133
David Majnemera96b7ee2014-05-13 00:44:44 +00003134void MicrosoftMangleContextImpl::mangleCXXRTTI(QualType T, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003135 msvc_hashing_ostream MHO(Out);
3136 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerfb931542018-03-16 20:36:49 +00003137 Mangler.getStream() << "??_R0";
David Majnemera96b7ee2014-05-13 00:44:44 +00003138 Mangler.mangleType(T, SourceRange(), MicrosoftCXXNameMangler::QMM_Result);
3139 Mangler.getStream() << "@8";
Guy Benyei11169dd2012-12-18 14:30:41 +00003140}
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003141
David Majnemera96b7ee2014-05-13 00:44:44 +00003142void MicrosoftMangleContextImpl::mangleCXXRTTIName(QualType T,
3143 raw_ostream &Out) {
3144 MicrosoftCXXNameMangler Mangler(*this, Out);
3145 Mangler.getStream() << '.';
3146 Mangler.mangleType(T, SourceRange(), MicrosoftCXXNameMangler::QMM_Result);
3147}
3148
David Majnemerc1709d32015-06-23 07:31:11 +00003149void MicrosoftMangleContextImpl::mangleCXXVirtualDisplacementMap(
3150 const CXXRecordDecl *SrcRD, const CXXRecordDecl *DstRD, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003151 msvc_hashing_ostream MHO(Out);
3152 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerfb931542018-03-16 20:36:49 +00003153 Mangler.getStream() << "??_K";
David Majnemerc1709d32015-06-23 07:31:11 +00003154 Mangler.mangleName(SrcRD);
3155 Mangler.getStream() << "$C";
3156 Mangler.mangleName(DstRD);
3157}
3158
David Majnemer526793d2016-07-12 04:42:50 +00003159void MicrosoftMangleContextImpl::mangleCXXThrowInfo(QualType T, bool IsConst,
David Majnemer7c237072015-03-05 00:46:22 +00003160 bool IsVolatile,
David Majnemer526793d2016-07-12 04:42:50 +00003161 bool IsUnaligned,
David Majnemer7c237072015-03-05 00:46:22 +00003162 uint32_t NumEntries,
3163 raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003164 msvc_hashing_ostream MHO(Out);
3165 MicrosoftCXXNameMangler Mangler(*this, MHO);
David Majnemer7c237072015-03-05 00:46:22 +00003166 Mangler.getStream() << "_TI";
3167 if (IsConst)
3168 Mangler.getStream() << 'C';
3169 if (IsVolatile)
3170 Mangler.getStream() << 'V';
David Majnemer526793d2016-07-12 04:42:50 +00003171 if (IsUnaligned)
3172 Mangler.getStream() << 'U';
David Majnemer7c237072015-03-05 00:46:22 +00003173 Mangler.getStream() << NumEntries;
3174 Mangler.mangleType(T, SourceRange(), MicrosoftCXXNameMangler::QMM_Result);
3175}
3176
3177void MicrosoftMangleContextImpl::mangleCXXCatchableTypeArray(
3178 QualType T, uint32_t NumEntries, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003179 msvc_hashing_ostream MHO(Out);
3180 MicrosoftCXXNameMangler Mangler(*this, MHO);
David Majnemer7c237072015-03-05 00:46:22 +00003181 Mangler.getStream() << "_CTA";
3182 Mangler.getStream() << NumEntries;
3183 Mangler.mangleType(T, SourceRange(), MicrosoftCXXNameMangler::QMM_Result);
3184}
3185
David Majnemere7a818f2015-03-06 18:53:55 +00003186void MicrosoftMangleContextImpl::mangleCXXCatchableType(
David Majnemerdfa6d202015-03-11 18:36:39 +00003187 QualType T, const CXXConstructorDecl *CD, CXXCtorType CT, uint32_t Size,
3188 uint32_t NVOffset, int32_t VBPtrOffset, uint32_t VBIndex,
3189 raw_ostream &Out) {
David Majnemer7c237072015-03-05 00:46:22 +00003190 MicrosoftCXXNameMangler Mangler(*this, Out);
David Majnemere7a818f2015-03-06 18:53:55 +00003191 Mangler.getStream() << "_CT";
3192
3193 llvm::SmallString<64> RTTIMangling;
3194 {
3195 llvm::raw_svector_ostream Stream(RTTIMangling);
David Majnemerfebb2722016-03-07 08:51:17 +00003196 msvc_hashing_ostream MHO(Stream);
3197 mangleCXXRTTI(T, MHO);
David Majnemere7a818f2015-03-06 18:53:55 +00003198 }
Reid Klecknerfb931542018-03-16 20:36:49 +00003199 Mangler.getStream() << RTTIMangling;
David Majnemere7a818f2015-03-06 18:53:55 +00003200
Nico Webere3b1f5d2019-05-29 13:48:19 +00003201 // VS2015 and VS2017.1 omit the copy-constructor in the mangled name but
3202 // both older and newer versions include it.
3203 // FIXME: It is known that the Ctor is present in 2013, and in 2017.7
3204 // (_MSC_VER 1914) and newer, and that it's omitted in 2015 and 2017.4
3205 // (_MSC_VER 1911), but it's unknown when exactly it reappeared (1914?
3206 // Or 1912, 1913 aleady?).
3207 bool OmitCopyCtor = getASTContext().getLangOpts().isCompatibleWithMSVC(
3208 LangOptions::MSVC2015) &&
3209 !getASTContext().getLangOpts().isCompatibleWithMSVC(
3210 LangOptions::MSVC2017_7);
David Majnemere7a818f2015-03-06 18:53:55 +00003211 llvm::SmallString<64> CopyCtorMangling;
Nico Webere3b1f5d2019-05-29 13:48:19 +00003212 if (!OmitCopyCtor && CD) {
David Majnemere7a818f2015-03-06 18:53:55 +00003213 llvm::raw_svector_ostream Stream(CopyCtorMangling);
David Majnemerfebb2722016-03-07 08:51:17 +00003214 msvc_hashing_ostream MHO(Stream);
3215 mangleCXXCtor(CD, CT, MHO);
David Majnemere7a818f2015-03-06 18:53:55 +00003216 }
Reid Klecknerfb931542018-03-16 20:36:49 +00003217 Mangler.getStream() << CopyCtorMangling;
David Majnemere7a818f2015-03-06 18:53:55 +00003218
David Majnemer7c237072015-03-05 00:46:22 +00003219 Mangler.getStream() << Size;
David Majnemer999cbf92015-03-10 19:01:51 +00003220 if (VBPtrOffset == -1) {
3221 if (NVOffset) {
3222 Mangler.getStream() << NVOffset;
3223 }
3224 } else {
3225 Mangler.getStream() << NVOffset;
3226 Mangler.getStream() << VBPtrOffset;
3227 Mangler.getStream() << VBIndex;
3228 }
David Majnemer7c237072015-03-05 00:46:22 +00003229}
3230
David Majnemera96b7ee2014-05-13 00:44:44 +00003231void MicrosoftMangleContextImpl::mangleCXXRTTIBaseClassDescriptor(
Warren Hunt5c2b4ea2014-05-23 16:07:43 +00003232 const CXXRecordDecl *Derived, uint32_t NVOffset, int32_t VBPtrOffset,
David Majnemeraeb55c92014-05-13 06:57:43 +00003233 uint32_t VBTableOffset, uint32_t Flags, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003234 msvc_hashing_ostream MHO(Out);
3235 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerfb931542018-03-16 20:36:49 +00003236 Mangler.getStream() << "??_R1";
David Majnemera96b7ee2014-05-13 00:44:44 +00003237 Mangler.mangleNumber(NVOffset);
3238 Mangler.mangleNumber(VBPtrOffset);
3239 Mangler.mangleNumber(VBTableOffset);
3240 Mangler.mangleNumber(Flags);
David Majnemeraeb55c92014-05-13 06:57:43 +00003241 Mangler.mangleName(Derived);
Warren Hunt5c2b4ea2014-05-23 16:07:43 +00003242 Mangler.getStream() << "8";
David Majnemera96b7ee2014-05-13 00:44:44 +00003243}
3244
3245void MicrosoftMangleContextImpl::mangleCXXRTTIBaseClassArray(
3246 const CXXRecordDecl *Derived, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003247 msvc_hashing_ostream MHO(Out);
3248 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerfb931542018-03-16 20:36:49 +00003249 Mangler.getStream() << "??_R2";
David Majnemera96b7ee2014-05-13 00:44:44 +00003250 Mangler.mangleName(Derived);
Warren Hunt5c2b4ea2014-05-23 16:07:43 +00003251 Mangler.getStream() << "8";
David Majnemera96b7ee2014-05-13 00:44:44 +00003252}
3253
3254void MicrosoftMangleContextImpl::mangleCXXRTTIClassHierarchyDescriptor(
3255 const CXXRecordDecl *Derived, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003256 msvc_hashing_ostream MHO(Out);
3257 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerfb931542018-03-16 20:36:49 +00003258 Mangler.getStream() << "??_R3";
David Majnemera96b7ee2014-05-13 00:44:44 +00003259 Mangler.mangleName(Derived);
Warren Hunt5c2b4ea2014-05-23 16:07:43 +00003260 Mangler.getStream() << "8";
David Majnemera96b7ee2014-05-13 00:44:44 +00003261}
3262
3263void MicrosoftMangleContextImpl::mangleCXXRTTICompleteObjectLocator(
David Majnemeraeb55c92014-05-13 06:57:43 +00003264 const CXXRecordDecl *Derived, ArrayRef<const CXXRecordDecl *> BasePath,
3265 raw_ostream &Out) {
3266 // <mangled-name> ::= ?_R4 <class-name> <storage-class>
3267 // <cvr-qualifiers> [<name>] @
3268 // NOTE: <cvr-qualifiers> here is always 'B' (const). <storage-class>
3269 // is always '6' for vftables.
David Majnemerfebb2722016-03-07 08:51:17 +00003270 llvm::SmallString<64> VFTableMangling;
3271 llvm::raw_svector_ostream Stream(VFTableMangling);
3272 mangleCXXVFTable(Derived, BasePath, Stream);
3273
Reid Klecknerfb931542018-03-16 20:36:49 +00003274 if (VFTableMangling.startswith("??@")) {
David Majnemerfebb2722016-03-07 08:51:17 +00003275 assert(VFTableMangling.endswith("@"));
3276 Out << VFTableMangling << "??_R4@";
3277 return;
3278 }
3279
Reid Klecknerfb931542018-03-16 20:36:49 +00003280 assert(VFTableMangling.startswith("??_7") ||
3281 VFTableMangling.startswith("??_S"));
David Majnemerfebb2722016-03-07 08:51:17 +00003282
Reid Klecknerfb931542018-03-16 20:36:49 +00003283 Out << "??_R4" << StringRef(VFTableMangling).drop_front(4);
Guy Benyei11169dd2012-12-18 14:30:41 +00003284}
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003285
Reid Kleckner1d59f992015-01-22 01:36:17 +00003286void MicrosoftMangleContextImpl::mangleSEHFilterExpression(
3287 const NamedDecl *EnclosingDecl, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003288 msvc_hashing_ostream MHO(Out);
3289 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Kleckner1d59f992015-01-22 01:36:17 +00003290 // The function body is in the same comdat as the function with the handler,
3291 // so the numbering here doesn't have to be the same across TUs.
3292 //
3293 // <mangled-name> ::= ?filt$ <filter-number> @0
Reid Klecknerfb931542018-03-16 20:36:49 +00003294 Mangler.getStream() << "?filt$" << SEHFilterIds[EnclosingDecl]++ << "@0@";
Reid Kleckner1d59f992015-01-22 01:36:17 +00003295 Mangler.mangleName(EnclosingDecl);
3296}
3297
Reid Klecknerebaf28d2015-04-14 20:59:00 +00003298void MicrosoftMangleContextImpl::mangleSEHFinallyBlock(
3299 const NamedDecl *EnclosingDecl, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003300 msvc_hashing_ostream MHO(Out);
3301 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerebaf28d2015-04-14 20:59:00 +00003302 // The function body is in the same comdat as the function with the handler,
3303 // so the numbering here doesn't have to be the same across TUs.
3304 //
3305 // <mangled-name> ::= ?fin$ <filter-number> @0
Reid Klecknerfb931542018-03-16 20:36:49 +00003306 Mangler.getStream() << "?fin$" << SEHFinallyIds[EnclosingDecl]++ << "@0@";
Reid Klecknerebaf28d2015-04-14 20:59:00 +00003307 Mangler.mangleName(EnclosingDecl);
3308}
3309
Reid Klecknercc99e262013-11-19 23:23:00 +00003310void MicrosoftMangleContextImpl::mangleTypeName(QualType T, raw_ostream &Out) {
3311 // This is just a made up unique string for the purposes of tbaa. undname
3312 // does *not* know how to demangle it.
3313 MicrosoftCXXNameMangler Mangler(*this, Out);
3314 Mangler.getStream() << '?';
3315 Mangler.mangleType(T, SourceRange());
3316}
3317
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003318void MicrosoftMangleContextImpl::mangleCXXCtor(const CXXConstructorDecl *D,
3319 CXXCtorType Type,
3320 raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003321 msvc_hashing_ostream MHO(Out);
3322 MicrosoftCXXNameMangler mangler(*this, MHO, D, Type);
Guy Benyei11169dd2012-12-18 14:30:41 +00003323 mangler.mangle(D);
3324}
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003325
3326void MicrosoftMangleContextImpl::mangleCXXDtor(const CXXDestructorDecl *D,
3327 CXXDtorType Type,
3328 raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003329 msvc_hashing_ostream MHO(Out);
3330 MicrosoftCXXNameMangler mangler(*this, MHO, D, Type);
Guy Benyei11169dd2012-12-18 14:30:41 +00003331 mangler.mangle(D);
3332}
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003333
David Majnemere77de752015-12-29 22:25:14 +00003334void MicrosoftMangleContextImpl::mangleReferenceTemporary(
3335 const VarDecl *VD, unsigned ManglingNumber, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003336 msvc_hashing_ostream MHO(Out);
3337 MicrosoftCXXNameMangler Mangler(*this, MHO);
David Majnemere77de752015-12-29 22:25:14 +00003338
Reid Klecknerfb931542018-03-16 20:36:49 +00003339 Mangler.getStream() << "?$RT" << ManglingNumber << '@';
David Majnemere77de752015-12-29 22:25:14 +00003340 Mangler.mangle(VD, "");
Guy Benyei11169dd2012-12-18 14:30:41 +00003341}
3342
David Majnemer8354eee2015-05-07 06:15:46 +00003343void MicrosoftMangleContextImpl::mangleThreadSafeStaticGuardVariable(
3344 const VarDecl *VD, unsigned GuardNum, raw_ostream &Out) {
David Majnemerfebb2722016-03-07 08:51:17 +00003345 msvc_hashing_ostream MHO(Out);
3346 MicrosoftCXXNameMangler Mangler(*this, MHO);
David Majnemer8354eee2015-05-07 06:15:46 +00003347
Reid Klecknerfb931542018-03-16 20:36:49 +00003348 Mangler.getStream() << "?$TSS" << GuardNum << '@';
David Majnemer8354eee2015-05-07 06:15:46 +00003349 Mangler.mangleNestedName(VD);
Dmitry Polukhin5fa681e2016-05-27 08:52:34 +00003350 Mangler.getStream() << "@4HA";
David Majnemer8354eee2015-05-07 06:15:46 +00003351}
3352
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003353void MicrosoftMangleContextImpl::mangleStaticGuardVariable(const VarDecl *VD,
3354 raw_ostream &Out) {
David Majnemer2206bf52014-03-05 08:57:59 +00003355 // <guard-name> ::= ?_B <postfix> @5 <scope-depth>
David Majnemerec8e54b2015-05-07 21:19:06 +00003356 // ::= ?__J <postfix> @5 <scope-depth>
Reid Klecknerd8110b62013-09-10 20:14:30 +00003357 // ::= ?$S <guard-num> @ <postfix> @4IA
3358
3359 // The first mangling is what MSVC uses to guard static locals in inline
3360 // functions. It uses a different mangling in external functions to support
3361 // guarding more than 32 variables. MSVC rejects inline functions with more
3362 // than 32 static locals. We don't fully implement the second mangling
3363 // because those guards are not externally visible, and instead use LLVM's
3364 // default renaming when creating a new guard variable.
David Majnemerfebb2722016-03-07 08:51:17 +00003365 msvc_hashing_ostream MHO(Out);
3366 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerd8110b62013-09-10 20:14:30 +00003367
3368 bool Visible = VD->isExternallyVisible();
David Majnemer8354eee2015-05-07 06:15:46 +00003369 if (Visible) {
Reid Klecknerfb931542018-03-16 20:36:49 +00003370 Mangler.getStream() << (VD->getTLSKind() ? "??__J" : "??_B");
David Majnemer8354eee2015-05-07 06:15:46 +00003371 } else {
Reid Klecknerfb931542018-03-16 20:36:49 +00003372 Mangler.getStream() << "?$S1@";
David Majnemer8354eee2015-05-07 06:15:46 +00003373 }
David Majnemer34b49892014-03-06 19:57:36 +00003374 unsigned ScopeDepth = 0;
3375 if (Visible && !getNextDiscriminator(VD, ScopeDepth))
3376 // If we do not have a discriminator and are emitting a guard variable for
3377 // use at global scope, then mangling the nested name will not be enough to
3378 // remove ambiguities.
3379 Mangler.mangle(VD, "");
3380 else
3381 Mangler.mangleNestedName(VD);
David Majnemer2206bf52014-03-05 08:57:59 +00003382 Mangler.getStream() << (Visible ? "@5" : "@4IA");
David Majnemer34b49892014-03-06 19:57:36 +00003383 if (ScopeDepth)
David Majnemer2206bf52014-03-05 08:57:59 +00003384 Mangler.mangleNumber(ScopeDepth);
Reid Klecknerd8110b62013-09-10 20:14:30 +00003385}
3386
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003387void MicrosoftMangleContextImpl::mangleInitFiniStub(const VarDecl *D,
David Majnemerfebb2722016-03-07 08:51:17 +00003388 char CharCode,
3389 raw_ostream &Out) {
3390 msvc_hashing_ostream MHO(Out);
3391 MicrosoftCXXNameMangler Mangler(*this, MHO);
Reid Klecknerfb931542018-03-16 20:36:49 +00003392 Mangler.getStream() << "??__" << CharCode;
David Majnemerf55feec2014-03-06 19:10:27 +00003393 if (D->isStaticDataMember()) {
Zachary Turnerf4f767d2018-08-30 20:53:11 +00003394 Mangler.getStream() << '?';
3395 Mangler.mangleName(D);
David Majnemerf55feec2014-03-06 19:10:27 +00003396 Mangler.mangleVariableEncoding(D);
Zachary Turnerf4f767d2018-08-30 20:53:11 +00003397 Mangler.getStream() << "@@";
3398 } else {
3399 Mangler.mangleName(D);
David Majnemerf55feec2014-03-06 19:10:27 +00003400 }
Reid Kleckner1ece9fc2013-09-10 20:43:12 +00003401 // This is the function class mangling. These stubs are global, non-variadic,
3402 // cdecl functions that return void and take no args.
3403 Mangler.getStream() << "YAXXZ";
3404}
3405
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003406void MicrosoftMangleContextImpl::mangleDynamicInitializer(const VarDecl *D,
3407 raw_ostream &Out) {
Reid Kleckner1ece9fc2013-09-10 20:43:12 +00003408 // <initializer-name> ::= ?__E <name> YAXXZ
David Majnemerfebb2722016-03-07 08:51:17 +00003409 mangleInitFiniStub(D, 'E', Out);
Reid Kleckner1ece9fc2013-09-10 20:43:12 +00003410}
3411
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003412void
3413MicrosoftMangleContextImpl::mangleDynamicAtExitDestructor(const VarDecl *D,
3414 raw_ostream &Out) {
Reid Kleckner1ece9fc2013-09-10 20:43:12 +00003415 // <destructor-name> ::= ?__F <name> YAXXZ
David Majnemerfebb2722016-03-07 08:51:17 +00003416 mangleInitFiniStub(D, 'F', Out);
Reid Klecknerd8110b62013-09-10 20:14:30 +00003417}
3418
David Majnemer58e5bee2014-03-24 21:43:36 +00003419void MicrosoftMangleContextImpl::mangleStringLiteral(const StringLiteral *SL,
3420 raw_ostream &Out) {
Nico Webera2d9c2b2018-07-02 12:31:20 +00003421 // <char-type> ::= 0 # char, char16_t, char32_t
3422 // # (little endian char data in mangling)
3423 // ::= 1 # wchar_t (big endian char data in mangling)
David Majnemer58e5bee2014-03-24 21:43:36 +00003424 //
3425 // <literal-length> ::= <non-negative integer> # the length of the literal
3426 //
3427 // <encoded-crc> ::= <hex digit>+ @ # crc of the literal including
Hans Wennborg7525edc2018-07-06 06:54:16 +00003428 // # trailing null bytes
David Majnemer58e5bee2014-03-24 21:43:36 +00003429 //
3430 // <encoded-string> ::= <simple character> # uninteresting character
3431 // ::= '?$' <hex digit> <hex digit> # these two nibbles
3432 // # encode the byte for the
3433 // # character
3434 // ::= '?' [a-z] # \xe1 - \xfa
3435 // ::= '?' [A-Z] # \xc1 - \xda
3436 // ::= '?' [0-9] # [,/\:. \n\t'-]
3437 //
3438 // <literal> ::= '??_C@_' <char-type> <literal-length> <encoded-crc>
3439 // <encoded-string> '@'
3440 MicrosoftCXXNameMangler Mangler(*this, Out);
Reid Klecknerfb931542018-03-16 20:36:49 +00003441 Mangler.getStream() << "??_C@_";
David Majnemer58e5bee2014-03-24 21:43:36 +00003442
Hans Wennborg7525edc2018-07-06 06:54:16 +00003443 // The actual string length might be different from that of the string literal
3444 // in cases like:
3445 // char foo[3] = "foobar";
3446 // char bar[42] = "foobar";
3447 // Where it is truncated or zero-padded to fit the array. This is the length
3448 // used for mangling, and any trailing null-bytes also need to be mangled.
3449 unsigned StringLength = getASTContext()
3450 .getAsConstantArrayType(SL->getType())
3451 ->getSize()
3452 .getZExtValue();
3453 unsigned StringByteLength = StringLength * SL->getCharByteWidth();
3454
David Majnemer58e5bee2014-03-24 21:43:36 +00003455 // <char-type>: The "kind" of string literal is encoded into the mangled name.
David Majnemer98e77a22014-11-21 19:57:25 +00003456 if (SL->isWide())
David Majnemer58e5bee2014-03-24 21:43:36 +00003457 Mangler.getStream() << '1';
3458 else
David Majnemer98e77a22014-11-21 19:57:25 +00003459 Mangler.getStream() << '0';
David Majnemer58e5bee2014-03-24 21:43:36 +00003460
3461 // <literal-length>: The next part of the mangled name consists of the length
Hans Wennborg7525edc2018-07-06 06:54:16 +00003462 // of the string in bytes.
3463 Mangler.mangleNumber(StringByteLength);
David Majnemer58e5bee2014-03-24 21:43:36 +00003464
Malcolm Parsonsc6e45832017-01-13 18:55:32 +00003465 auto GetLittleEndianByte = [&SL](unsigned Index) {
David Majnemer4b293eb2014-03-31 17:18:53 +00003466 unsigned CharByteWidth = SL->getCharByteWidth();
Hans Wennborg7525edc2018-07-06 06:54:16 +00003467 if (Index / CharByteWidth >= SL->getLength())
3468 return static_cast<char>(0);
David Majnemer4b293eb2014-03-31 17:18:53 +00003469 uint32_t CodeUnit = SL->getCodeUnit(Index / CharByteWidth);
David Majnemere49c4b62014-03-31 21:46:05 +00003470 unsigned OffsetInCodeUnit = Index % CharByteWidth;
3471 return static_cast<char>((CodeUnit >> (8 * OffsetInCodeUnit)) & 0xff);
David Majnemer4b293eb2014-03-31 17:18:53 +00003472 };
3473
Malcolm Parsonsc6e45832017-01-13 18:55:32 +00003474 auto GetBigEndianByte = [&SL](unsigned Index) {
David Majnemer4b293eb2014-03-31 17:18:53 +00003475 unsigned CharByteWidth = SL->getCharByteWidth();
Hans Wennborg7525edc2018-07-06 06:54:16 +00003476 if (Index / CharByteWidth >= SL->getLength())
3477 return static_cast<char>(0);
David Majnemer4b293eb2014-03-31 17:18:53 +00003478 uint32_t CodeUnit = SL->getCodeUnit(Index / CharByteWidth);
David Majnemere49c4b62014-03-31 21:46:05 +00003479 unsigned OffsetInCodeUnit = (CharByteWidth - 1) - (Index % CharByteWidth);
3480 return static_cast<char>((CodeUnit >> (8 * OffsetInCodeUnit)) & 0xff);
David Majnemer4b293eb2014-03-31 17:18:53 +00003481 };
3482
David Majnemer58e5bee2014-03-24 21:43:36 +00003483 // CRC all the bytes of the StringLiteral.
David Majnemerb8ed3642015-09-01 21:24:07 +00003484 llvm::JamCRC JC;
Hans Wennborg7525edc2018-07-06 06:54:16 +00003485 for (unsigned I = 0, E = StringByteLength; I != E; ++I)
David Majnemerb8ed3642015-09-01 21:24:07 +00003486 JC.update(GetLittleEndianByte(I));
David Majnemer58e5bee2014-03-24 21:43:36 +00003487
David Majnemer58e5bee2014-03-24 21:43:36 +00003488 // <encoded-crc>: The CRC is encoded utilizing the standard number mangling
3489 // scheme.
David Majnemerb8ed3642015-09-01 21:24:07 +00003490 Mangler.mangleNumber(JC.getCRC());
David Majnemer58e5bee2014-03-24 21:43:36 +00003491
Nico Webera2d9c2b2018-07-02 12:31:20 +00003492 // <encoded-string>: The mangled name also contains the first 32 bytes
3493 // (including null-terminator bytes) of the encoded StringLiteral.
David Majnemer58e5bee2014-03-24 21:43:36 +00003494 // Each character is encoded by splitting them into bytes and then encoding
3495 // the constituent bytes.
3496 auto MangleByte = [&Mangler](char Byte) {
3497 // There are five different manglings for characters:
3498 // - [a-zA-Z0-9_$]: A one-to-one mapping.
3499 // - ?[a-z]: The range from \xe1 to \xfa.
3500 // - ?[A-Z]: The range from \xc1 to \xda.
3501 // - ?[0-9]: The set of [,/\:. \n\t'-].
3502 // - ?$XX: A fallback which maps nibbles.
David Majnemer10befcf2014-04-01 00:05:57 +00003503 if (isIdentifierBody(Byte, /*AllowDollar=*/true)) {
David Majnemer58e5bee2014-03-24 21:43:36 +00003504 Mangler.getStream() << Byte;
David Majnemer10befcf2014-04-01 00:05:57 +00003505 } else if (isLetter(Byte & 0x7f)) {
3506 Mangler.getStream() << '?' << static_cast<char>(Byte & 0x7f);
David Majnemer58e5bee2014-03-24 21:43:36 +00003507 } else {
David Majnemer07bc6112014-12-22 06:24:49 +00003508 const char SpecialChars[] = {',', '/', '\\', ':', '.',
3509 ' ', '\n', '\t', '\'', '-'};
Fangrui Song75e74e02019-03-31 08:48:19 +00003510 const char *Pos = llvm::find(SpecialChars, Byte);
David Majnemer07bc6112014-12-22 06:24:49 +00003511 if (Pos != std::end(SpecialChars)) {
3512 Mangler.getStream() << '?' << (Pos - std::begin(SpecialChars));
3513 } else {
3514 Mangler.getStream() << "?$";
3515 Mangler.getStream() << static_cast<char>('A' + ((Byte >> 4) & 0xf));
3516 Mangler.getStream() << static_cast<char>('A' + (Byte & 0xf));
David Majnemer58e5bee2014-03-24 21:43:36 +00003517 }
3518 }
3519 };
3520
Nico Webera2d9c2b2018-07-02 12:31:20 +00003521 // Enforce our 32 bytes max, except wchar_t which gets 32 chars instead.
3522 unsigned MaxBytesToMangle = SL->isWide() ? 64U : 32U;
Hans Wennborg7525edc2018-07-06 06:54:16 +00003523 unsigned NumBytesToMangle = std::min(MaxBytesToMangle, StringByteLength);
3524 for (unsigned I = 0; I != NumBytesToMangle; ++I) {
David Majnemer98e77a22014-11-21 19:57:25 +00003525 if (SL->isWide())
3526 MangleByte(GetBigEndianByte(I));
3527 else
3528 MangleByte(GetLittleEndianByte(I));
Hans Wennborg7525edc2018-07-06 06:54:16 +00003529 }
David Majnemer58e5bee2014-03-24 21:43:36 +00003530
3531 Mangler.getStream() << '@';
3532}
3533
Timur Iskhodzhanov67455222013-10-03 06:26:13 +00003534MicrosoftMangleContext *
3535MicrosoftMangleContext::create(ASTContext &Context, DiagnosticsEngine &Diags) {
3536 return new MicrosoftMangleContextImpl(Context, Diags);
Guy Benyei11169dd2012-12-18 14:30:41 +00003537}