exhaustive 2 bits XOR search

src/tea3/cli.py

@@ -3,7 +3,7 @@ from tea3.cliutils import prompt_int, prompt_choice, prompt_list
 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.variable_xor import run_variable_xor, run_exhaustive_two_bit_xor
 from tea3.f31f32 import run_f31f32
 
 
@@ -70,6 +70,19 @@ def run_variable_xor_cli():
     print("\n" + "=" * 50)
     print("Done.")
 
+def run_exhaustive_two_bit_xor_cli():
+    print("\nR registers are indexed 0–7; bits within each register are 0–7.")
+    print("This mode checks all 2-bit XOR pairs and prints the best one(s) at each step.")
+
+    steps = prompt_int("How many steps? (1–100): ", 1, 100)
+    target_reg = prompt_int("Target register (0–7): ", 0, 7)
+
+    print("-" * 50)
+    run_exhaustive_two_bit_xor(steps, target_reg)
+
+    print("\n" + "=" * 50)
+    print("Done.")
+
 
 
 def main():
@@ -83,8 +96,9 @@ def main():
     print("  3) S box analysis")
     print("  4) variable XOR")
     print("  5) F31, F32 analysis")
+    print("  6) Exhaustive 2-bit XOR search")
 
-    mode = prompt_choice("Your choice (1, 2, 3, 4 or 5): ", {1, 2, 3, 4, 5})
+    mode = prompt_choice("Your choice (1, 2, 3, 4, 5 or 6): ", {1, 2, 3, 4, 5, 6})
 
     if mode == 1:
         run_classic_cli()
@@ -96,6 +110,8 @@ def main():
         run_variable_xor_cli()
     elif mode == 5:
         run_f31f32()
+    elif mode == 6:
+        run_exhaustive_two_bit_xor_cli()
 
 
 main()