【代码】Go语言RSA加解密

前言

Go语言RSA加解密工具函数

源代码

package main

import (
	"crypto/rand"
	"crypto/rsa"
	"crypto/x509"
	"encoding/base64"
	"encoding/pem"
	"fmt"
	"os"
)

// GenerateRSAKeys 生成密钥对
// bits 指定生成密钥对位数
// generateFile 是否生成密钥文件
func GenerateRSAKeys(bits int, generateFile bool) (publicKeyBase64String string, privateKeyBase64String string) {

	/* -- 生成私钥 -- */

	// 生成私钥
	privateKeyObject, _ := rsa.GenerateKey(rand.Reader, bits)
	// 通过x509对私钥编码为字符数组
	privateKeyByteArray := x509.MarshalPKCS1PrivateKey(privateKeyObject)
	// 转换为Base64编码的字符串
	privateKeyBase64String = base64.StdEncoding.EncodeToString(
		pem.EncodeToMemory(&pem.Block{
			Type:  "PUBLIC KEY",
			Bytes: privateKeyByteArray,
		}),
	)
	// 创建文件存储私钥
	if generateFile {
		privateKeyFile, _ := os.Create("private.pem")
		if err := pem.Encode(privateKeyFile, &pem.Block{
			Type:  "RSA Private Key",
			Bytes: privateKeyByteArray,
		}); err != nil {
			fmt.Println(err.Error())
		}
	}

	/* -- 生成公钥 -- */

	// 生成公钥
	publicKeyObject := &privateKeyObject.PublicKey
	// 通过x509对公钥编码
	publicKeyByteArray, _ := x509.MarshalPKIXPublicKey(publicKeyObject)
	// 转换为Base64编码的字符串
	publicKeyBase64String = base64.StdEncoding.EncodeToString(
		pem.EncodeToMemory(&pem.Block{
			Type:  "PRIVATE KEY",
			Bytes: publicKeyByteArray,
		}),
	)
	// 创建文件存储公钥
	if generateFile {
		publicKeyFile, _ := os.Create("public.pem")
		if err := pem.Encode(publicKeyFile, &pem.Block{
			Type:  "RSA Public Key",
			Bytes: publicKeyByteArray,
		}); err != nil {
			fmt.Println(err.Error())
		}
	}

	return privateKeyBase64String, publicKeyBase64String
}

// encrypt 通过RSA公钥加密字符串
// publicKeyFilePath 如果传递了公钥文件路径，就从公钥文件中获取私钥
// publicKeyBase64String 如果没有传递公钥文件路径，就从Base64字符串中获取公钥
func encrypt(plainText []byte, publicKeyFilePath string, publicKeyBase64String string) []byte {

	// 公钥字符数组
	var publicKeyByteArray []byte

	if publicKeyFilePath != "" {
		// 如果传递了文件路径，就从文件中获取公钥字符数组
		publicKeyFileObject, _ := os.Open(publicKeyFilePath)
		defer func(publicKeyFileObject *os.File) {
			if err := publicKeyFileObject.Close(); err != nil {
				fmt.Println(err.Error())
			}
		}(publicKeyFileObject)
		publicKeyFile, _ := publicKeyFileObject.Stat()
		publicKeyByteArray = make([]byte, publicKeyFile.Size())
		if _, err := publicKeyFileObject.Read(publicKeyByteArray); err != nil {
			fmt.Println(err.Error())
		}
	} else {
		// 如果没有传递文件路径，就从Base64字符串中获取公钥字符数组
		if res, err := base64.StdEncoding.DecodeString(publicKeyBase64String); err != nil {
			fmt.Println(err.Error())
		} else {
			publicKeyByteArray = res
		}
	}

	// pem解码
	block, _ := pem.Decode(publicKeyByteArray)
	// X509解码
	publicKey, _ := x509.ParsePKIXPublicKey(block.Bytes)
	// 对明文进行加密
	cipherText, _ := rsa.EncryptPKCS1v15(
		rand.Reader,
		publicKey.(*rsa.PublicKey),
		plainText,
	)

	//返回密文
	return cipherText
}

// decrypt 通过RSA私钥解密字符串
// privateKeyFilePath 如果传递了私钥文件路径，就从私钥文件中获取私钥
// privateKeyBase64String 如果没有传递私钥文件路径，就从Base64字符串中获取私钥
func decrypt(cipherText []byte, privateKeyFilePath string, privateKeyBase64String string) []byte {

	// 私钥字符数组
	var privateKeyByteArray []byte

	if privateKeyFilePath != "" {
		// 如果传递了文件路径，就从文件中获取公钥字符数组
		privateKeyFileObject, _ := os.Open(privateKeyFilePath)
		defer func(privateKeyFileObject *os.File) {
			if err := privateKeyFileObject.Close(); err != nil {
				fmt.Println(err.Error())
			}
		}(privateKeyFileObject)
		privateKeyFile, _ := privateKeyFileObject.Stat()
		privateKeyByteArray = make([]byte, privateKeyFile.Size())
		if _, err := privateKeyFileObject.Read(privateKeyByteArray); err != nil {
			fmt.Println(err.Error())
		}
	} else {
		// 如果没有传递文件路径，就从Base64字符串中获取公钥字符数组
		if res, err := base64.StdEncoding.DecodeString(privateKeyBase64String); err != nil {
			fmt.Println(err.Error())
		} else {
			privateKeyByteArray = res
		}
	}

	// pem解码
	block, _ := pem.Decode(privateKeyByteArray)
	// X509解码
	privateKey, _ := x509.ParsePKCS1PrivateKey(block.Bytes)
	// 对密文进行解密
	plainText, _ := rsa.DecryptPKCS1v15(
		rand.Reader,
		privateKey,
		cipherText,
	)

	//返回明文
	return plainText
}

func main() {

	// 生成密钥对
	publicKeyBase64String, privateKeyBase64String := GenerateRSAKeys(2048, true)
	fmt.Println(publicKeyBase64String)
	fmt.Println()
	fmt.Println(privateKeyBase64String)
	
	// 通过公钥文件加密字符串
	var res1 = encrypt(
		[]byte("字符串"),
		"public.pem",
		"",
	)
	fmt.Println(res1)

	// 通过公钥Base64加密字符串
	var res2 = encrypt(
		[]byte("字符串"),
		"",
		"LS0tLS1CRUdJTiBQUklWQVRFIEtFWS0tLS0tCk1JSUJJakFOQmdrcWhraUc5dzBCQVFFRkFBT0NBUThBTUlJQkNnS0NBUUVBeStmV3djSlRKQmNVd2V3Z3ZTMDkKL3hqNXY5VGFIczhseVpWN0JRbXgvYlQ2NHBqMW54bnBUdS9mNi9idjRFM0N6Z0RhMFBjV1gwdkM2UXEyVXc0cwpLOGk0QmlvNndpVHNIa0tiRzZVT1ZyUWJRUUN3VTNvNGpsWGxQZ0ZEREVvR2M4ZDhUNGV5UlBwOHZqV1lNdkI3CkkzOGZpYjNzWjcvNHgxVzdBNDZIdy9veFFOd2l4SHV3YUtNcFRlbkp0WjRGekFWaGQ3L1FwWFdSMDRGWUZzSTQKend1aVVlVVpKcnFyQnM3ZlhPWFd3VTV5NGQ4SEJZcGFESFhwbVl3dVB5UDE5SG1uSDhlQm5UR0Vod3U3L3dkQwpZaGY1VTBkTEFWY0M0ZTRZWHFsMHVRVjVHR0dBWkEyQ2ZPRWQ1clZRR1Baa25Wak1WRStZckpKMXk0TkNWOERpCit3SURBUUFCCi0tLS0tRU5EIFBSSVZBVEUgS0VZLS0tLS0K",
	)
	fmt.Println(res2)

	// 通过私钥文件解密字符串
	var res3 = decrypt(
		res1,
		"private.pem",
		"",
	)
	fmt.Println(string(res3))

	// 通过私钥Base64解密字符串
	var res4 = decrypt(
		res1,
		"",
		"LS0tLS1CRUdJTiBQVUJMSUMgS0VZLS0tLS0KTUlJRW93SUJBQUtDQVFFQXkrZld3Y0pUSkJjVXdld2d2UzA5L3hqNXY5VGFIczhseVpWN0JRbXgvYlQ2NHBqMQpueG5wVHUvZjYvYnY0RTNDemdEYTBQY1dYMHZDNlFxMlV3NHNLOGk0QmlvNndpVHNIa0tiRzZVT1ZyUWJRUUN3ClUzbzRqbFhsUGdGRERFb0djOGQ4VDRleVJQcDh2aldZTXZCN0kzOGZpYjNzWjcvNHgxVzdBNDZIdy9veFFOd2kKeEh1d2FLTXBUZW5KdFo0RnpBVmhkNy9RcFhXUjA0RllGc0k0end1aVVlVVpKcnFyQnM3ZlhPWFd3VTV5NGQ4SApCWXBhREhYcG1Zd3VQeVAxOUhtbkg4ZUJuVEdFaHd1Ny93ZENZaGY1VTBkTEFWY0M0ZTRZWHFsMHVRVjVHR0dBClpBMkNmT0VkNXJWUUdQWmtuVmpNVkUrWXJKSjF5NE5DVjhEaSt3SURBUUFCQW9JQkFITStKUEo5a2QvajY2ZHUKZlZKdUw0dnQ0Ym01NDAxbVprVDRadjhCNjhjb2NtTi9iOUxzSWlDckVqT2E3UmdvMnVEek9yaS83YlVPYktxcAprMEo5MEJrdTZGTVRxd0h2UEVRZUFPd3U3VnV6ajNPOW4rcU9PVWk0NDBlWUR2enpHWkoyUEp2MlpacjJ0Mm9KCmdHdkozczdCR21JUEpYMjkwaTFYRnN0NzB0Ky9sWXpNSzFpQVBHRUpwRVIvd29tUTArSDRTcmZ3ejRIMmgzMDMKU21JOTIzWGhjSjBFTHNibUdPR29jemxRWWVhaVRqQTk0YTNsbElZeWFGSHZuN2xRQklqQTFCTE9BbGNoTkVzNwplT1dkZ056cno5TnpSNG1KbE5CTGhCcEFWOHRyTFlBTThVWWp3eEhHczNTcEYyMUZnSFNMVmZFTW9xL0xTa3NuCkJBWnlLUUVDZ1lFQTdJV3Y3bm9Rb3hhU3VkZmZiMTdkSUNSYkVrOTNLMkk3am8zbURMWi8xbjFXUTFIZmFLQ0EKWmllQlpBZ1dnNEVuQjNTMDl4WUNWc2ZHTjB0T01neGpOZXByQmI5dk5DMDUxYlU2a3lsTDdtZGVJd3hpMnJFYgpUcU42MUxJSktMb2VXeEwyVmRqZUdla3JVMXI1RFdsbTRHcEgrSVBtQUMzUEtSL2laUjhHVExNQ2dZRUEzTEtKCkNhOFVValIvTS9ydS9YN0M1N3lYaUhBVlNKTWk4amRaamphRE02Q1RZSThWYStMYzArbk1naENPSFR6T3N0aDEKQmp2SWs1dkhKbitlTGxkYmxObnFtaW5RSTVuMjhNQjFBK1NCY3NLU21qeEJRS3lQK3RhY1FxT2pZUWtxbnpjYgozbkR4TW5hcmVJWnpoaGozczVINXBKNitLbjFXMjdoeWJNYXN4SmtDZ1lCOWw3K0Q4bTFETElsZEJiUXZmMEFNCkFVSWF4dmVRb2VqZUdacyt4T205QzNEeTdsYWpTQmJtSjdRZFdlNGFoa29qTzJXY2pxTlBVV1YvbHcwK3BqVkkKTytINGUxNmhaaUoveUd3Q0dMdXBMRnBnWWdOYi9ROXpBbndHMVVMbnpXeXhEN0NlaFRucXFGWEpQVlZpT2NYWApCSmlTN2ZjRHdVOXZsdTZ3bzNqRTRRS0JnUUNtNmlpV3VHR29JcWdxQ2hrQ2hBS28vWXVsMXc5aHlsWmQxdHJXCm9nZnMzZjlZd2g2emIrc1FJdjFYU1dvU1Z0RUZsb3c3WUsrOFk4N2xJUlBqZkpKNUs2OHhWWHBqMHFoZlFVKysKMSszSGU3MGhCWDFjaXFhQ1VobndsSTZXYnpaYklIVGxjWW9hWG9JbEw1cmp5ckpjajRTOXFvbXMzUXI2TnEvago4UWJ0MFFLQmdBbEtGbHRPZXEvT2I0R2dIYjByU3ZoYU5naUxhQ0JTS2VjeHJjMzJVMDNjMktaZGFWOHZmTXZWCndEbE5vYnFFV0pXUUUzMy9JemozVXFNOUw4bGxXZnNHVkJVc0dOZHdrRmUwaVdtbEZKVFArVHBienVFSG0yRlAKWmFYMHByNjk3VjJQa003THovRk5xbjN0UXBZWjdqeVJ4UFM2NnYrY1NBT0JWU21YWmd1aQotLS0tLUVORCBQVUJMSUMgS0VZLS0tLS0K",
	)
	fmt.Println(string(res4))

}

完成

参考文献

CSDN——起个破名真费劲..
CSDN——Axing丶