lib/libhttp
1
0
Fork 0
mirror of https://github.com/lammertb/libhttp.git synced 2025-07-29 21:01:13 +03:00
Code Activity

Updated indentation

Browse Source
This commit is contained in:
Thomas Davis
2013-09-23 21:48:20 -04:00
parent 55c27d4751
commit b73e2c198d
1 changed files with 75 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

168
examples/ws_server/ws_server.c
View File

@ -54,30 +54,25 @@ static void *ws_server_thread(void *parm)
fprintf(stderr, "ws_server_thread %d\n", wsd); fprintf(stderr, "ws_server_thread %d\n", wsd);
/* Send initial meter updates */ /* Send initial meter updates */
for (i=0; meter[i].period != 0; i++) for (i=0; meter[i].period != 0; i++) {
{
if (meter[i].value >= meter[i].limit) if (meter[i].value >= meter[i].limit)
meter[i].value = 0; meter[i].value = 0;
if (meter[i].value >= meter[i].limit) if (meter[i].value >= meter[i].limit)
meter[i].value = meter[i].limit; meter[i].value = meter[i].limit;
sprintf(tstr, "meter%d:%d,%d", i+1, sprintf(tstr, "meter%d:%d,%d", i+1,
meter[i].value, meter[i].limit); meter[i].value, meter[i].limit);
mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, tstr, strlen(tstr)); mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, tstr, strlen(tstr));
} }
/* While the connection is open, send periodic updates */ /* While the connection is open, send periodic updates */
while(!ws_conn[wsd].closing) while(!ws_conn[wsd].closing) {
{
usleep(100000); /* 0.1 second */ usleep(100000); /* 0.1 second */
timer++; timer++;
/* Send meter updates */ /* Send meter updates */
if (ws_conn[wsd].update) if (ws_conn[wsd].update) {
{ for (i=0; meter[i].period != 0; i++) {
for (i=0; meter[i].period != 0; i++) if (timer%meter[i].period == 0) {
{
if (timer%meter[i].period == 0)
{
if (meter[i].value >= meter[i].limit) if (meter[i].value >= meter[i].limit)
meter[i].value = 0; meter[i].value = 0;
else else
@ -85,9 +80,8 @@ static void *ws_server_thread(void *parm)
if (meter[i].value >= meter[i].limit) if (meter[i].value >= meter[i].limit)
meter[i].value = meter[i].limit; meter[i].value = meter[i].limit;
// if we are closing, server should not send new data // if we are closing, server should not send new data
if (!ws_conn[wsd].closing) if (!ws_conn[wsd].closing) {
{ sprintf(tstr, "meter%d:%d,%d", i+1,
sprintf(tstr, "meter%d:%d,%d", i+1,
meter[i].value, meter[i].limit); meter[i].value, meter[i].limit);
mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, tstr, strlen(tstr)); mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, tstr, strlen(tstr));
} }
@ -114,15 +108,14 @@ static void *ws_server_thread(void *parm)
// On new client connection, find next available server connection and store // On new client connection, find next available server connection and store
// new connection information. If no more server connections are available // new connection information. If no more server connections are available
// tell civetweb to not accept the client request. // tell civetweb to not accept the client request.
static int websocket_connect_handler(const struct mg_connection *conn) { static int websocket_connect_handler(const struct mg_connection *conn)
{
int i; int i;
fprintf(stderr, "connect handler\n"); fprintf(stderr, "connect handler\n");
for(i=0; i < CONNECTIONS; ++i) for(i=0; i < CONNECTIONS; ++i) {
{ if (ws_conn[i].conn == NULL) {
if (ws_conn[i].conn == NULL)
{
fprintf(stderr, "...prep for server %d\n", i); fprintf(stderr, "...prep for server %d\n", i);
ws_conn[i].conn = (struct mg_connection *)conn; ws_conn[i].conn = (struct mg_connection *)conn;
ws_conn[i].closing = 0; ws_conn[i].closing = 0;
@ -130,8 +123,7 @@ static int websocket_connect_handler(const struct mg_connection *conn) {
break; break;
} }
} }
if (i >= CONNECTIONS) if (i >= CONNECTIONS) {
{
fprintf(stderr, "Refused connection: Max connections exceeded\n"); fprintf(stderr, "Refused connection: Max connections exceeded\n");
return 1; return 1;
} }
@ -141,15 +133,14 @@ static int websocket_connect_handler(const struct mg_connection *conn) {
// websocket_ready_handler() // websocket_ready_handler()
// Once websocket negotiation is complete, start a server for the connection // Once websocket negotiation is complete, start a server for the connection
static void websocket_ready_handler(struct mg_connection *conn) { static void websocket_ready_handler(struct mg_connection *conn)
{
int i; int i;
fprintf(stderr, "ready handler\n"); fprintf(stderr, "ready handler\n");
for(i=0; i < CONNECTIONS; ++i) for(i=0; i < CONNECTIONS; ++i) {
{ if (ws_conn[i].conn == conn) {
if (ws_conn[i].conn == conn)
{
fprintf(stderr, "...start server %d\n", i); fprintf(stderr, "...start server %d\n", i);
mg_start_thread(ws_server_thread, (void *)(long)i); mg_start_thread(ws_server_thread, (void *)(long)i);
break; break;
@ -159,15 +150,14 @@ static void websocket_ready_handler(struct mg_connection *conn) {
// websocket_close_handler() // websocket_close_handler()
// When websocket is closed, tell the associated server to shut down // When websocket is closed, tell the associated server to shut down
static void websocket_close_handler(struct mg_connection *conn) { static void websocket_close_handler(struct mg_connection *conn)
{
int i; int i;
//fprintf(stderr, "close handler\n"); /* called for every close, not just websockets */ //fprintf(stderr, "close handler\n"); /* called for every close, not just websockets */
for(i=0; i < CONNECTIONS; ++i) for(i=0; i < CONNECTIONS; ++i) {
{ if (ws_conn[i].conn == conn) {
if (ws_conn[i].conn == conn)
{
fprintf(stderr, "...close server %d\n", i); fprintf(stderr, "...close server %d\n", i);
ws_conn[i].closing = 1; ws_conn[i].closing = 1;
} }
@ -179,74 +169,66 @@ static void websocket_close_handler(struct mg_connection *conn) {
// http://tools.ietf.org/html/rfc6455, section 5.2 // http://tools.ietf.org/html/rfc6455, section 5.2
// data, data_len: payload data. Mask, if any, is already applied. // data, data_len: payload data. Mask, if any, is already applied.
static int websocket_data_handler(struct mg_connection *conn, int flags, static int websocket_data_handler(struct mg_connection *conn, int flags,
char *data, size_t data_len) { char *data, size_t data_len)
{
int i; int i;
int wsd; int wsd;
for(i=0; i < CONNECTIONS; ++i) for(i=0; i < CONNECTIONS; ++i) {
{ if (ws_conn[i].conn == conn) {
if (ws_conn[i].conn == conn)
{
wsd = i; wsd = i;
break; break;
} }
} }
if (i >= CONNECTIONS) if (i >= CONNECTIONS) {
{
fprintf(stderr, "Received websocket data from unknown connection\n"); fprintf(stderr, "Received websocket data from unknown connection\n");
return 1; return 1;
} }
if (flags & 0x80) if (flags & 0x80) {
{ flags &= 0x7f;
flags &= 0x7f; switch (flags) {
switch (flags) case WEBSOCKET_OPCODE_CONTINUATION:
{ fprintf(stderr, "CONTINUATION...\n");
case WEBSOCKET_OPCODE_CONTINUATION: break;
fprintf(stderr, "CONTINUATION...\n"); case WEBSOCKET_OPCODE_TEXT:
break; fprintf(stderr, "TEXT: %-.*s\n", (int)data_len, data);
case WEBSOCKET_OPCODE_TEXT: /*** interpret data as commands here ***/
fprintf(stderr, "TEXT: %-.*s\n", (int)data_len, data); if (strncmp("update on", data, data_len)== 0) {
/*** interpret data as commands here ***/ /* turn on updates */
if (strncmp("update on", data, data_len)== 0) ws_conn[wsd].update = 1;
{ /* echo back */
/* turn on updates */ mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, data, data_len);
ws_conn[wsd].update = 1; } else if (strncmp("update off", data, data_len)== 0) {
/* echo back */ /* turn off updates */
mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, data, data_len); ws_conn[wsd].update = 0;
} /* echo back */
else if (strncmp("update off", data, data_len)== 0) mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, data, data_len);
{ }
/* turn off updates */ break;
ws_conn[wsd].update = 0; case WEBSOCKET_OPCODE_BINARY:
/* echo back */ fprintf(stderr, "BINARY...\n");
mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, data, data_len); break;
} case WEBSOCKET_OPCODE_CONNECTION_CLOSE:
break; fprintf(stderr, "CLOSE...\n");
case WEBSOCKET_OPCODE_BINARY: /* If client initiated close, respond with close message in acknowlegement */
fprintf(stderr, "BINARY...\n"); if (!ws_conn[wsd].closing) {
break; mg_websocket_write(conn, WEBSOCKET_OPCODE_CONNECTION_CLOSE, data, data_len);
case WEBSOCKET_OPCODE_CONNECTION_CLOSE: ws_conn[wsd].closing = 1; /* we should not send addional messages when close requested/acknowledged */
fprintf(stderr, "CLOSE...\n"); }
/* If client initiated close, respond with close message in acknowlegement */ return 0; /* time to close the connection */
if (!ws_conn[wsd].closing) break;
{ case WEBSOCKET_OPCODE_PING:
mg_websocket_write(conn, WEBSOCKET_OPCODE_CONNECTION_CLOSE, data, data_len); /* client sent PING, respond with PONG */
ws_conn[wsd].closing = 1; /* we should not send addional messages when close requested/acknowledged */ mg_websocket_write(conn, WEBSOCKET_OPCODE_PONG, data, data_len);
} break;
return 0; /* time to close the connection */ case WEBSOCKET_OPCODE_PONG:
break; /* received PONG to our PING, no action */
case WEBSOCKET_OPCODE_PING: break;
/* client sent PING, respond with PONG */ default:
mg_websocket_write(conn, WEBSOCKET_OPCODE_PONG, data, data_len); fprintf(stderr, "Unknown flags: %02x\n", flags);
break; break;
case WEBSOCKET_OPCODE_PONG: }
/* received PONG to our PING, no action */
break;
default:
fprintf(stderr, "Unknown flags: %02x\n", flags);
break;
}
} }
return 1; /* keep connection open */ return 1; /* keep connection open */
@ -265,9 +247,9 @@ int main(void)
}; };
/* get simple greeting for the web server */ /* get simple greeting for the web server */
snprintf(server_name, sizeof(server_name), snprintf(server_name, sizeof(server_name),
"Civetweb websocket server v. %s", "Civetweb websocket server v. %s",
mg_version()); mg_version());
memset(&callbacks, 0, sizeof(callbacks)); memset(&callbacks, 0, sizeof(callbacks));
callbacks.websocket_connect = websocket_connect_handler; callbacks.websocket_connect = websocket_connect_handler;
@ -279,8 +261,8 @@ int main(void)
/* show the greeting and some basic information */ /* show the greeting and some basic information */
printf("%s started on port(s) %s with web root [%s]\n", printf("%s started on port(s) %s with web root [%s]\n",
server_name, mg_get_option(ctx, "listening_ports"), server_name, mg_get_option(ctx, "listening_ports"),
mg_get_option(ctx, "document_root")); mg_get_option(ctx, "document_root"));
getchar(); // Wait until user hits "enter" getchar(); // Wait until user hits "enter"
mg_stop(ctx); mg_stop(ctx);