wt/mongoose
1
0 Fork 0
mirror of https://github.com/cesanta/mongoose.git synced 2026-11-13 17:00:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle #1523 - multiple WS messages

Browse Source
This commit is contained in:
Sergey Lyubka committed 2022-04-13 11:14:57 +01:00
1 parent 31698678fb
commit e9fabdcc96
1 file changed
+8 -5
Download Patch File Download Diff File Expand all files Collapse all files
test/unit_test.c
+8 -5
View File
@@ -558,20 +558,23 @@ static bool cmpheader(const char *buf, const char *name, const char *value) {
} }
static void wcb(struct mg_connection *c, int ev, void *ev_data, void *fn_data) { static void wcb(struct mg_connection *c, int ev, void *ev_data, void *fn_data) {
int *p = (int *) fn_data;
if (ev == MG_EV_WS_OPEN) { if (ev == MG_EV_WS_OPEN) {
struct mg_http_message *hm = (struct mg_http_message *) ev_data; struct mg_http_message *hm = (struct mg_http_message *) ev_data;
struct mg_str *wsproto = mg_http_get_header(hm, "Sec-WebSocket-Protocol"); struct mg_str *wsproto = mg_http_get_header(hm, "Sec-WebSocket-Protocol");
ASSERT(wsproto != NULL); ASSERT(wsproto != NULL);
mg_ws_send(c, "boo", 3, WEBSOCKET_OP_BINARY); mg_ws_send(c, "boo", 3, WEBSOCKET_OP_BINARY);
mg_ws_send(c, "foobar", 6, WEBSOCKET_OP_BINARY);
mg_ws_send(c, "", 0, WEBSOCKET_OP_PING); mg_ws_send(c, "", 0, WEBSOCKET_OP_PING);
((int *) fn_data)[0] += 100; p[0] += 100;
} else if (ev == MG_EV_WS_MSG) { } else if (ev == MG_EV_WS_MSG) {
struct mg_ws_message *wm = (struct mg_ws_message *) ev_data; struct mg_ws_message *wm = (struct mg_ws_message *) ev_data;
if (mg_strstr(wm->data, mg_str("boo"))) if (mg_strstr(wm->data, mg_str("foobar")))
mg_ws_send(c, "", 0, WEBSOCKET_OP_CLOSE); mg_ws_send(c, "", 0, WEBSOCKET_OP_CLOSE);
((int *) fn_data)[0]++; if (mg_strstr(wm->data, mg_str("boo"))) p[0] += 2;
if (mg_strstr(wm->data, mg_str("foobar"))) p[0] += 3;
} else if (ev == MG_EV_CLOSE) { } else if (ev == MG_EV_CLOSE) {
((int *) fn_data)[0] += 10; p[0] += 10;
} }
} }
@@ -586,7 +589,7 @@ static void test_ws(void) {
mg_ws_connect(&mgr, url, wcb, &done, "%s", "Sec-WebSocket-Protocol: meh\r\n"); mg_ws_connect(&mgr, url, wcb, &done, "%s", "Sec-WebSocket-Protocol: meh\r\n");
for (i = 0; i < 30; i++) mg_mgr_poll(&mgr, 1); for (i = 0; i < 30; i++) mg_mgr_poll(&mgr, 1);
// MG_INFO(("--> %d", done)); // MG_INFO(("--> %d", done));
ASSERT(done == 112); ASSERT(done == 115);
// Test that non-WS requests fail // Test that non-WS requests fail
ASSERT(fetch(&mgr, buf, url, "GET /ws HTTP/1.0\r\n\n") == 426); ASSERT(fetch(&mgr, buf, url, "GET /ws HTTP/1.0\r\n\n") == 426);