From 66eafcd558f6ab53b12a1b85012b18bddc4e82b8 Mon Sep 17 00:00:00 2001 From: Alex Vanin Date: Tue, 8 Sep 2020 11:37:58 +0300 Subject: [PATCH] [#21] Add neofs bind event Signed-off-by: Alex Vanin --- pkg/morph/event/neofs/bind.go | 69 +++++++++++++++++++++ pkg/morph/event/neofs/bind_test.go | 97 ++++++++++++++++++++++++++++++ 2 files changed, 166 insertions(+) create mode 100644 pkg/morph/event/neofs/bind.go create mode 100644 pkg/morph/event/neofs/bind_test.go diff --git a/pkg/morph/event/neofs/bind.go b/pkg/morph/event/neofs/bind.go new file mode 100644 index 000000000..27574b614 --- /dev/null +++ b/pkg/morph/event/neofs/bind.go @@ -0,0 +1,69 @@ +package neofs + +import ( + "crypto/elliptic" + + "github.com/nspcc-dev/neo-go/pkg/crypto/keys" + "github.com/nspcc-dev/neo-go/pkg/smartcontract" + "github.com/nspcc-dev/neo-go/pkg/util" + "github.com/nspcc-dev/neofs-node/pkg/morph/client" + "github.com/nspcc-dev/neofs-node/pkg/morph/event" + "github.com/pkg/errors" +) + +type Bind struct { + user util.Uint160 + keys []*keys.PublicKey +} + +// MorphEvent implements Neo:Morph Event interface. +func (Bind) MorphEvent() {} + +func (b Bind) Keys() []*keys.PublicKey { return b.keys } + +func (b Bind) User() util.Uint160 { return b.user } + +func ParseBind(params []smartcontract.Parameter) (event.Event, error) { + var ( + ev Bind + err error + ) + + if ln := len(params); ln != 2 { + return nil, event.WrongNumberOfParameters(2, ln) + } + + // parse user + user, err := client.BytesFromStackParameter(params[0]) + if err != nil { + return nil, errors.Wrap(err, "could not get bind user") + } + + ev.user, err = util.Uint160DecodeBytesBE(user) + if err != nil { + return nil, errors.Wrap(err, "could not convert bind user to uint160") + } + + // parse keys + bindKeys, err := client.ArrayFromStackParameter(params[1]) + if err != nil { + return nil, errors.Wrap(err, "could not get bind keys") + } + + ev.keys = make([]*keys.PublicKey, 0, len(bindKeys)) + for i := range bindKeys { + rawKey, err := client.BytesFromStackParameter(bindKeys[i]) + if err != nil { + return nil, errors.Wrap(err, "could not get bind public key") + } + + key, err := keys.NewPublicKeyFromBytes(rawKey, elliptic.P256()) + if err != nil { + return nil, errors.Wrap(err, "could not parse bind public key") + } + + ev.keys = append(ev.keys, key) + } + + return ev, nil +} diff --git a/pkg/morph/event/neofs/bind_test.go b/pkg/morph/event/neofs/bind_test.go new file mode 100644 index 000000000..6761b2eeb --- /dev/null +++ b/pkg/morph/event/neofs/bind_test.go @@ -0,0 +1,97 @@ +package neofs + +import ( + "crypto/ecdsa" + "crypto/elliptic" + "testing" + + "github.com/nspcc-dev/neo-go/pkg/crypto/keys" + "github.com/nspcc-dev/neo-go/pkg/smartcontract" + "github.com/nspcc-dev/neo-go/pkg/util" + crypto "github.com/nspcc-dev/neofs-crypto" + "github.com/nspcc-dev/neofs-node/pkg/morph/event" + "github.com/nspcc-dev/neofs-node/pkg/util/test" + "github.com/stretchr/testify/require" +) + +func TestParseBind(t *testing.T) { + var ( + user = util.Uint160{0x1, 0x2, 0x3} + publicKeys = []*ecdsa.PublicKey{ + &test.DecodeKey(1).PublicKey, + &test.DecodeKey(2).PublicKey, + &test.DecodeKey(3).PublicKey, + } + ) + + t.Run("wrong number of parameters", func(t *testing.T) { + prms := []smartcontract.Parameter{ + {}, + } + + _, err := ParseBind(prms) + require.EqualError(t, err, event.WrongNumberOfParameters(2, len(prms)).Error()) + }) + + t.Run("wrong first parameter", func(t *testing.T) { + _, err := ParseBind([]smartcontract.Parameter{ + { + Type: smartcontract.ArrayType, + }, + }) + + require.Error(t, err) + }) + + t.Run("wrong second parameter", func(t *testing.T) { + _, err := ParseBind([]smartcontract.Parameter{ + { + Type: smartcontract.ByteArrayType, + Value: user.BytesBE(), + }, + { + Type: smartcontract.ArrayType, + }, + }) + + require.Error(t, err) + }) + + t.Run("correct", func(t *testing.T) { + ev, err := ParseBind([]smartcontract.Parameter{ + { + Type: smartcontract.ByteArrayType, + Value: user.BytesBE(), + }, + { + Type: smartcontract.ArrayType, + Value: []smartcontract.Parameter{ + { + Type: smartcontract.ByteArrayType, + Value: crypto.MarshalPublicKey(publicKeys[0]), + }, + { + Type: smartcontract.ByteArrayType, + Value: crypto.MarshalPublicKey(publicKeys[1]), + }, + { + Type: smartcontract.ByteArrayType, + Value: crypto.MarshalPublicKey(publicKeys[2]), + }, + }, + }, + }) + require.NoError(t, err) + + expKeys := make([]*keys.PublicKey, len(publicKeys)) + for i := range publicKeys { + expKeys[i], err = keys.NewPublicKeyFromBytes(crypto.MarshalPublicKey(publicKeys[i]), elliptic.P256()) + require.NoError(t, err) + } + + require.Equal(t, Bind{ + user: user, + keys: expKeys, + }, ev) + }) +}