also callback closed if flushing

warmcat · Sep 8, 2014 · b944cc1 · b944cc1
1 parent 8721f4f
commit b944cc1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/libwebsockets.c b/lib/libwebsockets.c
@@ -237,7 +237,8 @@ libwebsocket_close_and_free_session(struct libwebsocket_context *context,
 	if (wsi->protocol && wsi->protocol->callback &&
 			((old_state == WSI_STATE_ESTABLISHED) ||
 			 (old_state == WSI_STATE_RETURNED_CLOSE_ALREADY) ||
-			 (old_state == WSI_STATE_AWAITING_CLOSE_ACK))) {
+			 (old_state == WSI_STATE_AWAITING_CLOSE_ACK) ||
+			 (old_state == WSI_STATE_FLUSHING_STORED_SEND_BEFORE_CLOSE))) {
 		lwsl_debug("calling back CLOSED\n");
 		wsi->protocol->callback(context, wsi, LWS_CALLBACK_CLOSED,
 						      wsi->user_space, NULL, 0);