Support for copy.deepcopy keys (#13759)

* fix: add deep copy for rsa keys

* fix: add deep copy for dh keys

* fix: add deep copy for ec keys

* fix: add deep copy for ed25519 keys

* fix: add deep copy for ed448 keys

* fix: add deep copy for x25519 keys

* fix: add deep copy for x448 keys

* fix: forgot skip test when x448 not available

* chore: removed __eq__ from dsa because unrelated + fmt on x448 + added deepcopy on example clouhsm

* fix: replaced all deepcopy implementations by self

* fix: modified tests for coverage

* chore: format test x448

* chore: removed cast for key in test_ec

* chore: replaced key generation in tests by pre generated keys

Signed-off-by: Courtcircuits <[email protected]>

* chore: replaced PyDict by PyAny in backend files for deepcopy

Signed-off-by: Courtcircuits <[email protected]>

---------

Signed-off-by: Courtcircuits <[email protected]>

Author: Courtcircuits

docs/hazmat/primitives/asymmetric/cloudhsm.rst

@@ -122,6 +122,9 @@ if you only need a subset of functionality.
     ...     def __copy__(self) -> "CloudRSAPrivateKey":
     ...         return self
     ...
+    ...     def __deepcopy__(self, memodict: dict) -> "CloudRSAPrivateKey":
+    ...         return self
+    ...
     >>> cloud_private_key = CloudRSAPrivateKey("creds", "key_id")
     >>> sig = cloud_private_key.sign(b"message", PKCS1v15(), hashes.SHA256())
     >>> isinstance(sig, bytes)

src/cryptography/hazmat/primitives/asymmetric/dh.py

@@ -87,6 +87,12 @@ def __copy__(self) -> DHPublicKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> DHPublicKey:
+        """
+        Returns a deep copy.
+        """
+
 
 DHPublicKeyWithSerialization = DHPublicKey
 DHPublicKey.register(rust_openssl.dh.DHPublicKey)
@@ -142,6 +148,12 @@ def __copy__(self) -> DHPrivateKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> DHPrivateKey:
+        """
+        Returns a deep copy.
+        """
+
 
 DHPrivateKeyWithSerialization = DHPrivateKey
 DHPrivateKey.register(rust_openssl.dh.DHPrivateKey)

src/cryptography/hazmat/primitives/asymmetric/dsa.py

@@ -84,6 +84,12 @@ def __copy__(self) -> DSAPrivateKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> DSAPrivateKey:
+        """
+        Returns a deep copy.
+        """
+
 
 DSAPrivateKeyWithSerialization = DSAPrivateKey
 DSAPrivateKey.register(rust_openssl.dsa.DSAPrivateKey)
@@ -142,6 +148,12 @@ def __copy__(self) -> DSAPublicKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> DSAPublicKey:
+        """
+        Returns a deep copy.
+        """
+
 
 DSAPublicKeyWithSerialization = DSAPublicKey
 DSAPublicKey.register(rust_openssl.dsa.DSAPublicKey)

src/cryptography/hazmat/primitives/asymmetric/ec.py

@@ -134,6 +134,12 @@ def __copy__(self) -> EllipticCurvePrivateKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> EllipticCurvePrivateKey:
+        """
+        Returns a deep copy.
+        """
+
 
 EllipticCurvePrivateKeyWithSerialization = EllipticCurvePrivateKey
 EllipticCurvePrivateKey.register(rust_openssl.ec.ECPrivateKey)
@@ -207,6 +213,12 @@ def __copy__(self) -> EllipticCurvePublicKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> EllipticCurvePublicKey:
+        """
+        Returns a deep copy.
+        """
+
 
 EllipticCurvePublicKeyWithSerialization = EllipticCurvePublicKey
 EllipticCurvePublicKey.register(rust_openssl.ec.ECPublicKey)

src/cryptography/hazmat/primitives/asymmetric/ed25519.py

@@ -60,6 +60,12 @@ def __copy__(self) -> Ed25519PublicKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> Ed25519PublicKey:
+        """
+        Returns a deep copy.
+        """
+
 
 Ed25519PublicKey.register(rust_openssl.ed25519.Ed25519PublicKey)
 
@@ -125,5 +131,11 @@ def __copy__(self) -> Ed25519PrivateKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> Ed25519PrivateKey:
+        """
+        Returns a deep copy.
+        """
+
 
 Ed25519PrivateKey.register(rust_openssl.ed25519.Ed25519PrivateKey)

src/cryptography/hazmat/primitives/asymmetric/ed448.py

@@ -60,6 +60,12 @@ def __copy__(self) -> Ed448PublicKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> Ed448PublicKey:
+        """
+        Returns a deep copy.
+        """
+
 
 if hasattr(rust_openssl, "ed448"):
     Ed448PublicKey.register(rust_openssl.ed448.Ed448PublicKey)
@@ -126,6 +132,12 @@ def __copy__(self) -> Ed448PrivateKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> Ed448PrivateKey:
+        """
+        Returns a deep copy.
+        """
+
 
 if hasattr(rust_openssl, "x448"):
     Ed448PrivateKey.register(rust_openssl.ed448.Ed448PrivateKey)

src/cryptography/hazmat/primitives/asymmetric/rsa.py

@@ -69,6 +69,12 @@ def __copy__(self) -> RSAPrivateKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> RSAPrivateKey:
+        """
+        Returns a deep copy.
+        """
+
 
 RSAPrivateKeyWithSerialization = RSAPrivateKey
 RSAPrivateKey.register(rust_openssl.rsa.RSAPrivateKey)
@@ -139,6 +145,12 @@ def __copy__(self) -> RSAPublicKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> RSAPublicKey:
+        """
+        Returns a deep copy.
+        """
+
 
 RSAPublicKeyWithSerialization = RSAPublicKey
 RSAPublicKey.register(rust_openssl.rsa.RSAPublicKey)

src/cryptography/hazmat/primitives/asymmetric/x25519.py

@@ -54,6 +54,12 @@ def __copy__(self) -> X25519PublicKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> X25519PublicKey:
+        """
+        Returns a deep copy.
+        """
+
 
 X25519PublicKey.register(rust_openssl.x25519.X25519PublicKey)
 
@@ -118,5 +124,11 @@ def __copy__(self) -> X25519PrivateKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> X25519PrivateKey:
+        """
+        Returns a deep copy.
+        """
+
 
 X25519PrivateKey.register(rust_openssl.x25519.X25519PrivateKey)

src/cryptography/hazmat/primitives/asymmetric/x448.py

@@ -54,6 +54,12 @@ def __copy__(self) -> X448PublicKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> X448PublicKey:
+        """
+        Returns a deep copy.
+        """
+
 
 if hasattr(rust_openssl, "x448"):
     X448PublicKey.register(rust_openssl.x448.X448PublicKey)
@@ -120,6 +126,12 @@ def __copy__(self) -> X448PrivateKey:
         Returns a copy.
         """
 
+    @abc.abstractmethod
+    def __deepcopy__(self) -> X448PrivateKey:
+        """
+        Returns a deep copy.
+        """
+
 
 if hasattr(rust_openssl, "x448"):
     X448PrivateKey.register(rust_openssl.x448.X448PrivateKey)

src/rust/src/backend/dh.rs

@@ -247,6 +247,13 @@ impl DHPrivateKey {
     fn __copy__(slf: pyo3::PyRef<'_, Self>) -> pyo3::PyRef<'_, Self> {
         slf
     }
+
+    fn __deepcopy__<'p>(
+        slf: pyo3::PyRef<'p, Self>,
+        _memo: &pyo3::Bound<'p, pyo3::PyAny>,
+    ) -> pyo3::PyRef<'p, Self> {
+        slf
+    }
 }
 
 #[pyo3::pymethods]
@@ -312,6 +319,13 @@ impl DHPublicKey {
     fn __copy__(slf: pyo3::PyRef<'_, Self>) -> pyo3::PyRef<'_, Self> {
         slf
     }
+
+    fn __deepcopy__<'p>(
+        slf: pyo3::PyRef<'p, Self>,
+        _memo: &pyo3::Bound<'p, pyo3::types::PyAny>,
+    ) -> pyo3::PyRef<'p, Self> {
+        slf
+    }
 }
 
 #[pyo3::pymethods]