1 /*
2 * Copyright 2016-2020 The OpenSSL Project Authors. All Rights Reserved.
3 *
4 * Licensed under the OpenSSL license (the "License"). You may not use
5 * this file except in compliance with the License. You can obtain a copy
6 * in the file LICENSE in the source distribution or at
7 * https://www.openssl.org/source/license.html
8 */
9
10 #include <string.h>
11 #include <openssl/bio.h>
12 #include <openssl/crypto.h>
13 #include <openssl/ssl.h>
14 #include <openssl/err.h>
15
16 #include "ssltestlib.h"
17 #include "testutil.h"
18
19 static char *cert = NULL;
20 static char *privkey = NULL;
21 static unsigned int timer_cb_count;
22
23 #define NUM_TESTS 2
24
25
26 #define DUMMY_CERT_STATUS_LEN 12
27
28 static unsigned char certstatus[] = {
29 SSL3_RT_HANDSHAKE, /* Content type */
30 0xfe, 0xfd, /* Record version */
31 0, 1, /* Epoch */
32 0, 0, 0, 0, 0, 0x0f, /* Record sequence number */
33 0, DTLS1_HM_HEADER_LENGTH + DUMMY_CERT_STATUS_LEN - 2,
34 SSL3_MT_CERTIFICATE_STATUS, /* Cert Status handshake message type */
35 0, 0, DUMMY_CERT_STATUS_LEN, /* Message len */
36 0, 5, /* Message sequence */
37 0, 0, 0, /* Fragment offset */
38 0, 0, DUMMY_CERT_STATUS_LEN - 2, /* Fragment len */
39 0x80, 0x80, 0x80, 0x80, 0x80,
40 0x80, 0x80, 0x80, 0x80, 0x80 /* Dummy data */
41 };
42
43 #define RECORD_SEQUENCE 10
44
timer_cb(SSL * s,unsigned int timer_us)45 static unsigned int timer_cb(SSL *s, unsigned int timer_us)
46 {
47 ++timer_cb_count;
48
49 if (timer_us == 0)
50 return 50000;
51 else
52 return 2 * timer_us;
53 }
54
test_dtls_unprocessed(int testidx)55 static int test_dtls_unprocessed(int testidx)
56 {
57 SSL_CTX *sctx = NULL, *cctx = NULL;
58 SSL *serverssl1 = NULL, *clientssl1 = NULL;
59 BIO *c_to_s_fbio, *c_to_s_mempacket;
60 int testresult = 0;
61
62 timer_cb_count = 0;
63
64 if (!TEST_true(create_ssl_ctx_pair(DTLS_server_method(),
65 DTLS_client_method(),
66 DTLS1_VERSION, DTLS_MAX_VERSION,
67 &sctx, &cctx, cert, privkey)))
68 return 0;
69
70 if (!TEST_true(SSL_CTX_set_cipher_list(cctx, "AES128-SHA")))
71 goto end;
72
73 c_to_s_fbio = BIO_new(bio_f_tls_dump_filter());
74 if (!TEST_ptr(c_to_s_fbio))
75 goto end;
76
77 /* BIO is freed by create_ssl_connection on error */
78 if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl1, &clientssl1,
79 NULL, c_to_s_fbio)))
80 goto end;
81
82 DTLS_set_timer_cb(clientssl1, timer_cb);
83
84 if (testidx == 1)
85 certstatus[RECORD_SEQUENCE] = 0xff;
86
87 /*
88 * Inject a dummy record from the next epoch. In test 0, this should never
89 * get used because the message sequence number is too big. In test 1 we set
90 * the record sequence number to be way off in the future.
91 */
92 c_to_s_mempacket = SSL_get_wbio(clientssl1);
93 c_to_s_mempacket = BIO_next(c_to_s_mempacket);
94 mempacket_test_inject(c_to_s_mempacket, (char *)certstatus,
95 sizeof(certstatus), 1, INJECT_PACKET_IGNORE_REC_SEQ);
96
97 /*
98 * Create the connection. We use "create_bare_ssl_connection" here so that
99 * we can force the connection to not do "SSL_read" once partly connected.
100 * We don't want to accidentally read the dummy records we injected because
101 * they will fail to decrypt.
102 */
103 if (!TEST_true(create_bare_ssl_connection(serverssl1, clientssl1,
104 SSL_ERROR_NONE, 0)))
105 goto end;
106
107 if (timer_cb_count == 0) {
108 printf("timer_callback was not called.\n");
109 goto end;
110 }
111
112 testresult = 1;
113 end:
114 SSL_free(serverssl1);
115 SSL_free(clientssl1);
116 SSL_CTX_free(sctx);
117 SSL_CTX_free(cctx);
118
119 return testresult;
120 }
121
122 #define CLI_TO_SRV_EPOCH_0_RECS 3
123 #define CLI_TO_SRV_EPOCH_1_RECS 1
124 #if !defined(OPENSSL_NO_EC) || !defined(OPENSSL_NO_DH)
125 # define SRV_TO_CLI_EPOCH_0_RECS 10
126 #else
127 /*
128 * In this case we have no ServerKeyExchange message, because we don't have
129 * ECDHE or DHE. When it is present it gets fragmented into 3 records in this
130 * test.
131 */
132 # define SRV_TO_CLI_EPOCH_0_RECS 9
133 #endif
134 #define SRV_TO_CLI_EPOCH_1_RECS 1
135 #define TOTAL_FULL_HAND_RECORDS \
136 (CLI_TO_SRV_EPOCH_0_RECS + CLI_TO_SRV_EPOCH_1_RECS + \
137 SRV_TO_CLI_EPOCH_0_RECS + SRV_TO_CLI_EPOCH_1_RECS)
138
139 #define CLI_TO_SRV_RESUME_EPOCH_0_RECS 3
140 #define CLI_TO_SRV_RESUME_EPOCH_1_RECS 1
141 #define SRV_TO_CLI_RESUME_EPOCH_0_RECS 2
142 #define SRV_TO_CLI_RESUME_EPOCH_1_RECS 1
143 #define TOTAL_RESUME_HAND_RECORDS \
144 (CLI_TO_SRV_RESUME_EPOCH_0_RECS + CLI_TO_SRV_RESUME_EPOCH_1_RECS + \
145 SRV_TO_CLI_RESUME_EPOCH_0_RECS + SRV_TO_CLI_RESUME_EPOCH_1_RECS)
146
147 #define TOTAL_RECORDS (TOTAL_FULL_HAND_RECORDS + TOTAL_RESUME_HAND_RECORDS)
148
test_dtls_drop_records(int idx)149 static int test_dtls_drop_records(int idx)
150 {
151 SSL_CTX *sctx = NULL, *cctx = NULL;
152 SSL *serverssl = NULL, *clientssl = NULL;
153 BIO *c_to_s_fbio, *mempackbio;
154 int testresult = 0;
155 int epoch = 0;
156 SSL_SESSION *sess = NULL;
157 int cli_to_srv_epoch0, cli_to_srv_epoch1, srv_to_cli_epoch0;
158
159 if (!TEST_true(create_ssl_ctx_pair(DTLS_server_method(),
160 DTLS_client_method(),
161 DTLS1_VERSION, DTLS_MAX_VERSION,
162 &sctx, &cctx, cert, privkey)))
163 return 0;
164
165 if (idx >= TOTAL_FULL_HAND_RECORDS) {
166 /* We're going to do a resumption handshake. Get a session first. */
167 if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl, &clientssl,
168 NULL, NULL))
169 || !TEST_true(create_ssl_connection(serverssl, clientssl,
170 SSL_ERROR_NONE))
171 || !TEST_ptr(sess = SSL_get1_session(clientssl)))
172 goto end;
173
174 SSL_shutdown(clientssl);
175 SSL_shutdown(serverssl);
176 SSL_free(serverssl);
177 SSL_free(clientssl);
178 serverssl = clientssl = NULL;
179
180 cli_to_srv_epoch0 = CLI_TO_SRV_RESUME_EPOCH_0_RECS;
181 cli_to_srv_epoch1 = CLI_TO_SRV_RESUME_EPOCH_1_RECS;
182 srv_to_cli_epoch0 = SRV_TO_CLI_RESUME_EPOCH_0_RECS;
183 idx -= TOTAL_FULL_HAND_RECORDS;
184 } else {
185 cli_to_srv_epoch0 = CLI_TO_SRV_EPOCH_0_RECS;
186 cli_to_srv_epoch1 = CLI_TO_SRV_EPOCH_1_RECS;
187 srv_to_cli_epoch0 = SRV_TO_CLI_EPOCH_0_RECS;
188 }
189
190 c_to_s_fbio = BIO_new(bio_f_tls_dump_filter());
191 if (!TEST_ptr(c_to_s_fbio))
192 goto end;
193
194 /* BIO is freed by create_ssl_connection on error */
195 if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl, &clientssl,
196 NULL, c_to_s_fbio)))
197 goto end;
198
199 if (sess != NULL) {
200 if (!TEST_true(SSL_set_session(clientssl, sess)))
201 goto end;
202 }
203
204 DTLS_set_timer_cb(clientssl, timer_cb);
205 DTLS_set_timer_cb(serverssl, timer_cb);
206
207 /* Work out which record to drop based on the test number */
208 if (idx >= cli_to_srv_epoch0 + cli_to_srv_epoch1) {
209 mempackbio = SSL_get_wbio(serverssl);
210 idx -= cli_to_srv_epoch0 + cli_to_srv_epoch1;
211 if (idx >= srv_to_cli_epoch0) {
212 epoch = 1;
213 idx -= srv_to_cli_epoch0;
214 }
215 } else {
216 mempackbio = SSL_get_wbio(clientssl);
217 if (idx >= cli_to_srv_epoch0) {
218 epoch = 1;
219 idx -= cli_to_srv_epoch0;
220 }
221 mempackbio = BIO_next(mempackbio);
222 }
223 BIO_ctrl(mempackbio, MEMPACKET_CTRL_SET_DROP_EPOCH, epoch, NULL);
224 BIO_ctrl(mempackbio, MEMPACKET_CTRL_SET_DROP_REC, idx, NULL);
225
226 if (!TEST_true(create_ssl_connection(serverssl, clientssl, SSL_ERROR_NONE)))
227 goto end;
228
229 if (sess != NULL && !TEST_true(SSL_session_reused(clientssl)))
230 goto end;
231
232 /* If the test did what we planned then it should have dropped a record */
233 if (!TEST_int_eq((int)BIO_ctrl(mempackbio, MEMPACKET_CTRL_GET_DROP_REC, 0,
234 NULL), -1))
235 goto end;
236
237 testresult = 1;
238 end:
239 SSL_SESSION_free(sess);
240 SSL_free(serverssl);
241 SSL_free(clientssl);
242 SSL_CTX_free(sctx);
243 SSL_CTX_free(cctx);
244
245 return testresult;
246 }
247
248 static const char dummy_cookie[] = "0123456";
249
generate_cookie_cb(SSL * ssl,unsigned char * cookie,unsigned int * cookie_len)250 static int generate_cookie_cb(SSL *ssl, unsigned char *cookie,
251 unsigned int *cookie_len)
252 {
253 memcpy(cookie, dummy_cookie, sizeof(dummy_cookie));
254 *cookie_len = sizeof(dummy_cookie);
255 return 1;
256 }
257
verify_cookie_cb(SSL * ssl,const unsigned char * cookie,unsigned int cookie_len)258 static int verify_cookie_cb(SSL *ssl, const unsigned char *cookie,
259 unsigned int cookie_len)
260 {
261 return TEST_mem_eq(cookie, cookie_len, dummy_cookie, sizeof(dummy_cookie));
262 }
263
test_cookie(void)264 static int test_cookie(void)
265 {
266 SSL_CTX *sctx = NULL, *cctx = NULL;
267 SSL *serverssl = NULL, *clientssl = NULL;
268 int testresult = 0;
269
270 if (!TEST_true(create_ssl_ctx_pair(DTLS_server_method(),
271 DTLS_client_method(),
272 DTLS1_VERSION, DTLS_MAX_VERSION,
273 &sctx, &cctx, cert, privkey)))
274 return 0;
275
276 SSL_CTX_set_options(sctx, SSL_OP_COOKIE_EXCHANGE);
277 SSL_CTX_set_cookie_generate_cb(sctx, generate_cookie_cb);
278 SSL_CTX_set_cookie_verify_cb(sctx, verify_cookie_cb);
279
280 if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl, &clientssl,
281 NULL, NULL))
282 || !TEST_true(create_ssl_connection(serverssl, clientssl,
283 SSL_ERROR_NONE)))
284 goto end;
285
286 testresult = 1;
287 end:
288 SSL_free(serverssl);
289 SSL_free(clientssl);
290 SSL_CTX_free(sctx);
291 SSL_CTX_free(cctx);
292
293 return testresult;
294 }
295
test_dtls_duplicate_records(void)296 static int test_dtls_duplicate_records(void)
297 {
298 SSL_CTX *sctx = NULL, *cctx = NULL;
299 SSL *serverssl = NULL, *clientssl = NULL;
300 int testresult = 0;
301
302 if (!TEST_true(create_ssl_ctx_pair(DTLS_server_method(),
303 DTLS_client_method(),
304 DTLS1_VERSION, DTLS_MAX_VERSION,
305 &sctx, &cctx, cert, privkey)))
306 return 0;
307
308 if (!TEST_true(create_ssl_objects(sctx, cctx, &serverssl, &clientssl,
309 NULL, NULL)))
310 goto end;
311
312 DTLS_set_timer_cb(clientssl, timer_cb);
313 DTLS_set_timer_cb(serverssl, timer_cb);
314
315 BIO_ctrl(SSL_get_wbio(clientssl), MEMPACKET_CTRL_SET_DUPLICATE_REC, 1, NULL);
316 BIO_ctrl(SSL_get_wbio(serverssl), MEMPACKET_CTRL_SET_DUPLICATE_REC, 1, NULL);
317
318 if (!TEST_true(create_ssl_connection(serverssl, clientssl, SSL_ERROR_NONE)))
319 goto end;
320
321 testresult = 1;
322 end:
323 SSL_free(serverssl);
324 SSL_free(clientssl);
325 SSL_CTX_free(sctx);
326 SSL_CTX_free(cctx);
327
328 return testresult;
329 }
330
setup_tests(void)331 int setup_tests(void)
332 {
333 if (!TEST_ptr(cert = test_get_argument(0))
334 || !TEST_ptr(privkey = test_get_argument(1)))
335 return 0;
336
337 ADD_ALL_TESTS(test_dtls_unprocessed, NUM_TESTS);
338 ADD_ALL_TESTS(test_dtls_drop_records, TOTAL_RECORDS);
339 ADD_TEST(test_cookie);
340 ADD_TEST(test_dtls_duplicate_records);
341
342 return 1;
343 }
344
cleanup_tests(void)345 void cleanup_tests(void)
346 {
347 bio_f_tls_dump_filter_free();
348 bio_s_mempacket_test_free();
349 }
350