neoneo-go/pkg/core/native/util.go

95 lines
2.4 KiB
Go
Raw Normal View History

2020-10-01 15:17:09 +00:00
package native
import (
2020-11-27 10:55:48 +00:00
"encoding/binary"
"encoding/hex"
"fmt"
"math/big"
2020-11-27 10:55:48 +00:00
2020-10-01 15:17:09 +00:00
"github.com/nspcc-dev/neo-go/pkg/core/dao"
2020-11-19 10:00:46 +00:00
"github.com/nspcc-dev/neo-go/pkg/core/state"
2020-10-01 15:17:09 +00:00
"github.com/nspcc-dev/neo-go/pkg/core/storage"
"github.com/nspcc-dev/neo-go/pkg/encoding/bigint"
2020-10-01 15:17:09 +00:00
"github.com/nspcc-dev/neo-go/pkg/io"
"github.com/nspcc-dev/neo-go/pkg/util"
2021-01-22 12:12:09 +00:00
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
2020-10-01 15:17:09 +00:00
)
var intOne = big.NewInt(1)
2020-10-01 15:17:09 +00:00
func getSerializableFromDAO(id int32, d dao.DAO, key []byte, item io.Serializable) error {
si := d.GetStorageItem(id, key)
if si == nil {
return storage.ErrKeyNotFound
}
r := io.NewBinReaderFromBuf(si)
2020-10-01 15:17:09 +00:00
item.DecodeBinary(r)
return r.Err
}
2020-11-19 10:00:46 +00:00
func putSerializableToDAO(id int32, d dao.DAO, key []byte, item io.Serializable) error {
w := io.NewBufBinWriter()
item.EncodeBinary(w.BinWriter)
if w.Err != nil {
return w.Err
}
return d.PutStorageItem(id, key, w.Bytes())
2020-11-19 10:00:46 +00:00
}
2020-11-27 10:55:48 +00:00
func getInt64WithKey(id int32, d dao.DAO, key []byte, defaultValue int64) int64 {
si := d.GetStorageItem(id, key)
if si == nil {
return defaultValue
}
return int64(binary.LittleEndian.Uint64(si))
2020-11-27 10:55:48 +00:00
}
func setInt64WithKey(id int32, dao dao.DAO, key []byte, value int64) error {
si := make(state.StorageItem, 8)
binary.LittleEndian.PutUint64(si, uint64(value))
2020-11-27 10:55:48 +00:00
return dao.PutStorageItem(id, key, si)
}
func getUint32WithKey(id int32, dao dao.DAO, key []byte, defaultValue uint32) uint32 {
si := dao.GetStorageItem(id, key)
if si == nil {
return defaultValue
}
return binary.LittleEndian.Uint32(si)
2020-11-27 10:55:48 +00:00
}
func setUint32WithKey(id int32, dao dao.DAO, key []byte, value uint32) error {
si := make(state.StorageItem, 4)
binary.LittleEndian.PutUint32(si, value)
2020-11-27 10:55:48 +00:00
return dao.PutStorageItem(id, key, si)
}
func setIntWithKey(id int32, dao dao.DAO, key []byte, value int64) error {
return dao.PutStorageItem(id, key, bigint.ToBytes(big.NewInt(value)))
}
func getIntWithKey(id int32, dao dao.DAO, key []byte) int64 {
si := dao.GetStorageItem(id, key)
if si == nil {
panic(fmt.Errorf("item with id = %d and key = %s is not initialized", id, hex.EncodeToString(key)))
}
return bigint.FromBytes(si).Int64()
}
// makeUint160Key creates a key from account script hash.
func makeUint160Key(prefix byte, h util.Uint160) []byte {
k := make([]byte, util.Uint160Size+1)
k[0] = prefix
copy(k[1:], h.BytesBE())
return k
}
2021-01-22 12:12:09 +00:00
func toString(item stackitem.Item) string {
s, err := stackitem.ToString(item)
if err != nil {
panic(err)
}
return s
}