Coverage Report

Created: 2024-07-27 06:39

/src/openssl32/ssl/quic/quic_cfq.c
Line
Count
Source (jump to first uncovered line)
1
/*
2
 * Copyright 2022-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_cfq.h"
11
#include "internal/numbers.h"
12
13
typedef struct quic_cfq_item_ex_st QUIC_CFQ_ITEM_EX;
14
15
struct quic_cfq_item_ex_st {
16
    QUIC_CFQ_ITEM           public;
17
    QUIC_CFQ_ITEM_EX       *prev, *next;
18
    unsigned char          *encoded;
19
    cfq_free_cb            *free_cb;
20
    void                   *free_cb_arg;
21
    uint64_t                frame_type;
22
    size_t                  encoded_len;
23
    uint32_t                priority, pn_space, flags;
24
    int                     state;
25
};
26
27
uint64_t ossl_quic_cfq_item_get_frame_type(const QUIC_CFQ_ITEM *item)
28
202M
{
29
202M
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
30
31
202M
    return ex->frame_type;
32
202M
}
33
34
const unsigned char *ossl_quic_cfq_item_get_encoded(const QUIC_CFQ_ITEM *item)
35
198M
{
36
198M
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
37
38
198M
    return ex->encoded;
39
198M
}
40
41
size_t ossl_quic_cfq_item_get_encoded_len(const QUIC_CFQ_ITEM *item)
42
198M
{
43
198M
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
44
45
198M
    return ex->encoded_len;
46
198M
}
47
48
int ossl_quic_cfq_item_get_state(const QUIC_CFQ_ITEM *item)
49
0
{
50
0
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
51
52
0
    return ex->state;
53
0
}
54
55
uint32_t ossl_quic_cfq_item_get_pn_space(const QUIC_CFQ_ITEM *item)
56
0
{
57
0
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
58
59
0
    return ex->pn_space;
60
0
}
61
62
int ossl_quic_cfq_item_is_unreliable(const QUIC_CFQ_ITEM *item)
63
22.1k
{
64
22.1k
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
65
66
22.1k
    return (ex->flags & QUIC_CFQ_ITEM_FLAG_UNRELIABLE) != 0;
67
22.1k
}
68
69
typedef struct quic_cfq_item_list_st {
70
    QUIC_CFQ_ITEM_EX *head, *tail;
71
} QUIC_CFQ_ITEM_LIST;
72
73
struct quic_cfq_st {
74
    /* 
75
     * Invariant: A CFQ item is always in exactly one of these lists, never more
76
     * or less than one.
77
     *
78
     * Invariant: The list the CFQ item is determined exactly by the state field
79
     * of the item.
80
     */
81
    QUIC_CFQ_ITEM_LIST                      new_list, tx_list, free_list;
82
};
83
84
static int compare(const QUIC_CFQ_ITEM_EX *a, const QUIC_CFQ_ITEM_EX *b)
85
226k
{
86
226k
    if (a->pn_space < b->pn_space)
87
0
        return -1;
88
226k
    else if (a->pn_space > b->pn_space)
89
0
        return 1;
90
91
226k
    if (a->priority > b->priority)
92
85.3k
        return -1;
93
140k
    else if (a->priority < b->priority)
94
467
        return 1;
95
96
140k
    return 0;
97
226k
}
98
99
static void list_remove(QUIC_CFQ_ITEM_LIST *l, QUIC_CFQ_ITEM_EX *n)
100
388k
{
101
388k
    if (l->head == n)
102
266k
        l->head = n->next;
103
388k
    if (l->tail == n)
104
225k
        l->tail = n->prev;
105
388k
    if (n->prev != NULL)
106
121k
        n->prev->next = n->next;
107
388k
    if (n->next != NULL)
108
162k
        n->next->prev = n->prev;
109
388k
    n->prev = n->next = NULL;
110
388k
}
111
112
static void list_insert_head(QUIC_CFQ_ITEM_LIST *l, QUIC_CFQ_ITEM_EX *n)
113
101k
{
114
101k
    n->next = l->head;
115
101k
    n->prev = NULL;
116
101k
    l->head = n;
117
101k
    if (n->next != NULL)
118
101k
        n->next->prev = n;
119
101k
    if (l->tail == NULL)
120
0
        l->tail = n;
121
101k
}
122
123
static void list_insert_tail(QUIC_CFQ_ITEM_LIST *l, QUIC_CFQ_ITEM_EX *n)
124
379k
{
125
379k
    n->prev = l->tail;
126
379k
    n->next = NULL;
127
379k
    l->tail = n;
128
379k
    if (n->prev != NULL)
129
237k
        n->prev->next = n;
130
379k
    if (l->head == NULL)
131
142k
        l->head = n;
132
379k
}
133
134
static void list_insert_after(QUIC_CFQ_ITEM_LIST *l,
135
                              QUIC_CFQ_ITEM_EX *ref,
136
                              QUIC_CFQ_ITEM_EX *n)
137
39.5k
{
138
39.5k
    n->prev = ref;
139
39.5k
    n->next = ref->next;
140
39.5k
    if (ref->next != NULL)
141
39.5k
        ref->next->prev = n;
142
39.5k
    ref->next = n;
143
39.5k
    if (l->tail == ref)
144
0
        l->tail = n;
145
39.5k
}
146
147
static void list_insert_sorted(QUIC_CFQ_ITEM_LIST *l, QUIC_CFQ_ITEM_EX *n,
148
                               int (*cmp)(const QUIC_CFQ_ITEM_EX *a,
149
                                          const QUIC_CFQ_ITEM_EX *b))
150
146k
{
151
146k
    QUIC_CFQ_ITEM_EX *p = l->head, *pprev = NULL;
152
153
146k
    if (p == NULL) {
154
4.68k
        l->head = l->tail = n;
155
4.68k
        n->prev = n->next = NULL;
156
4.68k
        return;
157
4.68k
    }
158
159
227k
    for (; p != NULL && cmp(p, n) < 0; pprev = p, p = p->next);
160
161
141k
    if (p == NULL)
162
1.12k
        list_insert_tail(l, n);
163
140k
    else if (pprev == NULL)
164
101k
        list_insert_head(l, n);
165
39.5k
    else
166
39.5k
        list_insert_after(l, pprev, n);
167
141k
}
168
169
QUIC_CFQ *ossl_quic_cfq_new(void)
170
21.1k
{
171
21.1k
    QUIC_CFQ *cfq = OPENSSL_zalloc(sizeof(*cfq));
172
173
21.1k
    if (cfq == NULL)
174
0
        return NULL;
175
176
21.1k
    return cfq;
177
21.1k
}
178
179
static void clear_item(QUIC_CFQ_ITEM_EX *item)
180
257k
{
181
257k
    if (item->free_cb != NULL) {
182
145k
        item->free_cb(item->encoded, item->encoded_len, item->free_cb_arg);
183
184
145k
        item->free_cb       = NULL;
185
145k
        item->encoded       = NULL;
186
145k
        item->encoded_len   = 0;
187
145k
    }
188
189
257k
    item->state = -1;
190
257k
}
191
192
static void free_list_items(QUIC_CFQ_ITEM_LIST *l)
193
63.5k
{
194
63.5k
    QUIC_CFQ_ITEM_EX *p, *pnext;
195
196
200k
    for (p = l->head; p != NULL; p = pnext) {
197
137k
        pnext = p->next;
198
137k
        clear_item(p);
199
137k
        OPENSSL_free(p);
200
137k
    }
201
63.5k
}
202
203
void ossl_quic_cfq_free(QUIC_CFQ *cfq)
204
21.1k
{
205
21.1k
    if (cfq == NULL)
206
0
        return;
207
208
21.1k
    free_list_items(&cfq->new_list);
209
21.1k
    free_list_items(&cfq->tx_list);
210
21.1k
    free_list_items(&cfq->free_list);
211
21.1k
    OPENSSL_free(cfq);
212
21.1k
}
213
214
static QUIC_CFQ_ITEM_EX *cfq_get_free(QUIC_CFQ *cfq)
215
145k
{
216
145k
    QUIC_CFQ_ITEM_EX *item = cfq->free_list.head;
217
218
145k
    if (item != NULL)
219
7.81k
        return item;
220
221
137k
    item = OPENSSL_zalloc(sizeof(*item));
222
137k
    if (item == NULL)
223
0
        return NULL;
224
225
137k
    item->state = -1;
226
137k
    list_insert_tail(&cfq->free_list, item);
227
137k
    return item;
228
137k
}
229
230
QUIC_CFQ_ITEM *ossl_quic_cfq_add_frame(QUIC_CFQ            *cfq,
231
                                       uint32_t             priority,
232
                                       uint32_t             pn_space,
233
                                       uint64_t             frame_type,
234
                                       uint32_t             flags,
235
                                       const unsigned char *encoded,
236
                                       size_t               encoded_len,
237
                                       cfq_free_cb         *free_cb,
238
                                       void                *free_cb_arg)
239
145k
{
240
145k
    QUIC_CFQ_ITEM_EX *item = cfq_get_free(cfq);
241
242
145k
    if (item == NULL)
243
0
        return NULL;
244
245
145k
    item->priority      = priority;
246
145k
    item->frame_type    = frame_type;
247
145k
    item->pn_space      = pn_space;
248
145k
    item->encoded       = (unsigned char *)encoded;
249
145k
    item->encoded_len   = encoded_len;
250
145k
    item->free_cb       = free_cb;
251
145k
    item->free_cb_arg   = free_cb_arg;
252
253
145k
    item->state = QUIC_CFQ_STATE_NEW;
254
145k
    item->flags = flags;
255
145k
    list_remove(&cfq->free_list, item);
256
145k
    list_insert_sorted(&cfq->new_list, item, compare);
257
145k
    return &item->public;
258
145k
}
259
260
void ossl_quic_cfq_mark_tx(QUIC_CFQ *cfq, QUIC_CFQ_ITEM *item)
261
121k
{
262
121k
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
263
264
121k
    switch (ex->state) {
265
121k
    case QUIC_CFQ_STATE_NEW:
266
121k
        list_remove(&cfq->new_list, ex);
267
121k
        list_insert_tail(&cfq->tx_list, ex);
268
121k
        ex->state = QUIC_CFQ_STATE_TX;
269
121k
        break;
270
0
    case QUIC_CFQ_STATE_TX:
271
0
        break; /* nothing to do */
272
0
    default:
273
0
        assert(0); /* invalid state (e.g. in free state) */
274
0
        break;
275
121k
    }
276
121k
}
277
278
void ossl_quic_cfq_mark_lost(QUIC_CFQ *cfq, QUIC_CFQ_ITEM *item,
279
                             uint32_t priority)
280
22.1k
{
281
22.1k
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
282
283
22.1k
    if (ossl_quic_cfq_item_is_unreliable(item)) {
284
20.6k
        ossl_quic_cfq_release(cfq, item);
285
20.6k
        return;
286
20.6k
    }
287
288
1.53k
    switch (ex->state) {
289
0
    case QUIC_CFQ_STATE_NEW:
290
0
        if (priority != UINT32_MAX && priority != ex->priority) {
291
0
            list_remove(&cfq->new_list, ex);
292
0
            ex->priority = priority;
293
0
            list_insert_sorted(&cfq->new_list, ex, compare);
294
0
        }
295
0
        break; /* nothing to do */
296
1.53k
    case QUIC_CFQ_STATE_TX:
297
1.53k
        if (priority != UINT32_MAX)
298
0
            ex->priority = priority;
299
1.53k
        list_remove(&cfq->tx_list, ex);
300
1.53k
        list_insert_sorted(&cfq->new_list, ex, compare);
301
1.53k
        ex->state = QUIC_CFQ_STATE_NEW;
302
1.53k
        break;
303
0
    default:
304
0
        assert(0); /* invalid state (e.g. in free state) */
305
0
        break;
306
1.53k
    }
307
1.53k
}
308
309
/*
310
 * Releases a CFQ item. The item may be in either state (NEW or TX) prior to the
311
 * call. The QUIC_CFQ_ITEM pointer must not be used following this call.
312
 */
313
void ossl_quic_cfq_release(QUIC_CFQ *cfq, QUIC_CFQ_ITEM *item)
314
119k
{
315
119k
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
316
317
119k
    switch (ex->state) {
318
0
    case QUIC_CFQ_STATE_NEW:
319
0
        list_remove(&cfq->new_list, ex);
320
0
        list_insert_tail(&cfq->free_list, ex);
321
0
        clear_item(ex);
322
0
        break;
323
119k
    case QUIC_CFQ_STATE_TX:
324
119k
        list_remove(&cfq->tx_list, ex);
325
119k
        list_insert_tail(&cfq->free_list, ex);
326
119k
        clear_item(ex);
327
119k
        break;
328
0
    default:
329
0
        assert(0); /* invalid state (e.g. in free state) */
330
0
        break;
331
119k
    }
332
119k
}
333
334
QUIC_CFQ_ITEM *ossl_quic_cfq_get_priority_head(const QUIC_CFQ *cfq,
335
                                               uint32_t pn_space)
336
34.7M
{
337
34.7M
    QUIC_CFQ_ITEM_EX *item = cfq->new_list.head;
338
339
84.5M
    for (; item != NULL && item->pn_space != pn_space; item = item->next);
340
341
34.7M
    if (item == NULL)
342
32.7M
        return NULL;
343
344
2.03M
    return &item->public;
345
34.7M
}
346
347
QUIC_CFQ_ITEM *ossl_quic_cfq_item_get_priority_next(const QUIC_CFQ_ITEM *item,
348
                                                    uint32_t pn_space)
349
202M
{
350
202M
    QUIC_CFQ_ITEM_EX *ex = (QUIC_CFQ_ITEM_EX *)item;
351
352
202M
    if (ex == NULL)
353
0
        return NULL;
354
355
202M
     ex = ex->next;
356
357
202M
     for (; ex != NULL && ex->pn_space != pn_space; ex = ex->next);
358
359
202M
     if (ex == NULL)
360
2.03M
         return NULL; /* ubsan */
361
362
200M
     return &ex->public;
363
202M
}