1*e7be843bSPierre Pronchery /*
2*e7be843bSPierre Pronchery * Copyright 2022-2025 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_demux.h"
11*e7be843bSPierre Pronchery #include "internal/quic_wire_pkt.h"
12*e7be843bSPierre Pronchery #include "internal/common.h"
13*e7be843bSPierre Pronchery #include <openssl/lhash.h>
14*e7be843bSPierre Pronchery #include <openssl/err.h>
15*e7be843bSPierre Pronchery
16*e7be843bSPierre Pronchery #define URXE_DEMUX_STATE_FREE 0 /* on urx_free list */
17*e7be843bSPierre Pronchery #define URXE_DEMUX_STATE_PENDING 1 /* on urx_pending list */
18*e7be843bSPierre Pronchery #define URXE_DEMUX_STATE_ISSUED 2 /* on neither list */
19*e7be843bSPierre Pronchery
20*e7be843bSPierre Pronchery #define DEMUX_MAX_MSGS_PER_CALL 32
21*e7be843bSPierre Pronchery
22*e7be843bSPierre Pronchery #define DEMUX_DEFAULT_MTU 1500
23*e7be843bSPierre Pronchery
24*e7be843bSPierre Pronchery struct quic_demux_st {
25*e7be843bSPierre Pronchery /* The underlying transport BIO with datagram semantics. */
26*e7be843bSPierre Pronchery BIO *net_bio;
27*e7be843bSPierre Pronchery
28*e7be843bSPierre Pronchery /*
29*e7be843bSPierre Pronchery * QUIC short packets do not contain the length of the connection ID field,
30*e7be843bSPierre Pronchery * therefore it must be known contextually. The demuxer requires connection
31*e7be843bSPierre Pronchery * IDs of the same length to be used for all incoming packets.
32*e7be843bSPierre Pronchery */
33*e7be843bSPierre Pronchery size_t short_conn_id_len;
34*e7be843bSPierre Pronchery
35*e7be843bSPierre Pronchery /*
36*e7be843bSPierre Pronchery * Our current understanding of the upper bound on an incoming datagram size
37*e7be843bSPierre Pronchery * in bytes.
38*e7be843bSPierre Pronchery */
39*e7be843bSPierre Pronchery size_t mtu;
40*e7be843bSPierre Pronchery
41*e7be843bSPierre Pronchery /* The datagram_id to use for the next datagram we receive. */
42*e7be843bSPierre Pronchery uint64_t next_datagram_id;
43*e7be843bSPierre Pronchery
44*e7be843bSPierre Pronchery /* Time retrieval callback. */
45*e7be843bSPierre Pronchery OSSL_TIME (*now)(void *arg);
46*e7be843bSPierre Pronchery void *now_arg;
47*e7be843bSPierre Pronchery
48*e7be843bSPierre Pronchery /* The default packet handler, if any. */
49*e7be843bSPierre Pronchery ossl_quic_demux_cb_fn *default_cb;
50*e7be843bSPierre Pronchery void *default_cb_arg;
51*e7be843bSPierre Pronchery
52*e7be843bSPierre Pronchery /*
53*e7be843bSPierre Pronchery * List of URXEs which are not currently in use (i.e., not filled with
54*e7be843bSPierre Pronchery * unconsumed data). These are moved to the pending list as they are filled.
55*e7be843bSPierre Pronchery */
56*e7be843bSPierre Pronchery QUIC_URXE_LIST urx_free;
57*e7be843bSPierre Pronchery
58*e7be843bSPierre Pronchery /*
59*e7be843bSPierre Pronchery * List of URXEs which are filled with received encrypted data. These are
60*e7be843bSPierre Pronchery * removed from this list as we invoke the callbacks for each of them. They
61*e7be843bSPierre Pronchery * are then not on any list managed by us; we forget about them until our
62*e7be843bSPierre Pronchery * user calls ossl_quic_demux_release_urxe to return the URXE to us, at
63*e7be843bSPierre Pronchery * which point we add it to the free list.
64*e7be843bSPierre Pronchery */
65*e7be843bSPierre Pronchery QUIC_URXE_LIST urx_pending;
66*e7be843bSPierre Pronchery
67*e7be843bSPierre Pronchery /* Whether to use local address support. */
68*e7be843bSPierre Pronchery char use_local_addr;
69*e7be843bSPierre Pronchery };
70*e7be843bSPierre Pronchery
ossl_quic_demux_new(BIO * net_bio,size_t short_conn_id_len,OSSL_TIME (* now)(void * arg),void * now_arg)71*e7be843bSPierre Pronchery QUIC_DEMUX *ossl_quic_demux_new(BIO *net_bio,
72*e7be843bSPierre Pronchery size_t short_conn_id_len,
73*e7be843bSPierre Pronchery OSSL_TIME (*now)(void *arg),
74*e7be843bSPierre Pronchery void *now_arg)
75*e7be843bSPierre Pronchery {
76*e7be843bSPierre Pronchery QUIC_DEMUX *demux;
77*e7be843bSPierre Pronchery
78*e7be843bSPierre Pronchery demux = OPENSSL_zalloc(sizeof(QUIC_DEMUX));
79*e7be843bSPierre Pronchery if (demux == NULL)
80*e7be843bSPierre Pronchery return NULL;
81*e7be843bSPierre Pronchery
82*e7be843bSPierre Pronchery demux->net_bio = net_bio;
83*e7be843bSPierre Pronchery demux->short_conn_id_len = short_conn_id_len;
84*e7be843bSPierre Pronchery /* We update this if possible when we get a BIO. */
85*e7be843bSPierre Pronchery demux->mtu = DEMUX_DEFAULT_MTU;
86*e7be843bSPierre Pronchery demux->now = now;
87*e7be843bSPierre Pronchery demux->now_arg = now_arg;
88*e7be843bSPierre Pronchery
89*e7be843bSPierre Pronchery if (net_bio != NULL
90*e7be843bSPierre Pronchery && BIO_dgram_get_local_addr_cap(net_bio)
91*e7be843bSPierre Pronchery && BIO_dgram_set_local_addr_enable(net_bio, 1))
92*e7be843bSPierre Pronchery demux->use_local_addr = 1;
93*e7be843bSPierre Pronchery
94*e7be843bSPierre Pronchery return demux;
95*e7be843bSPierre Pronchery }
96*e7be843bSPierre Pronchery
demux_free_urxl(QUIC_URXE_LIST * l)97*e7be843bSPierre Pronchery static void demux_free_urxl(QUIC_URXE_LIST *l)
98*e7be843bSPierre Pronchery {
99*e7be843bSPierre Pronchery QUIC_URXE *e, *enext;
100*e7be843bSPierre Pronchery
101*e7be843bSPierre Pronchery for (e = ossl_list_urxe_head(l); e != NULL; e = enext) {
102*e7be843bSPierre Pronchery enext = ossl_list_urxe_next(e);
103*e7be843bSPierre Pronchery ossl_list_urxe_remove(l, e);
104*e7be843bSPierre Pronchery OPENSSL_free(e);
105*e7be843bSPierre Pronchery }
106*e7be843bSPierre Pronchery }
107*e7be843bSPierre Pronchery
ossl_quic_demux_free(QUIC_DEMUX * demux)108*e7be843bSPierre Pronchery void ossl_quic_demux_free(QUIC_DEMUX *demux)
109*e7be843bSPierre Pronchery {
110*e7be843bSPierre Pronchery if (demux == NULL)
111*e7be843bSPierre Pronchery return;
112*e7be843bSPierre Pronchery
113*e7be843bSPierre Pronchery /* Free all URXEs we are holding. */
114*e7be843bSPierre Pronchery demux_free_urxl(&demux->urx_free);
115*e7be843bSPierre Pronchery demux_free_urxl(&demux->urx_pending);
116*e7be843bSPierre Pronchery
117*e7be843bSPierre Pronchery OPENSSL_free(demux);
118*e7be843bSPierre Pronchery }
119*e7be843bSPierre Pronchery
ossl_quic_demux_set_bio(QUIC_DEMUX * demux,BIO * net_bio)120*e7be843bSPierre Pronchery void ossl_quic_demux_set_bio(QUIC_DEMUX *demux, BIO *net_bio)
121*e7be843bSPierre Pronchery {
122*e7be843bSPierre Pronchery unsigned int mtu;
123*e7be843bSPierre Pronchery
124*e7be843bSPierre Pronchery demux->net_bio = net_bio;
125*e7be843bSPierre Pronchery
126*e7be843bSPierre Pronchery if (net_bio != NULL) {
127*e7be843bSPierre Pronchery /*
128*e7be843bSPierre Pronchery * Try to determine our MTU if possible. The BIO is not required to
129*e7be843bSPierre Pronchery * support this, in which case we remain at the last known MTU, or our
130*e7be843bSPierre Pronchery * initial default.
131*e7be843bSPierre Pronchery */
132*e7be843bSPierre Pronchery mtu = BIO_dgram_get_mtu(net_bio);
133*e7be843bSPierre Pronchery if (mtu >= QUIC_MIN_INITIAL_DGRAM_LEN)
134*e7be843bSPierre Pronchery ossl_quic_demux_set_mtu(demux, mtu); /* best effort */
135*e7be843bSPierre Pronchery }
136*e7be843bSPierre Pronchery }
137*e7be843bSPierre Pronchery
ossl_quic_demux_set_mtu(QUIC_DEMUX * demux,unsigned int mtu)138*e7be843bSPierre Pronchery int ossl_quic_demux_set_mtu(QUIC_DEMUX *demux, unsigned int mtu)
139*e7be843bSPierre Pronchery {
140*e7be843bSPierre Pronchery if (mtu < QUIC_MIN_INITIAL_DGRAM_LEN)
141*e7be843bSPierre Pronchery return 0;
142*e7be843bSPierre Pronchery
143*e7be843bSPierre Pronchery demux->mtu = mtu;
144*e7be843bSPierre Pronchery return 1;
145*e7be843bSPierre Pronchery }
146*e7be843bSPierre Pronchery
ossl_quic_demux_set_default_handler(QUIC_DEMUX * demux,ossl_quic_demux_cb_fn * cb,void * cb_arg)147*e7be843bSPierre Pronchery void ossl_quic_demux_set_default_handler(QUIC_DEMUX *demux,
148*e7be843bSPierre Pronchery ossl_quic_demux_cb_fn *cb,
149*e7be843bSPierre Pronchery void *cb_arg)
150*e7be843bSPierre Pronchery {
151*e7be843bSPierre Pronchery demux->default_cb = cb;
152*e7be843bSPierre Pronchery demux->default_cb_arg = cb_arg;
153*e7be843bSPierre Pronchery }
154*e7be843bSPierre Pronchery
demux_alloc_urxe(size_t alloc_len)155*e7be843bSPierre Pronchery static QUIC_URXE *demux_alloc_urxe(size_t alloc_len)
156*e7be843bSPierre Pronchery {
157*e7be843bSPierre Pronchery QUIC_URXE *e;
158*e7be843bSPierre Pronchery
159*e7be843bSPierre Pronchery if (alloc_len >= SIZE_MAX - sizeof(QUIC_URXE))
160*e7be843bSPierre Pronchery return NULL;
161*e7be843bSPierre Pronchery
162*e7be843bSPierre Pronchery e = OPENSSL_malloc(sizeof(QUIC_URXE) + alloc_len);
163*e7be843bSPierre Pronchery if (e == NULL)
164*e7be843bSPierre Pronchery return NULL;
165*e7be843bSPierre Pronchery
166*e7be843bSPierre Pronchery ossl_list_urxe_init_elem(e);
167*e7be843bSPierre Pronchery e->alloc_len = alloc_len;
168*e7be843bSPierre Pronchery e->data_len = 0;
169*e7be843bSPierre Pronchery return e;
170*e7be843bSPierre Pronchery }
171*e7be843bSPierre Pronchery
demux_resize_urxe(QUIC_DEMUX * demux,QUIC_URXE * e,size_t new_alloc_len)172*e7be843bSPierre Pronchery static QUIC_URXE *demux_resize_urxe(QUIC_DEMUX *demux, QUIC_URXE *e,
173*e7be843bSPierre Pronchery size_t new_alloc_len)
174*e7be843bSPierre Pronchery {
175*e7be843bSPierre Pronchery QUIC_URXE *e2, *prev;
176*e7be843bSPierre Pronchery
177*e7be843bSPierre Pronchery if (!ossl_assert(e->demux_state == URXE_DEMUX_STATE_FREE))
178*e7be843bSPierre Pronchery /* Never attempt to resize a URXE which is not on the free list. */
179*e7be843bSPierre Pronchery return NULL;
180*e7be843bSPierre Pronchery
181*e7be843bSPierre Pronchery prev = ossl_list_urxe_prev(e);
182*e7be843bSPierre Pronchery ossl_list_urxe_remove(&demux->urx_free, e);
183*e7be843bSPierre Pronchery
184*e7be843bSPierre Pronchery e2 = OPENSSL_realloc(e, sizeof(QUIC_URXE) + new_alloc_len);
185*e7be843bSPierre Pronchery if (e2 == NULL) {
186*e7be843bSPierre Pronchery /* Failed to resize, abort. */
187*e7be843bSPierre Pronchery if (prev == NULL)
188*e7be843bSPierre Pronchery ossl_list_urxe_insert_head(&demux->urx_free, e);
189*e7be843bSPierre Pronchery else
190*e7be843bSPierre Pronchery ossl_list_urxe_insert_after(&demux->urx_free, prev, e);
191*e7be843bSPierre Pronchery
192*e7be843bSPierre Pronchery return NULL;
193*e7be843bSPierre Pronchery }
194*e7be843bSPierre Pronchery
195*e7be843bSPierre Pronchery if (prev == NULL)
196*e7be843bSPierre Pronchery ossl_list_urxe_insert_head(&demux->urx_free, e2);
197*e7be843bSPierre Pronchery else
198*e7be843bSPierre Pronchery ossl_list_urxe_insert_after(&demux->urx_free, prev, e2);
199*e7be843bSPierre Pronchery
200*e7be843bSPierre Pronchery e2->alloc_len = new_alloc_len;
201*e7be843bSPierre Pronchery return e2;
202*e7be843bSPierre Pronchery }
203*e7be843bSPierre Pronchery
demux_reserve_urxe(QUIC_DEMUX * demux,QUIC_URXE * e,size_t alloc_len)204*e7be843bSPierre Pronchery static QUIC_URXE *demux_reserve_urxe(QUIC_DEMUX *demux, QUIC_URXE *e,
205*e7be843bSPierre Pronchery size_t alloc_len)
206*e7be843bSPierre Pronchery {
207*e7be843bSPierre Pronchery return e->alloc_len < alloc_len ? demux_resize_urxe(demux, e, alloc_len) : e;
208*e7be843bSPierre Pronchery }
209*e7be843bSPierre Pronchery
demux_ensure_free_urxe(QUIC_DEMUX * demux,size_t min_num_free)210*e7be843bSPierre Pronchery static int demux_ensure_free_urxe(QUIC_DEMUX *demux, size_t min_num_free)
211*e7be843bSPierre Pronchery {
212*e7be843bSPierre Pronchery QUIC_URXE *e;
213*e7be843bSPierre Pronchery
214*e7be843bSPierre Pronchery while (ossl_list_urxe_num(&demux->urx_free) < min_num_free) {
215*e7be843bSPierre Pronchery e = demux_alloc_urxe(demux->mtu);
216*e7be843bSPierre Pronchery if (e == NULL)
217*e7be843bSPierre Pronchery return 0;
218*e7be843bSPierre Pronchery
219*e7be843bSPierre Pronchery ossl_list_urxe_insert_tail(&demux->urx_free, e);
220*e7be843bSPierre Pronchery e->demux_state = URXE_DEMUX_STATE_FREE;
221*e7be843bSPierre Pronchery }
222*e7be843bSPierre Pronchery
223*e7be843bSPierre Pronchery return 1;
224*e7be843bSPierre Pronchery }
225*e7be843bSPierre Pronchery
226*e7be843bSPierre Pronchery /*
227*e7be843bSPierre Pronchery * Receive datagrams from network, placing them into URXEs.
228*e7be843bSPierre Pronchery *
229*e7be843bSPierre Pronchery * Returns 1 on success or 0 on failure.
230*e7be843bSPierre Pronchery *
231*e7be843bSPierre Pronchery * Precondition: at least one URXE is free
232*e7be843bSPierre Pronchery * Precondition: there are no pending URXEs
233*e7be843bSPierre Pronchery */
demux_recv(QUIC_DEMUX * demux)234*e7be843bSPierre Pronchery static int demux_recv(QUIC_DEMUX *demux)
235*e7be843bSPierre Pronchery {
236*e7be843bSPierre Pronchery BIO_MSG msg[DEMUX_MAX_MSGS_PER_CALL];
237*e7be843bSPierre Pronchery size_t rd, i;
238*e7be843bSPierre Pronchery QUIC_URXE *urxe = ossl_list_urxe_head(&demux->urx_free), *unext;
239*e7be843bSPierre Pronchery OSSL_TIME now;
240*e7be843bSPierre Pronchery
241*e7be843bSPierre Pronchery /* This should never be called when we have any pending URXE. */
242*e7be843bSPierre Pronchery assert(ossl_list_urxe_head(&demux->urx_pending) == NULL);
243*e7be843bSPierre Pronchery assert(urxe->demux_state == URXE_DEMUX_STATE_FREE);
244*e7be843bSPierre Pronchery
245*e7be843bSPierre Pronchery if (demux->net_bio == NULL)
246*e7be843bSPierre Pronchery /*
247*e7be843bSPierre Pronchery * If no BIO is plugged in, treat this as no datagram being available.
248*e7be843bSPierre Pronchery */
249*e7be843bSPierre Pronchery return QUIC_DEMUX_PUMP_RES_TRANSIENT_FAIL;
250*e7be843bSPierre Pronchery
251*e7be843bSPierre Pronchery /*
252*e7be843bSPierre Pronchery * Opportunistically receive as many messages as possible in a single
253*e7be843bSPierre Pronchery * syscall, determined by how many free URXEs are available.
254*e7be843bSPierre Pronchery */
255*e7be843bSPierre Pronchery for (i = 0; i < (ossl_ssize_t)OSSL_NELEM(msg);
256*e7be843bSPierre Pronchery ++i, urxe = ossl_list_urxe_next(urxe)) {
257*e7be843bSPierre Pronchery if (urxe == NULL) {
258*e7be843bSPierre Pronchery /* We need at least one URXE to receive into. */
259*e7be843bSPierre Pronchery if (!ossl_assert(i > 0))
260*e7be843bSPierre Pronchery return QUIC_DEMUX_PUMP_RES_PERMANENT_FAIL;
261*e7be843bSPierre Pronchery
262*e7be843bSPierre Pronchery break;
263*e7be843bSPierre Pronchery }
264*e7be843bSPierre Pronchery
265*e7be843bSPierre Pronchery /* Ensure the URXE is big enough. */
266*e7be843bSPierre Pronchery urxe = demux_reserve_urxe(demux, urxe, demux->mtu);
267*e7be843bSPierre Pronchery if (urxe == NULL)
268*e7be843bSPierre Pronchery /* Allocation error, fail. */
269*e7be843bSPierre Pronchery return QUIC_DEMUX_PUMP_RES_PERMANENT_FAIL;
270*e7be843bSPierre Pronchery
271*e7be843bSPierre Pronchery /* Ensure we zero any fields added to BIO_MSG at a later date. */
272*e7be843bSPierre Pronchery memset(&msg[i], 0, sizeof(BIO_MSG));
273*e7be843bSPierre Pronchery msg[i].data = ossl_quic_urxe_data(urxe);
274*e7be843bSPierre Pronchery msg[i].data_len = urxe->alloc_len;
275*e7be843bSPierre Pronchery msg[i].peer = &urxe->peer;
276*e7be843bSPierre Pronchery BIO_ADDR_clear(&urxe->peer);
277*e7be843bSPierre Pronchery if (demux->use_local_addr)
278*e7be843bSPierre Pronchery msg[i].local = &urxe->local;
279*e7be843bSPierre Pronchery else
280*e7be843bSPierre Pronchery BIO_ADDR_clear(&urxe->local);
281*e7be843bSPierre Pronchery }
282*e7be843bSPierre Pronchery
283*e7be843bSPierre Pronchery ERR_set_mark();
284*e7be843bSPierre Pronchery if (!BIO_recvmmsg(demux->net_bio, msg, sizeof(BIO_MSG), i, 0, &rd)) {
285*e7be843bSPierre Pronchery if (BIO_err_is_non_fatal(ERR_peek_last_error())) {
286*e7be843bSPierre Pronchery /* Transient error, clear the error and stop. */
287*e7be843bSPierre Pronchery ERR_pop_to_mark();
288*e7be843bSPierre Pronchery return QUIC_DEMUX_PUMP_RES_TRANSIENT_FAIL;
289*e7be843bSPierre Pronchery } else {
290*e7be843bSPierre Pronchery /* Non-transient error, do not clear the error. */
291*e7be843bSPierre Pronchery ERR_clear_last_mark();
292*e7be843bSPierre Pronchery return QUIC_DEMUX_PUMP_RES_PERMANENT_FAIL;
293*e7be843bSPierre Pronchery }
294*e7be843bSPierre Pronchery }
295*e7be843bSPierre Pronchery
296*e7be843bSPierre Pronchery ERR_clear_last_mark();
297*e7be843bSPierre Pronchery now = demux->now != NULL ? demux->now(demux->now_arg) : ossl_time_zero();
298*e7be843bSPierre Pronchery
299*e7be843bSPierre Pronchery urxe = ossl_list_urxe_head(&demux->urx_free);
300*e7be843bSPierre Pronchery for (i = 0; i < rd; ++i, urxe = unext) {
301*e7be843bSPierre Pronchery unext = ossl_list_urxe_next(urxe);
302*e7be843bSPierre Pronchery /* Set URXE with actual length of received datagram. */
303*e7be843bSPierre Pronchery urxe->data_len = msg[i].data_len;
304*e7be843bSPierre Pronchery /* Time we received datagram. */
305*e7be843bSPierre Pronchery urxe->time = now;
306*e7be843bSPierre Pronchery urxe->datagram_id = demux->next_datagram_id++;
307*e7be843bSPierre Pronchery /* Move from free list to pending list. */
308*e7be843bSPierre Pronchery ossl_list_urxe_remove(&demux->urx_free, urxe);
309*e7be843bSPierre Pronchery ossl_list_urxe_insert_tail(&demux->urx_pending, urxe);
310*e7be843bSPierre Pronchery urxe->demux_state = URXE_DEMUX_STATE_PENDING;
311*e7be843bSPierre Pronchery }
312*e7be843bSPierre Pronchery
313*e7be843bSPierre Pronchery return QUIC_DEMUX_PUMP_RES_OK;
314*e7be843bSPierre Pronchery }
315*e7be843bSPierre Pronchery
316*e7be843bSPierre Pronchery /* Extract destination connection ID from the first packet in a datagram. */
demux_identify_conn_id(QUIC_DEMUX * demux,QUIC_URXE * e,QUIC_CONN_ID * dst_conn_id)317*e7be843bSPierre Pronchery static int demux_identify_conn_id(QUIC_DEMUX *demux,
318*e7be843bSPierre Pronchery QUIC_URXE *e,
319*e7be843bSPierre Pronchery QUIC_CONN_ID *dst_conn_id)
320*e7be843bSPierre Pronchery {
321*e7be843bSPierre Pronchery return ossl_quic_wire_get_pkt_hdr_dst_conn_id(ossl_quic_urxe_data(e),
322*e7be843bSPierre Pronchery e->data_len,
323*e7be843bSPierre Pronchery demux->short_conn_id_len,
324*e7be843bSPierre Pronchery dst_conn_id);
325*e7be843bSPierre Pronchery }
326*e7be843bSPierre Pronchery
327*e7be843bSPierre Pronchery /*
328*e7be843bSPierre Pronchery * Process a single pending URXE.
329*e7be843bSPierre Pronchery * Returning 1 on success, 0 on failure.
330*e7be843bSPierre Pronchery */
demux_process_pending_urxe(QUIC_DEMUX * demux,QUIC_URXE * e)331*e7be843bSPierre Pronchery static int demux_process_pending_urxe(QUIC_DEMUX *demux, QUIC_URXE *e)
332*e7be843bSPierre Pronchery {
333*e7be843bSPierre Pronchery QUIC_CONN_ID dst_conn_id;
334*e7be843bSPierre Pronchery int dst_conn_id_ok = 0;
335*e7be843bSPierre Pronchery
336*e7be843bSPierre Pronchery /* The next URXE we process should be at the head of the pending list. */
337*e7be843bSPierre Pronchery if (!ossl_assert(e == ossl_list_urxe_head(&demux->urx_pending)))
338*e7be843bSPierre Pronchery return 0;
339*e7be843bSPierre Pronchery
340*e7be843bSPierre Pronchery assert(e->demux_state == URXE_DEMUX_STATE_PENDING);
341*e7be843bSPierre Pronchery
342*e7be843bSPierre Pronchery /* Determine the DCID of the first packet in the datagram. */
343*e7be843bSPierre Pronchery dst_conn_id_ok = demux_identify_conn_id(demux, e, &dst_conn_id);
344*e7be843bSPierre Pronchery
345*e7be843bSPierre Pronchery ossl_list_urxe_remove(&demux->urx_pending, e);
346*e7be843bSPierre Pronchery if (demux->default_cb != NULL) {
347*e7be843bSPierre Pronchery /*
348*e7be843bSPierre Pronchery * Pass to default handler for routing. The URXE now belongs to the
349*e7be843bSPierre Pronchery * callback.
350*e7be843bSPierre Pronchery */
351*e7be843bSPierre Pronchery e->demux_state = URXE_DEMUX_STATE_ISSUED;
352*e7be843bSPierre Pronchery demux->default_cb(e, demux->default_cb_arg,
353*e7be843bSPierre Pronchery dst_conn_id_ok ? &dst_conn_id : NULL);
354*e7be843bSPierre Pronchery } else {
355*e7be843bSPierre Pronchery /* Discard. */
356*e7be843bSPierre Pronchery ossl_list_urxe_insert_tail(&demux->urx_free, e);
357*e7be843bSPierre Pronchery e->demux_state = URXE_DEMUX_STATE_FREE;
358*e7be843bSPierre Pronchery }
359*e7be843bSPierre Pronchery
360*e7be843bSPierre Pronchery return 1; /* keep processing pending URXEs */
361*e7be843bSPierre Pronchery }
362*e7be843bSPierre Pronchery
363*e7be843bSPierre Pronchery /* Process pending URXEs to generate callbacks. */
demux_process_pending_urxl(QUIC_DEMUX * demux)364*e7be843bSPierre Pronchery static int demux_process_pending_urxl(QUIC_DEMUX *demux)
365*e7be843bSPierre Pronchery {
366*e7be843bSPierre Pronchery QUIC_URXE *e;
367*e7be843bSPierre Pronchery int ret;
368*e7be843bSPierre Pronchery
369*e7be843bSPierre Pronchery while ((e = ossl_list_urxe_head(&demux->urx_pending)) != NULL)
370*e7be843bSPierre Pronchery if ((ret = demux_process_pending_urxe(demux, e)) <= 0)
371*e7be843bSPierre Pronchery return ret;
372*e7be843bSPierre Pronchery
373*e7be843bSPierre Pronchery return 1;
374*e7be843bSPierre Pronchery }
375*e7be843bSPierre Pronchery
376*e7be843bSPierre Pronchery /*
377*e7be843bSPierre Pronchery * Drain the pending URXE list, processing any pending URXEs by making their
378*e7be843bSPierre Pronchery * callbacks. If no URXEs are pending, a network read is attempted first.
379*e7be843bSPierre Pronchery */
ossl_quic_demux_pump(QUIC_DEMUX * demux)380*e7be843bSPierre Pronchery int ossl_quic_demux_pump(QUIC_DEMUX *demux)
381*e7be843bSPierre Pronchery {
382*e7be843bSPierre Pronchery int ret;
383*e7be843bSPierre Pronchery
384*e7be843bSPierre Pronchery if (ossl_list_urxe_head(&demux->urx_pending) == NULL) {
385*e7be843bSPierre Pronchery ret = demux_ensure_free_urxe(demux, DEMUX_MAX_MSGS_PER_CALL);
386*e7be843bSPierre Pronchery if (ret != 1)
387*e7be843bSPierre Pronchery return QUIC_DEMUX_PUMP_RES_PERMANENT_FAIL;
388*e7be843bSPierre Pronchery
389*e7be843bSPierre Pronchery ret = demux_recv(demux);
390*e7be843bSPierre Pronchery if (ret != QUIC_DEMUX_PUMP_RES_OK)
391*e7be843bSPierre Pronchery return ret;
392*e7be843bSPierre Pronchery
393*e7be843bSPierre Pronchery /*
394*e7be843bSPierre Pronchery * If demux_recv returned successfully, we should always have something.
395*e7be843bSPierre Pronchery */
396*e7be843bSPierre Pronchery assert(ossl_list_urxe_head(&demux->urx_pending) != NULL);
397*e7be843bSPierre Pronchery }
398*e7be843bSPierre Pronchery
399*e7be843bSPierre Pronchery if ((ret = demux_process_pending_urxl(demux)) <= 0)
400*e7be843bSPierre Pronchery return QUIC_DEMUX_PUMP_RES_PERMANENT_FAIL;
401*e7be843bSPierre Pronchery
402*e7be843bSPierre Pronchery return QUIC_DEMUX_PUMP_RES_OK;
403*e7be843bSPierre Pronchery }
404*e7be843bSPierre Pronchery
405*e7be843bSPierre Pronchery /* Artificially inject a packet into the demuxer for testing purposes. */
ossl_quic_demux_inject(QUIC_DEMUX * demux,const unsigned char * buf,size_t buf_len,const BIO_ADDR * peer,const BIO_ADDR * local)406*e7be843bSPierre Pronchery int ossl_quic_demux_inject(QUIC_DEMUX *demux,
407*e7be843bSPierre Pronchery const unsigned char *buf,
408*e7be843bSPierre Pronchery size_t buf_len,
409*e7be843bSPierre Pronchery const BIO_ADDR *peer,
410*e7be843bSPierre Pronchery const BIO_ADDR *local)
411*e7be843bSPierre Pronchery {
412*e7be843bSPierre Pronchery int ret;
413*e7be843bSPierre Pronchery QUIC_URXE *urxe;
414*e7be843bSPierre Pronchery
415*e7be843bSPierre Pronchery ret = demux_ensure_free_urxe(demux, 1);
416*e7be843bSPierre Pronchery if (ret != 1)
417*e7be843bSPierre Pronchery return 0;
418*e7be843bSPierre Pronchery
419*e7be843bSPierre Pronchery urxe = ossl_list_urxe_head(&demux->urx_free);
420*e7be843bSPierre Pronchery
421*e7be843bSPierre Pronchery assert(urxe->demux_state == URXE_DEMUX_STATE_FREE);
422*e7be843bSPierre Pronchery
423*e7be843bSPierre Pronchery urxe = demux_reserve_urxe(demux, urxe, buf_len);
424*e7be843bSPierre Pronchery if (urxe == NULL)
425*e7be843bSPierre Pronchery return 0;
426*e7be843bSPierre Pronchery
427*e7be843bSPierre Pronchery memcpy(ossl_quic_urxe_data(urxe), buf, buf_len);
428*e7be843bSPierre Pronchery urxe->data_len = buf_len;
429*e7be843bSPierre Pronchery
430*e7be843bSPierre Pronchery if (peer != NULL)
431*e7be843bSPierre Pronchery urxe->peer = *peer;
432*e7be843bSPierre Pronchery else
433*e7be843bSPierre Pronchery BIO_ADDR_clear(&urxe->peer);
434*e7be843bSPierre Pronchery
435*e7be843bSPierre Pronchery if (local != NULL)
436*e7be843bSPierre Pronchery urxe->local = *local;
437*e7be843bSPierre Pronchery else
438*e7be843bSPierre Pronchery BIO_ADDR_clear(&urxe->local);
439*e7be843bSPierre Pronchery
440*e7be843bSPierre Pronchery urxe->time
441*e7be843bSPierre Pronchery = demux->now != NULL ? demux->now(demux->now_arg) : ossl_time_zero();
442*e7be843bSPierre Pronchery
443*e7be843bSPierre Pronchery /* Move from free list to pending list. */
444*e7be843bSPierre Pronchery ossl_list_urxe_remove(&demux->urx_free, urxe);
445*e7be843bSPierre Pronchery urxe->datagram_id = demux->next_datagram_id++;
446*e7be843bSPierre Pronchery ossl_list_urxe_insert_tail(&demux->urx_pending, urxe);
447*e7be843bSPierre Pronchery urxe->demux_state = URXE_DEMUX_STATE_PENDING;
448*e7be843bSPierre Pronchery
449*e7be843bSPierre Pronchery return demux_process_pending_urxl(demux) > 0;
450*e7be843bSPierre Pronchery }
451*e7be843bSPierre Pronchery
452*e7be843bSPierre Pronchery /* Called by our user to return a URXE to the free list. */
ossl_quic_demux_release_urxe(QUIC_DEMUX * demux,QUIC_URXE * e)453*e7be843bSPierre Pronchery void ossl_quic_demux_release_urxe(QUIC_DEMUX *demux,
454*e7be843bSPierre Pronchery QUIC_URXE *e)
455*e7be843bSPierre Pronchery {
456*e7be843bSPierre Pronchery assert(ossl_list_urxe_prev(e) == NULL && ossl_list_urxe_next(e) == NULL);
457*e7be843bSPierre Pronchery assert(e->demux_state == URXE_DEMUX_STATE_ISSUED);
458*e7be843bSPierre Pronchery ossl_list_urxe_insert_tail(&demux->urx_free, e);
459*e7be843bSPierre Pronchery e->demux_state = URXE_DEMUX_STATE_FREE;
460*e7be843bSPierre Pronchery }
461*e7be843bSPierre Pronchery
ossl_quic_demux_reinject_urxe(QUIC_DEMUX * demux,QUIC_URXE * e)462*e7be843bSPierre Pronchery void ossl_quic_demux_reinject_urxe(QUIC_DEMUX *demux,
463*e7be843bSPierre Pronchery QUIC_URXE *e)
464*e7be843bSPierre Pronchery {
465*e7be843bSPierre Pronchery assert(ossl_list_urxe_prev(e) == NULL && ossl_list_urxe_next(e) == NULL);
466*e7be843bSPierre Pronchery assert(e->demux_state == URXE_DEMUX_STATE_ISSUED);
467*e7be843bSPierre Pronchery ossl_list_urxe_insert_head(&demux->urx_pending, e);
468*e7be843bSPierre Pronchery e->demux_state = URXE_DEMUX_STATE_PENDING;
469*e7be843bSPierre Pronchery }
470*e7be843bSPierre Pronchery
ossl_quic_demux_has_pending(const QUIC_DEMUX * demux)471*e7be843bSPierre Pronchery int ossl_quic_demux_has_pending(const QUIC_DEMUX *demux)
472*e7be843bSPierre Pronchery {
473*e7be843bSPierre Pronchery return ossl_list_urxe_head(&demux->urx_pending) != NULL;
474*e7be843bSPierre Pronchery }
475