blob: 10313980ca9834485ef017e6a90812ce555451fd [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/raw_ostream.h"
John McCallefe6aee2009-09-05 07:56:18 +000027#include "llvm/Support/ErrorHandling.h"
Anders Carlsson461e3262010-04-08 16:30:25 +000028#include "CGVTables.h"
Anders Carlssonf98574b2010-02-05 07:31:37 +000029
30#define MANGLE_CHECKER 0
31
32#if MANGLE_CHECKER
33#include <cxxabi.h>
34#endif
35
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000036using namespace clang;
Anders Carlssonb73a5be2009-11-26 02:49:32 +000037using namespace CodeGen;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000038
39namespace {
Fariborz Jahanian57058532010-03-03 19:41:08 +000040
41static const DeclContext *GetLocalClassFunctionDeclContext(
42 const DeclContext *DC) {
43 if (isa<CXXRecordDecl>(DC)) {
44 while (!DC->isNamespace() && !DC->isTranslationUnit() &&
45 !isa<FunctionDecl>(DC))
46 DC = DC->getParent();
47 if (isa<FunctionDecl>(DC))
48 return DC;
49 }
50 return 0;
51}
52
Anders Carlsson7e120032009-11-24 05:36:32 +000053static const CXXMethodDecl *getStructor(const CXXMethodDecl *MD) {
54 assert((isa<CXXConstructorDecl>(MD) || isa<CXXDestructorDecl>(MD)) &&
55 "Passed in decl is not a ctor or dtor!");
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +000056
Anders Carlsson7e120032009-11-24 05:36:32 +000057 if (const TemplateDecl *TD = MD->getPrimaryTemplate()) {
58 MD = cast<CXXMethodDecl>(TD->getTemplatedDecl());
59
60 assert((isa<CXXConstructorDecl>(MD) || isa<CXXDestructorDecl>(MD)) &&
61 "Templated decl is not a ctor or dtor!");
62 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +000063
Anders Carlsson7e120032009-11-24 05:36:32 +000064 return MD;
65}
John McCall1dd73832010-02-04 01:42:13 +000066
67static const unsigned UnknownArity = ~0U;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +000068
Daniel Dunbar1b077112009-11-21 09:06:10 +000069/// CXXNameMangler - Manage the mangling of a single name.
Daniel Dunbarc0747712009-11-21 09:12:13 +000070class CXXNameMangler {
Daniel Dunbar1b077112009-11-21 09:06:10 +000071 MangleContext &Context;
Daniel Dunbar94fd26d2009-11-21 09:06:22 +000072 llvm::raw_svector_ostream Out;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000073
Daniel Dunbar1b077112009-11-21 09:06:10 +000074 const CXXMethodDecl *Structor;
75 unsigned StructorType;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +000076
Daniel Dunbar1b077112009-11-21 09:06:10 +000077 llvm::DenseMap<uintptr_t, unsigned> Substitutions;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +000078
John McCall1dd73832010-02-04 01:42:13 +000079 ASTContext &getASTContext() const { return Context.getASTContext(); }
80
Daniel Dunbarc0747712009-11-21 09:12:13 +000081public:
Daniel Dunbar94fd26d2009-11-21 09:06:22 +000082 CXXNameMangler(MangleContext &C, llvm::SmallVectorImpl<char> &Res)
83 : Context(C), Out(Res), Structor(0), StructorType(0) { }
Daniel Dunbar77939c92009-11-21 09:06:31 +000084 CXXNameMangler(MangleContext &C, llvm::SmallVectorImpl<char> &Res,
85 const CXXConstructorDecl *D, CXXCtorType Type)
Anders Carlsson7e120032009-11-24 05:36:32 +000086 : Context(C), Out(Res), Structor(getStructor(D)), StructorType(Type) { }
Daniel Dunbar77939c92009-11-21 09:06:31 +000087 CXXNameMangler(MangleContext &C, llvm::SmallVectorImpl<char> &Res,
88 const CXXDestructorDecl *D, CXXDtorType Type)
Anders Carlsson7e120032009-11-24 05:36:32 +000089 : Context(C), Out(Res), Structor(getStructor(D)), StructorType(Type) { }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000090
Anders Carlssonf98574b2010-02-05 07:31:37 +000091#if MANGLE_CHECKER
92 ~CXXNameMangler() {
93 if (Out.str()[0] == '\01')
94 return;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +000095
Anders Carlssonf98574b2010-02-05 07:31:37 +000096 int status = 0;
97 char *result = abi::__cxa_demangle(Out.str().str().c_str(), 0, 0, &status);
98 assert(status == 0 && "Could not demangle mangled name!");
99 free(result);
100 }
101#endif
Daniel Dunbarc0747712009-11-21 09:12:13 +0000102 llvm::raw_svector_ostream &getStream() { return Out; }
103
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000104 void mangle(const NamedDecl *D, llvm::StringRef Prefix = "_Z");
Anders Carlsson19879c92010-03-23 17:17:29 +0000105 void mangleCallOffset(int64_t NonVirtual, int64_t Virtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000106 void mangleNumber(int64_t Number);
Daniel Dunbarc0747712009-11-21 09:12:13 +0000107 void mangleFunctionEncoding(const FunctionDecl *FD);
108 void mangleName(const NamedDecl *ND);
109 void mangleType(QualType T);
Mike Stump1eb44332009-09-09 15:08:12 +0000110
Daniel Dunbarc0747712009-11-21 09:12:13 +0000111private:
Daniel Dunbar1b077112009-11-21 09:06:10 +0000112 bool mangleSubstitution(const NamedDecl *ND);
113 bool mangleSubstitution(QualType T);
Douglas Gregor1e9268e2010-04-28 05:58:56 +0000114 bool mangleSubstitution(TemplateName Template);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000115 bool mangleSubstitution(uintptr_t Ptr);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000116
Daniel Dunbar1b077112009-11-21 09:06:10 +0000117 bool mangleStandardSubstitution(const NamedDecl *ND);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000118
Daniel Dunbar1b077112009-11-21 09:06:10 +0000119 void addSubstitution(const NamedDecl *ND) {
120 ND = cast<NamedDecl>(ND->getCanonicalDecl());
Anders Carlsson433d1372009-11-07 04:26:04 +0000121
Daniel Dunbar1b077112009-11-21 09:06:10 +0000122 addSubstitution(reinterpret_cast<uintptr_t>(ND));
123 }
124 void addSubstitution(QualType T);
Douglas Gregor1e9268e2010-04-28 05:58:56 +0000125 void addSubstitution(TemplateName Template);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000126 void addSubstitution(uintptr_t Ptr);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000127
John McCall1dd73832010-02-04 01:42:13 +0000128 void mangleUnresolvedScope(NestedNameSpecifier *Qualifier);
129 void mangleUnresolvedName(NestedNameSpecifier *Qualifier,
130 DeclarationName Name,
131 unsigned KnownArity = UnknownArity);
132
Daniel Dunbar1b077112009-11-21 09:06:10 +0000133 void mangleName(const TemplateDecl *TD,
134 const TemplateArgument *TemplateArgs,
135 unsigned NumTemplateArgs);
John McCall1dd73832010-02-04 01:42:13 +0000136 void mangleUnqualifiedName(const NamedDecl *ND) {
137 mangleUnqualifiedName(ND, ND->getDeclName(), UnknownArity);
138 }
139 void mangleUnqualifiedName(const NamedDecl *ND, DeclarationName Name,
140 unsigned KnownArity);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000141 void mangleUnscopedName(const NamedDecl *ND);
142 void mangleUnscopedTemplateName(const TemplateDecl *ND);
Douglas Gregor1e9268e2010-04-28 05:58:56 +0000143 void mangleUnscopedTemplateName(TemplateName);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000144 void mangleSourceName(const IdentifierInfo *II);
145 void mangleLocalName(const NamedDecl *ND);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000146 void mangleNestedName(const NamedDecl *ND, const DeclContext *DC,
147 bool NoFunction=false);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000148 void mangleNestedName(const TemplateDecl *TD,
149 const TemplateArgument *TemplateArgs,
150 unsigned NumTemplateArgs);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000151 void manglePrefix(const DeclContext *DC, bool NoFunction=false);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000152 void mangleTemplatePrefix(const TemplateDecl *ND);
Douglas Gregor20f0cc72010-04-23 03:10:43 +0000153 void mangleTemplatePrefix(TemplateName Template);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000154 void mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity);
155 void mangleQualifiers(Qualifiers Quals);
John McCallefe6aee2009-09-05 07:56:18 +0000156
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000157 void mangleObjCMethodName(const ObjCMethodDecl *MD);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000158
Daniel Dunbar1b077112009-11-21 09:06:10 +0000159 // Declare manglers for every type class.
John McCallefe6aee2009-09-05 07:56:18 +0000160#define ABSTRACT_TYPE(CLASS, PARENT)
161#define NON_CANONICAL_TYPE(CLASS, PARENT)
162#define TYPE(CLASS, PARENT) void mangleType(const CLASS##Type *T);
163#include "clang/AST/TypeNodes.def"
164
Daniel Dunbar1b077112009-11-21 09:06:10 +0000165 void mangleType(const TagType*);
166 void mangleBareFunctionType(const FunctionType *T,
167 bool MangleReturnType);
Anders Carlssone170ba72009-12-14 01:45:37 +0000168
169 void mangleIntegerLiteral(QualType T, const llvm::APSInt &Value);
John McCall2f27bf82010-02-04 02:56:29 +0000170 void mangleMemberExpr(const Expr *Base, bool IsArrow,
171 NestedNameSpecifier *Qualifier,
172 DeclarationName Name,
173 unsigned KnownArity);
John McCall1dd73832010-02-04 01:42:13 +0000174 void mangleCalledExpression(const Expr *E, unsigned KnownArity);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000175 void mangleExpression(const Expr *E);
176 void mangleCXXCtorType(CXXCtorType T);
177 void mangleCXXDtorType(CXXDtorType T);
Mike Stump1eb44332009-09-09 15:08:12 +0000178
Douglas Gregor20f0cc72010-04-23 03:10:43 +0000179 void mangleTemplateArgs(TemplateName Template,
180 const TemplateArgument *TemplateArgs,
Sean Huntc3021132010-05-05 15:23:54 +0000181 unsigned NumTemplateArgs);
Rafael Espindolad9800722010-03-11 14:07:00 +0000182 void mangleTemplateArgs(const TemplateParameterList &PL,
183 const TemplateArgument *TemplateArgs,
Daniel Dunbar1b077112009-11-21 09:06:10 +0000184 unsigned NumTemplateArgs);
Rafael Espindolad9800722010-03-11 14:07:00 +0000185 void mangleTemplateArgs(const TemplateParameterList &PL,
186 const TemplateArgumentList &AL);
187 void mangleTemplateArg(const NamedDecl *P, const TemplateArgument &A);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000188
Daniel Dunbar1b077112009-11-21 09:06:10 +0000189 void mangleTemplateParameter(unsigned Index);
190};
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000191}
192
Anders Carlsson43f17402009-04-02 15:51:53 +0000193static bool isInCLinkageSpecification(const Decl *D) {
Douglas Gregor457e2812009-10-28 16:31:34 +0000194 D = D->getCanonicalDecl();
Mike Stump1eb44332009-09-09 15:08:12 +0000195 for (const DeclContext *DC = D->getDeclContext();
Anders Carlsson43f17402009-04-02 15:51:53 +0000196 !DC->isTranslationUnit(); DC = DC->getParent()) {
Mike Stump1eb44332009-09-09 15:08:12 +0000197 if (const LinkageSpecDecl *Linkage = dyn_cast<LinkageSpecDecl>(DC))
Anders Carlsson43f17402009-04-02 15:51:53 +0000198 return Linkage->getLanguage() == LinkageSpecDecl::lang_c;
199 }
Mike Stump1eb44332009-09-09 15:08:12 +0000200
Anders Carlsson43f17402009-04-02 15:51:53 +0000201 return false;
202}
203
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000204bool MangleContext::shouldMangleDeclName(const NamedDecl *D) {
205 // In C, functions with no attributes never need to be mangled. Fastpath them.
206 if (!getASTContext().getLangOptions().CPlusPlus && !D->hasAttrs())
207 return false;
208
209 // Any decl can be declared with __asm("foo") on it, and this takes precedence
210 // over all other naming in the .o file.
211 if (D->hasAttr<AsmLabelAttr>())
212 return true;
213
Mike Stump141c5af2009-09-02 00:25:38 +0000214 // Clang's "overloadable" attribute extension to C/C++ implies name mangling
Anders Carlssona1e16222009-11-07 07:15:03 +0000215 // (always) as does passing a C++ member function and a function
216 // whose name is not a simple identifier.
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000217 const FunctionDecl *FD = dyn_cast<FunctionDecl>(D);
218 if (FD && (FD->hasAttr<OverloadableAttr>() || isa<CXXMethodDecl>(FD) ||
219 !FD->getDeclName().isIdentifier()))
220 return true;
Mike Stump1eb44332009-09-09 15:08:12 +0000221
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000222 // Otherwise, no mangling is done outside C++ mode.
223 if (!getASTContext().getLangOptions().CPlusPlus)
224 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000225
Sean Hunt31455252010-01-24 03:04:27 +0000226 // Variables at global scope with non-internal linkage are not mangled
Eli Friedman7facf842009-12-02 20:32:49 +0000227 if (!FD) {
228 const DeclContext *DC = D->getDeclContext();
229 // Check for extern variable declared locally.
230 if (isa<FunctionDecl>(DC) && D->hasLinkage())
231 while (!DC->isNamespace() && !DC->isTranslationUnit())
232 DC = DC->getParent();
Douglas Gregor0b6bc8b2010-02-03 09:33:45 +0000233 if (DC->isTranslationUnit() && D->getLinkage() != InternalLinkage)
Eli Friedman7facf842009-12-02 20:32:49 +0000234 return false;
235 }
236
237 // C functions and "main" are not mangled.
238 if ((FD && FD->isMain()) || isInCLinkageSpecification(D))
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000239 return false;
240
Anders Carlsson43f17402009-04-02 15:51:53 +0000241 return true;
242}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000243
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000244void CXXNameMangler::mangle(const NamedDecl *D, llvm::StringRef Prefix) {
Mike Stump141c5af2009-09-02 00:25:38 +0000245 // Any decl can be declared with __asm("foo") on it, and this takes precedence
246 // over all other naming in the .o file.
Argyrios Kyrtzidis40b598e2009-06-30 02:34:44 +0000247 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000248 // If we have an asm name, then we use it as the mangling.
249 Out << '\01'; // LLVM IR Marker for __asm("foo")
250 Out << ALA->getLabel();
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000251 return;
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000252 }
Mike Stump1eb44332009-09-09 15:08:12 +0000253
Sean Hunt31455252010-01-24 03:04:27 +0000254 // <mangled-name> ::= _Z <encoding>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000255 // ::= <data name>
256 // ::= <special-name>
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000257 Out << Prefix;
258 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000259 mangleFunctionEncoding(FD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000260 else if (const VarDecl *VD = dyn_cast<VarDecl>(D))
261 mangleName(VD);
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000262 else
Rafael Espindolad9800722010-03-11 14:07:00 +0000263 mangleName(cast<FieldDecl>(D));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000264}
265
266void CXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD) {
267 // <encoding> ::= <function name> <bare-function-type>
268 mangleName(FD);
Mike Stump1eb44332009-09-09 15:08:12 +0000269
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000270 // Don't mangle in the type if this isn't a decl we should typically mangle.
271 if (!Context.shouldMangleDeclName(FD))
272 return;
273
Mike Stump141c5af2009-09-02 00:25:38 +0000274 // Whether the mangling of a function type includes the return type depends on
275 // the context and the nature of the function. The rules for deciding whether
276 // the return type is included are:
Mike Stump1eb44332009-09-09 15:08:12 +0000277 //
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000278 // 1. Template functions (names or types) have return types encoded, with
279 // the exceptions listed below.
Mike Stump1eb44332009-09-09 15:08:12 +0000280 // 2. Function types not appearing as part of a function name mangling,
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000281 // e.g. parameters, pointer types, etc., have return type encoded, with the
282 // exceptions listed below.
283 // 3. Non-template function names do not have return types encoded.
284 //
Mike Stump141c5af2009-09-02 00:25:38 +0000285 // The exceptions mentioned in (1) and (2) above, for which the return type is
286 // never included, are
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000287 // 1. Constructors.
288 // 2. Destructors.
289 // 3. Conversion operator functions, e.g. operator int.
290 bool MangleReturnType = false;
Anders Carlsson9234b7f2009-09-17 03:46:43 +0000291 if (FunctionTemplateDecl *PrimaryTemplate = FD->getPrimaryTemplate()) {
292 if (!(isa<CXXConstructorDecl>(FD) || isa<CXXDestructorDecl>(FD) ||
293 isa<CXXConversionDecl>(FD)))
294 MangleReturnType = true;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000295
Anders Carlsson9234b7f2009-09-17 03:46:43 +0000296 // Mangle the type of the primary template.
297 FD = PrimaryTemplate->getTemplatedDecl();
298 }
299
John McCall54e14c42009-10-22 22:37:11 +0000300 // Do the canonicalization out here because parameter types can
301 // undergo additional canonicalization (e.g. array decay).
302 FunctionType *FT = cast<FunctionType>(Context.getASTContext()
303 .getCanonicalType(FD->getType()));
304
305 mangleBareFunctionType(FT, MangleReturnType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000306}
307
Anders Carlsson47846d22009-12-04 06:23:23 +0000308/// isStd - Return whether a given namespace is the 'std' namespace.
309static bool isStd(const NamespaceDecl *NS) {
John McCall9aeed322009-10-01 00:25:31 +0000310 const IdentifierInfo *II = NS->getOriginalNamespace()->getIdentifier();
311 return II && II->isStr("std");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000312}
313
Anders Carlsson47846d22009-12-04 06:23:23 +0000314static const DeclContext *IgnoreLinkageSpecDecls(const DeclContext *DC) {
315 while (isa<LinkageSpecDecl>(DC)) {
Anders Carlsson47846d22009-12-04 06:23:23 +0000316 DC = DC->getParent();
317 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000318
Anders Carlsson47846d22009-12-04 06:23:23 +0000319 return DC;
320}
321
322// isStdNamespace - Return whether a given decl context is a toplevel 'std'
323// namespace.
Daniel Dunbar1308af92009-11-21 09:11:45 +0000324static bool isStdNamespace(const DeclContext *DC) {
Anders Carlsson47846d22009-12-04 06:23:23 +0000325 if (!DC->isNamespace())
326 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000327
Anders Carlsson47846d22009-12-04 06:23:23 +0000328 if (!IgnoreLinkageSpecDecls(DC->getParent())->isTranslationUnit())
329 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000330
Anders Carlsson47846d22009-12-04 06:23:23 +0000331 return isStd(cast<NamespaceDecl>(DC));
Daniel Dunbar1308af92009-11-21 09:11:45 +0000332}
333
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000334static const TemplateDecl *
335isTemplate(const NamedDecl *ND, const TemplateArgumentList *&TemplateArgs) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000336 // Check if we have a function template.
337 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)){
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000338 if (const TemplateDecl *TD = FD->getPrimaryTemplate()) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000339 TemplateArgs = FD->getTemplateSpecializationArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000340 return TD;
Anders Carlsson2744a062009-09-18 19:00:18 +0000341 }
342 }
343
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000344 // Check if we have a class template.
345 if (const ClassTemplateSpecializationDecl *Spec =
346 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
347 TemplateArgs = &Spec->getTemplateArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000348 return Spec->getSpecializedTemplate();
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000349 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000350
Anders Carlsson2744a062009-09-18 19:00:18 +0000351 return 0;
352}
353
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000354void CXXNameMangler::mangleName(const NamedDecl *ND) {
355 // <name> ::= <nested-name>
356 // ::= <unscoped-name>
357 // ::= <unscoped-template-name> <template-args>
Anders Carlsson201ce742009-09-17 03:17:01 +0000358 // ::= <local-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000359 //
Anders Carlssond58d6f72009-09-17 16:12:20 +0000360 const DeclContext *DC = ND->getDeclContext();
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000361
Fariborz Jahanian57058532010-03-03 19:41:08 +0000362 if (GetLocalClassFunctionDeclContext(DC)) {
363 mangleLocalName(ND);
364 return;
365 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000366
Eli Friedman7facf842009-12-02 20:32:49 +0000367 // If this is an extern variable declared locally, the relevant DeclContext
368 // is that of the containing namespace, or the translation unit.
369 if (isa<FunctionDecl>(DC) && ND->hasLinkage())
370 while (!DC->isNamespace() && !DC->isTranslationUnit())
371 DC = DC->getParent();
372
Anders Carlsson5cc58c62009-09-22 17:23:30 +0000373 while (isa<LinkageSpecDecl>(DC))
Anders Carlssond58d6f72009-09-17 16:12:20 +0000374 DC = DC->getParent();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000375
Anders Carlssond58d6f72009-09-17 16:12:20 +0000376 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000377 // Check if we have a template.
378 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000379 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000380 mangleUnscopedTemplateName(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000381 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
382 mangleTemplateArgs(*TemplateParameters, *TemplateArgs);
Anders Carlsson2744a062009-09-18 19:00:18 +0000383 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000384 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000385
Anders Carlsson7482e242009-09-18 04:29:09 +0000386 mangleUnscopedName(ND);
387 return;
388 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000389
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000390 if (isa<FunctionDecl>(DC) || isa<ObjCMethodDecl>(DC)) {
Anders Carlsson7482e242009-09-18 04:29:09 +0000391 mangleLocalName(ND);
392 return;
393 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000394
Eli Friedman7facf842009-12-02 20:32:49 +0000395 mangleNestedName(ND, DC);
Anders Carlsson7482e242009-09-18 04:29:09 +0000396}
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000397void CXXNameMangler::mangleName(const TemplateDecl *TD,
Anders Carlsson7624f212009-09-18 02:42:01 +0000398 const TemplateArgument *TemplateArgs,
399 unsigned NumTemplateArgs) {
Anders Carlsson47846d22009-12-04 06:23:23 +0000400 const DeclContext *DC = IgnoreLinkageSpecDecls(TD->getDeclContext());
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000401
Anders Carlsson7624f212009-09-18 02:42:01 +0000402 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000403 mangleUnscopedTemplateName(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000404 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
405 mangleTemplateArgs(*TemplateParameters, TemplateArgs, NumTemplateArgs);
Anders Carlsson7624f212009-09-18 02:42:01 +0000406 } else {
407 mangleNestedName(TD, TemplateArgs, NumTemplateArgs);
408 }
409}
410
Anders Carlsson201ce742009-09-17 03:17:01 +0000411void CXXNameMangler::mangleUnscopedName(const NamedDecl *ND) {
412 // <unscoped-name> ::= <unqualified-name>
413 // ::= St <unqualified-name> # ::std::
414 if (isStdNamespace(ND->getDeclContext()))
415 Out << "St";
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000416
Anders Carlsson201ce742009-09-17 03:17:01 +0000417 mangleUnqualifiedName(ND);
418}
419
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000420void CXXNameMangler::mangleUnscopedTemplateName(const TemplateDecl *ND) {
Anders Carlsson201ce742009-09-17 03:17:01 +0000421 // <unscoped-template-name> ::= <unscoped-name>
422 // ::= <substitution>
Anders Carlsson7624f212009-09-18 02:42:01 +0000423 if (mangleSubstitution(ND))
Anders Carlsson03c9d532009-09-17 04:02:31 +0000424 return;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000425
Douglas Gregor32fb4e12010-02-05 20:45:00 +0000426 // <template-template-param> ::= <template-param>
427 if (const TemplateTemplateParmDecl *TTP
428 = dyn_cast<TemplateTemplateParmDecl>(ND)) {
429 mangleTemplateParameter(TTP->getIndex());
430 return;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000431 }
Douglas Gregor32fb4e12010-02-05 20:45:00 +0000432
Anders Carlsson1668f202009-09-26 20:13:56 +0000433 mangleUnscopedName(ND->getTemplatedDecl());
Anders Carlsson7624f212009-09-18 02:42:01 +0000434 addSubstitution(ND);
Anders Carlsson201ce742009-09-17 03:17:01 +0000435}
436
Douglas Gregor1e9268e2010-04-28 05:58:56 +0000437void CXXNameMangler::mangleUnscopedTemplateName(TemplateName Template) {
438 // <unscoped-template-name> ::= <unscoped-name>
439 // ::= <substitution>
440 if (TemplateDecl *TD = Template.getAsTemplateDecl())
441 return mangleUnscopedTemplateName(TD);
Sean Huntc3021132010-05-05 15:23:54 +0000442
Douglas Gregor1e9268e2010-04-28 05:58:56 +0000443 if (mangleSubstitution(Template))
444 return;
445
446 // FIXME: How to cope with operators here?
447 DependentTemplateName *Dependent = Template.getAsDependentTemplateName();
448 assert(Dependent && "Not a dependent template name?");
449 if (!Dependent->isIdentifier()) {
450 // FIXME: We can't possibly know the arity of the operator here!
451 Diagnostic &Diags = Context.getDiags();
452 unsigned DiagID = Diags.getCustomDiagID(Diagnostic::Error,
453 "cannot mangle dependent operator name");
454 Diags.Report(FullSourceLoc(), DiagID);
455 return;
456 }
Sean Huntc3021132010-05-05 15:23:54 +0000457
Douglas Gregor1e9268e2010-04-28 05:58:56 +0000458 mangleSourceName(Dependent->getIdentifier());
459 addSubstitution(Template);
460}
461
Anders Carlssona94822e2009-11-26 02:32:05 +0000462void CXXNameMangler::mangleNumber(int64_t Number) {
463 // <number> ::= [n] <non-negative decimal integer>
464 if (Number < 0) {
465 Out << 'n';
466 Number = -Number;
467 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000468
Anders Carlssona94822e2009-11-26 02:32:05 +0000469 Out << Number;
470}
471
Anders Carlsson19879c92010-03-23 17:17:29 +0000472void CXXNameMangler::mangleCallOffset(int64_t NonVirtual, int64_t Virtual) {
Mike Stump141c5af2009-09-02 00:25:38 +0000473 // <call-offset> ::= h <nv-offset> _
474 // ::= v <v-offset> _
475 // <nv-offset> ::= <offset number> # non-virtual base override
Anders Carlssona94822e2009-11-26 02:32:05 +0000476 // <v-offset> ::= <offset number> _ <virtual offset number>
Mike Stump141c5af2009-09-02 00:25:38 +0000477 // # virtual base override, with vcall offset
Anders Carlsson19879c92010-03-23 17:17:29 +0000478 if (!Virtual) {
Anders Carlssona94822e2009-11-26 02:32:05 +0000479 Out << 'h';
Anders Carlsson19879c92010-03-23 17:17:29 +0000480 mangleNumber(NonVirtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000481 Out << '_';
482 return;
Mike Stump141c5af2009-09-02 00:25:38 +0000483 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000484
Anders Carlssona94822e2009-11-26 02:32:05 +0000485 Out << 'v';
Anders Carlsson19879c92010-03-23 17:17:29 +0000486 mangleNumber(NonVirtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000487 Out << '_';
Anders Carlsson19879c92010-03-23 17:17:29 +0000488 mangleNumber(Virtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000489 Out << '_';
Mike Stump9124bcc2009-09-02 00:56:18 +0000490}
491
John McCall1dd73832010-02-04 01:42:13 +0000492void CXXNameMangler::mangleUnresolvedScope(NestedNameSpecifier *Qualifier) {
493 Qualifier = getASTContext().getCanonicalNestedNameSpecifier(Qualifier);
494 switch (Qualifier->getKind()) {
495 case NestedNameSpecifier::Global:
496 // nothing
497 break;
498 case NestedNameSpecifier::Namespace:
499 mangleName(Qualifier->getAsNamespace());
500 break;
501 case NestedNameSpecifier::TypeSpec:
Rafael Espindola9b35b252010-03-17 04:28:11 +0000502 case NestedNameSpecifier::TypeSpecWithTemplate: {
503 const Type *QTy = Qualifier->getAsType();
504
505 if (const TemplateSpecializationType *TST =
506 dyn_cast<TemplateSpecializationType>(QTy)) {
507 if (!mangleSubstitution(QualType(TST, 0))) {
Douglas Gregor20f0cc72010-04-23 03:10:43 +0000508 mangleTemplatePrefix(TST->getTemplateName());
Sean Huntc3021132010-05-05 15:23:54 +0000509
Douglas Gregor20f0cc72010-04-23 03:10:43 +0000510 // FIXME: GCC does not appear to mangle the template arguments when
511 // the template in question is a dependent template name. Should we
512 // emulate that badness?
513 mangleTemplateArgs(TST->getTemplateName(), TST->getArgs(),
Rafael Espindola9b35b252010-03-17 04:28:11 +0000514 TST->getNumArgs());
515 addSubstitution(QualType(TST, 0));
516 }
517 } else {
518 // We use the QualType mangle type variant here because it handles
519 // substitutions.
520 mangleType(QualType(QTy, 0));
521 }
522 }
John McCall1dd73832010-02-04 01:42:13 +0000523 break;
524 case NestedNameSpecifier::Identifier:
John McCallad5e7382010-03-01 23:49:17 +0000525 // Member expressions can have these without prefixes.
526 if (Qualifier->getPrefix())
527 mangleUnresolvedScope(Qualifier->getPrefix());
John McCall1dd73832010-02-04 01:42:13 +0000528 mangleSourceName(Qualifier->getAsIdentifier());
529 break;
530 }
531}
532
533/// Mangles a name which was not resolved to a specific entity.
534void CXXNameMangler::mangleUnresolvedName(NestedNameSpecifier *Qualifier,
535 DeclarationName Name,
536 unsigned KnownArity) {
537 if (Qualifier)
538 mangleUnresolvedScope(Qualifier);
539 // FIXME: ambiguity of unqualified lookup with ::
540
541 mangleUnqualifiedName(0, Name, KnownArity);
542}
543
544void CXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND,
545 DeclarationName Name,
546 unsigned KnownArity) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000547 // <unqualified-name> ::= <operator-name>
Mike Stump1eb44332009-09-09 15:08:12 +0000548 // ::= <ctor-dtor-name>
549 // ::= <source-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000550 switch (Name.getNameKind()) {
Anders Carlssonc4355b62009-10-07 01:45:02 +0000551 case DeclarationName::Identifier: {
Anders Carlssonc4355b62009-10-07 01:45:02 +0000552 if (const IdentifierInfo *II = Name.getAsIdentifierInfo()) {
Sean Hunt31455252010-01-24 03:04:27 +0000553 // We must avoid conflicts between internally- and externally-
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000554 // linked variable declaration names in the same TU.
Anders Carlssonaec25232010-02-06 04:52:27 +0000555 // This naming convention is the same as that followed by GCC, though it
556 // shouldn't actually matter.
557 if (ND && isa<VarDecl>(ND) && ND->getLinkage() == InternalLinkage &&
Sean Hunt31455252010-01-24 03:04:27 +0000558 ND->getDeclContext()->isFileContext())
559 Out << 'L';
560
Anders Carlssonc4355b62009-10-07 01:45:02 +0000561 mangleSourceName(II);
562 break;
563 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000564
John McCall1dd73832010-02-04 01:42:13 +0000565 // Otherwise, an anonymous entity. We must have a declaration.
566 assert(ND && "mangling empty name without declaration");
567
568 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
569 if (NS->isAnonymousNamespace()) {
570 // This is how gcc mangles these names.
571 Out << "12_GLOBAL__N_1";
572 break;
573 }
574 }
575
Anders Carlssonc4355b62009-10-07 01:45:02 +0000576 // We must have an anonymous struct.
577 const TagDecl *TD = cast<TagDecl>(ND);
578 if (const TypedefDecl *D = TD->getTypedefForAnonDecl()) {
579 assert(TD->getDeclContext() == D->getDeclContext() &&
580 "Typedef should not be in another decl context!");
581 assert(D->getDeclName().getAsIdentifierInfo() &&
582 "Typedef was not named!");
583 mangleSourceName(D->getDeclName().getAsIdentifierInfo());
584 break;
585 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000586
Anders Carlssonc4355b62009-10-07 01:45:02 +0000587 // Get a unique id for the anonymous struct.
588 uint64_t AnonStructId = Context.getAnonymousStructId(TD);
589
590 // Mangle it as a source name in the form
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000591 // [n] $_<id>
Anders Carlssonc4355b62009-10-07 01:45:02 +0000592 // where n is the length of the string.
593 llvm::SmallString<8> Str;
594 Str += "$_";
595 Str += llvm::utostr(AnonStructId);
596
597 Out << Str.size();
598 Out << Str.str();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000599 break;
Anders Carlssonc4355b62009-10-07 01:45:02 +0000600 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000601
602 case DeclarationName::ObjCZeroArgSelector:
603 case DeclarationName::ObjCOneArgSelector:
604 case DeclarationName::ObjCMultiArgSelector:
605 assert(false && "Can't mangle Objective-C selector names here!");
606 break;
607
608 case DeclarationName::CXXConstructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000609 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000610 // If the named decl is the C++ constructor we're mangling, use the type
611 // we were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000612 mangleCXXCtorType(static_cast<CXXCtorType>(StructorType));
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000613 else
614 // Otherwise, use the complete constructor name. This is relevant if a
615 // class with a constructor is declared within a constructor.
616 mangleCXXCtorType(Ctor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000617 break;
618
619 case DeclarationName::CXXDestructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000620 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000621 // If the named decl is the C++ destructor we're mangling, use the type we
622 // were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000623 mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
624 else
625 // Otherwise, use the complete destructor name. This is relevant if a
626 // class with a destructor is declared within a destructor.
627 mangleCXXDtorType(Dtor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000628 break;
629
630 case DeclarationName::CXXConversionFunctionName:
Mike Stump1eb44332009-09-09 15:08:12 +0000631 // <operator-name> ::= cv <type> # (cast)
Douglas Gregor219cc612009-02-13 01:28:03 +0000632 Out << "cv";
Anders Carlssonb5404912009-10-07 01:06:45 +0000633 mangleType(Context.getASTContext().getCanonicalType(Name.getCXXNameType()));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000634 break;
635
Anders Carlsson8257d412009-12-22 06:36:32 +0000636 case DeclarationName::CXXOperatorName: {
John McCall1dd73832010-02-04 01:42:13 +0000637 unsigned Arity;
638 if (ND) {
639 Arity = cast<FunctionDecl>(ND)->getNumParams();
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000640
John McCall1dd73832010-02-04 01:42:13 +0000641 // If we have a C++ member function, we need to include the 'this' pointer.
642 // FIXME: This does not make sense for operators that are static, but their
643 // names stay the same regardless of the arity (operator new for instance).
644 if (isa<CXXMethodDecl>(ND))
645 Arity++;
646 } else
647 Arity = KnownArity;
648
Anders Carlsson8257d412009-12-22 06:36:32 +0000649 mangleOperatorName(Name.getCXXOverloadedOperator(), Arity);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000650 break;
Anders Carlsson8257d412009-12-22 06:36:32 +0000651 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000652
Sean Hunt3e518bd2009-11-29 07:34:05 +0000653 case DeclarationName::CXXLiteralOperatorName:
Sean Hunt5dd6b392009-12-04 21:11:13 +0000654 // FIXME: This mangling is not yet official.
Sean Hunt2421f662009-12-04 21:01:37 +0000655 Out << "li";
Sean Hunt3e518bd2009-11-29 07:34:05 +0000656 mangleSourceName(Name.getCXXLiteralIdentifier());
657 break;
658
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000659 case DeclarationName::CXXUsingDirective:
660 assert(false && "Can't mangle a using directive name!");
Douglas Gregor219cc612009-02-13 01:28:03 +0000661 break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000662 }
663}
664
665void CXXNameMangler::mangleSourceName(const IdentifierInfo *II) {
666 // <source-name> ::= <positive length number> <identifier>
667 // <number> ::= [n] <non-negative decimal integer>
668 // <identifier> ::= <unqualified source code identifier>
669 Out << II->getLength() << II->getName();
670}
671
Eli Friedman7facf842009-12-02 20:32:49 +0000672void CXXNameMangler::mangleNestedName(const NamedDecl *ND,
Fariborz Jahanian57058532010-03-03 19:41:08 +0000673 const DeclContext *DC,
674 bool NoFunction) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000675 // <nested-name> ::= N [<CV-qualifiers>] <prefix> <unqualified-name> E
676 // ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
Anders Carlssond99edc42009-09-26 03:55:37 +0000677
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000678 Out << 'N';
679 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(ND))
John McCall0953e762009-09-24 19:53:00 +0000680 mangleQualifiers(Qualifiers::fromCVRMask(Method->getTypeQualifiers()));
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000681
Anders Carlsson2744a062009-09-18 19:00:18 +0000682 // Check if we have a template.
683 const TemplateArgumentList *TemplateArgs = 0;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000684 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000685 mangleTemplatePrefix(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000686 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
687 mangleTemplateArgs(*TemplateParameters, *TemplateArgs);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000688 }
689 else {
690 manglePrefix(DC, NoFunction);
Anders Carlsson7482e242009-09-18 04:29:09 +0000691 mangleUnqualifiedName(ND);
692 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000693
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000694 Out << 'E';
695}
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000696void CXXNameMangler::mangleNestedName(const TemplateDecl *TD,
Anders Carlsson7624f212009-09-18 02:42:01 +0000697 const TemplateArgument *TemplateArgs,
698 unsigned NumTemplateArgs) {
Anders Carlssone45117b2009-09-27 19:53:49 +0000699 // <nested-name> ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
700
Anders Carlsson7624f212009-09-18 02:42:01 +0000701 Out << 'N';
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000702
Anders Carlssone45117b2009-09-27 19:53:49 +0000703 mangleTemplatePrefix(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000704 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
705 mangleTemplateArgs(*TemplateParameters, TemplateArgs, NumTemplateArgs);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000706
Anders Carlsson7624f212009-09-18 02:42:01 +0000707 Out << 'E';
708}
709
Anders Carlsson1b42c792009-04-02 16:24:45 +0000710void CXXNameMangler::mangleLocalName(const NamedDecl *ND) {
711 // <local-name> := Z <function encoding> E <entity name> [<discriminator>]
712 // := Z <function encoding> E s [<discriminator>]
Mike Stump1eb44332009-09-09 15:08:12 +0000713 // <discriminator> := _ <non-negative number>
Fariborz Jahanian57058532010-03-03 19:41:08 +0000714 const DeclContext *DC = ND->getDeclContext();
Anders Carlsson1b42c792009-04-02 16:24:45 +0000715 Out << 'Z';
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000716
Fariborz Jahanian57058532010-03-03 19:41:08 +0000717 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(DC))
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000718 mangleObjCMethodName(MD);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000719 else if (const DeclContext *CDC = GetLocalClassFunctionDeclContext(DC)) {
720 mangleFunctionEncoding(cast<FunctionDecl>(CDC));
721 Out << 'E';
722 mangleNestedName(ND, DC, true /*NoFunction*/);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000723
Fariborz Jahanian4819ac42010-03-04 01:02:03 +0000724 // FIXME. This still does not cover all cases.
725 unsigned disc;
726 if (Context.getNextDiscriminator(ND, disc)) {
727 if (disc < 10)
728 Out << '_' << disc;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000729 else
Fariborz Jahanian4819ac42010-03-04 01:02:03 +0000730 Out << "__" << disc << '_';
731 }
Fariborz Jahanian57058532010-03-03 19:41:08 +0000732
733 return;
734 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000735 else
Fariborz Jahanian57058532010-03-03 19:41:08 +0000736 mangleFunctionEncoding(cast<FunctionDecl>(DC));
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000737
Anders Carlsson1b42c792009-04-02 16:24:45 +0000738 Out << 'E';
Eli Friedman6f9f25d2009-12-11 20:21:38 +0000739 mangleUnqualifiedName(ND);
Anders Carlsson1b42c792009-04-02 16:24:45 +0000740}
741
Fariborz Jahanian57058532010-03-03 19:41:08 +0000742void CXXNameMangler::manglePrefix(const DeclContext *DC, bool NoFunction) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000743 // <prefix> ::= <prefix> <unqualified-name>
744 // ::= <template-prefix> <template-args>
745 // ::= <template-param>
746 // ::= # empty
747 // ::= <substitution>
Anders Carlsson6862fc72009-09-17 04:16:28 +0000748
Anders Carlssonadd28822009-09-22 20:33:31 +0000749 while (isa<LinkageSpecDecl>(DC))
750 DC = DC->getParent();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000751
Anders Carlsson9263e912009-09-18 18:39:58 +0000752 if (DC->isTranslationUnit())
753 return;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000754
Anders Carlsson6862fc72009-09-17 04:16:28 +0000755 if (mangleSubstitution(cast<NamedDecl>(DC)))
756 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000757
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000758 // Check if we have a template.
759 const TemplateArgumentList *TemplateArgs = 0;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000760 if (const TemplateDecl *TD = isTemplate(cast<NamedDecl>(DC), TemplateArgs)) {
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000761 mangleTemplatePrefix(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000762 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
763 mangleTemplateArgs(*TemplateParameters, *TemplateArgs);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000764 }
765 else if(NoFunction && isa<FunctionDecl>(DC))
766 return;
767 else {
768 manglePrefix(DC->getParent(), NoFunction);
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000769 mangleUnqualifiedName(cast<NamedDecl>(DC));
770 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000771
Anders Carlsson6862fc72009-09-17 04:16:28 +0000772 addSubstitution(cast<NamedDecl>(DC));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000773}
774
Douglas Gregor20f0cc72010-04-23 03:10:43 +0000775void CXXNameMangler::mangleTemplatePrefix(TemplateName Template) {
776 // <template-prefix> ::= <prefix> <template unqualified-name>
777 // ::= <template-param>
778 // ::= <substitution>
779 if (TemplateDecl *TD = Template.getAsTemplateDecl())
780 return mangleTemplatePrefix(TD);
781
782 if (QualifiedTemplateName *Qualified = Template.getAsQualifiedTemplateName())
783 mangleUnresolvedScope(Qualified->getQualifier());
Sean Huntc3021132010-05-05 15:23:54 +0000784
Douglas Gregor20f0cc72010-04-23 03:10:43 +0000785 if (OverloadedTemplateStorage *Overloaded
786 = Template.getAsOverloadedTemplate()) {
Sean Huntc3021132010-05-05 15:23:54 +0000787 mangleUnqualifiedName(0, (*Overloaded->begin())->getDeclName(),
Douglas Gregor20f0cc72010-04-23 03:10:43 +0000788 UnknownArity);
789 return;
790 }
Sean Huntc3021132010-05-05 15:23:54 +0000791
Douglas Gregor20f0cc72010-04-23 03:10:43 +0000792 DependentTemplateName *Dependent = Template.getAsDependentTemplateName();
793 assert(Dependent && "Unknown template name kind?");
794 mangleUnresolvedScope(Dependent->getQualifier());
Douglas Gregor1e9268e2010-04-28 05:58:56 +0000795 mangleUnscopedTemplateName(Template);
Douglas Gregor20f0cc72010-04-23 03:10:43 +0000796}
797
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000798void CXXNameMangler::mangleTemplatePrefix(const TemplateDecl *ND) {
Anders Carlsson7482e242009-09-18 04:29:09 +0000799 // <template-prefix> ::= <prefix> <template unqualified-name>
800 // ::= <template-param>
801 // ::= <substitution>
Douglas Gregor32fb4e12010-02-05 20:45:00 +0000802 // <template-template-param> ::= <template-param>
803 // <substitution>
Anders Carlsson7482e242009-09-18 04:29:09 +0000804
Anders Carlssonaeb85372009-09-26 22:18:22 +0000805 if (mangleSubstitution(ND))
806 return;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000807
Douglas Gregor32fb4e12010-02-05 20:45:00 +0000808 // <template-template-param> ::= <template-param>
809 if (const TemplateTemplateParmDecl *TTP
810 = dyn_cast<TemplateTemplateParmDecl>(ND)) {
811 mangleTemplateParameter(TTP->getIndex());
812 return;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000813 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000814
Anders Carlssonaa73ab12009-09-18 18:47:07 +0000815 manglePrefix(ND->getDeclContext());
Anders Carlsson1668f202009-09-26 20:13:56 +0000816 mangleUnqualifiedName(ND->getTemplatedDecl());
Anders Carlssonaeb85372009-09-26 22:18:22 +0000817 addSubstitution(ND);
Anders Carlsson7482e242009-09-18 04:29:09 +0000818}
819
Mike Stump1eb44332009-09-09 15:08:12 +0000820void
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000821CXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity) {
822 switch (OO) {
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000823 // <operator-name> ::= nw # new
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000824 case OO_New: Out << "nw"; break;
825 // ::= na # new[]
826 case OO_Array_New: Out << "na"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000827 // ::= dl # delete
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000828 case OO_Delete: Out << "dl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000829 // ::= da # delete[]
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000830 case OO_Array_Delete: Out << "da"; break;
831 // ::= ps # + (unary)
832 // ::= pl # +
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000833 case OO_Plus:
Anders Carlsson8257d412009-12-22 06:36:32 +0000834 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
835 Out << (Arity == 1? "ps" : "pl"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000836 // ::= ng # - (unary)
837 // ::= mi # -
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000838 case OO_Minus:
Anders Carlsson8257d412009-12-22 06:36:32 +0000839 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
840 Out << (Arity == 1? "ng" : "mi"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000841 // ::= ad # & (unary)
842 // ::= an # &
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000843 case OO_Amp:
Anders Carlsson8257d412009-12-22 06:36:32 +0000844 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
845 Out << (Arity == 1? "ad" : "an"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000846 // ::= de # * (unary)
847 // ::= ml # *
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000848 case OO_Star:
Anders Carlsson8257d412009-12-22 06:36:32 +0000849 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
850 Out << (Arity == 1? "de" : "ml"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000851 // ::= co # ~
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000852 case OO_Tilde: Out << "co"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000853 // ::= dv # /
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000854 case OO_Slash: Out << "dv"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000855 // ::= rm # %
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000856 case OO_Percent: Out << "rm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000857 // ::= or # |
858 case OO_Pipe: Out << "or"; break;
859 // ::= eo # ^
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000860 case OO_Caret: Out << "eo"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000861 // ::= aS # =
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000862 case OO_Equal: Out << "aS"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000863 // ::= pL # +=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000864 case OO_PlusEqual: Out << "pL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000865 // ::= mI # -=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000866 case OO_MinusEqual: Out << "mI"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000867 // ::= mL # *=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000868 case OO_StarEqual: Out << "mL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000869 // ::= dV # /=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000870 case OO_SlashEqual: Out << "dV"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000871 // ::= rM # %=
872 case OO_PercentEqual: Out << "rM"; break;
873 // ::= aN # &=
874 case OO_AmpEqual: Out << "aN"; break;
875 // ::= oR # |=
876 case OO_PipeEqual: Out << "oR"; break;
877 // ::= eO # ^=
878 case OO_CaretEqual: Out << "eO"; break;
879 // ::= ls # <<
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000880 case OO_LessLess: Out << "ls"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000881 // ::= rs # >>
882 case OO_GreaterGreater: Out << "rs"; break;
883 // ::= lS # <<=
884 case OO_LessLessEqual: Out << "lS"; break;
885 // ::= rS # >>=
886 case OO_GreaterGreaterEqual: Out << "rS"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000887 // ::= eq # ==
888 case OO_EqualEqual: Out << "eq"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000889 // ::= ne # !=
890 case OO_ExclaimEqual: Out << "ne"; break;
891 // ::= lt # <
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000892 case OO_Less: Out << "lt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000893 // ::= gt # >
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000894 case OO_Greater: Out << "gt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000895 // ::= le # <=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000896 case OO_LessEqual: Out << "le"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000897 // ::= ge # >=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000898 case OO_GreaterEqual: Out << "ge"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000899 // ::= nt # !
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000900 case OO_Exclaim: Out << "nt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000901 // ::= aa # &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000902 case OO_AmpAmp: Out << "aa"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000903 // ::= oo # ||
904 case OO_PipePipe: Out << "oo"; break;
905 // ::= pp # ++
906 case OO_PlusPlus: Out << "pp"; break;
907 // ::= mm # --
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000908 case OO_MinusMinus: Out << "mm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000909 // ::= cm # ,
910 case OO_Comma: Out << "cm"; break;
911 // ::= pm # ->*
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000912 case OO_ArrowStar: Out << "pm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000913 // ::= pt # ->
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000914 case OO_Arrow: Out << "pt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000915 // ::= cl # ()
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000916 case OO_Call: Out << "cl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000917 // ::= ix # []
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000918 case OO_Subscript: Out << "ix"; break;
Anders Carlssone170ba72009-12-14 01:45:37 +0000919
920 // ::= qu # ?
921 // The conditional operator can't be overloaded, but we still handle it when
922 // mangling expressions.
923 case OO_Conditional: Out << "qu"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000924
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000925 case OO_None:
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000926 case NUM_OVERLOADED_OPERATORS:
Mike Stump1eb44332009-09-09 15:08:12 +0000927 assert(false && "Not an overloaded operator");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000928 break;
929 }
930}
931
John McCall0953e762009-09-24 19:53:00 +0000932void CXXNameMangler::mangleQualifiers(Qualifiers Quals) {
Mike Stump1eb44332009-09-09 15:08:12 +0000933 // <CV-qualifiers> ::= [r] [V] [K] # restrict (C99), volatile, const
John McCall0953e762009-09-24 19:53:00 +0000934 if (Quals.hasRestrict())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000935 Out << 'r';
John McCall0953e762009-09-24 19:53:00 +0000936 if (Quals.hasVolatile())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000937 Out << 'V';
John McCall0953e762009-09-24 19:53:00 +0000938 if (Quals.hasConst())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000939 Out << 'K';
John McCall0953e762009-09-24 19:53:00 +0000940
941 // FIXME: For now, just drop all extension qualifiers on the floor.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000942}
943
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000944void CXXNameMangler::mangleObjCMethodName(const ObjCMethodDecl *MD) {
945 llvm::SmallString<64> Name;
946 llvm::raw_svector_ostream OS(Name);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000947
948 const ObjCContainerDecl *CD =
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000949 dyn_cast<ObjCContainerDecl>(MD->getDeclContext());
950 assert (CD && "Missing container decl in GetNameForMethod");
951 OS << (MD->isInstanceMethod() ? '-' : '+') << '[' << CD->getName();
952 if (const ObjCCategoryImplDecl *CID = dyn_cast<ObjCCategoryImplDecl>(CD))
Benjamin Kramer900fc632010-04-17 09:33:03 +0000953 OS << '(' << CID << ')';
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000954 OS << ' ' << MD->getSelector().getAsString() << ']';
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000955
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000956 Out << OS.str().size() << OS.str();
957}
958
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000959void CXXNameMangler::mangleType(QualType T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000960 // Only operate on the canonical type!
Anders Carlssonb5404912009-10-07 01:06:45 +0000961 T = Context.getASTContext().getCanonicalType(T);
Anders Carlsson4843e582009-03-10 17:07:44 +0000962
Douglas Gregora4923eb2009-11-16 21:35:15 +0000963 bool IsSubstitutable = T.hasLocalQualifiers() || !isa<BuiltinType>(T);
Anders Carlsson76967372009-09-17 00:43:46 +0000964 if (IsSubstitutable && mangleSubstitution(T))
965 return;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000966
Douglas Gregora4923eb2009-11-16 21:35:15 +0000967 if (Qualifiers Quals = T.getLocalQualifiers()) {
John McCall0953e762009-09-24 19:53:00 +0000968 mangleQualifiers(Quals);
969 // Recurse: even if the qualified type isn't yet substitutable,
970 // the unqualified type might be.
Douglas Gregora4923eb2009-11-16 21:35:15 +0000971 mangleType(T.getLocalUnqualifiedType());
Anders Carlsson76967372009-09-17 00:43:46 +0000972 } else {
973 switch (T->getTypeClass()) {
John McCallefe6aee2009-09-05 07:56:18 +0000974#define ABSTRACT_TYPE(CLASS, PARENT)
975#define NON_CANONICAL_TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000976 case Type::CLASS: \
Jeffrey Yasskin9f61aa92009-12-12 05:05:38 +0000977 llvm_unreachable("can't mangle non-canonical type " #CLASS "Type"); \
Anders Carlsson76967372009-09-17 00:43:46 +0000978 return;
John McCallefe6aee2009-09-05 07:56:18 +0000979#define TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000980 case Type::CLASS: \
John McCall0953e762009-09-24 19:53:00 +0000981 mangleType(static_cast<const CLASS##Type*>(T.getTypePtr())); \
Anders Carlsson76967372009-09-17 00:43:46 +0000982 break;
John McCallefe6aee2009-09-05 07:56:18 +0000983#include "clang/AST/TypeNodes.def"
Anders Carlsson76967372009-09-17 00:43:46 +0000984 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000985 }
Anders Carlsson76967372009-09-17 00:43:46 +0000986
987 // Add the substitution.
988 if (IsSubstitutable)
989 addSubstitution(T);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000990}
991
992void CXXNameMangler::mangleType(const BuiltinType *T) {
John McCallefe6aee2009-09-05 07:56:18 +0000993 // <type> ::= <builtin-type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000994 // <builtin-type> ::= v # void
995 // ::= w # wchar_t
996 // ::= b # bool
997 // ::= c # char
998 // ::= a # signed char
999 // ::= h # unsigned char
1000 // ::= s # short
1001 // ::= t # unsigned short
1002 // ::= i # int
1003 // ::= j # unsigned int
1004 // ::= l # long
1005 // ::= m # unsigned long
1006 // ::= x # long long, __int64
1007 // ::= y # unsigned long long, __int64
1008 // ::= n # __int128
1009 // UNSUPPORTED: ::= o # unsigned __int128
1010 // ::= f # float
1011 // ::= d # double
1012 // ::= e # long double, __float80
1013 // UNSUPPORTED: ::= g # __float128
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001014 // UNSUPPORTED: ::= Dd # IEEE 754r decimal floating point (64 bits)
1015 // UNSUPPORTED: ::= De # IEEE 754r decimal floating point (128 bits)
1016 // UNSUPPORTED: ::= Df # IEEE 754r decimal floating point (32 bits)
1017 // UNSUPPORTED: ::= Dh # IEEE 754r half-precision floating point (16 bits)
Alisdair Meredithf5c209d2009-07-14 06:30:34 +00001018 // ::= Di # char32_t
1019 // ::= Ds # char16_t
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001020 // ::= u <source-name> # vendor extended type
Sebastian Redl6e8ed162009-05-10 18:38:11 +00001021 // From our point of view, std::nullptr_t is a builtin, but as far as mangling
1022 // is concerned, it's a type called std::nullptr_t.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001023 switch (T->getKind()) {
1024 case BuiltinType::Void: Out << 'v'; break;
1025 case BuiltinType::Bool: Out << 'b'; break;
1026 case BuiltinType::Char_U: case BuiltinType::Char_S: Out << 'c'; break;
1027 case BuiltinType::UChar: Out << 'h'; break;
1028 case BuiltinType::UShort: Out << 't'; break;
1029 case BuiltinType::UInt: Out << 'j'; break;
1030 case BuiltinType::ULong: Out << 'm'; break;
1031 case BuiltinType::ULongLong: Out << 'y'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +00001032 case BuiltinType::UInt128: Out << 'o'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001033 case BuiltinType::SChar: Out << 'a'; break;
1034 case BuiltinType::WChar: Out << 'w'; break;
Alisdair Meredithf5c209d2009-07-14 06:30:34 +00001035 case BuiltinType::Char16: Out << "Ds"; break;
1036 case BuiltinType::Char32: Out << "Di"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001037 case BuiltinType::Short: Out << 's'; break;
1038 case BuiltinType::Int: Out << 'i'; break;
1039 case BuiltinType::Long: Out << 'l'; break;
1040 case BuiltinType::LongLong: Out << 'x'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +00001041 case BuiltinType::Int128: Out << 'n'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001042 case BuiltinType::Float: Out << 'f'; break;
1043 case BuiltinType::Double: Out << 'd'; break;
1044 case BuiltinType::LongDouble: Out << 'e'; break;
Sebastian Redl6e8ed162009-05-10 18:38:11 +00001045 case BuiltinType::NullPtr: Out << "St9nullptr_t"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001046
1047 case BuiltinType::Overload:
1048 case BuiltinType::Dependent:
Mike Stump1eb44332009-09-09 15:08:12 +00001049 assert(false &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001050 "Overloaded and dependent types shouldn't get to name mangling");
1051 break;
Anders Carlssone89d1592009-06-26 18:41:36 +00001052 case BuiltinType::UndeducedAuto:
1053 assert(0 && "Should not see undeduced auto here");
1054 break;
Steve Naroff9533a7f2009-07-22 17:14:51 +00001055 case BuiltinType::ObjCId: Out << "11objc_object"; break;
1056 case BuiltinType::ObjCClass: Out << "10objc_class"; break;
Fariborz Jahanian13dcd002009-11-21 19:53:08 +00001057 case BuiltinType::ObjCSel: Out << "13objc_selector"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001058 }
1059}
1060
John McCallefe6aee2009-09-05 07:56:18 +00001061// <type> ::= <function-type>
1062// <function-type> ::= F [Y] <bare-function-type> E
1063void CXXNameMangler::mangleType(const FunctionProtoType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001064 Out << 'F';
Mike Stumpf5408fe2009-05-16 07:57:57 +00001065 // FIXME: We don't have enough information in the AST to produce the 'Y'
1066 // encoding for extern "C" function types.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001067 mangleBareFunctionType(T, /*MangleReturnType=*/true);
1068 Out << 'E';
1069}
John McCallefe6aee2009-09-05 07:56:18 +00001070void CXXNameMangler::mangleType(const FunctionNoProtoType *T) {
Jeffrey Yasskin9f61aa92009-12-12 05:05:38 +00001071 llvm_unreachable("Can't mangle K&R function prototypes");
John McCallefe6aee2009-09-05 07:56:18 +00001072}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001073void CXXNameMangler::mangleBareFunctionType(const FunctionType *T,
1074 bool MangleReturnType) {
John McCallefe6aee2009-09-05 07:56:18 +00001075 // We should never be mangling something without a prototype.
1076 const FunctionProtoType *Proto = cast<FunctionProtoType>(T);
1077
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001078 // <bare-function-type> ::= <signature type>+
1079 if (MangleReturnType)
John McCallefe6aee2009-09-05 07:56:18 +00001080 mangleType(Proto->getResultType());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001081
Anders Carlssonc6c91bc2009-04-01 00:15:23 +00001082 if (Proto->getNumArgs() == 0) {
1083 Out << 'v';
1084 return;
1085 }
Mike Stump1eb44332009-09-09 15:08:12 +00001086
Douglas Gregor72564e72009-02-26 23:50:07 +00001087 for (FunctionProtoType::arg_type_iterator Arg = Proto->arg_type_begin(),
Mike Stump1eb44332009-09-09 15:08:12 +00001088 ArgEnd = Proto->arg_type_end();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001089 Arg != ArgEnd; ++Arg)
1090 mangleType(*Arg);
Douglas Gregor219cc612009-02-13 01:28:03 +00001091
1092 // <builtin-type> ::= z # ellipsis
1093 if (Proto->isVariadic())
1094 Out << 'z';
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001095}
1096
John McCallefe6aee2009-09-05 07:56:18 +00001097// <type> ::= <class-enum-type>
Mike Stump1eb44332009-09-09 15:08:12 +00001098// <class-enum-type> ::= <name>
John McCalled976492009-12-04 22:46:56 +00001099void CXXNameMangler::mangleType(const UnresolvedUsingType *T) {
1100 mangleName(T->getDecl());
1101}
1102
1103// <type> ::= <class-enum-type>
1104// <class-enum-type> ::= <name>
John McCallefe6aee2009-09-05 07:56:18 +00001105void CXXNameMangler::mangleType(const EnumType *T) {
1106 mangleType(static_cast<const TagType*>(T));
1107}
1108void CXXNameMangler::mangleType(const RecordType *T) {
1109 mangleType(static_cast<const TagType*>(T));
1110}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001111void CXXNameMangler::mangleType(const TagType *T) {
Eli Friedmanecb7e932009-12-11 18:00:57 +00001112 mangleName(T->getDecl());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001113}
1114
John McCallefe6aee2009-09-05 07:56:18 +00001115// <type> ::= <array-type>
1116// <array-type> ::= A <positive dimension number> _ <element type>
1117// ::= A [<dimension expression>] _ <element type>
1118void CXXNameMangler::mangleType(const ConstantArrayType *T) {
1119 Out << 'A' << T->getSize() << '_';
1120 mangleType(T->getElementType());
1121}
1122void CXXNameMangler::mangleType(const VariableArrayType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001123 Out << 'A';
John McCallefe6aee2009-09-05 07:56:18 +00001124 mangleExpression(T->getSizeExpr());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001125 Out << '_';
1126 mangleType(T->getElementType());
1127}
John McCallefe6aee2009-09-05 07:56:18 +00001128void CXXNameMangler::mangleType(const DependentSizedArrayType *T) {
1129 Out << 'A';
1130 mangleExpression(T->getSizeExpr());
1131 Out << '_';
1132 mangleType(T->getElementType());
1133}
1134void CXXNameMangler::mangleType(const IncompleteArrayType *T) {
1135 Out << 'A' << '_';
1136 mangleType(T->getElementType());
1137}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001138
John McCallefe6aee2009-09-05 07:56:18 +00001139// <type> ::= <pointer-to-member-type>
1140// <pointer-to-member-type> ::= M <class type> <member type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001141void CXXNameMangler::mangleType(const MemberPointerType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001142 Out << 'M';
1143 mangleType(QualType(T->getClass(), 0));
Anders Carlsson0e650012009-05-17 17:41:20 +00001144 QualType PointeeType = T->getPointeeType();
1145 if (const FunctionProtoType *FPT = dyn_cast<FunctionProtoType>(PointeeType)) {
John McCall0953e762009-09-24 19:53:00 +00001146 mangleQualifiers(Qualifiers::fromCVRMask(FPT->getTypeQuals()));
Anders Carlsson0e650012009-05-17 17:41:20 +00001147 mangleType(FPT);
Mike Stump1eb44332009-09-09 15:08:12 +00001148 } else
Anders Carlsson0e650012009-05-17 17:41:20 +00001149 mangleType(PointeeType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001150}
1151
John McCallefe6aee2009-09-05 07:56:18 +00001152// <type> ::= <template-param>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001153void CXXNameMangler::mangleType(const TemplateTypeParmType *T) {
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001154 mangleTemplateParameter(T->getIndex());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001155}
1156
John McCallefe6aee2009-09-05 07:56:18 +00001157// FIXME: <type> ::= <template-template-param> <template-args>
John McCallefe6aee2009-09-05 07:56:18 +00001158
1159// <type> ::= P <type> # pointer-to
1160void CXXNameMangler::mangleType(const PointerType *T) {
1161 Out << 'P';
1162 mangleType(T->getPointeeType());
1163}
1164void CXXNameMangler::mangleType(const ObjCObjectPointerType *T) {
1165 Out << 'P';
1166 mangleType(T->getPointeeType());
1167}
1168
1169// <type> ::= R <type> # reference-to
1170void CXXNameMangler::mangleType(const LValueReferenceType *T) {
1171 Out << 'R';
1172 mangleType(T->getPointeeType());
1173}
1174
1175// <type> ::= O <type> # rvalue reference-to (C++0x)
1176void CXXNameMangler::mangleType(const RValueReferenceType *T) {
1177 Out << 'O';
1178 mangleType(T->getPointeeType());
1179}
1180
1181// <type> ::= C <type> # complex pair (C 2000)
1182void CXXNameMangler::mangleType(const ComplexType *T) {
1183 Out << 'C';
1184 mangleType(T->getElementType());
1185}
1186
1187// GNU extension: vector types
Nick Lewycky0e5f0672010-03-26 07:18:04 +00001188// <type> ::= <vector-type>
1189// <vector-type> ::= Dv <positive dimension number> _ <element type>
1190// ::= Dv [<dimension expression>] _ <element type>
John McCallefe6aee2009-09-05 07:56:18 +00001191void CXXNameMangler::mangleType(const VectorType *T) {
Nick Lewycky0e5f0672010-03-26 07:18:04 +00001192 Out << "Dv" << T->getNumElements() << '_';
John McCallefe6aee2009-09-05 07:56:18 +00001193 mangleType(T->getElementType());
1194}
1195void CXXNameMangler::mangleType(const ExtVectorType *T) {
1196 mangleType(static_cast<const VectorType*>(T));
1197}
1198void CXXNameMangler::mangleType(const DependentSizedExtVectorType *T) {
Nick Lewycky0e5f0672010-03-26 07:18:04 +00001199 Out << "Dv";
1200 mangleExpression(T->getSizeExpr());
1201 Out << '_';
John McCallefe6aee2009-09-05 07:56:18 +00001202 mangleType(T->getElementType());
1203}
1204
Anders Carlssona40c5e42009-03-07 22:03:21 +00001205void CXXNameMangler::mangleType(const ObjCInterfaceType *T) {
1206 mangleSourceName(T->getDecl()->getIdentifier());
1207}
1208
John McCallefe6aee2009-09-05 07:56:18 +00001209void CXXNameMangler::mangleType(const BlockPointerType *T) {
Anders Carlssonf28c6872009-12-23 22:31:44 +00001210 Out << "U13block_pointer";
1211 mangleType(T->getPointeeType());
John McCallefe6aee2009-09-05 07:56:18 +00001212}
1213
John McCall31f17ec2010-04-27 00:57:59 +00001214void CXXNameMangler::mangleType(const InjectedClassNameType *T) {
1215 // Mangle injected class name types as if the user had written the
1216 // specialization out fully. It may not actually be possible to see
1217 // this mangling, though.
1218 mangleType(T->getInjectedSpecializationType());
1219}
1220
John McCallefe6aee2009-09-05 07:56:18 +00001221void CXXNameMangler::mangleType(const TemplateSpecializationType *T) {
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001222 if (TemplateDecl *TD = T->getTemplateName().getAsTemplateDecl()) {
1223 mangleName(TD, T->getArgs(), T->getNumArgs());
1224 } else {
1225 if (mangleSubstitution(QualType(T, 0)))
1226 return;
Sean Huntc3021132010-05-05 15:23:54 +00001227
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001228 mangleTemplatePrefix(T->getTemplateName());
Sean Huntc3021132010-05-05 15:23:54 +00001229
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001230 // FIXME: GCC does not appear to mangle the template arguments when
1231 // the template in question is a dependent template name. Should we
1232 // emulate that badness?
1233 mangleTemplateArgs(T->getTemplateName(), T->getArgs(), T->getNumArgs());
1234 addSubstitution(QualType(T, 0));
1235 }
John McCallefe6aee2009-09-05 07:56:18 +00001236}
1237
Douglas Gregor4714c122010-03-31 17:34:00 +00001238void CXXNameMangler::mangleType(const DependentNameType *T) {
Anders Carlssonae352482009-09-26 02:26:02 +00001239 // Typename types are always nested
1240 Out << 'N';
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001241 if (T->getIdentifier()) {
1242 mangleUnresolvedScope(T->getQualifier());
1243 mangleSourceName(T->getIdentifier());
1244 } else {
1245 const TemplateSpecializationType *TST = T->getTemplateId();
1246 if (!mangleSubstitution(QualType(TST, 0))) {
1247 mangleTemplatePrefix(TST->getTemplateName());
Sean Huntc3021132010-05-05 15:23:54 +00001248
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001249 // FIXME: GCC does not appear to mangle the template arguments when
1250 // the template in question is a dependent template name. Should we
1251 // emulate that badness?
1252 mangleTemplateArgs(TST->getTemplateName(), TST->getArgs(),
Sean Huntc3021132010-05-05 15:23:54 +00001253 TST->getNumArgs());
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001254 addSubstitution(QualType(TST, 0));
1255 }
1256 }
Sean Huntc3021132010-05-05 15:23:54 +00001257
Anders Carlssonae352482009-09-26 02:26:02 +00001258 Out << 'E';
John McCallefe6aee2009-09-05 07:56:18 +00001259}
1260
John McCallad5e7382010-03-01 23:49:17 +00001261void CXXNameMangler::mangleType(const TypeOfType *T) {
1262 // FIXME: this is pretty unsatisfactory, but there isn't an obvious
1263 // "extension with parameters" mangling.
1264 Out << "u6typeof";
1265}
1266
1267void CXXNameMangler::mangleType(const TypeOfExprType *T) {
1268 // FIXME: this is pretty unsatisfactory, but there isn't an obvious
1269 // "extension with parameters" mangling.
1270 Out << "u6typeof";
1271}
1272
1273void CXXNameMangler::mangleType(const DecltypeType *T) {
1274 Expr *E = T->getUnderlyingExpr();
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001275
John McCallad5e7382010-03-01 23:49:17 +00001276 // type ::= Dt <expression> E # decltype of an id-expression
1277 // # or class member access
1278 // ::= DT <expression> E # decltype of an expression
1279
1280 // This purports to be an exhaustive list of id-expressions and
1281 // class member accesses. Note that we do not ignore parentheses;
1282 // parentheses change the semantics of decltype for these
1283 // expressions (and cause the mangler to use the other form).
1284 if (isa<DeclRefExpr>(E) ||
1285 isa<MemberExpr>(E) ||
1286 isa<UnresolvedLookupExpr>(E) ||
1287 isa<DependentScopeDeclRefExpr>(E) ||
1288 isa<CXXDependentScopeMemberExpr>(E) ||
1289 isa<UnresolvedMemberExpr>(E))
1290 Out << "Dt";
1291 else
1292 Out << "DT";
1293 mangleExpression(E);
1294 Out << 'E';
1295}
1296
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001297void CXXNameMangler::mangleIntegerLiteral(QualType T,
Anders Carlssone170ba72009-12-14 01:45:37 +00001298 const llvm::APSInt &Value) {
1299 // <expr-primary> ::= L <type> <value number> E # integer literal
1300 Out << 'L';
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001301
Anders Carlssone170ba72009-12-14 01:45:37 +00001302 mangleType(T);
1303 if (T->isBooleanType()) {
1304 // Boolean values are encoded as 0/1.
1305 Out << (Value.getBoolValue() ? '1' : '0');
1306 } else {
1307 if (Value.isNegative())
1308 Out << 'n';
1309 Value.abs().print(Out, false);
1310 }
1311 Out << 'E';
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001312
Anders Carlssone170ba72009-12-14 01:45:37 +00001313}
1314
John McCall1dd73832010-02-04 01:42:13 +00001315void CXXNameMangler::mangleCalledExpression(const Expr *E, unsigned Arity) {
1316 if (E->getType() != getASTContext().OverloadTy)
1317 mangleExpression(E);
John McCall2f27bf82010-02-04 02:56:29 +00001318 // propagate arity to dependent overloads?
John McCall1dd73832010-02-04 01:42:13 +00001319
1320 llvm::PointerIntPair<OverloadExpr*,1> R
1321 = OverloadExpr::find(const_cast<Expr*>(E));
1322 if (R.getInt())
1323 Out << "an"; // &
1324 const OverloadExpr *Ovl = R.getPointer();
John McCall2f27bf82010-02-04 02:56:29 +00001325 if (const UnresolvedMemberExpr *ME = dyn_cast<UnresolvedMemberExpr>(Ovl)) {
1326 mangleMemberExpr(ME->getBase(), ME->isArrow(), ME->getQualifier(),
1327 ME->getMemberName(), Arity);
1328 return;
1329 }
John McCall1dd73832010-02-04 01:42:13 +00001330
1331 mangleUnresolvedName(Ovl->getQualifier(), Ovl->getName(), Arity);
1332}
1333
John McCall2f27bf82010-02-04 02:56:29 +00001334/// Mangles a member expression. Implicit accesses are not handled,
1335/// but that should be okay, because you shouldn't be able to
1336/// make an implicit access in a function template declaration.
John McCall2f27bf82010-02-04 02:56:29 +00001337void CXXNameMangler::mangleMemberExpr(const Expr *Base,
1338 bool IsArrow,
1339 NestedNameSpecifier *Qualifier,
1340 DeclarationName Member,
1341 unsigned Arity) {
John McCalle1e342f2010-03-01 19:12:25 +00001342 // gcc-4.4 uses 'dt' for dot expressions, which is reasonable.
1343 // OTOH, gcc also mangles the name as an expression.
1344 Out << (IsArrow ? "pt" : "dt");
John McCall2f27bf82010-02-04 02:56:29 +00001345 mangleExpression(Base);
1346 mangleUnresolvedName(Qualifier, Member, Arity);
1347}
1348
Anders Carlssond553f8c2009-09-21 01:21:10 +00001349void CXXNameMangler::mangleExpression(const Expr *E) {
1350 // <expression> ::= <unary operator-name> <expression>
John McCall09cc1412010-02-03 00:55:45 +00001351 // ::= <binary operator-name> <expression> <expression>
1352 // ::= <trinary operator-name> <expression> <expression> <expression>
1353 // ::= cl <expression>* E # call
Anders Carlssond553f8c2009-09-21 01:21:10 +00001354 // ::= cv <type> expression # conversion with one argument
1355 // ::= cv <type> _ <expression>* E # conversion with a different number of arguments
John McCall09cc1412010-02-03 00:55:45 +00001356 // ::= st <type> # sizeof (a type)
Anders Carlssond553f8c2009-09-21 01:21:10 +00001357 // ::= at <type> # alignof (a type)
1358 // ::= <template-param>
1359 // ::= <function-param>
1360 // ::= sr <type> <unqualified-name> # dependent name
1361 // ::= sr <type> <unqualified-name> <template-args> # dependent template-id
1362 // ::= sZ <template-param> # size of a parameter pack
John McCall09cc1412010-02-03 00:55:45 +00001363 // ::= <expr-primary>
John McCall1dd73832010-02-04 01:42:13 +00001364 // <expr-primary> ::= L <type> <value number> E # integer literal
1365 // ::= L <type <value float> E # floating literal
1366 // ::= L <mangled-name> E # external name
Anders Carlssond553f8c2009-09-21 01:21:10 +00001367 switch (E->getStmtClass()) {
John McCall6ae1f352010-04-09 22:26:14 +00001368 case Expr::NoStmtClass:
1369#define EXPR(Type, Base)
1370#define STMT(Type, Base) \
1371 case Expr::Type##Class:
Sean Hunt4bfe1962010-05-05 15:24:00 +00001372#include "clang/AST/StmtNodes.inc"
John McCall09cc1412010-02-03 00:55:45 +00001373 llvm_unreachable("unexpected statement kind");
1374 break;
1375
John McCall6ae1f352010-04-09 22:26:14 +00001376 default: {
1377 // As bad as this diagnostic is, it's better than crashing.
1378 Diagnostic &Diags = Context.getDiags();
1379 unsigned DiagID = Diags.getCustomDiagID(Diagnostic::Error,
1380 "cannot yet mangle expression type %0");
John McCall739bf092010-04-10 09:39:25 +00001381 Diags.Report(FullSourceLoc(E->getExprLoc(),
1382 getASTContext().getSourceManager()),
1383 DiagID)
1384 << E->getStmtClassName() << E->getSourceRange();
John McCall6ae1f352010-04-09 22:26:14 +00001385 break;
1386 }
1387
John McCall1dd73832010-02-04 01:42:13 +00001388 case Expr::CallExprClass: {
1389 const CallExpr *CE = cast<CallExpr>(E);
1390 Out << "cl";
1391 mangleCalledExpression(CE->getCallee(), CE->getNumArgs());
1392 for (unsigned I = 0, N = CE->getNumArgs(); I != N; ++I)
1393 mangleExpression(CE->getArg(I));
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001394 Out << 'E';
John McCall09cc1412010-02-03 00:55:45 +00001395 break;
John McCall1dd73832010-02-04 01:42:13 +00001396 }
John McCall09cc1412010-02-03 00:55:45 +00001397
John McCall2f27bf82010-02-04 02:56:29 +00001398 case Expr::MemberExprClass: {
1399 const MemberExpr *ME = cast<MemberExpr>(E);
1400 mangleMemberExpr(ME->getBase(), ME->isArrow(),
1401 ME->getQualifier(), ME->getMemberDecl()->getDeclName(),
1402 UnknownArity);
1403 break;
1404 }
1405
1406 case Expr::UnresolvedMemberExprClass: {
1407 const UnresolvedMemberExpr *ME = cast<UnresolvedMemberExpr>(E);
1408 mangleMemberExpr(ME->getBase(), ME->isArrow(),
1409 ME->getQualifier(), ME->getMemberName(),
1410 UnknownArity);
1411 break;
1412 }
1413
1414 case Expr::CXXDependentScopeMemberExprClass: {
1415 const CXXDependentScopeMemberExpr *ME
1416 = cast<CXXDependentScopeMemberExpr>(E);
1417 mangleMemberExpr(ME->getBase(), ME->isArrow(),
1418 ME->getQualifier(), ME->getMember(),
1419 UnknownArity);
1420 break;
1421 }
1422
John McCall1dd73832010-02-04 01:42:13 +00001423 case Expr::UnresolvedLookupExprClass: {
John McCalla3218e72010-02-04 01:48:38 +00001424 // The ABI doesn't cover how to mangle overload sets, so we mangle
1425 // using something as close as possible to the original lookup
1426 // expression.
John McCall1dd73832010-02-04 01:42:13 +00001427 const UnresolvedLookupExpr *ULE = cast<UnresolvedLookupExpr>(E);
1428 mangleUnresolvedName(ULE->getQualifier(), ULE->getName(), UnknownArity);
1429 break;
1430 }
1431
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001432 case Expr::CXXUnresolvedConstructExprClass: {
John McCall1dd73832010-02-04 01:42:13 +00001433 const CXXUnresolvedConstructExpr *CE = cast<CXXUnresolvedConstructExpr>(E);
1434 unsigned N = CE->arg_size();
1435
1436 Out << "cv";
1437 mangleType(CE->getType());
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001438 if (N != 1) Out << '_';
John McCall1dd73832010-02-04 01:42:13 +00001439 for (unsigned I = 0; I != N; ++I) mangleExpression(CE->getArg(I));
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001440 if (N != 1) Out << 'E';
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001441 break;
John McCall1dd73832010-02-04 01:42:13 +00001442 }
John McCall09cc1412010-02-03 00:55:45 +00001443
John McCall1dd73832010-02-04 01:42:13 +00001444 case Expr::CXXTemporaryObjectExprClass:
1445 case Expr::CXXConstructExprClass: {
1446 const CXXConstructExpr *CE = cast<CXXConstructExpr>(E);
1447 unsigned N = CE->getNumArgs();
1448
1449 Out << "cv";
1450 mangleType(CE->getType());
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001451 if (N != 1) Out << '_';
John McCall1dd73832010-02-04 01:42:13 +00001452 for (unsigned I = 0; I != N; ++I) mangleExpression(CE->getArg(I));
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001453 if (N != 1) Out << 'E';
John McCall09cc1412010-02-03 00:55:45 +00001454 break;
John McCall1dd73832010-02-04 01:42:13 +00001455 }
1456
1457 case Expr::SizeOfAlignOfExprClass: {
1458 const SizeOfAlignOfExpr *SAE = cast<SizeOfAlignOfExpr>(E);
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001459 if (SAE->isSizeOf()) Out << 's';
1460 else Out << 'a';
John McCall1dd73832010-02-04 01:42:13 +00001461 if (SAE->isArgumentType()) {
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001462 Out << 't';
John McCall1dd73832010-02-04 01:42:13 +00001463 mangleType(SAE->getArgumentType());
1464 } else {
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001465 Out << 'z';
John McCall1dd73832010-02-04 01:42:13 +00001466 mangleExpression(SAE->getArgumentExpr());
1467 }
1468 break;
1469 }
Anders Carlssona7694082009-11-06 02:50:19 +00001470
Anders Carlssone170ba72009-12-14 01:45:37 +00001471 case Expr::UnaryOperatorClass: {
1472 const UnaryOperator *UO = cast<UnaryOperator>(E);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001473 mangleOperatorName(UnaryOperator::getOverloadedOperator(UO->getOpcode()),
Anders Carlssone170ba72009-12-14 01:45:37 +00001474 /*Arity=*/1);
1475 mangleExpression(UO->getSubExpr());
1476 break;
1477 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001478
Anders Carlssone170ba72009-12-14 01:45:37 +00001479 case Expr::BinaryOperatorClass: {
1480 const BinaryOperator *BO = cast<BinaryOperator>(E);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001481 mangleOperatorName(BinaryOperator::getOverloadedOperator(BO->getOpcode()),
Anders Carlssone170ba72009-12-14 01:45:37 +00001482 /*Arity=*/2);
1483 mangleExpression(BO->getLHS());
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001484 mangleExpression(BO->getRHS());
Anders Carlssone170ba72009-12-14 01:45:37 +00001485 break;
John McCall2f27bf82010-02-04 02:56:29 +00001486 }
Anders Carlssone170ba72009-12-14 01:45:37 +00001487
1488 case Expr::ConditionalOperatorClass: {
1489 const ConditionalOperator *CO = cast<ConditionalOperator>(E);
1490 mangleOperatorName(OO_Conditional, /*Arity=*/3);
1491 mangleExpression(CO->getCond());
1492 mangleExpression(CO->getLHS());
1493 mangleExpression(CO->getRHS());
1494 break;
1495 }
1496
Douglas Gregor46287c72010-01-29 16:37:09 +00001497 case Expr::ImplicitCastExprClass: {
1498 mangleExpression(cast<ImplicitCastExpr>(E)->getSubExpr());
1499 break;
1500 }
1501
1502 case Expr::CStyleCastExprClass:
1503 case Expr::CXXStaticCastExprClass:
1504 case Expr::CXXDynamicCastExprClass:
1505 case Expr::CXXReinterpretCastExprClass:
1506 case Expr::CXXConstCastExprClass:
1507 case Expr::CXXFunctionalCastExprClass: {
1508 const ExplicitCastExpr *ECE = cast<ExplicitCastExpr>(E);
1509 Out << "cv";
1510 mangleType(ECE->getType());
1511 mangleExpression(ECE->getSubExpr());
1512 break;
1513 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001514
Anders Carlsson58040a52009-12-16 05:48:46 +00001515 case Expr::CXXOperatorCallExprClass: {
1516 const CXXOperatorCallExpr *CE = cast<CXXOperatorCallExpr>(E);
1517 unsigned NumArgs = CE->getNumArgs();
1518 mangleOperatorName(CE->getOperator(), /*Arity=*/NumArgs);
1519 // Mangle the arguments.
1520 for (unsigned i = 0; i != NumArgs; ++i)
1521 mangleExpression(CE->getArg(i));
1522 break;
1523 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001524
Anders Carlssona7694082009-11-06 02:50:19 +00001525 case Expr::ParenExprClass:
1526 mangleExpression(cast<ParenExpr>(E)->getSubExpr());
1527 break;
1528
Anders Carlssond553f8c2009-09-21 01:21:10 +00001529 case Expr::DeclRefExprClass: {
Douglas Gregor5ed1bc32010-02-28 21:40:32 +00001530 const NamedDecl *D = cast<DeclRefExpr>(E)->getDecl();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001531
Anders Carlssond553f8c2009-09-21 01:21:10 +00001532 switch (D->getKind()) {
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001533 default:
Douglas Gregor5ed1bc32010-02-28 21:40:32 +00001534 // <expr-primary> ::= L <mangled-name> E # external name
1535 Out << 'L';
1536 mangle(D, "_Z");
1537 Out << 'E';
1538 break;
1539
Anders Carlssond553f8c2009-09-21 01:21:10 +00001540 case Decl::NonTypeTemplateParm: {
1541 const NonTypeTemplateParmDecl *PD = cast<NonTypeTemplateParmDecl>(D);
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001542 mangleTemplateParameter(PD->getIndex());
Anders Carlssond553f8c2009-09-21 01:21:10 +00001543 break;
1544 }
1545
1546 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001547
Anders Carlsson50755b02009-09-27 20:11:34 +00001548 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001549 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001550
John McCall865d4472009-11-19 22:55:06 +00001551 case Expr::DependentScopeDeclRefExprClass: {
1552 const DependentScopeDeclRefExpr *DRE = cast<DependentScopeDeclRefExpr>(E);
Douglas Gregor4b2ccfc2010-02-28 22:05:49 +00001553 NestedNameSpecifier *NNS = DRE->getQualifier();
1554 const Type *QTy = NNS->getAsType();
1555
1556 // When we're dealing with a nested-name-specifier that has just a
1557 // dependent identifier in it, mangle that as a typename. FIXME:
1558 // It isn't clear that we ever actually want to have such a
1559 // nested-name-specifier; why not just represent it as a typename type?
1560 if (!QTy && NNS->getAsIdentifier() && NNS->getPrefix()) {
Douglas Gregor4a2023f2010-03-31 20:19:30 +00001561 QTy = getASTContext().getDependentNameType(ETK_Typename,
1562 NNS->getPrefix(),
1563 NNS->getAsIdentifier())
Douglas Gregor4b2ccfc2010-02-28 22:05:49 +00001564 .getTypePtr();
1565 }
Anders Carlsson50755b02009-09-27 20:11:34 +00001566 assert(QTy && "Qualifier was not type!");
1567
1568 // ::= sr <type> <unqualified-name> # dependent name
1569 Out << "sr";
1570 mangleType(QualType(QTy, 0));
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001571
Anders Carlsson50755b02009-09-27 20:11:34 +00001572 assert(DRE->getDeclName().getNameKind() == DeclarationName::Identifier &&
1573 "Unhandled decl name kind!");
1574 mangleSourceName(DRE->getDeclName().getAsIdentifierInfo());
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001575
Anders Carlsson50755b02009-09-27 20:11:34 +00001576 break;
1577 }
1578
John McCalld9307602010-04-09 22:54:09 +00001579 case Expr::CXXBindReferenceExprClass:
1580 mangleExpression(cast<CXXBindReferenceExpr>(E)->getSubExpr());
1581 break;
1582
1583 case Expr::CXXBindTemporaryExprClass:
1584 mangleExpression(cast<CXXBindTemporaryExpr>(E)->getSubExpr());
1585 break;
1586
1587 case Expr::CXXExprWithTemporariesClass:
1588 mangleExpression(cast<CXXExprWithTemporaries>(E)->getSubExpr());
1589 break;
1590
John McCall1dd73832010-02-04 01:42:13 +00001591 case Expr::FloatingLiteralClass: {
1592 const FloatingLiteral *FL = cast<FloatingLiteral>(E);
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001593 Out << 'L';
John McCall1dd73832010-02-04 01:42:13 +00001594 mangleType(FL->getType());
1595
1596 // TODO: avoid this copy with careful stream management.
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001597 llvm::SmallString<20> Buffer;
John McCall1dd73832010-02-04 01:42:13 +00001598 FL->getValue().bitcastToAPInt().toString(Buffer, 16, false);
1599 Out.write(Buffer.data(), Buffer.size());
1600
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001601 Out << 'E';
John McCall1dd73832010-02-04 01:42:13 +00001602 break;
1603 }
1604
John McCallde810632010-04-09 21:48:08 +00001605 case Expr::CharacterLiteralClass:
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001606 Out << 'L';
John McCallde810632010-04-09 21:48:08 +00001607 mangleType(E->getType());
1608 Out << cast<CharacterLiteral>(E)->getValue();
1609 Out << 'E';
1610 break;
1611
1612 case Expr::CXXBoolLiteralExprClass:
1613 Out << "Lb";
1614 Out << (cast<CXXBoolLiteralExpr>(E)->getValue() ? '1' : '0');
1615 Out << 'E';
1616 break;
1617
Anders Carlssone170ba72009-12-14 01:45:37 +00001618 case Expr::IntegerLiteralClass:
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001619 mangleIntegerLiteral(E->getType(),
Anders Carlssone170ba72009-12-14 01:45:37 +00001620 llvm::APSInt(cast<IntegerLiteral>(E)->getValue()));
1621 break;
1622
Anders Carlssond553f8c2009-09-21 01:21:10 +00001623 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001624}
1625
John McCallefe6aee2009-09-05 07:56:18 +00001626// FIXME: <type> ::= G <type> # imaginary (C 2000)
1627// FIXME: <type> ::= U <source-name> <type> # vendor extended type qualifier
1628
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001629void CXXNameMangler::mangleCXXCtorType(CXXCtorType T) {
1630 // <ctor-dtor-name> ::= C1 # complete object constructor
1631 // ::= C2 # base object constructor
1632 // ::= C3 # complete object allocating constructor
1633 //
1634 switch (T) {
1635 case Ctor_Complete:
1636 Out << "C1";
1637 break;
1638 case Ctor_Base:
1639 Out << "C2";
1640 break;
1641 case Ctor_CompleteAllocating:
1642 Out << "C3";
1643 break;
1644 }
1645}
1646
Anders Carlsson27ae5362009-04-17 01:58:57 +00001647void CXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
1648 // <ctor-dtor-name> ::= D0 # deleting destructor
1649 // ::= D1 # complete object destructor
1650 // ::= D2 # base object destructor
1651 //
1652 switch (T) {
1653 case Dtor_Deleting:
1654 Out << "D0";
1655 break;
1656 case Dtor_Complete:
1657 Out << "D1";
1658 break;
1659 case Dtor_Base:
1660 Out << "D2";
1661 break;
1662 }
1663}
1664
Douglas Gregor20f0cc72010-04-23 03:10:43 +00001665void CXXNameMangler::mangleTemplateArgs(TemplateName Template,
1666 const TemplateArgument *TemplateArgs,
1667 unsigned NumTemplateArgs) {
1668 if (TemplateDecl *TD = Template.getAsTemplateDecl())
1669 return mangleTemplateArgs(*TD->getTemplateParameters(), TemplateArgs,
1670 NumTemplateArgs);
Sean Huntc3021132010-05-05 15:23:54 +00001671
Douglas Gregor20f0cc72010-04-23 03:10:43 +00001672 // <template-args> ::= I <template-arg>+ E
1673 Out << 'I';
1674 for (unsigned i = 0; i != NumTemplateArgs; ++i)
1675 mangleTemplateArg(0, TemplateArgs[i]);
1676 Out << 'E';
1677}
1678
Rafael Espindolad9800722010-03-11 14:07:00 +00001679void CXXNameMangler::mangleTemplateArgs(const TemplateParameterList &PL,
1680 const TemplateArgumentList &AL) {
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001681 // <template-args> ::= I <template-arg>+ E
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001682 Out << 'I';
Rafael Espindolad9800722010-03-11 14:07:00 +00001683 for (unsigned i = 0, e = AL.size(); i != e; ++i)
1684 mangleTemplateArg(PL.getParam(i), AL[i]);
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001685 Out << 'E';
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001686}
1687
Rafael Espindolad9800722010-03-11 14:07:00 +00001688void CXXNameMangler::mangleTemplateArgs(const TemplateParameterList &PL,
1689 const TemplateArgument *TemplateArgs,
Anders Carlsson7624f212009-09-18 02:42:01 +00001690 unsigned NumTemplateArgs) {
1691 // <template-args> ::= I <template-arg>+ E
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001692 Out << 'I';
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001693 for (unsigned i = 0; i != NumTemplateArgs; ++i)
Rafael Espindolad9800722010-03-11 14:07:00 +00001694 mangleTemplateArg(PL.getParam(i), TemplateArgs[i]);
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001695 Out << 'E';
Anders Carlsson7624f212009-09-18 02:42:01 +00001696}
1697
Rafael Espindolad9800722010-03-11 14:07:00 +00001698void CXXNameMangler::mangleTemplateArg(const NamedDecl *P,
1699 const TemplateArgument &A) {
Mike Stump1eb44332009-09-09 15:08:12 +00001700 // <template-arg> ::= <type> # type or template
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001701 // ::= X <expression> E # expression
1702 // ::= <expr-primary> # simple expressions
1703 // ::= I <template-arg>* E # argument pack
1704 // ::= sp <expression> # pack expansion of (C++0x)
1705 switch (A.getKind()) {
1706 default:
1707 assert(0 && "Unknown template argument kind!");
1708 case TemplateArgument::Type:
1709 mangleType(A.getAsType());
1710 break;
Anders Carlsson9e85c742009-12-23 19:30:55 +00001711 case TemplateArgument::Template:
Douglas Gregor2d565b32010-02-06 01:09:36 +00001712 assert(A.getAsTemplate().getAsTemplateDecl() &&
Douglas Gregor20f0cc72010-04-23 03:10:43 +00001713 "Can't get dependent template names here");
Anders Carlsson9e85c742009-12-23 19:30:55 +00001714 mangleName(A.getAsTemplate().getAsTemplateDecl());
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001715 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001716 case TemplateArgument::Expression:
1717 Out << 'X';
1718 mangleExpression(A.getAsExpr());
1719 Out << 'E';
1720 break;
Anders Carlssone170ba72009-12-14 01:45:37 +00001721 case TemplateArgument::Integral:
1722 mangleIntegerLiteral(A.getIntegralType(), *A.getAsIntegral());
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001723 break;
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001724 case TemplateArgument::Declaration: {
Douglas Gregor20f0cc72010-04-23 03:10:43 +00001725 assert(P && "Missing template parameter for declaration argument");
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001726 // <expr-primary> ::= L <mangled-name> E # external name
1727
Rafael Espindolad9800722010-03-11 14:07:00 +00001728 // Clang produces AST's where pointer-to-member-function expressions
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001729 // and pointer-to-function expressions are represented as a declaration not
Rafael Espindolad9800722010-03-11 14:07:00 +00001730 // an expression. We compensate for it here to produce the correct mangling.
1731 NamedDecl *D = cast<NamedDecl>(A.getAsDecl());
1732 const NonTypeTemplateParmDecl *Parameter = cast<NonTypeTemplateParmDecl>(P);
1733 bool compensateMangling = D->isCXXClassMember() &&
1734 !Parameter->getType()->isReferenceType();
1735 if (compensateMangling) {
1736 Out << 'X';
1737 mangleOperatorName(OO_Amp, 1);
1738 }
1739
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001740 Out << 'L';
1741 // References to external entities use the mangled name; if the name would
1742 // not normally be manged then mangle it as unqualified.
1743 //
1744 // FIXME: The ABI specifies that external names here should have _Z, but
1745 // gcc leaves this off.
Rafael Espindolad9800722010-03-11 14:07:00 +00001746 if (compensateMangling)
1747 mangle(D, "_Z");
1748 else
1749 mangle(D, "Z");
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001750 Out << 'E';
Rafael Espindolad9800722010-03-11 14:07:00 +00001751
1752 if (compensateMangling)
1753 Out << 'E';
1754
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001755 break;
1756 }
1757 }
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001758}
1759
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001760void CXXNameMangler::mangleTemplateParameter(unsigned Index) {
1761 // <template-param> ::= T_ # first template parameter
1762 // ::= T <parameter-2 non-negative number> _
1763 if (Index == 0)
1764 Out << "T_";
1765 else
1766 Out << 'T' << (Index - 1) << '_';
1767}
1768
Anders Carlsson76967372009-09-17 00:43:46 +00001769// <substitution> ::= S <seq-id> _
1770// ::= S_
Anders Carlsson6862fc72009-09-17 04:16:28 +00001771bool CXXNameMangler::mangleSubstitution(const NamedDecl *ND) {
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001772 // Try one of the standard substitutions first.
1773 if (mangleStandardSubstitution(ND))
1774 return true;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001775
Anders Carlsson433d1372009-11-07 04:26:04 +00001776 ND = cast<NamedDecl>(ND->getCanonicalDecl());
Anders Carlsson6862fc72009-09-17 04:16:28 +00001777 return mangleSubstitution(reinterpret_cast<uintptr_t>(ND));
1778}
1779
Anders Carlsson76967372009-09-17 00:43:46 +00001780bool CXXNameMangler::mangleSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001781 if (!T.getCVRQualifiers()) {
1782 if (const RecordType *RT = T->getAs<RecordType>())
1783 return mangleSubstitution(RT->getDecl());
1784 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001785
Anders Carlsson76967372009-09-17 00:43:46 +00001786 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
1787
Anders Carlssond3a932a2009-09-17 03:53:28 +00001788 return mangleSubstitution(TypePtr);
1789}
1790
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001791bool CXXNameMangler::mangleSubstitution(TemplateName Template) {
1792 if (TemplateDecl *TD = Template.getAsTemplateDecl())
1793 return mangleSubstitution(TD);
Sean Huntc3021132010-05-05 15:23:54 +00001794
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001795 Template = Context.getASTContext().getCanonicalTemplateName(Template);
1796 return mangleSubstitution(
1797 reinterpret_cast<uintptr_t>(Template.getAsVoidPointer()));
1798}
1799
Anders Carlssond3a932a2009-09-17 03:53:28 +00001800bool CXXNameMangler::mangleSubstitution(uintptr_t Ptr) {
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001801 llvm::DenseMap<uintptr_t, unsigned>::iterator I = Substitutions.find(Ptr);
Anders Carlsson76967372009-09-17 00:43:46 +00001802 if (I == Substitutions.end())
1803 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001804
Anders Carlsson76967372009-09-17 00:43:46 +00001805 unsigned SeqID = I->second;
1806 if (SeqID == 0)
1807 Out << "S_";
1808 else {
1809 SeqID--;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001810
Anders Carlsson76967372009-09-17 00:43:46 +00001811 // <seq-id> is encoded in base-36, using digits and upper case letters.
1812 char Buffer[10];
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001813 char *BufferPtr = llvm::array_endof(Buffer);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001814
Anders Carlsson76967372009-09-17 00:43:46 +00001815 if (SeqID == 0) *--BufferPtr = '0';
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001816
Anders Carlsson76967372009-09-17 00:43:46 +00001817 while (SeqID) {
1818 assert(BufferPtr > Buffer && "Buffer overflow!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001819
Anders Carlsson76967372009-09-17 00:43:46 +00001820 unsigned char c = static_cast<unsigned char>(SeqID) % 36;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001821
Anders Carlsson76967372009-09-17 00:43:46 +00001822 *--BufferPtr = (c < 10 ? '0' + c : 'A' + c - 10);
1823 SeqID /= 36;
1824 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001825
Benjamin Kramer35f59b62010-04-10 16:03:31 +00001826 Out << 'S'
1827 << llvm::StringRef(BufferPtr, llvm::array_endof(Buffer)-BufferPtr)
1828 << '_';
Anders Carlsson76967372009-09-17 00:43:46 +00001829 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001830
Anders Carlsson76967372009-09-17 00:43:46 +00001831 return true;
1832}
1833
Anders Carlssonf514b542009-09-27 00:12:57 +00001834static bool isCharType(QualType T) {
1835 if (T.isNull())
1836 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001837
Anders Carlssonf514b542009-09-27 00:12:57 +00001838 return T->isSpecificBuiltinType(BuiltinType::Char_S) ||
1839 T->isSpecificBuiltinType(BuiltinType::Char_U);
1840}
1841
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001842/// isCharSpecialization - Returns whether a given type is a template
Anders Carlssonf514b542009-09-27 00:12:57 +00001843/// specialization of a given name with a single argument of type char.
1844static bool isCharSpecialization(QualType T, const char *Name) {
1845 if (T.isNull())
1846 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001847
Anders Carlssonf514b542009-09-27 00:12:57 +00001848 const RecordType *RT = T->getAs<RecordType>();
1849 if (!RT)
1850 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001851
1852 const ClassTemplateSpecializationDecl *SD =
Anders Carlssonf514b542009-09-27 00:12:57 +00001853 dyn_cast<ClassTemplateSpecializationDecl>(RT->getDecl());
1854 if (!SD)
1855 return false;
1856
1857 if (!isStdNamespace(SD->getDeclContext()))
1858 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001859
Anders Carlssonf514b542009-09-27 00:12:57 +00001860 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1861 if (TemplateArgs.size() != 1)
1862 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001863
Anders Carlssonf514b542009-09-27 00:12:57 +00001864 if (!isCharType(TemplateArgs[0].getAsType()))
1865 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001866
Daniel Dunbar01eb9b92009-10-18 21:17:35 +00001867 return SD->getIdentifier()->getName() == Name;
Anders Carlssonf514b542009-09-27 00:12:57 +00001868}
1869
Anders Carlsson91f88602009-12-07 19:56:42 +00001870template <std::size_t StrLen>
1871bool isStreamCharSpecialization(const ClassTemplateSpecializationDecl *SD,
1872 const char (&Str)[StrLen]) {
1873 if (!SD->getIdentifier()->isStr(Str))
1874 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001875
Anders Carlsson91f88602009-12-07 19:56:42 +00001876 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1877 if (TemplateArgs.size() != 2)
1878 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001879
Anders Carlsson91f88602009-12-07 19:56:42 +00001880 if (!isCharType(TemplateArgs[0].getAsType()))
1881 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001882
Anders Carlsson91f88602009-12-07 19:56:42 +00001883 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1884 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001885
Anders Carlsson91f88602009-12-07 19:56:42 +00001886 return true;
1887}
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001888
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001889bool CXXNameMangler::mangleStandardSubstitution(const NamedDecl *ND) {
1890 // <substitution> ::= St # ::std::
Anders Carlsson8c031552009-09-26 23:10:05 +00001891 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
Anders Carlsson47846d22009-12-04 06:23:23 +00001892 if (isStd(NS)) {
Anders Carlsson8c031552009-09-26 23:10:05 +00001893 Out << "St";
1894 return true;
1895 }
1896 }
1897
1898 if (const ClassTemplateDecl *TD = dyn_cast<ClassTemplateDecl>(ND)) {
1899 if (!isStdNamespace(TD->getDeclContext()))
1900 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001901
Anders Carlsson8c031552009-09-26 23:10:05 +00001902 // <substitution> ::= Sa # ::std::allocator
1903 if (TD->getIdentifier()->isStr("allocator")) {
1904 Out << "Sa";
1905 return true;
1906 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001907
Anders Carlsson189d59c2009-09-26 23:14:39 +00001908 // <<substitution> ::= Sb # ::std::basic_string
1909 if (TD->getIdentifier()->isStr("basic_string")) {
1910 Out << "Sb";
1911 return true;
1912 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001913 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001914
1915 if (const ClassTemplateSpecializationDecl *SD =
Anders Carlssonf514b542009-09-27 00:12:57 +00001916 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
Eli Friedman5370ee22010-02-23 18:25:09 +00001917 if (!isStdNamespace(SD->getDeclContext()))
1918 return false;
1919
Anders Carlssonf514b542009-09-27 00:12:57 +00001920 // <substitution> ::= Ss # ::std::basic_string<char,
1921 // ::std::char_traits<char>,
1922 // ::std::allocator<char> >
1923 if (SD->getIdentifier()->isStr("basic_string")) {
1924 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001925
Anders Carlssonf514b542009-09-27 00:12:57 +00001926 if (TemplateArgs.size() != 3)
1927 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001928
Anders Carlssonf514b542009-09-27 00:12:57 +00001929 if (!isCharType(TemplateArgs[0].getAsType()))
1930 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001931
Anders Carlssonf514b542009-09-27 00:12:57 +00001932 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1933 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001934
Anders Carlssonf514b542009-09-27 00:12:57 +00001935 if (!isCharSpecialization(TemplateArgs[2].getAsType(), "allocator"))
1936 return false;
1937
1938 Out << "Ss";
1939 return true;
1940 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001941
Anders Carlsson91f88602009-12-07 19:56:42 +00001942 // <substitution> ::= Si # ::std::basic_istream<char,
1943 // ::std::char_traits<char> >
1944 if (isStreamCharSpecialization(SD, "basic_istream")) {
1945 Out << "Si";
1946 return true;
1947 }
1948
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001949 // <substitution> ::= So # ::std::basic_ostream<char,
Anders Carlsson8f8fd8e2009-10-08 17:20:26 +00001950 // ::std::char_traits<char> >
Anders Carlsson91f88602009-12-07 19:56:42 +00001951 if (isStreamCharSpecialization(SD, "basic_ostream")) {
Anders Carlsson8f8fd8e2009-10-08 17:20:26 +00001952 Out << "So";
1953 return true;
1954 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001955
Anders Carlsson91f88602009-12-07 19:56:42 +00001956 // <substitution> ::= Sd # ::std::basic_iostream<char,
1957 // ::std::char_traits<char> >
1958 if (isStreamCharSpecialization(SD, "basic_iostream")) {
1959 Out << "Sd";
1960 return true;
1961 }
Anders Carlssonf514b542009-09-27 00:12:57 +00001962 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001963 return false;
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001964}
1965
Anders Carlsson76967372009-09-17 00:43:46 +00001966void CXXNameMangler::addSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001967 if (!T.getCVRQualifiers()) {
1968 if (const RecordType *RT = T->getAs<RecordType>()) {
1969 addSubstitution(RT->getDecl());
1970 return;
1971 }
1972 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001973
Anders Carlsson76967372009-09-17 00:43:46 +00001974 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
Anders Carlssond3a932a2009-09-17 03:53:28 +00001975 addSubstitution(TypePtr);
1976}
1977
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001978void CXXNameMangler::addSubstitution(TemplateName Template) {
1979 if (TemplateDecl *TD = Template.getAsTemplateDecl())
1980 return addSubstitution(TD);
Sean Huntc3021132010-05-05 15:23:54 +00001981
Douglas Gregor1e9268e2010-04-28 05:58:56 +00001982 Template = Context.getASTContext().getCanonicalTemplateName(Template);
1983 addSubstitution(reinterpret_cast<uintptr_t>(Template.getAsVoidPointer()));
1984}
1985
Anders Carlssond3a932a2009-09-17 03:53:28 +00001986void CXXNameMangler::addSubstitution(uintptr_t Ptr) {
Anders Carlsson76967372009-09-17 00:43:46 +00001987 unsigned SeqID = Substitutions.size();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001988
Anders Carlssond3a932a2009-09-17 03:53:28 +00001989 assert(!Substitutions.count(Ptr) && "Substitution already exists!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001990 Substitutions[Ptr] = SeqID;
Anders Carlsson76967372009-09-17 00:43:46 +00001991}
1992
Daniel Dunbar1b077112009-11-21 09:06:10 +00001993//
Mike Stump1eb44332009-09-09 15:08:12 +00001994
Daniel Dunbar1b077112009-11-21 09:06:10 +00001995/// \brief Mangles the name of the declaration D and emits that name to the
1996/// given output stream.
1997///
1998/// If the declaration D requires a mangled name, this routine will emit that
1999/// mangled name to \p os and return true. Otherwise, \p os will be unchanged
2000/// and this routine will return false. In this case, the caller should just
2001/// emit the identifier of the declaration (\c D->getIdentifier()) as its
2002/// name.
Daniel Dunbarf981bf82009-11-21 09:14:52 +00002003void MangleContext::mangleName(const NamedDecl *D,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002004 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc02ab4c2009-11-21 09:14:44 +00002005 assert((isa<FunctionDecl>(D) || isa<VarDecl>(D)) &&
2006 "Invalid mangleName() call, argument is not a variable or function!");
2007 assert(!isa<CXXConstructorDecl>(D) && !isa<CXXDestructorDecl>(D) &&
2008 "Invalid mangleName() call on 'structor decl!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00002009
Daniel Dunbar1b077112009-11-21 09:06:10 +00002010 PrettyStackTraceDecl CrashInfo(D, SourceLocation(),
2011 getASTContext().getSourceManager(),
2012 "Mangling declaration");
Mike Stump1eb44332009-09-09 15:08:12 +00002013
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002014 CXXNameMangler Mangler(*this, Res);
2015 return Mangler.mangle(D);
Daniel Dunbar1b077112009-11-21 09:06:10 +00002016}
Mike Stump1eb44332009-09-09 15:08:12 +00002017
Daniel Dunbar1b077112009-11-21 09:06:10 +00002018void MangleContext::mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002019 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbar77939c92009-11-21 09:06:31 +00002020 CXXNameMangler Mangler(*this, Res, D, Type);
2021 Mangler.mangle(D);
Daniel Dunbar1b077112009-11-21 09:06:10 +00002022}
Mike Stump1eb44332009-09-09 15:08:12 +00002023
Daniel Dunbar1b077112009-11-21 09:06:10 +00002024void MangleContext::mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002025 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbar77939c92009-11-21 09:06:31 +00002026 CXXNameMangler Mangler(*this, Res, D, Type);
2027 Mangler.mangle(D);
Daniel Dunbar1b077112009-11-21 09:06:10 +00002028}
Mike Stumpf1216772009-07-31 18:25:34 +00002029
Anders Carlsson19879c92010-03-23 17:17:29 +00002030void MangleContext::mangleThunk(const CXXMethodDecl *MD,
2031 const ThunkInfo &Thunk,
2032 llvm::SmallVectorImpl<char> &Res) {
2033 // <special-name> ::= T <call-offset> <base encoding>
2034 // # base is the nominal target function of thunk
2035 // <special-name> ::= Tc <call-offset> <call-offset> <base encoding>
2036 // # base is the nominal target function of thunk
2037 // # first call-offset is 'this' adjustment
2038 // # second call-offset is result adjustment
Sean Huntc3021132010-05-05 15:23:54 +00002039
Anders Carlsson19879c92010-03-23 17:17:29 +00002040 assert(!isa<CXXDestructorDecl>(MD) &&
2041 "Use mangleCXXDtor for destructor decls!");
Sean Huntc3021132010-05-05 15:23:54 +00002042
Anders Carlsson19879c92010-03-23 17:17:29 +00002043 CXXNameMangler Mangler(*this, Res);
2044 Mangler.getStream() << "_ZT";
2045 if (!Thunk.Return.isEmpty())
2046 Mangler.getStream() << 'c';
Sean Huntc3021132010-05-05 15:23:54 +00002047
Anders Carlsson19879c92010-03-23 17:17:29 +00002048 // Mangle the 'this' pointer adjustment.
2049 Mangler.mangleCallOffset(Thunk.This.NonVirtual, Thunk.This.VCallOffsetOffset);
Sean Huntc3021132010-05-05 15:23:54 +00002050
Anders Carlsson19879c92010-03-23 17:17:29 +00002051 // Mangle the return pointer adjustment if there is one.
2052 if (!Thunk.Return.isEmpty())
2053 Mangler.mangleCallOffset(Thunk.Return.NonVirtual,
2054 Thunk.Return.VBaseOffsetOffset);
Sean Huntc3021132010-05-05 15:23:54 +00002055
Anders Carlsson19879c92010-03-23 17:17:29 +00002056 Mangler.mangleFunctionEncoding(MD);
2057}
2058
Sean Huntc3021132010-05-05 15:23:54 +00002059void
Anders Carlsson19879c92010-03-23 17:17:29 +00002060MangleContext::mangleCXXDtorThunk(const CXXDestructorDecl *DD, CXXDtorType Type,
2061 const ThisAdjustment &ThisAdjustment,
2062 llvm::SmallVectorImpl<char> &Res) {
2063 // <special-name> ::= T <call-offset> <base encoding>
2064 // # base is the nominal target function of thunk
Sean Huntc3021132010-05-05 15:23:54 +00002065
Anders Carlsson19879c92010-03-23 17:17:29 +00002066 CXXNameMangler Mangler(*this, Res, DD, Type);
2067 Mangler.getStream() << "_ZT";
2068
2069 // Mangle the 'this' pointer adjustment.
Sean Huntc3021132010-05-05 15:23:54 +00002070 Mangler.mangleCallOffset(ThisAdjustment.NonVirtual,
Anders Carlsson19879c92010-03-23 17:17:29 +00002071 ThisAdjustment.VCallOffsetOffset);
2072
2073 Mangler.mangleFunctionEncoding(DD);
2074}
2075
Daniel Dunbarc0747712009-11-21 09:12:13 +00002076/// mangleGuardVariable - Returns the mangled name for a guard variable
2077/// for the passed in VarDecl.
2078void MangleContext::mangleGuardVariable(const VarDecl *D,
2079 llvm::SmallVectorImpl<char> &Res) {
2080 // <special-name> ::= GV <object name> # Guard variable for one-time
2081 // # initialization
2082 CXXNameMangler Mangler(*this, Res);
2083 Mangler.getStream() << "_ZGV";
2084 Mangler.mangleName(D);
2085}
2086
Anders Carlsson046c2942010-04-17 20:15:18 +00002087void MangleContext::mangleCXXVTable(const CXXRecordDecl *RD,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002088 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00002089 // <special-name> ::= TV <type> # virtual table
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002090 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00002091 Mangler.getStream() << "_ZTV";
2092 Mangler.mangleName(RD);
Daniel Dunbar1b077112009-11-21 09:06:10 +00002093}
Mike Stump82d75b02009-11-10 01:58:37 +00002094
Daniel Dunbar1b077112009-11-21 09:06:10 +00002095void MangleContext::mangleCXXVTT(const CXXRecordDecl *RD,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002096 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00002097 // <special-name> ::= TT <type> # VTT structure
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002098 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00002099 Mangler.getStream() << "_ZTT";
2100 Mangler.mangleName(RD);
Daniel Dunbar1b077112009-11-21 09:06:10 +00002101}
Mike Stumpab3f7e92009-11-10 01:41:59 +00002102
Anders Carlsson046c2942010-04-17 20:15:18 +00002103void MangleContext::mangleCXXCtorVTable(const CXXRecordDecl *RD, int64_t Offset,
Daniel Dunbar1b077112009-11-21 09:06:10 +00002104 const CXXRecordDecl *Type,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002105 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00002106 // <special-name> ::= TC <type> <offset number> _ <base type>
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002107 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00002108 Mangler.getStream() << "_ZTC";
2109 Mangler.mangleName(RD);
2110 Mangler.getStream() << Offset;
Benjamin Kramer35f59b62010-04-10 16:03:31 +00002111 Mangler.getStream() << '_';
Daniel Dunbarc0747712009-11-21 09:12:13 +00002112 Mangler.mangleName(Type);
Daniel Dunbar1b077112009-11-21 09:06:10 +00002113}
Mike Stump738f8c22009-07-31 23:15:31 +00002114
Mike Stumpde050572009-12-02 18:57:08 +00002115void MangleContext::mangleCXXRTTI(QualType Ty,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002116 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00002117 // <special-name> ::= TI <type> # typeinfo structure
Douglas Gregor154fe982009-12-23 22:04:40 +00002118 assert(!Ty.hasQualifiers() && "RTTI info cannot have top-level qualifiers");
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002119 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00002120 Mangler.getStream() << "_ZTI";
2121 Mangler.mangleType(Ty);
Daniel Dunbar1b077112009-11-21 09:06:10 +00002122}
Mike Stump67795982009-11-14 00:14:13 +00002123
Mike Stumpde050572009-12-02 18:57:08 +00002124void MangleContext::mangleCXXRTTIName(QualType Ty,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002125 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00002126 // <special-name> ::= TS <type> # typeinfo name (null terminated byte string)
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002127 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00002128 Mangler.getStream() << "_ZTS";
2129 Mangler.mangleType(Ty);
Mike Stumpf1216772009-07-31 18:25:34 +00002130}