GO语言实现AES-CFB加密的操作方法

package encrypt
import (
	"crypto/aes"        // 提供 AES 加密算法实现
	"crypto/cipher"     // 提供加密算法接口和模式（如 CFB）
	"crypto/rand"       // 提供生成随机数功能
	"encoding/base64"   // 提供 Base64 编码和解码
	"fmt"               // 格式化 I/O
	"io"                // 基本 I/O 接口
)

// Encrypt 对明文进行AES-CFB加密
func Encrypt(plaintext []byte, key string) (string, error) {
	block, err := aes.NewCipher([]byte(key))
	if err != nil {
		return "", err
	}
	iv := make([]byte, aes.BlockSize)
	if _, err := io.ReadFull(rand.Reader, iv); err != nil {
		return "", err
	}
	stream := cipher.NewCFBEncrypter(block, iv)
	ciphertext := make([]byte, len(plaintext))
	stream.XORKeyStream(ciphertext, plaintext)
	// 合并 IV 和密文
	result := append(iv, ciphertext...)
	// 使用 Base64 编码返回
	return base64.StdEncoding.EncodeToString(result), nil
}

// Decrypt 对密文进行AES-CFB解密
func Decrypt(ciphertextBase64 string, key string) ([]byte, error) {
	ciphertext, err := base64.StdEncoding.DecodeString(ciphertextBase64)
	if err != nil {
		return nil, err
	}
	block, err := aes.NewCipher([]byte(key))
	if err != nil {
		return nil, err
	}
	if len(ciphertext) < aes.BlockSize {
		return nil, fmt.Errorf("ciphertext too short")
	}
	iv := ciphertext[:aes.BlockSize]
	ciphertext = ciphertext[aes.BlockSize:]
	stream := cipher.NewCFBDecrypter(block, iv)
	plaintext := make([]byte, len(ciphertext))
	stream.XORKeyStream(plaintext, ciphertext)
	return plaintext, nil
}