netbird/relay/messages/message.go

157 lines
3.5 KiB
Go
Raw Normal View History

2024-05-17 17:43:28 +02:00
package messages
import (
"bytes"
2024-05-17 17:43:28 +02:00
"fmt"
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
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")
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
func (m MsgType) String() string {
switch m {
case MsgTypeHello:
return "hello"
2024-05-23 13:24:02 +02:00
case MsgTypeHelloResponse:
return "hello response"
case MsgTypeTransport:
return "transport"
2024-06-27 18:40:12 +02:00
case MsgTypeClose:
2024-06-05 19:49:30 +02:00
return "close"
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 {
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:
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 {
return nil, fmt.Errorf("invalid peerID length: %d", len(peerID))
2024-05-17 17:43:28 +02:00
}
msg := make([]byte, 5, headerSizeHello)
2024-05-17 17:43:28 +02:00
msg[0] = byte(MsgTypeHello)
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) {
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
}
bytes.Equal(msg[1:5], magicHeader)
return msg[5:], nil
2024-05-17 17:43:28 +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
func MarshalTransportMsg(peerID []byte, payload []byte) ([]byte, error) {
2024-05-23 13:24:02 +02:00
if len(peerID) != IDSize {
return nil, fmt.Errorf("invalid peerID length: %d", len(peerID))
2024-05-17 17:43:28 +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...)
return msg, nil
2024-05-17 17:43:28 +02:00
}
func UnmarshalTransportMsg(buf []byte) ([]byte, []byte, error) {
if len(buf) < headerSizeTransport {
return nil, nil, ErrInvalidMessageLength
2024-05-17 17:43:28 +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) {
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
}
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
}