1 /* $FreeBSD: src/sys/contrib/pf/net/pf_if.c,v 1.6 2004/09/14 15:20:24 mlaier Exp $ */ 2 /* $OpenBSD: pf_if.c,v 1.11 2004/03/15 11:38:23 cedric Exp $ */ 3 /* add $OpenBSD: pf_if.c,v 1.19 2004/08/11 12:06:44 henning Exp $ */ 4 /* $DragonFly: src/sys/net/pf/pf_if.c,v 1.9 2008/01/05 14:02:38 swildner Exp $ */ 5 6 /* 7 * Copyright (c) 2004 The DragonFly Project. All rights reserved. 8 * 9 * Copyright (c) 2001 Daniel Hartmeier 10 * Copyright (c) 2003 Cedric Berger 11 * All rights reserved. 12 * 13 * Redistribution and use in source and binary forms, with or without 14 * modification, are permitted provided that the following conditions 15 * are met: 16 * 17 * - Redistributions of source code must retain the above copyright 18 * notice, this list of conditions and the following disclaimer. 19 * - Redistributions in binary form must reproduce the above 20 * copyright notice, this list of conditions and the following 21 * disclaimer in the documentation and/or other materials provided 22 * with the distribution. 23 * 24 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 25 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 26 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 27 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 28 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 29 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 30 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 31 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER 32 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 33 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 34 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 35 * POSSIBILITY OF SUCH DAMAGE. 36 */ 37 38 #include "opt_inet.h" 39 #include "opt_inet6.h" 40 41 #include <sys/param.h> 42 #include <sys/systm.h> 43 #include <sys/malloc.h> 44 #include <sys/mbuf.h> 45 #include <sys/eventhandler.h> 46 #include <sys/filio.h> 47 #include <sys/socket.h> 48 #include <sys/socketvar.h> 49 #include <sys/kernel.h> 50 #include <sys/thread2.h> 51 #include <sys/time.h> 52 #include <vm/vm_zone.h> 53 54 #include <net/if.h> 55 #include <net/if_types.h> 56 #include <net/route.h> 57 58 #include <netinet/in.h> 59 #include <netinet/in_var.h> 60 #include <netinet/in_systm.h> 61 #include <netinet/ip.h> 62 #include <netinet/ip_var.h> 63 64 #include <net/pf/pfvar.h> 65 66 #ifdef INET6 67 #include <netinet/ip6.h> 68 #endif /* INET6 */ 69 70 #define ACCEPT_FLAGS(oklist) \ 71 do { \ 72 if ((flags & ~(oklist)) & \ 73 PFI_FLAG_ALLMASK) \ 74 return (EINVAL); \ 75 } while (0) 76 77 #define senderr(e) do { rv = (e); goto _bad; } while (0) 78 79 struct pfi_kif **pfi_index2kif; 80 struct pfi_kif *pfi_self, *pfi_dummy; 81 int pfi_indexlim; 82 struct pfi_ifhead pfi_ifs; 83 struct pfi_statehead pfi_statehead; 84 int pfi_ifcnt; 85 vm_zone_t pfi_addr_pl; 86 long pfi_update = 1; 87 struct pfr_addr *pfi_buffer; 88 int pfi_buffer_cnt; 89 int pfi_buffer_max; 90 char pfi_reserved_anchor[PF_ANCHOR_NAME_SIZE] = 91 PF_RESERVED_ANCHOR; 92 char pfi_interface_ruleset[PF_RULESET_NAME_SIZE] = 93 PF_INTERFACE_RULESET; 94 95 eventhandler_tag pfi_clone_cookie = NULL; 96 eventhandler_tag pfi_attach_cookie = NULL; 97 eventhandler_tag pfi_detach_cookie = NULL; 98 99 void pfi_dynaddr_update(void *); 100 void pfi_kifaddr_update(void *); 101 void pfi_table_update(struct pfr_ktable *, struct pfi_kif *, 102 int, int); 103 void pfi_instance_add(struct ifnet *, int, int); 104 void pfi_address_add(struct sockaddr *, int, int); 105 int pfi_if_compare(struct pfi_kif *, struct pfi_kif *); 106 struct pfi_kif *pfi_if_create(const char *, struct pfi_kif *, int); 107 void pfi_copy_group(char *, const char *, int); 108 void pfi_dynamic_drivers(void); 109 void pfi_newgroup(const char *, int); 110 int pfi_skip_if(const char *, struct pfi_kif *, int); 111 int pfi_unmask(void *); 112 void pfi_dohooks(struct pfi_kif *); 113 void pfi_kifaddr_update_event(void *, struct ifnet *); 114 void pfi_attach_clone_event(void *, struct if_clone *); 115 void pfi_attach_ifnet_event(void *, struct ifnet *); 116 void pfi_detach_ifnet_event(void *, struct ifnet *); 117 118 RB_PROTOTYPE(pfi_ifhead, pfi_kif, pfik_tree, pfi_if_compare); 119 RB_GENERATE(pfi_ifhead, pfi_kif, pfik_tree, pfi_if_compare); 120 121 #define PFI_DYNAMIC_BUSES { "pcmcia", "cardbus", "uhub" } 122 #define PFI_BUFFER_MAX 0x10000 123 MALLOC_DEFINE(PFI_MTYPE, "pf_if", "pf interface table"); 124 125 void 126 pfi_initialize(void) 127 { 128 struct ifnet *ifp; 129 130 if (pfi_self != NULL) /* already initialized */ 131 return; 132 133 TAILQ_INIT(&pfi_statehead); 134 pfi_buffer_max = 64; 135 pfi_buffer = kmalloc(pfi_buffer_max * sizeof(*pfi_buffer), 136 PFI_MTYPE, M_WAITOK); 137 pfi_self = pfi_if_create("self", NULL, PFI_IFLAG_GROUP); 138 pfi_dynamic_drivers(); 139 140 TAILQ_FOREACH(ifp, &ifnet, if_link) { 141 if (ifp->if_dunit != IF_DUNIT_NONE) 142 pfi_attach_ifnet(ifp); 143 } 144 pfi_dummy = pfi_if_create("notyet", pfi_self, 145 PFI_IFLAG_GROUP | PFI_IFLAG_DYNAMIC); 146 pfi_attach_cookie = EVENTHANDLER_REGISTER(ifnet_attach_event, 147 pfi_attach_ifnet_event, NULL, EVENTHANDLER_PRI_ANY); 148 pfi_detach_cookie = EVENTHANDLER_REGISTER(ifnet_detach_event, 149 pfi_detach_ifnet_event, NULL, EVENTHANDLER_PRI_ANY); 150 pfi_clone_cookie = EVENTHANDLER_REGISTER(if_clone_event, 151 pfi_attach_clone_event, NULL, EVENTHANDLER_PRI_ANY); 152 } 153 154 void 155 pfi_cleanup(void) 156 { 157 struct pfi_kif *p, key; 158 struct ifnet *ifp; 159 160 EVENTHANDLER_DEREGISTER(ifnet_attach_event, pfi_attach_cookie); 161 EVENTHANDLER_DEREGISTER(ifnet_detach_event, pfi_detach_cookie); 162 EVENTHANDLER_DEREGISTER(if_clone_event, pfi_clone_cookie); 163 164 /* release PFI_IFLAG_INSTANCE */ 165 TAILQ_FOREACH(ifp, &ifnet, if_link) { 166 strlcpy(key.pfik_name, ifp->if_xname, sizeof(key.pfik_name)); 167 p = RB_FIND(pfi_ifhead, &pfi_ifs, &key); 168 if (p != NULL) 169 pfi_detach_ifnet(ifp); 170 } 171 172 /* XXX clear all other interface group */ 173 while ((p = RB_MIN(pfi_ifhead, &pfi_ifs))) { 174 RB_REMOVE(pfi_ifhead, &pfi_ifs, p); 175 176 kfree(p->pfik_ah_head, PFI_MTYPE); 177 kfree(p, PFI_MTYPE); 178 } 179 kfree(pfi_index2kif, PFI_MTYPE); 180 kfree(pfi_buffer, PFI_MTYPE); 181 pfi_index2kif = NULL; 182 pfi_buffer = NULL; 183 pfi_self = NULL; 184 } 185 186 /* 187 * Wrapper functions for FreeBSD eventhandler 188 */ 189 void 190 pfi_kifaddr_update_event(void *arg, struct ifnet *ifp) 191 { 192 struct pfi_kif *p = arg; 193 194 /* 195 * Check to see if it is 'our' interface as we do not have per 196 * interface hooks and thus get an update for every interface. 197 */ 198 if (p && p->pfik_ifp == ifp) 199 pfi_kifaddr_update(p); 200 } 201 202 void 203 pfi_attach_clone_event(void *arg __unused, struct if_clone *ifc) 204 { 205 pfi_attach_clone(ifc); 206 } 207 208 void 209 pfi_attach_ifnet_event(void *arg __unused, struct ifnet *ifp) 210 { 211 if (ifp->if_dunit != IF_DUNIT_NONE) 212 pfi_attach_ifnet(ifp); 213 } 214 215 void 216 pfi_detach_ifnet_event(void *arg __unused, struct ifnet *ifp) 217 { 218 pfi_detach_ifnet(ifp); 219 } 220 221 void 222 pfi_attach_clone(struct if_clone *ifc) 223 { 224 pfi_initialize(); 225 pfi_newgroup(ifc->ifc_name, PFI_IFLAG_CLONABLE); 226 } 227 228 void 229 pfi_attach_ifnet(struct ifnet *ifp) 230 { 231 struct pfi_kif *p, *q, key; 232 int realname; 233 234 pfi_initialize(); 235 crit_enter(); 236 pfi_update++; 237 if (ifp->if_index >= pfi_indexlim) { 238 /* 239 * grow pfi_index2kif, similar to ifindex2ifnet code in if.c 240 */ 241 size_t m, n, oldlim; 242 struct pfi_kif **mp, **np; 243 244 oldlim = pfi_indexlim; 245 if (pfi_indexlim == 0) 246 pfi_indexlim = 64; 247 while (ifp->if_index >= pfi_indexlim) 248 pfi_indexlim <<= 1; 249 250 m = oldlim * sizeof(struct pfi_kif *); 251 mp = pfi_index2kif; 252 n = pfi_indexlim * sizeof(struct pfi_kif *); 253 np = kmalloc(n, PFI_MTYPE, M_NOWAIT | M_ZERO); 254 if (np == NULL) 255 panic("pfi_attach_ifnet: " 256 "cannot allocate translation table"); 257 if (mp != NULL) 258 bcopy(mp, np, m); 259 pfi_index2kif = np; 260 if (mp != NULL) 261 kfree(mp, PFI_MTYPE); 262 } 263 264 strlcpy(key.pfik_name, ifp->if_xname, sizeof(key.pfik_name)); 265 p = RB_FIND(pfi_ifhead, &pfi_ifs, &key); 266 /* some additional trickery for placeholders */ 267 if ((p == NULL) || (p->pfik_parent == pfi_dummy)) { 268 /* are we looking at a renamed instance or not? */ 269 pfi_copy_group(key.pfik_name, ifp->if_xname, 270 sizeof(key.pfik_name)); 271 realname = (strncmp(key.pfik_name, ifp->if_dname, 272 sizeof(key.pfik_name)) == 0); 273 /* add group */ 274 /* we can change if_xname, hence use if_dname as group id */ 275 pfi_copy_group(key.pfik_name, ifp->if_dname, 276 sizeof(key.pfik_name)); 277 q = RB_FIND(pfi_ifhead, &pfi_ifs, &key); 278 if (q == NULL) 279 q = pfi_if_create(key.pfik_name, pfi_self, 280 PFI_IFLAG_GROUP|PFI_IFLAG_DYNAMIC); 281 else if (q->pfik_parent == pfi_dummy) { 282 q->pfik_parent = pfi_self; 283 q->pfik_flags = (PFI_IFLAG_GROUP | PFI_IFLAG_DYNAMIC); 284 } 285 if (q == NULL) 286 panic("pfi_attach_ifnet: " 287 "cannot allocate '%s' group", key.pfik_name); 288 289 /* add/modify interface */ 290 if (p == NULL) 291 p = pfi_if_create(ifp->if_xname, q, 292 realname?PFI_IFLAG_INSTANCE:PFI_IFLAG_PLACEHOLDER); 293 else { 294 /* remove from the dummy group */ 295 /* XXX: copy stats? We should not have any!!! */ 296 pfi_dummy->pfik_delcnt++; 297 TAILQ_REMOVE(&pfi_dummy->pfik_grouphead, p, 298 pfik_instances); 299 /* move to the right group */ 300 p->pfik_parent = q; 301 q->pfik_addcnt++; 302 TAILQ_INSERT_TAIL(&q->pfik_grouphead, p, 303 pfik_instances); 304 if (realname) { 305 p->pfik_flags &= ~PFI_IFLAG_PLACEHOLDER; 306 p->pfik_flags |= PFI_IFLAG_INSTANCE; 307 } 308 } 309 if (p == NULL) 310 panic("pfi_attach_ifnet: " 311 "cannot allocate '%s' interface", ifp->if_xname); 312 } else 313 q = p->pfik_parent; 314 p->pfik_ifp = ifp; 315 p->pfik_flags |= PFI_IFLAG_ATTACHED; 316 p->pfik_ah_cookie = EVENTHANDLER_REGISTER(ifaddr_event, 317 pfi_kifaddr_update_event, p, EVENTHANDLER_PRI_ANY); 318 pfi_index2kif[ifp->if_index] = p; 319 pfi_dohooks(p); 320 crit_exit(); 321 } 322 323 void 324 pfi_detach_ifnet(struct ifnet *ifp) 325 { 326 struct pfi_kif *p, *q, key; 327 328 strlcpy(key.pfik_name, ifp->if_xname, sizeof(key.pfik_name)); 329 330 crit_enter(); 331 pfi_update++; 332 p = RB_FIND(pfi_ifhead, &pfi_ifs, &key); 333 if (p == NULL) { 334 kprintf("pfi_detach_ifnet: cannot find %s", ifp->if_xname); 335 crit_exit(); 336 return; 337 } 338 EVENTHANDLER_DEREGISTER(ifaddr_event, p->pfik_ah_cookie); 339 q = p->pfik_parent; 340 p->pfik_ifp = NULL; 341 p->pfik_flags &= ~PFI_IFLAG_ATTACHED; 342 pfi_index2kif[ifp->if_index] = NULL; 343 pfi_dohooks(p); 344 pfi_maybe_destroy(p); 345 crit_exit(); 346 } 347 348 struct pfi_kif * 349 pfi_lookup_create(const char *name) 350 { 351 struct pfi_kif *p, *q, key; 352 353 crit_enter(); 354 p = pfi_lookup_if(name); 355 if (p == NULL) { 356 pfi_copy_group(key.pfik_name, name, sizeof(key.pfik_name)); 357 q = pfi_lookup_if(key.pfik_name); 358 if ((q != NULL) && (q->pfik_parent != pfi_dummy)) 359 p = pfi_if_create(name, q, PFI_IFLAG_INSTANCE); 360 else { 361 if (pfi_dummy == NULL) 362 panic("no 'notyet' dummy group"); 363 p = pfi_if_create(name, pfi_dummy, 364 PFI_IFLAG_PLACEHOLDER); 365 } 366 } 367 crit_exit(); 368 return (p); 369 } 370 371 struct pfi_kif * 372 pfi_attach_rule(const char *name) 373 { 374 struct pfi_kif *p; 375 376 p = pfi_lookup_create(name); 377 if (p != NULL) 378 p->pfik_rules++; 379 return (p); 380 } 381 382 void 383 pfi_detach_rule(struct pfi_kif *p) 384 { 385 if (p == NULL) 386 return; 387 if (p->pfik_rules > 0) 388 p->pfik_rules--; 389 else 390 kprintf("pfi_detach_rule: reference count at 0\n"); 391 pfi_maybe_destroy(p); 392 } 393 394 void 395 pfi_attach_state(struct pfi_kif *p) 396 { 397 if (!p->pfik_states++) 398 TAILQ_INSERT_TAIL(&pfi_statehead, p, pfik_w_states); 399 } 400 401 void 402 pfi_detach_state(struct pfi_kif *p) 403 { 404 if (p == NULL) 405 return; 406 if (p->pfik_states <= 0) { 407 kprintf("pfi_detach_state: reference count <= 0\n"); 408 return; 409 } 410 if (!--p->pfik_states) 411 TAILQ_REMOVE(&pfi_statehead, p, pfik_w_states); 412 pfi_maybe_destroy(p); 413 } 414 415 int 416 pfi_dynaddr_setup(struct pf_addr_wrap *aw, sa_family_t af) 417 { 418 struct pfi_dynaddr *dyn; 419 char tblname[PF_TABLE_NAME_SIZE]; 420 struct pf_ruleset *ruleset = NULL; 421 int rv = 0; 422 423 if (aw->type != PF_ADDR_DYNIFTL) 424 return (0); 425 dyn = pool_get(&pfi_addr_pl, PR_NOWAIT); 426 if (dyn == NULL) 427 return (1); 428 bzero(dyn, sizeof(*dyn)); 429 430 crit_enter(); 431 dyn->pfid_kif = pfi_attach_rule(aw->v.ifname); 432 if (dyn->pfid_kif == NULL) 433 senderr(1); 434 435 dyn->pfid_net = pfi_unmask(&aw->v.a.mask); 436 if (af == AF_INET && dyn->pfid_net == 32) 437 dyn->pfid_net = 128; 438 strlcpy(tblname, aw->v.ifname, sizeof(tblname)); 439 if (aw->iflags & PFI_AFLAG_NETWORK) 440 strlcat(tblname, ":network", sizeof(tblname)); 441 if (aw->iflags & PFI_AFLAG_BROADCAST) 442 strlcat(tblname, ":broadcast", sizeof(tblname)); 443 if (aw->iflags & PFI_AFLAG_PEER) 444 strlcat(tblname, ":peer", sizeof(tblname)); 445 if (aw->iflags & PFI_AFLAG_NOALIAS) 446 strlcat(tblname, ":0", sizeof(tblname)); 447 if (dyn->pfid_net != 128) 448 ksnprintf(tblname + strlen(tblname), 449 sizeof(tblname) - strlen(tblname), "/%d", dyn->pfid_net); 450 ruleset = pf_find_or_create_ruleset(pfi_reserved_anchor, 451 pfi_interface_ruleset); 452 if (ruleset == NULL) 453 senderr(1); 454 455 dyn->pfid_kt = pfr_attach_table(ruleset, tblname); 456 if (dyn->pfid_kt == NULL) 457 senderr(1); 458 459 dyn->pfid_kt->pfrkt_flags |= PFR_TFLAG_ACTIVE; 460 dyn->pfid_iflags = aw->iflags; 461 dyn->pfid_af = af; 462 dyn->pfid_hook_cookie = hook_establish(dyn->pfid_kif->pfik_ah_head, 1, 463 pfi_dynaddr_update, dyn); 464 if (dyn->pfid_hook_cookie == NULL) 465 senderr(1); 466 467 aw->p.dyn = dyn; 468 pfi_dynaddr_update(aw->p.dyn); 469 crit_exit(); 470 return (0); 471 472 _bad: 473 if (dyn->pfid_kt != NULL) 474 pfr_detach_table(dyn->pfid_kt); 475 if (ruleset != NULL) 476 pf_remove_if_empty_ruleset(ruleset); 477 if (dyn->pfid_kif != NULL) 478 pfi_detach_rule(dyn->pfid_kif); 479 pool_put(&pfi_addr_pl, dyn); 480 crit_exit(); 481 return (rv); 482 } 483 484 void 485 pfi_dynaddr_update(void *p) 486 { 487 struct pfi_dynaddr *dyn = (struct pfi_dynaddr *)p; 488 struct pfi_kif *kif = dyn->pfid_kif; 489 struct pfr_ktable *kt = dyn->pfid_kt; 490 491 if (dyn == NULL || kif == NULL || kt == NULL) 492 panic("pfi_dynaddr_update"); 493 if (kt->pfrkt_larg != pfi_update) { 494 /* this table needs to be brought up-to-date */ 495 pfi_table_update(kt, kif, dyn->pfid_net, dyn->pfid_iflags); 496 kt->pfrkt_larg = pfi_update; 497 } 498 pfr_dynaddr_update(kt, dyn); 499 } 500 501 void 502 pfi_table_update(struct pfr_ktable *kt, struct pfi_kif *kif, int net, int flags) 503 { 504 int e, size2 = 0; 505 struct pfi_kif *p; 506 struct pfr_table t; 507 508 if ((kif->pfik_flags & PFI_IFLAG_INSTANCE) && kif->pfik_ifp == NULL) { 509 pfr_clr_addrs(&kt->pfrkt_t, NULL, 0); 510 return; 511 } 512 pfi_buffer_cnt = 0; 513 if ((kif->pfik_flags & PFI_IFLAG_INSTANCE)) 514 pfi_instance_add(kif->pfik_ifp, net, flags); 515 else if (strcmp(kif->pfik_name, "self")) { 516 TAILQ_FOREACH(p, &kif->pfik_grouphead, pfik_instances) 517 pfi_instance_add(p->pfik_ifp, net, flags); 518 } else { 519 RB_FOREACH(p, pfi_ifhead, &pfi_ifs) 520 if (p->pfik_flags & PFI_IFLAG_INSTANCE) 521 pfi_instance_add(p->pfik_ifp, net, flags); 522 } 523 t = kt->pfrkt_t; 524 t.pfrt_flags = 0; 525 if ((e = pfr_set_addrs(&t, pfi_buffer, pfi_buffer_cnt, &size2, 526 NULL, NULL, NULL, 0))) 527 kprintf("pfi_table_update: cannot set %d new addresses " 528 "into table %s: %d\n", pfi_buffer_cnt, kt->pfrkt_name, e); 529 } 530 531 void 532 pfi_instance_add(struct ifnet *ifp, int net, int flags) 533 { 534 struct ifaddr *ia; 535 int got4 = 0, got6 = 0; 536 int net2, af; 537 538 if (ifp == NULL) 539 return; 540 TAILQ_FOREACH(ia, &ifp->if_addrlist, ifa_list) { 541 if (ia->ifa_addr == NULL) 542 continue; 543 af = ia->ifa_addr->sa_family; 544 if (af != AF_INET && af != AF_INET6) 545 continue; 546 /* 547 * XXX: For point-to-point interfaces, (ifname:0) and IPv4, 548 * jump over address without a proper route to work 549 * around a problem with ppp not fully removing the 550 * address used during IPCP. 551 */ 552 if ((ifp->if_flags & IFF_POINTOPOINT) && 553 !(ia->ifa_flags & IFA_ROUTE) && 554 (flags & PFI_AFLAG_NOALIAS) && (af == AF_INET)) 555 continue; 556 if ((flags & PFI_AFLAG_BROADCAST) && af == AF_INET6) 557 continue; 558 if ((flags & PFI_AFLAG_BROADCAST) && 559 !(ifp->if_flags & IFF_BROADCAST)) 560 continue; 561 if ((flags & PFI_AFLAG_PEER) && 562 !(ifp->if_flags & IFF_POINTOPOINT)) 563 continue; 564 if ((flags & PFI_AFLAG_NETWORK) && af == AF_INET6 && 565 IN6_IS_ADDR_LINKLOCAL( 566 &((struct sockaddr_in6 *)ia->ifa_addr)->sin6_addr)) 567 continue; 568 if (flags & PFI_AFLAG_NOALIAS) { 569 if (af == AF_INET && got4) 570 continue; 571 if (af == AF_INET6 && got6) 572 continue; 573 } 574 if (af == AF_INET) 575 got4 = 1; 576 else 577 got6 = 1; 578 net2 = net; 579 if (net2 == 128 && (flags & PFI_AFLAG_NETWORK)) { 580 if (af == AF_INET) { 581 net2 = pfi_unmask(&((struct sockaddr_in *) 582 ia->ifa_netmask)->sin_addr); 583 } else { 584 net2 = pfi_unmask(&((struct sockaddr_in6 *) 585 ia->ifa_netmask)->sin6_addr); 586 } 587 } 588 if (af == AF_INET && net2 > 32) 589 net2 = 32; 590 if (flags & PFI_AFLAG_BROADCAST) 591 pfi_address_add(ia->ifa_broadaddr, af, net2); 592 else if (flags & PFI_AFLAG_PEER) 593 pfi_address_add(ia->ifa_dstaddr, af, net2); 594 else 595 pfi_address_add(ia->ifa_addr, af, net2); 596 } 597 } 598 599 void 600 pfi_address_add(struct sockaddr *sa, int af, int net) 601 { 602 struct pfr_addr *p; 603 int i; 604 605 if (pfi_buffer_cnt >= pfi_buffer_max) { 606 int new_max = pfi_buffer_max * 2; 607 608 if (new_max > PFI_BUFFER_MAX) { 609 kprintf("pfi_address_add: address buffer full (%d/%d)\n", 610 pfi_buffer_cnt, PFI_BUFFER_MAX); 611 return; 612 } 613 p = kmalloc(new_max * sizeof(*pfi_buffer), PFI_MTYPE, 614 M_NOWAIT); 615 if (p == NULL) { 616 kprintf("pfi_address_add: no memory to grow buffer " 617 "(%d/%d)\n", pfi_buffer_cnt, PFI_BUFFER_MAX); 618 return; 619 } 620 memcpy(pfi_buffer, p, pfi_buffer_cnt * sizeof(*pfi_buffer)); 621 /* no need to zero buffer */ 622 kfree(pfi_buffer, PFI_MTYPE); 623 pfi_buffer = p; 624 pfi_buffer_max = new_max; 625 } 626 if (af == AF_INET && net > 32) 627 net = 128; 628 p = pfi_buffer + pfi_buffer_cnt++; 629 bzero(p, sizeof(*p)); 630 p->pfra_af = af; 631 p->pfra_net = net; 632 if (af == AF_INET) 633 p->pfra_ip4addr = ((struct sockaddr_in *)sa)->sin_addr; 634 if (af == AF_INET6) { 635 p->pfra_ip6addr = ((struct sockaddr_in6 *)sa)->sin6_addr; 636 if (IN6_IS_ADDR_LINKLOCAL(&p->pfra_ip6addr)) 637 p->pfra_ip6addr.s6_addr16[1] = 0; 638 } 639 /* mask network address bits */ 640 if (net < 128) 641 ((caddr_t)p)[p->pfra_net/8] &= ~(0xFF >> (p->pfra_net%8)); 642 for (i = (p->pfra_net+7)/8; i < sizeof(p->pfra_u); i++) 643 ((caddr_t)p)[i] = 0; 644 } 645 646 void 647 pfi_dynaddr_remove(struct pf_addr_wrap *aw) 648 { 649 if (aw->type != PF_ADDR_DYNIFTL || aw->p.dyn == NULL || 650 aw->p.dyn->pfid_kif == NULL || aw->p.dyn->pfid_kt == NULL) 651 return; 652 653 crit_enter(); 654 hook_disestablish(aw->p.dyn->pfid_kif->pfik_ah_head, 655 aw->p.dyn->pfid_hook_cookie); 656 pfi_detach_rule(aw->p.dyn->pfid_kif); 657 aw->p.dyn->pfid_kif = NULL; 658 pfr_detach_table(aw->p.dyn->pfid_kt); 659 aw->p.dyn->pfid_kt = NULL; 660 pool_put(&pfi_addr_pl, aw->p.dyn); 661 aw->p.dyn = NULL; 662 crit_exit(); 663 } 664 665 void 666 pfi_dynaddr_copyout(struct pf_addr_wrap *aw) 667 { 668 if (aw->type != PF_ADDR_DYNIFTL || aw->p.dyn == NULL || 669 aw->p.dyn->pfid_kif == NULL) 670 return; 671 aw->p.dyncnt = aw->p.dyn->pfid_acnt4 + aw->p.dyn->pfid_acnt6; 672 } 673 674 void 675 pfi_kifaddr_update(void *v) 676 { 677 crit_enter(); 678 pfi_update++; 679 pfi_dohooks(v); 680 crit_exit(); 681 } 682 683 int 684 pfi_if_compare(struct pfi_kif *p, struct pfi_kif *q) 685 { 686 return (strncmp(p->pfik_name, q->pfik_name, IFNAMSIZ)); 687 } 688 689 struct pfi_kif * 690 pfi_if_create(const char *name, struct pfi_kif *q, int flags) 691 { 692 struct pfi_kif *p; 693 694 p = kmalloc(sizeof(*p), PFI_MTYPE, M_NOWAIT | M_ZERO); 695 if (p == NULL) 696 return (NULL); 697 p->pfik_ah_head = kmalloc(sizeof(*p->pfik_ah_head), PFI_MTYPE, 698 M_NOWAIT | M_ZERO); 699 if (p->pfik_ah_head == NULL) { 700 kfree(p, PFI_MTYPE); 701 return (NULL); 702 } 703 TAILQ_INIT(p->pfik_ah_head); 704 TAILQ_INIT(&p->pfik_grouphead); 705 strlcpy(p->pfik_name, name, sizeof(p->pfik_name)); 706 RB_INIT(&p->pfik_lan_ext); 707 RB_INIT(&p->pfik_ext_gwy); 708 p->pfik_flags = flags; 709 p->pfik_parent = q; 710 p->pfik_tzero = time_second; 711 712 RB_INSERT(pfi_ifhead, &pfi_ifs, p); 713 if (q != NULL) { 714 q->pfik_addcnt++; 715 TAILQ_INSERT_TAIL(&q->pfik_grouphead, p, pfik_instances); 716 } 717 pfi_ifcnt++; 718 return (p); 719 } 720 721 int 722 pfi_maybe_destroy(struct pfi_kif *p) 723 { 724 int i, j, k; 725 struct pfi_kif *q = p->pfik_parent; 726 727 if ((p->pfik_flags & (PFI_IFLAG_ATTACHED | PFI_IFLAG_GROUP)) || 728 p->pfik_rules > 0 || p->pfik_states > 0) 729 if (!(p->pfik_flags & PFI_IFLAG_PLACEHOLDER)) 730 return (0); 731 732 crit_enter(); 733 if (q != NULL) { 734 for (i = 0; i < 2; i++) 735 for (j = 0; j < 2; j++) 736 for (k = 0; k < 2; k++) { 737 q->pfik_bytes[i][j][k] += 738 p->pfik_bytes[i][j][k]; 739 q->pfik_packets[i][j][k] += 740 p->pfik_packets[i][j][k]; 741 /* clear stats in case we return to the dummy group */ 742 p->pfik_bytes[i][j][k] = 0; 743 p->pfik_packets[i][j][k] = 0; 744 } 745 q->pfik_delcnt++; 746 TAILQ_REMOVE(&q->pfik_grouphead, p, pfik_instances); 747 } 748 if (p->pfik_rules > 0 || p->pfik_states > 0) { 749 /* move back to the dummy group */ 750 p->pfik_parent = pfi_dummy; 751 pfi_dummy->pfik_addcnt++; 752 TAILQ_INSERT_TAIL(&pfi_dummy->pfik_grouphead, p, 753 pfik_instances); 754 crit_exit(); 755 return (0); 756 } 757 pfi_ifcnt--; 758 RB_REMOVE(pfi_ifhead, &pfi_ifs, p); 759 crit_exit(); 760 761 kfree(p->pfik_ah_head, PFI_MTYPE); 762 kfree(p, PFI_MTYPE); 763 return (1); 764 } 765 766 void 767 pfi_copy_group(char *p, const char *q, int m) 768 { 769 while (m > 1 && *q && !(*q >= '0' && *q <= '9')) { 770 *p++ = *q++; 771 m--; 772 } 773 if (m > 0) 774 *p++ = '\0'; 775 } 776 777 void 778 pfi_dynamic_drivers(void) 779 { 780 struct ifnet *ifp; 781 782 /* 783 * For FreeBSD basically every interface is "dynamic" as we can unload 784 * modules e.g. 785 */ 786 TAILQ_FOREACH(ifp, &ifnet, if_link) { 787 if (ifp->if_dunit == IF_DUNIT_NONE) 788 continue; 789 pfi_newgroup(ifp->if_dname, PFI_IFLAG_DYNAMIC); 790 } 791 } 792 793 void 794 pfi_newgroup(const char *name, int flags) 795 { 796 struct pfi_kif *p; 797 798 p = pfi_lookup_if(name); 799 if (p == NULL) 800 p = pfi_if_create(name, pfi_self, PFI_IFLAG_GROUP); 801 if (p == NULL) { 802 kprintf("pfi_newgroup: cannot allocate '%s' group", name); 803 return; 804 } 805 p->pfik_flags |= flags; 806 } 807 808 void 809 pfi_fill_oldstatus(struct pf_status *pfs) 810 { 811 struct pfi_kif *p, key; 812 int i, j, k; 813 814 strlcpy(key.pfik_name, pfs->ifname, sizeof(key.pfik_name)); 815 crit_enter(); 816 p = RB_FIND(pfi_ifhead, &pfi_ifs, &key); 817 if (p == NULL) { 818 crit_exit(); 819 return; 820 } 821 bzero(pfs->pcounters, sizeof(pfs->pcounters)); 822 bzero(pfs->bcounters, sizeof(pfs->bcounters)); 823 for (i = 0; i < 2; i++) 824 for (j = 0; j < 2; j++) 825 for (k = 0; k < 2; k++) { 826 pfs->pcounters[i][j][k] = 827 p->pfik_packets[i][j][k]; 828 pfs->bcounters[i][j] += 829 p->pfik_bytes[i][j][k]; 830 } 831 crit_exit(); 832 } 833 834 int 835 pfi_clr_istats(const char *name, int *nzero, int flags) 836 { 837 struct pfi_kif *p; 838 int n = 0; 839 long tzero = time_second; 840 841 crit_enter(); 842 ACCEPT_FLAGS(PFI_FLAG_GROUP|PFI_FLAG_INSTANCE); 843 RB_FOREACH(p, pfi_ifhead, &pfi_ifs) { 844 if (pfi_skip_if(name, p, flags)) 845 continue; 846 bzero(p->pfik_packets, sizeof(p->pfik_packets)); 847 bzero(p->pfik_bytes, sizeof(p->pfik_bytes)); 848 p->pfik_tzero = tzero; 849 n++; 850 } 851 crit_exit(); 852 if (nzero != NULL) 853 *nzero = n; 854 return (0); 855 } 856 857 int 858 pfi_get_ifaces(const char *name, struct pfi_if *buf, int *size, int flags) 859 { 860 struct pfi_kif *p; 861 int n = 0; 862 863 ACCEPT_FLAGS(PFI_FLAG_GROUP|PFI_FLAG_INSTANCE); 864 crit_enter(); 865 RB_FOREACH(p, pfi_ifhead, &pfi_ifs) { 866 if (pfi_skip_if(name, p, flags)) 867 continue; 868 if (*size > n++) { 869 if (!p->pfik_tzero) 870 p->pfik_tzero = boottime.tv_sec; 871 if (copyout(p, buf++, sizeof(*buf))) { 872 crit_exit(); 873 return (EFAULT); 874 } 875 } 876 } 877 crit_exit(); 878 *size = n; 879 return (0); 880 } 881 882 struct pfi_kif * 883 pfi_lookup_if(const char *name) 884 { 885 struct pfi_kif *p, key; 886 887 strlcpy(key.pfik_name, name, sizeof(key.pfik_name)); 888 p = RB_FIND(pfi_ifhead, &pfi_ifs, &key); 889 return (p); 890 } 891 892 int 893 pfi_skip_if(const char *filter, struct pfi_kif *p, int f) 894 { 895 int n; 896 897 if ((p->pfik_flags & PFI_IFLAG_GROUP) && !(f & PFI_FLAG_GROUP)) 898 return (1); 899 if ((p->pfik_flags & PFI_IFLAG_INSTANCE) && !(f & PFI_FLAG_INSTANCE)) 900 return (1); 901 if (filter == NULL || !*filter) 902 return (0); 903 if (!strcmp(p->pfik_name, filter)) 904 return (0); /* exact match */ 905 n = strlen(filter); 906 if (n < 1 || n >= IFNAMSIZ) 907 return (1); /* sanity check */ 908 if (filter[n-1] >= '0' && filter[n-1] <= '9') 909 return (1); /* only do exact match in that case */ 910 if (strncmp(p->pfik_name, filter, n)) 911 return (1); /* prefix doesn't match */ 912 return (p->pfik_name[n] < '0' || p->pfik_name[n] > '9'); 913 } 914 915 /* from pf_print_state.c */ 916 int 917 pfi_unmask(void *addr) 918 { 919 struct pf_addr *m = addr; 920 int i = 31, j = 0, b = 0; 921 u_int32_t tmp; 922 923 while (j < 4 && m->addr32[j] == 0xffffffff) { 924 b += 32; 925 j++; 926 } 927 if (j < 4) { 928 tmp = ntohl(m->addr32[j]); 929 for (i = 31; tmp & (1 << i); --i) 930 b++; 931 } 932 return (b); 933 } 934 935 void 936 pfi_dohooks(struct pfi_kif *p) 937 { 938 for (; p != NULL; p = p->pfik_parent) 939 dohooks(p->pfik_ah_head, 0); 940 } 941 942 int 943 pfi_match_addr(struct pfi_dynaddr *dyn, struct pf_addr *a, sa_family_t af) 944 { 945 if (af == AF_INET) { 946 switch (dyn->pfid_acnt4) { 947 case 0: 948 return (0); 949 case 1: 950 return (PF_MATCHA(0, &dyn->pfid_addr4, 951 &dyn->pfid_mask4, a, AF_INET)); 952 default: 953 return (pfr_match_addr(dyn->pfid_kt, a, AF_INET)); 954 } 955 } else { 956 switch (dyn->pfid_acnt6) { 957 case 0: 958 return (0); 959 case 1: 960 return (PF_MATCHA(0, &dyn->pfid_addr6, 961 &dyn->pfid_mask6, a, AF_INET6)); 962 default: 963 return (pfr_match_addr(dyn->pfid_kt, a, AF_INET6)); 964 } 965 } 966 } 967