mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-15 03:11:02 +01:00
30f025e7dd
When the remote peer switches the Relay instance then must to close the proxy connection to the old instance. It can cause issues when the remote peer switch connects to the Relay instance multiple times and then reconnects to an instance it had previously connected to.
76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
package client
|
|
|
|
import (
|
|
"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, net.ErrClosed
|
|
}
|
|
|
|
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")
|
|
}
|