mirror/gmsm
1
0
Fork 0
mirror of https://github.com/emmansun/gmsm.git synced 2025-08-29 19:53:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

sm9: add back SetMasterPublic methods

Browse Source
This commit is contained in:
Sun Yimin 2025-08-29 10:34:47 +08:00
parent 8d98e3bd90
commit 86cfa50941
2 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
sm9/sm9_key.go
View File

@ -235,6 +235,13 @@ func (priv *SignPrivateKey) MasterPublic() *SignMasterPublicKey {
return &SignMasterPublicKey{internal: masterKey, publicKey: masterKey.Bytes()}
}
// SetMasterPublic sets the master public key for the SignPrivateKey.
// The caller should ensure that the provided master public key is valid and corresponds
// to the SignPrivateKey. This method is NOT safe for concurrent use.
func (priv *SignPrivateKey) SetMasterPublic(master *SignMasterPublicKey) {
priv.internal.SetMasterPublicKey(master.internal)
}
// MarshalASN1 marshal signature private key to asn.1 format data according
// SM9 cryptographic algorithm application specification
func (priv *SignPrivateKey) MarshalASN1() ([]byte, error) {
@ -463,6 +470,13 @@ func (priv *EncryptPrivateKey) MasterPublic() *EncryptMasterPublicKey {
return &EncryptMasterPublicKey{publicKey: master.Bytes(), internal: master}
}
// SetMasterPublic sets the master public key for the EncryptPrivateKey.
// The caller should ensure that the provided master public key is valid and corresponds
// to the EncryptPrivateKey. This method is NOT safe for concurrent use.
func (priv *EncryptPrivateKey) SetMasterPublic(master *EncryptMasterPublicKey) {
priv.internal.SetMasterPublicKey(master.internal)
}
// MarshalASN1 marshal encryption private key to asn.1 format data according
// SM9 cryptographic algorithm application specification
func (priv *EncryptPrivateKey) MarshalASN1() ([]byte, error) {

14
sm9/sm9_test.go
View File

@ -21,6 +21,13 @@ func TestSignASN1(t *testing.T) {
if err != nil {
t.Fatal(err)
}
// Test Marshal and Unmarshal
userKeyBytes := userKey.Bytes()
userKey, err = sm9.UnmarshalSignPrivateKeyRaw(userKeyBytes)
if err != nil {
t.Fatal(err)
}
userKey.SetMasterPublic(masterKey.PublicKey())
sig, err := userKey.Sign(rand.Reader, hashed, nil)
if err != nil {
t.Fatal(err)
@ -103,6 +110,13 @@ func TestEncryptDecrypt(t *testing.T) {
if err != nil {
t.Fatal(err)
}
// Test Marshal and Unmarshal
userKeyBytes := userKey.Bytes()
userKey, err = sm9.UnmarshalEncryptPrivateKeyRaw(userKeyBytes)
if err != nil {
t.Fatal(err)
}
userKey.SetMasterPublic(masterKey.PublicKey())
encTypes := []sm9.EncrypterOpts{
sm9.DefaultEncrypterOpts, sm9.SM4ECBEncrypterOpts, sm9.SM4CBCEncrypterOpts, sm9.SM4CFBEncrypterOpts, sm9.SM4OFBEncrypterOpts,
}