forked from TrueCloudLab/neoneo-go
ec7e17ffa6
Simplifies a lot of code and removes some duplication. Unfortunately I had to move test_util random functions in same commit to avoid cycle dependencies. One of these random functions was also used in core/transaction testing, to simplify things I've just dropped it there and used a static string (which is nice to have for a test anyway). There is still sha256 left in wallet (but it needs to pass Hash structure into the signing function).
81 lines
2 KiB
Go
81 lines
2 KiB
Go
package util
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
const uint160Size = 20
|
|
|
|
// Uint160 is a 20 byte long unsigned integer.
|
|
type Uint160 [uint160Size]uint8
|
|
|
|
// Uint160DecodeString attempts to decode the given string into an Uint160.
|
|
func Uint160DecodeString(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 Uint160DecodeBytes(b)
|
|
}
|
|
|
|
// Uint160DecodeBytes attempts to decode the given bytes into an Uint160.
|
|
func Uint160DecodeBytes(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
|
|
}
|
|
|
|
// Bytes returns the byte slice representation of u.
|
|
func (u Uint160) Bytes() []byte {
|
|
return u[:]
|
|
}
|
|
|
|
// BytesReverse return a reversed byte representation of u.
|
|
func (u Uint160) BytesReverse() []byte {
|
|
return ArrayReverse(u.Bytes())
|
|
}
|
|
|
|
// String implements the stringer interface.
|
|
func (u Uint160) String() string {
|
|
return hex.EncodeToString(u.Bytes())
|
|
}
|
|
|
|
// ReverseString is the same as String, but returnes an inversed representation.
|
|
func (u Uint160) ReverseString() string {
|
|
return hex.EncodeToString(u.BytesReverse())
|
|
}
|
|
|
|
// Equals returns true if both Uint256 values are the same.
|
|
func (u Uint160) Equals(other Uint160) bool {
|
|
return u == other
|
|
}
|
|
|
|
// 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 = Uint160DecodeString(js)
|
|
return err
|
|
}
|
|
|
|
// Size returns the lenght of the bytes representation of Uint160
|
|
func (u Uint160) Size() int {
|
|
return uint160Size
|
|
}
|
|
|
|
// MarshalJSON implements the json marshaller interface.
|
|
func (u Uint160) MarshalJSON() ([]byte, error) {
|
|
return []byte(`"0x` + u.String() + `"`), nil
|
|
}
|