mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-03 05:24:56 +01:00
0c039274a4
This update adds new relay integration for NetBird clients. The new relay is based on web sockets and listens on a single port. - Adds new relay implementation with websocket with single port relaying mechanism - refactor peer connection logic, allowing upgrade and downgrade from/to P2P connection - peer connections are faster since it connects first to relay and then upgrades to P2P - maintains compatibility with old clients by not using the new relay - updates infrastructure scripts with new relay service
77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
package client
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"time"
|
|
)
|
|
|
|
// Conn represent a connection to a relayed remote peer.
|
|
type Conn struct {
|
|
client *Client
|
|
dstID []byte
|
|
dstStringID string
|
|
messageChan chan Msg
|
|
instanceURL *RelayAddr
|
|
}
|
|
|
|
// NewConn creates a new connection to a relayed remote peer.
|
|
// client: the client instance, it used to send messages to the destination peer
|
|
// dstID: the destination peer ID
|
|
// dstStringID: the destination peer ID in string format
|
|
// messageChan: the channel where the messages will be received
|
|
// instanceURL: the relay instance URL, it used to get the proper server instance address for the remote peer
|
|
func NewConn(client *Client, dstID []byte, dstStringID string, messageChan chan Msg, instanceURL *RelayAddr) *Conn {
|
|
c := &Conn{
|
|
client: client,
|
|
dstID: dstID,
|
|
dstStringID: dstStringID,
|
|
messageChan: messageChan,
|
|
instanceURL: instanceURL,
|
|
}
|
|
|
|
return c
|
|
}
|
|
|
|
func (c *Conn) Write(p []byte) (n int, err error) {
|
|
return c.client.writeTo(c, c.dstStringID, c.dstID, p)
|
|
}
|
|
|
|
func (c *Conn) Read(b []byte) (n int, err error) {
|
|
msg, ok := <-c.messageChan
|
|
if !ok {
|
|
return 0, io.EOF
|
|
}
|
|
|
|
n = copy(b, msg.Payload)
|
|
msg.Free()
|
|
return n, nil
|
|
}
|
|
|
|
func (c *Conn) Close() error {
|
|
return c.client.closeConn(c, c.dstStringID)
|
|
}
|
|
|
|
func (c *Conn) LocalAddr() net.Addr {
|
|
return c.client.relayConn.LocalAddr()
|
|
}
|
|
|
|
func (c *Conn) RemoteAddr() net.Addr {
|
|
return c.instanceURL
|
|
}
|
|
|
|
func (c *Conn) SetDeadline(t time.Time) error {
|
|
//TODO implement me
|
|
panic("SetDeadline is not implemented")
|
|
}
|
|
|
|
func (c *Conn) SetReadDeadline(t time.Time) error {
|
|
//TODO implement me
|
|
panic("SetReadDeadline is not implemented")
|
|
}
|
|
|
|
func (c *Conn) SetWriteDeadline(t time.Time) error {
|
|
//TODO implement me
|
|
panic("SetReadDeadline is not implemented")
|
|
}
|