Commit cfd334ac authored by Sergey Lyubka's avatar Sergey Lyubka

Moved websocket echo server example

parent b5f6254c
# 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
// 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) { ...@@ -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) { ...@@ -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);
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment