Agent: Pass str (not None) to VictimHost constructor

This commit is contained in:
Mike Salvatore 2021-12-16 13:50:39 -05:00
parent 7cb1f761d8
commit b6f2bab15b
1 changed files with 2 additions and 1 deletions

View File

@ -23,7 +23,8 @@ class VictimHostFactory:
self.on_island = on_island
def build_victim_host(self, network_address: NetworkAddress) -> VictimHost:
victim_host = VictimHost(network_address.ip, network_address.domain)
domain = network_address.domain or ""
victim_host = VictimHost(network_address.ip, domain)
if self.tunnel:
victim_host.default_tunnel = self.tunnel.get_tunnel_for_ip(victim_host.ip_addr)