F31 F32 analysis

src/tea3/cli.py

@@ -3,6 +3,7 @@ from tea3.tea3model import Tea3Model
 from tea3.variable_search import run_exhaustive
 from tea3.sbox import run_sbox
 from tea3.variable_xor import run_variable_xor
+from tea3.f31f32 import run_f31f32
 
 def prompt_int(message: str, lo: int, hi: int) -> int:
     while True:
@@ -155,8 +156,9 @@ def main():
     print("  2) Exhaustive variable-change search")
     print("  3) S box analysis")
     print("  4) variable XOR")
+    print("  5) F31, F32 analysis")
 
-    mode = prompt_choice("Your choice (1, 2, 3 or 4): ", {1, 2, 3, 4})
+    mode = prompt_choice("Your choice (1, 2, 3, 4 or 5): ", {1, 2, 3, 4, 5})
 
     if mode == 1:
         run_classic_cli()
@@ -164,8 +166,10 @@ def main():
         run_exhaustive_cli()
     elif mode == 3:
         run_sbox()
-    else:
+    elif mode == 4:
         run_variable_xor_cli()
+    elif mode == 5:
+        run_f31f32()
 
 
 main()