/
sm4.go
72 lines (57 loc) · 2.07 KB
/
sm4.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
/*
Copyright Suzhou Tongji Fintech Research Institute 2017 All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package sw
import (
"github.com/hyperledger/fabric/bccsp"
"github.com/tjfoc/gmsm/sm4"
)
// AESCBCPKCS7Encrypt combines CBC encryption and PKCS7 padding
func SM4Encrypt(key, src []byte) ([]byte, error) {
// // First pad
// tmp := pkcs7Padding(src)
// // Then encrypt
// return aesCBCEncrypt(key, tmp)
dst := make([]byte, len(src))
sm4.EncryptBlock(key, dst, src)
return dst, nil
}
// AESCBCPKCS7Decrypt combines CBC decryption and PKCS7 unpadding
func SM4Decrypt(key, src []byte) ([]byte, error) {
// First decrypt
// pt, err := aesCBCDecrypt(key, src)
// if err == nil {
// return pkcs7UnPadding(pt)
// }
dst := make([]byte, len(src))
sm4.DecryptBlock(key, dst, src)
return dst, nil
}
type gmsm4Encryptor struct{}
//实现 Encryptor 接口
func (*gmsm4Encryptor) Encrypt(k bccsp.Key, plaintext []byte, opts bccsp.EncrypterOpts) (ciphertext []byte, err error) {
return SM4Encrypt(k.(*gmsm4PrivateKey).privKey, plaintext)
//return AESCBCPKCS7Encrypt(k.(*sm4PrivateKey).privKey, plaintext)
// key := k.(*gmsm4PrivateKey).privKey
// var en = make([]byte, 16)
// sms4(plaintext, 16, key, en, 1)
// return en, nil
}
type gmsm4Decryptor struct{}
//实现 Decryptor 接口
func (*gmsm4Decryptor) Decrypt(k bccsp.Key, ciphertext []byte, opts bccsp.DecrypterOpts) (plaintext []byte, err error) {
return SM4Decrypt(k.(*gmsm4PrivateKey).privKey, ciphertext)
// var dc = make([]byte, 16)
// key := k.(*gmsm4PrivateKey).privKey
// sms4(ciphertext, 16, key, dc, 0)
// return dc, nil
}