1*e7be843bSPierre Pronchery /*
2*e7be843bSPierre Pronchery * Copyright 2023-2024 The OpenSSL Project Authors. All Rights Reserved.
3*e7be843bSPierre Pronchery *
4*e7be843bSPierre Pronchery * Licensed under the Apache License 2.0 (the "License"). You may not use
5*e7be843bSPierre Pronchery * this file except in compliance with the License. You can obtain a copy
6*e7be843bSPierre Pronchery * in the file LICENSE in the source distribution or at
7*e7be843bSPierre Pronchery * https://www.openssl.org/source/license.html
8*e7be843bSPierre Pronchery */
9*e7be843bSPierre Pronchery
10*e7be843bSPierre Pronchery #include "internal/quic_rcidm.h"
11*e7be843bSPierre Pronchery #include "testutil.h"
12*e7be843bSPierre Pronchery
13*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_1 = { 8, { 1 } };
14*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_2 = { 8, { 2 } };
15*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_3 = { 8, { 3 } };
16*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_4 = { 8, { 4 } };
17*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_5 = { 8, { 5 } };
18*e7be843bSPierre Pronchery
19*e7be843bSPierre Pronchery /*
20*e7be843bSPierre Pronchery * 0: Client, Initial ODCID
21*e7be843bSPierre Pronchery * 1: Client, Initial ODCID + Retry ODCID
22*e7be843bSPierre Pronchery * 2: Server, doesn't start with Initial ODCID
23*e7be843bSPierre Pronchery */
test_rcidm(int idx)24*e7be843bSPierre Pronchery static int test_rcidm(int idx)
25*e7be843bSPierre Pronchery {
26*e7be843bSPierre Pronchery int testresult = 0;
27*e7be843bSPierre Pronchery QUIC_RCIDM *rcidm;
28*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_NEW_CONN_ID ncid_frame_1 = {0}, ncid_frame_2 = {0};
29*e7be843bSPierre Pronchery QUIC_CONN_ID dcid_out;
30*e7be843bSPierre Pronchery const QUIC_CONN_ID *odcid = NULL;
31*e7be843bSPierre Pronchery uint64_t seq_num_out;
32*e7be843bSPierre Pronchery
33*e7be843bSPierre Pronchery ncid_frame_1.seq_num = 2;
34*e7be843bSPierre Pronchery ncid_frame_1.conn_id.id_len = 8;
35*e7be843bSPierre Pronchery ncid_frame_1.conn_id.id[0] = 3;
36*e7be843bSPierre Pronchery
37*e7be843bSPierre Pronchery ncid_frame_2.seq_num = 3;
38*e7be843bSPierre Pronchery ncid_frame_2.conn_id.id_len = 8;
39*e7be843bSPierre Pronchery ncid_frame_2.conn_id.id[0] = 4;
40*e7be843bSPierre Pronchery
41*e7be843bSPierre Pronchery odcid = ((idx == 2) ? NULL : &cid8_1);
42*e7be843bSPierre Pronchery if (!TEST_ptr(rcidm = ossl_quic_rcidm_new(odcid)))
43*e7be843bSPierre Pronchery goto err;
44*e7be843bSPierre Pronchery
45*e7be843bSPierre Pronchery if (idx != 2) {
46*e7be843bSPierre Pronchery if (/* ODCID not counted */
47*e7be843bSPierre Pronchery !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
48*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
49*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
50*e7be843bSPierre Pronchery
51*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_1))
52*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 0))
53*e7be843bSPierre Pronchery goto err;
54*e7be843bSPierre Pronchery } else {
55*e7be843bSPierre Pronchery if (!TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
56*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 0))
57*e7be843bSPierre Pronchery goto err;
58*e7be843bSPierre Pronchery }
59*e7be843bSPierre Pronchery
60*e7be843bSPierre Pronchery if (idx == 1) {
61*e7be843bSPierre Pronchery if (!TEST_true(ossl_quic_rcidm_add_from_server_retry(rcidm, &cid8_5))
62*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
63*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
64*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
65*e7be843bSPierre Pronchery
66*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_5))
67*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 0))
68*e7be843bSPierre Pronchery goto err;
69*e7be843bSPierre Pronchery }
70*e7be843bSPierre Pronchery
71*e7be843bSPierre Pronchery if (!TEST_true(ossl_quic_rcidm_add_from_initial(rcidm, &cid8_2))
72*e7be843bSPierre Pronchery /* Initial SCID (seq=0) is counted */
73*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 1)
74*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
75*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
76*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
77*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_2))
78*e7be843bSPierre Pronchery
79*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_add_from_ncid(rcidm, &ncid_frame_1))
80*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 2)
81*e7be843bSPierre Pronchery /* Not changed over yet - handshake not confirmed */
82*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
83*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
84*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_2))
85*e7be843bSPierre Pronchery
86*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_add_from_ncid(rcidm, &ncid_frame_2))
87*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_rcidm_get_num_active(rcidm), 3)
88*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_rcidm_get_num_retiring(rcidm), 0)
89*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_rcidm_pop_retire_seq_num(rcidm, &seq_num_out))
90*e7be843bSPierre Pronchery /* Not changed over yet - handshake not confirmed */
91*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 0))
92*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
93*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_2)))
94*e7be843bSPierre Pronchery goto err;
95*e7be843bSPierre Pronchery
96*e7be843bSPierre Pronchery ossl_quic_rcidm_on_handshake_complete(rcidm);
97*e7be843bSPierre Pronchery
98*e7be843bSPierre Pronchery if (!TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
99*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
100*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
101*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_3))
102*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_rcidm_get_num_retiring(rcidm), 1)
103*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_peek_retire_seq_num(rcidm, &seq_num_out))
104*e7be843bSPierre Pronchery || !TEST_uint64_t_eq(seq_num_out, 0))
105*e7be843bSPierre Pronchery goto err;
106*e7be843bSPierre Pronchery
107*e7be843bSPierre Pronchery ossl_quic_rcidm_request_roll(rcidm);
108*e7be843bSPierre Pronchery
109*e7be843bSPierre Pronchery if (!TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
110*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_rcidm_get_preferred_tx_dcid_changed(rcidm, 1))
111*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_get_preferred_tx_dcid(rcidm, &dcid_out))
112*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_conn_id_eq(&dcid_out, &cid8_4))
113*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_rcidm_get_num_retiring(rcidm), 2)
114*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_peek_retire_seq_num(rcidm, &seq_num_out))
115*e7be843bSPierre Pronchery || !TEST_uint64_t_eq(seq_num_out, 0)
116*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_pop_retire_seq_num(rcidm, &seq_num_out))
117*e7be843bSPierre Pronchery || !TEST_uint64_t_eq(seq_num_out, 0)
118*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_rcidm_pop_retire_seq_num(rcidm, &seq_num_out))
119*e7be843bSPierre Pronchery || !TEST_uint64_t_eq(seq_num_out, 2))
120*e7be843bSPierre Pronchery goto err;
121*e7be843bSPierre Pronchery
122*e7be843bSPierre Pronchery testresult = 1;
123*e7be843bSPierre Pronchery err:
124*e7be843bSPierre Pronchery ossl_quic_rcidm_free(rcidm);
125*e7be843bSPierre Pronchery return testresult;
126*e7be843bSPierre Pronchery }
127*e7be843bSPierre Pronchery
setup_tests(void)128*e7be843bSPierre Pronchery int setup_tests(void)
129*e7be843bSPierre Pronchery {
130*e7be843bSPierre Pronchery ADD_ALL_TESTS(test_rcidm, 3);
131*e7be843bSPierre Pronchery return 1;
132*e7be843bSPierre Pronchery }
133