blob: 85eca90d7df0a4ff4e42594b91f390af5e2cf26a [file] [log] [blame]
Frédéric Lécailleccac11f2021-03-03 16:09:02 +01001/*
2 * HTTP/3 protocol processing
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation, version 2.1
7 * exclusively.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19#include <haproxy/buf.h>
Amaury Denoyelle99043552021-08-24 15:36:02 +020020#include <haproxy/connection.h>
Frédéric Lécailleccac11f2021-03-03 16:09:02 +010021#include <haproxy/dynbuf.h>
22#include <haproxy/h3.h>
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +020023#include <haproxy/http.h>
24#include <haproxy/htx.h>
Frédéric Lécailleccac11f2021-03-03 16:09:02 +010025#include <haproxy/istbuf.h>
Amaury Denoyelledeed7772021-12-03 11:36:46 +010026#include <haproxy/mux_quic-t.h>
Frédéric Lécailleccac11f2021-03-03 16:09:02 +010027#include <haproxy/pool.h>
28#include <haproxy/qpack-dec.h>
Amaury Denoyelle15b09612021-08-24 16:20:27 +020029#include <haproxy/qpack-enc.h>
30#include <haproxy/quic_enc.h>
Amaury Denoyelle99043552021-08-24 15:36:02 +020031#include <haproxy/stream.h>
Frédéric Lécailleccac11f2021-03-03 16:09:02 +010032#include <haproxy/tools.h>
33#include <haproxy/xprt_quic.h>
34
35#define DEBUG_H3
36
37#if defined(DEBUG_H3)
38#define h3_debug_printf fprintf
39#define h3_debug_hexdump debug_hexdump
40#else
41#define h3_debug_printf(...) do { } while (0)
42#define h3_debug_hexdump(...) do { } while (0)
43#endif
44
45#define H3_CF_SETTINGS_SENT 0x00000001
46
47/* Default settings */
Amaury Denoyelle33949392021-08-24 15:16:58 +020048static uint64_t h3_settings_qpack_max_table_capacity = 0;
49static uint64_t h3_settings_qpack_blocked_streams = 4096;
50static uint64_t h3_settings_max_field_section_size = QUIC_VARINT_8_BYTE_MAX; /* Unlimited */
Frédéric Lécailleccac11f2021-03-03 16:09:02 +010051
52struct h3 {
53 struct qcc *qcc;
54 enum h3_err err;
55 uint32_t flags;
56 /* Locally initiated uni-streams */
57 struct h3_uqs lqpack_enc;
58 struct h3_uqs lqpack_dec;
59 struct h3_uqs lctrl;
60 /* Remotely initiated uni-streams */
61 struct h3_uqs rqpack_enc;
62 struct h3_uqs rqpack_dec;
63 struct h3_uqs rctrl;
64 /* Settings */
65 uint64_t qpack_max_table_capacity;
66 uint64_t qpack_blocked_streams;
67 uint64_t max_field_section_size;
68 struct buffer_wait buf_wait; /* wait list for buffer allocations */
69};
70
71DECLARE_STATIC_POOL(pool_head_h3, "h3", sizeof(struct h3));
72
73/* Simple function to duplicate a buffer */
74static inline struct buffer h3_b_dup(struct buffer *b)
75{
76 return b_make(b->area, b->size, b->head, b->data);
77}
78
Frédéric Lécailleccac11f2021-03-03 16:09:02 +010079/* Decode a h3 frame header made of two QUIC varints from <b> buffer.
80 * Returns the number of bytes consumed if there was enough data in <b>, 0 if not.
81 * Note that this function update <b> buffer to reflect the number of bytes consumed
82 * to decode the h3 frame header.
83 */
84static inline size_t h3_decode_frm_header(uint64_t *ftype, uint64_t *flen,
85 struct buffer *b)
86{
87 size_t hlen;
88
89 hlen = 0;
90 if (!b_quic_dec_int(ftype, b, &hlen) || !b_quic_dec_int(flen, b, &hlen))
91 return 0;
92
93 return hlen;
94}
95
Amaury Denoyelleb9ce14e2021-11-08 09:13:42 +010096/* Decode <qcs> remotely initiated bidi-stream.
97 * Returns <0 on error else 0.
98 */
Amaury Denoyelledb443382021-11-30 11:23:29 +010099static int h3_decode_qcs(struct qcs *qcs, int fin, void *ctx)
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100100{
101 struct buffer *rxbuf = &qcs->rx.buf;
102 struct h3 *h3 = ctx;
Amaury Denoyelle7059ebc2021-12-08 15:51:04 +0100103 struct htx *htx = NULL;
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200104 struct htx_sl *sl;
Amaury Denoyelle99043552021-08-24 15:36:02 +0200105 struct conn_stream *cs;
Amaury Denoyellefd7cdc32021-08-24 15:13:20 +0200106 struct http_hdr list[global.tune.max_http_hdr];
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200107 unsigned int flags = HTX_SL_F_NONE;
Amaury Denoyellefd7cdc32021-08-24 15:13:20 +0200108 int hdr_idx;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100109
110 h3_debug_printf(stderr, "%s: STREAM ID: %llu\n", __func__, qcs->by_id.key);
111 if (!b_data(rxbuf))
112 return 0;
113
114 while (b_data(rxbuf)) {
115 size_t hlen;
116 uint64_t ftype, flen;
117 struct buffer b;
118
119 /* Work on a copy of <rxbuf> */
120 b = h3_b_dup(rxbuf);
121 hlen = h3_decode_frm_header(&ftype, &flen, &b);
122 if (!hlen)
123 break;
124
125 h3_debug_printf(stderr, "%s: ftype: %llu, flen: %llu\n", __func__,
126 (unsigned long long)ftype, (unsigned long long)flen);
127 if (flen > b_data(&b))
128 break;
129
130 b_del(rxbuf, hlen);
131 switch (ftype) {
132 case H3_FT_DATA:
133 break;
134 case H3_FT_HEADERS:
135 {
136 const unsigned char *buf = (const unsigned char *)b_head(rxbuf);
Amaury Denoyelle3cae4042021-11-08 08:57:18 +0100137 struct buffer htx_buf = BUF_NULL;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100138 struct buffer *tmp = get_trash_chunk();
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200139 struct ist meth = IST_NULL, path = IST_NULL;
Amaury Denoyelle3cae4042021-11-08 08:57:18 +0100140 //struct ist scheme = IST_NULL, authority = IST_NULL;
141 struct ist authority = IST_NULL;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100142
Amaury Denoyelle4ac6d372022-02-14 14:38:55 +0100143 if (qpack_decode_fs(buf, flen, tmp, list) < 0) {
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100144 h3->err = QPACK_DECOMPRESSION_FAILED;
145 return -1;
146 }
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200147
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200148 b_alloc(&htx_buf);
149 htx = htx_from_buf(&htx_buf);
150
151 /* first treat pseudo-header to build the start line */
152 hdr_idx = 0;
153 while (1) {
154 if (isteq(list[hdr_idx].n, ist("")))
155 break;
156
157 if (istmatch(list[hdr_idx].n, ist(":"))) {
158 /* pseudo-header */
159 if (isteq(list[hdr_idx].n, ist(":method")))
160 meth = list[hdr_idx].v;
161 else if (isteq(list[hdr_idx].n, ist(":path")))
162 path = list[hdr_idx].v;
Amaury Denoyelle3cae4042021-11-08 08:57:18 +0100163 //else if (isteq(list[hdr_idx].n, ist(":scheme")))
164 // scheme = list[hdr_idx].v;
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200165 else if (isteq(list[hdr_idx].n, ist(":authority")))
166 authority = list[hdr_idx].v;
167 }
168
169 ++hdr_idx;
170 }
171
172 flags |= HTX_SL_F_VER_11;
173
174 sl = htx_add_stline(htx, HTX_BLK_REQ_SL, flags, meth, path, ist("HTTP/3.0"));
Amaury Denoyelleb9ce14e2021-11-08 09:13:42 +0100175 if (!sl)
176 goto fail;
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200177 sl->flags |= HTX_SL_F_BODYLESS;
178 sl->info.req.meth = find_http_meth(meth.ptr, meth.len);
179 BUG_ON(sl->info.req.meth == HTTP_METH_OTHER);
180
181 if (isttest(authority))
182 htx_add_header(htx, ist("host"), authority);
183
184 /* now treat standard headers */
185 hdr_idx = 0;
186 while (1) {
187 if (isteq(list[hdr_idx].n, ist("")))
188 break;
189
190 if (!istmatch(list[hdr_idx].n, ist(":")))
191 htx_add_header(htx, list[hdr_idx].n, list[hdr_idx].v);
192
193 ++hdr_idx;
194 }
195
196 htx_add_endof(htx, HTX_BLK_EOH);
197 htx_to_buf(htx, &htx_buf);
Amaury Denoyelle99043552021-08-24 15:36:02 +0200198
199 cs = cs_new(qcs->qcc->conn, qcs->qcc->conn->target);
200 cs->ctx = qcs;
201 stream_create_from_cs(cs, &htx_buf);
202
Ilya Shipitsinbd6b4be2021-10-15 16:18:21 +0500203 /* buffer is transferred to conn_stream and set to NULL
Amaury Denoyelle99043552021-08-24 15:36:02 +0200204 * except on stream creation error.
205 */
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200206 b_free(&htx_buf);
207
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100208 break;
209 }
210 case H3_FT_PUSH_PROMISE:
211 /* Not supported */
212 break;
213 default:
Amaury Denoyelled1acaf92021-11-15 15:52:55 +0100214 /* draft-ietf-quic-http34 9. Extensions to HTTP/3
215 * unknown frame types MUST be ignored
216 */
217 h3_debug_printf(stderr, "ignore unknown frame type 0x%lx\n", ftype);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100218 }
219 b_del(rxbuf, flen);
220 }
221
Amaury Denoyelle7059ebc2021-12-08 15:51:04 +0100222 if (htx) {
223 if (fin && !b_data(rxbuf))
224 htx->flags |= HTX_FL_EOM;
225 }
Amaury Denoyelledb443382021-11-30 11:23:29 +0100226
Amaury Denoyelleb9ce14e2021-11-08 09:13:42 +0100227 return 0;
228
229 fail:
230 return -1;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100231}
232
233/* Parse a SETTINGS frame which must not be truncated with <flen> as length from
234 * <rxbuf> buffer. This function does not update this buffer.
235 * Returns 0 if something wrong happened, 1 if not.
236 */
237static int h3_parse_settings_frm(struct h3 *h3, const struct buffer *rxbuf, size_t flen)
238{
239 uint64_t id, value;
240 const unsigned char *buf, *end;
241
242 buf = (const unsigned char *)b_head(rxbuf);
243 end = buf + flen;
244
245 while (buf <= end) {
246 if (!quic_dec_int(&id, &buf, end) || !quic_dec_int(&value, &buf, end))
247 return 0;
248
249 h3_debug_printf(stderr, "%s id: %llu value: %llu\n",
250 __func__, (unsigned long long)id, (unsigned long long)value);
251 switch (id) {
252 case H3_SETTINGS_QPACK_MAX_TABLE_CAPACITY:
253 h3->qpack_max_table_capacity = value;
254 break;
255 case H3_SETTINGS_MAX_FIELD_SECTION_SIZE:
256 h3->max_field_section_size = value;
257 break;
258 case H3_SETTINGS_QPACK_BLOCKED_STREAMS:
259 h3->qpack_blocked_streams = value;
260 break;
261 case H3_SETTINGS_RESERVED_2 ... H3_SETTINGS_RESERVED_5:
262 h3->err = H3_SETTINGS_ERROR;
263 return 0;
264 default:
265 /* MUST be ignored */
266 break;
267 }
268 }
269
270 return 1;
271}
272
273/* Decode <qcs> remotely initiated uni-stream. We stop parsing a frame as soon as
274 * there is not enough received data.
275 * Returns 0 if something wrong happened, 1 if not.
276 */
277static int h3_control_recv(struct h3_uqs *h3_uqs, void *ctx)
278{
279 struct buffer *rxbuf = &h3_uqs->qcs->rx.buf;
280 struct h3 *h3 = ctx;
281
282 h3_debug_printf(stderr, "%s STREAM ID: %llu\n", __func__, h3_uqs->qcs->by_id.key);
283 if (!b_data(rxbuf))
284 return 1;
285
286 while (b_data(rxbuf)) {
287 size_t hlen;
288 uint64_t ftype, flen;
289 struct buffer b;
290
291 /* Work on a copy of <rxbuf> */
292 b = h3_b_dup(rxbuf);
293 hlen = h3_decode_frm_header(&ftype, &flen, &b);
294 if (!hlen)
295 break;
296
297 h3_debug_printf(stderr, "%s: ftype: %llu, flen: %llu\n", __func__,
298 (unsigned long long)ftype, (unsigned long long)flen);
299 if (flen > b_data(&b))
300 break;
301
302 b_del(rxbuf, hlen);
303 /* From here, a frame must not be truncated */
304 switch (ftype) {
305 case H3_FT_CANCEL_PUSH:
Amaury Denoyellee1f3ff02021-12-06 14:26:52 +0100306 /* XXX TODO XXX */
307 ABORT_NOW();
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100308 break;
309 case H3_FT_SETTINGS:
310 if (!h3_parse_settings_frm(h3, rxbuf, flen))
311 return 0;
312 break;
313 case H3_FT_GOAWAY:
Amaury Denoyellee1f3ff02021-12-06 14:26:52 +0100314 /* XXX TODO XXX */
315 ABORT_NOW();
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100316 break;
317 case H3_FT_MAX_PUSH_ID:
Amaury Denoyellee1f3ff02021-12-06 14:26:52 +0100318 /* XXX TODO XXX */
319 ABORT_NOW();
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100320 break;
321 default:
322 /* Error */
323 h3->err = H3_FRAME_UNEXPECTED;
324 return 0;
325 }
326 b_del(rxbuf, flen);
327 }
328
Amaury Denoyellea3f222d2021-12-06 11:24:00 +0100329 /* Handle the case where remaining data are present in the buffer. This
330 * can happen if there is an incomplete frame. In this case, subscribe
331 * on the lower layer to restart receive operation.
Amaury Denoyelledeed7772021-12-03 11:36:46 +0100332 */
Amaury Denoyellea3f222d2021-12-06 11:24:00 +0100333 if (b_data(rxbuf))
334 qcs_subscribe(h3_uqs->qcs, SUB_RETRY_RECV, &h3_uqs->wait_event);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100335
336 return 1;
337}
338
Amaury Denoyellea5871362021-10-07 16:26:12 +0200339/* Returns buffer for data sending.
340 * May be NULL if the allocation failed.
341 */
342static struct buffer *mux_get_buf(struct qcs *qcs)
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100343{
Amaury Denoyellea5871362021-10-07 16:26:12 +0200344 if (!b_size(&qcs->tx.buf))
345 b_alloc(&qcs->tx.buf);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100346
Amaury Denoyellea5871362021-10-07 16:26:12 +0200347 return &qcs->tx.buf;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100348}
349
350/* Function used to emit stream data from <h3_uqs> control uni-stream */
351static int h3_control_send(struct h3_uqs *h3_uqs, void *ctx)
352{
353 int ret;
354 struct h3 *h3 = ctx;
355 unsigned char data[(2 + 3) * 2 * QUIC_VARINT_MAX_SIZE]; /* enough for 3 settings */
Amaury Denoyellea5871362021-10-07 16:26:12 +0200356 struct buffer pos, *res;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100357
358 ret = 0;
Amaury Denoyellea5871362021-10-07 16:26:12 +0200359 pos = b_make((char *)data, sizeof(data), 0, 0);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100360 if (!(h3->flags & H3_CF_SETTINGS_SENT)) {
361 struct qcs *qcs = h3_uqs->qcs;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100362 size_t frm_len;
363
364 frm_len = quic_int_getsize(H3_SETTINGS_QPACK_MAX_TABLE_CAPACITY) +
365 quic_int_getsize(h3_settings_qpack_max_table_capacity) +
366 quic_int_getsize(H3_SETTINGS_QPACK_BLOCKED_STREAMS) +
367 quic_int_getsize(h3_settings_qpack_blocked_streams);
368 if (h3_settings_max_field_section_size) {
369 frm_len += quic_int_getsize(H3_SETTINGS_MAX_FIELD_SECTION_SIZE) +
370 quic_int_getsize(h3_settings_max_field_section_size);
371 }
372
Amaury Denoyellea5871362021-10-07 16:26:12 +0200373 b_quic_enc_int(&pos, H3_UNI_STRM_TP_CONTROL_STREAM);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100374 /* Build a SETTINGS frame */
Amaury Denoyellea5871362021-10-07 16:26:12 +0200375 b_quic_enc_int(&pos, H3_FT_SETTINGS);
376 b_quic_enc_int(&pos, frm_len);
377 b_quic_enc_int(&pos, H3_SETTINGS_QPACK_MAX_TABLE_CAPACITY);
378 b_quic_enc_int(&pos, h3_settings_qpack_max_table_capacity);
379 b_quic_enc_int(&pos, H3_SETTINGS_QPACK_BLOCKED_STREAMS);
380 b_quic_enc_int(&pos, h3_settings_qpack_blocked_streams);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100381 if (h3_settings_max_field_section_size) {
Amaury Denoyellea5871362021-10-07 16:26:12 +0200382 b_quic_enc_int(&pos, H3_SETTINGS_MAX_FIELD_SECTION_SIZE);
383 b_quic_enc_int(&pos, h3_settings_max_field_section_size);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100384 }
Amaury Denoyellea5871362021-10-07 16:26:12 +0200385
386 res = mux_get_buf(qcs);
387 if (b_room(res) < b_data(&pos)) {
388 // TODO the mux should be put in blocked state, with
389 // the stream in state waiting for settings to be sent
390 ABORT_NOW();
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100391 }
392
Amaury Denoyellea5871362021-10-07 16:26:12 +0200393 ret = b_force_xfer(res, &pos, b_data(&pos));
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100394 if (ret > 0) {
395 h3->flags |= H3_CF_SETTINGS_SENT;
Amaury Denoyellea5871362021-10-07 16:26:12 +0200396 if (!(qcs->qcc->wait_event.events & SUB_RETRY_SEND))
397 tasklet_wakeup(qcs->qcc->wait_event.tasklet);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100398 }
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100399 }
400
401 return ret;
402}
403
Amaury Denoyelle15b09612021-08-24 16:20:27 +0200404static int h3_resp_headers_send(struct qcs *qcs, struct htx *htx)
405{
406 struct buffer outbuf;
407 struct buffer headers_buf = BUF_NULL;
408 struct buffer *res;
409 struct http_hdr list[global.tune.max_http_hdr];
410 struct htx_sl *sl;
411 struct htx_blk *blk;
412 enum htx_blk_type type;
413 int frame_length_size; /* size in bytes of frame length varint field */
414 int ret = 0;
415 int hdr;
416 int status = 0;
417
418 sl = NULL;
419 hdr = 0;
420 for (blk = htx_get_head_blk(htx); blk; blk = htx_get_next_blk(htx, blk)) {
421 type = htx_get_blk_type(blk);
422
423 if (type == HTX_BLK_UNUSED)
424 continue;
425
426 if (type == HTX_BLK_EOH)
427 break;
428
429 if (type == HTX_BLK_RES_SL) {
430 /* start-line -> HEADERS h3 frame */
431 BUG_ON(sl);
432 sl = htx_get_blk_ptr(htx, blk);
433 /* TODO should be on h3 layer */
434 status = sl->info.res.status;
435 }
436 else if (type == HTX_BLK_HDR) {
437 list[hdr].n = htx_get_blk_name(htx, blk);
438 list[hdr].v = htx_get_blk_value(htx, blk);
439 hdr++;
440 }
441 else {
442 ABORT_NOW();
443 goto err;
444 }
445 }
446
447 BUG_ON(!sl);
448
449 list[hdr].n = ist("");
450
Amaury Denoyelled3d97c62021-10-05 11:45:58 +0200451 res = mux_get_buf(qcs);
Amaury Denoyelle15b09612021-08-24 16:20:27 +0200452
453 /* At least 5 bytes to store frame type + length as a varint max size */
454 if (b_room(res) < 5)
455 ABORT_NOW();
456
457 b_reset(&outbuf);
458 outbuf = b_make(b_tail(res), b_contig_space(res), 0, 0);
459 /* Start the headers after frame type + length */
460 headers_buf = b_make(b_head(res) + 5, b_size(res) - 5, 0, 0);
461
462 if (qpack_encode_field_section_line(&headers_buf))
463 ABORT_NOW();
464 if (qpack_encode_int_status(&headers_buf, status))
465 ABORT_NOW();
466
467 for (hdr = 0; hdr < sizeof(list) / sizeof(list[0]); ++hdr) {
468 if (isteq(list[hdr].n, ist("")))
469 break;
470
Amaury Denoyelleffafb3d2022-02-15 16:10:42 +0100471 /* draft-ietf-quic-http34 4.1. HTTP Message Exchanges
472 * Transfer codings (see Section 6.1 of [HTTP11]) are not
473 * defined for HTTP/3; the Transfer-Encoding header field MUST
474 * NOT be used.
475 */
476 if (isteq(list[hdr].n, ist("transfer-encoding")))
477 continue;
478
Amaury Denoyelle15b09612021-08-24 16:20:27 +0200479 if (qpack_encode_header(&headers_buf, list[hdr].n, list[hdr].v))
480 ABORT_NOW();
481 }
482
483 /* Now that all headers are encoded, we are certain that res buffer is
484 * big enough
485 */
486 frame_length_size = quic_int_getsize(b_data(&headers_buf));
487 res->head += 4 - frame_length_size;
488 b_putchr(res, 0x01); /* h3 HEADERS frame type */
489 if (!b_quic_enc_int(res, b_data(&headers_buf)))
490 ABORT_NOW();
491 b_add(res, b_data(&headers_buf));
492
493 ret = 0;
494 blk = htx_get_head_blk(htx);
495 while (blk) {
496 type = htx_get_blk_type(blk);
497 ret += htx_get_blksz(blk);
498 blk = htx_remove_blk(htx, blk);
499 if (type == HTX_BLK_EOH)
500 break;
501 }
502
503 return ret;
504
505 err:
506 return 0;
507}
508
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200509/* Returns the total of bytes sent. */
510static int h3_resp_data_send(struct qcs *qcs, struct buffer *buf, size_t count)
511{
512 struct buffer outbuf;
513 struct buffer *res;
514 size_t total = 0;
515 struct htx *htx;
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200516 int bsize, fsize, hsize;
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200517 struct htx_blk *blk;
518 enum htx_blk_type type;
519
520 htx = htx_from_buf(buf);
521
522 new_frame:
523 if (!count || htx_is_empty(htx))
524 goto end;
525
526 blk = htx_get_head_blk(htx);
527 type = htx_get_blk_type(blk);
528 fsize = bsize = htx_get_blksz(blk);
529
530 if (type != HTX_BLK_DATA)
531 goto end;
532
Amaury Denoyelled3d97c62021-10-05 11:45:58 +0200533 res = mux_get_buf(qcs);
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200534
535 if (fsize > count)
536 fsize = count;
537
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200538 /* h3 DATA headers : 1-byte frame type + varint frame length */
539 hsize = 1 + QUIC_VARINT_MAX_SIZE;
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200540
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200541 while (1) {
542 b_reset(&outbuf);
543 outbuf = b_make(b_tail(res), b_contig_space(res), 0, 0);
544 if (b_size(&outbuf) > hsize || !b_space_wraps(res))
545 break;
546 b_slow_realign(res, trash.area, b_data(res));
547 }
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200548
Amaury Denoyelle84ea8dc2021-12-03 14:40:01 +0100549 /* Not enough room for headers and at least one data byte, block the
550 * stream. It is expected that the conn-stream layer will subscribe on
551 * SEND.
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200552 */
Amaury Denoyelle84ea8dc2021-12-03 14:40:01 +0100553 if (b_size(&outbuf) <= hsize) {
554 qcs->flags |= QC_SF_BLK_MROOM;
555 goto end;
556 }
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200557
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200558 if (b_size(&outbuf) < hsize + fsize)
559 fsize = b_size(&outbuf) - hsize;
560 BUG_ON(fsize <= 0);
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200561
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200562 b_putchr(&outbuf, 0x00); /* h3 frame type = DATA */
563 b_quic_enc_int(&outbuf, fsize); /* h3 frame length */
564
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200565 b_putblk(&outbuf, htx_get_blk_ptr(htx, blk), fsize);
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200566 total += fsize;
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200567 count -= fsize;
568
569 if (fsize == bsize)
570 htx_remove_blk(htx, blk);
571 else
572 htx_cut_data_blk(htx, blk, fsize);
573
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200574 /* commit the buffer */
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200575 b_add(res, b_data(&outbuf));
576 goto new_frame;
577
578 end:
579 return total;
580}
581
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200582size_t h3_snd_buf(struct conn_stream *cs, struct buffer *buf, size_t count, int flags)
583{
584 size_t total = 0;
585 struct qcs *qcs = cs->ctx;
586 struct htx *htx;
587 enum htx_blk_type btype;
588 struct htx_blk *blk;
589 uint32_t bsize;
590 int32_t idx;
591 int ret;
592
Amaury Denoyelledeed7772021-12-03 11:36:46 +0100593 fprintf(stderr, "%s\n", __func__);
594
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200595 htx = htx_from_buf(buf);
596
Amaury Denoyelle84ea8dc2021-12-03 14:40:01 +0100597 while (count && !htx_is_empty(htx) && !(qcs->flags & QC_SF_BLK_MROOM)) {
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200598 idx = htx_get_head(htx);
599 blk = htx_get_blk(htx, idx);
600 btype = htx_get_blk_type(blk);
601 bsize = htx_get_blksz(blk);
602
603 /* Not implemented : QUIC on backend side */
604 BUG_ON(btype == HTX_BLK_REQ_SL);
605
606 switch (btype) {
607 case HTX_BLK_RES_SL:
Amaury Denoyelle15b09612021-08-24 16:20:27 +0200608 /* start-line -> HEADERS h3 frame */
609 ret = h3_resp_headers_send(qcs, htx);
610 if (ret > 0) {
611 total += ret;
612 count -= ret;
613 if (ret < bsize)
614 goto out;
615 }
616 break;
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200617
618 case HTX_BLK_DATA:
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200619 ret = h3_resp_data_send(qcs, buf, count);
620 if (ret > 0) {
621 htx = htx_from_buf(buf);
622 total += ret;
623 count -= ret;
624 if (ret < bsize)
625 goto out;
626 }
627 break;
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200628
629 case HTX_BLK_TLR:
630 case HTX_BLK_EOT:
631 /* TODO trailers */
632
633 default:
634 htx_remove_blk(htx, blk);
635 total += bsize;
636 count -= bsize;
637 break;
638 }
639 }
640
Amaury Denoyellec2025c12021-12-03 15:03:36 +0100641 if ((htx->flags & HTX_FL_EOM) && htx_is_empty(htx))
642 qcs->flags |= QC_SF_FIN_STREAM;
643
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200644 out:
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200645 if (total) {
646 if (!(qcs->qcc->wait_event.events & SUB_RETRY_SEND))
647 tasklet_wakeup(qcs->qcc->wait_event.tasklet);
648 }
649
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200650 return total;
Amaury Denoyellef52151d2021-08-24 16:11:18 +0200651}
652
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100653/* Finalize the initialization of remotely initiated uni-stream <qcs>.
654 * Return 1 if succeeded, 0 if not. In this latter case, set the ->err h3 error
655 * to inform the QUIC mux layer of the encountered error.
656 */
657static int h3_attach_ruqs(struct qcs *qcs, void *ctx)
658{
659 uint64_t strm_type;
660 struct h3 *h3 = ctx;
661 struct buffer *rxbuf = &qcs->rx.buf;
662
663 /* First octets: the uni-stream type */
664 if (!b_quic_dec_int(&strm_type, rxbuf, NULL) || strm_type > H3_UNI_STRM_TP_MAX)
665 return 0;
666
667 /* Note that for all the uni-streams below, this is an error to receive two times the
668 * same type of uni-stream (even for Push stream which is not supported at this time.
669 */
670 switch (strm_type) {
671 case H3_UNI_STRM_TP_CONTROL_STREAM:
672 if (h3->rctrl.qcs) {
673 h3->err = H3_STREAM_CREATION_ERROR;
674 return 0;
675 }
676
677 h3->rctrl.qcs = qcs;
678 h3->rctrl.cb = h3_control_recv;
Amaury Denoyellea3f222d2021-12-06 11:24:00 +0100679 qcs_subscribe(qcs, SUB_RETRY_RECV, &h3->rctrl.wait_event);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100680 break;
681 case H3_UNI_STRM_TP_PUSH_STREAM:
682 /* NOT SUPPORTED */
683 break;
684 case H3_UNI_STRM_TP_QPACK_ENCODER:
685 if (h3->rqpack_enc.qcs) {
686 h3->err = H3_STREAM_CREATION_ERROR;
687 return 0;
688 }
689
690 h3->rqpack_enc.qcs = qcs;
691 h3->rqpack_enc.cb = qpack_decode_enc;
Amaury Denoyellea3f222d2021-12-06 11:24:00 +0100692 qcs_subscribe(qcs, SUB_RETRY_RECV, &h3->rqpack_enc.wait_event);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100693 break;
694 case H3_UNI_STRM_TP_QPACK_DECODER:
695 if (h3->rqpack_dec.qcs) {
696 h3->err = H3_STREAM_CREATION_ERROR;
697 return 0;
698 }
699
700 h3->rqpack_dec.qcs = qcs;
701 h3->rqpack_dec.cb = qpack_decode_dec;
Amaury Denoyellea3f222d2021-12-06 11:24:00 +0100702 qcs_subscribe(qcs, SUB_RETRY_RECV, &h3->rqpack_dec.wait_event);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100703 break;
704 default:
705 /* Error */
706 h3->err = H3_STREAM_CREATION_ERROR;
707 return 0;
708 }
709
710 return 1;
711}
712
713static int h3_finalize(void *ctx)
714{
715 struct h3 *h3 = ctx;
716
Amaury Denoyelleb7880542022-02-09 10:28:53 +0100717 h3->lctrl.qcs = qcs_new(h3->qcc, 0x3, QCS_SRV_UNI);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100718 if (!h3->lctrl.qcs)
719 return 0;
720
721 /* Wakeup ->lctrl uni-stream */
Frédéric Lécaillee16f0bd2021-08-23 09:50:29 +0200722 h3_control_send(&h3->lctrl, h3);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100723
724 return 1;
725}
726
727/* Tasklet dedicated to h3 incoming uni-streams */
728static struct task *h3_uqs_task(struct task *t, void *ctx, unsigned int state)
729{
730 struct h3_uqs *h3_uqs = ctx;
731 struct h3 *h3 = h3_uqs->qcs->qcc->ctx;
732
733 h3_uqs->cb(h3_uqs, h3);
734 return NULL;
735}
736
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100737/* Release all the tasklet attached to <h3_uqs> uni-stream */
738static inline void h3_uqs_tasklet_release(struct h3_uqs *h3_uqs)
739{
740 struct tasklet *t = h3_uqs->wait_event.tasklet;
741
742 if (t)
743 tasklet_free(t);
744}
745
746/* Release all the tasklet attached to <h3> uni-streams */
747static void h3_uqs_tasklets_release(struct h3 *h3)
748{
749 h3_uqs_tasklet_release(&h3->rqpack_enc);
750 h3_uqs_tasklet_release(&h3->rqpack_dec);
751 h3_uqs_tasklet_release(&h3->rctrl);
752}
753
754/* Tasklet dedicated to h3 outgoing uni-streams */
755__maybe_unused
756static struct task *h3_uqs_send_task(struct task *t, void *ctx, unsigned int state)
757{
758 struct h3_uqs *h3_uqs = ctx;
759 struct h3 *h3 = h3_uqs->qcs->qcc->ctx;
760
761 h3_uqs->cb(h3_uqs, h3);
762 return NULL;
763}
764
Ilya Shipitsinbd6b4be2021-10-15 16:18:21 +0500765/* Initialize <h3_uqs> uni-stream with <t> as tasklet */
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100766static int h3_uqs_init(struct h3_uqs *h3_uqs, struct h3 *h3,
767 int (*cb)(struct h3_uqs *h3_uqs, void *ctx),
768 struct task *(*t)(struct task *, void *, unsigned int))
769{
770 h3_uqs->qcs = NULL;
771 h3_uqs->cb = cb;
772 h3_uqs->wait_event.tasklet = tasklet_new();
773 if (!h3_uqs->wait_event.tasklet)
774 return 0;
775
776 h3_uqs->wait_event.tasklet->process = t;
777 h3_uqs->wait_event.tasklet->context = h3_uqs;
778 return 1;
779
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100780 tasklet_free(h3_uqs->wait_event.tasklet);
781 return 0;
782}
783
784static inline void h3_uqs_release(struct h3_uqs *h3_uqs)
785{
786 if (h3_uqs->qcs)
Amaury Denoyelledeed7772021-12-03 11:36:46 +0100787 uni_qcs_free(h3_uqs->qcs);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100788}
789
790static inline void h3_uqs_release_all(struct h3 *h3)
791{
792 h3_uqs_tasklet_release(&h3->lctrl);
793 h3_uqs_release(&h3->lctrl);
794 h3_uqs_tasklet_release(&h3->lqpack_enc);
795 h3_uqs_release(&h3->lqpack_enc);
796 h3_uqs_tasklet_release(&h3->lqpack_dec);
797 h3_uqs_release(&h3->lqpack_dec);
798}
799
800/* Initialize the HTTP/3 context for <qcc> mux.
801 * Return 1 if succeeded, 0 if not.
802 */
803static int h3_init(struct qcc *qcc)
804{
805 struct h3 *h3;
806
807 h3 = pool_alloc(pool_head_h3);
808 if (!h3)
809 goto fail_no_h3;
810
811 h3->qcc = qcc;
812 h3->err = H3_NO_ERROR;
813 h3->flags = 0;
814
815 if (!h3_uqs_init(&h3->rqpack_enc, h3, NULL, h3_uqs_task) ||
816 !h3_uqs_init(&h3->rqpack_dec, h3, NULL, h3_uqs_task) ||
817 !h3_uqs_init(&h3->rctrl, h3, h3_control_recv, h3_uqs_task))
818 goto fail_no_h3_ruqs;
819
820 if (!h3_uqs_init(&h3->lctrl, h3, h3_control_send, h3_uqs_task) ||
821 !h3_uqs_init(&h3->lqpack_enc, h3, NULL, h3_uqs_task) ||
822 !h3_uqs_init(&h3->lqpack_dec, h3, NULL, h3_uqs_task))
823 goto fail_no_h3_luqs;
824
825 qcc->ctx = h3;
826 LIST_INIT(&h3->buf_wait.list);
827
828 return 1;
829
830 fail_no_h3_ruqs:
831 h3_uqs_release_all(h3);
832 fail_no_h3_luqs:
833 h3_uqs_tasklets_release(h3);
834 pool_free(pool_head_h3, h3);
835 fail_no_h3:
836 return 0;
837}
838
839/* HTTP/3 application layer operations */
840const struct qcc_app_ops h3_ops = {
841 .init = h3_init,
842 .attach_ruqs = h3_attach_ruqs,
843 .decode_qcs = h3_decode_qcs,
Amaury Denoyelleabbe91e2021-11-12 16:09:29 +0100844 .snd_buf = h3_snd_buf,
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100845 .finalize = h3_finalize,
846};