include/llvm/CodeGen/MachineModuleInfoImpls.h

   1 //===-- llvm/CodeGen/MachineModuleInfoImpls.h -------------------*- C++ -*-===//
   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 // This file defines object-file format specific implementations of
  11 // MachineModuleInfoImpl.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef LLVM_CODEGEN_MACHINEMODULEINFOIMPLS_H
  16 #define LLVM_CODEGEN_MACHINEMODULEINFOIMPLS_H
  17
  18 #include "llvm/CodeGen/MachineModuleInfo.h"
  19
  20 namespace llvm {
  21   class MCSymbol;
  22
  23   /// MachineModuleInfoMachO - This is a MachineModuleInfoImpl implementation
  24   /// for MachO targets.
  25   class MachineModuleInfoMachO : public MachineModuleInfoImpl {
  26     /// FnStubs - Darwin '$stub' stubs.  The key is something like "Lfoo$stub",
  27     /// the value is something like "_foo".
  28     DenseMap<MCSymbol*, MCSymbol*> FnStubs;
  29
  30     /// GVStubs - Darwin '$non_lazy_ptr' stubs.  The key is something like
  31     /// "Lfoo$non_lazy_ptr", the value is something like "_foo".
  32     DenseMap<MCSymbol*, MCSymbol*> GVStubs;
  33
  34     /// HiddenGVStubs - Darwin '$non_lazy_ptr' stubs.  The key is something like
  35     /// "Lfoo$non_lazy_ptr", the value is something like "_foo".  Unlike GVStubs
  36     /// these are for things with hidden visibility.
  37     DenseMap<MCSymbol*, MCSymbol*> HiddenGVStubs;
  38
  39     virtual void Anchor();  // Out of line virtual method.
  40   public:
  41     MachineModuleInfoMachO(const MachineModuleInfo &) {}
  42
  43     MCSymbol *&getFnStubEntry(MCSymbol *Sym) {
  44       assert(Sym && "Key cannot be null");
  45       return FnStubs[Sym];
  46     }
  47
  48     MCSymbol *&getGVStubEntry(MCSymbol *Sym) {
  49       assert(Sym && "Key cannot be null");
  50       return GVStubs[Sym];
  51     }
  52
  53     MCSymbol *&getHiddenGVStubEntry(MCSymbol *Sym) {
  54       assert(Sym && "Key cannot be null");
  55       return HiddenGVStubs[Sym];
  56     }
  57
  58     /// Accessor methods to return the set of stubs in sorted order.
  59     typedef std::vector<std::pair<MCSymbol*, MCSymbol*> > SymbolListTy;
  60
  61     SymbolListTy GetFnStubList() const {
  62       return GetSortedStubs(FnStubs);
  63     }
  64     SymbolListTy GetGVStubList() const {
  65       return GetSortedStubs(GVStubs);
  66     }
  67     SymbolListTy GetHiddenGVStubList() const {
  68       return GetSortedStubs(HiddenGVStubs);
  69     }
  70
  71   private:
  72     static SymbolListTy
  73     GetSortedStubs(const DenseMap<MCSymbol*, MCSymbol*> &Map);
  74   };
  75
  76 } // end namespace llvm
  77
  78 #endif