mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-10-06 13:49:30 +00:00
cecf43d648
* Pyupgrade to Python 3.9
* updating DIRECTORY.md
Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com>
33 lines
952 B
Python
33 lines
952 B
Python
from __future__ import annotations
|
|
|
|
|
|
def find_primitive(n: int) -> int | None:
|
|
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)
|