blob: 081156cb6e3c99179aa3745e1f1c2ddabb72b733 [file] [log] [blame]
Douglas Gregor6ec36682009-02-18 23:53:56 +00001//===--- Mangle.cpp - Mangle C++ Names --------------------------*- C++ -*-===//
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// Implements C++ name mangling according to the Itanium C++ ABI,
11// which is used in GCC 3.2 and newer (and many compilers that are
12// ABI-compatible with GCC):
13//
14// http://www.codesourcery.com/public/cxx-abi/abi.html
15//
16//===----------------------------------------------------------------------===//
17#include "Mangle.h"
18#include "clang/AST/ASTContext.h"
19#include "clang/AST/Decl.h"
20#include "clang/AST/DeclCXX.h"
Anders Carlssona40c5e42009-03-07 22:03:21 +000021#include "clang/AST/DeclObjC.h"
Anders Carlsson7a0ba872009-05-15 16:09:15 +000022#include "clang/AST/DeclTemplate.h"
Anders Carlsson50755b02009-09-27 20:11:34 +000023#include "clang/AST/ExprCXX.h"
Douglas Gregor6ec36682009-02-18 23:53:56 +000024#include "clang/Basic/SourceManager.h"
Anders Carlssonc4355b62009-10-07 01:45:02 +000025#include "llvm/ADT/StringExtras.h"
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000026#include "llvm/Support/Compiler.h"
27#include "llvm/Support/raw_ostream.h"
John McCallefe6aee2009-09-05 07:56:18 +000028#include "llvm/Support/ErrorHandling.h"
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000029using namespace clang;
30
31namespace {
32 class VISIBILITY_HIDDEN CXXNameMangler {
Anders Carlssonb5404912009-10-07 01:06:45 +000033 MangleContext &Context;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000034 llvm::raw_ostream &Out;
35
Anders Carlsson27ae5362009-04-17 01:58:57 +000036 const CXXMethodDecl *Structor;
37 unsigned StructorType;
Anders Carlsson3ac86b52009-04-15 05:36:58 +000038 CXXCtorType CtorType;
Mike Stump1eb44332009-09-09 15:08:12 +000039
Anders Carlsson76967372009-09-17 00:43:46 +000040 llvm::DenseMap<uintptr_t, unsigned> Substitutions;
41
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000042 public:
Anders Carlssonb5404912009-10-07 01:06:45 +000043 CXXNameMangler(MangleContext &C, llvm::raw_ostream &os)
Anders Carlsson27ae5362009-04-17 01:58:57 +000044 : Context(C), Out(os), Structor(0), StructorType(0) { }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000045
46 bool mangle(const NamedDecl *D);
Mike Stump77ca8f62009-09-05 07:20:32 +000047 void mangleCalloffset(int64_t nv, int64_t v);
Mike Stumpdec025b2009-09-07 04:27:52 +000048 void mangleThunk(const FunctionDecl *FD, int64_t nv, int64_t v);
49 void mangleCovariantThunk(const FunctionDecl *FD,
Mike Stump77ca8f62009-09-05 07:20:32 +000050 int64_t nv_t, int64_t v_t,
Mike Stump9124bcc2009-09-02 00:56:18 +000051 int64_t nv_r, int64_t v_r);
Anders Carlsson41aa8c12009-04-13 18:02:10 +000052 void mangleGuardVariable(const VarDecl *D);
Mike Stump1eb44332009-09-09 15:08:12 +000053
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +000054 void mangleCXXVtable(const CXXRecordDecl *RD);
Mike Stump88e91712009-10-28 01:51:46 +000055 void mangleCXXVTT(const CXXRecordDecl *RD);
Anders Carlssond313e402009-10-30 01:52:02 +000056 void mangleCXXRtti(QualType Ty);
Anders Carlsson3ac86b52009-04-15 05:36:58 +000057 void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type);
Anders Carlsson27ae5362009-04-17 01:58:57 +000058 void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type);
Anders Carlsson3ac86b52009-04-15 05:36:58 +000059
Anders Carlsson43f17402009-04-02 15:51:53 +000060 private:
Anders Carlsson6862fc72009-09-17 04:16:28 +000061 bool mangleSubstitution(const NamedDecl *ND);
Anders Carlsson76967372009-09-17 00:43:46 +000062 bool mangleSubstitution(QualType T);
Anders Carlssond3a932a2009-09-17 03:53:28 +000063 bool mangleSubstitution(uintptr_t Ptr);
64
Anders Carlssone7c8cb62009-09-26 20:53:44 +000065 bool mangleStandardSubstitution(const NamedDecl *ND);
66
Anders Carlssond3a932a2009-09-17 03:53:28 +000067 void addSubstitution(const NamedDecl *ND) {
Anders Carlsson433d1372009-11-07 04:26:04 +000068 ND = cast<NamedDecl>(ND->getCanonicalDecl());
69
Anders Carlssond3a932a2009-09-17 03:53:28 +000070 addSubstitution(reinterpret_cast<uintptr_t>(ND));
71 }
Anders Carlsson76967372009-09-17 00:43:46 +000072 void addSubstitution(QualType T);
Anders Carlssond3a932a2009-09-17 03:53:28 +000073 void addSubstitution(uintptr_t Ptr);
Anders Carlsson76967372009-09-17 00:43:46 +000074
Anders Carlsson43f17402009-04-02 15:51:53 +000075 bool mangleFunctionDecl(const FunctionDecl *FD);
Mike Stump1eb44332009-09-09 15:08:12 +000076
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000077 void mangleFunctionEncoding(const FunctionDecl *FD);
78 void mangleName(const NamedDecl *ND);
Anders Carlsson7624f212009-09-18 02:42:01 +000079 void mangleName(const TemplateDecl *TD,
80 const TemplateArgument *TemplateArgs,
81 unsigned NumTemplateArgs);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000082 void mangleUnqualifiedName(const NamedDecl *ND);
Anders Carlsson201ce742009-09-17 03:17:01 +000083 void mangleUnscopedName(const NamedDecl *ND);
Anders Carlsson0fa6df42009-09-26 19:45:45 +000084 void mangleUnscopedTemplateName(const TemplateDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000085 void mangleSourceName(const IdentifierInfo *II);
Anders Carlsson1b42c792009-04-02 16:24:45 +000086 void mangleLocalName(const NamedDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000087 void mangleNestedName(const NamedDecl *ND);
Anders Carlsson7624f212009-09-18 02:42:01 +000088 void mangleNestedName(const TemplateDecl *TD,
89 const TemplateArgument *TemplateArgs,
90 unsigned NumTemplateArgs);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000091 void manglePrefix(const DeclContext *DC);
Anders Carlsson0fa6df42009-09-26 19:45:45 +000092 void mangleTemplatePrefix(const TemplateDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000093 void mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity);
John McCall0953e762009-09-24 19:53:00 +000094 void mangleQualifiers(Qualifiers Quals);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000095 void mangleType(QualType T);
John McCallefe6aee2009-09-05 07:56:18 +000096
97 // Declare manglers for every type class.
98#define ABSTRACT_TYPE(CLASS, PARENT)
99#define NON_CANONICAL_TYPE(CLASS, PARENT)
100#define TYPE(CLASS, PARENT) void mangleType(const CLASS##Type *T);
101#include "clang/AST/TypeNodes.def"
102
103 void mangleType(const TagType*);
104 void mangleBareFunctionType(const FunctionType *T,
105 bool MangleReturnType);
Anders Carlssond553f8c2009-09-21 01:21:10 +0000106 void mangleExpression(const Expr *E);
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000107 void mangleCXXCtorType(CXXCtorType T);
Anders Carlsson27ae5362009-04-17 01:58:57 +0000108 void mangleCXXDtorType(CXXDtorType T);
Mike Stump1eb44332009-09-09 15:08:12 +0000109
Anders Carlsson7624f212009-09-18 02:42:01 +0000110 void mangleTemplateArgs(const TemplateArgument *TemplateArgs,
111 unsigned NumTemplateArgs);
Anders Carlsson068f3472009-09-17 05:31:47 +0000112 void mangleTemplateArgumentList(const TemplateArgumentList &L);
113 void mangleTemplateArgument(const TemplateArgument &A);
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +0000114
115 void mangleTemplateParameter(unsigned Index);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000116 };
117}
118
Anders Carlsson43f17402009-04-02 15:51:53 +0000119static bool isInCLinkageSpecification(const Decl *D) {
Douglas Gregor457e2812009-10-28 16:31:34 +0000120 D = D->getCanonicalDecl();
Mike Stump1eb44332009-09-09 15:08:12 +0000121 for (const DeclContext *DC = D->getDeclContext();
Anders Carlsson43f17402009-04-02 15:51:53 +0000122 !DC->isTranslationUnit(); DC = DC->getParent()) {
Mike Stump1eb44332009-09-09 15:08:12 +0000123 if (const LinkageSpecDecl *Linkage = dyn_cast<LinkageSpecDecl>(DC))
Anders Carlsson43f17402009-04-02 15:51:53 +0000124 return Linkage->getLanguage() == LinkageSpecDecl::lang_c;
125 }
Mike Stump1eb44332009-09-09 15:08:12 +0000126
Anders Carlsson43f17402009-04-02 15:51:53 +0000127 return false;
128}
129
130bool CXXNameMangler::mangleFunctionDecl(const FunctionDecl *FD) {
Mike Stump141c5af2009-09-02 00:25:38 +0000131 // Clang's "overloadable" attribute extension to C/C++ implies name mangling
132 // (always).
Argyrios Kyrtzidis40b598e2009-06-30 02:34:44 +0000133 if (!FD->hasAttr<OverloadableAttr>()) {
Chris Lattner783601d2009-06-13 23:34:16 +0000134 // C functions are not mangled, and "main" is never mangled.
Anders Carlssonb5404912009-10-07 01:06:45 +0000135 if (!Context.getASTContext().getLangOptions().CPlusPlus || FD->isMain())
Chris Lattner783601d2009-06-13 23:34:16 +0000136 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000137
138 // No mangling in an "implicit extern C" header.
Chris Lattner783601d2009-06-13 23:34:16 +0000139 if (FD->getLocation().isValid() &&
Anders Carlssonb5404912009-10-07 01:06:45 +0000140 Context.getASTContext().getSourceManager().
141 isInExternCSystemHeader(FD->getLocation()))
Chris Lattner783601d2009-06-13 23:34:16 +0000142 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000143
Chris Lattner783601d2009-06-13 23:34:16 +0000144 // No name mangling in a C linkage specification.
Anders Carlssonadd28822009-09-22 20:33:31 +0000145 if (!isa<CXXMethodDecl>(FD) && isInCLinkageSpecification(FD))
Chris Lattner783601d2009-06-13 23:34:16 +0000146 return false;
147 }
Anders Carlsson43f17402009-04-02 15:51:53 +0000148
149 // If we get here, mangle the decl name!
150 Out << "_Z";
151 mangleFunctionEncoding(FD);
152 return true;
153}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000154
155bool CXXNameMangler::mangle(const NamedDecl *D) {
Mike Stump141c5af2009-09-02 00:25:38 +0000156 // Any decl can be declared with __asm("foo") on it, and this takes precedence
157 // over all other naming in the .o file.
Argyrios Kyrtzidis40b598e2009-06-30 02:34:44 +0000158 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000159 // If we have an asm name, then we use it as the mangling.
160 Out << '\01'; // LLVM IR Marker for __asm("foo")
161 Out << ALA->getLabel();
162 return true;
163 }
Mike Stump1eb44332009-09-09 15:08:12 +0000164
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000165 // <mangled-name> ::= _Z <encoding>
166 // ::= <data name>
167 // ::= <special-name>
168
169 // FIXME: Actually use a visitor to decode these?
Anders Carlsson43f17402009-04-02 15:51:53 +0000170 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
171 return mangleFunctionDecl(FD);
Mike Stump1eb44332009-09-09 15:08:12 +0000172
Anders Carlsson329749c2009-04-02 16:05:20 +0000173 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
Anders Carlssonb5404912009-10-07 01:06:45 +0000174 if (!Context.getASTContext().getLangOptions().CPlusPlus ||
Anders Carlsson9ccb0652009-04-11 01:19:45 +0000175 isInCLinkageSpecification(D) ||
176 D->getDeclContext()->isTranslationUnit())
Anders Carlsson329749c2009-04-02 16:05:20 +0000177 return false;
Mike Stump1eb44332009-09-09 15:08:12 +0000178
Anders Carlsson329749c2009-04-02 16:05:20 +0000179 Out << "_Z";
180 mangleName(VD);
181 return true;
182 }
Mike Stump1eb44332009-09-09 15:08:12 +0000183
Anders Carlsson43f17402009-04-02 15:51:53 +0000184 return false;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000185}
186
Mike Stump1eb44332009-09-09 15:08:12 +0000187void CXXNameMangler::mangleCXXCtor(const CXXConstructorDecl *D,
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000188 CXXCtorType Type) {
Anders Carlsson27ae5362009-04-17 01:58:57 +0000189 assert(!Structor && "Structor already set!");
190 Structor = D;
191 StructorType = Type;
Mike Stump1eb44332009-09-09 15:08:12 +0000192
Anders Carlsson27ae5362009-04-17 01:58:57 +0000193 mangle(D);
194}
195
Mike Stump1eb44332009-09-09 15:08:12 +0000196void CXXNameMangler::mangleCXXDtor(const CXXDestructorDecl *D,
Anders Carlsson27ae5362009-04-17 01:58:57 +0000197 CXXDtorType Type) {
198 assert(!Structor && "Structor already set!");
199 Structor = D;
200 StructorType = Type;
Mike Stump1eb44332009-09-09 15:08:12 +0000201
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000202 mangle(D);
203}
204
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +0000205void CXXNameMangler::mangleCXXVtable(const CXXRecordDecl *RD) {
Mike Stumpf1216772009-07-31 18:25:34 +0000206 // <special-name> ::= TV <type> # virtual table
207 Out << "_ZTV";
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +0000208 mangleName(RD);
Mike Stumpf1216772009-07-31 18:25:34 +0000209}
210
Mike Stump88e91712009-10-28 01:51:46 +0000211void CXXNameMangler::mangleCXXVTT(const CXXRecordDecl *RD) {
212 // <special-name> ::= TT <type> # VTT structure
213 Out << "_ZTT";
214 mangleName(RD);
215}
216
Anders Carlssond313e402009-10-30 01:52:02 +0000217void CXXNameMangler::mangleCXXRtti(QualType Ty) {
Mike Stump738f8c22009-07-31 23:15:31 +0000218 // <special-name> ::= TI <type> # typeinfo structure
219 Out << "_ZTI";
Anders Carlsson731f7f92009-10-30 01:26:12 +0000220
Anders Carlssond313e402009-10-30 01:52:02 +0000221 mangleType(Ty);
Mike Stump738f8c22009-07-31 23:15:31 +0000222}
223
Mike Stump1eb44332009-09-09 15:08:12 +0000224void CXXNameMangler::mangleGuardVariable(const VarDecl *D) {
225 // <special-name> ::= GV <object name> # Guard variable for one-time
Mike Stumpf1216772009-07-31 18:25:34 +0000226 // # initialization
Anders Carlsson41aa8c12009-04-13 18:02:10 +0000227
228 Out << "_ZGV";
229 mangleName(D);
230}
231
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000232void CXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD) {
233 // <encoding> ::= <function name> <bare-function-type>
234 mangleName(FD);
Mike Stump1eb44332009-09-09 15:08:12 +0000235
Mike Stump141c5af2009-09-02 00:25:38 +0000236 // Whether the mangling of a function type includes the return type depends on
237 // the context and the nature of the function. The rules for deciding whether
238 // the return type is included are:
Mike Stump1eb44332009-09-09 15:08:12 +0000239 //
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000240 // 1. Template functions (names or types) have return types encoded, with
241 // the exceptions listed below.
Mike Stump1eb44332009-09-09 15:08:12 +0000242 // 2. Function types not appearing as part of a function name mangling,
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000243 // e.g. parameters, pointer types, etc., have return type encoded, with the
244 // exceptions listed below.
245 // 3. Non-template function names do not have return types encoded.
246 //
Mike Stump141c5af2009-09-02 00:25:38 +0000247 // The exceptions mentioned in (1) and (2) above, for which the return type is
248 // never included, are
Douglas Gregor1fd2dd12009-06-29 22:39:32 +0000249 // 1. Constructors.
250 // 2. Destructors.
251 // 3. Conversion operator functions, e.g. operator int.
252 bool MangleReturnType = false;
Anders Carlsson9234b7f2009-09-17 03:46:43 +0000253 if (FunctionTemplateDecl *PrimaryTemplate = FD->getPrimaryTemplate()) {
254 if (!(isa<CXXConstructorDecl>(FD) || isa<CXXDestructorDecl>(FD) ||
255 isa<CXXConversionDecl>(FD)))
256 MangleReturnType = true;
257
258 // Mangle the type of the primary template.
259 FD = PrimaryTemplate->getTemplatedDecl();
260 }
261
John McCall54e14c42009-10-22 22:37:11 +0000262 // Do the canonicalization out here because parameter types can
263 // undergo additional canonicalization (e.g. array decay).
264 FunctionType *FT = cast<FunctionType>(Context.getASTContext()
265 .getCanonicalType(FD->getType()));
266
267 mangleBareFunctionType(FT, MangleReturnType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000268}
269
270static bool isStdNamespace(const DeclContext *DC) {
271 if (!DC->isNamespace() || !DC->getParent()->isTranslationUnit())
272 return false;
273
274 const NamespaceDecl *NS = cast<NamespaceDecl>(DC);
John McCall9aeed322009-10-01 00:25:31 +0000275 const IdentifierInfo *II = NS->getOriginalNamespace()->getIdentifier();
276 return II && II->isStr("std");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000277}
278
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000279static const TemplateDecl *
280isTemplate(const NamedDecl *ND, const TemplateArgumentList *&TemplateArgs) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000281 // Check if we have a function template.
282 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(ND)){
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000283 if (const TemplateDecl *TD = FD->getPrimaryTemplate()) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000284 TemplateArgs = FD->getTemplateSpecializationArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000285 return TD;
Anders Carlsson2744a062009-09-18 19:00:18 +0000286 }
287 }
288
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000289 // Check if we have a class template.
290 if (const ClassTemplateSpecializationDecl *Spec =
291 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
292 TemplateArgs = &Spec->getTemplateArgs();
Anders Carlssonbb36ba42009-09-26 03:24:57 +0000293 return Spec->getSpecializedTemplate();
Anders Carlssoneafc6dc2009-09-18 19:44:50 +0000294 }
295
Anders Carlsson2744a062009-09-18 19:00:18 +0000296 return 0;
297}
298
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000299void CXXNameMangler::mangleName(const NamedDecl *ND) {
300 // <name> ::= <nested-name>
301 // ::= <unscoped-name>
302 // ::= <unscoped-template-name> <template-args>
Anders Carlsson201ce742009-09-17 03:17:01 +0000303 // ::= <local-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000304 //
Anders Carlssond58d6f72009-09-17 16:12:20 +0000305 const DeclContext *DC = ND->getDeclContext();
Anders Carlsson5cc58c62009-09-22 17:23:30 +0000306 while (isa<LinkageSpecDecl>(DC))
Anders Carlssond58d6f72009-09-17 16:12:20 +0000307 DC = DC->getParent();
Anders Carlssond58d6f72009-09-17 16:12:20 +0000308
309 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000310 // Check if we have a template.
311 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000312 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000313 mangleUnscopedTemplateName(TD);
314 mangleTemplateArgumentList(*TemplateArgs);
315 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000316 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000317
Anders Carlsson7482e242009-09-18 04:29:09 +0000318 mangleUnscopedName(ND);
319 return;
320 }
321
322 if (isa<FunctionDecl>(DC)) {
323 mangleLocalName(ND);
324 return;
325 }
326
327 mangleNestedName(ND);
328}
Anders Carlsson7624f212009-09-18 02:42:01 +0000329void CXXNameMangler::mangleName(const TemplateDecl *TD,
330 const TemplateArgument *TemplateArgs,
331 unsigned NumTemplateArgs) {
332 const DeclContext *DC = TD->getDeclContext();
333 while (isa<LinkageSpecDecl>(DC)) {
334 assert(cast<LinkageSpecDecl>(DC)->getLanguage() ==
335 LinkageSpecDecl::lang_cxx && "Unexpected linkage decl!");
336 DC = DC->getParent();
337 }
338
339 if (DC->isTranslationUnit() || isStdNamespace(DC)) {
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000340 mangleUnscopedTemplateName(TD);
Anders Carlsson7624f212009-09-18 02:42:01 +0000341 mangleTemplateArgs(TemplateArgs, NumTemplateArgs);
342 } else {
343 mangleNestedName(TD, TemplateArgs, NumTemplateArgs);
344 }
345}
346
Anders Carlsson201ce742009-09-17 03:17:01 +0000347void CXXNameMangler::mangleUnscopedName(const NamedDecl *ND) {
348 // <unscoped-name> ::= <unqualified-name>
349 // ::= St <unqualified-name> # ::std::
350 if (isStdNamespace(ND->getDeclContext()))
351 Out << "St";
352
353 mangleUnqualifiedName(ND);
354}
355
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000356void CXXNameMangler::mangleUnscopedTemplateName(const TemplateDecl *ND) {
Anders Carlsson201ce742009-09-17 03:17:01 +0000357 // <unscoped-template-name> ::= <unscoped-name>
358 // ::= <substitution>
Anders Carlsson7624f212009-09-18 02:42:01 +0000359 if (mangleSubstitution(ND))
Anders Carlsson03c9d532009-09-17 04:02:31 +0000360 return;
361
Anders Carlsson1668f202009-09-26 20:13:56 +0000362 mangleUnscopedName(ND->getTemplatedDecl());
Anders Carlsson7624f212009-09-18 02:42:01 +0000363 addSubstitution(ND);
Anders Carlsson201ce742009-09-17 03:17:01 +0000364}
365
Mike Stump77ca8f62009-09-05 07:20:32 +0000366void CXXNameMangler::mangleCalloffset(int64_t nv, int64_t v) {
Mike Stump141c5af2009-09-02 00:25:38 +0000367 // <call-offset> ::= h <nv-offset> _
368 // ::= v <v-offset> _
369 // <nv-offset> ::= <offset number> # non-virtual base override
370 // <v-offset> ::= <offset nubmer> _ <virtual offset number>
371 // # virtual base override, with vcall offset
Mike Stump77ca8f62009-09-05 07:20:32 +0000372 if (v == 0) {
Mike Stump9124bcc2009-09-02 00:56:18 +0000373 Out << "h";
Mike Stump141c5af2009-09-02 00:25:38 +0000374 if (nv < 0) {
375 Out << "n";
376 nv = -nv;
377 }
378 Out << nv;
379 } else {
Mike Stump9124bcc2009-09-02 00:56:18 +0000380 Out << "v";
Mike Stump141c5af2009-09-02 00:25:38 +0000381 if (nv < 0) {
382 Out << "n";
383 nv = -nv;
384 }
385 Out << nv;
386 Out << "_";
387 if (v < 0) {
388 Out << "n";
389 v = -v;
390 }
391 Out << v;
392 }
393 Out << "_";
Mike Stump9124bcc2009-09-02 00:56:18 +0000394}
395
Mike Stumpdec025b2009-09-07 04:27:52 +0000396void CXXNameMangler::mangleThunk(const FunctionDecl *FD, int64_t nv,
397 int64_t v) {
Mike Stump9124bcc2009-09-02 00:56:18 +0000398 // <special-name> ::= T <call-offset> <base encoding>
399 // # base is the nominal target function of thunk
Mike Stumpdec025b2009-09-07 04:27:52 +0000400 Out << "_ZT";
Mike Stump77ca8f62009-09-05 07:20:32 +0000401 mangleCalloffset(nv, v);
Mike Stumpdec025b2009-09-07 04:27:52 +0000402 mangleFunctionEncoding(FD);
Mike Stump9124bcc2009-09-02 00:56:18 +0000403}
404
Mike Stumpdec025b2009-09-07 04:27:52 +0000405 void CXXNameMangler::mangleCovariantThunk(const FunctionDecl *FD,
Mike Stump77ca8f62009-09-05 07:20:32 +0000406 int64_t nv_t, int64_t v_t,
Mike Stump9124bcc2009-09-02 00:56:18 +0000407 int64_t nv_r, int64_t v_r) {
408 // <special-name> ::= Tc <call-offset> <call-offset> <base encoding>
409 // # base is the nominal target function of thunk
410 // # first call-offset is 'this' adjustment
411 // # second call-offset is result adjustment
Mike Stumpdec025b2009-09-07 04:27:52 +0000412 Out << "_ZTc";
Mike Stump77ca8f62009-09-05 07:20:32 +0000413 mangleCalloffset(nv_t, v_t);
414 mangleCalloffset(nv_r, v_r);
Mike Stumpdec025b2009-09-07 04:27:52 +0000415 mangleFunctionEncoding(FD);
Mike Stump141c5af2009-09-02 00:25:38 +0000416}
417
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000418void CXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND) {
419 // <unqualified-name> ::= <operator-name>
Mike Stump1eb44332009-09-09 15:08:12 +0000420 // ::= <ctor-dtor-name>
421 // ::= <source-name>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000422 DeclarationName Name = ND->getDeclName();
423 switch (Name.getNameKind()) {
Anders Carlssonc4355b62009-10-07 01:45:02 +0000424 case DeclarationName::Identifier: {
425 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
John McCall9aeed322009-10-01 00:25:31 +0000426 if (NS->isAnonymousNamespace()) {
427 // This is how gcc mangles these names. It's apparently
428 // always '1', no matter how many different anonymous
429 // namespaces appear in a context.
430 Out << "12_GLOBAL__N_1";
431 break;
432 }
Anders Carlssonc4355b62009-10-07 01:45:02 +0000433 }
434
435 if (const IdentifierInfo *II = Name.getAsIdentifierInfo()) {
436 mangleSourceName(II);
437 break;
438 }
439
440 // We must have an anonymous struct.
441 const TagDecl *TD = cast<TagDecl>(ND);
442 if (const TypedefDecl *D = TD->getTypedefForAnonDecl()) {
443 assert(TD->getDeclContext() == D->getDeclContext() &&
444 "Typedef should not be in another decl context!");
445 assert(D->getDeclName().getAsIdentifierInfo() &&
446 "Typedef was not named!");
447 mangleSourceName(D->getDeclName().getAsIdentifierInfo());
448 break;
449 }
450
451 // Get a unique id for the anonymous struct.
452 uint64_t AnonStructId = Context.getAnonymousStructId(TD);
453
454 // Mangle it as a source name in the form
455 // [n] $_<id>
456 // where n is the length of the string.
457 llvm::SmallString<8> Str;
458 Str += "$_";
459 Str += llvm::utostr(AnonStructId);
460
461 Out << Str.size();
462 Out << Str.str();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000463 break;
Anders Carlssonc4355b62009-10-07 01:45:02 +0000464 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000465
466 case DeclarationName::ObjCZeroArgSelector:
467 case DeclarationName::ObjCOneArgSelector:
468 case DeclarationName::ObjCMultiArgSelector:
469 assert(false && "Can't mangle Objective-C selector names here!");
470 break;
471
472 case DeclarationName::CXXConstructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000473 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000474 // If the named decl is the C++ constructor we're mangling, use the type
475 // we were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000476 mangleCXXCtorType(static_cast<CXXCtorType>(StructorType));
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000477 else
478 // Otherwise, use the complete constructor name. This is relevant if a
479 // class with a constructor is declared within a constructor.
480 mangleCXXCtorType(Ctor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000481 break;
482
483 case DeclarationName::CXXDestructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000484 if (ND == Structor)
Mike Stump141c5af2009-09-02 00:25:38 +0000485 // If the named decl is the C++ destructor we're mangling, use the type we
486 // were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000487 mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
488 else
489 // Otherwise, use the complete destructor name. This is relevant if a
490 // class with a destructor is declared within a destructor.
491 mangleCXXDtorType(Dtor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000492 break;
493
494 case DeclarationName::CXXConversionFunctionName:
Mike Stump1eb44332009-09-09 15:08:12 +0000495 // <operator-name> ::= cv <type> # (cast)
Douglas Gregor219cc612009-02-13 01:28:03 +0000496 Out << "cv";
Anders Carlssonb5404912009-10-07 01:06:45 +0000497 mangleType(Context.getASTContext().getCanonicalType(Name.getCXXNameType()));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000498 break;
499
500 case DeclarationName::CXXOperatorName:
501 mangleOperatorName(Name.getCXXOverloadedOperator(),
502 cast<FunctionDecl>(ND)->getNumParams());
503 break;
504
505 case DeclarationName::CXXUsingDirective:
506 assert(false && "Can't mangle a using directive name!");
Douglas Gregor219cc612009-02-13 01:28:03 +0000507 break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000508 }
509}
510
511void CXXNameMangler::mangleSourceName(const IdentifierInfo *II) {
512 // <source-name> ::= <positive length number> <identifier>
513 // <number> ::= [n] <non-negative decimal integer>
514 // <identifier> ::= <unqualified source code identifier>
515 Out << II->getLength() << II->getName();
516}
517
518void CXXNameMangler::mangleNestedName(const NamedDecl *ND) {
519 // <nested-name> ::= N [<CV-qualifiers>] <prefix> <unqualified-name> E
520 // ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
Anders Carlssond99edc42009-09-26 03:55:37 +0000521
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000522 Out << 'N';
523 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(ND))
John McCall0953e762009-09-24 19:53:00 +0000524 mangleQualifiers(Qualifiers::fromCVRMask(Method->getTypeQualifiers()));
Anders Carlsson7482e242009-09-18 04:29:09 +0000525
Anders Carlsson2744a062009-09-18 19:00:18 +0000526 // Check if we have a template.
527 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000528 if (const TemplateDecl *TD = isTemplate(ND, TemplateArgs)) {
Anders Carlsson2744a062009-09-18 19:00:18 +0000529 mangleTemplatePrefix(TD);
530 mangleTemplateArgumentList(*TemplateArgs);
Anders Carlsson7482e242009-09-18 04:29:09 +0000531 } else {
532 manglePrefix(ND->getDeclContext());
533 mangleUnqualifiedName(ND);
534 }
535
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000536 Out << 'E';
537}
Anders Carlsson7624f212009-09-18 02:42:01 +0000538void CXXNameMangler::mangleNestedName(const TemplateDecl *TD,
539 const TemplateArgument *TemplateArgs,
540 unsigned NumTemplateArgs) {
Anders Carlssone45117b2009-09-27 19:53:49 +0000541 // <nested-name> ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
542
Anders Carlsson7624f212009-09-18 02:42:01 +0000543 Out << 'N';
Anders Carlsson7624f212009-09-18 02:42:01 +0000544
Anders Carlssone45117b2009-09-27 19:53:49 +0000545 mangleTemplatePrefix(TD);
Anders Carlsson7624f212009-09-18 02:42:01 +0000546 mangleTemplateArgs(TemplateArgs, NumTemplateArgs);
Anders Carlssone45117b2009-09-27 19:53:49 +0000547
Anders Carlsson7624f212009-09-18 02:42:01 +0000548 Out << 'E';
549}
550
Anders Carlsson1b42c792009-04-02 16:24:45 +0000551void CXXNameMangler::mangleLocalName(const NamedDecl *ND) {
552 // <local-name> := Z <function encoding> E <entity name> [<discriminator>]
553 // := Z <function encoding> E s [<discriminator>]
Mike Stump1eb44332009-09-09 15:08:12 +0000554 // <discriminator> := _ <non-negative number>
Anders Carlsson1b42c792009-04-02 16:24:45 +0000555 Out << 'Z';
556 mangleFunctionEncoding(cast<FunctionDecl>(ND->getDeclContext()));
557 Out << 'E';
558 mangleSourceName(ND->getIdentifier());
559}
560
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000561void CXXNameMangler::manglePrefix(const DeclContext *DC) {
562 // <prefix> ::= <prefix> <unqualified-name>
563 // ::= <template-prefix> <template-args>
564 // ::= <template-param>
565 // ::= # empty
566 // ::= <substitution>
567 // FIXME: We only handle mangling of namespaces and classes at the moment.
Anders Carlsson6862fc72009-09-17 04:16:28 +0000568
Anders Carlssonadd28822009-09-22 20:33:31 +0000569 while (isa<LinkageSpecDecl>(DC))
570 DC = DC->getParent();
571
Anders Carlsson9263e912009-09-18 18:39:58 +0000572 if (DC->isTranslationUnit())
573 return;
574
Anders Carlsson6862fc72009-09-17 04:16:28 +0000575 if (mangleSubstitution(cast<NamedDecl>(DC)))
576 return;
Anders Carlsson7482e242009-09-18 04:29:09 +0000577
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000578 // Check if we have a template.
579 const TemplateArgumentList *TemplateArgs = 0;
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000580 if (const TemplateDecl *TD = isTemplate(cast<NamedDecl>(DC), TemplateArgs)) {
Anders Carlsson2ee3fca2009-09-18 20:11:09 +0000581 mangleTemplatePrefix(TD);
582 mangleTemplateArgumentList(*TemplateArgs);
583 } else {
584 manglePrefix(DC->getParent());
585 mangleUnqualifiedName(cast<NamedDecl>(DC));
586 }
Anders Carlsson6862fc72009-09-17 04:16:28 +0000587
588 addSubstitution(cast<NamedDecl>(DC));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000589}
590
Anders Carlsson0fa6df42009-09-26 19:45:45 +0000591void CXXNameMangler::mangleTemplatePrefix(const TemplateDecl *ND) {
Anders Carlsson7482e242009-09-18 04:29:09 +0000592 // <template-prefix> ::= <prefix> <template unqualified-name>
593 // ::= <template-param>
594 // ::= <substitution>
595
Anders Carlssonaeb85372009-09-26 22:18:22 +0000596 if (mangleSubstitution(ND))
597 return;
598
599 // FIXME: <template-param>
Anders Carlssonaa73ab12009-09-18 18:47:07 +0000600
601 manglePrefix(ND->getDeclContext());
Anders Carlsson1668f202009-09-26 20:13:56 +0000602 mangleUnqualifiedName(ND->getTemplatedDecl());
Anders Carlssonaeb85372009-09-26 22:18:22 +0000603
604 addSubstitution(ND);
Anders Carlsson7482e242009-09-18 04:29:09 +0000605}
606
Mike Stump1eb44332009-09-09 15:08:12 +0000607void
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000608CXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity) {
609 switch (OO) {
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000610 // <operator-name> ::= nw # new
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000611 case OO_New: Out << "nw"; break;
612 // ::= na # new[]
613 case OO_Array_New: Out << "na"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000614 // ::= dl # delete
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000615 case OO_Delete: Out << "dl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000616 // ::= da # delete[]
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000617 case OO_Array_Delete: Out << "da"; break;
618 // ::= ps # + (unary)
619 // ::= pl # +
620 case OO_Plus: Out << (Arity == 1? "ps" : "pl"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000621 // ::= ng # - (unary)
622 // ::= mi # -
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000623 case OO_Minus: Out << (Arity == 1? "ng" : "mi"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000624 // ::= ad # & (unary)
625 // ::= an # &
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000626 case OO_Amp: Out << (Arity == 1? "ad" : "an"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000627 // ::= de # * (unary)
628 // ::= ml # *
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000629 case OO_Star: Out << (Arity == 1? "de" : "ml"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000630 // ::= co # ~
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000631 case OO_Tilde: Out << "co"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000632 // ::= dv # /
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000633 case OO_Slash: Out << "dv"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000634 // ::= rm # %
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000635 case OO_Percent: Out << "rm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000636 // ::= or # |
637 case OO_Pipe: Out << "or"; break;
638 // ::= eo # ^
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000639 case OO_Caret: Out << "eo"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000640 // ::= aS # =
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000641 case OO_Equal: Out << "aS"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000642 // ::= pL # +=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000643 case OO_PlusEqual: Out << "pL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000644 // ::= mI # -=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000645 case OO_MinusEqual: Out << "mI"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000646 // ::= mL # *=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000647 case OO_StarEqual: Out << "mL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000648 // ::= dV # /=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000649 case OO_SlashEqual: Out << "dV"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000650 // ::= rM # %=
651 case OO_PercentEqual: Out << "rM"; break;
652 // ::= aN # &=
653 case OO_AmpEqual: Out << "aN"; break;
654 // ::= oR # |=
655 case OO_PipeEqual: Out << "oR"; break;
656 // ::= eO # ^=
657 case OO_CaretEqual: Out << "eO"; break;
658 // ::= ls # <<
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000659 case OO_LessLess: Out << "ls"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000660 // ::= rs # >>
661 case OO_GreaterGreater: Out << "rs"; break;
662 // ::= lS # <<=
663 case OO_LessLessEqual: Out << "lS"; break;
664 // ::= rS # >>=
665 case OO_GreaterGreaterEqual: Out << "rS"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000666 // ::= eq # ==
667 case OO_EqualEqual: Out << "eq"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000668 // ::= ne # !=
669 case OO_ExclaimEqual: Out << "ne"; break;
670 // ::= lt # <
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000671 case OO_Less: Out << "lt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000672 // ::= gt # >
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000673 case OO_Greater: Out << "gt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000674 // ::= le # <=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000675 case OO_LessEqual: Out << "le"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000676 // ::= ge # >=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000677 case OO_GreaterEqual: Out << "ge"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000678 // ::= nt # !
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000679 case OO_Exclaim: Out << "nt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000680 // ::= aa # &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000681 case OO_AmpAmp: Out << "aa"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000682 // ::= oo # ||
683 case OO_PipePipe: Out << "oo"; break;
684 // ::= pp # ++
685 case OO_PlusPlus: Out << "pp"; break;
686 // ::= mm # --
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000687 case OO_MinusMinus: Out << "mm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000688 // ::= cm # ,
689 case OO_Comma: Out << "cm"; break;
690 // ::= pm # ->*
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000691 case OO_ArrowStar: Out << "pm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000692 // ::= pt # ->
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000693 case OO_Arrow: Out << "pt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000694 // ::= cl # ()
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000695 case OO_Call: Out << "cl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000696 // ::= ix # []
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000697 case OO_Subscript: Out << "ix"; break;
698 // UNSUPPORTED: ::= qu # ?
699
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000700 case OO_None:
701 case OO_Conditional:
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000702 case NUM_OVERLOADED_OPERATORS:
Mike Stump1eb44332009-09-09 15:08:12 +0000703 assert(false && "Not an overloaded operator");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000704 break;
705 }
706}
707
John McCall0953e762009-09-24 19:53:00 +0000708void CXXNameMangler::mangleQualifiers(Qualifiers Quals) {
Mike Stump1eb44332009-09-09 15:08:12 +0000709 // <CV-qualifiers> ::= [r] [V] [K] # restrict (C99), volatile, const
John McCall0953e762009-09-24 19:53:00 +0000710 if (Quals.hasRestrict())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000711 Out << 'r';
John McCall0953e762009-09-24 19:53:00 +0000712 if (Quals.hasVolatile())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000713 Out << 'V';
John McCall0953e762009-09-24 19:53:00 +0000714 if (Quals.hasConst())
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000715 Out << 'K';
John McCall0953e762009-09-24 19:53:00 +0000716
717 // FIXME: For now, just drop all extension qualifiers on the floor.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000718}
719
720void CXXNameMangler::mangleType(QualType T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000721 // Only operate on the canonical type!
Anders Carlssonb5404912009-10-07 01:06:45 +0000722 T = Context.getASTContext().getCanonicalType(T);
Anders Carlsson4843e582009-03-10 17:07:44 +0000723
Anders Carlsson410f7642009-10-16 02:06:06 +0000724 bool IsSubstitutable = T.hasQualifiers() || !isa<BuiltinType>(T);
Anders Carlsson76967372009-09-17 00:43:46 +0000725 if (IsSubstitutable && mangleSubstitution(T))
726 return;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000727
John McCall0953e762009-09-24 19:53:00 +0000728 if (Qualifiers Quals = T.getQualifiers()) {
729 mangleQualifiers(Quals);
730 // Recurse: even if the qualified type isn't yet substitutable,
731 // the unqualified type might be.
Anders Carlsson76967372009-09-17 00:43:46 +0000732 mangleType(T.getUnqualifiedType());
733 } else {
734 switch (T->getTypeClass()) {
John McCallefe6aee2009-09-05 07:56:18 +0000735#define ABSTRACT_TYPE(CLASS, PARENT)
736#define NON_CANONICAL_TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000737 case Type::CLASS: \
738 llvm::llvm_unreachable("can't mangle non-canonical type " #CLASS "Type"); \
739 return;
John McCallefe6aee2009-09-05 07:56:18 +0000740#define TYPE(CLASS, PARENT) \
Anders Carlsson76967372009-09-17 00:43:46 +0000741 case Type::CLASS: \
John McCall0953e762009-09-24 19:53:00 +0000742 mangleType(static_cast<const CLASS##Type*>(T.getTypePtr())); \
Anders Carlsson76967372009-09-17 00:43:46 +0000743 break;
John McCallefe6aee2009-09-05 07:56:18 +0000744#include "clang/AST/TypeNodes.def"
Anders Carlsson76967372009-09-17 00:43:46 +0000745 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000746 }
Anders Carlsson76967372009-09-17 00:43:46 +0000747
748 // Add the substitution.
749 if (IsSubstitutable)
750 addSubstitution(T);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000751}
752
753void CXXNameMangler::mangleType(const BuiltinType *T) {
John McCallefe6aee2009-09-05 07:56:18 +0000754 // <type> ::= <builtin-type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000755 // <builtin-type> ::= v # void
756 // ::= w # wchar_t
757 // ::= b # bool
758 // ::= c # char
759 // ::= a # signed char
760 // ::= h # unsigned char
761 // ::= s # short
762 // ::= t # unsigned short
763 // ::= i # int
764 // ::= j # unsigned int
765 // ::= l # long
766 // ::= m # unsigned long
767 // ::= x # long long, __int64
768 // ::= y # unsigned long long, __int64
769 // ::= n # __int128
770 // UNSUPPORTED: ::= o # unsigned __int128
771 // ::= f # float
772 // ::= d # double
773 // ::= e # long double, __float80
774 // UNSUPPORTED: ::= g # __float128
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000775 // UNSUPPORTED: ::= Dd # IEEE 754r decimal floating point (64 bits)
776 // UNSUPPORTED: ::= De # IEEE 754r decimal floating point (128 bits)
777 // UNSUPPORTED: ::= Df # IEEE 754r decimal floating point (32 bits)
778 // UNSUPPORTED: ::= Dh # IEEE 754r half-precision floating point (16 bits)
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000779 // ::= Di # char32_t
780 // ::= Ds # char16_t
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000781 // ::= u <source-name> # vendor extended type
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000782 // From our point of view, std::nullptr_t is a builtin, but as far as mangling
783 // is concerned, it's a type called std::nullptr_t.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000784 switch (T->getKind()) {
785 case BuiltinType::Void: Out << 'v'; break;
786 case BuiltinType::Bool: Out << 'b'; break;
787 case BuiltinType::Char_U: case BuiltinType::Char_S: Out << 'c'; break;
788 case BuiltinType::UChar: Out << 'h'; break;
789 case BuiltinType::UShort: Out << 't'; break;
790 case BuiltinType::UInt: Out << 'j'; break;
791 case BuiltinType::ULong: Out << 'm'; break;
792 case BuiltinType::ULongLong: Out << 'y'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000793 case BuiltinType::UInt128: Out << 'o'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000794 case BuiltinType::SChar: Out << 'a'; break;
795 case BuiltinType::WChar: Out << 'w'; break;
Alisdair Meredithf5c209d2009-07-14 06:30:34 +0000796 case BuiltinType::Char16: Out << "Ds"; break;
797 case BuiltinType::Char32: Out << "Di"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000798 case BuiltinType::Short: Out << 's'; break;
799 case BuiltinType::Int: Out << 'i'; break;
800 case BuiltinType::Long: Out << 'l'; break;
801 case BuiltinType::LongLong: Out << 'x'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000802 case BuiltinType::Int128: Out << 'n'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000803 case BuiltinType::Float: Out << 'f'; break;
804 case BuiltinType::Double: Out << 'd'; break;
805 case BuiltinType::LongDouble: Out << 'e'; break;
Sebastian Redl6e8ed162009-05-10 18:38:11 +0000806 case BuiltinType::NullPtr: Out << "St9nullptr_t"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000807
808 case BuiltinType::Overload:
809 case BuiltinType::Dependent:
Mike Stump1eb44332009-09-09 15:08:12 +0000810 assert(false &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000811 "Overloaded and dependent types shouldn't get to name mangling");
812 break;
Anders Carlssone89d1592009-06-26 18:41:36 +0000813 case BuiltinType::UndeducedAuto:
814 assert(0 && "Should not see undeduced auto here");
815 break;
Steve Naroff9533a7f2009-07-22 17:14:51 +0000816 case BuiltinType::ObjCId: Out << "11objc_object"; break;
817 case BuiltinType::ObjCClass: Out << "10objc_class"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000818 }
819}
820
John McCallefe6aee2009-09-05 07:56:18 +0000821// <type> ::= <function-type>
822// <function-type> ::= F [Y] <bare-function-type> E
823void CXXNameMangler::mangleType(const FunctionProtoType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000824 Out << 'F';
Mike Stumpf5408fe2009-05-16 07:57:57 +0000825 // FIXME: We don't have enough information in the AST to produce the 'Y'
826 // encoding for extern "C" function types.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000827 mangleBareFunctionType(T, /*MangleReturnType=*/true);
828 Out << 'E';
829}
John McCallefe6aee2009-09-05 07:56:18 +0000830void CXXNameMangler::mangleType(const FunctionNoProtoType *T) {
831 llvm::llvm_unreachable("Can't mangle K&R function prototypes");
832}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000833void CXXNameMangler::mangleBareFunctionType(const FunctionType *T,
834 bool MangleReturnType) {
John McCallefe6aee2009-09-05 07:56:18 +0000835 // We should never be mangling something without a prototype.
836 const FunctionProtoType *Proto = cast<FunctionProtoType>(T);
837
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000838 // <bare-function-type> ::= <signature type>+
839 if (MangleReturnType)
John McCallefe6aee2009-09-05 07:56:18 +0000840 mangleType(Proto->getResultType());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000841
Anders Carlssonc6c91bc2009-04-01 00:15:23 +0000842 if (Proto->getNumArgs() == 0) {
843 Out << 'v';
844 return;
845 }
Mike Stump1eb44332009-09-09 15:08:12 +0000846
Douglas Gregor72564e72009-02-26 23:50:07 +0000847 for (FunctionProtoType::arg_type_iterator Arg = Proto->arg_type_begin(),
Mike Stump1eb44332009-09-09 15:08:12 +0000848 ArgEnd = Proto->arg_type_end();
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000849 Arg != ArgEnd; ++Arg)
850 mangleType(*Arg);
Douglas Gregor219cc612009-02-13 01:28:03 +0000851
852 // <builtin-type> ::= z # ellipsis
853 if (Proto->isVariadic())
854 Out << 'z';
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000855}
856
John McCallefe6aee2009-09-05 07:56:18 +0000857// <type> ::= <class-enum-type>
Mike Stump1eb44332009-09-09 15:08:12 +0000858// <class-enum-type> ::= <name>
John McCallefe6aee2009-09-05 07:56:18 +0000859void CXXNameMangler::mangleType(const EnumType *T) {
860 mangleType(static_cast<const TagType*>(T));
861}
862void CXXNameMangler::mangleType(const RecordType *T) {
863 mangleType(static_cast<const TagType*>(T));
864}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000865void CXXNameMangler::mangleType(const TagType *T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000866 if (!T->getDecl()->getIdentifier())
867 mangleName(T->getDecl()->getTypedefForAnonDecl());
868 else
869 mangleName(T->getDecl());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000870}
871
John McCallefe6aee2009-09-05 07:56:18 +0000872// <type> ::= <array-type>
873// <array-type> ::= A <positive dimension number> _ <element type>
874// ::= A [<dimension expression>] _ <element type>
875void CXXNameMangler::mangleType(const ConstantArrayType *T) {
876 Out << 'A' << T->getSize() << '_';
877 mangleType(T->getElementType());
878}
879void CXXNameMangler::mangleType(const VariableArrayType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000880 Out << 'A';
John McCallefe6aee2009-09-05 07:56:18 +0000881 mangleExpression(T->getSizeExpr());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000882 Out << '_';
883 mangleType(T->getElementType());
884}
John McCallefe6aee2009-09-05 07:56:18 +0000885void CXXNameMangler::mangleType(const DependentSizedArrayType *T) {
886 Out << 'A';
887 mangleExpression(T->getSizeExpr());
888 Out << '_';
889 mangleType(T->getElementType());
890}
891void CXXNameMangler::mangleType(const IncompleteArrayType *T) {
892 Out << 'A' << '_';
893 mangleType(T->getElementType());
894}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000895
John McCallefe6aee2009-09-05 07:56:18 +0000896// <type> ::= <pointer-to-member-type>
897// <pointer-to-member-type> ::= M <class type> <member type>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000898void CXXNameMangler::mangleType(const MemberPointerType *T) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000899 Out << 'M';
900 mangleType(QualType(T->getClass(), 0));
Anders Carlsson0e650012009-05-17 17:41:20 +0000901 QualType PointeeType = T->getPointeeType();
902 if (const FunctionProtoType *FPT = dyn_cast<FunctionProtoType>(PointeeType)) {
John McCall0953e762009-09-24 19:53:00 +0000903 mangleQualifiers(Qualifiers::fromCVRMask(FPT->getTypeQuals()));
Anders Carlsson0e650012009-05-17 17:41:20 +0000904 mangleType(FPT);
Mike Stump1eb44332009-09-09 15:08:12 +0000905 } else
Anders Carlsson0e650012009-05-17 17:41:20 +0000906 mangleType(PointeeType);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000907}
908
John McCallefe6aee2009-09-05 07:56:18 +0000909// <type> ::= <template-param>
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000910void CXXNameMangler::mangleType(const TemplateTypeParmType *T) {
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +0000911 mangleTemplateParameter(T->getIndex());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000912}
913
John McCallefe6aee2009-09-05 07:56:18 +0000914// FIXME: <type> ::= <template-template-param> <template-args>
John McCallefe6aee2009-09-05 07:56:18 +0000915
916// <type> ::= P <type> # pointer-to
917void CXXNameMangler::mangleType(const PointerType *T) {
918 Out << 'P';
919 mangleType(T->getPointeeType());
920}
921void CXXNameMangler::mangleType(const ObjCObjectPointerType *T) {
922 Out << 'P';
923 mangleType(T->getPointeeType());
924}
925
926// <type> ::= R <type> # reference-to
927void CXXNameMangler::mangleType(const LValueReferenceType *T) {
928 Out << 'R';
929 mangleType(T->getPointeeType());
930}
931
932// <type> ::= O <type> # rvalue reference-to (C++0x)
933void CXXNameMangler::mangleType(const RValueReferenceType *T) {
934 Out << 'O';
935 mangleType(T->getPointeeType());
936}
937
938// <type> ::= C <type> # complex pair (C 2000)
939void CXXNameMangler::mangleType(const ComplexType *T) {
940 Out << 'C';
941 mangleType(T->getElementType());
942}
943
944// GNU extension: vector types
945void CXXNameMangler::mangleType(const VectorType *T) {
946 Out << "U8__vector";
947 mangleType(T->getElementType());
948}
949void CXXNameMangler::mangleType(const ExtVectorType *T) {
950 mangleType(static_cast<const VectorType*>(T));
951}
952void CXXNameMangler::mangleType(const DependentSizedExtVectorType *T) {
953 Out << "U8__vector";
954 mangleType(T->getElementType());
955}
956
Anders Carlssona40c5e42009-03-07 22:03:21 +0000957void CXXNameMangler::mangleType(const ObjCInterfaceType *T) {
958 mangleSourceName(T->getDecl()->getIdentifier());
959}
960
John McCallefe6aee2009-09-05 07:56:18 +0000961void CXXNameMangler::mangleType(const BlockPointerType *T) {
962 assert(false && "can't mangle block pointer types yet");
963}
964
965void CXXNameMangler::mangleType(const FixedWidthIntType *T) {
966 assert(false && "can't mangle arbitary-precision integer type yet");
967}
968
969void CXXNameMangler::mangleType(const TemplateSpecializationType *T) {
Anders Carlsson7624f212009-09-18 02:42:01 +0000970 TemplateDecl *TD = T->getTemplateName().getAsTemplateDecl();
971 assert(TD && "FIXME: Support dependent template names!");
972
973 mangleName(TD, T->getArgs(), T->getNumArgs());
John McCallefe6aee2009-09-05 07:56:18 +0000974}
975
976void CXXNameMangler::mangleType(const TypenameType *T) {
Anders Carlssonae352482009-09-26 02:26:02 +0000977 // Typename types are always nested
978 Out << 'N';
979
980 const Type *QTy = T->getQualifier()->getAsType();
981 if (const TemplateSpecializationType *TST =
982 dyn_cast<TemplateSpecializationType>(QTy)) {
Anders Carlsson88599172009-09-27 01:06:07 +0000983 if (!mangleSubstitution(QualType(TST, 0))) {
984 TemplateDecl *TD = TST->getTemplateName().getAsTemplateDecl();
Anders Carlssonae352482009-09-26 02:26:02 +0000985
Anders Carlsson88599172009-09-27 01:06:07 +0000986 mangleTemplatePrefix(TD);
987 mangleTemplateArgs(TST->getArgs(), TST->getNumArgs());
988 addSubstitution(QualType(TST, 0));
989 }
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +0000990 } else if (const TemplateTypeParmType *TTPT =
991 dyn_cast<TemplateTypeParmType>(QTy)) {
992 // We use the QualType mangle type variant here because it handles
993 // substitutions.
994 mangleType(QualType(TTPT, 0));
Anders Carlssonae352482009-09-26 02:26:02 +0000995 } else
996 assert(false && "Unhandled type!");
997
998 mangleSourceName(T->getIdentifier());
999
1000 Out << 'E';
John McCallefe6aee2009-09-05 07:56:18 +00001001}
1002
Anders Carlssond553f8c2009-09-21 01:21:10 +00001003void CXXNameMangler::mangleExpression(const Expr *E) {
1004 // <expression> ::= <unary operator-name> <expression>
1005 // ::= <binary operator-name> <expression> <expression>
1006 // ::= <trinary operator-name> <expression> <expression> <expression>
1007 // ::= cl <expression>* E # call
1008 // ::= cv <type> expression # conversion with one argument
1009 // ::= cv <type> _ <expression>* E # conversion with a different number of arguments
1010 // ::= st <type> # sizeof (a type)
1011 // ::= at <type> # alignof (a type)
1012 // ::= <template-param>
1013 // ::= <function-param>
1014 // ::= sr <type> <unqualified-name> # dependent name
1015 // ::= sr <type> <unqualified-name> <template-args> # dependent template-id
1016 // ::= sZ <template-param> # size of a parameter pack
1017 // ::= <expr-primary>
1018 switch (E->getStmtClass()) {
1019 default: assert(false && "Unhandled expression kind!");
Anders Carlssona7694082009-11-06 02:50:19 +00001020
1021 case Expr::ParenExprClass:
1022 mangleExpression(cast<ParenExpr>(E)->getSubExpr());
1023 break;
1024
Anders Carlssond553f8c2009-09-21 01:21:10 +00001025 case Expr::DeclRefExprClass: {
1026 const Decl *D = cast<DeclRefExpr>(E)->getDecl();
1027
1028 switch (D->getKind()) {
1029 default: assert(false && "Unhandled decl kind!");
1030 case Decl::NonTypeTemplateParm: {
1031 const NonTypeTemplateParmDecl *PD = cast<NonTypeTemplateParmDecl>(D);
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001032 mangleTemplateParameter(PD->getIndex());
Anders Carlssond553f8c2009-09-21 01:21:10 +00001033 break;
1034 }
1035
1036 }
Anders Carlsson50755b02009-09-27 20:11:34 +00001037
1038 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001039 }
Anders Carlsson50755b02009-09-27 20:11:34 +00001040
1041 case Expr::UnresolvedDeclRefExprClass: {
1042 const UnresolvedDeclRefExpr *DRE = cast<UnresolvedDeclRefExpr>(E);
1043 const Type *QTy = DRE->getQualifier()->getAsType();
1044 assert(QTy && "Qualifier was not type!");
1045
1046 // ::= sr <type> <unqualified-name> # dependent name
1047 Out << "sr";
1048 mangleType(QualType(QTy, 0));
1049
1050 assert(DRE->getDeclName().getNameKind() == DeclarationName::Identifier &&
1051 "Unhandled decl name kind!");
1052 mangleSourceName(DRE->getDeclName().getAsIdentifierInfo());
1053
1054 break;
1055 }
1056
Anders Carlssond553f8c2009-09-21 01:21:10 +00001057 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001058}
1059
John McCallefe6aee2009-09-05 07:56:18 +00001060// FIXME: <type> ::= G <type> # imaginary (C 2000)
1061// FIXME: <type> ::= U <source-name> <type> # vendor extended type qualifier
1062
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001063void CXXNameMangler::mangleCXXCtorType(CXXCtorType T) {
1064 // <ctor-dtor-name> ::= C1 # complete object constructor
1065 // ::= C2 # base object constructor
1066 // ::= C3 # complete object allocating constructor
1067 //
1068 switch (T) {
1069 case Ctor_Complete:
1070 Out << "C1";
1071 break;
1072 case Ctor_Base:
1073 Out << "C2";
1074 break;
1075 case Ctor_CompleteAllocating:
1076 Out << "C3";
1077 break;
1078 }
1079}
1080
Anders Carlsson27ae5362009-04-17 01:58:57 +00001081void CXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
1082 // <ctor-dtor-name> ::= D0 # deleting destructor
1083 // ::= D1 # complete object destructor
1084 // ::= D2 # base object destructor
1085 //
1086 switch (T) {
1087 case Dtor_Deleting:
1088 Out << "D0";
1089 break;
1090 case Dtor_Complete:
1091 Out << "D1";
1092 break;
1093 case Dtor_Base:
1094 Out << "D2";
1095 break;
1096 }
1097}
1098
Anders Carlsson068f3472009-09-17 05:31:47 +00001099void CXXNameMangler::mangleTemplateArgumentList(const TemplateArgumentList &L) {
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001100 // <template-args> ::= I <template-arg>+ E
1101 Out << "I";
Mike Stump1eb44332009-09-09 15:08:12 +00001102
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001103 for (unsigned i = 0, e = L.size(); i != e; ++i) {
1104 const TemplateArgument &A = L[i];
Mike Stump1eb44332009-09-09 15:08:12 +00001105
Anders Carlsson068f3472009-09-17 05:31:47 +00001106 mangleTemplateArgument(A);
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001107 }
Mike Stump1eb44332009-09-09 15:08:12 +00001108
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001109 Out << "E";
1110}
1111
Anders Carlsson7624f212009-09-18 02:42:01 +00001112void CXXNameMangler::mangleTemplateArgs(const TemplateArgument *TemplateArgs,
1113 unsigned NumTemplateArgs) {
1114 // <template-args> ::= I <template-arg>+ E
1115 Out << "I";
1116
1117 for (unsigned i = 0; i != NumTemplateArgs; ++i) {
1118 mangleTemplateArgument(TemplateArgs[i]);
1119 }
1120
1121 Out << "E";
1122}
1123
Anders Carlsson068f3472009-09-17 05:31:47 +00001124void CXXNameMangler::mangleTemplateArgument(const TemplateArgument &A) {
Mike Stump1eb44332009-09-09 15:08:12 +00001125 // <template-arg> ::= <type> # type or template
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001126 // ::= X <expression> E # expression
1127 // ::= <expr-primary> # simple expressions
1128 // ::= I <template-arg>* E # argument pack
1129 // ::= sp <expression> # pack expansion of (C++0x)
1130 switch (A.getKind()) {
1131 default:
1132 assert(0 && "Unknown template argument kind!");
1133 case TemplateArgument::Type:
1134 mangleType(A.getAsType());
1135 break;
Anders Carlssond553f8c2009-09-21 01:21:10 +00001136 case TemplateArgument::Expression:
1137 Out << 'X';
1138 mangleExpression(A.getAsExpr());
1139 Out << 'E';
1140 break;
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001141 case TemplateArgument::Integral:
1142 // <expr-primary> ::= L <type> <value number> E # integer literal
1143
1144 Out << 'L';
Mike Stump1eb44332009-09-09 15:08:12 +00001145
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001146 mangleType(A.getIntegralType());
Mike Stump1eb44332009-09-09 15:08:12 +00001147
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001148 const llvm::APSInt *Integral = A.getAsIntegral();
1149 if (A.getIntegralType()->isBooleanType()) {
1150 // Boolean values are encoded as 0/1.
1151 Out << (Integral->getBoolValue() ? '1' : '0');
1152 } else {
1153 if (Integral->isNegative())
1154 Out << 'n';
1155 Integral->abs().print(Out, false);
1156 }
Mike Stump1eb44332009-09-09 15:08:12 +00001157
Anders Carlsson7a0ba872009-05-15 16:09:15 +00001158 Out << 'E';
1159 break;
1160 }
1161}
1162
Anders Carlsson0ccdf8d2009-09-27 00:38:53 +00001163void CXXNameMangler::mangleTemplateParameter(unsigned Index) {
1164 // <template-param> ::= T_ # first template parameter
1165 // ::= T <parameter-2 non-negative number> _
1166 if (Index == 0)
1167 Out << "T_";
1168 else
1169 Out << 'T' << (Index - 1) << '_';
1170}
1171
Anders Carlsson76967372009-09-17 00:43:46 +00001172// <substitution> ::= S <seq-id> _
1173// ::= S_
Anders Carlsson6862fc72009-09-17 04:16:28 +00001174bool CXXNameMangler::mangleSubstitution(const NamedDecl *ND) {
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001175 // Try one of the standard substitutions first.
1176 if (mangleStandardSubstitution(ND))
1177 return true;
1178
Anders Carlsson433d1372009-11-07 04:26:04 +00001179 ND = cast<NamedDecl>(ND->getCanonicalDecl());
Anders Carlsson6862fc72009-09-17 04:16:28 +00001180 return mangleSubstitution(reinterpret_cast<uintptr_t>(ND));
1181}
1182
Anders Carlsson76967372009-09-17 00:43:46 +00001183bool CXXNameMangler::mangleSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001184 if (!T.getCVRQualifiers()) {
1185 if (const RecordType *RT = T->getAs<RecordType>())
1186 return mangleSubstitution(RT->getDecl());
1187 }
1188
Anders Carlsson76967372009-09-17 00:43:46 +00001189 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
1190
Anders Carlssond3a932a2009-09-17 03:53:28 +00001191 return mangleSubstitution(TypePtr);
1192}
1193
1194bool CXXNameMangler::mangleSubstitution(uintptr_t Ptr) {
Anders Carlsson76967372009-09-17 00:43:46 +00001195 llvm::DenseMap<uintptr_t, unsigned>::iterator I =
Anders Carlssond3a932a2009-09-17 03:53:28 +00001196 Substitutions.find(Ptr);
Anders Carlsson76967372009-09-17 00:43:46 +00001197 if (I == Substitutions.end())
1198 return false;
1199
1200 unsigned SeqID = I->second;
1201 if (SeqID == 0)
1202 Out << "S_";
1203 else {
1204 SeqID--;
1205
1206 // <seq-id> is encoded in base-36, using digits and upper case letters.
1207 char Buffer[10];
1208 char *BufferPtr = Buffer + 9;
1209
1210 *BufferPtr = 0;
1211 if (SeqID == 0) *--BufferPtr = '0';
1212
1213 while (SeqID) {
1214 assert(BufferPtr > Buffer && "Buffer overflow!");
1215
1216 unsigned char c = static_cast<unsigned char>(SeqID) % 36;
1217
1218 *--BufferPtr = (c < 10 ? '0' + c : 'A' + c - 10);
1219 SeqID /= 36;
1220 }
1221
1222 Out << 'S' << BufferPtr << '_';
1223 }
1224
1225 return true;
1226}
1227
Anders Carlssonf514b542009-09-27 00:12:57 +00001228static bool isCharType(QualType T) {
1229 if (T.isNull())
1230 return false;
1231
1232 return T->isSpecificBuiltinType(BuiltinType::Char_S) ||
1233 T->isSpecificBuiltinType(BuiltinType::Char_U);
1234}
1235
1236/// isCharSpecialization - Returns whether a given type is a template
1237/// specialization of a given name with a single argument of type char.
1238static bool isCharSpecialization(QualType T, const char *Name) {
1239 if (T.isNull())
1240 return false;
1241
1242 const RecordType *RT = T->getAs<RecordType>();
1243 if (!RT)
1244 return false;
1245
1246 const ClassTemplateSpecializationDecl *SD =
1247 dyn_cast<ClassTemplateSpecializationDecl>(RT->getDecl());
1248 if (!SD)
1249 return false;
1250
1251 if (!isStdNamespace(SD->getDeclContext()))
1252 return false;
1253
1254 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1255 if (TemplateArgs.size() != 1)
1256 return false;
1257
1258 if (!isCharType(TemplateArgs[0].getAsType()))
1259 return false;
1260
Daniel Dunbar01eb9b92009-10-18 21:17:35 +00001261 return SD->getIdentifier()->getName() == Name;
Anders Carlssonf514b542009-09-27 00:12:57 +00001262}
1263
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001264bool CXXNameMangler::mangleStandardSubstitution(const NamedDecl *ND) {
1265 // <substitution> ::= St # ::std::
Anders Carlsson8c031552009-09-26 23:10:05 +00001266 if (const NamespaceDecl *NS = dyn_cast<NamespaceDecl>(ND)) {
John McCall9aeed322009-10-01 00:25:31 +00001267 if (isStdNamespace(NS)) {
Anders Carlsson8c031552009-09-26 23:10:05 +00001268 Out << "St";
1269 return true;
1270 }
1271 }
1272
1273 if (const ClassTemplateDecl *TD = dyn_cast<ClassTemplateDecl>(ND)) {
1274 if (!isStdNamespace(TD->getDeclContext()))
1275 return false;
1276
1277 // <substitution> ::= Sa # ::std::allocator
1278 if (TD->getIdentifier()->isStr("allocator")) {
1279 Out << "Sa";
1280 return true;
1281 }
Anders Carlsson189d59c2009-09-26 23:14:39 +00001282
1283 // <<substitution> ::= Sb # ::std::basic_string
1284 if (TD->getIdentifier()->isStr("basic_string")) {
1285 Out << "Sb";
1286 return true;
1287 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001288 }
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001289
Anders Carlssonf514b542009-09-27 00:12:57 +00001290 if (const ClassTemplateSpecializationDecl *SD =
1291 dyn_cast<ClassTemplateSpecializationDecl>(ND)) {
1292 // <substitution> ::= Ss # ::std::basic_string<char,
1293 // ::std::char_traits<char>,
1294 // ::std::allocator<char> >
1295 if (SD->getIdentifier()->isStr("basic_string")) {
1296 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1297
1298 if (TemplateArgs.size() != 3)
1299 return false;
1300
1301 if (!isCharType(TemplateArgs[0].getAsType()))
1302 return false;
1303
1304 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1305 return false;
1306
1307 if (!isCharSpecialization(TemplateArgs[2].getAsType(), "allocator"))
1308 return false;
1309
1310 Out << "Ss";
1311 return true;
1312 }
Anders Carlsson8f8fd8e2009-10-08 17:20:26 +00001313
1314 // <substitution> ::= So # ::std::basic_ostream<char,
1315 // ::std::char_traits<char> >
1316 if (SD->getIdentifier()->isStr("basic_ostream")) {
1317 const TemplateArgumentList &TemplateArgs = SD->getTemplateArgs();
1318
1319 if (TemplateArgs.size() != 2)
1320 return false;
1321
1322 if (!isCharType(TemplateArgs[0].getAsType()))
1323 return false;
1324
1325 if (!isCharSpecialization(TemplateArgs[1].getAsType(), "char_traits"))
1326 return false;
1327
1328 Out << "So";
1329 return true;
1330 }
Anders Carlssonf514b542009-09-27 00:12:57 +00001331 }
Anders Carlsson8c031552009-09-26 23:10:05 +00001332 return false;
Anders Carlssone7c8cb62009-09-26 20:53:44 +00001333}
1334
Anders Carlsson76967372009-09-17 00:43:46 +00001335void CXXNameMangler::addSubstitution(QualType T) {
Anders Carlssond99edc42009-09-26 03:55:37 +00001336 if (!T.getCVRQualifiers()) {
1337 if (const RecordType *RT = T->getAs<RecordType>()) {
1338 addSubstitution(RT->getDecl());
1339 return;
1340 }
1341 }
1342
Anders Carlsson76967372009-09-17 00:43:46 +00001343 uintptr_t TypePtr = reinterpret_cast<uintptr_t>(T.getAsOpaquePtr());
Anders Carlssond3a932a2009-09-17 03:53:28 +00001344 addSubstitution(TypePtr);
1345}
1346
1347void CXXNameMangler::addSubstitution(uintptr_t Ptr) {
Anders Carlsson76967372009-09-17 00:43:46 +00001348 unsigned SeqID = Substitutions.size();
1349
Anders Carlssond3a932a2009-09-17 03:53:28 +00001350 assert(!Substitutions.count(Ptr) && "Substitution already exists!");
1351 Substitutions[Ptr] = SeqID;
Anders Carlsson76967372009-09-17 00:43:46 +00001352}
1353
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001354namespace clang {
Mike Stump141c5af2009-09-02 00:25:38 +00001355 /// \brief Mangles the name of the declaration D and emits that name to the
1356 /// given output stream.
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001357 ///
Mike Stump141c5af2009-09-02 00:25:38 +00001358 /// If the declaration D requires a mangled name, this routine will emit that
1359 /// mangled name to \p os and return true. Otherwise, \p os will be unchanged
1360 /// and this routine will return false. In this case, the caller should just
1361 /// emit the identifier of the declaration (\c D->getIdentifier()) as its
1362 /// name.
Anders Carlssonb5404912009-10-07 01:06:45 +00001363 bool mangleName(MangleContext &Context, const NamedDecl *D,
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001364 llvm::raw_ostream &os) {
Anders Carlsson578aa642009-05-03 16:51:04 +00001365 assert(!isa<CXXConstructorDecl>(D) &&
1366 "Use mangleCXXCtor for constructor decls!");
1367 assert(!isa<CXXDestructorDecl>(D) &&
1368 "Use mangleCXXDtor for destructor decls!");
Mike Stump1eb44332009-09-09 15:08:12 +00001369
Anders Carlssond99edc42009-09-26 03:55:37 +00001370 PrettyStackTraceDecl CrashInfo(const_cast<NamedDecl *>(D), SourceLocation(),
Anders Carlssonb5404912009-10-07 01:06:45 +00001371 Context.getASTContext().getSourceManager(),
Anders Carlssond99edc42009-09-26 03:55:37 +00001372 "Mangling declaration");
1373
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001374 CXXNameMangler Mangler(Context, os);
Douglas Gregor457e2812009-10-28 16:31:34 +00001375 if (!Mangler.mangle(D))
Douglas Gregor6ec36682009-02-18 23:53:56 +00001376 return false;
Mike Stump1eb44332009-09-09 15:08:12 +00001377
Douglas Gregor6ec36682009-02-18 23:53:56 +00001378 os.flush();
1379 return true;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001380 }
Mike Stump1eb44332009-09-09 15:08:12 +00001381
Mike Stump141c5af2009-09-02 00:25:38 +00001382 /// \brief Mangles the a thunk with the offset n for the declaration D and
1383 /// emits that name to the given output stream.
Anders Carlssonb5404912009-10-07 01:06:45 +00001384 void mangleThunk(MangleContext &Context, const FunctionDecl *FD,
1385 int64_t nv, int64_t v, llvm::raw_ostream &os) {
Mike Stump141c5af2009-09-02 00:25:38 +00001386 // FIXME: Hum, we might have to thunk these, fix.
Mike Stumpdec025b2009-09-07 04:27:52 +00001387 assert(!isa<CXXDestructorDecl>(FD) &&
Mike Stump141c5af2009-09-02 00:25:38 +00001388 "Use mangleCXXDtor for destructor decls!");
Mike Stump1eb44332009-09-09 15:08:12 +00001389
Mike Stump141c5af2009-09-02 00:25:38 +00001390 CXXNameMangler Mangler(Context, os);
Mike Stumpdec025b2009-09-07 04:27:52 +00001391 Mangler.mangleThunk(FD, nv, v);
Mike Stump141c5af2009-09-02 00:25:38 +00001392 os.flush();
1393 }
Mike Stump1eb44332009-09-09 15:08:12 +00001394
Mike Stump9124bcc2009-09-02 00:56:18 +00001395 /// \brief Mangles the a covariant thunk for the declaration D and emits that
1396 /// name to the given output stream.
Anders Carlssonb5404912009-10-07 01:06:45 +00001397 void mangleCovariantThunk(MangleContext &Context, const FunctionDecl *FD,
1398 int64_t nv_t, int64_t v_t,
1399 int64_t nv_r, int64_t v_r,
Mike Stump9124bcc2009-09-02 00:56:18 +00001400 llvm::raw_ostream &os) {
1401 // FIXME: Hum, we might have to thunk these, fix.
Mike Stumpdec025b2009-09-07 04:27:52 +00001402 assert(!isa<CXXDestructorDecl>(FD) &&
Mike Stump9124bcc2009-09-02 00:56:18 +00001403 "Use mangleCXXDtor for destructor decls!");
Mike Stump1eb44332009-09-09 15:08:12 +00001404
Mike Stump9124bcc2009-09-02 00:56:18 +00001405 CXXNameMangler Mangler(Context, os);
Mike Stumpdec025b2009-09-07 04:27:52 +00001406 Mangler.mangleCovariantThunk(FD, nv_t, v_t, nv_r, v_r);
Mike Stump9124bcc2009-09-02 00:56:18 +00001407 os.flush();
1408 }
Mike Stump1eb44332009-09-09 15:08:12 +00001409
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001410 /// mangleGuardVariable - Returns the mangled name for a guard variable
1411 /// for the passed in VarDecl.
Anders Carlssonb5404912009-10-07 01:06:45 +00001412 void mangleGuardVariable(MangleContext &Context, const VarDecl *D,
Anders Carlsson41aa8c12009-04-13 18:02:10 +00001413 llvm::raw_ostream &os) {
1414 CXXNameMangler Mangler(Context, os);
1415 Mangler.mangleGuardVariable(D);
1416
1417 os.flush();
1418 }
Mike Stump1eb44332009-09-09 15:08:12 +00001419
Anders Carlssonb5404912009-10-07 01:06:45 +00001420 void mangleCXXCtor(MangleContext &Context, const CXXConstructorDecl *D,
1421 CXXCtorType Type, llvm::raw_ostream &os) {
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001422 CXXNameMangler Mangler(Context, os);
1423 Mangler.mangleCXXCtor(D, Type);
Mike Stump1eb44332009-09-09 15:08:12 +00001424
Anders Carlsson3ac86b52009-04-15 05:36:58 +00001425 os.flush();
1426 }
Mike Stump1eb44332009-09-09 15:08:12 +00001427
Anders Carlssonb5404912009-10-07 01:06:45 +00001428 void mangleCXXDtor(MangleContext &Context, const CXXDestructorDecl *D,
1429 CXXDtorType Type, llvm::raw_ostream &os) {
Anders Carlsson27ae5362009-04-17 01:58:57 +00001430 CXXNameMangler Mangler(Context, os);
1431 Mangler.mangleCXXDtor(D, Type);
Mike Stump1eb44332009-09-09 15:08:12 +00001432
Anders Carlsson27ae5362009-04-17 01:58:57 +00001433 os.flush();
1434 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +00001435
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +00001436 void mangleCXXVtable(MangleContext &Context, const CXXRecordDecl *RD,
Mike Stumpf1216772009-07-31 18:25:34 +00001437 llvm::raw_ostream &os) {
1438 CXXNameMangler Mangler(Context, os);
Anders Carlssonfc3bf4d2009-10-11 21:24:51 +00001439 Mangler.mangleCXXVtable(RD);
Mike Stumpf1216772009-07-31 18:25:34 +00001440
1441 os.flush();
1442 }
Mike Stump738f8c22009-07-31 23:15:31 +00001443
Anders Carlssond313e402009-10-30 01:52:02 +00001444 void mangleCXXRtti(MangleContext &Context, QualType Ty,
Mike Stump738f8c22009-07-31 23:15:31 +00001445 llvm::raw_ostream &os) {
1446 CXXNameMangler Mangler(Context, os);
Anders Carlsson731f7f92009-10-30 01:26:12 +00001447 Mangler.mangleCXXRtti(Ty);
Mike Stump738f8c22009-07-31 23:15:31 +00001448
1449 os.flush();
1450 }
Mike Stumpf1216772009-07-31 18:25:34 +00001451}