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.
59 lines
1.9 KiB
59 lines
1.9 KiB
package consensus
|
|
|
|
import (
|
|
"github.com/harmony-one/harmony/api/proto"
|
|
msg_pb "github.com/harmony-one/harmony/api/proto/message"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
)
|
|
|
|
// Construct the prepare message to send to leader (assumption the consensus data is already verified)
|
|
func (consensus *Consensus) constructPrepareMessage() []byte {
|
|
message := &msg_pb.Message{
|
|
ServiceType: msg_pb.ServiceType_CONSENSUS,
|
|
Type: msg_pb.MessageType_PREPARE,
|
|
Request: &msg_pb.Message_Consensus{
|
|
Consensus: &msg_pb.ConsensusRequest{},
|
|
},
|
|
}
|
|
|
|
consensusMsg := message.GetConsensus()
|
|
consensus.populateMessageFields(consensusMsg)
|
|
|
|
// 96 byte of bls signature
|
|
sign := consensus.priKey.SignHash(consensusMsg.BlockHash)
|
|
if sign != nil {
|
|
consensusMsg.Payload = sign.Serialize()
|
|
}
|
|
|
|
marshaledMessage, err := consensus.signAndMarshalConsensusMessage(message)
|
|
if err != nil {
|
|
utils.Logger().Error().Err(err).Msg("Failed to sign and marshal the Prepare message")
|
|
}
|
|
return proto.ConstructConsensusMessage(marshaledMessage)
|
|
}
|
|
|
|
// Construct the commit message which contains the signature on the multi-sig of prepare phase.
|
|
func (consensus *Consensus) constructCommitMessage(commitPayload []byte) []byte {
|
|
message := &msg_pb.Message{
|
|
ServiceType: msg_pb.ServiceType_CONSENSUS,
|
|
Type: msg_pb.MessageType_COMMIT,
|
|
Request: &msg_pb.Message_Consensus{
|
|
Consensus: &msg_pb.ConsensusRequest{},
|
|
},
|
|
}
|
|
|
|
consensusMsg := message.GetConsensus()
|
|
consensus.populateMessageFields(consensusMsg)
|
|
|
|
// 96 byte of bls signature
|
|
sign := consensus.priKey.SignHash(commitPayload)
|
|
if sign != nil {
|
|
consensusMsg.Payload = sign.Serialize()
|
|
}
|
|
|
|
marshaledMessage, err := consensus.signAndMarshalConsensusMessage(message)
|
|
if err != nil {
|
|
utils.Logger().Error().Err(err).Msg("Failed to sign and marshal the Commit message")
|
|
}
|
|
return proto.ConstructConsensusMessage(marshaledMessage)
|
|
}
|
|
|