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