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 5908
vcc_isconst(const struct expr *e)
63
{
64 5908
        AN(e->constant);
65 5908
        return (e->constant & EXPR_CONST);
66
}
67
68
static inline int
69 69685
vcc_islit(const struct expr *e)
70
{
71 69685
        AN(e->constant);
72 69685
        return (e->constant & EXPR_STR_CONST);
73
}
74
75
static const char *
76 67
vcc_utype(vcc_type_t t)
77
{
78 67
        if (t == STRINGS || t->stringform)
79 18
                t = STRING;
80 67
        return (t->name);
81
}
82
83
static vcc_type_t
84 603851
vcc_stringstype(vcc_type_t t)
85
{
86 603851
        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 680823
vcc_new_expr(vcc_type_t fmt)
97
{
98
        struct expr *e;
99
100 680823
        ALLOC_OBJ(e, EXPR_MAGIC);
101 680823
        AN(e);
102 680823
        e->vsb = VSB_new_auto();
103 680823
        e->fmt = fmt;
104 680823
        e->constant = EXPR_VAR;
105 680823
        return (e);
106
}
107
108
static struct expr * v_printflike_(2, 3)
109 182386
vcc_mk_expr(vcc_type_t fmt, const char *str, ...)
110
{
111
        va_list ap;
112
        struct expr *e;
113
114 182386
        e = vcc_new_expr(fmt);
115 182386
        va_start(ap, str);
116 182386
        VSB_vprintf(e->vsb, str, ap);
117 182386
        va_end(ap);
118 182386
        AZ(VSB_finish(e->vsb));
119 182386
        return (e);
120
}
121
122
static void
123 886395
vcc_delete_expr(struct expr *e)
124
{
125 886395
        if (e == NULL)
126 206426
                return;
127 679969
        CHECK_OBJ(e, EXPR_MAGIC);
128 679969
        VSB_destroy(&e->vsb);
129 679969
        FREE_OBJ(e);
130 886395
}
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 60608
vcc_strands_edit(const struct expr *e1, const struct expr *e2)
159
{
160
161 60608
        assert(e2->fmt == STRANDS || e2->fmt == STRINGS);
162
163 60608
        if (e2->fmt == STRANDS)
164 4
                VSB_cat(e1->vsb, VSB_data(e2->vsb));
165 60604
        else if (e2->nstr == 0)
166 0
                VSB_printf(e1->vsb, "vrt_null_strands");
167 60604
        else if (e2->nstr == 1)
168 54626
                VSB_printf(e1->vsb, "TOSTRAND(%s)", VSB_data(e2->vsb));
169
        else {
170 11956
                VSB_printf(e1->vsb, "TOSTRANDS(%d,\v+\n%s\v-)",
171 5978
                   e2->nstr, VSB_data(e2->vsb));
172
        }
173 60608
}
174
175
static struct expr *
176 383063
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 383063
        int nl = 1;
181
182 383063
        (void) tl;
183
184 383063
        AN(e1);
185 383063
        e = vcc_new_expr(fmt);
186 4337818
        while (*p != '\0') {
187 3954755
                if (*p != '\v') {
188 3270719
                        if (*p != '\n' || !nl)
189 3270669
                                VSB_putc(e->vsb, *p);
190 3270719
                        nl = (*p == '\n');
191 3270719
                        p++;
192 3270719
                        continue;
193
                }
194 684036
                assert(*p == '\v');
195 684036
                switch (*++p) {
196 60583
                case '+': VSB_cat(e->vsb, "\v+"); nl = 0; break;
197 63748
                case '-': VSB_cat(e->vsb, "\v-"); nl = 0; break;
198
                case 'S':
199
                case 's':
200 21037
                        e3 = (*p == 'S' ? e1 : e2);
201 21037
                        AN(e3);
202 21037
                        assert(e1->fmt == STRINGS);
203 21037
                        if (e3->nstr > 1) {
204 26
                                VSB_cat(e->vsb,
205
                                    "\nVRT_STRANDS_string(ctx,\v+\n");
206 26
                                vcc_strands_edit(e, e3);
207 26
                                VSB_cat(e->vsb,
208
                                    "\v-\n)\n");
209 26
                        } else {
210 21011
                                VSB_cat(e->vsb, VSB_data(e3->vsb));
211
                        }
212 21037
                        break;
213
                case 'T':
214
                case 't':
215 60582
                        e3 = (*p == 'T' ? e1 : e2);
216 60582
                        AN(e3);
217 60582
                        vcc_strands_edit(e, e3);
218 60582
                        break;
219
                case '1':
220 301475
                        VSB_cat(e->vsb, VSB_data(e1->vsb));
221 301475
                        break;
222
                case '2':
223 176611
                        AN(e2);
224 176611
                        VSB_cat(e->vsb, VSB_data(e2->vsb));
225 176611
                        break;
226
                default:
227 0
                        WRONG("Illegal edit in VCC expression");
228 0
                }
229 684036
                p++;
230
        }
231 383063
        AZ(VSB_finish(e->vsb));
232 383063
        e->t1 = e1->t1;
233 383063
        e->t2 = e1->t2;
234 383063
        if (e2 != NULL)
235 176642
                e->t2 = e2->t2;
236 383063
        vcc_delete_expr(e1);
237 383063
        vcc_delete_expr(e2);
238 383063
        return (e);
239
}
240
241
/*--------------------------------------------------------------------
242
 * Expand finished expression into C-source code
243
 */
244
245
static void
246 120262
vcc_expr_fmt(struct vsb *d, int ind, const struct expr *e1)
247
{
248
        char *p;
249
        int i;
250
251 120262
        if (!e1->fmt->noindent) {
252 1153358
                for (i = 0; i < ind; i++)
253 1038882
                        VSB_putc(d, ' ');
254 114476
        }
255 120262
        p = VSB_data(e1->vsb);
256 10199147
        while (*p != '\0') {
257 10078889
                if (*p == '\n') {
258 288772
                        VSB_putc(d, '\n');
259 288772
                        if (*++p == '\0')
260 4
                                break;
261 3086168
                        for (i = 0; i < ind; i++)
262 2797400
                                VSB_putc(d, ' ');
263 10078885
                } else if (*p != '\v') {
264 9650629
                        VSB_putc(d, *p++);
265 9650629
                } else {
266 139488
                        switch (*++p) {
267 69744
                        case '+': ind += INDENT; break;
268 69744
                        case '-': ind -= INDENT; break;
269 0
                        default:  WRONG("Illegal format in VCC expression");
270 0
                        }
271 139488
                        p++;
272
                }
273
        }
274 120262
}
275
276
/*--------------------------------------------------------------------
277
 */
278
279
static void
280 86650
vcc_expr_tobool(struct vcc *tl, struct expr **e)
281
{
282
283 86650
        if ((*e)->fmt == BOOL)
284 63475
                return;
285 23175
        if ((*e)->fmt == BACKEND || (*e)->fmt == INT)
286 10
                *e = vcc_expr_edit(tl, BOOL, "(\v1 != 0)", *e, NULL);
287 23165
        else if ((*e)->fmt == DURATION)
288 2
                *e = vcc_expr_edit(tl, BOOL, "(\v1 > 0)", *e, NULL);
289 23163
        else if ((*e)->fmt == STRINGS)
290 23157
                *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 86650
}
297
298
/*--------------------------------------------------------------------
299
 */
300
301
static void
302 62265
vcc_expr_tostring(struct vcc *tl, struct expr **e)
303
{
304
        const char *p;
305 62265
        uint8_t constant = EXPR_VAR;
306
307 62265
        CHECK_OBJ_NOTNULL(*e, EXPR_MAGIC);
308 62265
        assert((*e)->fmt != STRINGS);
309
310 62265
        p = (*e)->fmt->tostring;
311 62265
        if (p != NULL) {
312 62264
                AN(*p);
313 62264
                *e = vcc_expr_edit(tl, STRINGS, p, *e, NULL);
314 62264
                (*e)->constant = constant;
315 62264
                (*e)->nstr = 1;
316 62264
        } else {
317 2
                VSB_printf(tl->sb,
318
                    "Cannot convert %s to STRING.\n",
319 1
                    vcc_utype((*e)->fmt));
320 1
                vcc_ErrWhere2(tl, (*e)->t1, tl->t);
321
        }
322 62265
}
323
324
/*--------------------------------------------------------------------
325
 */
326
327
void v_matchproto_(sym_expr_t)
328 423
vcc_Eval_Handle(struct vcc *tl, struct expr **e, struct token *t,
329
    struct symbol *sym, vcc_type_t type)
330
{
331
332 423
        (void)t;
333 423
        (void)tl;
334 423
        AN(sym->rname);
335 423
        AZ(type->stringform);
336
337 423
        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 423
                *e = vcc_mk_expr(sym->type, "%s", sym->rname);
344 423
                (*e)->constant = EXPR_VAR;
345 423
                (*e)->nstr = 1;
346 423
                if ((*e)->fmt == STRING)
347 0
                        (*e)->fmt = STRINGS;
348
        }
349 423
}
350
351
void v_matchproto_(sym_expr_t)
352 25
vcc_Eval_Sub(struct vcc *tl, struct expr **e, struct token *t,
353
    struct symbol *sym, vcc_type_t type)
354
{
355
356 25
        (void)t;
357 25
        (void)tl;
358 25
        AN(sym->rname);
359 25
        AZ(type->stringform);
360
361 25
        assert (sym->type == SUB);
362
363 25
        if (type == SUB) {
364 22
                *e = vcc_mk_expr(sym->type, "%s", sym->rname);
365 22
                (*e)->constant = EXPR_CONST;
366 22
                return;
367
        }
368
369 6
        VSB_printf(tl->sb, "Symbol '%s' can only be used as a %s expression\n",
370 3
            sym->name, sym->type->name);
371 3
        vcc_ErrWhere(tl, tl->t);
372 25
}
373
374
/*--------------------------------------------------------------------
375
 */
376
377
void v_matchproto_(sym_expr_t)
378 140484
vcc_Eval_Var(struct vcc *tl, struct expr **e, struct token *t,
379
    struct symbol *sym, vcc_type_t type)
380
{
381
382 140484
        (void)type;
383 140484
        vcc_AddUses(tl, t, NULL, sym, XREF_READ);
384 140484
        ERRCHK(tl);
385 140484
        *e = vcc_mk_expr(sym->type, "%s", sym->rname);
386 140484
        (*e)->constant = EXPR_VAR;
387 140484
        (*e)->nstr = 1;
388 140484
        if ((*e)->fmt == STRING)
389 55292
                (*e)->fmt = STRINGS;
390 140484
}
391
392
void v_matchproto_(sym_expr_t)
393 11
vcc_Eval_ProtectedHeader(struct vcc *tl, struct expr **e, struct token *t,
394
    struct symbol *sym, vcc_type_t type)
395
{
396
397 11
        AN(sym);
398 11
        AZ(sym->lorev);
399
400 11
        vcc_Header_Fh(tl, sym);
401 11
        sym->eval = vcc_Eval_Var;
402 11
        vcc_Eval_Var(tl, e, t, sym, type);
403 11
}
404
405
/*--------------------------------------------------------------------
406
 */
407
408
static struct expr *
409 172
vcc_priv_arg(struct vcc *tl, const char *p, struct symbol *sym)
410
{
411
        char buf[64];
412
        struct inifin *ifp;
413 172
        const char *f = NULL;
414
415 172
        AN(sym);
416 172
        AN(sym->vmod_name);
417
418 172
        if (!strcmp(p, "PRIV_VCL"))
419 15
                return (vcc_mk_expr(VOID, "&vmod_priv_%s", sym->vmod_name));
420
421 157
        if (!strcmp(p, "PRIV_CALL")) {
422 22
                bprintf(buf, "vmod_priv_%u", tl->unique++);
423 22
                ifp = New_IniFin(tl);
424 22
                Fh(tl, 0, "static struct vmod_priv %s;\n", buf);
425 22
                VSB_printf(ifp->fin, "\tVRT_priv_fini(ctx, &%s);", buf);
426 22
                return (vcc_mk_expr(VOID, "&%s", buf));
427
        }
428
429 135
        if (!strcmp(p, "PRIV_TASK"))
430 127
                f = "task";
431 8
        else if (!strcmp(p, "PRIV_TOP")) {
432 8
                f = "top";
433 8
                sym->r_methods &= VCL_MET_TASK_C;
434 8
        } else {
435 0
                WRONG("Wrong PRIV_ type");
436
        }
437 135
        AN(f);
438
439 135
        return (vcc_mk_expr(VOID, "VRT_priv_%s(ctx, &VGC_vmod_%s)",
440 135
            f, sym->vmod_name));
441 172
}
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 1819
vcc_do_enum(struct vcc *tl, const char *cfunc, int len, const char *ptr)
457
{
458
        const char *r;
459
460 1819
        (void)tl;
461 1819
        r = strchr(cfunc, '.');
462 1819
        AN(r);
463 1819
        return (vcc_mk_expr(VOID, "*%.*s.enum_%.*s",
464 1819
            (int)(r - cfunc), cfunc, len, ptr));
465
}
466
467
static void
468 3995
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 3995
        if (fa->type == ENUM) {
474 1337
                ExpectErr(tl, ID);
475 1337
                ERRCHK(tl);
476 4571
                VTAILQ_FOREACH(vv, &fa->enums->children, list)
477 4569
                        if (vcc_IdIs(tl->t, vv->value))
478 1335
                                break;
479 1337
                if (vv == NULL) {
480 2
                        VSB_cat(tl->sb, "Wrong enum value.");
481 2
                        VSB_cat(tl->sb, "  Expected one of:\n");
482 10
                        VTAILQ_FOREACH(vv, &fa->enums->children, list)
483 8
                                VSB_printf(tl->sb, "\t%s\n", vv->value);
484 2
                        vcc_ErrWhere(tl, tl->t);
485 2
                        return;
486
                }
487 1335
                fa->result = vcc_do_enum(tl, cfunc, PF(tl->t));
488 1335
                SkipToken(tl, ID);
489 1335
        } else {
490 2658
                if (fa->type == SUB)
491 22
                        tl->subref++;
492 2658
                vcc_expr0(tl, &e2, fa->type);
493 2658
                ERRCHK(tl);
494 2650
                assert(e2->fmt == fa->type);
495 2650
                fa->result = e2;
496
        }
497 3985
        fa->avail = 1;
498 3995
}
499
500
static void
501 2771
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 2771
        CAST_OBJ_NOTNULL(vv, priv, VJSN_VAL_MAGIC);
516 2771
        assert(vjsn_is_array(vv));
517 2771
        vv = VTAILQ_FIRST(&vv->children);
518 2771
        rfmt = VCC_Type(VTAILQ_FIRST(&vv->children)->value);
519 2771
        AN(rfmt);
520 2771
        vv = VTAILQ_NEXT(vv, list);
521 2771
        cfunc = vv->value;
522 2771
        vv = VTAILQ_NEXT(vv, list);
523 2771
        sa = vv->value;
524 2771
        if (*sa == '\0') {
525 2357
                sa = NULL;
526 2357
        }
527 2771
        vv = VTAILQ_NEXT(vv, list);
528 2771
        if (sym->kind == SYM_METHOD) {
529 774
                if (*e == NULL) {
530 2
                        VSB_cat(tl->sb, "Syntax error.");
531 2
                        tl->err = 1;
532 2
                        return;
533
                }
534 772
                vcc_NextToken(tl);
535 772
                AZ(extra);
536 772
                AN((*e)->instance);
537 772
                extra = (*e)->instance->rname;
538 772
        }
539 2769
        tf = VTAILQ_PREV(tl->t, tokenhead, list);
540 2769
        SkipToken(tl, '(');
541 2769
        if (extra == NULL) {
542 1796
                extra = "";
543 1796
                extra_sep = "";
544 1796
        } else {
545 973
                AN(*extra);
546 973
                extra_sep = ", ";
547
        }
548 2769
        VTAILQ_INIT(&head);
549 9493
        for (;vv != NULL; vv = VTAILQ_NEXT(vv, list)) {
550 6724
                assert(vjsn_is_array(vv));
551 6724
                fa = calloc(1, sizeof *fa);
552 6724
                AN(fa);
553 6724
                VTAILQ_INSERT_TAIL(&head, fa, list);
554
555 6724
                vvp = VTAILQ_FIRST(&vv->children);
556 6724
                if (!memcmp(vvp->value, "PRIV_", 5)) {
557 172
                        fa->result = vcc_priv_arg(tl, vvp->value, sym);
558 172
                        vvp = VTAILQ_NEXT(vvp, list);
559 172
                        if (vvp != NULL)
560 172
                                fa->cname = fa->name = vvp->value;
561 172
                        continue;
562
                }
563 6552
                fa->type = VCC_Type(vvp->value);
564 6552
                AN(fa->type);
565 6552
                vvp = VTAILQ_NEXT(vvp, list);
566 6552
                if (vvp != NULL) {
567 6552
                        fa->name = vvp->value;
568 6552
                        vvp = VTAILQ_NEXT(vvp, list);
569 6552
                        AN(vvp); /* vmod_syntax 2.0 */
570 6552
                        fa->cname = vvp->value;
571 6552
                        vvp = VTAILQ_NEXT(vvp, list);
572 6552
                        if (vvp != NULL) {
573 4581
                                fa->val = vvp->value;
574 4581
                                vvp = VTAILQ_NEXT(vvp, list);
575 4581
                                if (vvp != NULL) {
576 3761
                                        fa->enums = vvp;
577 3761
                                        vvp = VTAILQ_NEXT(vvp, list);
578 3761
                                }
579 4581
                        }
580 6552
                }
581 6552
                if (sa != NULL && vvp != NULL && vjsn_is_true(vvp)) {
582 2065
                        fa->optional = 1;
583 2065
                        vvp = VTAILQ_NEXT(vvp, list);
584 2065
                }
585 6552
                AZ(vvp);
586 6552
        }
587
588 4446
        VTAILQ_FOREACH(fa, &head, list) {
589 3988
                if (tl->t->tok == ')')
590 89
                        break;
591 3899
                if (fa->result != NULL)
592 123
                        continue;
593 3776
                if (tl->t->tok == ID) {
594 2776
                        t1 = VTAILQ_NEXT(tl->t, list);
595 2776
                        if (t1->tok == '=')
596 760
                                break;
597 2016
                }
598 3016
                vcc_do_arg(tl, cfunc, fa);
599 3016
                if (tl->err)
600 20
                        VSB_printf(tl->sb, "Expected argument: %s %s\n\n",
601 10
                            fa->type->name,
602 10
                            fa->name ? fa->name : "(unnamed argument)");
603 3016
                ERRCHK(tl);
604 3006
                if (tl->t->tok == ')')
605 1452
                        break;
606 1554
                SkipToken(tl, ',');
607 1554
        }
608 2980
        while (tl->t->tok == ID) {
609 3224
                VTAILQ_FOREACH(fa, &head, list) {
610 3223
                        if (fa->name == NULL)
611 12
                                continue;
612 3211
                        if (vcc_IdIs(tl->t, fa->name))
613 980
                                break;
614 2231
                }
615 981
                if (fa == NULL) {
616 2
                        VSB_printf(tl->sb, "Unknown argument '%.*s'\n",
617 1
                            PF(tl->t));
618 1
                        vcc_ErrWhere(tl, tl->t);
619 1
                        return;
620
                }
621 980
                if (fa->result != NULL) {
622 1
                        AN(fa->name);
623 2
                        VSB_printf(tl->sb, "Argument '%s' already used\n",
624 1
                            fa->name);
625 1
                        vcc_ErrWhere(tl, tl->t);
626 1
                        return;
627
                }
628 979
                vcc_NextToken(tl);
629 979
                SkipToken(tl, '=');
630 979
                vcc_do_arg(tl, cfunc, fa);
631 979
                ERRCHK(tl);
632 979
                if (tl->t->tok == ')')
633 758
                        break;
634 221
                SkipToken(tl, ',');
635
        }
636
637 2757
        if (sa != NULL)
638 820
                e1 = vcc_mk_expr(rfmt, "%s(ctx%s%s,\v+\n&(%s)\v+ {\n",
639 410
                    cfunc, extra_sep, extra, sa);
640
        else
641 4694
                e1 = vcc_mk_expr(rfmt, "%s(ctx%s%s\v+",
642 2347
                    cfunc, extra_sep, extra);
643 2757
        n = 0;
644 9410
        VTAILQ_FOREACH_SAFE(fa, &head, list, fa2) {
645 6653
                n++;
646 6653
                if (fa->optional) {
647 2021
                        AN(fa->cname);
648 2021
                        bprintf(ssa, "\v1.valid_%s = %d,\n",
649
                            fa->cname, fa->avail);
650 2021
                        e1 = vcc_expr_edit(tl, e1->fmt, ssa, e1, NULL);
651 2021
                }
652 6653
                if (fa->result == NULL && fa->type == ENUM && fa->val != NULL)
653 484
                        fa->result = vcc_do_enum(tl, cfunc, strlen(fa->val), fa->val);
654 6653
                if (fa->result == NULL && fa->val != NULL)
655 609
                        fa->result = vcc_mk_expr(fa->type, "%s", fa->val);
656 6653
                if (fa->result != NULL && sa != NULL) {
657 886
                        if (fa->cname)
658 865
                                bprintf(ssa, "\v1.%s = \v2,\n", fa->cname);
659
                        else
660 21
                                bprintf(ssa, "\v1.arg%d = \v2,\n", n);
661 886
                        e1 = vcc_expr_edit(tl, e1->fmt, ssa, e1, fa->result);
662 6653
                } else if (fa->result != NULL) {
663 8718
                        e1 = vcc_expr_edit(tl, e1->fmt, "\v1,\n\v2",
664 4359
                            e1, fa->result);
665 5767
                } else if (!fa->optional) {
666 1
                        if (fa->name)
667 2
                                VSB_printf(tl->sb, "Argument '%s' missing\n",
668 1
                                    fa->name);
669
                        else
670 0
                                VSB_printf(tl->sb, "Argument %d missing\n", n);
671 1
                        vcc_ErrWhere(tl, tl->t);
672 1
                }
673 6653
                free(fa);
674 6653
        }
675 2757
        if (sa != NULL) {
676 410
                *e = vcc_expr_edit(tl, e1->fmt, "\v1\v-\n}\v-\n)", e1, NULL);
677 410
        } else {
678 2347
                *e = vcc_expr_edit(tl, e1->fmt, "\v1\v-\n)", e1, NULL);
679
        }
680 2757
        SkipToken(tl, ')');
681 2756
        vcc_AddUses(tl, tf, NULL, sym, XREF_READ);
682 2771
}
683
684
685
/*--------------------------------------------------------------------
686
 */
687
688
void
689 201
vcc_Eval_Func(struct vcc *tl, const struct vjsn_val *spec,
690
    const char *extra, struct symbol *sym)
691
{
692 201
        struct expr *e = NULL;
693
694 201
        vcc_func(tl, &e, spec, extra, sym);
695 201
        if (tl->err)
696 0
                VSB_cat(tl->sb, "While compiling function call:\n");
697 201
        ERRCHK(tl);
698 201
        vcc_expr_fmt(tl->fb, tl->indent, e);
699 201
        VSB_cat(tl->fb, ";\n");
700 201
        vcc_delete_expr(e);
701 201
}
702
703
/*--------------------------------------------------------------------
704
 */
705
706
void v_matchproto_(sym_expr_t)
707 2081
vcc_Eval_SymFunc(struct vcc *tl, struct expr **e, struct token *t,
708
    struct symbol *sym, vcc_type_t fmt)
709
{
710
711 2081
        (void)t;
712 2081
        (void)fmt;
713 2081
        assert(sym->kind == SYM_FUNC || sym->kind == SYM_METHOD);
714 2081
        AN(sym->eval_priv);
715
716 2081
        vcc_func(tl, e, sym->eval_priv, sym->extra, sym);
717 2081
        ERRCHK(tl);
718 2071
        if ((*e)->fmt == STRING) {
719 752
                (*e)->fmt = STRINGS;
720 752
                (*e)->nstr = 1;
721 752
        }
722 2081
}
723
724
/*--------------------------------------------------------------------
725
 */
726
727
static void
728 32944
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 32944
        assert(fmt != VOID);
735 32944
        if (fmt == BYTES) {
736 50
                vcc_ByteVal(tl, &vi);
737 50
                ERRCHK(tl);
738 48
                e1 = vcc_mk_expr(BYTES, "%ju", (intmax_t)vi);
739 48
        } else {
740 32894
                t = tl->t;
741 32894
                vcc_NextToken(tl);
742 32894
                if (tl->t->tok == ID) {
743 3148
                        e1 = vcc_mk_expr(DURATION, "%s%.3f * %g",
744 3148
                            sign, t->num, vcc_DurationUnit(tl));
745 3148
                        ERRCHK(tl);
746 32893
                } else if (fmt == REAL || t->tok == FNUM) {
747 154
                        e1 = vcc_mk_expr(REAL, "%s%.3f", sign, t->num);
748 154
                } else {
749 29592
                        e1 = vcc_mk_expr(INT, "%s%.0f", sign, t->num);
750
                }
751
        }
752 32941
        e1->constant = EXPR_CONST;
753 32941
        *e = e1;
754 32944
}
755
756
/*--------------------------------------------------------------------
757
 * SYNTAX:
758
 *    Expr5:
759
 *      '(' ExprCor ')'
760
 *      symbol
761
 *      CNUM
762
 *      FNUM
763
 *      CSTR
764
 *      CBLOB
765
 */
766
767
static void
768 296717
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 296717
        sign = "";
776 296717
        *e = NULL;
777 296717
        if (tl->t->tok == '(') {
778 2925
                SkipToken(tl, '(');
779 2925
                vcc_expr_cor(tl, &e2, fmt);
780 2925
                ERRCHK(tl);
781 2925
                SkipToken(tl, ')');
782 2925
                if (e2->fmt == STRINGS)
783 6
                        *e = e2;
784
                else
785 2919
                        *e = vcc_expr_edit(tl, e2->fmt, "(\v1)", e2, NULL);
786 2925
                return;
787
        }
788 293792
        switch (tl->t->tok) {
789
        case ID:
790 146205
                t = tl->t;
791 146205
                t1 = vcc_PeekToken(tl);
792 146205
                AN(t1);
793 146205
                sym = VCC_SymbolGet(tl, SYM_MAIN, SYM_NONE,
794
                    SYMTAB_PARTIAL_NOERR, XREF_REF);
795 146205
                if (sym == NULL && fmt->global_pfx != NULL && t1->tok != '.') {
796 9
                        sym = VCC_SymbolGet(tl, SYM_MAIN, SYM_NONE,
797
                            SYMTAB_CREATE, XREF_REF);
798 9
                        ERRCHK(tl);
799 7
                        AN(sym);
800 7
                        VCC_GlobalSymbol(sym, fmt);
801 7
                }
802 146203
                ERRCHK(tl);
803 146203
                if (sym == NULL)
804 12
                        AZ(VCC_SymbolGet(tl, SYM_MAIN, SYM_NONE,
805
                            SYMTAB_PARTIAL, XREF_REF));
806 146203
                ERRCHK(tl);
807 146191
                AN(sym);
808 146191
                if (sym->kind == SYM_INSTANCE) {
809 774
                        AZ(*e);
810 774
                        *e = vcc_new_expr(sym->type);
811 774
                        (*e)->instance = sym;
812 774
                        return;
813
                }
814 145417
                if (sym->kind == SYM_FUNC && sym->type == VOID) {
815 1
                        VSB_cat(tl->sb, "Function returns VOID:\n");
816 1
                        vcc_ErrWhere(tl, tl->t);
817 1
                        return;
818
                }
819 145416
                if (sym->eval != NULL) {
820 145413
                        AN(sym->eval);
821 145413
                        AZ(*e);
822 145413
                        sym->eval(tl, e, t, sym, fmt);
823 145413
                        if (tl->err) {
824 16
                                VSB_cat(tl->sb,
825
                                    "While compiling function call:\n\n");
826 16
                                vcc_ErrWhere2(tl, t, tl->t);
827 16
                        }
828 145413
                        ERRCHK(tl);
829
                        /* Unless asked for a HEADER, fold to string here */
830 145397
                        if (*e && fmt != HEADER && (*e)->fmt == HEADER) {
831 41378
                                vcc_expr_tostring(tl, e);
832 41378
                                ERRCHK(tl);
833 41378
                        }
834 145397
                        return;
835
                }
836 6
                VSB_printf(tl->sb,
837
                    "Symbol '%.*s' type (%s) cannot be used in expression.\n",
838 3
                    PF(t), sym->kind->name);
839 3
                vcc_ErrWhere(tl, t);
840 3
                if (sym->def_b != NULL) {
841 1
                        VSB_cat(tl->sb, "That symbol was defined here:\n");
842 1
                        vcc_ErrWhere(tl, sym->def_b);
843 1
                }
844 3
                return;
845
        case CSTR:
846 114628
                assert(fmt != VOID);
847 114628
                if (fmt == IP) {
848 30
                        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 1
                                VSB_cat(tl->sb,
856
                                    "Cannot convert to an IP address: ");
857 1
                                vcc_ErrToken(tl, tl->t);
858 1
                                vcc_ErrWhere(tl, tl->t);
859 1
                                return;
860
                        }
861 58
                        Resolve_Sockaddr(tl, tl->t->dec, "80",
862
                            &ip, NULL, &ip, NULL, NULL, 1,
863 29
                            tl->t, "IP constant");
864 29
                        ERRCHK(tl);
865 26
                        e1 = vcc_mk_expr(IP, "%s", ip);
866 26
                        ERRCHK(tl);
867 114624
                } else if (fmt == REGEX) {
868 8780
                        e1 = vcc_new_expr(REGEX);
869 8780
                        vcc_regexp(tl, e1->vsb);
870 8780
                        AZ(VSB_finish(e1->vsb));
871 8780
                } else {
872 105818
                        e1 = vcc_new_expr(STRINGS);
873 105818
                        EncToken(e1->vsb, tl->t);
874 105818
                        AZ(VSB_finish(e1->vsb));
875 105818
                        e1->constant |= EXPR_STR_CONST;
876 105818
                        e1->nstr = 1;
877
                }
878 114624
                e1->t1 = tl->t;
879 114624
                e1->constant |= EXPR_CONST;
880 114624
                vcc_NextToken(tl);
881 114624
                *e = e1;
882 114624
                return;
883
        case '-':
884 62
                if (fmt != INT &&
885 16
                    fmt != REAL &&
886 4
                    fmt != DURATION &&
887 1
                    fmt != STRINGS)
888 0
                        break;
889 61
                vcc_NextToken(tl);
890 61
                if (tl->t->tok != FNUM && tl->t->tok != CNUM) {
891 9
                        vcc_expr_cor(tl, &e1, fmt);
892 9
                        ERRCHK(tl);
893 9
                        *e = vcc_expr_edit(tl, e1->fmt, "-(\v1)", e1, NULL);
894 9
                        return;
895
                }
896 52
                sign = "-";
897
                /* FALLTHROUGH */
898
        case FNUM:
899
        case CNUM:
900 32944
                vcc_number(tl, e, fmt, sign);
901 32944
                return;
902
        case CBLOB:
903 2
                e1 = vcc_new_expr(BLOB);
904 2
                VSB_printf(e1->vsb, "%s", tl->t->dec);
905 2
                AZ(VSB_finish(e1->vsb));
906 2
                e1->constant |= EXPR_STR_CONST;
907 2
                e1->t1 = tl->t;
908 2
                vcc_NextToken(tl);
909 2
                *e = e1;
910 2
                return;
911
        default:
912 4
                break;
913
        }
914 4
        VSB_cat(tl->sb, "Unknown token ");
915 4
        vcc_ErrToken(tl, tl->t);
916 4
        VSB_printf(tl->sb, " when looking for %s\n\n", vcc_utype(fmt));
917 4
        vcc_ErrWhere(tl, tl->t);
918 296717
}
919
920
/*--------------------------------------------------------------------
921
 * SYNTAX:
922
 *    Expr4:
923
 *      Expr5 [ '.' (type_attribute | type_method()) ]*
924
 */
925
926
void
927 2902
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 2902
        (void)t;
933 2902
        impl = VCC_Type_EvalMethod(tl, sym);
934 2902
        ERRCHK(tl);
935 2902
        AN(impl);
936 2902
        *e = vcc_expr_edit(tl, fmt, impl, *e, NULL);
937 2902
}
938
939
static void
940 296717
vcc_expr4(struct vcc *tl, struct expr **e, vcc_type_t fmt)
941
{
942
        struct symbol *sym;
943
944 296717
        *e = NULL;
945 296717
        vcc_expr5(tl, e, fmt);
946 296717
        ERRCHK(tl);
947 296669
        AN(*e);
948 300343
        while (tl->t->tok == '.') {
949 3677
                vcc_NextToken(tl);
950 3677
                ExpectErr(tl, ID);
951
952 3676
                sym = VCC_TypeSymbol(tl, SYM_METHOD, (*e)->fmt);
953 3676
                if (sym == NULL) {
954 2
                        VSB_cat(tl->sb, "Unknown property ");
955 2
                        vcc_ErrToken(tl, tl->t);
956 2
                        VSB_printf(tl->sb, " for type %s\n", (*e)->fmt->name);
957 2
                        vcc_ErrWhere(tl, tl->t);
958 2
                        return;
959
                }
960
961 3674
                AN(sym->eval);
962 3674
                sym->eval(tl, e, tl->t, sym, sym->type);
963 3674
                ERRCHK(tl);
964 3674
                if ((*e)->fmt == STRING) {
965 2891
                        (*e)->fmt = STRINGS;
966 2891
                        (*e)->nstr = 1;
967 2891
                }
968
        }
969 296717
}
970
971
/*--------------------------------------------------------------------
972
 * SYNTAX:
973
 *    ExprMul:
974
 *      Expr4 { {'*'|'/'|'%'} Expr4 } *
975
 */
976
977
static void
978 287564
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 287564
        *e = NULL;
986 287564
        vcc_expr4(tl, e, fmt);
987 287564
        ERRCHK(tl);
988 287520
        AN(*e);
989
990 287557
        while (tl->t->tok == '*' || tl->t->tok == '/' || tl->t->tok == '%') {
991 41
                if (tl->t->tok == '%' && ((*e)->fmt != INT)) {
992 1
                        VSB_cat(tl->sb, "Operator % only possible on INT.\n");
993 1
                        vcc_ErrWhere(tl, tl->t);
994 1
                        return;
995
                }
996 40
                f2 = (*e)->fmt->multype;
997 40
                if (f2 == NULL) {
998 2
                        VSB_printf(tl->sb,
999
                            "Operator %.*s not possible on type %s.\n",
1000 1
                            PF(tl->t), vcc_utype((*e)->fmt));
1001 1
                        vcc_ErrWhere(tl, tl->t);
1002 1
                        return;
1003
                }
1004 39
                tk = tl->t;
1005 39
                vcc_NextToken(tl);
1006 39
                vcc_expr4(tl, &e2, f2);
1007 39
                ERRCHK(tl);
1008 39
                if (e2->fmt != INT && e2->fmt != f2) {
1009 4
                        VSB_printf(tl->sb, "%s %.*s %s not possible.\n",
1010 2
                            vcc_utype((*e)->fmt), PF(tk), vcc_utype(e2->fmt));
1011 2
                        vcc_ErrWhere(tl, tk);
1012 2
                        return;
1013
                }
1014 37
                bprintf(buf, "(\v1%c\v2)", tk->tok);
1015 37
                *e = vcc_expr_edit(tl, (*e)->fmt, buf, *e, e2);
1016
        }
1017 287564
}
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 217853
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 217853
        *e = NULL;
1059 217853
        vcc_expr_mul(tl, e, fmt);
1060 217853
        ERRCHK(tl);
1061
1062 287505
        while (tl->t->tok == '+' || tl->t->tok == '-') {
1063 69711
                tk = tl->t;
1064 1114622
                for (ap = vcc_adds; ap->op != EOI; ap++)
1065 1044986
                        if (tk->tok == ap->op && (*e)->fmt == ap->a)
1066 75
                                break;
1067 69711
                vcc_NextToken(tl);
1068 69711
                if (ap->op == EOI && fmt == STRINGS)
1069 506
                        vcc_expr_mul(tl, &e2, STRINGS);
1070
                else
1071 69205
                        vcc_expr_mul(tl, &e2, (*e)->fmt);
1072 69711
                ERRCHK(tl);
1073
1074 1114744
                for (ap = vcc_adds; ap->op != EOI; ap++)
1075 1045098
                        if (tk->tok == ap->op && (*e)->fmt == ap->a &&
1076 1045098
                            e2->fmt == ap->b)
1077 65
                                break;
1078
1079 69711
                if (ap->op == '+') {
1080 45
                        *e = vcc_expr_edit(tl, ap->fmt, "(\v1 + \v2)", *e, e2);
1081 69711
                } else if (ap->op == '-') {
1082 20
                        *e = vcc_expr_edit(tl, ap->fmt, "(\v1 - \v2)", *e, e2);
1083 69682
                } else if (tk->tok == '+' &&
1084 69645
                    ((*e)->fmt == STRINGS || fmt == STRINGS)) {
1085 69635
                        if ((*e)->fmt != STRINGS)
1086 6
                                vcc_expr_tostring(tl, e);
1087 69635
                        if (e2->fmt != STRINGS)
1088 17337
                                vcc_expr_tostring(tl, &e2);
1089 69635
                        if (vcc_islit(*e) && vcc_isconst(e2)) {
1090 50
                                lit = vcc_islit(e2);
1091 100
                                *e = vcc_expr_edit(tl, STRINGS,
1092 50
                                    "\v1\n\v2", *e, e2);
1093 50
                                (*e)->constant = EXPR_CONST;
1094 50
                                (*e)->nstr = 1;
1095 50
                                if (lit)
1096 50
                                        (*e)->constant |= EXPR_STR_CONST;
1097 50
                        } else {
1098 69585
                                n = (*e)->nstr + e2->nstr;
1099 139170
                                *e = vcc_expr_edit(tl, STRINGS,
1100 69585
                                    "\v1,\n\v2", *e, e2);
1101 69585
                                (*e)->constant = EXPR_VAR;
1102 69585
                                (*e)->nstr = n;
1103
                        }
1104 69635
                } else {
1105 22
                        VSB_printf(tl->sb, "%s %.*s %s not possible.\n",
1106 11
                            vcc_utype((*e)->fmt), PF(tk), vcc_utype(e2->fmt));
1107 11
                        vcc_ErrWhere2(tl, tk, tl->t);
1108 11
                        return;
1109
                }
1110
        }
1111 217853
}
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 14550
cmp_simple(struct vcc *tl, struct expr **e, const struct cmps *cp)
1139
{
1140
        struct expr *e2;
1141
        struct token *tk;
1142
1143 14550
        tk = tl->t;
1144 14550
        vcc_NextToken(tl);
1145 14550
        vcc_expr_add(tl, &e2, (*e)->fmt);
1146 14550
        ERRCHK(tl);
1147
1148 14547
        if (e2->fmt != (*e)->fmt) {
1149 4
                VSB_printf(tl->sb,
1150
                    "Comparison of different types: %s '%.*s' %s\n",
1151 2
                    vcc_utype((*e)->fmt), PF(tk), vcc_utype(e2->fmt));
1152 2
                vcc_ErrWhere(tl, tk);
1153 2
        } else
1154 14545
                *e = vcc_expr_edit(tl, BOOL, cp->emit, *e, e2);
1155 14550
}
1156
1157
static void v_matchproto_(cmp_f)
1158 8734
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 8734
        *e = vcc_expr_edit(tl, STRING, "\vS", *e, NULL);
1165 8734
        vcc_NextToken(tl);
1166 8734
        t1 = tl->t;
1167 8734
        vcc_expr4(tl, &e2, REGEX);
1168 8734
        ERRCHK(tl);
1169 8730
        vcc_expr_typecheck(tl, &e2, REGEX, t1);
1170 8730
        ERRCHK(tl);
1171 8729
        bprintf(buf, "%sVRT_re_match(ctx, \v1, \v2)", cp->emit);
1172 8729
        *e = vcc_expr_edit(tl, BOOL, buf, *e, e2);
1173 8734
}
1174
1175
static void v_matchproto_(cmp_f)
1176 47
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 47
        vcc_NextToken(tl);
1183 47
        t1 = tl->t;
1184 47
        vcc_expr4(tl, &e2, ACL);
1185 47
        ERRCHK(tl);
1186 44
        vcc_expr_typecheck(tl, &e2, ACL, t1);
1187 44
        ERRCHK(tl);
1188 41
        bprintf(buf, "%sVRT_acl_match(ctx, \v1, \v2)", cp->emit);
1189 41
        *e = vcc_expr_edit(tl, BOOL, buf, e2, *e);
1190 47
}
1191
1192
static void v_matchproto_(cmp_f)
1193 37922
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 37922
        tk = tl->t;
1200 37922
        vcc_NextToken(tl);
1201 37922
        vcc_expr_add(tl, &e2, STRINGS);
1202 37922
        ERRCHK(tl);
1203 37922
        if (vcc_stringstype(e2->fmt) != STRINGS) {
1204 6
                VSB_printf(tl->sb,
1205
                    "Comparison of different types: %s '%.*s' %s\n",
1206 3
                    vcc_utype((*e)->fmt), PF(tk), vcc_utype(e2->fmt));
1207 3
                vcc_ErrWhere(tl, tk);
1208 37922
        } else if ((*e)->nstr == 1 && e2->nstr == 1) {
1209 37888
                bprintf(buf, "(%s VRT_strcmp(\v1, \v2))", cp->emit);
1210 37888
                *e = vcc_expr_edit(tl, BOOL, buf, *e, e2);
1211 37888
        } else {
1212 31
                bprintf(buf, "(%s VRT_CompareStrands(\vT, \vt))", cp->emit);
1213 31
                *e = vcc_expr_edit(tl, BOOL, buf, *e, e2);
1214
        }
1215 37922
}
1216
1217
#define IDENT_REL(typ)                                                  \
1218
        {typ,           T_EQ,           cmp_simple, "(\v1 == \v2)" },   \
1219
        {typ,           T_NEQ,          cmp_simple, "(\v1 != \v2)" }
1220
1221
#define NUM_REL(typ)                                                    \
1222
        IDENT_REL(typ),                                                 \
1223
        {typ,           T_LEQ,          cmp_simple, "(\v1 <= \v2)" },   \
1224
        {typ,           T_GEQ,          cmp_simple, "(\v1 >= \v2)" },   \
1225
        {typ,           '<',            cmp_simple, "(\v1 < \v2)" },    \
1226
        {typ,           '>',            cmp_simple, "(\v1 > \v2)" }
1227
1228
static const struct cmps vcc_cmps[] = {
1229
        NUM_REL(INT),
1230
        NUM_REL(DURATION),
1231
        NUM_REL(BYTES),
1232
        NUM_REL(REAL),
1233
        NUM_REL(TIME),
1234
        IDENT_REL(BACKEND),
1235
        IDENT_REL(ACL),
1236
        IDENT_REL(PROBE),
1237
        IDENT_REL(STEVEDORE),
1238
        IDENT_REL(SUB),
1239
        IDENT_REL(INSTANCE),
1240
1241
        {BOOL,          T_EQ,           cmp_simple, "((!(\v1)) == (!(\v2)))" },
1242
        {BOOL,          T_NEQ,          cmp_simple, "((!(\v1)) != (!(\v2)))" },
1243
        {IP,            T_EQ,           cmp_simple, "!VRT_ipcmp(ctx, \v1, \v2)" },
1244
        {IP,            T_NEQ,          cmp_simple, "VRT_ipcmp(ctx, \v1, \v2)" },
1245
1246
        {IP,            '~',            cmp_acl, "" },
1247
        {IP,            T_NOMATCH,      cmp_acl, "!" },
1248
1249
        {STRINGS,       T_EQ,           cmp_string, "0 =="},
1250
        {STRINGS,       T_NEQ,          cmp_string, "0 !="},
1251
        {STRINGS,       '<',            cmp_string, "0 > "},
1252
        {STRINGS,       '>',            cmp_string, "0 < "},
1253
        {STRINGS,       T_LEQ,          cmp_string, "0 >="},
1254
        {STRINGS,       T_GEQ,          cmp_string, "0 <="},
1255
1256
        {STRINGS,       '~',            cmp_regexp, "" },
1257
        {STRINGS,       T_NOMATCH,      cmp_regexp, "!" },
1258
1259
        {VOID,          0,              NULL, NULL}
1260
};
1261
1262
#undef IDENT_REL
1263
#undef NUM_REL
1264
1265
static void
1266 165381
vcc_expr_cmp(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1267
{
1268
        const struct cmps *cp;
1269
        struct token *tk;
1270
1271 165381
        *e = NULL;
1272 165381
        vcc_expr_add(tl, e, fmt);
1273 165381
        ERRCHK(tl);
1274 165325
        tk = tl->t;
1275
1276 8349095
        for (cp = vcc_cmps; cp->fmt != VOID; cp++) {
1277 8245023
                if (tl->t->tok != cp->token)
1278 7679094
                        continue;
1279 565929
                if (vcc_stringstype((*e)->fmt) != cp->fmt)
1280 504676
                        continue;
1281 61253
                AN(cp->func);
1282 61253
                cp->func(tl, e, cp);
1283 61253
                return;
1284
        }
1285
1286 104072
        switch (tk->tok) {
1287
        case T_EQ:
1288
        case T_NEQ:
1289
        case '<':
1290
        case T_LEQ:
1291
        case '>':
1292
        case T_GEQ:
1293
        case '~':
1294
        case T_NOMATCH:
1295 4
                VSB_printf(tl->sb, "Operator %.*s not possible on %s\n",
1296 2
                    PF(tl->t), vcc_utype((*e)->fmt));
1297 2
                vcc_ErrWhere(tl, tl->t);
1298 2
                return;
1299
        default:
1300 104070
                break;
1301
        }
1302 165381
}
1303
1304
/*--------------------------------------------------------------------
1305
 * SYNTAX:
1306
 *    ExprNot:
1307
 *      '!' ExprCmp
1308
 */
1309
1310
static void
1311 165381
vcc_expr_not(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1312
{
1313
        struct token *tk;
1314
1315 165381
        *e = NULL;
1316 165381
        tk = tl->t;
1317 165381
        if (tl->t->tok == '!')
1318 14483
                vcc_NextToken(tl);
1319 165381
        vcc_expr_cmp(tl, e, fmt);
1320 165381
        ERRCHK(tl);
1321 165304
        if (tk->tok != '!')
1322 150822
                return;
1323 14482
        vcc_expr_tobool(tl, e);
1324 14482
        ERRCHK(tl);
1325 14482
        if ((*e)->fmt != BOOL) {
1326 1
                VSB_cat(tl->sb, "'!' must be followed by BOOL, found ");
1327 1
                VSB_printf(tl->sb, "%s.\n", vcc_utype((*e)->fmt));
1328 1
                vcc_ErrWhere2(tl, tk, tl->t);
1329 1
        } else {
1330 14481
                *e = vcc_expr_edit(tl, BOOL, "!(\v1)", *e, NULL);
1331
        }
1332 165381
}
1333
1334
/*--------------------------------------------------------------------
1335
 * CAND and COR are identical save for a few details, but they are
1336
 * stacked so handling them in the same function is not simpler.
1337
 * Instead have them both call this helper function to do everything.
1338
 */
1339
1340
typedef void upfunc(struct vcc *tl, struct expr **e, vcc_type_t fmt);
1341
1342
static void
1343 252949
vcc_expr_bin_bool(struct vcc *tl, struct expr **e, vcc_type_t fmt,
1344
    unsigned ourtok, upfunc *up, const char *tokstr)
1345
{
1346
        struct expr *e2;
1347
        struct token *tk;
1348
        char buf[32];
1349
1350 252949
        *e = NULL;
1351 252949
        tk = tl->t;
1352 252949
        up(tl, e, fmt);
1353 252949
        ERRCHK(tl);
1354 252791
        if (tl->t->tok != ourtok)
1355 232595
                return;
1356 20196
        vcc_expr_tobool(tl, e);
1357 20196
        ERRCHK(tl);
1358 20196
        if ((*e)->fmt != BOOL) {
1359 4
                VSB_printf(tl->sb,
1360
                    "'%s' must be preceded by BOOL,"
1361 2
                    " found %s.\n", tokstr, vcc_utype((*e)->fmt));
1362 2
                vcc_ErrWhere2(tl, tk, tl->t);
1363 2
                return;
1364
        }
1365 20194
        *e = vcc_expr_edit(tl, BOOL, "(\v+\n\v1", *e, NULL);
1366 60546
        while (tl->t->tok == ourtok) {
1367 40354
                vcc_NextToken(tl);
1368 40354
                tk = tl->t;
1369 40354
                up(tl, &e2, fmt);
1370 40354
                ERRCHK(tl);
1371 40354
                vcc_expr_tobool(tl, &e2);
1372 40354
                ERRCHK(tl);
1373 40354
                if (e2->fmt != BOOL) {
1374 4
                        VSB_printf(tl->sb,
1375
                            "'%s' must be followed by BOOL,"
1376 2
                            " found %s.\n", tokstr, vcc_utype(e2->fmt));
1377 2
                        vcc_ErrWhere2(tl, tk, tl->t);
1378 2
                        vcc_delete_expr(e2);
1379 2
                        return;
1380
                }
1381 40352
                bprintf(buf, "\v1\v-\n%s\v+\n\v2", tokstr);
1382 40352
                *e = vcc_expr_edit(tl, BOOL, buf, *e, e2);
1383
        }
1384 20192
        *e = vcc_expr_edit(tl, BOOL, "\v1\v-\n)", *e, NULL);
1385 252949
}
1386
1387
/*--------------------------------------------------------------------
1388
 * SYNTAX:
1389
 *    ExprCand:
1390
 *      ExprNot { '&&' ExprNot } *
1391
 */
1392
1393
static void
1394 127922
vcc_expr_cand(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1395
{
1396
1397 127922
        vcc_expr_bin_bool(tl, e, fmt, T_CAND, vcc_expr_not, "&&");
1398 127922
}
1399
1400
/*--------------------------------------------------------------------
1401
 * SYNTAX:
1402
 *    ExprCOR:
1403
 *      ExprCand { '||' ExprCand } *
1404
 */
1405
1406
static void
1407 125027
vcc_expr_cor(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1408
{
1409
1410 125027
        vcc_expr_bin_bool(tl, e, fmt, T_COR, vcc_expr_cand, "||");
1411 125027
}
1412
1413
/*--------------------------------------------------------------------
1414
 * This function is the entry-point for getting an expression with
1415
 * a particular type, ready for inclusion in the VGC.
1416
 */
1417
1418
static void
1419 122093
vcc_expr0(struct vcc *tl, struct expr **e, vcc_type_t fmt)
1420
{
1421
        struct token *t1;
1422
1423 122093
        assert(fmt != VOID);
1424 122093
        assert(fmt != STRINGS);
1425 122093
        *e = NULL;
1426 122093
        t1 = tl->t;
1427 122093
        if (fmt->stringform)
1428 41026
                vcc_expr_cor(tl, e, STRINGS);
1429
        else
1430 81067
                vcc_expr_cor(tl, e, fmt);
1431 122093
        ERRCHK(tl);
1432
1433 122011
        if ((*e)->fmt == fmt)
1434 63610
                return;
1435
1436 58401
        if ((*e)->fmt != STRINGS && fmt->stringform)
1437 3543
                vcc_expr_tostring(tl, e);
1438
1439 58401
        if ((*e)->fmt->stringform) {
1440 0
                VSB_printf(tl->sb, "Cannot convert type %s(%s) to %s(%s)\n",
1441 0
                    vcc_utype((*e)->fmt), (*e)->fmt->name,
1442 0
                    vcc_utype(fmt), fmt->name);
1443 0
                vcc_ErrWhere2(tl, t1, tl->t);
1444 0
                return;
1445
        }
1446
1447 58401
        if (fmt == BODY && !(*e)->fmt->bodyform)
1448 1
                vcc_expr_tostring(tl, e);
1449
1450 58401
        if (fmt == BODY && (*e)->fmt->bodyform) {
1451 5786
                if ((*e)->fmt == STRINGS)
1452 5782
                        *e = vcc_expr_edit(tl, BODY, "STRING, 0, \vT", *e, NULL);
1453 4
                else if ((*e)->fmt == BLOB)
1454 4
                        *e = vcc_expr_edit(tl, BODY, "BLOB, 0, \v1", *e, NULL);
1455
                else
1456 0
                        WRONG("Unhandled bodyform");
1457 5786
        }
1458
1459 58401
        if ((*e)->fmt == STRINGS && fmt->stringform) {
1460 40993
                if (fmt == STRING)
1461 12303
                        *e = vcc_expr_edit(tl, STRING, "\vS", *e, NULL);
1462 28690
                else if (fmt == STRANDS)
1463 28690
                        *e = vcc_expr_edit(tl, STRANDS, "\vT", (*e), NULL);
1464
                else
1465 0
                        WRONG("Unhandled stringform");
1466 40993
        }
1467
1468 58401
        if (fmt == BOOL) {
1469 11618
                vcc_expr_tobool(tl, e);
1470 11618
                ERRCHK(tl);
1471 11618
        }
1472
1473 58401
        vcc_expr_typecheck(tl, e, fmt, t1);
1474 122093
}
1475
1476
static void
1477 67175
vcc_expr_typecheck(struct vcc *tl, struct expr **e, vcc_type_t fmt,
1478
    struct token *t1)
1479
{
1480
1481 67175
        assert(fmt != VOID);
1482 67175
        assert(fmt != STRINGS);
1483
1484 67175
        if (fmt != (*e)->fmt)  {
1485 18
                VSB_printf(tl->sb, "Expression has type %s, expected %s\n",
1486 9
                    vcc_utype((*e)->fmt), vcc_utype(fmt));
1487 9
                vcc_ErrWhere2(tl, t1, tl->t);
1488 9
        }
1489 67175
}
1490
1491
/*--------------------------------------------------------------------
1492
 * This function parses and emits the C-code to evaluate an expression
1493
 *
1494
 * We know up front what kind of type we want the expression to be,
1495
 * and this function is the backstop if that doesn't succeed.
1496
 */
1497
1498
void
1499 119322
vcc_Expr(struct vcc *tl, vcc_type_t fmt)
1500
{
1501 119322
        struct expr *e = NULL;
1502
1503 119322
        assert(fmt != VOID);
1504 119322
        assert(fmt != STRINGS);
1505 119322
        vcc_expr0(tl, &e, fmt);
1506 119322
        ERRCHK(tl);
1507 119244
        assert(e->fmt == fmt);
1508
1509 119244
        vcc_expr_fmt(tl->fb, tl->indent, e);
1510 119244
        VSB_cat(tl->fb, "\n");
1511 119244
        vcc_delete_expr(e);
1512 119322
}
1513
1514
/*--------------------------------------------------------------------
1515
 */
1516
1517
void v_matchproto_(sym_act_f)
1518 489
vcc_Act_Call(struct vcc *tl, struct token *t, struct symbol *sym)
1519
{
1520
1521
        struct expr *e;
1522
1523 489
        e = NULL;
1524 489
        vcc_func(tl, &e, sym->eval_priv, sym->extra, sym);
1525 489
        if (!tl->err) {
1526 484
                vcc_expr_fmt(tl->fb, tl->indent, e);
1527 484
                SkipToken(tl, ';');
1528 484
                VSB_cat(tl->fb, ";\n");
1529 489
        } else if (t != tl->t) {
1530 5
                VSB_cat(tl->sb, "While compiling function call:\n\n");
1531 5
                vcc_ErrWhere2(tl, t, tl->t);
1532 5
        }
1533 489
        vcc_delete_expr(e);
1534 489
}
1535
1536
void v_matchproto_(sym_act_f)
1537 334
vcc_Act_Obj(struct vcc *tl, struct token *t, struct symbol *sym)
1538
{
1539
1540 334
        struct expr *e = NULL;
1541
1542 334
        assert(sym->kind == SYM_INSTANCE);
1543 334
        ExpectErr(tl, '.');
1544 333
        tl->t = t;
1545 333
        vcc_expr4(tl, &e, sym->type);
1546 333
        ERRCHK(tl);
1547 333
        vcc_expr_fmt(tl->fb, tl->indent, e);
1548 333
        vcc_delete_expr(e);
1549 333
        SkipToken(tl, ';');
1550 333
        VSB_cat(tl->fb, ";\n");
1551 334
}
1552
1553
/*--------------------------------------------------------------------
1554
 */
1555
1556
static void v_matchproto_(sym_expr_t)
1557 38
vcc_Eval_Regsub(struct vcc *tl, struct expr **e, struct token *t,
1558
    struct symbol *sym, vcc_type_t fmt)
1559
{
1560
        struct expr *e2, *e3;
1561 38
        int all = sym->eval_priv == NULL ? 0 : 1;
1562
        char buf[128];
1563
1564 38
        (void)t;
1565 38
        (void)fmt;
1566 38
        SkipToken(tl, '(');
1567 38
        vcc_expr0(tl, &e2, STRING);
1568 38
        ERRCHK(tl);
1569 38
        SkipToken(tl, ',');
1570 38
        vcc_expr0(tl, &e3, REGEX);
1571 38
        ERRCHK(tl);
1572
1573 37
        bprintf(buf, "VRT_regsub(ctx, %d,\v+\n\v1,\n\v2", all);
1574 37
        *e = vcc_expr_edit(tl, STRING, buf, e2, e3);
1575 37
        SkipToken(tl, ',');
1576 37
        vcc_expr0(tl, &e2, STRING);
1577 37
        ERRCHK(tl);
1578 37
        *e = vcc_expr_edit(tl, STRINGS, "\v1,\n\v2)\v-", *e, e2);
1579 37
        (*e)->nstr = 1;
1580 37
        SkipToken(tl, ')');
1581 38
}
1582
1583
/*--------------------------------------------------------------------
1584
 */
1585
1586
static void v_matchproto_(sym_expr_t)
1587 3130
vcc_Eval_BoolConst(struct vcc *tl, struct expr **e, struct token *t,
1588
    struct symbol *sym, vcc_type_t fmt)
1589
{
1590
1591 3130
        (void)t;
1592 3130
        (void)tl;
1593 3130
        (void)fmt;
1594 3130
        *e = vcc_mk_expr(BOOL, "(0==%d)", sym->eval_priv == NULL ? 1 : 0);
1595 3130
        (*e)->constant = EXPR_CONST;
1596 3130
}
1597
1598
/*--------------------------------------------------------------------
1599
 */
1600
1601
static void v_matchproto_(sym_expr_t)
1602 4
vcc_Eval_Default(struct vcc *tl, struct expr **e, struct token *t,
1603
    struct symbol *sym, vcc_type_t fmt)
1604
{
1605 4
        (void)e;
1606 4
        (void)fmt;
1607 4
        (void)sym;
1608 4
        (void)t;
1609
1610 4
        if (fmt->default_sym == NULL) {
1611 2
                VSB_cat(tl->sb, "Symbol 'default' is a reserved word.\n");
1612 2
                vcc_ErrWhere(tl, t);
1613 2
                return;
1614
        }
1615
1616 2
        *e = vcc_mk_expr(fmt, "%s", fmt->default_sym->rname);
1617 4
}
1618
1619
/*--------------------------------------------------------------------
1620
 */
1621
1622
void
1623 3119
vcc_Expr_Init(struct vcc *tl)
1624
{
1625
        struct symbol *sym;
1626
1627 3119
        sym = VCC_MkSym(tl, "regsub", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1628 3119
        AN(sym);
1629 3119
        sym->type = STRING;
1630 3119
        sym->eval = vcc_Eval_Regsub;
1631 3119
        sym->eval_priv = NULL;
1632
1633 3119
        sym = VCC_MkSym(tl, "regsuball", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1634 3119
        AN(sym);
1635 3119
        sym->type = STRING;
1636 3119
        sym->eval = vcc_Eval_Regsub;
1637 3119
        sym->eval_priv = sym;
1638
1639 3119
        sym = VCC_MkSym(tl, "true", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1640 3119
        AN(sym);
1641 3119
        sym->type = BOOL;
1642 3119
        sym->eval = vcc_Eval_BoolConst;
1643 3119
        sym->eval_priv = sym;
1644
1645 3119
        sym = VCC_MkSym(tl, "false", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1646 3119
        AN(sym);
1647 3119
        sym->type = BOOL;
1648 3119
        sym->eval = vcc_Eval_BoolConst;
1649 3119
        sym->eval_priv = NULL;
1650
1651 3119
        sym = VCC_MkSym(tl, "default", SYM_MAIN, SYM_FUNC, VCL_LOW, VCL_HIGH);
1652 3119
        AN(sym);
1653 3119
        sym->type = DEFAULT;
1654 3119
        sym->eval = vcc_Eval_Default;
1655 3119
}