extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-08-09 15:25:20 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add udp listener and did some change for debug purpose.

Browse Source
This commit is contained in:
Zoltan Papp
2024-05-19 12:41:06 +02:00
parent d4eaec5cbd
commit 9ac5a1ed3f
9 changed files with 224 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
relay/client/client.go
View File

@ -10,12 +10,12 @@ import (
log "github.com/sirupsen/logrus"
"github.com/netbirdio/netbird/relay/client/dialer/ws"
"github.com/netbirdio/netbird/relay/client/dialer/udp"
"github.com/netbirdio/netbird/relay/messages"
)
const (
bufferSize = 65535 // optimise the buffer size
bufferSize = 1500 // optimise the buffer size
)
type connContainer struct {
@ -52,7 +52,7 @@ func NewClient(serverAddress, peerID string) *Client {
}
func (c *Client) Connect() error {
conn, err := ws.Dial(c.serverAddress)
conn, err := udp.Dial(c.serverAddress)
if err != nil {
return err
}
@ -128,6 +128,7 @@ func (c *Client) readLoop() {
buf := c.msgPool.Get().([]byte)
n, errExit = c.relayConn.Read(buf)
if errExit != nil {
log.Debugf("failed to read message from relay server: %s", errExit)
break
}
@ -155,7 +156,8 @@ func (c *Client) readLoop() {
c.msgPool.Put(buf)
continue
}
c.handleTransport(channelId, buf[:n])
go c.handleTransport(channelId, buf[:n])
}
}