| //===--- Mangle.h - Mangle C++ Names ----------------------------*- C++ -*-===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is distributed under the University of Illinois Open Source |
| // License. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // Implements C++ name mangling according to the Itanium C++ ABI, |
| // which is used in GCC 3.2 and newer (and many compilers that are |
| // ABI-compatible with GCC): |
| // |
| // http://www.codesourcery.com/public/cxx-abi/abi.html |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #ifndef LLVM_CLANG_CODEGEN_MANGLE_H |
| #define LLVM_CLANG_CODEGEN_MANGLE_H |
| |
| #include "CGCXX.h" |
| #include "GlobalDecl.h" |
| #include "clang/AST/Type.h" |
| #include "llvm/ADT/DenseMap.h" |
| #include "llvm/ADT/StringRef.h" |
| #include "llvm/ADT/SmallString.h" |
| #include "llvm/Support/raw_ostream.h" |
| |
| namespace clang { |
| class ASTContext; |
| class BlockDecl; |
| class CXXConstructorDecl; |
| class CXXDestructorDecl; |
| class CXXMethodDecl; |
| class FunctionDecl; |
| class NamedDecl; |
| class ObjCMethodDecl; |
| class VarDecl; |
| |
| namespace CodeGen { |
| struct ThisAdjustment; |
| struct ThunkInfo; |
| |
| /// MangleBuffer - a convenient class for storing a name which is |
| /// either the result of a mangling or is a constant string with |
| /// external memory ownership. |
| class MangleBuffer { |
| public: |
| void setString(llvm::StringRef Ref) { |
| String = Ref; |
| } |
| |
| llvm::SmallVectorImpl<char> &getBuffer() { |
| return Buffer; |
| } |
| |
| llvm::StringRef getString() const { |
| if (!String.empty()) return String; |
| return Buffer.str(); |
| } |
| |
| operator llvm::StringRef() const { |
| return getString(); |
| } |
| |
| private: |
| llvm::StringRef String; |
| llvm::SmallString<256> Buffer; |
| }; |
| |
| /// MangleContext - Context for tracking state which persists across multiple |
| /// calls to the C++ name mangler. |
| class MangleContext { |
| ASTContext &Context; |
| Diagnostic &Diags; |
| |
| llvm::DenseMap<const TagDecl *, uint64_t> AnonStructIds; |
| unsigned Discriminator; |
| llvm::DenseMap<const NamedDecl*, unsigned> Uniquifier; |
| llvm::DenseMap<const BlockDecl*, unsigned> GlobalBlockIds; |
| llvm::DenseMap<const BlockDecl*, unsigned> LocalBlockIds; |
| |
| public: |
| explicit MangleContext(ASTContext &Context, |
| Diagnostic &Diags) |
| : Context(Context), Diags(Diags) { } |
| |
| virtual ~MangleContext() { } |
| |
| ASTContext &getASTContext() const { return Context; } |
| |
| Diagnostic &getDiags() const { return Diags; } |
| |
| void startNewFunction() { LocalBlockIds.clear(); } |
| |
| uint64_t getAnonymousStructId(const TagDecl *TD) { |
| std::pair<llvm::DenseMap<const TagDecl *, |
| uint64_t>::iterator, bool> Result = |
| AnonStructIds.insert(std::make_pair(TD, AnonStructIds.size())); |
| return Result.first->second; |
| } |
| |
| unsigned getBlockId(const BlockDecl *BD, bool Local) { |
| llvm::DenseMap<const BlockDecl *, unsigned> &BlockIds |
| = Local? LocalBlockIds : GlobalBlockIds; |
| std::pair<llvm::DenseMap<const BlockDecl *, unsigned>::iterator, bool> |
| Result = BlockIds.insert(std::make_pair(BD, BlockIds.size())); |
| return Result.first->second; |
| } |
| |
| /// @name Mangler Entry Points |
| /// @{ |
| |
| virtual bool shouldMangleDeclName(const NamedDecl *D); |
| virtual void mangleName(const NamedDecl *D, llvm::SmallVectorImpl<char> &); |
| virtual void mangleThunk(const CXXMethodDecl *MD, |
| const ThunkInfo &Thunk, |
| llvm::SmallVectorImpl<char> &); |
| virtual void mangleCXXDtorThunk(const CXXDestructorDecl *DD, CXXDtorType Type, |
| const ThisAdjustment &ThisAdjustment, |
| llvm::SmallVectorImpl<char> &); |
| virtual void mangleGuardVariable(const VarDecl *D, |
| llvm::SmallVectorImpl<char> &); |
| virtual void mangleCXXVTable(const CXXRecordDecl *RD, |
| llvm::SmallVectorImpl<char> &); |
| virtual void mangleCXXVTT(const CXXRecordDecl *RD, |
| llvm::SmallVectorImpl<char> &); |
| virtual void mangleCXXCtorVTable(const CXXRecordDecl *RD, int64_t Offset, |
| const CXXRecordDecl *Type, |
| llvm::SmallVectorImpl<char> &); |
| virtual void mangleCXXRTTI(QualType T, llvm::SmallVectorImpl<char> &); |
| virtual void mangleCXXRTTIName(QualType T, llvm::SmallVectorImpl<char> &); |
| virtual void mangleCXXCtor(const CXXConstructorDecl *D, CXXCtorType Type, |
| llvm::SmallVectorImpl<char> &); |
| virtual void mangleCXXDtor(const CXXDestructorDecl *D, CXXDtorType Type, |
| llvm::SmallVectorImpl<char> &); |
| void mangleBlock(GlobalDecl GD, |
| const BlockDecl *BD, llvm::SmallVectorImpl<char> &); |
| |
| void mangleInitDiscriminator() { |
| Discriminator = 0; |
| } |
| |
| bool getNextDiscriminator(const NamedDecl *ND, unsigned &disc) { |
| unsigned &discriminator = Uniquifier[ND]; |
| if (!discriminator) |
| discriminator = ++Discriminator; |
| if (discriminator == 1) |
| return false; |
| disc = discriminator-2; |
| return true; |
| } |
| /// @} |
| }; |
| |
| /// MiscNameMangler - Mangles Objective-C method names and blocks. |
| class MiscNameMangler { |
| MangleContext &Context; |
| llvm::raw_svector_ostream Out; |
| |
| ASTContext &getASTContext() const { return Context.getASTContext(); } |
| |
| public: |
| MiscNameMangler(MangleContext &C, llvm::SmallVectorImpl<char> &Res); |
| |
| llvm::raw_svector_ostream &getStream() { return Out; } |
| |
| void mangleBlock(GlobalDecl GD, const BlockDecl *BD); |
| void mangleObjCMethodName(const ObjCMethodDecl *MD); |
| }; |
| |
| } |
| } |
| |
| #endif |