TiagoRG
/
uaveiro-leci
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[LABI] tema01: added publicKey_RSA.py

This commit is contained in:
brunombpereira 2023-05-16 18:31:21 +01:00
parent 1367c7c474
commit d2e535ef37
Signed by untrusted user who does not match committer: TiagoRG
GPG Key ID: DFCD48E3F420DB42
1 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
1ano/2semestre/labi/tema01/src/publicKey_RSA.py Normal file
View File

@ -0,0 +1,23 @@
from Crypto.PublicKey import RSA
def generate_key_pair(file_name, encryption_key, key_size):
# Generate the RSA key pair
key = RSA.generate(key_size)
# Write the private key to a file
private_key = key.exportKey("PEM", encryption_key )
with open(file_name + '_private.pem', 'wb') as f:
f.write(private_key)
# Write the public key to a file
public_key = key.publickey().exportKey()
with open(file_name + '_public.pem', 'wb') as f:
f.write(public_key)
if __name__ == '__main__':
file_name = input('Enter file name to save the key pair: ')
encryption_key = input('Enter encryption key for the private key (press enter for no encryption): ')
key_size = int(input('Enter the key size (in bits): '))
generate_key_pair(file_name, encryption_key, key_size)