Changes to online concealment

Author: afmurillo

dhalsim/network_attacks/concealment_ae_model.py

@@ -15,6 +15,7 @@
 from sklearn.model_selection import train_test_split
 
 import joblib
+import numpy as np
 
 # This module is based on the implementation by Alessandro Erba, original is found here:
 # https://github.com/scy-phy/ICS-Evasion-Attacks/blob/master/Adversarial_Attacks/Black_Box_Attack/adversarial_AE.py
@@ -81,12 +82,12 @@ def save_model(self, model_filename, scaler_filename):
 
     def init_generator(self, training_path):
         # Load and preprocess training data
-        training_path = Path(__file__).parent/training_path/'training_data.csv'
+        #training_path = Path(__file__).parent/training_path/'training_data.csv'
+        training_path = Path(__file__).parent/training_path/'ground_truth_dataset.csv'
         # print('Reading training data from: ' + str(training_path))
         self.physical_pd = self.preprocess_physical(training_path)
 
         # Adversarial model for concealment
-        # toDo: Ask about this parameter
         hide_layers = 160
         self.hide_layers = hide_layers
         self.generator_layers = [self.feature_dims,
@@ -129,14 +130,19 @@ def fix_sample(self, gen_examples):
         return gen_examples
 
     def predict(self, received_values_df):
-        print('Attempting to predict concealment values')
+        #print('Attempting to predict concealment values')
         # print('Features received to predict: ' + str(received_values_df.columns))
         # print('Features received to train: ' + str(self.sensor_cols))
 
         gen_examples = self.generator.predict(self.attacker_scaler.transform(received_values_df))
+        #print('nan predicted values: ')
+        # print(gen_examples)
+        #print(np.isnan(gen_examples))
         gen_examples = self.fix_sample(pd.DataFrame(columns=self.sensor_cols,
                                                     data=self.attacker_scaler.inverse_transform(gen_examples)))
 
+        #print('Model fixed values')
+        #print(gen_examples)
         return gen_examples
 
     def __init__(self, features_list):
@@ -145,3 +151,4 @@ def __init__(self, features_list):
         self.sensor_cols = [col for col in features_list if
                             col not in ['Unnamed: 0', 'iteration', 'timestamp', 'Attack']]
         self.feature_dims = len(self.sensor_cols)
+        print(f'Model has {self.feature_dims } input features' )

dhalsim/network_attacks/mitm_netfilter_queue.py

@@ -58,7 +58,7 @@ def capture(self, packet):
                         del p[TCP].chksum
 
                         packet.set_payload(bytes(p))
-                        self.logger.debug(f"Value of network packet for {p[IP].dst} overwritten.")
+                        # self.logger.debug(f"Value of network packet for {p[IP].dst} overwritten.")
 
             packet.accept()
         except Exception as exc: