13007eb12a
* renamed gem unicorn to unicorn-engine * renamed modules to unicornengine * renamed Module Unicorn to UnicornEngine and the gem unicorn-engine to unicornengine * unicornengine -> unicorn_engine
61 lines
1.5 KiB
Ruby
61 lines
1.5 KiB
Ruby
#!/usr/bin/env ruby
|
|
require 'unicorn_engine'
|
|
require 'unicorn_engine/x86_const'
|
|
require 'weakref'
|
|
|
|
include UnicornEngine
|
|
|
|
X86_CODE32 = "\x41" # INC ecx; DEC edx
|
|
|
|
# memory address where emulation starts
|
|
ADDRESS = 0x1000000
|
|
|
|
# callback for tracing instructions
|
|
hook_code = Proc.new do |uc, address, size, user_data|
|
|
puts("proc was run")
|
|
end
|
|
|
|
hook_code_weak = WeakRef.new hook_code
|
|
|
|
begin
|
|
# Initialize emulator in X86-32bit mode
|
|
mu = Uc.new UC_ARCH_X86, UC_MODE_32
|
|
# map 2MB memory for this emulation
|
|
mu.mem_map(ADDRESS, 2 * 1024 * 1024)
|
|
|
|
# write machine code to be emulated to memory
|
|
mu.mem_write(ADDRESS, X86_CODE32)
|
|
|
|
# initialize machine registers
|
|
mu.reg_write(UC_X86_REG_ECX, 0x1234)
|
|
mu.reg_write(UC_X86_REG_EDX, 0x7890)
|
|
|
|
# tracing all instructions with customized callback
|
|
mu.hook_add(UC_HOOK_CODE, hook_code)
|
|
|
|
hook_code = nil # erase reference to proc
|
|
|
|
GC.start() # force garbage collection to test if proc is garbage collected
|
|
|
|
# emulate machine code in infinite time
|
|
mu.emu_start(ADDRESS, ADDRESS + X86_CODE32.bytesize)
|
|
|
|
mu = nil # erase reference to Uc because apparently it doesn't go out of scope after this?
|
|
rescue UcError => e
|
|
puts("ERROR: %s" % e)
|
|
exit 1
|
|
rescue NoMethodError => e
|
|
puts("proc was garbage collected and we tried to invoke `call` on something strange")
|
|
exit 1
|
|
end
|
|
|
|
GC.start()
|
|
|
|
if hook_code_weak.weakref_alive?() then
|
|
puts("proc was not garbage collected")
|
|
exit 1
|
|
end
|
|
|
|
puts "test passed"
|
|
exit 0
|
|
|