/
main.go
110 lines (90 loc) · 2.14 KB
/
main.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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
package main
import (
"bytes"
"crypto/aes"
"crypto/cipher"
"crypto/md5"
"crypto/sha1"
"encoding/base64"
"fmt"
)
func main() {
md5 := md5.Sum([]byte("5d4b7a49339eb3191688116224871548"))
sha1 := sha1.Sum([]byte("5d4b7a49339eb3191688116224871548"))
key := GetByteArray(converArr1(sha1), 32)
iv := converArr2(md5)
result, _ := NewAESCrypt(key, iv, AESModeCBC).Encrypt("{\"user_code\":\"Admin\"}")
fmt.Println(result)
}
const (
AESModeCBC AESMode = "CBC"
)
type AESMode string
type AESCrypt struct {
Key []byte
IV []byte
Mode AESMode
}
func NewAESCrypt(key []byte, iv []byte, mode AESMode) AESCrypt {
return AESCrypt{Key: key, IV: iv, Mode: mode}
}
func (a AESCrypt) Encrypt(value string) (result string, err error) {
originData := []byte(value)
var encrypted []byte
switch a.Mode {
case AESModeCBC:
encrypted, err = a.aesEncryptCBC(originData)
}
if err != nil {
return
}
result = base64.StdEncoding.EncodeToString(encrypted)
return
}
func (a AESCrypt) aesEncryptCBC(originData []byte) (encrypted []byte, err error) {
var myCipher cipher.Block
myCipher, err = aes.NewCipher(a.Key)
if err != nil {
return
}
blockSize := myCipher.BlockSize() // 获取秘钥块的长度
originData = pkcs5Padding(originData, blockSize) // 补全码
blockMode := cipher.NewCBCEncrypter(myCipher, a.IV) // 加密模式
encrypted = make([]byte, len(originData)) // 创建数组
blockMode.CryptBlocks(encrypted, originData)
return
}
func pkcs5Padding(ciphertext []byte, blockSize int) []byte {
padding := blockSize - len(ciphertext)%blockSize
padText := bytes.Repeat([]byte{byte(padding)}, padding)
return append(ciphertext, padText...)
}
func converArr1(array [20]byte) []byte {
s := []byte{}
for _, elem := range array {
s = append(s, elem)
}
return s
}
func converArr2(array [16]byte) []byte {
s := []byte{}
for _, elem := range array {
s = append(s, elem)
}
return s
}
func GetByteArray(src []byte, destLen int) []byte {
dest := make([]byte, destLen)
p := 0
for p < destLen {
for _, b := range src {
if p >= destLen {
return dest
} else {
dest[p] = b
p++
}
}
}
return dest
}