1*1c60b9acSAndroid Build Coastguard Worker /*
2*1c60b9acSAndroid Build Coastguard Worker  * lws-minimal-ws-server
3*1c60b9acSAndroid Build Coastguard Worker  *
4*1c60b9acSAndroid Build Coastguard Worker  * Written in 2010-2019 by Andy Green <[email protected]>
5*1c60b9acSAndroid Build Coastguard Worker  *
6*1c60b9acSAndroid Build Coastguard Worker  * This file is made available under the Creative Commons CC0 1.0
7*1c60b9acSAndroid Build Coastguard Worker  * Universal Public Domain Dedication.
8*1c60b9acSAndroid Build Coastguard Worker  *
9*1c60b9acSAndroid Build Coastguard Worker  * This demonstrates the most minimal http server you can make with lws,
10*1c60b9acSAndroid Build Coastguard Worker  * with an added websocket chat server.
11*1c60b9acSAndroid Build Coastguard Worker  *
12*1c60b9acSAndroid Build Coastguard Worker  * To keep it simple, it serves stuff in the subdirectory "./mount-origin" of
13*1c60b9acSAndroid Build Coastguard Worker  * the directory it was started in.
14*1c60b9acSAndroid Build Coastguard Worker  * You can change that by changing mount.origin.
15*1c60b9acSAndroid Build Coastguard Worker  */
16*1c60b9acSAndroid Build Coastguard Worker 
17*1c60b9acSAndroid Build Coastguard Worker #include <libwebsockets.h>
18*1c60b9acSAndroid Build Coastguard Worker #include <string.h>
19*1c60b9acSAndroid Build Coastguard Worker #include <signal.h>
20*1c60b9acSAndroid Build Coastguard Worker 
21*1c60b9acSAndroid Build Coastguard Worker #define LWS_PLUGIN_STATIC
22*1c60b9acSAndroid Build Coastguard Worker #include "protocol_lws_minimal.c"
23*1c60b9acSAndroid Build Coastguard Worker 
24*1c60b9acSAndroid Build Coastguard Worker static struct lws_protocols protocols[] = {
25*1c60b9acSAndroid Build Coastguard Worker 	{ "http", lws_callback_http_dummy, 0, 0, 0, NULL, 0},
26*1c60b9acSAndroid Build Coastguard Worker 	LWS_PLUGIN_PROTOCOL_MINIMAL,
27*1c60b9acSAndroid Build Coastguard Worker 	LWS_PROTOCOL_LIST_TERM
28*1c60b9acSAndroid Build Coastguard Worker };
29*1c60b9acSAndroid Build Coastguard Worker 
30*1c60b9acSAndroid Build Coastguard Worker static const lws_retry_bo_t retry = {
31*1c60b9acSAndroid Build Coastguard Worker 	.secs_since_valid_ping = 3,
32*1c60b9acSAndroid Build Coastguard Worker 	.secs_since_valid_hangup = 10,
33*1c60b9acSAndroid Build Coastguard Worker };
34*1c60b9acSAndroid Build Coastguard Worker 
35*1c60b9acSAndroid Build Coastguard Worker static int interrupted;
36*1c60b9acSAndroid Build Coastguard Worker 
37*1c60b9acSAndroid Build Coastguard Worker static const struct lws_http_mount mount = {
38*1c60b9acSAndroid Build Coastguard Worker 	/* .mount_next */		NULL,		/* linked-list "next" */
39*1c60b9acSAndroid Build Coastguard Worker 	/* .mountpoint */		"/",		/* mountpoint URL */
40*1c60b9acSAndroid Build Coastguard Worker 	/* .origin */			"./mount-origin",  /* serve from dir */
41*1c60b9acSAndroid Build Coastguard Worker 	/* .def */			"index.html",	/* default filename */
42*1c60b9acSAndroid Build Coastguard Worker 	/* .protocol */			NULL,
43*1c60b9acSAndroid Build Coastguard Worker 	/* .cgienv */			NULL,
44*1c60b9acSAndroid Build Coastguard Worker 	/* .extra_mimetypes */		NULL,
45*1c60b9acSAndroid Build Coastguard Worker 	/* .interpret */		NULL,
46*1c60b9acSAndroid Build Coastguard Worker 	/* .cgi_timeout */		0,
47*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_max_age */		0,
48*1c60b9acSAndroid Build Coastguard Worker 	/* .auth_mask */		0,
49*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_reusable */		0,
50*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_revalidate */		0,
51*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_intermediaries */	0,
52*1c60b9acSAndroid Build Coastguard Worker 	/* .origin_protocol */		LWSMPRO_FILE,	/* files in a dir */
53*1c60b9acSAndroid Build Coastguard Worker 	/* .mountpoint_len */		1,		/* char count */
54*1c60b9acSAndroid Build Coastguard Worker 	/* .basic_auth_login_file */	NULL,
55*1c60b9acSAndroid Build Coastguard Worker };
56*1c60b9acSAndroid Build Coastguard Worker 
57*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_PLUGINS)
58*1c60b9acSAndroid Build Coastguard Worker /* if plugins enabled, only protocols explicitly named in pvo bind to vhost */
59*1c60b9acSAndroid Build Coastguard Worker static struct lws_protocol_vhost_options pvo = { NULL, NULL, "lws-minimal", "" };
60*1c60b9acSAndroid Build Coastguard Worker #endif
61*1c60b9acSAndroid Build Coastguard Worker 
sigint_handler(int sig)62*1c60b9acSAndroid Build Coastguard Worker void sigint_handler(int sig)
63*1c60b9acSAndroid Build Coastguard Worker {
64*1c60b9acSAndroid Build Coastguard Worker 	interrupted = 1;
65*1c60b9acSAndroid Build Coastguard Worker }
66*1c60b9acSAndroid Build Coastguard Worker 
main(int argc,const char ** argv)67*1c60b9acSAndroid Build Coastguard Worker int main(int argc, const char **argv)
68*1c60b9acSAndroid Build Coastguard Worker {
69*1c60b9acSAndroid Build Coastguard Worker 	struct lws_context_creation_info info;
70*1c60b9acSAndroid Build Coastguard Worker 	struct lws_context *context;
71*1c60b9acSAndroid Build Coastguard Worker 	const char *p;
72*1c60b9acSAndroid Build Coastguard Worker 	int n = 0, logs = LLL_USER | LLL_ERR | LLL_WARN | LLL_NOTICE
73*1c60b9acSAndroid Build Coastguard Worker 			/* for LLL_ verbosity above NOTICE to be built into lws,
74*1c60b9acSAndroid Build Coastguard Worker 			 * lws must have been configured and built with
75*1c60b9acSAndroid Build Coastguard Worker 			 * -DCMAKE_BUILD_TYPE=DEBUG instead of =RELEASE */
76*1c60b9acSAndroid Build Coastguard Worker 			/* | LLL_INFO */ /* | LLL_PARSER */ /* | LLL_HEADER */
77*1c60b9acSAndroid Build Coastguard Worker 			/* | LLL_EXT */ /* | LLL_CLIENT */ /* | LLL_LATENCY */
78*1c60b9acSAndroid Build Coastguard Worker 			/* | LLL_DEBUG */;
79*1c60b9acSAndroid Build Coastguard Worker 
80*1c60b9acSAndroid Build Coastguard Worker 	signal(SIGINT, sigint_handler);
81*1c60b9acSAndroid Build Coastguard Worker 
82*1c60b9acSAndroid Build Coastguard Worker 	if ((p = lws_cmdline_option(argc, argv, "-d")))
83*1c60b9acSAndroid Build Coastguard Worker 		logs = atoi(p);
84*1c60b9acSAndroid Build Coastguard Worker 
85*1c60b9acSAndroid Build Coastguard Worker 	lws_set_log_level(logs, NULL);
86*1c60b9acSAndroid Build Coastguard Worker 	lwsl_user("LWS minimal ws server | visit http://localhost:7681 (-s = use TLS / https)\n");
87*1c60b9acSAndroid Build Coastguard Worker 
88*1c60b9acSAndroid Build Coastguard Worker 	memset(&info, 0, sizeof info); /* otherwise uninitialized garbage */
89*1c60b9acSAndroid Build Coastguard Worker 	info.port = 7681;
90*1c60b9acSAndroid Build Coastguard Worker 	info.mounts = &mount;
91*1c60b9acSAndroid Build Coastguard Worker 	info.protocols = protocols;
92*1c60b9acSAndroid Build Coastguard Worker 	info.vhost_name = "localhost";
93*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_PLUGINS)
94*1c60b9acSAndroid Build Coastguard Worker 	info.pvo = &pvo;
95*1c60b9acSAndroid Build Coastguard Worker #endif
96*1c60b9acSAndroid Build Coastguard Worker 	info.options =
97*1c60b9acSAndroid Build Coastguard Worker 		LWS_SERVER_OPTION_HTTP_HEADERS_SECURITY_BEST_PRACTICES_ENFORCE;
98*1c60b9acSAndroid Build Coastguard Worker 
99*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_TLS)
100*1c60b9acSAndroid Build Coastguard Worker 	if (lws_cmdline_option(argc, argv, "-s")) {
101*1c60b9acSAndroid Build Coastguard Worker 		lwsl_user("Server using TLS\n");
102*1c60b9acSAndroid Build Coastguard Worker 		info.options |= LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT;
103*1c60b9acSAndroid Build Coastguard Worker 		info.ssl_cert_filepath = "localhost-100y.cert";
104*1c60b9acSAndroid Build Coastguard Worker 		info.ssl_private_key_filepath = "localhost-100y.key";
105*1c60b9acSAndroid Build Coastguard Worker 	}
106*1c60b9acSAndroid Build Coastguard Worker #endif
107*1c60b9acSAndroid Build Coastguard Worker 
108*1c60b9acSAndroid Build Coastguard Worker 	if (lws_cmdline_option(argc, argv, "-h"))
109*1c60b9acSAndroid Build Coastguard Worker 		info.options |= LWS_SERVER_OPTION_VHOST_UPG_STRICT_HOST_CHECK;
110*1c60b9acSAndroid Build Coastguard Worker 
111*1c60b9acSAndroid Build Coastguard Worker 	if (lws_cmdline_option(argc, argv, "-v"))
112*1c60b9acSAndroid Build Coastguard Worker 		info.retry_and_idle_policy = &retry;
113*1c60b9acSAndroid Build Coastguard Worker 
114*1c60b9acSAndroid Build Coastguard Worker 	context = lws_create_context(&info);
115*1c60b9acSAndroid Build Coastguard Worker 	if (!context) {
116*1c60b9acSAndroid Build Coastguard Worker 		lwsl_err("lws init failed\n");
117*1c60b9acSAndroid Build Coastguard Worker 		return 1;
118*1c60b9acSAndroid Build Coastguard Worker 	}
119*1c60b9acSAndroid Build Coastguard Worker 
120*1c60b9acSAndroid Build Coastguard Worker 	while (n >= 0 && !interrupted)
121*1c60b9acSAndroid Build Coastguard Worker 		n = lws_service(context, 0);
122*1c60b9acSAndroid Build Coastguard Worker 
123*1c60b9acSAndroid Build Coastguard Worker 	lws_context_destroy(context);
124*1c60b9acSAndroid Build Coastguard Worker 
125*1c60b9acSAndroid Build Coastguard Worker 	return 0;
126*1c60b9acSAndroid Build Coastguard Worker }
127