syncthing/lib/relay/protocol/protocol.go

132 lines
3.0 KiB
Go
Raw Normal View History

2015-06-28 00:52:01 +00:00
// Copyright (C) 2015 Audrius Butkevicius and Contributors (see the CONTRIBUTORS file).
package protocol
import (
"errors"
2015-06-28 00:52:01 +00:00
"io"
)
const (
magic = 0x9E79BC40
ProtocolName = "bep-relay"
)
var (
ResponseSuccess = Response{0, "success"}
ResponseNotFound = Response{1, "not found"}
ResponseAlreadyConnected = Response{2, "already connected"}
ResponseUnexpectedMessage = Response{100, "unexpected message"}
)
func WriteMessage(w io.Writer, message interface{}) error {
header := header{
magic: magic,
}
var payload []byte
var err error
switch msg := message.(type) {
case Ping:
payload, err = msg.MarshalXDR()
header.messageType = messageTypePing
case Pong:
payload, err = msg.MarshalXDR()
header.messageType = messageTypePong
case JoinRelayRequest:
payload, err = msg.MarshalXDR()
header.messageType = messageTypeJoinRelayRequest
case JoinSessionRequest:
payload, err = msg.MarshalXDR()
header.messageType = messageTypeJoinSessionRequest
case Response:
payload, err = msg.MarshalXDR()
header.messageType = messageTypeResponse
case ConnectRequest:
payload, err = msg.MarshalXDR()
header.messageType = messageTypeConnectRequest
case SessionInvitation:
payload, err = msg.MarshalXDR()
header.messageType = messageTypeSessionInvitation
2015-11-20 23:42:49 +00:00
case RelayFull:
payload, err = msg.MarshalXDR()
header.messageType = messageTypeRelayFull
2015-06-28 00:52:01 +00:00
default:
err = errors.New("unknown message type")
2015-06-28 00:52:01 +00:00
}
if err != nil {
return err
}
header.messageLength = int32(len(payload))
headerpayload, err := header.MarshalXDR()
if err != nil {
return err
}
_, err = w.Write(append(headerpayload, payload...))
return err
}
func ReadMessage(r io.Reader) (interface{}, error) {
var header header
2016-02-02 11:48:09 +00:00
buf := make([]byte, header.XDRSize())
if _, err := io.ReadFull(r, buf); err != nil {
return nil, err
}
if err := header.UnmarshalXDR(buf); err != nil {
2015-06-28 00:52:01 +00:00
return nil, err
}
if header.magic != magic {
return nil, errors.New("magic mismatch")
2015-06-28 00:52:01 +00:00
}
2016-02-02 11:48:09 +00:00
buf = make([]byte, int(header.messageLength))
if _, err := io.ReadFull(r, buf); err != nil {
return nil, err
}
2015-06-28 00:52:01 +00:00
switch header.messageType {
case messageTypePing:
var msg Ping
2016-02-02 11:48:09 +00:00
err := msg.UnmarshalXDR(buf)
2015-06-28 00:52:01 +00:00
return msg, err
case messageTypePong:
var msg Pong
2016-02-02 11:48:09 +00:00
err := msg.UnmarshalXDR(buf)
2015-06-28 00:52:01 +00:00
return msg, err
case messageTypeJoinRelayRequest:
var msg JoinRelayRequest
2016-02-02 11:48:09 +00:00
err := msg.UnmarshalXDR(buf)
2015-06-28 00:52:01 +00:00
return msg, err
case messageTypeJoinSessionRequest:
var msg JoinSessionRequest
2016-02-02 11:48:09 +00:00
err := msg.UnmarshalXDR(buf)
2015-06-28 00:52:01 +00:00
return msg, err
case messageTypeResponse:
var msg Response
2016-02-02 11:48:09 +00:00
err := msg.UnmarshalXDR(buf)
2015-06-28 00:52:01 +00:00
return msg, err
case messageTypeConnectRequest:
var msg ConnectRequest
2016-02-02 11:48:09 +00:00
err := msg.UnmarshalXDR(buf)
2015-06-28 00:52:01 +00:00
return msg, err
case messageTypeSessionInvitation:
var msg SessionInvitation
2016-02-02 11:48:09 +00:00
err := msg.UnmarshalXDR(buf)
2015-06-28 00:52:01 +00:00
return msg, err
2015-11-20 23:42:49 +00:00
case messageTypeRelayFull:
var msg RelayFull
2016-02-02 11:48:09 +00:00
err := msg.UnmarshalXDR(buf)
2015-11-20 23:42:49 +00:00
return msg, err
2015-06-28 00:52:01 +00:00
}
return nil, errors.New("unknown message type")
2015-06-28 00:52:01 +00:00
}