Browse Source

Fix imports and define abstract method inherited from BaseAttack

dustin.born 6 years ago
parent
commit
fbb87226b1
2 changed files with 8 additions and 4 deletions
  1. 7 3
      code/Attack/MembersMgmtCommAttack.py
  2. 1 1
      code/ID2TLib/PcapAddressOperations.py

+ 7 - 3
code/Attack/MembersMgmtCommAttack.py

@@ -7,7 +7,7 @@ from datetime import datetime
 import os
 import sys
 
-import ID2TLib.libbotnetcomm as lb
+import ID2TLib.Botnet.libbotnetcomm as lb
 from Attack import BaseAttack
 from Attack.AttackParameters import Parameter as Param
 from Attack.AttackParameters import ParameterTypes
@@ -65,10 +65,10 @@ class Message():
 from ID2TLib import FileUtils, Generator
 from ID2TLib.IPv4 import IPAddress
 from ID2TLib.PcapAddressOperations import PcapAddressOperations
-from ID2TLib.CommunicationProcessor import CommunicationProcessor
+from ID2TLib.Botnet.CommunicationProcessor import CommunicationProcessor
 from ID2TLib.Botnet.MessageMapping import MessageMapping
 from ID2TLib.PcapFile import PcapFile
-from ID2TLib.Statistics import Statistics
+from Core.Statistics import Statistics
 from scapy.layers.inet import IP, IPOption_Security
 
 
@@ -288,6 +288,10 @@ class MembersMgmtCommAttack(BaseAttack.BaseAttack):
         return total_pkts , path_attack_pcap
 
 
+    def generate_attack_packets(self):
+        pass
+
+
     def _create_messages(self, context):
         """
         Creates the messages that are to be injected into the PCAP.

+ 1 - 1
code/ID2TLib/PcapAddressOperations.py

@@ -1,6 +1,6 @@
 from random import choice
 
-from ID2TLib import Statistics
+from Core import Statistics
 from ID2TLib.IPv4 import IPAddress
 
 is_ipv4 = IPAddress.is_ipv4