package consensus import ( "encoding/hex" "github.com/Secured-Finance/dione/beacon" "github.com/sirupsen/logrus" "github.com/Secured-Finance/dione/blockchain" types2 "github.com/Secured-Finance/dione/consensus/types" ) type ConsensusValidator struct { validationFuncMap map[types2.ConsensusMessageType]func(msg types2.ConsensusMessage) bool miner *blockchain.Miner beacon beacon.BeaconNetwork blockchain *blockchain.BlockChain } func NewConsensusValidator(miner *blockchain.Miner, bc *blockchain.BlockChain, b beacon.BeaconNetwork) *ConsensusValidator { cv := &ConsensusValidator{ miner: miner, blockchain: bc, beacon: b, } cv.validationFuncMap = map[types2.ConsensusMessageType]func(msg types2.ConsensusMessage) bool{ types2.ConsensusMessageTypePrePrepare: func(msg types2.ConsensusMessage) bool { if err := cv.blockchain.ValidateBlock(msg.Block); err != nil { logrus.WithFields(logrus.Fields{ "blockHash": hex.EncodeToString(msg.Block.Header.Hash), "err": err.Error(), }).Error("failed to validate block from PrePrepare message") return false } return true }, types2.ConsensusMessageTypePrepare: checkSignatureForBlockhash, types2.ConsensusMessageTypeCommit: checkSignatureForBlockhash, } return cv } func (cv *ConsensusValidator) Valid(msg types2.ConsensusMessage) bool { return cv.validationFuncMap[msg.Type](msg) } func checkSignatureForBlockhash(msg types2.ConsensusMessage) bool { pubKey, err := msg.From.ExtractPublicKey() if err != nil { // TODO logging return false } ok, err := pubKey.Verify(msg.Blockhash, msg.Signature) if err != nil { // TODO logging return false } return ok }