blob: 6ee1223a0059409b5f732c43cad782723f14795c [file] [log] [blame]
Douglas Gregor3c3c4542009-02-18 23:53:56 +00001//===--- Mangle.cpp - Mangle C++ Names --------------------------*- C++ -*-===//
Douglas Gregor3556bc72009-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 Carlssonfae45862009-03-07 22:03:21 +000021#include "clang/AST/DeclObjC.h"
Anders Carlssonb1008e42009-05-15 16:09:15 +000022#include "clang/AST/DeclTemplate.h"
Douglas Gregor3c3c4542009-02-18 23:53:56 +000023#include "clang/Basic/SourceManager.h"
Douglas Gregor3556bc72009-02-13 00:10:09 +000024#include "llvm/Support/Compiler.h"
25#include "llvm/Support/raw_ostream.h"
26using namespace clang;
27
28namespace {
29 class VISIBILITY_HIDDEN CXXNameMangler {
30 ASTContext &Context;
31 llvm::raw_ostream &Out;
32
Anders Carlsson4811c302009-04-17 01:58:57 +000033 const CXXMethodDecl *Structor;
34 unsigned StructorType;
Anders Carlsson6b6adf22009-04-15 05:36:58 +000035 CXXCtorType CtorType;
36
Douglas Gregor3556bc72009-02-13 00:10:09 +000037 public:
38 CXXNameMangler(ASTContext &C, llvm::raw_ostream &os)
Anders Carlsson4811c302009-04-17 01:58:57 +000039 : Context(C), Out(os), Structor(0), StructorType(0) { }
Douglas Gregor3556bc72009-02-13 00:10:09 +000040
41 bool mangle(const NamedDecl *D);
Anders Carlsson53f73bc2009-04-13 18:02:10 +000042 void mangleGuardVariable(const VarDecl *D);
Anders Carlsson4811c302009-04-17 01:58:57 +000043
Anders Carlsson6b6adf22009-04-15 05:36:58 +000044 void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type);
Anders Carlsson4811c302009-04-17 01:58:57 +000045 void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type);
Anders Carlsson6b6adf22009-04-15 05:36:58 +000046
Anders Carlsson70cc0d22009-04-02 15:51:53 +000047 private:
48 bool mangleFunctionDecl(const FunctionDecl *FD);
49
Douglas Gregor3556bc72009-02-13 00:10:09 +000050 void mangleFunctionEncoding(const FunctionDecl *FD);
51 void mangleName(const NamedDecl *ND);
52 void mangleUnqualifiedName(const NamedDecl *ND);
53 void mangleSourceName(const IdentifierInfo *II);
Anders Carlssonfd8c56b2009-04-02 16:24:45 +000054 void mangleLocalName(const NamedDecl *ND);
Douglas Gregor3556bc72009-02-13 00:10:09 +000055 void mangleNestedName(const NamedDecl *ND);
56 void manglePrefix(const DeclContext *DC);
57 void mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity);
58 void mangleCVQualifiers(unsigned Quals);
59 void mangleType(QualType T);
60 void mangleType(const BuiltinType *T);
61 void mangleType(const FunctionType *T);
62 void mangleBareFunctionType(const FunctionType *T, bool MangleReturnType);
63 void mangleType(const TagType *T);
64 void mangleType(const ArrayType *T);
65 void mangleType(const MemberPointerType *T);
66 void mangleType(const TemplateTypeParmType *T);
Anders Carlssonfae45862009-03-07 22:03:21 +000067 void mangleType(const ObjCInterfaceType *T);
Douglas Gregor3556bc72009-02-13 00:10:09 +000068 void mangleExpression(Expr *E);
Anders Carlsson6b6adf22009-04-15 05:36:58 +000069 void mangleCXXCtorType(CXXCtorType T);
Anders Carlsson4811c302009-04-17 01:58:57 +000070 void mangleCXXDtorType(CXXDtorType T);
Anders Carlssonb1008e42009-05-15 16:09:15 +000071
72 void mangleTemplateArgumentList(const TemplateArgumentList &L);
73 void mangleTemplateArgument(const TemplateArgument &A);
Douglas Gregor3556bc72009-02-13 00:10:09 +000074 };
75}
76
Anders Carlsson70cc0d22009-04-02 15:51:53 +000077static bool isInCLinkageSpecification(const Decl *D) {
78 for (const DeclContext *DC = D->getDeclContext();
79 !DC->isTranslationUnit(); DC = DC->getParent()) {
80 if (const LinkageSpecDecl *Linkage = dyn_cast<LinkageSpecDecl>(DC))
81 return Linkage->getLanguage() == LinkageSpecDecl::lang_c;
82 }
83
84 return false;
85}
86
87bool CXXNameMangler::mangleFunctionDecl(const FunctionDecl *FD) {
88 // Clang's "overloadable" attribute extension to C/C++ implies
89 // name mangling (always).
Daniel Dunbar78582862009-04-13 21:08:27 +000090 if (FD->hasAttr<OverloadableAttr>()) {
Anders Carlsson70cc0d22009-04-02 15:51:53 +000091 ; // fall into mangling code unconditionally.
92 } else if (// C functions are not mangled
93 !Context.getLangOptions().CPlusPlus ||
94 // "main" is not mangled in C++
95 FD->isMain() ||
Anders Carlsson4bf5bbe2009-05-31 20:19:23 +000096 // No mangling in an "implicit extern C" header.
97 (FD->getLocation().isValid() &&
98 Context.getSourceManager().getFileCharacteristic(FD->getLocation()))
Anders Carlsson70cc0d22009-04-02 15:51:53 +000099 == SrcMgr::C_ExternCSystem ||
100 // No name mangling in a C linkage specification.
101 isInCLinkageSpecification(FD))
102 return false;
103
104 // If we get here, mangle the decl name!
105 Out << "_Z";
106 mangleFunctionEncoding(FD);
107 return true;
108}
Douglas Gregor3556bc72009-02-13 00:10:09 +0000109
110bool CXXNameMangler::mangle(const NamedDecl *D) {
Chris Lattner8af0e262009-03-21 08:24:40 +0000111 // Any decl can be declared with __asm("foo") on it, and this takes
112 // precedence over all other naming in the .o file.
113 if (const AsmLabelAttr *ALA = D->getAttr<AsmLabelAttr>()) {
114 // If we have an asm name, then we use it as the mangling.
115 Out << '\01'; // LLVM IR Marker for __asm("foo")
116 Out << ALA->getLabel();
117 return true;
118 }
119
Douglas Gregor3556bc72009-02-13 00:10:09 +0000120 // <mangled-name> ::= _Z <encoding>
121 // ::= <data name>
122 // ::= <special-name>
123
124 // FIXME: Actually use a visitor to decode these?
Anders Carlsson70cc0d22009-04-02 15:51:53 +0000125 if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
126 return mangleFunctionDecl(FD);
Chris Lattner5cbf5882009-03-21 06:19:20 +0000127
Anders Carlssonaf6bb352009-04-02 16:05:20 +0000128 if (const VarDecl *VD = dyn_cast<VarDecl>(D)) {
129 if (!Context.getLangOptions().CPlusPlus ||
Anders Carlssonc3e84ae2009-04-11 01:19:45 +0000130 isInCLinkageSpecification(D) ||
131 D->getDeclContext()->isTranslationUnit())
Anders Carlssonaf6bb352009-04-02 16:05:20 +0000132 return false;
133
134 Out << "_Z";
135 mangleName(VD);
136 return true;
137 }
138
Anders Carlsson70cc0d22009-04-02 15:51:53 +0000139 return false;
Douglas Gregor3556bc72009-02-13 00:10:09 +0000140}
141
Anders Carlsson6b6adf22009-04-15 05:36:58 +0000142void CXXNameMangler::mangleCXXCtor(const CXXConstructorDecl *D,
143 CXXCtorType Type) {
Anders Carlsson4811c302009-04-17 01:58:57 +0000144 assert(!Structor && "Structor already set!");
145 Structor = D;
146 StructorType = Type;
147
148 mangle(D);
149}
150
151void CXXNameMangler::mangleCXXDtor(const CXXDestructorDecl *D,
152 CXXDtorType Type) {
153 assert(!Structor && "Structor already set!");
154 Structor = D;
155 StructorType = Type;
Anders Carlsson6b6adf22009-04-15 05:36:58 +0000156
157 mangle(D);
158}
159
Anders Carlsson53f73bc2009-04-13 18:02:10 +0000160void CXXNameMangler::mangleGuardVariable(const VarDecl *D)
161{
162 // <special-name> ::= GV <object name> # Guard variable for one-time
163 // # initialization
164
165 Out << "_ZGV";
166 mangleName(D);
167}
168
Douglas Gregor3556bc72009-02-13 00:10:09 +0000169void CXXNameMangler::mangleFunctionEncoding(const FunctionDecl *FD) {
170 // <encoding> ::= <function name> <bare-function-type>
171 mangleName(FD);
172 mangleBareFunctionType(FD->getType()->getAsFunctionType(), false);
173}
174
175static bool isStdNamespace(const DeclContext *DC) {
176 if (!DC->isNamespace() || !DC->getParent()->isTranslationUnit())
177 return false;
178
179 const NamespaceDecl *NS = cast<NamespaceDecl>(DC);
Douglas Gregor3c3c4542009-02-18 23:53:56 +0000180 return NS->getOriginalNamespace()->getIdentifier()->isStr("std");
Douglas Gregor3556bc72009-02-13 00:10:09 +0000181}
182
183void CXXNameMangler::mangleName(const NamedDecl *ND) {
184 // <name> ::= <nested-name>
185 // ::= <unscoped-name>
186 // ::= <unscoped-template-name> <template-args>
187 // ::= <local-name> # See Scope Encoding below
188 //
189 // <unscoped-name> ::= <unqualified-name>
190 // ::= St <unqualified-name> # ::std::
191 if (ND->getDeclContext()->isTranslationUnit())
192 mangleUnqualifiedName(ND);
193 else if (isStdNamespace(ND->getDeclContext())) {
194 Out << "St";
195 mangleUnqualifiedName(ND);
Anders Carlssonfd8c56b2009-04-02 16:24:45 +0000196 } else if (isa<FunctionDecl>(ND->getDeclContext()))
197 mangleLocalName(ND);
198 else
Douglas Gregor3556bc72009-02-13 00:10:09 +0000199 mangleNestedName(ND);
Douglas Gregor3556bc72009-02-13 00:10:09 +0000200}
201
202void CXXNameMangler::mangleUnqualifiedName(const NamedDecl *ND) {
203 // <unqualified-name> ::= <operator-name>
204 // ::= <ctor-dtor-name>
205 // ::= <source-name>
206 DeclarationName Name = ND->getDeclName();
207 switch (Name.getNameKind()) {
208 case DeclarationName::Identifier:
209 mangleSourceName(Name.getAsIdentifierInfo());
210 break;
211
212 case DeclarationName::ObjCZeroArgSelector:
213 case DeclarationName::ObjCOneArgSelector:
214 case DeclarationName::ObjCMultiArgSelector:
215 assert(false && "Can't mangle Objective-C selector names here!");
216 break;
217
218 case DeclarationName::CXXConstructorName:
Anders Carlsson4811c302009-04-17 01:58:57 +0000219 if (ND == Structor)
Anders Carlsson6b6adf22009-04-15 05:36:58 +0000220 // If the named decl is the C++ constructor we're mangling, use the
221 // type we were given.
Anders Carlsson4811c302009-04-17 01:58:57 +0000222 mangleCXXCtorType(static_cast<CXXCtorType>(StructorType));
Anders Carlsson6b6adf22009-04-15 05:36:58 +0000223 else
224 // Otherwise, use the complete constructor name. This is relevant if a
225 // class with a constructor is declared within a constructor.
226 mangleCXXCtorType(Ctor_Complete);
Douglas Gregor3556bc72009-02-13 00:10:09 +0000227 break;
228
229 case DeclarationName::CXXDestructorName:
Anders Carlsson4811c302009-04-17 01:58:57 +0000230 if (ND == Structor)
231 // If the named decl is the C++ destructor we're mangling, use the
232 // type we were given.
233 mangleCXXDtorType(static_cast<CXXDtorType>(StructorType));
234 else
235 // Otherwise, use the complete destructor name. This is relevant if a
236 // class with a destructor is declared within a destructor.
237 mangleCXXDtorType(Dtor_Complete);
Douglas Gregor3556bc72009-02-13 00:10:09 +0000238 break;
239
240 case DeclarationName::CXXConversionFunctionName:
Douglas Gregor77cfb3c2009-02-13 01:28:03 +0000241 // <operator-name> ::= cv <type> # (cast)
242 Out << "cv";
243 mangleType(Context.getCanonicalType(Name.getCXXNameType()));
Douglas Gregor3556bc72009-02-13 00:10:09 +0000244 break;
245
246 case DeclarationName::CXXOperatorName:
247 mangleOperatorName(Name.getCXXOverloadedOperator(),
248 cast<FunctionDecl>(ND)->getNumParams());
249 break;
250
251 case DeclarationName::CXXUsingDirective:
252 assert(false && "Can't mangle a using directive name!");
Douglas Gregor77cfb3c2009-02-13 01:28:03 +0000253 break;
Douglas Gregor3556bc72009-02-13 00:10:09 +0000254 }
255}
256
257void CXXNameMangler::mangleSourceName(const IdentifierInfo *II) {
258 // <source-name> ::= <positive length number> <identifier>
259 // <number> ::= [n] <non-negative decimal integer>
260 // <identifier> ::= <unqualified source code identifier>
261 Out << II->getLength() << II->getName();
262}
263
264void CXXNameMangler::mangleNestedName(const NamedDecl *ND) {
265 // <nested-name> ::= N [<CV-qualifiers>] <prefix> <unqualified-name> E
266 // ::= N [<CV-qualifiers>] <template-prefix> <template-args> E
267 // FIXME: no template support
268 Out << 'N';
269 if (const CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(ND))
270 mangleCVQualifiers(Method->getTypeQualifiers());
271 manglePrefix(ND->getDeclContext());
272 mangleUnqualifiedName(ND);
273 Out << 'E';
274}
275
Anders Carlssonfd8c56b2009-04-02 16:24:45 +0000276void CXXNameMangler::mangleLocalName(const NamedDecl *ND) {
277 // <local-name> := Z <function encoding> E <entity name> [<discriminator>]
278 // := Z <function encoding> E s [<discriminator>]
279 // <discriminator> := _ <non-negative number>
280 Out << 'Z';
281 mangleFunctionEncoding(cast<FunctionDecl>(ND->getDeclContext()));
282 Out << 'E';
283 mangleSourceName(ND->getIdentifier());
284}
285
Douglas Gregor3556bc72009-02-13 00:10:09 +0000286void CXXNameMangler::manglePrefix(const DeclContext *DC) {
287 // <prefix> ::= <prefix> <unqualified-name>
288 // ::= <template-prefix> <template-args>
289 // ::= <template-param>
290 // ::= # empty
291 // ::= <substitution>
292 // FIXME: We only handle mangling of namespaces and classes at the moment.
Anders Carlsson8c4c12a2009-04-01 00:42:16 +0000293 if (!DC->getParent()->isTranslationUnit())
294 manglePrefix(DC->getParent());
Douglas Gregor3556bc72009-02-13 00:10:09 +0000295
296 if (const NamespaceDecl *Namespace = dyn_cast<NamespaceDecl>(DC))
297 mangleSourceName(Namespace->getIdentifier());
Anders Carlssonb1008e42009-05-15 16:09:15 +0000298 else if (const RecordDecl *Record = dyn_cast<RecordDecl>(DC)) {
299 if (const ClassTemplateSpecializationDecl *D =
300 dyn_cast<ClassTemplateSpecializationDecl>(Record)) {
301 mangleType(QualType(D->getTypeForDecl(), 0));
302 } else
303 mangleSourceName(Record->getIdentifier());
304 }
Douglas Gregor3556bc72009-02-13 00:10:09 +0000305}
306
307void
308CXXNameMangler::mangleOperatorName(OverloadedOperatorKind OO, unsigned Arity) {
309 switch (OO) {
Sebastian Redlbd261962009-04-16 17:51:27 +0000310 // <operator-name> ::= nw # new
Douglas Gregor3556bc72009-02-13 00:10:09 +0000311 case OO_New: Out << "nw"; break;
312 // ::= na # new[]
313 case OO_Array_New: Out << "na"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000314 // ::= dl # delete
Douglas Gregor3556bc72009-02-13 00:10:09 +0000315 case OO_Delete: Out << "dl"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000316 // ::= da # delete[]
Douglas Gregor3556bc72009-02-13 00:10:09 +0000317 case OO_Array_Delete: Out << "da"; break;
318 // ::= ps # + (unary)
319 // ::= pl # +
320 case OO_Plus: Out << (Arity == 1? "ps" : "pl"); break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000321 // ::= ng # - (unary)
322 // ::= mi # -
Douglas Gregor3556bc72009-02-13 00:10:09 +0000323 case OO_Minus: Out << (Arity == 1? "ng" : "mi"); break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000324 // ::= ad # & (unary)
325 // ::= an # &
Douglas Gregor3556bc72009-02-13 00:10:09 +0000326 case OO_Amp: Out << (Arity == 1? "ad" : "an"); break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000327 // ::= de # * (unary)
328 // ::= ml # *
Douglas Gregor3556bc72009-02-13 00:10:09 +0000329 case OO_Star: Out << (Arity == 1? "de" : "ml"); break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000330 // ::= co # ~
Douglas Gregor3556bc72009-02-13 00:10:09 +0000331 case OO_Tilde: Out << "co"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000332 // ::= dv # /
Douglas Gregor3556bc72009-02-13 00:10:09 +0000333 case OO_Slash: Out << "dv"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000334 // ::= rm # %
Douglas Gregor3556bc72009-02-13 00:10:09 +0000335 case OO_Percent: Out << "rm"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000336 // ::= or # |
337 case OO_Pipe: Out << "or"; break;
338 // ::= eo # ^
Douglas Gregor3556bc72009-02-13 00:10:09 +0000339 case OO_Caret: Out << "eo"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000340 // ::= aS # =
Douglas Gregor3556bc72009-02-13 00:10:09 +0000341 case OO_Equal: Out << "aS"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000342 // ::= pL # +=
Douglas Gregor3556bc72009-02-13 00:10:09 +0000343 case OO_PlusEqual: Out << "pL"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000344 // ::= mI # -=
Douglas Gregor3556bc72009-02-13 00:10:09 +0000345 case OO_MinusEqual: Out << "mI"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000346 // ::= mL # *=
Douglas Gregor3556bc72009-02-13 00:10:09 +0000347 case OO_StarEqual: Out << "mL"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000348 // ::= dV # /=
Douglas Gregor3556bc72009-02-13 00:10:09 +0000349 case OO_SlashEqual: Out << "dV"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000350 // ::= rM # %=
351 case OO_PercentEqual: Out << "rM"; break;
352 // ::= aN # &=
353 case OO_AmpEqual: Out << "aN"; break;
354 // ::= oR # |=
355 case OO_PipeEqual: Out << "oR"; break;
356 // ::= eO # ^=
357 case OO_CaretEqual: Out << "eO"; break;
358 // ::= ls # <<
Douglas Gregor3556bc72009-02-13 00:10:09 +0000359 case OO_LessLess: Out << "ls"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000360 // ::= rs # >>
361 case OO_GreaterGreater: Out << "rs"; break;
362 // ::= lS # <<=
363 case OO_LessLessEqual: Out << "lS"; break;
364 // ::= rS # >>=
365 case OO_GreaterGreaterEqual: Out << "rS"; break;
Douglas Gregor3556bc72009-02-13 00:10:09 +0000366 // ::= eq # ==
367 case OO_EqualEqual: Out << "eq"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000368 // ::= ne # !=
369 case OO_ExclaimEqual: Out << "ne"; break;
370 // ::= lt # <
Douglas Gregor3556bc72009-02-13 00:10:09 +0000371 case OO_Less: Out << "lt"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000372 // ::= gt # >
Douglas Gregor3556bc72009-02-13 00:10:09 +0000373 case OO_Greater: Out << "gt"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000374 // ::= le # <=
Douglas Gregor3556bc72009-02-13 00:10:09 +0000375 case OO_LessEqual: Out << "le"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000376 // ::= ge # >=
Douglas Gregor3556bc72009-02-13 00:10:09 +0000377 case OO_GreaterEqual: Out << "ge"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000378 // ::= nt # !
Douglas Gregor3556bc72009-02-13 00:10:09 +0000379 case OO_Exclaim: Out << "nt"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000380 // ::= aa # &&
Douglas Gregor3556bc72009-02-13 00:10:09 +0000381 case OO_AmpAmp: Out << "aa"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000382 // ::= oo # ||
383 case OO_PipePipe: Out << "oo"; break;
384 // ::= pp # ++
385 case OO_PlusPlus: Out << "pp"; break;
386 // ::= mm # --
Douglas Gregor3556bc72009-02-13 00:10:09 +0000387 case OO_MinusMinus: Out << "mm"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000388 // ::= cm # ,
389 case OO_Comma: Out << "cm"; break;
390 // ::= pm # ->*
Douglas Gregor3556bc72009-02-13 00:10:09 +0000391 case OO_ArrowStar: Out << "pm"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000392 // ::= pt # ->
Douglas Gregor3556bc72009-02-13 00:10:09 +0000393 case OO_Arrow: Out << "pt"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000394 // ::= cl # ()
Douglas Gregor3556bc72009-02-13 00:10:09 +0000395 case OO_Call: Out << "cl"; break;
Sebastian Redlbd261962009-04-16 17:51:27 +0000396 // ::= ix # []
Douglas Gregor3556bc72009-02-13 00:10:09 +0000397 case OO_Subscript: Out << "ix"; break;
398 // UNSUPPORTED: ::= qu # ?
399
Sebastian Redlbd261962009-04-16 17:51:27 +0000400 case OO_None:
401 case OO_Conditional:
Douglas Gregor3556bc72009-02-13 00:10:09 +0000402 case NUM_OVERLOADED_OPERATORS:
Douglas Gregor3c3c4542009-02-18 23:53:56 +0000403 assert(false && "Not an overloaded operator");
Douglas Gregor3556bc72009-02-13 00:10:09 +0000404 break;
405 }
406}
407
408void CXXNameMangler::mangleCVQualifiers(unsigned Quals) {
409 // <CV-qualifiers> ::= [r] [V] [K] # restrict (C99), volatile, const
410 if (Quals & QualType::Restrict)
411 Out << 'r';
412 if (Quals & QualType::Volatile)
413 Out << 'V';
414 if (Quals & QualType::Const)
415 Out << 'K';
416}
417
418void CXXNameMangler::mangleType(QualType T) {
Anders Carlssonfdaed4c2009-03-10 17:07:44 +0000419 // Only operate on the canonical type!
420 T = Context.getCanonicalType(T);
421
Douglas Gregor3556bc72009-02-13 00:10:09 +0000422 // FIXME: Should we have a TypeNodes.def to make this easier? (YES!)
423
424 // <type> ::= <CV-qualifiers> <type>
425 mangleCVQualifiers(T.getCVRQualifiers());
426
427 // ::= <builtin-type>
Anders Carlssonfdaed4c2009-03-10 17:07:44 +0000428 if (const BuiltinType *BT = dyn_cast<BuiltinType>(T.getTypePtr()))
Douglas Gregor3556bc72009-02-13 00:10:09 +0000429 mangleType(BT);
430 // ::= <function-type>
431 else if (const FunctionType *FT = dyn_cast<FunctionType>(T.getTypePtr()))
432 mangleType(FT);
433 // ::= <class-enum-type>
434 else if (const TagType *TT = dyn_cast<TagType>(T.getTypePtr()))
435 mangleType(TT);
436 // ::= <array-type>
437 else if (const ArrayType *AT = dyn_cast<ArrayType>(T.getTypePtr()))
438 mangleType(AT);
439 // ::= <pointer-to-member-type>
440 else if (const MemberPointerType *MPT
441 = dyn_cast<MemberPointerType>(T.getTypePtr()))
442 mangleType(MPT);
443 // ::= <template-param>
444 else if (const TemplateTypeParmType *TypeParm
445 = dyn_cast<TemplateTypeParmType>(T.getTypePtr()))
446 mangleType(TypeParm);
447 // FIXME: ::= <template-template-param> <template-args>
448 // FIXME: ::= <substitution> # See Compression below
449 // ::= P <type> # pointer-to
450 else if (const PointerType *PT = dyn_cast<PointerType>(T.getTypePtr())) {
451 Out << 'P';
452 mangleType(PT->getPointeeType());
453 }
454 // ::= R <type> # reference-to
Sebastian Redlce6fff02009-03-16 23:22:08 +0000455 else if (const LValueReferenceType *RT =
456 dyn_cast<LValueReferenceType>(T.getTypePtr())) {
Douglas Gregor3556bc72009-02-13 00:10:09 +0000457 Out << 'R';
458 mangleType(RT->getPointeeType());
459 }
Sebastian Redlce6fff02009-03-16 23:22:08 +0000460 // ::= O <type> # rvalue reference-to (C++0x)
461 else if (const RValueReferenceType *RT =
462 dyn_cast<RValueReferenceType>(T.getTypePtr())) {
463 Out << 'O';
464 mangleType(RT->getPointeeType());
465 }
Douglas Gregor3556bc72009-02-13 00:10:09 +0000466 // ::= C <type> # complex pair (C 2000)
467 else if (const ComplexType *CT = dyn_cast<ComplexType>(T.getTypePtr())) {
468 Out << 'C';
469 mangleType(CT->getElementType());
470 } else if (const VectorType *VT = dyn_cast<VectorType>(T.getTypePtr())) {
471 // GNU extension: vector types
472 Out << "U8__vector";
473 mangleType(VT->getElementType());
Anders Carlssonfae45862009-03-07 22:03:21 +0000474 } else if (const ObjCInterfaceType *IT =
475 dyn_cast<ObjCInterfaceType>(T.getTypePtr())) {
476 mangleType(IT);
Anders Carlssonfdaed4c2009-03-10 17:07:44 +0000477 }
Douglas Gregor3556bc72009-02-13 00:10:09 +0000478 // FIXME: ::= G <type> # imaginary (C 2000)
479 // FIXME: ::= U <source-name> <type> # vendor extended type qualifier
480 else
481 assert(false && "Cannot mangle unknown type");
482}
483
484void CXXNameMangler::mangleType(const BuiltinType *T) {
485 // <builtin-type> ::= v # void
486 // ::= w # wchar_t
487 // ::= b # bool
488 // ::= c # char
489 // ::= a # signed char
490 // ::= h # unsigned char
491 // ::= s # short
492 // ::= t # unsigned short
493 // ::= i # int
494 // ::= j # unsigned int
495 // ::= l # long
496 // ::= m # unsigned long
497 // ::= x # long long, __int64
498 // ::= y # unsigned long long, __int64
499 // ::= n # __int128
500 // UNSUPPORTED: ::= o # unsigned __int128
501 // ::= f # float
502 // ::= d # double
503 // ::= e # long double, __float80
504 // UNSUPPORTED: ::= g # __float128
Douglas Gregor3556bc72009-02-13 00:10:09 +0000505 // UNSUPPORTED: ::= Dd # IEEE 754r decimal floating point (64 bits)
506 // UNSUPPORTED: ::= De # IEEE 754r decimal floating point (128 bits)
507 // UNSUPPORTED: ::= Df # IEEE 754r decimal floating point (32 bits)
508 // UNSUPPORTED: ::= Dh # IEEE 754r half-precision floating point (16 bits)
509 // UNSUPPORTED: ::= Di # char32_t
510 // UNSUPPORTED: ::= Ds # char16_t
511 // ::= u <source-name> # vendor extended type
Sebastian Redl5d0ead72009-05-10 18:38:11 +0000512 // From our point of view, std::nullptr_t is a builtin, but as far as mangling
513 // is concerned, it's a type called std::nullptr_t.
Douglas Gregor3556bc72009-02-13 00:10:09 +0000514 switch (T->getKind()) {
515 case BuiltinType::Void: Out << 'v'; break;
516 case BuiltinType::Bool: Out << 'b'; break;
517 case BuiltinType::Char_U: case BuiltinType::Char_S: Out << 'c'; break;
518 case BuiltinType::UChar: Out << 'h'; break;
519 case BuiltinType::UShort: Out << 't'; break;
520 case BuiltinType::UInt: Out << 'j'; break;
521 case BuiltinType::ULong: Out << 'm'; break;
522 case BuiltinType::ULongLong: Out << 'y'; break;
Chris Lattner6cc7e412009-04-30 02:43:43 +0000523 case BuiltinType::UInt128: Out << 'o'; break;
Douglas Gregor3556bc72009-02-13 00:10:09 +0000524 case BuiltinType::SChar: Out << 'a'; break;
525 case BuiltinType::WChar: Out << 'w'; break;
526 case BuiltinType::Short: Out << 's'; break;
527 case BuiltinType::Int: Out << 'i'; break;
528 case BuiltinType::Long: Out << 'l'; break;
529 case BuiltinType::LongLong: Out << 'x'; break;
Chris Lattner6cc7e412009-04-30 02:43:43 +0000530 case BuiltinType::Int128: Out << 'n'; break;
Douglas Gregor3556bc72009-02-13 00:10:09 +0000531 case BuiltinType::Float: Out << 'f'; break;
532 case BuiltinType::Double: Out << 'd'; break;
533 case BuiltinType::LongDouble: Out << 'e'; break;
Sebastian Redl5d0ead72009-05-10 18:38:11 +0000534 case BuiltinType::NullPtr: Out << "St9nullptr_t"; break;
Douglas Gregor3556bc72009-02-13 00:10:09 +0000535
536 case BuiltinType::Overload:
537 case BuiltinType::Dependent:
538 assert(false &&
539 "Overloaded and dependent types shouldn't get to name mangling");
540 break;
541 }
542}
543
544void CXXNameMangler::mangleType(const FunctionType *T) {
545 // <function-type> ::= F [Y] <bare-function-type> E
546 Out << 'F';
Mike Stumpba2cb0e2009-05-16 07:57:57 +0000547 // FIXME: We don't have enough information in the AST to produce the 'Y'
548 // encoding for extern "C" function types.
Douglas Gregor3556bc72009-02-13 00:10:09 +0000549 mangleBareFunctionType(T, /*MangleReturnType=*/true);
550 Out << 'E';
551}
552
553void CXXNameMangler::mangleBareFunctionType(const FunctionType *T,
554 bool MangleReturnType) {
555 // <bare-function-type> ::= <signature type>+
556 if (MangleReturnType)
557 mangleType(T->getResultType());
558
Douglas Gregor4fa58902009-02-26 23:50:07 +0000559 const FunctionProtoType *Proto = dyn_cast<FunctionProtoType>(T);
Douglas Gregor3556bc72009-02-13 00:10:09 +0000560 assert(Proto && "Can't mangle K&R function prototypes");
561
Anders Carlsson51e65522009-04-01 00:15:23 +0000562 if (Proto->getNumArgs() == 0) {
563 Out << 'v';
564 return;
565 }
566
Douglas Gregor4fa58902009-02-26 23:50:07 +0000567 for (FunctionProtoType::arg_type_iterator Arg = Proto->arg_type_begin(),
Douglas Gregor3556bc72009-02-13 00:10:09 +0000568 ArgEnd = Proto->arg_type_end();
569 Arg != ArgEnd; ++Arg)
570 mangleType(*Arg);
Douglas Gregor77cfb3c2009-02-13 01:28:03 +0000571
572 // <builtin-type> ::= z # ellipsis
573 if (Proto->isVariadic())
574 Out << 'z';
Douglas Gregor3556bc72009-02-13 00:10:09 +0000575}
576
577void CXXNameMangler::mangleType(const TagType *T) {
578 // <class-enum-type> ::= <name>
Anders Carlssonfdaed4c2009-03-10 17:07:44 +0000579
580 if (!T->getDecl()->getIdentifier())
581 mangleName(T->getDecl()->getTypedefForAnonDecl());
582 else
583 mangleName(T->getDecl());
Anders Carlssonb1008e42009-05-15 16:09:15 +0000584
585 // If this is a class template specialization, mangle the template
586 // arguments.
587 if (ClassTemplateSpecializationDecl *Spec
588 = dyn_cast<ClassTemplateSpecializationDecl>(T->getDecl()))
589 mangleTemplateArgumentList(Spec->getTemplateArgs());
Douglas Gregor3556bc72009-02-13 00:10:09 +0000590}
591
592void CXXNameMangler::mangleType(const ArrayType *T) {
593 // <array-type> ::= A <positive dimension number> _ <element type>
594 // ::= A [<dimension expression>] _ <element type>
595 Out << 'A';
596 if (const ConstantArrayType *CAT = dyn_cast<ConstantArrayType>(T))
597 Out << CAT->getSize();
598 else if (const VariableArrayType *VAT = dyn_cast<VariableArrayType>(T))
599 mangleExpression(VAT->getSizeExpr());
600 else if (const DependentSizedArrayType *DSAT
601 = dyn_cast<DependentSizedArrayType>(T))
602 mangleExpression(DSAT->getSizeExpr());
603
604 Out << '_';
605 mangleType(T->getElementType());
606}
607
608void CXXNameMangler::mangleType(const MemberPointerType *T) {
609 // <pointer-to-member-type> ::= M <class type> <member type>
610 Out << 'M';
611 mangleType(QualType(T->getClass(), 0));
Anders Carlsson0b17ef42009-05-17 17:41:20 +0000612 QualType PointeeType = T->getPointeeType();
613 if (const FunctionProtoType *FPT = dyn_cast<FunctionProtoType>(PointeeType)) {
614 mangleCVQualifiers(FPT->getTypeQuals());
615 mangleType(FPT);
616 } else
617 mangleType(PointeeType);
Douglas Gregor3556bc72009-02-13 00:10:09 +0000618}
619
620void CXXNameMangler::mangleType(const TemplateTypeParmType *T) {
621 // <template-param> ::= T_ # first template parameter
622 // ::= T <parameter-2 non-negative number> _
623 if (T->getIndex() == 0)
624 Out << "T_";
625 else
626 Out << 'T' << (T->getIndex() - 1) << '_';
627}
628
Anders Carlssonfae45862009-03-07 22:03:21 +0000629void CXXNameMangler::mangleType(const ObjCInterfaceType *T) {
630 mangleSourceName(T->getDecl()->getIdentifier());
631}
632
Douglas Gregor3556bc72009-02-13 00:10:09 +0000633void CXXNameMangler::mangleExpression(Expr *E) {
634 assert(false && "Cannot mangle expressions yet");
635}
636
Anders Carlsson6b6adf22009-04-15 05:36:58 +0000637void CXXNameMangler::mangleCXXCtorType(CXXCtorType T) {
638 // <ctor-dtor-name> ::= C1 # complete object constructor
639 // ::= C2 # base object constructor
640 // ::= C3 # complete object allocating constructor
641 //
642 switch (T) {
643 case Ctor_Complete:
644 Out << "C1";
645 break;
646 case Ctor_Base:
647 Out << "C2";
648 break;
649 case Ctor_CompleteAllocating:
650 Out << "C3";
651 break;
652 }
653}
654
Anders Carlsson4811c302009-04-17 01:58:57 +0000655void CXXNameMangler::mangleCXXDtorType(CXXDtorType T) {
656 // <ctor-dtor-name> ::= D0 # deleting destructor
657 // ::= D1 # complete object destructor
658 // ::= D2 # base object destructor
659 //
660 switch (T) {
661 case Dtor_Deleting:
662 Out << "D0";
663 break;
664 case Dtor_Complete:
665 Out << "D1";
666 break;
667 case Dtor_Base:
668 Out << "D2";
669 break;
670 }
671}
672
Anders Carlssonb1008e42009-05-15 16:09:15 +0000673void CXXNameMangler::mangleTemplateArgumentList(const TemplateArgumentList &L) {
674 // <template-args> ::= I <template-arg>+ E
675 Out << "I";
676
677 for (unsigned i = 0, e = L.size(); i != e; ++i) {
678 const TemplateArgument &A = L[i];
679
680 mangleTemplateArgument(A);
681 }
682
683 Out << "E";
684}
685
686void CXXNameMangler::mangleTemplateArgument(const TemplateArgument &A) {
687 // <template-arg> ::= <type> # type or template
688 // ::= X <expression> E # expression
689 // ::= <expr-primary> # simple expressions
690 // ::= I <template-arg>* E # argument pack
691 // ::= sp <expression> # pack expansion of (C++0x)
692 switch (A.getKind()) {
693 default:
694 assert(0 && "Unknown template argument kind!");
695 case TemplateArgument::Type:
696 mangleType(A.getAsType());
697 break;
698 case TemplateArgument::Integral:
699 // <expr-primary> ::= L <type> <value number> E # integer literal
700
701 Out << 'L';
702
703 mangleType(A.getIntegralType());
704
705 const llvm::APSInt *Integral = A.getAsIntegral();
706 if (A.getIntegralType()->isBooleanType()) {
707 // Boolean values are encoded as 0/1.
708 Out << (Integral->getBoolValue() ? '1' : '0');
709 } else {
710 if (Integral->isNegative())
711 Out << 'n';
712 Integral->abs().print(Out, false);
713 }
714
715 Out << 'E';
716 break;
717 }
718}
719
Douglas Gregor3556bc72009-02-13 00:10:09 +0000720namespace clang {
721 /// \brief Mangles the name of the declaration D and emits that name
722 /// to the given output stream.
723 ///
724 /// If the declaration D requires a mangled name, this routine will
725 /// emit that mangled name to \p os and return true. Otherwise, \p
726 /// os will be unchanged and this routine will return false. In this
727 /// case, the caller should just emit the identifier of the declaration
728 /// (\c D->getIdentifier()) as its name.
729 bool mangleName(const NamedDecl *D, ASTContext &Context,
730 llvm::raw_ostream &os) {
Anders Carlsson7bb19882009-05-03 16:51:04 +0000731 assert(!isa<CXXConstructorDecl>(D) &&
732 "Use mangleCXXCtor for constructor decls!");
733 assert(!isa<CXXDestructorDecl>(D) &&
734 "Use mangleCXXDtor for destructor decls!");
735
Douglas Gregor3556bc72009-02-13 00:10:09 +0000736 CXXNameMangler Mangler(Context, os);
Douglas Gregor3c3c4542009-02-18 23:53:56 +0000737 if (!Mangler.mangle(D))
738 return false;
739
740 os.flush();
741 return true;
Douglas Gregor3556bc72009-02-13 00:10:09 +0000742 }
Anders Carlsson53f73bc2009-04-13 18:02:10 +0000743
Anders Carlsson6b6adf22009-04-15 05:36:58 +0000744 /// mangleGuardVariable - Returns the mangled name for a guard variable
745 /// for the passed in VarDecl.
Anders Carlsson53f73bc2009-04-13 18:02:10 +0000746 void mangleGuardVariable(const VarDecl *D, ASTContext &Context,
747 llvm::raw_ostream &os) {
748 CXXNameMangler Mangler(Context, os);
749 Mangler.mangleGuardVariable(D);
750
751 os.flush();
752 }
Anders Carlsson6b6adf22009-04-15 05:36:58 +0000753
754 void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type,
755 ASTContext &Context, llvm::raw_ostream &os) {
756 CXXNameMangler Mangler(Context, os);
757 Mangler.mangleCXXCtor(D, Type);
758
759 os.flush();
760 }
Anders Carlsson4811c302009-04-17 01:58:57 +0000761
762 void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type,
763 ASTContext &Context, llvm::raw_ostream &os) {
764 CXXNameMangler Mangler(Context, os);
765 Mangler.mangleCXXDtor(D, Type);
766
767 os.flush();
768 }
769
770
Douglas Gregor3556bc72009-02-13 00:10:09 +0000771}
772