forked from p15670423/monkey
i think this is also a bug
This commit is contained in:
parent
29fac1a960
commit
807606dae2
|
@ -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):
|
||||
|
|
Loading…
Reference in New Issue