添加rsa非对称加密的助手方法

.vscode/settings.json

@@ -8,4 +8,5 @@
       "go.gotoSymbol.includeImports": true,
       "go.inferGopath": true,
       "go.gotoSymbol.includeGoroot": true,
+      "C_Cpp.intelliSenseEngineFallback": "Disabled",
 }

test/crypto_test.go

@@ -0,0 +1,15 @@
+package test
+
+import (
+	"gobase/utils"
+	"log"
+	"testing"
+)
+
+func TestCrypto(t *testing.T) {
+	pub, pri := utils.GenRsaKey(1024)
+	enc, e := utils.RSAEncrypt([]byte{1, 2, 3}, pub)
+	log.Print(enc, e)
+	dec, e := utils.RSADecrypt(enc, pri)
+	log.Print(dec, e)
+}

utils/utils.go

@@ -1,8 +1,13 @@
 package utils
 
 import (
+	"bytes"
 	"crypto/md5"
+	"crypto/rand"
+	"crypto/rsa"
+	"crypto/x509"
 	"encoding/hex"
+	"encoding/pem"
 	"os"
 	"path/filepath"
 	"reflect"
@@ -41,6 +46,7 @@ func InArrary(arr interface{}, ele interface{}) int {
 	}
 	return 1
 }
+
 // 将程序变成守护进程。
 func Daemonize() {
 	if runtime.GOOS == "windows" {
@@ -57,3 +63,71 @@ func Daemonize()  {
 		}
 	}
 }
+
+/*
+ * RSA私钥解密
+ */
+func RSADecrypt(src []byte, buf []byte) ([]byte, error) {
+	block, _ := pem.Decode(buf)
+	if block == nil {
+		return nil, nil
+	}
+	// 解析出一个der编码的私钥
+	privateKey, err := x509.ParsePKCS1PrivateKey(block.Bytes)
+
+	// 私钥解密
+	result, err := rsa.DecryptPKCS1v15(rand.Reader, privateKey, src)
+	if err != nil {
+		return nil, err
+	}
+	return result, nil
+}
+
+func RSAEncrypt(src []byte, buf []byte) ([]byte, error) {
+
+	block, _ := pem.Decode(buf)
+	if block == nil {
+		return nil, nil
+	}
+	// 解析一个der编码的公钥
+	publicKey, err := x509.ParsePKCS1PublicKey(block.Bytes)
+	if err != nil {
+		return nil, err
+	}
+
+	// 公钥加密
+	result, _ := rsa.EncryptPKCS1v15(rand.Reader, publicKey, src)
+	return result, nil
+}
+
+func GenRsaKey(bits int) ([]byte, []byte) {
+	pribuf := bytes.NewBuffer([]byte{})
+	pubbuf := bytes.NewBuffer([]byte{})
+	// 生成私钥文件
+	privateKey, err := rsa.GenerateKey(rand.Reader, bits)
+	if err != nil {
+		return nil, nil
+	}
+	derPrivateStream := x509.MarshalPKCS1PrivateKey(privateKey)
+	block := &pem.Block{
+		Type:  "RSA PRIVATE KEY",
+		Bytes: derPrivateStream,
+	}
+	err = pem.Encode(pribuf, block)
+	if err != nil {
+		return nil, nil
+	}
+	// 生成公钥文件
+	publicKey := &privateKey.PublicKey
+	derPkix := x509.MarshalPKCS1PublicKey(publicKey)
+	block = &pem.Block{
+		Type:  "RSA PUBLIC KEY",
+		Bytes: derPkix,
+	}
+
+	err = pem.Encode(pubbuf, block)
+	if err != nil {
+		return nil, nil
+	}
+	return pubbuf.Bytes(), pribuf.Bytes()
+}