go实现AES对字符串或文件内容加密,复制即用

package aessecret

import (
	"bytes"
	"crypto/aes"
	"crypto/cipher"
	"crypto/rand"
	"encoding/base64"
	"fmt"
	"io"
)

var key = []byte("此处填写密钥，32字符")

// zeroPadding zeroPadding 函数用于在字符串的末尾添加零填充，使其长度达到指定的块大小。
// 参数：
//
//	data (string) - 需要进行填充的字符串。
//	blockSize (int) - 块大小，是一个整数，表示需要将字符串填充到的长度。
//
// 返回值（string）- 返回填充后的字符串，如果原始字符串已经达到或超过了指定的块大小，则不会再进行填充。
func zeroPadding(data string, blockSize int) string {
	padding := blockSize - (len(data) % blockSize)
	pad := bytes.Repeat([]byte{0}, padding)
	return data + string(pad)
}

// Encrypt Encrypt 加密一个字符串，返回加密后的字符串和错误信息（如果有）。
// 参数：
//
//	data (string) - 需要加密的字符串。
//
// 返回值：
//
//	(string, error) - 第一个元素是加密后的字符串，第二个元素是错误信息（如果有）。
func Encrypt(data string) (string, error) {
	data = zeroPadding(data, aes.BlockSize)

	block, err := aes.NewCipher(key)
	if err != nil {
		return "", err
	}

	cipherText := make([]byte, aes.BlockSize+len(data))
	iv := cipherText[:aes.BlockSize]
	if _, err := io.ReadFull(rand.Reader, iv); err != nil {
		return "", err
	}

	mode := cipher.NewCBCEncrypter(block, iv)
	mode.CryptBlocks(cipherText[aes.BlockSize:], []byte(data))

	return base64.StdEncoding.EncodeToString(cipherText), nil
}

// Decrypt Decrypt 函数用于解密字符串，返回解密后的字符串和错误信息（如果有）。
// 参数 data 为需要解密的字符串，类型为 string。
// 返回值1：string，解密后的字符串。
// 返回值2：error，解密过程中出现的错误信息，如果没有错误则为 nil。
func Decrypt(data string) (string, error) {
	cipherText, err := base64.StdEncoding.DecodeString(data)
	if err != nil {
		return "", err
	}

	block, err := aes.NewCipher(key)
	if err != nil {
		return "", err
	}

	if len(cipherText) < aes.BlockSize {
		return "", fmt.Errorf("cipherText too short")
	}

	iv := cipherText[:aes.BlockSize]
	cipherText = cipherText[aes.BlockSize:]

	mode := cipher.NewCBCDecrypter(block, iv)
	mode.CryptBlocks(cipherText, cipherText)

	return string(cipherText), nil
}