client: use right state machine

[platform/upstream/libwebsockets.git] / lib / client.c

diff --git a/lib/client.c b/lib/client.c
index 0af13aa..4bb37cb 100755 (executable)
--- a/lib/client.c
+++ b/lib/client.c
@@ -42,7 +42,12 @@ lws_handshake_client(struct lws *wsi, unsigned char **buf, size_t len)
                                return 0;
                        }
                        if (wsi->u.ws.rx_draining_ext) {
-                               m = lws_rx_sm(wsi, 0);
+#if !defined(LWS_NO_CLIENT)
+                               if (wsi->mode == LWSCM_WS_CLIENT)
+                                       m = lws_client_rx_sm(wsi, 0);
+                               else
+#endif
+                                       m = lws_rx_sm(wsi, 0);
                                if (m < 0)
                                        return -1;
                                continue;