1*1c60b9acSAndroid Build Coastguard Worker /*
2*1c60b9acSAndroid Build Coastguard Worker * lws-minimal-secure-streams
3*1c60b9acSAndroid Build Coastguard Worker *
4*1c60b9acSAndroid Build Coastguard Worker * Written in 2010-2021 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 * This demonstrates a minimal http client using secure streams api.
11*1c60b9acSAndroid Build Coastguard Worker *
12*1c60b9acSAndroid Build Coastguard Worker * It visits https://warmcat.com/ and receives the html page there.
13*1c60b9acSAndroid Build Coastguard Worker *
14*1c60b9acSAndroid Build Coastguard Worker * This example is built two different ways from the same source... one includes
15*1c60b9acSAndroid Build Coastguard Worker * the policy everything needed to fulfil the stream directly. The other -client
16*1c60b9acSAndroid Build Coastguard Worker * variant has no policy itself and some other minor init changes, and connects
17*1c60b9acSAndroid Build Coastguard Worker * to the -proxy example to actually get the connection done.
18*1c60b9acSAndroid Build Coastguard Worker *
19*1c60b9acSAndroid Build Coastguard Worker * In the -client build case, the example does not even init the tls libraries
20*1c60b9acSAndroid Build Coastguard Worker * since the proxy part will take care of all that.
21*1c60b9acSAndroid Build Coastguard Worker */
22*1c60b9acSAndroid Build Coastguard Worker
23*1c60b9acSAndroid Build Coastguard Worker #include <libwebsockets.h>
24*1c60b9acSAndroid Build Coastguard Worker #include <string.h>
25*1c60b9acSAndroid Build Coastguard Worker #include <signal.h>
26*1c60b9acSAndroid Build Coastguard Worker
27*1c60b9acSAndroid Build Coastguard Worker // #define FORCE_OS_TRUST_STORE
28*1c60b9acSAndroid Build Coastguard Worker
29*1c60b9acSAndroid Build Coastguard Worker /*
30*1c60b9acSAndroid Build Coastguard Worker * uncomment to force network traffic through 127.0.0.1:1080
31*1c60b9acSAndroid Build Coastguard Worker *
32*1c60b9acSAndroid Build Coastguard Worker * On your local machine, you can run a SOCKS5 proxy like this
33*1c60b9acSAndroid Build Coastguard Worker *
34*1c60b9acSAndroid Build Coastguard Worker * $ ssh -N -D 0.0.0.0:1080 localhost -v
35*1c60b9acSAndroid Build Coastguard Worker *
36*1c60b9acSAndroid Build Coastguard Worker * If enabled, this also fetches a remote policy that also
37*1c60b9acSAndroid Build Coastguard Worker * specifies that all traffic should go through the remote
38*1c60b9acSAndroid Build Coastguard Worker * proxy.
39*1c60b9acSAndroid Build Coastguard Worker */
40*1c60b9acSAndroid Build Coastguard Worker // #define VIA_LOCALHOST_SOCKS
41*1c60b9acSAndroid Build Coastguard Worker
42*1c60b9acSAndroid Build Coastguard Worker static int interrupted, bad = 1, force_cpd_fail_portal,
43*1c60b9acSAndroid Build Coastguard Worker force_cpd_fail_no_internet, test_respmap, test_ots,
44*1c60b9acSAndroid Build Coastguard Worker budget = 1, predicted_good = 1, good, orig_budget;
45*1c60b9acSAndroid Build Coastguard Worker static unsigned int timeout_ms = 8000;
46*1c60b9acSAndroid Build Coastguard Worker static lws_state_notify_link_t nl;
47*1c60b9acSAndroid Build Coastguard Worker struct lws_context *context;
48*1c60b9acSAndroid Build Coastguard Worker static lws_sorted_usec_list_t sul_timeout; /* for each process to complete */
49*1c60b9acSAndroid Build Coastguard Worker
50*1c60b9acSAndroid Build Coastguard Worker /*
51*1c60b9acSAndroid Build Coastguard Worker * If the -proxy app is fulfilling our connection, then we don't need to have
52*1c60b9acSAndroid Build Coastguard Worker * the policy in the client.
53*1c60b9acSAndroid Build Coastguard Worker *
54*1c60b9acSAndroid Build Coastguard Worker * When we build with LWS_SS_USE_SSPC, the apis hook up to a proxy process over
55*1c60b9acSAndroid Build Coastguard Worker * a Unix Domain Socket. To test that, you need to separately run the
56*1c60b9acSAndroid Build Coastguard Worker * ./lws-minimal-secure-streams-proxy test app on the same machine.
57*1c60b9acSAndroid Build Coastguard Worker */
58*1c60b9acSAndroid Build Coastguard Worker
59*1c60b9acSAndroid Build Coastguard Worker #if !defined(LWS_SS_USE_SSPC)
60*1c60b9acSAndroid Build Coastguard Worker static const char * const default_ss_policy =
61*1c60b9acSAndroid Build Coastguard Worker "{"
62*1c60b9acSAndroid Build Coastguard Worker "\"release\":" "\"01234567\","
63*1c60b9acSAndroid Build Coastguard Worker "\"product\":" "\"myproduct\","
64*1c60b9acSAndroid Build Coastguard Worker "\"schema-version\":" "1,"
65*1c60b9acSAndroid Build Coastguard Worker #if defined(VIA_LOCALHOST_SOCKS)
66*1c60b9acSAndroid Build Coastguard Worker "\"via-socks5\":" "\"127.0.0.1:1080\","
67*1c60b9acSAndroid Build Coastguard Worker #endif
68*1c60b9acSAndroid Build Coastguard Worker
69*1c60b9acSAndroid Build Coastguard Worker "\"retry\": [" /* named backoff / retry strategies */
70*1c60b9acSAndroid Build Coastguard Worker "{\"default\": {"
71*1c60b9acSAndroid Build Coastguard Worker "\"backoff\": [" "1000,"
72*1c60b9acSAndroid Build Coastguard Worker "2000,"
73*1c60b9acSAndroid Build Coastguard Worker "3000,"
74*1c60b9acSAndroid Build Coastguard Worker "5000,"
75*1c60b9acSAndroid Build Coastguard Worker "10000"
76*1c60b9acSAndroid Build Coastguard Worker "],"
77*1c60b9acSAndroid Build Coastguard Worker "\"conceal\":" "5,"
78*1c60b9acSAndroid Build Coastguard Worker "\"jitterpc\":" "20,"
79*1c60b9acSAndroid Build Coastguard Worker "\"svalidping\":" "30,"
80*1c60b9acSAndroid Build Coastguard Worker "\"svalidhup\":" "35"
81*1c60b9acSAndroid Build Coastguard Worker "}}"
82*1c60b9acSAndroid Build Coastguard Worker "],"
83*1c60b9acSAndroid Build Coastguard Worker "\"certs\": [" /* named individual certificates in BASE64 DER */
84*1c60b9acSAndroid Build Coastguard Worker /*
85*1c60b9acSAndroid Build Coastguard Worker * Let's Encrypt certs for warmcat.com / libwebsockets.org
86*1c60b9acSAndroid Build Coastguard Worker *
87*1c60b9acSAndroid Build Coastguard Worker * We fetch the real policy from there using SS and switch to
88*1c60b9acSAndroid Build Coastguard Worker * using that.
89*1c60b9acSAndroid Build Coastguard Worker */
90*1c60b9acSAndroid Build Coastguard Worker #if !defined(FORCE_OS_TRUST_STORE)
91*1c60b9acSAndroid Build Coastguard Worker "{\"isrg_root_x1\": \""
92*1c60b9acSAndroid Build Coastguard Worker "MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw"
93*1c60b9acSAndroid Build Coastguard Worker "TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh"
94*1c60b9acSAndroid Build Coastguard Worker "cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4"
95*1c60b9acSAndroid Build Coastguard Worker "WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu"
96*1c60b9acSAndroid Build Coastguard Worker "ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY"
97*1c60b9acSAndroid Build Coastguard Worker "MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc"
98*1c60b9acSAndroid Build Coastguard Worker "h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+"
99*1c60b9acSAndroid Build Coastguard Worker "0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U"
100*1c60b9acSAndroid Build Coastguard Worker "A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW"
101*1c60b9acSAndroid Build Coastguard Worker "T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH"
102*1c60b9acSAndroid Build Coastguard Worker "B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC"
103*1c60b9acSAndroid Build Coastguard Worker "B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv"
104*1c60b9acSAndroid Build Coastguard Worker "KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn"
105*1c60b9acSAndroid Build Coastguard Worker "OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn"
106*1c60b9acSAndroid Build Coastguard Worker "jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw"
107*1c60b9acSAndroid Build Coastguard Worker "qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI"
108*1c60b9acSAndroid Build Coastguard Worker "rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV"
109*1c60b9acSAndroid Build Coastguard Worker "HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq"
110*1c60b9acSAndroid Build Coastguard Worker "hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL"
111*1c60b9acSAndroid Build Coastguard Worker "ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ"
112*1c60b9acSAndroid Build Coastguard Worker "3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK"
113*1c60b9acSAndroid Build Coastguard Worker "NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5"
114*1c60b9acSAndroid Build Coastguard Worker "ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur"
115*1c60b9acSAndroid Build Coastguard Worker "TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC"
116*1c60b9acSAndroid Build Coastguard Worker "jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc"
117*1c60b9acSAndroid Build Coastguard Worker "oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq"
118*1c60b9acSAndroid Build Coastguard Worker "4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA"
119*1c60b9acSAndroid Build Coastguard Worker "mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d"
120*1c60b9acSAndroid Build Coastguard Worker "emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc="
121*1c60b9acSAndroid Build Coastguard Worker "\"}"
122*1c60b9acSAndroid Build Coastguard Worker #endif
123*1c60b9acSAndroid Build Coastguard Worker "],"
124*1c60b9acSAndroid Build Coastguard Worker "\"trust_stores\": [" /* named cert chains */
125*1c60b9acSAndroid Build Coastguard Worker #if !defined(FORCE_OS_TRUST_STORE)
126*1c60b9acSAndroid Build Coastguard Worker "{"
127*1c60b9acSAndroid Build Coastguard Worker "\"name\": \"le_via_isrg\","
128*1c60b9acSAndroid Build Coastguard Worker "\"stack\": ["
129*1c60b9acSAndroid Build Coastguard Worker "\"isrg_root_x1\""
130*1c60b9acSAndroid Build Coastguard Worker "]"
131*1c60b9acSAndroid Build Coastguard Worker "}"
132*1c60b9acSAndroid Build Coastguard Worker #endif
133*1c60b9acSAndroid Build Coastguard Worker "],"
134*1c60b9acSAndroid Build Coastguard Worker "\"s\": ["
135*1c60b9acSAndroid Build Coastguard Worker #if !defined(LWS_WITH_SS_DIRECT_PROTOCOL_STR)
136*1c60b9acSAndroid Build Coastguard Worker /*
137*1c60b9acSAndroid Build Coastguard Worker * "fetch_policy" decides from where the real policy
138*1c60b9acSAndroid Build Coastguard Worker * will be fetched, if present. Otherwise the initial
139*1c60b9acSAndroid Build Coastguard Worker * policy is treated as the whole, hardcoded, policy.
140*1c60b9acSAndroid Build Coastguard Worker */
141*1c60b9acSAndroid Build Coastguard Worker "{\"fetch_policy\": {"
142*1c60b9acSAndroid Build Coastguard Worker "\"endpoint\":" "\"warmcat.com\","
143*1c60b9acSAndroid Build Coastguard Worker "\"port\":" "443,"
144*1c60b9acSAndroid Build Coastguard Worker "\"protocol\":" "\"h1\","
145*1c60b9acSAndroid Build Coastguard Worker "\"http_method\":" "\"GET\","
146*1c60b9acSAndroid Build Coastguard Worker #if defined(VIA_LOCALHOST_SOCKS)
147*1c60b9acSAndroid Build Coastguard Worker "\"http_url\":" "\"policy/minimal-proxy-socks.json\","
148*1c60b9acSAndroid Build Coastguard Worker #else
149*1c60b9acSAndroid Build Coastguard Worker "\"http_url\":" "\"policy/minimal-proxy-v4.2-v2.json\","
150*1c60b9acSAndroid Build Coastguard Worker #endif
151*1c60b9acSAndroid Build Coastguard Worker "\"tls\":" "true,"
152*1c60b9acSAndroid Build Coastguard Worker "\"opportunistic\":" "true,"
153*1c60b9acSAndroid Build Coastguard Worker #if !defined(FORCE_OS_TRUST_STORE)
154*1c60b9acSAndroid Build Coastguard Worker "\"tls_trust_store\":" "\"le_via_isrg\","
155*1c60b9acSAndroid Build Coastguard Worker #endif
156*1c60b9acSAndroid Build Coastguard Worker "\"retry\":" "\"default\""
157*1c60b9acSAndroid Build Coastguard Worker #else
158*1c60b9acSAndroid Build Coastguard Worker "{\"mintest\": {"
159*1c60b9acSAndroid Build Coastguard Worker "\"endpoint\": \"warmcat.com\","
160*1c60b9acSAndroid Build Coastguard Worker "\"port\": 443,"
161*1c60b9acSAndroid Build Coastguard Worker "\"protocol\": \"h1\","
162*1c60b9acSAndroid Build Coastguard Worker "\"http_method\": \"GET\","
163*1c60b9acSAndroid Build Coastguard Worker "\"http_url\": \"index.html?uptag=${uptag}\","
164*1c60b9acSAndroid Build Coastguard Worker "\"metadata\": [{"
165*1c60b9acSAndroid Build Coastguard Worker " \"uptag\": \"X-Upload-Tag:\""
166*1c60b9acSAndroid Build Coastguard Worker "}, {"
167*1c60b9acSAndroid Build Coastguard Worker " \"xctype\": \"X-Content-Type:\""
168*1c60b9acSAndroid Build Coastguard Worker "}],"
169*1c60b9acSAndroid Build Coastguard Worker "\"tls\": true,"
170*1c60b9acSAndroid Build Coastguard Worker "\"opportunistic\": true,"
171*1c60b9acSAndroid Build Coastguard Worker "\"retry\": \"default\","
172*1c60b9acSAndroid Build Coastguard Worker "\"timeout_ms\": 2000,"
173*1c60b9acSAndroid Build Coastguard Worker "\"direct_proto_str\": true,"
174*1c60b9acSAndroid Build Coastguard Worker "\"tls_trust_store\": \"le_via_isrg\""
175*1c60b9acSAndroid Build Coastguard Worker #endif
176*1c60b9acSAndroid Build Coastguard Worker "}},{"
177*1c60b9acSAndroid Build Coastguard Worker /*
178*1c60b9acSAndroid Build Coastguard Worker * "captive_portal_detect" describes
179*1c60b9acSAndroid Build Coastguard Worker * what to do in order to check if the path to
180*1c60b9acSAndroid Build Coastguard Worker * the Internet is being interrupted by a
181*1c60b9acSAndroid Build Coastguard Worker * captive portal. If there's a larger policy
182*1c60b9acSAndroid Build Coastguard Worker * fetched from elsewhere, it should also include
183*1c60b9acSAndroid Build Coastguard Worker * this since it needs to be done at least after
184*1c60b9acSAndroid Build Coastguard Worker * every DHCP acquisition
185*1c60b9acSAndroid Build Coastguard Worker */
186*1c60b9acSAndroid Build Coastguard Worker "\"captive_portal_detect\": {"
187*1c60b9acSAndroid Build Coastguard Worker "\"endpoint\": \"connectivitycheck.android.com\","
188*1c60b9acSAndroid Build Coastguard Worker "\"http_url\": \"generate_204\","
189*1c60b9acSAndroid Build Coastguard Worker "\"port\": 80,"
190*1c60b9acSAndroid Build Coastguard Worker "\"protocol\": \"h1\","
191*1c60b9acSAndroid Build Coastguard Worker "\"http_method\": \"GET\","
192*1c60b9acSAndroid Build Coastguard Worker "\"opportunistic\": true,"
193*1c60b9acSAndroid Build Coastguard Worker "\"http_expect\": 204,"
194*1c60b9acSAndroid Build Coastguard Worker "\"http_fail_redirect\": true"
195*1c60b9acSAndroid Build Coastguard Worker "}}"
196*1c60b9acSAndroid Build Coastguard Worker "]}"
197*1c60b9acSAndroid Build Coastguard Worker ;
198*1c60b9acSAndroid Build Coastguard Worker
199*1c60b9acSAndroid Build Coastguard Worker #endif
200*1c60b9acSAndroid Build Coastguard Worker
201*1c60b9acSAndroid Build Coastguard Worker typedef struct myss {
202*1c60b9acSAndroid Build Coastguard Worker struct lws_ss_handle *ss;
203*1c60b9acSAndroid Build Coastguard Worker void *opaque_data;
204*1c60b9acSAndroid Build Coastguard Worker /* ... application specific state ... */
205*1c60b9acSAndroid Build Coastguard Worker lws_sorted_usec_list_t sul;
206*1c60b9acSAndroid Build Coastguard Worker size_t amt;
207*1c60b9acSAndroid Build Coastguard Worker
208*1c60b9acSAndroid Build Coastguard Worker struct lws_genhash_ctx hash_ctx;
209*1c60b9acSAndroid Build Coastguard Worker } myss_t;
210*1c60b9acSAndroid Build Coastguard Worker
211*1c60b9acSAndroid Build Coastguard Worker static int
212*1c60b9acSAndroid Build Coastguard Worker create_ss(struct lws_context *cx);
213*1c60b9acSAndroid Build Coastguard Worker
214*1c60b9acSAndroid Build Coastguard Worker #if !defined(LWS_SS_USE_SSPC)
215*1c60b9acSAndroid Build Coastguard Worker
216*1c60b9acSAndroid Build Coastguard Worker static const char *canned_root_token_payload =
217*1c60b9acSAndroid Build Coastguard Worker "grant_type=refresh_token"
218*1c60b9acSAndroid Build Coastguard Worker "&refresh_token=Atzr|IwEBIJedGXjDqsU_vMxykqOMg"
219*1c60b9acSAndroid Build Coastguard Worker "SHfYe3CPcedueWEMWSDMaDnEmiW8RlR1Kns7Cb4B-TOSnqp7ifVsY4BMY2B8tpHfO39XP"
220*1c60b9acSAndroid Build Coastguard Worker "zfu9HapGjTR458IyHX44FE71pWJkGZ79uVBpljP4sazJuk8XS3Oe_yLnm_DIO6fU1nU3Y"
221*1c60b9acSAndroid Build Coastguard Worker "0flYmsOiOAQE_gRk_pdlmEtHnpMA-9rLw3mkY5L89Ty9kUygBsiFaYatouROhbsTn8-jW"
222*1c60b9acSAndroid Build Coastguard Worker "k1zZLUDpT6ICtBXSnrCIg0pUbZevPFhTwdXd6eX-u4rq0W-XaDvPWFO7au-iPb4Zk5eZE"
223*1c60b9acSAndroid Build Coastguard Worker "iX6sissYrtNmuEXc2uHu7MnQO1hHCaTdIO2CANVumf-PHSD8xseamyh04sLV5JgFzY45S"
224*1c60b9acSAndroid Build Coastguard Worker "KvKMajiUZuLkMokOx86rjC2Hdkx5DO7G-dbG1ufBDG-N79pFMSs7Ck5pc283IdLoJkCQc"
225*1c60b9acSAndroid Build Coastguard Worker "AGvTX8o8I29QqkcGou-9TKhOJmpX8As94T61ok0UqqEKPJ7RhfQHHYdCtsdwxgvfVr9qI"
226*1c60b9acSAndroid Build Coastguard Worker "xL_hDCcTho8opCVX-6QhJHl6SQFlTw13"
227*1c60b9acSAndroid Build Coastguard Worker "&client_id="
228*1c60b9acSAndroid Build Coastguard Worker "amzn1.application-oa2-client.4823334c434b4190a2b5a42c07938a2d";
229*1c60b9acSAndroid Build Coastguard Worker
230*1c60b9acSAndroid Build Coastguard Worker #endif
231*1c60b9acSAndroid Build Coastguard Worker
232*1c60b9acSAndroid Build Coastguard Worker static void
process_timeout(lws_sorted_usec_list_t * sul)233*1c60b9acSAndroid Build Coastguard Worker process_timeout(lws_sorted_usec_list_t *sul)
234*1c60b9acSAndroid Build Coastguard Worker {
235*1c60b9acSAndroid Build Coastguard Worker lwsl_err("%s: process timed out\n", __func__);
236*1c60b9acSAndroid Build Coastguard Worker
237*1c60b9acSAndroid Build Coastguard Worker exit(1);
238*1c60b9acSAndroid Build Coastguard Worker }
239*1c60b9acSAndroid Build Coastguard Worker
240*1c60b9acSAndroid Build Coastguard Worker /* secure streams payload interface */
241*1c60b9acSAndroid Build Coastguard Worker
242*1c60b9acSAndroid Build Coastguard Worker static lws_ss_state_return_t
myss_rx(void * userobj,const uint8_t * buf,size_t len,int flags)243*1c60b9acSAndroid Build Coastguard Worker myss_rx(void *userobj, const uint8_t *buf, size_t len, int flags)
244*1c60b9acSAndroid Build Coastguard Worker {
245*1c60b9acSAndroid Build Coastguard Worker
246*1c60b9acSAndroid Build Coastguard Worker if (flags & LWSSS_FLAG_PERF_JSON)
247*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_OK;
248*1c60b9acSAndroid Build Coastguard Worker
249*1c60b9acSAndroid Build Coastguard Worker #if !defined(LWS_WITH_SS_DIRECT_PROTOCOL_STR)
250*1c60b9acSAndroid Build Coastguard Worker myss_t *m = (myss_t *)userobj;
251*1c60b9acSAndroid Build Coastguard Worker const char *md_srv = "not set", *md_test = "not set";
252*1c60b9acSAndroid Build Coastguard Worker size_t md_srv_len = 7, md_test_len = 7;
253*1c60b9acSAndroid Build Coastguard Worker
254*1c60b9acSAndroid Build Coastguard Worker lws_ss_get_metadata(m->ss, "srv", (const void **)&md_srv, &md_srv_len);
255*1c60b9acSAndroid Build Coastguard Worker lws_ss_get_metadata(m->ss, "test", (const void **)&md_test, &md_test_len);
256*1c60b9acSAndroid Build Coastguard Worker lwsl_ss_user(m->ss, "len %d, flags: %d, srv: %.*s, test: %.*s",
257*1c60b9acSAndroid Build Coastguard Worker (int)len, flags, (int)md_srv_len, md_srv,
258*1c60b9acSAndroid Build Coastguard Worker (int)md_test_len, md_test);
259*1c60b9acSAndroid Build Coastguard Worker
260*1c60b9acSAndroid Build Coastguard Worker lwsl_hexdump_ss_info(m->ss, buf, len);
261*1c60b9acSAndroid Build Coastguard Worker #endif
262*1c60b9acSAndroid Build Coastguard Worker
263*1c60b9acSAndroid Build Coastguard Worker /*
264*1c60b9acSAndroid Build Coastguard Worker * If we received the whole message, for our example it means
265*1c60b9acSAndroid Build Coastguard Worker * we are done.
266*1c60b9acSAndroid Build Coastguard Worker */
267*1c60b9acSAndroid Build Coastguard Worker if (flags & LWSSS_FLAG_EOM) {
268*1c60b9acSAndroid Build Coastguard Worker bad = 0;
269*1c60b9acSAndroid Build Coastguard Worker }
270*1c60b9acSAndroid Build Coastguard Worker
271*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_OK;
272*1c60b9acSAndroid Build Coastguard Worker }
273*1c60b9acSAndroid Build Coastguard Worker
274*1c60b9acSAndroid Build Coastguard Worker static lws_ss_state_return_t
myss_tx(void * userobj,lws_ss_tx_ordinal_t ord,uint8_t * buf,size_t * len,int * flags)275*1c60b9acSAndroid Build Coastguard Worker myss_tx(void *userobj, lws_ss_tx_ordinal_t ord, uint8_t *buf, size_t *len,
276*1c60b9acSAndroid Build Coastguard Worker int *flags)
277*1c60b9acSAndroid Build Coastguard Worker {
278*1c60b9acSAndroid Build Coastguard Worker //myss_t *m = (myss_t *)userobj;
279*1c60b9acSAndroid Build Coastguard Worker
280*1c60b9acSAndroid Build Coastguard Worker /* in this example, we don't send stuff */
281*1c60b9acSAndroid Build Coastguard Worker
282*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_TX_DONT_SEND;
283*1c60b9acSAndroid Build Coastguard Worker }
284*1c60b9acSAndroid Build Coastguard Worker
285*1c60b9acSAndroid Build Coastguard Worker static lws_ss_state_return_t
myss_state(void * userobj,void * sh,lws_ss_constate_t state,lws_ss_tx_ordinal_t ack)286*1c60b9acSAndroid Build Coastguard Worker myss_state(void *userobj, void *sh, lws_ss_constate_t state,
287*1c60b9acSAndroid Build Coastguard Worker lws_ss_tx_ordinal_t ack)
288*1c60b9acSAndroid Build Coastguard Worker {
289*1c60b9acSAndroid Build Coastguard Worker myss_t *m = (myss_t *)userobj;
290*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_SS_DIRECT_PROTOCOL_STR)
291*1c60b9acSAndroid Build Coastguard Worker const char *md_test = "not set";
292*1c60b9acSAndroid Build Coastguard Worker size_t md_test_len = 7;
293*1c60b9acSAndroid Build Coastguard Worker int i;
294*1c60b9acSAndroid Build Coastguard Worker static const char * imd_test_keys[8] = {
295*1c60b9acSAndroid Build Coastguard Worker "server:",
296*1c60b9acSAndroid Build Coastguard Worker "content-security-policy:",
297*1c60b9acSAndroid Build Coastguard Worker "strict-transport-security:",
298*1c60b9acSAndroid Build Coastguard Worker "test-custom-header:",
299*1c60b9acSAndroid Build Coastguard Worker "x-xss-protection:",
300*1c60b9acSAndroid Build Coastguard Worker "x-content-type-options:",
301*1c60b9acSAndroid Build Coastguard Worker "x-frame-options:",
302*1c60b9acSAndroid Build Coastguard Worker "x-non-exist:",
303*1c60b9acSAndroid Build Coastguard Worker };
304*1c60b9acSAndroid Build Coastguard Worker #endif
305*1c60b9acSAndroid Build Coastguard Worker
306*1c60b9acSAndroid Build Coastguard Worker lwsl_ss_user(m->ss, "%s (%d), ord 0x%x",
307*1c60b9acSAndroid Build Coastguard Worker lws_ss_state_name((int)state), state, (unsigned int)ack);
308*1c60b9acSAndroid Build Coastguard Worker
309*1c60b9acSAndroid Build Coastguard Worker switch (state) {
310*1c60b9acSAndroid Build Coastguard Worker case LWSSSCS_CREATING:
311*1c60b9acSAndroid Build Coastguard Worker return lws_ss_client_connect(m->ss);
312*1c60b9acSAndroid Build Coastguard Worker
313*1c60b9acSAndroid Build Coastguard Worker case LWSSSCS_CONNECTING:
314*1c60b9acSAndroid Build Coastguard Worker lws_ss_start_timeout(m->ss, timeout_ms);
315*1c60b9acSAndroid Build Coastguard Worker
316*1c60b9acSAndroid Build Coastguard Worker if (lws_ss_set_metadata(m->ss, "uptag", "myuptag123", 10))
317*1c60b9acSAndroid Build Coastguard Worker /* can fail, eg due to OOM, retry later if so */
318*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_DISCONNECT_ME;
319*1c60b9acSAndroid Build Coastguard Worker #if !defined(LWS_WITH_SS_DIRECT_PROTOCOL_STR)
320*1c60b9acSAndroid Build Coastguard Worker if (lws_ss_set_metadata(m->ss, "ctype", "myctype", 7))
321*1c60b9acSAndroid Build Coastguard Worker /* can fail, eg due to OOM, retry later if so */
322*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_DISCONNECT_ME;
323*1c60b9acSAndroid Build Coastguard Worker #else
324*1c60b9acSAndroid Build Coastguard Worker if (lws_ss_set_metadata(m->ss, "X-Test-Type1:", "myctype1", 8))
325*1c60b9acSAndroid Build Coastguard Worker /* can fail, eg due to OOM, retry later if so */
326*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_DISCONNECT_ME;
327*1c60b9acSAndroid Build Coastguard Worker if (lws_ss_set_metadata(m->ss, "X-Test-Type2:", "myctype2", 8))
328*1c60b9acSAndroid Build Coastguard Worker /* can fail, eg due to OOM, retry later if so */
329*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_DISCONNECT_ME;
330*1c60b9acSAndroid Build Coastguard Worker if (lws_ss_set_metadata(m->ss, "Content-Type:", "myctype", 7))
331*1c60b9acSAndroid Build Coastguard Worker /* can fail, eg due to OOM, retry later if so */
332*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_DISCONNECT_ME;
333*1c60b9acSAndroid Build Coastguard Worker #endif
334*1c60b9acSAndroid Build Coastguard Worker break;
335*1c60b9acSAndroid Build Coastguard Worker
336*1c60b9acSAndroid Build Coastguard Worker case LWSSSCS_ALL_RETRIES_FAILED:
337*1c60b9acSAndroid Build Coastguard Worker /* if we're out of retries, we want to close the app and FAIL */
338*1c60b9acSAndroid Build Coastguard Worker interrupted = 1;
339*1c60b9acSAndroid Build Coastguard Worker bad = 2;
340*1c60b9acSAndroid Build Coastguard Worker break;
341*1c60b9acSAndroid Build Coastguard Worker
342*1c60b9acSAndroid Build Coastguard Worker case LWSSSCS_CONNECTED:
343*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_SS_DIRECT_PROTOCOL_STR)
344*1c60b9acSAndroid Build Coastguard Worker lwsl_cx_user(context, "get direct metadata");
345*1c60b9acSAndroid Build Coastguard Worker for (i = 0; i < 8; i++) {
346*1c60b9acSAndroid Build Coastguard Worker md_test = "not set";
347*1c60b9acSAndroid Build Coastguard Worker lws_ss_get_metadata(m->ss, imd_test_keys[i],
348*1c60b9acSAndroid Build Coastguard Worker (const void **)&md_test, &md_test_len);
349*1c60b9acSAndroid Build Coastguard Worker lwsl_ss_user(m->ss, " test key:[%s], got [%s]",
350*1c60b9acSAndroid Build Coastguard Worker imd_test_keys[i], md_test);
351*1c60b9acSAndroid Build Coastguard Worker }
352*1c60b9acSAndroid Build Coastguard Worker #endif
353*1c60b9acSAndroid Build Coastguard Worker break;
354*1c60b9acSAndroid Build Coastguard Worker
355*1c60b9acSAndroid Build Coastguard Worker case LWSSSCS_QOS_ACK_REMOTE: /* transaction assertively succeeded */
356*1c60b9acSAndroid Build Coastguard Worker lwsl_ss_notice(m->ss, "LWSSSCS_QOS_ACK_REMOTE");
357*1c60b9acSAndroid Build Coastguard Worker good++;
358*1c60b9acSAndroid Build Coastguard Worker break; /* disconnected will move us on */
359*1c60b9acSAndroid Build Coastguard Worker
360*1c60b9acSAndroid Build Coastguard Worker case LWSSSCS_QOS_NACK_REMOTE: /* transaction assertively failed */
361*1c60b9acSAndroid Build Coastguard Worker lwsl_ss_notice(m->ss, "LWSSSCS_QOS_NACK_REMOTE");
362*1c60b9acSAndroid Build Coastguard Worker break; /* disconnected will move us on */
363*1c60b9acSAndroid Build Coastguard Worker
364*1c60b9acSAndroid Build Coastguard Worker case LWSSSCS_DISCONNECTED: /* attempt is over */
365*1c60b9acSAndroid Build Coastguard Worker if (budget)
366*1c60b9acSAndroid Build Coastguard Worker create_ss(context);
367*1c60b9acSAndroid Build Coastguard Worker else
368*1c60b9acSAndroid Build Coastguard Worker interrupted = 1;
369*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_DESTROY_ME;
370*1c60b9acSAndroid Build Coastguard Worker
371*1c60b9acSAndroid Build Coastguard Worker case LWSSSCS_TIMEOUT:
372*1c60b9acSAndroid Build Coastguard Worker lwsl_ss_notice(m->ss, "LWSSSCS_TIMEOUT");
373*1c60b9acSAndroid Build Coastguard Worker bad = 3;
374*1c60b9acSAndroid Build Coastguard Worker if (budget)
375*1c60b9acSAndroid Build Coastguard Worker create_ss(context);
376*1c60b9acSAndroid Build Coastguard Worker else
377*1c60b9acSAndroid Build Coastguard Worker interrupted = 1;
378*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_DESTROY_ME;
379*1c60b9acSAndroid Build Coastguard Worker
380*1c60b9acSAndroid Build Coastguard Worker case LWSSSCS_USER_BASE:
381*1c60b9acSAndroid Build Coastguard Worker lwsl_ss_notice(m->ss, "LWSSSCS_USER_BASE");
382*1c60b9acSAndroid Build Coastguard Worker break;
383*1c60b9acSAndroid Build Coastguard Worker
384*1c60b9acSAndroid Build Coastguard Worker default:
385*1c60b9acSAndroid Build Coastguard Worker break;
386*1c60b9acSAndroid Build Coastguard Worker }
387*1c60b9acSAndroid Build Coastguard Worker
388*1c60b9acSAndroid Build Coastguard Worker return LWSSSSRET_OK;
389*1c60b9acSAndroid Build Coastguard Worker }
390*1c60b9acSAndroid Build Coastguard Worker
391*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_SECURE_STREAMS_BUFFER_DUMP)
392*1c60b9acSAndroid Build Coastguard Worker static void
myss_headers_dump(void * userobj,const uint8_t * buf,size_t len,int done)393*1c60b9acSAndroid Build Coastguard Worker myss_headers_dump(void *userobj, const uint8_t *buf, size_t len, int done)
394*1c60b9acSAndroid Build Coastguard Worker {
395*1c60b9acSAndroid Build Coastguard Worker lwsl_cx_user(context, "%lu done: %s", len, done ? "true" : "false");
396*1c60b9acSAndroid Build Coastguard Worker
397*1c60b9acSAndroid Build Coastguard Worker lwsl_hexdump_err(buf, len);
398*1c60b9acSAndroid Build Coastguard Worker }
399*1c60b9acSAndroid Build Coastguard Worker #endif
400*1c60b9acSAndroid Build Coastguard Worker
401*1c60b9acSAndroid Build Coastguard Worker static int
create_ss(struct lws_context * cx)402*1c60b9acSAndroid Build Coastguard Worker create_ss(struct lws_context *cx)
403*1c60b9acSAndroid Build Coastguard Worker {
404*1c60b9acSAndroid Build Coastguard Worker lws_ss_info_t ssi;
405*1c60b9acSAndroid Build Coastguard Worker
406*1c60b9acSAndroid Build Coastguard Worker budget--;
407*1c60b9acSAndroid Build Coastguard Worker lwsl_cx_notice(cx, "starting");
408*1c60b9acSAndroid Build Coastguard Worker
409*1c60b9acSAndroid Build Coastguard Worker /* We're making an outgoing secure stream ourselves */
410*1c60b9acSAndroid Build Coastguard Worker
411*1c60b9acSAndroid Build Coastguard Worker memset(&ssi, 0, sizeof(ssi));
412*1c60b9acSAndroid Build Coastguard Worker ssi.handle_offset = offsetof(myss_t, ss);
413*1c60b9acSAndroid Build Coastguard Worker ssi.opaque_user_data_offset = offsetof(myss_t, opaque_data);
414*1c60b9acSAndroid Build Coastguard Worker ssi.rx = myss_rx;
415*1c60b9acSAndroid Build Coastguard Worker ssi.tx = myss_tx;
416*1c60b9acSAndroid Build Coastguard Worker ssi.state = myss_state;
417*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_SECURE_STREAMS_BUFFER_DUMP)
418*1c60b9acSAndroid Build Coastguard Worker ssi.dump = myss_headers_dump;
419*1c60b9acSAndroid Build Coastguard Worker #endif
420*1c60b9acSAndroid Build Coastguard Worker ssi.user_alloc = sizeof(myss_t);
421*1c60b9acSAndroid Build Coastguard Worker ssi.streamtype = test_ots ? "mintest-ots" :
422*1c60b9acSAndroid Build Coastguard Worker (test_respmap ? "respmap" : "mintest");
423*1c60b9acSAndroid Build Coastguard Worker
424*1c60b9acSAndroid Build Coastguard Worker if (lws_ss_create(cx, 0, &ssi, NULL, NULL, NULL, NULL)) {
425*1c60b9acSAndroid Build Coastguard Worker lwsl_cx_err(context, "failed to create ss");
426*1c60b9acSAndroid Build Coastguard Worker return -1;
427*1c60b9acSAndroid Build Coastguard Worker }
428*1c60b9acSAndroid Build Coastguard Worker
429*1c60b9acSAndroid Build Coastguard Worker lwsl_cx_notice(cx, "started");
430*1c60b9acSAndroid Build Coastguard Worker
431*1c60b9acSAndroid Build Coastguard Worker return 0;
432*1c60b9acSAndroid Build Coastguard Worker }
433*1c60b9acSAndroid Build Coastguard Worker
434*1c60b9acSAndroid Build Coastguard Worker
435*1c60b9acSAndroid Build Coastguard Worker static int
app_system_state_nf(lws_state_manager_t * mgr,lws_state_notify_link_t * link,int current,int target)436*1c60b9acSAndroid Build Coastguard Worker app_system_state_nf(lws_state_manager_t *mgr, lws_state_notify_link_t *link,
437*1c60b9acSAndroid Build Coastguard Worker int current, int target)
438*1c60b9acSAndroid Build Coastguard Worker {
439*1c60b9acSAndroid Build Coastguard Worker struct lws_context *cx = lws_system_context_from_system_mgr(mgr);
440*1c60b9acSAndroid Build Coastguard Worker #if !defined(LWS_SS_USE_SSPC)
441*1c60b9acSAndroid Build Coastguard Worker
442*1c60b9acSAndroid Build Coastguard Worker lws_system_blob_t *ab = lws_system_get_blob(context,
443*1c60b9acSAndroid Build Coastguard Worker LWS_SYSBLOB_TYPE_AUTH, 1 /* AUTH_IDX_ROOT */);
444*1c60b9acSAndroid Build Coastguard Worker size_t size;
445*1c60b9acSAndroid Build Coastguard Worker #endif
446*1c60b9acSAndroid Build Coastguard Worker
447*1c60b9acSAndroid Build Coastguard Worker /*
448*1c60b9acSAndroid Build Coastguard Worker * For the things we care about, let's notice if we are trying to get
449*1c60b9acSAndroid Build Coastguard Worker * past them when we haven't solved them yet, and make the system
450*1c60b9acSAndroid Build Coastguard Worker * state wait while we trigger the dependent action.
451*1c60b9acSAndroid Build Coastguard Worker */
452*1c60b9acSAndroid Build Coastguard Worker switch (target) {
453*1c60b9acSAndroid Build Coastguard Worker
454*1c60b9acSAndroid Build Coastguard Worker #if !defined(LWS_SS_USE_SSPC)
455*1c60b9acSAndroid Build Coastguard Worker
456*1c60b9acSAndroid Build Coastguard Worker /*
457*1c60b9acSAndroid Build Coastguard Worker * The proxy takes responsibility for this stuff if we get things
458*1c60b9acSAndroid Build Coastguard Worker * done through that
459*1c60b9acSAndroid Build Coastguard Worker */
460*1c60b9acSAndroid Build Coastguard Worker
461*1c60b9acSAndroid Build Coastguard Worker case LWS_SYSTATE_INITIALIZED: /* overlay on the hardcoded policy */
462*1c60b9acSAndroid Build Coastguard Worker case LWS_SYSTATE_POLICY_VALID: /* overlay on the loaded policy */
463*1c60b9acSAndroid Build Coastguard Worker
464*1c60b9acSAndroid Build Coastguard Worker if (target != current)
465*1c60b9acSAndroid Build Coastguard Worker break;
466*1c60b9acSAndroid Build Coastguard Worker
467*1c60b9acSAndroid Build Coastguard Worker if (force_cpd_fail_portal)
468*1c60b9acSAndroid Build Coastguard Worker
469*1c60b9acSAndroid Build Coastguard Worker /* this makes it look like we're behind a captive portal
470*1c60b9acSAndroid Build Coastguard Worker * because the overriden address does a redirect */
471*1c60b9acSAndroid Build Coastguard Worker
472*1c60b9acSAndroid Build Coastguard Worker lws_ss_policy_overlay(context,
473*1c60b9acSAndroid Build Coastguard Worker "{\"s\": [{\"captive_portal_detect\": {"
474*1c60b9acSAndroid Build Coastguard Worker "\"endpoint\": \"google.com\","
475*1c60b9acSAndroid Build Coastguard Worker "\"http_url\": \"/\","
476*1c60b9acSAndroid Build Coastguard Worker "\"port\": 80"
477*1c60b9acSAndroid Build Coastguard Worker "}}]}");
478*1c60b9acSAndroid Build Coastguard Worker
479*1c60b9acSAndroid Build Coastguard Worker if (force_cpd_fail_no_internet)
480*1c60b9acSAndroid Build Coastguard Worker
481*1c60b9acSAndroid Build Coastguard Worker /* this looks like no internet, because the overridden
482*1c60b9acSAndroid Build Coastguard Worker * port doesn't have anything that will connect to us */
483*1c60b9acSAndroid Build Coastguard Worker
484*1c60b9acSAndroid Build Coastguard Worker lws_ss_policy_overlay(context,
485*1c60b9acSAndroid Build Coastguard Worker "{\"s\": [{\"captive_portal_detect\": {"
486*1c60b9acSAndroid Build Coastguard Worker "\"endpoint\": \"warmcat.com\","
487*1c60b9acSAndroid Build Coastguard Worker "\"http_url\": \"/\","
488*1c60b9acSAndroid Build Coastguard Worker "\"port\": 999"
489*1c60b9acSAndroid Build Coastguard Worker "}}]}");
490*1c60b9acSAndroid Build Coastguard Worker break;
491*1c60b9acSAndroid Build Coastguard Worker
492*1c60b9acSAndroid Build Coastguard Worker case LWS_SYSTATE_REGISTERED:
493*1c60b9acSAndroid Build Coastguard Worker size = lws_system_blob_get_size(ab);
494*1c60b9acSAndroid Build Coastguard Worker if (size)
495*1c60b9acSAndroid Build Coastguard Worker break;
496*1c60b9acSAndroid Build Coastguard Worker
497*1c60b9acSAndroid Build Coastguard Worker /* let's register our canned root token so auth can use it */
498*1c60b9acSAndroid Build Coastguard Worker lws_system_blob_direct_set(ab,
499*1c60b9acSAndroid Build Coastguard Worker (const uint8_t *)canned_root_token_payload,
500*1c60b9acSAndroid Build Coastguard Worker strlen(canned_root_token_payload));
501*1c60b9acSAndroid Build Coastguard Worker break;
502*1c60b9acSAndroid Build Coastguard Worker
503*1c60b9acSAndroid Build Coastguard Worker #endif
504*1c60b9acSAndroid Build Coastguard Worker
505*1c60b9acSAndroid Build Coastguard Worker case LWS_SYSTATE_OPERATIONAL:
506*1c60b9acSAndroid Build Coastguard Worker if (current == LWS_SYSTATE_OPERATIONAL) {
507*1c60b9acSAndroid Build Coastguard Worker create_ss(cx);
508*1c60b9acSAndroid Build Coastguard Worker }
509*1c60b9acSAndroid Build Coastguard Worker break;
510*1c60b9acSAndroid Build Coastguard Worker }
511*1c60b9acSAndroid Build Coastguard Worker
512*1c60b9acSAndroid Build Coastguard Worker return 0;
513*1c60b9acSAndroid Build Coastguard Worker }
514*1c60b9acSAndroid Build Coastguard Worker
515*1c60b9acSAndroid Build Coastguard Worker static lws_state_notify_link_t * const app_notifier_list[] = {
516*1c60b9acSAndroid Build Coastguard Worker &nl, NULL
517*1c60b9acSAndroid Build Coastguard Worker };
518*1c60b9acSAndroid Build Coastguard Worker
519*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_SYS_METRICS)
520*1c60b9acSAndroid Build Coastguard Worker
521*1c60b9acSAndroid Build Coastguard Worker static int
my_metric_report(lws_metric_pub_t * mp)522*1c60b9acSAndroid Build Coastguard Worker my_metric_report(lws_metric_pub_t *mp)
523*1c60b9acSAndroid Build Coastguard Worker {
524*1c60b9acSAndroid Build Coastguard Worker lws_metric_bucket_t *sub = mp->u.hist.head;
525*1c60b9acSAndroid Build Coastguard Worker char buf[192];
526*1c60b9acSAndroid Build Coastguard Worker
527*1c60b9acSAndroid Build Coastguard Worker do {
528*1c60b9acSAndroid Build Coastguard Worker if (lws_metrics_format(mp, &sub, buf, sizeof(buf)))
529*1c60b9acSAndroid Build Coastguard Worker lwsl_cx_user(context, "%s: %s\n", __func__, buf);
530*1c60b9acSAndroid Build Coastguard Worker } while ((mp->flags & LWSMTFL_REPORT_HIST) && sub);
531*1c60b9acSAndroid Build Coastguard Worker
532*1c60b9acSAndroid Build Coastguard Worker /* 0 = leave metric to accumulate, 1 = reset the metric */
533*1c60b9acSAndroid Build Coastguard Worker
534*1c60b9acSAndroid Build Coastguard Worker return 1;
535*1c60b9acSAndroid Build Coastguard Worker }
536*1c60b9acSAndroid Build Coastguard Worker
537*1c60b9acSAndroid Build Coastguard Worker static const lws_system_ops_t system_ops = {
538*1c60b9acSAndroid Build Coastguard Worker .metric_report = my_metric_report,
539*1c60b9acSAndroid Build Coastguard Worker };
540*1c60b9acSAndroid Build Coastguard Worker
541*1c60b9acSAndroid Build Coastguard Worker #endif
542*1c60b9acSAndroid Build Coastguard Worker
543*1c60b9acSAndroid Build Coastguard Worker static void
sigint_handler(int sig)544*1c60b9acSAndroid Build Coastguard Worker sigint_handler(int sig)
545*1c60b9acSAndroid Build Coastguard Worker {
546*1c60b9acSAndroid Build Coastguard Worker interrupted = 1;
547*1c60b9acSAndroid Build Coastguard Worker }
548*1c60b9acSAndroid Build Coastguard Worker
549*1c60b9acSAndroid Build Coastguard Worker static lws_log_cx_t my_log_cx = {
550*1c60b9acSAndroid Build Coastguard Worker .lll_flags = LLLF_LOG_CONTEXT_AWARE |
551*1c60b9acSAndroid Build Coastguard Worker LLL_ERR | LLL_WARN | LLL_NOTICE | LLL_USER,
552*1c60b9acSAndroid Build Coastguard Worker .refcount_cb = lws_log_use_cx_file,
553*1c60b9acSAndroid Build Coastguard Worker .u.emit_cx = lws_log_emit_cx_file,
554*1c60b9acSAndroid Build Coastguard Worker };
555*1c60b9acSAndroid Build Coastguard Worker
main(int argc,const char ** argv)556*1c60b9acSAndroid Build Coastguard Worker int main(int argc, const char **argv)
557*1c60b9acSAndroid Build Coastguard Worker {
558*1c60b9acSAndroid Build Coastguard Worker struct lws_context_creation_info info;
559*1c60b9acSAndroid Build Coastguard Worker int n = 0, expected = 0, concurrent = 1;
560*1c60b9acSAndroid Build Coastguard Worker char cxname[16], logpath[128];
561*1c60b9acSAndroid Build Coastguard Worker const char *p;
562*1c60b9acSAndroid Build Coastguard Worker
563*1c60b9acSAndroid Build Coastguard Worker signal(SIGINT, sigint_handler);
564*1c60b9acSAndroid Build Coastguard Worker
565*1c60b9acSAndroid Build Coastguard Worker memset(&info, 0, sizeof info);
566*1c60b9acSAndroid Build Coastguard Worker lws_cmdline_option_handle_builtin(argc, argv, &info);
567*1c60b9acSAndroid Build Coastguard Worker
568*1c60b9acSAndroid Build Coastguard Worker if ((p = lws_cmdline_option(argc, argv, "-c")))
569*1c60b9acSAndroid Build Coastguard Worker concurrent = atoi(p);
570*1c60b9acSAndroid Build Coastguard Worker
571*1c60b9acSAndroid Build Coastguard Worker if (concurrent < 0 || concurrent > 100)
572*1c60b9acSAndroid Build Coastguard Worker return 1;
573*1c60b9acSAndroid Build Coastguard Worker
574*1c60b9acSAndroid Build Coastguard Worker if ((p = lws_cmdline_option(argc, argv, "-d")))
575*1c60b9acSAndroid Build Coastguard Worker my_log_cx.lll_flags = (uint32_t)(LLLF_LOG_CONTEXT_AWARE | atoi(p));
576*1c60b9acSAndroid Build Coastguard Worker
577*1c60b9acSAndroid Build Coastguard Worker lws_strncpy(cxname, "ctx0", sizeof(cxname));
578*1c60b9acSAndroid Build Coastguard Worker
579*1c60b9acSAndroid Build Coastguard Worker for (n = 0; n < concurrent - 1; n++) {
580*1c60b9acSAndroid Build Coastguard Worker if (fork()) {
581*1c60b9acSAndroid Build Coastguard Worker #if defined(WIN32)
582*1c60b9acSAndroid Build Coastguard Worker Sleep(1);
583*1c60b9acSAndroid Build Coastguard Worker #else
584*1c60b9acSAndroid Build Coastguard Worker usleep(1000);
585*1c60b9acSAndroid Build Coastguard Worker #endif
586*1c60b9acSAndroid Build Coastguard Worker lws_snprintf(cxname, sizeof(cxname), "ctx%d", n + 1);
587*1c60b9acSAndroid Build Coastguard Worker break;
588*1c60b9acSAndroid Build Coastguard Worker }
589*1c60b9acSAndroid Build Coastguard Worker }
590*1c60b9acSAndroid Build Coastguard Worker
591*1c60b9acSAndroid Build Coastguard Worker /*
592*1c60b9acSAndroid Build Coastguard Worker * Arrange that each process's context logs to a different file
593*1c60b9acSAndroid Build Coastguard Worker */
594*1c60b9acSAndroid Build Coastguard Worker
595*1c60b9acSAndroid Build Coastguard Worker info.log_cx = &my_log_cx;
596*1c60b9acSAndroid Build Coastguard Worker info.vhost_name = cxname;
597*1c60b9acSAndroid Build Coastguard Worker lws_snprintf(logpath, sizeof(logpath), "/tmp/%s.log", cxname);
598*1c60b9acSAndroid Build Coastguard Worker my_log_cx.opaque = (void *)logpath;
599*1c60b9acSAndroid Build Coastguard Worker
600*1c60b9acSAndroid Build Coastguard Worker lwsl_user("LWS secure streams test client [-d<verb>]\n");
601*1c60b9acSAndroid Build Coastguard Worker
602*1c60b9acSAndroid Build Coastguard Worker /* these options are mutually exclusive if given */
603*1c60b9acSAndroid Build Coastguard Worker
604*1c60b9acSAndroid Build Coastguard Worker if (lws_cmdline_option(argc, argv, "--force-portal"))
605*1c60b9acSAndroid Build Coastguard Worker force_cpd_fail_portal = 1;
606*1c60b9acSAndroid Build Coastguard Worker
607*1c60b9acSAndroid Build Coastguard Worker if (lws_cmdline_option(argc, argv, "--force-no-internet"))
608*1c60b9acSAndroid Build Coastguard Worker force_cpd_fail_no_internet = 1;
609*1c60b9acSAndroid Build Coastguard Worker
610*1c60b9acSAndroid Build Coastguard Worker if (lws_cmdline_option(argc, argv, "--respmap"))
611*1c60b9acSAndroid Build Coastguard Worker test_respmap = 1;
612*1c60b9acSAndroid Build Coastguard Worker
613*1c60b9acSAndroid Build Coastguard Worker if (lws_cmdline_option(argc, argv, "--ots"))
614*1c60b9acSAndroid Build Coastguard Worker /*
615*1c60b9acSAndroid Build Coastguard Worker * Use a streamtype that relies on the OS trust store for
616*1c60b9acSAndroid Build Coastguard Worker * validation
617*1c60b9acSAndroid Build Coastguard Worker */
618*1c60b9acSAndroid Build Coastguard Worker test_ots = 1;
619*1c60b9acSAndroid Build Coastguard Worker
620*1c60b9acSAndroid Build Coastguard Worker if ((p = lws_cmdline_option(argc, argv, "--timeout_ms")))
621*1c60b9acSAndroid Build Coastguard Worker timeout_ms = (unsigned int)atoi(p);
622*1c60b9acSAndroid Build Coastguard Worker
623*1c60b9acSAndroid Build Coastguard Worker if ((p = lws_cmdline_option(argc, argv, "--budget")))
624*1c60b9acSAndroid Build Coastguard Worker budget = atoi(p);
625*1c60b9acSAndroid Build Coastguard Worker
626*1c60b9acSAndroid Build Coastguard Worker predicted_good = budget;
627*1c60b9acSAndroid Build Coastguard Worker orig_budget = budget;
628*1c60b9acSAndroid Build Coastguard Worker
629*1c60b9acSAndroid Build Coastguard Worker if ((p = lws_cmdline_option(argc, argv, "--pass-limit")))
630*1c60b9acSAndroid Build Coastguard Worker predicted_good = atoi(p);
631*1c60b9acSAndroid Build Coastguard Worker
632*1c60b9acSAndroid Build Coastguard Worker info.fd_limit_per_thread = 1 + 26 + 1;
633*1c60b9acSAndroid Build Coastguard Worker info.port = CONTEXT_PORT_NO_LISTEN;
634*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_SS_USE_SSPC)
635*1c60b9acSAndroid Build Coastguard Worker info.protocols = lws_sspc_protocols;
636*1c60b9acSAndroid Build Coastguard Worker {
637*1c60b9acSAndroid Build Coastguard Worker const char *p;
638*1c60b9acSAndroid Build Coastguard Worker
639*1c60b9acSAndroid Build Coastguard Worker /* connect to ssproxy via UDS by default, else via
640*1c60b9acSAndroid Build Coastguard Worker * tcp connection to this port */
641*1c60b9acSAndroid Build Coastguard Worker if ((p = lws_cmdline_option(argc, argv, "-p")))
642*1c60b9acSAndroid Build Coastguard Worker info.ss_proxy_port = (uint16_t)atoi(p);
643*1c60b9acSAndroid Build Coastguard Worker
644*1c60b9acSAndroid Build Coastguard Worker /* UDS "proxy.ss.lws" in abstract namespace, else this socket
645*1c60b9acSAndroid Build Coastguard Worker * path; when -p given this can specify the network interface
646*1c60b9acSAndroid Build Coastguard Worker * to bind to */
647*1c60b9acSAndroid Build Coastguard Worker if ((p = lws_cmdline_option(argc, argv, "-i")))
648*1c60b9acSAndroid Build Coastguard Worker info.ss_proxy_bind = p;
649*1c60b9acSAndroid Build Coastguard Worker
650*1c60b9acSAndroid Build Coastguard Worker /* if -p given, -a specifies the proxy address to connect to */
651*1c60b9acSAndroid Build Coastguard Worker if ((p = lws_cmdline_option(argc, argv, "-a")))
652*1c60b9acSAndroid Build Coastguard Worker info.ss_proxy_address = p;
653*1c60b9acSAndroid Build Coastguard Worker }
654*1c60b9acSAndroid Build Coastguard Worker #else
655*1c60b9acSAndroid Build Coastguard Worker info.pss_policies_json = default_ss_policy;
656*1c60b9acSAndroid Build Coastguard Worker info.options = LWS_SERVER_OPTION_EXPLICIT_VHOSTS |
657*1c60b9acSAndroid Build Coastguard Worker LWS_SERVER_OPTION_H2_JUST_FIX_WINDOW_UPDATE_OVERFLOW |
658*1c60b9acSAndroid Build Coastguard Worker LWS_SERVER_OPTION_DO_SSL_GLOBAL_INIT;
659*1c60b9acSAndroid Build Coastguard Worker #endif
660*1c60b9acSAndroid Build Coastguard Worker
661*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_MBEDTLS)
662*1c60b9acSAndroid Build Coastguard Worker
663*1c60b9acSAndroid Build Coastguard Worker /* uncomment to force mbedtls to load a system trust store like
664*1c60b9acSAndroid Build Coastguard Worker * openssl does
665*1c60b9acSAndroid Build Coastguard Worker *
666*1c60b9acSAndroid Build Coastguard Worker * info.mbedtls_client_preload_filepath =
667*1c60b9acSAndroid Build Coastguard Worker * "/etc/pki/ca-trust/extracted/pem/tls-ca-bundle.pem";
668*1c60b9acSAndroid Build Coastguard Worker */
669*1c60b9acSAndroid Build Coastguard Worker #endif
670*1c60b9acSAndroid Build Coastguard Worker
671*1c60b9acSAndroid Build Coastguard Worker /* integrate us with lws system state management when context created */
672*1c60b9acSAndroid Build Coastguard Worker
673*1c60b9acSAndroid Build Coastguard Worker nl.name = "app";
674*1c60b9acSAndroid Build Coastguard Worker nl.notify_cb = app_system_state_nf;
675*1c60b9acSAndroid Build Coastguard Worker info.register_notifier_list = app_notifier_list;
676*1c60b9acSAndroid Build Coastguard Worker
677*1c60b9acSAndroid Build Coastguard Worker #if defined(LWS_WITH_SYS_METRICS)
678*1c60b9acSAndroid Build Coastguard Worker info.system_ops = &system_ops;
679*1c60b9acSAndroid Build Coastguard Worker info.metrics_prefix = "ssmex";
680*1c60b9acSAndroid Build Coastguard Worker #endif
681*1c60b9acSAndroid Build Coastguard Worker
682*1c60b9acSAndroid Build Coastguard Worker /* create the context */
683*1c60b9acSAndroid Build Coastguard Worker
684*1c60b9acSAndroid Build Coastguard Worker context = lws_create_context(&info);
685*1c60b9acSAndroid Build Coastguard Worker if (!context) {
686*1c60b9acSAndroid Build Coastguard Worker lwsl_err("lws init failed\n");
687*1c60b9acSAndroid Build Coastguard Worker goto bail;
688*1c60b9acSAndroid Build Coastguard Worker }
689*1c60b9acSAndroid Build Coastguard Worker
690*1c60b9acSAndroid Build Coastguard Worker /* timeout for each forked process */
691*1c60b9acSAndroid Build Coastguard Worker
692*1c60b9acSAndroid Build Coastguard Worker lws_sul_schedule(context, 0, &sul_timeout, process_timeout,
693*1c60b9acSAndroid Build Coastguard Worker (lws_usec_t)((lws_usec_t)budget *
694*1c60b9acSAndroid Build Coastguard Worker (lws_usec_t)timeout_ms * LWS_US_PER_MS));
695*1c60b9acSAndroid Build Coastguard Worker
696*1c60b9acSAndroid Build Coastguard Worker #if !defined(LWS_SS_USE_SSPC)
697*1c60b9acSAndroid Build Coastguard Worker /*
698*1c60b9acSAndroid Build Coastguard Worker * If we're being a proxied client, the proxy does all this
699*1c60b9acSAndroid Build Coastguard Worker */
700*1c60b9acSAndroid Build Coastguard Worker
701*1c60b9acSAndroid Build Coastguard Worker /*
702*1c60b9acSAndroid Build Coastguard Worker * Set the related lws_system blobs
703*1c60b9acSAndroid Build Coastguard Worker *
704*1c60b9acSAndroid Build Coastguard Worker * ...direct_set() sets a pointer, so the thing pointed to has to have
705*1c60b9acSAndroid Build Coastguard Worker * a suitable lifetime, eg, something that already exists on the heap or
706*1c60b9acSAndroid Build Coastguard Worker * a const string in .rodata like this
707*1c60b9acSAndroid Build Coastguard Worker */
708*1c60b9acSAndroid Build Coastguard Worker
709*1c60b9acSAndroid Build Coastguard Worker lws_system_blob_direct_set(lws_system_get_blob(context,
710*1c60b9acSAndroid Build Coastguard Worker LWS_SYSBLOB_TYPE_DEVICE_SERIAL, 0),
711*1c60b9acSAndroid Build Coastguard Worker (const uint8_t *)"SN12345678", 10);
712*1c60b9acSAndroid Build Coastguard Worker lws_system_blob_direct_set(lws_system_get_blob(context,
713*1c60b9acSAndroid Build Coastguard Worker LWS_SYSBLOB_TYPE_DEVICE_FW_VERSION, 0),
714*1c60b9acSAndroid Build Coastguard Worker (const uint8_t *)"v0.01", 5);
715*1c60b9acSAndroid Build Coastguard Worker
716*1c60b9acSAndroid Build Coastguard Worker /*
717*1c60b9acSAndroid Build Coastguard Worker * ..._heap_append() appends to a buflist kind of arrangement on heap,
718*1c60b9acSAndroid Build Coastguard Worker * just one block is fine, otherwise it will concatenate the fragments
719*1c60b9acSAndroid Build Coastguard Worker * in the order they were appended (and take care of freeing them at
720*1c60b9acSAndroid Build Coastguard Worker * context destroy time). ..._heap_empty() is also available to remove
721*1c60b9acSAndroid Build Coastguard Worker * everything that was already allocated.
722*1c60b9acSAndroid Build Coastguard Worker *
723*1c60b9acSAndroid Build Coastguard Worker * Here we use _heap_append() just so it's tested as well as direct set.
724*1c60b9acSAndroid Build Coastguard Worker */
725*1c60b9acSAndroid Build Coastguard Worker
726*1c60b9acSAndroid Build Coastguard Worker lws_system_blob_heap_append(lws_system_get_blob(context,
727*1c60b9acSAndroid Build Coastguard Worker LWS_SYSBLOB_TYPE_DEVICE_TYPE, 0),
728*1c60b9acSAndroid Build Coastguard Worker (const uint8_t *)"spacerocket", 11);
729*1c60b9acSAndroid Build Coastguard Worker #endif
730*1c60b9acSAndroid Build Coastguard Worker
731*1c60b9acSAndroid Build Coastguard Worker /* the event loop */
732*1c60b9acSAndroid Build Coastguard Worker
733*1c60b9acSAndroid Build Coastguard Worker n = 0;
734*1c60b9acSAndroid Build Coastguard Worker while (n >= 0 && !interrupted)
735*1c60b9acSAndroid Build Coastguard Worker n = lws_service(context, 0);
736*1c60b9acSAndroid Build Coastguard Worker
737*1c60b9acSAndroid Build Coastguard Worker lws_sul_cancel(&sul_timeout);
738*1c60b9acSAndroid Build Coastguard Worker lws_context_destroy(context);
739*1c60b9acSAndroid Build Coastguard Worker
740*1c60b9acSAndroid Build Coastguard Worker bail:
741*1c60b9acSAndroid Build Coastguard Worker lwsl_user(" good: %d / %d budget, pass limit %d\n", good, orig_budget,
742*1c60b9acSAndroid Build Coastguard Worker predicted_good);
743*1c60b9acSAndroid Build Coastguard Worker if (good < predicted_good)
744*1c60b9acSAndroid Build Coastguard Worker bad = 1;
745*1c60b9acSAndroid Build Coastguard Worker
746*1c60b9acSAndroid Build Coastguard Worker if ((p = lws_cmdline_option(argc, argv, "--expected-exit")))
747*1c60b9acSAndroid Build Coastguard Worker expected = atoi(p);
748*1c60b9acSAndroid Build Coastguard Worker
749*1c60b9acSAndroid Build Coastguard Worker if (bad == expected) {
750*1c60b9acSAndroid Build Coastguard Worker lwsl_user("Completed: OK (seen expected %d)\n", expected);
751*1c60b9acSAndroid Build Coastguard Worker return 0;
752*1c60b9acSAndroid Build Coastguard Worker } else
753*1c60b9acSAndroid Build Coastguard Worker lwsl_err("Completed: failed: exit %d, expected %d\n", bad, expected);
754*1c60b9acSAndroid Build Coastguard Worker
755*1c60b9acSAndroid Build Coastguard Worker return 1;
756*1c60b9acSAndroid Build Coastguard Worker }
757