Fix #373

abcea076 · Sergey Lyubka · 6a54704b · abcea076
Commit abcea076 authored Jun 11, 2014 by Sergey Lyubka
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

mongoose.c mongoose.c +6 -0

No files found.
--- a/mongoose.c
+++ b/mongoose.c
@@ -2712,6 +2712,12 @@ size_t mg_websocket_write(struct mg_connection* conn, int opcode,
      free(copy);
    }

+    // If we send closing frame, schedule a connection to be closed after
+    // data is drained to the client.
+    if (opcode == WEBSOCKET_OPCODE_CONNECTION_CLOSE) {
+      MG_CONN_2_CONN(conn)->ns_conn->flags |= NSF_FINISHED_SENDING_DATA;
+    }
+
    return MG_CONN_2_CONN(conn)->ns_conn->send_iobuf.len;
 }