Merge pull request #2 from AYM1607/fix-key-size

Fix key size when decoding from b64.
This commit is contained in:
Mariano Uvalle 2023-11-12 00:04:19 -08:00 committed by GitHub
commit 02680eb0e3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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))