LLVMExtra.cpp

// Copyright (c) 2012, Richard Osborne, All rights reserved
// This software is freely distributable under a derivative of the
// University of Illinois/NCSA Open Source License posted in
// LICENSE.txt and at <http://github.xcore.com/>

#include "LLVMExtra.h"
#include "llvm-c/Disassembler.h"
#include "llvm/ExecutionEngine/ExecutionEngine.h"
#include "llvm/ExecutionEngine/JITEventListener.h"
#include "llvm/Support/CallSite.h"
#include "llvm/Support/MemoryBuffer.h"
#include "llvm/Transforms/Utils/Cloning.h"
#include "llvm/Transforms/Scalar.h"
#include "llvm/PassManager.h"
#include <iostream>

using namespace llvm;

LLVMMemoryBufferRef
LLVMExtraCreateMemoryBufferWithPtr(const char *ptr, size_t length)
{
  return wrap(MemoryBuffer::getMemBuffer(StringRef(ptr, length)));
}

LLVMBool LLVMExtraInlineFunction(LLVMValueRef call)
{
  InlineFunctionInfo IFI;
  return InlineFunction(CallSite(unwrap(call)), IFI);
}

void LLVMExtraAddDeadCodeEliminationPass(LLVMPassManagerRef PM) {
  unwrap(PM)->add(createDeadCodeEliminationPass());
}

#if 0
class JitDisassembler : public JITEventListener {
  LLVMDisasmContextRef DC;
public:
  JitDisassembler(const char *triple) {
    DC = LLVMCreateDisasm(triple, 0, 0, 0, 0);
  }
  ~JitDisassembler() {
    LLVMDisasmDispose(DC);
  }
  virtual void NotifyFunctionEmitted(const Function &f, void *code,
                                     size_t size,
                                     const EmittedFunctionDetails &) {
    uint8_t *bytes = (uint8_t*)code;
    uint8_t *end = bytes + size;
    char buf[256];
    while (bytes < end) {
      size_t instSize = LLVMDisasmInstruction(DC, bytes, end - bytes,
                                              (uint64_t)bytes, &buf[0], 256);
      if (instSize == 0) {
        std::cerr << "\t???\n";
        return;
      }
      bytes += instSize;
      std::cerr << buf << '\n';
    }
  }
};
#endif

void
LLVMExtraRegisterJitDisassembler(LLVMExecutionEngineRef EE,
                                 const char *triple) {
#if 0
  static JitDisassembler disassembler(triple);
  unwrap(EE)->RegisterJITEventListener(&disassembler);
#endif
}