mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-10-07 06:09:28 +00:00
14bcb580d5
* Initial fix for mypy errors in some cipher algorithms * fix(mypy): Update type hints * fix(mypy): Update type hints for enigma_machine2.py * Update as per the suggestion Co-authored-by: Christian Clauss <cclauss@me.com> Co-authored-by: Christian Clauss <cclauss@me.com>
33 lines
948 B
Python
33 lines
948 B
Python
from typing import Optional
|
|
|
|
|
|
def find_primitive(n: int) -> Optional[int]:
|
|
for r in range(1, n):
|
|
li = []
|
|
for x in range(n - 1):
|
|
val = pow(r, x, n)
|
|
if val in li:
|
|
break
|
|
li.append(val)
|
|
else:
|
|
return r
|
|
return None
|
|
|
|
|
|
if __name__ == "__main__":
|
|
q = int(input("Enter a prime number q: "))
|
|
a = find_primitive(q)
|
|
if a is None:
|
|
print(f"Cannot find the primitive for the value: {a!r}")
|
|
else:
|
|
a_private = int(input("Enter private key of A: "))
|
|
a_public = pow(a, a_private, q)
|
|
b_private = int(input("Enter private key of B: "))
|
|
b_public = pow(a, b_private, q)
|
|
|
|
a_secret = pow(b_public, a_private, q)
|
|
b_secret = pow(a_public, b_private, q)
|
|
|
|
print("The key value generated by A is: ", a_secret)
|
|
print("The key value generated by B is: ", b_secret)
|