mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-24 05:21:09 +00:00
9d745b6156
* updating DIRECTORY.md * updating DIRECTORY.md * Fixed accidental commit of file I have't touched * fixup! Format Python code with psf/black push * updating DIRECTORY.md * updating DIRECTORY.md * Fixed some suggested coding style issues * Update rsa_key_generator.py * Update rsa_key_generator.py Co-authored-by: github-actions <${GITHUB_ACTOR}@users.noreply.github.com> Co-authored-by: John Law <johnlaw.po@gmail.com>
26 lines
760 B
Python
26 lines
760 B
Python
def find_primitive(n: int) -> 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
|
|
|
|
|
|
if __name__ == "__main__":
|
|
q = int(input("Enter a prime number q: "))
|
|
a = find_primitive(q)
|
|
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)
|