csle-attack-profiler 0.6.0__py3-none-any.whl → 0.6.1__py3-none-any.whl

csle_attack_profiler/__version__.py

@@ -1 +1 @@
-__version__ = '0.6.0'
+__version__ = '0.6.1'

csle_attack_profiler/hmm_profiling.py

@@ -1,12 +1,12 @@
+from typing import List, Union, Tuple, Any
+import numpy as np
+import sys
 from csle_common.dao.system_identification.emulation_statistics import EmulationStatistics
 from csle_common.dao.emulation_action.attacker.emulation_attacker_action_id import EmulationAttackerActionId
 from csle_common.dao.emulation_action.attacker.emulation_attacker_action_type import EmulationAttackerActionType
 from csle_common.dao.emulation_action.attacker.emulation_attacker_action_outcome import EmulationAttackerActionOutcome
 from csle_common.dao.emulation_action.attacker.emulation_attacker_action import EmulationAttackerAction
 from csle_attack_profiler.attack_profiler import AttackProfiler
-from typing import List, Union, Tuple, Any
-import numpy as np
-import sys
 
 
 class HMMProfiler:
@@ -20,6 +20,7 @@ class HMMProfiler:
 
         :param statistics: The list of EmulationStatistics objects
         :param model_name: The name of the model
+        :return: None
         """
         self.statistics = statistics
         self.transition_matrix: List[List[float]] = []
@@ -38,11 +39,12 @@ class HMMProfiler:
 
         :param transition_matrix: The transition matrix
         :param states: The list of states of the HMM (format: 'A:attack_name' or
-        'no_intrusion' based on emulation statistics file)
+                        'no_intrusion' based on emulation statistics file)
         :param metrics: The list of metrics to profile
         :param save: Whether to save the matrices to a file
         :param location: The location to save the matrices, if save = True, e.g "./resources",
-        default is current directory
+                         default is current directory
+        :return: None
         """
         emission_matrix, emission_matrix_observations = self.get_matrices_of_observation(self.statistics,
                                                                                          metric, hidden_states)
@@ -63,6 +65,7 @@ class HMMProfiler:
         Loads the HMM model from the given location.
 
         :param location: The location of the model files, default is current directory
+        :return: None
         """
         self.transition_matrix = np.load(f'{location}/transition_matrix.npy')
         self.hidden_states = np.load(f'{location}/hidden_states.npy')
@@ -75,18 +78,16 @@ class HMMProfiler:
         Profiles a sequence of observations based on the HMM model.
 
         :param sequence: The sequence of observations
-        
-
         :return: The most likely sequence of states
         """
-        
+
         path = HMMProfiler.viterbi(self.hidden_states, self.start_state_probs,
                                    self.transition_matrix, self.emission_matrix,
                                    sequence, self.emission_matrix_observations)
         profiled_sequence = []
         for i in range(len(path)):
             profiled_sequence.append(self.hidden_states[int(path[i])])
-        
+
         return profiled_sequence
 
     def get_matrices_of_observation(self, statistics: List[EmulationStatistics],
@@ -97,14 +98,13 @@ class HMMProfiler:
         :param statistics: The list of EmulationStatistics objects
         :param metric: The metric to get the emission matrix for
         :param states: The list of states
-
         :return: The emission matrix, the list of observations, the list of states
         """
         emission_matrix = []
         attack_observations = {}
         attack_observations_total_counts = {}
         all_keys = set()
-        
+
         for stats in statistics:
             for condition, metric_distribution in stats.conditionals_counts.items():
                 action = condition.split('_')
@@ -118,7 +118,7 @@ class HMMProfiler:
                         # Add the observations of the attack to the dictionary
                         if metric in metric_distribution:
                             attack_observations[action[0]] = metric_distribution[metric]
-                        # Sum the total counts of the observations
+                            # Sum the total counts of the observations
                             attack_observations_total_counts[action[0]] = sum(attack_observations[action[0]].values())
                 # Aggregate the counts from the metric distribution
                 else:
@@ -135,7 +135,7 @@ class HMMProfiler:
                 # Store all possible values for the observation
                 if action[0] in attack_observations:
                     all_keys.update(attack_observations[action[0]])
-        
+
         # Normalize the counts
         for attack, _ in attack_observations.items():
             attack_observations_total_counts[attack] = sum(attack_observations[attack].values())
@@ -182,7 +182,6 @@ class HMMProfiler:
         Converts a list of states to a list of AttackProfiles.
 
         :param states: The list of states to convert
-
         :return: The list of EmulationAttackerActionId
         """
 
@@ -198,8 +197,8 @@ class HMMProfiler:
                     id=EmulationAttackerActionId.CVE_2015_1427_EXPLOIT, name="CVE-2015-1427 exploit", cmds=None,
                     type=EmulationAttackerActionType.EXPLOIT,
                     descr="Uses the CVE-2015-1427 vulnerability to "
-                    "get remote code execution and then sets up a SSH backdoor"
-                    "to upgrade the channel", index=None, ips=[],
+                          "get remote code execution and then sets up a SSH backdoor"
+                          "to upgrade the channel", index=None, ips=[],
                     action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
                 p = AttackProfiler.get_attack_profile(action)
                 new_states.append(p)
@@ -232,7 +231,7 @@ class HMMProfiler:
                     id=EmulationAttackerActionId.PING_SCAN_HOST, name="Ping Scan",
                     cmds=None, type=EmulationAttackerActionType.RECON,
                     descr="A host discovery scan, it is quick because it only checks of hosts "
-                    "are up with Ping, without scanning the ports.", ips=None, index=None,
+                          "are up with Ping, without scanning the ports.", ips=None, index=None,
                     action_outcome=EmulationAttackerActionOutcome.INFORMATION_GATHERING, backdoor=False)
                 p = AttackProfiler.get_attack_profile(action)
                 new_states.append(p)
@@ -241,7 +240,7 @@ class HMMProfiler:
                     id=EmulationAttackerActionId.SAMBACRY_EXPLOIT, name="Sambacry Explolit", cmds=None,
                     type=EmulationAttackerActionType.EXPLOIT,
                     descr="Uses the sambacry shell to get remote code execution and then"
-                    "sets up a SSH backdoor to upgrade the channel",
+                          "sets up a SSH backdoor to upgrade the channel",
                     index=None, ips=[], action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
                 p = AttackProfiler.get_attack_profile(action)
                 new_states.append(p)
@@ -259,7 +258,8 @@ class HMMProfiler:
                     name="SSH dictionary attack for username=pw", cmds=None,
                     type=EmulationAttackerActionType.EXPLOIT, index=None,
                     descr="A dictionary attack that tries common passwords and usernames for SSH"
-                    "where username=password", ips=None, action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
+                          "where username=password", ips=None,
+                    action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
                 p = AttackProfiler.get_attack_profile(action)
                 new_states.append(p)
             elif state == 'A:FTP dictionary attack for username=pw':
@@ -267,7 +267,7 @@ class HMMProfiler:
                     id=EmulationAttackerActionId.FTP_SAME_USER_PASS_DICTIONARY_HOST,
                     name="FTP dictionary attack for username=pw", cmds=None, type=EmulationAttackerActionType.EXPLOIT,
                     index=None, descr="A dictionary attack that tries common passwords and"
-                    "usernames for FTP where username=password", ips=None,
+                                      "usernames for FTP where username=password", ips=None,
                     action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
                 p = AttackProfiler.get_attack_profile(action)
                 new_states.append(p)
@@ -277,7 +277,7 @@ class HMMProfiler:
                     name="Telnet dictionary attack for username=pw", cmds=None,
                     type=EmulationAttackerActionType.EXPLOIT, index=None,
                     descr="A dictionary attack that tries common passwords and usernames for"
-                    "Telnet where username=password", ips=None,
+                          "Telnet where username=password", ips=None,
                     action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
                 p = AttackProfiler.get_attack_profile(action)
                 new_states.append(p)
@@ -302,7 +302,7 @@ class HMMProfiler:
                     id=EmulationAttackerActionId.SSH_BACKDOOR, name="Install SSH backdoor",
                     cmds=None, type=EmulationAttackerActionType.POST_EXPLOIT,
                     descr="If taken root on remote machine, installs a ssh backdoor useful for"
-                    "upgrading telnetor weaker channels", index=None, ips=[],
+                          "upgrading telnetor weaker channels", index=None, ips=[],
                     action_outcome=EmulationAttackerActionOutcome.PIVOTING, alt_cmds=None, backdoor=True)
                 p = AttackProfiler.get_attack_profile(action)
                 new_states.append(p)
@@ -310,7 +310,7 @@ class HMMProfiler:
                 new_states.append(state)
 
         return new_states
-   
+
     def calculate_initial_states(self, transition_matrix: List[List[float]]) -> List[float]:
         """
         Calculates the initial states probabilities based on the transition matrix.
@@ -318,7 +318,6 @@ class HMMProfiler:
         1 / (# of states)
 
         :param transition_matrix: The transition matrix
-
         :return: The start states probabilities
         """
         start_states = []
@@ -341,7 +340,6 @@ class HMMProfiler:
         :param emission_matrix: The emission matrix
         :param obs: The observation sequence
         :param emissions_list: The list of possible observations
-
         :return: The most likely sequence of hidden states
         """
         # Convert the emissions list to a numpy array, to use the where function
@@ -407,470 +405,18 @@ class HMMProfiler:
         :param intrusion_length: The length of the intrusion
         :param initial_state_index: The index of the initial state
         :param seed: The seed for the random number generator
-        
         return: The sequence of states and observations
         """
-        
-        P_obs = self.emission_matrix
-        P_states = self.transition_matrix
-        states = self.hidden_states
-        observations = self.emission_matrix_observations
-        
-        if seed:
-            np.random.seed(seed)
-        obs_len = len(observations)
-        states_len = len(states)
-        # Return the geometric distribution of the initial state
-        dist = np.random.geometric(p=P_states[initial_state_index][0], size=1000)
-        T_i = round(sum(dist) / len(dist))
-
-        state_seq = [states[initial_state_index]] * T_i
-        obs_seq = []
-        for i in range(T_i):
-
-            o_i = np.random.choice(obs_len, p=P_obs[initial_state_index])
-            obs_seq.append(observations[o_i])
-
-        recon_states_sum = np.sum(P_states[initial_state_index][1:])
-        recon_states = P_states[initial_state_index][1:] / recon_states_sum
-
-        intrusion_start_state = np.random.choice(states_len - 1, p=recon_states) + 1
-        intrusion_start_observation = np.random.choice(obs_len, p=P_obs[intrusion_start_state])
-        state_seq.append(states[intrusion_start_state])
-        obs_seq.append(observations[intrusion_start_observation])
-
-        s_i = intrusion_start_state
-        if intrusion_length == 1:
-            return state_seq, obs_seq
-        for i in range(intrusion_length - 1):
-            # si ~ Ps(si | si-1)
-            s_i = np.random.choice(states_len, p=P_states[s_i])
-            # oi ~ Po(oi | si)
-            o_i = np.random.choice(obs_len, p=P_obs[s_i])
-            state_seq.append(states[s_i])
-            obs_seq.append(observations[o_i])
-        return state_seq, obs_seq
-
-from csle_common.dao.system_identification.emulation_statistics import EmulationStatistics
-from csle_common.dao.emulation_action.attacker.emulation_attacker_action_id import EmulationAttackerActionId
-from csle_common.dao.emulation_action.attacker.emulation_attacker_action_type import EmulationAttackerActionType
-from csle_common.dao.emulation_action.attacker.emulation_attacker_action_outcome import EmulationAttackerActionOutcome
-from csle_common.dao.emulation_action.attacker.emulation_attacker_action import EmulationAttackerAction
-from csle_attack_profiler.attack_profiler import AttackProfiler
-from typing import List, Union, Tuple, Any
-import numpy as np
-import sys
-
-
-class HMMProfiler:
-    """
-    The HMMProfiler class is used to profile a sequence of observations based on a Hidden Markov Model (HMM).
-    """
-
-    def __init__(self, statistics: List[EmulationStatistics], model_name: Union[str, None] = None) -> None:
-        """
-        Class constructor
-
-        :param statistics: The list of EmulationStatistics objects
-        :param model_name: The name of the model
-        """
-        self.statistics = statistics
-        self.transition_matrix: List[List[float]] = []
-        self.emission_matrix: List[List[float]] = []
-        self.hidden_states: List[str] = []
-        self.emission_matrix_observations: List[int] = []
-        self.start_state_probs: List[float] = []
-        self.model_name = None
-
-    def create_model(self, transition_matrix: List[List[float]],
-                     hidden_states: List[str], metric: str,
-                     save_model: bool = False, location: str = ".") -> None:
-        """
-        Creates the HMM model based on the given transition matrix, states and metrics.
-        If save = True, matrices are saved to given location
-
-        :param transition_matrix: The transition matrix
-        :param states: The list of states of the HMM (format: 'A:attack_name' or
-                       'no_intrusion' based on emulation statistics file)
-        :param metrics: The list of metrics to profile
-        :param save: Whether to save the matrices to a file
-        :param location: The location to save the matrices, if save = True, e.g "./resources",
-        default is current directory
-        """
-        emission_matrix, emission_matrix_observations = self.get_matrices_of_observation(self.statistics,
-                                                                                         metric, hidden_states)
-        self.emission_matrix = emission_matrix
-        self.emission_matrix_observations = emission_matrix_observations
-        self.transition_matrix = transition_matrix
-        self.start_state_probs = self.calculate_initial_states(self.transition_matrix)
-        self.hidden_states = hidden_states
-        if save_model and location:
-            np.save(f'{location}/transition_matrix.npy', transition_matrix)
-            np.save(f'{location}/hidden_states.npy', hidden_states)
-            np.save(f'{location}/start_state_probs.npy', self.start_state_probs)
-            np.save(f'{location}/emission_matrix_{metric}.npy', emission_matrix)
-            np.save(f'{location}/emission_matrix_observations_{metric}.npy', emission_matrix_observations)
-
-    def load_model(self, location: str, metric: str) -> None:
-        """
-        Loads the HMM model from the given location.
-
-        :param location: The location of the model files, default is current directory
-        :param metric: The data metric to use
-        """
-        self.transition_matrix = np.load(f'{location}/transition_matrix.npy')
-        self.hidden_states = np.load(f'{location}/hidden_states.npy')
-        self.start_state_probs = np.load(f'{location}/start_state_probs.npy')
-        self.emission_matrix = np.load(f'{location}/emission_matrix_{metric}.npy')
-        self.emission_matrix_observations = np.load(f'{location}/emission_matrix_observations_{metric}.npy')
-
-    def profile_sequence(self, sequence: List[int]) -> List[str]:
-        """
-        Profiles a sequence of observations based on the HMM model.
-
-        :param sequence: The sequence of observations
-        :return: The most likely sequence of states
-        """
-        
-        path = HMMProfiler.viterbi(self.hidden_states, self.start_state_probs,
-                                   self.transition_matrix, self.emission_matrix,
-                                   sequence, self.emission_matrix_observations)
-        profiled_sequence = []
-        for i in range(len(path)):
-            profiled_sequence.append(self.hidden_states[int(path[i])])
-        
-        return profiled_sequence
-
-    def get_matrices_of_observation(self, statistics: List[EmulationStatistics],
-                                    metric: str, states: List[str]) -> Tuple[List[List[float]], List[int]]:
-        """
-        Creates the emission matrix for a given metric based on the statistics from the EmulationStatistics objects.
-
-        :param statistics: The list of EmulationStatistics objects
-        :param metric: The metric to get the emission matrix for
-        :return: The emission matrix, the list of observations, the list of states
-        """
-        emission_matrix = []
-        attack_observations = {}
-        attack_observations_total_counts = {}
-        all_keys = set()
-        
-        for stats in statistics:
-            for condition, metric_distribution in stats.conditionals_counts.items():
-                action = condition.split('_')
-                if action[0] == 'no':
-                    action[0] = 'no_intrusion'
-                if action[0] not in attack_observations:
-                    # We are not intrested in the observations from 'intrusion' or 'A:Continue'
-                    if action[0] == 'intrusion' or action[0] == 'A:Continue':
-                        continue
-                    else:
-                        # Add the observations of the attack to the dictionary
-                        if metric in metric_distribution:
-                            attack_observations[action[0]] = metric_distribution[metric]
-                        # Sum the total counts of the observations
-                            attack_observations_total_counts[action[0]] = sum(attack_observations[action[0]].values())
-                # Aggregate the counts from the metric distribution
-                else:
-                    counts_observation = metric_distribution[metric]
-                    for element in counts_observation:
-                        if element in attack_observations[action[0]]:
-                            # Aggregate the counts if the element is already in the dictionary
-                            attack_observations[action[0]][element] += counts_observation[element]
-                        else:
-                            attack_observations[action[0]][element] = counts_observation[element]
-                    # Sum the total counts of the observations
-                    attack_observations_total_counts[action[0]] += sum(attack_observations[action[0]].values())
-
-                # Store all possible values for the observation
-                if action[0] in attack_observations:
-                    all_keys.update(attack_observations[action[0]])
-        
-        # Normalize the counts
-        for attack, _ in attack_observations.items():
-            attack_observations_total_counts[attack] = sum(attack_observations[attack].values())
-            for key in all_keys:
-                int_key = int(key)
-                if key in attack_observations[attack]:
-                    count = attack_observations[attack].pop(key, 0)
-                    attack_observations[attack][int_key] = count / attack_observations_total_counts[attack]
-                else:
-                    attack_observations[attack][int_key] = 0
-            # Sort the dictionary by key
-            attack_observations[attack] = dict(sorted(attack_observations[attack].items()))
-
-        # Take any attack as the reference to get the keys
-        emission_matrix_observations = []
-        emission_matrix_states = []
-        # Create the emission matrix
-        for state in states:
-            if state in attack_observations:
-                # Normalize the and then append
-                emission_matrix.append(list(attack_observations[state].values()))
-                # Get the keys of all observations
-                emission_matrix_observations = list(attack_observations[state].keys())
-                emission_matrix_states.append(state)
-            else:
-                # LaPlace smoothing for missing observations
-                num_keys = len(all_keys)
-                laplace_probability = 1 / (num_keys + 2)
-                laplace_sum = laplace_probability * num_keys
-                laplace_probability_adj = laplace_probability / laplace_sum
-                emission_matrix.append([laplace_probability_adj] * num_keys)
-                emission_matrix_states.append(state)
-
-        # Check if the sum of the probabilities is 1
-        for i in range(len(emission_matrix)):
-            sum_prob = round(sum(emission_matrix[i]), 10)
-            if sum_prob != 1:
-                print(f'Sum of probabilities for state {emission_matrix_states[i]} is {sum_prob}')
-
-        return (emission_matrix, emission_matrix_observations)
-
-    def convert_states_to_profiles(self, states: List[str]) -> List[Union[AttackProfiler, str]]:
-        """
-        Converts a list of states to a list of AttackProfiles.
-
-        :param states: The list of states to convert
-        :return: The list of EmulationAttackerActionId
-        """
-
-        new_states: List[Union[AttackProfiler, str]] = []
-        for state in states:
-            if state == 'A:Continue':
-                action = EmulationAttackerAction(id=EmulationAttackerActionId.CONTINUE, name="Continue", cmds=[],
-                                                 type=EmulationAttackerActionType.CONTINUE, descr="CONTINUE", ips=[],
-                                                 index=0, action_outcome=EmulationAttackerActionOutcome.CONTINUE)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:CVE-2015-1427 exploit':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.CVE_2015_1427_EXPLOIT, name="CVE-2015-1427 exploit", cmds=[],
-                    type=EmulationAttackerActionType.EXPLOIT,
-                    descr="Uses the CVE-2015-1427 vulnerability to "
-                    "get remote code execution and then sets up a SSH backdoor"
-                    "to upgrade the channel", index=-1, ips=[],
-                    action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:DVWA SQL Injection Exploit':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.DVWA_SQL_INJECTION, name="DVWA SQL Injection Exploit",
-                    cmds=[], type=EmulationAttackerActionType.EXPLOIT,
-                    descr="Uses the DVWA SQL Injection exploit to extract secret passwords",
-                    index=-1, ips=[], action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:Install tools':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.INSTALL_TOOLS, name="Install tools", cmds=[],
-                    type=EmulationAttackerActionType.POST_EXPLOIT,
-                    descr="If taken root on remote machine, installs pentest tools, e.g. nmap",
-                    index=-1, ips=[], action_outcome=EmulationAttackerActionOutcome.PIVOTING)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:Network service login':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.NETWORK_SERVICE_LOGIN, name="Network service login",
-                    cmds=[], type=EmulationAttackerActionType.POST_EXPLOIT,
-                    descr="Uses known credentials to login to network services on a server",
-                    index=-1, ips=[], action_outcome=EmulationAttackerActionOutcome.LOGIN)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:Ping Scan':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.PING_SCAN_HOST, name="Ping Scan",
-                    cmds=[], type=EmulationAttackerActionType.RECON,
-                    descr="A host discovery scan, it is quick because it only checks of hosts "
-                    "are up with Ping, without scanning the ports.", ips=[], index=-1,
-                    action_outcome=EmulationAttackerActionOutcome.INFORMATION_GATHERING, backdoor=False)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:Sambacry Explolit':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.SAMBACRY_EXPLOIT, name="Sambacry Explolit", cmds=[],
-                    type=EmulationAttackerActionType.EXPLOIT,
-                    descr="Uses the sambacry shell to get remote code execution and then"
-                    "sets up a SSH backdoor to upgrade the channel",
-                    index=-1, ips=[], action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:ShellShock Explolit':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.SHELLSHOCK_EXPLOIT, name="ShellShock Exploit",
-                    cmds=[], type=EmulationAttackerActionType.EXPLOIT,
-                    descr="Uses the Shellshock exploit and curl to do remote code execution and create a backdoor",
-                    index=-1, ips=[], action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:SSH dictionary attack for username=pw':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.SSH_SAME_USER_PASS_DICTIONARY_HOST,
-                    name="SSH dictionary attack for username=pw", cmds=[],
-                    type=EmulationAttackerActionType.EXPLOIT, index=-1,
-                    descr="A dictionary attack that tries common passwords and usernames for SSH"
-                    "where username=password", ips=[], action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:FTP dictionary attack for username=pw':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.FTP_SAME_USER_PASS_DICTIONARY_HOST,
-                    name="FTP dictionary attack for username=pw", cmds=[], type=EmulationAttackerActionType.EXPLOIT,
-                    index=-1, descr="A dictionary attack that tries common passwords and"
-                    "usernames for FTP where username=password", ips=[],
-                    action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:Telnet dictionary attack for username=pw':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.TELNET_SAME_USER_PASS_DICTIONARY_HOST,
-                    name="Telnet dictionary attack for username=pw", cmds=[],
-                    type=EmulationAttackerActionType.EXPLOIT, index=-1,
-                    descr="A dictionary attack that tries common passwords and usernames for"
-                    "Telnet where username=password", ips=[],
-                    action_outcome=EmulationAttackerActionOutcome.SHELL_ACCESS)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:CVE-2010-0426 exploit':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.CVE_2010_0426_PRIV_ESC,
-                    name="CVE-2010-0426 exploit", cmds=[], type=EmulationAttackerActionType.PRIVILEGE_ESCALATION,
-                    descr="Uses the CVE-2010-0426 vulnerability to perform privilege escalation to get root access",
-                    index=-1, ips=[], action_outcome=EmulationAttackerActionOutcome.PRIVILEGE_ESCALATION_ROOT)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'A:TCP SYN (Stealth) Scan':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.TCP_SYN_STEALTH_SCAN_HOST, name="TCP SYN (Stealth) Scan",
-                    cmds=[], type=EmulationAttackerActionType.RECON,
-                    descr="A stealthy and fast TCP SYN scan to detect open TCP ports on the subnet", ips=[],
-                    index=-1, action_outcome=EmulationAttackerActionOutcome.INFORMATION_GATHERING, backdoor=False)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            elif state == 'ssh backdoor':
-                action = EmulationAttackerAction(
-                    id=EmulationAttackerActionId.SSH_BACKDOOR, name="Install SSH backdoor",
-                    cmds=[], type=EmulationAttackerActionType.POST_EXPLOIT,
-                    descr="If taken root on remote machine, installs a ssh backdoor useful for"
-                    "upgrading telnetor weaker channels", index=-1, ips=[],
-                    action_outcome=EmulationAttackerActionOutcome.PIVOTING, alt_cmds=None, backdoor=True)
-                p = AttackProfiler.get_attack_profile(action)
-                new_states.append(p)
-            else:
-                new_states.append(state)
-
-        return new_states
-   
-    def calculate_initial_states(self, transition_matrix: List[List[float]]) -> List[float]:
-        """
-        Calculates the initial states probabilities based on the transition matrix.
-
-        1 / (# of states)
-
-        :param transition_matrix: The transition matrix
-
-        :return: The start states probabilities
-        """
-        start_states = []
-        total_states = len(transition_matrix)
-        for _ in range(total_states):
-            start_states.append(1 / total_states)
-
-        return start_states
-
-    @staticmethod
-    def viterbi(hidden_states: List[EmulationAttackerActionId], init_probs: List[float],
-                trans_matrix: List[List[float]], emission_matrix: List[List[float]],
-                obs: List[int], emissions_list: List[int]) -> List[float]:
-        """
-        Viterbi algorithm for Hidden Markov Models (HMM).
-
-        :param hidden_states: The hidden states
-        :param init_probs: The initial probabilities of the hidden states
-        :param trans_matrix: The transition matrix
-        :param emission_matrix: The emission matrix
-        :param obs: The observation sequence
-        :param emissions_list: The list of possible observations
-
-        :return: The most likely sequence of hidden states
-        """
-        # Convert the emissions list to a numpy array, to use the where function
-        emissions_list_typed: np.ndarray[int, Any] = np.array(emissions_list)
-
-        # Check that the sum equals 1
-        for i in range(len(emission_matrix)):
-            if round(sum(emission_matrix[i]), 10) != 1:
-                print(f'Sum of probabilities for state {hidden_states[i]} is not 1')
-                print(f'Sum of probabilities: {sum(emission_matrix[i])}')
-
-        # The number of hidden states
-        S = len(hidden_states)
-        # The number of observations
-        T = len(obs)
-
-        # The Viterbi matrix (prob) T x S matrix of zeroes
-        prob = np.zeros((T, S))
-        # The backpointer matrix (prev)
-        prev = np.empty((T, S))
-        # Initialization
-        for i in range(S):
-            # Fetch the index of the observation in the emission_matrix
-            index, = np.where(emissions_list_typed == obs[0])
-            if index[0].size > 0:
-                prob[0][i] = init_probs[i] * emission_matrix[i][index[0]]
-            else:
-                print(f'Observation {obs[0]} not found in the emission matrix')
-                sys.exit(1)
-
-        # Recursion
-        for t in range(1, T):
-            index, = np.where(emissions_list_typed == obs[t])
-            for i in range(S):
-                max_prob = -1
-                max_state = -1
-                for j in range(S):
-                    new_prob = prob[t - 1][j] * trans_matrix[j][i] * emission_matrix[i][index[0]]
-                    if new_prob > max_prob:
-                        max_prob = new_prob
-                        max_state = j
-                prob[t][i] = max_prob
-                prev[t][i] = max_state
-
-        path = np.zeros(T)
-        path[T - 1] = np.argmax(prob[T - 1])
-        for t in range(T - 2, -1, -1):
-            path[t] = prev[t + 1][int(path[t + 1])]
-        # Convert the path to a list
-        typed_path: List[float] = path.tolist()
-
-        return typed_path
-
-    def generate_sequence(self, intrusion_length: int, initial_state_index: int,
-                          seed: Union[int, None] = None) -> Tuple[List[str], List[int]]:
-        """
-        Generates a sequence of states and corresponding observations based on the given emission matrix,
-        and transition matrix. First, a sequence of observation from 'no intrusion' is generated
-        based on the geometric distribution of the initial state. Then, a sequence observations and states are
-        generated based on emission matrix and transition matrix. The length of this intrusion
-        sequence is given by the intrusion_length parameter.
 
-        :param P_obs: The emission matrix
-        :param P_states: The transition matrix
-        :param states: The list of states
-        :param observations: The list of observations
-        :param intrusion_length: The length of the intrusion
-        :param initial_state: The initial state
-        return: The sequence of states and observations
-        """
         P_obs = self.emission_matrix
         P_states = self.transition_matrix
         states = self.hidden_states
         observations = self.emission_matrix_observations
+
         if seed:
             np.random.seed(seed)
         obs_len = len(observations)
         states_len = len(states)
-
         # Return the geometric distribution of the initial state
         dist = np.random.geometric(p=P_states[initial_state_index][0], size=1000)
         T_i = round(sum(dist) / len(dist))
@@ -878,7 +424,6 @@ class HMMProfiler:
         state_seq = [states[initial_state_index]] * T_i
         obs_seq = []
         for i in range(T_i):
-
             o_i = np.random.choice(obs_len, p=P_obs[initial_state_index])
             obs_seq.append(observations[o_i])
 
@@ -893,12 +438,11 @@ class HMMProfiler:
         s_i = intrusion_start_state
         if intrusion_length == 1:
             return state_seq, obs_seq
-        for i in range(intrusion_length):
+        for i in range(intrusion_length - 1):
             # si ~ Ps(si | si-1)
             s_i = np.random.choice(states_len, p=P_states[s_i])
             # oi ~ Po(oi | si)
             o_i = np.random.choice(obs_len, p=P_obs[s_i])
             state_seq.append(states[s_i])
             obs_seq.append(observations[o_i])
-        
         return state_seq, obs_seq

{csle_attack_profiler-0.6.0.dist-info → csle_attack_profiler-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: csle-attack-profiler
-Version: 0.6.0
+Version: 0.6.1
 Summary: Library with MITRE attack profiler for CSLE
 Author: Bength Pappila
 Author-email: brpa@kth.se
@@ -15,8 +15,8 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Intended Audience :: Science/Research
 Requires-Python: >=3.8
 Requires-Dist: mitreattack-python ==2.0.14
-Requires-Dist: csle-base ==0.6.0
-Requires-Dist: csle-common ==0.6.0
+Requires-Dist: csle-base ==0.6.1
+Requires-Dist: csle-common ==0.6.1
 Provides-Extra: testing
 Requires-Dist: pytest >=6.0 ; extra == 'testing'
 Requires-Dist: pytest-cov >=2.0 ; extra == 'testing'

csle_attack_profiler-0.6.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+csle_attack_profiler/__init__.py,sha256=C7_gE0lIQJ8Wh2jgU8C8P_xyvq76bKTf0gm8bGYhMBk,37
+csle_attack_profiler/__version__.py,sha256=gd3s3RotD0_KL90Tua-YkOr60Jm2C2_wvlEhAT08068,22
+csle_attack_profiler/attack_profiler.py,sha256=bYqSeItJkhswronvIRFPyvf1rtvUKEWPPF4K2t1JVqE,10535
+csle_attack_profiler/hmm_profiling.py,sha256=cHE2wImgBJ3xwW33S0JcTwsiccstijR9uRMeBPXIT18,24167
+csle_attack_profiler-0.6.1.dist-info/METADATA,sha256=qcoiNvO6GxPrQRxB9P9IiaIxwR46ZKAIRu2nDd0_svA,2006
+csle_attack_profiler-0.6.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+csle_attack_profiler-0.6.1.dist-info/top_level.txt,sha256=OuI4zvPo3MQYLQ7Dqm32oM0V8rNdwHe9tjtngx2KVtA,21
+csle_attack_profiler-0.6.1.dist-info/RECORD,,

csle_attack_profiler-0.6.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-csle_attack_profiler/__init__.py,sha256=C7_gE0lIQJ8Wh2jgU8C8P_xyvq76bKTf0gm8bGYhMBk,37
-csle_attack_profiler/__version__.py,sha256=CBY3jsC-9HCm7eZ6CKD-sYLCejqOJ1pYWPQM4LGIXcI,22
-csle_attack_profiler/attack_profiler.py,sha256=bYqSeItJkhswronvIRFPyvf1rtvUKEWPPF4K2t1JVqE,10535
-csle_attack_profiler/hmm_profiling.py,sha256=891sNt1bWlqIIUKCnzZaEGnEsIhB6TY3ob46MSHymoI,48264
-csle_attack_profiler-0.6.0.dist-info/METADATA,sha256=biQd6Up9qFTEihixwI4t_jGFAhkr9UWsTPG8bIdL2kM,2006
-csle_attack_profiler-0.6.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-csle_attack_profiler-0.6.0.dist-info/top_level.txt,sha256=OuI4zvPo3MQYLQ7Dqm32oM0V8rNdwHe9tjtngx2KVtA,21
-csle_attack_profiler-0.6.0.dist-info/RECORD,,