PSG
/
civetweb
mirror of https://github.com/civetweb/civetweb.git


			
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234
							/*
* Copyright (c) 2013-2015 the CivetWeb developers
* Copyright (c) 2013 No Face Press, LLC
* License http://opensource.org/licenses/mit-license.php MIT License
*/

// Simple example program on how to use Embedded C interface.
#ifdef _WIN32
#include <Windows.h>
#else
#include <unistd.h>
#endif

#include <string.h>
#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");
    mg_printf(conn, "<html><body>");
    mg_printf(conn, "<h2>This is an example text from a C handler</h2>");
    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");
    mg_printf(conn, "Bye!\n");
    exitNow = 1;
    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");
    mg_printf(conn, "<html><body>");
    mg_printf(conn, "<h2>This is the A handler!!!</h2>");
    mg_printf(conn, "</body></html>\n");
    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");
    mg_printf(conn, "<html><body>");
    mg_printf(conn, "<h2>This is the AB handler!!!</h2>");
    mg_printf(conn, "</body></html>\n");
    return 1;
}


int FooHandler(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);

    mg_printf(conn, "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\n");
    mg_printf(conn, "<html><body>");
    mg_printf(conn, "<h2>This is the Foo handler!!!</h2>");
    mg_printf(conn, "<p>The request was:<br><pre>%s %s HTTP/%s</pre></p>",
              req_info->request_method, req_info->uri, req_info->http_version);
    mg_printf(conn, "</body></html>\n");
    return 1;
}


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');\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
#define MAX_WS_CLIENTS 1024
static struct mg_connection * ws_clients[MAX_WS_CLIENTS];
static unsigned long cnt;

int WebSocketConnectHandler(const struct mg_connection * conn, void *cbdata)
{
    int reject = 0;
    fprintf(stdout, "Websocket client %s\r\n\r\n", reject ? "rejected" : "accepted");
    return reject;
}

void WebSocketReadyHandler(struct mg_connection * conn, void *cbdata)
{
    struct mg_context *ctx = mg_get_context(conn);
    int i;

    const char * text = "Hello from the websocket ready handler";
    mg_websocket_write(conn, WEBSOCKET_OPCODE_TEXT, text, strlen(text));
    fprintf(stdout, "Client added to the set of webserver connections\r\n\r\n");
    mg_lock_context(ctx);
    for (i=0; i<MAX_WS_CLIENTS; i++) {
        if (ws_clients[i] == NULL) {
            ws_clients[i] = conn;
            break;
        }
    }
    mg_unlock_context(ctx);
}

int WebsocketDataHandler(struct mg_connection * conn, int bits, char * data, size_t len, void *cbdata)
{
    fprintf(stdout, "Websocket got data:\r\n");
    fwrite(data, len, 1, stdout);
    fprintf(stdout, "\r\n\r\n");

    return 1;
}

void WebSocketCloseHandler(const struct mg_connection * conn, void *cbdata)
{
    struct mg_context *ctx = mg_get_context((struct mg_connection *) /* TODO: check const_casts */ conn);
    int i;

    mg_lock_context(ctx);
    for (i=0; i<MAX_WS_CLIENTS; i++) {
        if (ws_clients[i] == conn) {
            ws_clients[i] = NULL;
            break;
        }
    }
    mg_unlock_context(ctx);
    fprintf(stdout, "Client droped from the set of webserver connections\r\n\r\n");
}

void InformWebsockets(struct mg_context *ctx)
{
    char text[32];
    int i;

    sprintf(text, "%lu", ++cnt);

    mg_lock_context(ctx);
    for (i=0; i<MAX_WS_CLIENTS; i++) {
        if (ws_clients[i] != NULL) {
            mg_websocket_write(ws_clients[i], WEBSOCKET_OPCODE_TEXT, text, strlen(text));
        }
    }
    mg_unlock_context(ctx);
}
#endif


int main(int argc, char *argv[])
{
    const char * options[] = { "document_root", DOCUMENT_ROOT,
                               "listening_ports", PORT, 0
                             };
    struct mg_callbacks callbacks;
    struct mg_context *ctx;

    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, "/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);
    printf("Exit at http://localhost:%s%s\n", PORT, EXIT_URI);

    while (!exitNow) {
#ifdef _WIN32
        Sleep(1000);
#else
        sleep(1);
#endif
#ifdef USE_WEBSOCKET
        InformWebsockets(ctx);
#endif
    }

    mg_stop(ctx);
    printf("Bye!\n");

    return 0;
}