GCOV - Varnish Cache


/*-
 * Copyright (c) 2006 Verdens Gang AS
 * Copyright (c) 2006-2015 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.
 */

/*
 * XXX:
 *      Better error messages, throughout.
 *      >It also occurred to me that we could link the errors to the error
 *      >documentation.
 *      >
 *      >Unreferenced  function 'request_policy', first mention is
 *      >         Line 8 Pos 4
 *      >         sub request_policy {
 *      >         ----##############--
 *      >Read more about this type of error:
 *      >http://varnish/doc/error.html#Unreferenced%20function
 *      >
 *      >
 *      >         Unknown variable 'obj.bandwidth'
 *      >         At: Line 88 Pos 12
 *      >                 if (obj.bandwidth < 1 kb/h) {
 *      >         ------------#############------------
 *      >Read more about this type of error:
 *      >http://varnish/doc/error.html#Unknown%20variable
 *
 */

#include "config.h"

#include <fcntl.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "vcc_compile.h"

#include "libvcc.h"
#include "vfil.h"
#include "vct.h"

static const struct method method_tab[] = {
        { "none", 0U, 0},
#define VCL_MET_MAC(l,U,t,b)    { "vcl_"#l, b, VCL_MET_##U },
#include "tbl/vcl_returns.h"
        { NULL, 0U, 0}
};

struct vcc *vcc_builtin;

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

static void
vcc_vcl_met2c(struct vsb *vsb, unsigned method)
{
        int d = 0;

        //lint -e{774} Boolean within 'if' always evaluates to False
#define VCL_MET_MAC(l,U,t,b)                            \
        do {                                            \
                if (method & VCL_MET_##U) {             \
                        if (d)                          \
                                VSB_putc(vsb, '|');     \
                        VSB_cat(vsb, "VCL_MET_" #U);    \
                        d = 1;                          \
                }                                       \
        } while (0);
#include "tbl/vcl_returns.h"
        AN(d);
}


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

void * v_matchproto_(TlAlloc)
TlAlloc(struct vcc *tl, unsigned len)
{
        void *p;

        (void)tl;
        p = calloc(1, len);
        assert(p != NULL);
        return (p);
}

char *
TlDup(struct vcc *tl, const char *s)
{
        char *p;

        p = TlAlloc(tl, strlen(s) + 1);
        AN(p);
        strcpy(p, s);
        return (p);
}

static int
TLWriteVSB(struct vcc *tl, const char *fn, const struct vsb *vsb,
    const char *what)
{
        int fo;
        int i;

        fo = open(fn, O_WRONLY|O_TRUNC|O_CREAT, 0600);
        if (fo < 0) {
                VSB_printf(tl->sb,
                    "Could not open %s file %s: %s\n",
                    what, fn, strerror(errno));
                return (-1);
        }
        i = VSB_tofile(vsb, fo);
        if (i) {
                VSB_printf(tl->sb,
                    "Could not write %s to %s: %s\n",
                    what, fn, strerror(errno));
        }
        closefd(&fo);
        return (i);
}

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

struct proc *
vcc_NewProc(struct vcc *tl, struct symbol *sym)
{
        struct proc *p;

        ALLOC_OBJ(p, PROC_MAGIC);
        AN(p);
        VTAILQ_INIT(&p->calls);
        VTAILQ_INIT(&p->uses);
        VTAILQ_INSERT_TAIL(&tl->procs, p, list);
        p->prologue = VSB_new_auto();
        AN(p->prologue);
        p->body = VSB_new_auto();
        AN(p->body);
        p->cname = VSB_new_auto();
        AN(p->cname);
        p->okmask = VCL_MET_TASK_ALL;
        sym->proc = p;
        p->sym = sym;
        return (p);
}

static void
vcc_EmitProc(struct vcc *tl, struct proc *p)
{
        struct vsb *vsbm;
        unsigned mask, nsub;
        const char *maskcmp;
        const char *cc_adv;
        int dyn = (p->sym->nref > p->called);

        AN(p->okmask);
        AZ(VSB_finish(p->cname));
        AZ(VSB_finish(p->prologue));
        AZ(VSB_finish(p->body));
        AN(p->sym);

        if (p->method) {
                mask = p->method->bitval;
                maskcmp = "==";
        } else {
                mask = p->okmask;
                maskcmp = "&";
        }

        if (dyn == 0 && (p->calledfrom & VCL_MET_TASK_H) == p->calledfrom)
                cc_adv = "v_dont_optimize ";
        else
                cc_adv = "";

        nsub = tl->nsub++;

        Fh(tl, 1, "vcl_func_f %s;\n", VSB_data(p->cname));
        Fh(tl, 1, "extern const struct vcl_sub sub_%s[1];\n",
            VSB_data(p->cname));
        Fh(tl, 1, "const struct vcl_sub sub_%s[1] = {{\n", VSB_data(p->cname));
        Fh(tl, 1, "\t.magic\t\t= VCL_SUB_MAGIC,\n");
        Fh(tl, 1, "\t.methods\t= 0x%x,\n", p->okmask);
        Fh(tl, 1, "\t.name\t\t= \"%.*s\",\n", PF(p->name));
        Fh(tl, 1, "\t.vcl_conf\t= &VCL_conf,\n");
        Fh(tl, 1, "\t.func\t\t= %s,\n", VSB_data(p->cname));
        Fh(tl, 1, "\t.n\t\t= %d,\n", nsub);
        Fh(tl, 1, "\t.nref\t\t= %d,\n", p->sym->nref);
        Fh(tl, 1, "\t.called\t\t= %d\n", p->called);
        Fh(tl, 1, "\t// calledfrom\t  0x%x\n", p->calledfrom);
        Fh(tl, 1, "}};\n");

        if (dyn) {
                Fc(tl, 1, "\nstatic inline void %s\n", cc_adv);
                Fc(tl, 1, "%s_checked(VRT_CTX)\n{\n", VSB_data(p->cname));
        } else {
                Fc(tl, 1, "\nvoid %sv_matchproto_(vcl_func_f)\n", cc_adv);
                Fc(tl, 1, "%s(VRT_CTX, enum vcl_func_call_e call,\n",
                    VSB_data(p->cname));
                Fc(tl, 1, "    enum vcl_func_fail_e *failp)\n{\n");
                Fc(tl, 1, "  assert(call == VSUB_STATIC);\n");
                Fc(tl, 1, "  assert(failp == NULL);\n");
        }

        vsbm = VSB_new_auto();
        AN(vsbm);
        vcc_vcl_met2c(vsbm, mask);
        AZ(VSB_finish(vsbm));
        Fc(tl, 1, "  assert(ctx->method %s (%s));\n", maskcmp, VSB_data(vsbm));
        VSB_destroy(&vsbm);
        Fc(tl, 1, "%s\n%s}\n", VSB_data(p->prologue), VSB_data(p->body));
        VSB_destroy(&p->body);
        VSB_destroy(&p->prologue);

        if (! dyn) {
                VSB_destroy(&p->cname);
                return;
        }

        /* wrapper to call the actual (_checked) function */
        Fc(tl, 1, "\nvoid v_matchproto_(vcl_func_f)\n");
        Fc(tl, 1, "%s(VRT_CTX, enum vcl_func_call_e call,\n",
            VSB_data(p->cname));
        Fc(tl, 1, "    enum vcl_func_fail_e *failp)\n{\n");
        Fc(tl, 1, "  enum vcl_func_fail_e fail;\n\n");
        Fc(tl, 1, "  fail = VPI_Call_Check(ctx, &VCL_conf, 0x%x, %d);\n",
            mask, nsub);
        Fc(tl, 1, "  if (failp)\n");
        Fc(tl, 1, "    *failp = fail;\n");
        Fc(tl, 1, "  else if (fail == VSUB_E_METHOD)\n");
        Fc(tl, 1, "    VRT_fail(ctx, \"call to \\\"sub %.*s{}\\\""
            " not allowed from here\");\n", PF(p->name));
        Fc(tl, 1, "  else if (fail == VSUB_E_RECURSE)\n");
        Fc(tl, 1, "    VRT_fail(ctx, \"Recursive call to "
            "\\\"sub %.*s{}\\\"\");\n", PF(p->name));
        Fc(tl, 1, "  else\n");
        Fc(tl, 1, "    assert (fail == VSUB_E_OK);\n");
        Fc(tl, 1, "  if (fail != VSUB_E_OK || call == VSUB_CHECK)\n");
        Fc(tl, 1, "    return;\n");
        Fc(tl, 1, "  VPI_Call_Begin(ctx, %d);\n", nsub);
        Fc(tl, 1, "  %s_checked(ctx);\n", VSB_data(p->cname));
        Fc(tl, 1, "  VPI_Call_End(ctx, %d);\n", nsub);
        Fc(tl, 1, "}\n");
        VSB_destroy(&p->cname);
}

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

struct inifin *
New_IniFin(struct vcc *tl)
{
        struct inifin *p;

        ALLOC_OBJ(p, INIFIN_MAGIC);
        AN(p);
        p->ini = VSB_new_auto();
        AN(p->ini);
        p->fin = VSB_new_auto();
        AN(p->fin);
        p->final = VSB_new_auto();
        AN(p->final);
        p->event = VSB_new_auto();
        AN(p->event);
        p->n = ++tl->ninifin;
        VTAILQ_INSERT_TAIL(&tl->inifin, p, list);
        return (p);
}

/*--------------------------------------------------------------------
 * Printf output to the vsbs, possibly indented
 */

void
Fh(const struct vcc *tl, int indent, const char *fmt, ...)
{
        va_list ap;

        if (indent)
                VSB_printf(tl->fh, "%*.*s", tl->hindent, tl->hindent, "");
        va_start(ap, fmt);
        VSB_vprintf(tl->fh, fmt, ap);
        va_end(ap);
}

void
Fb(const struct vcc *tl, int indent, const char *fmt, ...)
{
        va_list ap;

        assert(tl->fb != NULL);
        if (indent)
                VSB_printf(tl->fb, "%*.*s", tl->indent, tl->indent, "");
        va_start(ap, fmt);
        VSB_vprintf(tl->fb, fmt, ap);
        va_end(ap);
}

void
Fc(const struct vcc *tl, int indent, const char *fmt, ...)
{
        va_list ap;

        if (indent)
                VSB_printf(tl->fc, "%*.*s", tl->indent, tl->indent, "");
        va_start(ap, fmt);
        VSB_vprintf(tl->fc, fmt, ap);
        va_end(ap);
}

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

void
EncToken(struct vsb *sb, const struct token *t)
{

        assert(t->tok == CSTR);
        VSB_quote(sb, t->dec, -1, VSB_QUOTE_CSTR);
}

/*--------------------------------------------------------------------
 * Output the location/profiling table.  For each counted token, we
 * record source+line+charpos for the first character in the token.
 */

static void
EmitCoordinates(const struct vcc *tl, struct vsb *vsb)
{
        struct token *t;
        unsigned lin, pos;
        const struct source *sp;
        const char *p;

        VSB_cat(vsb, "/* ---===### Source Code ###===---*/\n");

        VSB_printf(vsb, "\n#define VGC_NSRCS %u\n", tl->nsources);

        VSB_cat(vsb, "\nstatic const char *srcname[VGC_NSRCS] = {\n");
        VTAILQ_FOREACH(sp, &tl->sources, list) {
                VSB_cat(vsb, "\t");
                VSB_quote(vsb, sp->name, -1, VSB_QUOTE_CSTR);
                VSB_cat(vsb, ",\n");
        }
        VSB_cat(vsb, "};\n");

        VSB_printf(vsb, "\nstatic const char *srcbody[%u] = {\n", tl->nsources);
        VTAILQ_FOREACH(sp, &tl->sources, list) {
                VSB_cat(vsb, "    /* ");
                VSB_quote(vsb, sp->name, -1, VSB_QUOTE_CSTR);
                VSB_cat(vsb, " */\n");
                VSB_quote_pfx(vsb, "\t", sp->b, sp->e - sp->b, VSB_QUOTE_CSTR);
                VSB_cat(vsb, ",\n");
        }
        VSB_cat(vsb, "};\n\n");

        VSB_cat(vsb, "/* ---===### Location Counters ###===---*/\n");

        VSB_printf(vsb, "\n#define VGC_NREFS %u\n\n", tl->cnt + 1);

        VSB_cat(vsb, "static const struct vpi_ref VGC_ref[VGC_NREFS] = {\n");
        lin = 1;
        pos = 0;
        sp = 0;
        p = NULL;
        VTAILQ_FOREACH(t, &tl->tokens, list) {
                if (t->cnt == 0)
                        continue;
                assert(t->src != NULL);
                if (t->src != sp) {
                        lin = 1;
                        pos = 0;
                        sp = t->src;
                        p = sp->b;
                }
                assert(sp != NULL);
                assert(p != NULL);
                for (;p < t->b; p++) {
                        if (*p == '\n') {
                                lin++;
                                pos = 0;
                        } else if (*p == '\t') {
                                pos &= ~7;
                                pos += 8;
                        } else
                                pos++;

                }
                VSB_printf(vsb, "  [%3u] = { VPI_REF_MAGIC, %u, %8tu, %4u, %3u, ",
                    t->cnt, sp->idx, t->b - sp->b, lin, pos + 1);
                if (t->tok == CSRC)
                        VSB_cat(vsb, " \"C{\"},\n");
                else
                        VSB_printf(vsb, " \"%.*s\" },\n", PF(t));
        }
        VSB_cat(vsb, "};\n\n");
}

/*--------------------------------------------------------------------
 * Init/Fini/Event
 *
 * We call DISCARD and COLD events in the opposite order of LOAD and
 * WARM.
 */

static void
EmitInitFini(const struct vcc *tl)
{
        struct inifin *p, *q = NULL;
        unsigned has_event = 0;
        struct symbol *sy;

        Fh(tl, 0, "\n");
        Fh(tl, 0, "static unsigned vgc_inistep;\n");
        Fh(tl, 0, "static unsigned vgc_warmupstep;\n");

        /*
         * LOAD
         */
        Fc(tl, 0, "\nstatic int\nVGC_Load(VRT_CTX)\n{\n\n");
        Fc(tl, 0, "\tvgc_inistep = 0;\n");
        Fc(tl, 0, "\tsize_t ndirector = %dUL;\n", tl->ndirector);
        Fc(tl, 0, "\n");
        VTAILQ_FOREACH(p, &tl->inifin, list) {
                AZ(VSB_finish(p->ini));
                assert(p->n > 0);
                if (VSB_len(p->ini))
                        Fc(tl, 0, "\t/* %u */\n%s\n", p->n, VSB_data(p->ini));
                if (p->ignore_errors == 0) {
                        Fc(tl, 0, "\tif (ctx->vpi->handling == VCL_RET_FAIL)\n");
                        Fc(tl, 0, "\t\treturn(1);\n");
                }
                Fc(tl, 0, "\tvgc_inistep = %u;\n\n", p->n);
                VSB_destroy(&p->ini);

                AZ(VSB_finish(p->event));
                if (VSB_len(p->event))
                        has_event = 1;
        }

        /* Handle failures from vcl_init */
        Fc(tl, 0, "\n");
        Fc(tl, 0, "\tif (ctx->vpi->handling != VCL_RET_OK)\n");
        Fc(tl, 0, "\t\treturn(1);\n");
        Fc(tl, 0, "\tctx->vpi->handling = 0;\n");

        VTAILQ_FOREACH(sy, &tl->sym_objects, sideways) {
                Fc(tl, 0, "\tif (!%s) {\n", sy->rname);
                Fc(tl, 0, "\t\tVRT_fail(ctx, "
                    "\"Object %s not initialized\");\n" , sy->name);
                Fc(tl, 0, "\t\treturn(1);\n");
                Fc(tl, 0, "\t}\n");
        }

        Fc(tl, 0, "\treturn(0);\n");
        Fc(tl, 0, "}\n");

        /*
         * DISCARD
         */
        Fc(tl, 0, "\nstatic int\nVGC_Discard(VRT_CTX)\n{\n\n");

        Fc(tl, 0, "\tswitch (vgc_inistep) {\n");
        VTAILQ_FOREACH_REVERSE(p, &tl->inifin, inifinhead, list) {
                AZ(VSB_finish(p->fin));
                if (q)
                        assert(q->n > p->n);
                q = p;
                Fc(tl, 0, "\t\tcase %u:\n", p->n);
                if (VSB_len(p->fin))
                        Fc(tl, 0, "\t%s\n", VSB_data(p->fin));
                Fc(tl, 0, "\t\t\t/* FALLTHROUGH */\n");
                VSB_destroy(&p->fin);
        }
        Fc(tl, 0, "\t\tdefault:\n\t\t\tbreak;\n");
        Fc(tl, 0, "\t}\n\n");
        Fc(tl, 0, "\tswitch (vgc_inistep) {\n");
        VTAILQ_FOREACH_REVERSE(p, &tl->inifin, inifinhead, list) {
                AZ(VSB_finish(p->final));
                Fc(tl, 0, "\t\tcase %u:\n", p->n);
                if (VSB_len(p->final))
                        Fc(tl, 0, "\t%s\n", VSB_data(p->final));
                Fc(tl, 0, "\t\t\t/* FALLTHROUGH */\n");
                VSB_destroy(&p->final);
        }
        Fc(tl, 0, "\t\tdefault:\n\t\t\tbreak;\n");
        Fc(tl, 0, "\t}\n\n");

        Fc(tl, 0, "\treturn (0);\n");
        Fc(tl, 0, "}\n");

        if (has_event) {
                /*
                 * WARM
                 */
                Fc(tl, 0, "\nstatic int\n");
                Fc(tl, 0, "VGC_Warmup(VRT_CTX, enum vcl_event_e ev)\n{\n\n");

                Fc(tl, 0, "\tvgc_warmupstep = 0;\n\n");
                VTAILQ_FOREACH(p, &tl->inifin, list) {
                        assert(p->n > 0);
                        if (VSB_len(p->event)) {
                                Fc(tl, 0, "\t/* %u */\n", p->n);
                                Fc(tl, 0, "\tif (%s)\n", VSB_data(p->event));
                                Fc(tl, 0, "\t\treturn (1);\n");
                                Fc(tl, 0, "\tvgc_warmupstep = %u;\n\n", p->n);
                        }
                }

                Fc(tl, 0, "\treturn (0);\n");
                Fc(tl, 0, "}\n");

                /*
                 * COLD
                 */
                Fc(tl, 0, "\nstatic int\n");
                Fc(tl, 0, "VGC_Cooldown(VRT_CTX, enum vcl_event_e ev)\n{\n");
                Fc(tl, 0, "\tint retval = 0;\n\n");

                VTAILQ_FOREACH_REVERSE(p, &tl->inifin, inifinhead, list) {
                        if (VSB_len(p->event)) {
                                Fc(tl, 0, "\t/* %u */\n", p->n);
                                Fc(tl, 0,
                                    "\tif (vgc_warmupstep >= %u &&\n", p->n);
                                Fc(tl, 0,
                                    "\t    %s != 0)\n", VSB_data(p->event));
                                Fc(tl, 0, "\t\tretval = 1;\n\n");
                        }
                        VSB_destroy(&p->event);
                }

                Fc(tl, 0, "\treturn (retval);\n");
                Fc(tl, 0, "}\n");
        }

        /*
         * EVENTS
         */
        Fc(tl, 0, "\nstatic int\n");
        Fc(tl, 0, "VGC_Event(VRT_CTX, enum vcl_event_e ev)\n");
        Fc(tl, 0, "{\n");
        Fc(tl, 0, "\tif (ev == VCL_EVENT_LOAD)\n");
        Fc(tl, 0, "\t\treturn (VGC_Load(ctx));\n");
        if (has_event) {
                Fc(tl, 0, "\tif (ev == VCL_EVENT_WARM)\n");
                Fc(tl, 0, "\t\treturn (VGC_Warmup(ctx, ev));\n");
                Fc(tl, 0, "\tif (ev == VCL_EVENT_COLD)\n");
                Fc(tl, 0, "\t\treturn (VGC_Cooldown(ctx, ev));\n");
        }
        Fc(tl, 0, "\tif (ev == VCL_EVENT_DISCARD)\n");
        Fc(tl, 0, "\t\treturn (VGC_Discard(ctx));\n");
        Fc(tl, 0, "\n");
        if (!has_event)
                Fc(tl, 0, "\t(void)vgc_warmupstep;\n");
        Fc(tl, 0, "\treturn (%d);\n", has_event ? 1 : 0);
        Fc(tl, 0, "}\n");
}

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

static void
EmitStruct(const struct vcc *tl)
{
        Fc(tl, 0, "\nconst struct VCL_conf VCL_conf = {\n");
        Fc(tl, 0, "\t.magic = VCL_CONF_MAGIC,\n");
        Fc(tl, 0, "\t.syntax = %u,\n", tl->syntax);
        Fc(tl, 0, "\t.event_vcl = VGC_Event,\n");
        Fc(tl, 0, "\t.default_director = &%s,\n", tl->default_director);
        if (tl->default_probe != NULL)
                Fc(tl, 0, "\t.default_probe = %s,\n", tl->default_probe);
        Fc(tl, 0, "\t.ref = VGC_ref,\n");
        Fc(tl, 0, "\t.nref = VGC_NREFS,\n");
        Fc(tl, 0, "\t.nsrc = VGC_NSRCS,\n");
        Fc(tl, 0, "\t.nsub = %d,\n", tl->subref > 0 ? tl->nsub : 0);
        Fc(tl, 0, "\t.srcname = srcname,\n");
        Fc(tl, 0, "\t.srcbody = srcbody,\n");
        Fc(tl, 0, "\t.nvmod = %u,\n", tl->vmod_count);
#define VCL_MET_MAC(l,u,t,b) \
        Fc(tl, 0, "\t." #l "_func = VGC_function_vcl_" #l ",\n");
#include "tbl/vcl_returns.h"
        Fc(tl, 0, "\t.instance_info = VGC_instance_info\n");
        Fc(tl, 0, "};\n");
}

/*--------------------------------------------------------------------
 * Compile the VCL code from the given source and return the C-source
 */

static struct vsb *
vcc_CompileSource(struct vcc *tl, struct source *sp, const char *jfile)
{
        struct proc *p;
        struct vsb *vsb;
        struct inifin *ifp;

        Fh(tl, 0, "/* ---===### VCC generated .h code ###===---*/\n");
        Fc(tl, 0, "\n/* ---===### VCC generated .c code ###===---*/\n");

        Fc(tl, 0, "\n#define END_ if (ctx->vpi->handling) return\n");

        vcc_Parse_Init(tl);

        vcc_Expr_Init(tl);

        vcc_Action_Init(tl);

        vcc_Backend_Init(tl);

        vcc_Var_Init(tl);

        vcc_Type_Init(tl);

        Fh(tl, 0, "\nextern const struct VCL_conf VCL_conf;\n");

        /* Register and lex the main source */
        if (sp != NULL) {
                AN(vcc_builtin);
                vcc_lex_source(tl, sp, 0);
                if (tl->err)
                        return (NULL);
        }

        /* Register and lex the builtin VCL */
        sp = vcc_new_source(tl->builtin_vcl, "builtin", "<builtin>");
        assert(sp != NULL);
        vcc_lex_source(tl, sp, 1);
        if (tl->err)
                return (NULL);

        /* Expand and lex any includes in the token string */
        if (tl->err)
                return (NULL);

        /* Parse the token string */
        tl->t = VTAILQ_FIRST(&tl->tokens);
        vcc_Parse(tl);
        if (tl->err)
                return (NULL);

        /* Check for orphans */
        if (vcc_CheckReferences(tl))
                return (NULL);

        /* Check that all action returns are legal */
        if (vcc_CheckAction(tl) || tl->err)
                return (NULL);

        /* Check that all variable uses are legal */
        if (vcc_CheckUses(tl) || tl->err)
                return (NULL);

        if (vcc_builtin == NULL)
                return (NULL);

        /* Check if we have any backends at all */
        if (tl->default_director == NULL) {
                VSB_cat(tl->sb,
                    "No backends or directors found in VCL program, "
                    "at least one is necessary.\n");
                tl->err = 1;
                return (NULL);
        }

        /* Tie vcl_init/fini in */
        ifp = New_IniFin(tl);
        VSB_cat(ifp->ini, "\tVGC_function_vcl_init(ctx, VSUB_STATIC, NULL);\n");
        /*
         * Because the failure could be half way into vcl_init{} so vcl_fini{}
         * must always be called, also on failure.
         */
        ifp->ignore_errors = 1;
        VSB_cat(ifp->fin, "\t\tVGC_function_vcl_fini(ctx, VSUB_STATIC, NULL);\n");
        VSB_cat(ifp->fin, "\t\t\tVPI_vcl_fini(ctx);");

        /* Emit method functions */
        Fh(tl, 1, "\n");
        VTAILQ_FOREACH(p, &tl->procs, list)
                if (p->method == NULL)
                        vcc_EmitProc(tl, p);
        VTAILQ_FOREACH(p, &tl->procs, list)
                if (p->method != NULL)
                        vcc_EmitProc(tl, p);

        EmitInitFini(tl);

        VCC_InstanceInfo(tl);

        EmitStruct(tl);

        VCC_XrefTable(tl);

        VSB_cat(tl->symtab, "\n]\n");
        AZ(VSB_finish(tl->symtab));
        if (TLWriteVSB(tl, jfile, tl->symtab, "Symbol table"))
                return (NULL);

        /* Combine it all */

        vsb = VSB_new_auto();
        AN(vsb);

        vcl_output_lang_h(vsb);

        EmitCoordinates(tl, vsb);

        AZ(VSB_finish(tl->fh));
        VSB_cat(vsb, VSB_data(tl->fh));

        AZ(VSB_finish(tl->fc));
        VSB_cat(vsb, VSB_data(tl->fc));

        AZ(VSB_finish(vsb));
        return (vsb);
}

static struct vcc *
vcc_ParseBuiltin(struct vcc *tl)
{
        struct vcc *tl_builtin;

        CHECK_OBJ_NOTNULL(tl, VCC_MAGIC);
        tl_builtin = VCC_New();
        AN(tl_builtin);
        VCC_Builtin_VCL(tl_builtin, tl->builtin_vcl);
        AZ(vcc_CompileSource(tl_builtin, NULL, NULL));
        return (tl_builtin);
}

/*--------------------------------------------------------------------
 * Report the range of VCL language we support
 */
void
VCC_VCL_Range(unsigned *lo, unsigned *hi)
{

        AN(lo);
        *lo = VCL_LOW;
        AN(hi);
        *hi = VCL_HIGH;
}

/*--------------------------------------------------------------------
 * Compile the VCL code in the argument.  Error messages, if any are
 * formatted into the vsb.
 */

int
VCC_Compile(struct vcc *tl, struct vsb **sb,
    const char *vclsrc, const char *vclsrcfile,
    const char *ofile, const char *jfile)
{
        struct source *sp;
        struct vsb *r = NULL;
        int retval = 0;

        CHECK_OBJ_NOTNULL(tl, VCC_MAGIC);
        AN(sb);
        AN(vclsrcfile);
        AN(ofile);
        AN(jfile);

        AZ(vcc_builtin);
        vcc_builtin = vcc_ParseBuiltin(tl);
        AN(vcc_builtin);
        if (vcc_builtin->err) {
                AZ(VSB_finish(vcc_builtin->sb));
                *sb = vcc_builtin->sb;
                return (-1);
        }

        if (vclsrc != NULL)
                sp = vcc_new_source(vclsrc, "vcl.inline", vclsrcfile);
        else
                sp = vcc_file_source(tl, vclsrcfile);

        if (sp != NULL)
                r = vcc_CompileSource(tl, sp, jfile);

        if (r != NULL) {
                retval = TLWriteVSB(tl, ofile, r, "C-source");
                VSB_destroy(&r);
        } else {
                retval = -1;
        }
        AZ(VSB_finish(tl->sb));
        *sb = tl->sb;
        return (retval);
}

/*--------------------------------------------------------------------
 * Allocate a compiler instance
 */

struct vcc *
VCC_New(void)
{
        struct vcc *tl;
        struct symbol *sym;
        struct proc *p;
        struct vsb *vsb;
        int i;

        ALLOC_OBJ(tl, VCC_MAGIC);
        AN(tl);
        VTAILQ_INIT(&tl->inifin);
        VTAILQ_INIT(&tl->tokens);
        VTAILQ_INIT(&tl->sources);
        VTAILQ_INIT(&tl->procs);
        VTAILQ_INIT(&tl->sym_objects);
        VTAILQ_INIT(&tl->sym_vmods);
        VTAILQ_INIT(&tl->vmod_objects);

        tl->nsources = 0;

        tl->symtab = VSB_new_auto();
        assert(tl->symtab != NULL);
        VSB_cat(tl->symtab, "[\n    {\"version\": 0}");

        tl->fc = VSB_new_auto();
        assert(tl->fc != NULL);

        tl->fh = VSB_new_auto();
        assert(tl->fh != NULL);

        for (i = 1; i < VCL_MET_MAX; i++) {
                sym = VCC_MkSym(tl, method_tab[i].name,
                    SYM_MAIN, SYM_SUB, VCL_LOW, VCL_HIGH);
                p = vcc_NewProc(tl, sym);
                p->method = &method_tab[i];

                // see also VCC_GlobalSymbol()
                vsb = VSB_new_auto();
                AN(vsb);
                VSB_printf(vsb, "%s_%s", SUB->global_pfx, p->method->name);
                AZ(VSB_finish(vsb));

                AZ(VSB_bcat(p->cname, VSB_data(vsb), VSB_len(vsb)));

                sym->lname = strdup(VSB_data(vsb));
                AN(sym->lname);

                VSB_clear(vsb);
                VSB_printf(vsb, "sub_%s", sym->lname);
                AZ(VSB_finish(vsb));

                sym->rname = strdup(VSB_data(vsb));
                AN(sym->rname);
                VSB_destroy(&vsb);

                sym->type = SUB;
                sym->kind = VCC_HandleKind(SUB);
                AZ(VCT_invalid_name(sym->rname, NULL));
                sym->eval = vcc_Eval_Sub;
        }
        tl->sb = VSB_new_auto();
        AN(tl->sb);
        return (tl);
}

/*--------------------------------------------------------------------
 * Configure builtin VCL source code
 */

void
VCC_Builtin_VCL(struct vcc *vcc, const char *str)
{

        CHECK_OBJ_NOTNULL(vcc, VCC_MAGIC);
        REPLACE(vcc->builtin_vcl, str);
}

/*--------------------------------------------------------------------
 * Configure default VCL source path
 */

void
VCC_VCL_path(struct vcc *vcc, const char *str)
{

        CHECK_OBJ_NOTNULL(vcc, VCC_MAGIC);
        VFIL_setpath(&vcc->vcl_path, str);
}

/*--------------------------------------------------------------------
 * Configure default VMOD path
 */

void
VCC_VMOD_path(struct vcc *vcc, const char *str)
{

        CHECK_OBJ_NOTNULL(vcc, VCC_MAGIC);
        VFIL_setpath(&vcc->vmod_path, str);
}

/*--------------------------------------------------------------------
 * Configure settings
 */

#define VCC_FEATURE_BIT(U, l, d)                                \
        void VCC_Opt_ ## l(struct vcc *vcc, unsigned val)       \
        {                                                       \
                CHECK_OBJ_NOTNULL(vcc, VCC_MAGIC);              \
                vcc->l = val;                                   \
        }
#include "tbl/vcc_feature_bits.h"

/*--------------------------------------------------------------------
 * Configure settings
 */

static void
vcc_predef_vcl(struct vcc *vcc, const char *name)
{
        struct symbol *sym;

        sym = VCC_MkSym(vcc, name, SYM_MAIN, SYM_VCL, VCL_LOW, VCL_HIGH);
        AN(sym);
        sym->type = VCL;
        sym->r_methods = VCL_MET_RECV;
}

void
VCC_Predef(struct vcc *vcc, const char *type, const char *name)
{

        CHECK_OBJ_NOTNULL(vcc, VCC_MAGIC);
        if (!strcmp(type, "VCL_STEVEDORE"))
                vcc_stevedore(vcc, name);
        else if (!strcmp(type, "VCL_VCL"))
                vcc_predef_vcl(vcc, name);
        else
                WRONG("Unknown VCC predef type");
}

void
VCC_VEXT(struct vcc *vcc, const char *filename)
{
        CHECK_OBJ_NOTNULL(vcc, VCC_MAGIC);
        vcc_ImportVext(vcc, filename);
}

		varnish-cache/lib/libvcc/vcc_compile.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		/*
32		* XXX:
33		* Better error messages, throughout.
34		* >It also occurred to me that we could link the errors to the error
35		* >documentation.
36		* >
37		* >Unreferenced function 'request_policy', first mention is
38		* > Line 8 Pos 4
39		* > sub request_policy {
40		* > ----##############--
41		* >Read more about this type of error:
42		* >http://varnish/doc/error.html#Unreferenced%20function
43		* >
44		* >
45		* > Unknown variable 'obj.bandwidth'
46		* > At: Line 88 Pos 12
47		* > if (obj.bandwidth < 1 kb/h) {
48		* > ------------#############------------
49		* >Read more about this type of error:
50		* >http://varnish/doc/error.html#Unknown%20variable
51		*
52		*/
53
54		#include "config.h"
55
56		#include <fcntl.h>
57		#include <stdarg.h>
58		#include <stdlib.h>
59		#include <string.h>
60		#include <unistd.h>
61
62		#include "vcc_compile.h"
63
64		#include "libvcc.h"
65		#include "vfil.h"
66		#include "vct.h"
67
68		static const struct method method_tab[] = {
69		{ "none", 0U, 0},
70		#define VCL_MET_MAC(l,U,t,b) { "vcl_"#l, b, VCL_MET_##U },
71		#include "tbl/vcl_returns.h"
72		{ NULL, 0U, 0}
73		};
74
75		struct vcc *vcc_builtin;
76
77		/--------------------------------------------------------------------/
78
79		static void
80	1673920	vcc_vcl_met2c(struct vsb *vsb, unsigned method)
81		{
82	1673920	int d = 0;
83
84		//lint -e{774} Boolean within 'if' always evaluates to False
85		#define VCL_MET_MAC(l,U,t,b) \
86		do { \
87		if (method & VCL_MET_##U) { \
88		if (d) \
89		VSB_putc(vsb, '\|'); \
90		VSB_cat(vsb, "VCL_MET_" #U); \
91		d = 1; \
92		} \
93		} while (0);
94		#include "tbl/vcl_returns.h"
95	1673920	AN(d);
96		}
97
98
99		/--------------------------------------------------------------------/
100
101		void * v_matchproto_(TlAlloc)
102	136008080	TlAlloc(struct vcc *tl, unsigned len)
103		{
104		void *p;
105
106	136008080	(void)tl;
107	136008080	p = calloc(1, len);
108	136008080	assert(p != NULL);
109	136008080	return (p);
110		}
111
112	1673920	char *
113	6942160	TlDup(struct vcc tl, const char s)
114		{
115		char *p;
116
117	6942160	p = TlAlloc(tl, strlen(s) + 1);
118	6942160	AN(p);
119	8616080	strcpy(p, s);
120	6942160	return (p);
121		}
122
123		static int
124	95520	TLWriteVSB(struct vcc tl, const char fn, const struct vsb *vsb,
125		const char *what)
126	1673920	{
127		int fo;
128		int i;
129
130	95520	fo = open(fn, O_WRONLY\|O_TRUNC\|O_CREAT, 0600);
131	95520	if (fo < 0) {
132	0	VSB_printf(tl->sb,
133		"Could not open %s file %s: %s\n",
134	0	what, fn, strerror(errno));
135	1673920	return (-1);
136		}
137	95520	i = VSB_tofile(vsb, fo);
138	95520	if (i) {
139	0	VSB_printf(tl->sb,
140		"Could not write %s to %s: %s\n",
141	0	what, fn, strerror(errno));
142	1673920	}
143	95520	closefd(&fo);
144	95520	return (i);
145	95520	}
146	1673920
147		/--------------------------------------------------------------------/
148
149		struct proc *
150	3959720	vcc_NewProc(struct vcc tl, struct symbol sym)
151	1673920	{
152		struct proc *p;
153
154	3959720	ALLOC_OBJ(p, PROC_MAGIC);
155	3959720	AN(p);
156	3959720	VTAILQ_INIT(&p->calls);
157	3959720	VTAILQ_INIT(&p->uses);
158	3959720	VTAILQ_INSERT_TAIL(&tl->procs, p, list);
159	5633640	p->prologue = VSB_new_auto();
160	3959720	AN(p->prologue);
161	3959720	p->body = VSB_new_auto();
162	3959720	AN(p->body);
163	3959720	p->cname = VSB_new_auto();
164	5633640	AN(p->cname);
165	3959720	p->okmask = VCL_MET_TASK_ALL;
166	3959720	sym->proc = p;
167	3959720	p->sym = sym;
168	3959720	return (p);
169		}
170
171		static void
172	3347840	vcc_EmitProc(struct vcc tl, struct proc p)
173		{
174		struct vsb *vsbm;
175		unsigned mask, nsub;
176		const char *maskcmp;
177		const char *cc_adv;
178	1673920	int dyn = (p->sym->nref > p->called);
179	1673920
180	1673920	AN(p->okmask);
181	1673920	AZ(VSB_finish(p->cname));
182	1673920	AZ(VSB_finish(p->prologue));
183	1673920	AZ(VSB_finish(p->body));
184	1673920	AN(p->sym);
185	1673920
186	1673920	if (p->method) {
187	668640	mask = p->method->bitval;
188	668640	maskcmp = "==";
189	668640	} else {
190	1005280	mask = p->okmask;
191	2679200	maskcmp = "&";
192		}
193
194	1673920	if (dyn == 0 && (p->calledfrom & VCL_MET_TASK_H) == p->calledfrom)
195	96000	cc_adv = "v_dont_optimize ";
196		else
197	1577920	cc_adv = "";
198
199	1673920	nsub = tl->nsub++;
200
201	1673920	Fh(tl, 1, "vcl_func_f %s;\n", VSB_data(p->cname));
202	5021760	Fh(tl, 1, "extern const struct vcl_sub sub_%s[1];\n",
203	1673920	VSB_data(p->cname));
204	1673920	Fh(tl, 1, "const struct vcl_sub sub_%s[1] = {{\n", VSB_data(p->cname));
205	1673920	Fh(tl, 1, "\t.magic\t\t= VCL_SUB_MAGIC,\n");
206	1673920	Fh(tl, 1, "\t.methods\t= 0x%x,\n", p->okmask);
207	1673920	Fh(tl, 1, "\t.name\t\t= \"%.*s\",\n", PF(p->name));
208	1673920	Fh(tl, 1, "\t.vcl_conf\t= &VCL_conf,\n");
209	1673920	Fh(tl, 1, "\t.func\t\t= %s,\n", VSB_data(p->cname));
210	1673920	Fh(tl, 1, "\t.n\t\t= %d,\n", nsub);
211	1673920	Fh(tl, 1, "\t.nref\t\t= %d,\n", p->sym->nref);
212	1673920	Fh(tl, 1, "\t.called\t\t= %d\n", p->called);
213	1673920	Fh(tl, 1, "\t// calledfrom\t 0x%x\n", p->calledfrom);
214	1673920	Fh(tl, 1, "}};\n");
215
216	1673920	if (dyn) {
217	1120	Fc(tl, 1, "\nstatic inline void %s\n", cc_adv);
218	1120	Fc(tl, 1, "%s_checked(VRT_CTX)\n{\n", VSB_data(p->cname));
219	1120	} else {
220	1672800	Fc(tl, 1, "\nvoid %sv_matchproto_(vcl_func_f)\n", cc_adv);
221	3345600	Fc(tl, 1, "%s(VRT_CTX, enum vcl_func_call_e call,\n",
222	1672800	VSB_data(p->cname));
223	1672800	Fc(tl, 1, " enum vcl_func_fail_e *failp)\n{\n");
224	1672800	Fc(tl, 1, " assert(call == VSUB_STATIC);\n");
225	1672800	Fc(tl, 1, " assert(failp == NULL);\n");
226		}
227
228	1673920	vsbm = VSB_new_auto();
229	1673920	AN(vsbm);
230	1673920	vcc_vcl_met2c(vsbm, mask);
231	1673920	AZ(VSB_finish(vsbm));
232	1673920	Fc(tl, 1, " assert(ctx->method %s (%s));\n", maskcmp, VSB_data(vsbm));
233	1673920	VSB_destroy(&vsbm);
234	1673920	Fc(tl, 1, "%s\n%s}\n", VSB_data(p->prologue), VSB_data(p->body));
235	1673920	VSB_destroy(&p->body);
236	1673920	VSB_destroy(&p->prologue);
237
238	1673920	if (! dyn) {
239	1672800	VSB_destroy(&p->cname);
240	1672800	return;
241		}
242
243		/* wrapper to call the actual (_checked) function */
244	1120	Fc(tl, 1, "\nvoid v_matchproto_(vcl_func_f)\n");
245	2240	Fc(tl, 1, "%s(VRT_CTX, enum vcl_func_call_e call,\n",
246	1120	VSB_data(p->cname));
247	1120	Fc(tl, 1, " enum vcl_func_fail_e *failp)\n{\n");
248	1120	Fc(tl, 1, " enum vcl_func_fail_e fail;\n\n");
249	2240	Fc(tl, 1, " fail = VPI_Call_Check(ctx, &VCL_conf, 0x%x, %d);\n",
250	1120	mask, nsub);
251	1120	Fc(tl, 1, " if (failp)\n");
252	1120	Fc(tl, 1, " *failp = fail;\n");
253	1120	Fc(tl, 1, " else if (fail == VSUB_E_METHOD)\n");
254	2240	Fc(tl, 1, " VRT_fail(ctx, \"call to \\\"sub %.*s{}\\\""
255	1120	" not allowed from here\");\n", PF(p->name));
256	1120	Fc(tl, 1, " else if (fail == VSUB_E_RECURSE)\n");
257	2240	Fc(tl, 1, " VRT_fail(ctx, \"Recursive call to "
258	1120	"\\\"sub %.*s{}\\\"\");\n", PF(p->name));
259	1120	Fc(tl, 1, " else\n");
260	1120	Fc(tl, 1, " assert (fail == VSUB_E_OK);\n");
261	1120	Fc(tl, 1, " if (fail != VSUB_E_OK \|\| call == VSUB_CHECK)\n");
262	1120	Fc(tl, 1, " return;\n");
263	1120	Fc(tl, 1, " VPI_Call_Begin(ctx, %d);\n", nsub);
264	1120	Fc(tl, 1, " %s_checked(ctx);\n", VSB_data(p->cname));
265	1120	Fc(tl, 1, " VPI_Call_End(ctx, %d);\n", nsub);
266	1120	Fc(tl, 1, "}\n");
267	1120	VSB_destroy(&p->cname);
268	1673920	}
269
270		/--------------------------------------------------------------------/
271
272		struct inifin *
273	599520	New_IniFin(struct vcc *tl)
274		{
275		struct inifin *p;
276
277	599520	ALLOC_OBJ(p, INIFIN_MAGIC);
278	599520	AN(p);
279	599520	p->ini = VSB_new_auto();
280	599520	AN(p->ini);
281	599520	p->fin = VSB_new_auto();
282	599520	AN(p->fin);
283	599520	p->final = VSB_new_auto();
284	599520	AN(p->final);
285	599520	p->event = VSB_new_auto();
286	599520	AN(p->event);
287	599520	p->n = ++tl->ninifin;
288	599520	VTAILQ_INSERT_TAIL(&tl->inifin, p, list);
289	599520	return (p);
290		}
291
292		/*--------------------------------------------------------------------
293		* Printf output to the vsbs, possibly indented
294		*/
295
296		void
297	30054440	Fh(const struct vcc tl, int indent, const char fmt, ...)
298		{
299		va_list ap;
300
301	30054440	if (indent)
302	21808880	VSB_printf(tl->fh, "%.s", tl->hindent, tl->hindent, "");
303	30054440	va_start(ap, fmt);
304	30054440	VSB_vprintf(tl->fh, fmt, ap);
305	30054440	va_end(ap);
306	30054440	}
307
308		void
309	75694040	Fb(const struct vcc tl, int indent, const char fmt, ...)
310		{
311		va_list ap;
312
313	75694040	assert(tl->fb != NULL);
314	75694040	if (indent)
315	66350000	VSB_printf(tl->fb, "%.s", tl->indent, tl->indent, "");
316	75694040	va_start(ap, fmt);
317	75694040	VSB_vprintf(tl->fb, fmt, ap);
318	75694040	va_end(ap);
319	75694040	}
320
321		void
322	60016040	Fc(const struct vcc tl, int indent, const char fmt, ...)
323		{
324		va_list ap;
325
326	60016040	if (indent)
327	11735360	VSB_printf(tl->fc, "%.s", tl->indent, tl->indent, "");
328	60016040	va_start(ap, fmt);
329	60016040	VSB_vprintf(tl->fc, fmt, ap);
330	60016040	va_end(ap);
331	60016040	}
332
333		/--------------------------------------------------------------------/
334
335		void
336	3745240	EncToken(struct vsb sb, const struct token t)
337		{
338
339	3745240	assert(t->tok == CSTR);
340	3745240	VSB_quote(sb, t->dec, -1, VSB_QUOTE_CSTR);
341	3745240	}
342
343		/*--------------------------------------------------------------------
344		* Output the location/profiling table. For each counted token, we
345		* record source+line+charpos for the first character in the token.
346		*/
347
348		static void
349	47760	EmitCoordinates(const struct vcc tl, struct vsb vsb)
350		{
351		struct token *t;
352		unsigned lin, pos;
353		const struct source *sp;
354		const char *p;
355
356	47760	VSB_cat(vsb, "/* ---===### Source Code ###===---*/\n");
357
358	47760	VSB_printf(vsb, "\n#define VGC_NSRCS %u\n", tl->nsources);
359
360	47760	VSB_cat(vsb, "\nstatic const char *srcname[VGC_NSRCS] = {\n");
361	144520	VTAILQ_FOREACH(sp, &tl->sources, list) {
362	96760	VSB_cat(vsb, "\t");
363	96760	VSB_quote(vsb, sp->name, -1, VSB_QUOTE_CSTR);
364	96760	VSB_cat(vsb, ",\n");
365	96760	}
366	47760	VSB_cat(vsb, "};\n");
367
368	47760	VSB_printf(vsb, "\nstatic const char *srcbody[%u] = {\n", tl->nsources);
369	144520	VTAILQ_FOREACH(sp, &tl->sources, list) {
370	96760	VSB_cat(vsb, " /* ");
371	96760	VSB_quote(vsb, sp->name, -1, VSB_QUOTE_CSTR);
372	96760	VSB_cat(vsb, " */\n");
373	96760	VSB_quote_pfx(vsb, "\t", sp->b, sp->e - sp->b, VSB_QUOTE_CSTR);
374	96760	VSB_cat(vsb, ",\n");
375	96760	}
376	47760	VSB_cat(vsb, "};\n\n");
377
378	47760	VSB_cat(vsb, "/* ---===### Location Counters ###===---*/\n");
379
380	47760	VSB_printf(vsb, "\n#define VGC_NREFS %u\n\n", tl->cnt + 1);
381
382	47760	VSB_cat(vsb, "static const struct vpi_ref VGC_ref[VGC_NREFS] = {\n");
383	47760	lin = 1;
384	47760	pos = 0;
385	47760	sp = 0;
386	47760	p = NULL;
387	39300320	VTAILQ_FOREACH(t, &tl->tokens, list) {
388	39252560	if (t->cnt == 0)
389	36126840	continue;
390	3125720	assert(t->src != NULL);
391	3125720	if (t->src != sp) {
392	80760	lin = 1;
393	80760	pos = 0;
394	80760	sp = t->src;
395	80760	p = sp->b;
396	80760	}
397	3125720	assert(sp != NULL);
398	3125720	assert(p != NULL);
399	308121221	for (;p < t->b; p++) {
400	304995501	if (*p == '\n') {
401	14104800	lin++;
402	14104800	pos = 0;
403	304995501	} else if (*p == '\t') {
404	6811400	pos &= ~7;
405	6811400	pos += 8;
406	6811400	} else
407	284079301	pos++;
408
409	304995501	}
410	6251440	VSB_printf(vsb, " [%3u] = { VPI_REF_MAGIC, %u, %8tu, %4u, %3u, ",
411	3125720	t->cnt, sp->idx, t->b - sp->b, lin, pos + 1);
412	3125720	if (t->tok == CSRC)
413	280	VSB_cat(vsb, " \"C{\"},\n");
414		else
415	3125440	VSB_printf(vsb, " \"%.*s\" },\n", PF(t));
416	3125720	}
417	47760	VSB_cat(vsb, "};\n\n");
418	47760	}
419
420		/*--------------------------------------------------------------------
421		* Init/Fini/Event
422		*
423		* We call DISCARD and COLD events in the opposite order of LOAD and
424		* WARM.
425		*/
426
427		static void
428	47760	EmitInitFini(const struct vcc *tl)
429		{
430	47760	struct inifin p, q = NULL;
431	47760	unsigned has_event = 0;
432		struct symbol *sy;
433
434	47760	Fh(tl, 0, "\n");
435	47760	Fh(tl, 0, "static unsigned vgc_inistep;\n");
436	47760	Fh(tl, 0, "static unsigned vgc_warmupstep;\n");
437
438		/*
439		* LOAD
440		*/
441	47760	Fc(tl, 0, "\nstatic int\nVGC_Load(VRT_CTX)\n{\n\n");
442	47760	Fc(tl, 0, "\tvgc_inistep = 0;\n");
443	47760	Fc(tl, 0, "\tsize_t ndirector = %dUL;\n", tl->ndirector);
444	47760	Fc(tl, 0, "\n");
445	382800	VTAILQ_FOREACH(p, &tl->inifin, list) {
446	335040	AZ(VSB_finish(p->ini));
447	335040	assert(p->n > 0);
448	335040	if (VSB_len(p->ini))
449	326400	Fc(tl, 0, "\t/* %u */\n%s\n", p->n, VSB_data(p->ini));
450	335040	if (p->ignore_errors == 0) {
451	287280	Fc(tl, 0, "\tif (ctx->vpi->handling == VCL_RET_FAIL)\n");
452	287280	Fc(tl, 0, "\t\treturn(1);\n");
453	287280	}
454	335040	Fc(tl, 0, "\tvgc_inistep = %u;\n\n", p->n);
455	335040	VSB_destroy(&p->ini);
456
457	335040	AZ(VSB_finish(p->event));
458	335040	if (VSB_len(p->event))
459	4000	has_event = 1;
460	335040	}
461
462		/* Handle failures from vcl_init */
463	47760	Fc(tl, 0, "\n");
464	47760	Fc(tl, 0, "\tif (ctx->vpi->handling != VCL_RET_OK)\n");
465	47760	Fc(tl, 0, "\t\treturn(1);\n");
466	47760	Fc(tl, 0, "\tctx->vpi->handling = 0;\n");
467
468	55360	VTAILQ_FOREACH(sy, &tl->sym_objects, sideways) {
469	7600	Fc(tl, 0, "\tif (!%s) {\n", sy->rname);
470	15200	Fc(tl, 0, "\t\tVRT_fail(ctx, "
471	7600	"\"Object %s not initialized\");\n" , sy->name);
472	7600	Fc(tl, 0, "\t\treturn(1);\n");
473	7600	Fc(tl, 0, "\t}\n");
474	7600	}
475
476	47760	Fc(tl, 0, "\treturn(0);\n");
477	47760	Fc(tl, 0, "}\n");
478
479		/*
480		* DISCARD
481		*/
482	47760	Fc(tl, 0, "\nstatic int\nVGC_Discard(VRT_CTX)\n{\n\n");
483
484	47760	Fc(tl, 0, "\tswitch (vgc_inistep) {\n");
485	382800	VTAILQ_FOREACH_REVERSE(p, &tl->inifin, inifinhead, list) {
486	335040	AZ(VSB_finish(p->fin));
487	335040	if (q)
488	287280	assert(q->n > p->n);
489	335040	q = p;
490	335040	Fc(tl, 0, "\t\tcase %u:\n", p->n);
491	335040	if (VSB_len(p->fin))
492	334960	Fc(tl, 0, "\t%s\n", VSB_data(p->fin));
493	335040	Fc(tl, 0, "\t\t\t/* FALLTHROUGH */\n");
494	335040	VSB_destroy(&p->fin);
495	335040	}
496	47760	Fc(tl, 0, "\t\tdefault:\n\t\t\tbreak;\n");
497	47760	Fc(tl, 0, "\t}\n\n");
498	47760	Fc(tl, 0, "\tswitch (vgc_inistep) {\n");
499	382800	VTAILQ_FOREACH_REVERSE(p, &tl->inifin, inifinhead, list) {