go使用aes加密算法

超维Ai编程2023-12-08 8:12

工具代码

go 复制代码

package tool

import (
	"bytes"
	"crypto/aes"
	"crypto/cipher"
)

// AES加密函数
var key []byte = []byte("0@#3456789ABCDEF") //todo 记住这个长度只能是16 24 32  如果不是的话话会报错
func Encrypt(data []byte) ([]byte, error) {
	block, err := aes.NewCipher(key)
	if err != nil {
		return nil, err
	}
	// 填充明文字节数组
	paddingLen := aes.BlockSize - (len(data) % aes.BlockSize)
	paddedData := append(data, bytes.Repeat([]byte{byte(paddingLen)}, paddingLen)...)
	// 执行加密操作
	encrypted := make([]byte, len(paddedData))
	iv := make([]byte, aes.BlockSize) // 初始化向量，通常为全零
	mode := cipher.NewCBCEncrypter(block, iv)
	mode.CryptBlocks(encrypted, paddedData)
	return encrypted, nil
}

// AES解密函数
func Decrypt(encryptedData []byte) ([]byte, error) {
	block, err := aes.NewCipher(key)
	if err != nil {
		return nil, err
	}
	// 执行解密操作
	decrypted := make([]byte, len(encryptedData))
	iv := make([]byte, aes.BlockSize) // 初始化向量，通常为全零
	mode := cipher.NewCBCDecrypter(block, iv)
	mode.CryptBlocks(decrypted, encryptedData)
	// 去除填充字节
	paddingLen := int(decrypted[len(decrypted)-1])
	decrypted = decrypted[:len(decrypted)-paddingLen]
	return decrypted, nil
}

// 计算加密后的字节数组长度
func CalculateEncryptedLength(dataLen int) int {
	paddingLen := aes.BlockSize - (dataLen % aes.BlockSize)
	encryptedLen := dataLen + paddingLen
	return encryptedLen
}

测试类

go 复制代码

package test
import (
	"beego_test/tool"
	"testing"
)
// todo 字节解密和解密测试
func Test_encryption(t *testing.T) {
	originalBytes := []byte{115, 91, 212, 128, 220, 50, 0, 247, 207, 91, 178, 183, 207, 59, 115, 91}
	encryptedBytes, err := tool.Encrypt(originalBytes)
	if err != nil {
		log.Fatal(err)
	}
	d, err := tool.Decrypt(encryptedBytes)
	fmt.Println("原始的字节为", originalBytes)
	fmt.Println("加密后的字节为", encryptedBytes)
	fmt.Println("加密后的字节的长度为", len(encryptedBytes))
	fmt.Println("字节49加密后的长度", tool.CalculateEncryptedLength(49))
	fmt.Println("解密后的字节为", d)
	fmt.Println("无法根据密文长度推出原文长度")
	fmt.Println("可以根据密文退出原文")
	fmt.Println("可以根据原文长度推出密文长度")
	fmt.Println("如果字节无法解密会报错")
	tool.Decrypt(originalBytes)
}