blob: e6d643f7ca8b33d2c95bc5c4e68077f8df8ab182 [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"
Douglas Gregor6ec36682009-02-18 23:53:56 +000022#include "clang/Basic/SourceManager.h"
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000023#include "llvm/Support/Compiler.h"
24#include "llvm/Support/raw_ostream.h"
25using namespace clang;
26
27namespace {
28 class VISIBILITY_HIDDEN CXXNameMangler {
29 ASTContext &Context;
30 llvm::raw_ostream &Out;
31
Anders Carlsson27ae5362009-04-17 01:58:57 +000032 const CXXMethodDecl *Structor;
33 unsigned StructorType;
Anders Carlsson3ac86b52009-04-15 05:36:58 +000034 CXXCtorType CtorType;
35
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000036 public:
37 CXXNameMangler(ASTContext &C, llvm::raw_ostream &os)
Anders Carlsson27ae5362009-04-17 01:58:57 +000038 : Context(C), Out(os), Structor(0), StructorType(0) { }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000039
40 bool mangle(const NamedDecl *D);
Anders Carlsson41aa8c12009-04-13 18:02:10 +000041 void mangleGuardVariable(const VarDecl *D);
Anders Carlsson27ae5362009-04-17 01:58:57 +000042
Anders Carlsson3ac86b52009-04-15 05:36:58 +000043 void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type);
Anders Carlsson27ae5362009-04-17 01:58:57 +000044 void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type);
Anders Carlsson3ac86b52009-04-15 05:36:58 +000045
Anders Carlsson43f17402009-04-02 15:51:53 +000046 private:
47 bool mangleFunctionDecl(const FunctionDecl *FD);
48
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000049 void mangleFunctionEncoding(const FunctionDecl *FD);
50 void mangleName(const NamedDecl *ND);
51 void mangleUnqualifiedName(const NamedDecl *ND);
52 void mangleSourceName(const IdentifierInfo *II);
Anders Carlsson1b42c792009-04-02 16:24:45 +000053 void mangleLocalName(const NamedDecl *ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000054 void mangleNestedName(const NamedDecl *ND);
55 void manglePrefix(const DeclContext *DC);
56 void mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity);
57 void mangleCVQualifiers(unsigned Quals);
58 void mangleType(QualType T);
59 void mangleType(const BuiltinType *T);
60 void mangleType(const FunctionType *T);
61 void mangleBareFunctionType(const FunctionType *T, bool MangleReturnType);
62 void mangleType(const TagType *T);
63 void mangleType(const ArrayType *T);
64 void mangleType(const MemberPointerType *T);
65 void mangleType(const TemplateTypeParmType *T);
Anders Carlssona40c5e42009-03-07 22:03:21 +000066 void mangleType(const ObjCInterfaceType *T);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000067 void mangleExpression(Expr *E);
Anders Carlsson3ac86b52009-04-15 05:36:58 +000068 void mangleCXXCtorType(CXXCtorType T);
Anders Carlsson27ae5362009-04-17 01:58:57 +000069 void mangleCXXDtorType(CXXDtorType T);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +000070 };
71}
72
Anders Carlsson43f17402009-04-02 15:51:53 +000073static bool isInCLinkageSpecification(const Decl *D) {
74 for (const DeclContext *DC = D->getDeclContext();
75 !DC->isTranslationUnit(); DC = DC->getParent()) {
76 if (const LinkageSpecDecl *Linkage = dyn_cast<LinkageSpecDecl>(DC))
77 return Linkage->getLanguage() == LinkageSpecDecl::lang_c;
78 }
79
80 return false;
81}
82
83bool CXXNameMangler::mangleFunctionDecl(const FunctionDecl *FD) {
84 // Clang's "overloadable" attribute extension to C/C++ implies
85 // name mangling (always).
Daniel Dunbarb11fa0d2009-04-13 21:08:27 +000086 if (FD->hasAttr<OverloadableAttr>()) {
Anders Carlsson43f17402009-04-02 15:51:53 +000087 ; // fall into mangling code unconditionally.
88 } else if (// C functions are not mangled
89 !Context.getLangOptions().CPlusPlus ||
90 // "main" is not mangled in C++
91 FD->isMain() ||
92 // No mangling in an "implicit extern C" header.
93 Context.getSourceManager().getFileCharacteristic(FD->getLocation())
94 == SrcMgr::C_ExternCSystem ||
95 // No name mangling in a C linkage specification.
96 isInCLinkageSpecification(FD))
97 return false;
98
99 // If we get here, mangle the decl name!
100 Out << "_Z";
101 mangleFunctionEncoding(FD);
102 return true;
103}
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000104
105bool CXXNameMangler::mangle(const NamedDecl *D) {
Chris Lattnerca3f25c2009-03-21 08:24:40 +0000106 // Any decl can be declared with __asm("foo") on it, and this takes
107 // precedence over all other naming in the .o file.
108 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
109 // If we have an asm name, then we use it as the mangling.
110 Out << '\01'; // LLVM IR Marker for __asm("foo")
111 Out << ALA->getLabel();
112 return true;
113 }
114
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000115 // <mangled-name> ::= _Z <encoding>
116 // ::= <data name>
117 // ::= <special-name>
118
119 // FIXME: Actually use a visitor to decode these?
Anders Carlsson43f17402009-04-02 15:51:53 +0000120 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
121 return mangleFunctionDecl(FD);
Chris Lattnerbc7a0292009-03-21 06:19:20 +0000122
Anders Carlsson329749c2009-04-02 16:05:20 +0000123 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
124 if (!Context.getLangOptions().CPlusPlus ||
Anders Carlsson9ccb0652009-04-11 01:19:45 +0000125 isInCLinkageSpecification(D) ||
126 D->getDeclContext()->isTranslationUnit())
Anders Carlsson329749c2009-04-02 16:05:20 +0000127 return false;
128
129 Out << "_Z";
130 mangleName(VD);
131 return true;
132 }
133
Anders Carlsson43f17402009-04-02 15:51:53 +0000134 return false;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000135}
136
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000137void CXXNameMangler::mangleCXXCtor(const CXXConstructorDecl *D,
138 CXXCtorType Type) {
Anders Carlsson27ae5362009-04-17 01:58:57 +0000139 assert(!Structor && "Structor already set!");
140 Structor = D;
141 StructorType = Type;
142
143 mangle(D);
144}
145
146void CXXNameMangler::mangleCXXDtor(const CXXDestructorDecl *D,
147 CXXDtorType Type) {
148 assert(!Structor && "Structor already set!");
149 Structor = D;
150 StructorType = Type;
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000151
152 mangle(D);
153}
154
Anders Carlsson41aa8c12009-04-13 18:02:10 +0000155void CXXNameMangler::mangleGuardVariable(const VarDecl *D)
156{
157 // <special-name> ::= GV <object name> # Guard variable for one-time
158 // # initialization
159
160 Out << "_ZGV";
161 mangleName(D);
162}
163
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000164void CXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD) {
165 // <encoding> ::= <function name> <bare-function-type>
166 mangleName(FD);
167 mangleBareFunctionType(FD->getType()->getAsFunctionType(), false);
168}
169
170static bool isStdNamespace(const DeclContext *DC) {
171 if (!DC->isNamespace() || !DC->getParent()->isTranslationUnit())
172 return false;
173
174 const NamespaceDecl *NS = cast<NamespaceDecl>(DC);
Douglas Gregor6ec36682009-02-18 23:53:56 +0000175 return NS->getOriginalNamespace()->getIdentifier()->isStr("std");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000176}
177
178void CXXNameMangler::mangleName(const NamedDecl *ND) {
179 // <name> ::= <nested-name>
180 // ::= <unscoped-name>
181 // ::= <unscoped-template-name> <template-args>
182 // ::= <local-name> # See Scope Encoding below
183 //
184 // <unscoped-name> ::= <unqualified-name>
185 // ::= St <unqualified-name> # ::std::
186 if (ND->getDeclContext()->isTranslationUnit())
187 mangleUnqualifiedName(ND);
188 else if (isStdNamespace(ND->getDeclContext())) {
189 Out << "St";
190 mangleUnqualifiedName(ND);
Anders Carlsson1b42c792009-04-02 16:24:45 +0000191 } else if (isa<FunctionDecl>(ND->getDeclContext()))
192 mangleLocalName(ND);
193 else
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000194 mangleNestedName(ND);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000195}
196
197void CXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND) {
198 // <unqualified-name> ::= <operator-name>
199 // ::= <ctor-dtor-name>
200 // ::= <source-name>
201 DeclarationName Name = ND->getDeclName();
202 switch (Name.getNameKind()) {
203 case DeclarationName::Identifier:
204 mangleSourceName(Name.getAsIdentifierInfo());
205 break;
206
207 case DeclarationName::ObjCZeroArgSelector:
208 case DeclarationName::ObjCOneArgSelector:
209 case DeclarationName::ObjCMultiArgSelector:
210 assert(false && "Can't mangle Objective-C selector names here!");
211 break;
212
213 case DeclarationName::CXXConstructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000214 if (ND == Structor)
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000215 // If the named decl is the C++ constructor we're mangling, use the
216 // type we were given.
Anders Carlsson27ae5362009-04-17 01:58:57 +0000217 mangleCXXCtorType(static_cast<CXXCtorType>(StructorType));
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000218 else
219 // Otherwise, use the complete constructor name. This is relevant if a
220 // class with a constructor is declared within a constructor.
221 mangleCXXCtorType(Ctor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000222 break;
223
224 case DeclarationName::CXXDestructorName:
Anders Carlsson27ae5362009-04-17 01:58:57 +0000225 if (ND == Structor)
226 // If the named decl is the C++ destructor we're mangling, use the
227 // type we were given.
228 mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
229 else
230 // Otherwise, use the complete destructor name. This is relevant if a
231 // class with a destructor is declared within a destructor.
232 mangleCXXDtorType(Dtor_Complete);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000233 break;
234
235 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor219cc612009-02-13 01:28:03 +0000236 // <operator-name> ::= cv <type> # (cast)
237 Out << "cv";
238 mangleType(Context.getCanonicalType(Name.getCXXNameType()));
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000239 break;
240
241 case DeclarationName::CXXOperatorName:
242 mangleOperatorName(Name.getCXXOverloadedOperator(),
243 cast<FunctionDecl>(ND)->getNumParams());
244 break;
245
246 case DeclarationName::CXXUsingDirective:
247 assert(false && "Can't mangle a using directive name!");
Douglas Gregor219cc612009-02-13 01:28:03 +0000248 break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000249 }
250}
251
252void CXXNameMangler::mangleSourceName(const IdentifierInfo *II) {
253 // <source-name> ::= <positive length number> <identifier>
254 // <number> ::= [n] <non-negative decimal integer>
255 // <identifier> ::= <unqualified source code identifier>
256 Out << II->getLength() << II->getName();
257}
258
259void CXXNameMangler::mangleNestedName(const NamedDecl *ND) {
260 // <nested-name> ::= N [<CV-qualifiers>] <prefix> <unqualified-name> E
261 // ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
262 // FIXME: no template support
263 Out << 'N';
264 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(ND))
265 mangleCVQualifiers(Method->getTypeQualifiers());
266 manglePrefix(ND->getDeclContext());
267 mangleUnqualifiedName(ND);
268 Out << 'E';
269}
270
Anders Carlsson1b42c792009-04-02 16:24:45 +0000271void CXXNameMangler::mangleLocalName(const NamedDecl *ND) {
272 // <local-name> := Z <function encoding> E <entity name> [<discriminator>]
273 // := Z <function encoding> E s [<discriminator>]
274 // <discriminator> := _ <non-negative number>
275 Out << 'Z';
276 mangleFunctionEncoding(cast<FunctionDecl>(ND->getDeclContext()));
277 Out << 'E';
278 mangleSourceName(ND->getIdentifier());
279}
280
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000281void CXXNameMangler::manglePrefix(const DeclContext *DC) {
282 // <prefix> ::= <prefix> <unqualified-name>
283 // ::= <template-prefix> <template-args>
284 // ::= <template-param>
285 // ::= # empty
286 // ::= <substitution>
287 // FIXME: We only handle mangling of namespaces and classes at the moment.
Anders Carlssonc8dee9c2009-04-01 00:42:16 +0000288 if (!DC->getParent()->isTranslationUnit())
289 manglePrefix(DC->getParent());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000290
291 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(DC))
292 mangleSourceName(Namespace->getIdentifier());
293 else if (const RecordDecl *Record = dyn_cast<RecordDecl>(DC))
294 mangleSourceName(Record->getIdentifier());
295}
296
297void
298CXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity) {
299 switch (OO) {
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000300 // <operator-name> ::= nw # new
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000301 case OO_New: Out << "nw"; break;
302 // ::= na # new[]
303 case OO_Array_New: Out << "na"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000304 // ::= dl # delete
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000305 case OO_Delete: Out << "dl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000306 // ::= da # delete[]
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000307 case OO_Array_Delete: Out << "da"; break;
308 // ::= ps # + (unary)
309 // ::= pl # +
310 case OO_Plus: Out << (Arity == 1? "ps" : "pl"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000311 // ::= ng # - (unary)
312 // ::= mi # -
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000313 case OO_Minus: Out << (Arity == 1? "ng" : "mi"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000314 // ::= ad # & (unary)
315 // ::= an # &
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000316 case OO_Amp: Out << (Arity == 1? "ad" : "an"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000317 // ::= de # * (unary)
318 // ::= ml # *
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000319 case OO_Star: Out << (Arity == 1? "de" : "ml"); break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000320 // ::= co # ~
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000321 case OO_Tilde: Out << "co"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000322 // ::= dv # /
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000323 case OO_Slash: Out << "dv"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000324 // ::= rm # %
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000325 case OO_Percent: Out << "rm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000326 // ::= or # |
327 case OO_Pipe: Out << "or"; break;
328 // ::= eo # ^
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000329 case OO_Caret: Out << "eo"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000330 // ::= aS # =
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000331 case OO_Equal: Out << "aS"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000332 // ::= pL # +=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000333 case OO_PlusEqual: Out << "pL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000334 // ::= mI # -=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000335 case OO_MinusEqual: Out << "mI"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000336 // ::= mL # *=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000337 case OO_StarEqual: Out << "mL"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000338 // ::= dV # /=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000339 case OO_SlashEqual: Out << "dV"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000340 // ::= rM # %=
341 case OO_PercentEqual: Out << "rM"; break;
342 // ::= aN # &=
343 case OO_AmpEqual: Out << "aN"; break;
344 // ::= oR # |=
345 case OO_PipeEqual: Out << "oR"; break;
346 // ::= eO # ^=
347 case OO_CaretEqual: Out << "eO"; break;
348 // ::= ls # <<
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000349 case OO_LessLess: Out << "ls"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000350 // ::= rs # >>
351 case OO_GreaterGreater: Out << "rs"; break;
352 // ::= lS # <<=
353 case OO_LessLessEqual: Out << "lS"; break;
354 // ::= rS # >>=
355 case OO_GreaterGreaterEqual: Out << "rS"; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000356 // ::= eq # ==
357 case OO_EqualEqual: Out << "eq"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000358 // ::= ne # !=
359 case OO_ExclaimEqual: Out << "ne"; break;
360 // ::= lt # <
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000361 case OO_Less: Out << "lt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000362 // ::= gt # >
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000363 case OO_Greater: Out << "gt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000364 // ::= le # <=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000365 case OO_LessEqual: Out << "le"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000366 // ::= ge # >=
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000367 case OO_GreaterEqual: Out << "ge"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000368 // ::= nt # !
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000369 case OO_Exclaim: Out << "nt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000370 // ::= aa # &&
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000371 case OO_AmpAmp: Out << "aa"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000372 // ::= oo # ||
373 case OO_PipePipe: Out << "oo"; break;
374 // ::= pp # ++
375 case OO_PlusPlus: Out << "pp"; break;
376 // ::= mm # --
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000377 case OO_MinusMinus: Out << "mm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000378 // ::= cm # ,
379 case OO_Comma: Out << "cm"; break;
380 // ::= pm # ->*
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000381 case OO_ArrowStar: Out << "pm"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000382 // ::= pt # ->
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000383 case OO_Arrow: Out << "pt"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000384 // ::= cl # ()
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000385 case OO_Call: Out << "cl"; break;
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000386 // ::= ix # []
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000387 case OO_Subscript: Out << "ix"; break;
388 // UNSUPPORTED: ::= qu # ?
389
Sebastian Redl3201f6b2009-04-16 17:51:27 +0000390 case OO_None:
391 case OO_Conditional:
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000392 case NUM_OVERLOADED_OPERATORS:
Douglas Gregor6ec36682009-02-18 23:53:56 +0000393 assert(false && "Not an overloaded operator");
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000394 break;
395 }
396}
397
398void CXXNameMangler::mangleCVQualifiers(unsigned Quals) {
399 // <CV-qualifiers> ::= [r] [V] [K] # restrict (C99), volatile, const
400 if (Quals & QualType::Restrict)
401 Out << 'r';
402 if (Quals & QualType::Volatile)
403 Out << 'V';
404 if (Quals & QualType::Const)
405 Out << 'K';
406}
407
408void CXXNameMangler::mangleType(QualType T) {
Anders Carlsson4843e582009-03-10 17:07:44 +0000409 // Only operate on the canonical type!
410 T = Context.getCanonicalType(T);
411
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000412 // FIXME: Should we have a TypeNodes.def to make this easier? (YES!)
413
414 // <type> ::= <CV-qualifiers> <type>
415 mangleCVQualifiers(T.getCVRQualifiers());
416
417 // ::= <builtin-type>
Anders Carlsson4843e582009-03-10 17:07:44 +0000418 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T.getTypePtr()))
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000419 mangleType(BT);
420 // ::= <function-type>
421 else if (const FunctionType *FT = dyn_cast<FunctionType>(T.getTypePtr()))
422 mangleType(FT);
423 // ::= <class-enum-type>
424 else if (const TagType *TT = dyn_cast<TagType>(T.getTypePtr()))
425 mangleType(TT);
426 // ::= <array-type>
427 else if (const ArrayType *AT = dyn_cast<ArrayType>(T.getTypePtr()))
428 mangleType(AT);
429 // ::= <pointer-to-member-type>
430 else if (const MemberPointerType *MPT
431 = dyn_cast<MemberPointerType>(T.getTypePtr()))
432 mangleType(MPT);
433 // ::= <template-param>
434 else if (const TemplateTypeParmType *TypeParm
435 = dyn_cast<TemplateTypeParmType>(T.getTypePtr()))
436 mangleType(TypeParm);
437 // FIXME: ::= <template-template-param> <template-args>
438 // FIXME: ::= <substitution> # See Compression below
439 // ::= P <type> # pointer-to
440 else if (const PointerType *PT = dyn_cast<PointerType>(T.getTypePtr())) {
441 Out << 'P';
442 mangleType(PT->getPointeeType());
443 }
444 // ::= R <type> # reference-to
Sebastian Redl7c80bd62009-03-16 23:22:08 +0000445 else if (const LValueReferenceType *RT =
446 dyn_cast<LValueReferenceType>(T.getTypePtr())) {
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000447 Out << 'R';
448 mangleType(RT->getPointeeType());
449 }
Sebastian Redl7c80bd62009-03-16 23:22:08 +0000450 // ::= O <type> # rvalue reference-to (C++0x)
451 else if (const RValueReferenceType *RT =
452 dyn_cast<RValueReferenceType>(T.getTypePtr())) {
453 Out << 'O';
454 mangleType(RT->getPointeeType());
455 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000456 // ::= C <type> # complex pair (C 2000)
457 else if (const ComplexType *CT = dyn_cast<ComplexType>(T.getTypePtr())) {
458 Out << 'C';
459 mangleType(CT->getElementType());
460 } else if (const VectorType *VT = dyn_cast<VectorType>(T.getTypePtr())) {
461 // GNU extension: vector types
462 Out << "U8__vector";
463 mangleType(VT->getElementType());
Anders Carlssona40c5e42009-03-07 22:03:21 +0000464 } else if (const ObjCInterfaceType *IT =
465 dyn_cast<ObjCInterfaceType>(T.getTypePtr())) {
466 mangleType(IT);
Anders Carlsson4843e582009-03-10 17:07:44 +0000467 }
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000468 // FIXME: ::= G <type> # imaginary (C 2000)
469 // FIXME: ::= U <source-name> <type> # vendor extended type qualifier
470 else
471 assert(false && "Cannot mangle unknown type");
472}
473
474void CXXNameMangler::mangleType(const BuiltinType *T) {
475 // <builtin-type> ::= v # void
476 // ::= w # wchar_t
477 // ::= b # bool
478 // ::= c # char
479 // ::= a # signed char
480 // ::= h # unsigned char
481 // ::= s # short
482 // ::= t # unsigned short
483 // ::= i # int
484 // ::= j # unsigned int
485 // ::= l # long
486 // ::= m # unsigned long
487 // ::= x # long long, __int64
488 // ::= y # unsigned long long, __int64
489 // ::= n # __int128
490 // UNSUPPORTED: ::= o # unsigned __int128
491 // ::= f # float
492 // ::= d # double
493 // ::= e # long double, __float80
494 // UNSUPPORTED: ::= g # __float128
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000495 // UNSUPPORTED: ::= Dd # IEEE 754r decimal floating point (64 bits)
496 // UNSUPPORTED: ::= De # IEEE 754r decimal floating point (128 bits)
497 // UNSUPPORTED: ::= Df # IEEE 754r decimal floating point (32 bits)
498 // UNSUPPORTED: ::= Dh # IEEE 754r half-precision floating point (16 bits)
499 // UNSUPPORTED: ::= Di # char32_t
500 // UNSUPPORTED: ::= Ds # char16_t
501 // ::= u <source-name> # vendor extended type
502 switch (T->getKind()) {
503 case BuiltinType::Void: Out << 'v'; break;
504 case BuiltinType::Bool: Out << 'b'; break;
505 case BuiltinType::Char_U: case BuiltinType::Char_S: Out << 'c'; break;
506 case BuiltinType::UChar: Out << 'h'; break;
507 case BuiltinType::UShort: Out << 't'; break;
508 case BuiltinType::UInt: Out << 'j'; break;
509 case BuiltinType::ULong: Out << 'm'; break;
510 case BuiltinType::ULongLong: Out << 'y'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000511 case BuiltinType::UInt128: Out << 'o'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000512 case BuiltinType::SChar: Out << 'a'; break;
513 case BuiltinType::WChar: Out << 'w'; break;
514 case BuiltinType::Short: Out << 's'; break;
515 case BuiltinType::Int: Out << 'i'; break;
516 case BuiltinType::Long: Out << 'l'; break;
517 case BuiltinType::LongLong: Out << 'x'; break;
Chris Lattner2df9ced2009-04-30 02:43:43 +0000518 case BuiltinType::Int128: Out << 'n'; break;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000519 case BuiltinType::Float: Out << 'f'; break;
520 case BuiltinType::Double: Out << 'd'; break;
521 case BuiltinType::LongDouble: Out << 'e'; break;
522
523 case BuiltinType::Overload:
524 case BuiltinType::Dependent:
525 assert(false &&
526 "Overloaded and dependent types shouldn't get to name mangling");
527 break;
528 }
529}
530
531void CXXNameMangler::mangleType(const FunctionType *T) {
532 // <function-type> ::= F [Y] <bare-function-type> E
533 Out << 'F';
534 // FIXME: We don't have enough information in the AST to produce the
535 // 'Y' encoding for extern "C" function types.
536 mangleBareFunctionType(T, /*MangleReturnType=*/true);
537 Out << 'E';
538}
539
540void CXXNameMangler::mangleBareFunctionType(const FunctionType *T,
541 bool MangleReturnType) {
542 // <bare-function-type> ::= <signature type>+
543 if (MangleReturnType)
544 mangleType(T->getResultType());
545
Douglas Gregor72564e72009-02-26 23:50:07 +0000546 const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(T);
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000547 assert(Proto && "Can't mangle K&R function prototypes");
548
Anders Carlssonc6c91bc2009-04-01 00:15:23 +0000549 if (Proto->getNumArgs() == 0) {
550 Out << 'v';
551 return;
552 }
553
Douglas Gregor72564e72009-02-26 23:50:07 +0000554 for (FunctionProtoType::arg_type_iterator Arg = Proto->arg_type_begin(),
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000555 ArgEnd = Proto->arg_type_end();
556 Arg != ArgEnd; ++Arg)
557 mangleType(*Arg);
Douglas Gregor219cc612009-02-13 01:28:03 +0000558
559 // <builtin-type> ::= z # ellipsis
560 if (Proto->isVariadic())
561 Out << 'z';
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000562}
563
564void CXXNameMangler::mangleType(const TagType *T) {
565 // <class-enum-type> ::= <name>
Anders Carlsson4843e582009-03-10 17:07:44 +0000566
567 if (!T->getDecl()->getIdentifier())
568 mangleName(T->getDecl()->getTypedefForAnonDecl());
569 else
570 mangleName(T->getDecl());
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000571}
572
573void CXXNameMangler::mangleType(const ArrayType *T) {
574 // <array-type> ::= A <positive dimension number> _ <element type>
575 // ::= A [<dimension expression>] _ <element type>
576 Out << 'A';
577 if (const ConstantArrayType *CAT = dyn_cast<ConstantArrayType>(T))
578 Out << CAT->getSize();
579 else if (const VariableArrayType *VAT = dyn_cast<VariableArrayType>(T))
580 mangleExpression(VAT->getSizeExpr());
581 else if (const DependentSizedArrayType *DSAT
582 = dyn_cast<DependentSizedArrayType>(T))
583 mangleExpression(DSAT->getSizeExpr());
584
585 Out << '_';
586 mangleType(T->getElementType());
587}
588
589void CXXNameMangler::mangleType(const MemberPointerType *T) {
590 // <pointer-to-member-type> ::= M <class type> <member type>
591 Out << 'M';
592 mangleType(QualType(T->getClass(), 0));
593 mangleType(T->getPointeeType());
594}
595
596void CXXNameMangler::mangleType(const TemplateTypeParmType *T) {
597 // <template-param> ::= T_ # first template parameter
598 // ::= T <parameter-2 non-negative number> _
599 if (T->getIndex() == 0)
600 Out << "T_";
601 else
602 Out << 'T' << (T->getIndex() - 1) << '_';
603}
604
Anders Carlssona40c5e42009-03-07 22:03:21 +0000605void CXXNameMangler::mangleType(const ObjCInterfaceType *T) {
606 mangleSourceName(T->getDecl()->getIdentifier());
607}
608
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000609void CXXNameMangler::mangleExpression(Expr *E) {
610 assert(false && "Cannot mangle expressions yet");
611}
612
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000613void CXXNameMangler::mangleCXXCtorType(CXXCtorType T) {
614 // <ctor-dtor-name> ::= C1 # complete object constructor
615 // ::= C2 # base object constructor
616 // ::= C3 # complete object allocating constructor
617 //
618 switch (T) {
619 case Ctor_Complete:
620 Out << "C1";
621 break;
622 case Ctor_Base:
623 Out << "C2";
624 break;
625 case Ctor_CompleteAllocating:
626 Out << "C3";
627 break;
628 }
629}
630
Anders Carlsson27ae5362009-04-17 01:58:57 +0000631void CXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
632 // <ctor-dtor-name> ::= D0 # deleting destructor
633 // ::= D1 # complete object destructor
634 // ::= D2 # base object destructor
635 //
636 switch (T) {
637 case Dtor_Deleting:
638 Out << "D0";
639 break;
640 case Dtor_Complete:
641 Out << "D1";
642 break;
643 case Dtor_Base:
644 Out << "D2";
645 break;
646 }
647}
648
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000649namespace clang {
650 /// \brief Mangles the name of the declaration D and emits that name
651 /// to the given output stream.
652 ///
653 /// If the declaration D requires a mangled name, this routine will
654 /// emit that mangled name to \p os and return true. Otherwise, \p
655 /// os will be unchanged and this routine will return false. In this
656 /// case, the caller should just emit the identifier of the declaration
657 /// (\c D->getIdentifier()) as its name.
658 bool mangleName(const NamedDecl *D, ASTContext &Context,
659 llvm::raw_ostream &os) {
660 CXXNameMangler Mangler(Context, os);
Douglas Gregor6ec36682009-02-18 23:53:56 +0000661 if (!Mangler.mangle(D))
662 return false;
663
664 os.flush();
665 return true;
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000666 }
Anders Carlsson41aa8c12009-04-13 18:02:10 +0000667
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000668 /// mangleGuardVariable - Returns the mangled name for a guard variable
669 /// for the passed in VarDecl.
Anders Carlsson41aa8c12009-04-13 18:02:10 +0000670 void mangleGuardVariable(const VarDecl *D, ASTContext &Context,
671 llvm::raw_ostream &os) {
672 CXXNameMangler Mangler(Context, os);
673 Mangler.mangleGuardVariable(D);
674
675 os.flush();
676 }
Anders Carlsson3ac86b52009-04-15 05:36:58 +0000677
678 void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type,
679 ASTContext &Context, llvm::raw_ostream &os) {
680 CXXNameMangler Mangler(Context, os);
681 Mangler.mangleCXXCtor(D, Type);
682
683 os.flush();
684 }
Anders Carlsson27ae5362009-04-17 01:58:57 +0000685
686 void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type,
687 ASTContext &Context, llvm::raw_ostream &os) {
688 CXXNameMangler Mangler(Context, os);
689 Mangler.mangleCXXDtor(D, Type);
690
691 os.flush();
692 }
693
694
Douglas Gregor5f2bfd42009-02-13 00:10:09 +0000695}
696