当前位置：首页 > news >正文

go使用aes加密算法

news 来源：原创 2024/5/20 3:02:42

工具代码

package toolimport ("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 + paddingLenreturn encryptedLen
}

测试类

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)
}