neo-go/pkg/consensus/block.go

74 lines
2.1 KiB
Go
Raw Normal View History

2019-11-15 10:32:40 +00:00
package consensus
import (
"errors"
"github.com/nspcc-dev/dbft"
"github.com/nspcc-dev/neo-go/pkg/config/netmode"
coreb "github.com/nspcc-dev/neo-go/pkg/core/block"
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
"github.com/nspcc-dev/neo-go/pkg/util"
2019-11-15 10:32:40 +00:00
)
// neoBlock is a wrapper of a core.Block which implements
2019-11-15 10:32:40 +00:00
// methods necessary for dBFT library.
type neoBlock struct {
coreb.Block
2019-11-15 10:32:40 +00:00
network netmode.Magic
2019-11-15 10:32:40 +00:00
signature []byte
}
var _ dbft.Block[util.Uint256] = (*neoBlock)(nil)
2019-11-15 10:32:40 +00:00
// Sign implements the block.Block interface.
func (n *neoBlock) Sign(key dbft.PrivateKey) error {
k := key.(*privateKey)
sig := k.PrivateKey.SignHashable(uint32(n.network), &n.Block)
2019-11-15 10:32:40 +00:00
n.signature = sig
return nil
}
// Verify implements the block.Block interface.
func (n *neoBlock) Verify(key dbft.PublicKey, sign []byte) error {
k := key.(*keys.PublicKey)
if k.VerifyHashable(sign, uint32(n.network), &n.Block) {
return nil
}
return errors.New("verification failed")
2019-11-15 10:32:40 +00:00
}
// Transactions implements the block.Block interface.
func (n *neoBlock) Transactions() []dbft.Transaction[util.Uint256] {
txes := make([]dbft.Transaction[util.Uint256], len(n.Block.Transactions))
2019-11-15 10:32:40 +00:00
for i, tx := range n.Block.Transactions {
txes[i] = tx
}
return txes
}
// SetTransactions implements the block.Block interface.
func (n *neoBlock) SetTransactions(txes []dbft.Transaction[util.Uint256]) {
2019-11-15 10:32:40 +00:00
n.Block.Transactions = make([]*transaction.Transaction, len(txes))
for i, tx := range txes {
n.Block.Transactions[i] = tx.(*transaction.Transaction)
}
}
// PrevHash implements the block.Block interface.
2019-11-15 10:32:40 +00:00
func (n *neoBlock) PrevHash() util.Uint256 { return n.Block.PrevHash }
// MerkleRoot implements the block.Block interface.
2019-11-15 10:32:40 +00:00
func (n *neoBlock) MerkleRoot() util.Uint256 { return n.Block.MerkleRoot }
// Timestamp implements the block.Block interface.
func (n *neoBlock) Timestamp() uint64 { return n.Block.Timestamp * nsInMs }
2019-11-15 10:32:40 +00:00
// Index implements the block.Block interface.
2019-11-15 10:32:40 +00:00
func (n *neoBlock) Index() uint32 { return n.Block.Index }
// Signature implements the block.Block interface.
2019-11-15 10:32:40 +00:00
func (n *neoBlock) Signature() []byte { return n.signature }