GCOV - Varnish Cache


/*-
 * Copyright (c) 2013-2016 Varnish Software AS
 * All rights reserved.
 *
 * Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 * Lifetime of an objcore:
 *      phase 0 - nonexistent
 *      phase 1 - created, but no stevedore associated
 *      phase 2 - stevedore associated, being filled out
 *      phase 3 - stable, no changes happening
 *      phase 4 - unavailable, being dismantled
 *      phase 5 - stevedore disassociated
 *      phase 6 - nonexistent
 *
 * 0->1 ObjNew()        creates objcore
 *
 * 1->2 STV_NewObject() associates a stevedore
 *
 * 2    ObjSetState()   sets state
 * 2    ObjWaitState()  waits for particular state
 *                      INVALID->REQ_DONE->STREAM->FINISHED->FAILED
 *
 * 2    ObjGetSpace()   allocates space
 * 2    ObjExtend()     commits content
 * 2    ObjWaitExtend() waits for content - used to implement ObjIterate())
 *
 * 2    ObjSetAttr()
 * 2      ObjCopyAttr()
 * 2      ObjSetFlag()
 * 2      ObjSetDouble()
 * 2      ObjSetU32()
 * 2      ObjSetU64()
 *
 * 2->3 ObjBocDone()    Boc removed from OC, clean it up
 *
 * 23   ObjHasAttr()
 * 23   ObjGetAttr()
 * 23     ObjCheckFlag()
 * 23     ObjGetDouble()
 * 23     ObjGetU32()
 * 23     ObjGetU64()
 * 23     ObjGetLen()
 * 23     ObjGetXID()
 *
 * 23   ObjIterate()    ... over body
 *
 * 23   ObjTouch()      Signal to LRU(-like) facilities
 *
 * 3->4 HSH_Snipe()     kill if not in use
 * 3->4 HSH_Kill()      make unavailable
 *
 * 234  ObjSlim()       Release body storage (but retain attribute storage)
 *
 * 4->5 ObjFreeObj()    disassociates stevedore
 *
 * 5->6 FREE_OBJ()      ...in HSH_DerefObjCore()
 */

#include "config.h"

#include <stdlib.h>

#include "cache_varnishd.h"
#include "cache_obj.h"
#include "vend.h"
#include "storage/storage.h"

static const struct obj_methods *
obj_getmethods(const struct objcore *oc)
{

        CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
        CHECK_OBJ_NOTNULL(oc->stobj->stevedore, STEVEDORE_MAGIC);
        AN(oc->stobj->stevedore->methods);
        return (oc->stobj->stevedore->methods);
}

static struct boc *
obj_newboc(void)
{
        struct boc *boc;

        ALLOC_OBJ(boc, BOC_MAGIC);
        AN(boc);
        Lck_New(&boc->mtx, lck_busyobj);
        PTOK(pthread_cond_init(&boc->cond, NULL));
        boc->refcount = 1;
        boc->transit_buffer = cache_param->transit_buffer;
        return (boc);
}

static void
obj_deleteboc(struct boc **p)
{
        struct boc *boc;

        TAKE_OBJ_NOTNULL(boc, p, BOC_MAGIC);
        Lck_Delete(&boc->mtx);
        PTOK(pthread_cond_destroy(&boc->cond));
        free(boc->vary);
        FREE_OBJ(boc);
}

/*====================================================================
 * ObjNew()
 *
 */

struct objcore *
ObjNew(const struct worker *wrk)
{
        struct objcore *oc;

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);

        ALLOC_OBJ(oc, OBJCORE_MAGIC);
        AN(oc);
        wrk->stats->n_objectcore++;
        oc->last_lru = NAN;
        oc->flags = OC_F_BUSY;

        oc->boc = obj_newboc();

        return (oc);
}

/*====================================================================
 * ObjDestroy()
 *
 */

void
ObjDestroy(const struct worker *wrk, struct objcore **p)
{
        struct objcore *oc;

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        TAKE_OBJ_NOTNULL(oc, p, OBJCORE_MAGIC);
        if (oc->boc != NULL)
                obj_deleteboc(&oc->boc);
        FREE_OBJ(oc);
        wrk->stats->n_objectcore--;
}

/*====================================================================
 * ObjIterate()
 *
 */

int
ObjIterate(struct worker *wrk, struct objcore *oc,
    void *priv, objiterate_f *func, int final)
{
        const struct obj_methods *om = obj_getmethods(oc);

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        AN(func);
        AN(om->objiterator);
        return (om->objiterator(wrk, oc, priv, func, final));
}

/*====================================================================
 * ObjGetSpace()
 *
 * This function returns a pointer and length of free space.  If there
 * is no free space, some will be added first.
 *
 * The "sz" argument is an input hint of how much space is desired.
 * 0 means "unknown", return some default size (maybe fetch_chunksize)
 */

int
ObjGetSpace(struct worker *wrk, struct objcore *oc, ssize_t *sz, uint8_t **ptr)
{
        const struct obj_methods *om = obj_getmethods(oc);

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
        AN(sz);
        AN(ptr);
        assert(*sz >= 0);

        AN(om->objgetspace);
        return (om->objgetspace(wrk, oc, sz, ptr));
}

/*====================================================================
 * ObjExtend()
 *
 * This function extends the used part of the object a number of bytes
 * into the last space returned by ObjGetSpace()
 *
 * The final flag must be set on the last call, and it will release any
 * surplus space allocated.
 */

static void
obj_extend_condwait(const struct objcore *oc)
{

        if (oc->boc->transit_buffer == 0)
                return;

        assert(oc->flags & OC_F_TRANSIENT);
        while (!(oc->flags & OC_F_CANCEL) && oc->boc->fetched_so_far >
            oc->boc->delivered_so_far + oc->boc->transit_buffer)
                (void)Lck_CondWait(&oc->boc->cond, &oc->boc->mtx);
}

void
ObjExtend(struct worker *wrk, struct objcore *oc, ssize_t l, int final)
{
        const struct obj_methods *om = obj_getmethods(oc);

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
        AN(om->objextend);
        assert(l >= 0);

        Lck_Lock(&oc->boc->mtx);
        if (l > 0) {
                obj_extend_condwait(oc);
                om->objextend(wrk, oc, l);
                oc->boc->fetched_so_far += l;
                PTOK(pthread_cond_broadcast(&oc->boc->cond));
        }
        Lck_Unlock(&oc->boc->mtx);

        assert(oc->boc->state < BOS_FINISHED);
        if (final && om->objtrimstore != NULL)
                om->objtrimstore(wrk, oc);
}

/*====================================================================
 */

uint64_t
ObjWaitExtend(const struct worker *wrk, const struct objcore *oc, uint64_t l,
    enum boc_state_e *statep)
{
        enum boc_state_e state;
        uint64_t rv;

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
        CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
        Lck_Lock(&oc->boc->mtx);
        while (1) {
                rv = oc->boc->fetched_so_far;
                assert(l <= rv || oc->boc->state == BOS_FAILED);
                if (oc->boc->transit_buffer > 0) {
                        assert(oc->flags & OC_F_TRANSIENT);
                        /* Signal the new client position */
                        oc->boc->delivered_so_far = l;
                        PTOK(pthread_cond_signal(&oc->boc->cond));
                }
                state = oc->boc->state;
                if (rv > l || state >= BOS_FINISHED)
                        break;
                (void)Lck_CondWait(&oc->boc->cond, &oc->boc->mtx);
        }
        Lck_Unlock(&oc->boc->mtx);
        if (statep != NULL)
                *statep = state;
        return (rv);
}
/*====================================================================
 */

void
ObjSetState(struct worker *wrk, const struct objcore *oc,
    enum boc_state_e next)
{
        const struct obj_methods *om;

        CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
        CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
        assert(next > oc->boc->state);

        CHECK_OBJ_ORNULL(oc->stobj->stevedore, STEVEDORE_MAGIC);
        assert(next != BOS_STREAM || oc->boc->state == BOS_PREP_STREAM);
        assert(next != BOS_FINISHED || (oc->oa_present & (1 << OA_LEN)));

        if (oc->stobj->stevedore != NULL) {
                om = oc->stobj->stevedore->methods;
                if (om->objsetstate != NULL)
                        om->objsetstate(wrk, oc, next);
        }

        Lck_Lock(&oc->boc->mtx);
        oc->boc->state = next;
        PTOK(pthread_cond_broadcast(&oc->boc->cond));
        Lck_Unlock(&oc->boc->mtx);
}

/*====================================================================
 */

void
ObjWaitState(const struct objcore *oc, enum boc_state_e want)
{

        CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
        CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);

        Lck_Lock(&oc->boc->mtx);
        /* wake up obj_extend_condwait() */
        if (oc->flags & OC_F_CANCEL)
                PTOK(pthread_cond_signal(&oc->boc->cond));
        while (1) {
                if (oc->boc->state >= want)
                        break;
                (void)Lck_CondWait(&oc->boc->cond, &oc->boc->mtx);
        }
        Lck_Unlock(&oc->boc->mtx);
}

/*====================================================================
 * ObjGetlen()
 *
 * This is a separate function because it may need locking
 */

uint64_t
ObjGetLen(struct worker *wrk, struct objcore *oc)
{
        uint64_t len;

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);

        AZ(ObjGetU64(wrk, oc, OA_LEN, &len));
        return (len);
}

/*====================================================================
 * ObjSlim()
 *
 * Free the whatever storage can be freed, without freeing the actual
 * object yet.
 */

void
ObjSlim(struct worker *wrk, struct objcore *oc)
{
        const struct obj_methods *om = obj_getmethods(oc);

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);

        if (om->objslim != NULL)
                om->objslim(wrk, oc);
}

/*====================================================================
 * Called when the boc used to populate the objcore is going away.
 * Useful for releasing any leftovers from Trim.
 */

void
ObjBocDone(struct worker *wrk, struct objcore *oc, struct boc **boc)
{
        const struct obj_methods *m;

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
        AN(boc);
        CHECK_OBJ_NOTNULL(*boc, BOC_MAGIC);
        CHECK_OBJ_ORNULL(oc->stobj->stevedore, STEVEDORE_MAGIC);
        if (oc->stobj->stevedore != NULL) {
                m = obj_getmethods(oc);
                if (m->objbocdone != NULL)
                        m->objbocdone(wrk, oc, *boc);
        }
        obj_deleteboc(boc);
}

/*====================================================================
 */
void
ObjFreeObj(struct worker *wrk, struct objcore *oc)
{
        const struct obj_methods *m = obj_getmethods(oc);

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);

        AN(m->objfree);
        m->objfree(wrk, oc);
        AZ(oc->stobj->stevedore);
}

/*====================================================================
 * ObjHasAttr()
 *
 * Check if object has this attribute
 */

int
ObjHasAttr(struct worker *wrk, struct objcore *oc, enum obj_attr attr)
{

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);

        if (oc->oa_present)
                return (oc->oa_present & (1 << attr));

        /* resurrected persistent objects don't have oa_present set */
        return (ObjGetAttr(wrk, oc, attr, NULL) != NULL ? 1 : 0);
}

/*====================================================================
 * ObjGetAttr()
 *
 * Get an attribute of the object.
 *
 * Returns NULL on unset or zero length attributes and len set to
 * zero. Returns Non-NULL otherwise and len is updated with the attributes
 * length.
 */

const void *
ObjGetAttr(struct worker *wrk, struct objcore *oc, enum obj_attr attr,
   ssize_t *len)
{
        const struct obj_methods *om = obj_getmethods(oc);

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);

        AN(om->objgetattr);
        return (om->objgetattr(wrk, oc, attr, len));
}

/*====================================================================
 * ObjSetAttr()
 *
 * Setting fixed size attributes always succeeds.
 *
 * Setting a variable size attribute asserts if the combined size of the
 * variable attributes exceeds the total variable attribute space set at
 * object creation. If there is space it always succeeds.
 *
 * Setting an auxiliary attribute can fail.
 *
 * Resetting any variable asserts if the new length does not match the
 * previous length exactly.
 *
 * If ptr is Non-NULL, it points to the new content which is copied into
 * the attribute.  Otherwise the caller will have to do the copying.
 *
 * Return value is non-NULL on success and NULL on failure. If ptr was
 * non-NULL, it is an error to use the returned pointer to set the
 * attribute data, it is only a success indicator in that case.
 */

void *
ObjSetAttr(struct worker *wrk, struct objcore *oc, enum obj_attr attr,
    ssize_t len, const void *ptr)
{
        const struct obj_methods *om = obj_getmethods(oc);
        void *r;

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);

        AN(om->objsetattr);
        assert((int)attr < 16);
        r = om->objsetattr(wrk, oc, attr, len, ptr);
        if (r)
                oc->oa_present |= (1 << attr);
        return (r);
}

/*====================================================================
 * ObjTouch()
 */

void
ObjTouch(struct worker *wrk, struct objcore *oc, vtim_real now)
{
        const struct obj_methods *om = obj_getmethods(oc);

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        if (om->objtouch != NULL)
                om->objtouch(wrk, oc, now);
}

/*====================================================================
 * Utility functions which work on top of the previous ones
 */

int
ObjCopyAttr(struct worker *wrk, struct objcore *oc, struct objcore *ocs,
    enum obj_attr attr)
{
        const void *vps;
        void *vpd;
        ssize_t l;

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
        CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
        CHECK_OBJ_NOTNULL(ocs, OBJCORE_MAGIC);

        vps = ObjGetAttr(wrk, ocs, attr, &l);
        // XXX: later we want to have zero-length OA's too
        if (vps == NULL || l <= 0)
                return (-1);
        vpd = ObjSetAttr(wrk, oc, attr, l, vps);
        if (vpd == NULL)
                return (-1);
        return (0);
}

int
ObjSetXID(struct worker *wrk, struct objcore *oc, vxid_t xid)
{
        uint64_t u;

        u = VXID(xid);
        AZ(ObjSetU64(wrk, oc, OA_VXID, u));
        return (0);
}


vxid_t
ObjGetXID(struct worker *wrk, struct objcore *oc)
{
        vxid_t u;

        AZ(ObjGetU64(wrk, oc, OA_VXID, &u.vxid));
        return (u);
}

/*--------------------------------------------------------------------
 * There is no well-defined byteorder for IEEE-754 double and the
 * correct solution (frexp(3) and manual encoding) is more work
 * than our (weak) goal of being endian-agnostic requires at this point.
 * We give it a shot by memcpy'ing doubles over a uint64_t and then
 * BE encode that.
 */

int
ObjSetDouble(struct worker *wrk, struct objcore *oc, enum obj_attr a, double t)
{
        void *vp;
        uint64_t u;

        assert(sizeof t == sizeof u);
        memcpy(&u, &t, sizeof u);
        vp = ObjSetAttr(wrk, oc, a, sizeof u, NULL);
        if (vp == NULL)
                return (-1);
        vbe64enc(vp, u);
        return (0);
}

int
ObjGetDouble(struct worker *wrk, struct objcore *oc, enum obj_attr a, double *d)
{
        const void *vp;
        uint64_t u;
        ssize_t l;

        assert(sizeof *d == sizeof u);
        vp = ObjGetAttr(wrk, oc, a, &l);
        if (vp == NULL)
                return (-1);
        if (d != NULL) {
                assert(l == sizeof u);
                u = vbe64dec(vp);
                memcpy(d, &u, sizeof *d);
        }
        return (0);
}

/*--------------------------------------------------------------------
 */

int
ObjSetU64(struct worker *wrk, struct objcore *oc, enum obj_attr a, uint64_t t)
{
        void *vp;

        vp = ObjSetAttr(wrk, oc, a, sizeof t, NULL);
        if (vp == NULL)
                return (-1);
        vbe64enc(vp, t);
        return (0);
}

int
ObjGetU64(struct worker *wrk, struct objcore *oc, enum obj_attr a, uint64_t *d)
{
        const void *vp;
        ssize_t l;

        vp = ObjGetAttr(wrk, oc, a, &l);
        if (vp == NULL || l != sizeof *d)
                return (-1);
        if (d != NULL)
                *d = vbe64dec(vp);
        return (0);
}

/*--------------------------------------------------------------------
 */

int
ObjCheckFlag(struct worker *wrk, struct objcore *oc, enum obj_flags of)
{
        const uint8_t *fp;

        fp = ObjGetAttr(wrk, oc, OA_FLAGS, NULL);
        AN(fp);
        return ((*fp) & of);
}

void
ObjSetFlag(struct worker *wrk, struct objcore *oc, enum obj_flags of, int val)
{
        uint8_t *fp;

        fp = ObjSetAttr(wrk, oc, OA_FLAGS, 1, NULL);
        AN(fp);
        if (val)
                (*fp) |= of;
        else
                (*fp) &= ~of;
}

/*====================================================================
 * Object event subscription mechanism.
 *
 * XXX: it is extremely unclear what the locking circumstances are here.
 */

struct oev_entry {
        unsigned                        magic;
#define OEV_MAGIC                       0xb0b7c5a1
        unsigned                        mask;
        obj_event_f                     *func;
        void                            *priv;
        VTAILQ_ENTRY(oev_entry)         list;
};

static VTAILQ_HEAD(,oev_entry)          oev_list;
static pthread_rwlock_t                 oev_rwl;
static unsigned                         oev_mask;

/*
 * NB: ObjSubscribeEvents() is not atomic:
 * oev_mask is checked optimistically in ObjSendEvent()
 */
uintptr_t
ObjSubscribeEvents(obj_event_f *func, void *priv, unsigned mask)
{
        struct oev_entry *oev;

        AN(func);
        AZ(mask & ~OEV_MASK);

        ALLOC_OBJ(oev, OEV_MAGIC);
        AN(oev);
        oev->func = func;
        oev->priv = priv;
        oev->mask = mask;
        PTOK(pthread_rwlock_wrlock(&oev_rwl));
        VTAILQ_INSERT_TAIL(&oev_list, oev, list);
        oev_mask |= mask;
        PTOK(pthread_rwlock_unlock(&oev_rwl));
        return ((uintptr_t)oev);
}

void
ObjUnsubscribeEvents(uintptr_t *handle)
{
        struct oev_entry *oev, *oev2 = NULL;
        unsigned newmask = 0;

        AN(handle);
        AN(*handle);
        PTOK(pthread_rwlock_wrlock(&oev_rwl));
        VTAILQ_FOREACH(oev, &oev_list, list) {
                CHECK_OBJ_NOTNULL(oev, OEV_MAGIC);
                if ((uintptr_t)oev == *handle)
                        oev2 = oev;
                else
                        newmask |= oev->mask;
        }
        AN(oev2);
        VTAILQ_REMOVE(&oev_list, oev2, list);
        oev_mask = newmask;
        AZ(newmask & ~OEV_MASK);
        PTOK(pthread_rwlock_unlock(&oev_rwl));
        FREE_OBJ(oev2);
        *handle = 0;
}

void
ObjSendEvent(struct worker *wrk, struct objcore *oc, unsigned event)
{
        struct oev_entry *oev;

        CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
        CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
        AN(event & OEV_MASK);
        AZ(event & ~OEV_MASK);
        if (!(event & oev_mask))
                return;

        PTOK(pthread_rwlock_rdlock(&oev_rwl));
        VTAILQ_FOREACH(oev, &oev_list, list) {
                CHECK_OBJ_NOTNULL(oev, OEV_MAGIC);
                if (event & oev->mask)
                        oev->func(wrk, oev->priv, oc, event);
        }
        PTOK(pthread_rwlock_unlock(&oev_rwl));

}

void
ObjInit(void)
{
        VTAILQ_INIT(&oev_list);
        PTOK(pthread_rwlock_init(&oev_rwl, NULL));
}

		varnish-cache/bin/varnishd/cache/cache_obj.c
0		/*-
1		* Copyright (c) 2013-2016 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		* Lifetime of an objcore:
30		* phase 0 - nonexistent
31		* phase 1 - created, but no stevedore associated
32		* phase 2 - stevedore associated, being filled out
33		* phase 3 - stable, no changes happening
34		* phase 4 - unavailable, being dismantled
35		* phase 5 - stevedore disassociated
36		* phase 6 - nonexistent
37		*
38		* 0->1 ObjNew() creates objcore
39		*
40		* 1->2 STV_NewObject() associates a stevedore
41		*
42		* 2 ObjSetState() sets state
43		* 2 ObjWaitState() waits for particular state
44		* INVALID->REQ_DONE->STREAM->FINISHED->FAILED
45		*
46		* 2 ObjGetSpace() allocates space
47		* 2 ObjExtend() commits content
48		* 2 ObjWaitExtend() waits for content - used to implement ObjIterate())
49		*
50		* 2 ObjSetAttr()
51		* 2 ObjCopyAttr()
52		* 2 ObjSetFlag()
53		* 2 ObjSetDouble()
54		* 2 ObjSetU32()
55		* 2 ObjSetU64()
56		*
57		* 2->3 ObjBocDone() Boc removed from OC, clean it up
58		*
59		* 23 ObjHasAttr()
60		* 23 ObjGetAttr()
61		* 23 ObjCheckFlag()
62		* 23 ObjGetDouble()
63		* 23 ObjGetU32()
64		* 23 ObjGetU64()
65		* 23 ObjGetLen()
66		* 23 ObjGetXID()
67		*
68		* 23 ObjIterate() ... over body
69		*
70		* 23 ObjTouch() Signal to LRU(-like) facilities
71		*
72		* 3->4 HSH_Snipe() kill if not in use
73		* 3->4 HSH_Kill() make unavailable
74		*
75		* 234 ObjSlim() Release body storage (but retain attribute storage)
76		*
77		* 4->5 ObjFreeObj() disassociates stevedore
78		*
79		* 5->6 FREE_OBJ() ...in HSH_DerefObjCore()
80		*/
81
82		#include "config.h"
83
84		#include <stdlib.h>
85
86		#include "cache_varnishd.h"
87		#include "cache_obj.h"
88		#include "vend.h"
89		#include "storage/storage.h"
90
91		static const struct obj_methods *
92	159179	obj_getmethods(const struct objcore *oc)
93		{
94
95	159179	CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
96	159179	CHECK_OBJ_NOTNULL(oc->stobj->stevedore, STEVEDORE_MAGIC);
97	159179	AN(oc->stobj->stevedore->methods);
98	159179	return (oc->stobj->stevedore->methods);
99		}
100
101		static struct boc *
102	3198	obj_newboc(void)
103		{
104		struct boc *boc;
105
106	3198	ALLOC_OBJ(boc, BOC_MAGIC);
107	3198	AN(boc);
108	3198	Lck_New(&boc->mtx, lck_busyobj);
109	3198	PTOK(pthread_cond_init(&boc->cond, NULL));
110	3198	boc->refcount = 1;
111	3198	boc->transit_buffer = cache_param->transit_buffer;
112	3198	return (boc);
113		}
114
115		static void
116	2921	obj_deleteboc(struct boc **p)
117		{
118		struct boc *boc;
119
120	2921	TAKE_OBJ_NOTNULL(boc, p, BOC_MAGIC);
121	2921	Lck_Delete(&boc->mtx);
122	2921	PTOK(pthread_cond_destroy(&boc->cond));
123	2921	free(boc->vary);
124	2921	FREE_OBJ(boc);
125	2921	}
126
127		/*====================================================================
128		* ObjNew()
129		*
130		*/
131
132		struct objcore *
133	3198	ObjNew(const struct worker *wrk)
134		{
135		struct objcore *oc;
136
137	3198	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
138
139	3198	ALLOC_OBJ(oc, OBJCORE_MAGIC);
140	3198	AN(oc);
141	3198	wrk->stats->n_objectcore++;
142	3198	oc->last_lru = NAN;
143	3198	oc->flags = OC_F_BUSY;
144
145	3198	oc->boc = obj_newboc();
146
147	3198	return (oc);
148		}
149
150		/*====================================================================
151		* ObjDestroy()
152		*
153		*/
154
155		void
156	1856	ObjDestroy(const struct worker wrk, struct objcore *p)
157		{
158		struct objcore *oc;
159
160	1856	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
161	1856	TAKE_OBJ_NOTNULL(oc, p, OBJCORE_MAGIC);
162	1856	if (oc->boc != NULL)
163	34	obj_deleteboc(&oc->boc);
164	1856	FREE_OBJ(oc);
165	1856	wrk->stats->n_objectcore--;
166	1856	}
167
168		/*====================================================================
169		* ObjIterate()
170		*
171		*/
172
173		int
174	2404	ObjIterate(struct worker wrk, struct objcore oc,
175		void priv, objiterate_f func, int final)
176		{
177	2404	const struct obj_methods *om = obj_getmethods(oc);
178
179	2404	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
180	2404	AN(func);
181	2404	AN(om->objiterator);
182	2404	return (om->objiterator(wrk, oc, priv, func, final));
183		}
184
185		/*====================================================================
186		* ObjGetSpace()
187		*
188		* This function returns a pointer and length of free space. If there
189		* is no free space, some will be added first.
190		*
191		* The "sz" argument is an input hint of how much space is desired.
192		* 0 means "unknown", return some default size (maybe fetch_chunksize)
193		*/
194
195		int
196	57894	ObjGetSpace(struct worker wrk, struct objcore oc, ssize_t sz, uint8_t *ptr)
197		{
198	57894	const struct obj_methods *om = obj_getmethods(oc);
199
200	57894	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
201	57894	CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
202	57894	AN(sz);
203	57894	AN(ptr);
204	57894	assert(*sz >= 0);
205
206	57894	AN(om->objgetspace);
207	57894	return (om->objgetspace(wrk, oc, sz, ptr));
208		}
209
210		/*====================================================================
211		* ObjExtend()
212		*
213		* This function extends the used part of the object a number of bytes
214		* into the last space returned by ObjGetSpace()
215		*
216		* The final flag must be set on the last call, and it will release any
217		* surplus space allocated.
218		*/
219
220		static void
221	56427	obj_extend_condwait(const struct objcore *oc)
222		{
223
224	56427	if (oc->boc->transit_buffer == 0)
225	55941	return;
226
227	486	assert(oc->flags & OC_F_TRANSIENT);
228	509	while (!(oc->flags & OC_F_CANCEL) && oc->boc->fetched_so_far >
229	507	oc->boc->delivered_so_far + oc->boc->transit_buffer)
230	23	(void)Lck_CondWait(&oc->boc->cond, &oc->boc->mtx);
231	56427	}
232
233		void
234	57113	ObjExtend(struct worker wrk, struct objcore oc, ssize_t l, int final)
235		{
236	57113	const struct obj_methods *om = obj_getmethods(oc);
237
238	57113	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
239	57113	CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
240	57113	AN(om->objextend);
241	57113	assert(l >= 0);
242
243	57113	Lck_Lock(&oc->boc->mtx);
244	57113	if (l > 0) {
245	56427	obj_extend_condwait(oc);
246	56427	om->objextend(wrk, oc, l);
247	56427	oc->boc->fetched_so_far += l;
248	56427	PTOK(pthread_cond_broadcast(&oc->boc->cond));
249	56427	}
250	57113	Lck_Unlock(&oc->boc->mtx);
251
252	57113	assert(oc->boc->state < BOS_FINISHED);
253	57113	if (final && om->objtrimstore != NULL)
254	1878	om->objtrimstore(wrk, oc);
255	57113	}
256
257		/*====================================================================
258		*/
259
260		uint64_t
261	2173	ObjWaitExtend(const struct worker wrk, const struct objcore oc, uint64_t l,
262		enum boc_state_e *statep)
263		{
264		enum boc_state_e state;
265		uint64_t rv;
266
267	2173	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
268	2173	CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
269	2173	CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
270	2173	Lck_Lock(&oc->boc->mtx);
271	3255	while (1) {
272	3255	rv = oc->boc->fetched_so_far;
273	3255	assert(l <= rv \|\| oc->boc->state == BOS_FAILED);
274	3255	if (oc->boc->transit_buffer > 0) {
275	680	assert(oc->flags & OC_F_TRANSIENT);
276		/* Signal the new client position */
277	680	oc->boc->delivered_so_far = l;
278	680	PTOK(pthread_cond_signal(&oc->boc->cond));
279	680	}
280	3255	state = oc->boc->state;
281	3255	if (rv > l \|\| state >= BOS_FINISHED)
282	2173	break;
283	1082	(void)Lck_CondWait(&oc->boc->cond, &oc->boc->mtx);
284		}
285	2173	Lck_Unlock(&oc->boc->mtx);
286	2173	if (statep != NULL)
287	2173	*statep = state;
288	2173	return (rv);
289		}
290		/*====================================================================
291		*/
292
293		void
294	7494	ObjSetState(struct worker wrk, const struct objcore oc,
295		enum boc_state_e next)
296		{
297		const struct obj_methods *om;
298
299	7494	CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
300	7494	CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
301	7494	assert(next > oc->boc->state);
302
303	7494	CHECK_OBJ_ORNULL(oc->stobj->stevedore, STEVEDORE_MAGIC);
304	7494	assert(next != BOS_STREAM \|\| oc->boc->state == BOS_PREP_STREAM);
305	7494	assert(next != BOS_FINISHED \|\| (oc->oa_present & (1 << OA_LEN)));
306
307	7494	if (oc->stobj->stevedore != NULL) {
308	5220	om = oc->stobj->stevedore->methods;
309	5220	if (om->objsetstate != NULL)
310	0	om->objsetstate(wrk, oc, next);
311	5220	}
312
313	7494	Lck_Lock(&oc->boc->mtx);
314	7494	oc->boc->state = next;
315	7494	PTOK(pthread_cond_broadcast(&oc->boc->cond));
316	7494	Lck_Unlock(&oc->boc->mtx);
317	7494	}
318
319		/*====================================================================
320		*/
321
322		void
323	2616	ObjWaitState(const struct objcore *oc, enum boc_state_e want)
324		{
325
326	2616	CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
327	2616	CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
328
329	2616	Lck_Lock(&oc->boc->mtx);
330		/* wake up obj_extend_condwait() */
331	2616	if (oc->flags & OC_F_CANCEL)
332	356	PTOK(pthread_cond_signal(&oc->boc->cond));
333	34835	while (1) {
334	34835	if (oc->boc->state >= want)
335	2616	break;
336	32219	(void)Lck_CondWait(&oc->boc->cond, &oc->boc->mtx);
337		}
338	2616	Lck_Unlock(&oc->boc->mtx);
339	2616	}
340
341		/*====================================================================
342		* ObjGetlen()
343		*
344		* This is a separate function because it may need locking
345		*/
346
347		uint64_t
348	5325	ObjGetLen(struct worker wrk, struct objcore oc)
349		{
350		uint64_t len;
351
352	5325	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
353
354	5325	AZ(ObjGetU64(wrk, oc, OA_LEN, &len));
355	5325	return (len);
356		}
357
358		/*====================================================================
359		* ObjSlim()
360		*
361		* Free the whatever storage can be freed, without freeing the actual
362		* object yet.
363		*/
364
365		void
366	1405	ObjSlim(struct worker wrk, struct objcore oc)
367		{
368	1405	const struct obj_methods *om = obj_getmethods(oc);
369
370	1405	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
371
372	1405	if (om->objslim != NULL)
373	1405	om->objslim(wrk, oc);
374	1405	}
375
376		/*====================================================================
377		* Called when the boc used to populate the objcore is going away.
378		* Useful for releasing any leftovers from Trim.
379		*/
380
381		void
382	2887	ObjBocDone(struct worker wrk, struct objcore oc, struct boc **boc)
383		{
384		const struct obj_methods *m;
385
386	2887	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
387	2887	CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
388	2887	AN(boc);
389	2887	CHECK_OBJ_NOTNULL(*boc, BOC_MAGIC);
390	2887	CHECK_OBJ_ORNULL(oc->stobj->stevedore, STEVEDORE_MAGIC);
391	2887	if (oc->stobj->stevedore != NULL) {
392	2834	m = obj_getmethods(oc);
393	2834	if (m->objbocdone != NULL)
394	2834	m->objbocdone(wrk, oc, *boc);
395	2834	}
396	2887	obj_deleteboc(boc);
397	2887	}
398
399		/*====================================================================
400		*/
401		void
402	1790	ObjFreeObj(struct worker wrk, struct objcore oc)
403		{
404	1790	const struct obj_methods *m = obj_getmethods(oc);
405
406	1790	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
407
408	1790	AN(m->objfree);
409	1790	m->objfree(wrk, oc);
410	1790	AZ(oc->stobj->stevedore);
411	1790	}
412
413		/*====================================================================
414		* ObjHasAttr()
415		*
416		* Check if object has this attribute
417		*/
418
419		int
420	7678	ObjHasAttr(struct worker wrk, struct objcore oc, enum obj_attr attr)
421		{
422
423	7678	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
424	7678	CHECK_OBJ_NOTNULL(oc, OBJCORE_MAGIC);
425
426	7678	if (oc->oa_present)
427	7660	return (oc->oa_present & (1 << attr));
428
429		/* resurrected persistent objects don't have oa_present set */
430	18	return (ObjGetAttr(wrk, oc, attr, NULL) != NULL ? 1 : 0);
431	7678	}
432
433		/*====================================================================
434		* ObjGetAttr()
435		*
436		* Get an attribute of the object.
437		*
438		* Returns NULL on unset or zero length attributes and len set to
439		* zero. Returns Non-NULL otherwise and len is updated with the attributes
440		* length.
441		*/
442
443		const void *
444	21978	ObjGetAttr(struct worker wrk, struct objcore oc, enum obj_attr attr,
445		ssize_t *len)
446		{
447	21978	const struct obj_methods *om = obj_getmethods(oc);
448
449	21978	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
450
451	21978	AN(om->objgetattr);
452	21978	return (om->objgetattr(wrk, oc, attr, len));
453		}
454
455		/*====================================================================
456		* ObjSetAttr()
457		*
458		* Setting fixed size attributes always succeeds.
459		*
460		* Setting a variable size attribute asserts if the combined size of the
461		* variable attributes exceeds the total variable attribute space set at
462		* object creation. If there is space it always succeeds.
463		*
464		* Setting an auxiliary attribute can fail.
465		*
466		* Resetting any variable asserts if the new length does not match the
467		* previous length exactly.
468		*
469		* If ptr is Non-NULL, it points to the new content which is copied into
470		* the attribute. Otherwise the caller will have to do the copying.
471		*
472		* Return value is non-NULL on success and NULL on failure. If ptr was
473		* non-NULL, it is an error to use the returned pointer to set the
474		* attribute data, it is only a success indicator in that case.
475		*/
476
477		void *
478	10560	ObjSetAttr(struct worker wrk, struct objcore oc, enum obj_attr attr,
479		ssize_t len, const void *ptr)
480		{
481	10560	const struct obj_methods *om = obj_getmethods(oc);
482		void *r;
483
484	10560	CHECK_OBJ_NOTNULL(wrk, WORKER_MAGIC);
485	10560	CHECK_OBJ_NOTNULL(oc->boc, BOC_MAGIC);
486
487	10560	AN(om->objsetattr);
488	10560	assert((int)attr < 16);
489	10560	r = om->objsetattr(wrk, oc, attr, len, ptr);
490	10560	if (r)
491	10558	oc->oa_present \|= (1 << attr);
492	10560	return (r);
493		}
494
495		/*====================================================================
496		* ObjTouch()
497		*/
498
499		void