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.
77 lines
1.9 KiB
77 lines
1.9 KiB
package message
|
|
|
|
import (
|
|
"errors"
|
|
)
|
|
|
|
/*
|
|
|
|
Node will process the content of the p2p message
|
|
|
|
---- content start -----
|
|
1 byte - message category
|
|
0x00: COMMITTEE
|
|
0x01: NODE...
|
|
1 byte - message type
|
|
- for COMMITTEE category
|
|
0x00: consensus
|
|
0x01: sharding ...
|
|
- for NODE category
|
|
0x00: transaction ...
|
|
n - 2 bytes - actual message payload
|
|
---- content end -----
|
|
|
|
*/
|
|
|
|
const NODE_TYPE_BYTES = 1
|
|
const ACTION_TYPE_BYTES = 1
|
|
|
|
// The category of messages
|
|
type MessageCategory byte
|
|
|
|
const (
|
|
COMMITTEE MessageCategory = iota
|
|
NODE
|
|
// TODO: add more types
|
|
)
|
|
|
|
// The specific types of message under committee category
|
|
type CommitteeMessageType byte
|
|
|
|
const (
|
|
CONSENSUS CommitteeMessageType = iota
|
|
// TODO: add more types
|
|
)
|
|
|
|
// The specific types of message under node category
|
|
type NodeMessageType byte
|
|
|
|
const (
|
|
TRANSACTION NodeMessageType = iota
|
|
CONTROL
|
|
// TODO: add more types
|
|
)
|
|
|
|
// Get the message category from the p2p message content
|
|
func GetMessageCategory(message []byte) (MessageCategory, error) {
|
|
if len(message) < NODE_TYPE_BYTES {
|
|
return 0, errors.New("Failed to get node type: no data available.")
|
|
}
|
|
return MessageCategory(message[NODE_TYPE_BYTES-1]), nil
|
|
}
|
|
|
|
// Get the action type from the p2p message content
|
|
func GetMessageType(message []byte) (byte, error) {
|
|
if len(message) < NODE_TYPE_BYTES+ACTION_TYPE_BYTES {
|
|
return 0, errors.New("Failed to get action type: no data available.")
|
|
}
|
|
return byte(message[NODE_TYPE_BYTES+ACTION_TYPE_BYTES-1]), nil
|
|
}
|
|
|
|
// Get the node message payload from the p2p message content
|
|
func GetMessagePayload(message []byte) ([]byte, error) {
|
|
if len(message) < NODE_TYPE_BYTES+ACTION_TYPE_BYTES {
|
|
return []byte{}, errors.New("Failed to get message payload: no data available.")
|
|
}
|
|
return message[NODE_TYPE_BYTES+ACTION_TYPE_BYTES:], nil
|
|
}
|
|
|