Nick Lewycky | 4288f9e | 2013-03-09 09:32:16 +0000 | [diff] [blame] | 1 | //===- lib/MC/ELFObjectWriter.cpp - ELF File Writer -----------------------===// |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 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 implements ELF object file writer information. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 14 | #include "llvm/ADT/ArrayRef.h" |
| 15 | #include "llvm/ADT/DenseMap.h" |
Chandler Carruth | 6bda14b | 2017-06-06 11:49:48 +0000 | [diff] [blame] | 16 | #include "llvm/ADT/STLExtras.h" |
Rafael Espindola | 29abd97 | 2011-12-22 03:38:00 +0000 | [diff] [blame] | 17 | #include "llvm/ADT/SmallString.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 18 | #include "llvm/ADT/SmallVector.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 19 | #include "llvm/ADT/StringRef.h" |
| 20 | #include "llvm/ADT/Twine.h" |
Zachary Turner | 264b5d9 | 2017-06-07 03:48:56 +0000 | [diff] [blame] | 21 | #include "llvm/BinaryFormat/ELF.h" |
Rafael Espindola | ceecfe5b | 2017-07-11 23:56:10 +0000 | [diff] [blame] | 22 | #include "llvm/MC/MCAsmBackend.h" |
David Blaikie | 8019bf8 | 2014-04-10 21:53:53 +0000 | [diff] [blame] | 23 | #include "llvm/MC/MCAsmInfo.h" |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 24 | #include "llvm/MC/MCAsmLayout.h" |
Rafael Espindola | 29abd97 | 2011-12-22 03:38:00 +0000 | [diff] [blame] | 25 | #include "llvm/MC/MCAssembler.h" |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 26 | #include "llvm/MC/MCContext.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 27 | #include "llvm/MC/MCELFObjectWriter.h" |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 28 | #include "llvm/MC/MCExpr.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 29 | #include "llvm/MC/MCFixup.h" |
Rafael Espindola | ceecfe5b | 2017-07-11 23:56:10 +0000 | [diff] [blame] | 30 | #include "llvm/MC/MCFixupKindInfo.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 31 | #include "llvm/MC/MCFragment.h" |
Craig Topper | 6e80c28 | 2012-03-26 06:58:25 +0000 | [diff] [blame] | 32 | #include "llvm/MC/MCObjectWriter.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 33 | #include "llvm/MC/MCSection.h" |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 34 | #include "llvm/MC/MCSectionELF.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 35 | #include "llvm/MC/MCSymbol.h" |
Rafael Espindola | a869576 | 2015-06-02 00:25:12 +0000 | [diff] [blame] | 36 | #include "llvm/MC/MCSymbolELF.h" |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 37 | #include "llvm/MC/MCValue.h" |
Rafael Espindola | 97de474 | 2014-07-03 02:01:39 +0000 | [diff] [blame] | 38 | #include "llvm/MC/StringTableBuilder.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 39 | #include "llvm/Support/Allocator.h" |
| 40 | #include "llvm/Support/Casting.h" |
David Blaikie | 8019bf8 | 2014-04-10 21:53:53 +0000 | [diff] [blame] | 41 | #include "llvm/Support/Compression.h" |
Chandler Carruth | d990388 | 2015-01-14 11:23:27 +0000 | [diff] [blame] | 42 | #include "llvm/Support/Endian.h" |
George Rimar | 167ca4a | 2017-01-17 15:45:07 +0000 | [diff] [blame] | 43 | #include "llvm/Support/Error.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 44 | #include "llvm/Support/ErrorHandling.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 45 | #include "llvm/Support/Host.h" |
| 46 | #include "llvm/Support/MathExtras.h" |
| 47 | #include "llvm/Support/SMLoc.h" |
Rafael Espindola | fc063e8 | 2015-10-22 18:32:06 +0000 | [diff] [blame] | 48 | #include "llvm/Support/StringSaver.h" |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 49 | #include "llvm/Support/SwapByteOrder.h" |
| 50 | #include "llvm/Support/raw_ostream.h" |
| 51 | #include <algorithm> |
| 52 | #include <cassert> |
| 53 | #include <cstddef> |
| 54 | #include <cstdint> |
| 55 | #include <map> |
| 56 | #include <memory> |
| 57 | #include <string> |
| 58 | #include <utility> |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 59 | #include <vector> |
Eugene Zelenko | ecefe5a | 2016-02-02 18:20:45 +0000 | [diff] [blame] | 60 | |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 61 | using namespace llvm; |
| 62 | |
Jason W Kim | c09e726 | 2011-05-11 22:53:06 +0000 | [diff] [blame] | 63 | #undef DEBUG_TYPE |
| 64 | #define DEBUG_TYPE "reloc-info" |
| 65 | |
Rafael Espindola | 29abd97 | 2011-12-22 03:38:00 +0000 | [diff] [blame] | 66 | namespace { |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 67 | |
Eugene Zelenko | 7975b99 | 2017-04-26 22:31:39 +0000 | [diff] [blame] | 68 | using SectionIndexMapTy = DenseMap<const MCSectionELF *, uint32_t>; |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 69 | |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 70 | class ELFObjectWriter; |
| 71 | |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 72 | class SymbolTableWriter { |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 73 | ELFObjectWriter &EWriter; |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 74 | bool Is64Bit; |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 75 | |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 76 | // indexes we are going to write to .symtab_shndx. |
| 77 | std::vector<uint32_t> ShndxIndexes; |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 78 | |
| 79 | // The numbel of symbols written so far. |
| 80 | unsigned NumWritten; |
| 81 | |
| 82 | void createSymtabShndx(); |
| 83 | |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 84 | template <typename T> void write(T Value); |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 85 | |
| 86 | public: |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 87 | SymbolTableWriter(ELFObjectWriter &EWriter, bool Is64Bit); |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 88 | |
| 89 | void writeSymbol(uint32_t name, uint8_t info, uint64_t value, uint64_t size, |
| 90 | uint8_t other, uint32_t shndx, bool Reserved); |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 91 | |
| 92 | ArrayRef<uint32_t> getShndxIndexes() const { return ShndxIndexes; } |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 93 | }; |
| 94 | |
Rafael Espindola | 29abd97 | 2011-12-22 03:38:00 +0000 | [diff] [blame] | 95 | class ELFObjectWriter : public MCObjectWriter { |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 96 | static uint64_t SymbolValue(const MCSymbol &Sym, const MCAsmLayout &Layout); |
| 97 | static bool isInSymtab(const MCAsmLayout &Layout, const MCSymbolELF &Symbol, |
| 98 | bool Used, bool Renamed); |
Rafael Espindola | 29abd97 | 2011-12-22 03:38:00 +0000 | [diff] [blame] | 99 | |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 100 | /// Helper struct for containing some precomputed information on symbols. |
| 101 | struct ELFSymbolData { |
| 102 | const MCSymbolELF *Symbol; |
| 103 | uint32_t SectionIndex; |
| 104 | StringRef Name; |
Rafael Espindola | 29abd97 | 2011-12-22 03:38:00 +0000 | [diff] [blame] | 105 | |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 106 | // Support lexicographic sorting. |
| 107 | bool operator<(const ELFSymbolData &RHS) const { |
| 108 | unsigned LHSType = Symbol->getType(); |
| 109 | unsigned RHSType = RHS.Symbol->getType(); |
| 110 | if (LHSType == ELF::STT_SECTION && RHSType != ELF::STT_SECTION) |
| 111 | return false; |
| 112 | if (LHSType != ELF::STT_SECTION && RHSType == ELF::STT_SECTION) |
| 113 | return true; |
| 114 | if (LHSType == ELF::STT_SECTION && RHSType == ELF::STT_SECTION) |
| 115 | return SectionIndex < RHS.SectionIndex; |
| 116 | return Name < RHS.Name; |
Rafael Espindola | 29abd97 | 2011-12-22 03:38:00 +0000 | [diff] [blame] | 117 | } |
Rafael Espindola | 29abd97 | 2011-12-22 03:38:00 +0000 | [diff] [blame] | 118 | }; |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 119 | |
| 120 | /// The target specific ELF writer instance. |
| 121 | std::unique_ptr<MCELFObjectTargetWriter> TargetObjectWriter; |
| 122 | |
| 123 | DenseMap<const MCSymbolELF *, const MCSymbolELF *> Renames; |
| 124 | |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 125 | DenseMap<const MCSectionELF *, std::vector<ELFRelocationEntry>> Relocations; |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 126 | |
| 127 | /// @} |
| 128 | /// @name Symbol Table Data |
| 129 | /// @{ |
| 130 | |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 131 | StringTableBuilder StrTabBuilder{StringTableBuilder::ELF}; |
| 132 | |
| 133 | /// @} |
| 134 | |
| 135 | // This holds the symbol table index of the last local symbol. |
| 136 | unsigned LastLocalSymbolIndex; |
| 137 | // This holds the .strtab section index. |
| 138 | unsigned StringTableIndex; |
| 139 | // This holds the .symtab section index. |
| 140 | unsigned SymbolTableIndex; |
| 141 | |
| 142 | // Sections in the order they are to be output in the section table. |
| 143 | std::vector<const MCSectionELF *> SectionTable; |
| 144 | unsigned addToSectionTable(const MCSectionELF *Sec); |
| 145 | |
| 146 | // TargetObjectWriter wrappers. |
| 147 | bool is64Bit() const { return TargetObjectWriter->is64Bit(); } |
| 148 | bool hasRelocationAddend() const { |
| 149 | return TargetObjectWriter->hasRelocationAddend(); |
| 150 | } |
| 151 | unsigned getRelocType(MCContext &Ctx, const MCValue &Target, |
| 152 | const MCFixup &Fixup, bool IsPCRel) const { |
| 153 | return TargetObjectWriter->getRelocType(Ctx, Target, Fixup, IsPCRel); |
| 154 | } |
| 155 | |
| 156 | void align(unsigned Alignment); |
| 157 | |
George Rimar | c91e38c | 2016-05-27 12:27:32 +0000 | [diff] [blame] | 158 | bool maybeWriteCompression(uint64_t Size, |
| 159 | SmallVectorImpl<char> &CompressedContents, |
| 160 | bool ZLibStyle, unsigned Alignment); |
| 161 | |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 162 | public: |
Lang Hames | dcb312b | 2017-10-09 23:53:15 +0000 | [diff] [blame] | 163 | ELFObjectWriter(std::unique_ptr<MCELFObjectTargetWriter> MOTW, |
| 164 | raw_pwrite_stream &OS, bool IsLittleEndian) |
| 165 | : MCObjectWriter(OS, IsLittleEndian), |
| 166 | TargetObjectWriter(std::move(MOTW)) {} |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 167 | |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 168 | ~ELFObjectWriter() override = default; |
| 169 | |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 170 | void reset() override { |
| 171 | Renames.clear(); |
| 172 | Relocations.clear(); |
| 173 | StrTabBuilder.clear(); |
| 174 | SectionTable.clear(); |
| 175 | MCObjectWriter::reset(); |
| 176 | } |
| 177 | |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 178 | void WriteWord(uint64_t W) { |
| 179 | if (is64Bit()) |
| 180 | write64(W); |
| 181 | else |
| 182 | write32(W); |
| 183 | } |
| 184 | |
| 185 | template <typename T> void write(T Val) { |
| 186 | if (IsLittleEndian) |
| 187 | support::endian::Writer<support::little>(getStream()).write(Val); |
| 188 | else |
| 189 | support::endian::Writer<support::big>(getStream()).write(Val); |
| 190 | } |
| 191 | |
| 192 | void writeHeader(const MCAssembler &Asm); |
| 193 | |
| 194 | void writeSymbol(SymbolTableWriter &Writer, uint32_t StringIndex, |
| 195 | ELFSymbolData &MSD, const MCAsmLayout &Layout); |
| 196 | |
| 197 | // Start and end offset of each section |
Eugene Zelenko | 7975b99 | 2017-04-26 22:31:39 +0000 | [diff] [blame] | 198 | using SectionOffsetsTy = |
| 199 | std::map<const MCSectionELF *, std::pair<uint64_t, uint64_t>>; |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 200 | |
| 201 | bool shouldRelocateWithSymbol(const MCAssembler &Asm, |
| 202 | const MCSymbolRefExpr *RefA, |
| 203 | const MCSymbol *Sym, uint64_t C, |
| 204 | unsigned Type) const; |
| 205 | |
| 206 | void recordRelocation(MCAssembler &Asm, const MCAsmLayout &Layout, |
| 207 | const MCFragment *Fragment, const MCFixup &Fixup, |
Rafael Espindola | ceecfe5b | 2017-07-11 23:56:10 +0000 | [diff] [blame] | 208 | MCValue Target, uint64_t &FixedValue) override; |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 209 | |
| 210 | // Map from a signature symbol to the group section index |
Eugene Zelenko | 7975b99 | 2017-04-26 22:31:39 +0000 | [diff] [blame] | 211 | using RevGroupMapTy = DenseMap<const MCSymbol *, unsigned>; |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 212 | |
| 213 | /// Compute the symbol table data |
| 214 | /// |
| 215 | /// \param Asm - The assembler. |
| 216 | /// \param SectionIndexMap - Maps a section to its index. |
| 217 | /// \param RevGroupMap - Maps a signature symbol to the group section. |
| 218 | void computeSymbolTable(MCAssembler &Asm, const MCAsmLayout &Layout, |
| 219 | const SectionIndexMapTy &SectionIndexMap, |
| 220 | const RevGroupMapTy &RevGroupMap, |
| 221 | SectionOffsetsTy &SectionOffsets); |
| 222 | |
| 223 | MCSectionELF *createRelocationSection(MCContext &Ctx, |
| 224 | const MCSectionELF &Sec); |
| 225 | |
| 226 | const MCSectionELF *createStringTable(MCContext &Ctx); |
| 227 | |
| 228 | void executePostLayoutBinding(MCAssembler &Asm, |
| 229 | const MCAsmLayout &Layout) override; |
| 230 | |
| 231 | void writeSectionHeader(const MCAsmLayout &Layout, |
| 232 | const SectionIndexMapTy &SectionIndexMap, |
| 233 | const SectionOffsetsTy &SectionOffsets); |
| 234 | |
| 235 | void writeSectionData(const MCAssembler &Asm, MCSection &Sec, |
| 236 | const MCAsmLayout &Layout); |
| 237 | |
| 238 | void WriteSecHdrEntry(uint32_t Name, uint32_t Type, uint64_t Flags, |
| 239 | uint64_t Address, uint64_t Offset, uint64_t Size, |
| 240 | uint32_t Link, uint32_t Info, uint64_t Alignment, |
| 241 | uint64_t EntrySize); |
| 242 | |
| 243 | void writeRelocations(const MCAssembler &Asm, const MCSectionELF &Sec); |
| 244 | |
Vassil Vassilev | 89e36cc | 2017-03-06 15:50:59 +0000 | [diff] [blame] | 245 | using MCObjectWriter::isSymbolRefDifferenceFullyResolvedImpl; |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 246 | bool isSymbolRefDifferenceFullyResolvedImpl(const MCAssembler &Asm, |
| 247 | const MCSymbol &SymA, |
| 248 | const MCFragment &FB, bool InSet, |
| 249 | bool IsPCRel) const override; |
| 250 | |
David Blaikie | a0fa262 | 2016-04-12 21:45:53 +0000 | [diff] [blame] | 251 | void writeObject(MCAssembler &Asm, const MCAsmLayout &Layout) override; |
| 252 | void writeSection(const SectionIndexMapTy &SectionIndexMap, |
| 253 | uint32_t GroupSymbolIndex, uint64_t Offset, uint64_t Size, |
| 254 | const MCSectionELF &Section); |
| 255 | }; |
Eugene Zelenko | d3a6c89 | 2017-02-11 00:27:28 +0000 | [diff] [blame] | 256 | |
Eugene Zelenko | ecefe5a | 2016-02-02 18:20:45 +0000 | [diff] [blame] | 257 | } // end anonymous namespace |
Rafael Espindola | 29abd97 | 2011-12-22 03:38:00 +0000 | [diff] [blame] | 258 | |
Rafael Espindola | b20fbb8 | 2015-06-05 18:21:00 +0000 | [diff] [blame] | 259 | void ELFObjectWriter::align(unsigned Alignment) { |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 260 | uint64_t Padding = OffsetToAlignment(getStream().tell(), Alignment); |
Rafael Espindola | b20fbb8 | 2015-06-05 18:21:00 +0000 | [diff] [blame] | 261 | WriteZeros(Padding); |
| 262 | } |
| 263 | |
Rafael Espindola | 08850b3 | 2015-05-25 21:56:55 +0000 | [diff] [blame] | 264 | unsigned ELFObjectWriter::addToSectionTable(const MCSectionELF *Sec) { |
Rafael Espindola | 03d7abb | 2015-04-30 20:53:27 +0000 | [diff] [blame] | 265 | SectionTable.push_back(Sec); |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 266 | StrTabBuilder.add(Sec->getSectionName()); |
Rafael Espindola | 03d7abb | 2015-04-30 20:53:27 +0000 | [diff] [blame] | 267 | return SectionTable.size(); |
| 268 | } |
| 269 | |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 270 | void SymbolTableWriter::createSymtabShndx() { |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 271 | if (!ShndxIndexes.empty()) |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 272 | return; |
| 273 | |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 274 | ShndxIndexes.resize(NumWritten); |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 275 | } |
| 276 | |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 277 | template <typename T> void SymbolTableWriter::write(T Value) { |
| 278 | EWriter.write(Value); |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 279 | } |
| 280 | |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 281 | SymbolTableWriter::SymbolTableWriter(ELFObjectWriter &EWriter, bool Is64Bit) |
| 282 | : EWriter(EWriter), Is64Bit(Is64Bit), NumWritten(0) {} |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 283 | |
| 284 | void SymbolTableWriter::writeSymbol(uint32_t name, uint8_t info, uint64_t value, |
| 285 | uint64_t size, uint8_t other, |
| 286 | uint32_t shndx, bool Reserved) { |
| 287 | bool LargeIndex = shndx >= ELF::SHN_LORESERVE && !Reserved; |
| 288 | |
| 289 | if (LargeIndex) |
| 290 | createSymtabShndx(); |
| 291 | |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 292 | if (!ShndxIndexes.empty()) { |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 293 | if (LargeIndex) |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 294 | ShndxIndexes.push_back(shndx); |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 295 | else |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 296 | ShndxIndexes.push_back(0); |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 297 | } |
| 298 | |
| 299 | uint16_t Index = LargeIndex ? uint16_t(ELF::SHN_XINDEX) : shndx; |
| 300 | |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 301 | if (Is64Bit) { |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 302 | write(name); // st_name |
| 303 | write(info); // st_info |
| 304 | write(other); // st_other |
| 305 | write(Index); // st_shndx |
| 306 | write(value); // st_value |
| 307 | write(size); // st_size |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 308 | } else { |
Rafael Espindola | 91fd277 | 2015-04-29 21:09:32 +0000 | [diff] [blame] | 309 | write(name); // st_name |
| 310 | write(uint32_t(value)); // st_value |
| 311 | write(uint32_t(size)); // st_size |
| 312 | write(info); // st_info |
| 313 | write(other); // st_other |
| 314 | write(Index); // st_shndx |
Rafael Espindola | 10be083 | 2014-03-25 23:44:25 +0000 | [diff] [blame] | 315 | } |
| 316 | |
| 317 | ++NumWritten; |
| 318 | } |
| 319 | |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 320 | // Emit the ELF header. |
Rafael Espindola | 8c7829b | 2015-04-29 20:39:37 +0000 | [diff] [blame] | 321 | void ELFObjectWriter::writeHeader(const MCAssembler &Asm) { |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 322 | // ELF Header |
| 323 | // ---------- |
| 324 | // |
| 325 | // Note |
| 326 | // ---- |
| 327 | // emitWord method behaves differently for ELF32 and ELF64, writing |
| 328 | // 4 bytes in the former and 8 in the latter. |
| 329 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 330 | writeBytes(ELF::ElfMagic); // e_ident[EI_MAG0] to e_ident[EI_MAG3] |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 331 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 332 | write8(is64Bit() ? ELF::ELFCLASS64 : ELF::ELFCLASS32); // e_ident[EI_CLASS] |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 333 | |
| 334 | // e_ident[EI_DATA] |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 335 | write8(isLittleEndian() ? ELF::ELFDATA2LSB : ELF::ELFDATA2MSB); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 336 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 337 | write8(ELF::EV_CURRENT); // e_ident[EI_VERSION] |
Roman Divacky | 3b727f5 | 2010-09-09 17:57:50 +0000 | [diff] [blame] | 338 | // e_ident[EI_OSABI] |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 339 | write8(TargetObjectWriter->getOSABI()); |
| 340 | write8(0); // e_ident[EI_ABIVERSION] |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 341 | |
| 342 | WriteZeros(ELF::EI_NIDENT - ELF::EI_PAD); |
| 343 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 344 | write16(ELF::ET_REL); // e_type |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 345 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 346 | write16(TargetObjectWriter->getEMachine()); // e_machine = target |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 347 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 348 | write32(ELF::EV_CURRENT); // e_version |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 349 | WriteWord(0); // e_entry, no entry point in .o file |
| 350 | WriteWord(0); // e_phoff, no program header for .o |
Rafael Espindola | 642a221 | 2015-04-14 22:54:16 +0000 | [diff] [blame] | 351 | WriteWord(0); // e_shoff = sec hdr table off in bytes |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 352 | |
Jason W Kim | 4761fba | 2011-02-04 21:41:11 +0000 | [diff] [blame] | 353 | // e_flags = whatever the target wants |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 354 | write32(Asm.getELFHeaderEFlags()); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 355 | |
| 356 | // e_ehsize = ELF header size |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 357 | write16(is64Bit() ? sizeof(ELF::Elf64_Ehdr) : sizeof(ELF::Elf32_Ehdr)); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 358 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 359 | write16(0); // e_phentsize = prog header entry size |
| 360 | write16(0); // e_phnum = # prog header entries = 0 |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 361 | |
| 362 | // e_shentsize = Section header entry size |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 363 | write16(is64Bit() ? sizeof(ELF::Elf64_Shdr) : sizeof(ELF::Elf32_Shdr)); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 364 | |
| 365 | // e_shnum = # of section header ents |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 366 | write16(0); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 367 | |
| 368 | // e_shstrndx = Section # of '.shstrtab' |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 369 | assert(StringTableIndex < ELF::SHN_LORESERVE); |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 370 | write16(StringTableIndex); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 371 | } |
| 372 | |
Duncan P. N. Exon Smith | 469f9db | 2015-05-20 04:39:01 +0000 | [diff] [blame] | 373 | uint64_t ELFObjectWriter::SymbolValue(const MCSymbol &Sym, |
Jan Sjödin | 30a52de | 2011-02-28 21:45:04 +0000 | [diff] [blame] | 374 | const MCAsmLayout &Layout) { |
Rafael Espindola | 4d37b2a | 2015-05-29 21:45:01 +0000 | [diff] [blame] | 375 | if (Sym.isCommon() && Sym.isExternal()) |
Rafael Espindola | 1467250 | 2015-05-29 17:48:04 +0000 | [diff] [blame] | 376 | return Sym.getCommonAlignment(); |
Rafael Espindola | 9735f4f | 2010-09-27 21:23:02 +0000 | [diff] [blame] | 377 | |
Rafael Espindola | fee224f | 2014-04-30 21:51:13 +0000 | [diff] [blame] | 378 | uint64_t Res; |
Duncan P. N. Exon Smith | 469f9db | 2015-05-20 04:39:01 +0000 | [diff] [blame] | 379 | if (!Layout.getSymbolOffset(Sym, Res)) |
Rafael Espindola | 553e5eb | 2014-04-30 16:59:35 +0000 | [diff] [blame] | 380 | return 0; |
| 381 | |
Duncan P. N. Exon Smith | 469f9db | 2015-05-20 04:39:01 +0000 | [diff] [blame] | 382 | if (Layout.getAssembler().isThumbFunc(&Sym)) |
Rafael Espindola | 66f96fe | 2014-03-21 22:00:29 +0000 | [diff] [blame] | 383 | Res |= 1; |
Rafael Espindola | 9735f4f | 2010-09-27 21:23:02 +0000 | [diff] [blame] | 384 | |
Rafael Espindola | 66f96fe | 2014-03-21 22:00:29 +0000 | [diff] [blame] | 385 | return Res; |
Rafael Espindola | 9735f4f | 2010-09-27 21:23:02 +0000 | [diff] [blame] | 386 | } |
| 387 | |
Jim Grosbach | 56ed0bb | 2015-06-04 23:25:54 +0000 | [diff] [blame] | 388 | void ELFObjectWriter::executePostLayoutBinding(MCAssembler &Asm, |
Rafael Espindola | 93e3cf0 | 2010-12-07 00:27:36 +0000 | [diff] [blame] | 389 | const MCAsmLayout &Layout) { |
Rafael Espindola | eb0c2c1 | 2010-10-27 15:18:17 +0000 | [diff] [blame] | 390 | // The presence of symbol versions causes undefined symbols and |
| 391 | // versions declared with @@@ to be renamed. |
Rafael Espindola | 47b4d6b | 2018-03-09 18:42:25 +0000 | [diff] [blame] | 392 | for (const std::pair<StringRef, const MCSymbol *> &P : Asm.Symvers) { |
| 393 | StringRef AliasName = P.first; |
| 394 | const auto &Symbol = cast<MCSymbolELF>(*P.second); |
Rafael Espindola | eb0c2c1 | 2010-10-27 15:18:17 +0000 | [diff] [blame] | 395 | size_t Pos = AliasName.find('@'); |
Rafael Espindola | 47b4d6b | 2018-03-09 18:42:25 +0000 | [diff] [blame] | 396 | assert(Pos != StringRef::npos); |
| 397 | |
| 398 | StringRef Prefix = AliasName.substr(0, Pos); |
| 399 | StringRef Rest = AliasName.substr(Pos); |
| 400 | StringRef Tail = Rest; |
| 401 | if (Rest.startswith("@@@")) |
| 402 | Tail = Rest.substr(Symbol.isUndefined() ? 2 : 1); |
| 403 | |
| 404 | auto *Alias = |
| 405 | cast<MCSymbolELF>(Asm.getContext().getOrCreateSymbol(Prefix + Tail)); |
| 406 | Asm.registerSymbol(*Alias); |
| 407 | const MCExpr *Value = MCSymbolRefExpr::create(&Symbol, Asm.getContext()); |
| 408 | Alias->setVariableValue(Value); |
Rafael Espindola | eb0c2c1 | 2010-10-27 15:18:17 +0000 | [diff] [blame] | 409 | |
Rafael Espindola | 6cd76e6 | 2010-10-28 18:33:03 +0000 | [diff] [blame] | 410 | // Aliases defined with .symvar copy the binding from the symbol they alias. |
| 411 | // This is the first place we are able to copy this information. |
Rafael Espindola | 47b4d6b | 2018-03-09 18:42:25 +0000 | [diff] [blame] | 412 | Alias->setExternal(Symbol.isExternal()); |
| 413 | Alias->setBinding(Symbol.getBinding()); |
Rafael Espindola | 6cd76e6 | 2010-10-28 18:33:03 +0000 | [diff] [blame] | 414 | |
Rafael Espindola | eb0c2c1 | 2010-10-27 15:18:17 +0000 | [diff] [blame] | 415 | if (!Symbol.isUndefined() && !Rest.startswith("@@@")) |
| 416 | continue; |
| 417 | |
Rafael Espindola | 26496e6 | 2010-10-27 17:56:18 +0000 | [diff] [blame] | 418 | // FIXME: produce a better error message. |
| 419 | if (Symbol.isUndefined() && Rest.startswith("@@") && |
| 420 | !Rest.startswith("@@@")) |
| 421 | report_fatal_error("A @@ version cannot be undefined"); |
| 422 | |
Vlad Tsyrklevich | 201a108 | 2018-04-27 20:32:34 +0000 | [diff] [blame] | 423 | if (Renames.count(&Symbol) && Renames[&Symbol] != Alias) |
| 424 | report_fatal_error(llvm::Twine("Multiple symbol versions defined for ") + |
| 425 | Symbol.getName()); |
| 426 | |
Rafael Espindola | 47b4d6b | 2018-03-09 18:42:25 +0000 | [diff] [blame] | 427 | Renames.insert(std::make_pair(&Symbol, Alias)); |
Rafael Espindola | eb0c2c1 | 2010-10-27 15:18:17 +0000 | [diff] [blame] | 428 | } |
| 429 | } |
| 430 | |
Roman Divacky | 5a1c549 | 2014-01-07 20:17:03 +0000 | [diff] [blame] | 431 | static uint8_t mergeTypeForSet(uint8_t origType, uint8_t newType) { |
| 432 | uint8_t Type = newType; |
| 433 | |
| 434 | // Propagation rules: |
| 435 | // IFUNC > FUNC > OBJECT > NOTYPE |
| 436 | // TLS_OBJECT > OBJECT > NOTYPE |
| 437 | // |
| 438 | // dont let the new type degrade the old type |
| 439 | switch (origType) { |
| 440 | default: |
| 441 | break; |
| 442 | case ELF::STT_GNU_IFUNC: |
| 443 | if (Type == ELF::STT_FUNC || Type == ELF::STT_OBJECT || |
| 444 | Type == ELF::STT_NOTYPE || Type == ELF::STT_TLS) |
| 445 | Type = ELF::STT_GNU_IFUNC; |
| 446 | break; |
| 447 | case ELF::STT_FUNC: |
| 448 | if (Type == ELF::STT_OBJECT || Type == ELF::STT_NOTYPE || |
| 449 | Type == ELF::STT_TLS) |
| 450 | Type = ELF::STT_FUNC; |
| 451 | break; |
| 452 | case ELF::STT_OBJECT: |
| 453 | if (Type == ELF::STT_NOTYPE) |
| 454 | Type = ELF::STT_OBJECT; |
| 455 | break; |
| 456 | case ELF::STT_TLS: |
| 457 | if (Type == ELF::STT_OBJECT || Type == ELF::STT_NOTYPE || |
| 458 | Type == ELF::STT_GNU_IFUNC || Type == ELF::STT_FUNC) |
| 459 | Type = ELF::STT_TLS; |
| 460 | break; |
| 461 | } |
| 462 | |
| 463 | return Type; |
| 464 | } |
| 465 | |
Rafael Espindola | e48421f | 2015-05-28 20:25:29 +0000 | [diff] [blame] | 466 | void ELFObjectWriter::writeSymbol(SymbolTableWriter &Writer, |
| 467 | uint32_t StringIndex, ELFSymbolData &MSD, |
Daniel Dunbar | fe0c28f | 2010-11-13 07:33:40 +0000 | [diff] [blame] | 468 | const MCAsmLayout &Layout) { |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 469 | const auto &Symbol = cast<MCSymbolELF>(*MSD.Symbol); |
Rafael Espindola | a869576 | 2015-06-02 00:25:12 +0000 | [diff] [blame] | 470 | const MCSymbolELF *Base = |
| 471 | cast_or_null<MCSymbolELF>(Layout.getBaseSymbol(Symbol)); |
Rafael Espindola | 85a8491 | 2014-03-26 00:16:43 +0000 | [diff] [blame] | 472 | |
| 473 | // This has to be in sync with when computeSymbolTable uses SHN_ABS or |
| 474 | // SHN_COMMON. |
Rafael Espindola | 4d37b2a | 2015-05-29 21:45:01 +0000 | [diff] [blame] | 475 | bool IsReserved = !Base || Symbol.isCommon(); |
Rafael Espindola | 3fe87a1 | 2010-10-31 00:16:26 +0000 | [diff] [blame] | 476 | |
Jack Carter | 2f8d9d9 | 2013-02-19 21:57:35 +0000 | [diff] [blame] | 477 | // Binding and Type share the same byte as upper and lower nibbles |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 478 | uint8_t Binding = Symbol.getBinding(); |
| 479 | uint8_t Type = Symbol.getType(); |
Rafael Espindola | a6e3a59 | 2014-03-23 03:33:20 +0000 | [diff] [blame] | 480 | if (Base) { |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 481 | Type = mergeTypeForSet(Type, Base->getType()); |
Rafael Espindola | a6e3a59 | 2014-03-23 03:33:20 +0000 | [diff] [blame] | 482 | } |
Rafael Espindola | f8794ff | 2015-06-04 00:47:43 +0000 | [diff] [blame] | 483 | uint8_t Info = (Binding << 4) | Type; |
Jack Carter | 2f8d9d9 | 2013-02-19 21:57:35 +0000 | [diff] [blame] | 484 | |
Joerg Sonnenberger | fc18473 | 2013-10-29 01:06:17 +0000 | [diff] [blame] | 485 | // Other and Visibility share the same byte with Visibility using the lower |
Jack Carter | 2f8d9d9 | 2013-02-19 21:57:35 +0000 | [diff] [blame] | 486 | // 2 bits |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 487 | uint8_t Visibility = Symbol.getVisibility(); |
Rafael Espindola | 8c006ee | 2015-06-04 05:59:23 +0000 | [diff] [blame] | 488 | uint8_t Other = Symbol.getOther() | Visibility; |
Rafael Espindola | 5f2d6a5 | 2010-10-06 21:02:29 +0000 | [diff] [blame] | 489 | |
Duncan P. N. Exon Smith | 469f9db | 2015-05-20 04:39:01 +0000 | [diff] [blame] | 490 | uint64_t Value = SymbolValue(*MSD.Symbol, Layout); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 491 | uint64_t Size = 0; |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 492 | |
Rafael Espindola | 7c23cba | 2015-05-29 17:24:52 +0000 | [diff] [blame] | 493 | const MCExpr *ESize = MSD.Symbol->getSize(); |
Rafael Espindola | a041ef1 | 2014-03-27 00:28:24 +0000 | [diff] [blame] | 494 | if (!ESize && Base) |
Rafael Espindola | 7c23cba | 2015-05-29 17:24:52 +0000 | [diff] [blame] | 495 | ESize = Base->getSize(); |
Rafael Espindola | f0591c1 | 2010-09-21 00:24:38 +0000 | [diff] [blame] | 496 | |
Rafael Espindola | 73c0ae7 | 2010-12-22 16:03:00 +0000 | [diff] [blame] | 497 | if (ESize) { |
| 498 | int64_t Res; |
Rafael Espindola | 94a88d7 | 2015-04-06 15:27:57 +0000 | [diff] [blame] | 499 | if (!ESize->evaluateKnownAbsolute(Res, Layout)) |
Rafael Espindola | 73c0ae7 | 2010-12-22 16:03:00 +0000 | [diff] [blame] | 500 | report_fatal_error("Size expression must be absolute."); |
| 501 | Size = Res; |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 502 | } |
| 503 | |
| 504 | // Write out the symbol table entry |
Rafael Espindola | e48421f | 2015-05-28 20:25:29 +0000 | [diff] [blame] | 505 | Writer.writeSymbol(StringIndex, Info, Value, Size, Other, MSD.SectionIndex, |
| 506 | IsReserved); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 507 | } |
| 508 | |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 509 | // It is always valid to create a relocation with a symbol. It is preferable |
| 510 | // to use a relocation with a section if that is possible. Using the section |
| 511 | // allows us to omit some local symbols from the symbol table. |
Rafael Espindola | b60c829 | 2014-04-29 12:46:50 +0000 | [diff] [blame] | 512 | bool ELFObjectWriter::shouldRelocateWithSymbol(const MCAssembler &Asm, |
| 513 | const MCSymbolRefExpr *RefA, |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 514 | const MCSymbol *S, uint64_t C, |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 515 | unsigned Type) const { |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 516 | const auto *Sym = cast_or_null<MCSymbolELF>(S); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 517 | // A PCRel relocation to an absolute value has no symbol (or section). We |
| 518 | // represent that with a relocation to a null section. |
| 519 | if (!RefA) |
| 520 | return false; |
Rafael Espindola | 240028d | 2010-11-14 23:53:26 +0000 | [diff] [blame] | 521 | |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 522 | MCSymbolRefExpr::VariantKind Kind = RefA->getKind(); |
| 523 | switch (Kind) { |
| 524 | default: |
| 525 | break; |
| 526 | // The .odp creation emits a relocation against the symbol ".TOC." which |
| 527 | // create a R_PPC64_TOC relocation. However the relocation symbol name |
| 528 | // in final object creation should be NULL, since the symbol does not |
| 529 | // really exist, it is just the reference to TOC base for the current |
| 530 | // object file. Since the symbol is undefined, returning false results |
| 531 | // in a relocation with a null section which is the desired result. |
| 532 | case MCSymbolRefExpr::VK_PPC_TOCBASE: |
| 533 | return false; |
| 534 | |
| 535 | // These VariantKind cause the relocation to refer to something other than |
| 536 | // the symbol itself, like a linker generated table. Since the address of |
| 537 | // symbol is not relevant, we cannot replace the symbol with the |
| 538 | // section and patch the difference in the addend. |
| 539 | case MCSymbolRefExpr::VK_GOT: |
| 540 | case MCSymbolRefExpr::VK_PLT: |
| 541 | case MCSymbolRefExpr::VK_GOTPCREL: |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 542 | case MCSymbolRefExpr::VK_PPC_GOT_LO: |
| 543 | case MCSymbolRefExpr::VK_PPC_GOT_HI: |
| 544 | case MCSymbolRefExpr::VK_PPC_GOT_HA: |
| 545 | return true; |
Rafael Espindola | 240028d | 2010-11-14 23:53:26 +0000 | [diff] [blame] | 546 | } |
Rafael Espindola | 240028d | 2010-11-14 23:53:26 +0000 | [diff] [blame] | 547 | |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 548 | // An undefined symbol is not in any section, so the relocation has to point |
| 549 | // to the symbol itself. |
Duncan P. N. Exon Smith | 469f9db | 2015-05-20 04:39:01 +0000 | [diff] [blame] | 550 | assert(Sym && "Expected a symbol"); |
| 551 | if (Sym->isUndefined()) |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 552 | return true; |
| 553 | |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 554 | unsigned Binding = Sym->getBinding(); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 555 | switch(Binding) { |
| 556 | default: |
| 557 | llvm_unreachable("Invalid Binding"); |
| 558 | case ELF::STB_LOCAL: |
| 559 | break; |
| 560 | case ELF::STB_WEAK: |
| 561 | // If the symbol is weak, it might be overridden by a symbol in another |
| 562 | // file. The relocation has to point to the symbol so that the linker |
| 563 | // can update it. |
| 564 | return true; |
| 565 | case ELF::STB_GLOBAL: |
| 566 | // Global ELF symbols can be preempted by the dynamic linker. The relocation |
| 567 | // has to point to the symbol for a reason analogous to the STB_WEAK case. |
| 568 | return true; |
Rafael Espindola | 8c3039b | 2010-11-15 16:33:49 +0000 | [diff] [blame] | 569 | } |
Rafael Espindola | 75d65b9 | 2010-09-25 05:42:19 +0000 | [diff] [blame] | 570 | |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 571 | // If a relocation points to a mergeable section, we have to be careful. |
| 572 | // If the offset is zero, a relocation with the section will encode the |
| 573 | // same information. With a non-zero offset, the situation is different. |
| 574 | // For example, a relocation can point 42 bytes past the end of a string. |
| 575 | // If we change such a relocation to use the section, the linker would think |
| 576 | // that it pointed to another string and subtracting 42 at runtime will |
| 577 | // produce the wrong value. |
Saleem Abdulrasool | 9b106ea | 2016-11-22 04:32:54 +0000 | [diff] [blame] | 578 | if (Sym->isInSection()) { |
| 579 | auto &Sec = cast<MCSectionELF>(Sym->getSection()); |
| 580 | unsigned Flags = Sec.getFlags(); |
| 581 | if (Flags & ELF::SHF_MERGE) { |
| 582 | if (C != 0) |
| 583 | return true; |
Rafael Espindola | b1b4978 | 2014-04-02 12:15:20 +0000 | [diff] [blame] | 584 | |
Saleem Abdulrasool | 9b106ea | 2016-11-22 04:32:54 +0000 | [diff] [blame] | 585 | // It looks like gold has a bug (http://sourceware.org/PR16794) and can |
| 586 | // only handle section relocations to mergeable sections if using RELA. |
| 587 | if (!hasRelocationAddend()) |
| 588 | return true; |
| 589 | } |
| 590 | |
| 591 | // Most TLS relocations use a got, so they need the symbol. Even those that |
| 592 | // are just an offset (@tpoff), require a symbol in gold versions before |
| 593 | // 5efeedf61e4fe720fd3e9a08e6c91c10abb66d42 (2014-09-26) which fixed |
| 594 | // http://sourceware.org/PR16773. |
| 595 | if (Flags & ELF::SHF_TLS) |
Rafael Espindola | b1b4978 | 2014-04-02 12:15:20 +0000 | [diff] [blame] | 596 | return true; |
Rafael Espindola | 9f75d5d | 2010-11-24 21:57:39 +0000 | [diff] [blame] | 597 | } |
| 598 | |
Rafael Espindola | 9ef8441 | 2014-04-11 19:18:01 +0000 | [diff] [blame] | 599 | // If the symbol is a thumb function the final relocation must set the lowest |
| 600 | // bit. With a symbol that is done by just having the symbol have that bit |
| 601 | // set, so we would lose the bit if we relocated with the section. |
| 602 | // FIXME: We could use the section but add the bit to the relocation value. |
Duncan P. N. Exon Smith | 469f9db | 2015-05-20 04:39:01 +0000 | [diff] [blame] | 603 | if (Asm.isThumbFunc(Sym)) |
Rafael Espindola | 9ef8441 | 2014-04-11 19:18:01 +0000 | [diff] [blame] | 604 | return true; |
| 605 | |
Rafael Espindola | ece40ca | 2015-05-29 18:26:09 +0000 | [diff] [blame] | 606 | if (TargetObjectWriter->needsRelocateWithSymbol(*Sym, Type)) |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 607 | return true; |
| 608 | return false; |
Rafael Espindola | 75d65b9 | 2010-09-25 05:42:19 +0000 | [diff] [blame] | 609 | } |
| 610 | |
Rafael Espindola | db8a586 | 2015-04-17 20:05:17 +0000 | [diff] [blame] | 611 | // True if the assembler knows nothing about the final value of the symbol. |
| 612 | // This doesn't cover the comdat issues, since in those cases the assembler |
| 613 | // can at least know that all symbols in the section will move together. |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 614 | static bool isWeak(const MCSymbolELF &Sym) { |
| 615 | if (Sym.getType() == ELF::STT_GNU_IFUNC) |
Rafael Espindola | a635d83 | 2015-04-17 08:46:11 +0000 | [diff] [blame] | 616 | return true; |
| 617 | |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 618 | switch (Sym.getBinding()) { |
Rafael Espindola | a635d83 | 2015-04-17 08:46:11 +0000 | [diff] [blame] | 619 | default: |
| 620 | llvm_unreachable("Unknown binding"); |
| 621 | case ELF::STB_LOCAL: |
| 622 | return false; |
| 623 | case ELF::STB_GLOBAL: |
Rafael Espindola | db8a586 | 2015-04-17 20:05:17 +0000 | [diff] [blame] | 624 | return false; |
Rafael Espindola | a635d83 | 2015-04-17 08:46:11 +0000 | [diff] [blame] | 625 | case ELF::STB_WEAK: |
| 626 | case ELF::STB_GNU_UNIQUE: |
| 627 | return true; |
| 628 | } |
Rafael Espindola | c9e7068 | 2015-03-24 23:48:44 +0000 | [diff] [blame] | 629 | } |
| 630 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 631 | void ELFObjectWriter::recordRelocation(MCAssembler &Asm, |
Jason W Kim | 495c2bb | 2010-12-06 21:57:34 +0000 | [diff] [blame] | 632 | const MCAsmLayout &Layout, |
| 633 | const MCFragment *Fragment, |
Rafael Espindola | 2658554 | 2015-01-19 21:11:14 +0000 | [diff] [blame] | 634 | const MCFixup &Fixup, MCValue Target, |
Rafael Espindola | ceecfe5b | 2017-07-11 23:56:10 +0000 | [diff] [blame] | 635 | uint64_t &FixedValue) { |
| 636 | MCAsmBackend &Backend = Asm.getBackend(); |
| 637 | bool IsPCRel = Backend.getFixupKindInfo(Fixup.getKind()).Flags & |
| 638 | MCFixupKindInfo::FKF_IsPCRel; |
Rafael Espindola | 7549f87 | 2015-05-26 00:36:57 +0000 | [diff] [blame] | 639 | const MCSectionELF &FixupSection = cast<MCSectionELF>(*Fragment->getParent()); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 640 | uint64_t C = Target.getConstant(); |
| 641 | uint64_t FixupOffset = Layout.getFragmentOffset(Fragment) + Fixup.getOffset(); |
Rafael Espindola | 00ebfd4 | 2016-01-13 22:23:36 +0000 | [diff] [blame] | 642 | MCContext &Ctx = Asm.getContext(); |
Jason W Kim | 495c2bb | 2010-12-06 21:57:34 +0000 | [diff] [blame] | 643 | |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 644 | if (const MCSymbolRefExpr *RefB = Target.getSymB()) { |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 645 | // Let A, B and C being the components of Target and R be the location of |
| 646 | // the fixup. If the fixup is not pcrel, we want to compute (A - B + C). |
| 647 | // If it is pcrel, we want to compute (A - B + C - R). |
Jason W Kim | 495c2bb | 2010-12-06 21:57:34 +0000 | [diff] [blame] | 648 | |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 649 | // In general, ELF has no relocations for -B. It can only represent (A + C) |
| 650 | // or (A + C - R). If B = R + K and the relocation is not pcrel, we can |
| 651 | // replace B to implement it: (A - R - K + C) |
Oliver Stannard | 9be59af | 2015-11-17 10:00:43 +0000 | [diff] [blame] | 652 | if (IsPCRel) { |
Rafael Espindola | 00ebfd4 | 2016-01-13 22:23:36 +0000 | [diff] [blame] | 653 | Ctx.reportError( |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 654 | Fixup.getLoc(), |
| 655 | "No relocation available to represent this relative expression"); |
Oliver Stannard | 9be59af | 2015-11-17 10:00:43 +0000 | [diff] [blame] | 656 | return; |
| 657 | } |
David Majnemer | a45a176 | 2014-01-06 07:39:46 +0000 | [diff] [blame] | 658 | |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 659 | const auto &SymB = cast<MCSymbolELF>(RefB->getSymbol()); |
Jason W Kim | 495c2bb | 2010-12-06 21:57:34 +0000 | [diff] [blame] | 660 | |
Oliver Stannard | 9be59af | 2015-11-17 10:00:43 +0000 | [diff] [blame] | 661 | if (SymB.isUndefined()) { |
Rafael Espindola | 00ebfd4 | 2016-01-13 22:23:36 +0000 | [diff] [blame] | 662 | Ctx.reportError(Fixup.getLoc(), |
| 663 | Twine("symbol '") + SymB.getName() + |
| 664 | "' can not be undefined in a subtraction expression"); |
Oliver Stannard | 9be59af | 2015-11-17 10:00:43 +0000 | [diff] [blame] | 665 | return; |
| 666 | } |
Jason W Kim | 495c2bb | 2010-12-06 21:57:34 +0000 | [diff] [blame] | 667 | |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 668 | assert(!SymB.isAbsolute() && "Should have been folded"); |
| 669 | const MCSection &SecB = SymB.getSection(); |
Oliver Stannard | 9be59af | 2015-11-17 10:00:43 +0000 | [diff] [blame] | 670 | if (&SecB != &FixupSection) { |
Rafael Espindola | 00ebfd4 | 2016-01-13 22:23:36 +0000 | [diff] [blame] | 671 | Ctx.reportError(Fixup.getLoc(), |
| 672 | "Cannot represent a difference across sections"); |
Oliver Stannard | 9be59af | 2015-11-17 10:00:43 +0000 | [diff] [blame] | 673 | return; |
| 674 | } |
Jason W Kim | 495c2bb | 2010-12-06 21:57:34 +0000 | [diff] [blame] | 675 | |
Duncan P. N. Exon Smith | 2a40483 | 2015-05-19 23:53:20 +0000 | [diff] [blame] | 676 | uint64_t SymBOffset = Layout.getSymbolOffset(SymB); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 677 | uint64_t K = SymBOffset - FixupOffset; |
| 678 | IsPCRel = true; |
| 679 | C -= K; |
Jason W Kim | 495c2bb | 2010-12-06 21:57:34 +0000 | [diff] [blame] | 680 | } |
| 681 | |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 682 | // We either rejected the fixup or folded B into C at this point. |
| 683 | const MCSymbolRefExpr *RefA = Target.getSymA(); |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 684 | const auto *SymA = RefA ? cast<MCSymbolELF>(&RefA->getSymbol()) : nullptr; |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 685 | |
Rafael Espindola | 212fdde | 2015-06-03 21:52:06 +0000 | [diff] [blame] | 686 | bool ViaWeakRef = false; |
| 687 | if (SymA && SymA->isVariable()) { |
| 688 | const MCExpr *Expr = SymA->getVariableValue(); |
| 689 | if (const auto *Inner = dyn_cast<MCSymbolRefExpr>(Expr)) { |
| 690 | if (Inner->getKind() == MCSymbolRefExpr::VK_WEAKREF) { |
| 691 | SymA = cast<MCSymbolELF>(&Inner->getSymbol()); |
| 692 | ViaWeakRef = true; |
| 693 | } |
| 694 | } |
| 695 | } |
| 696 | |
Rafael Espindola | 8340f94 | 2016-01-13 22:56:57 +0000 | [diff] [blame] | 697 | unsigned Type = getRelocType(Ctx, Target, Fixup, IsPCRel); |
Daniel Sanders | a463d31 | 2016-05-06 13:49:25 +0000 | [diff] [blame] | 698 | uint64_t OriginalC = C; |
Duncan P. N. Exon Smith | 469f9db | 2015-05-20 04:39:01 +0000 | [diff] [blame] | 699 | bool RelocateWithSymbol = shouldRelocateWithSymbol(Asm, RefA, SymA, C, Type); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 700 | if (!RelocateWithSymbol && SymA && !SymA->isUndefined()) |
Duncan P. N. Exon Smith | 2a40483 | 2015-05-19 23:53:20 +0000 | [diff] [blame] | 701 | C += Layout.getSymbolOffset(*SymA); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 702 | |
| 703 | uint64_t Addend = 0; |
| 704 | if (hasRelocationAddend()) { |
| 705 | Addend = C; |
| 706 | C = 0; |
| 707 | } |
| 708 | |
| 709 | FixedValue = C; |
| 710 | |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 711 | if (!RelocateWithSymbol) { |
| 712 | const MCSection *SecA = |
| 713 | (SymA && !SymA->isUndefined()) ? &SymA->getSection() : nullptr; |
Rafael Espindola | b661302 | 2014-10-17 01:48:58 +0000 | [diff] [blame] | 714 | auto *ELFSec = cast_or_null<MCSectionELF>(SecA); |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 715 | const auto *SectionSymbol = |
| 716 | ELFSec ? cast<MCSymbolELF>(ELFSec->getBeginSymbol()) : nullptr; |
Rafael Espindola | a401eee | 2015-06-04 15:33:30 +0000 | [diff] [blame] | 717 | if (SectionSymbol) |
| 718 | SectionSymbol->setUsedInReloc(); |
Daniel Sanders | a463d31 | 2016-05-06 13:49:25 +0000 | [diff] [blame] | 719 | ELFRelocationEntry Rec(FixupOffset, SectionSymbol, Type, Addend, SymA, |
| 720 | OriginalC); |
Rafael Espindola | 34948e5 | 2015-04-30 00:45:46 +0000 | [diff] [blame] | 721 | Relocations[&FixupSection].push_back(Rec); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 722 | return; |
| 723 | } |
Roman Divacky | dfbecd1 | 2011-08-04 19:08:19 +0000 | [diff] [blame] | 724 | |
Daniel Sanders | a463d31 | 2016-05-06 13:49:25 +0000 | [diff] [blame] | 725 | const auto *RenamedSymA = SymA; |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 726 | if (SymA) { |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 727 | if (const MCSymbolELF *R = Renames.lookup(SymA)) |
Daniel Sanders | a463d31 | 2016-05-06 13:49:25 +0000 | [diff] [blame] | 728 | RenamedSymA = R; |
Rafael Espindola | d7facaf | 2011-08-04 13:05:26 +0000 | [diff] [blame] | 729 | |
Rafael Espindola | 212fdde | 2015-06-03 21:52:06 +0000 | [diff] [blame] | 730 | if (ViaWeakRef) |
Daniel Sanders | a463d31 | 2016-05-06 13:49:25 +0000 | [diff] [blame] | 731 | RenamedSymA->setIsWeakrefUsedInReloc(); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 732 | else |
Daniel Sanders | a463d31 | 2016-05-06 13:49:25 +0000 | [diff] [blame] | 733 | RenamedSymA->setUsedInReloc(); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 734 | } |
Daniel Sanders | a463d31 | 2016-05-06 13:49:25 +0000 | [diff] [blame] | 735 | ELFRelocationEntry Rec(FixupOffset, RenamedSymA, Type, Addend, SymA, |
| 736 | OriginalC); |
Rafael Espindola | 34948e5 | 2015-04-30 00:45:46 +0000 | [diff] [blame] | 737 | Relocations[&FixupSection].push_back(Rec); |
Jason W Kim | 495c2bb | 2010-12-06 21:57:34 +0000 | [diff] [blame] | 738 | } |
| 739 | |
Rafael Espindola | 3b5ee55 | 2014-04-28 13:39:57 +0000 | [diff] [blame] | 740 | bool ELFObjectWriter::isInSymtab(const MCAsmLayout &Layout, |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 741 | const MCSymbolELF &Symbol, bool Used, |
Rafael Espindola | 3b5ee55 | 2014-04-28 13:39:57 +0000 | [diff] [blame] | 742 | bool Renamed) { |
Rafael Espindola | 7fadc0e | 2014-03-20 02:12:01 +0000 | [diff] [blame] | 743 | if (Symbol.isVariable()) { |
| 744 | const MCExpr *Expr = Symbol.getVariableValue(); |
| 745 | if (const MCSymbolRefExpr *Ref = dyn_cast<MCSymbolRefExpr>(Expr)) { |
| 746 | if (Ref->getKind() == MCSymbolRefExpr::VK_WEAKREF) |
| 747 | return false; |
| 748 | } |
| 749 | } |
Rafael Espindola | 1614597 | 2010-11-01 14:28:48 +0000 | [diff] [blame] | 750 | |
Rafael Espindola | ee8d151 | 2010-10-19 19:31:37 +0000 | [diff] [blame] | 751 | if (Used) |
| 752 | return true; |
| 753 | |
Rafael Espindola | eb0c2c1 | 2010-10-27 15:18:17 +0000 | [diff] [blame] | 754 | if (Renamed) |
| 755 | return false; |
| 756 | |
Rafael Espindola | ef1e863 | 2015-06-03 21:23:21 +0000 | [diff] [blame] | 757 | if (Symbol.isVariable() && Symbol.isUndefined()) { |
| 758 | // FIXME: this is here just to diagnose the case of a var = commmon_sym. |
| 759 | Layout.getBaseSymbol(Symbol); |
| 760 | return false; |
Rafael Espindola | 3b5ee55 | 2014-04-28 13:39:57 +0000 | [diff] [blame] | 761 | } |
Rafael Espindola | 9e18e96 | 2011-02-23 20:22:07 +0000 | [diff] [blame] | 762 | |
Rafael Espindola | ef1e863 | 2015-06-03 21:23:21 +0000 | [diff] [blame] | 763 | if (Symbol.isUndefined() && !Symbol.isBindingSet()) |
Rafael Espindola | a5efd6a | 2010-10-27 14:44:52 +0000 | [diff] [blame] | 764 | return false; |
| 765 | |
Rafael Espindola | ee8d151 | 2010-10-19 19:31:37 +0000 | [diff] [blame] | 766 | if (Symbol.isTemporary()) |
Rafael Espindola | b1d0789 | 2010-10-05 18:01:23 +0000 | [diff] [blame] | 767 | return false; |
| 768 | |
Rafael Espindola | a401eee | 2015-06-04 15:33:30 +0000 | [diff] [blame] | 769 | if (Symbol.getType() == ELF::STT_SECTION) |
| 770 | return false; |
| 771 | |
Rafael Espindola | b1d0789 | 2010-10-05 18:01:23 +0000 | [diff] [blame] | 772 | return true; |
| 773 | } |
| 774 | |
Rafael Espindola | ea1b394 | 2015-04-07 19:00:17 +0000 | [diff] [blame] | 775 | void ELFObjectWriter::computeSymbolTable( |
| 776 | MCAssembler &Asm, const MCAsmLayout &Layout, |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 777 | const SectionIndexMapTy &SectionIndexMap, const RevGroupMapTy &RevGroupMap, |
| 778 | SectionOffsetsTy &SectionOffsets) { |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 779 | MCContext &Ctx = Asm.getContext(); |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 780 | SymbolTableWriter Writer(*this, is64Bit()); |
| 781 | |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 782 | // Symbol table |
| 783 | unsigned EntrySize = is64Bit() ? ELF::SYMENTRY_SIZE64 : ELF::SYMENTRY_SIZE32; |
| 784 | MCSectionELF *SymtabSection = |
| 785 | Ctx.getELFSection(".symtab", ELF::SHT_SYMTAB, 0, EntrySize, ""); |
| 786 | SymtabSection->setAlignment(is64Bit() ? 8 : 4); |
| 787 | SymbolTableIndex = addToSectionTable(SymtabSection); |
| 788 | |
Rafael Espindola | b20fbb8 | 2015-06-05 18:21:00 +0000 | [diff] [blame] | 789 | align(SymtabSection->getAlignment()); |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 790 | uint64_t SecStart = getStream().tell(); |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 791 | |
| 792 | // The first entry is the undefined symbol entry. |
| 793 | Writer.writeSymbol(0, 0, 0, 0, 0, 0, false); |
| 794 | |
Rafael Espindola | cfbd35c | 2015-05-28 20:11:34 +0000 | [diff] [blame] | 795 | std::vector<ELFSymbolData> LocalSymbolData; |
| 796 | std::vector<ELFSymbolData> ExternalSymbolData; |
Rafael Espindola | cfbd35c | 2015-05-28 20:11:34 +0000 | [diff] [blame] | 797 | |
Rafael Espindola | bee6e9f | 2010-10-14 16:34:44 +0000 | [diff] [blame] | 798 | // Add the data for the symbols. |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 799 | bool HasLargeSectionIndex = false; |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 800 | for (const MCSymbol &S : Asm.symbols()) { |
| 801 | const auto &Symbol = cast<MCSymbolELF>(S); |
Rafael Espindola | ada43f6 | 2015-06-03 21:30:10 +0000 | [diff] [blame] | 802 | bool Used = Symbol.isUsedInReloc(); |
Rafael Espindola | 212fdde | 2015-06-03 21:52:06 +0000 | [diff] [blame] | 803 | bool WeakrefUsed = Symbol.isWeakrefUsedInReloc(); |
Rafael Espindola | 8c52a9b | 2015-06-03 21:41:59 +0000 | [diff] [blame] | 804 | bool isSignature = Symbol.isSignature(); |
Rafael Espindola | 7d0ba34 | 2010-11-14 04:17:37 +0000 | [diff] [blame] | 805 | |
Duncan P. N. Exon Smith | 469f9db | 2015-05-20 04:39:01 +0000 | [diff] [blame] | 806 | if (!isInSymtab(Layout, Symbol, Used || WeakrefUsed || isSignature, |
Rafael Espindola | eb0c2c1 | 2010-10-27 15:18:17 +0000 | [diff] [blame] | 807 | Renames.count(&Symbol))) |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 808 | continue; |
| 809 | |
Oliver Stannard | 9be59af | 2015-11-17 10:00:43 +0000 | [diff] [blame] | 810 | if (Symbol.isTemporary() && Symbol.isUndefined()) { |
| 811 | Ctx.reportError(SMLoc(), "Undefined temporary symbol"); |
| 812 | continue; |
| 813 | } |
Rafael Espindola | 6dff814 | 2015-06-25 20:10:45 +0000 | [diff] [blame] | 814 | |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 815 | ELFSymbolData MSD; |
Rafael Espindola | a869576 | 2015-06-02 00:25:12 +0000 | [diff] [blame] | 816 | MSD.Symbol = cast<MCSymbolELF>(&Symbol); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 817 | |
Rafael Espindola | 8c52a9b | 2015-06-03 21:41:59 +0000 | [diff] [blame] | 818 | bool Local = Symbol.getBinding() == ELF::STB_LOCAL; |
Rafael Espindola | 6dff814 | 2015-06-25 20:10:45 +0000 | [diff] [blame] | 819 | assert(Local || !Symbol.isTemporary()); |
| 820 | |
Rafael Espindola | f4b4430 | 2015-05-29 15:07:27 +0000 | [diff] [blame] | 821 | if (Symbol.isAbsolute()) { |
Rafael Espindola | 022bb76 | 2014-03-24 03:43:21 +0000 | [diff] [blame] | 822 | MSD.SectionIndex = ELF::SHN_ABS; |
Rafael Espindola | 1467250 | 2015-05-29 17:48:04 +0000 | [diff] [blame] | 823 | } else if (Symbol.isCommon()) { |
Rafael Espindola | bee6e9f | 2010-10-14 16:34:44 +0000 | [diff] [blame] | 824 | assert(!Local); |
Rafael Espindola | f0591c1 | 2010-09-21 00:24:38 +0000 | [diff] [blame] | 825 | MSD.SectionIndex = ELF::SHN_COMMON; |
Rafael Espindola | f4b4430 | 2015-05-29 15:07:27 +0000 | [diff] [blame] | 826 | } else if (Symbol.isUndefined()) { |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 827 | if (isSignature && !Used) { |
Rafael Espindola | 89feff3 | 2015-04-28 22:59:58 +0000 | [diff] [blame] | 828 | MSD.SectionIndex = RevGroupMap.lookup(&Symbol); |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 829 | if (MSD.SectionIndex >= ELF::SHN_LORESERVE) |
| 830 | HasLargeSectionIndex = true; |
| 831 | } else { |
Rafael Espindola | 7d0ba34 | 2010-11-14 04:17:37 +0000 | [diff] [blame] | 832 | MSD.SectionIndex = ELF::SHN_UNDEF; |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 833 | } |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 834 | } else { |
Rafael Espindola | d634003 | 2010-11-10 21:51:05 +0000 | [diff] [blame] | 835 | const MCSectionELF &Section = |
Rafael Espindola | f4b4430 | 2015-05-29 15:07:27 +0000 | [diff] [blame] | 836 | static_cast<const MCSectionELF &>(Symbol.getSection()); |
Rafael Espindola | d634003 | 2010-11-10 21:51:05 +0000 | [diff] [blame] | 837 | MSD.SectionIndex = SectionIndexMap.lookup(&Section); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 838 | assert(MSD.SectionIndex && "Invalid section index!"); |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 839 | if (MSD.SectionIndex >= ELF::SHN_LORESERVE) |
| 840 | HasLargeSectionIndex = true; |
Rafael Espindola | fbcf0db | 2010-10-15 15:39:06 +0000 | [diff] [blame] | 841 | } |
| 842 | |
Rafael Espindola | eb0c2c1 | 2010-10-27 15:18:17 +0000 | [diff] [blame] | 843 | StringRef Name = Symbol.getName(); |
Rafael Espindola | b661302 | 2014-10-17 01:48:58 +0000 | [diff] [blame] | 844 | |
| 845 | // Sections have their own string table |
Rafael Espindola | fc063e8 | 2015-10-22 18:32:06 +0000 | [diff] [blame] | 846 | if (Symbol.getType() != ELF::STT_SECTION) { |
| 847 | MSD.Name = Name; |
| 848 | StrTabBuilder.add(Name); |
| 849 | } |
Rafael Espindola | eb0c2c1 | 2010-10-27 15:18:17 +0000 | [diff] [blame] | 850 | |
Rafael Espindola | 10d2387 | 2015-05-29 14:20:40 +0000 | [diff] [blame] | 851 | if (Local) |
Rafael Espindola | a5efd6a | 2010-10-27 14:44:52 +0000 | [diff] [blame] | 852 | LocalSymbolData.push_back(MSD); |
| 853 | else |
| 854 | ExternalSymbolData.push_back(MSD); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 855 | } |
| 856 | |
Rafael Espindola | 2cd1951 | 2015-07-02 16:59:57 +0000 | [diff] [blame] | 857 | // This holds the .symtab_shndx section index. |
| 858 | unsigned SymtabShndxSectionIndex = 0; |
| 859 | |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 860 | if (HasLargeSectionIndex) { |
| 861 | MCSectionELF *SymtabShndxSection = |
| 862 | Ctx.getELFSection(".symtab_shndxr", ELF::SHT_SYMTAB_SHNDX, 0, 4, ""); |
| 863 | SymtabShndxSectionIndex = addToSectionTable(SymtabShndxSection); |
| 864 | SymtabShndxSection->setAlignment(4); |
| 865 | } |
| 866 | |
Rafael Espindola | 1fd3627 | 2015-05-28 19:29:15 +0000 | [diff] [blame] | 867 | ArrayRef<std::string> FileNames = Asm.getFileNames(); |
| 868 | for (const std::string &Name : FileNames) |
Rafael Espindola | 66f3c9c | 2015-05-28 18:03:20 +0000 | [diff] [blame] | 869 | StrTabBuilder.add(Name); |
Hans Wennborg | 83e6e1e | 2014-04-30 16:25:02 +0000 | [diff] [blame] | 870 | |
Rafael Espindola | 21956e4 | 2015-10-23 21:48:05 +0000 | [diff] [blame] | 871 | StrTabBuilder.finalize(); |
Hans Wennborg | 83e6e1e | 2014-04-30 16:25:02 +0000 | [diff] [blame] | 872 | |
Peter Collingbourne | 5b75fd9 | 2017-03-03 21:22:06 +0000 | [diff] [blame] | 873 | // File symbols are emitted first and handled separately from normal symbols, |
| 874 | // i.e. a non-STT_FILE symbol with the same name may appear. |
Rafael Espindola | 0cbea29 | 2015-05-28 20:00:13 +0000 | [diff] [blame] | 875 | for (const std::string &Name : FileNames) |
| 876 | Writer.writeSymbol(StrTabBuilder.getOffset(Name), |
| 877 | ELF::STT_FILE | ELF::STB_LOCAL, 0, 0, ELF::STV_DEFAULT, |
| 878 | ELF::SHN_ABS, true); |
| 879 | |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 880 | // Symbols are required to be in lexicographic order. |
| 881 | array_pod_sort(LocalSymbolData.begin(), LocalSymbolData.end()); |
| 882 | array_pod_sort(ExternalSymbolData.begin(), ExternalSymbolData.end()); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 883 | |
| 884 | // Set the symbol indices. Local symbols must come before all other |
| 885 | // symbols with non-local bindings. |
Rafael Espindola | 1fd3627 | 2015-05-28 19:29:15 +0000 | [diff] [blame] | 886 | unsigned Index = FileNames.size() + 1; |
Rafael Espindola | 0e3decf | 2010-11-14 03:12:24 +0000 | [diff] [blame] | 887 | |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 888 | for (ELFSymbolData &MSD : LocalSymbolData) { |
Daniel Jasper | 41de802 | 2015-06-23 11:31:32 +0000 | [diff] [blame] | 889 | unsigned StringIndex = MSD.Symbol->getType() == ELF::STT_SECTION |
| 890 | ? 0 |
| 891 | : StrTabBuilder.getOffset(MSD.Name); |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 892 | MSD.Symbol->setIndex(Index++); |
Rafael Espindola | e48421f | 2015-05-28 20:25:29 +0000 | [diff] [blame] | 893 | writeSymbol(Writer, StringIndex, MSD, Layout); |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 894 | } |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 895 | |
| 896 | // Write the symbol table entries. |
Rafael Espindola | 0cbea29 | 2015-05-28 20:00:13 +0000 | [diff] [blame] | 897 | LastLocalSymbolIndex = Index; |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 898 | |
Rafael Espindola | dcda997 | 2015-05-28 19:43:20 +0000 | [diff] [blame] | 899 | for (ELFSymbolData &MSD : ExternalSymbolData) { |
Rafael Espindola | e48421f | 2015-05-28 20:25:29 +0000 | [diff] [blame] | 900 | unsigned StringIndex = StrTabBuilder.getOffset(MSD.Name); |
Rafael Espindola | 0cbea29 | 2015-05-28 20:00:13 +0000 | [diff] [blame] | 901 | MSD.Symbol->setIndex(Index++); |
Rafael Espindola | e48421f | 2015-05-28 20:25:29 +0000 | [diff] [blame] | 902 | writeSymbol(Writer, StringIndex, MSD, Layout); |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 903 | assert(MSD.Symbol->getBinding() != ELF::STB_LOCAL); |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 904 | } |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 905 | |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 906 | uint64_t SecEnd = getStream().tell(); |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 907 | SectionOffsets[SymtabSection] = std::make_pair(SecStart, SecEnd); |
| 908 | |
| 909 | ArrayRef<uint32_t> ShndxIndexes = Writer.getShndxIndexes(); |
| 910 | if (ShndxIndexes.empty()) { |
| 911 | assert(SymtabShndxSectionIndex == 0); |
| 912 | return; |
| 913 | } |
| 914 | assert(SymtabShndxSectionIndex != 0); |
| 915 | |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 916 | SecStart = getStream().tell(); |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 917 | const MCSectionELF *SymtabShndxSection = |
| 918 | SectionTable[SymtabShndxSectionIndex - 1]; |
| 919 | for (uint32_t Index : ShndxIndexes) |
| 920 | write(Index); |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 921 | SecEnd = getStream().tell(); |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 922 | SectionOffsets[SymtabShndxSection] = std::make_pair(SecStart, SecEnd); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 923 | } |
| 924 | |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 925 | MCSectionELF * |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 926 | ELFObjectWriter::createRelocationSection(MCContext &Ctx, |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 927 | const MCSectionELF &Sec) { |
Rafael Espindola | 34948e5 | 2015-04-30 00:45:46 +0000 | [diff] [blame] | 928 | if (Relocations[&Sec].empty()) |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 929 | return nullptr; |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 930 | |
Rafael Espindola | 34948e5 | 2015-04-30 00:45:46 +0000 | [diff] [blame] | 931 | const StringRef SectionName = Sec.getSectionName(); |
Rafael Espindola | ead8549 | 2015-02-17 20:31:13 +0000 | [diff] [blame] | 932 | std::string RelaSectionName = hasRelocationAddend() ? ".rela" : ".rel"; |
| 933 | RelaSectionName += SectionName; |
Benjamin Kramer | fd05415 | 2010-08-17 17:56:13 +0000 | [diff] [blame] | 934 | |
Rafael Espindola | ead8549 | 2015-02-17 20:31:13 +0000 | [diff] [blame] | 935 | unsigned EntrySize; |
| 936 | if (hasRelocationAddend()) |
| 937 | EntrySize = is64Bit() ? sizeof(ELF::Elf64_Rela) : sizeof(ELF::Elf32_Rela); |
| 938 | else |
| 939 | EntrySize = is64Bit() ? sizeof(ELF::Elf64_Rel) : sizeof(ELF::Elf32_Rel); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 940 | |
Rafael Espindola | ead8549 | 2015-02-17 20:31:13 +0000 | [diff] [blame] | 941 | unsigned Flags = 0; |
Rafael Espindola | 34948e5 | 2015-04-30 00:45:46 +0000 | [diff] [blame] | 942 | if (Sec.getFlags() & ELF::SHF_GROUP) |
Rafael Espindola | ead8549 | 2015-02-17 20:31:13 +0000 | [diff] [blame] | 943 | Flags = ELF::SHF_GROUP; |
Rafael Espindola | ead8549 | 2015-02-17 20:31:13 +0000 | [diff] [blame] | 944 | |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 945 | MCSectionELF *RelaSection = Ctx.createELFRelSection( |
Rafael Espindola | ead8549 | 2015-02-17 20:31:13 +0000 | [diff] [blame] | 946 | RelaSectionName, hasRelocationAddend() ? ELF::SHT_RELA : ELF::SHT_REL, |
Rafael Espindola | 34948e5 | 2015-04-30 00:45:46 +0000 | [diff] [blame] | 947 | Flags, EntrySize, Sec.getGroup(), &Sec); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 948 | RelaSection->setAlignment(is64Bit() ? 8 : 4); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 949 | return RelaSection; |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 950 | } |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 951 | |
George Rimar | c91e38c | 2016-05-27 12:27:32 +0000 | [diff] [blame] | 952 | // Include the debug info compression header. |
| 953 | bool ELFObjectWriter::maybeWriteCompression( |
| 954 | uint64_t Size, SmallVectorImpl<char> &CompressedContents, bool ZLibStyle, |
| 955 | unsigned Alignment) { |
| 956 | if (ZLibStyle) { |
| 957 | uint64_t HdrSize = |
| 958 | is64Bit() ? sizeof(ELF::Elf32_Chdr) : sizeof(ELF::Elf64_Chdr); |
| 959 | if (Size <= HdrSize + CompressedContents.size()) |
| 960 | return false; |
| 961 | // Platform specific header is followed by compressed data. |
| 962 | if (is64Bit()) { |
| 963 | // Write Elf64_Chdr header. |
| 964 | write(static_cast<ELF::Elf64_Word>(ELF::ELFCOMPRESS_ZLIB)); |
| 965 | write(static_cast<ELF::Elf64_Word>(0)); // ch_reserved field. |
| 966 | write(static_cast<ELF::Elf64_Xword>(Size)); |
| 967 | write(static_cast<ELF::Elf64_Xword>(Alignment)); |
| 968 | } else { |
| 969 | // Write Elf32_Chdr header otherwise. |
| 970 | write(static_cast<ELF::Elf32_Word>(ELF::ELFCOMPRESS_ZLIB)); |
| 971 | write(static_cast<ELF::Elf32_Word>(Size)); |
| 972 | write(static_cast<ELF::Elf32_Word>(Alignment)); |
| 973 | } |
| 974 | return true; |
| 975 | } |
| 976 | |
| 977 | // "ZLIB" followed by 8 bytes representing the uncompressed size of the section, |
| 978 | // useful for consumers to preallocate a buffer to decompress into. |
Richard Smith | b910e56 | 2016-05-25 00:14:12 +0000 | [diff] [blame] | 979 | const StringRef Magic = "ZLIB"; |
| 980 | if (Size <= Magic.size() + sizeof(Size) + CompressedContents.size()) |
| 981 | return false; |
George Rimar | c91e38c | 2016-05-27 12:27:32 +0000 | [diff] [blame] | 982 | write(ArrayRef<char>(Magic.begin(), Magic.size())); |
| 983 | writeBE64(Size); |
Richard Smith | b910e56 | 2016-05-25 00:14:12 +0000 | [diff] [blame] | 984 | return true; |
| 985 | } |
| 986 | |
Rafael Espindola | e15b1b7 | 2015-05-27 13:30:50 +0000 | [diff] [blame] | 987 | void ELFObjectWriter::writeSectionData(const MCAssembler &Asm, MCSection &Sec, |
Rafael Espindola | 868b3f4 | 2015-04-30 21:51:58 +0000 | [diff] [blame] | 988 | const MCAsmLayout &Layout) { |
Rafael Espindola | e15b1b7 | 2015-05-27 13:30:50 +0000 | [diff] [blame] | 989 | MCSectionELF &Section = static_cast<MCSectionELF &>(Sec); |
Rafael Espindola | 868b3f4 | 2015-04-30 21:51:58 +0000 | [diff] [blame] | 990 | StringRef SectionName = Section.getSectionName(); |
David Blaikie | 8019bf8 | 2014-04-10 21:53:53 +0000 | [diff] [blame] | 991 | |
Saleem Abdulrasool | 1f62f57 | 2017-06-09 00:40:19 +0000 | [diff] [blame] | 992 | auto &MC = Asm.getContext(); |
| 993 | const auto &MAI = MC.getAsmInfo(); |
| 994 | |
Rafael Espindola | 868b3f4 | 2015-04-30 21:51:58 +0000 | [diff] [blame] | 995 | // Compressing debug_frame requires handling alignment fragments which is |
| 996 | // more work (possibly generalizing MCAssembler.cpp:writeFragment to allow |
| 997 | // for writing to arbitrary buffers) for little benefit. |
George Rimar | c91e38c | 2016-05-27 12:27:32 +0000 | [diff] [blame] | 998 | bool CompressionEnabled = |
Saleem Abdulrasool | 1f62f57 | 2017-06-09 00:40:19 +0000 | [diff] [blame] | 999 | MAI->compressDebugSections() != DebugCompressionType::None; |
George Rimar | c91e38c | 2016-05-27 12:27:32 +0000 | [diff] [blame] | 1000 | if (!CompressionEnabled || !SectionName.startswith(".debug_") || |
| 1001 | SectionName == ".debug_frame") { |
Rafael Espindola | 64acc7f | 2015-05-26 02:17:21 +0000 | [diff] [blame] | 1002 | Asm.writeSectionData(&Section, Layout); |
Rafael Espindola | 868b3f4 | 2015-04-30 21:51:58 +0000 | [diff] [blame] | 1003 | return; |
| 1004 | } |
| 1005 | |
Saleem Abdulrasool | 1f62f57 | 2017-06-09 00:40:19 +0000 | [diff] [blame] | 1006 | assert((MAI->compressDebugSections() == DebugCompressionType::Z || |
| 1007 | MAI->compressDebugSections() == DebugCompressionType::GNU) && |
| 1008 | "expected zlib or zlib-gnu style compression"); |
| 1009 | |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1010 | SmallVector<char, 128> UncompressedData; |
| 1011 | raw_svector_ostream VecOS(UncompressedData); |
| 1012 | raw_pwrite_stream &OldStream = getStream(); |
| 1013 | setStream(VecOS); |
| 1014 | Asm.writeSectionData(&Section, Layout); |
| 1015 | setStream(OldStream); |
David Blaikie | 8019bf8 | 2014-04-10 21:53:53 +0000 | [diff] [blame] | 1016 | |
Rafael Espindola | 868b3f4 | 2015-04-30 21:51:58 +0000 | [diff] [blame] | 1017 | SmallVector<char, 128> CompressedContents; |
George Rimar | 167ca4a | 2017-01-17 15:45:07 +0000 | [diff] [blame] | 1018 | if (Error E = zlib::compress( |
| 1019 | StringRef(UncompressedData.data(), UncompressedData.size()), |
| 1020 | CompressedContents)) { |
| 1021 | consumeError(std::move(E)); |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1022 | getStream() << UncompressedData; |
David Blaikie | 8019bf8 | 2014-04-10 21:53:53 +0000 | [diff] [blame] | 1023 | return; |
Rafael Espindola | 868b3f4 | 2015-04-30 21:51:58 +0000 | [diff] [blame] | 1024 | } |
David Blaikie | 8019bf8 | 2014-04-10 21:53:53 +0000 | [diff] [blame] | 1025 | |
Saleem Abdulrasool | 1f62f57 | 2017-06-09 00:40:19 +0000 | [diff] [blame] | 1026 | bool ZlibStyle = MAI->compressDebugSections() == DebugCompressionType::Z; |
George Rimar | c91e38c | 2016-05-27 12:27:32 +0000 | [diff] [blame] | 1027 | if (!maybeWriteCompression(UncompressedData.size(), CompressedContents, |
| 1028 | ZlibStyle, Sec.getAlignment())) { |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1029 | getStream() << UncompressedData; |
Rafael Espindola | 868b3f4 | 2015-04-30 21:51:58 +0000 | [diff] [blame] | 1030 | return; |
| 1031 | } |
George Rimar | c91e38c | 2016-05-27 12:27:32 +0000 | [diff] [blame] | 1032 | |
| 1033 | if (ZlibStyle) |
| 1034 | // Set the compressed flag. That is zlib style. |
| 1035 | Section.setFlags(Section.getFlags() | ELF::SHF_COMPRESSED); |
| 1036 | else |
| 1037 | // Add "z" prefix to section name. This is zlib-gnu style. |
Saleem Abdulrasool | 1f62f57 | 2017-06-09 00:40:19 +0000 | [diff] [blame] | 1038 | MC.renameELFSection(&Section, (".z" + SectionName.drop_front(1)).str()); |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1039 | getStream() << CompressedContents; |
David Blaikie | 8019bf8 | 2014-04-10 21:53:53 +0000 | [diff] [blame] | 1040 | } |
| 1041 | |
Daniel Dunbar | fe0c28f | 2010-11-13 07:33:40 +0000 | [diff] [blame] | 1042 | void ELFObjectWriter::WriteSecHdrEntry(uint32_t Name, uint32_t Type, |
| 1043 | uint64_t Flags, uint64_t Address, |
| 1044 | uint64_t Offset, uint64_t Size, |
| 1045 | uint32_t Link, uint32_t Info, |
| 1046 | uint64_t Alignment, |
| 1047 | uint64_t EntrySize) { |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 1048 | write32(Name); // sh_name: index into string table |
| 1049 | write32(Type); // sh_type |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 1050 | WriteWord(Flags); // sh_flags |
| 1051 | WriteWord(Address); // sh_addr |
| 1052 | WriteWord(Offset); // sh_offset |
| 1053 | WriteWord(Size); // sh_size |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 1054 | write32(Link); // sh_link |
| 1055 | write32(Info); // sh_info |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 1056 | WriteWord(Alignment); // sh_addralign |
| 1057 | WriteWord(EntrySize); // sh_entsize |
| 1058 | } |
| 1059 | |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1060 | void ELFObjectWriter::writeRelocations(const MCAssembler &Asm, |
Rafael Espindola | 34948e5 | 2015-04-30 00:45:46 +0000 | [diff] [blame] | 1061 | const MCSectionELF &Sec) { |
| 1062 | std::vector<ELFRelocationEntry> &Relocs = Relocations[&Sec]; |
Akira Hatanaka | 64ad2cf | 2012-03-23 23:06:45 +0000 | [diff] [blame] | 1063 | |
Rafael Espindola | f44db24 | 2015-12-17 16:22:06 +0000 | [diff] [blame] | 1064 | // We record relocations by pushing to the end of a vector. Reverse the vector |
| 1065 | // to get the relocations in the order they were created. |
| 1066 | // In most cases that is not important, but it can be for special sections |
| 1067 | // (.eh_frame) or specific relocations (TLS optimizations on SystemZ). |
| 1068 | std::reverse(Relocs.begin(), Relocs.end()); |
Rafael Espindola | d0e1652 | 2015-12-17 15:08:24 +0000 | [diff] [blame] | 1069 | |
Rafael Espindola | f44db24 | 2015-12-17 16:22:06 +0000 | [diff] [blame] | 1070 | // Sort the relocation entries. MIPS needs this. |
Petar Jovanovic | 0380d0b | 2015-04-14 13:23:34 +0000 | [diff] [blame] | 1071 | TargetObjectWriter->sortRelocs(Asm, Relocs); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 1072 | |
| 1073 | for (unsigned i = 0, e = Relocs.size(); i != e; ++i) { |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 1074 | const ELFRelocationEntry &Entry = Relocs[e - i - 1]; |
Rafael Espindola | 5e9ed90 | 2015-05-28 20:53:09 +0000 | [diff] [blame] | 1075 | unsigned Index = Entry.Symbol ? Entry.Symbol->getIndex() : 0; |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 1076 | |
Rafael Espindola | fdaae0d | 2010-12-18 03:27:34 +0000 | [diff] [blame] | 1077 | if (is64Bit()) { |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1078 | write(Entry.Offset); |
Simon Atanasyan | 9f676a7 | 2017-09-21 14:04:47 +0000 | [diff] [blame] | 1079 | if (TargetObjectWriter->getEMachine() == ELF::EM_MIPS) { |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1080 | write(uint32_t(Index)); |
Benjamin Kramer | 6c3c349 | 2010-09-09 18:01:29 +0000 | [diff] [blame] | 1081 | |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1082 | write(TargetObjectWriter->getRSsym(Entry.Type)); |
| 1083 | write(TargetObjectWriter->getRType3(Entry.Type)); |
| 1084 | write(TargetObjectWriter->getRType2(Entry.Type)); |
| 1085 | write(TargetObjectWriter->getRType(Entry.Type)); |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 1086 | } else { |
Jack Carter | 8ad0c27 | 2012-06-27 22:28:30 +0000 | [diff] [blame] | 1087 | struct ELF::Elf64_Rela ERE64; |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 1088 | ERE64.setSymbolAndType(Index, Entry.Type); |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1089 | write(ERE64.r_info); |
Jack Carter | 8ad0c27 | 2012-06-27 22:28:30 +0000 | [diff] [blame] | 1090 | } |
Rafael Espindola | fdaae0d | 2010-12-18 03:27:34 +0000 | [diff] [blame] | 1091 | if (hasRelocationAddend()) |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1092 | write(Entry.Addend); |
Benjamin Kramer | 6c3c349 | 2010-09-09 18:01:29 +0000 | [diff] [blame] | 1093 | } else { |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1094 | write(uint32_t(Entry.Offset)); |
Benjamin Kramer | 6c3c349 | 2010-09-09 18:01:29 +0000 | [diff] [blame] | 1095 | |
Rafael Espindola | bce26a1 | 2010-10-05 15:11:03 +0000 | [diff] [blame] | 1096 | struct ELF::Elf32_Rela ERE32; |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 1097 | ERE32.setSymbolAndType(Index, Entry.Type); |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1098 | write(ERE32.r_info); |
Benjamin Kramer | 6c3c349 | 2010-09-09 18:01:29 +0000 | [diff] [blame] | 1099 | |
Rafael Espindola | fdaae0d | 2010-12-18 03:27:34 +0000 | [diff] [blame] | 1100 | if (hasRelocationAddend()) |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1101 | write(uint32_t(Entry.Addend)); |
Simon Atanasyan | ede43b7 | 2017-09-21 14:04:53 +0000 | [diff] [blame] | 1102 | |
| 1103 | if (TargetObjectWriter->getEMachine() == ELF::EM_MIPS) { |
| 1104 | if (uint32_t RType = TargetObjectWriter->getRType2(Entry.Type)) { |
| 1105 | write(uint32_t(Entry.Offset)); |
| 1106 | |
| 1107 | ERE32.setSymbolAndType(0, RType); |
| 1108 | write(ERE32.r_info); |
| 1109 | write(uint32_t(0)); |
| 1110 | } |
| 1111 | if (uint32_t RType = TargetObjectWriter->getRType3(Entry.Type)) { |
| 1112 | write(uint32_t(Entry.Offset)); |
| 1113 | |
| 1114 | ERE32.setSymbolAndType(0, RType); |
| 1115 | write(ERE32.r_info); |
| 1116 | write(uint32_t(0)); |
| 1117 | } |
| 1118 | } |
Benjamin Kramer | 6c3c349 | 2010-09-09 18:01:29 +0000 | [diff] [blame] | 1119 | } |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 1120 | } |
| 1121 | } |
| 1122 | |
Rafael Espindola | b186391 | 2015-04-30 21:10:06 +0000 | [diff] [blame] | 1123 | const MCSectionELF *ELFObjectWriter::createStringTable(MCContext &Ctx) { |
Rafael Espindola | 08850b3 | 2015-05-25 21:56:55 +0000 | [diff] [blame] | 1124 | const MCSectionELF *StrtabSection = SectionTable[StringTableIndex - 1]; |
Rafael Espindola | 39751af | 2016-10-04 22:43:25 +0000 | [diff] [blame] | 1125 | StrTabBuilder.write(getStream()); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1126 | return StrtabSection; |
Rafael Espindola | a3e9a22 | 2010-11-11 18:13:52 +0000 | [diff] [blame] | 1127 | } |
| 1128 | |
Rafael Espindola | e92c1bf | 2015-05-21 19:42:35 +0000 | [diff] [blame] | 1129 | void ELFObjectWriter::writeSection(const SectionIndexMapTy &SectionIndexMap, |
| 1130 | uint32_t GroupSymbolIndex, uint64_t Offset, |
| 1131 | uint64_t Size, const MCSectionELF &Section) { |
Rafael Espindola | 5a8d781 | 2010-11-10 23:36:59 +0000 | [diff] [blame] | 1132 | uint64_t sh_link = 0; |
| 1133 | uint64_t sh_info = 0; |
| 1134 | |
| 1135 | switch(Section.getType()) { |
Rafael Espindola | 86fe2fe | 2015-04-06 03:16:51 +0000 | [diff] [blame] | 1136 | default: |
| 1137 | // Nothing to do. |
| 1138 | break; |
| 1139 | |
Rafael Espindola | 5a8d781 | 2010-11-10 23:36:59 +0000 | [diff] [blame] | 1140 | case ELF::SHT_DYNAMIC: |
Rafael Espindola | 74ef480 | 2015-04-30 21:20:06 +0000 | [diff] [blame] | 1141 | llvm_unreachable("SHT_DYNAMIC in a relocatable object"); |
Rafael Espindola | 5a8d781 | 2010-11-10 23:36:59 +0000 | [diff] [blame] | 1142 | |
| 1143 | case ELF::SHT_REL: |
| 1144 | case ELF::SHT_RELA: { |
Rafael Espindola | 3a7c0eb | 2015-02-17 20:37:50 +0000 | [diff] [blame] | 1145 | sh_link = SymbolTableIndex; |
Rafael Espindola | 5a8d781 | 2010-11-10 23:36:59 +0000 | [diff] [blame] | 1146 | assert(sh_link && ".symtab not found"); |
Evgeniy Stepanov | 43dcf4d | 2017-03-14 19:28:51 +0000 | [diff] [blame] | 1147 | const MCSection *InfoSection = Section.getAssociatedSection(); |
| 1148 | sh_info = SectionIndexMap.lookup(cast<MCSectionELF>(InfoSection)); |
Rafael Espindola | 5a8d781 | 2010-11-10 23:36:59 +0000 | [diff] [blame] | 1149 | break; |
| 1150 | } |
| 1151 | |
| 1152 | case ELF::SHT_SYMTAB: |
| 1153 | case ELF::SHT_DYNSYM: |
| 1154 | sh_link = StringTableIndex; |
| 1155 | sh_info = LastLocalSymbolIndex; |
| 1156 | break; |
| 1157 | |
| 1158 | case ELF::SHT_SYMTAB_SHNDX: |
| 1159 | sh_link = SymbolTableIndex; |
| 1160 | break; |
| 1161 | |
Nick Lewycky | c6ac5f7 | 2011-10-07 23:28:32 +0000 | [diff] [blame] | 1162 | case ELF::SHT_GROUP: |
Rafael Espindola | a3e9a22 | 2010-11-11 18:13:52 +0000 | [diff] [blame] | 1163 | sh_link = SymbolTableIndex; |
| 1164 | sh_info = GroupSymbolIndex; |
| 1165 | break; |
Rafael Espindola | 5a8d781 | 2010-11-10 23:36:59 +0000 | [diff] [blame] | 1166 | } |
| 1167 | |
Evgeniy Stepanov | 43dcf4d | 2017-03-14 19:28:51 +0000 | [diff] [blame] | 1168 | if (Section.getFlags() & ELF::SHF_LINK_ORDER) { |
| 1169 | const MCSymbol *Sym = Section.getAssociatedSymbol(); |
| 1170 | const MCSectionELF *Sec = cast<MCSectionELF>(&Sym->getSection()); |
| 1171 | sh_link = SectionIndexMap.lookup(Sec); |
| 1172 | } |
Logan Chien | 4b72442 | 2013-02-05 14:18:59 +0000 | [diff] [blame] | 1173 | |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 1174 | WriteSecHdrEntry(StrTabBuilder.getOffset(Section.getSectionName()), |
Rafael Espindola | 2868758 | 2015-05-21 19:36:43 +0000 | [diff] [blame] | 1175 | Section.getType(), Section.getFlags(), 0, Offset, Size, |
| 1176 | sh_link, sh_info, Section.getAlignment(), |
| 1177 | Section.getEntrySize()); |
Rafael Espindola | 5a8d781 | 2010-11-10 23:36:59 +0000 | [diff] [blame] | 1178 | } |
| 1179 | |
Rafael Espindola | 7fe7e05 | 2015-02-17 20:40:59 +0000 | [diff] [blame] | 1180 | void ELFObjectWriter::writeSectionHeader( |
Rafael Espindola | 57c8083 | 2015-06-04 20:55:49 +0000 | [diff] [blame] | 1181 | const MCAsmLayout &Layout, const SectionIndexMapTy &SectionIndexMap, |
Rafael Espindola | a820169 | 2015-04-28 15:26:21 +0000 | [diff] [blame] | 1182 | const SectionOffsetsTy &SectionOffsets) { |
Rafael Espindola | b186391 | 2015-04-30 21:10:06 +0000 | [diff] [blame] | 1183 | const unsigned NumSections = SectionTable.size(); |
Rafael Espindola | a3e9a22 | 2010-11-11 18:13:52 +0000 | [diff] [blame] | 1184 | |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 1185 | // Null section first. |
Rafael Espindola | 3fe87a1 | 2010-10-31 00:16:26 +0000 | [diff] [blame] | 1186 | uint64_t FirstSectionSize = |
Rafael Espindola | bf0db6c | 2015-04-15 13:07:47 +0000 | [diff] [blame] | 1187 | (NumSections + 1) >= ELF::SHN_LORESERVE ? NumSections + 1 : 0; |
Rafael Espindola | 88d1f63 | 2015-04-29 20:25:24 +0000 | [diff] [blame] | 1188 | WriteSecHdrEntry(0, 0, 0, 0, 0, FirstSectionSize, 0, 0, 0, 0); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 1189 | |
Rafael Espindola | 08850b3 | 2015-05-25 21:56:55 +0000 | [diff] [blame] | 1190 | for (const MCSectionELF *Section : SectionTable) { |
Rafael Espindola | a3e9a22 | 2010-11-11 18:13:52 +0000 | [diff] [blame] | 1191 | uint32_t GroupSymbolIndex; |
Rafael Espindola | 03d7abb | 2015-04-30 20:53:27 +0000 | [diff] [blame] | 1192 | unsigned Type = Section->getType(); |
| 1193 | if (Type != ELF::SHT_GROUP) |
Rafael Espindola | a3e9a22 | 2010-11-11 18:13:52 +0000 | [diff] [blame] | 1194 | GroupSymbolIndex = 0; |
| 1195 | else |
Rafael Espindola | 5e9ed90 | 2015-05-28 20:53:09 +0000 | [diff] [blame] | 1196 | GroupSymbolIndex = Section->getGroup()->getIndex(); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 1197 | |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1198 | const std::pair<uint64_t, uint64_t> &Offsets = |
Rafael Espindola | 03d7abb | 2015-04-30 20:53:27 +0000 | [diff] [blame] | 1199 | SectionOffsets.find(Section)->second; |
Rafael Espindola | 1aa20fc | 2015-05-21 19:46:39 +0000 | [diff] [blame] | 1200 | uint64_t Size; |
Rafael Espindola | 5a1e80b | 2015-05-26 02:00:36 +0000 | [diff] [blame] | 1201 | if (Type == ELF::SHT_NOBITS) |
| 1202 | Size = Layout.getSectionAddressSize(Section); |
| 1203 | else |
Rafael Espindola | 1aa20fc | 2015-05-21 19:46:39 +0000 | [diff] [blame] | 1204 | Size = Offsets.second - Offsets.first; |
Rafael Espindola | 60ebca9 | 2010-12-02 03:09:06 +0000 | [diff] [blame] | 1205 | |
Rafael Espindola | e92c1bf | 2015-05-21 19:42:35 +0000 | [diff] [blame] | 1206 | writeSection(SectionIndexMap, GroupSymbolIndex, Offsets.first, Size, |
Rafael Espindola | 2868758 | 2015-05-21 19:36:43 +0000 | [diff] [blame] | 1207 | *Section); |
Matt Fleming | 6c1ad48 | 2010-08-16 18:57:57 +0000 | [diff] [blame] | 1208 | } |
| 1209 | } |
| 1210 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 1211 | void ELFObjectWriter::writeObject(MCAssembler &Asm, |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 1212 | const MCAsmLayout &Layout) { |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1213 | MCContext &Ctx = Asm.getContext(); |
Rafael Espindola | 5960cee | 2015-05-22 23:58:30 +0000 | [diff] [blame] | 1214 | MCSectionELF *StrtabSection = |
| 1215 | Ctx.getELFSection(".strtab", ELF::SHT_STRTAB, 0); |
| 1216 | StringTableIndex = addToSectionTable(StrtabSection); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1217 | |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 1218 | RevGroupMapTy RevGroupMap; |
| 1219 | SectionIndexMapTy SectionIndexMap; |
| 1220 | |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1221 | std::map<const MCSymbol *, std::vector<const MCSectionELF *>> GroupMembers; |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 1222 | |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 1223 | // Write out the ELF header ... |
Rafael Espindola | 8c7829b | 2015-04-29 20:39:37 +0000 | [diff] [blame] | 1224 | writeHeader(Asm); |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 1225 | |
Rafael Espindola | 7230f80 | 2015-04-08 11:41:24 +0000 | [diff] [blame] | 1226 | // ... then the sections ... |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1227 | SectionOffsetsTy SectionOffsets; |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1228 | std::vector<MCSectionELF *> Groups; |
| 1229 | std::vector<MCSectionELF *> Relocations; |
Rafael Espindola | e15b1b7 | 2015-05-27 13:30:50 +0000 | [diff] [blame] | 1230 | for (MCSection &Sec : Asm) { |
| 1231 | MCSectionELF &Section = static_cast<MCSectionELF &>(Sec); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1232 | |
Rafael Espindola | b20fbb8 | 2015-06-05 18:21:00 +0000 | [diff] [blame] | 1233 | align(Section.getAlignment()); |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 1234 | |
Rafael Espindola | 642a221 | 2015-04-14 22:54:16 +0000 | [diff] [blame] | 1235 | // Remember the offset into the file for this section. |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1236 | uint64_t SecStart = getStream().tell(); |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1237 | |
Rafael Espindola | 0ccf9b7 | 2015-06-02 21:30:13 +0000 | [diff] [blame] | 1238 | const MCSymbolELF *SignatureSymbol = Section.getGroup(); |
Rafael Espindola | e15b1b7 | 2015-05-27 13:30:50 +0000 | [diff] [blame] | 1239 | writeSectionData(Asm, Section, Layout); |
Rafael Espindola | b8cbb26 | 2015-04-30 00:30:40 +0000 | [diff] [blame] | 1240 | |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1241 | uint64_t SecEnd = getStream().tell(); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1242 | SectionOffsets[&Section] = std::make_pair(SecStart, SecEnd); |
| 1243 | |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1244 | MCSectionELF *RelSection = createRelocationSection(Ctx, Section); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1245 | |
| 1246 | if (SignatureSymbol) { |
Rafael Espindola | b5d316b | 2015-05-29 20:21:02 +0000 | [diff] [blame] | 1247 | Asm.registerSymbol(*SignatureSymbol); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1248 | unsigned &GroupIdx = RevGroupMap[SignatureSymbol]; |
| 1249 | if (!GroupIdx) { |
Rafael Espindola | 0709a7b | 2015-05-21 19:20:38 +0000 | [diff] [blame] | 1250 | MCSectionELF *Group = Ctx.createELFGroupSection(SignatureSymbol); |
Rafael Espindola | 03d7abb | 2015-04-30 20:53:27 +0000 | [diff] [blame] | 1251 | GroupIdx = addToSectionTable(Group); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1252 | Group->setAlignment(4); |
| 1253 | Groups.push_back(Group); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1254 | } |
Rafael Espindola | 7830fc8 | 2015-06-05 17:54:25 +0000 | [diff] [blame] | 1255 | std::vector<const MCSectionELF *> &Members = |
| 1256 | GroupMembers[SignatureSymbol]; |
| 1257 | Members.push_back(&Section); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1258 | if (RelSection) |
Rafael Espindola | 7830fc8 | 2015-06-05 17:54:25 +0000 | [diff] [blame] | 1259 | Members.push_back(RelSection); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1260 | } |
| 1261 | |
Rafael Espindola | 03d7abb | 2015-04-30 20:53:27 +0000 | [diff] [blame] | 1262 | SectionIndexMap[&Section] = addToSectionTable(&Section); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1263 | if (RelSection) { |
Rafael Espindola | 03d7abb | 2015-04-30 20:53:27 +0000 | [diff] [blame] | 1264 | SectionIndexMap[RelSection] = addToSectionTable(RelSection); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1265 | Relocations.push_back(RelSection); |
| 1266 | } |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1267 | } |
| 1268 | |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1269 | for (MCSectionELF *Group : Groups) { |
Rafael Espindola | b20fbb8 | 2015-06-05 18:21:00 +0000 | [diff] [blame] | 1270 | align(Group->getAlignment()); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1271 | |
| 1272 | // Remember the offset into the file for this section. |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1273 | uint64_t SecStart = getStream().tell(); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1274 | |
| 1275 | const MCSymbol *SignatureSymbol = Group->getGroup(); |
| 1276 | assert(SignatureSymbol); |
| 1277 | write(uint32_t(ELF::GRP_COMDAT)); |
| 1278 | for (const MCSectionELF *Member : GroupMembers[SignatureSymbol]) { |
| 1279 | uint32_t SecIndex = SectionIndexMap.lookup(Member); |
| 1280 | write(SecIndex); |
| 1281 | } |
| 1282 | |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1283 | uint64_t SecEnd = getStream().tell(); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1284 | SectionOffsets[Group] = std::make_pair(SecStart, SecEnd); |
| 1285 | } |
| 1286 | |
| 1287 | // Compute symbol table information. |
Rafael Espindola | aa486e9 | 2015-05-28 17:54:01 +0000 | [diff] [blame] | 1288 | computeSymbolTable(Asm, Layout, SectionIndexMap, RevGroupMap, SectionOffsets); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1289 | |
| 1290 | for (MCSectionELF *RelSection : Relocations) { |
Rafael Espindola | b20fbb8 | 2015-06-05 18:21:00 +0000 | [diff] [blame] | 1291 | align(RelSection->getAlignment()); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1292 | |
| 1293 | // Remember the offset into the file for this section. |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1294 | uint64_t SecStart = getStream().tell(); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1295 | |
Evgeniy Stepanov | 43dcf4d | 2017-03-14 19:28:51 +0000 | [diff] [blame] | 1296 | writeRelocations(Asm, |
| 1297 | cast<MCSectionELF>(*RelSection->getAssociatedSection())); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1298 | |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1299 | uint64_t SecEnd = getStream().tell(); |
Rafael Espindola | 0a82ad7 | 2015-05-21 20:43:13 +0000 | [diff] [blame] | 1300 | SectionOffsets[RelSection] = std::make_pair(SecStart, SecEnd); |
Rafael Espindola | 642a221 | 2015-04-14 22:54:16 +0000 | [diff] [blame] | 1301 | } |
| 1302 | |
Rafael Espindola | 88d1f63 | 2015-04-29 20:25:24 +0000 | [diff] [blame] | 1303 | { |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1304 | uint64_t SecStart = getStream().tell(); |
Rafael Espindola | b186391 | 2015-04-30 21:10:06 +0000 | [diff] [blame] | 1305 | const MCSectionELF *Sec = createStringTable(Ctx); |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1306 | uint64_t SecEnd = getStream().tell(); |
Rafael Espindola | bda1980 | 2015-04-30 14:21:49 +0000 | [diff] [blame] | 1307 | SectionOffsets[Sec] = std::make_pair(SecStart, SecEnd); |
Rafael Espindola | 88abc39 | 2015-04-29 20:34:31 +0000 | [diff] [blame] | 1308 | } |
| 1309 | |
Rafael Espindola | 642a221 | 2015-04-14 22:54:16 +0000 | [diff] [blame] | 1310 | uint64_t NaturalAlignment = is64Bit() ? 8 : 4; |
Rafael Espindola | b20fbb8 | 2015-06-05 18:21:00 +0000 | [diff] [blame] | 1311 | align(NaturalAlignment); |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 1312 | |
Rafael Espindola | 5568c83 | 2016-01-19 15:19:08 +0000 | [diff] [blame] | 1313 | const uint64_t SectionHeaderOffset = getStream().tell(); |
Rafael Espindola | 642a221 | 2015-04-14 22:54:16 +0000 | [diff] [blame] | 1314 | |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 1315 | // ... then the section header table ... |
Rafael Espindola | 57c8083 | 2015-06-04 20:55:49 +0000 | [diff] [blame] | 1316 | writeSectionHeader(Layout, SectionIndexMap, SectionOffsets); |
Rafael Espindola | 642a221 | 2015-04-14 22:54:16 +0000 | [diff] [blame] | 1317 | |
Rafael Espindola | 03d7abb | 2015-04-30 20:53:27 +0000 | [diff] [blame] | 1318 | uint16_t NumSections = (SectionTable.size() + 1 >= ELF::SHN_LORESERVE) |
Aaron Ballman | 9cab732 | 2015-04-30 14:03:12 +0000 | [diff] [blame] | 1319 | ? (uint16_t)ELF::SHN_UNDEF |
Rafael Espindola | 03d7abb | 2015-04-30 20:53:27 +0000 | [diff] [blame] | 1320 | : SectionTable.size() + 1; |
Rafael Espindola | 8c7829b | 2015-04-29 20:39:37 +0000 | [diff] [blame] | 1321 | if (sys::IsLittleEndianHost != IsLittleEndian) |
| 1322 | sys::swapByteOrder(NumSections); |
| 1323 | unsigned NumSectionsOffset; |
| 1324 | |
Rafael Espindola | 642a221 | 2015-04-14 22:54:16 +0000 | [diff] [blame] | 1325 | if (is64Bit()) { |
| 1326 | uint64_t Val = SectionHeaderOffset; |
| 1327 | if (sys::IsLittleEndianHost != IsLittleEndian) |
| 1328 | sys::swapByteOrder(Val); |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1329 | getStream().pwrite(reinterpret_cast<char *>(&Val), sizeof(Val), |
| 1330 | offsetof(ELF::Elf64_Ehdr, e_shoff)); |
Rafael Espindola | 8c7829b | 2015-04-29 20:39:37 +0000 | [diff] [blame] | 1331 | NumSectionsOffset = offsetof(ELF::Elf64_Ehdr, e_shnum); |
Rafael Espindola | 642a221 | 2015-04-14 22:54:16 +0000 | [diff] [blame] | 1332 | } else { |
| 1333 | uint32_t Val = SectionHeaderOffset; |
| 1334 | if (sys::IsLittleEndianHost != IsLittleEndian) |
| 1335 | sys::swapByteOrder(Val); |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1336 | getStream().pwrite(reinterpret_cast<char *>(&Val), sizeof(Val), |
| 1337 | offsetof(ELF::Elf32_Ehdr, e_shoff)); |
Rafael Espindola | 8c7829b | 2015-04-29 20:39:37 +0000 | [diff] [blame] | 1338 | NumSectionsOffset = offsetof(ELF::Elf32_Ehdr, e_shnum); |
Rafael Espindola | 642a221 | 2015-04-14 22:54:16 +0000 | [diff] [blame] | 1339 | } |
David Majnemer | abdb2d2a | 2015-09-01 16:19:03 +0000 | [diff] [blame] | 1340 | getStream().pwrite(reinterpret_cast<char *>(&NumSections), |
| 1341 | sizeof(NumSections), NumSectionsOffset); |
Rafael Espindola | 1557fd6 | 2011-03-20 18:44:20 +0000 | [diff] [blame] | 1342 | } |
| 1343 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 1344 | bool ELFObjectWriter::isSymbolRefDifferenceFullyResolvedImpl( |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 1345 | const MCAssembler &Asm, const MCSymbol &SA, const MCFragment &FB, |
Rafael Espindola | 35d6189 | 2015-04-17 21:15:17 +0000 | [diff] [blame] | 1346 | bool InSet, bool IsPCRel) const { |
Rafael Espindola | 95fb9b9 | 2015-06-02 20:38:46 +0000 | [diff] [blame] | 1347 | const auto &SymA = cast<MCSymbolELF>(SA); |
Rafael Espindola | 35d6189 | 2015-04-17 21:15:17 +0000 | [diff] [blame] | 1348 | if (IsPCRel) { |
| 1349 | assert(!InSet); |
Peter Collingbourne | 3600305 | 2017-04-08 23:35:49 +0000 | [diff] [blame] | 1350 | if (isWeak(SymA)) |
Rafael Espindola | 35d6189 | 2015-04-17 21:15:17 +0000 | [diff] [blame] | 1351 | return false; |
| 1352 | } |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame] | 1353 | return MCObjectWriter::isSymbolRefDifferenceFullyResolvedImpl(Asm, SymA, FB, |
Rafael Espindola | 35d6189 | 2015-04-17 21:15:17 +0000 | [diff] [blame] | 1354 | InSet, IsPCRel); |
Eli Friedman | d92d17b | 2011-03-03 07:24:36 +0000 | [diff] [blame] | 1355 | } |
| 1356 | |
Lang Hames | 60fbc7c | 2017-10-10 16:28:07 +0000 | [diff] [blame] | 1357 | std::unique_ptr<MCObjectWriter> |
Lang Hames | dcb312b | 2017-10-09 23:53:15 +0000 | [diff] [blame] | 1358 | llvm::createELFObjectWriter(std::unique_ptr<MCELFObjectTargetWriter> MOTW, |
| 1359 | raw_pwrite_stream &OS, bool IsLittleEndian) { |
Lang Hames | 60fbc7c | 2017-10-10 16:28:07 +0000 | [diff] [blame] | 1360 | return llvm::make_unique<ELFObjectWriter>(std::move(MOTW), OS, |
| 1361 | IsLittleEndian); |
Rafael Espindola | b264d33 | 2011-12-21 17:30:17 +0000 | [diff] [blame] | 1362 | } |