Make sure debug info hook gets called when emitting
[oota-llvm.git] / lib / Target / PowerPC / AsmPrinter / PPCAsmPrinter.cpp
1 //===-- PPCAsmPrinter.cpp - Print machine instrs to PowerPC assembly --------=//
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 a printer that converts from our internal representation
11 // of machine-dependent LLVM code to PowerPC assembly language. This printer is
12 // the output mechanism used by `llc'.
13 //
14 // Documentation at http://developer.apple.com/documentation/DeveloperTools/
15 // Reference/Assembler/ASMIntroduction/chapter_1_section_1.html
16 //
17 //===----------------------------------------------------------------------===//
18
19 #define DEBUG_TYPE "asmprinter"
20 #include "PPC.h"
21 #include "PPCPredicates.h"
22 #include "PPCTargetMachine.h"
23 #include "PPCSubtarget.h"
24 #include "llvm/Constants.h"
25 #include "llvm/DerivedTypes.h"
26 #include "llvm/Module.h"
27 #include "llvm/Assembly/Writer.h"
28 #include "llvm/CodeGen/AsmPrinter.h"
29 #include "llvm/CodeGen/DwarfWriter.h"
30 #include "llvm/CodeGen/MachineModuleInfo.h"
31 #include "llvm/CodeGen/MachineFunctionPass.h"
32 #include "llvm/CodeGen/MachineInstr.h"
33 #include "llvm/CodeGen/MachineInstrBuilder.h"
34 #include "llvm/MC/MCAsmInfo.h"
35 #include "llvm/MC/MCSectionMachO.h"
36 #include "llvm/MC/MCStreamer.h"
37 #include "llvm/MC/MCSymbol.h"
38 #include "llvm/Target/TargetLoweringObjectFile.h"
39 #include "llvm/Target/TargetRegisterInfo.h"
40 #include "llvm/Target/TargetInstrInfo.h"
41 #include "llvm/Target/TargetOptions.h"
42 #include "llvm/Target/TargetRegistry.h"
43 #include "llvm/Support/Mangler.h"
44 #include "llvm/Support/MathExtras.h"
45 #include "llvm/Support/CommandLine.h"
46 #include "llvm/Support/Debug.h"
47 #include "llvm/Support/ErrorHandling.h"
48 #include "llvm/Support/FormattedStream.h"
49 #include "llvm/ADT/Statistic.h"
50 #include "llvm/ADT/StringExtras.h"
51 #include "llvm/ADT/StringSet.h"
52 using namespace llvm;
53
54 STATISTIC(EmittedInsts, "Number of machine instrs printed");
55
56 namespace {
57   class PPCAsmPrinter : public AsmPrinter {
58   protected:
59     struct FnStubInfo {
60       std::string Stub, LazyPtr, AnonSymbol;
61       
62       FnStubInfo() {}
63       
64       void Init(const GlobalValue *GV, Mangler *Mang) {
65         // Already initialized.
66         if (!Stub.empty()) return;
67         Stub = Mang->getMangledName(GV, "$stub", true);
68         LazyPtr = Mang->getMangledName(GV, "$lazy_ptr", true);
69         AnonSymbol = Mang->getMangledName(GV, "$stub$tmp", true);
70       }
71
72       void Init(const std::string &GV, Mangler *Mang) {
73         // Already initialized.
74         if (!Stub.empty()) return;
75         Stub = Mang->makeNameProper(GV + "$stub",
76                                     Mangler::Private);
77         LazyPtr = Mang->makeNameProper(GV + "$lazy_ptr",
78                                        Mangler::Private);
79         AnonSymbol = Mang->makeNameProper(GV + "$stub$tmp",
80                                           Mangler::Private);
81       }
82     };
83     
84     StringMap<FnStubInfo> FnStubs;
85     StringMap<std::string> GVStubs, HiddenGVStubs, TOC;
86     const PPCSubtarget &Subtarget;
87     uint64_t LabelID;
88   public:
89     explicit PPCAsmPrinter(formatted_raw_ostream &O, TargetMachine &TM,
90                            const MCAsmInfo *T, bool V)
91       : AsmPrinter(O, TM, T, V),
92         Subtarget(TM.getSubtarget<PPCSubtarget>()), LabelID(0) {}
93
94     virtual const char *getPassName() const {
95       return "PowerPC Assembly Printer";
96     }
97
98     PPCTargetMachine &getTM() {
99       return static_cast<PPCTargetMachine&>(TM);
100     }
101
102     unsigned enumRegToMachineReg(unsigned enumReg) {
103       switch (enumReg) {
104       default: llvm_unreachable("Unhandled register!");
105       case PPC::CR0:  return  0;
106       case PPC::CR1:  return  1;
107       case PPC::CR2:  return  2;
108       case PPC::CR3:  return  3;
109       case PPC::CR4:  return  4;
110       case PPC::CR5:  return  5;
111       case PPC::CR6:  return  6;
112       case PPC::CR7:  return  7;
113       }
114       llvm_unreachable(0);
115     }
116
117     /// printInstruction - This method is automatically generated by tablegen
118     /// from the instruction set description.  This method returns true if the
119     /// machine instruction was sufficiently described to print it, otherwise it
120     /// returns false.
121     void printInstruction(const MachineInstr *MI);
122     static const char *getRegisterName(unsigned RegNo);
123
124
125     void printMachineInstruction(const MachineInstr *MI);
126     void printOp(const MachineOperand &MO);
127
128     /// stripRegisterPrefix - This method strips the character prefix from a
129     /// register name so that only the number is left.  Used by for linux asm.
130     const char *stripRegisterPrefix(const char *RegName) {
131       switch (RegName[0]) {
132       case 'r':
133       case 'f':
134       case 'v': return RegName + 1;
135       case 'c': if (RegName[1] == 'r') return RegName + 2;
136       }
137
138       return RegName;
139     }
140
141     /// printRegister - Print register according to target requirements.
142     ///
143     void printRegister(const MachineOperand &MO, bool R0AsZero) {
144       unsigned RegNo = MO.getReg();
145       assert(TargetRegisterInfo::isPhysicalRegister(RegNo) && "Not physreg??");
146
147       // If we should use 0 for R0.
148       if (R0AsZero && RegNo == PPC::R0) {
149         O << "0";
150         return;
151       }
152
153       const char *RegName = getRegisterName(RegNo);
154       // Linux assembler (Others?) does not take register mnemonics.
155       // FIXME - What about special registers used in mfspr/mtspr?
156       if (!Subtarget.isDarwin()) RegName = stripRegisterPrefix(RegName);
157       O << RegName;
158     }
159
160     void printOperand(const MachineInstr *MI, unsigned OpNo) {
161       const MachineOperand &MO = MI->getOperand(OpNo);
162       if (MO.isReg()) {
163         printRegister(MO, false);
164       } else if (MO.isImm()) {
165         O << MO.getImm();
166       } else {
167         printOp(MO);
168       }
169     }
170
171     bool PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
172                          unsigned AsmVariant, const char *ExtraCode);
173     bool PrintAsmMemoryOperand(const MachineInstr *MI, unsigned OpNo,
174                                unsigned AsmVariant, const char *ExtraCode);
175
176
177     void printS5ImmOperand(const MachineInstr *MI, unsigned OpNo) {
178       char value = MI->getOperand(OpNo).getImm();
179       value = (value << (32-5)) >> (32-5);
180       O << (int)value;
181     }
182     void printU5ImmOperand(const MachineInstr *MI, unsigned OpNo) {
183       unsigned char value = MI->getOperand(OpNo).getImm();
184       assert(value <= 31 && "Invalid u5imm argument!");
185       O << (unsigned int)value;
186     }
187     void printU6ImmOperand(const MachineInstr *MI, unsigned OpNo) {
188       unsigned char value = MI->getOperand(OpNo).getImm();
189       assert(value <= 63 && "Invalid u6imm argument!");
190       O << (unsigned int)value;
191     }
192     void printS16ImmOperand(const MachineInstr *MI, unsigned OpNo) {
193       O << (short)MI->getOperand(OpNo).getImm();
194     }
195     void printU16ImmOperand(const MachineInstr *MI, unsigned OpNo) {
196       O << (unsigned short)MI->getOperand(OpNo).getImm();
197     }
198     void printS16X4ImmOperand(const MachineInstr *MI, unsigned OpNo) {
199       if (MI->getOperand(OpNo).isImm()) {
200         O << (short)(MI->getOperand(OpNo).getImm()*4);
201       } else {
202         O << "lo16(";
203         printOp(MI->getOperand(OpNo));
204         if (TM.getRelocationModel() == Reloc::PIC_)
205           O << "-\"L" << getFunctionNumber() << "$pb\")";
206         else
207           O << ')';
208       }
209     }
210     void printBranchOperand(const MachineInstr *MI, unsigned OpNo) {
211       // Branches can take an immediate operand.  This is used by the branch
212       // selection pass to print $+8, an eight byte displacement from the PC.
213       if (MI->getOperand(OpNo).isImm()) {
214         O << "$+" << MI->getOperand(OpNo).getImm()*4;
215       } else {
216         printOp(MI->getOperand(OpNo));
217       }
218     }
219     void printCallOperand(const MachineInstr *MI, unsigned OpNo) {
220       const MachineOperand &MO = MI->getOperand(OpNo);
221       if (TM.getRelocationModel() != Reloc::Static) {
222         if (MO.getType() == MachineOperand::MO_GlobalAddress) {
223           GlobalValue *GV = MO.getGlobal();
224           if (GV->isDeclaration() || GV->isWeakForLinker()) {
225             // Dynamically-resolved functions need a stub for the function.
226             FnStubInfo &FnInfo = FnStubs[Mang->getMangledName(GV)];
227             FnInfo.Init(GV, Mang);
228             O << FnInfo.Stub;
229             return;
230           }
231         }
232         if (MO.getType() == MachineOperand::MO_ExternalSymbol) {
233           FnStubInfo &FnInfo =FnStubs[Mang->makeNameProper(MO.getSymbolName())];
234           FnInfo.Init(MO.getSymbolName(), Mang);
235           O << FnInfo.Stub;
236           return;
237         }
238       }
239
240       printOp(MI->getOperand(OpNo));
241     }
242     void printAbsAddrOperand(const MachineInstr *MI, unsigned OpNo) {
243      O << (int)MI->getOperand(OpNo).getImm()*4;
244     }
245     void printPICLabel(const MachineInstr *MI, unsigned OpNo) {
246       O << "\"L" << getFunctionNumber() << "$pb\"\n";
247       O << "\"L" << getFunctionNumber() << "$pb\":";
248     }
249     void printSymbolHi(const MachineInstr *MI, unsigned OpNo) {
250       if (MI->getOperand(OpNo).isImm()) {
251         printS16ImmOperand(MI, OpNo);
252       } else {
253         if (Subtarget.isDarwin()) O << "ha16(";
254         printOp(MI->getOperand(OpNo));
255         if (TM.getRelocationModel() == Reloc::PIC_)
256           O << "-\"L" << getFunctionNumber() << "$pb\"";
257         if (Subtarget.isDarwin())
258           O << ')';
259         else
260           O << "@ha";
261       }
262     }
263     void printSymbolLo(const MachineInstr *MI, unsigned OpNo) {
264       if (MI->getOperand(OpNo).isImm()) {
265         printS16ImmOperand(MI, OpNo);
266       } else {
267         if (Subtarget.isDarwin()) O << "lo16(";
268         printOp(MI->getOperand(OpNo));
269         if (TM.getRelocationModel() == Reloc::PIC_)
270           O << "-\"L" << getFunctionNumber() << "$pb\"";
271         if (Subtarget.isDarwin())
272           O << ')';
273         else
274           O << "@l";
275       }
276     }
277     void printcrbitm(const MachineInstr *MI, unsigned OpNo) {
278       unsigned CCReg = MI->getOperand(OpNo).getReg();
279       unsigned RegNo = enumRegToMachineReg(CCReg);
280       O << (0x80 >> RegNo);
281     }
282     // The new addressing mode printers.
283     void printMemRegImm(const MachineInstr *MI, unsigned OpNo) {
284       printSymbolLo(MI, OpNo);
285       O << '(';
286       if (MI->getOperand(OpNo+1).isReg() &&
287           MI->getOperand(OpNo+1).getReg() == PPC::R0)
288         O << "0";
289       else
290         printOperand(MI, OpNo+1);
291       O << ')';
292     }
293     void printMemRegImmShifted(const MachineInstr *MI, unsigned OpNo) {
294       if (MI->getOperand(OpNo).isImm())
295         printS16X4ImmOperand(MI, OpNo);
296       else
297         printSymbolLo(MI, OpNo);
298       O << '(';
299       if (MI->getOperand(OpNo+1).isReg() &&
300           MI->getOperand(OpNo+1).getReg() == PPC::R0)
301         O << "0";
302       else
303         printOperand(MI, OpNo+1);
304       O << ')';
305     }
306
307     void printMemRegReg(const MachineInstr *MI, unsigned OpNo) {
308       // When used as the base register, r0 reads constant zero rather than
309       // the value contained in the register.  For this reason, the darwin
310       // assembler requires that we print r0 as 0 (no r) when used as the base.
311       const MachineOperand &MO = MI->getOperand(OpNo);
312       printRegister(MO, true);
313       O << ", ";
314       printOperand(MI, OpNo+1);
315     }
316
317     void printTOCEntryLabel(const MachineInstr *MI, unsigned OpNo) {
318       const MachineOperand &MO = MI->getOperand(OpNo);
319
320       assert(MO.getType() == MachineOperand::MO_GlobalAddress);
321
322       GlobalValue *GV = MO.getGlobal();
323
324       std::string Name = Mang->getMangledName(GV);
325
326       // Map symbol -> label of TOC entry.
327       if (TOC.count(Name) == 0) {
328         std::string Label;
329         Label += MAI->getPrivateGlobalPrefix();
330         Label += "C";
331         Label += utostr(LabelID++);
332
333         TOC[Name] = Label;
334       }
335
336       O << TOC[Name] << "@toc";
337     }
338
339     void printPredicateOperand(const MachineInstr *MI, unsigned OpNo,
340                                const char *Modifier);
341
342     virtual bool runOnMachineFunction(MachineFunction &F) = 0;
343   };
344
345   /// PPCLinuxAsmPrinter - PowerPC assembly printer, customized for Linux
346   class PPCLinuxAsmPrinter : public PPCAsmPrinter {
347   public:
348     explicit PPCLinuxAsmPrinter(formatted_raw_ostream &O, TargetMachine &TM,
349                                 const MCAsmInfo *T, bool V)
350       : PPCAsmPrinter(O, TM, T, V){}
351
352     virtual const char *getPassName() const {
353       return "Linux PPC Assembly Printer";
354     }
355
356     bool runOnMachineFunction(MachineFunction &F);
357     bool doFinalization(Module &M);
358
359     void getAnalysisUsage(AnalysisUsage &AU) const {
360       AU.setPreservesAll();
361       AU.addRequired<MachineModuleInfo>();
362       AU.addRequired<DwarfWriter>();
363       PPCAsmPrinter::getAnalysisUsage(AU);
364     }
365
366     void PrintGlobalVariable(const GlobalVariable *GVar);
367   };
368
369   /// PPCDarwinAsmPrinter - PowerPC assembly printer, customized for Darwin/Mac
370   /// OS X
371   class PPCDarwinAsmPrinter : public PPCAsmPrinter {
372     formatted_raw_ostream &OS;
373   public:
374     explicit PPCDarwinAsmPrinter(formatted_raw_ostream &O, TargetMachine &TM,
375                                  const MCAsmInfo *T, bool V)
376       : PPCAsmPrinter(O, TM, T, V), OS(O) {}
377
378     virtual const char *getPassName() const {
379       return "Darwin PPC Assembly Printer";
380     }
381
382     bool runOnMachineFunction(MachineFunction &F);
383     bool doFinalization(Module &M);
384     void EmitStartOfAsmFile(Module &M);
385
386     void getAnalysisUsage(AnalysisUsage &AU) const {
387       AU.setPreservesAll();
388       AU.addRequired<MachineModuleInfo>();
389       AU.addRequired<DwarfWriter>();
390       PPCAsmPrinter::getAnalysisUsage(AU);
391     }
392
393     void PrintGlobalVariable(const GlobalVariable *GVar);
394   };
395 } // end of anonymous namespace
396
397 // Include the auto-generated portion of the assembly writer
398 #include "PPCGenAsmWriter.inc"
399
400 void PPCAsmPrinter::printOp(const MachineOperand &MO) {
401   switch (MO.getType()) {
402   case MachineOperand::MO_Immediate:
403     llvm_unreachable("printOp() does not handle immediate values");
404
405   case MachineOperand::MO_MachineBasicBlock:
406     GetMBBSymbol(MO.getMBB()->getNumber())->print(O, MAI);
407     return;
408   case MachineOperand::MO_JumpTableIndex:
409     O << MAI->getPrivateGlobalPrefix() << "JTI" << getFunctionNumber()
410       << '_' << MO.getIndex();
411     // FIXME: PIC relocation model
412     return;
413   case MachineOperand::MO_ConstantPoolIndex:
414     O << MAI->getPrivateGlobalPrefix() << "CPI" << getFunctionNumber()
415       << '_' << MO.getIndex();
416     return;
417   case MachineOperand::MO_BlockAddress:
418     GetBlockAddressSymbol(MO.getBlockAddress())->print(O, MAI);
419     return;
420   case MachineOperand::MO_ExternalSymbol: {
421     // Computing the address of an external symbol, not calling it.
422     std::string Name(MAI->getGlobalPrefix());
423     Name += MO.getSymbolName();
424     
425     if (TM.getRelocationModel() != Reloc::Static) {
426       GVStubs[Name] = Name+"$non_lazy_ptr";
427       Name += "$non_lazy_ptr";
428     }
429     O << Name;
430     return;
431   }
432   case MachineOperand::MO_GlobalAddress: {
433     // Computing the address of a global symbol, not calling it.
434     GlobalValue *GV = MO.getGlobal();
435     std::string Name;
436
437     // External or weakly linked global variables need non-lazily-resolved stubs
438     if (TM.getRelocationModel() != Reloc::Static &&
439         (GV->isDeclaration() || GV->isWeakForLinker())) {
440       if (!GV->hasHiddenVisibility()) {
441         Name = Mang->getMangledName(GV, "$non_lazy_ptr", true);
442         GVStubs[Mang->getMangledName(GV)] = Name;
443       } else if (GV->isDeclaration() || GV->hasCommonLinkage() ||
444                  GV->hasAvailableExternallyLinkage()) {
445         Name = Mang->getMangledName(GV, "$non_lazy_ptr", true);
446         HiddenGVStubs[Mang->getMangledName(GV)] = Name;
447       } else {
448         Name = Mang->getMangledName(GV);
449       }
450     } else {
451       Name = Mang->getMangledName(GV);
452     }
453     O << Name;
454
455     printOffset(MO.getOffset());
456     return;
457   }
458
459   default:
460     O << "<unknown operand type: " << MO.getType() << ">";
461     return;
462   }
463 }
464
465 /// PrintAsmOperand - Print out an operand for an inline asm expression.
466 ///
467 bool PPCAsmPrinter::PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
468                                     unsigned AsmVariant,
469                                     const char *ExtraCode) {
470   // Does this asm operand have a single letter operand modifier?
471   if (ExtraCode && ExtraCode[0]) {
472     if (ExtraCode[1] != 0) return true; // Unknown modifier.
473
474     switch (ExtraCode[0]) {
475     default: return true;  // Unknown modifier.
476     case 'c': // Don't print "$" before a global var name or constant.
477       // PPC never has a prefix.
478       printOperand(MI, OpNo);
479       return false;
480     case 'L': // Write second word of DImode reference.
481       // Verify that this operand has two consecutive registers.
482       if (!MI->getOperand(OpNo).isReg() ||
483           OpNo+1 == MI->getNumOperands() ||
484           !MI->getOperand(OpNo+1).isReg())
485         return true;
486       ++OpNo;   // Return the high-part.
487       break;
488     case 'I':
489       // Write 'i' if an integer constant, otherwise nothing.  Used to print
490       // addi vs add, etc.
491       if (MI->getOperand(OpNo).isImm())
492         O << "i";
493       return false;
494     }
495   }
496
497   printOperand(MI, OpNo);
498   return false;
499 }
500
501 // At the moment, all inline asm memory operands are a single register.
502 // In any case, the output of this routine should always be just one
503 // assembler operand.
504
505 bool PPCAsmPrinter::PrintAsmMemoryOperand(const MachineInstr *MI, unsigned OpNo,
506                                           unsigned AsmVariant,
507                                           const char *ExtraCode) {
508   if (ExtraCode && ExtraCode[0])
509     return true; // Unknown modifier.
510   assert (MI->getOperand(OpNo).isReg());
511   O << "0(";
512   printOperand(MI, OpNo);
513   O << ")";
514   return false;
515 }
516
517 void PPCAsmPrinter::printPredicateOperand(const MachineInstr *MI, unsigned OpNo,
518                                           const char *Modifier) {
519   assert(Modifier && "Must specify 'cc' or 'reg' as predicate op modifier!");
520   unsigned Code = MI->getOperand(OpNo).getImm();
521   if (!strcmp(Modifier, "cc")) {
522     switch ((PPC::Predicate)Code) {
523     case PPC::PRED_ALWAYS: return; // Don't print anything for always.
524     case PPC::PRED_LT: O << "lt"; return;
525     case PPC::PRED_LE: O << "le"; return;
526     case PPC::PRED_EQ: O << "eq"; return;
527     case PPC::PRED_GE: O << "ge"; return;
528     case PPC::PRED_GT: O << "gt"; return;
529     case PPC::PRED_NE: O << "ne"; return;
530     case PPC::PRED_UN: O << "un"; return;
531     case PPC::PRED_NU: O << "nu"; return;
532     }
533
534   } else {
535     assert(!strcmp(Modifier, "reg") &&
536            "Need to specify 'cc' or 'reg' as predicate op modifier!");
537     // Don't print the register for 'always'.
538     if (Code == PPC::PRED_ALWAYS) return;
539     printOperand(MI, OpNo+1);
540   }
541 }
542
543
544 /// printMachineInstruction -- Print out a single PowerPC MI in Darwin syntax to
545 /// the current output stream.
546 ///
547 void PPCAsmPrinter::printMachineInstruction(const MachineInstr *MI) {
548   ++EmittedInsts;
549   
550   processDebugLoc(MI, true);
551
552   // Check for slwi/srwi mnemonics.
553   bool useSubstituteMnemonic = false;
554   if (MI->getOpcode() == PPC::RLWINM) {
555     unsigned char SH = MI->getOperand(2).getImm();
556     unsigned char MB = MI->getOperand(3).getImm();
557     unsigned char ME = MI->getOperand(4).getImm();
558     if (SH <= 31 && MB == 0 && ME == (31-SH)) {
559       O << "\tslwi "; useSubstituteMnemonic = true;
560     }
561     if (SH <= 31 && MB == (32-SH) && ME == 31) {
562       O << "\tsrwi "; useSubstituteMnemonic = true;
563       SH = 32-SH;
564     }
565     if (useSubstituteMnemonic) {
566       printOperand(MI, 0);
567       O << ", ";
568       printOperand(MI, 1);
569       O << ", " << (unsigned int)SH;
570     }
571   } else if (MI->getOpcode() == PPC::OR || MI->getOpcode() == PPC::OR8) {
572     if (MI->getOperand(1).getReg() == MI->getOperand(2).getReg()) {
573       useSubstituteMnemonic = true;
574       O << "\tmr ";
575       printOperand(MI, 0);
576       O << ", ";
577       printOperand(MI, 1);
578     }
579   } else if (MI->getOpcode() == PPC::RLDICR) {
580     unsigned char SH = MI->getOperand(2).getImm();
581     unsigned char ME = MI->getOperand(3).getImm();
582     // rldicr RA, RS, SH, 63-SH == sldi RA, RS, SH
583     if (63-SH == ME) {
584       useSubstituteMnemonic = true;
585       O << "\tsldi ";
586       printOperand(MI, 0);
587       O << ", ";
588       printOperand(MI, 1);
589       O << ", " << (unsigned int)SH;
590     }
591   }
592
593   if (!useSubstituteMnemonic)
594     printInstruction(MI);
595
596   if (VerboseAsm)
597     EmitComments(*MI);
598   O << '\n';
599
600   processDebugLoc(MI, false);
601 }
602
603 /// runOnMachineFunction - This uses the printMachineInstruction()
604 /// method to print assembly for each instruction.
605 ///
606 bool PPCLinuxAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
607   this->MF = &MF;
608
609   SetupMachineFunction(MF);
610   O << "\n\n";
611
612   // Print out constants referenced by the function
613   EmitConstantPool(MF.getConstantPool());
614
615   // Print out labels for the function.
616   const Function *F = MF.getFunction();
617   OutStreamer.SwitchSection(getObjFileLowering().SectionForGlobal(F, Mang, TM));
618
619   switch (F->getLinkage()) {
620   default: llvm_unreachable("Unknown linkage type!");
621   case Function::PrivateLinkage:
622   case Function::InternalLinkage:  // Symbols default to internal.
623     break;
624   case Function::ExternalLinkage:
625     O << "\t.global\t" << CurrentFnName << '\n'
626       << "\t.type\t" << CurrentFnName << ", @function\n";
627     break;
628   case Function::LinkerPrivateLinkage:
629   case Function::WeakAnyLinkage:
630   case Function::WeakODRLinkage:
631   case Function::LinkOnceAnyLinkage:
632   case Function::LinkOnceODRLinkage:
633     O << "\t.global\t" << CurrentFnName << '\n';
634     O << "\t.weak\t" << CurrentFnName << '\n';
635     break;
636   }
637
638   printVisibility(CurrentFnName, F->getVisibility());
639
640   EmitAlignment(MF.getAlignment(), F);
641
642   if (Subtarget.isPPC64()) {
643     // Emit an official procedure descriptor.
644     // FIXME 64-bit SVR4: Use MCSection here?
645     O << "\t.section\t\".opd\",\"aw\"\n";
646     O << "\t.align 3\n";
647     O << CurrentFnName << ":\n";
648     O << "\t.quad .L." << CurrentFnName << ",.TOC.@tocbase\n";
649     O << "\t.previous\n";
650     O << ".L." << CurrentFnName << ":\n";
651   } else {
652     O << CurrentFnName << ":\n";
653   }
654
655   // Emit pre-function debug information.
656   DW->BeginFunction(&MF);
657
658   // Print out code for the function.
659   for (MachineFunction::const_iterator I = MF.begin(), E = MF.end();
660        I != E; ++I) {
661     // Print a label for the basic block.
662     if (I != MF.begin()) {
663       EmitBasicBlockStart(I);
664     }
665     for (MachineBasicBlock::const_iterator II = I->begin(), E = I->end();
666          II != E; ++II) {
667       // Print the assembly for the instruction.
668       printMachineInstruction(II);
669     }
670   }
671
672   O << "\t.size\t" << CurrentFnName << ",.-" << CurrentFnName << '\n';
673
674   OutStreamer.SwitchSection(getObjFileLowering().SectionForGlobal(F, Mang, TM));
675
676   // Emit post-function debug information.
677   DW->EndFunction(&MF);
678
679   // Print out jump tables referenced by the function.
680   EmitJumpTableInfo(MF.getJumpTableInfo(), MF);
681
682   // We didn't modify anything.
683   return false;
684 }
685
686 void PPCLinuxAsmPrinter::PrintGlobalVariable(const GlobalVariable *GVar) {
687   const TargetData *TD = TM.getTargetData();
688
689   if (!GVar->hasInitializer())
690     return;   // External global require no code
691
692   // Check to see if this is a special global used by LLVM, if so, emit it.
693   if (EmitSpecialLLVMGlobal(GVar))
694     return;
695
696   std::string name = Mang->getMangledName(GVar);
697
698   printVisibility(name, GVar->getVisibility());
699
700   Constant *C = GVar->getInitializer();
701   const Type *Type = C->getType();
702   unsigned Size = TD->getTypeAllocSize(Type);
703   unsigned Align = TD->getPreferredAlignmentLog(GVar);
704
705   OutStreamer.SwitchSection(getObjFileLowering().SectionForGlobal(GVar, Mang,
706                                                                   TM));
707
708   if (C->isNullValue() && /* FIXME: Verify correct */
709       !GVar->hasSection() &&
710       (GVar->hasLocalLinkage() || GVar->hasExternalLinkage() ||
711        GVar->isWeakForLinker())) {
712       if (Size == 0) Size = 1;   // .comm Foo, 0 is undefined, avoid it.
713
714       if (GVar->hasExternalLinkage()) {
715         O << "\t.global " << name << '\n';
716         O << "\t.type " << name << ", @object\n";
717         O << name << ":\n";
718         O << "\t.zero " << Size << '\n';
719       } else if (GVar->hasLocalLinkage()) {
720         O << MAI->getLCOMMDirective() << name << ',' << Size;
721       } else {
722         O << ".comm " << name << ',' << Size;
723       }
724       if (VerboseAsm) {
725         O << "\t\t" << MAI->getCommentString() << " '";
726         WriteAsOperand(O, GVar, /*PrintType=*/false, GVar->getParent());
727         O << "'";
728       }
729       O << '\n';
730       return;
731   }
732
733   switch (GVar->getLinkage()) {
734    case GlobalValue::LinkOnceAnyLinkage:
735    case GlobalValue::LinkOnceODRLinkage:
736    case GlobalValue::WeakAnyLinkage:
737    case GlobalValue::WeakODRLinkage:
738    case GlobalValue::CommonLinkage:
739    case GlobalValue::LinkerPrivateLinkage:
740     O << "\t.global " << name << '\n'
741       << "\t.type " << name << ", @object\n"
742       << "\t.weak " << name << '\n';
743     break;
744    case GlobalValue::AppendingLinkage:
745     // FIXME: appending linkage variables should go into a section of
746     // their name or something.  For now, just emit them as external.
747    case GlobalValue::ExternalLinkage:
748     // If external or appending, declare as a global symbol
749     O << "\t.global " << name << '\n'
750       << "\t.type " << name << ", @object\n";
751     // FALL THROUGH
752    case GlobalValue::InternalLinkage:
753    case GlobalValue::PrivateLinkage:
754     break;
755    default:
756     llvm_unreachable("Unknown linkage type!");
757   }
758
759   EmitAlignment(Align, GVar);
760   O << name << ":";
761   if (VerboseAsm) {
762     O << "\t\t\t\t" << MAI->getCommentString() << " '";
763     WriteAsOperand(O, GVar, /*PrintType=*/false, GVar->getParent());
764     O << "'";
765   }
766   O << '\n';
767
768   EmitGlobalConstant(C);
769   O << '\n';
770 }
771
772 bool PPCLinuxAsmPrinter::doFinalization(Module &M) {
773   const TargetData *TD = TM.getTargetData();
774
775   bool isPPC64 = TD->getPointerSizeInBits() == 64;
776
777   if (isPPC64 && !TOC.empty()) {
778     // FIXME 64-bit SVR4: Use MCSection here?
779     O << "\t.section\t\".toc\",\"aw\"\n";
780
781     for (StringMap<std::string>::iterator I = TOC.begin(), E = TOC.end();
782          I != E; ++I) {
783       O << I->second << ":\n";
784       O << "\t.tc " << I->getKeyData() << "[TC]," << I->getKeyData() << '\n';
785     }
786   }
787
788   return AsmPrinter::doFinalization(M);
789 }
790
791 /// runOnMachineFunction - This uses the printMachineInstruction()
792 /// method to print assembly for each instruction.
793 ///
794 bool PPCDarwinAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
795   this->MF = &MF;
796
797   SetupMachineFunction(MF);
798   O << "\n\n";
799
800   // Print out constants referenced by the function
801   EmitConstantPool(MF.getConstantPool());
802
803   // Print out labels for the function.
804   const Function *F = MF.getFunction();
805   OutStreamer.SwitchSection(getObjFileLowering().SectionForGlobal(F, Mang, TM));
806
807   switch (F->getLinkage()) {
808   default: llvm_unreachable("Unknown linkage type!");
809   case Function::PrivateLinkage:
810   case Function::InternalLinkage:  // Symbols default to internal.
811     break;
812   case Function::ExternalLinkage:
813     O << "\t.globl\t" << CurrentFnName << '\n';
814     break;
815   case Function::WeakAnyLinkage:
816   case Function::WeakODRLinkage:
817   case Function::LinkOnceAnyLinkage:
818   case Function::LinkOnceODRLinkage:
819   case Function::LinkerPrivateLinkage:
820     O << "\t.globl\t" << CurrentFnName << '\n';
821     O << "\t.weak_definition\t" << CurrentFnName << '\n';
822     break;
823   }
824
825   printVisibility(CurrentFnName, F->getVisibility());
826
827   EmitAlignment(MF.getAlignment(), F);
828   O << CurrentFnName << ":\n";
829
830   // Emit pre-function debug information.
831   DW->BeginFunction(&MF);
832
833   // If the function is empty, then we need to emit *something*. Otherwise, the
834   // function's label might be associated with something that it wasn't meant to
835   // be associated with. We emit a noop in this situation.
836   MachineFunction::iterator I = MF.begin();
837
838   if (++I == MF.end() && MF.front().empty())
839     O << "\tnop\n";
840
841   // Print out code for the function.
842   for (MachineFunction::const_iterator I = MF.begin(), E = MF.end();
843        I != E; ++I) {
844     // Print a label for the basic block.
845     if (I != MF.begin()) {
846       EmitBasicBlockStart(I);
847     }
848     for (MachineBasicBlock::const_iterator II = I->begin(), IE = I->end();
849          II != IE; ++II) {
850       // Print the assembly for the instruction.
851       printMachineInstruction(II);
852     }
853   }
854
855   // Emit post-function debug information.
856   DW->EndFunction(&MF);
857
858   // Print out jump tables referenced by the function.
859   EmitJumpTableInfo(MF.getJumpTableInfo(), MF);
860
861   // We didn't modify anything.
862   return false;
863 }
864
865
866 void PPCDarwinAsmPrinter::EmitStartOfAsmFile(Module &M) {
867   static const char *const CPUDirectives[] = {
868     "",
869     "ppc",
870     "ppc601",
871     "ppc602",
872     "ppc603",
873     "ppc7400",
874     "ppc750",
875     "ppc970",
876     "ppc64"
877   };
878
879   unsigned Directive = Subtarget.getDarwinDirective();
880   if (Subtarget.isGigaProcessor() && Directive < PPC::DIR_970)
881     Directive = PPC::DIR_970;
882   if (Subtarget.hasAltivec() && Directive < PPC::DIR_7400)
883     Directive = PPC::DIR_7400;
884   if (Subtarget.isPPC64() && Directive < PPC::DIR_970)
885     Directive = PPC::DIR_64;
886   assert(Directive <= PPC::DIR_64 && "Directive out of range.");
887   O << "\t.machine " << CPUDirectives[Directive] << '\n';
888
889   // Prime text sections so they are adjacent.  This reduces the likelihood a
890   // large data or debug section causes a branch to exceed 16M limit.
891   TargetLoweringObjectFileMachO &TLOFMacho = 
892     static_cast<TargetLoweringObjectFileMachO &>(getObjFileLowering());
893   OutStreamer.SwitchSection(TLOFMacho.getTextCoalSection());
894   if (TM.getRelocationModel() == Reloc::PIC_) {
895     OutStreamer.SwitchSection(
896             TLOFMacho.getMachOSection("__TEXT", "__picsymbolstub1",
897                                       MCSectionMachO::S_SYMBOL_STUBS |
898                                       MCSectionMachO::S_ATTR_PURE_INSTRUCTIONS,
899                                       32, SectionKind::getText()));
900   } else if (TM.getRelocationModel() == Reloc::DynamicNoPIC) {
901     OutStreamer.SwitchSection(
902             TLOFMacho.getMachOSection("__TEXT","__symbol_stub1",
903                                       MCSectionMachO::S_SYMBOL_STUBS |
904                                       MCSectionMachO::S_ATTR_PURE_INSTRUCTIONS,
905                                       16, SectionKind::getText()));
906   }
907   OutStreamer.SwitchSection(getObjFileLowering().getTextSection());
908 }
909
910 void PPCDarwinAsmPrinter::PrintGlobalVariable(const GlobalVariable *GVar) {
911   const TargetData *TD = TM.getTargetData();
912
913   if (!GVar->hasInitializer())
914     return;   // External global require no code
915
916   // Check to see if this is a special global used by LLVM, if so, emit it.
917   if (EmitSpecialLLVMGlobal(GVar)) {
918     if (TM.getRelocationModel() == Reloc::Static) {
919       if (GVar->getName() == "llvm.global_ctors")
920         O << ".reference .constructors_used\n";
921       else if (GVar->getName() == "llvm.global_dtors")
922         O << ".reference .destructors_used\n";
923     }
924     return;
925   }
926
927   std::string name = Mang->getMangledName(GVar);
928   printVisibility(name, GVar->getVisibility());
929
930   Constant *C = GVar->getInitializer();
931   const Type *Type = C->getType();
932   unsigned Size = TD->getTypeAllocSize(Type);
933   unsigned Align = TD->getPreferredAlignmentLog(GVar);
934
935   const MCSection *TheSection =
936     getObjFileLowering().SectionForGlobal(GVar, Mang, TM);
937   OutStreamer.SwitchSection(TheSection);
938
939   /// FIXME: Drive this off the section!
940   if (C->isNullValue() && /* FIXME: Verify correct */
941       !GVar->hasSection() &&
942       (GVar->hasLocalLinkage() || GVar->hasExternalLinkage() ||
943        GVar->isWeakForLinker()) &&
944       // Don't put things that should go in the cstring section into "comm".
945       !TheSection->getKind().isMergeableCString()) {
946     if (Size == 0) Size = 1;   // .comm Foo, 0 is undefined, avoid it.
947
948     if (GVar->hasExternalLinkage()) {
949       O << "\t.globl " << name << '\n';
950       O << "\t.zerofill __DATA, __common, " << name << ", "
951         << Size << ", " << Align;
952     } else if (GVar->hasLocalLinkage()) {
953       O << MAI->getLCOMMDirective() << name << ',' << Size << ',' << Align;
954     } else if (!GVar->hasCommonLinkage()) {
955       O << "\t.globl " << name << '\n'
956         << MAI->getWeakDefDirective() << name << '\n';
957       EmitAlignment(Align, GVar);
958       O << name << ":";
959       if (VerboseAsm) {
960         O << "\t\t\t\t" << MAI->getCommentString() << " ";
961         WriteAsOperand(O, GVar, /*PrintType=*/false, GVar->getParent());
962       }
963       O << '\n';
964       EmitGlobalConstant(C);
965       return;
966     } else {
967       O << ".comm " << name << ',' << Size;
968       // Darwin 9 and above support aligned common data.
969       if (Subtarget.isDarwin9())
970         O << ',' << Align;
971     }
972     if (VerboseAsm) {
973       O << "\t\t" << MAI->getCommentString() << " '";
974       WriteAsOperand(O, GVar, /*PrintType=*/false, GVar->getParent());
975       O << "'";
976     }
977     O << '\n';
978     return;
979   }
980
981   switch (GVar->getLinkage()) {
982    case GlobalValue::LinkOnceAnyLinkage:
983    case GlobalValue::LinkOnceODRLinkage:
984    case GlobalValue::WeakAnyLinkage:
985    case GlobalValue::WeakODRLinkage:
986    case GlobalValue::CommonLinkage:
987    case GlobalValue::LinkerPrivateLinkage:
988     O << "\t.globl " << name << '\n'
989       << "\t.weak_definition " << name << '\n';
990     break;
991    case GlobalValue::AppendingLinkage:
992     // FIXME: appending linkage variables should go into a section of
993     // their name or something.  For now, just emit them as external.
994    case GlobalValue::ExternalLinkage:
995     // If external or appending, declare as a global symbol
996     O << "\t.globl " << name << '\n';
997     // FALL THROUGH
998    case GlobalValue::InternalLinkage:
999    case GlobalValue::PrivateLinkage:
1000     break;
1001    default:
1002     llvm_unreachable("Unknown linkage type!");
1003   }
1004
1005   EmitAlignment(Align, GVar);
1006   O << name << ":";
1007   if (VerboseAsm) {
1008     O << "\t\t\t\t" << MAI->getCommentString() << " '";
1009     WriteAsOperand(O, GVar, /*PrintType=*/false, GVar->getParent());
1010     O << "'";
1011   }
1012   O << '\n';
1013
1014   EmitGlobalConstant(C);
1015   O << '\n';
1016 }
1017
1018 bool PPCDarwinAsmPrinter::doFinalization(Module &M) {
1019   const TargetData *TD = TM.getTargetData();
1020
1021   bool isPPC64 = TD->getPointerSizeInBits() == 64;
1022
1023   // Darwin/PPC always uses mach-o.
1024   TargetLoweringObjectFileMachO &TLOFMacho = 
1025     static_cast<TargetLoweringObjectFileMachO &>(getObjFileLowering());
1026
1027   
1028   const MCSection *LSPSection = 0;
1029   if (!FnStubs.empty()) // .lazy_symbol_pointer
1030     LSPSection = TLOFMacho.getLazySymbolPointerSection();
1031     
1032   
1033   // Output stubs for dynamically-linked functions
1034   if (TM.getRelocationModel() == Reloc::PIC_ && !FnStubs.empty()) {
1035     const MCSection *StubSection = 
1036       TLOFMacho.getMachOSection("__TEXT", "__picsymbolstub1",
1037                                 MCSectionMachO::S_SYMBOL_STUBS |
1038                                 MCSectionMachO::S_ATTR_PURE_INSTRUCTIONS,
1039                                 32, SectionKind::getText());
1040      for (StringMap<FnStubInfo>::iterator I = FnStubs.begin(), E = FnStubs.end();
1041          I != E; ++I) {
1042       OutStreamer.SwitchSection(StubSection);
1043       EmitAlignment(4);
1044       const FnStubInfo &Info = I->second;
1045       O << Info.Stub << ":\n";
1046       O << "\t.indirect_symbol " << I->getKeyData() << '\n';
1047       O << "\tmflr r0\n";
1048       O << "\tbcl 20,31," << Info.AnonSymbol << '\n';
1049       O << Info.AnonSymbol << ":\n";
1050       O << "\tmflr r11\n";
1051       O << "\taddis r11,r11,ha16(" << Info.LazyPtr << "-" << Info.AnonSymbol;
1052       O << ")\n";
1053       O << "\tmtlr r0\n";
1054       O << (isPPC64 ? "\tldu" : "\tlwzu") << " r12,lo16(";
1055       O << Info.LazyPtr << "-" << Info.AnonSymbol << ")(r11)\n";
1056       O << "\tmtctr r12\n";
1057       O << "\tbctr\n";
1058       
1059       OutStreamer.SwitchSection(LSPSection);
1060       O << Info.LazyPtr << ":\n";
1061       O << "\t.indirect_symbol " << I->getKeyData() << '\n';
1062       O << (isPPC64 ? "\t.quad" : "\t.long") << " dyld_stub_binding_helper\n";
1063     }
1064   } else if (!FnStubs.empty()) {
1065     const MCSection *StubSection =
1066       TLOFMacho.getMachOSection("__TEXT","__symbol_stub1",
1067                                 MCSectionMachO::S_SYMBOL_STUBS |
1068                                 MCSectionMachO::S_ATTR_PURE_INSTRUCTIONS,
1069                                 16, SectionKind::getText());
1070     
1071     for (StringMap<FnStubInfo>::iterator I = FnStubs.begin(), E = FnStubs.end();
1072          I != E; ++I) {
1073       OutStreamer.SwitchSection(StubSection);
1074       EmitAlignment(4);
1075       const FnStubInfo &Info = I->second;
1076       O << Info.Stub << ":\n";
1077       O << "\t.indirect_symbol " << I->getKeyData() << '\n';
1078       O << "\tlis r11,ha16(" << Info.LazyPtr << ")\n";
1079       O << (isPPC64 ? "\tldu" :  "\tlwzu") << " r12,lo16(";
1080       O << Info.LazyPtr << ")(r11)\n";
1081       O << "\tmtctr r12\n";
1082       O << "\tbctr\n";
1083       OutStreamer.SwitchSection(LSPSection);
1084       O << Info.LazyPtr << ":\n";
1085       O << "\t.indirect_symbol " << I->getKeyData() << '\n';
1086       O << (isPPC64 ? "\t.quad" : "\t.long") << " dyld_stub_binding_helper\n";
1087     }
1088   }
1089
1090   O << '\n';
1091
1092   if (MAI->doesSupportExceptionHandling() && MMI) {
1093     // Add the (possibly multiple) personalities to the set of global values.
1094     // Only referenced functions get into the Personalities list.
1095     const std::vector<Function *> &Personalities = MMI->getPersonalities();
1096     for (std::vector<Function *>::const_iterator I = Personalities.begin(),
1097          E = Personalities.end(); I != E; ++I) {
1098       if (*I)
1099         GVStubs[Mang->getMangledName(*I)] =
1100           Mang->getMangledName(*I, "$non_lazy_ptr", true);
1101     }
1102   }
1103
1104   // Output macho stubs for external and common global variables.
1105   if (!GVStubs.empty()) {
1106     // Switch with ".non_lazy_symbol_pointer" directive.
1107     OutStreamer.SwitchSection(TLOFMacho.getNonLazySymbolPointerSection());
1108     EmitAlignment(isPPC64 ? 3 : 2);
1109     
1110     for (StringMap<std::string>::iterator I = GVStubs.begin(),
1111          E = GVStubs.end(); I != E; ++I) {
1112       O << I->second << ":\n";
1113       O << "\t.indirect_symbol " << I->getKeyData() << '\n';
1114       O << (isPPC64 ? "\t.quad\t0\n" : "\t.long\t0\n");
1115     }
1116   }
1117
1118   if (!HiddenGVStubs.empty()) {
1119     OutStreamer.SwitchSection(getObjFileLowering().getDataSection());
1120     EmitAlignment(isPPC64 ? 3 : 2);
1121     for (StringMap<std::string>::iterator I = HiddenGVStubs.begin(),
1122          E = HiddenGVStubs.end(); I != E; ++I) {
1123       O << I->second << ":\n";
1124       O << (isPPC64 ? "\t.quad\t" : "\t.long\t") << I->getKeyData() << '\n';
1125     }
1126   }
1127
1128   // Funny Darwin hack: This flag tells the linker that no global symbols
1129   // contain code that falls through to other global symbols (e.g. the obvious
1130   // implementation of multiple entry points).  If this doesn't occur, the
1131   // linker can safely perform dead code stripping.  Since LLVM never generates
1132   // code that does this, it is always safe to set.
1133   OutStreamer.EmitAssemblerFlag(MCStreamer::SubsectionsViaSymbols);
1134
1135   return AsmPrinter::doFinalization(M);
1136 }
1137
1138
1139
1140 /// createPPCAsmPrinterPass - Returns a pass that prints the PPC assembly code
1141 /// for a MachineFunction to the given output stream, in a format that the
1142 /// Darwin assembler can deal with.
1143 ///
1144 static AsmPrinter *createPPCAsmPrinterPass(formatted_raw_ostream &o,
1145                                            TargetMachine &tm,
1146                                            const MCAsmInfo *tai,
1147                                            bool verbose) {
1148   const PPCSubtarget *Subtarget = &tm.getSubtarget<PPCSubtarget>();
1149
1150   if (Subtarget->isDarwin())
1151     return new PPCDarwinAsmPrinter(o, tm, tai, verbose);
1152   return new PPCLinuxAsmPrinter(o, tm, tai, verbose);
1153 }
1154
1155 // Force static initialization.
1156 extern "C" void LLVMInitializePowerPCAsmPrinter() { 
1157   TargetRegistry::RegisterAsmPrinter(ThePPC32Target, createPPCAsmPrinterPass);
1158   TargetRegistry::RegisterAsmPrinter(ThePPC64Target, createPPCAsmPrinterPass);
1159 }