1*1c60b9acSAndroid Build Coastguard Worker /*
2*1c60b9acSAndroid Build Coastguard Worker  * lws-minimal-http-server-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 multithreaded http server you can make with lws.
10*1c60b9acSAndroid Build Coastguard Worker  *
11*1c60b9acSAndroid Build Coastguard Worker  * To keep it simple, it serves stuff in the subdirectory "./mount-origin" of
12*1c60b9acSAndroid Build Coastguard Worker  * the directory it was started in.
13*1c60b9acSAndroid Build Coastguard Worker  * You can change that by changing mount.origin.
14*1c60b9acSAndroid Build Coastguard Worker  *
15*1c60b9acSAndroid Build Coastguard Worker  * Also for simplicity the number of threads is set in the code... note that
16*1c60b9acSAndroid Build Coastguard Worker  * the real number of threads possible is decided by the LWS_MAX_SMP that lws
17*1c60b9acSAndroid Build Coastguard Worker  * was configured with, by default that is 1.  Lws will limit the number of
18*1c60b9acSAndroid Build Coastguard Worker  * requested threads to the number possible.
19*1c60b9acSAndroid Build Coastguard Worker  */
20*1c60b9acSAndroid Build Coastguard Worker 
21*1c60b9acSAndroid Build Coastguard Worker #include <libwebsockets.h>
22*1c60b9acSAndroid Build Coastguard Worker #include <string.h>
23*1c60b9acSAndroid Build Coastguard Worker #include <signal.h>
24*1c60b9acSAndroid Build Coastguard Worker #if defined(WIN32)
25*1c60b9acSAndroid Build Coastguard Worker #define HAVE_STRUCT_TIMESPEC
26*1c60b9acSAndroid Build Coastguard Worker #if defined(pid_t)
27*1c60b9acSAndroid Build Coastguard Worker #undef pid_t
28*1c60b9acSAndroid Build Coastguard Worker #endif
29*1c60b9acSAndroid Build Coastguard Worker #endif
30*1c60b9acSAndroid Build Coastguard Worker #include <pthread.h>
31*1c60b9acSAndroid Build Coastguard Worker 
32*1c60b9acSAndroid Build Coastguard Worker #define COUNT_THREADS 8
33*1c60b9acSAndroid Build Coastguard Worker 
34*1c60b9acSAndroid Build Coastguard Worker static struct lws_context *context;
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 
thread_service(void * threadid)57*1c60b9acSAndroid Build Coastguard Worker void *thread_service(void *threadid)
58*1c60b9acSAndroid Build Coastguard Worker {
59*1c60b9acSAndroid Build Coastguard Worker 	while (lws_service_tsi(context, 10000,
60*1c60b9acSAndroid Build Coastguard Worker 			       (int)(lws_intptr_t)threadid) >= 0 &&
61*1c60b9acSAndroid Build Coastguard Worker 	       !interrupted)
62*1c60b9acSAndroid Build Coastguard Worker 		;
63*1c60b9acSAndroid Build Coastguard Worker 
64*1c60b9acSAndroid Build Coastguard Worker 	pthread_exit(NULL);
65*1c60b9acSAndroid Build Coastguard Worker 
66*1c60b9acSAndroid Build Coastguard Worker 	return NULL;
67*1c60b9acSAndroid Build Coastguard Worker }
68*1c60b9acSAndroid Build Coastguard Worker 
sigint_handler(int sig)69*1c60b9acSAndroid Build Coastguard Worker void sigint_handler(int sig)
70*1c60b9acSAndroid Build Coastguard Worker {
71*1c60b9acSAndroid Build Coastguard Worker 	interrupted = 1;
72*1c60b9acSAndroid Build Coastguard Worker 	lws_cancel_service(context);
73*1c60b9acSAndroid Build Coastguard Worker }
74*1c60b9acSAndroid Build Coastguard Worker 
main(int argc,const char ** argv)75*1c60b9acSAndroid Build Coastguard Worker int main(int argc, const char **argv)
76*1c60b9acSAndroid Build Coastguard Worker {
77*1c60b9acSAndroid Build Coastguard Worker 	pthread_t pthread_service[COUNT_THREADS];
78*1c60b9acSAndroid Build Coastguard Worker 	struct lws_context_creation_info info;
79*1c60b9acSAndroid Build Coastguard Worker 	void *retval;
80*1c60b9acSAndroid Build Coastguard Worker 	const char *p;
81*1c60b9acSAndroid Build Coastguard Worker 	int n = 0, logs = LLL_USER | LLL_ERR | LLL_WARN | LLL_NOTICE
82*1c60b9acSAndroid Build Coastguard Worker 			/* for LLL_ verbosity above NOTICE to be built into lws,
83*1c60b9acSAndroid Build Coastguard Worker 			 * lws must have been configured and built with
84*1c60b9acSAndroid Build Coastguard Worker 			 * -DCMAKE_BUILD_TYPE=DEBUG instead of =RELEASE */
85*1c60b9acSAndroid Build Coastguard Worker 			/* | LLL_INFO */ /* | LLL_PARSER */ /* | LLL_HEADER */
86*1c60b9acSAndroid Build Coastguard Worker 			/* | LLL_EXT */ /* | LLL_CLIENT */ /* | LLL_LATENCY */
87*1c60b9acSAndroid Build Coastguard Worker 			/* | LLL_DEBUG */;
88*1c60b9acSAndroid Build Coastguard Worker 
89*1c60b9acSAndroid Build Coastguard Worker 	if ((p = lws_cmdline_option(argc, argv, "-d")))
90*1c60b9acSAndroid Build Coastguard Worker 		logs = atoi(p);
91*1c60b9acSAndroid Build Coastguard Worker 
92*1c60b9acSAndroid Build Coastguard Worker 	lws_set_log_level(logs, NULL);
93*1c60b9acSAndroid Build Coastguard Worker 	lwsl_user("LWS minimal http server SMP | visit http://127.0.0.1:7681\n");
94*1c60b9acSAndroid Build Coastguard Worker 
95*1c60b9acSAndroid Build Coastguard Worker 	signal(SIGINT, sigint_handler);
96*1c60b9acSAndroid Build Coastguard Worker 
97*1c60b9acSAndroid Build Coastguard Worker 	memset(&info, 0, sizeof info); /* otherwise uninitialized garbage */
98*1c60b9acSAndroid Build Coastguard Worker 	info.port = 7681;
99*1c60b9acSAndroid Build Coastguard Worker 	info.mounts = &mount;
100*1c60b9acSAndroid Build Coastguard Worker 	info.options =
101*1c60b9acSAndroid Build Coastguard Worker 		LWS_SERVER_OPTION_HTTP_HEADERS_SECURITY_BEST_PRACTICES_ENFORCE;
102*1c60b9acSAndroid Build Coastguard Worker 	if ((p = lws_cmdline_option(argc, argv, "-t"))) {
103*1c60b9acSAndroid Build Coastguard Worker 		info.count_threads = (unsigned int)atoi(p);
104*1c60b9acSAndroid Build Coastguard Worker 		if (info.count_threads < 1 || info.count_threads > LWS_MAX_SMP)
105*1c60b9acSAndroid Build Coastguard Worker 			return 1;
106*1c60b9acSAndroid Build Coastguard Worker 	} else
107*1c60b9acSAndroid Build Coastguard Worker 		info.count_threads = COUNT_THREADS;
108*1c60b9acSAndroid Build Coastguard Worker 
109*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_TLS)
110*1c60b9acSAndroid Build Coastguard Worker 	if (lws_cmdline_option(argc, argv, "-s")) {
111*1c60b9acSAndroid Build Coastguard Worker 		info.options = LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT;
112*1c60b9acSAndroid Build Coastguard Worker 		info.ssl_cert_filepath = "localhost-100y.cert";
113*1c60b9acSAndroid Build Coastguard Worker 		info.ssl_private_key_filepath = "localhost-100y.key";
114*1c60b9acSAndroid Build Coastguard Worker 	}
115*1c60b9acSAndroid Build Coastguard Worker #endif
116*1c60b9acSAndroid Build Coastguard Worker 
117*1c60b9acSAndroid Build Coastguard Worker 	context = lws_create_context(&info);
118*1c60b9acSAndroid Build Coastguard Worker 	if (!context) {
119*1c60b9acSAndroid Build Coastguard Worker 		lwsl_err("lws init failed\n");
120*1c60b9acSAndroid Build Coastguard Worker 		return 1;
121*1c60b9acSAndroid Build Coastguard Worker 	}
122*1c60b9acSAndroid Build Coastguard Worker 
123*1c60b9acSAndroid Build Coastguard Worker 	lwsl_notice("  Service threads: %d\n", lws_get_count_threads(context));
124*1c60b9acSAndroid Build Coastguard Worker 
125*1c60b9acSAndroid Build Coastguard Worker 	/* start all the service threads */
126*1c60b9acSAndroid Build Coastguard Worker 
127*1c60b9acSAndroid Build Coastguard Worker 	for (n = 0; n < lws_get_count_threads(context); n++)
128*1c60b9acSAndroid Build Coastguard Worker 		if (pthread_create(&pthread_service[n], NULL, thread_service,
129*1c60b9acSAndroid Build Coastguard Worker 				   (void *)(lws_intptr_t)n))
130*1c60b9acSAndroid Build Coastguard Worker 			lwsl_err("Failed to start service thread\n");
131*1c60b9acSAndroid Build Coastguard Worker 
132*1c60b9acSAndroid Build Coastguard Worker 	/* wait for all the service threads to exit */
133*1c60b9acSAndroid Build Coastguard Worker 
134*1c60b9acSAndroid Build Coastguard Worker 	while ((--n) >= 0)
135*1c60b9acSAndroid Build Coastguard Worker 		pthread_join(pthread_service[n], &retval);
136*1c60b9acSAndroid Build Coastguard Worker 
137*1c60b9acSAndroid Build Coastguard Worker 	lws_context_destroy(context);
138*1c60b9acSAndroid Build Coastguard Worker 
139*1c60b9acSAndroid Build Coastguard Worker 	return 0;
140*1c60b9acSAndroid Build Coastguard Worker }
141