module Msf

###
#
# This module provides methods for reconaissance modules
#
###

module Auxiliary::Scanner

#
# Initializes an instance of a recon auxiliary module
#
def initialize(info = {})
	super

	register_advanced_options(
		[
			OptAddressRange.new('RHOSTS', [ true, "The address range or netmask of the target"]),
		], Auxiliary::Scanner)
end

#
# The command handler when launched from the console
#
def run
	if (self.respond_to?('run_range'))
		return run_range(datastore['RHOSTS'])
	end
	
	# Add support for multiple threads
	if (self.respond_to?('run_host'))
		ar = Rex::Socket::RangeWalker.new(datastore['RHOSTS'])
		while(ip = ar.next_ip)
			run_host(ip)
		end
		return
	end

	# Add support for multiple threads
	if (self.respond_to?('run_batch'))
	
		if (! self.respond_to?('run_batch_size'))
			print_status("This module needs to export run_batch_size()")
			return
		end
		
		size = run_batch_size()

		ar = Rex::Socket::RangeWalker.new(datastore['RHOSTS'])
		while(true)
			batch = []
			
			while (batch.length < size)
				ip = ar.next_ip
				break if not ip
				batch << ip
			end
			return if batch.length == 0
			run_batch(batch)
		end
		return
	end
		
	print_status("This module defined no run_host or run_range methods")
end

end
end