varnish-cache/lib/libvcc/vcc_expr.c
0
/*-
1
 * Copyright (c) 2006 Verdens Gang AS
2
 * Copyright (c) 2006-2011 Varnish Software AS
3
 * All rights reserved.
4
 *
5
 * Author: Poul-Henning Kamp <phk@phk.freebsd.dk>
6
 *
7
 * SPDX-License-Identifier: BSD-2-Clause
8
 *
9
 * Redistribution and use in source and binary forms, with or without
10
 * modification, are permitted provided that the following conditions
11
 * are met:
12
 * 1. Redistributions of source code must retain the above copyright
13
 *    notice, this list of conditions and the following disclaimer.
14
 * 2. Redistributions in binary form must reproduce the above copyright
15
 *    notice, this list of conditions and the following disclaimer in the
16
 *    documentation and/or other materials provided with the distribution.
17
 *
18
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21
 * ARE DISCLAIMED.  IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE
22
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28
 * SUCH DAMAGE.
29
 *
30
 */
31
32
#include "config.h"
33
34
#include <math.h>
35
#include <stdarg.h>
36
#include <stdlib.h>
37
#include <string.h>
38
39
#include "vcc_compile.h"
40
#include "vjsn.h"
41
42
struct expr {
43
        unsigned        magic;
44
#define EXPR_MAGIC      0x38c794ab
45
        vcc_type_t      fmt;
46
        struct vsb      *vsb;
47
        uint8_t         constant;
48
#define EXPR_VAR        (1<<0)
49
#define EXPR_CONST      (1<<1)
50
#define EXPR_STR_CONST  (1<<2)          // Last string elem is "..."
51
        struct token    *t1, *t2;
52
        struct symbol   *instance;
53
        int             nstr;
54
};
55
56
/*--------------------------------------------------------------------
57
 * Facility for carrying expressions around and do text-processing on
58
 * them.
59
 */
60
61
static inline int
62 65252
vcc_isconst(const struct expr *e)
63
{
64 65252
        AN(e->constant);
65 65252
        return (e->constant & EXPR_CONST);
66
}
67
68
static inline int
69 769703
vcc_islit(const struct expr *e)
70
{
71 769703
        AN(e->constant);
72 769703
        return (e->constant & EXPR_STR_CONST);
73
}
74
75
static const char *
76 737
vcc_utype(vcc_type_t t)
77
{
78 737
        if (t == STRINGS || t->stringform)
79 198
                t = STRING;
80 737
        return (t->name);
81
}
82
83
static vcc_type_t
84 6670015
vcc_stringstype(vcc_type_t t)
85
{
86 6670015
        return (t->stringform ? STRINGS : t);
87
}
88
89
90
static void vcc_expr0(struct vcc *tl, struct expr **e, vcc_type_t fmt);
91
static void vcc_expr_cor(struct vcc *tl, struct expr **e, vcc_type_t fmt);
92
static void vcc_expr_typecheck(struct vcc *tl, struct expr **e, vcc_type_t fmt,
93
    struct token *t1);
94
95
static struct expr *
96 7519512
vcc_new_expr(vcc_type_t fmt)
97
{
98
        struct expr *e;
99
100 7519512
        ALLOC_OBJ(e, EXPR_MAGIC);
101 7519512
        AN(e);
102 7519512
        e->vsb = VSB_new_auto();
103 7519512
        e->fmt = fmt;
104 7519512
        e->constant = EXPR_VAR;
105 7519512
        return (e);
106
}
107
108
static struct expr * v_printflike_(2, 3)
109 2014408
vcc_mk_expr(vcc_type_t fmt, const char *str, ...)
110
{
111
        va_list ap;
112
        struct expr *e;
113
114 2014408
        e = vcc_new_expr(fmt);
115 2014408
        va_start(ap, str);
116 2014408
        VSB_vprintf(e->vsb, str, ap);
117 2014408
        va_end(ap);
118 2014408
        AZ(VSB_finish(e->vsb));
119 2014408
        return (e);
120
}
121
122
static void
123 9790011
vcc_delete_expr(struct expr *e)
124
{
125 9790011
        if (e == NULL)
126 2279893
                return;
127 7510118
        CHECK_OBJ(e, EXPR_MAGIC);
128 7510118
        VSB_destroy(&e->vsb);
129 7510118
        FREE_OBJ(e);
130 9790011
}
131
132
/*--------------------------------------------------------------------
133
 * We want to get the indentation right in the emitted C code so we have
134
 * to represent it symbolically until we are ready to render.
135
 *
136
 * Many of the operations have very schematic output syntaxes, so we
137
 * use the same facility to simplify the text-processing of emitting
138
 * a given operation on two subexpressions.
139
 *
140
 * We use '\v' as the magic escape character.
141
 *      \v1  insert subexpression 1
142
 *      \v2  insert subexpression 2
143
 *      \vS  insert subexpression 1(STRINGS) as STRING
144
 *      \vs  insert subexpression 2(STRINGS) as STRING
145
 *      \vT  insert subexpression 1(STRINGS) as STRANDS
146
 *      \vt  insert subexpression 2(STRINGS) as STRANDS
147
 *      \v+  increase indentation
148
 *      \v-  decrease indentation
149
 *      anything else is literal
150
 *
151
 * When editing, we check if any of the subexpressions contain a newline
152
 * and issue it as an indented block of so.
153
 *
154
 * XXX: check line lengths in edit, should pass indent in for this
155
 */
156
157
static void
158 669416
vcc_strands_edit(const struct expr *e1, const struct expr *e2)
159
{
160
161 669416
        assert(e2->fmt == STRANDS || e2->fmt == STRINGS);
162
163 669416
        if (e2->fmt == STRANDS)
164 44
                VSB_cat(e1->vsb, VSB_data(e2->vsb));
165 669372
        else if (e2->nstr == 0)
166 0
                VSB_printf(e1->vsb, "vrt_null_strands");
167 669372
        else if (e2->nstr == 1)
168 603350
                VSB_printf(e1->vsb, "TOSTRAND(%s)", VSB_data(e2->vsb));
169
        else {
170 132044
                VSB_printf(e1->vsb, "TOSTRANDS(%d,\v+\n%s\v-)",
171 66022
                   e2->nstr, VSB_data(e2->vsb));
172
        }
173 669416
}
174
175
static struct expr *
176 4230787
vcc_expr_edit(struct vcc *tl, vcc_type_t fmt, const char *p, struct expr *e1,
177
    struct expr *e2)
178
{
179
        struct expr *e, *e3;
180 4230787
        int nl = 1;
181
182 4230787
        (void) tl;
183
184 4230787
        AN(e1);
185 4230787
        e = vcc_new_expr(fmt);
186 47911886
        while (*p != '\0') {
187 43681099
                if (*p != '\v') {
188 36126123
                        if (*p != '\n' || !nl)
189 36125573
                                VSB_putc(e->vsb, *p);
190 36126123
                        nl = (*p == '\n');
191 36126123
                        p++;
192 36126123
                        continue;
193
                }
194 7554976
                assert(*p == '\v');
195 7554976
                switch (*++p) {
196 669185
                case '+': VSB_cat(e->vsb, "\v+"); nl = 0; break;
197 704055
                case '-': VSB_cat(e->vsb, "\v-"); nl = 0; break;
198
                case 'S':
199
                case 's':
200 232342
                        e3 = (*p == 'S' ? e1 : e2);
201 232342
                        AN(e3);
202 232342
                        assert(e1->fmt == STRINGS);
203 232342
                        if (e3->nstr > 1) {
204 286
                                VSB_cat(e->vsb,
205
                                    "\nVRT_STRANDS_string(ctx,\v+\n");
206 286
                                vcc_strands_edit(e, e3);
207 286
                                VSB_cat(e->vsb,
208
                                    "\v-\n)\n");
209 286
                        } else {
210 232056
                                VSB_cat(e->vsb, VSB_data(e3->vsb));
211
                        }
212 232342
                        break;
213
                case 'T':
214
                case 't':
215 669130
                        e3 = (*p == 'T' ? e1 : e2);
216 669130
                        AN(e3);
217 669130
                        vcc_strands_edit(e, e3);
218 669130
                        break;
219
                case '1':
220 3329656
                        VSB_cat(e->vsb, VSB_data(e1->vsb));
221 3329656
                        break;
222
                case '2':
223 1950608
                        AN(e2);
224 1950608
                        VSB_cat(e->vsb, VSB_data(e2->vsb));
225 1950608
                        break;
226
                default:
227 0
                        WRONG("Illegal edit in VCC expression");
228 0
                }
229 7554976
                p++;
230
        }
231 4230787
        AZ(VSB_finish(e->vsb));
232 4230787
        e->t1 = e1->t1;
233 4230787
        e->t2 = e1->t2;
234 4230787
        if (e2 != NULL)
235 1950949
                e->t2 = e2->t2;
236 4230787
        vcc_delete_expr(e1);
237 4230787
        vcc_delete_expr(e2);
238 4230787
        return (e);
239
}
240
241
/*--------------------------------------------------------------------
242
 * Expand finished expression into C-source code
243
 */
244
245
static void
246 1328360
vcc_expr_fmt(struct vsb *d, int ind, const struct expr *e1)
247
{
248
        char *p;
249
        int i;
250
251 1328360
        if (!e1->fmt->noindent) {
252 12739716
                for (i = 0; i < ind; i++)
253 11475266
                        VSB_putc(d, ' ');
254 1264450
        }
255 1328360
        p = VSB_data(e1->vsb);
256 113299901
        while (*p != '\0') {
257 111971585
                if (*p == '\n') {
258 3189329
                        VSB_putc(d, '\n');
259 3189329
                        if (*++p == '\0')
260 44
                                break;
261 34085073
                        for (i = 0; i < ind; i++)
262 30895788
                                VSB_putc(d, ' ');
263 111971541
                } else if (*p != '\v') {
264 107241706
                        VSB_putc(d, *p++);
265 107241706
                } else {
266 1540550
                        switch (*++p) {
267 770275
                        case '+': ind += INDENT; break;
268 770275
                        case '-': ind -= INDENT; break;
269 0
                        default:  WRONG("Illegal format in VCC expression");
270 0
                        }
271 1540550
                        p++;
272
                }
273
        }
274 1328360
}
275
276
/*--------------------------------------------------------------------
277
 */
278
279
static void
280 957121
vcc_expr_tobool(struct vcc *tl, struct expr **e)
281
{
282
283 957121
        if ((*e)->fmt == BOOL)
284 701129
                return;
285 255992
        if ((*e)->fmt == BACKEND || (*e)->fmt == INT)
286 110
                *e = vcc_expr_edit(tl, BOOL, "(\v1 != 0)", *e, NULL);
287 255882
        else if ((*e)->fmt == DURATION)
288 22
                *e = vcc_expr_edit(tl, BOOL, "(\v1 > 0)", *e, NULL);
289 255860
        else if ((*e)->fmt == STRINGS)
290 255794
                *e = vcc_expr_edit(tl, BOOL, "VRT_Strands2Bool(\vT)", *e, NULL);
291
        /*
292
         * We do not provide automatic folding from REAL to BOOL
293
         * because comparing to zero is seldom an exact science
294
         * and we want to force people to explicitly get it right.
295
         */
296 957121
}
297
298
/*--------------------------------------------------------------------
299
 */
300
301
static void
302 687731
vcc_expr_tostring(struct vcc *tl, struct expr **e)
303
{
304
        const char *p;
305 687731
        uint8_t constant = EXPR_VAR;
306
307 687731
        CHECK_OBJ_NOTNULL(*e, EXPR_MAGIC);
308 687731
        assert((*e)->fmt != STRINGS);
309
310 687731
        p = (*e)->fmt->tostring;
311 687731
        if (p != NULL) {
312 687720
                AN(*p);
313 687720
                *e = vcc_expr_edit(tl, STRINGS, p, *e, NULL);
314 687720
                (*e)->constant = constant;
315 687720
                (*e)->nstr = 1;
316 687720
        } else {
317 22
                VSB_printf(tl->sb,
318
                    "Cannot convert %s to STRING.\n",
319 11
                    vcc_utype((*e)->fmt));
320 11
                vcc_ErrWhere2(tl, (*e)->t1, tl->t);
321
        }
322 687731
}
323
324
/*--------------------------------------------------------------------
325
 */
326
327
void v_matchproto_(sym_expr_t)
328 4741
vcc_Eval_Handle(struct vcc *tl, struct expr **e, struct token *t,
329
    struct symbol *sym, vcc_type_t type)
330
{
331
332 4741
        (void)t;
333 4741
        (void)tl;
334 4741
        AN(sym->rname);
335 4741
        AZ(type->stringform);
336
337 4741
        if (sym->type->tostring == NULL &&
338 0
            sym->type != STRING && type == STRINGS) {
339 0
                *e = vcc_mk_expr(STRINGS, "\"%s\"", sym->name);
340 0
                (*e)->nstr = 1;
341 0
                (*e)->constant |= EXPR_CONST | EXPR_STR_CONST;
342 0
        } else {
343 4741
                *e = vcc_mk_expr(sym->type, "%s", sym->rname);
344 4741
                (*e)->constant = EXPR_VAR;
345 4741
                (*e)->nstr = 1;
346 4741
                if ((*e)->fmt == STRING)
347 0
                        (*e)->fmt = STRINGS;
348
        }
349 4741
}
350
351
void v_matchproto_(sym_expr_t)
352 275
vcc_Eval_Sub(struct vcc *tl, struct expr **e, struct token *t,
353
    struct symbol *sym, vcc_type_t type)
354
{
355
356 275
        (void)t;
357 275
        (void)tl;
358 275
        AN(sym->rname);
359 275
        AZ(type->stringform);
360
361 275
        assert (sym->type == SUB);
362
363 275
        if (type == SUB) {
364 242
                *e = vcc_mk_expr(sym->type, "%s", sym->rname);
365 242
                (*e)->constant = EXPR_CONST;
366 242
                return;
367
        }
368
369 66
        VSB_printf(tl->sb, "Symbol '%s' can only be used as a %s expression\n",
370 33
            sym->name, sym->type->name);
371 33
        vcc_ErrWhere(tl, tl->t);
372 275
}
373
374
/*--------------------------------------------------------------------
375
 */
376
377
void v_matchproto_(sym_expr_t)
378 1551693
vcc_Eval_Var(struct vcc *tl, struct expr **e, struct token *t,
379
    struct symbol *sym, vcc_type_t type)
380
{
381
382 1551693
        (void)type;
383 1551693
        vcc_AddUses(tl, t, NULL, sym, XREF_READ);
384 1551693
        ERRCHK(tl);
385 1551693
        *e = vcc_mk_expr(sym->type, "%s", sym->rname);
386 1551693
        (*e)->constant = EXPR_VAR;
387 1551693
        (*e)->nstr = 1;
388 1551693
        if ((*e)->fmt == STRING)
389 610720
                (*e)->fmt = STRINGS;
390 1551693
}
391
392
void v_matchproto_(sym_expr_t)
393 121
vcc_Eval_ProtectedHeader(struct vcc *tl, struct expr **e, struct token *t,
394
    struct symbol *sym, vcc_type_t type)
395
{
396
397 121
        AN(sym);
398 121
        AZ(sym->lorev);
399
400 121
        vcc_Header_Fh(tl, sym);
401 121
        sym->eval = vcc_Eval_Var;
402 121
        vcc_Eval_Var(tl, e, t, sym, type);
403 121
}
404
405
/*--------------------------------------------------------------------
406
 */
407
408
static struct expr *
409 1892
vcc_priv_arg(struct vcc *tl, const char *p, struct symbol *sym)
410
{
411
        char buf[64];
412
        struct inifin *ifp;
413 1892
        const char *f = NULL;
414
415 1892
        AN(sym);
416 1892
        AN(sym->vmod_name);
417
418 1892
        if (!strcmp(p, "PRIV_VCL"))
419 165
                return (vcc_mk_expr(VOID, "&vmod_priv_%s", sym->vmod_name));
420
421 1727
        if (!strcmp(p, "PRIV_CALL")) {
422 242
                bprintf(buf, "vmod_priv_%u", tl->unique++);
423 242
                ifp = New_IniFin(tl);
424 242
                Fh(tl, 0, "static struct vmod_priv %s;\n", buf);
425 242
                VSB_printf(ifp->fin, "\tVRT_priv_fini(ctx, &%s);", buf);
426 242
                return (vcc_mk_expr(VOID, "&%s", buf));
427
        }
428
429 1485
        if (!strcmp(p, "PRIV_TASK"))
430 1397
                f = "task";
431 88
        else if (!strcmp(p, "PRIV_TOP")) {
432 88
                f = "top";
433 88
                sym->r_methods &= VCL_MET_TASK_C;
434 88
        } else {
435 0
                WRONG("Wrong PRIV_ type");
436
        }
437 1485
        AN(f);
438
439 1485
        return (vcc_mk_expr(VOID, "VRT_priv_%s(ctx, &VGC_vmod_%s)",
440 1485
            f, sym->vmod_name));
441 1892
}
442
443
struct func_arg {
444
        vcc_type_t              type;
445
        const struct vjsn_val   *enums;
446
        const char              *cname;
447
        const char              *name;
448
        const char              *val;
449
        struct expr             *result;
450
        int                     avail;
451
        int                     optional;
452
        VTAILQ_ENTRY(func_arg)  list;
453
};
454
455
static struct expr *
456 20009
vcc_do_enum(struct vcc *tl, const char *cfunc, int len, const char *ptr)
457
{
458
        const char *r;
459
460 20009
        (void)tl;
461 20009
        r = strchr(cfunc, '.');
462 20009
        AN(r);
463 20009
        return (vcc_mk_expr(VOID, "*%.*s.enum_%.*s",
464 20009
            (int)(r - cfunc), cfunc, len, ptr));
465
}
466
467
static void
468 43989
vcc_do_arg(struct vcc *tl, const char *cfunc, struct func_arg *fa)
469
{
470
        struct expr *e2;
471
        struct vjsn_val *vv;
472
473 43989
        if (fa->type == ENUM) {
474 14707
                ExpectErr(tl, ID);
475 14707
                ERRCHK(tl);
476 50281
                VTAILQ_FOREACH(vv, &fa->enums->children, list)
477 50259
                        if (vcc_IdIs(tl->t, vv->value))
478 14685
                                break;
479 14707
                if (vv == NULL) {
480 22
                        VSB_cat(tl->sb, "Wrong enum value.");
481 22
                        VSB_cat(tl->sb, "  Expected one of:\n");
482 110
                        VTAILQ_FOREACH(vv, &fa->enums->children, list)
483 88
                                VSB_printf(tl->sb, "\t%s\n", vv->value);
484 22
                        vcc_ErrWhere(tl, tl->t);
485 22
                        return;
486
                }
487 14685
                fa->result = vcc_do_enum(tl, cfunc, PF(tl->t));
488 14685
                SkipToken(tl, ID);
489 14685
        } else {
490 29282
                if (fa->type == SUB)
491 242
                        tl->subref++;
492 29282
                vcc_expr0(tl, &e2, fa->type);
493 29282
                ERRCHK(tl);
494 29194
                assert(e2->fmt == fa->type);
495 29194
                fa->result = e2;
496
        }
497 43879
        fa->avail = 1;
498 43989
}
499
500
static void
501 30525
vcc_func(struct vcc *tl, struct expr **e, const void *priv,
502
    const char *extra, struct symbol *sym)
503
{
504
        vcc_type_t rfmt;
505
        const char *cfunc;
506
        struct expr *e1;
507
        struct func_arg *fa, *fa2;
508
        VTAILQ_HEAD(,func_arg) head;
509
        struct token *tf, *t1;
510
        const struct vjsn_val *vv, *vvp;
511
        const char *sa, *extra_sep;
512
        char ssa[64];
513
        int n;
514
515 30525
        CAST_OBJ_NOTNULL(vv, priv, VJSN_VAL_MAGIC);
516 30525
        assert(vjsn_is_array(vv));
517 30525
        vv = VTAILQ_FIRST(&vv->children);
518 30525
        rfmt = VCC_Type(VTAILQ_FIRST(&vv->children)->value);
519 30525
        AN(rfmt);
520 30525
        vv = VTAILQ_NEXT(vv, list);
521 30525
        cfunc = vv->value;
522 30525
        vv = VTAILQ_NEXT(vv, list);
523 30525
        sa = vv->value;
524 30525
        if (*sa == '\0') {
525 25960
                sa = NULL;
526 25960
        }
527 30525
        vv = VTAILQ_NEXT(vv, list);
528 30525
        if (sym->kind == SYM_METHOD) {
529 8514
                if (*e == NULL) {
530 22
                        VSB_cat(tl->sb, "Syntax error.");
531 22
                        tl->err = 1;
532 22
                        return;
533
                }
534 8492
                vcc_NextToken(tl);
535 8492
                AZ(extra);
536 8492
                AN((*e)->instance);
537 8492
                extra = (*e)->instance->rname;
538 8492
        }
539 30503
        tf = VTAILQ_PREV(tl->t, tokenhead, list);
540 30503
        SkipToken(tl, '(');
541 30503
        if (extra == NULL) {
542 19800
                extra = "";
543 19800
                extra_sep = "";
544 19800
        } else {
545 10703
                AN(*extra);
546 10703
                extra_sep = ", ";
547
        }
548 30503
        VTAILQ_INIT(&head);
549 104533
        for (;vv != NULL; vv = VTAILQ_NEXT(vv, list)) {
550 74030
                assert(vjsn_is_array(vv));
551 74030
                fa = calloc(1, sizeof *fa);
552 74030
                AN(fa);
553 74030
                VTAILQ_INSERT_TAIL(&head, fa, list);
554
555 74030
                vvp = VTAILQ_FIRST(&vv->children);
556 74030
                if (!memcmp(vvp->value, "PRIV_", 5)) {
557 1892
                        fa->result = vcc_priv_arg(tl, vvp->value, sym);
558 1892
                        vvp = VTAILQ_NEXT(vvp, list);
559 1892
                        if (vvp != NULL)
560 1892
                                fa->cname = fa->name = vvp->value;
561 1892
                        continue;
562
                }
563 72138
                fa->type = VCC_Type(vvp->value);
564 72138
                AN(fa->type);
565 72138
                vvp = VTAILQ_NEXT(vvp, list);
566 72138
                if (vvp != NULL) {
567 72138
                        fa->name = vvp->value;
568 72138
                        vvp = VTAILQ_NEXT(vvp, list);
569 72138
                        AN(vvp); /* vmod_syntax 2.0 */
570 72138
                        fa->cname = vvp->value;
571 72138
                        vvp = VTAILQ_NEXT(vvp, list);
572 72138
                        if (vvp != NULL) {
573 50435
                                fa->val = vvp->value;
574 50435
                                vvp = VTAILQ_NEXT(vvp, list);
575 50435
                                if (vvp != NULL) {
576 41415
                                        fa->enums = vvp;
577 41415
                                        vvp = VTAILQ_NEXT(vvp, list);
578 41415
                                }
579 50435
                        }
580 72138
                }
581 72138
                if (sa != NULL && vvp != NULL && vjsn_is_true(vvp)) {
582 22759
                        fa->optional = 1;
583 22759
                        vvp = VTAILQ_NEXT(vvp, list);
584 22759
                }
585 72138
                AZ(vvp);
586 72138
        }
587
588 48972
        VTAILQ_FOREACH(fa, &head, list) {
589 43912
                if (tl->t->tok == ')')
590 979
                        break;
591 42933
                if (fa->result != NULL)
592 1353
                        continue;
593 41580
                if (tl->t->tok == ID) {
594 30547
                        t1 = VTAILQ_NEXT(tl->t, list);
595 30547
                        if (t1->tok == '=')
596 8360
                                break;
597 22187
                }
598 33220
                vcc_do_arg(tl, cfunc, fa);
599 33220
                if (tl->err)
600 220
                        VSB_printf(tl->sb, "Expected argument: %s %s\n\n",
601 110
                            fa->type->name,
602 110
                            fa->name ? fa->name : "(unnamed argument)");
603 33220
                ERRCHK(tl);
604 33110
                if (tl->t->tok == ')')
605 15994
                        break;
606 17116
                SkipToken(tl, ',');
607 17116
        }
608 32824
        while (tl->t->tok == ID) {
609 35464
                VTAILQ_FOREACH(fa, &head, list) {
610 35453
                        if (fa->name == NULL)
611 132
                                continue;
612 35321
                        if (vcc_IdIs(tl->t, fa->name))
613 10780
                                break;
614 24541
                }
615 10791
                if (fa == NULL) {
616 22
                        VSB_printf(tl->sb, "Unknown argument '%.*s'\n",
617 11
                            PF(tl->t));
618 11
                        vcc_ErrWhere(tl, tl->t);
619 11
                        return;
620
                }
621 10780
                if (fa->result != NULL) {
622 11
                        AN(fa->name);
623 22
                        VSB_printf(tl->sb, "Argument '%s' already used\n",
624 11
                            fa->name);
625 11
                        vcc_ErrWhere(tl, tl->t);
626 11
                        return;
627
                }
628 10769
                vcc_NextToken(tl);
629 10769
                SkipToken(tl, '=');
630 10769
                vcc_do_arg(tl, cfunc, fa);
631 10769
                ERRCHK(tl);
632 10769
                if (tl->t->tok == ')')
633 8338
                        break;
634 2431
                SkipToken(tl, ',');
635
        }
636
637 30371
        if (sa != NULL)
638 9042
                e1 = vcc_mk_expr(rfmt, "%s(ctx%s%s,\v+\n&(%s)\v+ {\n",
639 4521
                    cfunc, extra_sep, extra, sa);
640
        else
641 51700
                e1 = vcc_mk_expr(rfmt, "%s(ctx%s%s\v+",
642 25850
                    cfunc, extra_sep, extra);
643 30371
        n = 0;
644 103620
        VTAILQ_FOREACH_SAFE(fa, &head, list, fa2) {
645 73249
                n++;
646 73249
                if (fa->optional) {
647 22275
                        AN(fa->cname);
648 22275
                        bprintf(ssa, "\v1.valid_%s = %d,\n",
649
                            fa->cname, fa->avail);
650 22275
                        e1 = vcc_expr_edit(tl, e1->fmt, ssa, e1, NULL);
651 22275
                }
652 73249
                if (fa->result == NULL && fa->type == ENUM && fa->val != NULL)
653 5324
                        fa->result = vcc_do_enum(tl, cfunc, strlen(fa->val), fa->val);
654 73249
                if (fa->result == NULL && fa->val != NULL)
655 6699
                        fa->result = vcc_mk_expr(fa->type, "%s", fa->val);
656 73249
                if (fa->result != NULL && sa != NULL) {
657 9768
                        if (fa->cname)
658 9537
                                bprintf(ssa, "\v1.%s = \v2,\n", fa->cname);
659
                        else
660 231
                                bprintf(ssa, "\v1.arg%d = \v2,\n", n);
661 9768
                        e1 = vcc_expr_edit(tl, e1->fmt, ssa, e1, fa->result);
662 73249
                } else if (fa->result != NULL) {
663 95942
                        e1 = vcc_expr_edit(tl, e1->fmt, "\v1,\n\v2",
664 47971
                            e1, fa->result);
665 63481
                } else if (!fa->optional) {
666 11
                        if (fa->name)
667 22
                                VSB_printf(tl->sb, "Argument '%s' missing\n",
668 11
                                    fa->name);
669
                        else
670 0
                                VSB_printf(tl->sb, "Argument %d missing\n", n);
671 11
                        vcc_ErrWhere(tl, tl->t);
672 11
                }
673 73249
                free(fa);
674 73249
        }
675 30371
        if (sa != NULL) {
676 4521
                *e = vcc_expr_edit(tl, e1->fmt, "\v1\v-\n}\v-\n)", e1, NULL);
677 4521
        } else {
678 25850
                *e = vcc_expr_edit(tl, e1->fmt, "\v1\v-\n)", e1, NULL);
679
        }
680 30371
        SkipToken(tl, ')');
681 30360
        vcc_AddUses(tl, tf, NULL, sym, XREF_READ);
682 30525
}
683
684
685
/*--------------------------------------------------------------------
686
 */
687
688
void
689 2211
vcc_Eval_Func(struct vcc *tl, const struct vjsn_val *spec,
690
    const char *extra, struct symbol *sym)
691
{
692 2211
        struct expr *e = NULL;
693
694 2211
        vcc_func(tl, &e, spec, extra, sym);
695 2211
        if (tl->err)
696 0
                VSB_cat(tl->sb, "While compiling function call:\n");
697 2211
        ERRCHK(tl);
698 2211
        vcc_expr_fmt(tl->fb, tl->indent, e);
699 2211
        VSB_cat(tl->fb, ";\n");
700 2211
        vcc_delete_expr(e);
701 2211
}
702
703
/*--------------------------------------------------------------------
704
 */
705
706
void v_matchproto_(sym_expr_t)
707 22913
vcc_Eval_SymFunc(struct vcc *tl, struct expr **e, struct token *t,
708
    struct symbol *sym, vcc_type_t fmt)
709
{
710
711 22913
        (void)t;
712 22913
        (void)fmt;
713 22913
        assert(sym->kind == SYM_FUNC || sym->kind == SYM_METHOD);
714 22913
        AN(sym->eval_priv);
715
716 22913
        vcc_func(tl, e, sym->eval_priv, sym->extra, sym);
717 22913
        ERRCHK(tl);
718 22803
        if ((*e)->fmt == STRING) {
719 8272
                (*e)->fmt = STRINGS;
720 8272
                (*e)->nstr = 1;
721 8272
        }
722 22913
}
723
724
/*--------------------------------------------------------------------
725
 */
726
727
static void
728 363913
vcc_number(struct vcc *tl, struct expr **e, vcc_type_t fmt, const char *sign)
729
{
730
        VCL_INT vi;
731
        struct expr *e1;
732
        struct token *t;
733
734 363913
        assert(fmt != VOID);
735 363913
        if (fmt == BYTES) {
736 550
                vcc_ByteVal(tl, &vi);
737 550
                ERRCHK(tl);
738 528
                e1 = vcc_mk_expr(BYTES, "%ju", (intmax_t)vi);
739 528
        } else {
740 363363
                t = tl->t;
741 363363
                vcc_NextToken(tl);
742 363363
                if (tl->t->tok == ID) {
743 34782
                        e1 = vcc_mk_expr(DURATION, "%s%.3f * %g",
744 34782
                            sign, t->num, vcc_DurationUnit(tl));
745 34782
                        ERRCHK(tl);
746 363352
                } else if (fmt == REAL || t->tok == FNUM) {
747 1716
                        e1 = vcc_mk_expr(REAL, "%s%.3f", sign, t->num);
748 1716
                } else {
749 326865
                        e1 = vcc_mk_expr(INT, "%s%.0fLL", sign, t->num);
750
                }
751
        }
752 363880
        e1->constant = EXPR_CONST;
753 363880
        *e = e1;
754 363913
}
755
756
/*--------------------------------------------------------------------
757
 * SYNTAX:
758
 *    Expr5:
759
 *      '(' ExprCor ')'
760
 *      symbol
761
 *      CNUM
762
 *      FNUM
763
 *      CSTR
764
 *      CBLOB
765
 */
766
767
static void
768 3277362
vcc_expr5(struct vcc *tl, struct expr **e, vcc_type_t fmt)
769
{
770
        struct expr *e1, *e2;
771
        const char *ip, *sign;
772
        struct token *t, *t1;
773
        struct symbol *sym;
774
775 3277362
        sign = "";
776 3277362
        *e = NULL;
777 3277362
        if (tl->t->tok == '(') {
778 32307
                SkipToken(tl, '(');
779 32307
                vcc_expr_cor(tl, &e2, fmt);
780 32307
                ERRCHK(tl);
781 32307
                SkipToken(tl, ')');
782 32307
                if (e2->fmt == STRINGS)
783 66
                        *e = e2;
784
                else
785 32241
                        *e = vcc_expr_edit(tl, e2->fmt, "(\v1)", e2, NULL);
786 32307
                return;
787
        }
788 3245055
        switch (tl->t->tok) {
789
        case ID:
790 1614866
                t = tl->t;
791 1614866
                t1 = vcc_PeekToken(tl);
792 1614866
                AN(t1);
793 1614866
                sym = VCC_SymbolGet(tl, SYM_MAIN, SYM_NONE,
794
                    SYMTAB_PARTIAL_NOERR, XREF_REF);
795 1614866
                if (sym == NULL && fmt->global_pfx != NULL && t1->tok != '.') {
796 99
                        sym = VCC_SymbolGet(tl, SYM_MAIN, SYM_NONE,
797
                            SYMTAB_CREATE, XREF_REF);
798 99
                        ERRCHK(tl);
799 77
                        AN(sym);
800 77
                        VCC_GlobalSymbol(sym, fmt);
801 77
                }
802 1614844
                ERRCHK(tl);
803 1614844
                if (sym == NULL)
804 132
                        AZ(VCC_SymbolGet(tl, SYM_MAIN, SYM_NONE,
805
                            SYMTAB_PARTIAL, XREF_REF));
806 1614844
                ERRCHK(tl);
807 1614712
                AN(sym);
808 1614712
                if (sym->kind == SYM_INSTANCE) {
809 8514
                        AZ(*e);
810 8514
                        *e = vcc_new_expr(sym->type);
811 8514
                        (*e)->instance = sym;
812 8514
                        return;
813
                }
814 1606198
                if (sym->kind == SYM_FUNC && sym->type == VOID) {
815 11
                        VSB_cat(tl->sb, "Function returns VOID:\n");
816 11
                        vcc_ErrWhere(tl, tl->t);
817 11
                        return;
818
                }
819 1606187
                if (sym->eval != NULL) {
820 1606154
                        AN(sym->eval);
821 1606154
                        AZ(*e);
822 1606154
                        sym->eval(tl, e, t, sym, fmt);
823 1606154
                        if (tl->err) {
824 176
                                VSB_cat(tl->sb,
825
                                    "While compiling function call:\n\n");
826 176
                                vcc_ErrWhere2(tl, t, tl->t);
827 176
                        }
828 1606154
                        ERRCHK(tl);
829
                        /* Unless asked for a HEADER, fold to string here */
830 1605978
                        if (*e && fmt != HEADER && (*e)->fmt == HEADER) {
831 457028
                                vcc_expr_tostring(tl, e);
832 457028
                                ERRCHK(tl);
833 457028
                        }
834 1605978
                        return;
835
                }
836 66
                VSB_printf(tl->sb,
837
                    "Symbol '%.*s' type (%s) cannot be used in expression.\n",
838 33
                    PF(t), sym->kind->name);
839 33
                vcc_ErrWhere(tl, t);
840 33
                if (sym->def_b != NULL) {
841 11
                        VSB_cat(tl->sb, "That symbol was defined here:\n");
842 11
                        vcc_ErrWhere(tl, sym->def_b);
843 11
                }
844 33
                return;
845
        case CSTR:
846 1266111
                assert(fmt != VOID);
847 1266111
                if (fmt == IP) {
848 330
                        if (*tl->t->dec == '/') {
849
                                /*
850
                                 * On some platforms (e.g. FreeBSD),
851
                                 * getaddrinfo(3) may resolve a path to a
852
                                 * sockaddr_un if it happens to exist and
853
                                 * is a socket. So don't let that happen.
854
                                 */
855 11
                                VSB_cat(tl->sb,
856
                                    "Cannot convert to an IP address: ");
857 11
                                vcc_ErrToken(tl, tl->t);
858 11
                                vcc_ErrWhere(tl, tl->t);
859 11
                                return;
860
                        }
861 638
                        Resolve_Sockaddr(tl, tl->t->dec, "80",
862
                            &ip, NULL, &ip, NULL, NULL, 1,
863 319
                            tl->t, "IP constant");
864 319
                        ERRCHK(tl);
865 286
                        e1 = vcc_mk_expr(IP, "%s", ip);
866 286
                        ERRCHK(tl);
867 1266067
                } else if (fmt == REGEX) {
868 96976
                        e1 = vcc_new_expr(REGEX);
869 96976
                        vcc_regexp(tl, e1->vsb);
870 96976
                        AZ(VSB_finish(e1->vsb));
871 96976
                } else {
872 1168805
                        e1 = vcc_new_expr(STRINGS);
873 1168805
                        EncToken(e1->vsb, tl->t);
874 1168805
                        AZ(VSB_finish(e1->vsb));
875 1168805
                        e1->constant |= EXPR_STR_CONST;
876 1168805
                        e1->nstr = 1;
877
                }
878 1266067
                e1->t1 = tl->t;
879 1266067
                e1->constant |= EXPR_CONST;
880 1266067
                vcc_NextToken(tl);
881 1266067
                *e = e1;
882 1266067
                return;
883
        case '-':
884 682
                if (fmt != INT &&
885 176
                    fmt != REAL &&
886 44
                    fmt != DURATION &&
887 11
                    fmt != STRINGS)
888 0
                        break;
889 671
                vcc_NextToken(tl);
890 671
                if (tl->t->tok != FNUM && tl->t->tok != CNUM) {
891 99
                        vcc_expr_cor(tl, &e1, fmt);
892 99
                        ERRCHK(tl);
893 99
                        *e = vcc_expr_edit(tl, e1->fmt, "-(\v1)", e1, NULL);
894 99
                        return;
895
                }
896 572
                sign = "-";
897
                /* FALLTHROUGH */
898
        case FNUM:
899
        case CNUM:
900 363913
                vcc_number(tl, e, fmt, sign);
901 363913
                return;
902
        case CBLOB:
903 22
                e1 = vcc_new_expr(BLOB);
904 22
                VSB_printf(e1->vsb, "%s", tl->t->dec);
905 22
                AZ(VSB_finish(e1->vsb));
906 22
                e1->constant |= EXPR_STR_CONST;
907 22
                e1->t1 = tl->t;
908 22
                vcc_NextToken(tl);
909 22
                *e = e1;
910 22
                return;
911
        default:
912 44
                break;
913
        }
914 44
        VSB_cat(tl->sb, "Unknown token ");
915 44
        vcc_ErrToken(tl, tl->t);
916 44
        VSB_printf(tl->sb, " when looking for %s\n\n", vcc_utype(fmt));
917 44
        vcc_ErrWhere(tl, tl->t);
918 3277362
}
919
920
/*--------------------------------------------------------------------
921
 * SYNTAX:
922
 *    Expr4:
923
 *      Expr5 [ '.' (type_attribute | type_method()) ]*
924
 */
925
926
void
927 32054
vcc_Eval_TypeMethod(struct vcc *tl, struct expr **e, struct token *t,
928
    struct symbol *sym, vcc_type_t fmt)
929
{
930
        const char *impl;
931
932 32054
        (void)t;
933 32054
        impl = VCC_Type_EvalMethod(tl, sym);
934 32054
        ERRCHK(tl);
935 32054
        AN(impl);
936 32054
        *e = vcc_expr_edit(tl, fmt, impl, *e, NULL);
937 32054
}
938
939
static void
940 3277362
vcc_expr4(struct vcc *tl, struct expr **e, vcc_type_t fmt)
941
{
942
        struct symbol *sym;
943
944 3277362
        *e = NULL;
945 3277362
        vcc_expr5(tl, e, fmt);
946 3277362
        ERRCHK(tl);
947 3276834
        AN(*e);
948 3317380
        while (tl->t->tok == '.') {
949 40579
                vcc_NextToken(tl);
950 40579
                ExpectErr(tl, ID);
951
952 40568
                sym = VCC_TypeSymbol(tl, SYM_METHOD, (*e)->fmt);
953 40568
                if (sym == NULL) {
954 22
                        VSB_cat(tl->sb, "Unknown property ");
955 22
                        vcc_ErrToken(tl, tl->t);
956 22
                        VSB_printf(tl->sb, " for type %s\n", (*e)->fmt->name);
957 22
                        vcc_ErrWhere(tl, tl->t);
958 22
                        return;
959
                }
960
961 40546
                AN(sym->eval);
962 40546
                sym->eval(tl, e, tl->t, sym, sym->type);
963 40546
                ERRCHK(tl);
964 40546
                if ((*e)->fmt == STRING) {
965 31933
                        (*e)->fmt = STRINGS;
966 31933
                        (*e)->nstr = 1;
967 31933
                }
968
        }
969 3277362
}
970
971
/*--------------------------------------------------------------------
972
 * SYNTAX:
973
 *    ExprMul:
974
 *      Expr4 { {'*'|'/'|'%'} Expr4 } *
975
 */
976
977
static void
978 3176283
vcc_expr_mul(struct vcc *tl, struct expr **e, vcc_type_t fmt)
979
{
980
        struct expr *e2;
981
        vcc_type_t f2;
982
        struct token *tk;
983
        char buf[24];
984
985 3176283
        *e = NULL;
986 3176283
        vcc_expr4(tl, e, fmt);
987 3176283
        ERRCHK(tl);
988 3175799
        AN(*e);
989
990 3176206
        while (tl->t->tok == '*' || tl->t->tok == '/' || tl->t->tok == '%') {
991 451
                if (tl->t->tok == '%' && ((*e)->fmt != INT)) {
992 11
                        VSB_cat(tl->sb, "Operator % only possible on INT.\n");
993 11
                        vcc_ErrWhere(tl, tl->t);
994 11
                        return;
995
                }
996 440
                f2 = (*e)->fmt->multype;
997 440
                if (f2 == NULL) {
998 22
                        VSB_printf(tl->sb,
999
                            "Operator %.*s not possible on type %s.\n",
1000 11
                            PF(tl->t), vcc_utype((*e)->fmt));
1001 11
                        vcc_ErrWhere(tl, tl->t);
1002 11
                        return;
1003
                }
1004 429
                tk = tl->t;
1005 429
                vcc_NextToken(tl);
1006 429
                vcc_expr4(tl, &e2, f2);
1007 429
                ERRCHK(tl);
1008 429
                if (e2->fmt != INT && e2->fmt != f2) {
1009 44
                        VSB_printf(tl->sb, "%s %.*s %s not possible.\n",
1010 22
                            vcc_utype((*e)->fmt), PF(tk), vcc_utype(e2->fmt));
1011 22
                        vcc_ErrWhere(tl, tk);
1012 22
                        return;
1013
                }
1014 407
                bprintf(buf, "(\v1%c\v2)", tk->tok);
1015 407
                *e = vcc_expr_edit(tl, (*e)->fmt, buf, *e, e2);
1016
        }
1017 3176283
}
1018
1019
/*--------------------------------------------------------------------
1020
 * SYNTAX:
1021
 *    ExprAdd:
1022
 *      ExprMul { {'+'|'-'} ExprMul } *
1023
 */
1024
1025
static const struct adds {
1026
        unsigned        op;
1027
        vcc_type_t      a;
1028
        vcc_type_t      b;
1029
        vcc_type_t      fmt;
1030
} vcc_adds[] = {
1031
        { '+', BYTES,           BYTES,          BYTES },
1032
        { '-', BYTES,           BYTES,          BYTES },
1033
        { '+', DURATION,        DURATION,       DURATION },
1034
        { '-', DURATION,        DURATION,       DURATION },
1035
        { '+', INT,             INT,            INT },
1036
        { '-', INT,             INT,            INT },
1037
        { '+', INT,             REAL,           REAL },
1038
        { '-', INT,             REAL,           REAL },
1039
        { '+', REAL,            INT,            REAL },
1040
        { '-', REAL,            INT,            REAL },
1041
        { '+', REAL,            REAL,           REAL },
1042
        { '-', REAL,            REAL,           REAL },
1043
        { '-', TIME,            TIME,           DURATION },
1044
        { '+', TIME,            DURATION,       TIME },
1045
        { '-', TIME,            DURATION,       TIME },
1046
1047
        { EOI, VOID,            VOID,           VOID }
1048
};
1049
1050
static void
1051 2406294
vcc_expr_add(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1052
{
1053
        const struct adds *ap;
1054
        struct expr  *e2;
1055
        struct token *tk;
1056
        int lit, n;
1057
1058 2406294
        *e = NULL;
1059 2406294
        vcc_expr_mul(tl, e, fmt);
1060 2406294
        ERRCHK(tl);
1061
1062 3175634
        while (tl->t->tok == '+' || tl->t->tok == '-') {
1063 769989
                tk = tl->t;
1064 12311530
                for (ap = vcc_adds; ap->op != EOI; ap++)
1065 11542366
                        if (tk->tok == ap->op && (*e)->fmt == ap->a)
1066 825
                                break;
1067 769989
                vcc_NextToken(tl);
1068 769989
                if (ap->op == EOI && fmt == STRINGS)
1069 5566
                        vcc_expr_mul(tl, &e2, STRINGS);
1070
                else
1071 764423
                        vcc_expr_mul(tl, &e2, (*e)->fmt);
1072 769989
                ERRCHK(tl);
1073
1074 12312872
                for (ap = vcc_adds; ap->op != EOI; ap++)
1075 11543598
                        if (tk->tok == ap->op && (*e)->fmt == ap->a &&
1076 11543598
                            e2->fmt == ap->b)
1077 715
                                break;
1078
1079 769989
                if (ap->op == '+') {
1080 495
                        *e = vcc_expr_edit(tl, ap->fmt, "(\v1 + \v2)", *e, e2);
1081 769989
                } else if (ap->op == '-') {
1082 220
                        *e = vcc_expr_edit(tl, ap->fmt, "(\v1 - \v2)", *e, e2);
1083 769670
                } else if (tk->tok == '+' &&
1084 769263
                    ((*e)->fmt == STRINGS || fmt == STRINGS)) {
1085 769153
                        if ((*e)->fmt != STRINGS)
1086 66
                                vcc_expr_tostring(tl, e);
1087 769153
                        if (e2->fmt != STRINGS)
1088 191499
                                vcc_expr_tostring(tl, &e2);
1089 769153
                        if (vcc_islit(*e) && vcc_isconst(e2)) {
1090 550
                                lit = vcc_islit(e2);
1091 1100
                                *e = vcc_expr_edit(tl, STRINGS,
1092 550
                                    "\v1\n\v2", *e, e2);
1093 550
                                (*e)->constant = EXPR_CONST;
1094 550
                                (*e)->nstr = 1;
1095 550
                                if (lit)
1096 550
                                        (*e)->constant |= EXPR_STR_CONST;
1097 550
                        } else {
1098 768603
                                n = (*e)->nstr + e2->nstr;
1099 1537206
                                *e = vcc_expr_edit(tl, STRINGS,
1100 768603
                                    "\v1,\n\v2", *e, e2);
1101 768603
                                (*e)->constant = EXPR_VAR;
1102 768603
                                (*e)->nstr = n;
1103
                        }
1104 769153
                } else {
1105 242
                        VSB_printf(tl->sb, "%s %.*s %s not possible.\n",
1106 121
                            vcc_utype((*e)->fmt), PF(tk), vcc_utype(e2->fmt));
1107 121
                        vcc_ErrWhere2(tl, tk, tl->t);
1108 121
                        return;
1109
                }
1110
        }
1111 2406294
}
1112
1113
/*--------------------------------------------------------------------
1114
 * SYNTAX:
1115
 *    ExprCmp:
1116
 *      ExprAdd
1117
 *      ExprAdd Relation ExprAdd
1118
 *      ExprAdd(STRING) '~' CString
1119
 *      ExprAdd(STRING) '!~' CString
1120
 *      ExprAdd(IP) '==' ExprAdd(IP)
1121
 *      ExprAdd(IP) '!=' ExprAdd(IP)
1122
 *      ExprAdd(IP) '~' ACL
1123
 *      ExprAdd(IP) '!~' ACL
1124
 */
1125
1126
struct cmps;
1127
1128
typedef void cmp_f(struct vcc *, struct expr **, const struct cmps *);
1129
1130
struct cmps {
1131
        vcc_type_t              fmt;
1132
        unsigned                token;
1133
        cmp_f                   *func;
1134
        const char              *emit;
1135
};
1136
1137
static void v_matchproto_(cmp_f)
1138 160765
cmp_simple(struct vcc *tl, struct expr **e, const struct cmps *cp)
1139
{
1140
        struct expr *e2;
1141
        struct token *tk;
1142
1143 160765
        tk = tl->t;
1144 160765
        vcc_NextToken(tl);
1145 160765
        vcc_expr_add(tl, &e2, (*e)->fmt);
1146 160765
        ERRCHK(tl);
1147
1148 160732
        if (e2->fmt != (*e)->fmt) {
1149 44
                VSB_printf(tl->sb,
1150
                    "Comparison of different types: %s '%.*s' %s\n",
1151 22
                    vcc_utype((*e)->fmt), PF(tk), vcc_utype(e2->fmt));
1152 22
                vcc_ErrWhere(tl, tk);
1153 22
        } else
1154 160710
                *e = vcc_expr_edit(tl, BOOL, cp->emit, *e, e2);
1155 160765
}
1156
1157
static void v_matchproto_(cmp_f)
1158 96470
cmp_regexp(struct vcc *tl, struct expr **e, const struct cmps *cp)
1159
{
1160
        struct token *t1;
1161
        struct expr *e2;
1162
        char buf[128];
1163
1164 96470
        *e = vcc_expr_edit(tl, STRING, "\vS", *e, NULL);
1165 96470
        vcc_NextToken(tl);
1166 96470
        t1 = tl->t;
1167 96470
        vcc_expr4(tl, &e2, REGEX);
1168 96470
        ERRCHK(tl);
1169 96426
        vcc_expr_typecheck(tl, &e2, REGEX, t1);
1170 96426
        ERRCHK(tl);
1171 96415
        bprintf(buf, "%sVRT_re_match(ctx, \v1, \v2)", cp->emit);
1172 96415
        *e = vcc_expr_edit(tl, BOOL, buf, *e, e2);
1173 96470
}
1174
1175
static void v_matchproto_(cmp_f)
1176 517
cmp_acl(struct vcc *tl, struct expr **e, const struct cmps *cp)
1177
{
1178
        struct token *t1;
1179
        struct expr *e2;
1180
        char buf[256];
1181
1182 517
        vcc_NextToken(tl);
1183 517
        t1 = tl->t;
1184 517
        vcc_expr4(tl, &e2, ACL);
1185 517
        ERRCHK(tl);
1186 484
        vcc_expr_typecheck(tl, &e2, ACL, t1);
1187 484
        ERRCHK(tl);
1188 451
        bprintf(buf, "%sVRT_acl_match(ctx, \v1, \v2)", cp->emit);
1189 451
        *e = vcc_expr_edit(tl, BOOL, buf, e2, *e);
1190 517
}
1191
1192
static void v_matchproto_(cmp_f)
1193 418858
cmp_string(struct vcc *tl, struct expr **e, const struct cmps *cp)
1194
{
1195
        struct expr *e2;
1196
        struct token *tk;
1197
        char buf[128];
1198
1199 418858
        tk = tl->t;
1200 418858
        vcc_NextToken(tl);
1201 418858
        vcc_expr_add(tl, &e2, STRINGS);
1202 418858
        ERRCHK(tl);
1203 418858
        if (vcc_stringstype(e2->fmt) != STRINGS) {
1204 66
                VSB_printf(tl->sb,
1205
                    "Comparison of different types: %s '%.*s' %s\n",
1206 33
                    vcc_utype((*e)->fmt), PF(tk), vcc_utype(e2->fmt));
1207 33
                vcc_ErrWhere(tl, tk);
1208 418858
        } else if ((*e)->nstr == 1 && e2->nstr == 1) {
1209 418484
                bprintf(buf, "(%s VRT_strcmp(\v1, \v2))", cp->emit);
1210 418484
                *e = vcc_expr_edit(tl, BOOL, buf, *e, e2);
1211 418484
        } else {
1212 341
                bprintf(buf, "(%s VRT_CompareStrands(\vT, \vt))", cp->emit);
1213 341
                *e = vcc_expr_edit(tl, BOOL, buf, *e, e2);
1214
        }
1215 418858
}
1216
1217
#define PTR_REL(typ)                                                            \
1218
        {typ,           T_EQ,           cmp_simple, "!VPI_PtrCmp(\v1, \v2)" },  \
1219
        {typ,           T_NEQ,          cmp_simple, "VPI_PtrCmp(\v1, \v2)" }
1220
1221
#define IDENT_REL(typ)                                                  \
1222
        {typ,           T_EQ,           cmp_simple, "(\v1 == \v2)" },   \
1223
        {typ,           T_NEQ,          cmp_simple, "(\v1 != \v2)" }
1224
1225
#define NUM_REL(typ)                                                    \
1226
        IDENT_REL(typ),                                                 \
1227
        {typ,           T_LEQ,          cmp_simple, "(\v1 <= \v2)" },   \
1228
        {typ,           T_GEQ,          cmp_simple, "(\v1 >= \v2)" },   \
1229
        {typ,           '<',            cmp_simple, "(\v1 < \v2)" },    \
1230
        {typ,           '>',            cmp_simple, "(\v1 > \v2)" }
1231
1232
static const struct cmps vcc_cmps[] = {
1233
        NUM_REL(INT),
1234
        NUM_REL(DURATION),
1235
        NUM_REL(BYTES),
1236
        NUM_REL(REAL),
1237
        NUM_REL(TIME),
1238
        PTR_REL(BACKEND),
1239
        PTR_REL(ACL),
1240
        PTR_REL(PROBE),
1241
        IDENT_REL(STEVEDORE),
1242
        IDENT_REL(SUB),
1243
        IDENT_REL(INSTANCE),
1244
1245
        {BOOL,          T_EQ,           cmp_simple, "((!(\v1)) == (!(\v2)))" },
1246
        {BOOL,          T_NEQ,          cmp_simple, "((!(\v1)) != (!(\v2)))" },
1247
        {IP,            T_EQ,           cmp_simple, "!VRT_ipcmp(ctx, \v1, \v2)" },
1248
        {IP,            T_NEQ,          cmp_simple, "VRT_ipcmp(ctx, \v1, \v2)" },
1249
1250
        {IP,            '~',            cmp_acl, "" },
1251
        {IP,            T_NOMATCH,      cmp_acl, "!" },
1252
1253
        {STRINGS,       T_EQ,           cmp_string, "0 =="},
1254
        {STRINGS,       T_NEQ,          cmp_string, "0 !="},
1255
        {STRINGS,       '<',            cmp_string, "0 > "},
1256
        {STRINGS,       '>',            cmp_string, "0 < "},
1257
        {STRINGS,       T_LEQ,          cmp_string, "0 >="},
1258
        {STRINGS,       T_GEQ,          cmp_string, "0 <="},
1259
1260
        {STRINGS,       '~',            cmp_regexp, "" },
1261
        {STRINGS,       T_NOMATCH,      cmp_regexp, "!" },
1262
1263
        {VOID,          0,              NULL, NULL}
1264
};
1265
1266
#undef IDENT_REL
1267
#undef NUM_REL
1268
1269
static void
1270 1826671
vcc_expr_cmp(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1271
{
1272
        const struct cmps *cp;
1273
        struct token *tk;
1274
1275 1826671
        *e = NULL;
1276 1826671
        vcc_expr_add(tl, e, fmt);
1277 1826671
        ERRCHK(tl);
1278 1826055
        tk = tl->t;
1279
1280 92216047
        for (cp = vcc_cmps; cp->fmt != VOID; cp++) {
1281 91066602
                if (tl->t->tok != cp->token)
1282 84815445
                        continue;
1283 6251157
                if (vcc_stringstype((*e)->fmt) != cp->fmt)
1284 5574547
                        continue;
1285 676610
                AN(cp->func);
1286 676610
                cp->func(tl, e, cp);
1287 676610
                return;
1288
        }
1289
1290 1149445
        switch (tk->tok) {
1291
        case T_EQ:
1292
        case T_NEQ:
1293
        case '<':
1294
        case T_LEQ:
1295
        case '>':
1296
        case T_GEQ:
1297
        case '~':
1298
        case T_NOMATCH:
1299 44
                VSB_printf(tl->sb, "Operator %.*s not possible on %s\n",
1300 22
                    PF(tl->t), vcc_utype((*e)->fmt));
1301 22
                vcc_ErrWhere(tl, tl->t);
1302 22
                return;
1303
        default:
1304 1149423
                break;
1305
        }
1306 1826671
}
1307
1308
/*--------------------------------------------------------------------
1309
 * SYNTAX:
1310
 *    ExprNot:
1311
 *      '!' ExprCmp
1312
 */
1313
1314
static void
1315 1826671
vcc_expr_not(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1316
{
1317
        struct token *tk;
1318
1319 1826671
        *e = NULL;
1320 1826671
        tk = tl->t;
1321 1826671
        if (tl->t->tok == '!')
1322 159973
                vcc_NextToken(tl);
1323 1826671
        vcc_expr_cmp(tl, e, fmt);
1324 1826671
        ERRCHK(tl);
1325 1825824
        if (tk->tok != '!')
1326 1665862
                return;
1327 159962
        vcc_expr_tobool(tl, e);
1328 159962
        ERRCHK(tl);
1329 159962
        if ((*e)->fmt != BOOL) {
1330 11
                VSB_cat(tl->sb, "'!' must be followed by BOOL, found ");
1331 11
                VSB_printf(tl->sb, "%s.\n", vcc_utype((*e)->fmt));
1332 11
                vcc_ErrWhere2(tl, tk, tl->t);
1333 11
        } else {
1334 159951
                *e = vcc_expr_edit(tl, BOOL, "!(\v1)", *e, NULL);
1335
        }
1336 1826671
}
1337
1338
/*--------------------------------------------------------------------
1339
 * CAND and COR are identical save for a few details, but they are
1340
 * stacked so handling them in the same function is not simpler.
1341
 * Instead have them both call this helper function to do everything.
1342
 */
1343
1344
typedef void upfunc(struct vcc *tl, struct expr **e, vcc_type_t fmt);
1345
1346
static void
1347 2793835
vcc_expr_bin_bool(struct vcc *tl, struct expr **e, vcc_type_t fmt,
1348
    unsigned ourtok, upfunc *up, const char *tokstr)
1349
{
1350
        struct expr *e2;
1351
        struct token *tk;
1352
        char buf[32];
1353
1354 2793835
        *e = NULL;
1355 2793835
        tk = tl->t;
1356 2793835
        up(tl, e, fmt);
1357 2793835
        ERRCHK(tl);
1358 2792097
        if (tl->t->tok != ourtok)
1359 2569017
                return;
1360 223080
        vcc_expr_tobool(tl, e);
1361 223080
        ERRCHK(tl);
1362 223080
        if ((*e)->fmt != BOOL) {
1363 44
                VSB_printf(tl->sb,
1364
                    "'%s' must be preceded by BOOL,"
1365 22
                    " found %s.\n", tokstr, vcc_utype((*e)->fmt));
1366 22
                vcc_ErrWhere2(tl, tk, tl->t);
1367 22
                return;
1368
        }
1369 223058
        *e = vcc_expr_edit(tl, BOOL, "(\v+\n\v1", *e, NULL);
1370 668778
        while (tl->t->tok == ourtok) {
1371 445742
                vcc_NextToken(tl);
1372 445742
                tk = tl->t;
1373 445742
                up(tl, &e2, fmt);
1374 445742
                ERRCHK(tl);
1375 445742
                vcc_expr_tobool(tl, &e2);
1376 445742
                ERRCHK(tl);
1377 445742
                if (e2->fmt != BOOL) {
1378 44
                        VSB_printf(tl->sb,
1379
                            "'%s' must be followed by BOOL,"
1380 22
                            " found %s.\n", tokstr, vcc_utype(e2->fmt));
1381 22
                        vcc_ErrWhere2(tl, tk, tl->t);
1382 22
                        vcc_delete_expr(e2);
1383 22
                        return;
1384
                }
1385 445720
                bprintf(buf, "\v1\v-\n%s\v+\n\v2", tokstr);
1386 445720
                *e = vcc_expr_edit(tl, BOOL, buf, *e, e2);
1387
        }
1388 223036
        *e = vcc_expr_edit(tl, BOOL, "\v1\v-\n)", *e, NULL);
1389 2793835
}
1390
1391
/*--------------------------------------------------------------------
1392
 * SYNTAX:
1393
 *    ExprCand:
1394
 *      ExprNot { '&&' ExprNot } *
1395
 */
1396
1397
static void
1398 1412906
vcc_expr_cand(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1399
{
1400
1401 1412906
        vcc_expr_bin_bool(tl, e, fmt, T_CAND, vcc_expr_not, "&&");
1402 1412906
}
1403
1404
/*--------------------------------------------------------------------
1405
 * SYNTAX:
1406
 *    ExprCOR:
1407
 *      ExprCand { '||' ExprCand } *
1408
 */
1409
1410
static void
1411 1380929
vcc_expr_cor(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1412
{
1413
1414 1380929
        vcc_expr_bin_bool(tl, e, fmt, T_COR, vcc_expr_cand, "||");
1415 1380929
}
1416
1417
/*--------------------------------------------------------------------
1418
 * This function is the entry-point for getting an expression with
1419
 * a particular type, ready for inclusion in the VGC.
1420
 */
1421
1422
static void
1423 1348523
vcc_expr0(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1424
{
1425
        struct token *t1;
1426
1427 1348523
        assert(fmt != VOID);
1428 1348523
        assert(fmt != STRINGS);
1429 1348523
        *e = NULL;
1430 1348523
        t1 = tl->t;
1431 1348523
        if (fmt->stringform)
1432 453090
                vcc_expr_cor(tl, e, STRINGS);
1433
        else
1434 895433
                vcc_expr_cor(tl, e, fmt);
1435 1348523
        ERRCHK(tl);
1436
1437 1347621
        if ((*e)->fmt == fmt)
1438 702603
                return;
1439
1440 645018
        if ((*e)->fmt != STRINGS && fmt->stringform)
1441 39127
                vcc_expr_tostring(tl, e);
1442
1443 645018
        if ((*e)->fmt->stringform) {
1444 0
                VSB_printf(tl->sb, "Cannot convert type %s(%s) to %s(%s)\n",
1445 0
                    vcc_utype((*e)->fmt), (*e)->fmt->name,
1446 0
                    vcc_utype(fmt), fmt->name);
1447 0
                vcc_ErrWhere2(tl, t1, tl->t);
1448 0
                return;
1449
        }
1450
1451 645018
        if (fmt == BODY && !(*e)->fmt->bodyform)
1452 11
                vcc_expr_tostring(tl, e);
1453
1454 645018
        if (fmt == BODY && (*e)->fmt->bodyform) {
1455 63910
                if ((*e)->fmt == STRINGS)
1456 63866
                        *e = vcc_expr_edit(tl, BODY, "STRING, 0, \vT", *e, NULL);
1457 44
                else if ((*e)->fmt == BLOB)
1458 44
                        *e = vcc_expr_edit(tl, BODY, "BLOB, 0, \v1", *e, NULL);
1459
                else
1460 0
                        WRONG("Unhandled bodyform");
1461 63910
        }
1462
1463 645018
        if ((*e)->fmt == STRINGS && fmt->stringform) {
1464 452727
                if (fmt == STRING)
1465 135872
                        *e = vcc_expr_edit(tl, STRING, "\vS", *e, NULL);
1466 316855
                else if (fmt == STRANDS)
1467 316855
                        *e = vcc_expr_edit(tl, STRANDS, "\vT", (*e), NULL);
1468
                else
1469 0
                        WRONG("Unhandled stringform");
1470 452727
        }
1471
1472 645018
        if (fmt == BOOL) {
1473 128337
                vcc_expr_tobool(tl, e);
1474 128337
                ERRCHK(tl);
1475 128337
        }
1476
1477 645018
        vcc_expr_typecheck(tl, e, fmt, t1);
1478 1348523
}
1479
1480
static void
1481 741928
vcc_expr_typecheck(struct vcc *tl, struct expr **e, vcc_type_t fmt,
1482
    struct token *t1)
1483
{
1484
1485 741928
        assert(fmt != VOID);
1486 741928
        assert(fmt != STRINGS);
1487
1488 741928
        if (fmt != (*e)->fmt)  {
1489 198
                VSB_printf(tl->sb, "Expression has type %s, expected %s\n",
1490 99
                    vcc_utype((*e)->fmt), vcc_utype(fmt));
1491 99
                vcc_ErrWhere2(tl, t1, tl->t);
1492 99
        }
1493 741928
}
1494
1495
/*--------------------------------------------------------------------
1496
 * This function parses and emits the C-code to evaluate an expression
1497
 *
1498
 * We know up front what kind of type we want the expression to be,
1499
 * and this function is the backstop if that doesn't succeed.
1500
 */
1501
1502
void
1503 1317998
vcc_Expr(struct vcc *tl, vcc_type_t fmt)
1504
{
1505 1317998
        struct expr *e = NULL;
1506
1507 1317998
        assert(fmt != VOID);
1508 1317998
        assert(fmt != STRINGS);
1509 1317998
        vcc_expr0(tl, &e, fmt);
1510 1317998
        ERRCHK(tl);
1511 1317140
        assert(e->fmt == fmt);
1512
1513 1317140
        vcc_expr_fmt(tl->fb, tl->indent, e);
1514 1317140
        VSB_cat(tl->fb, "\n");
1515 1317140
        vcc_delete_expr(e);
1516 1317998
}
1517
1518
/*--------------------------------------------------------------------
1519
 */
1520
1521
void v_matchproto_(sym_act_f)
1522 5401
vcc_Act_Call(struct vcc *tl, struct token *t, struct symbol *sym)
1523
{
1524
1525
        struct expr *e;
1526
1527 5401
        e = NULL;
1528 5401
        vcc_func(tl, &e, sym->eval_priv, sym->extra, sym);
1529 5401
        if (!tl->err) {
1530 5346
                vcc_expr_fmt(tl->fb, tl->indent, e);
1531 5346
                SkipToken(tl, ';');
1532 5346
                VSB_cat(tl->fb, ";\n");
1533 5401
        } else if (t != tl->t) {
1534 55
                VSB_cat(tl->sb, "While compiling function call:\n\n");
1535 55
                vcc_ErrWhere2(tl, t, tl->t);
1536 55
        }
1537 5401
        vcc_delete_expr(e);
1538 5401
}
1539
1540
void v_matchproto_(sym_act_f)
1541 3674
vcc_Act_Obj(struct vcc *tl, struct token *t, struct symbol *sym)
1542
{
1543
1544 3674
        struct expr *e = NULL;
1545
1546 3674
        assert(sym->kind == SYM_INSTANCE);
1547 3674
        ExpectErr(tl, '.');
1548 3663
        tl->t = t;
1549 3663
        vcc_expr4(tl, &e, sym->type);
1550 3663
        ERRCHK(tl);
1551 3663
        vcc_expr_fmt(tl->fb, tl->indent, e);
1552 3663
        vcc_delete_expr(e);
1553 3663
        SkipToken(tl, ';');
1554 3663
        VSB_cat(tl->fb, ";\n");
1555 3674
}
1556
1557
/*--------------------------------------------------------------------
1558
 */
1559
1560
static void v_matchproto_(sym_expr_t)
1561 418
vcc_Eval_Regsub(struct vcc *tl, struct expr **e, struct token *t,
1562
    struct symbol *sym, vcc_type_t fmt)
1563
{
1564
        struct expr *e2, *e3;
1565 418
        int all = sym->eval_priv == NULL ? 0 : 1;
1566
        char buf[128];
1567
1568 418
        (void)t;
1569 418
        (void)fmt;
1570 418
        SkipToken(tl, '(');
1571 418
        vcc_expr0(tl, &e2, STRING);
1572 418
        ERRCHK(tl);
1573 418
        SkipToken(tl, ',');
1574 418
        vcc_expr0(tl, &e3, REGEX);
1575 418
        ERRCHK(tl);
1576
1577 407
        bprintf(buf, "VRT_regsub(ctx, %d,\v+\n\v1,\n\v2", all);
1578 407
        *e = vcc_expr_edit(tl, STRING, buf, e2, e3);
1579 407
        SkipToken(tl, ',');
1580 407
        vcc_expr0(tl, &e2, STRING);
1581 407
        ERRCHK(tl);
1582 407
        *e = vcc_expr_edit(tl, STRINGS, "\v1,\n\v2)\v-", *e, e2);
1583 407
        (*e)->nstr = 1;
1584 407
        SkipToken(tl, ')');
1585 418
}
1586
1587
/*--------------------------------------------------------------------
1588
 */
1589
1590
static void v_matchproto_(sym_expr_t)
1591 34562
vcc_Eval_BoolConst(struct vcc *tl, struct expr **e, struct token *t,
1592
    struct symbol *sym, vcc_type_t fmt)
1593
{
1594
1595 34562
        (void)t;
1596 34562
        (void)tl;
1597 34562
        (void)fmt;
1598 34562
        *e = vcc_mk_expr(BOOL, "(0==%d)", sym->eval_priv == NULL ? 1 : 0);
1599 34562
        (*e)->constant = EXPR_CONST;
1600 34562
}
1601
1602
/*--------------------------------------------------------------------
1603
 */
1604
1605
static void v_matchproto_(sym_expr_t)
1606 44
vcc_Eval_Default(struct vcc *tl, struct expr **e, struct token *t,
1607
    struct symbol *sym, vcc_type_t fmt)
1608
{
1609 44
        (void)e;
1610 44
        (void)fmt;
1611 44
        (void)sym;
1612 44
        (void)t;
1613
1614 44
        if (fmt->default_sym == NULL) {
1615 22
                VSB_cat(tl->sb, "Symbol 'default' is a reserved word.\n");
1616 22
                vcc_ErrWhere(tl, t);
1617 22
                return;
1618
        }
1619
1620 22
        *e = vcc_mk_expr(fmt, "%s", fmt->default_sym->rname);
1621 44
}
1622
1623
/*--------------------------------------------------------------------
1624
 */
1625
1626
void
1627 34485
vcc_Expr_Init(struct vcc *tl)
1628
{
1629
        struct symbol *sym;
1630
1631 34485
        sym = VCC_MkSym(tl, "regsub", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1632 34485
        AN(sym);
1633 34485
        sym->type = STRING;
1634 34485
        sym->eval = vcc_Eval_Regsub;
1635 34485
        sym->eval_priv = NULL;
1636
1637 34485
        sym = VCC_MkSym(tl, "regsuball", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1638 34485
        AN(sym);
1639 34485
        sym->type = STRING;
1640 34485
        sym->eval = vcc_Eval_Regsub;
1641 34485
        sym->eval_priv = sym;
1642
1643 34485
        sym = VCC_MkSym(tl, "true", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1644 34485
        AN(sym);
1645 34485
        sym->type = BOOL;
1646 34485
        sym->eval = vcc_Eval_BoolConst;
1647 34485
        sym->eval_priv = sym;
1648
1649 34485
        sym = VCC_MkSym(tl, "false", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1650 34485
        AN(sym);
1651 34485
        sym->type = BOOL;
1652 34485
        sym->eval = vcc_Eval_BoolConst;
1653 34485
        sym->eval_priv = NULL;
1654
1655 34485
        sym = VCC_MkSym(tl, "default", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1656 34485
        AN(sym);
1657 34485
        sym->type = DEFAULT;
1658 34485
        sym->eval = vcc_Eval_Default;
1659 34485
}