commit d4238b8f97b83963523cecf18cd5a031bf8f06d3
author Jack R. Dunaway <jack@wirebirdlabs.com> Thu Nov 05 10:58:01 2015 -0600
committer Garrett D'Amore <garrett@damore.org> Tue Nov 10 10:39:08 2015 -0800
tree 8072b876c1aba61ff7c031839243258bc020a79a
parent 5511bc316cb968c2d1175163b8b14ec8982a127b

fixes #505 hang in websocket test

src/transports/ws/cws.c

diff --git a/src/transports/ws/cws.c b/src/transports/ws/cws.c
index 7842b6f..08b32a9 100644
--- a/src/transports/ws/cws.c
+++ b/src/transports/ws/cws.c
@@ -86,6 +86,9 @@
     /*  Used to wait before retrying to connect. */
     struct nn_backoff retry;
 
+    /*  Defines message validation and framing. */
+    uint8_t msg_type;
+
     /*  State machine that handles the active part of the connection
         lifetime. */
     struct nn_sws sws;
@@ -143,6 +146,7 @@
     struct nn_cws *self;
     int reconnect_ivl;
     int reconnect_ivl_max;
+    int msg_type;
     size_t sz;
 
     /*  Allocate the new endpoint object. */
@@ -235,6 +239,12 @@
         nn_epbase_getctx (&self->epbase));
     self->state = NN_CWS_STATE_IDLE;
     nn_usock_init (&self->usock, NN_CWS_SRC_USOCK, &self->fsm);
+    
+    sz = sizeof (msg_type);
+    nn_epbase_getopt (&self->epbase, NN_WS, NN_WS_MSG_TYPE,
+        &msg_type, &sz);
+    nn_assert (sz == sizeof (msg_type));
+    self->msg_type = (uint8_t) msg_type;
 
     sz = sizeof (reconnect_ivl);
     nn_epbase_getopt (&self->epbase, NN_SOL_SOCKET, NN_RECONNECT_IVL,
@@ -330,8 +340,6 @@
     NN_UNUSED void *srcptr)
 {
     struct nn_cws *cws;
-    int msg_type;
-    size_t sz;
 
     cws = nn_cont (self, struct nn_cws, fsm);
 
@@ -414,13 +422,9 @@
         case NN_CWS_SRC_USOCK:
             switch (type) {
             case NN_USOCK_CONNECTED:
-                sz = sizeof (msg_type);
-                nn_epbase_getopt (&cws->epbase, NN_WS, NN_WS_MSG_TYPE,
-                    &msg_type, &sz);
-                nn_assert(sz == sizeof (msg_type));
                 nn_sws_start (&cws->sws, &cws->usock, NN_WS_CLIENT,
                     nn_chunkref_data (&cws->resource),
-                    nn_chunkref_data (&cws->remote_host), (uint8_t)msg_type);
+                    nn_chunkref_data (&cws->remote_host), cws->msg_type);
                 cws->state = NN_CWS_STATE_ACTIVE;
                 cws->peer_gone = 0;
                 nn_epbase_stat_increment (&cws->epbase,