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