forked from TrueCloudLab/neoneo-go
dba248236c
And use it where appropriate. Some of our code was just plain wrong (like the one in GAS contract) and unification is always useful here.
96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
package testchain
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/hash"
|
|
"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/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/emit"
|
|
)
|
|
|
|
// privNetKeys is a list of unencrypted WIFs sorted by public key.
|
|
var privNetKeys = []string{
|
|
"KzfPUYDC9n2yf4fK5ro4C8KMcdeXtFuEnStycbZgX3GomiUsvX6W",
|
|
"KzgWE3u3EDp13XPXXuTKZxeJ3Gi8Bsm8f9ijY3ZsCKKRvZUo1Cdn",
|
|
"KxyjQ8eUa4FHt3Gvioyt1Wz29cTUrE4eTqX3yFSk1YFCsPL8uNsY",
|
|
"L2oEXKRAAMiPEZukwR5ho2S6SMeQLhcK9mF71ZnF7GvT8dU4Kkgz",
|
|
}
|
|
|
|
var (
|
|
// ids maps validators order by public key sorting to validators ID.
|
|
// which is an order of the validator in the StandByValidators list.
|
|
ids = []int{1, 3, 0, 2}
|
|
// orders maps to validators id to it's order by public key sorting.
|
|
orders = []int{2, 0, 3, 1}
|
|
)
|
|
|
|
// Size returns testchain initial validators amount.
|
|
func Size() int {
|
|
return len(privNetKeys)
|
|
}
|
|
|
|
// IDToOrder returns node's order in privnet.
|
|
func IDToOrder(id int) int {
|
|
return orders[id]
|
|
}
|
|
|
|
// WIF returns unencrypted wif of the specified validator.
|
|
func WIF(i int) string {
|
|
return privNetKeys[i]
|
|
}
|
|
|
|
// PrivateKey returns private key of node #i.
|
|
func PrivateKey(i int) *keys.PrivateKey {
|
|
wif := WIF(i)
|
|
priv, err := keys.NewPrivateKeyFromWIF(wif)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return priv
|
|
}
|
|
|
|
// PrivateKeyByID returns private keys of a node with the specified id.
|
|
func PrivateKeyByID(id int) *keys.PrivateKey {
|
|
return PrivateKey(IDToOrder(id))
|
|
}
|
|
|
|
// MultisigVerificationScript returns script hash of the consensus multisig address.
|
|
func MultisigVerificationScript() []byte {
|
|
var pubs keys.PublicKeys
|
|
for i := range privNetKeys {
|
|
priv := PrivateKey(ids[i])
|
|
pubs = append(pubs, priv.PublicKey())
|
|
}
|
|
|
|
script, err := smartcontract.CreateDefaultMultiSigRedeemScript(pubs)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return script
|
|
}
|
|
|
|
// MultisigScriptHash returns consensus address as Uint160.
|
|
func MultisigScriptHash() util.Uint160 {
|
|
return hash.Hash160(MultisigVerificationScript())
|
|
}
|
|
|
|
// MultisigAddress return consensus address as string.
|
|
func MultisigAddress() string {
|
|
return address.Uint160ToString(MultisigScriptHash())
|
|
}
|
|
|
|
// Sign signs data by all consensus nodes and returns invocation script.
|
|
func Sign(data []byte) []byte {
|
|
buf := io.NewBufBinWriter()
|
|
for i := 0; i < Size(); i++ {
|
|
pKey := PrivateKey(i)
|
|
sig := pKey.Sign(data)
|
|
if len(sig) != 64 {
|
|
panic("wrong signature length")
|
|
}
|
|
emit.Bytes(buf.BinWriter, sig)
|
|
}
|
|
return buf.Bytes()
|
|
}
|