From f6f1e8afdf4c476734e09b30d5eb9f86cb2c5309 Mon Sep 17 00:00:00 2001 From: Erik Larsson Date: Sun, 18 May 2025 14:01:20 +0200 Subject: [PATCH] cryptograpy: add copy dunder for private keys cryptography >= 45.0.0 requires the copy dunder for private key implementations. Signed-off-by: Erik Larsson --- src/tpm2_pytss/cryptography.py | 12 ++++++++++++ test/test_cryptography.py | 19 +++++++++++++++++++ 2 files changed, 31 insertions(+) diff --git a/src/tpm2_pytss/cryptography.py b/src/tpm2_pytss/cryptography.py index dd75623a..5b8432cc 100644 --- a/src/tpm2_pytss/cryptography.py +++ b/src/tpm2_pytss/cryptography.py @@ -257,6 +257,12 @@ def private_bytes( """Always raises a NotImplementedError.""" raise NotImplementedError() + def __copy__(self) -> "tpm_rsa_private_key": + """Returns a shallow copy of the private key.""" + return tpm_rsa_private_key( + ectx=self._ectx, handle=self._handle, session=self._session + ) + class tpm_ecc_private_key(ec.EllipticCurvePrivateKey): """Interface to a TPM ECC key for use with the cryptography module. @@ -428,3 +434,9 @@ def private_bytes( ) -> None: """Always raises a NotImplementedError.""" raise NotImplementedError() + + def __copy__(self) -> "tpm_ecc_private_key": + """Returns a shallow copy of the private key.""" + return tpm_ecc_private_key( + ectx=self._ectx, handle=self._handle, session=self._session + ) diff --git a/test/test_cryptography.py b/test/test_cryptography.py index 3f81f6a6..3e4f30d4 100644 --- a/test/test_cryptography.py +++ b/test/test_cryptography.py @@ -11,6 +11,7 @@ from cryptography.hazmat.primitives.asymmetric.utils import Prehashed from cryptography import x509 import datetime +import copy rsa_template = TPM2B_PUBLIC.parse( @@ -502,3 +503,21 @@ def test_csr_builder_ecc(self): halg = privkey.get_digest_algorithm() csr = builder.sign(privkey, algorithm=halg()) self.assertEqual(csr.is_signature_valid, True) + + def test_rsa_copy(self): + handle, _, _, _, _ = self.ectx.create_primary( + in_sensitive=None, in_public=rsa_template + ) + privkey = tpm_rsa_private_key(self.ectx, handle) + privkey_copy = copy.copy(privkey) + + self.assertEqual(privkey.key_size, privkey_copy.key_size) + + def test_ecc_copy(self): + handle, _, _, _, _ = self.ectx.create_primary( + in_sensitive=None, in_public=ecc_template + ) + privkey = tpm_ecc_private_key(self.ectx, handle) + privkey_copy = copy.copy(privkey) + + self.assertEqual(type(privkey.curve), type(privkey_copy.curve))