mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-11-30 09:33:36 +00:00
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
package smartcontract
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"sort"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/crypto/keys"
|
|
"github.com/CityOfZion/neo-go/pkg/vm"
|
|
)
|
|
|
|
// CreateSignatureRedeemScript creates a check signature script runnable by VM.
|
|
func CreateSignatureRedeemScript(key *keys.PublicKey) ([]byte, error) {
|
|
buf := new(bytes.Buffer)
|
|
err := vm.EmitBytes(buf, key.Bytes())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = vm.EmitOpcode(buf, vm.CHECKSIG)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
// CreateMultiSigRedeemScript creates a script runnable by the VM.
|
|
func CreateMultiSigRedeemScript(m int, publicKeys keys.PublicKeys) ([]byte, error) {
|
|
if m <= 1 {
|
|
return nil, fmt.Errorf("param m cannot be smaller or equal to 1 got %d", m)
|
|
}
|
|
if m > len(publicKeys) {
|
|
return nil, fmt.Errorf("length of the signatures (%d) is higher then the number of public keys", m)
|
|
}
|
|
if m > 1024 {
|
|
return nil, fmt.Errorf("public key count %d exceeds maximum of length 1024", len(publicKeys))
|
|
}
|
|
|
|
buf := new(bytes.Buffer)
|
|
if err := vm.EmitInt(buf, int64(m)); err != nil {
|
|
return nil, err
|
|
}
|
|
sort.Sort(publicKeys)
|
|
for _, pubKey := range publicKeys {
|
|
if err := vm.EmitBytes(buf, pubKey.Bytes()); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
if err := vm.EmitInt(buf, int64(len(publicKeys))); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := vm.EmitOpcode(buf, vm.CHECKMULTISIG); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return buf.Bytes(), nil
|
|
}
|