GCOV - Varnish Cache


/*-
 * Copyright (c) 2008-2015 Varnish Software AS
 * All rights reserved.
 *
 * Author: Martin Blix Grydeland <martin@varnish-software.com>
 *
 * 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.
 */

#ifdef VTEST_WITH_VTC_LOGEXPECT

/* SECTION: logexpect logexpect
 *
 * Reads the VSL and looks for records matching a given specification. It will
 * process records trying to match the first pattern, and when done, will
 * continue processing, trying to match the following pattern. If a pattern
 * isn't matched, the test will fail.
 *
 * logexpect threads are declared this way::
 *
 *         logexpect lNAME -v <id> [-g <grouping>] [-d 0|1] [-q query] \
 *                 [vsl arguments] {
 *                         expect <skip> <vxid> <tag> <regex>
 *                         expect <skip> <vxid> <tag> <regex>
 *                         fail add <vxid> <tag> <regex>
 *                         fail clear
 *                         abort
 *                         ...
 *                 } [-start|-wait|-run]
 *
 * And once declared, you can start them, or wait on them::
 *
 *         logexpect lNAME <-start|-wait>
 *
 * With:
 *
 * lNAME
 *         Name the logexpect thread, it must start with 'l'.
 *
 * \-v id
 *         Specify the varnish instance to use (most of the time, id=v1).
 *
 * \-g <session|request|vxid|raw
 *         Decide how records are grouped, see -g in ``man varnishlog`` for more
 *         information.
 *
 * \-d <0|1>
 *         Start processing log records at the head of the log instead of the
 *         tail.
 *
 * \-q query
 *         Filter records using a query expression, see ``man vsl-query`` for
 *         more information. Multiple -q options are not supported.
 *
 * \-m
 *         Also emit log records for misses (only for debugging)
 *
 * \-err
 *         Invert the meaning of success. Usually called once to expect the
 *         logexpect to fail
 *
 * \-start
 *         Start the logexpect thread in the background.
 *
 * \-wait
 *         Wait for the logexpect thread to finish
 *
 * \-run
 *        Equivalent to "-start -wait".
 *
 * VSL arguments (similar to the varnishlog options):
 *
 * \-C
 *         Use caseless regex
 *
 * \-i <taglist>
 *         Include tags
 *
 * \-I <[taglist:]regex>
 *         Include by regex
 *
 * \-T <seconds>
 *         Transaction end timeout
 *
 * expect specification:
 *
 * skip: [uint|*|?]
 *         Max number of record to skip
 *
 * vxid: [uint|*|=]
 *         vxid to match
 *
 * tag:  [tagname|*|=]
 *         Tag to match against
 *
 * regex:
 *         regular expression to match against (optional)
 *
 * For skip, vxid and tag, '*' matches anything, '=' expects the value of the
 * previous matched record. The '?' marker is equivalent to zero, expecting a
 * match on the next record. The difference is that '?' can be used when the
 * order of individual consecutive logs is not deterministic. In other words,
 * lines from a block of alternatives marked by '?' can be matched in any order,
 * but all need to match eventually.
 *
 * fail specification:
 *
 * add: Add to the fail list
 *
 *      Arguments are equivalent to expect, except for skip missing
 *
 * clear: Clear the fail list
 *
 * Any number of fail specifications can be active during execution of
 * a logexpect. All active fail specifications are matched against every
 * log line and, if any match, the logexpect fails immediately.
 *
 * For a logexpect to end successfully, there must be no specs on the fail list,
 * so logexpects should always end with
 *
 *      expect <skip> <vxid> <tag> <termination-condition>
 *      fail clear
 *
 * .. XXX can we come up with a better solution which is still safe?
 *
 * abort specification:
 *
 * abort(3) varnishtest, intended to help debugging of the VSL client library
 * itself.
 */

#include "config.h"

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <stdint.h>

#include "vapi/vsm.h"
#include "vapi/vsl.h"

#include "vtc.h"

#include "vtim.h"
#include "vre.h"

#define LE_ANY   (-1)
#define LE_LAST  (-2)
#define LE_ALT   (-3)
#define LE_SEEN  (-4)
#define LE_FAIL  (-5)
#define LE_CLEAR (-6)   // clear fail list
#define LE_ABORT (-7)

struct logexp_test {
        unsigned                        magic;
#define LOGEXP_TEST_MAGIC               0x6F62B350
        VTAILQ_ENTRY(logexp_test)       list;
        VTAILQ_ENTRY(logexp_test)       faillist;

        struct vsb                      *str;
        int64_t                         vxid;
        int                             tag;
        vre_t                           *vre;
        int                             skip_max;
};

VTAILQ_HEAD(tests_head,logexp_test);

struct logexp {
        unsigned                        magic;
#define LOGEXP_MAGIC                    0xE81D9F1B
        VTAILQ_ENTRY(logexp)            list;

        char                            *name;
        char                            *vname;
        struct vtclog                   *vl;
        char                            run;
        struct tests_head               tests;

        struct logexp_test              *test;
        int                             skip_cnt;
        int64_t                         vxid_last;
        int                             tag_last;

        struct tests_head               fail;

        int                             m_arg;
        int                             err_arg;
        int                             d_arg;
        enum VSL_grouping_e             g_arg;
        char                            *query;

        struct vsm                      *vsm;
        struct VSL_data                 *vsl;
        struct VSLQ                     *vslq;
        pthread_t                       tp;
};

static VTAILQ_HEAD(, logexp)            logexps =
        VTAILQ_HEAD_INITIALIZER(logexps);

static cmd_f cmd_logexp_expect;
static cmd_f cmd_logexp_fail;
static cmd_f cmd_logexp_abort;

static const struct cmds logexp_cmds[] = {
        { "expect",             cmd_logexp_expect },
        { "fail",               cmd_logexp_fail },
        { "abort",              cmd_logexp_abort },
        { NULL,                 NULL },
};

static void
logexp_delete_tests(struct logexp *le)
{
        struct logexp_test *test;

        CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);
        while (!VTAILQ_EMPTY(&le->tests)) {
                test = VTAILQ_FIRST(&le->tests);
                CHECK_OBJ_NOTNULL(test, LOGEXP_TEST_MAGIC);
                VTAILQ_REMOVE(&le->tests, test, list);
                VSB_destroy(&test->str);
                if (test->vre)
                        VRE_free(&test->vre);
                FREE_OBJ(test);
        }
}

static void
logexp_delete(struct logexp *le)
{
        CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);
        AZ(le->run);
        AN(le->vsl);
        VSL_Delete(le->vsl);
        AZ(le->vslq);
        logexp_delete_tests(le);
        free(le->name);
        free(le->vname);
        free(le->query);
        VSM_Destroy(&le->vsm);
        vtc_logclose(le->vl);
        FREE_OBJ(le);
}

static struct logexp *
logexp_new(const char *name, const char *varg)
{
        struct logexp *le;
        struct vsb *n_arg;

        ALLOC_OBJ(le, LOGEXP_MAGIC);
        AN(le);
        REPLACE(le->name, name);
        le->vl = vtc_logopen("%s", name);
        vtc_log_set_cmd(le->vl, logexp_cmds);
        VTAILQ_INIT(&le->tests);

        le->d_arg = 0;
        le->g_arg = VSL_g_vxid;
        le->vsm = VSM_New();
        le->vsl = VSL_New();
        AN(le->vsm);
        AN(le->vsl);

        VTAILQ_INSERT_TAIL(&logexps, le, list);

        REPLACE(le->vname, varg);

        n_arg = macro_expandf(le->vl, "${tmpdir}/%s", varg);
        if (n_arg == NULL)
                vtc_fatal(le->vl, "-v argument problems");
        if (VSM_Arg(le->vsm, 'n', VSB_data(n_arg)) <= 0)
                vtc_fatal(le->vl, "-v argument error: %s",
                    VSM_Error(le->vsm));
        VSB_destroy(&n_arg);
        if (VSM_Attach(le->vsm, -1))
                vtc_fatal(le->vl, "VSM_Attach: %s", VSM_Error(le->vsm));
        return (le);
}

static void
logexp_clean(const struct tests_head *head)
{
        struct logexp_test *test;

        VTAILQ_FOREACH(test, head, list)
                if (test->skip_max == LE_SEEN)
                        test->skip_max = LE_ALT;
}

static struct logexp_test *
logexp_alt(struct logexp_test *test)
{
        assert(test->skip_max == LE_ALT);

        do
                test = VTAILQ_NEXT(test, list);
        while (test != NULL && test->skip_max == LE_SEEN);

        if (test == NULL || test->skip_max != LE_ALT)
                return (NULL);

        return (test);
}

static void
logexp_next(struct logexp *le)
{
        CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);

        if (le->test && le->test->skip_max == LE_ALT) {
                /*
                 * if an alternative was not seen, continue at this expection
                 * with the next vsl
                 */
                (void)0;
        } else if (le->test) {
                CHECK_OBJ_NOTNULL(le->test, LOGEXP_TEST_MAGIC);
                le->test = VTAILQ_NEXT(le->test, list);
        } else {
                logexp_clean(&le->tests);
                VTAILQ_INIT(&le->fail);
                le->test = VTAILQ_FIRST(&le->tests);
        }

        if (le->test == NULL)
                return;

        CHECK_OBJ(le->test, LOGEXP_TEST_MAGIC);

        switch (le->test->skip_max) {
        case LE_SEEN:
                logexp_next(le);
                return;
        case LE_CLEAR:
                vtc_log(le->vl, 3, "cond | fail clear");
                VTAILQ_INIT(&le->fail);
                logexp_next(le);
                return;
        case LE_FAIL:
                vtc_log(le->vl, 3, "cond | %s", VSB_data(le->test->str));
                VTAILQ_INSERT_TAIL(&le->fail, le->test, faillist);
                logexp_next(le);
                return;
        case LE_ABORT:
                abort();
                NEEDLESS(return);
        default:
                vtc_log(le->vl, 3, "test | %s", VSB_data(le->test->str));
        }
}

enum le_match_e {
        LEM_OK,
        LEM_SKIP,
        LEM_FAIL
};

static enum le_match_e
logexp_match(const struct logexp *le, struct logexp_test *test,
    const char *data, int64_t vxid, int tag, int type, int len)
{
        const char *legend;
        int ok = 1, skip = 0, alt, fail, vxid_ok = 0;

        AN(le);
        AN(test);
        assert(test->skip_max != LE_SEEN);
        assert(test->skip_max != LE_CLEAR);

        if (test->vxid == LE_LAST) {
                if (le->vxid_last != vxid)
                        ok = 0;
                vxid_ok = ok;
        } else if (test->vxid >= 0) {
                if (test->vxid != vxid)
                        ok = 0;
                vxid_ok = ok;
        }
        if (test->tag == LE_LAST) {
                if (le->tag_last != tag)
                        ok = 0;
        } else if (test->tag >= 0) {
                if (test->tag != tag)
                        ok = 0;
        }
        if (test->vre &&
            test->tag >= 0 &&
            test->tag == tag &&
            VRE_ERROR_NOMATCH == VRE_match(test->vre, data, len, 0, NULL))
                ok = 0;

        alt = (test->skip_max == LE_ALT);
        fail = (test->skip_max == LE_FAIL);

        if (!ok && !alt && (test->skip_max == LE_ANY ||
                    test->skip_max > le->skip_cnt))
                skip = 1;

        if (skip && vxid_ok && tag == SLT_End)
                fail = 1;

        if (fail) {
                if (ok) {
                        legend = "fail";
                } else if (skip) {
                        legend = "end";
                        skip = 0;
                } else if (le->m_arg) {
                        legend = "fmiss";
                } else {
                        legend = NULL;
                }
        }
        else if (ok)
                legend = "match";
        else if (skip && le->m_arg)
                legend = "miss";
        else if (skip || alt)
                legend = NULL;
        else
                legend = "err";

        if (legend != NULL)
                vtc_log(le->vl, 4, "%-5s| %10ju %-15s %c %.*s",
                    legend, (intmax_t)vxid, VSL_tags[tag], type, len,
                    data);

        if (ok) {
                if (alt)
                        test->skip_max = LE_SEEN;
                return (LEM_OK);
        }
        if (alt) {
                test = logexp_alt(test);
                if (test == NULL)
                        return (LEM_FAIL);
                vtc_log(le->vl, 3, "alt  | %s", VSB_data(test->str));
                return (logexp_match(le, test, data, vxid, tag, type, len));
        }
        if (skip)
                return (LEM_SKIP);
        return (LEM_FAIL);
}

static enum le_match_e
logexp_failchk(const struct logexp *le,
    const char *data, int64_t vxid, int tag, int type, int len)
{
        struct logexp_test *test;
        static enum le_match_e r;

        if (VTAILQ_FIRST(&le->fail) == NULL)
                return (LEM_SKIP);

        VTAILQ_FOREACH(test, &le->fail, faillist) {
                r = logexp_match(le, test, data, vxid, tag, type, len);
                if (r == LEM_OK)
                        return (LEM_FAIL);
                assert (r == LEM_FAIL);
        }
        return (LEM_OK);
}

static int
logexp_done(const struct logexp *le)
{
        return ((VTAILQ_FIRST(&le->fail) == NULL) && le->test == NULL);
}

static int v_matchproto_(VSLQ_dispatch_f)
logexp_dispatch(struct VSL_data *vsl, struct VSL_transaction * const pt[],
    void *priv)
{
        struct logexp *le;
        struct VSL_transaction *t;
        int i;
        enum le_match_e r;
        int64_t vxid;
        int tag, type, len;
        const char *data;

        CAST_OBJ_NOTNULL(le, priv, LOGEXP_MAGIC);

        for (i = 0; (t = pt[i]) != NULL; i++) {
                while (1 == VSL_Next(t->c)) {
                        if (!VSL_Match(vsl, t->c))
                                continue;

                        AN(t->c->rec.ptr);
                        tag = VSL_TAG(t->c->rec.ptr);

                        if (tag == SLT__Batch || tag == SLT_Witness)
                                continue;

                        vxid = VSL_ID(t->c->rec.ptr);
                        data = VSL_CDATA(t->c->rec.ptr);
                        len = VSL_LEN(t->c->rec.ptr) - 1;
                        type = VSL_CLIENT(t->c->rec.ptr) ? 'c' :
                            VSL_BACKEND(t->c->rec.ptr) ? 'b' : '-';

                        r = logexp_failchk(le, data, vxid, tag, type, len);
                        if (r == LEM_FAIL)
                                return (r);
                        if (le->test == NULL) {
                                assert (r == LEM_OK);
                                continue;
                        }

                        CHECK_OBJ_NOTNULL(le->test, LOGEXP_TEST_MAGIC);

                        r = logexp_match(le, le->test,
                            data, vxid, tag, type, len);
                        if (r == LEM_FAIL)
                                return (r);
                        if (r == LEM_SKIP) {
                                le->skip_cnt++;
                                continue;
                        }
                        assert(r == LEM_OK);
                        le->vxid_last = vxid;
                        le->tag_last = tag;
                        le->skip_cnt = 0;
                        logexp_next(le);
                        if (logexp_done(le))
                                return (1);
                }
        }
        return (0);
}

static void *
logexp_thread(void *priv)
{
        struct logexp *le;
        int i;

        CAST_OBJ_NOTNULL(le, priv, LOGEXP_MAGIC);
        AN(le->run);
        AN(le->vsm);
        AN(le->vslq);

        AZ(le->test);
        vtc_log(le->vl, 4, "begin|");
        if (le->query != NULL)
                vtc_log(le->vl, 4, "qry  | %s", le->query);
        logexp_next(le);
        while (!logexp_done(le) && !vtc_stop && !vtc_error) {
                i = VSLQ_Dispatch(le->vslq, logexp_dispatch, le);
                if (i == 2 && le->err_arg) {
                        vtc_log(le->vl, 4, "done | failed as expected");
                        return (NULL);
                }
                if (i == 2)
                        vtc_fatal(le->vl, "bad  | expectation failed");
                else if (i < 0)
                        vtc_fatal(le->vl, "bad  | dispatch failed (%d)", i);
                else if (i == 0 && ! logexp_done(le))
                        VTIM_sleep(0.01);
        }
        if (!logexp_done(le))
                vtc_fatal(le->vl, "bad  | outstanding expectations");
        vtc_log(le->vl, 4, "done |");

        return (NULL);
}

static void
logexp_close(struct logexp *le)
{

        CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);
        AN(le->vsm);
        if (le->vslq)
                VSLQ_Delete(&le->vslq);
        AZ(le->vslq);
}

static void
logexp_start(struct logexp *le)
{
        struct VSL_cursor *c;

        CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);
        AN(le->vsl);
        AZ(le->vslq);

        AN(le->vsl);
        (void)VSM_Status(le->vsm);
        c = VSL_CursorVSM(le->vsl, le->vsm,
            (le->d_arg ? 0 : VSL_COPT_TAIL) | VSL_COPT_BATCH);
        if (c == NULL)
                vtc_fatal(le->vl, "VSL_CursorVSM: %s", VSL_Error(le->vsl));
        le->vslq = VSLQ_New(le->vsl, &c, le->g_arg, le->query);
        if (le->vslq == NULL) {
                VSL_DeleteCursor(c);
                vtc_fatal(le->vl, "VSLQ_New: %s", VSL_Error(le->vsl));
        }
        AZ(c);

        le->test = NULL;
        le->skip_cnt = 0;
        le->vxid_last = le->tag_last = -1;
        le->run = 1;
        PTOK(pthread_create(&le->tp, NULL, logexp_thread, le));
}

static void
logexp_wait(struct logexp *le)
{
        void *res;

        CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);
        vtc_log(le->vl, 2, "Waiting for logexp");
        PTOK(pthread_join(le->tp, &res));
        logexp_close(le);
        if (res != NULL && !vtc_stop)
                vtc_fatal(le->vl, "logexp returned \"%p\"", (char *)res);
        le->run = 0;
}

/* shared by expect and fail: parse from av[2] (vxid) onwards */

static void
cmd_logexp_common(struct logexp *le, struct vtclog *vl,
    int skip_max, char * const *av)
{
        vre_t *vre;
        struct vsb vsb[1];
        int64_t vxid;
        int err, pos, tag;
        struct logexp_test *test;
        char *end, errbuf[VRE_ERROR_LEN];

        if (!strcmp(av[2], "*"))
                vxid = LE_ANY;
        else if (!strcmp(av[2], "="))
                vxid = LE_LAST;
        else {
                vxid = strtoll(av[2], &end, 10);
                if (*end != '\0' || vxid < 0)
                        vtc_fatal(vl, "Not a positive integer: '%s'", av[2]);
        }
        if (!strcmp(av[3], "*"))
                tag = LE_ANY;
        else if (!strcmp(av[3], "="))
                tag = LE_LAST;
        else {
                tag = VSL_Name2Tag(av[3], strlen(av[3]));
                if (tag < 0)
                        vtc_fatal(vl, "Unknown tag name: '%s'", av[3]);
        }
        vre = NULL;
        if (av[4]) {
                vre = VRE_compile(av[4], 0, &err, &pos, 1);
                if (vre == NULL) {
                        AN(VSB_init(vsb, errbuf, sizeof errbuf));
                        AZ(VRE_error(vsb, err));
                        AZ(VSB_finish(vsb));
                        VSB_fini(vsb);
                        vtc_fatal(vl, "Regex error (%s): '%s' pos %d",
                            errbuf, av[4], pos);
                }
        }

        ALLOC_OBJ(test, LOGEXP_TEST_MAGIC);
        AN(test);
        test->str = VSB_new_auto();
        AN(test->str);
        AZ(VSB_printf(test->str, "%s %s %s %s ", av[0], av[1], av[2], av[3]));
        if (av[4])
                VSB_quote(test->str, av[4], -1, 0);
        AZ(VSB_finish(test->str));
        test->skip_max = skip_max;
        test->vxid = vxid;
        test->tag = tag;
        test->vre = vre;
        VTAILQ_INSERT_TAIL(&le->tests, test, list);
}

static void
cmd_logexp_expect(CMD_ARGS)
{
        struct logexp *le;
        int skip_max;
        char *end;

        CAST_OBJ_NOTNULL(le, priv, LOGEXP_MAGIC);
        if (av[1] == NULL || av[2] == NULL || av[3] == NULL)
                vtc_fatal(vl, "Syntax error");

        if (av[4] != NULL && av[5] != NULL)
                vtc_fatal(vl, "Syntax error");

        if (!strcmp(av[1], "*"))
                skip_max = LE_ANY;
        else if (!strcmp(av[1], "?"))
                skip_max = LE_ALT;
        else {
                skip_max = (int)strtol(av[1], &end, 10);
                if (*end != '\0' || skip_max < 0)
                        vtc_fatal(vl, "Not a positive integer: '%s'", av[1]);
        }
        cmd_logexp_common(le, vl, skip_max, av);
}

static void
cmd_logexp_fail(CMD_ARGS)
{
        struct logexp *le;
        struct logexp_test *test;

        CAST_OBJ_NOTNULL(le, priv, LOGEXP_MAGIC);

        if (av[1] == NULL)
                vtc_fatal(vl, "Syntax error");

        if (!strcmp(av[1], "clear")) {
                ALLOC_OBJ(test, LOGEXP_TEST_MAGIC);
                AN(test);
                test->skip_max = LE_CLEAR;
                test->str = VSB_new_auto();
                AN(test->str);
                AZ(VSB_printf(test->str, "%s %s",
                    av[0], av[1]));
                AZ(VSB_finish(test->str));

                VTAILQ_INSERT_TAIL(&le->tests, test, list);
                return;
        }

        if (strcmp(av[1], "add"))
                vtc_fatal(vl, "Unknown fail argument '%s'", av[1]);

        if (av[2] == NULL || av[3] == NULL)
                vtc_fatal(vl, "Syntax error");

        cmd_logexp_common(le, vl, LE_FAIL, av);
}

/* aid vsl debugging */
static void
cmd_logexp_abort(CMD_ARGS)
{

        struct logexp *le;

        CAST_OBJ_NOTNULL(le, priv, LOGEXP_MAGIC);

        cmd_logexp_common(le, vl, LE_ABORT, av);
}

static void
logexp_spec(struct logexp *le, const char *spec)
{
        CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);

        logexp_delete_tests(le);

        parse_string(le->vl, le, spec);
}

void
cmd_logexpect(CMD_ARGS)
{
        struct logexp *le, *le2;
        int i;

        (void)priv;

        if (av == NULL) {
                /* Reset and free */
                VTAILQ_FOREACH_SAFE(le, &logexps, list, le2) {
                        CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);
                        VTAILQ_REMOVE(&logexps, le, list);
                        if (le->run) {
                                (void)pthread_cancel(le->tp);
                                logexp_wait(le);
                        }
                        logexp_delete(le);
                }
                return;
        }

        AZ(strcmp(av[0], "logexpect"));
        av++;

        VTC_CHECK_NAME(vl, av[0], "Logexpect", 'l');
        VTAILQ_FOREACH(le, &logexps, list) {
                if (!strcmp(le->name, av[0]))
                        break;
        }
        if (le == NULL) {
                if (strcmp(av[1], "-v") || av[2] == NULL)
                        vtc_fatal(vl, "new logexp lacks -v");
                le = logexp_new(av[0], av[2]);
                av += 2;
        }
        av++;

        for (; *av != NULL; av++) {
                if (vtc_error)
                        break;
                if (!strcmp(*av, "-wait")) {
                        if (!le->run)
                                vtc_fatal(le->vl, "logexp not -started '%s'",
                                        *av);
                        logexp_wait(le);
                        continue;
                }

                /*
                 * We do an implicit -wait if people muck about with a
                 * running logexp.
                 */
                if (le->run)
                        logexp_wait(le);
                AZ(le->run);

                if (!strcmp(*av, "-v")) {
                        if (av[1] == NULL || strcmp(av[1], le->vname))
                                vtc_fatal(le->vl, "-v argument cannot change");
                        av++;
                        continue;
                }
                if (!strcmp(*av, "-d")) {
                        if (av[1] == NULL)
                                vtc_fatal(le->vl, "Missing -d argument");
                        le->d_arg = atoi(av[1]);
                        av++;
                        continue;
                }
                if (!strcmp(*av, "-g")) {
                        if (av[1] == NULL)
                                vtc_fatal(le->vl, "Missing -g argument");
                        i = VSLQ_Name2Grouping(av[1], strlen(av[1]));
                        if (i < 0)
                                vtc_fatal(le->vl, "Unknown grouping '%s'",
                                    av[1]);
                        le->g_arg = (enum VSL_grouping_e)i;
                        av++;
                        continue;
                }
                if (!strcmp(*av, "-q")) {
                        if (av[1] == NULL)
                                vtc_fatal(le->vl, "Missing -q argument");
                        REPLACE(le->query, av[1]);
                        av++;
                        continue;
                }
                if (!strcmp(*av, "-m")) {
                        le->m_arg = !le->m_arg;
                        continue;
                }
                if (!strcmp(*av, "-err")) {
                        le->err_arg = !le->err_arg;
                        continue;
                }
                if (!strcmp(*av, "-start")) {
                        logexp_start(le);
                        continue;
                }
                if (!strcmp(*av, "-run")) {
                        logexp_start(le);
                        logexp_wait(le);
                        continue;
                }
                if (**av == '-') {
                        if (av[1] != NULL) {
                                if (VSL_Arg(le->vsl, av[0][1], av[1])) {
                                        av++;
                                        continue;
                                }
                                vtc_fatal(le->vl, "%s", VSL_Error(le->vsl));
                        }
                        vtc_fatal(le->vl, "Unknown logexp argument: %s", *av);
                }
                logexp_spec(le, *av);
        }
}

#endif /* VTEST_WITH_VTC_LOGEXPECT */

		varnish-cache/bin/varnishtest/vtc_logexp.c
0		/*-
1		* Copyright (c) 2008-2015 Varnish Software AS
2		* All rights reserved.
3		*
4		* Author: Martin Blix Grydeland <martin@varnish-software.com>
5		*
6		* SPDX-License-Identifier: BSD-2-Clause
7		*
8		* Redistribution and use in source and binary forms, with or without
9		* modification, are permitted provided that the following conditions
10		* are met:
11		* 1. Redistributions of source code must retain the above copyright
12		* notice, this list of conditions and the following disclaimer.
13		* 2. Redistributions in binary form must reproduce the above copyright
14		* notice, this list of conditions and the following disclaimer in the
15		* documentation and/or other materials provided with the distribution.
16		*
17		* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18		* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19		* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20		* ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
21		* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22		* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23		* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24		* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25		* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26		* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27		* SUCH DAMAGE.
28		*/
29
30		#ifdef VTEST_WITH_VTC_LOGEXPECT
31
32		/* SECTION: logexpect logexpect
33		*
34		* Reads the VSL and looks for records matching a given specification. It will
35		* process records trying to match the first pattern, and when done, will
36		* continue processing, trying to match the following pattern. If a pattern
37		* isn't matched, the test will fail.
38		*
39		* logexpect threads are declared this way::
40		*
41		* logexpect lNAME -v <id> [-g <grouping>] [-d 0\|1] [-q query] \
42		* [vsl arguments] {
43		* expect <skip> <vxid> <tag> <regex>
44		* expect <skip> <vxid> <tag> <regex>
45		* fail add <vxid> <tag> <regex>
46		* fail clear
47		* abort
48		* ...
49		* } [-start\|-wait\|-run]
50		*
51		* And once declared, you can start them, or wait on them::
52		*
53		* logexpect lNAME <-start\|-wait>
54		*
55		* With:
56		*
57		* lNAME
58		* Name the logexpect thread, it must start with 'l'.
59		*
60		* \-v id
61		* Specify the varnish instance to use (most of the time, id=v1).
62		*
63		* \-g <session\|request\|vxid\|raw
64		* Decide how records are grouped, see -g in ``man varnishlog`` for more
65		* information.
66		*
67		* \-d <0\|1>
68		* Start processing log records at the head of the log instead of the
69		* tail.
70		*
71		* \-q query
72		* Filter records using a query expression, see ``man vsl-query`` for
73		* more information. Multiple -q options are not supported.
74		*
75		* \-m
76		* Also emit log records for misses (only for debugging)
77		*
78		* \-err
79		* Invert the meaning of success. Usually called once to expect the
80		* logexpect to fail
81		*
82		* \-start
83		* Start the logexpect thread in the background.
84		*
85		* \-wait
86		* Wait for the logexpect thread to finish
87		*
88		* \-run
89		* Equivalent to "-start -wait".
90		*
91		* VSL arguments (similar to the varnishlog options):
92		*
93		* \-C
94		* Use caseless regex
95		*
96		* \-i <taglist>
97		* Include tags
98		*
99		* \-I <[taglist:]regex>
100		* Include by regex
101		*
102		* \-T <seconds>
103		* Transaction end timeout
104		*
105		* expect specification:
106		*
107		* skip: [uint\|*\|?]
108		* Max number of record to skip
109		*
110		* vxid: [uint\|*\|=]
111		* vxid to match
112		*
113		* tag: [tagname\|*\|=]
114		* Tag to match against
115		*
116		* regex:
117		* regular expression to match against (optional)
118		*
119		* For skip, vxid and tag, '*' matches anything, '=' expects the value of the
120		* previous matched record. The '?' marker is equivalent to zero, expecting a
121		* match on the next record. The difference is that '?' can be used when the
122		* order of individual consecutive logs is not deterministic. In other words,
123		* lines from a block of alternatives marked by '?' can be matched in any order,
124		* but all need to match eventually.
125		*
126		* fail specification:
127		*
128		* add: Add to the fail list
129		*
130		* Arguments are equivalent to expect, except for skip missing
131		*
132		* clear: Clear the fail list
133		*
134		* Any number of fail specifications can be active during execution of
135		* a logexpect. All active fail specifications are matched against every
136		* log line and, if any match, the logexpect fails immediately.
137		*
138		* For a logexpect to end successfully, there must be no specs on the fail list,
139		* so logexpects should always end with
140		*
141		* expect <skip> <vxid> <tag> <termination-condition>
142		* fail clear
143		*
144		* .. XXX can we come up with a better solution which is still safe?
145		*
146		* abort specification:
147		*
148		* abort(3) varnishtest, intended to help debugging of the VSL client library
149		* itself.
150		*/
151
152		#include "config.h"
153
154		#include <stdlib.h>
155		#include <stdio.h>
156		#include <string.h>
157		#include <stdint.h>
158
159		#include "vapi/vsm.h"
160		#include "vapi/vsl.h"
161
162		#include "vtc.h"
163
164		#include "vtim.h"
165		#include "vre.h"
166
167		#define LE_ANY (-1)
168		#define LE_LAST (-2)
169		#define LE_ALT (-3)
170		#define LE_SEEN (-4)
171		#define LE_FAIL (-5)
172		#define LE_CLEAR (-6) // clear fail list
173		#define LE_ABORT (-7)
174
175		struct logexp_test {
176		unsigned magic;
177		#define LOGEXP_TEST_MAGIC 0x6F62B350
178		VTAILQ_ENTRY(logexp_test) list;
179		VTAILQ_ENTRY(logexp_test) faillist;
180
181		struct vsb *str;
182		int64_t vxid;
183		int tag;
184		vre_t *vre;
185		int skip_max;
186		};
187
188		VTAILQ_HEAD(tests_head,logexp_test);
189
190		struct logexp {
191		unsigned magic;
192		#define LOGEXP_MAGIC 0xE81D9F1B
193		VTAILQ_ENTRY(logexp) list;
194
195		char *name;
196		char *vname;
197		struct vtclog *vl;
198		char run;
199		struct tests_head tests;
200
201		struct logexp_test *test;
202		int skip_cnt;
203		int64_t vxid_last;
204		int tag_last;
205
206		struct tests_head fail;
207
208		int m_arg;
209		int err_arg;
210		int d_arg;
211		enum VSL_grouping_e g_arg;
212		char *query;
213
214		struct vsm *vsm;
215		struct VSL_data *vsl;
216		struct VSLQ *vslq;
217		pthread_t tp;
218		};
219
220		static VTAILQ_HEAD(, logexp) logexps =
221		VTAILQ_HEAD_INITIALIZER(logexps);
222
223		static cmd_f cmd_logexp_expect;
224		static cmd_f cmd_logexp_fail;
225		static cmd_f cmd_logexp_abort;
226
227		static const struct cmds logexp_cmds[] = {
228		{ "expect", cmd_logexp_expect },
229		{ "fail", cmd_logexp_fail },
230		{ "abort", cmd_logexp_abort },
231		{ NULL, NULL },
232		};
233
234		static void
235	19000	logexp_delete_tests(struct logexp *le)
236		{
237		struct logexp_test *test;
238
239	19000	CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);
240	84920	while (!VTAILQ_EMPTY(&le->tests)) {
241	65920	test = VTAILQ_FIRST(&le->tests);
242	65920	CHECK_OBJ_NOTNULL(test, LOGEXP_TEST_MAGIC);
243	65920	VTAILQ_REMOVE(&le->tests, test, list);
244	65920	VSB_destroy(&test->str);
245	65920	if (test->vre)
246	58960	VRE_free(&test->vre);
247	65920	FREE_OBJ(test);
248		}
249	19000	}
250
251		static void
252	8400	logexp_delete(struct logexp *le)
253		{
254	8400	CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);
255	8400	AZ(le->run);
256	8400	AN(le->vsl);
257	8400	VSL_Delete(le->vsl);
258	8400	AZ(le->vslq);
259	8400	logexp_delete_tests(le);
260	8400	free(le->name);
261	8400	free(le->vname);
262	8400	free(le->query);
263	8400	VSM_Destroy(&le->vsm);
264	8400	vtc_logclose(le->vl);
265	8400	FREE_OBJ(le);
266	8400	}
267
268		static struct logexp *
269	8400	logexp_new(const char name, const char varg)
270		{
271		struct logexp *le;
272		struct vsb *n_arg;
273
274	8400	ALLOC_OBJ(le, LOGEXP_MAGIC);
275	8400	AN(le);
276	8400	REPLACE(le->name, name);
277	8400	le->vl = vtc_logopen("%s", name);
278	8400	vtc_log_set_cmd(le->vl, logexp_cmds);
279	8400	VTAILQ_INIT(&le->tests);
280
281	8400	le->d_arg = 0;
282	8400	le->g_arg = VSL_g_vxid;
283	8400	le->vsm = VSM_New();
284	8400	le->vsl = VSL_New();
285	8400	AN(le->vsm);
286	8400	AN(le->vsl);
287
288	8400	VTAILQ_INSERT_TAIL(&logexps, le, list);
289
290	8400	REPLACE(le->vname, varg);
291
292	8400	n_arg = macro_expandf(le->vl, "${tmpdir}/%s", varg);
293	8400	if (n_arg == NULL)
294	0	vtc_fatal(le->vl, "-v argument problems");
295	8400	if (VSM_Arg(le->vsm, 'n', VSB_data(n_arg)) <= 0)
296	0	vtc_fatal(le->vl, "-v argument error: %s",
297	0	VSM_Error(le->vsm));
298	8400	VSB_destroy(&n_arg);
299	8400	if (VSM_Attach(le->vsm, -1))
300	0	vtc_fatal(le->vl, "VSM_Attach: %s", VSM_Error(le->vsm));
301	8400	return (le);
302		}
303
304		static void
305	10680	logexp_clean(const struct tests_head *head)
306		{
307		struct logexp_test *test;
308
309	76600	VTAILQ_FOREACH(test, head, list)
310	65920	if (test->skip_max == LE_SEEN)
311	0	test->skip_max = LE_ALT;
312	10680	}
313
314		static struct logexp_test *
315	200	logexp_alt(struct logexp_test *test)
316		{
317	200	assert(test->skip_max == LE_ALT);
318
319	200	do
320	440	test = VTAILQ_NEXT(test, list);
321	240	while (test != NULL && test->skip_max == LE_SEEN);
322
323	200	if (test == NULL \|\| test->skip_max != LE_ALT)
324	0	return (NULL);
325
326	200	return (test);
327	200	}
328
329		static void
330	76720	logexp_next(struct logexp *le)
331		{
332	76720	CHECK_OBJ_NOTNULL(le, LOGEXP_MAGIC);
333
334	76720	if (le->test && le->test->skip_max == LE_ALT) {
335		/*
336		* if an alternative was not seen, continue at this expection
337		* with the next vsl
338		*/
339		(void)0;
340	76720	} else if (le->test) {
341	65920	CHECK_OBJ_NOTNULL(le->test, LOGEXP_TEST_MAGIC);
342	65920	le->test = VTAILQ_NEXT(le->test, list);
343	65920	} else {
344	10680	logexp_clean(&le->tests);
345	10680	VTAILQ_INIT(&le->fail);
346	10680	le->test = VTAILQ_FIRST(&le->tests);
347		}
348
349	76720	if (le->test == NULL)
350	10680	return;
351
352	66040	CHECK_OBJ(le->test, LOGEXP_TEST_MAGIC);
353
354	66040	switch (le->test->skip_max) {
355		case LE_SEEN:
356	120	logexp_next(le);
357	120	return;
358		case LE_CLEAR:
359	480	vtc_log(le->vl, 3, "cond \| fail clear");
360	480	VTAILQ_INIT(&le->fail);
361	480	logexp_next(le);
362	480	return;
363		case LE_FAIL:
364	600	vtc_log(le->vl, 3, "cond \| %s", VSB_data(le->test->str));
365	600	VTAILQ_INSERT_TAIL(&le->fail, le->test, faillist);
366	600	logexp_next(le);
367	600	return;
368		case LE_ABORT:
369	0	abort();
370		NEEDLESS(return);
371		default:
372	64840	vtc_log(le->vl, 3, "test \| %s", VSB_data(le->test->str));
373	64840	}
374	76720	}
375
376		enum le_match_e {
377		LEM_OK,
378		LEM_SKIP,
379		LEM_FAIL
380		};
381
382		static enum le_match_e
383	1075461	logexp_match(const struct logexp le, struct logexp_test test,
384		const char *data, int64_t vxid, int tag, int type, int len)
385		{
386		const char *legend;
387	1075461	int ok = 1, skip = 0, alt, fail, vxid_ok = 0;
388
389	1075461	AN(le);
390	1075461	AN(test);
391	1075461	assert(test->skip_max != LE_SEEN);
392	1075461	assert(test->skip_max != LE_CLEAR);
393
394	1075461	if (test->vxid == LE_LAST) {
395	189652	if (le->vxid_last != vxid)
396	14041	ok = 0;
397	189652	vxid_ok = ok;
398	1075461	} else if (test->vxid >= 0) {
399	318334	if (test->vxid != vxid)
400	213160	ok = 0;
401	318334	vxid_ok = ok;
402	318334	}
403	1075461	if (test->tag == LE_LAST) {
404	15	if (le->tag_last != tag)
405	0	ok = 0;
406	1075461	} else if (test->tag >= 0) {
407	1075446	if (test->tag != tag)
408	985914	ok = 0;
409	1075446	}
410	1158442	if (test->vre &&
411	964208	test->tag >= 0 &&
412	964208	test->tag == tag &&
413	82981	VRE_ERROR_NOMATCH == VRE_match(test->vre, data, len, 0, NULL))
414	24261	ok = 0;
415
416	1075445	alt = (test->skip_max == LE_ALT);
417	1075445	fail = (test->skip_max == LE_FAIL);
418
419	1075445	if (!ok && !alt && (test->skip_max == LE_ANY \|\|
420	23120	test->skip_max > le->skip_cnt))
421	991891	skip = 1;
422
423	1075445	if (skip && vxid_ok && tag == SLT_End)
424	0	fail = 1;
425
426	1075445	if (fail) {
427	18520	if (ok) {
428	40	legend = "fail";
429	18520	} else if (skip) {
430	0	legend = "end";
431	0	skip = 0;
432	18480	} else if (le->m_arg) {
433	0	legend = "fmiss";
434	0	} else {
435	18480	legend = NULL;
436		}
437	18520	}
438	1056925	else if (ok)
439	64838	legend = "match";
440	992087	else if (skip && le->m_arg)
441	920	legend = "miss";
442	991167	else if (skip \|\| alt)
443	991167	legend = NULL;
444		else
445	0	legend = "err";
446
447	1075445	if (legend != NULL)
448	131596	vtc_log(le->vl, 4, "%-5s\| %10ju %-15s %c %.*s",
449	65798	legend, (intmax_t)vxid, VSL_tags[tag], type, len,
450	65798	data);
451
452	1075445	if (ok) {
453	64878	if (alt)
454	880	test->skip_max = LE_SEEN;
455	64878	return (LEM_OK);
456		}
457	1010567	if (alt) {
458	200	test = logexp_alt(test);
459	200	if (test == NULL)
460	0	return (LEM_FAIL);
461	200	vtc_log(le->vl, 3, "alt \| %s", VSB_data(test->str));
462	200	return (logexp_match(le, test, data, vxid, tag, type, len));
463		}
464	1010367	if (skip)
465	991887	return (LEM_SKIP);
466	18480	return (LEM_FAIL);
467	1075445	}
468
469		static enum le_match_e
470	1057331	logexp_failchk(const struct logexp *le,
471		const char *data, int64_t vxid, int tag, int type, int len)
472		{
473		struct logexp_test *test;
474		static enum le_match_e r;
475
476	1057331	if (VTAILQ_FIRST(&le->fail) == NULL)
477	1044891	return (LEM_SKIP);
478
479	30920	VTAILQ_FOREACH(test, &le->fail, faillist) {
480	18520	r = logexp_match(le, test, data, vxid, tag, type, len);
481	18520	if (r == LEM_OK)
482	40	return (LEM_FAIL);
483	18480	assert (r == LEM_FAIL);
484	18480	}
485	12400	return (LEM_OK);
486	1057331	}
487
488		static int
489	1551363	logexp_done(const struct logexp *le)
490		{
491	1551363	return ((VTAILQ_FIRST(&le->fail) == NULL) && le->test == NULL);
492		}
493
494		static int v_matchproto_(VSLQ_dispatch_f)
495	392232	logexp_dispatch(struct VSL_data vsl, struct VSL_transaction const pt[],
496		void *priv)
497		{
498		struct logexp *le;
499		struct VSL_transaction *t;