varnish-cache/bin/varnishtest/vtc_http2.c
0
/*-
1
 * Copyright (c) 2008-2016 Varnish Software AS
2
 * All rights reserved.
3
 *
4
 * Author: Guillaume Quintard <guillaume.quintard@gmail.com>
5
 *
6
 * SPDX-License-Identifier: BSD-2-Clause
7
 *
8
 * Redistribution and use in source and binary forms, with or without
9
 * modification, are permitted provided that the following conditions
10
 * are met:
11
 * 1. Redistributions of source code must retain the above copyright
12
 *    notice, this list of conditions and the following disclaimer.
13
 * 2. Redistributions in binary form must reproduce the above copyright
14
 *    notice, this list of conditions and the following disclaimer in the
15
 *    documentation and/or other materials provided with the distribution.
16
 *
17
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
21
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27
 * SUCH DAMAGE.
28
 */
29
30
#include "config.h"
31
32
#include <sys/types.h>
33
#include <sys/socket.h>
34
35
#include <math.h>
36
#include <poll.h>
37 27056
#include <stdio.h>
38
#include <stdlib.h>
39
#include <unistd.h>
40
#include <string.h>
41
#include <netinet/in.h>
42
43
#include "vtc.h"
44
#include "vtc_http.h"
45
46 26936
#include "vfil.h"
47
#include "hpack.h"
48
#include "vend.h"
49
50
#define ERR_MAX         13
51
#define BUF_SIZE        (1024*2048)
52
53
static const char *const h2_errs[] = {
54
#define H2_ERROR(n,v,sc,g,r,t) [v] = #n,
55 25256
#include <tbl/h2_error.h>
56
        NULL
57
};
58
59
static const char *const h2_types[] = {
60
#define H2_FRAME(l,u,t,f,...) [t] = #u,
61
#include <tbl/h2_frames.h>
62
        NULL
63
};
64 25216
65
static const char * const h2_settings[] = {
66
        [0] = "unknown",
67
#define H2_SETTING(U,l,v,...) [v] = #U,
68
#include <tbl/h2_settings.h>
69
        NULL
70
};
71
72
enum h2_settings_e {
73 25096
#define H2_SETTING(U,l,v,...) SETTINGS_##U = v,
74
#include <tbl/h2_settings.h>
75
        SETTINGS_MAX
76
};
77
78
79
enum h2_type_e {
80
#define H2_FRAME(l,u,t,f,...) TYPE_##u = t,
81
#include <tbl/h2_frames.h>
82 25096
        TYPE_MAX
83
};
84
85
//lint -save -e849      Same enum value
86
enum {
87
        ACK = 0x1,
88
        END_STREAM = 0x1,
89
        PADDED = 0x8,
90
        END_HEADERS = 0x4,
91 25016
        PRIORITY = 0x20,
92
};
93
//lint -restore
94
95
struct stream {
96
        unsigned                magic;
97
#define STREAM_MAGIC            0x63f1fac2
98
        uint32_t                id;
99
        struct vtclog           *vl;
100 24856
        char                    *spec;
101
        char                    *name;
102
        VTAILQ_ENTRY(stream)    list;
103
        unsigned                running;
104
        pthread_cond_t          cond;
105
        struct frame            *frame;
106
        pthread_t               tp;
107
        struct http             *hp;
108
        int64_t                 win_self;
109 24776
        int64_t                 win_peer;
110
        int                     wf;
111
112
        VTAILQ_HEAD(, frame)   fq;
113
114
        char                    *body;
115
        long                    bodylen;
116
        struct hpk_hdr          req[MAX_HDR];
117
        struct hpk_hdr          resp[MAX_HDR];
118 24656
119
        int                     dependency;
120
        int                     weight;
121
};
122
123
static void
124 43800
clean_headers(struct hpk_hdr *h)
125
{
126 43800
        unsigned n = MAX_HDR;
127 24536
128 79879
        while (h->t && n > 0) {
129 36079
                if (h->key.len)
130 36000
                        free(h->key.ptr);
131 36079
                if (h->value.len)
132 36000
                        free(h->value.ptr);
133 36079
                memset(h, 0, sizeof(*h));
134 36079
                h++;
135 36079
                n--;
136 24536
        }
137 43800
}
138
139
#define ONLY_H2_CLIENT(hp, av)                                          \
140
        do {                                                            \
141
                if (hp->sfd != NULL)                                    \
142
                        vtc_fatal(s->vl,                                \
143
                            "\"%s\" only possible in client", av[0]);   \
144
        } while (0)
145 24336
146
#define ONLY_H2_SERVER(hp, av)                                          \
147
        do {                                                            \
148
                if (hp->sfd == NULL)                                    \
149
                        vtc_fatal(s->vl,                                \
150
                            "\"%s\" only possible in server", av[0]);   \
151
        } while (0)
152
153
static void
154 26576
http_write(const struct http *hp, int lvl,
155
    const char *buf, int s, const char *pfx)
156
{
157
        ssize_t l;
158
159 2240
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
160 2240
        AN(buf);
161 2240
        AN(pfx);
162
163 2240
        vtc_dump(hp->vl, lvl, pfx, buf, s);
164 2240
        l = write(hp->sess->fd, buf, s);
165 2240
        if (l != s)
166 0
                vtc_log(hp->vl, hp->fatal, "Write failed: (%zd vs %d) %s",
167 0
                    l, s, strerror(errno));
168 2240
}
169
170
static int
171 57096
get_bytes(const struct http *hp, char *buf, size_t n)
172
{
173
        int i;
174
        struct pollfd pfd[1];
175
176 57096
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
177 57096
        AN(buf);
178
179 114189
        while (n > 0) {
180 57093
                pfd[0].fd = hp->sess->fd;
181 57093
                pfd[0].events = POLLIN;
182 57093
                pfd[0].revents = 0;
183 57093
                i = poll(pfd, 1, (int)(hp->timeout * 1000));
184 57093
                if (i < 0 && errno == EINTR)
185 0
                        continue;
186 57093
                if (i == 0)
187 0
                        vtc_log(hp->vl, 3,
188
                            "HTTP2 rx timeout (fd:%d %.3fs)",
189 0
                            hp->sess->fd, hp->timeout);
190 57093
                if (i < 0)
191 0
                        vtc_log(hp->vl, 3,
192
                            "HTTP2 rx failed (fd:%d poll: %s)",
193 0
                            hp->sess->fd, strerror(errno));
194 57093
                if (i <= 0)
195 0
                        return (i);
196 57093
                i = read(hp->sess->fd, buf, n);
197 57093
                if (!(pfd[0].revents & POLLIN))
198 0
                        vtc_log(hp->vl, 4,
199
                            "HTTP2 rx poll (fd:%d revents: %x n=%zu, i=%d)",
200 0
                            hp->sess->fd, pfd[0].revents, n, i);
201 57093
                if (i == 0)
202 0
                        vtc_log(hp->vl, 3,
203
                            "HTTP2 rx EOF (fd:%d read: %s)",
204 0
                            hp->sess->fd, strerror(errno));
205 57093
                if (i < 0)
206 0
                        vtc_log(hp->vl, 3,
207
                            "HTTP2 rx failed (fd:%d read: %s)",
208 0
                            hp->sess->fd, strerror(errno));
209 57093
                if (i <= 0)
210 0
                        return (i);
211 57093
                n -= i;
212
        }
213 57096
        return (1);
214
215 57096
}
216
217
VTAILQ_HEAD(fq_head, frame);
218
219
struct frame {
220
        unsigned        magic;
221
#define FRAME_MAGIC     0x5dd3ec4
222
        uint32_t        size;
223
        uint32_t        stid;
224
        uint8_t         type;
225
        uint8_t         flags;
226
        char            *data;
227
228
        VTAILQ_ENTRY(frame)    list;
229
230
        union {
231
                struct {
232
                        uint32_t stream;
233
                        uint8_t  exclusive;
234
                        uint8_t  weight;
235
                }               prio;
236
                uint32_t        rst_err;
237
                double settings[SETTINGS_MAX+1];
238
                struct {
239
                        char data[9];
240
                        int ack;
241
                }               ping;
242
                struct {
243
                        uint32_t err;
244
                        uint32_t stream;
245
                        char     *debug;
246
                }               goaway;
247
                uint32_t        winup_size;
248
                uint32_t        promised;
249
                uint8_t         padded;
250
        } md;
251
};
252
253
static void
254 34002
readFrameHeader(struct frame *f, const char *buf)
255
{
256 34002
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
257 34002
        AN(buf);
258
259 34002
        f->size  = (unsigned char)buf[0] << 16;
260 34002
        f->size += (unsigned char)buf[1] << 8;
261 34002
        f->size += (unsigned char)buf[2];
262
263 34002
        f->type = (unsigned char)buf[3];
264
265 34002
        f->flags = (unsigned char)buf[4];
266
267 34002
        f->stid  = vbe32dec(buf+5);
268 34002
}
269
270
static void
271 82439
writeFrameHeader(char *buf, const struct frame *f)
272
{
273 82439
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
274 82439
        AN(buf);
275 82439
        buf[0] = (f->size >> 16) & 0xff;
276 82439
        buf[1] = (f->size >>  8) & 0xff;
277 82439
        buf[2] = (f->size      ) & 0xff;
278
279 82439
        buf[3] = f->type;
280
281 82439
        buf[4] = f->flags;
282
283 82439
        vbe32enc(buf + 5, f->stid);
284 82439
}
285
286
#define INIT_FRAME(f, ty, sz, id, fl) \
287
do { \
288
        f.magic = FRAME_MAGIC; \
289
        f.type = TYPE_ ## ty; \
290
        f.size = sz; \
291
        f.stid = id; \
292
        f.flags = fl; \
293
        f.data = NULL; \
294
} while(0)
295
296
static void
297 75065
replace_frame(struct frame **fp, struct frame *new)
298
{
299
        struct frame *old;
300
301 75065
        AN(fp);
302 75065
        CHECK_OBJ_ORNULL(new, FRAME_MAGIC);
303
304 75065
        old = *fp;
305 75065
        *fp = new;
306 75065
        if (old == NULL)
307 41036
                return;
308
309 34029
        CHECK_OBJ(old, FRAME_MAGIC);
310 34029
        if (old->type == TYPE_GOAWAY)
311 1400
                free(old->md.goaway.debug);
312 34029
        free(old->data);
313 34029
        FREE_OBJ(old);
314 75065
}
315
316
static void
317 55556
clean_frame(struct frame **fp)
318
{
319
320 55556
        replace_frame(fp, NULL);
321 55556
}
322
323
static void
324 82438
write_frame(struct stream *sp, const struct frame *f, const unsigned lock)
325
{
326
        struct http *hp;
327
        ssize_t l;
328
        char hdr[9];
329
330 82438
        CHECK_OBJ_NOTNULL(sp, STREAM_MAGIC);
331 82438
        hp = sp->hp;
332 82438
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
333 82438
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
334
335 82438
        writeFrameHeader(hdr, f);
336
337 164876
        vtc_log(sp->vl, 3,
338
            "tx: stream: %d, type: %s (%d), flags: 0x%02x, size: %d",
339 82438
            f->stid,
340 82438
            f->type < TYPE_MAX ? h2_types[f->type] : "?",
341 82438
            f->type, f->flags, f->size);
342
343 82438
        if (f->type == TYPE_DATA) {
344 13440
                sp->win_peer -= f->size;
345 13440
                hp->h2_win_peer->size -= f->size;
346 13440
        }
347
348 82438
        if (lock)
349 66920
                PTOK(pthread_mutex_lock(&hp->mtx));
350 82438
        l = write(hp->sess->fd, hdr, sizeof(hdr));
351 82438
        if (l != sizeof(hdr))
352 78960
                vtc_log(sp->vl, hp->fatal, "Write failed: (%zd vs %zd) %s",
353 39480
                    l, sizeof(hdr), strerror(errno));
354
355 82438
        if (f->size) {
356 67280
                AN(f->data);
357 67280
                l = write(hp->sess->fd, f->data, f->size);
358 67280
                if (l != f->size)
359 78960
                        vtc_log(sp->vl, hp->fatal,
360
                                        "Write failed: (%zd vs %d) %s",
361 39480
                                        l, f->size, strerror(errno));
362 67280
        }
363 82438
        if (lock)
364 66920
                PTOK(pthread_mutex_unlock(&hp->mtx));
365 82438
}
366
367
static void
368 240
exclusive_stream_dependency(const struct stream *s)
369
{
370
        struct stream *target;
371 240
        struct http *hp = s->hp;
372
373 240
        if (s->id == 0)
374 0
                return;
375
376 1120
        VTAILQ_FOREACH(target, &hp->streams, list) {
377 880
                if (target->id != s->id && target->dependency == s->dependency)
378 320
                        target->dependency = s->id;
379 880
        }
380 240
}
381
382
static void
383 34053
explain_flags(uint8_t flags, uint8_t type, struct vtclog *vl)
384
{
385 34053
        if (flags & ACK && (type == TYPE_PING || type == TYPE_SETTINGS)) {
386 8000
                vtc_log(vl, 3, "flag: ACK");
387 36013
        } else if (flags & END_STREAM && (type == TYPE_HEADERS ||
388 1960
            type == TYPE_PUSH_PROMISE || type == TYPE_DATA)) {
389 6920
                vtc_log(vl, 3, "flag: END_STREAM");
390 26293
        } else if (flags & END_HEADERS && (type == TYPE_HEADERS ||
391 280
            type == TYPE_PUSH_PROMISE || type == TYPE_CONTINUATION)) {
392 2320
                vtc_log(vl, 3, "flag: END_TYPE_HEADERS");
393 19133
        } else if (flags & PRIORITY && (type == TYPE_HEADERS ||
394 0
            type == TYPE_PUSH_PROMISE)) {
395 0
                vtc_log(vl, 3, "flag: END_PRIORITY");
396 16813
        } else if (flags & PADDED && (type == TYPE_DATA || type ==
397 0
            TYPE_HEADERS || type == TYPE_PUSH_PROMISE)) {
398 40
                vtc_log(vl, 3, "flag: PADDED");
399 16813
        } else if (flags)
400 0
                vtc_log(vl, 3, "UNKNOWN FLAG(S): 0x%02x", flags);
401 34053
}
402
403
static void
404 4000
parse_data(struct stream *s, struct frame *f)
405
{
406
        struct http *hp;
407 4000
        uint32_t size = f->size;
408 4000
        char *data = f->data;
409
410 4000
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
411 4000
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
412 4000
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
413
414 4000
        if (f->flags & PADDED) {
415 160
                f->md.padded = *((uint8_t *)data);
416 160
                if (f->md.padded >= size) {
417 0
                        vtc_log(s->vl, hp->fatal,
418
                                        "invalid padding: %d reported,"
419
                                        "but size is only %d",
420 0
                                        f->md.padded, size);
421 0
                        size = 0;
422 0
                        f->md.padded = 0;
423 0
                }
424 160
                data++;
425 160
                size -= f->md.padded + 1;
426 160
                vtc_log(s->vl, 4, "padding: %3d", f->md.padded);
427 160
        }
428
429 4000
        if (s->id)
430 3920
                s->win_self -= size;
431
432 4000
        s->hp->h2_win_self->size -= size;
433
434 4000
        if (!size) {
435 1120
                AZ(data);
436 1120
                vtc_log(s->vl, 4, "s%u - no data", s->id);
437 1120
                return;
438
        }
439
440 2880
        s->body = realloc(s->body, s->bodylen + size + 1L);
441 2880
        AN(s->body);
442 2880
        memcpy(s->body + s->bodylen, data, size);
443 2880
        s->bodylen += size;
444 2880
        s->body[s->bodylen] = '\0';
445 4000
}
446
447
static void
448 7080
decode_hdr(struct http *hp, struct hpk_hdr *h, const struct vsb *vsb)
449
{
450
        struct hpk_iter *iter;
451 7080
        enum hpk_result r = hpk_err;
452
        int n;
453
454 7080
        CHECK_OBJ_NOTNULL(vsb, VSB_MAGIC);
455 7080
        CAST_OBJ_NOTNULL(hp, hp, HTTP_MAGIC);;
456
457 7080
        if (VSB_len(vsb) == 0)
458 0
                return;
459
460 7080
        iter = HPK_NewIter(hp->decctx, VSB_data(vsb), VSB_len(vsb));
461
462 7080
        n = 0;
463 7080
        while (n < MAX_HDR && h[n].t)
464 0
                n++;
465 36080
        while (n < MAX_HDR) {
466 36080
                r = HPK_DecHdr(iter, h + n);
467 36080
                if (r == hpk_err )
468 80
                        break;
469 72000
                vtc_log(hp->vl, 4, "header[%2d]: %s: %s",
470 36000
                    n, h[n].key.ptr, h[n].value.ptr);
471 36000
                n++;
472 36000
                if (r == hpk_done)
473 7000
                        break;
474
        }
475
476 7080
        if (r != hpk_done) {
477 160
                vtc_log(hp->vl, hp->fatal ? 4 : 0,
478 80
                    "Header decoding failed (%d) %d", r, hp->fatal);
479 7080
        } else if (n == MAX_HDR) {
480 0
                vtc_log(hp->vl, hp->fatal,
481
                    "Max number of headers reached (%d)", MAX_HDR);
482 0
        }
483
484 7080
        HPK_FreeIter(iter);
485 7080
}
486
487
static void
488 7640
parse_hdr(struct stream *s, struct frame *f, struct vsb *vsb)
489
{
490 7640
        int shift = 0;
491 7640
        int exclusive = 0;
492 7640
        uint32_t size = f->size;
493 7640
        char *data = f->data;
494
        struct http *hp;
495
        uint32_t n;
496
497 7640
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
498 7640
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
499 7640
        CHECK_OBJ_NOTNULL(vsb, VSB_MAGIC);
500 7640
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
501
502 7640
        if (f->flags & PADDED && f->type != TYPE_CONTINUATION) {
503 80
                f->md.padded = *((uint8_t *)data);
504 80
                if (f->md.padded >= size) {
505 0
                        vtc_log(s->vl, hp->fatal,
506
                                        "invalid padding: %d reported,"
507
                                        "but size is only %d",
508 0
                                        f->md.padded, size);
509 0
                        size = 0;
510 0
                        f->md.padded = 0;
511 0
                }
512 80
                shift += 1;
513 80
                size -= f->md.padded;
514 80
                vtc_log(s->vl, 4, "padding: %3d", f->md.padded);
515 80
        }
516
517 7640
        if (f->type == TYPE_HEADERS && f->flags & PRIORITY){
518 80
                shift += 5;
519 80
                n = vbe32dec(f->data);
520 80
                s->dependency = n & ~(1U << 31);
521 80
                exclusive = n >> 31;
522
523 80
                s->weight = f->data[4];
524 80
                if (exclusive)
525 80
                        exclusive_stream_dependency(s);
526
527 80
                vtc_log(s->vl, 4, "stream->dependency: %u", s->dependency);
528 80
                vtc_log(s->vl, 4, "stream->weight: %u", s->weight);
529 7640
        } else if (f->type == TYPE_PUSH_PROMISE){
530 40
                shift += 4;
531 40
                n = vbe32dec(f->data);
532 40
                f->md.promised = n & ~(1U << 31);
533 40
        }
534
535 7640
        AZ(VSB_bcat(vsb, data + shift, size - shift));
536 7640
}
537
538
static void
539 360
parse_prio(struct stream *s, struct frame *f)
540
{
541
        struct http *hp;
542
        char *buf;
543
        uint32_t n;
544
545 360
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
546 360
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
547 360
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
548
549 360
        if (f->size != 5)
550 0
                vtc_fatal(s->vl, "Size should be 5, but isn't (%d)", f->size);
551
552 360
        buf = f->data;
553 360
        AN(buf);
554
555 360
        n = vbe32dec(f->data);
556 360
        f->md.prio.stream = n & ~(1U << 31);
557
558 360
        s->dependency = f->md.prio.stream;
559 360
        if (n >> 31){
560 40
                f->md.prio.exclusive = 1;
561 40
                exclusive_stream_dependency(s);
562 40
        }
563
564 360
        buf += 4;
565 360
        f->md.prio.weight = *buf;
566 360
        s->weight = f->md.prio.weight;
567
568 360
        vtc_log(s->vl, 3, "prio->stream: %u", f->md.prio.stream);
569 360
        vtc_log(s->vl, 3, "prio->weight: %u", f->md.prio.weight);
570 360
}
571
572
static void
573 2173
parse_rst(const struct stream *s, struct frame *f)
574
{
575
        struct http *hp;
576
        uint32_t err;
577
        const char *buf;
578 2173
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
579 2173
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
580 2173
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
581
582 2173
        if (f->size != 4)
583 0
                vtc_fatal(s->vl, "Size should be 4, but isn't (%d)", f->size);
584
585 2173
        err = vbe32dec(f->data);
586 2173
        f->md.rst_err = err;
587
588 2173
        vtc_log(s->vl, 2, "ouch");
589 2173
        if (err <= ERR_MAX)
590 2173
                buf = h2_errs[err];
591
        else
592 0
                buf = "unknown";
593 2173
        vtc_log(s->vl, 4, "rst->err: %s (%d)", buf, err);
594
595 2173
}
596
597
static void
598 15599
parse_settings(const struct stream *s, struct frame *f)
599
{
600
        struct http *hp;
601
        int v;
602
        unsigned u, t;
603
        const char *buf;
604
        enum hpk_result r;
605 15599
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
606 15599
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
607 15599
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
608
609 15599
        if (f->size % 6)
610 0
                vtc_fatal(s->vl,
611 0
                    "Size should be a multiple of 6, but isn't (%d)", f->size);
612
613 15599
        if (s->id != 0)
614 0
                vtc_fatal(s->vl,
615 0
                    "Setting frames should only be on stream 0, but received on stream: %d", s->id);
616
617 140390
        for (u = 0; u <= SETTINGS_MAX; u++)
618 124791
                f->md.settings[u] = NAN;
619
620 31759
        for (u = 0; u < f->size;) {
621 16160
                t = vbe16dec(f->data + u);
622 16160
                u += 2;
623 16160
                v = vbe32dec(f->data + u);
624 16160
                if (t <= SETTINGS_MAX) {
625 16160
                        buf = h2_settings[t];
626 16160
                        f->md.settings[t] = v;
627 16160
                } else
628 0
                        buf = "unknown";
629 16160
                u += 4;
630
631 16160
                if (t == 1) {
632 200
                        r = HPK_ResizeTbl(s->hp->encctx, v);
633 200
                        assert(r == hpk_done);
634 200
                }
635
636 16160
                vtc_log(s->vl, 4, "settings->%s (%u): %d", buf, t, v);
637
        }
638
639 15599
}
640
641
static void
642 240
parse_ping(const struct stream *s, struct frame *f)
643
{
644
        struct http *hp;
645 240
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
646 240
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
647 240
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
648 240
        if (f->size != 8)
649 0
                vtc_fatal(s->vl, "Size should be 8, but isn't (%d)", f->size);
650 240
        f->md.ping.ack = f->flags & ACK;
651 240
        memcpy(f->md.ping.data, f->data, 8);
652 240
        f->md.ping.data[8] = '\0';
653
654 240
        vtc_log(s->vl, 4, "ping->data: %s", f->md.ping.data);
655
656 240
}
657
658
static void
659 1400
parse_goaway(const struct stream *s, struct frame *f)
660
{
661
        struct http *hp;
662
        const char *err_buf;
663
        uint32_t err, stid;
664 1400
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
665 1400
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
666 1400
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
667
668 1400
        if (f->size < 8)
669 0
                vtc_fatal(s->vl,
670 0
                    "Size should be at least 8, but isn't (%d)", f->size);
671 1400
        if (f->data[0] & (1<<7))
672 0
                vtc_fatal(s->vl,
673
                    "First bit of data is reserved and should be 0");
674
675 1400
        stid = vbe32dec(f->data);
676 1400
        err = vbe32dec(f->data + 4);
677 1400
        f->md.goaway.err = err;
678 1400
        f->md.goaway.stream = stid;
679
680 1400
        if (err <= ERR_MAX)
681 1400
                err_buf = h2_errs[err];
682
        else
683 0
                err_buf = "unknown";
684
685 1400
        if (f->size > 8) {
686 80
                f->md.goaway.debug = malloc((f->size - 8) + 1L);
687 80
                AN(f->md.goaway.debug);
688 80
                f->md.goaway.debug[f->size - 8] = '\0';
689
690 80
                memcpy(f->md.goaway.debug, f->data + 8, f->size - 8);
691 80
        }
692
693 1400
        vtc_log(s->vl, 3, "goaway->laststream: %d", stid);
694 1400
        vtc_log(s->vl, 3, "goaway->err: %s (%d)", err_buf, err);
695 1400
        if (f->md.goaway.debug)
696 80
                vtc_log(s->vl, 3, "goaway->debug: %s", f->md.goaway.debug);
697 1400
}
698
699
static void
700 2640
parse_winup(const struct stream *s, struct frame *f)
701
{
702
        struct http *hp;
703
        uint32_t size;
704 2640
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
705 2640
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
706 2640
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
707
708 2640
        if (f->size != 4)
709 0
                vtc_fatal(s->vl, "Size should be 4, but isn't (%d)", f->size);
710 2640
        if (f->data[0] & (1<<7))
711 0
                vtc_log(s->vl, s->hp->fatal,
712
                    "First bit of data is reserved and should be 0");
713
714 2640
        size = vbe32dec(f->data);
715 2640
        f->md.winup_size = size;
716
717 2640
        vtc_log(s->vl, 3, "winup->size: %d", size);
718 2640
}
719
720
/* read a frame and queue it in the relevant stream, wait if not present yet.
721
 */
722
static void *
723 7920
receive_frame(void *priv)
724
{
725
        struct http *hp;
726
        char hdr[9];
727
        struct frame *f;
728
        struct stream *s;
729 7920
        int expect_cont = 0;
730 7920
        struct vsb *vsb = NULL;
731 7920
        struct hpk_hdr *hdrs = NULL;
732
733 7920
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
734
735 7920
        PTOK(pthread_mutex_lock(&hp->mtx));
736 81308
        while (hp->h2) {
737
                /*no wanted frames? */
738 73390
                assert(hp->wf >= 0);
739 73390
                if (hp->wf == 0) {
740 39335
                        PTOK(pthread_cond_wait(&hp->cond, &hp->mtx));
741 39335
                        continue;
742
                }
743 34055
                PTOK(pthread_mutex_unlock(&hp->mtx));
744
745 34055
                if (get_bytes(hp, hdr, sizeof hdr) <= 0) {
746 0
                        PTOK(pthread_mutex_lock(&hp->mtx));
747 0
                        VTAILQ_FOREACH(s, &hp->streams, list)
748 0
                                PTOK(pthread_cond_signal(&s->cond));
749 0
                        PTOK(pthread_mutex_unlock(&hp->mtx));
750 0
                        vtc_log(hp->vl, hp->fatal,
751
                            "could not get frame header");
752 0
                        return (NULL);
753
                }
754 34055
                ALLOC_OBJ(f, FRAME_MAGIC);
755 34055
                AN(f);
756 34055
                readFrameHeader(f, hdr);
757
758 68110
                vtc_log(hp->vl, 3, "rx: stream: %d, type: %s (%d), "
759
                                "flags: 0x%02x, size: %d",
760 34055
                                f->stid,
761 34055
                                f->type < TYPE_MAX ? h2_types[f->type] : "?",
762 34055
                                f->type, f->flags, f->size);
763 34055
                explain_flags(f->flags, f->type, hp->vl);
764
765 34055
                if (f->size) {
766 23053
                        f->data = malloc(f->size + 1L);
767 23053
                        AN(f->data);
768 23053
                        f->data[f->size] = '\0';
769 23053
                        if (get_bytes(hp, f->data, f->size) <= 0) {
770 0
                                PTOK(pthread_mutex_lock(&hp->mtx));
771 0
                                VTAILQ_FOREACH(s, &hp->streams, list)
772 0
                                        PTOK(pthread_cond_signal(&s->cond));
773 0
                                clean_frame(&f);
774 0
                                PTOK(pthread_mutex_unlock(&hp->mtx));
775 0
                                vtc_log(hp->vl, hp->fatal,
776
                                    "could not get frame body");
777 0
                                return (NULL);
778
                        }
779 23053
                }
780
781
                /* is the corresponding stream waiting? */
782 34055
                PTOK(pthread_mutex_lock(&hp->mtx));
783 34055
                s = NULL;
784 68107
                while (!s) {
785 39654
                        VTAILQ_FOREACH(s, &hp->streams, list)
786 39653
                                if (s->id == f->stid)
787 34053
                                        break;
788 34054
                        if (!s)
789 0
                                PTOK(pthread_cond_wait(&hp->cond, &hp->mtx));
790 34052
                        if (!hp->h2) {
791 0
                                clean_frame(&f);
792 0
                                PTOK(pthread_mutex_unlock(&hp->mtx));
793 0
                                return (NULL);
794
                        }
795
                }
796 34053
                PTOK(pthread_mutex_unlock(&hp->mtx));
797
798 34053
                AN(s);
799 34613
                if (expect_cont &&
800 560
                    (f->type != TYPE_CONTINUATION || expect_cont != s->id))
801 0
                        vtc_fatal(s->vl, "Expected CONTINUATION frame for "
802 0
                            "stream %u", expect_cont);
803
804
                /* parse the frame according to it type, and fill the metadata */
805 34053
                switch (f->type) {
806
                        case TYPE_DATA:
807 4000
                                parse_data(s, f);
808 4000
                                break;
809
                        case TYPE_PUSH_PROMISE:
810 40
                                hdrs = s->req;
811
                                /*FALLTHROUGH*/
812
                        case TYPE_HEADERS:
813 7080
                                if (!hdrs) {
814 7040
                                        if (hp->sfd)
815 1840
                                                hdrs = s->req;
816
                                        else
817 5200
                                                hdrs = s->resp;
818 7040
                                }
819 7080
                                clean_headers(hdrs);
820 7080
                                hdrs[0].t = hpk_unset;
821 7080
                                AZ(vsb);
822 7080
                                vsb = VSB_new_auto();
823
                                /*FALLTHROUGH*/
824
                        case TYPE_CONTINUATION:
825 7640
                                AN(hdrs);
826 7640
                                expect_cont = s->id;
827 7640
                                parse_hdr(s, f, vsb);
828 7640
                                if (f->flags & END_HEADERS) {
829 7080
                                        expect_cont = 0;
830 7080
                                        AZ(VSB_finish(vsb));
831 7080
                                        decode_hdr(hp, hdrs, vsb);
832 7080
                                        VSB_destroy(&vsb);
833 7080
                                        hdrs = NULL;
834 7080
                                }
835 7640
                                break;
836
                        case TYPE_PRIORITY:
837 360
                                parse_prio(s, f);
838 360
                                break;
839
                        case TYPE_RST_STREAM:
840 2173
                                parse_rst(s, f);
841 2173
                                break;
842
                        case TYPE_SETTINGS:
843 15600
                                parse_settings(s, f);
844 15600
                                break;
845
                        case TYPE_PING:
846 240
                                parse_ping(s, f);
847 240
                                break;
848
                        case TYPE_GOAWAY:
849 1400
                                parse_goaway(s, f);
850 1400
                                break;
851
                        case TYPE_WINDOW_UPDATE:
852 2640
                                parse_winup(s, f);
853 2640
                                break;
854
                        default:
855 0
                                WRONG("wrong frame type");
856 0
                }
857
858 34053
                PTOK(pthread_mutex_lock(&hp->mtx));
859 34053
                VTAILQ_INSERT_HEAD(&s->fq, f, list);
860 34053
                if (s->wf) {
861 32371
                        assert(hp->wf > 0);
862 32371
                        hp->wf--;
863 32371
                        s->wf = 0;
864 32371
                        PTOK(pthread_cond_signal(&s->cond));
865 32371
                }
866 34053
                continue;
867
        }
868 7918
        PTOK(pthread_mutex_unlock(&hp->mtx));
869 7918
        if (vsb != NULL)
870 0
                VSB_destroy(&vsb);
871 7918
        return (NULL);
872 7918
}
873
874
#define STRTOU32(n, ss, p, v, c)                                        \
875
        do {                                                            \
876
                n = strtoul(ss, &p, 0);                                 \
877
                if (*p != '\0')                                         \
878
                        vtc_fatal(v, "%s takes an integer as argument " \
879
                                "(found %s)", c, ss);                   \
880
        } while (0)
881
882
#define STRTOU32_CHECK(n, sp, p, v, c, l)                               \
883
do {                                                                    \
884
        sp++;                                                           \
885
        AN(*sp);                                                        \
886
        STRTOU32(n, *sp, p, v, c);                                      \
887
        if (l && n >= (1U << l))                                        \
888
                vtc_fatal(v,                                            \
889
                    c " must be a %d-bits integer (found %s)", l, *sp); \
890
} while (0)
891
892
#define CHECK_LAST_FRAME(TYPE) \
893
        if (!f || f->type != TYPE_ ## TYPE) {                              \
894
                vtc_fatal(s->vl, "Last frame was not of type " #TYPE); \
895
        }
896
897
#define RETURN_SETTINGS(idx) \
898
do { \
899
        if (isnan(f->md.settings[idx])) { \
900
                return (NULL); \
901
        } \
902
        snprintf(buf, 20, "%.0f", f->md.settings[idx]); \
903
        return (buf); \
904
} while (0)
905
906
#define RETURN_BUFFED(val) \
907
do { \
908
        snprintf(buf, 20, "%ld", (long)val); \
909
        return (buf); \
910
} while (0)
911
912
static char *
913 5440
find_header(const struct hpk_hdr *h, const char *k)
914
{
915 5440
        AN(k);
916
917 5440
        int kl = strlen(k);
918 15360
        while (h->t) {
919 14960
                if (kl == h->key.len  && !strncasecmp(h->key.ptr, k, kl))
920 5040
                        return (h->value.ptr);
921 9920
                h++;
922
        }
923 400
        return (NULL);
924 5440
}
925
/* SECTION: stream.spec.zexpect expect
926
 *
927
 * expect in stream works as it does in client or server, except that the
928
 * elements compared will be different.
929
 *
930
 * Most of these elements will be frame specific, meaning that the last frame
931
 * received on that stream must of the correct type.
932
 *
933
 * Here the list of keywords you can look at.
934
 */
935
static const char *
936 54272
cmd_var_resolve(const struct stream *s, const char *spec, char *buf)
937
{
938
        uint32_t idx;
939
        int n;
940
        const struct hpk_hdr *h;
941
        struct hpk_ctx *ctx;
942 54272
        struct frame *f = s->frame;
943
944 54272
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
945 54272
        CHECK_OBJ_NOTNULL(s->hp, HTTP_MAGIC);
946 54272
        AN(spec);
947 54272
        AN(buf);
948
949 54272
        n = 0;
950
        /* SECTION: stream.spec.zexpect.ping PING specific
951
         *
952
         * ping.data
953
         *      The 8-bytes string of the PING frame payload.
954
         * ping.ack (PING)
955
         *      "true" if the ACK flag was set, "false" otherwise.
956
         */
957 54272
        if (!strcmp(spec, "ping.data")) {
958 240
                CHECK_LAST_FRAME(PING);
959 240
                return (f->md.ping.data);
960
        }
961 54032
        if (!strcmp(spec, "ping.ack")) {
962 160
                CHECK_LAST_FRAME(PING);
963 160
                snprintf(buf, 20, (f->flags & ACK) ? "true" : "false");
964 160
                return (buf);
965
        }
966
        /* SECTION: stream.spec.zexpect.winup WINDOW_UPDATE specific
967
         *
968
         * winup.size
969
         *      The size of the upgrade given by the WINDOW_UPDATE frame.
970
         */
971 53872
        if (!strcmp(spec, "winup.size")) {
972 40
                CHECK_LAST_FRAME(WINDOW_UPDATE);
973 40
                RETURN_BUFFED(f->md.winup_size);
974 0
        }
975
        /* SECTION: stream.spec.zexpect.prio PRIORITY specific
976
         *
977
         * prio.stream
978
         *      The stream ID announced.
979
         *
980
         * prio.exclusive
981
         *      "true" if the priority is exclusive, else "false".
982
         *
983
         * prio.weight
984
         *      The dependency weight.
985
         */
986 53832
        if (!strcmp(spec, "prio.stream")) {
987 80
                CHECK_LAST_FRAME(PRIORITY);
988 80
                RETURN_BUFFED(f->md.prio.stream);
989 0
        }
990 53752
        if (!strcmp(spec, "prio.exclusive")) {
991 0
                CHECK_LAST_FRAME(PRIORITY);
992 0
                snprintf(buf, 20, f->md.prio.exclusive ? "true" : "false");
993 0
                return (buf);
994
        }
995 53752
        if (!strcmp(spec, "prio.weight")) {
996 80
                CHECK_LAST_FRAME(PRIORITY);
997 80
                RETURN_BUFFED(f->md.prio.weight);
998 0
        }
999
        /* SECTION: stream.spec.zexpect.rst RESET_STREAM specific
1000
         *
1001
         * rst.err
1002
         *      The error code (as integer) of the RESET_STREAM frame.
1003
         */
1004 53672
        if (!strcmp(spec, "rst.err")) {
1005 1640
                CHECK_LAST_FRAME(RST_STREAM);
1006 1640
                RETURN_BUFFED(f->md.rst_err);
1007 0
        }
1008
        /* SECTION: stream.spec.zexpect.settings SETTINGS specific
1009
         *
1010
         * settings.ack
1011
         *      "true" if the ACK flag was set, else "false".
1012
         *
1013
         * settings.push
1014
         *      "true" if the push settings was set to yes, "false" if set to
1015
         *      no, and <undef> if not present.
1016
         *
1017
         * settings.hdrtbl
1018
         *      Value of HEADER_TABLE_SIZE if set, <undef> otherwise.
1019
         *
1020
         * settings.maxstreams
1021
         *      Value of MAX_CONCURRENT_STREAMS if set, <undef> otherwise.
1022
         *
1023
         * settings.winsize
1024
         *      Value of INITIAL_WINDOW_SIZE if set, <undef> otherwise.
1025
         *
1026
         * setting.framesize
1027
         *      Value of MAX_FRAME_SIZE if set, <undef> otherwise.
1028
         *
1029
         * settings.hdrsize
1030
         *      Value of MAX_HEADER_LIST_SIZE if set, <undef> otherwise.
1031
         */
1032 52032
        if (!strncmp(spec, "settings.", 9)) {
1033 8376
                CHECK_LAST_FRAME(SETTINGS);
1034 8376
                spec += 9;
1035 8376
                if (!strcmp(spec, "ack")) {
1036 7736
                        snprintf(buf, 20, (f->flags & ACK) ? "true" : "false");
1037 7736
                        return (buf);
1038
                }
1039 640
                if (!strcmp(spec, "push")) {
1040 40
                        if (isnan(f->md.settings[SETTINGS_ENABLE_PUSH]))
1041 40
                                return (NULL);
1042 0
                        else if (f->md.settings[SETTINGS_ENABLE_PUSH] == 1)
1043 0
                                snprintf(buf, 20, "true");
1044
                        else
1045 0
                                snprintf(buf, 20, "false");
1046 0
                        return (buf);
1047
                }
1048 600
                if (!strcmp(spec, "hdrtbl"))     { RETURN_SETTINGS(1); }
1049 480
                if (!strcmp(spec, "maxstreams")) { RETURN_SETTINGS(3); }
1050 360
                if (!strcmp(spec, "winsize"))    { RETURN_SETTINGS(4); }
1051 240
                if (!strcmp(spec, "framesize"))  { RETURN_SETTINGS(5); }
1052 120
                if (!strcmp(spec, "hdrsize"))    { RETURN_SETTINGS(6); }
1053 0
        }
1054
        /* SECTION: stream.spec.zexpect.push PUSH_PROMISE specific
1055
         *
1056
         * push.id
1057
         *      The id of the promised stream.
1058
         */
1059 43656
        if (!strcmp(spec, "push.id")) {
1060 40
                CHECK_LAST_FRAME(PUSH_PROMISE);
1061 40
                RETURN_BUFFED(f->md.promised);
1062 0
        }
1063
        /* SECTION: stream.spec.zexpect.goaway GOAWAY specific
1064
         *
1065
         * goaway.err
1066
         *      The error code (as integer) of the GOAWAY frame.
1067
         *
1068
         * goaway.laststream
1069
         *      Last-Stream-ID
1070
         *
1071
         * goaway.debug
1072
         *      Debug data, if any.
1073
         */
1074 43616
        if (!strncmp(spec, "goaway.", 7)) {
1075 2320
                spec += 7;
1076 2320
                CHECK_LAST_FRAME(GOAWAY);
1077
1078 2320
                if (!strcmp(spec, "err"))
1079 1240
                        RETURN_BUFFED(f->md.goaway.err);
1080 1080
                else if (!strcmp(spec, "laststream"))
1081 1000
                        RETURN_BUFFED(f->md.goaway.stream);
1082 80
                else if (!strcmp(spec, "debug"))
1083 80
                        return (f->md.goaway.debug);
1084 0
        }
1085
        /* SECTION: stream.spec.zexpect.zframe Generic frame
1086
         *
1087
         * frame.data
1088
         *      Payload of the last frame
1089
         *
1090
         * frame.type
1091
         *      Type of the frame, as integer.
1092
         *
1093
         * frame.size
1094
         *      Size of the frame.
1095
         *
1096
         * frame.stream
1097
         *      Stream of the frame (correspond to the one you are executing
1098
         *      this from, obviously).
1099
         *
1100
         * frame.padding (for DATA, HEADERS, PUSH_PROMISE frames)
1101
         *      Number of padded bytes.
1102
         */
1103 41296
        if (!strncmp(spec, "frame.", 6)) {
1104 320
                spec += 6;
1105 320
                if (!f)
1106 0
                        vtc_fatal(s->vl, "No frame received yet.");
1107 320
                if (!strcmp(spec, "data"))   { return (f->data); }
1108 280
                else if (!strcmp(spec, "type"))   { RETURN_BUFFED(f->type); }
1109 240
                else if (!strcmp(spec, "size"))   { RETURN_BUFFED(f->size); }
1110 80
                else if (!strcmp(spec, "stream")) { RETURN_BUFFED(f->stid); }
1111 40
                else if (!strcmp(spec, "padding")) {
1112 40
                        if (f->type != TYPE_DATA &&
1113 40
                                        f->type != TYPE_HEADERS &&
1114 0
                                        f->type != TYPE_PUSH_PROMISE)
1115 0
                                vtc_fatal(s->vl,
1116
                                                "Last frame was not of type "
1117
                                                "DATA, HEADERS or PUSH");
1118 40
                        RETURN_BUFFED(f->md.padded);
1119 0
                }
1120 0
        }
1121
        /* SECTION: stream.spec.zexpect.zstream Stream
1122
         *
1123
         * stream.window
1124
         *      The current local window size of the stream, or, if on stream 0,
1125
         *      of the connection.
1126
         *
1127
         * stream.peer_window
1128
         *      The current peer window size of the stream, or, if on stream 0,
1129
         *      of the connection.
1130
         *
1131
         * stream.weight
1132
         *      Weight of the stream
1133
         *
1134
         * stream.dependency
1135
         *      Id of the stream this one depends on.
1136
         */
1137 40976
        if (!strcmp(spec, "stream.window")) {
1138 1040
                snprintf(buf, 20, "%jd",
1139 520
                    (intmax_t)(s->id ? s->win_self : s->hp->h2_win_self->size));
1140 520
                return (buf);
1141
        }
1142 40456
        if (!strcmp(spec, "stream.peer_window")) {
1143 1280
                snprintf(buf, 20, "%jd",
1144 640
                    (intmax_t)(s->id ? s->win_peer : s->hp->h2_win_peer->size));
1145 640
                return (buf);
1146
        }
1147 39816
        if (!strcmp(spec, "stream.weight")) {
1148 320
                if (s->id) {
1149 280
                        snprintf(buf, 20, "%d", s->weight);
1150 280
                        return (buf);
1151
                } else
1152 40
                        return (NULL);
1153
        }
1154 39496
        if (!strcmp(spec, "stream.dependency")) {
1155 640
                if (s->id) {
1156 600
                        snprintf(buf, 20, "%d", s->dependency);
1157 600
                        return (buf);
1158
                } else
1159 40
                        return (NULL);
1160
        }
1161
        /* SECTION: stream.spec.zexpect.ztable Index tables
1162
         *
1163
         * tbl.dec.size / tbl.enc.size
1164
         *      Size (bytes) of the decoding/encoding table.
1165
         *
1166
         * tbl.dec.size / tbl.enc.maxsize
1167
         *      Maximum size (bytes) of the decoding/encoding table.
1168
         *
1169
         * tbl.dec.length / tbl.enc.length
1170
         *      Number of headers in decoding/encoding table.
1171
         *
1172
         * tbl.dec[INT].key / tbl.enc[INT].key
1173
         *      Name of the header at index INT of the decoding/encoding
1174
         *      table.
1175
         *
1176
         * tbl.dec[INT].value / tbl.enc[INT].value
1177
         *      Value of the header at index INT of the decoding/encoding
1178
         *      table.
1179
         */
1180 38856
        if (!strncmp(spec, "tbl.dec", 7) || !strncmp(spec, "tbl.enc", 7)) {
1181 4920
                if (spec[4] == 'd')
1182 2520
                        ctx = s->hp->decctx;
1183
                else
1184 2400
                        ctx = s->hp->encctx;
1185 4920
                spec += 7;
1186
1187 4920
                if (1 == sscanf(spec, "[%u].key%n", &idx, &n) &&
1188 3840
                                spec[n] == '\0') {
1189 1920
                        h = HPK_GetHdr(ctx, idx + 61);
1190 1920
                        return (h ? h->key.ptr : NULL);
1191
                }
1192 3000
                else if (1 == sscanf(spec, "[%u].value%n", &idx, &n) &&
1193 1920
                                spec[n] == '\0') {
1194 1920
                        h = HPK_GetHdr(ctx, idx + 61);
1195 1920
                        return (h ? h->value.ptr : NULL);
1196
                }
1197 1080
                else if (!strcmp(spec, ".size"))
1198 1040
                        RETURN_BUFFED(HPK_GetTblSize(ctx));
1199 40
                else if (!strcmp(spec, ".maxsize"))
1200 0
                        RETURN_BUFFED(HPK_GetTblMaxSize(ctx));
1201 40
                else if (!strcmp(spec, ".length"))
1202 40
                        RETURN_BUFFED(HPK_GetTblLength(ctx));
1203 0
        }
1204
        /* SECTION: stream.spec.zexpect.zre Request and response
1205
         *
1206
         * Note: it's possible to inspect a request or response while it is
1207
         * still being construct (in-between two frames for example).
1208
         *
1209
         * req.bodylen / resp.bodylen
1210
         *      Length in bytes of the request/response so far.
1211
         *
1212
         * req.body / resp.body
1213
         *      Body of the request/response so far.
1214
         *
1215
         * req.http.STRING / resp.http.STRING
1216
         *      Value of the header STRING in the request/response.
1217
         *
1218
         * req.status / resp.status
1219
         *      :status pseudo-header's value.
1220
         *
1221
         * req.url / resp.url
1222
         *      :path pseudo-header's value.
1223
         *
1224
         * req.method / resp.method
1225
         *      :method pseudo-header's value.
1226
         *
1227
         * req.authority / resp.authority
1228
         *      :method pseudo-header's value.
1229
         *
1230
         * req.scheme / resp.scheme
1231
         *      :method pseudo-header's value.
1232
         */
1233 33936
        if (!strncmp(spec, "req.", 4) || !strncmp(spec, "resp.", 5)) {
1234 6880
                if (spec[2] == 'q') {
1235 600
                        h = s->req;
1236 600
                        spec += 4;
1237 600
                } else {
1238 6280
                        h = s->resp;
1239 6280
                        spec += 5;
1240
                }
1241 6880
                if (!strcmp(spec, "body"))
1242 680
                        return (s->body);
1243 6200
                else if (!strcmp(spec, "bodylen"))
1244 760
                        RETURN_BUFFED(s->bodylen);
1245 5440
                else if (!strcmp(spec, "status"))
1246 2520
                        return (find_header(h, ":status"));
1247 2920
                else if (!strcmp(spec, "url"))
1248 80
                        return (find_header(h, ":path"));
1249 2840
                else if (!strcmp(spec, "method"))
1250 80
                        return (find_header(h, ":method"));
1251 2760
                else if (!strcmp(spec, "authority"))
1252 40
                        return (find_header(h, ":authority"));
1253 2720
                else if (!strcmp(spec, "scheme"))
1254 40
                        return (find_header(h, ":scheme"));
1255 2680
                else if (!strncmp(spec, "http.", 5))
1256 2680
                        return (find_header(h, spec + 5));
1257
                else
1258 0
                        return (NULL);
1259 0
        }
1260
#define H2_ERROR(U,v,sc,g,r,t) \
1261
        if (!strcmp(spec, #U)) { return (#v); }
1262
#include "tbl/h2_error.h"
1263
        return (spec);
1264
}
1265
1266
/* SECTION: stream.spec.frame_sendhex sendhex
1267
 *
1268
 * Push bytes directly on the wire. sendhex takes exactly one argument: a string
1269
 * describing the bytes, in hex notation, with possible whitespaces between
1270
 * them. Here's an example::
1271
 *
1272
 *      sendhex "00 00 08 00 0900       8d"
1273
 */
1274
static void
1275 2240
cmd_sendhex(CMD_ARGS)
1276
{
1277
        struct http *hp;
1278
        struct stream *s;
1279
        struct vsb *vsb;
1280
1281 2240
        (void)vl;
1282 2240
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1283 2240
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);
1284 2240
        AN(av[1]);
1285 2240
        AZ(av[2]);
1286 2240
        vsb = vtc_hex_to_bin(hp->vl, av[1]);
1287 2240
        assert(VSB_len(vsb) >= 0);
1288 2240
        vtc_hexdump(hp->vl, 4, "sendhex", VSB_data(vsb), VSB_len(vsb));
1289 2240
        PTOK(pthread_mutex_lock(&hp->mtx));
1290 2240
        http_write(hp, 4, VSB_data(vsb), VSB_len(vsb), "sendhex");
1291 2240
        PTOK(pthread_mutex_unlock(&hp->mtx));
1292 2240
        VSB_destroy(&vsb);
1293 2240
}
1294
1295
#define ENC(hdr, k, v)                                  \
1296
{                                                       \
1297
        AN(k);                                          \
1298
        hdr.key.ptr = TRUST_ME(k);                      \
1299
        hdr.key.len = strlen(k);                        \
1300
        AN(v);                                          \
1301
        hdr.value.ptr = TRUST_ME(v);                    \
1302
        hdr.value.len = strlen(v);                      \
1303
        assert(HPK_EncHdr(iter, &hdr) != hpk_err);      \
1304
}
1305
1306
#define STR_ENC(av, field, str)                                                \
1307
{                                                                              \
1308
        av++;                                                                  \
1309
             if (AV_IS("plain")) { hdr.field.huff = 0; }                       \
1310
        else if (AV_IS("huf"))   { hdr.field.huff = 1; }                       \
1311
        else                                                                   \
1312
                vtc_fatal(vl, str " arg can be huf or plain (got: %s)", *av); \
1313
        av++;                                                                  \
1314
        AN(*av);                                                               \
1315
        hdr.field.ptr = *av;                                                   \
1316
        hdr.field.len = strlen(*av);                                           \
1317
}
1318
1319
1320
/* SECTION: stream.spec.data_0 txreq, txresp, txcont, txpush
1321
 *
1322
 * These four commands are about sending headers. txreq and txresp
1323
 * will send HEADER frames; txcont will send CONTINUATION frames; txpush
1324
 * PUSH frames.
1325
 * The only difference between txreq and txresp are the default headers
1326
 * set by each of them.
1327
 *
1328
 * \-noadd
1329
 *      Do not add default headers. Useful to avoid duplicates when sending
1330
 *      default headers using ``-hdr``, ``-idxHdr`` and ``-litIdxHdr``.
1331
 *
1332
 * \-status INT (txresp)
1333
 *      Set the :status pseudo-header.
1334
 *
1335
 * \-url STRING (txreq, txpush)
1336
 *      Set the :path pseudo-header.
1337
 *
1338
 * \-method STRING (txreq, txpush)
1339
 *      Set the :method pseudo-header.
1340
 *
1341
 * \-req STRING (txreq, txpush)
1342
 *      Alias for -method.
1343
 *
1344
 * \-scheme STRING (txreq, txpush)
1345
 *      Set the :scheme pseudo-header.
1346
 *
1347
 * \-hdr STRING1 STRING2
1348
 *      Insert a header, STRING1 being the name, and STRING2 the value.
1349
 *
1350
 * \-idxHdr INT
1351
 *      Insert an indexed header, using INT as index.
1352
 *
1353
 * \-litIdxHdr inc|not|never INT huf|plain STRING
1354
 *      Insert an literal, indexed header. The first argument specify if the
1355
 *      header should be added to the table, shouldn't, or mustn't be
1356
 *      compressed if/when retransmitted.
1357
 *
1358
 *      INT is the index of the header name to use.
1359
 *
1360
 *      The third argument informs about the Huffman encoding: yes (huf) or
1361
 *      no (plain).
1362
 *
1363
 *      The last term is the literal value of the header.
1364
 *
1365
 * \-litHdr inc|not|never huf|plain STRING1 huf|plain STRING2
1366
 *      Insert a literal header, with the same first argument as
1367
 *      ``-litIdxHdr``.
1368
 *
1369
 *      The second and third terms tell what the name of the header is and if
1370
 *      it should be Huffman-encoded, while the last two do the same
1371
 *      regarding the value.
1372
 *
1373
 * \-body STRING (txreq, txresp)
1374
 *      Specify a body, effectively putting STRING into a DATA frame after
1375
 *      the HEADER frame is sent.
1376
 *
1377
 * \-bodyfrom FILE (txreq, txresp)
1378
 *      Same as ``-body`` but content is read from FILE.
1379
 *
1380
 * \-bodylen INT (txreq, txresp)
1381
 *      Do the same thing as ``-body`` but generate a string of INT length
1382
 *      for you.
1383
 *
1384
 * \-gzipbody STRING (txreq, txresp)
1385
 *      Gzip STRING and send it as body.
1386
 *
1387
 * \-gziplen NUMBER (txreq, txresp)
1388
 *      Combine -bodylen and -gzipbody: generate a string of length NUMBER,
1389
 *      gzip it and send as body.
1390
 *
1391
 * \-nostrend (txreq, txresp)
1392
 *      Don't set the END_STREAM flag automatically, making the peer expect
1393
 *      a body after the headers.
1394
 *
1395
 * \-nohdrend
1396
 *      Don't set the END_HEADERS flag automatically, making the peer expect
1397
 *      more HEADER frames.
1398
 *
1399
 * \-dep INT (txreq, txresp)
1400
 *      Tell the peer that this content depends on the stream with the INT
1401
 *      id.
1402
 *
1403
 * \-ex (txreq, txresp)
1404
 *      Make the dependency exclusive (``-dep`` is still needed).
1405
 *
1406
 * \-weight (txreq, txresp)
1407
 *      Set the weight for the dependency.
1408
 *
1409
 * \-promised INT (txpush)
1410
 *      The id of the promised stream.
1411
 *
1412
 * \-pad STRING / -padlen INT (txreq, txresp, txpush)
1413
 *      Add string as padding to the frame, either the one you provided with
1414
 *      \-pad, or one that is generated for you, of length INT is -padlen
1415
 *      case.
1416
 */
1417
1418
#define cmd_txreq       cmd_tx11obj
1419
#define cmd_txresp      cmd_tx11obj
1420
#define cmd_txpush      cmd_tx11obj
1421
#define cmd_txcont      cmd_tx11obj
1422
1423
static void
1424 50760
cmd_tx11obj(CMD_ARGS)
1425
{
1426
        struct stream *s;
1427
        int i;
1428 50760
        int status_done = 1;
1429 50760
        int method_done = 1;
1430 50760
        int path_done = 1;
1431 50760
        int scheme_done = 1;
1432 50760
        long bodylen = 0;
1433
        ssize_t len;
1434 50760
        uint32_t stid = 0, pstid;
1435 50760
        uint32_t weight = 16;
1436 50760
        uint32_t exclusive = 0;
1437
        char *buf;
1438
        struct hpk_iter *iter;
1439
        struct frame f;
1440 50760
        char *body = NULL, *pad = NULL;
1441
        /*XXX: do we need a better api? yes we do */
1442
        struct hpk_hdr hdr;
1443 50760
        char *cmd_str = *av;
1444
        char *p;
1445
1446 50760
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1447 50760
        INIT_FRAME(f, CONTINUATION, 0, s->id, END_HEADERS);
1448 50760
        buf = malloc(BUF_SIZE);
1449 50760
        AN(buf);
1450
1451 50760
        if (!strcmp(cmd_str, "txreq")) {
1452 8160
                ONLY_H2_CLIENT(s->hp, av);
1453 8160
                f.type = TYPE_HEADERS;
1454 8160
                f.flags |= END_STREAM;
1455 8160
                method_done = 0;
1456 8160
                path_done = 0;
1457 8160
                scheme_done = 0;
1458 50760
        } else if (!strcmp(cmd_str, "txresp")) {
1459 1880
                ONLY_H2_SERVER(s->hp, av);
1460 1880
                f.type = TYPE_HEADERS;
1461 1880
                f.flags |= END_STREAM;
1462 1880
                status_done = 0;
1463 42600
        } else if (!strcmp(cmd_str, "txpush")) {
1464 40
                ONLY_H2_SERVER(s->hp, av);
1465 40
                f.type = TYPE_PUSH_PROMISE;
1466 40
                method_done = 0;
1467 40
                path_done = 0;
1468 40
                scheme_done = 0;
1469 40
        }
1470
1471 50760
        if (f.type == TYPE_PUSH_PROMISE) {
1472 40
                *buf = 0;
1473 40
                iter = HPK_NewIter(s->hp->encctx, buf + 4, BUF_SIZE - 4);
1474 40
        } else
1475 50720
                iter = HPK_NewIter(s->hp->encctx, buf, BUF_SIZE);
1476
1477
#define AV_IS(str) !strcmp(*av, str)
1478
#define CMD_IS(str) !strcmp(cmd_str, str)
1479 148480
        while (*++av) {
1480 97720
                memset(&hdr, 0, sizeof(hdr));
1481 97720
                hdr.t = hpk_not;
1482 97720
                if (AV_IS("-noadd")) {
1483 480
                        path_done = 1;
1484 480
                        status_done = 1;
1485 480
                        method_done = 1;
1486 480
                        scheme_done = 1;
1487 480
                }
1488 97240
                else if (AV_IS("-status") && CMD_IS("txresp")) {
1489 40
                        ENC(hdr, ":status", av[1]);
1490 40
                        av++;
1491 40
                        status_done = 1;
1492 40
                }
1493 97240
                else if (AV_IS("-url") &&
1494 2200
                                (CMD_IS("txreq") || CMD_IS("txpush"))) {
1495 2200
                        ENC(hdr, ":path", av[1]);
1496 2200
                        av++;
1497 2200
                        path_done = 1;
1498 2200
                }
1499 95000
                else if ((AV_IS("-method") || AV_IS("-req")) &&
1500 1480
                                (CMD_IS("txreq") || CMD_IS("txpush"))) {
1501 1480
                        ENC(hdr, ":method", av[1]);
1502 1480
                        av++;
1503 1480
                        method_done = 1;
1504 1480
                }
1505 93520
                else if (AV_IS("-scheme") &&
1506 280
                                (CMD_IS("txreq") || CMD_IS("txpush"))) {
1507 280
                        ENC(hdr, ":scheme", av[1]);
1508 280
                        av++;
1509 280
                        scheme_done = 1;
1510 280
                }
1511 93240
                else if (AV_IS("-hdr")) {
1512 47360
                        if (av[2] == NULL)
1513 0
                                vtc_fatal(vl, "-hdr takes two arguments in http2");
1514 47360
                        ENC(hdr, av[1], av[2]);
1515 47360
                        av += 2;
1516 47360
                }
1517 45880
                else if (AV_IS("-idxHdr")) {
1518 1680
                        hdr.t = hpk_idx;
1519 1680
                        STRTOU32_CHECK(hdr.i, av, p, vl, "-idxHdr", 0);
1520 1680
                        assert(HPK_EncHdr(iter, &hdr) != hpk_err);
1521 1680
                }
1522 44200
                else if (AV_IS("-litIdxHdr")) {
1523 640
                        av++;
1524 640
                        if      (AV_IS("inc"))   { hdr.t = hpk_inc;   }
1525 40
                        else if (AV_IS("not"))   { hdr.t = hpk_not;   }
1526 40
                        else if (AV_IS("never")) { hdr.t = hpk_never; }
1527
                        else
1528 0
                                vtc_fatal(vl, "first -litidxHdr arg can be "
1529 0
                                    "inc, not, never (got: %s)", *av);
1530
1531 640
                        STRTOU32_CHECK(hdr.i, av, p, vl,
1532
                            "second -litidxHdr arg", 0);
1533
1534 640
                        hdr.key.ptr = NULL;
1535 640
                        hdr.key.len = 0;
1536 960
                        STR_ENC(av, value,   "third -litHdr");
1537 640
                        assert(HPK_EncHdr(iter, &hdr) != hpk_err);
1538 640
                }
1539 43560
                else if (AV_IS("-litHdr")) {
1540 280
                        av++;
1541 280
                        if      (AV_IS("inc"))   { hdr.t = hpk_inc;   }
1542 0
                        else if (AV_IS("not"))   { hdr.t = hpk_not;   }
1543 0
                        else if (AV_IS("never")) { hdr.t = hpk_never; }
1544
                        else
1545 0
                                vtc_fatal(vl, "first -litHdr arg can be inc, "
1546 0
                                    "not, never (got: %s)", *av);
1547
1548 400
                        STR_ENC(av, key,   "second -litHdr");
1549 400
                        STR_ENC(av, value, "fourth -litHdr");
1550 280
                        assert(HPK_EncHdr(iter, &hdr) != hpk_err);
1551 280
                }
1552 43280
                else if (AV_IS("-nostrend")) {
1553 1520
                        f.flags &= ~END_STREAM;
1554 1520
                }
1555 41760
                else if (AV_IS("-nohdrend")) {
1556 40600
                        f.flags &= ~END_HEADERS;
1557 40600
                }
1558 1160
                else if (AV_IS("-promised") && CMD_IS("txpush")) {
1559 40
                        STRTOU32_CHECK(pstid, av, p, vl, "-promised", 31);
1560 40
                        vbe32enc(buf, pstid);
1561 40
                }
1562 1120
                else if (AV_IS("-pad") && !CMD_IS("txcont")) {
1563 240
                        AZ(pad);
1564 240
                        av++;
1565 240
                        AN(*av);
1566 240
                        pad = strdup(*av);
1567 240
                }
1568 880
                else if (AV_IS("-padlen") && !CMD_IS("txcont")) {
1569 0
                        AZ(pad);
1570 0
                        av++;
1571 0
                        pad = synth_body(*av, 0);
1572 0
                }
1573 880
                else if (CMD_IS("txreq") || CMD_IS("txresp")) {
1574 880
                        if (AV_IS("-body")) {
1575 320
                                AZ(body);
1576 320
                                REPLACE(body, av[1]);
1577 320
                                AN(body);
1578 320
                                bodylen = strlen(body);
1579 320
                                f.flags &= ~END_STREAM;
1580 320
                                av++;
1581 320
                        }
1582 560
                        else if (AV_IS("-bodyfrom")) {
1583 80
                                AZ(body);
1584 80
                                body = VFIL_readfile(NULL, av[1], &len);
1585 80
                                AN(body);
1586 80
                                assert(len < INT_MAX);
1587 80
                                bodylen = len;
1588 80
                                f.flags &= ~END_STREAM;
1589 80
                                av++;
1590 80
                        }
1591 480
                        else if (AV_IS("-bodylen")) {
1592 160
                                AZ(body);
1593 160
                                body = synth_body(av[1], 0);
1594 160
                                bodylen = strlen(body);
1595 160
                                f.flags &= ~END_STREAM;
1596 160
                                av++;
1597 160
                        }
1598 320
                        else if (!strncmp(*av, "-gzip", 5)) {
1599 80
                                i = vtc_gzip_cmd(s->hp, av, &body, &bodylen);
1600 80
                                if (i == 0)
1601 0
                                        break;
1602 80
                                av += i;
1603 80
                                if (i > 1) {
1604 80
                                        ENC(hdr, ":content-encoding", "gzip");
1605 80
                                        f.flags &= ~END_STREAM;
1606 80
                                }
1607 80
                        }
1608 240
                        else if (AV_IS("-dep")) {
1609 120
                                STRTOU32_CHECK(stid, av, p, vl, "-dep", 0);
1610 120
                                f.flags |= PRIORITY;
1611 120
                        }
1612 120
                        else if (AV_IS("-ex")) {
1613 80
                                exclusive = 1U << 31;
1614 80
                                f.flags |= PRIORITY;
1615 80
                        }
1616 40
                        else if (AV_IS("-weight")) {
1617 40
                                STRTOU32_CHECK(weight, av, p, vl, "-weight", 8);
1618 40
                                f.flags |= PRIORITY;
1619 40
                        } else
1620 0
                                break;
1621 880
                } else
1622 0
                        break;
1623
        }
1624
#undef CMD_IS
1625
#undef AV_IS
1626 50760
        if (*av != NULL)
1627 0
                vtc_fatal(vl, "Unknown %s spec: %s\n", cmd_str, *av);
1628
1629 50760
        memset(&hdr, 0, sizeof(hdr));
1630 50760
        hdr.t = hpk_not;
1631
1632 50760
        if (!status_done) { ENC(hdr, ":status", "200"); }
1633 50760
        if (!path_done)   { ENC(hdr, ":path",   "/"); }
1634 50760
        if (!method_done) { ENC(hdr, ":method", "GET"); }
1635 50760
        if (!scheme_done) { ENC(hdr, ":scheme", "http"); }
1636
1637 50760
        f.size = gethpk_iterLen(iter);
1638 50760
        if (f.flags & PRIORITY) {
1639 160
                s->weight = weight & 0xff;
1640 160
                s->dependency = stid;
1641
1642 160
                assert(f.size + 5 < BUF_SIZE);
1643 160
                memmove(buf + 5, buf, f.size);
1644 160
                vbe32enc(buf, (stid | exclusive));
1645 160
                buf[4] = s->weight;
1646 160
                f.size += 5;
1647
1648 160
                vtc_log(vl, 4, "stream->dependency: %u", s->dependency);
1649 160
                vtc_log(vl, 4, "stream->weight: %u", s->weight);
1650 160
                if (exclusive)
1651 80
                        exclusive_stream_dependency(s);
1652 160
        }
1653 50760
        if (pad) {
1654 240
                if (strlen(pad) > 255)
1655 0
                        vtc_fatal(vl, "Padding is limited to 255 bytes");
1656 240
                f.flags |= PADDED;
1657 240
                assert(f.size + strlen(pad) < BUF_SIZE);
1658 240
                memmove(buf + 1, buf, f.size);
1659 240
                buf[0] = strlen(pad);
1660 240
                f.size += 1;
1661 240
                memcpy(buf + f.size, pad, strlen(pad));
1662 240
                f.size += strlen(pad);
1663 240
                free(pad);
1664 240
        }
1665 50760
        if (f.type == TYPE_PUSH_PROMISE)
1666 40
                f.size += 4;
1667 50760
        f.data = buf;
1668 50760
        HPK_FreeIter(iter);
1669 50760
        write_frame(s, &f, 1);
1670 50760
        free(buf);
1671
1672 50760
        if (!body)
1673 50120
                return;
1674
1675 640
        INIT_FRAME(f, DATA, bodylen, s->id, END_STREAM);
1676 640
        f.data = body;
1677
1678 640
        write_frame(s, &f, 1);
1679 640
        free(body);
1680 50760
}
1681
1682
/* SECTION: stream.spec.data_1 txdata
1683
 *
1684
 * By default, data frames are empty. The receiving end will know the whole body
1685
 * has been delivered thanks to the END_STREAM flag set in the last DATA frame,
1686
 * and txdata automatically set it.
1687
 *
1688
 * \-data STRING
1689
 *      Data to be embedded into the frame.
1690
 *
1691
 * \-datalen INT
1692
 *      Generate and INT-bytes long string to be sent in the frame.
1693
 *
1694
 * \-pad STRING / -padlen INT
1695
 *      Add string as padding to the frame, either the one you provided with
1696
 *      \-pad, or one that is generated for you, of length INT is -padlen
1697
 *      case.
1698
 *
1699
 * \-nostrend
1700
 *      Don't set the END_STREAM flag, allowing to send more data on this
1701
 *      stream.
1702
 */
1703
static void
1704 12800
cmd_txdata(CMD_ARGS)
1705
{
1706
        struct stream *s;
1707 12800
        char *pad = NULL;
1708
        struct frame f;
1709 12800
        char *body = NULL;
1710 12800
        char *data = NULL;
1711
1712 12800
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1713
1714 12800
        INIT_FRAME(f, DATA, 0, s->id, END_STREAM);
1715
1716 37480
        while (*++av) {
1717 24680
                if (!strcmp(*av, "-data")) {
1718 1000
                        AZ(body);
1719 1000
                        av++;
1720 1000
                        body = strdup(*av);
1721 24680
                } else if (!strcmp(*av, "-datalen")) {
1722 1400
                        AZ(body);
1723 1400
                        av++;
1724 1400
                        body = synth_body(*av, 0);
1725 23680
                } else if (!strcmp(*av, "-pad")) {
1726 40
                        AZ(pad);
1727 40
                        av++;
1728 40
                        AN(*av);
1729 40
                        pad = strdup(*av);
1730 22280
                } else if (!strcmp(*av, "-padlen")) {
1731 10720
                        AZ(pad);
1732 10720
                        av++;
1733 10720
                        pad = synth_body(*av, 0);
1734 22240
                } else if (!strcmp(*av, "-nostrend"))
1735 11520
                        f.flags &= ~END_STREAM;
1736
                else
1737 0
                        break;
1738
        }
1739 12800
        if (*av != NULL)
1740 0
                vtc_fatal(vl, "Unknown txdata spec: %s\n", *av);
1741
1742 12800
        if (!body)
1743 10400
                body = strdup("");
1744
1745 12800
        if (pad) {
1746 10760
                f.flags |= PADDED;
1747 10760
                if (strlen(pad) > 255)
1748 0
                        vtc_fatal(vl, "Padding is limited to 255 bytes");
1749 10760
                data = malloc( 1 + strlen(body) + strlen(pad));
1750 10760
                AN(data);
1751 10760
                *((uint8_t *)data) = strlen(pad);
1752 10760
                f.size = 1;
1753 10760
                memcpy(data + f.size, body, strlen(body));
1754 10760
                f.size += strlen(body);
1755 10760
                memcpy(data + f.size, pad, strlen(pad));
1756 10760
                f.size += strlen(pad);
1757 10760
                f.data = data;
1758 10760
        } else {
1759 2040
                f.size = strlen(body);
1760 2040
                f.data = body;
1761
        }
1762 12800
        write_frame(s, &f, 1);
1763 12800
        free(body);
1764 12800
        free(pad);
1765 12800
        free(data);
1766 12800
}
1767
1768
/* SECTION: stream.spec.reset_txrst txrst
1769
 *
1770
 * Send a RST_STREAM frame. By default, txrst will send a 0 error code
1771
 * (NO_ERROR).
1772
 *
1773
 * \-err STRING|INT
1774
 *      Sets the error code to be sent. The argument can be an integer or a
1775
 *      string describing the error, such as NO_ERROR, or CANCEL (see
1776
 *      rfc7540#11.4 for more strings).
1777
 */
1778
static void
1779 760
cmd_txrst(CMD_ARGS)
1780
{
1781
        struct stream *s;
1782
        char *p;
1783 760
        uint32_t err = 0;
1784
        struct frame f;
1785
1786 760
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1787
1788 760
        INIT_FRAME(f, RST_STREAM, 4, s->id, 0);
1789
1790 1080
        while (*++av) {
1791 320
                if (!strcmp(*av, "-err")) {
1792 320
                        ++av;
1793 4800
                        for (err = 0; h2_errs[err]; err++) {
1794 4480
                                if (!strcmp(h2_errs[err], *av))
1795 0
                                        break;
1796 4480
                        }
1797
1798 320
                        if (h2_errs[err])
1799 0
                                continue;
1800
1801 320
                        STRTOU32(err, *av, p, vl, "-err");
1802 320
                } else
1803 0
                        break;
1804
        }
1805 760
        if (*av != NULL)
1806 0
                vtc_fatal(vl, "Unknown txrst spec: %s\n", *av);
1807
1808 760
        err = htonl(err);
1809 760
        f.data = (void *)&err;
1810 760
        write_frame(s, &f, 1);
1811 760
}
1812
1813
/* SECTION: stream.spec.prio_txprio txprio
1814
 *
1815
 * Send a PRIORITY frame
1816
 *
1817
 * \-stream INT
1818
 *      indicate the id of the stream the sender stream depends on.
1819
 *
1820
 * \-ex
1821
 *      the dependency should be made exclusive (only this streams depends on
1822
 *      the parent stream).
1823
 *
1824
 * \-weight INT
1825
 *      an 8-bits integer is used to balance priority between streams
1826
 *      depending on the same streams.
1827
 */
1828
static void
1829 520
cmd_txprio(CMD_ARGS)
1830
{
1831
        struct stream *s;
1832
        char *p;
1833 520
        uint32_t stid = 0;
1834
        struct frame f;
1835 520
        uint32_t weight = 0;
1836 520
        uint32_t exclusive = 0;
1837
        uint8_t buf[5];
1838
1839 520
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1840
1841 520
        INIT_FRAME(f, PRIORITY, 5, s->id, 0);
1842 520
        f.data = (void *)buf;
1843
1844 1240
        while (*++av) {
1845 720
                if (!strcmp(*av, "-stream")) {
1846 360
                        STRTOU32_CHECK(stid, av, p, vl, "-stream", 0);
1847 720
                } else if (!strcmp(*av, "-ex")) {
1848 40
                        exclusive = 1U << 31;
1849 360
                } else if (!strcmp(*av, "-weight")) {
1850 320
                        STRTOU32_CHECK(weight, av, p, vl, "-weight", 8);
1851 320
                } else
1852 0
                        break;
1853
        }
1854 520
        if (*av != NULL)
1855 0
                vtc_fatal(vl, "Unknown txprio spec: %s\n", *av);
1856 520
        s->weight = weight & 0xff;
1857 520
        s->dependency = stid;
1858
1859 520
        if (exclusive)
1860 40
                exclusive_stream_dependency(s);
1861
1862 520
        vbe32enc(buf, (stid | exclusive));
1863 520
        buf[4] = s->weight;
1864 520
        write_frame(s, &f, 1);
1865 520
}
1866
1867
#define PUT_KV(av, vl, name, val, code) \
1868
        do {\
1869
                STRTOU32_CHECK(val, av, p, vl, #name, 0);       \
1870
                vbe16enc(cursor, code);                         \
1871
                cursor += sizeof(uint16_t);                     \
1872
                vbe32enc(cursor, val);                          \
1873
                cursor += sizeof(uint32_t);                     \
1874
                f.size += 6;                                    \
1875
        } while(0)
1876
1877
/* SECTION: stream.spec.settings_txsettings txsettings
1878
 *
1879
 * SETTINGS frames must be acknowledge, arguments are as follow (most of them
1880
 * are from  rfc7540#6.5.2):
1881
 *
1882
 * \-hdrtbl INT
1883
 *      headers table size
1884
 *
1885
 * \-push BOOL
1886
 *      whether push frames are accepted or not
1887
 *
1888
 * \-maxstreams INT
1889
 *      maximum concurrent streams allowed
1890
 *
1891
 * \-winsize INT
1892
 *      sender's initial window size
1893
 *
1894
 * \-framesize INT
1895
 *      largest frame size authorized
1896
 *
1897
 * \-hdrsize INT
1898
 *      maximum size of the header list authorized
1899
 *
1900
 * \-ack
1901
 *      set the ack bit
1902
 */
1903
static void
1904 15519
cmd_txsettings(CMD_ARGS)
1905
{
1906
        struct stream *s, *s2;
1907
        struct http *hp;
1908
        char *p;
1909 15519
        uint32_t val = 0;
1910
        struct frame f;
1911
        //TODO dynamic alloc
1912
        char buf[512];
1913 15519
        char *cursor = buf;
1914
1915 15519
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1916 15519
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);
1917
1918 15519
        memset(buf, 0, 512);
1919 15519
        INIT_FRAME(f, SETTINGS, 0, s->id, 0);
1920 15519
        f.data = buf;
1921
1922 15519
        PTOK(pthread_mutex_lock(&hp->mtx));
1923 23959
        while (*++av) {
1924 8440
                if (!strcmp(*av, "-push")) {
1925 40
                        ++av;
1926 40
                        vbe16enc(cursor, 0x2);
1927 40
                        cursor += sizeof(uint16_t);
1928 40
                        if (!strcmp(*av, "false"))
1929 0
                                vbe32enc(cursor, 0);
1930 40
                        else if (!strcmp(*av, "true"))
1931 40
                                vbe32enc(cursor, 1);
1932
                        else
1933 0
                                vtc_fatal(vl, "Push parameter is either "
1934 0
                                    "\"true\" or \"false\", not %s", *av);
1935 40
                        cursor += sizeof(uint32_t);
1936 40
                        f.size += 6;
1937 40
                }
1938 8400
                else if (!strcmp(*av, "-hdrtbl")) {
1939 200
                        PUT_KV(av, vl, hdrtbl, val, 0x1);
1940 200
                        assert(HPK_ResizeTbl(s->hp->decctx, val) != hpk_err);
1941 200
                }
1942 8200
                else if (!strcmp(*av, "-maxstreams"))
1943 40
                        PUT_KV(av, vl, maxstreams, val, 0x3);
1944 8160
                else if (!strcmp(*av, "-winsize"))      {
1945 320
                        PUT_KV(av, vl, winsize, val, 0x4);
1946 680
                        VTAILQ_FOREACH(s2, &hp->streams, list)
1947 360
                                s2->win_self += (val - hp->h2_win_self->init);
1948 320
                        hp->h2_win_self->init = val;
1949 320
                }
1950 7840
                else if (!strcmp(*av, "-framesize"))
1951 120
                        PUT_KV(av, vl, framesize, val, 0x5);
1952 7720
                else if (!strcmp(*av, "-hdrsize"))
1953 40
                        PUT_KV(av, vl, hdrsize, val, 0x6);
1954 7680
                else if (!strcmp(*av, "-ack"))
1955 7680
                        f.flags |= 1;
1956
                else
1957 0
                        break;
1958
        }
1959 15519
        if (*av != NULL)
1960 0
                vtc_fatal(vl, "Unknown txsettings spec: %s\n", *av);
1961
1962 15519
        AN(s->hp);
1963 15519
        write_frame(s, &f, 0);
1964 15519
        PTOK(pthread_mutex_unlock(&hp->mtx));
1965 15519
}
1966
1967
/* SECTION: stream.spec.ping_txping txping
1968
 *
1969
 * Send PING frame.
1970
 *
1971
 * \-data STRING
1972
 *      specify the payload of the frame, with STRING being an 8-char string.
1973
 *
1974
 * \-ack
1975
 *      set the ACK flag.
1976
 */
1977
static void
1978 280
cmd_txping(CMD_ARGS)
1979
{
1980
        struct stream *s;
1981
        struct frame f;
1982
        char buf[8];
1983
1984 280
        memset(buf, 0, 8);
1985 280
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1986 280
        INIT_FRAME(f, PING, 8, s->id, 0);
1987
1988 560
        while (*++av) {
1989 280
                if (!strcmp(*av, "-data")) {
1990 200
                        av++;
1991 200
                        if (f.data)
1992 0
                                vtc_fatal(vl, "this frame already has data");
1993 200
                        if (strlen(*av) != 8)
1994 0
                                vtc_fatal(vl, "data must be a 8-char string, found  (%s)", *av);
1995 200
                        f.data = *av;
1996 280
                } else if (!strcmp(*av, "-ack"))
1997 80
                        f.flags |= 1;
1998
                else
1999 0
                        break;
2000
        }
2001 280
        if (*av != NULL)
2002 0
                vtc_fatal(vl, "Unknown txping spec: %s\n", *av);
2003 280
        if (!f.data)
2004 80
                f.data = buf;
2005 280
        write_frame(s, &f, 1);
2006 280
}
2007
2008
/*
2009
 * SECTION: stream.spec.goaway_txgoaway txgoaway
2010
 *
2011
 * Possible options include:
2012
 *
2013
 * \-err STRING|INT
2014
 *      set the error code to explain the termination. The second argument
2015
 *      can be a integer or the string version of the error code as found
2016
 *      in rfc7540#7.
2017
 *
2018
 * \-laststream INT
2019
 *      the id of the "highest-numbered stream identifier for which the
2020
 *      sender of the GOAWAY frame might have taken some action on or might
2021
 *      yet take action on".
2022
 *
2023
 * \-debug
2024
 *      specify the debug data, if any to append to the frame.
2025
 */
2026
static void
2027 200
cmd_txgoaway(CMD_ARGS)
2028
{
2029
        struct stream *s;
2030
        char *p;
2031 200
        uint32_t err = 0;
2032 200
        uint32_t ls = 0;
2033
        struct frame f;
2034
2035 200
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2036
2037 200
        INIT_FRAME(f, GOAWAY, 8, s->id, 0);
2038
2039 560
        while (*++av) {
2040 360
                if (!strcmp(*av, "-err")) {
2041 200
                        ++av;
2042 3000
                        for (err = 0; h2_errs[err]; err++)
2043 2800
                                if (!strcmp(h2_errs[err], *av))
2044 0
                                        break;
2045
2046 200
                        if (h2_errs[err])
2047 0
                                continue;
2048
2049 200
                        STRTOU32(err, *av, p, vl, "-err");
2050 360
                } else if (!strcmp(*av, "-laststream")) {
2051 80
                        STRTOU32_CHECK(ls, av, p, vl, "-laststream", 31);
2052 160
                } else if (!strcmp(*av, "-debug")) {
2053 80
                        ++av;
2054 80
                        if (f.data)
2055 0
                                vtc_fatal(vl, "this frame already has debug data");
2056 80
                        f.size = 8 + strlen(*av);
2057 80
                        f.data = malloc(f.size);
2058 80
                        AN(f.data);
2059 80
                        memcpy(f.data + 8, *av, f.size - 8);
2060 80
                } else
2061 0
                        break;
2062
        }
2063 200
        if (*av != NULL)
2064 0
                vtc_fatal(vl, "Unknown txgoaway spec: %s\n", *av);
2065
2066 200
        if (!f.data) {
2067 120
                f.data = malloc(8);
2068 120
                AN(f.data);
2069 120
        }
2070 200
        vbe32enc(f.data, ls);
2071 200
        vbe32enc(f.data + 4, err);
2072 200
        write_frame(s, &f, 1);
2073 200
        free(f.data);
2074 200
}
2075
2076
/* SECTION: stream.spec.winup_txwinup txwinup
2077
 *
2078
 * Transmit a WINDOW_UPDATE frame, increasing the amount of credit of the
2079
 * connection (from stream 0) or of the stream (any other stream).
2080
 *
2081
 * \-size INT
2082
 *      give INT credits to the peer.
2083
 */
2084
static void
2085 960
cmd_txwinup(CMD_ARGS)
2086
{
2087
        struct http *hp;
2088
        struct stream *s;
2089
        char *p;
2090
        struct frame f;
2091
        char buf[8];
2092 960
        uint32_t size = 0;
2093
2094 960
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2095 960
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);
2096 960
        memset(buf, 0, 8);
2097
2098 960
        AN(av[1]);
2099 960
        AN(av[2]);
2100
2101 960
        INIT_FRAME(f, WINDOW_UPDATE, 4, s->id, 0);
2102 960
        f.data = buf;
2103
2104 1920
        while (*++av)
2105 960
                if (!strcmp(*av, "-size")) {
2106 960
                        STRTOU32_CHECK(size, av, p, vl, "-size", 0);
2107 960
                } else
2108 0
                        break;
2109 960
        if (*av != NULL)
2110 0
                vtc_fatal(vl, "Unknown txwinup spec: %s\n", *av);
2111
2112 960
        PTOK(pthread_mutex_lock(&hp->mtx));
2113 960
        if (s->id == 0)
2114 400
                hp->h2_win_self->size += size;
2115 960
        s->win_self += size;
2116 960
        PTOK(pthread_mutex_unlock(&hp->mtx));
2117
2118 960
        size = htonl(size);
2119 960
        f.data = (void *)&size;
2120 960
        write_frame(s, &f, 1);
2121 960
}
2122
2123
static struct frame *
2124 32464
rxstuff(struct stream *s)
2125
{
2126
        struct frame *f;
2127
2128 32464
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
2129
2130 32464
        PTOK(pthread_mutex_lock(&s->hp->mtx));
2131 32464
        if (VTAILQ_EMPTY(&s->fq)) {
2132 32375
                assert(s->hp->wf >= 0);
2133 32375
                s->hp->wf++;
2134 32375
                s->wf = 1;
2135 32375
                PTOK(pthread_cond_signal(&s->hp->cond));
2136 32375
                PTOK(pthread_cond_wait(&s->cond, &s->hp->mtx));
2137 32375
        }
2138 32464
        if (VTAILQ_EMPTY(&s->fq)) {
2139 0
                PTOK(pthread_mutex_unlock(&s->hp->mtx));
2140 0
                return (NULL);
2141
        }
2142 32464
        clean_frame(&s->frame);
2143 32464
        f = VTAILQ_LAST(&s->fq, fq_head);
2144 32464
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
2145 32464
        VTAILQ_REMOVE(&s->fq, f, list);
2146 32464
        PTOK(pthread_mutex_unlock(&s->hp->mtx));
2147 32464
        return (f);
2148 32464
}
2149
2150
#define CHKFRAME(rt, wt, rcv, func) \
2151
        do { \
2152
        if (rt != wt) \
2153
                vtc_fatal(vl, "Frame #%d for %s was of type %s (%d) " \
2154
                    "instead of %s (%d)", \
2155
                    rcv, func, \
2156
                    rt < TYPE_MAX ? h2_types[rt] : "?", rt, \
2157
                    wt < TYPE_MAX ? h2_types[wt] : "?", wt); \
2158
        } while (0);
2159
2160
/* SECTION: stream.spec.data_11 rxhdrs
2161
 *
2162
 * ``rxhdrs`` will expect one HEADER frame, then, depending on the arguments,
2163
 * zero or more CONTINUATION frame.
2164
 *
2165
 * \-all
2166
 *      Keep waiting for CONTINUATION frames until END_HEADERS flag is seen.
2167
 *
2168
 * \-some INT
2169
 *      Retrieve INT - 1 CONTINUATION frames after the HEADER frame.
2170
 *
2171
 */
2172
static void
2173 560
cmd_rxhdrs(CMD_ARGS)
2174
{
2175
        struct stream *s;
2176 560
        struct frame *f = NULL;
2177
        char *p;
2178 560
        int loop = 0;
2179 560
        unsigned long int times = 1;
2180 560
        unsigned rcv = 0;
2181 560
        enum h2_type_e expect = TYPE_HEADERS;
2182
2183 560
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2184
2185 640
        while (*++av) {
2186 80
                if (!strcmp(*av, "-some")) {
2187 40
                        STRTOU32_CHECK(times, av, p, vl, "-some", 0);
2188 40
                        if (!times)
2189 0
                                vtc_fatal(vl, "-some argument must be more"
2190 0
                                               "than 0 (found \"%s\")\n", *av);
2191 80
                } else if (!strcmp(*av, "-all"))
2192 40
                        loop = 1;
2193
                else
2194 0
                        break;
2195
        }
2196 560
        if (*av != NULL)
2197 0
                vtc_fatal(vl, "Unknown rxhdrs spec: %s\n", *av);
2198
2199 560
        do {
2200 680
                replace_frame(&f, rxstuff(s));
2201 680
                if (f == NULL)
2202 0
                        break;
2203 680
                rcv++;
2204 680
                CHKFRAME(f->type, expect, rcv, "rxhdrs");
2205 680
                expect = TYPE_CONTINUATION;
2206 680
        } while (rcv < times || (loop && !(f->flags & END_HEADERS)));
2207 560
        replace_frame(&s->frame, f);
2208 560
}
2209
2210
static void
2211 200
cmd_rxcont(CMD_ARGS)
2212
{
2213
        struct stream *s;
2214 200
        struct frame *f = NULL;
2215
        char *p;
2216 200
        int loop = 0;
2217 200
        unsigned long int times = 1;
2218 200
        unsigned rcv = 0;
2219
2220 200
        (void)av;
2221 200
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2222
2223 200
        while (*++av)
2224 0
                if (!strcmp(*av, "-some")) {
2225 0
                        STRTOU32(times, *av, p, vl, "-some");
2226 0
                        if (!times)
2227 0
                                vtc_fatal(vl, "-some argument must be more"
2228 0
                                               "than 0 (found \"%s\")\n", *av);
2229 0
                } else if (!strcmp(*av, "-all"))
2230 0
                        loop = 1;
2231
                else
2232 0
                        break;
2233 200
        if (*av != NULL)
2234 0
                vtc_fatal(vl, "Unknown rxcont spec: %s\n", *av);
2235
2236 200
        do {
2237 200
                replace_frame(&f, rxstuff(s));
2238 200
                if (f == NULL)
2239 0
                        break;
2240 200
                rcv++;
2241 200
                CHKFRAME(f->type, TYPE_CONTINUATION, rcv, "rxcont");
2242 200
        } while (rcv < times || (loop && !(f->flags & END_HEADERS)));
2243 200
        replace_frame(&s->frame, f);
2244 200
}
2245
2246
2247
/* SECTION: stream.spec.data_13 rxdata
2248
 *
2249
 * Receiving data is done using the ``rxdata`` keywords and will retrieve one
2250
 * DATA frame, if you wish to receive more, you can use these two convenience
2251
 * arguments:
2252
 *
2253
 * \-all
2254
 *      keep waiting for DATA frame until one sets the END_STREAM flag
2255
 *
2256
 * \-some INT
2257
 *      retrieve INT DATA frames.
2258
 *
2259
 */
2260
static void
2261 600
cmd_rxdata(CMD_ARGS)
2262
{
2263
        struct stream *s;
2264 600
        struct frame *f = NULL;
2265
        char *p;
2266 600
        int loop = 0;
2267 600
        unsigned long int times = 1;
2268 600
        unsigned rcv = 0;
2269
2270 600
        (void)av;
2271 600
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2272
2273 640
        while (*++av)
2274 40
                if (!strcmp(*av, "-some")) {
2275 40
                        av++;
2276 40
                        STRTOU32(times, *av, p, vl, "-some");
2277 40
                        if (!times)
2278 0
                                vtc_fatal(vl, "-some argument must be more"
2279 0
                                               "than 0 (found \"%s\")\n", *av);
2280 40
                } else if (!strcmp(*av, "-all"))
2281 0
                        loop = 1;
2282
                else
2283 0
                        break;
2284 600
        if (*av != NULL)
2285 0
                vtc_fatal(vl, "Unknown rxdata spec: %s\n", *av);
2286
2287 600
        do {
2288 680
                replace_frame(&f, rxstuff(s));
2289 680
                if (f == NULL)
2290 0
                        break;
2291 680
                rcv++;
2292 680
                CHKFRAME(f->type, TYPE_DATA, rcv, "rxhdata");
2293 680
        } while (rcv < times || (loop && !(f->flags & END_STREAM)));
2294 600
        replace_frame(&s->frame, f);
2295 600
}
2296
2297
/* SECTION: stream.spec.data_10 rxreq, rxresp
2298
 *
2299
 * These are two convenience functions to receive headers and body of an
2300
 * incoming request or response. The only difference is that rxreq can only be
2301
 * by a server, and rxresp by a client.
2302
 *
2303
 */
2304
2305
#define cmd_rxreq       cmd_rxmsg
2306
#define cmd_rxresp      cmd_rxmsg
2307
2308
static void
2309 6480
cmd_rxmsg(CMD_ARGS)
2310
{
2311
        struct stream *s;
2312 6480
        struct frame *f = NULL;
2313
        int end_stream;
2314 6480
        int rcv = 0;
2315
2316 6480
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2317
2318 6480
        if (!strcmp(av[0], "rxreq"))
2319 1840
                ONLY_H2_SERVER(s->hp, av);
2320
        else
2321 4640
                ONLY_H2_CLIENT(s->hp, av);
2322
2323 6480
        do {
2324 6560
                replace_frame(&f, rxstuff(s));
2325 6560
                CHECK_OBJ_ORNULL(f, FRAME_MAGIC);
2326 6560
                if (f == NULL)
2327 0
                        return;
2328 6560
        } while (f->type == TYPE_WINDOW_UPDATE);
2329
2330 6480
        rcv++;
2331 6480
        CHKFRAME(f->type, TYPE_HEADERS, rcv, *av);
2332
2333 6480
        end_stream = f->flags & END_STREAM;
2334
2335 6720
        while (!(f->flags & END_HEADERS)) {
2336 240
                replace_frame(&f, rxstuff(s));
2337 240
                CHECK_OBJ_ORNULL(f, FRAME_MAGIC);
2338 240
                if (f == NULL)
2339 0
                        return;
2340 240
                rcv++;
2341 240
                CHKFRAME(f->type, TYPE_CONTINUATION, rcv, *av);
2342
        }
2343
2344 9800
        while (!end_stream) {
2345 3320
                replace_frame(&f, rxstuff(s));
2346 3320
                CHECK_OBJ_ORNULL(f, FRAME_MAGIC);
2347 3320
                if (f == NULL)
2348 0
                        break;
2349 3320
                rcv++;
2350 3320
                CHKFRAME(f->type, TYPE_DATA, rcv, *av);
2351 3320
                end_stream = f->flags & END_STREAM;
2352
        }
2353 6480
        replace_frame(&s->frame, f);
2354 6480
}
2355
2356
/* SECTION: stream.spec.data_12 rxpush
2357
 *
2358
 * This works like ``rxhdrs``, expecting a PUSH frame and then zero or more
2359
 * CONTINUATION frames.
2360
 *
2361
 * \-all
2362
 *      Keep waiting for CONTINUATION frames until END_HEADERS flag is seen.
2363
 *
2364
 * \-some INT
2365
 *      Retrieve INT - 1 CONTINUATION frames after the PUSH frame.
2366
 *
2367
 */
2368
static void
2369 40
cmd_rxpush(CMD_ARGS)
2370
{
2371
        struct stream *s;
2372 40
        struct frame *f = NULL;
2373
        char *p;
2374 40
        int loop = 0;
2375 40
        unsigned long int times = 1;
2376 40
        unsigned rcv = 0;
2377 40
        enum h2_type_e expect = TYPE_PUSH_PROMISE;
2378
2379 40
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2380
2381 40
        while (*++av) {
2382 0
                if (!strcmp(*av, "-some")) {
2383 0
                        STRTOU32_CHECK(times, av, p, vl, "-some", 0);
2384 0
                        if (!times)
2385 0
                                vtc_fatal(vl, "-some argument must be more"
2386 0
                                               "than 0 (found \"%s\")\n", *av);
2387 0
                } else if (!strcmp(*av, "-all")) {
2388 0
                        loop = 1;
2389 0
                } else
2390 0
                        break;
2391
        }
2392 40
        if (*av != NULL)
2393 0
                vtc_fatal(vl, "Unknown rxpush spec: %s\n", *av);
2394
2395 40
        do {
2396 40
                f = rxstuff(s);
2397 40
                if (!f)
2398 0
                        return;
2399 40
                rcv++;
2400 40
                CHKFRAME(f->type, expect, rcv, "rxpush");
2401 40
                expect = TYPE_CONTINUATION;
2402 40
        } while (rcv < times || (loop && !(f->flags & END_HEADERS)));
2403 40
        s->frame = f;
2404 40
}
2405
2406
/* SECTION: stream.spec.winup_rxwinup rxwinup
2407
 *
2408
 * Receive a WINDOW_UPDATE frame.
2409
 */
2410
static void
2411 1000
cmd_rxwinup(CMD_ARGS)
2412
{
2413
        struct stream *s;
2414
        struct frame *f;
2415
2416 1000
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2417 1000
        s->frame = rxstuff(s);
2418 1000
        CAST_OBJ_NOTNULL(f, s->frame, FRAME_MAGIC);
2419 1000
        CHKFRAME(f->type, TYPE_WINDOW_UPDATE, 0, *av);
2420 1000
        if (s->id == 0)
2421 240
                s->hp->h2_win_peer->size += s->frame->md.winup_size;
2422 1000
        s->win_peer += s->frame->md.winup_size;
2423 1000
}
2424
2425
/* SECTION: stream.spec.settings_rxsettings rxsettings
2426
 *
2427
 * Receive a SETTINGS frame.
2428
 */
2429
static void
2430 15596
cmd_rxsettings(CMD_ARGS)
2431
{
2432
        struct stream *s, *s2;
2433 15596
        uint32_t val = 0;
2434
        struct http *hp;
2435
        struct frame *f;
2436
2437 15596
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2438 15596
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);
2439 15596
        s->frame = rxstuff(s);
2440 15596
        CAST_OBJ_NOTNULL(f, s->frame, FRAME_MAGIC);
2441 15596
        CHKFRAME(f->type, TYPE_SETTINGS, 0, *av);
2442 15596
        if (! isnan(f->md.settings[SETTINGS_INITIAL_WINDOW_SIZE])) {
2443 4960
                val = (uint32_t)f->md.settings[SETTINGS_INITIAL_WINDOW_SIZE];
2444 10200
                VTAILQ_FOREACH(s2, &hp->streams, list)
2445 5240
                        s2->win_peer += (val - hp->h2_win_peer->init);
2446 4960
                hp->h2_win_peer->init = val;
2447 4960
        }
2448 15596
}
2449
2450
#define RXFUNC(lctype, upctype) \
2451
        static void \
2452
        cmd_rx ## lctype(CMD_ARGS) { \
2453
                struct stream *s; \
2454
                (void)av; \
2455
                CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC); \
2456
                s->frame = rxstuff(s); \
2457
                if (s->frame != NULL && s->frame->type != TYPE_ ## upctype) \
2458
                        vtc_fatal(vl, \
2459
                            "Wrong frame type %s (%d) wanted %s", \
2460
                            s->frame->type < TYPE_MAX ? \
2461
                            h2_types[s->frame->type] : "?", \
2462
                            s->frame->type, #upctype); \
2463
        }
2464
2465
/* SECTION: stream.spec.prio_rxprio rxprio
2466
 *
2467
 * Receive a PRIORITY frame.
2468
 */
2469 360
RXFUNC(prio,    PRIORITY)
2470
2471
/* SECTION: stream.spec.reset_rxrst rxrst
2472
 *
2473
 * Receive a RST_STREAM frame.
2474
 */
2475 2160
RXFUNC(rst,     RST_STREAM)
2476
2477
/* SECTION: stream.spec.ping_rxping rxping
2478
 *
2479
 * Receive a PING frame.
2480
 */
2481 240
RXFUNC(ping,    PING)
2482
2483
/* SECTION: stream.spec.goaway_rxgoaway rxgoaway
2484
 *
2485
 * Receive a GOAWAY frame.
2486
 */
2487 1400
RXFUNC(goaway,  GOAWAY)
2488
2489
/* SECTION: stream.spec.frame_rxframe
2490
 *
2491
 * Receive a frame, any frame.
2492
 */
2493
static void
2494 0
cmd_rxframe(CMD_ARGS)
2495
{
2496
        struct stream *s;
2497
2498 0
        (void)vl;
2499 0
        (void)av;
2500 0
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2501 0
        if (rxstuff(s) == NULL)
2502 0
                vtc_fatal(s->vl, "No frame received");
2503 0
}
2504
2505
static void
2506 27160
cmd_expect(CMD_ARGS)
2507
{
2508
        struct http *hp;
2509
        struct stream *s;
2510
        const char *lhs;
2511
        char *cmp;
2512
        const char *rhs;
2513
        char buf[20];
2514
2515 27160
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2516 27160
        hp = s->hp;
2517 27160
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
2518
2519 27160
        AZ(strcmp(av[0], "expect"));
2520 27160
        av++;
2521
2522 27160
        AN(av[0]);
2523 27160
        AN(av[1]);
2524 27160
        AN(av[2]);
2525 27160
        AZ(av[3]);
2526 27160
        PTOK(pthread_mutex_lock(&s->hp->mtx));
2527 27160
        lhs = cmd_var_resolve(s, av[0], buf);
2528 27160
        cmp = av[1];
2529 27160
        rhs = cmd_var_resolve(s, av[2], buf);
2530 27160
        vtc_expect(vl, av[0], lhs, cmp, av[2], rhs);
2531 27160
        PTOK(pthread_mutex_unlock(&s->hp->mtx));
2532 27160
}
2533
2534
/* SECTION: stream.spec.gunzip gunzip
2535
 *
2536
 * Same as the ``gunzip`` command for HTTP/1.
2537
 */
2538
static void
2539 80
cmd_gunzip(CMD_ARGS)
2540
{
2541
        struct http *hp;
2542
        struct stream *s;
2543
2544 80
        (void)av;
2545 80
        (void)vl;
2546
2547 80
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2548 80
        hp = s->hp;
2549 80
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
2550
2551 80
        vtc_gunzip(s->hp, s->body, &s->bodylen);
2552 80
}
2553
2554
/* SECTION: stream.spec.write_body
2555
 *
2556
 * write_body STRING
2557
 *      Same as the ``write_body`` command for HTTP/1.
2558
 */
2559
static void
2560 80
cmd_write_body(CMD_ARGS)
2561
{
2562
        struct stream *s;
2563
2564 80
        (void)vl;
2565 80
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2566 80
        AN(av[0]);
2567 80
        AN(av[1]);
2568 80
        AZ(av[2]);
2569 80
        AZ(strcmp(av[0], "write_body"));
2570 80
        if (VFIL_writefile(NULL, av[1], s->body, s->bodylen) != 0)
2571 0
                vtc_fatal(s->vl, "failed to write body: %s (%d)",
2572 0
                    strerror(errno), errno);
2573 80
}
2574
2575
/* SECTION: stream.spec Specification
2576
 *
2577
 * The specification of a stream follows the exact same rules as one for a
2578
 * client or a server.
2579
 */
2580
static const struct cmds stream_cmds[] = {
2581
#define CMD_STREAM(n) { #n, cmd_##n },
2582
        /* spec */
2583
        CMD_STREAM(expect)
2584
        CMD_STREAM(gunzip)
2585
        CMD_STREAM(rxcont)
2586
        CMD_STREAM(rxdata)
2587
        CMD_STREAM(rxframe)
2588
        CMD_STREAM(rxgoaway)
2589
        CMD_STREAM(rxhdrs)
2590
        CMD_STREAM(rxping)
2591
        CMD_STREAM(rxprio)
2592
        CMD_STREAM(rxpush)
2593
        CMD_STREAM(rxreq)
2594
        CMD_STREAM(rxresp)
2595
        CMD_STREAM(rxrst)
2596
        CMD_STREAM(rxsettings)
2597
        CMD_STREAM(rxwinup)
2598
        CMD_STREAM(sendhex)
2599
        CMD_STREAM(txcont)
2600
        CMD_STREAM(txdata)
2601
        CMD_STREAM(txgoaway)
2602
        CMD_STREAM(txping)
2603
        CMD_STREAM(txprio)
2604
        CMD_STREAM(txpush)
2605
        CMD_STREAM(txreq)
2606
        CMD_STREAM(txresp)
2607
        CMD_STREAM(txrst)
2608
        CMD_STREAM(txsettings)
2609
        CMD_STREAM(txwinup)
2610
        CMD_STREAM(write_body)
2611
        { NULL, NULL }
2612
#undef CMD_STREAM
2613
};
2614
2615
static void *
2616 21517
stream_thread(void *priv)
2617
{
2618
        struct stream *s;
2619
2620 21517
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2621 21517
        parse_string(s->vl, s, s->spec);
2622 21517
        vtc_log(s->vl, 2, "Ending stream %u", s->id);
2623 21517
        return (NULL);
2624
}
2625
/**********************************************************************
2626
 * Allocate and initialize a stream
2627
 */
2628
2629
static struct stream *
2630 18360
stream_new(const char *name, struct http *h)
2631
{
2632
        char *p, buf[20];
2633
        struct stream *s;
2634
2635 18360
        if (!strcmp("next", name)) {
2636 799
                if (h->last_stream > 0)
2637 520
                        bprintf(buf, "%d", h->last_stream + 2);
2638
                else
2639 279
                        bprintf(buf, "%d", 1);
2640 799
                name = buf;
2641 799
        }
2642
2643 18360
        ALLOC_OBJ(s, STREAM_MAGIC);
2644 18360
        AN(s);
2645 18360
        PTOK(pthread_cond_init(&s->cond, NULL));
2646 18360
        REPLACE(s->name, name);
2647 18360
        AN(s->name);
2648 18360
        VTAILQ_INIT(&s->fq);
2649 18360
        s->win_self = h->h2_win_self->init;
2650 18360
        s->win_peer = h->h2_win_peer->init;
2651 18360
        s->vl = vtc_logopen("%s.%s", h->sess->name, name);
2652 18360
        vtc_log_set_cmd(s->vl, stream_cmds);
2653
2654 18360
        s->weight = 16;
2655 18360
        s->dependency = 0;
2656
2657 18360
        STRTOU32(s->id, name, p, s->vl, "stream");
2658 18360
        if (s->id & (1U << 31))
2659 0
                vtc_fatal(s->vl, "Stream id must be a 31-bits integer "
2660 0
                    "(found %s)", name);
2661
2662 18360
        CHECK_OBJ_NOTNULL(h, HTTP_MAGIC);
2663 18360
        s->hp = h;
2664 18360
        h->last_stream = s->id;
2665
2666
        //bprintf(s->connect, "%s", "${v1_sock}");
2667 18360
        PTOK(pthread_mutex_lock(&h->mtx));
2668 18360
        VTAILQ_INSERT_HEAD(&h->streams, s, list);
2669 18360
        PTOK(pthread_mutex_unlock(&h->mtx));
2670 18360
        return (s);
2671
}
2672
2673
/**********************************************************************
2674
 * Clean up stream
2675
 */
2676
2677
static void
2678 18360
stream_delete(struct stream *s)
2679
{
2680
        struct frame *f, *f2;
2681
2682 18360
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
2683
2684 19680
        VTAILQ_FOREACH_SAFE(f, &s->fq, list, f2) {
2685 1320
                VTAILQ_REMOVE(&s->fq, f, list);
2686 1320
                clean_frame(&f);
2687 1320
        }
2688 18360
        vtc_logclose(s->vl);
2689 18360
        clean_headers(s->req);
2690 18360
        clean_headers(s->resp);
2691 18360
        AZ(s->frame);
2692 18360
        free(s->body);
2693 18360
        free(s->spec);
2694 18360
        free(s->name);
2695 18360
        FREE_OBJ(s);
2696 18360
}
2697
2698
/**********************************************************************
2699
 * Start the stream thread
2700
 */
2701
2702
static void
2703 21520
stream_start(struct stream *s)
2704
{
2705 21520
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
2706 21520
        vtc_log(s->hp->vl, 2, "Starting stream %s (%p)", s->name, s);
2707 21520
        PTOK(pthread_create(&s->tp, NULL, stream_thread, s));
2708 21520
        s->running = 1;
2709 21520
}
2710
2711
/**********************************************************************
2712
 * Wait for stream thread to stop
2713
 */
2714
static void
2715 21520
stream_wait(struct stream *s)
2716
{
2717
        void *res;
2718
        struct frame *f, *f2;
2719
2720 21520
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
2721 21520
        vtc_log(s->hp->vl, 2, "Waiting for stream %u", s->id);
2722 21520
        PTOK(pthread_join(s->tp, &res));
2723 21520
        if (res != NULL)
2724 0
                vtc_fatal(s->hp->vl, "Stream %u returned \"%s\"", s->id,
2725 0
                    (char *)res);
2726
2727 21773
        VTAILQ_FOREACH_SAFE(f, &s->fq, list, f2) {
2728 253
                VTAILQ_REMOVE(&s->fq, f, list);
2729 253
                clean_frame(&f);
2730 253
        }
2731 21520
        clean_frame(&s->frame);
2732 21520
        s->tp = 0;
2733 21520
        s->running = 0;
2734 21520
}
2735
2736
/**********************************************************************
2737
 * Run the stream thread
2738
 */
2739
2740
static void
2741 19600
stream_run(struct stream *s)
2742
{
2743 19600
        stream_start(s);
2744 19600
        stream_wait(s);
2745 19600
}
2746
2747
2748
2749
/* SECTION: client-server.spec.stream
2750
 *
2751
 * stream
2752
 *      HTTP/2 introduces the concept of streams, and these come with
2753
 *      their own specification, and as it's quite big, have been moved
2754
 *      to their own chapter.
2755
 *
2756
 * SECTION: stream stream
2757
 *
2758
 * (note: this section is at the top-level for easier navigation, but
2759
 * it's part of the client/server specification)
2760
 *
2761
 * Streams map roughly to a request in HTTP/2, a request is sent on
2762
 * stream N, the response too, then the stream is discarded. The main
2763
 * exception is the first stream, 0, that serves as coordinator.
2764
 *
2765
 * Stream syntax follow the client/server one::
2766
 *
2767
 *      stream ID [SPEC] [ACTION]
2768
 *
2769
 * ID is the HTTP/2 stream number, while SPEC describes what will be
2770
 * done in that stream. If ID has the value ``next``, the actual stream
2771
 * number is computed based on the last one.
2772
 *
2773
 * Note that, when parsing a stream action, if the entity isn't operating
2774
 * in HTTP/2 mode, these spec is ran before::
2775
 *
2776
 *      txpri/rxpri # client/server
2777
 *      stream 0 {
2778
 *          txsettings
2779
 *          rxsettings
2780
 *          txsettings -ack
2781
 *          rxsettings
2782
 *          expect settings.ack == true
2783
 *      } -run
2784
 *
2785
 * And HTTP/2 mode is then activated before parsing the specification.
2786
 *
2787
 * SECTION: stream.actions Actions
2788
 *
2789
 * \-start
2790
 *      Run the specification in a thread, giving back control immediately.
2791
 *
2792
 * \-wait
2793
 *      Wait for the started thread to finish running the spec.
2794
 *
2795
 * \-run
2796
 *      equivalent to calling ``-start`` then ``-wait``.
2797
 */
2798
2799
void
2800 22356
cmd_stream(CMD_ARGS)
2801
{
2802
        struct stream *s;
2803
        struct http *h;
2804
2805 22356
        (void)vl;
2806 22356
        CAST_OBJ_NOTNULL(h, priv, HTTP_MAGIC);
2807
2808 22356
        AZ(strcmp(av[0], "stream"));
2809 22356
        av++;
2810
2811 43695
        VTAILQ_FOREACH(s, &h->streams, list)
2812 25336
                if (!strcmp(s->name, av[0]))
2813 3997
                        break;
2814 22356
        if (s == NULL)
2815 18342
                s = stream_new(av[0], h);
2816 22356
        av++;
2817
2818 66274
        for (; *av != NULL; av++) {
2819 43918
                if (vtc_error)
2820 0
                        break;
2821
2822 43918
                if (!strcmp(*av, "-wait")) {
2823 880
                        stream_wait(s);
2824 880
                        continue;
2825
                }
2826
2827
                /* Don't muck about with a running client */
2828 43038
                if (s->running)
2829 0
                        stream_wait(s);
2830
2831 43038
                if (!strcmp(*av, "-start")) {
2832 1920
                        stream_start(s);
2833 1920
                        continue;
2834
                }
2835 41118
                if (!strcmp(*av, "-run")) {
2836 19600
                        stream_run(s);
2837 19600
                        continue;
2838
                }
2839 21518
                if (**av == '-')
2840 0
                        vtc_fatal(vl, "Unknown stream argument: %s", *av);
2841 21518
                REPLACE(s->spec, *av);
2842 21518
        }
2843 22356
}
2844
2845
void
2846 40
b64_settings(const struct http *hp, const char *s)
2847
{
2848
        uint16_t i;
2849
        uint64_t v, vv;
2850
        const char *buf;
2851
        int shift;
2852
2853 120
        while (*s) {
2854 80
                v = 0;
2855 720
                for (shift = 42; shift >= 0; shift -= 6) {
2856 640
                        if (*s >= 'A' && *s <= 'Z')
2857 520
                                vv = (*s - 'A');
2858 120
                        else if (*s >= 'a' && *s <= 'z')
2859 40
                                vv = (*s - 'a') + 26;
2860 80
                        else if (*s >= '0' && *s <= '9')
2861 0
                                vv = (*s - '0') + 52;
2862 80
                        else if (*s == '-')
2863 0
                                vv = 62;
2864 80
                        else if (*s == '_')
2865 80
                                vv = 63;
2866
                        else
2867 0
                                vtc_fatal(hp->vl,
2868
                                    "Bad \"HTTP2-Settings\" header");
2869 640
                        v |= vv << shift;
2870 640
                        s++;
2871 640
                }
2872 80
                i = v >> 32;
2873 80
                v &= 0xffff;
2874
2875 80
                if (i <= SETTINGS_MAX)
2876 80
                        buf = h2_settings[i];
2877
                else
2878 0
                        buf = "unknown";
2879
2880 80
                if (v == 1) {
2881 0
                        if (hp->sfd)
2882 0
                                assert(HPK_ResizeTbl(hp->encctx, v) != hpk_err);
2883
                        else
2884 0
                                assert(HPK_ResizeTbl(hp->decctx, v) != hpk_err);
2885 0
                }
2886
2887 160
                vtc_log(hp->vl, 4, "Upgrade: %s (%d): %ju",
2888 80
                    buf, i, (intmax_t)v);
2889
        }
2890 40
}
2891
2892
void
2893 7919
start_h2(struct http *hp)
2894
{
2895 7919
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
2896 7919
        PTOK(pthread_mutex_init(&hp->mtx, NULL));
2897 7919
        PTOK(pthread_cond_init(&hp->cond, NULL));
2898 7919
        VTAILQ_INIT(&hp->streams);
2899 7919
        hp->h2_win_self->init = 0xffff;
2900 7919
        hp->h2_win_self->size = 0xffff;
2901 7919
        hp->h2_win_peer->init = 0xffff;
2902 7919
        hp->h2_win_peer->size = 0xffff;
2903 7919
        hp->h2 = 1;
2904
2905 7919
        hp->decctx = HPK_NewCtx(4096);
2906 7919
        hp->encctx = HPK_NewCtx(4096);
2907 7919
        PTOK(pthread_create(&hp->tp, NULL, receive_frame, hp));
2908 7919
}
2909
2910
void
2911 7920
stop_h2(struct http *hp)
2912
{
2913
        struct stream *s, *s2;
2914
2915 7920
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
2916 26279
        VTAILQ_FOREACH_SAFE(s, &hp->streams, list, s2) {
2917 18359
                if (s->running)
2918 1040
                        stream_wait(s);
2919 18359
                PTOK(pthread_mutex_lock(&hp->mtx));
2920 18359
                VTAILQ_REMOVE(&hp->streams, s, list);
2921 18359
                PTOK(pthread_mutex_unlock(&hp->mtx));
2922 18359
                stream_delete(s);
2923 18359
        }
2924
2925 7920
        PTOK(pthread_mutex_lock(&hp->mtx));
2926 7920
        hp->h2 = 0;
2927 7920
        PTOK(pthread_cond_signal(&hp->cond));
2928 7920
        PTOK(pthread_mutex_unlock(&hp->mtx));
2929 7920
        PTOK(pthread_join(hp->tp, NULL));
2930
2931 7920
        HPK_FreeCtx(hp->decctx);
2932 7920
        HPK_FreeCtx(hp->encctx);
2933
2934 7920
        PTOK(pthread_mutex_destroy(&hp->mtx));
2935 7920
        PTOK(pthread_cond_destroy(&hp->cond));
2936 7920
}