From 35bf532e4b1dc721d271fd508a5e0b381f8729eb Mon Sep 17 00:00:00 2001 From: Mariano Uvalle Date: Sun, 12 Nov 2023 08:03:53 +0000 Subject: [PATCH] Fix key size when decoding from b64. --- pkg/crypto/keys.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkg/crypto/keys.go b/pkg/crypto/keys.go index db983ec..fca1b3c 100644 --- a/pkg/crypto/keys.go +++ b/pkg/crypto/keys.go @@ -65,7 +65,7 @@ func NewSharedKey(local *ecdh.PrivateKey, remote *ecdh.PublicKey, direction Dire } func PrivateKeyFromB64(encodedKey []byte) *ecdh.PrivateKey { - keyBytes := make([]byte, base64.StdEncoding.DecodedLen(len(encodedKey))) + keyBytes := make([]byte, KeySize) _, err := base64.StdEncoding.Decode(keyBytes, encodedKey) if err != nil { panic(fmt.Errorf("could not decode base64 private key: %w", err)) @@ -74,7 +74,7 @@ func PrivateKeyFromB64(encodedKey []byte) *ecdh.PrivateKey { } func PublicKeyFromB64(encodedKey []byte) *ecdh.PublicKey { - keyBytes := make([]byte, base64.StdEncoding.DecodedLen(len(encodedKey))) + keyBytes := make([]byte, KeySize) _, err := base64.StdEncoding.Decode(keyBytes, encodedKey) if err != nil { panic(fmt.Errorf("could not decode base64 public key: %w", err))