diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/hazmat/backends/test_openssl.py | 38 | ||||
| -rw-r--r-- | tests/hazmat/primitives/test_3des.py | 37 | ||||
| -rw-r--r-- | tests/hazmat/primitives/test_rsa.py | 190 | ||||
| -rw-r--r-- | tests/hazmat/primitives/test_serialization.py | 52 | ||||
| -rw-r--r-- | tests/hazmat/primitives/utils.py | 32 | 
5 files changed, 188 insertions, 161 deletions
diff --git a/tests/hazmat/backends/test_openssl.py b/tests/hazmat/backends/test_openssl.py index aa2122fb..75369efc 100644 --- a/tests/hazmat/backends/test_openssl.py +++ b/tests/hazmat/backends/test_openssl.py @@ -13,6 +13,10 @@  from __future__ import absolute_import, division, print_function +import subprocess +import sys +import textwrap +  import pretend  import pytest @@ -212,14 +216,32 @@ class TestOpenSSLRandomEngine(object):          name = backend._lib.ENGINE_get_name(current_default)          assert name == backend._lib.Cryptography_osrandom_engine_name -    # This must be the first test in the class so that the teardown method -    # has not (potentially) altered the default engine. -    def test_osrandom_engine_is_default(self): -        e = backend._lib.ENGINE_get_default_RAND() -        name = backend._lib.ENGINE_get_name(e) -        assert name == backend._lib.Cryptography_osrandom_engine_name -        res = backend._lib.ENGINE_free(e) -        assert res == 1 +    def test_osrandom_engine_is_default(self, tmpdir): +        engine_printer = textwrap.dedent( +            """ +            import sys +            from cryptography.hazmat.backends.openssl.backend import backend + +            e = backend._lib.ENGINE_get_default_RAND() +            name = backend._lib.ENGINE_get_name(e) +            sys.stdout.write(backend._ffi.string(name).decode('ascii')) +            res = backend._lib.ENGINE_free(e) +            assert res == 1 +            """ +        ) +        engine_name = tmpdir.join('engine_name') + +        with engine_name.open('w') as out: +            subprocess.check_call( +                [sys.executable, "-c", engine_printer], +                stdout=out +            ) + +        osrandom_engine_name = backend._ffi.string( +            backend._lib.Cryptography_osrandom_engine_name +        ) + +        assert engine_name.read().encode('ascii') == osrandom_engine_name      def test_osrandom_sanity_check(self):          # This test serves as a check against catastrophic failure. diff --git a/tests/hazmat/primitives/test_3des.py b/tests/hazmat/primitives/test_3des.py index b9354f0e..65660386 100644 --- a/tests/hazmat/primitives/test_3des.py +++ b/tests/hazmat/primitives/test_3des.py @@ -174,3 +174,40 @@ class TestTripleDESModeCFB8(object):          ),          lambda iv, **kwargs: modes.CFB8(binascii.unhexlify(iv)),      ) + + +@pytest.mark.supported( +    only_if=lambda backend: backend.cipher_supported( +        algorithms.TripleDES("\x00" * 8), modes.ECB() +    ), +    skip_message="Does not support TripleDES ECB", +) +@pytest.mark.cipher +class TestTripleDESModeECB(object): +    test_KAT = generate_encrypt_test( +        load_nist_vectors, +        os.path.join("ciphers", "3DES", "ECB"), +        [ +            "TECBinvperm.rsp", +            "TECBpermop.rsp", +            "TECBsubtab.rsp", +            "TECBvarkey.rsp", +            "TECBvartext.rsp", +        ], +        lambda keys, **kwargs: algorithms.TripleDES(binascii.unhexlify(keys)), +        lambda **kwargs: modes.ECB(), +    ) + +    test_MMT = generate_encrypt_test( +        load_nist_vectors, +        os.path.join("ciphers", "3DES", "ECB"), +        [ +            "TECBMMT1.rsp", +            "TECBMMT2.rsp", +            "TECBMMT3.rsp", +        ], +        lambda key1, key2, key3, **kwargs: algorithms.TripleDES( +            binascii.unhexlify(key1 + key2 + key3) +        ), +        lambda **kwargs: modes.ECB(), +    ) diff --git a/tests/hazmat/primitives/test_rsa.py b/tests/hazmat/primitives/test_rsa.py index 8f10fb10..ffaaf587 100644 --- a/tests/hazmat/primitives/test_rsa.py +++ b/tests/hazmat/primitives/test_rsa.py @@ -38,7 +38,7 @@ from .fixtures_rsa import (      RSA_KEY_768,  )  from .utils import ( -    _check_rsa_private_key, generate_rsa_verification_test +    _check_rsa_private_numbers, generate_rsa_verification_test  )  from ...utils import (      load_pkcs1_vectors, load_rsa_nist_vectors, load_vectors_from_file, @@ -98,13 +98,13 @@ class TestRSA(object):      )      def test_generate_rsa_keys(self, backend, public_exponent, key_size):          skey = rsa.generate_private_key(public_exponent, key_size, backend) -        _check_rsa_private_key(skey)          assert skey.key_size == key_size -        assert skey.public_exponent == public_exponent + +        if isinstance(skey, interfaces.RSAPrivateKeyWithNumbers): +            _check_rsa_private_numbers(skey.private_numbers())      def test_generate_rsa_key_class_method(self, backend):          skey = rsa.RSAPrivateKey.generate(65537, 512, backend) -        _check_rsa_private_key(skey)          assert skey.key_size == 512          assert skey.public_exponent == 65537 @@ -141,43 +141,31 @@ class TestRSA(object):      def test_load_pss_vect_example_keys(self, pkcs1_example):          secret, public = pkcs1_example -        skey = rsa.RSAPrivateKey( +        private_num = rsa.RSAPrivateNumbers(              p=secret["p"],              q=secret["q"], -            private_exponent=secret["private_exponent"], +            d=secret["private_exponent"],              dmp1=secret["dmp1"],              dmq1=secret["dmq1"],              iqmp=secret["iqmp"], -            public_exponent=secret["public_exponent"], -            modulus=secret["modulus"] +            public_numbers=rsa.RSAPublicNumbers( +                e=secret["public_exponent"], +                n=secret["modulus"] +            )          ) -        assert skey -        _check_rsa_private_key(skey) +        _check_rsa_private_numbers(private_num) -        pkey = rsa.RSAPublicKey( -            public_exponent=public["public_exponent"], -            modulus=public["modulus"] +        public_num = rsa.RSAPublicNumbers( +            e=public["public_exponent"], +            n=public["modulus"]          ) -        assert pkey - -        pkey2 = skey.public_key() -        assert pkey2 +        assert public_num -        assert skey.modulus == pkey.modulus -        assert skey.modulus == skey.n -        assert skey.public_exponent == pkey.public_exponent -        assert skey.public_exponent == skey.e -        assert skey.private_exponent == skey.d +        public_num2 = private_num.public_numbers +        assert public_num2 -        assert pkey.modulus -        assert pkey.modulus == pkey2.modulus -        assert pkey.modulus == pkey.n -        assert pkey.public_exponent == pkey2.public_exponent -        assert pkey.public_exponent == pkey.e - -        assert skey.key_size -        assert skey.key_size == pkey.key_size -        assert skey.key_size == pkey2.key_size +        assert public_num.n == public_num2.n +        assert public_num.e == public_num2.e      def test_invalid_private_key_argument_types(self):          with pytest.raises(TypeError): @@ -504,8 +492,7 @@ class TestRSASignature(object):                      salt_length=padding.MGF1.MAX_LENGTH                  )              ), -            hashes.SHA1(), -            backend +            hashes.SHA1()          )          signer.update(b"so deprecated")          signature = signer.finalize() @@ -520,8 +507,7 @@ class TestRSASignature(object):                      salt_length=padding.MGF1.MAX_LENGTH                  )              ), -            hashes.SHA1(), -            backend +            hashes.SHA1()          )          verifier.update(b"so deprecated")          verifier.verify() @@ -546,19 +532,10 @@ class TestRSASignature(object):              mgf=padding.MGF1(hash_alg),              salt_length=padding.PSS.MAX_LENGTH          ) -        signer = private_key.signer( -            pss, -            hash_alg, -            backend -        ) +        signer = private_key.signer(pss, hash_alg)          signer.update(b"testing signature")          signature = signer.finalize() -        verifier = public_key.verifier( -            signature, -            pss, -            hash_alg, -            backend -        ) +        verifier = public_key.verifier(signature, pss, hash_alg)          verifier.update(b"testing signature")          verifier.verify() @@ -581,8 +558,7 @@ class TestRSASignature(object):                  mgf=padding.MGF1(hashes.SHA1()),                  salt_length=padding.PSS.MAX_LENGTH              ), -            hashes.SHA512(), -            backend +            hashes.SHA512()          )          signer.update(b"no failure")          signer.finalize() @@ -608,8 +584,7 @@ class TestRSASignature(object):                      mgf=padding.MGF1(hashes.SHA1()),                      salt_length=padding.PSS.MAX_LENGTH                  ), -                hashes.SHA512(), -                backend +                hashes.SHA512()              )      @pytest.mark.supported( @@ -628,8 +603,7 @@ class TestRSASignature(object):                  mgf=padding.MGF1(hashes.SHA1()),                  salt_length=1000000              ), -            hashes.SHA1(), -            backend +            hashes.SHA1()          )          signer.update(b"failure coming")          with pytest.raises(ValueError): @@ -643,7 +617,7 @@ class TestRSASignature(object):      )      def test_use_after_finalize(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend) -        signer = private_key.signer(padding.PKCS1v15(), hashes.SHA1(), backend) +        signer = private_key.signer(padding.PKCS1v15(), hashes.SHA1())          signer.update(b"sign me")          signer.finalize()          with pytest.raises(AlreadyFinalized): @@ -654,16 +628,25 @@ class TestRSASignature(object):      def test_unsupported_padding(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend)          with raises_unsupported_algorithm(_Reasons.UNSUPPORTED_PADDING): -            private_key.signer(DummyPadding(), hashes.SHA1(), backend) +            private_key.signer(DummyPadding(), hashes.SHA1())      def test_padding_incorrect_type(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend)          with pytest.raises(TypeError): -            private_key.signer("notpadding", hashes.SHA1(), backend) +            private_key.signer("notpadding", hashes.SHA1())      def test_rsa_signer_invalid_backend(self, backend):          pretend_backend = object() -        private_key = load_rsa_private_numbers(RSA_KEY_2048, backend) +        private_key = rsa.RSAPrivateKey( +            p=RSA_KEY_512.p, +            q=RSA_KEY_512.q, +            private_exponent=RSA_KEY_512.d, +            dmp1=RSA_KEY_512.dmp1, +            dmq1=RSA_KEY_512.dmq1, +            iqmp=RSA_KEY_512.iqmp, +            public_exponent=RSA_KEY_512.public_numbers.e, +            modulus=RSA_KEY_512.public_numbers.n +        )          with raises_unsupported_algorithm(_Reasons.BACKEND_MISSING_INTERFACE):              private_key.signer( @@ -678,8 +661,7 @@ class TestRSASignature(object):      def test_unsupported_pss_mgf(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend)          with raises_unsupported_algorithm(_Reasons.UNSUPPORTED_MGF): -            private_key.signer(padding.PSS(mgf=DummyMGF()), hashes.SHA1(), -                               backend) +            private_key.signer(padding.PSS(mgf=DummyMGF()), hashes.SHA1())      @pytest.mark.supported(          only_if=lambda backend: backend.rsa_padding_supported( @@ -691,8 +673,7 @@ class TestRSASignature(object):          private_key = load_rsa_private_numbers(RSA_KEY_599, backend)          signer = private_key.signer(              padding.PKCS1v15(), -            hashes.SHA512(), -            backend +            hashes.SHA512()          )          signer.update(b"failure coming")          with pytest.raises(ValueError): @@ -708,8 +689,7 @@ class TestRSASignature(object):          private_key = load_rsa_private_numbers(RSA_KEY_745, backend)          signer = private_key.signer(              padding.PKCS1v15(), -            hashes.SHA512(), -            backend +            hashes.SHA512()          )          signer.update(b"no failure")          signer.finalize() @@ -755,14 +735,13 @@ class TestRSAVerification(object):      def test_invalid_pkcs1v15_signature_wrong_data(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend)          public_key = private_key.public_key() -        signer = private_key.signer(padding.PKCS1v15(), hashes.SHA1(), backend) +        signer = private_key.signer(padding.PKCS1v15(), hashes.SHA1())          signer.update(b"sign me")          signature = signer.finalize()          verifier = public_key.verifier(              signature,              padding.PKCS1v15(), -            hashes.SHA1(), -            backend +            hashes.SHA1()          )          verifier.update(b"incorrect data")          with pytest.raises(InvalidSignature): @@ -778,14 +757,13 @@ class TestRSAVerification(object):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend)          private_key2 = load_rsa_private_numbers(RSA_KEY_512_ALT, backend)          public_key = private_key2.public_key() -        signer = private_key.signer(padding.PKCS1v15(), hashes.SHA1(), backend) +        signer = private_key.signer(padding.PKCS1v15(), hashes.SHA1())          signer.update(b"sign me")          signature = signer.finalize()          verifier = public_key.verifier(              signature,              padding.PKCS1v15(), -            hashes.SHA1(), -            backend +            hashes.SHA1()          )          verifier.update(b"sign me")          with pytest.raises(InvalidSignature): @@ -944,15 +922,14 @@ class TestRSAVerification(object):      def test_use_after_finalize(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend)          public_key = private_key.public_key() -        signer = private_key.signer(padding.PKCS1v15(), hashes.SHA1(), backend) +        signer = private_key.signer(padding.PKCS1v15(), hashes.SHA1())          signer.update(b"sign me")          signature = signer.finalize()          verifier = public_key.verifier(              signature,              padding.PKCS1v15(), -            hashes.SHA1(), -            backend +            hashes.SHA1()          )          verifier.update(b"sign me")          verifier.verify() @@ -965,17 +942,17 @@ class TestRSAVerification(object):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend)          public_key = private_key.public_key()          with raises_unsupported_algorithm(_Reasons.UNSUPPORTED_PADDING): -            public_key.verifier(b"sig", DummyPadding(), hashes.SHA1(), backend) +            public_key.verifier(b"sig", DummyPadding(), hashes.SHA1())      def test_padding_incorrect_type(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend)          public_key = private_key.public_key()          with pytest.raises(TypeError): -            public_key.verifier(b"sig", "notpadding", hashes.SHA1(), backend) +            public_key.verifier(b"sig", "notpadding", hashes.SHA1())      def test_rsa_verifier_invalid_backend(self, backend):          pretend_backend = object() -        private_key = rsa.generate_private_key(65537, 2048, backend) +        private_key = rsa.RSAPrivateKey.generate(65537, 2048, backend)          public_key = private_key.public_key()          with raises_unsupported_algorithm(_Reasons.BACKEND_MISSING_INTERFACE): @@ -993,7 +970,7 @@ class TestRSAVerification(object):          public_key = private_key.public_key()          with raises_unsupported_algorithm(_Reasons.UNSUPPORTED_MGF):              public_key.verifier(b"sig", padding.PSS(mgf=DummyMGF()), -                                hashes.SHA1(), backend) +                                hashes.SHA1())      @pytest.mark.supported(          only_if=lambda backend: backend.rsa_padding_supported( @@ -1022,8 +999,7 @@ class TestRSAVerification(object):                      mgf=padding.MGF1(algorithm=hashes.SHA1()),                      salt_length=padding.PSS.MAX_LENGTH                  ), -                hashes.SHA512(), -                backend +                hashes.SHA512()              )      @pytest.mark.supported( @@ -1413,7 +1389,7 @@ class TestRSADecryption(object):      def test_unsupported_padding(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend)          with raises_unsupported_algorithm(_Reasons.UNSUPPORTED_PADDING): -            private_key.decrypt(b"0" * 64, DummyPadding(), backend) +            private_key.decrypt(b"0" * 64, DummyPadding())      @pytest.mark.supported(          only_if=lambda backend: backend.rsa_padding_supported( @@ -1426,8 +1402,7 @@ class TestRSADecryption(object):          with pytest.raises(ValueError):              private_key.decrypt(                  b"\x00" * 64, -                padding.PKCS1v15(), -                backend +                padding.PKCS1v15()              )      @pytest.mark.supported( @@ -1441,8 +1416,7 @@ class TestRSADecryption(object):          with pytest.raises(ValueError):              private_key.decrypt(                  b"\x00" * 65, -                padding.PKCS1v15(), -                backend +                padding.PKCS1v15()              )      @pytest.mark.supported( @@ -1460,8 +1434,7 @@ class TestRSADecryption(object):          with pytest.raises(ValueError):              private_key.decrypt(                  ct, -                padding.PKCS1v15(), -                backend +                padding.PKCS1v15()              )      def test_rsa_decrypt_invalid_backend(self, backend): @@ -1525,8 +1498,7 @@ class TestRSADecryption(object):                      mgf=DummyMGF(),                      algorithm=hashes.SHA1(),                      label=None -                ), -                backend +                )              ) @@ -1558,21 +1530,22 @@ class TestRSAEncryption(object):          )      )      def test_rsa_encrypt_oaep(self, key_data, pad, backend): -        private_key = load_rsa_private_numbers(key_data, backend) +        private_key = rsa.RSAPrivateKey( +            p=key_data.p, +            q=key_data.q, +            private_exponent=key_data.d, +            dmp1=key_data.dmp1, +            dmq1=key_data.dmq1, +            iqmp=key_data.iqmp, +            public_exponent=key_data.public_numbers.e, +            modulus=key_data.public_numbers.n +        )          pt = b"encrypt me!"          public_key = private_key.public_key() -        ct = public_key.encrypt( -            pt, -            pad, -            backend -        ) +        ct = public_key.encrypt(pt, pad, backend)          assert ct != pt          assert len(ct) == math.ceil(public_key.key_size / 8.0) -        recovered_pt = private_key.decrypt( -            ct, -            pad, -            backend -        ) +        recovered_pt = private_key.decrypt(ct, pad, backend)          assert recovered_pt == pt      @pytest.mark.supported( @@ -1594,18 +1567,10 @@ class TestRSAEncryption(object):          private_key = load_rsa_private_numbers(key_data, backend)          pt = b"encrypt me!"          public_key = private_key.public_key() -        ct = public_key.encrypt( -            pt, -            pad, -            backend -        ) +        ct = public_key.encrypt(pt, pad)          assert ct != pt          assert len(ct) == math.ceil(public_key.key_size / 8.0) -        recovered_pt = private_key.decrypt( -            ct, -            pad, -            backend -        ) +        recovered_pt = private_key.decrypt(ct, pad)          assert recovered_pt == pt      @pytest.mark.parametrize( @@ -1631,16 +1596,14 @@ class TestRSAEncryption(object):          with pytest.raises(ValueError):              public_key.encrypt(                  b"\x00" * (private_key.key_size // 8 - 1), -                pad, -                backend +                pad              )          # Larger than the key size.          with pytest.raises(ValueError):              public_key.encrypt(                  b"\x00" * (private_key.key_size // 8 + 5), -                pad, -                backend +                pad              )      def test_rsa_encrypt_invalid_backend(self, backend): @@ -1660,7 +1623,7 @@ class TestRSAEncryption(object):          public_key = private_key.public_key()          with raises_unsupported_algorithm(_Reasons.UNSUPPORTED_PADDING): -            public_key.encrypt(b"somedata", DummyPadding(), backend) +            public_key.encrypt(b"somedata", DummyPadding())      def test_unsupported_oaep_mgf(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_512, backend) @@ -1673,8 +1636,7 @@ class TestRSAEncryption(object):                      mgf=DummyMGF(),                      algorithm=hashes.SHA1(),                      label=None -                ), -                backend +                )              ) diff --git a/tests/hazmat/primitives/test_serialization.py b/tests/hazmat/primitives/test_serialization.py index 4aa9707f..ae990b64 100644 --- a/tests/hazmat/primitives/test_serialization.py +++ b/tests/hazmat/primitives/test_serialization.py @@ -20,7 +20,8 @@ import textwrap  import pytest  from cryptography.exceptions import _Reasons -from cryptography.hazmat.primitives.asymmetric import dsa, rsa +from cryptography.hazmat.primitives import interfaces +from cryptography.hazmat.primitives.asymmetric import dsa  from cryptography.hazmat.primitives.serialization import (      load_pem_pkcs8_private_key,      load_pem_traditional_openssl_private_key, @@ -29,7 +30,7 @@ from cryptography.hazmat.primitives.serialization import (  )  from .fixtures_rsa import RSA_KEY_1024 -from .utils import _check_rsa_private_key, load_vectors_from_file +from .utils import _check_rsa_private_numbers, load_vectors_from_file  from ...utils import raises_unsupported_algorithm @@ -54,8 +55,9 @@ class TestTraditionalOpenSSLSerialisation(object):          )          assert key -        assert isinstance(key, rsa.RSAPrivateKey) -        _check_rsa_private_key(key) +        assert isinstance(key, interfaces.RSAPrivateKey) +        if isinstance(key, interfaces.RSAPrivateKeyWithNumbers): +            _check_rsa_private_numbers(key.private_numbers())      @pytest.mark.parametrize(          ("key_file", "password"), @@ -87,33 +89,34 @@ class TestTraditionalOpenSSLSerialisation(object):          )          assert pkey -        assert pkey.p == int( +        numbers = pkey.private_numbers() +        assert numbers.p == int(              "fb7d316fc51531b36d93adaefaf52db6ad5beb793d37c4cf9dfc1ddd17cfbafb",              16          ) -        assert pkey.q == int( +        assert numbers.q == int(              "df98264e646de9a0fbeab094e31caad5bc7adceaaae3c800ca0275dd4bb307f5",              16          ) -        assert pkey.private_exponent == int( +        assert numbers.d == int(              "db4848c36f478dd5d38f35ae519643b6b810d404bcb76c00e44015e56ca1cab0"              "7bb7ae91f6b4b43fcfc82a47d7ed55b8c575152116994c2ce5325ec24313b911",              16          ) -        assert pkey.dmp1 == int( +        assert numbers.dmp1 == int(              "ce997f967192c2bcc3853186f1559fd355c190c58ddc15cbf5de9b6df954c727",              16          ) -        assert pkey.dmq1 == int( +        assert numbers.dmq1 == int(              "b018a57ab20ffaa3862435445d863369b852cf70a67c55058213e3fe10e3848d",              16          ) -        assert pkey.iqmp == int( +        assert numbers.iqmp == int(              "6a8d830616924f5cf2d1bc1973f97fde6b63e052222ac7be06aa2532d10bac76",              16          ) -        assert pkey.public_exponent == 65537 -        assert pkey.modulus == int( +        assert numbers.public_numbers.e == 65537 +        assert numbers.public_numbers.n == int(              "dba786074f2f0350ce1d99f5aed5b520cfe0deb5429ec8f2a88563763f566e77"              "9814b7c310e5326edae31198eed439b845dd2db99eaa60f5c16a43f4be6bcf37",              16 @@ -283,8 +286,9 @@ class TestPKCS8Serialisation(object):          )          assert key -        assert isinstance(key, rsa.RSAPrivateKey) -        _check_rsa_private_key(key) +        assert isinstance(key, interfaces.RSAPrivateKey) +        if isinstance(key, interfaces.RSAPrivateKeyWithNumbers): +            _check_rsa_private_numbers(key.private_numbers())      def test_unused_password(self, backend):          key_file = os.path.join( @@ -415,7 +419,9 @@ class TestPKCS8Serialisation(object):          )          assert pkey -        assert pkey.modulus == int( +        numbers = pkey.private_numbers() + +        assert numbers.public_numbers.n == int(              "00beec64d6db5760ac2fd4c971145641b9bd7f5c56558ece608795c79807"              "376a7fe5b19f95b35ca358ea5c8abd7ae051d49cd2f1e45969a1ae945460"              "3c14b278664a0e414ebc8913acb6203626985525e17a600611b028542dd0" @@ -423,9 +429,9 @@ class TestPKCS8Serialisation(object):              "dd68480567c99b1a57", 16          ) -        assert pkey.public_exponent == 65537 +        assert numbers.public_numbers.e == 65537 -        assert pkey.private_exponent == int( +        assert numbers.d == int(              "0cfe316e9dc6b8817f4fcfd5ae38a0886f68f773b8a6db4c9e6d8703c599"              "f3d9785c3a2c09e4c8090909fb3721e19a3009ec21221523a729265707a5"              "8f13063671c42a4096cad378ef2510cb59e23071489d8893ac4934dd149f" @@ -433,31 +439,31 @@ class TestPKCS8Serialisation(object):              "cf7d995688c86c81", 16          ) -        assert pkey.p == int( +        assert numbers.p == int(              "00db122ac857b2c0437d7616daa98e597bb75ca9ad3a47a70bec10c10036"              "03328794b225c8e3eee6ffd3fd6d2253d28e071fe27d629ab072faa14377"              "ce6118cb67", 16          ) -        assert pkey.q == int( +        assert numbers.q == int(              "00df1b8aa8506fcbbbb9d00257f2975e38b33d2698fd0f37e82d7ef38c56"              "f21b6ced63c825383782a7115cfcc093300987dbd2853b518d1c8f26382a"              "2d2586d391", 16          ) -        assert pkey.dmp1 == int( +        assert numbers.dmp1 == int(              "00be18aca13e60712fdf5daa85421eb10d86d654b269e1255656194fb0c4"              "2dd01a1070ea12c19f5c39e09587af02f7b1a1030d016a9ffabf3b36d699"              "ceaf38d9bf", 16          ) -        assert pkey.dmq1 == int( +        assert numbers.dmq1 == int(              "71aa8978f90a0c050744b77cf1263725b203ac9f730606d8ae1d289dce4a"              "28b8d534e9ea347aeb808c73107e583eb80c546d2bddadcdb3c82693a4c1"              "3d863451", 16          ) -        assert pkey.iqmp == int( +        assert numbers.iqmp == int(              "136b7b1afac6e6279f71b24217b7083485a5e827d156024609dae39d48a6"              "bdb55af2f062cc4a3b077434e6fffad5faa29a2b5dba2bed3e4621e478c0"              "97ccfe7f", 16 @@ -554,9 +560,11 @@ class TestLoadRSANumbers(object):      def test_load_private_numbers(self, backend):          private_key = load_rsa_private_numbers(RSA_KEY_1024, backend)          assert private_key +        assert private_key.private_numbers()      def test_load_public_numbers(self, backend):          public_key = load_rsa_public_numbers(              RSA_KEY_1024.public_numbers, backend          )          assert public_key +        assert public_key.public_numbers() diff --git a/tests/hazmat/primitives/utils.py b/tests/hazmat/primitives/utils.py index a496459b..54659aa9 100644 --- a/tests/hazmat/primitives/utils.py +++ b/tests/hazmat/primitives/utils.py @@ -28,6 +28,9 @@ from cryptography.hazmat.primitives.asymmetric import rsa  from cryptography.hazmat.primitives.ciphers import Cipher  from cryptography.hazmat.primitives.kdf.hkdf import HKDF, HKDFExpand  from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC +from cryptography.hazmat.primitives.serialization import ( +    load_rsa_public_numbers +)  from ...utils import load_vectors_from_file @@ -388,16 +391,16 @@ def generate_rsa_verification_test(param_loader, path, file_names, hash_alg,  def rsa_verification_test(backend, params, hash_alg, pad_factory): -    public_key = rsa.RSAPublicKey( -        public_exponent=params["public_exponent"], -        modulus=params["modulus"] +    public_numbers = rsa.RSAPublicNumbers( +        e=params["public_exponent"], +        n=params["modulus"]      ) +    public_key = load_rsa_public_numbers(public_numbers, backend)      pad = pad_factory(params, hash_alg)      verifier = public_key.verifier(          binascii.unhexlify(params["s"]),          pad, -        hash_alg, -        backend +        hash_alg      )      verifier.update(binascii.unhexlify(params["msg"]))      if params["fail"]: @@ -407,19 +410,14 @@ def rsa_verification_test(backend, params, hash_alg, pad_factory):          verifier.verify() -def _check_rsa_private_key(skey): +def _check_rsa_private_numbers(skey):      assert skey -    assert skey.modulus -    assert skey.public_exponent -    assert skey.private_exponent -    assert skey.p * skey.q == skey.modulus -    assert skey.key_size +    pkey = skey.public_numbers +    assert pkey +    assert pkey.e +    assert pkey.n +    assert skey.d +    assert skey.p * skey.q == pkey.n      assert skey.dmp1 == rsa.rsa_crt_dmp1(skey.d, skey.p)      assert skey.dmq1 == rsa.rsa_crt_dmq1(skey.d, skey.q)      assert skey.iqmp == rsa.rsa_crt_iqmp(skey.p, skey.q) - -    pkey = skey.public_key() -    assert pkey -    assert skey.modulus == pkey.modulus -    assert skey.public_exponent == pkey.public_exponent -    assert skey.key_size == pkey.key_size  | 
