61 lines
1.1 KiB
Ruby
61 lines
1.1 KiB
Ruby
|
require 'rex/socket'
|
||
|
|
|
||
|
|
module Rex
|
||
|
|
module Socket
|
||
|
|
|
||
|
|
###
|
||
|
|
#
|
||
|
|
# This class provides an interface to enumerating a subnet with a supplied
|
||
|
|
# netmask.
|
||
|
|
#
|
||
|
|
###
|
||
|
|
class SubnetWalker
|
||
|
|
|
||
|
|
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)) - 1
|
||
|
|
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
|
||
|
|
|
||
|
|
attr_reader :subnet, :netmask, :num_ips
|
||
|
|
|
||
|
|
protected
|
||
|
|
|
||
|
|
attr_writer :subnet, :netmask, :num_ips
|
||
|
|
attr_accessor :curr_ip, :curr_ip_idx
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
end
|