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.
100 lines
3.0 KiB
100 lines
3.0 KiB
package consensus
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/harmony-one/bls/ffi/go/bls"
|
|
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"
|
|
)
|
|
|
|
// Constructs the announce message
|
|
func (consensus *Consensus) constructAnnounceMessage() []byte {
|
|
message := &msg_pb.Message{
|
|
ReceiverType: msg_pb.ReceiverType_VALIDATOR,
|
|
ServiceType: msg_pb.ServiceType_CONSENSUS,
|
|
Type: msg_pb.MessageType_ANNOUNCE,
|
|
Request: &msg_pb.Message_Consensus{
|
|
Consensus: &msg_pb.ConsensusRequest{},
|
|
},
|
|
}
|
|
consensusMsg := message.GetConsensus()
|
|
consensus.populateMessageFields(consensusMsg)
|
|
// n byte of block header
|
|
consensusMsg.Payload = consensus.block // TODO: send only block header in the announce phase.
|
|
|
|
marshaledMessage, err := consensus.signAndMarshalConsensusMessage(message)
|
|
if err != nil {
|
|
utils.GetLogInstance().Error("Failed to sign and marshal the Announce message", "error", err)
|
|
}
|
|
return marshaledMessage
|
|
}
|
|
|
|
// Construct the prepared message, returning prepared message in bytes.
|
|
func (consensus *Consensus) constructPreparedMessage() ([]byte, *bls.Sign) {
|
|
message := &msg_pb.Message{
|
|
ReceiverType: msg_pb.ReceiverType_VALIDATOR,
|
|
ServiceType: msg_pb.ServiceType_CONSENSUS,
|
|
Type: msg_pb.MessageType_PREPARED,
|
|
Request: &msg_pb.Message_Consensus{
|
|
Consensus: &msg_pb.ConsensusRequest{},
|
|
},
|
|
}
|
|
|
|
consensusMsg := message.GetConsensus()
|
|
consensus.populateMessageFields(consensusMsg)
|
|
|
|
//// Payload
|
|
buffer := bytes.NewBuffer([]byte{})
|
|
|
|
// 48 bytes aggregated signature
|
|
aggSig := bls_cosi.AggregateSig(consensus.GetPrepareSigsArray())
|
|
buffer.Write(aggSig.Serialize())
|
|
|
|
// Bitmap
|
|
buffer.Write(consensus.prepareBitmap.Bitmap)
|
|
|
|
consensusMsg.Payload = buffer.Bytes()
|
|
//// END Payload
|
|
|
|
marshaledMessage, err := consensus.signAndMarshalConsensusMessage(message)
|
|
if err != nil {
|
|
utils.GetLogInstance().Error("Failed to sign and marshal the Prepared message", "error", err)
|
|
}
|
|
return marshaledMessage, aggSig
|
|
}
|
|
|
|
// Construct the committed message, returning committed message in bytes.
|
|
func (consensus *Consensus) constructCommittedMessage() ([]byte, *bls.Sign) {
|
|
message := &msg_pb.Message{
|
|
ReceiverType: msg_pb.ReceiverType_VALIDATOR,
|
|
ServiceType: msg_pb.ServiceType_CONSENSUS,
|
|
Type: msg_pb.MessageType_COMMITTED,
|
|
Request: &msg_pb.Message_Consensus{
|
|
Consensus: &msg_pb.ConsensusRequest{},
|
|
},
|
|
}
|
|
|
|
consensusMsg := message.GetConsensus()
|
|
consensus.populateMessageFields(consensusMsg)
|
|
|
|
//// Payload
|
|
buffer := bytes.NewBuffer([]byte{})
|
|
|
|
// 48 bytes aggregated signature
|
|
aggSig := bls_cosi.AggregateSig(consensus.GetCommitSigsArray())
|
|
buffer.Write(aggSig.Serialize())
|
|
|
|
// Bitmap
|
|
buffer.Write(consensus.commitBitmap.Bitmap)
|
|
|
|
consensusMsg.Payload = buffer.Bytes()
|
|
//// END Payload
|
|
|
|
marshaledMessage, err := consensus.signAndMarshalConsensusMessage(message)
|
|
if err != nil {
|
|
utils.GetLogInstance().Error("Failed to sign and marshal the Committed message", "error", err)
|
|
}
|
|
return marshaledMessage, aggSig
|
|
}
|
|
|