Spaces:
Sleeping
Sleeping
| from Crypto.PublicKey import RSA | |
| from Crypto.Random import get_random_bytes | |
| from Crypto.Cipher import AES, PKCS1_OAEP | |
| from huggingface_hub import hf_hub_download | |
| import yaml | |
| import io | |
| loc = hf_hub_download(repo_id="Team8/dataset", repo_type="dataset", filename="receiver.pem") | |
| data = "I met aliens in UFO. Here is the map.".encode("utf-8") | |
| file_out = open("encrypted_data.bin", "wb") | |
| recipient_key = RSA.import_key(open(loc).read()) | |
| session_key = get_random_bytes(16) | |
| # Encrypt the session key with the public RSA key | |
| cipher_rsa = PKCS1_OAEP.new(recipient_key) | |
| enc_session_key = cipher_rsa.encrypt(session_key) | |
| # Encrypt the data with the AES session key | |
| cipher_aes = AES.new(session_key, AES.MODE_EAX) | |
| ciphertext, tag = cipher_aes.encrypt_and_digest(data) | |
| [ file_out.write(x) for x in (enc_session_key, cipher_aes.nonce, tag, ciphertext) ] | |
| file_out.close() | |