updated code

Author: desktop machine

dhalsim/network_attacks/concealment_ae_model.py

@@ -74,10 +74,10 @@ def load_scaler(self, scaler_path):
     # Saves the model and the scaler used to train the model
     def save_model(self, filename):
         print('saving trained model at: ', str(filename))
-        self.generator.save(str(model_path))
+        self.generator.save(str(filename))
 
         scaler_path = Path.cwd()
-        print('saved scaler model at: ', filename)
+        print('saved scaler model at: ', scaler_path)
         joblib.dump(self.attacker_scaler, 'ctown_attacker_scaler.gz')
 
     def init_generator(self, training_path):

dhalsim/network_attacks/unconstrained_blackbox_netfilter_queue.py

@@ -140,13 +140,13 @@ def handle_sync(self):
             while not self.get_sync(2):
                 pass
 
-            # self.logger.debug('Sync is 2. Keeping attack sync in 2, until we get all SCADA flags')
+            self.logger.debug('Sync is 2. Keeping attack sync in 2, until we get all SCADA flags')
 
             # We stay in 2, to conceal the values exchanged remotely from the PLCs, until we make a prediction
             while self.missing_scada_tags and self.sync_flag:
                 pass
 
-            # self.logger.debug('Setting attack sync in 3')
+            self.logger.debug('Setting attack sync in 3')
             self.set_sync(3)
 
         self.logger.debug('Netfilter sync thread while finished')
@@ -184,7 +184,7 @@ def process_tag_in_missing(self, session, ip_payload):
         self.received_scada_tags_df.loc[current_clock, session['tag']] = translate_payload_to_float(
             ip_payload[Raw].load)
         self.missing_scada_tags.remove(session['tag'])
-        # self.logger.debug('Missing tags len after removing: ' + str(len(self.missing_scada_tags)))
+        self.logger.debug('Missing tags len after removing: ' + str(len(self.missing_scada_tags)))
 
     def scada_tag_list_empty(self):
         # self.logger.debug('SCADA set empty')