vinyl-cache/bin/vinyld/cache/cache_deliver_proc.c
0
/*-
1
 * Copyright (c) 2006 Verdens Gang AS
2
 * Copyright (c) 2006-2015 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
31
#include "config.h"
32
33
#include "cache_int.h"
34
#include "cache_filter.h"
35
#include "cache_objhead.h"
36
37
void
38 24
VDP_Panic(struct vsb *vsb, const struct vdp_ctx *vdc)
39
{
40
        struct vdp_entry *vde;
41
42 24
        if (PAN_dump_struct(vsb, vdc, VDP_CTX_MAGIC, "vdc"))
43 16
                return;
44 8
        VSB_printf(vsb, "nxt = %p,\n", vdc->nxt);
45 8
        VSB_printf(vsb, "retval = %d,\n", vdc->retval);
46
47 8
        if (!VTAILQ_EMPTY(&vdc->vdp)) {
48 4
                VSB_cat(vsb, "filters = {\n");
49 4
                VSB_indent(vsb, 2);
50 20
                VTAILQ_FOREACH(vde, &vdc->vdp, list)
51 32
                        VSB_printf(vsb, "%s = %p { priv = %p }\n",
52 16
                            vde->vdp->name, vde, vde->priv);
53 4
                VSB_indent(vsb, -2);
54 4
                VSB_cat(vsb, "},\n");
55 4
        }
56
57 8
        VSB_indent(vsb, -2);
58 8
        VSB_cat(vsb, "},\n");
59 24
}
60
61
/*
62
 * Ensure that transports have called VDP_Close()
63
 * to avoid leaks in VDPs
64
 */
65
void
66 17099
VDP_Fini(const struct vdp_ctx *vdc)
67
{
68 17099
        assert(VTAILQ_EMPTY(&vdc->vdp));
69 17099
}
70
71
void
72 23731
VDP_Init(struct vdp_ctx *vdc, struct worker *wrk, struct vsl_log *vsl,
73
    const struct req *req, const struct busyobj *bo, intmax_t *clen)
74
{
75 23731
        AN(vdc);
76 23731
        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
77 23731
        AN(vsl);
78
79 23731
        AN(clen);
80
81 23731
        assert((req ? 1 : 0) ^ (bo ? 1 : 0));
82
83 23731
        AN(clen);
84 23731
        assert(*clen >= -1);
85
86 23731
        INIT_OBJ(vdc, VDP_CTX_MAGIC);
87 23731
        VTAILQ_INIT(&vdc->vdp);
88 23731
        vdc->wrk = wrk;
89 23731
        vdc->vsl = vsl;
90 23731
        vdc->clen = clen;
91
92 23731
        if (req != NULL) {
93 14986
                CHECK_OBJ(req, REQ_MAGIC);
94 14986
                vdc->oc = req->objcore;
95 14986
                vdc->hp = req->resp;
96 14986
        }
97
        else {
98 8745
                CHECK_OBJ_NOTNULL(bo, BUSYOBJ_MAGIC);
99 8745
                vdc->oc = bo->bereq_body;
100 8745
                vdc->hp = bo->bereq;
101
        }
102 23731
}
103
104
/* VDP_bytes
105
 *
106
 * Pushes len bytes at ptr down the delivery processor list.
107
 *
108
 * This function picks and calls the next delivery processor from the
109
 * list. The return value is the return value of the delivery
110
 * processor. Upon seeing a non-zero return value, that lowest value
111
 * observed is latched in ->retval and all subsequent calls to
112
 * VDP_bytes will return that value directly without calling the next
113
 * processor.
114
 *
115
 * VDP_END marks the end of successful processing, it is issued by
116
 * VDP_DeliverObj() and may also be sent downstream by processors ending the
117
 * stream (for return value != 0)
118
 *
119
 * VDP_END must at most be received once per processor, so any VDP sending it
120
 * downstream must itself not forward it a second time.
121
 *
122
 * Valid return values (of VDP_bytes and any VDP function):
123
 * r < 0:  Error, breaks out early on an error condition
124
 * r == 0: Continue
125
 * r > 0:  Stop, breaks out early without error condition
126
 */
127
128
int
129 108929
VDP_bytes(struct vdp_ctx *vdc, enum vdp_action act,
130
    const void *ptr, ssize_t len)
131
{
132
        int retval;
133
        struct vdp_entry *vdpe;
134
135 108929
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
136 108929
        if (vdc->retval)
137 99
                return (vdc->retval);
138 108830
        vdpe = vdc->nxt;
139 108830
        CHECK_OBJ_NOTNULL(vdpe, VDP_ENTRY_MAGIC);
140
141
        /* at most one VDP_END call */
142 108830
        assert(vdpe->end == VDP_NULL);
143
144 108830
        if (act == VDP_NULL)
145 23208
                assert(len > 0);
146 85622
        else if (act == VDP_END)
147 11586
                vdpe->end = VDP_END;
148
        else
149 74036
                assert(act == VDP_FLUSH);
150
151
        /* Call the present layer, while pointing to the next layer down */
152 108830
        vdc->nxt = VTAILQ_NEXT(vdpe, list);
153 108830
        vdpe->calls++;
154 108830
        vdc->bytes_done = len;
155 108830
        retval = vdpe->vdp->bytes(vdc, act, &vdpe->priv, ptr, len);
156 108830
        vdpe->bytes_in += vdc->bytes_done;
157 108830
        if (retval && (vdc->retval == 0 || retval < vdc->retval))
158 243
                vdc->retval = retval; /* Latch error value */
159 108830
        vdc->nxt = vdpe;
160 108830
        return (vdc->retval);
161 108929
}
162
163
int
164 21484
VDP_Push(VRT_CTX, struct vdp_ctx *vdc, struct ws *ws, const struct vdp *vdp,
165
    void *priv)
166
{
167
        struct vdp_entry *vdpe;
168
169 21484
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
170 21484
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
171 21484
        CHECK_OBJ_ORNULL(vdc->oc, OBJCORE_MAGIC);
172 21484
        CHECK_OBJ_NOTNULL(vdc->hp, HTTP_MAGIC);
173 21484
        AN(vdc->clen);
174 21484
        assert(*vdc->clen >= -1);
175 21484
        AN(ws);
176 21484
        AN(vdp);
177 21484
        AN(vdp->name);
178
179 21484
        if (vdc->retval)
180 0
                return (vdc->retval);
181
182 21484
        if (DO_DEBUG(DBG_PROCESSORS))
183 196
                VSLb(vdc->vsl, SLT_Debug, "VDP_push(%s)", vdp->name);
184
185 21484
        vdpe = WS_Alloc(ws, sizeof *vdpe);
186 21484
        if (vdpe == NULL) {
187 512
                AZ(vdc->retval);
188 512
                vdc->retval = -1;
189 512
                return (vdc->retval);
190
        }
191 20972
        INIT_OBJ(vdpe, VDP_ENTRY_MAGIC);
192 20972
        vdpe->vdp = vdp;
193 20972
        vdpe->priv = priv;
194 20972
        VTAILQ_INSERT_TAIL(&vdc->vdp, vdpe, list);
195 20972
        vdc->nxt = VTAILQ_FIRST(&vdc->vdp);
196
197 20972
        AZ(vdc->retval);
198 20972
        if (vdpe->vdp->init != NULL)
199 19726
                vdc->retval = vdpe->vdp->init(ctx, vdc, &vdpe->priv);
200 20972
        vdc->oc = NULL;
201
202 20972
        if (vdc->retval) {
203 256
                VTAILQ_REMOVE(&vdc->vdp, vdpe, list);
204 256
                vdc->nxt = VTAILQ_FIRST(&vdc->vdp);
205 256
                vdc->retval = vdc->retval > 0 ? 0 : vdc->retval;
206 256
        } else
207 20716
                AN(vdp->bytes);
208 20972
        return (vdc->retval);
209 21484
}
210
211
uint64_t
212 23736
VDP_Close(struct vdp_ctx *vdc, struct objcore *oc, struct boc *boc)
213
{
214
        struct vdp_entry *vdpe;
215 23736
        uint64_t rv = 0;
216
217 23736
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
218 23736
        CHECK_OBJ_NOTNULL(vdc->wrk, WORKER_MAGIC);
219 23736
        CHECK_OBJ_ORNULL(oc, OBJCORE_MAGIC);
220 23736
        CHECK_OBJ_ORNULL(boc, BOC_MAGIC);
221
222 44373
        while ((vdpe = VTAILQ_FIRST(&vdc->vdp)) != NULL) {
223 20637
                CHECK_OBJ(vdpe, VDP_ENTRY_MAGIC);
224 20637
                rv = vdpe->bytes_in;
225 41274
                VSLb(vdc->vsl, SLT_VdpAcct, "%s %ju %ju", vdpe->vdp->name,
226 20637
                    (uintmax_t)vdpe->calls, (uintmax_t)rv);
227 20637
                if (vdpe->vdp->fini != NULL)
228 3848
                        AZ(vdpe->vdp->fini(vdc, &vdpe->priv));
229 20637
                AZ(vdpe->priv);
230 20637
                VTAILQ_REMOVE(&vdc->vdp, vdpe, list);
231 20637
                vdc->nxt = VTAILQ_FIRST(&vdc->vdp);
232
#ifdef VDP_PEDANTIC_ARMED
233
                // enable when we are confident to get VDP_END right
234
                if (vdc->nxt == NULL && vdc->retval >= 0)
235
                        assert(vdpe->end == VDP_END);
236
#endif
237
        }
238 23736
        if (oc != NULL)
239 14989
                HSH_Cancel(vdc->wrk, oc, boc);
240 23736
        return (rv);
241
}
242
243
/*--------------------------------------------------------------------*/
244
245
/*
246
 * Push a VDPIO vdp. This can only be used with only vdpio-enabled VDPs or
247
 * after a successful upgrade
248
 */
249
int
250 513
VDPIO_Push(VRT_CTX, struct vdp_ctx *vdc, struct ws *ws, const struct vdp *vdp,
251
    void *priv)
252
{
253
        struct vdp_entry *vdpe;
254
        int r;
255
256 513
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
257 513
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
258 513
        CHECK_OBJ_ORNULL(vdc->oc, OBJCORE_MAGIC);
259 513
        CHECK_OBJ_NOTNULL(vdc->hp, HTTP_MAGIC);
260 513
        AN(vdc->clen);
261 513
        assert(*vdc->clen >= -1);
262 513
        AN(ws);
263 513
        AN(vdp);
264 513
        AN(vdp->name);
265
266 513
        if (vdc->retval < 0)
267 0
                return (vdc->retval);
268
269 513
        AN(vdp->io_init);
270
271
        // the first VDP (which leases from storage) only gets the minimum
272
        // capacity requirement of 1
273 513
        if (vdc->retval == 0) {
274 0
                assert(VTAILQ_EMPTY(&vdc->vdp));
275 0
                vdc->retval = 1;
276 0
        }
277
278 513
        if (DO_DEBUG(DBG_PROCESSORS))
279 512
                VSLb(vdc->vsl, SLT_Debug, "VDPIO_push(%s)", vdp->name);
280
281 513
        vdpe = WS_Alloc(ws, sizeof *vdpe);
282 513
        if (vdpe == NULL) {
283 0
                vdc->retval = -ENOMEM;
284 0
                return (vdc->retval);
285
        }
286 511
        INIT_OBJ(vdpe, VDP_ENTRY_MAGIC);
287 511
        vdpe->vdp = vdp;
288 511
        vdpe->priv = priv;
289 511
        VTAILQ_INSERT_TAIL(&vdc->vdp, vdpe, list);
290 511
        vdc->nxt = VTAILQ_FIRST(&vdc->vdp);
291
292 511
        assert(vdc->retval > 0);
293 511
        if (vdpe->vdp->io_init != NULL) {
294 511
                r = vdpe->vdp->io_init(ctx, vdc, &vdpe->priv, vdc->retval);
295 511
                if (r <= 0) {
296 0
                        VTAILQ_REMOVE(&vdc->vdp, vdpe, list);
297 0
                        vdc->nxt = VTAILQ_FIRST(&vdc->vdp);
298 0
                }
299
                else
300 511
                        AN(vdp->io_lease);
301 511
                if (r != 0)
302 511
                        vdc->retval = r;
303 511
        }
304 511
        vdc->oc = NULL;
305 511
        return (vdc->retval);
306 511
}
307
308
/*
309
 * upgrade an already initialized VDP filter chain to VDPIO, if possible
310
 * returns:
311
 * > 0 cap
312
 * -ENOTSUP io_upgrade missing for at least one filter
313
 * vdc->retval if < 0
314
 */
315
int
316 160
VDPIO_Upgrade(VRT_CTX, struct vdp_ctx *vdc)
317
{
318
        struct vdp_entry *vdpe;
319
        int cap, r;
320
321 160
        CHECK_OBJ_NOTNULL(ctx, VRT_CTX_MAGIC);
322 160
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
323
324 256
        VTAILQ_FOREACH(vdpe, &vdc->vdp, list)
325 128
                if (vdpe->vdp->io_upgrade == NULL)
326 32
                        return (-ENOTSUP);
327
328 128
        if (vdc->retval < 0)
329 0
                return (vdc->retval);
330
331
        // minimum capacity requirement for the first filter (after storage)
332 128
        r = cap = 1;
333 192
        VTAILQ_FOREACH(vdpe, &vdc->vdp, list) {
334 64
                r = vdpe->vdp->io_upgrade(ctx, vdc, &vdpe->priv, cap);
335 64
                if (DO_DEBUG(DBG_PROCESSORS)) {
336 128
                        VSLb(vdc->vsl, SLT_Debug, "VDPIO_Upgrade "
337
                            "%d = %s(cap = %d)",
338 64
                            r, vdpe->vdp->name, cap);
339 64
                }
340 64
                if (r < 0)
341 0
                        return ((vdc->retval = r));
342
                // XXX remove if filter does not want to be pushed?
343 64
                assert(r != 0);
344 64
                cap = r;
345 64
        }
346 128
        return ((vdc->retval = r));
347 160
}
348
349
uint64_t
350 576
VDPIO_Close1(struct vdp_ctx *vdc, struct vdp_entry *vdpe)
351
{
352
        uint64_t rv;
353
354 576
        CHECK_OBJ_NOTNULL(vdpe, VDP_ENTRY_MAGIC);
355 576
        rv = vdpe->bytes_in;
356 1152
        VSLb(vdc->vsl, SLT_VdpAcct, "%s %ju %ju", vdpe->vdp->name,
357 576
            (uintmax_t)vdpe->calls, (uintmax_t)rv);
358 576
        if (vdpe->vdp->io_fini != NULL)
359 320
                vdpe->vdp->io_fini(vdc, &vdpe->priv);
360 576
        AZ(vdpe->priv);
361 576
        VTAILQ_REMOVE(&vdc->vdp, vdpe, list);
362 576
        vdc->nxt = VTAILQ_FIRST(&vdc->vdp);
363 576
        return (rv);
364
}
365
366
uint64_t
367 128
VDPIO_Close(struct vdp_ctx *vdc, struct objcore *oc, struct boc *boc)
368
{
369
        struct vdp_entry *vdpe;
370 128
        uint64_t rv = 0;
371
372 128
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
373 128
        CHECK_OBJ_NOTNULL(vdc->wrk, WORKER_MAGIC);
374 128
        CHECK_OBJ_ORNULL(oc, OBJCORE_MAGIC);
375 128
        CHECK_OBJ_ORNULL(boc, BOC_MAGIC);
376
377 576
        while ((vdpe = VTAILQ_FIRST(&vdc->vdp)) != NULL)
378 448
                rv = VDPIO_Close1(vdc, vdpe);
379
380 128
        if (oc != NULL)
381 128
                HSH_Cancel(vdc->wrk, oc, boc);
382 128
        return (rv);
383
}
384
385
/*
386
 * ============================================================
387
 * VDPIO helpers: VAI management
388
 *
389
 * Transports should not need to talk to the VAI Object interface directly,
390
 * because its state is kept in the vdp_ctx
391
 *
392
 * So we wrap init, return and fini
393
 */
394
395
// return true if error
396
int
397 128
VDPIO_Init(struct vdp_ctx *vdc, struct objcore *oc, struct ws *ws,
398
    vai_notify_cb *notify_cb, void *notify_priv, struct vscaret *scaret)
399
{
400 128
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
401 128
        VSCARET_CHECK_NOTNULL(scaret);
402 128
        AN(scaret->capacity);
403 128
        AZ(scaret->used);
404
405 128
        AZ(vdc->vai_hdl);
406 128
        vdc->vai_hdl = ObjVAIinit(vdc->wrk, oc, ws, notify_cb, notify_priv);
407 128
        if (vdc->vai_hdl == NULL)
408 0
                return (1);
409 128
        vdc->scaret = scaret;
410 128
        return (0);
411 128
}
412
413
// return leases stashed in scaret
414
void
415 640
VDPIO_Return(const struct vdp_ctx *vdc)
416
{
417 640
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
418
419 640
        ObjVAIreturn(vdc->wrk, vdc->vai_hdl, vdc->scaret);
420 640
}
421
422
// Notify after a VDP's io_lease has return -EAGAIN
423
// usually not called from the io thread, so it has a separate wrk
424
void
425 512
VDPIO_Notify(struct worker *wrk, const struct vdp_ctx *vdc)
426
{
427 512
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
428
429 512
        ObjVAInotify(wrk, vdc->vai_hdl);
430 512
}
431
432
void
433 128
VDPIO_Fini(struct vdp_ctx *vdc)
434
{
435 128
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
436
437 128
        VDPIO_Return(vdc);
438 128
        ObjVAIfini(vdc->wrk, &vdc->vai_hdl);
439 128
}
440
441
/*--------------------------------------------------------------------*/
442
int v_matchproto_(objiterate_f)
443 40334
VDP_ObjIterate(void *priv, unsigned flush, const void *ptr, ssize_t len)
444
{
445
        enum vdp_action act;
446
447 40334
        if (flush == 0)
448 22089
                act = VDP_NULL;
449 18245
        else if ((flush & OBJ_ITER_END) != 0)
450 9916
                act = VDP_END;
451
        else
452 8329
                act = VDP_FLUSH;
453
454 40334
        return (VDP_bytes(priv, act, ptr, len));
455
}
456
457
458
int
459 9632
VDP_DeliverObj(struct vdp_ctx *vdc, struct objcore *oc)
460
{
461
        int r, final;
462
463 9632
        CHECK_OBJ_NOTNULL(vdc, VDP_CTX_MAGIC);
464 9632
        CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
465 9632
        CHECK_OBJ_NOTNULL(vdc->wrk, WORKER_MAGIC);
466 9632
        AN(vdc->vsl);
467 9632
        AZ(vdc->oc);
468 9632
        vdc->hp = NULL;
469 9632
        vdc->clen = NULL;
470 9632
        final = oc->flags & OC_F_TRANSIENT ? 1 : 0;
471 9632
        r = ObjIterate(vdc->wrk, oc, vdc, VDP_ObjIterate, final);
472 9632
        if (r < 0)
473 162
                return (r);
474 9470
        return (0);
475 9632
}