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
def GetVictimsByAttacker(self, attacker):
if type(victim) != unicode:
victim = victim.monkey_guid
victims = set()
for atck, vic, _ in self.edges:
if atck == attacker:
victims.add(vic)
return victims
return set(map(Machine, victims))
@cache
def GetInPathCountByVictim(self, victim, already_processed=None):