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

Handler for websocket connection (Step 1/?)

Browse Source 
Add an interface to register handler for websocket connections.
See enhancement #30 and question #101
This commit is contained in:
bel
2015-04-18 18:14:20 +02:00
parent 00877a1649
commit 0016b36204
4 changed files with 156 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

83
examples/embedded_c/embedded_c.c
View File

@ -1,4 +1,5 @@
/*
* Copyright (c) 2013-2015 the CivetWeb developers
* Copyright (c) 2013 No Face Press, LLC
* License http://opensource.org/licenses/mit-license.php MIT License
*/
@ -14,13 +15,13 @@
#include "civetweb.h"
#define DOCUMENT_ROOT "."
#define PORT "8888"
#define EXAMPLE_URI "/example"
#define EXIT_URI "/exit"
int exitNow = 0;
int ExampleHandler(struct mg_connection *conn, void *cbdata)
{
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n");
@ -29,12 +30,14 @@ int ExampleHandler(struct mg_connection *conn, void *cbdata)
mg_printf(conn, "<p>To see a page from the A handler <a href=\"A\">click here</a></p>");
mg_printf(conn, "<p>To see a page from the A/B handler <a href=\"A/B\">click here</a></p>");
mg_printf(conn, "<p>To see a page from the *.foo handler <a href=\"xy.foo\">click here</a></p>");
mg_printf(conn, "<p>To test websocket handler <a href=\"/websocket\">click here</a></p>");
mg_printf(conn, "<p>To exit <a href=\"%s\">click here</a></p>",
EXIT_URI);
mg_printf(conn, "</body></html>\n");
return 1;
}
int ExitHandler(struct mg_connection *conn, void *cbdata)
{
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\n\r\n");
@ -43,6 +46,7 @@ int ExitHandler(struct mg_connection *conn, void *cbdata)
return 1;
}
int AHandler(struct mg_connection *conn, void *cbdata)
{
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n");
@ -52,6 +56,7 @@ int AHandler(struct mg_connection *conn, void *cbdata)
return 1;
}
int ABHandler(struct mg_connection *conn, void *cbdata)
{
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n");
@ -61,6 +66,7 @@ int ABHandler(struct mg_connection *conn, void *cbdata)
return 1;
}
int FooHandler(struct mg_connection *conn, void *cbdata)
{
/* Handler may access the request info using mg_get_request_info */
@ -76,9 +82,70 @@ int FooHandler(struct mg_connection *conn, void *cbdata)
}
int WebSocketStartHandler(struct mg_connection *conn, void *cbdata)
{
mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n");
mg_printf(conn, "<!DOCTYPE html>\n");
mg_printf(conn, "<html>\n<head>\n");
mg_printf(conn, "<meta charset=\"UTF-8\">\n");
mg_printf(conn, "<title>Embedded websocket example</title>\n");
#ifdef USE_WEBSOCKET
/* mg_printf(conn, "<script type=\"text/javascript\"><![CDATA[\n"); ... xhtml style */
mg_printf(conn, "<script>\n");
mg_printf(conn,
"function load() {\n"
" var wsproto = (location.protocol === 'https:') ? 'wss:' : 'ws:';\n"
" connection = new WebSocket(wsproto + '//' + window.location.host + '/websocket.lua');\n"
" websock_text_field = document.getElementById('websock_text_field');\n"
" connection.onmessage = function (e) {\n"
" websock_text_field.innerHtml=e.data;\n"
" }\n"
" connection.onerror = function (error) {\n"
" alert('WebSocket error');\n"
" connection.close();\n"
" }\n"
"}\n"
);
/* mg_printf(conn, "]]></script>\n"); ... xhtml style */
mg_printf(conn, "</script>\n");
mg_printf(conn, "</head>\n<body onload=\"load()\">\n");
mg_printf(conn, "<div id='websock_text_field'>No websocket connection yet</div>\n");
#else
mg_printf(conn, "</head>\n<body>\n");
mg_printf(conn, "Example not compiled with USE_WEBSOCKET\n");
#endif
mg_printf(conn, "</body>\n</html>\n");
return 1;
}
#ifdef USE_WEBSOCKET
int WebSocketConnectHandler(const struct mg_connection * conn, void *cbdata)
{
return 0;
}
void WebSocketReadyHandler(const struct mg_connection * conn, void *cbdata)
{
const char * text = "Hello from the websocket ready handler";
mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, text, strlen(text));
}
int WebsocketDataHandler(const struct mg_connection * conn, int bits, char * data, size_t len, void *cbdata)
{
return 1;
}
void WebSocketCloseHandler(const struct mg_connection * conn, void *cbdata)
{
}
#endif
int main(int argc, char *argv[])
{
const char * options[] = { "document_root", DOCUMENT_ROOT,
"listening_ports", PORT, 0
};
@ -88,11 +155,13 @@ int main(int argc, char *argv[])
memset(&callbacks, 0, sizeof(callbacks));
ctx = mg_start(&callbacks, 0, options);
mg_set_request_handler(ctx,EXAMPLE_URI, ExampleHandler,0);
mg_set_request_handler(ctx,EXIT_URI, ExitHandler,0);
mg_set_request_handler(ctx,"/a", AHandler,0);
mg_set_request_handler(ctx,"/a/b", ABHandler,0);
mg_set_request_handler(ctx,"**.foo$", FooHandler,0);
mg_set_request_handler(ctx, EXAMPLE_URI, ExampleHandler, 0);
mg_set_request_handler(ctx, EXIT_URI, ExitHandler, 0);
mg_set_request_handler(ctx, "/a", AHandler, 0);
mg_set_request_handler(ctx, "/a/b", ABHandler, 0);
mg_set_request_handler(ctx, "**.foo$", FooHandler, 0);
mg_set_request_handler(ctx, "/websocket", WebSocketStartHandler, 0);
mg_set_websocket_handler(ctx, "/websocket", WebSocketConnectHandler, WebSocketReadyHandler, WebsocketDataHandler, WebSocketCloseHandler, 0);
printf("Browse files at http://localhost:%s/\n", PORT);
printf("Run example at http://localhost:%s%s\n", PORT, EXAMPLE_URI);