import random
isprime = lambda number: all(number % i for i in range(2, number))
class RSA:
e = 3
def __init__(self, *args, **kwargs):
self.initialize(*args, **kwargs)
def initialize(self, *args, **kwargs):
if args: self.p, self.q = args
else:
kwargs = kwargs or {'start': 100, 'end':400}
simples = [i for i in range(kwargs['start'], kwargs['end']) if isprime(i)]
self.p = random.choice(simples)
while True:
self.q = random.choice(simples)
if self.q != self.p: break;
print (self.p, self.q)
self.n = self.q * self.p
self.euler_value = (self.p-1)*(self.q-1)
if not self.create_secret_key():
if args:
arg = int(input('Worth p/q pair for secret_exp calculate. Input other q value: '))
self.initialize(args[0], arg)
else:
self.initialize()
def create_secret_key(self, start=100, end=100000):
for i in range(start, end):
if (i * self.e) % self.euler_value == 1:
self.secret_exp = i
return self.secret_exp, self.n
return None
generate_open_key = lambda self: (self.e, self.n,)
encrypt_value = lambda self, value: (value ** self.e) % self.n
decrypt_value = lambda self, value: (value ** self.secret_exp) % self.n
def main():
rsa = RSA()
print(rsa)
# s_key = rsa.create_secret_key()
s_key = rsa.secret_exp
print(s_key)
o_key = rsa.generate_open_key()
print(o_key)
val = int(input("Enter value for decryption: ")) or 111111
enc_value = (rsa.encrypt_value(val))
print (enc_value)
input("Enter any key for decryption ")
print(rsa.decrypt_value(enc_value))
# print( isprime(6111579))
if __name__ == '__main__':
main()
