mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-12 01:10:36 +00:00
b0744c2b21
name old time/op new time/op delta Uint256MarshalJSON-8 230ns ± 6% 83ns ±13% -63.78% (p=0.000 n=8+10) name old alloc/op new alloc/op delta Uint256MarshalJSON-8 320B ± 0% 80B ± 0% -75.00% (p=0.000 n=10+10) name old allocs/op new allocs/op delta Uint256MarshalJSON-8 5.00 ± 0% 1.00 ± 0% -80.00% (p=0.000 n=10+10)
170 lines
4.1 KiB
Go
170 lines
4.1 KiB
Go
package util
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/util/slice"
|
|
)
|
|
|
|
// Uint160Size is the size of Uint160 in bytes.
|
|
const Uint160Size = 20
|
|
|
|
// Uint160 is a 20 byte long unsigned integer.
|
|
type Uint160 [Uint160Size]uint8
|
|
|
|
// Uint160DecodeStringBE attempts to decode the given string into a Uint160.
|
|
func Uint160DecodeStringBE(s string) (Uint160, error) {
|
|
var u Uint160
|
|
if len(s) != Uint160Size*2 {
|
|
return u, fmt.Errorf("expected string size of %d got %d", Uint160Size*2, len(s))
|
|
}
|
|
b, err := hex.DecodeString(s)
|
|
if err != nil {
|
|
return u, err
|
|
}
|
|
return Uint160DecodeBytesBE(b)
|
|
}
|
|
|
|
// Uint160DecodeStringLE attempts to decode the given string
|
|
// in little-endian hex encoding into a Uint160.
|
|
func Uint160DecodeStringLE(s string) (Uint160, error) {
|
|
var u Uint160
|
|
if len(s) != Uint160Size*2 {
|
|
return u, fmt.Errorf("expected string size of %d got %d", Uint160Size*2, len(s))
|
|
}
|
|
|
|
b, err := hex.DecodeString(s)
|
|
if err != nil {
|
|
return u, err
|
|
}
|
|
|
|
return Uint160DecodeBytesLE(b)
|
|
}
|
|
|
|
// Uint160DecodeBytesBE attempts to decode the given bytes into a Uint160.
|
|
func Uint160DecodeBytesBE(b []byte) (u Uint160, err error) {
|
|
if len(b) != Uint160Size {
|
|
return u, fmt.Errorf("expected byte size of %d got %d", Uint160Size, len(b))
|
|
}
|
|
copy(u[:], b)
|
|
return
|
|
}
|
|
|
|
// Uint160DecodeBytesLE attempts to decode the given bytes in little-endian
|
|
// into a Uint160.
|
|
func Uint160DecodeBytesLE(b []byte) (u Uint160, err error) {
|
|
if len(b) != Uint160Size {
|
|
return u, fmt.Errorf("expected byte size of %d got %d", Uint160Size, len(b))
|
|
}
|
|
|
|
for i := range b {
|
|
u[Uint160Size-i-1] = b[i]
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// BytesBE returns a big-endian byte representation of u.
|
|
func (u Uint160) BytesBE() []byte {
|
|
return u[:]
|
|
}
|
|
|
|
// BytesLE returns a little-endian byte representation of u.
|
|
func (u Uint160) BytesLE() []byte {
|
|
return slice.CopyReverse(u.BytesBE())
|
|
}
|
|
|
|
// String implements the stringer interface.
|
|
func (u Uint160) String() string {
|
|
return u.StringBE()
|
|
}
|
|
|
|
// StringBE returns string representations of u with big-endian byte order.
|
|
func (u Uint160) StringBE() string {
|
|
return hex.EncodeToString(u.BytesBE())
|
|
}
|
|
|
|
// StringLE returns string representations of u with little-endian byte order.
|
|
func (u Uint160) StringLE() string {
|
|
return hex.EncodeToString(u.BytesLE())
|
|
}
|
|
|
|
// Reverse returns a reversed representation of u.
|
|
func (u Uint160) Reverse() (r Uint160) {
|
|
for i := 0; i < Uint160Size; i++ {
|
|
r[i] = u[Uint160Size-i-1]
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Equals returns true if both Uint160 values are the same.
|
|
func (u Uint160) Equals(other Uint160) bool {
|
|
return u == other
|
|
}
|
|
|
|
// Less returns true if this value is less than the given Uint160 value. It's
|
|
// primarily intended to be used for sorting purposes.
|
|
func (u Uint160) Less(other Uint160) bool {
|
|
for k := range u {
|
|
if u[k] == other[k] {
|
|
continue
|
|
}
|
|
return u[k] < other[k]
|
|
}
|
|
return false
|
|
}
|
|
|
|
// UnmarshalJSON implements the json unmarshaller interface.
|
|
func (u *Uint160) UnmarshalJSON(data []byte) (err error) {
|
|
var js string
|
|
if err = json.Unmarshal(data, &js); err != nil {
|
|
return err
|
|
}
|
|
js = strings.TrimPrefix(js, "0x")
|
|
*u, err = Uint160DecodeStringLE(js)
|
|
return err
|
|
}
|
|
|
|
// MarshalJSON implements the json marshaller interface.
|
|
func (u Uint160) MarshalJSON() ([]byte, error) {
|
|
r := make([]byte, 3+Uint160Size*2+1)
|
|
copy(r, `"0x`)
|
|
r[len(r)-1] = '"'
|
|
slice.Reverse(u[:]) // u is a copy, so we can mangle it in any way.
|
|
hex.Encode(r[3:], u[:])
|
|
return r, nil
|
|
}
|
|
|
|
// UnmarshalYAML implements the YAML Unmarshaler interface.
|
|
func (u *Uint160) UnmarshalYAML(unmarshal func(interface{}) error) error {
|
|
var s string
|
|
|
|
err := unmarshal(&s)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
s = strings.TrimPrefix(s, "0x")
|
|
*u, err = Uint160DecodeStringLE(s)
|
|
return err
|
|
}
|
|
|
|
// MarshalYAML implements the YAML marshaller interface.
|
|
func (u Uint160) MarshalYAML() (interface{}, error) {
|
|
return "0x" + u.StringLE(), nil
|
|
}
|
|
|
|
// EncodeBinary implements the Serializable interface.
|
|
func (u *Uint160) EncodeBinary(bw *io.BinWriter) {
|
|
bw.WriteBytes(u[:])
|
|
}
|
|
|
|
// DecodeBinary implements the Serializable interface.
|
|
func (u *Uint160) DecodeBinary(br *io.BinReader) {
|
|
br.ReadBytes(u[:])
|
|
}
|