This commit is contained in:
Oran Nadler 2018-04-17 14:59:06 +03:00
parent 0d4e28b55b
commit 29fac1a960
1 changed files with 5 additions and 2 deletions

View File

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