1 files changed, 48 insertions, 0 deletions
diff --git a/ujit_iface.c b/ujit_iface.c
index fac401a7b2..349aa40b1b 100644
--- a/ujit_iface.c
+++ b/ujit_iface.c
@@ -13,8 +13,12 @@
 #include "ujit_codegen.h"
 #include "ujit_core.h"
 #include "ujit_hooks.inc"
+#include "ujit.rbinc"
+#include <capstone/capstone.h>
 
 VALUE cUjitBlock;
+VALUE cUjitDisasm;
+VALUE cUjitDisasmInsn;
 
 extern st_table * version_tbl;
 extern codeblock_t *cb;
@@ -25,6 +29,12 @@ static const rb_data_type_t ujit_block_type = {
     0, 0, RUBY_TYPED_FREE_IMMEDIATELY
 };
 
+static const rb_data_type_t ujit_disasm_type = {
+    "UJIT/Disasm",
+    {0, (void(*)(void *))cs_close, 0, },
+    0, 0, RUBY_TYPED_FREE_IMMEDIATELY
+};
+
 bool rb_ujit_enabled;
 
 // Hash table of encoded instructions
@@ -366,6 +376,36 @@ iseq_end_index(VALUE self)
     return INT2NUM(block->end_idx);
 }
 
+static VALUE
+ujit_disasm_open(VALUE mod, VALUE arch, VALUE mode)
+{
+    csh * handle;
+    VALUE disasm = TypedData_Make_Struct(cUjitDisasm, csh, &ujit_disasm_type, handle);
+    cs_open(NUM2INT(arch), NUM2INT(mode), handle);
+    return disasm;
+}
+
+static VALUE
+ujit_disasm(VALUE self, VALUE code, VALUE from)
+{
+    size_t count;
+    csh * handle;
+    cs_insn *insns;
+
+    TypedData_Get_Struct(self, csh, &ujit_disasm_type, handle);
+    count = cs_disasm(*handle, StringValuePtr(code), RSTRING_LEN(code), NUM2INT(from), 0, &insns);
+    VALUE insn_list = rb_ary_new_capa(count);
+
+    for (size_t i = 0; i < count; i++) {
+        VALUE vals = rb_ary_new_from_args(3, LONG2NUM(insns[i].address),
+                rb_str_new2(insns[i].mnemonic),
+                rb_str_new2(insns[i].op_str));
+        rb_ary_push(insn_list, rb_struct_alloc(cUjitDisasmInsn, vals));
+    }
+    cs_free(insns, count);
+    return insn_list;
+}
+
 void
 rb_ujit_init(void)
 {
@@ -389,6 +429,14 @@ rb_ujit_init(void)
     rb_define_method(cUjitBlock, "iseq_start_index", iseq_start_index, 0);
     rb_define_method(cUjitBlock, "iseq_end_index", iseq_end_index, 0);
 
+    cUjitDisasm = rb_define_class_under(mUjit, "Disasm", rb_cObject);
+    rb_define_const(cUjitDisasm, "ARCH_X86", INT2NUM(CS_ARCH_X86));
+    rb_define_const(cUjitDisasm, "MODE_64", INT2NUM(CS_MODE_64));
+    rb_define_module_function(cUjitDisasm, "open", ujit_disasm_open, 2);
+    rb_define_method(cUjitDisasm, "disasm", ujit_disasm, 2);
+
+    cUjitDisasmInsn = rb_struct_define_under(cUjitDisasm, "Insn", "address", "mnemonic", "op_str", NULL);
+
     // Initialize the GC hooks
     method_lookup_dependency = st_init_numtable();
     struct ujit_root_struct *root;