refine get_ips_from_interfaces

This commit is contained in:
Itay Mizeretz 2017-10-24 11:51:53 +03:00
parent 75a399874f
commit bf343ee24b
1 changed files with 6 additions and 4 deletions

View File

@ -138,12 +138,14 @@ def get_ips_from_interfaces():
res = [] res = []
ifs = get_host_subnets() ifs = get_host_subnets()
for net_interface in ifs: for net_interface in ifs:
host_addr = ipaddress.ip_address(u"%s" % net_interface['addr']) address_str = unicode(net_interface['addr'])
ip_interface = ipaddress.ip_interface(u"%s/%s" % (net_interface['addr'], net_interface['netmask'])) netmask_str = unicode(net_interface['netmask'])
host_address = ipaddress.ip_address(address_str)
ip_interface = ipaddress.ip_interface(u"%s/%s" % (address_str, netmask_str))
# limit subnet scans to class C only # limit subnet scans to class C only
if ip_interface.network.num_addresses > 255: if ip_interface.network.num_addresses > 255:
ip_interface = ipaddress.ip_interface(u"%s/24" % net_interface['addr']) ip_interface = ipaddress.ip_interface(u"%s/24" % address_str)
addrs = [str(addr) for addr in ip_interface.network.hosts() if addr != host_addr] addrs = [str(addr) for addr in ip_interface.network.hosts() if addr != host_address]
res.extend(addrs) res.extend(addrs)
return res return res