require 'msf/core'

module Msf

class Auxiliary::ReconTest < Msf::Auxiliary

	include Auxiliary::Recon
	include Exploit::Remote::Tcp
	
	def initialize
		super(
			'Name'        => 'Simple Recon Module Tester',
			'Version'     => '$Revision$',
			'Description' => 'Simple Recon Module Tester',
			'Author'      => 'hdm',
			'License'     => MSF_LICENSE,
			'Actions'     =>
				[
					['Single Port Probe'],
					['Continuous Port Sweep']
				],
			'PassiveActions' => 
				[
					'Continuous Port Sweep'
				]
		)

		register_options(
			[
				Opt::RHOST,
				Opt::RPORT,
			], self.class)	

	end

	def run
		print_status("Running the simple recon module with action #{action.name}")
	
		case action.name
		when 'Single Port Probe'
			prober()
			
		when 'Continuous Port Sweep'
			while (true)
				1.upto(65535) do |port|
					datastore['RPORT'] = port
					prober()
				end
			end
		end
	end
	
	def prober
		begin
			connect
			disconnect
			report_host(:host => datastore['RHOST'])
			report_service(
				:host  => datastore['RHOST'], 
				:port  => datastore['RPORT'],
				:proto => 'tcp'
			)
		rescue ::Exception => e
			case e.to_s
			when /connection was refused/
				report_host(:host => datastore['RHOST'])
			else
				print_status(e.to_s)
			end
		end	
	end

	
end
end