lib/ReaderWriter/ELF/X86/X86DynamicLibraryWriter.h

   1 //===- lib/ReaderWriter/ELF/X86/X86DynamicLibraryWriter.h -----------------===//
   2 //
   3 //                             The LLVM Linker
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 #ifndef X86_X86_DYNAMIC_LIBRARY_WRITER_H
  10 #define X86_X86_DYNAMIC_LIBRARY_WRITER_H
  11
  12 #include "DynamicLibraryWriter.h"
  13 #include "X86LinkingContext.h"
  14
  15 namespace lld {
  16 namespace elf {
  17
  18 template <class ELFT>
  19 class X86DynamicLibraryWriter : public DynamicLibraryWriter<ELFT> {
  20 public:
  21   X86DynamicLibraryWriter(X86LinkingContext &context,
  22                           X86TargetLayout<ELFT> &layout);
  23
  24 protected:
  25   // Add any runtime files and their atoms to the output
  26   virtual bool createImplicitFiles(std::vector<std::unique_ptr<File>> &);
  27
  28   virtual void finalizeDefaultAtomValues() {
  29     return DynamicLibraryWriter<ELFT>::finalizeDefaultAtomValues();
  30   }
  31
  32   virtual void addDefaultAtoms() {
  33     return DynamicLibraryWriter<ELFT>::addDefaultAtoms();
  34   }
  35
  36 private:
  37   class GOTFile : public SimpleFile {
  38   public:
  39     GOTFile(const ELFLinkingContext &eti) : SimpleFile("GOTFile") {}
  40     llvm::BumpPtrAllocator _alloc;
  41   };
  42
  43   std::unique_ptr<GOTFile> _gotFile;
  44   X86LinkingContext &_context;
  45   X86TargetLayout<ELFT> &_x86Layout;
  46 };
  47
  48 template <class ELFT>
  49 X86DynamicLibraryWriter<ELFT>::X86DynamicLibraryWriter(
  50     X86LinkingContext &context, X86TargetLayout<ELFT> &layout)
  51     : DynamicLibraryWriter<ELFT>(context, layout),
  52       _gotFile(new GOTFile(context)), _context(context), _x86Layout(layout) {}
  53
  54 template <class ELFT>
  55 bool X86DynamicLibraryWriter<ELFT>::createImplicitFiles(
  56     std::vector<std::unique_ptr<File>> &result) {
  57   DynamicLibraryWriter<ELFT>::createImplicitFiles(result);
  58   _gotFile->addAtom(*new (_gotFile->_alloc) GLOBAL_OFFSET_TABLEAtom(*_gotFile));
  59   _gotFile->addAtom(*new (_gotFile->_alloc) DYNAMICAtom(*_gotFile));
  60   result.push_back(std::move(_gotFile));
  61   return true;
  62 }
  63
  64 } // namespace elf
  65 } // namespace lld
  66
  67 #endif