Douglas Gregor | 6ec3668 | 2009-02-18 23:53:56 +0000 | [diff] [blame] | 1 | //===--- Mangle.h - Mangle C++ Names ----------------------------*- C++ -*-===// |
Douglas Gregor | 5f2bfd4 | 2009-02-13 00:10:09 +0000 | [diff] [blame] | 2 | // |
| 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 | //===----------------------------------------------------------------------===// |
Chris Lattner | ca3f25c | 2009-03-21 08:24:40 +0000 | [diff] [blame] | 17 | |
Douglas Gregor | 5f2bfd4 | 2009-02-13 00:10:09 +0000 | [diff] [blame] | 18 | #ifndef LLVM_CLANG_CODEGEN_MANGLE_H |
| 19 | #define LLVM_CLANG_CODEGEN_MANGLE_H |
| 20 | |
Anders Carlsson | 3ac86b5 | 2009-04-15 05:36:58 +0000 | [diff] [blame] | 21 | #include "CGCXX.h" |
Mike Stump | f121677 | 2009-07-31 18:25:34 +0000 | [diff] [blame] | 22 | #include "clang/AST/Type.h" |
Anders Carlsson | c4355b6 | 2009-10-07 01:45:02 +0000 | [diff] [blame] | 23 | #include "llvm/ADT/DenseMap.h" |
John McCall | f746aa6 | 2010-03-19 23:29:14 +0000 | [diff] [blame] | 24 | #include "llvm/ADT/StringRef.h" |
| 25 | #include "llvm/ADT/SmallString.h" |
Douglas Gregor | 5f2bfd4 | 2009-02-13 00:10:09 +0000 | [diff] [blame] | 26 | |
| 27 | namespace clang { |
Anders Carlsson | b73a5be | 2009-11-26 02:49:32 +0000 | [diff] [blame] | 28 | class ASTContext; |
| 29 | class CXXConstructorDecl; |
| 30 | class CXXDestructorDecl; |
Anders Carlsson | 19879c9 | 2010-03-23 17:17:29 +0000 | [diff] [blame] | 31 | class CXXMethodDecl; |
Anders Carlsson | b73a5be | 2009-11-26 02:49:32 +0000 | [diff] [blame] | 32 | class FunctionDecl; |
| 33 | class NamedDecl; |
| 34 | class VarDecl; |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 35 | |
Anders Carlsson | b73a5be | 2009-11-26 02:49:32 +0000 | [diff] [blame] | 36 | namespace CodeGen { |
Douglas Gregor | 9ff659b | 2010-03-24 16:59:08 +0000 | [diff] [blame] | 37 | struct ThisAdjustment; |
| 38 | struct ThunkInfo; |
John McCall | f746aa6 | 2010-03-19 23:29:14 +0000 | [diff] [blame] | 39 | |
| 40 | /// MangleBuffer - a convenient class for storing a name which is |
| 41 | /// either the result of a mangling or is a constant string with |
| 42 | /// external memory ownership. |
| 43 | class MangleBuffer { |
| 44 | public: |
| 45 | void setString(llvm::StringRef Ref) { |
| 46 | String = Ref; |
| 47 | } |
| 48 | |
| 49 | llvm::SmallVectorImpl<char> &getBuffer() { |
| 50 | return Buffer; |
| 51 | } |
| 52 | |
| 53 | llvm::StringRef getString() const { |
| 54 | if (!String.empty()) return String; |
| 55 | return Buffer.str(); |
| 56 | } |
| 57 | |
| 58 | operator llvm::StringRef() const { |
| 59 | return getString(); |
| 60 | } |
| 61 | |
| 62 | private: |
| 63 | llvm::StringRef String; |
| 64 | llvm::SmallString<256> Buffer; |
| 65 | }; |
Anders Carlsson | b73a5be | 2009-11-26 02:49:32 +0000 | [diff] [blame] | 66 | |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 67 | /// MangleContext - Context for tracking state which persists across multiple |
| 68 | /// calls to the C++ name mangler. |
| 69 | class MangleContext { |
| 70 | ASTContext &Context; |
John McCall | 6ae1f35 | 2010-04-09 22:26:14 +0000 | [diff] [blame^] | 71 | Diagnostic &Diags; |
Anders Carlsson | c4355b6 | 2009-10-07 01:45:02 +0000 | [diff] [blame] | 72 | |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 73 | llvm::DenseMap<const TagDecl *, uint64_t> AnonStructIds; |
Fariborz Jahanian | 4819ac4 | 2010-03-04 01:02:03 +0000 | [diff] [blame] | 74 | unsigned Discriminator; |
| 75 | llvm::DenseMap<const NamedDecl*, unsigned> Uniquifier; |
| 76 | |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 77 | public: |
John McCall | 6ae1f35 | 2010-04-09 22:26:14 +0000 | [diff] [blame^] | 78 | explicit MangleContext(ASTContext &Context, |
| 79 | Diagnostic &Diags) |
| 80 | : Context(Context), Diags(Diags) { } |
Anders Carlsson | b540491 | 2009-10-07 01:06:45 +0000 | [diff] [blame] | 81 | |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 82 | ASTContext &getASTContext() const { return Context; } |
| 83 | |
John McCall | 6ae1f35 | 2010-04-09 22:26:14 +0000 | [diff] [blame^] | 84 | Diagnostic &getDiags() const { return Diags; } |
| 85 | |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 86 | uint64_t getAnonymousStructId(const TagDecl *TD) { |
| 87 | std::pair<llvm::DenseMap<const TagDecl *, |
| 88 | uint64_t>::iterator, bool> Result = |
| 89 | AnonStructIds.insert(std::make_pair(TD, AnonStructIds.size())); |
| 90 | return Result.first->second; |
| 91 | } |
| 92 | |
| 93 | /// @name Mangler Entry Points |
| 94 | /// @{ |
| 95 | |
Daniel Dunbar | f981bf8 | 2009-11-21 09:14:52 +0000 | [diff] [blame] | 96 | bool shouldMangleDeclName(const NamedDecl *D); |
| 97 | |
| 98 | void mangleName(const NamedDecl *D, llvm::SmallVectorImpl<char> &); |
Anders Carlsson | 19879c9 | 2010-03-23 17:17:29 +0000 | [diff] [blame] | 99 | void mangleThunk(const CXXMethodDecl *MD, |
| 100 | const ThunkInfo &Thunk, |
| 101 | llvm::SmallVectorImpl<char> &); |
Anders Carlsson | 19879c9 | 2010-03-23 17:17:29 +0000 | [diff] [blame] | 102 | void mangleCXXDtorThunk(const CXXDestructorDecl *DD, CXXDtorType Type, |
| 103 | const ThisAdjustment &ThisAdjustment, |
| 104 | llvm::SmallVectorImpl<char> &); |
Daniel Dunbar | 94fd26d | 2009-11-21 09:06:22 +0000 | [diff] [blame] | 105 | void mangleGuardVariable(const VarDecl *D, llvm::SmallVectorImpl<char> &); |
| 106 | void mangleCXXVtable(const CXXRecordDecl *RD, llvm::SmallVectorImpl<char> &); |
| 107 | void mangleCXXVTT(const CXXRecordDecl *RD, llvm::SmallVectorImpl<char> &); |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 108 | void mangleCXXCtorVtable(const CXXRecordDecl *RD, int64_t Offset, |
Daniel Dunbar | 94fd26d | 2009-11-21 09:06:22 +0000 | [diff] [blame] | 109 | const CXXRecordDecl *Type, |
| 110 | llvm::SmallVectorImpl<char> &); |
Mike Stump | de05057 | 2009-12-02 18:57:08 +0000 | [diff] [blame] | 111 | void mangleCXXRTTI(QualType T, llvm::SmallVectorImpl<char> &); |
| 112 | void mangleCXXRTTIName(QualType T, llvm::SmallVectorImpl<char> &); |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 113 | void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type, |
Daniel Dunbar | 94fd26d | 2009-11-21 09:06:22 +0000 | [diff] [blame] | 114 | llvm::SmallVectorImpl<char> &); |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 115 | void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type, |
Daniel Dunbar | 94fd26d | 2009-11-21 09:06:22 +0000 | [diff] [blame] | 116 | llvm::SmallVectorImpl<char> &); |
Fariborz Jahanian | 4819ac4 | 2010-03-04 01:02:03 +0000 | [diff] [blame] | 117 | |
| 118 | void mangleInitDiscriminator() { |
| 119 | Discriminator = 0; |
| 120 | } |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 121 | |
Fariborz Jahanian | 4819ac4 | 2010-03-04 01:02:03 +0000 | [diff] [blame] | 122 | bool getNextDiscriminator(const NamedDecl *ND, unsigned &disc) { |
| 123 | unsigned &discriminator = Uniquifier[ND]; |
| 124 | if (!discriminator) |
| 125 | discriminator = ++Discriminator; |
| 126 | if (discriminator == 1) |
| 127 | return false; |
| 128 | disc = discriminator-2; |
| 129 | return true; |
| 130 | } |
Daniel Dunbar | 1b07711 | 2009-11-21 09:06:10 +0000 | [diff] [blame] | 131 | /// @} |
| 132 | }; |
Anders Carlsson | b73a5be | 2009-11-26 02:49:32 +0000 | [diff] [blame] | 133 | |
| 134 | } |
Douglas Gregor | 5f2bfd4 | 2009-02-13 00:10:09 +0000 | [diff] [blame] | 135 | } |
| 136 | |
Mike Stump | 1eb4433 | 2009-09-09 15:08:12 +0000 | [diff] [blame] | 137 | #endif |