Initial commit

lab1-Diffie-Hellman/app/utils/__init__.py

@@ -0,0 +1 @@
+#

lab1-Diffie-Hellman/app/utils/check_case.py

@@ -0,0 +1,21 @@
+from core.public_params import PublicParams
+from core.participant import Participant
+
+
+def check_case(prime_modulus, generator, x, y, K_expected):
+    """
+    Строит участников, имитирует обмен и проверяет совпадение ключа с эталоном.
+    Возвращает: ok (bool), (A, B, K_a, K_b)
+    """
+    params = PublicParams(prime_modulus, generator)
+    alice = Participant(params, secret=x)
+    bob = Participant(params, secret=y)
+
+    A = alice.public_value
+    B = bob.public_value
+
+    K_a = alice.compute_shared_key(B)
+    K_b = bob.compute_shared_key(A)
+
+    ok = (K_a == K_b == K_expected)
+    return ok, (A, B, K_a, K_b)

lab1-Diffie-Hellman/app/utils/format_line.py

@@ -0,0 +1,13 @@
+def format_line(i, A, B, K_a, K_b, ok):
+    """
+    Форматирует вывод строки для теста.
+    """
+    status = "OK" if ok else "FAIL"
+    return (
+        f"#{i:>2}:  "
+        f"A={A:>6}  "
+        f"B={B:>6}  "
+        f"K_A={K_a:>6}  "
+        f"K_B={K_b:>6}  "
+        f"-> {status}"
+    )