llvm/lib/CodeGen/MachineCodeEmitter.cpp

178 lines
4.8 KiB
C++
Raw Normal View History

//===-- 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>
#include <ios>
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);
}