blob: 9be2a5b11c18f16b12a14c5ea26c717fb0eb0783 [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>
26#include <haproxy/mux_quic.h>
27#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 */
Frédéric Lécailleccac11f2021-03-03 16:09:02 +010099static int h3_decode_qcs(struct qcs *qcs, void *ctx)
100{
101 struct buffer *rxbuf = &qcs->rx.buf;
102 struct h3 *h3 = ctx;
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200103 struct htx *htx;
104 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);
137 size_t len = b_data(rxbuf);
Amaury Denoyelle3cae4042021-11-08 08:57:18 +0100138 struct buffer htx_buf = BUF_NULL;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100139 struct buffer *tmp = get_trash_chunk();
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200140 struct ist meth = IST_NULL, path = IST_NULL;
Amaury Denoyelle3cae4042021-11-08 08:57:18 +0100141 //struct ist scheme = IST_NULL, authority = IST_NULL;
142 struct ist authority = IST_NULL;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100143
Amaury Denoyellefd7cdc32021-08-24 15:13:20 +0200144 if (qpack_decode_fs(buf, len, tmp, list) < 0) {
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100145 h3->err = QPACK_DECOMPRESSION_FAILED;
146 return -1;
147 }
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200148
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200149 b_alloc(&htx_buf);
150 htx = htx_from_buf(&htx_buf);
151
152 /* first treat pseudo-header to build the start line */
153 hdr_idx = 0;
154 while (1) {
155 if (isteq(list[hdr_idx].n, ist("")))
156 break;
157
158 if (istmatch(list[hdr_idx].n, ist(":"))) {
159 /* pseudo-header */
160 if (isteq(list[hdr_idx].n, ist(":method")))
161 meth = list[hdr_idx].v;
162 else if (isteq(list[hdr_idx].n, ist(":path")))
163 path = list[hdr_idx].v;
Amaury Denoyelle3cae4042021-11-08 08:57:18 +0100164 //else if (isteq(list[hdr_idx].n, ist(":scheme")))
165 // scheme = list[hdr_idx].v;
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200166 else if (isteq(list[hdr_idx].n, ist(":authority")))
167 authority = list[hdr_idx].v;
168 }
169
170 ++hdr_idx;
171 }
172
173 flags |= HTX_SL_F_VER_11;
174
175 sl = htx_add_stline(htx, HTX_BLK_REQ_SL, flags, meth, path, ist("HTTP/3.0"));
Amaury Denoyelleb9ce14e2021-11-08 09:13:42 +0100176 if (!sl)
177 goto fail;
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200178 sl->flags |= HTX_SL_F_BODYLESS;
179 sl->info.req.meth = find_http_meth(meth.ptr, meth.len);
180 BUG_ON(sl->info.req.meth == HTTP_METH_OTHER);
181
182 if (isttest(authority))
183 htx_add_header(htx, ist("host"), authority);
184
185 /* now treat standard headers */
186 hdr_idx = 0;
187 while (1) {
188 if (isteq(list[hdr_idx].n, ist("")))
189 break;
190
191 if (!istmatch(list[hdr_idx].n, ist(":")))
192 htx_add_header(htx, list[hdr_idx].n, list[hdr_idx].v);
193
194 ++hdr_idx;
195 }
196
197 htx_add_endof(htx, HTX_BLK_EOH);
198 htx_to_buf(htx, &htx_buf);
Amaury Denoyelle99043552021-08-24 15:36:02 +0200199
200 cs = cs_new(qcs->qcc->conn, qcs->qcc->conn->target);
201 cs->ctx = qcs;
202 stream_create_from_cs(cs, &htx_buf);
203
Ilya Shipitsinbd6b4be2021-10-15 16:18:21 +0500204 /* buffer is transferred to conn_stream and set to NULL
Amaury Denoyelle99043552021-08-24 15:36:02 +0200205 * except on stream creation error.
206 */
Amaury Denoyelleb49fa1a2021-08-24 15:30:12 +0200207 b_free(&htx_buf);
208
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100209 break;
210 }
211 case H3_FT_PUSH_PROMISE:
212 /* Not supported */
213 break;
214 default:
Amaury Denoyelled1acaf92021-11-15 15:52:55 +0100215 /* draft-ietf-quic-http34 9. Extensions to HTTP/3
216 * unknown frame types MUST be ignored
217 */
218 h3_debug_printf(stderr, "ignore unknown frame type 0x%lx\n", ftype);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100219 }
220 b_del(rxbuf, flen);
221 }
222
Amaury Denoyelleb9ce14e2021-11-08 09:13:42 +0100223 return 0;
224
225 fail:
226 return -1;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100227}
228
229/* Parse a SETTINGS frame which must not be truncated with <flen> as length from
230 * <rxbuf> buffer. This function does not update this buffer.
231 * Returns 0 if something wrong happened, 1 if not.
232 */
233static int h3_parse_settings_frm(struct h3 *h3, const struct buffer *rxbuf, size_t flen)
234{
235 uint64_t id, value;
236 const unsigned char *buf, *end;
237
238 buf = (const unsigned char *)b_head(rxbuf);
239 end = buf + flen;
240
241 while (buf <= end) {
242 if (!quic_dec_int(&id, &buf, end) || !quic_dec_int(&value, &buf, end))
243 return 0;
244
245 h3_debug_printf(stderr, "%s id: %llu value: %llu\n",
246 __func__, (unsigned long long)id, (unsigned long long)value);
247 switch (id) {
248 case H3_SETTINGS_QPACK_MAX_TABLE_CAPACITY:
249 h3->qpack_max_table_capacity = value;
250 break;
251 case H3_SETTINGS_MAX_FIELD_SECTION_SIZE:
252 h3->max_field_section_size = value;
253 break;
254 case H3_SETTINGS_QPACK_BLOCKED_STREAMS:
255 h3->qpack_blocked_streams = value;
256 break;
257 case H3_SETTINGS_RESERVED_2 ... H3_SETTINGS_RESERVED_5:
258 h3->err = H3_SETTINGS_ERROR;
259 return 0;
260 default:
261 /* MUST be ignored */
262 break;
263 }
264 }
265
266 return 1;
267}
268
269/* Decode <qcs> remotely initiated uni-stream. We stop parsing a frame as soon as
270 * there is not enough received data.
271 * Returns 0 if something wrong happened, 1 if not.
272 */
273static int h3_control_recv(struct h3_uqs *h3_uqs, void *ctx)
274{
275 struct buffer *rxbuf = &h3_uqs->qcs->rx.buf;
276 struct h3 *h3 = ctx;
277
278 h3_debug_printf(stderr, "%s STREAM ID: %llu\n", __func__, h3_uqs->qcs->by_id.key);
279 if (!b_data(rxbuf))
280 return 1;
281
282 while (b_data(rxbuf)) {
283 size_t hlen;
284 uint64_t ftype, flen;
285 struct buffer b;
286
287 /* Work on a copy of <rxbuf> */
288 b = h3_b_dup(rxbuf);
289 hlen = h3_decode_frm_header(&ftype, &flen, &b);
290 if (!hlen)
291 break;
292
293 h3_debug_printf(stderr, "%s: ftype: %llu, flen: %llu\n", __func__,
294 (unsigned long long)ftype, (unsigned long long)flen);
295 if (flen > b_data(&b))
296 break;
297
298 b_del(rxbuf, hlen);
299 /* From here, a frame must not be truncated */
300 switch (ftype) {
301 case H3_FT_CANCEL_PUSH:
302 break;
303 case H3_FT_SETTINGS:
304 if (!h3_parse_settings_frm(h3, rxbuf, flen))
305 return 0;
306 break;
307 case H3_FT_GOAWAY:
308 break;
309 case H3_FT_MAX_PUSH_ID:
310 break;
311 default:
312 /* Error */
313 h3->err = H3_FRAME_UNEXPECTED;
314 return 0;
315 }
316 b_del(rxbuf, flen);
317 }
318
319 if (b_data(rxbuf))
320 h3->qcc->conn->mux->ruqs_subscribe(h3_uqs->qcs, SUB_RETRY_RECV, &h3->rctrl.wait_event);
321
322 return 1;
323}
324
Amaury Denoyellea5871362021-10-07 16:26:12 +0200325/* Returns buffer for data sending.
326 * May be NULL if the allocation failed.
327 */
328static struct buffer *mux_get_buf(struct qcs *qcs)
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100329{
Amaury Denoyellea5871362021-10-07 16:26:12 +0200330 if (!b_size(&qcs->tx.buf))
331 b_alloc(&qcs->tx.buf);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100332
Amaury Denoyellea5871362021-10-07 16:26:12 +0200333 return &qcs->tx.buf;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100334}
335
336/* Function used to emit stream data from <h3_uqs> control uni-stream */
337static int h3_control_send(struct h3_uqs *h3_uqs, void *ctx)
338{
339 int ret;
340 struct h3 *h3 = ctx;
341 unsigned char data[(2 + 3) * 2 * QUIC_VARINT_MAX_SIZE]; /* enough for 3 settings */
Amaury Denoyellea5871362021-10-07 16:26:12 +0200342 struct buffer pos, *res;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100343
344 ret = 0;
Amaury Denoyellea5871362021-10-07 16:26:12 +0200345 pos = b_make((char *)data, sizeof(data), 0, 0);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100346 if (!(h3->flags & H3_CF_SETTINGS_SENT)) {
347 struct qcs *qcs = h3_uqs->qcs;
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100348 size_t frm_len;
349
350 frm_len = quic_int_getsize(H3_SETTINGS_QPACK_MAX_TABLE_CAPACITY) +
351 quic_int_getsize(h3_settings_qpack_max_table_capacity) +
352 quic_int_getsize(H3_SETTINGS_QPACK_BLOCKED_STREAMS) +
353 quic_int_getsize(h3_settings_qpack_blocked_streams);
354 if (h3_settings_max_field_section_size) {
355 frm_len += quic_int_getsize(H3_SETTINGS_MAX_FIELD_SECTION_SIZE) +
356 quic_int_getsize(h3_settings_max_field_section_size);
357 }
358
Amaury Denoyellea5871362021-10-07 16:26:12 +0200359 b_quic_enc_int(&pos, H3_UNI_STRM_TP_CONTROL_STREAM);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100360 /* Build a SETTINGS frame */
Amaury Denoyellea5871362021-10-07 16:26:12 +0200361 b_quic_enc_int(&pos, H3_FT_SETTINGS);
362 b_quic_enc_int(&pos, frm_len);
363 b_quic_enc_int(&pos, H3_SETTINGS_QPACK_MAX_TABLE_CAPACITY);
364 b_quic_enc_int(&pos, h3_settings_qpack_max_table_capacity);
365 b_quic_enc_int(&pos, H3_SETTINGS_QPACK_BLOCKED_STREAMS);
366 b_quic_enc_int(&pos, h3_settings_qpack_blocked_streams);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100367 if (h3_settings_max_field_section_size) {
Amaury Denoyellea5871362021-10-07 16:26:12 +0200368 b_quic_enc_int(&pos, H3_SETTINGS_MAX_FIELD_SECTION_SIZE);
369 b_quic_enc_int(&pos, h3_settings_max_field_section_size);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100370 }
Amaury Denoyellea5871362021-10-07 16:26:12 +0200371
372 res = mux_get_buf(qcs);
373 if (b_room(res) < b_data(&pos)) {
374 // TODO the mux should be put in blocked state, with
375 // the stream in state waiting for settings to be sent
376 ABORT_NOW();
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100377 }
378
Amaury Denoyellea5871362021-10-07 16:26:12 +0200379 ret = b_force_xfer(res, &pos, b_data(&pos));
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100380 if (ret > 0) {
381 h3->flags |= H3_CF_SETTINGS_SENT;
Amaury Denoyellea5871362021-10-07 16:26:12 +0200382 if (!(qcs->qcc->wait_event.events & SUB_RETRY_SEND))
383 tasklet_wakeup(qcs->qcc->wait_event.tasklet);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100384 }
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100385 }
386
387 return ret;
388}
389
Amaury Denoyelle15b09612021-08-24 16:20:27 +0200390static int h3_resp_headers_send(struct qcs *qcs, struct htx *htx)
391{
392 struct buffer outbuf;
393 struct buffer headers_buf = BUF_NULL;
394 struct buffer *res;
395 struct http_hdr list[global.tune.max_http_hdr];
396 struct htx_sl *sl;
397 struct htx_blk *blk;
398 enum htx_blk_type type;
399 int frame_length_size; /* size in bytes of frame length varint field */
400 int ret = 0;
401 int hdr;
402 int status = 0;
403
404 sl = NULL;
405 hdr = 0;
406 for (blk = htx_get_head_blk(htx); blk; blk = htx_get_next_blk(htx, blk)) {
407 type = htx_get_blk_type(blk);
408
409 if (type == HTX_BLK_UNUSED)
410 continue;
411
412 if (type == HTX_BLK_EOH)
413 break;
414
415 if (type == HTX_BLK_RES_SL) {
416 /* start-line -> HEADERS h3 frame */
417 BUG_ON(sl);
418 sl = htx_get_blk_ptr(htx, blk);
419 /* TODO should be on h3 layer */
420 status = sl->info.res.status;
421 }
422 else if (type == HTX_BLK_HDR) {
423 list[hdr].n = htx_get_blk_name(htx, blk);
424 list[hdr].v = htx_get_blk_value(htx, blk);
425 hdr++;
426 }
427 else {
428 ABORT_NOW();
429 goto err;
430 }
431 }
432
433 BUG_ON(!sl);
434
435 list[hdr].n = ist("");
436
Amaury Denoyelled3d97c62021-10-05 11:45:58 +0200437 res = mux_get_buf(qcs);
Amaury Denoyelle15b09612021-08-24 16:20:27 +0200438
439 /* At least 5 bytes to store frame type + length as a varint max size */
440 if (b_room(res) < 5)
441 ABORT_NOW();
442
443 b_reset(&outbuf);
444 outbuf = b_make(b_tail(res), b_contig_space(res), 0, 0);
445 /* Start the headers after frame type + length */
446 headers_buf = b_make(b_head(res) + 5, b_size(res) - 5, 0, 0);
447
448 if (qpack_encode_field_section_line(&headers_buf))
449 ABORT_NOW();
450 if (qpack_encode_int_status(&headers_buf, status))
451 ABORT_NOW();
452
453 for (hdr = 0; hdr < sizeof(list) / sizeof(list[0]); ++hdr) {
454 if (isteq(list[hdr].n, ist("")))
455 break;
456
457 if (qpack_encode_header(&headers_buf, list[hdr].n, list[hdr].v))
458 ABORT_NOW();
459 }
460
461 /* Now that all headers are encoded, we are certain that res buffer is
462 * big enough
463 */
464 frame_length_size = quic_int_getsize(b_data(&headers_buf));
465 res->head += 4 - frame_length_size;
466 b_putchr(res, 0x01); /* h3 HEADERS frame type */
467 if (!b_quic_enc_int(res, b_data(&headers_buf)))
468 ABORT_NOW();
469 b_add(res, b_data(&headers_buf));
470
471 ret = 0;
472 blk = htx_get_head_blk(htx);
473 while (blk) {
474 type = htx_get_blk_type(blk);
475 ret += htx_get_blksz(blk);
476 blk = htx_remove_blk(htx, blk);
477 if (type == HTX_BLK_EOH)
478 break;
479 }
480
Amaury Denoyelle42bb8aa2021-08-24 16:28:47 +0200481 if ((htx->flags & HTX_FL_EOM) && htx_is_empty(htx) && status >= 200)
482 qcs->flags |= QC_SF_FIN_STREAM;
483
Amaury Denoyelle15b09612021-08-24 16:20:27 +0200484 return ret;
485
486 err:
487 return 0;
488}
489
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200490/* Returns the total of bytes sent. */
491static int h3_resp_data_send(struct qcs *qcs, struct buffer *buf, size_t count)
492{
493 struct buffer outbuf;
494 struct buffer *res;
495 size_t total = 0;
496 struct htx *htx;
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200497 int bsize, fsize, hsize;
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200498 struct htx_blk *blk;
499 enum htx_blk_type type;
500
501 htx = htx_from_buf(buf);
502
503 new_frame:
504 if (!count || htx_is_empty(htx))
505 goto end;
506
507 blk = htx_get_head_blk(htx);
508 type = htx_get_blk_type(blk);
509 fsize = bsize = htx_get_blksz(blk);
510
511 if (type != HTX_BLK_DATA)
512 goto end;
513
Amaury Denoyelled3d97c62021-10-05 11:45:58 +0200514 res = mux_get_buf(qcs);
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200515
516 if (fsize > count)
517 fsize = count;
518
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200519 /* h3 DATA headers : 1-byte frame type + varint frame length */
520 hsize = 1 + QUIC_VARINT_MAX_SIZE;
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200521
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200522 while (1) {
523 b_reset(&outbuf);
524 outbuf = b_make(b_tail(res), b_contig_space(res), 0, 0);
525 if (b_size(&outbuf) > hsize || !b_space_wraps(res))
526 break;
527 b_slow_realign(res, trash.area, b_data(res));
528 }
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200529
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200530 /* not enough room for headers and at least one data byte, block the
531 * stream
532 */
533 if (b_size(&outbuf) <= hsize) {
534 qcs->flags |= QC_SF_BLK_MROOM;
535 goto end;
536 }
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200537
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200538 if (b_size(&outbuf) < hsize + fsize)
539 fsize = b_size(&outbuf) - hsize;
540 BUG_ON(fsize <= 0);
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200541
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200542 b_putchr(&outbuf, 0x00); /* h3 frame type = DATA */
543 b_quic_enc_int(&outbuf, fsize); /* h3 frame length */
544
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200545 b_putblk(&outbuf, htx_get_blk_ptr(htx, blk), fsize);
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200546 total += fsize;
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200547 count -= fsize;
548
549 if (fsize == bsize)
550 htx_remove_blk(htx, blk);
551 else
552 htx_cut_data_blk(htx, blk, fsize);
553
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200554 /* commit the buffer */
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200555 b_add(res, b_data(&outbuf));
556 goto new_frame;
557
558 end:
559 return total;
560}
561
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200562size_t h3_snd_buf(struct conn_stream *cs, struct buffer *buf, size_t count, int flags)
563{
564 size_t total = 0;
565 struct qcs *qcs = cs->ctx;
566 struct htx *htx;
567 enum htx_blk_type btype;
568 struct htx_blk *blk;
569 uint32_t bsize;
570 int32_t idx;
571 int ret;
572
573 htx = htx_from_buf(buf);
574
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200575 while (count && !htx_is_empty(htx) && !(qcs->flags & QC_SF_BLK_MROOM)) {
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200576 idx = htx_get_head(htx);
577 blk = htx_get_blk(htx, idx);
578 btype = htx_get_blk_type(blk);
579 bsize = htx_get_blksz(blk);
580
581 /* Not implemented : QUIC on backend side */
582 BUG_ON(btype == HTX_BLK_REQ_SL);
583
584 switch (btype) {
585 case HTX_BLK_RES_SL:
Amaury Denoyelle15b09612021-08-24 16:20:27 +0200586 /* start-line -> HEADERS h3 frame */
587 ret = h3_resp_headers_send(qcs, htx);
588 if (ret > 0) {
589 total += ret;
590 count -= ret;
591 if (ret < bsize)
592 goto out;
593 }
594 break;
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200595
596 case HTX_BLK_DATA:
Amaury Denoyelle8e2a9982021-08-24 16:24:37 +0200597 ret = h3_resp_data_send(qcs, buf, count);
598 if (ret > 0) {
599 htx = htx_from_buf(buf);
600 total += ret;
601 count -= ret;
602 if (ret < bsize)
603 goto out;
604 }
605 break;
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200606
607 case HTX_BLK_TLR:
608 case HTX_BLK_EOT:
609 /* TODO trailers */
610
611 default:
612 htx_remove_blk(htx, blk);
613 total += bsize;
614 count -= bsize;
615 break;
616 }
617 }
618
Amaury Denoyelle42bb8aa2021-08-24 16:28:47 +0200619 if ((htx->flags & HTX_FL_EOM) && htx_is_empty(htx))
620 qcs->flags |= QC_SF_FIN_STREAM;
Amaury Denoyellef52151d2021-08-24 16:11:18 +0200621
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200622 out:
Amaury Denoyellea543eb12021-10-06 14:53:13 +0200623 if (total) {
624 if (!(qcs->qcc->wait_event.events & SUB_RETRY_SEND))
625 tasklet_wakeup(qcs->qcc->wait_event.tasklet);
626 }
627
Amaury Denoyelle26dfd902021-08-24 16:33:53 +0200628 return total;
Amaury Denoyellef52151d2021-08-24 16:11:18 +0200629}
630
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100631/* Finalize the initialization of remotely initiated uni-stream <qcs>.
632 * Return 1 if succeeded, 0 if not. In this latter case, set the ->err h3 error
633 * to inform the QUIC mux layer of the encountered error.
634 */
635static int h3_attach_ruqs(struct qcs *qcs, void *ctx)
636{
637 uint64_t strm_type;
638 struct h3 *h3 = ctx;
639 struct buffer *rxbuf = &qcs->rx.buf;
640
641 /* First octets: the uni-stream type */
642 if (!b_quic_dec_int(&strm_type, rxbuf, NULL) || strm_type > H3_UNI_STRM_TP_MAX)
643 return 0;
644
645 /* Note that for all the uni-streams below, this is an error to receive two times the
646 * same type of uni-stream (even for Push stream which is not supported at this time.
647 */
648 switch (strm_type) {
649 case H3_UNI_STRM_TP_CONTROL_STREAM:
650 if (h3->rctrl.qcs) {
651 h3->err = H3_STREAM_CREATION_ERROR;
652 return 0;
653 }
654
655 h3->rctrl.qcs = qcs;
656 h3->rctrl.cb = h3_control_recv;
657 h3->qcc->conn->mux->ruqs_subscribe(qcs, SUB_RETRY_RECV, &h3->rctrl.wait_event);
658 break;
659 case H3_UNI_STRM_TP_PUSH_STREAM:
660 /* NOT SUPPORTED */
661 break;
662 case H3_UNI_STRM_TP_QPACK_ENCODER:
663 if (h3->rqpack_enc.qcs) {
664 h3->err = H3_STREAM_CREATION_ERROR;
665 return 0;
666 }
667
668 h3->rqpack_enc.qcs = qcs;
669 h3->rqpack_enc.cb = qpack_decode_enc;
670 h3->qcc->conn->mux->ruqs_subscribe(qcs, SUB_RETRY_RECV, &h3->rqpack_enc.wait_event);
671 break;
672 case H3_UNI_STRM_TP_QPACK_DECODER:
673 if (h3->rqpack_dec.qcs) {
674 h3->err = H3_STREAM_CREATION_ERROR;
675 return 0;
676 }
677
678 h3->rqpack_dec.qcs = qcs;
679 h3->rqpack_dec.cb = qpack_decode_dec;
680 h3->qcc->conn->mux->ruqs_subscribe(qcs, SUB_RETRY_RECV, &h3->rqpack_dec.wait_event);
681 break;
682 default:
683 /* Error */
684 h3->err = H3_STREAM_CREATION_ERROR;
685 return 0;
686 }
687
688 return 1;
689}
690
691static int h3_finalize(void *ctx)
692{
693 struct h3 *h3 = ctx;
694
695 h3->lctrl.qcs = luqs_new(h3->qcc);
696 if (!h3->lctrl.qcs)
697 return 0;
698
699 /* Wakeup ->lctrl uni-stream */
Frédéric Lécaillee16f0bd2021-08-23 09:50:29 +0200700 h3_control_send(&h3->lctrl, h3);
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100701
702 return 1;
703}
704
705/* Tasklet dedicated to h3 incoming uni-streams */
706static struct task *h3_uqs_task(struct task *t, void *ctx, unsigned int state)
707{
708 struct h3_uqs *h3_uqs = ctx;
709 struct h3 *h3 = h3_uqs->qcs->qcc->ctx;
710
711 h3_uqs->cb(h3_uqs, h3);
712 return NULL;
713}
714
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100715/* Release all the tasklet attached to <h3_uqs> uni-stream */
716static inline void h3_uqs_tasklet_release(struct h3_uqs *h3_uqs)
717{
718 struct tasklet *t = h3_uqs->wait_event.tasklet;
719
720 if (t)
721 tasklet_free(t);
722}
723
724/* Release all the tasklet attached to <h3> uni-streams */
725static void h3_uqs_tasklets_release(struct h3 *h3)
726{
727 h3_uqs_tasklet_release(&h3->rqpack_enc);
728 h3_uqs_tasklet_release(&h3->rqpack_dec);
729 h3_uqs_tasklet_release(&h3->rctrl);
730}
731
732/* Tasklet dedicated to h3 outgoing uni-streams */
733__maybe_unused
734static struct task *h3_uqs_send_task(struct task *t, void *ctx, unsigned int state)
735{
736 struct h3_uqs *h3_uqs = ctx;
737 struct h3 *h3 = h3_uqs->qcs->qcc->ctx;
738
739 h3_uqs->cb(h3_uqs, h3);
740 return NULL;
741}
742
Ilya Shipitsinbd6b4be2021-10-15 16:18:21 +0500743/* Initialize <h3_uqs> uni-stream with <t> as tasklet */
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100744static int h3_uqs_init(struct h3_uqs *h3_uqs, struct h3 *h3,
745 int (*cb)(struct h3_uqs *h3_uqs, void *ctx),
746 struct task *(*t)(struct task *, void *, unsigned int))
747{
748 h3_uqs->qcs = NULL;
749 h3_uqs->cb = cb;
750 h3_uqs->wait_event.tasklet = tasklet_new();
751 if (!h3_uqs->wait_event.tasklet)
752 return 0;
753
754 h3_uqs->wait_event.tasklet->process = t;
755 h3_uqs->wait_event.tasklet->context = h3_uqs;
756 return 1;
757
758 err:
759 tasklet_free(h3_uqs->wait_event.tasklet);
760 return 0;
761}
762
763static inline void h3_uqs_release(struct h3_uqs *h3_uqs)
764{
765 if (h3_uqs->qcs)
766 qcs_release(h3_uqs->qcs);
767}
768
769static inline void h3_uqs_release_all(struct h3 *h3)
770{
771 h3_uqs_tasklet_release(&h3->lctrl);
772 h3_uqs_release(&h3->lctrl);
773 h3_uqs_tasklet_release(&h3->lqpack_enc);
774 h3_uqs_release(&h3->lqpack_enc);
775 h3_uqs_tasklet_release(&h3->lqpack_dec);
776 h3_uqs_release(&h3->lqpack_dec);
777}
778
779/* Initialize the HTTP/3 context for <qcc> mux.
780 * Return 1 if succeeded, 0 if not.
781 */
782static int h3_init(struct qcc *qcc)
783{
784 struct h3 *h3;
785
786 h3 = pool_alloc(pool_head_h3);
787 if (!h3)
788 goto fail_no_h3;
789
790 h3->qcc = qcc;
791 h3->err = H3_NO_ERROR;
792 h3->flags = 0;
793
794 if (!h3_uqs_init(&h3->rqpack_enc, h3, NULL, h3_uqs_task) ||
795 !h3_uqs_init(&h3->rqpack_dec, h3, NULL, h3_uqs_task) ||
796 !h3_uqs_init(&h3->rctrl, h3, h3_control_recv, h3_uqs_task))
797 goto fail_no_h3_ruqs;
798
799 if (!h3_uqs_init(&h3->lctrl, h3, h3_control_send, h3_uqs_task) ||
800 !h3_uqs_init(&h3->lqpack_enc, h3, NULL, h3_uqs_task) ||
801 !h3_uqs_init(&h3->lqpack_dec, h3, NULL, h3_uqs_task))
802 goto fail_no_h3_luqs;
803
804 qcc->ctx = h3;
805 LIST_INIT(&h3->buf_wait.list);
806
807 return 1;
808
809 fail_no_h3_ruqs:
810 h3_uqs_release_all(h3);
811 fail_no_h3_luqs:
812 h3_uqs_tasklets_release(h3);
813 pool_free(pool_head_h3, h3);
814 fail_no_h3:
815 return 0;
816}
817
818/* HTTP/3 application layer operations */
819const struct qcc_app_ops h3_ops = {
820 .init = h3_init,
821 .attach_ruqs = h3_attach_ruqs,
822 .decode_qcs = h3_decode_qcs,
Amaury Denoyelleabbe91e2021-11-12 16:09:29 +0100823 .snd_buf = h3_snd_buf,
Frédéric Lécailleccac11f2021-03-03 16:09:02 +0100824 .finalize = h3_finalize,
825};