AES_CBC.go

package goencrypt

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

/*
@Time : 2018/11/1 22:53
@Author : wuman
@File : AES_CBC
@Software: GoLand
*/
/**
加密
	注意 : 这里采用key长度为16字节
*/
func init() {
	log.SetFlags(log.Ldate | log.Lshortfile)
}
func AesCBC_Encrypt(plainText, key []byte) []byte {
	//判断用户传过来的key是否符合16字节，如果不符合16字节加以处理
	keylen := len(key)
	if keylen == 0 { //如果用户传入的密钥为空那么就用默认密钥
		key = []byte("wumansgygoaescbc") //默认密钥
	} else if keylen > 0 && keylen < 16 { //如果密钥长度在0到16之间，那么用0补齐剩余的
		key = append(key, bytes.Repeat([]byte{0}, (16-keylen))...)
	} else if keylen > 16 {
		key = key[:16]
	}
	//1.指定一个aes算法,返回一个block接口
	block, err := aes.NewCipher(key)
	if err != nil {
		panic(err)
	}
	//2.分组填充数据 blockSize 16
	paddingText := PKCS5Padding(plainText, block.BlockSize())
	//3.创建使用cbc分组模式的blockMode接口
	iv := []byte("wumansgy12345678") //初始化向量，需要和block.blocksize长度一样
	blockMode := cipher.NewCBCEncrypter(block, iv)
	//4. 加密
	cipherText := make([]byte, len(paddingText))
	blockMode.CryptBlocks(cipherText, paddingText)
	// 5. 返回数据
	return cipherText
}

//解密
func AesCBC_Decrypt(cipherText, key []byte) []byte {
	//判断用户传过来的key是否符合16字节，如果不符合16字节加以处理
	keylen := len(key)
	if keylen == 0 { //如果用户传入的密钥为空那么就用默认密钥
		key = []byte("wumansgygoaescbc") //默认密钥
	} else if keylen > 0 && keylen < 16 { //如果密钥长度在0到16之间，那么用0补齐剩余的
		key = append(key, bytes.Repeat([]byte{0}, (16-keylen))...)
	} else if keylen > 16 {
		key = key[:16]
	}

	//1.指定使用aes算法
	block, err := aes.NewCipher(key)
	if err != nil {
		panic(err)
	}
	//2.获取一个CBC模式的blockMode接口
	iv := []byte("wumansgy12345678") //初始化向量，需要和block.blocksize长度一样
	blockMode := cipher.NewCBCDecrypter(block, iv)
	//3.解密,内容包含有填充的数据
	paddingText := make([]byte, len(cipherText))
	blockMode.CryptBlocks(paddingText, cipherText)
	//4.删除填充的数据,防止出错，错误检查一下

	defer func() {
		if err := recover(); err != nil {
			switch err.(type) {
			case runtime.Error:
				log.Println("runtime err:", err, "请检查密钥是否正确")
			default:
				log.Println("error:", err)
			}
		}
	}()

	plainText := PKCS5UnPadding(paddingText)

	//5.返回明文
	return plainText
}