package main
import (
"bytes"
"crypto/aes"
"crypto/cipher"
"encoding/base64"
"errors"
"fmt"
)
// 加密过程:// 1、解决数据,对数据进行填充,采纳 PKCS7(当密钥长度不够时,缺几位补几个几)的形式。// 2、对数据进行加密,采纳 AES 加密办法中 CBC 加密模式
// 3、对失去的加密数据,进行 base64 加密,失去字符串
// 解密过程相同
//16,24,32 位字符串的话,别离对应 AES-128,AES-192,AES-256 加密办法
//key 不能泄露
var PwdKey = []byte("1234asdf1234asdf")
//pkcs7Padding 填充
func pkcs7Padding(data []byte, blockSize int) []byte {
// 判断短少几位长度。起码 1,最多 blockSize
padding := blockSize - len(data)%blockSize
// 补足位数。把切片 []byte{byte(padding)} 复制 padding 个
padText := bytes.Repeat([]byte{byte(padding)}, padding)
return append(data, padText...)
}
//pkcs7UnPadding 填充的反向操作
func pkcs7UnPadding(data []byte) ([]byte, error) {length := len(data)
if length == 0 {return nil, errors.New("加密字符串谬误!")
}
// 获取填充的个数
unPadding := int(data[length-1])
return data[:(length - unPadding)], nil
}
//AesEncrypt 加密
func AesEncrypt(data []byte, key []byte) ([]byte, error) {
// 创立加密实例
block, err := aes.NewCipher(key)
if err != nil {return nil, err}
// 判断加密快的大小
blockSize := block.BlockSize()
// 填充
encryptBytes := pkcs7Padding(data, blockSize)
// 初始化加密数据接管切片
crypted := make([]byte, len(encryptBytes))
// 应用 cbc 加密模式
blockMode := cipher.NewCBCEncrypter(block, key[:blockSize])
// 执行加密
blockMode.CryptBlocks(crypted, encryptBytes)
return crypted, nil
}
//AesDecrypt 解密
func AesDecrypt(data []byte, key []byte) ([]byte, error) {
// 创立实例
block, err := aes.NewCipher(key)
if err != nil {return nil, err}
// 获取块的大小
blockSize := block.BlockSize()
// 应用 cbc
blockMode := cipher.NewCBCDecrypter(block, key[:blockSize])
// 初始化解密数据接管切片
crypted := make([]byte, len(data))
// 执行解密
blockMode.CryptBlocks(crypted, data)
// 去除填充
crypted, err = pkcs7UnPadding(crypted)
if err != nil {return nil, err}
return crypted, nil
}
//EncryptByAes Aes 加密 后 base64
func EncryptByAes(data []byte, key []byte) (string, error) {res, err := AesEncrypt(data, key)
if err != nil {return "", err}
return base64.StdEncoding.EncodeToString(res), nil
}
//DecryptByAes base64 解码后 Aes 解密
func DecryptByAes(data string, key []byte) ([]byte, error) {dataByte, err := base64.StdEncoding.DecodeString(data)
if err != nil {return nil, err}
return AesDecrypt(dataByte, key)
}
func main() {var origin = []byte("123456")
// 加密
encrypt, _ := EncryptByAes(origin, PwdKey)
// 解密
decrypt, _ := DecryptByAes(encrypt, PwdKey)
// 打印
fmt.Printf("加密前: %s\n", origin)
fmt.Printf("加密后: %s\n", encrypt)
fmt.Printf("解密后:%s\n", decrypt)
}