瀏覽代碼

Fix imports and define abstract method inherited from BaseAttack

dustin.born 7 年之前
父節點
當前提交
fbb87226b1
共有 2 個文件被更改,包括 8 次插入4 次删除
  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 os
 import sys
 import sys
 
 
-import ID2TLib.libbotnetcomm as lb
+import ID2TLib.Botnet.libbotnetcomm as lb
 from Attack import BaseAttack
 from Attack import BaseAttack
 from Attack.AttackParameters import Parameter as Param
 from Attack.AttackParameters import Parameter as Param
 from Attack.AttackParameters import ParameterTypes
 from Attack.AttackParameters import ParameterTypes
@@ -65,10 +65,10 @@ class Message():
 from ID2TLib import FileUtils, Generator
 from ID2TLib import FileUtils, Generator
 from ID2TLib.IPv4 import IPAddress
 from ID2TLib.IPv4 import IPAddress
 from ID2TLib.PcapAddressOperations import PcapAddressOperations
 from ID2TLib.PcapAddressOperations import PcapAddressOperations
-from ID2TLib.CommunicationProcessor import CommunicationProcessor
+from ID2TLib.Botnet.CommunicationProcessor import CommunicationProcessor
 from ID2TLib.Botnet.MessageMapping import MessageMapping
 from ID2TLib.Botnet.MessageMapping import MessageMapping
 from ID2TLib.PcapFile import PcapFile
 from ID2TLib.PcapFile import PcapFile
-from ID2TLib.Statistics import Statistics
+from Core.Statistics import Statistics
 from scapy.layers.inet import IP, IPOption_Security
 from scapy.layers.inet import IP, IPOption_Security
 
 
 
 
@@ -288,6 +288,10 @@ class MembersMgmtCommAttack(BaseAttack.BaseAttack):
         return total_pkts , path_attack_pcap
         return total_pkts , path_attack_pcap
 
 
 
 
+    def generate_attack_packets(self):
+        pass
+
+
     def _create_messages(self, context):
     def _create_messages(self, context):
         """
         """
         Creates the messages that are to be injected into the PCAP.
         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 random import choice
 
 
-from ID2TLib import Statistics
+from Core import Statistics
 from ID2TLib.IPv4 import IPAddress
 from ID2TLib.IPv4 import IPAddress
 
 
 is_ipv4 = IPAddress.is_ipv4
 is_ipv4 = IPAddress.is_ipv4