sys/netpfil/ipfw/ip_fw_pfil.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
   3  *
   4  * Copyright (c) 2004 Andre Oppermann, Internet Business Solutions AG
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  */
  28
  29 #include <sys/cdefs.h>
  30 __FBSDID("$FreeBSD$");
  31
  32 #include "opt_ipfw.h"
  33 #include "opt_inet.h"
  34 #include "opt_inet6.h"
  35 #ifndef INET
  36 #error IPFIREWALL requires INET.
  37 #endif /* INET */
  38
  39 #include <sys/param.h>
  40 #include <sys/systm.h>
  41 #include <sys/malloc.h>
  42 #include <sys/mbuf.h>
  43 #include <sys/module.h>
  44 #include <sys/kernel.h>
  45 #include <sys/lock.h>
  46 #include <sys/rwlock.h>
  47 #include <sys/socket.h>
  48 #include <sys/sysctl.h>
  49
  50 #include <net/if.h>
  51 #include <net/if_var.h>
  52 #include <net/route.h>
  53 #include <net/ethernet.h>
  54 #include <net/pfil.h>
  55 #include <net/vnet.h>
  56
  57 #include <netinet/in.h>
  58 #include <netinet/in_systm.h>
  59 #include <netinet/ip.h>
  60 #include <netinet/ip_var.h>
  61 #include <netinet/ip_fw.h>
  62 #ifdef INET6
  63 #include <netinet/ip6.h>
  64 #include <netinet6/ip6_var.h>
  65 #include <netinet6/scope6_var.h>
  66 #endif
  67
  68 #include <netgraph/ng_ipfw.h>
  69
  70 #include <netpfil/ipfw/ip_fw_private.h>
  71
  72 #include <machine/in_cksum.h>
  73
  74 VNET_DEFINE_STATIC(int, fw_enable) = 1;
  75 #define V_fw_enable     VNET(fw_enable)
  76
  77 #ifdef INET6
  78 VNET_DEFINE_STATIC(int, fw6_enable) = 1;
  79 #define V_fw6_enable    VNET(fw6_enable)
  80 #endif
  81
  82 VNET_DEFINE_STATIC(int, fwlink_enable) = 0;
  83 #define V_fwlink_enable VNET(fwlink_enable)
  84
  85 int ipfw_chg_hook(SYSCTL_HANDLER_ARGS);
  86
  87 /* Forward declarations. */
  88 static int ipfw_divert(struct mbuf **, bool, struct ipfw_rule_ref *, int);
  89
  90 #ifdef SYSCTL_NODE
  91
  92 SYSBEGIN(f1)
  93
  94 SYSCTL_DECL(_net_inet_ip_fw);
  95 SYSCTL_PROC(_net_inet_ip_fw, OID_AUTO, enable,
  96     CTLFLAG_VNET | CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_SECURE3,
  97     &VNET_NAME(fw_enable), 0, ipfw_chg_hook, "I", "Enable ipfw");
  98 #ifdef INET6
  99 SYSCTL_DECL(_net_inet6_ip6_fw);
 100 SYSCTL_PROC(_net_inet6_ip6_fw, OID_AUTO, enable,
 101     CTLFLAG_VNET | CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_SECURE3,
 102     &VNET_NAME(fw6_enable), 0, ipfw_chg_hook, "I", "Enable ipfw+6");
 103 #endif /* INET6 */
 104
 105 SYSCTL_DECL(_net_link_ether);
 106 SYSCTL_PROC(_net_link_ether, OID_AUTO, ipfw,
 107     CTLFLAG_VNET | CTLTYPE_INT | CTLFLAG_RW | CTLFLAG_SECURE3,
 108     &VNET_NAME(fwlink_enable), 0, ipfw_chg_hook, "I",
 109     "Pass ether pkts through firewall");
 110
 111 SYSEND
 112
 113 #endif /* SYSCTL_NODE */
 114
 115 /*
 116  * The pfilter hook to pass packets to ipfw_chk and then to
 117  * dummynet, divert, netgraph or other modules.
 118  * The packet may be consumed.
 119  */
 120 static pfil_return_t
 121 ipfw_check_packet(struct mbuf **m0, struct ifnet *ifp, int flags,
 122     void *ruleset __unused, struct inpcb *inp)
 123 {
 124         struct ip_fw_args args;
 125         struct m_tag *tag;
 126         pfil_return_t ret;
 127         int ipfw, dir;
 128
 129         args.flags = (flags & PFIL_IN) ? IPFW_ARGS_IN : IPFW_ARGS_OUT;
 130         dir = (flags & PFIL_IN) ? DIR_IN : DIR_OUT;
 131 again:
 132         /*
 133          * extract and remove the tag if present. If we are left
 134          * with onepass, optimize the outgoing path.
 135          */
 136         tag = m_tag_locate(*m0, MTAG_IPFW_RULE, 0, NULL);
 137         if (tag != NULL) {
 138                 args.rule = *((struct ipfw_rule_ref *)(tag+1));
 139                 m_tag_delete(*m0, tag);
 140                 if (args.rule.info & IPFW_ONEPASS)
 141                         return (0);
 142                 args.flags |= IPFW_ARGS_REF;
 143         }
 144
 145         args.m = *m0;
 146         args.ifp = ifp;
 147         args.inp = inp;
 148
 149         ipfw = ipfw_chk(&args);
 150         *m0 = args.m;
 151
 152         KASSERT(*m0 != NULL || ipfw == IP_FW_DENY, ("%s: m0 is NULL",
 153             __func__));
 154
 155         ret = PFIL_PASS;
 156         switch (ipfw) {
 157         case IP_FW_PASS:
 158                 /* next_hop may be set by ipfw_chk */
 159                 if ((args.flags & (IPFW_ARGS_NH4 | IPFW_ARGS_NH4PTR |
 160                     IPFW_ARGS_NH6 | IPFW_ARGS_NH6PTR)) == 0)
 161                         break;
 162 #if (!defined(INET6) && !defined(INET))
 163                 ret = PFIL_DROPPED;
 164 #else
 165             {
 166                 void *psa;
 167                 size_t len;
 168 #ifdef INET
 169                 if (args.flags & (IPFW_ARGS_NH4 | IPFW_ARGS_NH4PTR)) {
 170                         MPASS((args.flags & (IPFW_ARGS_NH4 |
 171                             IPFW_ARGS_NH4PTR)) != (IPFW_ARGS_NH4 |
 172                             IPFW_ARGS_NH4PTR));
 173                         MPASS((args.flags & (IPFW_ARGS_NH6 |
 174                             IPFW_ARGS_NH6PTR)) == 0);
 175                         len = sizeof(struct sockaddr_in);
 176                         psa = (args.flags & IPFW_ARGS_NH4) ?
 177                             &args.hopstore : args.next_hop;
 178                         if (in_localip(satosin(psa)->sin_addr))
 179                                 (*m0)->m_flags |= M_FASTFWD_OURS;
 180                         (*m0)->m_flags |= M_IP_NEXTHOP;
 181                 }
 182 #endif /* INET */
 183 #ifdef INET6
 184                 if (args.flags & (IPFW_ARGS_NH6 | IPFW_ARGS_NH6PTR)) {
 185                         MPASS((args.flags & (IPFW_ARGS_NH6 |
 186                             IPFW_ARGS_NH6PTR)) != (IPFW_ARGS_NH6 |
 187                             IPFW_ARGS_NH6PTR));
 188                         MPASS((args.flags & (IPFW_ARGS_NH4 |
 189                             IPFW_ARGS_NH4PTR)) == 0);
 190                         len = sizeof(struct sockaddr_in6);
 191                         psa = args.next_hop6;
 192                         (*m0)->m_flags |= M_IP6_NEXTHOP;
 193                 }
 194 #endif /* INET6 */
 195                 /*
 196                  * Incoming packets should not be tagged so we do not
 197                  * m_tag_find. Outgoing packets may be tagged, so we
 198                  * reuse the tag if present.
 199                  */
 200                 tag = (flags & PFIL_IN) ? NULL :
 201                         m_tag_find(*m0, PACKET_TAG_IPFORWARD, NULL);
 202                 if (tag != NULL) {
 203                         m_tag_unlink(*m0, tag);
 204                 } else {
 205                         tag = m_tag_get(PACKET_TAG_IPFORWARD, len,
 206                             M_NOWAIT);
 207                         if (tag == NULL) {
 208                                 ret = PFIL_DROPPED;
 209                                 break;
 210                         }
 211                 }
 212                 if ((args.flags & IPFW_ARGS_NH6) == 0)
 213                         bcopy(psa, tag + 1, len);
 214                 m_tag_prepend(*m0, tag);
 215                 ret = 0;
 216 #ifdef INET6
 217                 /* IPv6 next hop needs additional handling */
 218                 if (args.flags & (IPFW_ARGS_NH6 | IPFW_ARGS_NH6PTR)) {
 219                         struct sockaddr_in6 *sa6;
 220
 221                         sa6 = satosin6(tag + 1);
 222                         if (args.flags & IPFW_ARGS_NH6) {
 223                                 sa6->sin6_family = AF_INET6;
 224                                 sa6->sin6_len = sizeof(*sa6);
 225                                 sa6->sin6_addr = args.hopstore6.sin6_addr;
 226                                 sa6->sin6_port = args.hopstore6.sin6_port;
 227                                 sa6->sin6_scope_id =
 228                                     args.hopstore6.sin6_scope_id;
 229                         }
 230                         /*
 231                          * If nh6 address is link-local we should convert
 232                          * it to kernel internal form before doing any
 233                          * comparisons.
 234                          */
 235                         if (sa6_embedscope(sa6, V_ip6_use_defzone) != 0) {
 236                                 ret = PFIL_DROPPED;
 237                                 break;
 238                         }
 239                         if (in6_localip(&sa6->sin6_addr))
 240                                 (*m0)->m_flags |= M_FASTFWD_OURS;
 241                 }
 242 #endif /* INET6 */
 243             }
 244 #endif /* INET || INET6 */
 245                 break;
 246
 247         case IP_FW_DENY:
 248                 ret = PFIL_DROPPED;
 249                 break;
 250
 251         case IP_FW_DUMMYNET:
 252                 if (ip_dn_io_ptr == NULL) {
 253                         ret = PFIL_DROPPED;
 254                         break;
 255                 }
 256                 MPASS(args.flags & IPFW_ARGS_REF);
 257                 if (mtod(*m0, struct ip *)->ip_v == 4)
 258                         (void )ip_dn_io_ptr(m0, dir, &args);
 259                 else if (mtod(*m0, struct ip *)->ip_v == 6)
 260                         (void )ip_dn_io_ptr(m0, dir | PROTO_IPV6, &args);
 261                 else {
 262                         ret = PFIL_DROPPED;
 263                         break;
 264                 }
 265                 /*
 266                  * XXX should read the return value.
 267                  * dummynet normally eats the packet and sets *m0=NULL
 268                  * unless the packet can be sent immediately. In this
 269                  * case args is updated and we should re-run the
 270                  * check without clearing args.
 271                  */
 272                 if (*m0 != NULL)
 273                         goto again;
 274                 ret = PFIL_CONSUMED;
 275                 break;
 276
 277         case IP_FW_TEE:
 278         case IP_FW_DIVERT:
 279                 if (ip_divert_ptr == NULL) {
 280                         ret = PFIL_DROPPED;
 281                         break;
 282                 }
 283                 MPASS(args.flags & IPFW_ARGS_REF);
 284                 (void )ipfw_divert(m0, dir == DIR_IN, &args.rule,
 285                         (ipfw == IP_FW_TEE) ? 1 : 0);
 286                 /* continue processing for the original packet (tee). */
 287                 if (*m0)
 288                         goto again;
 289                 ret = PFIL_CONSUMED;
 290                 break;
 291
 292         case IP_FW_NGTEE:
 293         case IP_FW_NETGRAPH:
 294                 if (ng_ipfw_input_p == NULL) {
 295                         ret = PFIL_DROPPED;
 296                         break;
 297                 }
 298                 MPASS(args.flags & IPFW_ARGS_REF);
 299                 (void )ng_ipfw_input_p(m0, dir, &args,
 300                         (ipfw == IP_FW_NGTEE) ? 1 : 0);
 301                 if (ipfw == IP_FW_NGTEE) /* ignore errors for NGTEE */
 302                         goto again;     /* continue with packet */
 303                 ret = PFIL_CONSUMED;
 304                 break;
 305
 306         case IP_FW_NAT:
 307                 /* honor one-pass in case of successful nat */
 308                 if (V_fw_one_pass)
 309                         break;
 310                 goto again;
 311
 312         case IP_FW_REASS:
 313                 goto again;             /* continue with packet */
 314
 315         case IP_FW_NAT64:
 316                 ret = PFIL_CONSUMED;
 317                 break;
 318
 319         default:
 320                 KASSERT(0, ("%s: unknown retval", __func__));
 321         }
 322
 323         if (ret != PFIL_PASS) {
 324                 if (*m0)
 325                         FREE_PKT(*m0);
 326                 *m0 = NULL;
 327         }
 328
 329         return (ret);
 330 }
 331
 332 /*
 333  * ipfw processing for ethernet packets (in and out).
 334  */
 335 static pfil_return_t
 336 ipfw_check_frame(struct mbuf **m0, struct ifnet *ifp, int dir,
 337     void *ruleset __unused, struct inpcb *inp)
 338 {
 339         struct ip_fw_args args;
 340         struct ether_header save_eh;
 341         struct ether_header *eh;
 342         struct m_tag *mtag;
 343         struct mbuf *m;
 344         pfil_return_t ret;
 345         int i;
 346
 347         args.flags = IPFW_ARGS_ETHER;
 348         args.flags |= (dir & PFIL_IN) ? IPFW_ARGS_IN : IPFW_ARGS_OUT;
 349 again:
 350         /* fetch start point from rule, if any.  remove the tag if present. */
 351         mtag = m_tag_locate(*m0, MTAG_IPFW_RULE, 0, NULL);
 352         if (mtag != NULL) {
 353                 args.rule = *((struct ipfw_rule_ref *)(mtag+1));
 354                 m_tag_delete(*m0, mtag);
 355                 if (args.rule.info & IPFW_ONEPASS)
 356                         return (0);
 357                 args.flags |= IPFW_ARGS_REF;
 358         }
 359
 360         /* I need some amt of data to be contiguous */
 361         m = *m0;
 362         i = min(m->m_pkthdr.len, max_protohdr);
 363         if (m->m_len < i) {
 364                 m = m_pullup(m, i);
 365                 if (m == NULL) {
 366                         *m0 = m;
 367                         return (0);
 368                 }
 369         }
 370         eh = mtod(m, struct ether_header *);
 371         save_eh = *eh;                  /* save copy for restore below */
 372         m_adj(m, ETHER_HDR_LEN);        /* strip ethernet header */
 373
 374         args.m = m;             /* the packet we are looking at         */
 375         args.ifp = ifp;
 376         args.eh = &save_eh;     /* MAC header for bridged/MAC packets   */
 377         args.inp = inp; /* used by ipfw uid/gid/jail rules      */
 378         i = ipfw_chk(&args);
 379         m = args.m;
 380         if (m != NULL) {
 381                 /*
 382                  * Restore Ethernet header, as needed, in case the
 383                  * mbuf chain was replaced by ipfw.
 384                  */
 385                 M_PREPEND(m, ETHER_HDR_LEN, M_NOWAIT);
 386                 if (m == NULL) {
 387                         *m0 = NULL;
 388                         return (0);
 389                 }
 390                 if (eh != mtod(m, struct ether_header *))
 391                         bcopy(&save_eh, mtod(m, struct ether_header *),
 392                                 ETHER_HDR_LEN);
 393         }
 394         *m0 = m;
 395
 396         ret = PFIL_PASS;
 397         /* Check result of ipfw_chk() */
 398         switch (i) {
 399         case IP_FW_PASS:
 400                 break;
 401
 402         case IP_FW_DENY:
 403                 ret = PFIL_DROPPED;
 404                 break;
 405
 406         case IP_FW_DUMMYNET:
 407                 if (ip_dn_io_ptr == NULL) {
 408                         ret = PFIL_DROPPED;
 409                         break;
 410                 }
 411                 *m0 = NULL;
 412                 dir = (dir & PFIL_IN) ? DIR_IN : DIR_OUT;
 413                 MPASS(args.flags & IPFW_ARGS_REF);
 414                 ip_dn_io_ptr(&m, dir | PROTO_LAYER2, &args);
 415                 return (PFIL_CONSUMED);
 416
 417         case IP_FW_NGTEE:
 418         case IP_FW_NETGRAPH:
 419                 if (ng_ipfw_input_p == NULL) {
 420                         ret = PFIL_DROPPED;
 421                         break;
 422                 }
 423                 MPASS(args.flags & IPFW_ARGS_REF);
 424                 (void )ng_ipfw_input_p(m0, (dir & PFIL_IN) ? DIR_IN : DIR_OUT,
 425                         &args, (i == IP_FW_NGTEE) ? 1 : 0);
 426                 if (i == IP_FW_NGTEE) /* ignore errors for NGTEE */
 427                         goto again;     /* continue with packet */
 428                 ret = PFIL_CONSUMED;
 429                 break;
 430
 431         default:
 432                 KASSERT(0, ("%s: unknown retval", __func__));
 433         }
 434
 435         if (ret != PFIL_PASS) {
 436                 if (*m0)
 437                         FREE_PKT(*m0);
 438                 *m0 = NULL;
 439         }
 440
 441         return (ret);
 442 }
 443
 444 /* do the divert, return 1 on error 0 on success */
 445 static int
 446 ipfw_divert(struct mbuf **m0, bool incoming, struct ipfw_rule_ref *rule,
 447         int tee)
 448 {
 449         /*
 450          * ipfw_chk() has already tagged the packet with the divert tag.
 451          * If tee is set, copy packet and return original.
 452          * If not tee, consume packet and send it to divert socket.
 453          */
 454         struct mbuf *clone;
 455         struct ip *ip = mtod(*m0, struct ip *);
 456         struct m_tag *tag;
 457
 458         /* Cloning needed for tee? */
 459         if (tee == 0) {
 460                 clone = *m0;    /* use the original mbuf */
 461                 *m0 = NULL;
 462         } else {
 463                 clone = m_dup(*m0, M_NOWAIT);
 464                 /* If we cannot duplicate the mbuf, we sacrifice the divert
 465                  * chain and continue with the tee-ed packet.
 466                  */
 467                 if (clone == NULL)
 468                         return 1;
 469         }
 470
 471         /*
 472          * Divert listeners can normally handle non-fragmented packets,
 473          * but we can only reass in the non-tee case.
 474          * This means that listeners on a tee rule may get fragments,
 475          * and have to live with that.
 476          * Note that we now have the 'reass' ipfw option so if we care
 477          * we can do it before a 'tee'.
 478          */
 479         if (!tee) switch (ip->ip_v) {
 480         case IPVERSION:
 481             if (ntohs(ip->ip_off) & (IP_MF | IP_OFFMASK)) {
 482                 int hlen;
 483                 struct mbuf *reass;
 484
 485                 reass = ip_reass(clone); /* Reassemble packet. */
 486                 if (reass == NULL)
 487                         return 0; /* not an error */
 488                 /* if reass = NULL then it was consumed by ip_reass */
 489                 /*
 490                  * IP header checksum fixup after reassembly and leave header
 491                  * in network byte order.
 492                  */
 493                 ip = mtod(reass, struct ip *);
 494                 hlen = ip->ip_hl << 2;
 495                 ip->ip_sum = 0;
 496                 if (hlen == sizeof(struct ip))
 497                         ip->ip_sum = in_cksum_hdr(ip);
 498                 else
 499                         ip->ip_sum = in_cksum(reass, hlen);
 500                 clone = reass;
 501             }
 502             break;
 503 #ifdef INET6
 504         case IPV6_VERSION >> 4:
 505             {
 506             struct ip6_hdr *const ip6 = mtod(clone, struct ip6_hdr *);
 507
 508                 if (ip6->ip6_nxt == IPPROTO_FRAGMENT) {
 509                         int nxt, off;
 510
 511                         off = sizeof(struct ip6_hdr);
 512                         nxt = frag6_input(&clone, &off, 0);
 513                         if (nxt == IPPROTO_DONE)
 514                                 return (0);
 515                 }
 516                 break;
 517             }
 518 #endif
 519         }
 520
 521         /* attach a tag to the packet with the reinject info */
 522         tag = m_tag_alloc(MTAG_IPFW_RULE, 0,
 523                     sizeof(struct ipfw_rule_ref), M_NOWAIT);
 524         if (tag == NULL) {
 525                 FREE_PKT(clone);
 526                 return 1;
 527         }
 528         *((struct ipfw_rule_ref *)(tag+1)) = *rule;
 529         m_tag_prepend(clone, tag);
 530
 531         /* Do the dirty job... */
 532         ip_divert_ptr(clone, incoming);
 533         return 0;
 534 }
 535
 536 /*
 537  * attach or detach hooks for a given protocol family
 538  */
 539 VNET_DEFINE_STATIC(pfil_hook_t, ipfw_inet_hook);
 540 #define V_ipfw_inet_hook        VNET(ipfw_inet_hook)
 541 #ifdef INET6
 542 VNET_DEFINE_STATIC(pfil_hook_t, ipfw_inet6_hook);
 543 #define V_ipfw_inet6_hook       VNET(ipfw_inet6_hook)
 544 #endif
 545 VNET_DEFINE_STATIC(pfil_hook_t, ipfw_link_hook);
 546 #define V_ipfw_link_hook        VNET(ipfw_link_hook)
 547
 548 static int
 549 ipfw_hook(int onoff, int pf)
 550 {
 551         struct pfil_hook_args pha;
 552         struct pfil_link_args pla;
 553         pfil_hook_t *h;
 554
 555         pha.pa_version = PFIL_VERSION;
 556         pha.pa_flags = PFIL_IN | PFIL_OUT;
 557         pha.pa_modname = "ipfw";
 558         pha.pa_ruleset = NULL;
 559
 560         pla.pa_version = PFIL_VERSION;
 561         pla.pa_flags = PFIL_IN | PFIL_OUT |
 562             PFIL_HEADPTR | PFIL_HOOKPTR;
 563
 564         switch (pf) {
 565         case AF_INET:
 566                 pha.pa_func = ipfw_check_packet;
 567                 pha.pa_type = PFIL_TYPE_IP4;
 568                 pha.pa_rulname = "default";
 569                 h = &V_ipfw_inet_hook;
 570                 pla.pa_head = V_inet_pfil_head;
 571                 break;
 572 #ifdef INET6
 573         case AF_INET6:
 574                 pha.pa_func = ipfw_check_packet;
 575                 pha.pa_type = PFIL_TYPE_IP6;
 576                 pha.pa_rulname = "default6";
 577                 h = &V_ipfw_inet6_hook;
 578                 pla.pa_head = V_inet6_pfil_head;
 579                 break;
 580 #endif
 581         case AF_LINK:
 582                 pha.pa_func = ipfw_check_frame;
 583                 pha.pa_type = PFIL_TYPE_ETHERNET;
 584                 pha.pa_rulname = "default-link";
 585                 h = &V_ipfw_link_hook;
 586                 pla.pa_head = V_link_pfil_head;
 587                 break;
 588         }
 589
 590         if (onoff) {
 591                 *h = pfil_add_hook(&pha);
 592                 pla.pa_hook = *h;
 593                 (void)pfil_link(&pla);
 594         } else
 595                 if (*h != NULL)
 596                         pfil_remove_hook(*h);
 597
 598         return 0;
 599 }
 600
 601 int
 602 ipfw_attach_hooks(int arg)
 603 {
 604         int error = 0;
 605
 606         if (arg == 0) /* detach */
 607                 ipfw_hook(0, AF_INET);
 608         else if (V_fw_enable && ipfw_hook(1, AF_INET) != 0) {
 609                 error = ENOENT; /* see ip_fw_pfil.c::ipfw_hook() */
 610                 printf("ipfw_hook() error\n");
 611         }
 612 #ifdef INET6
 613         if (arg == 0) /* detach */
 614                 ipfw_hook(0, AF_INET6);
 615         else if (V_fw6_enable && ipfw_hook(1, AF_INET6) != 0) {
 616                 error = ENOENT;
 617                 printf("ipfw6_hook() error\n");
 618         }
 619 #endif
 620         if (arg == 0) /* detach */
 621                 ipfw_hook(0, AF_LINK);
 622         else if (V_fwlink_enable && ipfw_hook(1, AF_LINK) != 0) {
 623                 error = ENOENT;
 624                 printf("ipfw_link_hook() error\n");
 625         }
 626         return error;
 627 }
 628
 629 int
 630 ipfw_chg_hook(SYSCTL_HANDLER_ARGS)
 631 {
 632         int newval;
 633         int error;
 634         int af;
 635
 636         if (arg1 == &V_fw_enable)
 637                 af = AF_INET;
 638 #ifdef INET6
 639         else if (arg1 == &V_fw6_enable)
 640                 af = AF_INET6;
 641 #endif
 642         else if (arg1 == &V_fwlink_enable)
 643                 af = AF_LINK;
 644         else
 645                 return (EINVAL);
 646
 647         newval = *(int *)arg1;
 648         /* Handle sysctl change */
 649         error = sysctl_handle_int(oidp, &newval, 0, req);
 650
 651         if (error)
 652                 return (error);
 653
 654         /* Formalize new value */
 655         newval = (newval) ? 1 : 0;
 656
 657         if (*(int *)arg1 == newval)
 658                 return (0);
 659
 660         error = ipfw_hook(newval, af);
 661         if (error)
 662                 return (error);
 663         *(int *)arg1 = newval;
 664
 665         return (0);
 666 }
 667 /* end of file */