duran 0.1.0

data/client_src/duran.c

@@ -0,0 +1,57 @@
+#include <stdlib.h>
+#include <stdio.h>
+#include <ruby.h>
+#include "duran.h"
+#include "dr_api.h"
+
+VALUE cDuran;
+
+int 
+wrap_ruby_load(char *script)
+{
+  int status;
+  rb_load_protect(rb_str_new2(script), 0, &status);
+  if(status != 0) {
+    rb_eval_string("STDERR.puts 'Exception raised:', $!.inspect, $!.backtrace");
+    exit(status);
+  }
+  return(status);
+}
+
+DR_EXPORT void
+dr_init(client_id_t id)
+{
+  char *duran_script;
+  VALUE library_path;
+
+  ruby_init();
+  ruby_script("embedded");
+
+  cDuran = rb_define_module("Duran");
+  ruby_init_loadpath();
+  rb_define_const(cDuran, "ARCH", 
+#ifdef X64
+      rb_eval_string(":x64")
+#else
+      rb_eval_string(":x32")
+#endif
+  );
+
+  rb_define_const(cDuran, "PLATFORM", 
+#ifdef WINDOWS
+      rb_eval_string(":windows")
+#else
+      rb_eval_string(":linux")
+#endif
+  );
+
+  /* always load the duran runtime ruby library */
+  library_path = rb_gv_get("$:");
+  rb_ary_push(library_path, rb_str_new2(DURAN_RUBYLIBPATH));
+  wrap_ruby_load("duran.rb");
+
+  if(duran_script=getenv("DURAN_SCRIPT"))
+    wrap_ruby_load(duran_script);
+
+}
+

data/client_src/extconf.rb

@@ -0,0 +1,28 @@
+require 'mkmf'
+
+ARCHBITS = "32"
+#ARCHBITS = "64"
+
+$defs.push("-DX86_#{ARCHBITS}")
+
+# Indicate the DynamoRIO platform based on our RUBY_PLATFORM
+if RUBY_PLATFORM =~ /linux$/
+  $defs.push("-DLINUX")
+elsif RUBY_PLATFORM =~ /win32$/
+  $defs.push("-DWINDOWS")
+else
+  puts("Sorry, only Windows and Linux are supported by DynamoRIO")
+  exit 1
+end
+
+# Set the DURAN_PATH define based on cwd
+duran_path = File.expand_path(File.join(File.dirname(__FILE__), "..", "lib"))
+$defs.push("-DDURAN_RUBYLIBPATH=\\\"#{duran_path}\\\"")
+
+$CFLAGS = ENV["CFLAGS"].to_s
+$CFLAGS << " -I./dr_include"
+
+create_header("duran.h")
+dir_config("duran#{ARCHBITS}")
+create_makefile("duran#{ARCHBITS}")
+

data/lib/duran.rb

@@ -0,0 +1,18 @@
+begin ; require 'rubygems' ; rescue LoadError; end
+require 'ffi'
+require 'ffi/dry'
+
+module Duran
+  extend FFI::Library
+end
+
+require 'duran/defines'
+require 'duran/structs'
+
+require 'duran/events'
+require 'duran/app'
+require 'duran/ir_opnd'
+require 'duran/tools'
+require 'duran/proc'
+require 'duran/ir_utils'
+require 'duran/ir_opcodes'

data/lib/duran/app.rb

@@ -0,0 +1,8 @@
+
+module Duran
+  attach_function :dr_app_setup, [], :int
+  attach_function :dr_app_cleanup, [], :int
+  attach_function :dr_app_start, [], :void
+  attach_function :dr_app_stop, [], :void
+  attach_function :dr_app_take_over, [], :void
+end

data/lib/duran/defines.rb

@@ -0,0 +1,39 @@
+
+module Duran
+  if ::Duran::ARCH == :x64
+    typedef :int64, :ssize_t
+    typedef :uint64, :reg_t
+    typedef :int64, :ptr_int_t
+  else
+    typedef :int, :ssize_t
+    typedef :uint, :reg_t
+    typedef :int, :ptr_int_t
+  end
+
+
+  if ::Duran::PLATFORM == :windows
+    typedef :pointer, :file_t
+    typedef :reg_t, :thread_id_t
+    typedef :reg_t, :process_id_t
+  else
+    typedef :int, :file_t
+    typedef :pid_t, :thread_id_t
+    typedef :pid_t, :process_id_t
+  end
+
+  typedef :reg_t, :ptr_uint_t
+
+  typedef :uchar, :byte
+  typedef :uint, :client_id_t
+  typedef :uchar, :reg_id_t
+  typedef :uchar, :opnd_size_t
+  typedef :pointer, :app_pc
+  typedef :uint, :spill_slot_t
+
+  typedef :pointer, :instrlist_t
+  typedef :pointer, :module_data_t
+  typedef :pointer, :module_handle_t
+  typedef :pointer, :dr_module_iterator_t
+
+end
+

data/lib/duran/events.rb

@@ -0,0 +1,156 @@
+
+module Duran
+  typedef :pointer, :dr_ctx_t
+  typedef :pointer, :dr_tag_t
+
+  callback :void_event_cb, [], :void
+
+  attach_function :dr_register_exit_event, [:void_event_cb], :void
+  attach_function :dr_unregister_exit_event, [:void_event_cb], :bool
+
+  enum :emit_flags, [:default, :store_translations]
+
+  # dr_emit_flags_t (*func)
+  #   (void *drcontext, void *tag, instrlist_t *bb,
+  #   bool for_trace, bool translating)
+  callback :bb_event_cb, 
+           [:dr_ctx_t, :dr_tag_t, :instrlist_t, :bool, :bool],
+           :emit_flags
+
+  attach_function :dr_register_bb_event, [:bb_event_cb], :void
+  attach_function :dr_unregister_bb_event, [:bb_event_cb], :bool
+
+
+  # dr_emit_flags_t (*func)
+  #   (void *drcontext, void *tag, instrlist_t *trace, bool translating)
+  callback :trace_event_cb,
+           [:dr_ctx_t, :dr_tag_t, :instrlist_t, :bool], 
+           :emit_flags
+
+  attach_function :dr_register_trace_event, [:trace_event_cb], :void
+  attach_function :dr_unregister_trace_event, [:trace_event_cb], :bool
+
+  enum :custom_trace_action, [:dr_decides, :end_now, :continue]
+
+  # dr_custom_trace_action_t (*func)
+  #   (void *drcontext, void *tag, void *next_tag)
+  callback :end_trace_event_cb, 
+           [:dr_ctx_t, :dr_tag_t, :dr_tag_t],
+           :custom_trace_action
+
+  attach_function :dr_register_end_trace_event, [:end_trace_event_cb], :void
+  attach_function :dr_unregister_end_trace_event, [:end_trace_event_cb], :bool
+
+  # void (*func)(void *drcontext, void *tag)
+  callback :delete_event_cb, [:dr_ctx_t, :dr_tag_t], :void
+
+  attach_function :dr_register_delete_event, [:delete_event_cb], :void
+  attach_function :dr_unregister_delete_event, [:delete_event_cb], :bool
+
+  # void (*func)
+  #   (void *drcontext, void *tag, dr_mcontext_t *mcontext,
+  #   bool restore_memory, bool app_code_consistent)
+  callback :restore_state_event_cb,
+           [:dr_ctx_t, :dr_tag_t, MachineContext, :bool, :bool],
+           :void
+
+  attach_function :dr_register_restore_state_event,
+                  [:restore_state_event_cb],
+                  :void
+
+  attach_function :dr_unregister_restore_state_event,
+                  [:restore_state_event_cb],
+                  :bool
+
+
+  # bool (*func) (void *drcontext, bool restore_memory, 
+  #               dr_restore_state_info_t *info)
+  callback :restore_state_event_ex_cb,
+           [:dr_ctx_t, :bool, RestoreStateInfo],
+           :bool
+
+  attach_function :dr_register_restore_state_ex_event,
+                  [:restore_state_event_ex_cb],
+                  :void
+
+  attach_function :dr_unregister_restore_state_ex_event,
+                  [:restore_state_event_ex_cb],
+                  :bool
+
+  # void (*func)(void *drcontext)
+  callback :ctx_event_cb, [:dr_ctx_t], :void
+
+  attach_function :dr_register_thread_init_event, [:ctx_event_cb], :void
+  attach_function :dr_unregister_thread_init_event, [:ctx_event_cb], :bool
+
+  attach_function :dr_register_thread_exit_event, [:ctx_event_cb], :void
+  attach_function :dr_unregister_thread_exit_event, [:ctx_event_cb], :bool
+
+  attach_function :dr_register_fork_init_event, [:ctx_event_cb], :void
+  attach_function :dr_unregister_fork_init_event, [:ctx_event_cb], :bool
+
+  # void (*func)(void *drcontext, const module_data_t *info, bool loaded)
+  callback :modload_event_cb, [:dr_ctx_t, ModuleData, :bool], :void
+
+  attach_function :dr_register_module_load_event, [:modload_event_cb], :void
+  attach_function :dr_unregister_module_load_event, [:modload_event_cb], :bool
+
+
+  # void (*func)(void *drcontext, int sysnum)
+  callback :bool_syscall_event_cb, [:dr_ctx_t, :int], :bool
+
+  attach_function :dr_register_filter_syscall_event, 
+                  [:bool_syscall_event_cb], 
+                  :void
+
+  attach_function :dr_unregister_filter_syscall_event, 
+                  [:bool_syscall_event_cb], 
+                  :bool
+
+  attach_function :dr_register_pre_syscall_event, 
+                  [:bool_syscall_event_cb], 
+                  :void
+
+  attach_function :dr_unregister_pre_syscall_event, 
+                  [:bool_syscall_event_cb], 
+                  :bool
+
+  # bool (*func)(void *drcontext, int sysnum)
+  callback :void_syscall_event_cb, [:dr_ctx_t, :int], :void
+
+  attach_function :dr_register_post_syscall_event, 
+                  [:void_syscall_event_cb], 
+                  :void
+
+  attach_function :dr_unregister_post_syscall_event, 
+                  [:void_syscall_event_cb], 
+                  :bool
+
+
+  # void (*func)(void *drcontext, uint64 argument)
+  callback :nudge_event_cb, [:dr_ctx_t, :uint64], :void
+
+  attach_function :dr_register_nudge_event, 
+                  [:nudge_event_cb, :client_id_t], :void
+
+  attach_function :dr_unregister_nudge_event,
+                  [:nudge_event_cb, :client_id_t], :bool
+
+  attach_function :dr_nudge_client, [:client_id_t, :uint64], :bool
+
+  if ::Duran::PLATFORM == :windows
+    # bool (*func)(void *drcontext, dr_exception_t *excpt)
+    callback :exception_event_cb, [:dr_ctx_t, Exception], :bool
+
+    attach_function :dr_register_exception_event, [:exception_event_cb], :void
+    attach_function :dr_unregister_exception_event, [:exception_event_cb], :bool
+  elsif ::Duran::PLATFORM == :linux
+    enum :signal_action, [:deliver, :supress, :bypass, :redirect]
+    
+    # dr_signal_action_t (*func)(void *drcontext, dr_siginfo_t *siginfo)
+    callback :signal_event_cb, [:dr_ctx_t, SignalInfo], :signal_action
+
+    attach_function :dr_register_signal_event, [:signal_event_cb], :void
+    attach_function :dr_unregister_signal_event, [:signal_event_cb], :bool
+  end
+end

data/lib/duran/ir_opcodes.rb

@@ -0,0 +1,616 @@
+module Duran
+  OPCODES = [
+    :INVALID,
+    :UNDECODED,
+    :CONTD,
+    :LABEL,
+    :add,
+    :or,
+    :adc,
+    :sbb,
+    :and,
+    :daa,
+    :sub,
+    :das,
+    :xor,
+    :aaa,
+    :cmp,
+    :aas,
+    :inc,
+    :dec,
+    :push,
+    :push_imm,
+    :pop,
+    :pusha,
+    :popa,
+    :bound,
+    :arpl,
+    :imul,
+    :jo_short,
+    :jno_short,
+    :jb_short,
+    :jnb_short,
+    :jz_short,
+    :jnz_short,
+    :jbe_short,
+    :jnbe_short,
+    :js_short,
+    :jns_short,
+    :jp_short,
+    :jnp_short,
+    :jl_short,
+    :jnl_short,
+    :jle_short,
+    :jnle_short,
+    :call,
+    :call_ind,
+    :call_far,
+    :call_far_ind,
+    :jmp,
+    :jmp_short,
+    :jmp_ind,
+    :jmp_far,
+    :jmp_far_ind,
+    :loopne,
+    :loope,
+    :loop,
+    :jecxz,
+    :mov_ld,
+    :mov_st,
+    :mov_imm,
+    :mov_seg,
+    :mov_priv,
+    :test,
+    :lea,
+    :xchg,
+    :cwde,
+    :cdq,
+    :fwait,
+    :pushf,
+    :popf,
+    :sahf,
+    :lahf,
+    :ret,
+    :ret_far,
+    :les,
+    :lds,
+    :enter,
+    :leave,
+    :int3,
+    :int,
+    :into,
+    :iret,
+    :aam,
+    :aad,
+    :xlat,
+    :in,
+    :out,
+    :hlt,
+    :cmc,
+    :clc,
+    :stc,
+    :cli,
+    :sti,
+    :cld,
+    :std,
+    :lar,
+    :lsl,
+    :syscall,
+    :clts,
+    :sysret,
+    :invd,
+    :wbinvd,
+    :ud2a,
+    :nop_modrm,
+    :movntps,
+    :movntpd,
+    :wrmsr,
+    :rdtsc,
+    :rdmsr,
+    :rdpmc,
+    :sysenter,
+    :sysexit,
+    :cmovo,
+    :cmovno,
+    :cmovb,
+    :cmovnb,
+    :cmovz,
+    :cmovnz,
+    :cmovbe,
+    :cmovnbe,
+    :cmovs,
+    :cmovns,
+    :cmovp,
+    :cmovnp,
+    :cmovl,
+    :cmovnl,
+    :cmovle,
+    :cmovnle,
+    :punpcklbw,
+    :punpcklwd,
+    :punpckldq,
+    :packsswb,
+    :pcmpgtb,
+    :pcmpgtw,
+    :pcmpgtd,
+    :packuswb,
+    :punpckhbw,
+    :punpckhwd,
+    :punpckhdq,
+    :packssdw,
+    :punpcklqdq,
+    :punpckhqdq,
+    :movd,
+    :movq,
+    :movdqu,
+    :movdqa,
+    :pshufw,
+    :pshufd,
+    :pshufhw,
+    :pshuflw,
+    :pcmpeqb,
+    :pcmpeqw,
+    :pcmpeqd,
+    :emms,
+    :jo,
+    :jno,
+    :jb,
+    :jnb,
+    :jz,
+    :jnz,
+    :jbe,
+    :jnbe,
+    :js,
+    :jns,
+    :jp,
+    :jnp,
+    :jl,
+    :jnl,
+    :jle,
+    :jnle,
+    :seto,
+    :setno,
+    :setb,
+    :setnb,
+    :setz,
+    :setnz,
+    :setbe,
+    :setnbe,
+    :sets,
+    :setns,
+    :setp,
+    :setnp,
+    :setl,
+    :setnl,
+    :setle,
+    :setnle,
+    :cpuid,
+    :bt,
+    :shld,
+    :rsm,
+    :bts,
+    :shrd,
+    :cmpxchg,
+    :lss,
+    :btr,
+    :lfs,
+    :lgs,
+    :movzx,
+    :ud2b,
+    :btc,
+    :bsf,
+    :bsr,
+    :movsx,
+    :xadd,
+    :movnti,
+    :pinsrw,
+    :pextrw,
+    :bswap,
+    :psrlw,
+    :psrld,
+    :psrlq,
+    :paddq,
+    :pmullw,
+    :pmovmskb,
+    :psubusb,
+    :psubusw,
+    :pminub,
+    :pand,
+    :paddusb,
+    :paddusw,
+    :pmaxub,
+    :pandn,
+    :pavgb,
+    :psraw,
+    :psrad,
+    :pavgw,
+    :pmulhuw,
+    :pmulhw,
+    :movntq,
+    :movntdq,
+    :psubsb,
+    :psubsw,
+    :pminsw,
+    :por,
+    :paddsb,
+    :paddsw,
+    :pmaxsw,
+    :pxor,
+    :psllw,
+    :pslld,
+    :psllq,
+    :pmuludq,
+    :pmaddwd,
+    :psadbw,
+    :maskmovq,
+    :maskmovdqu,
+    :psubb,
+    :psubw,
+    :psubd,
+    :psubq,
+    :paddb,
+    :paddw,
+    :paddd,
+    :psrldq,
+    :pslldq,
+    :rol,
+    :ror,
+    :rcl,
+    :rcr,
+    :shl,
+    :shr,
+    :sar,
+    :not,
+    :neg,
+    :mul,
+    :div,
+    :idiv,
+    :sldt,
+    :str,
+    :lldt,
+    :ltr,
+    :verr,
+    :verw,
+    :sgdt,
+    :sidt,
+    :lgdt,
+    :lidt,
+    :smsw,
+    :lmsw,
+    :invlpg,
+    :cmpxchg8b,
+    :fxsave,
+    :fxrstor,
+    :ldmxcsr,
+    :stmxcsr,
+    :lfence,
+    :mfence,
+    :clflush,
+    :sfence,
+    :prefetchnta,
+    :prefetcht0,
+    :prefetcht1,
+    :prefetcht2,
+    :prefetch,
+    :prefetchw,
+    :movups,
+    :movss,
+    :movupd,
+    :movsd,
+    :movlps,
+    :movlpd,
+    :unpcklps,
+    :unpcklpd,
+    :unpckhps,
+    :unpckhpd,
+    :movhps,
+    :movhpd,
+    :movaps,
+    :movapd,
+    :cvtpi2ps,
+    :cvtsi2ss,
+    :cvtpi2pd,
+    :cvtsi2sd,
+    :cvttps2pi,
+    :cvttss2si,
+    :cvttpd2pi,
+    :cvttsd2si,
+    :cvtps2pi,
+    :cvtss2si,
+    :cvtpd2pi,
+    :cvtsd2si,
+    :ucomiss,
+    :ucomisd,
+    :comiss,
+    :comisd,
+    :movmskps,
+    :movmskpd,
+    :sqrtps,
+    :sqrtss,
+    :sqrtpd,
+    :sqrtsd,
+    :rsqrtps,
+    :rsqrtss,
+    :rcpps,
+    :rcpss,
+    :andps,
+    :andpd,
+    :andnps,
+    :andnpd,
+    :orps,
+    :orpd,
+    :xorps,
+    :xorpd,
+    :addps,
+    :addss,
+    :addpd,
+    :addsd,
+    :mulps,
+    :mulss,
+    :mulpd,
+    :mulsd,
+    :cvtps2pd,
+    :cvtss2sd,
+    :cvtpd2ps,
+    :cvtsd2ss,
+    :cvtdq2ps,
+    :cvttps2dq,
+    :cvtps2dq,
+    :subps,
+    :subss,
+    :subpd,
+    :subsd,
+    :minps,
+    :minss,
+    :minpd,
+    :minsd,
+    :divps,
+    :divss,
+    :divpd,
+    :divsd,
+    :maxps,
+    :maxss,
+    :maxpd,
+    :maxsd,
+    :cmpps,
+    :cmpss,
+    :cmppd,
+    :cmpsd,
+    :shufps,
+    :shufpd,
+    :cvtdq2pd,
+    :cvttpd2dq,
+    :cvtpd2dq,
+    :nop,
+    :pause,
+    :ins,
+    :rep_ins,
+    :outs,
+    :rep_outs,
+    :movs,
+    :rep_movs,
+    :stos,
+    :rep_stos,
+    :lods,
+    :rep_lods,
+    :cmps,
+    :rep_cmps,
+    :repne_cmps,
+    :scas,
+    :rep_scas,
+    :repne_scas,
+    :fadd,
+    :fmul,
+    :fcom,
+    :fcomp,
+    :fsub,
+    :fsubr,
+    :fdiv,
+    :fdivr,
+    :fld,
+    :fst,
+    :fstp,
+    :fldenv,
+    :fldcw,
+    :fnstenv,
+    :fnstcw,
+    :fiadd,
+    :fimul,
+    :ficom,
+    :ficomp,
+    :fisub,
+    :fisubr,
+    :fidiv,
+    :fidivr,
+    :fild,
+    :fist,
+    :fistp,
+    :frstor,
+    :fnsave,
+    :fnstsw,
+    :fbld,
+    :fbstp,
+    :fxch,
+    :fnop,
+    :fchs,
+    :fabs,
+    :ftst,
+    :fxam,
+    :fld1,
+    :fldl2t,
+    :fldl2e,
+    :fldpi,
+    :fldlg2,
+    :fldln2,
+    :fldz,
+    :f2xm1,
+    :fyl2x,
+    :fptan,
+    :fpatan,
+    :fxtract,
+    :fprem1,
+    :fdecstp,
+    :fincstp,
+    :fprem,
+    :fyl2xp1,
+    :fsqrt,
+    :fsincos,
+    :frndint,
+    :fscale,
+    :fsin,
+    :fcos,
+    :fcmovb,
+    :fcmove,
+    :fcmovbe,
+    :fcmovu,
+    :fucompp,
+    :fcmovnb,
+    :fcmovene,
+    :fcmovnbe,
+    :fcmovnu,
+    :fnclex,
+    :fninit,
+    :fucomi,
+    :fcomi,
+    :ffree,
+    :fucom,
+    :fucomp,
+    :faddp,
+    :fmulp,
+    :fcompp,
+    :fsubrp,
+    :fsubp,
+    :fdivrp,
+    :fdivp,
+    :fucomip,
+    :fcomip,
+    :fisttp,
+    :haddpd,
+    :haddps,
+    :hsubpd,
+    :hsubps,
+    :addsubpd,
+    :addsubps,
+    :lddqu,
+    :monitor,
+    :mwait,
+    :movsldup,
+    :movshdup,
+    :movddup,
+    :femms,
+    :unknown_3dnow,
+    :pavgusb,
+    :pfadd,
+    :pfacc,
+    :pfcmpge,
+    :pfcmpgt,
+    :pfcmpeq,
+    :pfmin,
+    :pfmax,
+    :pfmul,
+    :pfrcp,
+    :pfrcpit1,
+    :pfrcpit2,
+    :pfrsqrt,
+    :pfrsqit1,
+    :pmulhrw,
+    :pfsub,
+    :pfsubr,
+    :pi2fd,
+    :pf2id,
+    :pi2fw,
+    :pf2iw,
+    :pfnacc,
+    :pfpnacc,
+    :pswapd,
+    :pshufb,
+    :phaddw,
+    :phaddd,
+    :phaddsw,
+    :pmaddubsw,
+    :phsubw,
+    :phsubd,
+    :phsubsw,
+    :psignb,
+    :psignw,
+    :psignd,
+    :pmulhrsw,
+    :pabsb,
+    :pabsw,
+    :pabsd,
+    :palignr,
+    :popcnt,
+    :movntss,
+    :movntsd,
+    :extrq,
+    :insertq,
+    :lzcnt,
+    :pblendvb,
+    :blendvps,
+    :blendvpd,
+    :ptest,
+    :pmovsxbw,
+    :pmovsxbd,
+    :pmovsxbq,
+    :pmovsxdw,
+    :pmovsxwq,
+    :pmovsxdq,
+    :pmuldq,
+    :pcmpeqq,
+    :movntdqa,
+    :packusdw,
+    :pmovzxbw,
+    :pmovzxbd,
+    :pmovzxbq,
+    :pmovzxdw,
+    :pmovzxwq,
+    :pmovzxdq,
+    :pcmpgtq,
+    :pminsb,
+    :pminsd,
+    :pminuw,
+    :pminud,
+    :pmaxsb,
+    :pmaxsd,
+    :pmaxuw,
+    :pmaxud,
+    :pmulld,
+    :phminposuw,
+    :crc32,
+    :pextrb,
+    :pextrd,
+    :extractps,
+    :roundps,
+    :roundpd,
+    :roundss,
+    :roundsd,
+    :blendps,
+    :blendpd,
+    :pblendw,
+    :pinsrb,
+    :insertps,
+    :pinsrd,
+    :dpps,
+    :dppd,
+    :mpsadbw,
+    :pcmpestrm,
+    :pcmpestri,
+    :pcmpistrm,
+    :pcmpistri,
+    :movsxd,
+    :swapgs,
+    :vmcall,
+    :vmlaunch,
+    :vmresume,
+    :vmxoff,
+    :vmptrst,
+    :vmptrld,
+    :vmxon,
+    :vmclear,
+    :vmread,
+    :vmwrite,
+    :int1,
+    :salc,
+    :ffreep,
+  ]
+end