sbox analysis

src/tea3/cli.py

@@ -1,6 +1,7 @@
 from tea3.pretty_print import pretty_print
 from tea3.tea3model import Tea3Model
 from tea3.variable_search import run_exhaustive
+from tea3.sbox import run_sbox
 
 
 def prompt_int(message: str, lo: int, hi: int) -> int:
@@ -83,13 +84,16 @@ def main():
     print("\nChoose a mode:")
     print("  1) Classic inspection")
     print("  2) Exhaustive variable-change search")
+    print("  3) S box analysis")
 
-    mode = prompt_choice("Your choice (1 or 2): ", {1, 2})
+    mode = prompt_choice("Your choice (1, 2 or 3): ", {1, 2, 3})
 
     if mode == 1:
         run_classic_cli()
-    else:
+    elif mode == 2:
         run_exhaustive_cli()
+    else:
+        run_sbox()
 
 
 main()