extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2024-12-12 09:50:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix relay instance address indication

Browse Source
This commit is contained in:
Zoltán Papp 2024-07-10 22:33:15 +02:00
parent 83b83ccfd2
commit cb77ff4661
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
relay/client/client.go
View File

@ -111,6 +111,7 @@ type Client struct {
readLoopMutex sync.Mutex
wgReadLoop sync.WaitGroup
instanceURL string
muInstanceURL sync.Mutex
onDisconnectListener func()
listenerMutex sync.Mutex
@ -190,8 +191,8 @@ func (c *Client) OpenConn(dstPeerID string) (net.Conn, error) {
// ServerInstanceURL returns the address of the relay server. It could change after the close and reopen the connection.
func (c *Client) ServerInstanceURL() (string, error) {
c.mu.Lock()
defer c.mu.Unlock()
c.muInstanceURL.Lock()
defer c.muInstanceURL.Unlock()
if c.instanceURL == "" {
return "", fmt.Errorf("relay connection is not established")
}
@ -272,7 +273,9 @@ func (c *Client) handShake() error {
if err != nil {
return err
}
c.muInstanceURL.Lock()
c.instanceURL = ia
c.muInstanceURL.Unlock()
return nil
}
@ -310,6 +313,11 @@ func (c *Client) readLoop(relayConn net.Conn) {
}
hc.Stop()
c.muInstanceURL.Lock()
c.instanceURL = ""
c.muInstanceURL.Unlock()
c.notifyDisconnected()
c.wgReadLoop.Done()
_ = c.close(false)