feat(): add generate rsa privatekey

Author: zyjblockchain

signer.go

@@ -56,6 +56,16 @@ func NewSigner(b []byte) (*Signer, error) {
 	}, nil
 }
 
+func NewSignerByPrivateKey(privateKey *rsa.PrivateKey) *Signer {
+	pub := &privateKey.PublicKey
+	addr := sha256.Sum256(pub.N.Bytes())
+	return &Signer{
+		Address: utils.Base64Encode(addr[:]),
+		PubKey:  pub,
+		PrvKey:  privateKey,
+	}
+}
+
 func (s *Signer) SignTx(tx *types.Transaction) error {
 	return utils.SignTransaction(tx, s.PrvKey)
 }

utils/crypto.go

@@ -7,6 +7,10 @@ import (
 	"crypto/sha256"
 )
 
+func GenerateRsaKey(bits int) (*rsa.PrivateKey, error) {
+	return rsa.GenerateKey(rand.Reader, bits)
+}
+
 func Sign(msg []byte, prvKey *rsa.PrivateKey) ([]byte, error) {
 	hashed := sha256.Sum256(msg)
 

utils/crypto_test.go

@@ -1,17 +1,14 @@
 package utils
 
 import (
-	"crypto/rand"
-	"crypto/rsa"
-	"testing"
-
 	"github.com/stretchr/testify/assert"
+	"testing"
 )
 
 func TestSignAndVerify(t *testing.T) {
-	rightKey, err := rsa.GenerateKey(rand.Reader, 2048)
+	rightKey, err := GenerateRsaKey(4096)
 	assert.NoError(t, err)
-	wrongKey, err := rsa.GenerateKey(rand.Reader, 2048)
+	wrongKey, err := GenerateRsaKey(4096)
 	assert.NoError(t, err)
 	msg := []byte("123")