lib/Target/WebAssembly/InstPrinter/WebAssemblyInstPrinter.cpp

   1 //=- WebAssemblyInstPrinter.cpp - WebAssembly assembly instruction printing -=//
   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 /// \file
  11 /// \brief Print MCInst instructions to wasm format.
  12 ///
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "InstPrinter/WebAssemblyInstPrinter.h"
  16 #include "WebAssembly.h"
  17 #include "WebAssemblyMachineFunctionInfo.h"
  18 #include "llvm/MC/MCExpr.h"
  19 #include "llvm/MC/MCInst.h"
  20 #include "llvm/MC/MCInstrInfo.h"
  21 #include "llvm/MC/MCSubtargetInfo.h"
  22 #include "llvm/MC/MCSymbol.h"
  23 #include "llvm/Support/ErrorHandling.h"
  24 #include "llvm/Support/FormattedStream.h"
  25 #include "llvm/Target/TargetRegisterInfo.h"
  26 #include <cctype>
  27 using namespace llvm;
  28
  29 #define DEBUG_TYPE "asm-printer"
  30
  31 #include "WebAssemblyGenAsmWriter.inc"
  32
  33 WebAssemblyInstPrinter::WebAssemblyInstPrinter(const MCAsmInfo &MAI,
  34                                                const MCInstrInfo &MII,
  35                                                const MCRegisterInfo &MRI)
  36     : MCInstPrinter(MAI, MII, MRI) {}
  37
  38 void WebAssemblyInstPrinter::printRegName(raw_ostream &OS,
  39                                           unsigned RegNo) const {
  40   assert(RegNo != WebAssemblyFunctionInfo::UnusedReg);
  41   // FIXME: Revisit whether we actually print the get_local explicitly.
  42   OS << "(get_local " << RegNo << ")";
  43 }
  44
  45 void WebAssemblyInstPrinter::printInst(const MCInst *MI, raw_ostream &OS,
  46                                        StringRef Annot,
  47                                        const MCSubtargetInfo &STI) {
  48   printInstruction(MI, OS);
  49
  50   const MCInstrDesc &Desc = MII.get(MI->getOpcode());
  51   if (Desc.isVariadic())
  52     for (unsigned i = Desc.getNumOperands(), e = MI->getNumOperands(); i < e;
  53          ++i) {
  54       OS << ", ";
  55       printOperand(MI, i, OS);
  56     }
  57
  58   printAnnotation(OS, Annot);
  59
  60   unsigned NumDefs = MII.get(MI->getOpcode()).getNumDefs();
  61   assert(NumDefs <= 1 &&
  62          "Instructions with multiple result values not implemented");
  63
  64   // FIXME: Revisit whether we actually print the set_local explicitly.
  65   if (NumDefs != 0) {
  66     unsigned WAReg = MI->getOperand(0).getReg();
  67     // Only print the set_local if the register is used.
  68     // TODO: Revisit this once the spec explains what should happen here.
  69     if (WAReg != WebAssemblyFunctionInfo::UnusedReg)
  70       OS << "\n"
  71             "\t" "set_local " << WAReg << ", $pop";
  72   }
  73 }
  74
  75 static std::string toString(const APFloat &FP) {
  76   static const size_t BufBytes = 128;
  77   char buf[BufBytes];
  78   if (FP.isNaN())
  79     assert((FP.bitwiseIsEqual(APFloat::getQNaN(FP.getSemantics())) ||
  80             FP.bitwiseIsEqual(
  81                 APFloat::getQNaN(FP.getSemantics(), /*Negative=*/true))) &&
  82            "convertToHexString handles neither SNaN nor NaN payloads");
  83   // Use C99's hexadecimal floating-point representation.
  84   auto Written = FP.convertToHexString(
  85       buf, /*hexDigits=*/0, /*upperCase=*/false, APFloat::rmNearestTiesToEven);
  86   (void)Written;
  87   assert(Written != 0);
  88   assert(Written < BufBytes);
  89   return buf;
  90 }
  91
  92 void WebAssemblyInstPrinter::printOperand(const MCInst *MI, unsigned OpNo,
  93                                           raw_ostream &O) {
  94   const MCOperand &Op = MI->getOperand(OpNo);
  95   if (Op.isReg()) {
  96     if (OpNo >= MII.get(MI->getOpcode()).getNumDefs())
  97       printRegName(O, Op.getReg());
  98     else {
  99       if (Op.getReg() != WebAssemblyFunctionInfo::UnusedReg)
 100         O << "$push";
 101       else
 102         O << "$discard";
 103     }
 104   } else if (Op.isImm())
 105     O << Op.getImm();
 106   else if (Op.isFPImm())
 107     O << toString(APFloat(Op.getFPImm()));
 108   else {
 109     assert(Op.isExpr() && "unknown operand kind in printOperand");
 110     Op.getExpr()->print(O, &MAI);
 111   }
 112 }