forked from eddycjy/go-gin-example
/
aes.go
59 lines (54 loc) · 1.35 KB
/
aes.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
package util
import (
"crypto/aes"
"crypto/cipher"
"encoding/base64"
)
type AesEncrypt struct {
}
func (this *AesEncrypt) getKey() []byte {
strKey := "60f2fa84aed5a7a766c37e07817966dd"
keyLen := len(strKey)
if keyLen < 16 {
panic("res key 长度不能小于16")
}
arrKey := []byte(strKey)
if keyLen >= 32 {
//取前32个字节
return arrKey[:32]
}
if keyLen >= 24 {
//取前24个字节
return arrKey[:24]
}
//取前16个字节
return arrKey[:16]
}
//加密字符串
func (this *AesEncrypt) AesEncrypt(strMesg string) string {
key := this.getKey()
var iv = []byte(key)[:aes.BlockSize]
encrypted := make([]byte, len(strMesg))
aesBlockEncrypter, err := aes.NewCipher(key)
if err != nil {
return ""
}
aesEncrypter := cipher.NewCFBEncrypter(aesBlockEncrypter, iv)
aesEncrypter.XORKeyStream(encrypted, []byte(strMesg))
return base64.URLEncoding.EncodeToString(encrypted)
}
//解密字符串
func (this *AesEncrypt) AesDecrypt(str string) string {
key := this.getKey()
var iv = []byte(key)[:aes.BlockSize]
src, _ := base64.URLEncoding.DecodeString(str)
decrypted := make([]byte, len(src))
var aesBlockDecrypter cipher.Block
aesBlockDecrypter, err := aes.NewCipher([]byte(key))
if err != nil {
return ""
}
aesDecrypter := cipher.NewCFBDecrypter(aesBlockDecrypter, iv)
aesDecrypter.XORKeyStream(decrypted, src)
return string(decrypted)
}