add sample of recovering secret key for Go

2 files changed, 68 insertions, 0 deletions

diff --git a/go/bls/bls.go b/go/bls/bls.go
index 5ba2f24..c7749f5 100644
--- a/go/bls/bls.go
+++ b/go/bls/bls.go
@@ -94,6 +94,38 @@ func (sec *SecretKey) Add(rhs *SecretKey) {
 	C.blsSecretKeyAdd(sec.self, rhs.self);
 }
 
+func (sec *SecretKey) GetMasterSecretKey(k int) (msk []SecretKey) {
+	msk = make([]SecretKey, k)
+	msk[0] = *sec
+	for i := 1; i < k; i++ {
+		msk[i] = *NewSecretKey()
+		msk[i].Init()
+	}
+	return msk
+}
+
+func (sec *SecretKey) Set(msk []SecretKey, id Id) {
+	n := len(msk)
+	v := make([]*C.blsSecretKey, n)
+	for i := 0; i < n; i++ {
+		v[i] = msk[i].self
+	}
+	C.blsSecretKeySet(sec.self, (**C.blsSecretKey)(unsafe.Pointer(&v[0])), C.size_t(n), id.self)
+}
+
+func (sec *SecretKey) Recover(secVec []SecretKey, idVec []Id) {
+	n := len(secVec)
+	sv := make([]*C.blsSecretKey, n)
+	for i := 0; i < n; i++ {
+		sv[i] = secVec[i].self
+	}
+	iv := make([]*C.blsId, n)
+	for i := 0; i < n; i++ {
+		iv[i] = idVec[i].self
+	}
+	C.blsSecretKeyRecover(sec.self, (**C.blsSecretKey)(unsafe.Pointer(&sv[0])), (**C.blsId)(unsafe.Pointer(&iv[0])), C.size_t(n))
+}
+
 type PublicKey struct {
 	self *C.blsPublicKey
 }
diff --git a/go/main.go b/go/main.go
index a43d9ac..e769aea 100644
--- a/go/main.go
+++ b/go/main.go
@@ -3,6 +3,31 @@ package main
 import "fmt"
 import "./bls"
 
+func testRecoverSecretKey() {
+	fmt.Println("testRecoverSecretKey")
+	k := 5
+	sec := bls.NewSecretKey()
+	sec.Init()
+	fmt.Println("sec =", sec)
+
+	// make master secret key
+	msk := sec.GetMasterSecretKey(k)
+
+	n := k
+	secVec := make([]bls.SecretKey, n)
+	idVec := make([]bls.Id, n)
+	for i := 0; i < n; i++ {
+		idVec[i] = *bls.NewId()
+		idVec[i].Set([]uint64{1, 2, 3, uint64(i)})
+		secVec[i] = *bls.NewSecretKey()
+		secVec[i].Set(msk, idVec[i])
+	}
+	// recover sec2 from secVec and idVec
+	sec2 := bls.NewSecretKey()
+	sec2.Recover(secVec, idVec)
+	fmt.Println("sec2=", sec2)
+}
+
 func main() {
 	fmt.Println("init")
 	bls.Init()
@@ -26,4 +51,15 @@ func main() {
 	sign := sec.Sign(m)
 	fmt.Println("sign:", sign)
 	fmt.Println("verify:", sign.Verify(pub, m))
+
+	// How to make array of SecretKey
+	{
+		sec := make([]bls.SecretKey, 3)
+		for i := 0; i < len(sec); i++ {
+			sec[i] = *bls.NewSecretKey()
+			sec[i].Init()
+			fmt.Println("sec=", sec[i].String())
+		}
+	}
+	testRecoverSecretKey()
 }