From e4d8acc85a5c6427d30af0f988aaa5573acd8f5d Mon Sep 17 00:00:00 2001
From: Boutoukoat <pierre@user.github.invalid.com>
Date: Wed, 28 Jun 2017 22:25:57 +0800
Subject: [PATCH] client: use right state machine

https://github.com/warmcat/libwebsockets/issues/951
---
 lib/client.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/lib/client.c b/lib/client.c
index 0af13aa4..4bb37cb8 100755
--- 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;