[MC][Target] Implement support for R_X86_64_SIZE{32,64}.
[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_signed_4byte:
195       case FK_PCRel_4:
196       case FK_Data_4:
197         switch (Modifier) {
198         default:
199           llvm_unreachable("Unimplemented");
200         case MCSymbolRefExpr::VK_None:
201           Type = ELF::R_386_PC32;
202           break;
203         case MCSymbolRefExpr::VK_PLT:
204           Type = ELF::R_386_PLT32;
205           break;
206         }
207         break;
208       }
209     } else {
210       switch ((unsigned)Fixup.getKind()) {
211       default: llvm_unreachable("invalid fixup kind!");
212
213       case X86::reloc_global_offset_table:
214         Type = ELF::R_386_GOTPC;
215         break;
216
217       // FIXME: Should we avoid selecting reloc_signed_4byte in 32 bit mode
218       // instead?
219       case X86::reloc_signed_4byte:
220       case FK_PCRel_4:
221       case FK_Data_4:
222         switch (Modifier) {
223         default:
224           llvm_unreachable("Unimplemented");
225         case MCSymbolRefExpr::VK_None:
226           Type = ELF::R_386_32;
227           break;
228         case MCSymbolRefExpr::VK_GOT:
229           Type = ELF::R_386_GOT32;
230           break;
231         case MCSymbolRefExpr::VK_PLT:
232           Type = ELF::R_386_PLT32;
233           break;
234         case MCSymbolRefExpr::VK_GOTOFF:
235           Type = ELF::R_386_GOTOFF;
236           break;
237         case MCSymbolRefExpr::VK_TLSGD:
238           Type = ELF::R_386_TLS_GD;
239           break;
240         case MCSymbolRefExpr::VK_TPOFF:
241           Type = ELF::R_386_TLS_LE_32;
242           break;
243         case MCSymbolRefExpr::VK_INDNTPOFF:
244           Type = ELF::R_386_TLS_IE;
245           break;
246         case MCSymbolRefExpr::VK_NTPOFF:
247           Type = ELF::R_386_TLS_LE;
248           break;
249         case MCSymbolRefExpr::VK_GOTNTPOFF:
250           Type = ELF::R_386_TLS_GOTIE;
251           break;
252         case MCSymbolRefExpr::VK_TLSLDM:
253           Type = ELF::R_386_TLS_LDM;
254           break;
255         case MCSymbolRefExpr::VK_DTPOFF:
256           Type = ELF::R_386_TLS_LDO_32;
257           break;
258         case MCSymbolRefExpr::VK_GOTTPOFF:
259           Type = ELF::R_386_TLS_IE_32;
260           break;
261         }
262         break;
263       case FK_Data_2: Type = ELF::R_386_16; break;
264       case FK_PCRel_1:
265       case FK_Data_1: Type = ELF::R_386_8; break;
266       }
267     }
268   } else
269     llvm_unreachable("Unsupported ELF machine type.");
270
271   return Type;
272 }
273
274 MCObjectWriter *llvm::createX86ELFObjectWriter(raw_ostream &OS,
275                                                bool IsELF64,
276                                                uint8_t OSABI,
277                                                uint16_t EMachine) {
278   MCELFObjectTargetWriter *MOTW =
279     new X86ELFObjectWriter(IsELF64, OSABI, EMachine);
280   return createELFObjectWriter(MOTW, OS,  /*IsLittleEndian=*/true);
281 }