106 lines
2.3 KiB
Go
106 lines
2.3 KiB
Go
package encoding
|
||
|
||
import (
|
||
"bytes"
|
||
"crypto/aes"
|
||
"crypto/cipher"
|
||
"encoding/base64"
|
||
"encoding/json"
|
||
"errors"
|
||
)
|
||
|
||
//EncryptConfig 加密配置
|
||
type EncryptConfig struct {
|
||
Key string
|
||
IV string
|
||
}
|
||
|
||
//JSON JSON格式化增加对[]byte的支持
|
||
func JSON(o interface{}) string {
|
||
var v []byte
|
||
switch o1 := o.(type) {
|
||
case [][]byte:
|
||
o2 := make([]interface{}, len(o1))
|
||
for i := 0; i < len(o1); i++ {
|
||
o2[i] = string(o1[i])
|
||
}
|
||
v, _ = json.Marshal(o2)
|
||
case []interface{}:
|
||
for i := 0; i < len(o1); i++ {
|
||
if b, ok := o1[i].([]byte); ok {
|
||
o1[i] = string(b)
|
||
}
|
||
}
|
||
v, _ = json.Marshal(o1)
|
||
default:
|
||
v, _ = json.Marshal(o)
|
||
}
|
||
if len(v) > 0 {
|
||
return string(v)
|
||
}
|
||
return ""
|
||
}
|
||
|
||
//填充到BlockSize整数倍长度，如果正好就是对的长度，再多填充一个BlockSize长度
|
||
func pad(src []byte) []byte {
|
||
padding := aes.BlockSize - len(src)%aes.BlockSize
|
||
padtext := bytes.Repeat([]byte{byte(padding)}, padding)
|
||
return append(src, padtext...)
|
||
}
|
||
|
||
//去除填充的字节
|
||
func unpad(src []byte) ([]byte, error) {
|
||
length := len(src)
|
||
if length == 0 {
|
||
return []byte{0}, nil
|
||
}
|
||
unpadding := int(src[length-1])
|
||
if unpadding > length {
|
||
return nil, errors.New("unpad error. This could happen when incorrect encryption key is used")
|
||
}
|
||
return src[:(length - unpadding)], nil
|
||
}
|
||
|
||
//Encrypt 加密
|
||
func Encrypt(text string, c *EncryptConfig) (string, error) {
|
||
if "" == text {
|
||
return "", nil
|
||
}
|
||
block, err := aes.NewCipher([]byte(c.Key))
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
msg := pad([]byte(text))
|
||
ciphertext := make([]byte, len(msg))
|
||
mode := cipher.NewCBCEncrypter(block, []byte(c.IV))
|
||
mode.CryptBlocks(ciphertext, msg)
|
||
finalMsg := base64.StdEncoding.EncodeToString(ciphertext)
|
||
return finalMsg, nil
|
||
}
|
||
|
||
//Decrypt 解密
|
||
func Decrypt(text string, c *EncryptConfig) (string, error) {
|
||
if "" == text {
|
||
return "", nil
|
||
}
|
||
block, err := aes.NewCipher([]byte(c.Key))
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
decodedMsg, err := base64.StdEncoding.DecodeString(text)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
if (len(decodedMsg) % aes.BlockSize) != 0 {
|
||
return "", errors.New("blocksize must be multipe of decoded message length")
|
||
}
|
||
msg := decodedMsg
|
||
mode := cipher.NewCBCDecrypter(block, []byte(c.IV))
|
||
mode.CryptBlocks(msg, msg)
|
||
unpadMsg, err := unpad(msg)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
return string(unpadMsg), nil
|
||
}
|