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.
103 lines
3.2 KiB
103 lines
3.2 KiB
package consensus
|
|
|
|
import (
|
|
"encoding/binary"
|
|
|
|
"github.com/harmony-one/harmony/api/proto"
|
|
msg_pb "github.com/harmony-one/harmony/api/proto/message"
|
|
bls_cosi "github.com/harmony-one/harmony/crypto/bls"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
)
|
|
|
|
// construct the view change message
|
|
func (consensus *Consensus) constructViewChangeMessage() []byte {
|
|
message := &msg_pb.Message{
|
|
ServiceType: msg_pb.ServiceType_CONSENSUS,
|
|
Type: msg_pb.MessageType_VIEWCHANGE,
|
|
Request: &msg_pb.Message_Viewchange{
|
|
Viewchange: &msg_pb.ViewChangeRequest{},
|
|
},
|
|
}
|
|
|
|
vcMsg := message.GetViewchange()
|
|
vcMsg.ViewId = consensus.mode.GetViewID()
|
|
vcMsg.BlockNum = consensus.blockNum
|
|
// sender address
|
|
vcMsg.SenderPubkey = consensus.PubKey.Serialize()
|
|
|
|
// next leader key already updated
|
|
vcMsg.LeaderPubkey = consensus.LeaderPubKey.Serialize()
|
|
|
|
preparedMsgs := consensus.PbftLog.GetMessagesByTypeSeqHash(msg_pb.MessageType_PREPARED, consensus.blockNum, consensus.blockHash)
|
|
preparedMsg := consensus.PbftLog.FindMessageByMaxViewID(preparedMsgs)
|
|
|
|
var msgToSign []byte
|
|
if preparedMsg == nil {
|
|
msgToSign = NIL // m2 type message
|
|
vcMsg.Payload = []byte{}
|
|
} else {
|
|
// m1 type message
|
|
msgToSign = append(preparedMsg.BlockHash[:], preparedMsg.Payload...)
|
|
vcMsg.Payload = append(msgToSign[:0:0], msgToSign...)
|
|
}
|
|
|
|
sign := consensus.priKey.SignHash(msgToSign)
|
|
if sign != nil {
|
|
vcMsg.ViewchangeSig = sign.Serialize()
|
|
} else {
|
|
utils.GetLogger().Error("unable to serialize m1/m2 view change message signature")
|
|
}
|
|
|
|
viewIDHash := make([]byte, 4)
|
|
binary.LittleEndian.PutUint32(viewIDHash, consensus.mode.ViewID())
|
|
sign1 := consensus.priKey.SignHash(viewIDHash)
|
|
if sign1 != nil {
|
|
vcMsg.ViewidSig = sign1.Serialize()
|
|
} else {
|
|
utils.GetLogger().Error("unable to serialize viewID signature")
|
|
}
|
|
|
|
marshaledMessage, err := consensus.signAndMarshalConsensusMessage(message)
|
|
if err != nil {
|
|
utils.GetLogInstance().Error("constructViewChangeMessage failed to sign and marshal the viewchange message", "error", err)
|
|
}
|
|
return proto.ConstructConsensusMessage(marshaledMessage)
|
|
}
|
|
|
|
// new leader construct newview message
|
|
func (consensus *Consensus) constructNewViewMessage() []byte {
|
|
message := &msg_pb.Message{
|
|
ServiceType: msg_pb.ServiceType_CONSENSUS,
|
|
Type: msg_pb.MessageType_NEWVIEW,
|
|
Request: &msg_pb.Message_Viewchange{
|
|
Viewchange: &msg_pb.ViewChangeRequest{},
|
|
},
|
|
}
|
|
|
|
vcMsg := message.GetViewchange()
|
|
vcMsg.ViewId = consensus.mode.GetViewID()
|
|
vcMsg.BlockNum = consensus.blockNum
|
|
// sender address
|
|
vcMsg.SenderPubkey = consensus.PubKey.Serialize()
|
|
vcMsg.Payload = consensus.m1Payload
|
|
|
|
sig2arr := consensus.GetNilSigsArray()
|
|
if len(sig2arr) > 0 {
|
|
m2Sig := bls_cosi.AggregateSig(sig2arr)
|
|
vcMsg.M2Aggsigs = m2Sig.Serialize()
|
|
vcMsg.M2Bitmap = consensus.nilBitmap.Bitmap
|
|
}
|
|
|
|
sig3arr := consensus.GetViewIDSigsArray()
|
|
if len(sig3arr) > 0 {
|
|
m3Sig := bls_cosi.AggregateSig(sig3arr)
|
|
vcMsg.M3Aggsigs = m3Sig.Serialize()
|
|
vcMsg.M3Bitmap = consensus.viewIDBitmap.Bitmap
|
|
}
|
|
|
|
marshaledMessage, err := consensus.signAndMarshalConsensusMessage(message)
|
|
if err != nil {
|
|
utils.GetLogInstance().Error("constructNewViewMessage failed to sign and marshal the new view message", "error", err)
|
|
}
|
|
return proto.ConstructConsensusMessage(marshaledMessage)
|
|
}
|
|
|