i think this is also a bug

This commit is contained in:
Oran Nadler 2018-04-17 15:00:22 +03:00
parent 29fac1a960
commit 807606dae2
1 changed files with 4 additions and 1 deletions

View File

@ -776,13 +776,16 @@ class PassTheHashMap(object):
@cache @cache
def GetVictimsByAttacker(self, attacker): def GetVictimsByAttacker(self, attacker):
if type(victim) != unicode:
victim = victim.monkey_guid
victims = set() victims = set()
for atck, vic, _ in self.edges: for atck, vic, _ in self.edges:
if atck == attacker: if atck == attacker:
victims.add(vic) victims.add(vic)
return victims return set(map(Machine, victims))
@cache @cache
def GetInPathCountByVictim(self, victim, already_processed=None): def GetInPathCountByVictim(self, victim, already_processed=None):