172887a48ec827b028d04b361378969119e8aeb5
[oota-llvm.git] / lib / Target / X86 / MCTargetDesc / X86ELFObjectWriter.cpp
1 //===-- X86ELFObjectWriter.cpp - X86 ELF Writer ---------------------------===//
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 #include "MCTargetDesc/X86FixupKinds.h"
11 #include "MCTargetDesc/X86MCTargetDesc.h"
12 #include "llvm/MC/MCELFObjectWriter.h"
13 #include "llvm/MC/MCExpr.h"
14 #include "llvm/MC/MCValue.h"
15 #include "llvm/Support/ELF.h"
16 #include "llvm/Support/ErrorHandling.h"
17
18 using namespace llvm;
19
20 namespace {
21   class X86ELFObjectWriter : public MCELFObjectTargetWriter {
22   public:
23     X86ELFObjectWriter(bool IsELF64, uint8_t OSABI, uint16_t EMachine);
24
25     virtual ~X86ELFObjectWriter();
26   protected:
27     unsigned GetRelocType(const MCValue &Target, const MCFixup &Fixup,
28                           bool IsPCRel) const override;
29   };
30 }
31
32 X86ELFObjectWriter::X86ELFObjectWriter(bool IsELF64, uint8_t OSABI,
33                                        uint16_t EMachine)
34   : MCELFObjectTargetWriter(IsELF64, OSABI, EMachine,
35                             // Only i386 uses Rel instead of RelA.
36                             /*HasRelocationAddend*/ EMachine != ELF::EM_386) {}
37
38 X86ELFObjectWriter::~X86ELFObjectWriter()
39 {}
40
41 unsigned X86ELFObjectWriter::GetRelocType(const MCValue &Target,
42                                           const MCFixup &Fixup,
43                                           bool IsPCRel) const {
44   // determine the type of the relocation
45
46   MCSymbolRefExpr::VariantKind Modifier = Target.getAccessVariant();
47   unsigned Type;
48   if (getEMachine() == ELF::EM_X86_64) {
49     if (IsPCRel) {
50       switch ((unsigned)Fixup.getKind()) {
51       default: llvm_unreachable("invalid fixup kind!");
52
53       case FK_Data_8: Type = ELF::R_X86_64_PC64; break;
54       case FK_Data_4: Type = ELF::R_X86_64_PC32; break;
55       case FK_Data_2: Type = ELF::R_X86_64_PC16; break;
56       case FK_Data_1: Type = ELF::R_X86_64_PC8; break;
57
58       case FK_PCRel_8:
59         assert(Modifier == MCSymbolRefExpr::VK_None);
60         Type = ELF::R_X86_64_PC64;
61         break;
62       case X86::reloc_signed_4byte:
63       case X86::reloc_riprel_4byte_movq_load:
64       case X86::reloc_riprel_4byte:
65       case FK_PCRel_4:
66         switch (Modifier) {
67         default:
68           llvm_unreachable("Unimplemented");
69         case MCSymbolRefExpr::VK_None:
70           Type = ELF::R_X86_64_PC32;
71           break;
72         case MCSymbolRefExpr::VK_PLT:
73           Type = ELF::R_X86_64_PLT32;
74           break;
75         case MCSymbolRefExpr::VK_GOTPCREL:
76           Type = ELF::R_X86_64_GOTPCREL;
77           break;
78         case MCSymbolRefExpr::VK_GOTTPOFF:
79           Type = ELF::R_X86_64_GOTTPOFF;
80           break;
81         case MCSymbolRefExpr::VK_TLSGD:
82           Type = ELF::R_X86_64_TLSGD;
83           break;
84         case MCSymbolRefExpr::VK_TLSLD:
85           Type = ELF::R_X86_64_TLSLD;
86           break;
87         }
88         break;
89       case FK_PCRel_2:
90         assert(Modifier == MCSymbolRefExpr::VK_None);
91         Type = ELF::R_X86_64_PC16;
92         break;
93       case FK_PCRel_1:
94         assert(Modifier == MCSymbolRefExpr::VK_None);
95         Type = ELF::R_X86_64_PC8;
96         break;
97       }
98     } else {
99       switch ((unsigned)Fixup.getKind()) {
100       default: llvm_unreachable("invalid fixup kind!");
101       case X86::reloc_global_offset_table8:
102         Type = ELF::R_X86_64_GOTPC64;
103         break;
104       case X86::reloc_global_offset_table:
105         Type = ELF::R_X86_64_GOTPC32;
106         break;
107       case FK_Data_8:
108         switch (Modifier) {
109         default:
110           llvm_unreachable("Unimplemented");
111         case MCSymbolRefExpr::VK_None:
112           Type = ELF::R_X86_64_64;
113           break;
114         case MCSymbolRefExpr::VK_GOT:
115           Type = ELF::R_X86_64_GOT64;
116           break;
117         case MCSymbolRefExpr::VK_GOTOFF:
118           Type = ELF::R_X86_64_GOTOFF64;
119           break;
120         case MCSymbolRefExpr::VK_TPOFF:
121           Type = ELF::R_X86_64_TPOFF64;
122           break;
123         case MCSymbolRefExpr::VK_DTPOFF:
124           Type = ELF::R_X86_64_DTPOFF64;
125           break;
126         case MCSymbolRefExpr::VK_SIZE:
127           Type = ELF::R_X86_64_SIZE64;
128           break;
129         }
130         break;
131       case X86::reloc_signed_4byte:
132         switch (Modifier) {
133         default:
134           llvm_unreachable("Unimplemented");
135         case MCSymbolRefExpr::VK_None:
136           Type = ELF::R_X86_64_32S;
137           break;
138         case MCSymbolRefExpr::VK_GOT:
139           Type = ELF::R_X86_64_GOT32;
140           break;
141         case MCSymbolRefExpr::VK_GOTPCREL:
142           Type = ELF::R_X86_64_GOTPCREL;
143           break;
144         case MCSymbolRefExpr::VK_TPOFF:
145           Type = ELF::R_X86_64_TPOFF32;
146           break;
147         case MCSymbolRefExpr::VK_DTPOFF:
148           Type = ELF::R_X86_64_DTPOFF32;
149           break;
150         case MCSymbolRefExpr::VK_SIZE:
151           Type = ELF::R_X86_64_SIZE32;
152           break;
153         }
154         break;
155       case FK_Data_4:
156         Type = ELF::R_X86_64_32;
157         break;
158       case FK_Data_2: Type = ELF::R_X86_64_16; break;
159       case FK_PCRel_1:
160       case FK_Data_1: Type = ELF::R_X86_64_8; break;
161       }
162     }
163   } else if (getEMachine() == ELF::EM_386) {
164     if (IsPCRel) {
165       switch ((unsigned)Fixup.getKind()) {
166       default: llvm_unreachable("invalid fixup kind!");
167
168       case X86::reloc_global_offset_table:
169         Type = ELF::R_386_GOTPC;
170         break;
171
172       case FK_PCRel_1:
173       case FK_Data_1:
174         switch (Modifier) {
175         default:
176           llvm_unreachable("Unimplemented");
177         case MCSymbolRefExpr::VK_None:
178           Type = ELF::R_386_PC8;
179           break;
180         }
181         break;
182
183       case FK_PCRel_2:
184       case FK_Data_2:
185         switch (Modifier) {
186         default:
187           llvm_unreachable("Unimplemented");
188         case MCSymbolRefExpr::VK_None:
189           Type = ELF::R_386_PC16;
190           break;
191         }
192         break;
193
194       case X86::reloc_riprel_4byte:
195       case X86::reloc_signed_4byte:
196       case FK_PCRel_4:
197       case FK_Data_4:
198         switch (Modifier) {
199         default:
200           llvm_unreachable("Unimplemented");
201         case MCSymbolRefExpr::VK_None:
202           Type = ELF::R_386_PC32;
203           break;
204         case MCSymbolRefExpr::VK_PLT:
205           Type = ELF::R_386_PLT32;
206           break;
207         }
208         break;
209       }
210     } else {
211       switch ((unsigned)Fixup.getKind()) {
212       default: llvm_unreachable("invalid fixup kind!");
213
214       case X86::reloc_global_offset_table:
215         Type = ELF::R_386_GOTPC;
216         break;
217
218       // FIXME: Should we avoid selecting reloc_signed_4byte in 32 bit mode
219       // instead?
220       case X86::reloc_signed_4byte:
221       case FK_PCRel_4:
222       case FK_Data_4:
223         switch (Modifier) {
224         default:
225           llvm_unreachable("Unimplemented");
226         case MCSymbolRefExpr::VK_None:
227           Type = ELF::R_386_32;
228           break;
229         case MCSymbolRefExpr::VK_GOT:
230           Type = ELF::R_386_GOT32;
231           break;
232         case MCSymbolRefExpr::VK_PLT:
233           Type = ELF::R_386_PLT32;
234           break;
235         case MCSymbolRefExpr::VK_GOTOFF:
236           Type = ELF::R_386_GOTOFF;
237           break;
238         case MCSymbolRefExpr::VK_TLSGD:
239           Type = ELF::R_386_TLS_GD;
240           break;
241         case MCSymbolRefExpr::VK_TPOFF:
242           Type = ELF::R_386_TLS_LE_32;
243           break;
244         case MCSymbolRefExpr::VK_INDNTPOFF:
245           Type = ELF::R_386_TLS_IE;
246           break;
247         case MCSymbolRefExpr::VK_NTPOFF:
248           Type = ELF::R_386_TLS_LE;
249           break;
250         case MCSymbolRefExpr::VK_GOTNTPOFF:
251           Type = ELF::R_386_TLS_GOTIE;
252           break;
253         case MCSymbolRefExpr::VK_TLSLDM:
254           Type = ELF::R_386_TLS_LDM;
255           break;
256         case MCSymbolRefExpr::VK_DTPOFF:
257           Type = ELF::R_386_TLS_LDO_32;
258           break;
259         case MCSymbolRefExpr::VK_GOTTPOFF:
260           Type = ELF::R_386_TLS_IE_32;
261           break;
262         }
263         break;
264       case FK_Data_2: Type = ELF::R_386_16; break;
265       case FK_PCRel_1:
266       case FK_Data_1: Type = ELF::R_386_8; break;
267       }
268     }
269   } else
270     llvm_unreachable("Unsupported ELF machine type.");
271
272   return Type;
273 }
274
275 MCObjectWriter *llvm::createX86ELFObjectWriter(raw_ostream &OS,
276                                                bool IsELF64,
277                                                uint8_t OSABI,
278                                                uint16_t EMachine) {
279   MCELFObjectTargetWriter *MOTW =
280     new X86ELFObjectWriter(IsELF64, OSABI, EMachine);
281   return createELFObjectWriter(MOTW, OS,  /*IsLittleEndian=*/true);
282 }