civetweb/examples/websocket/websocket.c

/* This example uses deprecated interfaces: global websocket callbacks.
   They have been superseeded by URI specific callbacks.
   See examples/embedded_c for an up to date example.
   */

#include <stdio.h>
#include <ctype.h>
#include <string.h>

#include "civetweb.h"
#include "WebSockCallbacks.h"

int main(void)
{
    struct mg_context *ctx = 0;
    struct mg_callbacks callback_funcs = {0};
    tWebSockContext ws_ctx;
    char inbuf[4];

    const char *server_options[] = {
        /* document_root: The path to the test function websock.htm */
        "document_root",     "../../examples/websocket",

        /* port: use http standard to match websocket url in websock.htm: ws://127.0.0.1/MyWebSock  */
        /*       if the port is changed here, it needs to be changed in websock.htm as well         */
        "listening_ports",   "80",

        NULL
    };

    callback_funcs.init_context = websock_init_lib;
    callback_funcs.exit_context = websock_exit_lib;

    ctx = mg_start(&callback_funcs, &ws_ctx, server_options);

    mg_set_websocket_handler(ctx, "/MyWebSock",
                             NULL,
                             websocket_ready_handler,
                             websocket_data_handler,
                             connection_close_handler,
                             NULL);

    printf("Connect to localhost:%s/websock.htm\n", mg_get_option(ctx, "listening_ports"));

    puts("Enter an (ASCII) character or * to exit:");
    for (;;) {
        fgets(inbuf, sizeof(inbuf), stdin);

        if (inbuf[0]=='*') {
           break;
        }
        inbuf[0] = toupper(inbuf[0]);
        websock_send_broadcast(ctx, inbuf, 1);
    }

    mg_stop(ctx);

    return 0;
}