mirror of
https://github.com/saymrwulf/cryptography.git
synced 2026-05-14 20:37:55 +00:00
36 lines
997 B
Python
36 lines
997 B
Python
import binascii
|
|
|
|
import botan
|
|
|
|
from tests.utils import load_nist_vectors
|
|
|
|
BLOCK_SIZE = 64
|
|
|
|
|
|
def encrypt(mode, key, iv, plaintext):
|
|
encryptor = botan.Cipher(
|
|
f"IDEA/{mode}/NoPadding", "encrypt", binascii.unhexlify(key)
|
|
)
|
|
|
|
cipher_text = encryptor.cipher(
|
|
binascii.unhexlify(plaintext), binascii.unhexlify(iv)
|
|
)
|
|
return binascii.hexlify(cipher_text)
|
|
|
|
|
|
def verify_vectors(mode, filename):
|
|
with open(filename) as f:
|
|
vector_file = f.read().splitlines()
|
|
|
|
vectors = load_nist_vectors(vector_file)
|
|
for vector in vectors:
|
|
ct = encrypt(mode, vector["key"], vector["iv"], vector["plaintext"])
|
|
assert ct == vector["ciphertext"]
|
|
|
|
|
|
cbc_path = "tests/hazmat/primitives/vectors/ciphers/IDEA/idea-cbc.txt"
|
|
verify_vectors("CBC", cbc_path)
|
|
ofb_path = "tests/hazmat/primitives/vectors/ciphers/IDEA/idea-ofb.txt"
|
|
verify_vectors("OFB", ofb_path)
|
|
cfb_path = "tests/hazmat/primitives/vectors/ciphers/IDEA/idea-cfb.txt"
|
|
verify_vectors("CFB", cfb_path)
|