state: store notary deposit as stackitem
Which is less effective, but makes it more similar to other native contracts that are supposed to be contracts anyway.
This commit is contained in:
parent
70ddbf7180
commit
2d993d0da5
2 changed files with 101 additions and 5 deletions
|
@ -1,10 +1,13 @@
|
||||||
package state
|
package state
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"math"
|
||||||
"math/big"
|
"math/big"
|
||||||
|
|
||||||
"github.com/nspcc-dev/neo-go/pkg/encoding/bigint"
|
|
||||||
"github.com/nspcc-dev/neo-go/pkg/io"
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
||||||
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Deposit represents GAS deposit from Notary contract.
|
// Deposit represents GAS deposit from Notary contract.
|
||||||
|
@ -15,12 +18,46 @@ type Deposit struct {
|
||||||
|
|
||||||
// EncodeBinary implements io.Serializable interface.
|
// EncodeBinary implements io.Serializable interface.
|
||||||
func (d *Deposit) EncodeBinary(w *io.BinWriter) {
|
func (d *Deposit) EncodeBinary(w *io.BinWriter) {
|
||||||
w.WriteVarBytes(bigint.ToBytes(d.Amount))
|
stackitem.EncodeBinary(d.toStackItem(), w)
|
||||||
w.WriteU32LE(d.Till)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// DecodeBinary implements io.Serializable interface.
|
// DecodeBinary implements io.Serializable interface.
|
||||||
func (d *Deposit) DecodeBinary(r *io.BinReader) {
|
func (d *Deposit) DecodeBinary(r *io.BinReader) {
|
||||||
d.Amount = bigint.FromBytes(r.ReadVarBytes())
|
si := stackitem.DecodeBinary(r)
|
||||||
d.Till = r.ReadU32LE()
|
if r.Err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
r.Err = d.fromStackItem(si)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Deposit) toStackItem() stackitem.Item {
|
||||||
|
return stackitem.NewStruct([]stackitem.Item{
|
||||||
|
stackitem.NewBigInteger(d.Amount),
|
||||||
|
stackitem.Make(d.Till),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *Deposit) fromStackItem(it stackitem.Item) error {
|
||||||
|
items, ok := it.Value().([]stackitem.Item)
|
||||||
|
if !ok {
|
||||||
|
return errors.New("not a struct")
|
||||||
|
}
|
||||||
|
if len(items) != 2 {
|
||||||
|
return errors.New("wrong number of elements")
|
||||||
|
}
|
||||||
|
amount, err := items[0].TryInteger()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("invalid amount: %w", err)
|
||||||
|
}
|
||||||
|
till, err := items[1].TryInteger()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("invalid till: %w", err)
|
||||||
|
}
|
||||||
|
ti64 := till.Int64()
|
||||||
|
if !till.IsInt64() || ti64 > math.MaxUint32 || ti64 < 0 {
|
||||||
|
return errors.New("wrong till value")
|
||||||
|
}
|
||||||
|
d.Amount = amount
|
||||||
|
d.Till = uint32(ti64)
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
59
pkg/core/state/deposit_test.go
Normal file
59
pkg/core/state/deposit_test.go
Normal file
|
@ -0,0 +1,59 @@
|
||||||
|
package state
|
||||||
|
|
||||||
|
import (
|
||||||
|
"math/big"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/nspcc-dev/neo-go/internal/testserdes"
|
||||||
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestEncodeDecodeDeposit(t *testing.T) {
|
||||||
|
d := &Deposit{Amount: big.NewInt(100500), Till: 888}
|
||||||
|
|
||||||
|
depo := new(Deposit)
|
||||||
|
testserdes.EncodeDecodeBinary(t, d, depo)
|
||||||
|
item := stackitem.Make(42)
|
||||||
|
data, err := stackitem.Serialize(item)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Error(t, testserdes.DecodeBinary(data, depo))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDepositFromStackItem(t *testing.T) {
|
||||||
|
var d Deposit
|
||||||
|
|
||||||
|
item := stackitem.Make(42)
|
||||||
|
require.Error(t, d.fromStackItem(item))
|
||||||
|
|
||||||
|
item = stackitem.NewStruct(nil)
|
||||||
|
require.Error(t, d.fromStackItem(item))
|
||||||
|
|
||||||
|
item = stackitem.NewStruct([]stackitem.Item{
|
||||||
|
stackitem.NewStruct(nil),
|
||||||
|
stackitem.NewStruct(nil),
|
||||||
|
})
|
||||||
|
require.Error(t, d.fromStackItem(item))
|
||||||
|
|
||||||
|
item = stackitem.NewStruct([]stackitem.Item{
|
||||||
|
stackitem.Make(777),
|
||||||
|
stackitem.NewStruct(nil),
|
||||||
|
})
|
||||||
|
require.Error(t, d.fromStackItem(item))
|
||||||
|
|
||||||
|
item = stackitem.NewStruct([]stackitem.Item{
|
||||||
|
stackitem.Make(777),
|
||||||
|
stackitem.Make(-1),
|
||||||
|
})
|
||||||
|
require.Error(t, d.fromStackItem(item))
|
||||||
|
item = stackitem.NewStruct([]stackitem.Item{
|
||||||
|
stackitem.Make(777),
|
||||||
|
stackitem.Make("somenonu64value"),
|
||||||
|
})
|
||||||
|
require.Error(t, d.fromStackItem(item))
|
||||||
|
item = stackitem.NewStruct([]stackitem.Item{
|
||||||
|
stackitem.Make(777),
|
||||||
|
stackitem.Make(888),
|
||||||
|
})
|
||||||
|
require.NoError(t, d.fromStackItem(item))
|
||||||
|
}
|
Loading…
Reference in a new issue