fix

2018-04-17 14:59:06 +03:00 · 2018-04-17 14:59:06 +03:00 · 29fac1a960
parent 0d4e28b55b
commit 29fac1a960
1 changed files with 5 additions and 2 deletions
--- a/monkey_island/cc/resources/pthmap.py
+++ b/monkey_island/cc/resources/pthmap.py
@ -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):