1*1c60b9acSAndroid Build Coastguard Worker /*
2*1c60b9acSAndroid Build Coastguard Worker  * lws-minimal-ws-server-timer
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 static int
callback_protocol(struct lws * wsi,enum lws_callback_reasons reason,void * user,void * in,size_t len)22*1c60b9acSAndroid Build Coastguard Worker callback_protocol(struct lws *wsi, enum lws_callback_reasons reason,
23*1c60b9acSAndroid Build Coastguard Worker 			  void *user, void *in, size_t len)
24*1c60b9acSAndroid Build Coastguard Worker {
25*1c60b9acSAndroid Build Coastguard Worker 	switch (reason) {
26*1c60b9acSAndroid Build Coastguard Worker 
27*1c60b9acSAndroid Build Coastguard Worker 	case LWS_CALLBACK_ESTABLISHED:
28*1c60b9acSAndroid Build Coastguard Worker 		lwsl_user("LWS_CALLBACK_ESTABLISHED\n");
29*1c60b9acSAndroid Build Coastguard Worker 		lws_set_timer_usecs(wsi, 20 * LWS_USEC_PER_SEC);
30*1c60b9acSAndroid Build Coastguard Worker 		lws_set_timeout(wsi, 1, 60);
31*1c60b9acSAndroid Build Coastguard Worker 		break;
32*1c60b9acSAndroid Build Coastguard Worker 
33*1c60b9acSAndroid Build Coastguard Worker 	case LWS_CALLBACK_TIMER:
34*1c60b9acSAndroid Build Coastguard Worker 		lwsl_user("LWS_CALLBACK_TIMER\n");
35*1c60b9acSAndroid Build Coastguard Worker 		lws_set_timer_usecs(wsi, 20 * LWS_USEC_PER_SEC);
36*1c60b9acSAndroid Build Coastguard Worker 		lws_set_timeout(wsi, 1, 60);
37*1c60b9acSAndroid Build Coastguard Worker 		break;
38*1c60b9acSAndroid Build Coastguard Worker 
39*1c60b9acSAndroid Build Coastguard Worker 	case LWS_CALLBACK_CLOSED:
40*1c60b9acSAndroid Build Coastguard Worker 		lwsl_user("LWS_CALLBACK_CLOSED\n");
41*1c60b9acSAndroid Build Coastguard Worker 		break;
42*1c60b9acSAndroid Build Coastguard Worker 
43*1c60b9acSAndroid Build Coastguard Worker 	default:
44*1c60b9acSAndroid Build Coastguard Worker 		break;
45*1c60b9acSAndroid Build Coastguard Worker 	}
46*1c60b9acSAndroid Build Coastguard Worker 
47*1c60b9acSAndroid Build Coastguard Worker 	return 0;
48*1c60b9acSAndroid Build Coastguard Worker }
49*1c60b9acSAndroid Build Coastguard Worker 
50*1c60b9acSAndroid Build Coastguard Worker static struct lws_protocols protocols[] = {
51*1c60b9acSAndroid Build Coastguard Worker 	{ "http", lws_callback_http_dummy, 0, 0, 0, NULL, 0 },
52*1c60b9acSAndroid Build Coastguard Worker 	{ "timer", callback_protocol, 0, 0, 0, NULL, 0 },
53*1c60b9acSAndroid Build Coastguard Worker 	LWS_PROTOCOL_LIST_TERM
54*1c60b9acSAndroid Build Coastguard Worker };
55*1c60b9acSAndroid Build Coastguard Worker 
56*1c60b9acSAndroid Build Coastguard Worker static const lws_retry_bo_t retry = {
57*1c60b9acSAndroid Build Coastguard Worker 	.secs_since_valid_ping = 3,
58*1c60b9acSAndroid Build Coastguard Worker 	.secs_since_valid_hangup = 10,
59*1c60b9acSAndroid Build Coastguard Worker };
60*1c60b9acSAndroid Build Coastguard Worker 
61*1c60b9acSAndroid Build Coastguard Worker static int interrupted;
62*1c60b9acSAndroid Build Coastguard Worker 
63*1c60b9acSAndroid Build Coastguard Worker static const struct lws_http_mount mount = {
64*1c60b9acSAndroid Build Coastguard Worker 	/* .mount_next */		NULL,		/* linked-list "next" */
65*1c60b9acSAndroid Build Coastguard Worker 	/* .mountpoint */		"/",		/* mountpoint URL */
66*1c60b9acSAndroid Build Coastguard Worker 	/* .origin */			"./mount-origin",  /* serve from dir */
67*1c60b9acSAndroid Build Coastguard Worker 	/* .def */			"index.html",	/* default filename */
68*1c60b9acSAndroid Build Coastguard Worker 	/* .protocol */			NULL,
69*1c60b9acSAndroid Build Coastguard Worker 	/* .cgienv */			NULL,
70*1c60b9acSAndroid Build Coastguard Worker 	/* .extra_mimetypes */		NULL,
71*1c60b9acSAndroid Build Coastguard Worker 	/* .interpret */		NULL,
72*1c60b9acSAndroid Build Coastguard Worker 	/* .cgi_timeout */		0,
73*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_max_age */		0,
74*1c60b9acSAndroid Build Coastguard Worker 	/* .auth_mask */		0,
75*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_reusable */		0,
76*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_revalidate */		0,
77*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_intermediaries */	0,
78*1c60b9acSAndroid Build Coastguard Worker 	/* .origin_protocol */		LWSMPRO_FILE,	/* files in a dir */
79*1c60b9acSAndroid Build Coastguard Worker 	/* .mountpoint_len */		1,		/* char count */
80*1c60b9acSAndroid Build Coastguard Worker 	/* .basic_auth_login_file */	NULL,
81*1c60b9acSAndroid Build Coastguard Worker };
82*1c60b9acSAndroid Build Coastguard Worker 
sigint_handler(int sig)83*1c60b9acSAndroid Build Coastguard Worker void sigint_handler(int sig)
84*1c60b9acSAndroid Build Coastguard Worker {
85*1c60b9acSAndroid Build Coastguard Worker 	interrupted = 1;
86*1c60b9acSAndroid Build Coastguard Worker }
87*1c60b9acSAndroid Build Coastguard Worker 
main(int argc,const char ** argv)88*1c60b9acSAndroid Build Coastguard Worker int main(int argc, const char **argv)
89*1c60b9acSAndroid Build Coastguard Worker {
90*1c60b9acSAndroid Build Coastguard Worker 	struct lws_context_creation_info info;
91*1c60b9acSAndroid Build Coastguard Worker 	struct lws_context *context;
92*1c60b9acSAndroid Build Coastguard Worker 	const char *p;
93*1c60b9acSAndroid Build Coastguard Worker 	int n = 0, logs = LLL_USER | LLL_ERR | LLL_WARN | LLL_NOTICE
94*1c60b9acSAndroid Build Coastguard Worker 			/* for LLL_ verbosity above NOTICE to be built into lws,
95*1c60b9acSAndroid Build Coastguard Worker 			 * lws must have been configured and built with
96*1c60b9acSAndroid Build Coastguard Worker 			 * -DCMAKE_BUILD_TYPE=DEBUG instead of =RELEASE */
97*1c60b9acSAndroid Build Coastguard Worker 			/* | LLL_INFO */ /* | LLL_PARSER */ /* | LLL_HEADER */
98*1c60b9acSAndroid Build Coastguard Worker 			/* | LLL_EXT */ /* | LLL_CLIENT */ /* | LLL_LATENCY */
99*1c60b9acSAndroid Build Coastguard Worker 			/* | LLL_DEBUG */;
100*1c60b9acSAndroid Build Coastguard Worker 
101*1c60b9acSAndroid Build Coastguard Worker 	signal(SIGINT, sigint_handler);
102*1c60b9acSAndroid Build Coastguard Worker 
103*1c60b9acSAndroid Build Coastguard Worker 	if ((p = lws_cmdline_option(argc, argv, "-d")))
104*1c60b9acSAndroid Build Coastguard Worker 		logs = atoi(p);
105*1c60b9acSAndroid Build Coastguard Worker 
106*1c60b9acSAndroid Build Coastguard Worker 	lws_set_log_level(logs, NULL);
107*1c60b9acSAndroid Build Coastguard Worker 	lwsl_user("LWS minimal ws server | visit http://localhost:7681 (-s = use TLS / https)\n");
108*1c60b9acSAndroid Build Coastguard Worker 
109*1c60b9acSAndroid Build Coastguard Worker 	memset(&info, 0, sizeof info); /* otherwise uninitialized garbage */
110*1c60b9acSAndroid Build Coastguard Worker 	info.port = 7681;
111*1c60b9acSAndroid Build Coastguard Worker 	info.mounts = &mount;
112*1c60b9acSAndroid Build Coastguard Worker 	info.protocols = protocols;
113*1c60b9acSAndroid Build Coastguard Worker 	info.vhost_name = "localhost";
114*1c60b9acSAndroid Build Coastguard Worker 	info.options =
115*1c60b9acSAndroid Build Coastguard Worker 		LWS_SERVER_OPTION_HTTP_HEADERS_SECURITY_BEST_PRACTICES_ENFORCE;
116*1c60b9acSAndroid Build Coastguard Worker 
117*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_TLS)
118*1c60b9acSAndroid Build Coastguard Worker 	if (lws_cmdline_option(argc, argv, "-s")) {
119*1c60b9acSAndroid Build Coastguard Worker 		lwsl_user("Server using TLS\n");
120*1c60b9acSAndroid Build Coastguard Worker 		info.options |= LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT;
121*1c60b9acSAndroid Build Coastguard Worker 		info.ssl_cert_filepath = "localhost-100y.cert";
122*1c60b9acSAndroid Build Coastguard Worker 		info.ssl_private_key_filepath = "localhost-100y.key";
123*1c60b9acSAndroid Build Coastguard Worker 	}
124*1c60b9acSAndroid Build Coastguard Worker #endif
125*1c60b9acSAndroid Build Coastguard Worker 
126*1c60b9acSAndroid Build Coastguard Worker 	if (lws_cmdline_option(argc, argv, "-h"))
127*1c60b9acSAndroid Build Coastguard Worker 		info.options |= LWS_SERVER_OPTION_VHOST_UPG_STRICT_HOST_CHECK;
128*1c60b9acSAndroid Build Coastguard Worker 
129*1c60b9acSAndroid Build Coastguard Worker 	if (lws_cmdline_option(argc, argv, "-v"))
130*1c60b9acSAndroid Build Coastguard Worker 		info.retry_and_idle_policy = &retry;
131*1c60b9acSAndroid Build Coastguard Worker 
132*1c60b9acSAndroid Build Coastguard Worker 	context = lws_create_context(&info);
133*1c60b9acSAndroid Build Coastguard Worker 	if (!context) {
134*1c60b9acSAndroid Build Coastguard Worker 		lwsl_err("lws init failed\n");
135*1c60b9acSAndroid Build Coastguard Worker 		return 1;
136*1c60b9acSAndroid Build Coastguard Worker 	}
137*1c60b9acSAndroid Build Coastguard Worker 
138*1c60b9acSAndroid Build Coastguard Worker 	while (n >= 0 && !interrupted)
139*1c60b9acSAndroid Build Coastguard Worker 		n = lws_service(context, 0);
140*1c60b9acSAndroid Build Coastguard Worker 
141*1c60b9acSAndroid Build Coastguard Worker 	lws_context_destroy(context);
142*1c60b9acSAndroid Build Coastguard Worker 
143*1c60b9acSAndroid Build Coastguard Worker 	return 0;
144*1c60b9acSAndroid Build Coastguard Worker }
145