diff --git a/message/validation/rsa_benchmark_test.go b/message/validation/rsa_benchmark_test.go index 19ba8b0b62..02d80bb0eb 100644 --- a/message/validation/rsa_benchmark_test.go +++ b/message/validation/rsa_benchmark_test.go @@ -77,8 +77,9 @@ func BenchmarkVerifyRSASignature(b *testing.B) { hash := sha256.Sum256(encodedMsg) privateKey, err := rsa.GenerateKey(crand.Reader, 2048) require.NoError(b, err) + pubkey := privateKey.Public().(*rsa.PublicKey) - pubKey, err := rsaencryption.ExtractPublicKey(privateKey) + pubKey, err := rsaencryption.ExtractPublicKey(pubkey) require.NoError(b, err) od := ®istrystorage.OperatorData{ @@ -113,7 +114,7 @@ func BenchmarkVerifyRSASignature(b *testing.B) { b.ResetTimer() for i := 0; i < b.N; i++ { - err := mv.verifyRSASignature(messageData, operatorIDX, signature) + err := mv.verifySignature(messageData, operatorIDX, signature) require.NoError(b, err) } } diff --git a/utils/rsaencryption/rsa_encryption.go b/utils/rsaencryption/rsa_encryption.go index 7fda24a1b0..4a52c9f467 100644 --- a/utils/rsaencryption/rsa_encryption.go +++ b/utils/rsaencryption/rsa_encryption.go @@ -114,7 +114,7 @@ func PrivateKeyToByte(sk *rsa.PrivateKey) []byte { ) } -// ExtractPublicKey get public key from private key and return base64 encoded public key +// ExtractPublicKey gets public key and returns base64 encoded public key func ExtractPublicKey(pk *rsa.PublicKey) (string, error) { pkBytes, err := x509.MarshalPKIXPublicKey(pk) if err != nil {