/
signer_rsa.go
91 lines (71 loc) · 2.03 KB
/
signer_rsa.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
package pkcs7
import (
"errors"
"crypto"
"crypto/rsa"
"crypto/rand"
"encoding/asn1"
)
// rsa 签名
type KeySignWithRSA struct {
isRSAPSS bool
hashFunc crypto.Hash
hashId asn1.ObjectIdentifier
identifier asn1.ObjectIdentifier
}
// oid
func (this KeySignWithRSA) HashOID() asn1.ObjectIdentifier {
return this.hashId
}
// oid
func (this KeySignWithRSA) OID() asn1.ObjectIdentifier {
return this.identifier
}
// 签名
func (this KeySignWithRSA) Sign(pkey crypto.PrivateKey, data []byte) ([]byte, []byte, error) {
var priv *rsa.PrivateKey
var ok bool
if priv, ok = pkey.(*rsa.PrivateKey); !ok {
return nil, nil, errors.New("pkcs7: PrivateKey is not rsa PrivateKey")
}
hashType := this.hashFunc
hashData := hashSignData(hashType, data)
var signData []byte
var err error
if this.isRSAPSS {
signData, err = rsa.SignPSS(rand.Reader, priv, hashType, hashData, &rsa.PSSOptions{SaltLength: rsa.PSSSaltLengthEqualsHash})
} else {
signData, err = rsa.SignPKCS1v15(rand.Reader, priv, hashType, hashData)
}
return hashData, signData, err
}
// 验证
func (this KeySignWithRSA) Verify(pkey crypto.PublicKey, data []byte, signature []byte) (bool, error) {
var pub *rsa.PublicKey
var ok bool
if pub, ok = pkey.(*rsa.PublicKey); !ok {
return false, errors.New("pkcs7: PublicKey is not rsa PublicKey")
}
hashType := this.hashFunc
hashData := hashSignData(hashType, data)
var err error
if this.isRSAPSS {
err = rsa.VerifyPSS(pub, hashType, hashData, signature, &rsa.PSSOptions{SaltLength: rsa.PSSSaltLengthEqualsHash})
} else {
err = rsa.VerifyPKCS1v15(pub, hashType, hashData, signature)
}
if err != nil {
return false, err
}
return true, nil
}
// 检测证书
func (this KeySignWithRSA) Check(pkey any) bool {
if _, ok := pkey.(*rsa.PrivateKey); ok {
return true
}
if _, ok := pkey.(*rsa.PublicKey); ok {
return true
}
return false
}