def test_master_keys(self):
"""
Generate and test master keys.
"""
# test keys are correct sizes
self.assertEqual(32, len(MyTestCase.master_key_256))
self.assertEqual(64, len(MyTestCase.master_key_512))
self.assertGreater(len(MyTestCase.master_key_512),
len(MyTestCase.master_key_256))
print("\nMaster encryption keys:\n256:")
print(hexlify(MyTestCase.master_key_256).decode())
print("512:")
print(hexlify(MyTestCase.master_key_512).decode())
print()
print("Expected TypeError:")
self.assertRaises(TypeError, (pyCryptoFrost.generate_master_key(1)))
# Test keys for uniqueness.
for i in range(0, 3):
master_key_256 = pyCryptoFrost.generate_master_key('sha256')
master_key_512 = pyCryptoFrost.generate_master_key('sha512')
master_key_256_next = pyCryptoFrost.generate_master_key('sha256')
master_key_512_next = pyCryptoFrost.generate_master_key('sha512')
self.assertNotEqual(master_key_256, master_key_256_next)
self.assertNotEqual(master_key_512, master_key_512_next)
def test_encrypt_aes(self):
"""
Test the AES256 and AES512 encryption algorithms
for correctness, uniqueness and functionality.
Tests decryption as well, ensures the text can
make the round trip.
"""
text = b'**ENCRYPT ME AES**'
# Tests AES256.
master_key = pyCryptoFrost.generate_master_key('sha256')
key256 = pyCryptoFrost.generate_encryption_key('sha256', master_key)
pyCryptoFrost.encrypt_aes(text, key256, "aes128")
self.assertEqual(len(key256), 16)
# Tests AES512.
master_key = pyCryptoFrost.generate_master_key('sha512')
key512 = pyCryptoFrost.generate_encryption_key('sha512', master_key)
pyCryptoFrost.encrypt_aes(text, key512, "aes256")
self.assertEqual(len(key512), 32)
# Tests default encryption with dynamic assignments.
print(pyCryptoFrost.encrypt_aes(text))
pyCryptoFrost.decrypt()
pyCryptoFrost.decrypt("ff08c21b4c7045a64defca6136244fbf")
pyCryptoFrost.decrypt("cdf042cefb0cffeda749da3c2a4bc137RRR")
def test_generate_hmac_key(self):
"""
Generate and test derived HMAC keys
"""
# derived keys for sha256
key_length = 16
master_key = pyCryptoFrost.generate_master_key('sha256')
key256 = pyCryptoFrost.generate_hmac_key(key_length, master_key)
print("\nDerived HMAC key sha256:")
print(hexlify(key256).decode())
self.assertEqual(key_length, len(key256))
# derived keys for sha512
key_length = 32
master_key = pyCryptoFrost.generate_master_key('sha512')
key512 = pyCryptoFrost.generate_hmac_key(key_length, master_key)
print("\nDerived HMAC key sha512:")
print(hexlify(key512).decode())
self.assertEqual(key_length, len(key512))
# Test key differences
self.assertNotEqual(key256, key512)
self.assertGreater(len(key512), len(key256))
def test_generate_hmac(self):
"""
Ensures HMAC can be properly created and used.
"""
text = b'**HMAC TEST**'
master_key = pyCryptoFrost.generate_master_key('sha512')
hmac_key = pyCryptoFrost.generate_hmac_key(16, master_key)
print("hmac key:")
print(hmac_key)
print()
test = pyCryptoFrost.generate_hmac(hmac_key, text)
self.assertGreater(len(test), 0)
self.assertGreater(len(hmac_key), 0)
def setUp(self):
"""
Generate unencrypted text and master keys.
"""
MyTestCase.master_key_256 = pyCryptoFrost.generate_master_key('sha256')
MyTestCase.master_key_512 = pyCryptoFrost.generate_master_key('sha512')