2013-06-24 20:41:05 +00:00
|
|
|
from binascii import hexlify
|
2013-07-08 21:38:27 +00:00
|
|
|
from attic.testsuite import AtticTestCase
|
|
|
|
|
from attic.crypto import AES, bytes_to_long, bytes_to_int, long_to_bytes, pbkdf2_sha256, get_random_bytes
|
2013-06-24 20:41:05 +00:00
|
|
|
|
|
|
|
|
|
2013-07-08 21:38:27 +00:00
|
|
|
class CryptoTestCase(AtticTestCase):
|
2013-06-24 20:41:05 +00:00
|
|
|
|
|
|
|
|
def test_bytes_to_int(self):
|
|
|
|
|
self.assert_equal(bytes_to_int(b'\0\0\0\1'), 1)
|
|
|
|
|
|
|
|
|
|
def test_bytes_to_long(self):
|
|
|
|
|
self.assert_equal(bytes_to_long(b'\0\0\0\0\0\0\0\1'), 1)
|
|
|
|
|
self.assert_equal(long_to_bytes(1), b'\0\0\0\0\0\0\0\1')
|
|
|
|
|
|
|
|
|
|
def test_pbkdf2_sha256(self):
|
|
|
|
|
self.assert_equal(hexlify(pbkdf2_sha256(b'password', b'salt', 1, 32)),
|
|
|
|
|
b'120fb6cffcf8b32c43e7225256c4f837a86548c92ccc35480805987cb70be17b')
|
|
|
|
|
self.assert_equal(hexlify(pbkdf2_sha256(b'password', b'salt', 2, 32)),
|
|
|
|
|
b'ae4d0c95af6b46d32d0adff928f06dd02a303f8ef3c251dfd6e2d85a95474c43')
|
|
|
|
|
self.assert_equal(hexlify(pbkdf2_sha256(b'password', b'salt', 4096, 32)),
|
|
|
|
|
b'c5e478d59288c841aa530db6845c4c8d962893a001ce4e11a4963873aa98134a')
|
|
|
|
|
|
|
|
|
|
def test_get_random_bytes(self):
|
|
|
|
|
bytes = get_random_bytes(10)
|
|
|
|
|
bytes2 = get_random_bytes(10)
|
|
|
|
|
self.assert_equal(len(bytes), 10)
|
|
|
|
|
self.assert_equal(len(bytes2), 10)
|
|
|
|
|
self.assert_not_equal(bytes, bytes2)
|
|
|
|
|
|
|
|
|
|
def test_aes(self):
|
|
|
|
|
key = b'X' * 32
|
|
|
|
|
data = b'foo' * 10
|
|
|
|
|
aes = AES(key)
|
|
|
|
|
self.assert_equal(bytes_to_long(aes.iv.raw, 8), 0)
|
|
|
|
|
cdata = aes.encrypt(data)
|
|
|
|
|
self.assert_equal(hexlify(cdata), b'c6efb702de12498f34a2c2bbc8149e759996d08bf6dc5c610aefc0c3a466')
|
|
|
|
|
self.assert_equal(bytes_to_long(aes.iv.raw, 8), 2)
|
|
|
|
|
self.assert_not_equal(data, aes.decrypt(cdata))
|
|
|
|
|
aes.reset(iv=b'\0' * 16)
|
|
|
|
|
self.assert_equal(data, aes.decrypt(cdata))
|
