Split the "operand -> symbol" logic from the "get offset and other munging
[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 MCOperand X86ATTAsmPrinter::LowerGlobalAddressOperand(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   // Handle dllimport linkage.
74   if (MO.getTargetFlags() == X86II::MO_DLLIMPORT)
75     Name = "__imp_" + Name;
76   
77   if (MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY ||
78       MO.getTargetFlags() == X86II::MO_DARWIN_NONLAZY_PIC_BASE)
79     GVStubs[Name] = Mang->getMangledName(GV);
80   else if (MO.getTargetFlags() == X86II::MO_DARWIN_HIDDEN_NONLAZY ||
81            MO.getTargetFlags() == X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE)
82     HiddenGVStubs[Name] = Mang->getMangledName(GV);
83   else if (MO.getTargetFlags() == X86II::MO_DARWIN_STUB)
84     FnStubs[Name] = Mang->getMangledName(GV);
85   
86   
87   // Handle target operand flags.
88   // FIXME: This should be common between external symbols, constant pool etc.
89   MCSymbol *NegatedSymbol = 0;
90   
91   switch (MO.getTargetFlags()) {
92     default:
93       llvm_unreachable("Unknown target flag on GV operand");
94     case X86II::MO_NO_FLAG:    // No flag.
95       break;
96     case X86II::MO_DARWIN_NONLAZY:
97     case X86II::MO_DARWIN_HIDDEN_NONLAZY:
98     case X86II::MO_DLLIMPORT:
99     case X86II::MO_DARWIN_STUB:
100       // These affect the name of the symbol, not any suffix.
101       break;
102     case X86II::MO_GOT_ABSOLUTE_ADDRESS:
103       assert(0 && "Reloc mode unimp!");
104       //O << " + [.-";
105       //PrintPICBaseSymbol();
106       //O << ']';
107       break;      
108     case X86II::MO_PIC_BASE_OFFSET:
109     case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
110     case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
111       // Subtract the pic base.
112       NegatedSymbol = GetPICBaseSymbol();
113       break;
114       
115       // FIXME: These probably should be a modifier on the symbol or something??
116     case X86II::MO_TLSGD:     Name += "@TLSGD";     break;
117     case X86II::MO_GOTTPOFF:  Name += "@GOTTPOFF";  break;
118     case X86II::MO_INDNTPOFF: Name += "@INDNTPOFF"; break;
119     case X86II::MO_TPOFF:     Name += "@TPOFF";     break;
120     case X86II::MO_NTPOFF:    Name += "@NTPOFF";    break;
121     case X86II::MO_GOTPCREL:  Name += "@GOTPCREL";  break;
122     case X86II::MO_GOT:       Name += "@GOT";       break;
123     case X86II::MO_GOTOFF:    Name += "@GOTOFF";    break;
124     case X86II::MO_PLT:       Name += "@PLT";       break;
125   }
126   
127   // Create a symbol for the name.
128   MCSymbol *Sym = OutContext.GetOrCreateSymbol(Name);
129   // FIXME: We would like an efficient form for this, so we don't have to do a
130   // lot of extra uniquing.
131   const MCExpr *Expr = MCSymbolRefExpr::Create(Sym, OutContext);
132   if (NegatedSymbol)
133     Expr = MCBinaryExpr::CreateSub(Expr, MCSymbolRefExpr::Create(NegatedSymbol,
134                                                                  OutContext),
135                                    OutContext);
136   if (MO.getOffset())
137     Expr = MCBinaryExpr::CreateAdd(Expr, MCConstantExpr::Create(MO.getOffset(),
138                                                                 OutContext),
139                                    OutContext);
140   return MCOperand::CreateExpr(Expr);
141 }
142
143 MCSymbol *X86ATTAsmPrinter::GetExternalSymbolSymbol(const MachineOperand &MO) {
144   std::string Name = Mang->makeNameProper(MO.getSymbolName());
145   if (MO.getTargetFlags() == X86II::MO_DARWIN_STUB) {
146     FnStubs[Name+"$stub"] = Name;
147     Name += "$stub";
148   }
149   
150   return OutContext.GetOrCreateSymbol(Name);
151 }
152
153 MCSymbol *X86ATTAsmPrinter::GetJumpTableSymbol(const MachineOperand &MO) {
154   SmallString<256> Name;
155   raw_svector_ostream(Name) << MAI->getPrivateGlobalPrefix() << "JTI"
156     << getFunctionNumber() << '_' << MO.getIndex();
157   
158   MCSymbol *NegatedSymbol = 0;
159   switch (MO.getTargetFlags()) {
160   default:
161     llvm_unreachable("Unknown target flag on GV operand");
162   case X86II::MO_NO_FLAG:    // No flag.
163   case X86II::MO_PIC_BASE_OFFSET:
164   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
165   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
166     break;
167     // Subtract the pic base.
168     NegatedSymbol = GetPICBaseSymbol();
169     break;
170   }
171   
172   // Create a symbol for the name.
173   return OutContext.GetOrCreateSymbol(Name.str());
174 }
175
176
177 MCSymbol *X86ATTAsmPrinter::
178 GetConstantPoolIndexSymbol(const MachineOperand &MO) {
179   SmallString<256> Name;
180   raw_svector_ostream(Name) << MAI->getPrivateGlobalPrefix() << "CPI"
181   << getFunctionNumber() << '_' << MO.getIndex();
182   
183   switch (MO.getTargetFlags()) {
184   default:
185     llvm_unreachable("Unknown target flag on GV operand");
186   case X86II::MO_NO_FLAG:    // No flag.
187   case X86II::MO_PIC_BASE_OFFSET:
188   case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
189   case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
190     break;
191   }
192   
193   // Create a symbol for the name.
194   return OutContext.GetOrCreateSymbol(Name.str());
195 }
196
197 MCOperand X86ATTAsmPrinter::LowerSymbolOperand(const MachineOperand &MO,
198                                                MCSymbol *Sym) {
199   MCSymbol *NegatedSymbol = 0;
200   switch (MO.getTargetFlags()) {
201     default:
202       llvm_unreachable("Unknown target flag on GV operand");
203     case X86II::MO_NO_FLAG:    // No flag.
204       break;
205     case X86II::MO_PIC_BASE_OFFSET:
206     case X86II::MO_DARWIN_NONLAZY_PIC_BASE:
207     case X86II::MO_DARWIN_HIDDEN_NONLAZY_PIC_BASE:
208       // Subtract the pic base.
209       NegatedSymbol = GetPICBaseSymbol();
210       break;
211   }
212   
213   // FIXME: We would like an efficient form for this, so we don't have to do a
214   // lot of extra uniquing.
215   const MCExpr *Expr = MCSymbolRefExpr::Create(Sym, OutContext);
216   if (NegatedSymbol)
217     Expr = MCBinaryExpr::CreateSub(Expr, MCSymbolRefExpr::Create(NegatedSymbol,
218                                                                  OutContext),
219                                    OutContext);
220   return MCOperand::CreateExpr(Expr);
221 }
222
223
224 void X86ATTAsmPrinter::
225 printInstructionThroughMCStreamer(const MachineInstr *MI) {
226   
227   MCInst TmpInst;
228   
229   switch (MI->getOpcode()) {
230   case TargetInstrInfo::DBG_LABEL:
231   case TargetInstrInfo::EH_LABEL:
232   case TargetInstrInfo::GC_LABEL:
233     printLabel(MI);
234     return;
235   case TargetInstrInfo::INLINEASM:
236     O << '\t';
237     printInlineAsm(MI);
238     return;
239   case TargetInstrInfo::IMPLICIT_DEF:
240     printImplicitDef(MI);
241     return;
242   case X86::MOVPC32r: {
243     // This is a pseudo op for a two instruction sequence with a label, which
244     // looks like:
245     //     call "L1$pb"
246     // "L1$pb":
247     //     popl %esi
248     
249     // Emit the call.
250     MCSymbol *PICBase = GetPICBaseSymbol();
251     TmpInst.setOpcode(X86::CALLpcrel32);
252     // FIXME: We would like an efficient form for this, so we don't have to do a
253     // lot of extra uniquing.
254     TmpInst.addOperand(MCOperand::CreateExpr(MCSymbolRefExpr::Create(PICBase,
255                                                                  OutContext)));
256     printInstruction(&TmpInst);
257     
258     // Emit the label.
259     OutStreamer.EmitLabel(PICBase);
260     
261     // popl $reg
262     TmpInst.setOpcode(X86::POP32r);
263     TmpInst.getOperand(0) = MCOperand::CreateReg(MI->getOperand(0).getReg());
264     printInstruction(&TmpInst);
265     return;
266     }
267   }
268   
269   TmpInst.setOpcode(MI->getOpcode());
270   
271   for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) {
272     const MachineOperand &MO = MI->getOperand(i);
273     
274     MCOperand MCOp;
275     switch (MO.getType()) {
276     default:
277       O.flush();
278       errs() << "Cannot lower operand #" << i << " of :" << *MI;
279       llvm_unreachable("Unimp");
280     case MachineOperand::MO_Register:
281       MCOp = MCOperand::CreateReg(MO.getReg());
282       break;
283     case MachineOperand::MO_Immediate:
284       MCOp = MCOperand::CreateImm(MO.getImm());
285       break;
286     case MachineOperand::MO_MachineBasicBlock:
287       MCOp = MCOperand::CreateMBBLabel(getFunctionNumber(), 
288                                        MO.getMBB()->getNumber());
289       break;
290     case MachineOperand::MO_GlobalAddress:
291       MCOp = LowerGlobalAddressOperand(MO);
292       break;
293     case MachineOperand::MO_ExternalSymbol:
294       MCOp = LowerSymbolOperand(MO, GetExternalSymbolSymbol(MO));
295       break;
296     case MachineOperand::MO_JumpTableIndex:
297       MCOp = LowerSymbolOperand(MO, GetJumpTableSymbol(MO));
298       break;
299     case MachineOperand::MO_ConstantPoolIndex:
300       MCOp = LowerSymbolOperand(MO, GetConstantPoolIndexSymbol(MO));
301       break;
302     }
303     
304     TmpInst.addOperand(MCOp);
305   }
306   
307   switch (TmpInst.getOpcode()) {
308   case X86::LEA64_32r:
309     // Handle the 'subreg rewriting' for the lea64_32mem operand.
310     lower_lea64_32mem(&TmpInst, 1);
311     break;
312   }
313   
314   printInstruction(&TmpInst);
315 }