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 138
vcl_acl_free(struct acl_e **aep)
111
{
112
        struct acl_e *a;
113
114 138
        TAKE_OBJ_NOTNULL(a, aep, VCC_ACL_E_MAGIC);
115 138
        free(a->addr);
116 138
        FREE_OBJ(a);
117 138
}
118
119
static enum acl_cmp_e
120 11385
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 11385
        CHECK_OBJ_NOTNULL(ae1, VCC_ACL_E_MAGIC);
127 11385
        CHECK_OBJ_NOTNULL(ae2, VCC_ACL_E_MAGIC);
128
129 11385
        p1 = ae1->data;
130 11385
        p2 = ae2->data;
131 11385
        m = vmin_t(unsigned, ae1->mask, ae2->mask);
132 59179
        for (; m >= 8; m -= 8) {
133 54073
                if (m == 8 && ae1->mask == ae2->mask)
134 897
                        CMPA(*p1, *p2);
135 101660
                CMP(ACL_GT, *p1, *p2);
136 47794
                p1++;
137 47794
                p2++;
138 47794
        }
139 5106
        if (m) {
140 4876
                assert (m < 8);
141 4876
                a1 = *p1 >> (8 - m);
142 4876
                a2 = *p2 >> (8 - m);
143 4876
                if (ae1->mask == ae2->mask)
144 1564
                        CMPA(a1, a2);
145 6532
                CMP(ACL_GT, a1, a2);
146 2254
        } else if (ae1->mask == ae2->mask) {
147 230
                CMPA(*p1, *p2);
148 115
        }
149
        /* Long mask is less than short mask */
150 2392
        CMP(ACL_CONTAINS, ae2->mask, ae1->mask);
151
152 138
        return (ACL_EQ);
153 11385
}
154
155
static int
156 2691
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 2691
        CHECK_OBJ_NOTNULL(ae1, VCC_ACL_E_MAGIC);
162 2691
        CHECK_OBJ_NOTNULL(ae2, VCC_ACL_E_MAGIC);
163
164 2691
        p1 = ae1->data;
165 2691
        p2 = ae2->data;
166 2691
        m = vmin_t(unsigned, ae1->mask, ae2->mask);
167 20332
        for (; m >= 8; m -= 8) {
168 35972
                CMP(ACL_GT, *p1, *p2);
169 17641
                p1++;
170 17641
                p2++;
171 17641
        }
172 2001
        if (m) {
173 1840
                m = 0xff00 >> m;
174 1840
                m &= 0xff;
175 3473
                CMP(ACL_GT, *p1 & m, *p2 & m);
176 1633
        }
177 1794
        return (0);
178 2691
}
179
180 8119
VRBT_GENERATE_INSERT_COLOR(acl_tree, acl_e, branch, static)
181 3979
VRBT_GENERATE_INSERT_FINISH(acl_tree, acl_e, branch, static)
182 14214
VRBT_GENERATE_INSERT(acl_tree, acl_e, branch, vcl_acl_cmp, static)
183 667
VRBT_GENERATE_REMOVE_COLOR(acl_tree, acl_e, branch, static)
184 1127
VRBT_GENERATE_REMOVE(acl_tree, acl_e, branch, static)
185 4048
VRBT_GENERATE_MINMAX(acl_tree, acl_e, branch, static)
186 10304
VRBT_GENERATE_NEXT(acl_tree, acl_e, branch, static)
187 6417
VRBT_GENERATE_PREV(acl_tree, acl_e, branch, static)
188
189
static char *
190 3910
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 3910
        char s[vsa_suckaddr_len];
197 3910
        char *r = NULL;
198
        const struct suckaddr *sa;
199
        unsigned m;
200 3910
        int ll, ret = 0;
201
202 3910
        u = p;
203 3910
        ll = l;
204 3910
        m = ae->mask;
205
206 3910
        p += m / 8;
207 3910
        ll -= m / 8;
208 3910
        assert (ll >= 0);
209 3910
        m %= 8;
210
211 3910
        if (m && ((unsigned)*p << m & 0xff) != 0) {
212 92
                ret = 1;
213 92
                m = 0xff00 >> m;
214 92
                *p &= m;
215 92
        }
216 3910
        if (m) {
217 2185
                p++;
218 2185
                ll--;
219 2185
        }
220
221 7314
        for ( ; ll > 0; p++, ll--) {
222 3404
                if (*p == 0)
223 3289
                        continue;
224 115
                ret = 1;
225 115
                *p = 0;
226 115
        }
227 3910
        if (ret == 0)
228 3703
                return (NULL);
229
230 207
        sa = VSA_BuildFAP(s, fam, u, l, NULL, 0);
231 207
        AN(sa);
232 207
        VTCP_name(sa, h, sizeof h, NULL, 0);
233 207
        bprintf(t, "%s/%d", h, ae->mask);
234 207
        if (tl->acl->flag_pedantic != 0) {
235 46
                VSB_cat(tl->sb, "Non-zero bits in masked part, ");
236 46
                VSB_printf(tl->sb, "(maybe use %s ?)\n", t);
237 46
                vcc_ErrWhere(tl, ae->t_addr);
238 46
        }
239 207
        REPLACE(r, t);
240 207
        return (r);
241 3910
}
242
243
static void
244 1012
vcl_acl_fold(struct vcc *tl, struct acl_e **l, struct acl_e **r)
245
{
246
        enum acl_cmp_e cmp;
247
248 1012
        AN(l);
249 1012
        AN(r);
250 1012
        CHECK_OBJ_NOTNULL(*l, VCC_ACL_E_MAGIC);
251 1012
        CHECK_OBJ_NOTNULL(*r, VCC_ACL_E_MAGIC);
252
253 1012
        if ((*l)->not || (*r)->not)
254 0
                return;
255
256 1012
        cmp = vcl_acl_cmp(*l, *r);
257
258 1012
        assert(cmp < 0);
259 1012
        if (cmp == ACL_LT)
260 667
                return;
261
262 345
        do {
263 368
                switch (cmp) {
264
                case ACL_CONTAINED:
265 161
                        VSB_cat(tl->sb, "ACL entry:\n");
266 161
                        vcc_ErrWhere(tl, (*r)->t_addr);
267 161
                        VSB_cat(tl->sb, "supersedes / removes:\n");
268 161
                        vcc_ErrWhere(tl, (*l)->t_addr);
269 161
                        vcc_Warn(tl);
270 161
                        VRBT_REMOVE(acl_tree, &tl->acl->acl_tree, *l);
271 161
                        FREE_OBJ(*l);
272 161
                        *l = VRBT_PREV(acl_tree, &tl->acl->acl_tree, *r);
273 161
                        break;
274
                case ACL_LEFT:
275 207
                        (*l)->mask--;
276 207
                        (*l)->fixed = "folded";
277 207
                        VSB_cat(tl->sb, "ACL entry:\n");
278 207
                        vcc_ErrWhere(tl, (*l)->t_addr);
279 207
                        VSB_cat(tl->sb, "left of:\n");
280 207
                        vcc_ErrWhere(tl, (*r)->t_addr);
281 414
                        VSB_printf(tl->sb, "removing the latter and expanding "
282
                            "mask of the former by one to /%u\n",
283 207
                            (*l)->mask - 8);
284 207
                        vcc_Warn(tl);
285 207
                        VRBT_REMOVE(acl_tree, &tl->acl->acl_tree, *r);
286 207
                        FREE_OBJ(*r);
287 207
                        VRBT_REMOVE(acl_tree, &tl->acl->acl_tree, *l);
288 207
                        vcc_acl_insert_entry(tl, l);
289 207
                        return;
290
                default:
291 0
                        INCOMPL();
292 0
                }
293 161
                if (*l == NULL || *r == NULL)
294 23
                        break;
295 138
                cmp = vcl_acl_cmp(*l, *r);
296 138
        } while (cmp != ACL_LT);
297 1012
}
298
299
static void
300 4117
vcc_acl_insert_entry(struct vcc *tl, struct acl_e **aenp)
301
{
302
        struct acl_e *ae2, *l, *r;
303
304 4117
        CHECK_OBJ_NOTNULL(*aenp, VCC_ACL_E_MAGIC);
305 4117
        ae2 = VRBT_INSERT(acl_tree, &tl->acl->acl_tree, *aenp);
306 4117
        if (ae2 != NULL) {
307 138
                if (ae2->not != (*aenp)->not) {
308 23
                        VSB_cat(tl->sb, "Conflicting ACL entries:\n");
309 23
                        vcc_ErrWhere(tl, ae2->t_addr);
310 23
                        VSB_cat(tl->sb, "vs:\n");
311 23
                        vcc_ErrWhere(tl, (*aenp)->t_addr);
312 23
                }
313 138
                return;
314
        }
315
316 3979
        r = *aenp;
317 3979
        *aenp = NULL;
318
319 3979
        if (tl->acl->flag_fold == 0)
320 3128
                return;
321
322 851
        l = VRBT_PREV(acl_tree, &tl->acl->acl_tree, r);
323 851
        if (l != NULL) {
324 805
                vcl_acl_fold(tl, &l, &r);
325 805
        }
326 851
        if (r == NULL)
327 184
                return;
328 667
        l = r;
329 667
        r = VRBT_NEXT(acl_tree, &tl->acl->acl_tree, l);
330 667
        if (r == NULL)
331 460
                return;
332 207
        vcl_acl_fold(tl, &l, &r);
333 4117
}
334
335
static void
336 3956
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 3956
        if (fam == PF_INET && ae->mask > 32) {
342 46
                VSB_printf(tl->sb,
343 23
                    "Too wide mask (/%u) for IPv4 address\n", ae->mask);
344 23
                if (ae->t_mask != NULL)
345 23
                        vcc_ErrWhere(tl, ae->t_mask);
346
                else
347 0
                        vcc_ErrWhere(tl, ae->t_addr);
348 23
                return;
349
        }
350 3933
        if (fam == PF_INET6 && ae->mask > 128) {
351 46
                VSB_printf(tl->sb,
352 23
                    "Too wide mask (/%u) for IPv6 address\n", ae->mask);
353 23
                vcc_ErrWhere(tl, ae->t_mask);
354 23
                return;
355
        }
356
357
        /* Make a copy from the template */
358 3910
        ALLOC_OBJ(aen, VCC_ACL_E_MAGIC);
359 3910
        AN(aen);
360 3910
        *aen = *ae;
361 3910
        aen->addr = strdup(ae->addr);
362 3910
        AN(aen->addr);
363
364 3910
        aen->fixed = vcc_acl_chk(tl, ae, l, u, fam);
365
366
        /* We treat family as part of address, it saves code */
367 3910
        assert(fam <= 0xff);
368 3910
        aen->data[0] = fam & 0xff;
369 3910
        aen->mask += 8;
370
371 3910
        assert(l + 1UL <= sizeof aen->data);
372 3910
        memcpy(aen->data + 1L, u, l);
373
374 3910
        vcc_acl_insert_entry(tl, &aen);
375 3910
        if (aen != NULL)
376 138
                vcl_acl_free(&aen);
377 3956
}
378
379
static void
380 1311
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 1311
        CHECK_OBJ_NOTNULL(ae, VCC_ACL_E_MAGIC);
389 1311
        memset(&hint, 0, sizeof hint);
390 1311
        hint.ai_family = PF_UNSPEC;
391 1311
        hint.ai_socktype = SOCK_STREAM;
392 1311
        error = getaddrinfo(ae->addr, "0", &hint, &res0);
393 1311
        if (error) {
394 92
                if (ae->para) {
395 92
                        VSB_printf(tl->sb,
396
                            "Warning: %s ignored\n  -- %s\n",
397 46
                            ae->addr, gai_strerror(error));
398 92
                        Fh(tl, 1, "/* Ignored ACL entry: %s%s",
399 46
                            ae->para ? "\"(\" " : "", ae->not ? "\"!\" " : "");
400 46
                        EncToken(tl->fh, ae->t_addr);
401 46
                        if (ae->t_mask)
402 23
                                Fh(tl, 0, "/%u", ae->mask);
403 46
                        Fh(tl, 0, "%s\n", ae->para ? " \")\"" : "");
404 92
                        Fh(tl, 1, " * getaddrinfo:  %s */\n",
405 46
                             gai_strerror(error));
406 46
                } else {
407 92
                        VSB_printf(tl->sb,
408
                            "DNS lookup(%s): %s\n",
409 46
                            ae->addr, gai_strerror(error));
410 46
                        vcc_ErrWhere(tl, ae->t_addr);
411
                }
412 92
                return;
413
        }
414
415 1219
        i4 = i6 = 0;
416 2484
        for (res = res0; res != NULL; res = res->ai_next) {
417 1265
                switch (res->ai_family) {
418
                case PF_INET:
419 46
                        i4++;
420 46
                        break;
421
                case PF_INET6:
422 1219
                        i6++;
423 1219
                        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 1265
        }
431
432 1219
        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 2438
        for (res = res0; res != NULL; res = res->ai_next) {
442 1265
                switch (res->ai_family) {
443
                case PF_INET:
444 46
                        assert(PF_INET < 256);
445 46
                        sin4 = (void*)res->ai_addr;
446 46
                        assert(sizeof(sin4->sin_addr) == 4);
447 46
                        u = (void*)&sin4->sin_addr;
448 46
                        if (ae->t_mask == NULL)
449 46
                                ae->mask = 32;
450 46
                        vcc_acl_add_entry(tl, ae, 4, u, res->ai_family);
451 46
                        break;
452
                case PF_INET6:
453 1219
                        assert(PF_INET6 < 256);
454 1219
                        sin6 = (void*)res->ai_addr;
455 1219
                        assert(sizeof(sin6->sin6_addr) == 16);
456 1219
                        u = (void*)&sin6->sin6_addr;
457 1219
                        if (ae->t_mask == NULL)
458 184
                                ae->mask = 128;
459 1219
                        vcc_acl_add_entry(tl, ae, 16, u, res->ai_family);
460 1219
                        break;
461
                default:
462 0
                        continue;
463
                }
464 1265
                if (tl->err)
465 46
                        freeaddrinfo(res0);
466 1265
                ERRCHK(tl);
467 1219
        }
468 1173
        freeaddrinfo(res0);
469
470 1311
}
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 4002
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 4002
        CHECK_OBJ_NOTNULL(ae, VCC_ACL_E_MAGIC);
488 4002
        memset(b, 0, sizeof b);
489 4002
        p = ae->addr;
490 11638
        for (i = 0; i < 4; i++) {
491 11638
                j = sscanf(p, "%u%n", &u, &k);
492 11638
                if (j != 1)
493 1196
                        return (0);
494 10442
                if (u & ~0xff)
495 46
                        return (0);
496 10396
                b[i] = (unsigned char)u;
497 10396
                if (p[k] == '\0')
498 2691
                        break;
499 7705
                if (p[k] != '.')
500 69
                        return (0);
501 7636
                p += k + 1;
502 7636
        }
503 2691
        if (ae->t_mask == NULL)
504 644
                ae->mask = 8 + 8 * i;
505 2691
        vcc_acl_add_entry(tl, ae, 4, b, AF_INET);
506 2691
        return (1);
507 4002
}
508
509
static void
510 4094
vcc_acl_entry(struct vcc *tl)
511
{
512
        struct acl_e ae[1];
513
        char *sl, *e;
514
515 4094
        INIT_OBJ(ae, VCC_ACL_E_MAGIC);
516
517 4094
        if (tl->t->tok == '!') {
518 920
                ae->not = 1;
519 920
                vcc_NextToken(tl);
520 920
        }
521
522 4094
        if (tl->t->tok == '(') {
523 69
                ae->para = 1;
524 69
                vcc_NextToken(tl);
525 69
        }
526
527 4094
        if (!ae->not && tl->t->tok == '!') {
528 23
                ae->not = 1;
529 23
                vcc_NextToken(tl);
530 23
        }
531
532 4094
        ExpectErr(tl, CSTR);
533 4094
        ae->t_addr = tl->t;
534 4094
        ae->addr = ae->t_addr->dec;
535 4094
        vcc_NextToken(tl);
536
537 4094
        if (strchr(ae->t_addr->dec, '/') != NULL) {
538 69
                sl = strchr(ae->addr, '/');
539 69
                AN(sl);
540 69
                *sl++ = '\0';
541 69
                e = NULL;
542 69
                ae->mask = strtoul(sl, &e, 10);
543 69
                if (*e != '\0') {
544 46
                        VSB_cat(tl->sb, ".../mask is not numeric.\n");
545 46
                        vcc_ErrWhere(tl, ae->t_addr);
546 46
                        return;
547
                }
548 23
                ae->t_mask = ae->t_addr;
549 23
                if (tl->t->tok == '/') {
550 23
                        VSB_cat(tl->sb, "/mask only allowed once.\n");
551 23
                        vcc_ErrWhere(tl, tl->t);
552 23
                        return;
553
                }
554 4025
        } else if (tl->t->tok == '/') {
555 3105
                vcc_NextToken(tl);
556 3105
                ae->t_mask = tl->t;
557 3105
                ExpectErr(tl, CNUM);
558 3105
                ae->mask = vcc_UintVal(tl);
559 3105
        }
560
561 4025
        if (ae->para)
562 69
                SkipToken(tl, ')');
563
564 4002
        if (!vcc_acl_try_netnotation(tl, ae)) {
565 1311
                ERRCHK(tl);
566 1311
                vcc_acl_try_getaddrinfo(tl, ae);
567 1311
        }
568 4002
        ERRCHK(tl);
569 4094
}
570
571
/*********************************************************************
572
 * Emit the tokens making up an entry as C-strings
573
 */
574
575
static void
576 1173
vcc_acl_emit_tokens(const struct vcc *tl, const struct acl_e *ae)
577
{
578
        struct token *t;
579 1173
        const char *sep = "";
580
581 1173
        CHECK_OBJ_NOTNULL(ae, VCC_ACL_E_MAGIC);
582 1173
        t = ae->t_addr;
583 1173
        do {
584 3427
                if (t->tok == CSTR) {
585 1173
                        Fh(tl, 0, "%s\"\\\"\" ", sep);
586 1173
                        EncToken(tl->fh, t);
587 1173
                        Fh(tl, 0, " \"\\\"\"");
588 3427
                } else if (t == ae->t_mask) {
589 1127
                        Fh(tl, 0, " \"%u\"", ae->mask - 8);
590 1127
                } else {
591 1127
                        Fh(tl, 0, "%s\"%.*s\"", sep, PF(t));
592
                }
593 3427
                if (t == ae->t_mask)
594 1127
                        break;
595 2300
                t = vcc_PeekTokenFrom(tl, t);
596 2300
                AN(t);
597 2300
                sep = " ";
598 2300
        } while (ae->t_mask != NULL);
599 1173
        if (ae->fixed)
600 230
                Fh(tl, 0, "\" fixed: %s\"", ae->fixed);
601 1173
}
602
603
/*********************************************************************
604
 * Emit ACL on table format
605
 */
606
607
static unsigned
608 46
vcc_acl_emit_tables(const struct vcc *tl, unsigned n, const char *name)
609
{
610
        struct acl_e *ae;
611 46
        unsigned rv = sizeof(ae->data) + 3;
612 46
        unsigned nn = 0;
613
        size_t sz;
614
615 92
        Fh(tl, 0, "\nstatic unsigned char acl_tbl_%s[%u*%u] = {\n",
616 46
            name, n, rv);
617 1150
        VRBT_FOREACH(ae, acl_tree, &tl->acl->acl_tree) {
618 1104
                if (ae->overlapped)
619 920
                        continue;
620 184
                Fh(tl, 0, "\t0x%02x,", ae->not ? 0 : 1);
621 184
                Fh(tl, 0, "0x%02x,", (ae->mask >> 3) - 1);
622 184
                Fh(tl, 0, "0x%02x,", (0xff00 >> (ae->mask & 7)) & 0xff);
623 3312
                for (sz = 0; sz < sizeof(ae->data); sz++)
624 3128
                        Fh(tl, 0, "0x%02x,", ae->data[sz]);
625 184
                for (; sz < rv - 3; sz++)
626 0
                        Fh(tl, 0, "0,");
627 184
                Fh(tl, 0, "\n");
628 184
                nn++;
629 184
        }
630 46
        assert(n == nn);
631 46
        Fh(tl, 0, "};\n");
632 46
        if (tl->acl->flag_log) {
633 46
                Fh(tl, 0, "\nstatic const char *acl_str_%s[%d] = {\n",
634 23
                    name, n);
635 575
                VRBT_FOREACH(ae, acl_tree, &tl->acl->acl_tree) {
636 552
                        if (ae->overlapped)
637 460
                                continue;
638 92
                        Fh(tl, 0, "\t");
639 184
                        Fh(tl, 0, "\"%sMATCH %s \" ",
640 92
                            ae->not ? "NEG_" : "", name);
641 92
                        vcc_acl_emit_tokens(tl, ae);
642 92
                        Fh(tl, 0, ",\n");
643 92
                }
644 23
                Fh(tl, 0, "};\n");
645 23
        }
646 46
        return (rv);
647
}
648
649
/*********************************************************************
650
 * Emit a function to match the ACL we have collected
651
 */
652
653
static void
654 644
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 644
        struct inifin *ifp = NULL;
660
        struct vsb *func;
661 644
        unsigned n, no, nw = 0;
662
663 644
        func = VSB_new_auto();
664 644
        AN(func);
665 644
        VSB_cat(func, "match_acl_");
666 644
        VCC_PrintCName(func, sym->name, NULL);
667 644
        AZ(VSB_finish(func));
668
669 644
        depth = -1;
670 644
        at[0] = 256;
671 644
        ae2 = NULL;
672 644
        n = no = 0;
673 3956
        VRBT_FOREACH_REVERSE(ae, acl_tree, &tl->acl->acl_tree) {
674 3312
                n++;
675 3312
                if (ae2 == NULL) {
676 621
                        ae2 = ae;
677 3312
                } else if (vcl_acl_disjoint(ae, ae2)) {
678 897
                        ae2 = ae;
679 897
                } else {
680 1794
                        no++;
681 1794
                        ae->overlapped = 1;
682
                }
683 3312
        }
684
685 644
        Fh(tl, 0, "/* acl_n_%s n %u no %u */\n", sym->name, n, no);
686 644
        if (n - no < (1<<1))
687 345
                no = n;
688 299
        else if (!tl->acl->flag_table)
689 253
                no = n;
690
691 644
        if (no < n)
692 46
                nw = vcc_acl_emit_tables(tl, n - no, sym->name);
693
694
695 644
        Fh(tl, 0, "\nstatic int v_matchproto_(acl_match_f)\n");
696 644
        Fh(tl, 0, "%s(VRT_CTX, const VCL_IP p)\n", VSB_data(func));
697 644
        Fh(tl, 0, "{\n");
698 644
        Fh(tl, 0, "\tconst unsigned char *a;\n");
699 644
        Fh(tl, 0, "\tint fam;\n");
700 644
        Fh(tl, 0, "\n");
701 644
        Fh(tl, 0, "\tfam = VRT_VSA_GetPtr(ctx, p, &a);\n");
702 644
        Fh(tl, 0, "\tif (fam < 0) {\n");
703 644
        Fh(tl, 0, "\t\tVRT_fail(ctx,");
704 644
        Fh(tl, 0, " \"ACL %s: no protocol family\");\n", sym->name);
705 644
        Fh(tl, 0, "\t\treturn(0);\n");
706 644
        Fh(tl, 0, "\t}\n\n");
707 644
        if (!tl->err_unref) {
708 46
                ifp = New_IniFin(tl);
709 92
                VSB_printf(ifp->ini,
710 46
                        "\t(void)%s;\n", VSB_data(func));
711 46
        }
712
713 3956
        VRBT_FOREACH(ae, acl_tree, &tl->acl->acl_tree) {
714
715 3312
                if (no < n && !ae->overlapped)
716 184
                        continue;
717
718
                /* Find how much common prefix we have */
719 19044
                for (l = 0; l <= depth && l * 8 < (int)ae->mask - 7; l++) {
720 17020
                        assert(l >= 0);
721 17020
                        if (ae->data[l] != at[l])
722 1104
                                break;
723 15916
                }
724
725
                /* Back down, if necessary */
726 7222
                while (l <= depth) {
727 4094
                        Fh(tl, 0, "\t%*s}\n", -depth, "");
728 4094
                        depth--;
729
                }
730
731 3128
                m = (int)ae->mask;
732 3128
                assert(m >= l*8);
733 3128
                m -= l * 8;
734
735
                /* Do whole byte compares */
736 11086
                for (i = l; m >= 8; m -= 8, i++) {
737 7958
                        if (i == 0)
738 1794
                                Fh(tl, 0, "\t%*s%sif (fam == %d) {\n",
739 897
                                    -i, "", "", ae->data[i]);
740
                        else
741 14122
                                Fh(tl, 0, "\t%*s%sif (a[%d] == %d) {\n",
742 7061
                                    -i, "", "", i - 1, ae->data[i]);
743 7958
                        at[i] = ae->data[i];
744 7958
                        depth = i;
745 7958
                }
746
747 3128
                if (m > 0) {
748
                        // XXX can remove masking due to fixup
749
                        /* Do fractional byte compares */
750 3496
                        Fh(tl, 0, "\t%*s%sif ((a[%d] & 0x%x) == %d) {\n",
751 1748
                            -i, "", "", i - 1, (0xff00 >> m) & 0xff,
752 1748
                            ae->data[i] & ((0xff00 >> m) & 0xff));
753 1748
                        at[i] = 256;
754 1748
                        depth = i;
755 1748
                }
756
757 3128
                i = ((int)ae->mask + 7) / 8;
758
759 3128
                if (tl->acl->flag_log) {
760 2162
                        Fh(tl, 0, "\t%*sVPI_acl_log(ctx, \"%sMATCH %s \" ",
761 1081
                            -i, "", ae->not ? "NEG_" : "", sym->name);
762 1081
                        vcc_acl_emit_tokens(tl, ae);
763 1081
                        Fh(tl, 0, ");\n");
764 1081
                }
765
766 3128
                Fh(tl, 0, "\t%*sreturn (%d);\n", -i, "", ae->not ? 0 : 1);
767 3128
        }
768
769
        /* Unwind */
770 6256
        for (; 0 <= depth; depth--)
771 5612
                Fh(tl, 0, "\t%*.*s}\n", depth, depth, "");
772
773 644
        if (no < n) {
774 46
                Fh(tl, 0, "\treturn(\n\t    VPI_acl_table(ctx,\n");
775 46
                Fh(tl, 0, "\t\tp,\n");
776 46
                Fh(tl, 0, "\t\t%u, %u,\n", n - no, nw);
777 46
                Fh(tl, 0, "\t\tacl_tbl_%s,\n", sym->name);
778 46
                if (tl->acl->flag_log)
779 23
                        Fh(tl, 0, "\t\tacl_str_%s,\n", sym->name);
780
                else
781 23
                        Fh(tl, 0, "\t\tNULL,\n");
782 46
                Fh(tl, 0, "\t\t\"NO MATCH %s\"\n\t    )\n\t);\n", sym->name);
783 46
        } else {
784
                /* Deny by default */
785 598
                if (tl->acl->flag_log)
786 138
                        Fh(tl, 0, "\tVPI_acl_log(ctx, \"NO_MATCH %s\");\n",
787 69
                            sym->name);
788 598
                Fh(tl, 0, "\treturn(0);\n");
789
        }
790 644
        Fh(tl, 0, "}\n");
791
792
        /* Emit the struct that will be referenced */
793 644
        Fh(tl, 0, "\nstatic const struct vrt_acl %s[] = {{\n", sym->rname);
794 644
        Fh(tl, 0, "\t.magic = VRT_ACL_MAGIC,\n");
795 644
        Fh(tl, 0, "\t.match = &%s,\n", VSB_data(func));
796 644
        Fh(tl, 0, "\t.name = \"%s\",\n", sym->name);
797 644
        Fh(tl, 0, "}};\n\n");
798 644
        if (!tl->err_unref) {
799 46
                AN(ifp);
800 46
                VSB_printf(ifp->ini, "\t(void)%s;", sym->rname);
801 46
        }
802 644
        VSB_destroy(&func);
803 644
}
804
805
void
806 989
vcc_ParseAcl(struct vcc *tl)
807
{
808
        struct symbol *sym;
809
        int sign;
810
        struct acl acl[1];
811
812 989
        INIT_OBJ(acl, VCC_ACL_MAGIC);
813 989
        tl->acl = acl;
814 989
        acl->flag_pedantic = 1;
815 989
        vcc_NextToken(tl);
816 989
        VRBT_INIT(&acl->acl_tree);
817
818 989
        vcc_ExpectVid(tl, "ACL");
819 989
        ERRCHK(tl);
820 966
        sym = VCC_HandleSymbol(tl, ACL);
821 966
        ERRCHK(tl);
822 966
        AN(sym);
823
824 1311
        while (1) {
825 1311
                sign = vcc_IsFlag(tl);
826 1311
                if (tl->err) {
827 23
                        VSB_cat(tl->sb,
828
                            "Valid ACL flags are `log` and `table`:\n");
829 23
                        return;
830
                }
831 1288
                if (sign < 0)
832 920
                        break;
833 368
                if (vcc_IdIs(tl->t, "log")) {
834 92
                        acl->flag_log = sign;
835 92
                        vcc_NextToken(tl);
836 368
                } else if (vcc_IdIs(tl->t, "fold")) {
837 23
                        acl->flag_fold = sign;
838 23
                        vcc_NextToken(tl);
839 276
                } else if (vcc_IdIs(tl->t, "pedantic")) {
840 184
                        acl->flag_pedantic = sign;
841 184
                        vcc_NextToken(tl);
842 253
                } else if (vcc_IdIs(tl->t, "table")) {
843 46
                        acl->flag_table = sign;
844 46
                        vcc_NextToken(tl);
845 46
                } else {
846 23
                        VSB_cat(tl->sb, "Unknown ACL flag:\n");
847 23
                        vcc_ErrWhere(tl, tl->t);
848 23
                        return;
849
                }
850
        }
851
852 920
        SkipToken(tl, '{');
853
854 4738
        while (tl->t->tok != '}') {
855 4094
                vcc_acl_entry(tl);
856 4094
                ERRCHK(tl);
857 3841
                SkipToken(tl, ';');
858
        }
859 644
        SkipToken(tl, '}');
860
861 644
        vcc_acl_emit(tl, sym);
862 989
}