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 755
#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 748
#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 706
#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 705
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 702
#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 702
        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 700
        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 696
        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 694
        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 691
119
        int                     dependency;
120
        int                     weight;
121
};
122
123
static void
124 1181
clean_headers(struct hpk_hdr *h)
125
{
126 1181
        unsigned n = MAX_HDR;
127 688
128 2324
        while (h->t && n > 0) {
129 1143
                if (h->key.len)
130 1141
                        free(h->key.ptr);
131 1143
                if (h->value.len)
132 1136
                        free(h->value.ptr);
133 1143
                memset(h, 0, sizeof(*h));
134 1143
                h++;
135 1143
                n--;
136 688
        }
137 1181
}
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 683
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 740
http_write(const struct http *hp, int lvl,
155
    const char *buf, int s, const char *pfx)
156
{
157
        ssize_t l;
158
159 57
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
160 57
        AN(buf);
161 57
        AN(pfx);
162
163 57
        vtc_dump(hp->vl, lvl, pfx, buf, s);
164 57
        l = write(hp->sess->fd, buf, s);
165 57
        if (l != s)
166 0
                vtc_log(hp->vl, hp->fatal, "Write failed: (%zd vs %d) %s",
167 0
                    l, s, strerror(errno));
168 57
}
169
170
static int
171 1506
get_bytes(const struct http *hp, char *buf, size_t n)
172
{
173
        int i;
174
        struct pollfd pfd[1];
175
176 1506
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
177 1506
        AN(buf);
178
179 3012
        while (n > 0) {
180 1506
                pfd[0].fd = hp->sess->fd;
181 1506
                pfd[0].events = POLLIN;
182 1506
                pfd[0].revents = 0;
183 1506
                i = poll(pfd, 1, (int)(hp->timeout * 1000));
184 1506
                if (i < 0 && errno == EINTR)
185 0
                        continue;
186 1506
                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 1506
                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 1506
                if (i <= 0)
195 0
                        return (i);
196 1506
                i = read(hp->sess->fd, buf, n);
197 1506
                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 1506
                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 1506
                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 1506
                if (i <= 0)
210 0
                        return (i);
211 1506
                n -= i;
212
        }
213 1506
        return (1);
214
215 1506
}
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 893
readFrameHeader(struct frame *f, const char *buf)
255
{
256 893
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
257 893
        AN(buf);
258
259 893
        f->size  = (unsigned char)buf[0] << 16;
260 893
        f->size += (unsigned char)buf[1] << 8;
261 893
        f->size += (unsigned char)buf[2];
262
263 893
        f->type = (unsigned char)buf[3];
264
265 893
        f->flags = (unsigned char)buf[4];
266
267 893
        f->stid  = vbe32dec(buf+5);
268 893
}
269
270
static void
271 2103
writeFrameHeader(char *buf, const struct frame *f)
272
{
273 2103
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
274 2103
        AN(buf);
275 2103
        buf[0] = (f->size >> 16) & 0xff;
276 2103
        buf[1] = (f->size >>  8) & 0xff;
277 2103
        buf[2] = (f->size      ) & 0xff;
278
279 2103
        buf[3] = f->type;
280
281 2103
        buf[4] = f->flags;
282
283 2103
        vbe32enc(buf + 5, f->stid);
284 2103
}
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 1999
replace_frame(struct frame **fp, struct frame *new)
298
{
299
        struct frame *old;
300
301 1999
        AN(fp);
302 1999
        CHECK_OBJ_ORNULL(new, FRAME_MAGIC);
303
304 1999
        old = *fp;
305 1999
        *fp = new;
306 1999
        if (old == NULL)
307 1103
                return;
308
309 896
        CHECK_OBJ(old, FRAME_MAGIC);
310 896
        if (old->type == TYPE_GOAWAY)
311 39
                free(old->md.goaway.debug);
312 896
        free(old->data);
313 896
        FREE_OBJ(old);
314 1999
}
315
316
static void
317 1471
clean_frame(struct frame **fp)
318
{
319
320 1471
        replace_frame(fp, NULL);
321 1471
}
322
323
static void
324 2103
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 2103
        CHECK_OBJ_NOTNULL(sp, STREAM_MAGIC);
331 2103
        hp = sp->hp;
332 2103
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
333 2103
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
334
335 2103
        writeFrameHeader(hdr, f);
336
337 4206
        vtc_log(sp->vl, 3,
338
            "tx: stream: %d, type: %s (%d), flags: 0x%02x, size: %d",
339 2103
            f->stid,
340 2103
            f->type < TYPE_MAX ? h2_types[f->type] : "?",
341 2103
            f->type, f->flags, f->size);
342
343 2103
        if (f->type == TYPE_DATA) {
344 336
                sp->win_peer -= f->size;
345 336
                hp->h2_win_peer->size -= f->size;
346 336
        }
347
348 2103
        if (lock)
349 1695
                PTOK(pthread_mutex_lock(&hp->mtx));
350 2103
        l = write(hp->sess->fd, hdr, sizeof(hdr));
351 2103
        if (l != sizeof(hdr))
352 1974
                vtc_log(sp->vl, hp->fatal, "Write failed: (%zd vs %zd) %s",
353 987
                    l, sizeof(hdr), strerror(errno));
354
355 2103
        if (f->size) {
356 1704
                AN(f->data);
357 1704
                l = write(hp->sess->fd, f->data, f->size);
358 1704
                if (l != f->size)
359 1974
                        vtc_log(sp->vl, hp->fatal,
360
                                        "Write failed: (%zd vs %d) %s",
361 987
                                        l, f->size, strerror(errno));
362 1704
        }
363 2103
        if (lock)
364 1695
                PTOK(pthread_mutex_unlock(&hp->mtx));
365 2103
}
366
367
static void
368 6
exclusive_stream_dependency(const struct stream *s)
369
{
370
        struct stream *target;
371 6
        struct http *hp = s->hp;
372
373 6
        if (s->id == 0)
374 0
                return;
375
376 28
        VTAILQ_FOREACH(target, &hp->streams, list) {
377 22
                if (target->id != s->id && target->dependency == s->dependency)
378 8
                        target->dependency = s->id;
379 22
        }
380 6
}
381
382
static void
383 896
explain_flags(uint8_t flags, uint8_t type, struct vtclog *vl)
384
{
385 896
        if (flags & ACK && (type == TYPE_PING || type == TYPE_SETTINGS)) {
386 210
                vtc_log(vl, 3, "flag: ACK");
387 945
        } else if (flags & END_STREAM && (type == TYPE_HEADERS ||
388 49
            type == TYPE_PUSH_PROMISE || type == TYPE_DATA)) {
389 193
                vtc_log(vl, 3, "flag: END_STREAM");
390 692
        } else if (flags & END_HEADERS && (type == TYPE_HEADERS ||
391 7
            type == TYPE_PUSH_PROMISE || type == TYPE_CONTINUATION)) {
392 58
                vtc_log(vl, 3, "flag: END_TYPE_HEADERS");
393 493
        } else if (flags & PRIORITY && (type == TYPE_HEADERS ||
394 0
            type == TYPE_PUSH_PROMISE)) {
395 0
                vtc_log(vl, 3, "flag: END_PRIORITY");
396 435
        } else if (flags & PADDED && (type == TYPE_DATA || type ==
397 0
            TYPE_HEADERS || type == TYPE_PUSH_PROMISE)) {
398 1
                vtc_log(vl, 3, "flag: PADDED");
399 435
        } else if (flags)
400 0
                vtc_log(vl, 3, "UNKNOWN FLAG(S): 0x%02x", flags);
401 896
}
402
403
static void
404 100
parse_data(struct stream *s, struct frame *f)
405
{
406
        struct http *hp;
407 100
        uint32_t size = f->size;
408 100
        char *data = f->data;
409
410 100
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
411 100
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
412 100
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
413
414 100
        if (f->flags & PADDED) {
415 4
                f->md.padded = *((uint8_t *)data);
416 4
                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 4
                data++;
425 4
                size -= f->md.padded + 1;
426 4
                vtc_log(s->vl, 4, "padding: %3d", f->md.padded);
427 4
        }
428
429 100
        if (s->id)
430 98
                s->win_self -= size;
431
432 100
        s->hp->h2_win_self->size -= size;
433
434 100
        if (!size) {
435 28
                AZ(data);
436 28
                vtc_log(s->vl, 4, "s%u - no data", s->id);
437 28
                return;
438
        }
439
440 72
        s->body = realloc(s->body, s->bodylen + size + 1L);
441 72
        AN(s->body);
442 72
        memcpy(s->body + s->bodylen, data, size);
443 72
        s->bodylen += size;
444 72
        s->body[s->bodylen] = '\0';
445 100
}
446
447
static void
448 197
decode_hdr(struct http *hp, struct hpk_hdr *h, const struct vsb *vsb)
449
{
450
        struct hpk_iter *iter;
451 197
        enum hpk_result r = hpk_err;
452
        int n;
453
454 197
        CHECK_OBJ_NOTNULL(vsb, VSB_MAGIC);
455 197
        CAST_OBJ_NOTNULL(hp, hp, HTTP_MAGIC);;
456
457 197
        if (VSB_len(vsb) == 0)
458 0
                return;
459
460 197
        iter = HPK_NewIter(hp->decctx, VSB_data(vsb), VSB_len(vsb));
461
462 197
        n = 0;
463 197
        while (n < MAX_HDR && h[n].t)
464 0
                n++;
465 1146
        while (n < MAX_HDR) {
466 1146
                r = HPK_DecHdr(iter, h + n);
467 1146
                if (r == hpk_err )
468 2
                        break;
469 2288
                vtc_log(hp->vl, 4, "header[%2d]: %s: %s",
470 1144
                    n, h[n].key.ptr, h[n].value.ptr);
471 1144
                n++;
472 1144
                if (r == hpk_done)
473 195
                        break;
474
        }
475
476 197
        if (r != hpk_done) {
477 4
                vtc_log(hp->vl, hp->fatal ? 4 : 0,
478 2
                    "Header decoding failed (%d) %d", r, hp->fatal);
479 197
        } 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 197
        HPK_FreeIter(iter);
485 197
}
486
487
static void
488 211
parse_hdr(struct stream *s, struct frame *f, struct vsb *vsb)
489
{
490 211
        int shift = 0;
491 211
        int exclusive = 0;
492 211
        uint32_t size = f->size;
493 211
        char *data = f->data;
494
        struct http *hp;
495
        uint32_t n;
496
497 211
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
498 211
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
499 211
        CHECK_OBJ_NOTNULL(vsb, VSB_MAGIC);
500 211
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
501
502 211
        if (f->flags & PADDED && f->type != TYPE_CONTINUATION) {
503 2
                f->md.padded = *((uint8_t *)data);
504 2
                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 2
                shift += 1;
513 2
                size -= f->md.padded;
514 2
                vtc_log(s->vl, 4, "padding: %3d", f->md.padded);
515 2
        }
516
517 211
        if (f->type == TYPE_HEADERS && f->flags & PRIORITY){
518 2
                shift += 5;
519 2
                n = vbe32dec(f->data);
520 2
                s->dependency = n & ~(1U << 31);
521 2
                exclusive = n >> 31;
522
523 2
                s->weight = f->data[4];
524 2
                if (exclusive)
525 2
                        exclusive_stream_dependency(s);
526
527 2
                vtc_log(s->vl, 4, "stream->dependency: %u", s->dependency);
528 2
                vtc_log(s->vl, 4, "stream->weight: %u", s->weight);
529 211
        } else if (f->type == TYPE_PUSH_PROMISE){
530 1
                shift += 4;
531 1
                n = vbe32dec(f->data);
532 1
                f->md.promised = n & ~(1U << 31);
533 1
        }
534
535 211
        AZ(VSB_bcat(vsb, data + shift, size - shift));
536 211
}
537
538
static void
539 9
parse_prio(struct stream *s, struct frame *f)
540
{
541
        struct http *hp;
542
        char *buf;
543
        uint32_t n;
544
545 9
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
546 9
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
547 9
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
548
549 9
        if (f->size != 5)
550 0
                vtc_fatal(s->vl, "Size should be 5, but isn't (%d)", f->size);
551
552 9
        buf = f->data;
553 9
        AN(buf);
554
555 9
        n = vbe32dec(f->data);
556 9
        f->md.prio.stream = n & ~(1U << 31);
557
558 9
        s->dependency = f->md.prio.stream;
559 9
        if (n >> 31){
560 1
                f->md.prio.exclusive = 1;
561 1
                exclusive_stream_dependency(s);
562 1
        }
563
564 9
        buf += 4;
565 9
        f->md.prio.weight = *buf;
566 9
        s->weight = f->md.prio.weight;
567
568 9
        vtc_log(s->vl, 3, "prio->stream: %u", f->md.prio.stream);
569 9
        vtc_log(s->vl, 3, "prio->weight: %u", f->md.prio.weight);
570 9
}
571
572
static void
573 54
parse_rst(const struct stream *s, struct frame *f)
574
{
575
        struct http *hp;
576
        uint32_t err;
577
        const char *buf;
578 54
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
579 54
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
580 54
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
581
582 54
        if (f->size != 4)
583 0
                vtc_fatal(s->vl, "Size should be 4, but isn't (%d)", f->size);
584
585 54
        err = vbe32dec(f->data);
586 54
        f->md.rst_err = err;
587
588 54
        vtc_log(s->vl, 2, "ouch");
589 54
        if (err <= ERR_MAX)
590 54
                buf = h2_errs[err];
591
        else
592 0
                buf = "unknown";
593 54
        vtc_log(s->vl, 4, "rst->err: %s (%d)", buf, err);
594
595 54
}
596
597
static void
598 410
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 410
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
606 410
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
607 410
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
608
609 410
        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 410
        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 3698
        for (u = 0; u <= SETTINGS_MAX; u++)
618 3288
                f->md.settings[u] = NAN;
619
620 847
        for (u = 0; u < f->size;) {
621 437
                t = vbe16dec(f->data + u);
622 437
                u += 2;
623 437
                v = vbe32dec(f->data + u);
624 437
                if (t <= SETTINGS_MAX) {
625 437
                        buf = h2_settings[t];
626 437
                        f->md.settings[t] = v;
627 437
                } else
628 0
                        buf = "unknown";
629 437
                u += 4;
630
631 437
                if (t == 1) {
632 5
                        r = HPK_ResizeTbl(s->hp->encctx, v);
633 5
                        assert(r == hpk_done);
634 5
                }
635
636 437
                vtc_log(s->vl, 4, "settings->%s (%u): %d", buf, t, v);
637
        }
638
639 410
}
640
641
static void
642 6
parse_ping(const struct stream *s, struct frame *f)
643
{
644
        struct http *hp;
645 6
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
646 6
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
647 6
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
648 6
        if (f->size != 8)
649 0
                vtc_fatal(s->vl, "Size should be 8, but isn't (%d)", f->size);
650 6
        f->md.ping.ack = f->flags & ACK;
651 6
        memcpy(f->md.ping.data, f->data, 8);
652 6
        f->md.ping.data[8] = '\0';
653
654 6
        vtc_log(s->vl, 4, "ping->data: %s", f->md.ping.data);
655
656 6
}
657
658
static void
659 39
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 39
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
665 39
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
666 39
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
667
668 39
        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 39
        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 39
        stid = vbe32dec(f->data);
676 39
        err = vbe32dec(f->data + 4);
677 39
        f->md.goaway.err = err;
678 39
        f->md.goaway.stream = stid;
679
680 39
        if (err <= ERR_MAX)
681 39
                err_buf = h2_errs[err];
682
        else
683 0
                err_buf = "unknown";
684
685 39
        if (f->size > 8) {
686 2
                f->md.goaway.debug = malloc((f->size - 8) + 1L);
687 2
                AN(f->md.goaway.debug);
688 2
                f->md.goaway.debug[f->size - 8] = '\0';
689
690 2
                memcpy(f->md.goaway.debug, f->data + 8, f->size - 8);
691 2
        }
692
693 39
        vtc_log(s->vl, 3, "goaway->laststream: %d", stid);
694 39
        vtc_log(s->vl, 3, "goaway->err: %s (%d)", err_buf, err);
695 39
        if (f->md.goaway.debug)
696 2
                vtc_log(s->vl, 3, "goaway->debug: %s", f->md.goaway.debug);
697 39
}
698
699
static void
700 66
parse_winup(const struct stream *s, struct frame *f)
701
{
702
        struct http *hp;
703
        uint32_t size;
704 66
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
705 66
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
706 66
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);;
707
708 66
        if (f->size != 4)
709 0
                vtc_fatal(s->vl, "Size should be 4, but isn't (%d)", f->size);
710 66
        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 66
        size = vbe32dec(f->data);
715 66
        f->md.winup_size = size;
716
717 66
        vtc_log(s->vl, 3, "winup->size: %d", size);
718 66
}
719
720
/* read a frame and queue it in the relevant stream, wait if not present yet.
721
 */
722
static void *
723 209
receive_frame(void *priv)
724
{
725
        struct http *hp;
726
        char hdr[9];
727
        struct frame *f;
728
        struct stream *s;
729 209
        int expect_cont = 0;
730 209
        struct vsb *vsb = NULL;
731 209
        struct hpk_hdr *hdrs = NULL;
732
733 209
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
734
735 209
        PTOK(pthread_mutex_lock(&hp->mtx));
736 2147
        while (hp->h2) {
737
                /*no wanted frames? */
738 1938
                assert(hp->wf >= 0);
739 1938
                if (hp->wf == 0) {
740 1042
                        PTOK(pthread_cond_wait(&hp->cond, &hp->mtx));
741 1042
                        continue;
742
                }
743 896
                PTOK(pthread_mutex_unlock(&hp->mtx));
744
745 896
                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 896
                ALLOC_OBJ(f, FRAME_MAGIC);
755 896
                AN(f);
756 896
                readFrameHeader(f, hdr);
757
758 1792
                vtc_log(hp->vl, 3, "rx: stream: %d, type: %s (%d), "
759
                                "flags: 0x%02x, size: %d",
760 896
                                f->stid,
761 896
                                f->type < TYPE_MAX ? h2_types[f->type] : "?",
762 896
                                f->type, f->flags, f->size);
763 896
                explain_flags(f->flags, f->type, hp->vl);
764
765 896
                if (f->size) {
766 611
                        f->data = malloc(f->size + 1L);
767 611
                        AN(f->data);
768 611
                        f->data[f->size] = '\0';
769 611
                        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 611
                }
780
781
                /* is the corresponding stream waiting? */
782 896
                PTOK(pthread_mutex_lock(&hp->mtx));
783 896
                s = NULL;
784 1791
                while (!s) {
785 1037
                        VTAILQ_FOREACH(s, &hp->streams, list)
786 1037
                                if (s->id == f->stid)
787 895
                                        break;
788 895
                        if (!s)
789 0
                                PTOK(pthread_cond_wait(&hp->cond, &hp->mtx));
790 895
                        if (!hp->h2) {
791 0
                                clean_frame(&f);
792 0
                                PTOK(pthread_mutex_unlock(&hp->mtx));
793 0
                                return (NULL);
794
                        }
795
                }
796 896
                PTOK(pthread_mutex_unlock(&hp->mtx));
797
798 896
                AN(s);
799 910
                if (expect_cont &&
800 14
                    (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 896
                switch (f->type) {
806
                        case TYPE_DATA:
807 100
                                parse_data(s, f);
808 100
                                break;
809
                        case TYPE_PUSH_PROMISE:
810 1
                                hdrs = s->req;
811
                                /*FALLTHROUGH*/
812
                        case TYPE_HEADERS:
813 197
                                if (!hdrs) {
814 196
                                        if (hp->sfd)
815 46
                                                hdrs = s->req;
816
                                        else
817 150
                                                hdrs = s->resp;
818 196
                                }
819 197
                                clean_headers(hdrs);
820 197
                                hdrs[0].t = hpk_unset;
821 197
                                AZ(vsb);
822 197
                                vsb = VSB_new_auto();
823
                                /*FALLTHROUGH*/
824
                        case TYPE_CONTINUATION:
825 211
                                AN(hdrs);
826 211
                                expect_cont = s->id;
827 211
                                parse_hdr(s, f, vsb);
828 211
                                if (f->flags & END_HEADERS) {
829 197
                                        expect_cont = 0;
830 197
                                        AZ(VSB_finish(vsb));
831 197
                                        decode_hdr(hp, hdrs, vsb);
832 197
                                        VSB_destroy(&vsb);
833 197
                                        hdrs = NULL;
834 197
                                }
835 211
                                break;
836
                        case TYPE_PRIORITY:
837 9
                                parse_prio(s, f);
838 9
                                break;
839
                        case TYPE_RST_STREAM:
840 54
                                parse_rst(s, f);
841 54
                                break;
842
                        case TYPE_SETTINGS:
843 411
                                parse_settings(s, f);
844 411
                                break;
845
                        case TYPE_PING:
846 6
                                parse_ping(s, f);
847 6
                                break;
848
                        case TYPE_GOAWAY:
849 39
                                parse_goaway(s, f);
850 39
                                break;
851
                        case TYPE_WINDOW_UPDATE:
852 66
                                parse_winup(s, f);
853 66
                                break;
854
                        default:
855 0
                                WRONG("wrong frame type");
856 0
                }
857
858 896
                PTOK(pthread_mutex_lock(&hp->mtx));
859 896
                VTAILQ_INSERT_HEAD(&s->fq, f, list);
860 896
                if (s->wf) {
861 855
                        assert(hp->wf > 0);
862 855
                        hp->wf--;
863 855
                        s->wf = 0;
864 855
                        PTOK(pthread_cond_signal(&s->cond));
865 855
                }
866 896
                continue;
867
        }
868 209
        PTOK(pthread_mutex_unlock(&hp->mtx));
869 209
        if (vsb != NULL)
870 0
                VSB_destroy(&vsb);
871 209
        return (NULL);
872 209
}
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 198
find_header(const struct hpk_hdr *h, const char *k)
914
{
915 198
        AN(k);
916
917 198
        int kl = strlen(k);
918 1062
        while (h->t) {
919 1052
                if (kl == h->key.len  && !strncasecmp(h->key.ptr, k, kl))
920 188
                        return (h->value.ptr);
921 864
                h++;
922
        }
923 10
        return (NULL);
924 198
}
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 1516
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 1516
        struct frame *f = s->frame;
943
944 1516
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
945 1516
        CHECK_OBJ_NOTNULL(s->hp, HTTP_MAGIC);
946 1516
        AN(spec);
947 1516
        AN(buf);
948
949 1516
        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 1516
        if (!strcmp(spec, "ping.data")) {
958 6
                CHECK_LAST_FRAME(PING);
959 6
                return (f->md.ping.data);
960
        }
961 1510
        if (!strcmp(spec, "ping.ack")) {
962 4
                CHECK_LAST_FRAME(PING);
963 4
                snprintf(buf, 20, (f->flags & ACK) ? "true" : "false");
964 4
                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 1506
        if (!strcmp(spec, "winup.size")) {
972 1
                CHECK_LAST_FRAME(WINDOW_UPDATE);
973 1
                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 1505
        if (!strcmp(spec, "prio.stream")) {
987 2
                CHECK_LAST_FRAME(PRIORITY);
988 2
                RETURN_BUFFED(f->md.prio.stream);
989 0
        }
990 1503
        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 1503
        if (!strcmp(spec, "prio.weight")) {
996 2
                CHECK_LAST_FRAME(PRIORITY);
997 2
                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 1501
        if (!strcmp(spec, "rst.err")) {
1005 41
                CHECK_LAST_FRAME(RST_STREAM);
1006 41
                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 1460
        if (!strncmp(spec, "settings.", 9)) {
1033 220
                CHECK_LAST_FRAME(SETTINGS);
1034 220
                spec += 9;
1035 220
                if (!strcmp(spec, "ack")) {
1036 204
                        snprintf(buf, 20, (f->flags & ACK) ? "true" : "false");
1037 204
                        return (buf);
1038
                }
1039 16
                if (!strcmp(spec, "push")) {
1040 1
                        if (isnan(f->md.settings[SETTINGS_ENABLE_PUSH]))
1041 1
                                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 15
                if (!strcmp(spec, "hdrtbl"))     { RETURN_SETTINGS(1); }
1049 12
                if (!strcmp(spec, "maxstreams")) { RETURN_SETTINGS(3); }
1050 9
                if (!strcmp(spec, "winsize"))    { RETURN_SETTINGS(4); }
1051 6
                if (!strcmp(spec, "framesize"))  { RETURN_SETTINGS(5); }
1052 3
                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 1240
        if (!strcmp(spec, "push.id")) {
1060 1
                CHECK_LAST_FRAME(PUSH_PROMISE);
1061 1
                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 1239
        if (!strncmp(spec, "goaway.", 7)) {
1075 66
                spec += 7;
1076 66
                CHECK_LAST_FRAME(GOAWAY);
1077
1078 66
                if (!strcmp(spec, "err"))
1079 35
                        RETURN_BUFFED(f->md.goaway.err);
1080 31
                else if (!strcmp(spec, "laststream"))
1081 29
                        RETURN_BUFFED(f->md.goaway.stream);
1082 2
                else if (!strcmp(spec, "debug"))
1083 2
                        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 1173
        if (!strncmp(spec, "frame.", 6)) {
1104 8
                spec += 6;
1105 8
                if (!f)
1106 0
                        vtc_fatal(s->vl, "No frame received yet.");
1107 8
                if (!strcmp(spec, "data"))   { return (f->data); }
1108 7
                else if (!strcmp(spec, "type"))   { RETURN_BUFFED(f->type); }
1109 6
                else if (!strcmp(spec, "size"))   { RETURN_BUFFED(f->size); }
1110 2
                else if (!strcmp(spec, "stream")) { RETURN_BUFFED(f->stid); }
1111 1
                else if (!strcmp(spec, "padding")) {
1112 1
                        if (f->type != TYPE_DATA &&
1113 1
                                        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 1
                        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 1165
        if (!strcmp(spec, "stream.window")) {
1138 26
                snprintf(buf, 20, "%jd",
1139 13
                    (intmax_t)(s->id ? s->win_self : s->hp->h2_win_self->size));
1140 13
                return (buf);
1141
        }
1142 1152
        if (!strcmp(spec, "stream.peer_window")) {
1143 32
                snprintf(buf, 20, "%jd",
1144 16
                    (intmax_t)(s->id ? s->win_peer : s->hp->h2_win_peer->size));
1145 16
                return (buf);
1146
        }
1147 1136
        if (!strcmp(spec, "stream.weight")) {
1148 8
                if (s->id) {
1149 7
                        snprintf(buf, 20, "%d", s->weight);
1150 7
                        return (buf);
1151
                } else
1152 1
                        return (NULL);
1153
        }
1154 1128
        if (!strcmp(spec, "stream.dependency")) {
1155 16
                if (s->id) {
1156 15
                        snprintf(buf, 20, "%d", s->dependency);
1157 15
                        return (buf);
1158
                } else
1159 1
                        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 1112
        if (!strncmp(spec, "tbl.dec", 7) || !strncmp(spec, "tbl.enc", 7)) {
1181 123
                if (spec[4] == 'd')
1182 63
                        ctx = s->hp->decctx;
1183
                else
1184 60
                        ctx = s->hp->encctx;
1185 123
                spec += 7;
1186
1187 123
                if (1 == sscanf(spec, "[%u].key%n", &idx, &n) &&
1188 96
                                spec[n] == '\0') {
1189 48
                        h = HPK_GetHdr(ctx, idx + 61);
1190 48
                        return (h ? h->key.ptr : NULL);
1191
                }
1192 75
                else if (1 == sscanf(spec, "[%u].value%n", &idx, &n) &&
1193 48
                                spec[n] == '\0') {
1194 48
                        h = HPK_GetHdr(ctx, idx + 61);
1195 48
                        return (h ? h->value.ptr : NULL);
1196
                }
1197 27
                else if (!strcmp(spec, ".size"))
1198 26
                        RETURN_BUFFED(HPK_GetTblSize(ctx));
1199 1
                else if (!strcmp(spec, ".maxsize"))
1200 0
                        RETURN_BUFFED(HPK_GetTblMaxSize(ctx));
1201 1
                else if (!strcmp(spec, ".length"))
1202 1
                        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 989
        if (!strncmp(spec, "req.", 4) || !strncmp(spec, "resp.", 5)) {
1234 234
                if (spec[2] == 'q') {
1235 15
                        h = s->req;
1236 15
                        spec += 4;
1237 15
                } else {
1238 219
                        h = s->resp;
1239 219
                        spec += 5;
1240
                }
1241 234
                if (!strcmp(spec, "body"))
1242 17
                        return (s->body);
1243 217
                else if (!strcmp(spec, "bodylen"))
1244 19
                        RETURN_BUFFED(s->bodylen);
1245 198
                else if (!strcmp(spec, "status"))
1246 82
                        return (find_header(h, ":status"));
1247 116
                else if (!strcmp(spec, "url"))
1248 2
                        return (find_header(h, ":path"));
1249 114
                else if (!strcmp(spec, "method"))
1250 2
                        return (find_header(h, ":method"));
1251 112
                else if (!strcmp(spec, "authority"))
1252 1
                        return (find_header(h, ":authority"));
1253 111
                else if (!strcmp(spec, "scheme"))
1254 1
                        return (find_header(h, ":scheme"));
1255 110
                else if (!strncmp(spec, "http.", 5))
1256 110
                        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 57
cmd_sendhex(CMD_ARGS)
1276
{
1277
        struct http *hp;
1278
        struct stream *s;
1279
        struct vsb *vsb;
1280
1281 57
        (void)vl;
1282 57
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1283 57
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);
1284 57
        AN(av[1]);
1285 57
        AZ(av[2]);
1286 57
        vsb = vtc_hex_to_bin(hp->vl, av[1]);
1287 57
        assert(VSB_len(vsb) >= 0);
1288 57
        vtc_hexdump(hp->vl, 4, "sendhex", VSB_data(vsb), VSB_len(vsb));
1289 57
        PTOK(pthread_mutex_lock(&hp->mtx));
1290 57
        http_write(hp, 4, VSB_data(vsb), VSB_len(vsb), "sendhex");
1291 57
        PTOK(pthread_mutex_unlock(&hp->mtx));
1292 57
        VSB_destroy(&vsb);
1293 57
}
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 1291
cmd_tx11obj(CMD_ARGS)
1425
{
1426
        struct stream *s;
1427
        int i;
1428 1291
        int status_done = 1;
1429 1291
        int method_done = 1;
1430 1291
        int path_done = 1;
1431 1291
        int scheme_done = 1;
1432 1291
        long bodylen = 0;
1433
        ssize_t len;
1434 1291
        uint32_t stid = 0, pstid;
1435 1291
        uint32_t weight = 16;
1436 1291
        uint32_t exclusive = 0;
1437
        char *buf;
1438
        struct hpk_iter *iter;
1439
        struct frame f;
1440 1291
        char *body = NULL, *pad = NULL;
1441
        /*XXX: do we need a better api? yes we do */
1442
        struct hpk_hdr hdr;
1443 1291
        char *cmd_str = *av;
1444
        char *p;
1445
1446 1291
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1447 1291
        INIT_FRAME(f, CONTINUATION, 0, s->id, END_HEADERS);
1448 1291
        buf = malloc(BUF_SIZE);
1449 1291
        AN(buf);
1450
1451 1291
        if (!strcmp(cmd_str, "txreq")) {
1452 226
                ONLY_H2_CLIENT(s->hp, av);
1453 226
                f.type = TYPE_HEADERS;
1454 226
                f.flags |= END_STREAM;
1455 226
                method_done = 0;
1456 226
                path_done = 0;
1457 226
                scheme_done = 0;
1458 1291
        } else if (!strcmp(cmd_str, "txresp")) {
1459 47
                ONLY_H2_SERVER(s->hp, av);
1460 47
                f.type = TYPE_HEADERS;
1461 47
                f.flags |= END_STREAM;
1462 47
                status_done = 0;
1463 1065
        } else if (!strcmp(cmd_str, "txpush")) {
1464 1
                ONLY_H2_SERVER(s->hp, av);
1465 1
                f.type = TYPE_PUSH_PROMISE;
1466 1
                method_done = 0;
1467 1
                path_done = 0;
1468 1
                scheme_done = 0;
1469 1
        }
1470
1471 1291
        if (f.type == TYPE_PUSH_PROMISE) {
1472 1
                *buf = 0;
1473 1
                iter = HPK_NewIter(s->hp->encctx, buf + 4, BUF_SIZE - 4);
1474 1
        } else
1475 1290
                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 3743
        while (*++av) {
1480 2452
                memset(&hdr, 0, sizeof(hdr));
1481 2452
                hdr.t = hpk_not;
1482 2452
                if (AV_IS("-noadd")) {
1483 12
                        path_done = 1;
1484 12
                        status_done = 1;
1485 12
                        method_done = 1;
1486 12
                        scheme_done = 1;
1487 12
                }
1488 2440
                else if (AV_IS("-status") && CMD_IS("txresp")) {
1489 1
                        ENC(hdr, ":status", av[1]);
1490 1
                        av++;
1491 1
                        status_done = 1;
1492 1
                }
1493 2440
                else if (AV_IS("-url") &&
1494 62
                                (CMD_IS("txreq") || CMD_IS("txpush"))) {
1495 62
                        ENC(hdr, ":path", av[1]);
1496 62
                        av++;
1497 62
                        path_done = 1;
1498 62
                }
1499 2377
                else if ((AV_IS("-method") || AV_IS("-req")) &&
1500 37
                                (CMD_IS("txreq") || CMD_IS("txpush"))) {
1501 37
                        ENC(hdr, ":method", av[1]);
1502 37
                        av++;
1503 37
                        method_done = 1;
1504 37
                }
1505 2340
                else if (AV_IS("-scheme") &&
1506 7
                                (CMD_IS("txreq") || CMD_IS("txpush"))) {
1507 7
                        ENC(hdr, ":scheme", av[1]);
1508 7
                        av++;
1509 7
                        scheme_done = 1;
1510 7
                }
1511 2333
                else if (AV_IS("-hdr")) {
1512 1184
                        if (av[2] == NULL)
1513 0
                                vtc_fatal(vl, "-hdr takes two arguments in http2");
1514 1184
                        ENC(hdr, av[1], av[2]);
1515 1184
                        av += 2;
1516 1184
                }
1517 1149
                else if (AV_IS("-idxHdr")) {
1518 42
                        hdr.t = hpk_idx;
1519 42
                        STRTOU32_CHECK(hdr.i, av, p, vl, "-idxHdr", 0);
1520 42
                        assert(HPK_EncHdr(iter, &hdr) != hpk_err);
1521 42
                }
1522 1107
                else if (AV_IS("-litIdxHdr")) {
1523 16
                        av++;
1524 16
                        if      (AV_IS("inc"))   { hdr.t = hpk_inc;   }
1525 1
                        else if (AV_IS("not"))   { hdr.t = hpk_not;   }
1526 1
                        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 16
                        STRTOU32_CHECK(hdr.i, av, p, vl,
1532
                            "second -litidxHdr arg", 0);
1533
1534 16
                        hdr.key.ptr = NULL;
1535 16
                        hdr.key.len = 0;
1536 24
                        STR_ENC(av, value,   "third -litHdr");
1537 16
                        assert(HPK_EncHdr(iter, &hdr) != hpk_err);
1538 16
                }
1539 1091
                else if (AV_IS("-litHdr")) {
1540 7
                        av++;
1541 7
                        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 10
                        STR_ENC(av, key,   "second -litHdr");
1549 10
                        STR_ENC(av, value, "fourth -litHdr");
1550 7
                        assert(HPK_EncHdr(iter, &hdr) != hpk_err);
1551 7
                }
1552 1084
                else if (AV_IS("-nostrend")) {
1553 39
                        f.flags &= ~END_STREAM;
1554 39
                }
1555 1045
                else if (AV_IS("-nohdrend")) {
1556 1016
                        f.flags &= ~END_HEADERS;
1557 1016
                }
1558 29
                else if (AV_IS("-promised") && CMD_IS("txpush")) {
1559 1
                        STRTOU32_CHECK(pstid, av, p, vl, "-promised", 31);
1560 1
                        vbe32enc(buf, pstid);
1561 1
                }
1562 28
                else if (AV_IS("-pad") && !CMD_IS("txcont")) {
1563 6
                        AZ(pad);
1564 6
                        av++;
1565 6
                        AN(*av);
1566 6
                        pad = strdup(*av);
1567 6
                }
1568 22
                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 22
                else if (CMD_IS("txreq") || CMD_IS("txresp")) {
1574 22
                        if (AV_IS("-body")) {
1575 8
                                AZ(body);
1576 8
                                REPLACE(body, av[1]);
1577 8
                                AN(body);
1578 8
                                bodylen = strlen(body);
1579 8
                                f.flags &= ~END_STREAM;
1580 8
                                av++;
1581 8
                        }
1582 14
                        else if (AV_IS("-bodyfrom")) {
1583 2
                                AZ(body);
1584 2
                                body = VFIL_readfile(NULL, av[1], &len);
1585 2
                                AN(body);
1586 2
                                assert(len < INT_MAX);
1587 2
                                bodylen = len;
1588 2
                                f.flags &= ~END_STREAM;
1589 2
                                av++;
1590 2
                        }
1591 12
                        else if (AV_IS("-bodylen")) {
1592 4
                                AZ(body);
1593 4
                                body = synth_body(av[1], 0);
1594 4
                                bodylen = strlen(body);
1595 4
                                f.flags &= ~END_STREAM;
1596 4
                                av++;
1597 4
                        }
1598 8
                        else if (!strncmp(*av, "-gzip", 5)) {
1599 2
                                i = vtc_gzip_cmd(s->hp, av, &body, &bodylen);
1600 2
                                if (i == 0)
1601 0
                                        break;
1602 2
                                av += i;
1603 2
                                if (i > 1) {
1604 2
                                        ENC(hdr, ":content-encoding", "gzip");
1605 2
                                        f.flags &= ~END_STREAM;
1606 2
                                }
1607 2
                        }
1608 6
                        else if (AV_IS("-dep")) {
1609 3
                                STRTOU32_CHECK(stid, av, p, vl, "-dep", 0);
1610 3
                                f.flags |= PRIORITY;
1611 3
                        }
1612 3
                        else if (AV_IS("-ex")) {
1613 2
                                exclusive = 1U << 31;
1614 2
                                f.flags |= PRIORITY;
1615 2
                        }
1616 1
                        else if (AV_IS("-weight")) {
1617 1
                                STRTOU32_CHECK(weight, av, p, vl, "-weight", 8);
1618 1
                                f.flags |= PRIORITY;
1619 1
                        } else
1620 0
                                break;
1621 22
                } else
1622 0
                        break;
1623
        }
1624
#undef CMD_IS
1625
#undef AV_IS
1626 1291
        if (*av != NULL)
1627 0
                vtc_fatal(vl, "Unknown %s spec: %s\n", cmd_str, *av);
1628
1629 1291
        memset(&hdr, 0, sizeof(hdr));
1630 1291
        hdr.t = hpk_not;
1631
1632 1291
        if (!status_done) { ENC(hdr, ":status", "200"); }
1633 1291
        if (!path_done)   { ENC(hdr, ":path",   "/"); }
1634 1291
        if (!method_done) { ENC(hdr, ":method", "GET"); }
1635 1291
        if (!scheme_done) { ENC(hdr, ":scheme", "http"); }
1636
1637 1291
        f.size = gethpk_iterLen(iter);
1638 1291
        if (f.flags & PRIORITY) {
1639 4
                s->weight = weight & 0xff;
1640 4
                s->dependency = stid;
1641
1642 4
                assert(f.size + 5 < BUF_SIZE);
1643 4
                memmove(buf + 5, buf, f.size);
1644 4
                vbe32enc(buf, (stid | exclusive));
1645 4
                buf[4] = s->weight;
1646 4
                f.size += 5;
1647
1648 4
                vtc_log(vl, 4, "stream->dependency: %u", s->dependency);
1649 4
                vtc_log(vl, 4, "stream->weight: %u", s->weight);
1650 4
                if (exclusive)
1651 2
                        exclusive_stream_dependency(s);
1652 4
        }
1653 1291
        if (pad) {
1654 6
                if (strlen(pad) > 255)
1655 0
                        vtc_fatal(vl, "Padding is limited to 255 bytes");
1656 6
                f.flags |= PADDED;
1657 6
                assert(f.size + strlen(pad) < BUF_SIZE);
1658 6
                memmove(buf + 1, buf, f.size);
1659 6
                buf[0] = strlen(pad);
1660 6
                f.size += 1;
1661 6
                memcpy(buf + f.size, pad, strlen(pad));
1662 6
                f.size += strlen(pad);
1663 6
                free(pad);
1664 6
        }
1665 1291
        if (f.type == TYPE_PUSH_PROMISE)
1666 1
                f.size += 4;
1667 1291
        f.data = buf;
1668 1291
        HPK_FreeIter(iter);
1669 1291
        write_frame(s, &f, 1);
1670 1291
        free(buf);
1671
1672 1291
        if (!body)
1673 1275
                return;
1674
1675 16
        INIT_FRAME(f, DATA, bodylen, s->id, END_STREAM);
1676 16
        f.data = body;
1677
1678 16
        write_frame(s, &f, 1);
1679 16
        free(body);
1680 1291
}
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 320
cmd_txdata(CMD_ARGS)
1705
{
1706
        struct stream *s;
1707 320
        char *pad = NULL;
1708
        struct frame f;
1709 320
        char *body = NULL;
1710 320
        char *data = NULL;
1711
1712 320
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1713
1714 320
        INIT_FRAME(f, DATA, 0, s->id, END_STREAM);
1715
1716 937
        while (*++av) {
1717 617
                if (!strcmp(*av, "-data")) {
1718 25
                        AZ(body);
1719 25
                        av++;
1720 25
                        body = strdup(*av);
1721 617
                } else if (!strcmp(*av, "-datalen")) {
1722 35
                        AZ(body);
1723 35
                        av++;
1724 35
                        body = synth_body(*av, 0);
1725 592
                } else if (!strcmp(*av, "-pad")) {
1726 1
                        AZ(pad);
1727 1
                        av++;
1728 1
                        AN(*av);
1729 1
                        pad = strdup(*av);
1730 557
                } else if (!strcmp(*av, "-padlen")) {
1731 268
                        AZ(pad);
1732 268
                        av++;
1733 268
                        pad = synth_body(*av, 0);
1734 556
                } else if (!strcmp(*av, "-nostrend"))
1735 288
                        f.flags &= ~END_STREAM;
1736
                else
1737 0
                        break;
1738
        }
1739 320
        if (*av != NULL)
1740 0
                vtc_fatal(vl, "Unknown txdata spec: %s\n", *av);
1741
1742 320
        if (!body)
1743 260
                body = strdup("");
1744
1745 320
        if (pad) {
1746 269
                f.flags |= PADDED;
1747 269
                if (strlen(pad) > 255)
1748 0
                        vtc_fatal(vl, "Padding is limited to 255 bytes");
1749 269
                data = malloc( 1 + strlen(body) + strlen(pad));
1750 269
                AN(data);
1751 269
                *((uint8_t *)data) = strlen(pad);
1752 269
                f.size = 1;
1753 269
                memcpy(data + f.size, body, strlen(body));
1754 269
                f.size += strlen(body);
1755 269
                memcpy(data + f.size, pad, strlen(pad));
1756 269
                f.size += strlen(pad);
1757 269
                f.data = data;
1758 269
        } else {
1759 51
                f.size = strlen(body);
1760 51
                f.data = body;
1761
        }
1762 320
        write_frame(s, &f, 1);
1763 320
        free(body);
1764 320
        free(pad);
1765 320
        free(data);
1766 320
}
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 19
cmd_txrst(CMD_ARGS)
1780
{
1781
        struct stream *s;
1782
        char *p;
1783 19
        uint32_t err = 0;
1784
        struct frame f;
1785
1786 19
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1787
1788 19
        INIT_FRAME(f, RST_STREAM, 4, s->id, 0);
1789
1790 27
        while (*++av) {
1791 8
                if (!strcmp(*av, "-err")) {
1792 8
                        ++av;
1793 120
                        for (err = 0; h2_errs[err]; err++) {
1794 112
                                if (!strcmp(h2_errs[err], *av))
1795 0
                                        break;
1796 112
                        }
1797
1798 8
                        if (h2_errs[err])
1799 0
                                continue;
1800
1801 8
                        STRTOU32(err, *av, p, vl, "-err");
1802 8
                } else
1803 0
                        break;
1804
        }
1805 19
        if (*av != NULL)
1806 0
                vtc_fatal(vl, "Unknown txrst spec: %s\n", *av);
1807
1808 19
        err = htonl(err);
1809 19
        f.data = (void *)&err;
1810 19
        write_frame(s, &f, 1);
1811 19
}
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 13
cmd_txprio(CMD_ARGS)
1830
{
1831
        struct stream *s;
1832
        char *p;
1833 13
        uint32_t stid = 0;
1834
        struct frame f;
1835 13
        uint32_t weight = 0;
1836 13
        uint32_t exclusive = 0;
1837
        uint8_t buf[5];
1838
1839 13
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1840
1841 13
        INIT_FRAME(f, PRIORITY, 5, s->id, 0);
1842 13
        f.data = (void *)buf;
1843
1844 31
        while (*++av) {
1845 18
                if (!strcmp(*av, "-stream")) {
1846 9
                        STRTOU32_CHECK(stid, av, p, vl, "-stream", 0);
1847 18
                } else if (!strcmp(*av, "-ex")) {
1848 1
                        exclusive = 1U << 31;
1849 9
                } else if (!strcmp(*av, "-weight")) {
1850 8
                        STRTOU32_CHECK(weight, av, p, vl, "-weight", 8);
1851 8
                } else
1852 0
                        break;
1853
        }
1854 13
        if (*av != NULL)
1855 0
                vtc_fatal(vl, "Unknown txprio spec: %s\n", *av);
1856 13
        s->weight = weight & 0xff;
1857 13
        s->dependency = stid;
1858
1859 13
        if (exclusive)
1860 1
                exclusive_stream_dependency(s);
1861
1862 13
        vbe32enc(buf, (stid | exclusive));
1863 13
        buf[4] = s->weight;
1864 13
        write_frame(s, &f, 1);
1865 13
}
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
 * \-0xHH[HH] INT
1901
 *      tx arbitraty settings with tag xx
1902
 *
1903
 * \-ack
1904
 *      set the ack bit
1905
 */
1906
static void
1907 408
cmd_txsettings(CMD_ARGS)
1908
{
1909
        struct stream *s, *s2;
1910
        struct http *hp;
1911
        char *p, *e;
1912
        unsigned long u;
1913 408
        uint32_t val = 0;
1914
        struct frame f;
1915
        //TODO dynamic alloc
1916
        char buf[512];
1917 408
        char *cursor = buf;
1918
1919 408
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1920 408
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);
1921
1922 408
        memset(buf, 0, 512);
1923 408
        INIT_FRAME(f, SETTINGS, 0, s->id, 0);
1924 408
        f.data = buf;
1925
1926 408
        PTOK(pthread_mutex_lock(&hp->mtx));
1927 629
        while (*++av) {
1928 221
                if (!strcmp(*av, "-push")) {
1929 1
                        ++av;
1930 1
                        vbe16enc(cursor, 0x2);
1931 1
                        cursor += sizeof(uint16_t);
1932 1
                        if (!strcmp(*av, "false"))
1933 0
                                vbe32enc(cursor, 0);
1934 1
                        else if (!strcmp(*av, "true"))
1935 1
                                vbe32enc(cursor, 1);
1936
                        else
1937 0
                                vtc_fatal(vl, "Push parameter is either "
1938 0
                                    "\"true\" or \"false\", not %s", *av);
1939 1
                        cursor += sizeof(uint32_t);
1940 1
                        f.size += 6;
1941 1
                }
1942 220
                else if (!strcmp(*av, "-hdrtbl")) {
1943 5
                        PUT_KV(av, vl, hdrtbl, val, 0x1);
1944 5
                        assert(HPK_ResizeTbl(s->hp->decctx, val) != hpk_err);
1945 5
                }
1946 215
                else if (!strcmp(*av, "-maxstreams"))
1947 1
                        PUT_KV(av, vl, maxstreams, val, 0x3);
1948 214
                else if (!strcmp(*av, "-winsize"))      {
1949 8
                        PUT_KV(av, vl, winsize, val, 0x4);
1950 17
                        VTAILQ_FOREACH(s2, &hp->streams, list)
1951 9
                                s2->win_self += (val - hp->h2_win_self->init);
1952 8
                        hp->h2_win_self->init = val;
1953 8
                }
1954 206
                else if (!strcmp(*av, "-framesize"))
1955 3
                        PUT_KV(av, vl, framesize, val, 0x5);
1956 203
                else if (!strcmp(*av, "-hdrsize"))
1957 1
                        PUT_KV(av, vl, hdrsize, val, 0x6);
1958 202
                else if (!strncmp(*av, "-0x", 3)) {
1959 0
                        p = *av + 3;
1960 0
                        errno = 0;
1961 0
                        u = strtoul(p, &e, 16);
1962 0
                        if (*p == '\0' || *e != '\0' || u > 0xffff || errno != 0)
1963 0
                                vtc_fatal(vl, "Invalid settings tag %s", p);
1964 0
                        assert(u <= 0xffff);
1965 0
                        PUT_KV(av, vl, hdrtbl, val, (uint16_t)u);
1966 0
                }
1967 202
                else if (!strcmp(*av, "-ack"))
1968 202
                        f.flags |= 1;
1969
                else
1970 0
                        break;
1971
        }
1972 408
        if (*av != NULL)
1973 0
                vtc_fatal(vl, "Unknown txsettings spec: %s\n", *av);
1974
1975 408
        AN(s->hp);
1976 408
        write_frame(s, &f, 0);
1977 408
        PTOK(pthread_mutex_unlock(&hp->mtx));
1978 408
}
1979
1980
/* SECTION: stream.spec.ping_txping txping
1981
 *
1982
 * Send PING frame.
1983
 *
1984
 * \-data STRING
1985
 *      specify the payload of the frame, with STRING being an 8-char string.
1986
 *
1987
 * \-ack
1988
 *      set the ACK flag.
1989
 */
1990
static void
1991 7
cmd_txping(CMD_ARGS)
1992
{
1993
        struct stream *s;
1994
        struct frame f;
1995
        char buf[8];
1996
1997 7
        memset(buf, 0, 8);
1998 7
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
1999 7
        INIT_FRAME(f, PING, 8, s->id, 0);
2000
2001 14
        while (*++av) {
2002 7
                if (!strcmp(*av, "-data")) {
2003 5
                        av++;
2004 5
                        if (f.data)
2005 0
                                vtc_fatal(vl, "this frame already has data");
2006 5
                        if (strlen(*av) != 8)
2007 0
                                vtc_fatal(vl, "data must be a 8-char string, found  (%s)", *av);
2008 5
                        f.data = *av;
2009 7
                } else if (!strcmp(*av, "-ack"))
2010 2
                        f.flags |= 1;
2011
                else
2012 0
                        break;
2013
        }
2014 7
        if (*av != NULL)
2015 0
                vtc_fatal(vl, "Unknown txping spec: %s\n", *av);
2016 7
        if (!f.data)
2017 2
                f.data = buf;
2018 7
        write_frame(s, &f, 1);
2019 7
}
2020
2021
/*
2022
 * SECTION: stream.spec.goaway_txgoaway txgoaway
2023
 *
2024
 * Possible options include:
2025
 *
2026
 * \-err STRING|INT
2027
 *      set the error code to explain the termination. The second argument
2028
 *      can be a integer or the string version of the error code as found
2029
 *      in rfc7540#7.
2030
 *
2031
 * \-laststream INT
2032
 *      the id of the "highest-numbered stream identifier for which the
2033
 *      sender of the GOAWAY frame might have taken some action on or might
2034
 *      yet take action on".
2035
 *
2036
 * \-debug
2037
 *      specify the debug data, if any to append to the frame.
2038
 */
2039
static void
2040 5
cmd_txgoaway(CMD_ARGS)
2041
{
2042
        struct stream *s;
2043
        char *p;
2044 5
        uint32_t err = 0;
2045 5
        uint32_t ls = 0;
2046
        struct frame f;
2047
2048 5
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2049
2050 5
        INIT_FRAME(f, GOAWAY, 8, s->id, 0);
2051
2052 14
        while (*++av) {
2053 9
                if (!strcmp(*av, "-err")) {
2054 5
                        ++av;
2055 75
                        for (err = 0; h2_errs[err]; err++)
2056 70
                                if (!strcmp(h2_errs[err], *av))
2057 0
                                        break;
2058
2059 5
                        if (h2_errs[err])
2060 0
                                continue;
2061
2062 5
                        STRTOU32(err, *av, p, vl, "-err");
2063 9
                } else if (!strcmp(*av, "-laststream")) {
2064 2
                        STRTOU32_CHECK(ls, av, p, vl, "-laststream", 31);
2065 4
                } else if (!strcmp(*av, "-debug")) {
2066 2
                        ++av;
2067 2
                        if (f.data)
2068 0
                                vtc_fatal(vl, "this frame already has debug data");
2069 2
                        f.size = 8 + strlen(*av);
2070 2
                        f.data = malloc(f.size);
2071 2
                        AN(f.data);
2072 2
                        memcpy(f.data + 8, *av, f.size - 8);
2073 2
                } else
2074 0
                        break;
2075
        }
2076 5
        if (*av != NULL)
2077 0
                vtc_fatal(vl, "Unknown txgoaway spec: %s\n", *av);
2078
2079 5
        if (!f.data) {
2080 3
                f.data = malloc(8);
2081 3
                AN(f.data);
2082 3
        }
2083 5
        vbe32enc(f.data, ls);
2084 5
        vbe32enc(f.data + 4, err);
2085 5
        write_frame(s, &f, 1);
2086 5
        free(f.data);
2087 5
}
2088
2089
/* SECTION: stream.spec.winup_txwinup txwinup
2090
 *
2091
 * Transmit a WINDOW_UPDATE frame, increasing the amount of credit of the
2092
 * connection (from stream 0) or of the stream (any other stream).
2093
 *
2094
 * \-size INT
2095
 *      give INT credits to the peer.
2096
 */
2097
static void
2098 24
cmd_txwinup(CMD_ARGS)
2099
{
2100
        struct http *hp;
2101
        struct stream *s;
2102
        char *p;
2103
        struct frame f;
2104
        char buf[8];
2105 24
        uint32_t size = 0;
2106
2107 24
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2108 24
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);
2109 24
        memset(buf, 0, 8);
2110
2111 24
        AN(av[1]);
2112 24
        AN(av[2]);
2113
2114 24
        INIT_FRAME(f, WINDOW_UPDATE, 4, s->id, 0);
2115 24
        f.data = buf;
2116
2117 48
        while (*++av)
2118 24
                if (!strcmp(*av, "-size")) {
2119 24
                        STRTOU32_CHECK(size, av, p, vl, "-size", 0);
2120 24
                } else
2121 0
                        break;
2122 24
        if (*av != NULL)
2123 0
                vtc_fatal(vl, "Unknown txwinup spec: %s\n", *av);
2124
2125 24
        PTOK(pthread_mutex_lock(&hp->mtx));
2126 24
        if (s->id == 0)
2127 10
                hp->h2_win_self->size += size;
2128 24
        s->win_self += size;
2129 24
        PTOK(pthread_mutex_unlock(&hp->mtx));
2130
2131 24
        size = htonl(size);
2132 24
        f.data = (void *)&size;
2133 24
        write_frame(s, &f, 1);
2134 24
}
2135
2136
static struct frame *
2137 856
rxstuff(struct stream *s)
2138
{
2139
        struct frame *f;
2140
2141 856
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
2142
2143 856
        PTOK(pthread_mutex_lock(&s->hp->mtx));
2144 856
        if (VTAILQ_EMPTY(&s->fq)) {
2145 855
                assert(s->hp->wf >= 0);
2146 855
                s->hp->wf++;
2147 855
                s->wf = 1;
2148 855
                PTOK(pthread_cond_signal(&s->hp->cond));
2149 855
                PTOK(pthread_cond_wait(&s->cond, &s->hp->mtx));
2150 855
        }
2151 856
        if (VTAILQ_EMPTY(&s->fq)) {
2152 0
                PTOK(pthread_mutex_unlock(&s->hp->mtx));
2153 0
                return (NULL);
2154
        }
2155 856
        clean_frame(&s->frame);
2156 856
        f = VTAILQ_LAST(&s->fq, fq_head);
2157 856
        CHECK_OBJ_NOTNULL(f, FRAME_MAGIC);
2158 856
        VTAILQ_REMOVE(&s->fq, f, list);
2159 856
        PTOK(pthread_mutex_unlock(&s->hp->mtx));
2160 856
        return (f);
2161 856
}
2162
2163
#define CHKFRAME(rt, wt, rcv, func) \
2164
        do { \
2165
        if (rt != wt) \
2166
                vtc_fatal(vl, "Frame #%d for %s was of type %s (%d) " \
2167
                    "instead of %s (%d)", \
2168
                    rcv, func, \
2169
                    rt < TYPE_MAX ? h2_types[rt] : "?", rt, \
2170
                    wt < TYPE_MAX ? h2_types[wt] : "?", wt); \
2171
        } while (0);
2172
2173
/* SECTION: stream.spec.data_11 rxhdrs
2174
 *
2175
 * ``rxhdrs`` will expect one HEADER frame, then, depending on the arguments,
2176
 * zero or more CONTINUATION frame.
2177
 *
2178
 * \-all
2179
 *      Keep waiting for CONTINUATION frames until END_HEADERS flag is seen.
2180
 *
2181
 * \-some INT
2182
 *      Retrieve INT - 1 CONTINUATION frames after the HEADER frame.
2183
 *
2184
 */
2185
static void
2186 14
cmd_rxhdrs(CMD_ARGS)
2187
{
2188
        struct stream *s;
2189 14
        struct frame *f = NULL;
2190
        char *p;
2191 14
        int loop = 0;
2192 14
        unsigned long int times = 1;
2193 14
        unsigned rcv = 0;
2194 14
        enum h2_type_e expect = TYPE_HEADERS;
2195
2196 14
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2197
2198 16
        while (*++av) {
2199 2
                if (!strcmp(*av, "-some")) {
2200 1
                        STRTOU32_CHECK(times, av, p, vl, "-some", 0);
2201 1
                        if (!times)
2202 0
                                vtc_fatal(vl, "-some argument must be more"
2203 0
                                               "than 0 (found \"%s\")\n", *av);
2204 2
                } else if (!strcmp(*av, "-all"))
2205 1
                        loop = 1;
2206
                else
2207 0
                        break;
2208
        }
2209 14
        if (*av != NULL)
2210 0
                vtc_fatal(vl, "Unknown rxhdrs spec: %s\n", *av);
2211
2212 14
        do {
2213 17
                replace_frame(&f, rxstuff(s));
2214 17
                if (f == NULL)
2215 0
                        break;
2216 17
                rcv++;
2217 17
                CHKFRAME(f->type, expect, rcv, "rxhdrs");
2218 17
                expect = TYPE_CONTINUATION;
2219 17
        } while (rcv < times || (loop && !(f->flags & END_HEADERS)));
2220 14
        replace_frame(&s->frame, f);
2221 14
}
2222
2223
static void
2224 5
cmd_rxcont(CMD_ARGS)
2225
{
2226
        struct stream *s;
2227 5
        struct frame *f = NULL;
2228
        char *p;
2229 5
        int loop = 0;
2230 5
        unsigned long int times = 1;
2231 5
        unsigned rcv = 0;
2232
2233 5
        (void)av;
2234 5
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2235
2236 5
        while (*++av)
2237 0
                if (!strcmp(*av, "-some")) {
2238 0
                        STRTOU32(times, *av, p, vl, "-some");
2239 0
                        if (!times)
2240 0
                                vtc_fatal(vl, "-some argument must be more"
2241 0
                                               "than 0 (found \"%s\")\n", *av);
2242 0
                } else if (!strcmp(*av, "-all"))
2243 0
                        loop = 1;
2244
                else
2245 0
                        break;
2246 5
        if (*av != NULL)
2247 0
                vtc_fatal(vl, "Unknown rxcont spec: %s\n", *av);
2248
2249 5
        do {
2250 5
                replace_frame(&f, rxstuff(s));
2251 5
                if (f == NULL)
2252 0
                        break;
2253 5
                rcv++;
2254 5
                CHKFRAME(f->type, TYPE_CONTINUATION, rcv, "rxcont");
2255 5
        } while (rcv < times || (loop && !(f->flags & END_HEADERS)));
2256 5
        replace_frame(&s->frame, f);
2257 5
}
2258
2259
2260
/* SECTION: stream.spec.data_13 rxdata
2261
 *
2262
 * Receiving data is done using the ``rxdata`` keywords and will retrieve one
2263
 * DATA frame, if you wish to receive more, you can use these two convenience
2264
 * arguments:
2265
 *
2266
 * \-all
2267
 *      keep waiting for DATA frame until one sets the END_STREAM flag
2268
 *
2269
 * \-some INT
2270
 *      retrieve INT DATA frames.
2271
 *
2272
 */
2273
static void
2274 15
cmd_rxdata(CMD_ARGS)
2275
{
2276
        struct stream *s;
2277 15
        struct frame *f = NULL;
2278
        char *p;
2279 15
        int loop = 0;
2280 15
        unsigned long int times = 1;
2281 15
        unsigned rcv = 0;
2282
2283 15
        (void)av;
2284 15
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2285
2286 16
        while (*++av)
2287 1
                if (!strcmp(*av, "-some")) {
2288 1
                        av++;
2289 1
                        STRTOU32(times, *av, p, vl, "-some");
2290 1
                        if (!times)
2291 0
                                vtc_fatal(vl, "-some argument must be more"
2292 0
                                               "than 0 (found \"%s\")\n", *av);
2293 1
                } else if (!strcmp(*av, "-all"))
2294 0
                        loop = 1;
2295
                else
2296 0
                        break;
2297 15
        if (*av != NULL)
2298 0
                vtc_fatal(vl, "Unknown rxdata spec: %s\n", *av);
2299
2300 15
        do {
2301 17
                replace_frame(&f, rxstuff(s));
2302 17
                if (f == NULL)
2303 0
                        break;
2304 17
                rcv++;
2305 17
                CHKFRAME(f->type, TYPE_DATA, rcv, "rxhdata");
2306 17
        } while (rcv < times || (loop && !(f->flags & END_STREAM)));
2307 15
        replace_frame(&s->frame, f);
2308 15
}
2309
2310
/* SECTION: stream.spec.data_10 rxreq, rxresp
2311
 *
2312
 * These are two convenience functions to receive headers and body of an
2313
 * incoming request or response. The only difference is that rxreq can only be
2314
 * by a server, and rxresp by a client.
2315
 *
2316
 */
2317
2318
#define cmd_rxreq       cmd_rxmsg
2319
#define cmd_rxresp      cmd_rxmsg
2320
2321
static void
2322 182
cmd_rxmsg(CMD_ARGS)
2323
{
2324
        struct stream *s;
2325 182
        struct frame *f = NULL;
2326
        int end_stream;
2327 182
        int rcv = 0;
2328
2329 182
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2330
2331 182
        if (!strcmp(av[0], "rxreq"))
2332 46
                ONLY_H2_SERVER(s->hp, av);
2333
        else
2334 136
                ONLY_H2_CLIENT(s->hp, av);
2335
2336 182
        do {
2337 184
                replace_frame(&f, rxstuff(s));
2338 184
                CHECK_OBJ_ORNULL(f, FRAME_MAGIC);
2339 184
                if (f == NULL)
2340 0
                        return;
2341 184
        } while (f->type == TYPE_WINDOW_UPDATE);
2342
2343 182
        rcv++;
2344 182
        CHKFRAME(f->type, TYPE_HEADERS, rcv, *av);
2345
2346 182
        end_stream = f->flags & END_STREAM;
2347
2348 188
        while (!(f->flags & END_HEADERS)) {
2349 6
                replace_frame(&f, rxstuff(s));
2350 6
                CHECK_OBJ_ORNULL(f, FRAME_MAGIC);
2351 6
                if (f == NULL)
2352 0
                        return;
2353 6
                rcv++;
2354 6
                CHKFRAME(f->type, TYPE_CONTINUATION, rcv, *av);
2355
        }
2356
2357 265
        while (!end_stream) {
2358 83
                replace_frame(&f, rxstuff(s));
2359 83
                CHECK_OBJ_ORNULL(f, FRAME_MAGIC);
2360 83
                if (f == NULL)
2361 0
                        break;
2362 83
                rcv++;
2363 83
                CHKFRAME(f->type, TYPE_DATA, rcv, *av);
2364 83
                end_stream = f->flags & END_STREAM;
2365
        }
2366 182
        replace_frame(&s->frame, f);
2367 182
}
2368
2369
/* SECTION: stream.spec.data_12 rxpush
2370
 *
2371
 * This works like ``rxhdrs``, expecting a PUSH frame and then zero or more
2372
 * CONTINUATION frames.
2373
 *
2374
 * \-all
2375
 *      Keep waiting for CONTINUATION frames until END_HEADERS flag is seen.
2376
 *
2377
 * \-some INT
2378
 *      Retrieve INT - 1 CONTINUATION frames after the PUSH frame.
2379
 *
2380
 */
2381
static void
2382 1
cmd_rxpush(CMD_ARGS)
2383
{
2384
        struct stream *s;
2385 1
        struct frame *f = NULL;
2386
        char *p;
2387 1
        int loop = 0;
2388 1
        unsigned long int times = 1;
2389 1
        unsigned rcv = 0;
2390 1
        enum h2_type_e expect = TYPE_PUSH_PROMISE;
2391
2392 1
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2393
2394 1
        while (*++av) {
2395 0
                if (!strcmp(*av, "-some")) {
2396 0
                        STRTOU32_CHECK(times, av, p, vl, "-some", 0);
2397 0
                        if (!times)
2398 0
                                vtc_fatal(vl, "-some argument must be more"
2399 0
                                               "than 0 (found \"%s\")\n", *av);
2400 0
                } else if (!strcmp(*av, "-all")) {
2401 0
                        loop = 1;
2402 0
                } else
2403 0
                        break;
2404
        }
2405 1
        if (*av != NULL)
2406 0
                vtc_fatal(vl, "Unknown rxpush spec: %s\n", *av);
2407
2408 1
        do {
2409 1
                f = rxstuff(s);
2410 1
                if (!f)
2411 0
                        return;
2412 1
                rcv++;
2413 1
                CHKFRAME(f->type, expect, rcv, "rxpush");
2414 1
                expect = TYPE_CONTINUATION;
2415 1
        } while (rcv < times || (loop && !(f->flags & END_HEADERS)));
2416 1
        s->frame = f;
2417 1
}
2418
2419
/* SECTION: stream.spec.winup_rxwinup rxwinup
2420
 *
2421
 * Receive a WINDOW_UPDATE frame.
2422
 */
2423
static void
2424 25
cmd_rxwinup(CMD_ARGS)
2425
{
2426
        struct stream *s;
2427
        struct frame *f;
2428
2429 25
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2430 25
        s->frame = rxstuff(s);
2431 25
        CAST_OBJ_NOTNULL(f, s->frame, FRAME_MAGIC);
2432 25
        CHKFRAME(f->type, TYPE_WINDOW_UPDATE, 0, *av);
2433 25
        if (s->id == 0)
2434 6
                s->hp->h2_win_peer->size += s->frame->md.winup_size;
2435 25
        s->win_peer += s->frame->md.winup_size;
2436 25
}
2437
2438
/* SECTION: stream.spec.settings_rxsettings rxsettings
2439
 *
2440
 * Receive a SETTINGS frame.
2441
 */
2442
static void
2443 411
cmd_rxsettings(CMD_ARGS)
2444
{
2445
        struct stream *s, *s2;
2446 411
        uint32_t val = 0;
2447
        struct http *hp;
2448
        struct frame *f;
2449
2450 411
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2451 411
        CAST_OBJ_NOTNULL(hp, s->hp, HTTP_MAGIC);
2452 411
        s->frame = rxstuff(s);
2453 411
        CAST_OBJ_NOTNULL(f, s->frame, FRAME_MAGIC);
2454 411
        CHKFRAME(f->type, TYPE_SETTINGS, 0, *av);
2455 411
        if (! isnan(f->md.settings[SETTINGS_INITIAL_WINDOW_SIZE])) {
2456 135
                val = (uint32_t)f->md.settings[SETTINGS_INITIAL_WINDOW_SIZE];
2457 277
                VTAILQ_FOREACH(s2, &hp->streams, list)
2458 142
                        s2->win_peer += (val - hp->h2_win_peer->init);
2459 135
                hp->h2_win_peer->init = val;
2460 135
        }
2461 411
}
2462
2463
#define RXFUNC(lctype, upctype) \
2464
        static void \
2465
        cmd_rx ## lctype(CMD_ARGS) { \
2466
                struct stream *s; \
2467
                (void)av; \
2468
                CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC); \
2469
                s->frame = rxstuff(s); \
2470
                if (s->frame != NULL && s->frame->type != TYPE_ ## upctype) \
2471
                        vtc_fatal(vl, \
2472
                            "Wrong frame type %s (%d) wanted %s", \
2473
                            s->frame->type < TYPE_MAX ? \
2474
                            h2_types[s->frame->type] : "?", \
2475
                            s->frame->type, #upctype); \
2476
        }
2477
2478
/* SECTION: stream.spec.prio_rxprio rxprio
2479
 *
2480
 * Receive a PRIORITY frame.
2481
 */
2482 9
RXFUNC(prio,    PRIORITY)
2483
2484
/* SECTION: stream.spec.reset_rxrst rxrst
2485
 *
2486
 * Receive a RST_STREAM frame.
2487
 */
2488 54
RXFUNC(rst,     RST_STREAM)
2489
2490
/* SECTION: stream.spec.ping_rxping rxping
2491
 *
2492
 * Receive a PING frame.
2493
 */
2494 6
RXFUNC(ping,    PING)
2495
2496
/* SECTION: stream.spec.goaway_rxgoaway rxgoaway
2497
 *
2498
 * Receive a GOAWAY frame.
2499
 */
2500 39
RXFUNC(goaway,  GOAWAY)
2501
2502
/* SECTION: stream.spec.frame_rxframe
2503
 *
2504
 * Receive a frame, any frame.
2505
 */
2506
static void
2507 0
cmd_rxframe(CMD_ARGS)
2508
{
2509
        struct stream *s;
2510
2511 0
        (void)vl;
2512 0
        (void)av;
2513 0
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2514 0
        if (rxstuff(s) == NULL)
2515 0
                vtc_fatal(s->vl, "No frame received");
2516 0
}
2517
2518
static void
2519 759
cmd_expect(CMD_ARGS)
2520
{
2521
        struct http *hp;
2522
        struct stream *s;
2523
        const char *lhs;
2524
        char *cmp;
2525
        const char *rhs;
2526
        char buf[20];
2527
2528 759
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2529 759
        hp = s->hp;
2530 759
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
2531
2532 759
        AZ(strcmp(av[0], "expect"));
2533 759
        av++;
2534
2535 759
        AN(av[0]);
2536 759
        AN(av[1]);
2537 759
        AN(av[2]);
2538 759
        AZ(av[3]);
2539 759
        PTOK(pthread_mutex_lock(&s->hp->mtx));
2540 759
        lhs = cmd_var_resolve(s, av[0], buf);
2541 759
        cmp = av[1];
2542 759
        rhs = cmd_var_resolve(s, av[2], buf);
2543 759
        vtc_expect(vl, av[0], lhs, cmp, av[2], rhs);
2544 759
        PTOK(pthread_mutex_unlock(&s->hp->mtx));
2545 759
}
2546
2547
/* SECTION: stream.spec.gunzip gunzip
2548
 *
2549
 * Same as the ``gunzip`` command for HTTP/1.
2550
 */
2551
static void
2552 2
cmd_gunzip(CMD_ARGS)
2553
{
2554
        struct http *hp;
2555
        struct stream *s;
2556
2557 2
        (void)av;
2558 2
        (void)vl;
2559
2560 2
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2561 2
        hp = s->hp;
2562 2
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
2563
2564 2
        vtc_gunzip(s->hp, s->body, &s->bodylen);
2565 2
}
2566
2567
/* SECTION: stream.spec.write_body
2568
 *
2569
 * write_body STRING
2570
 *      Same as the ``write_body`` command for HTTP/1.
2571
 */
2572
static void
2573 2
cmd_write_body(CMD_ARGS)
2574
{
2575
        struct stream *s;
2576
2577 2
        (void)vl;
2578 2
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2579 2
        AN(av[0]);
2580 2
        AN(av[1]);
2581 2
        AZ(av[2]);
2582 2
        AZ(strcmp(av[0], "write_body"));
2583 2
        if (VFIL_writefile(NULL, av[1], s->body, s->bodylen) != 0)
2584 0
                vtc_fatal(s->vl, "failed to write body: %s (%d)",
2585 0
                    strerror(errno), errno);
2586 2
}
2587
2588
/* SECTION: stream.spec Specification
2589
 *
2590
 * The specification of a stream follows the exact same rules as one for a
2591
 * client or a server.
2592
 */
2593
static const struct cmds stream_cmds[] = {
2594
#define CMD_STREAM(n) { #n, cmd_##n },
2595
        /* spec */
2596
        CMD_STREAM(expect)
2597
        CMD_STREAM(gunzip)
2598
        CMD_STREAM(rxcont)
2599
        CMD_STREAM(rxdata)
2600
        CMD_STREAM(rxframe)
2601
        CMD_STREAM(rxgoaway)
2602
        CMD_STREAM(rxhdrs)
2603
        CMD_STREAM(rxping)
2604
        CMD_STREAM(rxprio)
2605
        CMD_STREAM(rxpush)
2606
        CMD_STREAM(rxreq)
2607
        CMD_STREAM(rxresp)
2608
        CMD_STREAM(rxrst)
2609
        CMD_STREAM(rxsettings)
2610
        CMD_STREAM(rxwinup)
2611
        CMD_STREAM(sendhex)
2612
        CMD_STREAM(txcont)
2613
        CMD_STREAM(txdata)
2614
        CMD_STREAM(txgoaway)
2615
        CMD_STREAM(txping)
2616
        CMD_STREAM(txprio)
2617
        CMD_STREAM(txpush)
2618
        CMD_STREAM(txreq)
2619
        CMD_STREAM(txresp)
2620
        CMD_STREAM(txrst)
2621
        CMD_STREAM(txsettings)
2622
        CMD_STREAM(txwinup)
2623
        CMD_STREAM(write_body)
2624
        { NULL, NULL }
2625
#undef CMD_STREAM
2626
};
2627
2628
static void *
2629 575
stream_thread(void *priv)
2630
{
2631
        struct stream *s;
2632
2633 575
        CAST_OBJ_NOTNULL(s, priv, STREAM_MAGIC);
2634 575
        parse_string(s->vl, s, s->spec);
2635 575
        vtc_log(s->vl, 2, "Ending stream %u", s->id);
2636 575
        return (NULL);
2637
}
2638
/**********************************************************************
2639
 * Allocate and initialize a stream
2640
 */
2641
2642
static struct stream *
2643 493
stream_new(const char *name, struct http *h)
2644
{
2645
        char *p, buf[20];
2646
        struct stream *s;
2647
2648 493
        if (!strcmp("next", name)) {
2649 20
                if (h->last_stream > 0)
2650 13
                        bprintf(buf, "%d", h->last_stream + 2);
2651
                else
2652 7
                        bprintf(buf, "%d", 1);
2653 20
                name = buf;
2654 20
        }
2655
2656 493
        ALLOC_OBJ(s, STREAM_MAGIC);
2657 493
        AN(s);
2658 493
        PTOK(pthread_cond_init(&s->cond, NULL));
2659 493
        REPLACE(s->name, name);
2660 493
        AN(s->name);
2661 493
        VTAILQ_INIT(&s->fq);
2662 493
        s->win_self = h->h2_win_self->init;
2663 493
        s->win_peer = h->h2_win_peer->init;
2664 493
        s->vl = vtc_logopen("%s.%s", h->sess->name, name);
2665 493
        vtc_log_set_cmd(s->vl, stream_cmds);
2666
2667 493
        s->weight = 16;
2668 493
        s->dependency = 0;
2669
2670 493
        STRTOU32(s->id, name, p, s->vl, "stream");
2671 493
        if (s->id & (1U << 31))
2672 0
                vtc_fatal(s->vl, "Stream id must be a 31-bits integer "
2673 0
                    "(found %s)", name);
2674
2675 493
        CHECK_OBJ_NOTNULL(h, HTTP_MAGIC);
2676 493
        s->hp = h;
2677 493
        h->last_stream = s->id;
2678
2679
        //bprintf(s->connect, "%s", "${v1_sock}");
2680 493
        PTOK(pthread_mutex_lock(&h->mtx));
2681 493
        VTAILQ_INSERT_HEAD(&h->streams, s, list);
2682 493
        PTOK(pthread_mutex_unlock(&h->mtx));
2683 493
        return (s);
2684
}
2685
2686
/**********************************************************************
2687
 * Clean up stream
2688
 */
2689
2690
static void
2691 493
stream_delete(struct stream *s)
2692
{
2693
        struct frame *f, *f2;
2694
2695 493
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
2696
2697 526
        VTAILQ_FOREACH_SAFE(f, &s->fq, list, f2) {
2698 33
                VTAILQ_REMOVE(&s->fq, f, list);
2699 33
                clean_frame(&f);
2700 33
        }
2701 493
        vtc_logclose(s->vl);
2702 493
        clean_headers(s->req);
2703 493
        clean_headers(s->resp);
2704 493
        AZ(s->frame);
2705 493
        free(s->body);
2706 493
        free(s->spec);
2707 493
        free(s->name);
2708 493
        FREE_OBJ(s);
2709 493
}
2710
2711
/**********************************************************************
2712
 * Start the stream thread
2713
 */
2714
2715
static void
2716 575
stream_start(struct stream *s)
2717
{
2718 575
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
2719 575
        vtc_log(s->hp->vl, 2, "Starting stream %s (%p)", s->name, s);
2720 575
        PTOK(pthread_create(&s->tp, NULL, stream_thread, s));
2721 575
        s->running = 1;
2722 575
}
2723
2724
/**********************************************************************
2725
 * Wait for stream thread to stop
2726
 */
2727
static void
2728 575
stream_wait(struct stream *s)
2729
{
2730
        void *res;
2731
        struct frame *f, *f2;
2732
2733 575
        CHECK_OBJ_NOTNULL(s, STREAM_MAGIC);
2734 575
        vtc_log(s->hp->vl, 2, "Waiting for stream %u", s->id);
2735 575
        PTOK(pthread_join(s->tp, &res));
2736 575
        if (res != NULL)
2737 0
                vtc_fatal(s->hp->vl, "Stream %u returned \"%s\"", s->id,
2738 0
                    (char *)res);
2739
2740 581
        VTAILQ_FOREACH_SAFE(f, &s->fq, list, f2) {
2741 6
                VTAILQ_REMOVE(&s->fq, f, list);
2742 6
                clean_frame(&f);
2743 6
        }
2744 575
        clean_frame(&s->frame);
2745 575
        s->tp = 0;
2746 575
        s->running = 0;
2747 575
}
2748
2749
/**********************************************************************
2750
 * Run the stream thread
2751
 */
2752
2753
static void
2754 527
stream_run(struct stream *s)
2755
{
2756 527
        stream_start(s);
2757 527
        stream_wait(s);
2758 527
}
2759
2760
2761
2762
/* SECTION: client-server.spec.stream
2763
 *
2764
 * stream
2765
 *      HTTP/2 introduces the concept of streams, and these come with
2766
 *      their own specification, and as it's quite big, have been moved
2767
 *      to their own chapter.
2768
 *
2769
 * SECTION: stream stream
2770
 *
2771
 * (note: this section is at the top-level for easier navigation, but
2772
 * it's part of the client/server specification)
2773
 *
2774
 * Streams map roughly to a request in HTTP/2, a request is sent on
2775
 * stream N, the response too, then the stream is discarded. The main
2776
 * exception is the first stream, 0, that serves as coordinator.
2777
 *
2778
 * Stream syntax follow the client/server one::
2779
 *
2780
 *      stream ID [SPEC] [ACTION]
2781
 *
2782
 * ID is the HTTP/2 stream number, while SPEC describes what will be
2783
 * done in that stream. If ID has the value ``next``, the actual stream
2784
 * number is computed based on the last one.
2785
 *
2786
 * Note that, when parsing a stream action, if the entity isn't operating
2787
 * in HTTP/2 mode, these spec is ran before::
2788
 *
2789
 *      txpri/rxpri # client/server
2790
 *      stream 0 {
2791
 *          txsettings
2792
 *          rxsettings
2793
 *          txsettings -ack
2794
 *          rxsettings
2795
 *          expect settings.ack == true
2796
 *      } -run
2797
 *
2798
 * And HTTP/2 mode is then activated before parsing the specification.
2799
 *
2800
 * SECTION: stream.actions Actions
2801
 *
2802
 * \-start
2803
 *      Run the specification in a thread, giving back control immediately.
2804
 *
2805
 * \-wait
2806
 *      Wait for the started thread to finish running the spec.
2807
 *
2808
 * \-run
2809
 *      equivalent to calling ``-start`` then ``-wait``.
2810
 */
2811
2812
void
2813 596
cmd_stream(CMD_ARGS)
2814
{
2815
        struct stream *s;
2816
        struct http *h;
2817
2818 596
        (void)vl;
2819 596
        CAST_OBJ_NOTNULL(h, priv, HTTP_MAGIC);
2820
2821 596
        AZ(strcmp(av[0], "stream"));
2822 596
        av++;
2823
2824 1169
        VTAILQ_FOREACH(s, &h->streams, list)
2825 676
                if (!strcmp(s->name, av[0]))
2826 103
                        break;
2827 596
        if (s == NULL)
2828 493
                s = stream_new(av[0], h);
2829 596
        av++;
2830
2831 1767
        for (; *av != NULL; av++) {
2832 1171
                if (vtc_error)
2833 0
                        break;
2834
2835 1171
                if (!strcmp(*av, "-wait")) {
2836 22
                        stream_wait(s);
2837 22
                        continue;
2838
                }
2839
2840
                /* Don't muck about with a running client */
2841 1149
                if (s->running)
2842 0
                        stream_wait(s);
2843
2844 1149
                if (!strcmp(*av, "-start")) {
2845 48
                        stream_start(s);
2846 48
                        continue;
2847
                }
2848 1101
                if (!strcmp(*av, "-run")) {
2849 527
                        stream_run(s);
2850 527
                        continue;
2851
                }
2852 574
                if (**av == '-')
2853 0
                        vtc_fatal(vl, "Unknown stream argument: %s", *av);
2854 574
                REPLACE(s->spec, *av);
2855 574
        }
2856 596
}
2857
2858
void
2859 1
b64_settings(const struct http *hp, const char *s)
2860
{
2861
        uint16_t i;
2862
        uint64_t v, vv;
2863
        const char *buf;
2864
        int shift;
2865
2866 3
        while (*s) {
2867 2
                v = 0;
2868 18
                for (shift = 42; shift >= 0; shift -= 6) {
2869 16
                        if (*s >= 'A' && *s <= 'Z')
2870 13
                                vv = (*s - 'A');
2871 3
                        else if (*s >= 'a' && *s <= 'z')
2872 1
                                vv = (*s - 'a') + 26;
2873 2
                        else if (*s >= '0' && *s <= '9')
2874 0
                                vv = (*s - '0') + 52;
2875 2
                        else if (*s == '-')
2876 0
                                vv = 62;
2877 2
                        else if (*s == '_')
2878 2
                                vv = 63;
2879
                        else
2880 0
                                vtc_fatal(hp->vl,
2881
                                    "Bad \"HTTP2-Settings\" header");
2882 16
                        v |= vv << shift;
2883 16
                        s++;
2884 16
                }
2885 2
                i = v >> 32;
2886 2
                v &= 0xffff;
2887
2888 2
                if (i <= SETTINGS_MAX)
2889 2
                        buf = h2_settings[i];
2890
                else
2891 0
                        buf = "unknown";
2892
2893 2
                if (v == 1) {
2894 0
                        if (hp->sfd)
2895 0
                                assert(HPK_ResizeTbl(hp->encctx, v) != hpk_err);
2896
                        else
2897 0
                                assert(HPK_ResizeTbl(hp->decctx, v) != hpk_err);
2898 0
                }
2899
2900 4
                vtc_log(hp->vl, 4, "Upgrade: %s (%d): %ju",
2901 2
                    buf, i, (intmax_t)v);
2902
        }
2903 1
}
2904
2905
void
2906 209
start_h2(struct http *hp)
2907
{
2908 209
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
2909 209
        PTOK(pthread_mutex_init(&hp->mtx, NULL));
2910 209
        PTOK(pthread_cond_init(&hp->cond, NULL));
2911 209
        VTAILQ_INIT(&hp->streams);
2912 209
        hp->h2_win_self->init = 0xffff;
2913 209
        hp->h2_win_self->size = 0xffff;
2914 209
        hp->h2_win_peer->init = 0xffff;
2915 209
        hp->h2_win_peer->size = 0xffff;
2916 209
        hp->h2 = 1;
2917
2918 209
        hp->decctx = HPK_NewCtx(4096);
2919 209
        hp->encctx = HPK_NewCtx(4096);
2920 209
        PTOK(pthread_create(&hp->tp, NULL, receive_frame, hp));
2921 209
}
2922
2923
void
2924 209
stop_h2(struct http *hp)
2925
{
2926
        struct stream *s, *s2;
2927
2928 209
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
2929 701
        VTAILQ_FOREACH_SAFE(s, &hp->streams, list, s2) {
2930 492
                if (s->running)
2931 26
                        stream_wait(s);
2932 492
                PTOK(pthread_mutex_lock(&hp->mtx));
2933 492
                VTAILQ_REMOVE(&hp->streams, s, list);
2934 492
                PTOK(pthread_mutex_unlock(&hp->mtx));
2935 492
                stream_delete(s);
2936 492
        }
2937
2938 209
        PTOK(pthread_mutex_lock(&hp->mtx));
2939 209
        hp->h2 = 0;
2940 209
        PTOK(pthread_cond_signal(&hp->cond));
2941 209
        PTOK(pthread_mutex_unlock(&hp->mtx));
2942 209
        PTOK(pthread_join(hp->tp, NULL));
2943
2944 209
        HPK_FreeCtx(hp->decctx);
2945 209
        HPK_FreeCtx(hp->encctx);
2946
2947 209
        PTOK(pthread_mutex_destroy(&hp->mtx));
2948 209
        PTOK(pthread_cond_destroy(&hp->cond));
2949 209
}