Skip to content
Snippets Groups Projects
minimal-ws-client-echo.c 3.89 KiB
Newer Older
  • Learn to ignore specific revisions
  • Andy Green's avatar
    Andy Green committed
    /*
     * lws-minimal-ws-client-echo
     *
     * Copyright (C) 2018 Andy Green <andy@warmcat.com>
     *
     * This file is made available under the Creative Commons CC0 1.0
     * Universal Public Domain Dedication.
     *
     * This demonstrates a ws client that echoes back what it was sent, in a
     * way compatible with autobahn -m fuzzingserver
     */
    
    #include <libwebsockets.h>
    #include <string.h>
    #include <signal.h>
    
    #define LWS_PLUGIN_STATIC
    #include "protocol_lws_minimal_client_echo.c"
    
    static struct lws_protocols protocols[] = {
    	LWS_PLUGIN_PROTOCOL_MINIMAL_CLIENT_ECHO,
    	{ NULL, NULL, 0, 0 } /* terminator */
    };
    
    
    static struct lws_context *context;
    
    Andy Green's avatar
    Andy Green committed
    static int interrupted, port = 7681, options = 0;
    static const char *url = "/", *ads = "localhost";
    
    /* pass pointers to shared vars to the protocol */
    
    static const struct lws_protocol_vhost_options pvo_ads = {
    	NULL,
    	NULL,
    	"ads",		/* pvo name */
    	(void *)&ads	/* pvo value */
    };
    
    static const struct lws_protocol_vhost_options pvo_url = {
    	&pvo_ads,
    	NULL,
    	"url",		/* pvo name */
    	(void *)&url	/* pvo value */
    };
    
    static const struct lws_protocol_vhost_options pvo_options = {
    	&pvo_url,
    	NULL,
    	"options",		/* pvo name */
    	(void *)&options	/* pvo value */
    };
    
    static const struct lws_protocol_vhost_options pvo_port = {
    	&pvo_options,
    	NULL,
    	"port",		/* pvo name */
    	(void *)&port	/* pvo value */
    };
    
    static const struct lws_protocol_vhost_options pvo_interrupted = {
    	&pvo_port,
    	NULL,
    	"interrupted",		/* pvo name */
    	(void *)&interrupted	/* pvo value */
    };
    
    static const struct lws_protocol_vhost_options pvo = {
    	NULL,		/* "next" pvo linked-list */
    	&pvo_interrupted,	/* "child" pvo linked-list */
    	"lws-minimal-client-echo",	/* protocol name we belong to on this vhost */
    	""		/* ignored */
    };
    static const struct lws_extension extensions[] = {
    	{
    		"permessage-deflate",
    		lws_extension_callback_pm_deflate,
    		"permessage-deflate"
    		 "; client_no_context_takeover"
    		 "; client_max_window_bits"
    	},
    	{ NULL, NULL, NULL /* terminator */ }
    };
    
    void sigint_handler(int sig)
    {
    	interrupted = 1;
    }
    
    int main(int argc, const char **argv)
    {
    	struct lws_context_creation_info info;
    	const char *p;
    
    	int n, logs = LLL_USER | LLL_ERR | LLL_WARN | LLL_NOTICE
    
    Andy Green's avatar
    Andy Green committed
    			/* for LLL_ verbosity above NOTICE to be built into lws,
    			 * lws must have been configured and built with
    			 * -DCMAKE_BUILD_TYPE=DEBUG instead of =RELEASE */
    			/* | LLL_INFO */ /* | LLL_PARSER */ /* | LLL_HEADER */
    			/* | LLL_EXT */ /* | LLL_CLIENT */ /* | LLL_LATENCY */
    			/* | LLL_DEBUG */;
    
    	if ((p = lws_cmdline_option(argc, argv, "-d")))
    		logs = atoi(p);
    
    	lws_set_log_level(logs, NULL);
    	lwsl_user("LWS minimal ws client echo + permessage-deflate + multifragment bulk message\n");
    	lwsl_user("   lws-minimal-ws-client-echo [-n (no exts)] [-u url] [-p port] [-o (once)]\n");
    
    	if ((p = lws_cmdline_option(argc, argv, "-u")))
    		url = p;
    
    	if ((p = lws_cmdline_option(argc, argv, "-p")))
    		port = atoi(p);
    
    	if (lws_cmdline_option(argc, argv, "-o"))
    		options |= 1;
    
    
    	if (lws_cmdline_option(argc, argv, "--ssl"))
    		options |= 2;
    
    
    	if ((p = lws_cmdline_option(argc, argv, "-s")))
    		ads = p;
    
    	lwsl_user("options %d, ads %s\n", options, ads);
    
    Andy Green's avatar
    Andy Green committed
    
    	memset(&info, 0, sizeof info); /* otherwise uninitialized garbage */
    	info.port = CONTEXT_PORT_NO_LISTEN;
    	info.protocols = protocols;
    	info.pvo = &pvo;
    	if (!lws_cmdline_option(argc, argv, "-n"))
    		info.extensions = extensions;
    	info.pt_serv_buf_size = 32 * 1024;
    
    	info.options = LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT |
    		       LWS_SERVER_OPTION_VALIDATE_UTF8;
    
    	if (lws_cmdline_option(argc, argv, "--libuv"))
    		info.options |= LWS_SERVER_OPTION_LIBUV;
    	else
    		signal(SIGINT, sigint_handler);
    
    Andy Green's avatar
    Andy Green committed
    
    	context = lws_create_context(&info);
    	if (!context) {
    		lwsl_err("lws init failed\n");
    		return 1;
    	}
    
    
    	while (!lws_service(context, 1000) && !interrupted)
    		;
    
    Andy Green's avatar
    Andy Green committed
    
    	lws_context_destroy(context);
    
    
    	n = (options & 1) ? interrupted != 2 : interrupted == 3;
    	lwsl_user("Completed %d %s\n", interrupted, !n ? "OK" : "failed");
    
    Andy Green's avatar
    Andy Green committed
    
    
    Andy Green's avatar
    Andy Green committed
    }