user
110 lines
4.1 KiB
Python
110 lines
4.1 KiB
Python
from argparse import ArgumentParser
|
||
from secrets import randbelow
|
||
|
||
from utils.check_case import check_case
|
||
from utils.format_line import format_line
|
||
from orchestration.param_generator import ParamGenerator
|
||
|
||
|
||
def _random_secret() -> int:
|
||
# 2..9999 включительно (меньше 10000, исключаем 0/1)
|
||
return 2 + randbelow(9998)
|
||
|
||
|
||
def run_auto(count: int) -> None:
|
||
"""
|
||
Автоматический режим:
|
||
- случайно генерируем (p, q, g);
|
||
- случайно выбираем x, y (<10000);
|
||
- вычисляем K и проверяем через check_case;
|
||
- печатаем результат через format_line
|
||
"""
|
||
for i in range(1, count + 1):
|
||
p, q, g = ParamGenerator.generate() # q выводим для информации
|
||
x = _random_secret()
|
||
y = _random_secret()
|
||
|
||
# Вычислим K заранее, чтобы передать как эталон в check_case
|
||
# (тем самым используем общий хелпер единообразно)
|
||
ok, (A, B, K_a, K_b) = check_case(p, g, x, y, K_expected=None)
|
||
# Когда K_expected=None, наша старая версия check_case
|
||
# сравнивала строго с эталоном. Если у тебя именно так,
|
||
# просто заменим на повторный вызов с «эталоном = K_a»
|
||
if K_a != K_b:
|
||
ok = False
|
||
else:
|
||
ok, (A, B, K_a, K_b) = check_case(p, g, x, y, K_expected=K_a)
|
||
|
||
print(f"[auto #{i}] p={p} q={q} g={g}")
|
||
print(format_line(i, A, B, K_a, K_b, ok))
|
||
|
||
|
||
def run_manual(p: int, g: int, x: int, y: int, k: int | None) -> None:
|
||
"""
|
||
Ручной режим:
|
||
- принимаем p, g, x, y;
|
||
- если указан k — проверяем с эталоном;
|
||
- если k не указан — просто считаем и показываем
|
||
"""
|
||
if k is None:
|
||
ok, (A, B, K_a, K_b) = check_case(p, g, x, y, K_expected=None)
|
||
ok = (K_a == K_b)
|
||
print(f"[manual] p={p} g={g}")
|
||
print(format_line(1, A, B, K_a, K_b, ok))
|
||
else:
|
||
ok, (A, B, K_a, K_b) = check_case(p, g, x, y, K_expected=k)
|
||
print(f"[manual] p={p} g={g} K_expected={k}")
|
||
print(format_line(1, A, B, K_a, K_b, ok))
|
||
|
||
|
||
def parse_args():
|
||
ap = ArgumentParser(
|
||
description=(
|
||
"Demo Diffie–Hellman: режимы manual и auto\n"
|
||
"Auto: случайно генерирует (p,q,g) и секреты (<10000)\n"
|
||
"Manual: используется заданный набор параметров"
|
||
)
|
||
)
|
||
ap.add_argument(
|
||
"--mode",
|
||
choices=["auto", "manual"],
|
||
default="auto",
|
||
help="Режим работы (auto|manual). По умолчанию auto",
|
||
)
|
||
ap.add_argument(
|
||
"--count",
|
||
type=int,
|
||
default=5,
|
||
help="Сколько генераций в auto-режиме (по умолчанию 5)",
|
||
)
|
||
ap.add_argument("--p", type=int, help="Модуль (prime_modulus) в manual")
|
||
ap.add_argument("--g", type=int, help="Генератор (generator) в manual")
|
||
ap.add_argument("--x", type=int, help="Секрет Алисы (<10000) в manual")
|
||
ap.add_argument("--y", type=int, help="Секрет Боба (<10000) в manual")
|
||
ap.add_argument(
|
||
"--k",
|
||
type=int,
|
||
help="Эталонный общий ключ для проверки в manual (необязательно)",
|
||
)
|
||
return ap.parse_args()
|
||
|
||
|
||
def main():
|
||
args = parse_args()
|
||
if args.mode == "auto":
|
||
run_auto(count=args.count)
|
||
return
|
||
|
||
# manual
|
||
missing = [n for n, v in [("p", args.p), ("g", args.g),
|
||
("x", args.x), ("y", args.y)] if v is None]
|
||
if missing:
|
||
need = ", ".join(missing)
|
||
raise SystemExit(f"manual-режим: необходимо указать параметры: {need}")
|
||
run_manual(p=args.p, g=args.g, x=args.x, y=args.y, k=args.k)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|
||
|