Adding disassembler interface and external hook to udis86 library.

llvm-svn: 33358

1 files changed, 53 insertions, 0 deletions

diff --git a/llvm/lib/Support/Disassembler.cpp b/llvm/lib/Support/Disassembler.cpp
new file mode 100644
index 00000000000..cc1f5e4390e
--- /dev/null
+++ b/llvm/lib/Support/Disassembler.cpp
@@ -0,0 +1,53 @@
+//===- lib/Support/Disassembler.cpp -----------------------------*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file was developed by Anton Korobeynikov and is distributed under the
+// University of Illinois Open Source License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This file implements the necessary glue to call external disassembler
+// libraries.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/Config/config.h"
+#include "llvm/Support/Disassembler.h"
+
+#include <cassert>
+#include <iomanip>
+#include <string>
+#include <sstream>
+
+#if USE_UDIS86
+#include <udis86.h>
+#endif
+
+using namespace llvm;
+
+std::string llvm::disassembleBuffer(uint8_t* start, size_t length,
+                              Disassembler::Type type, uint64_t pc) {
+  std::stringstream res;
+  
+  if (type == Disassembler::X86_32 || type == Disassembler::X86_64) {
+#if USE_UDIS86
+    ud_t ud_obj;
+   
+    ud_init(&ud_obj);
+    ud_set_input_buffer(&ud_obj, start, length);
+    ud_set_mode(&ud_obj, (type == Disassembler::X86_32 ? 32 : 64));
+    ud_set_pc(&ud_obj, pc);
+    ud_set_syntax(&ud_obj, UD_SYN_ATT);
+
+    res << std::setbase(16)
+        << std::setw((type == Disassembler::X86_32 ? 8 : 16));
+
+    while (ud_disassemble(&ud_obj)) {
+      res << ud_insn_off(&ud_obj) << ":\t" << ud_insn_asm(&ud_obj) << "\n";
+    }
+#endif
+  }
+
+  return res.str();
+}