mirror/mongoose
1
0
Fork 0
mirror of https://github.com/cesanta/mongoose.git synced 2024-11-25 03:29:00 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Moved websocket echo server example

Browse Source
This commit is contained in:
Sergey Lyubka 2014-09-09 09:27:35 +01:00
parent b5f6254c1e
commit cfd334ac6c
3 changed files with 31 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/websocket_echo_server/Makefile Normal file
View File

@ -0,0 +1,12 @@
# Copyright (c) 2014 Cesanta Software
# All rights reserved
PROG = websocket_echo_server
CFLAGS = -W -Wall -I../.. -g -O0 $(CFLAGS_EXTRA)
SOURCES = $(PROG).c ../../mongoose.c
$(PROG): $(SOURCES)
$(CC) -o $(PROG) $(SOURCES) $(CFLAGS)
clean:
rm -rf $(PROG) *.exe *.dSYM *.obj *.exp .*o *.lib

0
examples/websocket.html → examples/websocket_echo_server/index.html
View File

42
examples/websocket.c → examples/websocket_echo_server/websocket_echo_server.c
View File

@ -1,27 +1,24 @@
// Copyright (c) 2013-2014 Cesanta Software Limited
// $Date: 2014-09-09 08:27:35 UTC $
#include <string.h> #include <string.h>
#include <time.h> #include <time.h>
#include "mongoose.h" #include "mongoose.h"
#ifdef _WIN32 static void push_message(struct mg_server *server, time_t current_time) {
#define snprintf _snprintf struct mg_connection *c;
#endif
extern const char *find_embedded_file(const char *, size_t *);
static int iterate_callback(struct mg_connection *c, enum mg_event ev) {
if (ev == MG_POLL && c->is_websocket) {
char buf[20]; char buf[20];
int len = snprintf(buf, sizeof(buf), "%lu", int len = sprintf(buf, "%lu", (unsigned long) current_time);
(unsigned long) * (time_t *) c->callback_param);
// Iterate over all connections, and push current time message to websocket ones.
for (c = mg_next(server, NULL); c != NULL; c = mg_next(server, c)) {
if (c->is_websocket) {
mg_websocket_write(c, 1, buf, len); mg_websocket_write(c, 1, buf, len);
} }
return MG_TRUE; }
} }
static int send_reply(struct mg_connection *conn) { static int send_reply(struct mg_connection *conn) {
size_t index_size;
const char *index_html = find_embedded_file("websocket.html", &index_size);
if (conn->is_websocket) { if (conn->is_websocket) {
// This handler is called for each incoming websocket frame, one or more // This handler is called for each incoming websocket frame, one or more
// times for connection lifetime. // times for connection lifetime.
@ -30,19 +27,16 @@ static int send_reply(struct mg_connection *conn) {
return conn->content_len == 4 && !memcmp(conn->content, "exit", 4) ? return conn->content_len == 4 && !memcmp(conn->content, "exit", 4) ?
MG_FALSE : MG_TRUE; MG_FALSE : MG_TRUE;
} else { } else {
mg_send_header(conn, "Content-Type", "text/html"); mg_send_file(conn, "index.html");
mg_send_data(conn, index_html, index_size); return MG_MORE;
return MG_TRUE;
} }
} }
static int ev_handler(struct mg_connection *conn, enum mg_event ev) { static int ev_handler(struct mg_connection *conn, enum mg_event ev) {
if (ev == MG_REQUEST) { switch (ev) {
return send_reply(conn); case MG_AUTH: return MG_TRUE;
} else if (ev == MG_AUTH) { case MG_REQUEST: return send_reply(conn);
return MG_TRUE; default: return MG_FALSE;
} else {
return MG_FALSE;
} }
} }
@ -58,7 +52,7 @@ int main(void) {
current_timer = time(NULL); current_timer = time(NULL);
if (current_timer - last_timer > 0) { if (current_timer - last_timer > 0) {
last_timer = current_timer; last_timer = current_timer;
mg_iterate_over_connections(server, iterate_callback, &current_timer); push_message(server, current_timer);
} }
} }