vinyl-cache/bin/vinyltest/vtest2/src/vtc_http.c
0
/*-
1
 * Copyright (c) 2008-2019 Varnish Software AS
2
 * All rights reserved.
3
 *
4
 * Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
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/socket.h>
33
34
#include <math.h>
35
#include <poll.h>
36
#include <stdio.h>
37
#include <stdlib.h>
38
#include <unistd.h>
39
#include <string.h>
40
41
#include "vtc.h"
42
#include "vtc_http.h"
43
44
#include "vct.h"
45
#include "vfil.h"
46
#include "vnum.h"
47
#include "vrnd.h"
48
#include "vtcp.h"
49
#include "vtim.h"
50
#include "hpack.h"
51
52
extern const struct cmds http_cmds[];
53
54
/* SECTION: client-server client/server
55
 *
56
 * Client and server threads are fake HTTP entities used to test your Varnish
57
 * and VCL. They take any number of arguments, and the one that are not
58
 * recognized, assuming they don't start with '-', are treated as
59
 * specifications, laying out the actions to undertake::
60
 *
61
 *         client cNAME [...]
62
 *         server sNAME [...]
63
 *
64
 * Clients and server are identified by a string that's the first argument,
65
 * clients' names start with 'c' and servers' names start with 's'.
66
 *
67
 * As the client and server commands share a good deal of arguments and
68
 * specification actions, they are grouped in this single section, specific
69
 * items will be explicitly marked as such.
70
 *
71
 * SECTION: client-server.macros Macros and automatic behaviour
72
 *
73
 * To make things easier in the general case, clients will connect by default
74
 * to a Varnish server called v1. To connect to a different Varnish server, use
75
 * '-connect ${vNAME_sock}'.
76
 *
77
 * The -vcl+backend switch of the ``varnish`` command will add all the declared
78
 * servers as backends. Be careful though, servers will by default listen to
79
 * the 127.0.0.1 IP and will pick a random port, and publish 3 macros:
80
 * sNAME_addr, sNAME_port and sNAME_sock, but only once they are started. For
81
 * 'varnish -vcl+backend' to create the vcl with the correct values, the server
82
 * must be started first.
83
 *
84
 * SECTION: client-server.args Arguments
85
 *
86
 * \-start
87
 *        Start the thread in background, processing the last given
88
 *        specification.
89
 *
90
 * \-wait
91
 *        Block until the thread finishes.
92
 *
93
 * \-run (client only)
94
 *        Equivalent to "-start -wait".
95
 *
96
 * \-repeat NUMBER
97
 *        Instead of processing the specification only once, do it NUMBER times.
98
 *
99
 * \-keepalive
100
 *        For repeat, do not open new connections but rather run all
101
 *        iterations in the same connection
102
 *
103
 * \-break (server only)
104
 *        Stop the server.
105
 *
106
 * \-listen STRING (server only)
107
 *        Dictate the listening socket for the server. STRING is of the form
108
 *        "IP PORT", or "/PATH/TO/SOCKET" for a Unix domain socket. In the
109
 *        latter case, the path must begin with '/', and the server must be
110
 *        able to create it.
111
 *
112
 * \-connect STRING (client only)
113
 *        Indicate the server to connect to. STRING is also of the form
114
 *        "IP PORT", or "/PATH/TO/SOCKET". As with "server -listen", a
115
 *        Unix domain socket is recognized when STRING begins with a '/'.
116
 *
117
 * \-dispatch (server only, s0 only)
118
 *        Normally, to keep things simple, server threads only handle one
119
 *        connection at a time, but the -dispatch switch allows to accept
120
 *        any number of connection and handle them following the given spec.
121
 *
122
 *        However, -dispatch is only allowed for the server name "s0".
123
 *
124
 * \-proxy1 STRING (client only)
125
 *        Use the PROXY protocol version 1 for this connection. STRING
126
 *        is of the form "CLIENTIP:PORT SERVERIP:PORT".
127
 *
128
 * \-proxy2 STRING (client only)
129
 *        Use the PROXY protocol version 2 for this connection. STRING
130
 *        is of the form "CLIENTIP:PORT SERVERIP:PORT [TLV [TLV ... ]]".
131
 *
132
 *        TLV is in the form name=val
133
 *
134
 *        name: 0xID or alpn, authority, crc32c, noop, unique_id, netns
135
 *        val: 0x... or string
136
 *
137
 *        ssl is currently not implemented (can be sent as hex)
138
 *
139
 * SECTION: client-server.spec Specification
140
 *
141
 * It's a string, either double-quoted "like this", but most of the time
142
 * enclosed in curly brackets, allowing multilining. Write a command per line in
143
 * it, empty line are ignored, and long line can be wrapped by using a
144
 * backslash. For example::
145
 *
146
 *     client c1 {
147
 *         txreq -url /foo \
148
 *               -hdr "bar: baz"
149
 *
150
 *         rxresp
151
 *     } -run
152
 */
153
154
#define ONLY_CLIENT(hp, av)                                             \
155
        do {                                                            \
156
                if (hp->h2)                                             \
157
                        vtc_fatal(hp->vl,                               \
158
                            "\"%s\" only possible before H/2 upgrade",  \
159
                                        av[0]);                         \
160
                if (hp->sfd != NULL)                                    \
161
                        vtc_fatal(hp->vl,                               \
162
                            "\"%s\" only possible in client", av[0]);   \
163
        } while (0)
164
165
#define ONLY_SERVER(hp, av)                                             \
166
        do {                                                            \
167
                if (hp->h2)                                             \
168
                        vtc_fatal(hp->vl,                               \
169
                            "\"%s\" only possible before H/2 upgrade",  \
170
                                        av[0]);                         \
171
                if (hp->sfd == NULL)                                    \
172
                        vtc_fatal(hp->vl,                               \
173
                            "\"%s\" only possible in server", av[0]);   \
174
        } while (0)
175
176
177
/* XXX: we may want to vary this */
178
static const char * const nl = "\r\n";
179
180
/**********************************************************************
181
 * Generate a synthetic body
182
 */
183
184
char *
185 2860
synth_body(const char *len, int rnd)
186
{
187
        int i, j, k, l;
188
        char *b;
189
190
191 2860
        AN(len);
192 2860
        i = strtoul(len, NULL, 0);
193 2860
        assert(i > 0);
194 2860
        b = malloc(i + 1L);
195 2860
        AN(b);
196 2860
        l = k = '!';
197 171151110
        for (j = 0; j < i; j++) {
198 171148250
                if ((j % 64) == 63) {
199 2672550
                        b[j] = '\n';
200 2672550
                        k++;
201 2672550
                        if (k == '~')
202 28470
                                k = '!';
203 2672550
                        l = k;
204 171148250
                } else if (rnd) {
205 85155
                        b[j] = (VRND_RandomTestable() % 95) + ' ';
206 85155
                } else {
207 168390545
                        b[j] = (char)l;
208 168390545
                        if (++l == '~')
209 1802915
                                l = '!';
210
                }
211 171148250
        }
212 2860
        b[i - 1] = '\n';
213 2860
        b[i] = '\0';
214 2860
        return (b);
215
}
216
217
/**********************************************************************
218
 * Finish and write the vsb to the fd
219
 */
220
221
static void
222 40610
http_write(const struct http *hp, int lvl, const char *pfx)
223
{
224
225 40610
        AZ(VSB_finish(hp->vsb));
226 40610
        vtc_dump(hp->vl, lvl, pfx, VSB_data(hp->vsb), VSB_len(hp->vsb));
227 40610
        if (VSB_tofile(hp->vsb, *hp->sess->fd))
228 50
                vtc_log(hp->vl, hp->fatal, "Write failed: %s",
229 25
                    strerror(errno));
230 40610
}
231
232
/**********************************************************************
233
 * find header
234
 */
235
236
static char *
237 76487
http_find_header(char * const *hh, const char *hdr)
238
{
239
        int n, l;
240
        char *r;
241
242 76487
        l = strlen(hdr);
243
244 555826
        for (n = 3; hh[n] != NULL; n++) {
245 512882
                if (strncasecmp(hdr, hh[n], l) || hh[n][l] != ':')
246 479339
                        continue;
247 67134
                for (r = hh[n] + l + 1; vct_issp(*r); r++)
248 33591
                        continue;
249 33543
                return (r);
250
        }
251 42944
        return (NULL);
252 76487
}
253
254
/**********************************************************************
255
 * count header
256
 */
257
258
static int
259 29892
http_count_header(char * const *hh, const char *hdr)
260
{
261 29892
        int n, l, r = 0;
262
263 29892
        l = strlen(hdr);
264
265 239272
        for (n = 3; hh[n] != NULL; n++) {
266 209380
                if (strncasecmp(hdr, hh[n], l) || hh[n][l] != ':')
267 194286
                        continue;
268 15094
                r++;
269 15094
        }
270 29892
        return (r);
271
}
272
273
/* SECTION: client-server.spec.expect
274
 *
275
 * expect STRING1 OP STRING2
276
 *         Test if "STRING1 OP STRING2" is true, and if not, fails the test.
277
 *         OP can be ==, <, <=, >, >= when STRING1 and STRING2 represent numbers
278
 *         in which case it's an order operator. If STRING1 and STRING2 are
279
 *         meant as strings OP is a matching operator, either == (exact match)
280
 *         or ~ (regex match).
281
 *
282
 *         varnishtest will first try to resolve STRING1 and STRING2 by looking
283
 *         if they have special meanings, in which case, the resolved value is
284
 *         use for the test. Note that this value can be a string representing a
285
 *         number, allowing for tests such as::
286
 *
287
 *                 expect req.http.x-num > 2
288
 *
289
 *         Here's the list of recognized strings, most should be obvious as they
290
 *         either match VCL logic, or the txreq/txresp options:
291
 *
292
 *         - remote.ip
293
 *         - remote.port
294
 *         - remote.path
295
 *         - req.method
296
 *         - req.url
297
 *         - req.proto
298
 *         - resp.proto
299
 *         - resp.status
300
 *         - resp.reason
301
 *         - resp.chunklen
302
 *         - req.bodylen
303
 *         - req.body
304
 *         - resp.bodylen
305
 *         - resp.body
306
 *         - req.http.NAME
307
 *         - resp.http.NAME
308
 */
309
310
static const char *
311 73185
cmd_var_resolve(struct http *hp, char *spec)
312
{
313
        char **hh, *hdr;
314 73185
        if (!strcmp(spec, "remote.ip"))
315 20
                return (hp->rem_ip);
316 73165
        if (!strcmp(spec, "remote.port"))
317 20
                return (hp->rem_port);
318 73145
        if (!strcmp(spec, "remote.path"))
319 20
                return (hp->rem_path);
320 73125
        if (!strcmp(spec, "req.method"))
321 285
                return (hp->req[0]);
322 72840
        if (!strcmp(spec, "req.url"))
323 3285
                return (hp->req[1]);
324 69555
        if (!strcmp(spec, "req.proto"))
325 225
                return (hp->req[2]);
326 69330
        if (!strcmp(spec, "resp.proto"))
327 170
                return (hp->resp[0]);
328 69160
        if (!strcmp(spec, "resp.status"))
329 8824
                return (hp->resp[1]);
330 60336
        if (!strcmp(spec, "resp.reason"))
331 1149
                return (hp->resp[2]);
332 59187
        if (!strcmp(spec, "resp.chunklen"))
333 0
                return (hp->chunklen);
334 59187
        if (!strcmp(spec, "req.bodylen"))
335 165
                return (hp->bodylen);
336 59022
        if (!strcmp(spec, "req.body"))
337 195
                return (hp->body != NULL ? hp->body : spec);
338 58827
        if (!strcmp(spec, "resp.bodylen"))
339 3550
                return (hp->bodylen);
340 55277
        if (!strcmp(spec, "resp.body"))
341 1180
                return (hp->body != NULL ? hp->body : spec);
342 54097
        if (!strncmp(spec, "req.http.", 9)) {
343 2185
                hh = hp->req;
344 2185
                hdr = spec + 9;
345 54097
        } else if (!strncmp(spec, "resp.http.", 10)) {
346 16210
                hh = hp->resp;
347 16210
                hdr = spec + 10;
348 51912
        } else if (!strcmp(spec, "h2.state")) {
349 36
                if (hp->h2)
350 16
                        return ("true");
351
                else
352 20
                        return ("false");
353
        } else
354 35666
                return (spec);
355 18395
        hdr = http_find_header(hh, hdr);
356 18395
        return (hdr);
357 73185
}
358
359
static void
360 36596
cmd_http_expect(CMD_ARGS)
361
{
362
        struct http *hp;
363
        const char *lhs;
364
        char *cmp;
365
        const char *rhs;
366
367 36596
        (void)vl;
368 36596
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
369 36596
        AZ(strcmp(av[0], "expect"));
370
        av++;
371 36596
372 36596
        AN(av[0]);
373 36596
        AN(av[1]);
374 36596
        AN(av[2]);
375 36596
        AZ(av[3]);
376 36596
        lhs = cmd_var_resolve(hp, av[0]);
377 36596
        cmp = av[1];
378
        rhs = cmd_var_resolve(hp, av[2]);
379 36596
380 36596
        vtc_expect(vl, av[0], lhs, cmp, av[2], rhs);
381
}
382
383
/* SECTION: client-server.spec.expect_pattern
384
 *
385
 * expect_pattern
386
 *
387
 * Expect as the http body the test pattern generated by chunkedlen ('0'..'7'
388
 * repeating).
389
 */
390 20
static void
391
cmd_http_expect_pattern(CMD_ARGS)
392
{
393
        char *p;
394 20
        struct http *hp;
395
        char t = '0';
396 20
397 20
        (void)vl;
398 20
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
399 20
        AZ(strcmp(av[0], "expect_pattern"));
400 1310740
        av++;
401 1310720
        AZ(av[0]);
402 0
        for (p = hp->body; *p != '\0'; p++) {
403
                if (*p != t)
404 0
                        vtc_fatal(hp->vl,
405 1310720
                            "EXPECT PATTERN FAIL @%zd should 0x%02x is 0x%02x",
406 1310720
                            (ssize_t) (p - hp->body), t, *p);
407 1310720
                t += 1;
408 20
                t &= ~0x08;
409 20
        }
410
        vtc_log(hp->vl, 4, "EXPECT PATTERN SUCCESS");
411
}
412
413
/**********************************************************************
414
 * Split a HTTP protocol header
415
 */
416 29892
417
static void
418
http_splitheader(struct http *hp, int req)
419
{
420
        char *p, *q, **hh;
421
        int n;
422 29892
        char buf[20];
423 29892
424 12447
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
425 12447
        if (req) {
426 12447
                memset(hp->req, 0, sizeof hp->req);
427 17445
                hh = hp->req;
428 17445
        } else {
429
                memset(hp->resp, 0, sizeof hp->resp);
430
                hh = hp->resp;
431 29892
        }
432 29892
433 29892
        n = 0;
434 405
        p = hp->rx_b;
435 405
        if (*p == '\0') {
436
                vtc_log(hp->vl, 4, "No headers");
437
                return;
438
        }
439 29492
440 5
        /* REQ/PROTO */
441 29487
        while (vct_islws(*p))
442 204296
                p++;
443 174809
        hh[n++] = p;
444 29487
        while (!vct_islws(*p))
445 29487
                p++;
446
        AZ(vct_iscrlf(p, hp->rx_e));
447
        *p++ = '\0';
448 29487
449 0
        /* URL/STATUS */
450 29487
        while (vct_issp(*p))            /* XXX: H space only */
451 29487
                p++;
452 142810
        AZ(vct_iscrlf(p, hp->rx_e));
453 113323
        hh[n++] = p;
454 29487
        while (!vct_islws(*p))
455 5
                p++;
456 5
        if (vct_iscrlf(p, hp->rx_e)) {
457 5
                hh[n++] = NULL;
458 5
                q = p;
459 5
                p = vct_skipcrlf(p, hp->rx_e);
460 29482
                *q = '\0';
461
        } else {
462 29482
                *p++ = '\0';
463 0
                /* PROTO/MSG */
464 29482
                while (vct_issp(*p))            /* XXX: H space only */
465 226871
                        p++;
466 197389
                hh[n++] = p;
467 29482
                while (!vct_iscrlf(p, hp->rx_e))
468 29482
                        p++;
469 29482
                q = p;
470
                p = vct_skipcrlf(p, hp->rx_e);
471 29487
                *q = '\0';
472
        }
473 238843
        assert(n == 3);
474 238843
475 238843
        while (*p != '\0') {
476 29487
                assert(n < MAX_HDR);
477 209356
                if (vct_iscrlf(p, hp->rx_e))
478 4496567
                        break;
479 4287211
                hh[n++] = p++;
480 209356
                while (*p != '\0' && !vct_iscrlf(p, hp->rx_e))
481 0
                        p++;
482
                if (*p == '\0') {
483 209356
                        break;
484 209356
                }
485 209356
                q = p;
486
                p = vct_skipcrlf(p, hp->rx_e);
487 29487
                *q = '\0';
488 29487
        }
489
        p = vct_skipcrlf(p, hp->rx_e);
490 327329
        assert(*p == '\0');
491 297842
492 297842
        for (n = 0; n < 3 || hh[n] != NULL; n++) {
493 297842
                bprintf(buf, "http[%2d] ", n);
494 29892
                vtc_dump(hp->vl, 4, buf, hh[n], -1);
495
        }
496
}
497
498
499
/**********************************************************************
500
 * Receive another character
501
 */
502 6981704
503
static int
504
http_rxchar(struct http *hp, int n, int eof)
505
{
506
        int i;
507 13990826
        struct pollfd pfd[1];
508 7009657
509 7009657
        while (n > 0) {
510 7009657
                pfd[0].fd = *hp->sess->fd;
511 7009657
                pfd[0].events = POLLIN;
512 7009657
                pfd[0].revents = 0;
513 0
                i = poll(pfd, 1, (int)(hp->timeout * 1000));
514 7009657
                if (i < 0 && errno == EINTR)
515 0
                        continue;
516
                if (i == 0) {
517 0
                        vtc_log(hp->vl, hp->fatal,
518 0
                            "HTTP rx timeout (fd:%d %.3fs)",
519
                            *hp->sess->fd, hp->timeout);
520 7009657
                        continue;
521 0
                }
522
                if (i < 0) {
523 0
                        vtc_log(hp->vl, hp->fatal,
524 0
                            "HTTP rx failed (fd:%d poll: %s)",
525
                            *hp->sess->fd, strerror(errno));
526 7009657
                        continue;
527 7009657
                }
528 7009657
                assert(i > 0);
529 7009657
                assert(hp->rx_p + n < hp->rx_e);
530 0
                i = read(*hp->sess->fd, hp->rx_p, n);
531
                if (!(pfd[0].revents & POLLIN))
532 0
                        vtc_log(hp->vl, 4,
533 7009657
                            "HTTP rx poll (fd:%d revents: %x n=%d, i=%d)",
534 435
                            *hp->sess->fd, pfd[0].revents, n, i);
535 7009222
                if (i == 0 && eof)
536 200
                        return (i);
537
                if (i == 0) {
538 100
                        vtc_log(hp->vl, hp->fatal,
539 100
                            "HTTP rx EOF (fd:%d read: %s) %d",
540
                            *hp->sess->fd, strerror(errno), n);
541 7009122
                        return (-1);
542 0
                }
543
                if (i < 0) {
544 0
                        vtc_log(hp->vl, hp->fatal,
545 0
                            "HTTP rx failed (fd:%d read: %s)",
546
                            *hp->sess->fd, strerror(errno));
547 7009122
                        return (-1);
548 7009122
                }
549 7009122
                hp->rx_p += i;
550
                *hp->rx_p = '\0';
551 6981169
                n -= i;
552 6981704
        }
553
        return (1);
554
}
555 10553
556
static int
557
http_rxchunk(struct http *hp)
558
{
559
        char *q, *old;
560 10553
        int i;
561 10553
562 57344
        old = hp->rx_p;
563 51
        do {
564 57293
                if (http_rxchar(hp, 1, 0) < 0)
565 10502
                        return (-1);
566 10502
        } while (hp->rx_p[-1] != '\n');
567 10502
        vtc_dump(hp->vl, 4, "len", old, -1);
568 10502
        i = strtoul(old, &q, 16);
569 0
        bprintf(hp->chunklen, "%d", i);
570 0
        if ((q == old) || (q == hp->rx_p) || (*q != '\0' && !vct_islws(*q))) {
571 0
                vtc_log(hp->vl, hp->fatal, "Chunklen fail (%02x @ %td)",
572
                    (*q & 0xff), q - old);
573 10502
                return (-1);
574 10502
        }
575 10502
        assert(*q == '\0' || vct_islws(*q));
576 9080
        hp->rx_p = old;
577 9
        if (i > 0) {
578 9071
                if (http_rxchar(hp, i, 0) < 0)
579 9071
                        return (-1);
580 10493
                vtc_dump(hp->vl, 4, "chunk", old, i);
581 10493
        }
582 0
        old = hp->rx_p;
583 10493
        if (http_rxchar(hp, 2, 0) < 0)
584 0
                return (-1);
585 0
        if (!vct_iscrlf(old, hp->rx_e)) {
586
                vtc_log(hp->vl, hp->fatal, "Chunklen without CRLF");
587 10493
                return (-1);
588 10493
        }
589 10493
        hp->rx_p = old;
590 10553
        *hp->rx_p = '\0';
591
        return (i);
592
}
593
594
/**********************************************************************
595
 * Swallow a HTTP message body
596
 *
597
 * max: 0 is all
598
 */
599 29031
600
static void
601
http_swallow_body(struct http *hp, char * const *hh, int body, int max)
602
{
603
        const char *p, *q;
604 29031
        int i, l, ll;
605
606 29031
        l = hp->rx_p - hp->body;
607 29031
608 29031
        p = http_find_header(hh, "transfer-encoding");
609 1506
        q = http_find_header(hh, "content-length");
610 0
        if (p != NULL && !strcasecmp(p, "chunked")) {
611 0
                if (q != NULL) {
612
                        vtc_log(hp->vl, hp->fatal, "Both C-E: Chunked and C-L");
613 1506
                        return;
614 10488
                }
615 9022
                ll = 0;
616 9022
                while (http_rxchunk(hp) > 0) {
617 40
                        ll = (hp->rx_p - hp->body) - l;
618
                        if (max && ll >= max)
619 1506
                                break;
620 29031
                }
621 14833
                p = "chunked";
622 14833
        } else if (q != NULL) {
623 30
                ll = strtoul(q, NULL, 10);
624
                if (max && ll > l + max)
625 14803
                        ll = max;
626 14833
                else
627 14833
                        ll -= l;
628 40
                i = http_rxchar(hp, ll, 0);
629 14793
                if (i < 0)
630 27485
                        return;
631 50
                p = "c-l";
632 50
        } else if (body) {
633 1311310
                ll = 0;
634 1311310
                do  {
635 0
                        i = http_rxchar(hp, 1, 1);
636 1311310
                        if (i < 0)
637 1311310
                                return;
638 20
                        ll += i;
639 1311290
                        if (max && ll >= max)
640 50
                                break;
641 50
                } while (i > 0);
642 12642
                p = "eof";
643 12642
        } else {
644
                p = "none";
645 28991
                ll = l = 0;
646 28991
        }
647 28991
        vtc_dump(hp->vl, 4, p, hp->body + l, ll);
648 28991
        l += ll;
649 29031
        hp->bodyl = l;
650
        bprintf(hp->bodylen, "%d", l);
651
}
652
653
/**********************************************************************
654
 * Receive a HTTP protocol header
655
 */
656 29892
657
static void
658 29892
http_rxhdr(struct http *hp)
659
{
660
        int i, s = 0;
661
        char *p;
662 29892
        ssize_t l;
663 29892
664 29892
        CHECK_OBJ_NOTNULL(hp, HTTP_MAGIC);
665 29892
        hp->rx_p = hp->rx_b;
666 29892
        *hp->rx_p = '\0';
667 5577016
        hp->body = NULL;
668 5577016
        bprintf(hp->bodylen, "%s", "<undef>");
669 5577016
        while (1) {
670 5577016
                p = hp->rx_p;
671 405
                i = http_rxchar(hp, 1, 1);
672 5576611
                if (i < 1)
673 238854
                        break;
674 5337757
                if (s == 0 && *p == '\r')
675 238866
                        s = 1;
676 5098891
                else if ((s == 0 || s == 1) && *p == '\n')
677 29482
                        s = 2;
678 5069409
                else if (s == 2 && *p == '\r')
679 29487
                        s = 3;
680
                else if ((s == 2 || s == 3) && *p == '\n')
681 5039922
                        break;
682
                else
683 29892
                        s = 0;
684 29892
        }
685 29892
        l = hp->rx_p - hp->rx_b;
686 29892
        vtc_dump(hp->vl, 4, "rxhdr", hp->rx_b, l);
687 405
        vtc_log(hp->vl, 4, "rxhdrlen = %zd", l);
688 29892
        if (i < 1)
689 29892
                vtc_log(hp->vl, hp->fatal, "HTTP header is incomplete");
690 29892
        *hp->rx_p = '\0';
691
        hp->body = hp->rx_p;
692
}
693
694
/* SECTION: client-server.spec.rxresp
695
 *
696
 * rxresp [-no_obj] (client only)
697
 *         Receive and parse a response's headers and body. If -no_obj is
698
 *         present, only get the headers.
699
 */
700 17160
701
static void
702
cmd_http_rxresp(CMD_ARGS)
703 17160
{
704
        struct http *hp;
705 17160
        int has_obj = 1;
706 17160
707 17160
        (void)vl;
708 17160
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
709 17160
        ONLY_CLIENT(hp, av);
710
        AZ(strcmp(av[0], "rxresp"));
711 17300
        av++;
712 280
713 140
        for (; *av != NULL; av++)
714
                if (!strcmp(*av, "-no_obj"))
715 0
                        has_obj = 0;
716 0
                else
717 17160
                        vtc_fatal(hp->vl,
718 17160
                            "Unknown http rxresp spec: %s\n", *av);
719 17160
        http_rxhdr(hp);
720 0
        http_splitheader(hp, 0);
721
        if (http_count_header(hp->resp, "Content-Length") > 1)
722 17160
                vtc_fatal(hp->vl,
723 140
                    "Multiple Content-Length headers.\n");
724 17020
        if (!has_obj)
725 320
                return;
726 16700
        if (!hp->resp[0] || !hp->resp[1])
727 350
                return;
728 16350
        if (hp->head_method)
729 12400
                return;
730
        if (!strcmp(hp->resp[1], "200"))
731 3950
                http_swallow_body(hp, hp->resp, 1, 0);
732 16350
        else
733 17160
                http_swallow_body(hp, hp->resp, 0, 0);
734
        vtc_log(hp->vl, 4, "bodylen = %s", hp->bodylen);
735
}
736
737
/* SECTION: client-server.spec.rxresphdrs
738
 *
739
 * rxresphdrs (client only)
740
 *         Receive and parse a response's headers.
741
 */
742 285
743
static void
744
cmd_http_rxresphdrs(CMD_ARGS)
745
{
746 285
        struct http *hp;
747 285
748 285
        (void)vl;
749 285
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
750 285
        ONLY_CLIENT(hp, av);
751
        AZ(strcmp(av[0], "rxresphdrs"));
752 285
        av++;
753 0
754 285
        for (; *av != NULL; av++)
755 285
                vtc_fatal(hp->vl, "Unknown http rxresp spec: %s\n", *av);
756 285
        http_rxhdr(hp);
757 0
        http_splitheader(hp, 0);
758
        if (http_count_header(hp->resp, "Content-Length") > 1)
759 285
                vtc_fatal(hp->vl,
760
                    "Multiple Content-Length headers.\n");
761
}
762
763
/* SECTION: client-server.spec.gunzip
764
 *
765
 * gunzip
766
 *         Gunzip the body in place.
767 195
 */
768
static void
769
cmd_http_gunzip(CMD_ARGS)
770
{
771 195
        struct http *hp;
772 195
773
        (void)av;
774 195
        (void)vl;
775 195
776 195
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
777
        vtc_gunzip(hp, hp->body, &hp->bodyl);
778
}
779
780
/**********************************************************************
781
 * Handle common arguments of a transmitted request or response
782
 */
783 28803
784
static char* const *
785
http_tx_parse_args(char * const *av, struct vtclog *vl, struct http *hp,
786 28803
    char *body, unsigned nohost, unsigned nodate, unsigned noserver, unsigned nouseragent)
787
{
788
        long bodylen = 0;
789
        char *b, *c;
790 28803
        char *nullbody;
791
        ssize_t len;
792
        int nolen = 0;
793 28803
        int l;
794
795 40985
        nullbody = body;
796 18236
797 755
        for (; *av != NULL; av++) {
798 18236
                if (!strcmp(*av, "-nolen")) {
799 10
                        nolen = 1;
800 17481
                } else if (!strcmp(*av, "-nohost")) {
801 40
                        nohost = 1;
802 17471
                } else if (!strcmp(*av, "-nodate")) {
803 11357
                        nodate = 1;
804 11002
                } else if (!strcmp(*av, "-hdr")) {
805 870
                        if (!strncasecmp(av[1], "Content-Length:", 15) ||
806 11357
                            !strncasecmp(av[1], "Transfer-Encoding:", 18))
807 235
                                nolen = 1;
808 11357
                        if (!strncasecmp(av[1], "Host:", 5))
809 40
                                nohost = 1;
810 11357
                        if (!strncasecmp(av[1], "Date:", 5))
811 80
                                nodate = 1;
812 11357
                        if (!strncasecmp(av[1], "Server:", 7))
813 15
                                noserver = 1;
814 11357
                        if (!strncasecmp(av[1], "User-Agent:", 11))
815 11357
                                nouseragent = 1;
816 17431
                        VSB_printf(hp->vsb, "%s%s", av[1], nl);
817 20
                        av++;
818 20
                } else if (!strcmp(*av, "-hdrlen")) {
819 3770
                        VSB_printf(hp->vsb, "%s: ", av[1]);
820 3750
                        l = atoi(av[2]);
821 20
                        while (l-- > 0)
822 20
                                VSB_putc(hp->vsb, '0' + (l % 10));
823 20
                        VSB_printf(hp->vsb, "%s", nl);
824 6054
                        av+=2;
825 12182
                } else
826 34948
                        break;
827 6145
        }
828 3905
        for (; *av != NULL; av++) {
829 3905
                if (!strcmp(*av, "-body")) {
830
                        assert(body == nullbody);
831 3905
                        REPLACE(body, av[1]);
832 3905
833 3905
                        AN(body);
834 117725
                        av++;
835 113820
                        bodylen = strlen(body);
836 25
                        for (b = body; *b != '\0'; b++) {
837 225
                                if (*b == '\\' && b[1] == '0') {
838 200
                                        *b = '\0';
839 200
                                        for (c = b+1; *c != '\0'; c++) {
840 25
                                                *c = c[1];
841 25
                                        }
842 25
                                        b++;
843 113820
                                        bodylen--;
844 6145
                                }
845 20
                        }
846 20
                } else if (!strcmp(*av, "-bodyfrom")) {
847 20
                        assert(body == nullbody);
848 20
                        free(body);
849 20
                        body = VFIL_readfile(NULL, av[1], &len);
850 20
                        AN(body);
851 20
                        assert(len < INT_MAX);
852 2240
                        bodylen = len;
853 1300
                        av++;
854 1300
                } else if (!strcmp(*av, "-bodylen")) {
855 1300
                        assert(body == nullbody);
856 1300
                        free(body);
857 1300
                        body = synth_body(av[1], 0);
858 2220
                        bodylen = strlen(body);
859 920
                        av++;
860 920
                } else if (!strncmp(*av, "-gzip", 5)) {
861 0
                        l = vtc_gzip_cmd(hp, av, &body, &bodylen);
862 920
                        if (l == 0)
863 920
                                break;
864 830
                        av++;
865 920
                        if (l > 1)
866 0
                                VSB_printf(hp->vsb, "Content-Encoding: gzip%s", nl);
867 6145
                } else
868 28803
                        break;
869 16770
        }
870 16770
        if (!nohost) {
871 16770
                VSB_cat(hp->vsb, "Host: ");
872 16770
                macro_cat(vl, hp->vsb, "localhost", NULL);
873 28803
                VSB_cat(hp->vsb, nl);
874 11501
        }
875 11501
        if (!nodate) {
876 11501
                VSB_cat(hp->vsb, "Date: ");
877 11501
                macro_cat(vl, hp->vsb, "date", NULL);
878 28803
                VSB_cat(hp->vsb, nl);
879 11426
        }
880 28803
        if (!noserver)
881 17185
                VSB_printf(hp->vsb, "Server: %s%s", hp->sess->name, nl);
882 28803
        if (!nouseragent)
883 11251
                VSB_printf(hp->vsb, "User-Agent: %s%s", hp->sess->name, nl);
884 28803
        if (body != NULL && !nolen)
885 28803
                VSB_printf(hp->vsb, "Content-Length: %ld%s", bodylen, nl);
886 11956
        VSB_cat(hp->vsb, nl);
887 11956
        if (body != NULL) {
888 11956
                VSB_bcat(hp->vsb, body, bodylen);
889 28803
                free(body);
890
        }
891
        return (av);
892
}
893
894
/* SECTION: client-server.spec.txreq
895
 *
896
 * txreq|txresp [...]
897
 *         Send a minimal request or response, but overload it if necessary.
898
 *
899
 *         txreq is client-specific and txresp is server-specific.
900
 *
901
 *         The only thing different between a request and a response, apart
902
 *         from who can send them is that the first line (request line vs
903
 *         status line), so all the options are pretty much the same.
904
 *
905
 *         \-method STRING (txreq only)
906
 *                 What method to use (default: "GET").
907
 *
908
 *         \-req STRING (txreq only)
909
 *                 Alias for -method.
910
 *
911
 *         \-url STRING (txreq only)
912
 *                 What location to use (default "/").
913
 *
914
 *         \-proto STRING
915
 *                 What protocol use in the status line.
916
 *                 (default: "HTTP/1.1").
917
 *
918
 *         \-status NUMBER (txresp only)
919
 *                 What status code to return (default 200).
920
 *
921
 *         \-reason STRING (txresp only)
922
 *                 What message to put in the status line (default: "OK").
923
 *
924
 *         \-noserver (txresp only)
925
 *                 Don't include a Server header with the id of the server.
926
 *
927
 *         \-nouseragent (txreq only)
928
 *                 Don't include a User-Agent header with the id of the client.
929
 *
930
 *         These three switches can appear in any order but must come before the
931
 *         following ones.
932
 *
933
 *         \-nohost
934
 *                 Don't include a Host header in the request. Also Implied
935
 *                 by the addition of a Host header with ``-hdr``.
936
 *
937
 *         \-nolen
938
 *                 Don't include a Content-Length header. Also implied by the
939
 *                 addition of a Content-Length or Transfer-Encoding header
940
 *                 with ``-hdr``.
941
 *
942
 *         \-nodate
943
 *                 Don't include a Date header in the response. Also implied
944
 *                 by the addition of a Date header with ``-hdr``.
945
 *
946
 *         \-hdr STRING
947
 *                 Add STRING as a header, it must follow this format:
948
 *                 "name: value". It can be called multiple times.
949
 *
950
 *         \-hdrlen STRING NUMBER
951
 *                 Add STRING as a header with NUMBER bytes of content.
952
 *
953
 *         You can then use the arguments related to the body:
954
 *
955
 *         \-body STRING
956
 *                 Input STRING as body.
957
 *
958
 *         \-bodyfrom FILE
959
 *                 Same as -body but content is read from FILE.
960
 *
961
 *         \-bodylen NUMBER
962
 *                 Generate and input a body that is NUMBER bytes-long.
963
 *
964
 *         \-gziplevel NUMBER
965
 *                 Set the gzip level (call it before any of the other gzip
966
 *                 switches).
967
 *
968
 *         \-gzipresidual NUMBER
969
 *                 Add extra gzip bits. You should never need it.
970
 *
971
 *         \-gzipbody STRING
972
 *                 Gzip STRING and send it as body.
973
 *
974
 *         \-gziplen NUMBER
975
 *                 Combine -bodylen and -gzipbody: generate a string of length
976
 *                 NUMBER, gzip it and send as body.
977
 */
978
979
/**********************************************************************
980
 * Transmit a response
981
 */
982 11580
983
static void
984
cmd_http_txresp(CMD_ARGS)
985 11580
{
986 11580
        struct http *hp;
987 11580
        const char *proto = "HTTP/1.1";
988 11580
        const char *status = "200";
989 11580
        const char *reason = "OK";
990
        char* body = NULL;
991 11580
        unsigned noserver = 0;
992 11580
993 11580
        (void)vl;
994 11580
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
995 11580
        ONLY_SERVER(hp, av);
996
        AZ(strcmp(av[0], "txresp"));
997 11580
        av++;
998
999 12890
        VSB_clear(hp->vsb);
1000 8890
1001 205
        for (; *av != NULL; av++) {
1002 205
                if (!strcmp(*av, "-proto")) {
1003 8890
                        proto = av[1];
1004 890
                        av++;
1005 890
                } else if (!strcmp(*av, "-status")) {
1006 8685
                        status = av[1];
1007 140
                        av++;
1008 140
                } else if (!strcmp(*av, "-reason")) {
1009 140
                        reason = av[1];
1010 7655
                        av++;
1011 75
                        continue;
1012 75
                } else if (!strcmp(*av, "-noserver")) {
1013
                        noserver = 1;
1014 7580
                        continue;
1015 1095
                } else
1016
                        break;
1017 11580
        }
1018
1019
        VSB_printf(hp->vsb, "%s %s %s%s", proto, status, reason, nl);
1020 11580
1021
        /* send a "Content-Length: 0" header unless something else happens */
1022 11580
        REPLACE(body, "");
1023 11580
1024 0
        av = http_tx_parse_args(av, vl, hp, body, 1, 0, noserver, 1);
1025
        if (*av != NULL)
1026 11580
                vtc_fatal(hp->vl, "Unknown http txresp spec: %s\n", *av);
1027 11580
1028
        http_write(hp, 4, "txresp");
1029
}
1030 10
1031
static void
1032
cmd_http_upgrade(CMD_ARGS)
1033
{
1034
        char *h;
1035 10
        struct http *hp;
1036 10
1037 10
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1038
        ONLY_SERVER(hp, av);
1039 10
        AN(hp->sfd);
1040 10
1041 0
        h = http_find_header(hp->req, "Upgrade");
1042
        if (!h || strcmp(h, "h2c"))
1043 10
                vtc_fatal(vl, "Req misses \"Upgrade: h2c\" header");
1044 10
1045 0
        h = http_find_header(hp->req, "Connection");
1046
        if (!h || strcmp(h, "Upgrade, HTTP2-Settings"))
1047
                vtc_fatal(vl, "Req misses \"Connection: "
1048 10
                        "Upgrade, HTTP2-Settings\" header");
1049 10
1050 0
        h = http_find_header(hp->req, "HTTP2-Settings");
1051
        if (!h)
1052 10
                vtc_fatal(vl, "Req misses \"HTTP2-Settings\" header");
1053
1054
        parse_string(vl, hp,
1055
            "txresp -status 101"
1056
            " -hdr \"Connection: Upgrade\""
1057
            " -hdr \"Upgrade: h2c\"\n"
1058 10
        );
1059
1060
        parse_string(vl, hp,
1061
            "rxpri\n"
1062
            "stream 0 {\n"
1063
            "    txsettings\n"
1064
            "    rxsettings\n"
1065
            "    txsettings -ack\n"
1066
            "    rxsettings\n"
1067
            "    expect settings.ack == true\n"
1068
            "} -start\n"
1069 10
        );
1070
1071 10
        b64_settings(hp, h);
1072
1073
}
1074
1075
/**********************************************************************
1076
 * Receive a request
1077
 */
1078
1079
/* SECTION: client-server.spec.rxreq
1080
 *
1081
 * rxreq (server only)
1082
 *         Receive and parse a request's headers and body.
1083 12522
 */
1084
static void
1085
cmd_http_rxreq(CMD_ARGS)
1086
{
1087 12522
        struct http *hp;
1088 12522
1089 12522
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1090 12522
        ONLY_SERVER(hp, av);
1091
        AZ(strcmp(av[0], "rxreq"));
1092 12522
        av++;
1093 0
1094 12522
        for (; *av != NULL; av++)
1095 12522
                vtc_fatal(vl, "Unknown http rxreq spec: %s\n", *av);
1096 12522
        http_rxhdr(hp);
1097 0
        http_splitheader(hp, 1);
1098 12522
        if (http_count_header(hp->req, "Content-Length") > 1)
1099 12522
                vtc_fatal(vl, "Multiple Content-Length headers.\n");
1100 12522
        http_swallow_body(hp, hp->req, 0, 0);
1101
        vtc_log(vl, 4, "bodylen = %s", hp->bodylen);
1102
}
1103
1104
/* SECTION: client-server.spec.rxreqhdrs
1105
 *
1106
 * rxreqhdrs (server only)
1107
 *         Receive and parse a request's headers (but not the body).
1108
 */
1109 25
1110
static void
1111
cmd_http_rxreqhdrs(CMD_ARGS)
1112
{
1113 25
        struct http *hp;
1114 25
1115 25
        (void)vl;
1116 25
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1117
        AZ(strcmp(av[0], "rxreqhdrs"));
1118 25
        av++;
1119 0
1120 25
        for (; *av != NULL; av++)
1121 25
                vtc_fatal(hp->vl, "Unknown http rxreq spec: %s\n", *av);
1122 25
        http_rxhdr(hp);
1123 0
        http_splitheader(hp, 1);
1124 25
        if (http_count_header(hp->req, "Content-Length") > 1)
1125
                vtc_fatal(hp->vl, "Multiple Content-Length headers.\n");
1126
}
1127
1128
/* SECTION: client-server.spec.rxreqbody
1129
 *
1130
 * rxreqbody (server only)
1131
 *         Receive a request's body.
1132
 */
1133 20
1134
static void
1135
cmd_http_rxreqbody(CMD_ARGS)
1136
{
1137 20
        struct http *hp;
1138 20
1139 20
        (void)vl;
1140 20
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1141 20
        ONLY_SERVER(hp, av);
1142
        AZ(strcmp(av[0], "rxreqbody"));
1143 20
        av++;
1144 0
1145 20
        for (; *av != NULL; av++)
1146 20
                vtc_fatal(hp->vl, "Unknown http rxreq spec: %s\n", *av);
1147 20
        http_swallow_body(hp, hp->req, 0, 0);
1148
        vtc_log(hp->vl, 4, "bodylen = %s", hp->bodylen);
1149
}
1150
1151
/* SECTION: client-server.spec.rxrespbody
1152
 *
1153
 * rxrespbody (client only)
1154
 *         Receive (part of) a response's body.
1155
 *
1156
 * -max : max length of this receive, 0 for all
1157
 */
1158 240
1159
static void
1160
cmd_http_rxrespbody(CMD_ARGS)
1161 240
{
1162
        struct http *hp;
1163 240
        int max = 0;
1164 240
1165 240
        (void)vl;
1166 240
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1167 240
        ONLY_CLIENT(hp, av);
1168
        AZ(strcmp(av[0], "rxrespbody"));
1169 330
        av++;
1170 180
1171 90
        for (; *av != NULL; av++)
1172 90
                if (!strcmp(*av, "-max")) {
1173 90
                        max = atoi(av[1]);
1174 0
                        av++;
1175 0
                } else
1176
                        vtc_fatal(hp->vl,
1177 240
                            "Unknown http rxrespbody spec: %s\n", *av);
1178 240
1179 240
        http_swallow_body(hp, hp->resp, 1, max);
1180
        vtc_log(hp->vl, 4, "bodylen = %s", hp->bodylen);
1181
}
1182
1183
/* SECTION: client-server.spec.rxchunk
1184
 *
1185
 * rxchunk
1186
 *         Receive an HTTP chunk.
1187
 */
1188 65
1189
static void
1190
cmd_http_rxchunk(CMD_ARGS)
1191
{
1192
        struct http *hp;
1193 65
        int ll, i;
1194 65
1195 65
        (void)vl;
1196
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1197 65
        ONLY_CLIENT(hp, av);
1198 65
1199 0
        i = http_rxchunk(hp);
1200 0
        if (i == 0) {
1201 0
                ll = hp->rx_p - hp->body;
1202 0
                hp->bodyl = ll;
1203 0
                bprintf(hp->bodylen, "%d", ll);
1204 65
                vtc_log(hp->vl, 4, "bodylen = %s", hp->bodylen);
1205
        }
1206
}
1207
1208
/**********************************************************************
1209
 * Transmit a request
1210
 */
1211 17225
1212
static void
1213
cmd_http_txreq(CMD_ARGS)
1214 17225
{
1215 17225
        struct http *hp;
1216 17225
        const char *req = "GET";
1217 17225
        const char *url = "/";
1218
        const char *proto = "HTTP/1.1";
1219 17225
        const char *up = NULL;
1220
        unsigned nohost;
1221 17225
        unsigned nouseragent = 0;
1222 17225
1223 17225
        (void)vl;
1224 17225
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1225 17225
        ONLY_CLIENT(hp, av);
1226
        AZ(strcmp(av[0], "txreq"));
1227 17225
        av++;
1228
1229 17225
        VSB_clear(hp->vsb);
1230 27880
1231 15640
        hp->head_method = 0;
1232 8875
        for (; *av != NULL; av++) {
1233 8875
                if (!strcmp(*av, "-url")) {
1234 15640
                        url = av[1];
1235 250
                        av++;
1236 250
                } else if (!strcmp(*av, "-proto")) {
1237 6765
                        proto = av[1];
1238 5875
                        av++;
1239 1500
                } else if (!strcmp(*av, "-method") ||
1240 1500
                    !strcmp(*av, "-req")) {
1241 1500
                        req = av[1];
1242 6515
                        hp->head_method = !strcmp(av[1], "HEAD") ;
1243 5
                        av++;
1244 5
                } else if (!hp->sfd && !strcmp(*av, "-up")) {
1245 5015
                        up = av[1];
1246 25
                        av++;
1247 25
                } else if (!strcmp(*av, "-nouseragent")) {
1248 4985
                        nouseragent = 1;
1249 10655
                } else
1250 17225
                        break;
1251
        }
1252 17225
        VSB_printf(hp->vsb, "%s %s %s%s", req, url, proto, nl);
1253 10
1254
        if (up)
1255 5
                VSB_printf(hp->vsb, "Connection: Upgrade, HTTP2-Settings%s"
1256
                                "Upgrade: h2c%s"
1257 17225
                                "HTTP2-Settings: %s%s", nl, nl, up, nl);
1258 17225
1259 17225
        nohost = strcmp(proto, "HTTP/1.1") != 0;
1260 0
        av = http_tx_parse_args(av, vl, hp, NULL, nohost, 1, 1, nouseragent);
1261 17225
        if (*av != NULL)
1262
                vtc_fatal(hp->vl, "Unknown http txreq spec: %s\n", *av);
1263 17225
        http_write(hp, 4, "txreq");
1264 5
1265
        if (up) {
1266
                parse_string(vl, hp,
1267
                    "rxresp\n"
1268
                    "expect resp.status == 101\n"
1269
                    "expect resp.http.connection == Upgrade\n"
1270
                    "expect resp.http.upgrade == h2c\n"
1271 5
                    "txpri\n"
1272 5
                );
1273
                b64_settings(hp, up);
1274
                parse_string(vl, hp,
1275
                    "stream 0 {\n"
1276
                    "    txsettings\n"
1277
                    "    rxsettings\n"
1278
                    "    txsettings -ack\n"
1279
                    "    rxsettings\n"
1280
                    "    expect settings.ack == true"
1281 5
                    "} -start\n"
1282 17225
                );
1283
        }
1284
}
1285
1286
/* SECTION: client-server.spec.recv
1287
 *
1288
 * recv NUMBER
1289
 *         Read NUMBER bytes from the connection.
1290
 */
1291 40
1292
static void
1293
cmd_http_recv(CMD_ARGS)
1294
{
1295
        struct http *hp;
1296
        int i, n;
1297 40
        char u[32];
1298 40
1299 40
        (void)vl;
1300 40
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1301 40
        AN(av[1]);
1302 120
        AZ(av[2]);
1303 80
        n = strtoul(av[1], NULL, 0);
1304 80
        while (n > 0) {
1305 80
                i = read(*hp->sess->fd, u, n > 32 ? 32 : n);
1306
                if (i > 0)
1307 0
                        vtc_dump(hp->vl, 4, "recv", u, i);
1308 0
                else
1309 80
                        vtc_log(hp->vl, hp->fatal, "recv() got %d (%s)", i,
1310
                            strerror(errno));
1311 40
                n -= i;
1312
        }
1313
}
1314
1315
/* SECTION: client-server.spec.send
1316
 *
1317
 * send STRING
1318
 *         Push STRING on the connection.
1319
 */
1320 4669
1321
static void
1322
cmd_http_send(CMD_ARGS)
1323
{
1324
        struct http *hp;
1325 4669
        int i;
1326 4669
1327 4669
        (void)vl;
1328 4669
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1329 4669
        AN(av[1]);
1330 4669
        AZ(av[2]);
1331 4669
        vtc_dump(hp->vl, 4, "send", av[1], -1);
1332 2
        i = write(*hp->sess->fd, av[1], strlen(av[1]));
1333 1
        if (i != strlen(av[1]))
1334 4669
                vtc_log(hp->vl, hp->fatal, "Write error in http_send(): %s",
1335
                    strerror(errno));
1336
}
1337
1338
/* SECTION: client-server.spec.send_n
1339
 *
1340
 * send_n NUMBER STRING
1341
 *         Write STRING on the socket NUMBER times.
1342
 */
1343 20
1344
static void
1345
cmd_http_send_n(CMD_ARGS)
1346
{
1347
        struct http *hp;
1348 20
        int i, n, l;
1349 20
1350 20
        (void)vl;
1351 20
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1352 20
        AN(av[1]);
1353 20
        AN(av[2]);
1354 20
        AZ(av[3]);
1355 20
        n = strtoul(av[1], NULL, 0);
1356 10470
                vtc_dump(hp->vl, 4, "send_n", av[2], -1);
1357 10450
        l = strlen(av[2]);
1358 10450
        while (n--) {
1359 910
                i = write(*hp->sess->fd, av[2], l);
1360
                if (i != l)
1361 455
                        vtc_log(hp->vl, hp->fatal,
1362
                            "Write error in http_send(): %s",
1363 20
                            strerror(errno));
1364
        }
1365
}
1366
1367
/* SECTION: client-server.spec.send_urgent
1368
 *
1369
 * send_urgent STRING
1370
 *         Send string as TCP OOB urgent data. You will never need this.
1371
 */
1372 50
1373
static void
1374
cmd_http_send_urgent(CMD_ARGS)
1375
{
1376
        struct http *hp;
1377 50
        int i;
1378 50
1379 50
        (void)vl;
1380 50
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1381 50
        AN(av[1]);
1382 50
        AZ(av[2]);
1383 50
        vtc_dump(hp->vl, 4, "send_urgent", av[1], -1);
1384 0
        i = send(*hp->sess->fd, av[1], strlen(av[1]), MSG_OOB);
1385 0
        if (i != strlen(av[1]))
1386 50
                vtc_log(hp->vl, hp->fatal,
1387
                    "Write error in http_send_urgent(): %s", strerror(errno));
1388
}
1389
1390
/* SECTION: client-server.spec.sendhex
1391
 *
1392
 * sendhex STRING
1393
 *         Send bytes as described by STRING. STRING should consist of hex pairs
1394
 *         possibly separated by whitespace or newlines. For example:
1395
 *         "0F EE a5    3df2".
1396
 */
1397 895
1398
static void
1399
cmd_http_sendhex(CMD_ARGS)
1400
{
1401
        struct vsb *vsb;
1402 895
        struct http *hp;
1403 895
1404 895
        (void)vl;
1405 895
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1406 895
        AN(av[1]);
1407 895
        AZ(av[2]);
1408 895
        vsb = vtc_hex_to_bin(hp->vl, av[1]);
1409 895
        assert(VSB_len(vsb) >= 0);
1410 0
        vtc_hexdump(hp->vl, 4, "sendhex", VSB_data(vsb), VSB_len(vsb));
1411 0
        if (VSB_tofile(vsb, *hp->sess->fd))
1412 895
                vtc_log(hp->vl, hp->fatal, "Write failed: %s",
1413 895
                    strerror(errno));
1414
        VSB_destroy(&vsb);
1415
}
1416
1417
/* SECTION: client-server.spec.chunked
1418
 *
1419
 * chunked STRING
1420
 *         Send STRING as chunked encoding.
1421
 */
1422 490
1423
static void
1424
cmd_http_chunked(CMD_ARGS)
1425
{
1426 490
        struct http *hp;
1427 490
1428 490
        (void)vl;
1429 490
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1430 490
        AN(av[1]);
1431 980
        AZ(av[2]);
1432 490
        VSB_clear(hp->vsb);
1433 490
        VSB_printf(hp->vsb, "%jx%s%s%s",
1434 490
            (uintmax_t)strlen(av[1]), nl, av[1], nl);
1435
        http_write(hp, 4, "chunked");
1436
}
1437
1438
/* SECTION: client-server.spec.chunkedlen
1439
 *
1440
 * chunkedlen NUMBER
1441
 *         Do as ``chunked`` except that the string will be generated
1442
 *         for you, with a length of NUMBER characters.
1443
 */
1444 11315
1445
static void
1446
cmd_http_chunkedlen(CMD_ARGS)
1447
{
1448
        unsigned len;
1449
        unsigned u, v;
1450
        char buf[16384];
1451 11315
        struct http *hp;
1452 11315
1453 11315
        (void)vl;
1454 11315
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1455 11315
        AN(av[1]);
1456
        AZ(av[2]);
1457 11315
        VSB_clear(hp->vsb);
1458
1459 11315
        len = atoi(av[1]);
1460 430
1461 430
        if (len == 0) {
1462 101735807
                VSB_printf(hp->vsb, "0%s%s", nl, nl);
1463 101724922
        } else {
1464
                for (u = 0; u < sizeof buf; u++)
1465 10885
                        buf[u] = (u & 7) + '0';
1466 22660
1467 11775
                VSB_printf(hp->vsb, "%x%s", len, nl);
1468 11775
                for (u = 0; u < len; u += v) {
1469 11775
                        v = vmin_t(unsigned, len - u, sizeof buf);
1470 10885
                        VSB_bcat(hp->vsb, buf, v);
1471
                }
1472 11315
                VSB_printf(hp->vsb, "%s", nl);
1473 11315
        }
1474
        http_write(hp, 4, "chunked");
1475
}
1476
1477
1478
/* SECTION: client-server.spec.timeout
1479
 *
1480
 * timeout NUMBER
1481
 *         Set the TCP timeout for this entity.
1482
 */
1483 170
1484
static void
1485
cmd_http_timeout(CMD_ARGS)
1486
{
1487
        struct http *hp;
1488 170
        double d;
1489 170
1490 170
        (void)vl;
1491 170
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1492 170
        AN(av[1]);
1493 170
        AZ(av[2]);
1494 0
        d = VNUM(av[1]);
1495 170
        if (isnan(d))
1496 170
                vtc_fatal(vl, "timeout is not a number (%s)", av[1]);
1497
        hp->timeout = d;
1498
}
1499
1500
/* SECTION: client-server.spec.expect_close
1501
 *
1502
 * expect_close
1503
 *      Reads from the connection, expecting nothing to read but an EOF.
1504 915
 */
1505
static void
1506
cmd_http_expect_close(CMD_ARGS)
1507
{
1508
        struct http *hp;
1509
        struct pollfd fds[1];
1510
        char c;
1511 915
        int i;
1512 915
1513 915
        (void)vl;
1514
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1515 915
        AZ(av[1]);
1516 915
1517 75
        vtc_log(vl, 4, "Expecting close (fd = %d)", *hp->sess->fd);
1518 915
        if (hp->h2)
1519 915
                stop_h2(hp);
1520 915
        while (1) {
1521 915
                fds[0].fd = *hp->sess->fd;
1522 915
                fds[0].events = POLLIN;
1523 915
                fds[0].revents = 0;
1524 0
                i = poll(fds, 1, (int)(hp->timeout * 1000));
1525 915
                if (i < 0 && errno == EINTR)
1526 0
                        continue;
1527 915
                if (i == 0)
1528 60
                        vtc_log(vl, hp->fatal, "Expected close: timeout");
1529
                if (i != 1 || !(fds[0].revents & (POLLIN|POLLERR|POLLHUP)))
1530 30
                        vtc_log(vl, hp->fatal,
1531 885
                            "Expected close: poll = %d, revents = 0x%x",
1532 885
                            i, fds[0].revents);
1533 885
                i = read(*hp->sess->fd, &c, 1);
1534 30
                if (i <= 0 && VTCP_Check(i))
1535 0
                        break;
1536 0
                if (i == 1 && vct_islws(c))
1537 0
                        continue;
1538
                vtc_log(vl, hp->fatal,
1539 885
                    "Expecting close: read = %d, c = 0x%02x", i, c);
1540 885
        }
1541
        vtc_log(vl, 4, "fd=%d EOF, as expected", *hp->sess->fd);
1542
}
1543
1544
/* SECTION: client-server.spec.close
1545
 *
1546
 * close (server only)
1547
 *      Close the connection. Note that if operating in HTTP/2 mode no
1548
 *      extra (GOAWAY) frame is sent, it's simply a TCP close.
1549 125
 */
1550
static void
1551
cmd_http_close(CMD_ARGS)
1552
{
1553 125
        struct http *hp;
1554 125
1555 125
        (void)vl;
1556 125
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1557 125
        ONLY_SERVER(hp, av);
1558 125
        AZ(av[1]);
1559 125
        assert(hp->sfd != NULL);
1560 0
        assert(*hp->sfd >= 0);
1561 125
        if (hp->h2)
1562 125
                stop_h2(hp);
1563 125
        VTCP_close(hp->sess->fd);
1564
        vtc_log(vl, 4, "Closed");
1565
}
1566
1567
/* SECTION: client-server.spec.accept
1568
 *
1569
 * accept (server only)
1570
 *      Close the current connection, if any, and accept a new one. Note
1571
 *      that this new connection is HTTP/1.x.
1572 889
 */
1573
static void
1574
cmd_http_accept(CMD_ARGS)
1575
{
1576 889
        struct http *hp;
1577 889
1578 889
        (void)vl;
1579 889
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1580 889
        ONLY_SERVER(hp, av);
1581 889
        AZ(av[1]);
1582 889
        assert(hp->sfd != NULL);
1583 0
        assert(*hp->sfd >= 0);
1584 889
        if (hp->h2)
1585 769
                stop_h2(hp);
1586 889
        if (*hp->sess->fd >= 0)
1587 889
                VTCP_close(hp->sess->fd);
1588 889
        vtc_log(vl, 4, "Accepting");
1589 0
        *hp->sess->fd = accept(*hp->sfd, NULL, NULL);
1590 889
        if (*hp->sess->fd < 0)
1591 889
                vtc_log(vl, hp->fatal, "Accepted failed: %s", strerror(errno));
1592
        vtc_log(vl, 3, "Accepted socket fd is %d", *hp->sess->fd);
1593
}
1594
1595
/* SECTION: client-server.spec.shutdown
1596
 *
1597
 * shutdown
1598
 *      Initiate shutdown.
1599
 *
1600
 *      \-read
1601
 *              Shutdown the read direction.
1602
 *      \-write
1603
 *              Shutdown the write direction.
1604
 *      \-notconn
1605
 *              OK to get ENOTCONN
1606
 *
1607
 *      The default is both direction.
1608 70
 */
1609
static void
1610
cmd_http_shutdown(CMD_ARGS)
1611 70
{
1612 70
        struct http *hp;
1613
        int how = SHUT_RDWR;
1614 70
        int notconn = 0;
1615
        int i, j;
1616
        const char *str[] = {
1617
                [SHUT_RD]       = "RD",
1618
                [SHUT_WR]       = "WR",
1619
                [SHUT_RDWR]     = "RDWR",
1620 70
        };
1621 70
1622 70
        (void)vl;
1623 70
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1624
        AZ(strcmp(av[0], "shutdown"));
1625 140
        av++;
1626 70
1627 15
        while (*av != NULL) {
1628 15
                if (!strcmp(*av, "-read")) {
1629 70
                        how = SHUT_RD;
1630 35
                        av++;
1631 35
                } else if (!strcmp(*av, "-write")) {
1632 55
                        how = SHUT_WR;
1633 20
                        av++;
1634 20
                } else if (!strcmp(*av, "-notconn")) {
1635 20
                        notconn = 1;
1636 0
                        av++;
1637
                } else {
1638
                        break;
1639
                }
1640 70
        }
1641 0
1642
        if (*av != NULL)
1643 70
                vtc_fatal(hp->vl, "Unknown http shutdown spec: %s\n", *av);
1644 70
1645 70
        vtc_log(vl, 4, "Shutting down fd (%s): %d", str[how], *hp->sess->fd);
1646 70
        i = shutdown(*hp->sess->fd, how);
1647 0
        j = errno;
1648
        if (i < 0 && j == ENOTCONN && notconn) {
1649 0
                vtc_log(vl, 3,
1650
                    "Shutdown(%s) socket fd %d, (was already closed)",
1651 70
                    str[how], *hp->sess->fd
1652 0
                );
1653
        } else if (i < 0) {
1654 0
                vtc_log(vl, hp->fatal,
1655
                    "Shutdown(%s) socket fd %d, failed, %s",
1656 0
                    str[how], *hp->sess->fd, strerror(j)
1657 140
                );
1658 70
        } else {
1659
                vtc_log(vl, 3, "Shutdown(%s) socket fd %d",
1660
                    str[how], *hp->sess->fd
1661 70
                );
1662
        }
1663
}
1664
1665
/* SECTION: client-server.spec.fatal
1666
 *
1667
 * fatal|non_fatal
1668
 *         Control whether a failure of this entity should stop the test.
1669
 */
1670 2114
1671
static void
1672
cmd_http_fatal(CMD_ARGS)
1673 2114
{
1674
        struct http *hp;
1675 2114
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1676 2114
1677 2114
        (void)vl;
1678 120
        AZ(av[1]);
1679 120
        if (!strcmp(av[0], "fatal")) {
1680 1994
                hp->fatal = 0;
1681 1994
        } else {
1682
                assert(!strcmp(av[0], "non_fatal"));
1683 2114
                hp->fatal = -1;
1684
        }
1685
}
1686
1687
#define cmd_http_non_fatal cmd_http_fatal
1688
1689
static const char PREFACE[24] = {
1690
        0x50, 0x52, 0x49, 0x20, 0x2a, 0x20, 0x48, 0x54,
1691
        0x54, 0x50, 0x2f, 0x32, 0x2e, 0x30, 0x0d, 0x0a,
1692
        0x0d, 0x0a, 0x53, 0x4d, 0x0d, 0x0a, 0x0d, 0x0a
1693
};
1694
1695
/* SECTION: client-server.spec.txpri
1696
 *
1697
 * txpri (client only)
1698
 *      Send an HTTP/2 preface ("PRI * HTTP/2.0\\r\\n\\r\\nSM\\r\\n\\r\\n")
1699
 *      and set client to HTTP/2.
1700 914
 */
1701
static void
1702
cmd_http_txpri(CMD_ARGS)
1703
{
1704
        size_t l;
1705 914
        struct http *hp;
1706 914
1707
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1708 914
        ONLY_CLIENT(hp, av);
1709
1710 914
        vtc_dump(hp->vl, 4, "txpri", PREFACE, sizeof(PREFACE));
1711 914
        /* Dribble out the preface */
1712 0
        l = write(*hp->sess->fd, PREFACE, 18);
1713 0
        if (l != 18)
1714 914
                vtc_log(vl, hp->fatal, "Write failed: (%zd vs %zd) %s",
1715 914
                    l, sizeof(PREFACE), strerror(errno));
1716 914
        VTIM_sleep(0.01);
1717 0
        l = write(*hp->sess->fd, PREFACE + 18, sizeof(PREFACE) - 18);
1718 0
        if (l != sizeof(PREFACE) - 18)
1719
                vtc_log(vl, hp->fatal, "Write failed: (%zd vs %zd) %s",
1720 914
                    l, sizeof(PREFACE), strerror(errno));
1721 914
1722 914
        start_h2(hp);
1723
        AN(hp->h2);
1724
}
1725
1726
/* SECTION: client-server.spec.rxpri
1727
 *
1728
 * rxpri (server only)
1729
 *      Receive a preface. If valid set the server to HTTP/2, abort
1730
 *      otherwise.
1731 150
 */
1732
static void
1733
cmd_http_rxpri(CMD_ARGS)
1734
{
1735 150
        struct http *hp;
1736 150
1737
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1738 150
        ONLY_SERVER(hp, av);
1739 150
1740 0
        hp->rx_p = hp->rx_b;
1741 150
        if (!http_rxchar(hp, sizeof(PREFACE), 0))
1742 0
                vtc_fatal(vl, "Couldn't retrieve connection preface");
1743 150
        if (memcmp(hp->rx_b, PREFACE, sizeof(PREFACE)))
1744 150
                vtc_fatal(vl, "Received invalid preface\n");
1745 150
        start_h2(hp);
1746
        AN(hp->h2);
1747
}
1748
1749
/* SECTION: client-server.spec.settings
1750
 *
1751
 * settings -dectbl INT
1752
 *      Force internal HTTP/2 settings to certain values. Currently only
1753
 *      support setting the decoding table size.
1754 0
 */
1755
static void
1756
cmd_http_settings(CMD_ARGS)
1757
{
1758
        uint32_t n;
1759 0
        char *p;
1760
        struct http *hp;
1761 0
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1762 0
1763
        if (!hp->h2)
1764 0
                vtc_fatal(hp->vl, "Only possible in H/2 mode");
1765
1766 0
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1767 0
1768 0
        for (; *av != NULL; av++) {
1769 0
                if (!strcmp(*av, "-dectbl")) {
1770 0
                        n = strtoul(av[1], &p, 0);
1771 0
                        if (*p != '\0')
1772 0
                                vtc_fatal(hp->vl, "-dectbl takes an integer as "
1773 0
                                    "argument (found %s)", av[1]);
1774 0
                        assert(HPK_ResizeTbl(hp->decctx, n) != hpk_err);
1775 0
                        av++;
1776 0
                } else
1777 0
                        vtc_fatal(vl, "Unknown settings spec: %s\n", *av);
1778
        }
1779
}
1780 3070
1781
static void
1782
cmd_http_stream(CMD_ARGS)
1783 3070
{
1784 3070
        struct http *hp;
1785 945
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1786 945
        if (!hp->h2) {
1787 130
                vtc_log(hp->vl, 4, "Not in H/2 mode, do what's needed");
1788
                if (hp->sfd)
1789 815
                        parse_string(vl, hp, "rxpri");
1790 945
                else
1791
                        parse_string(vl, hp, "txpri");
1792
                parse_string(vl, hp,
1793
                    "stream 0 {\n"
1794
                    "    txsettings\n"
1795
                    "    rxsettings\n"
1796
                    "    txsettings -ack\n"
1797
                    "    rxsettings\n"
1798
                    "    expect settings.ack == true"
1799 945
                    "} -run\n"
1800 3070
                );
1801 3070
        }
1802
        cmd_stream(av, hp, vl);
1803
}
1804
1805
/* SECTION: client-server.spec.write_body
1806
 *
1807
 * write_body STRING
1808
 *      Write the body of a request or a response to a file. By using the
1809
 *      shell command, higher-level checks on the body can be performed
1810
 *      (eg. XML, JSON, ...) provided that such checks can be delegated
1811
 *      to an external program.
1812 20
 */
1813
static void
1814
cmd_http_write_body(CMD_ARGS)
1815
{
1816 20
        struct http *hp;
1817 20
1818 20
        (void)vl;
1819 20
        CAST_OBJ_NOTNULL(hp, priv, HTTP_MAGIC);
1820 20
        AN(av[0]);
1821 20
        AN(av[1]);
1822 20
        AZ(av[2]);
1823 0
        AZ(strcmp(av[0], "write_body"));
1824 0
        if (VFIL_writefile(NULL, av[1], hp->body, hp->bodyl) != 0)
1825 20
                vtc_fatal(hp->vl, "failed to write body: %s (%d)",
1826
                    strerror(errno), errno);
1827
}
1828
1829
/**********************************************************************
1830
 * Execute HTTP specifications
1831
 */
1832
1833
const struct cmds http_cmds[] = {
1834
#define CMD_HTTP(n) { CMDS_MAGIC, #n, cmd_http_##n },
1835
        /* session */
1836
        CMD_HTTP(accept)
1837
        CMD_HTTP(close)
1838
        CMD_HTTP(recv)
1839
        CMD_HTTP(send)
1840
        CMD_HTTP(send_n)
1841
        CMD_HTTP(send_urgent)
1842
        CMD_HTTP(sendhex)
1843
        CMD_HTTP(shutdown)
1844
        CMD_HTTP(timeout)
1845
1846
        /* spec */
1847
        CMD_HTTP(fatal)
1848
        CMD_HTTP(non_fatal)
1849
1850
        /* body */
1851
        CMD_HTTP(gunzip)
1852
        CMD_HTTP(write_body)
1853
1854
        /* HTTP/1.x */
1855
        CMD_HTTP(chunked)
1856
        CMD_HTTP(chunkedlen)
1857
        CMD_HTTP(rxchunk)
1858
1859
        /* HTTP/2 */
1860
        CMD_HTTP(stream)
1861
        CMD_HTTP(settings)
1862
1863
        /* client */
1864
        CMD_HTTP(rxresp)
1865
        CMD_HTTP(rxrespbody)
1866
        CMD_HTTP(rxresphdrs)
1867
        CMD_HTTP(txpri)
1868
        CMD_HTTP(txreq)
1869
1870
        /* server */
1871
        CMD_HTTP(rxpri)
1872
        CMD_HTTP(rxreq)
1873
        CMD_HTTP(rxreqbody)
1874
        CMD_HTTP(rxreqhdrs)
1875
        CMD_HTTP(txresp)
1876
        CMD_HTTP(upgrade)
1877
1878
        /* expect */
1879
        CMD_HTTP(expect)
1880
        CMD_HTTP(expect_close)
1881
        CMD_HTTP(expect_pattern)
1882
#undef CMD_HTTP
1883
        { CMDS_MAGIC, NULL, NULL }
1884
};
1885 21039
1886
static void
1887
http_process_cleanup(void *arg)
1888
{
1889 21039
        struct http *hp;
1890
1891 21039
        CAST_OBJ_NOTNULL(hp, arg, HTTP_MAGIC);
1892 0
1893 21039
        if (vtc_error && *hp->sess->fd >= 0)
1894 990
                VTCP_close(hp->sess->fd);
1895 21039
        if (hp->h2)
1896 21039
                stop_h2(hp);
1897 21039
        VSB_destroy(&hp->vsb);
1898 21039
        free(hp->rx_b);
1899 21039
        free(hp->rem_ip);
1900 21039
        free(hp->rem_port);
1901 21039
        free(hp->rem_path);
1902
        FREE_OBJ(hp);
1903
}
1904 21042
1905
void
1906
http_process(struct vtclog *vl, struct vtc_sess *vsp, const char *spec,
1907
    int *sock, int *sfd, const char *addr, int rcvbuf)
1908
{
1909 21042
        struct http *hp;
1910
        int oldbuf;
1911 21042
        socklen_t intlen = sizeof(int);
1912 21042
1913 21042
        (void)sfd;
1914 21042
        ALLOC_OBJ(hp, HTTP_MAGIC);
1915 21042
        AN(hp);
1916 21042
        hp->sess = vsp;
1917
        hp->sess->fd = sock;
1918 21042
        hp->timeout = vtc_maxdur * .5;
1919
1920
        if (rcvbuf) {
1921 40
                // XXX setsockopt() too late on SunOS
1922
                // https://github.com/varnishcache/varnish-cache/pull/2980#issuecomment-486214661
1923 40
                hp->rcvbuf = rcvbuf;
1924 40
1925 40
                oldbuf = 0;
1926 40
                AZ(getsockopt(*hp->sess->fd, SOL_SOCKET, SO_RCVBUF, &oldbuf, &intlen));
1927
                AZ(setsockopt(*hp->sess->fd, SOL_SOCKET, SO_RCVBUF, &rcvbuf, intlen));
1928 80
                AZ(getsockopt(*hp->sess->fd, SOL_SOCKET, SO_RCVBUF, &rcvbuf, &intlen));
1929 40
1930 40
                vtc_log(vl, 3, "-rcvbuf fd=%d old=%d new=%d actual=%d",
1931
                    *hp->sess->fd, oldbuf, hp->rcvbuf, rcvbuf);
1932 21042
        }
1933 21042
1934 21042
        hp->nrxbuf = 2048*1024;
1935 21042
        hp->rx_b = malloc(hp->nrxbuf);
1936 21042
        AN(hp->rx_b);
1937 21042
        hp->rx_e = hp->rx_b + hp->nrxbuf;
1938
        hp->rx_p = hp->rx_b;
1939 21042
        *hp->rx_p = '\0';
1940 21042
1941
        hp->vsb = VSB_new_auto();
1942 21042
        AN(hp->vsb);
1943
1944 21042
        hp->sfd = sfd;
1945 21042
1946
        hp->rem_ip = malloc(VTCP_ADDRBUFSIZE);
1947 21042
        AN(hp->rem_ip);
1948 21042
1949
        hp->rem_port = malloc(VTCP_PORTBUFSIZE);
1950 21042
        AN(hp->rem_port);
1951 21042
1952 21042
        hp->vl = vl;
1953 21042
        vtc_log_set_cmd(hp->vl, http_cmds);
1954
        hp->gziplevel = 0;
1955 21042
        hp->gzipresidual = -1;
1956 18913
1957
        if (*addr != '/') {
1958 18913
                VTCP_hisname(*sock, hp->rem_ip, VTCP_ADDRBUFSIZE, hp->rem_port,
1959 18913
                             VTCP_PORTBUFSIZE);
1960 2129
                hp->rem_path = NULL;
1961 2129
        } else {
1962 2129
                strcpy(hp->rem_ip, "0.0.0.0");
1963
                strcpy(hp->rem_port, "0");
1964
                hp->rem_path = strdup(addr);
1965
        }
1966
        /* XXX: After an upgrade to HTTP/2 the cleanup of a server that is
1967
         * not -wait'ed before the test resets is subject to a race where the
1968
         * cleanup does not happen, so ASAN reports leaks despite the push
1969
         * of a cleanup handler. To easily reproduce, remove the server wait
1970 21042
         * from a02022.vtc and run with ASAN enabled.
1971 21042
         */
1972 21042
        pthread_cleanup_push(http_process_cleanup, hp);
1973 21042
        parse_string(vl, hp, spec);
1974 21042
        pthread_cleanup_pop(0);
1975
        http_process_cleanup(hp);
1976
}
1977
1978
/**********************************************************************
1979
 * Magic test routine
1980
 *
1981
 * This function brute-forces some short strings through gzip(9) to
1982
 * find candidates for all possible 8 bit positions of the stopbit.
1983
 *
1984
 * Here is some good short output strings:
1985
 *
1986
 *      0 184 <e04c8d0fd604c>
1987
 *      1 257 <1ea86e6cf31bf4ec3d7a86>
1988
 *      2 106 <10>
1989
 *      3 163 <a5e2e2e1c2e2>
1990
 *      4 180 <71c5d18ec5d5d1>
1991
 *      5 189 <39886d28a6d2988>
1992
 *      6 118 <80000>
1993
 *      7 151 <386811868>
1994
 *
1995
 */
1996
1997
#if 0
1998
void xxx(void);
1999
2000
void
2001
xxx(void)
2002
{
2003
        z_stream vz;
2004
        int n;
2005
        char ibuf[200];
2006
        char obuf[200];
2007
        int fl[8];
2008
        int i, j;
2009
2010
        for (n = 0; n < 8; n++)
2011
                fl[n] = 9999;
2012
2013
        memset(&vz, 0, sizeof vz);
2014
2015
        for (n = 0;  n < 999999999; n++) {
2016
                *ibuf = 0;
2017
                for (j = 0; j < 7; j++) {
2018
                        snprintf(strchr(ibuf, 0), 5, "%x",
2019
                            (unsigned)VRND_RandomTestable() & 0xffff);
2020
                        vz.next_in = TRUST_ME(ibuf);
2021
                        vz.avail_in = strlen(ibuf);
2022
                        vz.next_out = TRUST_ME(obuf);
2023
                        vz.avail_out = sizeof obuf;
2024
                        assert(Z_OK == deflateInit2(&vz,
2025
                            9, Z_DEFLATED, 31, 9, Z_DEFAULT_STRATEGY));
2026
                        assert(Z_STREAM_END == deflate(&vz, Z_FINISH));
2027
                        i = vz.stop_bit & 7;
2028
                        if (fl[i] > strlen(ibuf)) {
2029
                                printf("%d %jd <%s>\n", i, vz.stop_bit, ibuf);
2030
                                fl[i] = strlen(ibuf);
2031
                        }
2032
                        assert(Z_OK == deflateEnd(&vz));
2033
                }
2034
        }
2035
2036
        printf("FOO\n");
2037
}
2038
#endif