blob: 4f84edd908bf5e47eea7d2d85f1591e986003f54 [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"
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000029using namespace clang;
Anders Carlssonb73a5be2009-11-26 02:49:32 +000030using namespace CodeGen;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000031
32namespace {
Anders Carlsson7e120032009-11-24 05:36:32 +000033
34static const CXXMethodDecl *getStructor(const CXXMethodDecl *MD) {
35 assert((isa<CXXConstructorDecl>(MD) || isa<CXXDestructorDecl>(MD)) &&
36 "Passed in decl is not a ctor or dtor!");
37
38 if (const TemplateDecl *TD = MD->getPrimaryTemplate()) {
39 MD = cast<CXXMethodDecl>(TD->getTemplatedDecl());
40
41 assert((isa<CXXConstructorDecl>(MD) || isa<CXXDestructorDecl>(MD)) &&
42 "Templated decl is not a ctor or dtor!");
43 }
44
45 return MD;
46}
47
Daniel Dunbar1b077112009-11-21 09:06:10 +000048/// CXXNameMangler - Manage the mangling of a single name.
Daniel Dunbarc0747712009-11-21 09:12:13 +000049class CXXNameMangler {
Daniel Dunbar1b077112009-11-21 09:06:10 +000050 MangleContext &Context;
Daniel Dunbar94fd26d2009-11-21 09:06:22 +000051 llvm::raw_svector_ostream Out;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000052
Daniel Dunbar1b077112009-11-21 09:06:10 +000053 const CXXMethodDecl *Structor;
54 unsigned StructorType;
Mike Stump1eb44332009-09-09 15:08:12 +000055
Daniel Dunbar1b077112009-11-21 09:06:10 +000056 llvm::DenseMap<uintptr_t, unsigned> Substitutions;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +000057
Daniel Dunbarc0747712009-11-21 09:12:13 +000058public:
Daniel Dunbar94fd26d2009-11-21 09:06:22 +000059 CXXNameMangler(MangleContext &C, llvm::SmallVectorImpl<char> &Res)
60 : Context(C), Out(Res), Structor(0), StructorType(0) { }
Daniel Dunbar77939c92009-11-21 09:06:31 +000061 CXXNameMangler(MangleContext &C, llvm::SmallVectorImpl<char> &Res,
62 const CXXConstructorDecl *D, CXXCtorType Type)
Anders Carlsson7e120032009-11-24 05:36:32 +000063 : Context(C), Out(Res), Structor(getStructor(D)), StructorType(Type) { }
Daniel Dunbar77939c92009-11-21 09:06:31 +000064 CXXNameMangler(MangleContext &C, llvm::SmallVectorImpl<char> &Res,
65 const CXXDestructorDecl *D, CXXDtorType Type)
Anders Carlsson7e120032009-11-24 05:36:32 +000066 : Context(C), Out(Res), Structor(getStructor(D)), StructorType(Type) { }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000067
Daniel Dunbarc0747712009-11-21 09:12:13 +000068 llvm::raw_svector_ostream &getStream() { return Out; }
69
Daniel Dunbar7e0c1952009-11-21 09:17:15 +000070 void mangle(const NamedDecl *D, llvm::StringRef Prefix = "_Z");
Anders Carlssonb73a5be2009-11-26 02:49:32 +000071 void mangleCallOffset(const ThunkAdjustment &Adjustment);
Anders Carlssona94822e2009-11-26 02:32:05 +000072 void mangleNumber(int64_t Number);
Daniel Dunbarc0747712009-11-21 09:12:13 +000073 void mangleFunctionEncoding(const FunctionDecl *FD);
74 void mangleName(const NamedDecl *ND);
75 void mangleType(QualType T);
Mike Stump1eb44332009-09-09 15:08:12 +000076
Daniel Dunbarc0747712009-11-21 09:12:13 +000077private:
Daniel Dunbar1b077112009-11-21 09:06:10 +000078 bool mangleSubstitution(const NamedDecl *ND);
79 bool mangleSubstitution(QualType T);
80 bool mangleSubstitution(uintptr_t Ptr);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +000081
Daniel Dunbar1b077112009-11-21 09:06:10 +000082 bool mangleStandardSubstitution(const NamedDecl *ND);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +000083
Daniel Dunbar1b077112009-11-21 09:06:10 +000084 void addSubstitution(const NamedDecl *ND) {
85 ND = cast<NamedDecl>(ND->getCanonicalDecl());
Anders Carlsson433d1372009-11-07 04:26:04 +000086
Daniel Dunbar1b077112009-11-21 09:06:10 +000087 addSubstitution(reinterpret_cast<uintptr_t>(ND));
88 }
89 void addSubstitution(QualType T);
90 void addSubstitution(uintptr_t Ptr);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +000091
Daniel Dunbar1b077112009-11-21 09:06:10 +000092 void mangleName(const TemplateDecl *TD,
93 const TemplateArgument *TemplateArgs,
94 unsigned NumTemplateArgs);
95 void mangleUnqualifiedName(const NamedDecl *ND);
96 void mangleUnscopedName(const NamedDecl *ND);
97 void mangleUnscopedTemplateName(const TemplateDecl *ND);
98 void mangleSourceName(const IdentifierInfo *II);
99 void mangleLocalName(const NamedDecl *ND);
Eli Friedman7facf842009-12-02 20:32:49 +0000100 void mangleNestedName(const NamedDecl *ND, const DeclContext *DC);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000101 void mangleNestedName(const TemplateDecl *TD,
102 const TemplateArgument *TemplateArgs,
103 unsigned NumTemplateArgs);
104 void manglePrefix(const DeclContext *DC);
105 void mangleTemplatePrefix(const TemplateDecl *ND);
106 void mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity);
107 void mangleQualifiers(Qualifiers Quals);
John McCallefe6aee2009-09-05 07:56:18 +0000108
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000109 void mangleObjCMethodName(const ObjCMethodDecl *MD);
110
Daniel Dunbar1b077112009-11-21 09:06:10 +0000111 // Declare manglers for every type class.
John McCallefe6aee2009-09-05 07:56:18 +0000112#define ABSTRACT_TYPE(CLASS, PARENT)
113#define NON_CANONICAL_TYPE(CLASS, PARENT)
114#define TYPE(CLASS, PARENT) void mangleType(const CLASS##Type *T);
115#include "clang/AST/TypeNodes.def"
116
Daniel Dunbar1b077112009-11-21 09:06:10 +0000117 void mangleType(const TagType*);
118 void mangleBareFunctionType(const FunctionType *T,
119 bool MangleReturnType);
Anders Carlssone170ba72009-12-14 01:45:37 +0000120
121 void mangleIntegerLiteral(QualType T, const llvm::APSInt &Value);
Daniel Dunbar1b077112009-11-21 09:06:10 +0000122 void mangleExpression(const Expr *E);
123 void mangleCXXCtorType(CXXCtorType T);
124 void mangleCXXDtorType(CXXDtorType T);
Mike Stump1eb44332009-09-09 15:08:12 +0000125
Daniel Dunbar1b077112009-11-21 09:06:10 +0000126 void mangleTemplateArgs(const TemplateArgument *TemplateArgs,
127 unsigned NumTemplateArgs);
Anders Carlsson9e85c742009-12-23 19:30:55 +0000128 void mangleTemplateArgs(const TemplateArgumentList &L);
129 void mangleTemplateArg(const TemplateArgument &A);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000130
Daniel Dunbar1b077112009-11-21 09:06:10 +0000131 void mangleTemplateParameter(unsigned Index);
132};
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000133}
134
Anders Carlsson43f17402009-04-02 15:51:53 +0000135static bool isInCLinkageSpecification(const Decl *D) {
Douglas Gregor457e2812009-10-28 16:31:34 +0000136 D = D->getCanonicalDecl();
Mike Stump1eb44332009-09-09 15:08:12 +0000137 for (const DeclContext *DC = D->getDeclContext();
Anders Carlsson43f17402009-04-02 15:51:53 +0000138 !DC->isTranslationUnit(); DC = DC->getParent()) {
Mike Stump1eb44332009-09-09 15:08:12 +0000139 if (const LinkageSpecDecl *Linkage = dyn_cast<LinkageSpecDecl>(DC))
Anders Carlsson43f17402009-04-02 15:51:53 +0000140 return Linkage->getLanguage() == LinkageSpecDecl::lang_c;
141 }
Mike Stump1eb44332009-09-09 15:08:12 +0000142
Anders Carlsson43f17402009-04-02 15:51:53 +0000143 return false;
144}
145
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000146bool MangleContext::shouldMangleDeclName(const NamedDecl *D) {
147 // In C, functions with no attributes never need to be mangled. Fastpath them.
148 if (!getASTContext().getLangOptions().CPlusPlus && !D->hasAttrs())
149 return false;
150
151 // Any decl can be declared with __asm("foo") on it, and this takes precedence
152 // over all other naming in the .o file.
153 if (D->hasAttr<AsmLabelAttr>())
154 return true;
155
Mike Stump141c5af2009-09-02 00:25:38 +0000156 // Clang's "overloadable" attribute extension to C/C++ implies name mangling
Anders Carlssona1e16222009-11-07 07:15:03 +0000157 // (always) as does passing a C++ member function and a function
158 // whose name is not a simple identifier.
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000159 const FunctionDecl *FD = dyn_cast<FunctionDecl>(D);
160 if (FD && (FD->hasAttr<OverloadableAttr>() || isa<CXXMethodDecl>(FD) ||
161 !FD->getDeclName().isIdentifier()))
162 return true;
Mike Stump1eb44332009-09-09 15:08:12 +0000163
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000164 // Otherwise, no mangling is done outside C++ mode.
165 if (!getASTContext().getLangOptions().CPlusPlus)
166 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000167
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000168 // No mangling in an "implicit extern C" header.
169 if (D->getLocation().isValid() &&
170 getASTContext().getSourceManager().
171 isInExternCSystemHeader(D->getLocation()))
172 return false;
Anders Carlsson43f17402009-04-02 15:51:53 +0000173
Sean Hunt31455252010-01-24 03:04:27 +0000174 // Variables at global scope with non-internal linkage are not mangled
Eli Friedman7facf842009-12-02 20:32:49 +0000175 if (!FD) {
176 const DeclContext *DC = D->getDeclContext();
177 // Check for extern variable declared locally.
178 if (isa<FunctionDecl>(DC) && D->hasLinkage())
179 while (!DC->isNamespace() && !DC->isTranslationUnit())
180 DC = DC->getParent();
Sean Hunt31455252010-01-24 03:04:27 +0000181 if (DC->isTranslationUnit() &&
182 D->getLinkage() != NamedDecl::InternalLinkage)
Eli Friedman7facf842009-12-02 20:32:49 +0000183 return false;
184 }
185
186 // C functions and "main" are not mangled.
187 if ((FD && FD->isMain()) || isInCLinkageSpecification(D))
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000188 return false;
189
Anders Carlsson43f17402009-04-02 15:51:53 +0000190 return true;
191}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000192
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000193void CXXNameMangler::mangle(const NamedDecl *D, llvm::StringRef Prefix) {
Mike Stump141c5af2009-09-02 00:25:38 +0000194 // Any decl can be declared with __asm("foo") on it, and this takes precedence
195 // over all other naming in the .o file.
Argyrios Kyrtzidis40b598e2009-06-30 02:34:44 +0000196 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000197 // If we have an asm name, then we use it as the mangling.
198 Out << '\01'; // LLVM IR Marker for __asm("foo")
199 Out << ALA->getLabel();
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000200 return;
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000201 }
Mike Stump1eb44332009-09-09 15:08:12 +0000202
Sean Hunt31455252010-01-24 03:04:27 +0000203 // <mangled-name> ::= _Z <encoding>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000204 // ::= <data name>
205 // ::= <special-name>
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000206 Out << Prefix;
207 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
Daniel Dunbarf981bf82009-11-21 09:14:52 +0000208 mangleFunctionEncoding(FD);
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000209 else
210 mangleName(cast<VarDecl>(D));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000211}
212
213void CXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD) {
214 // <encoding> ::= <function name> <bare-function-type>
215 mangleName(FD);
Mike Stump1eb44332009-09-09 15:08:12 +0000216
Daniel Dunbar7e0c1952009-11-21 09:17:15 +0000217 // Don't mangle in the type if this isn't a decl we should typically mangle.
218 if (!Context.shouldMangleDeclName(FD))
219 return;
220
Mike Stump141c5af2009-09-02 00:25:38 +0000221 // Whether the mangling of a function type includes the return type depends on
222 // the context and the nature of the function. The rules for deciding whether
223 // the return type is included are:
Mike Stump1eb44332009-09-09 15:08:12 +0000224 //
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000225 // 1. Template functions (names or types) have return types encoded, with
226 // the exceptions listed below.
Mike Stump1eb44332009-09-09 15:08:12 +0000227 // 2. Function types not appearing as part of a function name mangling,
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000228 // e.g. parameters, pointer types, etc., have return type encoded, with the
229 // exceptions listed below.
230 // 3. Non-template function names do not have return types encoded.
231 //
Mike Stump141c5af2009-09-02 00:25:38 +0000232 // The exceptions mentioned in (1) and (2) above, for which the return type is
233 // never included, are
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000234 // 1. Constructors.
235 // 2. Destructors.
236 // 3. Conversion operator functions, e.g. operator int.
237 bool MangleReturnType = false;
Anders Carlsson9234b7f2009-09-17 03:46:43 +0000238 if (FunctionTemplateDecl *PrimaryTemplate = FD->getPrimaryTemplate()) {
239 if (!(isa<CXXConstructorDecl>(FD) || isa<CXXDestructorDecl>(FD) ||
240 isa<CXXConversionDecl>(FD)))
241 MangleReturnType = true;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000242
Anders Carlsson9234b7f2009-09-17 03:46:43 +0000243 // Mangle the type of the primary template.
244 FD = PrimaryTemplate->getTemplatedDecl();
245 }
246
John McCall54e14c42009-10-22 22:37:11 +0000247 // Do the canonicalization out here because parameter types can
248 // undergo additional canonicalization (e.g. array decay).
249 FunctionType *FT = cast<FunctionType>(Context.getASTContext()
250 .getCanonicalType(FD->getType()));
251
252 mangleBareFunctionType(FT, MangleReturnType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000253}
254
Anders Carlsson47846d22009-12-04 06:23:23 +0000255/// isStd - Return whether a given namespace is the 'std' namespace.
256static bool isStd(const NamespaceDecl *NS) {
John McCall9aeed322009-10-01 00:25:31 +0000257 const IdentifierInfo *II = NS->getOriginalNamespace()->getIdentifier();
258 return II && II->isStr("std");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000259}
260
Anders Carlsson47846d22009-12-04 06:23:23 +0000261static const DeclContext *IgnoreLinkageSpecDecls(const DeclContext *DC) {
262 while (isa<LinkageSpecDecl>(DC)) {
263 assert(cast<LinkageSpecDecl>(DC)->getLanguage() ==
264 LinkageSpecDecl::lang_cxx && "Unexpected linkage decl!");
265 DC = DC->getParent();
266 }
267
268 return DC;
269}
270
271// isStdNamespace - Return whether a given decl context is a toplevel 'std'
272// namespace.
Daniel Dunbar1308af92009-11-21 09:11:45 +0000273static bool isStdNamespace(const DeclContext *DC) {
Anders Carlsson47846d22009-12-04 06:23:23 +0000274 if (!DC->isNamespace())
275 return false;
276
277 if (!IgnoreLinkageSpecDecls(DC->getParent())->isTranslationUnit())
278 return false;
279
280 return isStd(cast<NamespaceDecl>(DC));
Daniel Dunbar1308af92009-11-21 09:11:45 +0000281}
282
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000283static const TemplateDecl *
284isTemplate(const NamedDecl *ND, const TemplateArgumentList *&TemplateArgs) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000285 // Check if we have a function template.
286 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)){
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000287 if (const TemplateDecl *TD = FD->getPrimaryTemplate()) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000288 TemplateArgs = FD->getTemplateSpecializationArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000289 return TD;
Anders Carlsson2744a062009-09-18 19:00:18 +0000290 }
291 }
292
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000293 // Check if we have a class template.
294 if (const ClassTemplateSpecializationDecl *Spec =
295 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
296 TemplateArgs = &Spec->getTemplateArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000297 return Spec->getSpecializedTemplate();
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000298 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000299
Anders Carlsson2744a062009-09-18 19:00:18 +0000300 return 0;
301}
302
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000303void CXXNameMangler::mangleName(const NamedDecl *ND) {
304 // <name> ::= <nested-name>
305 // ::= <unscoped-name>
306 // ::= <unscoped-template-name> <template-args>
Anders Carlsson201ce742009-09-17 03:17:01 +0000307 // ::= <local-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000308 //
Anders Carlssond58d6f72009-09-17 16:12:20 +0000309 const DeclContext *DC = ND->getDeclContext();
Eli Friedman7facf842009-12-02 20:32:49 +0000310
311 // If this is an extern variable declared locally, the relevant DeclContext
312 // is that of the containing namespace, or the translation unit.
313 if (isa<FunctionDecl>(DC) && ND->hasLinkage())
314 while (!DC->isNamespace() && !DC->isTranslationUnit())
315 DC = DC->getParent();
316
Anders Carlsson5cc58c62009-09-22 17:23:30 +0000317 while (isa<LinkageSpecDecl>(DC))
Anders Carlssond58d6f72009-09-17 16:12:20 +0000318 DC = DC->getParent();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000319
Anders Carlssond58d6f72009-09-17 16:12:20 +0000320 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000321 // Check if we have a template.
322 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000323 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000324 mangleUnscopedTemplateName(TD);
Anders Carlsson9e85c742009-12-23 19:30:55 +0000325 mangleTemplateArgs(*TemplateArgs);
Anders Carlsson2744a062009-09-18 19:00:18 +0000326 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000327 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000328
Anders Carlsson7482e242009-09-18 04:29:09 +0000329 mangleUnscopedName(ND);
330 return;
331 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000332
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000333 if (isa<FunctionDecl>(DC) || isa<ObjCMethodDecl>(DC)) {
Anders Carlsson7482e242009-09-18 04:29:09 +0000334 mangleLocalName(ND);
335 return;
336 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000337
Eli Friedman7facf842009-12-02 20:32:49 +0000338 mangleNestedName(ND, DC);
Anders Carlsson7482e242009-09-18 04:29:09 +0000339}
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000340void CXXNameMangler::mangleName(const TemplateDecl *TD,
Anders Carlsson7624f212009-09-18 02:42:01 +0000341 const TemplateArgument *TemplateArgs,
342 unsigned NumTemplateArgs) {
Anders Carlsson47846d22009-12-04 06:23:23 +0000343 const DeclContext *DC = IgnoreLinkageSpecDecls(TD->getDeclContext());
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000344
Anders Carlsson7624f212009-09-18 02:42:01 +0000345 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000346 mangleUnscopedTemplateName(TD);
Anders Carlsson7624f212009-09-18 02:42:01 +0000347 mangleTemplateArgs(TemplateArgs, NumTemplateArgs);
348 } else {
349 mangleNestedName(TD, TemplateArgs, NumTemplateArgs);
350 }
351}
352
Anders Carlsson201ce742009-09-17 03:17:01 +0000353void CXXNameMangler::mangleUnscopedName(const NamedDecl *ND) {
354 // <unscoped-name> ::= <unqualified-name>
355 // ::= St <unqualified-name> # ::std::
356 if (isStdNamespace(ND->getDeclContext()))
357 Out << "St";
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000358
Anders Carlsson201ce742009-09-17 03:17:01 +0000359 mangleUnqualifiedName(ND);
360}
361
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000362void CXXNameMangler::mangleUnscopedTemplateName(const TemplateDecl *ND) {
Anders Carlsson201ce742009-09-17 03:17:01 +0000363 // <unscoped-template-name> ::= <unscoped-name>
364 // ::= <substitution>
Anders Carlsson7624f212009-09-18 02:42:01 +0000365 if (mangleSubstitution(ND))
Anders Carlsson03c9d532009-09-17 04:02:31 +0000366 return;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000367
Anders Carlsson1668f202009-09-26 20:13:56 +0000368 mangleUnscopedName(ND->getTemplatedDecl());
Anders Carlsson7624f212009-09-18 02:42:01 +0000369 addSubstitution(ND);
Anders Carlsson201ce742009-09-17 03:17:01 +0000370}
371
Anders Carlssona94822e2009-11-26 02:32:05 +0000372void CXXNameMangler::mangleNumber(int64_t Number) {
373 // <number> ::= [n] <non-negative decimal integer>
374 if (Number < 0) {
375 Out << 'n';
376 Number = -Number;
377 }
378
379 Out << Number;
380}
381
Anders Carlssonb73a5be2009-11-26 02:49:32 +0000382void CXXNameMangler::mangleCallOffset(const ThunkAdjustment &Adjustment) {
Mike Stump141c5af2009-09-02 00:25:38 +0000383 // <call-offset> ::= h <nv-offset> _
384 // ::= v <v-offset> _
385 // <nv-offset> ::= <offset number> # non-virtual base override
Anders Carlssona94822e2009-11-26 02:32:05 +0000386 // <v-offset> ::= <offset number> _ <virtual offset number>
Mike Stump141c5af2009-09-02 00:25:38 +0000387 // # virtual base override, with vcall offset
Anders Carlssonb73a5be2009-11-26 02:49:32 +0000388 if (!Adjustment.Virtual) {
Anders Carlssona94822e2009-11-26 02:32:05 +0000389 Out << 'h';
Anders Carlssonb73a5be2009-11-26 02:49:32 +0000390 mangleNumber(Adjustment.NonVirtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000391 Out << '_';
392 return;
Mike Stump141c5af2009-09-02 00:25:38 +0000393 }
Anders Carlssona94822e2009-11-26 02:32:05 +0000394
395 Out << 'v';
Anders Carlssonb73a5be2009-11-26 02:49:32 +0000396 mangleNumber(Adjustment.NonVirtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000397 Out << '_';
Anders Carlssonb73a5be2009-11-26 02:49:32 +0000398 mangleNumber(Adjustment.Virtual);
Anders Carlssona94822e2009-11-26 02:32:05 +0000399 Out << '_';
Mike Stump9124bcc2009-09-02 00:56:18 +0000400}
401
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000402void CXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND) {
403 // <unqualified-name> ::= <operator-name>
Mike Stump1eb44332009-09-09 15:08:12 +0000404 // ::= <ctor-dtor-name>
405 // ::= <source-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000406 DeclarationName Name = ND->getDeclName();
407 switch (Name.getNameKind()) {
Anders Carlssonc4355b62009-10-07 01:45:02 +0000408 case DeclarationName::Identifier: {
409 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
John McCall9aeed322009-10-01 00:25:31 +0000410 if (NS->isAnonymousNamespace()) {
411 // This is how gcc mangles these names. It's apparently
412 // always '1', no matter how many different anonymous
413 // namespaces appear in a context.
414 Out << "12_GLOBAL__N_1";
415 break;
416 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000417 }
Anders Carlssonc4355b62009-10-07 01:45:02 +0000418
419 if (const IdentifierInfo *II = Name.getAsIdentifierInfo()) {
Sean Hunt31455252010-01-24 03:04:27 +0000420 // We must avoid conflicts between internally- and externally-
421 // linked names in the same TU. This naming convention is the
422 // same as that followed by GCC, though it shouldn't actually matter.
423 if (ND->getLinkage() == NamedDecl::InternalLinkage &&
424 ND->getDeclContext()->isFileContext())
425 Out << 'L';
426
Anders Carlssonc4355b62009-10-07 01:45:02 +0000427 mangleSourceName(II);
428 break;
429 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000430
Anders Carlssonc4355b62009-10-07 01:45:02 +0000431 // We must have an anonymous struct.
432 const TagDecl *TD = cast<TagDecl>(ND);
433 if (const TypedefDecl *D = TD->getTypedefForAnonDecl()) {
434 assert(TD->getDeclContext() == D->getDeclContext() &&
435 "Typedef should not be in another decl context!");
436 assert(D->getDeclName().getAsIdentifierInfo() &&
437 "Typedef was not named!");
438 mangleSourceName(D->getDeclName().getAsIdentifierInfo());
439 break;
440 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000441
Anders Carlssonc4355b62009-10-07 01:45:02 +0000442 // Get a unique id for the anonymous struct.
443 uint64_t AnonStructId = Context.getAnonymousStructId(TD);
444
445 // Mangle it as a source name in the form
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000446 // [n] $_<id>
Anders Carlssonc4355b62009-10-07 01:45:02 +0000447 // where n is the length of the string.
448 llvm::SmallString<8> Str;
449 Str += "$_";
450 Str += llvm::utostr(AnonStructId);
451
452 Out << Str.size();
453 Out << Str.str();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000454 break;
Anders Carlssonc4355b62009-10-07 01:45:02 +0000455 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000456
457 case DeclarationName::ObjCZeroArgSelector:
458 case DeclarationName::ObjCOneArgSelector:
459 case DeclarationName::ObjCMultiArgSelector:
460 assert(false && "Can't mangle Objective-C selector names here!");
461 break;
462
463 case DeclarationName::CXXConstructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000464 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000465 // If the named decl is the C++ constructor we're mangling, use the type
466 // we were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000467 mangleCXXCtorType(static_cast<CXXCtorType>(StructorType));
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000468 else
469 // Otherwise, use the complete constructor name. This is relevant if a
470 // class with a constructor is declared within a constructor.
471 mangleCXXCtorType(Ctor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000472 break;
473
474 case DeclarationName::CXXDestructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000475 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000476 // If the named decl is the C++ destructor we're mangling, use the type we
477 // were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000478 mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
479 else
480 // Otherwise, use the complete destructor name. This is relevant if a
481 // class with a destructor is declared within a destructor.
482 mangleCXXDtorType(Dtor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000483 break;
484
485 case DeclarationName::CXXConversionFunctionName:
Mike Stump1eb44332009-09-09 15:08:12 +0000486 // <operator-name> ::= cv <type> # (cast)
Douglas Gregor219cc612009-02-13 01:28:03 +0000487 Out << "cv";
Anders Carlssonb5404912009-10-07 01:06:45 +0000488 mangleType(Context.getASTContext().getCanonicalType(Name.getCXXNameType()));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000489 break;
490
Anders Carlsson8257d412009-12-22 06:36:32 +0000491 case DeclarationName::CXXOperatorName: {
492 unsigned Arity = cast<FunctionDecl>(ND)->getNumParams();
493
494 // If we have a C++ member function, we need to include the 'this' pointer.
495 // FIXME: This does not make sense for operators that are static, but their
496 // names stay the same regardless of the arity (operator new for instance).
497 if (isa<CXXMethodDecl>(ND))
498 Arity++;
499 mangleOperatorName(Name.getCXXOverloadedOperator(), Arity);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000500 break;
Anders Carlsson8257d412009-12-22 06:36:32 +0000501 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000502
Sean Hunt3e518bd2009-11-29 07:34:05 +0000503 case DeclarationName::CXXLiteralOperatorName:
Sean Hunt5dd6b392009-12-04 21:11:13 +0000504 // FIXME: This mangling is not yet official.
Sean Hunt2421f662009-12-04 21:01:37 +0000505 Out << "li";
Sean Hunt3e518bd2009-11-29 07:34:05 +0000506 mangleSourceName(Name.getCXXLiteralIdentifier());
507 break;
508
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000509 case DeclarationName::CXXUsingDirective:
510 assert(false && "Can't mangle a using directive name!");
Douglas Gregor219cc612009-02-13 01:28:03 +0000511 break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000512 }
513}
514
515void CXXNameMangler::mangleSourceName(const IdentifierInfo *II) {
516 // <source-name> ::= <positive length number> <identifier>
517 // <number> ::= [n] <non-negative decimal integer>
518 // <identifier> ::= <unqualified source code identifier>
519 Out << II->getLength() << II->getName();
520}
521
Eli Friedman7facf842009-12-02 20:32:49 +0000522void CXXNameMangler::mangleNestedName(const NamedDecl *ND,
523 const DeclContext *DC) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000524 // <nested-name> ::= N [<CV-qualifiers>] <prefix> <unqualified-name> E
525 // ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
Anders Carlssond99edc42009-09-26 03:55:37 +0000526
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000527 Out << 'N';
528 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(ND))
John McCall0953e762009-09-24 19:53:00 +0000529 mangleQualifiers(Qualifiers::fromCVRMask(Method->getTypeQualifiers()));
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000530
Anders Carlsson2744a062009-09-18 19:00:18 +0000531 // Check if we have a template.
532 const TemplateArgumentList *TemplateArgs = 0;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000533 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000534 mangleTemplatePrefix(TD);
Anders Carlsson9e85c742009-12-23 19:30:55 +0000535 mangleTemplateArgs(*TemplateArgs);
Anders Carlsson7482e242009-09-18 04:29:09 +0000536 } else {
Eli Friedman7facf842009-12-02 20:32:49 +0000537 manglePrefix(DC);
Anders Carlsson7482e242009-09-18 04:29:09 +0000538 mangleUnqualifiedName(ND);
539 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000540
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000541 Out << 'E';
542}
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000543void CXXNameMangler::mangleNestedName(const TemplateDecl *TD,
Anders Carlsson7624f212009-09-18 02:42:01 +0000544 const TemplateArgument *TemplateArgs,
545 unsigned NumTemplateArgs) {
Anders Carlssone45117b2009-09-27 19:53:49 +0000546 // <nested-name> ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
547
Anders Carlsson7624f212009-09-18 02:42:01 +0000548 Out << 'N';
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000549
Anders Carlssone45117b2009-09-27 19:53:49 +0000550 mangleTemplatePrefix(TD);
Anders Carlsson7624f212009-09-18 02:42:01 +0000551 mangleTemplateArgs(TemplateArgs, NumTemplateArgs);
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000552
Anders Carlsson7624f212009-09-18 02:42:01 +0000553 Out << 'E';
554}
555
Anders Carlsson1b42c792009-04-02 16:24:45 +0000556void CXXNameMangler::mangleLocalName(const NamedDecl *ND) {
557 // <local-name> := Z <function encoding> E <entity name> [<discriminator>]
558 // := Z <function encoding> E s [<discriminator>]
Mike Stump1eb44332009-09-09 15:08:12 +0000559 // <discriminator> := _ <non-negative number>
Anders Carlsson1b42c792009-04-02 16:24:45 +0000560 Out << 'Z';
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000561
562 if (const ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(ND->getDeclContext()))
563 mangleObjCMethodName(MD);
564 else
565 mangleFunctionEncoding(cast<FunctionDecl>(ND->getDeclContext()));
566
Anders Carlsson1b42c792009-04-02 16:24:45 +0000567 Out << 'E';
Eli Friedman6f9f25d2009-12-11 20:21:38 +0000568 mangleUnqualifiedName(ND);
Anders Carlsson1b42c792009-04-02 16:24:45 +0000569}
570
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000571void CXXNameMangler::manglePrefix(const DeclContext *DC) {
572 // <prefix> ::= <prefix> <unqualified-name>
573 // ::= <template-prefix> <template-args>
574 // ::= <template-param>
575 // ::= # empty
576 // ::= <substitution>
Anders Carlsson6862fc72009-09-17 04:16:28 +0000577
Anders Carlssonadd28822009-09-22 20:33:31 +0000578 while (isa<LinkageSpecDecl>(DC))
579 DC = DC->getParent();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000580
Anders Carlsson9263e912009-09-18 18:39:58 +0000581 if (DC->isTranslationUnit())
582 return;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000583
Anders Carlsson6862fc72009-09-17 04:16:28 +0000584 if (mangleSubstitution(cast<NamedDecl>(DC)))
585 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000586
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000587 // Check if we have a template.
588 const TemplateArgumentList *TemplateArgs = 0;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000589 if (const TemplateDecl *TD = isTemplate(cast<NamedDecl>(DC), TemplateArgs)) {
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000590 mangleTemplatePrefix(TD);
Anders Carlsson9e85c742009-12-23 19:30:55 +0000591 mangleTemplateArgs(*TemplateArgs);
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000592 } else {
593 manglePrefix(DC->getParent());
594 mangleUnqualifiedName(cast<NamedDecl>(DC));
595 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000596
Anders Carlsson6862fc72009-09-17 04:16:28 +0000597 addSubstitution(cast<NamedDecl>(DC));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000598}
599
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000600void CXXNameMangler::mangleTemplatePrefix(const TemplateDecl *ND) {
Anders Carlsson7482e242009-09-18 04:29:09 +0000601 // <template-prefix> ::= <prefix> <template unqualified-name>
602 // ::= <template-param>
603 // ::= <substitution>
604
Anders Carlssonaeb85372009-09-26 22:18:22 +0000605 if (mangleSubstitution(ND))
606 return;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000607
Anders Carlssonaeb85372009-09-26 22:18:22 +0000608 // FIXME: <template-param>
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000609
Anders Carlssonaa73ab12009-09-18 18:47:07 +0000610 manglePrefix(ND->getDeclContext());
Anders Carlsson1668f202009-09-26 20:13:56 +0000611 mangleUnqualifiedName(ND->getTemplatedDecl());
Daniel Dunbar3c9e4632009-11-21 09:05:47 +0000612
Anders Carlssonaeb85372009-09-26 22:18:22 +0000613 addSubstitution(ND);
Anders Carlsson7482e242009-09-18 04:29:09 +0000614}
615
Mike Stump1eb44332009-09-09 15:08:12 +0000616void
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000617CXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity) {
618 switch (OO) {
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000619 // <operator-name> ::= nw # new
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000620 case OO_New: Out << "nw"; break;
621 // ::= na # new[]
622 case OO_Array_New: Out << "na"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000623 // ::= dl # delete
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000624 case OO_Delete: Out << "dl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000625 // ::= da # delete[]
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000626 case OO_Array_Delete: Out << "da"; break;
627 // ::= ps # + (unary)
628 // ::= pl # +
Anders Carlsson8257d412009-12-22 06:36:32 +0000629 case OO_Plus:
630 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
631 Out << (Arity == 1? "ps" : "pl"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000632 // ::= ng # - (unary)
633 // ::= mi # -
Anders Carlsson8257d412009-12-22 06:36:32 +0000634 case OO_Minus:
635 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
636 Out << (Arity == 1? "ng" : "mi"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000637 // ::= ad # & (unary)
638 // ::= an # &
Anders Carlsson8257d412009-12-22 06:36:32 +0000639 case OO_Amp:
640 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
641 Out << (Arity == 1? "ad" : "an"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000642 // ::= de # * (unary)
643 // ::= ml # *
Anders Carlsson8257d412009-12-22 06:36:32 +0000644 case OO_Star:
645 assert((Arity == 1 || Arity == 2) && "Invalid arity!");
646 Out << (Arity == 1? "de" : "ml"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000647 // ::= co # ~
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000648 case OO_Tilde: Out << "co"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000649 // ::= dv # /
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000650 case OO_Slash: Out << "dv"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000651 // ::= rm # %
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000652 case OO_Percent: Out << "rm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000653 // ::= or # |
654 case OO_Pipe: Out << "or"; break;
655 // ::= eo # ^
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000656 case OO_Caret: Out << "eo"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000657 // ::= aS # =
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000658 case OO_Equal: Out << "aS"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000659 // ::= pL # +=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000660 case OO_PlusEqual: Out << "pL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000661 // ::= mI # -=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000662 case OO_MinusEqual: Out << "mI"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000663 // ::= mL # *=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000664 case OO_StarEqual: Out << "mL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000665 // ::= dV # /=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000666 case OO_SlashEqual: Out << "dV"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000667 // ::= rM # %=
668 case OO_PercentEqual: Out << "rM"; break;
669 // ::= aN # &=
670 case OO_AmpEqual: Out << "aN"; break;
671 // ::= oR # |=
672 case OO_PipeEqual: Out << "oR"; break;
673 // ::= eO # ^=
674 case OO_CaretEqual: Out << "eO"; break;
675 // ::= ls # <<
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000676 case OO_LessLess: Out << "ls"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000677 // ::= rs # >>
678 case OO_GreaterGreater: Out << "rs"; break;
679 // ::= lS # <<=
680 case OO_LessLessEqual: Out << "lS"; break;
681 // ::= rS # >>=
682 case OO_GreaterGreaterEqual: Out << "rS"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000683 // ::= eq # ==
684 case OO_EqualEqual: Out << "eq"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000685 // ::= ne # !=
686 case OO_ExclaimEqual: Out << "ne"; break;
687 // ::= lt # <
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000688 case OO_Less: Out << "lt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000689 // ::= gt # >
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000690 case OO_Greater: Out << "gt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000691 // ::= le # <=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000692 case OO_LessEqual: Out << "le"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000693 // ::= ge # >=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000694 case OO_GreaterEqual: Out << "ge"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000695 // ::= nt # !
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000696 case OO_Exclaim: Out << "nt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000697 // ::= aa # &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000698 case OO_AmpAmp: Out << "aa"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000699 // ::= oo # ||
700 case OO_PipePipe: Out << "oo"; break;
701 // ::= pp # ++
702 case OO_PlusPlus: Out << "pp"; break;
703 // ::= mm # --
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000704 case OO_MinusMinus: Out << "mm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000705 // ::= cm # ,
706 case OO_Comma: Out << "cm"; break;
707 // ::= pm # ->*
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000708 case OO_ArrowStar: Out << "pm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000709 // ::= pt # ->
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000710 case OO_Arrow: Out << "pt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000711 // ::= cl # ()
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000712 case OO_Call: Out << "cl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000713 // ::= ix # []
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000714 case OO_Subscript: Out << "ix"; break;
Anders Carlssone170ba72009-12-14 01:45:37 +0000715
716 // ::= qu # ?
717 // The conditional operator can't be overloaded, but we still handle it when
718 // mangling expressions.
719 case OO_Conditional: Out << "qu"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000720
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000721 case OO_None:
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000722 case NUM_OVERLOADED_OPERATORS:
Mike Stump1eb44332009-09-09 15:08:12 +0000723 assert(false && "Not an overloaded operator");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000724 break;
725 }
726}
727
John McCall0953e762009-09-24 19:53:00 +0000728void CXXNameMangler::mangleQualifiers(Qualifiers Quals) {
Mike Stump1eb44332009-09-09 15:08:12 +0000729 // <CV-qualifiers> ::= [r] [V] [K] # restrict (C99), volatile, const
John McCall0953e762009-09-24 19:53:00 +0000730 if (Quals.hasRestrict())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000731 Out << 'r';
John McCall0953e762009-09-24 19:53:00 +0000732 if (Quals.hasVolatile())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000733 Out << 'V';
John McCall0953e762009-09-24 19:53:00 +0000734 if (Quals.hasConst())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000735 Out << 'K';
John McCall0953e762009-09-24 19:53:00 +0000736
737 // FIXME: For now, just drop all extension qualifiers on the floor.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000738}
739
Anders Carlsson7b06f6c2009-12-10 03:14:39 +0000740void CXXNameMangler::mangleObjCMethodName(const ObjCMethodDecl *MD) {
741 llvm::SmallString<64> Name;
742 llvm::raw_svector_ostream OS(Name);
743
744 const ObjCContainerDecl *CD =
745 dyn_cast<ObjCContainerDecl>(MD->getDeclContext());
746 assert (CD && "Missing container decl in GetNameForMethod");
747 OS << (MD->isInstanceMethod() ? '-' : '+') << '[' << CD->getName();
748 if (const ObjCCategoryImplDecl *CID = dyn_cast<ObjCCategoryImplDecl>(CD))
749 OS << '(' << CID->getNameAsString() << ')';
750 OS << ' ' << MD->getSelector().getAsString() << ']';
751
752 Out << OS.str().size() << OS.str();
753}
754
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000755void CXXNameMangler::mangleType(QualType T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000756 // Only operate on the canonical type!
Anders Carlssonb5404912009-10-07 01:06:45 +0000757 T = Context.getASTContext().getCanonicalType(T);
Anders Carlsson4843e582009-03-10 17:07:44 +0000758
Douglas Gregora4923eb2009-11-16 21:35:15 +0000759 bool IsSubstitutable = T.hasLocalQualifiers() || !isa<BuiltinType>(T);
Anders Carlsson76967372009-09-17 00:43:46 +0000760 if (IsSubstitutable && mangleSubstitution(T))
761 return;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000762
Douglas Gregora4923eb2009-11-16 21:35:15 +0000763 if (Qualifiers Quals = T.getLocalQualifiers()) {
John McCall0953e762009-09-24 19:53:00 +0000764 mangleQualifiers(Quals);
765 // Recurse: even if the qualified type isn't yet substitutable,
766 // the unqualified type might be.
Douglas Gregora4923eb2009-11-16 21:35:15 +0000767 mangleType(T.getLocalUnqualifiedType());
Anders Carlsson76967372009-09-17 00:43:46 +0000768 } else {
769 switch (T->getTypeClass()) {
John McCallefe6aee2009-09-05 07:56:18 +0000770#define ABSTRACT_TYPE(CLASS, PARENT)
771#define NON_CANONICAL_TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000772 case Type::CLASS: \
Jeffrey Yasskin9f61aa92009-12-12 05:05:38 +0000773 llvm_unreachable("can't mangle non-canonical type " #CLASS "Type"); \
Anders Carlsson76967372009-09-17 00:43:46 +0000774 return;
John McCallefe6aee2009-09-05 07:56:18 +0000775#define TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000776 case Type::CLASS: \
John McCall0953e762009-09-24 19:53:00 +0000777 mangleType(static_cast<const CLASS##Type*>(T.getTypePtr())); \
Anders Carlsson76967372009-09-17 00:43:46 +0000778 break;
John McCallefe6aee2009-09-05 07:56:18 +0000779#include "clang/AST/TypeNodes.def"
Anders Carlsson76967372009-09-17 00:43:46 +0000780 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000781 }
Anders Carlsson76967372009-09-17 00:43:46 +0000782
783 // Add the substitution.
784 if (IsSubstitutable)
785 addSubstitution(T);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000786}
787
788void CXXNameMangler::mangleType(const BuiltinType *T) {
John McCallefe6aee2009-09-05 07:56:18 +0000789 // <type> ::= <builtin-type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000790 // <builtin-type> ::= v # void
791 // ::= w # wchar_t
792 // ::= b # bool
793 // ::= c # char
794 // ::= a # signed char
795 // ::= h # unsigned char
796 // ::= s # short
797 // ::= t # unsigned short
798 // ::= i # int
799 // ::= j # unsigned int
800 // ::= l # long
801 // ::= m # unsigned long
802 // ::= x # long long, __int64
803 // ::= y # unsigned long long, __int64
804 // ::= n # __int128
805 // UNSUPPORTED: ::= o # unsigned __int128
806 // ::= f # float
807 // ::= d # double
808 // ::= e # long double, __float80
809 // UNSUPPORTED: ::= g # __float128
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000810 // UNSUPPORTED: ::= Dd # IEEE 754r decimal floating point (64 bits)
811 // UNSUPPORTED: ::= De # IEEE 754r decimal floating point (128 bits)
812 // UNSUPPORTED: ::= Df # IEEE 754r decimal floating point (32 bits)
813 // UNSUPPORTED: ::= Dh # IEEE 754r half-precision floating point (16 bits)
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000814 // ::= Di # char32_t
815 // ::= Ds # char16_t
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000816 // ::= u <source-name> # vendor extended type
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000817 // From our point of view, std::nullptr_t is a builtin, but as far as mangling
818 // is concerned, it's a type called std::nullptr_t.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000819 switch (T->getKind()) {
820 case BuiltinType::Void: Out << 'v'; break;
821 case BuiltinType::Bool: Out << 'b'; break;
822 case BuiltinType::Char_U: case BuiltinType::Char_S: Out << 'c'; break;
823 case BuiltinType::UChar: Out << 'h'; break;
824 case BuiltinType::UShort: Out << 't'; break;
825 case BuiltinType::UInt: Out << 'j'; break;
826 case BuiltinType::ULong: Out << 'm'; break;
827 case BuiltinType::ULongLong: Out << 'y'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000828 case BuiltinType::UInt128: Out << 'o'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000829 case BuiltinType::SChar: Out << 'a'; break;
830 case BuiltinType::WChar: Out << 'w'; break;
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000831 case BuiltinType::Char16: Out << "Ds"; break;
832 case BuiltinType::Char32: Out << "Di"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000833 case BuiltinType::Short: Out << 's'; break;
834 case BuiltinType::Int: Out << 'i'; break;
835 case BuiltinType::Long: Out << 'l'; break;
836 case BuiltinType::LongLong: Out << 'x'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000837 case BuiltinType::Int128: Out << 'n'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000838 case BuiltinType::Float: Out << 'f'; break;
839 case BuiltinType::Double: Out << 'd'; break;
840 case BuiltinType::LongDouble: Out << 'e'; break;
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000841 case BuiltinType::NullPtr: Out << "St9nullptr_t"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000842
843 case BuiltinType::Overload:
844 case BuiltinType::Dependent:
Mike Stump1eb44332009-09-09 15:08:12 +0000845 assert(false &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000846 "Overloaded and dependent types shouldn't get to name mangling");
847 break;
Anders Carlssone89d1592009-06-26 18:41:36 +0000848 case BuiltinType::UndeducedAuto:
849 assert(0 && "Should not see undeduced auto here");
850 break;
Steve Naroff9533a7f2009-07-22 17:14:51 +0000851 case BuiltinType::ObjCId: Out << "11objc_object"; break;
852 case BuiltinType::ObjCClass: Out << "10objc_class"; break;
Fariborz Jahanian13dcd002009-11-21 19:53:08 +0000853 case BuiltinType::ObjCSel: Out << "13objc_selector"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000854 }
855}
856
John McCallefe6aee2009-09-05 07:56:18 +0000857// <type> ::= <function-type>
858// <function-type> ::= F [Y] <bare-function-type> E
859void CXXNameMangler::mangleType(const FunctionProtoType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000860 Out << 'F';
Mike Stumpf5408fe2009-05-16 07:57:57 +0000861 // FIXME: We don't have enough information in the AST to produce the 'Y'
862 // encoding for extern "C" function types.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000863 mangleBareFunctionType(T, /*MangleReturnType=*/true);
864 Out << 'E';
865}
John McCallefe6aee2009-09-05 07:56:18 +0000866void CXXNameMangler::mangleType(const FunctionNoProtoType *T) {
Jeffrey Yasskin9f61aa92009-12-12 05:05:38 +0000867 llvm_unreachable("Can't mangle K&R function prototypes");
John McCallefe6aee2009-09-05 07:56:18 +0000868}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000869void CXXNameMangler::mangleBareFunctionType(const FunctionType *T,
870 bool MangleReturnType) {
John McCallefe6aee2009-09-05 07:56:18 +0000871 // We should never be mangling something without a prototype.
872 const FunctionProtoType *Proto = cast<FunctionProtoType>(T);
873
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000874 // <bare-function-type> ::= <signature type>+
875 if (MangleReturnType)
John McCallefe6aee2009-09-05 07:56:18 +0000876 mangleType(Proto->getResultType());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000877
Anders Carlssonc6c91bc2009-04-01 00:15:23 +0000878 if (Proto->getNumArgs() == 0) {
879 Out << 'v';
880 return;
881 }
Mike Stump1eb44332009-09-09 15:08:12 +0000882
Douglas Gregor72564e72009-02-26 23:50:07 +0000883 for (FunctionProtoType::arg_type_iterator Arg = Proto->arg_type_begin(),
Mike Stump1eb44332009-09-09 15:08:12 +0000884 ArgEnd = Proto->arg_type_end();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000885 Arg != ArgEnd; ++Arg)
886 mangleType(*Arg);
Douglas Gregor219cc612009-02-13 01:28:03 +0000887
888 // <builtin-type> ::= z # ellipsis
889 if (Proto->isVariadic())
890 Out << 'z';
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000891}
892
John McCallefe6aee2009-09-05 07:56:18 +0000893// <type> ::= <class-enum-type>
Mike Stump1eb44332009-09-09 15:08:12 +0000894// <class-enum-type> ::= <name>
John McCalled976492009-12-04 22:46:56 +0000895void CXXNameMangler::mangleType(const UnresolvedUsingType *T) {
896 mangleName(T->getDecl());
897}
898
899// <type> ::= <class-enum-type>
900// <class-enum-type> ::= <name>
John McCallefe6aee2009-09-05 07:56:18 +0000901void CXXNameMangler::mangleType(const EnumType *T) {
902 mangleType(static_cast<const TagType*>(T));
903}
904void CXXNameMangler::mangleType(const RecordType *T) {
905 mangleType(static_cast<const TagType*>(T));
906}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000907void CXXNameMangler::mangleType(const TagType *T) {
Eli Friedmanecb7e932009-12-11 18:00:57 +0000908 mangleName(T->getDecl());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000909}
910
John McCallefe6aee2009-09-05 07:56:18 +0000911// <type> ::= <array-type>
912// <array-type> ::= A <positive dimension number> _ <element type>
913// ::= A [<dimension expression>] _ <element type>
914void CXXNameMangler::mangleType(const ConstantArrayType *T) {
915 Out << 'A' << T->getSize() << '_';
916 mangleType(T->getElementType());
917}
918void CXXNameMangler::mangleType(const VariableArrayType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000919 Out << 'A';
John McCallefe6aee2009-09-05 07:56:18 +0000920 mangleExpression(T->getSizeExpr());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000921 Out << '_';
922 mangleType(T->getElementType());
923}
John McCallefe6aee2009-09-05 07:56:18 +0000924void CXXNameMangler::mangleType(const DependentSizedArrayType *T) {
925 Out << 'A';
926 mangleExpression(T->getSizeExpr());
927 Out << '_';
928 mangleType(T->getElementType());
929}
930void CXXNameMangler::mangleType(const IncompleteArrayType *T) {
931 Out << 'A' << '_';
932 mangleType(T->getElementType());
933}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000934
John McCallefe6aee2009-09-05 07:56:18 +0000935// <type> ::= <pointer-to-member-type>
936// <pointer-to-member-type> ::= M <class type> <member type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000937void CXXNameMangler::mangleType(const MemberPointerType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000938 Out << 'M';
939 mangleType(QualType(T->getClass(), 0));
Anders Carlsson0e650012009-05-17 17:41:20 +0000940 QualType PointeeType = T->getPointeeType();
941 if (const FunctionProtoType *FPT = dyn_cast<FunctionProtoType>(PointeeType)) {
John McCall0953e762009-09-24 19:53:00 +0000942 mangleQualifiers(Qualifiers::fromCVRMask(FPT->getTypeQuals()));
Anders Carlsson0e650012009-05-17 17:41:20 +0000943 mangleType(FPT);
Mike Stump1eb44332009-09-09 15:08:12 +0000944 } else
Anders Carlsson0e650012009-05-17 17:41:20 +0000945 mangleType(PointeeType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000946}
947
John McCallefe6aee2009-09-05 07:56:18 +0000948// <type> ::= <template-param>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000949void CXXNameMangler::mangleType(const TemplateTypeParmType *T) {
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +0000950 mangleTemplateParameter(T->getIndex());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000951}
952
John McCallefe6aee2009-09-05 07:56:18 +0000953// FIXME: <type> ::= <template-template-param> <template-args>
John McCallefe6aee2009-09-05 07:56:18 +0000954
955// <type> ::= P <type> # pointer-to
956void CXXNameMangler::mangleType(const PointerType *T) {
957 Out << 'P';
958 mangleType(T->getPointeeType());
959}
960void CXXNameMangler::mangleType(const ObjCObjectPointerType *T) {
961 Out << 'P';
962 mangleType(T->getPointeeType());
963}
964
965// <type> ::= R <type> # reference-to
966void CXXNameMangler::mangleType(const LValueReferenceType *T) {
967 Out << 'R';
968 mangleType(T->getPointeeType());
969}
970
971// <type> ::= O <type> # rvalue reference-to (C++0x)
972void CXXNameMangler::mangleType(const RValueReferenceType *T) {
973 Out << 'O';
974 mangleType(T->getPointeeType());
975}
976
977// <type> ::= C <type> # complex pair (C 2000)
978void CXXNameMangler::mangleType(const ComplexType *T) {
979 Out << 'C';
980 mangleType(T->getElementType());
981}
982
983// GNU extension: vector types
984void CXXNameMangler::mangleType(const VectorType *T) {
985 Out << "U8__vector";
986 mangleType(T->getElementType());
987}
988void CXXNameMangler::mangleType(const ExtVectorType *T) {
989 mangleType(static_cast<const VectorType*>(T));
990}
991void CXXNameMangler::mangleType(const DependentSizedExtVectorType *T) {
992 Out << "U8__vector";
993 mangleType(T->getElementType());
994}
995
Anders Carlssona40c5e42009-03-07 22:03:21 +0000996void CXXNameMangler::mangleType(const ObjCInterfaceType *T) {
997 mangleSourceName(T->getDecl()->getIdentifier());
998}
999
John McCallefe6aee2009-09-05 07:56:18 +00001000void CXXNameMangler::mangleType(const BlockPointerType *T) {
Anders Carlssonf28c6872009-12-23 22:31:44 +00001001 Out << "U13block_pointer";
1002 mangleType(T->getPointeeType());
John McCallefe6aee2009-09-05 07:56:18 +00001003}
1004
John McCallefe6aee2009-09-05 07:56:18 +00001005void CXXNameMangler::mangleType(const TemplateSpecializationType *T) {
Anders Carlsson7624f212009-09-18 02:42:01 +00001006 TemplateDecl *TD = T->getTemplateName().getAsTemplateDecl();
1007 assert(TD && "FIXME: Support dependent template names!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001008
Anders Carlsson7624f212009-09-18 02:42:01 +00001009 mangleName(TD, T->getArgs(), T->getNumArgs());
John McCallefe6aee2009-09-05 07:56:18 +00001010}
1011
1012void CXXNameMangler::mangleType(const TypenameType *T) {
Anders Carlssonae352482009-09-26 02:26:02 +00001013 // Typename types are always nested
1014 Out << 'N';
1015
1016 const Type *QTy = T->getQualifier()->getAsType();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001017 if (const TemplateSpecializationType *TST =
Anders Carlssonae352482009-09-26 02:26:02 +00001018 dyn_cast<TemplateSpecializationType>(QTy)) {
Anders Carlsson88599172009-09-27 01:06:07 +00001019 if (!mangleSubstitution(QualType(TST, 0))) {
1020 TemplateDecl *TD = TST->getTemplateName().getAsTemplateDecl();
Anders Carlssonae352482009-09-26 02:26:02 +00001021
Anders Carlsson88599172009-09-27 01:06:07 +00001022 mangleTemplatePrefix(TD);
1023 mangleTemplateArgs(TST->getArgs(), TST->getNumArgs());
1024 addSubstitution(QualType(TST, 0));
1025 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001026 } else if (const TemplateTypeParmType *TTPT =
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001027 dyn_cast<TemplateTypeParmType>(QTy)) {
1028 // We use the QualType mangle type variant here because it handles
1029 // substitutions.
1030 mangleType(QualType(TTPT, 0));
Anders Carlssonae352482009-09-26 02:26:02 +00001031 } else
1032 assert(false && "Unhandled type!");
1033
1034 mangleSourceName(T->getIdentifier());
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001035
Anders Carlssonae352482009-09-26 02:26:02 +00001036 Out << 'E';
John McCallefe6aee2009-09-05 07:56:18 +00001037}
1038
Anders Carlssone170ba72009-12-14 01:45:37 +00001039void CXXNameMangler::mangleIntegerLiteral(QualType T,
1040 const llvm::APSInt &Value) {
1041 // <expr-primary> ::= L <type> <value number> E # integer literal
1042 Out << 'L';
1043
1044 mangleType(T);
1045 if (T->isBooleanType()) {
1046 // Boolean values are encoded as 0/1.
1047 Out << (Value.getBoolValue() ? '1' : '0');
1048 } else {
1049 if (Value.isNegative())
1050 Out << 'n';
1051 Value.abs().print(Out, false);
1052 }
1053 Out << 'E';
1054
1055}
1056
Anders Carlssond553f8c2009-09-21 01:21:10 +00001057void CXXNameMangler::mangleExpression(const Expr *E) {
1058 // <expression> ::= <unary operator-name> <expression>
John McCall09cc1412010-02-03 00:55:45 +00001059 // ::= <binary operator-name> <expression> <expression>
1060 // ::= <trinary operator-name> <expression> <expression> <expression>
1061 // ::= cl <expression>* E # call
Anders Carlssond553f8c2009-09-21 01:21:10 +00001062 // ::= cv <type> expression # conversion with one argument
1063 // ::= cv <type> _ <expression>* E # conversion with a different number of arguments
John McCall09cc1412010-02-03 00:55:45 +00001064 // ::= st <type> # sizeof (a type)
Anders Carlssond553f8c2009-09-21 01:21:10 +00001065 // ::= at <type> # alignof (a type)
1066 // ::= <template-param>
1067 // ::= <function-param>
1068 // ::= sr <type> <unqualified-name> # dependent name
1069 // ::= sr <type> <unqualified-name> <template-args> # dependent template-id
1070 // ::= sZ <template-param> # size of a parameter pack
John McCall09cc1412010-02-03 00:55:45 +00001071 // ::= <expr-primary>
Anders Carlssond553f8c2009-09-21 01:21:10 +00001072 switch (E->getStmtClass()) {
John McCall09cc1412010-02-03 00:55:45 +00001073 default:
1074 llvm_unreachable("unexpected statement kind");
1075 break;
1076
1077 case Expr::CallExprClass:
1078 // FIXME: implement
1079 llvm_unreachable("mangling not implemented for calls");
1080 break;
1081
1082 case Expr::CXXConstructExprClass:
1083 // FIXME: implement
1084 llvm_unreachable("mangling not implemented for construct exprs");
1085 break;
1086
1087 case Expr::SizeOfAlignOfExprClass:
1088 // FIXME: implement
1089 llvm_unreachable("mangling not implemented for sizeof/alignof");
1090 break;
Anders Carlssona7694082009-11-06 02:50:19 +00001091
Anders Carlssone170ba72009-12-14 01:45:37 +00001092 case Expr::UnaryOperatorClass: {
1093 const UnaryOperator *UO = cast<UnaryOperator>(E);
1094 mangleOperatorName(UnaryOperator::getOverloadedOperator(UO->getOpcode()),
1095 /*Arity=*/1);
1096 mangleExpression(UO->getSubExpr());
1097 break;
1098 }
1099
1100 case Expr::BinaryOperatorClass: {
1101 const BinaryOperator *BO = cast<BinaryOperator>(E);
1102 mangleOperatorName(BinaryOperator::getOverloadedOperator(BO->getOpcode()),
1103 /*Arity=*/2);
1104 mangleExpression(BO->getLHS());
1105 mangleExpression(BO->getRHS());
1106 break;
1107 }
1108
1109 case Expr::ConditionalOperatorClass: {
1110 const ConditionalOperator *CO = cast<ConditionalOperator>(E);
1111 mangleOperatorName(OO_Conditional, /*Arity=*/3);
1112 mangleExpression(CO->getCond());
1113 mangleExpression(CO->getLHS());
1114 mangleExpression(CO->getRHS());
1115 break;
1116 }
1117
Douglas Gregor46287c72010-01-29 16:37:09 +00001118 case Expr::ImplicitCastExprClass: {
1119 mangleExpression(cast<ImplicitCastExpr>(E)->getSubExpr());
1120 break;
1121 }
1122
1123 case Expr::CStyleCastExprClass:
1124 case Expr::CXXStaticCastExprClass:
1125 case Expr::CXXDynamicCastExprClass:
1126 case Expr::CXXReinterpretCastExprClass:
1127 case Expr::CXXConstCastExprClass:
1128 case Expr::CXXFunctionalCastExprClass: {
1129 const ExplicitCastExpr *ECE = cast<ExplicitCastExpr>(E);
1130 Out << "cv";
1131 mangleType(ECE->getType());
1132 mangleExpression(ECE->getSubExpr());
1133 break;
1134 }
1135
Anders Carlsson58040a52009-12-16 05:48:46 +00001136 case Expr::CXXOperatorCallExprClass: {
1137 const CXXOperatorCallExpr *CE = cast<CXXOperatorCallExpr>(E);
1138 unsigned NumArgs = CE->getNumArgs();
1139 mangleOperatorName(CE->getOperator(), /*Arity=*/NumArgs);
1140 // Mangle the arguments.
1141 for (unsigned i = 0; i != NumArgs; ++i)
1142 mangleExpression(CE->getArg(i));
1143 break;
1144 }
1145
Anders Carlssona7694082009-11-06 02:50:19 +00001146 case Expr::ParenExprClass:
1147 mangleExpression(cast<ParenExpr>(E)->getSubExpr());
1148 break;
1149
Anders Carlssond553f8c2009-09-21 01:21:10 +00001150 case Expr::DeclRefExprClass: {
1151 const Decl *D = cast<DeclRefExpr>(E)->getDecl();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001152
Anders Carlssond553f8c2009-09-21 01:21:10 +00001153 switch (D->getKind()) {
1154 default: assert(false && "Unhandled decl kind!");
1155 case Decl::NonTypeTemplateParm: {
1156 const NonTypeTemplateParmDecl *PD = cast<NonTypeTemplateParmDecl>(D);
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001157 mangleTemplateParameter(PD->getIndex());
Anders Carlssond553f8c2009-09-21 01:21:10 +00001158 break;
1159 }
1160
1161 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001162
Anders Carlsson50755b02009-09-27 20:11:34 +00001163 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001164 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001165
John McCall865d4472009-11-19 22:55:06 +00001166 case Expr::DependentScopeDeclRefExprClass: {
1167 const DependentScopeDeclRefExpr *DRE = cast<DependentScopeDeclRefExpr>(E);
Anders Carlsson50755b02009-09-27 20:11:34 +00001168 const Type *QTy = DRE->getQualifier()->getAsType();
1169 assert(QTy && "Qualifier was not type!");
1170
1171 // ::= sr <type> <unqualified-name> # dependent name
1172 Out << "sr";
1173 mangleType(QualType(QTy, 0));
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001174
Anders Carlsson50755b02009-09-27 20:11:34 +00001175 assert(DRE->getDeclName().getNameKind() == DeclarationName::Identifier &&
1176 "Unhandled decl name kind!");
1177 mangleSourceName(DRE->getDeclName().getAsIdentifierInfo());
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001178
Anders Carlsson50755b02009-09-27 20:11:34 +00001179 break;
1180 }
1181
Anders Carlssone170ba72009-12-14 01:45:37 +00001182 case Expr::IntegerLiteralClass:
1183 mangleIntegerLiteral(E->getType(),
1184 llvm::APSInt(cast<IntegerLiteral>(E)->getValue()));
1185 break;
1186
Anders Carlssond553f8c2009-09-21 01:21:10 +00001187 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001188}
1189
John McCallefe6aee2009-09-05 07:56:18 +00001190// FIXME: <type> ::= G <type> # imaginary (C 2000)
1191// FIXME: <type> ::= U <source-name> <type> # vendor extended type qualifier
1192
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001193void CXXNameMangler::mangleCXXCtorType(CXXCtorType T) {
1194 // <ctor-dtor-name> ::= C1 # complete object constructor
1195 // ::= C2 # base object constructor
1196 // ::= C3 # complete object allocating constructor
1197 //
1198 switch (T) {
1199 case Ctor_Complete:
1200 Out << "C1";
1201 break;
1202 case Ctor_Base:
1203 Out << "C2";
1204 break;
1205 case Ctor_CompleteAllocating:
1206 Out << "C3";
1207 break;
1208 }
1209}
1210
Anders Carlsson27ae5362009-04-17 01:58:57 +00001211void CXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
1212 // <ctor-dtor-name> ::= D0 # deleting destructor
1213 // ::= D1 # complete object destructor
1214 // ::= D2 # base object destructor
1215 //
1216 switch (T) {
1217 case Dtor_Deleting:
1218 Out << "D0";
1219 break;
1220 case Dtor_Complete:
1221 Out << "D1";
1222 break;
1223 case Dtor_Base:
1224 Out << "D2";
1225 break;
1226 }
1227}
1228
Anders Carlsson9e85c742009-12-23 19:30:55 +00001229void CXXNameMangler::mangleTemplateArgs(const TemplateArgumentList &L) {
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001230 // <template-args> ::= I <template-arg>+ E
1231 Out << "I";
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001232 for (unsigned i = 0, e = L.size(); i != e; ++i)
Anders Carlsson9e85c742009-12-23 19:30:55 +00001233 mangleTemplateArg(L[i]);
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001234 Out << "E";
1235}
1236
Anders Carlsson7624f212009-09-18 02:42:01 +00001237void CXXNameMangler::mangleTemplateArgs(const TemplateArgument *TemplateArgs,
1238 unsigned NumTemplateArgs) {
1239 // <template-args> ::= I <template-arg>+ E
1240 Out << "I";
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001241 for (unsigned i = 0; i != NumTemplateArgs; ++i)
Anders Carlsson9e85c742009-12-23 19:30:55 +00001242 mangleTemplateArg(TemplateArgs[i]);
Anders Carlsson7624f212009-09-18 02:42:01 +00001243 Out << "E";
1244}
1245
Anders Carlsson9e85c742009-12-23 19:30:55 +00001246void CXXNameMangler::mangleTemplateArg(const TemplateArgument &A) {
Mike Stump1eb44332009-09-09 15:08:12 +00001247 // <template-arg> ::= <type> # type or template
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001248 // ::= X <expression> E # expression
1249 // ::= <expr-primary> # simple expressions
1250 // ::= I <template-arg>* E # argument pack
1251 // ::= sp <expression> # pack expansion of (C++0x)
1252 switch (A.getKind()) {
1253 default:
1254 assert(0 && "Unknown template argument kind!");
1255 case TemplateArgument::Type:
1256 mangleType(A.getAsType());
1257 break;
Anders Carlsson9e85c742009-12-23 19:30:55 +00001258 case TemplateArgument::Template:
1259 mangleName(A.getAsTemplate().getAsTemplateDecl());
1260 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001261 case TemplateArgument::Expression:
1262 Out << 'X';
1263 mangleExpression(A.getAsExpr());
1264 Out << 'E';
1265 break;
Anders Carlssone170ba72009-12-14 01:45:37 +00001266 case TemplateArgument::Integral:
1267 mangleIntegerLiteral(A.getIntegralType(), *A.getAsIntegral());
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001268 break;
Daniel Dunbar7e0c1952009-11-21 09:17:15 +00001269 case TemplateArgument::Declaration: {
1270 // <expr-primary> ::= L <mangled-name> E # external name
1271
1272 // FIXME: Clang produces AST's where pointer-to-member-function expressions
1273 // and pointer-to-function expressions are represented as a declaration not
1274 // an expression; this is not how gcc represents them and this changes the
1275 // mangling.
1276 Out << 'L';
1277 // References to external entities use the mangled name; if the name would
1278 // not normally be manged then mangle it as unqualified.
1279 //
1280 // FIXME: The ABI specifies that external names here should have _Z, but
1281 // gcc leaves this off.
1282 mangle(cast<NamedDecl>(A.getAsDecl()), "Z");
1283 Out << 'E';
1284 break;
1285 }
1286 }
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001287}
1288
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001289void CXXNameMangler::mangleTemplateParameter(unsigned Index) {
1290 // <template-param> ::= T_ # first template parameter
1291 // ::= T <parameter-2 non-negative number> _
1292 if (Index == 0)
1293 Out << "T_";
1294 else
1295 Out << 'T' << (Index - 1) << '_';
1296}
1297
Anders Carlsson76967372009-09-17 00:43:46 +00001298// <substitution> ::= S <seq-id> _
1299// ::= S_
Anders Carlsson6862fc72009-09-17 04:16:28 +00001300bool CXXNameMangler::mangleSubstitution(const NamedDecl *ND) {
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001301 // Try one of the standard substitutions first.
1302 if (mangleStandardSubstitution(ND))
1303 return true;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001304
Anders Carlsson433d1372009-11-07 04:26:04 +00001305 ND = cast<NamedDecl>(ND->getCanonicalDecl());
Anders Carlsson6862fc72009-09-17 04:16:28 +00001306 return mangleSubstitution(reinterpret_cast<uintptr_t>(ND));
1307}
1308
Anders Carlsson76967372009-09-17 00:43:46 +00001309bool CXXNameMangler::mangleSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001310 if (!T.getCVRQualifiers()) {
1311 if (const RecordType *RT = T->getAs<RecordType>())
1312 return mangleSubstitution(RT->getDecl());
1313 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001314
Anders Carlsson76967372009-09-17 00:43:46 +00001315 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
1316
Anders Carlssond3a932a2009-09-17 03:53:28 +00001317 return mangleSubstitution(TypePtr);
1318}
1319
1320bool CXXNameMangler::mangleSubstitution(uintptr_t Ptr) {
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001321 llvm::DenseMap<uintptr_t, unsigned>::iterator I =
Anders Carlssond3a932a2009-09-17 03:53:28 +00001322 Substitutions.find(Ptr);
Anders Carlsson76967372009-09-17 00:43:46 +00001323 if (I == Substitutions.end())
1324 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001325
Anders Carlsson76967372009-09-17 00:43:46 +00001326 unsigned SeqID = I->second;
1327 if (SeqID == 0)
1328 Out << "S_";
1329 else {
1330 SeqID--;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001331
Anders Carlsson76967372009-09-17 00:43:46 +00001332 // <seq-id> is encoded in base-36, using digits and upper case letters.
1333 char Buffer[10];
1334 char *BufferPtr = Buffer + 9;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001335
Anders Carlsson76967372009-09-17 00:43:46 +00001336 *BufferPtr = 0;
1337 if (SeqID == 0) *--BufferPtr = '0';
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001338
Anders Carlsson76967372009-09-17 00:43:46 +00001339 while (SeqID) {
1340 assert(BufferPtr > Buffer && "Buffer overflow!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001341
Anders Carlsson76967372009-09-17 00:43:46 +00001342 unsigned char c = static_cast<unsigned char>(SeqID) % 36;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001343
Anders Carlsson76967372009-09-17 00:43:46 +00001344 *--BufferPtr = (c < 10 ? '0' + c : 'A' + c - 10);
1345 SeqID /= 36;
1346 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001347
Anders Carlsson76967372009-09-17 00:43:46 +00001348 Out << 'S' << BufferPtr << '_';
1349 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001350
Anders Carlsson76967372009-09-17 00:43:46 +00001351 return true;
1352}
1353
Anders Carlssonf514b542009-09-27 00:12:57 +00001354static bool isCharType(QualType T) {
1355 if (T.isNull())
1356 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001357
Anders Carlssonf514b542009-09-27 00:12:57 +00001358 return T->isSpecificBuiltinType(BuiltinType::Char_S) ||
1359 T->isSpecificBuiltinType(BuiltinType::Char_U);
1360}
1361
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001362/// isCharSpecialization - Returns whether a given type is a template
Anders Carlssonf514b542009-09-27 00:12:57 +00001363/// specialization of a given name with a single argument of type char.
1364static bool isCharSpecialization(QualType T, const char *Name) {
1365 if (T.isNull())
1366 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001367
Anders Carlssonf514b542009-09-27 00:12:57 +00001368 const RecordType *RT = T->getAs<RecordType>();
1369 if (!RT)
1370 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001371
1372 const ClassTemplateSpecializationDecl *SD =
Anders Carlssonf514b542009-09-27 00:12:57 +00001373 dyn_cast<ClassTemplateSpecializationDecl>(RT->getDecl());
1374 if (!SD)
1375 return false;
1376
1377 if (!isStdNamespace(SD->getDeclContext()))
1378 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001379
Anders Carlssonf514b542009-09-27 00:12:57 +00001380 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1381 if (TemplateArgs.size() != 1)
1382 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001383
Anders Carlssonf514b542009-09-27 00:12:57 +00001384 if (!isCharType(TemplateArgs[0].getAsType()))
1385 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001386
Daniel Dunbar01eb9b92009-10-18 21:17:35 +00001387 return SD->getIdentifier()->getName() == Name;
Anders Carlssonf514b542009-09-27 00:12:57 +00001388}
1389
Anders Carlsson91f88602009-12-07 19:56:42 +00001390template <std::size_t StrLen>
1391bool isStreamCharSpecialization(const ClassTemplateSpecializationDecl *SD,
1392 const char (&Str)[StrLen]) {
1393 if (!SD->getIdentifier()->isStr(Str))
1394 return false;
1395
1396 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1397 if (TemplateArgs.size() != 2)
1398 return false;
1399
1400 if (!isCharType(TemplateArgs[0].getAsType()))
1401 return false;
1402
1403 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1404 return false;
1405
1406 return true;
1407}
1408
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001409bool CXXNameMangler::mangleStandardSubstitution(const NamedDecl *ND) {
1410 // <substitution> ::= St # ::std::
Anders Carlsson8c031552009-09-26 23:10:05 +00001411 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
Anders Carlsson47846d22009-12-04 06:23:23 +00001412 if (isStd(NS)) {
Anders Carlsson8c031552009-09-26 23:10:05 +00001413 Out << "St";
1414 return true;
1415 }
1416 }
1417
1418 if (const ClassTemplateDecl *TD = dyn_cast<ClassTemplateDecl>(ND)) {
1419 if (!isStdNamespace(TD->getDeclContext()))
1420 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001421
Anders Carlsson8c031552009-09-26 23:10:05 +00001422 // <substitution> ::= Sa # ::std::allocator
1423 if (TD->getIdentifier()->isStr("allocator")) {
1424 Out << "Sa";
1425 return true;
1426 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001427
Anders Carlsson189d59c2009-09-26 23:14:39 +00001428 // <<substitution> ::= Sb # ::std::basic_string
1429 if (TD->getIdentifier()->isStr("basic_string")) {
1430 Out << "Sb";
1431 return true;
1432 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001433 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001434
1435 if (const ClassTemplateSpecializationDecl *SD =
Anders Carlssonf514b542009-09-27 00:12:57 +00001436 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
1437 // <substitution> ::= Ss # ::std::basic_string<char,
1438 // ::std::char_traits<char>,
1439 // ::std::allocator<char> >
1440 if (SD->getIdentifier()->isStr("basic_string")) {
1441 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001442
Anders Carlssonf514b542009-09-27 00:12:57 +00001443 if (TemplateArgs.size() != 3)
1444 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001445
Anders Carlssonf514b542009-09-27 00:12:57 +00001446 if (!isCharType(TemplateArgs[0].getAsType()))
1447 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001448
Anders Carlssonf514b542009-09-27 00:12:57 +00001449 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1450 return false;
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001451
Anders Carlssonf514b542009-09-27 00:12:57 +00001452 if (!isCharSpecialization(TemplateArgs[2].getAsType(), "allocator"))
1453 return false;
1454
1455 Out << "Ss";
1456 return true;
1457 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001458
Anders Carlsson91f88602009-12-07 19:56:42 +00001459 // <substitution> ::= Si # ::std::basic_istream<char,
1460 // ::std::char_traits<char> >
1461 if (isStreamCharSpecialization(SD, "basic_istream")) {
1462 Out << "Si";
1463 return true;
1464 }
1465
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001466 // <substitution> ::= So # ::std::basic_ostream<char,
Anders Carlsson8f8fd8e2009-10-08 17:20:26 +00001467 // ::std::char_traits<char> >
Anders Carlsson91f88602009-12-07 19:56:42 +00001468 if (isStreamCharSpecialization(SD, "basic_ostream")) {
Anders Carlsson8f8fd8e2009-10-08 17:20:26 +00001469 Out << "So";
1470 return true;
1471 }
Anders Carlsson91f88602009-12-07 19:56:42 +00001472
1473 // <substitution> ::= Sd # ::std::basic_iostream<char,
1474 // ::std::char_traits<char> >
1475 if (isStreamCharSpecialization(SD, "basic_iostream")) {
1476 Out << "Sd";
1477 return true;
1478 }
Anders Carlssonf514b542009-09-27 00:12:57 +00001479 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001480 return false;
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001481}
1482
Anders Carlsson76967372009-09-17 00:43:46 +00001483void CXXNameMangler::addSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001484 if (!T.getCVRQualifiers()) {
1485 if (const RecordType *RT = T->getAs<RecordType>()) {
1486 addSubstitution(RT->getDecl());
1487 return;
1488 }
1489 }
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001490
Anders Carlsson76967372009-09-17 00:43:46 +00001491 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
Anders Carlssond3a932a2009-09-17 03:53:28 +00001492 addSubstitution(TypePtr);
1493}
1494
1495void CXXNameMangler::addSubstitution(uintptr_t Ptr) {
Anders Carlsson76967372009-09-17 00:43:46 +00001496 unsigned SeqID = Substitutions.size();
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001497
Anders Carlssond3a932a2009-09-17 03:53:28 +00001498 assert(!Substitutions.count(Ptr) && "Substitution already exists!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001499 Substitutions[Ptr] = SeqID;
Anders Carlsson76967372009-09-17 00:43:46 +00001500}
1501
Daniel Dunbar1b077112009-11-21 09:06:10 +00001502//
Mike Stump1eb44332009-09-09 15:08:12 +00001503
Daniel Dunbar1b077112009-11-21 09:06:10 +00001504/// \brief Mangles the name of the declaration D and emits that name to the
1505/// given output stream.
1506///
1507/// If the declaration D requires a mangled name, this routine will emit that
1508/// mangled name to \p os and return true. Otherwise, \p os will be unchanged
1509/// and this routine will return false. In this case, the caller should just
1510/// emit the identifier of the declaration (\c D->getIdentifier()) as its
1511/// name.
Daniel Dunbarf981bf82009-11-21 09:14:52 +00001512void MangleContext::mangleName(const NamedDecl *D,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001513 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc02ab4c2009-11-21 09:14:44 +00001514 assert((isa<FunctionDecl>(D) || isa<VarDecl>(D)) &&
1515 "Invalid mangleName() call, argument is not a variable or function!");
1516 assert(!isa<CXXConstructorDecl>(D) && !isa<CXXDestructorDecl>(D) &&
1517 "Invalid mangleName() call on 'structor decl!");
Daniel Dunbar3c9e4632009-11-21 09:05:47 +00001518
Daniel Dunbar1b077112009-11-21 09:06:10 +00001519 PrettyStackTraceDecl CrashInfo(D, SourceLocation(),
1520 getASTContext().getSourceManager(),
1521 "Mangling declaration");
Mike Stump1eb44332009-09-09 15:08:12 +00001522
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001523 CXXNameMangler Mangler(*this, Res);
1524 return Mangler.mangle(D);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001525}
Mike Stump1eb44332009-09-09 15:08:12 +00001526
Daniel Dunbar1b077112009-11-21 09:06:10 +00001527void MangleContext::mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001528 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbar77939c92009-11-21 09:06:31 +00001529 CXXNameMangler Mangler(*this, Res, D, Type);
1530 Mangler.mangle(D);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001531}
Mike Stump1eb44332009-09-09 15:08:12 +00001532
Daniel Dunbar1b077112009-11-21 09:06:10 +00001533void MangleContext::mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001534 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbar77939c92009-11-21 09:06:31 +00001535 CXXNameMangler Mangler(*this, Res, D, Type);
1536 Mangler.mangle(D);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001537}
Mike Stumpf1216772009-07-31 18:25:34 +00001538
Daniel Dunbarc0747712009-11-21 09:12:13 +00001539/// \brief Mangles the a thunk with the offset n for the declaration D and
1540/// emits that name to the given output stream.
Anders Carlssona94822e2009-11-26 02:32:05 +00001541void MangleContext::mangleThunk(const FunctionDecl *FD,
Anders Carlssonb73a5be2009-11-26 02:49:32 +00001542 const ThunkAdjustment &ThisAdjustment,
Daniel Dunbarc0747712009-11-21 09:12:13 +00001543 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001544 assert(!isa<CXXDestructorDecl>(FD) &&
1545 "Use mangleCXXDtor for destructor decls!");
1546
1547 // <special-name> ::= T <call-offset> <base encoding>
1548 // # base is the nominal target function of thunk
1549 CXXNameMangler Mangler(*this, Res);
1550 Mangler.getStream() << "_ZT";
Anders Carlssonb73a5be2009-11-26 02:49:32 +00001551 Mangler.mangleCallOffset(ThisAdjustment);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001552 Mangler.mangleFunctionEncoding(FD);
1553}
1554
Eli Friedman61d89b62009-12-03 00:03:05 +00001555void MangleContext::mangleCXXDtorThunk(const CXXDestructorDecl *D,
1556 CXXDtorType Type,
1557 const ThunkAdjustment &ThisAdjustment,
1558 llvm::SmallVectorImpl<char> &Res) {
1559 // <special-name> ::= T <call-offset> <base encoding>
1560 // # base is the nominal target function of thunk
1561 CXXNameMangler Mangler(*this, Res, D, Type);
1562 Mangler.getStream() << "_ZT";
1563 Mangler.mangleCallOffset(ThisAdjustment);
1564 Mangler.mangleFunctionEncoding(D);
1565}
1566
Daniel Dunbarc0747712009-11-21 09:12:13 +00001567/// \brief Mangles the a covariant thunk for the declaration D and emits that
1568/// name to the given output stream.
Anders Carlsson7622cd32009-11-26 03:09:37 +00001569void
1570MangleContext::mangleCovariantThunk(const FunctionDecl *FD,
1571 const CovariantThunkAdjustment& Adjustment,
1572 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001573 assert(!isa<CXXDestructorDecl>(FD) &&
Eli Friedman61d89b62009-12-03 00:03:05 +00001574 "No such thing as a covariant thunk for a destructor!");
Daniel Dunbarc0747712009-11-21 09:12:13 +00001575
1576 // <special-name> ::= Tc <call-offset> <call-offset> <base encoding>
1577 // # base is the nominal target function of thunk
1578 // # first call-offset is 'this' adjustment
1579 // # second call-offset is result adjustment
1580 CXXNameMangler Mangler(*this, Res);
1581 Mangler.getStream() << "_ZTc";
Anders Carlsson7622cd32009-11-26 03:09:37 +00001582 Mangler.mangleCallOffset(Adjustment.ThisAdjustment);
1583 Mangler.mangleCallOffset(Adjustment.ReturnAdjustment);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001584 Mangler.mangleFunctionEncoding(FD);
1585}
1586
1587/// mangleGuardVariable - Returns the mangled name for a guard variable
1588/// for the passed in VarDecl.
1589void MangleContext::mangleGuardVariable(const VarDecl *D,
1590 llvm::SmallVectorImpl<char> &Res) {
1591 // <special-name> ::= GV <object name> # Guard variable for one-time
1592 // # initialization
1593 CXXNameMangler Mangler(*this, Res);
1594 Mangler.getStream() << "_ZGV";
1595 Mangler.mangleName(D);
1596}
1597
Daniel Dunbar1b077112009-11-21 09:06:10 +00001598void MangleContext::mangleCXXVtable(const CXXRecordDecl *RD,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001599 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001600 // <special-name> ::= TV <type> # virtual table
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001601 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001602 Mangler.getStream() << "_ZTV";
1603 Mangler.mangleName(RD);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001604}
Mike Stump82d75b02009-11-10 01:58:37 +00001605
Daniel Dunbar1b077112009-11-21 09:06:10 +00001606void MangleContext::mangleCXXVTT(const CXXRecordDecl *RD,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001607 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001608 // <special-name> ::= TT <type> # VTT structure
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001609 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001610 Mangler.getStream() << "_ZTT";
1611 Mangler.mangleName(RD);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001612}
Mike Stumpab3f7e92009-11-10 01:41:59 +00001613
Daniel Dunbar1b077112009-11-21 09:06:10 +00001614void MangleContext::mangleCXXCtorVtable(const CXXRecordDecl *RD, int64_t Offset,
1615 const CXXRecordDecl *Type,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001616 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001617 // <special-name> ::= TC <type> <offset number> _ <base type>
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001618 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001619 Mangler.getStream() << "_ZTC";
1620 Mangler.mangleName(RD);
1621 Mangler.getStream() << Offset;
1622 Mangler.getStream() << "_";
1623 Mangler.mangleName(Type);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001624}
Mike Stump738f8c22009-07-31 23:15:31 +00001625
Mike Stumpde050572009-12-02 18:57:08 +00001626void MangleContext::mangleCXXRTTI(QualType Ty,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001627 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001628 // <special-name> ::= TI <type> # typeinfo structure
Douglas Gregor154fe982009-12-23 22:04:40 +00001629 assert(!Ty.hasQualifiers() && "RTTI info cannot have top-level qualifiers");
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001630 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001631 Mangler.getStream() << "_ZTI";
1632 Mangler.mangleType(Ty);
Daniel Dunbar1b077112009-11-21 09:06:10 +00001633}
Mike Stump67795982009-11-14 00:14:13 +00001634
Mike Stumpde050572009-12-02 18:57:08 +00001635void MangleContext::mangleCXXRTTIName(QualType Ty,
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001636 llvm::SmallVectorImpl<char> &Res) {
Daniel Dunbarc0747712009-11-21 09:12:13 +00001637 // <special-name> ::= TS <type> # typeinfo name (null terminated byte string)
Daniel Dunbar94fd26d2009-11-21 09:06:22 +00001638 CXXNameMangler Mangler(*this, Res);
Daniel Dunbarc0747712009-11-21 09:12:13 +00001639 Mangler.getStream() << "_ZTS";
1640 Mangler.mangleType(Ty);
Mike Stumpf1216772009-07-31 18:25:34 +00001641}