libsecp256k1: use a static global context and add verification tests.
This commit is contained in:
+11
-18
@@ -14,45 +14,38 @@ import (
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
type Context struct {
|
||||
ctx *C.secp256k1_context
|
||||
}
|
||||
var globalSecp256k1Context *C.secp256k1_context
|
||||
|
||||
func NewContext() (*Context, error) {
|
||||
ctx := C.secp256k1_context_create(C.SECP256K1_CONTEXT_SIGN | C.SECP256K1_CONTEXT_VERIFY)
|
||||
if ctx == nil {
|
||||
return nil, errors.New("failed to create secp256k1 context")
|
||||
func init() {
|
||||
globalSecp256k1Context = C.secp256k1_context_create(C.SECP256K1_CONTEXT_SIGN | C.SECP256K1_CONTEXT_VERIFY)
|
||||
if globalSecp256k1Context == nil {
|
||||
panic("failed to create secp256k1 context")
|
||||
}
|
||||
return &Context{ctx: ctx}, nil
|
||||
}
|
||||
|
||||
func (c *Context) Destroy() {
|
||||
C.secp256k1_context_destroy(c.ctx)
|
||||
}
|
||||
|
||||
func (c *Context) Sign(msg [32]byte, sk [32]byte) ([64]byte, error) {
|
||||
func Sign(msg [32]byte, sk [32]byte) ([64]byte, error) {
|
||||
var sig [64]byte
|
||||
|
||||
var keypair C.secp256k1_keypair
|
||||
if C.secp256k1_keypair_create(c.ctx, &keypair, (*C.uchar)(unsafe.Pointer(&sk[0]))) != 1 {
|
||||
if C.secp256k1_keypair_create(globalSecp256k1Context, &keypair, (*C.uchar)(unsafe.Pointer(&sk[0]))) != 1 {
|
||||
return sig, errors.New("failed to parse private key")
|
||||
}
|
||||
|
||||
var random [32]byte
|
||||
rand.Read(random[:])
|
||||
|
||||
if C.secp256k1_schnorrsig_sign32(c.ctx, (*C.uchar)(unsafe.Pointer(&sig[0])), (*C.uchar)(unsafe.Pointer(&msg[0])), &keypair, (*C.uchar)(unsafe.Pointer(&random[0]))) != 1 {
|
||||
if C.secp256k1_schnorrsig_sign32(globalSecp256k1Context, (*C.uchar)(unsafe.Pointer(&sig[0])), (*C.uchar)(unsafe.Pointer(&msg[0])), &keypair, (*C.uchar)(unsafe.Pointer(&random[0]))) != 1 {
|
||||
return sig, errors.New("failed to sign message")
|
||||
}
|
||||
|
||||
return sig, nil
|
||||
}
|
||||
|
||||
func (c *Context) Verify(msg [32]byte, sig [64]byte, pk [32]byte) bool {
|
||||
func Verify(msg [32]byte, sig [64]byte, pk [32]byte) bool {
|
||||
var xonly C.secp256k1_xonly_pubkey
|
||||
if C.secp256k1_xonly_pubkey_parse(c.ctx, &xonly, (*C.uchar)(unsafe.Pointer(&pk[0]))) != 1 {
|
||||
if C.secp256k1_xonly_pubkey_parse(globalSecp256k1Context, &xonly, (*C.uchar)(unsafe.Pointer(&pk[0]))) != 1 {
|
||||
return false
|
||||
}
|
||||
|
||||
return C.secp256k1_schnorrsig_verify(c.ctx, (*C.uchar)(unsafe.Pointer(&sig[0])), (*C.uchar)(unsafe.Pointer(&msg[0])), 32, &xonly) == 1
|
||||
return C.secp256k1_schnorrsig_verify(globalSecp256k1Context, (*C.uchar)(unsafe.Pointer(&sig[0])), (*C.uchar)(unsafe.Pointer(&msg[0])), 32, &xonly) == 1
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user