1*e7be843bSPierre Pronchery /*
2*e7be843bSPierre Pronchery * Copyright 2023 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_lcidm.h"
11*e7be843bSPierre Pronchery #include "testutil.h"
12*e7be843bSPierre Pronchery
13*e7be843bSPierre Pronchery static char ptrs[8];
14*e7be843bSPierre Pronchery
15*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_1 = { 8, { 1 } };
16*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_2 = { 8, { 2 } };
17*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_3 = { 8, { 3 } };
18*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_4 = { 8, { 4 } };
19*e7be843bSPierre Pronchery static const QUIC_CONN_ID cid8_5 = { 8, { 5 } };
20*e7be843bSPierre Pronchery
test_lcidm(void)21*e7be843bSPierre Pronchery static int test_lcidm(void)
22*e7be843bSPierre Pronchery {
23*e7be843bSPierre Pronchery int testresult = 0;
24*e7be843bSPierre Pronchery QUIC_LCIDM *lcidm;
25*e7be843bSPierre Pronchery size_t lcid_len = 10; /* != ODCID len */
26*e7be843bSPierre Pronchery QUIC_CONN_ID lcid_1, lcid_dummy, lcid_init = {0};
27*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_NEW_CONN_ID ncid_frame_1, ncid_frame_2, ncid_frame_3;
28*e7be843bSPierre Pronchery void *opaque = NULL;
29*e7be843bSPierre Pronchery uint64_t seq_num = UINT64_MAX;
30*e7be843bSPierre Pronchery int did_retire = 0;
31*e7be843bSPierre Pronchery
32*e7be843bSPierre Pronchery if (!TEST_ptr(lcidm = ossl_quic_lcidm_new(NULL, lcid_len)))
33*e7be843bSPierre Pronchery goto err;
34*e7be843bSPierre Pronchery
35*e7be843bSPierre Pronchery if (!TEST_size_t_eq(ossl_quic_lcidm_get_lcid_len(lcidm), lcid_len))
36*e7be843bSPierre Pronchery goto err;
37*e7be843bSPierre Pronchery
38*e7be843bSPierre Pronchery if (!TEST_true(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_1))
39*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_2))
40*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 1, &cid8_1))
41*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 1), 0)
42*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 1, &cid8_3))
43*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 1, &cid8_4))
44*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 0), 1)
45*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 1), 1)
46*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_retire_odcid(lcidm, ptrs + 0))
47*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 0), 0)
48*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_1))
49*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_lcidm_enrol_odcid(lcidm, ptrs + 0, &cid8_5))
50*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 0), 0)
51*e7be843bSPierre Pronchery
52*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_generate_initial(lcidm, ptrs + 2, &lcid_1))
53*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 1)
54*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_lcidm_generate_initial(lcidm, ptrs + 2, &lcid_init))
55*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_generate(lcidm, ptrs + 2, &ncid_frame_1))
56*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_generate(lcidm, ptrs + 2, &ncid_frame_2))
57*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_generate(lcidm, ptrs + 2, &ncid_frame_3))
58*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 4)
59*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &lcid_1, &seq_num, &opaque))
60*e7be843bSPierre Pronchery || !TEST_ptr_eq(opaque, ptrs + 2)
61*e7be843bSPierre Pronchery || !TEST_uint64_t_eq(seq_num, 0)
62*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_1.conn_id,
63*e7be843bSPierre Pronchery &seq_num, &opaque))
64*e7be843bSPierre Pronchery || !TEST_ptr_eq(opaque, ptrs + 2)
65*e7be843bSPierre Pronchery || !TEST_uint64_t_eq(seq_num, 1)
66*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_2.conn_id,
67*e7be843bSPierre Pronchery &seq_num, &opaque))
68*e7be843bSPierre Pronchery || !TEST_ptr_eq(opaque, ptrs + 2)
69*e7be843bSPierre Pronchery || !TEST_uint64_t_eq(seq_num, 2)
70*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_3.conn_id,
71*e7be843bSPierre Pronchery &seq_num, &opaque))
72*e7be843bSPierre Pronchery || !TEST_ptr_eq(opaque, ptrs + 2)
73*e7be843bSPierre Pronchery || !TEST_uint64_t_eq(seq_num, 3)
74*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_retire(lcidm, ptrs + 2, 2, NULL,
75*e7be843bSPierre Pronchery &lcid_dummy, &seq_num, &did_retire))
76*e7be843bSPierre Pronchery || !TEST_true(did_retire)
77*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 3)
78*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_retire(lcidm, ptrs + 2, 2, NULL,
79*e7be843bSPierre Pronchery &lcid_dummy, &seq_num, &did_retire))
80*e7be843bSPierre Pronchery || !TEST_true(did_retire)
81*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 2)
82*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_retire(lcidm, ptrs + 2, 2, NULL,
83*e7be843bSPierre Pronchery &lcid_dummy, &seq_num, &did_retire))
84*e7be843bSPierre Pronchery || !TEST_false(did_retire)
85*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 2)
86*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_lcidm_lookup(lcidm, &lcid_init,
87*e7be843bSPierre Pronchery &seq_num, &opaque))
88*e7be843bSPierre Pronchery || !TEST_false(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_1.conn_id,
89*e7be843bSPierre Pronchery &seq_num, &opaque))
90*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_lookup(lcidm, &ncid_frame_2.conn_id,
91*e7be843bSPierre Pronchery &seq_num, &opaque))
92*e7be843bSPierre Pronchery || !TEST_true(ossl_quic_lcidm_cull(lcidm, ptrs + 2))
93*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_lcidm_get_num_active_lcid(lcidm, ptrs + 2), 0))
94*e7be843bSPierre Pronchery goto err;
95*e7be843bSPierre Pronchery
96*e7be843bSPierre Pronchery testresult = 1;
97*e7be843bSPierre Pronchery err:
98*e7be843bSPierre Pronchery ossl_quic_lcidm_free(lcidm);
99*e7be843bSPierre Pronchery return testresult;
100*e7be843bSPierre Pronchery }
101*e7be843bSPierre Pronchery
setup_tests(void)102*e7be843bSPierre Pronchery int setup_tests(void)
103*e7be843bSPierre Pronchery {
104*e7be843bSPierre Pronchery ADD_TEST(test_lcidm);
105*e7be843bSPierre Pronchery return 1;
106*e7be843bSPierre Pronchery }
107