You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
175 lines
4.4 KiB
175 lines
4.4 KiB
package types
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/tendermint/tendermint/account"
|
|
)
|
|
|
|
type AccountGetter interface {
|
|
GetAccount(addr []byte) *account.Account
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
// SendTx interface for adding inputs/outputs and adding signatures
|
|
|
|
func NewSendTx() *SendTx {
|
|
return &SendTx{
|
|
Inputs: []*TxInput{},
|
|
Outputs: []*TxOutput{},
|
|
}
|
|
}
|
|
|
|
func (tx *SendTx) AddInput(st AccountGetter, pubkey account.PubKey, amt uint64) error {
|
|
addr := pubkey.Address()
|
|
acc := st.GetAccount(addr)
|
|
if acc == nil {
|
|
return fmt.Errorf("Invalid address %X from pubkey %X", addr, pubkey)
|
|
}
|
|
|
|
tx.Inputs = append(tx.Inputs, &TxInput{
|
|
Address: addr,
|
|
Amount: amt,
|
|
Sequence: uint(acc.Sequence) + 1,
|
|
Signature: account.SignatureEd25519{},
|
|
PubKey: pubkey,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
func (tx *SendTx) AddInputWithNonce(pubkey account.PubKey, amt, nonce uint64) error {
|
|
addr := pubkey.Address()
|
|
tx.Inputs = append(tx.Inputs, &TxInput{
|
|
Address: addr,
|
|
Amount: amt,
|
|
Sequence: uint(nonce) + 1,
|
|
Signature: account.SignatureEd25519{},
|
|
PubKey: pubkey,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
func (tx *SendTx) AddOutput(addr []byte, amt uint64) error {
|
|
tx.Outputs = append(tx.Outputs, &TxOutput{
|
|
Address: addr,
|
|
Amount: amt,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
func (tx *SendTx) SignInput(i int, privAccount *account.PrivAccount) error {
|
|
if i >= len(tx.Inputs) {
|
|
return fmt.Errorf("Index %v is greater than number of inputs (%v)", i, len(tx.Inputs))
|
|
}
|
|
tx.Inputs[i].PubKey = privAccount.PubKey
|
|
tx.Inputs[i].Signature = privAccount.Sign(tx)
|
|
return nil
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
// CallTx interface for creating tx
|
|
|
|
func NewCallTx(st AccountGetter, from account.PubKey, to, data []byte, amt, gasLimit, fee uint64) (*CallTx, error) {
|
|
addr := from.Address()
|
|
acc := st.GetAccount(addr)
|
|
if acc == nil {
|
|
return nil, fmt.Errorf("Invalid address %X from pubkey %X", addr, from)
|
|
}
|
|
|
|
nonce := uint64(acc.Sequence)
|
|
return NewCallTxWithNonce(from, to, data, amt, gasLimit, fee, nonce), nil
|
|
}
|
|
|
|
func NewCallTxWithNonce(from account.PubKey, to, data []byte, amt, gasLimit, fee, nonce uint64) *CallTx {
|
|
addr := from.Address()
|
|
input := &TxInput{
|
|
Address: addr,
|
|
Amount: amt,
|
|
Sequence: uint(nonce) + 1,
|
|
Signature: account.SignatureEd25519{},
|
|
PubKey: from,
|
|
}
|
|
|
|
return &CallTx{
|
|
Input: input,
|
|
Address: to,
|
|
GasLimit: gasLimit,
|
|
Fee: fee,
|
|
Data: data,
|
|
}
|
|
}
|
|
|
|
func (tx *CallTx) Sign(privAccount *account.PrivAccount) {
|
|
tx.Input.PubKey = privAccount.PubKey
|
|
tx.Input.Signature = privAccount.Sign(tx)
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
// BondTx interface for adding inputs/outputs and adding signatures
|
|
|
|
func NewBondTx(pubkey account.PubKey) (*BondTx, error) {
|
|
pubkeyEd, ok := pubkey.(account.PubKeyEd25519)
|
|
if !ok {
|
|
return nil, fmt.Errorf("Pubkey must be ed25519")
|
|
}
|
|
return &BondTx{
|
|
PubKey: pubkeyEd,
|
|
Inputs: []*TxInput{},
|
|
UnbondTo: []*TxOutput{},
|
|
}, nil
|
|
}
|
|
|
|
func (tx *BondTx) AddInput(st AccountGetter, pubkey account.PubKey, amt uint64) error {
|
|
addr := pubkey.Address()
|
|
acc := st.GetAccount(addr)
|
|
if acc == nil {
|
|
return fmt.Errorf("Invalid address %X from pubkey %X", addr, pubkey)
|
|
}
|
|
|
|
tx.Inputs = append(tx.Inputs, &TxInput{
|
|
Address: addr,
|
|
Amount: amt,
|
|
Sequence: uint(acc.Sequence) + 1,
|
|
Signature: account.SignatureEd25519{},
|
|
PubKey: pubkey,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
func (tx *BondTx) AddInputWithNonce(pubkey account.PubKey, amt, nonce uint64) error {
|
|
addr := pubkey.Address()
|
|
tx.Inputs = append(tx.Inputs, &TxInput{
|
|
Address: addr,
|
|
Amount: amt,
|
|
Sequence: uint(nonce) + 1,
|
|
Signature: account.SignatureEd25519{},
|
|
PubKey: pubkey,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
func (tx *BondTx) AddOutput(addr []byte, amt uint64) error {
|
|
tx.UnbondTo = append(tx.UnbondTo, &TxOutput{
|
|
Address: addr,
|
|
Amount: amt,
|
|
})
|
|
return nil
|
|
}
|
|
|
|
func (tx *BondTx) SignBond(privAccount *account.PrivAccount) error {
|
|
sig := privAccount.Sign(tx)
|
|
sigEd, ok := sig.(account.SignatureEd25519)
|
|
if !ok {
|
|
return fmt.Errorf("Bond signer must be ED25519")
|
|
}
|
|
tx.Signature = sigEd
|
|
return nil
|
|
}
|
|
|
|
func (tx *BondTx) SignInput(i int, privAccount *account.PrivAccount) error {
|
|
if i >= len(tx.Inputs) {
|
|
return fmt.Errorf("Index %v is greater than number of inputs (%v)", i, len(tx.Inputs))
|
|
}
|
|
tx.Inputs[i].PubKey = privAccount.PubKey
|
|
tx.Inputs[i].Signature = privAccount.Sign(tx)
|
|
return nil
|
|
}
|