Contents

require 'rex/socket'

module Rex
module Socket

###
#
# This class provides an interface to enumerating a subnet with a supplied
# netmask.
#
###
class SubnetWalker

	#
	# Initializes a subnet walker instance using the supplied subnet
	# information.
	#
	def initialize(subnet, netmask)
		self.subnet  = Socket.resolv_to_dotted(subnet)
		self.netmask = Socket.resolv_to_dotted(netmask)

		reset
	end

	#
	# Resets the subnet walker back to its original state.
	#
	def reset
		self.curr_ip     = self.subnet.split('.')
		self.num_ips     = (1 << (32 - Socket.net2bitmask(self.netmask).to_i))
		self.curr_ip_idx = 0
	end

	#
	# Returns the next IP address.
	#
	def next_ip
		if (curr_ip_idx >= num_ips)
			return nil
		end

		if (curr_ip_idx > 0)
			self.curr_ip[3] = (curr_ip[3].to_i + 1) % 256
			self.curr_ip[2] = (curr_ip[2].to_i + 1) % 256 if (curr_ip[3] == 0)
			self.curr_ip[1] = (curr_ip[1].to_i + 1) % 256 if (curr_ip[2] == 0)
			self.curr_ip[0] = (curr_ip[0].to_i + 1) % 256 if (curr_ip[1] == 0)
		end

		self.curr_ip_idx += 1
	
		self.curr_ip.join('.')
	end

	#
	# The subnet that is being enumerated.
	#
	attr_reader :subnet
	#
	# The netmask of the subnet.
	#
	attr_reader :netmask
	#
	# The total number of IPs within the subnet.
	#
	attr_reader :num_ips

protected

	attr_writer   :subnet, :netmask, :num_ips # :nodoc:
	attr_accessor :curr_ip, :curr_ip_idx # :nodoc:

end

end
end

Version data entries

43 entries across 43 versions & 1 rubygems

Version	Path
librex-0.0.65	lib/rex/socket/subnet_walker.rb
librex-0.0.63	lib/rex/socket/subnet_walker.rb
librex-0.0.54	lib/rex/socket/subnet_walker.rb
librex-0.0.53	lib/rex/socket/subnet_walker.rb
librex-0.0.52	lib/rex/socket/subnet_walker.rb
librex-0.0.51	lib/rex/socket/subnet_walker.rb
librex-0.0.50	lib/rex/socket/subnet_walker.rb
librex-0.0.49	lib/rex/socket/subnet_walker.rb
librex-0.0.48	lib/rex/socket/subnet_walker.rb
librex-0.0.47	lib/rex/socket/subnet_walker.rb
librex-0.0.46	lib/rex/socket/subnet_walker.rb
librex-0.0.44	lib/rex/socket/subnet_walker.rb
librex-0.0.43	lib/rex/socket/subnet_walker.rb
librex-0.0.42	lib/rex/socket/subnet_walker.rb
librex-0.0.41	lib/rex/socket/subnet_walker.rb
librex-0.0.40	lib/rex/socket/subnet_walker.rb
librex-0.0.39	lib/rex/socket/subnet_walker.rb
librex-0.0.38	lib/rex/socket/subnet_walker.rb
librex-0.0.37	lib/rex/socket/subnet_walker.rb
librex-0.0.36	lib/rex/socket/subnet_walker.rb