emonti-jdi_hook 1.0.0

data/lib/jdi_hook/method_tracer.rb

@@ -0,0 +1,95 @@
+
+module JdiHook
+  class MethodTracer < BaseDebugger
+
+    def initialize(vm, opts={})
+      super(vm, opts)
+      @meth_hooks = Array.new
+      if hks = opts[:meth_hooks]
+        self.meth_hooks=hks
+      end
+    end
+
+    attr_reader :meth_hooks
+
+    def meth_hooks=(enu)
+      @meth_hooks = Array.new
+      raise ":method_hooks must be Enumerable" unless enu.kind_of? Enumerable
+      enu.each {|k,v| append_method_hook(k,v) }
+    end
+
+    def append_method_hook(m, val)
+      unless m.is_a? String or m.is_a? Regexp
+        raise "hook method key must be a String or Regex - got #{m.class}" 
+      end
+      unless val.is_a? Hash
+        raise "hook value must me a kind of Hash - got #{block.class}" 
+      end
+      @meth_hooks << [m, val]
+    end
+
+    def find_method_hook(meth)
+      @meth_hooks.each do |h|
+        k, val = h
+        if (k.is_a?(String) and meth == k) or (k.is_a?(Regexp) and meth =~ k)
+          return h
+        end
+      end
+      return nil
+    end
+
+    def create_event_requests(mgr)
+      reqs = [ mgr.createMethodEntryRequest(), mgr.createMethodExitRequest() ]
+      reqs.each do |req|
+        filter_classes(req)
+        req.setSuspendPolicy(EventRequest::SUSPEND_ALL)
+        req.enable()
+      end
+    end
+
+    def receive_event(event)
+      case event
+      when MethodEntryEvent  : handle_method_entry(event)
+      when MethodExitEvent   : handle_method_exit(event)
+      else
+        STDERR.puts " [DEBUG] Received Event: #{event.java_class}" if $DEBUG
+      end
+    end
+
+    def meth_name(meth)
+      "#{meth.declaringType.name}.#{meth.name}"
+    end
+
+    def notify_entry(meth)
+      m_name = meth_name(meth)
+      args = meth.argumentTypeNames.join(', ')
+      "MethodEntry: #{m_name}(#{args})"
+    end
+
+    def notify_exit(meth)
+      m_name = meth_name(meth)
+      "MethodExit: #{m_name}"
+    end
+
+    def handle_method_entry(event)
+      meth = event.method
+      m_name = meth_name(meth)
+      if h=find_method_hook(m_name) 
+        if block=h[1][:on_entry]
+          block.call(self, event)
+        end
+      end
+    end
+
+    def handle_method_exit(event)
+      meth = event.method
+      m_name = meth_name(meth)
+      if h=find_method_hook(m_name)
+        if block=h[1][:on_exit]
+          block.call(self, event)
+        end
+      end
+    end
+
+  end
+end

data/lib/jdi_hook/stream_redirect_thread.rb

@@ -0,0 +1,39 @@
+
+module JdiHook
+  class StreamRedirectThread < java.lang.Thread
+    include_class "java.io.IOException"
+
+    BUFFER_SIZE = 2048
+    PRIORITY = java.lang.Thread::MAX_PRIORITY-1
+
+    # Parameters:
+    #   name   = a name for this thread for display purposes
+    #   input  = java input stream
+    #   label  = label for output
+    #   output = ruby output object
+    #   bufsz  = read buffer size (Default: BUFFER_SIZE)
+    #   pri    = thread priority (Default: PRIORITY)
+    def initialize(name, input, label, output, bufsz=nil, pri=nil)
+      super(name)
+      @input = java.io.InputStreamReader.new(input)
+      @label = label
+      @output = output
+      @bufsz = bufsz || BUFFER_SIZE
+      setPriority(pri || PRIORITY)
+    end
+
+    def run()
+      begin
+        cbuf = Array.new(@bufsz).to_java(:char)
+        while ((count = @input.read(cbuf, 0, @bufsz)) >= 0 )
+          dat = cbuf[0,count].map {|x| x.chr}.join().chomp
+          @output.puts dat.split("\n").map {|l| "** #{@label} => #{l}"}
+        end
+        @output.flush()
+      rescue IOException => exc
+        STDERR.puts("Child I/O Transfer - #{exc}")
+      end
+    end
+  end
+end
+

data/samples/base_test.rb

@@ -0,0 +1,7 @@
+$: << "lib"
+require 'jdi_hook'
+
+vm = JdiHook.command_line_launch("HelloWorld")
+dbg = JdiHook::BaseDebugger.new(vm, :redirect_stdio => true)
+dbg.go
+

data/samples/meth_test.rb

@@ -0,0 +1,17 @@
+$: << "lib"
+require 'jdi_hook'
+
+vm = JdiHook.command_line_launch("HelloWorld")
+dbg = JdiHook::MethodTracer.new vm, 
+  :redirect_stdio => true
+
+en_proc = lambda {|this, evt| puts " [*] " << this.notify_entry(evt.method) }
+ex_proc = lambda {|this, evt| puts " [*] " << this.notify_exit(evt.method) }
+
+dbg.meth_hooks = { 
+  /\.main$/ => { :on_entry => en_proc, :on_exit => ex_proc },
+  /.*/      => { :on_entry => en_proc },
+}
+
+dbg.go
+

data/tasks/ann.rake

@@ -0,0 +1,80 @@
+
+begin
+  require 'bones/smtp_tls'
+rescue LoadError
+  require 'net/smtp'
+end
+require 'time'
+
+namespace :ann do
+
+  # A prerequisites task that all other tasks depend upon
+  task :prereqs
+
+  file PROJ.ann.file do
+    ann = PROJ.ann
+    puts "Generating #{ann.file}"
+    File.open(ann.file,'w') do |fd|
+      fd.puts("#{PROJ.name} version #{PROJ.version}")
+      fd.puts("    by #{Array(PROJ.authors).first}") if PROJ.authors
+      fd.puts("    #{PROJ.url}") if PROJ.url.valid?
+      fd.puts("    (the \"#{PROJ.release_name}\" release)") if PROJ.release_name
+      fd.puts
+      fd.puts("== DESCRIPTION")
+      fd.puts
+      fd.puts(PROJ.description)
+      fd.puts
+      fd.puts(PROJ.changes.sub(%r/^.*$/, '== CHANGES'))
+      fd.puts
+      ann.paragraphs.each do |p|
+        fd.puts "== #{p.upcase}"
+        fd.puts
+        fd.puts paragraphs_of(PROJ.readme_file, p).join("\n\n")
+        fd.puts
+      end
+      fd.puts ann.text if ann.text
+    end
+  end
+
+  desc "Create an announcement file"
+  task :announcement => ['ann:prereqs', PROJ.ann.file]
+
+  desc "Send an email announcement"
+  task :email => ['ann:prereqs', PROJ.ann.file] do
+    ann = PROJ.ann
+    from = ann.email[:from] || Array(PROJ.authors).first || PROJ.email
+    to   = Array(ann.email[:to])
+
+    ### build a mail header for RFC 822
+    rfc822msg =  "From: #{from}\n"
+    rfc822msg << "To: #{to.join(',')}\n"
+    rfc822msg << "Subject: [ANN] #{PROJ.name} #{PROJ.version}"
+    rfc822msg << " (#{PROJ.release_name})" if PROJ.release_name
+    rfc822msg << "\n"
+    rfc822msg << "Date: #{Time.new.rfc822}\n"
+    rfc822msg << "Message-Id: "
+    rfc822msg << "<#{"%.8f" % Time.now.to_f}@#{ann.email[:domain]}>\n\n"
+    rfc822msg << File.read(ann.file)
+
+    params = [:server, :port, :domain, :acct, :passwd, :authtype].map do |key|
+      ann.email[key]
+    end
+
+    params[3] = PROJ.email if params[3].nil?
+
+    if params[4].nil?
+      STDOUT.write "Please enter your e-mail password (#{params[3]}): "
+      params[4] = STDIN.gets.chomp
+    end
+
+    ### send email
+    Net::SMTP.start(*params) {|smtp| smtp.sendmail(rfc822msg, from, to)}
+  end
+end  # namespace :ann
+
+desc 'Alias to ann:announcement'
+task :ann => 'ann:announcement'
+
+CLOBBER << PROJ.ann.file
+
+# EOF

data/tasks/bones.rake

@@ -0,0 +1,20 @@
+
+if HAVE_BONES
+
+namespace :bones do
+
+  desc 'Show the PROJ open struct'
+  task :debug do |t|
+    atr = if t.application.top_level_tasks.length == 2
+      t.application.top_level_tasks.pop
+    end
+
+    if atr then Bones::Debug.show_attr(PROJ, atr)
+    else Bones::Debug.show PROJ end
+  end
+
+end  # namespace :bones
+
+end  # HAVE_BONES
+
+# EOF

data/tasks/gem.rake

@@ -0,0 +1,201 @@
+
+require 'find'
+require 'rake/packagetask'
+require 'rubygems/user_interaction'
+require 'rubygems/builder'
+
+module Bones
+class GemPackageTask < Rake::PackageTask
+  # Ruby GEM spec containing the metadata for this package.  The
+  # name, version and package_files are automatically determined
+  # from the GEM spec and don't need to be explicitly provided.
+  #
+  attr_accessor :gem_spec
+
+  # Tasks from the Bones gem directory
+  attr_reader :bones_files
+
+  # Create a GEM Package task library.  Automatically define the gem
+  # if a block is given.  If no block is supplied, then +define+
+  # needs to be called to define the task.
+  #
+  def initialize(gem_spec)
+    init(gem_spec)
+    yield self if block_given?
+    define if block_given?
+  end
+
+  # Initialization tasks without the "yield self" or define
+  # operations.
+  #
+  def init(gem)
+    super(gem.name, gem.version)
+    @gem_spec = gem
+    @package_files += gem_spec.files if gem_spec.files
+    @bones_files = []
+
+    local_setup = File.join(Dir.pwd, %w[tasks setup.rb])
+    if !test(?e, local_setup)
+      Dir.glob(::Bones.path(%w[lib bones tasks *])).each {|fn| bones_files << fn}
+    end
+  end
+
+  # Create the Rake tasks and actions specified by this
+  # GemPackageTask.  (+define+ is automatically called if a block is
+  # given to +new+).
+  #
+  def define
+    super
+    task :prereqs
+    task :package => ['gem:prereqs', "#{package_dir_path}/#{gem_file}"]
+    file "#{package_dir_path}/#{gem_file}" => [package_dir_path] + package_files + bones_files do
+      when_writing("Creating GEM") {
+        chdir(package_dir_path) do
+          Gem::Builder.new(gem_spec).build
+          verbose(true) {
+            mv gem_file, "../#{gem_file}"
+          }
+        end
+      }
+    end
+
+    file package_dir_path => bones_files do
+      mkdir_p package_dir rescue nil
+
+      gem_spec.files = (gem_spec.files +
+          bones_files.map {|fn| File.join('tasks', File.basename(fn))}).sort
+
+      bones_files.each do |fn|
+        base_fn = File.join('tasks', File.basename(fn))
+        f = File.join(package_dir_path, base_fn)
+        fdir = File.dirname(f)
+        mkdir_p(fdir) if !File.exist?(fdir)
+        if File.directory?(fn)
+          mkdir_p(f)
+        else
+          raise "file name conflict for '#{base_fn}' (conflicts with '#{fn}')" if test(?e, f)
+          safe_ln(fn, f)
+        end
+      end
+    end
+  end
+  
+  def gem_file
+    if @gem_spec.platform == Gem::Platform::RUBY
+      "#{package_name}.gem"
+    else
+      "#{package_name}-#{@gem_spec.platform}.gem"
+    end
+  end
+end  # class GemPackageTask
+end  # module Bones
+
+namespace :gem do
+
+  PROJ.gem._spec = Gem::Specification.new do |s|
+    s.name = PROJ.name
+    s.version = PROJ.version
+    s.summary = PROJ.summary
+    s.authors = Array(PROJ.authors)
+    s.email = PROJ.email
+    s.homepage = Array(PROJ.url).first
+    s.rubyforge_project = PROJ.rubyforge.name
+
+    s.description = PROJ.description
+
+    PROJ.gem.dependencies.each do |dep|
+      s.add_dependency(*dep)
+    end
+
+    PROJ.gem.development_dependencies.each do |dep|
+      s.add_development_dependency(*dep)
+    end
+
+    s.files = PROJ.gem.files
+    s.executables = PROJ.gem.executables.map {|fn| File.basename(fn)}
+    s.extensions = PROJ.gem.files.grep %r/extconf\.rb$/
+
+    s.bindir = 'bin'
+    dirs = Dir["{#{PROJ.libs.join(',')}}"]
+    s.require_paths = dirs unless dirs.empty?
+
+    incl = Regexp.new(PROJ.rdoc.include.join('|'))
+    excl = PROJ.rdoc.exclude.dup.concat %w[\.rb$ ^(\.\/|\/)?ext]
+    excl = Regexp.new(excl.join('|'))
+    rdoc_files = PROJ.gem.files.find_all do |fn|
+                   case fn
+                   when excl; false
+                   when incl; true
+                   else false end
+                 end
+    s.rdoc_options = PROJ.rdoc.opts + ['--main', PROJ.rdoc.main]
+    s.extra_rdoc_files = rdoc_files
+    s.has_rdoc = true
+
+    if test ?f, PROJ.test.file
+      s.test_file = PROJ.test.file
+    else
+      s.test_files = PROJ.test.files.to_a
+    end
+
+    # Do any extra stuff the user wants
+    PROJ.gem.extras.each do |msg, val|
+      case val
+      when Proc
+        val.call(s.send(msg))
+      else
+        s.send "#{msg}=", val
+      end
+    end
+  end  # Gem::Specification.new
+
+  Bones::GemPackageTask.new(PROJ.gem._spec) do |pkg|
+    pkg.need_tar = PROJ.gem.need_tar
+    pkg.need_zip = PROJ.gem.need_zip
+  end
+
+  desc 'Show information about the gem'
+  task :debug => 'gem:prereqs' do
+    puts PROJ.gem._spec.to_ruby
+  end
+
+  desc 'Write the gemspec '
+  task :spec => 'gem:prereqs' do
+    File.open("#{PROJ.name}.gemspec", 'w') do |f|
+      f.write PROJ.gem._spec.to_ruby
+    end
+  end
+
+  desc 'Install the gem'
+  task :install => [:clobber, 'gem:package'] do
+    sh "#{SUDO} #{GEM} install --local pkg/#{PROJ.gem._spec.full_name}"
+
+    # use this version of the command for rubygems > 1.0.0
+    #sh "#{SUDO} #{GEM} install --no-update-sources pkg/#{PROJ.gem._spec.full_name}"
+  end
+
+  desc 'Uninstall the gem'
+  task :uninstall do
+    installed_list = Gem.source_index.find_name(PROJ.name)
+    if installed_list and installed_list.collect { |s| s.version.to_s}.include?(PROJ.version) then
+      sh "#{SUDO} #{GEM} uninstall --version '#{PROJ.version}' --ignore-dependencies --executables #{PROJ.name}"
+    end
+  end
+
+  desc 'Reinstall the gem'
+  task :reinstall => [:uninstall, :install]
+
+  desc 'Cleanup the gem'
+  task :cleanup do
+    sh "#{SUDO} #{GEM} cleanup #{PROJ.gem._spec.name}"
+  end
+end  # namespace :gem
+
+
+desc 'Alias to gem:package'
+task :gem => 'gem:package'
+
+task :clobber => 'gem:clobber_package'
+remove_desc_for_task 'gem:clobber_package'
+
+# EOF