varnish-cache/lib/libvcc/vcc_acl.c
0
/*-
1
 * Copyright (c) 2006 Verdens Gang AS
2
 * Copyright (c) 2006-2010 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
#include "config.h"
32
33
#include <sys/socket.h>
34
35
#include <netinet/in.h>
36
37
#include <netdb.h>
38
#include <stdlib.h>
39
#include <string.h>
40
41
#include "vcc_compile.h"
42
#include <vtcp.h>
43
#include <vtree.h>
44
#include <vsa.h>
45
46
#define ACL_MAXADDR     (sizeof(struct in6_addr) + 1)
47
48
VRBT_HEAD(acl_tree, acl_e);
49
50
struct acl {
51
        unsigned                magic;
52
#define VCC_ACL_MAGIC           0xb9fb3cd0
53
54
        int                     flag_log;
55
        int                     flag_fold;
56
        int                     flag_pedantic;
57
        int                     flag_table;
58
59
        struct acl_tree         acl_tree;
60
};
61
62
struct acl_e {
63
        unsigned                magic;
64
#define VCC_ACL_E_MAGIC 0xcac81e23
65
        VRBT_ENTRY(acl_e)       branch;
66
        unsigned char           data[ACL_MAXADDR];
67
        unsigned                mask;
68
        unsigned                not;
69
        unsigned                para;
70
        unsigned                overlapped;
71
        char                    *addr;
72
        const char              *fixed;
73
        struct token            *t_addr;
74
        struct token            *t_mask;
75
};
76
77
enum acl_cmp_e {
78
        ACL_EQ = 0,
79
        ACL_LT = -1,            // a < b
80
        ACL_GT = 1,             // b > a
81
        ACL_CONTAINED = -2,     // b contains a
82
        ACL_CONTAINS = 2,       // a contains b
83
        ACL_LEFT = -3,          // a + 1 == b
84
        ACL_RIGHT = 3           // a == b + 1
85
};
86
87
static void vcc_acl_insert_entry(struct vcc *, struct acl_e **);
88
89
/*
90
 * Compare two acl rules for relation
91
 */
92
93
#define CMP(n, a, b)                                                    \
94
        do {                                                            \
95
                if ((a) < (b))                                          \
96
                        return (enum acl_cmp_e)(-n);                    \
97
                else if ((b) < (a))                                     \
98
                        return (n);                                     \
99
        } while (0)
100
101
#define CMPA(a, b)                                                      \
102
        do {                                                            \
103
                if (((a) | 1) == (b))                                   \
104
                        return (ACL_LEFT);                              \
105
                else if (((b) | 1) == (a))                              \
106
                        return (ACL_RIGHT);                             \
107
        } while (0)
108
109
static void
110 150
vcl_acl_free(struct acl_e **aep)
111
{
112
        struct acl_e *a;
113
114 150
        TAKE_OBJ_NOTNULL(a, aep, VCC_ACL_E_MAGIC);
115 150
        free(a->addr);
116 150
        FREE_OBJ(a);
117 150
}
118
119
static enum acl_cmp_e
120 12325
vcl_acl_cmp(const struct acl_e *ae1, const struct acl_e *ae2)
121
{
122
        const unsigned char *p1, *p2;
123
        unsigned m;
124
        unsigned char a1, a2;
125
126 12325
        CHECK_OBJ_NOTNULL(ae1, VCC_ACL_E_MAGIC);
127 12325
        CHECK_OBJ_NOTNULL(ae2, VCC_ACL_E_MAGIC);
128
129 12325
        p1 = ae1->data;
130 12325
        p2 = ae2->data;
131 12325
        m = vmin_t(unsigned, ae1->mask, ae2->mask);
132 64275
        for (; m >= 8; m -= 8) {
133 58725
                if (m == 8 && ae1->mask == ae2->mask)
134 975
                        CMPA(*p1, *p2);
135 110450
                CMP(ACL_GT, *p1, *p2);
136 51950
                p1++;
137 51950
                p2++;
138 51950
        }
139 5550
        if (m) {
140 5300
                assert (m < 8);
141 5300
                a1 = *p1 >> (8 - m);
142 5300
                a2 = *p2 >> (8 - m);
143 5300
                if (ae1->mask == ae2->mask)
144 1700
                        CMPA(a1, a2);
145 7100
                CMP(ACL_GT, a1, a2);
146 2450
        } else if (ae1->mask == ae2->mask) {
147 250
                CMPA(*p1, *p2);
148 125
        }
149
        /* Long mask is less than short mask */
150 2600
        CMP(ACL_CONTAINS, ae2->mask, ae1->mask);
151
152 150
        return (ACL_EQ);
153 12325
}
154
155
static int
156 2875
vcl_acl_disjoint(const struct acl_e *ae1, const struct acl_e *ae2)
157
{
158
        const unsigned char *p1, *p2;
159
        unsigned m;
160
161 2875
        CHECK_OBJ_NOTNULL(ae1, VCC_ACL_E_MAGIC);
162 2875
        CHECK_OBJ_NOTNULL(ae2, VCC_ACL_E_MAGIC);
163
164 2875
        p1 = ae1->data;
165 2875
        p2 = ae2->data;
166 2875
        m = vmin_t(unsigned, ae1->mask, ae2->mask);
167 22050
        for (; m >= 8; m -= 8) {
168 39050
                CMP(ACL_GT, *p1, *p2);
169 19175
                p1++;
170 19175
                p2++;
171 19175
        }
172 2175
        if (m) {
173 2000
                m = 0xff00 >> m;
174 2000
                m &= 0xff;
175 3775
                CMP(ACL_GT, *p1 & m, *p2 & m);
176 1775
        }
177 1950
        return (0);
178 2875
}
179
180 8775
VRBT_GENERATE_INSERT_COLOR(acl_tree, acl_e, branch, static)
181 4225
VRBT_GENERATE_INSERT_FINISH(acl_tree, acl_e, branch, static)
182 15300
VRBT_GENERATE_INSERT(acl_tree, acl_e, branch, vcl_acl_cmp, static)
183 725
VRBT_GENERATE_REMOVE_COLOR(acl_tree, acl_e, branch, static)
184 1225
VRBT_GENERATE_REMOVE(acl_tree, acl_e, branch, static)
185 4150
VRBT_GENERATE_MINMAX(acl_tree, acl_e, branch, static)
186 11050
VRBT_GENERATE_NEXT(acl_tree, acl_e, branch, static)
187 6875
VRBT_GENERATE_PREV(acl_tree, acl_e, branch, static)
188
189
static char *
190 4150
vcc_acl_chk(struct vcc *tl, const struct acl_e *ae, const int l,
191
    unsigned char *p, int fam)
192
{
193
        const unsigned char *u;
194
        char h[VTCP_ADDRBUFSIZE];
195
        char t[VTCP_ADDRBUFSIZE + 10];
196 4150
        char s[vsa_suckaddr_len];
197 4150
        char *r = NULL;
198
        const struct suckaddr *sa;
199
        unsigned m;
200 4150
        int ll, ret = 0;
201
202 4150
        u = p;
203 4150
        ll = l;
204 4150
        m = ae->mask;
205
206 4150
        p += m / 8;
207 4150
        ll -= m / 8;
208 4150
        assert (ll >= 0);
209 4150
        m %= 8;
210
211 4150
        if (m && ((unsigned)*p << m & 0xff) != 0) {
212 100
                ret = 1;
213 100
                m = 0xff00 >> m;
214 100
                *p &= m;
215 100
        }
216 4150
        if (m) {
217 2375
                p++;
218 2375
                ll--;
219 2375
        }
220
221 7850
        for ( ; ll > 0; p++, ll--) {
222 3700
                if (*p == 0)
223 3575
                        continue;
224 125
                ret = 1;
225 125
                *p = 0;
226 125
        }
227 4150
        if (ret == 0)
228 3925
                return (NULL);
229
230 225
        sa = VSA_BuildFAP(s, fam, u, l, NULL, 0);
231 225
        AN(sa);
232 225
        VTCP_name(sa, h, sizeof h, NULL, 0);
233 225
        bprintf(t, "%s/%d", h, ae->mask);
234 225
        if (tl->acl->flag_pedantic != 0) {
235 50
                VSB_cat(tl->sb, "Non-zero bits in masked part, ");
236 50
                VSB_printf(tl->sb, "(maybe use %s ?)\n", t);
237 50
                vcc_ErrWhere(tl, ae->t_addr);
238 50
        }
239 225
        REPLACE(r, t);
240 225
        return (r);
241 4150
}
242
243
static void
244 1100
vcl_acl_fold(struct vcc *tl, struct acl_e **l, struct acl_e **r)
245
{
246
        enum acl_cmp_e cmp;
247
248 1100
        AN(l);
249 1100
        AN(r);
250 1100
        CHECK_OBJ_NOTNULL(*l, VCC_ACL_E_MAGIC);
251 1100
        CHECK_OBJ_NOTNULL(*r, VCC_ACL_E_MAGIC);
252
253 1100
        if ((*l)->not || (*r)->not)
254 0
                return;
255
256 1100
        cmp = vcl_acl_cmp(*l, *r);
257
258 1100
        assert(cmp < 0);
259 1100
        if (cmp == ACL_LT)
260 725
                return;
261
262 375
        do {
263 400
                switch (cmp) {
264
                case ACL_CONTAINED:
265 175
                        VSB_cat(tl->sb, "ACL entry:\n");
266 175
                        vcc_ErrWhere(tl, (*r)->t_addr);
267 175
                        VSB_cat(tl->sb, "supersedes / removes:\n");
268 175
                        vcc_ErrWhere(tl, (*l)->t_addr);
269 175
                        vcc_Warn(tl);
270 175
                        VRBT_REMOVE(acl_tree, &tl->acl->acl_tree, *l);
271 175
                        FREE_OBJ(*l);
272 175
                        *l = VRBT_PREV(acl_tree, &tl->acl->acl_tree, *r);
273 175
                        break;
274
                case ACL_LEFT:
275 225
                        (*l)->mask--;
276 225
                        (*l)->fixed = "folded";
277 225
                        VSB_cat(tl->sb, "ACL entry:\n");
278 225
                        vcc_ErrWhere(tl, (*l)->t_addr);
279 225
                        VSB_cat(tl->sb, "left of:\n");
280 225
                        vcc_ErrWhere(tl, (*r)->t_addr);
281 450
                        VSB_printf(tl->sb, "removing the latter and expanding "
282
                            "mask of the former by one to /%u\n",
283 225
                            (*l)->mask - 8);
284 225
                        vcc_Warn(tl);
285 225
                        VRBT_REMOVE(acl_tree, &tl->acl->acl_tree, *r);
286 225
                        FREE_OBJ(*r);
287 225
                        VRBT_REMOVE(acl_tree, &tl->acl->acl_tree, *l);
288 225
                        vcc_acl_insert_entry(tl, l);
289 225
                        return;
290
                default:
291 0
                        INCOMPL();
292 0
                }
293 175
                if (*l == NULL || *r == NULL)
294 25
                        break;
295 150
                cmp = vcl_acl_cmp(*l, *r);
296 150
        } while (cmp != ACL_LT);
297 1100
}
298
299
static void
300 4375
vcc_acl_insert_entry(struct vcc *tl, struct acl_e **aenp)
301
{
302
        struct acl_e *ae2, *l, *r;
303
304 4375
        CHECK_OBJ_NOTNULL(*aenp, VCC_ACL_E_MAGIC);
305 4375
        ae2 = VRBT_INSERT(acl_tree, &tl->acl->acl_tree, *aenp);
306 4375
        if (ae2 != NULL) {
307 150
                if (ae2->not != (*aenp)->not) {
308 25
                        VSB_cat(tl->sb, "Conflicting ACL entries:\n");
309 25
                        vcc_ErrWhere(tl, ae2->t_addr);
310 25
                        VSB_cat(tl->sb, "vs:\n");
311 25
                        vcc_ErrWhere(tl, (*aenp)->t_addr);
312 25
                }
313 150
                return;
314
        }
315
316 4225
        r = *aenp;
317 4225
        *aenp = NULL;
318
319 4225
        if (tl->acl->flag_fold == 0)
320 3300
                return;
321
322 925
        l = VRBT_PREV(acl_tree, &tl->acl->acl_tree, r);
323 925
        if (l != NULL) {
324 875
                vcl_acl_fold(tl, &l, &r);
325 875
        }
326 925
        if (r == NULL)
327 200
                return;
328 725
        l = r;
329 725
        r = VRBT_NEXT(acl_tree, &tl->acl->acl_tree, l);
330 725
        if (r == NULL)
331 500
                return;
332 225
        vcl_acl_fold(tl, &l, &r);
333 4375
}
334
335
static void
336 4200
vcc_acl_add_entry(struct vcc *tl, const struct acl_e *ae, int l,
337
    unsigned char *u, int fam)
338
{
339
        struct acl_e *aen;
340
341 4200
        if (fam == PF_INET && ae->mask > 32) {
342 50
                VSB_printf(tl->sb,
343 25
                    "Too wide mask (/%u) for IPv4 address\n", ae->mask);
344 25
                if (ae->t_mask != NULL)
345 25
                        vcc_ErrWhere(tl, ae->t_mask);
346
                else
347 0
                        vcc_ErrWhere(tl, ae->t_addr);
348 25
                return;
349
        }
350 4175
        if (fam == PF_INET6 && ae->mask > 128) {
351 50
                VSB_printf(tl->sb,
352 25
                    "Too wide mask (/%u) for IPv6 address\n", ae->mask);
353 25
                vcc_ErrWhere(tl, ae->t_mask);
354 25
                return;
355
        }
356
357
        /* Make a copy from the template */
358 4150
        ALLOC_OBJ(aen, VCC_ACL_E_MAGIC);
359 4150
        AN(aen);
360 4150
        *aen = *ae;
361 4150
        aen->addr = strdup(ae->addr);
362 4150
        AN(aen->addr);
363
364 4150
        aen->fixed = vcc_acl_chk(tl, ae, l, u, fam);
365
366
        /* We treat family as part of address, it saves code */
367 4150
        assert(fam <= 0xff);
368 4150
        aen->data[0] = fam & 0xff;
369 4150
        aen->mask += 8;
370
371 4150
        assert(l + 1UL <= sizeof aen->data);
372 4150
        memcpy(aen->data + 1L, u, l);
373
374 4150
        vcc_acl_insert_entry(tl, &aen);
375 4150
        if (aen != NULL)
376 150
                vcl_acl_free(&aen);
377 4200
}
378
379
static void
380 1375
vcc_acl_try_getaddrinfo(struct vcc *tl, struct acl_e *ae)
381
{
382
        struct addrinfo *res0, *res, hint;
383
        struct sockaddr_in *sin4;
384
        struct sockaddr_in6 *sin6;
385
        unsigned char *u, i4, i6;
386
        int error;
387
388 1375
        CHECK_OBJ_NOTNULL(ae, VCC_ACL_E_MAGIC);
389 1375
        memset(&hint, 0, sizeof hint);
390 1375
        hint.ai_family = PF_UNSPEC;
391 1375
        hint.ai_socktype = SOCK_STREAM;
392 1375
        error = getaddrinfo(ae->addr, "0", &hint, &res0);
393 1375
        if (error) {
394 100
                if (ae->para) {
395 100
                        VSB_printf(tl->sb,
396
                            "Warning: %s ignored\n  -- %s\n",
397 50
                            ae->addr, gai_strerror(error));
398 100
                        Fh(tl, 1, "/* Ignored ACL entry: %s%s",
399 50
                            ae->para ? "\"(\" " : "", ae->not ? "\"!\" " : "");
400 50
                        EncToken(tl->fh, ae->t_addr);
401 50
                        if (ae->t_mask)
402 25
                                Fh(tl, 0, "/%u", ae->mask);
403 50
                        Fh(tl, 0, "%s\n", ae->para ? " \")\"" : "");
404 100
                        Fh(tl, 1, " * getaddrinfo:  %s */\n",
405 50
                             gai_strerror(error));
406 50
                } else {
407 100
                        VSB_printf(tl->sb,
408
                            "DNS lookup(%s): %s\n",
409 50
                            ae->addr, gai_strerror(error));
410 50
                        vcc_ErrWhere(tl, ae->t_addr);
411
                }
412 100
                return;
413
        }
414
415 1275
        i4 = i6 = 0;
416 2600
        for (res = res0; res != NULL; res = res->ai_next) {
417 1325
                switch (res->ai_family) {
418
                case PF_INET:
419 50
                        i4++;
420 50
                        break;
421
                case PF_INET6:
422 1275
                        i6++;
423 1275
                        break;
424
                default:
425 0
                        VSB_printf(tl->sb,
426
                            "Ignoring unknown protocol family (%d) for %.*s\n",
427 0
                                res->ai_family, PF(ae->t_addr));
428 0
                        continue;
429
                }
430 1325
        }
431
432 1275
        if (ae->t_mask != NULL && i4 > 0 && i6 > 0) {
433 0
                VSB_printf(tl->sb,
434
                    "Mask (/%u) specified, but string resolves to"
435 0
                    " both IPv4 and IPv6 addresses.\n", ae->mask);
436 0
                vcc_ErrWhere(tl, ae->t_mask);
437 0
                freeaddrinfo(res0);
438 0
                return;
439
        }
440
441 2550
        for (res = res0; res != NULL; res = res->ai_next) {
442 1325
                switch (res->ai_family) {
443
                case PF_INET:
444 50
                        assert(PF_INET < 256);
445 50
                        sin4 = (void*)res->ai_addr;
446 50
                        assert(sizeof(sin4->sin_addr) == 4);
447 50
                        u = (void*)&sin4->sin_addr;
448 50
                        if (ae->t_mask == NULL)
449 50
                                ae->mask = 32;
450 50
                        vcc_acl_add_entry(tl, ae, 4, u, res->ai_family);
451 50
                        break;
452
                case PF_INET6:
453 1275
                        assert(PF_INET6 < 256);
454 1275
                        sin6 = (void*)res->ai_addr;
455 1275
                        assert(sizeof(sin6->sin6_addr) == 16);
456 1275
                        u = (void*)&sin6->sin6_addr;
457 1275
                        if (ae->t_mask == NULL)
458 150
                                ae->mask = 128;
459 1275
                        vcc_acl_add_entry(tl, ae, 16, u, res->ai_family);
460 1275
                        break;
461
                default:
462 0
                        continue;
463
                }
464 1325
                if (tl->err)
465 50
                        freeaddrinfo(res0);
466 1325
                ERRCHK(tl);
467 1275
        }
468 1225
        freeaddrinfo(res0);
469
470 1375
}
471
472
/*--------------------------------------------------------------------
473
 * Ancient stupidity on the part of X/Open and other standards orgs
474
 * dictate that "192.168" be translated to 192.0.0.168.  Ever since
475
 * CIDR happened, "192.168/16" notation has been used, but apparently
476
 * no API supports parsing this, so roll our own.
477
 */
478
479
static int
480 4250
vcc_acl_try_netnotation(struct vcc *tl, struct acl_e *ae)
481
{
482
        unsigned char b[4];
483
        int i, j, k;
484
        unsigned u;
485
        const char *p;
486
487 4250
        CHECK_OBJ_NOTNULL(ae, VCC_ACL_E_MAGIC);
488 4250
        memset(b, 0, sizeof b);
489 4250
        p = ae->addr;
490 12400
        for (i = 0; i < 4; i++) {
491 12400
                j = sscanf(p, "%u%n", &u, &k);
492 12400
                if (j != 1)
493 1300
                        return (0);
494 11100
                if (u & ~0xff)
495 25
                        return (0);
496 11075
                b[i] = (unsigned char)u;
497 11075
                if (p[k] == '\0')
498 2875
                        break;
499 8200
                if (p[k] != '.')
500 50
                        return (0);
501 8150
                p += k + 1;
502 8150
        }
503 2875
        if (ae->t_mask == NULL)
504 650
                ae->mask = 8 + 8 * i;
505 2875
        vcc_acl_add_entry(tl, ae, 4, b, AF_INET);
506 2875
        return (1);
507 4250
}
508
509
static void
510 4350
vcc_acl_entry(struct vcc *tl)
511
{
512
        struct acl_e ae[1];
513
        char *sl, *e;
514
515 4350
        INIT_OBJ(ae, VCC_ACL_E_MAGIC);
516
517 4350
        if (tl->t->tok == '!') {
518 1000
                ae->not = 1;
519 1000
                vcc_NextToken(tl);
520 1000
        }
521
522 4350
        if (tl->t->tok == '(') {
523 75
                ae->para = 1;
524 75
                vcc_NextToken(tl);
525 75
        }
526
527 4350
        if (!ae->not && tl->t->tok == '!') {
528 25
                ae->not = 1;
529 25
                vcc_NextToken(tl);
530 25
        }
531
532 4350
        ExpectErr(tl, CSTR);
533 4350
        ae->t_addr = tl->t;
534 4350
        ae->addr = ae->t_addr->dec;
535 4350
        vcc_NextToken(tl);
536
537 4350
        if (strchr(ae->t_addr->dec, '/') != NULL) {
538 75
                sl = strchr(ae->addr, '/');
539 75
                AN(sl);
540 75
                *sl++ = '\0';
541 75
                e = NULL;
542 75
                ae->mask = strtoul(sl, &e, 10);
543 75
                if (*e != '\0') {
544 50
                        VSB_cat(tl->sb, ".../mask is not numeric.\n");
545 50
                        vcc_ErrWhere(tl, ae->t_addr);
546 50
                        return;
547
                }
548 25
                ae->t_mask = ae->t_addr;
549 25
                if (tl->t->tok == '/') {
550 25
                        VSB_cat(tl->sb, "/mask only allowed once.\n");
551 25
                        vcc_ErrWhere(tl, tl->t);
552 25
                        return;
553
                }
554 4275
        } else if (tl->t->tok == '/') {
555 3375
                vcc_NextToken(tl);
556 3375
                ae->t_mask = tl->t;
557 3375
                ExpectErr(tl, CNUM);
558 3375
                ae->mask = vcc_UintVal(tl);
559 3375
        }
560
561 4275
        if (ae->para)
562 75
                SkipToken(tl, ')');
563
564 4250
        if (!vcc_acl_try_netnotation(tl, ae)) {
565 1375
                ERRCHK(tl);
566 1375
                vcc_acl_try_getaddrinfo(tl, ae);
567 1375
        }
568 4250
        ERRCHK(tl);
569 4350
}
570
571
/*********************************************************************
572
 * Emit the tokens making up an entry as C-strings
573
 */
574
575
static void
576 1275
vcc_acl_emit_tokens(const struct vcc *tl, const struct acl_e *ae)
577
{
578
        struct token *t;
579 1275
        const char *sep = "";
580
581 1275
        CHECK_OBJ_NOTNULL(ae, VCC_ACL_E_MAGIC);
582 1275
        t = ae->t_addr;
583 1275
        do {
584 3725
                if (t->tok == CSTR) {
585 1275
                        Fh(tl, 0, "%s\"\\\"\" ", sep);
586 1275
                        EncToken(tl->fh, t);
587 1275
                        Fh(tl, 0, " \"\\\"\"");
588 3725
                } else if (t == ae->t_mask) {
589 1225
                        Fh(tl, 0, " \"%u\"", ae->mask - 8);
590 1225
                } else {
591 1225
                        Fh(tl, 0, "%s\"%.*s\"", sep, PF(t));
592
                }
593 3725
                if (t == ae->t_mask)
594 1225
                        break;
595 2500
                t = vcc_PeekTokenFrom(tl, t);
596 2500
                AN(t);
597 2500
                sep = " ";
598 2500
        } while (ae->t_mask != NULL);
599 1275
        if (ae->fixed)
600 250
                Fh(tl, 0, "\" fixed: %s\"", ae->fixed);
601 1275
}
602
603
/*********************************************************************
604
 * Emit ACL on table format
605
 */
606
607
static unsigned
608 50
vcc_acl_emit_tables(const struct vcc *tl, unsigned n, const char *name)
609
{
610
        struct acl_e *ae;
611 50
        unsigned rv = sizeof(ae->data) + 3;
612 50
        unsigned nn = 0;
613
        size_t sz;
614
615 100
        Fh(tl, 0, "\nstatic unsigned char acl_tbl_%s[%u*%u] = {\n",
616 50
            name, n, rv);
617 1250
        VRBT_FOREACH(ae, acl_tree, &tl->acl->acl_tree) {
618 1200
                if (ae->overlapped)
619 1000
                        continue;
620 200
                Fh(tl, 0, "\t0x%02x,", ae->not ? 0 : 1);
621 200
                Fh(tl, 0, "0x%02x,", (ae->mask >> 3) - 1);
622 200
                Fh(tl, 0, "0x%02x,", (0xff00 >> (ae->mask & 7)) & 0xff);
623 3600
                for (sz = 0; sz < sizeof(ae->data); sz++)
624 3400
                        Fh(tl, 0, "0x%02x,", ae->data[sz]);
625 200
                for (; sz < rv - 3; sz++)
626 0
                        Fh(tl, 0, "0,");
627 200
                Fh(tl, 0, "\n");
628 200
                nn++;
629 200
        }
630 50
        assert(n == nn);
631 50
        Fh(tl, 0, "};\n");
632 50
        if (tl->acl->flag_log) {
633 50
                Fh(tl, 0, "\nstatic const char *acl_str_%s[%d] = {\n",
634 25
                    name, n);
635 625
                VRBT_FOREACH(ae, acl_tree, &tl->acl->acl_tree) {
636 600
                        if (ae->overlapped)
637 500
                                continue;
638 100
                        Fh(tl, 0, "\t");
639 200
                        Fh(tl, 0, "\"%sMATCH %s \" ",
640 100
                            ae->not ? "NEG_" : "", name);
641 100
                        vcc_acl_emit_tokens(tl, ae);
642 100
                        Fh(tl, 0, ",\n");
643 100
                }
644 25
                Fh(tl, 0, "};\n");
645 25
        }
646 50
        return (rv);
647
}
648
649
/*********************************************************************
650
 * Emit a function to match the ACL we have collected
651
 */
652
653
static void
654 650
vcc_acl_emit(struct vcc *tl, const struct symbol *sym)
655
{
656
        struct acl_e *ae, *ae2;
657
        int depth, l, m, i;
658
        unsigned at[ACL_MAXADDR];
659 650
        struct inifin *ifp = NULL;
660
        struct vsb *func;
661 650
        unsigned n, no, nw = 0;
662
663 650
        func = VSB_new_auto();
664 650
        AN(func);
665 650
        VSB_cat(func, "match_acl_");
666 650
        VCC_PrintCName(func, sym->name, NULL);
667 650
        AZ(VSB_finish(func));
668
669 650
        depth = -1;
670 650
        at[0] = 256;
671 650
        ae2 = NULL;
672 650
        n = no = 0;
673 4150
        VRBT_FOREACH_REVERSE(ae, acl_tree, &tl->acl->acl_tree) {
674 3500
                n++;
675 3500
                if (ae2 == NULL) {
676 625
                        ae2 = ae;
677 3500
                } else if (vcl_acl_disjoint(ae, ae2)) {
678 925
                        ae2 = ae;
679 925
                } else {
680 1950
                        no++;
681 1950
                        ae->overlapped = 1;
682
                }
683 3500
        }
684
685 650
        Fh(tl, 0, "/* acl_n_%s n %u no %u */\n", sym->name, n, no);
686 650
        if (n - no < (1<<1))
687 375
                no = n;
688 275
        else if (!tl->acl->flag_table)
689 225
                no = n;
690
691 650
        if (no < n)
692 50
                nw = vcc_acl_emit_tables(tl, n - no, sym->name);
693
694
695 650
        Fh(tl, 0, "\nstatic int v_matchproto_(acl_match_f)\n");
696 650
        Fh(tl, 0, "%s(VRT_CTX, const VCL_IP p)\n", VSB_data(func));
697 650
        Fh(tl, 0, "{\n");
698 650
        Fh(tl, 0, "\tconst unsigned char *a;\n");
699 650
        Fh(tl, 0, "\tint fam;\n");
700 650
        Fh(tl, 0, "\n");
701 650
        Fh(tl, 0, "\tfam = VRT_VSA_GetPtr(ctx, p, &a);\n");
702 650
        Fh(tl, 0, "\tif (fam < 0) {\n");
703 650
        Fh(tl, 0, "\t\tVRT_fail(ctx,");
704 650
        Fh(tl, 0, " \"ACL %s: no protocol family\");\n", sym->name);
705 650
        Fh(tl, 0, "\t\treturn(0);\n");
706 650
        Fh(tl, 0, "\t}\n\n");
707 650
        if (!tl->err_unref) {
708 50
                ifp = New_IniFin(tl);
709 100
                VSB_printf(ifp->ini,
710 50
                        "\t(void)%s;\n", VSB_data(func));
711 50
        }
712
713 4150
        VRBT_FOREACH(ae, acl_tree, &tl->acl->acl_tree) {
714
715 3500
                if (no < n && !ae->overlapped)
716 200
                        continue;
717
718
                /* Find how much common prefix we have */
719 20600
                for (l = 0; l <= depth && l * 8 < (int)ae->mask - 7; l++) {
720 18450
                        assert(l >= 0);
721 18450
                        if (ae->data[l] != at[l])
722 1150
                                break;
723 17300
                }
724
725
                /* Back down, if necessary */
726 7500
                while (l <= depth) {
727 4200
                        Fh(tl, 0, "\t%*s}\n", -depth, "");
728 4200
                        depth--;
729
                }
730
731 3300
                m = (int)ae->mask;
732 3300
                assert(m >= l*8);
733 3300
                m -= l * 8;
734
735
                /* Do whole byte compares */
736 10850
                for (i = l; m >= 8; m -= 8, i++) {
737 7550
                        if (i == 0)
738 1750
                                Fh(tl, 0, "\t%*s%sif (fam == %d) {\n",
739 875
                                    -i, "", "", ae->data[i]);
740
                        else
741 13350
                                Fh(tl, 0, "\t%*s%sif (a[%d] == %d) {\n",
742 6675
                                    -i, "", "", i - 1, ae->data[i]);
743 7550
                        at[i] = ae->data[i];
744 7550
                        depth = i;
745 7550
                }
746
747 3300
                if (m > 0) {
748
                        // XXX can remove masking due to fixup
749
                        /* Do fractional byte compares */
750 3800
                        Fh(tl, 0, "\t%*s%sif ((a[%d] & 0x%x) == %d) {\n",
751 1900
                            -i, "", "", i - 1, (0xff00 >> m) & 0xff,
752 1900
                            ae->data[i] & ((0xff00 >> m) & 0xff));
753 1900
                        at[i] = 256;
754 1900
                        depth = i;
755 1900
                }
756
757 3300
                i = ((int)ae->mask + 7) / 8;
758
759 3300
                if (tl->acl->flag_log) {
760 2350
                        Fh(tl, 0, "\t%*sVPI_acl_log(ctx, \"%sMATCH %s \" ",
761 1175
                            -i, "", ae->not ? "NEG_" : "", sym->name);
762 1175
                        vcc_acl_emit_tokens(tl, ae);
763 1175
                        Fh(tl, 0, ");\n");
764 1175
                }
765
766 3300
                Fh(tl, 0, "\t%*sreturn (%d);\n", -i, "", ae->not ? 0 : 1);
767 3300
        }
768
769
        /* Unwind */
770 5900
        for (; 0 <= depth; depth--)
771 5250
                Fh(tl, 0, "\t%*.*s}\n", depth, depth, "");
772
773 650
        if (no < n) {
774 50
                Fh(tl, 0, "\treturn(\n\t    VPI_acl_table(ctx,\n");
775 50
                Fh(tl, 0, "\t\tp,\n");
776 50
                Fh(tl, 0, "\t\t%u, %u,\n", n - no, nw);
777 50
                Fh(tl, 0, "\t\tacl_tbl_%s,\n", sym->name);
778 50
                if (tl->acl->flag_log)
779 25
                        Fh(tl, 0, "\t\tacl_str_%s,\n", sym->name);
780
                else
781 25
                        Fh(tl, 0, "\t\tNULL,\n");
782 50
                Fh(tl, 0, "\t\t\"NO MATCH %s\"\n\t    )\n\t);\n", sym->name);
783 50
        } else {
784
                /* Deny by default */
785 600
                if (tl->acl->flag_log)
786 150
                        Fh(tl, 0, "\tVPI_acl_log(ctx, \"NO_MATCH %s\");\n",
787 75
                            sym->name);
788 600
                Fh(tl, 0, "\treturn(0);\n");
789
        }
790 650
        Fh(tl, 0, "}\n");
791
792
        /* Emit the struct that will be referenced */
793 650
        Fh(tl, 0, "\nstatic const struct vrt_acl %s[] = {{\n", sym->rname);
794 650
        Fh(tl, 0, "\t.magic = VRT_ACL_MAGIC,\n");
795 650
        Fh(tl, 0, "\t.match = &%s,\n", VSB_data(func));
796 650
        Fh(tl, 0, "\t.name = \"%s\",\n", sym->name);
797 650
        Fh(tl, 0, "}};\n\n");
798 650
        if (!tl->err_unref) {
799 50
                AN(ifp);
800 50
                VSB_printf(ifp->ini, "\t(void)%s;", sym->rname);
801 50
        }
802 650
        VSB_destroy(&func);
803 650
}
804
805
void
806 1025
vcc_ParseAcl(struct vcc *tl)
807
{
808
        struct symbol *sym;
809
        int sign;
810
        struct acl acl[1];
811
812 1025
        INIT_OBJ(acl, VCC_ACL_MAGIC);
813 1025
        tl->acl = acl;
814 1025
        acl->flag_pedantic = 1;
815 1025
        vcc_NextToken(tl);
816 1025
        VRBT_INIT(&acl->acl_tree);
817
818 1025
        vcc_ExpectVid(tl, "ACL");
819 1025
        ERRCHK(tl);
820 1000
        sym = VCC_HandleSymbol(tl, ACL);
821 1000
        ERRCHK(tl);
822 1000
        AN(sym);
823
824 1375
        while (1) {
825 1375
                sign = vcc_IsFlag(tl);
826 1375
                if (tl->err) {
827 25
                        VSB_cat(tl->sb,
828
                            "Valid ACL flags are `log` and `table`:\n");
829 25
                        return;
830
                }
831 1350
                if (sign < 0)
832 950
                        break;
833 400
                if (vcc_IdIs(tl->t, "log")) {
834 100
                        acl->flag_log = sign;
835 100
                        vcc_NextToken(tl);
836 400
                } else if (vcc_IdIs(tl->t, "fold")) {
837 25
                        acl->flag_fold = sign;
838 25
                        vcc_NextToken(tl);
839 300
                } else if (vcc_IdIs(tl->t, "pedantic")) {
840 200
                        acl->flag_pedantic = sign;
841 200
                        vcc_NextToken(tl);
842 275
                } else if (vcc_IdIs(tl->t, "table")) {
843 50
                        acl->flag_table = sign;
844 50
                        vcc_NextToken(tl);
845 50
                } else {
846 25
                        VSB_cat(tl->sb, "Unknown ACL flag:\n");
847 25
                        vcc_ErrWhere(tl, tl->t);
848 25
                        return;
849
                }
850
        }
851
852 950
        SkipToken(tl, '{');
853
854 5000
        while (tl->t->tok != '}') {
855 4350
                vcc_acl_entry(tl);
856 4350
                ERRCHK(tl);
857 4075
                SkipToken(tl, ';');
858
        }
859 650
        SkipToken(tl, '}');
860
861 650
        vcc_acl_emit(tl, sym);
862 1025
}