def _get_hosts_by_ids(self, ids_string):
if ids_string == 'all':
return self.hosts.copy()
ids = ids_string.split(',')
hosts = set()
for id_ in ids:
is_mac = netutils.validate_mac_address(id_)
is_ip = netutils.validate_ip_address(id_)
is_id_ = id_.isdigit()
if not is_mac and not is_ip and not is_id_:
IO.error('invalid identifier(s): \'{}\'.'.format(ids_string))
return
if is_mac or is_ip:
found = False
for host in self.hosts:
if host.mac == id_.lower() or host.ip == id_:
found = True
hosts.add(host)
break
if not found:
IO.error('no host matching {}{}{}.'.format(IO.Fore.LIGHTYELLOW_EX, id_, IO.Style.RESET_ALL))
return
else:
id_ = int(id_)
if len(self.hosts) == 0 or id_ not in range(len(self.hosts)):
IO.error('no host with id {}{}{}.'.format(IO.Fore.LIGHTYELLOW_EX, id_, IO.Style.RESET_ALL))
return
hosts.add(self.hosts[id_])
return hosts
def _add_handler(self, args):
"""
Handles 'add' command-line argument
Adds custom host to host list
"""
ip = args.ip
if not netutils.validate_ip_address(ip):
IO.error('invalid ip address.')
return
if args.mac:
mac = args.mac
if not netutils.validate_mac_address(mac):
IO.error('invalid mac address.')
return
else:
mac = netutils.get_mac_by_ip(self.interface, ip)
if mac is None:
IO.error('unable to resolve mac address. specify manually (--mac).')
return
name = None
try:
host_info = socket.gethostbyaddr(ip)
name = None if host_info is None else host_info[0]
except socket.herror:
pass
host = Host(ip, mac, name)
if host in self.hosts:
IO.error('host does already exist.')
return
self.hosts.append(host)
IO.ok('host added.')
