mirror/gmsm
1
0
Fork 0
mirror of https://github.com/emmansun/gmsm.git synced 2025-04-26 04:06:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

pkcs7: update comments #276

Browse Source
This commit is contained in:
Sun Yimin 2024-11-21 17:53:35 +08:00 committed by GitHub
parent bc0e11e9b1
commit 5c2a22ec2a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkcs7/envelope.go
View File

@ -210,7 +210,7 @@ func newEnvelopedData(cipher pkcs.Cipher, content []byte, contentType asn1.Objec
// AddRecipient adds a recipient to the EnvelopedData structure. // AddRecipient adds a recipient to the EnvelopedData structure.
// version 0: IssuerAndSerialNumber // version 0: IssuerAndSerialNumber
// version 1: SM2 // version 1: SM2GB/T 35275-2017
// version 2: SubjectKeyIdentifier // version 2: SubjectKeyIdentifier
func (ed *EnvelopedData) AddRecipient(cert *smx509.Certificate, version int, encryptKeyFunc func(cert *smx509.Certificate, key []byte) ([]byte, error)) error { func (ed *EnvelopedData) AddRecipient(cert *smx509.Certificate, version int, encryptKeyFunc func(cert *smx509.Certificate, key []byte) ([]byte, error)) error {
if version < 0 || version > 2 { if version < 0 || version > 2 {

13
pkcs7/session.go
View File

@ -1,3 +1,7 @@
// Copyright 2024 Sun Yimin. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package pkcs7 package pkcs7
import ( import (
@ -11,6 +15,7 @@ import (
"github.com/emmansun/gmsm/smx509" "github.com/emmansun/gmsm/smx509"
) )
// Session is an interface that provides methods to generate and encrypt/decrypt data keys
type Session interface { type Session interface {
// GenerateDataKey returns the data key to be used for encryption // GenerateDataKey returns the data key to be used for encryption
GenerateDataKey(size int) ([]byte, error) GenerateDataKey(size int) ([]byte, error)
@ -22,11 +27,11 @@ type Session interface {
DecryptDataKey(key []byte, priv crypto.PrivateKey, cert *smx509.Certificate, opts any) ([]byte, error) DecryptDataKey(key []byte, priv crypto.PrivateKey, cert *smx509.Certificate, opts any) ([]byte, error)
} }
// DefaultSession is the default implementation of Session without any special handling // DefaultSession is the default implementation of Session without any special handling (stateless).
// Custom implementations can be provided to handle key reuse, cache, etc. // Custom implementations can be provided to handle key reuse, cache, etc.
type DefaultSession struct{} type DefaultSession struct{}
func (d DefaultSession) GenerateDataKey(size int) ([]byte, error) { func (DefaultSession) GenerateDataKey(size int) ([]byte, error) {
key := make([]byte, size) key := make([]byte, size)
if _, err := rand.Read(key); err != nil { if _, err := rand.Read(key); err != nil {
return nil, err return nil, err
@ -34,7 +39,7 @@ func (d DefaultSession) GenerateDataKey(size int) ([]byte, error) {
return key, nil return key, nil
} }
func (d DefaultSession) EncryptdDataKey(key []byte, cert *smx509.Certificate, opts any) ([]byte, error) { func (DefaultSession) EncryptdDataKey(key []byte, cert *smx509.Certificate, opts any) ([]byte, error) {
switch pub := cert.PublicKey.(type) { switch pub := cert.PublicKey.(type) {
case *rsa.PublicKey: case *rsa.PublicKey:
return rsa.EncryptPKCS1v15(rand.Reader, pub, key) return rsa.EncryptPKCS1v15(rand.Reader, pub, key)
@ -54,7 +59,7 @@ func (d DefaultSession) EncryptdDataKey(key []byte, cert *smx509.Certificate, op
return nil, errors.New("pkcs7: only supports RSA/SM2 key") return nil, errors.New("pkcs7: only supports RSA/SM2 key")
} }
func (d DefaultSession) DecryptDataKey(key []byte, priv crypto.PrivateKey, cert *smx509.Certificate, opts any) ([]byte, error) { func (DefaultSession) DecryptDataKey(key []byte, priv crypto.PrivateKey, cert *smx509.Certificate, opts any) ([]byte, error) {
switch pkey := priv.(type) { switch pkey := priv.(type) {
case crypto.Decrypter: case crypto.Decrypter:
// Generic case to handle anything that provides the crypto.Decrypter interface. // Generic case to handle anything that provides the crypto.Decrypter interface.