fix: edge case null websocket

koraykoska · Aug 14, 2023 · c79e692 · c79e692
1 parent c705625
commit c79e692
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/Sources/libwebsockets/WebsocketClient.swift b/Sources/libwebsockets/WebsocketClient.swift
@@ -457,7 +457,16 @@ public class WebsocketClient: WebsocketConnection {
         }
 
         // Connect
-        websocket = lws_client_connect_via_info(&lwsCCInfo)
+        guard let websocket = lws_client_connect_via_info(&lwsCCInfo) else {
+            eventLoop.execute {
+                if let onConnect = self.onConnect {
+                    onConnect.fail(Error.connectionError(description: "lws_client_connect_via_info failed"))
+                    self.onConnect = nil
+                }
+            }
+            return
+        }
+        self.websocket = websocket
 
         // Polling of Events, including connection success
         scheduleServiceCall()