lib/Target/X86/X86AsmPrinter.cpp

   1 //===-- X86AsmPrinter.cpp - Convert X86 LLVM IR to X86 assembly -----------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file the shared super class printer that converts from our internal
  11 // representation of machine-dependent LLVM code to Intel and AT&T format
  12 // assembly language.
  13 // This printer is the output mechanism used by `llc'.
  14 //
  15 //===----------------------------------------------------------------------===//
  16
  17 #include "X86ATTAsmPrinter.h"
  18 #include "X86IntelAsmPrinter.h"
  19 #include "X86.h"
  20 #include "llvm/Module.h"
  21 #include "llvm/Type.h"
  22 #include "llvm/Assembly/Writer.h"
  23 #include "llvm/CodeGen/MachineConstantPool.h"
  24 #include "llvm/Support/Mangler.h"
  25 #include "llvm/Support/CommandLine.h"
  26 using namespace llvm;
  27 using namespace x86;
  28
  29 Statistic<> llvm::x86::EmittedInsts("asm-printer",
  30                                     "Number of machine instrs printed");
  31
  32 enum AsmWriterFlavorTy { att, intel };
  33 cl::opt<AsmWriterFlavorTy>
  34 AsmWriterFlavor("x86-asm-syntax",
  35                 cl::desc("Choose style of code to emit from X86 backend:"),
  36                 cl::values(
  37                            clEnumVal(att,   "  Emit AT&T-style assembly"),
  38                            clEnumVal(intel, "  Emit Intel-style assembly"),
  39                            clEnumValEnd),
  40                 cl::init(att));
  41
  42 /// doInitialization
  43 bool X86SharedAsmPrinter::doInitialization(Module& M) {
  44   bool leadingUnderscore = false;
  45   forCygwin = false;
  46   const std::string& TT = M.getTargetTriple();
  47   if (TT.length() > 5) {
  48     forCygwin = TT.find("cygwin") != std::string::npos ||
  49                 TT.find("mingw")  != std::string::npos;
  50     forDarwin = TT.find("darwin") != std::string::npos;
  51   } else if (TT.empty()) {
  52 #if defined(__CYGWIN__) || defined(__MINGW32__)
  53     forCygwin = true;
  54 #elif defined(__APPLE__)
  55     forDarwin = true;
  56 #elif defined(_WIN32)
  57     leadingUnderscore = true;
  58 #else
  59     leadingUnderscore = false;
  60 #endif
  61   }
  62
  63   if (leadingUnderscore || forCygwin || forDarwin)
  64     GlobalPrefix = "_";
  65
  66   if (forDarwin) {
  67     AlignmentIsInBytes = false;
  68     Data64bitsDirective = 0;       // we can't emit a 64-bit unit
  69     ZeroDirective = "\t.space\t";  // ".space N" emits N zeros.
  70     PrivateGlobalPrefix = "L";     // Marker for constant pool idxs
  71   }
  72
  73   return AsmPrinter::doInitialization(M);
  74 }
  75
  76 /// printConstantPool - Print to the current output stream assembly
  77 /// representations of the constants in the constant pool MCP. This is
  78 /// used to print out constants which have been "spilled to memory" by
  79 /// the code generator.
  80 ///
  81 void X86SharedAsmPrinter::printConstantPool(MachineConstantPool *MCP) {
  82   const std::vector<Constant*> &CP = MCP->getConstants();
  83   const TargetData &TD = TM.getTargetData();
  84
  85   if (CP.empty()) return;
  86
  87   if (forDarwin) {
  88     O << "\t.const\n";
  89   } else {
  90     O << "\t.section .rodata\n";
  91   }
  92
  93   for (unsigned i = 0, e = CP.size(); i != e; ++i) {
  94     // FIXME: force doubles to be naturally aligned.  We should handle this
  95     // more correctly in the future.
  96     if (CP[i]->getType() == Type::DoubleTy)
  97       emitAlignment(3);
  98     else
  99       emitAlignment(TD.getTypeAlignmentShift(CP[i]->getType()));
 100     O << PrivateGlobalPrefix << "CPI" << CurrentFnName << "_" << i
 101       << ":\t\t\t\t\t" << CommentString << *CP[i] << "\n";
 102     emitGlobalConstant(CP[i]);
 103   }
 104 }
 105
 106 bool X86SharedAsmPrinter::doFinalization(Module &M) {
 107   const TargetData &TD = TM.getTargetData();
 108   std::string CurSection;
 109
 110   // Print out module-level global variables here.
 111   for (Module::const_global_iterator I = M.global_begin(),
 112          E = M.global_end(); I != E; ++I)
 113     if (I->hasInitializer()) {   // External global require no code
 114       O << "\n\n";
 115       std::string name = Mang->getValueName(I);
 116       Constant *C = I->getInitializer();
 117       unsigned Size = TD.getTypeSize(C->getType());
 118       unsigned Align = TD.getTypeAlignmentShift(C->getType());
 119
 120       if (C->isNullValue() &&
 121           (I->hasLinkOnceLinkage() || I->hasInternalLinkage() ||
 122            I->hasWeakLinkage() /* FIXME: Verify correct */)) {
 123         switchSection(O, CurSection, ".data");
 124         if (!forCygwin && !forDarwin && I->hasInternalLinkage())
 125           O << "\t.local " << name << "\n";
 126         if (forDarwin && I->hasInternalLinkage())
 127           O << "\t.lcomm " << name << "," << Size << "," << Align;
 128         else
 129           O << "\t.comm " << name << "," << Size;
 130         if (!forCygwin && !forDarwin)
 131           O << "," << (1 << Align);
 132         O << "\t\t# ";
 133         WriteAsOperand(O, I, true, true, &M);
 134         O << "\n";
 135       } else {
 136         switch (I->getLinkage()) {
 137         default: assert(0 && "Unknown linkage type!");
 138         case GlobalValue::LinkOnceLinkage:
 139         case GlobalValue::WeakLinkage:   // FIXME: Verify correct for weak.
 140           // Nonnull linkonce -> weak
 141           O << "\t.weak " << name << "\n";
 142           switchSection(O, CurSection, "");
 143           O << "\t.section\t.llvm.linkonce.d." << name << ",\"aw\",@progbits\n";
 144           break;
 145         case GlobalValue::AppendingLinkage:
 146           // FIXME: appending linkage variables should go into a section of
 147           // their name or something.  For now, just emit them as external.
 148         case GlobalValue::ExternalLinkage:
 149           // If external or appending, declare as a global symbol
 150           O << "\t.globl " << name << "\n";
 151           // FALL THROUGH
 152         case GlobalValue::InternalLinkage:
 153           if (C->isNullValue())
 154             switchSection(O, CurSection, ".bss");
 155           else
 156             switchSection(O, CurSection, ".data");
 157           break;
 158         }
 159
 160         emitAlignment(Align);
 161         if (!forCygwin && !forDarwin) {
 162           O << "\t.type " << name << ",@object\n";
 163           O << "\t.size " << name << "," << Size << "\n";
 164         }
 165         O << name << ":\t\t\t\t# ";
 166         WriteAsOperand(O, I, true, true, &M);
 167         O << " = ";
 168         WriteAsOperand(O, C, false, false, &M);
 169         O << "\n";
 170         emitGlobalConstant(C);
 171       }
 172     }
 173
 174   if (forDarwin) {
 175     // Output stubs for external global variables
 176     if (GVStubs.begin() != GVStubs.end())
 177       O << "\t.non_lazy_symbol_pointer\n";
 178     for (std::set<std::string>::iterator i = GVStubs.begin(), e = GVStubs.end();
 179          i != e; ++i) {
 180       O << "L" << *i << "$non_lazy_ptr:\n";
 181       O << "\t.indirect_symbol " << *i << "\n";
 182       O << "\t.long\t0\n";
 183     }
 184
 185     // Output stubs for dynamically-linked functions
 186     unsigned j = 1;
 187     for (std::set<std::string>::iterator i = FnStubs.begin(), e = FnStubs.end();
 188          i != e; ++i, ++j) {
 189       O << "\t.symbol_stub\n";
 190       O << "L" << *i << "$stub:\n";
 191       O << "\t.indirect_symbol " << *i << "\n";
 192       O << "\tjmp\t*L" << j << "$lz\n";
 193       O << "L" << *i << "$stub_binder:\n";
 194       O << "\tpushl\t$L" << j << "$lz\n";
 195       O << "\tjmp\tdyld_stub_binding_helper\n";
 196       O << "\t.section __DATA, __la_sym_ptr3,lazy_symbol_pointers\n";
 197       O << "L" << j << "$lz:\n";
 198       O << "\t.indirect_symbol " << *i << "\n";
 199       O << "\t.long\tL" << *i << "$stub_binder\n";
 200     }
 201
 202     O << "\n";
 203
 204     // Output stubs for link-once variables
 205     if (LinkOnceStubs.begin() != LinkOnceStubs.end())
 206       O << ".data\n.align 2\n";
 207     for (std::set<std::string>::iterator i = LinkOnceStubs.begin(),
 208          e = LinkOnceStubs.end(); i != e; ++i) {
 209       O << "L" << *i << "$non_lazy_ptr:\n"
 210         << "\t.long\t" << *i << '\n';
 211     }
 212   }
 213
 214   AsmPrinter::doFinalization(M);
 215   return false; // success
 216 }
 217
 218 /// createX86CodePrinterPass - Returns a pass that prints the X86 assembly code
 219 /// for a MachineFunction to the given output stream, using the given target
 220 /// machine description.
 221 ///
 222 FunctionPass *llvm::createX86CodePrinterPass(std::ostream &o,TargetMachine &tm){
 223   switch (AsmWriterFlavor) {
 224   default:
 225     assert(0 && "Unknown asm flavor!");
 226   case intel:
 227     return new X86IntelAsmPrinter(o, tm);
 228   case att:
 229     return new X86ATTAsmPrinter(o, tm);
 230   }
 231 }