Cleanup relocation sorting for ELF.
[oota-llvm.git] / lib / Target / PowerPC / MCTargetDesc / PPCELFObjectWriter.cpp
1 //===-- PPCELFObjectWriter.cpp - PPC 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/PPCMCTargetDesc.h"
11 #include "MCTargetDesc/PPCFixupKinds.h"
12 #include "llvm/ADT/STLExtras.h"
13 #include "llvm/MC/MCELFObjectWriter.h"
14 #include "llvm/MC/MCExpr.h"
15 #include "llvm/MC/MCValue.h"
16 #include "llvm/Support/ErrorHandling.h"
17
18 using namespace llvm;
19
20 namespace {
21   class PPCELFObjectWriter : public MCELFObjectTargetWriter {
22   public:
23     PPCELFObjectWriter(bool Is64Bit, uint8_t OSABI);
24
25     virtual ~PPCELFObjectWriter();
26   protected:
27     virtual unsigned getRelocTypeInner(const MCValue &Target,
28                                        const MCFixup &Fixup,
29                                        bool IsPCRel) const;
30     virtual unsigned GetRelocType(const MCValue &Target, const MCFixup &Fixup,
31                                   bool IsPCRel, bool IsRelocWithSymbol,
32                                   int64_t Addend) const;
33     virtual const MCSymbol *undefinedExplicitRelSym(const MCValue &Target,
34                                                     const MCFixup &Fixup,
35                                                     bool IsPCRel) const;
36   };
37 }
38
39 PPCELFObjectWriter::PPCELFObjectWriter(bool Is64Bit, uint8_t OSABI)
40   : MCELFObjectTargetWriter(Is64Bit, OSABI,
41                             Is64Bit ?  ELF::EM_PPC64 : ELF::EM_PPC,
42                             /*HasRelocationAddend*/ true) {}
43
44 PPCELFObjectWriter::~PPCELFObjectWriter() {
45 }
46
47 unsigned PPCELFObjectWriter::getRelocTypeInner(const MCValue &Target,
48                                                const MCFixup &Fixup,
49                                                bool IsPCRel) const
50 {
51   MCSymbolRefExpr::VariantKind Modifier = Target.isAbsolute() ?
52     MCSymbolRefExpr::VK_None : Target.getSymA()->getKind();
53
54   // determine the type of the relocation
55   unsigned Type;
56   if (IsPCRel) {
57     switch ((unsigned)Fixup.getKind()) {
58     default:
59       llvm_unreachable("Unimplemented");
60     case PPC::fixup_ppc_br24:
61       Type = ELF::R_PPC_REL24;
62       break;
63     case PPC::fixup_ppc_brcond14:
64       Type = ELF::R_PPC_REL14;
65       break;
66     case FK_Data_4:
67     case FK_PCRel_4:
68       Type = ELF::R_PPC_REL32;
69       break;
70     case FK_Data_8:
71     case FK_PCRel_8:
72       Type = ELF::R_PPC64_REL64;
73       break;
74     }
75   } else {
76     switch ((unsigned)Fixup.getKind()) {
77       default: llvm_unreachable("invalid fixup kind!");
78     case PPC::fixup_ppc_br24:
79       Type = ELF::R_PPC_ADDR24;
80       break;
81     case PPC::fixup_ppc_brcond14:
82       Type = ELF::R_PPC_ADDR14; // XXX: or BRNTAKEN?_
83       break;
84     case PPC::fixup_ppc_ha16:
85       switch (Modifier) {
86       default: llvm_unreachable("Unsupported Modifier");
87       case MCSymbolRefExpr::VK_PPC_TPREL16_HA:
88         Type = ELF::R_PPC_TPREL16_HA;
89         break;
90       case MCSymbolRefExpr::VK_PPC_DTPREL16_HA:
91         Type = ELF::R_PPC64_DTPREL16_HA;
92         break;
93       case MCSymbolRefExpr::VK_PPC_GAS_HA16:
94       case MCSymbolRefExpr::VK_PPC_DARWIN_HA16:
95         Type = ELF::R_PPC_ADDR16_HA;
96         break;
97       case MCSymbolRefExpr::VK_PPC_TOC16_HA:
98         Type = ELF::R_PPC64_TOC16_HA;
99         break;
100       case MCSymbolRefExpr::VK_PPC_GOT_TPREL16_HA:
101         Type = ELF::R_PPC64_GOT_TPREL16_HA;
102         break;
103       case MCSymbolRefExpr::VK_PPC_GOT_TLSGD16_HA:
104         Type = ELF::R_PPC64_GOT_TLSGD16_HA;
105         break;
106       case MCSymbolRefExpr::VK_PPC_GOT_TLSLD16_HA:
107         Type = ELF::R_PPC64_GOT_TLSLD16_HA;
108         break;
109       }
110       break;
111     case PPC::fixup_ppc_lo16:
112       switch (Modifier) {
113       default: llvm_unreachable("Unsupported Modifier");
114       case MCSymbolRefExpr::VK_PPC_TPREL16_LO:
115         Type = ELF::R_PPC_TPREL16_LO;
116         break;
117       case MCSymbolRefExpr::VK_PPC_DTPREL16_LO:
118         Type = ELF::R_PPC64_DTPREL16_LO;
119         break;
120       case MCSymbolRefExpr::VK_None:
121         Type = ELF::R_PPC_ADDR16;
122         break;
123       case MCSymbolRefExpr::VK_PPC_GAS_LO16:
124       case MCSymbolRefExpr::VK_PPC_DARWIN_LO16:
125         Type = ELF::R_PPC_ADDR16_LO;
126         break;
127       case MCSymbolRefExpr::VK_PPC_TOC_ENTRY:
128         Type = ELF::R_PPC64_TOC16;
129         break;
130       case MCSymbolRefExpr::VK_PPC_TOC16_LO:
131         Type = ELF::R_PPC64_TOC16_LO;
132         break;
133       case MCSymbolRefExpr::VK_PPC_GOT_TLSGD16_LO:
134         Type = ELF::R_PPC64_GOT_TLSGD16_LO;
135         break;
136       case MCSymbolRefExpr::VK_PPC_GOT_TLSLD16_LO:
137         Type = ELF::R_PPC64_GOT_TLSLD16_LO;
138         break;
139       }
140       break;
141     case PPC::fixup_ppc_lo16_ds:
142       switch (Modifier) {
143       default: llvm_unreachable("Unsupported Modifier");
144       case MCSymbolRefExpr::VK_None:
145         Type = ELF::R_PPC64_ADDR16_DS;
146         break;
147       case MCSymbolRefExpr::VK_PPC_GAS_LO16:
148       case MCSymbolRefExpr::VK_PPC_DARWIN_LO16:
149         Type = ELF::R_PPC64_ADDR16_LO_DS;
150         break;
151       case MCSymbolRefExpr::VK_PPC_TOC_ENTRY:
152         Type = ELF::R_PPC64_TOC16_DS;
153         break;
154       case MCSymbolRefExpr::VK_PPC_TOC16_LO:
155         Type = ELF::R_PPC64_TOC16_LO_DS;
156         break;
157       case MCSymbolRefExpr::VK_PPC_GOT_TPREL16_LO:
158         Type = ELF::R_PPC64_GOT_TPREL16_LO_DS;
159         break;
160       }
161       break;
162     case PPC::fixup_ppc_tlsreg:
163       Type = ELF::R_PPC64_TLS;
164       break;
165     case PPC::fixup_ppc_nofixup:
166       switch (Modifier) {
167       default: llvm_unreachable("Unsupported Modifier");
168       case MCSymbolRefExpr::VK_PPC_TLSGD:
169         Type = ELF::R_PPC64_TLSGD;
170         break;
171       case MCSymbolRefExpr::VK_PPC_TLSLD:
172         Type = ELF::R_PPC64_TLSLD;
173         break;
174       }
175       break;
176     case FK_Data_8:
177       switch (Modifier) {
178       default: llvm_unreachable("Unsupported Modifier");
179       case MCSymbolRefExpr::VK_PPC_TOC:
180         Type = ELF::R_PPC64_TOC;
181         break;
182       case MCSymbolRefExpr::VK_None:
183         Type = ELF::R_PPC64_ADDR64;
184         break;
185       }
186       break;
187     case FK_Data_4:
188       Type = ELF::R_PPC_ADDR32;
189       break;
190     case FK_Data_2:
191       Type = ELF::R_PPC_ADDR16;
192       break;
193     }
194   }
195   return Type;
196 }
197
198 unsigned PPCELFObjectWriter::GetRelocType(const MCValue &Target,
199                                           const MCFixup &Fixup,
200                                           bool IsPCRel,
201                                           bool IsRelocWithSymbol,
202                                           int64_t Addend) const {
203   return getRelocTypeInner(Target, Fixup, IsPCRel);
204 }
205
206 const MCSymbol *PPCELFObjectWriter::undefinedExplicitRelSym(const MCValue &Target,
207                                                             const MCFixup &Fixup,
208                                                             bool IsPCRel) const {
209   assert(Target.getSymA() && "SymA cannot be 0");
210   const MCSymbol &Symbol = Target.getSymA()->getSymbol().AliasedSymbol();
211
212   unsigned RelocType = getRelocTypeInner(Target, Fixup, IsPCRel);
213
214   // The .odp creation emits a relocation against the symbol ".TOC." which
215   // create a R_PPC64_TOC relocation. However the relocation symbol name
216   // in final object creation should be NULL, since the symbol does not
217   // really exist, it is just the reference to TOC base for the current
218   // object file.
219   bool EmitThisSym = RelocType != ELF::R_PPC64_TOC;
220
221   if (EmitThisSym && !Symbol.isTemporary())
222     return &Symbol;
223   return NULL;
224 }
225
226 MCObjectWriter *llvm::createPPCELFObjectWriter(raw_ostream &OS,
227                                                bool Is64Bit,
228                                                uint8_t OSABI) {
229   MCELFObjectTargetWriter *MOTW = new PPCELFObjectWriter(Is64Bit, OSABI);
230   return createELFObjectWriter(MOTW, OS,  /*IsLittleEndian=*/false);
231 }