-
Notifications
You must be signed in to change notification settings - Fork 68
/
aes.go
51 lines (48 loc) · 1.05 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
package dongle
import (
"crypto/aes"
)
// ByAes encrypts by aes.
// 通过 aes 加密
func (e Encrypter) ByAes(c *Cipher) Encrypter {
if len(e.src) == 0 || e.Error != nil {
return e
}
block, err := aes.NewCipher(c.key)
if err != nil {
e.Error = invalidAesKeyError()
return e
}
if c.mode != ECB && len(c.iv) != block.BlockSize() {
e.Error = invalidAesIVError()
return e
}
if c.padding == No && len(e.src)%block.BlockSize() != 0 {
e.Error = invalidAesSrcError()
return e
}
e.dst, e.Error = c.Encrypt(e.src, block)
return e
}
// ByAes decrypts by aes.
// 通过 aes 解密
func (d Decrypter) ByAes(c *Cipher) Decrypter {
if len(d.src) == 0 || d.Error != nil {
return d
}
block, err := aes.NewCipher(c.key)
if err != nil {
d.Error = invalidAesKeyError()
return d
}
if c.mode != ECB && len(c.iv) != block.BlockSize() {
d.Error = invalidAesIVError()
return d
}
if (c.mode == CBC || c.padding == No) && len(d.src)%block.BlockSize() != 0 {
d.Error = invalidAesSrcError()
return d
}
d.dst, d.Error = c.Decrypt(d.src, block)
return d
}