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.
143 lines
3.8 KiB
143 lines
3.8 KiB
package consensus
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
|
|
"github.com/simple-rules/harmony-benchmark/proto"
|
|
)
|
|
|
|
/*
|
|
Consensus message is the payload of p2p message.
|
|
Consensus message data structure:
|
|
|
|
|
|
Announce:
|
|
---- message start -----
|
|
1 byte - consensus.MessageType
|
|
0x00 - Announce
|
|
0x01 - Commit
|
|
...
|
|
4 byte - consensus id
|
|
32 byte - block hash
|
|
2 byte - leader id
|
|
(n bytes) - consensus payload (the data to run consensus with, e.g. block header data)
|
|
4 byte - payload size
|
|
64 byte - signature
|
|
---- message end -----
|
|
|
|
Commit:
|
|
---- message start -----
|
|
1 byte - consensus.MessageType
|
|
0x00 - Announce
|
|
0x01 - Commit
|
|
...
|
|
4 byte - consensus id
|
|
32 byte - block hash
|
|
2 byte - validator id
|
|
32 byte - commit message (Note it's different than Zilliqa's ECPoint which takes 33 bytes: https://crypto.stackexchange.com/questions/51703/how-to-convert-from-curve25519-33-byte-to-32-byte-representation)
|
|
64 byte - signature
|
|
---- message end -----
|
|
|
|
Challenge:
|
|
---- message start -----
|
|
1 byte - consensus.MessageType
|
|
0x00 - Announce
|
|
0x01 - Commit
|
|
...
|
|
4 byte - consensus id
|
|
32 byte - block hash
|
|
2 byte - leader id
|
|
33 byte - aggregated commit
|
|
33 byte - aggregated key
|
|
32 byte - challenge
|
|
64 byte - signature
|
|
---- message end -----
|
|
|
|
Response:
|
|
---- message start -----
|
|
1 byte - consensus.MessageType
|
|
0x00 - Announce
|
|
0x01 - Commit
|
|
...
|
|
4 byte - consensus id
|
|
32 byte - block hash
|
|
2 byte - validator id
|
|
32 byte - response
|
|
64 byte - signature
|
|
---- message end -----
|
|
*/
|
|
|
|
// MessageTypeBytes is the number of bytes consensus message type occupies
|
|
const MessageTypeBytes = 1
|
|
|
|
// ConsensusMessageType is the specific types of message under Consensus category
|
|
type ConsensusMessageType byte
|
|
|
|
// Consensus message type constants.
|
|
const (
|
|
Consensus ConsensusMessageType = iota
|
|
// TODO: add more types
|
|
)
|
|
|
|
// MessageType is the consensus communication message type.
|
|
// Leader and validator dispatch messages based on incoming message type
|
|
type MessageType int
|
|
|
|
// Message type constants.
|
|
const (
|
|
Announce MessageType = iota
|
|
Commit
|
|
Challenge
|
|
Response
|
|
CollectiveSig
|
|
FinalCommit
|
|
FinalChallenge
|
|
FinalResponse
|
|
StartConsensus
|
|
)
|
|
|
|
// Returns string name for the MessageType enum
|
|
func (msgType MessageType) String() string {
|
|
names := [...]string{
|
|
"Announce",
|
|
"Commit",
|
|
"Challenge",
|
|
"Response",
|
|
"CollectiveSig",
|
|
"FinalCommit",
|
|
"FinalChallenge",
|
|
"FinalResponse",
|
|
"StartConsensus",
|
|
}
|
|
|
|
if msgType < Announce || msgType > StartConsensus {
|
|
return "Unknown"
|
|
}
|
|
return names[msgType]
|
|
}
|
|
|
|
// Get the consensus message type from the consensus message
|
|
func GetConsensusMessageType(message []byte) (MessageType, error) {
|
|
if len(message) < 1 {
|
|
return 0, errors.New("Failed to get consensus message type: no data available.")
|
|
}
|
|
return MessageType(message[0]), nil
|
|
}
|
|
|
|
// Get the consensus message payload from the consensus message
|
|
func GetConsensusMessagePayload(message []byte) ([]byte, error) {
|
|
if len(message) < 2 {
|
|
return []byte{}, errors.New("Failed to get consensus message payload: no data available.")
|
|
}
|
|
return message[MessageTypeBytes:], nil
|
|
}
|
|
|
|
// Concatenate msgType as one byte with payload, and return the whole byte array
|
|
func ConstructConsensusMessage(consensusMsgType MessageType, payload []byte) []byte {
|
|
byteBuffer := bytes.NewBuffer([]byte{byte(proto.Consensus)})
|
|
byteBuffer.WriteByte(byte(Consensus))
|
|
byteBuffer.WriteByte(byte(consensusMsgType))
|
|
byteBuffer.Write(payload)
|
|
return byteBuffer.Bytes()
|
|
}
|
|
|