Переглянути джерело

First fix for ports, more to follow

Denis Waßmann 6 роки тому
батько
коміт
d8d73e92a7
1 змінених файлів з 4 додано та 1 видалено
  1. 4 1
      code/Attack/MembersMgmtCommAttack.py

+ 4 - 1
code/Attack/MembersMgmtCommAttack.py

@@ -187,7 +187,10 @@ class MembersMgmtCommAttack(BaseAttack.BaseAttack):
             id_src, id_dst = msg.src["ID"], msg.dst["ID"]
             ip_src, ip_dst = msg.src["IP"], msg.dst["IP"]
             mac_src, mac_dst = msg.src["MAC"], msg.dst["MAC"]
-            port_src, port_dst = int(msg.src["SrcPort"]), int(msg.dst["DstPort"])
+            if msg.type.is_request():
+                port_src, port_dst = int(msg.src["SrcPort"]), int(msg.dst["DstPort"])
+            else:
+                port_src, port_dst = int(msg.src["DstPort"]), int(msg.dst["SrcPort"])
             ttl = int(msg.src["TTL"])
 
             # update duration