1 files changed, 176 insertions, 0 deletions
diff --git a/lib/CodeGen/MachineCodeEmitter.cpp b/lib/CodeGen/MachineCodeEmitter.cpp
new file mode 100644
index 0000000000..1090d762d7
--- /dev/null
+++ b/lib/CodeGen/MachineCodeEmitter.cpp
@@ -0,0 +1,176 @@
+//===-- MachineCodeEmitter.cpp - Implement the MachineCodeEmitter itf -----===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file was developed by the LLVM research group and is distributed under
+// the University of Illinois Open Source License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file implements the MachineCodeEmitter interface.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/CodeGen/MachineCodeEmitter.h"
+#include "llvm/CodeGen/MachineFunction.h"
+#include "llvm/Function.h"
+#include <fstream>
+#include <iostream>
+
+using namespace llvm;
+
+namespace {
+  struct DebugMachineCodeEmitter : public MachineCodeEmitter {
+    void startFunction(MachineFunction &F) {
+      std::cout << "\n**** Writing machine code for function: "
+                << F.getFunction()->getName() << "\n";
+    }
+    void finishFunction(MachineFunction &F) {
+      std::cout << "\n";
+    }
+    void startFunctionStub(unsigned StubSize) {
+      std::cout << "\n--- Function stub:\n";
+    }
+    void *finishFunctionStub(const Function *F) {
+      std::cout << "\n--- End of stub for Function\n";
+      return 0;
+    }
+
+    void emitByte(unsigned char B) {
+      std::cout << "0x" << std::hex << (unsigned int)B << std::dec << " ";
+    }
+    void emitWord(unsigned W) {
+      std::cout << "0x" << std::hex << W << std::dec << " ";
+    }
+    void emitWordAt(unsigned W, unsigned *Ptr) {
+      std::cout << "0x" << std::hex << W << std::dec << " (at "
+                << (void*) Ptr << ") ";
+    }
+
+    void addRelocation(const MachineRelocation &MR) {
+      std::cout << "<relocation> ";
+    }
+
+    virtual unsigned char* allocateGlobal(unsigned size, unsigned alignment)
+    { return 0; }
+
+    uint64_t getConstantPoolEntryAddress(unsigned Num) { return 0; }
+    uint64_t getCurrentPCValue() { return 0; }
+    uint64_t getCurrentPCOffset() { return 0; }
+  };
+
+  class FilePrinterEmitter : public MachineCodeEmitter {
+    std::ofstream actual;
+    std::ostream &o;
+    MachineCodeEmitter &MCE;
+    unsigned counter;
+    unsigned values[4];
+
+  public:
+    FilePrinterEmitter(MachineCodeEmitter &M, std::ostream &os)
+      : o(os), MCE(M), counter(0) {
+      openActual();
+    }
+
+    ~FilePrinterEmitter() {
+      o << "\n";
+      actual.close();
+    }
+
+    void openActual() {
+      actual.open("lli.actual.obj");
+      if (!actual.good()) {
+        std::cerr << "Cannot open 'lli.actual.obj' for writing\n";
+        abort();
+      }
+    }
+
+    void startFunction(MachineFunction &F) {
+      // resolve any outstanding calls
+      MCE.startFunction(F);
+    }
+    void finishFunction(MachineFunction &F) {
+      MCE.finishFunction(F);
+    }
+
+    void emitConstantPool(MachineConstantPool *MCP) {
+      MCE.emitConstantPool(MCP);
+    }
+
+    void startFunctionStub(unsigned StubSize) {
+      MCE.startFunctionStub(StubSize);
+    }
+
+    void *finishFunctionStub(const Function *F) {
+      return MCE.finishFunctionStub(F);
+    }
+
+    void emitByte(unsigned char B) {
+      MCE.emitByte(B);
+      actual << B; actual.flush();
+
+      values[counter] = (unsigned int) B;
+      if (++counter % 4 == 0 && counter != 0) {
+        o << std::hex;
+        for (unsigned i=0; i<4; ++i) {
+          if (values[i] < 16) o << "0";
+          o << values[i] << " ";
+        }
+
+        o << std::dec << "\t";
+        for (unsigned i=0; i<4; ++i) {
+          for (int j=7; j>=0; --j) {
+            o << ((values[i] >> j) & 1);
+          }
+          o << " ";
+        }
+
+        o << "\n";
+
+        unsigned instr = 0;
+        for (unsigned i=0; i<4; ++i)
+          instr |= values[i] << (i*8);
+
+        o << "--- * --- * --- * --- * ---\n";
+        counter %= 4;
+      }
+    }
+
+    void emitWord(unsigned W) {
+      MCE.emitWord(W);
+    }
+    void emitWordAt(unsigned W, unsigned *Ptr) {
+      MCE.emitWordAt(W, Ptr);
+    }
+    uint64_t getConstantPoolEntryAddress(unsigned Num) {
+      return MCE.getConstantPoolEntryAddress(Num);
+    }
+
+    virtual unsigned char* allocateGlobal(unsigned size, unsigned alignment)
+    { return MCE.allocateGlobal(size, alignment); }
+
+    uint64_t getCurrentPCValue() {
+      return MCE.getCurrentPCValue();
+    }
+    uint64_t getCurrentPCOffset() {
+      return MCE.getCurrentPCOffset();
+    }
+    void addRelocation(const MachineRelocation &MR) {
+      return MCE.addRelocation(MR);
+    }
+  };
+}
+
+/// createDebugMachineCodeEmitter - Return a dynamically allocated machine
+/// code emitter, which just prints the opcodes and fields out the cout.  This
+/// can be used for debugging users of the MachineCodeEmitter interface.
+///
+MachineCodeEmitter *
+MachineCodeEmitter::createDebugEmitter() {
+  return new DebugMachineCodeEmitter();
+}
+
+MachineCodeEmitter *
+MachineCodeEmitter::createFilePrinterEmitter(MachineCodeEmitter &MCE) {
+  return new FilePrinterEmitter(MCE, std::cerr);
+}