diff --git a/ujit.rb b/ujit.rb index 43be881c05..3697e50b1c 100644 --- a/ujit.rb +++ b/ujit.rb @@ -1,5 +1,9 @@ module UJIT def self.disasm(iseq) + if iseq.is_a? Method + iseq = RubyVM::InstructionSequence.of(iseq) + end + blocks = UJIT.blocks_for(iseq) return if blocks.empty? @@ -10,13 +14,14 @@ module UJIT str << iseq.disasm str << "\n" + # Sort the blocks by increasing addresses blocks.sort_by(&:address).reverse.each do |block| - str << "== ISEQ RANGE: #{block.iseq_start_index} -> #{block.iseq_end_index} ".ljust(80, "=") + str << "== ISEQ RANGE: [#{block.iseq_start_index},#{block.iseq_end_index}[ ".ljust(80, "=") str << "\n" cs.disasm(block.code, 0).each do |i| str << sprintf( - "\t0x%
x:\t%s\t%
s\n", + "\t%
04X:\t%s\t%
s\n", address: i.address, instruction: i.mnemonic, details: i.op_str diff --git a/ujit_iface.c b/ujit_iface.c index 591d9c044c..981c882ceb 100644 --- a/ujit_iface.c +++ b/ujit_iface.c @@ -397,7 +397,7 @@ ujit_disasm(VALUE self, VALUE code, VALUE from) cs_insn *insns; TypedData_Get_Struct(self, csh, &ujit_disasm_type, handle); - count = cs_disasm(*handle, (uint8_t *)StringValuePtr(code), RSTRING_LEN(code), NUM2INT(from), 0, &insns); + count = cs_disasm(*handle, (uint8_t*)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++) {