2024-05-17 17:43:28 +02:00
|
|
|
package messages
|
|
|
|
|
|
|
|
import (
|
2024-06-25 17:36:04 +02:00
|
|
|
"bytes"
|
2024-05-17 17:43:28 +02:00
|
|
|
"fmt"
|
2024-06-03 21:38:37 +02:00
|
|
|
|
|
|
|
log "github.com/sirupsen/logrus"
|
2024-05-17 17:43:28 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2024-05-23 13:24:02 +02:00
|
|
|
MsgTypeHello MsgType = 0
|
|
|
|
MsgTypeHelloResponse MsgType = 1
|
|
|
|
MsgTypeTransport MsgType = 2
|
2024-06-27 18:40:12 +02:00
|
|
|
MsgTypeClose MsgType = 3
|
|
|
|
MsgTypeHealthCheck MsgType = 4
|
2024-06-25 17:36:04 +02:00
|
|
|
|
|
|
|
headerSizeTransport = 1 + IDSize // 1 byte for msg type, IDSize for peerID
|
|
|
|
headerSizeHello = 1 + 4 + IDSize // 1 byte for msg type, 4 byte for magic header, IDSize for peerID
|
2024-06-26 16:22:26 +02:00
|
|
|
|
|
|
|
MaxHandshakeSize = 90
|
2024-05-17 17:43:28 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
ErrInvalidMessageLength = fmt.Errorf("invalid message length")
|
2024-06-25 17:36:04 +02:00
|
|
|
|
|
|
|
magicHeader = []byte{0x21, 0x12, 0xA4, 0x42}
|
2024-06-27 18:40:12 +02:00
|
|
|
|
|
|
|
healthCheckMsg = []byte{byte(MsgTypeHealthCheck)}
|
2024-05-17 17:43:28 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
type MsgType byte
|
|
|
|
|
2024-05-19 12:41:06 +02:00
|
|
|
func (m MsgType) String() string {
|
|
|
|
switch m {
|
|
|
|
case MsgTypeHello:
|
|
|
|
return "hello"
|
2024-05-23 13:24:02 +02:00
|
|
|
case MsgTypeHelloResponse:
|
|
|
|
return "hello response"
|
2024-05-19 12:41:06 +02:00
|
|
|
case MsgTypeTransport:
|
|
|
|
return "transport"
|
2024-06-27 18:40:12 +02:00
|
|
|
case MsgTypeClose:
|
2024-06-05 19:49:30 +02:00
|
|
|
return "close"
|
2024-05-19 12:41:06 +02:00
|
|
|
default:
|
|
|
|
return "unknown"
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2024-05-17 17:43:28 +02:00
|
|
|
func DetermineClientMsgType(msg []byte) (MsgType, error) {
|
|
|
|
msgType := MsgType(msg[0])
|
|
|
|
switch msgType {
|
|
|
|
case MsgTypeHello:
|
|
|
|
return msgType, nil
|
|
|
|
case MsgTypeTransport:
|
|
|
|
return msgType, nil
|
2024-06-27 18:40:12 +02:00
|
|
|
case MsgTypeClose:
|
|
|
|
return msgType, nil
|
|
|
|
case MsgTypeHealthCheck:
|
2024-06-05 19:49:30 +02:00
|
|
|
return msgType, nil
|
2024-05-17 17:43:28 +02:00
|
|
|
default:
|
2024-06-03 20:14:39 +02:00
|
|
|
return 0, fmt.Errorf("invalid msg type, len: %d", len(msg))
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func DetermineServerMsgType(msg []byte) (MsgType, error) {
|
|
|
|
msgType := MsgType(msg[0])
|
|
|
|
switch msgType {
|
2024-05-21 15:51:37 +02:00
|
|
|
case MsgTypeHelloResponse:
|
|
|
|
return msgType, nil
|
2024-05-17 17:43:28 +02:00
|
|
|
case MsgTypeTransport:
|
|
|
|
return msgType, nil
|
2024-06-27 18:40:12 +02:00
|
|
|
case MsgTypeClose:
|
|
|
|
return msgType, nil
|
|
|
|
case MsgTypeHealthCheck:
|
2024-06-05 19:49:30 +02:00
|
|
|
return msgType, nil
|
2024-05-17 17:43:28 +02:00
|
|
|
default:
|
2024-05-21 15:51:37 +02:00
|
|
|
return 0, fmt.Errorf("invalid msg type (len: %d)", len(msg))
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// MarshalHelloMsg initial hello message
|
2024-05-23 13:24:02 +02:00
|
|
|
func MarshalHelloMsg(peerID []byte) ([]byte, error) {
|
|
|
|
if len(peerID) != IDSize {
|
2024-06-25 17:36:04 +02:00
|
|
|
return nil, fmt.Errorf("invalid peerID length: %d", len(peerID))
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
2024-06-25 17:36:04 +02:00
|
|
|
msg := make([]byte, 5, headerSizeHello)
|
2024-05-17 17:43:28 +02:00
|
|
|
msg[0] = byte(MsgTypeHello)
|
2024-06-25 17:36:04 +02:00
|
|
|
copy(msg[1:5], magicHeader)
|
2024-05-23 13:24:02 +02:00
|
|
|
msg = append(msg, peerID...)
|
2024-05-17 17:43:28 +02:00
|
|
|
return msg, nil
|
|
|
|
}
|
|
|
|
|
2024-05-23 13:24:02 +02:00
|
|
|
func UnmarshalHelloMsg(msg []byte) ([]byte, error) {
|
2024-06-25 17:36:04 +02:00
|
|
|
if len(msg) < headerSizeHello {
|
2024-05-23 13:24:02 +02:00
|
|
|
return nil, fmt.Errorf("invalid 'hello' messge")
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
2024-06-25 17:36:04 +02:00
|
|
|
bytes.Equal(msg[1:5], magicHeader)
|
|
|
|
return msg[5:], nil
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
|
|
|
|
2024-05-21 15:51:37 +02:00
|
|
|
func MarshalHelloResponse() []byte {
|
|
|
|
msg := make([]byte, 1)
|
|
|
|
msg[0] = byte(MsgTypeHelloResponse)
|
|
|
|
return msg
|
|
|
|
}
|
|
|
|
|
2024-06-05 19:49:30 +02:00
|
|
|
// Close message
|
|
|
|
|
|
|
|
func MarshalCloseMsg() []byte {
|
|
|
|
msg := make([]byte, 1)
|
2024-06-27 18:40:12 +02:00
|
|
|
msg[0] = byte(MsgTypeClose)
|
|
|
|
return healthCheckMsg
|
2024-06-05 19:49:30 +02:00
|
|
|
}
|
|
|
|
|
2024-05-23 13:24:02 +02:00
|
|
|
// Transport message
|
2024-05-17 17:43:28 +02:00
|
|
|
|
2024-06-25 17:36:04 +02:00
|
|
|
func MarshalTransportMsg(peerID []byte, payload []byte) ([]byte, error) {
|
2024-05-23 13:24:02 +02:00
|
|
|
if len(peerID) != IDSize {
|
2024-06-25 17:36:04 +02:00
|
|
|
return nil, fmt.Errorf("invalid peerID length: %d", len(peerID))
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
|
|
|
|
2024-06-25 17:36:04 +02:00
|
|
|
msg := make([]byte, headerSizeTransport, headerSizeTransport+len(payload))
|
2024-05-17 17:43:28 +02:00
|
|
|
msg[0] = byte(MsgTypeTransport)
|
2024-05-23 13:24:02 +02:00
|
|
|
copy(msg[1:], peerID)
|
2024-05-17 17:43:28 +02:00
|
|
|
msg = append(msg, payload...)
|
2024-06-25 17:36:04 +02:00
|
|
|
return msg, nil
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
|
|
|
|
2024-06-09 20:27:40 +02:00
|
|
|
func UnmarshalTransportMsg(buf []byte) ([]byte, []byte, error) {
|
2024-06-25 17:36:04 +02:00
|
|
|
if len(buf) < headerSizeTransport {
|
2024-06-09 20:27:40 +02:00
|
|
|
return nil, nil, ErrInvalidMessageLength
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
2024-06-09 20:27:40 +02:00
|
|
|
|
2024-06-25 17:36:04 +02:00
|
|
|
return buf[1:headerSizeTransport], buf[headerSizeTransport:], nil
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
|
|
|
|
2024-05-23 13:24:02 +02:00
|
|
|
func UnmarshalTransportID(buf []byte) ([]byte, error) {
|
2024-06-25 17:36:04 +02:00
|
|
|
if len(buf) < headerSizeTransport {
|
|
|
|
log.Debugf("invalid message length: %d, expected: %d, %x", len(buf), headerSizeTransport, buf)
|
2024-05-23 13:24:02 +02:00
|
|
|
return nil, ErrInvalidMessageLength
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
2024-06-25 17:36:04 +02:00
|
|
|
return buf[1:headerSizeTransport], nil
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
|
|
|
|
2024-05-23 13:24:02 +02:00
|
|
|
func UpdateTransportMsg(msg []byte, peerID []byte) error {
|
|
|
|
if len(msg) < 1+len(peerID) {
|
2024-05-17 17:43:28 +02:00
|
|
|
return ErrInvalidMessageLength
|
|
|
|
}
|
2024-05-23 13:24:02 +02:00
|
|
|
copy(msg[1:], peerID)
|
2024-05-17 17:43:28 +02:00
|
|
|
return nil
|
|
|
|
}
|
2024-06-27 18:40:12 +02:00
|
|
|
|
|
|
|
// health check message
|
|
|
|
|
|
|
|
func MarshalHealthcheck() []byte {
|
|
|
|
return healthCheckMsg
|
|
|
|
}
|