From a3e4f9dbaa85531d17a807f55f9a96b49ee1846c Mon Sep 17 00:00:00 2001
From: thecliguy <adam@thecliguy.co.uk>
Date: Wed, 28 Oct 2020 15:56:12 +0000
Subject: [PATCH] Added similar algorithm suggestions to --lookup (#80)

---
 src/ssh_audit/ssh_audit.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/ssh_audit/ssh_audit.py b/src/ssh_audit/ssh_audit.py
index 344498c..8fe1330 100755
--- a/src/ssh_audit/ssh_audit.py
+++ b/src/ssh_audit/ssh_audit.py
@@ -899,12 +899,30 @@ def algorithm_lookup(alg_names: str) -> int:
         if alg_name not in algorithms_dict_flattened
     ]
 
+    similar_algorithms = [
+        alg_unknown + " --> (" + alg_type + ") " + alg_name
+        for alg_unknown in algorithms_not_found
+        for alg_type in adb.keys()
+        for alg_name in adb[alg_type]
+        # Perform a case-insensitive comparison using 'casefold'
+        # and match substrings using the 'in' operator.
+        if alg_unknown.casefold() in alg_name.casefold()
+    ]
+
     if len(algorithms_not_found) > 0:
         retval = exitcodes.FAILURE
         out.head('# unknown algorithms')
         for algorithm_not_found in algorithms_not_found:
             out.fail(algorithm_not_found)
 
+    print()
+
+    if len(similar_algorithms) > 0:
+        retval = exitcodes.FAILURE
+        out.head('# suggested similar algorithms')
+        for similar_algorithm in similar_algorithms:
+            out.warn(similar_algorithm)
+
     return retval