1*1c60b9acSAndroid Build Coastguard Worker /*
2*1c60b9acSAndroid Build Coastguard Worker  * lws-minimal-http-server-eventlib-smp
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 a minimal http[s] server that can work with any of the
10*1c60b9acSAndroid Build Coastguard Worker  * supported event loop backends, or the default poll() one.
11*1c60b9acSAndroid Build Coastguard Worker  *
12*1c60b9acSAndroid Build Coastguard Worker  * To keep it simple, it serves stuff from the subdirectory
13*1c60b9acSAndroid Build Coastguard Worker  * "./mount-origin" of the directory it was started in.
14*1c60b9acSAndroid Build Coastguard Worker  * You can change that by changing mount.origin below.
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 #if defined(WIN32)
22*1c60b9acSAndroid Build Coastguard Worker #define HAVE_STRUCT_TIMESPEC
23*1c60b9acSAndroid Build Coastguard Worker #if defined(pid_t)
24*1c60b9acSAndroid Build Coastguard Worker #undef pid_t
25*1c60b9acSAndroid Build Coastguard Worker #endif
26*1c60b9acSAndroid Build Coastguard Worker #endif
27*1c60b9acSAndroid Build Coastguard Worker 
28*1c60b9acSAndroid Build Coastguard Worker #include <pthread.h>
29*1c60b9acSAndroid Build Coastguard Worker 
30*1c60b9acSAndroid Build Coastguard Worker #define COUNT_THREADS 8
31*1c60b9acSAndroid Build Coastguard Worker 
32*1c60b9acSAndroid Build Coastguard Worker static struct lws_context *context;
33*1c60b9acSAndroid Build Coastguard Worker static volatile int interrupted;
34*1c60b9acSAndroid Build Coastguard Worker 
35*1c60b9acSAndroid Build Coastguard Worker static const struct lws_http_mount mount = {
36*1c60b9acSAndroid Build Coastguard Worker 	/* .mount_next */		NULL,		/* linked-list "next" */
37*1c60b9acSAndroid Build Coastguard Worker 	/* .mountpoint */		"/",		/* mountpoint URL */
38*1c60b9acSAndroid Build Coastguard Worker 	/* .origin */			"./mount-origin", /* serve from dir */
39*1c60b9acSAndroid Build Coastguard Worker 	/* .def */			"index.html",	/* default filename */
40*1c60b9acSAndroid Build Coastguard Worker 	/* .protocol */			NULL,
41*1c60b9acSAndroid Build Coastguard Worker 	/* .cgienv */			NULL,
42*1c60b9acSAndroid Build Coastguard Worker 	/* .extra_mimetypes */		NULL,
43*1c60b9acSAndroid Build Coastguard Worker 	/* .interpret */		NULL,
44*1c60b9acSAndroid Build Coastguard Worker 	/* .cgi_timeout */		0,
45*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_max_age */		0,
46*1c60b9acSAndroid Build Coastguard Worker 	/* .auth_mask */		0,
47*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_reusable */		0,
48*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_revalidate */		0,
49*1c60b9acSAndroid Build Coastguard Worker 	/* .cache_intermediaries */	0,
50*1c60b9acSAndroid Build Coastguard Worker 	/* .origin_protocol */		LWSMPRO_FILE,	/* files in a dir */
51*1c60b9acSAndroid Build Coastguard Worker 	/* .mountpoint_len */		1,		/* char count */
52*1c60b9acSAndroid Build Coastguard Worker 	/* .basic_auth_login_file */	NULL,
53*1c60b9acSAndroid Build Coastguard Worker };
54*1c60b9acSAndroid Build Coastguard Worker 
thread_service(void * threadid)55*1c60b9acSAndroid Build Coastguard Worker void *thread_service(void *threadid)
56*1c60b9acSAndroid Build Coastguard Worker {
57*1c60b9acSAndroid Build Coastguard Worker 	while (lws_service_tsi(context, 10000,
58*1c60b9acSAndroid Build Coastguard Worker 			       (int)(lws_intptr_t)threadid) >= 0 &&
59*1c60b9acSAndroid Build Coastguard Worker 	       !interrupted)
60*1c60b9acSAndroid Build Coastguard Worker 		;
61*1c60b9acSAndroid Build Coastguard Worker 
62*1c60b9acSAndroid Build Coastguard Worker 	pthread_exit(NULL);
63*1c60b9acSAndroid Build Coastguard Worker 
64*1c60b9acSAndroid Build Coastguard Worker 	return NULL;
65*1c60b9acSAndroid Build Coastguard Worker }
66*1c60b9acSAndroid Build Coastguard Worker 
signal_cb(void * handle,int signum)67*1c60b9acSAndroid Build Coastguard Worker void signal_cb(void *handle, int signum)
68*1c60b9acSAndroid Build Coastguard Worker {
69*1c60b9acSAndroid Build Coastguard Worker 	interrupted = 1;
70*1c60b9acSAndroid Build Coastguard Worker 
71*1c60b9acSAndroid Build Coastguard Worker 	switch (signum) {
72*1c60b9acSAndroid Build Coastguard Worker 	case SIGTERM:
73*1c60b9acSAndroid Build Coastguard Worker 	case SIGINT:
74*1c60b9acSAndroid Build Coastguard Worker 		break;
75*1c60b9acSAndroid Build Coastguard Worker 	default:
76*1c60b9acSAndroid Build Coastguard Worker 		lwsl_err("%s: signal %d\n", __func__, signum);
77*1c60b9acSAndroid Build Coastguard Worker 		break;
78*1c60b9acSAndroid Build Coastguard Worker 	}
79*1c60b9acSAndroid Build Coastguard Worker 	lws_context_destroy(context);
80*1c60b9acSAndroid Build Coastguard Worker }
81*1c60b9acSAndroid Build Coastguard Worker 
sigint_handler(int sig)82*1c60b9acSAndroid Build Coastguard Worker void sigint_handler(int sig)
83*1c60b9acSAndroid Build Coastguard Worker {
84*1c60b9acSAndroid Build Coastguard Worker 	signal_cb(NULL, sig);
85*1c60b9acSAndroid Build Coastguard Worker }
86*1c60b9acSAndroid Build Coastguard Worker 
main(int argc,const char ** argv)87*1c60b9acSAndroid Build Coastguard Worker int main(int argc, const char **argv)
88*1c60b9acSAndroid Build Coastguard Worker {
89*1c60b9acSAndroid Build Coastguard Worker 	pthread_t pthread_service[COUNT_THREADS];
90*1c60b9acSAndroid Build Coastguard Worker 	struct lws_context_creation_info info;
91*1c60b9acSAndroid Build Coastguard Worker 	const char *p;
92*1c60b9acSAndroid Build Coastguard Worker 	void *retval;
93*1c60b9acSAndroid Build Coastguard Worker 	int n, 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 	if ((p = lws_cmdline_option(argc, argv, "-d")))
102*1c60b9acSAndroid Build Coastguard Worker 		logs = atoi(p);
103*1c60b9acSAndroid Build Coastguard Worker 
104*1c60b9acSAndroid Build Coastguard Worker 	lws_set_log_level(logs, NULL);
105*1c60b9acSAndroid Build Coastguard Worker 	lwsl_user("LWS minimal http server eventlib SMP | visit http://localhost:7681\n");
106*1c60b9acSAndroid Build Coastguard Worker 	lwsl_user(" [-s (ssl)] [--uv (libuv)] [--ev (libev)] [--event (libevent)]\n");
107*1c60b9acSAndroid Build Coastguard Worker 	lwsl_user("WARNING: Not stable, under development!\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.error_document_404 = "/404.html";
113*1c60b9acSAndroid Build Coastguard Worker 	info.pcontext = &context;
114*1c60b9acSAndroid Build Coastguard Worker 	info.signal_cb = signal_cb;
115*1c60b9acSAndroid Build Coastguard Worker 	info.options =
116*1c60b9acSAndroid Build Coastguard Worker 		LWS_SERVER_OPTION_HTTP_HEADERS_SECURITY_BEST_PRACTICES_ENFORCE;
117*1c60b9acSAndroid Build Coastguard Worker 
118*1c60b9acSAndroid Build Coastguard Worker 	if ((p = lws_cmdline_option(argc, argv, "-t"))) {
119*1c60b9acSAndroid Build Coastguard Worker 		info.count_threads = (unsigned int)atoi(p);
120*1c60b9acSAndroid Build Coastguard Worker 		if (info.count_threads < 1 || info.count_threads > LWS_MAX_SMP)
121*1c60b9acSAndroid Build Coastguard Worker 			return 1;
122*1c60b9acSAndroid Build Coastguard Worker 	} else
123*1c60b9acSAndroid Build Coastguard Worker 		info.count_threads = COUNT_THREADS;
124*1c60b9acSAndroid Build Coastguard Worker 
125*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_TLS)
126*1c60b9acSAndroid Build Coastguard Worker 	if (lws_cmdline_option(argc, argv, "-s")) {
127*1c60b9acSAndroid Build Coastguard Worker 		info.options |= LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT;
128*1c60b9acSAndroid Build Coastguard Worker 		info.ssl_cert_filepath = "localhost-100y.cert";
129*1c60b9acSAndroid Build Coastguard Worker 		info.ssl_private_key_filepath = "localhost-100y.key";
130*1c60b9acSAndroid Build Coastguard Worker 	}
131*1c60b9acSAndroid Build Coastguard Worker #endif
132*1c60b9acSAndroid Build Coastguard Worker 
133*1c60b9acSAndroid Build Coastguard Worker 	if (lws_cmdline_option(argc, argv, "--uv"))
134*1c60b9acSAndroid Build Coastguard Worker 		info.options |= LWS_SERVER_OPTION_LIBUV;
135*1c60b9acSAndroid Build Coastguard Worker 	else
136*1c60b9acSAndroid Build Coastguard Worker 		if (lws_cmdline_option(argc, argv, "--event"))
137*1c60b9acSAndroid Build Coastguard Worker 			info.options |= LWS_SERVER_OPTION_LIBEVENT;
138*1c60b9acSAndroid Build Coastguard Worker 		else
139*1c60b9acSAndroid Build Coastguard Worker 			if (lws_cmdline_option(argc, argv, "--ev"))
140*1c60b9acSAndroid Build Coastguard Worker 				info.options |= LWS_SERVER_OPTION_LIBEV;
141*1c60b9acSAndroid Build Coastguard Worker 			else
142*1c60b9acSAndroid Build Coastguard Worker 				if (lws_cmdline_option(argc, argv, "--glib"))
143*1c60b9acSAndroid Build Coastguard Worker 					info.options |= LWS_SERVER_OPTION_GLIB;
144*1c60b9acSAndroid Build Coastguard Worker 				else
145*1c60b9acSAndroid Build Coastguard Worker 					signal(SIGINT, sigint_handler);
146*1c60b9acSAndroid Build Coastguard Worker 
147*1c60b9acSAndroid Build Coastguard Worker 	context = lws_create_context(&info);
148*1c60b9acSAndroid Build Coastguard Worker 	if (!context) {
149*1c60b9acSAndroid Build Coastguard Worker 		lwsl_err("lws init failed\n");
150*1c60b9acSAndroid Build Coastguard Worker 		return 1;
151*1c60b9acSAndroid Build Coastguard Worker 	}
152*1c60b9acSAndroid Build Coastguard Worker 
153*1c60b9acSAndroid Build Coastguard Worker 	lwsl_notice("  Service threads: %d\n", lws_get_count_threads(context));
154*1c60b9acSAndroid Build Coastguard Worker 
155*1c60b9acSAndroid Build Coastguard Worker 	/* start all the service threads */
156*1c60b9acSAndroid Build Coastguard Worker 
157*1c60b9acSAndroid Build Coastguard Worker 	for (n = 0; n < lws_get_count_threads(context); n++)
158*1c60b9acSAndroid Build Coastguard Worker 		if (pthread_create(&pthread_service[n], NULL, thread_service,
159*1c60b9acSAndroid Build Coastguard Worker 				   (void *)(lws_intptr_t)n))
160*1c60b9acSAndroid Build Coastguard Worker 			lwsl_err("Failed to start service thread\n");
161*1c60b9acSAndroid Build Coastguard Worker 
162*1c60b9acSAndroid Build Coastguard Worker 	/* wait for all the service threads to exit */
163*1c60b9acSAndroid Build Coastguard Worker 
164*1c60b9acSAndroid Build Coastguard Worker 	while ((--n) >= 0)
165*1c60b9acSAndroid Build Coastguard Worker 		pthread_join(pthread_service[n], &retval);
166*1c60b9acSAndroid Build Coastguard Worker 
167*1c60b9acSAndroid Build Coastguard Worker 	lwsl_notice("%s: calling external context destroy\n", __func__);
168*1c60b9acSAndroid Build Coastguard Worker 	lws_context_destroy(context);
169*1c60b9acSAndroid Build Coastguard Worker 
170*1c60b9acSAndroid Build Coastguard Worker 	return 0;
171*1c60b9acSAndroid Build Coastguard Worker }
172