1 files changed, 131 insertions, 22 deletions

diff --git a/go/main.go b/go/main.go
index 6852d60..c75c7df 100644
--- a/go/main.go
+++ b/go/main.go
@@ -47,9 +47,131 @@ func (id *BlsId) setStr(s string) {
 }
 
 func (id *BlsId) set(v []uint64) {
+	if len(v) != 4 {
+		fmt.Println("BlsId:set bad size", len(v))
+		return
+	}
 	C.blsIdSet(id.self, (*C.uint64_t)(unsafe.Pointer(&v[0])))
 }
 
+type BlsSecretKey struct {
+	self *C.blsSecretKey
+}
+
+func destroyBlsSecretKey(p *BlsSecretKey) {
+	C.blsSecretKeyDestroy(p.self)
+}
+
+func newBlsSecretKey() *BlsSecretKey {
+	p := new(BlsSecretKey)
+	p.self = C.blsSecretKeyCreate()
+	runtime.SetFinalizer(p, destroyBlsSecretKey)
+	return p
+}
+
+func (sec *BlsSecretKey) String() string {
+	buf := make([]byte, 1024)
+	n := C.blsSecretKeyGetStr(sec.self, (*C.char)(unsafe.Pointer(&buf[0])), C.size_t(len(buf)))
+	if n == 0 {
+		return "err"
+	}
+	return string(buf[:n])
+}
+
+func (sec *BlsSecretKey) setStr(s string) {
+	buf := []byte(s)
+	err := C.blsSecretKeySetStr(sec.self, (*C.char)(unsafe.Pointer(&buf[0])), C.size_t(len(buf)))
+	if err > 0 {
+		fmt.Println("BlsSecretKey:SetStr", err)
+	}
+}
+
+func (sec *BlsSecretKey) init() {
+	C.blsSecretKeyInit(sec.self)
+}
+
+type BlsPublicKey struct {
+	self *C.blsPublicKey
+}
+
+func destroyBlsPublicKey(p *BlsPublicKey) {
+	C.blsPublicKeyDestroy(p.self)
+}
+
+func newBlsPublicKey() *BlsPublicKey {
+	p := new(BlsPublicKey)
+	p.self = C.blsPublicKeyCreate()
+	runtime.SetFinalizer(p, destroyBlsPublicKey)
+	return p
+}
+
+func (pub *BlsPublicKey) String() string {
+	buf := make([]byte, 1024)
+	n := C.blsPublicKeyGetStr(pub.self, (*C.char)(unsafe.Pointer(&buf[0])), C.size_t(len(buf)))
+	if n == 0 {
+		return "err"
+	}
+	return string(buf[:n])
+}
+
+func (pub *BlsPublicKey) setStr(s string) {
+	buf := []byte(s)
+	err := C.blsPublicKeySetStr(pub.self, (*C.char)(unsafe.Pointer(&buf[0])), C.size_t(len(buf)))
+	if err > 0 {
+		fmt.Println("BlsPublicKey:SetStr", err)
+	}
+}
+
+type BlsSign struct {
+	self *C.blsSign
+}
+
+func destroyBlsSign(p *BlsSign) {
+	C.blsSignDestroy(p.self)
+}
+
+func newBlsSign() *BlsSign {
+	p := new(BlsSign)
+	p.self = C.blsSignCreate()
+	runtime.SetFinalizer(p, destroyBlsSign)
+	return p
+}
+
+func (sign *BlsSign) String() string {
+	buf := make([]byte, 1024)
+	n := C.blsSignGetStr(sign.self, (*C.char)(unsafe.Pointer(&buf[0])), C.size_t(len(buf)))
+	if n == 0 {
+		return "err"
+	}
+	return string(buf[:n])
+}
+
+func (sign *BlsSign) setStr(s string) {
+	buf := []byte(s)
+	err := C.blsSignSetStr(sign.self, (*C.char)(unsafe.Pointer(&buf[0])), C.size_t(len(buf)))
+	if err > 0 {
+		fmt.Println("BlsSign:SetStr", err)
+	}
+}
+
+func (sec *BlsSecretKey) getPublicKey() (pub *BlsPublicKey) {
+	pub = newBlsPublicKey()
+	C.blsSecretKeyGetPublicKey(sec.self, pub.self)
+	return pub
+}
+
+func (sec *BlsSecretKey) sign(m string) (sign *BlsSign) {
+	sign = newBlsSign()
+	buf := []byte(m)
+	C.blsSecretKeySign(sec.self, sign.self, (*C.char)(unsafe.Pointer(&buf[0])), C.size_t(len(buf)))
+	return sign
+}
+
+func (sign *BlsSign) verify(pub *BlsPublicKey, m string) bool {
+	buf := []byte(m)
+	return C.blsSignVerify(sign.self, pub.self, (*C.char)(unsafe.Pointer(&buf[0])), C.size_t(len(buf))) == 1
+}
+
 func main() {
 	fmt.Println("init")
 	BlsInit()
@@ -63,27 +185,14 @@ func main() {
 	}
 
 	fmt.Println("create secret key")
-	sec := C.blsSecretKeyCreate()
-	defer C.blsSecretKeyDestroy(sec)
-	C.blsSecretKeyInit(sec)
-	C.blsSecretKeyPut(sec)
-
+	m := "this is a bls sample for go"
+	sec := newBlsSecretKey()
+	sec.init()
+	fmt.Println("sec:", sec)
 	fmt.Println("create public key")
-	pub := C.blsPublicKeyCreate()
-	defer C.blsPublicKeyDestroy(pub)
-	C.blsSecretKeyGetPublicKey(sec, pub)
-
-	C.blsPublicKeyPut(pub)
-
-	sign := C.blsSignCreate()
-	defer C.blsSignDestroy(sign)
-
-	msg := []byte("Hello bls")
-	fmt.Println("sign message")
-	C.blsSecretKeySign(sec, sign, (*C.char)(unsafe.Pointer(&msg[0])), C.size_t(len(msg)))
-
-	C.blsSignPut(sign)
-
-	fmt.Println("verify:", C.blsSignVerify(sign, pub, (*C.char)(unsafe.Pointer(&msg[0])), C.size_t(len(msg))))
-
+	pub := sec.getPublicKey()
+	fmt.Println("pub:", pub)
+	sign := sec.sign(m)
+	fmt.Println("sign:", sign)
+	fmt.Println("verify:", sign.verify(pub, m))
 }