1/*
2 * lws-minimal-secure-streams-avs
3 *
4 * This file is made available under the Creative Commons CC0 1.0
5 * Universal Public Domain Dedication.
6 */
7
8#include <libwebsockets.h>
9#include <string.h>
10#include <signal.h>
11
12extern int
13avs_example_start(struct lws_context *context);
14
15int interrupted, bad = 1;
16static lws_state_notify_link_t nl;
17
18static const char *canned_root_token_payload =
19	"grant_type=refresh_token"
20	"&refresh_token=Atzr|IwEBIJedGXjDqsU_vMxykqOMg"
21	"SHfYe3CPcedueWEMWSDMaDnEmiW8RlR1Kns7Cb4B-TOSnqp7ifVsY4BMY2B8tpHfO39XP"
22	"zfu9HapGjTR458IyHX44FE71pWJkGZ79uVBpljP4sazJuk8XS3Oe_yLnm_DIO6fU1nU3Y"
23	"0flYmsOiOAQE_gRk_pdlmEtHnpMA-9rLw3mkY5L89Ty9kUygBsiFaYatouROhbsTn8-jW"
24	"k1zZLUDpT6ICtBXSnrCIg0pUbZevPFhTwdXd6eX-u4rq0W-XaDvPWFO7au-iPb4Zk5eZE"
25	"iX6sissYrtNmuEXc2uHu7MnQO1hHCaTdIO2CANVumf-PHSD8xseamyh04sLV5JgFzY45S"
26	"KvKMajiUZuLkMokOx86rjC2Hdkx5DO7G-dbG1ufBDG-N79pFMSs7Ck5pc283IdLoJkCQc"
27	"AGvTX8o8I29QqkcGou-9TKhOJmpX8As94T61ok0UqqEKPJ7RhfQHHYdCtsdwxgvfVr9qI"
28	"xL_hDCcTho8opCVX-6QhJHl6SQFlTw13"
29	"&client_id="
30		"amzn1.application-oa2-client.4823334c434b4190a2b5a42c07938a2d";
31
32static int
33app_system_state_nf(lws_state_manager_t *mgr, lws_state_notify_link_t *link,
34		    int current, int target)
35{
36	struct lws_context *context = lws_system_context_from_system_mgr(mgr);
37	lws_system_blob_t *ab = lws_system_get_blob(context,
38				LWS_SYSBLOB_TYPE_AUTH, 1 /* AUTH_IDX_ROOT */);
39	size_t size;
40
41	/*
42	 * For the things we care about, let's notice if we are trying to get
43	 * past them when we haven't solved them yet, and make the system
44	 * state wait while we trigger the dependent action.
45	 */
46	switch (target) {
47	case LWS_SYSTATE_REGISTERED:
48		size = lws_system_blob_get_size(ab);
49		if (size)
50			break;
51
52		/* let's register our canned root token so auth can use it */
53		lws_system_blob_direct_set(ab,
54				(const uint8_t *)canned_root_token_payload,
55				strlen(canned_root_token_payload));
56		break;
57	case LWS_SYSTATE_OPERATIONAL:
58		if (current == LWS_SYSTATE_OPERATIONAL)
59			avs_example_start(context);
60		break;
61	case LWS_SYSTATE_POLICY_INVALID:
62		/*
63		 * This is a NOP since we used direct set... but in a real
64		 * system this could easily change to be done on the heap, then
65		 * this would be important
66		 */
67		lws_system_blob_destroy(lws_system_get_blob(context,
68					LWS_SYSBLOB_TYPE_AUTH,
69					1 /* AUTH_IDX_ROOT */));
70		break;
71	}
72
73	return 0;
74}
75
76static void
77sigint_handler(int sig)
78{
79	interrupted = 1;
80}
81
82static lws_state_notify_link_t * const app_notifier_list[] = {
83	&nl, NULL
84};
85
86int main(int argc, const char **argv)
87{
88	struct lws_context_creation_info info;
89	struct lws_context *context;
90	int n = 0;
91
92	signal(SIGINT, sigint_handler);
93	memset(&info, 0, sizeof info); /* otherwise uninitialized garbage */
94	lws_cmdline_option_handle_builtin(argc, argv, &info);
95
96	lwsl_user("LWS secure streams - AVS test client [-d<verb>]\n");
97
98	info.options = LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT;
99	info.fd_limit_per_thread = 1 + 6 + 1;
100	info.protocols = lws_sspc_protocols;
101	info.port = CONTEXT_PORT_NO_LISTEN;
102
103	/* integrate us with lws system state management when context created */
104	nl.name = "app";
105	nl.notify_cb = app_system_state_nf;
106	info.register_notifier_list = app_notifier_list;
107
108	context = lws_create_context(&info);
109	if (!context) {
110		lwsl_err("lws init failed\n");
111		return 1;
112	}
113
114	/* the event loop */
115
116	while (n >= 0 && !interrupted)
117		n = lws_service(context, 0);
118
119	lws_context_destroy(context);
120	lwsl_user("Completed: %s\n", bad ? "failed" : "OK");
121
122	return bad;
123}
124