blob: 11bbaae01836ab2511c57ce1df29675863fda6f9 [file] [log] [blame]
Douglas Gregor6ec36682009-02-18 23:53:56 +00001//===--- Mangle.cpp - Mangle C++ Names --------------------------*- C++ -*-===//
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// Implements C++ name mangling according to the Itanium C++ ABI,
11// which is used in GCC 3.2 and newer (and many compilers that are
12// ABI-compatible with GCC):
13//
14// http://www.codesourcery.com/public/cxx-abi/abi.html
15//
16//===----------------------------------------------------------------------===//
17#include "Mangle.h"
18#include "clang/AST/ASTContext.h"
19#include "clang/AST/Decl.h"
20#include "clang/AST/DeclCXX.h"
Anders Carlssona40c5e42009-03-07 22:03:21 +000021#include "clang/AST/DeclObjC.h"
Anders Carlsson7a0ba872009-05-15 16:09:15 +000022#include "clang/AST/DeclTemplate.h"
Anders Carlsson50755b02009-09-27 20:11:34 +000023#include "clang/AST/ExprCXX.h"
Douglas Gregor6ec36682009-02-18 23:53:56 +000024#include "clang/Basic/SourceManager.h"
Anders Carlssonc4355b62009-10-07 01:45:02 +000025#include "llvm/ADT/StringExtras.h"
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000026#include "llvm/Support/Compiler.h"
27#include "llvm/Support/raw_ostream.h"
John McCallefe6aee2009-09-05 07:56:18 +000028#include "llvm/Support/ErrorHandling.h"
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000029using namespace clang;
30
31namespace {
32 class VISIBILITY_HIDDEN CXXNameMangler {
Anders Carlssonb5404912009-10-07 01:06:45 +000033 MangleContext &Context;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000034 llvm::raw_ostream &Out;
35
Anders Carlsson27ae5362009-04-17 01:58:57 +000036 const CXXMethodDecl *Structor;
37 unsigned StructorType;
Anders Carlsson3ac86b52009-04-15 05:36:58 +000038 CXXCtorType CtorType;
Mike Stump1eb44332009-09-09 15:08:12 +000039
Anders Carlsson76967372009-09-17 00:43:46 +000040 llvm::DenseMap<uintptr_t, unsigned> Substitutions;
41
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000042 public:
Anders Carlssonb5404912009-10-07 01:06:45 +000043 CXXNameMangler(MangleContext &C, llvm::raw_ostream &os)
Anders Carlsson27ae5362009-04-17 01:58:57 +000044 : Context(C), Out(os), Structor(0), StructorType(0) { }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000045
46 bool mangle(const NamedDecl *D);
Mike Stump77ca8f62009-09-05 07:20:32 +000047 void mangleCalloffset(int64_t nv, int64_t v);
Mike Stumpdec025b2009-09-07 04:27:52 +000048 void mangleThunk(const FunctionDecl *FD, int64_t nv, int64_t v);
49 void mangleCovariantThunk(const FunctionDecl *FD,
Mike Stump77ca8f62009-09-05 07:20:32 +000050 int64_t nv_t, int64_t v_t,
Mike Stump9124bcc2009-09-02 00:56:18 +000051 int64_t nv_r, int64_t v_r);
Anders Carlsson41aa8c12009-04-13 18:02:10 +000052 void mangleGuardVariable(const VarDecl *D);
Mike Stump1eb44332009-09-09 15:08:12 +000053
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +000054 void mangleCXXVtable(const CXXRecordDecl *RD);
Mike Stump88e91712009-10-28 01:51:46 +000055 void mangleCXXVTT(const CXXRecordDecl *RD);
Anders Carlssond313e402009-10-30 01:52:02 +000056 void mangleCXXRtti(QualType Ty);
Anders Carlsson3ac86b52009-04-15 05:36:58 +000057 void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type);
Anders Carlsson27ae5362009-04-17 01:58:57 +000058 void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type);
Anders Carlsson3ac86b52009-04-15 05:36:58 +000059
Anders Carlsson43f17402009-04-02 15:51:53 +000060 private:
Anders Carlsson6862fc72009-09-17 04:16:28 +000061 bool mangleSubstitution(const NamedDecl *ND);
Anders Carlsson76967372009-09-17 00:43:46 +000062 bool mangleSubstitution(QualType T);
Anders Carlssond3a932a2009-09-17 03:53:28 +000063 bool mangleSubstitution(uintptr_t Ptr);
64
Anders Carlssone7c8cb62009-09-26 20:53:44 +000065 bool mangleStandardSubstitution(const NamedDecl *ND);
66
Anders Carlssond3a932a2009-09-17 03:53:28 +000067 void addSubstitution(const NamedDecl *ND) {
68 addSubstitution(reinterpret_cast<uintptr_t>(ND));
69 }
Anders Carlsson76967372009-09-17 00:43:46 +000070 void addSubstitution(QualType T);
Anders Carlssond3a932a2009-09-17 03:53:28 +000071 void addSubstitution(uintptr_t Ptr);
Anders Carlsson76967372009-09-17 00:43:46 +000072
Anders Carlsson43f17402009-04-02 15:51:53 +000073 bool mangleFunctionDecl(const FunctionDecl *FD);
Mike Stump1eb44332009-09-09 15:08:12 +000074
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000075 void mangleFunctionEncoding(const FunctionDecl *FD);
76 void mangleName(const NamedDecl *ND);
Anders Carlsson7624f212009-09-18 02:42:01 +000077 void mangleName(const TemplateDecl *TD,
78 const TemplateArgument *TemplateArgs,
79 unsigned NumTemplateArgs);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000080 void mangleUnqualifiedName(const NamedDecl *ND);
Anders Carlsson201ce742009-09-17 03:17:01 +000081 void mangleUnscopedName(const NamedDecl *ND);
Anders Carlsson0fa6df42009-09-26 19:45:45 +000082 void mangleUnscopedTemplateName(const TemplateDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000083 void mangleSourceName(const IdentifierInfo *II);
Anders Carlsson1b42c792009-04-02 16:24:45 +000084 void mangleLocalName(const NamedDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000085 void mangleNestedName(const NamedDecl *ND);
Anders Carlsson7624f212009-09-18 02:42:01 +000086 void mangleNestedName(const TemplateDecl *TD,
87 const TemplateArgument *TemplateArgs,
88 unsigned NumTemplateArgs);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000089 void manglePrefix(const DeclContext *DC);
Anders Carlsson0fa6df42009-09-26 19:45:45 +000090 void mangleTemplatePrefix(const TemplateDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000091 void mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity);
John McCall0953e762009-09-24 19:53:00 +000092 void mangleQualifiers(Qualifiers Quals);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000093 void mangleType(QualType T);
John McCallefe6aee2009-09-05 07:56:18 +000094
95 // Declare manglers for every type class.
96#define ABSTRACT_TYPE(CLASS, PARENT)
97#define NON_CANONICAL_TYPE(CLASS, PARENT)
98#define TYPE(CLASS, PARENT) void mangleType(const CLASS##Type *T);
99#include "clang/AST/TypeNodes.def"
100
101 void mangleType(const TagType*);
102 void mangleBareFunctionType(const FunctionType *T,
103 bool MangleReturnType);
Anders Carlssond553f8c2009-09-21 01:21:10 +0000104 void mangleExpression(const Expr *E);
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000105 void mangleCXXCtorType(CXXCtorType T);
Anders Carlsson27ae5362009-04-17 01:58:57 +0000106 void mangleCXXDtorType(CXXDtorType T);
Mike Stump1eb44332009-09-09 15:08:12 +0000107
Anders Carlsson7624f212009-09-18 02:42:01 +0000108 void mangleTemplateArgs(const TemplateArgument *TemplateArgs,
109 unsigned NumTemplateArgs);
Anders Carlsson068f3472009-09-17 05:31:47 +0000110 void mangleTemplateArgumentList(const TemplateArgumentList &L);
111 void mangleTemplateArgument(const TemplateArgument &A);
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +0000112
113 void mangleTemplateParameter(unsigned Index);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000114 };
115}
116
Anders Carlsson43f17402009-04-02 15:51:53 +0000117static bool isInCLinkageSpecification(const Decl *D) {
Douglas Gregor457e2812009-10-28 16:31:34 +0000118 D = D->getCanonicalDecl();
Mike Stump1eb44332009-09-09 15:08:12 +0000119 for (const DeclContext *DC = D->getDeclContext();
Anders Carlsson43f17402009-04-02 15:51:53 +0000120 !DC->isTranslationUnit(); DC = DC->getParent()) {
Mike Stump1eb44332009-09-09 15:08:12 +0000121 if (const LinkageSpecDecl *Linkage = dyn_cast<LinkageSpecDecl>(DC))
Anders Carlsson43f17402009-04-02 15:51:53 +0000122 return Linkage->getLanguage() == LinkageSpecDecl::lang_c;
123 }
Mike Stump1eb44332009-09-09 15:08:12 +0000124
Anders Carlsson43f17402009-04-02 15:51:53 +0000125 return false;
126}
127
128bool CXXNameMangler::mangleFunctionDecl(const FunctionDecl *FD) {
Mike Stump141c5af2009-09-02 00:25:38 +0000129 // Clang's "overloadable" attribute extension to C/C++ implies name mangling
130 // (always).
Argyrios Kyrtzidis40b598e2009-06-30 02:34:44 +0000131 if (!FD->hasAttr<OverloadableAttr>()) {
Chris Lattner783601d2009-06-13 23:34:16 +0000132 // C functions are not mangled, and "main" is never mangled.
Anders Carlssonb5404912009-10-07 01:06:45 +0000133 if (!Context.getASTContext().getLangOptions().CPlusPlus || FD->isMain())
Chris Lattner783601d2009-06-13 23:34:16 +0000134 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000135
136 // No mangling in an "implicit extern C" header.
Chris Lattner783601d2009-06-13 23:34:16 +0000137 if (FD->getLocation().isValid() &&
Anders Carlssonb5404912009-10-07 01:06:45 +0000138 Context.getASTContext().getSourceManager().
139 isInExternCSystemHeader(FD->getLocation()))
Chris Lattner783601d2009-06-13 23:34:16 +0000140 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000141
Chris Lattner783601d2009-06-13 23:34:16 +0000142 // No name mangling in a C linkage specification.
Anders Carlssonadd28822009-09-22 20:33:31 +0000143 if (!isa<CXXMethodDecl>(FD) && isInCLinkageSpecification(FD))
Chris Lattner783601d2009-06-13 23:34:16 +0000144 return false;
145 }
Anders Carlsson43f17402009-04-02 15:51:53 +0000146
147 // If we get here, mangle the decl name!
148 Out << "_Z";
149 mangleFunctionEncoding(FD);
150 return true;
151}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000152
153bool CXXNameMangler::mangle(const NamedDecl *D) {
Mike Stump141c5af2009-09-02 00:25:38 +0000154 // Any decl can be declared with __asm("foo") on it, and this takes precedence
155 // over all other naming in the .o file.
Argyrios Kyrtzidis40b598e2009-06-30 02:34:44 +0000156 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000157 // If we have an asm name, then we use it as the mangling.
158 Out << '\01'; // LLVM IR Marker for __asm("foo")
159 Out << ALA->getLabel();
160 return true;
161 }
Mike Stump1eb44332009-09-09 15:08:12 +0000162
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000163 // <mangled-name> ::= _Z <encoding>
164 // ::= <data name>
165 // ::= <special-name>
166
167 // FIXME: Actually use a visitor to decode these?
Anders Carlsson43f17402009-04-02 15:51:53 +0000168 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
169 return mangleFunctionDecl(FD);
Mike Stump1eb44332009-09-09 15:08:12 +0000170
Anders Carlsson329749c2009-04-02 16:05:20 +0000171 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Anders Carlssonb5404912009-10-07 01:06:45 +0000172 if (!Context.getASTContext().getLangOptions().CPlusPlus ||
Anders Carlsson9ccb0652009-04-11 01:19:45 +0000173 isInCLinkageSpecification(D) ||
174 D->getDeclContext()->isTranslationUnit())
Anders Carlsson329749c2009-04-02 16:05:20 +0000175 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000176
Anders Carlsson329749c2009-04-02 16:05:20 +0000177 Out << "_Z";
178 mangleName(VD);
179 return true;
180 }
Mike Stump1eb44332009-09-09 15:08:12 +0000181
Anders Carlsson43f17402009-04-02 15:51:53 +0000182 return false;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000183}
184
Mike Stump1eb44332009-09-09 15:08:12 +0000185void CXXNameMangler::mangleCXXCtor(const CXXConstructorDecl *D,
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000186 CXXCtorType Type) {
Anders Carlsson27ae5362009-04-17 01:58:57 +0000187 assert(!Structor && "Structor already set!");
188 Structor = D;
189 StructorType = Type;
Mike Stump1eb44332009-09-09 15:08:12 +0000190
Anders Carlsson27ae5362009-04-17 01:58:57 +0000191 mangle(D);
192}
193
Mike Stump1eb44332009-09-09 15:08:12 +0000194void CXXNameMangler::mangleCXXDtor(const CXXDestructorDecl *D,
Anders Carlsson27ae5362009-04-17 01:58:57 +0000195 CXXDtorType Type) {
196 assert(!Structor && "Structor already set!");
197 Structor = D;
198 StructorType = Type;
Mike Stump1eb44332009-09-09 15:08:12 +0000199
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000200 mangle(D);
201}
202
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +0000203void CXXNameMangler::mangleCXXVtable(const CXXRecordDecl *RD) {
Mike Stumpf1216772009-07-31 18:25:34 +0000204 // <special-name> ::= TV <type> # virtual table
205 Out << "_ZTV";
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +0000206 mangleName(RD);
Mike Stumpf1216772009-07-31 18:25:34 +0000207}
208
Mike Stump88e91712009-10-28 01:51:46 +0000209void CXXNameMangler::mangleCXXVTT(const CXXRecordDecl *RD) {
210 // <special-name> ::= TT <type> # VTT structure
211 Out << "_ZTT";
212 mangleName(RD);
213}
214
Anders Carlssond313e402009-10-30 01:52:02 +0000215void CXXNameMangler::mangleCXXRtti(QualType Ty) {
Mike Stump738f8c22009-07-31 23:15:31 +0000216 // <special-name> ::= TI <type> # typeinfo structure
217 Out << "_ZTI";
Anders Carlsson731f7f92009-10-30 01:26:12 +0000218
Anders Carlssond313e402009-10-30 01:52:02 +0000219 mangleType(Ty);
Mike Stump738f8c22009-07-31 23:15:31 +0000220}
221
Mike Stump1eb44332009-09-09 15:08:12 +0000222void CXXNameMangler::mangleGuardVariable(const VarDecl *D) {
223 // <special-name> ::= GV <object name> # Guard variable for one-time
Mike Stumpf1216772009-07-31 18:25:34 +0000224 // # initialization
Anders Carlsson41aa8c12009-04-13 18:02:10 +0000225
226 Out << "_ZGV";
227 mangleName(D);
228}
229
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000230void CXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD) {
231 // <encoding> ::= <function name> <bare-function-type>
232 mangleName(FD);
Mike Stump1eb44332009-09-09 15:08:12 +0000233
Mike Stump141c5af2009-09-02 00:25:38 +0000234 // Whether the mangling of a function type includes the return type depends on
235 // the context and the nature of the function. The rules for deciding whether
236 // the return type is included are:
Mike Stump1eb44332009-09-09 15:08:12 +0000237 //
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000238 // 1. Template functions (names or types) have return types encoded, with
239 // the exceptions listed below.
Mike Stump1eb44332009-09-09 15:08:12 +0000240 // 2. Function types not appearing as part of a function name mangling,
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000241 // e.g. parameters, pointer types, etc., have return type encoded, with the
242 // exceptions listed below.
243 // 3. Non-template function names do not have return types encoded.
244 //
Mike Stump141c5af2009-09-02 00:25:38 +0000245 // The exceptions mentioned in (1) and (2) above, for which the return type is
246 // never included, are
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000247 // 1. Constructors.
248 // 2. Destructors.
249 // 3. Conversion operator functions, e.g. operator int.
250 bool MangleReturnType = false;
Anders Carlsson9234b7f2009-09-17 03:46:43 +0000251 if (FunctionTemplateDecl *PrimaryTemplate = FD->getPrimaryTemplate()) {
252 if (!(isa<CXXConstructorDecl>(FD) || isa<CXXDestructorDecl>(FD) ||
253 isa<CXXConversionDecl>(FD)))
254 MangleReturnType = true;
255
256 // Mangle the type of the primary template.
257 FD = PrimaryTemplate->getTemplatedDecl();
258 }
259
John McCall54e14c42009-10-22 22:37:11 +0000260 // Do the canonicalization out here because parameter types can
261 // undergo additional canonicalization (e.g. array decay).
262 FunctionType *FT = cast<FunctionType>(Context.getASTContext()
263 .getCanonicalType(FD->getType()));
264
265 mangleBareFunctionType(FT, MangleReturnType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000266}
267
268static bool isStdNamespace(const DeclContext *DC) {
269 if (!DC->isNamespace() || !DC->getParent()->isTranslationUnit())
270 return false;
271
272 const NamespaceDecl *NS = cast<NamespaceDecl>(DC);
John McCall9aeed322009-10-01 00:25:31 +0000273 const IdentifierInfo *II = NS->getOriginalNamespace()->getIdentifier();
274 return II && II->isStr("std");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000275}
276
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000277static const TemplateDecl *
278isTemplate(const NamedDecl *ND, const TemplateArgumentList *&TemplateArgs) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000279 // Check if we have a function template.
280 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)){
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000281 if (const TemplateDecl *TD = FD->getPrimaryTemplate()) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000282 TemplateArgs = FD->getTemplateSpecializationArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000283 return TD;
Anders Carlsson2744a062009-09-18 19:00:18 +0000284 }
285 }
286
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000287 // Check if we have a class template.
288 if (const ClassTemplateSpecializationDecl *Spec =
289 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
290 TemplateArgs = &Spec->getTemplateArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000291 return Spec->getSpecializedTemplate();
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000292 }
293
Anders Carlsson2744a062009-09-18 19:00:18 +0000294 return 0;
295}
296
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000297void CXXNameMangler::mangleName(const NamedDecl *ND) {
298 // <name> ::= <nested-name>
299 // ::= <unscoped-name>
300 // ::= <unscoped-template-name> <template-args>
Anders Carlsson201ce742009-09-17 03:17:01 +0000301 // ::= <local-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000302 //
Anders Carlssond58d6f72009-09-17 16:12:20 +0000303 const DeclContext *DC = ND->getDeclContext();
Anders Carlsson5cc58c62009-09-22 17:23:30 +0000304 while (isa<LinkageSpecDecl>(DC))
Anders Carlssond58d6f72009-09-17 16:12:20 +0000305 DC = DC->getParent();
Anders Carlssond58d6f72009-09-17 16:12:20 +0000306
307 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000308 // Check if we have a template.
309 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000310 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000311 mangleUnscopedTemplateName(TD);
312 mangleTemplateArgumentList(*TemplateArgs);
313 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000314 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000315
Anders Carlsson7482e242009-09-18 04:29:09 +0000316 mangleUnscopedName(ND);
317 return;
318 }
319
320 if (isa<FunctionDecl>(DC)) {
321 mangleLocalName(ND);
322 return;
323 }
324
325 mangleNestedName(ND);
326}
Anders Carlsson7624f212009-09-18 02:42:01 +0000327void CXXNameMangler::mangleName(const TemplateDecl *TD,
328 const TemplateArgument *TemplateArgs,
329 unsigned NumTemplateArgs) {
330 const DeclContext *DC = TD->getDeclContext();
331 while (isa<LinkageSpecDecl>(DC)) {
332 assert(cast<LinkageSpecDecl>(DC)->getLanguage() ==
333 LinkageSpecDecl::lang_cxx && "Unexpected linkage decl!");
334 DC = DC->getParent();
335 }
336
337 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000338 mangleUnscopedTemplateName(TD);
Anders Carlsson7624f212009-09-18 02:42:01 +0000339 mangleTemplateArgs(TemplateArgs, NumTemplateArgs);
340 } else {
341 mangleNestedName(TD, TemplateArgs, NumTemplateArgs);
342 }
343}
344
Anders Carlsson201ce742009-09-17 03:17:01 +0000345void CXXNameMangler::mangleUnscopedName(const NamedDecl *ND) {
346 // <unscoped-name> ::= <unqualified-name>
347 // ::= St <unqualified-name> # ::std::
348 if (isStdNamespace(ND->getDeclContext()))
349 Out << "St";
350
351 mangleUnqualifiedName(ND);
352}
353
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000354void CXXNameMangler::mangleUnscopedTemplateName(const TemplateDecl *ND) {
Anders Carlsson201ce742009-09-17 03:17:01 +0000355 // <unscoped-template-name> ::= <unscoped-name>
356 // ::= <substitution>
Anders Carlsson7624f212009-09-18 02:42:01 +0000357 if (mangleSubstitution(ND))
Anders Carlsson03c9d532009-09-17 04:02:31 +0000358 return;
359
Anders Carlsson1668f202009-09-26 20:13:56 +0000360 mangleUnscopedName(ND->getTemplatedDecl());
Anders Carlsson7624f212009-09-18 02:42:01 +0000361 addSubstitution(ND);
Anders Carlsson201ce742009-09-17 03:17:01 +0000362}
363
Mike Stump77ca8f62009-09-05 07:20:32 +0000364void CXXNameMangler::mangleCalloffset(int64_t nv, int64_t v) {
Mike Stump141c5af2009-09-02 00:25:38 +0000365 // <call-offset> ::= h <nv-offset> _
366 // ::= v <v-offset> _
367 // <nv-offset> ::= <offset number> # non-virtual base override
368 // <v-offset> ::= <offset nubmer> _ <virtual offset number>
369 // # virtual base override, with vcall offset
Mike Stump77ca8f62009-09-05 07:20:32 +0000370 if (v == 0) {
Mike Stump9124bcc2009-09-02 00:56:18 +0000371 Out << "h";
Mike Stump141c5af2009-09-02 00:25:38 +0000372 if (nv < 0) {
373 Out << "n";
374 nv = -nv;
375 }
376 Out << nv;
377 } else {
Mike Stump9124bcc2009-09-02 00:56:18 +0000378 Out << "v";
Mike Stump141c5af2009-09-02 00:25:38 +0000379 if (nv < 0) {
380 Out << "n";
381 nv = -nv;
382 }
383 Out << nv;
384 Out << "_";
385 if (v < 0) {
386 Out << "n";
387 v = -v;
388 }
389 Out << v;
390 }
391 Out << "_";
Mike Stump9124bcc2009-09-02 00:56:18 +0000392}
393
Mike Stumpdec025b2009-09-07 04:27:52 +0000394void CXXNameMangler::mangleThunk(const FunctionDecl *FD, int64_t nv,
395 int64_t v) {
Mike Stump9124bcc2009-09-02 00:56:18 +0000396 // <special-name> ::= T <call-offset> <base encoding>
397 // # base is the nominal target function of thunk
Mike Stumpdec025b2009-09-07 04:27:52 +0000398 Out << "_ZT";
Mike Stump77ca8f62009-09-05 07:20:32 +0000399 mangleCalloffset(nv, v);
Mike Stumpdec025b2009-09-07 04:27:52 +0000400 mangleFunctionEncoding(FD);
Mike Stump9124bcc2009-09-02 00:56:18 +0000401}
402
Mike Stumpdec025b2009-09-07 04:27:52 +0000403 void CXXNameMangler::mangleCovariantThunk(const FunctionDecl *FD,
Mike Stump77ca8f62009-09-05 07:20:32 +0000404 int64_t nv_t, int64_t v_t,
Mike Stump9124bcc2009-09-02 00:56:18 +0000405 int64_t nv_r, int64_t v_r) {
406 // <special-name> ::= Tc <call-offset> <call-offset> <base encoding>
407 // # base is the nominal target function of thunk
408 // # first call-offset is 'this' adjustment
409 // # second call-offset is result adjustment
Mike Stumpdec025b2009-09-07 04:27:52 +0000410 Out << "_ZTc";
Mike Stump77ca8f62009-09-05 07:20:32 +0000411 mangleCalloffset(nv_t, v_t);
412 mangleCalloffset(nv_r, v_r);
Mike Stumpdec025b2009-09-07 04:27:52 +0000413 mangleFunctionEncoding(FD);
Mike Stump141c5af2009-09-02 00:25:38 +0000414}
415
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000416void CXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND) {
417 // <unqualified-name> ::= <operator-name>
Mike Stump1eb44332009-09-09 15:08:12 +0000418 // ::= <ctor-dtor-name>
419 // ::= <source-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000420 DeclarationName Name = ND->getDeclName();
421 switch (Name.getNameKind()) {
Anders Carlssonc4355b62009-10-07 01:45:02 +0000422 case DeclarationName::Identifier: {
423 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
John McCall9aeed322009-10-01 00:25:31 +0000424 if (NS->isAnonymousNamespace()) {
425 // This is how gcc mangles these names. It's apparently
426 // always '1', no matter how many different anonymous
427 // namespaces appear in a context.
428 Out << "12_GLOBAL__N_1";
429 break;
430 }
Anders Carlssonc4355b62009-10-07 01:45:02 +0000431 }
432
433 if (const IdentifierInfo *II = Name.getAsIdentifierInfo()) {
434 mangleSourceName(II);
435 break;
436 }
437
438 // We must have an anonymous struct.
439 const TagDecl *TD = cast<TagDecl>(ND);
440 if (const TypedefDecl *D = TD->getTypedefForAnonDecl()) {
441 assert(TD->getDeclContext() == D->getDeclContext() &&
442 "Typedef should not be in another decl context!");
443 assert(D->getDeclName().getAsIdentifierInfo() &&
444 "Typedef was not named!");
445 mangleSourceName(D->getDeclName().getAsIdentifierInfo());
446 break;
447 }
448
449 // Get a unique id for the anonymous struct.
450 uint64_t AnonStructId = Context.getAnonymousStructId(TD);
451
452 // Mangle it as a source name in the form
453 // [n] $_<id>
454 // where n is the length of the string.
455 llvm::SmallString<8> Str;
456 Str += "$_";
457 Str += llvm::utostr(AnonStructId);
458
459 Out << Str.size();
460 Out << Str.str();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000461 break;
Anders Carlssonc4355b62009-10-07 01:45:02 +0000462 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000463
464 case DeclarationName::ObjCZeroArgSelector:
465 case DeclarationName::ObjCOneArgSelector:
466 case DeclarationName::ObjCMultiArgSelector:
467 assert(false && "Can't mangle Objective-C selector names here!");
468 break;
469
470 case DeclarationName::CXXConstructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000471 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000472 // If the named decl is the C++ constructor we're mangling, use the type
473 // we were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000474 mangleCXXCtorType(static_cast<CXXCtorType>(StructorType));
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000475 else
476 // Otherwise, use the complete constructor name. This is relevant if a
477 // class with a constructor is declared within a constructor.
478 mangleCXXCtorType(Ctor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000479 break;
480
481 case DeclarationName::CXXDestructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000482 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000483 // If the named decl is the C++ destructor we're mangling, use the type we
484 // were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000485 mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
486 else
487 // Otherwise, use the complete destructor name. This is relevant if a
488 // class with a destructor is declared within a destructor.
489 mangleCXXDtorType(Dtor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000490 break;
491
492 case DeclarationName::CXXConversionFunctionName:
Mike Stump1eb44332009-09-09 15:08:12 +0000493 // <operator-name> ::= cv <type> # (cast)
Douglas Gregor219cc612009-02-13 01:28:03 +0000494 Out << "cv";
Anders Carlssonb5404912009-10-07 01:06:45 +0000495 mangleType(Context.getASTContext().getCanonicalType(Name.getCXXNameType()));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000496 break;
497
498 case DeclarationName::CXXOperatorName:
499 mangleOperatorName(Name.getCXXOverloadedOperator(),
500 cast<FunctionDecl>(ND)->getNumParams());
501 break;
502
503 case DeclarationName::CXXUsingDirective:
504 assert(false && "Can't mangle a using directive name!");
Douglas Gregor219cc612009-02-13 01:28:03 +0000505 break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000506 }
507}
508
509void CXXNameMangler::mangleSourceName(const IdentifierInfo *II) {
510 // <source-name> ::= <positive length number> <identifier>
511 // <number> ::= [n] <non-negative decimal integer>
512 // <identifier> ::= <unqualified source code identifier>
513 Out << II->getLength() << II->getName();
514}
515
516void CXXNameMangler::mangleNestedName(const NamedDecl *ND) {
517 // <nested-name> ::= N [<CV-qualifiers>] <prefix> <unqualified-name> E
518 // ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
Anders Carlssond99edc42009-09-26 03:55:37 +0000519
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000520 Out << 'N';
521 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(ND))
John McCall0953e762009-09-24 19:53:00 +0000522 mangleQualifiers(Qualifiers::fromCVRMask(Method->getTypeQualifiers()));
Anders Carlsson7482e242009-09-18 04:29:09 +0000523
Anders Carlsson2744a062009-09-18 19:00:18 +0000524 // Check if we have a template.
525 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000526 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000527 mangleTemplatePrefix(TD);
528 mangleTemplateArgumentList(*TemplateArgs);
Anders Carlsson7482e242009-09-18 04:29:09 +0000529 } else {
530 manglePrefix(ND->getDeclContext());
531 mangleUnqualifiedName(ND);
532 }
533
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000534 Out << 'E';
535}
Anders Carlsson7624f212009-09-18 02:42:01 +0000536void CXXNameMangler::mangleNestedName(const TemplateDecl *TD,
537 const TemplateArgument *TemplateArgs,
538 unsigned NumTemplateArgs) {
Anders Carlssone45117b2009-09-27 19:53:49 +0000539 // <nested-name> ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
540
Anders Carlsson7624f212009-09-18 02:42:01 +0000541 Out << 'N';
Anders Carlsson7624f212009-09-18 02:42:01 +0000542
Anders Carlssone45117b2009-09-27 19:53:49 +0000543 mangleTemplatePrefix(TD);
Anders Carlsson7624f212009-09-18 02:42:01 +0000544 mangleTemplateArgs(TemplateArgs, NumTemplateArgs);
Anders Carlssone45117b2009-09-27 19:53:49 +0000545
Anders Carlsson7624f212009-09-18 02:42:01 +0000546 Out << 'E';
547}
548
Anders Carlsson1b42c792009-04-02 16:24:45 +0000549void CXXNameMangler::mangleLocalName(const NamedDecl *ND) {
550 // <local-name> := Z <function encoding> E <entity name> [<discriminator>]
551 // := Z <function encoding> E s [<discriminator>]
Mike Stump1eb44332009-09-09 15:08:12 +0000552 // <discriminator> := _ <non-negative number>
Anders Carlsson1b42c792009-04-02 16:24:45 +0000553 Out << 'Z';
554 mangleFunctionEncoding(cast<FunctionDecl>(ND->getDeclContext()));
555 Out << 'E';
556 mangleSourceName(ND->getIdentifier());
557}
558
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000559void CXXNameMangler::manglePrefix(const DeclContext *DC) {
560 // <prefix> ::= <prefix> <unqualified-name>
561 // ::= <template-prefix> <template-args>
562 // ::= <template-param>
563 // ::= # empty
564 // ::= <substitution>
565 // FIXME: We only handle mangling of namespaces and classes at the moment.
Anders Carlsson6862fc72009-09-17 04:16:28 +0000566
Anders Carlssonadd28822009-09-22 20:33:31 +0000567 while (isa<LinkageSpecDecl>(DC))
568 DC = DC->getParent();
569
Anders Carlsson9263e912009-09-18 18:39:58 +0000570 if (DC->isTranslationUnit())
571 return;
572
Anders Carlsson6862fc72009-09-17 04:16:28 +0000573 if (mangleSubstitution(cast<NamedDecl>(DC)))
574 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000575
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000576 // Check if we have a template.
577 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000578 if (const TemplateDecl *TD = isTemplate(cast<NamedDecl>(DC), TemplateArgs)) {
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000579 mangleTemplatePrefix(TD);
580 mangleTemplateArgumentList(*TemplateArgs);
581 } else {
582 manglePrefix(DC->getParent());
583 mangleUnqualifiedName(cast<NamedDecl>(DC));
584 }
Anders Carlsson6862fc72009-09-17 04:16:28 +0000585
586 addSubstitution(cast<NamedDecl>(DC));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000587}
588
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000589void CXXNameMangler::mangleTemplatePrefix(const TemplateDecl *ND) {
Anders Carlsson7482e242009-09-18 04:29:09 +0000590 // <template-prefix> ::= <prefix> <template unqualified-name>
591 // ::= <template-param>
592 // ::= <substitution>
593
Anders Carlssonaeb85372009-09-26 22:18:22 +0000594 if (mangleSubstitution(ND))
595 return;
596
597 // FIXME: <template-param>
Anders Carlssonaa73ab12009-09-18 18:47:07 +0000598
599 manglePrefix(ND->getDeclContext());
Anders Carlsson1668f202009-09-26 20:13:56 +0000600 mangleUnqualifiedName(ND->getTemplatedDecl());
Anders Carlssonaeb85372009-09-26 22:18:22 +0000601
602 addSubstitution(ND);
Anders Carlsson7482e242009-09-18 04:29:09 +0000603}
604
Mike Stump1eb44332009-09-09 15:08:12 +0000605void
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000606CXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity) {
607 switch (OO) {
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000608 // <operator-name> ::= nw # new
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000609 case OO_New: Out << "nw"; break;
610 // ::= na # new[]
611 case OO_Array_New: Out << "na"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000612 // ::= dl # delete
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000613 case OO_Delete: Out << "dl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000614 // ::= da # delete[]
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000615 case OO_Array_Delete: Out << "da"; break;
616 // ::= ps # + (unary)
617 // ::= pl # +
618 case OO_Plus: Out << (Arity == 1? "ps" : "pl"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000619 // ::= ng # - (unary)
620 // ::= mi # -
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000621 case OO_Minus: Out << (Arity == 1? "ng" : "mi"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000622 // ::= ad # & (unary)
623 // ::= an # &
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000624 case OO_Amp: Out << (Arity == 1? "ad" : "an"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000625 // ::= de # * (unary)
626 // ::= ml # *
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000627 case OO_Star: Out << (Arity == 1? "de" : "ml"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000628 // ::= co # ~
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000629 case OO_Tilde: Out << "co"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000630 // ::= dv # /
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000631 case OO_Slash: Out << "dv"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000632 // ::= rm # %
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000633 case OO_Percent: Out << "rm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000634 // ::= or # |
635 case OO_Pipe: Out << "or"; break;
636 // ::= eo # ^
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000637 case OO_Caret: Out << "eo"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000638 // ::= aS # =
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000639 case OO_Equal: Out << "aS"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000640 // ::= pL # +=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000641 case OO_PlusEqual: Out << "pL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000642 // ::= mI # -=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000643 case OO_MinusEqual: Out << "mI"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000644 // ::= mL # *=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000645 case OO_StarEqual: Out << "mL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000646 // ::= dV # /=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000647 case OO_SlashEqual: Out << "dV"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000648 // ::= rM # %=
649 case OO_PercentEqual: Out << "rM"; break;
650 // ::= aN # &=
651 case OO_AmpEqual: Out << "aN"; break;
652 // ::= oR # |=
653 case OO_PipeEqual: Out << "oR"; break;
654 // ::= eO # ^=
655 case OO_CaretEqual: Out << "eO"; break;
656 // ::= ls # <<
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000657 case OO_LessLess: Out << "ls"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000658 // ::= rs # >>
659 case OO_GreaterGreater: Out << "rs"; break;
660 // ::= lS # <<=
661 case OO_LessLessEqual: Out << "lS"; break;
662 // ::= rS # >>=
663 case OO_GreaterGreaterEqual: Out << "rS"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000664 // ::= eq # ==
665 case OO_EqualEqual: Out << "eq"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000666 // ::= ne # !=
667 case OO_ExclaimEqual: Out << "ne"; break;
668 // ::= lt # <
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000669 case OO_Less: Out << "lt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000670 // ::= gt # >
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000671 case OO_Greater: Out << "gt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000672 // ::= le # <=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000673 case OO_LessEqual: Out << "le"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000674 // ::= ge # >=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000675 case OO_GreaterEqual: Out << "ge"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000676 // ::= nt # !
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000677 case OO_Exclaim: Out << "nt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000678 // ::= aa # &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000679 case OO_AmpAmp: Out << "aa"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000680 // ::= oo # ||
681 case OO_PipePipe: Out << "oo"; break;
682 // ::= pp # ++
683 case OO_PlusPlus: Out << "pp"; break;
684 // ::= mm # --
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000685 case OO_MinusMinus: Out << "mm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000686 // ::= cm # ,
687 case OO_Comma: Out << "cm"; break;
688 // ::= pm # ->*
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000689 case OO_ArrowStar: Out << "pm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000690 // ::= pt # ->
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000691 case OO_Arrow: Out << "pt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000692 // ::= cl # ()
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000693 case OO_Call: Out << "cl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000694 // ::= ix # []
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000695 case OO_Subscript: Out << "ix"; break;
696 // UNSUPPORTED: ::= qu # ?
697
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000698 case OO_None:
699 case OO_Conditional:
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000700 case NUM_OVERLOADED_OPERATORS:
Mike Stump1eb44332009-09-09 15:08:12 +0000701 assert(false && "Not an overloaded operator");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000702 break;
703 }
704}
705
John McCall0953e762009-09-24 19:53:00 +0000706void CXXNameMangler::mangleQualifiers(Qualifiers Quals) {
Mike Stump1eb44332009-09-09 15:08:12 +0000707 // <CV-qualifiers> ::= [r] [V] [K] # restrict (C99), volatile, const
John McCall0953e762009-09-24 19:53:00 +0000708 if (Quals.hasRestrict())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000709 Out << 'r';
John McCall0953e762009-09-24 19:53:00 +0000710 if (Quals.hasVolatile())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000711 Out << 'V';
John McCall0953e762009-09-24 19:53:00 +0000712 if (Quals.hasConst())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000713 Out << 'K';
John McCall0953e762009-09-24 19:53:00 +0000714
715 // FIXME: For now, just drop all extension qualifiers on the floor.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000716}
717
718void CXXNameMangler::mangleType(QualType T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000719 // Only operate on the canonical type!
Anders Carlssonb5404912009-10-07 01:06:45 +0000720 T = Context.getASTContext().getCanonicalType(T);
Anders Carlsson4843e582009-03-10 17:07:44 +0000721
Anders Carlsson410f7642009-10-16 02:06:06 +0000722 bool IsSubstitutable = T.hasQualifiers() || !isa<BuiltinType>(T);
Anders Carlsson76967372009-09-17 00:43:46 +0000723 if (IsSubstitutable && mangleSubstitution(T))
724 return;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000725
John McCall0953e762009-09-24 19:53:00 +0000726 if (Qualifiers Quals = T.getQualifiers()) {
727 mangleQualifiers(Quals);
728 // Recurse: even if the qualified type isn't yet substitutable,
729 // the unqualified type might be.
Anders Carlsson76967372009-09-17 00:43:46 +0000730 mangleType(T.getUnqualifiedType());
731 } else {
732 switch (T->getTypeClass()) {
John McCallefe6aee2009-09-05 07:56:18 +0000733#define ABSTRACT_TYPE(CLASS, PARENT)
734#define NON_CANONICAL_TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000735 case Type::CLASS: \
736 llvm::llvm_unreachable("can't mangle non-canonical type " #CLASS "Type"); \
737 return;
John McCallefe6aee2009-09-05 07:56:18 +0000738#define TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000739 case Type::CLASS: \
John McCall0953e762009-09-24 19:53:00 +0000740 mangleType(static_cast<const CLASS##Type*>(T.getTypePtr())); \
Anders Carlsson76967372009-09-17 00:43:46 +0000741 break;
John McCallefe6aee2009-09-05 07:56:18 +0000742#include "clang/AST/TypeNodes.def"
Anders Carlsson76967372009-09-17 00:43:46 +0000743 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000744 }
Anders Carlsson76967372009-09-17 00:43:46 +0000745
746 // Add the substitution.
747 if (IsSubstitutable)
748 addSubstitution(T);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000749}
750
751void CXXNameMangler::mangleType(const BuiltinType *T) {
John McCallefe6aee2009-09-05 07:56:18 +0000752 // <type> ::= <builtin-type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000753 // <builtin-type> ::= v # void
754 // ::= w # wchar_t
755 // ::= b # bool
756 // ::= c # char
757 // ::= a # signed char
758 // ::= h # unsigned char
759 // ::= s # short
760 // ::= t # unsigned short
761 // ::= i # int
762 // ::= j # unsigned int
763 // ::= l # long
764 // ::= m # unsigned long
765 // ::= x # long long, __int64
766 // ::= y # unsigned long long, __int64
767 // ::= n # __int128
768 // UNSUPPORTED: ::= o # unsigned __int128
769 // ::= f # float
770 // ::= d # double
771 // ::= e # long double, __float80
772 // UNSUPPORTED: ::= g # __float128
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000773 // UNSUPPORTED: ::= Dd # IEEE 754r decimal floating point (64 bits)
774 // UNSUPPORTED: ::= De # IEEE 754r decimal floating point (128 bits)
775 // UNSUPPORTED: ::= Df # IEEE 754r decimal floating point (32 bits)
776 // UNSUPPORTED: ::= Dh # IEEE 754r half-precision floating point (16 bits)
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000777 // ::= Di # char32_t
778 // ::= Ds # char16_t
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000779 // ::= u <source-name> # vendor extended type
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000780 // From our point of view, std::nullptr_t is a builtin, but as far as mangling
781 // is concerned, it's a type called std::nullptr_t.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000782 switch (T->getKind()) {
783 case BuiltinType::Void: Out << 'v'; break;
784 case BuiltinType::Bool: Out << 'b'; break;
785 case BuiltinType::Char_U: case BuiltinType::Char_S: Out << 'c'; break;
786 case BuiltinType::UChar: Out << 'h'; break;
787 case BuiltinType::UShort: Out << 't'; break;
788 case BuiltinType::UInt: Out << 'j'; break;
789 case BuiltinType::ULong: Out << 'm'; break;
790 case BuiltinType::ULongLong: Out << 'y'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000791 case BuiltinType::UInt128: Out << 'o'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000792 case BuiltinType::SChar: Out << 'a'; break;
793 case BuiltinType::WChar: Out << 'w'; break;
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000794 case BuiltinType::Char16: Out << "Ds"; break;
795 case BuiltinType::Char32: Out << "Di"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000796 case BuiltinType::Short: Out << 's'; break;
797 case BuiltinType::Int: Out << 'i'; break;
798 case BuiltinType::Long: Out << 'l'; break;
799 case BuiltinType::LongLong: Out << 'x'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000800 case BuiltinType::Int128: Out << 'n'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000801 case BuiltinType::Float: Out << 'f'; break;
802 case BuiltinType::Double: Out << 'd'; break;
803 case BuiltinType::LongDouble: Out << 'e'; break;
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000804 case BuiltinType::NullPtr: Out << "St9nullptr_t"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000805
806 case BuiltinType::Overload:
807 case BuiltinType::Dependent:
Mike Stump1eb44332009-09-09 15:08:12 +0000808 assert(false &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000809 "Overloaded and dependent types shouldn't get to name mangling");
810 break;
Anders Carlssone89d1592009-06-26 18:41:36 +0000811 case BuiltinType::UndeducedAuto:
812 assert(0 && "Should not see undeduced auto here");
813 break;
Steve Naroff9533a7f2009-07-22 17:14:51 +0000814 case BuiltinType::ObjCId: Out << "11objc_object"; break;
815 case BuiltinType::ObjCClass: Out << "10objc_class"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000816 }
817}
818
John McCallefe6aee2009-09-05 07:56:18 +0000819// <type> ::= <function-type>
820// <function-type> ::= F [Y] <bare-function-type> E
821void CXXNameMangler::mangleType(const FunctionProtoType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000822 Out << 'F';
Mike Stumpf5408fe2009-05-16 07:57:57 +0000823 // FIXME: We don't have enough information in the AST to produce the 'Y'
824 // encoding for extern "C" function types.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000825 mangleBareFunctionType(T, /*MangleReturnType=*/true);
826 Out << 'E';
827}
John McCallefe6aee2009-09-05 07:56:18 +0000828void CXXNameMangler::mangleType(const FunctionNoProtoType *T) {
829 llvm::llvm_unreachable("Can't mangle K&R function prototypes");
830}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000831void CXXNameMangler::mangleBareFunctionType(const FunctionType *T,
832 bool MangleReturnType) {
John McCallefe6aee2009-09-05 07:56:18 +0000833 // We should never be mangling something without a prototype.
834 const FunctionProtoType *Proto = cast<FunctionProtoType>(T);
835
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000836 // <bare-function-type> ::= <signature type>+
837 if (MangleReturnType)
John McCallefe6aee2009-09-05 07:56:18 +0000838 mangleType(Proto->getResultType());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000839
Anders Carlssonc6c91bc2009-04-01 00:15:23 +0000840 if (Proto->getNumArgs() == 0) {
841 Out << 'v';
842 return;
843 }
Mike Stump1eb44332009-09-09 15:08:12 +0000844
Douglas Gregor72564e72009-02-26 23:50:07 +0000845 for (FunctionProtoType::arg_type_iterator Arg = Proto->arg_type_begin(),
Mike Stump1eb44332009-09-09 15:08:12 +0000846 ArgEnd = Proto->arg_type_end();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000847 Arg != ArgEnd; ++Arg)
848 mangleType(*Arg);
Douglas Gregor219cc612009-02-13 01:28:03 +0000849
850 // <builtin-type> ::= z # ellipsis
851 if (Proto->isVariadic())
852 Out << 'z';
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000853}
854
John McCallefe6aee2009-09-05 07:56:18 +0000855// <type> ::= <class-enum-type>
Mike Stump1eb44332009-09-09 15:08:12 +0000856// <class-enum-type> ::= <name>
John McCallefe6aee2009-09-05 07:56:18 +0000857void CXXNameMangler::mangleType(const EnumType *T) {
858 mangleType(static_cast<const TagType*>(T));
859}
860void CXXNameMangler::mangleType(const RecordType *T) {
861 mangleType(static_cast<const TagType*>(T));
862}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000863void CXXNameMangler::mangleType(const TagType *T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000864 if (!T->getDecl()->getIdentifier())
865 mangleName(T->getDecl()->getTypedefForAnonDecl());
866 else
867 mangleName(T->getDecl());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000868}
869
John McCallefe6aee2009-09-05 07:56:18 +0000870// <type> ::= <array-type>
871// <array-type> ::= A <positive dimension number> _ <element type>
872// ::= A [<dimension expression>] _ <element type>
873void CXXNameMangler::mangleType(const ConstantArrayType *T) {
874 Out << 'A' << T->getSize() << '_';
875 mangleType(T->getElementType());
876}
877void CXXNameMangler::mangleType(const VariableArrayType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000878 Out << 'A';
John McCallefe6aee2009-09-05 07:56:18 +0000879 mangleExpression(T->getSizeExpr());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000880 Out << '_';
881 mangleType(T->getElementType());
882}
John McCallefe6aee2009-09-05 07:56:18 +0000883void CXXNameMangler::mangleType(const DependentSizedArrayType *T) {
884 Out << 'A';
885 mangleExpression(T->getSizeExpr());
886 Out << '_';
887 mangleType(T->getElementType());
888}
889void CXXNameMangler::mangleType(const IncompleteArrayType *T) {
890 Out << 'A' << '_';
891 mangleType(T->getElementType());
892}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000893
John McCallefe6aee2009-09-05 07:56:18 +0000894// <type> ::= <pointer-to-member-type>
895// <pointer-to-member-type> ::= M <class type> <member type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000896void CXXNameMangler::mangleType(const MemberPointerType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000897 Out << 'M';
898 mangleType(QualType(T->getClass(), 0));
Anders Carlsson0e650012009-05-17 17:41:20 +0000899 QualType PointeeType = T->getPointeeType();
900 if (const FunctionProtoType *FPT = dyn_cast<FunctionProtoType>(PointeeType)) {
John McCall0953e762009-09-24 19:53:00 +0000901 mangleQualifiers(Qualifiers::fromCVRMask(FPT->getTypeQuals()));
Anders Carlsson0e650012009-05-17 17:41:20 +0000902 mangleType(FPT);
Mike Stump1eb44332009-09-09 15:08:12 +0000903 } else
Anders Carlsson0e650012009-05-17 17:41:20 +0000904 mangleType(PointeeType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000905}
906
John McCallefe6aee2009-09-05 07:56:18 +0000907// <type> ::= <template-param>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000908void CXXNameMangler::mangleType(const TemplateTypeParmType *T) {
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +0000909 mangleTemplateParameter(T->getIndex());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000910}
911
John McCallefe6aee2009-09-05 07:56:18 +0000912// FIXME: <type> ::= <template-template-param> <template-args>
John McCallefe6aee2009-09-05 07:56:18 +0000913
914// <type> ::= P <type> # pointer-to
915void CXXNameMangler::mangleType(const PointerType *T) {
916 Out << 'P';
917 mangleType(T->getPointeeType());
918}
919void CXXNameMangler::mangleType(const ObjCObjectPointerType *T) {
920 Out << 'P';
921 mangleType(T->getPointeeType());
922}
923
924// <type> ::= R <type> # reference-to
925void CXXNameMangler::mangleType(const LValueReferenceType *T) {
926 Out << 'R';
927 mangleType(T->getPointeeType());
928}
929
930// <type> ::= O <type> # rvalue reference-to (C++0x)
931void CXXNameMangler::mangleType(const RValueReferenceType *T) {
932 Out << 'O';
933 mangleType(T->getPointeeType());
934}
935
936// <type> ::= C <type> # complex pair (C 2000)
937void CXXNameMangler::mangleType(const ComplexType *T) {
938 Out << 'C';
939 mangleType(T->getElementType());
940}
941
942// GNU extension: vector types
943void CXXNameMangler::mangleType(const VectorType *T) {
944 Out << "U8__vector";
945 mangleType(T->getElementType());
946}
947void CXXNameMangler::mangleType(const ExtVectorType *T) {
948 mangleType(static_cast<const VectorType*>(T));
949}
950void CXXNameMangler::mangleType(const DependentSizedExtVectorType *T) {
951 Out << "U8__vector";
952 mangleType(T->getElementType());
953}
954
Anders Carlssona40c5e42009-03-07 22:03:21 +0000955void CXXNameMangler::mangleType(const ObjCInterfaceType *T) {
956 mangleSourceName(T->getDecl()->getIdentifier());
957}
958
John McCallefe6aee2009-09-05 07:56:18 +0000959void CXXNameMangler::mangleType(const BlockPointerType *T) {
960 assert(false && "can't mangle block pointer types yet");
961}
962
963void CXXNameMangler::mangleType(const FixedWidthIntType *T) {
964 assert(false && "can't mangle arbitary-precision integer type yet");
965}
966
967void CXXNameMangler::mangleType(const TemplateSpecializationType *T) {
Anders Carlsson7624f212009-09-18 02:42:01 +0000968 TemplateDecl *TD = T->getTemplateName().getAsTemplateDecl();
969 assert(TD && "FIXME: Support dependent template names!");
970
971 mangleName(TD, T->getArgs(), T->getNumArgs());
John McCallefe6aee2009-09-05 07:56:18 +0000972}
973
974void CXXNameMangler::mangleType(const TypenameType *T) {
Anders Carlssonae352482009-09-26 02:26:02 +0000975 // Typename types are always nested
976 Out << 'N';
977
978 const Type *QTy = T->getQualifier()->getAsType();
979 if (const TemplateSpecializationType *TST =
980 dyn_cast<TemplateSpecializationType>(QTy)) {
Anders Carlsson88599172009-09-27 01:06:07 +0000981 if (!mangleSubstitution(QualType(TST, 0))) {
982 TemplateDecl *TD = TST->getTemplateName().getAsTemplateDecl();
Anders Carlssonae352482009-09-26 02:26:02 +0000983
Anders Carlsson88599172009-09-27 01:06:07 +0000984 mangleTemplatePrefix(TD);
985 mangleTemplateArgs(TST->getArgs(), TST->getNumArgs());
986 addSubstitution(QualType(TST, 0));
987 }
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +0000988 } else if (const TemplateTypeParmType *TTPT =
989 dyn_cast<TemplateTypeParmType>(QTy)) {
990 // We use the QualType mangle type variant here because it handles
991 // substitutions.
992 mangleType(QualType(TTPT, 0));
Anders Carlssonae352482009-09-26 02:26:02 +0000993 } else
994 assert(false && "Unhandled type!");
995
996 mangleSourceName(T->getIdentifier());
997
998 Out << 'E';
John McCallefe6aee2009-09-05 07:56:18 +0000999}
1000
Anders Carlssond553f8c2009-09-21 01:21:10 +00001001void CXXNameMangler::mangleExpression(const Expr *E) {
1002 // <expression> ::= <unary operator-name> <expression>
1003 // ::= <binary operator-name> <expression> <expression>
1004 // ::= <trinary operator-name> <expression> <expression> <expression>
1005 // ::= cl <expression>* E # call
1006 // ::= cv <type> expression # conversion with one argument
1007 // ::= cv <type> _ <expression>* E # conversion with a different number of arguments
1008 // ::= st <type> # sizeof (a type)
1009 // ::= at <type> # alignof (a type)
1010 // ::= <template-param>
1011 // ::= <function-param>
1012 // ::= sr <type> <unqualified-name> # dependent name
1013 // ::= sr <type> <unqualified-name> <template-args> # dependent template-id
1014 // ::= sZ <template-param> # size of a parameter pack
1015 // ::= <expr-primary>
1016 switch (E->getStmtClass()) {
1017 default: assert(false && "Unhandled expression kind!");
Anders Carlssona7694082009-11-06 02:50:19 +00001018
1019 case Expr::ParenExprClass:
1020 mangleExpression(cast<ParenExpr>(E)->getSubExpr());
1021 break;
1022
Anders Carlssond553f8c2009-09-21 01:21:10 +00001023 case Expr::DeclRefExprClass: {
1024 const Decl *D = cast<DeclRefExpr>(E)->getDecl();
1025
1026 switch (D->getKind()) {
1027 default: assert(false && "Unhandled decl kind!");
1028 case Decl::NonTypeTemplateParm: {
1029 const NonTypeTemplateParmDecl *PD = cast<NonTypeTemplateParmDecl>(D);
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001030 mangleTemplateParameter(PD->getIndex());
Anders Carlssond553f8c2009-09-21 01:21:10 +00001031 break;
1032 }
1033
1034 }
Anders Carlsson50755b02009-09-27 20:11:34 +00001035
1036 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001037 }
Anders Carlsson50755b02009-09-27 20:11:34 +00001038
1039 case Expr::UnresolvedDeclRefExprClass: {
1040 const UnresolvedDeclRefExpr *DRE = cast<UnresolvedDeclRefExpr>(E);
1041 const Type *QTy = DRE->getQualifier()->getAsType();
1042 assert(QTy && "Qualifier was not type!");
1043
1044 // ::= sr <type> <unqualified-name> # dependent name
1045 Out << "sr";
1046 mangleType(QualType(QTy, 0));
1047
1048 assert(DRE->getDeclName().getNameKind() == DeclarationName::Identifier &&
1049 "Unhandled decl name kind!");
1050 mangleSourceName(DRE->getDeclName().getAsIdentifierInfo());
1051
1052 break;
1053 }
1054
Anders Carlssond553f8c2009-09-21 01:21:10 +00001055 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001056}
1057
John McCallefe6aee2009-09-05 07:56:18 +00001058// FIXME: <type> ::= G <type> # imaginary (C 2000)
1059// FIXME: <type> ::= U <source-name> <type> # vendor extended type qualifier
1060
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001061void CXXNameMangler::mangleCXXCtorType(CXXCtorType T) {
1062 // <ctor-dtor-name> ::= C1 # complete object constructor
1063 // ::= C2 # base object constructor
1064 // ::= C3 # complete object allocating constructor
1065 //
1066 switch (T) {
1067 case Ctor_Complete:
1068 Out << "C1";
1069 break;
1070 case Ctor_Base:
1071 Out << "C2";
1072 break;
1073 case Ctor_CompleteAllocating:
1074 Out << "C3";
1075 break;
1076 }
1077}
1078
Anders Carlsson27ae5362009-04-17 01:58:57 +00001079void CXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
1080 // <ctor-dtor-name> ::= D0 # deleting destructor
1081 // ::= D1 # complete object destructor
1082 // ::= D2 # base object destructor
1083 //
1084 switch (T) {
1085 case Dtor_Deleting:
1086 Out << "D0";
1087 break;
1088 case Dtor_Complete:
1089 Out << "D1";
1090 break;
1091 case Dtor_Base:
1092 Out << "D2";
1093 break;
1094 }
1095}
1096
Anders Carlsson068f3472009-09-17 05:31:47 +00001097void CXXNameMangler::mangleTemplateArgumentList(const TemplateArgumentList &L) {
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001098 // <template-args> ::= I <template-arg>+ E
1099 Out << "I";
Mike Stump1eb44332009-09-09 15:08:12 +00001100
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001101 for (unsigned i = 0, e = L.size(); i != e; ++i) {
1102 const TemplateArgument &A = L[i];
Mike Stump1eb44332009-09-09 15:08:12 +00001103
Anders Carlsson068f3472009-09-17 05:31:47 +00001104 mangleTemplateArgument(A);
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001105 }
Mike Stump1eb44332009-09-09 15:08:12 +00001106
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001107 Out << "E";
1108}
1109
Anders Carlsson7624f212009-09-18 02:42:01 +00001110void CXXNameMangler::mangleTemplateArgs(const TemplateArgument *TemplateArgs,
1111 unsigned NumTemplateArgs) {
1112 // <template-args> ::= I <template-arg>+ E
1113 Out << "I";
1114
1115 for (unsigned i = 0; i != NumTemplateArgs; ++i) {
1116 mangleTemplateArgument(TemplateArgs[i]);
1117 }
1118
1119 Out << "E";
1120}
1121
Anders Carlsson068f3472009-09-17 05:31:47 +00001122void CXXNameMangler::mangleTemplateArgument(const TemplateArgument &A) {
Mike Stump1eb44332009-09-09 15:08:12 +00001123 // <template-arg> ::= <type> # type or template
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001124 // ::= X <expression> E # expression
1125 // ::= <expr-primary> # simple expressions
1126 // ::= I <template-arg>* E # argument pack
1127 // ::= sp <expression> # pack expansion of (C++0x)
1128 switch (A.getKind()) {
1129 default:
1130 assert(0 && "Unknown template argument kind!");
1131 case TemplateArgument::Type:
1132 mangleType(A.getAsType());
1133 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001134 case TemplateArgument::Expression:
1135 Out << 'X';
1136 mangleExpression(A.getAsExpr());
1137 Out << 'E';
1138 break;
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001139 case TemplateArgument::Integral:
1140 // <expr-primary> ::= L <type> <value number> E # integer literal
1141
1142 Out << 'L';
Mike Stump1eb44332009-09-09 15:08:12 +00001143
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001144 mangleType(A.getIntegralType());
Mike Stump1eb44332009-09-09 15:08:12 +00001145
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001146 const llvm::APSInt *Integral = A.getAsIntegral();
1147 if (A.getIntegralType()->isBooleanType()) {
1148 // Boolean values are encoded as 0/1.
1149 Out << (Integral->getBoolValue() ? '1' : '0');
1150 } else {
1151 if (Integral->isNegative())
1152 Out << 'n';
1153 Integral->abs().print(Out, false);
1154 }
Mike Stump1eb44332009-09-09 15:08:12 +00001155
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001156 Out << 'E';
1157 break;
1158 }
1159}
1160
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001161void CXXNameMangler::mangleTemplateParameter(unsigned Index) {
1162 // <template-param> ::= T_ # first template parameter
1163 // ::= T <parameter-2 non-negative number> _
1164 if (Index == 0)
1165 Out << "T_";
1166 else
1167 Out << 'T' << (Index - 1) << '_';
1168}
1169
Anders Carlsson76967372009-09-17 00:43:46 +00001170// <substitution> ::= S <seq-id> _
1171// ::= S_
Anders Carlsson6862fc72009-09-17 04:16:28 +00001172bool CXXNameMangler::mangleSubstitution(const NamedDecl *ND) {
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001173 // Try one of the standard substitutions first.
1174 if (mangleStandardSubstitution(ND))
1175 return true;
1176
Anders Carlsson6862fc72009-09-17 04:16:28 +00001177 return mangleSubstitution(reinterpret_cast<uintptr_t>(ND));
1178}
1179
Anders Carlsson76967372009-09-17 00:43:46 +00001180bool CXXNameMangler::mangleSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001181 if (!T.getCVRQualifiers()) {
1182 if (const RecordType *RT = T->getAs<RecordType>())
1183 return mangleSubstitution(RT->getDecl());
1184 }
1185
Anders Carlsson76967372009-09-17 00:43:46 +00001186 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
1187
Anders Carlssond3a932a2009-09-17 03:53:28 +00001188 return mangleSubstitution(TypePtr);
1189}
1190
1191bool CXXNameMangler::mangleSubstitution(uintptr_t Ptr) {
Anders Carlsson76967372009-09-17 00:43:46 +00001192 llvm::DenseMap<uintptr_t, unsigned>::iterator I =
Anders Carlssond3a932a2009-09-17 03:53:28 +00001193 Substitutions.find(Ptr);
Anders Carlsson76967372009-09-17 00:43:46 +00001194 if (I == Substitutions.end())
1195 return false;
1196
1197 unsigned SeqID = I->second;
1198 if (SeqID == 0)
1199 Out << "S_";
1200 else {
1201 SeqID--;
1202
1203 // <seq-id> is encoded in base-36, using digits and upper case letters.
1204 char Buffer[10];
1205 char *BufferPtr = Buffer + 9;
1206
1207 *BufferPtr = 0;
1208 if (SeqID == 0) *--BufferPtr = '0';
1209
1210 while (SeqID) {
1211 assert(BufferPtr > Buffer && "Buffer overflow!");
1212
1213 unsigned char c = static_cast<unsigned char>(SeqID) % 36;
1214
1215 *--BufferPtr = (c < 10 ? '0' + c : 'A' + c - 10);
1216 SeqID /= 36;
1217 }
1218
1219 Out << 'S' << BufferPtr << '_';
1220 }
1221
1222 return true;
1223}
1224
Anders Carlssonf514b542009-09-27 00:12:57 +00001225static bool isCharType(QualType T) {
1226 if (T.isNull())
1227 return false;
1228
1229 return T->isSpecificBuiltinType(BuiltinType::Char_S) ||
1230 T->isSpecificBuiltinType(BuiltinType::Char_U);
1231}
1232
1233/// isCharSpecialization - Returns whether a given type is a template
1234/// specialization of a given name with a single argument of type char.
1235static bool isCharSpecialization(QualType T, const char *Name) {
1236 if (T.isNull())
1237 return false;
1238
1239 const RecordType *RT = T->getAs<RecordType>();
1240 if (!RT)
1241 return false;
1242
1243 const ClassTemplateSpecializationDecl *SD =
1244 dyn_cast<ClassTemplateSpecializationDecl>(RT->getDecl());
1245 if (!SD)
1246 return false;
1247
1248 if (!isStdNamespace(SD->getDeclContext()))
1249 return false;
1250
1251 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1252 if (TemplateArgs.size() != 1)
1253 return false;
1254
1255 if (!isCharType(TemplateArgs[0].getAsType()))
1256 return false;
1257
Daniel Dunbar01eb9b92009-10-18 21:17:35 +00001258 return SD->getIdentifier()->getName() == Name;
Anders Carlssonf514b542009-09-27 00:12:57 +00001259}
1260
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001261bool CXXNameMangler::mangleStandardSubstitution(const NamedDecl *ND) {
1262 // <substitution> ::= St # ::std::
Anders Carlsson8c031552009-09-26 23:10:05 +00001263 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
John McCall9aeed322009-10-01 00:25:31 +00001264 if (isStdNamespace(NS)) {
Anders Carlsson8c031552009-09-26 23:10:05 +00001265 Out << "St";
1266 return true;
1267 }
1268 }
1269
1270 if (const ClassTemplateDecl *TD = dyn_cast<ClassTemplateDecl>(ND)) {
1271 if (!isStdNamespace(TD->getDeclContext()))
1272 return false;
1273
1274 // <substitution> ::= Sa # ::std::allocator
1275 if (TD->getIdentifier()->isStr("allocator")) {
1276 Out << "Sa";
1277 return true;
1278 }
Anders Carlsson189d59c2009-09-26 23:14:39 +00001279
1280 // <<substitution> ::= Sb # ::std::basic_string
1281 if (TD->getIdentifier()->isStr("basic_string")) {
1282 Out << "Sb";
1283 return true;
1284 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001285 }
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001286
Anders Carlssonf514b542009-09-27 00:12:57 +00001287 if (const ClassTemplateSpecializationDecl *SD =
1288 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
1289 // <substitution> ::= Ss # ::std::basic_string<char,
1290 // ::std::char_traits<char>,
1291 // ::std::allocator<char> >
1292 if (SD->getIdentifier()->isStr("basic_string")) {
1293 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1294
1295 if (TemplateArgs.size() != 3)
1296 return false;
1297
1298 if (!isCharType(TemplateArgs[0].getAsType()))
1299 return false;
1300
1301 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1302 return false;
1303
1304 if (!isCharSpecialization(TemplateArgs[2].getAsType(), "allocator"))
1305 return false;
1306
1307 Out << "Ss";
1308 return true;
1309 }
Anders Carlsson8f8fd8e2009-10-08 17:20:26 +00001310
1311 // <substitution> ::= So # ::std::basic_ostream<char,
1312 // ::std::char_traits<char> >
1313 if (SD->getIdentifier()->isStr("basic_ostream")) {
1314 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1315
1316 if (TemplateArgs.size() != 2)
1317 return false;
1318
1319 if (!isCharType(TemplateArgs[0].getAsType()))
1320 return false;
1321
1322 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1323 return false;
1324
1325 Out << "So";
1326 return true;
1327 }
Anders Carlssonf514b542009-09-27 00:12:57 +00001328 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001329 return false;
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001330}
1331
Anders Carlsson76967372009-09-17 00:43:46 +00001332void CXXNameMangler::addSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001333 if (!T.getCVRQualifiers()) {
1334 if (const RecordType *RT = T->getAs<RecordType>()) {
1335 addSubstitution(RT->getDecl());
1336 return;
1337 }
1338 }
1339
Anders Carlsson76967372009-09-17 00:43:46 +00001340 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
Anders Carlssond3a932a2009-09-17 03:53:28 +00001341 addSubstitution(TypePtr);
1342}
1343
1344void CXXNameMangler::addSubstitution(uintptr_t Ptr) {
Anders Carlsson76967372009-09-17 00:43:46 +00001345 unsigned SeqID = Substitutions.size();
1346
Anders Carlssond3a932a2009-09-17 03:53:28 +00001347 assert(!Substitutions.count(Ptr) && "Substitution already exists!");
1348 Substitutions[Ptr] = SeqID;
Anders Carlsson76967372009-09-17 00:43:46 +00001349}
1350
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001351namespace clang {
Mike Stump141c5af2009-09-02 00:25:38 +00001352 /// \brief Mangles the name of the declaration D and emits that name to the
1353 /// given output stream.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001354 ///
Mike Stump141c5af2009-09-02 00:25:38 +00001355 /// If the declaration D requires a mangled name, this routine will emit that
1356 /// mangled name to \p os and return true. Otherwise, \p os will be unchanged
1357 /// and this routine will return false. In this case, the caller should just
1358 /// emit the identifier of the declaration (\c D->getIdentifier()) as its
1359 /// name.
Anders Carlssonb5404912009-10-07 01:06:45 +00001360 bool mangleName(MangleContext &Context, const NamedDecl *D,
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001361 llvm::raw_ostream &os) {
Anders Carlsson578aa642009-05-03 16:51:04 +00001362 assert(!isa<CXXConstructorDecl>(D) &&
1363 "Use mangleCXXCtor for constructor decls!");
1364 assert(!isa<CXXDestructorDecl>(D) &&
1365 "Use mangleCXXDtor for destructor decls!");
Mike Stump1eb44332009-09-09 15:08:12 +00001366
Anders Carlssond99edc42009-09-26 03:55:37 +00001367 PrettyStackTraceDecl CrashInfo(const_cast<NamedDecl *>(D), SourceLocation(),
Anders Carlssonb5404912009-10-07 01:06:45 +00001368 Context.getASTContext().getSourceManager(),
Anders Carlssond99edc42009-09-26 03:55:37 +00001369 "Mangling declaration");
1370
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001371 CXXNameMangler Mangler(Context, os);
Douglas Gregor457e2812009-10-28 16:31:34 +00001372 if (!Mangler.mangle(D))
Douglas Gregor6ec36682009-02-18 23:53:56 +00001373 return false;
Mike Stump1eb44332009-09-09 15:08:12 +00001374
Douglas Gregor6ec36682009-02-18 23:53:56 +00001375 os.flush();
1376 return true;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001377 }
Mike Stump1eb44332009-09-09 15:08:12 +00001378
Mike Stump141c5af2009-09-02 00:25:38 +00001379 /// \brief Mangles the a thunk with the offset n for the declaration D and
1380 /// emits that name to the given output stream.
Anders Carlssonb5404912009-10-07 01:06:45 +00001381 void mangleThunk(MangleContext &Context, const FunctionDecl *FD,
1382 int64_t nv, int64_t v, llvm::raw_ostream &os) {
Mike Stump141c5af2009-09-02 00:25:38 +00001383 // FIXME: Hum, we might have to thunk these, fix.
Mike Stumpdec025b2009-09-07 04:27:52 +00001384 assert(!isa<CXXDestructorDecl>(FD) &&
Mike Stump141c5af2009-09-02 00:25:38 +00001385 "Use mangleCXXDtor for destructor decls!");
Mike Stump1eb44332009-09-09 15:08:12 +00001386
Mike Stump141c5af2009-09-02 00:25:38 +00001387 CXXNameMangler Mangler(Context, os);
Mike Stumpdec025b2009-09-07 04:27:52 +00001388 Mangler.mangleThunk(FD, nv, v);
Mike Stump141c5af2009-09-02 00:25:38 +00001389 os.flush();
1390 }
Mike Stump1eb44332009-09-09 15:08:12 +00001391
Mike Stump9124bcc2009-09-02 00:56:18 +00001392 /// \brief Mangles the a covariant thunk for the declaration D and emits that
1393 /// name to the given output stream.
Anders Carlssonb5404912009-10-07 01:06:45 +00001394 void mangleCovariantThunk(MangleContext &Context, const FunctionDecl *FD,
1395 int64_t nv_t, int64_t v_t,
1396 int64_t nv_r, int64_t v_r,
Mike Stump9124bcc2009-09-02 00:56:18 +00001397 llvm::raw_ostream &os) {
1398 // FIXME: Hum, we might have to thunk these, fix.
Mike Stumpdec025b2009-09-07 04:27:52 +00001399 assert(!isa<CXXDestructorDecl>(FD) &&
Mike Stump9124bcc2009-09-02 00:56:18 +00001400 "Use mangleCXXDtor for destructor decls!");
Mike Stump1eb44332009-09-09 15:08:12 +00001401
Mike Stump9124bcc2009-09-02 00:56:18 +00001402 CXXNameMangler Mangler(Context, os);
Mike Stumpdec025b2009-09-07 04:27:52 +00001403 Mangler.mangleCovariantThunk(FD, nv_t, v_t, nv_r, v_r);
Mike Stump9124bcc2009-09-02 00:56:18 +00001404 os.flush();
1405 }
Mike Stump1eb44332009-09-09 15:08:12 +00001406
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001407 /// mangleGuardVariable - Returns the mangled name for a guard variable
1408 /// for the passed in VarDecl.
Anders Carlssonb5404912009-10-07 01:06:45 +00001409 void mangleGuardVariable(MangleContext &Context, const VarDecl *D,
Anders Carlsson41aa8c12009-04-13 18:02:10 +00001410 llvm::raw_ostream &os) {
1411 CXXNameMangler Mangler(Context, os);
1412 Mangler.mangleGuardVariable(D);
1413
1414 os.flush();
1415 }
Mike Stump1eb44332009-09-09 15:08:12 +00001416
Anders Carlssonb5404912009-10-07 01:06:45 +00001417 void mangleCXXCtor(MangleContext &Context, const CXXConstructorDecl *D,
1418 CXXCtorType Type, llvm::raw_ostream &os) {
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001419 CXXNameMangler Mangler(Context, os);
1420 Mangler.mangleCXXCtor(D, Type);
Mike Stump1eb44332009-09-09 15:08:12 +00001421
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001422 os.flush();
1423 }
Mike Stump1eb44332009-09-09 15:08:12 +00001424
Anders Carlssonb5404912009-10-07 01:06:45 +00001425 void mangleCXXDtor(MangleContext &Context, const CXXDestructorDecl *D,
1426 CXXDtorType Type, llvm::raw_ostream &os) {
Anders Carlsson27ae5362009-04-17 01:58:57 +00001427 CXXNameMangler Mangler(Context, os);
1428 Mangler.mangleCXXDtor(D, Type);
Mike Stump1eb44332009-09-09 15:08:12 +00001429
Anders Carlsson27ae5362009-04-17 01:58:57 +00001430 os.flush();
1431 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001432
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +00001433 void mangleCXXVtable(MangleContext &Context, const CXXRecordDecl *RD,
Mike Stumpf1216772009-07-31 18:25:34 +00001434 llvm::raw_ostream &os) {
1435 CXXNameMangler Mangler(Context, os);
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +00001436 Mangler.mangleCXXVtable(RD);
Mike Stumpf1216772009-07-31 18:25:34 +00001437
1438 os.flush();
1439 }
Mike Stump738f8c22009-07-31 23:15:31 +00001440
Anders Carlssond313e402009-10-30 01:52:02 +00001441 void mangleCXXRtti(MangleContext &Context, QualType Ty,
Mike Stump738f8c22009-07-31 23:15:31 +00001442 llvm::raw_ostream &os) {
1443 CXXNameMangler Mangler(Context, os);
Anders Carlsson731f7f92009-10-30 01:26:12 +00001444 Mangler.mangleCXXRtti(Ty);
Mike Stump738f8c22009-07-31 23:15:31 +00001445
1446 os.flush();
1447 }
Mike Stumpf1216772009-07-31 18:25:34 +00001448}