forked from golang/hotime
43 lines
1.7 KiB
Go
43 lines
1.7 KiB
Go
package aes
|
||
|
||
import "bytes"
|
||
|
||
// 加密填充模式（添加补全码） PKCS5Padding
|
||
// 加密时，如果加密bytes的length不是blockSize的整数倍，需要在最后面添加填充byte
|
||
func PKCS5Padding(ciphertext []byte, blockSize int) []byte {
|
||
paddingCount := blockSize - len(ciphertext)%blockSize //需要padding的数目
|
||
paddingBytes := []byte{byte(paddingCount)}
|
||
padtext := bytes.Repeat(paddingBytes, paddingCount) //生成填充的文本
|
||
return append(ciphertext, padtext...)
|
||
}
|
||
|
||
// 解密填充模式（去除补全码） PKCS5UnPadding
|
||
// 解密时，需要在最后面去掉加密时添加的填充byte
|
||
func PKCS5UnPadding(origData []byte) []byte {
|
||
length := len(origData)
|
||
unpadding := int(origData[length-1]) //找到Byte数组最后的填充byte
|
||
return origData[:(length - unpadding)] //只截取返回有效数字内的byte数组
|
||
}
|
||
|
||
// 加密填充模式（添加补全码） PKCS5Padding
|
||
// 加密时，如果加密bytes的length不是blockSize的整数倍，需要在最后面添加填充byte
|
||
func PKCS7Padding(ciphertext []byte, blockSize int) []byte {
|
||
paddingCount := blockSize - len(ciphertext)%blockSize //需要padding的数目
|
||
paddingBytes := []byte{byte(paddingCount)}
|
||
padtext := bytes.Repeat(paddingBytes, paddingCount) //生成填充的文本
|
||
return append(ciphertext, padtext...)
|
||
}
|
||
|
||
// 解密填充模式（去除补全码） PKCS7UnPadding
|
||
// 解密时，需要在最后面去掉加密时添加的填充byte
|
||
func PKCS7UnPadding(origData []byte) (bs []byte) {
|
||
length := len(origData)
|
||
unPaddingNumber := int(origData[length-1]) // 找到Byte数组最后的填充byte 数字
|
||
if unPaddingNumber <= 16 {
|
||
bs = origData[:(length - unPaddingNumber)] // 只截取返回有效数字内的byte数组
|
||
} else {
|
||
bs = origData
|
||
}
|
||
return
|
||
}
|