forked from TrueCloudLab/neoneo-go
6ccb518ab0
* Optimizations + some improvements - optimized pkg/core/storage.HeaderHashes - optimized pkg/rpc.performRequest (used json.Encoder) - fixes for pkg/util.ReadVarUint and pkg/util.WriteVarUint - optimized and fix fixed8 (Fixed8DecodeString / MarshalJSON) + tests - optimized and fix uint160 (Bytes / Uint160DecodeString / Equal / MarshalJSON) + tests - optimized and fix uint256 (Bytes / Equal / MarshalJSON) + tests - preallocate for pkg/vm.buildStackOutput - add go.mod / go.sum * update version
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package util
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
const uint256Size = 32
|
|
|
|
// Uint256 is a 32 byte long unsigned integer.
|
|
type Uint256 [uint256Size]uint8
|
|
|
|
// Uint256DecodeString attempts to decode the given string into an Uint256.
|
|
func Uint256DecodeString(s string) (u Uint256, err error) {
|
|
if len(s) != uint256Size*2 {
|
|
return u, fmt.Errorf("expected string size of %d got %d", uint256Size*2, len(s))
|
|
}
|
|
b, err := hex.DecodeString(s)
|
|
if err != nil {
|
|
return u, err
|
|
}
|
|
return Uint256DecodeBytes(b)
|
|
}
|
|
|
|
// Uint256DecodeBytes attempts to decode the given string into an Uint256.
|
|
func Uint256DecodeBytes(b []byte) (u Uint256, err error) {
|
|
b = ArrayReverse(b)
|
|
if len(b) != uint256Size {
|
|
return u, fmt.Errorf("expected []byte of size %d got %d", uint256Size, len(b))
|
|
}
|
|
for i := 0; i < uint256Size; i++ {
|
|
u[i] = b[i]
|
|
}
|
|
return u, nil
|
|
}
|
|
|
|
// Bytes returns a byte slice representation of u.
|
|
func (u Uint256) Bytes() []byte {
|
|
return u[:]
|
|
}
|
|
|
|
// BytesReverse return a reversed byte representation of u.
|
|
func (u Uint256) BytesReverse() []byte {
|
|
return ArrayReverse(u.Bytes())
|
|
}
|
|
|
|
// Equals returns true if both Uint256 values are the same.
|
|
func (u Uint256) Equals(other Uint256) bool {
|
|
return u == other
|
|
}
|
|
|
|
// String implements the stringer interface.
|
|
func (u Uint256) String() string {
|
|
return hex.EncodeToString(ArrayReverse(u.Bytes()))
|
|
}
|
|
|
|
// UnmarshalJSON implements the json unmarshaller interface.
|
|
func (u *Uint256) UnmarshalJSON(data []byte) (err error) {
|
|
var js string
|
|
if err = json.Unmarshal(data, &js); err != nil {
|
|
return err
|
|
}
|
|
if strings.HasPrefix(js, "0x") {
|
|
js = js[2:]
|
|
}
|
|
*u, err = Uint256DecodeString(js)
|
|
return err
|
|
}
|
|
|
|
// MarshalJSON implements the json marshaller interface.
|
|
func (u Uint256) MarshalJSON() ([]byte, error) {
|
|
return []byte(`"0x` + u.String() + `"`), nil
|
|
}
|