| # This file is dual licensed under the terms of the Apache License, Version |
| # 2.0, and the BSD License. See the LICENSE file in the root of this repository |
| # for complete details. |
| |
| from __future__ import absolute_import, division, print_function |
| |
| import base64 |
| import itertools |
| import os |
| import textwrap |
| |
| import pytest |
| |
| import six |
| |
| from cryptography.exceptions import UnsupportedAlgorithm |
| from cryptography.hazmat.backends.interfaces import ( |
| DERSerializationBackend, |
| DSABackend, |
| EllipticCurveBackend, |
| PEMSerializationBackend, |
| RSABackend, |
| ) |
| from cryptography.hazmat.primitives.asymmetric import ( |
| dsa, |
| ec, |
| ed25519, |
| ed448, |
| rsa, |
| x25519, |
| x448, |
| ) |
| from cryptography.hazmat.primitives.serialization import ( |
| BestAvailableEncryption, |
| Encoding, |
| NoEncryption, |
| PrivateFormat, |
| PublicFormat, |
| load_der_parameters, |
| load_der_private_key, |
| load_der_public_key, |
| load_pem_parameters, |
| load_pem_private_key, |
| load_pem_public_key, |
| load_ssh_private_key, |
| load_ssh_public_key, |
| ssh, |
| ) |
| |
| |
| from .test_ec import _skip_curve_unsupported |
| from .utils import ( |
| _check_dsa_private_numbers, |
| _check_rsa_private_numbers, |
| load_vectors_from_file, |
| ) |
| from ...doubles import DummyKeySerializationEncryption |
| |
| |
| def _skip_fips_format(key_path, password, backend): |
| if backend._fips_enabled: |
| if key_path[0] == "Traditional_OpenSSL_Serialization": |
| pytest.skip("Traditional OpenSSL format blocked in FIPS mode") |
| if key_path[0] == "PEM_Serialization" and password is not None: |
| pytest.skip("Encrypted PEM_Serialization blocked in FIPS mode") |
| |
| |
| class TestBufferProtocolSerialization(object): |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| @pytest.mark.parametrize( |
| ("key_path", "password"), |
| [ |
| (["DER_Serialization", "enc-rsa-pkcs8.der"], bytearray(b"foobar")), |
| (["DER_Serialization", "enc2-rsa-pkcs8.der"], bytearray(b"baz")), |
| (["DER_Serialization", "unenc-rsa-pkcs8.der"], None), |
| (["DER_Serialization", "testrsa.der"], None), |
| ], |
| ) |
| def test_load_der_rsa_private_key(self, key_path, password, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda derfile: derfile.read(), |
| mode="rb", |
| ) |
| key = load_der_private_key(bytearray(data), password, backend) |
| assert key |
| assert isinstance(key, rsa.RSAPrivateKey) |
| _check_rsa_private_numbers(key.private_numbers()) |
| |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| @pytest.mark.parametrize( |
| ("key_path", "password"), |
| [ |
| ( |
| ["PEM_Serialization", "rsa_private_key.pem"], |
| bytearray(b"123456"), |
| ), |
| (["PKCS8", "unenc-rsa-pkcs8.pem"], None), |
| (["PKCS8", "enc-rsa-pkcs8.pem"], bytearray(b"foobar")), |
| (["PKCS8", "enc2-rsa-pkcs8.pem"], bytearray(b"baz")), |
| ( |
| ["Traditional_OpenSSL_Serialization", "key1.pem"], |
| bytearray(b"123456"), |
| ), |
| ], |
| ) |
| def test_load_pem_rsa_private_key(self, key_path, password, backend): |
| _skip_fips_format(key_path, password, backend) |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| key = load_pem_private_key(bytearray(data), password, backend) |
| assert key |
| assert isinstance(key, rsa.RSAPrivateKey) |
| _check_rsa_private_numbers(key.private_numbers()) |
| |
| |
| @pytest.mark.requires_backend_interface(interface=DERSerializationBackend) |
| class TestDERSerialization(object): |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| @pytest.mark.parametrize( |
| ("key_path", "password"), |
| [ |
| (["DER_Serialization", "enc-rsa-pkcs8.der"], b"foobar"), |
| (["DER_Serialization", "enc2-rsa-pkcs8.der"], b"baz"), |
| (["DER_Serialization", "unenc-rsa-pkcs8.der"], None), |
| (["DER_Serialization", "testrsa.der"], None), |
| ], |
| ) |
| def test_load_der_rsa_private_key(self, key_path, password, backend): |
| key = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda derfile: load_der_private_key( |
| derfile.read(), password, backend |
| ), |
| mode="rb", |
| ) |
| assert key |
| assert isinstance(key, rsa.RSAPrivateKey) |
| _check_rsa_private_numbers(key.private_numbers()) |
| |
| @pytest.mark.requires_backend_interface(interface=DSABackend) |
| @pytest.mark.parametrize( |
| ("key_path", "password"), |
| [ |
| (["DER_Serialization", "unenc-dsa-pkcs8.der"], None), |
| (["DER_Serialization", "dsa.1024.der"], None), |
| (["DER_Serialization", "dsa.2048.der"], None), |
| (["DER_Serialization", "dsa.3072.der"], None), |
| ], |
| ) |
| def test_load_der_dsa_private_key(self, key_path, password, backend): |
| key = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda derfile: load_der_private_key( |
| derfile.read(), password, backend |
| ), |
| mode="rb", |
| ) |
| assert key |
| assert isinstance(key, dsa.DSAPrivateKey) |
| _check_dsa_private_numbers(key.private_numbers()) |
| |
| @pytest.mark.parametrize( |
| "key_path", [["DER_Serialization", "enc-rsa-pkcs8.der"]] |
| ) |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| def test_password_not_bytes(self, key_path, backend): |
| key_file = os.path.join("asymmetric", *key_path) |
| password = u"this password is not bytes" |
| |
| with pytest.raises(TypeError): |
| load_vectors_from_file( |
| key_file, |
| lambda derfile: load_der_private_key( |
| derfile.read(), password, backend |
| ), |
| mode="rb", |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "password"), |
| [ |
| (["DER_Serialization", "ec_private_key.der"], None), |
| (["DER_Serialization", "ec_private_key_encrypted.der"], b"123456"), |
| ], |
| ) |
| @pytest.mark.requires_backend_interface(interface=EllipticCurveBackend) |
| def test_load_der_ec_private_key(self, key_path, password, backend): |
| _skip_curve_unsupported(backend, ec.SECP256R1()) |
| key = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda derfile: load_der_private_key( |
| derfile.read(), password, backend |
| ), |
| mode="rb", |
| ) |
| |
| assert key |
| assert isinstance(key, ec.EllipticCurvePrivateKey) |
| assert key.curve.name == "secp256r1" |
| assert key.curve.key_size == 256 |
| |
| @pytest.mark.parametrize( |
| "key_path", [["DER_Serialization", "enc-rsa-pkcs8.der"]] |
| ) |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| def test_wrong_password(self, key_path, backend): |
| key_file = os.path.join("asymmetric", *key_path) |
| password = b"this password is wrong" |
| |
| with pytest.raises(ValueError): |
| load_vectors_from_file( |
| key_file, |
| lambda derfile: load_der_private_key( |
| derfile.read(), password, backend |
| ), |
| mode="rb", |
| ) |
| |
| @pytest.mark.parametrize( |
| "key_path", [["DER_Serialization", "unenc-rsa-pkcs8.der"]] |
| ) |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| def test_unused_password(self, key_path, backend): |
| key_file = os.path.join("asymmetric", *key_path) |
| password = b"this password will not be used" |
| |
| with pytest.raises(TypeError): |
| load_vectors_from_file( |
| key_file, |
| lambda derfile: load_der_private_key( |
| derfile.read(), password, backend |
| ), |
| mode="rb", |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "password"), |
| itertools.product( |
| [["DER_Serialization", "enc-rsa-pkcs8.der"]], [b"", None] |
| ), |
| ) |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| def test_missing_password(self, key_path, password, backend): |
| key_file = os.path.join("asymmetric", *key_path) |
| |
| with pytest.raises(TypeError): |
| load_vectors_from_file( |
| key_file, |
| lambda derfile: load_der_private_key( |
| derfile.read(), password, backend |
| ), |
| mode="rb", |
| ) |
| |
| def test_wrong_format(self, backend): |
| key_data = b"---- NOT A KEY ----\n" |
| |
| with pytest.raises(ValueError): |
| load_der_private_key(key_data, None, backend) |
| |
| with pytest.raises(ValueError): |
| load_der_private_key( |
| key_data, b"this password will not be used", backend |
| ) |
| |
| def test_corrupt_der_pkcs8(self, backend): |
| # unenc-rsa-pkcs8 with a bunch of data missing. |
| key_data = textwrap.dedent( |
| """\ |
| MIICdQIBADALBgkqhkiG9w0BAQEEggJhMIICXQIBAAKBgQC7JHoJfg6yNzLMOWet |
| 8Z49a4KD0dCspMAYvo2YAMB7/wdEycocujbhJ2n/seONi+5XqTqqFkM5VBl8rmkk |
| FPZk/7x0xmdsTPECSWnHK+HhoaNDFPR3j8jQhVo1laxiqcEhAHegi5cwtFosuJAv |
| FiRC0Cgz+frQPFQEBsAV9RuasyQxqzxrR0Ow0qncBeGBWbYE6WZhqtcLAI895b+i |
| +F4lbB4iD7T9QeIDMU/aIMXA81UO4cns1z4qDAHKeyLLrPQrJ/B4X7XC+egUWm5+ |
| hr1qmyAMusyXIBECQQDJWZ8piluf4yrYfsJAn6hF5T4RjTztbqvO0GVG2McHY7Uj |
| NPSffhzHx/ll0fQEQji+OgydCCX8o3HZrgw5YfSJAkEA7e+rqdU5nO5ZG//PSEQb |
| tjLnRiTzBH/elQhtdZ5nF7pcpNTi4k13zutmKcWW4GK75azcRGJUhu1kDM7QYAOd |
| SQJAVNkYcifkvna7GmooL5VYEsQsqLbM4v0NF2TIGNfG3z1MGp75KrC5LhL97MNR |
| we2p/bd2k0HYyCKUGnf2nMPDiQJBAI75pwittSoE240EobUGIDTSz8CJsXIxuDmL |
| z+KOpdpPRR5TQmbEMEspjsFpFymMiuYPgmihQbO2cJl1qScY5OkCQQCJ6m5tcN8l |
| Xxg/SNpjEIv+qAyUD96XVlOJlOIeLHQ8kYE0C6ZA+MsqYIzgAreJk88Yn0lU/X0/ |
| mu/UpE/BRZmR |
| """ |
| ).encode() |
| bad_der = base64.b64decode(b"".join(key_data.splitlines())) |
| |
| with pytest.raises(ValueError): |
| load_der_private_key(bad_der, None, backend) |
| |
| with pytest.raises(ValueError): |
| load_der_private_key( |
| bad_der, b"this password will not be used", backend |
| ) |
| |
| def test_corrupt_traditional_format_der(self, backend): |
| # privkey with a bunch of data missing. |
| key_data = textwrap.dedent( |
| """\ |
| MIIBPAIBAAJBAKrbeqkuRk8VcRmWFmtP+LviMB3+6dizWW3DwaffznyHGAFwUJ/I |
| Tv0XtbsCyl3QoyKGhrOAy3RvPK5M38iuXT0CAwEAAQJAZ3cnzaHXM/bxGaR5CR1R |
| rD1qFBAVfoQFiOH9uPJgMaoAuoQEisPHVcZDKcOv4wEg6/TInAIXBnEigtqvRzuy |
| mvcpHZwQJdmdHHkGKAs37Dfxi67HbkUCIQCeZGliHXFa071Fp06ZeWlR2ADonTZz |
| rJBhdTe0v5pCeQIhAIZfkiGgGBX4cIuuckzEm43g9WMUjxP/0GlK39vIyihxAiEA |
| mymehFRT0MvqW5xAKAx7Pgkt8HVKwVhc2LwGKHE0DZM= |
| """ |
| ).encode() |
| bad_der = base64.b64decode(b"".join(key_data.splitlines())) |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key(bad_der, None, backend) |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key( |
| bad_der, b"this password will not be used", backend |
| ) |
| |
| @pytest.mark.parametrize( |
| "key_file", |
| [ |
| os.path.join( |
| "asymmetric", "DER_Serialization", "unenc-rsa-pkcs8.pub.der" |
| ), |
| os.path.join( |
| "asymmetric", "DER_Serialization", "rsa_public_key.der" |
| ), |
| os.path.join("asymmetric", "public", "PKCS1", "rsa.pub.der"), |
| ], |
| ) |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| def test_load_der_rsa_public_key(self, key_file, backend): |
| key = load_vectors_from_file( |
| key_file, |
| lambda derfile: load_der_public_key(derfile.read(), backend), |
| mode="rb", |
| ) |
| assert key |
| assert isinstance(key, rsa.RSAPublicKey) |
| numbers = key.public_numbers() |
| assert numbers.e == 65537 |
| |
| def test_load_der_invalid_public_key(self, backend): |
| with pytest.raises(ValueError): |
| load_der_public_key(b"invalid data", backend) |
| |
| @pytest.mark.parametrize( |
| "key_file", |
| [ |
| os.path.join( |
| "asymmetric", "DER_Serialization", "unenc-dsa-pkcs8.pub.der" |
| ), |
| os.path.join( |
| "asymmetric", "DER_Serialization", "dsa_public_key.der" |
| ), |
| ], |
| ) |
| @pytest.mark.requires_backend_interface(interface=DSABackend) |
| def test_load_der_dsa_public_key(self, key_file, backend): |
| key = load_vectors_from_file( |
| key_file, |
| lambda derfile: load_der_public_key(derfile.read(), backend), |
| mode="rb", |
| ) |
| assert key |
| assert isinstance(key, dsa.DSAPublicKey) |
| |
| @pytest.mark.requires_backend_interface(interface=EllipticCurveBackend) |
| def test_load_ec_public_key(self, backend): |
| _skip_curve_unsupported(backend, ec.SECP256R1()) |
| key = load_vectors_from_file( |
| os.path.join( |
| "asymmetric", "DER_Serialization", "ec_public_key.der" |
| ), |
| lambda derfile: load_der_public_key(derfile.read(), backend), |
| mode="rb", |
| ) |
| assert key |
| assert isinstance(key, ec.EllipticCurvePublicKey) |
| assert key.curve.name == "secp256r1" |
| assert key.curve.key_size == 256 |
| |
| def test_wrong_parameters_format(self, backend): |
| param_data = b"---- NOT A KEY ----\n" |
| |
| with pytest.raises(ValueError): |
| load_der_parameters(param_data, backend) |
| |
| |
| @pytest.mark.requires_backend_interface(interface=PEMSerializationBackend) |
| class TestPEMSerialization(object): |
| @pytest.mark.parametrize( |
| ("key_file", "password"), |
| [ |
| (["PEM_Serialization", "rsa_private_key.pem"], b"123456"), |
| (["PKCS8", "unenc-rsa-pkcs8.pem"], None), |
| (["PKCS8", "enc-rsa-pkcs8.pem"], b"foobar"), |
| (["PKCS8", "enc2-rsa-pkcs8.pem"], b"baz"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9607.pem"], b"123456"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9671.pem"], b"123456"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9925.pem"], b"123456"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9926.pem"], b"123456"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9927.pem"], b"123456"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9928.pem"], b"123456"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9929.pem"], b"123456"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9930.pem"], b"123456"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9931.pem"], b"123456"), |
| (["PKCS8", "pkcs12_s2k_pem-X_9932.pem"], b"123456"), |
| (["Traditional_OpenSSL_Serialization", "key1.pem"], b"123456"), |
| (["Traditional_OpenSSL_Serialization", "key2.pem"], b"a123456"), |
| (["Traditional_OpenSSL_Serialization", "testrsa.pem"], None), |
| ( |
| ["Traditional_OpenSSL_Serialization", "testrsa-encrypted.pem"], |
| b"password", |
| ), |
| ], |
| ) |
| def test_load_pem_rsa_private_key(self, key_file, password, backend): |
| _skip_fips_format(key_file, password, backend) |
| key = load_vectors_from_file( |
| os.path.join("asymmetric", *key_file), |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), password, backend |
| ), |
| ) |
| |
| assert key |
| assert isinstance(key, rsa.RSAPrivateKey) |
| _check_rsa_private_numbers(key.private_numbers()) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "password"), |
| [ |
| (["Traditional_OpenSSL_Serialization", "dsa.1024.pem"], None), |
| (["Traditional_OpenSSL_Serialization", "dsa.2048.pem"], None), |
| (["Traditional_OpenSSL_Serialization", "dsa.3072.pem"], None), |
| (["PKCS8", "unenc-dsa-pkcs8.pem"], None), |
| (["PEM_Serialization", "dsa_private_key.pem"], b"123456"), |
| ], |
| ) |
| def test_load_dsa_private_key(self, key_path, password, backend): |
| _skip_fips_format(key_path, password, backend) |
| key = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), password, backend |
| ), |
| ) |
| assert key |
| assert isinstance(key, dsa.DSAPrivateKey) |
| _check_dsa_private_numbers(key.private_numbers()) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "password"), |
| [ |
| (["PKCS8", "ec_private_key.pem"], None), |
| (["PKCS8", "ec_private_key_encrypted.pem"], b"123456"), |
| (["PEM_Serialization", "ec_private_key.pem"], None), |
| (["PEM_Serialization", "ec_private_key_encrypted.pem"], b"123456"), |
| ], |
| ) |
| @pytest.mark.requires_backend_interface(interface=EllipticCurveBackend) |
| def test_load_pem_ec_private_key(self, key_path, password, backend): |
| _skip_curve_unsupported(backend, ec.SECP256R1()) |
| _skip_fips_format(key_path, password, backend) |
| key = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), password, backend |
| ), |
| ) |
| |
| assert key |
| assert isinstance(key, ec.EllipticCurvePrivateKey) |
| assert key.curve.name == "secp256r1" |
| assert key.curve.key_size == 256 |
| |
| @pytest.mark.parametrize( |
| ("key_file"), |
| [ |
| os.path.join("asymmetric", "PKCS8", "unenc-rsa-pkcs8.pub.pem"), |
| os.path.join( |
| "asymmetric", "PEM_Serialization", "rsa_public_key.pem" |
| ), |
| os.path.join("asymmetric", "public", "PKCS1", "rsa.pub.pem"), |
| ], |
| ) |
| def test_load_pem_rsa_public_key(self, key_file, backend): |
| key = load_vectors_from_file( |
| key_file, |
| lambda pemfile: load_pem_public_key( |
| pemfile.read().encode(), backend |
| ), |
| ) |
| assert key |
| assert isinstance(key, rsa.RSAPublicKey) |
| numbers = key.public_numbers() |
| assert numbers.e == 65537 |
| |
| @pytest.mark.parametrize( |
| ("key_file"), |
| [ |
| os.path.join("asymmetric", "PKCS8", "unenc-dsa-pkcs8.pub.pem"), |
| os.path.join( |
| "asymmetric", "PEM_Serialization", "dsa_public_key.pem" |
| ), |
| ], |
| ) |
| def test_load_pem_dsa_public_key(self, key_file, backend): |
| key = load_vectors_from_file( |
| key_file, |
| lambda pemfile: load_pem_public_key( |
| pemfile.read().encode(), backend |
| ), |
| ) |
| assert key |
| assert isinstance(key, dsa.DSAPublicKey) |
| |
| @pytest.mark.requires_backend_interface(interface=EllipticCurveBackend) |
| def test_load_ec_public_key(self, backend): |
| _skip_curve_unsupported(backend, ec.SECP256R1()) |
| key = load_vectors_from_file( |
| os.path.join( |
| "asymmetric", "PEM_Serialization", "ec_public_key.pem" |
| ), |
| lambda pemfile: load_pem_public_key( |
| pemfile.read().encode(), backend |
| ), |
| ) |
| assert key |
| assert isinstance(key, ec.EllipticCurvePublicKey) |
| assert key.curve.name == "secp256r1" |
| assert key.curve.key_size == 256 |
| |
| @pytest.mark.skip_fips( |
| reason="Traditional OpenSSL format blocked in FIPS mode" |
| ) |
| def test_rsa_traditional_encrypted_values(self, backend): |
| pkey = load_vectors_from_file( |
| os.path.join( |
| "asymmetric", "Traditional_OpenSSL_Serialization", "key1.pem" |
| ), |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), b"123456", backend |
| ), |
| ) |
| assert pkey |
| |
| numbers = pkey.private_numbers() |
| assert numbers.p == int( |
| "fb7d316fc51531b36d93adaefaf52db6ad5beb793d37c4cf9dfc1ddd17cfbafb", |
| 16, |
| ) |
| assert numbers.q == int( |
| "df98264e646de9a0fbeab094e31caad5bc7adceaaae3c800ca0275dd4bb307f5", |
| 16, |
| ) |
| assert numbers.d == int( |
| "db4848c36f478dd5d38f35ae519643b6b810d404bcb76c00e44015e56ca1cab0" |
| "7bb7ae91f6b4b43fcfc82a47d7ed55b8c575152116994c2ce5325ec24313b911", |
| 16, |
| ) |
| assert numbers.dmp1 == int( |
| "ce997f967192c2bcc3853186f1559fd355c190c58ddc15cbf5de9b6df954c727", |
| 16, |
| ) |
| assert numbers.dmq1 == int( |
| "b018a57ab20ffaa3862435445d863369b852cf70a67c55058213e3fe10e3848d", |
| 16, |
| ) |
| assert numbers.iqmp == int( |
| "6a8d830616924f5cf2d1bc1973f97fde6b63e052222ac7be06aa2532d10bac76", |
| 16, |
| ) |
| assert numbers.public_numbers.e == 65537 |
| assert numbers.public_numbers.n == int( |
| "dba786074f2f0350ce1d99f5aed5b520cfe0deb5429ec8f2a88563763f566e77" |
| "9814b7c310e5326edae31198eed439b845dd2db99eaa60f5c16a43f4be6bcf37", |
| 16, |
| ) |
| |
| @pytest.mark.parametrize( |
| "key_path", |
| [ |
| ["Traditional_OpenSSL_Serialization", "testrsa.pem"], |
| ["PKCS8", "unenc-rsa-pkcs8.pem"], |
| ], |
| ) |
| def test_unused_password(self, key_path, backend): |
| key_file = os.path.join("asymmetric", *key_path) |
| password = b"this password will not be used" |
| |
| with pytest.raises(TypeError): |
| load_vectors_from_file( |
| key_file, |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), password, backend |
| ), |
| ) |
| |
| def test_invalid_encoding_with_traditional(self, backend): |
| key_file = os.path.join( |
| "asymmetric", "Traditional_OpenSSL_Serialization", "testrsa.pem" |
| ) |
| key = load_vectors_from_file( |
| key_file, |
| lambda pemfile: load_pem_private_key( |
| pemfile.read(), None, backend |
| ), |
| mode="rb", |
| ) |
| |
| for enc in (Encoding.OpenSSH, Encoding.Raw, Encoding.X962): |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| enc, PrivateFormat.TraditionalOpenSSL, NoEncryption() |
| ) |
| |
| @pytest.mark.parametrize( |
| "key_path", |
| [ |
| ["Traditional_OpenSSL_Serialization", "testrsa-encrypted.pem"], |
| ["PKCS8", "enc-rsa-pkcs8.pem"], |
| ], |
| ) |
| def test_password_not_bytes(self, key_path, backend): |
| key_file = os.path.join("asymmetric", *key_path) |
| password = u"this password is not bytes" |
| |
| with pytest.raises(TypeError): |
| load_vectors_from_file( |
| key_file, |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), password, backend |
| ), |
| ) |
| |
| @pytest.mark.parametrize( |
| "key_path", |
| [ |
| ["Traditional_OpenSSL_Serialization", "testrsa-encrypted.pem"], |
| ["PKCS8", "enc-rsa-pkcs8.pem"], |
| ], |
| ) |
| def test_wrong_password(self, key_path, backend): |
| key_file = os.path.join("asymmetric", *key_path) |
| password = b"this password is wrong" |
| |
| with pytest.raises(ValueError): |
| load_vectors_from_file( |
| key_file, |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), password, backend |
| ), |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "password"), |
| itertools.product( |
| [ |
| ["Traditional_OpenSSL_Serialization", "testrsa-encrypted.pem"], |
| ["PKCS8", "enc-rsa-pkcs8.pem"], |
| ], |
| [b"", None], |
| ), |
| ) |
| def test_missing_password(self, key_path, password, backend): |
| key_file = os.path.join("asymmetric", *key_path) |
| |
| with pytest.raises(TypeError): |
| load_vectors_from_file( |
| key_file, |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), password, backend |
| ), |
| ) |
| |
| def test_wrong_private_format(self, backend): |
| key_data = b"---- NOT A KEY ----\n" |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key(key_data, None, backend) |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key( |
| key_data, b"this password will not be used", backend |
| ) |
| |
| def test_wrong_public_format(self, backend): |
| key_data = b"---- NOT A KEY ----\n" |
| |
| with pytest.raises(ValueError): |
| load_pem_public_key(key_data, backend) |
| |
| def test_wrong_parameters_format(self, backend): |
| param_data = b"---- NOT A KEY ----\n" |
| |
| with pytest.raises(ValueError): |
| load_pem_parameters(param_data, backend) |
| |
| def test_corrupt_traditional_format(self, backend): |
| # privkey.pem with a bunch of data missing. |
| key_data = textwrap.dedent( |
| """\ |
| -----BEGIN RSA PRIVATE KEY----- |
| MIIBPAIBAAJBAKrbeqkuRk8VcRmWFmtP+LviMB3+6dizWW3DwaffznyHGAFwUJ/I |
| Tv0XtbsCyl3QoyKGhrOAy3RvPK5M38iuXT0CAwEAAQJAZ3cnzaHXM/bxGaR5CR1R |
| rD1qFBAVfoQFiOH9uPJgMaoAuoQEisPHVcZDKcOv4wEg6/TInAIXBnEigtqvRzuy |
| mvcpHZwQJdmdHHkGKAs37Dfxi67HbkUCIQCeZGliHXFa071Fp06ZeWlR2ADonTZz |
| rJBhdTe0v5pCeQIhAIZfkiGgGBX4cIuuckzEm43g9WMUjxP/0GlK39vIyihxAiEA |
| mymehFRT0MvqW5xAKAx7Pgkt8HVKwVhc2LwGKHE0DZM= |
| -----END RSA PRIVATE KEY----- |
| """ |
| ).encode() |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key(key_data, None, backend) |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key( |
| key_data, b"this password will not be used", backend |
| ) |
| |
| def test_traditional_encrypted_corrupt_format(self, backend): |
| # privkey.pem with a single bit flipped |
| key_data = textwrap.dedent( |
| """\ |
| -----BEGIN RSA PRIVATE KEY----- |
| Proc-Type: <,ENCRYPTED |
| DEK-Info: AES-128-CBC,5E22A2BD85A653FB7A3ED20DE84F54CD |
| |
| hAqtb5ZkTMGcs4BBDQ1SKZzdQThWRDzEDxM3qBfjvYa35KxZ54aic013mW/lwj2I |
| v5bbpOjrHYHNAiZYZ7RNb+ztbF6F/g5PA5g7mFwEq+LFBY0InIplYBSv9QtE+lot |
| Dy4AlZa/+NzJwgdKDb+JVfk5SddyD4ywnyeORnMPy4xXKvjXwmW+iLibZVKsjIgw |
| H8hSxcD+FhWyJm9h9uLtmpuqhQo0jTUYpnTezZx2xeVPB53Ev7YCxR9Nsgj5GsVf |
| 9Z/hqLB7IFgM3pa0z3PQeUIZF/cEf72fISWIOBwwkzVrPUkXWfbuWeJXQXSs3amE |
| 5A295jD9BQp9CY0nNFSsy+qiXWToq2xT3y5zVNEStmN0SCGNaIlUnJzL9IHW+oMI |
| kPmXZMnAYBWeeCF1gf3J3aE5lZInegHNfEI0+J0LazC2aNU5Dg/BNqrmRqKWEIo/ |
| -----END RSA PRIVATE KEY----- |
| """ |
| ).encode() |
| |
| password = b"this password is wrong" |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key(key_data, None, backend) |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key(key_data, password, backend) |
| |
| def test_unsupported_key_encryption(self, backend): |
| key_data = textwrap.dedent( |
| """\ |
| -----BEGIN RSA PRIVATE KEY----- |
| Proc-Type: 4,ENCRYPTED |
| DEK-Info: FAKE-123,5E22A2BD85A653FB7A3ED20DE84F54CD |
| |
| hAqtb5ZkTMGcs4BBDQ1SKZzdQThWRDzEDxM3qBfjvYa35KxZ54aic013mW/lwj2I |
| v5bbpOjrHYHNAiZYZ7RNb+ztbF6F/g5PA5g7mFwEq+LFBY0InIplYBSv9QtE+lot |
| Dy4AlZa/+NzJwgdKDb+JVfk5SddyD4ywnyeORnMPy4xXKvjXwmW+iLibZVKsjIgw |
| H8hSxcD+FhWyJm9h9uLtmpuqhQo0jTUYpnTezZx2xeVPB53Ev7YCxR9Nsgj5GsVf |
| 9Z/hqLB7IFgM3pa0z3PQeUIZF/cEf72fISWIOBwwkzVrPUkXWfbuWeJXQXSs3amE |
| 5A295jD9BQp9CY0nNFSsy+qiXWToq2xT3y5zVNEStmN0SCGNaIlUnJzL9IHW+oMI |
| kPmXZMnAYBWeeCF1gf3J3aE5lZInegHNfEI0+J0LazC2aNU5Dg/BNqrmRqKWEIo/ |
| -----END RSA PRIVATE KEY----- |
| """ |
| ).encode() |
| |
| password = b"password" |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key(key_data, password, backend) |
| |
| def test_corrupt_pkcs8_format(self, backend): |
| # unenc-rsa-pkcs8.pem with a bunch of data missing. |
| key_data = textwrap.dedent( |
| """\ |
| -----BEGIN PRIVATE KEY----- |
| MIICdQIBADALBgkqhkiG9w0BAQEEggJhMIICXQIBAAKBgQC7JHoJfg6yNzLMOWet |
| 8Z49a4KD0dCspMAYvo2YAMB7/wdEycocujbhJ2n/seONi+5XqTqqFkM5VBl8rmkk |
| FPZk/7x0xmdsTPECSWnHK+HhoaNDFPR3j8jQhVo1laxiqcEhAHegi5cwtFosuJAv |
| FiRC0Cgz+frQPFQEBsAV9RuasyQxqzxrR0Ow0qncBeGBWbYE6WZhqtcLAI895b+i |
| +F4lbB4iD7T9QeIDMU/aIMXA81UO4cns1z4qDAHKeyLLrPQrJ/B4X7XC+egUWm5+ |
| hr1qmyAMusyXIBECQQDJWZ8piluf4yrYfsJAn6hF5T4RjTztbqvO0GVG2McHY7Uj |
| NPSffhzHx/ll0fQEQji+OgydCCX8o3HZrgw5YfSJAkEA7e+rqdU5nO5ZG//PSEQb |
| tjLnRiTzBH/elQhtdZ5nF7pcpNTi4k13zutmKcWW4GK75azcRGJUhu1kDM7QYAOd |
| SQJAVNkYcifkvna7GmooL5VYEsQsqLbM4v0NF2TIGNfG3z1MGp75KrC5LhL97MNR |
| we2p/bd2k0HYyCKUGnf2nMPDiQJBAI75pwittSoE240EobUGIDTSz8CJsXIxuDmL |
| z+KOpdpPRR5TQmbEMEspjsFpFymMiuYPgmihQbO2cJl1qScY5OkCQQCJ6m5tcN8l |
| Xxg/SNpjEIv+qAyUD96XVlOJlOIeLHQ8kYE0C6ZA+MsqYIzgAreJk88Yn0lU/X0/ |
| mu/UpE/BRZmR |
| -----END PRIVATE KEY----- |
| """ |
| ).encode() |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key(key_data, None, backend) |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key( |
| key_data, b"this password will not be used", backend |
| ) |
| |
| def test_pks8_encrypted_corrupt_format(self, backend): |
| # enc-rsa-pkcs8.pem with some bits flipped. |
| key_data = textwrap.dedent( |
| """\ |
| -----BEGIN ENCRYPTED PRIVATE KEY----- |
| MIICojAcBgoqhkiG9w0BDAEDMA4ECHK0M0+QuEL9AgIBIcSCAoDRq+KRY+0XP0tO |
| lwBTzViiXSXoyNnKAZKt5r5K/fGNntv22g/1s/ZNCetrqsJDC5eMUPPacz06jFq/ |
| Ipsep4/OgjQ9UAOzXNrWEoNyrHnWDo7usgD3CW0mKyqER4+wG0adVMbt3N+CJHGB |
| 85jzRmQTfkdx1rSWeSx+XyswHn8ER4+hQ+omKWMVm7AFkjjmP/KnhUnLT98J8rhU |
| ArQoFPHz/6HVkypFccNaPPNg6IA4aS2A+TU9vJYOaXSVfFB2yf99hfYYzC+ukmuU |
| 5Lun0cysK5s/5uSwDueUmDQKspnaNyiaMGDxvw8hilJc7vg0fGObfnbIpizhxJwq |
| gKBfR7Zt0Hv8OYi1He4MehfMGdbHskztF+yQ40LplBGXQrvAqpU4zShga1BoQ98T |
| 0ekbBmqj7hg47VFsppXR7DKhx7G7rpMmdKbFhAZVCjae7rRGpUtD52cpFdPhMyAX |
| huhMkoczwUW8B/rM4272lkHo6Br0yk/TQfTEGkvryflNVu6lniPTV151WV5U1M3o |
| 3G3a44eDyt7Ln+WSOpWtbPQMTrpKhur6WXgJvrpa/m02oOGdvOlDsoOCgavgQMWg |
| 7xKKL7620pHl7p7f/8tlE8q6vLXVvyNtAOgt/JAr2rgvrHaZSzDE0DwgCjBXEm+7 |
| cVMVNkHod7bLQefVanVtWqPzbmr8f7gKeuGwWSG9oew/lN2hxcLEPJHAQlnLgx3P |
| 0GdGjK9NvwA0EP2gYIeE4+UtSder7xQ7bVh25VB20R4TTIIs4aXXCVOoQPagnzaT |
| 6JLgl8FrvdfjHwIvmSOO1YMNmILBq000Q8WDqyErBDs4hsvtO6VQ4LeqJj6gClX3 |
| qeJNaJFu |
| -----END ENCRYPTED PRIVATE KEY----- |
| """ |
| ).encode() |
| |
| password = b"this password is wrong" |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key(key_data, None, backend) |
| |
| with pytest.raises(ValueError): |
| load_pem_private_key(key_data, password, backend) |
| |
| def test_rsa_pkcs8_encrypted_values(self, backend): |
| pkey = load_vectors_from_file( |
| os.path.join("asymmetric", "PKCS8", "enc-rsa-pkcs8.pem"), |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), b"foobar", backend |
| ), |
| ) |
| assert pkey |
| |
| numbers = pkey.private_numbers() |
| |
| assert numbers.public_numbers.n == int( |
| "00beec64d6db5760ac2fd4c971145641b9bd7f5c56558ece608795c79807" |
| "376a7fe5b19f95b35ca358ea5c8abd7ae051d49cd2f1e45969a1ae945460" |
| "3c14b278664a0e414ebc8913acb6203626985525e17a600611b028542dd0" |
| "562aad787fb4f1650aa318cdcff751e1b187cbf6785fbe164e9809491b95" |
| "dd68480567c99b1a57", |
| 16, |
| ) |
| |
| assert numbers.public_numbers.e == 65537 |
| |
| assert numbers.d == int( |
| "0cfe316e9dc6b8817f4fcfd5ae38a0886f68f773b8a6db4c9e6d8703c599" |
| "f3d9785c3a2c09e4c8090909fb3721e19a3009ec21221523a729265707a5" |
| "8f13063671c42a4096cad378ef2510cb59e23071489d8893ac4934dd149f" |
| "34f2d094bea57f1c8027c3a77248ac9b91218737d0c3c3dfa7d7829e6977" |
| "cf7d995688c86c81", |
| 16, |
| ) |
| |
| assert numbers.p == int( |
| "00db122ac857b2c0437d7616daa98e597bb75ca9ad3a47a70bec10c10036" |
| "03328794b225c8e3eee6ffd3fd6d2253d28e071fe27d629ab072faa14377" |
| "ce6118cb67", |
| 16, |
| ) |
| |
| assert numbers.q == int( |
| "00df1b8aa8506fcbbbb9d00257f2975e38b33d2698fd0f37e82d7ef38c56" |
| "f21b6ced63c825383782a7115cfcc093300987dbd2853b518d1c8f26382a" |
| "2d2586d391", |
| 16, |
| ) |
| |
| assert numbers.dmp1 == int( |
| "00be18aca13e60712fdf5daa85421eb10d86d654b269e1255656194fb0c4" |
| "2dd01a1070ea12c19f5c39e09587af02f7b1a1030d016a9ffabf3b36d699" |
| "ceaf38d9bf", |
| 16, |
| ) |
| |
| assert numbers.dmq1 == int( |
| "71aa8978f90a0c050744b77cf1263725b203ac9f730606d8ae1d289dce4a" |
| "28b8d534e9ea347aeb808c73107e583eb80c546d2bddadcdb3c82693a4c1" |
| "3d863451", |
| 16, |
| ) |
| |
| assert numbers.iqmp == int( |
| "136b7b1afac6e6279f71b24217b7083485a5e827d156024609dae39d48a6" |
| "bdb55af2f062cc4a3b077434e6fffad5faa29a2b5dba2bed3e4621e478c0" |
| "97ccfe7f", |
| 16, |
| ) |
| |
| def test_load_pem_dsa_private_key(self, backend): |
| key = load_vectors_from_file( |
| os.path.join("asymmetric", "PKCS8", "unenc-dsa-pkcs8.pem"), |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), None, backend |
| ), |
| ) |
| assert key |
| assert isinstance(key, dsa.DSAPrivateKey) |
| |
| params = key.parameters() |
| assert isinstance(params, dsa.DSAParameters) |
| |
| num = key.private_numbers() |
| pub = num.public_numbers |
| parameter_numbers = pub.parameter_numbers |
| assert num.x == int("00a535a8e1d0d91beafc8bee1d9b2a3a8de3311203", 16) |
| assert pub.y == int( |
| "2b260ea97dc6a12ae932c640e7df3d8ff04a8a05a0324f8d5f1b23f15fa1" |
| "70ff3f42061124eff2586cb11b49a82dcdc1b90fc6a84fb10109cb67db5d" |
| "2da971aeaf17be5e37284563e4c64d9e5fc8480258b319f0de29d54d8350" |
| "70d9e287914d77df81491f4423b62da984eb3f45eb2a29fcea5dae525ac6" |
| "ab6bcce04bfdf5b6", |
| 16, |
| ) |
| |
| assert parameter_numbers.p == int( |
| "00aa0930cc145825221caffa28ac2894196a27833de5ec21270791689420" |
| "7774a2e7b238b0d36f1b2499a2c2585083eb01432924418d867faa212dd1" |
| "071d4dceb2782794ad393cc08a4d4ada7f68d6e839a5fcd34b4e402d82cb" |
| "8a8cb40fec31911bf9bd360b034caacb4c5e947992573c9e90099c1b0f05" |
| "940cabe5d2de49a167", |
| 16, |
| ) |
| |
| assert parameter_numbers.q == int( |
| "00adc0e869b36f0ac013a681fdf4d4899d69820451", 16 |
| ) |
| |
| assert parameter_numbers.g == int( |
| "008c6b4589afa53a4d1048bfc346d1f386ca75521ccf72ddaa251286880e" |
| "e13201ff48890bbfc33d79bacaec71e7a778507bd5f1a66422e39415be03" |
| "e71141ba324f5b93131929182c88a9fa4062836066cebe74b5c6690c7d10" |
| "1106c240ab7ebd54e4e3301fd086ce6adac922fb2713a2b0887cba13b9bc" |
| "68ce5cfff241cd3246", |
| 16, |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_file", "password"), [("bad-oid-dsa-key.pem", None)] |
| ) |
| def test_load_bad_oid_key(self, key_file, password, backend): |
| with pytest.raises(ValueError): |
| load_vectors_from_file( |
| os.path.join("asymmetric", "PKCS8", key_file), |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), password, backend |
| ), |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_file", "password"), [("bad-encryption-oid.pem", b"password")] |
| ) |
| def test_load_bad_encryption_oid_key(self, key_file, password, backend): |
| with pytest.raises(ValueError): |
| load_vectors_from_file( |
| os.path.join("asymmetric", "PKCS8", key_file), |
| lambda pemfile: load_pem_private_key( |
| pemfile.read().encode(), password, backend |
| ), |
| ) |
| |
| |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| class TestRSASSHSerialization(object): |
| def test_load_ssh_public_key_unsupported(self, backend): |
| ssh_key = b"ecdsa-sha2-junk AAAAE2VjZHNhLXNoYTItbmlzdHAyNTY=" |
| |
| with pytest.raises(UnsupportedAlgorithm): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_bad_format(self, backend): |
| ssh_key = b"ssh-rsa not-a-real-key" |
| |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_rsa_too_short(self, backend): |
| ssh_key = b"ssh-rsa" |
| |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_truncated_int(self, backend): |
| ssh_key = b"ssh-rsa AAAAB3NzaC1yc2EAAAA=" |
| |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| ssh_key = b"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAACKr+IHXo" |
| |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_rsa_comment_with_spaces(self, backend): |
| ssh_key = ( |
| b"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDDu/XRP1kyK6Cgt36gts9XAk" |
| b"FiiuJLW6RU0j3KKVZSs1I7Z3UmU9/9aVh/rZV43WQG8jaR6kkcP4stOR0DEtll" |
| b"PDA7ZRBnrfiHpSQYQ874AZaAoIjgkv7DBfsE6gcDQLub0PFjWyrYQUJhtOLQEK" |
| b"vY/G0vt2iRL3juawWmCFdTK3W3XvwAdgGk71i6lHt+deOPNEPN2H58E4odrZ2f" |
| b"sxn/adpDqfb2sM0kPwQs0aWvrrKGvUaustkivQE4XWiSFnB0oJB/lKK/CKVKuy" |
| b"///ImSCGHQRvhwariN2tvZ6CBNSLh3iQgeB0AkyJlng7MXB2qYq/Ci2FUOryCX" |
| # Extra section appended |
| b"2MzHvnbv testkey@localhost extra" |
| ) |
| |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_rsa_extra_data_after_modulo(self, backend): |
| ssh_key = ( |
| b"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDDu/XRP1kyK6Cgt36gts9XAk" |
| b"FiiuJLW6RU0j3KKVZSs1I7Z3UmU9/9aVh/rZV43WQG8jaR6kkcP4stOR0DEtll" |
| b"PDA7ZRBnrfiHpSQYQ874AZaAoIjgkv7DBfsE6gcDQLub0PFjWyrYQUJhtOLQEK" |
| b"vY/G0vt2iRL3juawWmCFdTK3W3XvwAdgGk71i6lHt+deOPNEPN2H58E4odrZ2f" |
| b"sxn/adpDqfb2sM0kPwQs0aWvrrKGvUaustkivQE4XWiSFnB0oJB/lKK/CKVKuy" |
| b"///ImSCGHQRvhwariN2tvZ6CBNSLh3iQgeB0AkyJlng7MXB2qYq/Ci2FUOryCX" |
| b"2MzHvnbvAQ== testkey@localhost" |
| ) |
| |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_rsa_different_string(self, backend): |
| ssh_key = ( |
| # "AAAAB3NzA" the final A is capitalized here to cause the string |
| # ssh-rsa inside the base64 encoded blob to be incorrect. It should |
| # be a lower case 'a'. |
| b"ssh-rsa AAAAB3NzAC1yc2EAAAADAQABAAABAQDDu/XRP1kyK6Cgt36gts9XAk" |
| b"FiiuJLW6RU0j3KKVZSs1I7Z3UmU9/9aVh/rZV43WQG8jaR6kkcP4stOR0DEtll" |
| b"PDA7ZRBnrfiHpSQYQ874AZaAoIjgkv7DBfsE6gcDQLub0PFjWyrYQUJhtOLQEK" |
| b"vY/G0vt2iRL3juawWmCFdTK3W3XvwAdgGk71i6lHt+deOPNEPN2H58E4odrZ2f" |
| b"sxn/adpDqfb2sM0kPwQs0aWvrrKGvUaustkivQE4XWiSFnB0oJB/lKK/CKVKuy" |
| b"///ImSCGHQRvhwariN2tvZ6CBNSLh3iQgeB0AkyJlng7MXB2qYq/Ci2FUOryCX" |
| b"2MzHvnbvAQ== testkey@localhost" |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_rsa(self, backend): |
| ssh_key = ( |
| b"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDDu/XRP1kyK6Cgt36gts9XAk" |
| b"FiiuJLW6RU0j3KKVZSs1I7Z3UmU9/9aVh/rZV43WQG8jaR6kkcP4stOR0DEtll" |
| b"PDA7ZRBnrfiHpSQYQ874AZaAoIjgkv7DBfsE6gcDQLub0PFjWyrYQUJhtOLQEK" |
| b"vY/G0vt2iRL3juawWmCFdTK3W3XvwAdgGk71i6lHt+deOPNEPN2H58E4odrZ2f" |
| b"sxn/adpDqfb2sM0kPwQs0aWvrrKGvUaustkivQE4XWiSFnB0oJB/lKK/CKVKuy" |
| b"///ImSCGHQRvhwariN2tvZ6CBNSLh3iQgeB0AkyJlng7MXB2qYq/Ci2FUOryCX" |
| b"2MzHvnbv testkey@localhost" |
| ) |
| |
| key = load_ssh_public_key(ssh_key, backend) |
| |
| assert key is not None |
| assert isinstance(key, rsa.RSAPublicKey) |
| |
| numbers = key.public_numbers() |
| |
| expected_e = 0x10001 |
| expected_n = int( |
| "00C3BBF5D13F59322BA0A0B77EA0B6CF570241628AE24B5BA454D" |
| "23DCA295652B3523B67752653DFFD69587FAD9578DD6406F23691" |
| "EA491C3F8B2D391D0312D9653C303B651067ADF887A5241843CEF" |
| "8019680A088E092FEC305FB04EA070340BB9BD0F1635B2AD84142" |
| "61B4E2D010ABD8FC6D2FB768912F78EE6B05A60857532B75B75EF" |
| "C007601A4EF58BA947B7E75E38F3443CDD87E7C138A1DAD9D9FB3" |
| "19FF69DA43A9F6F6B0CD243F042CD1A5AFAEB286BD46AEB2D922B" |
| "D01385D6892167074A0907F94A2BF08A54ABB2FFFFC89920861D0" |
| "46F8706AB88DDADBD9E8204D48B87789081E074024C8996783B31" |
| "7076A98ABF0A2D8550EAF2097D8CCC7BE76EF", |
| 16, |
| ) |
| |
| expected = rsa.RSAPublicNumbers(expected_e, expected_n) |
| |
| assert numbers == expected |
| |
| |
| @pytest.mark.requires_backend_interface(interface=DSABackend) |
| class TestDSSSSHSerialization(object): |
| def test_load_ssh_public_key_dss_too_short(self, backend): |
| ssh_key = b"ssh-dss" |
| |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_dss_comment_with_spaces(self, backend): |
| ssh_key = ( |
| b"ssh-dss AAAAB3NzaC1kc3MAAACBALmwUtfwdjAUjU2Dixd5DvT0NDcjjr69UD" |
| b"LqSD/Xt5Al7D3GXr1WOrWGpjO0NE9qzRCvMTU7zykRH6XjuNXB6Hvv48Zfm4vm" |
| b"nHQHFmmMg2bI75JbnOwdzWnnPZJrVU4rS23dFFPqs5ug+EbhVVrcwzxahjcSjJ" |
| b"7WEQSkVQWnSPbbAAAAFQDXmpD3DIkGvLSBf1GdUF4PHKtUrQAAAIB/bJFwss+2" |
| b"fngmfG/Li5OyL7A9iVoGdkUaFaxEUROTp7wkm2z49fXFAir+/U31v50Tu98YLf" |
| b"WvKlxdHcdgQYV9Ww5LIrhWwwD4UKOwC6w5S3KHVbi3pWUi7vxJFXOWfeu1mC/J" |
| b"TWqMKR91j+rmOtdppWIZRyIVIqLcMdGO3m+2VgAAAIANFDz5KQH5NvoljpoRQi" |
| b"RgyPjxWXiE7vjLElKj4v8KrpanAywBzdhIW1y/tzpGuwRwj5ihi8iNTHgSsoTa" |
| b"j5AG5HPomJf5vJElxpu/2O9pHA52wcNObIQ7j+JA5uWusxNIbl+pF6sSiP8abr" |
| b"z53N7tPF/IhHTjBHb1Ol7IFu9p9A== testkey@localhost extra" |
| ) |
| |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_dss_extra_data_after_modulo(self, backend): |
| ssh_key = ( |
| b"ssh-dss AAAAB3NzaC1kc3MAAACBALmwUtfwdjAUjU2Dixd5DvT0NDcjjr69UD" |
| b"LqSD/Xt5Al7D3GXr1WOrWGpjO0NE9qzRCvMTU7zykRH6XjuNXB6Hvv48Zfm4vm" |
| b"nHQHFmmMg2bI75JbnOwdzWnnPZJrVU4rS23dFFPqs5ug+EbhVVrcwzxahjcSjJ" |
| b"7WEQSkVQWnSPbbAAAAFQDXmpD3DIkGvLSBf1GdUF4PHKtUrQAAAIB/bJFwss+2" |
| b"fngmfG/Li5OyL7A9iVoGdkUaFaxEUROTp7wkm2z49fXFAir+/U31v50Tu98YLf" |
| b"WvKlxdHcdgQYV9Ww5LIrhWwwD4UKOwC6w5S3KHVbi3pWUi7vxJFXOWfeu1mC/J" |
| b"TWqMKR91j+rmOtdppWIZRyIVIqLcMdGO3m+2VgAAAIANFDz5KQH5NvoljpoRQi" |
| b"RgyPjxWXiE7vjLElKj4v8KrpanAywBzdhIW1y/tzpGuwRwj5ihi8iNTHgSsoTa" |
| b"j5AG5HPomJf5vJElxpu/2O9pHA52wcNObIQ7j+JA5uWusxNIbl+pF6sSiP8abr" |
| b"z53N7tPF/IhHTjBHb1Ol7IFu9p9AAwMD== testkey@localhost" |
| ) |
| |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_dss_different_string(self, backend): |
| ssh_key = ( |
| # "AAAAB3NzA" the final A is capitalized here to cause the string |
| # ssh-dss inside the base64 encoded blob to be incorrect. It should |
| # be a lower case 'a'. |
| b"ssh-dss AAAAB3NzAC1kc3MAAACBALmwUtfwdjAUjU2Dixd5DvT0NDcjjr69UD" |
| b"LqSD/Xt5Al7D3GXr1WOrWGpjO0NE9qzRCvMTU7zykRH6XjuNXB6Hvv48Zfm4vm" |
| b"nHQHFmmMg2bI75JbnOwdzWnnPZJrVU4rS23dFFPqs5ug+EbhVVrcwzxahjcSjJ" |
| b"7WEQSkVQWnSPbbAAAAFQDXmpD3DIkGvLSBf1GdUF4PHKtUrQAAAIB/bJFwss+2" |
| b"fngmfG/Li5OyL7A9iVoGdkUaFaxEUROTp7wkm2z49fXFAir+/U31v50Tu98YLf" |
| b"WvKlxdHcdgQYV9Ww5LIrhWwwD4UKOwC6w5S3KHVbi3pWUi7vxJFXOWfeu1mC/J" |
| b"TWqMKR91j+rmOtdppWIZRyIVIqLcMdGO3m+2VgAAAIANFDz5KQH5NvoljpoRQi" |
| b"RgyPjxWXiE7vjLElKj4v8KrpanAywBzdhIW1y/tzpGuwRwj5ihi8iNTHgSsoTa" |
| b"j5AG5HPomJf5vJElxpu/2O9pHA52wcNObIQ7j+JA5uWusxNIbl+pF6sSiP8abr" |
| b"z53N7tPF/IhHTjBHb1Ol7IFu9p9A== testkey@localhost" |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_dss(self, backend): |
| ssh_key = ( |
| b"ssh-dss AAAAB3NzaC1kc3MAAACBALmwUtfwdjAUjU2Dixd5DvT0NDcjjr69UD" |
| b"LqSD/Xt5Al7D3GXr1WOrWGpjO0NE9qzRCvMTU7zykRH6XjuNXB6Hvv48Zfm4vm" |
| b"nHQHFmmMg2bI75JbnOwdzWnnPZJrVU4rS23dFFPqs5ug+EbhVVrcwzxahjcSjJ" |
| b"7WEQSkVQWnSPbbAAAAFQDXmpD3DIkGvLSBf1GdUF4PHKtUrQAAAIB/bJFwss+2" |
| b"fngmfG/Li5OyL7A9iVoGdkUaFaxEUROTp7wkm2z49fXFAir+/U31v50Tu98YLf" |
| b"WvKlxdHcdgQYV9Ww5LIrhWwwD4UKOwC6w5S3KHVbi3pWUi7vxJFXOWfeu1mC/J" |
| b"TWqMKR91j+rmOtdppWIZRyIVIqLcMdGO3m+2VgAAAIANFDz5KQH5NvoljpoRQi" |
| b"RgyPjxWXiE7vjLElKj4v8KrpanAywBzdhIW1y/tzpGuwRwj5ihi8iNTHgSsoTa" |
| b"j5AG5HPomJf5vJElxpu/2O9pHA52wcNObIQ7j+JA5uWusxNIbl+pF6sSiP8abr" |
| b"z53N7tPF/IhHTjBHb1Ol7IFu9p9A== testkey@localhost" |
| ) |
| |
| key = load_ssh_public_key(ssh_key, backend) |
| |
| assert key is not None |
| assert isinstance(key, dsa.DSAPublicKey) |
| |
| numbers = key.public_numbers() |
| |
| expected_y = int( |
| "d143cf92901f936fa258e9a11422460c8f8f1597884eef8cb1252a3e2ff0aae" |
| "96a7032c01cdd8485b5cbfb73a46bb04708f98a18bc88d4c7812b284da8f900" |
| "6e473e89897f9bc9125c69bbfd8ef691c0e76c1c34e6c843b8fe240e6e5aeb3" |
| "13486e5fa917ab1288ff1a6ebcf9dcdeed3c5fc88474e30476f53a5ec816ef6" |
| "9f4", |
| 16, |
| ) |
| expected_p = int( |
| "b9b052d7f07630148d4d838b17790ef4f43437238ebebd5032ea483fd7b7902" |
| "5ec3dc65ebd563ab586a633b4344f6acd10af31353bcf29111fa5e3b8d5c1e8" |
| "7befe3c65f9b8be69c740716698c8366c8ef925b9cec1dcd69e73d926b554e2" |
| "b4b6ddd1453eab39ba0f846e1555adcc33c5a8637128c9ed61104a45505a748" |
| "f6db", |
| 16, |
| ) |
| expected_q = 1230879958723280233885494314531920096931919647917 |
| expected_g = int( |
| "7f6c9170b2cfb67e78267c6fcb8b93b22fb03d895a0676451a15ac44511393a" |
| "7bc249b6cf8f5f5c5022afefd4df5bf9d13bbdf182df5af2a5c5d1dc7604185" |
| "7d5b0e4b22b856c300f850a3b00bac394b728755b8b7a56522eefc491573967" |
| "debb5982fc94d6a8c291f758feae63ad769a5621947221522a2dc31d18ede6f" |
| "b656", |
| 16, |
| ) |
| expected = dsa.DSAPublicNumbers( |
| expected_y, |
| dsa.DSAParameterNumbers(expected_p, expected_q, expected_g), |
| ) |
| |
| assert numbers == expected |
| |
| |
| @pytest.mark.requires_backend_interface(interface=EllipticCurveBackend) |
| class TestECDSASSHSerialization(object): |
| def test_load_ssh_public_key_ecdsa_nist_p256(self, backend): |
| _skip_curve_unsupported(backend, ec.SECP256R1()) |
| |
| ssh_key = ( |
| b"ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAy" |
| b"NTYAAABBBGG2MfkHXp0UkxUyllDzWNBAImsvt5t7pFtTXegZK2WbGxml8zMrgWi5" |
| b"teIg1TO03/FD9hbpBFgBeix3NrCFPls= root@cloud-server-01" |
| ) |
| key = load_ssh_public_key(ssh_key, backend) |
| assert isinstance(key, ec.EllipticCurvePublicKey) |
| |
| expected_x = int( |
| "44196257377740326295529888716212621920056478823906609851236662550" |
| "785814128027", |
| 10, |
| ) |
| expected_y = int( |
| "12257763433170736656417248739355923610241609728032203358057767672" |
| "925775019611", |
| 10, |
| ) |
| |
| assert key.public_numbers() == ec.EllipticCurvePublicNumbers( |
| expected_x, expected_y, ec.SECP256R1() |
| ) |
| |
| def test_load_ssh_public_key_byteslike(self, backend): |
| _skip_curve_unsupported(backend, ec.SECP256R1()) |
| |
| ssh_key = ( |
| b"ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAy" |
| b"NTYAAABBBGG2MfkHXp0UkxUyllDzWNBAImsvt5t7pFtTXegZK2WbGxml8zMrgWi5" |
| b"teIg1TO03/FD9hbpBFgBeix3NrCFPls= root@cloud-server-01" |
| ) |
| assert load_ssh_public_key(bytearray(ssh_key), backend) |
| if six.PY3: |
| assert load_ssh_public_key(memoryview(ssh_key), backend) |
| assert load_ssh_public_key(memoryview(bytearray(ssh_key)), backend) |
| |
| def test_load_ssh_public_key_ecdsa_nist_p384(self, backend): |
| _skip_curve_unsupported(backend, ec.SECP384R1()) |
| ssh_key = ( |
| b"ecdsa-sha2-nistp384 AAAAE2VjZHNhLXNoYTItbmlzdHAzODQAAAAIbmlzdHAz" |
| b"ODQAAABhBMzucOm9wbwg4iMr5QL0ya0XNQGXpw4wM5f12E3tWhdcrzyGHyel71t1" |
| b"4bvF9JZ2/WIuSxUr33XDl8jYo+lMQ5N7Vanc7f7i3AR1YydatL3wQfZStQ1I3rBa" |
| b"qQtRSEU8Tg== root@cloud-server-01" |
| ) |
| key = load_ssh_public_key(ssh_key, backend) |
| |
| expected_x = int( |
| "31541830871345183397582554827482786756220448716666815789487537666" |
| "592636882822352575507883817901562613492450642523901", |
| 10, |
| ) |
| expected_y = int( |
| "15111413269431823234030344298767984698884955023183354737123929430" |
| "995703524272335782455051101616329050844273733614670", |
| 10, |
| ) |
| |
| assert key.public_numbers() == ec.EllipticCurvePublicNumbers( |
| expected_x, expected_y, ec.SECP384R1() |
| ) |
| |
| def test_load_ssh_public_key_ecdsa_nist_p521(self, backend): |
| _skip_curve_unsupported(backend, ec.SECP521R1()) |
| ssh_key = ( |
| b"ecdsa-sha2-nistp521 AAAAE2VjZHNhLXNoYTItbmlzdHA1MjEAAAAIbmlzdHA1" |
| b"MjEAAACFBAGTrRhMSEgF6Ni+PXNz+5fjS4lw3ypUILVVQ0Av+0hQxOx+MyozELon" |
| b"I8NKbrbBjijEs1GuImsmkTmWsMXS1j2A7wB4Kseh7W9KA9IZJ1+TMrzWUEwvOOXi" |
| b"wT23pbaWWXG4NaM7vssWfZBnvz3S174TCXnJ+DSccvWBFnKP0KchzLKxbg== " |
| b"root@cloud-server-01" |
| ) |
| key = load_ssh_public_key(ssh_key, backend) |
| |
| expected_x = int( |
| "54124123120178189598842622575230904027376313369742467279346415219" |
| "77809037378785192537810367028427387173980786968395921877911964629" |
| "142163122798974160187785455", |
| 10, |
| ) |
| expected_y = int( |
| "16111775122845033200938694062381820957441843014849125660011303579" |
| "15284560361402515564433711416776946492019498546572162801954089916" |
| "006665939539407104638103918", |
| 10, |
| ) |
| |
| assert key.public_numbers() == ec.EllipticCurvePublicNumbers( |
| expected_x, expected_y, ec.SECP521R1() |
| ) |
| |
| def test_load_ssh_public_key_ecdsa_nist_p256_trailing_data(self, backend): |
| ssh_key = ( |
| b"ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAy" |
| b"NTYAAABBBGG2MfkHXp0UkxUyllDzWNBAImsvt5t7pFtTXegZK2WbGxml8zMrgWi5" |
| b"teIg1TO03/FD9hbpBFgBeix3NrCFPltB= root@cloud-server-01" |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_ecdsa_nist_p256_missing_data(self, backend): |
| ssh_key = ( |
| b"ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAy" |
| b"NTYAAABBBGG2MfkHXp0UkxUyllDzWNBAImsvt5t7pFtTXegZK2WbGxml8zMrgWi5" |
| b"teIg1TO03/FD9hbpBFgBeix3NrCF= root@cloud-server-01" |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_ecdsa_nist_p256_compressed(self, backend): |
| # If we ever implement compressed points, note that this is not a valid |
| # one, it just has the compressed marker in the right place. |
| ssh_key = ( |
| b"ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAy" |
| b"NTYAAABBAWG2MfkHXp0UkxUyllDzWNBAImsvt5t7pFtTXegZK2WbGxml8zMrgWi5" |
| b"teIg1TO03/FD9hbpBFgBeix3NrCFPls= root@cloud-server-01" |
| ) |
| with pytest.raises(NotImplementedError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_ecdsa_nist_p256_bad_curve_name(self, backend): |
| ssh_key = ( |
| # The curve name in here is changed to be "nistp255". |
| b"ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAy" |
| b"NTUAAABBBGG2MfkHXp0UkxUyllDzWNBAImsvt5t7pFtTXegZK2WbGxml8zMrgWi5" |
| b"teIg1TO03/FD9hbpBFgBeix3NrCFPls= root@cloud-server-01" |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| |
| @pytest.mark.supported( |
| only_if=lambda backend: backend.ed25519_supported(), |
| skip_message="Requires OpenSSL with Ed25519 support", |
| ) |
| class TestEd25519SSHSerialization(object): |
| def test_load_ssh_public_key(self, backend): |
| ssh_key = ( |
| b"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIG2fgpmpYO61qeAxGd0wgRaN/E4" |
| b"GR+xWvBmvxjxrB1vG [email protected]" |
| ) |
| key = load_ssh_public_key(ssh_key, backend) |
| assert isinstance(key, ed25519.Ed25519PublicKey) |
| assert key.public_bytes(Encoding.Raw, PublicFormat.Raw) == ( |
| b"m\x9f\x82\x99\xa9`\xee\xb5\xa9\xe01\x19\xdd0\x81\x16\x8d\xfc" |
| b"N\x06G\xecV\xbc\x19\xaf\xc6<k\x07[\xc6" |
| ) |
| |
| def test_public_bytes_openssh(self, backend): |
| ssh_key = ( |
| b"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIG2fgpmpYO61qeAxGd0wgRaN/E4" |
| b"GR+xWvBmvxjxrB1vG" |
| ) |
| key = load_ssh_public_key(ssh_key, backend) |
| assert isinstance(key, ed25519.Ed25519PublicKey) |
| assert ( |
| key.public_bytes(Encoding.OpenSSH, PublicFormat.OpenSSH) == ssh_key |
| ) |
| |
| def test_load_ssh_public_key_not_32_bytes(self, backend): |
| ssh_key = ( |
| b"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAI22fgpmpYO61qeAxGd0wgRaN/E4" |
| b"GR+xWvBmvxjxrB1vGaGVs [email protected]" |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| def test_load_ssh_public_key_trailing_data(self, backend): |
| ssh_key = ( |
| b"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIG2fgpmpYO61qeAxGd0wgRa" |
| b"N/E4GR+xWvBmvxjxrB1vGdHJhaWxpbmdkYXRh [email protected]" |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_public_key(ssh_key, backend) |
| |
| |
| class TestKeySerializationEncryptionTypes(object): |
| def test_non_bytes_password(self): |
| with pytest.raises(ValueError): |
| BestAvailableEncryption(object()) |
| |
| def test_encryption_with_zero_length_password(self): |
| with pytest.raises(ValueError): |
| BestAvailableEncryption(b"") |
| |
| |
| @pytest.mark.supported( |
| only_if=lambda backend: backend.ed25519_supported(), |
| skip_message="Requires OpenSSL with Ed25519 support", |
| ) |
| class TestEd25519Serialization(object): |
| def test_load_der_private_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "Ed25519", "ed25519-pkcs8-enc.der"), |
| lambda derfile: derfile.read(), |
| mode="rb", |
| ) |
| unencrypted = load_vectors_from_file( |
| os.path.join("asymmetric", "Ed25519", "ed25519-pkcs8.der"), |
| lambda derfile: derfile.read(), |
| mode="rb", |
| ) |
| key = load_der_private_key(data, b"password", backend) |
| assert ( |
| key.private_bytes( |
| Encoding.DER, PrivateFormat.PKCS8, NoEncryption() |
| ) |
| == unencrypted |
| ) |
| |
| def test_load_pem_private_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "Ed25519", "ed25519-pkcs8-enc.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| unencrypted = load_vectors_from_file( |
| os.path.join("asymmetric", "Ed25519", "ed25519-pkcs8.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| key = load_pem_private_key(data, b"password", backend) |
| assert ( |
| key.private_bytes( |
| Encoding.PEM, PrivateFormat.PKCS8, NoEncryption() |
| ) |
| == unencrypted |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "encoding", "loader"), |
| [ |
| ( |
| ["Ed25519", "ed25519-pub.pem"], |
| Encoding.PEM, |
| load_pem_public_key, |
| ), |
| ( |
| ["Ed25519", "ed25519-pub.der"], |
| Encoding.DER, |
| load_der_public_key, |
| ), |
| ], |
| ) |
| def test_load_public_key(self, key_path, encoding, loader, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| public_key = loader(data, backend) |
| assert ( |
| public_key.public_bytes( |
| encoding, PublicFormat.SubjectPublicKeyInfo |
| ) |
| == data |
| ) |
| |
| def test_openssl_serialization_unsupported(self, backend): |
| key = ed25519.Ed25519PrivateKey.generate() |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.PEM, |
| PrivateFormat.TraditionalOpenSSL, |
| NoEncryption(), |
| ) |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.DER, |
| PrivateFormat.TraditionalOpenSSL, |
| NoEncryption(), |
| ) |
| |
| |
| @pytest.mark.supported( |
| only_if=lambda backend: backend.x448_supported(), |
| skip_message="Requires OpenSSL with X448 support", |
| ) |
| class TestX448Serialization(object): |
| def test_load_der_private_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "X448", "x448-pkcs8-enc.der"), |
| lambda derfile: derfile.read(), |
| mode="rb", |
| ) |
| unencrypted = load_vectors_from_file( |
| os.path.join("asymmetric", "X448", "x448-pkcs8.der"), |
| lambda derfile: derfile.read(), |
| mode="rb", |
| ) |
| key = load_der_private_key(data, b"password", backend) |
| assert ( |
| key.private_bytes( |
| Encoding.DER, PrivateFormat.PKCS8, NoEncryption() |
| ) |
| == unencrypted |
| ) |
| |
| def test_load_pem_private_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "X448", "x448-pkcs8-enc.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| unencrypted = load_vectors_from_file( |
| os.path.join("asymmetric", "X448", "x448-pkcs8.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| key = load_pem_private_key(data, b"password", backend) |
| assert ( |
| key.private_bytes( |
| Encoding.PEM, PrivateFormat.PKCS8, NoEncryption() |
| ) |
| == unencrypted |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "encoding", "loader"), |
| [ |
| (["X448", "x448-pub.pem"], Encoding.PEM, load_pem_public_key), |
| (["X448", "x448-pub.der"], Encoding.DER, load_der_public_key), |
| ], |
| ) |
| def test_load_public_key(self, key_path, encoding, loader, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| public_key = loader(data, backend) |
| assert ( |
| public_key.public_bytes( |
| encoding, PublicFormat.SubjectPublicKeyInfo |
| ) |
| == data |
| ) |
| |
| def test_openssl_serialization_unsupported(self, backend): |
| key = x448.X448PrivateKey.generate() |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.PEM, |
| PrivateFormat.TraditionalOpenSSL, |
| NoEncryption(), |
| ) |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.DER, |
| PrivateFormat.TraditionalOpenSSL, |
| NoEncryption(), |
| ) |
| |
| def test_openssh_serialization_unsupported(self, backend): |
| key = x448.X448PrivateKey.generate() |
| with pytest.raises(ValueError): |
| key.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.PEM, PrivateFormat.OpenSSH, NoEncryption() |
| ) |
| |
| |
| @pytest.mark.supported( |
| only_if=lambda backend: backend.x25519_supported(), |
| skip_message="Requires OpenSSL with X25519 support", |
| ) |
| class TestX25519Serialization(object): |
| def test_load_der_private_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "X25519", "x25519-pkcs8-enc.der"), |
| lambda derfile: derfile.read(), |
| mode="rb", |
| ) |
| unencrypted = load_vectors_from_file( |
| os.path.join("asymmetric", "X25519", "x25519-pkcs8.der"), |
| lambda derfile: derfile.read(), |
| mode="rb", |
| ) |
| key = load_der_private_key(data, b"password", backend) |
| assert ( |
| key.private_bytes( |
| Encoding.DER, PrivateFormat.PKCS8, NoEncryption() |
| ) |
| == unencrypted |
| ) |
| |
| def test_load_pem_private_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "X25519", "x25519-pkcs8-enc.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| unencrypted = load_vectors_from_file( |
| os.path.join("asymmetric", "X25519", "x25519-pkcs8.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| key = load_pem_private_key(data, b"password", backend) |
| assert ( |
| key.private_bytes( |
| Encoding.PEM, PrivateFormat.PKCS8, NoEncryption() |
| ) |
| == unencrypted |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "encoding", "loader"), |
| [ |
| (["X25519", "x25519-pub.pem"], Encoding.PEM, load_pem_public_key), |
| (["X25519", "x25519-pub.der"], Encoding.DER, load_der_public_key), |
| ], |
| ) |
| def test_load_public_key(self, key_path, encoding, loader, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| public_key = loader(data, backend) |
| assert ( |
| public_key.public_bytes( |
| encoding, PublicFormat.SubjectPublicKeyInfo |
| ) |
| == data |
| ) |
| |
| def test_openssl_serialization_unsupported(self, backend): |
| key = x25519.X25519PrivateKey.generate() |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.PEM, |
| PrivateFormat.TraditionalOpenSSL, |
| NoEncryption(), |
| ) |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.DER, |
| PrivateFormat.TraditionalOpenSSL, |
| NoEncryption(), |
| ) |
| |
| def test_openssh_serialization_unsupported(self, backend): |
| key = x25519.X25519PrivateKey.generate() |
| with pytest.raises(ValueError): |
| key.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.PEM, PrivateFormat.OpenSSH, NoEncryption() |
| ) |
| |
| |
| @pytest.mark.supported( |
| only_if=lambda backend: backend.ed448_supported(), |
| skip_message="Requires OpenSSL with Ed448 support", |
| ) |
| class TestEd448Serialization(object): |
| def test_load_der_private_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "Ed448", "ed448-pkcs8-enc.der"), |
| lambda derfile: derfile.read(), |
| mode="rb", |
| ) |
| unencrypted = load_vectors_from_file( |
| os.path.join("asymmetric", "Ed448", "ed448-pkcs8.der"), |
| lambda derfile: derfile.read(), |
| mode="rb", |
| ) |
| key = load_der_private_key(data, b"password", backend) |
| assert ( |
| key.private_bytes( |
| Encoding.DER, PrivateFormat.PKCS8, NoEncryption() |
| ) |
| == unencrypted |
| ) |
| |
| def test_load_pem_private_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "Ed448", "ed448-pkcs8-enc.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| unencrypted = load_vectors_from_file( |
| os.path.join("asymmetric", "Ed448", "ed448-pkcs8.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| key = load_pem_private_key(data, b"password", backend) |
| assert ( |
| key.private_bytes( |
| Encoding.PEM, PrivateFormat.PKCS8, NoEncryption() |
| ) |
| == unencrypted |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "encoding", "loader"), |
| [ |
| (["Ed448", "ed448-pub.pem"], Encoding.PEM, load_pem_public_key), |
| (["Ed448", "ed448-pub.der"], Encoding.DER, load_der_public_key), |
| ], |
| ) |
| def test_load_public_key(self, key_path, encoding, loader, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| public_key = loader(data, backend) |
| assert ( |
| public_key.public_bytes( |
| encoding, PublicFormat.SubjectPublicKeyInfo |
| ) |
| == data |
| ) |
| |
| def test_openssl_serialization_unsupported(self, backend): |
| key = ed448.Ed448PrivateKey.generate() |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.PEM, |
| PrivateFormat.TraditionalOpenSSL, |
| NoEncryption(), |
| ) |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.DER, |
| PrivateFormat.TraditionalOpenSSL, |
| NoEncryption(), |
| ) |
| |
| def test_openssh_serialization_unsupported(self, backend): |
| key = ed448.Ed448PrivateKey.generate() |
| with pytest.raises(ValueError): |
| key.public_key().public_bytes( |
| Encoding.OpenSSH, |
| PublicFormat.OpenSSH, |
| ) |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.PEM, |
| PrivateFormat.OpenSSH, |
| NoEncryption(), |
| ) |
| |
| |
| class TestDHSerialization(object): |
| """Test all options with least-supported key type.""" |
| |
| @pytest.mark.skip_fips(reason="non-FIPS parameters") |
| def test_dh_public_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "DH", "dhkey.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| public_key = load_pem_private_key(data, None, backend).public_key() |
| for enc in ( |
| Encoding.PEM, |
| Encoding.DER, |
| Encoding.OpenSSH, |
| Encoding.Raw, |
| Encoding.X962, |
| ): |
| for fmt in ( |
| PublicFormat.SubjectPublicKeyInfo, |
| PublicFormat.PKCS1, |
| PublicFormat.OpenSSH, |
| PublicFormat.Raw, |
| PublicFormat.CompressedPoint, |
| PublicFormat.UncompressedPoint, |
| ): |
| if ( |
| enc in (Encoding.PEM, Encoding.DER) |
| and fmt == PublicFormat.SubjectPublicKeyInfo |
| ): |
| # tested elsewhere |
| continue |
| with pytest.raises(ValueError): |
| public_key.public_bytes(enc, fmt) |
| |
| @pytest.mark.skip_fips(reason="non-FIPS parameters") |
| def test_dh_private_key(self, backend): |
| data = load_vectors_from_file( |
| os.path.join("asymmetric", "DH", "dhkey.pem"), |
| lambda pemfile: pemfile.read(), |
| mode="rb", |
| ) |
| private_key = load_pem_private_key(data, None, backend) |
| for enc in ( |
| Encoding.PEM, |
| Encoding.DER, |
| Encoding.OpenSSH, |
| Encoding.Raw, |
| Encoding.X962, |
| ): |
| for fmt in ( |
| PrivateFormat.PKCS8, |
| PrivateFormat.TraditionalOpenSSL, |
| PrivateFormat.Raw, |
| ): |
| if ( |
| enc in (Encoding.PEM, Encoding.DER) |
| and fmt is PrivateFormat.PKCS8 |
| ): |
| # tested elsewhere |
| continue |
| with pytest.raises(ValueError): |
| private_key.private_bytes(enc, fmt, NoEncryption()) |
| |
| |
| @pytest.mark.requires_backend_interface(interface=RSABackend) |
| @pytest.mark.requires_backend_interface(interface=DSABackend) |
| @pytest.mark.requires_backend_interface(interface=EllipticCurveBackend) |
| class TestOpenSSHSerialization(object): |
| @pytest.mark.parametrize( |
| ("key_file", "cert_file"), |
| [ |
| ("rsa-psw.key.pub", None), |
| ("rsa-nopsw.key.pub", "rsa-nopsw.key-cert.pub"), |
| ("dsa-psw.key.pub", None), |
| ("dsa-nopsw.key.pub", "dsa-nopsw.key-cert.pub"), |
| ("ecdsa-psw.key.pub", None), |
| ("ecdsa-nopsw.key.pub", "ecdsa-nopsw.key-cert.pub"), |
| ("ed25519-psw.key.pub", None), |
| ("ed25519-nopsw.key.pub", "ed25519-nopsw.key-cert.pub"), |
| ], |
| ) |
| def test_load_ssh_public_key(self, key_file, cert_file, backend): |
| if "ed25519" in key_file and not backend.ed25519_supported(): |
| pytest.skip("Requires OpenSSL with Ed25519 support") |
| |
| # normal public key |
| pub_data = load_vectors_from_file( |
| os.path.join("asymmetric", "OpenSSH", key_file), |
| lambda f: f.read(), |
| mode="rb", |
| ) |
| public_key = load_ssh_public_key(pub_data, backend) |
| nocomment_data = b" ".join(pub_data.split()[:2]) |
| assert ( |
| public_key.public_bytes(Encoding.OpenSSH, PublicFormat.OpenSSH) |
| == nocomment_data |
| ) |
| |
| self.run_partial_pubkey(pub_data, backend) |
| |
| # parse public key with ssh certificate |
| if cert_file: |
| cert_data = load_vectors_from_file( |
| os.path.join("asymmetric", "OpenSSH", cert_file), |
| lambda f: f.read(), |
| mode="rb", |
| ) |
| cert_key = load_ssh_public_key(cert_data, backend) |
| assert ( |
| cert_key.public_bytes(Encoding.OpenSSH, PublicFormat.OpenSSH) |
| == nocomment_data |
| ) |
| |
| # try with more spaces |
| cert_data = b" \t ".join(cert_data.split()) |
| cert_key = load_ssh_public_key(cert_data, backend) |
| assert ( |
| cert_key.public_bytes(Encoding.OpenSSH, PublicFormat.OpenSSH) |
| == nocomment_data |
| ) |
| |
| self.run_partial_pubkey(cert_data, backend) |
| |
| def run_partial_pubkey(self, pubdata, backend): |
| parts = pubdata.split() |
| raw = base64.b64decode(parts[1]) |
| for i in range(1, len(raw)): |
| frag = base64.b64encode(raw[:i]) |
| new_pub = b" ".join([parts[0], frag]) |
| with pytest.raises(ValueError): |
| load_ssh_public_key(new_pub, backend) |
| |
| @pytest.mark.parametrize( |
| ("key_file",), |
| [ |
| ("rsa-nopsw.key",), |
| ("rsa-psw.key",), |
| ("dsa-nopsw.key",), |
| ("dsa-psw.key",), |
| ("ecdsa-nopsw.key",), |
| ("ecdsa-psw.key",), |
| ("ed25519-nopsw.key",), |
| ("ed25519-psw.key",), |
| ], |
| ) |
| def test_load_ssh_private_key(self, key_file, backend): |
| if "ed25519" in key_file and not backend.ed25519_supported(): |
| pytest.skip("Requires OpenSSL with Ed25519 support") |
| if "-psw" in key_file and not ssh._bcrypt_supported: |
| pytest.skip("Requires bcrypt module") |
| |
| # read public and private key from ssh-keygen |
| priv_data = load_vectors_from_file( |
| os.path.join("asymmetric", "OpenSSH", key_file), |
| lambda f: f.read(), |
| mode="rb", |
| ) |
| pub_data = load_vectors_from_file( |
| os.path.join("asymmetric", "OpenSSH", key_file + ".pub"), |
| lambda f: f.read(), |
| mode="rb", |
| ) |
| nocomment_data = b" ".join(pub_data.split()[:2]) |
| |
| # load and compare |
| password = None |
| if "-psw" in key_file: |
| password = b"password" |
| private_key = load_ssh_private_key(priv_data, password, backend) |
| assert ( |
| private_key.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| == nocomment_data |
| ) |
| |
| # bytearray |
| private_key = load_ssh_private_key( |
| bytearray(priv_data), password, backend |
| ) |
| assert ( |
| private_key.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| == nocomment_data |
| ) |
| |
| if six.PY3: |
| # memoryview(bytes) |
| private_key = load_ssh_private_key( |
| memoryview(priv_data), password, backend |
| ) |
| assert ( |
| private_key.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| == nocomment_data |
| ) |
| |
| # memoryview(bytearray) |
| private_key = load_ssh_private_key( |
| memoryview(bytearray(priv_data)), password, backend |
| ) |
| assert ( |
| private_key.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| == nocomment_data |
| ) |
| |
| # serialize with own code and reload |
| encryption = NoEncryption() |
| if password: |
| encryption = BestAvailableEncryption(password) |
| priv_data2 = private_key.private_bytes( |
| Encoding.PEM, |
| PrivateFormat.OpenSSH, |
| encryption, |
| ) |
| private_key2 = load_ssh_private_key(priv_data2, password, backend) |
| assert ( |
| private_key2.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| == nocomment_data |
| ) |
| |
| # make sure multi-line base64 is used |
| maxline = max(map(len, priv_data2.split(b"\n"))) |
| assert maxline < 80 |
| |
| @pytest.mark.supported( |
| only_if=lambda backend: ssh._bcrypt_supported, |
| skip_message="Requires that bcrypt exists", |
| ) |
| def test_bcrypt_encryption(self, backend): |
| private_key = ec.generate_private_key(ec.SECP256R1(), backend) |
| pub1 = private_key.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| |
| for psw in ( |
| b"1", |
| b"1234", |
| b"1234" * 4, |
| b"x" * 72, |
| ): |
| # BestAvailableEncryption does not handle bytes-like? |
| best = BestAvailableEncryption(psw) |
| encdata = private_key.private_bytes( |
| Encoding.PEM, PrivateFormat.OpenSSH, best |
| ) |
| decoded_key = load_ssh_private_key(encdata, psw, backend) |
| pub2 = decoded_key.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| assert pub1 == pub2 |
| |
| # bytearray |
| decoded_key2 = load_ssh_private_key( |
| bytearray(encdata), psw, backend |
| ) |
| pub2 = decoded_key2.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| assert pub1 == pub2 |
| |
| if six.PY3: |
| # memoryview(bytes) |
| decoded_key2 = load_ssh_private_key( |
| memoryview(encdata), psw, backend |
| ) |
| pub2 = decoded_key2.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| assert pub1 == pub2 |
| |
| # memoryview(bytearray) |
| decoded_key2 = load_ssh_private_key( |
| memoryview(bytearray(encdata)), psw, backend |
| ) |
| pub2 = decoded_key2.public_key().public_bytes( |
| Encoding.OpenSSH, PublicFormat.OpenSSH |
| ) |
| assert pub1 == pub2 |
| |
| with pytest.raises(ValueError): |
| decoded_key = load_ssh_private_key(encdata, None, backend) |
| with pytest.raises(ValueError): |
| decoded_key = load_ssh_private_key(encdata, b"wrong", backend) |
| |
| @pytest.mark.supported( |
| only_if=lambda backend: not ssh._bcrypt_supported, |
| skip_message="Requires that bcrypt is missing", |
| ) |
| def test_missing_bcrypt(self, backend): |
| priv_data = load_vectors_from_file( |
| os.path.join("asymmetric", "OpenSSH", "ecdsa-psw.key"), |
| lambda f: f.read(), |
| mode="rb", |
| ) |
| with pytest.raises(UnsupportedAlgorithm): |
| load_ssh_private_key(priv_data, b"password", backend) |
| |
| private_key = ec.generate_private_key(ec.SECP256R1(), backend) |
| with pytest.raises(UnsupportedAlgorithm): |
| private_key.private_bytes( |
| Encoding.PEM, |
| PrivateFormat.OpenSSH, |
| BestAvailableEncryption(b"x"), |
| ) |
| |
| def test_fraglist_corners(self): |
| f = ssh._FragList() |
| with pytest.raises(ValueError): |
| f.put_mpint(-1) |
| f.put_mpint(0) |
| f.put_mpint(0x80) |
| assert f.tobytes() == b"\0\0\0\0" + b"\0\0\0\x02" + b"\0\x80" |
| |
| def make_file( |
| self, |
| magic=b"openssh-key-v1\0", |
| ciphername=b"none", |
| kdfname=b"none", |
| kdfoptions=b"", |
| nkeys=1, |
| pub_type=b"ecdsa-sha2-nistp256", |
| pub_fields=( |
| b"nistp256", |
| b"\x04" * 65, |
| ), |
| priv_type=None, |
| priv_fields=(b"nistp256", b"\x04" * 65, b"\x7F" * 32), |
| comment=b"comment", |
| checkval1=b"1234", |
| checkval2=b"1234", |
| pad=None, |
| header=b"-----BEGIN OPENSSH PRIVATE KEY-----\n", |
| footer=b"-----END OPENSSH PRIVATE KEY-----\n", |
| cut=8192, |
| ): |
| """Create private key file""" |
| if not priv_type: |
| priv_type = pub_type |
| |
| pub = ssh._FragList() |
| for elem in (pub_type,) + pub_fields: |
| pub.put_sshstr(elem) |
| |
| secret = ssh._FragList([checkval1, checkval2]) |
| for i in range(nkeys): |
| for elem in (priv_type,) + priv_fields + (comment,): |
| secret.put_sshstr(elem) |
| |
| if pad is None: |
| pad_len = 8 - (secret.size() % 8) |
| pad = bytearray(range(1, 1 + pad_len)) |
| secret.put_raw(pad) |
| |
| main = ssh._FragList([magic]) |
| main.put_sshstr(ciphername) |
| main.put_sshstr(kdfname) |
| main.put_sshstr(kdfoptions) |
| main.put_u32(nkeys) |
| for i in range(nkeys): |
| main.put_sshstr(pub) |
| main.put_sshstr(secret) |
| |
| res = main.tobytes() |
| return ssh._ssh_pem_encode(res[:cut], header, footer) |
| |
| def test_ssh_make_file(self, backend): |
| # check if works by default |
| data = self.make_file() |
| key = load_ssh_private_key(data, None, backend) |
| assert isinstance(key, ec.EllipticCurvePrivateKey) |
| |
| def test_load_ssh_private_key_errors(self, backend): |
| # bad kdf |
| data = self.make_file(kdfname=b"unknown", ciphername=b"aes256-ctr") |
| with pytest.raises(UnsupportedAlgorithm): |
| load_ssh_private_key(data, None, backend) |
| |
| # bad cipher |
| data = self.make_file(ciphername=b"unknown", kdfname=b"bcrypt") |
| with pytest.raises(UnsupportedAlgorithm): |
| load_ssh_private_key(data, None, backend) |
| |
| # bad magic |
| data = self.make_file(magic=b"unknown") |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| # too few keys |
| data = self.make_file(nkeys=0) |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| # too many keys |
| data = self.make_file(nkeys=2) |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| def test_ssh_errors_bad_values(self, backend): |
| # bad curve |
| data = self.make_file(pub_type=b"ecdsa-sha2-nistp444") |
| with pytest.raises(UnsupportedAlgorithm): |
| load_ssh_private_key(data, None, backend) |
| |
| # curve mismatch |
| data = self.make_file(priv_type=b"ecdsa-sha2-nistp384") |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| # invalid bigint |
| data = self.make_file( |
| priv_fields=(b"nistp256", b"\x04" * 65, b"\x80" * 32) |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| def test_ssh_errors_pubpriv_mismatch(self, backend): |
| # ecdsa public-private mismatch |
| data = self.make_file( |
| pub_fields=( |
| b"nistp256", |
| b"\x04" + b"\x05" * 64, |
| ) |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| # rsa public-private mismatch |
| data = self.make_file( |
| pub_type=b"ssh-rsa", |
| pub_fields=(b"x" * 32,) * 2, |
| priv_fields=(b"z" * 32,) * 6, |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| # dsa public-private mismatch |
| data = self.make_file( |
| pub_type=b"ssh-dss", |
| pub_fields=(b"x" * 32,) * 4, |
| priv_fields=(b"z" * 32,) * 5, |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| # ed25519 public-private mismatch |
| sk = b"x" * 32 |
| pk1 = b"y" * 32 |
| pk2 = b"z" * 32 |
| data = self.make_file( |
| pub_type=b"ssh-ed25519", |
| pub_fields=(pk1,), |
| priv_fields=( |
| pk1, |
| sk + pk2, |
| ), |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| data = self.make_file( |
| pub_type=b"ssh-ed25519", |
| pub_fields=(pk1,), |
| priv_fields=( |
| pk2, |
| sk + pk1, |
| ), |
| ) |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| def test_ssh_errors_bad_wrapper(self, backend): |
| # wrong header |
| data = self.make_file(header=b"-----BEGIN RSA PRIVATE KEY-----\n") |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| # wring footer |
| data = self.make_file(footer=b"-----END RSA PRIVATE KEY-----\n") |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| def test_ssh_no_padding(self, backend): |
| # no padding must work, if data is on block boundary |
| data = self.make_file(pad=b"", comment=b"") |
| key = load_ssh_private_key(data, None, backend) |
| assert isinstance(key, ec.EllipticCurvePrivateKey) |
| |
| # no padding with right last byte |
| data = self.make_file(pad=b"", comment=b"\x08" * 8) |
| key = load_ssh_private_key(data, None, backend) |
| assert isinstance(key, ec.EllipticCurvePrivateKey) |
| |
| # avoid unexpected padding removal |
| data = self.make_file(pad=b"", comment=b"1234\x01\x02\x03\x04") |
| key = load_ssh_private_key(data, None, backend) |
| assert isinstance(key, ec.EllipticCurvePrivateKey) |
| |
| # bad padding with right size |
| data = self.make_file(pad=b"\x08" * 8, comment=b"") |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| def test_ssh_errors_bad_secrets(self, backend): |
| # checkval mismatch |
| data = self.make_file(checkval2=b"4321") |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| # bad padding, correct=1 |
| data = self.make_file(pad=b"\x01\x02") |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| data = self.make_file(pad=b"") |
| with pytest.raises(ValueError): |
| load_ssh_private_key(data, None, backend) |
| |
| @pytest.mark.supported( |
| only_if=lambda backend: backend.elliptic_curve_supported( |
| ec.SECP192R1() |
| ), |
| skip_message="Requires backend support for ec.SECP192R1", |
| ) |
| def test_serialize_ssh_private_key_errors_bad_curve(self, backend): |
| private_key = ec.generate_private_key(ec.SECP192R1(), backend) |
| with pytest.raises(ValueError): |
| private_key.private_bytes( |
| Encoding.PEM, PrivateFormat.OpenSSH, NoEncryption() |
| ) |
| |
| def test_serialize_ssh_private_key_errors(self, backend): |
| # bad encoding |
| private_key = ec.generate_private_key(ec.SECP256R1(), backend) |
| with pytest.raises(ValueError): |
| private_key.private_bytes( |
| Encoding.DER, PrivateFormat.OpenSSH, NoEncryption() |
| ) |
| |
| # bad object type |
| with pytest.raises(ValueError): |
| ssh.serialize_ssh_private_key(object(), None) |
| |
| private_key = ec.generate_private_key(ec.SECP256R1(), backend) |
| |
| # too long password |
| with pytest.raises(ValueError): |
| private_key.private_bytes( |
| Encoding.PEM, |
| PrivateFormat.OpenSSH, |
| BestAvailableEncryption(b"p" * 73), |
| ) |
| |
| # unknown encryption class |
| with pytest.raises(ValueError): |
| private_key.private_bytes( |
| Encoding.PEM, |
| PrivateFormat.OpenSSH, |
| DummyKeySerializationEncryption(), |
| ) |
| |
| @pytest.mark.parametrize( |
| ("key_path", "supported"), |
| [ |
| (["Traditional_OpenSSL_Serialization", "dsa.1024.pem"], True), |
| (["Traditional_OpenSSL_Serialization", "dsa.2048.pem"], False), |
| (["Traditional_OpenSSL_Serialization", "dsa.3072.pem"], False), |
| ], |
| ) |
| def test_dsa_private_key_sizes(self, key_path, supported, backend): |
| key = load_vectors_from_file( |
| os.path.join("asymmetric", *key_path), |
| lambda pemfile: load_pem_private_key( |
| pemfile.read(), None, backend |
| ), |
| mode="rb", |
| ) |
| assert isinstance(key, dsa.DSAPrivateKey) |
| if supported: |
| res = key.private_bytes( |
| Encoding.PEM, PrivateFormat.OpenSSH, NoEncryption() |
| ) |
| assert isinstance(res, bytes) |
| else: |
| with pytest.raises(ValueError): |
| key.private_bytes( |
| Encoding.PEM, PrivateFormat.OpenSSH, NoEncryption() |
| ) |
