golang Aes

package models
import (
 "bytes"
 "crypto/aes"
 "crypto/cipher"
 "errors" 
)
const (
 ivDefValue = "0102030405060708"
)
func AesEncrypt(plaintext []byte, key []byte) ([]byte, error) {
 block, err := aes.NewCipher(key)
 if err != nil {
 return nil, errors.New("invalid decrypt key")
 } 
 blockSize := block.BlockSize()
 plaintext = PKCS5Padding(plaintext, blockSize)
 iv := []byte(ivDefValue)
 blockMode := cipher.NewCBCEncrypter(block, iv) 
 ciphertext := make([]byte, len(plaintext))
 blockMode.CryptBlocks(ciphertext, plaintext)
 return ciphertext, nil 
}
func AesDecrypt(ciphertext []byte, key []byte) ([]byte, error) {
 block, err := aes.NewCipher(key)
 if err != nil {
 return nil, errors.New("invalid decrypt key")
 } 
 blockSize := block.BlockSize()
 if len(ciphertext) < blockSize {
 return nil, errors.New("ciphertext too short")
 } 
 iv := []byte(ivDefValue)
 if len(ciphertext)%blockSize != 0 { 
 return nil, errors.New("ciphertext is not a multiple of the block size")
 } 
 blockModel := cipher.NewCBCDecrypter(block, iv)
 plaintext := make([]byte, len(ciphertext))
 blockModel.CryptBlocks(plaintext, ciphertext)
 plaintext = PKCS5UnPadding(plaintext)
 return plaintext, nil
}
func PKCS5Padding(src []byte, blockSize int) []byte {
 padding := blockSize - len(src)%blockSize
 padtext := bytes.Repeat([]byte{byte(padding)}, padding)
 return append(src, padtext...)
}
func PKCS5UnPadding(src []byte) []byte {
 length := len(src)
 unpadding := int(src[length-1])
 return src[:(length - unpadding)]
} 

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传