Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 1 | //===-- llvm/MC/WinCOFFObjectWriter.cpp -------------------------*- C++ -*-===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file contains an implementation of a Win32 COFF object file writer. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
Rafael Espindola | 908d2ed | 2011-12-24 02:14:02 +0000 | [diff] [blame] | 14 | #include "llvm/MC/MCWinCOFFObjectWriter.h" |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 15 | #include "llvm/ADT/DenseMap.h" |
Chandler Carruth | d990388 | 2015-01-14 11:23:27 +0000 | [diff] [blame] | 16 | #include "llvm/ADT/STLExtras.h" |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 17 | #include "llvm/ADT/StringMap.h" |
| 18 | #include "llvm/ADT/StringRef.h" |
Nico Rieck | a37acf7 | 2013-07-06 12:13:10 +0000 | [diff] [blame] | 19 | #include "llvm/ADT/Twine.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 20 | #include "llvm/MC/MCAsmLayout.h" |
| 21 | #include "llvm/MC/MCAssembler.h" |
| 22 | #include "llvm/MC/MCContext.h" |
| 23 | #include "llvm/MC/MCExpr.h" |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 24 | #include "llvm/MC/MCObjectFileInfo.h" |
Chandler Carruth | ed0881b | 2012-12-03 16:50:05 +0000 | [diff] [blame] | 25 | #include "llvm/MC/MCObjectWriter.h" |
| 26 | #include "llvm/MC/MCSection.h" |
| 27 | #include "llvm/MC/MCSectionCOFF.h" |
| 28 | #include "llvm/MC/MCSymbol.h" |
| 29 | #include "llvm/MC/MCValue.h" |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 30 | #include "llvm/MC/StringTableBuilder.h" |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 31 | #include "llvm/Support/COFF.h" |
| 32 | #include "llvm/Support/Debug.h" |
Hans Wennborg | ba80b5d | 2014-09-28 00:22:27 +0000 | [diff] [blame] | 33 | #include "llvm/Support/Endian.h" |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 34 | #include "llvm/Support/ErrorHandling.h" |
Michael J. Spencer | 447762d | 2010-11-29 18:16:10 +0000 | [diff] [blame] | 35 | #include "llvm/Support/TimeValue.h" |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 36 | #include <cstdio> |
| 37 | |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 38 | using namespace llvm; |
| 39 | |
Chandler Carruth | f58e376 | 2014-04-22 03:04:17 +0000 | [diff] [blame] | 40 | #define DEBUG_TYPE "WinCOFFObjectWriter" |
| 41 | |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 42 | namespace { |
Dmitri Gribenko | 226fea5 | 2013-01-13 16:01:15 +0000 | [diff] [blame] | 43 | typedef SmallString<COFF::NameSize> name; |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 44 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 45 | enum AuxiliaryType { |
| 46 | ATFunctionDefinition, |
| 47 | ATbfAndefSymbol, |
| 48 | ATWeakExternal, |
| 49 | ATFile, |
| 50 | ATSectionDefinition |
| 51 | }; |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 52 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 53 | struct AuxSymbol { |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 54 | AuxiliaryType AuxType; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 55 | COFF::Auxiliary Aux; |
| 56 | }; |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 57 | |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 58 | class COFFSymbol; |
| 59 | class COFFSection; |
| 60 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 61 | class COFFSymbol { |
| 62 | public: |
| 63 | COFF::symbol Data; |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 64 | |
Dmitri Gribenko | 226fea5 | 2013-01-13 16:01:15 +0000 | [diff] [blame] | 65 | typedef SmallVector<AuxSymbol, 1> AuxiliarySymbols; |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 66 | |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 67 | name Name; |
| 68 | int Index; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 69 | AuxiliarySymbols Aux; |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 70 | COFFSymbol *Other; |
| 71 | COFFSection *Section; |
| 72 | int Relocations; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 73 | |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 74 | const MCSymbol *MC; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 75 | |
Dmitri Gribenko | 226fea5 | 2013-01-13 16:01:15 +0000 | [diff] [blame] | 76 | COFFSymbol(StringRef name); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 77 | void set_name_offset(uint32_t Offset); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 78 | |
| 79 | bool should_keep() const; |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 80 | |
| 81 | int64_t getIndex() const { return Index; } |
| 82 | void setIndex(int Value) { |
| 83 | Index = Value; |
| 84 | if (MC) |
| 85 | MC->setIndex(static_cast<uint32_t>(Value)); |
| 86 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 87 | }; |
| 88 | |
| 89 | // This class contains staging data for a COFF relocation entry. |
| 90 | struct COFFRelocation { |
| 91 | COFF::relocation Data; |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 92 | COFFSymbol *Symb; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 93 | |
Craig Topper | bb694de | 2014-04-13 04:57:38 +0000 | [diff] [blame] | 94 | COFFRelocation() : Symb(nullptr) {} |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 95 | static size_t size() { return COFF::RelocationSize; } |
| 96 | }; |
| 97 | |
| 98 | typedef std::vector<COFFRelocation> relocations; |
| 99 | |
| 100 | class COFFSection { |
| 101 | public: |
| 102 | COFF::section Header; |
| 103 | |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 104 | std::string Name; |
| 105 | int Number; |
Rafael Espindola | f59264f | 2015-05-27 14:45:54 +0000 | [diff] [blame] | 106 | MCSectionCOFF const *MCSection; |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 107 | COFFSymbol *Symbol; |
| 108 | relocations Relocations; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 109 | |
Dmitri Gribenko | 226fea5 | 2013-01-13 16:01:15 +0000 | [diff] [blame] | 110 | COFFSection(StringRef name); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 111 | static size_t size(); |
| 112 | }; |
| 113 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 114 | class WinCOFFObjectWriter : public MCObjectWriter { |
| 115 | public: |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 116 | typedef std::vector<std::unique_ptr<COFFSymbol>> symbols; |
David Blaikie | f564ab6 | 2014-04-15 05:25:03 +0000 | [diff] [blame] | 117 | typedef std::vector<std::unique_ptr<COFFSection>> sections; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 118 | |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 119 | typedef DenseMap<MCSymbol const *, COFFSymbol *> symbol_map; |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 120 | typedef DenseMap<MCSection const *, COFFSection *> section_map; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 121 | |
Ahmed Charles | 56440fd | 2014-03-06 05:51:42 +0000 | [diff] [blame] | 122 | std::unique_ptr<MCWinCOFFObjectTargetWriter> TargetObjectWriter; |
Rafael Espindola | 908d2ed | 2011-12-24 02:14:02 +0000 | [diff] [blame] | 123 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 124 | // Root level file contents. |
| 125 | COFF::header Header; |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 126 | sections Sections; |
| 127 | symbols Symbols; |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 128 | StringTableBuilder Strings; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 129 | |
| 130 | // Maps used during object file creation. |
| 131 | section_map SectionMap; |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 132 | symbol_map SymbolMap; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 133 | |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 134 | bool UseBigObj; |
| 135 | |
Rafael Espindola | 5560a4c | 2015-04-14 22:14:34 +0000 | [diff] [blame] | 136 | WinCOFFObjectWriter(MCWinCOFFObjectTargetWriter *MOTW, raw_pwrite_stream &OS); |
Rafael Espindola | 37099d9 | 2015-04-09 18:08:15 +0000 | [diff] [blame] | 137 | |
Yaron Keren | cca43c1 | 2014-09-16 21:31:04 +0000 | [diff] [blame] | 138 | void reset() override { |
| 139 | memset(&Header, 0, sizeof(Header)); |
| 140 | Header.Machine = TargetObjectWriter->getMachine(); |
| 141 | Sections.clear(); |
| 142 | Symbols.clear(); |
| 143 | Strings.clear(); |
| 144 | SectionMap.clear(); |
| 145 | SymbolMap.clear(); |
| 146 | MCObjectWriter::reset(); |
| 147 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 148 | |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 149 | COFFSymbol *createSymbol(StringRef Name); |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 150 | COFFSymbol *GetOrCreateCOFFSymbol(const MCSymbol *Symbol); |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 151 | COFFSection *createSection(StringRef Name); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 152 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 153 | template <typename object_t, typename list_t> |
Dmitri Gribenko | 226fea5 | 2013-01-13 16:01:15 +0000 | [diff] [blame] | 154 | object_t *createCOFFEntity(StringRef Name, list_t &List); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 155 | |
Rafael Espindola | f59264f | 2015-05-27 14:45:54 +0000 | [diff] [blame] | 156 | void defineSection(MCSectionCOFF const &Sec); |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 157 | void DefineSymbol(const MCSymbol &Symbol, MCAssembler &Assembler, |
Reid Kleckner | c1e7621 | 2013-09-17 23:18:05 +0000 | [diff] [blame] | 158 | const MCAsmLayout &Layout); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 159 | |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 160 | void SetSymbolName(COFFSymbol &S); |
| 161 | void SetSectionName(COFFSection &S); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 162 | |
David Majnemer | 299674e | 2014-07-13 04:31:19 +0000 | [diff] [blame] | 163 | bool ExportSymbol(const MCSymbol &Symbol, MCAssembler &Asm); |
Reid Kleckner | 8b2ad2a | 2013-11-18 23:08:12 +0000 | [diff] [blame] | 164 | |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 165 | bool IsPhysicalSection(COFFSection *S); |
| 166 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 167 | // Entity writing methods. |
| 168 | |
| 169 | void WriteFileHeader(const COFF::header &Header); |
David Blaikie | f564ab6 | 2014-04-15 05:25:03 +0000 | [diff] [blame] | 170 | void WriteSymbol(const COFFSymbol &S); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 171 | void WriteAuxiliarySymbols(const COFFSymbol::AuxiliarySymbols &S); |
| 172 | void WriteSectionHeader(const COFF::section &S); |
| 173 | void WriteRelocation(const COFF::relocation &R); |
| 174 | |
| 175 | // MCObjectWriter interface implementation. |
| 176 | |
Craig Topper | 34a61bc | 2014-03-08 07:02:02 +0000 | [diff] [blame] | 177 | void ExecutePostLayoutBinding(MCAssembler &Asm, |
| 178 | const MCAsmLayout &Layout) override; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 179 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 180 | bool isSymbolRefDifferenceFullyResolvedImpl(const MCAssembler &Asm, |
Duncan P. N. Exon Smith | d81ba53 | 2015-05-16 01:01:55 +0000 | [diff] [blame] | 181 | const MCSymbol &SymA, |
David Majnemer | 2cc4bc77 | 2014-11-11 08:43:57 +0000 | [diff] [blame] | 182 | const MCFragment &FB, bool InSet, |
| 183 | bool IsPCRel) const override; |
| 184 | |
Duncan P. N. Exon Smith | 5266ad9 | 2015-05-20 15:10:03 +0000 | [diff] [blame] | 185 | bool isWeak(const MCSymbol &Sym) const override; |
Rafael Espindola | aeed3cb | 2015-03-26 21:11:00 +0000 | [diff] [blame] | 186 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 187 | void recordRelocation(MCAssembler &Asm, const MCAsmLayout &Layout, |
Craig Topper | 34a61bc | 2014-03-08 07:02:02 +0000 | [diff] [blame] | 188 | const MCFragment *Fragment, const MCFixup &Fixup, |
Rafael Espindola | 5904e12 | 2014-03-29 06:26:49 +0000 | [diff] [blame] | 189 | MCValue Target, bool &IsPCRel, |
| 190 | uint64_t &FixedValue) override; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 191 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 192 | void writeObject(MCAssembler &Asm, const MCAsmLayout &Layout) override; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 193 | }; |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 194 | } |
| 195 | |
Hans Wennborg | ba80b5d | 2014-09-28 00:22:27 +0000 | [diff] [blame] | 196 | static inline void write_uint32_le(void *Data, uint32_t Value) { |
| 197 | support::endian::write<uint32_t, support::little, support::unaligned>(Data, |
| 198 | Value); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 199 | } |
| 200 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 201 | //------------------------------------------------------------------------------ |
| 202 | // Symbol class implementation |
| 203 | |
Dmitri Gribenko | 226fea5 | 2013-01-13 16:01:15 +0000 | [diff] [blame] | 204 | COFFSymbol::COFFSymbol(StringRef name) |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 205 | : Name(name.begin(), name.end()), Other(nullptr), Section(nullptr), |
| 206 | Relocations(0), MC(nullptr) { |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 207 | memset(&Data, 0, sizeof(Data)); |
| 208 | } |
| 209 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 210 | // In the case that the name does not fit within 8 bytes, the offset |
| 211 | // into the string table is stored in the last 4 bytes instead, leaving |
| 212 | // the first 4 bytes as 0. |
| 213 | void COFFSymbol::set_name_offset(uint32_t Offset) { |
| 214 | write_uint32_le(Data.Name + 0, 0); |
| 215 | write_uint32_le(Data.Name + 4, Offset); |
| 216 | } |
| 217 | |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 218 | /// logic to decide if the symbol should be reported in the symbol table |
| 219 | bool COFFSymbol::should_keep() const { |
| 220 | // no section means its external, keep it |
Craig Topper | bb694de | 2014-04-13 04:57:38 +0000 | [diff] [blame] | 221 | if (!Section) |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 222 | return true; |
| 223 | |
| 224 | // if it has relocations pointing at it, keep it |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 225 | if (Relocations > 0) { |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 226 | assert(Section->Number != -1 && "Sections with relocations must be real!"); |
| 227 | return true; |
| 228 | } |
| 229 | |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 230 | // if this is a safeseh handler, keep it |
| 231 | if (MC && (MC->getFlags() & COFF::SF_SafeSEH)) |
| 232 | return true; |
| 233 | |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 234 | // if the section its in is being droped, drop it |
| 235 | if (Section->Number == -1) |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 236 | return false; |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 237 | |
| 238 | // if it is the section symbol, keep it |
| 239 | if (Section->Symbol == this) |
| 240 | return true; |
| 241 | |
| 242 | // if its temporary, drop it |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 243 | if (MC && MC->isTemporary()) |
| 244 | return false; |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 245 | |
| 246 | // otherwise, keep it |
| 247 | return true; |
| 248 | } |
| 249 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 250 | //------------------------------------------------------------------------------ |
| 251 | // Section class implementation |
| 252 | |
Dmitri Gribenko | 226fea5 | 2013-01-13 16:01:15 +0000 | [diff] [blame] | 253 | COFFSection::COFFSection(StringRef name) |
Rafael Espindola | f59264f | 2015-05-27 14:45:54 +0000 | [diff] [blame] | 254 | : Name(name), MCSection(nullptr), Symbol(nullptr) { |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 255 | memset(&Header, 0, sizeof(Header)); |
| 256 | } |
| 257 | |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 258 | size_t COFFSection::size() { return COFF::SectionSize; } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 259 | |
| 260 | //------------------------------------------------------------------------------ |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 261 | // WinCOFFObjectWriter class implementation |
| 262 | |
Rafael Espindola | 908d2ed | 2011-12-24 02:14:02 +0000 | [diff] [blame] | 263 | WinCOFFObjectWriter::WinCOFFObjectWriter(MCWinCOFFObjectTargetWriter *MOTW, |
Rafael Espindola | 5560a4c | 2015-04-14 22:14:34 +0000 | [diff] [blame] | 264 | raw_pwrite_stream &OS) |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 265 | : MCObjectWriter(OS, true), TargetObjectWriter(MOTW) { |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 266 | memset(&Header, 0, sizeof(Header)); |
Michael J. Spencer | 377aa20 | 2010-08-21 05:58:13 +0000 | [diff] [blame] | 267 | |
Rafael Espindola | 908d2ed | 2011-12-24 02:14:02 +0000 | [diff] [blame] | 268 | Header.Machine = TargetObjectWriter->getMachine(); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 269 | } |
| 270 | |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 271 | COFFSymbol *WinCOFFObjectWriter::createSymbol(StringRef Name) { |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 272 | return createCOFFEntity<COFFSymbol>(Name, Symbols); |
| 273 | } |
| 274 | |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 275 | COFFSymbol *WinCOFFObjectWriter::GetOrCreateCOFFSymbol(const MCSymbol *Symbol) { |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 276 | symbol_map::iterator i = SymbolMap.find(Symbol); |
| 277 | if (i != SymbolMap.end()) |
| 278 | return i->second; |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 279 | COFFSymbol *RetSymbol = |
| 280 | createCOFFEntity<COFFSymbol>(Symbol->getName(), Symbols); |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 281 | SymbolMap[Symbol] = RetSymbol; |
| 282 | return RetSymbol; |
| 283 | } |
| 284 | |
Dmitri Gribenko | 226fea5 | 2013-01-13 16:01:15 +0000 | [diff] [blame] | 285 | COFFSection *WinCOFFObjectWriter::createSection(StringRef Name) { |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 286 | return createCOFFEntity<COFFSection>(Name, Sections); |
| 287 | } |
| 288 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 289 | /// A template used to lookup or create a symbol/section, and initialize it if |
| 290 | /// needed. |
| 291 | template <typename object_t, typename list_t> |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 292 | object_t *WinCOFFObjectWriter::createCOFFEntity(StringRef Name, list_t &List) { |
David Blaikie | f564ab6 | 2014-04-15 05:25:03 +0000 | [diff] [blame] | 293 | List.push_back(make_unique<object_t>(Name)); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 294 | |
David Blaikie | f564ab6 | 2014-04-15 05:25:03 +0000 | [diff] [blame] | 295 | return List.back().get(); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 296 | } |
| 297 | |
| 298 | /// This function takes a section data object from the assembler |
| 299 | /// and creates the associated COFF section staging object. |
Rafael Espindola | f59264f | 2015-05-27 14:45:54 +0000 | [diff] [blame] | 300 | void WinCOFFObjectWriter::defineSection(MCSectionCOFF const &Sec) { |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 301 | COFFSection *coff_section = createSection(Sec.getSectionName()); |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 302 | COFFSymbol *coff_symbol = createSymbol(Sec.getSectionName()); |
Rafael Espindola | 0766ae0 | 2014-06-06 19:26:12 +0000 | [diff] [blame] | 303 | if (Sec.getSelection() != COFF::IMAGE_COMDAT_SELECT_ASSOCIATIVE) { |
| 304 | if (const MCSymbol *S = Sec.getCOMDATSymbol()) { |
| 305 | COFFSymbol *COMDATSymbol = GetOrCreateCOFFSymbol(S); |
| 306 | if (COMDATSymbol->Section) |
| 307 | report_fatal_error("two sections have the same comdat"); |
| 308 | COMDATSymbol->Section = coff_section; |
| 309 | } |
| 310 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 311 | |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 312 | coff_section->Symbol = coff_symbol; |
| 313 | coff_symbol->Section = coff_section; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 314 | coff_symbol->Data.StorageClass = COFF::IMAGE_SYM_CLASS_STATIC; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 315 | |
| 316 | // In this case the auxiliary symbol is a Section Definition. |
| 317 | coff_symbol->Aux.resize(1); |
| 318 | memset(&coff_symbol->Aux[0], 0, sizeof(coff_symbol->Aux[0])); |
| 319 | coff_symbol->Aux[0].AuxType = ATSectionDefinition; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 320 | coff_symbol->Aux[0].Aux.SectionDefinition.Selection = Sec.getSelection(); |
| 321 | |
| 322 | coff_section->Header.Characteristics = Sec.getCharacteristics(); |
| 323 | |
| 324 | uint32_t &Characteristics = coff_section->Header.Characteristics; |
Rafael Espindola | 967d6a6 | 2015-05-21 21:02:35 +0000 | [diff] [blame] | 325 | switch (Sec.getAlignment()) { |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 326 | case 1: |
| 327 | Characteristics |= COFF::IMAGE_SCN_ALIGN_1BYTES; |
| 328 | break; |
| 329 | case 2: |
| 330 | Characteristics |= COFF::IMAGE_SCN_ALIGN_2BYTES; |
| 331 | break; |
| 332 | case 4: |
| 333 | Characteristics |= COFF::IMAGE_SCN_ALIGN_4BYTES; |
| 334 | break; |
| 335 | case 8: |
| 336 | Characteristics |= COFF::IMAGE_SCN_ALIGN_8BYTES; |
| 337 | break; |
| 338 | case 16: |
| 339 | Characteristics |= COFF::IMAGE_SCN_ALIGN_16BYTES; |
| 340 | break; |
| 341 | case 32: |
| 342 | Characteristics |= COFF::IMAGE_SCN_ALIGN_32BYTES; |
| 343 | break; |
| 344 | case 64: |
| 345 | Characteristics |= COFF::IMAGE_SCN_ALIGN_64BYTES; |
| 346 | break; |
| 347 | case 128: |
| 348 | Characteristics |= COFF::IMAGE_SCN_ALIGN_128BYTES; |
| 349 | break; |
| 350 | case 256: |
| 351 | Characteristics |= COFF::IMAGE_SCN_ALIGN_256BYTES; |
| 352 | break; |
| 353 | case 512: |
| 354 | Characteristics |= COFF::IMAGE_SCN_ALIGN_512BYTES; |
| 355 | break; |
| 356 | case 1024: |
| 357 | Characteristics |= COFF::IMAGE_SCN_ALIGN_1024BYTES; |
| 358 | break; |
| 359 | case 2048: |
| 360 | Characteristics |= COFF::IMAGE_SCN_ALIGN_2048BYTES; |
| 361 | break; |
| 362 | case 4096: |
| 363 | Characteristics |= COFF::IMAGE_SCN_ALIGN_4096BYTES; |
| 364 | break; |
| 365 | case 8192: |
| 366 | Characteristics |= COFF::IMAGE_SCN_ALIGN_8192BYTES; |
| 367 | break; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 368 | default: |
| 369 | llvm_unreachable("unsupported section alignment"); |
| 370 | } |
| 371 | |
| 372 | // Bind internal COFF section to MC section. |
Rafael Espindola | f59264f | 2015-05-27 14:45:54 +0000 | [diff] [blame] | 373 | coff_section->MCSection = &Sec; |
| 374 | SectionMap[&Sec] = coff_section; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 375 | } |
| 376 | |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 377 | static uint64_t getSymbolValue(const MCSymbol &Symbol, |
Rafael Espindola | ff68cb7 | 2014-05-01 00:10:17 +0000 | [diff] [blame] | 378 | const MCAsmLayout &Layout) { |
Rafael Espindola | 4d37b2a | 2015-05-29 21:45:01 +0000 | [diff] [blame] | 379 | if (Symbol.isCommon() && Symbol.isExternal()) |
Rafael Espindola | 1467250 | 2015-05-29 17:48:04 +0000 | [diff] [blame] | 380 | return Symbol.getCommonSize(); |
Rafael Espindola | ff68cb7 | 2014-05-01 00:10:17 +0000 | [diff] [blame] | 381 | |
| 382 | uint64_t Res; |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 383 | if (!Layout.getSymbolOffset(Symbol, Res)) |
Rafael Espindola | ff68cb7 | 2014-05-01 00:10:17 +0000 | [diff] [blame] | 384 | return 0; |
| 385 | |
| 386 | return Res; |
| 387 | } |
| 388 | |
David Majnemer | a9bdb32 | 2014-04-08 22:33:40 +0000 | [diff] [blame] | 389 | /// This function takes a symbol data object from the assembler |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 390 | /// and creates the associated COFF symbol staging object. |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 391 | void WinCOFFObjectWriter::DefineSymbol(const MCSymbol &Symbol, |
Reid Kleckner | c1e7621 | 2013-09-17 23:18:05 +0000 | [diff] [blame] | 392 | MCAssembler &Assembler, |
| 393 | const MCAsmLayout &Layout) { |
Michael J. Spencer | 54b24e1 | 2013-01-29 22:10:07 +0000 | [diff] [blame] | 394 | COFFSymbol *coff_symbol = GetOrCreateCOFFSymbol(&Symbol); |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 395 | SymbolMap[&Symbol] = coff_symbol; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 396 | |
Rafael Espindola | 2229d33 | 2015-05-29 19:07:51 +0000 | [diff] [blame] | 397 | if (Symbol.getFlags() & COFF::SF_WeakExternal) { |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 398 | coff_symbol->Data.StorageClass = COFF::IMAGE_SYM_CLASS_WEAK_EXTERNAL; |
| 399 | |
Michael J. Spencer | 54b24e1 | 2013-01-29 22:10:07 +0000 | [diff] [blame] | 400 | if (Symbol.isVariable()) { |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 401 | const MCSymbolRefExpr *SymRef = |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 402 | dyn_cast<MCSymbolRefExpr>(Symbol.getVariableValue()); |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 403 | |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 404 | if (!SymRef) |
| 405 | report_fatal_error("Weak externals may only alias symbols"); |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 406 | |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 407 | coff_symbol->Other = GetOrCreateCOFFSymbol(&SymRef->getSymbol()); |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 408 | } else { |
Yaron Keren | 075759a | 2015-03-30 15:42:36 +0000 | [diff] [blame] | 409 | std::string WeakName = (".weak." + Symbol.getName() + ".default").str(); |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 410 | COFFSymbol *WeakDefault = createSymbol(WeakName); |
| 411 | WeakDefault->Data.SectionNumber = COFF::IMAGE_SYM_ABSOLUTE; |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 412 | WeakDefault->Data.StorageClass = COFF::IMAGE_SYM_CLASS_EXTERNAL; |
| 413 | WeakDefault->Data.Type = 0; |
| 414 | WeakDefault->Data.Value = 0; |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 415 | coff_symbol->Other = WeakDefault; |
| 416 | } |
| 417 | |
| 418 | // Setup the Weak External auxiliary symbol. |
| 419 | coff_symbol->Aux.resize(1); |
| 420 | memset(&coff_symbol->Aux[0], 0, sizeof(coff_symbol->Aux[0])); |
| 421 | coff_symbol->Aux[0].AuxType = ATWeakExternal; |
| 422 | coff_symbol->Aux[0].Aux.WeakExternal.TagIndex = 0; |
| 423 | coff_symbol->Aux[0].Aux.WeakExternal.Characteristics = |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 424 | COFF::IMAGE_WEAK_EXTERN_SEARCH_LIBRARY; |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 425 | |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 426 | coff_symbol->MC = &Symbol; |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 427 | } else { |
Rafael Espindola | 575f79a | 2014-05-01 13:37:57 +0000 | [diff] [blame] | 428 | const MCSymbol *Base = Layout.getBaseSymbol(Symbol); |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 429 | coff_symbol->Data.Value = getSymbolValue(Symbol, Layout); |
Reid Kleckner | c1e7621 | 2013-09-17 23:18:05 +0000 | [diff] [blame] | 430 | |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 431 | coff_symbol->Data.Type = |
| 432 | (Symbol.getFlags() & COFF::SF_TypeMask) >> COFF::SF_TypeShift; |
| 433 | coff_symbol->Data.StorageClass = |
| 434 | (Symbol.getFlags() & COFF::SF_ClassMask) >> COFF::SF_ClassShift; |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 435 | |
| 436 | // If no storage class was specified in the streamer, define it here. |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 437 | if (coff_symbol->Data.StorageClass == COFF::IMAGE_SYM_CLASS_NULL) { |
Rafael Espindola | 4d37b2a | 2015-05-29 21:45:01 +0000 | [diff] [blame] | 438 | bool IsExternal = Symbol.isExternal() || |
| 439 | (!Symbol.getFragment() && !Symbol.isVariable()); |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 440 | |
David Majnemer | 299674e | 2014-07-13 04:31:19 +0000 | [diff] [blame] | 441 | coff_symbol->Data.StorageClass = IsExternal |
| 442 | ? COFF::IMAGE_SYM_CLASS_EXTERNAL |
| 443 | : COFF::IMAGE_SYM_CLASS_STATIC; |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 444 | } |
| 445 | |
Rafael Espindola | 575f79a | 2014-05-01 13:37:57 +0000 | [diff] [blame] | 446 | if (!Base) { |
Reid Kleckner | c1e7621 | 2013-09-17 23:18:05 +0000 | [diff] [blame] | 447 | coff_symbol->Data.SectionNumber = COFF::IMAGE_SYM_ABSOLUTE; |
Rafael Espindola | 575f79a | 2014-05-01 13:37:57 +0000 | [diff] [blame] | 448 | } else { |
Rafael Espindola | 4d37b2a | 2015-05-29 21:45:01 +0000 | [diff] [blame] | 449 | if (Base->getFragment()) { |
| 450 | COFFSection *Sec = SectionMap[Base->getFragment()->getParent()]; |
Rafael Espindola | 0766ae0 | 2014-06-06 19:26:12 +0000 | [diff] [blame] | 451 | |
| 452 | if (coff_symbol->Section && coff_symbol->Section != Sec) |
| 453 | report_fatal_error("conflicting sections for symbol"); |
| 454 | |
| 455 | coff_symbol->Section = Sec; |
| 456 | } |
Rafael Espindola | 575f79a | 2014-05-01 13:37:57 +0000 | [diff] [blame] | 457 | } |
Peter Collingbourne | 8988687 | 2013-04-22 18:48:56 +0000 | [diff] [blame] | 458 | |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 459 | coff_symbol->MC = &Symbol; |
Michael J. Spencer | 17990d5 | 2010-10-16 08:25:57 +0000 | [diff] [blame] | 460 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 461 | } |
| 462 | |
Nico Rieck | 01143f9 | 2014-02-25 09:50:40 +0000 | [diff] [blame] | 463 | // Maximum offsets for different string table entry encodings. |
| 464 | static const unsigned Max6DecimalOffset = 999999; |
| 465 | static const unsigned Max7DecimalOffset = 9999999; |
| 466 | static const uint64_t MaxBase64Offset = 0xFFFFFFFFFULL; // 64^6, including 0 |
| 467 | |
Nico Rieck | 9d2c15e | 2014-02-22 16:12:20 +0000 | [diff] [blame] | 468 | // Encode a string table entry offset in base 64, padded to 6 chars, and |
| 469 | // prefixed with a double slash: '//AAAAAA', '//AAAAAB', ... |
| 470 | // Buffer must be at least 8 bytes large. No terminating null appended. |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 471 | static void encodeBase64StringEntry(char *Buffer, uint64_t Value) { |
Nico Rieck | 01143f9 | 2014-02-25 09:50:40 +0000 | [diff] [blame] | 472 | assert(Value > Max7DecimalOffset && Value <= MaxBase64Offset && |
Nico Rieck | 9d2c15e | 2014-02-22 16:12:20 +0000 | [diff] [blame] | 473 | "Illegal section name encoding for value"); |
| 474 | |
| 475 | static const char Alphabet[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZ" |
| 476 | "abcdefghijklmnopqrstuvwxyz" |
| 477 | "0123456789+/"; |
| 478 | |
| 479 | Buffer[0] = '/'; |
| 480 | Buffer[1] = '/'; |
| 481 | |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 482 | char *Ptr = Buffer + 7; |
Nico Rieck | 9d2c15e | 2014-02-22 16:12:20 +0000 | [diff] [blame] | 483 | for (unsigned i = 0; i < 6; ++i) { |
| 484 | unsigned Rem = Value % 64; |
| 485 | Value /= 64; |
| 486 | *(Ptr--) = Alphabet[Rem]; |
| 487 | } |
| 488 | } |
| 489 | |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 490 | void WinCOFFObjectWriter::SetSectionName(COFFSection &S) { |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 491 | if (S.Name.size() > COFF::NameSize) { |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 492 | uint64_t StringTableEntry = Strings.getOffset(S.Name); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 493 | |
Nico Rieck | 01143f9 | 2014-02-25 09:50:40 +0000 | [diff] [blame] | 494 | if (StringTableEntry <= Max6DecimalOffset) { |
Nico Rieck | 2c9c89b | 2013-07-29 12:30:12 +0000 | [diff] [blame] | 495 | std::sprintf(S.Header.Name, "/%d", unsigned(StringTableEntry)); |
Nico Rieck | 01143f9 | 2014-02-25 09:50:40 +0000 | [diff] [blame] | 496 | } else if (StringTableEntry <= Max7DecimalOffset) { |
Nico Rieck | 2c9c89b | 2013-07-29 12:30:12 +0000 | [diff] [blame] | 497 | // With seven digits, we have to skip the terminating null. Because |
| 498 | // sprintf always appends it, we use a larger temporary buffer. |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 499 | char buffer[9] = {}; |
Nico Rieck | 2c9c89b | 2013-07-29 12:30:12 +0000 | [diff] [blame] | 500 | std::sprintf(buffer, "/%d", unsigned(StringTableEntry)); |
| 501 | std::memcpy(S.Header.Name, buffer, 8); |
Nico Rieck | 01143f9 | 2014-02-25 09:50:40 +0000 | [diff] [blame] | 502 | } else if (StringTableEntry <= MaxBase64Offset) { |
Nico Rieck | 9d2c15e | 2014-02-22 16:12:20 +0000 | [diff] [blame] | 503 | // Starting with 10,000,000, offsets are encoded as base64. |
| 504 | encodeBase64StringEntry(S.Header.Name, StringTableEntry); |
Nico Rieck | 2c9c89b | 2013-07-29 12:30:12 +0000 | [diff] [blame] | 505 | } else { |
Nico Rieck | 9d2c15e | 2014-02-22 16:12:20 +0000 | [diff] [blame] | 506 | report_fatal_error("COFF string table is greater than 64 GB."); |
Nico Rieck | 2c9c89b | 2013-07-29 12:30:12 +0000 | [diff] [blame] | 507 | } |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 508 | } else |
| 509 | std::memcpy(S.Header.Name, S.Name.c_str(), S.Name.size()); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 510 | } |
| 511 | |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 512 | void WinCOFFObjectWriter::SetSymbolName(COFFSymbol &S) { |
| 513 | if (S.Name.size() > COFF::NameSize) |
| 514 | S.set_name_offset(Strings.getOffset(S.Name)); |
| 515 | else |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 516 | std::memcpy(S.Data.Name, S.Name.c_str(), S.Name.size()); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 517 | } |
| 518 | |
David Majnemer | 299674e | 2014-07-13 04:31:19 +0000 | [diff] [blame] | 519 | bool WinCOFFObjectWriter::ExportSymbol(const MCSymbol &Symbol, |
Reid Kleckner | 8b2ad2a | 2013-11-18 23:08:12 +0000 | [diff] [blame] | 520 | MCAssembler &Asm) { |
| 521 | // This doesn't seem to be right. Strings referred to from the .data section |
| 522 | // need symbols so they can be linked to code in the .text section right? |
| 523 | |
David Majnemer | 299674e | 2014-07-13 04:31:19 +0000 | [diff] [blame] | 524 | // return Asm.isSymbolLinkerVisible(Symbol); |
| 525 | |
| 526 | // Non-temporary labels should always be visible to the linker. |
| 527 | if (!Symbol.isTemporary()) |
| 528 | return true; |
| 529 | |
| 530 | // Absolute temporary labels are never visible. |
| 531 | if (!Symbol.isInSection()) |
| 532 | return false; |
Reid Kleckner | 8b2ad2a | 2013-11-18 23:08:12 +0000 | [diff] [blame] | 533 | |
| 534 | // For now, all non-variable symbols are exported, |
| 535 | // the linker will sort the rest out for us. |
David Majnemer | 299674e | 2014-07-13 04:31:19 +0000 | [diff] [blame] | 536 | return !Symbol.isVariable(); |
Reid Kleckner | 8b2ad2a | 2013-11-18 23:08:12 +0000 | [diff] [blame] | 537 | } |
| 538 | |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 539 | bool WinCOFFObjectWriter::IsPhysicalSection(COFFSection *S) { |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 540 | return (S->Header.Characteristics & COFF::IMAGE_SCN_CNT_UNINITIALIZED_DATA) == |
| 541 | 0; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 542 | } |
| 543 | |
| 544 | //------------------------------------------------------------------------------ |
| 545 | // entity writing methods |
| 546 | |
| 547 | void WinCOFFObjectWriter::WriteFileHeader(const COFF::header &Header) { |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 548 | if (UseBigObj) { |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 549 | writeLE16(COFF::IMAGE_FILE_MACHINE_UNKNOWN); |
| 550 | writeLE16(0xFFFF); |
| 551 | writeLE16(COFF::BigObjHeader::MinBigObjectVersion); |
| 552 | writeLE16(Header.Machine); |
| 553 | writeLE32(Header.TimeDateStamp); |
| 554 | writeBytes(StringRef(COFF::BigObjMagic, sizeof(COFF::BigObjMagic))); |
| 555 | writeLE32(0); |
| 556 | writeLE32(0); |
| 557 | writeLE32(0); |
| 558 | writeLE32(0); |
| 559 | writeLE32(Header.NumberOfSections); |
| 560 | writeLE32(Header.PointerToSymbolTable); |
| 561 | writeLE32(Header.NumberOfSymbols); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 562 | } else { |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 563 | writeLE16(Header.Machine); |
| 564 | writeLE16(static_cast<int16_t>(Header.NumberOfSections)); |
| 565 | writeLE32(Header.TimeDateStamp); |
| 566 | writeLE32(Header.PointerToSymbolTable); |
| 567 | writeLE32(Header.NumberOfSymbols); |
| 568 | writeLE16(Header.SizeOfOptionalHeader); |
| 569 | writeLE16(Header.Characteristics); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 570 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 571 | } |
| 572 | |
David Blaikie | f564ab6 | 2014-04-15 05:25:03 +0000 | [diff] [blame] | 573 | void WinCOFFObjectWriter::WriteSymbol(const COFFSymbol &S) { |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 574 | writeBytes(StringRef(S.Data.Name, COFF::NameSize)); |
| 575 | writeLE32(S.Data.Value); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 576 | if (UseBigObj) |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 577 | writeLE32(S.Data.SectionNumber); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 578 | else |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 579 | writeLE16(static_cast<int16_t>(S.Data.SectionNumber)); |
| 580 | writeLE16(S.Data.Type); |
| 581 | write8(S.Data.StorageClass); |
| 582 | write8(S.Data.NumberOfAuxSymbols); |
David Blaikie | f564ab6 | 2014-04-15 05:25:03 +0000 | [diff] [blame] | 583 | WriteAuxiliarySymbols(S.Aux); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 584 | } |
| 585 | |
| 586 | void WinCOFFObjectWriter::WriteAuxiliarySymbols( |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 587 | const COFFSymbol::AuxiliarySymbols &S) { |
| 588 | for (COFFSymbol::AuxiliarySymbols::const_iterator i = S.begin(), e = S.end(); |
| 589 | i != e; ++i) { |
| 590 | switch (i->AuxType) { |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 591 | case ATFunctionDefinition: |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 592 | writeLE32(i->Aux.FunctionDefinition.TagIndex); |
| 593 | writeLE32(i->Aux.FunctionDefinition.TotalSize); |
| 594 | writeLE32(i->Aux.FunctionDefinition.PointerToLinenumber); |
| 595 | writeLE32(i->Aux.FunctionDefinition.PointerToNextFunction); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 596 | WriteZeros(sizeof(i->Aux.FunctionDefinition.unused)); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 597 | if (UseBigObj) |
| 598 | WriteZeros(COFF::Symbol32Size - COFF::Symbol16Size); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 599 | break; |
| 600 | case ATbfAndefSymbol: |
| 601 | WriteZeros(sizeof(i->Aux.bfAndefSymbol.unused1)); |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 602 | writeLE16(i->Aux.bfAndefSymbol.Linenumber); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 603 | WriteZeros(sizeof(i->Aux.bfAndefSymbol.unused2)); |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 604 | writeLE32(i->Aux.bfAndefSymbol.PointerToNextFunction); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 605 | WriteZeros(sizeof(i->Aux.bfAndefSymbol.unused3)); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 606 | if (UseBigObj) |
| 607 | WriteZeros(COFF::Symbol32Size - COFF::Symbol16Size); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 608 | break; |
| 609 | case ATWeakExternal: |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 610 | writeLE32(i->Aux.WeakExternal.TagIndex); |
| 611 | writeLE32(i->Aux.WeakExternal.Characteristics); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 612 | WriteZeros(sizeof(i->Aux.WeakExternal.unused)); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 613 | if (UseBigObj) |
| 614 | WriteZeros(COFF::Symbol32Size - COFF::Symbol16Size); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 615 | break; |
| 616 | case ATFile: |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 617 | writeBytes( |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 618 | StringRef(reinterpret_cast<const char *>(&i->Aux), |
| 619 | UseBigObj ? COFF::Symbol32Size : COFF::Symbol16Size)); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 620 | break; |
| 621 | case ATSectionDefinition: |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 622 | writeLE32(i->Aux.SectionDefinition.Length); |
| 623 | writeLE16(i->Aux.SectionDefinition.NumberOfRelocations); |
| 624 | writeLE16(i->Aux.SectionDefinition.NumberOfLinenumbers); |
| 625 | writeLE32(i->Aux.SectionDefinition.CheckSum); |
| 626 | writeLE16(static_cast<int16_t>(i->Aux.SectionDefinition.Number)); |
| 627 | write8(i->Aux.SectionDefinition.Selection); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 628 | WriteZeros(sizeof(i->Aux.SectionDefinition.unused)); |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 629 | writeLE16(static_cast<int16_t>(i->Aux.SectionDefinition.Number >> 16)); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 630 | if (UseBigObj) |
| 631 | WriteZeros(COFF::Symbol32Size - COFF::Symbol16Size); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 632 | break; |
| 633 | } |
| 634 | } |
| 635 | } |
| 636 | |
| 637 | void WinCOFFObjectWriter::WriteSectionHeader(const COFF::section &S) { |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 638 | writeBytes(StringRef(S.Name, COFF::NameSize)); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 639 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 640 | writeLE32(S.VirtualSize); |
| 641 | writeLE32(S.VirtualAddress); |
| 642 | writeLE32(S.SizeOfRawData); |
| 643 | writeLE32(S.PointerToRawData); |
| 644 | writeLE32(S.PointerToRelocations); |
| 645 | writeLE32(S.PointerToLineNumbers); |
| 646 | writeLE16(S.NumberOfRelocations); |
| 647 | writeLE16(S.NumberOfLineNumbers); |
| 648 | writeLE32(S.Characteristics); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 649 | } |
| 650 | |
| 651 | void WinCOFFObjectWriter::WriteRelocation(const COFF::relocation &R) { |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 652 | writeLE32(R.VirtualAddress); |
| 653 | writeLE32(R.SymbolTableIndex); |
| 654 | writeLE16(R.Type); |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 655 | } |
| 656 | |
| 657 | //////////////////////////////////////////////////////////////////////////////// |
| 658 | // MCObjectWriter interface implementations |
| 659 | |
Rafael Espindola | 93e3cf0 | 2010-12-07 00:27:36 +0000 | [diff] [blame] | 660 | void WinCOFFObjectWriter::ExecutePostLayoutBinding(MCAssembler &Asm, |
| 661 | const MCAsmLayout &Layout) { |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 662 | // "Define" each section & symbol. This creates section & symbol |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 663 | // entries in the staging area. |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 664 | for (const auto &Section : Asm) |
Rafael Espindola | f59264f | 2015-05-27 14:45:54 +0000 | [diff] [blame] | 665 | defineSection(static_cast<const MCSectionCOFF &>(Section)); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 666 | |
Duncan P. N. Exon Smith | f48de1c | 2015-05-16 00:35:24 +0000 | [diff] [blame] | 667 | for (const MCSymbol &Symbol : Asm.symbols()) |
| 668 | if (ExportSymbol(Symbol, Asm)) |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 669 | DefineSymbol(Symbol, Asm, Layout); |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 670 | } |
| 671 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 672 | bool WinCOFFObjectWriter::isSymbolRefDifferenceFullyResolvedImpl( |
Duncan P. N. Exon Smith | d81ba53 | 2015-05-16 01:01:55 +0000 | [diff] [blame] | 673 | const MCAssembler &Asm, const MCSymbol &SymA, const MCFragment &FB, |
Rafael Espindola | 35d6189 | 2015-04-17 21:15:17 +0000 | [diff] [blame] | 674 | bool InSet, bool IsPCRel) const { |
David Majnemer | 2cc4bc77 | 2014-11-11 08:43:57 +0000 | [diff] [blame] | 675 | // MS LINK expects to be able to replace all references to a function with a |
| 676 | // thunk to implement their /INCREMENTAL feature. Make sure we don't optimize |
| 677 | // away any relocations to functions. |
Rafael Espindola | 2229d33 | 2015-05-29 19:07:51 +0000 | [diff] [blame] | 678 | if ((((SymA.getFlags() & COFF::SF_TypeMask) >> COFF::SF_TypeShift) >> |
David Majnemer | 2cc4bc77 | 2014-11-11 08:43:57 +0000 | [diff] [blame] | 679 | COFF::SCT_COMPLEX_TYPE_SHIFT) == COFF::IMAGE_SYM_DTYPE_FUNCTION) |
| 680 | return false; |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 681 | return MCObjectWriter::isSymbolRefDifferenceFullyResolvedImpl(Asm, SymA, FB, |
Rafael Espindola | 35d6189 | 2015-04-17 21:15:17 +0000 | [diff] [blame] | 682 | InSet, IsPCRel); |
David Majnemer | 2cc4bc77 | 2014-11-11 08:43:57 +0000 | [diff] [blame] | 683 | } |
| 684 | |
Duncan P. N. Exon Smith | 5266ad9 | 2015-05-20 15:10:03 +0000 | [diff] [blame] | 685 | bool WinCOFFObjectWriter::isWeak(const MCSymbol &Sym) const { |
Rafael Espindola | 4d37b2a | 2015-05-29 21:45:01 +0000 | [diff] [blame] | 686 | if (!Sym.isExternal()) |
Rafael Espindola | 88af411 | 2015-04-17 11:27:13 +0000 | [diff] [blame] | 687 | return false; |
| 688 | |
Rafael Espindola | 88af411 | 2015-04-17 11:27:13 +0000 | [diff] [blame] | 689 | if (!Sym.isInSection()) |
| 690 | return false; |
| 691 | |
| 692 | const auto &Sec = cast<MCSectionCOFF>(Sym.getSection()); |
| 693 | if (!Sec.getCOMDATSymbol()) |
| 694 | return false; |
| 695 | |
| 696 | // It looks like for COFF it is invalid to replace a reference to a global |
| 697 | // in a comdat with a reference to a local. |
| 698 | // FIXME: Add a specification reference if available. |
| 699 | return true; |
Rafael Espindola | aeed3cb | 2015-03-26 21:11:00 +0000 | [diff] [blame] | 700 | } |
| 701 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 702 | void WinCOFFObjectWriter::recordRelocation( |
Rafael Espindola | 2658554 | 2015-01-19 21:11:14 +0000 | [diff] [blame] | 703 | MCAssembler &Asm, const MCAsmLayout &Layout, const MCFragment *Fragment, |
| 704 | const MCFixup &Fixup, MCValue Target, bool &IsPCRel, uint64_t &FixedValue) { |
Craig Topper | bb694de | 2014-04-13 04:57:38 +0000 | [diff] [blame] | 705 | assert(Target.getSymA() && "Relocation must reference a symbol!"); |
Michael J. Spencer | ccd28d0 | 2010-08-24 21:04:52 +0000 | [diff] [blame] | 706 | |
Reid Kleckner | dae7b4e | 2013-07-15 19:41:21 +0000 | [diff] [blame] | 707 | const MCSymbol &Symbol = Target.getSymA()->getSymbol(); |
Rafael Espindola | 88af411 | 2015-04-17 11:27:13 +0000 | [diff] [blame] | 708 | const MCSymbol &A = Symbol; |
Rafael Espindola | 2641014 | 2015-06-01 01:52:18 +0000 | [diff] [blame] | 709 | if (!A.isRegistered()) |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 710 | Asm.getContext().reportFatalError(Fixup.getLoc(), |
| 711 | Twine("symbol '") + A.getName() + |
| 712 | "' can not be undefined"); |
Timur Iskhodzhanov | 3e4ac4e | 2014-01-30 21:13:05 +0000 | [diff] [blame] | 713 | |
Rafael Espindola | 7549f87 | 2015-05-26 00:36:57 +0000 | [diff] [blame] | 714 | MCSection *Section = Fragment->getParent(); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 715 | |
Michael J. Spencer | ccd28d0 | 2010-08-24 21:04:52 +0000 | [diff] [blame] | 716 | // Mark this symbol as requiring an entry in the symbol table. |
Rafael Espindola | 7549f87 | 2015-05-26 00:36:57 +0000 | [diff] [blame] | 717 | assert(SectionMap.find(Section) != SectionMap.end() && |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 718 | "Section must already have been defined in ExecutePostLayoutBinding!"); |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 719 | assert(SymbolMap.find(&A) != SymbolMap.end() && |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 720 | "Symbol must already have been defined in ExecutePostLayoutBinding!"); |
| 721 | |
Rafael Espindola | 7549f87 | 2015-05-26 00:36:57 +0000 | [diff] [blame] | 722 | COFFSection *coff_section = SectionMap[Section]; |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 723 | COFFSymbol *coff_symbol = SymbolMap[&A]; |
Rafael Espindola | ed16477 | 2011-04-20 14:01:45 +0000 | [diff] [blame] | 724 | const MCSymbolRefExpr *SymB = Target.getSymB(); |
David Majnemer | a45a176 | 2014-01-06 07:39:46 +0000 | [diff] [blame] | 725 | bool CrossSection = false; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 726 | |
David Majnemer | a45a176 | 2014-01-06 07:39:46 +0000 | [diff] [blame] | 727 | if (SymB) { |
| 728 | const MCSymbol *B = &SymB->getSymbol(); |
Rafael Espindola | 4d37b2a | 2015-05-29 21:45:01 +0000 | [diff] [blame] | 729 | if (!B->getFragment()) |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 730 | Asm.getContext().reportFatalError( |
David Majnemer | a45a176 | 2014-01-06 07:39:46 +0000 | [diff] [blame] | 731 | Fixup.getLoc(), |
| 732 | Twine("symbol '") + B->getName() + |
| 733 | "' can not be undefined in a subtraction expression"); |
| 734 | |
Rafael Espindola | 4d37b2a | 2015-05-29 21:45:01 +0000 | [diff] [blame] | 735 | if (!A.getFragment()) |
Jim Grosbach | 6f48200 | 2015-05-18 18:43:14 +0000 | [diff] [blame] | 736 | Asm.getContext().reportFatalError( |
David Majnemer | a45a176 | 2014-01-06 07:39:46 +0000 | [diff] [blame] | 737 | Fixup.getLoc(), |
| 738 | Twine("symbol '") + Symbol.getName() + |
| 739 | "' can not be undefined in a subtraction expression"); |
| 740 | |
| 741 | CrossSection = &Symbol.getSection() != &B->getSection(); |
Michael J. Spencer | ccd28d0 | 2010-08-24 21:04:52 +0000 | [diff] [blame] | 742 | |
Rafael Espindola | c3dc486 | 2011-04-21 18:36:50 +0000 | [diff] [blame] | 743 | // Offset of the symbol in the section |
Duncan P. N. Exon Smith | 2a40483 | 2015-05-19 23:53:20 +0000 | [diff] [blame] | 744 | int64_t OffsetOfB = Layout.getSymbolOffset(*B); |
Michael J. Spencer | ccd28d0 | 2010-08-24 21:04:52 +0000 | [diff] [blame] | 745 | |
| 746 | // In the case where we have SymbA and SymB, we just need to store the delta |
| 747 | // between the two symbols. Update FixedValue to account for the delta, and |
| 748 | // skip recording the relocation. |
David Majnemer | 1de3094 | 2015-02-09 06:31:31 +0000 | [diff] [blame] | 749 | if (!CrossSection) { |
Duncan P. N. Exon Smith | 2a40483 | 2015-05-19 23:53:20 +0000 | [diff] [blame] | 750 | int64_t OffsetOfA = Layout.getSymbolOffset(A); |
David Majnemer | 1de3094 | 2015-02-09 06:31:31 +0000 | [diff] [blame] | 751 | FixedValue = (OffsetOfA - OffsetOfB) + Target.getConstant(); |
Rafael Espindola | ed16477 | 2011-04-20 14:01:45 +0000 | [diff] [blame] | 752 | return; |
David Majnemer | 1de3094 | 2015-02-09 06:31:31 +0000 | [diff] [blame] | 753 | } |
| 754 | |
| 755 | // Offset of the relocation in the section |
| 756 | int64_t OffsetOfRelocation = |
| 757 | Layout.getFragmentOffset(Fragment) + Fixup.getOffset(); |
| 758 | |
Andy Ayers | 9e5c851 | 2015-05-14 01:10:41 +0000 | [diff] [blame] | 759 | FixedValue = (OffsetOfRelocation - OffsetOfB) + Target.getConstant(); |
Michael J. Spencer | ccd28d0 | 2010-08-24 21:04:52 +0000 | [diff] [blame] | 760 | } else { |
| 761 | FixedValue = Target.getConstant(); |
| 762 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 763 | |
| 764 | COFFRelocation Reloc; |
| 765 | |
Daniel Dunbar | 727be43 | 2010-07-31 21:08:54 +0000 | [diff] [blame] | 766 | Reloc.Data.SymbolTableIndex = 0; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 767 | Reloc.Data.VirtualAddress = Layout.getFragmentOffset(Fragment); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 768 | |
Michael J. Spencer | a65d17a | 2010-10-05 19:48:12 +0000 | [diff] [blame] | 769 | // Turn relocations for temporary symbols into section relocations. |
Duncan P. N. Exon Smith | e8fb3a2 | 2015-05-20 19:34:08 +0000 | [diff] [blame] | 770 | if (coff_symbol->MC->isTemporary() || CrossSection) { |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 771 | Reloc.Symb = coff_symbol->Section->Symbol; |
Rafael Espindola | 4d37b2a | 2015-05-29 21:45:01 +0000 | [diff] [blame] | 772 | FixedValue += Layout.getFragmentOffset(coff_symbol->MC->getFragment()) + |
| 773 | coff_symbol->MC->getOffset(); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 774 | } else |
| 775 | Reloc.Symb = coff_symbol; |
| 776 | |
| 777 | ++Reloc.Symb->Relocations; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 778 | |
| 779 | Reloc.Data.VirtualAddress += Fixup.getOffset(); |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 780 | Reloc.Data.Type = TargetObjectWriter->getRelocType( |
| 781 | Target, Fixup, CrossSection, Asm.getBackend()); |
Rafael Espindola | e61724a | 2011-12-22 22:21:47 +0000 | [diff] [blame] | 782 | |
| 783 | // FIXME: Can anyone explain what this does other than adjust for the size |
| 784 | // of the offset? |
Saleem Abdulrasool | 2c08051 | 2014-04-13 20:47:55 +0000 | [diff] [blame] | 785 | if ((Header.Machine == COFF::IMAGE_FILE_MACHINE_AMD64 && |
| 786 | Reloc.Data.Type == COFF::IMAGE_REL_AMD64_REL32) || |
| 787 | (Header.Machine == COFF::IMAGE_FILE_MACHINE_I386 && |
| 788 | Reloc.Data.Type == COFF::IMAGE_REL_I386_REL32)) |
Michael J. Spencer | ccd28d0 | 2010-08-24 21:04:52 +0000 | [diff] [blame] | 789 | FixedValue += 4; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 790 | |
Saleem Abdulrasool | 84b952b | 2014-04-27 03:48:22 +0000 | [diff] [blame] | 791 | if (Header.Machine == COFF::IMAGE_FILE_MACHINE_ARMNT) { |
| 792 | switch (Reloc.Data.Type) { |
| 793 | case COFF::IMAGE_REL_ARM_ABSOLUTE: |
| 794 | case COFF::IMAGE_REL_ARM_ADDR32: |
| 795 | case COFF::IMAGE_REL_ARM_ADDR32NB: |
| 796 | case COFF::IMAGE_REL_ARM_TOKEN: |
| 797 | case COFF::IMAGE_REL_ARM_SECTION: |
| 798 | case COFF::IMAGE_REL_ARM_SECREL: |
| 799 | break; |
| 800 | case COFF::IMAGE_REL_ARM_BRANCH11: |
| 801 | case COFF::IMAGE_REL_ARM_BLX11: |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 802 | // IMAGE_REL_ARM_BRANCH11 and IMAGE_REL_ARM_BLX11 are only used for |
| 803 | // pre-ARMv7, which implicitly rules it out of ARMNT (it would be valid |
| 804 | // for Windows CE). |
Saleem Abdulrasool | 84b952b | 2014-04-27 03:48:22 +0000 | [diff] [blame] | 805 | case COFF::IMAGE_REL_ARM_BRANCH24: |
| 806 | case COFF::IMAGE_REL_ARM_BLX24: |
| 807 | case COFF::IMAGE_REL_ARM_MOV32A: |
| 808 | // IMAGE_REL_ARM_BRANCH24, IMAGE_REL_ARM_BLX24, IMAGE_REL_ARM_MOV32A are |
| 809 | // only used for ARM mode code, which is documented as being unsupported |
Alp Toker | beaca19 | 2014-05-15 01:52:21 +0000 | [diff] [blame] | 810 | // by Windows on ARM. Empirical proof indicates that masm is able to |
Saleem Abdulrasool | 84b952b | 2014-04-27 03:48:22 +0000 | [diff] [blame] | 811 | // generate the relocations however the rest of the MSVC toolchain is |
| 812 | // unable to handle it. |
| 813 | llvm_unreachable("unsupported relocation"); |
| 814 | break; |
| 815 | case COFF::IMAGE_REL_ARM_MOV32T: |
| 816 | break; |
| 817 | case COFF::IMAGE_REL_ARM_BRANCH20T: |
| 818 | case COFF::IMAGE_REL_ARM_BRANCH24T: |
| 819 | case COFF::IMAGE_REL_ARM_BLX23T: |
| 820 | // IMAGE_REL_BRANCH20T, IMAGE_REL_ARM_BRANCH24T, IMAGE_REL_ARM_BLX23T all |
| 821 | // perform a 4 byte adjustment to the relocation. Relative branches are |
| 822 | // offset by 4 on ARM, however, because there is no RELA relocations, all |
| 823 | // branches are offset by 4. |
| 824 | FixedValue = FixedValue + 4; |
| 825 | break; |
| 826 | } |
| 827 | } |
| 828 | |
Saleem Abdulrasool | 54bed12 | 2014-05-21 23:17:50 +0000 | [diff] [blame] | 829 | if (TargetObjectWriter->recordRelocation(Fixup)) |
| 830 | coff_section->Relocations.push_back(Reloc); |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 831 | } |
| 832 | |
Jim Grosbach | 36e60e9 | 2015-06-04 22:24:41 +0000 | [diff] [blame^] | 833 | void WinCOFFObjectWriter::writeObject(MCAssembler &Asm, |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 834 | const MCAsmLayout &Layout) { |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 835 | size_t SectionsSize = Sections.size(); |
| 836 | if (SectionsSize > static_cast<size_t>(INT32_MAX)) |
| 837 | report_fatal_error( |
| 838 | "PE COFF object files can't have more than 2147483647 sections"); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 839 | |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 840 | // Assign symbol and section indexes and offsets. |
| 841 | int32_t NumberOfSections = static_cast<int32_t>(SectionsSize); |
| 842 | |
| 843 | UseBigObj = NumberOfSections > COFF::MaxNumberOfSections16; |
| 844 | |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 845 | // Assign section numbers. |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 846 | size_t Number = 1; |
David Majnemer | 9ab5ff1 | 2014-08-28 04:02:50 +0000 | [diff] [blame] | 847 | for (const auto &Section : Sections) { |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 848 | Section->Number = Number; |
| 849 | Section->Symbol->Data.SectionNumber = Number; |
| 850 | Section->Symbol->Aux[0].Aux.SectionDefinition.Number = Number; |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 851 | ++Number; |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 852 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 853 | |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 854 | Header.NumberOfSections = NumberOfSections; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 855 | Header.NumberOfSymbols = 0; |
| 856 | |
Rafael Espindola | 66f3c9c | 2015-05-28 18:03:20 +0000 | [diff] [blame] | 857 | for (const std::string &Name : Asm.getFileNames()) { |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 858 | // round up to calculate the number of auxiliary symbols required |
| 859 | unsigned SymbolSize = UseBigObj ? COFF::Symbol32Size : COFF::Symbol16Size; |
Rafael Espindola | 66f3c9c | 2015-05-28 18:03:20 +0000 | [diff] [blame] | 860 | unsigned Count = (Name.size() + SymbolSize - 1) / SymbolSize; |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 861 | |
| 862 | COFFSymbol *file = createSymbol(".file"); |
| 863 | file->Data.SectionNumber = COFF::IMAGE_SYM_DEBUG; |
| 864 | file->Data.StorageClass = COFF::IMAGE_SYM_CLASS_FILE; |
| 865 | file->Aux.resize(Count); |
| 866 | |
| 867 | unsigned Offset = 0; |
Rafael Espindola | 66f3c9c | 2015-05-28 18:03:20 +0000 | [diff] [blame] | 868 | unsigned Length = Name.size(); |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 869 | for (auto &Aux : file->Aux) { |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 870 | Aux.AuxType = ATFile; |
| 871 | |
| 872 | if (Length > SymbolSize) { |
Rafael Espindola | 66f3c9c | 2015-05-28 18:03:20 +0000 | [diff] [blame] | 873 | memcpy(&Aux.Aux, Name.c_str() + Offset, SymbolSize); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 874 | Length = Length - SymbolSize; |
| 875 | } else { |
Rafael Espindola | 66f3c9c | 2015-05-28 18:03:20 +0000 | [diff] [blame] | 876 | memcpy(&Aux.Aux, Name.c_str() + Offset, Length); |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 877 | memset((char *)&Aux.Aux + Length, 0, SymbolSize - Length); |
| 878 | break; |
| 879 | } |
| 880 | |
| 881 | Offset += SymbolSize; |
| 882 | } |
| 883 | } |
| 884 | |
David Majnemer | 9ab5ff1 | 2014-08-28 04:02:50 +0000 | [diff] [blame] | 885 | for (auto &Symbol : Symbols) { |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 886 | // Update section number & offset for symbols that have them. |
Rafael Espindola | 575f79a | 2014-05-01 13:37:57 +0000 | [diff] [blame] | 887 | if (Symbol->Section) |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 888 | Symbol->Data.SectionNumber = Symbol->Section->Number; |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 889 | if (Symbol->should_keep()) { |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 890 | Symbol->setIndex(Header.NumberOfSymbols++); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 891 | // Update auxiliary symbol info. |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 892 | Symbol->Data.NumberOfAuxSymbols = Symbol->Aux.size(); |
| 893 | Header.NumberOfSymbols += Symbol->Data.NumberOfAuxSymbols; |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 894 | } else { |
| 895 | Symbol->setIndex(-1); |
| 896 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 897 | } |
| 898 | |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 899 | // Build string table. |
| 900 | for (const auto &S : Sections) |
| 901 | if (S->Name.size() > COFF::NameSize) |
| 902 | Strings.add(S->Name); |
| 903 | for (const auto &S : Symbols) |
| 904 | if (S->should_keep() && S->Name.size() > COFF::NameSize) |
| 905 | Strings.add(S->Name); |
| 906 | Strings.finalize(StringTableBuilder::WinCOFF); |
| 907 | |
| 908 | // Set names. |
| 909 | for (const auto &S : Sections) |
| 910 | SetSectionName(*S); |
| 911 | for (auto &S : Symbols) |
| 912 | if (S->should_keep()) |
| 913 | SetSymbolName(*S); |
| 914 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 915 | // Fixup weak external references. |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 916 | for (auto &Symbol : Symbols) { |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 917 | if (Symbol->Other) { |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 918 | assert(Symbol->getIndex() != -1); |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 919 | assert(Symbol->Aux.size() == 1 && "Symbol must contain one aux symbol!"); |
| 920 | assert(Symbol->Aux[0].AuxType == ATWeakExternal && |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 921 | "Symbol's aux symbol must be a Weak External!"); |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 922 | Symbol->Aux[0].Aux.WeakExternal.TagIndex = Symbol->Other->getIndex(); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 923 | } |
| 924 | } |
| 925 | |
Nico Rieck | a37acf7 | 2013-07-06 12:13:10 +0000 | [diff] [blame] | 926 | // Fixup associative COMDAT sections. |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 927 | for (auto &Section : Sections) { |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 928 | if (Section->Symbol->Aux[0].Aux.SectionDefinition.Selection != |
Nico Rieck | a37acf7 | 2013-07-06 12:13:10 +0000 | [diff] [blame] | 929 | COFF::IMAGE_COMDAT_SELECT_ASSOCIATIVE) |
| 930 | continue; |
| 931 | |
Rafael Espindola | f59264f | 2015-05-27 14:45:54 +0000 | [diff] [blame] | 932 | const MCSectionCOFF &MCSec = *Section->MCSection; |
Nico Rieck | a37acf7 | 2013-07-06 12:13:10 +0000 | [diff] [blame] | 933 | |
Rafael Espindola | 0766ae0 | 2014-06-06 19:26:12 +0000 | [diff] [blame] | 934 | const MCSymbol *COMDAT = MCSec.getCOMDATSymbol(); |
| 935 | assert(COMDAT); |
| 936 | COFFSymbol *COMDATSymbol = GetOrCreateCOFFSymbol(COMDAT); |
| 937 | assert(COMDATSymbol); |
| 938 | COFFSection *Assoc = COMDATSymbol->Section; |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 939 | if (!Assoc) |
Rafael Espindola | 0766ae0 | 2014-06-06 19:26:12 +0000 | [diff] [blame] | 940 | report_fatal_error( |
| 941 | Twine("Missing associated COMDAT section for section ") + |
| 942 | MCSec.getSectionName()); |
Nico Rieck | a37acf7 | 2013-07-06 12:13:10 +0000 | [diff] [blame] | 943 | |
| 944 | // Skip this section if the associated section is unused. |
| 945 | if (Assoc->Number == -1) |
| 946 | continue; |
| 947 | |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 948 | Section->Symbol->Aux[0].Aux.SectionDefinition.Number = Assoc->Number; |
Nico Rieck | a37acf7 | 2013-07-06 12:13:10 +0000 | [diff] [blame] | 949 | } |
| 950 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 951 | // Assign file offsets to COFF object file structures. |
| 952 | |
| 953 | unsigned offset = 0; |
| 954 | |
David Majnemer | 4d57159 | 2014-09-15 19:42:42 +0000 | [diff] [blame] | 955 | if (UseBigObj) |
| 956 | offset += COFF::Header32Size; |
| 957 | else |
| 958 | offset += COFF::Header16Size; |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 959 | offset += COFF::SectionSize * Header.NumberOfSections; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 960 | |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 961 | for (const auto &Section : Asm) { |
Rafael Espindola | a554c05 | 2015-05-25 23:14:17 +0000 | [diff] [blame] | 962 | COFFSection *Sec = SectionMap[&Section]; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 963 | |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 964 | if (Sec->Number == -1) |
| 965 | continue; |
| 966 | |
Rafael Espindola | 5a1e80b | 2015-05-26 02:00:36 +0000 | [diff] [blame] | 967 | Sec->Header.SizeOfRawData = Layout.getSectionAddressSize(&Section); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 968 | |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 969 | if (IsPhysicalSection(Sec)) { |
David Majnemer | 3df3c61 | 2015-02-11 22:22:30 +0000 | [diff] [blame] | 970 | // Align the section data to a four byte boundary. |
David Majnemer | ab2b25b | 2015-02-11 22:51:55 +0000 | [diff] [blame] | 971 | offset = RoundUpToAlignment(offset, 4); |
| 972 | Sec->Header.PointerToRawData = offset; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 973 | |
| 974 | offset += Sec->Header.SizeOfRawData; |
| 975 | } |
| 976 | |
| 977 | if (Sec->Relocations.size() > 0) { |
Michael J. Spencer | fa39bd2 | 2012-03-15 09:03:03 +0000 | [diff] [blame] | 978 | bool RelocationsOverflow = Sec->Relocations.size() >= 0xffff; |
| 979 | |
| 980 | if (RelocationsOverflow) { |
David Majnemer | 9ab5ff1 | 2014-08-28 04:02:50 +0000 | [diff] [blame] | 981 | // Signal overflow by setting NumberOfRelocations to max value. Actual |
Michael J. Spencer | fa39bd2 | 2012-03-15 09:03:03 +0000 | [diff] [blame] | 982 | // size is found in reloc #0. Microsoft tools understand this. |
| 983 | Sec->Header.NumberOfRelocations = 0xffff; |
| 984 | } else { |
| 985 | Sec->Header.NumberOfRelocations = Sec->Relocations.size(); |
| 986 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 987 | Sec->Header.PointerToRelocations = offset; |
| 988 | |
Michael J. Spencer | fa39bd2 | 2012-03-15 09:03:03 +0000 | [diff] [blame] | 989 | if (RelocationsOverflow) { |
| 990 | // Reloc #0 will contain actual count, so make room for it. |
| 991 | offset += COFF::RelocationSize; |
| 992 | } |
| 993 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 994 | offset += COFF::RelocationSize * Sec->Relocations.size(); |
| 995 | |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 996 | for (auto &Relocation : Sec->Relocations) { |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 997 | assert(Relocation.Symb->getIndex() != -1); |
| 998 | Relocation.Data.SymbolTableIndex = Relocation.Symb->getIndex(); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 999 | } |
| 1000 | } |
| 1001 | |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 1002 | assert(Sec->Symbol->Aux.size() == 1 && |
| 1003 | "Section's symbol must have one aux!"); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 1004 | AuxSymbol &Aux = Sec->Symbol->Aux[0]; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1005 | assert(Aux.AuxType == ATSectionDefinition && |
| 1006 | "Section's symbol's aux symbol must be a Section Definition!"); |
| 1007 | Aux.Aux.SectionDefinition.Length = Sec->Header.SizeOfRawData; |
| 1008 | Aux.Aux.SectionDefinition.NumberOfRelocations = |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 1009 | Sec->Header.NumberOfRelocations; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1010 | Aux.Aux.SectionDefinition.NumberOfLinenumbers = |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 1011 | Sec->Header.NumberOfLineNumbers; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1012 | } |
| 1013 | |
| 1014 | Header.PointerToSymbolTable = offset; |
| 1015 | |
Rafael Espindola | 5ac4ad2 | 2013-12-04 02:26:54 +0000 | [diff] [blame] | 1016 | // We want a deterministic output. It looks like GNU as also writes 0 in here. |
Rafael Espindola | 9201e6b | 2013-12-04 02:02:55 +0000 | [diff] [blame] | 1017 | Header.TimeDateStamp = 0; |
Michael J. Spencer | a6cfbeb | 2010-08-03 04:43:33 +0000 | [diff] [blame] | 1018 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1019 | // Write it all to disk... |
| 1020 | WriteFileHeader(Header); |
| 1021 | |
| 1022 | { |
| 1023 | sections::iterator i, ie; |
Rafael Espindola | 63702e2 | 2015-06-01 01:30:01 +0000 | [diff] [blame] | 1024 | MCAssembler::iterator j, je; |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1025 | |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 1026 | for (auto &Section : Sections) { |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 1027 | if (Section->Number != -1) { |
| 1028 | if (Section->Relocations.size() >= 0xffff) |
| 1029 | Section->Header.Characteristics |= COFF::IMAGE_SCN_LNK_NRELOC_OVFL; |
| 1030 | WriteSectionHeader(Section->Header); |
Michael J. Spencer | fa39bd2 | 2012-03-15 09:03:03 +0000 | [diff] [blame] | 1031 | } |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 1032 | } |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1033 | |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 1034 | for (i = Sections.begin(), ie = Sections.end(), j = Asm.begin(), |
| 1035 | je = Asm.end(); |
Michael J. Spencer | d628377 | 2010-09-27 08:58:26 +0000 | [diff] [blame] | 1036 | (i != ie) && (j != je); ++i, ++j) { |
| 1037 | |
| 1038 | if ((*i)->Number == -1) |
| 1039 | continue; |
| 1040 | |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1041 | if ((*i)->Header.PointerToRawData != 0) { |
David Majnemer | 3df3c61 | 2015-02-11 22:22:30 +0000 | [diff] [blame] | 1042 | assert(OS.tell() <= (*i)->Header.PointerToRawData && |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1043 | "Section::PointerToRawData is insane!"); |
| 1044 | |
David Majnemer | 3df3c61 | 2015-02-11 22:22:30 +0000 | [diff] [blame] | 1045 | unsigned SectionDataPadding = (*i)->Header.PointerToRawData - OS.tell(); |
| 1046 | assert(SectionDataPadding < 4 && |
| 1047 | "Should only need at most three bytes of padding!"); |
| 1048 | |
| 1049 | WriteZeros(SectionDataPadding); |
| 1050 | |
Rafael Espindola | 64acc7f | 2015-05-26 02:17:21 +0000 | [diff] [blame] | 1051 | Asm.writeSectionData(&*j, Layout); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1052 | } |
| 1053 | |
| 1054 | if ((*i)->Relocations.size() > 0) { |
| 1055 | assert(OS.tell() == (*i)->Header.PointerToRelocations && |
| 1056 | "Section::PointerToRelocations is insane!"); |
| 1057 | |
Michael J. Spencer | fa39bd2 | 2012-03-15 09:03:03 +0000 | [diff] [blame] | 1058 | if ((*i)->Relocations.size() >= 0xffff) { |
| 1059 | // In case of overflow, write actual relocation count as first |
| 1060 | // relocation. Including the synthetic reloc itself (+ 1). |
| 1061 | COFF::relocation r; |
| 1062 | r.VirtualAddress = (*i)->Relocations.size() + 1; |
| 1063 | r.SymbolTableIndex = 0; |
| 1064 | r.Type = 0; |
| 1065 | WriteRelocation(r); |
| 1066 | } |
| 1067 | |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 1068 | for (const auto &Relocation : (*i)->Relocations) |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 1069 | WriteRelocation(Relocation.Data); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1070 | } else |
| 1071 | assert((*i)->Header.PointerToRelocations == 0 && |
| 1072 | "Section::PointerToRelocations is insane!"); |
| 1073 | } |
| 1074 | } |
| 1075 | |
| 1076 | assert(OS.tell() == Header.PointerToSymbolTable && |
| 1077 | "Header::PointerToSymbolTable is insane!"); |
| 1078 | |
Yaron Keren | 56919ef | 2014-12-04 08:30:39 +0000 | [diff] [blame] | 1079 | for (auto &Symbol : Symbols) |
David Majnemer | 4eecd30 | 2015-05-30 04:56:02 +0000 | [diff] [blame] | 1080 | if (Symbol->getIndex() != -1) |
Saleem Abdulrasool | 09ced5f | 2014-04-28 03:34:48 +0000 | [diff] [blame] | 1081 | WriteSymbol(*Symbol); |
Michael J. Spencer | b5fc138 | 2010-07-26 02:17:32 +0000 | [diff] [blame] | 1082 | |
Hans Wennborg | f26bfc1 | 2014-09-29 22:43:20 +0000 | [diff] [blame] | 1083 | OS.write(Strings.data().data(), Strings.data().size()); |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 1084 | } |
| 1085 | |
Rafael Espindola | 11e9e21 | 2015-05-27 14:37:12 +0000 | [diff] [blame] | 1086 | MCWinCOFFObjectTargetWriter::MCWinCOFFObjectTargetWriter(unsigned Machine_) |
| 1087 | : Machine(Machine_) {} |
Rafael Espindola | 908d2ed | 2011-12-24 02:14:02 +0000 | [diff] [blame] | 1088 | |
Juergen Ributzka | d12ccbd | 2013-11-19 00:57:56 +0000 | [diff] [blame] | 1089 | // Pin the vtable to this file. |
| 1090 | void MCWinCOFFObjectTargetWriter::anchor() {} |
| 1091 | |
Chris Lattner | 2c52b79 | 2010-07-11 22:07:02 +0000 | [diff] [blame] | 1092 | //------------------------------------------------------------------------------ |
| 1093 | // WinCOFFObjectWriter factory function |
| 1094 | |
Rafael Espindola | 37099d9 | 2015-04-09 18:08:15 +0000 | [diff] [blame] | 1095 | MCObjectWriter * |
| 1096 | llvm::createWinCOFFObjectWriter(MCWinCOFFObjectTargetWriter *MOTW, |
Rafael Espindola | 5560a4c | 2015-04-14 22:14:34 +0000 | [diff] [blame] | 1097 | raw_pwrite_stream &OS) { |
Rafael Espindola | 37099d9 | 2015-04-09 18:08:15 +0000 | [diff] [blame] | 1098 | return new WinCOFFObjectWriter(MOTW, OS); |
Michael J. Spencer | c212937 | 2010-07-26 03:01:28 +0000 | [diff] [blame] | 1099 | } |