crypto/noise: Generate keypair only for Curve25519

Signed-off-by: Alexandru Vasile <[email protected]>
paritytech · Aug 21, 2024 · dd0659d · dd0659d
1 parent 677e151
commit dd0659d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/crypto/noise/protocol.rs b/src/crypto/noise/protocol.rs
@@ -75,8 +75,12 @@ impl snow::resolvers::CryptoResolver for Resolver {
         Some(Box::new(Rng(rand::rngs::StdRng::from_entropy())))
     }
 
-    fn resolve_dh(&self, _: &snow::params::DHChoice) -> Option<Box<dyn snow::types::Dh>> {
-        Some(Box::new(Keypair::<x25519_spec::X25519Spec>::default()))
+    fn resolve_dh(&self, choice: &snow::params::DHChoice) -> Option<Box<dyn snow::types::Dh>> {
+        if let snow::params::DHChoice::Curve25519 = choice {
+            Some(Box::new(Keypair::<x25519_spec::X25519Spec>::default()))
+        } else {
+            None
+        }
     }
 
     fn resolve_hash(