lib/Target/X86/X86AsmPrinter.cpp

   1 //===-- X86/Printer.cpp - Convert X86 code to human readable rep. ---------===//
   2 //
   3 // This file contains a printer that converts from our internal representation
   4 // of LLVM code to a nice human readable form that is suitable for debuggging.
   5 //
   6 //===----------------------------------------------------------------------===//
   7
   8 #include "X86.h"
   9 #include "X86InstrInfo.h"
  10 #include "llvm/Pass.h"
  11 #include "llvm/Function.h"
  12 #include "llvm/Target/TargetMachine.h"
  13 #include "llvm/CodeGen/MachineFunction.h"
  14 #include "llvm/CodeGen/MachineInstr.h"
  15 #include "Support/Statistic.h"
  16
  17 namespace {
  18   struct Printer : public FunctionPass {
  19     TargetMachine &TM;
  20     std::ostream &O;
  21
  22     Printer(TargetMachine &tm, std::ostream &o) : TM(tm), O(o) {}
  23
  24     bool runOnFunction(Function &F);
  25   };
  26 }
  27
  28 /// createX86CodePrinterPass - Print out the specified machine code function to
  29 /// the specified stream.  This function should work regardless of whether or
  30 /// not the function is in SSA form or not.
  31 ///
  32 Pass *createX86CodePrinterPass(TargetMachine &TM, std::ostream &O) {
  33   return new Printer(TM, O);
  34 }
  35
  36
  37 /// runOnFunction - This uses the X86InstructionInfo::print method
  38 /// to print assembly for each instruction.
  39 bool Printer::runOnFunction (Function & F)
  40 {
  41   static unsigned bbnumber = 0;
  42   MachineFunction & MF = MachineFunction::get (&F);
  43   const MachineInstrInfo & MII = TM.getInstrInfo ();
  44
  45   // Print out labels for the function.
  46   O << "\t.globl\t" << F.getName () << "\n";
  47   O << "\t.type\t" << F.getName () << ", @function\n";
  48   O << F.getName () << ":\n";
  49
  50   // Print out code for the function.
  51   for (MachineFunction::const_iterator bb_i = MF.begin (), bb_e = MF.end ();
  52        bb_i != bb_e; ++bb_i)
  53     {
  54       // Print a label for the basic block.
  55       O << ".BB" << bbnumber++ << ":\n";
  56       for (MachineBasicBlock::const_iterator i_i = bb_i->begin (), i_e =
  57            bb_i->end (); i_i != i_e; ++i_i)
  58         {
  59           // Print the assembly for the instruction.
  60           O << "\t";
  61           MII.print(*i_i, O, TM);
  62         }
  63     }
  64
  65   // We didn't modify anything.
  66   return false;
  67 }
  68
  69 static bool isReg(const MachineOperand &MO) {
  70   return MO.getType() == MachineOperand::MO_VirtualRegister ||
  71          MO.getType() == MachineOperand::MO_MachineRegister;
  72 }
  73
  74 static bool isImmediate(const MachineOperand &MO) {
  75   return MO.getType() == MachineOperand::MO_SignExtendedImmed ||
  76          MO.getType() == MachineOperand::MO_UnextendedImmed;
  77 }
  78
  79 static bool isPCRelativeDisp(const MachineOperand &MO) {
  80   return MO.getType() == MachineOperand::MO_PCRelativeDisp;
  81 }
  82
  83 static bool isScale(const MachineOperand &MO) {
  84   return isImmediate(MO) &&
  85            (MO.getImmedValue() == 1 || MO.getImmedValue() == 2 ||
  86             MO.getImmedValue() == 4 || MO.getImmedValue() == 8);
  87 }
  88
  89 static bool isMem(const MachineInstr *MI, unsigned Op) {
  90   return Op+4 <= MI->getNumOperands() &&
  91          isReg(MI->getOperand(Op  )) && isScale(MI->getOperand(Op+1)) &&
  92          isReg(MI->getOperand(Op+2)) && isImmediate(MI->getOperand(Op+3));
  93 }
  94
  95 static void printOp(std::ostream &O, const MachineOperand &MO,
  96                     const MRegisterInfo &RI) {
  97   switch (MO.getType()) {
  98   case MachineOperand::MO_VirtualRegister:
  99     if (Value *V = MO.getVRegValueOrNull()) {
 100       O << "<" << V->getName() << ">";
 101       return;
 102     }
 103   case MachineOperand::MO_MachineRegister:
 104     if (MO.getReg() < MRegisterInfo::FirstVirtualRegister)
 105       O << RI.get(MO.getReg()).Name;
 106     else
 107       O << "%reg" << MO.getReg();
 108     return;
 109
 110   case MachineOperand::MO_SignExtendedImmed:
 111   case MachineOperand::MO_UnextendedImmed:
 112     O << (int)MO.getImmedValue();
 113     return;
 114   case MachineOperand::MO_PCRelativeDisp:
 115     O << "<" << MO.getVRegValue()->getName() << ">";
 116     return;
 117   default:
 118     O << "<unknown op ty>"; return;
 119   }
 120 }
 121
 122 static const std::string sizePtr (const MachineInstrDescriptor &Desc) {
 123   switch (Desc.TSFlags & X86II::ArgMask) {
 124     case X86II::Arg8:   return "BYTE PTR";
 125     case X86II::Arg16:  return "WORD PTR";
 126     case X86II::Arg32:  return "DWORD PTR";
 127     case X86II::Arg64:  return "QWORD PTR";
 128     case X86II::Arg80:  return "XWORD PTR";
 129     case X86II::Arg128: return "128BIT PTR";  // dunno what the real one is
 130     default: return "<SIZE?> PTR"; // crack being smoked
 131   }
 132 }
 133
 134 static void printMemReference(std::ostream &O, const MachineInstr *MI,
 135                               unsigned Op, const MRegisterInfo &RI) {
 136   assert(isMem(MI, Op) && "Invalid memory reference!");
 137   const MachineOperand &BaseReg  = MI->getOperand(Op);
 138   const MachineOperand &Scale    = MI->getOperand(Op+1);
 139   const MachineOperand &IndexReg = MI->getOperand(Op+2);
 140   const MachineOperand &Disp     = MI->getOperand(Op+3);
 141
 142   O << "[";
 143   bool NeedPlus = false;
 144   if (BaseReg.getReg()) {
 145     printOp(O, BaseReg, RI);
 146     NeedPlus = true;
 147   }
 148
 149   if (IndexReg.getReg()) {
 150     if (NeedPlus) O << " + ";
 151     if (Scale.getImmedValue() != 1)
 152       O << Scale.getImmedValue() << "*";
 153     printOp(O, IndexReg, RI);
 154     NeedPlus = true;
 155   }
 156
 157   if (Disp.getImmedValue()) {
 158     if (NeedPlus) O << " + ";
 159     printOp(O, Disp, RI);
 160   }
 161   O << "]";
 162 }
 163
 164 // print - Print out an x86 instruction in intel syntax
 165 void X86InstrInfo::print(const MachineInstr *MI, std::ostream &O,
 166                          const TargetMachine &TM) const {
 167   unsigned Opcode = MI->getOpcode();
 168   const MachineInstrDescriptor &Desc = get(Opcode);
 169
 170   if (Opcode == X86::PHI) {
 171     printOp(O, MI->getOperand(0), RI);
 172     O << " = phi ";
 173     for (unsigned i = 1, e = MI->getNumOperands(); i != e; i+=2) {
 174       if (i != 1) O << ", ";
 175       O << "[";
 176       printOp(O, MI->getOperand(i), RI);
 177       O << ", ";
 178       printOp(O, MI->getOperand(i+1), RI);
 179       O << "]";
 180     }
 181     O << "\n";
 182     return;
 183   }
 184
 185
 186   switch (Desc.TSFlags & X86II::FormMask) {
 187   case X86II::RawFrm:
 188     // The accepted forms of Raw instructions are:
 189     //   1. nop     - No operand required
 190     //   2. jmp foo - PC relative displacement operand
 191     //
 192     assert(MI->getNumOperands() == 0 ||
 193            (MI->getNumOperands() == 1 && isPCRelativeDisp(MI->getOperand(0))) &&
 194            "Illegal raw instruction!");
 195     O << getName(MI->getOpCode()) << " ";
 196
 197     if (MI->getNumOperands() == 1) {
 198       printOp(O, MI->getOperand(0), RI);
 199     }
 200     O << "\n";
 201     return;
 202
 203   case X86II::AddRegFrm: {
 204     // There are currently two forms of acceptable AddRegFrm instructions.
 205     // Either the instruction JUST takes a single register (like inc, dec, etc),
 206     // or it takes a register and an immediate of the same size as the register
 207     // (move immediate f.e.).  Note that this immediate value might be stored as
 208     // an LLVM value, to represent, for example, loading the address of a global
 209     // into a register.
 210     //
 211     assert(isReg(MI->getOperand(0)) &&
 212            (MI->getNumOperands() == 1 ||
 213             (MI->getNumOperands() == 2 &&
 214              (MI->getOperand(1).getVRegValueOrNull() ||
 215               isImmediate(MI->getOperand(1))))) &&
 216            "Illegal form for AddRegFrm instruction!");
 217
 218     unsigned Reg = MI->getOperand(0).getReg();
 219
 220     O << getName(MI->getOpCode()) << " ";
 221     printOp(O, MI->getOperand(0), RI);
 222     if (MI->getNumOperands() == 2) {
 223       O << ", ";
 224       printOp(O, MI->getOperand(1), RI);
 225     }
 226     O << "\n";
 227     return;
 228   }
 229   case X86II::MRMDestReg: {
 230     // There are two acceptable forms of MRMDestReg instructions, those with 3
 231     // and 2 operands:
 232     //
 233     // 3 Operands: in this form, the first two registers (the destination, and
 234     // the first operand) should be the same, post register allocation.  The 3rd
 235     // operand is an additional input.  This should be for things like add
 236     // instructions.
 237     //
 238     // 2 Operands: this is for things like mov that do not read a second input
 239     //
 240     assert(isReg(MI->getOperand(0)) &&
 241            (MI->getNumOperands() == 2 ||
 242             (MI->getNumOperands() == 3 && isReg(MI->getOperand(1)))) &&
 243            isReg(MI->getOperand(MI->getNumOperands()-1))
 244            && "Bad format for MRMDestReg!");
 245     if (MI->getNumOperands() == 3 &&
 246         MI->getOperand(0).getReg() != MI->getOperand(1).getReg())
 247       O << "**";
 248
 249     O << getName(MI->getOpCode()) << " ";
 250     printOp(O, MI->getOperand(0), RI);
 251     O << ", ";
 252     printOp(O, MI->getOperand(MI->getNumOperands()-1), RI);
 253     O << "\n";
 254     return;
 255   }
 256
 257   case X86II::MRMDestMem: {
 258     // These instructions are the same as MRMDestReg, but instead of having a
 259     // register reference for the mod/rm field, it's a memory reference.
 260     //
 261     assert(isMem(MI, 0) && MI->getNumOperands() == 4+1 &&
 262            isReg(MI->getOperand(4)) && "Bad format for MRMDestMem!");
 263
 264     O << getName(MI->getOpCode()) << " " << sizePtr (Desc) << " ";
 265     printMemReference(O, MI, 0, RI);
 266     O << ", ";
 267     printOp(O, MI->getOperand(4), RI);
 268     O << "\n";
 269     return;
 270   }
 271
 272   case X86II::MRMSrcReg: {
 273     // There is a two forms that are acceptable for MRMSrcReg instructions,
 274     // those with 3 and 2 operands:
 275     //
 276     // 3 Operands: in this form, the last register (the second input) is the
 277     // ModR/M input.  The first two operands should be the same, post register
 278     // allocation.  This is for things like: add r32, r/m32
 279     //
 280     // 2 Operands: this is for things like mov that do not read a second input
 281     //
 282     assert(isReg(MI->getOperand(0)) &&
 283            isReg(MI->getOperand(1)) &&
 284            (MI->getNumOperands() == 2 ||
 285             (MI->getNumOperands() == 3 && isReg(MI->getOperand(2))))
 286            && "Bad format for MRMDestReg!");
 287     if (MI->getNumOperands() == 3 &&
 288         MI->getOperand(0).getReg() != MI->getOperand(1).getReg())
 289       O << "**";
 290
 291     O << getName(MI->getOpCode()) << " ";
 292     printOp(O, MI->getOperand(0), RI);
 293     O << ", ";
 294     printOp(O, MI->getOperand(MI->getNumOperands()-1), RI);
 295     O << "\n";
 296     return;
 297   }
 298
 299   case X86II::MRMSrcMem: {
 300     // These instructions are the same as MRMSrcReg, but instead of having a
 301     // register reference for the mod/rm field, it's a memory reference.
 302     //
 303     assert(isReg(MI->getOperand(0)) &&
 304            (MI->getNumOperands() == 1+4 && isMem(MI, 1)) ||
 305            (MI->getNumOperands() == 2+4 && isReg(MI->getOperand(1)) &&
 306             isMem(MI, 2))
 307            && "Bad format for MRMDestReg!");
 308     if (MI->getNumOperands() == 2+4 &&
 309         MI->getOperand(0).getReg() != MI->getOperand(1).getReg())
 310       O << "**";
 311
 312     O << getName(MI->getOpCode()) << " ";
 313     printOp(O, MI->getOperand(0), RI);
 314     O << ", " << sizePtr (Desc) << " ";
 315     printMemReference(O, MI, MI->getNumOperands()-4, RI);
 316     O << "\n";
 317     return;
 318   }
 319
 320   case X86II::MRMS0r: case X86II::MRMS1r:
 321   case X86II::MRMS2r: case X86II::MRMS3r:
 322   case X86II::MRMS4r: case X86II::MRMS5r:
 323   case X86II::MRMS6r: case X86II::MRMS7r: {
 324     // In this form, the following are valid formats:
 325     //  1. sete r
 326     //  2. cmp reg, immediate
 327     //  2. shl rdest, rinput  <implicit CL or 1>
 328     //  3. sbb rdest, rinput, immediate   [rdest = rinput]
 329     //
 330     assert(MI->getNumOperands() > 0 && MI->getNumOperands() < 4 &&
 331            isReg(MI->getOperand(0)) && "Bad MRMSxR format!");
 332     assert((MI->getNumOperands() != 2 ||
 333             isReg(MI->getOperand(1)) || isImmediate(MI->getOperand(1))) &&
 334            "Bad MRMSxR format!");
 335     assert((MI->getNumOperands() < 3 ||
 336             (isReg(MI->getOperand(1)) && isImmediate(MI->getOperand(2)))) &&
 337            "Bad MRMSxR format!");
 338
 339     if (MI->getNumOperands() > 1 && isReg(MI->getOperand(1)) &&
 340         MI->getOperand(0).getReg() != MI->getOperand(1).getReg())
 341       O << "**";
 342
 343     O << getName(MI->getOpCode()) << " ";
 344     printOp(O, MI->getOperand(0), RI);
 345     if (isImmediate(MI->getOperand(MI->getNumOperands()-1))) {
 346       O << ", ";
 347       printOp(O, MI->getOperand(MI->getNumOperands()-1), RI);
 348     }
 349     O << "\n";
 350
 351     return;
 352   }
 353
 354   default:
 355     O << "\t\t\t-"; MI->print(O, TM); break;
 356   }
 357 }