1*e7be843bSPierre Pronchery /*
2*e7be843bSPierre Pronchery * Copyright 2022-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/packet.h"
11*e7be843bSPierre Pronchery #include "internal/quic_cfq.h"
12*e7be843bSPierre Pronchery #include "internal/quic_wire.h"
13*e7be843bSPierre Pronchery #include "testutil.h"
14*e7be843bSPierre Pronchery
15*e7be843bSPierre Pronchery static const unsigned char ref_buf[] = {
16*e7be843bSPierre Pronchery 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19
17*e7be843bSPierre Pronchery };
18*e7be843bSPierre Pronchery
19*e7be843bSPierre Pronchery static const uint32_t ref_priority[] = {
20*e7be843bSPierre Pronchery 90, 80, 70, 60, 95, 40, 94, 20, 10, 0
21*e7be843bSPierre Pronchery };
22*e7be843bSPierre Pronchery
23*e7be843bSPierre Pronchery static const uint32_t ref_pn_space[] = {
24*e7be843bSPierre Pronchery QUIC_PN_SPACE_INITIAL,
25*e7be843bSPierre Pronchery QUIC_PN_SPACE_HANDSHAKE,
26*e7be843bSPierre Pronchery QUIC_PN_SPACE_HANDSHAKE,
27*e7be843bSPierre Pronchery QUIC_PN_SPACE_INITIAL,
28*e7be843bSPierre Pronchery QUIC_PN_SPACE_INITIAL,
29*e7be843bSPierre Pronchery QUIC_PN_SPACE_INITIAL,
30*e7be843bSPierre Pronchery QUIC_PN_SPACE_INITIAL,
31*e7be843bSPierre Pronchery QUIC_PN_SPACE_INITIAL,
32*e7be843bSPierre Pronchery QUIC_PN_SPACE_APP,
33*e7be843bSPierre Pronchery QUIC_PN_SPACE_APP,
34*e7be843bSPierre Pronchery };
35*e7be843bSPierre Pronchery
36*e7be843bSPierre Pronchery static const uint64_t ref_frame_type[] = {
37*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_NEW_CONN_ID,
38*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_NEW_CONN_ID,
39*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_NEW_CONN_ID,
40*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_NEW_CONN_ID,
41*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_NEW_CONN_ID,
42*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_NEW_CONN_ID,
43*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_NEW_CONN_ID,
44*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_NEW_CONN_ID,
45*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_NEW_CONN_ID,
46*e7be843bSPierre Pronchery OSSL_QUIC_FRAME_TYPE_RETIRE_CONN_ID,
47*e7be843bSPierre Pronchery };
48*e7be843bSPierre Pronchery
49*e7be843bSPierre Pronchery static const uint32_t expect[QUIC_PN_SPACE_NUM][11] = {
50*e7be843bSPierre Pronchery { 4, 6, 0, 3, 5, 7, UINT32_MAX },
51*e7be843bSPierre Pronchery { 1, 2, UINT32_MAX },
52*e7be843bSPierre Pronchery { 8, 9, UINT32_MAX },
53*e7be843bSPierre Pronchery };
54*e7be843bSPierre Pronchery
55*e7be843bSPierre Pronchery static QUIC_CFQ_ITEM *items[QUIC_PN_SPACE_NUM][10];
56*e7be843bSPierre Pronchery
57*e7be843bSPierre Pronchery static unsigned char *g_free;
58*e7be843bSPierre Pronchery static size_t g_free_len;
59*e7be843bSPierre Pronchery
free_cb(unsigned char * buf,size_t buf_len,void * arg)60*e7be843bSPierre Pronchery static void free_cb(unsigned char *buf, size_t buf_len, void *arg)
61*e7be843bSPierre Pronchery {
62*e7be843bSPierre Pronchery g_free = buf;
63*e7be843bSPierre Pronchery g_free_len = buf_len;
64*e7be843bSPierre Pronchery }
65*e7be843bSPierre Pronchery
check(QUIC_CFQ * cfq)66*e7be843bSPierre Pronchery static int check(QUIC_CFQ *cfq)
67*e7be843bSPierre Pronchery {
68*e7be843bSPierre Pronchery int testresult = 0;
69*e7be843bSPierre Pronchery QUIC_CFQ_ITEM *item;
70*e7be843bSPierre Pronchery size_t i;
71*e7be843bSPierre Pronchery uint32_t pn_space;
72*e7be843bSPierre Pronchery
73*e7be843bSPierre Pronchery for (pn_space = QUIC_PN_SPACE_INITIAL; pn_space < QUIC_PN_SPACE_NUM; ++pn_space)
74*e7be843bSPierre Pronchery for (i = 0, item = ossl_quic_cfq_get_priority_head(cfq, pn_space);;
75*e7be843bSPierre Pronchery ++i, item = ossl_quic_cfq_item_get_priority_next(item, pn_space)) {
76*e7be843bSPierre Pronchery
77*e7be843bSPierre Pronchery if (expect[pn_space][i] == UINT32_MAX) {
78*e7be843bSPierre Pronchery if (!TEST_ptr_null(item))
79*e7be843bSPierre Pronchery goto err;
80*e7be843bSPierre Pronchery
81*e7be843bSPierre Pronchery break;
82*e7be843bSPierre Pronchery }
83*e7be843bSPierre Pronchery
84*e7be843bSPierre Pronchery items[pn_space][i] = item;
85*e7be843bSPierre Pronchery
86*e7be843bSPierre Pronchery if (!TEST_ptr(item)
87*e7be843bSPierre Pronchery || !TEST_ptr_eq(ossl_quic_cfq_item_get_encoded(item),
88*e7be843bSPierre Pronchery ref_buf + expect[pn_space][i])
89*e7be843bSPierre Pronchery || !TEST_int_eq(ossl_quic_cfq_item_get_pn_space(item), pn_space)
90*e7be843bSPierre Pronchery || !TEST_int_eq(ossl_quic_cfq_item_get_state(item),
91*e7be843bSPierre Pronchery QUIC_CFQ_STATE_NEW))
92*e7be843bSPierre Pronchery goto err;
93*e7be843bSPierre Pronchery }
94*e7be843bSPierre Pronchery
95*e7be843bSPierre Pronchery testresult = 1;
96*e7be843bSPierre Pronchery err:
97*e7be843bSPierre Pronchery return testresult;
98*e7be843bSPierre Pronchery }
99*e7be843bSPierre Pronchery
test_cfq(void)100*e7be843bSPierre Pronchery static int test_cfq(void)
101*e7be843bSPierre Pronchery {
102*e7be843bSPierre Pronchery int testresult = 0;
103*e7be843bSPierre Pronchery QUIC_CFQ *cfq = NULL;
104*e7be843bSPierre Pronchery QUIC_CFQ_ITEM *item, *inext;
105*e7be843bSPierre Pronchery size_t i;
106*e7be843bSPierre Pronchery uint32_t pn_space;
107*e7be843bSPierre Pronchery
108*e7be843bSPierre Pronchery if (!TEST_ptr(cfq = ossl_quic_cfq_new()))
109*e7be843bSPierre Pronchery goto err;
110*e7be843bSPierre Pronchery
111*e7be843bSPierre Pronchery g_free = NULL;
112*e7be843bSPierre Pronchery g_free_len = 0;
113*e7be843bSPierre Pronchery
114*e7be843bSPierre Pronchery for (i = 0; i < OSSL_NELEM(ref_buf); ++i) {
115*e7be843bSPierre Pronchery if (!TEST_ptr(item = ossl_quic_cfq_add_frame(cfq, ref_priority[i],
116*e7be843bSPierre Pronchery ref_pn_space[i],
117*e7be843bSPierre Pronchery ref_frame_type[i], 0,
118*e7be843bSPierre Pronchery ref_buf + i,
119*e7be843bSPierre Pronchery 1,
120*e7be843bSPierre Pronchery free_cb,
121*e7be843bSPierre Pronchery NULL))
122*e7be843bSPierre Pronchery || !TEST_int_eq(ossl_quic_cfq_item_get_state(item),
123*e7be843bSPierre Pronchery QUIC_CFQ_STATE_NEW)
124*e7be843bSPierre Pronchery || !TEST_uint_eq(ossl_quic_cfq_item_get_pn_space(item),
125*e7be843bSPierre Pronchery ref_pn_space[i])
126*e7be843bSPierre Pronchery || !TEST_uint64_t_eq(ossl_quic_cfq_item_get_frame_type(item),
127*e7be843bSPierre Pronchery ref_frame_type[i])
128*e7be843bSPierre Pronchery || !TEST_ptr_eq(ossl_quic_cfq_item_get_encoded(item),
129*e7be843bSPierre Pronchery ref_buf + i)
130*e7be843bSPierre Pronchery || !TEST_size_t_eq(ossl_quic_cfq_item_get_encoded_len(item),
131*e7be843bSPierre Pronchery 1))
132*e7be843bSPierre Pronchery goto err;
133*e7be843bSPierre Pronchery }
134*e7be843bSPierre Pronchery
135*e7be843bSPierre Pronchery if (!check(cfq))
136*e7be843bSPierre Pronchery goto err;
137*e7be843bSPierre Pronchery
138*e7be843bSPierre Pronchery for (pn_space = QUIC_PN_SPACE_INITIAL; pn_space < QUIC_PN_SPACE_NUM; ++pn_space)
139*e7be843bSPierre Pronchery for (item = ossl_quic_cfq_get_priority_head(cfq, pn_space);
140*e7be843bSPierre Pronchery item != NULL; item = inext) {
141*e7be843bSPierre Pronchery inext = ossl_quic_cfq_item_get_priority_next(item, pn_space);
142*e7be843bSPierre Pronchery
143*e7be843bSPierre Pronchery ossl_quic_cfq_mark_tx(cfq, item);
144*e7be843bSPierre Pronchery }
145*e7be843bSPierre Pronchery
146*e7be843bSPierre Pronchery for (pn_space = QUIC_PN_SPACE_INITIAL; pn_space < QUIC_PN_SPACE_NUM; ++pn_space)
147*e7be843bSPierre Pronchery if (!TEST_ptr_null(ossl_quic_cfq_get_priority_head(cfq, pn_space)))
148*e7be843bSPierre Pronchery goto err;
149*e7be843bSPierre Pronchery
150*e7be843bSPierre Pronchery for (pn_space = QUIC_PN_SPACE_INITIAL; pn_space < QUIC_PN_SPACE_NUM; ++pn_space)
151*e7be843bSPierre Pronchery for (i = 0; i < OSSL_NELEM(items[0]); ++i)
152*e7be843bSPierre Pronchery if (items[pn_space][i] != NULL)
153*e7be843bSPierre Pronchery ossl_quic_cfq_mark_lost(cfq, items[pn_space][i], UINT32_MAX);
154*e7be843bSPierre Pronchery
155*e7be843bSPierre Pronchery if (!check(cfq))
156*e7be843bSPierre Pronchery goto err;
157*e7be843bSPierre Pronchery
158*e7be843bSPierre Pronchery for (pn_space = QUIC_PN_SPACE_INITIAL; pn_space < QUIC_PN_SPACE_NUM; ++pn_space)
159*e7be843bSPierre Pronchery for (i = 0; i < OSSL_NELEM(items[0]); ++i)
160*e7be843bSPierre Pronchery if (items[pn_space][i] != NULL)
161*e7be843bSPierre Pronchery ossl_quic_cfq_release(cfq, items[pn_space][i]);
162*e7be843bSPierre Pronchery
163*e7be843bSPierre Pronchery for (pn_space = QUIC_PN_SPACE_INITIAL; pn_space < QUIC_PN_SPACE_NUM; ++pn_space)
164*e7be843bSPierre Pronchery if (!TEST_ptr_null(ossl_quic_cfq_get_priority_head(cfq, pn_space)))
165*e7be843bSPierre Pronchery goto err;
166*e7be843bSPierre Pronchery
167*e7be843bSPierre Pronchery testresult = 1;
168*e7be843bSPierre Pronchery err:
169*e7be843bSPierre Pronchery ossl_quic_cfq_free(cfq);
170*e7be843bSPierre Pronchery return testresult;
171*e7be843bSPierre Pronchery }
172*e7be843bSPierre Pronchery
setup_tests(void)173*e7be843bSPierre Pronchery int setup_tests(void)
174*e7be843bSPierre Pronchery {
175*e7be843bSPierre Pronchery ADD_TEST(test_cfq);
176*e7be843bSPierre Pronchery return 1;
177*e7be843bSPierre Pronchery }
178