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.
106 lines
3.6 KiB
106 lines
3.6 KiB
package node
|
|
|
|
import (
|
|
"encoding/binary"
|
|
|
|
"github.com/ethereum/go-ethereum/rlp"
|
|
protobuf "github.com/golang/protobuf/proto"
|
|
msg_pb "github.com/harmony-one/harmony/api/proto/message"
|
|
"github.com/harmony-one/harmony/api/service/explorer"
|
|
"github.com/harmony-one/harmony/consensus"
|
|
"github.com/harmony-one/harmony/core/types"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
)
|
|
|
|
// ExplorerMessageHandler passes received message in node_handler to explorer service
|
|
func (node *Node) ExplorerMessageHandler(payload []byte) {
|
|
if len(payload) == 0 {
|
|
utils.GetLogger().Debug("Payload is empty")
|
|
return
|
|
}
|
|
msg := &msg_pb.Message{}
|
|
err := protobuf.Unmarshal(payload, msg)
|
|
if err != nil {
|
|
utils.GetLogger().Error("Failed to unmarshal message payload.", "err", err)
|
|
return
|
|
}
|
|
|
|
if msg.Type == msg_pb.MessageType_COMMITTED {
|
|
recvMsg, err := consensus.ParsePbftMessage(msg)
|
|
if err != nil {
|
|
utils.GetLogInstance().Debug("[Explorer] onCommitted unable to parse msg", "error", err)
|
|
return
|
|
}
|
|
|
|
aggSig, mask, err := node.Consensus.ReadSignatureBitmapPayload(recvMsg.Payload, 0)
|
|
if err != nil {
|
|
utils.GetLogInstance().Debug("[Explorer] readSignatureBitmapPayload failed", "error", err)
|
|
return
|
|
}
|
|
|
|
// check has 2f+1 signatures
|
|
if count := utils.CountOneBits(mask.Bitmap); count < node.Consensus.Quorum() {
|
|
utils.GetLogInstance().Debug("[Explorer] not have enough signature", "need", node.Consensus.Quorum(), "have", count)
|
|
return
|
|
}
|
|
|
|
blockNumHash := make([]byte, 8)
|
|
binary.LittleEndian.PutUint64(blockNumHash, recvMsg.BlockNum)
|
|
commitPayload := append(blockNumHash, recvMsg.BlockHash[:]...)
|
|
if !aggSig.VerifyHash(mask.AggregatePublic, commitPayload) {
|
|
utils.GetLogInstance().Debug("[Explorer] Failed to verify the multi signature for commit phase", "msgBlock", recvMsg.BlockNum)
|
|
return
|
|
}
|
|
|
|
block := node.Consensus.PbftLog.GetBlockByHash(recvMsg.BlockHash)
|
|
|
|
if block == nil {
|
|
utils.GetLogInstance().Info("Haven't received the block before the committed msg", "msgBlock", recvMsg.BlockNum)
|
|
node.Consensus.PbftLog.AddMessage(recvMsg)
|
|
return
|
|
}
|
|
|
|
node.AddNewBlockForExplorer(block)
|
|
node.commitBlockForExplorer(block)
|
|
} else if msg.Type == msg_pb.MessageType_PREPARED {
|
|
|
|
recvMsg, err := consensus.ParsePbftMessage(msg)
|
|
if err != nil {
|
|
utils.GetLogInstance().Debug("[Explorer] Unable to parse Prepared msg", "error", err)
|
|
return
|
|
}
|
|
block := recvMsg.Block
|
|
|
|
var blockObj types.Block
|
|
err = rlp.DecodeBytes(block, &blockObj)
|
|
msgs := node.Consensus.PbftLog.GetMessagesByTypeSeqHash(msg_pb.MessageType_COMMITTED, blockObj.NumberU64(), blockObj.Hash())
|
|
if len(msgs) > 0 {
|
|
node.commitBlockForExplorer(&blockObj)
|
|
}
|
|
node.Consensus.PbftLog.AddBlock(&blockObj)
|
|
}
|
|
return
|
|
}
|
|
|
|
// AddNewBlockForExplorer add new block for explorer.
|
|
func (node *Node) AddNewBlockForExplorer(block *types.Block) {
|
|
// TODO: Currently assumping blocks come in order.
|
|
utils.GetLogInstance().Info("Add new block for explorer")
|
|
node.AddNewBlock(block)
|
|
}
|
|
|
|
// ExplorerMessageHandler passes received message in node_handler to explorer service
|
|
func (node *Node) commitBlockForExplorer(block *types.Block) {
|
|
if block.ShardID() != node.NodeConfig.ShardID {
|
|
return
|
|
}
|
|
// Dump new block into level db.
|
|
utils.GetLogInstance().Info("[Explorer] Committing block into explorer DB", "blockNum", block.NumberU64())
|
|
explorer.GetStorageInstance(node.SelfPeer.IP, node.SelfPeer.Port, true).Dump(block, block.NumberU64())
|
|
|
|
curNum := block.NumberU64()
|
|
if curNum-100 > 0 {
|
|
node.Consensus.PbftLog.DeleteBlocksLessThan(curNum - 100)
|
|
node.Consensus.PbftLog.DeleteMessagesLessThan(curNum - 100)
|
|
}
|
|
}
|
|
|