contrib/llvm/tools/lld/ELF/OutputSections.h

   1 //===- OutputSections.h -----------------------------------------*- C++ -*-===//
   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
  10 #ifndef LLD_ELF_OUTPUT_SECTIONS_H
  11 #define LLD_ELF_OUTPUT_SECTIONS_H
  12
  13 #include "Config.h"
  14 #include "InputSection.h"
  15 #include "Relocations.h"
  16
  17 #include "lld/Core/LLVM.h"
  18 #include "llvm/MC/StringTableBuilder.h"
  19 #include "llvm/Object/ELF.h"
  20
  21 namespace lld {
  22 namespace elf {
  23
  24 struct PhdrEntry;
  25 class SymbolBody;
  26 struct EhSectionPiece;
  27 class EhInputSection;
  28 class InputSection;
  29 class InputSectionBase;
  30 class MergeInputSection;
  31 class OutputSection;
  32 template <class ELFT> class ObjectFile;
  33 template <class ELFT> class SharedFile;
  34 class SharedSymbol;
  35 class DefinedRegular;
  36
  37 // This represents a section in an output file.
  38 // It is composed of multiple InputSections.
  39 // The writer creates multiple OutputSections and assign them unique,
  40 // non-overlapping file offsets and VAs.
  41 class OutputSection final : public SectionBase {
  42 public:
  43   OutputSection(StringRef Name, uint32_t Type, uint64_t Flags);
  44
  45   static bool classof(const SectionBase *S) {
  46     return S->kind() == SectionBase::Output;
  47   }
  48
  49   uint64_t getLMA() const { return Addr + LMAOffset; }
  50   template <typename ELFT> void writeHeaderTo(typename ELFT::Shdr *SHdr);
  51
  52   unsigned SectionIndex;
  53
  54   uint32_t getPhdrFlags() const;
  55
  56   void updateAlignment(uint32_t Val) {
  57     if (Val > Alignment)
  58       Alignment = Val;
  59   }
  60
  61   // If true, this section will be page aligned on disk.
  62   // Typically the first section of each PT_LOAD segment has this flag.
  63   bool PageAlign = false;
  64
  65   // Pointer to the first section in PT_LOAD segment, which this section
  66   // also resides in. This field is used to correctly compute file offset
  67   // of a section. When two sections share the same load segment, difference
  68   // between their file offsets should be equal to difference between their
  69   // virtual addresses. To compute some section offset we use the following
  70   // formula: Off = Off_first + VA - VA_first.
  71   OutputSection *FirstInPtLoad = nullptr;
  72
  73   // The following fields correspond to Elf_Shdr members.
  74   uint64_t Size = 0;
  75   uint64_t Offset = 0;
  76   uint64_t LMAOffset = 0;
  77   uint64_t Addr = 0;
  78   uint32_t ShName = 0;
  79
  80   void addSection(InputSection *S);
  81   void sort(std::function<int(InputSectionBase *S)> Order);
  82   void sortInitFini();
  83   void sortCtorsDtors();
  84   uint32_t getFiller();
  85   template <class ELFT> void writeTo(uint8_t *Buf);
  86   template <class ELFT> void finalize();
  87   void assignOffsets();
  88   std::vector<InputSection *> Sections;
  89
  90   // Location in the output buffer.
  91   uint8_t *Loc = nullptr;
  92 };
  93
  94 // All output sections that are handled by the linker specially are
  95 // globally accessible. Writer initializes them, so don't use them
  96 // until Writer is initialized.
  97 struct Out {
  98   static uint8_t First;
  99   static OutputSection *Opd;
 100   static uint8_t *OpdBuf;
 101   static PhdrEntry *TlsPhdr;
 102   static OutputSection *DebugInfo;
 103   static OutputSection *ElfHeader;
 104   static OutputSection *ProgramHeaders;
 105   static OutputSection *PreinitArray;
 106   static OutputSection *InitArray;
 107   static OutputSection *FiniArray;
 108 };
 109
 110 struct SectionKey {
 111   StringRef Name;
 112   uint64_t Flags;
 113   uint32_t Alignment;
 114 };
 115 }
 116 }
 117 namespace llvm {
 118 template <> struct DenseMapInfo<lld::elf::SectionKey> {
 119   static lld::elf::SectionKey getEmptyKey();
 120   static lld::elf::SectionKey getTombstoneKey();
 121   static unsigned getHashValue(const lld::elf::SectionKey &Val);
 122   static bool isEqual(const lld::elf::SectionKey &LHS,
 123                       const lld::elf::SectionKey &RHS);
 124 };
 125 }
 126 namespace lld {
 127 namespace elf {
 128
 129 // This class knows how to create an output section for a given
 130 // input section. Output section type is determined by various
 131 // factors, including input section's sh_flags, sh_type and
 132 // linker scripts.
 133 class OutputSectionFactory {
 134 public:
 135   OutputSectionFactory(std::vector<OutputSection *> &OutputSections);
 136   ~OutputSectionFactory();
 137
 138   void addInputSec(InputSectionBase *IS, StringRef OutsecName);
 139
 140 private:
 141   llvm::SmallDenseMap<SectionKey, OutputSection *> Map;
 142   std::vector<OutputSection *> &OutputSections;
 143 };
 144
 145 uint64_t getHeaderSize();
 146
 147 } // namespace elf
 148 } // namespace lld
 149
 150
 151 #endif