sys/netpfil/ipfw/ip_fw_sockopt.c

   1 /*-
   2  * Copyright (c) 2002-2009 Luigi Rizzo, Universita` di Pisa
   3  *
   4  * Supported by: Valeria Paoli
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  25  * SUCH DAMAGE.
  26  */
  27
  28 #include <sys/cdefs.h>
  29 __FBSDID("$FreeBSD$");
  30
  31 /*
  32  * Sockopt support for ipfw. The routines here implement
  33  * the upper half of the ipfw code.
  34  */
  35
  36 #include "opt_ipfw.h"
  37 #include "opt_inet.h"
  38 #ifndef INET
  39 #error IPFIREWALL requires INET.
  40 #endif /* INET */
  41 #include "opt_inet6.h"
  42
  43 #include <sys/param.h>
  44 #include <sys/systm.h>
  45 #include <sys/malloc.h>
  46 #include <sys/mbuf.h>   /* struct m_tag used by nested headers */
  47 #include <sys/kernel.h>
  48 #include <sys/lock.h>
  49 #include <sys/priv.h>
  50 #include <sys/proc.h>
  51 #include <sys/rwlock.h>
  52 #include <sys/socket.h>
  53 #include <sys/socketvar.h>
  54 #include <sys/sysctl.h>
  55 #include <sys/syslog.h>
  56 #include <net/if.h>
  57 #include <net/route.h>
  58 #include <net/vnet.h>
  59
  60 #include <netinet/in.h>
  61 #include <netinet/ip_var.h> /* hooks */
  62 #include <netinet/ip_fw.h>
  63
  64 #include <netpfil/ipfw/ip_fw_private.h>
  65
  66 #ifdef MAC
  67 #include <security/mac/mac_framework.h>
  68 #endif
  69
  70 MALLOC_DEFINE(M_IPFW, "IpFw/IpAcct", "IpFw/IpAcct chain's");
  71
  72 /*
  73  * static variables followed by global ones (none in this file)
  74  */
  75
  76 /*
  77  * Find the smallest rule >= key, id.
  78  * We could use bsearch but it is so simple that we code it directly
  79  */
  80 int
  81 ipfw_find_rule(struct ip_fw_chain *chain, uint32_t key, uint32_t id)
  82 {
  83         int i, lo, hi;
  84         struct ip_fw *r;
  85
  86         for (lo = 0, hi = chain->n_rules - 1; lo < hi;) {
  87                 i = (lo + hi) / 2;
  88                 r = chain->map[i];
  89                 if (r->rulenum < key)
  90                         lo = i + 1;     /* continue from the next one */
  91                 else if (r->rulenum > key)
  92                         hi = i;         /* this might be good */
  93                 else if (r->id < id)
  94                         lo = i + 1;     /* continue from the next one */
  95                 else /* r->id >= id */
  96                         hi = i;         /* this might be good */
  97         };
  98         return hi;
  99 }
 100
 101 /*
 102  * allocate a new map, returns the chain locked. extra is the number
 103  * of entries to add or delete.
 104  */
 105 static struct ip_fw **
 106 get_map(struct ip_fw_chain *chain, int extra, int locked)
 107 {
 108
 109         for (;;) {
 110                 struct ip_fw **map;
 111                 int i;
 112
 113                 i = chain->n_rules + extra;
 114                 map = malloc(i * sizeof(struct ip_fw *), M_IPFW,
 115                         locked ? M_NOWAIT : M_WAITOK);
 116                 if (map == NULL) {
 117                         printf("%s: cannot allocate map\n", __FUNCTION__);
 118                         return NULL;
 119                 }
 120                 if (!locked)
 121                         IPFW_UH_WLOCK(chain);
 122                 if (i >= chain->n_rules + extra) /* good */
 123                         return map;
 124                 /* otherwise we lost the race, free and retry */
 125                 if (!locked)
 126                         IPFW_UH_WUNLOCK(chain);
 127                 free(map, M_IPFW);
 128         }
 129 }
 130
 131 /*
 132  * swap the maps. It is supposed to be called with IPFW_UH_WLOCK
 133  */
 134 static struct ip_fw **
 135 swap_map(struct ip_fw_chain *chain, struct ip_fw **new_map, int new_len)
 136 {
 137         struct ip_fw **old_map;
 138
 139         IPFW_WLOCK(chain);
 140         chain->id++;
 141         chain->n_rules = new_len;
 142         old_map = chain->map;
 143         chain->map = new_map;
 144         IPFW_WUNLOCK(chain);
 145         return old_map;
 146 }
 147
 148 /*
 149  * Add a new rule to the list. Copy the rule into a malloc'ed area, then
 150  * possibly create a rule number and add the rule to the list.
 151  * Update the rule_number in the input struct so the caller knows it as well.
 152  * XXX DO NOT USE FOR THE DEFAULT RULE.
 153  * Must be called without IPFW_UH held
 154  */
 155 int
 156 ipfw_add_rule(struct ip_fw_chain *chain, struct ip_fw *input_rule)
 157 {
 158         struct ip_fw *rule;
 159         int i, l, insert_before;
 160         struct ip_fw **map;     /* the new array of pointers */
 161
 162         if (chain->rules == NULL || input_rule->rulenum > IPFW_DEFAULT_RULE-1)
 163                 return (EINVAL);
 164
 165         l = RULESIZE(input_rule);
 166         rule = malloc(l, M_IPFW, M_WAITOK | M_ZERO);
 167         if (rule == NULL)
 168                 return (ENOSPC);
 169         /* get_map returns with IPFW_UH_WLOCK if successful */
 170         map = get_map(chain, 1, 0 /* not locked */);
 171         if (map == NULL) {
 172                 free(rule, M_IPFW);
 173                 return ENOSPC;
 174         }
 175
 176         bcopy(input_rule, rule, l);
 177         /* clear fields not settable from userland */
 178         rule->x_next = NULL;
 179         rule->next_rule = NULL;
 180         IPFW_ZERO_RULE_COUNTER(rule);
 181
 182         if (V_autoinc_step < 1)
 183                 V_autoinc_step = 1;
 184         else if (V_autoinc_step > 1000)
 185                 V_autoinc_step = 1000;
 186         /* find the insertion point, we will insert before */
 187         insert_before = rule->rulenum ? rule->rulenum + 1 : IPFW_DEFAULT_RULE;
 188         i = ipfw_find_rule(chain, insert_before, 0);
 189         /* duplicate first part */
 190         if (i > 0)
 191                 bcopy(chain->map, map, i * sizeof(struct ip_fw *));
 192         map[i] = rule;
 193         /* duplicate remaining part, we always have the default rule */
 194         bcopy(chain->map + i, map + i + 1,
 195                 sizeof(struct ip_fw *) *(chain->n_rules - i));
 196         if (rule->rulenum == 0) {
 197                 /* write back the number */
 198                 rule->rulenum = i > 0 ? map[i-1]->rulenum : 0;
 199                 if (rule->rulenum < IPFW_DEFAULT_RULE - V_autoinc_step)
 200                         rule->rulenum += V_autoinc_step;
 201                 input_rule->rulenum = rule->rulenum;
 202         }
 203
 204         rule->id = chain->id + 1;
 205         map = swap_map(chain, map, chain->n_rules + 1);
 206         chain->static_len += l;
 207         IPFW_UH_WUNLOCK(chain);
 208         if (map)
 209                 free(map, M_IPFW);
 210         return (0);
 211 }
 212
 213 /*
 214  * Reclaim storage associated with a list of rules.  This is
 215  * typically the list created using remove_rule.
 216  * A NULL pointer on input is handled correctly.
 217  */
 218 void
 219 ipfw_reap_rules(struct ip_fw *head)
 220 {
 221         struct ip_fw *rule;
 222
 223         while ((rule = head) != NULL) {
 224                 head = head->x_next;
 225                 free(rule, M_IPFW);
 226         }
 227 }
 228
 229 /*
 230  * Used by del_entry() to check if a rule should be kept.
 231  * Returns 1 if the rule must be kept, 0 otherwise.
 232  *
 233  * Called with cmd = {0,1,5}.
 234  * cmd == 0 matches on rule numbers, excludes rules in RESVD_SET if n == 0 ;
 235  * cmd == 1 matches on set numbers only, rule numbers are ignored;
 236  * cmd == 5 matches on rule and set numbers.
 237  *
 238  * n == 0 is a wildcard for rule numbers, there is no wildcard for sets.
 239  *
 240  * Rules to keep are
 241  *      (default || reserved || !match_set || !match_number)
 242  * where
 243  *   default ::= (rule->rulenum == IPFW_DEFAULT_RULE)
 244  *      // the default rule is always protected
 245  *
 246  *   reserved ::= (cmd == 0 && n == 0 && rule->set == RESVD_SET)
 247  *      // RESVD_SET is protected only if cmd == 0 and n == 0 ("ipfw flush")
 248  *
 249  *   match_set ::= (cmd == 0 || rule->set == set)
 250  *      // set number is ignored for cmd == 0
 251  *
 252  *   match_number ::= (cmd == 1 || n == 0 || n == rule->rulenum)
 253  *      // number is ignored for cmd == 1 or n == 0
 254  *
 255  */
 256 static int
 257 keep_rule(struct ip_fw *rule, uint8_t cmd, uint8_t set, uint32_t n)
 258 {
 259         return
 260                  (rule->rulenum == IPFW_DEFAULT_RULE)           ||
 261                  (cmd == 0 && n == 0 && rule->set == RESVD_SET) ||
 262                 !(cmd == 0 || rule->set == set)                 ||
 263                 !(cmd == 1 || n == 0 || n == rule->rulenum);
 264 }
 265
 266 /**
 267  * Remove all rules with given number, or do set manipulation.
 268  * Assumes chain != NULL && *chain != NULL.
 269  *
 270  * The argument is an uint32_t. The low 16 bit are the rule or set number;
 271  * the next 8 bits are the new set; the top 8 bits indicate the command:
 272  *
 273  *      0       delete rules numbered "rulenum"
 274  *      1       delete rules in set "rulenum"
 275  *      2       move rules "rulenum" to set "new_set"
 276  *      3       move rules from set "rulenum" to set "new_set"
 277  *      4       swap sets "rulenum" and "new_set"
 278  *      5       delete rules "rulenum" and set "new_set"
 279  */
 280 static int
 281 del_entry(struct ip_fw_chain *chain, uint32_t arg)
 282 {
 283         struct ip_fw *rule;
 284         uint32_t num;   /* rule number or old_set */
 285         uint8_t cmd, new_set;
 286         int start, end, i, ofs, n;
 287         struct ip_fw **map = NULL;
 288         int error = 0;
 289
 290         num = arg & 0xffff;
 291         cmd = (arg >> 24) & 0xff;
 292         new_set = (arg >> 16) & 0xff;
 293
 294         if (cmd > 5 || new_set > RESVD_SET)
 295                 return EINVAL;
 296         if (cmd == 0 || cmd == 2 || cmd == 5) {
 297                 if (num >= IPFW_DEFAULT_RULE)
 298                         return EINVAL;
 299         } else {
 300                 if (num > RESVD_SET)    /* old_set */
 301                         return EINVAL;
 302         }
 303
 304         IPFW_UH_WLOCK(chain);   /* arbitrate writers */
 305         chain->reap = NULL;     /* prepare for deletions */
 306
 307         switch (cmd) {
 308         case 0: /* delete rules "num" (num == 0 matches all) */
 309         case 1: /* delete all rules in set N */
 310         case 5: /* delete rules with number N and set "new_set". */
 311
 312                 /*
 313                  * Locate first rule to delete (start), the rule after
 314                  * the last one to delete (end), and count how many
 315                  * rules to delete (n). Always use keep_rule() to
 316                  * determine which rules to keep.
 317                  */
 318                 n = 0;
 319                 if (cmd == 1) {
 320                         /* look for a specific set including RESVD_SET.
 321                          * Must scan the entire range, ignore num.
 322                          */
 323                         new_set = num;
 324                         for (start = -1, end = i = 0; i < chain->n_rules; i++) {
 325                                 if (keep_rule(chain->map[i], cmd, new_set, 0))
 326                                         continue;
 327                                 if (start < 0)
 328                                         start = i;
 329                                 end = i;
 330                                 n++;
 331                         }
 332                         end++;  /* first non-matching */
 333                 } else {
 334                         /* Optimized search on rule numbers */
 335                         start = ipfw_find_rule(chain, num, 0);
 336                         for (end = start; end < chain->n_rules; end++) {
 337                                 rule = chain->map[end];
 338                                 if (num > 0 && rule->rulenum != num)
 339                                         break;
 340                                 if (!keep_rule(rule, cmd, new_set, num))
 341                                         n++;
 342                         }
 343                 }
 344
 345                 if (n == 0) {
 346                         /* A flush request (arg == 0 or cmd == 1) on empty
 347                          * ruleset returns with no error. On the contrary,
 348                          * if there is no match on a specific request,
 349                          * we return EINVAL.
 350                          */
 351                         if (arg != 0 && cmd != 1)
 352                                 error = EINVAL;
 353                         break;
 354                 }
 355
 356                 /* We have something to delete. Allocate the new map */
 357                 map = get_map(chain, -n, 1 /* locked */);
 358                 if (map == NULL) {
 359                         error = EINVAL;
 360                         break;
 361                 }
 362
 363                 /* 1. bcopy the initial part of the map */
 364                 if (start > 0)
 365                         bcopy(chain->map, map, start * sizeof(struct ip_fw *));
 366                 /* 2. copy active rules between start and end */
 367                 for (i = ofs = start; i < end; i++) {
 368                         rule = chain->map[i];
 369                         if (keep_rule(rule, cmd, new_set, num))
 370                                 map[ofs++] = rule;
 371                 }
 372                 /* 3. copy the final part of the map */
 373                 bcopy(chain->map + end, map + ofs,
 374                         (chain->n_rules - end) * sizeof(struct ip_fw *));
 375                 /* 4. swap the maps (under BH_LOCK) */
 376                 map = swap_map(chain, map, chain->n_rules - n);
 377                 /* 5. now remove the rules deleted from the old map */
 378                 if (cmd == 1)
 379                         ipfw_expire_dyn_rules(chain, NULL, new_set);
 380                 for (i = start; i < end; i++) {
 381                         rule = map[i];
 382                         if (keep_rule(rule, cmd, new_set, num))
 383                                 continue;
 384                         chain->static_len -= RULESIZE(rule);
 385                         if (cmd != 1)
 386                                 ipfw_expire_dyn_rules(chain, rule, RESVD_SET);
 387                         rule->x_next = chain->reap;
 388                         chain->reap = rule;
 389                 }
 390                 break;
 391
 392         /*
 393          * In the next 3 cases the loop stops at (n_rules - 1)
 394          * because the default rule is never eligible..
 395          */
 396
 397         case 2: /* move rules with given RULE number to new set */
 398                 for (i = 0; i < chain->n_rules - 1; i++) {
 399                         rule = chain->map[i];
 400                         if (rule->rulenum == num)
 401                                 rule->set = new_set;
 402                 }
 403                 break;
 404
 405         case 3: /* move rules with given SET number to new set */
 406                 for (i = 0; i < chain->n_rules - 1; i++) {
 407                         rule = chain->map[i];
 408                         if (rule->set == num)
 409                                 rule->set = new_set;
 410                 }
 411                 break;
 412
 413         case 4: /* swap two sets */
 414                 for (i = 0; i < chain->n_rules - 1; i++) {
 415                         rule = chain->map[i];
 416                         if (rule->set == num)
 417                                 rule->set = new_set;
 418                         else if (rule->set == new_set)
 419                                 rule->set = num;
 420                 }
 421                 break;
 422         }
 423
 424         rule = chain->reap;
 425         chain->reap = NULL;
 426         IPFW_UH_WUNLOCK(chain);
 427         ipfw_reap_rules(rule);
 428         if (map)
 429                 free(map, M_IPFW);
 430         return error;
 431 }
 432
 433 /*
 434  * Clear counters for a specific rule.
 435  * Normally run under IPFW_UH_RLOCK, but these are idempotent ops
 436  * so we only care that rules do not disappear.
 437  */
 438 static void
 439 clear_counters(struct ip_fw *rule, int log_only)
 440 {
 441         ipfw_insn_log *l = (ipfw_insn_log *)ACTION_PTR(rule);
 442
 443         if (log_only == 0)
 444                 IPFW_ZERO_RULE_COUNTER(rule);
 445         if (l->o.opcode == O_LOG)
 446                 l->log_left = l->max_log;
 447 }
 448
 449 /**
 450  * Reset some or all counters on firewall rules.
 451  * The argument `arg' is an u_int32_t. The low 16 bit are the rule number,
 452  * the next 8 bits are the set number, the top 8 bits are the command:
 453  *      0       work with rules from all set's;
 454  *      1       work with rules only from specified set.
 455  * Specified rule number is zero if we want to clear all entries.
 456  * log_only is 1 if we only want to reset logs, zero otherwise.
 457  */
 458 static int
 459 zero_entry(struct ip_fw_chain *chain, u_int32_t arg, int log_only)
 460 {
 461         struct ip_fw *rule;
 462         char *msg;
 463         int i;
 464
 465         uint16_t rulenum = arg & 0xffff;
 466         uint8_t set = (arg >> 16) & 0xff;
 467         uint8_t cmd = (arg >> 24) & 0xff;
 468
 469         if (cmd > 1)
 470                 return (EINVAL);
 471         if (cmd == 1 && set > RESVD_SET)
 472                 return (EINVAL);
 473
 474         IPFW_UH_RLOCK(chain);
 475         if (rulenum == 0) {
 476                 V_norule_counter = 0;
 477                 for (i = 0; i < chain->n_rules; i++) {
 478                         rule = chain->map[i];
 479                         /* Skip rules not in our set. */
 480                         if (cmd == 1 && rule->set != set)
 481                                 continue;
 482                         clear_counters(rule, log_only);
 483                 }
 484                 msg = log_only ? "All logging counts reset" :
 485                     "Accounting cleared";
 486         } else {
 487                 int cleared = 0;
 488                 for (i = 0; i < chain->n_rules; i++) {
 489                         rule = chain->map[i];
 490                         if (rule->rulenum == rulenum) {
 491                                 if (cmd == 0 || rule->set == set)
 492                                         clear_counters(rule, log_only);
 493                                 cleared = 1;
 494                         }
 495                         if (rule->rulenum > rulenum)
 496                                 break;
 497                 }
 498                 if (!cleared) { /* we did not find any matching rules */
 499                         IPFW_UH_RUNLOCK(chain);
 500                         return (EINVAL);
 501                 }
 502                 msg = log_only ? "logging count reset" : "cleared";
 503         }
 504         IPFW_UH_RUNLOCK(chain);
 505
 506         if (V_fw_verbose) {
 507                 int lev = LOG_SECURITY | LOG_NOTICE;
 508
 509                 if (rulenum)
 510                         log(lev, "ipfw: Entry %d %s.\n", rulenum, msg);
 511                 else
 512                         log(lev, "ipfw: %s.\n", msg);
 513         }
 514         return (0);
 515 }
 516
 517 /*
 518  * Check validity of the structure before insert.
 519  * Rules are simple, so this mostly need to check rule sizes.
 520  */
 521 static int
 522 check_ipfw_struct(struct ip_fw *rule, int size)
 523 {
 524         int l, cmdlen = 0;
 525         int have_action=0;
 526         ipfw_insn *cmd;
 527
 528         if (size < sizeof(*rule)) {
 529                 printf("ipfw: rule too short\n");
 530                 return (EINVAL);
 531         }
 532         /* first, check for valid size */
 533         l = RULESIZE(rule);
 534         if (l != size) {
 535                 printf("ipfw: size mismatch (have %d want %d)\n", size, l);
 536                 return (EINVAL);
 537         }
 538         if (rule->act_ofs >= rule->cmd_len) {
 539                 printf("ipfw: bogus action offset (%u > %u)\n",
 540                     rule->act_ofs, rule->cmd_len - 1);
 541                 return (EINVAL);
 542         }
 543         /*
 544          * Now go for the individual checks. Very simple ones, basically only
 545          * instruction sizes.
 546          */
 547         for (l = rule->cmd_len, cmd = rule->cmd ;
 548                         l > 0 ; l -= cmdlen, cmd += cmdlen) {
 549                 cmdlen = F_LEN(cmd);
 550                 if (cmdlen > l) {
 551                         printf("ipfw: opcode %d size truncated\n",
 552                             cmd->opcode);
 553                         return EINVAL;
 554                 }
 555                 switch (cmd->opcode) {
 556                 case O_PROBE_STATE:
 557                 case O_KEEP_STATE:
 558                 case O_PROTO:
 559                 case O_IP_SRC_ME:
 560                 case O_IP_DST_ME:
 561                 case O_LAYER2:
 562                 case O_IN:
 563                 case O_FRAG:
 564                 case O_DIVERTED:
 565                 case O_IPOPT:
 566                 case O_IPTOS:
 567                 case O_IPPRECEDENCE:
 568                 case O_IPVER:
 569                 case O_SOCKARG:
 570                 case O_TCPFLAGS:
 571                 case O_TCPOPTS:
 572                 case O_ESTAB:
 573                 case O_VERREVPATH:
 574                 case O_VERSRCREACH:
 575                 case O_ANTISPOOF:
 576                 case O_IPSEC:
 577 #ifdef INET6
 578                 case O_IP6_SRC_ME:
 579                 case O_IP6_DST_ME:
 580                 case O_EXT_HDR:
 581                 case O_IP6:
 582 #endif
 583                 case O_IP4:
 584                 case O_TAG:
 585                         if (cmdlen != F_INSN_SIZE(ipfw_insn))
 586                                 goto bad_size;
 587                         break;
 588
 589                 case O_FIB:
 590                         if (cmdlen != F_INSN_SIZE(ipfw_insn))
 591                                 goto bad_size;
 592                         if (cmd->arg1 >= rt_numfibs) {
 593                                 printf("ipfw: invalid fib number %d\n",
 594                                         cmd->arg1);
 595                                 return EINVAL;
 596                         }
 597                         break;
 598
 599                 case O_SETFIB:
 600                         if (cmdlen != F_INSN_SIZE(ipfw_insn))
 601                                 goto bad_size;
 602                         if ((cmd->arg1 != IP_FW_TABLEARG) &&
 603                             (cmd->arg1 >= rt_numfibs)) {
 604                                 printf("ipfw: invalid fib number %d\n",
 605                                         cmd->arg1);
 606                                 return EINVAL;
 607                         }
 608                         goto check_action;
 609
 610                 case O_UID:
 611                 case O_GID:
 612                 case O_JAIL:
 613                 case O_IP_SRC:
 614                 case O_IP_DST:
 615                 case O_TCPSEQ:
 616                 case O_TCPACK:
 617                 case O_PROB:
 618                 case O_ICMPTYPE:
 619                         if (cmdlen != F_INSN_SIZE(ipfw_insn_u32))
 620                                 goto bad_size;
 621                         break;
 622
 623                 case O_LIMIT:
 624                         if (cmdlen != F_INSN_SIZE(ipfw_insn_limit))
 625                                 goto bad_size;
 626                         break;
 627
 628                 case O_LOG:
 629                         if (cmdlen != F_INSN_SIZE(ipfw_insn_log))
 630                                 goto bad_size;
 631
 632                         ((ipfw_insn_log *)cmd)->log_left =
 633                             ((ipfw_insn_log *)cmd)->max_log;
 634
 635                         break;
 636
 637                 case O_IP_SRC_MASK:
 638                 case O_IP_DST_MASK:
 639                         /* only odd command lengths */
 640                         if ( !(cmdlen & 1) || cmdlen > 31)
 641                                 goto bad_size;
 642                         break;
 643
 644                 case O_IP_SRC_SET:
 645                 case O_IP_DST_SET:
 646                         if (cmd->arg1 == 0 || cmd->arg1 > 256) {
 647                                 printf("ipfw: invalid set size %d\n",
 648                                         cmd->arg1);
 649                                 return EINVAL;
 650                         }
 651                         if (cmdlen != F_INSN_SIZE(ipfw_insn_u32) +
 652                             (cmd->arg1+31)/32 )
 653                                 goto bad_size;
 654                         break;
 655
 656                 case O_IP_SRC_LOOKUP:
 657                 case O_IP_DST_LOOKUP:
 658                         if (cmd->arg1 >= IPFW_TABLES_MAX) {
 659                                 printf("ipfw: invalid table number %d\n",
 660                                     cmd->arg1);
 661                                 return (EINVAL);
 662                         }
 663                         if (cmdlen != F_INSN_SIZE(ipfw_insn) &&
 664                             cmdlen != F_INSN_SIZE(ipfw_insn_u32) + 1 &&
 665                             cmdlen != F_INSN_SIZE(ipfw_insn_u32))
 666                                 goto bad_size;
 667                         break;
 668                 case O_MACADDR2:
 669                         if (cmdlen != F_INSN_SIZE(ipfw_insn_mac))
 670                                 goto bad_size;
 671                         break;
 672
 673                 case O_NOP:
 674                 case O_IPID:
 675                 case O_IPTTL:
 676                 case O_IPLEN:
 677                 case O_TCPDATALEN:
 678                 case O_TCPWIN:
 679                 case O_TAGGED:
 680                         if (cmdlen < 1 || cmdlen > 31)
 681                                 goto bad_size;
 682                         break;
 683
 684                 case O_DSCP:
 685                         if (cmdlen != F_INSN_SIZE(ipfw_insn_u32) + 1)
 686                                 goto bad_size;
 687                         break;
 688
 689                 case O_MAC_TYPE:
 690                 case O_IP_SRCPORT:
 691                 case O_IP_DSTPORT: /* XXX artificial limit, 30 port pairs */
 692                         if (cmdlen < 2 || cmdlen > 31)
 693                                 goto bad_size;
 694                         break;
 695
 696                 case O_RECV:
 697                 case O_XMIT:
 698                 case O_VIA:
 699                         if (cmdlen != F_INSN_SIZE(ipfw_insn_if))
 700                                 goto bad_size;
 701                         break;
 702
 703                 case O_ALTQ:
 704                         if (cmdlen != F_INSN_SIZE(ipfw_insn_altq))
 705                                 goto bad_size;
 706                         break;
 707
 708                 case O_PIPE:
 709                 case O_QUEUE:
 710                         if (cmdlen != F_INSN_SIZE(ipfw_insn))
 711                                 goto bad_size;
 712                         goto check_action;
 713
 714                 case O_FORWARD_IP:
 715                         if (cmdlen != F_INSN_SIZE(ipfw_insn_sa))
 716                                 goto bad_size;
 717                         goto check_action;
 718 #ifdef INET6
 719                 case O_FORWARD_IP6:
 720                         if (cmdlen != F_INSN_SIZE(ipfw_insn_sa6))
 721                                 goto bad_size;
 722                         goto check_action;
 723 #endif /* INET6 */
 724
 725                 case O_DIVERT:
 726                 case O_TEE:
 727                         if (ip_divert_ptr == NULL)
 728                                 return EINVAL;
 729                         else
 730                                 goto check_size;
 731                 case O_NETGRAPH:
 732                 case O_NGTEE:
 733                         if (ng_ipfw_input_p == NULL)
 734                                 return EINVAL;
 735                         else
 736                                 goto check_size;
 737                 case O_NAT:
 738                         if (!IPFW_NAT_LOADED)
 739                                 return EINVAL;
 740                         if (cmdlen != F_INSN_SIZE(ipfw_insn_nat))
 741                                 goto bad_size;
 742                         goto check_action;
 743                 case O_FORWARD_MAC: /* XXX not implemented yet */
 744                 case O_CHECK_STATE:
 745                 case O_COUNT:
 746                 case O_ACCEPT:
 747                 case O_DENY:
 748                 case O_REJECT:
 749                 case O_SETDSCP:
 750 #ifdef INET6
 751                 case O_UNREACH6:
 752 #endif
 753                 case O_SKIPTO:
 754                 case O_REASS:
 755                 case O_CALLRETURN:
 756 check_size:
 757                         if (cmdlen != F_INSN_SIZE(ipfw_insn))
 758                                 goto bad_size;
 759 check_action:
 760                         if (have_action) {
 761                                 printf("ipfw: opcode %d, multiple actions"
 762                                         " not allowed\n",
 763                                         cmd->opcode);
 764                                 return EINVAL;
 765                         }
 766                         have_action = 1;
 767                         if (l != cmdlen) {
 768                                 printf("ipfw: opcode %d, action must be"
 769                                         " last opcode\n",
 770                                         cmd->opcode);
 771                                 return EINVAL;
 772                         }
 773                         break;
 774 #ifdef INET6
 775                 case O_IP6_SRC:
 776                 case O_IP6_DST:
 777                         if (cmdlen != F_INSN_SIZE(struct in6_addr) +
 778                             F_INSN_SIZE(ipfw_insn))
 779                                 goto bad_size;
 780                         break;
 781
 782                 case O_FLOW6ID:
 783                         if (cmdlen != F_INSN_SIZE(ipfw_insn_u32) +
 784                             ((ipfw_insn_u32 *)cmd)->o.arg1)
 785                                 goto bad_size;
 786                         break;
 787
 788                 case O_IP6_SRC_MASK:
 789                 case O_IP6_DST_MASK:
 790                         if ( !(cmdlen & 1) || cmdlen > 127)
 791                                 goto bad_size;
 792                         break;
 793                 case O_ICMP6TYPE:
 794                         if( cmdlen != F_INSN_SIZE( ipfw_insn_icmp6 ) )
 795                                 goto bad_size;
 796                         break;
 797 #endif
 798
 799                 default:
 800                         switch (cmd->opcode) {
 801 #ifndef INET6
 802                         case O_IP6_SRC_ME:
 803                         case O_IP6_DST_ME:
 804                         case O_EXT_HDR:
 805                         case O_IP6:
 806                         case O_UNREACH6:
 807                         case O_IP6_SRC:
 808                         case O_IP6_DST:
 809                         case O_FLOW6ID:
 810                         case O_IP6_SRC_MASK:
 811                         case O_IP6_DST_MASK:
 812                         case O_ICMP6TYPE:
 813                                 printf("ipfw: no IPv6 support in kernel\n");
 814                                 return EPROTONOSUPPORT;
 815 #endif
 816                         default:
 817                                 printf("ipfw: opcode %d, unknown opcode\n",
 818                                         cmd->opcode);
 819                                 return EINVAL;
 820                         }
 821                 }
 822         }
 823         if (have_action == 0) {
 824                 printf("ipfw: missing action\n");
 825                 return EINVAL;
 826         }
 827         return 0;
 828
 829 bad_size:
 830         printf("ipfw: opcode %d size %d wrong\n",
 831                 cmd->opcode, cmdlen);
 832         return EINVAL;
 833 }
 834
 835
 836 /*
 837  * Translation of requests for compatibility with FreeBSD 7.2/8.
 838  * a static variable tells us if we have an old client from userland,
 839  * and if necessary we translate requests and responses between the
 840  * two formats.
 841  */
 842 static int is7 = 0;
 843
 844 struct ip_fw7 {
 845         struct ip_fw7   *next;          /* linked list of rules     */
 846         struct ip_fw7   *next_rule;     /* ptr to next [skipto] rule    */
 847         /* 'next_rule' is used to pass up 'set_disable' status      */
 848
 849         uint16_t        act_ofs;        /* offset of action in 32-bit units */
 850         uint16_t        cmd_len;        /* # of 32-bit words in cmd */
 851         uint16_t        rulenum;        /* rule number          */
 852         uint8_t         set;            /* rule set (0..31)     */
 853         // #define RESVD_SET   31  /* set for default and persistent rules */
 854         uint8_t         _pad;           /* padding          */
 855         // uint32_t        id;             /* rule id, only in v.8 */
 856         /* These fields are present in all rules.           */
 857         uint64_t        pcnt;           /* Packet counter       */
 858         uint64_t        bcnt;           /* Byte counter         */
 859         uint32_t        timestamp;      /* tv_sec of last match     */
 860
 861         ipfw_insn       cmd[1];         /* storage for commands     */
 862 };
 863
 864         int convert_rule_to_7(struct ip_fw *rule);
 865 int convert_rule_to_8(struct ip_fw *rule);
 866
 867 #ifndef RULESIZE7
 868 #define RULESIZE7(rule)  (sizeof(struct ip_fw7) + \
 869         ((struct ip_fw7 *)(rule))->cmd_len * 4 - 4)
 870 #endif
 871
 872
 873 /*
 874  * Copy the static and dynamic rules to the supplied buffer
 875  * and return the amount of space actually used.
 876  * Must be run under IPFW_UH_RLOCK
 877  */
 878 static size_t
 879 ipfw_getrules(struct ip_fw_chain *chain, void *buf, size_t space)
 880 {
 881         char *bp = buf;
 882         char *ep = bp + space;
 883         struct ip_fw *rule, *dst;
 884         int l, i;
 885         time_t  boot_seconds;
 886
 887         boot_seconds = boottime.tv_sec;
 888         for (i = 0; i < chain->n_rules; i++) {
 889                 rule = chain->map[i];
 890
 891                 if (is7) {
 892                     /* Convert rule to FreeBSd 7.2 format */
 893                     l = RULESIZE7(rule);
 894                     if (bp + l + sizeof(uint32_t) <= ep) {
 895                         int error;
 896                         bcopy(rule, bp, l + sizeof(uint32_t));
 897                         error = convert_rule_to_7((struct ip_fw *) bp);
 898                         if (error)
 899                                 return 0; /*XXX correct? */
 900                         /*
 901                          * XXX HACK. Store the disable mask in the "next"
 902                          * pointer in a wild attempt to keep the ABI the same.
 903                          * Why do we do this on EVERY rule?
 904                          */
 905                         bcopy(&V_set_disable,
 906                                 &(((struct ip_fw7 *)bp)->next_rule),
 907                                 sizeof(V_set_disable));
 908                         if (((struct ip_fw7 *)bp)->timestamp)
 909                             ((struct ip_fw7 *)bp)->timestamp += boot_seconds;
 910                         bp += l;
 911                     }
 912                     continue; /* go to next rule */
 913                 }
 914
 915                 /* normal mode, don't touch rules */
 916                 l = RULESIZE(rule);
 917                 if (bp + l > ep) { /* should not happen */
 918                         printf("overflow dumping static rules\n");
 919                         break;
 920                 }
 921                 dst = (struct ip_fw *)bp;
 922                 bcopy(rule, dst, l);
 923                 /*
 924                  * XXX HACK. Store the disable mask in the "next"
 925                  * pointer in a wild attempt to keep the ABI the same.
 926                  * Why do we do this on EVERY rule?
 927                  */
 928                 bcopy(&V_set_disable, &dst->next_rule, sizeof(V_set_disable));
 929                 if (dst->timestamp)
 930                         dst->timestamp += boot_seconds;
 931                 bp += l;
 932         }
 933         ipfw_get_dynamic(chain, &bp, ep); /* protected by the dynamic lock */
 934         return (bp - (char *)buf);
 935 }
 936
 937
 938 #define IP_FW3_OPLENGTH(x)      ((x)->sopt_valsize - sizeof(ip_fw3_opheader))
 939 /**
 940  * {set|get}sockopt parser.
 941  */
 942 int
 943 ipfw_ctl(struct sockopt *sopt)
 944 {
 945 #define RULE_MAXSIZE    (256*sizeof(u_int32_t))
 946         int error;
 947         size_t size, len, valsize;
 948         struct ip_fw *buf, *rule;
 949         struct ip_fw_chain *chain;
 950         u_int32_t rulenum[2];
 951         uint32_t opt;
 952         char xbuf[128];
 953         ip_fw3_opheader *op3 = NULL;
 954
 955         error = priv_check(sopt->sopt_td, PRIV_NETINET_IPFW);
 956         if (error)
 957                 return (error);
 958
 959         /*
 960          * Disallow modifications in really-really secure mode, but still allow
 961          * the logging counters to be reset.
 962          */
 963         if (sopt->sopt_name == IP_FW_ADD ||
 964             (sopt->sopt_dir == SOPT_SET && sopt->sopt_name != IP_FW_RESETLOG)) {
 965                 error = securelevel_ge(sopt->sopt_td->td_ucred, 3);
 966                 if (error)
 967                         return (error);
 968         }
 969
 970         chain = &V_layer3_chain;
 971         error = 0;
 972
 973         /* Save original valsize before it is altered via sooptcopyin() */
 974         valsize = sopt->sopt_valsize;
 975         if ((opt = sopt->sopt_name) == IP_FW3) {
 976                 /*
 977                  * Copy not less than sizeof(ip_fw3_opheader).
 978                  * We hope any IP_FW3 command will fit into 128-byte buffer.
 979                  */
 980                 if ((error = sooptcopyin(sopt, xbuf, sizeof(xbuf),
 981                         sizeof(ip_fw3_opheader))) != 0)
 982                         return (error);
 983                 op3 = (ip_fw3_opheader *)xbuf;
 984                 opt = op3->opcode;
 985         }
 986
 987         switch (opt) {
 988         case IP_FW_GET:
 989                 /*
 990                  * pass up a copy of the current rules. Static rules
 991                  * come first (the last of which has number IPFW_DEFAULT_RULE),
 992                  * followed by a possibly empty list of dynamic rule.
 993                  * The last dynamic rule has NULL in the "next" field.
 994                  *
 995                  * Note that the calculated size is used to bound the
 996                  * amount of data returned to the user.  The rule set may
 997                  * change between calculating the size and returning the
 998                  * data in which case we'll just return what fits.
 999                  */
1000                 for (;;) {
1001                         int len = 0, want;
1002
1003                         size = chain->static_len;
1004                         size += ipfw_dyn_len();
1005                         if (size >= sopt->sopt_valsize)
1006                                 break;
1007                         buf = malloc(size, M_TEMP, M_WAITOK);
1008                         if (buf == NULL)
1009                                 break;
1010                         IPFW_UH_RLOCK(chain);
1011                         /* check again how much space we need */
1012                         want = chain->static_len + ipfw_dyn_len();
1013                         if (size >= want)
1014                                 len = ipfw_getrules(chain, buf, size);
1015                         IPFW_UH_RUNLOCK(chain);
1016                         if (size >= want)
1017                                 error = sooptcopyout(sopt, buf, len);
1018                         free(buf, M_TEMP);
1019                         if (size >= want)
1020                                 break;
1021                 }
1022                 break;
1023
1024         case IP_FW_FLUSH:
1025                 /* locking is done within del_entry() */
1026                 error = del_entry(chain, 0); /* special case, rule=0, cmd=0 means all */
1027                 break;
1028
1029         case IP_FW_ADD:
1030                 rule = malloc(RULE_MAXSIZE, M_TEMP, M_WAITOK);
1031                 error = sooptcopyin(sopt, rule, RULE_MAXSIZE,
1032                         sizeof(struct ip_fw7) );
1033
1034                 /*
1035                  * If the size of commands equals RULESIZE7 then we assume
1036                  * a FreeBSD7.2 binary is talking to us (set is7=1).
1037                  * is7 is persistent so the next 'ipfw list' command
1038                  * will use this format.
1039                  * NOTE: If wrong version is guessed (this can happen if
1040                  *       the first ipfw command is 'ipfw [pipe] list')
1041                  *       the ipfw binary may crash or loop infinitly...
1042                  */
1043                 if (sopt->sopt_valsize == RULESIZE7(rule)) {
1044                     is7 = 1;
1045                     error = convert_rule_to_8(rule);
1046                     if (error)
1047                         return error;
1048                     if (error == 0)
1049                         error = check_ipfw_struct(rule, RULESIZE(rule));
1050                 } else {
1051                     is7 = 0;
1052                 if (error == 0)
1053                         error = check_ipfw_struct(rule, sopt->sopt_valsize);
1054                 }
1055                 if (error == 0) {
1056                         /* locking is done within ipfw_add_rule() */
1057                         error = ipfw_add_rule(chain, rule);
1058                         size = RULESIZE(rule);
1059                         if (!error && sopt->sopt_dir == SOPT_GET) {
1060                                 if (is7) {
1061                                         error = convert_rule_to_7(rule);
1062                                         size = RULESIZE7(rule);
1063                                         if (error)
1064                                                 return error;
1065                                 }
1066                                 error = sooptcopyout(sopt, rule, size);
1067                 }
1068                 }
1069                 free(rule, M_TEMP);
1070                 break;
1071
1072         case IP_FW_DEL:
1073                 /*
1074                  * IP_FW_DEL is used for deleting single rules or sets,
1075                  * and (ab)used to atomically manipulate sets. Argument size
1076                  * is used to distinguish between the two:
1077                  *    sizeof(u_int32_t)
1078                  *      delete single rule or set of rules,
1079                  *      or reassign rules (or sets) to a different set.
1080                  *    2*sizeof(u_int32_t)
1081                  *      atomic disable/enable sets.
1082                  *      first u_int32_t contains sets to be disabled,
1083                  *      second u_int32_t contains sets to be enabled.
1084                  */
1085                 error = sooptcopyin(sopt, rulenum,
1086                         2*sizeof(u_int32_t), sizeof(u_int32_t));
1087                 if (error)
1088                         break;
1089                 size = sopt->sopt_valsize;
1090                 if (size == sizeof(u_int32_t) && rulenum[0] != 0) {
1091                         /* delete or reassign, locking done in del_entry() */
1092                         error = del_entry(chain, rulenum[0]);
1093                 } else if (size == 2*sizeof(u_int32_t)) { /* set enable/disable */
1094                         IPFW_UH_WLOCK(chain);
1095                         V_set_disable =
1096                             (V_set_disable | rulenum[0]) & ~rulenum[1] &
1097                             ~(1<<RESVD_SET); /* set RESVD_SET always enabled */
1098                         IPFW_UH_WUNLOCK(chain);
1099                 } else
1100                         error = EINVAL;
1101                 break;
1102
1103         case IP_FW_ZERO:
1104         case IP_FW_RESETLOG: /* argument is an u_int_32, the rule number */
1105                 rulenum[0] = 0;
1106                 if (sopt->sopt_val != 0) {
1107                     error = sooptcopyin(sopt, rulenum,
1108                             sizeof(u_int32_t), sizeof(u_int32_t));
1109                     if (error)
1110                         break;
1111                 }
1112                 error = zero_entry(chain, rulenum[0],
1113                         sopt->sopt_name == IP_FW_RESETLOG);
1114                 break;
1115
1116         /*--- TABLE manipulations are protected by the IPFW_LOCK ---*/
1117         case IP_FW_TABLE_ADD:
1118                 {
1119                         ipfw_table_entry ent;
1120
1121                         error = sooptcopyin(sopt, &ent,
1122                             sizeof(ent), sizeof(ent));
1123                         if (error)
1124                                 break;
1125                         error = ipfw_add_table_entry(chain, ent.tbl,
1126                             &ent.addr, sizeof(ent.addr), ent.masklen,
1127                             IPFW_TABLE_CIDR, ent.value);
1128                 }
1129                 break;
1130
1131         case IP_FW_TABLE_DEL:
1132                 {
1133                         ipfw_table_entry ent;
1134
1135                         error = sooptcopyin(sopt, &ent,
1136                             sizeof(ent), sizeof(ent));
1137                         if (error)
1138                                 break;
1139                         error = ipfw_del_table_entry(chain, ent.tbl,
1140                             &ent.addr, sizeof(ent.addr), ent.masklen, IPFW_TABLE_CIDR);
1141                 }
1142                 break;
1143
1144         case IP_FW_TABLE_XADD: /* IP_FW3 */
1145         case IP_FW_TABLE_XDEL: /* IP_FW3 */
1146                 {
1147                         ipfw_table_xentry *xent = (ipfw_table_xentry *)(op3 + 1);
1148
1149                         /* Check minimum header size */
1150                         if (IP_FW3_OPLENGTH(sopt) < offsetof(ipfw_table_xentry, k)) {
1151                                 error = EINVAL;
1152                                 break;
1153                         }
1154
1155                         /* Check if len field is valid */
1156                         if (xent->len > sizeof(ipfw_table_xentry)) {
1157                                 error = EINVAL;
1158                                 break;
1159                         }
1160
1161                         len = xent->len - offsetof(ipfw_table_xentry, k);
1162
1163                         error = (opt == IP_FW_TABLE_XADD) ?
1164                                 ipfw_add_table_entry(chain, xent->tbl, &xent->k,
1165                                         len, xent->masklen, xent->type, xent->value) :
1166                                 ipfw_del_table_entry(chain, xent->tbl, &xent->k,
1167                                         len, xent->masklen, xent->type);
1168                 }
1169                 break;
1170
1171         case IP_FW_TABLE_FLUSH:
1172                 {
1173                         u_int16_t tbl;
1174
1175                         error = sooptcopyin(sopt, &tbl,
1176                             sizeof(tbl), sizeof(tbl));
1177                         if (error)
1178                                 break;
1179                         error = ipfw_flush_table(chain, tbl);
1180                 }
1181                 break;
1182
1183         case IP_FW_TABLE_GETSIZE:
1184                 {
1185                         u_int32_t tbl, cnt;
1186
1187                         if ((error = sooptcopyin(sopt, &tbl, sizeof(tbl),
1188                             sizeof(tbl))))
1189                                 break;
1190                         IPFW_RLOCK(chain);
1191                         error = ipfw_count_table(chain, tbl, &cnt);
1192                         IPFW_RUNLOCK(chain);
1193                         if (error)
1194                                 break;
1195                         error = sooptcopyout(sopt, &cnt, sizeof(cnt));
1196                 }
1197                 break;
1198
1199         case IP_FW_TABLE_LIST:
1200                 {
1201                         ipfw_table *tbl;
1202
1203                         if (sopt->sopt_valsize < sizeof(*tbl)) {
1204                                 error = EINVAL;
1205                                 break;
1206                         }
1207                         size = sopt->sopt_valsize;
1208                         tbl = malloc(size, M_TEMP, M_WAITOK);
1209                         error = sooptcopyin(sopt, tbl, size, sizeof(*tbl));
1210                         if (error) {
1211                                 free(tbl, M_TEMP);
1212                                 break;
1213                         }
1214                         tbl->size = (size - sizeof(*tbl)) /
1215                             sizeof(ipfw_table_entry);
1216                         IPFW_RLOCK(chain);
1217                         error = ipfw_dump_table(chain, tbl);
1218                         IPFW_RUNLOCK(chain);
1219                         if (error) {
1220                                 free(tbl, M_TEMP);
1221                                 break;
1222                         }
1223                         error = sooptcopyout(sopt, tbl, size);
1224                         free(tbl, M_TEMP);
1225                 }
1226                 break;
1227
1228         case IP_FW_TABLE_XGETSIZE: /* IP_FW3 */
1229                 {
1230                         uint32_t *tbl;
1231
1232                         if (IP_FW3_OPLENGTH(sopt) < sizeof(uint32_t)) {
1233                                 error = EINVAL;
1234                                 break;
1235                         }
1236
1237                         tbl = (uint32_t *)(op3 + 1);
1238
1239                         IPFW_RLOCK(chain);
1240                         error = ipfw_count_xtable(chain, *tbl, tbl);
1241                         IPFW_RUNLOCK(chain);
1242                         if (error)
1243                                 break;
1244                         error = sooptcopyout(sopt, op3, sopt->sopt_valsize);
1245                 }
1246                 break;
1247
1248         case IP_FW_TABLE_XLIST: /* IP_FW3 */
1249                 {
1250                         ipfw_xtable *tbl;
1251
1252                         if ((size = valsize) < sizeof(ipfw_xtable)) {
1253                                 error = EINVAL;
1254                                 break;
1255                         }
1256
1257                         tbl = malloc(size, M_TEMP, M_ZERO | M_WAITOK);
1258                         memcpy(tbl, op3, sizeof(ipfw_xtable));
1259
1260                         /* Get maximum number of entries we can store */
1261                         tbl->size = (size - sizeof(ipfw_xtable)) /
1262                             sizeof(ipfw_table_xentry);
1263                         IPFW_RLOCK(chain);
1264                         error = ipfw_dump_xtable(chain, tbl);
1265                         IPFW_RUNLOCK(chain);
1266                         if (error) {
1267                                 free(tbl, M_TEMP);
1268                                 break;
1269                         }
1270
1271                         /* Revert size field back to bytes */
1272                         tbl->size = tbl->size * sizeof(ipfw_table_xentry) +
1273                                 sizeof(ipfw_table);
1274                         /*
1275                          * Since we call sooptcopyin() with small buffer, sopt_valsize is
1276                          * decreased to reflect supplied buffer size. Set it back to original value
1277                          */
1278                         sopt->sopt_valsize = valsize;
1279                         error = sooptcopyout(sopt, tbl, size);
1280                         free(tbl, M_TEMP);
1281                 }
1282                 break;
1283
1284         /*--- NAT operations are protected by the IPFW_LOCK ---*/
1285         case IP_FW_NAT_CFG:
1286                 if (IPFW_NAT_LOADED)
1287                         error = ipfw_nat_cfg_ptr(sopt);
1288                 else {
1289                         printf("IP_FW_NAT_CFG: %s\n",
1290                             "ipfw_nat not present, please load it");
1291                         error = EINVAL;
1292                 }
1293                 break;
1294
1295         case IP_FW_NAT_DEL:
1296                 if (IPFW_NAT_LOADED)
1297                         error = ipfw_nat_del_ptr(sopt);
1298                 else {
1299                         printf("IP_FW_NAT_DEL: %s\n",
1300                             "ipfw_nat not present, please load it");
1301                         error = EINVAL;
1302                 }
1303                 break;
1304
1305         case IP_FW_NAT_GET_CONFIG:
1306                 if (IPFW_NAT_LOADED)
1307                         error = ipfw_nat_get_cfg_ptr(sopt);
1308                 else {
1309                         printf("IP_FW_NAT_GET_CFG: %s\n",
1310                             "ipfw_nat not present, please load it");
1311                         error = EINVAL;
1312                 }
1313                 break;
1314
1315         case IP_FW_NAT_GET_LOG:
1316                 if (IPFW_NAT_LOADED)
1317                         error = ipfw_nat_get_log_ptr(sopt);
1318                 else {
1319                         printf("IP_FW_NAT_GET_LOG: %s\n",
1320                             "ipfw_nat not present, please load it");
1321                         error = EINVAL;
1322                 }
1323                 break;
1324
1325         default:
1326                 printf("ipfw: ipfw_ctl invalid option %d\n", sopt->sopt_name);
1327                 error = EINVAL;
1328         }
1329
1330         return (error);
1331 #undef RULE_MAXSIZE
1332 }
1333
1334
1335 #define RULE_MAXSIZE    (256*sizeof(u_int32_t))
1336
1337 /* Functions to convert rules 7.2 <==> 8.0 */
1338 int
1339 convert_rule_to_7(struct ip_fw *rule)
1340 {
1341         /* Used to modify original rule */
1342         struct ip_fw7 *rule7 = (struct ip_fw7 *)rule;
1343         /* copy of original rule, version 8 */
1344         struct ip_fw *tmp;
1345
1346         /* Used to copy commands */
1347         ipfw_insn *ccmd, *dst;
1348         int ll = 0, ccmdlen = 0;
1349
1350         tmp = malloc(RULE_MAXSIZE, M_TEMP, M_NOWAIT | M_ZERO);
1351         if (tmp == NULL) {
1352                 return 1; //XXX error
1353         }
1354         bcopy(rule, tmp, RULE_MAXSIZE);
1355
1356         /* Copy fields */
1357         rule7->_pad = tmp->_pad;
1358         rule7->set = tmp->set;
1359         rule7->rulenum = tmp->rulenum;
1360         rule7->cmd_len = tmp->cmd_len;
1361         rule7->act_ofs = tmp->act_ofs;
1362         rule7->next_rule = (struct ip_fw7 *)tmp->next_rule;
1363         rule7->next = (struct ip_fw7 *)tmp->x_next;
1364         rule7->cmd_len = tmp->cmd_len;
1365         rule7->pcnt = tmp->pcnt;
1366         rule7->bcnt = tmp->bcnt;
1367         rule7->timestamp = tmp->timestamp;
1368
1369         /* Copy commands */
1370         for (ll = tmp->cmd_len, ccmd = tmp->cmd, dst = rule7->cmd ;
1371                         ll > 0 ; ll -= ccmdlen, ccmd += ccmdlen, dst += ccmdlen) {
1372                 ccmdlen = F_LEN(ccmd);
1373
1374                 bcopy(ccmd, dst, F_LEN(ccmd)*sizeof(uint32_t));
1375
1376                 if (dst->opcode > O_NAT)
1377                         /* O_REASS doesn't exists in 7.2 version, so
1378                          * decrement opcode if it is after O_REASS
1379                          */
1380                         dst->opcode--;
1381
1382                 if (ccmdlen > ll) {
1383                         printf("ipfw: opcode %d size truncated\n",
1384                                 ccmd->opcode);
1385                         return EINVAL;
1386                 }
1387         }
1388         free(tmp, M_TEMP);
1389
1390         return 0;
1391 }
1392
1393 int
1394 convert_rule_to_8(struct ip_fw *rule)
1395 {
1396         /* Used to modify original rule */
1397         struct ip_fw7 *rule7 = (struct ip_fw7 *) rule;
1398
1399         /* Used to copy commands */
1400         ipfw_insn *ccmd, *dst;
1401         int ll = 0, ccmdlen = 0;
1402
1403         /* Copy of original rule */
1404         struct ip_fw7 *tmp = malloc(RULE_MAXSIZE, M_TEMP, M_NOWAIT | M_ZERO);
1405         if (tmp == NULL) {
1406                 return 1; //XXX error
1407         }
1408
1409         bcopy(rule7, tmp, RULE_MAXSIZE);
1410
1411         for (ll = tmp->cmd_len, ccmd = tmp->cmd, dst = rule->cmd ;
1412                         ll > 0 ; ll -= ccmdlen, ccmd += ccmdlen, dst += ccmdlen) {
1413                 ccmdlen = F_LEN(ccmd);
1414
1415                 bcopy(ccmd, dst, F_LEN(ccmd)*sizeof(uint32_t));
1416
1417                 if (dst->opcode > O_NAT)
1418                         /* O_REASS doesn't exists in 7.2 version, so
1419                          * increment opcode if it is after O_REASS
1420                          */
1421                         dst->opcode++;
1422
1423                 if (ccmdlen > ll) {
1424                         printf("ipfw: opcode %d size truncated\n",
1425                             ccmd->opcode);
1426                         return EINVAL;
1427                 }
1428         }
1429
1430         rule->_pad = tmp->_pad;
1431         rule->set = tmp->set;
1432         rule->rulenum = tmp->rulenum;
1433         rule->cmd_len = tmp->cmd_len;
1434         rule->act_ofs = tmp->act_ofs;
1435         rule->next_rule = (struct ip_fw *)tmp->next_rule;
1436         rule->x_next = (struct ip_fw *)tmp->next;
1437         rule->cmd_len = tmp->cmd_len;
1438         rule->id = 0; /* XXX see if is ok = 0 */
1439         rule->pcnt = tmp->pcnt;
1440         rule->bcnt = tmp->bcnt;
1441         rule->timestamp = tmp->timestamp;
1442
1443         free (tmp, M_TEMP);
1444         return 0;
1445 }
1446
1447 /* end of file */