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     SymbolListTy GetFnStubList() const {
  60       return GetSortedStubs(FnStubs);
  61     }
  62     SymbolListTy GetGVStubList() const {
  63       return GetSortedStubs(GVStubs);
  64     }
  65     SymbolListTy GetHiddenGVStubList() const {
  66       return GetSortedStubs(HiddenGVStubs);
  67     }
  68   };
  69
  70   /// MachineModuleInfoELF - This is a MachineModuleInfoImpl implementation
  71   /// for ELF targets.
  72   class MachineModuleInfoELF : public MachineModuleInfoImpl {
  73     /// GVStubs - These stubs are used to materialize global addresses in PIC
  74     /// mode.
  75     DenseMap<MCSymbol*, MCSymbol*> GVStubs;
  76
  77     virtual void Anchor();  // Out of line virtual method.
  78   public:
  79     MachineModuleInfoELF(const MachineModuleInfo &) {}
  80
  81     MCSymbol *&getGVStubEntry(MCSymbol *Sym) {
  82       assert(Sym && "Key cannot be null");
  83       return GVStubs[Sym];
  84     }
  85
  86     /// Accessor methods to return the set of stubs in sorted order.
  87
  88     SymbolListTy GetGVStubList() const {
  89       return GetSortedStubs(GVStubs);
  90     }
  91   };
  92
  93 } // end namespace llvm
  94
  95 #endif