|
@@ -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
|