varnish-cache/bin/varnishd/http1/cache_http1_line.c
0
/*-
1
 * Copyright (c) 2006 Verdens Gang AS
2
 * Copyright (c) 2006-2011 Varnish Software AS
3
 * All rights reserved.
4
 *
5
 * Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
6
 *
7
 * SPDX-License-Identifier: BSD-2-Clause
8
 *
9
 * Redistribution and use in source and binary forms, with or without
10
 * modification, are permitted provided that the following conditions
11
 * are met:
12
 * 1. Redistributions of source code must retain the above copyright
13
 *    notice, this list of conditions and the following disclaimer.
14
 * 2. Redistributions in binary form must reproduce the above copyright
15
 *    notice, this list of conditions and the following disclaimer in the
16
 *    documentation and/or other materials provided with the distribution.
17
 *
18
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
22
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28
 * SUCH DAMAGE.
29
 *
30
 * Write data to fd
31
 * We try to use writev() if possible in order to minimize number of
32
 * syscalls made and packets sent.  It also just might allow the worker
33
 * thread to complete the request without holding stuff locked.
34
 *
35
 * XXX: chunked header (generated in Flush) and Tail (EndChunk)
36
 *      are not accounted by means of the size_t returned. Obvious ideas:
37
 *      - add size_t return value to Flush and EndChunk
38
 *      - base accounting on (struct v1l).cnt
39
 */
40
41
#include "config.h"
42
43
#include <sys/uio.h>
44
#include "cache/cache_varnishd.h"
45
#include "cache/cache_filter.h"
46
47
#include <stdio.h>
48
49
#include "cache_http1.h"
50
#include "vtim.h"
51
52
/*--------------------------------------------------------------------*/
53
54
struct v1l {
55
        unsigned                magic;
56
#define V1L_MAGIC               0x2f2142e5
57
        int                     *wfd;
58
        stream_close_t          werr;   /* valid after V1L_Flush() */
59
        struct iovec            *iov;
60
        int                     siov;
61
        int                     niov;
62
        size_t                  liov;
63
        size_t                  cliov;
64
        int                     ciov;   /* Chunked header marker */
65
        vtim_real               deadline;
66
        struct vsl_log          *vsl;
67
        uint64_t                cnt;    /* Flushed byte count */
68
        struct ws               *ws;
69
        uintptr_t               ws_snap;
70
        void                    **vdp_priv;
71
};
72
73
/*--------------------------------------------------------------------
74
 * for niov == 0, reserve the ws for max number of iovs
75
 * otherwise, up to niov
76
 */
77
78
struct v1l *
79 10601
V1L_Open(struct ws *ws, int *fd, struct vsl_log *vsl,
80
    vtim_real deadline, unsigned niov)
81
{
82
        struct v1l *v1l;
83
        unsigned u;
84
        uintptr_t ws_snap;
85
        size_t sz;
86
87 10601
        if (WS_Overflowed(ws))
88 0
                return (NULL);
89
90 10601
        if (niov != 0)
91 6289
                assert(niov >= 3);
92
93 10601
        ws_snap = WS_Snapshot(ws);
94
95 10601
        v1l = WS_Alloc(ws, sizeof *v1l);
96 10601
        if (v1l == NULL)
97 2
                return (NULL);
98 10599
        INIT_OBJ(v1l, V1L_MAGIC);
99
100 10599
        v1l->ws = ws;
101 10599
        v1l->ws_snap = ws_snap;
102
103 10599
        u = WS_ReserveLumps(ws, sizeof(struct iovec));
104 10599
        if (u < 3) {
105
                /* Must have at least 3 in case of chunked encoding */
106 0
                WS_Release(ws, 0);
107 0
                WS_MarkOverflow(ws);
108 0
                return (NULL);
109
        }
110 10599
        if (u > IOV_MAX)
111 144
                u = IOV_MAX;
112 10599
        if (niov != 0 && u > niov)
113 6136
                u = niov;
114 10599
        v1l->iov = WS_Reservation(ws);
115 10599
        v1l->siov = (int)u;
116 10599
        v1l->ciov = (int)u;
117 10599
        v1l->wfd = fd;
118 10599
        v1l->deadline = deadline;
119 10599
        v1l->vsl = vsl;
120 10599
        v1l->werr = SC_NULL;
121
122 10599
        sz = u * sizeof(struct iovec);
123 10599
        assert(sz < UINT_MAX);
124 10599
        WS_Release(ws, (unsigned)sz);
125 10599
        return (v1l);
126 10601
}
127
128
void
129 144
V1L_NoRollback(struct v1l *v1l)
130
{
131
132 144
        CHECK_OBJ_NOTNULL(v1l, V1L_MAGIC);
133 144
        v1l->ws_snap = 0;
134 144
}
135
136
stream_close_t
137 10600
V1L_Close(struct v1l **v1lp, uint64_t *cnt)
138
{
139
        struct v1l *v1l;
140
        struct ws *ws;
141
        uintptr_t ws_snap;
142
        stream_close_t sc;
143
144 10600
        AN(cnt);
145 10600
        TAKE_OBJ_NOTNULL(v1l, v1lp, V1L_MAGIC);
146 10600
        if (v1l->vdp_priv != NULL) {
147 8330
                assert(*v1l->vdp_priv == v1l);
148 8330
                *v1l->vdp_priv = NULL;
149 8330
        }
150 10600
        sc = V1L_Flush(v1l);
151 10600
        *cnt = v1l->cnt;
152 10600
        ws = v1l->ws;
153 10600
        ws_snap = v1l->ws_snap;
154 10600
        ZERO_OBJ(v1l, sizeof *v1l);
155 10600
        if (ws_snap != 0)
156 10455
                WS_Rollback(ws, ws_snap);
157 10600
        return (sc);
158
}
159
160
static void
161 78
v1l_prune(struct v1l *v1l, ssize_t abytes)
162
{
163 78
        size_t used = 0;
164
        size_t sz, bytes, used_here;
165
        int j;
166
167 78
        assert(abytes > 0);
168 78
        bytes = (size_t)abytes;
169
170 226
        for (j = 0; j < v1l->niov; j++) {
171 226
                if (used + v1l->iov[j].iov_len > bytes) {
172
                        /* Cutoff is in this iov */
173 78
                        used_here = bytes - used;
174 78
                        v1l->iov[j].iov_len -= used_here;
175 78
                        v1l->iov[j].iov_base =
176 78
                            (char*)v1l->iov[j].iov_base + used_here;
177 78
                        sz = (unsigned)v1l->niov - (unsigned)j;
178 78
                        sz *= sizeof(struct iovec);
179 78
                        memmove(v1l->iov, &v1l->iov[j], sz);
180 78
                        v1l->niov -= j;
181 78
                        assert(v1l->liov >= bytes);
182 78
                        v1l->liov -= bytes;
183 78
                        return;
184
                }
185 148
                used += v1l->iov[j].iov_len;
186 148
        }
187 0
        AZ(v1l->liov);
188 78
}
189
190
stream_close_t
191 47118
V1L_Flush(struct v1l *v1l)
192
{
193
        ssize_t i;
194
        size_t sz;
195
        int err;
196
        char cbuf[32];
197
198 47118
        CHECK_OBJ_NOTNULL(v1l, V1L_MAGIC);
199 47118
        CHECK_OBJ_NOTNULL(v1l->werr, STREAM_CLOSE_MAGIC);
200 47118
        AN(v1l->wfd);
201
202 47118
        assert(v1l->niov <= v1l->siov);
203
204 47118
        if (*v1l->wfd >= 0 && v1l->liov > 0 && v1l->werr == SC_NULL) {
205 40195
                if (v1l->ciov < v1l->siov && v1l->cliov > 0) {
206
                        /* Add chunk head & tail */
207 2163
                        bprintf(cbuf, "00%zx\r\n", v1l->cliov);
208 2163
                        sz = strlen(cbuf);
209 2163
                        v1l->iov[v1l->ciov].iov_base = cbuf;
210 2163
                        v1l->iov[v1l->ciov].iov_len = sz;
211 2163
                        v1l->liov += sz;
212
213
                        /* This is OK, because siov was --'ed */
214 2163
                        v1l->iov[v1l->niov].iov_base = cbuf + sz - 2;
215 2163
                        v1l->iov[v1l->niov++].iov_len = 2;
216 2163
                        v1l->liov += 2;
217 40195
                } else if (v1l->ciov < v1l->siov) {
218 136
                        v1l->iov[v1l->ciov].iov_base = cbuf;
219 136
                        v1l->iov[v1l->ciov].iov_len = 0;
220 136
                }
221
222 40195
                i = 0;
223 40195
                err = 0;
224 40195
                do {
225 40326
                        if (VTIM_real() > v1l->deadline) {
226 16
                                VSLb(v1l->vsl, SLT_Debug,
227
                                    "Hit total send timeout, "
228
                                    "wrote = %zd/%zd; not retrying",
229 8
                                    i, v1l->liov);
230 8
                                i = -1;
231 8
                                break;
232
                        }
233
234 40318
                        i = writev(*v1l->wfd, v1l->iov, v1l->niov);
235 40318
                        if (i > 0) {
236 40249
                                v1l->cnt += (size_t)i;
237 40249
                                if ((size_t)i == v1l->liov)
238 40171
                                        break;
239 78
                        }
240
241
                        /* we hit a timeout, and some data may have been sent:
242
                         * Remove sent data from start of I/O vector, then retry
243
                         *
244
                         * XXX: Add a "minimum sent data per timeout counter to
245
                         * prevent slowloris attacks
246
                         */
247
248 147
                        err = errno;
249
250 147
                        if (err == EWOULDBLOCK) {
251 106
                                VSLb(v1l->vsl, SLT_Debug,
252
                                    "Hit idle send timeout, "
253
                                    "wrote = %zd/%zd; retrying",
254 53
                                    i, v1l->liov);
255 53
                        }
256
257 147
                        if (i > 0)
258 78
                                v1l_prune(v1l, i);
259 147
                } while (i > 0 || err == EWOULDBLOCK);
260
261 40195
                if (i <= 0) {
262 48
                        VSLb(v1l->vsl, SLT_Debug,
263
                            "Write error, retval = %zd, len = %zd, errno = %s",
264 24
                            i, v1l->liov, VAS_errtxt(err));
265 24
                        assert(v1l->werr == SC_NULL);
266 24
                        if (err == EPIPE)
267 15
                                v1l->werr = SC_REM_CLOSE;
268
                        else
269 9
                                v1l->werr = SC_TX_ERROR;
270 24
                        errno = err;
271 24
                }
272 40195
        }
273 47120
        v1l->liov = 0;
274 47120
        v1l->cliov = 0;
275 47120
        v1l->niov = 0;
276 47120
        if (v1l->ciov < v1l->siov)
277 4226
                v1l->ciov = v1l->niov++;
278 47120
        CHECK_OBJ_NOTNULL(v1l->werr, STREAM_CLOSE_MAGIC);
279 47120
        return (v1l->werr);
280
}
281
282
size_t
283 264201
V1L_Write(struct v1l *v1l, const void *ptr, ssize_t alen)
284
{
285 264201
        size_t len = 0;
286
287 264201
        CHECK_OBJ_NOTNULL(v1l, V1L_MAGIC);
288 264201
        AN(v1l->wfd);
289 264201
        if (alen == 0 || *v1l->wfd < 0)
290 52
                return (0);
291 264201
        if (alen > 0)
292 143444
                len = (size_t)alen;
293 120757
        else if (alen == -1)
294 120757
                len = strlen(ptr);
295
        else
296 0
                WRONG("alen");
297
298 264201
        assert(v1l->niov < v1l->siov);
299 264201
        v1l->iov[v1l->niov].iov_base = TRUST_ME(ptr);
300 264201
        v1l->iov[v1l->niov].iov_len = len;
301 264201
        v1l->liov += len;
302 264201
        v1l->niov++;
303 264201
        v1l->cliov += len;
304 264201
        if (v1l->niov >= v1l->siov) {
305 220
                (void)V1L_Flush(v1l);
306 220
                VSC_C_main->http1_iovs_flush++;
307 220
        }
308 264201
        return (len);
309 264201
}
310
311
void
312 586
V1L_Chunked(struct v1l *v1l)
313
{
314
315 586
        CHECK_OBJ_NOTNULL(v1l, V1L_MAGIC);
316
317 586
        assert(v1l->ciov == v1l->siov);
318 586
        assert(v1l->siov >= 3);
319
        /*
320
         * If there is no space for chunked header, a chunk of data and
321
         * a chunk tail, we might as well flush right away.
322
         */
323 586
        if (v1l->niov + 3 >= v1l->siov) {
324 0
                (void)V1L_Flush(v1l);
325 0
                VSC_C_main->http1_iovs_flush++;
326 0
        }
327 586
        v1l->siov--;
328 586
        v1l->ciov = v1l->niov++;
329 586
        v1l->cliov = 0;
330 586
        assert(v1l->ciov < v1l->siov);
331 586
        assert(v1l->niov < v1l->siov);
332 586
}
333
334
/*
335
 * XXX: It is not worth the complexity to attempt to get the
336
 * XXX: end of chunk into the V1L_Flush(), because most of the time
337
 * XXX: if not always, that is a no-op anyway, because the calling
338
 * XXX: code already called V1L_Flush() to release local storage.
339
 */
340
341
void
342 544
V1L_EndChunk(struct v1l *v1l)
343
{
344
345 544
        CHECK_OBJ_NOTNULL(v1l, V1L_MAGIC);
346
347 544
        assert(v1l->ciov < v1l->siov);
348 544
        (void)V1L_Flush(v1l);
349 544
        v1l->siov++;
350 544
        v1l->ciov = v1l->siov;
351 544
        v1l->niov = 0;
352 544
        v1l->cliov = 0;
353 544
        (void)V1L_Write(v1l, "0\r\n\r\n", -1);
354 544
}
355
356
/*--------------------------------------------------------------------
357
 * VDP using V1L
358
 */
359
360
/* remember priv pointer for V1L_Close() to clear */
361
static int v_matchproto_(vdp_init_f)
362 8265
v1l_init(VRT_CTX, struct vdp_ctx *vdc, void **priv)
363
{
364
        struct v1l *v1l;
365
366 8265
        (void) ctx;
367 8265
        (void) vdc;
368 8265
        AN(priv);
369 8265
        CAST_OBJ_NOTNULL(v1l, *priv, V1L_MAGIC);
370
371 8265
        v1l->vdp_priv = priv;
372 8265
        return (0);
373
}
374
375
static int v_matchproto_(vdp_bytes_f)
376 36189
v1l_bytes(struct vdp_ctx *vdc, enum vdp_action act, void **priv,
377
    const void *ptr, ssize_t len)
378
{
379 36189
        size_t wl = 0;
380
381 36189
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
382 36189
        AN(priv);
383
384 36189
        AZ(vdc->nxt);           /* always at the bottom of the pile */
385
386 36189
        if (len > 0)
387 33453
                wl = V1L_Write(*priv, ptr, len);
388 36189
        if (act > VDP_NULL && V1L_Flush(*priv) != SC_NULL)
389 24
                return (-1);
390 36165
        if ((size_t)len != wl)
391 0
                return (-1);
392 36165
        return (0);
393 36189
}
394
395
/*--------------------------------------------------------------------
396
 * VDPIO using V1L
397
 *
398
 * this is deliverately half-baked to reduce work in progress while heading
399
 * towards VAI/VDPIO: we update the v1l with the scarab, which we
400
 * return unmodified.
401
 *
402
 */
403
404
/* remember priv pointer for V1L_Close() to clear */
405
static int v_matchproto_(vpio_init_f)
406 64
v1l_io_init(VRT_CTX, struct vdp_ctx *vdc, void **priv, int capacity)
407
{
408
        struct v1l *v1l;
409
410 64
        (void) ctx;
411 64
        (void) vdc;
412 64
        AN(priv);
413
414 64
        CAST_OBJ_NOTNULL(v1l, *priv, V1L_MAGIC);
415
416 64
        v1l->vdp_priv = priv;
417 64
        return (capacity);
418
}
419
420
static int v_matchproto_(vpio_init_f)
421 0
v1l_io_upgrade(VRT_CTX, struct vdp_ctx *vdc, void **priv, int capacity)
422
{
423 0
        return (v1l_io_init(ctx, vdc, priv, capacity));
424
}
425
426
/*
427
 * API note
428
 *
429
 * this VDP is special in that it does not transform data, but prepares
430
 * the write. From the perspective of VDPIO, its current state is only
431
 * transitional.
432
 *
433
 * Because the VDP prepares the actual writes, but the caller needs
434
 * to return the scarab's leases, the caller in this case is
435
 * required to empty the scarab after V1L_Flush()'ing.
436
 */
437
438
static int v_matchproto_(vdpio_lease_f)
439 140
v1l_io_lease(struct vdp_ctx *vdc, struct vdp_entry *this, struct vscarab *scarab)
440
{
441
        struct v1l *v1l;
442
        struct viov *v;
443
        int r;
444
445 140
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
446 140
        CHECK_OBJ_NOTNULL(this, VDP_ENTRY_MAGIC);
447 140
        CAST_OBJ_NOTNULL(v1l, this->priv, V1L_MAGIC);
448 140
        VSCARAB_CHECK(scarab);
449 140
        AZ(scarab->used);       // see note above
450 140
        this->calls++;
451 140
        r = vdpio_pull(vdc, this, scarab);
452 140
        if (r < 0)
453 6
                return (r);
454 262
        VSCARAB_FOREACH(v, scarab)
455 128
                this->bytes_in += V1L_Write(v1l, v->iov.iov_base, v->iov.iov_len);
456 134
        return (r);
457 140
}
458
459
const struct vdp * const VDP_v1l = &(struct vdp){
460
        .name =         "V1B",
461
        .init =         v1l_init,
462
        .bytes =        v1l_bytes,
463
464
        .io_init =      v1l_io_init,
465
        .io_upgrade =   v1l_io_upgrade,
466
        .io_lease =     v1l_io_lease,
467
};