blob: e644ff3e4ea0a3ddbe5b05deb02516decf61830e [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);
Mike Stumpab3f7e92009-11-10 01:41:59 +000056 void mangleCXXCtorVtable(const CXXRecordDecl *RD, int64_t Offset,
57 const CXXRecordDecl *Type);
Anders Carlssond313e402009-10-30 01:52:02 +000058 void mangleCXXRtti(QualType Ty);
Mike Stump67795982009-11-14 00:14:13 +000059 void mangleCXXRttiName(QualType Ty);
Anders Carlsson3ac86b52009-04-15 05:36:58 +000060 void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type);
Anders Carlsson27ae5362009-04-17 01:58:57 +000061 void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type);
Anders Carlsson3ac86b52009-04-15 05:36:58 +000062
Anders Carlsson43f17402009-04-02 15:51:53 +000063 private:
Anders Carlsson6862fc72009-09-17 04:16:28 +000064 bool mangleSubstitution(const NamedDecl *ND);
Anders Carlsson76967372009-09-17 00:43:46 +000065 bool mangleSubstitution(QualType T);
Anders Carlssond3a932a2009-09-17 03:53:28 +000066 bool mangleSubstitution(uintptr_t Ptr);
67
Anders Carlssone7c8cb62009-09-26 20:53:44 +000068 bool mangleStandardSubstitution(const NamedDecl *ND);
69
Anders Carlssond3a932a2009-09-17 03:53:28 +000070 void addSubstitution(const NamedDecl *ND) {
Anders Carlsson433d1372009-11-07 04:26:04 +000071 ND = cast<NamedDecl>(ND->getCanonicalDecl());
72
Anders Carlssond3a932a2009-09-17 03:53:28 +000073 addSubstitution(reinterpret_cast<uintptr_t>(ND));
74 }
Anders Carlsson76967372009-09-17 00:43:46 +000075 void addSubstitution(QualType T);
Anders Carlssond3a932a2009-09-17 03:53:28 +000076 void addSubstitution(uintptr_t Ptr);
Anders Carlsson76967372009-09-17 00:43:46 +000077
Anders Carlsson43f17402009-04-02 15:51:53 +000078 bool mangleFunctionDecl(const FunctionDecl *FD);
Mike Stump1eb44332009-09-09 15:08:12 +000079
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000080 void mangleFunctionEncoding(const FunctionDecl *FD);
81 void mangleName(const NamedDecl *ND);
Anders Carlsson7624f212009-09-18 02:42:01 +000082 void mangleName(const TemplateDecl *TD,
83 const TemplateArgument *TemplateArgs,
84 unsigned NumTemplateArgs);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000085 void mangleUnqualifiedName(const NamedDecl *ND);
Anders Carlsson201ce742009-09-17 03:17:01 +000086 void mangleUnscopedName(const NamedDecl *ND);
Anders Carlsson0fa6df42009-09-26 19:45:45 +000087 void mangleUnscopedTemplateName(const TemplateDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000088 void mangleSourceName(const IdentifierInfo *II);
Anders Carlsson1b42c792009-04-02 16:24:45 +000089 void mangleLocalName(const NamedDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000090 void mangleNestedName(const NamedDecl *ND);
Anders Carlsson7624f212009-09-18 02:42:01 +000091 void mangleNestedName(const TemplateDecl *TD,
92 const TemplateArgument *TemplateArgs,
93 unsigned NumTemplateArgs);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000094 void manglePrefix(const DeclContext *DC);
Anders Carlsson0fa6df42009-09-26 19:45:45 +000095 void mangleTemplatePrefix(const TemplateDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000096 void mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity);
John McCall0953e762009-09-24 19:53:00 +000097 void mangleQualifiers(Qualifiers Quals);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000098 void mangleType(QualType T);
John McCallefe6aee2009-09-05 07:56:18 +000099
100 // Declare manglers for every type class.
101#define ABSTRACT_TYPE(CLASS, PARENT)
102#define NON_CANONICAL_TYPE(CLASS, PARENT)
103#define TYPE(CLASS, PARENT) void mangleType(const CLASS##Type *T);
104#include "clang/AST/TypeNodes.def"
105
106 void mangleType(const TagType*);
107 void mangleBareFunctionType(const FunctionType *T,
108 bool MangleReturnType);
Anders Carlssond553f8c2009-09-21 01:21:10 +0000109 void mangleExpression(const Expr *E);
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000110 void mangleCXXCtorType(CXXCtorType T);
Anders Carlsson27ae5362009-04-17 01:58:57 +0000111 void mangleCXXDtorType(CXXDtorType T);
Mike Stump1eb44332009-09-09 15:08:12 +0000112
Anders Carlsson7624f212009-09-18 02:42:01 +0000113 void mangleTemplateArgs(const TemplateArgument *TemplateArgs,
114 unsigned NumTemplateArgs);
Anders Carlsson068f3472009-09-17 05:31:47 +0000115 void mangleTemplateArgumentList(const TemplateArgumentList &L);
116 void mangleTemplateArgument(const TemplateArgument &A);
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +0000117
118 void mangleTemplateParameter(unsigned Index);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000119 };
120}
121
Anders Carlsson43f17402009-04-02 15:51:53 +0000122static bool isInCLinkageSpecification(const Decl *D) {
Douglas Gregor457e2812009-10-28 16:31:34 +0000123 D = D->getCanonicalDecl();
Mike Stump1eb44332009-09-09 15:08:12 +0000124 for (const DeclContext *DC = D->getDeclContext();
Anders Carlsson43f17402009-04-02 15:51:53 +0000125 !DC->isTranslationUnit(); DC = DC->getParent()) {
Mike Stump1eb44332009-09-09 15:08:12 +0000126 if (const LinkageSpecDecl *Linkage = dyn_cast<LinkageSpecDecl>(DC))
Anders Carlsson43f17402009-04-02 15:51:53 +0000127 return Linkage->getLanguage() == LinkageSpecDecl::lang_c;
128 }
Mike Stump1eb44332009-09-09 15:08:12 +0000129
Anders Carlsson43f17402009-04-02 15:51:53 +0000130 return false;
131}
132
133bool CXXNameMangler::mangleFunctionDecl(const FunctionDecl *FD) {
Mike Stump141c5af2009-09-02 00:25:38 +0000134 // Clang's "overloadable" attribute extension to C/C++ implies name mangling
Anders Carlssona1e16222009-11-07 07:15:03 +0000135 // (always) as does passing a C++ member function and a function
136 // whose name is not a simple identifier.
137 if (!FD->hasAttr<OverloadableAttr>() && !isa<CXXMethodDecl>(FD) &&
138 FD->getDeclName().isIdentifier()) {
Chris Lattner783601d2009-06-13 23:34:16 +0000139 // C functions are not mangled, and "main" is never mangled.
Anders Carlssonb5404912009-10-07 01:06:45 +0000140 if (!Context.getASTContext().getLangOptions().CPlusPlus || FD->isMain())
Chris Lattner783601d2009-06-13 23:34:16 +0000141 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000142
143 // No mangling in an "implicit extern C" header.
Chris Lattner783601d2009-06-13 23:34:16 +0000144 if (FD->getLocation().isValid() &&
Anders Carlssonb5404912009-10-07 01:06:45 +0000145 Context.getASTContext().getSourceManager().
146 isInExternCSystemHeader(FD->getLocation()))
Chris Lattner783601d2009-06-13 23:34:16 +0000147 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000148
Chris Lattner783601d2009-06-13 23:34:16 +0000149 // No name mangling in a C linkage specification.
Anders Carlssona1e16222009-11-07 07:15:03 +0000150 if (isInCLinkageSpecification(FD))
Chris Lattner783601d2009-06-13 23:34:16 +0000151 return false;
152 }
Anders Carlsson43f17402009-04-02 15:51:53 +0000153
154 // If we get here, mangle the decl name!
155 Out << "_Z";
156 mangleFunctionEncoding(FD);
157 return true;
158}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000159
160bool CXXNameMangler::mangle(const NamedDecl *D) {
Mike Stump141c5af2009-09-02 00:25:38 +0000161 // Any decl can be declared with __asm("foo") on it, and this takes precedence
162 // over all other naming in the .o file.
Argyrios Kyrtzidis40b598e2009-06-30 02:34:44 +0000163 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000164 // If we have an asm name, then we use it as the mangling.
165 Out << '\01'; // LLVM IR Marker for __asm("foo")
166 Out << ALA->getLabel();
167 return true;
168 }
Mike Stump1eb44332009-09-09 15:08:12 +0000169
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000170 // <mangled-name> ::= _Z <encoding>
171 // ::= <data name>
172 // ::= <special-name>
173
174 // FIXME: Actually use a visitor to decode these?
Anders Carlsson43f17402009-04-02 15:51:53 +0000175 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
176 return mangleFunctionDecl(FD);
Mike Stump1eb44332009-09-09 15:08:12 +0000177
Anders Carlsson329749c2009-04-02 16:05:20 +0000178 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Anders Carlssonb5404912009-10-07 01:06:45 +0000179 if (!Context.getASTContext().getLangOptions().CPlusPlus ||
Anders Carlsson9ccb0652009-04-11 01:19:45 +0000180 isInCLinkageSpecification(D) ||
181 D->getDeclContext()->isTranslationUnit())
Anders Carlsson329749c2009-04-02 16:05:20 +0000182 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000183
Anders Carlsson329749c2009-04-02 16:05:20 +0000184 Out << "_Z";
185 mangleName(VD);
186 return true;
187 }
Mike Stump1eb44332009-09-09 15:08:12 +0000188
Anders Carlsson43f17402009-04-02 15:51:53 +0000189 return false;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000190}
191
Mike Stump1eb44332009-09-09 15:08:12 +0000192void CXXNameMangler::mangleCXXCtor(const CXXConstructorDecl *D,
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000193 CXXCtorType Type) {
Anders Carlsson27ae5362009-04-17 01:58:57 +0000194 assert(!Structor && "Structor already set!");
195 Structor = D;
196 StructorType = Type;
Mike Stump1eb44332009-09-09 15:08:12 +0000197
Anders Carlsson27ae5362009-04-17 01:58:57 +0000198 mangle(D);
199}
200
Mike Stump1eb44332009-09-09 15:08:12 +0000201void CXXNameMangler::mangleCXXDtor(const CXXDestructorDecl *D,
Anders Carlsson27ae5362009-04-17 01:58:57 +0000202 CXXDtorType Type) {
203 assert(!Structor && "Structor already set!");
204 Structor = D;
205 StructorType = Type;
Mike Stump1eb44332009-09-09 15:08:12 +0000206
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000207 mangle(D);
208}
209
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +0000210void CXXNameMangler::mangleCXXVtable(const CXXRecordDecl *RD) {
Mike Stumpf1216772009-07-31 18:25:34 +0000211 // <special-name> ::= TV <type> # virtual table
212 Out << "_ZTV";
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +0000213 mangleName(RD);
Mike Stumpf1216772009-07-31 18:25:34 +0000214}
215
Mike Stump88e91712009-10-28 01:51:46 +0000216void CXXNameMangler::mangleCXXVTT(const CXXRecordDecl *RD) {
217 // <special-name> ::= TT <type> # VTT structure
218 Out << "_ZTT";
219 mangleName(RD);
220}
221
Mike Stumpab3f7e92009-11-10 01:41:59 +0000222void CXXNameMangler::mangleCXXCtorVtable(const CXXRecordDecl *RD,
223 int64_t Offset,
224 const CXXRecordDecl *Type) {
225 // <special-name> ::= TC <type> <offset number> _ <base type>
226 Out << "_ZTC";
227 mangleName(RD);
228 Out << Offset;
229 Out << "_";
230 mangleName(Type);
231}
232
Anders Carlssond313e402009-10-30 01:52:02 +0000233void CXXNameMangler::mangleCXXRtti(QualType Ty) {
Mike Stump738f8c22009-07-31 23:15:31 +0000234 // <special-name> ::= TI <type> # typeinfo structure
235 Out << "_ZTI";
Anders Carlsson731f7f92009-10-30 01:26:12 +0000236
Anders Carlssond313e402009-10-30 01:52:02 +0000237 mangleType(Ty);
Mike Stump738f8c22009-07-31 23:15:31 +0000238}
239
Mike Stump67795982009-11-14 00:14:13 +0000240void CXXNameMangler::mangleCXXRttiName(QualType Ty) {
241 // <special-name> ::= TS <type> # typeinfo name (null terminated byte string)
242 Out << "_ZTS";
243
244 mangleType(Ty);
245}
246
Mike Stump1eb44332009-09-09 15:08:12 +0000247void CXXNameMangler::mangleGuardVariable(const VarDecl *D) {
248 // <special-name> ::= GV <object name> # Guard variable for one-time
Mike Stumpf1216772009-07-31 18:25:34 +0000249 // # initialization
Anders Carlsson41aa8c12009-04-13 18:02:10 +0000250
251 Out << "_ZGV";
252 mangleName(D);
253}
254
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000255void CXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD) {
256 // <encoding> ::= <function name> <bare-function-type>
257 mangleName(FD);
Mike Stump1eb44332009-09-09 15:08:12 +0000258
Mike Stump141c5af2009-09-02 00:25:38 +0000259 // Whether the mangling of a function type includes the return type depends on
260 // the context and the nature of the function. The rules for deciding whether
261 // the return type is included are:
Mike Stump1eb44332009-09-09 15:08:12 +0000262 //
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000263 // 1. Template functions (names or types) have return types encoded, with
264 // the exceptions listed below.
Mike Stump1eb44332009-09-09 15:08:12 +0000265 // 2. Function types not appearing as part of a function name mangling,
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000266 // e.g. parameters, pointer types, etc., have return type encoded, with the
267 // exceptions listed below.
268 // 3. Non-template function names do not have return types encoded.
269 //
Mike Stump141c5af2009-09-02 00:25:38 +0000270 // The exceptions mentioned in (1) and (2) above, for which the return type is
271 // never included, are
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000272 // 1. Constructors.
273 // 2. Destructors.
274 // 3. Conversion operator functions, e.g. operator int.
275 bool MangleReturnType = false;
Anders Carlsson9234b7f2009-09-17 03:46:43 +0000276 if (FunctionTemplateDecl *PrimaryTemplate = FD->getPrimaryTemplate()) {
277 if (!(isa<CXXConstructorDecl>(FD) || isa<CXXDestructorDecl>(FD) ||
278 isa<CXXConversionDecl>(FD)))
279 MangleReturnType = true;
280
281 // Mangle the type of the primary template.
282 FD = PrimaryTemplate->getTemplatedDecl();
283 }
284
John McCall54e14c42009-10-22 22:37:11 +0000285 // Do the canonicalization out here because parameter types can
286 // undergo additional canonicalization (e.g. array decay).
287 FunctionType *FT = cast<FunctionType>(Context.getASTContext()
288 .getCanonicalType(FD->getType()));
289
290 mangleBareFunctionType(FT, MangleReturnType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000291}
292
293static bool isStdNamespace(const DeclContext *DC) {
294 if (!DC->isNamespace() || !DC->getParent()->isTranslationUnit())
295 return false;
296
297 const NamespaceDecl *NS = cast<NamespaceDecl>(DC);
John McCall9aeed322009-10-01 00:25:31 +0000298 const IdentifierInfo *II = NS->getOriginalNamespace()->getIdentifier();
299 return II && II->isStr("std");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000300}
301
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000302static const TemplateDecl *
303isTemplate(const NamedDecl *ND, const TemplateArgumentList *&TemplateArgs) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000304 // Check if we have a function template.
305 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)){
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000306 if (const TemplateDecl *TD = FD->getPrimaryTemplate()) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000307 TemplateArgs = FD->getTemplateSpecializationArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000308 return TD;
Anders Carlsson2744a062009-09-18 19:00:18 +0000309 }
310 }
311
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000312 // Check if we have a class template.
313 if (const ClassTemplateSpecializationDecl *Spec =
314 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
315 TemplateArgs = &Spec->getTemplateArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000316 return Spec->getSpecializedTemplate();
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000317 }
318
Anders Carlsson2744a062009-09-18 19:00:18 +0000319 return 0;
320}
321
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000322void CXXNameMangler::mangleName(const NamedDecl *ND) {
323 // <name> ::= <nested-name>
324 // ::= <unscoped-name>
325 // ::= <unscoped-template-name> <template-args>
Anders Carlsson201ce742009-09-17 03:17:01 +0000326 // ::= <local-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000327 //
Anders Carlssond58d6f72009-09-17 16:12:20 +0000328 const DeclContext *DC = ND->getDeclContext();
Anders Carlsson5cc58c62009-09-22 17:23:30 +0000329 while (isa<LinkageSpecDecl>(DC))
Anders Carlssond58d6f72009-09-17 16:12:20 +0000330 DC = DC->getParent();
Anders Carlssond58d6f72009-09-17 16:12:20 +0000331
332 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000333 // Check if we have a template.
334 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000335 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000336 mangleUnscopedTemplateName(TD);
337 mangleTemplateArgumentList(*TemplateArgs);
338 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000339 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000340
Anders Carlsson7482e242009-09-18 04:29:09 +0000341 mangleUnscopedName(ND);
342 return;
343 }
344
345 if (isa<FunctionDecl>(DC)) {
346 mangleLocalName(ND);
347 return;
348 }
349
350 mangleNestedName(ND);
351}
Anders Carlsson7624f212009-09-18 02:42:01 +0000352void CXXNameMangler::mangleName(const TemplateDecl *TD,
353 const TemplateArgument *TemplateArgs,
354 unsigned NumTemplateArgs) {
355 const DeclContext *DC = TD->getDeclContext();
356 while (isa<LinkageSpecDecl>(DC)) {
357 assert(cast<LinkageSpecDecl>(DC)->getLanguage() ==
358 LinkageSpecDecl::lang_cxx && "Unexpected linkage decl!");
359 DC = DC->getParent();
360 }
361
362 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000363 mangleUnscopedTemplateName(TD);
Anders Carlsson7624f212009-09-18 02:42:01 +0000364 mangleTemplateArgs(TemplateArgs, NumTemplateArgs);
365 } else {
366 mangleNestedName(TD, TemplateArgs, NumTemplateArgs);
367 }
368}
369
Anders Carlsson201ce742009-09-17 03:17:01 +0000370void CXXNameMangler::mangleUnscopedName(const NamedDecl *ND) {
371 // <unscoped-name> ::= <unqualified-name>
372 // ::= St <unqualified-name> # ::std::
373 if (isStdNamespace(ND->getDeclContext()))
374 Out << "St";
375
376 mangleUnqualifiedName(ND);
377}
378
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000379void CXXNameMangler::mangleUnscopedTemplateName(const TemplateDecl *ND) {
Anders Carlsson201ce742009-09-17 03:17:01 +0000380 // <unscoped-template-name> ::= <unscoped-name>
381 // ::= <substitution>
Anders Carlsson7624f212009-09-18 02:42:01 +0000382 if (mangleSubstitution(ND))
Anders Carlsson03c9d532009-09-17 04:02:31 +0000383 return;
384
Anders Carlsson1668f202009-09-26 20:13:56 +0000385 mangleUnscopedName(ND->getTemplatedDecl());
Anders Carlsson7624f212009-09-18 02:42:01 +0000386 addSubstitution(ND);
Anders Carlsson201ce742009-09-17 03:17:01 +0000387}
388
Mike Stump77ca8f62009-09-05 07:20:32 +0000389void CXXNameMangler::mangleCalloffset(int64_t nv, int64_t v) {
Mike Stump141c5af2009-09-02 00:25:38 +0000390 // <call-offset> ::= h <nv-offset> _
391 // ::= v <v-offset> _
392 // <nv-offset> ::= <offset number> # non-virtual base override
393 // <v-offset> ::= <offset nubmer> _ <virtual offset number>
394 // # virtual base override, with vcall offset
Mike Stump77ca8f62009-09-05 07:20:32 +0000395 if (v == 0) {
Mike Stump9124bcc2009-09-02 00:56:18 +0000396 Out << "h";
Mike Stump141c5af2009-09-02 00:25:38 +0000397 if (nv < 0) {
398 Out << "n";
399 nv = -nv;
400 }
401 Out << nv;
402 } else {
Mike Stump9124bcc2009-09-02 00:56:18 +0000403 Out << "v";
Mike Stump141c5af2009-09-02 00:25:38 +0000404 if (nv < 0) {
405 Out << "n";
406 nv = -nv;
407 }
408 Out << nv;
409 Out << "_";
410 if (v < 0) {
411 Out << "n";
412 v = -v;
413 }
414 Out << v;
415 }
416 Out << "_";
Mike Stump9124bcc2009-09-02 00:56:18 +0000417}
418
Mike Stumpdec025b2009-09-07 04:27:52 +0000419void CXXNameMangler::mangleThunk(const FunctionDecl *FD, int64_t nv,
420 int64_t v) {
Mike Stump9124bcc2009-09-02 00:56:18 +0000421 // <special-name> ::= T <call-offset> <base encoding>
422 // # base is the nominal target function of thunk
Mike Stumpdec025b2009-09-07 04:27:52 +0000423 Out << "_ZT";
Mike Stump77ca8f62009-09-05 07:20:32 +0000424 mangleCalloffset(nv, v);
Mike Stumpdec025b2009-09-07 04:27:52 +0000425 mangleFunctionEncoding(FD);
Mike Stump9124bcc2009-09-02 00:56:18 +0000426}
427
Mike Stumpdec025b2009-09-07 04:27:52 +0000428 void CXXNameMangler::mangleCovariantThunk(const FunctionDecl *FD,
Mike Stump77ca8f62009-09-05 07:20:32 +0000429 int64_t nv_t, int64_t v_t,
Mike Stump9124bcc2009-09-02 00:56:18 +0000430 int64_t nv_r, int64_t v_r) {
431 // <special-name> ::= Tc <call-offset> <call-offset> <base encoding>
432 // # base is the nominal target function of thunk
433 // # first call-offset is 'this' adjustment
434 // # second call-offset is result adjustment
Mike Stumpdec025b2009-09-07 04:27:52 +0000435 Out << "_ZTc";
Mike Stump77ca8f62009-09-05 07:20:32 +0000436 mangleCalloffset(nv_t, v_t);
437 mangleCalloffset(nv_r, v_r);
Mike Stumpdec025b2009-09-07 04:27:52 +0000438 mangleFunctionEncoding(FD);
Mike Stump141c5af2009-09-02 00:25:38 +0000439}
440
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000441void CXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND) {
442 // <unqualified-name> ::= <operator-name>
Mike Stump1eb44332009-09-09 15:08:12 +0000443 // ::= <ctor-dtor-name>
444 // ::= <source-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000445 DeclarationName Name = ND->getDeclName();
446 switch (Name.getNameKind()) {
Anders Carlssonc4355b62009-10-07 01:45:02 +0000447 case DeclarationName::Identifier: {
448 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
John McCall9aeed322009-10-01 00:25:31 +0000449 if (NS->isAnonymousNamespace()) {
450 // This is how gcc mangles these names. It's apparently
451 // always '1', no matter how many different anonymous
452 // namespaces appear in a context.
453 Out << "12_GLOBAL__N_1";
454 break;
455 }
Anders Carlssonc4355b62009-10-07 01:45:02 +0000456 }
457
458 if (const IdentifierInfo *II = Name.getAsIdentifierInfo()) {
459 mangleSourceName(II);
460 break;
461 }
462
463 // We must have an anonymous struct.
464 const TagDecl *TD = cast<TagDecl>(ND);
465 if (const TypedefDecl *D = TD->getTypedefForAnonDecl()) {
466 assert(TD->getDeclContext() == D->getDeclContext() &&
467 "Typedef should not be in another decl context!");
468 assert(D->getDeclName().getAsIdentifierInfo() &&
469 "Typedef was not named!");
470 mangleSourceName(D->getDeclName().getAsIdentifierInfo());
471 break;
472 }
473
474 // Get a unique id for the anonymous struct.
475 uint64_t AnonStructId = Context.getAnonymousStructId(TD);
476
477 // Mangle it as a source name in the form
478 // [n] $_<id>
479 // where n is the length of the string.
480 llvm::SmallString<8> Str;
481 Str += "$_";
482 Str += llvm::utostr(AnonStructId);
483
484 Out << Str.size();
485 Out << Str.str();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000486 break;
Anders Carlssonc4355b62009-10-07 01:45:02 +0000487 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000488
489 case DeclarationName::ObjCZeroArgSelector:
490 case DeclarationName::ObjCOneArgSelector:
491 case DeclarationName::ObjCMultiArgSelector:
492 assert(false && "Can't mangle Objective-C selector names here!");
493 break;
494
495 case DeclarationName::CXXConstructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000496 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000497 // If the named decl is the C++ constructor we're mangling, use the type
498 // we were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000499 mangleCXXCtorType(static_cast<CXXCtorType>(StructorType));
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000500 else
501 // Otherwise, use the complete constructor name. This is relevant if a
502 // class with a constructor is declared within a constructor.
503 mangleCXXCtorType(Ctor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000504 break;
505
506 case DeclarationName::CXXDestructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000507 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000508 // If the named decl is the C++ destructor we're mangling, use the type we
509 // were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000510 mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
511 else
512 // Otherwise, use the complete destructor name. This is relevant if a
513 // class with a destructor is declared within a destructor.
514 mangleCXXDtorType(Dtor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000515 break;
516
517 case DeclarationName::CXXConversionFunctionName:
Mike Stump1eb44332009-09-09 15:08:12 +0000518 // <operator-name> ::= cv <type> # (cast)
Douglas Gregor219cc612009-02-13 01:28:03 +0000519 Out << "cv";
Anders Carlssonb5404912009-10-07 01:06:45 +0000520 mangleType(Context.getASTContext().getCanonicalType(Name.getCXXNameType()));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000521 break;
522
523 case DeclarationName::CXXOperatorName:
524 mangleOperatorName(Name.getCXXOverloadedOperator(),
525 cast<FunctionDecl>(ND)->getNumParams());
526 break;
527
528 case DeclarationName::CXXUsingDirective:
529 assert(false && "Can't mangle a using directive name!");
Douglas Gregor219cc612009-02-13 01:28:03 +0000530 break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000531 }
532}
533
534void CXXNameMangler::mangleSourceName(const IdentifierInfo *II) {
535 // <source-name> ::= <positive length number> <identifier>
536 // <number> ::= [n] <non-negative decimal integer>
537 // <identifier> ::= <unqualified source code identifier>
538 Out << II->getLength() << II->getName();
539}
540
541void CXXNameMangler::mangleNestedName(const NamedDecl *ND) {
542 // <nested-name> ::= N [<CV-qualifiers>] <prefix> <unqualified-name> E
543 // ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
Anders Carlssond99edc42009-09-26 03:55:37 +0000544
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000545 Out << 'N';
546 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(ND))
John McCall0953e762009-09-24 19:53:00 +0000547 mangleQualifiers(Qualifiers::fromCVRMask(Method->getTypeQualifiers()));
Anders Carlsson7482e242009-09-18 04:29:09 +0000548
Anders Carlsson2744a062009-09-18 19:00:18 +0000549 // Check if we have a template.
550 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000551 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000552 mangleTemplatePrefix(TD);
553 mangleTemplateArgumentList(*TemplateArgs);
Anders Carlsson7482e242009-09-18 04:29:09 +0000554 } else {
555 manglePrefix(ND->getDeclContext());
556 mangleUnqualifiedName(ND);
557 }
558
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000559 Out << 'E';
560}
Anders Carlsson7624f212009-09-18 02:42:01 +0000561void CXXNameMangler::mangleNestedName(const TemplateDecl *TD,
562 const TemplateArgument *TemplateArgs,
563 unsigned NumTemplateArgs) {
Anders Carlssone45117b2009-09-27 19:53:49 +0000564 // <nested-name> ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
565
Anders Carlsson7624f212009-09-18 02:42:01 +0000566 Out << 'N';
Anders Carlsson7624f212009-09-18 02:42:01 +0000567
Anders Carlssone45117b2009-09-27 19:53:49 +0000568 mangleTemplatePrefix(TD);
Anders Carlsson7624f212009-09-18 02:42:01 +0000569 mangleTemplateArgs(TemplateArgs, NumTemplateArgs);
Anders Carlssone45117b2009-09-27 19:53:49 +0000570
Anders Carlsson7624f212009-09-18 02:42:01 +0000571 Out << 'E';
572}
573
Anders Carlsson1b42c792009-04-02 16:24:45 +0000574void CXXNameMangler::mangleLocalName(const NamedDecl *ND) {
575 // <local-name> := Z <function encoding> E <entity name> [<discriminator>]
576 // := Z <function encoding> E s [<discriminator>]
Mike Stump1eb44332009-09-09 15:08:12 +0000577 // <discriminator> := _ <non-negative number>
Anders Carlsson1b42c792009-04-02 16:24:45 +0000578 Out << 'Z';
579 mangleFunctionEncoding(cast<FunctionDecl>(ND->getDeclContext()));
580 Out << 'E';
581 mangleSourceName(ND->getIdentifier());
582}
583
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000584void CXXNameMangler::manglePrefix(const DeclContext *DC) {
585 // <prefix> ::= <prefix> <unqualified-name>
586 // ::= <template-prefix> <template-args>
587 // ::= <template-param>
588 // ::= # empty
589 // ::= <substitution>
590 // FIXME: We only handle mangling of namespaces and classes at the moment.
Anders Carlsson6862fc72009-09-17 04:16:28 +0000591
Anders Carlssonadd28822009-09-22 20:33:31 +0000592 while (isa<LinkageSpecDecl>(DC))
593 DC = DC->getParent();
594
Anders Carlsson9263e912009-09-18 18:39:58 +0000595 if (DC->isTranslationUnit())
596 return;
597
Anders Carlsson6862fc72009-09-17 04:16:28 +0000598 if (mangleSubstitution(cast<NamedDecl>(DC)))
599 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000600
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000601 // Check if we have a template.
602 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000603 if (const TemplateDecl *TD = isTemplate(cast<NamedDecl>(DC), TemplateArgs)) {
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000604 mangleTemplatePrefix(TD);
605 mangleTemplateArgumentList(*TemplateArgs);
606 } else {
607 manglePrefix(DC->getParent());
608 mangleUnqualifiedName(cast<NamedDecl>(DC));
609 }
Anders Carlsson6862fc72009-09-17 04:16:28 +0000610
611 addSubstitution(cast<NamedDecl>(DC));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000612}
613
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000614void CXXNameMangler::mangleTemplatePrefix(const TemplateDecl *ND) {
Anders Carlsson7482e242009-09-18 04:29:09 +0000615 // <template-prefix> ::= <prefix> <template unqualified-name>
616 // ::= <template-param>
617 // ::= <substitution>
618
Anders Carlssonaeb85372009-09-26 22:18:22 +0000619 if (mangleSubstitution(ND))
620 return;
621
622 // FIXME: <template-param>
Anders Carlssonaa73ab12009-09-18 18:47:07 +0000623
624 manglePrefix(ND->getDeclContext());
Anders Carlsson1668f202009-09-26 20:13:56 +0000625 mangleUnqualifiedName(ND->getTemplatedDecl());
Anders Carlssonaeb85372009-09-26 22:18:22 +0000626
627 addSubstitution(ND);
Anders Carlsson7482e242009-09-18 04:29:09 +0000628}
629
Mike Stump1eb44332009-09-09 15:08:12 +0000630void
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000631CXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity) {
632 switch (OO) {
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000633 // <operator-name> ::= nw # new
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000634 case OO_New: Out << "nw"; break;
635 // ::= na # new[]
636 case OO_Array_New: Out << "na"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000637 // ::= dl # delete
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000638 case OO_Delete: Out << "dl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000639 // ::= da # delete[]
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000640 case OO_Array_Delete: Out << "da"; break;
641 // ::= ps # + (unary)
642 // ::= pl # +
643 case OO_Plus: Out << (Arity == 1? "ps" : "pl"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000644 // ::= ng # - (unary)
645 // ::= mi # -
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000646 case OO_Minus: Out << (Arity == 1? "ng" : "mi"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000647 // ::= ad # & (unary)
648 // ::= an # &
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000649 case OO_Amp: Out << (Arity == 1? "ad" : "an"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000650 // ::= de # * (unary)
651 // ::= ml # *
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000652 case OO_Star: Out << (Arity == 1? "de" : "ml"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000653 // ::= co # ~
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000654 case OO_Tilde: Out << "co"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000655 // ::= dv # /
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000656 case OO_Slash: Out << "dv"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000657 // ::= rm # %
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000658 case OO_Percent: Out << "rm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000659 // ::= or # |
660 case OO_Pipe: Out << "or"; break;
661 // ::= eo # ^
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000662 case OO_Caret: Out << "eo"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000663 // ::= aS # =
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000664 case OO_Equal: Out << "aS"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000665 // ::= pL # +=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000666 case OO_PlusEqual: Out << "pL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000667 // ::= mI # -=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000668 case OO_MinusEqual: Out << "mI"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000669 // ::= mL # *=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000670 case OO_StarEqual: Out << "mL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000671 // ::= dV # /=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000672 case OO_SlashEqual: Out << "dV"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000673 // ::= rM # %=
674 case OO_PercentEqual: Out << "rM"; break;
675 // ::= aN # &=
676 case OO_AmpEqual: Out << "aN"; break;
677 // ::= oR # |=
678 case OO_PipeEqual: Out << "oR"; break;
679 // ::= eO # ^=
680 case OO_CaretEqual: Out << "eO"; break;
681 // ::= ls # <<
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000682 case OO_LessLess: Out << "ls"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000683 // ::= rs # >>
684 case OO_GreaterGreater: Out << "rs"; break;
685 // ::= lS # <<=
686 case OO_LessLessEqual: Out << "lS"; break;
687 // ::= rS # >>=
688 case OO_GreaterGreaterEqual: Out << "rS"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000689 // ::= eq # ==
690 case OO_EqualEqual: Out << "eq"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000691 // ::= ne # !=
692 case OO_ExclaimEqual: Out << "ne"; break;
693 // ::= lt # <
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000694 case OO_Less: Out << "lt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000695 // ::= gt # >
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000696 case OO_Greater: Out << "gt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000697 // ::= le # <=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000698 case OO_LessEqual: Out << "le"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000699 // ::= ge # >=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000700 case OO_GreaterEqual: Out << "ge"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000701 // ::= nt # !
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000702 case OO_Exclaim: Out << "nt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000703 // ::= aa # &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000704 case OO_AmpAmp: Out << "aa"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000705 // ::= oo # ||
706 case OO_PipePipe: Out << "oo"; break;
707 // ::= pp # ++
708 case OO_PlusPlus: Out << "pp"; break;
709 // ::= mm # --
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000710 case OO_MinusMinus: Out << "mm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000711 // ::= cm # ,
712 case OO_Comma: Out << "cm"; break;
713 // ::= pm # ->*
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000714 case OO_ArrowStar: Out << "pm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000715 // ::= pt # ->
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000716 case OO_Arrow: Out << "pt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000717 // ::= cl # ()
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000718 case OO_Call: Out << "cl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000719 // ::= ix # []
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000720 case OO_Subscript: Out << "ix"; break;
721 // UNSUPPORTED: ::= qu # ?
722
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000723 case OO_None:
724 case OO_Conditional:
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000725 case NUM_OVERLOADED_OPERATORS:
Mike Stump1eb44332009-09-09 15:08:12 +0000726 assert(false && "Not an overloaded operator");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000727 break;
728 }
729}
730
John McCall0953e762009-09-24 19:53:00 +0000731void CXXNameMangler::mangleQualifiers(Qualifiers Quals) {
Mike Stump1eb44332009-09-09 15:08:12 +0000732 // <CV-qualifiers> ::= [r] [V] [K] # restrict (C99), volatile, const
John McCall0953e762009-09-24 19:53:00 +0000733 if (Quals.hasRestrict())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000734 Out << 'r';
John McCall0953e762009-09-24 19:53:00 +0000735 if (Quals.hasVolatile())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000736 Out << 'V';
John McCall0953e762009-09-24 19:53:00 +0000737 if (Quals.hasConst())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000738 Out << 'K';
John McCall0953e762009-09-24 19:53:00 +0000739
740 // FIXME: For now, just drop all extension qualifiers on the floor.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000741}
742
743void CXXNameMangler::mangleType(QualType T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000744 // Only operate on the canonical type!
Anders Carlssonb5404912009-10-07 01:06:45 +0000745 T = Context.getASTContext().getCanonicalType(T);
Anders Carlsson4843e582009-03-10 17:07:44 +0000746
Anders Carlsson410f7642009-10-16 02:06:06 +0000747 bool IsSubstitutable = T.hasQualifiers() || !isa<BuiltinType>(T);
Anders Carlsson76967372009-09-17 00:43:46 +0000748 if (IsSubstitutable && mangleSubstitution(T))
749 return;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000750
John McCall0953e762009-09-24 19:53:00 +0000751 if (Qualifiers Quals = T.getQualifiers()) {
752 mangleQualifiers(Quals);
753 // Recurse: even if the qualified type isn't yet substitutable,
754 // the unqualified type might be.
Anders Carlsson76967372009-09-17 00:43:46 +0000755 mangleType(T.getUnqualifiedType());
756 } else {
757 switch (T->getTypeClass()) {
John McCallefe6aee2009-09-05 07:56:18 +0000758#define ABSTRACT_TYPE(CLASS, PARENT)
759#define NON_CANONICAL_TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000760 case Type::CLASS: \
761 llvm::llvm_unreachable("can't mangle non-canonical type " #CLASS "Type"); \
762 return;
John McCallefe6aee2009-09-05 07:56:18 +0000763#define TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000764 case Type::CLASS: \
John McCall0953e762009-09-24 19:53:00 +0000765 mangleType(static_cast<const CLASS##Type*>(T.getTypePtr())); \
Anders Carlsson76967372009-09-17 00:43:46 +0000766 break;
John McCallefe6aee2009-09-05 07:56:18 +0000767#include "clang/AST/TypeNodes.def"
Anders Carlsson76967372009-09-17 00:43:46 +0000768 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000769 }
Anders Carlsson76967372009-09-17 00:43:46 +0000770
771 // Add the substitution.
772 if (IsSubstitutable)
773 addSubstitution(T);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000774}
775
776void CXXNameMangler::mangleType(const BuiltinType *T) {
John McCallefe6aee2009-09-05 07:56:18 +0000777 // <type> ::= <builtin-type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000778 // <builtin-type> ::= v # void
779 // ::= w # wchar_t
780 // ::= b # bool
781 // ::= c # char
782 // ::= a # signed char
783 // ::= h # unsigned char
784 // ::= s # short
785 // ::= t # unsigned short
786 // ::= i # int
787 // ::= j # unsigned int
788 // ::= l # long
789 // ::= m # unsigned long
790 // ::= x # long long, __int64
791 // ::= y # unsigned long long, __int64
792 // ::= n # __int128
793 // UNSUPPORTED: ::= o # unsigned __int128
794 // ::= f # float
795 // ::= d # double
796 // ::= e # long double, __float80
797 // UNSUPPORTED: ::= g # __float128
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000798 // UNSUPPORTED: ::= Dd # IEEE 754r decimal floating point (64 bits)
799 // UNSUPPORTED: ::= De # IEEE 754r decimal floating point (128 bits)
800 // UNSUPPORTED: ::= Df # IEEE 754r decimal floating point (32 bits)
801 // UNSUPPORTED: ::= Dh # IEEE 754r half-precision floating point (16 bits)
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000802 // ::= Di # char32_t
803 // ::= Ds # char16_t
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000804 // ::= u <source-name> # vendor extended type
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000805 // From our point of view, std::nullptr_t is a builtin, but as far as mangling
806 // is concerned, it's a type called std::nullptr_t.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000807 switch (T->getKind()) {
808 case BuiltinType::Void: Out << 'v'; break;
809 case BuiltinType::Bool: Out << 'b'; break;
810 case BuiltinType::Char_U: case BuiltinType::Char_S: Out << 'c'; break;
811 case BuiltinType::UChar: Out << 'h'; break;
812 case BuiltinType::UShort: Out << 't'; break;
813 case BuiltinType::UInt: Out << 'j'; break;
814 case BuiltinType::ULong: Out << 'm'; break;
815 case BuiltinType::ULongLong: Out << 'y'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000816 case BuiltinType::UInt128: Out << 'o'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000817 case BuiltinType::SChar: Out << 'a'; break;
818 case BuiltinType::WChar: Out << 'w'; break;
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000819 case BuiltinType::Char16: Out << "Ds"; break;
820 case BuiltinType::Char32: Out << "Di"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000821 case BuiltinType::Short: Out << 's'; break;
822 case BuiltinType::Int: Out << 'i'; break;
823 case BuiltinType::Long: Out << 'l'; break;
824 case BuiltinType::LongLong: Out << 'x'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000825 case BuiltinType::Int128: Out << 'n'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000826 case BuiltinType::Float: Out << 'f'; break;
827 case BuiltinType::Double: Out << 'd'; break;
828 case BuiltinType::LongDouble: Out << 'e'; break;
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000829 case BuiltinType::NullPtr: Out << "St9nullptr_t"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000830
831 case BuiltinType::Overload:
832 case BuiltinType::Dependent:
Mike Stump1eb44332009-09-09 15:08:12 +0000833 assert(false &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000834 "Overloaded and dependent types shouldn't get to name mangling");
835 break;
Anders Carlssone89d1592009-06-26 18:41:36 +0000836 case BuiltinType::UndeducedAuto:
837 assert(0 && "Should not see undeduced auto here");
838 break;
Steve Naroff9533a7f2009-07-22 17:14:51 +0000839 case BuiltinType::ObjCId: Out << "11objc_object"; break;
840 case BuiltinType::ObjCClass: Out << "10objc_class"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000841 }
842}
843
John McCallefe6aee2009-09-05 07:56:18 +0000844// <type> ::= <function-type>
845// <function-type> ::= F [Y] <bare-function-type> E
846void CXXNameMangler::mangleType(const FunctionProtoType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000847 Out << 'F';
Mike Stumpf5408fe2009-05-16 07:57:57 +0000848 // FIXME: We don't have enough information in the AST to produce the 'Y'
849 // encoding for extern "C" function types.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000850 mangleBareFunctionType(T, /*MangleReturnType=*/true);
851 Out << 'E';
852}
John McCallefe6aee2009-09-05 07:56:18 +0000853void CXXNameMangler::mangleType(const FunctionNoProtoType *T) {
854 llvm::llvm_unreachable("Can't mangle K&R function prototypes");
855}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000856void CXXNameMangler::mangleBareFunctionType(const FunctionType *T,
857 bool MangleReturnType) {
John McCallefe6aee2009-09-05 07:56:18 +0000858 // We should never be mangling something without a prototype.
859 const FunctionProtoType *Proto = cast<FunctionProtoType>(T);
860
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000861 // <bare-function-type> ::= <signature type>+
862 if (MangleReturnType)
John McCallefe6aee2009-09-05 07:56:18 +0000863 mangleType(Proto->getResultType());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000864
Anders Carlssonc6c91bc2009-04-01 00:15:23 +0000865 if (Proto->getNumArgs() == 0) {
866 Out << 'v';
867 return;
868 }
Mike Stump1eb44332009-09-09 15:08:12 +0000869
Douglas Gregor72564e72009-02-26 23:50:07 +0000870 for (FunctionProtoType::arg_type_iterator Arg = Proto->arg_type_begin(),
Mike Stump1eb44332009-09-09 15:08:12 +0000871 ArgEnd = Proto->arg_type_end();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000872 Arg != ArgEnd; ++Arg)
873 mangleType(*Arg);
Douglas Gregor219cc612009-02-13 01:28:03 +0000874
875 // <builtin-type> ::= z # ellipsis
876 if (Proto->isVariadic())
877 Out << 'z';
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000878}
879
John McCallefe6aee2009-09-05 07:56:18 +0000880// <type> ::= <class-enum-type>
Mike Stump1eb44332009-09-09 15:08:12 +0000881// <class-enum-type> ::= <name>
John McCallefe6aee2009-09-05 07:56:18 +0000882void CXXNameMangler::mangleType(const EnumType *T) {
883 mangleType(static_cast<const TagType*>(T));
884}
885void CXXNameMangler::mangleType(const RecordType *T) {
886 mangleType(static_cast<const TagType*>(T));
887}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000888void CXXNameMangler::mangleType(const TagType *T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000889 if (!T->getDecl()->getIdentifier())
890 mangleName(T->getDecl()->getTypedefForAnonDecl());
891 else
892 mangleName(T->getDecl());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000893}
894
John McCallefe6aee2009-09-05 07:56:18 +0000895// <type> ::= <array-type>
896// <array-type> ::= A <positive dimension number> _ <element type>
897// ::= A [<dimension expression>] _ <element type>
898void CXXNameMangler::mangleType(const ConstantArrayType *T) {
899 Out << 'A' << T->getSize() << '_';
900 mangleType(T->getElementType());
901}
902void CXXNameMangler::mangleType(const VariableArrayType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000903 Out << 'A';
John McCallefe6aee2009-09-05 07:56:18 +0000904 mangleExpression(T->getSizeExpr());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000905 Out << '_';
906 mangleType(T->getElementType());
907}
John McCallefe6aee2009-09-05 07:56:18 +0000908void CXXNameMangler::mangleType(const DependentSizedArrayType *T) {
909 Out << 'A';
910 mangleExpression(T->getSizeExpr());
911 Out << '_';
912 mangleType(T->getElementType());
913}
914void CXXNameMangler::mangleType(const IncompleteArrayType *T) {
915 Out << 'A' << '_';
916 mangleType(T->getElementType());
917}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000918
John McCallefe6aee2009-09-05 07:56:18 +0000919// <type> ::= <pointer-to-member-type>
920// <pointer-to-member-type> ::= M <class type> <member type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000921void CXXNameMangler::mangleType(const MemberPointerType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000922 Out << 'M';
923 mangleType(QualType(T->getClass(), 0));
Anders Carlsson0e650012009-05-17 17:41:20 +0000924 QualType PointeeType = T->getPointeeType();
925 if (const FunctionProtoType *FPT = dyn_cast<FunctionProtoType>(PointeeType)) {
John McCall0953e762009-09-24 19:53:00 +0000926 mangleQualifiers(Qualifiers::fromCVRMask(FPT->getTypeQuals()));
Anders Carlsson0e650012009-05-17 17:41:20 +0000927 mangleType(FPT);
Mike Stump1eb44332009-09-09 15:08:12 +0000928 } else
Anders Carlsson0e650012009-05-17 17:41:20 +0000929 mangleType(PointeeType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000930}
931
John McCallefe6aee2009-09-05 07:56:18 +0000932// <type> ::= <template-param>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000933void CXXNameMangler::mangleType(const TemplateTypeParmType *T) {
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +0000934 mangleTemplateParameter(T->getIndex());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000935}
936
John McCallefe6aee2009-09-05 07:56:18 +0000937// FIXME: <type> ::= <template-template-param> <template-args>
John McCallefe6aee2009-09-05 07:56:18 +0000938
939// <type> ::= P <type> # pointer-to
940void CXXNameMangler::mangleType(const PointerType *T) {
941 Out << 'P';
942 mangleType(T->getPointeeType());
943}
944void CXXNameMangler::mangleType(const ObjCObjectPointerType *T) {
945 Out << 'P';
946 mangleType(T->getPointeeType());
947}
948
949// <type> ::= R <type> # reference-to
950void CXXNameMangler::mangleType(const LValueReferenceType *T) {
951 Out << 'R';
952 mangleType(T->getPointeeType());
953}
954
955// <type> ::= O <type> # rvalue reference-to (C++0x)
956void CXXNameMangler::mangleType(const RValueReferenceType *T) {
957 Out << 'O';
958 mangleType(T->getPointeeType());
959}
960
961// <type> ::= C <type> # complex pair (C 2000)
962void CXXNameMangler::mangleType(const ComplexType *T) {
963 Out << 'C';
964 mangleType(T->getElementType());
965}
966
967// GNU extension: vector types
968void CXXNameMangler::mangleType(const VectorType *T) {
969 Out << "U8__vector";
970 mangleType(T->getElementType());
971}
972void CXXNameMangler::mangleType(const ExtVectorType *T) {
973 mangleType(static_cast<const VectorType*>(T));
974}
975void CXXNameMangler::mangleType(const DependentSizedExtVectorType *T) {
976 Out << "U8__vector";
977 mangleType(T->getElementType());
978}
979
Anders Carlssona40c5e42009-03-07 22:03:21 +0000980void CXXNameMangler::mangleType(const ObjCInterfaceType *T) {
981 mangleSourceName(T->getDecl()->getIdentifier());
982}
983
John McCallefe6aee2009-09-05 07:56:18 +0000984void CXXNameMangler::mangleType(const BlockPointerType *T) {
985 assert(false && "can't mangle block pointer types yet");
986}
987
988void CXXNameMangler::mangleType(const FixedWidthIntType *T) {
989 assert(false && "can't mangle arbitary-precision integer type yet");
990}
991
992void CXXNameMangler::mangleType(const TemplateSpecializationType *T) {
Anders Carlsson7624f212009-09-18 02:42:01 +0000993 TemplateDecl *TD = T->getTemplateName().getAsTemplateDecl();
994 assert(TD && "FIXME: Support dependent template names!");
995
996 mangleName(TD, T->getArgs(), T->getNumArgs());
John McCallefe6aee2009-09-05 07:56:18 +0000997}
998
999void CXXNameMangler::mangleType(const TypenameType *T) {
Anders Carlssonae352482009-09-26 02:26:02 +00001000 // Typename types are always nested
1001 Out << 'N';
1002
1003 const Type *QTy = T->getQualifier()->getAsType();
1004 if (const TemplateSpecializationType *TST =
1005 dyn_cast<TemplateSpecializationType>(QTy)) {
Anders Carlsson88599172009-09-27 01:06:07 +00001006 if (!mangleSubstitution(QualType(TST, 0))) {
1007 TemplateDecl *TD = TST->getTemplateName().getAsTemplateDecl();
Anders Carlssonae352482009-09-26 02:26:02 +00001008
Anders Carlsson88599172009-09-27 01:06:07 +00001009 mangleTemplatePrefix(TD);
1010 mangleTemplateArgs(TST->getArgs(), TST->getNumArgs());
1011 addSubstitution(QualType(TST, 0));
1012 }
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001013 } else if (const TemplateTypeParmType *TTPT =
1014 dyn_cast<TemplateTypeParmType>(QTy)) {
1015 // We use the QualType mangle type variant here because it handles
1016 // substitutions.
1017 mangleType(QualType(TTPT, 0));
Anders Carlssonae352482009-09-26 02:26:02 +00001018 } else
1019 assert(false && "Unhandled type!");
1020
1021 mangleSourceName(T->getIdentifier());
1022
1023 Out << 'E';
John McCallefe6aee2009-09-05 07:56:18 +00001024}
1025
Anders Carlssond553f8c2009-09-21 01:21:10 +00001026void CXXNameMangler::mangleExpression(const Expr *E) {
1027 // <expression> ::= <unary operator-name> <expression>
1028 // ::= <binary operator-name> <expression> <expression>
1029 // ::= <trinary operator-name> <expression> <expression> <expression>
1030 // ::= cl <expression>* E # call
1031 // ::= cv <type> expression # conversion with one argument
1032 // ::= cv <type> _ <expression>* E # conversion with a different number of arguments
1033 // ::= st <type> # sizeof (a type)
1034 // ::= at <type> # alignof (a type)
1035 // ::= <template-param>
1036 // ::= <function-param>
1037 // ::= sr <type> <unqualified-name> # dependent name
1038 // ::= sr <type> <unqualified-name> <template-args> # dependent template-id
1039 // ::= sZ <template-param> # size of a parameter pack
1040 // ::= <expr-primary>
1041 switch (E->getStmtClass()) {
1042 default: assert(false && "Unhandled expression kind!");
Anders Carlssona7694082009-11-06 02:50:19 +00001043
1044 case Expr::ParenExprClass:
1045 mangleExpression(cast<ParenExpr>(E)->getSubExpr());
1046 break;
1047
Anders Carlssond553f8c2009-09-21 01:21:10 +00001048 case Expr::DeclRefExprClass: {
1049 const Decl *D = cast<DeclRefExpr>(E)->getDecl();
1050
1051 switch (D->getKind()) {
1052 default: assert(false && "Unhandled decl kind!");
1053 case Decl::NonTypeTemplateParm: {
1054 const NonTypeTemplateParmDecl *PD = cast<NonTypeTemplateParmDecl>(D);
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001055 mangleTemplateParameter(PD->getIndex());
Anders Carlssond553f8c2009-09-21 01:21:10 +00001056 break;
1057 }
1058
1059 }
Anders Carlsson50755b02009-09-27 20:11:34 +00001060
1061 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001062 }
Anders Carlsson50755b02009-09-27 20:11:34 +00001063
1064 case Expr::UnresolvedDeclRefExprClass: {
1065 const UnresolvedDeclRefExpr *DRE = cast<UnresolvedDeclRefExpr>(E);
1066 const Type *QTy = DRE->getQualifier()->getAsType();
1067 assert(QTy && "Qualifier was not type!");
1068
1069 // ::= sr <type> <unqualified-name> # dependent name
1070 Out << "sr";
1071 mangleType(QualType(QTy, 0));
1072
1073 assert(DRE->getDeclName().getNameKind() == DeclarationName::Identifier &&
1074 "Unhandled decl name kind!");
1075 mangleSourceName(DRE->getDeclName().getAsIdentifierInfo());
1076
1077 break;
1078 }
1079
Anders Carlssond553f8c2009-09-21 01:21:10 +00001080 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001081}
1082
John McCallefe6aee2009-09-05 07:56:18 +00001083// FIXME: <type> ::= G <type> # imaginary (C 2000)
1084// FIXME: <type> ::= U <source-name> <type> # vendor extended type qualifier
1085
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001086void CXXNameMangler::mangleCXXCtorType(CXXCtorType T) {
1087 // <ctor-dtor-name> ::= C1 # complete object constructor
1088 // ::= C2 # base object constructor
1089 // ::= C3 # complete object allocating constructor
1090 //
1091 switch (T) {
1092 case Ctor_Complete:
1093 Out << "C1";
1094 break;
1095 case Ctor_Base:
1096 Out << "C2";
1097 break;
1098 case Ctor_CompleteAllocating:
1099 Out << "C3";
1100 break;
1101 }
1102}
1103
Anders Carlsson27ae5362009-04-17 01:58:57 +00001104void CXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
1105 // <ctor-dtor-name> ::= D0 # deleting destructor
1106 // ::= D1 # complete object destructor
1107 // ::= D2 # base object destructor
1108 //
1109 switch (T) {
1110 case Dtor_Deleting:
1111 Out << "D0";
1112 break;
1113 case Dtor_Complete:
1114 Out << "D1";
1115 break;
1116 case Dtor_Base:
1117 Out << "D2";
1118 break;
1119 }
1120}
1121
Anders Carlsson068f3472009-09-17 05:31:47 +00001122void CXXNameMangler::mangleTemplateArgumentList(const TemplateArgumentList &L) {
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001123 // <template-args> ::= I <template-arg>+ E
1124 Out << "I";
Mike Stump1eb44332009-09-09 15:08:12 +00001125
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001126 for (unsigned i = 0, e = L.size(); i != e; ++i) {
1127 const TemplateArgument &A = L[i];
Mike Stump1eb44332009-09-09 15:08:12 +00001128
Anders Carlsson068f3472009-09-17 05:31:47 +00001129 mangleTemplateArgument(A);
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001130 }
Mike Stump1eb44332009-09-09 15:08:12 +00001131
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001132 Out << "E";
1133}
1134
Anders Carlsson7624f212009-09-18 02:42:01 +00001135void CXXNameMangler::mangleTemplateArgs(const TemplateArgument *TemplateArgs,
1136 unsigned NumTemplateArgs) {
1137 // <template-args> ::= I <template-arg>+ E
1138 Out << "I";
1139
1140 for (unsigned i = 0; i != NumTemplateArgs; ++i) {
1141 mangleTemplateArgument(TemplateArgs[i]);
1142 }
1143
1144 Out << "E";
1145}
1146
Anders Carlsson068f3472009-09-17 05:31:47 +00001147void CXXNameMangler::mangleTemplateArgument(const TemplateArgument &A) {
Mike Stump1eb44332009-09-09 15:08:12 +00001148 // <template-arg> ::= <type> # type or template
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001149 // ::= X <expression> E # expression
1150 // ::= <expr-primary> # simple expressions
1151 // ::= I <template-arg>* E # argument pack
1152 // ::= sp <expression> # pack expansion of (C++0x)
1153 switch (A.getKind()) {
1154 default:
1155 assert(0 && "Unknown template argument kind!");
1156 case TemplateArgument::Type:
1157 mangleType(A.getAsType());
1158 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001159 case TemplateArgument::Expression:
1160 Out << 'X';
1161 mangleExpression(A.getAsExpr());
1162 Out << 'E';
1163 break;
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001164 case TemplateArgument::Integral:
1165 // <expr-primary> ::= L <type> <value number> E # integer literal
1166
1167 Out << 'L';
Mike Stump1eb44332009-09-09 15:08:12 +00001168
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001169 mangleType(A.getIntegralType());
Mike Stump1eb44332009-09-09 15:08:12 +00001170
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001171 const llvm::APSInt *Integral = A.getAsIntegral();
1172 if (A.getIntegralType()->isBooleanType()) {
1173 // Boolean values are encoded as 0/1.
1174 Out << (Integral->getBoolValue() ? '1' : '0');
1175 } else {
1176 if (Integral->isNegative())
1177 Out << 'n';
1178 Integral->abs().print(Out, false);
1179 }
Mike Stump1eb44332009-09-09 15:08:12 +00001180
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001181 Out << 'E';
1182 break;
1183 }
1184}
1185
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001186void CXXNameMangler::mangleTemplateParameter(unsigned Index) {
1187 // <template-param> ::= T_ # first template parameter
1188 // ::= T <parameter-2 non-negative number> _
1189 if (Index == 0)
1190 Out << "T_";
1191 else
1192 Out << 'T' << (Index - 1) << '_';
1193}
1194
Anders Carlsson76967372009-09-17 00:43:46 +00001195// <substitution> ::= S <seq-id> _
1196// ::= S_
Anders Carlsson6862fc72009-09-17 04:16:28 +00001197bool CXXNameMangler::mangleSubstitution(const NamedDecl *ND) {
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001198 // Try one of the standard substitutions first.
1199 if (mangleStandardSubstitution(ND))
1200 return true;
1201
Anders Carlsson433d1372009-11-07 04:26:04 +00001202 ND = cast<NamedDecl>(ND->getCanonicalDecl());
Anders Carlsson6862fc72009-09-17 04:16:28 +00001203 return mangleSubstitution(reinterpret_cast<uintptr_t>(ND));
1204}
1205
Anders Carlsson76967372009-09-17 00:43:46 +00001206bool CXXNameMangler::mangleSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001207 if (!T.getCVRQualifiers()) {
1208 if (const RecordType *RT = T->getAs<RecordType>())
1209 return mangleSubstitution(RT->getDecl());
1210 }
1211
Anders Carlsson76967372009-09-17 00:43:46 +00001212 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
1213
Anders Carlssond3a932a2009-09-17 03:53:28 +00001214 return mangleSubstitution(TypePtr);
1215}
1216
1217bool CXXNameMangler::mangleSubstitution(uintptr_t Ptr) {
Anders Carlsson76967372009-09-17 00:43:46 +00001218 llvm::DenseMap<uintptr_t, unsigned>::iterator I =
Anders Carlssond3a932a2009-09-17 03:53:28 +00001219 Substitutions.find(Ptr);
Anders Carlsson76967372009-09-17 00:43:46 +00001220 if (I == Substitutions.end())
1221 return false;
1222
1223 unsigned SeqID = I->second;
1224 if (SeqID == 0)
1225 Out << "S_";
1226 else {
1227 SeqID--;
1228
1229 // <seq-id> is encoded in base-36, using digits and upper case letters.
1230 char Buffer[10];
1231 char *BufferPtr = Buffer + 9;
1232
1233 *BufferPtr = 0;
1234 if (SeqID == 0) *--BufferPtr = '0';
1235
1236 while (SeqID) {
1237 assert(BufferPtr > Buffer && "Buffer overflow!");
1238
1239 unsigned char c = static_cast<unsigned char>(SeqID) % 36;
1240
1241 *--BufferPtr = (c < 10 ? '0' + c : 'A' + c - 10);
1242 SeqID /= 36;
1243 }
1244
1245 Out << 'S' << BufferPtr << '_';
1246 }
1247
1248 return true;
1249}
1250
Anders Carlssonf514b542009-09-27 00:12:57 +00001251static bool isCharType(QualType T) {
1252 if (T.isNull())
1253 return false;
1254
1255 return T->isSpecificBuiltinType(BuiltinType::Char_S) ||
1256 T->isSpecificBuiltinType(BuiltinType::Char_U);
1257}
1258
1259/// isCharSpecialization - Returns whether a given type is a template
1260/// specialization of a given name with a single argument of type char.
1261static bool isCharSpecialization(QualType T, const char *Name) {
1262 if (T.isNull())
1263 return false;
1264
1265 const RecordType *RT = T->getAs<RecordType>();
1266 if (!RT)
1267 return false;
1268
1269 const ClassTemplateSpecializationDecl *SD =
1270 dyn_cast<ClassTemplateSpecializationDecl>(RT->getDecl());
1271 if (!SD)
1272 return false;
1273
1274 if (!isStdNamespace(SD->getDeclContext()))
1275 return false;
1276
1277 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1278 if (TemplateArgs.size() != 1)
1279 return false;
1280
1281 if (!isCharType(TemplateArgs[0].getAsType()))
1282 return false;
1283
Daniel Dunbar01eb9b92009-10-18 21:17:35 +00001284 return SD->getIdentifier()->getName() == Name;
Anders Carlssonf514b542009-09-27 00:12:57 +00001285}
1286
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001287bool CXXNameMangler::mangleStandardSubstitution(const NamedDecl *ND) {
1288 // <substitution> ::= St # ::std::
Anders Carlsson8c031552009-09-26 23:10:05 +00001289 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
John McCall9aeed322009-10-01 00:25:31 +00001290 if (isStdNamespace(NS)) {
Anders Carlsson8c031552009-09-26 23:10:05 +00001291 Out << "St";
1292 return true;
1293 }
1294 }
1295
1296 if (const ClassTemplateDecl *TD = dyn_cast<ClassTemplateDecl>(ND)) {
1297 if (!isStdNamespace(TD->getDeclContext()))
1298 return false;
1299
1300 // <substitution> ::= Sa # ::std::allocator
1301 if (TD->getIdentifier()->isStr("allocator")) {
1302 Out << "Sa";
1303 return true;
1304 }
Anders Carlsson189d59c2009-09-26 23:14:39 +00001305
1306 // <<substitution> ::= Sb # ::std::basic_string
1307 if (TD->getIdentifier()->isStr("basic_string")) {
1308 Out << "Sb";
1309 return true;
1310 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001311 }
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001312
Anders Carlssonf514b542009-09-27 00:12:57 +00001313 if (const ClassTemplateSpecializationDecl *SD =
1314 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
1315 // <substitution> ::= Ss # ::std::basic_string<char,
1316 // ::std::char_traits<char>,
1317 // ::std::allocator<char> >
1318 if (SD->getIdentifier()->isStr("basic_string")) {
1319 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1320
1321 if (TemplateArgs.size() != 3)
1322 return false;
1323
1324 if (!isCharType(TemplateArgs[0].getAsType()))
1325 return false;
1326
1327 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1328 return false;
1329
1330 if (!isCharSpecialization(TemplateArgs[2].getAsType(), "allocator"))
1331 return false;
1332
1333 Out << "Ss";
1334 return true;
1335 }
Anders Carlsson8f8fd8e2009-10-08 17:20:26 +00001336
1337 // <substitution> ::= So # ::std::basic_ostream<char,
1338 // ::std::char_traits<char> >
1339 if (SD->getIdentifier()->isStr("basic_ostream")) {
1340 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1341
1342 if (TemplateArgs.size() != 2)
1343 return false;
1344
1345 if (!isCharType(TemplateArgs[0].getAsType()))
1346 return false;
1347
1348 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1349 return false;
1350
1351 Out << "So";
1352 return true;
1353 }
Anders Carlssonf514b542009-09-27 00:12:57 +00001354 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001355 return false;
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001356}
1357
Anders Carlsson76967372009-09-17 00:43:46 +00001358void CXXNameMangler::addSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001359 if (!T.getCVRQualifiers()) {
1360 if (const RecordType *RT = T->getAs<RecordType>()) {
1361 addSubstitution(RT->getDecl());
1362 return;
1363 }
1364 }
1365
Anders Carlsson76967372009-09-17 00:43:46 +00001366 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
Anders Carlssond3a932a2009-09-17 03:53:28 +00001367 addSubstitution(TypePtr);
1368}
1369
1370void CXXNameMangler::addSubstitution(uintptr_t Ptr) {
Anders Carlsson76967372009-09-17 00:43:46 +00001371 unsigned SeqID = Substitutions.size();
1372
Anders Carlssond3a932a2009-09-17 03:53:28 +00001373 assert(!Substitutions.count(Ptr) && "Substitution already exists!");
1374 Substitutions[Ptr] = SeqID;
Anders Carlsson76967372009-09-17 00:43:46 +00001375}
1376
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001377namespace clang {
Mike Stump141c5af2009-09-02 00:25:38 +00001378 /// \brief Mangles the name of the declaration D and emits that name to the
1379 /// given output stream.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001380 ///
Mike Stump141c5af2009-09-02 00:25:38 +00001381 /// If the declaration D requires a mangled name, this routine will emit that
1382 /// mangled name to \p os and return true. Otherwise, \p os will be unchanged
1383 /// and this routine will return false. In this case, the caller should just
1384 /// emit the identifier of the declaration (\c D->getIdentifier()) as its
1385 /// name.
Anders Carlssonb5404912009-10-07 01:06:45 +00001386 bool mangleName(MangleContext &Context, const NamedDecl *D,
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001387 llvm::raw_ostream &os) {
Anders Carlsson578aa642009-05-03 16:51:04 +00001388 assert(!isa<CXXConstructorDecl>(D) &&
1389 "Use mangleCXXCtor for constructor decls!");
1390 assert(!isa<CXXDestructorDecl>(D) &&
1391 "Use mangleCXXDtor for destructor decls!");
Mike Stump1eb44332009-09-09 15:08:12 +00001392
Anders Carlssond99edc42009-09-26 03:55:37 +00001393 PrettyStackTraceDecl CrashInfo(const_cast<NamedDecl *>(D), SourceLocation(),
Anders Carlssonb5404912009-10-07 01:06:45 +00001394 Context.getASTContext().getSourceManager(),
Anders Carlssond99edc42009-09-26 03:55:37 +00001395 "Mangling declaration");
1396
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001397 CXXNameMangler Mangler(Context, os);
Douglas Gregor457e2812009-10-28 16:31:34 +00001398 if (!Mangler.mangle(D))
Douglas Gregor6ec36682009-02-18 23:53:56 +00001399 return false;
Mike Stump1eb44332009-09-09 15:08:12 +00001400
Douglas Gregor6ec36682009-02-18 23:53:56 +00001401 os.flush();
1402 return true;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001403 }
Mike Stump1eb44332009-09-09 15:08:12 +00001404
Mike Stump141c5af2009-09-02 00:25:38 +00001405 /// \brief Mangles the a thunk with the offset n for the declaration D and
1406 /// emits that name to the given output stream.
Anders Carlssonb5404912009-10-07 01:06:45 +00001407 void mangleThunk(MangleContext &Context, const FunctionDecl *FD,
1408 int64_t nv, int64_t v, llvm::raw_ostream &os) {
Mike Stump141c5af2009-09-02 00:25:38 +00001409 // FIXME: Hum, we might have to thunk these, fix.
Mike Stumpdec025b2009-09-07 04:27:52 +00001410 assert(!isa<CXXDestructorDecl>(FD) &&
Mike Stump141c5af2009-09-02 00:25:38 +00001411 "Use mangleCXXDtor for destructor decls!");
Mike Stump1eb44332009-09-09 15:08:12 +00001412
Mike Stump141c5af2009-09-02 00:25:38 +00001413 CXXNameMangler Mangler(Context, os);
Mike Stumpdec025b2009-09-07 04:27:52 +00001414 Mangler.mangleThunk(FD, nv, v);
Mike Stump141c5af2009-09-02 00:25:38 +00001415 os.flush();
1416 }
Mike Stump1eb44332009-09-09 15:08:12 +00001417
Mike Stump9124bcc2009-09-02 00:56:18 +00001418 /// \brief Mangles the a covariant thunk for the declaration D and emits that
1419 /// name to the given output stream.
Anders Carlssonb5404912009-10-07 01:06:45 +00001420 void mangleCovariantThunk(MangleContext &Context, const FunctionDecl *FD,
1421 int64_t nv_t, int64_t v_t,
1422 int64_t nv_r, int64_t v_r,
Mike Stump9124bcc2009-09-02 00:56:18 +00001423 llvm::raw_ostream &os) {
1424 // FIXME: Hum, we might have to thunk these, fix.
Mike Stumpdec025b2009-09-07 04:27:52 +00001425 assert(!isa<CXXDestructorDecl>(FD) &&
Mike Stump9124bcc2009-09-02 00:56:18 +00001426 "Use mangleCXXDtor for destructor decls!");
Mike Stump1eb44332009-09-09 15:08:12 +00001427
Mike Stump9124bcc2009-09-02 00:56:18 +00001428 CXXNameMangler Mangler(Context, os);
Mike Stumpdec025b2009-09-07 04:27:52 +00001429 Mangler.mangleCovariantThunk(FD, nv_t, v_t, nv_r, v_r);
Mike Stump9124bcc2009-09-02 00:56:18 +00001430 os.flush();
1431 }
Mike Stump1eb44332009-09-09 15:08:12 +00001432
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001433 /// mangleGuardVariable - Returns the mangled name for a guard variable
1434 /// for the passed in VarDecl.
Anders Carlssonb5404912009-10-07 01:06:45 +00001435 void mangleGuardVariable(MangleContext &Context, const VarDecl *D,
Anders Carlsson41aa8c12009-04-13 18:02:10 +00001436 llvm::raw_ostream &os) {
1437 CXXNameMangler Mangler(Context, os);
1438 Mangler.mangleGuardVariable(D);
1439
1440 os.flush();
1441 }
Mike Stump1eb44332009-09-09 15:08:12 +00001442
Anders Carlssonb5404912009-10-07 01:06:45 +00001443 void mangleCXXCtor(MangleContext &Context, const CXXConstructorDecl *D,
1444 CXXCtorType Type, llvm::raw_ostream &os) {
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001445 CXXNameMangler Mangler(Context, os);
1446 Mangler.mangleCXXCtor(D, Type);
Mike Stump1eb44332009-09-09 15:08:12 +00001447
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001448 os.flush();
1449 }
Mike Stump1eb44332009-09-09 15:08:12 +00001450
Anders Carlssonb5404912009-10-07 01:06:45 +00001451 void mangleCXXDtor(MangleContext &Context, const CXXDestructorDecl *D,
1452 CXXDtorType Type, llvm::raw_ostream &os) {
Anders Carlsson27ae5362009-04-17 01:58:57 +00001453 CXXNameMangler Mangler(Context, os);
1454 Mangler.mangleCXXDtor(D, Type);
Mike Stump1eb44332009-09-09 15:08:12 +00001455
Anders Carlsson27ae5362009-04-17 01:58:57 +00001456 os.flush();
1457 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001458
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +00001459 void mangleCXXVtable(MangleContext &Context, const CXXRecordDecl *RD,
Mike Stumpf1216772009-07-31 18:25:34 +00001460 llvm::raw_ostream &os) {
1461 CXXNameMangler Mangler(Context, os);
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +00001462 Mangler.mangleCXXVtable(RD);
Mike Stumpf1216772009-07-31 18:25:34 +00001463
1464 os.flush();
1465 }
Mike Stump738f8c22009-07-31 23:15:31 +00001466
Mike Stump82d75b02009-11-10 01:58:37 +00001467 void mangleCXXVTT(MangleContext &Context, const CXXRecordDecl *RD,
1468 llvm::raw_ostream &os) {
1469 CXXNameMangler Mangler(Context, os);
1470 Mangler.mangleCXXVTT(RD);
1471
1472 os.flush();
1473 }
1474
Mike Stumpab3f7e92009-11-10 01:41:59 +00001475 void mangleCXXCtorVtable(MangleContext &Context, const CXXRecordDecl *RD,
1476 int64_t Offset, const CXXRecordDecl *Type,
1477 llvm::raw_ostream &os) {
1478 CXXNameMangler Mangler(Context, os);
1479 Mangler.mangleCXXCtorVtable(RD, Offset, Type);
1480
1481 os.flush();
1482 }
1483
Anders Carlssond313e402009-10-30 01:52:02 +00001484 void mangleCXXRtti(MangleContext &Context, QualType Ty,
Mike Stump738f8c22009-07-31 23:15:31 +00001485 llvm::raw_ostream &os) {
1486 CXXNameMangler Mangler(Context, os);
Anders Carlsson731f7f92009-10-30 01:26:12 +00001487 Mangler.mangleCXXRtti(Ty);
Mike Stump738f8c22009-07-31 23:15:31 +00001488
1489 os.flush();
1490 }
Mike Stump67795982009-11-14 00:14:13 +00001491
1492 void mangleCXXRttiName(MangleContext &Context, QualType Ty,
1493 llvm::raw_ostream &os) {
1494 CXXNameMangler Mangler(Context, os);
1495 Mangler.mangleCXXRttiName(Ty);
1496
1497 os.flush();
1498 }
Mike Stumpf1216772009-07-31 18:25:34 +00001499}