mirror of
https://github.com/ComradCollective/Comrad
synced 2024-11-09 13:10:44 +00:00
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
|
import base64
|
||
|
|
import os
|
||
|
|
|
||
|
|
from . import syfr as crypto
|
||
|
|
|
||
|
|
|
||
|
|
def test_rsa_encrypt_and_decrypt():
|
||
|
|
priv = crypto.generate_rsa_key(complexity=512)
|
||
|
|
message = "Attack at Calais"
|
||
|
|
ciphertext = crypto.rsa_encrypt(message, priv.public_key())
|
||
|
|
assert crypto.rsa_decrypt(ciphertext, priv) == message
|
||
|
|
|
||
|
|
|
||
|
|
def test_aes_encrypt_decrypt():
|
||
|
|
priv = crypto.create_aes_key()
|
||
|
|
message = "Sell Watermelons"
|
||
|
|
ciphertext, iv = crypto.aes_encrypt(message, priv)
|
||
|
|
assert crypto.aes_decrypt(ciphertext, priv, iv) == message
|
||
|
|
|
||
|
|
|
||
|
|
def test_full_encrypt():
|
||
|
|
priv1 = crypto.generate_rsa_key()
|
||
|
|
priv2 = crypto.generate_rsa_key()
|
||
|
|
target_pubkey = crypto.serialize_pubkey(priv2.public_key())
|
||
|
|
message = "Santa is not real."
|
||
|
|
|
||
|
|
aes_ciphertext, encry_aes_key, hmac, hmac_signature, iv, metadata = \
|
||
|
|
crypto.encrypt(message, priv1, target_pubkey)
|
||
|
|
|
||
|
|
aes_key = crypto.rsa_decrypt(encry_aes_key, priv2)
|
||
|
|
|
||
|
|
assert crypto.aes_decrypt(aes_ciphertext, aes_key, iv) == message
|
||
|
|
|
||
|
|
assert message == \
|
||
|
|
crypto.decrypt(
|
||
|
|
aes_ciphertext, encry_aes_key, hmac, hmac_signature,
|
||
|
|
priv2, iv, metadata
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_sign():
|
||
|
|
priv = crypto.generate_rsa_key(complexity=512)
|
||
|
|
message = "Secret wish list"
|
||
|
|
sig = crypto.sign(message, priv)
|
||
|
|
assert crypto.verify_signature(sig, message, priv.public_key())
|
||
|
|
|
||
|
|
|
||
|
|
def test_long_pad():
|
||
|
|
complexity = 10**3 # won't create exactly this length
|
||
|
|
contents = base64.b64encode(os.urandom(complexity))
|
||
|
|
padded = crypto.long_pad(contents, 3*complexity)
|
||
|
|
assert crypto.long_unpad(padded) == contents
|