2024-05-17 17:43:28 +02:00
|
|
|
package messages
|
|
|
|
|
|
|
|
import (
|
2024-06-25 17:36:04 +02:00
|
|
|
"bytes"
|
2024-07-04 13:42:27 +02:00
|
|
|
"encoding/gob"
|
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
|
|
|
|
2024-07-05 16:12:30 +02:00
|
|
|
sizeOfMsgType = 1
|
|
|
|
sizeOfMagicBye = 4
|
|
|
|
headerSizeTransport = sizeOfMsgType + IDSize // 1 byte for msg type, IDSize for peerID
|
|
|
|
headerSizeHello = sizeOfMsgType + sizeOfMagicBye + 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-07-02 11:57:17 +02:00
|
|
|
type HelloResponse struct {
|
2024-07-05 16:12:30 +02:00
|
|
|
InstanceAddress string
|
2024-07-02 11:57:17 +02:00
|
|
|
}
|
|
|
|
|
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-07-05 16:12:30 +02:00
|
|
|
func MarshalHelloMsg(peerID []byte, additions []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-07-05 16:12:30 +02:00
|
|
|
msg := make([]byte, 5, headerSizeHello+len(additions))
|
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-07-05 16:12:30 +02:00
|
|
|
msg = append(msg, additions...)
|
2024-05-17 17:43:28 +02:00
|
|
|
return msg, nil
|
|
|
|
}
|
|
|
|
|
2024-07-05 16:12:30 +02:00
|
|
|
func UnmarshalHelloMsg(msg []byte) ([]byte, []byte, error) {
|
2024-06-25 17:36:04 +02:00
|
|
|
if len(msg) < headerSizeHello {
|
2024-07-05 16:12:30 +02:00
|
|
|
return nil, 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)
|
2024-07-05 16:12:30 +02:00
|
|
|
return msg[5:], msg[headerSizeHello:], nil
|
2024-05-17 17:43:28 +02:00
|
|
|
}
|
|
|
|
|
2024-07-02 11:57:17 +02:00
|
|
|
func MarshalHelloResponse(DomainAddress string) ([]byte, error) {
|
|
|
|
payload := HelloResponse{
|
2024-07-05 16:12:30 +02:00
|
|
|
InstanceAddress: DomainAddress,
|
2024-07-02 11:57:17 +02:00
|
|
|
}
|
2024-07-04 13:42:27 +02:00
|
|
|
|
|
|
|
buf := new(bytes.Buffer)
|
|
|
|
enc := gob.NewEncoder(buf)
|
|
|
|
|
|
|
|
err := enc.Encode(payload)
|
2024-07-02 11:57:17 +02:00
|
|
|
if err != nil {
|
2024-07-04 13:42:27 +02:00
|
|
|
log.Errorf("failed to gob encode hello response: %s", err)
|
2024-07-02 11:57:17 +02:00
|
|
|
return nil, err
|
|
|
|
}
|
2024-07-04 13:42:27 +02:00
|
|
|
|
|
|
|
msg := make([]byte, 1, 1+buf.Len())
|
2024-05-21 15:51:37 +02:00
|
|
|
msg[0] = byte(MsgTypeHelloResponse)
|
2024-07-04 13:42:27 +02:00
|
|
|
msg = append(msg, buf.Bytes()...)
|
2024-07-02 11:57:17 +02:00
|
|
|
return msg, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func UnmarshalHelloResponse(msg []byte) (string, error) {
|
|
|
|
if len(msg) < 2 {
|
|
|
|
return "", fmt.Errorf("invalid 'hello response' message")
|
|
|
|
}
|
|
|
|
payload := HelloResponse{}
|
2024-07-04 13:42:27 +02:00
|
|
|
buf := bytes.NewBuffer(msg[1:])
|
|
|
|
dec := gob.NewDecoder(buf)
|
|
|
|
|
|
|
|
err := dec.Decode(&payload)
|
2024-07-02 11:57:17 +02:00
|
|
|
if err != nil {
|
2024-07-04 13:42:27 +02:00
|
|
|
log.Errorf("failed to gob decode hello response: %s", err)
|
2024-07-02 11:57:17 +02:00
|
|
|
return "", err
|
|
|
|
}
|
2024-07-05 16:12:30 +02:00
|
|
|
return payload.InstanceAddress, nil
|
2024-05-21 15:51:37 +02:00
|
|
|
}
|
|
|
|
|
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
|
|
|
|
}
|