Secure_File_Tranfer_Protocol/RSA_encryption.py at master · Tejas-Rote/Secure_File_Tranfer_Protocol · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
from Crypto.PublicKey import RSA
from Crypto.Cipher import PKCS1_OAEP
import binascii

keyPair = RSA.generate(3072)

pubKey = keyPair.publickey()
print(f"Public key:  (n={hex(pubKey.n)}, e={hex(pubKey.e)})")
pubKeyPEM = pubKey.exportKey()
print(pubKeyPEM.decode('ascii'))

print(f"Private key: (n={hex(pubKey.n)}, d={hex(keyPair.d)})")
privKeyPEM = keyPair.exportKey()
print(privKeyPEM.decode('ascii'))


msg = b'A message for encryption'
encryptor = PKCS1_OAEP.new(pubKey)
encrypted = encryptor.encrypt(msg)
print("Encrypted:", binascii.hexlify(encrypted))

decryptor = PKCS1_OAEP.new(keyPair)
decrypted = decryptor.decrypt(encrypted)
print('Decrypted:', decrypted)


# print('RSA Encrypt')
# with open("Keys/publicKey.pem", "rb") as file:
#     file_data = file.read()
#     print(file_data)
#     encryptor = PKCS1_OAEP.new(pubKey)
#     encrypted = encryptor.encrypt(file_data)
#     print("Encrypted File:", binascii.hexlify(encrypted))


# print('RSA Decrypt')
# with open("Keys/privateKey.pem", "rb") as file:
#     file_data = file.read()
#     print(file_data)
#     decryptor = PKCS1_OAEP.new(keyPair)
#     decrypted = decryptor.decrypt(encrypted)
#     print('Decrypted:', decrypted)


def EncryptRSA():
    print('RSA Encrypt')
    with open("keys/publicKey.pem", "rb") as file:
        file_data = file.read()
        print(file_data)
        encryptor = PKCS1_OAEP.new(pubKey)
        encrypted = encryptor.encrypt(file_data)
        print("Encrypted File:", binascii.hexlify(encrypted))

    with open("Keys/encrypted.key", "wb") as file:
        file.write(encrypted)


def DecryptRSA():
    with open("keys/encrypted.key", "rb") as file:
        file_data = file.read()
    print(file_data)
    decryptor = PKCS1_OAEP.new(keyPair)
    decrypted = decryptor.decrypt(file_data)
    print('Decrypted:', decrypted)


EncryptRSA()
DecryptRSA()