add more test vectors for aes

adafruit · Apr 5, 2023 · 7b9f97d · 7b9f97d
1 parent e8711ee
commit 7b9f97d
Showing 1 changed file with 44 additions and 0 deletions.
diff --git a/tests/circuitpython/aes.py b/tests/circuitpython/aes.py
@@ -97,3 +97,47 @@
     cipher.decrypt_into(cyphertext[i : i + 16], output)
     print(str(hexlify(output), ""))
 print()
+
+print("truncated-CTR-1")
+## Truncated CTR test case
+plaintext = unhexlify("6bc1bee22e409f96e93d7e117393172a" "ae2d")
+
+key = unhexlify("2b7e151628aed2a6abf7158809cf4f3c")
+counter = unhexlify("f0f1f2f3f4f5f6f7f8f9fafbfcfdfeff")
+cyphertext = bytearray(len(plaintext))
+cipher = aesio.AES(key, aesio.MODE_CTR, IV=counter)
+for i in range(0, len(plaintext), 16):
+    output = memoryview(cyphertext)[i : i + 16]
+    cipher.encrypt_into(plaintext[i : i + 16], output)
+    print(str(hexlify(output), ""))
+print()
+
+plaintext = bytearray(len(plaintext))
+cipher = aesio.AES(key, aesio.MODE_CTR, IV=counter)
+for i in range(0, len(plaintext), 16):
+    output = memoryview(plaintext)[i : i + 16]
+    cipher.decrypt_into(cyphertext[i : i + 16], output)
+    print(str(hexlify(output), ""))
+print()
+
+print("truncated-CTR-2")
+## Truncated CTR test case #2
+plaintext = unhexlify("6bc1bee22e409f96e93d7e117393172a" "ae")
+
+key = unhexlify("2b7e151628aed2a6abf7158809cf4f3c")
+counter = unhexlify("f0f1f2f3f4f5f6f7f8f9fafbfcfdfeff")
+cyphertext = bytearray(len(plaintext))
+cipher = aesio.AES(key, aesio.MODE_CTR, IV=counter)
+cipher.encrypt_into(plaintext, cyphertext)
+for i in range(0, len(plaintext), 16):
+    output = memoryview(cyphertext)[i : i + 16]
+    print(str(hexlify(output), ""))
+print()
+
+plaintext = bytearray(len(plaintext))
+cipher = aesio.AES(key, aesio.MODE_CTR, IV=counter)
+cipher.decrypt_into(cyphertext, plaintext)
+for i in range(0, len(plaintext), 16):
+    output = memoryview(plaintext)[i : i + 16]
+    print(str(hexlify(output), ""))
+print()