2017-11-02 12:06:48 -06:00
|
|
|
package evpool
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/tendermint/tmlibs/log"
|
|
|
|
|
2017-11-19 02:02:58 +00:00
|
|
|
cfg "github.com/tendermint/tendermint/config"
|
2017-11-02 12:06:48 -06:00
|
|
|
"github.com/tendermint/tendermint/types"
|
|
|
|
)
|
|
|
|
|
2017-11-19 00:57:55 +00:00
|
|
|
// EvidencePool maintains a pool of valid evidence
|
|
|
|
// in an EvidenceStore.
|
2017-11-02 12:06:48 -06:00
|
|
|
type EvidencePool struct {
|
2017-11-19 02:02:58 +00:00
|
|
|
config *cfg.EvidenceConfig
|
2017-11-02 18:26:07 -06:00
|
|
|
logger log.Logger
|
2017-11-02 12:06:48 -06:00
|
|
|
|
2017-11-19 02:02:58 +00:00
|
|
|
state types.State
|
2017-11-19 00:57:55 +00:00
|
|
|
evidenceStore *EvidenceStore
|
2017-11-02 12:06:48 -06:00
|
|
|
|
2017-11-19 02:02:58 +00:00
|
|
|
evidenceChan chan types.Evidence
|
2017-11-02 12:06:48 -06:00
|
|
|
}
|
|
|
|
|
2017-11-19 02:02:58 +00:00
|
|
|
func NewEvidencePool(config *cfg.EvidenceConfig, evidenceStore *EvidenceStore, state types.State) *EvidencePool {
|
2017-11-02 12:06:48 -06:00
|
|
|
evpool := &EvidencePool{
|
2017-11-19 00:57:55 +00:00
|
|
|
config: config,
|
|
|
|
logger: log.NewNopLogger(),
|
|
|
|
evidenceStore: evidenceStore,
|
2017-11-19 02:02:58 +00:00
|
|
|
state: state,
|
2017-11-19 00:57:55 +00:00
|
|
|
evidenceChan: make(chan types.Evidence),
|
2017-11-02 12:06:48 -06:00
|
|
|
}
|
|
|
|
return evpool
|
|
|
|
}
|
|
|
|
|
|
|
|
// SetLogger sets the Logger.
|
|
|
|
func (evpool *EvidencePool) SetLogger(l log.Logger) {
|
|
|
|
evpool.logger = l
|
|
|
|
}
|
|
|
|
|
2017-11-19 00:57:55 +00:00
|
|
|
// EvidenceChan returns an unbuffered channel on which new evidence can be received.
|
|
|
|
func (evpool *EvidencePool) EvidenceChan() chan types.Evidence {
|
|
|
|
return evpool.evidenceChan
|
2017-11-02 12:06:48 -06:00
|
|
|
}
|
|
|
|
|
2017-11-02 18:26:07 -06:00
|
|
|
// PriorityEvidence returns the priority evidence.
|
|
|
|
func (evpool *EvidencePool) PriorityEvidence() []types.Evidence {
|
2017-11-03 01:50:05 -06:00
|
|
|
return evpool.evidenceStore.PriorityEvidence()
|
2017-11-02 12:06:48 -06:00
|
|
|
}
|
|
|
|
|
2017-11-02 18:26:07 -06:00
|
|
|
// PendingEvidence returns all uncommitted evidence.
|
|
|
|
func (evpool *EvidencePool) PendingEvidence() []types.Evidence {
|
2017-11-03 01:50:05 -06:00
|
|
|
return evpool.evidenceStore.PendingEvidence()
|
2017-11-02 12:06:48 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
// AddEvidence checks the evidence is valid and adds it to the pool.
|
2017-11-19 00:57:55 +00:00
|
|
|
// Blocks on the EvidenceChan.
|
2017-11-02 12:06:48 -06:00
|
|
|
func (evpool *EvidencePool) AddEvidence(evidence types.Evidence) (err error) {
|
2017-11-19 02:02:58 +00:00
|
|
|
|
|
|
|
priority, err := evpool.state.VerifyEvidence(evidence)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
added, err := evpool.evidenceStore.AddNewEvidence(evidence, priority)
|
2017-11-02 18:26:07 -06:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
} else if !added {
|
|
|
|
// evidence already known, just ignore
|
|
|
|
return
|
2017-11-02 12:06:48 -06:00
|
|
|
}
|
2017-11-02 18:26:07 -06:00
|
|
|
|
|
|
|
evpool.logger.Info("Verified new evidence of byzantine behaviour", "evidence", evidence)
|
|
|
|
|
2017-11-19 00:57:55 +00:00
|
|
|
evpool.evidenceChan <- evidence
|
2017-11-02 12:06:48 -06:00
|
|
|
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.
|
2017-11-02 18:26:07 -06:00
|
|
|
func (evpool *EvidencePool) Update(height int, evidence []types.Evidence) {
|
|
|
|
|
|
|
|
// First, create a lookup map of new committed evidence
|
2017-11-02 12:06:48 -06:00
|
|
|
|
|
|
|
evMap := make(map[string]struct{})
|
|
|
|
for _, ev := range evidence {
|
2017-11-02 18:26:07 -06:00
|
|
|
evpool.evidenceStore.MarkEvidenceAsCommitted(ev)
|
|
|
|
evMap[string(ev.Hash())] = struct{}{}
|
2017-11-02 12:06:48 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
// Remove evidence that is already committed .
|
|
|
|
goodEvidence := evpool.filterEvidence(evMap)
|
|
|
|
_ = goodEvidence
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2017-11-02 18:26:07 -06:00
|
|
|
func (evpool *EvidencePool) filterEvidence(blockEvidenceMap map[string]struct{}) []types.Evidence {
|
|
|
|
// TODO:
|
|
|
|
return nil
|
2017-11-02 12:06:48 -06:00
|
|
|
}
|