lib/Target/ARM/ARMAsmPrinter.cpp

   1 //===-- ARMAsmPrinter.cpp - ARM LLVM assembly writer ----------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the "Instituto Nokia de Tecnologia" and
   6 // is distributed under the University of Illinois Open Source
   7 // License. See LICENSE.TXT for details.
   8 //
   9 //===----------------------------------------------------------------------===//
  10 //
  11 // This file contains a printer that converts from our internal representation
  12 // of machine-dependent LLVM code to GAS-format ARM assembly language.
  13 //
  14 //===----------------------------------------------------------------------===//
  15
  16 #define DEBUG_TYPE "asm-printer"
  17 #include "ARM.h"
  18 #include "ARMTargetMachine.h"
  19 #include "ARMAddressingModes.h"
  20 #include "ARMConstantPoolValue.h"
  21 #include "ARMMachineFunctionInfo.h"
  22 #include "llvm/Constants.h"
  23 #include "llvm/Module.h"
  24 #include "llvm/CodeGen/AsmPrinter.h"
  25 #include "llvm/CodeGen/DwarfWriter.h"
  26 #include "llvm/CodeGen/MachineModuleInfo.h"
  27 #include "llvm/CodeGen/MachineFunctionPass.h"
  28 #include "llvm/CodeGen/MachineJumpTableInfo.h"
  29 #include "llvm/Target/TargetAsmInfo.h"
  30 #include "llvm/Target/TargetData.h"
  31 #include "llvm/Target/TargetMachine.h"
  32 #include "llvm/Target/TargetOptions.h"
  33 #include "llvm/ADT/Statistic.h"
  34 #include "llvm/ADT/StringExtras.h"
  35 #include "llvm/Support/Compiler.h"
  36 #include "llvm/Support/Mangler.h"
  37 #include "llvm/Support/MathExtras.h"
  38 #include <cctype>
  39 #include <iostream>
  40 using namespace llvm;
  41
  42 STATISTIC(EmittedInsts, "Number of machine instrs printed");
  43
  44 namespace {
  45   struct VISIBILITY_HIDDEN ARMAsmPrinter : public AsmPrinter {
  46     ARMAsmPrinter(std::ostream &O, TargetMachine &TM, const TargetAsmInfo *T)
  47       : AsmPrinter(O, TM, T), DW(O, this, T), AFI(NULL), InCPMode(false) {
  48       Subtarget = &TM.getSubtarget<ARMSubtarget>();
  49     }
  50
  51     DwarfWriter DW;
  52
  53     /// Subtarget - Keep a pointer to the ARMSubtarget around so that we can
  54     /// make the right decision when printing asm code for different targets.
  55     const ARMSubtarget *Subtarget;
  56
  57     /// AFI - Keep a pointer to ARMFunctionInfo for the current
  58     /// MachineFunction
  59     ARMFunctionInfo *AFI;
  60
  61     /// We name each basic block in a Function with a unique number, so
  62     /// that we can consistently refer to them later. This is cleared
  63     /// at the beginning of each call to runOnMachineFunction().
  64     ///
  65     typedef std::map<const Value *, unsigned> ValueMapTy;
  66     ValueMapTy NumberForBB;
  67
  68     /// Keeps the set of GlobalValues that require non-lazy-pointers for
  69     /// indirect access.
  70     std::set<std::string> GVNonLazyPtrs;
  71
  72     /// Keeps the set of external function GlobalAddresses that the asm
  73     /// printer should generate stubs for.
  74     std::set<std::string> FnStubs;
  75
  76     /// True if asm printer is printing a series of CONSTPOOL_ENTRY.
  77     bool InCPMode;
  78
  79     virtual const char *getPassName() const {
  80       return "ARM Assembly Printer";
  81     }
  82
  83     void printOperand(const MachineInstr *MI, int opNum,
  84                       const char *Modifier = 0);
  85     void printSOImmOperand(const MachineInstr *MI, int opNum);
  86     void printSORegOperand(const MachineInstr *MI, int opNum);
  87     void printAddrMode2Operand(const MachineInstr *MI, int OpNo);
  88     void printAddrMode2OffsetOperand(const MachineInstr *MI, int OpNo);
  89     void printAddrMode3Operand(const MachineInstr *MI, int OpNo);
  90     void printAddrMode3OffsetOperand(const MachineInstr *MI, int OpNo);
  91     void printAddrMode4Operand(const MachineInstr *MI, int OpNo,
  92                                const char *Modifier = 0);
  93     void printAddrMode5Operand(const MachineInstr *MI, int OpNo,
  94                                const char *Modifier = 0);
  95     void printAddrModePCOperand(const MachineInstr *MI, int OpNo,
  96                                 const char *Modifier = 0);
  97     void printThumbAddrModeRROperand(const MachineInstr *MI, int OpNo);
  98     void printThumbAddrModeRI5Operand(const MachineInstr *MI, int OpNo,
  99                                       unsigned Scale);
 100     void printThumbAddrModeS1Operand(const MachineInstr *MI, int OpNo);
 101     void printThumbAddrModeS2Operand(const MachineInstr *MI, int OpNo);
 102     void printThumbAddrModeS4Operand(const MachineInstr *MI, int OpNo);
 103     void printThumbAddrModeSPOperand(const MachineInstr *MI, int OpNo);
 104     void printCCOperand(const MachineInstr *MI, int opNum);
 105     void printPCLabel(const MachineInstr *MI, int opNum);
 106     void printRegisterList(const MachineInstr *MI, int opNum);
 107     void printCPInstOperand(const MachineInstr *MI, int opNum,
 108                             const char *Modifier);
 109     void printJTBlockOperand(const MachineInstr *MI, int opNum);
 110
 111     virtual bool PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
 112                                  unsigned AsmVariant, const char *ExtraCode);
 113
 114     bool printInstruction(const MachineInstr *MI);  // autogenerated.
 115     void printMachineInstruction(const MachineInstr *MI);
 116     bool runOnMachineFunction(MachineFunction &F);
 117     bool doInitialization(Module &M);
 118     bool doFinalization(Module &M);
 119
 120     virtual void EmitMachineConstantPoolValue(MachineConstantPoolValue *MCPV) {
 121       printDataDirective(MCPV->getType());
 122
 123       ARMConstantPoolValue *ACPV = (ARMConstantPoolValue*)MCPV;
 124       GlobalValue *GV = ACPV->getGV();
 125       std::string Name = Mang->getValueName(GV);
 126       if (ACPV->isNonLazyPointer()) {
 127         GVNonLazyPtrs.insert(Name);
 128         O << TAI->getPrivateGlobalPrefix() << Name << "$non_lazy_ptr";
 129       } else
 130         O << Name;
 131       if (ACPV->getPCAdjustment() != 0)
 132         O << "-(" << TAI->getPrivateGlobalPrefix() << "PC"
 133           << utostr(ACPV->getLabelId())
 134           << "+" << (unsigned)ACPV->getPCAdjustment() << ")";
 135       O << "\n";
 136
 137       // If the constant pool value is a extern weak symbol, remember to emit
 138       // the weak reference.
 139       if (GV->hasExternalWeakLinkage())
 140         ExtWeakSymbols.insert(GV);
 141     }
 142
 143     void getAnalysisUsage(AnalysisUsage &AU) const {
 144       AU.setPreservesAll();
 145       AU.addRequired<MachineModuleInfo>();
 146     }
 147   };
 148 } // end of anonymous namespace
 149
 150 #include "ARMGenAsmWriter.inc"
 151
 152 /// createARMCodePrinterPass - Returns a pass that prints the ARM
 153 /// assembly code for a MachineFunction to the given output stream,
 154 /// using the given target machine description.  This should work
 155 /// regardless of whether the function is in SSA form.
 156 ///
 157 FunctionPass *llvm::createARMCodePrinterPass(std::ostream &o,
 158                                              ARMTargetMachine &tm) {
 159   return new ARMAsmPrinter(o, tm, tm.getTargetAsmInfo());
 160 }
 161
 162 /// runOnMachineFunction - This uses the printInstruction()
 163 /// method to print assembly for each instruction.
 164 ///
 165 bool ARMAsmPrinter::runOnMachineFunction(MachineFunction &MF) {
 166   AFI = MF.getInfo<ARMFunctionInfo>();
 167
 168   if (Subtarget->isTargetDarwin()) {
 169     DW.SetModuleInfo(&getAnalysis<MachineModuleInfo>());
 170   }
 171
 172   SetupMachineFunction(MF);
 173   O << "\n";
 174
 175   // NOTE: we don't print out constant pools here, they are handled as
 176   // instructions.
 177
 178   O << "\n";
 179   // Print out labels for the function.
 180   const Function *F = MF.getFunction();
 181   switch (F->getLinkage()) {
 182   default: assert(0 && "Unknown linkage type!");
 183   case Function::InternalLinkage:
 184     SwitchToTextSection("\t.text", F);
 185     break;
 186   case Function::ExternalLinkage:
 187     SwitchToTextSection("\t.text", F);
 188     O << "\t.globl\t" << CurrentFnName << "\n";
 189     break;
 190   case Function::WeakLinkage:
 191   case Function::LinkOnceLinkage:
 192     if (Subtarget->isTargetDarwin()) {
 193       SwitchToTextSection(
 194                 ".section __TEXT,__textcoal_nt,coalesced,pure_instructions", F);
 195       O << "\t.globl\t" << CurrentFnName << "\n";
 196       O << "\t.weak_definition\t" << CurrentFnName << "\n";
 197     } else {
 198       O << TAI->getWeakRefDirective() << CurrentFnName << "\n";
 199     }
 200     break;
 201   }
 202
 203   if (AFI->isThumbFunction()) {
 204     EmitAlignment(1, F);
 205     O << "\t.code\t16\n";
 206     O << "\t.thumb_func\t" << CurrentFnName << "\n";
 207     InCPMode = false;
 208   } else
 209     EmitAlignment(2, F);
 210
 211   O << CurrentFnName << ":\n";
 212   if (Subtarget->isTargetDarwin()) {
 213     // Emit pre-function debug information.
 214     DW.BeginFunction(&MF);
 215   }
 216
 217   // Print out code for the function.
 218   for (MachineFunction::const_iterator I = MF.begin(), E = MF.end();
 219        I != E; ++I) {
 220     // Print a label for the basic block.
 221     if (I != MF.begin()) {
 222       printBasicBlockLabel(I, true);
 223       O << '\n';
 224     }
 225     for (MachineBasicBlock::const_iterator II = I->begin(), E = I->end();
 226          II != E; ++II) {
 227       // Print the assembly for the instruction.
 228       printMachineInstruction(II);
 229     }
 230   }
 231
 232   if (TAI->hasDotTypeDotSizeDirective())
 233     O << "\t.size " << CurrentFnName << ", .-" << CurrentFnName << "\n";
 234
 235   if (Subtarget->isTargetDarwin()) {
 236     // Emit post-function debug information.
 237     DW.EndFunction();
 238   }
 239
 240   return false;
 241 }
 242
 243 void ARMAsmPrinter::printOperand(const MachineInstr *MI, int opNum,
 244                                  const char *Modifier) {
 245   const MachineOperand &MO = MI->getOperand(opNum);
 246   switch (MO.getType()) {
 247   case MachineOperand::MO_Register:
 248     if (MRegisterInfo::isPhysicalRegister(MO.getReg()))
 249       O << TM.getRegisterInfo()->get(MO.getReg()).Name;
 250     else
 251       assert(0 && "not implemented");
 252     break;
 253   case MachineOperand::MO_Immediate: {
 254     if (!Modifier || strcmp(Modifier, "no_hash") != 0)
 255       O << "#";
 256
 257     O << (int)MO.getImmedValue();
 258     break;
 259   }
 260   case MachineOperand::MO_MachineBasicBlock:
 261     printBasicBlockLabel(MO.getMachineBasicBlock());
 262     return;
 263   case MachineOperand::MO_GlobalAddress: {
 264     bool isCallOp = Modifier && !strcmp(Modifier, "call");
 265     GlobalValue *GV = MO.getGlobal();
 266     std::string Name = Mang->getValueName(GV);
 267     bool isExt = (GV->isDeclaration() || GV->hasWeakLinkage() ||
 268                   GV->hasLinkOnceLinkage());
 269     if (isExt && isCallOp && Subtarget->isTargetDarwin() &&
 270         TM.getRelocationModel() != Reloc::Static) {
 271       O << TAI->getPrivateGlobalPrefix() << Name << "$stub";
 272       FnStubs.insert(Name);
 273     } else
 274       O << Name;
 275
 276     if (GV->hasExternalWeakLinkage())
 277       ExtWeakSymbols.insert(GV);
 278     break;
 279   }
 280   case MachineOperand::MO_ExternalSymbol: {
 281     bool isCallOp = Modifier && !strcmp(Modifier, "call");
 282     std::string Name(TAI->getGlobalPrefix());
 283     Name += MO.getSymbolName();
 284     if (isCallOp && Subtarget->isTargetDarwin() &&
 285         TM.getRelocationModel() != Reloc::Static) {
 286       O << TAI->getPrivateGlobalPrefix() << Name << "$stub";
 287       FnStubs.insert(Name);
 288     } else
 289       O << Name;
 290     break;
 291   }
 292   case MachineOperand::MO_ConstantPoolIndex:
 293     O << TAI->getPrivateGlobalPrefix() << "CPI" << getFunctionNumber()
 294       << '_' << MO.getConstantPoolIndex();
 295     break;
 296   case MachineOperand::MO_JumpTableIndex:
 297     O << TAI->getPrivateGlobalPrefix() << "JTI" << getFunctionNumber()
 298       << '_' << MO.getJumpTableIndex();
 299     break;
 300   default:
 301     O << "<unknown operand type>"; abort (); break;
 302   }
 303 }
 304
 305 /// printSOImmOperand - SOImm is 4-bit rotate amount in bits 8-11 with 8-bit
 306 /// immediate in bits 0-7.
 307 void ARMAsmPrinter::printSOImmOperand(const MachineInstr *MI, int OpNum) {
 308   const MachineOperand &MO = MI->getOperand(OpNum);
 309   assert(MO.isImmediate() && (MO.getImmedValue() < (1 << 12)) &&
 310          "Not a valid so_imm value!");
 311   unsigned Imm = ARM_AM::getSOImmValImm(MO.getImmedValue());
 312   unsigned Rot = ARM_AM::getSOImmValRot(MO.getImmedValue());
 313
 314   // Print low-level immediate formation info, per
 315   // A5.1.3: "Data-processing operands - Immediate".
 316   if (Rot) {
 317     O << "#" << Imm << ", " << Rot;
 318     // Pretty printed version.
 319     O << ' ' << TAI->getCommentString() << ' ' << (int)ARM_AM::rotr32(Imm, Rot);
 320   } else {
 321     O << "#" << Imm;
 322   }
 323 }
 324
 325 // so_reg is a 4-operand unit corresponding to register forms of the A5.1
 326 // "Addressing Mode 1 - Data-processing operands" forms.  This includes:
 327 //    REG 0   0    - e.g. R5
 328 //    REG REG 0,SH_OPC     - e.g. R5, ROR R3
 329 //    REG 0   IMM,SH_OPC  - e.g. R5, LSL #3
 330 void ARMAsmPrinter::printSORegOperand(const MachineInstr *MI, int Op) {
 331   const MachineOperand &MO1 = MI->getOperand(Op);
 332   const MachineOperand &MO2 = MI->getOperand(Op+1);
 333   const MachineOperand &MO3 = MI->getOperand(Op+2);
 334
 335   assert(MRegisterInfo::isPhysicalRegister(MO1.getReg()));
 336   O << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 337
 338   // Print the shift opc.
 339   O << ", "
 340     << ARM_AM::getShiftOpcStr(ARM_AM::getSORegShOp(MO3.getImmedValue()))
 341     << " ";
 342
 343   if (MO2.getReg()) {
 344     assert(MRegisterInfo::isPhysicalRegister(MO2.getReg()));
 345     O << TM.getRegisterInfo()->get(MO2.getReg()).Name;
 346     assert(ARM_AM::getSORegOffset(MO3.getImm()) == 0);
 347   } else {
 348     O << "#" << ARM_AM::getSORegOffset(MO3.getImm());
 349   }
 350 }
 351
 352 void ARMAsmPrinter::printAddrMode2Operand(const MachineInstr *MI, int Op) {
 353   const MachineOperand &MO1 = MI->getOperand(Op);
 354   const MachineOperand &MO2 = MI->getOperand(Op+1);
 355   const MachineOperand &MO3 = MI->getOperand(Op+2);
 356
 357   if (!MO1.isRegister()) {   // FIXME: This is for CP entries, but isn't right.
 358     printOperand(MI, Op);
 359     return;
 360   }
 361
 362   O << "[" << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 363
 364   if (!MO2.getReg()) {
 365     if (ARM_AM::getAM2Offset(MO3.getImm()))  // Don't print +0.
 366       O << ", #"
 367         << (char)ARM_AM::getAM2Op(MO3.getImm())
 368         << ARM_AM::getAM2Offset(MO3.getImm());
 369     O << "]";
 370     return;
 371   }
 372
 373   O << ", "
 374     << (char)ARM_AM::getAM2Op(MO3.getImm())
 375     << TM.getRegisterInfo()->get(MO2.getReg()).Name;
 376
 377   if (unsigned ShImm = ARM_AM::getAM2Offset(MO3.getImm()))
 378     O << ", "
 379       << ARM_AM::getShiftOpcStr(ARM_AM::getAM2ShiftOpc(MO3.getImmedValue()))
 380       << " #" << ShImm;
 381   O << "]";
 382 }
 383
 384 void ARMAsmPrinter::printAddrMode2OffsetOperand(const MachineInstr *MI, int Op){
 385   const MachineOperand &MO1 = MI->getOperand(Op);
 386   const MachineOperand &MO2 = MI->getOperand(Op+1);
 387
 388   if (!MO1.getReg()) {
 389     if (ARM_AM::getAM2Offset(MO2.getImm()))  // Don't print +0.
 390       O << "#"
 391         << (char)ARM_AM::getAM2Op(MO2.getImm())
 392         << ARM_AM::getAM2Offset(MO2.getImm());
 393     return;
 394   }
 395
 396   O << (char)ARM_AM::getAM2Op(MO2.getImm())
 397     << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 398
 399   if (unsigned ShImm = ARM_AM::getAM2Offset(MO2.getImm()))
 400     O << ", "
 401       << ARM_AM::getShiftOpcStr(ARM_AM::getAM2ShiftOpc(MO2.getImmedValue()))
 402       << " #" << ShImm;
 403 }
 404
 405 void ARMAsmPrinter::printAddrMode3Operand(const MachineInstr *MI, int Op) {
 406   const MachineOperand &MO1 = MI->getOperand(Op);
 407   const MachineOperand &MO2 = MI->getOperand(Op+1);
 408   const MachineOperand &MO3 = MI->getOperand(Op+2);
 409
 410   assert(MRegisterInfo::isPhysicalRegister(MO1.getReg()));
 411   O << "[" << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 412
 413   if (MO2.getReg()) {
 414     O << ", "
 415       << (char)ARM_AM::getAM3Op(MO3.getImm())
 416       << TM.getRegisterInfo()->get(MO2.getReg()).Name
 417       << "]";
 418     return;
 419   }
 420
 421   if (unsigned ImmOffs = ARM_AM::getAM3Offset(MO3.getImm()))
 422     O << ", #"
 423       << (char)ARM_AM::getAM3Op(MO3.getImm())
 424       << ImmOffs;
 425   O << "]";
 426 }
 427
 428 void ARMAsmPrinter::printAddrMode3OffsetOperand(const MachineInstr *MI, int Op){
 429   const MachineOperand &MO1 = MI->getOperand(Op);
 430   const MachineOperand &MO2 = MI->getOperand(Op+1);
 431
 432   if (MO1.getReg()) {
 433     O << (char)ARM_AM::getAM3Op(MO2.getImm())
 434       << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 435     return;
 436   }
 437
 438   unsigned ImmOffs = ARM_AM::getAM3Offset(MO2.getImm());
 439   O << "#"
 440   << (char)ARM_AM::getAM3Op(MO2.getImm())
 441     << ImmOffs;
 442 }
 443
 444 void ARMAsmPrinter::printAddrMode4Operand(const MachineInstr *MI, int Op,
 445                                           const char *Modifier) {
 446   const MachineOperand &MO1 = MI->getOperand(Op);
 447   const MachineOperand &MO2 = MI->getOperand(Op+1);
 448   ARM_AM::AMSubMode Mode = ARM_AM::getAM4SubMode(MO2.getImm());
 449   if (Modifier && strcmp(Modifier, "submode") == 0) {
 450     if (MO1.getReg() == ARM::SP) {
 451       bool isLDM = (MI->getOpcode() == ARM::LDM ||
 452                     MI->getOpcode() == ARM::LDM_RET);
 453       O << ARM_AM::getAMSubModeAltStr(Mode, isLDM);
 454     } else
 455       O << ARM_AM::getAMSubModeStr(Mode);
 456   } else {
 457     printOperand(MI, Op);
 458     if (ARM_AM::getAM4WBFlag(MO2.getImm()))
 459       O << "!";
 460   }
 461 }
 462
 463 void ARMAsmPrinter::printAddrMode5Operand(const MachineInstr *MI, int Op,
 464                                           const char *Modifier) {
 465   const MachineOperand &MO1 = MI->getOperand(Op);
 466   const MachineOperand &MO2 = MI->getOperand(Op+1);
 467
 468   if (!MO1.isRegister()) {   // FIXME: This is for CP entries, but isn't right.
 469     printOperand(MI, Op);
 470     return;
 471   }
 472
 473   assert(MRegisterInfo::isPhysicalRegister(MO1.getReg()));
 474
 475   if (Modifier && strcmp(Modifier, "submode") == 0) {
 476     ARM_AM::AMSubMode Mode = ARM_AM::getAM5SubMode(MO2.getImm());
 477     if (MO1.getReg() == ARM::SP) {
 478       bool isFLDM = (MI->getOpcode() == ARM::FLDMD ||
 479                      MI->getOpcode() == ARM::FLDMS);
 480       O << ARM_AM::getAMSubModeAltStr(Mode, isFLDM);
 481     } else
 482       O << ARM_AM::getAMSubModeStr(Mode);
 483     return;
 484   } else if (Modifier && strcmp(Modifier, "base") == 0) {
 485     // Used for FSTM{D|S} and LSTM{D|S} operations.
 486     O << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 487     if (ARM_AM::getAM5WBFlag(MO2.getImm()))
 488       O << "!";
 489     return;
 490   }
 491
 492   O << "[" << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 493
 494   if (unsigned ImmOffs = ARM_AM::getAM5Offset(MO2.getImm())) {
 495     O << ", #"
 496       << (char)ARM_AM::getAM5Op(MO2.getImm())
 497       << ImmOffs*4;
 498   }
 499   O << "]";
 500 }
 501
 502 void ARMAsmPrinter::printAddrModePCOperand(const MachineInstr *MI, int Op,
 503                                            const char *Modifier) {
 504   if (Modifier && strcmp(Modifier, "label") == 0) {
 505     printPCLabel(MI, Op+1);
 506     return;
 507   }
 508
 509   const MachineOperand &MO1 = MI->getOperand(Op);
 510   assert(MRegisterInfo::isPhysicalRegister(MO1.getReg()));
 511   O << "[pc, +" << TM.getRegisterInfo()->get(MO1.getReg()).Name << "]";
 512 }
 513
 514 void
 515 ARMAsmPrinter::printThumbAddrModeRROperand(const MachineInstr *MI, int Op) {
 516   const MachineOperand &MO1 = MI->getOperand(Op);
 517   const MachineOperand &MO2 = MI->getOperand(Op+1);
 518   O << "[" << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 519   O << ", " << TM.getRegisterInfo()->get(MO2.getReg()).Name << "]";
 520 }
 521
 522 void
 523 ARMAsmPrinter::printThumbAddrModeRI5Operand(const MachineInstr *MI, int Op,
 524                                             unsigned Scale) {
 525   const MachineOperand &MO1 = MI->getOperand(Op);
 526   const MachineOperand &MO2 = MI->getOperand(Op+1);
 527   const MachineOperand &MO3 = MI->getOperand(Op+2);
 528
 529   if (!MO1.isRegister()) {   // FIXME: This is for CP entries, but isn't right.
 530     printOperand(MI, Op);
 531     return;
 532   }
 533
 534   O << "[" << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 535   if (MO3.getReg())
 536     O << ", " << TM.getRegisterInfo()->get(MO3.getReg()).Name;
 537   else if (unsigned ImmOffs = MO2.getImm()) {
 538     O << ", #" << ImmOffs;
 539     if (Scale > 1)
 540       O << " * " << Scale;
 541   }
 542   O << "]";
 543 }
 544
 545 void
 546 ARMAsmPrinter::printThumbAddrModeS1Operand(const MachineInstr *MI, int Op) {
 547   printThumbAddrModeRI5Operand(MI, Op, 1);
 548 }
 549 void
 550 ARMAsmPrinter::printThumbAddrModeS2Operand(const MachineInstr *MI, int Op) {
 551   printThumbAddrModeRI5Operand(MI, Op, 2);
 552 }
 553 void
 554 ARMAsmPrinter::printThumbAddrModeS4Operand(const MachineInstr *MI, int Op) {
 555   printThumbAddrModeRI5Operand(MI, Op, 4);
 556 }
 557
 558 void ARMAsmPrinter::printThumbAddrModeSPOperand(const MachineInstr *MI,int Op) {
 559   const MachineOperand &MO1 = MI->getOperand(Op);
 560   const MachineOperand &MO2 = MI->getOperand(Op+1);
 561   O << "[" << TM.getRegisterInfo()->get(MO1.getReg()).Name;
 562   if (unsigned ImmOffs = MO2.getImm())
 563     O << ", #" << ImmOffs << " * 4";
 564   O << "]";
 565 }
 566
 567 void ARMAsmPrinter::printCCOperand(const MachineInstr *MI, int opNum) {
 568   int CC = (int)MI->getOperand(opNum).getImmedValue();
 569   O << ARMCondCodeToString((ARMCC::CondCodes)CC);
 570 }
 571
 572 void ARMAsmPrinter::printPCLabel(const MachineInstr *MI, int opNum) {
 573   int Id = (int)MI->getOperand(opNum).getImmedValue();
 574   O << TAI->getPrivateGlobalPrefix() << "PC" << Id;
 575 }
 576
 577 void ARMAsmPrinter::printRegisterList(const MachineInstr *MI, int opNum) {
 578   O << "{";
 579   for (unsigned i = opNum, e = MI->getNumOperands(); i != e; ++i) {
 580     printOperand(MI, i);
 581     if (i != e-1) O << ", ";
 582   }
 583   O << "}";
 584 }
 585
 586 void ARMAsmPrinter::printCPInstOperand(const MachineInstr *MI, int OpNo,
 587                                        const char *Modifier) {
 588   assert(Modifier && "This operand only works with a modifier!");
 589   // There are two aspects to a CONSTANTPOOL_ENTRY operand, the label and the
 590   // data itself.
 591   if (!strcmp(Modifier, "label")) {
 592     unsigned ID = MI->getOperand(OpNo).getImm();
 593     O << TAI->getPrivateGlobalPrefix() << "CPI" << getFunctionNumber()
 594       << '_' << ID << ":\n";
 595   } else {
 596     assert(!strcmp(Modifier, "cpentry") && "Unknown modifier for CPE");
 597     unsigned CPI = MI->getOperand(OpNo).getConstantPoolIndex();
 598
 599     const MachineConstantPoolEntry &MCPE =  // Chasing pointers is fun?
 600       MI->getParent()->getParent()->getConstantPool()->getConstants()[CPI];
 601
 602     if (MCPE.isMachineConstantPoolEntry())
 603       EmitMachineConstantPoolValue(MCPE.Val.MachineCPVal);
 604     else
 605       EmitGlobalConstant(MCPE.Val.ConstVal);
 606   }
 607 }
 608
 609 void ARMAsmPrinter::printJTBlockOperand(const MachineInstr *MI, int OpNo) {
 610   const MachineOperand &MO1 = MI->getOperand(OpNo);
 611   const MachineOperand &MO2 = MI->getOperand(OpNo+1); // Unique Id
 612   unsigned JTI = MO1.getJumpTableIndex();
 613   O << TAI->getPrivateGlobalPrefix() << "JTI" << getFunctionNumber()
 614     << '_' << JTI << '_' << MO2.getImmedValue() << ":\n";
 615
 616   const char *JTEntryDirective = TAI->getJumpTableDirective();
 617   if (!JTEntryDirective)
 618     JTEntryDirective = TAI->getData32bitsDirective();
 619
 620   const MachineFunction *MF = MI->getParent()->getParent();
 621   MachineJumpTableInfo *MJTI = MF->getJumpTableInfo();
 622   const std::vector<MachineJumpTableEntry> &JT = MJTI->getJumpTables();
 623   const std::vector<MachineBasicBlock*> &JTBBs = JT[JTI].MBBs;
 624   bool UseSet= TAI->getSetDirective() && TM.getRelocationModel() == Reloc::PIC_;
 625   std::set<MachineBasicBlock*> JTSets;
 626   for (unsigned i = 0, e = JTBBs.size(); i != e; ++i) {
 627     MachineBasicBlock *MBB = JTBBs[i];
 628     if (UseSet && JTSets.insert(MBB).second)
 629       printSetLabel(JTI, MO2.getImmedValue(), MBB);
 630
 631     O << JTEntryDirective << ' ';
 632     if (UseSet)
 633       O << TAI->getPrivateGlobalPrefix() << getFunctionNumber()
 634         << '_' << JTI << '_' << MO2.getImmedValue()
 635         << "_set_" << MBB->getNumber();
 636     else if (TM.getRelocationModel() == Reloc::PIC_) {
 637       printBasicBlockLabel(MBB, false, false);
 638       // If the arch uses custom Jump Table directives, don't calc relative to JT
 639       if (!TAI->getJumpTableDirective())
 640         O << '-' << TAI->getPrivateGlobalPrefix() << "JTI"
 641           << getFunctionNumber() << '_' << JTI << '_' << MO2.getImmedValue();
 642     } else
 643       printBasicBlockLabel(MBB, false, false);
 644     if (i != e-1)
 645       O << '\n';
 646   }
 647 }
 648
 649
 650 bool ARMAsmPrinter::PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
 651                                     unsigned AsmVariant, const char *ExtraCode){
 652   // Does this asm operand have a single letter operand modifier?
 653   if (ExtraCode && ExtraCode[0]) {
 654     if (ExtraCode[1] != 0) return true; // Unknown modifier.
 655
 656     switch (ExtraCode[0]) {
 657     default: return true;  // Unknown modifier.
 658     case 'Q':
 659       if (TM.getTargetData()->isLittleEndian())
 660         break;
 661       // Fallthrough
 662     case 'R':
 663       if (TM.getTargetData()->isBigEndian())
 664         break;
 665       // Fallthrough
 666     case 'H': // Write second word of DI / DF reference.
 667       // Verify that this operand has two consecutive registers.
 668       if (!MI->getOperand(OpNo).isRegister() ||
 669           OpNo+1 == MI->getNumOperands() ||
 670           !MI->getOperand(OpNo+1).isRegister())
 671         return true;
 672       ++OpNo;   // Return the high-part.
 673     }
 674   }
 675
 676   printOperand(MI, OpNo);
 677   return false;
 678 }
 679
 680 void ARMAsmPrinter::printMachineInstruction(const MachineInstr *MI) {
 681   ++EmittedInsts;
 682
 683   if (MI->getOpcode() == ARM::CONSTPOOL_ENTRY) {
 684     if (!InCPMode && AFI->isThumbFunction()) {
 685       EmitAlignment(2);
 686       InCPMode = true;
 687     }
 688   } else {
 689     if (InCPMode && AFI->isThumbFunction()) {
 690       EmitAlignment(1);
 691       InCPMode = false;
 692     }
 693     O << "\t";
 694   }
 695
 696   // Call the autogenerated instruction printer routines.
 697   printInstruction(MI);
 698 }
 699
 700 bool ARMAsmPrinter::doInitialization(Module &M) {
 701   if (Subtarget->isTargetDarwin()) {
 702     // Emit initial debug information.
 703     DW.BeginModule(&M);
 704   }
 705
 706   return AsmPrinter::doInitialization(M);
 707 }
 708
 709 bool ARMAsmPrinter::doFinalization(Module &M) {
 710   const TargetData *TD = TM.getTargetData();
 711
 712   for (Module::const_global_iterator I = M.global_begin(), E = M.global_end();
 713        I != E; ++I) {
 714     if (!I->hasInitializer())   // External global require no code
 715       continue;
 716
 717     if (EmitSpecialLLVMGlobal(I)) {
 718       if (Subtarget->isTargetDarwin() &&
 719           TM.getRelocationModel() == Reloc::Static) {
 720         if (I->getName() == "llvm.global_ctors")
 721           O << ".reference .constructors_used\n";
 722         else if (I->getName() == "llvm.global_dtors")
 723           O << ".reference .destructors_used\n";
 724       }
 725       continue;
 726     }
 727
 728     std::string name = Mang->getValueName(I);
 729     Constant *C = I->getInitializer();
 730     unsigned Size = TD->getTypeSize(C->getType());
 731     unsigned Align = TD->getPreferredAlignmentLog(I);
 732
 733     if (I->hasHiddenVisibility())
 734       if (const char *Directive = TAI->getHiddenDirective())
 735         O << Directive << name << "\n";
 736     if (Subtarget->isTargetELF())
 737       O << "\t.type " << name << ",%object\n";
 738
 739     if (C->isNullValue()) {
 740       if (I->hasExternalLinkage()) {
 741         if (const char *Directive = TAI->getZeroFillDirective()) {
 742           O << "\t.globl\t" << name << "\n";
 743           O << Directive << "__DATA__, __common, " << name << ", "
 744             << Size << ", " << Align << "\n";
 745           continue;
 746         }
 747       }
 748
 749       if (!I->hasSection() &&
 750           (I->hasInternalLinkage() || I->hasWeakLinkage() ||
 751            I->hasLinkOnceLinkage())) {
 752         if (Size == 0) Size = 1;   // .comm Foo, 0 is undefined, avoid it.
 753         if (!NoZerosInBSS && TAI->getBSSSection())
 754           SwitchToDataSection(TAI->getBSSSection(), I);
 755         else
 756           SwitchToDataSection(TAI->getDataSection(), I);
 757         if (TAI->getLCOMMDirective() != NULL) {
 758           if (I->hasInternalLinkage()) {
 759             O << TAI->getLCOMMDirective() << name << "," << Size;
 760             if (Subtarget->isTargetDarwin())
 761               O << "," << Align;
 762           } else
 763             O << TAI->getCOMMDirective()  << name << "," << Size;
 764         } else {
 765           if (I->hasInternalLinkage())
 766             O << "\t.local\t" << name << "\n";
 767           O << TAI->getCOMMDirective()  << name << "," << Size;
 768           if (TAI->getCOMMDirectiveTakesAlignment())
 769             O << "," << (TAI->getAlignmentIsInBytes() ? (1 << Align) : Align);
 770         }
 771         O << "\t\t" << TAI->getCommentString() << " " << I->getName() << "\n";
 772         continue;
 773       }
 774     }
 775
 776     switch (I->getLinkage()) {
 777     case GlobalValue::LinkOnceLinkage:
 778     case GlobalValue::WeakLinkage:
 779       if (Subtarget->isTargetDarwin()) {
 780         O << "\t.globl " << name << "\n"
 781           << "\t.weak_definition " << name << "\n";
 782         SwitchToDataSection("\t.section __DATA,__const_coal,coalesced", I);
 783       } else {
 784         std::string SectionName("\t.section\t.llvm.linkonce.d." +
 785                                 name +
 786                                 ",\"aw\",%progbits");
 787         SwitchToDataSection(SectionName.c_str(), I);
 788         O << "\t.weak " << name << "\n";
 789       }
 790       break;
 791     case GlobalValue::AppendingLinkage:
 792       // FIXME: appending linkage variables should go into a section of
 793       // their name or something.  For now, just emit them as external.
 794     case GlobalValue::ExternalLinkage:
 795       O << "\t.globl " << name << "\n";
 796       // FALL THROUGH
 797     case GlobalValue::InternalLinkage: {
 798       if (I->isConstant()) {
 799         const ConstantArray *CVA = dyn_cast<ConstantArray>(C);
 800         if (TAI->getCStringSection() && CVA && CVA->isCString()) {
 801           SwitchToDataSection(TAI->getCStringSection(), I);
 802           break;
 803         }
 804       }
 805       // FIXME: special handling for ".ctors" & ".dtors" sections
 806       if (I->hasSection() &&
 807           (I->getSection() == ".ctors" ||
 808            I->getSection() == ".dtors")) {
 809         assert(!Subtarget->isTargetDarwin());
 810         std::string SectionName = ".section " + I->getSection();
 811         SectionName += ",\"aw\",%progbits";
 812         SwitchToDataSection(SectionName.c_str());
 813       } else {
 814         if (C->isNullValue() && !NoZerosInBSS && TAI->getBSSSection())
 815           SwitchToDataSection(TAI->getBSSSection(), I);
 816         else
 817           SwitchToDataSection(TAI->getDataSection(), I);
 818       }
 819
 820       break;
 821     }
 822     default:
 823       assert(0 && "Unknown linkage type!");
 824       break;
 825     }
 826
 827     EmitAlignment(Align, I);
 828     O << name << ":\t\t\t\t" << TAI->getCommentString() << " " << I->getName()
 829       << "\n";
 830     if (TAI->hasDotTypeDotSizeDirective())
 831       O << "\t.size " << name << ", " << Size << "\n";
 832     // If the initializer is a extern weak symbol, remember to emit the weak
 833     // reference!
 834     if (const GlobalValue *GV = dyn_cast<GlobalValue>(C))
 835       if (GV->hasExternalWeakLinkage())
 836       ExtWeakSymbols.insert(GV);
 837
 838     EmitGlobalConstant(C);
 839     O << '\n';
 840   }
 841
 842   if (Subtarget->isTargetDarwin()) {
 843     SwitchToDataSection("");
 844
 845     // Output stubs for dynamically-linked functions
 846     unsigned j = 1;
 847     for (std::set<std::string>::iterator i = FnStubs.begin(), e = FnStubs.end();
 848          i != e; ++i, ++j) {
 849       if (TM.getRelocationModel() == Reloc::PIC_)
 850         SwitchToTextSection(".section __TEXT,__picsymbolstub4,symbol_stubs,"
 851                             "none,16", 0);
 852       else
 853         SwitchToTextSection(".section __TEXT,__symbol_stub4,symbol_stubs,"
 854                             "none,12", 0);
 855
 856       EmitAlignment(2);
 857       O << "\t.code\t32\n";
 858
 859       O << "L" << *i << "$stub:\n";
 860       O << "\t.indirect_symbol " << *i << "\n";
 861       O << "\tldr ip, L" << *i << "$slp\n";
 862       if (TM.getRelocationModel() == Reloc::PIC_) {
 863         O << "L" << *i << "$scv:\n";
 864         O << "\tadd ip, pc, ip\n";
 865       }
 866       O << "\tldr pc, [ip, #0]\n";
 867       O << "L" << *i << "$slp:\n";
 868       if (TM.getRelocationModel() == Reloc::PIC_)
 869         O << "\t.long\tL" << *i << "$lazy_ptr-(L" << *i << "$scv+8)\n";
 870       else
 871         O << "\t.long\tL" << *i << "$lazy_ptr\n";
 872       SwitchToDataSection(".lazy_symbol_pointer", 0);
 873       O << "L" << *i << "$lazy_ptr:\n";
 874       O << "\t.indirect_symbol " << *i << "\n";
 875       O << "\t.long\tdyld_stub_binding_helper\n";
 876     }
 877     O << "\n";
 878
 879     // Output non-lazy-pointers for external and common global variables.
 880     if (GVNonLazyPtrs.begin() != GVNonLazyPtrs.end())
 881       SwitchToDataSection(".non_lazy_symbol_pointer", 0);
 882     for (std::set<std::string>::iterator i = GVNonLazyPtrs.begin(),
 883            e = GVNonLazyPtrs.end(); i != e; ++i) {
 884       O << "L" << *i << "$non_lazy_ptr:\n";
 885       O << "\t.indirect_symbol " << *i << "\n";
 886       O << "\t.long\t0\n";
 887     }
 888
 889     // Emit initial debug information.
 890     DW.EndModule();
 891
 892     // Funny Darwin hack: This flag tells the linker that no global symbols
 893     // contain code that falls through to other global symbols (e.g. the obvious
 894     // implementation of multiple entry points).  If this doesn't occur, the
 895     // linker can safely perform dead code stripping.  Since LLVM never
 896     // generates code that does this, it is always safe to set.
 897     O << "\t.subsections_via_symbols\n";
 898   }
 899
 900   AsmPrinter::doFinalization(M);
 901   return false; // success
 902 }