You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

104 lines
2.8 KiB

package xaes
import (
"bytes"
"crypto/aes"
"crypto/cipher"
)
//#;AES加解密使用 长度 16, 24, 32
//const aesKey = "sfe023f_9fd&fwfl"
////加密字符串
//func AesEncrypt(hstring string) (string, error) {
// if result, err := aesEncrypt([]byte(hstring), []byte(aesKey)); err != nil {
// return "", err
// } else {
// return base64.StdEncoding.EncodeToString(result), nil
// }
//}
//
////解密字符串
//func AesDecrypt(hstring string) (string, error) {
// if bye, err := base64.StdEncoding.DecodeString(hstring); err != nil {
// return "", err
// } else {
// if result, err := aesDecrypt(bye, []byte(aesKey)); err != nil {
// return "", err
// } else {
// return string(result), nil
// }
// }
//}
//func AESTestCode() {
// // AES-128。key长度:16, 24, 32 bytes 对应 AES-128, AES-192, AES-256
// key := []byte("sfe023f_9fd&fwfl")
// result, err := aesEncrypt([]byte("polaris@studygolang"), key)
// if err != nil {
// panic(err)
// }
// fmt.Println(result)
// a, _ := AesEncrypt("polaris@studygolang")
// aa, _ := base64.StdEncoding.DecodeString(a)
// origData, err := aesDecrypt(aa, key)
// if err != nil {
// panic(err)
// }
// fmt.Println(string(origData))
//}
func Encrypt(origData, key []byte) ([]byte, error) {
block, err := aes.NewCipher(key)
if err != nil {
return nil, err
}
blockSize := block.BlockSize()
origData = pKCS5Padding(origData, blockSize)
// origData = ZeroPadding(origData, block.BlockSize())
blockMode := cipher.NewCBCEncrypter(block, key[:blockSize])
crypted := make([]byte, len(origData))
// 根据CryptBlocks方法的说明,如下方式初始化crypted也可以
// crypted := origData
blockMode.CryptBlocks(crypted, origData)
return crypted, nil
}
func Decrypt(crypted, key []byte) ([]byte, error) {
block, err := aes.NewCipher(key)
if err != nil {
return nil, err
}
blockSize := block.BlockSize()
blockMode := cipher.NewCBCDecrypter(block, key[:blockSize])
origData := make([]byte, len(crypted))
// origData := crypted
blockMode.CryptBlocks(origData, crypted)
origData = pKCS5UnPadding(origData)
// origData = ZeroUnPadding(origData)
return origData, nil
}
// func zeroPadding(ciphertext []byte, blockSize int) []byte {
// padding := blockSize - len(ciphertext)%blockSize
// padtext := bytes.Repeat([]byte{0}, padding)
// return append(ciphertext, padtext...)
// }
//
// func zeroUnPadding(origData []byte) []byte {
// length := len(origData)
// unpadding := int(origData[length-1])
// return origData[:(length - unpadding)]
// }
func pKCS5Padding(ciphertext []byte, blockSize int) []byte {
padding := blockSize - len(ciphertext)%blockSize
padtext := bytes.Repeat([]byte{byte(padding)}, padding)
return append(ciphertext, padtext...)
}
func pKCS5UnPadding(origData []byte) []byte {
length := len(origData)
unpadding := int(origData[length-1])
return origData[:(length - unpadding)]
}