error handling cube attack, less round than model precomputed rounds

src/tea3/cube_attack.py

@@ -52,7 +52,10 @@ def build_target_poly(
     if fixed_bits:
         model = specialize_model(model, fixed_bits)
 
-    for _ in range((rounds - base_model.step_count) if base_model is not None else rounds):
+    start = model.step_count
+    if rounds < start:
+        raise ValueError(f"rounds={rounds} is smaller than current step_count={start}")
+    for _ in range(rounds - start):
         model.step(skip_abstract=True)
 
     return model.R_bits[target_reg][target_bit]
@@ -203,13 +206,20 @@ def run_cube_attack(
     tmp_model = Tea3Model()
     public_vars = pick_public_vars(tmp_model, fixed_bits=fixed_bits)
 
-    poly = build_target_poly(
+    try:
-        rounds=rounds,
+        poly = build_target_poly(
-        target_reg=target_reg,
+            rounds=rounds,
-        target_bit=target_bit,
+            target_reg=target_reg,
-        fixed_bits=fixed_bits,
+            target_bit=target_bit,
-        base_model=model,
+            fixed_bits=fixed_bits,
-    )
+            base_model=model,
+        )
+    except ValueError as e:
+        print("=" * 50)
+        print("Cube attack aborted.")
+        print(f"Error: {e}")
+        print("=" * 50)
+        return
 
     print("=" * 50)
     if model is not None: