|
|
@@ -106,7 +106,7 @@ class EternalBlueExploit(BaseAttack.BaseAttack):
|
|
|
source_ttl_prob_dict = lea.Lea.fromValFreqsDict(source_ttl_dist)
|
|
|
source_ttl_value = source_ttl_prob_dict.random()
|
|
|
else:
|
|
|
- source_ttl_value = Util.handle_most_used_outputs(self.statistics.process_db_query("most_used(ttlValue)"))
|
|
|
+ source_ttl_value = Util.handle_most_used_outputs(self.statistics.get_most_used_ttl_value())
|
|
|
|
|
|
destination_ttl_dist = self.statistics.get_ttl_distribution(ip_destination)
|
|
|
if len(destination_ttl_dist) > 0:
|
|
|
@@ -114,7 +114,7 @@ class EternalBlueExploit(BaseAttack.BaseAttack):
|
|
|
destination_ttl_value = destination_ttl_prob_dict.random()
|
|
|
else:
|
|
|
destination_ttl_value = Util.handle_most_used_outputs(
|
|
|
- self.statistics.process_db_query("most_used(ttlValue)"))
|
|
|
+ self.statistics.get_most_used_ttl_value())
|
|
|
|
|
|
# Set Window Size based on Window Size distribution of IP address
|
|
|
source_win_dist = self.statistics.get_win_distribution(ip_source)
|
|
|
@@ -132,7 +132,7 @@ class EternalBlueExploit(BaseAttack.BaseAttack):
|
|
|
destination_win_prob_dict = lea.Lea.fromValFreqsDict(destination_win_dist)
|
|
|
|
|
|
# Set MSS (Maximum Segment Size) based on MSS distribution of IP address
|
|
|
- mss_value = Util.handle_most_used_outputs(self.statistics.process_db_query("most_used(mssValue)"))
|
|
|
+ mss_value = Util.handle_most_used_outputs(self.statistics.get_most_used_mss_value())
|
|
|
if not mss_value:
|
|
|
mss_value = 1465
|
|
|
|
