fix conn client check and validate engine

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