add methods to get master public key

opensvn · Jun 16, 2022 · ca87062 · ca87062
1 parent 5c6c980
commit ca87062
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/kgc.go b/kgc.go
@@ -28,10 +28,18 @@ func New() *Kgc {
 	}
 }
 
+func (k *Kgc) GetSignMasterPublicKey() *sm9.SignMasterPublicKey {
+	return k.signMasterKey.Public()
+}
+
 func (k *Kgc) GenerateUserSignKey(uid []byte, hid byte) (*sm9.SignPrivateKey, error) {
 	return k.signMasterKey.GenerateUserKey(uid, hid)
 }
 
+func (k *Kgc) GetEncryptMasterPublicKey() *sm9.EncryptMasterPublicKey {
+	return k.encryptMasterKey.Public()
+}
+
 func (k *Kgc) GenerateUserEncryptKey(uid []byte, hid byte) (*sm9.EncryptPrivateKey, error) {
 	return k.encryptMasterKey.GenerateUserKey(uid, hid)
 }
diff --git a/kgc_test.go b/kgc_test.go
@@ -17,6 +17,18 @@ func TestNewKgc(t *testing.T) {
 	}
 }
 
+func TestGetSignMasterPublicKey(t *testing.T) {
+	kgc := New()
+	if kgc == nil {
+		t.Error("kgc is nil")
+	}
+
+	signMasterPublicKey := kgc.GetSignMasterPublicKey()
+	if signMasterPublicKey == nil {
+		t.Error("signMasterPublicKey is nil")
+	}
+}
+
 func TestGenerateUserSignKey(t *testing.T) {
 	kgc := New()
 	if kgc == nil {
@@ -35,6 +47,18 @@ func TestGenerateUserSignKey(t *testing.T) {
 	}
 }
 
+func TestGetEncryptMasterPublicKey(t *testing.T) {
+	kgc := New()
+	if kgc == nil {
+		t.Error("kgc is nil")
+	}
+
+	encryptMasterPublicKey := kgc.GetEncryptMasterPublicKey()
+	if encryptMasterPublicKey == nil {
+		t.Error("signMasterPublicKey is nil")
+	}
+}
+
 func TestGenerateUserEncryptKey(t *testing.T) {
 	kgc := New()
 	if kgc == nil {