rename folders, amend code structure

Author: MoeBuTa

README.md

@@ -57,7 +57,9 @@ A research project on the simulation and evaluation of moving target defense in
 
 6. implement three [MTD Schemes](https://github.com/MoeBuTa/MTDSimTime/tree/main/mtdnetwork/network/mtd_scheme): simultaneously, randomly, alternatively.
 
-7. implement Mean Time to Compromise metric.
+7. implement Mean Time to Compromise.
+
+8. implement MTD Execute Frequency.
 
 
 ## Todos / Future works

docs/notes.md

@@ -0,0 +1,33 @@
+# Network Notes
+
+
+1. generate services
+
+2. setup users
+
+3. generate graph
+
+4. setup network
+
+
+vulnerabilities
+services
+hosts & compromised hosts
+users:
+
+
+
+
+Graph Neural Network (GNN):
+https://www.datacamp.com/tutorial/comprehensive-introduction-graph-neural-networks-gnns-tutorial
+
+Special types of neural networks capable of working with a graph data structure.
+
+Typical tasks:
+- graph classification: classify graphs into various categories
+- **node classification**: use neighboring node labels to predict missing node labels in a graph
+- edge prediction: predict edge between a pair of nodes with an incomplete adjacency matrix
+- cluster detection: divide nodes into various clusters based on edge structure
+- graph embedding: map graphs into vectors, preserving the relevant information on nodes, edges, structure
+- graph generation: learns from sample graph distribution to generate a new but similar graph structure
+

mtdnetwork/network/__init__.py mtdnetwork/component/__init__.py

@@ -1,5 +1,5 @@
 import logging
-from mtdnetwork.statistics.attack_statistics import AttackStatistics
+from mtdnetwork.statistic.attack_statistics import AttackStatistics
 from mtdnetwork.data.constants import HACKER_ATTACK_ATTEMPT_MULTIPLER
 
 

mtdnetwork/network/adversary.py mtdnetwork/component/adversary.py

@@ -3,8 +3,8 @@
 import numpy as np
 import random
 import mtdnetwork.data.constants as constants
-import mtdnetwork.network.services as services
-from mtdnetwork.statistics.scorer import Scorer
+import mtdnetwork.component.services as services
+from mtdnetwork.statistic.scorer import Scorer
 
 
 class Network:

mtdnetwork/network/copynetwork.py mtdnetwork/component/copynetwork.py

@@ -24,7 +24,8 @@ def __init__(self, scheme: str, network, alter_strategies=None):
                                 OSDiversity,
                                 PortShuffle,
                                 ServiceDiversity,
-                                UserShuffle]
+                                # UserShuffle
+                                ]
         self._mtd_alter_strategies = alter_strategies
         self.network = network
         self._init_mtd_scheme(scheme)

mtdnetwork/network/host.py mtdnetwork/component/host.py

@@ -4,9 +4,9 @@
 import numpy as np
 import random
 import mtdnetwork.data.constants as constants
-import mtdnetwork.network.services as services
-from mtdnetwork.network.host import Host
-from mtdnetwork.statistics.scorer import Scorer
+import mtdnetwork.component.services as services
+from mtdnetwork.component.host import Host
+from mtdnetwork.statistic.scorer import Scorer
 
 
 class TargetNetwork:

mtdnetwork/network/mtd_scheme.py mtdnetwork/component/mtd_scheme.py

@@ -0,0 +1,50 @@
+from mtdnetwork.component.copynetwork import Network
+from mtdnetwork.statistic.mtd_statistics import MTDStatistics
+from mtdnetwork.component.targetnetwork import TargetNetwork
+
+
+class TimeNetwork(Network):
+
+    def __init__(self, graph, pos, colour_map, total_nodes, total_endpoints, total_subnets, total_layers,
+                 node_per_layer, users_list, users_per_host):
+        # default parameters
+        self._mtd_stats = MTDStatistics()
+        self._mtd_queue = []
+        self._suspended_mtd = dict()
+        self._unfinished_mtd = dict()
+        super().__init__(graph, pos, colour_map, total_nodes, total_endpoints, total_subnets,
+                         total_layers, node_per_layer, users_list, users_per_host)
+
+    @staticmethod
+    def create_network():
+        target_network = TargetNetwork(total_nodes=200, total_endpoints=20, total_subnets=20, total_layers=5,
+                                       target_layer=2)
+        graph = target_network.get_graph_copy()
+        colour_map = target_network.get_colourmap()
+        pos = target_network.get_pos()
+        node_per_layer = target_network.get_node_per_layer()
+        users_list = target_network.get_users_list()
+        users_per_host = target_network.get_users_per_host()
+        time_network = TimeNetwork(graph, pos, colour_map, 200, 20, 20, 5, node_per_layer, users_list,
+                                   users_per_host)
+        return time_network
+
+    def is_compromised(self, compromised_hosts):
+        # TODO: refactor terminating condition
+        super().is_compromised(compromised_hosts)
+        pass
+
+    def get_mtd_stats(self):
+        return self._mtd_stats
+
+    def get_mtd_queue(self):
+        return self._mtd_queue
+
+    def get_suspended_mtd(self):
+        return self._suspended_mtd
+
+    def get_unfinished_mtd(self):
+        return self._unfinished_mtd
+
+    def set_unfinished_mtd(self, mtd):
+        self._unfinished_mtd[mtd.get_resource_type()] = mtd

mtdnetwork/network/services.py mtdnetwork/component/services.py

@@ -140,8 +140,7 @@
     'ENUM_HOST': 15,
     'SCAN_NEIGHBOR': 10,
     'SCAN_PORT': 20,
-    'EXPLOIT_VULN_MEAN': 25,
-    'EXPLOIT_VULN_STD': 0.8,
+    'EXPLOIT_VULN': (25, 0.8),
     'BRUTE_FORCE': 15,
-    'PENALTY': 2,
+    'PENALTY': 10,
 }

mtdnetwork/network/targetnetwork.py mtdnetwork/component/targetnetwork.py

@@ -1,5 +1,5 @@
 from mtdnetwork.mtd import MTD
-from mtdnetwork.network import host
+from mtdnetwork.component import host
 
 
 class IPShuffle(MTD):

mtdnetwork/operation/time_generator.py mtdnetwork/component/time_generator.py

@@ -1,5 +1,5 @@
 from mtdnetwork.mtd import MTD
-from mtdnetwork.network import host
+from mtdnetwork.component import host
 
 
 class PortShuffle(MTD):