diff --git a/plugins/amplification/plugin.py b/plugins/amplification/plugin.py index ff7d8b9..60edb9e 100644 --- a/plugins/amplification/plugin.py +++ b/plugins/amplification/plugin.py @@ -25,6 +25,8 @@ from scapy.all import * import actions.utils from plugins.plugin import Plugin +import layers.packet + BASEPATH = os.path.dirname(os.path.abspath(__file__)) PROJECT_ROOT = os.path.dirname(os.path.dirname(BASEPATH)) @@ -129,7 +131,7 @@ class AmplificationPluginRunner(Plugin): IP(dst=dst)/TCP(sport=sport, dport=dport, flags="A", ack=ack, seq=seq+1), IP(dst=dst)/TCP(sport=sport, dport=dport, flags="PA", ack=ack, seq=seq+1)/Raw(payload) ] - packets = [actions.packet.Packet(packet) for packet in packets] + packets = [layers.packet.Packet(packet) for packet in packets] packets_to_send = [] try: