90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
|
package utils
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"crypto/aes"
|
|||
|
|
"crypto/cipher"
|
|||
|
|
"crypto/rand"
|
|||
|
|
"encoding/hex"
|
|||
|
|
"math"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 对明文进行填充
|
|||
|
|
func Padding(plainText []byte, blockSize int) []byte {
|
|||
|
|
//计算要填充的长度
|
|||
|
|
n := blockSize - len(plainText)%blockSize
|
|||
|
|
//对原来的明文填充n个n
|
|||
|
|
temp := bytes.Repeat([]byte{byte(n)}, n)
|
|||
|
|
plainText = append(plainText, temp...)
|
|||
|
|
return plainText
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 对密文删除填充
|
|||
|
|
func UnPadding(cipherText []byte) []byte {
|
|||
|
|
//取出密文最后一个字节end
|
|||
|
|
end := cipherText[len(cipherText)-1]
|
|||
|
|
//删除填充
|
|||
|
|
cipherText = cipherText[:len(cipherText)-int(end)]
|
|||
|
|
return cipherText
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AEC加密（CBC模式）
|
|||
|
|
func AES_CBC_Encrypt(plainText []byte, key []byte) []byte {
|
|||
|
|
//指定加密算法，返回一个AES算法的Block接口对象
|
|||
|
|
block, err := aes.NewCipher(key)
|
|||
|
|
if err != nil {
|
|||
|
|
panic(err)
|
|||
|
|
}
|
|||
|
|
//进行填充
|
|||
|
|
plainText = Padding(plainText, block.BlockSize())
|
|||
|
|
//指定初始向量vi,长度和block的块尺寸一致
|
|||
|
|
iv := []byte("0000000000000000")
|
|||
|
|
//指定分组模式，返回一个BlockMode接口对象
|
|||
|
|
blockMode := cipher.NewCBCEncrypter(block, iv)
|
|||
|
|
//加密连续数据库
|
|||
|
|
cipherText := make([]byte, len(plainText))
|
|||
|
|
blockMode.CryptBlocks(cipherText, plainText)
|
|||
|
|
//返回base64密文
|
|||
|
|
return cipherText
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AEC解密（CBC模式）
|
|||
|
|
func AES_CBC_Decrypt(cipherText []byte, key []byte) []byte {
|
|||
|
|
//指定解密算法，返回一个AES算法的Block接口对象
|
|||
|
|
block, err := aes.NewCipher(key)
|
|||
|
|
if err != nil {
|
|||
|
|
panic(err)
|
|||
|
|
}
|
|||
|
|
//指定初始化向量IV,和加密的一致
|
|||
|
|
iv := []byte("0000000000000000")
|
|||
|
|
//指定分组模式，返回一个BlockMode接口对象
|
|||
|
|
blockMode := cipher.NewCBCDecrypter(block, iv)
|
|||
|
|
//解密
|
|||
|
|
plainText := make([]byte, len(cipherText))
|
|||
|
|
blockMode.CryptBlocks(plainText, cipherText)
|
|||
|
|
//删除填充
|
|||
|
|
plainText = UnPadding(plainText)
|
|||
|
|
return plainText
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateRandomString 生成一个32位的随机字符串订单号
|
|||
|
|
func GenerateRandomString() (string, error) {
|
|||
|
|
// 创建一个16字节的数组
|
|||
|
|
bytes := make([]byte, 16)
|
|||
|
|
// 读取随机字节到数组中
|
|||
|
|
if _, err := rand.Read(bytes); err != nil {
|
|||
|
|
return "", err
|
|||
|
|
}
|
|||
|
|
// 将字节数组编码为16进制字符串
|
|||
|
|
return hex.EncodeToString(bytes), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 定义一个函数，将分数从原始范围转换到 0-100 的范围
|
|||
|
|
func ConvertScore(originalScore, oldMin, oldMax float64) int {
|
|||
|
|
newMin := 0.0
|
|||
|
|
newMax := 100.0
|
|||
|
|
Score := ((originalScore - oldMin) * (newMax - newMin) / (oldMax - oldMin)) + newMin
|
|||
|
|
newScore := int(math.Round(100 - Score))
|
|||
|
|
return newScore
|
|||
|
|
}
|