blob: 37052a54622f7e36bfad0628439c5de9ac0248c1 [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 Carlssonb73a5be2009-11-26 02:49:32 +000028#include "CGVtable.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 Carlssonb73a5be2009-11-26 02:49:32 +0000105 void mangleCallOffset(const ThunkAdjustment &Adjustment);
Anders Carlsson19879c92010-03-23 17:17:29 +0000106 void mangleCallOffset(int64_t NonVirtual, int64_t Virtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000107 void mangleNumber(int64_t Number);
Daniel Dunbarc0747712009-11-21 09:12:13 +0000108 void mangleFunctionEncoding(const FunctionDecl *FD);
109 void mangleName(const NamedDecl *ND);
110 void mangleType(QualType T);
Mike Stump1eb44332009-09-09 15:08:12 +0000111
Daniel Dunbarc0747712009-11-21 09:12:13 +0000112private:
Daniel Dunbar1b077112009-11-21 09:06:10 +0000113 bool mangleSubstitution(const NamedDecl *ND);
114 bool mangleSubstitution(QualType T);
115 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);
125 void addSubstitution(uintptr_t Ptr);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000126
John McCall1dd73832010-02-04 01:42:13 +0000127 void mangleUnresolvedScope(NestedNameSpecifier *Qualifier);
128 void mangleUnresolvedName(NestedNameSpecifier *Qualifier,
129 DeclarationName Name,
130 unsigned KnownArity = UnknownArity);
131
Daniel Dunbar1b077112009-11-21 09:06:10 +0000132 void mangleName(const TemplateDecl *TD,
133 const TemplateArgument *TemplateArgs,
134 unsigned NumTemplateArgs);
John McCall1dd73832010-02-04 01:42:13 +0000135 void mangleUnqualifiedName(const NamedDecl *ND) {
136 mangleUnqualifiedName(ND, ND->getDeclName(), UnknownArity);
137 }
138 void mangleUnqualifiedName(const NamedDecl *ND, DeclarationName Name,
139 unsigned KnownArity);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000140 void mangleUnscopedName(const NamedDecl *ND);
141 void mangleUnscopedTemplateName(const TemplateDecl *ND);
142 void mangleSourceName(const IdentifierInfo *II);
143 void mangleLocalName(const NamedDecl *ND);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000144 void mangleNestedName(const NamedDecl *ND, const DeclContext *DC,
145 bool NoFunction=false);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000146 void mangleNestedName(const TemplateDecl *TD,
147 const TemplateArgument *TemplateArgs,
148 unsigned NumTemplateArgs);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000149 void manglePrefix(const DeclContext *DC, bool NoFunction=false);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000150 void mangleTemplatePrefix(const TemplateDecl *ND);
151 void mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity);
152 void mangleQualifiers(Qualifiers Quals);
John McCallefe6aee2009-09-05 07:56:18 +0000153
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000154 void mangleObjCMethodName(const ObjCMethodDecl *MD);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000155
Daniel Dunbar1b077112009-11-21 09:06:10 +0000156 // Declare manglers for every type class.
John McCallefe6aee2009-09-05 07:56:18 +0000157#define ABSTRACT_TYPE(CLASS, PARENT)
158#define NON_CANONICAL_TYPE(CLASS, PARENT)
159#define TYPE(CLASS, PARENT) void mangleType(const CLASS##Type *T);
160#include "clang/AST/TypeNodes.def"
161
Daniel Dunbar1b077112009-11-21 09:06:10 +0000162 void mangleType(const TagType*);
163 void mangleBareFunctionType(const FunctionType *T,
164 bool MangleReturnType);
Anders Carlssone170ba72009-12-14 01:45:37 +0000165
166 void mangleIntegerLiteral(QualType T, const llvm::APSInt &Value);
John McCall2f27bf82010-02-04 02:56:29 +0000167 void mangleMemberExpr(const Expr *Base, bool IsArrow,
168 NestedNameSpecifier *Qualifier,
169 DeclarationName Name,
170 unsigned KnownArity);
John McCall1dd73832010-02-04 01:42:13 +0000171 void mangleCalledExpression(const Expr *E, unsigned KnownArity);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000172 void mangleExpression(const Expr *E);
173 void mangleCXXCtorType(CXXCtorType T);
174 void mangleCXXDtorType(CXXDtorType T);
Mike Stump1eb44332009-09-09 15:08:12 +0000175
Rafael Espindolad9800722010-03-11 14:07:00 +0000176 void mangleTemplateArgs(const TemplateParameterList &PL,
177 const TemplateArgument *TemplateArgs,
Daniel Dunbar1b077112009-11-21 09:06:10 +0000178 unsigned NumTemplateArgs);
Rafael Espindolad9800722010-03-11 14:07:00 +0000179 void mangleTemplateArgs(const TemplateParameterList &PL,
180 const TemplateArgumentList &AL);
181 void mangleTemplateArg(const NamedDecl *P, const TemplateArgument &A);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000182
Daniel Dunbar1b077112009-11-21 09:06:10 +0000183 void mangleTemplateParameter(unsigned Index);
184};
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000185}
186
Anders Carlsson43f17402009-04-02 15:51:53 +0000187static bool isInCLinkageSpecification(const Decl *D) {
Douglas Gregor457e2812009-10-28 16:31:34 +0000188 D = D->getCanonicalDecl();
Mike Stump1eb44332009-09-09 15:08:12 +0000189 for (const DeclContext *DC = D->getDeclContext();
Anders Carlsson43f17402009-04-02 15:51:53 +0000190 !DC->isTranslationUnit(); DC = DC->getParent()) {
Mike Stump1eb44332009-09-09 15:08:12 +0000191 if (const LinkageSpecDecl *Linkage = dyn_cast<LinkageSpecDecl>(DC))
Anders Carlsson43f17402009-04-02 15:51:53 +0000192 return Linkage->getLanguage() == LinkageSpecDecl::lang_c;
193 }
Mike Stump1eb44332009-09-09 15:08:12 +0000194
Anders Carlsson43f17402009-04-02 15:51:53 +0000195 return false;
196}
197
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000198bool MangleContext::shouldMangleDeclName(const NamedDecl *D) {
199 // In C, functions with no attributes never need to be mangled. Fastpath them.
200 if (!getASTContext().getLangOptions().CPlusPlus && !D->hasAttrs())
201 return false;
202
203 // Any decl can be declared with __asm("foo") on it, and this takes precedence
204 // over all other naming in the .o file.
205 if (D->hasAttr<AsmLabelAttr>())
206 return true;
207
Mike Stump141c5af2009-09-02 00:25:38 +0000208 // Clang's "overloadable" attribute extension to C/C++ implies name mangling
Anders Carlssona1e16222009-11-07 07:15:03 +0000209 // (always) as does passing a C++ member function and a function
210 // whose name is not a simple identifier.
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000211 const FunctionDecl *FD = dyn_cast<FunctionDecl>(D);
212 if (FD && (FD->hasAttr<OverloadableAttr>() || isa<CXXMethodDecl>(FD) ||
213 !FD->getDeclName().isIdentifier()))
214 return true;
Mike Stump1eb44332009-09-09 15:08:12 +0000215
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000216 // Otherwise, no mangling is done outside C++ mode.
217 if (!getASTContext().getLangOptions().CPlusPlus)
218 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000219
Sean Hunt31455252010-01-24 03:04:27 +0000220 // Variables at global scope with non-internal linkage are not mangled
Eli Friedman7facf842009-12-02 20:32:49 +0000221 if (!FD) {
222 const DeclContext *DC = D->getDeclContext();
223 // Check for extern variable declared locally.
224 if (isa<FunctionDecl>(DC) && D->hasLinkage())
225 while (!DC->isNamespace() && !DC->isTranslationUnit())
226 DC = DC->getParent();
Douglas Gregor0b6bc8b2010-02-03 09:33:45 +0000227 if (DC->isTranslationUnit() && D->getLinkage() != InternalLinkage)
Eli Friedman7facf842009-12-02 20:32:49 +0000228 return false;
229 }
230
231 // C functions and "main" are not mangled.
232 if ((FD && FD->isMain()) || isInCLinkageSpecification(D))
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000233 return false;
234
Anders Carlsson43f17402009-04-02 15:51:53 +0000235 return true;
236}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000237
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000238void CXXNameMangler::mangle(const NamedDecl *D, llvm::StringRef Prefix) {
Mike Stump141c5af2009-09-02 00:25:38 +0000239 // Any decl can be declared with __asm("foo") on it, and this takes precedence
240 // over all other naming in the .o file.
Argyrios Kyrtzidis40b598e2009-06-30 02:34:44 +0000241 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000242 // If we have an asm name, then we use it as the mangling.
243 Out << '\01'; // LLVM IR Marker for __asm("foo")
244 Out << ALA->getLabel();
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000245 return;
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000246 }
Mike Stump1eb44332009-09-09 15:08:12 +0000247
Sean Hunt31455252010-01-24 03:04:27 +0000248 // <mangled-name> ::= _Z <encoding>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000249 // ::= <data name>
250 // ::= <special-name>
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000251 Out << Prefix;
252 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000253 mangleFunctionEncoding(FD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000254 else if (const VarDecl *VD = dyn_cast<VarDecl>(D))
255 mangleName(VD);
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000256 else
Rafael Espindolad9800722010-03-11 14:07:00 +0000257 mangleName(cast<FieldDecl>(D));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000258}
259
260void CXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD) {
261 // <encoding> ::= <function name> <bare-function-type>
262 mangleName(FD);
Mike Stump1eb44332009-09-09 15:08:12 +0000263
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000264 // Don't mangle in the type if this isn't a decl we should typically mangle.
265 if (!Context.shouldMangleDeclName(FD))
266 return;
267
Mike Stump141c5af2009-09-02 00:25:38 +0000268 // Whether the mangling of a function type includes the return type depends on
269 // the context and the nature of the function. The rules for deciding whether
270 // the return type is included are:
Mike Stump1eb44332009-09-09 15:08:12 +0000271 //
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000272 // 1. Template functions (names or types) have return types encoded, with
273 // the exceptions listed below.
Mike Stump1eb44332009-09-09 15:08:12 +0000274 // 2. Function types not appearing as part of a function name mangling,
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000275 // e.g. parameters, pointer types, etc., have return type encoded, with the
276 // exceptions listed below.
277 // 3. Non-template function names do not have return types encoded.
278 //
Mike Stump141c5af2009-09-02 00:25:38 +0000279 // The exceptions mentioned in (1) and (2) above, for which the return type is
280 // never included, are
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000281 // 1. Constructors.
282 // 2. Destructors.
283 // 3. Conversion operator functions, e.g. operator int.
284 bool MangleReturnType = false;
Anders Carlsson9234b7f2009-09-17 03:46:43 +0000285 if (FunctionTemplateDecl *PrimaryTemplate = FD->getPrimaryTemplate()) {
286 if (!(isa<CXXConstructorDecl>(FD) || isa<CXXDestructorDecl>(FD) ||
287 isa<CXXConversionDecl>(FD)))
288 MangleReturnType = true;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000289
Anders Carlsson9234b7f2009-09-17 03:46:43 +0000290 // Mangle the type of the primary template.
291 FD = PrimaryTemplate->getTemplatedDecl();
292 }
293
John McCall54e14c42009-10-22 22:37:11 +0000294 // Do the canonicalization out here because parameter types can
295 // undergo additional canonicalization (e.g. array decay).
296 FunctionType *FT = cast<FunctionType>(Context.getASTContext()
297 .getCanonicalType(FD->getType()));
298
299 mangleBareFunctionType(FT, MangleReturnType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000300}
301
Anders Carlsson47846d22009-12-04 06:23:23 +0000302/// isStd - Return whether a given namespace is the 'std' namespace.
303static bool isStd(const NamespaceDecl *NS) {
John McCall9aeed322009-10-01 00:25:31 +0000304 const IdentifierInfo *II = NS->getOriginalNamespace()->getIdentifier();
305 return II && II->isStr("std");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000306}
307
Anders Carlsson47846d22009-12-04 06:23:23 +0000308static const DeclContext *IgnoreLinkageSpecDecls(const DeclContext *DC) {
309 while (isa<LinkageSpecDecl>(DC)) {
310 assert(cast<LinkageSpecDecl>(DC)->getLanguage() ==
311 LinkageSpecDecl::lang_cxx && "Unexpected linkage decl!");
312 DC = DC->getParent();
313 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000314
Anders Carlsson47846d22009-12-04 06:23:23 +0000315 return DC;
316}
317
318// isStdNamespace - Return whether a given decl context is a toplevel 'std'
319// namespace.
Daniel Dunbar1308af92009-11-21 09:11:45 +0000320static bool isStdNamespace(const DeclContext *DC) {
Anders Carlsson47846d22009-12-04 06:23:23 +0000321 if (!DC->isNamespace())
322 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000323
Anders Carlsson47846d22009-12-04 06:23:23 +0000324 if (!IgnoreLinkageSpecDecls(DC->getParent())->isTranslationUnit())
325 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000326
Anders Carlsson47846d22009-12-04 06:23:23 +0000327 return isStd(cast<NamespaceDecl>(DC));
Daniel Dunbar1308af92009-11-21 09:11:45 +0000328}
329
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000330static const TemplateDecl *
331isTemplate(const NamedDecl *ND, const TemplateArgumentList *&TemplateArgs) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000332 // Check if we have a function template.
333 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)){
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000334 if (const TemplateDecl *TD = FD->getPrimaryTemplate()) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000335 TemplateArgs = FD->getTemplateSpecializationArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000336 return TD;
Anders Carlsson2744a062009-09-18 19:00:18 +0000337 }
338 }
339
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000340 // Check if we have a class template.
341 if (const ClassTemplateSpecializationDecl *Spec =
342 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
343 TemplateArgs = &Spec->getTemplateArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000344 return Spec->getSpecializedTemplate();
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000345 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000346
Anders Carlsson2744a062009-09-18 19:00:18 +0000347 return 0;
348}
349
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000350void CXXNameMangler::mangleName(const NamedDecl *ND) {
351 // <name> ::= <nested-name>
352 // ::= <unscoped-name>
353 // ::= <unscoped-template-name> <template-args>
Anders Carlsson201ce742009-09-17 03:17:01 +0000354 // ::= <local-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000355 //
Anders Carlssond58d6f72009-09-17 16:12:20 +0000356 const DeclContext *DC = ND->getDeclContext();
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000357
Fariborz Jahanian57058532010-03-03 19:41:08 +0000358 if (GetLocalClassFunctionDeclContext(DC)) {
359 mangleLocalName(ND);
360 return;
361 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000362
Eli Friedman7facf842009-12-02 20:32:49 +0000363 // If this is an extern variable declared locally, the relevant DeclContext
364 // is that of the containing namespace, or the translation unit.
365 if (isa<FunctionDecl>(DC) && ND->hasLinkage())
366 while (!DC->isNamespace() && !DC->isTranslationUnit())
367 DC = DC->getParent();
368
Anders Carlsson5cc58c62009-09-22 17:23:30 +0000369 while (isa<LinkageSpecDecl>(DC))
Anders Carlssond58d6f72009-09-17 16:12:20 +0000370 DC = DC->getParent();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000371
Anders Carlssond58d6f72009-09-17 16:12:20 +0000372 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000373 // Check if we have a template.
374 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000375 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000376 mangleUnscopedTemplateName(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000377 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
378 mangleTemplateArgs(*TemplateParameters, *TemplateArgs);
Anders Carlsson2744a062009-09-18 19:00:18 +0000379 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000380 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000381
Anders Carlsson7482e242009-09-18 04:29:09 +0000382 mangleUnscopedName(ND);
383 return;
384 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000385
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000386 if (isa<FunctionDecl>(DC) || isa<ObjCMethodDecl>(DC)) {
Anders Carlsson7482e242009-09-18 04:29:09 +0000387 mangleLocalName(ND);
388 return;
389 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000390
Eli Friedman7facf842009-12-02 20:32:49 +0000391 mangleNestedName(ND, DC);
Anders Carlsson7482e242009-09-18 04:29:09 +0000392}
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000393void CXXNameMangler::mangleName(const TemplateDecl *TD,
Anders Carlsson7624f212009-09-18 02:42:01 +0000394 const TemplateArgument *TemplateArgs,
395 unsigned NumTemplateArgs) {
Anders Carlsson47846d22009-12-04 06:23:23 +0000396 const DeclContext *DC = IgnoreLinkageSpecDecls(TD->getDeclContext());
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000397
Anders Carlsson7624f212009-09-18 02:42:01 +0000398 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000399 mangleUnscopedTemplateName(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000400 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
401 mangleTemplateArgs(*TemplateParameters, TemplateArgs, NumTemplateArgs);
Anders Carlsson7624f212009-09-18 02:42:01 +0000402 } else {
403 mangleNestedName(TD, TemplateArgs, NumTemplateArgs);
404 }
405}
406
Anders Carlsson201ce742009-09-17 03:17:01 +0000407void CXXNameMangler::mangleUnscopedName(const NamedDecl *ND) {
408 // <unscoped-name> ::= <unqualified-name>
409 // ::= St <unqualified-name> # ::std::
410 if (isStdNamespace(ND->getDeclContext()))
411 Out << "St";
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000412
Anders Carlsson201ce742009-09-17 03:17:01 +0000413 mangleUnqualifiedName(ND);
414}
415
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000416void CXXNameMangler::mangleUnscopedTemplateName(const TemplateDecl *ND) {
Anders Carlsson201ce742009-09-17 03:17:01 +0000417 // <unscoped-template-name> ::= <unscoped-name>
418 // ::= <substitution>
Anders Carlsson7624f212009-09-18 02:42:01 +0000419 if (mangleSubstitution(ND))
Anders Carlsson03c9d532009-09-17 04:02:31 +0000420 return;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000421
Douglas Gregor32fb4e12010-02-05 20:45:00 +0000422 // <template-template-param> ::= <template-param>
423 if (const TemplateTemplateParmDecl *TTP
424 = dyn_cast<TemplateTemplateParmDecl>(ND)) {
425 mangleTemplateParameter(TTP->getIndex());
426 return;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000427 }
Douglas Gregor32fb4e12010-02-05 20:45:00 +0000428
Anders Carlsson1668f202009-09-26 20:13:56 +0000429 mangleUnscopedName(ND->getTemplatedDecl());
Anders Carlsson7624f212009-09-18 02:42:01 +0000430 addSubstitution(ND);
Anders Carlsson201ce742009-09-17 03:17:01 +0000431}
432
Anders Carlssona94822e2009-11-26 02:32:05 +0000433void CXXNameMangler::mangleNumber(int64_t Number) {
434 // <number> ::= [n] <non-negative decimal integer>
435 if (Number < 0) {
436 Out << 'n';
437 Number = -Number;
438 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000439
Anders Carlssona94822e2009-11-26 02:32:05 +0000440 Out << Number;
441}
442
Anders Carlssonb73a5be2009-11-26 02:49:32 +0000443void CXXNameMangler::mangleCallOffset(const ThunkAdjustment &Adjustment) {
Anders Carlsson19879c92010-03-23 17:17:29 +0000444 mangleCallOffset(Adjustment.NonVirtual, Adjustment.Virtual);
445}
446
447void CXXNameMangler::mangleCallOffset(int64_t NonVirtual, int64_t Virtual) {
Mike Stump141c5af2009-09-02 00:25:38 +0000448 // <call-offset> ::= h <nv-offset> _
449 // ::= v <v-offset> _
450 // <nv-offset> ::= <offset number> # non-virtual base override
Anders Carlssona94822e2009-11-26 02:32:05 +0000451 // <v-offset> ::= <offset number> _ <virtual offset number>
Mike Stump141c5af2009-09-02 00:25:38 +0000452 // # virtual base override, with vcall offset
Anders Carlsson19879c92010-03-23 17:17:29 +0000453 if (!Virtual) {
Anders Carlssona94822e2009-11-26 02:32:05 +0000454 Out << 'h';
Anders Carlsson19879c92010-03-23 17:17:29 +0000455 mangleNumber(NonVirtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000456 Out << '_';
457 return;
Mike Stump141c5af2009-09-02 00:25:38 +0000458 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000459
Anders Carlssona94822e2009-11-26 02:32:05 +0000460 Out << 'v';
Anders Carlsson19879c92010-03-23 17:17:29 +0000461 mangleNumber(NonVirtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000462 Out << '_';
Anders Carlsson19879c92010-03-23 17:17:29 +0000463 mangleNumber(Virtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000464 Out << '_';
Mike Stump9124bcc2009-09-02 00:56:18 +0000465}
466
John McCall1dd73832010-02-04 01:42:13 +0000467void CXXNameMangler::mangleUnresolvedScope(NestedNameSpecifier *Qualifier) {
468 Qualifier = getASTContext().getCanonicalNestedNameSpecifier(Qualifier);
469 switch (Qualifier->getKind()) {
470 case NestedNameSpecifier::Global:
471 // nothing
472 break;
473 case NestedNameSpecifier::Namespace:
474 mangleName(Qualifier->getAsNamespace());
475 break;
476 case NestedNameSpecifier::TypeSpec:
Rafael Espindola9b35b252010-03-17 04:28:11 +0000477 case NestedNameSpecifier::TypeSpecWithTemplate: {
478 const Type *QTy = Qualifier->getAsType();
479
480 if (const TemplateSpecializationType *TST =
481 dyn_cast<TemplateSpecializationType>(QTy)) {
482 if (!mangleSubstitution(QualType(TST, 0))) {
483 TemplateDecl *TD = TST->getTemplateName().getAsTemplateDecl();
484 assert(TD && "FIXME: Support dependent template names");
485 mangleTemplatePrefix(TD);
486 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
487 mangleTemplateArgs(*TemplateParameters, TST->getArgs(),
488 TST->getNumArgs());
489 addSubstitution(QualType(TST, 0));
490 }
491 } else {
492 // We use the QualType mangle type variant here because it handles
493 // substitutions.
494 mangleType(QualType(QTy, 0));
495 }
496 }
John McCall1dd73832010-02-04 01:42:13 +0000497 break;
498 case NestedNameSpecifier::Identifier:
John McCallad5e7382010-03-01 23:49:17 +0000499 // Member expressions can have these without prefixes.
500 if (Qualifier->getPrefix())
501 mangleUnresolvedScope(Qualifier->getPrefix());
John McCall1dd73832010-02-04 01:42:13 +0000502 mangleSourceName(Qualifier->getAsIdentifier());
503 break;
504 }
505}
506
507/// Mangles a name which was not resolved to a specific entity.
508void CXXNameMangler::mangleUnresolvedName(NestedNameSpecifier *Qualifier,
509 DeclarationName Name,
510 unsigned KnownArity) {
511 if (Qualifier)
512 mangleUnresolvedScope(Qualifier);
513 // FIXME: ambiguity of unqualified lookup with ::
514
515 mangleUnqualifiedName(0, Name, KnownArity);
516}
517
518void CXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND,
519 DeclarationName Name,
520 unsigned KnownArity) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000521 // <unqualified-name> ::= <operator-name>
Mike Stump1eb44332009-09-09 15:08:12 +0000522 // ::= <ctor-dtor-name>
523 // ::= <source-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000524 switch (Name.getNameKind()) {
Anders Carlssonc4355b62009-10-07 01:45:02 +0000525 case DeclarationName::Identifier: {
Anders Carlssonc4355b62009-10-07 01:45:02 +0000526 if (const IdentifierInfo *II = Name.getAsIdentifierInfo()) {
Sean Hunt31455252010-01-24 03:04:27 +0000527 // We must avoid conflicts between internally- and externally-
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000528 // linked variable declaration names in the same TU.
Anders Carlssonaec25232010-02-06 04:52:27 +0000529 // This naming convention is the same as that followed by GCC, though it
530 // shouldn't actually matter.
531 if (ND && isa<VarDecl>(ND) && ND->getLinkage() == InternalLinkage &&
Sean Hunt31455252010-01-24 03:04:27 +0000532 ND->getDeclContext()->isFileContext())
533 Out << 'L';
534
Anders Carlssonc4355b62009-10-07 01:45:02 +0000535 mangleSourceName(II);
536 break;
537 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000538
John McCall1dd73832010-02-04 01:42:13 +0000539 // Otherwise, an anonymous entity. We must have a declaration.
540 assert(ND && "mangling empty name without declaration");
541
542 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
543 if (NS->isAnonymousNamespace()) {
544 // This is how gcc mangles these names.
545 Out << "12_GLOBAL__N_1";
546 break;
547 }
548 }
549
Anders Carlssonc4355b62009-10-07 01:45:02 +0000550 // We must have an anonymous struct.
551 const TagDecl *TD = cast<TagDecl>(ND);
552 if (const TypedefDecl *D = TD->getTypedefForAnonDecl()) {
553 assert(TD->getDeclContext() == D->getDeclContext() &&
554 "Typedef should not be in another decl context!");
555 assert(D->getDeclName().getAsIdentifierInfo() &&
556 "Typedef was not named!");
557 mangleSourceName(D->getDeclName().getAsIdentifierInfo());
558 break;
559 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000560
Anders Carlssonc4355b62009-10-07 01:45:02 +0000561 // Get a unique id for the anonymous struct.
562 uint64_t AnonStructId = Context.getAnonymousStructId(TD);
563
564 // Mangle it as a source name in the form
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000565 // [n] $_<id>
Anders Carlssonc4355b62009-10-07 01:45:02 +0000566 // where n is the length of the string.
567 llvm::SmallString<8> Str;
568 Str += "$_";
569 Str += llvm::utostr(AnonStructId);
570
571 Out << Str.size();
572 Out << Str.str();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000573 break;
Anders Carlssonc4355b62009-10-07 01:45:02 +0000574 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000575
576 case DeclarationName::ObjCZeroArgSelector:
577 case DeclarationName::ObjCOneArgSelector:
578 case DeclarationName::ObjCMultiArgSelector:
579 assert(false && "Can't mangle Objective-C selector names here!");
580 break;
581
582 case DeclarationName::CXXConstructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000583 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000584 // If the named decl is the C++ constructor we're mangling, use the type
585 // we were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000586 mangleCXXCtorType(static_cast<CXXCtorType>(StructorType));
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000587 else
588 // Otherwise, use the complete constructor name. This is relevant if a
589 // class with a constructor is declared within a constructor.
590 mangleCXXCtorType(Ctor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000591 break;
592
593 case DeclarationName::CXXDestructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000594 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000595 // If the named decl is the C++ destructor we're mangling, use the type we
596 // were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000597 mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
598 else
599 // Otherwise, use the complete destructor name. This is relevant if a
600 // class with a destructor is declared within a destructor.
601 mangleCXXDtorType(Dtor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000602 break;
603
604 case DeclarationName::CXXConversionFunctionName:
Mike Stump1eb44332009-09-09 15:08:12 +0000605 // <operator-name> ::= cv <type> # (cast)
Douglas Gregor219cc612009-02-13 01:28:03 +0000606 Out << "cv";
Anders Carlssonb5404912009-10-07 01:06:45 +0000607 mangleType(Context.getASTContext().getCanonicalType(Name.getCXXNameType()));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000608 break;
609
Anders Carlsson8257d412009-12-22 06:36:32 +0000610 case DeclarationName::CXXOperatorName: {
John McCall1dd73832010-02-04 01:42:13 +0000611 unsigned Arity;
612 if (ND) {
613 Arity = cast<FunctionDecl>(ND)->getNumParams();
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000614
John McCall1dd73832010-02-04 01:42:13 +0000615 // If we have a C++ member function, we need to include the 'this' pointer.
616 // FIXME: This does not make sense for operators that are static, but their
617 // names stay the same regardless of the arity (operator new for instance).
618 if (isa<CXXMethodDecl>(ND))
619 Arity++;
620 } else
621 Arity = KnownArity;
622
Anders Carlsson8257d412009-12-22 06:36:32 +0000623 mangleOperatorName(Name.getCXXOverloadedOperator(), Arity);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000624 break;
Anders Carlsson8257d412009-12-22 06:36:32 +0000625 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000626
Sean Hunt3e518bd2009-11-29 07:34:05 +0000627 case DeclarationName::CXXLiteralOperatorName:
Sean Hunt5dd6b392009-12-04 21:11:13 +0000628 // FIXME: This mangling is not yet official.
Sean Hunt2421f662009-12-04 21:01:37 +0000629 Out << "li";
Sean Hunt3e518bd2009-11-29 07:34:05 +0000630 mangleSourceName(Name.getCXXLiteralIdentifier());
631 break;
632
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000633 case DeclarationName::CXXUsingDirective:
634 assert(false && "Can't mangle a using directive name!");
Douglas Gregor219cc612009-02-13 01:28:03 +0000635 break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000636 }
637}
638
639void CXXNameMangler::mangleSourceName(const IdentifierInfo *II) {
640 // <source-name> ::= <positive length number> <identifier>
641 // <number> ::= [n] <non-negative decimal integer>
642 // <identifier> ::= <unqualified source code identifier>
643 Out << II->getLength() << II->getName();
644}
645
Eli Friedman7facf842009-12-02 20:32:49 +0000646void CXXNameMangler::mangleNestedName(const NamedDecl *ND,
Fariborz Jahanian57058532010-03-03 19:41:08 +0000647 const DeclContext *DC,
648 bool NoFunction) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000649 // <nested-name> ::= N [<CV-qualifiers>] <prefix> <unqualified-name> E
650 // ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
Anders Carlssond99edc42009-09-26 03:55:37 +0000651
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000652 Out << 'N';
653 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(ND))
John McCall0953e762009-09-24 19:53:00 +0000654 mangleQualifiers(Qualifiers::fromCVRMask(Method->getTypeQualifiers()));
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000655
Anders Carlsson2744a062009-09-18 19:00:18 +0000656 // Check if we have a template.
657 const TemplateArgumentList *TemplateArgs = 0;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000658 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000659 mangleTemplatePrefix(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000660 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
661 mangleTemplateArgs(*TemplateParameters, *TemplateArgs);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000662 }
663 else {
664 manglePrefix(DC, NoFunction);
Anders Carlsson7482e242009-09-18 04:29:09 +0000665 mangleUnqualifiedName(ND);
666 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000667
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000668 Out << 'E';
669}
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000670void CXXNameMangler::mangleNestedName(const TemplateDecl *TD,
Anders Carlsson7624f212009-09-18 02:42:01 +0000671 const TemplateArgument *TemplateArgs,
672 unsigned NumTemplateArgs) {
Anders Carlssone45117b2009-09-27 19:53:49 +0000673 // <nested-name> ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
674
Anders Carlsson7624f212009-09-18 02:42:01 +0000675 Out << 'N';
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000676
Anders Carlssone45117b2009-09-27 19:53:49 +0000677 mangleTemplatePrefix(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000678 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
679 mangleTemplateArgs(*TemplateParameters, TemplateArgs, NumTemplateArgs);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000680
Anders Carlsson7624f212009-09-18 02:42:01 +0000681 Out << 'E';
682}
683
Anders Carlsson1b42c792009-04-02 16:24:45 +0000684void CXXNameMangler::mangleLocalName(const NamedDecl *ND) {
685 // <local-name> := Z <function encoding> E <entity name> [<discriminator>]
686 // := Z <function encoding> E s [<discriminator>]
Mike Stump1eb44332009-09-09 15:08:12 +0000687 // <discriminator> := _ <non-negative number>
Fariborz Jahanian57058532010-03-03 19:41:08 +0000688 const DeclContext *DC = ND->getDeclContext();
Anders Carlsson1b42c792009-04-02 16:24:45 +0000689 Out << 'Z';
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000690
Fariborz Jahanian57058532010-03-03 19:41:08 +0000691 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(DC))
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000692 mangleObjCMethodName(MD);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000693 else if (const DeclContext *CDC = GetLocalClassFunctionDeclContext(DC)) {
694 mangleFunctionEncoding(cast<FunctionDecl>(CDC));
695 Out << 'E';
696 mangleNestedName(ND, DC, true /*NoFunction*/);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000697
Fariborz Jahanian4819ac42010-03-04 01:02:03 +0000698 // FIXME. This still does not cover all cases.
699 unsigned disc;
700 if (Context.getNextDiscriminator(ND, disc)) {
701 if (disc < 10)
702 Out << '_' << disc;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000703 else
Fariborz Jahanian4819ac42010-03-04 01:02:03 +0000704 Out << "__" << disc << '_';
705 }
Fariborz Jahanian57058532010-03-03 19:41:08 +0000706
707 return;
708 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000709 else
Fariborz Jahanian57058532010-03-03 19:41:08 +0000710 mangleFunctionEncoding(cast<FunctionDecl>(DC));
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000711
Anders Carlsson1b42c792009-04-02 16:24:45 +0000712 Out << 'E';
Eli Friedman6f9f25d2009-12-11 20:21:38 +0000713 mangleUnqualifiedName(ND);
Anders Carlsson1b42c792009-04-02 16:24:45 +0000714}
715
Fariborz Jahanian57058532010-03-03 19:41:08 +0000716void CXXNameMangler::manglePrefix(const DeclContext *DC, bool NoFunction) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000717 // <prefix> ::= <prefix> <unqualified-name>
718 // ::= <template-prefix> <template-args>
719 // ::= <template-param>
720 // ::= # empty
721 // ::= <substitution>
Anders Carlsson6862fc72009-09-17 04:16:28 +0000722
Anders Carlssonadd28822009-09-22 20:33:31 +0000723 while (isa<LinkageSpecDecl>(DC))
724 DC = DC->getParent();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000725
Anders Carlsson9263e912009-09-18 18:39:58 +0000726 if (DC->isTranslationUnit())
727 return;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000728
Anders Carlsson6862fc72009-09-17 04:16:28 +0000729 if (mangleSubstitution(cast<NamedDecl>(DC)))
730 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000731
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000732 // Check if we have a template.
733 const TemplateArgumentList *TemplateArgs = 0;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000734 if (const TemplateDecl *TD = isTemplate(cast<NamedDecl>(DC), TemplateArgs)) {
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000735 mangleTemplatePrefix(TD);
Rafael Espindolad9800722010-03-11 14:07:00 +0000736 TemplateParameterList *TemplateParameters = TD->getTemplateParameters();
737 mangleTemplateArgs(*TemplateParameters, *TemplateArgs);
Fariborz Jahanian57058532010-03-03 19:41:08 +0000738 }
739 else if(NoFunction && isa<FunctionDecl>(DC))
740 return;
741 else {
742 manglePrefix(DC->getParent(), NoFunction);
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000743 mangleUnqualifiedName(cast<NamedDecl>(DC));
744 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000745
Anders Carlsson6862fc72009-09-17 04:16:28 +0000746 addSubstitution(cast<NamedDecl>(DC));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000747}
748
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000749void CXXNameMangler::mangleTemplatePrefix(const TemplateDecl *ND) {
Anders Carlsson7482e242009-09-18 04:29:09 +0000750 // <template-prefix> ::= <prefix> <template unqualified-name>
751 // ::= <template-param>
752 // ::= <substitution>
Douglas Gregor32fb4e12010-02-05 20:45:00 +0000753 // <template-template-param> ::= <template-param>
754 // <substitution>
Anders Carlsson7482e242009-09-18 04:29:09 +0000755
Anders Carlssonaeb85372009-09-26 22:18:22 +0000756 if (mangleSubstitution(ND))
757 return;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000758
Douglas Gregor32fb4e12010-02-05 20:45:00 +0000759 // <template-template-param> ::= <template-param>
760 if (const TemplateTemplateParmDecl *TTP
761 = dyn_cast<TemplateTemplateParmDecl>(ND)) {
762 mangleTemplateParameter(TTP->getIndex());
763 return;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000764 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000765
Anders Carlssonaa73ab12009-09-18 18:47:07 +0000766 manglePrefix(ND->getDeclContext());
Anders Carlsson1668f202009-09-26 20:13:56 +0000767 mangleUnqualifiedName(ND->getTemplatedDecl());
Anders Carlssonaeb85372009-09-26 22:18:22 +0000768 addSubstitution(ND);
Anders Carlsson7482e242009-09-18 04:29:09 +0000769}
770
Mike Stump1eb44332009-09-09 15:08:12 +0000771void
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000772CXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity) {
773 switch (OO) {
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000774 // <operator-name> ::= nw # new
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000775 case OO_New: Out << "nw"; break;
776 // ::= na # new[]
777 case OO_Array_New: Out << "na"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000778 // ::= dl # delete
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000779 case OO_Delete: Out << "dl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000780 // ::= da # delete[]
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000781 case OO_Array_Delete: Out << "da"; break;
782 // ::= ps # + (unary)
783 // ::= pl # +
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000784 case OO_Plus:
Anders Carlsson8257d412009-12-22 06:36:32 +0000785 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
786 Out << (Arity == 1? "ps" : "pl"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000787 // ::= ng # - (unary)
788 // ::= mi # -
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000789 case OO_Minus:
Anders Carlsson8257d412009-12-22 06:36:32 +0000790 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
791 Out << (Arity == 1? "ng" : "mi"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000792 // ::= ad # & (unary)
793 // ::= an # &
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000794 case OO_Amp:
Anders Carlsson8257d412009-12-22 06:36:32 +0000795 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
796 Out << (Arity == 1? "ad" : "an"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000797 // ::= de # * (unary)
798 // ::= ml # *
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000799 case OO_Star:
Anders Carlsson8257d412009-12-22 06:36:32 +0000800 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
801 Out << (Arity == 1? "de" : "ml"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000802 // ::= co # ~
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000803 case OO_Tilde: Out << "co"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000804 // ::= dv # /
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000805 case OO_Slash: Out << "dv"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000806 // ::= rm # %
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000807 case OO_Percent: Out << "rm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000808 // ::= or # |
809 case OO_Pipe: Out << "or"; break;
810 // ::= eo # ^
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000811 case OO_Caret: Out << "eo"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000812 // ::= aS # =
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000813 case OO_Equal: Out << "aS"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000814 // ::= pL # +=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000815 case OO_PlusEqual: Out << "pL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000816 // ::= mI # -=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000817 case OO_MinusEqual: Out << "mI"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000818 // ::= mL # *=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000819 case OO_StarEqual: Out << "mL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000820 // ::= dV # /=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000821 case OO_SlashEqual: Out << "dV"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000822 // ::= rM # %=
823 case OO_PercentEqual: Out << "rM"; break;
824 // ::= aN # &=
825 case OO_AmpEqual: Out << "aN"; break;
826 // ::= oR # |=
827 case OO_PipeEqual: Out << "oR"; break;
828 // ::= eO # ^=
829 case OO_CaretEqual: Out << "eO"; break;
830 // ::= ls # <<
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000831 case OO_LessLess: Out << "ls"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000832 // ::= rs # >>
833 case OO_GreaterGreater: Out << "rs"; break;
834 // ::= lS # <<=
835 case OO_LessLessEqual: Out << "lS"; break;
836 // ::= rS # >>=
837 case OO_GreaterGreaterEqual: Out << "rS"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000838 // ::= eq # ==
839 case OO_EqualEqual: Out << "eq"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000840 // ::= ne # !=
841 case OO_ExclaimEqual: Out << "ne"; break;
842 // ::= lt # <
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000843 case OO_Less: Out << "lt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000844 // ::= gt # >
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000845 case OO_Greater: Out << "gt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000846 // ::= le # <=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000847 case OO_LessEqual: Out << "le"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000848 // ::= ge # >=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000849 case OO_GreaterEqual: Out << "ge"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000850 // ::= nt # !
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000851 case OO_Exclaim: Out << "nt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000852 // ::= aa # &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000853 case OO_AmpAmp: Out << "aa"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000854 // ::= oo # ||
855 case OO_PipePipe: Out << "oo"; break;
856 // ::= pp # ++
857 case OO_PlusPlus: Out << "pp"; break;
858 // ::= mm # --
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000859 case OO_MinusMinus: Out << "mm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000860 // ::= cm # ,
861 case OO_Comma: Out << "cm"; break;
862 // ::= pm # ->*
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000863 case OO_ArrowStar: Out << "pm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000864 // ::= pt # ->
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000865 case OO_Arrow: Out << "pt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000866 // ::= cl # ()
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000867 case OO_Call: Out << "cl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000868 // ::= ix # []
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000869 case OO_Subscript: Out << "ix"; break;
Anders Carlssone170ba72009-12-14 01:45:37 +0000870
871 // ::= qu # ?
872 // The conditional operator can't be overloaded, but we still handle it when
873 // mangling expressions.
874 case OO_Conditional: Out << "qu"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000875
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000876 case OO_None:
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000877 case NUM_OVERLOADED_OPERATORS:
Mike Stump1eb44332009-09-09 15:08:12 +0000878 assert(false && "Not an overloaded operator");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000879 break;
880 }
881}
882
John McCall0953e762009-09-24 19:53:00 +0000883void CXXNameMangler::mangleQualifiers(Qualifiers Quals) {
Mike Stump1eb44332009-09-09 15:08:12 +0000884 // <CV-qualifiers> ::= [r] [V] [K] # restrict (C99), volatile, const
John McCall0953e762009-09-24 19:53:00 +0000885 if (Quals.hasRestrict())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000886 Out << 'r';
John McCall0953e762009-09-24 19:53:00 +0000887 if (Quals.hasVolatile())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000888 Out << 'V';
John McCall0953e762009-09-24 19:53:00 +0000889 if (Quals.hasConst())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000890 Out << 'K';
John McCall0953e762009-09-24 19:53:00 +0000891
892 // FIXME: For now, just drop all extension qualifiers on the floor.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000893}
894
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000895void CXXNameMangler::mangleObjCMethodName(const ObjCMethodDecl *MD) {
896 llvm::SmallString<64> Name;
897 llvm::raw_svector_ostream OS(Name);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000898
899 const ObjCContainerDecl *CD =
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000900 dyn_cast<ObjCContainerDecl>(MD->getDeclContext());
901 assert (CD && "Missing container decl in GetNameForMethod");
902 OS << (MD->isInstanceMethod() ? '-' : '+') << '[' << CD->getName();
903 if (const ObjCCategoryImplDecl *CID = dyn_cast<ObjCCategoryImplDecl>(CD))
904 OS << '(' << CID->getNameAsString() << ')';
905 OS << ' ' << MD->getSelector().getAsString() << ']';
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +0000906
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000907 Out << OS.str().size() << OS.str();
908}
909
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000910void CXXNameMangler::mangleType(QualType T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000911 // Only operate on the canonical type!
Anders Carlssonb5404912009-10-07 01:06:45 +0000912 T = Context.getASTContext().getCanonicalType(T);
Anders Carlsson4843e582009-03-10 17:07:44 +0000913
Douglas Gregora4923eb2009-11-16 21:35:15 +0000914 bool IsSubstitutable = T.hasLocalQualifiers() || !isa<BuiltinType>(T);
Anders Carlsson76967372009-09-17 00:43:46 +0000915 if (IsSubstitutable && mangleSubstitution(T))
916 return;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000917
Douglas Gregora4923eb2009-11-16 21:35:15 +0000918 if (Qualifiers Quals = T.getLocalQualifiers()) {
John McCall0953e762009-09-24 19:53:00 +0000919 mangleQualifiers(Quals);
920 // Recurse: even if the qualified type isn't yet substitutable,
921 // the unqualified type might be.
Douglas Gregora4923eb2009-11-16 21:35:15 +0000922 mangleType(T.getLocalUnqualifiedType());
Anders Carlsson76967372009-09-17 00:43:46 +0000923 } else {
924 switch (T->getTypeClass()) {
John McCallefe6aee2009-09-05 07:56:18 +0000925#define ABSTRACT_TYPE(CLASS, PARENT)
926#define NON_CANONICAL_TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000927 case Type::CLASS: \
Jeffrey Yasskin9f61aa92009-12-12 05:05:38 +0000928 llvm_unreachable("can't mangle non-canonical type " #CLASS "Type"); \
Anders Carlsson76967372009-09-17 00:43:46 +0000929 return;
John McCallefe6aee2009-09-05 07:56:18 +0000930#define TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000931 case Type::CLASS: \
John McCall0953e762009-09-24 19:53:00 +0000932 mangleType(static_cast<const CLASS##Type*>(T.getTypePtr())); \
Anders Carlsson76967372009-09-17 00:43:46 +0000933 break;
John McCallefe6aee2009-09-05 07:56:18 +0000934#include "clang/AST/TypeNodes.def"
Anders Carlsson76967372009-09-17 00:43:46 +0000935 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000936 }
Anders Carlsson76967372009-09-17 00:43:46 +0000937
938 // Add the substitution.
939 if (IsSubstitutable)
940 addSubstitution(T);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000941}
942
943void CXXNameMangler::mangleType(const BuiltinType *T) {
John McCallefe6aee2009-09-05 07:56:18 +0000944 // <type> ::= <builtin-type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000945 // <builtin-type> ::= v # void
946 // ::= w # wchar_t
947 // ::= b # bool
948 // ::= c # char
949 // ::= a # signed char
950 // ::= h # unsigned char
951 // ::= s # short
952 // ::= t # unsigned short
953 // ::= i # int
954 // ::= j # unsigned int
955 // ::= l # long
956 // ::= m # unsigned long
957 // ::= x # long long, __int64
958 // ::= y # unsigned long long, __int64
959 // ::= n # __int128
960 // UNSUPPORTED: ::= o # unsigned __int128
961 // ::= f # float
962 // ::= d # double
963 // ::= e # long double, __float80
964 // UNSUPPORTED: ::= g # __float128
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000965 // UNSUPPORTED: ::= Dd # IEEE 754r decimal floating point (64 bits)
966 // UNSUPPORTED: ::= De # IEEE 754r decimal floating point (128 bits)
967 // UNSUPPORTED: ::= Df # IEEE 754r decimal floating point (32 bits)
968 // UNSUPPORTED: ::= Dh # IEEE 754r half-precision floating point (16 bits)
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000969 // ::= Di # char32_t
970 // ::= Ds # char16_t
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000971 // ::= u <source-name> # vendor extended type
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000972 // From our point of view, std::nullptr_t is a builtin, but as far as mangling
973 // is concerned, it's a type called std::nullptr_t.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000974 switch (T->getKind()) {
975 case BuiltinType::Void: Out << 'v'; break;
976 case BuiltinType::Bool: Out << 'b'; break;
977 case BuiltinType::Char_U: case BuiltinType::Char_S: Out << 'c'; break;
978 case BuiltinType::UChar: Out << 'h'; break;
979 case BuiltinType::UShort: Out << 't'; break;
980 case BuiltinType::UInt: Out << 'j'; break;
981 case BuiltinType::ULong: Out << 'm'; break;
982 case BuiltinType::ULongLong: Out << 'y'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000983 case BuiltinType::UInt128: Out << 'o'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000984 case BuiltinType::SChar: Out << 'a'; break;
985 case BuiltinType::WChar: Out << 'w'; break;
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000986 case BuiltinType::Char16: Out << "Ds"; break;
987 case BuiltinType::Char32: Out << "Di"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000988 case BuiltinType::Short: Out << 's'; break;
989 case BuiltinType::Int: Out << 'i'; break;
990 case BuiltinType::Long: Out << 'l'; break;
991 case BuiltinType::LongLong: Out << 'x'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000992 case BuiltinType::Int128: Out << 'n'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000993 case BuiltinType::Float: Out << 'f'; break;
994 case BuiltinType::Double: Out << 'd'; break;
995 case BuiltinType::LongDouble: Out << 'e'; break;
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000996 case BuiltinType::NullPtr: Out << "St9nullptr_t"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000997
998 case BuiltinType::Overload:
999 case BuiltinType::Dependent:
Mike Stump1eb44332009-09-09 15:08:12 +00001000 assert(false &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001001 "Overloaded and dependent types shouldn't get to name mangling");
1002 break;
Anders Carlssone89d1592009-06-26 18:41:36 +00001003 case BuiltinType::UndeducedAuto:
1004 assert(0 && "Should not see undeduced auto here");
1005 break;
Steve Naroff9533a7f2009-07-22 17:14:51 +00001006 case BuiltinType::ObjCId: Out << "11objc_object"; break;
1007 case BuiltinType::ObjCClass: Out << "10objc_class"; break;
Fariborz Jahanian13dcd002009-11-21 19:53:08 +00001008 case BuiltinType::ObjCSel: Out << "13objc_selector"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001009 }
1010}
1011
John McCallefe6aee2009-09-05 07:56:18 +00001012// <type> ::= <function-type>
1013// <function-type> ::= F [Y] <bare-function-type> E
1014void CXXNameMangler::mangleType(const FunctionProtoType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001015 Out << 'F';
Mike Stumpf5408fe2009-05-16 07:57:57 +00001016 // FIXME: We don't have enough information in the AST to produce the 'Y'
1017 // encoding for extern "C" function types.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001018 mangleBareFunctionType(T, /*MangleReturnType=*/true);
1019 Out << 'E';
1020}
John McCallefe6aee2009-09-05 07:56:18 +00001021void CXXNameMangler::mangleType(const FunctionNoProtoType *T) {
Jeffrey Yasskin9f61aa92009-12-12 05:05:38 +00001022 llvm_unreachable("Can't mangle K&R function prototypes");
John McCallefe6aee2009-09-05 07:56:18 +00001023}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001024void CXXNameMangler::mangleBareFunctionType(const FunctionType *T,
1025 bool MangleReturnType) {
John McCallefe6aee2009-09-05 07:56:18 +00001026 // We should never be mangling something without a prototype.
1027 const FunctionProtoType *Proto = cast<FunctionProtoType>(T);
1028
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001029 // <bare-function-type> ::= <signature type>+
1030 if (MangleReturnType)
John McCallefe6aee2009-09-05 07:56:18 +00001031 mangleType(Proto->getResultType());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001032
Anders Carlssonc6c91bc2009-04-01 00:15:23 +00001033 if (Proto->getNumArgs() == 0) {
1034 Out << 'v';
1035 return;
1036 }
Mike Stump1eb44332009-09-09 15:08:12 +00001037
Douglas Gregor72564e72009-02-26 23:50:07 +00001038 for (FunctionProtoType::arg_type_iterator Arg = Proto->arg_type_begin(),
Mike Stump1eb44332009-09-09 15:08:12 +00001039 ArgEnd = Proto->arg_type_end();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001040 Arg != ArgEnd; ++Arg)
1041 mangleType(*Arg);
Douglas Gregor219cc612009-02-13 01:28:03 +00001042
1043 // <builtin-type> ::= z # ellipsis
1044 if (Proto->isVariadic())
1045 Out << 'z';
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001046}
1047
John McCallefe6aee2009-09-05 07:56:18 +00001048// <type> ::= <class-enum-type>
Mike Stump1eb44332009-09-09 15:08:12 +00001049// <class-enum-type> ::= <name>
John McCalled976492009-12-04 22:46:56 +00001050void CXXNameMangler::mangleType(const UnresolvedUsingType *T) {
1051 mangleName(T->getDecl());
1052}
1053
1054// <type> ::= <class-enum-type>
1055// <class-enum-type> ::= <name>
John McCallefe6aee2009-09-05 07:56:18 +00001056void CXXNameMangler::mangleType(const EnumType *T) {
1057 mangleType(static_cast<const TagType*>(T));
1058}
1059void CXXNameMangler::mangleType(const RecordType *T) {
1060 mangleType(static_cast<const TagType*>(T));
1061}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001062void CXXNameMangler::mangleType(const TagType *T) {
Eli Friedmanecb7e932009-12-11 18:00:57 +00001063 mangleName(T->getDecl());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001064}
1065
John McCallefe6aee2009-09-05 07:56:18 +00001066// <type> ::= <array-type>
1067// <array-type> ::= A <positive dimension number> _ <element type>
1068// ::= A [<dimension expression>] _ <element type>
1069void CXXNameMangler::mangleType(const ConstantArrayType *T) {
1070 Out << 'A' << T->getSize() << '_';
1071 mangleType(T->getElementType());
1072}
1073void CXXNameMangler::mangleType(const VariableArrayType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001074 Out << 'A';
John McCallefe6aee2009-09-05 07:56:18 +00001075 mangleExpression(T->getSizeExpr());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001076 Out << '_';
1077 mangleType(T->getElementType());
1078}
John McCallefe6aee2009-09-05 07:56:18 +00001079void CXXNameMangler::mangleType(const DependentSizedArrayType *T) {
1080 Out << 'A';
1081 mangleExpression(T->getSizeExpr());
1082 Out << '_';
1083 mangleType(T->getElementType());
1084}
1085void CXXNameMangler::mangleType(const IncompleteArrayType *T) {
1086 Out << 'A' << '_';
1087 mangleType(T->getElementType());
1088}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001089
John McCallefe6aee2009-09-05 07:56:18 +00001090// <type> ::= <pointer-to-member-type>
1091// <pointer-to-member-type> ::= M <class type> <member type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001092void CXXNameMangler::mangleType(const MemberPointerType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001093 Out << 'M';
1094 mangleType(QualType(T->getClass(), 0));
Anders Carlsson0e650012009-05-17 17:41:20 +00001095 QualType PointeeType = T->getPointeeType();
1096 if (const FunctionProtoType *FPT = dyn_cast<FunctionProtoType>(PointeeType)) {
John McCall0953e762009-09-24 19:53:00 +00001097 mangleQualifiers(Qualifiers::fromCVRMask(FPT->getTypeQuals()));
Anders Carlsson0e650012009-05-17 17:41:20 +00001098 mangleType(FPT);
Mike Stump1eb44332009-09-09 15:08:12 +00001099 } else
Anders Carlsson0e650012009-05-17 17:41:20 +00001100 mangleType(PointeeType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001101}
1102
John McCallefe6aee2009-09-05 07:56:18 +00001103// <type> ::= <template-param>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001104void CXXNameMangler::mangleType(const TemplateTypeParmType *T) {
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001105 mangleTemplateParameter(T->getIndex());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001106}
1107
John McCallefe6aee2009-09-05 07:56:18 +00001108// FIXME: <type> ::= <template-template-param> <template-args>
John McCallefe6aee2009-09-05 07:56:18 +00001109
1110// <type> ::= P <type> # pointer-to
1111void CXXNameMangler::mangleType(const PointerType *T) {
1112 Out << 'P';
1113 mangleType(T->getPointeeType());
1114}
1115void CXXNameMangler::mangleType(const ObjCObjectPointerType *T) {
1116 Out << 'P';
1117 mangleType(T->getPointeeType());
1118}
1119
1120// <type> ::= R <type> # reference-to
1121void CXXNameMangler::mangleType(const LValueReferenceType *T) {
1122 Out << 'R';
1123 mangleType(T->getPointeeType());
1124}
1125
1126// <type> ::= O <type> # rvalue reference-to (C++0x)
1127void CXXNameMangler::mangleType(const RValueReferenceType *T) {
1128 Out << 'O';
1129 mangleType(T->getPointeeType());
1130}
1131
1132// <type> ::= C <type> # complex pair (C 2000)
1133void CXXNameMangler::mangleType(const ComplexType *T) {
1134 Out << 'C';
1135 mangleType(T->getElementType());
1136}
1137
1138// GNU extension: vector types
Nick Lewycky0e5f0672010-03-26 07:18:04 +00001139// <type> ::= <vector-type>
1140// <vector-type> ::= Dv <positive dimension number> _ <element type>
1141// ::= Dv [<dimension expression>] _ <element type>
John McCallefe6aee2009-09-05 07:56:18 +00001142void CXXNameMangler::mangleType(const VectorType *T) {
Nick Lewycky0e5f0672010-03-26 07:18:04 +00001143 Out << "Dv" << T->getNumElements() << '_';
John McCallefe6aee2009-09-05 07:56:18 +00001144 mangleType(T->getElementType());
1145}
1146void CXXNameMangler::mangleType(const ExtVectorType *T) {
1147 mangleType(static_cast<const VectorType*>(T));
1148}
1149void CXXNameMangler::mangleType(const DependentSizedExtVectorType *T) {
Nick Lewycky0e5f0672010-03-26 07:18:04 +00001150 Out << "Dv";
1151 mangleExpression(T->getSizeExpr());
1152 Out << '_';
John McCallefe6aee2009-09-05 07:56:18 +00001153 mangleType(T->getElementType());
1154}
1155
Anders Carlssona40c5e42009-03-07 22:03:21 +00001156void CXXNameMangler::mangleType(const ObjCInterfaceType *T) {
1157 mangleSourceName(T->getDecl()->getIdentifier());
1158}
1159
John McCallefe6aee2009-09-05 07:56:18 +00001160void CXXNameMangler::mangleType(const BlockPointerType *T) {
Anders Carlssonf28c6872009-12-23 22:31:44 +00001161 Out << "U13block_pointer";
1162 mangleType(T->getPointeeType());
John McCallefe6aee2009-09-05 07:56:18 +00001163}
1164
John McCallefe6aee2009-09-05 07:56:18 +00001165void CXXNameMangler::mangleType(const TemplateSpecializationType *T) {
Anders Carlsson7624f212009-09-18 02:42:01 +00001166 TemplateDecl *TD = T->getTemplateName().getAsTemplateDecl();
1167 assert(TD && "FIXME: Support dependent template names!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001168
Anders Carlsson7624f212009-09-18 02:42:01 +00001169 mangleName(TD, T->getArgs(), T->getNumArgs());
John McCallefe6aee2009-09-05 07:56:18 +00001170}
1171
1172void CXXNameMangler::mangleType(const TypenameType *T) {
Anders Carlssonae352482009-09-26 02:26:02 +00001173 // Typename types are always nested
1174 Out << 'N';
Rafael Espindola9b35b252010-03-17 04:28:11 +00001175 mangleUnresolvedScope(T->getQualifier());
Anders Carlssonae352482009-09-26 02:26:02 +00001176 mangleSourceName(T->getIdentifier());
Anders Carlssonae352482009-09-26 02:26:02 +00001177 Out << 'E';
John McCallefe6aee2009-09-05 07:56:18 +00001178}
1179
John McCallad5e7382010-03-01 23:49:17 +00001180void CXXNameMangler::mangleType(const TypeOfType *T) {
1181 // FIXME: this is pretty unsatisfactory, but there isn't an obvious
1182 // "extension with parameters" mangling.
1183 Out << "u6typeof";
1184}
1185
1186void CXXNameMangler::mangleType(const TypeOfExprType *T) {
1187 // FIXME: this is pretty unsatisfactory, but there isn't an obvious
1188 // "extension with parameters" mangling.
1189 Out << "u6typeof";
1190}
1191
1192void CXXNameMangler::mangleType(const DecltypeType *T) {
1193 Expr *E = T->getUnderlyingExpr();
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001194
John McCallad5e7382010-03-01 23:49:17 +00001195 // type ::= Dt <expression> E # decltype of an id-expression
1196 // # or class member access
1197 // ::= DT <expression> E # decltype of an expression
1198
1199 // This purports to be an exhaustive list of id-expressions and
1200 // class member accesses. Note that we do not ignore parentheses;
1201 // parentheses change the semantics of decltype for these
1202 // expressions (and cause the mangler to use the other form).
1203 if (isa<DeclRefExpr>(E) ||
1204 isa<MemberExpr>(E) ||
1205 isa<UnresolvedLookupExpr>(E) ||
1206 isa<DependentScopeDeclRefExpr>(E) ||
1207 isa<CXXDependentScopeMemberExpr>(E) ||
1208 isa<UnresolvedMemberExpr>(E))
1209 Out << "Dt";
1210 else
1211 Out << "DT";
1212 mangleExpression(E);
1213 Out << 'E';
1214}
1215
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001216void CXXNameMangler::mangleIntegerLiteral(QualType T,
Anders Carlssone170ba72009-12-14 01:45:37 +00001217 const llvm::APSInt &Value) {
1218 // <expr-primary> ::= L <type> <value number> E # integer literal
1219 Out << 'L';
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001220
Anders Carlssone170ba72009-12-14 01:45:37 +00001221 mangleType(T);
1222 if (T->isBooleanType()) {
1223 // Boolean values are encoded as 0/1.
1224 Out << (Value.getBoolValue() ? '1' : '0');
1225 } else {
1226 if (Value.isNegative())
1227 Out << 'n';
1228 Value.abs().print(Out, false);
1229 }
1230 Out << 'E';
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001231
Anders Carlssone170ba72009-12-14 01:45:37 +00001232}
1233
John McCall1dd73832010-02-04 01:42:13 +00001234void CXXNameMangler::mangleCalledExpression(const Expr *E, unsigned Arity) {
1235 if (E->getType() != getASTContext().OverloadTy)
1236 mangleExpression(E);
John McCall2f27bf82010-02-04 02:56:29 +00001237 // propagate arity to dependent overloads?
John McCall1dd73832010-02-04 01:42:13 +00001238
1239 llvm::PointerIntPair<OverloadExpr*,1> R
1240 = OverloadExpr::find(const_cast<Expr*>(E));
1241 if (R.getInt())
1242 Out << "an"; // &
1243 const OverloadExpr *Ovl = R.getPointer();
John McCall2f27bf82010-02-04 02:56:29 +00001244 if (const UnresolvedMemberExpr *ME = dyn_cast<UnresolvedMemberExpr>(Ovl)) {
1245 mangleMemberExpr(ME->getBase(), ME->isArrow(), ME->getQualifier(),
1246 ME->getMemberName(), Arity);
1247 return;
1248 }
John McCall1dd73832010-02-04 01:42:13 +00001249
1250 mangleUnresolvedName(Ovl->getQualifier(), Ovl->getName(), Arity);
1251}
1252
John McCall2f27bf82010-02-04 02:56:29 +00001253/// Mangles a member expression. Implicit accesses are not handled,
1254/// but that should be okay, because you shouldn't be able to
1255/// make an implicit access in a function template declaration.
John McCall2f27bf82010-02-04 02:56:29 +00001256void CXXNameMangler::mangleMemberExpr(const Expr *Base,
1257 bool IsArrow,
1258 NestedNameSpecifier *Qualifier,
1259 DeclarationName Member,
1260 unsigned Arity) {
John McCalle1e342f2010-03-01 19:12:25 +00001261 // gcc-4.4 uses 'dt' for dot expressions, which is reasonable.
1262 // OTOH, gcc also mangles the name as an expression.
1263 Out << (IsArrow ? "pt" : "dt");
John McCall2f27bf82010-02-04 02:56:29 +00001264 mangleExpression(Base);
1265 mangleUnresolvedName(Qualifier, Member, Arity);
1266}
1267
Anders Carlssond553f8c2009-09-21 01:21:10 +00001268void CXXNameMangler::mangleExpression(const Expr *E) {
1269 // <expression> ::= <unary operator-name> <expression>
John McCall09cc1412010-02-03 00:55:45 +00001270 // ::= <binary operator-name> <expression> <expression>
1271 // ::= <trinary operator-name> <expression> <expression> <expression>
1272 // ::= cl <expression>* E # call
Anders Carlssond553f8c2009-09-21 01:21:10 +00001273 // ::= cv <type> expression # conversion with one argument
1274 // ::= cv <type> _ <expression>* E # conversion with a different number of arguments
John McCall09cc1412010-02-03 00:55:45 +00001275 // ::= st <type> # sizeof (a type)
Anders Carlssond553f8c2009-09-21 01:21:10 +00001276 // ::= at <type> # alignof (a type)
1277 // ::= <template-param>
1278 // ::= <function-param>
1279 // ::= sr <type> <unqualified-name> # dependent name
1280 // ::= sr <type> <unqualified-name> <template-args> # dependent template-id
1281 // ::= sZ <template-param> # size of a parameter pack
John McCall09cc1412010-02-03 00:55:45 +00001282 // ::= <expr-primary>
John McCall1dd73832010-02-04 01:42:13 +00001283 // <expr-primary> ::= L <type> <value number> E # integer literal
1284 // ::= L <type <value float> E # floating literal
1285 // ::= L <mangled-name> E # external name
Anders Carlssond553f8c2009-09-21 01:21:10 +00001286 switch (E->getStmtClass()) {
John McCall09cc1412010-02-03 00:55:45 +00001287 default:
1288 llvm_unreachable("unexpected statement kind");
1289 break;
1290
John McCall1dd73832010-02-04 01:42:13 +00001291 case Expr::CallExprClass: {
1292 const CallExpr *CE = cast<CallExpr>(E);
1293 Out << "cl";
1294 mangleCalledExpression(CE->getCallee(), CE->getNumArgs());
1295 for (unsigned I = 0, N = CE->getNumArgs(); I != N; ++I)
1296 mangleExpression(CE->getArg(I));
1297 Out << "E";
John McCall09cc1412010-02-03 00:55:45 +00001298 break;
John McCall1dd73832010-02-04 01:42:13 +00001299 }
John McCall09cc1412010-02-03 00:55:45 +00001300
John McCall2f27bf82010-02-04 02:56:29 +00001301 case Expr::MemberExprClass: {
1302 const MemberExpr *ME = cast<MemberExpr>(E);
1303 mangleMemberExpr(ME->getBase(), ME->isArrow(),
1304 ME->getQualifier(), ME->getMemberDecl()->getDeclName(),
1305 UnknownArity);
1306 break;
1307 }
1308
1309 case Expr::UnresolvedMemberExprClass: {
1310 const UnresolvedMemberExpr *ME = cast<UnresolvedMemberExpr>(E);
1311 mangleMemberExpr(ME->getBase(), ME->isArrow(),
1312 ME->getQualifier(), ME->getMemberName(),
1313 UnknownArity);
1314 break;
1315 }
1316
1317 case Expr::CXXDependentScopeMemberExprClass: {
1318 const CXXDependentScopeMemberExpr *ME
1319 = cast<CXXDependentScopeMemberExpr>(E);
1320 mangleMemberExpr(ME->getBase(), ME->isArrow(),
1321 ME->getQualifier(), ME->getMember(),
1322 UnknownArity);
1323 break;
1324 }
1325
John McCall1dd73832010-02-04 01:42:13 +00001326 case Expr::UnresolvedLookupExprClass: {
John McCalla3218e72010-02-04 01:48:38 +00001327 // The ABI doesn't cover how to mangle overload sets, so we mangle
1328 // using something as close as possible to the original lookup
1329 // expression.
John McCall1dd73832010-02-04 01:42:13 +00001330 const UnresolvedLookupExpr *ULE = cast<UnresolvedLookupExpr>(E);
1331 mangleUnresolvedName(ULE->getQualifier(), ULE->getName(), UnknownArity);
1332 break;
1333 }
1334
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001335 case Expr::CXXUnresolvedConstructExprClass: {
John McCall1dd73832010-02-04 01:42:13 +00001336 const CXXUnresolvedConstructExpr *CE = cast<CXXUnresolvedConstructExpr>(E);
1337 unsigned N = CE->arg_size();
1338
1339 Out << "cv";
1340 mangleType(CE->getType());
1341 if (N != 1) Out << "_";
1342 for (unsigned I = 0; I != N; ++I) mangleExpression(CE->getArg(I));
1343 if (N != 1) Out << "E";
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001344 break;
John McCall1dd73832010-02-04 01:42:13 +00001345 }
John McCall09cc1412010-02-03 00:55:45 +00001346
John McCall1dd73832010-02-04 01:42:13 +00001347 case Expr::CXXTemporaryObjectExprClass:
1348 case Expr::CXXConstructExprClass: {
1349 const CXXConstructExpr *CE = cast<CXXConstructExpr>(E);
1350 unsigned N = CE->getNumArgs();
1351
1352 Out << "cv";
1353 mangleType(CE->getType());
1354 if (N != 1) Out << "_";
1355 for (unsigned I = 0; I != N; ++I) mangleExpression(CE->getArg(I));
1356 if (N != 1) Out << "E";
John McCall09cc1412010-02-03 00:55:45 +00001357 break;
John McCall1dd73832010-02-04 01:42:13 +00001358 }
1359
1360 case Expr::SizeOfAlignOfExprClass: {
1361 const SizeOfAlignOfExpr *SAE = cast<SizeOfAlignOfExpr>(E);
1362 if (SAE->isSizeOf()) Out << "s";
1363 else Out << "a";
1364 if (SAE->isArgumentType()) {
1365 Out << "t";
1366 mangleType(SAE->getArgumentType());
1367 } else {
1368 Out << "z";
1369 mangleExpression(SAE->getArgumentExpr());
1370 }
1371 break;
1372 }
Anders Carlssona7694082009-11-06 02:50:19 +00001373
Anders Carlssone170ba72009-12-14 01:45:37 +00001374 case Expr::UnaryOperatorClass: {
1375 const UnaryOperator *UO = cast<UnaryOperator>(E);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001376 mangleOperatorName(UnaryOperator::getOverloadedOperator(UO->getOpcode()),
Anders Carlssone170ba72009-12-14 01:45:37 +00001377 /*Arity=*/1);
1378 mangleExpression(UO->getSubExpr());
1379 break;
1380 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001381
Anders Carlssone170ba72009-12-14 01:45:37 +00001382 case Expr::BinaryOperatorClass: {
1383 const BinaryOperator *BO = cast<BinaryOperator>(E);
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001384 mangleOperatorName(BinaryOperator::getOverloadedOperator(BO->getOpcode()),
Anders Carlssone170ba72009-12-14 01:45:37 +00001385 /*Arity=*/2);
1386 mangleExpression(BO->getLHS());
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001387 mangleExpression(BO->getRHS());
Anders Carlssone170ba72009-12-14 01:45:37 +00001388 break;
John McCall2f27bf82010-02-04 02:56:29 +00001389 }
Anders Carlssone170ba72009-12-14 01:45:37 +00001390
1391 case Expr::ConditionalOperatorClass: {
1392 const ConditionalOperator *CO = cast<ConditionalOperator>(E);
1393 mangleOperatorName(OO_Conditional, /*Arity=*/3);
1394 mangleExpression(CO->getCond());
1395 mangleExpression(CO->getLHS());
1396 mangleExpression(CO->getRHS());
1397 break;
1398 }
1399
Douglas Gregor46287c72010-01-29 16:37:09 +00001400 case Expr::ImplicitCastExprClass: {
1401 mangleExpression(cast<ImplicitCastExpr>(E)->getSubExpr());
1402 break;
1403 }
1404
1405 case Expr::CStyleCastExprClass:
1406 case Expr::CXXStaticCastExprClass:
1407 case Expr::CXXDynamicCastExprClass:
1408 case Expr::CXXReinterpretCastExprClass:
1409 case Expr::CXXConstCastExprClass:
1410 case Expr::CXXFunctionalCastExprClass: {
1411 const ExplicitCastExpr *ECE = cast<ExplicitCastExpr>(E);
1412 Out << "cv";
1413 mangleType(ECE->getType());
1414 mangleExpression(ECE->getSubExpr());
1415 break;
1416 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001417
Anders Carlsson58040a52009-12-16 05:48:46 +00001418 case Expr::CXXOperatorCallExprClass: {
1419 const CXXOperatorCallExpr *CE = cast<CXXOperatorCallExpr>(E);
1420 unsigned NumArgs = CE->getNumArgs();
1421 mangleOperatorName(CE->getOperator(), /*Arity=*/NumArgs);
1422 // Mangle the arguments.
1423 for (unsigned i = 0; i != NumArgs; ++i)
1424 mangleExpression(CE->getArg(i));
1425 break;
1426 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001427
Anders Carlssona7694082009-11-06 02:50:19 +00001428 case Expr::ParenExprClass:
1429 mangleExpression(cast<ParenExpr>(E)->getSubExpr());
1430 break;
1431
Anders Carlssond553f8c2009-09-21 01:21:10 +00001432 case Expr::DeclRefExprClass: {
Douglas Gregor5ed1bc32010-02-28 21:40:32 +00001433 const NamedDecl *D = cast<DeclRefExpr>(E)->getDecl();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001434
Anders Carlssond553f8c2009-09-21 01:21:10 +00001435 switch (D->getKind()) {
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001436 default:
Douglas Gregor5ed1bc32010-02-28 21:40:32 +00001437 // <expr-primary> ::= L <mangled-name> E # external name
1438 Out << 'L';
1439 mangle(D, "_Z");
1440 Out << 'E';
1441 break;
1442
Anders Carlssond553f8c2009-09-21 01:21:10 +00001443 case Decl::NonTypeTemplateParm: {
1444 const NonTypeTemplateParmDecl *PD = cast<NonTypeTemplateParmDecl>(D);
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001445 mangleTemplateParameter(PD->getIndex());
Anders Carlssond553f8c2009-09-21 01:21:10 +00001446 break;
1447 }
1448
1449 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001450
Anders Carlsson50755b02009-09-27 20:11:34 +00001451 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001452 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001453
John McCall865d4472009-11-19 22:55:06 +00001454 case Expr::DependentScopeDeclRefExprClass: {
1455 const DependentScopeDeclRefExpr *DRE = cast<DependentScopeDeclRefExpr>(E);
Douglas Gregor4b2ccfc2010-02-28 22:05:49 +00001456 NestedNameSpecifier *NNS = DRE->getQualifier();
1457 const Type *QTy = NNS->getAsType();
1458
1459 // When we're dealing with a nested-name-specifier that has just a
1460 // dependent identifier in it, mangle that as a typename. FIXME:
1461 // It isn't clear that we ever actually want to have such a
1462 // nested-name-specifier; why not just represent it as a typename type?
1463 if (!QTy && NNS->getAsIdentifier() && NNS->getPrefix()) {
1464 QTy = getASTContext().getTypenameType(NNS->getPrefix(),
1465 NNS->getAsIdentifier())
1466 .getTypePtr();
1467 }
Anders Carlsson50755b02009-09-27 20:11:34 +00001468 assert(QTy && "Qualifier was not type!");
1469
1470 // ::= sr <type> <unqualified-name> # dependent name
1471 Out << "sr";
1472 mangleType(QualType(QTy, 0));
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001473
Anders Carlsson50755b02009-09-27 20:11:34 +00001474 assert(DRE->getDeclName().getNameKind() == DeclarationName::Identifier &&
1475 "Unhandled decl name kind!");
1476 mangleSourceName(DRE->getDeclName().getAsIdentifierInfo());
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001477
Anders Carlsson50755b02009-09-27 20:11:34 +00001478 break;
1479 }
1480
John McCall1dd73832010-02-04 01:42:13 +00001481 case Expr::FloatingLiteralClass: {
1482 const FloatingLiteral *FL = cast<FloatingLiteral>(E);
1483 Out << "L";
1484 mangleType(FL->getType());
1485
1486 // TODO: avoid this copy with careful stream management.
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001487 llvm::SmallString<20> Buffer;
John McCall1dd73832010-02-04 01:42:13 +00001488 FL->getValue().bitcastToAPInt().toString(Buffer, 16, false);
1489 Out.write(Buffer.data(), Buffer.size());
1490
1491 Out << "E";
1492 break;
1493 }
1494
Anders Carlssone170ba72009-12-14 01:45:37 +00001495 case Expr::IntegerLiteralClass:
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001496 mangleIntegerLiteral(E->getType(),
Anders Carlssone170ba72009-12-14 01:45:37 +00001497 llvm::APSInt(cast<IntegerLiteral>(E)->getValue()));
1498 break;
1499
Anders Carlssond553f8c2009-09-21 01:21:10 +00001500 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001501}
1502
John McCallefe6aee2009-09-05 07:56:18 +00001503// FIXME: <type> ::= G <type> # imaginary (C 2000)
1504// FIXME: <type> ::= U <source-name> <type> # vendor extended type qualifier
1505
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001506void CXXNameMangler::mangleCXXCtorType(CXXCtorType T) {
1507 // <ctor-dtor-name> ::= C1 # complete object constructor
1508 // ::= C2 # base object constructor
1509 // ::= C3 # complete object allocating constructor
1510 //
1511 switch (T) {
1512 case Ctor_Complete:
1513 Out << "C1";
1514 break;
1515 case Ctor_Base:
1516 Out << "C2";
1517 break;
1518 case Ctor_CompleteAllocating:
1519 Out << "C3";
1520 break;
1521 }
1522}
1523
Anders Carlsson27ae5362009-04-17 01:58:57 +00001524void CXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
1525 // <ctor-dtor-name> ::= D0 # deleting destructor
1526 // ::= D1 # complete object destructor
1527 // ::= D2 # base object destructor
1528 //
1529 switch (T) {
1530 case Dtor_Deleting:
1531 Out << "D0";
1532 break;
1533 case Dtor_Complete:
1534 Out << "D1";
1535 break;
1536 case Dtor_Base:
1537 Out << "D2";
1538 break;
1539 }
1540}
1541
Rafael Espindolad9800722010-03-11 14:07:00 +00001542void CXXNameMangler::mangleTemplateArgs(const TemplateParameterList &PL,
1543 const TemplateArgumentList &AL) {
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001544 // <template-args> ::= I <template-arg>+ E
1545 Out << "I";
Rafael Espindolad9800722010-03-11 14:07:00 +00001546 for (unsigned i = 0, e = AL.size(); i != e; ++i)
1547 mangleTemplateArg(PL.getParam(i), AL[i]);
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001548 Out << "E";
1549}
1550
Rafael Espindolad9800722010-03-11 14:07:00 +00001551void CXXNameMangler::mangleTemplateArgs(const TemplateParameterList &PL,
1552 const TemplateArgument *TemplateArgs,
Anders Carlsson7624f212009-09-18 02:42:01 +00001553 unsigned NumTemplateArgs) {
1554 // <template-args> ::= I <template-arg>+ E
1555 Out << "I";
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001556 for (unsigned i = 0; i != NumTemplateArgs; ++i)
Rafael Espindolad9800722010-03-11 14:07:00 +00001557 mangleTemplateArg(PL.getParam(i), TemplateArgs[i]);
Anders Carlsson7624f212009-09-18 02:42:01 +00001558 Out << "E";
1559}
1560
Rafael Espindolad9800722010-03-11 14:07:00 +00001561void CXXNameMangler::mangleTemplateArg(const NamedDecl *P,
1562 const TemplateArgument &A) {
Mike Stump1eb44332009-09-09 15:08:12 +00001563 // <template-arg> ::= <type> # type or template
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001564 // ::= X <expression> E # expression
1565 // ::= <expr-primary> # simple expressions
1566 // ::= I <template-arg>* E # argument pack
1567 // ::= sp <expression> # pack expansion of (C++0x)
1568 switch (A.getKind()) {
1569 default:
1570 assert(0 && "Unknown template argument kind!");
1571 case TemplateArgument::Type:
1572 mangleType(A.getAsType());
1573 break;
Anders Carlsson9e85c742009-12-23 19:30:55 +00001574 case TemplateArgument::Template:
Douglas Gregor2d565b32010-02-06 01:09:36 +00001575 assert(A.getAsTemplate().getAsTemplateDecl() &&
1576 "FIXME: Support dependent template names");
Anders Carlsson9e85c742009-12-23 19:30:55 +00001577 mangleName(A.getAsTemplate().getAsTemplateDecl());
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001578 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001579 case TemplateArgument::Expression:
1580 Out << 'X';
1581 mangleExpression(A.getAsExpr());
1582 Out << 'E';
1583 break;
Anders Carlssone170ba72009-12-14 01:45:37 +00001584 case TemplateArgument::Integral:
1585 mangleIntegerLiteral(A.getIntegralType(), *A.getAsIntegral());
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001586 break;
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001587 case TemplateArgument::Declaration: {
1588 // <expr-primary> ::= L <mangled-name> E # external name
1589
Rafael Espindolad9800722010-03-11 14:07:00 +00001590 // Clang produces AST's where pointer-to-member-function expressions
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001591 // and pointer-to-function expressions are represented as a declaration not
Rafael Espindolad9800722010-03-11 14:07:00 +00001592 // an expression. We compensate for it here to produce the correct mangling.
1593 NamedDecl *D = cast<NamedDecl>(A.getAsDecl());
1594 const NonTypeTemplateParmDecl *Parameter = cast<NonTypeTemplateParmDecl>(P);
1595 bool compensateMangling = D->isCXXClassMember() &&
1596 !Parameter->getType()->isReferenceType();
1597 if (compensateMangling) {
1598 Out << 'X';
1599 mangleOperatorName(OO_Amp, 1);
1600 }
1601
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001602 Out << 'L';
1603 // References to external entities use the mangled name; if the name would
1604 // not normally be manged then mangle it as unqualified.
1605 //
1606 // FIXME: The ABI specifies that external names here should have _Z, but
1607 // gcc leaves this off.
Rafael Espindolad9800722010-03-11 14:07:00 +00001608 if (compensateMangling)
1609 mangle(D, "_Z");
1610 else
1611 mangle(D, "Z");
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001612 Out << 'E';
Rafael Espindolad9800722010-03-11 14:07:00 +00001613
1614 if (compensateMangling)
1615 Out << 'E';
1616
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001617 break;
1618 }
1619 }
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001620}
1621
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001622void CXXNameMangler::mangleTemplateParameter(unsigned Index) {
1623 // <template-param> ::= T_ # first template parameter
1624 // ::= T <parameter-2 non-negative number> _
1625 if (Index == 0)
1626 Out << "T_";
1627 else
1628 Out << 'T' << (Index - 1) << '_';
1629}
1630
Anders Carlsson76967372009-09-17 00:43:46 +00001631// <substitution> ::= S <seq-id> _
1632// ::= S_
Anders Carlsson6862fc72009-09-17 04:16:28 +00001633bool CXXNameMangler::mangleSubstitution(const NamedDecl *ND) {
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001634 // Try one of the standard substitutions first.
1635 if (mangleStandardSubstitution(ND))
1636 return true;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001637
Anders Carlsson433d1372009-11-07 04:26:04 +00001638 ND = cast<NamedDecl>(ND->getCanonicalDecl());
Anders Carlsson6862fc72009-09-17 04:16:28 +00001639 return mangleSubstitution(reinterpret_cast<uintptr_t>(ND));
1640}
1641
Anders Carlsson76967372009-09-17 00:43:46 +00001642bool CXXNameMangler::mangleSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001643 if (!T.getCVRQualifiers()) {
1644 if (const RecordType *RT = T->getAs<RecordType>())
1645 return mangleSubstitution(RT->getDecl());
1646 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001647
Anders Carlsson76967372009-09-17 00:43:46 +00001648 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
1649
Anders Carlssond3a932a2009-09-17 03:53:28 +00001650 return mangleSubstitution(TypePtr);
1651}
1652
1653bool CXXNameMangler::mangleSubstitution(uintptr_t Ptr) {
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001654 llvm::DenseMap<uintptr_t, unsigned>::iterator I =
Anders Carlssond3a932a2009-09-17 03:53:28 +00001655 Substitutions.find(Ptr);
Anders Carlsson76967372009-09-17 00:43:46 +00001656 if (I == Substitutions.end())
1657 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001658
Anders Carlsson76967372009-09-17 00:43:46 +00001659 unsigned SeqID = I->second;
1660 if (SeqID == 0)
1661 Out << "S_";
1662 else {
1663 SeqID--;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001664
Anders Carlsson76967372009-09-17 00:43:46 +00001665 // <seq-id> is encoded in base-36, using digits and upper case letters.
1666 char Buffer[10];
1667 char *BufferPtr = Buffer + 9;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001668
Anders Carlsson76967372009-09-17 00:43:46 +00001669 *BufferPtr = 0;
1670 if (SeqID == 0) *--BufferPtr = '0';
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001671
Anders Carlsson76967372009-09-17 00:43:46 +00001672 while (SeqID) {
1673 assert(BufferPtr > Buffer && "Buffer overflow!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001674
Anders Carlsson76967372009-09-17 00:43:46 +00001675 unsigned char c = static_cast<unsigned char>(SeqID) % 36;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001676
Anders Carlsson76967372009-09-17 00:43:46 +00001677 *--BufferPtr = (c < 10 ? '0' + c : 'A' + c - 10);
1678 SeqID /= 36;
1679 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001680
Anders Carlsson76967372009-09-17 00:43:46 +00001681 Out << 'S' << BufferPtr << '_';
1682 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001683
Anders Carlsson76967372009-09-17 00:43:46 +00001684 return true;
1685}
1686
Anders Carlssonf514b542009-09-27 00:12:57 +00001687static bool isCharType(QualType T) {
1688 if (T.isNull())
1689 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001690
Anders Carlssonf514b542009-09-27 00:12:57 +00001691 return T->isSpecificBuiltinType(BuiltinType::Char_S) ||
1692 T->isSpecificBuiltinType(BuiltinType::Char_U);
1693}
1694
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001695/// isCharSpecialization - Returns whether a given type is a template
Anders Carlssonf514b542009-09-27 00:12:57 +00001696/// specialization of a given name with a single argument of type char.
1697static bool isCharSpecialization(QualType T, const char *Name) {
1698 if (T.isNull())
1699 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001700
Anders Carlssonf514b542009-09-27 00:12:57 +00001701 const RecordType *RT = T->getAs<RecordType>();
1702 if (!RT)
1703 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001704
1705 const ClassTemplateSpecializationDecl *SD =
Anders Carlssonf514b542009-09-27 00:12:57 +00001706 dyn_cast<ClassTemplateSpecializationDecl>(RT->getDecl());
1707 if (!SD)
1708 return false;
1709
1710 if (!isStdNamespace(SD->getDeclContext()))
1711 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001712
Anders Carlssonf514b542009-09-27 00:12:57 +00001713 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1714 if (TemplateArgs.size() != 1)
1715 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001716
Anders Carlssonf514b542009-09-27 00:12:57 +00001717 if (!isCharType(TemplateArgs[0].getAsType()))
1718 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001719
Daniel Dunbar01eb9b92009-10-18 21:17:35 +00001720 return SD->getIdentifier()->getName() == Name;
Anders Carlssonf514b542009-09-27 00:12:57 +00001721}
1722
Anders Carlsson91f88602009-12-07 19:56:42 +00001723template <std::size_t StrLen>
1724bool isStreamCharSpecialization(const ClassTemplateSpecializationDecl *SD,
1725 const char (&Str)[StrLen]) {
1726 if (!SD->getIdentifier()->isStr(Str))
1727 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001728
Anders Carlsson91f88602009-12-07 19:56:42 +00001729 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1730 if (TemplateArgs.size() != 2)
1731 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001732
Anders Carlsson91f88602009-12-07 19:56:42 +00001733 if (!isCharType(TemplateArgs[0].getAsType()))
1734 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001735
Anders Carlsson91f88602009-12-07 19:56:42 +00001736 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1737 return false;
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001738
Anders Carlsson91f88602009-12-07 19:56:42 +00001739 return true;
1740}
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001741
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001742bool CXXNameMangler::mangleStandardSubstitution(const NamedDecl *ND) {
1743 // <substitution> ::= St # ::std::
Anders Carlsson8c031552009-09-26 23:10:05 +00001744 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
Anders Carlsson47846d22009-12-04 06:23:23 +00001745 if (isStd(NS)) {
Anders Carlsson8c031552009-09-26 23:10:05 +00001746 Out << "St";
1747 return true;
1748 }
1749 }
1750
1751 if (const ClassTemplateDecl *TD = dyn_cast<ClassTemplateDecl>(ND)) {
1752 if (!isStdNamespace(TD->getDeclContext()))
1753 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001754
Anders Carlsson8c031552009-09-26 23:10:05 +00001755 // <substitution> ::= Sa # ::std::allocator
1756 if (TD->getIdentifier()->isStr("allocator")) {
1757 Out << "Sa";
1758 return true;
1759 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001760
Anders Carlsson189d59c2009-09-26 23:14:39 +00001761 // <<substitution> ::= Sb # ::std::basic_string
1762 if (TD->getIdentifier()->isStr("basic_string")) {
1763 Out << "Sb";
1764 return true;
1765 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001766 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001767
1768 if (const ClassTemplateSpecializationDecl *SD =
Anders Carlssonf514b542009-09-27 00:12:57 +00001769 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
Eli Friedman5370ee22010-02-23 18:25:09 +00001770 if (!isStdNamespace(SD->getDeclContext()))
1771 return false;
1772
Anders Carlssonf514b542009-09-27 00:12:57 +00001773 // <substitution> ::= Ss # ::std::basic_string<char,
1774 // ::std::char_traits<char>,
1775 // ::std::allocator<char> >
1776 if (SD->getIdentifier()->isStr("basic_string")) {
1777 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001778
Anders Carlssonf514b542009-09-27 00:12:57 +00001779 if (TemplateArgs.size() != 3)
1780 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001781
Anders Carlssonf514b542009-09-27 00:12:57 +00001782 if (!isCharType(TemplateArgs[0].getAsType()))
1783 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001784
Anders Carlssonf514b542009-09-27 00:12:57 +00001785 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1786 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001787
Anders Carlssonf514b542009-09-27 00:12:57 +00001788 if (!isCharSpecialization(TemplateArgs[2].getAsType(), "allocator"))
1789 return false;
1790
1791 Out << "Ss";
1792 return true;
1793 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001794
Anders Carlsson91f88602009-12-07 19:56:42 +00001795 // <substitution> ::= Si # ::std::basic_istream<char,
1796 // ::std::char_traits<char> >
1797 if (isStreamCharSpecialization(SD, "basic_istream")) {
1798 Out << "Si";
1799 return true;
1800 }
1801
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001802 // <substitution> ::= So # ::std::basic_ostream<char,
Anders Carlsson8f8fd8e2009-10-08 17:20:26 +00001803 // ::std::char_traits<char> >
Anders Carlsson91f88602009-12-07 19:56:42 +00001804 if (isStreamCharSpecialization(SD, "basic_ostream")) {
Anders Carlsson8f8fd8e2009-10-08 17:20:26 +00001805 Out << "So";
1806 return true;
1807 }
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001808
Anders Carlsson91f88602009-12-07 19:56:42 +00001809 // <substitution> ::= Sd # ::std::basic_iostream<char,
1810 // ::std::char_traits<char> >
1811 if (isStreamCharSpecialization(SD, "basic_iostream")) {
1812 Out << "Sd";
1813 return true;
1814 }
Anders Carlssonf514b542009-09-27 00:12:57 +00001815 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001816 return false;
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001817}
1818
Anders Carlsson76967372009-09-17 00:43:46 +00001819void CXXNameMangler::addSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001820 if (!T.getCVRQualifiers()) {
1821 if (const RecordType *RT = T->getAs<RecordType>()) {
1822 addSubstitution(RT->getDecl());
1823 return;
1824 }
1825 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001826
Anders Carlsson76967372009-09-17 00:43:46 +00001827 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
Anders Carlssond3a932a2009-09-17 03:53:28 +00001828 addSubstitution(TypePtr);
1829}
1830
1831void CXXNameMangler::addSubstitution(uintptr_t Ptr) {
Anders Carlsson76967372009-09-17 00:43:46 +00001832 unsigned SeqID = Substitutions.size();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001833
Anders Carlssond3a932a2009-09-17 03:53:28 +00001834 assert(!Substitutions.count(Ptr) && "Substitution already exists!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001835 Substitutions[Ptr] = SeqID;
Anders Carlsson76967372009-09-17 00:43:46 +00001836}
1837
Daniel Dunbar1b077112009-11-21 09:06:10 +00001838//
Mike Stump1eb44332009-09-09 15:08:12 +00001839
Daniel Dunbar1b077112009-11-21 09:06:10 +00001840/// \brief Mangles the name of the declaration D and emits that name to the
1841/// given output stream.
1842///
1843/// If the declaration D requires a mangled name, this routine will emit that
1844/// mangled name to \p os and return true. Otherwise, \p os will be unchanged
1845/// and this routine will return false. In this case, the caller should just
1846/// emit the identifier of the declaration (\c D->getIdentifier()) as its
1847/// name.
Daniel Dunbarf981bf82009-11-21 09:14:52 +00001848void MangleContext::mangleName(const NamedDecl *D,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001849 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc02ab4c2009-11-21 09:14:44 +00001850 assert((isa<FunctionDecl>(D) || isa<VarDecl>(D)) &&
1851 "Invalid mangleName() call, argument is not a variable or function!");
1852 assert(!isa<CXXConstructorDecl>(D) && !isa<CXXDestructorDecl>(D) &&
1853 "Invalid mangleName() call on 'structor decl!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001854
Daniel Dunbar1b077112009-11-21 09:06:10 +00001855 PrettyStackTraceDecl CrashInfo(D, SourceLocation(),
1856 getASTContext().getSourceManager(),
1857 "Mangling declaration");
Mike Stump1eb44332009-09-09 15:08:12 +00001858
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001859 CXXNameMangler Mangler(*this, Res);
1860 return Mangler.mangle(D);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001861}
Mike Stump1eb44332009-09-09 15:08:12 +00001862
Daniel Dunbar1b077112009-11-21 09:06:10 +00001863void MangleContext::mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001864 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbar77939c92009-11-21 09:06:31 +00001865 CXXNameMangler Mangler(*this, Res, D, Type);
1866 Mangler.mangle(D);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001867}
Mike Stump1eb44332009-09-09 15:08:12 +00001868
Daniel Dunbar1b077112009-11-21 09:06:10 +00001869void MangleContext::mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001870 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbar77939c92009-11-21 09:06:31 +00001871 CXXNameMangler Mangler(*this, Res, D, Type);
1872 Mangler.mangle(D);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001873}
Mike Stumpf1216772009-07-31 18:25:34 +00001874
Daniel Dunbarc0747712009-11-21 09:12:13 +00001875/// \brief Mangles the a thunk with the offset n for the declaration D and
1876/// emits that name to the given output stream.
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001877void MangleContext::mangleThunk(const FunctionDecl *FD,
Anders Carlssonb73a5be2009-11-26 02:49:32 +00001878 const ThunkAdjustment &ThisAdjustment,
Daniel Dunbarc0747712009-11-21 09:12:13 +00001879 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001880 assert(!isa<CXXDestructorDecl>(FD) &&
1881 "Use mangleCXXDtor for destructor decls!");
1882
1883 // <special-name> ::= T <call-offset> <base encoding>
1884 // # base is the nominal target function of thunk
1885 CXXNameMangler Mangler(*this, Res);
1886 Mangler.getStream() << "_ZT";
Anders Carlssonb73a5be2009-11-26 02:49:32 +00001887 Mangler.mangleCallOffset(ThisAdjustment);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001888 Mangler.mangleFunctionEncoding(FD);
1889}
1890
Anders Carlsson19879c92010-03-23 17:17:29 +00001891void MangleContext::mangleThunk(const CXXMethodDecl *MD,
1892 const ThunkInfo &Thunk,
1893 llvm::SmallVectorImpl<char> &Res) {
1894 // <special-name> ::= T <call-offset> <base encoding>
1895 // # base is the nominal target function of thunk
1896 // <special-name> ::= Tc <call-offset> <call-offset> <base encoding>
1897 // # base is the nominal target function of thunk
1898 // # first call-offset is 'this' adjustment
1899 // # second call-offset is result adjustment
1900
1901 assert(!isa<CXXDestructorDecl>(MD) &&
1902 "Use mangleCXXDtor for destructor decls!");
1903
1904 CXXNameMangler Mangler(*this, Res);
1905 Mangler.getStream() << "_ZT";
1906 if (!Thunk.Return.isEmpty())
1907 Mangler.getStream() << 'c';
1908
1909 // Mangle the 'this' pointer adjustment.
1910 Mangler.mangleCallOffset(Thunk.This.NonVirtual, Thunk.This.VCallOffsetOffset);
1911
1912 // Mangle the return pointer adjustment if there is one.
1913 if (!Thunk.Return.isEmpty())
1914 Mangler.mangleCallOffset(Thunk.Return.NonVirtual,
1915 Thunk.Return.VBaseOffsetOffset);
1916
1917 Mangler.mangleFunctionEncoding(MD);
1918}
1919
1920void
1921MangleContext::mangleCXXDtorThunk(const CXXDestructorDecl *DD, CXXDtorType Type,
1922 const ThisAdjustment &ThisAdjustment,
1923 llvm::SmallVectorImpl<char> &Res) {
1924 // <special-name> ::= T <call-offset> <base encoding>
1925 // # base is the nominal target function of thunk
1926
1927 CXXNameMangler Mangler(*this, Res, DD, Type);
1928 Mangler.getStream() << "_ZT";
1929
1930 // Mangle the 'this' pointer adjustment.
1931 Mangler.mangleCallOffset(ThisAdjustment.NonVirtual,
1932 ThisAdjustment.VCallOffsetOffset);
1933
1934 Mangler.mangleFunctionEncoding(DD);
1935}
1936
Eli Friedman61d89b62009-12-03 00:03:05 +00001937void MangleContext::mangleCXXDtorThunk(const CXXDestructorDecl *D,
1938 CXXDtorType Type,
1939 const ThunkAdjustment &ThisAdjustment,
1940 llvm::SmallVectorImpl<char> &Res) {
1941 // <special-name> ::= T <call-offset> <base encoding>
1942 // # base is the nominal target function of thunk
1943 CXXNameMangler Mangler(*this, Res, D, Type);
1944 Mangler.getStream() << "_ZT";
1945 Mangler.mangleCallOffset(ThisAdjustment);
1946 Mangler.mangleFunctionEncoding(D);
1947}
1948
Daniel Dunbarc0747712009-11-21 09:12:13 +00001949/// \brief Mangles the a covariant thunk for the declaration D and emits that
1950/// name to the given output stream.
Kovarththanan Rajaratnam19357542010-03-13 10:17:05 +00001951void
Anders Carlsson7622cd32009-11-26 03:09:37 +00001952MangleContext::mangleCovariantThunk(const FunctionDecl *FD,
1953 const CovariantThunkAdjustment& Adjustment,
1954 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001955 assert(!isa<CXXDestructorDecl>(FD) &&
Eli Friedman61d89b62009-12-03 00:03:05 +00001956 "No such thing as a covariant thunk for a destructor!");
Daniel Dunbarc0747712009-11-21 09:12:13 +00001957
1958 // <special-name> ::= Tc <call-offset> <call-offset> <base encoding>
1959 // # base is the nominal target function of thunk
1960 // # first call-offset is 'this' adjustment
1961 // # second call-offset is result adjustment
1962 CXXNameMangler Mangler(*this, Res);
1963 Mangler.getStream() << "_ZTc";
Anders Carlsson7622cd32009-11-26 03:09:37 +00001964 Mangler.mangleCallOffset(Adjustment.ThisAdjustment);
1965 Mangler.mangleCallOffset(Adjustment.ReturnAdjustment);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001966 Mangler.mangleFunctionEncoding(FD);
1967}
1968
1969/// mangleGuardVariable - Returns the mangled name for a guard variable
1970/// for the passed in VarDecl.
1971void MangleContext::mangleGuardVariable(const VarDecl *D,
1972 llvm::SmallVectorImpl<char> &Res) {
1973 // <special-name> ::= GV <object name> # Guard variable for one-time
1974 // # initialization
1975 CXXNameMangler Mangler(*this, Res);
1976 Mangler.getStream() << "_ZGV";
1977 Mangler.mangleName(D);
1978}
1979
Daniel Dunbar1b077112009-11-21 09:06:10 +00001980void MangleContext::mangleCXXVtable(const CXXRecordDecl *RD,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001981 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001982 // <special-name> ::= TV <type> # virtual table
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001983 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001984 Mangler.getStream() << "_ZTV";
1985 Mangler.mangleName(RD);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001986}
Mike Stump82d75b02009-11-10 01:58:37 +00001987
Daniel Dunbar1b077112009-11-21 09:06:10 +00001988void MangleContext::mangleCXXVTT(const CXXRecordDecl *RD,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001989 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001990 // <special-name> ::= TT <type> # VTT structure
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001991 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001992 Mangler.getStream() << "_ZTT";
1993 Mangler.mangleName(RD);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001994}
Mike Stumpab3f7e92009-11-10 01:41:59 +00001995
Daniel Dunbar1b077112009-11-21 09:06:10 +00001996void MangleContext::mangleCXXCtorVtable(const CXXRecordDecl *RD, int64_t Offset,
1997 const CXXRecordDecl *Type,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001998 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001999 // <special-name> ::= TC <type> <offset number> _ <base type>
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002000 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00002001 Mangler.getStream() << "_ZTC";
2002 Mangler.mangleName(RD);
2003 Mangler.getStream() << Offset;
2004 Mangler.getStream() << "_";
2005 Mangler.mangleName(Type);
Daniel Dunbar1b077112009-11-21 09:06:10 +00002006}
Mike Stump738f8c22009-07-31 23:15:31 +00002007
Mike Stumpde050572009-12-02 18:57:08 +00002008void MangleContext::mangleCXXRTTI(QualType Ty,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002009 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00002010 // <special-name> ::= TI <type> # typeinfo structure
Douglas Gregor154fe982009-12-23 22:04:40 +00002011 assert(!Ty.hasQualifiers() && "RTTI info cannot have top-level qualifiers");
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002012 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00002013 Mangler.getStream() << "_ZTI";
2014 Mangler.mangleType(Ty);
Daniel Dunbar1b077112009-11-21 09:06:10 +00002015}
Mike Stump67795982009-11-14 00:14:13 +00002016
Mike Stumpde050572009-12-02 18:57:08 +00002017void MangleContext::mangleCXXRTTIName(QualType Ty,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002018 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00002019 // <special-name> ::= TS <type> # typeinfo name (null terminated byte string)
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00002020 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00002021 Mangler.getStream() << "_ZTS";
2022 Mangler.mangleType(Ty);
Mike Stumpf1216772009-07-31 18:25:34 +00002023}