lib/libhttp
1
0
Fork 0
mirror of https://github.com/lammertb/libhttp.git synced 2025-08-07 16:02:55 +03:00
Code Activity

Add example for cookie handling in C

Browse Source
This commit is contained in:
bel
2016-04-25 23:01:07 +02:00
parent bec1213791
commit 2475a96f48
1 changed files with 59 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
examples/embedded_c/embedded_c.c
View File

@@ -13,6 +13,7 @@
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include <time.h>
#include "civetweb.h" #include "civetweb.h"
@@ -68,7 +69,12 @@ ExampleHandler(struct mg_connection *conn, void *cbdata)
"href=\"close\">click close</a></p>"); "href=\"close\">click close</a></p>");
mg_printf(conn, mg_printf(conn,
"<p>To see a page from the FileHandler handler <a " "<p>To see a page from the FileHandler handler <a "
"href=\"form\">click form</a> (this is the form test page)</p>"); "href=\"form\">click form</a> (the starting point of the "
"<b>form</b> test)</p>");
mg_printf(conn,
"<p>To see a page from the CookieHandler handler <a "
"href=\"cookie\">click cookie</a></p>");
#ifdef USE_WEBSOCKET #ifdef USE_WEBSOCKET
mg_printf(conn, mg_printf(conn,
"<p>To test websocket handler <a href=\"/websocket\">click " "<p>To test websocket handler <a href=\"/websocket\">click "
@@ -246,7 +252,8 @@ FormHandler(struct mg_connection *conn, void *cbdata)
int ret; int ret;
struct mg_form_data_handler fdh = {field_found, field_get, field_stored, 0}; struct mg_form_data_handler fdh = {field_found, field_get, field_stored, 0};
/* TODO: Checks before calling mg_handle_form_request ? */ /* It would be possible to check the request info here before calling
* mg_handle_form_request. */
(void)req_info; (void)req_info;
mg_printf(conn, "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\n"); mg_printf(conn, "HTTP/1.0 200 OK\r\nContent-Type: text/plain\r\n\r\n");
@@ -261,6 +268,51 @@ FormHandler(struct mg_connection *conn, void *cbdata)
} }
int
CookieHandler(struct mg_connection *conn, void *cbdata)
{
/* Handler may access the request info using mg_get_request_info */
const struct mg_request_info *req_info = mg_get_request_info(conn);
const char *cookie = mg_get_header(conn, "Cookie");
char first_str[64], count_str[64];
int count;
(void)mg_get_cookie(cookie, "first", first_str, sizeof(first_str));
(void)mg_get_cookie(cookie, "count", count_str, sizeof(count_str));
mg_printf(conn, "HTTP/1.1 200 OK\r\n");
if (first_str[0] == 0) {
time_t t = time(0);
struct tm *ptm = localtime(&t);
mg_printf(conn,
"Set-Cookie: first=%04i-%02i-%02iT%02i:%02i:%02i\r\n",
ptm->tm_year + 1900,
ptm->tm_mon + 1,
ptm->tm_mday,
ptm->tm_hour,
ptm->tm_min,
ptm->tm_sec);
}
count = (count_str[0] == 0) ? 0 : atoi(count_str);
mg_printf(conn, "Set-Cookie: count=%i\r\n", count + 1);
mg_printf(conn, "Content-Type: text/html\r\n\r\n");
mg_printf(conn, "<html><body>");
mg_printf(conn, "<h2>This is the CookieHandler.</h2>", cbdata);
mg_printf(conn, "<p>The actual uri is %s</p>", req_info->uri);
if (first_str[0] == 0) {
mg_printf(conn, "<p>This is the first time, you opened this page</p>");
} else {
mg_printf(conn, "<p>You opened this page %i times before.</p>", count);
mg_printf(conn, "<p>You first opened this page on %s.</p>", first_str);
}
mg_printf(conn, "</body></html>\n");
return 1;
}
int int
WebSocketStartHandler(struct mg_connection *conn, void *cbdata) WebSocketStartHandler(struct mg_connection *conn, void *cbdata)
{ {
@@ -344,7 +396,8 @@ WebSocketConnectHandler(const struct mg_connection *conn, void *cbdata)
if (ws_clients[i].conn == NULL) { if (ws_clients[i].conn == NULL) {
ws_clients[i].conn = (struct mg_connection *)conn; ws_clients[i].conn = (struct mg_connection *)conn;
ws_clients[i].state = 1; ws_clients[i].state = 1;
mg_set_user_connection_data(conn, (void *)(ws_clients + i)); mg_set_user_connection_data(ws_clients[i].conn,
(void *)(ws_clients + i));
reject = 0; reject = 0;
break; break;
} }
@@ -524,6 +577,9 @@ main(int argc, char *argv[])
FormHandler, FormHandler,
(void *)0); (void *)0);
/* Add handler for /cookie example */
mg_set_request_handler(ctx, "/cookie", CookieHandler, 0);
/* Add HTTP site to open a websocket connection */ /* Add HTTP site to open a websocket connection */
mg_set_request_handler(ctx, "/websocket", WebSocketStartHandler, 0); mg_set_request_handler(ctx, "/websocket", WebSocketStartHandler, 0);