diff options
author | Paul Kehrer <paul.l.kehrer@gmail.com> | 2013-09-27 13:43:06 -0500 |
---|---|---|
committer | Paul Kehrer <paul.l.kehrer@gmail.com> | 2013-10-06 17:31:31 -0500 |
commit | dff22d4707a50b8164c5c6acd5521bcd91160cd1 (patch) | |
tree | 761787ecb9a65b941fcfd4907c16ee244f6bf9b4 /tests/primitives/test_cryptrec.py | |
parent | 867b979b81aba0578d7241d6a38201214a976ace (diff) | |
download | cryptography-dff22d4707a50b8164c5c6acd5521bcd91160cd1.tar.gz cryptography-dff22d4707a50b8164c5c6acd5521bcd91160cd1.tar.bz2 cryptography-dff22d4707a50b8164c5c6acd5521bcd91160cd1.zip |
Camellia block cipher support
* Tests for CBC, OFB, CFB, and ECB
* Tests will be automatically skipped if camellia support is not present
in your OpenSSL library (e.g. OS X 10.8 with default OpenSSL)
* Test for unsupported cipher in create_block_cipher_context
* Docs for the cipher
Diffstat (limited to 'tests/primitives/test_cryptrec.py')
-rw-r--r-- | tests/primitives/test_cryptrec.py | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/tests/primitives/test_cryptrec.py b/tests/primitives/test_cryptrec.py new file mode 100644 index 00000000..54ae4d0c --- /dev/null +++ b/tests/primitives/test_cryptrec.py @@ -0,0 +1,64 @@ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or +# implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +""" +Test using the CRYPTREC (Camellia) Test Vectors +""" + +from __future__ import absolute_import, division, print_function + +import binascii +import itertools +import os + +import pytest + +from cryptography.bindings.openssl.api import api +from cryptography.primitives.block import BlockCipher, ciphers, modes + +from ..utils import load_cryptrec_vectors_from_file + +CAMELLIA_ECB_SUPPORTED = api.supports('camellia-128-ecb') + + +def parameterize_encrypt_test(cipher, vector_type, params, fnames): + return pytest.mark.parametrize(params, + list(itertools.chain.from_iterable( + load_cryptrec_vectors_from_file( + os.path.join(cipher, vector_type, fname), + ) + for fname in fnames + )) + ) + + +@pytest.mark.skipif("not CAMELLIA_ECB_SUPPORTED") +class TestCamelliaECB(object): + + @parameterize_encrypt_test( + "Camellia", "NTT", + ("key", "plaintext", "ciphertext"), + [ + "camellia-128-ecb.txt", + "camellia-192-ecb.txt", + "camellia-256-ecb.txt", + ] + ) + def test_NTT(self, key, plaintext, ciphertext): + cipher = BlockCipher( + ciphers.Camellia(binascii.unhexlify(key)), + modes.ECB(), + ) + actual_ciphertext = cipher.encrypt(binascii.unhexlify(plaintext)) + actual_ciphertext += cipher.finalize() + assert binascii.hexlify(actual_ciphertext).upper() == ciphertext |