merge globaladdress symbol processing stuff into other stuff. Now
[oota-llvm.git] / lib / Target / X86 / AsmPrinter / X86MCInstLower.cpp
1 //===-- X86MCInstLower.cpp - Convert X86 MachineInstr to an MCInst --------===//
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 code to lower X86 MachineInstrs to their corresponding
11 // MCInst records.
12 //
13 //===----------------------------------------------------------------------===//
14
15
16 #include "X86ATTAsmPrinter.h"
17 #include "X86MCAsmInfo.h"
18 #include "llvm/MC/MCContext.h"
19 #include "llvm/MC/MCExpr.h"
20 #include "llvm/MC/MCInst.h"
21 #include "llvm/MC/MCStreamer.h"
22 #include "llvm/Support/FormattedStream.h"
23 #include "llvm/Support/Mangler.h"
24 #include "llvm/ADT/SmallString.h"
25 using namespace llvm;
26
27 MCSymbol *X86ATTAsmPrinter::GetPICBaseSymbol() {
28   // FIXME: the actual label generated doesn't matter here!  Just mangle in
29   // something unique (the function number) with Private prefix.
30   SmallString<60> Name;
31   
32   if (Subtarget->isTargetDarwin()) {
33     raw_svector_ostream(Name) << 'L' << getFunctionNumber() << "$pb";
34   } else {
35     assert(Subtarget->isTargetELF() && "Don't know how to print PIC label!");
36     raw_svector_ostream(Name) << ".Lllvm$" << getFunctionNumber()<<".$piclabel";
37   }
38   return OutContext.GetOrCreateSymbol(Name.str());
39 }
40
41
42
43 static void lower_lea64_32mem(MCInst *MI, unsigned OpNo) {
44   // Convert registers in the addr mode according to subreg64.
45   for (unsigned i = 0; i != 4; ++i) {
46     if (!MI->getOperand(i).isReg()) continue;
47     
48     unsigned Reg = MI->getOperand(i).getReg();
49     if (Reg == 0) continue;
50     
51     MI->getOperand(i).setReg(getX86SubSuperRegister(Reg, MVT::i64));
52   }
53 }
54
55 /// LowerGlobalAddressOperand - Lower an MO_GlobalAddress operand to an
56 /// MCOperand.
57 MCSymbol *X86ATTAsmPrinter::GetGlobalAddressSymbol(const MachineOperand &MO) {
58   const GlobalValue *GV = MO.getGlobal();
59   
60   const char *Suffix = "";
61   if (MO.getTargetFlags() == X86II::MO_DARWIN_STUB)
62     Suffix = "$stub";
63   else if (MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY ||
64            MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY_PIC_BASE ||
65            MO.getTargetFlags() == X86II::MO_DARWIN_HIDDEN_NONLAZY ||
66            MO.getTargetFlags() == X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE)
67     Suffix = "$non_lazy_ptr";
68   
69   std::string Name = Mang->getMangledName(GV, Suffix, Suffix[0] != '\0');
70   if (Subtarget->isTargetCygMing())
71     DecorateCygMingName(Name, GV);
72   
73   switch (MO.getTargetFlags()) {
74   default: llvm_unreachable("Unknown target flag on GV operand");
75   case X86II::MO_NO_FLAG:                // No flag.
76   case X86II::MO_GOT_ABSOLUTE_ADDRESS:   // Doesn't modify symbol name.
77   case X86II::MO_PIC_BASE_OFFSET:        // Doesn't modify symbol name.
78     break;
79   case X86II::MO_DLLIMPORT:
80     // Handle dllimport linkage.
81     Name = "__imp_" + Name;
82     break;
83   case X86II::MO_DARWIN_NONLAZY:
84   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
85     GVStubs[Name] = Mang->getMangledName(GV);
86     break;
87   case X86II::MO_DARWIN_HIDDEN_NONLAZY:
88   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
89     HiddenGVStubs[Name] = Mang->getMangledName(GV);
90     break;
91   case X86II::MO_DARWIN_STUB:
92     FnStubs[Name] = Mang->getMangledName(GV);
93     break;
94   // FIXME: These probably should be a modifier on the symbol or something??
95   case X86II::MO_TLSGD:     Name += "@TLSGD";     break;
96   case X86II::MO_GOTTPOFF:  Name += "@GOTTPOFF";  break;
97   case X86II::MO_INDNTPOFF: Name += "@INDNTPOFF"; break;
98   case X86II::MO_TPOFF:     Name += "@TPOFF";     break;
99   case X86II::MO_NTPOFF:    Name += "@NTPOFF";    break;
100   case X86II::MO_GOTPCREL:  Name += "@GOTPCREL";  break;
101   case X86II::MO_GOT:       Name += "@GOT";       break;
102   case X86II::MO_GOTOFF:    Name += "@GOTOFF";    break;
103   case X86II::MO_PLT:       Name += "@PLT";       break;
104   }
105   
106   return OutContext.GetOrCreateSymbol(Name);
107 }
108
109 MCSymbol *X86ATTAsmPrinter::GetExternalSymbolSymbol(const MachineOperand &MO) {
110   std::string Name = Mang->makeNameProper(MO.getSymbolName());
111   if (MO.getTargetFlags() == X86II::MO_DARWIN_STUB) {
112     FnStubs[Name+"$stub"] = Name;
113     Name += "$stub";
114   }
115   
116   return OutContext.GetOrCreateSymbol(Name);
117 }
118
119 MCSymbol *X86ATTAsmPrinter::GetJumpTableSymbol(const MachineOperand &MO) {
120   SmallString<256> Name;
121   raw_svector_ostream(Name) << MAI->getPrivateGlobalPrefix() << "JTI"
122     << getFunctionNumber() << '_' << MO.getIndex();
123   
124   MCSymbol *NegatedSymbol = 0;
125   switch (MO.getTargetFlags()) {
126   default:
127     llvm_unreachable("Unknown target flag on GV operand");
128   case X86II::MO_NO_FLAG:    // No flag.
129   case X86II::MO_PIC_BASE_OFFSET:
130   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
131   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
132     break;
133     // Subtract the pic base.
134     NegatedSymbol = GetPICBaseSymbol();
135     break;
136   }
137   
138   // Create a symbol for the name.
139   return OutContext.GetOrCreateSymbol(Name.str());
140 }
141
142
143 MCSymbol *X86ATTAsmPrinter::
144 GetConstantPoolIndexSymbol(const MachineOperand &MO) {
145   SmallString<256> Name;
146   raw_svector_ostream(Name) << MAI->getPrivateGlobalPrefix() << "CPI"
147   << getFunctionNumber() << '_' << MO.getIndex();
148   
149   switch (MO.getTargetFlags()) {
150   default:
151     llvm_unreachable("Unknown target flag on GV operand");
152   case X86II::MO_NO_FLAG:    // No flag.
153   case X86II::MO_PIC_BASE_OFFSET:
154   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
155   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
156     break;
157   }
158   
159   // Create a symbol for the name.
160   return OutContext.GetOrCreateSymbol(Name.str());
161 }
162
163 MCOperand X86ATTAsmPrinter::LowerSymbolOperand(const MachineOperand &MO,
164                                                MCSymbol *Sym) {
165   MCSymbol *NegatedSymbol = 0;
166   switch (MO.getTargetFlags()) {
167   default: llvm_unreachable("Unknown target flag on GV operand");
168   case X86II::MO_NO_FLAG:    // No flag.
169       
170   // These affect the name of the symbol, not any suffix.
171   case X86II::MO_DARWIN_NONLAZY:
172   case X86II::MO_DARWIN_HIDDEN_NONLAZY:
173   case X86II::MO_DLLIMPORT:
174   case X86II::MO_DARWIN_STUB:
175   case X86II::MO_TLSGD:
176   case X86II::MO_GOTTPOFF:
177   case X86II::MO_INDNTPOFF:
178   case X86II::MO_TPOFF:
179   case X86II::MO_NTPOFF:
180   case X86II::MO_GOTPCREL:
181   case X86II::MO_GOT:
182   case X86II::MO_GOTOFF:
183   case X86II::MO_PLT:
184     break;
185   case X86II::MO_PIC_BASE_OFFSET:
186   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
187   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
188     // Subtract the pic base.
189     NegatedSymbol = GetPICBaseSymbol();
190     break;
191   case X86II::MO_GOT_ABSOLUTE_ADDRESS:
192     assert(0 && "Reloc mode unimp!");
193     //O << " + [.-";
194     //PrintPICBaseSymbol();
195     //O << ']';
196     break;      
197   }
198   
199   // FIXME: We would like an efficient form for this, so we don't have to do a
200   // lot of extra uniquing.
201   const MCExpr *Expr = MCSymbolRefExpr::Create(Sym, OutContext);
202   if (NegatedSymbol)
203     Expr = MCBinaryExpr::CreateSub(Expr, MCSymbolRefExpr::Create(NegatedSymbol,
204                                                                  OutContext),
205                                    OutContext);
206   if (!MO.isSymbol() && MO.getOffset())
207     Expr = MCBinaryExpr::CreateAdd(Expr, MCConstantExpr::Create(MO.getOffset(),
208                                                                 OutContext),
209                                    OutContext);
210   return MCOperand::CreateExpr(Expr);
211 }
212
213 void X86ATTAsmPrinter::
214 printInstructionThroughMCStreamer(const MachineInstr *MI) {
215   
216   MCInst TmpInst;
217   
218   switch (MI->getOpcode()) {
219   case TargetInstrInfo::DBG_LABEL:
220   case TargetInstrInfo::EH_LABEL:
221   case TargetInstrInfo::GC_LABEL:
222     printLabel(MI);
223     return;
224   case TargetInstrInfo::INLINEASM:
225     O << '\t';
226     printInlineAsm(MI);
227     return;
228   case TargetInstrInfo::IMPLICIT_DEF:
229     printImplicitDef(MI);
230     return;
231   case X86::MOVPC32r: {
232     // This is a pseudo op for a two instruction sequence with a label, which
233     // looks like:
234     //     call "L1$pb"
235     // "L1$pb":
236     //     popl %esi
237     
238     // Emit the call.
239     MCSymbol *PICBase = GetPICBaseSymbol();
240     TmpInst.setOpcode(X86::CALLpcrel32);
241     // FIXME: We would like an efficient form for this, so we don't have to do a
242     // lot of extra uniquing.
243     TmpInst.addOperand(MCOperand::CreateExpr(MCSymbolRefExpr::Create(PICBase,
244                                                                  OutContext)));
245     printInstruction(&TmpInst);
246     
247     // Emit the label.
248     OutStreamer.EmitLabel(PICBase);
249     
250     // popl $reg
251     TmpInst.setOpcode(X86::POP32r);
252     TmpInst.getOperand(0) = MCOperand::CreateReg(MI->getOperand(0).getReg());
253     printInstruction(&TmpInst);
254     return;
255     }
256   }
257   
258   TmpInst.setOpcode(MI->getOpcode());
259   
260   for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
261     const MachineOperand &MO = MI->getOperand(i);
262     
263     MCOperand MCOp;
264     switch (MO.getType()) {
265     default:
266       O.flush();
267       errs() << "Cannot lower operand #" << i << " of :" << *MI;
268       llvm_unreachable("Unimp");
269     case MachineOperand::MO_Register:
270       MCOp = MCOperand::CreateReg(MO.getReg());
271       break;
272     case MachineOperand::MO_Immediate:
273       MCOp = MCOperand::CreateImm(MO.getImm());
274       break;
275     case MachineOperand::MO_MachineBasicBlock:
276       MCOp = MCOperand::CreateMBBLabel(getFunctionNumber(), 
277                                        MO.getMBB()->getNumber());
278       break;
279     case MachineOperand::MO_GlobalAddress:
280       MCOp = LowerSymbolOperand(MO, GetGlobalAddressSymbol(MO));
281       break;
282     case MachineOperand::MO_ExternalSymbol:
283       MCOp = LowerSymbolOperand(MO, GetExternalSymbolSymbol(MO));
284       break;
285     case MachineOperand::MO_JumpTableIndex:
286       MCOp = LowerSymbolOperand(MO, GetJumpTableSymbol(MO));
287       break;
288     case MachineOperand::MO_ConstantPoolIndex:
289       MCOp = LowerSymbolOperand(MO, GetConstantPoolIndexSymbol(MO));
290       break;
291     }
292     
293     TmpInst.addOperand(MCOp);
294   }
295   
296   switch (TmpInst.getOpcode()) {
297   case X86::LEA64_32r:
298     // Handle the 'subreg rewriting' for the lea64_32mem operand.
299     lower_lea64_32mem(&TmpInst, 1);
300     break;
301   }
302   
303   printInstruction(&TmpInst);
304 }