aarifullin/neoneo-go
1
0
Fork 0
forked from TrueCloudLab/neoneo-go
Code Pull requests Activity

core: export (*NEOBalanceState).FromStackItem

Browse source
This commit is contained in:
Anna Shaleva 2021-05-31 10:39:27 +03:00
parent 9340a70aa8
commit 999fba81cd
1 changed files with 21 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
pkg/core/state/native_state.go
View file

@ -2,6 +2,8 @@ package state
import ( import (
"crypto/elliptic" "crypto/elliptic"
"errors"
"fmt"
"math/big" "math/big"
"github.com/nspcc-dev/neo-go/pkg/crypto/keys" "github.com/nspcc-dev/neo-go/pkg/crypto/keys"
@ -105,7 +107,7 @@ func (s *NEOBalanceState) DecodeBinary(r *io.BinReader) {
if r.Err != nil { if r.Err != nil {
return return
} }
r.Err = s.fromStackItem(si) r.Err = s.FromStackItem(si)
} }
func (s *NEOBalanceState) toStackItem() stackitem.Item { func (s *NEOBalanceState) toStackItem() stackitem.Item {
@ -119,21 +121,33 @@ func (s *NEOBalanceState) toStackItem() stackitem.Item {
return result return result
} }
func (s *NEOBalanceState) fromStackItem(item stackitem.Item) error { // FromStackItem converts stackitem.Item to NEOBalanceState.
structItem := item.Value().([]stackitem.Item) func (s *NEOBalanceState) FromStackItem(item stackitem.Item) error {
s.Balance = *structItem[0].Value().(*big.Int) structItem, ok := item.Value().([]stackitem.Item)
s.BalanceHeight = uint32(structItem[1].Value().(*big.Int).Int64()) if !ok || len(structItem) < 3 {
return errors.New("invalid stackitem length")
}
balance, err := structItem[0].TryInteger()
if err != nil {
return fmt.Errorf("invalid balance stackitem: %w", err)
}
s.Balance = *balance
h, err := structItem[1].TryInteger()
if err != nil {
return fmt.Errorf("invalid heigh stackitem")
}
s.BalanceHeight = uint32(h.Int64())
if _, ok := structItem[2].(stackitem.Null); ok { if _, ok := structItem[2].(stackitem.Null); ok {
s.VoteTo = nil s.VoteTo = nil
return nil return nil
} }
bs, err := structItem[2].TryBytes() bs, err := structItem[2].TryBytes()
if err != nil { if err != nil {
return err return fmt.Errorf("invalid public key stackitem: %w", err)
} }
pub, err := keys.NewPublicKeyFromBytes(bs, elliptic.P256()) pub, err := keys.NewPublicKeyFromBytes(bs, elliptic.P256())
if err != nil { if err != nil {
return err return fmt.Errorf("invalid public key bytes: %w", err)
} }
s.VoteTo = pub s.VoteTo = pub
return nil return nil