sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-07-17 12:09:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix use mgr without start

Browse Source
This commit is contained in:
Zoltan Papp
2025-02-28 16:12:09 +01:00
parent 3ff3bbe782
commit a89a6c00de
2 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
client/internal/conn_mgr.go
View File

@@ -71,7 +71,7 @@ func (e *ConnMgr) AddPeerConn(peerKey string, conn *peer.Conn) (exists bool) {
return true
}
if e.lazyConnMgr == nil {
if !e.isStarted() {
conn.Open()
return
}
@@ -101,7 +101,7 @@ func (e *ConnMgr) OnSignalMsg(peerKey string) (*peer.Conn, bool) {
return nil, false
}
if e.lazyConnMgr == nil {
if !e.isStarted() {
return conn, true
}
@@ -119,7 +119,7 @@ func (e *ConnMgr) RemovePeerConn(peerKey string) {
}
defer conn.Close()
if e.lazyConnMgr == nil {
if !e.isStarted() {
return
}
@@ -128,9 +128,10 @@ func (e *ConnMgr) RemovePeerConn(peerKey string) {
}
func (e *ConnMgr) Close() {
if e.lazyConnMgr == nil {
if !e.isStarted() {
return
}
e.ctxCancel()
e.lazyConnMgr.Close()
e.wg.Wait()
@@ -147,3 +148,7 @@ func (e *ConnMgr) receiveLazyConnEvents(ctx context.Context) {
e.peerStore.PeerConnOpen(peerID)
}
}
func (e *ConnMgr) isStarted() bool {
return e.lazyConnMgr != nil && e.ctxCancel != nil
}

3
client/internal/engine_test.go
View File

@@ -332,7 +332,6 @@ func TestEngine_SSH(t *testing.T) {
}
func TestEngine_UpdateNetworkMap(t *testing.T) {
t.Setenv(envDisableLazyConn, "true")
// test setup
key, err := wgtypes.GeneratePrivateKey()
if err != nil {
@@ -766,6 +765,7 @@ func TestEngine_UpdateNetworkMapWithRoutes(t *testing.T) {
engine.routeManager = mockRouteManager
engine.dnsServer = &dns.MockServer{}
engine.connMgr = NewConnMgr(engine.peerStore, engine.wgInterface)
defer func() {
exitErr := engine.Stop()
@@ -962,6 +962,7 @@ func TestEngine_UpdateNetworkMapWithDNSUpdate(t *testing.T) {
}
engine.dnsServer = mockDNSServer
engine.connMgr = NewConnMgr(engine.peerStore, engine.wgInterface)
defer func() {
exitErr := engine.Stop()