119 lines
2.6 KiB
Go
119 lines
2.6 KiB
Go
package keyer
|
|
|
|
import (
|
|
"crypto/elliptic"
|
|
"encoding/hex"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/mr-tron/base58"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
)
|
|
|
|
const (
|
|
NeoPrivateKeySize = 32
|
|
|
|
scriptHashSize = 20
|
|
addressSize = 25
|
|
publicKeyCompressedSize = 33
|
|
wifSize = 38
|
|
publicKeyUncompressedSize = 65
|
|
)
|
|
|
|
var errInputType = errors.New("unknown input type")
|
|
|
|
func (d *Dashboard) ParseString(data string) error {
|
|
// just in case remove 0x prefixes if there are some
|
|
data = strings.TrimPrefix(data, "0x")
|
|
data = strings.TrimPrefix(data, "0X")
|
|
|
|
var (
|
|
rawData []byte
|
|
err error
|
|
)
|
|
|
|
// data could be encoded in base58 or hex formats, try both
|
|
rawData, err = hex.DecodeString(data)
|
|
if err != nil {
|
|
rawData, err = base58.Decode(data)
|
|
if err != nil {
|
|
return fmt.Errorf("data is not hex or base58 encoded: %w", err)
|
|
}
|
|
}
|
|
|
|
return d.ParseBinary(rawData)
|
|
}
|
|
|
|
func (d *Dashboard) ParseBinary(data []byte) error {
|
|
var err error
|
|
|
|
switch len(data) {
|
|
case NeoPrivateKeySize:
|
|
d.privKey, err = keys.NewPrivateKeyFromBytes(data)
|
|
if err != nil {
|
|
return fmt.Errorf("can't parse private key: %w", err)
|
|
}
|
|
case wifSize:
|
|
d.privKey, err = keys.NewPrivateKeyFromWIF(base58.Encode(data))
|
|
if err != nil {
|
|
return fmt.Errorf("can't parse WIF: %w", err)
|
|
}
|
|
case publicKeyCompressedSize, publicKeyUncompressedSize:
|
|
d.pubKey, err = keys.NewPublicKeyFromBytes(data, elliptic.P256())
|
|
if err != nil {
|
|
return fmt.Errorf("can't parse public key: %w", err)
|
|
}
|
|
case addressSize:
|
|
d.scriptHash3, err = address.StringToUint160(base58.Encode(data))
|
|
if err != nil {
|
|
return fmt.Errorf("can't parse address: %w", err)
|
|
}
|
|
case scriptHashSize:
|
|
sc, err := util.Uint160DecodeBytesLE(data)
|
|
if err != nil {
|
|
return fmt.Errorf("can't parse script hash: %w", err)
|
|
}
|
|
|
|
d.scriptHash3 = sc
|
|
default:
|
|
return errInputType
|
|
}
|
|
|
|
d.fill()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Dashboard) ParseMultiSig(data []string) error {
|
|
d.multisigKeys = make(keys.PublicKeys, 0, len(data))
|
|
|
|
for i := range data {
|
|
data, err := hex.DecodeString(data[i])
|
|
if err != nil {
|
|
return fmt.Errorf("pass only hex encoded public keys: %w", err)
|
|
}
|
|
|
|
key, err := keys.NewPublicKeyFromBytes(data, elliptic.P256())
|
|
if err != nil {
|
|
return fmt.Errorf("pass only hex encoded public keys: %w", err)
|
|
}
|
|
|
|
d.multisigKeys = append(d.multisigKeys, key)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *Dashboard) fill() {
|
|
if d.privKey != nil {
|
|
d.pubKey = d.privKey.PublicKey()
|
|
}
|
|
|
|
if d.pubKey != nil {
|
|
d.scriptHash3 = d.pubKey.GetScriptHash()
|
|
}
|
|
}
|