mirror of
https://github.com/fluencelabs/tendermint
synced 2025-04-25 14:52:17 +00:00
introduce evidence store
This commit is contained in:
parent
fe4b53a463
commit
10c43c9edc
@ -17,6 +17,7 @@ import (
|
||||
|
||||
cfg "github.com/tendermint/tendermint/config"
|
||||
cstypes "github.com/tendermint/tendermint/consensus/types"
|
||||
evpool "github.com/tendermint/tendermint/evidence"
|
||||
"github.com/tendermint/tendermint/proxy"
|
||||
sm "github.com/tendermint/tendermint/state"
|
||||
"github.com/tendermint/tendermint/types"
|
||||
@ -78,7 +79,7 @@ type ConsensusState struct {
|
||||
proxyAppConn proxy.AppConnConsensus
|
||||
blockStore types.BlockStore
|
||||
mempool types.Mempool
|
||||
evpool types.EvidencePool
|
||||
evpool evpool.EvidencePool
|
||||
|
||||
// internal state
|
||||
mtx sync.Mutex
|
||||
@ -114,7 +115,7 @@ type ConsensusState struct {
|
||||
}
|
||||
|
||||
// NewConsensusState returns a new ConsensusState.
|
||||
func NewConsensusState(config *cfg.ConsensusConfig, state *sm.State, proxyAppConn proxy.AppConnConsensus, blockStore types.BlockStore, mempool types.Mempool, evpool types.Mempool) *ConsensusState {
|
||||
func NewConsensusState(config *cfg.ConsensusConfig, state *sm.State, proxyAppConn proxy.AppConnConsensus, blockStore types.BlockStore, mempool types.Mempool) *ConsensusState {
|
||||
cs := &ConsensusState{
|
||||
config: config,
|
||||
proxyAppConn: proxyAppConn,
|
||||
@ -126,7 +127,7 @@ func NewConsensusState(config *cfg.ConsensusConfig, state *sm.State, proxyAppCon
|
||||
done: make(chan struct{}),
|
||||
doWALCatchup: true,
|
||||
wal: nilWAL{},
|
||||
evpool: evpool,
|
||||
// evpool: evpool,
|
||||
}
|
||||
// set function defaults (may be overwritten before calling Start)
|
||||
cs.decideProposal = cs.defaultDecideProposal
|
||||
@ -866,7 +867,7 @@ func (cs *ConsensusState) createProposalBlock() (block *types.Block, blockParts
|
||||
// Mempool validated transactions
|
||||
txs := cs.mempool.Reap(cs.config.MaxBlockSizeTxs)
|
||||
block, parts := cs.state.MakeBlock(cs.Height, txs, commit)
|
||||
evidence := cs.evpool.Evidence()
|
||||
evidence := cs.evpool.PendingEvidence()
|
||||
block.AddEvidence(evidence)
|
||||
return block, parts
|
||||
}
|
||||
|
@ -1,14 +1,8 @@
|
||||
package evpool
|
||||
|
||||
import (
|
||||
"container/list"
|
||||
"fmt"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
|
||||
"github.com/tendermint/tmlibs/log"
|
||||
|
||||
cfg "github.com/tendermint/tendermint/config"
|
||||
"github.com/tendermint/tendermint/types"
|
||||
)
|
||||
|
||||
@ -16,25 +10,23 @@ const cacheSize = 100000
|
||||
|
||||
// EvidencePool maintains a set of valid uncommitted evidence.
|
||||
type EvidencePool struct {
|
||||
config *cfg.EvidencePoolConfig
|
||||
|
||||
mtx sync.Mutex
|
||||
height int // the last block Update()'d to
|
||||
evidence types.Evidences
|
||||
// TODO: evidenceCache
|
||||
|
||||
// TODO: need to persist evidence so we never lose it
|
||||
|
||||
config *EvidencePoolConfig
|
||||
logger log.Logger
|
||||
|
||||
evidenceStore *EvidenceStore
|
||||
newEvidenceChan chan types.Evidence
|
||||
}
|
||||
|
||||
func NewEvidencePool(config *cfg.EvidencePoolConfig, height int) *EvidencePool {
|
||||
type EvidencePoolConfig struct {
|
||||
}
|
||||
|
||||
func NewEvidencePool(config *EvidencePoolConfig, evidenceStore *EvidenceStore) *EvidencePool {
|
||||
evpool := &EvidencePool{
|
||||
config: config,
|
||||
height: height,
|
||||
logger: log.NewNopLogger(),
|
||||
config: config,
|
||||
logger: log.NewNopLogger(),
|
||||
evidenceStore: evidenceStore,
|
||||
newEvidenceChan: make(chan types.Evidence),
|
||||
}
|
||||
evpool.initWAL()
|
||||
return evpool
|
||||
}
|
||||
|
||||
@ -43,160 +35,59 @@ func (evpool *EvidencePool) SetLogger(l log.Logger) {
|
||||
evpool.logger = l
|
||||
}
|
||||
|
||||
// Evidence returns a copy of the pool's evidence.
|
||||
func (evpool *EvidencePool) Evidence() types.Evidences {
|
||||
evpool.mtx.Lock()
|
||||
defer evpool.mtx.Unlock()
|
||||
|
||||
evCopy := make(types.Evidences, len(evpool.evidence))
|
||||
for i, ev := range evpool.evidence {
|
||||
evCopy[i] = ev
|
||||
}
|
||||
return evCopy
|
||||
// NewEvidenceChan returns a channel on which new evidence is sent.
|
||||
func (evpool *EvidencePool) NewEvidenceChan() chan types.Evidence {
|
||||
return evpool.newEvidenceChan
|
||||
}
|
||||
|
||||
// Size returns the number of pieces of evidence in the evpool.
|
||||
func (evpool *EvidencePool) Size() int {
|
||||
evpool.mtx.Lock()
|
||||
defer evpool.mtx.Unlock()
|
||||
return len(evpool.evidence)
|
||||
// PriorityEvidence returns the priority evidence.
|
||||
func (evpool *EvidencePool) PriorityEvidence() []types.Evidence {
|
||||
// TODO
|
||||
return nil
|
||||
}
|
||||
|
||||
// Flush removes all evidence from the evpool
|
||||
func (evpool *EvidencePool) Flush() {
|
||||
evpool.mtx.Lock()
|
||||
defer evpool.mtx.Unlock()
|
||||
evpool.evidence = make(types.Evidence)
|
||||
// PendingEvidence returns all uncommitted evidence.
|
||||
func (evpool *EvidencePool) PendingEvidence() []types.Evidence {
|
||||
// TODO
|
||||
return nil
|
||||
}
|
||||
|
||||
// AddEvidence checks the evidence is valid and adds it to the pool.
|
||||
func (evpool *EvidencePool) AddEvidence(evidence types.Evidence) (err error) {
|
||||
evpool.mtx.Lock()
|
||||
defer evpool.mtx.Unlock()
|
||||
|
||||
if evpool.evidence.Has(evidence) {
|
||||
return fmt.Errorf("Evidence already exists", "evidence", evidence)
|
||||
idx := 1 // TODO
|
||||
added, err := evpool.evidenceStore.AddNewEvidence(idx, evidence)
|
||||
if err != nil {
|
||||
return err
|
||||
} else if !added {
|
||||
// evidence already known, just ignore
|
||||
return
|
||||
}
|
||||
cs.Logger.Info("Found conflicting vote. Recording evidence", "evidence", ev)
|
||||
evpool.evidence = append(evpool.evidence, ev)
|
||||
// TODO: write to disk ? WAL ?
|
||||
|
||||
evpool.logger.Info("Verified new evidence of byzantine behaviour", "evidence", evidence)
|
||||
|
||||
evpool.newEvidenceChan <- evidence
|
||||
return nil
|
||||
}
|
||||
|
||||
// Update informs the evpool that the given evidence was committed and can be discarded.
|
||||
// NOTE: this should be called *after* block is committed by consensus.
|
||||
func (evpool *EvidencePool) Update(height int, evidence types.Evidences) {
|
||||
func (evpool *EvidencePool) Update(height int, evidence []types.Evidence) {
|
||||
|
||||
// First, create a lookup map of new committed evidence
|
||||
|
||||
// First, create a lookup map of txns in new txs.
|
||||
evMap := make(map[string]struct{})
|
||||
for _, ev := range evidence {
|
||||
evMap[string(evidence.Hash())] = struct{}{}
|
||||
evpool.evidenceStore.MarkEvidenceAsCommitted(ev)
|
||||
evMap[string(ev.Hash())] = struct{}{}
|
||||
}
|
||||
|
||||
// Set height
|
||||
evpool.height = height
|
||||
|
||||
// Remove evidence that is already committed .
|
||||
goodEvidence := evpool.filterEvidence(evMap)
|
||||
_ = goodEvidence
|
||||
|
||||
}
|
||||
|
||||
// TODO:
|
||||
func (evpool *EvidencePool) filterTxs(blockTxsMap map[string]struct{}) []types.Tx {
|
||||
goodTxs := make([]types.Tx, 0, evpool.txs.Len())
|
||||
for e := evpool.txs.Front(); e != nil; e = e.Next() {
|
||||
memTx := e.Value.(*evpoolTx)
|
||||
// Remove the tx if it's alredy in a block.
|
||||
if _, ok := blockTxsMap[string(memTx.tx)]; ok {
|
||||
// remove from clist
|
||||
evpool.txs.Remove(e)
|
||||
e.DetachPrev()
|
||||
|
||||
// NOTE: we don't remove committed txs from the cache.
|
||||
continue
|
||||
}
|
||||
// Good tx!
|
||||
goodTxs = append(goodTxs, memTx.tx)
|
||||
}
|
||||
return goodTxs
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
|
||||
// evpoolTx is a transaction that successfully ran
|
||||
type evpoolEvidence struct {
|
||||
counter int64 // a simple incrementing counter
|
||||
height int64 // height that this tx had been validated in
|
||||
evidence types.Evidence //
|
||||
}
|
||||
|
||||
// Height returns the height for this transaction
|
||||
func (memTx *evpoolTx) Height() int {
|
||||
return int(atomic.LoadInt64(&memTx.height))
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
// TODO:
|
||||
|
||||
// txCache maintains a cache of evidence
|
||||
type txCache struct {
|
||||
mtx sync.Mutex
|
||||
size int
|
||||
map_ map[string]struct{}
|
||||
list *list.List // to remove oldest tx when cache gets too big
|
||||
}
|
||||
|
||||
// newTxCache returns a new txCache.
|
||||
func newTxCache(cacheSize int) *txCache {
|
||||
return &txCache{
|
||||
size: cacheSize,
|
||||
map_: make(map[string]struct{}, cacheSize),
|
||||
list: list.New(),
|
||||
}
|
||||
}
|
||||
|
||||
// Reset resets the txCache to empty.
|
||||
func (cache *txCache) Reset() {
|
||||
cache.mtx.Lock()
|
||||
cache.map_ = make(map[string]struct{}, cacheSize)
|
||||
cache.list.Init()
|
||||
cache.mtx.Unlock()
|
||||
}
|
||||
|
||||
// Exists returns true if the given tx is cached.
|
||||
func (cache *txCache) Exists(tx types.Tx) bool {
|
||||
cache.mtx.Lock()
|
||||
_, exists := cache.map_[string(tx)]
|
||||
cache.mtx.Unlock()
|
||||
return exists
|
||||
}
|
||||
|
||||
// Push adds the given tx to the txCache. It returns false if tx is already in the cache.
|
||||
func (cache *txCache) Push(tx types.Tx) bool {
|
||||
cache.mtx.Lock()
|
||||
defer cache.mtx.Unlock()
|
||||
|
||||
if _, exists := cache.map_[string(tx)]; exists {
|
||||
return false
|
||||
}
|
||||
|
||||
if cache.list.Len() >= cache.size {
|
||||
popped := cache.list.Front()
|
||||
poppedTx := popped.Value.(types.Tx)
|
||||
// NOTE: the tx may have already been removed from the map
|
||||
// but deleting a non-existent element is fine
|
||||
delete(cache.map_, string(poppedTx))
|
||||
cache.list.Remove(popped)
|
||||
}
|
||||
cache.map_[string(tx)] = struct{}{}
|
||||
cache.list.PushBack(tx)
|
||||
return true
|
||||
}
|
||||
|
||||
// Remove removes the given tx from the cache.
|
||||
func (cache *txCache) Remove(tx types.Tx) {
|
||||
cache.mtx.Lock()
|
||||
delete(cache.map_, string(tx))
|
||||
cache.mtx.Unlock()
|
||||
func (evpool *EvidencePool) filterEvidence(blockEvidenceMap map[string]struct{}) []types.Evidence {
|
||||
// TODO:
|
||||
return nil
|
||||
}
|
||||
|
@ -1,4 +1,4 @@
|
||||
package mempool
|
||||
package evpool
|
||||
|
||||
import (
|
||||
"crypto/rand"
|
||||
|
@ -4,11 +4,11 @@ import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"reflect"
|
||||
"time"
|
||||
|
||||
wire "github.com/tendermint/go-wire"
|
||||
"github.com/tendermint/tmlibs/log"
|
||||
|
||||
cfg "github.com/tendermint/tendermint/config"
|
||||
"github.com/tendermint/tendermint/p2p"
|
||||
"github.com/tendermint/tendermint/types"
|
||||
)
|
||||
@ -18,21 +18,22 @@ const (
|
||||
|
||||
maxEvidencePoolMessageSize = 1048576 // 1MB TODO make it configurable
|
||||
peerCatchupSleepIntervalMS = 100 // If peer is behind, sleep this amount
|
||||
broadcastEvidenceIntervalS = 60 // broadcast uncommitted evidence this often
|
||||
)
|
||||
|
||||
// EvidencePoolReactor handles evpool evidence broadcasting amongst peers.
|
||||
type EvidencePoolReactor struct {
|
||||
p2p.BaseReactor
|
||||
config *cfg.EvidencePoolConfig
|
||||
EvidencePool *EvidencePool
|
||||
evsw types.EventSwitch
|
||||
config *EvidencePoolConfig
|
||||
evpool *EvidencePool
|
||||
evsw types.EventSwitch
|
||||
}
|
||||
|
||||
// NewEvidencePoolReactor returns a new EvidencePoolReactor with the given config and evpool.
|
||||
func NewEvidencePoolReactor(config *cfg.EvidencePoolConfig, evpool *EvidencePool) *EvidencePoolReactor {
|
||||
func NewEvidencePoolReactor(config *EvidencePoolConfig, evpool *EvidencePool) *EvidencePoolReactor {
|
||||
evR := &EvidencePoolReactor{
|
||||
config: config,
|
||||
EvidencePool: evpool,
|
||||
config: config,
|
||||
evpool: evpool,
|
||||
}
|
||||
evR.BaseReactor = *p2p.NewBaseReactor("EvidencePoolReactor", evR)
|
||||
return evR
|
||||
@ -41,7 +42,16 @@ func NewEvidencePoolReactor(config *cfg.EvidencePoolConfig, evpool *EvidencePool
|
||||
// SetLogger sets the Logger on the reactor and the underlying EvidencePool.
|
||||
func (evR *EvidencePoolReactor) SetLogger(l log.Logger) {
|
||||
evR.Logger = l
|
||||
evR.EvidencePool.SetLogger(l)
|
||||
evR.evpool.SetLogger(l)
|
||||
}
|
||||
|
||||
// OnStart implements cmn.Service
|
||||
func (evR *EvidencePoolReactor) OnStart() error {
|
||||
if err := evR.BaseReactor.OnStart(); err != nil {
|
||||
return err
|
||||
}
|
||||
go evR.broadcastRoutine()
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetChannels implements Reactor.
|
||||
@ -57,13 +67,16 @@ func (evR *EvidencePoolReactor) GetChannels() []*p2p.ChannelDescriptor {
|
||||
|
||||
// AddPeer implements Reactor.
|
||||
func (evR *EvidencePoolReactor) AddPeer(peer p2p.Peer) {
|
||||
// send the new peer all current evidence
|
||||
evidence := evR.evpool.Evidence()
|
||||
// first send the peer high-priority evidence
|
||||
evidence := evR.evpool.PriorityEvidence()
|
||||
msg := EvidenceMessage{evidence}
|
||||
success := peer.Send(EvidencePoolChannel, struct{ EvidencePoolMessage }{msg})
|
||||
if !success {
|
||||
// TODO: remove peer ?
|
||||
}
|
||||
|
||||
// TODO: send the remaining pending evidence
|
||||
// or just let the broadcastRoutine do it ?
|
||||
}
|
||||
|
||||
// RemovePeer implements Reactor.
|
||||
@ -83,12 +96,10 @@ func (evR *EvidencePoolReactor) Receive(chID byte, src p2p.Peer, msgBytes []byte
|
||||
switch msg := msg.(type) {
|
||||
case *EvidenceMessage:
|
||||
for _, ev := range msg.Evidence {
|
||||
err := evR.EvidencePool.AddEvidence(msg.Evidence, nil)
|
||||
err := evR.evpool.AddEvidence(ev)
|
||||
if err != nil {
|
||||
evR.Logger.Info("Evidence is not valid", "evidence", msg.Evidence, "err", err)
|
||||
// TODO: punish peer
|
||||
} else {
|
||||
// TODO: broadcast good evidence to all peers (except sender? )
|
||||
}
|
||||
}
|
||||
default:
|
||||
@ -101,6 +112,30 @@ func (evR *EvidencePoolReactor) SetEventSwitch(evsw types.EventSwitch) {
|
||||
evR.evsw = evsw
|
||||
}
|
||||
|
||||
// broadcast new evidence to all peers
|
||||
func (evR *EvidencePoolReactor) broadcastRoutine() {
|
||||
ticker := time.NewTicker(time.Second * broadcastEvidenceIntervalS)
|
||||
for {
|
||||
select {
|
||||
case evidence := <-evR.evpool.NewEvidenceChan():
|
||||
// broadcast some new evidence
|
||||
msg := EvidenceMessage{[]types.Evidence{evidence}}
|
||||
evR.Switch.Broadcast(EvidencePoolChannel, struct{ EvidencePoolMessage }{msg})
|
||||
|
||||
// NOTE: Broadcast runs asynchronously, so this should wait on the successChan
|
||||
// in another routine before marking to be proper.
|
||||
idx := 1 // TODO
|
||||
evR.evpool.evidenceStore.MarkEvidenceAsBroadcasted(idx, evidence)
|
||||
case <-ticker.C:
|
||||
// broadcast all pending evidence
|
||||
msg := EvidenceMessage{evR.evpool.PendingEvidence()}
|
||||
evR.Switch.Broadcast(EvidencePoolChannel, struct{ EvidencePoolMessage }{msg})
|
||||
case <-evR.Quit:
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//-----------------------------------------------------------------------------
|
||||
// Messages
|
||||
|
||||
@ -129,7 +164,7 @@ func DecodeMessage(bz []byte) (msgType byte, msg EvidencePoolMessage, err error)
|
||||
|
||||
// EvidenceMessage contains a list of evidence.
|
||||
type EvidenceMessage struct {
|
||||
Evidence types.Evidences
|
||||
Evidence []types.Evidence
|
||||
}
|
||||
|
||||
// String returns a string representation of the EvidenceMessage.
|
||||
|
100
evidence/store.go
Normal file
100
evidence/store.go
Normal file
@ -0,0 +1,100 @@
|
||||
package evpool
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
wire "github.com/tendermint/go-wire"
|
||||
"github.com/tendermint/tendermint/types"
|
||||
dbm "github.com/tendermint/tmlibs/db"
|
||||
)
|
||||
|
||||
/*
|
||||
"evidence-lookup"/<evidence-height>/<evidence-hash> -> evidence struct
|
||||
"evidence-outqueue"/<index>/<evidence-height>/<evidence-hash> -> nil
|
||||
"evidence-pending"/<evidence-height>/evidence-hash> -> nil
|
||||
*/
|
||||
|
||||
var nullValue = []byte{0}
|
||||
|
||||
type evidenceInfo struct {
|
||||
Committed bool
|
||||
Priority int
|
||||
Evidence types.Evidence
|
||||
}
|
||||
|
||||
func keyLookup(evidence types.Evidence) []byte {
|
||||
return []byte(fmt.Sprintf("evidence-lookup/%d/%X", evidence.Height(), evidence.Hash()))
|
||||
}
|
||||
|
||||
func keyOutqueue(idx int, evidence types.Evidence) []byte {
|
||||
return []byte(fmt.Sprintf("evidence-outqueue/%d/%d/%X", idx, evidence.Height(), evidence.Hash()))
|
||||
}
|
||||
|
||||
func keyPending(evidence types.Evidence) []byte {
|
||||
return []byte(fmt.Sprintf("evidence-pending/%d/%X", evidence.Height(), evidence.Hash()))
|
||||
}
|
||||
|
||||
// EvidenceStore stores all the evidence we've seen, including
|
||||
// evidence that has been committed, evidence that has been seen but not broadcast,
|
||||
// and evidence that has been broadcast but not yet committed.
|
||||
type EvidenceStore struct {
|
||||
chainID string
|
||||
db dbm.DB
|
||||
}
|
||||
|
||||
func NewEvidenceStore(chainID string, db dbm.DB) *EvidenceStore {
|
||||
return &EvidenceStore{
|
||||
chainID: chainID,
|
||||
db: db,
|
||||
}
|
||||
}
|
||||
|
||||
// AddNewEvidence adds the given evidence to the database.
|
||||
func (store *EvidenceStore) AddNewEvidence(idx int, evidence types.Evidence) (bool, error) {
|
||||
// check if we already have seen it
|
||||
key := keyLookup(evidence)
|
||||
v := store.db.Get(key)
|
||||
if len(v) == 0 {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
// verify the evidence
|
||||
if err := evidence.Verify(store.chainID); err != nil {
|
||||
return false, err
|
||||
}
|
||||
|
||||
// add it to the store
|
||||
ei := evidenceInfo{
|
||||
Committed: false,
|
||||
Priority: idx,
|
||||
Evidence: evidence,
|
||||
}
|
||||
store.db.Set(key, wire.BinaryBytes(ei))
|
||||
|
||||
key = keyOutqueue(idx, evidence)
|
||||
store.db.Set(key, nullValue)
|
||||
|
||||
key = keyPending(evidence)
|
||||
store.db.Set(key, nullValue)
|
||||
|
||||
return true, nil
|
||||
}
|
||||
|
||||
// MarkEvidenceAsBroadcasted removes evidence from the outqueue.
|
||||
func (store *EvidenceStore) MarkEvidenceAsBroadcasted(idx int, evidence types.Evidence) {
|
||||
key := keyOutqueue(idx, evidence)
|
||||
store.db.Delete(key)
|
||||
}
|
||||
|
||||
// MarkEvidenceAsPending removes evidence from pending and sets the state to committed.
|
||||
func (store *EvidenceStore) MarkEvidenceAsCommitted(evidence types.Evidence) {
|
||||
key := keyPending(evidence)
|
||||
store.db.Delete(key)
|
||||
|
||||
key = keyLookup(evidence)
|
||||
var ei evidenceInfo
|
||||
b := store.db.Get(key)
|
||||
wire.ReadBinaryBytes(b, &ei)
|
||||
ei.Committed = true
|
||||
store.db.Set(key, wire.BinaryBytes(ei))
|
||||
}
|
@ -437,7 +437,8 @@ func (data *Data) StringIndented(indent string) string {
|
||||
|
||||
// EvidenceData contains any evidence of malicious wrong-doing by validators
|
||||
type EvidenceData struct {
|
||||
Evidences Evidences `json:"evidence"`
|
||||
// TODO: FIXME
|
||||
Evidences evidences `json:"evidence"`
|
||||
|
||||
// Volatile
|
||||
hash data.Bytes
|
||||
|
@ -3,6 +3,7 @@ package types
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"sync"
|
||||
|
||||
"github.com/tendermint/go-crypto"
|
||||
"github.com/tendermint/tmlibs/merkle"
|
||||
@ -27,6 +28,7 @@ func (err *ErrEvidenceInvalid) Error() string {
|
||||
|
||||
// Evidence represents any provable malicious activity by a validator
|
||||
type Evidence interface {
|
||||
Height() int
|
||||
Address() []byte
|
||||
Hash() []byte
|
||||
Verify(chainID string) error
|
||||
@ -37,9 +39,72 @@ type Evidence interface {
|
||||
|
||||
//-------------------------------------------
|
||||
|
||||
type Evidences []Evidence
|
||||
//EvidenceSet is a thread-safe set of evidence.
|
||||
type EvidenceSet struct {
|
||||
sync.RWMutex
|
||||
evidences evidences
|
||||
}
|
||||
|
||||
func (evs Evidences) Hash() []byte {
|
||||
//Evidence returns a copy of all the evidence.
|
||||
func (evset EvidenceSet) Evidence() []Evidence {
|
||||
evset.RLock()
|
||||
defer evset.RUnlock()
|
||||
evCopy := make([]Evidence, len(evset.evidences))
|
||||
for i, ev := range evset.evidences {
|
||||
evCopy[i] = ev
|
||||
}
|
||||
return evCopy
|
||||
}
|
||||
|
||||
// Size returns the number of pieces of evidence in the set.
|
||||
func (evset EvidenceSet) Size() int {
|
||||
evset.RLock()
|
||||
defer evset.RUnlock()
|
||||
return len(evset.evidences)
|
||||
}
|
||||
|
||||
// Hash returns a merkle hash of the evidence.
|
||||
func (evset EvidenceSet) Hash() []byte {
|
||||
evset.RLock()
|
||||
defer evset.RUnlock()
|
||||
return evset.evidences.Hash()
|
||||
}
|
||||
|
||||
// Has returns true if the given evidence is in the set.
|
||||
func (evset EvidenceSet) Has(evidence Evidence) bool {
|
||||
evset.RLock()
|
||||
defer evset.RUnlock()
|
||||
return evset.evidences.Has(evidence)
|
||||
}
|
||||
|
||||
// String returns a string representation of the evidence.
|
||||
func (evset EvidenceSet) String() string {
|
||||
evset.RLock()
|
||||
defer evset.RUnlock()
|
||||
return evset.evidences.String()
|
||||
}
|
||||
|
||||
// Add adds the given evidence to the set.
|
||||
// TODO: and persists it to disk.
|
||||
func (evset EvidenceSet) Add(evidence Evidence) {
|
||||
evset.Lock()
|
||||
defer evset.Unlock()
|
||||
evset.evidences = append(evset.evidences, evidence)
|
||||
}
|
||||
|
||||
// Reset empties the evidence set.
|
||||
func (evset EvidenceSet) Reset() {
|
||||
evset.Lock()
|
||||
defer evset.Unlock()
|
||||
evset.evidences = make(evidences, 0)
|
||||
|
||||
}
|
||||
|
||||
//-------------------------------------------
|
||||
|
||||
type evidences []Evidence
|
||||
|
||||
func (evs evidences) Hash() []byte {
|
||||
// Recursive impl.
|
||||
// Copied from tmlibs/merkle to avoid allocations
|
||||
switch len(evs) {
|
||||
@ -48,13 +113,13 @@ func (evs Evidences) Hash() []byte {
|
||||
case 1:
|
||||
return evs[0].Hash()
|
||||
default:
|
||||
left := Evidences(evs[:(len(evs)+1)/2]).Hash()
|
||||
right := Evidences(evs[(len(evs)+1)/2:]).Hash()
|
||||
left := evidences(evs[:(len(evs)+1)/2]).Hash()
|
||||
right := evidences(evs[(len(evs)+1)/2:]).Hash()
|
||||
return merkle.SimpleHashFromTwoHashes(left, right)
|
||||
}
|
||||
}
|
||||
|
||||
func (evs Evidences) String() string {
|
||||
func (evs evidences) String() string {
|
||||
s := ""
|
||||
for _, e := range evs {
|
||||
s += fmt.Sprintf("%s\t\t", e)
|
||||
@ -62,7 +127,7 @@ func (evs Evidences) String() string {
|
||||
return s
|
||||
}
|
||||
|
||||
func (evs Evidences) Has(evidence Evidence) bool {
|
||||
func (evs evidences) Has(evidence Evidence) bool {
|
||||
for _, ev := range evs {
|
||||
if ev.Equal(evidence) {
|
||||
return true
|
||||
@ -86,6 +151,11 @@ func (dve *DuplicateVoteEvidence) String() string {
|
||||
|
||||
}
|
||||
|
||||
// Height returns the height this evidence refers to.
|
||||
func (dve *DuplicateVoteEvidence) Height() int {
|
||||
return dve.VoteA.Height
|
||||
}
|
||||
|
||||
// Address returns the address of the validator.
|
||||
func (dve *DuplicateVoteEvidence) Address() []byte {
|
||||
return dve.PubKey.Address()
|
||||
|
Loading…
x
Reference in New Issue
Block a user