dearsky/pico-hsm
1
0
Fork 0
Code Issues 17 Pull Requests 4 Actions Packages Projects Releases 26 Wiki Activity

Added support for AES CTR.

Browse Source 
Note: the OID used by CTR does not exist.
Signed-off-by: Pol Henarejos <pol.henarejos@cttc.es>
This commit is contained in:
Pol Henarejos
2023-03-23 18:47:32 +01:00
parent ad3304a384
commit f5e875a6b7
3 changed files with 58 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
tests/pico-hsm/test_052_aes_ext.py
View File

@@ -259,3 +259,43 @@ def test_aes_xts_iv(device, size):
assert(dtA == dtB)
assert(dtA == MESSAGE)
device.delete_key(keyid)
@pytest.mark.parametrize(
"size", [128, 192, 256]
)
def test_aes_ctr_no_iv(device, size):
pkey, keyid = generate_key(device, size)
ctA = device.aes(keyid, EncryptionMode.ENCRYPT, AES.CTR, MESSAGE)
iv = b'\x00' * 16
cipher = Cipher(algorithms.AES(pkey), modes.CTR(iv))
encryptor = cipher.encryptor()
ctB = encryptor.update(MESSAGE) + encryptor.finalize()
assert(ctA == ctB)
dtA = device.aes(keyid, EncryptionMode.DECRYPT, AES.CTR, ctA)
decryptor = cipher.decryptor()
dtB = decryptor.update(ctB) + decryptor.finalize()
assert(dtA == dtB)
assert(dtA == MESSAGE)
device.delete_key(keyid)
@pytest.mark.parametrize(
"size", [128, 192, 256]
)
def test_aes_ctr_iv(device, size):
pkey, keyid = generate_key(device, size)
iv = os.urandom(16)
ctA = device.aes(keyid, EncryptionMode.ENCRYPT, AES.CTR, MESSAGE, iv=iv)
cipher = Cipher(algorithms.AES(pkey), modes.CTR(iv))
encryptor = cipher.encryptor()
ctB = encryptor.update(MESSAGE) + encryptor.finalize()
assert(ctA == ctB)
dtA = device.aes(keyid, EncryptionMode.DECRYPT, AES.CTR, ctA, iv=iv)
decryptor = cipher.decryptor()
dtB = decryptor.update(ctB) + decryptor.finalize()
assert(dtA == dtB)
assert(dtA == MESSAGE)
device.delete_key(keyid)