chore: improve ECDH public key converting

ldclabs · Sep 29, 2024 · c89833e · c89833e
1 parent 1ee1caf
commit c89833e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/key/ecdh/ecdh.go b/key/ecdh/ecdh.go
@@ -7,6 +7,7 @@ package ecdh
 
 import (
 	goecdh "crypto/ecdh"
+	"crypto/ecdsa"
 	"crypto/elliptic"
 	"crypto/rand"
 	"fmt"
@@ -119,8 +120,8 @@ func keyToPublic(pk key.Key) (*goecdh.PublicKey, error) {
 		copy(compressed[1:], x)
 		ix, iy = elliptic.UnmarshalCompressed(ecdsaCurve, compressed)
 	}
-
-	return curve.NewPublicKey(elliptic.Marshal(ecdsaCurve, ix, iy))
+	k := ecdsa.PublicKey{Curve: ecdsaCurve, X: ix, Y: iy}
+	return k.ECDH()
 }
 
 // KeyFromPublic returns a public Key with given ecdh.PublicKey.