lib/CodeGen/MachineCodeEmitter.cpp

   1 //===-- MachineCodeEmitter.cpp - Implement the MachineCodeEmitter itf -----===//
   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 implements the MachineCodeEmitter interface.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "llvm/CodeGen/MachineCodeEmitter.h"
  15 #include "llvm/CodeGen/MachineFunction.h"
  16 #include "llvm/Function.h"
  17 #include <fstream>
  18 using namespace llvm;
  19
  20 namespace {
  21   struct DebugMachineCodeEmitter : public MachineCodeEmitter {
  22     void startFunction(MachineFunction &F) {
  23       std::cout << "\n**** Writing machine code for function: "
  24                 << F.getFunction()->getName() << "\n";
  25     }
  26     void finishFunction(MachineFunction &F) {
  27       std::cout << "\n";
  28     }
  29     void startFunctionStub(const Function &F, unsigned StubSize) {
  30       std::cout << "\n--- Function stub for function: " << F.getName() << "\n";
  31     }
  32     void *finishFunctionStub(const Function &F) {
  33       std::cout << "\n";
  34       return 0;
  35     }
  36
  37     void emitByte(unsigned char B) {
  38       std::cout << "0x" << std::hex << (unsigned int)B << std::dec << " ";
  39     }
  40     void emitWord(unsigned W) {
  41       std::cout << "0x" << std::hex << W << std::dec << " ";
  42     }
  43
  44     uint64_t getGlobalValueAddress(GlobalValue *V) { return 0; }
  45     uint64_t getGlobalValueAddress(const std::string &Name) { return 0; }
  46     uint64_t getConstantPoolEntryAddress(unsigned Num) { return 0; }
  47     uint64_t getCurrentPCValue() { return 0; }
  48
  49     // forceCompilationOf - Force the compilation of the specified function, and
  50     // return its address, because we REALLY need the address now.
  51     //
  52     // FIXME: This is JIT specific!
  53     //
  54     virtual uint64_t forceCompilationOf(Function *F) {
  55       return 0;
  56     }
  57   };
  58
  59   class FilePrinterEmitter : public MachineCodeEmitter {
  60     std::ofstream actual;
  61     std::ostream &o;
  62     MachineCodeEmitter &MCE;
  63     unsigned counter;
  64     unsigned values[4];
  65
  66   public:
  67     FilePrinterEmitter(MachineCodeEmitter &M, std::ostream &os)
  68       : o(os), MCE(M), counter(0) {
  69       openActual();
  70     }
  71
  72     ~FilePrinterEmitter() {
  73       o << "\n";
  74       actual.close();
  75     }
  76
  77     void openActual() {
  78       actual.open("lli.actual.obj");
  79       if (!actual.good()) {
  80         std::cerr << "Cannot open 'lli.actual.obj' for writing\n";
  81         abort();
  82       }
  83     }
  84
  85     void startFunction(MachineFunction &F) {
  86       // resolve any outstanding calls
  87       MCE.startFunction(F);
  88     }
  89     void finishFunction(MachineFunction &F) {
  90       MCE.finishFunction(F);
  91     }
  92
  93     void emitConstantPool(MachineConstantPool *MCP) {
  94       MCE.emitConstantPool(MCP);
  95     }
  96
  97     void startFunctionStub(const Function &F, unsigned StubSize) {
  98       MCE.startFunctionStub(F, StubSize);
  99     }
 100
 101     void *finishFunctionStub(const Function &F) {
 102       return MCE.finishFunctionStub(F);
 103     }
 104
 105     void emitByte(unsigned char B) {
 106       MCE.emitByte(B);
 107       actual << B; actual.flush();
 108
 109       values[counter] = (unsigned int) B;
 110       if (++counter % 4 == 0 && counter != 0) {
 111         o << std::hex;
 112         for (unsigned i=0; i<4; ++i) {
 113           if (values[i] < 16) o << "0";
 114           o << values[i] << " ";
 115         }
 116
 117         o << std::dec << "\t";
 118         for (unsigned i=0; i<4; ++i) {
 119           for (int j=7; j>=0; --j) {
 120             o << ((values[i] >> j) & 1);
 121           }
 122           o << " ";
 123         }
 124
 125         o << "\n";
 126
 127         unsigned instr = 0;
 128         for (unsigned i=0; i<4; ++i)
 129           instr |= values[i] << (i*8);
 130
 131         o << "--- * --- * --- * --- * ---\n";
 132         counter %= 4;
 133       }
 134     }
 135
 136     void emitWord(unsigned W) {
 137       MCE.emitWord(W);
 138     }
 139     uint64_t getGlobalValueAddress(GlobalValue *V) {
 140       return MCE.getGlobalValueAddress(V);
 141     }
 142     uint64_t getGlobalValueAddress(const std::string &Name) {
 143       return MCE.getGlobalValueAddress(Name);
 144     }
 145     uint64_t getConstantPoolEntryAddress(unsigned Num) {
 146       return MCE.getConstantPoolEntryAddress(Num);
 147     }
 148     uint64_t getCurrentPCValue() {
 149       return MCE.getCurrentPCValue();
 150     }
 151     // forceCompilationOf - Force the compilation of the specified function, and
 152     // return its address, because we REALLY need the address now.
 153     //
 154     // FIXME: This is JIT specific!
 155     //
 156     virtual uint64_t forceCompilationOf(Function *F) {
 157       return MCE.forceCompilationOf(F);
 158     }
 159   };
 160 }
 161
 162 /// createDebugMachineCodeEmitter - Return a dynamically allocated machine
 163 /// code emitter, which just prints the opcodes and fields out the cout.  This
 164 /// can be used for debugging users of the MachineCodeEmitter interface.
 165 ///
 166 MachineCodeEmitter *
 167 MachineCodeEmitter::createDebugEmitter() {
 168   return new DebugMachineCodeEmitter();
 169 }
 170
 171 MachineCodeEmitter *
 172 MachineCodeEmitter::createFilePrinterEmitter(MachineCodeEmitter &MCE) {
 173   return new FilePrinterEmitter(MCE, std::cerr);
 174 }