metasploit-gs/test/functional/meterpreter/meterpreter_java_spec.rb

$:.unshift(File.join(File.dirname(__FILE__)))
$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..', 'lib'))
$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..', 'test', 'lib'))

require 'fileutils'
require 'meterpreter_spec_helper'
require 'meterpreter_specs'

module MsfTest
  describe "JavaMeterpreter" do
    # This include brings in all the spec helper methods
    include MsfTest::MeterpreterSpecHelper

    # This include brings in all the specs that are generic across the
    # meterpreter platforms
    include MsfTest::MeterpreterSpecs

    # This include brings in all the specs that are specific to the java
    # meterpreter
    include MsfTest::JavaMeterpreterSpecs

    before :all do
      @verbose = true

      @meterpreter_type = "java"

      ## Set up an outupt directory
      @output_directory = File.join(File.dirname(__FILE__), "test_output_#{@meterpreter_type}")

      if File.directory? @output_directory
        FileUtils.rm_rf(@output_directory)
      end

      Dir.mkdir(@output_directory)
      @default_file = "#{@output_directory}/default"

      create_session_java
    end

    before :each do
    end

    after :each do
      @session.init_ui(@input, @output)
    end

    after :all do
      # FileUtils.rm_rf("*.jpeg")
      # FileUtils.rm_rf("payload.jar")
      FileUtils.rm_rf(@output_directory)
    end

    def create_session_java
      ## Setup for win32
      @framework = Msf::Simple::Framework.create

      test_modules_path = File.join(File.dirname(__FILE__), '..', '..', 'modules')
      @framework.modules.add_module_path(test_modules_path)

      @exploit_name = 'test/java_tester'
      @payload_name = 'java/meterpreter/bind_tcp'
      @input = Rex::Ui::Text::Input::Stdio.new
      @output = Rex::Ui::Text::Output::File.new(@default_file)

      # Initialize the exploit instance
      exploit = @framework.exploits.create(@exploit_name)

      ## Fire it off against a known-vulnerable host
      @session = exploit.exploit_simple(
        'Options' => {},
        'Payload' => @payload_name,
        'LocalInput' => @input,
        'LocalOutput' => @output
      )

      puts @session.inspect

      ## If a session came back, try to interact with it.
      if @session
        @session.load_stdapi
      else
        raise Exception "Couldn't get a session!"
      end
    end
  end
end