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
|
@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):
|
||||||
|
|
Loading…
Reference in New Issue