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.
118 lines
4.1 KiB
118 lines
4.1 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()
|
|
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() {
|
|
utils.GetLogInstance().Info("Add new block for explorer")
|
|
// Search for the next block in PbftLog and commit the block into blockchain for explorer node.
|
|
for {
|
|
blocks := node.Consensus.PbftLog.GetBlocksByNumber(node.Blockchain().CurrentBlock().NumberU64() + 1)
|
|
if len(blocks) > 1 {
|
|
utils.GetLogInstance().Error("We should have not received more than one block with the same block height.")
|
|
} else if len(blocks) == 0 {
|
|
break
|
|
} else {
|
|
utils.GetLogInstance().Info("Adding new block for explorer node", "blockHeight", blocks[0].NumberU64())
|
|
node.AddNewBlock(blocks[0])
|
|
// Clean up the blocks to avoid OOM.
|
|
node.Consensus.PbftLog.DeleteBlockByNumber(blocks[0].NumberU64())
|
|
}
|
|
}
|
|
}
|
|
|
|
// 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)
|
|
}
|
|
}
|
|
|