1*1c60b9acSAndroid Build Coastguard Worker /*
2*1c60b9acSAndroid Build Coastguard Worker  * lws-minimal-secure-streams-server
3*1c60b9acSAndroid Build Coastguard Worker  *
4*1c60b9acSAndroid Build Coastguard Worker  * Written in 2010-2020 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 
10*1c60b9acSAndroid Build Coastguard Worker #include <libwebsockets.h>
11*1c60b9acSAndroid Build Coastguard Worker #include <string.h>
12*1c60b9acSAndroid Build Coastguard Worker #include <signal.h>
13*1c60b9acSAndroid Build Coastguard Worker 
14*1c60b9acSAndroid Build Coastguard Worker extern const lws_ss_info_t ssi_client, ssi_server;
15*1c60b9acSAndroid Build Coastguard Worker 
16*1c60b9acSAndroid Build Coastguard Worker static struct lws_context *context;
17*1c60b9acSAndroid Build Coastguard Worker int interrupted, bad = 1;
18*1c60b9acSAndroid Build Coastguard Worker static const char * const default_ss_policy =
19*1c60b9acSAndroid Build Coastguard Worker 	"{"
20*1c60b9acSAndroid Build Coastguard Worker 	  "\"release\":"			"\"01234567\","
21*1c60b9acSAndroid Build Coastguard Worker 	  "\"product\":"			"\"myproduct\","
22*1c60b9acSAndroid Build Coastguard Worker 	  "\"schema-version\":"			"1,"
23*1c60b9acSAndroid Build Coastguard Worker 	  "\"s\": ["
24*1c60b9acSAndroid Build Coastguard Worker 
25*1c60b9acSAndroid Build Coastguard Worker 		/*
26*1c60b9acSAndroid Build Coastguard Worker 		 * This streamtype represents a raw server listening on :7681,
27*1c60b9acSAndroid Build Coastguard Worker 		 * without tls
28*1c60b9acSAndroid Build Coastguard Worker 		 */
29*1c60b9acSAndroid Build Coastguard Worker 
30*1c60b9acSAndroid Build Coastguard Worker 		"{\"myrawserver\": {"
31*1c60b9acSAndroid Build Coastguard Worker 			/* if given, "endpoint" is network if to bind to */
32*1c60b9acSAndroid Build Coastguard Worker 			"\"server\":"		"true,"
33*1c60b9acSAndroid Build Coastguard Worker 			"\"port\":"		"7681,"
34*1c60b9acSAndroid Build Coastguard Worker 			"\"protocol\":"		"\"raw\""
35*1c60b9acSAndroid Build Coastguard Worker 		"}}"
36*1c60b9acSAndroid Build Coastguard Worker 
37*1c60b9acSAndroid Build Coastguard Worker 	  "]"
38*1c60b9acSAndroid Build Coastguard Worker 	"}"
39*1c60b9acSAndroid Build Coastguard Worker ;
40*1c60b9acSAndroid Build Coastguard Worker 
41*1c60b9acSAndroid Build Coastguard Worker static int
smd_cb(void * opaque,lws_smd_class_t c,lws_usec_t ts,void * buf,size_t len)42*1c60b9acSAndroid Build Coastguard Worker smd_cb(void *opaque, lws_smd_class_t c, lws_usec_t ts, void *buf, size_t len)
43*1c60b9acSAndroid Build Coastguard Worker {
44*1c60b9acSAndroid Build Coastguard Worker 	if ((c & LWSSMDCL_SYSTEM_STATE) &&
45*1c60b9acSAndroid Build Coastguard Worker 	    !lws_json_simple_strcmp(buf, len, "\"state\":", "OPERATIONAL")) {
46*1c60b9acSAndroid Build Coastguard Worker 
47*1c60b9acSAndroid Build Coastguard Worker 		/* create the secure streams */
48*1c60b9acSAndroid Build Coastguard Worker 
49*1c60b9acSAndroid Build Coastguard Worker 		lwsl_notice("%s: creating server stream\n", __func__);
50*1c60b9acSAndroid Build Coastguard Worker 
51*1c60b9acSAndroid Build Coastguard Worker 		if (lws_ss_create(context, 0, &ssi_server, NULL, NULL,
52*1c60b9acSAndroid Build Coastguard Worker 				  NULL, NULL)) {
53*1c60b9acSAndroid Build Coastguard Worker 			lwsl_err("%s: failed to create secure stream\n",
54*1c60b9acSAndroid Build Coastguard Worker 				 __func__);
55*1c60b9acSAndroid Build Coastguard Worker 			return -1;
56*1c60b9acSAndroid Build Coastguard Worker 		}
57*1c60b9acSAndroid Build Coastguard Worker 	}
58*1c60b9acSAndroid Build Coastguard Worker 
59*1c60b9acSAndroid Build Coastguard Worker 	return 0;
60*1c60b9acSAndroid Build Coastguard Worker }
61*1c60b9acSAndroid Build Coastguard Worker 
62*1c60b9acSAndroid Build Coastguard Worker static void
sigint_handler(int sig)63*1c60b9acSAndroid Build Coastguard Worker sigint_handler(int sig)
64*1c60b9acSAndroid Build Coastguard Worker {
65*1c60b9acSAndroid Build Coastguard Worker 	interrupted = 1;
66*1c60b9acSAndroid Build Coastguard Worker }
67*1c60b9acSAndroid Build Coastguard Worker 
main(int argc,const char ** argv)68*1c60b9acSAndroid Build Coastguard Worker int main(int argc, const char **argv)
69*1c60b9acSAndroid Build Coastguard Worker {
70*1c60b9acSAndroid Build Coastguard Worker 	struct lws_context_creation_info info;
71*1c60b9acSAndroid Build Coastguard Worker 	int n = 0;
72*1c60b9acSAndroid Build Coastguard Worker 
73*1c60b9acSAndroid Build Coastguard Worker 	signal(SIGINT, sigint_handler);
74*1c60b9acSAndroid Build Coastguard Worker 
75*1c60b9acSAndroid Build Coastguard Worker 	memset(&info, 0, sizeof info); /* otherwise uninitialized garbage */
76*1c60b9acSAndroid Build Coastguard Worker 	lws_cmdline_option_handle_builtin(argc, argv, &info);
77*1c60b9acSAndroid Build Coastguard Worker 	lwsl_user("LWS Secure Streams Server Raw\n");
78*1c60b9acSAndroid Build Coastguard Worker 
79*1c60b9acSAndroid Build Coastguard Worker 	info.options			= LWS_SERVER_OPTION_EXPLICIT_VHOSTS |
80*1c60b9acSAndroid Build Coastguard Worker 					  LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT;
81*1c60b9acSAndroid Build Coastguard Worker 	info.fd_limit_per_thread	= 1 + 6 + 1;
82*1c60b9acSAndroid Build Coastguard Worker 	info.pss_policies_json		= default_ss_policy;
83*1c60b9acSAndroid Build Coastguard Worker 	info.port			= CONTEXT_PORT_NO_LISTEN;
84*1c60b9acSAndroid Build Coastguard Worker 	info.early_smd_cb		= smd_cb;
85*1c60b9acSAndroid Build Coastguard Worker 	info.early_smd_class_filter	= LWSSMDCL_SYSTEM_STATE;
86*1c60b9acSAndroid Build Coastguard Worker 
87*1c60b9acSAndroid Build Coastguard Worker 	context = lws_create_context(&info);
88*1c60b9acSAndroid Build Coastguard Worker 	if (!context) {
89*1c60b9acSAndroid Build Coastguard Worker 		lwsl_err("lws init failed\n");
90*1c60b9acSAndroid Build Coastguard Worker 		return 1;
91*1c60b9acSAndroid Build Coastguard Worker 	}
92*1c60b9acSAndroid Build Coastguard Worker 
93*1c60b9acSAndroid Build Coastguard Worker 	/* the event loop */
94*1c60b9acSAndroid Build Coastguard Worker 
95*1c60b9acSAndroid Build Coastguard Worker 	while (n >= 0 && !interrupted)
96*1c60b9acSAndroid Build Coastguard Worker 		n = lws_service(context, 0);
97*1c60b9acSAndroid Build Coastguard Worker 
98*1c60b9acSAndroid Build Coastguard Worker 	bad = 0;
99*1c60b9acSAndroid Build Coastguard Worker 
100*1c60b9acSAndroid Build Coastguard Worker 	lws_context_destroy(context);
101*1c60b9acSAndroid Build Coastguard Worker 	lwsl_user("Completed: %s\n", bad ? "failed" : "OK");
102*1c60b9acSAndroid Build Coastguard Worker 
103*1c60b9acSAndroid Build Coastguard Worker 	return bad;
104*1c60b9acSAndroid Build Coastguard Worker }
105