1 /*
2  *  SSL client with options
3  *
4  *  Copyright The Mbed TLS Contributors
5  *  SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
6  */
7 
8 #define MBEDTLS_ALLOW_PRIVATE_ACCESS
9 
10 #include "ssl_test_lib.h"
11 
12 #if defined(MBEDTLS_SSL_TEST_IMPOSSIBLE)
main(void)13 int main(void)
14 {
15     mbedtls_printf(MBEDTLS_SSL_TEST_IMPOSSIBLE);
16     mbedtls_exit(0);
17 }
18 #elif !defined(MBEDTLS_SSL_SRV_C)
main(void)19 int main(void)
20 {
21     mbedtls_printf("MBEDTLS_SSL_SRV_C not defined.\n");
22     mbedtls_exit(0);
23 }
24 #else /* !MBEDTLS_SSL_TEST_IMPOSSIBLE && MBEDTLS_SSL_SRV_C */
25 
26 #include <stdint.h>
27 
28 #if !defined(_MSC_VER)
29 #include <inttypes.h>
30 #endif
31 
32 #if !defined(_WIN32)
33 #include <signal.h>
34 #endif
35 
36 #if defined(MBEDTLS_SSL_CACHE_C)
37 #include "mbedtls/ssl_cache.h"
38 #endif
39 
40 #if defined(MBEDTLS_SSL_SESSION_TICKETS) && defined(MBEDTLS_SSL_TICKET_C)
41 #include "mbedtls/ssl_ticket.h"
42 #endif
43 
44 #if defined(MBEDTLS_SSL_COOKIE_C)
45 #include "mbedtls/ssl_cookie.h"
46 #endif
47 
48 #if defined(MBEDTLS_SSL_SERVER_NAME_INDICATION) && defined(MBEDTLS_FS_IO)
49 #define SNI_OPTION
50 #endif
51 
52 #if defined(_WIN32)
53 #include <windows.h>
54 #endif
55 
56 #if defined(MBEDTLS_USE_PSA_CRYPTO) || defined(MBEDTLS_SSL_PROTO_TLS1_3)
57 #include "test/psa_crypto_helpers.h"
58 #endif
59 
60 #include "mbedtls/pk.h"
61 #include "mbedtls/dhm.h"
62 
63 /* Size of memory to be allocated for the heap, when using the library's memory
64  * management and MBEDTLS_MEMORY_BUFFER_ALLOC_C is enabled. */
65 #define MEMORY_HEAP_SIZE        180000
66 
67 #define DFL_SERVER_ADDR         NULL
68 #define DFL_SERVER_PORT         "4433"
69 #define DFL_RESPONSE_SIZE       -1
70 #define DFL_DEBUG_LEVEL         0
71 #define DFL_NBIO                0
72 #define DFL_EVENT               0
73 #define DFL_READ_TIMEOUT        0
74 #define DFL_CA_FILE             ""
75 #define DFL_CA_PATH             ""
76 #define DFL_CRT_FILE            ""
77 #define DFL_KEY_FILE            ""
78 #define DFL_KEY_OPAQUE          0
79 #define DFL_KEY_PWD             ""
80 #define DFL_CRT_FILE2           ""
81 #define DFL_KEY_FILE2           ""
82 #define DFL_KEY_PWD2            ""
83 #define DFL_ASYNC_OPERATIONS    "-"
84 #define DFL_ASYNC_PRIVATE_DELAY1 (-1)
85 #define DFL_ASYNC_PRIVATE_DELAY2 (-1)
86 #define DFL_ASYNC_PRIVATE_ERROR  (0)
87 #define DFL_PSK                 ""
88 #define DFL_PSK_OPAQUE          0
89 #define DFL_PSK_LIST_OPAQUE     0
90 #define DFL_PSK_IDENTITY        "Client_identity"
91 #define DFL_ECJPAKE_PW          NULL
92 #define DFL_ECJPAKE_PW_OPAQUE   0
93 #define DFL_PSK_LIST            NULL
94 #define DFL_FORCE_CIPHER        0
95 #define DFL_TLS1_3_KEX_MODES    MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_ALL
96 #define DFL_RENEGOTIATION       MBEDTLS_SSL_RENEGOTIATION_DISABLED
97 #define DFL_ALLOW_LEGACY        -2
98 #define DFL_RENEGOTIATE         0
99 #define DFL_RENEGO_DELAY        -2
100 #define DFL_RENEGO_PERIOD       ((uint64_t) -1)
101 #define DFL_EXCHANGES           1
102 #define DFL_MIN_VERSION         -1
103 #define DFL_MAX_VERSION         -1
104 #define DFL_SHA1                -1
105 #define DFL_CID_ENABLED         0
106 #define DFL_CID_VALUE           ""
107 #define DFL_CID_ENABLED_RENEGO  -1
108 #define DFL_CID_VALUE_RENEGO    NULL
109 #define DFL_AUTH_MODE           -1
110 #define DFL_CERT_REQ_CA_LIST    MBEDTLS_SSL_CERT_REQ_CA_LIST_ENABLED
111 #define DFL_CERT_REQ_DN_HINT    0
112 #define DFL_MFL_CODE            MBEDTLS_SSL_MAX_FRAG_LEN_NONE
113 #define DFL_TRUNC_HMAC          -1
114 #define DFL_TICKETS             MBEDTLS_SSL_SESSION_TICKETS_ENABLED
115 #define DFL_DUMMY_TICKET        0
116 #define DFL_TICKET_ROTATE       0
117 #define DFL_TICKET_TIMEOUT      86400
118 #define DFL_TICKET_AEAD         MBEDTLS_CIPHER_AES_256_GCM
119 #define DFL_CACHE_MAX           -1
120 #define DFL_CACHE_TIMEOUT       -1
121 #define DFL_CACHE_REMOVE        0
122 #define DFL_SNI                 NULL
123 #define DFL_ALPN_STRING         NULL
124 #define DFL_GROUPS              NULL
125 #define DFL_EARLY_DATA          -1
126 #define DFL_MAX_EARLY_DATA_SIZE ((uint32_t) -1)
127 #define DFL_SIG_ALGS            NULL
128 #define DFL_DHM_FILE            NULL
129 #define DFL_TRANSPORT           MBEDTLS_SSL_TRANSPORT_STREAM
130 #define DFL_COOKIES             1
131 #define DFL_ANTI_REPLAY         -1
132 #define DFL_HS_TO_MIN           0
133 #define DFL_HS_TO_MAX           0
134 #define DFL_DTLS_MTU            -1
135 #define DFL_BADMAC_LIMIT        -1
136 #define DFL_DGRAM_PACKING        1
137 #define DFL_EXTENDED_MS         -1
138 #define DFL_ETM                 -1
139 #define DFL_SERIALIZE           0
140 #define DFL_CONTEXT_FILE        ""
141 #define DFL_EXTENDED_MS_ENFORCE -1
142 #define DFL_CA_CALLBACK         0
143 #define DFL_EAP_TLS             0
144 #define DFL_REPRODUCIBLE        0
145 #define DFL_NSS_KEYLOG          0
146 #define DFL_NSS_KEYLOG_FILE     NULL
147 #define DFL_QUERY_CONFIG_MODE   0
148 #define DFL_USE_SRTP            0
149 #define DFL_SRTP_FORCE_PROFILE  0
150 #define DFL_SRTP_SUPPORT_MKI    0
151 #define DFL_KEY_OPAQUE_ALG      "none"
152 
153 #define LONG_RESPONSE "<p>01-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n" \
154                       "02-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
155                       "03-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
156                       "04-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
157                       "05-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
158                       "06-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah\r\n"  \
159                       "07-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah-blah</p>\r\n"
160 
161 /* Uncomment LONG_RESPONSE at the end of HTTP_RESPONSE to test sending longer
162  * packets (for fragmentation purposes) */
163 #define HTTP_RESPONSE \
164     "HTTP/1.0 200 OK\r\nContent-Type: text/html\r\n\r\n" \
165     "<h2>Mbed TLS Test Server</h2>\r\n" \
166     "<p>Successful connection using: %s</p>\r\n" // LONG_RESPONSE
167 
168 /*
169  * Size of the basic I/O buffer. Able to hold our default response.
170  */
171 #define DFL_IO_BUF_LEN      200
172 
173 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
174 #if defined(MBEDTLS_FS_IO)
175 #define USAGE_IO \
176     "    ca_file=%%s          The single file containing the top-level CA(s) you fully trust\n" \
177     "                        default: \"\" (pre-loaded)\n" \
178     "                        use \"none\" to skip loading any top-level CAs.\n" \
179     "    ca_path=%%s          The path containing the top-level CA(s) you fully trust\n" \
180     "                        default: \"\" (pre-loaded) (overrides ca_file)\n" \
181     "                        use \"none\" to skip loading any top-level CAs.\n" \
182     "    crt_file=%%s         Your own cert and chain (in bottom to top order, top may be omitted)\n" \
183     "                        default: see note after key_file2\n" \
184     "    key_file=%%s         default: see note after key_file2\n" \
185     "    key_pwd=%%s          Password for key specified by key_file argument\n" \
186     "                        default: none\n" \
187     "    crt_file2=%%s        Your second cert and chain (in bottom to top order, top may be omitted)\n" \
188     "                        default: see note after key_file2\n" \
189     "    key_file2=%%s        default: see note below\n" \
190     "                        note: if neither crt_file/key_file nor crt_file2/key_file2 are used,\n" \
191     "                              preloaded certificate(s) and key(s) are used if available\n" \
192     "    key_pwd2=%%s         Password for key specified by key_file2 argument\n" \
193     "                        default: none\n" \
194     "    dhm_file=%%s        File containing Diffie-Hellman parameters\n" \
195     "                       default: preloaded parameters\n"
196 #else
197 #define USAGE_IO \
198     "\n"                                                    \
199     "    No file operations available (MBEDTLS_FS_IO not defined)\n" \
200     "\n"
201 #endif /* MBEDTLS_FS_IO */
202 #else
203 #define USAGE_IO ""
204 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED */
205 #if defined(MBEDTLS_USE_PSA_CRYPTO) && defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
206 #define USAGE_KEY_OPAQUE \
207     "    key_opaque=%%d       Handle your private keys as if they were opaque\n" \
208     "                        default: 0 (disabled)\n"
209 #else
210 #define USAGE_KEY_OPAQUE ""
211 #endif
212 
213 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
214 #define USAGE_SSL_ASYNC \
215     "    async_operations=%%c...   d=decrypt, s=sign (default: -=off)\n" \
216     "    async_private_delay1=%%d  Asynchronous delay for key_file or preloaded key\n" \
217     "    async_private_delay2=%%d  Asynchronous delay for key_file2 and sni\n" \
218     "                              default: -1 (not asynchronous)\n" \
219     "    async_private_error=%%d   Async callback error injection (default=0=none,\n" \
220     "                              1=start, 2=cancel, 3=resume, negative=first time only)"
221 #else
222 #define USAGE_SSL_ASYNC ""
223 #endif /* MBEDTLS_SSL_ASYNC_PRIVATE */
224 
225 #if defined(MBEDTLS_SSL_DTLS_CONNECTION_ID)
226 #define USAGE_CID \
227     "    cid=%%d             Disable (0) or enable (1) the use of the DTLS Connection ID extension.\n" \
228     "                       default: 0 (disabled)\n"     \
229     "    cid_renego=%%d      Disable (0) or enable (1) the use of the DTLS Connection ID extension during renegotiation.\n" \
230     "                       default: same as 'cid' parameter\n"     \
231     "    cid_val=%%s          The CID to use for incoming messages (in hex, without 0x).\n"  \
232     "                        default: \"\"\n" \
233     "    cid_val_renego=%%s   The CID to use for incoming messages (in hex, without 0x) after renegotiation.\n"  \
234     "                        default: same as 'cid_val' parameter\n"
235 #else /* MBEDTLS_SSL_DTLS_CONNECTION_ID */
236 #define USAGE_CID ""
237 #endif /* MBEDTLS_SSL_DTLS_CONNECTION_ID */
238 
239 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED)
240 #define USAGE_PSK_RAW                                               \
241     "    psk=%%s              default: \"\" (disabled)\n"     \
242     "                          The PSK values are in hex, without 0x.\n" \
243     "    psk_list=%%s         default: \"\"\n"                          \
244     "                          A list of (PSK identity, PSK value) pairs.\n" \
245     "                          The PSK values are in hex, without 0x.\n" \
246     "                          id1,psk1[,id2,psk2[,...]]\n"             \
247     "    psk_identity=%%s     default: \"Client_identity\"\n"
248 #if defined(MBEDTLS_USE_PSA_CRYPTO)
249 #define USAGE_PSK_SLOT                          \
250     "    psk_opaque=%%d       default: 0 (don't use opaque static PSK)\n"     \
251     "                          Enable this to store the PSK configured through command line\n" \
252     "                          parameter `psk` in a PSA-based key slot.\n" \
253     "                          Note: Currently only supported in conjunction with\n"                  \
254     "                          the use of min_version to force TLS 1.2 and force_ciphersuite \n"      \
255     "                          to force a particular PSK-only ciphersuite.\n"                         \
256     "                          Note: This is to test integration of PSA-based opaque PSKs with\n"     \
257     "                          Mbed TLS only. Production systems are likely to configure Mbed TLS\n"  \
258     "                          with prepopulated key slots instead of importing raw key material.\n" \
259     "    psk_list_opaque=%%d  default: 0 (don't use opaque dynamic PSKs)\n"     \
260     "                          Enable this to store the list of dynamically chosen PSKs configured\n" \
261     "                          through the command line parameter `psk_list` in PSA-based key slots.\n" \
262     "                          Note: Currently only supported in conjunction with\n" \
263     "                          the use of min_version to force TLS 1.2 and force_ciphersuite \n" \
264     "                          to force a particular PSK-only ciphersuite.\n" \
265     "                          Note: This is to test integration of PSA-based opaque PSKs with\n" \
266     "                          Mbed TLS only. Production systems are likely to configure Mbed TLS\n" \
267     "                          with prepopulated key slots instead of importing raw key material.\n"
268 #else
269 #define USAGE_PSK_SLOT ""
270 #endif /* MBEDTLS_USE_PSA_CRYPTO */
271 #define USAGE_PSK USAGE_PSK_RAW USAGE_PSK_SLOT
272 #else
273 #define USAGE_PSK ""
274 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED */
275 
276 #if defined(MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK)
277 #define USAGE_CA_CALLBACK                       \
278     "   ca_callback=%%d       default: 0 (disabled)\n"      \
279     "                         Enable this to use the trusted certificate callback function\n"
280 #else
281 #define USAGE_CA_CALLBACK ""
282 #endif /* MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK */
283 
284 #if defined(MBEDTLS_SSL_SESSION_TICKETS) && defined(MBEDTLS_SSL_TICKET_C)
285 #define USAGE_TICKETS                                       \
286     "    tickets=%%d          default: 1 (enabled)\n"       \
287     "    ticket_rotate=%%d    default: 0 (disabled)\n"      \
288     "    ticket_timeout=%%d   default: 86400 (one day)\n"   \
289     "    ticket_aead=%%s      default: \"AES-256-GCM\"\n"
290 #else /* MBEDTLS_SSL_SESSION_TICKETS && MBEDTLS_SSL_TICKET_C */
291 #define USAGE_TICKETS ""
292 #endif /* MBEDTLS_SSL_SESSION_TICKETS && MBEDTLS_SSL_TICKET_C */
293 
294 #define USAGE_EAP_TLS                                       \
295     "    eap_tls=%%d          default: 0 (disabled)\n"
296 #define USAGE_NSS_KEYLOG                                    \
297     "    nss_keylog=%%d          default: 0 (disabled)\n"   \
298     "                             This cannot be used with eap_tls=1\n"
299 #define USAGE_NSS_KEYLOG_FILE                               \
300     "    nss_keylog_file=%%s\n"
301 #if defined(MBEDTLS_SSL_DTLS_SRTP)
302 #define USAGE_SRTP \
303     "    use_srtp=%%d         default: 0 (disabled)\n" \
304     "    srtp_force_profile=%%d  default: 0 (all enabled)\n"   \
305     "                        available profiles:\n"       \
306     "                        1 - SRTP_AES128_CM_HMAC_SHA1_80\n"  \
307     "                        2 - SRTP_AES128_CM_HMAC_SHA1_32\n"  \
308     "                        3 - SRTP_NULL_HMAC_SHA1_80\n"       \
309     "                        4 - SRTP_NULL_HMAC_SHA1_32\n"       \
310     "    support_mki=%%d     default: 0 (not supported)\n"
311 #else /* MBEDTLS_SSL_DTLS_SRTP */
312 #define USAGE_SRTP ""
313 #endif
314 
315 #if defined(MBEDTLS_SSL_CACHE_C)
316 #define USAGE_CACHE                                             \
317     "    cache_max=%%d        default: cache default (50)\n"    \
318     "    cache_remove=%%d     default: 0 (don't remove)\n"
319 #if defined(MBEDTLS_HAVE_TIME)
320 #define USAGE_CACHE_TIME \
321     "    cache_timeout=%%d    default: cache default (1d)\n"
322 #else
323 #define USAGE_CACHE_TIME ""
324 #endif
325 #else
326 #define USAGE_CACHE ""
327 #define USAGE_CACHE_TIME ""
328 #endif /* MBEDTLS_SSL_CACHE_C */
329 
330 #if defined(SNI_OPTION)
331 #if defined(MBEDTLS_X509_CRL_PARSE_C)
332 #define SNI_CRL              ",crl"
333 #else
334 #define SNI_CRL              ""
335 #endif
336 
337 #define USAGE_SNI                                                           \
338     "    sni=%%s              name1,cert1,key1,ca1"SNI_CRL ",auth1[,...]\n"  \
339                                                            "                        default: disabled\n"
340 #else
341 #define USAGE_SNI ""
342 #endif /* SNI_OPTION */
343 
344 #if defined(MBEDTLS_SSL_MAX_FRAGMENT_LENGTH)
345 #define USAGE_MAX_FRAG_LEN                                      \
346     "    max_frag_len=%%d     default: 16384 (tls default)\n"   \
347     "                        options: 512, 1024, 2048, 4096\n"
348 #else
349 #define USAGE_MAX_FRAG_LEN ""
350 #endif /* MBEDTLS_SSL_MAX_FRAGMENT_LENGTH */
351 
352 #if defined(MBEDTLS_SSL_ALPN)
353 #define USAGE_ALPN \
354     "    alpn=%%s             default: \"\" (disabled)\n"   \
355     "                        example: spdy/1,http/1.1\n"
356 #else
357 #define USAGE_ALPN ""
358 #endif /* MBEDTLS_SSL_ALPN */
359 
360 #if defined(MBEDTLS_SSL_DTLS_HELLO_VERIFY)
361 #define USAGE_COOKIES \
362     "    cookies=0/1/-1      default: 1 (enabled)\n"        \
363     "                        0: disabled, -1: library default (broken)\n"
364 #else
365 #define USAGE_COOKIES ""
366 #endif
367 
368 #if defined(MBEDTLS_SSL_DTLS_ANTI_REPLAY)
369 #define USAGE_ANTI_REPLAY \
370     "    anti_replay=0/1     default: (library default: enabled)\n"
371 #else
372 #define USAGE_ANTI_REPLAY ""
373 #endif
374 
375 #define USAGE_BADMAC_LIMIT \
376     "    badmac_limit=%%d     default: (library default: disabled)\n"
377 
378 #if defined(MBEDTLS_SSL_PROTO_DTLS)
379 #define USAGE_DTLS \
380     "    dtls=%%d             default: 0 (TLS)\n"                           \
381     "    hs_timeout=%%d-%%d    default: (library default: 1000-60000)\n"    \
382     "                        range of DTLS handshake timeouts in millisecs\n" \
383     "    mtu=%%d              default: (library default: unlimited)\n"  \
384     "    dgram_packing=%%d    default: 1 (allowed)\n"                   \
385     "                        allow or forbid packing of multiple\n" \
386     "                        records within a single datgram.\n"
387 #else
388 #define USAGE_DTLS ""
389 #endif
390 
391 #if defined(MBEDTLS_SSL_EXTENDED_MASTER_SECRET)
392 #define USAGE_EMS \
393     "    extended_ms=0/1     default: (library default: on)\n"
394 #else
395 #define USAGE_EMS ""
396 #endif
397 
398 #if defined(MBEDTLS_SSL_ENCRYPT_THEN_MAC)
399 #define USAGE_ETM \
400     "    etm=0/1             default: (library default: on)\n"
401 #else
402 #define USAGE_ETM ""
403 #endif
404 
405 #define USAGE_REPRODUCIBLE \
406     "    reproducible=0/1     default: 0 (disabled)\n"
407 
408 #if defined(MBEDTLS_SSL_RENEGOTIATION)
409 #define USAGE_RENEGO \
410     "    renegotiation=%%d    default: 0 (disabled)\n"      \
411     "    renegotiate=%%d      default: 0 (disabled)\n"      \
412     "    renego_delay=%%d     default: -2 (library default)\n" \
413     "    renego_period=%%d    default: (2^64 - 1 for TLS, 2^48 - 1 for DTLS)\n"
414 #else
415 #define USAGE_RENEGO ""
416 #endif
417 
418 #if defined(MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED)
419 #if defined(MBEDTLS_USE_PSA_CRYPTO)
420 #define USAGE_ECJPAKE \
421     "    ecjpake_pw=%%s           default: none (disabled)\n"   \
422     "    ecjpake_pw_opaque=%%d    default: 0 (disabled)\n"
423 #else /* MBEDTLS_USE_PSA_CRYPTO */
424 #define USAGE_ECJPAKE \
425     "    ecjpake_pw=%%s           default: none (disabled)\n"
426 #endif /* MBEDTLS_USE_PSA_CRYPTO */
427 #else /* MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED */
428 #define USAGE_ECJPAKE ""
429 #endif /* MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED */
430 
431 #if defined(MBEDTLS_SSL_EARLY_DATA)
432 #define USAGE_EARLY_DATA \
433     "    early_data=%%d      default: library default\n" \
434     "                        options: 0 (disabled), 1 (enabled)\n" \
435     "    max_early_data_size=%%d default: library default\n" \
436     "                            options: max amount of early data\n"
437 #else
438 #define USAGE_EARLY_DATA ""
439 #endif /* MBEDTLS_SSL_EARLY_DATA */
440 
441 #if defined(MBEDTLS_PK_HAVE_ECC_KEYS) || \
442     (defined(MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_SOME_EPHEMERAL_ENABLED) && \
443     defined(PSA_WANT_ALG_FFDH))
444 #define USAGE_GROUPS \
445     "    groups=a,b,c,d      default: \"default\" (library default)\n"  \
446     "                        example: \"secp521r1,brainpoolP512r1\"\n"  \
447     "                        - use \"none\" for empty list\n"           \
448     "                        - see mbedtls_ecp_curve_list()\n"                \
449     "                          for acceptable EC group names\n"               \
450     "                        - the following ffdh groups are supported:\n"    \
451     "                          ffdhe2048, ffdhe3072, ffdhe4096, ffdhe6144,\n" \
452     "                          ffdhe8192\n"
453 #else
454 #define USAGE_GROUPS ""
455 #endif
456 
457 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
458 #define USAGE_SIG_ALGS \
459     "    sig_algs=a,b,c,d      default: \"default\" (library default)\n"  \
460     "                          example: \"ecdsa_secp256r1_sha256,ecdsa_secp384r1_sha384\"\n"
461 #else
462 #define USAGE_SIG_ALGS ""
463 #endif
464 
465 #if defined(MBEDTLS_SSL_CONTEXT_SERIALIZATION)
466 #define USAGE_SERIALIZATION \
467     "    serialize=%%d        default: 0 (do not serialize/deserialize)\n"     \
468     "                        options: 1 (serialize)\n"                         \
469     "                                 2 (serialize with re-initialization)\n"  \
470     "    context_file=%%s     The file path to write a serialized connection\n" \
471     "                        in the form of base64 code (serialize option\n"   \
472     "                        must be set)\n"                                   \
473     "                         default: \"\" (do nothing)\n"                    \
474     "                         option: a file path\n"
475 #else
476 #define USAGE_SERIALIZATION ""
477 #endif
478 
479 #define USAGE_KEY_OPAQUE_ALGS \
480     "    key_opaque_algs=%%s  Allowed opaque key 1 algorithms.\n"                      \
481     "                        comma-separated pair of values among the following:\n"    \
482     "                        rsa-sign-pkcs1, rsa-sign-pss, rsa-sign-pss-sha256,\n"     \
483     "                        rsa-sign-pss-sha384, rsa-sign-pss-sha512, rsa-decrypt,\n" \
484     "                        ecdsa-sign, ecdh, none (only acceptable for\n"            \
485     "                        the second value).\n"                                     \
486     "    key_opaque_algs2=%%s Allowed opaque key 2 algorithms.\n"                      \
487     "                        comma-separated pair of values among the following:\n"    \
488     "                        rsa-sign-pkcs1, rsa-sign-pss, rsa-sign-pss-sha256,\n"     \
489     "                        rsa-sign-pss-sha384, rsa-sign-pss-sha512, rsa-decrypt,\n" \
490     "                        ecdsa-sign, ecdh, none (only acceptable for\n"            \
491     "                        the second value).\n"
492 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
493 #define USAGE_TLS1_3_KEY_EXCHANGE_MODES \
494     "    tls13_kex_modes=%%s   default: all\n"     \
495     "                          options: psk, psk_ephemeral, psk_all, ephemeral,\n"  \
496     "                                   ephemeral_all, all, psk_or_ephemeral\n"
497 #else
498 #define USAGE_TLS1_3_KEY_EXCHANGE_MODES ""
499 #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
500 
501 
502 /* USAGE is arbitrarily split to stay under the portable string literal
503  * length limit: 4095 bytes in C99. */
504 #define USAGE1 \
505     "\n usage: ssl_server2 param=<>...\n"                   \
506     "\n acceptable parameters:\n"                           \
507     "    server_addr=%%s      default: (all interfaces)\n"  \
508     "    server_port=%%d      default: 4433\n"              \
509     "    debug_level=%%d      default: 0 (disabled)\n"      \
510     "    build_version=%%d    default: none (disabled)\n"                     \
511     "                        option: 1 (print build version only and stop)\n" \
512     "    buffer_size=%%d      default: 200 \n" \
513     "                         (minimum: 1)\n" \
514     "    response_size=%%d    default: about 152 (basic response)\n" \
515     "                          (minimum: 0, max: 16384)\n" \
516     "                          increases buffer_size if bigger\n" \
517     "    nbio=%%d             default: 0 (blocking I/O)\n"  \
518     "                        options: 1 (non-blocking), 2 (added delays)\n" \
519     "    event=%%d            default: 0 (loop)\n"                            \
520     "                        options: 1 (level-triggered, implies nbio=1),\n" \
521     "    read_timeout=%%d     default: 0 ms (no timeout)\n"    \
522     "\n"                                                    \
523     USAGE_DTLS                                              \
524     USAGE_SRTP                                              \
525     USAGE_COOKIES                                           \
526     USAGE_ANTI_REPLAY                                       \
527     USAGE_BADMAC_LIMIT                                      \
528     "\n"
529 #define USAGE2 \
530     "    auth_mode=%%s        default: (library default: none)\n"      \
531     "                        options: none, optional, required\n" \
532     "    cert_req_ca_list=%%d default: 1 (send ca list)\n"  \
533     "                        options: 1 (send ca list), 0 (don't send)\n" \
534     "                                 2 (send conf dn hint), 3 (send hs dn hint)\n" \
535     USAGE_IO                                                \
536     USAGE_KEY_OPAQUE                                        \
537     "\n"                                                    \
538     USAGE_PSK                                               \
539     USAGE_CA_CALLBACK                                       \
540     USAGE_ECJPAKE                                           \
541     "\n"
542 #define USAGE3 \
543     "    allow_legacy=%%d     default: (library default: no)\n"      \
544     USAGE_RENEGO                                            \
545     "    exchanges=%%d        default: 1\n"                 \
546     "\n"                                                    \
547     USAGE_TICKETS                                           \
548     USAGE_EAP_TLS                                           \
549     USAGE_REPRODUCIBLE                                      \
550     USAGE_NSS_KEYLOG                                        \
551     USAGE_NSS_KEYLOG_FILE                                   \
552     USAGE_CACHE                                             \
553     USAGE_CACHE_TIME                                        \
554     USAGE_MAX_FRAG_LEN                                      \
555     USAGE_ALPN                                              \
556     USAGE_EMS                                               \
557     USAGE_ETM                                               \
558     USAGE_GROUPS                                            \
559     USAGE_SIG_ALGS                                          \
560     USAGE_KEY_OPAQUE_ALGS                                   \
561     USAGE_EARLY_DATA                                        \
562     "\n"
563 
564 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
565 #define TLS1_3_VERSION_OPTIONS  ", tls13"
566 #else /* MBEDTLS_SSL_PROTO_TLS1_3 */
567 #define TLS1_3_VERSION_OPTIONS  ""
568 #endif /* !MBEDTLS_SSL_PROTO_TLS1_3 */
569 
570 #define USAGE4 \
571     USAGE_SSL_ASYNC                                         \
572     USAGE_SNI                                               \
573     "    allow_sha1=%%d       default: 0\n"                                   \
574     "    min_version=%%s      default: (library default: tls12)\n"            \
575     "    max_version=%%s      default: (library default: tls12)\n"            \
576     "    force_version=%%s    default: \"\" (none)\n"                         \
577     "                         options: tls12, dtls12" TLS1_3_VERSION_OPTIONS  \
578     "\n\n"                                                                    \
579     "    force_ciphersuite=<name>    default: all enabled\n"                  \
580     USAGE_TLS1_3_KEY_EXCHANGE_MODES                                           \
581     "    query_config=<name>         return 0 if the specified\n"             \
582     "                                configuration macro is defined and 1\n"  \
583     "                                otherwise. The expansion of the macro\n" \
584     "                                is printed if it is defined\n"           \
585     USAGE_SERIALIZATION                                                       \
586     "\n"
587 
588 #define PUT_UINT64_BE(out_be, in_le, i)                                   \
589     {                                                                       \
590         (out_be)[(i) + 0] = (unsigned char) (((in_le) >> 56) & 0xFF);    \
591         (out_be)[(i) + 1] = (unsigned char) (((in_le) >> 48) & 0xFF);    \
592         (out_be)[(i) + 2] = (unsigned char) (((in_le) >> 40) & 0xFF);    \
593         (out_be)[(i) + 3] = (unsigned char) (((in_le) >> 32) & 0xFF);    \
594         (out_be)[(i) + 4] = (unsigned char) (((in_le) >> 24) & 0xFF);    \
595         (out_be)[(i) + 5] = (unsigned char) (((in_le) >> 16) & 0xFF);    \
596         (out_be)[(i) + 6] = (unsigned char) (((in_le) >> 8) & 0xFF);    \
597         (out_be)[(i) + 7] = (unsigned char) (((in_le) >> 0) & 0xFF);    \
598     }
599 
600 /* This is global so it can be easily accessed by callback functions */
601 rng_context_t rng;
602 
603 /*
604  * global options
605  */
606 struct options {
607     const char *server_addr;    /* address on which the ssl service runs    */
608     const char *server_port;    /* port on which the ssl service runs       */
609     int debug_level;            /* level of debugging                       */
610     int nbio;                   /* should I/O be blocking?                  */
611     int event;                  /* loop or event-driven IO? level or edge triggered? */
612     uint32_t read_timeout;      /* timeout on mbedtls_ssl_read() in milliseconds    */
613     int response_size;          /* pad response with header to requested size */
614     uint16_t buffer_size;       /* IO buffer size */
615     const char *ca_file;        /* the file with the CA certificate(s)      */
616     const char *ca_path;        /* the path with the CA certificate(s) reside */
617     const char *crt_file;       /* the file with the server certificate     */
618     const char *key_file;       /* the file with the server key             */
619     int key_opaque;             /* handle private key as if it were opaque  */
620     const char *key_pwd;        /* the password for the server key          */
621     const char *crt_file2;      /* the file with the 2nd server certificate */
622     const char *key_file2;      /* the file with the 2nd server key         */
623     const char *key_pwd2;       /* the password for the 2nd server key      */
624     const char *async_operations; /* supported SSL asynchronous operations  */
625     int async_private_delay1;   /* number of times f_async_resume needs to be called for key 1, or -1 for no async */
626     int async_private_delay2;   /* number of times f_async_resume needs to be called for key 2, or -1 for no async */
627     int async_private_error;    /* inject error in async private callback */
628 #if defined(MBEDTLS_USE_PSA_CRYPTO)
629     int psk_opaque;
630     int psk_list_opaque;
631 #endif
632 #if defined(MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK)
633     int ca_callback;            /* Use callback for trusted certificate list */
634 #endif
635     const char *psk;            /* the pre-shared key                       */
636     const char *psk_identity;   /* the pre-shared key identity              */
637     char *psk_list;             /* list of PSK id/key pairs for callback    */
638     const char *ecjpake_pw;     /* the EC J-PAKE password                   */
639 #if defined(MBEDTLS_USE_PSA_CRYPTO)
640     int ecjpake_pw_opaque;      /* set to 1 to use the opaque method for setting the password */
641 #endif
642     int force_ciphersuite[2];   /* protocol/ciphersuite to use, or all      */
643 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
644     int tls13_kex_modes;        /* supported TLS 1.3 key exchange modes     */
645 #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
646     int renegotiation;          /* enable / disable renegotiation           */
647     int allow_legacy;           /* allow legacy renegotiation               */
648     int renegotiate;            /* attempt renegotiation?                   */
649     int renego_delay;           /* delay before enforcing renegotiation     */
650     uint64_t renego_period;     /* period for automatic renegotiation       */
651     int exchanges;              /* number of data exchanges                 */
652     int min_version;            /* minimum protocol version accepted        */
653     int max_version;            /* maximum protocol version accepted        */
654     int allow_sha1;             /* flag for SHA-1 support                   */
655     int auth_mode;              /* verify mode for connection               */
656     int cert_req_ca_list;       /* should we send the CA list?              */
657     int cert_req_dn_hint;       /* mode to set DN hints for CA list to send */
658     unsigned char mfl_code;     /* code for maximum fragment length         */
659     int trunc_hmac;             /* accept truncated hmac?                   */
660     int tickets;                /* enable / disable session tickets         */
661     int dummy_ticket;           /* enable / disable dummy ticket generator  */
662     int ticket_rotate;          /* session ticket rotate (code coverage)    */
663     int ticket_timeout;         /* session ticket lifetime                  */
664     int ticket_aead;            /* session ticket protection                */
665     int cache_max;              /* max number of session cache entries      */
666 #if defined(MBEDTLS_HAVE_TIME)
667     int cache_timeout;          /* expiration delay of session cache entries*/
668 #endif
669     int cache_remove;           /* enable / disable cache entry removal     */
670     char *sni;                  /* string describing sni information        */
671     const char *groups;         /* list of supported groups                 */
672     const char *sig_algs;       /* supported TLS 1.3 signature algorithms   */
673     const char *alpn_string;    /* ALPN supported protocols                 */
674     const char *dhm_file;       /* the file with the DH parameters          */
675     int extended_ms;            /* allow negotiation of extended MS?        */
676     int etm;                    /* allow negotiation of encrypt-then-MAC?   */
677     int transport;              /* TLS or DTLS?                             */
678     int cookies;                /* Use cookies for DTLS? -1 to break them   */
679     int anti_replay;            /* Use anti-replay for DTLS? -1 for default */
680     uint32_t hs_to_min;         /* Initial value of DTLS handshake timer    */
681     uint32_t hs_to_max;         /* Max value of DTLS handshake timer        */
682     int dtls_mtu;               /* UDP Maximum transport unit for DTLS       */
683     int dgram_packing;          /* allow/forbid datagram packing            */
684     int badmac_limit;           /* Limit of records with bad MAC            */
685     int eap_tls;                /* derive EAP-TLS keying material?          */
686     int nss_keylog;             /* export NSS key log material              */
687     const char *nss_keylog_file; /* NSS key log file                        */
688     int cid_enabled;            /* whether to use the CID extension or not  */
689     int cid_enabled_renego;     /* whether to use the CID extension or not
690                                  * during renegotiation                     */
691     const char *cid_val;        /* the CID to use for incoming messages     */
692     int serialize;              /* serialize/deserialize connection         */
693     const char *context_file;   /* the file to write a serialized connection
694                                  * in the form of base64 code (serialize
695                                  * option must be set)                      */
696     const char *cid_val_renego; /* the CID to use for incoming messages
697                                  * after renegotiation                      */
698     int reproducible;           /* make communication reproducible          */
699 #if defined(MBEDTLS_SSL_EARLY_DATA)
700     int early_data;               /* early data enablement flag             */
701     uint32_t max_early_data_size; /* max amount of early data               */
702 #endif
703     int query_config_mode;      /* whether to read config                   */
704     int use_srtp;               /* Support SRTP                             */
705     int force_srtp_profile;     /* SRTP protection profile to use or all    */
706     int support_mki;            /* The dtls mki mki support                 */
707     const char *key1_opaque_alg1; /* Allowed opaque key 1 alg 1            */
708     const char *key1_opaque_alg2; /* Allowed opaque key 1 alg 2            */
709     const char *key2_opaque_alg1; /* Allowed opaque key 2 alg 1            */
710     const char *key2_opaque_alg2; /* Allowed opaque key 2 alg 2            */
711 } opt;
712 
713 #include "ssl_test_common_source.c"
714 
715 /*
716  * Return authmode from string, or -1 on error
717  */
get_auth_mode(const char * s)718 static int get_auth_mode(const char *s)
719 {
720     if (strcmp(s, "none") == 0) {
721         return MBEDTLS_SSL_VERIFY_NONE;
722     }
723     if (strcmp(s, "optional") == 0) {
724         return MBEDTLS_SSL_VERIFY_OPTIONAL;
725     }
726     if (strcmp(s, "required") == 0) {
727         return MBEDTLS_SSL_VERIFY_REQUIRED;
728     }
729 
730     return -1;
731 }
732 
733 /*
734  * Used by sni_parse and psk_parse to handle comma-separated lists
735  */
736 #define GET_ITEM(dst)         \
737     do                          \
738     {                           \
739         (dst) = p;              \
740         while (*p != ',')      \
741         if (++p > end)     \
742         goto error;     \
743         *p++ = '\0';            \
744     } while (0)
745 
746 #if defined(SNI_OPTION)
747 typedef struct _sni_entry sni_entry;
748 
749 struct _sni_entry {
750     const char *name;
751     mbedtls_x509_crt *cert;
752     mbedtls_pk_context *key;
753     mbedtls_x509_crt *ca;
754     mbedtls_x509_crl *crl;
755     int authmode;
756     sni_entry *next;
757 };
758 
sni_free(sni_entry * head)759 static void sni_free(sni_entry *head)
760 {
761     sni_entry *cur = head, *next;
762 
763     while (cur != NULL) {
764         mbedtls_x509_crt_free(cur->cert);
765         mbedtls_free(cur->cert);
766 
767         mbedtls_pk_free(cur->key);
768         mbedtls_free(cur->key);
769 
770         mbedtls_x509_crt_free(cur->ca);
771         mbedtls_free(cur->ca);
772 #if defined(MBEDTLS_X509_CRL_PARSE_C)
773         mbedtls_x509_crl_free(cur->crl);
774         mbedtls_free(cur->crl);
775 #endif
776         next = cur->next;
777         mbedtls_free(cur);
778         cur = next;
779     }
780 }
781 
782 /*
783  * Parse a string of sextuples name1,crt1,key1,ca1,crl1,auth1[,...]
784  * into a usable sni_entry list. For ca1, crl1, auth1, the special value
785  * '-' means unset. If ca1 is unset, then crl1 is ignored too.
786  *
787  * Modifies the input string! This is not production quality!
788  */
sni_parse(char * sni_string)789 static sni_entry *sni_parse(char *sni_string)
790 {
791     sni_entry *cur = NULL, *new = NULL;
792     char *p = sni_string;
793     char *end = p;
794     char *crt_file, *key_file, *ca_file, *auth_str;
795 #if defined(MBEDTLS_X509_CRL_PARSE_C)
796     char *crl_file;
797 #endif
798 
799     while (*end != '\0') {
800         ++end;
801     }
802     *end = ',';
803 
804     while (p <= end) {
805         if ((new = mbedtls_calloc(1, sizeof(sni_entry))) == NULL) {
806             sni_free(cur);
807             return NULL;
808         }
809 
810         GET_ITEM(new->name);
811         GET_ITEM(crt_file);
812         GET_ITEM(key_file);
813         GET_ITEM(ca_file);
814 #if defined(MBEDTLS_X509_CRL_PARSE_C)
815         GET_ITEM(crl_file);
816 #endif
817         GET_ITEM(auth_str);
818 
819         if ((new->cert = mbedtls_calloc(1, sizeof(mbedtls_x509_crt))) == NULL ||
820             (new->key = mbedtls_calloc(1, sizeof(mbedtls_pk_context))) == NULL) {
821             goto error;
822         }
823 
824         mbedtls_x509_crt_init(new->cert);
825         mbedtls_pk_init(new->key);
826 
827         if (mbedtls_x509_crt_parse_file(new->cert, crt_file) != 0 ||
828             mbedtls_pk_parse_keyfile(new->key, key_file, "", rng_get, &rng) != 0) {
829             goto error;
830         }
831 
832         if (strcmp(ca_file, "-") != 0) {
833             if ((new->ca = mbedtls_calloc(1, sizeof(mbedtls_x509_crt))) == NULL) {
834                 goto error;
835             }
836 
837             mbedtls_x509_crt_init(new->ca);
838 
839             if (mbedtls_x509_crt_parse_file(new->ca, ca_file) != 0) {
840                 goto error;
841             }
842         }
843 
844 #if defined(MBEDTLS_X509_CRL_PARSE_C)
845         if (strcmp(crl_file, "-") != 0) {
846             if ((new->crl = mbedtls_calloc(1, sizeof(mbedtls_x509_crl))) == NULL) {
847                 goto error;
848             }
849 
850             mbedtls_x509_crl_init(new->crl);
851 
852             if (mbedtls_x509_crl_parse_file(new->crl, crl_file) != 0) {
853                 goto error;
854             }
855         }
856 #endif
857 
858         if (strcmp(auth_str, "-") != 0) {
859             if ((new->authmode = get_auth_mode(auth_str)) < 0) {
860                 goto error;
861             }
862         } else {
863             new->authmode = DFL_AUTH_MODE;
864         }
865 
866         new->next = cur;
867         cur = new;
868     }
869 
870     return cur;
871 
872 error:
873     sni_free(new);
874     sni_free(cur);
875     return NULL;
876 }
877 
878 /*
879  * SNI callback.
880  */
sni_callback(void * p_info,mbedtls_ssl_context * ssl,const unsigned char * name,size_t name_len)881 static int sni_callback(void *p_info, mbedtls_ssl_context *ssl,
882                         const unsigned char *name, size_t name_len)
883 {
884     const sni_entry *cur = (const sni_entry *) p_info;
885 
886     /* preserve behavior which checks for SNI match in sni_callback() for
887      * the benefits of tests using sni_callback(), even though the actual
888      * certificate assignment has moved to certificate selection callback
889      * in this application.  This exercises sni_callback and cert_callback
890      * even though real applications might choose to do this differently.
891      * Application might choose to save name and name_len in user_data for
892      * later use in certificate selection callback.
893      */
894     while (cur != NULL) {
895         if (name_len == strlen(cur->name) &&
896             memcmp(name, cur->name, name_len) == 0) {
897             void *p;
898             *(const void **)&p = cur;
899             mbedtls_ssl_set_user_data_p(ssl, p);
900             return 0;
901         }
902 
903         cur = cur->next;
904     }
905 
906     return -1;
907 }
908 
909 /*
910  * server certificate selection callback.
911  */
cert_callback(mbedtls_ssl_context * ssl)912 static int cert_callback(mbedtls_ssl_context *ssl)
913 {
914     const sni_entry *cur = (sni_entry *) mbedtls_ssl_get_user_data_p(ssl);
915     if (cur != NULL) {
916         /*(exercise mbedtls_ssl_get_hs_sni(); not otherwise used here)*/
917         size_t name_len;
918         const unsigned char *name = mbedtls_ssl_get_hs_sni(ssl, &name_len);
919         if (strlen(cur->name) != name_len ||
920             memcmp(cur->name, name, name_len) != 0) {
921             return MBEDTLS_ERR_SSL_DECODE_ERROR;
922         }
923 
924         if (cur->ca != NULL) {
925             mbedtls_ssl_set_hs_ca_chain(ssl, cur->ca, cur->crl);
926         }
927 
928         if (cur->authmode != DFL_AUTH_MODE) {
929             mbedtls_ssl_set_hs_authmode(ssl, cur->authmode);
930         }
931 
932         return mbedtls_ssl_set_hs_own_cert(ssl, cur->cert, cur->key);
933     }
934 
935     return 0;
936 }
937 
938 #endif /* SNI_OPTION */
939 
940 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED)
941 
942 typedef struct _psk_entry psk_entry;
943 
944 struct _psk_entry {
945     const char *name;
946     size_t key_len;
947     unsigned char key[MBEDTLS_PSK_MAX_LEN];
948 #if defined(MBEDTLS_USE_PSA_CRYPTO)
949     mbedtls_svc_key_id_t slot;
950 #endif /* MBEDTLS_USE_PSA_CRYPTO */
951     psk_entry *next;
952 };
953 
954 /*
955  * Free a list of psk_entry's
956  */
psk_free(psk_entry * head)957 static int psk_free(psk_entry *head)
958 {
959     psk_entry *next;
960 
961     while (head != NULL) {
962 #if defined(MBEDTLS_USE_PSA_CRYPTO)
963         psa_status_t status;
964         mbedtls_svc_key_id_t const slot = head->slot;
965 
966         if (MBEDTLS_SVC_KEY_ID_GET_KEY_ID(slot) != 0) {
967             status = psa_destroy_key(slot);
968             if (status != PSA_SUCCESS) {
969                 return status;
970             }
971         }
972 #endif /* MBEDTLS_USE_PSA_CRYPTO */
973 
974         next = head->next;
975         mbedtls_free(head);
976         head = next;
977     }
978 
979     return 0;
980 }
981 
982 /*
983  * Parse a string of pairs name1,key1[,name2,key2[,...]]
984  * into a usable psk_entry list.
985  *
986  * Modifies the input string! This is not production quality!
987  */
psk_parse(char * psk_string)988 static psk_entry *psk_parse(char *psk_string)
989 {
990     psk_entry *cur = NULL, *new = NULL;
991     char *p = psk_string;
992     char *end = p;
993     char *key_hex;
994 
995     while (*end != '\0') {
996         ++end;
997     }
998     *end = ',';
999 
1000     while (p <= end) {
1001         if ((new = mbedtls_calloc(1, sizeof(psk_entry))) == NULL) {
1002             goto error;
1003         }
1004 
1005         memset(new, 0, sizeof(psk_entry));
1006 
1007         GET_ITEM(new->name);
1008         GET_ITEM(key_hex);
1009 
1010         if (mbedtls_test_unhexify(new->key, MBEDTLS_PSK_MAX_LEN,
1011                                   key_hex, &new->key_len) != 0) {
1012             goto error;
1013         }
1014 
1015         new->next = cur;
1016         cur = new;
1017     }
1018 
1019     return cur;
1020 
1021 error:
1022     psk_free(new);
1023     psk_free(cur);
1024     return 0;
1025 }
1026 
1027 /*
1028  * PSK callback
1029  */
psk_callback(void * p_info,mbedtls_ssl_context * ssl,const unsigned char * name,size_t name_len)1030 static int psk_callback(void *p_info, mbedtls_ssl_context *ssl,
1031                         const unsigned char *name, size_t name_len)
1032 {
1033     psk_entry *cur = (psk_entry *) p_info;
1034 
1035     while (cur != NULL) {
1036         if (name_len == strlen(cur->name) &&
1037             memcmp(name, cur->name, name_len) == 0) {
1038 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1039             if (MBEDTLS_SVC_KEY_ID_GET_KEY_ID(cur->slot) != 0) {
1040                 return mbedtls_ssl_set_hs_psk_opaque(ssl, cur->slot);
1041             } else
1042 #endif
1043             return mbedtls_ssl_set_hs_psk(ssl, cur->key, cur->key_len);
1044         }
1045 
1046         cur = cur->next;
1047     }
1048 
1049     return -1;
1050 }
1051 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED */
1052 
1053 static mbedtls_net_context listen_fd, client_fd;
1054 
1055 /* Interruption handler to ensure clean exit (for valgrind testing) */
1056 #if !defined(_WIN32)
1057 static int received_sigterm = 0;
term_handler(int sig)1058 static void term_handler(int sig)
1059 {
1060     ((void) sig);
1061     received_sigterm = 1;
1062     mbedtls_net_free(&listen_fd);   /* causes mbedtls_net_accept() to abort */
1063     mbedtls_net_free(&client_fd);   /* causes net_read() to abort */
1064 }
1065 #endif
1066 
1067 /** Return true if \p ret is a status code indicating that there is an
1068  * operation in progress on an SSL connection, and false if it indicates
1069  * success or a fatal error.
1070  *
1071  * The possible operations in progress are:
1072  *
1073  * - A read, when the SSL input buffer does not contain a full message.
1074  * - A write, when the SSL output buffer contains some data that has not
1075  *   been sent over the network yet.
1076  * - An asynchronous callback that has not completed yet. */
mbedtls_status_is_ssl_in_progress(int ret)1077 static int mbedtls_status_is_ssl_in_progress(int ret)
1078 {
1079     return ret == MBEDTLS_ERR_SSL_WANT_READ ||
1080            ret == MBEDTLS_ERR_SSL_WANT_WRITE ||
1081            ret == MBEDTLS_ERR_SSL_ASYNC_IN_PROGRESS;
1082 }
1083 
1084 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
1085 typedef struct {
1086     mbedtls_x509_crt *cert; /*!< Certificate corresponding to the key */
1087     mbedtls_pk_context *pk; /*!< Private key */
1088     unsigned delay; /*!< Number of resume steps to go through */
1089     unsigned pk_owned : 1; /*!< Whether to free the pk object on exit */
1090 } ssl_async_key_slot_t;
1091 
1092 typedef enum {
1093     SSL_ASYNC_INJECT_ERROR_NONE = 0, /*!< Let the callbacks succeed */
1094     SSL_ASYNC_INJECT_ERROR_START, /*!< Inject error during start */
1095     SSL_ASYNC_INJECT_ERROR_CANCEL, /*!< Close the connection after async start */
1096     SSL_ASYNC_INJECT_ERROR_RESUME, /*!< Inject error during resume */
1097 #define SSL_ASYNC_INJECT_ERROR_MAX SSL_ASYNC_INJECT_ERROR_RESUME
1098 } ssl_async_inject_error_t;
1099 
1100 typedef struct {
1101     ssl_async_key_slot_t slots[4]; /* key, key2, sni1, sni2 */
1102     size_t slots_used;
1103     ssl_async_inject_error_t inject_error;
1104     int (*f_rng)(void *, unsigned char *, size_t);
1105     void *p_rng;
1106 } ssl_async_key_context_t;
1107 
ssl_async_set_key(ssl_async_key_context_t * ctx,mbedtls_x509_crt * cert,mbedtls_pk_context * pk,int pk_take_ownership,unsigned delay)1108 static int ssl_async_set_key(ssl_async_key_context_t *ctx,
1109                              mbedtls_x509_crt *cert,
1110                              mbedtls_pk_context *pk,
1111                              int pk_take_ownership,
1112                              unsigned delay)
1113 {
1114     if (ctx->slots_used >= sizeof(ctx->slots) / sizeof(*ctx->slots)) {
1115         return -1;
1116     }
1117     ctx->slots[ctx->slots_used].cert = cert;
1118     ctx->slots[ctx->slots_used].pk = pk;
1119     ctx->slots[ctx->slots_used].delay = delay;
1120     ctx->slots[ctx->slots_used].pk_owned = pk_take_ownership;
1121     ++ctx->slots_used;
1122     return 0;
1123 }
1124 
1125 #define SSL_ASYNC_INPUT_MAX_SIZE 512
1126 
1127 typedef enum {
1128     ASYNC_OP_SIGN,
1129     ASYNC_OP_DECRYPT,
1130 } ssl_async_operation_type_t;
1131 
1132 typedef struct {
1133     unsigned slot;
1134     ssl_async_operation_type_t operation_type;
1135     mbedtls_md_type_t md_alg;
1136     unsigned char input[SSL_ASYNC_INPUT_MAX_SIZE];
1137     size_t input_len;
1138     unsigned remaining_delay;
1139 } ssl_async_operation_context_t;
1140 
1141 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
1142 
1143 /* Note that ssl_async_operation_type_t and the array below need to be kept in sync!
1144  * `ssl_async_operation_names[op]` is the name of op for each value `op`
1145  * of type `ssl_async_operation_type_t`. */
1146 static const char *const ssl_async_operation_names[] =
1147 {
1148     "sign",
1149     "decrypt",
1150 };
1151 
ssl_async_start(mbedtls_ssl_context * ssl,mbedtls_x509_crt * cert,ssl_async_operation_type_t op_type,mbedtls_md_type_t md_alg,const unsigned char * input,size_t input_len)1152 static int ssl_async_start(mbedtls_ssl_context *ssl,
1153                            mbedtls_x509_crt *cert,
1154                            ssl_async_operation_type_t op_type,
1155                            mbedtls_md_type_t md_alg,
1156                            const unsigned char *input,
1157                            size_t input_len)
1158 {
1159     ssl_async_key_context_t *config_data =
1160         mbedtls_ssl_conf_get_async_config_data(ssl->conf);
1161     unsigned slot;
1162     ssl_async_operation_context_t *ctx = NULL;
1163     const char *op_name = ssl_async_operation_names[op_type];
1164 
1165     {
1166         char dn[100];
1167         if (mbedtls_x509_dn_gets(dn, sizeof(dn), &cert->subject) > 0) {
1168             mbedtls_printf("Async %s callback: looking for DN=%s\n",
1169                            op_name, dn);
1170         }
1171     }
1172 
1173     /* Look for a private key that matches the public key in cert.
1174      * Since this test code has the private key inside Mbed TLS,
1175      * we call mbedtls_pk_check_pair to match a private key with the
1176      * public key. */
1177     for (slot = 0; slot < config_data->slots_used; slot++) {
1178         if (mbedtls_pk_check_pair(&cert->pk,
1179                                   config_data->slots[slot].pk,
1180                                   rng_get, &rng) == 0) {
1181             break;
1182         }
1183     }
1184     if (slot == config_data->slots_used) {
1185         mbedtls_printf("Async %s callback: no key matches this certificate.\n",
1186                        op_name);
1187         return MBEDTLS_ERR_SSL_HW_ACCEL_FALLTHROUGH;
1188     }
1189     mbedtls_printf("Async %s callback: using key slot %u, delay=%u.\n",
1190                    op_name, slot, config_data->slots[slot].delay);
1191 
1192     if (config_data->inject_error == SSL_ASYNC_INJECT_ERROR_START) {
1193         mbedtls_printf("Async %s callback: injected error\n", op_name);
1194         return MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE;
1195     }
1196 
1197     if (input_len > SSL_ASYNC_INPUT_MAX_SIZE) {
1198         return MBEDTLS_ERR_SSL_BAD_INPUT_DATA;
1199     }
1200 
1201     ctx = mbedtls_calloc(1, sizeof(*ctx));
1202     if (ctx == NULL) {
1203         return MBEDTLS_ERR_SSL_ALLOC_FAILED;
1204     }
1205     ctx->slot = slot;
1206     ctx->operation_type = op_type;
1207     ctx->md_alg = md_alg;
1208     memcpy(ctx->input, input, input_len);
1209     ctx->input_len = input_len;
1210     ctx->remaining_delay = config_data->slots[slot].delay;
1211     mbedtls_ssl_set_async_operation_data(ssl, ctx);
1212 
1213     if (ctx->remaining_delay == 0) {
1214         return 0;
1215     } else {
1216         return MBEDTLS_ERR_SSL_ASYNC_IN_PROGRESS;
1217     }
1218 }
1219 
ssl_async_sign(mbedtls_ssl_context * ssl,mbedtls_x509_crt * cert,mbedtls_md_type_t md_alg,const unsigned char * hash,size_t hash_len)1220 static int ssl_async_sign(mbedtls_ssl_context *ssl,
1221                           mbedtls_x509_crt *cert,
1222                           mbedtls_md_type_t md_alg,
1223                           const unsigned char *hash,
1224                           size_t hash_len)
1225 {
1226     return ssl_async_start(ssl, cert,
1227                            ASYNC_OP_SIGN, md_alg,
1228                            hash, hash_len);
1229 }
1230 
ssl_async_decrypt(mbedtls_ssl_context * ssl,mbedtls_x509_crt * cert,const unsigned char * input,size_t input_len)1231 static int ssl_async_decrypt(mbedtls_ssl_context *ssl,
1232                              mbedtls_x509_crt *cert,
1233                              const unsigned char *input,
1234                              size_t input_len)
1235 {
1236     return ssl_async_start(ssl, cert,
1237                            ASYNC_OP_DECRYPT, MBEDTLS_MD_NONE,
1238                            input, input_len);
1239 }
1240 
ssl_async_resume(mbedtls_ssl_context * ssl,unsigned char * output,size_t * output_len,size_t output_size)1241 static int ssl_async_resume(mbedtls_ssl_context *ssl,
1242                             unsigned char *output,
1243                             size_t *output_len,
1244                             size_t output_size)
1245 {
1246     ssl_async_operation_context_t *ctx = mbedtls_ssl_get_async_operation_data(ssl);
1247     ssl_async_key_context_t *config_data =
1248         mbedtls_ssl_conf_get_async_config_data(ssl->conf);
1249     ssl_async_key_slot_t *key_slot = &config_data->slots[ctx->slot];
1250     int ret;
1251     const char *op_name;
1252 
1253     if (ctx->remaining_delay > 0) {
1254         --ctx->remaining_delay;
1255         mbedtls_printf("Async resume (slot %u): call %u more times.\n",
1256                        ctx->slot, ctx->remaining_delay);
1257         return MBEDTLS_ERR_SSL_ASYNC_IN_PROGRESS;
1258     }
1259 
1260     switch (ctx->operation_type) {
1261         case ASYNC_OP_DECRYPT:
1262             ret = mbedtls_pk_decrypt(key_slot->pk,
1263                                      ctx->input, ctx->input_len,
1264                                      output, output_len, output_size,
1265                                      config_data->f_rng, config_data->p_rng);
1266             break;
1267         case ASYNC_OP_SIGN:
1268             ret = mbedtls_pk_sign(key_slot->pk,
1269                                   ctx->md_alg,
1270                                   ctx->input, ctx->input_len,
1271                                   output, output_size, output_len,
1272                                   config_data->f_rng, config_data->p_rng);
1273             break;
1274         default:
1275             mbedtls_printf(
1276                 "Async resume (slot %u): unknown operation type %ld. This shouldn't happen.\n",
1277                 ctx->slot,
1278                 (long) ctx->operation_type);
1279             mbedtls_free(ctx);
1280             return MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE;
1281             break;
1282     }
1283 
1284     op_name = ssl_async_operation_names[ctx->operation_type];
1285 
1286     if (config_data->inject_error == SSL_ASYNC_INJECT_ERROR_RESUME) {
1287         mbedtls_printf("Async resume callback: %s done but injected error\n",
1288                        op_name);
1289         mbedtls_free(ctx);
1290         return MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE;
1291     }
1292 
1293     mbedtls_printf("Async resume (slot %u): %s done, status=%d.\n",
1294                    ctx->slot, op_name, ret);
1295     mbedtls_free(ctx);
1296     return ret;
1297 }
1298 
ssl_async_cancel(mbedtls_ssl_context * ssl)1299 static void ssl_async_cancel(mbedtls_ssl_context *ssl)
1300 {
1301     ssl_async_operation_context_t *ctx = mbedtls_ssl_get_async_operation_data(ssl);
1302     mbedtls_printf("Async cancel callback.\n");
1303     mbedtls_free(ctx);
1304 }
1305 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED */
1306 #endif /* MBEDTLS_SSL_ASYNC_PRIVATE */
1307 
1308 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1309 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED)
psa_setup_psk_key_slot(mbedtls_svc_key_id_t * slot,psa_algorithm_t alg,unsigned char * psk,size_t psk_len)1310 static psa_status_t psa_setup_psk_key_slot(mbedtls_svc_key_id_t *slot,
1311                                            psa_algorithm_t alg,
1312                                            unsigned char *psk,
1313                                            size_t psk_len)
1314 {
1315     psa_status_t status;
1316     psa_key_attributes_t key_attributes;
1317 
1318     key_attributes = psa_key_attributes_init();
1319     psa_set_key_usage_flags(&key_attributes, PSA_KEY_USAGE_DERIVE);
1320     psa_set_key_algorithm(&key_attributes, alg);
1321     psa_set_key_type(&key_attributes, PSA_KEY_TYPE_DERIVE);
1322 
1323     status = psa_import_key(&key_attributes, psk, psk_len, slot);
1324     if (status != PSA_SUCCESS) {
1325         fprintf(stderr, "IMPORT\n");
1326         return status;
1327     }
1328 
1329     return PSA_SUCCESS;
1330 }
1331 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED */
1332 #endif /* MBEDTLS_USE_PSA_CRYPTO */
1333 
1334 #if defined(MBEDTLS_SSL_DTLS_CONNECTION_ID)
report_cid_usage(mbedtls_ssl_context * ssl,const char * additional_description)1335 static int report_cid_usage(mbedtls_ssl_context *ssl,
1336                             const char *additional_description)
1337 {
1338     int ret;
1339     unsigned char peer_cid[MBEDTLS_SSL_CID_OUT_LEN_MAX];
1340     size_t peer_cid_len;
1341     int cid_negotiated;
1342 
1343     if (opt.transport != MBEDTLS_SSL_TRANSPORT_DATAGRAM) {
1344         return 0;
1345     }
1346 
1347     /* Check if the use of a CID has been negotiated */
1348     ret = mbedtls_ssl_get_peer_cid(ssl, &cid_negotiated,
1349                                    peer_cid, &peer_cid_len);
1350     if (ret != 0) {
1351         mbedtls_printf(" failed\n  ! mbedtls_ssl_get_peer_cid returned -0x%x\n\n",
1352                        (unsigned int) -ret);
1353         return ret;
1354     }
1355 
1356     if (cid_negotiated == MBEDTLS_SSL_CID_DISABLED) {
1357         if (opt.cid_enabled == MBEDTLS_SSL_CID_ENABLED) {
1358             mbedtls_printf("(%s) Use of Connection ID was not offered by client.\n",
1359                            additional_description);
1360         }
1361     } else {
1362         size_t idx = 0;
1363         mbedtls_printf("(%s) Use of Connection ID has been negotiated.\n",
1364                        additional_description);
1365         mbedtls_printf("(%s) Peer CID (length %u Bytes): ",
1366                        additional_description,
1367                        (unsigned) peer_cid_len);
1368         while (idx < peer_cid_len) {
1369             mbedtls_printf("%02x ", peer_cid[idx]);
1370             idx++;
1371         }
1372         mbedtls_printf("\n");
1373     }
1374 
1375     return 0;
1376 }
1377 #endif /* MBEDTLS_SSL_DTLS_CONNECTION_ID */
1378 
1379 #if defined(MBEDTLS_SSL_SESSION_TICKETS) && defined(MBEDTLS_SSL_TICKET_C) && \
1380     defined(MBEDTLS_HAVE_TIME)
put_unaligned_uint32(void * p,uint32_t x)1381 static inline void put_unaligned_uint32(void *p, uint32_t x)
1382 {
1383     memcpy(p, &x, sizeof(x));
1384 }
1385 
1386 /* Functions for session ticket tests */
dummy_ticket_write(void * p_ticket,const mbedtls_ssl_session * session,unsigned char * start,const unsigned char * end,size_t * tlen,uint32_t * ticket_lifetime)1387 static int dummy_ticket_write(void *p_ticket, const mbedtls_ssl_session *session,
1388                               unsigned char *start, const unsigned char *end,
1389                               size_t *tlen, uint32_t *ticket_lifetime)
1390 {
1391     int ret;
1392     unsigned char *p = start;
1393     size_t clear_len;
1394     ((void) p_ticket);
1395 
1396     if (end - p < 4) {
1397         return MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL;
1398     }
1399     put_unaligned_uint32(p, 7 * 24 * 3600);
1400     *ticket_lifetime = 7 * 24 * 3600;
1401     p += 4;
1402 
1403     /* Dump session state */
1404     if ((ret = mbedtls_ssl_session_save(session, p, end - p,
1405                                         &clear_len)) != 0) {
1406         return ret;
1407     }
1408 
1409     *tlen = 4 + clear_len;
1410 
1411     return 0;
1412 }
1413 
dummy_ticket_parse(void * p_ticket,mbedtls_ssl_session * session,unsigned char * buf,size_t len)1414 static int dummy_ticket_parse(void *p_ticket, mbedtls_ssl_session *session,
1415                               unsigned char *buf, size_t len)
1416 {
1417     int ret;
1418     ((void) p_ticket);
1419 
1420     if ((ret = mbedtls_ssl_session_load(session, buf + 4, len - 4)) != 0) {
1421         return ret;
1422     }
1423 
1424     switch (opt.dummy_ticket % 11) {
1425         case 1:
1426             return MBEDTLS_ERR_SSL_INVALID_MAC;
1427         case 2:
1428             return MBEDTLS_ERR_SSL_SESSION_TICKET_EXPIRED;
1429         case 3:
1430             /* Creation time in the future. */
1431             session->ticket_creation_time = mbedtls_ms_time() + 1000;
1432             break;
1433         case 4:
1434             /* Ticket has reached the end of lifetime. */
1435             session->ticket_creation_time = mbedtls_ms_time() -
1436                                             (7 * 24 * 3600 * 1000 + 1000);
1437             break;
1438 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
1439         case 5:
1440             /* Ticket is valid, but client age is below the lower bound of the tolerance window. */
1441             session->ticket_age_add += MBEDTLS_SSL_TLS1_3_TICKET_AGE_TOLERANCE + 4 * 1000;
1442             /* Make sure the execution time does not affect the result */
1443             session->ticket_creation_time = mbedtls_ms_time();
1444             break;
1445 
1446         case 6:
1447             /* Ticket is valid, but client age is beyond the upper bound of the tolerance window. */
1448             session->ticket_age_add -= MBEDTLS_SSL_TLS1_3_TICKET_AGE_TOLERANCE + 4 * 1000;
1449             /* Make sure the execution time does not affect the result */
1450             session->ticket_creation_time = mbedtls_ms_time();
1451             break;
1452         case 7:
1453             session->ticket_flags = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_NONE;
1454             break;
1455         case 8:
1456             session->ticket_flags = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK;
1457             break;
1458         case 9:
1459             session->ticket_flags = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK_EPHEMERAL;
1460             break;
1461         case 10:
1462             session->ticket_flags = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK_ALL;
1463             break;
1464 #endif
1465         default:
1466             break;
1467     }
1468 
1469     return ret;
1470 }
1471 #endif /* MBEDTLS_SSL_SESSION_TICKETS && MBEDTLS_SSL_TICKET_C && MBEDTLS_HAVE_TIME */
1472 
parse_cipher(char * buf)1473 static int parse_cipher(char *buf)
1474 {
1475     if (strcmp(buf, "AES-128-CCM")) {
1476         return MBEDTLS_CIPHER_AES_128_CCM;
1477     } else if (strcmp(buf, "AES-128-GCM")) {
1478         return MBEDTLS_CIPHER_AES_128_GCM;
1479     } else if (strcmp(buf, "AES-192-CCM")) {
1480         return MBEDTLS_CIPHER_AES_192_CCM;
1481     } else if (strcmp(buf, "AES-192-GCM")) {
1482         return MBEDTLS_CIPHER_AES_192_GCM;
1483     } else if (strcmp(buf, "AES-256-CCM")) {
1484         return MBEDTLS_CIPHER_AES_256_CCM;
1485     } else if (strcmp(buf, "ARIA-128-CCM")) {
1486         return MBEDTLS_CIPHER_ARIA_128_CCM;
1487     } else if (strcmp(buf, "ARIA-128-GCM")) {
1488         return MBEDTLS_CIPHER_ARIA_128_GCM;
1489     } else if (strcmp(buf, "ARIA-192-CCM")) {
1490         return MBEDTLS_CIPHER_ARIA_192_CCM;
1491     } else if (strcmp(buf, "ARIA-192-GCM")) {
1492         return MBEDTLS_CIPHER_ARIA_192_GCM;
1493     } else if (strcmp(buf, "ARIA-256-CCM")) {
1494         return MBEDTLS_CIPHER_ARIA_256_CCM;
1495     } else if (strcmp(buf, "ARIA-256-GCM")) {
1496         return MBEDTLS_CIPHER_ARIA_256_GCM;
1497     } else if (strcmp(buf, "CAMELLIA-128-CCM")) {
1498         return MBEDTLS_CIPHER_CAMELLIA_128_CCM;
1499     } else if (strcmp(buf, "CAMELLIA-192-CCM")) {
1500         return MBEDTLS_CIPHER_CAMELLIA_192_CCM;
1501     } else if (strcmp(buf, "CAMELLIA-256-CCM")) {
1502         return MBEDTLS_CIPHER_CAMELLIA_256_CCM;
1503     } else if (strcmp(buf, "CHACHA20-POLY1305")) {
1504         return MBEDTLS_CIPHER_CHACHA20_POLY1305;
1505     }
1506     return MBEDTLS_CIPHER_NONE;
1507 }
1508 
main(int argc,char * argv[])1509 int main(int argc, char *argv[])
1510 {
1511     int ret = 0, len, written, frags, exchanges_left;
1512     int query_config_ret = 0;
1513     io_ctx_t io_ctx;
1514     unsigned char *buf = 0;
1515 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED)
1516 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1517     psa_algorithm_t alg = 0;
1518     mbedtls_svc_key_id_t psk_slot = MBEDTLS_SVC_KEY_ID_INIT;
1519 #endif /* MBEDTLS_USE_PSA_CRYPTO */
1520     unsigned char psk[MBEDTLS_PSK_MAX_LEN];
1521     size_t psk_len = 0;
1522     psk_entry *psk_info = NULL;
1523 #endif
1524     const char *pers = "ssl_server2";
1525     unsigned char client_ip[16] = { 0 };
1526     size_t cliip_len;
1527 #if defined(MBEDTLS_SSL_COOKIE_C)
1528     mbedtls_ssl_cookie_ctx cookie_ctx;
1529 #endif
1530 
1531     mbedtls_ssl_context ssl;
1532     mbedtls_ssl_config conf;
1533 #if defined(MBEDTLS_TIMING_C)
1534     mbedtls_timing_delay_context timer;
1535 #endif
1536 #if defined(MBEDTLS_SSL_RENEGOTIATION)
1537     unsigned char renego_period[8] = { 0 };
1538 #endif
1539 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
1540     uint32_t flags;
1541     mbedtls_x509_crt cacert;
1542     mbedtls_x509_crt srvcert;
1543     mbedtls_pk_context pkey;
1544     mbedtls_x509_crt srvcert2;
1545     mbedtls_pk_context pkey2;
1546     mbedtls_x509_crt_profile crt_profile_for_test = mbedtls_x509_crt_profile_default;
1547 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1548     mbedtls_svc_key_id_t key_slot = MBEDTLS_SVC_KEY_ID_INIT; /* invalid key slot */
1549     mbedtls_svc_key_id_t key_slot2 = MBEDTLS_SVC_KEY_ID_INIT; /* invalid key slot */
1550 #endif
1551     int key_cert_init = 0, key_cert_init2 = 0;
1552 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED */
1553 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
1554     ssl_async_key_context_t ssl_async_keys;
1555 #endif /* MBEDTLS_SSL_ASYNC_PRIVATE */
1556 #if defined(MBEDTLS_DHM_C) && defined(MBEDTLS_FS_IO)
1557     mbedtls_dhm_context dhm;
1558 #endif
1559 #if defined(MBEDTLS_SSL_CACHE_C)
1560     mbedtls_ssl_cache_context cache;
1561 #endif
1562 #if defined(MBEDTLS_SSL_SESSION_TICKETS) && defined(MBEDTLS_SSL_TICKET_C)
1563     mbedtls_ssl_ticket_context ticket_ctx;
1564 #endif /* MBEDTLS_SSL_SESSION_TICKETS && MBEDTLS_SSL_TICKET_C */
1565 #if defined(SNI_OPTION)
1566     sni_entry *sni_info = NULL;
1567 #endif
1568     uint16_t group_list[GROUP_LIST_SIZE];
1569 #if defined(MBEDTLS_SSL_ALPN)
1570     const char *alpn_list[ALPN_LIST_SIZE];
1571 #endif
1572 #if defined(MBEDTLS_MEMORY_BUFFER_ALLOC_C)
1573     unsigned char alloc_buf[MEMORY_HEAP_SIZE];
1574 #endif
1575 #if defined(MBEDTLS_SSL_DTLS_CONNECTION_ID)
1576     unsigned char cid[MBEDTLS_SSL_CID_IN_LEN_MAX];
1577     unsigned char cid_renego[MBEDTLS_SSL_CID_IN_LEN_MAX];
1578     size_t cid_len = 0;
1579     size_t cid_renego_len = 0;
1580 #endif
1581 #if defined(MBEDTLS_SSL_CONTEXT_SERIALIZATION)
1582     unsigned char *context_buf = NULL;
1583     size_t context_buf_len = 0;
1584 #endif
1585 #if defined(MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED) && \
1586     defined(MBEDTLS_USE_PSA_CRYPTO)
1587     mbedtls_svc_key_id_t ecjpake_pw_slot = MBEDTLS_SVC_KEY_ID_INIT; /* ecjpake password key slot */
1588 #endif /* MBEDTLS_USE_PSA_CRYPTO && MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED */
1589 
1590 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
1591     uint16_t sig_alg_list[SIG_ALG_LIST_SIZE];
1592 #endif
1593 
1594     int i;
1595     char *p, *q;
1596     const int *list;
1597 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1598     psa_status_t status;
1599 #endif
1600     unsigned char eap_tls_keymaterial[16];
1601     unsigned char eap_tls_iv[8];
1602     const char *eap_tls_label = "client EAP encryption";
1603     eap_tls_keys eap_tls_keying;
1604 #if defined(MBEDTLS_SSL_DTLS_SRTP)
1605     /*! master keys and master salt for SRTP generated during handshake */
1606     unsigned char dtls_srtp_key_material[MBEDTLS_TLS_SRTP_MAX_KEY_MATERIAL_LENGTH];
1607     const char *dtls_srtp_label = "EXTRACTOR-dtls_srtp";
1608     dtls_srtp_keys dtls_srtp_keying;
1609     const mbedtls_ssl_srtp_profile default_profiles[] = {
1610         MBEDTLS_TLS_SRTP_AES128_CM_HMAC_SHA1_80,
1611         MBEDTLS_TLS_SRTP_AES128_CM_HMAC_SHA1_32,
1612         MBEDTLS_TLS_SRTP_NULL_HMAC_SHA1_80,
1613         MBEDTLS_TLS_SRTP_NULL_HMAC_SHA1_32,
1614         MBEDTLS_TLS_SRTP_UNSET
1615     };
1616 #endif /* MBEDTLS_SSL_DTLS_SRTP */
1617 
1618 #if defined(MBEDTLS_MEMORY_BUFFER_ALLOC_C)
1619     mbedtls_memory_buffer_alloc_init(alloc_buf, sizeof(alloc_buf));
1620 #if defined(MBEDTLS_MEMORY_DEBUG)
1621     size_t current_heap_memory, peak_heap_memory, heap_blocks;
1622 #endif  /* MBEDTLS_MEMORY_DEBUG */
1623 #endif  /* MBEDTLS_MEMORY_BUFFER_ALLOC_C */
1624 
1625 #if defined(MBEDTLS_TEST_HOOKS)
1626     test_hooks_init();
1627 #endif /* MBEDTLS_TEST_HOOKS */
1628 
1629     /*
1630      * Make sure memory references are valid in case we exit early.
1631      */
1632     mbedtls_net_init(&client_fd);
1633     mbedtls_net_init(&listen_fd);
1634     mbedtls_ssl_init(&ssl);
1635     mbedtls_ssl_config_init(&conf);
1636     rng_init(&rng);
1637 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
1638     mbedtls_x509_crt_init(&cacert);
1639     mbedtls_x509_crt_init(&srvcert);
1640     mbedtls_pk_init(&pkey);
1641     mbedtls_x509_crt_init(&srvcert2);
1642     mbedtls_pk_init(&pkey2);
1643 #endif
1644 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
1645     memset(&ssl_async_keys, 0, sizeof(ssl_async_keys));
1646 #endif
1647 #if defined(MBEDTLS_DHM_C) && defined(MBEDTLS_FS_IO)
1648     mbedtls_dhm_init(&dhm);
1649 #endif
1650 #if defined(MBEDTLS_SSL_CACHE_C)
1651     mbedtls_ssl_cache_init(&cache);
1652 #endif
1653 #if defined(MBEDTLS_SSL_SESSION_TICKETS) && defined(MBEDTLS_SSL_TICKET_C)
1654     mbedtls_ssl_ticket_init(&ticket_ctx);
1655 #endif
1656 #if defined(MBEDTLS_SSL_ALPN)
1657     memset((void *) alpn_list, 0, sizeof(alpn_list));
1658 #endif
1659 #if defined(MBEDTLS_SSL_COOKIE_C)
1660     mbedtls_ssl_cookie_init(&cookie_ctx);
1661 #endif
1662 
1663     /* For builds with TLS 1.3 enabled but not MBEDTLS_USE_PSA_CRYPTO,
1664      * we deliberately do not call psa_crypto_init() here, to test that
1665      * the library is backward-compatible with versions prior to 3.6.0
1666      * where calling psa_crypto_init() was not required to open a TLS
1667      * connection in the default configuration. See
1668      * https://github.com/Mbed-TLS/mbedtls/issues/9072 and
1669      * mbedtls_ssl_tls13_crypto_init().
1670      */
1671 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1672     status = psa_crypto_init();
1673     if (status != PSA_SUCCESS) {
1674         mbedtls_fprintf(stderr, "Failed to initialize PSA Crypto implementation: %d\n",
1675                         (int) status);
1676         ret = MBEDTLS_ERR_SSL_HW_ACCEL_FAILED;
1677         goto exit;
1678     }
1679 #endif  /* MBEDTLS_USE_PSA_CRYPTO */
1680 #if defined(MBEDTLS_PSA_CRYPTO_EXTERNAL_RNG)
1681     mbedtls_test_enable_insecure_external_rng();
1682 #endif  /* MBEDTLS_PSA_CRYPTO_EXTERNAL_RNG */
1683 
1684 #if !defined(_WIN32)
1685     /* Abort cleanly on SIGTERM and SIGINT */
1686     signal(SIGTERM, term_handler);
1687     signal(SIGINT, term_handler);
1688 #endif
1689 
1690     opt.buffer_size         = DFL_IO_BUF_LEN;
1691     opt.server_addr         = DFL_SERVER_ADDR;
1692     opt.server_port         = DFL_SERVER_PORT;
1693     opt.debug_level         = DFL_DEBUG_LEVEL;
1694     opt.event               = DFL_EVENT;
1695     opt.response_size       = DFL_RESPONSE_SIZE;
1696     opt.nbio                = DFL_NBIO;
1697     opt.cid_enabled         = DFL_CID_ENABLED;
1698     opt.cid_enabled_renego  = DFL_CID_ENABLED_RENEGO;
1699     opt.cid_val             = DFL_CID_VALUE;
1700     opt.cid_val_renego      = DFL_CID_VALUE_RENEGO;
1701     opt.read_timeout        = DFL_READ_TIMEOUT;
1702     opt.ca_file             = DFL_CA_FILE;
1703     opt.ca_path             = DFL_CA_PATH;
1704     opt.crt_file            = DFL_CRT_FILE;
1705     opt.key_file            = DFL_KEY_FILE;
1706     opt.key_opaque          = DFL_KEY_OPAQUE;
1707     opt.key_pwd             = DFL_KEY_PWD;
1708     opt.crt_file2           = DFL_CRT_FILE2;
1709     opt.key_file2           = DFL_KEY_FILE2;
1710     opt.key_pwd2            = DFL_KEY_PWD2;
1711     opt.async_operations    = DFL_ASYNC_OPERATIONS;
1712     opt.async_private_delay1 = DFL_ASYNC_PRIVATE_DELAY1;
1713     opt.async_private_delay2 = DFL_ASYNC_PRIVATE_DELAY2;
1714     opt.async_private_error = DFL_ASYNC_PRIVATE_ERROR;
1715     opt.psk                 = DFL_PSK;
1716 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1717     opt.psk_opaque          = DFL_PSK_OPAQUE;
1718     opt.psk_list_opaque     = DFL_PSK_LIST_OPAQUE;
1719 #endif
1720 #if defined(MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK)
1721     opt.ca_callback         = DFL_CA_CALLBACK;
1722 #endif
1723     opt.psk_identity        = DFL_PSK_IDENTITY;
1724     opt.psk_list            = DFL_PSK_LIST;
1725     opt.ecjpake_pw          = DFL_ECJPAKE_PW;
1726 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1727     opt.ecjpake_pw_opaque   = DFL_ECJPAKE_PW_OPAQUE;
1728 #endif
1729     opt.force_ciphersuite[0] = DFL_FORCE_CIPHER;
1730 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
1731     opt.tls13_kex_modes     = DFL_TLS1_3_KEX_MODES;
1732 #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
1733     opt.renegotiation       = DFL_RENEGOTIATION;
1734     opt.allow_legacy        = DFL_ALLOW_LEGACY;
1735     opt.renegotiate         = DFL_RENEGOTIATE;
1736     opt.renego_delay        = DFL_RENEGO_DELAY;
1737     opt.renego_period       = DFL_RENEGO_PERIOD;
1738     opt.exchanges           = DFL_EXCHANGES;
1739     opt.min_version         = DFL_MIN_VERSION;
1740     opt.max_version         = DFL_MAX_VERSION;
1741     opt.allow_sha1          = DFL_SHA1;
1742     opt.auth_mode           = DFL_AUTH_MODE;
1743     opt.cert_req_ca_list    = DFL_CERT_REQ_CA_LIST;
1744     opt.cert_req_dn_hint    = DFL_CERT_REQ_DN_HINT;
1745     opt.mfl_code            = DFL_MFL_CODE;
1746     opt.trunc_hmac          = DFL_TRUNC_HMAC;
1747     opt.tickets             = DFL_TICKETS;
1748     opt.dummy_ticket        = DFL_DUMMY_TICKET;
1749     opt.ticket_rotate       = DFL_TICKET_ROTATE;
1750     opt.ticket_timeout      = DFL_TICKET_TIMEOUT;
1751     opt.ticket_aead         = DFL_TICKET_AEAD;
1752     opt.cache_max           = DFL_CACHE_MAX;
1753 #if defined(MBEDTLS_HAVE_TIME)
1754     opt.cache_timeout       = DFL_CACHE_TIMEOUT;
1755 #endif
1756     opt.cache_remove        = DFL_CACHE_REMOVE;
1757     opt.sni                 = DFL_SNI;
1758     opt.alpn_string         = DFL_ALPN_STRING;
1759     opt.groups              = DFL_GROUPS;
1760 #if defined(MBEDTLS_SSL_EARLY_DATA)
1761     opt.early_data          = DFL_EARLY_DATA;
1762     opt.max_early_data_size = DFL_MAX_EARLY_DATA_SIZE;
1763 #endif
1764     opt.sig_algs            = DFL_SIG_ALGS;
1765     opt.dhm_file            = DFL_DHM_FILE;
1766     opt.transport           = DFL_TRANSPORT;
1767     opt.cookies             = DFL_COOKIES;
1768     opt.anti_replay         = DFL_ANTI_REPLAY;
1769     opt.hs_to_min           = DFL_HS_TO_MIN;
1770     opt.hs_to_max           = DFL_HS_TO_MAX;
1771     opt.dtls_mtu            = DFL_DTLS_MTU;
1772     opt.dgram_packing       = DFL_DGRAM_PACKING;
1773     opt.badmac_limit        = DFL_BADMAC_LIMIT;
1774     opt.extended_ms         = DFL_EXTENDED_MS;
1775     opt.etm                 = DFL_ETM;
1776     opt.serialize           = DFL_SERIALIZE;
1777     opt.context_file        = DFL_CONTEXT_FILE;
1778     opt.eap_tls             = DFL_EAP_TLS;
1779     opt.reproducible        = DFL_REPRODUCIBLE;
1780     opt.nss_keylog          = DFL_NSS_KEYLOG;
1781     opt.nss_keylog_file     = DFL_NSS_KEYLOG_FILE;
1782     opt.query_config_mode   = DFL_QUERY_CONFIG_MODE;
1783     opt.use_srtp            = DFL_USE_SRTP;
1784     opt.force_srtp_profile  = DFL_SRTP_FORCE_PROFILE;
1785     opt.support_mki         = DFL_SRTP_SUPPORT_MKI;
1786     opt.key1_opaque_alg1   = DFL_KEY_OPAQUE_ALG;
1787     opt.key1_opaque_alg2   = DFL_KEY_OPAQUE_ALG;
1788     opt.key2_opaque_alg1   = DFL_KEY_OPAQUE_ALG;
1789     opt.key2_opaque_alg2   = DFL_KEY_OPAQUE_ALG;
1790 
1791     p = q = NULL;
1792     if (argc < 1) {
1793 usage:
1794         if (p != NULL && q != NULL) {
1795             printf("unrecognized value for '%s': '%s'\n", p, q);
1796         } else if (p != NULL && q == NULL) {
1797             printf("unrecognized param: '%s'\n", p);
1798         }
1799 
1800         mbedtls_printf("usage: ssl_client2 [param=value] [...]\n");
1801         mbedtls_printf("       ssl_client2 help[_theme]\n");
1802         mbedtls_printf("'help' lists acceptable 'param' and 'value'\n");
1803         mbedtls_printf("'help_ciphersuites' lists available ciphersuites\n");
1804         mbedtls_printf("\n");
1805 
1806         if (ret == 0) {
1807             ret = 1;
1808         }
1809         goto exit;
1810     }
1811 
1812     for (i = 1; i < argc; i++) {
1813         p = argv[i];
1814 
1815         if (strcmp(p, "help") == 0) {
1816             mbedtls_printf(USAGE1);
1817             mbedtls_printf(USAGE2);
1818             mbedtls_printf(USAGE3);
1819             mbedtls_printf(USAGE4);
1820 
1821             ret = 0;
1822             goto exit;
1823         }
1824         if (strcmp(p, "help_ciphersuites") == 0) {
1825             mbedtls_printf(" acceptable ciphersuite names:\n");
1826             for (list = mbedtls_ssl_list_ciphersuites();
1827                  *list != 0;
1828                  list++) {
1829                 mbedtls_printf(" %s\n", mbedtls_ssl_get_ciphersuite_name(*list));
1830             }
1831 
1832             ret = 0;
1833             goto exit;
1834         }
1835 
1836         if ((q = strchr(p, '=')) == NULL) {
1837             mbedtls_printf("param requires a value: '%s'\n", p);
1838             p = NULL; // avoid "unrecnognized param" message
1839             goto usage;
1840         }
1841         *q++ = '\0';
1842 
1843         if (strcmp(p, "server_port") == 0) {
1844             opt.server_port = q;
1845         } else if (strcmp(p, "server_addr") == 0) {
1846             opt.server_addr = q;
1847         } else if (strcmp(p, "dtls") == 0) {
1848             int t = atoi(q);
1849             if (t == 0) {
1850                 opt.transport = MBEDTLS_SSL_TRANSPORT_STREAM;
1851             } else if (t == 1) {
1852                 opt.transport = MBEDTLS_SSL_TRANSPORT_DATAGRAM;
1853             } else {
1854                 goto usage;
1855             }
1856         } else if (strcmp(p, "debug_level") == 0) {
1857             opt.debug_level = atoi(q);
1858             if (opt.debug_level < 0 || opt.debug_level > 65535) {
1859                 goto usage;
1860             }
1861         } else if (strcmp(p, "build_version") == 0) {
1862             if (strcmp(q, "1") == 0) {
1863                 mbedtls_printf("build version: %s (build %d)\n",
1864                                MBEDTLS_VERSION_STRING_FULL,
1865                                MBEDTLS_VERSION_NUMBER);
1866                 goto exit;
1867             }
1868         } else if (strcmp(p, "nbio") == 0) {
1869             opt.nbio = atoi(q);
1870             if (opt.nbio < 0 || opt.nbio > 2) {
1871                 goto usage;
1872             }
1873         } else if (strcmp(p, "event") == 0) {
1874             opt.event = atoi(q);
1875             if (opt.event < 0 || opt.event > 2) {
1876                 goto usage;
1877             }
1878         } else if (strcmp(p, "read_timeout") == 0) {
1879             opt.read_timeout = atoi(q);
1880         } else if (strcmp(p, "buffer_size") == 0) {
1881             opt.buffer_size = atoi(q);
1882             if (opt.buffer_size < 1) {
1883                 goto usage;
1884             }
1885         } else if (strcmp(p, "response_size") == 0) {
1886             opt.response_size = atoi(q);
1887             if (opt.response_size < 0 || opt.response_size > MBEDTLS_SSL_OUT_CONTENT_LEN) {
1888                 goto usage;
1889             }
1890             if (opt.buffer_size < opt.response_size) {
1891                 opt.buffer_size = opt.response_size;
1892             }
1893         } else if (strcmp(p, "ca_file") == 0) {
1894             opt.ca_file = q;
1895         } else if (strcmp(p, "ca_path") == 0) {
1896             opt.ca_path = q;
1897         } else if (strcmp(p, "crt_file") == 0) {
1898             opt.crt_file = q;
1899         } else if (strcmp(p, "key_file") == 0) {
1900             opt.key_file = q;
1901         } else if (strcmp(p, "key_pwd") == 0) {
1902             opt.key_pwd = q;
1903         }
1904 #if defined(MBEDTLS_USE_PSA_CRYPTO) && defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
1905         else if (strcmp(p, "key_opaque") == 0) {
1906             opt.key_opaque = atoi(q);
1907         }
1908 #endif
1909         else if (strcmp(p, "crt_file2") == 0) {
1910             opt.crt_file2 = q;
1911         } else if (strcmp(p, "key_file2") == 0) {
1912             opt.key_file2 = q;
1913         } else if (strcmp(p, "key_pwd2") == 0) {
1914             opt.key_pwd2 = q;
1915         } else if (strcmp(p, "dhm_file") == 0) {
1916             opt.dhm_file = q;
1917         }
1918 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
1919         else if (strcmp(p, "async_operations") == 0) {
1920             opt.async_operations = q;
1921         } else if (strcmp(p, "async_private_delay1") == 0) {
1922             opt.async_private_delay1 = atoi(q);
1923         } else if (strcmp(p, "async_private_delay2") == 0) {
1924             opt.async_private_delay2 = atoi(q);
1925         } else if (strcmp(p, "async_private_error") == 0) {
1926             int n = atoi(q);
1927             if (n < -SSL_ASYNC_INJECT_ERROR_MAX ||
1928                 n > SSL_ASYNC_INJECT_ERROR_MAX) {
1929                 ret = 2;
1930                 goto usage;
1931             }
1932             opt.async_private_error = n;
1933         }
1934 #endif /* MBEDTLS_SSL_ASYNC_PRIVATE */
1935 #if defined(MBEDTLS_SSL_DTLS_CONNECTION_ID)
1936         else if (strcmp(p, "cid") == 0) {
1937             opt.cid_enabled = atoi(q);
1938             if (opt.cid_enabled != 0 && opt.cid_enabled != 1) {
1939                 goto usage;
1940             }
1941         } else if (strcmp(p, "cid_renego") == 0) {
1942             opt.cid_enabled_renego = atoi(q);
1943             if (opt.cid_enabled_renego != 0 && opt.cid_enabled_renego != 1) {
1944                 goto usage;
1945             }
1946         } else if (strcmp(p, "cid_val") == 0) {
1947             opt.cid_val = q;
1948         } else if (strcmp(p, "cid_val_renego") == 0) {
1949             opt.cid_val_renego = q;
1950         }
1951 #endif /* MBEDTLS_SSL_DTLS_CONNECTION_ID */
1952         else if (strcmp(p, "psk") == 0) {
1953             opt.psk = q;
1954         }
1955 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1956         else if (strcmp(p, "psk_opaque") == 0) {
1957             opt.psk_opaque = atoi(q);
1958         } else if (strcmp(p, "psk_list_opaque") == 0) {
1959             opt.psk_list_opaque = atoi(q);
1960         }
1961 #endif
1962 #if defined(MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK)
1963         else if (strcmp(p, "ca_callback") == 0) {
1964             opt.ca_callback = atoi(q);
1965         }
1966 #endif
1967         else if (strcmp(p, "psk_identity") == 0) {
1968             opt.psk_identity = q;
1969         } else if (strcmp(p, "psk_list") == 0) {
1970             opt.psk_list = q;
1971         } else if (strcmp(p, "ecjpake_pw") == 0) {
1972             opt.ecjpake_pw = q;
1973         }
1974 #if defined(MBEDTLS_USE_PSA_CRYPTO)
1975         else if (strcmp(p, "ecjpake_pw_opaque") == 0) {
1976             opt.ecjpake_pw_opaque = atoi(q);
1977         }
1978 #endif
1979         else if (strcmp(p, "force_ciphersuite") == 0) {
1980             opt.force_ciphersuite[0] = mbedtls_ssl_get_ciphersuite_id(q);
1981 
1982             if (opt.force_ciphersuite[0] == 0) {
1983                 ret = 2;
1984                 goto usage;
1985             }
1986             opt.force_ciphersuite[1] = 0;
1987         } else if (strcmp(p, "groups") == 0) {
1988             opt.groups = q;
1989         }
1990 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
1991         else if (strcmp(p, "sig_algs") == 0) {
1992             opt.sig_algs = q;
1993         }
1994 #endif
1995 #if defined(MBEDTLS_SSL_EARLY_DATA)
1996         else if (strcmp(p, "early_data") == 0) {
1997             switch (atoi(q)) {
1998                 case 0:
1999                     opt.early_data = MBEDTLS_SSL_EARLY_DATA_DISABLED;
2000                     break;
2001                 case 1:
2002                     opt.early_data = MBEDTLS_SSL_EARLY_DATA_ENABLED;
2003                     break;
2004                 default: goto usage;
2005             }
2006         } else if (strcmp(p, "max_early_data_size") == 0) {
2007             opt.max_early_data_size = (uint32_t) atoll(q);
2008         }
2009 #endif /* MBEDTLS_SSL_EARLY_DATA */
2010         else if (strcmp(p, "renegotiation") == 0) {
2011             opt.renegotiation = (atoi(q)) ?
2012                                 MBEDTLS_SSL_RENEGOTIATION_ENABLED :
2013                                 MBEDTLS_SSL_RENEGOTIATION_DISABLED;
2014         } else if (strcmp(p, "allow_legacy") == 0) {
2015             switch (atoi(q)) {
2016                 case -1:
2017                     opt.allow_legacy = MBEDTLS_SSL_LEGACY_BREAK_HANDSHAKE;
2018                     break;
2019                 case 0:
2020                     opt.allow_legacy = MBEDTLS_SSL_LEGACY_NO_RENEGOTIATION;
2021                     break;
2022                 case 1:
2023                     opt.allow_legacy = MBEDTLS_SSL_LEGACY_ALLOW_RENEGOTIATION;
2024                     break;
2025                 default: goto usage;
2026             }
2027         } else if (strcmp(p, "renegotiate") == 0) {
2028             opt.renegotiate = atoi(q);
2029             if (opt.renegotiate < 0 || opt.renegotiate > 1) {
2030                 goto usage;
2031             }
2032         } else if (strcmp(p, "renego_delay") == 0) {
2033             opt.renego_delay = atoi(q);
2034         } else if (strcmp(p, "renego_period") == 0) {
2035 #if defined(_MSC_VER)
2036             opt.renego_period = _strtoui64(q, NULL, 10);
2037 #else
2038             if (sscanf(q, "%" SCNu64, &opt.renego_period) != 1) {
2039                 goto usage;
2040             }
2041 #endif /* _MSC_VER */
2042             if (opt.renego_period < 2) {
2043                 goto usage;
2044             }
2045         } else if (strcmp(p, "exchanges") == 0) {
2046             opt.exchanges = atoi(q);
2047             if (opt.exchanges < 0) {
2048                 goto usage;
2049             }
2050         }
2051 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
2052         else if (strcmp(p, "tls13_kex_modes") == 0) {
2053             if (strcmp(q, "psk") == 0) {
2054                 opt.tls13_kex_modes = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK;
2055             } else if (strcmp(q, "psk_ephemeral") == 0) {
2056                 opt.tls13_kex_modes = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK_EPHEMERAL;
2057             } else if (strcmp(q, "ephemeral") == 0) {
2058                 opt.tls13_kex_modes = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_EPHEMERAL;
2059             } else if (strcmp(q, "ephemeral_all") == 0) {
2060                 opt.tls13_kex_modes = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_EPHEMERAL_ALL;
2061             } else if (strcmp(q, "psk_all") == 0) {
2062                 opt.tls13_kex_modes = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK_ALL;
2063             } else if (strcmp(q, "all") == 0) {
2064                 opt.tls13_kex_modes = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_ALL;
2065             }
2066             /* The purpose of `psk_or_ephemeral` is to improve test coverage. That
2067              * is not recommended in practice.
2068              * `psk_or_ephemeral` exists in theory, we need this mode to test if
2069              * this setting work correctly. With this key exchange setting, server
2070              * should always perform `ephemeral` handshake. `psk` or `psk_ephemeral`
2071              * is not expected.
2072              */
2073             else if (strcmp(q, "psk_or_ephemeral") == 0) {
2074                 opt.tls13_kex_modes = MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_PSK |
2075                                       MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_EPHEMERAL;
2076             } else {
2077                 goto usage;
2078             }
2079         }
2080 #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
2081 
2082         else if (strcmp(p, "min_version") == 0) {
2083             if (strcmp(q, "tls12") == 0 ||
2084                 strcmp(q, "dtls12") == 0) {
2085                 opt.min_version = MBEDTLS_SSL_VERSION_TLS1_2;
2086             }
2087 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
2088             else if (strcmp(q, "tls13") == 0) {
2089                 opt.min_version = MBEDTLS_SSL_VERSION_TLS1_3;
2090             }
2091 #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
2092             else {
2093                 goto usage;
2094             }
2095         } else if (strcmp(p, "max_version") == 0) {
2096             if (strcmp(q, "tls12") == 0 ||
2097                 strcmp(q, "dtls12") == 0) {
2098                 opt.max_version = MBEDTLS_SSL_VERSION_TLS1_2;
2099             }
2100 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
2101             else if (strcmp(q, "tls13") == 0) {
2102                 opt.max_version = MBEDTLS_SSL_VERSION_TLS1_3;
2103             }
2104 #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
2105             else {
2106                 goto usage;
2107             }
2108         } else if (strcmp(p, "allow_sha1") == 0) {
2109             switch (atoi(q)) {
2110                 case 0:     opt.allow_sha1 = 0;   break;
2111                 case 1:     opt.allow_sha1 = 1;    break;
2112                 default:    goto usage;
2113             }
2114         } else if (strcmp(p, "force_version") == 0) {
2115             if (strcmp(q, "tls12") == 0) {
2116                 opt.min_version = MBEDTLS_SSL_VERSION_TLS1_2;
2117                 opt.max_version = MBEDTLS_SSL_VERSION_TLS1_2;
2118             } else if (strcmp(q, "dtls12") == 0) {
2119                 opt.min_version = MBEDTLS_SSL_VERSION_TLS1_2;
2120                 opt.max_version = MBEDTLS_SSL_VERSION_TLS1_2;
2121                 opt.transport = MBEDTLS_SSL_TRANSPORT_DATAGRAM;
2122             }
2123 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
2124             else if (strcmp(q, "tls13") == 0) {
2125                 opt.min_version = MBEDTLS_SSL_VERSION_TLS1_3;
2126                 opt.max_version = MBEDTLS_SSL_VERSION_TLS1_3;
2127             }
2128 #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
2129             else {
2130                 goto usage;
2131             }
2132         } else if (strcmp(p, "auth_mode") == 0) {
2133             if ((opt.auth_mode = get_auth_mode(q)) < 0) {
2134                 goto usage;
2135             }
2136         } else if (strcmp(p, "cert_req_ca_list") == 0) {
2137             opt.cert_req_ca_list = atoi(q);
2138             if (opt.cert_req_ca_list < 0 || opt.cert_req_ca_list > 3) {
2139                 goto usage;
2140             }
2141             if (opt.cert_req_ca_list > 1) {
2142                 opt.cert_req_dn_hint = opt.cert_req_ca_list;
2143                 opt.cert_req_ca_list = MBEDTLS_SSL_CERT_REQ_CA_LIST_ENABLED;
2144             }
2145         } else if (strcmp(p, "max_frag_len") == 0) {
2146             if (strcmp(q, "512") == 0) {
2147                 opt.mfl_code = MBEDTLS_SSL_MAX_FRAG_LEN_512;
2148             } else if (strcmp(q, "1024") == 0) {
2149                 opt.mfl_code = MBEDTLS_SSL_MAX_FRAG_LEN_1024;
2150             } else if (strcmp(q, "2048") == 0) {
2151                 opt.mfl_code = MBEDTLS_SSL_MAX_FRAG_LEN_2048;
2152             } else if (strcmp(q, "4096") == 0) {
2153                 opt.mfl_code = MBEDTLS_SSL_MAX_FRAG_LEN_4096;
2154             } else {
2155                 goto usage;
2156             }
2157         } else if (strcmp(p, "alpn") == 0) {
2158             opt.alpn_string = q;
2159         } else if (strcmp(p, "trunc_hmac") == 0) {
2160             switch (atoi(q)) {
2161                 case 0: opt.trunc_hmac = MBEDTLS_SSL_TRUNC_HMAC_DISABLED; break;
2162                 case 1: opt.trunc_hmac = MBEDTLS_SSL_TRUNC_HMAC_ENABLED; break;
2163                 default: goto usage;
2164             }
2165         } else if (strcmp(p, "extended_ms") == 0) {
2166             switch (atoi(q)) {
2167                 case 0:
2168                     opt.extended_ms = MBEDTLS_SSL_EXTENDED_MS_DISABLED;
2169                     break;
2170                 case 1:
2171                     opt.extended_ms = MBEDTLS_SSL_EXTENDED_MS_ENABLED;
2172                     break;
2173                 default: goto usage;
2174             }
2175         } else if (strcmp(p, "etm") == 0) {
2176             switch (atoi(q)) {
2177                 case 0: opt.etm = MBEDTLS_SSL_ETM_DISABLED; break;
2178                 case 1: opt.etm = MBEDTLS_SSL_ETM_ENABLED; break;
2179                 default: goto usage;
2180             }
2181         } else if (strcmp(p, "tickets") == 0) {
2182             opt.tickets = atoi(q);
2183             if (opt.tickets < 0) {
2184                 goto usage;
2185             }
2186         } else if (strcmp(p, "dummy_ticket") == 0) {
2187             opt.dummy_ticket = atoi(q);
2188             if (opt.dummy_ticket < 0) {
2189                 goto usage;
2190             }
2191         } else if (strcmp(p, "ticket_rotate") == 0) {
2192             opt.ticket_rotate = atoi(q);
2193             if (opt.ticket_rotate < 0 || opt.ticket_rotate > 1) {
2194                 goto usage;
2195             }
2196         } else if (strcmp(p, "ticket_timeout") == 0) {
2197             opt.ticket_timeout = atoi(q);
2198             if (opt.ticket_timeout < 0) {
2199                 goto usage;
2200             }
2201         } else if (strcmp(p, "ticket_aead") == 0) {
2202             opt.ticket_aead = parse_cipher(q);
2203 
2204             if (opt.ticket_aead == MBEDTLS_CIPHER_NONE) {
2205                 goto usage;
2206             }
2207         } else if (strcmp(p, "cache_max") == 0) {
2208             opt.cache_max = atoi(q);
2209             if (opt.cache_max < 0) {
2210                 goto usage;
2211             }
2212         }
2213 #if defined(MBEDTLS_HAVE_TIME)
2214         else if (strcmp(p, "cache_timeout") == 0) {
2215             opt.cache_timeout = atoi(q);
2216             if (opt.cache_timeout < 0) {
2217                 goto usage;
2218             }
2219         }
2220 #endif
2221         else if (strcmp(p, "cache_remove") == 0) {
2222             opt.cache_remove = atoi(q);
2223             if (opt.cache_remove < 0 || opt.cache_remove > 1) {
2224                 goto usage;
2225             }
2226         } else if (strcmp(p, "cookies") == 0) {
2227             opt.cookies = atoi(q);
2228             if (opt.cookies < -1 || opt.cookies > 1) {
2229                 goto usage;
2230             }
2231         } else if (strcmp(p, "anti_replay") == 0) {
2232             opt.anti_replay = atoi(q);
2233             if (opt.anti_replay < 0 || opt.anti_replay > 1) {
2234                 goto usage;
2235             }
2236         } else if (strcmp(p, "badmac_limit") == 0) {
2237             opt.badmac_limit = atoi(q);
2238             if (opt.badmac_limit < 0) {
2239                 goto usage;
2240             }
2241         } else if (strcmp(p, "hs_timeout") == 0) {
2242             if ((p = strchr(q, '-')) == NULL) {
2243                 goto usage;
2244             }
2245             *p++ = '\0';
2246             opt.hs_to_min = atoi(q);
2247             opt.hs_to_max = atoi(p);
2248             if (opt.hs_to_min == 0 || opt.hs_to_max < opt.hs_to_min) {
2249                 goto usage;
2250             }
2251         } else if (strcmp(p, "mtu") == 0) {
2252             opt.dtls_mtu = atoi(q);
2253             if (opt.dtls_mtu < 0) {
2254                 goto usage;
2255             }
2256         } else if (strcmp(p, "dgram_packing") == 0) {
2257             opt.dgram_packing = atoi(q);
2258             if (opt.dgram_packing != 0 &&
2259                 opt.dgram_packing != 1) {
2260                 goto usage;
2261             }
2262         } else if (strcmp(p, "sni") == 0) {
2263             opt.sni = q;
2264         } else if (strcmp(p, "query_config") == 0) {
2265             opt.query_config_mode = 1;
2266             query_config_ret = query_config(q);
2267             goto exit;
2268         } else if (strcmp(p, "serialize") == 0) {
2269             opt.serialize = atoi(q);
2270             if (opt.serialize < 0 || opt.serialize > 2) {
2271                 goto usage;
2272             }
2273         } else if (strcmp(p, "context_file") == 0) {
2274             opt.context_file = q;
2275         } else if (strcmp(p, "eap_tls") == 0) {
2276             opt.eap_tls = atoi(q);
2277             if (opt.eap_tls < 0 || opt.eap_tls > 1) {
2278                 goto usage;
2279             }
2280         } else if (strcmp(p, "reproducible") == 0) {
2281             opt.reproducible = 1;
2282         } else if (strcmp(p, "nss_keylog") == 0) {
2283             opt.nss_keylog = atoi(q);
2284             if (opt.nss_keylog < 0 || opt.nss_keylog > 1) {
2285                 goto usage;
2286             }
2287         } else if (strcmp(p, "nss_keylog_file") == 0) {
2288             opt.nss_keylog_file = q;
2289         } else if (strcmp(p, "use_srtp") == 0) {
2290             opt.use_srtp = atoi(q);
2291         } else if (strcmp(p, "srtp_force_profile") == 0) {
2292             opt.force_srtp_profile = atoi(q);
2293         } else if (strcmp(p, "support_mki") == 0) {
2294             opt.support_mki = atoi(q);
2295         } else if (strcmp(p, "key_opaque_algs") == 0) {
2296             if (key_opaque_alg_parse(q, &opt.key1_opaque_alg1,
2297                                      &opt.key1_opaque_alg2) != 0) {
2298                 goto usage;
2299             }
2300         } else if (strcmp(p, "key_opaque_algs2") == 0) {
2301             if (key_opaque_alg_parse(q, &opt.key2_opaque_alg1,
2302                                      &opt.key2_opaque_alg2) != 0) {
2303                 goto usage;
2304             }
2305         } else {
2306             /* This signals that the problem is with p not q */
2307             q = NULL;
2308             goto usage;
2309         }
2310     }
2311     /* This signals that any further erorrs are not with a single option */
2312     p = q = NULL;
2313 
2314     if (opt.nss_keylog != 0 && opt.eap_tls != 0) {
2315         mbedtls_printf("Error: eap_tls and nss_keylog options cannot be used together.\n");
2316         goto usage;
2317     }
2318 
2319     /* Event-driven IO is incompatible with the above custom
2320      * receive and send functions, as the polling builds on
2321      * refers to the underlying net_context. */
2322     if (opt.event == 1 && opt.nbio != 1) {
2323         mbedtls_printf("Warning: event-driven IO mandates nbio=1 - overwrite\n");
2324         opt.nbio = 1;
2325     }
2326 
2327 #if defined(MBEDTLS_DEBUG_C)
2328     mbedtls_debug_set_threshold(opt.debug_level);
2329 #endif
2330 
2331     /* buf will alternatively contain the input read from the client and the
2332      * response that's about to be sent, plus a null byte in each case. */
2333     size_t buf_content_size = opt.buffer_size;
2334     /* The default response contains the ciphersuite name. Leave enough
2335      * room for that plus some margin. */
2336     if (buf_content_size < strlen(HTTP_RESPONSE) + 80) {
2337         buf_content_size = strlen(HTTP_RESPONSE) + 80;
2338     }
2339     if (opt.response_size != DFL_RESPONSE_SIZE &&
2340         buf_content_size < (size_t) opt.response_size) {
2341         buf_content_size = opt.response_size;
2342     }
2343     buf = mbedtls_calloc(1, buf_content_size + 1);
2344     if (buf == NULL) {
2345         mbedtls_printf("Could not allocate %lu bytes\n",
2346                        (unsigned long) buf_content_size + 1);
2347         ret = 3;
2348         goto exit;
2349     }
2350 
2351 #if defined(MBEDTLS_USE_PSA_CRYPTO)
2352     if (opt.psk_opaque != 0) {
2353         if (strlen(opt.psk) == 0) {
2354             mbedtls_printf("psk_opaque set but no psk to be imported specified.\n");
2355             ret = 2;
2356             goto usage;
2357         }
2358 
2359         if (opt.force_ciphersuite[0] <= 0) {
2360             mbedtls_printf(
2361                 "opaque PSKs are only supported in conjunction with forcing TLS 1.2 and a PSK-only ciphersuite through the 'force_ciphersuite' option.\n");
2362             ret = 2;
2363             goto usage;
2364         }
2365     }
2366 
2367     if (opt.psk_list_opaque != 0) {
2368         if (opt.psk_list == NULL) {
2369             mbedtls_printf("psk_slot set but no psk to be imported specified.\n");
2370             ret = 2;
2371             goto usage;
2372         }
2373 
2374         if (opt.force_ciphersuite[0] <= 0) {
2375             mbedtls_printf(
2376                 "opaque PSKs are only supported in conjunction with forcing TLS 1.2 and a PSK-only ciphersuite through the 'force_ciphersuite' option.\n");
2377             ret = 2;
2378             goto usage;
2379         }
2380     }
2381 #endif /* MBEDTLS_USE_PSA_CRYPTO */
2382 
2383     if (opt.force_ciphersuite[0] > 0) {
2384         const mbedtls_ssl_ciphersuite_t *ciphersuite_info;
2385         ciphersuite_info =
2386             mbedtls_ssl_ciphersuite_from_id(opt.force_ciphersuite[0]);
2387 
2388         if (opt.max_version != -1 &&
2389             ciphersuite_info->min_tls_version > opt.max_version) {
2390             mbedtls_printf("forced ciphersuite not allowed with this protocol version\n");
2391             ret = 2;
2392             goto usage;
2393         }
2394         if (opt.min_version != -1 &&
2395             ciphersuite_info->max_tls_version < opt.min_version) {
2396             mbedtls_printf("forced ciphersuite not allowed with this protocol version\n");
2397             ret = 2;
2398             goto usage;
2399         }
2400 
2401         /* If we select a version that's not supported by
2402          * this suite, then there will be no common ciphersuite... */
2403         if (opt.max_version == -1 ||
2404             opt.max_version > ciphersuite_info->max_tls_version) {
2405             opt.max_version = ciphersuite_info->max_tls_version;
2406         }
2407         if (opt.min_version < ciphersuite_info->min_tls_version) {
2408             opt.min_version = ciphersuite_info->min_tls_version;
2409         }
2410 
2411 #if defined(MBEDTLS_USE_PSA_CRYPTO)
2412 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED)
2413         if (opt.psk_opaque != 0 || opt.psk_list_opaque != 0) {
2414             /* Determine KDF algorithm the opaque PSK will be used in. */
2415 #if defined(MBEDTLS_MD_CAN_SHA384)
2416             if (ciphersuite_info->mac == MBEDTLS_MD_SHA384) {
2417                 alg = PSA_ALG_TLS12_PSK_TO_MS(PSA_ALG_SHA_384);
2418             } else
2419 #endif /* MBEDTLS_MD_CAN_SHA384 */
2420             alg = PSA_ALG_TLS12_PSK_TO_MS(PSA_ALG_SHA_256);
2421         }
2422 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED */
2423 #endif /* MBEDTLS_USE_PSA_CRYPTO */
2424     }
2425 
2426 #if defined(MBEDTLS_SSL_DTLS_CONNECTION_ID)
2427     if (mbedtls_test_unhexify(cid, sizeof(cid),
2428                               opt.cid_val, &cid_len) != 0) {
2429         mbedtls_printf("CID not valid hex\n");
2430         goto exit;
2431     }
2432 
2433     /* Keep CID settings for renegotiation unless
2434      * specified otherwise. */
2435     if (opt.cid_enabled_renego == DFL_CID_ENABLED_RENEGO) {
2436         opt.cid_enabled_renego = opt.cid_enabled;
2437     }
2438     if (opt.cid_val_renego == DFL_CID_VALUE_RENEGO) {
2439         opt.cid_val_renego = opt.cid_val;
2440     }
2441 
2442     if (mbedtls_test_unhexify(cid_renego, sizeof(cid_renego),
2443                               opt.cid_val_renego, &cid_renego_len) != 0) {
2444         mbedtls_printf("CID not valid hex\n");
2445         goto exit;
2446     }
2447 #endif /* MBEDTLS_SSL_DTLS_CONNECTION_ID */
2448 
2449 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED)
2450     /*
2451      * Unhexify the pre-shared key and parse the list if any given
2452      */
2453     if (mbedtls_test_unhexify(psk, sizeof(psk),
2454                               opt.psk, &psk_len) != 0) {
2455         mbedtls_printf("pre-shared key not valid hex\n");
2456         goto exit;
2457     }
2458 
2459     if (opt.psk_list != NULL) {
2460         if ((psk_info = psk_parse(opt.psk_list)) == NULL) {
2461             mbedtls_printf("psk_list invalid");
2462             goto exit;
2463         }
2464     }
2465 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED */
2466 
2467     if (opt.groups != NULL) {
2468         if (parse_groups(opt.groups, group_list, GROUP_LIST_SIZE) != 0) {
2469             goto exit;
2470         }
2471     }
2472 
2473 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
2474     if (opt.sig_algs != NULL) {
2475         p = (char *) opt.sig_algs;
2476         i = 0;
2477 
2478         /* Leave room for a final MBEDTLS_TLS1_3_SIG_NONE in signature algorithm list (sig_alg_list). */
2479         while (i < SIG_ALG_LIST_SIZE - 1 && *p != '\0') {
2480             q = p;
2481 
2482             /* Terminate the current string */
2483             while (*p != ',' && *p != '\0') {
2484                 p++;
2485             }
2486             if (*p == ',') {
2487                 *p++ = '\0';
2488             }
2489 
2490             if (strcmp(q, "rsa_pkcs1_sha256") == 0) {
2491                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PKCS1_SHA256;
2492             } else if (strcmp(q, "rsa_pkcs1_sha384") == 0) {
2493                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PKCS1_SHA384;
2494             } else if (strcmp(q, "rsa_pkcs1_sha512") == 0) {
2495                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PKCS1_SHA512;
2496             } else if (strcmp(q, "ecdsa_secp256r1_sha256") == 0) {
2497                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_ECDSA_SECP256R1_SHA256;
2498             } else if (strcmp(q, "ecdsa_secp384r1_sha384") == 0) {
2499                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_ECDSA_SECP384R1_SHA384;
2500             } else if (strcmp(q, "ecdsa_secp521r1_sha512") == 0) {
2501                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_ECDSA_SECP521R1_SHA512;
2502             } else if (strcmp(q, "rsa_pss_rsae_sha256") == 0) {
2503                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PSS_RSAE_SHA256;
2504             } else if (strcmp(q, "rsa_pss_rsae_sha384") == 0) {
2505                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PSS_RSAE_SHA384;
2506             } else if (strcmp(q, "rsa_pss_rsae_sha512") == 0) {
2507                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PSS_RSAE_SHA512;
2508             } else if (strcmp(q, "ed25519") == 0) {
2509                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_ED25519;
2510             } else if (strcmp(q, "ed448") == 0) {
2511                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_ED448;
2512             } else if (strcmp(q, "rsa_pss_pss_sha256") == 0) {
2513                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PSS_PSS_SHA256;
2514             } else if (strcmp(q, "rsa_pss_pss_sha384") == 0) {
2515                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PSS_PSS_SHA384;
2516             } else if (strcmp(q, "rsa_pss_pss_sha512") == 0) {
2517                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PSS_PSS_SHA512;
2518             } else if (strcmp(q, "rsa_pkcs1_sha1") == 0) {
2519                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_RSA_PKCS1_SHA1;
2520             } else if (strcmp(q, "ecdsa_sha1") == 0) {
2521                 sig_alg_list[i++] = MBEDTLS_TLS1_3_SIG_ECDSA_SHA1;
2522             } else {
2523                 ret = -1;
2524                 mbedtls_printf("unknown signature algorithm \"%s\"\n", q);
2525                 mbedtls_print_supported_sig_algs();
2526                 goto exit;
2527             }
2528         }
2529 
2530         if (i == (SIG_ALG_LIST_SIZE - 1) && *p != '\0') {
2531             mbedtls_printf("signature algorithm list too long, maximum %d",
2532                            SIG_ALG_LIST_SIZE - 1);
2533             goto exit;
2534         }
2535 
2536         sig_alg_list[i] = MBEDTLS_TLS1_3_SIG_NONE;
2537     }
2538 #endif
2539 
2540 #if defined(MBEDTLS_SSL_ALPN)
2541     if (opt.alpn_string != NULL) {
2542         p = (char *) opt.alpn_string;
2543         i = 0;
2544 
2545         /* Leave room for a final NULL in alpn_list */
2546         while (i < ALPN_LIST_SIZE - 1 && *p != '\0') {
2547             alpn_list[i++] = p;
2548 
2549             /* Terminate the current string and move on to next one */
2550             while (*p != ',' && *p != '\0') {
2551                 p++;
2552             }
2553             if (*p == ',') {
2554                 *p++ = '\0';
2555             }
2556         }
2557     }
2558 #endif /* MBEDTLS_SSL_ALPN */
2559 
2560     mbedtls_printf("build version: %s (build %d)\n",
2561                    MBEDTLS_VERSION_STRING_FULL, MBEDTLS_VERSION_NUMBER);
2562 
2563     /*
2564      * 0. Initialize the RNG and the session data
2565      */
2566     mbedtls_printf("\n  . Seeding the random number generator...");
2567     fflush(stdout);
2568 
2569     ret = rng_seed(&rng, opt.reproducible, pers);
2570     if (ret != 0) {
2571         goto exit;
2572     }
2573     mbedtls_printf(" ok\n");
2574 
2575 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
2576     /*
2577      * 1.1. Load the trusted CA
2578      */
2579     mbedtls_printf("  . Loading the CA root certificate ...");
2580     fflush(stdout);
2581 
2582     if (strcmp(opt.ca_path, "none") == 0 ||
2583         strcmp(opt.ca_file, "none") == 0) {
2584         ret = 0;
2585     } else
2586 #if defined(MBEDTLS_FS_IO)
2587     if (strlen(opt.ca_path)) {
2588         ret = mbedtls_x509_crt_parse_path(&cacert, opt.ca_path);
2589     } else if (strlen(opt.ca_file)) {
2590         ret = mbedtls_x509_crt_parse_file(&cacert, opt.ca_file);
2591     } else
2592 #endif
2593     {
2594 #if defined(MBEDTLS_PEM_PARSE_C)
2595         for (i = 0; mbedtls_test_cas[i] != NULL; i++) {
2596             ret = mbedtls_x509_crt_parse(&cacert,
2597                                          (const unsigned char *) mbedtls_test_cas[i],
2598                                          mbedtls_test_cas_len[i]);
2599             if (ret != 0) {
2600                 break;
2601             }
2602         }
2603 #endif /* MBEDTLS_PEM_PARSE_C */
2604         if (ret == 0) {
2605             for (i = 0; mbedtls_test_cas_der[i] != NULL; i++) {
2606                 ret = mbedtls_x509_crt_parse_der(&cacert,
2607                                                  (const unsigned char *) mbedtls_test_cas_der[i],
2608                                                  mbedtls_test_cas_der_len[i]);
2609                 if (ret != 0) {
2610                     break;
2611                 }
2612             }
2613         }
2614     }
2615     if (ret < 0) {
2616         mbedtls_printf(" failed\n  !  mbedtls_x509_crt_parse returned -0x%x\n\n",
2617                        (unsigned int) -ret);
2618         goto exit;
2619     }
2620 
2621     mbedtls_printf(" ok (%d skipped)\n", ret);
2622 
2623     /*
2624      * 1.2. Load own certificate and private key
2625      */
2626     mbedtls_printf("  . Loading the server cert. and key...");
2627     fflush(stdout);
2628 
2629 #if defined(MBEDTLS_FS_IO)
2630     if (strlen(opt.crt_file) && strcmp(opt.crt_file, "none") != 0) {
2631         key_cert_init++;
2632         if ((ret = mbedtls_x509_crt_parse_file(&srvcert, opt.crt_file)) != 0) {
2633             mbedtls_printf(" failed\n  !  mbedtls_x509_crt_parse_file returned -0x%x\n\n",
2634                            (unsigned int) -ret);
2635             goto exit;
2636         }
2637     }
2638     if (strlen(opt.key_file) && strcmp(opt.key_file, "none") != 0) {
2639         key_cert_init++;
2640         if ((ret = mbedtls_pk_parse_keyfile(&pkey, opt.key_file,
2641                                             opt.key_pwd, rng_get, &rng)) != 0) {
2642             mbedtls_printf(" failed\n  !  mbedtls_pk_parse_keyfile returned -0x%x\n\n",
2643                            (unsigned int) -ret);
2644             goto exit;
2645         }
2646     }
2647     if (key_cert_init == 1) {
2648         mbedtls_printf(" failed\n  !  crt_file without key_file or vice-versa\n\n");
2649         goto exit;
2650     }
2651 
2652     if (strlen(opt.crt_file2) && strcmp(opt.crt_file2, "none") != 0) {
2653         key_cert_init2++;
2654         if ((ret = mbedtls_x509_crt_parse_file(&srvcert2, opt.crt_file2)) != 0) {
2655             mbedtls_printf(" failed\n  !  mbedtls_x509_crt_parse_file(2) returned -0x%x\n\n",
2656                            (unsigned int) -ret);
2657             goto exit;
2658         }
2659     }
2660     if (strlen(opt.key_file2) && strcmp(opt.key_file2, "none") != 0) {
2661         key_cert_init2++;
2662         if ((ret = mbedtls_pk_parse_keyfile(&pkey2, opt.key_file2,
2663                                             opt.key_pwd2, rng_get, &rng)) != 0) {
2664             mbedtls_printf(" failed\n  !  mbedtls_pk_parse_keyfile(2) returned -0x%x\n\n",
2665                            (unsigned int) -ret);
2666             goto exit;
2667         }
2668     }
2669     if (key_cert_init2 == 1) {
2670         mbedtls_printf(" failed\n  !  crt_file2 without key_file2 or vice-versa\n\n");
2671         goto exit;
2672     }
2673 #endif
2674     if (key_cert_init == 0 &&
2675         strcmp(opt.crt_file, "none") != 0 &&
2676         strcmp(opt.key_file, "none") != 0 &&
2677         key_cert_init2 == 0 &&
2678         strcmp(opt.crt_file2, "none") != 0 &&
2679         strcmp(opt.key_file2, "none") != 0) {
2680 #if defined(MBEDTLS_RSA_C)
2681         if ((ret = mbedtls_x509_crt_parse(&srvcert,
2682                                           (const unsigned char *) mbedtls_test_srv_crt_rsa,
2683                                           mbedtls_test_srv_crt_rsa_len)) != 0) {
2684             mbedtls_printf(" failed\n  !  mbedtls_x509_crt_parse returned -0x%x\n\n",
2685                            (unsigned int) -ret);
2686             goto exit;
2687         }
2688         if ((ret = mbedtls_pk_parse_key(&pkey,
2689                                         (const unsigned char *) mbedtls_test_srv_key_rsa,
2690                                         mbedtls_test_srv_key_rsa_len, NULL, 0,
2691                                         rng_get, &rng)) != 0) {
2692             mbedtls_printf(" failed\n  !  mbedtls_pk_parse_key returned -0x%x\n\n",
2693                            (unsigned int) -ret);
2694             goto exit;
2695         }
2696         key_cert_init = 2;
2697 #endif /* MBEDTLS_RSA_C */
2698 #if defined(MBEDTLS_PK_CAN_ECDSA_SOME)
2699         if ((ret = mbedtls_x509_crt_parse(&srvcert2,
2700                                           (const unsigned char *) mbedtls_test_srv_crt_ec,
2701                                           mbedtls_test_srv_crt_ec_len)) != 0) {
2702             mbedtls_printf(" failed\n  !  x509_crt_parse2 returned -0x%x\n\n",
2703                            (unsigned int) -ret);
2704             goto exit;
2705         }
2706         if ((ret = mbedtls_pk_parse_key(&pkey2,
2707                                         (const unsigned char *) mbedtls_test_srv_key_ec,
2708                                         mbedtls_test_srv_key_ec_len, NULL, 0,
2709                                         rng_get, &rng)) != 0) {
2710             mbedtls_printf(" failed\n  !  pk_parse_key2 returned -0x%x\n\n",
2711                            (unsigned int) -ret);
2712             goto exit;
2713         }
2714         key_cert_init2 = 2;
2715 #endif /* MBEDTLS_PK_CAN_ECDSA_SOME */
2716     }
2717 
2718 #if defined(MBEDTLS_USE_PSA_CRYPTO)
2719     if (opt.key_opaque != 0) {
2720         psa_algorithm_t psa_alg, psa_alg2 = PSA_ALG_NONE;
2721         psa_key_usage_t psa_usage = 0;
2722 
2723         if (key_opaque_set_alg_usage(opt.key1_opaque_alg1,
2724                                      opt.key1_opaque_alg2,
2725                                      &psa_alg, &psa_alg2,
2726                                      &psa_usage,
2727                                      mbedtls_pk_get_type(&pkey)) == 0) {
2728             ret = pk_wrap_as_opaque(&pkey, psa_alg, psa_alg2, psa_usage, &key_slot);
2729             if (ret != 0) {
2730                 mbedtls_printf(" failed\n  !  "
2731                                "pk_wrap_as_opaque returned -0x%x\n\n",
2732                                (unsigned int)  -ret);
2733                 goto exit;
2734             }
2735         }
2736 
2737         psa_alg = PSA_ALG_NONE; psa_alg2 = PSA_ALG_NONE;
2738         psa_usage = 0;
2739 
2740         if (key_opaque_set_alg_usage(opt.key2_opaque_alg1,
2741                                      opt.key2_opaque_alg2,
2742                                      &psa_alg, &psa_alg2,
2743                                      &psa_usage,
2744                                      mbedtls_pk_get_type(&pkey2)) == 0) {
2745             ret = pk_wrap_as_opaque(&pkey2, psa_alg, psa_alg2, psa_usage, &key_slot2);
2746             if (ret != 0) {
2747                 mbedtls_printf(" failed\n  !  "
2748                                "mbedtls_pk_get_psa_attributes returned -0x%x\n\n",
2749                                (unsigned int)  -ret);
2750                 goto exit;
2751             }
2752         }
2753     }
2754 #endif /* MBEDTLS_USE_PSA_CRYPTO */
2755 
2756     mbedtls_printf(" ok (key types: %s, %s)\n",
2757                    key_cert_init ? mbedtls_pk_get_name(&pkey) : "none",
2758                    key_cert_init2 ? mbedtls_pk_get_name(&pkey2) : "none");
2759 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED */
2760 
2761 #if defined(MBEDTLS_DHM_C) && defined(MBEDTLS_FS_IO)
2762     if (opt.dhm_file != NULL) {
2763         mbedtls_printf("  . Loading DHM parameters...");
2764         fflush(stdout);
2765 
2766         if ((ret = mbedtls_dhm_parse_dhmfile(&dhm, opt.dhm_file)) != 0) {
2767             mbedtls_printf(" failed\n  ! mbedtls_dhm_parse_dhmfile returned -0x%04X\n\n",
2768                            (unsigned int) -ret);
2769             goto exit;
2770         }
2771 
2772         mbedtls_printf(" ok\n");
2773     }
2774 #endif
2775 
2776 #if defined(SNI_OPTION)
2777     if (opt.sni != NULL) {
2778         mbedtls_printf("  . Setting up SNI information...");
2779         fflush(stdout);
2780 
2781         if ((sni_info = sni_parse(opt.sni)) == NULL) {
2782             mbedtls_printf(" failed\n");
2783             goto exit;
2784         }
2785 
2786         mbedtls_printf(" ok\n");
2787     }
2788 #endif /* SNI_OPTION */
2789 
2790     /*
2791      * 2. Setup stuff
2792      */
2793     mbedtls_printf("  . Setting up the SSL/TLS structure...");
2794     fflush(stdout);
2795 
2796     if ((ret = mbedtls_ssl_config_defaults(&conf,
2797                                            MBEDTLS_SSL_IS_SERVER,
2798                                            opt.transport,
2799                                            MBEDTLS_SSL_PRESET_DEFAULT)) != 0) {
2800         mbedtls_printf(" failed\n  ! mbedtls_ssl_config_defaults returned -0x%x\n\n",
2801                        (unsigned int) -ret);
2802         goto exit;
2803     }
2804 
2805 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
2806     /* The default algorithms profile disables SHA-1, but our tests still
2807        rely on it heavily. Hence we allow it here. A real-world server
2808        should use the default profile unless there is a good reason not to. */
2809     if (opt.allow_sha1 > 0) {
2810         crt_profile_for_test.allowed_mds |= MBEDTLS_X509_ID_FLAG(MBEDTLS_MD_SHA1);
2811         mbedtls_ssl_conf_cert_profile(&conf, &crt_profile_for_test);
2812         mbedtls_ssl_conf_sig_algs(&conf, ssl_sig_algs_for_test);
2813     }
2814 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED */
2815 
2816     if (opt.auth_mode != DFL_AUTH_MODE) {
2817         mbedtls_ssl_conf_authmode(&conf, opt.auth_mode);
2818     }
2819 
2820     if (opt.cert_req_ca_list != DFL_CERT_REQ_CA_LIST) {
2821         mbedtls_ssl_conf_cert_req_ca_list(&conf, opt.cert_req_ca_list);
2822     }
2823 
2824 #if defined(MBEDTLS_SSL_EARLY_DATA)
2825     if (opt.early_data != DFL_EARLY_DATA) {
2826         mbedtls_ssl_conf_early_data(&conf, opt.early_data);
2827     }
2828     if (opt.max_early_data_size != DFL_MAX_EARLY_DATA_SIZE) {
2829         mbedtls_ssl_conf_max_early_data_size(
2830             &conf, opt.max_early_data_size);
2831     }
2832 #endif /* MBEDTLS_SSL_EARLY_DATA */
2833 
2834 #if defined(MBEDTLS_KEY_EXCHANGE_CERT_REQ_ALLOWED_ENABLED)
2835     /* exercise setting DN hints for server certificate request
2836      * (Intended for use where the client cert expected has been signed by
2837      *  a specific CA which is an intermediate in a CA chain, not the root) */
2838     if (opt.cert_req_dn_hint == 2 && key_cert_init2) {
2839         mbedtls_ssl_conf_dn_hints(&conf, &srvcert2);
2840     }
2841 #endif
2842 
2843 #if defined(MBEDTLS_SSL_PROTO_DTLS)
2844     if (opt.hs_to_min != DFL_HS_TO_MIN || opt.hs_to_max != DFL_HS_TO_MAX) {
2845         mbedtls_ssl_conf_handshake_timeout(&conf, opt.hs_to_min, opt.hs_to_max);
2846     }
2847 
2848     if (opt.dgram_packing != DFL_DGRAM_PACKING) {
2849         mbedtls_ssl_set_datagram_packing(&ssl, opt.dgram_packing);
2850     }
2851 #endif /* MBEDTLS_SSL_PROTO_DTLS */
2852 
2853 #if defined(MBEDTLS_SSL_MAX_FRAGMENT_LENGTH)
2854     if ((ret = mbedtls_ssl_conf_max_frag_len(&conf, opt.mfl_code)) != 0) {
2855         mbedtls_printf(" failed\n  ! mbedtls_ssl_conf_max_frag_len returned %d\n\n", ret);
2856         goto exit;
2857     }
2858 #endif
2859 
2860 #if defined(MBEDTLS_SSL_DTLS_CONNECTION_ID)
2861     if (opt.cid_enabled == 1 || opt.cid_enabled_renego == 1) {
2862         if (opt.cid_enabled == 1        &&
2863             opt.cid_enabled_renego == 1 &&
2864             cid_len != cid_renego_len) {
2865             mbedtls_printf("CID length must not change during renegotiation\n");
2866             goto usage;
2867         }
2868 
2869         if (opt.cid_enabled == 1) {
2870             ret = mbedtls_ssl_conf_cid(&conf, cid_len,
2871                                        MBEDTLS_SSL_UNEXPECTED_CID_IGNORE);
2872         } else {
2873             ret = mbedtls_ssl_conf_cid(&conf, cid_renego_len,
2874                                        MBEDTLS_SSL_UNEXPECTED_CID_IGNORE);
2875         }
2876 
2877         if (ret != 0) {
2878             mbedtls_printf(" failed\n  ! mbedtls_ssl_conf_cid_len returned -%#04x\n\n",
2879                            (unsigned int) -ret);
2880             goto exit;
2881         }
2882     }
2883 #endif /* MBEDTLS_SSL_DTLS_CONNECTION_ID */
2884 
2885 #if defined(MBEDTLS_SSL_DTLS_SRTP)
2886     const mbedtls_ssl_srtp_profile forced_profile[] =
2887     { opt.force_srtp_profile, MBEDTLS_TLS_SRTP_UNSET };
2888     if (opt.use_srtp == 1) {
2889         if (opt.force_srtp_profile != 0) {
2890             ret = mbedtls_ssl_conf_dtls_srtp_protection_profiles(&conf, forced_profile);
2891         } else {
2892             ret = mbedtls_ssl_conf_dtls_srtp_protection_profiles(&conf, default_profiles);
2893         }
2894 
2895         if (ret != 0) {
2896             mbedtls_printf(
2897                 " failed\n  ! mbedtls_ssl_conf_dtls_srtp_protection_profiles returned %d\n\n",
2898                 ret);
2899             goto exit;
2900         }
2901 
2902         mbedtls_ssl_conf_srtp_mki_value_supported(&conf,
2903                                                   opt.support_mki ?
2904                                                   MBEDTLS_SSL_DTLS_SRTP_MKI_SUPPORTED :
2905                                                   MBEDTLS_SSL_DTLS_SRTP_MKI_UNSUPPORTED);
2906 
2907     } else if (opt.force_srtp_profile != 0) {
2908         mbedtls_printf(" failed\n  ! must enable use_srtp to force srtp profile\n\n");
2909         goto exit;
2910     }
2911 #endif /* MBEDTLS_SSL_DTLS_SRTP */
2912 
2913 #if defined(MBEDTLS_SSL_EXTENDED_MASTER_SECRET)
2914     if (opt.extended_ms != DFL_EXTENDED_MS) {
2915         mbedtls_ssl_conf_extended_master_secret(&conf, opt.extended_ms);
2916     }
2917 #endif
2918 
2919 #if defined(MBEDTLS_SSL_ENCRYPT_THEN_MAC)
2920     if (opt.etm != DFL_ETM) {
2921         mbedtls_ssl_conf_encrypt_then_mac(&conf, opt.etm);
2922     }
2923 #endif
2924 
2925 #if defined(MBEDTLS_SSL_ALPN)
2926     if (opt.alpn_string != NULL) {
2927         if ((ret = mbedtls_ssl_conf_alpn_protocols(&conf, alpn_list)) != 0) {
2928             mbedtls_printf(" failed\n  ! mbedtls_ssl_conf_alpn_protocols returned %d\n\n", ret);
2929             goto exit;
2930         }
2931     }
2932 #endif
2933 
2934     if (opt.reproducible) {
2935 #if defined(MBEDTLS_HAVE_TIME)
2936 #if defined(MBEDTLS_PLATFORM_TIME_ALT)
2937         mbedtls_platform_set_time(dummy_constant_time);
2938 #else
2939         fprintf(stderr, "Warning: reproducible option used without constant time\n");
2940 #endif
2941 #endif  /* MBEDTLS_HAVE_TIME */
2942     }
2943     mbedtls_ssl_conf_rng(&conf, rng_get, &rng);
2944     mbedtls_ssl_conf_dbg(&conf, my_debug, stdout);
2945 
2946 #if defined(MBEDTLS_SSL_CACHE_C)
2947     if (opt.cache_max != -1) {
2948         mbedtls_ssl_cache_set_max_entries(&cache, opt.cache_max);
2949     }
2950 
2951 #if defined(MBEDTLS_HAVE_TIME)
2952     if (opt.cache_timeout != -1) {
2953         mbedtls_ssl_cache_set_timeout(&cache, opt.cache_timeout);
2954     }
2955 #endif
2956 
2957     mbedtls_ssl_conf_session_cache(&conf, &cache,
2958                                    mbedtls_ssl_cache_get,
2959                                    mbedtls_ssl_cache_set);
2960 #endif
2961 
2962 #if defined(MBEDTLS_SSL_SESSION_TICKETS) && defined(MBEDTLS_SSL_TICKET_C)
2963     if (opt.tickets != MBEDTLS_SSL_SESSION_TICKETS_DISABLED) {
2964 #if defined(MBEDTLS_HAVE_TIME)
2965         if (opt.dummy_ticket) {
2966             mbedtls_ssl_conf_session_tickets_cb(&conf,
2967                                                 dummy_ticket_write,
2968                                                 dummy_ticket_parse,
2969                                                 NULL);
2970         } else
2971 #endif /* MBEDTLS_HAVE_TIME */
2972         {
2973             if ((ret = mbedtls_ssl_ticket_setup(&ticket_ctx,
2974                                                 rng_get, &rng,
2975                                                 opt.ticket_aead,
2976                                                 opt.ticket_timeout)) != 0) {
2977                 mbedtls_printf(
2978                     " failed\n  ! mbedtls_ssl_ticket_setup returned %d\n\n",
2979                     ret);
2980                 goto exit;
2981             }
2982 
2983             mbedtls_ssl_conf_session_tickets_cb(&conf,
2984                                                 mbedtls_ssl_ticket_write,
2985                                                 mbedtls_ssl_ticket_parse,
2986                                                 &ticket_ctx);
2987         }
2988 
2989 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
2990         mbedtls_ssl_conf_new_session_tickets(&conf, opt.tickets);
2991 #endif
2992         /* exercise manual ticket rotation (not required for typical use)
2993          * (used for external synchronization of session ticket encryption keys)
2994          */
2995         if (opt.ticket_rotate) {
2996             unsigned char kbuf[MBEDTLS_SSL_TICKET_MAX_KEY_BYTES];
2997             unsigned char name[MBEDTLS_SSL_TICKET_KEY_NAME_BYTES];
2998             if ((ret = rng_get(&rng, name, sizeof(name))) != 0 ||
2999                 (ret = rng_get(&rng, kbuf, sizeof(kbuf))) != 0 ||
3000                 (ret = mbedtls_ssl_ticket_rotate(&ticket_ctx,
3001                                                  name, sizeof(name), kbuf, sizeof(kbuf),
3002                                                  opt.ticket_timeout)) != 0) {
3003                 mbedtls_printf(" failed\n  ! mbedtls_ssl_ticket_rotate returned %d\n\n", ret);
3004                 goto exit;
3005             }
3006         }
3007     }
3008 #endif
3009 
3010 #if defined(MBEDTLS_SSL_PROTO_DTLS)
3011     if (opt.transport == MBEDTLS_SSL_TRANSPORT_DATAGRAM) {
3012 #if defined(MBEDTLS_SSL_COOKIE_C)
3013         if (opt.cookies > 0) {
3014             if ((ret = mbedtls_ssl_cookie_setup(&cookie_ctx,
3015                                                 rng_get, &rng)) != 0) {
3016                 mbedtls_printf(" failed\n  ! mbedtls_ssl_cookie_setup returned %d\n\n", ret);
3017                 goto exit;
3018             }
3019 
3020             mbedtls_ssl_conf_dtls_cookies(&conf, mbedtls_ssl_cookie_write, mbedtls_ssl_cookie_check,
3021                                           &cookie_ctx);
3022         } else
3023 #endif /* MBEDTLS_SSL_COOKIE_C */
3024 #if defined(MBEDTLS_SSL_DTLS_HELLO_VERIFY)
3025         if (opt.cookies == 0) {
3026             mbedtls_ssl_conf_dtls_cookies(&conf, NULL, NULL, NULL);
3027         } else
3028 #endif /* MBEDTLS_SSL_DTLS_HELLO_VERIFY */
3029         {
3030             ; /* Nothing to do */
3031         }
3032 
3033 #if defined(MBEDTLS_SSL_DTLS_ANTI_REPLAY)
3034         if (opt.anti_replay != DFL_ANTI_REPLAY) {
3035             mbedtls_ssl_conf_dtls_anti_replay(&conf, opt.anti_replay);
3036         }
3037 #endif
3038 
3039         if (opt.badmac_limit != DFL_BADMAC_LIMIT) {
3040             mbedtls_ssl_conf_dtls_badmac_limit(&conf, opt.badmac_limit);
3041         }
3042     }
3043 #endif /* MBEDTLS_SSL_PROTO_DTLS */
3044 
3045     if (opt.force_ciphersuite[0] != DFL_FORCE_CIPHER) {
3046         mbedtls_ssl_conf_ciphersuites(&conf, opt.force_ciphersuite);
3047     }
3048 
3049 #if defined(MBEDTLS_SSL_PROTO_TLS1_3)
3050     mbedtls_ssl_conf_tls13_key_exchange_modes(&conf, opt.tls13_kex_modes);
3051 #endif /* MBEDTLS_SSL_PROTO_TLS1_3 */
3052 
3053     if (opt.allow_legacy != DFL_ALLOW_LEGACY) {
3054         mbedtls_ssl_conf_legacy_renegotiation(&conf, opt.allow_legacy);
3055     }
3056 #if defined(MBEDTLS_SSL_RENEGOTIATION)
3057     mbedtls_ssl_conf_renegotiation(&conf, opt.renegotiation);
3058 
3059     if (opt.renego_delay != DFL_RENEGO_DELAY) {
3060         mbedtls_ssl_conf_renegotiation_enforced(&conf, opt.renego_delay);
3061     }
3062 
3063     if (opt.renego_period != DFL_RENEGO_PERIOD) {
3064         PUT_UINT64_BE(renego_period, opt.renego_period, 0);
3065         mbedtls_ssl_conf_renegotiation_period(&conf, renego_period);
3066     }
3067 #endif
3068 
3069 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
3070     if (strcmp(opt.ca_path, "none") != 0 &&
3071         strcmp(opt.ca_file, "none") != 0) {
3072 #if defined(MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK)
3073         if (opt.ca_callback != 0) {
3074             mbedtls_ssl_conf_ca_cb(&conf, ca_callback, &cacert);
3075         } else
3076 #endif
3077         mbedtls_ssl_conf_ca_chain(&conf, &cacert, NULL);
3078     }
3079     if (key_cert_init) {
3080         mbedtls_pk_context *pk = &pkey;
3081 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
3082         if (opt.async_private_delay1 >= 0) {
3083             ret = ssl_async_set_key(&ssl_async_keys, &srvcert, pk, 0,
3084                                     opt.async_private_delay1);
3085             if (ret < 0) {
3086                 mbedtls_printf("  Test error: ssl_async_set_key failed (%d)\n",
3087                                ret);
3088                 goto exit;
3089             }
3090             pk = NULL;
3091         }
3092 #endif /* MBEDTLS_SSL_ASYNC_PRIVATE */
3093         if ((ret = mbedtls_ssl_conf_own_cert(&conf, &srvcert, pk)) != 0) {
3094             mbedtls_printf(" failed\n  ! mbedtls_ssl_conf_own_cert returned %d\n\n", ret);
3095             goto exit;
3096         }
3097     }
3098     if (key_cert_init2) {
3099         mbedtls_pk_context *pk = &pkey2;
3100 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
3101         if (opt.async_private_delay2 >= 0) {
3102             ret = ssl_async_set_key(&ssl_async_keys, &srvcert2, pk, 0,
3103                                     opt.async_private_delay2);
3104             if (ret < 0) {
3105                 mbedtls_printf("  Test error: ssl_async_set_key failed (%d)\n",
3106                                ret);
3107                 goto exit;
3108             }
3109             pk = NULL;
3110         }
3111 #endif /* MBEDTLS_SSL_ASYNC_PRIVATE */
3112         if ((ret = mbedtls_ssl_conf_own_cert(&conf, &srvcert2, pk)) != 0) {
3113             mbedtls_printf(" failed\n  ! mbedtls_ssl_conf_own_cert returned %d\n\n", ret);
3114             goto exit;
3115         }
3116     }
3117 
3118 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
3119     if (opt.async_operations[0] != '-') {
3120         mbedtls_ssl_async_sign_t *sign = NULL;
3121         mbedtls_ssl_async_decrypt_t *decrypt = NULL;
3122         const char *r;
3123         for (r = opt.async_operations; *r; r++) {
3124             switch (*r) {
3125                 case 'd':
3126                     decrypt = ssl_async_decrypt;
3127                     break;
3128                 case 's':
3129                     sign = ssl_async_sign;
3130                     break;
3131             }
3132         }
3133         ssl_async_keys.inject_error = (opt.async_private_error < 0 ?
3134                                        -opt.async_private_error :
3135                                        opt.async_private_error);
3136         ssl_async_keys.f_rng = rng_get;
3137         ssl_async_keys.p_rng = &rng;
3138         mbedtls_ssl_conf_async_private_cb(&conf,
3139                                           sign,
3140                                           decrypt,
3141                                           ssl_async_resume,
3142                                           ssl_async_cancel,
3143                                           &ssl_async_keys);
3144     }
3145 #endif /* MBEDTLS_SSL_ASYNC_PRIVATE */
3146 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED */
3147 
3148 #if defined(SNI_OPTION)
3149     if (opt.sni != NULL) {
3150         mbedtls_ssl_conf_sni(&conf, sni_callback, sni_info);
3151         mbedtls_ssl_conf_cert_cb(&conf, cert_callback);
3152 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
3153         if (opt.async_private_delay2 >= 0) {
3154             sni_entry *cur;
3155             for (cur = sni_info; cur != NULL; cur = cur->next) {
3156                 ret = ssl_async_set_key(&ssl_async_keys,
3157                                         cur->cert, cur->key, 1,
3158                                         opt.async_private_delay2);
3159                 if (ret < 0) {
3160                     mbedtls_printf("  Test error: ssl_async_set_key failed (%d)\n",
3161                                    ret);
3162                     goto exit;
3163                 }
3164                 cur->key = NULL;
3165             }
3166         }
3167 #endif /* MBEDTLS_SSL_ASYNC_PRIVATE */
3168     }
3169 #endif
3170 
3171 #if defined(MBEDTLS_PK_HAVE_ECC_KEYS) || \
3172     (defined(MBEDTLS_SSL_TLS1_3_KEY_EXCHANGE_MODE_SOME_EPHEMERAL_ENABLED) && \
3173     defined(PSA_WANT_ALG_FFDH))
3174     if (opt.groups != NULL &&
3175         strcmp(opt.groups, "default") != 0) {
3176         mbedtls_ssl_conf_groups(&conf, group_list);
3177     }
3178 #endif
3179 
3180 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
3181     if (opt.sig_algs != NULL) {
3182         mbedtls_ssl_conf_sig_algs(&conf, sig_alg_list);
3183     }
3184 #endif
3185 
3186 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED)
3187 
3188     if (strlen(opt.psk) != 0 && strlen(opt.psk_identity) != 0) {
3189 #if defined(MBEDTLS_USE_PSA_CRYPTO)
3190         if (opt.psk_opaque != 0) {
3191             /* The algorithm has already been determined earlier. */
3192             status = psa_setup_psk_key_slot(&psk_slot, alg, psk, psk_len);
3193             if (status != PSA_SUCCESS) {
3194                 fprintf(stderr, "SETUP FAIL\n");
3195                 ret = MBEDTLS_ERR_SSL_HW_ACCEL_FAILED;
3196                 goto exit;
3197             }
3198             if ((ret = mbedtls_ssl_conf_psk_opaque(&conf, psk_slot,
3199                                                    (const unsigned char *) opt.psk_identity,
3200                                                    strlen(opt.psk_identity))) != 0) {
3201                 mbedtls_printf(" failed\n  ! mbedtls_ssl_conf_psk_opaque returned %d\n\n",
3202                                ret);
3203                 goto exit;
3204             }
3205         } else
3206 #endif /* MBEDTLS_USE_PSA_CRYPTO */
3207         if (psk_len > 0) {
3208             ret = mbedtls_ssl_conf_psk(&conf, psk, psk_len,
3209                                        (const unsigned char *) opt.psk_identity,
3210                                        strlen(opt.psk_identity));
3211             if (ret != 0) {
3212                 mbedtls_printf("  failed\n  mbedtls_ssl_conf_psk returned -0x%04X\n\n",
3213                                (unsigned int) -ret);
3214                 goto exit;
3215             }
3216         }
3217     }
3218 
3219     if (opt.psk_list != NULL) {
3220 #if defined(MBEDTLS_USE_PSA_CRYPTO)
3221         if (opt.psk_list_opaque != 0) {
3222             psk_entry *cur_psk;
3223             for (cur_psk = psk_info; cur_psk != NULL; cur_psk = cur_psk->next) {
3224 
3225                 status = psa_setup_psk_key_slot(&cur_psk->slot, alg,
3226                                                 cur_psk->key,
3227                                                 cur_psk->key_len);
3228                 if (status != PSA_SUCCESS) {
3229                     ret = MBEDTLS_ERR_SSL_HW_ACCEL_FAILED;
3230                     goto exit;
3231                 }
3232             }
3233         }
3234 #endif /* MBEDTLS_USE_PSA_CRYPTO */
3235 
3236         mbedtls_ssl_conf_psk_cb(&conf, psk_callback, psk_info);
3237     }
3238 #endif
3239 
3240 #if defined(MBEDTLS_DHM_C)
3241     /*
3242      * Use different group than default DHM group
3243      */
3244 #if defined(MBEDTLS_FS_IO)
3245     if (opt.dhm_file != NULL) {
3246         ret = mbedtls_ssl_conf_dh_param_ctx(&conf, &dhm);
3247     }
3248 #endif
3249     if (ret != 0) {
3250         mbedtls_printf("  failed\n  mbedtls_ssl_conf_dh_param returned -0x%04X\n\n",
3251                        (unsigned int) -ret);
3252         goto exit;
3253     }
3254 #endif
3255 
3256     if (opt.min_version != DFL_MIN_VERSION) {
3257         mbedtls_ssl_conf_min_tls_version(&conf, opt.min_version);
3258     }
3259 
3260     if (opt.max_version != DFL_MIN_VERSION) {
3261         mbedtls_ssl_conf_max_tls_version(&conf, opt.max_version);
3262     }
3263 
3264     if ((ret = mbedtls_ssl_setup(&ssl, &conf)) != 0) {
3265         mbedtls_printf(" failed\n  ! mbedtls_ssl_setup returned -0x%x\n\n", (unsigned int) -ret);
3266         goto exit;
3267     }
3268 
3269     if (opt.eap_tls != 0) {
3270         mbedtls_ssl_set_export_keys_cb(&ssl, eap_tls_key_derivation,
3271                                        &eap_tls_keying);
3272     } else if (opt.nss_keylog != 0) {
3273         mbedtls_ssl_set_export_keys_cb(&ssl,
3274                                        nss_keylog_export,
3275                                        NULL);
3276     }
3277 #if defined(MBEDTLS_SSL_DTLS_SRTP)
3278     else if (opt.use_srtp != 0) {
3279         mbedtls_ssl_set_export_keys_cb(&ssl, dtls_srtp_key_derivation,
3280                                        &dtls_srtp_keying);
3281     }
3282 #endif /* MBEDTLS_SSL_DTLS_SRTP */
3283 
3284     io_ctx.ssl = &ssl;
3285     io_ctx.net = &client_fd;
3286     mbedtls_ssl_set_bio(&ssl, &io_ctx, send_cb, recv_cb,
3287                         opt.nbio == 0 ? recv_timeout_cb : NULL);
3288 
3289 #if defined(MBEDTLS_SSL_DTLS_CONNECTION_ID)
3290     if (opt.transport == MBEDTLS_SSL_TRANSPORT_DATAGRAM) {
3291         if ((ret = mbedtls_ssl_set_cid(&ssl, opt.cid_enabled,
3292                                        cid, cid_len)) != 0) {
3293             mbedtls_printf(" failed\n  ! mbedtls_ssl_set_cid returned %d\n\n",
3294                            ret);
3295             goto exit;
3296         }
3297     }
3298 #endif /* MBEDTLS_SSL_DTLS_CONNECTION_ID */
3299 
3300 #if defined(MBEDTLS_SSL_PROTO_DTLS)
3301     if (opt.dtls_mtu != DFL_DTLS_MTU) {
3302         mbedtls_ssl_set_mtu(&ssl, opt.dtls_mtu);
3303     }
3304 #endif
3305 
3306 #if defined(MBEDTLS_TIMING_C)
3307     mbedtls_ssl_set_timer_cb(&ssl, &timer, mbedtls_timing_set_delay,
3308                              mbedtls_timing_get_delay);
3309 #endif
3310 
3311     mbedtls_printf(" ok\n");
3312 
3313     /*
3314      * 3. Setup the listening TCP socket
3315      */
3316     mbedtls_printf("  . Bind on %s://%s:%s/ ...",
3317                    opt.transport == MBEDTLS_SSL_TRANSPORT_STREAM ? "tcp" : "udp",
3318                    opt.server_addr ? opt.server_addr : "*",
3319                    opt.server_port);
3320     fflush(stdout);
3321 
3322     if ((ret = mbedtls_net_bind(&listen_fd, opt.server_addr, opt.server_port,
3323                                 opt.transport == MBEDTLS_SSL_TRANSPORT_STREAM ?
3324                                 MBEDTLS_NET_PROTO_TCP : MBEDTLS_NET_PROTO_UDP)) != 0) {
3325         mbedtls_printf(" failed\n  ! mbedtls_net_bind returned -0x%x\n\n", (unsigned int) -ret);
3326         goto exit;
3327     }
3328     mbedtls_printf(" ok\n");
3329 
3330 reset:
3331 #if !defined(_WIN32)
3332     if (received_sigterm) {
3333         mbedtls_printf(" interrupted by SIGTERM (not in net_accept())\n");
3334         if (ret == MBEDTLS_ERR_NET_INVALID_CONTEXT) {
3335             ret = 0;
3336         }
3337 
3338         goto exit;
3339     }
3340 #endif
3341 
3342     if (ret == MBEDTLS_ERR_SSL_CLIENT_RECONNECT) {
3343         mbedtls_printf("  ! Client initiated reconnection from same port\n");
3344         goto handshake;
3345     }
3346 
3347 #ifdef MBEDTLS_ERROR_C
3348     if (ret != 0) {
3349         char error_buf[100];
3350         mbedtls_strerror(ret, error_buf, 100);
3351         mbedtls_printf("Last error was: %d - %s\n\n", ret, error_buf);
3352     }
3353 #endif
3354 
3355     mbedtls_net_free(&client_fd);
3356 
3357     mbedtls_ssl_session_reset(&ssl);
3358 
3359     /*
3360      * 3. Wait until a client connects
3361      */
3362     mbedtls_printf("  . Waiting for a remote connection ...");
3363     fflush(stdout);
3364 
3365     if ((ret = mbedtls_net_accept(&listen_fd, &client_fd,
3366                                   client_ip, sizeof(client_ip), &cliip_len)) != 0) {
3367 #if !defined(_WIN32)
3368         if (received_sigterm) {
3369             mbedtls_printf(" interrupted by SIGTERM (in net_accept())\n");
3370             if (ret == MBEDTLS_ERR_NET_ACCEPT_FAILED) {
3371                 ret = 0;
3372             }
3373 
3374             goto exit;
3375         }
3376 #endif
3377 
3378         mbedtls_printf(" failed\n  ! mbedtls_net_accept returned -0x%x\n\n", (unsigned int) -ret);
3379         goto exit;
3380     }
3381 
3382     if (opt.nbio > 0) {
3383         ret = mbedtls_net_set_nonblock(&client_fd);
3384     } else {
3385         ret = mbedtls_net_set_block(&client_fd);
3386     }
3387     if (ret != 0) {
3388         mbedtls_printf(" failed\n  ! net_set_(non)block() returned -0x%x\n\n", (unsigned int) -ret);
3389         goto exit;
3390     }
3391 
3392     mbedtls_ssl_conf_read_timeout(&conf, opt.read_timeout);
3393 
3394 #if defined(MBEDTLS_SSL_DTLS_HELLO_VERIFY)
3395     if (opt.transport == MBEDTLS_SSL_TRANSPORT_DATAGRAM) {
3396         if ((ret = mbedtls_ssl_set_client_transport_id(&ssl,
3397                                                        client_ip, cliip_len)) != 0) {
3398             mbedtls_printf(" failed\n  ! mbedtls_ssl_set_client_transport_id() returned -0x%x\n\n",
3399                            (unsigned int) -ret);
3400             goto exit;
3401         }
3402     }
3403 #endif /* MBEDTLS_SSL_DTLS_HELLO_VERIFY */
3404 
3405 #if defined(MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED)
3406     if (opt.ecjpake_pw != DFL_ECJPAKE_PW) {
3407 #if defined(MBEDTLS_USE_PSA_CRYPTO)
3408         if (opt.ecjpake_pw_opaque != DFL_ECJPAKE_PW_OPAQUE) {
3409             psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;
3410 
3411             psa_set_key_usage_flags(&attributes, PSA_KEY_USAGE_DERIVE);
3412             psa_set_key_algorithm(&attributes, PSA_ALG_JPAKE);
3413             psa_set_key_type(&attributes, PSA_KEY_TYPE_PASSWORD);
3414 
3415             status = psa_import_key(&attributes,
3416                                     (const unsigned char *) opt.ecjpake_pw,
3417                                     strlen(opt.ecjpake_pw),
3418                                     &ecjpake_pw_slot);
3419             if (status != PSA_SUCCESS) {
3420                 mbedtls_printf(" failed\n  ! psa_import_key returned %d\n\n",
3421                                status);
3422                 goto exit;
3423             }
3424             if ((ret = mbedtls_ssl_set_hs_ecjpake_password_opaque(&ssl,
3425                                                                   ecjpake_pw_slot)) != 0) {
3426                 mbedtls_printf(
3427                     " failed\n  ! mbedtls_ssl_set_hs_ecjpake_password_opaque returned %d\n\n",
3428                     ret);
3429                 goto exit;
3430             }
3431             mbedtls_printf("using opaque password\n");
3432         } else
3433 #endif  /* MBEDTLS_USE_PSA_CRYPTO */
3434         {
3435             if ((ret = mbedtls_ssl_set_hs_ecjpake_password(&ssl,
3436                                                            (const unsigned char *) opt.ecjpake_pw,
3437                                                            strlen(opt.ecjpake_pw))) != 0) {
3438                 mbedtls_printf(" failed\n  ! mbedtls_ssl_set_hs_ecjpake_password returned %d\n\n",
3439                                ret);
3440                 goto exit;
3441             }
3442         }
3443     }
3444 #endif /* MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED */
3445 
3446 #if defined(MBEDTLS_SSL_SERVER_NAME_INDICATION)
3447 #if defined(MBEDTLS_KEY_EXCHANGE_CERT_REQ_ALLOWED_ENABLED)
3448     /* exercise setting DN hints for server certificate request
3449      * (Intended for use where the client cert expected has been signed by
3450      *  a specific CA which is an intermediate in a CA chain, not the root)
3451      * (Additionally, the CA choice would typically be influenced by SNI
3452      *  if being set per-handshake using mbedtls_ssl_set_hs_dn_hints()) */
3453     if (opt.cert_req_dn_hint == 3 && key_cert_init2) {
3454         mbedtls_ssl_set_hs_dn_hints(&ssl, &srvcert2);
3455     }
3456 #endif
3457 #endif
3458 
3459     mbedtls_printf(" ok\n");
3460 
3461     /*
3462      * 4. Handshake
3463      */
3464 handshake:
3465     mbedtls_printf("  . Performing the SSL/TLS handshake...");
3466     fflush(stdout);
3467 
3468     while ((ret = mbedtls_ssl_handshake(&ssl)) != 0) {
3469 #if defined(MBEDTLS_SSL_EARLY_DATA)
3470         if (ret == MBEDTLS_ERR_SSL_RECEIVED_EARLY_DATA) {
3471             memset(buf, 0, opt.buffer_size);
3472             ret = mbedtls_ssl_read_early_data(&ssl, buf, opt.buffer_size);
3473             if (ret > 0) {
3474                 buf[ret] = '\0';
3475                 mbedtls_printf(" %d early data bytes read\n\n%s\n",
3476                                ret, (char *) buf);
3477             }
3478             continue;
3479         }
3480 #endif /* MBEDTLS_SSL_EARLY_DATA */
3481 
3482 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
3483         if (ret == MBEDTLS_ERR_SSL_ASYNC_IN_PROGRESS &&
3484             ssl_async_keys.inject_error == SSL_ASYNC_INJECT_ERROR_CANCEL) {
3485             mbedtls_printf(" cancelling on injected error\n");
3486             break;
3487         }
3488 #endif /* MBEDTLS_SSL_ASYNC_PRIVATE */
3489 
3490         if (!mbedtls_status_is_ssl_in_progress(ret)) {
3491             break;
3492         }
3493 
3494         /* For event-driven IO, wait for socket to become available */
3495         if (opt.event == 1 /* level triggered IO */) {
3496 #if defined(MBEDTLS_TIMING_C)
3497             ret = idle(&client_fd, &timer, ret);
3498 #else
3499             ret = idle(&client_fd, ret);
3500 #endif
3501             if (ret != 0) {
3502                 goto reset;
3503             }
3504         }
3505     }
3506 
3507     if (ret == MBEDTLS_ERR_SSL_HELLO_VERIFY_REQUIRED) {
3508         mbedtls_printf(" hello verification requested\n");
3509         ret = 0;
3510         goto reset;
3511     } else if (ret != 0) {
3512         mbedtls_printf(" failed\n  ! mbedtls_ssl_handshake returned -0x%x\n\n",
3513                        (unsigned int) -ret);
3514 
3515 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
3516         if (ret == MBEDTLS_ERR_X509_CERT_VERIFY_FAILED ||
3517             ret == MBEDTLS_ERR_SSL_BAD_CERTIFICATE) {
3518             char vrfy_buf[512];
3519             flags = mbedtls_ssl_get_verify_result(&ssl);
3520 
3521             x509_crt_verify_info(vrfy_buf, sizeof(vrfy_buf), "  ! ", flags);
3522 
3523             mbedtls_printf("%s\n", vrfy_buf);
3524         }
3525 #endif
3526 
3527 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
3528         if (opt.async_private_error < 0) {
3529             /* Injected error only the first time round, to test reset */
3530             ssl_async_keys.inject_error = SSL_ASYNC_INJECT_ERROR_NONE;
3531         }
3532 #endif
3533         goto reset;
3534     } else { /* ret == 0 */
3535         int suite_id = mbedtls_ssl_get_ciphersuite_id_from_ssl(&ssl);
3536         const mbedtls_ssl_ciphersuite_t *ciphersuite_info;
3537         ciphersuite_info = mbedtls_ssl_ciphersuite_from_id(suite_id);
3538 
3539         mbedtls_printf(" ok\n    [ Protocol is %s ]\n"
3540                        "    [ Ciphersuite is %s ]\n"
3541                        "    [ Key size is %u ]\n",
3542                        mbedtls_ssl_get_version(&ssl),
3543                        mbedtls_ssl_ciphersuite_get_name(ciphersuite_info),
3544                        (unsigned int)
3545                        mbedtls_ssl_ciphersuite_get_cipher_key_bitlen(ciphersuite_info));
3546     }
3547 
3548     if ((ret = mbedtls_ssl_get_record_expansion(&ssl)) >= 0) {
3549         mbedtls_printf("    [ Record expansion is %d ]\n", ret);
3550     } else {
3551         mbedtls_printf("    [ Record expansion is unknown ]\n");
3552     }
3553 
3554 #if defined(MBEDTLS_SSL_MAX_FRAGMENT_LENGTH) || defined(MBEDTLS_SSL_RECORD_SIZE_LIMIT)
3555     mbedtls_printf("    [ Maximum incoming record payload length is %u ]\n",
3556                    (unsigned int) mbedtls_ssl_get_max_in_record_payload(&ssl));
3557     mbedtls_printf("    [ Maximum outgoing record payload length is %u ]\n",
3558                    (unsigned int) mbedtls_ssl_get_max_out_record_payload(&ssl));
3559 #endif
3560 
3561 #if defined(MBEDTLS_SSL_ALPN)
3562     if (opt.alpn_string != NULL) {
3563         const char *alp = mbedtls_ssl_get_alpn_protocol(&ssl);
3564         mbedtls_printf("    [ Application Layer Protocol is %s ]\n",
3565                        alp ? alp : "(none)");
3566     }
3567 #endif
3568 
3569 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
3570     /*
3571      * 5. Verify the client certificate
3572      */
3573     mbedtls_printf("  . Verifying peer X.509 certificate...");
3574 
3575     if ((flags = mbedtls_ssl_get_verify_result(&ssl)) != 0) {
3576         char vrfy_buf[512];
3577 
3578         mbedtls_printf(" failed\n");
3579 
3580         x509_crt_verify_info(vrfy_buf, sizeof(vrfy_buf), "  ! ", flags);
3581         mbedtls_printf("%s\n", vrfy_buf);
3582     } else {
3583         mbedtls_printf(" ok\n");
3584     }
3585 
3586 #if !defined(MBEDTLS_X509_REMOVE_INFO)
3587     if (mbedtls_ssl_get_peer_cert(&ssl) != NULL) {
3588         char crt_buf[512];
3589 
3590         mbedtls_printf("  . Peer certificate information    ...\n");
3591         mbedtls_x509_crt_info(crt_buf, sizeof(crt_buf), "      ",
3592                               mbedtls_ssl_get_peer_cert(&ssl));
3593         mbedtls_printf("%s\n", crt_buf);
3594     }
3595 #endif /* MBEDTLS_X509_REMOVE_INFO */
3596 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED */
3597 
3598     if (opt.eap_tls != 0) {
3599         size_t j = 0;
3600 
3601         if ((ret = mbedtls_ssl_tls_prf(eap_tls_keying.tls_prf_type,
3602                                        eap_tls_keying.master_secret,
3603                                        sizeof(eap_tls_keying.master_secret),
3604                                        eap_tls_label,
3605                                        eap_tls_keying.randbytes,
3606                                        sizeof(eap_tls_keying.randbytes),
3607                                        eap_tls_keymaterial,
3608                                        sizeof(eap_tls_keymaterial)))
3609             != 0) {
3610             mbedtls_printf(" failed\n  ! mbedtls_ssl_tls_prf returned -0x%x\n\n",
3611                            (unsigned int) -ret);
3612             goto reset;
3613         }
3614 
3615         mbedtls_printf("    EAP-TLS key material is:");
3616         for (j = 0; j < sizeof(eap_tls_keymaterial); j++) {
3617             if (j % 8 == 0) {
3618                 mbedtls_printf("\n    ");
3619             }
3620             mbedtls_printf("%02x ", eap_tls_keymaterial[j]);
3621         }
3622         mbedtls_printf("\n");
3623 
3624         if ((ret = mbedtls_ssl_tls_prf(eap_tls_keying.tls_prf_type, NULL, 0,
3625                                        eap_tls_label,
3626                                        eap_tls_keying.randbytes,
3627                                        sizeof(eap_tls_keying.randbytes),
3628                                        eap_tls_iv,
3629                                        sizeof(eap_tls_iv))) != 0) {
3630             mbedtls_printf(" failed\n  ! mbedtls_ssl_tls_prf returned -0x%x\n\n",
3631                            (unsigned int) -ret);
3632             goto reset;
3633         }
3634 
3635         mbedtls_printf("    EAP-TLS IV is:");
3636         for (j = 0; j < sizeof(eap_tls_iv); j++) {
3637             if (j % 8 == 0) {
3638                 mbedtls_printf("\n    ");
3639             }
3640             mbedtls_printf("%02x ", eap_tls_iv[j]);
3641         }
3642         mbedtls_printf("\n");
3643     }
3644 
3645 #if defined(MBEDTLS_SSL_DTLS_SRTP)
3646     else if (opt.use_srtp != 0) {
3647         size_t j = 0;
3648         mbedtls_dtls_srtp_info dtls_srtp_negotiation_result;
3649         mbedtls_ssl_get_dtls_srtp_negotiation_result(&ssl, &dtls_srtp_negotiation_result);
3650 
3651         if (dtls_srtp_negotiation_result.chosen_dtls_srtp_profile
3652             == MBEDTLS_TLS_SRTP_UNSET) {
3653             mbedtls_printf("    Unable to negotiate "
3654                            "the use of DTLS-SRTP\n");
3655         } else {
3656             if ((ret = mbedtls_ssl_tls_prf(dtls_srtp_keying.tls_prf_type,
3657                                            dtls_srtp_keying.master_secret,
3658                                            sizeof(dtls_srtp_keying.master_secret),
3659                                            dtls_srtp_label,
3660                                            dtls_srtp_keying.randbytes,
3661                                            sizeof(dtls_srtp_keying.randbytes),
3662                                            dtls_srtp_key_material,
3663                                            sizeof(dtls_srtp_key_material)))
3664                 != 0) {
3665                 mbedtls_printf(" failed\n  ! mbedtls_ssl_tls_prf returned -0x%x\n\n",
3666                                (unsigned int) -ret);
3667                 goto exit;
3668             }
3669 
3670             mbedtls_printf("    DTLS-SRTP key material is:");
3671             for (j = 0; j < sizeof(dtls_srtp_key_material); j++) {
3672                 if (j % 8 == 0) {
3673                     mbedtls_printf("\n    ");
3674                 }
3675                 mbedtls_printf("%02x ", dtls_srtp_key_material[j]);
3676             }
3677             mbedtls_printf("\n");
3678 
3679             /* produce a less readable output used to perform automatic checks
3680              * - compare client and server output
3681              * - interop test with openssl which client produces this kind of output
3682              */
3683             mbedtls_printf("    Keying material: ");
3684             for (j = 0; j < sizeof(dtls_srtp_key_material); j++) {
3685                 mbedtls_printf("%02X", dtls_srtp_key_material[j]);
3686             }
3687             mbedtls_printf("\n");
3688 
3689             if (dtls_srtp_negotiation_result.mki_len > 0) {
3690                 mbedtls_printf("    DTLS-SRTP mki value: ");
3691                 for (j = 0; j < dtls_srtp_negotiation_result.mki_len; j++) {
3692                     mbedtls_printf("%02X", dtls_srtp_negotiation_result.mki_value[j]);
3693                 }
3694             } else {
3695                 mbedtls_printf("    DTLS-SRTP no mki value negotiated");
3696             }
3697             mbedtls_printf("\n");
3698 
3699         }
3700     }
3701 #endif /* MBEDTLS_SSL_DTLS_SRTP */
3702 
3703 #if defined(MBEDTLS_SSL_DTLS_CONNECTION_ID)
3704     ret = report_cid_usage(&ssl, "initial handshake");
3705     if (ret != 0) {
3706         goto exit;
3707     }
3708 
3709     if (opt.transport == MBEDTLS_SSL_TRANSPORT_DATAGRAM) {
3710         if ((ret = mbedtls_ssl_set_cid(&ssl, opt.cid_enabled_renego,
3711                                        cid_renego, cid_renego_len)) != 0) {
3712             mbedtls_printf(" failed\n  ! mbedtls_ssl_set_cid returned %d\n\n",
3713                            ret);
3714             goto exit;
3715         }
3716     }
3717 #endif /* MBEDTLS_SSL_DTLS_CONNECTION_ID */
3718 
3719 #if defined(MBEDTLS_MEMORY_DEBUG)
3720     mbedtls_memory_buffer_alloc_cur_get(&current_heap_memory, &heap_blocks);
3721     mbedtls_memory_buffer_alloc_max_get(&peak_heap_memory, &heap_blocks);
3722     mbedtls_printf("Heap memory usage after handshake: %lu bytes. Peak memory usage was %lu\n",
3723                    (unsigned long) current_heap_memory, (unsigned long) peak_heap_memory);
3724 #endif  /* MBEDTLS_MEMORY_DEBUG */
3725 
3726     if (opt.exchanges == 0) {
3727         goto close_notify;
3728     }
3729 
3730     exchanges_left = opt.exchanges;
3731 data_exchange:
3732     /*
3733      * 6. Read the HTTP Request
3734      */
3735     mbedtls_printf("  < Read from client:");
3736     fflush(stdout);
3737 
3738     /*
3739      * TLS and DTLS need different reading styles (stream vs datagram)
3740      */
3741     if (opt.transport == MBEDTLS_SSL_TRANSPORT_STREAM) {
3742         do {
3743             int terminated = 0;
3744             len = opt.buffer_size;
3745             memset(buf, 0, opt.buffer_size);
3746             ret = mbedtls_ssl_read(&ssl, buf, len);
3747 
3748             if (mbedtls_status_is_ssl_in_progress(ret)) {
3749                 if (opt.event == 1 /* level triggered IO */) {
3750 #if defined(MBEDTLS_TIMING_C)
3751                     idle(&client_fd, &timer, ret);
3752 #else
3753                     idle(&client_fd, ret);
3754 #endif
3755                 }
3756 
3757                 continue;
3758             }
3759 
3760             if (ret <= 0) {
3761                 switch (ret) {
3762                     case MBEDTLS_ERR_SSL_PEER_CLOSE_NOTIFY:
3763                         mbedtls_printf(" connection was closed gracefully\n");
3764                         goto close_notify;
3765 
3766                     case 0:
3767                     case MBEDTLS_ERR_NET_CONN_RESET:
3768                         mbedtls_printf(" connection was reset by peer\n");
3769                         ret = MBEDTLS_ERR_NET_CONN_RESET;
3770                         goto reset;
3771 
3772                     default:
3773                         mbedtls_printf(" mbedtls_ssl_read returned -0x%x\n", (unsigned int) -ret);
3774                         goto reset;
3775                 }
3776             }
3777 
3778             if (mbedtls_ssl_get_bytes_avail(&ssl) == 0) {
3779                 len = ret;
3780                 buf[len] = '\0';
3781                 mbedtls_printf(" %d bytes read\n\n%s\n", len, (char *) buf);
3782 
3783                 /* End of message should be detected according to the syntax of the
3784                  * application protocol (eg HTTP), just use a dummy test here. */
3785                 if (buf[len - 1] == '\n') {
3786                     terminated = 1;
3787                 }
3788             } else {
3789                 int extra_len, ori_len;
3790                 unsigned char *larger_buf;
3791 
3792                 ori_len = ret;
3793                 extra_len = (int) mbedtls_ssl_get_bytes_avail(&ssl);
3794 
3795                 larger_buf = mbedtls_calloc(1, ori_len + extra_len + 1);
3796                 if (larger_buf == NULL) {
3797                     mbedtls_printf("  ! memory allocation failed\n");
3798                     ret = 1;
3799                     goto reset;
3800                 }
3801 
3802                 memset(larger_buf, 0, ori_len + extra_len);
3803                 memcpy(larger_buf, buf, ori_len);
3804 
3805                 /* This read should never fail and get the whole cached data */
3806                 ret = mbedtls_ssl_read(&ssl, larger_buf + ori_len, extra_len);
3807                 if (ret != extra_len ||
3808                     mbedtls_ssl_get_bytes_avail(&ssl) != 0) {
3809                     mbedtls_printf("  ! mbedtls_ssl_read failed on cached data\n");
3810                     ret = 1;
3811                     goto reset;
3812                 }
3813 
3814                 larger_buf[ori_len + extra_len] = '\0';
3815                 mbedtls_printf(" %d bytes read (%d + %d)\n\n%s\n",
3816                                ori_len + extra_len, ori_len, extra_len,
3817                                (char *) larger_buf);
3818 
3819                 /* End of message should be detected according to the syntax of the
3820                  * application protocol (eg HTTP), just use a dummy test here. */
3821                 if (larger_buf[ori_len + extra_len - 1] == '\n') {
3822                     terminated = 1;
3823                 }
3824 
3825                 mbedtls_free(larger_buf);
3826             }
3827 
3828             if (terminated) {
3829                 ret = 0;
3830                 break;
3831             }
3832         } while (1);
3833     } else { /* Not stream, so datagram */
3834         len = opt.buffer_size;
3835         memset(buf, 0, opt.buffer_size);
3836 
3837         do {
3838             /* Without the call to `mbedtls_ssl_check_pending`, it might
3839              * happen that the client sends application data in the same
3840              * datagram as the Finished message concluding the handshake.
3841              * In this case, the application data would be ready to be
3842              * processed while the underlying transport wouldn't signal
3843              * any further incoming data.
3844              *
3845              * See the test 'Event-driven I/O: session-id resume, UDP packing'
3846              * in tests/ssl-opt.sh.
3847              */
3848 
3849             /* For event-driven IO, wait for socket to become available */
3850             if (mbedtls_ssl_check_pending(&ssl) == 0 &&
3851                 opt.event == 1 /* level triggered IO */) {
3852 #if defined(MBEDTLS_TIMING_C)
3853                 idle(&client_fd, &timer, MBEDTLS_ERR_SSL_WANT_READ);
3854 #else
3855                 idle(&client_fd, MBEDTLS_ERR_SSL_WANT_READ);
3856 #endif
3857             }
3858 
3859             ret = mbedtls_ssl_read(&ssl, buf, len);
3860 
3861             /* Note that even if `mbedtls_ssl_check_pending` returns true,
3862              * it can happen that the subsequent call to `mbedtls_ssl_read`
3863              * returns `MBEDTLS_ERR_SSL_WANT_READ`, because the pending messages
3864              * might be discarded (e.g. because they are retransmissions). */
3865         } while (mbedtls_status_is_ssl_in_progress(ret));
3866 
3867         if (ret <= 0) {
3868             switch (ret) {
3869                 case MBEDTLS_ERR_SSL_PEER_CLOSE_NOTIFY:
3870                     mbedtls_printf(" connection was closed gracefully\n");
3871                     goto close_notify;
3872 
3873                 default:
3874                     mbedtls_printf(" mbedtls_ssl_read returned -0x%x\n", (unsigned int) -ret);
3875                     goto reset;
3876             }
3877         }
3878 
3879         len = ret;
3880         buf[len] = '\0';
3881         mbedtls_printf(" %d bytes read\n\n%s", len, (char *) buf);
3882         ret = 0;
3883     }
3884 
3885     /*
3886      * 7a. Request renegotiation while client is waiting for input from us.
3887      * (only on the first exchange, to be able to test retransmission)
3888      */
3889 #if defined(MBEDTLS_SSL_RENEGOTIATION)
3890     if (opt.renegotiate && exchanges_left == opt.exchanges) {
3891         mbedtls_printf("  . Requestion renegotiation...");
3892         fflush(stdout);
3893 
3894         while ((ret = mbedtls_ssl_renegotiate(&ssl)) != 0) {
3895             if (!mbedtls_status_is_ssl_in_progress(ret)) {
3896                 mbedtls_printf(" failed\n  ! mbedtls_ssl_renegotiate returned %d\n\n", ret);
3897                 goto reset;
3898             }
3899 
3900             /* For event-driven IO, wait for socket to become available */
3901             if (opt.event == 1 /* level triggered IO */) {
3902 #if defined(MBEDTLS_TIMING_C)
3903                 idle(&client_fd, &timer, ret);
3904 #else
3905                 idle(&client_fd, ret);
3906 #endif
3907             }
3908         }
3909 
3910         mbedtls_printf(" ok\n");
3911     }
3912 #endif /* MBEDTLS_SSL_RENEGOTIATION */
3913 
3914 #if defined(MBEDTLS_SSL_DTLS_CONNECTION_ID)
3915     ret = report_cid_usage(&ssl, "after renegotiation");
3916     if (ret != 0) {
3917         goto exit;
3918     }
3919 #endif /* MBEDTLS_SSL_DTLS_CONNECTION_ID */
3920 
3921     /*
3922      * 7. Write the 200 Response
3923      */
3924     mbedtls_printf("  > Write to client:");
3925     fflush(stdout);
3926 
3927     /* If the format of the response changes, make sure there is enough
3928      * room in buf (buf_content_size calculation above). */
3929     len = sprintf((char *) buf, HTTP_RESPONSE,
3930                   mbedtls_ssl_get_ciphersuite(&ssl));
3931 
3932     /* Add padding to the response to reach opt.response_size in length */
3933     if (opt.response_size != DFL_RESPONSE_SIZE &&
3934         len < opt.response_size) {
3935         memset(buf + len, 'B', opt.response_size - len);
3936         len += opt.response_size - len;
3937     }
3938 
3939     /* Truncate if response size is smaller than the "natural" size */
3940     if (opt.response_size != DFL_RESPONSE_SIZE &&
3941         len > opt.response_size) {
3942         len = opt.response_size;
3943 
3944         /* Still end with \r\n unless that's really not possible */
3945         if (len >= 2) {
3946             buf[len - 2] = '\r';
3947         }
3948         if (len >= 1) {
3949             buf[len - 1] = '\n';
3950         }
3951     }
3952 
3953     if (opt.transport == MBEDTLS_SSL_TRANSPORT_STREAM) {
3954         for (written = 0, frags = 0; written < len; written += ret, frags++) {
3955             while ((ret = mbedtls_ssl_write(&ssl, buf + written, len - written))
3956                    <= 0) {
3957                 if (ret == MBEDTLS_ERR_NET_CONN_RESET) {
3958                     mbedtls_printf(" failed\n  ! peer closed the connection\n\n");
3959                     goto reset;
3960                 }
3961 
3962                 if (!mbedtls_status_is_ssl_in_progress(ret)) {
3963                     mbedtls_printf(" failed\n  ! mbedtls_ssl_write returned %d\n\n", ret);
3964                     goto reset;
3965                 }
3966 
3967                 /* For event-driven IO, wait for socket to become available */
3968                 if (opt.event == 1 /* level triggered IO */) {
3969 #if defined(MBEDTLS_TIMING_C)
3970                     idle(&client_fd, &timer, ret);
3971 #else
3972                     idle(&client_fd, ret);
3973 #endif
3974                 }
3975             }
3976         }
3977     } else { /* Not stream, so datagram */
3978         while (1) {
3979             ret = mbedtls_ssl_write(&ssl, buf, len);
3980 
3981             if (!mbedtls_status_is_ssl_in_progress(ret)) {
3982                 break;
3983             }
3984 
3985             /* For event-driven IO, wait for socket to become available */
3986             if (opt.event == 1 /* level triggered IO */) {
3987 #if defined(MBEDTLS_TIMING_C)
3988                 idle(&client_fd, &timer, ret);
3989 #else
3990                 idle(&client_fd, ret);
3991 #endif
3992             }
3993         }
3994 
3995         if (ret < 0) {
3996             mbedtls_printf(" failed\n  ! mbedtls_ssl_write returned %d\n\n", ret);
3997             goto reset;
3998         }
3999 
4000         frags = 1;
4001         written = ret;
4002     }
4003 
4004     buf[written] = '\0';
4005     mbedtls_printf(" %d bytes written in %d fragments\n\n%s\n", written, frags, (char *) buf);
4006     ret = 0;
4007 
4008     /*
4009      * 7b. Simulate serialize/deserialize and go back to data exchange
4010      */
4011 #if defined(MBEDTLS_SSL_CONTEXT_SERIALIZATION)
4012     if (opt.serialize != 0) {
4013         size_t buf_len;
4014 
4015         mbedtls_printf("  . Serializing live connection...");
4016 
4017         ret = mbedtls_ssl_context_save(&ssl, NULL, 0, &buf_len);
4018         if (ret != MBEDTLS_ERR_SSL_BUFFER_TOO_SMALL) {
4019             mbedtls_printf(" failed\n  ! mbedtls_ssl_context_save returned "
4020                            "-0x%x\n\n", (unsigned int) -ret);
4021 
4022             goto exit;
4023         }
4024 
4025         if ((context_buf = mbedtls_calloc(1, buf_len)) == NULL) {
4026             mbedtls_printf(" failed\n  ! Couldn't allocate buffer for "
4027                            "serialized context");
4028 
4029             goto exit;
4030         }
4031         context_buf_len = buf_len;
4032 
4033         if ((ret = mbedtls_ssl_context_save(&ssl, context_buf,
4034                                             buf_len, &buf_len)) != 0) {
4035             mbedtls_printf(" failed\n  ! mbedtls_ssl_context_save returned "
4036                            "-0x%x\n\n", (unsigned int) -ret);
4037 
4038             goto exit;
4039         }
4040 
4041         mbedtls_printf(" ok\n");
4042 
4043         /* Save serialized context to the 'opt.context_file' as a base64 code */
4044         if (0 < strlen(opt.context_file)) {
4045             FILE *b64_file;
4046             uint8_t *b64_buf;
4047             size_t b64_len;
4048 
4049             mbedtls_printf("  . Save serialized context to a file... ");
4050 
4051             mbedtls_base64_encode(NULL, 0, &b64_len, context_buf, buf_len);
4052 
4053             if ((b64_buf = mbedtls_calloc(1, b64_len)) == NULL) {
4054                 mbedtls_printf("failed\n  ! Couldn't allocate buffer for "
4055                                "the base64 code\n");
4056                 goto exit;
4057             }
4058 
4059             if ((ret = mbedtls_base64_encode(b64_buf, b64_len, &b64_len,
4060                                              context_buf, buf_len)) != 0) {
4061                 mbedtls_printf("failed\n  ! mbedtls_base64_encode returned "
4062                                "-0x%x\n", (unsigned int) -ret);
4063                 mbedtls_free(b64_buf);
4064                 goto exit;
4065             }
4066 
4067             if ((b64_file = fopen(opt.context_file, "w")) == NULL) {
4068                 mbedtls_printf("failed\n  ! Cannot open '%s' for writing.\n",
4069                                opt.context_file);
4070                 mbedtls_free(b64_buf);
4071                 goto exit;
4072             }
4073 
4074             if (b64_len != fwrite(b64_buf, 1, b64_len, b64_file)) {
4075                 mbedtls_printf("failed\n  ! fwrite(%ld bytes) failed\n",
4076                                (long) b64_len);
4077                 mbedtls_free(b64_buf);
4078                 fclose(b64_file);
4079                 goto exit;
4080             }
4081 
4082             mbedtls_free(b64_buf);
4083             fclose(b64_file);
4084 
4085             mbedtls_printf("ok\n");
4086         }
4087 
4088         /*
4089          * This simulates a workflow where you have a long-lived server
4090          * instance, potentially with a pool of ssl_context objects, and you
4091          * just want to re-use one while the connection is inactive: in that
4092          * case you can just reset() it, and then it's ready to receive
4093          * serialized data from another connection (or the same here).
4094          */
4095         if (opt.serialize == 1) {
4096             /* nothing to do here, done by context_save() already */
4097             mbedtls_printf("  . Context has been reset... ok\n");
4098         }
4099 
4100         /*
4101          * This simulates a workflow where you have one server instance per
4102          * connection, and want to release it entire when the connection is
4103          * inactive, and spawn it again when needed again - this would happen
4104          * between ssl_free() and ssl_init() below, together with any other
4105          * teardown/startup code needed - for example, preparing the
4106          * ssl_config again (see section 3 "setup stuff" in this file).
4107          */
4108         if (opt.serialize == 2) {
4109             mbedtls_printf("  . Freeing and reinitializing context...");
4110 
4111             mbedtls_ssl_free(&ssl);
4112 
4113             mbedtls_ssl_init(&ssl);
4114 
4115             if ((ret = mbedtls_ssl_setup(&ssl, &conf)) != 0) {
4116                 mbedtls_printf(" failed\n  ! mbedtls_ssl_setup returned "
4117                                "-0x%x\n\n", (unsigned int) -ret);
4118                 goto exit;
4119             }
4120 
4121             /*
4122              * This illustrates the minimum amount of things you need to set
4123              * up, however you could set up much more if desired, for example
4124              * if you want to share your set up code between the case of
4125              * establishing a new connection and this case.
4126              */
4127             if (opt.nbio == 2) {
4128                 mbedtls_ssl_set_bio(&ssl, &client_fd, delayed_send,
4129                                     delayed_recv, NULL);
4130             } else {
4131                 mbedtls_ssl_set_bio(&ssl, &client_fd, mbedtls_net_send,
4132                                     mbedtls_net_recv,
4133                                     opt.nbio == 0 ? mbedtls_net_recv_timeout : NULL);
4134             }
4135 
4136 #if defined(MBEDTLS_TIMING_C)
4137             mbedtls_ssl_set_timer_cb(&ssl, &timer,
4138                                      mbedtls_timing_set_delay,
4139                                      mbedtls_timing_get_delay);
4140 #endif /* MBEDTLS_TIMING_C */
4141 
4142             mbedtls_printf(" ok\n");
4143         }
4144 
4145         mbedtls_printf("  . Deserializing connection...");
4146 
4147         if ((ret = mbedtls_ssl_context_load(&ssl, context_buf,
4148                                             buf_len)) != 0) {
4149             mbedtls_printf("failed\n  ! mbedtls_ssl_context_load returned "
4150                            "-0x%x\n\n", (unsigned int) -ret);
4151 
4152             goto exit;
4153         }
4154 
4155         mbedtls_free(context_buf);
4156         context_buf = NULL;
4157         context_buf_len = 0;
4158 
4159         mbedtls_printf(" ok\n");
4160     }
4161 #endif /* MBEDTLS_SSL_CONTEXT_SERIALIZATION */
4162 
4163     /*
4164      * 7c. Continue doing data exchanges?
4165      */
4166     if (--exchanges_left > 0) {
4167         goto data_exchange;
4168     }
4169 
4170     /*
4171      * 8. Done, cleanly close the connection
4172      */
4173 close_notify:
4174     mbedtls_printf("  . Closing the connection...");
4175 
4176     /* No error checking, the connection might be closed already */
4177     do {
4178         ret = mbedtls_ssl_close_notify(&ssl);
4179     } while (ret == MBEDTLS_ERR_SSL_WANT_WRITE);
4180     ret = 0;
4181 
4182     mbedtls_printf(" done\n");
4183 
4184 #if defined(MBEDTLS_SSL_CACHE_C)
4185     if (opt.cache_remove > 0) {
4186         mbedtls_ssl_cache_remove(&cache, ssl.session->id, ssl.session->id_len);
4187     }
4188 #endif
4189 
4190     goto reset;
4191 
4192     /*
4193      * Cleanup and exit
4194      */
4195 exit:
4196 #ifdef MBEDTLS_ERROR_C
4197     if (ret != 0) {
4198         char error_buf[100];
4199         mbedtls_strerror(ret, error_buf, 100);
4200         mbedtls_printf("Last error was: -0x%X - %s\n\n", (unsigned int) -ret, error_buf);
4201     }
4202 #endif
4203 
4204     if (opt.query_config_mode == DFL_QUERY_CONFIG_MODE) {
4205         mbedtls_printf("  . Cleaning up...");
4206         fflush(stdout);
4207     }
4208 
4209     mbedtls_net_free(&client_fd);
4210     mbedtls_net_free(&listen_fd);
4211 
4212     mbedtls_ssl_free(&ssl);
4213     mbedtls_ssl_config_free(&conf);
4214 
4215 #if defined(MBEDTLS_SSL_CACHE_C)
4216     mbedtls_ssl_cache_free(&cache);
4217 #endif
4218 #if defined(MBEDTLS_SSL_SESSION_TICKETS) && defined(MBEDTLS_SSL_TICKET_C)
4219     mbedtls_ssl_ticket_free(&ticket_ctx);
4220 #endif
4221 #if defined(MBEDTLS_SSL_COOKIE_C)
4222     mbedtls_ssl_cookie_free(&cookie_ctx);
4223 #endif
4224 
4225 #if defined(MBEDTLS_SSL_CONTEXT_SERIALIZATION)
4226     if (context_buf != NULL) {
4227         mbedtls_platform_zeroize(context_buf, context_buf_len);
4228     }
4229     mbedtls_free(context_buf);
4230 #endif
4231 
4232 #if defined(SNI_OPTION)
4233     sni_free(sni_info);
4234 #endif
4235 
4236 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED)
4237     ret = psk_free(psk_info);
4238     if ((ret != 0) && (opt.query_config_mode == DFL_QUERY_CONFIG_MODE)) {
4239         mbedtls_printf("Failed to list of opaque PSKs - error was %d\n", ret);
4240     }
4241 #endif
4242 
4243 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_CERT_ENABLED)
4244     mbedtls_x509_crt_free(&cacert);
4245     mbedtls_x509_crt_free(&srvcert);
4246     mbedtls_pk_free(&pkey);
4247     mbedtls_x509_crt_free(&srvcert2);
4248     mbedtls_pk_free(&pkey2);
4249 #if defined(MBEDTLS_USE_PSA_CRYPTO)
4250     psa_destroy_key(key_slot);
4251     psa_destroy_key(key_slot2);
4252 #endif
4253 #endif
4254 
4255 #if defined(MBEDTLS_DHM_C) && defined(MBEDTLS_FS_IO)
4256     mbedtls_dhm_free(&dhm);
4257 #endif
4258 
4259 #if defined(MBEDTLS_SSL_ASYNC_PRIVATE)
4260     for (i = 0; (size_t) i < ssl_async_keys.slots_used; i++) {
4261         if (ssl_async_keys.slots[i].pk_owned) {
4262             mbedtls_pk_free(ssl_async_keys.slots[i].pk);
4263             mbedtls_free(ssl_async_keys.slots[i].pk);
4264             ssl_async_keys.slots[i].pk = NULL;
4265         }
4266     }
4267 #endif
4268 
4269 #if defined(MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED) && \
4270     defined(MBEDTLS_USE_PSA_CRYPTO)
4271     if (opt.psk_opaque != 0) {
4272         /* This is ok even if the slot hasn't been
4273          * initialized (we might have jumed here
4274          * immediately because of bad cmd line params,
4275          * for example). */
4276         status = psa_destroy_key(psk_slot);
4277         if ((status != PSA_SUCCESS) &&
4278             (opt.query_config_mode == DFL_QUERY_CONFIG_MODE)) {
4279             mbedtls_printf("Failed to destroy key slot %u - error was %d",
4280                            (unsigned) MBEDTLS_SVC_KEY_ID_GET_KEY_ID(psk_slot),
4281                            (int) status);
4282         }
4283     }
4284 #endif /* MBEDTLS_SSL_HANDSHAKE_WITH_PSK_ENABLED &&
4285           MBEDTLS_USE_PSA_CRYPTO */
4286 
4287 #if defined(MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED) && \
4288     defined(MBEDTLS_USE_PSA_CRYPTO)
4289     /*
4290      * In case opaque keys it's the user responsibility to keep the key valid
4291      * for the duration of the handshake and destroy it at the end
4292      */
4293     if ((opt.ecjpake_pw_opaque != DFL_ECJPAKE_PW_OPAQUE)) {
4294         psa_key_attributes_t check_attributes = PSA_KEY_ATTRIBUTES_INIT;
4295 
4296         /* Verify that the key is still valid before destroying it */
4297         if (psa_get_key_attributes(ecjpake_pw_slot, &check_attributes) !=
4298             PSA_SUCCESS) {
4299             if (ret == 0) {
4300                 ret = 1;
4301             }
4302             mbedtls_printf("The EC J-PAKE password key has unexpectedly been already destroyed\n");
4303         } else {
4304             psa_destroy_key(ecjpake_pw_slot);
4305         }
4306     }
4307 #endif  /* MBEDTLS_KEY_EXCHANGE_ECJPAKE_ENABLED && MBEDTLS_USE_PSA_CRYPTO */
4308 
4309 #if defined(MBEDTLS_USE_PSA_CRYPTO) || defined(MBEDTLS_SSL_PROTO_TLS1_3)
4310     const char *message = mbedtls_test_helper_is_psa_leaking();
4311     if (message) {
4312         if (ret == 0) {
4313             ret = 1;
4314         }
4315         mbedtls_printf("PSA memory leak detected: %s\n",  message);
4316     }
4317 #endif
4318 
4319     /* For builds with MBEDTLS_TEST_USE_PSA_CRYPTO_RNG psa crypto
4320      * resources are freed by rng_free(). */
4321     /* For builds with MBEDTLS_SSL_PROTO_TLS1_3, PSA may have been
4322      * initialized under the hood by the TLS layer. See
4323      * mbedtls_ssl_tls13_crypto_init(). */
4324 #if (defined(MBEDTLS_USE_PSA_CRYPTO) || defined(MBEDTLS_SSL_PROTO_TLS1_3)) \
4325     && !defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
4326     mbedtls_psa_crypto_free();
4327 #endif
4328 
4329     rng_free(&rng);
4330 
4331     mbedtls_free(buf);
4332 
4333 #if defined(MBEDTLS_TEST_HOOKS)
4334     /* Let test hooks detect errors such as resource leaks.
4335      * Don't do it in query_config mode, because some test code prints
4336      * information to stdout and this gets mixed with the regular output. */
4337     if (opt.query_config_mode == DFL_QUERY_CONFIG_MODE) {
4338         if (test_hooks_failure_detected()) {
4339             if (ret == 0) {
4340                 ret = 1;
4341             }
4342             mbedtls_printf("Test hooks detected errors.\n");
4343         }
4344     }
4345     test_hooks_free();
4346 #endif /* MBEDTLS_TEST_HOOKS */
4347 
4348 #if defined(MBEDTLS_MEMORY_BUFFER_ALLOC_C)
4349 #if defined(MBEDTLS_MEMORY_DEBUG)
4350     mbedtls_memory_buffer_alloc_status();
4351 #endif
4352     mbedtls_memory_buffer_alloc_free();
4353 #endif  /* MBEDTLS_MEMORY_BUFFER_ALLOC_C */
4354 
4355     if (opt.query_config_mode == DFL_QUERY_CONFIG_MODE) {
4356         mbedtls_printf(" done.\n");
4357     }
4358 
4359     // Shell can not handle large exit numbers -> 1 for errors
4360     if (ret < 0) {
4361         ret = 1;
4362     }
4363 
4364     if (opt.query_config_mode == DFL_QUERY_CONFIG_MODE) {
4365         mbedtls_exit(ret);
4366     } else {
4367         mbedtls_exit(query_config_ret);
4368     }
4369 }
4370 #endif /* !MBEDTLS_SSL_TEST_IMPOSSIBLE && MBEDTLS_SSL_SRV_C */
4371