[client] Avoid panic when there is no conn client (#2541)

netbirdio · Sep 5, 2024 · bdbd1db · bdbd1db
1 parent f2b5b2e
commit bdbd1db
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/client/internal/connect.go b/client/internal/connect.go
@@ -297,6 +297,9 @@ func (c *ConnectClient) run(
 }
 
 func (c *ConnectClient) Engine() *Engine {
+	if c == nil {
+		return nil
+	}
 	var e *Engine
 	c.engineMutex.Lock()
 	e = c.engine
@@ -305,8 +308,15 @@ func (c *ConnectClient) Engine() *Engine {
 }
 
 func (c *ConnectClient) Stop() error {
+	if c == nil {
+		return nil
+	}
 	c.engineMutex.Lock()
 	defer c.engineMutex.Unlock()
+
+	if c.engine == nil {
+		return nil
+	}
 	return c.engine.Stop()
 }