sbin/ipfw/nat.c

   1 /*-
   2  * Copyright (c) 2002-2003 Luigi Rizzo
   3  * Copyright (c) 1996 Alex Nash, Paul Traina, Poul-Henning Kamp
   4  * Copyright (c) 1994 Ugen J.S.Antsilevich
   5  *
   6  * Idea and grammar partially left from:
   7  * Copyright (c) 1993 Daniel Boulet
   8  *
   9  * Redistribution and use in source forms, with and without modification,
  10  * are permitted provided that this entire comment appears intact.
  11  *
  12  * Redistribution in binary form may occur without any restrictions.
  13  * Obviously, it would be nice if you gave credit where credit is due
  14  * but requiring it would be too onerous.
  15  *
  16  * This software is provided ``AS IS'' without any warranties of any kind.
  17  *
  18  * NEW command line interface for IP firewall facility
  19  *
  20  * $FreeBSD$
  21  *
  22  * In-kernel nat support
  23  */
  24
  25 #include <sys/types.h>
  26 #include <sys/socket.h>
  27 #include <sys/sysctl.h>
  28
  29 #include "ipfw2.h"
  30
  31 #include <ctype.h>
  32 #include <err.h>
  33 #include <errno.h>
  34 #include <netdb.h>
  35 #include <stdio.h>
  36 #include <stdlib.h>
  37 #include <string.h>
  38 #include <sysexits.h>
  39
  40 #include <net/if.h>
  41 #include <net/if_dl.h>
  42 #include <net/route.h> /* def. of struct route */
  43 #include <netinet/in.h>
  44 #include <netinet/ip_fw.h>
  45 #include <arpa/inet.h>
  46 #include <alias.h>
  47
  48 typedef int (nat_cb_t)(struct nat44_cfg_nat *cfg, void *arg);
  49 static void nat_show_cfg(struct nat44_cfg_nat *n, void *arg);
  50 static void nat_show_log(struct nat44_cfg_nat *n, void *arg);
  51 static int nat_show_data(struct nat44_cfg_nat *cfg, void *arg);
  52 static int natname_cmp(const void *a, const void *b);
  53 static int nat_foreach(nat_cb_t *f, void *arg, int sort);
  54 static int nat_get_cmd(char *name, uint16_t cmd, ipfw_obj_header **ooh);
  55
  56 static struct _s_x nat_params[] = {
  57         { "ip",                 TOK_IP },
  58         { "if",                 TOK_IF },
  59         { "log",                TOK_ALOG },
  60         { "deny_in",            TOK_DENY_INC },
  61         { "same_ports",         TOK_SAME_PORTS },
  62         { "unreg_only",         TOK_UNREG_ONLY },
  63         { "unreg_cgn",          TOK_UNREG_CGN },
  64         { "skip_global",        TOK_SKIP_GLOBAL },
  65         { "reset",              TOK_RESET_ADDR },
  66         { "reverse",            TOK_ALIAS_REV },
  67         { "proxy_only",         TOK_PROXY_ONLY },
  68         { "port_range",         TOK_PORT_ALIAS },
  69         { "redirect_addr",      TOK_REDIR_ADDR },
  70         { "redirect_port",      TOK_REDIR_PORT },
  71         { "redirect_proto",     TOK_REDIR_PROTO },
  72         { NULL, 0 }     /* terminator */
  73 };
  74
  75
  76 /*
  77  * Search for interface with name "ifn", and fill n accordingly:
  78  *
  79  * n->ip        ip address of interface "ifn"
  80  * n->if_name   copy of interface name "ifn"
  81  */
  82 static void
  83 set_addr_dynamic(const char *ifn, struct nat44_cfg_nat *n)
  84 {
  85         size_t needed;
  86         int mib[6];
  87         char *buf, *lim, *next;
  88         struct if_msghdr *ifm;
  89         struct ifa_msghdr *ifam;
  90         struct sockaddr_dl *sdl;
  91         struct sockaddr_in *sin;
  92         int ifIndex;
  93
  94         mib[0] = CTL_NET;
  95         mib[1] = PF_ROUTE;
  96         mib[2] = 0;
  97         mib[3] = AF_INET;
  98         mib[4] = NET_RT_IFLIST;
  99         mib[5] = 0;
 100 /*
 101  * Get interface data.
 102  */
 103         if (sysctl(mib, 6, NULL, &needed, NULL, 0) == -1)
 104                 err(1, "iflist-sysctl-estimate");
 105         buf = safe_calloc(1, needed);
 106         if (sysctl(mib, 6, buf, &needed, NULL, 0) == -1)
 107                 err(1, "iflist-sysctl-get");
 108         lim = buf + needed;
 109 /*
 110  * Loop through interfaces until one with
 111  * given name is found. This is done to
 112  * find correct interface index for routing
 113  * message processing.
 114  */
 115         ifIndex = 0;
 116         next = buf;
 117         while (next < lim) {
 118                 ifm = (struct if_msghdr *)next;
 119                 next += ifm->ifm_msglen;
 120                 if (ifm->ifm_version != RTM_VERSION) {
 121                         if (g_co.verbose)
 122                                 warnx("routing message version %d "
 123                                     "not understood", ifm->ifm_version);
 124                         continue;
 125                 }
 126                 if (ifm->ifm_type == RTM_IFINFO) {
 127                         sdl = (struct sockaddr_dl *)(ifm + 1);
 128                         if (strlen(ifn) == sdl->sdl_nlen &&
 129                             strncmp(ifn, sdl->sdl_data, sdl->sdl_nlen) == 0) {
 130                                 ifIndex = ifm->ifm_index;
 131                                 break;
 132                         }
 133                 }
 134         }
 135         if (!ifIndex)
 136                 errx(1, "unknown interface name %s", ifn);
 137 /*
 138  * Get interface address.
 139  */
 140         sin = NULL;
 141         while (next < lim) {
 142                 ifam = (struct ifa_msghdr *)next;
 143                 next += ifam->ifam_msglen;
 144                 if (ifam->ifam_version != RTM_VERSION) {
 145                         if (g_co.verbose)
 146                                 warnx("routing message version %d "
 147                                     "not understood", ifam->ifam_version);
 148                         continue;
 149                 }
 150                 if (ifam->ifam_type != RTM_NEWADDR)
 151                         break;
 152                 if (ifam->ifam_addrs & RTA_IFA) {
 153                         int i;
 154                         char *cp = (char *)(ifam + 1);
 155
 156                         for (i = 1; i < RTA_IFA; i <<= 1) {
 157                                 if (ifam->ifam_addrs & i)
 158                                         cp += SA_SIZE((struct sockaddr *)cp);
 159                         }
 160                         if (((struct sockaddr *)cp)->sa_family == AF_INET) {
 161                                 sin = (struct sockaddr_in *)cp;
 162                                 break;
 163                         }
 164                 }
 165         }
 166         if (sin == NULL)
 167                 n->ip.s_addr = htonl(INADDR_ANY);
 168         else
 169                 n->ip = sin->sin_addr;
 170         strncpy(n->if_name, ifn, IF_NAMESIZE);
 171
 172         free(buf);
 173 }
 174
 175 /*
 176  * XXX - The following functions, macros and definitions come from natd.c:
 177  * it would be better to move them outside natd.c, in a file
 178  * (redirect_support.[ch]?) shared by ipfw and natd, but for now i can live
 179  * with it.
 180  */
 181
 182 /*
 183  * Definition of a port range, and macros to deal with values.
 184  * FORMAT:  HI 16-bits == first port in range, 0 == all ports.
 185  *        LO 16-bits == number of ports in range
 186  * NOTES:   - Port values are not stored in network byte order.
 187  */
 188
 189 #define port_range u_long
 190
 191 #define GETLOPORT(x)    ((x) >> 0x10)
 192 #define GETNUMPORTS(x)  ((x) & 0x0000ffff)
 193 #define GETHIPORT(x)    (GETLOPORT((x)) + GETNUMPORTS((x)))
 194
 195 /* Set y to be the low-port value in port_range variable x. */
 196 #define SETLOPORT(x,y)   ((x) = ((x) & 0x0000ffff) | ((y) << 0x10))
 197
 198 /* Set y to be the number of ports in port_range variable x. */
 199 #define SETNUMPORTS(x,y) ((x) = ((x) & 0xffff0000) | (y))
 200
 201 static void
 202 StrToAddr (const char* str, struct in_addr* addr)
 203 {
 204         struct hostent* hp;
 205
 206         if (inet_aton (str, addr))
 207                 return;
 208
 209         hp = gethostbyname (str);
 210         if (!hp)
 211                 errx (1, "unknown host %s", str);
 212
 213         memcpy (addr, hp->h_addr, sizeof (struct in_addr));
 214 }
 215
 216 static int
 217 StrToPortRange (const char* str, const char* proto, port_range *portRange)
 218 {
 219         char*      sep;
 220         struct servent* sp;
 221         char*           end;
 222         u_short  loPort;
 223         u_short  hiPort;
 224
 225         /* First see if this is a service, return corresponding port if so. */
 226         sp = getservbyname (str,proto);
 227         if (sp) {
 228                 SETLOPORT(*portRange, ntohs(sp->s_port));
 229                 SETNUMPORTS(*portRange, 1);
 230                 return 0;
 231         }
 232
 233         /* Not a service, see if it's a single port or port range. */
 234         sep = strchr (str, '-');
 235         if (sep == NULL) {
 236                 SETLOPORT(*portRange, strtol(str, &end, 10));
 237                 if (end != str) {
 238                         /* Single port. */
 239                         SETNUMPORTS(*portRange, 1);
 240                         return 0;
 241                 }
 242
 243                 /* Error in port range field. */
 244                 errx (EX_DATAERR, "%s/%s: unknown service", str, proto);
 245         }
 246
 247         /* Port range, get the values and sanity check. */
 248         sscanf (str, "%hu-%hu", &loPort, &hiPort);
 249         SETLOPORT(*portRange, loPort);
 250         SETNUMPORTS(*portRange, 0);     /* Error by default */
 251         if (loPort <= hiPort)
 252                 SETNUMPORTS(*portRange, hiPort - loPort + 1);
 253
 254         if (GETNUMPORTS(*portRange) == 0)
 255                 errx (EX_DATAERR, "invalid port range %s", str);
 256
 257         return 0;
 258 }
 259
 260 static int
 261 StrToProto (const char* str)
 262 {
 263         if (!strcmp (str, "tcp"))
 264                 return IPPROTO_TCP;
 265
 266         if (!strcmp (str, "udp"))
 267                 return IPPROTO_UDP;
 268
 269         if (!strcmp (str, "sctp"))
 270                 return IPPROTO_SCTP;
 271         errx (EX_DATAERR, "unknown protocol %s. Expected sctp, tcp or udp", str);
 272 }
 273
 274 static int
 275 StrToAddrAndPortRange (const char* str, struct in_addr* addr, char* proto,
 276                         port_range *portRange)
 277 {
 278         char*   ptr;
 279
 280         ptr = strchr (str, ':');
 281         if (!ptr)
 282                 errx (EX_DATAERR, "%s is missing port number", str);
 283
 284         *ptr = '\0';
 285         ++ptr;
 286
 287         StrToAddr (str, addr);
 288         return StrToPortRange (ptr, proto, portRange);
 289 }
 290
 291 /* End of stuff taken from natd.c. */
 292
 293 /*
 294  * The next 3 functions add support for the addr, port and proto redirect and
 295  * their logic is loosely based on SetupAddressRedirect(), SetupPortRedirect()
 296  * and SetupProtoRedirect() from natd.c.
 297  *
 298  * Every setup_* function fills at least one redirect entry
 299  * (struct nat44_cfg_redir) and zero or more server pool entry
 300  * (struct nat44_cfg_spool) in buf.
 301  *
 302  * The format of data in buf is:
 303  *
 304  *  nat44_cfg_nat nat44_cfg_redir nat44_cfg_spool    ......  nat44_cfg_spool
 305  *
 306  *    -------------------------------------        ------------
 307  *   |           | .....X ..... |          |         |           |  .....
 308  *    ------------------------------------- ...... ------------
 309  *                     ^
 310  *                spool_cnt       n=0       ......   n=(X-1)
 311  *
 312  * len points to the amount of available space in buf
 313  * space counts the memory consumed by every function
 314  *
 315  * XXX - Every function get all the argv params so it
 316  * has to check, in optional parameters, that the next
 317  * args is a valid option for the redir entry and not
 318  * another token. Only redir_port and redir_proto are
 319  * affected by this.
 320  */
 321
 322 static int
 323 estimate_redir_addr(int *ac, char ***av)
 324 {
 325         size_t space = sizeof(struct nat44_cfg_redir);
 326         char *sep = **av;
 327         u_int c = 0;
 328
 329         (void)ac;       /* UNUSED */
 330         while ((sep = strchr(sep, ',')) != NULL) {
 331                 c++;
 332                 sep++;
 333         }
 334
 335         if (c > 0)
 336                 c++;
 337
 338         space += c * sizeof(struct nat44_cfg_spool);
 339
 340         return (space);
 341 }
 342
 343 static int
 344 setup_redir_addr(char *buf, int *ac, char ***av)
 345 {
 346         struct nat44_cfg_redir *r;
 347         char *sep;
 348         size_t space;
 349
 350         r = (struct nat44_cfg_redir *)buf;
 351         r->mode = REDIR_ADDR;
 352         /* Skip nat44_cfg_redir at beginning of buf. */
 353         buf = &buf[sizeof(struct nat44_cfg_redir)];
 354         space = sizeof(struct nat44_cfg_redir);
 355
 356         /* Extract local address. */
 357         if (strchr(**av, ',') != NULL) {
 358                 struct nat44_cfg_spool *spool;
 359
 360                 /* Setup LSNAT server pool. */
 361                 r->laddr.s_addr = INADDR_NONE;
 362                 sep = strtok(**av, ",");
 363                 while (sep != NULL) {
 364                         spool = (struct nat44_cfg_spool *)buf;
 365                         space += sizeof(struct nat44_cfg_spool);
 366                         StrToAddr(sep, &spool->addr);
 367                         spool->port = ~0;
 368                         r->spool_cnt++;
 369                         /* Point to the next possible nat44_cfg_spool. */
 370                         buf = &buf[sizeof(struct nat44_cfg_spool)];
 371                         sep = strtok(NULL, ",");
 372                 }
 373         } else
 374                 StrToAddr(**av, &r->laddr);
 375         (*av)++; (*ac)--;
 376
 377         /* Extract public address. */
 378         StrToAddr(**av, &r->paddr);
 379         (*av)++; (*ac)--;
 380
 381         return (space);
 382 }
 383
 384 static int
 385 estimate_redir_port(int *ac, char ***av)
 386 {
 387         size_t space = sizeof(struct nat44_cfg_redir);
 388         char *sep = **av;
 389         u_int c = 0;
 390
 391         (void)ac;       /* UNUSED */
 392         while ((sep = strchr(sep, ',')) != NULL) {
 393                 c++;
 394                 sep++;
 395         }
 396
 397         if (c > 0)
 398                 c++;
 399
 400         space += c * sizeof(struct nat44_cfg_spool);
 401
 402         return (space);
 403 }
 404
 405 static int
 406 setup_redir_port(char *buf, int *ac, char ***av)
 407 {
 408         struct nat44_cfg_redir *r;
 409         char *sep, *protoName, *lsnat = NULL;
 410         size_t space;
 411         u_short numLocalPorts;
 412         port_range portRange;
 413
 414         numLocalPorts = 0;
 415
 416         r = (struct nat44_cfg_redir *)buf;
 417         r->mode = REDIR_PORT;
 418         /* Skip nat44_cfg_redir at beginning of buf. */
 419         buf = &buf[sizeof(struct nat44_cfg_redir)];
 420         space = sizeof(struct nat44_cfg_redir);
 421
 422         /*
 423          * Extract protocol.
 424          */
 425         r->proto = StrToProto(**av);
 426         protoName = **av;
 427         (*av)++; (*ac)--;
 428
 429         /*
 430          * Extract local address.
 431          */
 432         if (strchr(**av, ',') != NULL) {
 433                 r->laddr.s_addr = INADDR_NONE;
 434                 r->lport = ~0;
 435                 numLocalPorts = 1;
 436                 lsnat = **av;
 437         } else {
 438                 /*
 439                  * The sctp nat does not allow the port numbers to be mapped to
 440                  * new port numbers. Therefore, no ports are to be specified
 441                  * in the target port field.
 442                  */
 443                 if (r->proto == IPPROTO_SCTP) {
 444                         if (strchr(**av, ':'))
 445                                 errx(EX_DATAERR, "redirect_port:"
 446                                     "port numbers do not change in sctp, so do "
 447                                     "not specify them as part of the target");
 448                         else
 449                                 StrToAddr(**av, &r->laddr);
 450                 } else {
 451                         if (StrToAddrAndPortRange(**av, &r->laddr, protoName,
 452                             &portRange) != 0)
 453                                 errx(EX_DATAERR, "redirect_port: "
 454                                     "invalid local port range");
 455
 456                         r->lport = GETLOPORT(portRange);
 457                         numLocalPorts = GETNUMPORTS(portRange);
 458                 }
 459         }
 460         (*av)++; (*ac)--;
 461
 462         /*
 463          * Extract public port and optionally address.
 464          */
 465         if (strchr(**av, ':') != NULL) {
 466                 if (StrToAddrAndPortRange(**av, &r->paddr, protoName,
 467                     &portRange) != 0)
 468                         errx(EX_DATAERR, "redirect_port: "
 469                             "invalid public port range");
 470         } else {
 471                 r->paddr.s_addr = INADDR_ANY;
 472                 if (StrToPortRange(**av, protoName, &portRange) != 0)
 473                         errx(EX_DATAERR, "redirect_port: "
 474                             "invalid public port range");
 475         }
 476
 477         r->pport = GETLOPORT(portRange);
 478         if (r->proto == IPPROTO_SCTP) { /* so the logic below still works */
 479                 numLocalPorts = GETNUMPORTS(portRange);
 480                 r->lport = r->pport;
 481         }
 482         r->pport_cnt = GETNUMPORTS(portRange);
 483         (*av)++; (*ac)--;
 484
 485         /*
 486          * Extract remote address and optionally port.
 487          */
 488         /*
 489          * NB: isdigit(**av) => we've to check that next parameter is really an
 490          * option for this redirect entry, else stop here processing arg[cv].
 491          */
 492         if (*ac != 0 && isdigit(***av)) {
 493                 if (strchr(**av, ':') != NULL) {
 494                         if (StrToAddrAndPortRange(**av, &r->raddr, protoName,
 495                             &portRange) != 0)
 496                                 errx(EX_DATAERR, "redirect_port: "
 497                                     "invalid remote port range");
 498                 } else {
 499                         SETLOPORT(portRange, 0);
 500                         SETNUMPORTS(portRange, 1);
 501                         StrToAddr(**av, &r->raddr);
 502                 }
 503                 (*av)++; (*ac)--;
 504         } else {
 505                 SETLOPORT(portRange, 0);
 506                 SETNUMPORTS(portRange, 1);
 507                 r->raddr.s_addr = INADDR_ANY;
 508         }
 509         r->rport = GETLOPORT(portRange);
 510         r->rport_cnt = GETNUMPORTS(portRange);
 511
 512         /*
 513          * Make sure port ranges match up, then add the redirect ports.
 514          */
 515         if (numLocalPorts != r->pport_cnt)
 516                 errx(EX_DATAERR, "redirect_port: "
 517                     "port ranges must be equal in size");
 518
 519         /* Remote port range is allowed to be '0' which means all ports. */
 520         if (r->rport_cnt != numLocalPorts &&
 521             (r->rport_cnt != 1 || r->rport != 0))
 522                 errx(EX_DATAERR, "redirect_port: remote port must"
 523                     "be 0 or equal to local port range in size");
 524
 525         /* Setup LSNAT server pool. */
 526         if (lsnat != NULL) {
 527                 struct nat44_cfg_spool *spool;
 528
 529                 sep = strtok(lsnat, ",");
 530                 while (sep != NULL) {
 531                         spool = (struct nat44_cfg_spool *)buf;
 532                         space += sizeof(struct nat44_cfg_spool);
 533                         /*
 534                          * The sctp nat does not allow the port numbers to
 535                          * be mapped to new port numbers. Therefore, no ports
 536                          * are to be specified in the target port field.
 537                          */
 538                         if (r->proto == IPPROTO_SCTP) {
 539                                 if (strchr (sep, ':')) {
 540                                         errx(EX_DATAERR, "redirect_port:"
 541                                             "port numbers do not change in "
 542                                             "sctp, so do not specify them as "
 543                                             "part of the target");
 544                                 } else {
 545                                         StrToAddr(sep, &spool->addr);
 546                                         spool->port = r->pport;
 547                                 }
 548                         } else {
 549                                 if (StrToAddrAndPortRange(sep, &spool->addr,
 550                                         protoName, &portRange) != 0)
 551                                         errx(EX_DATAERR, "redirect_port:"
 552                                             "invalid local port range");
 553                                 if (GETNUMPORTS(portRange) != 1)
 554                                         errx(EX_DATAERR, "redirect_port: "
 555                                             "local port must be single in "
 556                                             "this context");
 557                                 spool->port = GETLOPORT(portRange);
 558                         }
 559                         r->spool_cnt++;
 560                         /* Point to the next possible nat44_cfg_spool. */
 561                         buf = &buf[sizeof(struct nat44_cfg_spool)];
 562                         sep = strtok(NULL, ",");
 563                 }
 564         }
 565
 566         return (space);
 567 }
 568
 569 static int
 570 setup_redir_proto(char *buf, int *ac, char ***av)
 571 {
 572         struct nat44_cfg_redir *r;
 573         struct protoent *protoent;
 574         size_t space;
 575
 576         r = (struct nat44_cfg_redir *)buf;
 577         r->mode = REDIR_PROTO;
 578         /* Skip nat44_cfg_redir at beginning of buf. */
 579         buf = &buf[sizeof(struct nat44_cfg_redir)];
 580         space = sizeof(struct nat44_cfg_redir);
 581
 582         /*
 583          * Extract protocol.
 584          */
 585         protoent = getprotobyname(**av);
 586         if (protoent == NULL)
 587                 errx(EX_DATAERR, "redirect_proto: unknown protocol %s", **av);
 588         else
 589                 r->proto = protoent->p_proto;
 590
 591         (*av)++; (*ac)--;
 592
 593         /*
 594          * Extract local address.
 595          */
 596         StrToAddr(**av, &r->laddr);
 597
 598         (*av)++; (*ac)--;
 599
 600         /*
 601          * Extract optional public address.
 602          */
 603         if (*ac == 0) {
 604                 r->paddr.s_addr = INADDR_ANY;
 605                 r->raddr.s_addr = INADDR_ANY;
 606         } else {
 607                 /* see above in setup_redir_port() */
 608                 if (isdigit(***av)) {
 609                         StrToAddr(**av, &r->paddr);
 610                         (*av)++; (*ac)--;
 611
 612                         /*
 613                          * Extract optional remote address.
 614                          */
 615                         /* see above in setup_redir_port() */
 616                         if (*ac != 0 && isdigit(***av)) {
 617                                 StrToAddr(**av, &r->raddr);
 618                                 (*av)++; (*ac)--;
 619                         }
 620                 }
 621         }
 622
 623         return (space);
 624 }
 625
 626 static void
 627 nat_show_log(struct nat44_cfg_nat *n, void *arg __unused)
 628 {
 629         char *buf;
 630
 631         buf = (char *)(n + 1);
 632         if (buf[0] != '\0')
 633                 printf("nat %s: %s\n", n->name, buf);
 634 }
 635
 636 static void
 637 nat_show_cfg(struct nat44_cfg_nat *n, void *arg __unused)
 638 {
 639         struct nat44_cfg_redir *t;
 640         struct nat44_cfg_spool *s;
 641         caddr_t buf;
 642         struct protoent *p;
 643         uint32_t cnt;
 644         int i, off;
 645
 646         buf = (caddr_t)n;
 647         off = sizeof(*n);
 648         printf("ipfw nat %s config", n->name);
 649         if (strlen(n->if_name) != 0)
 650                 printf(" if %s", n->if_name);
 651         else if (n->ip.s_addr != 0)
 652                 printf(" ip %s", inet_ntoa(n->ip));
 653         while (n->mode != 0) {
 654                 if (n->mode & PKT_ALIAS_LOG) {
 655                         printf(" log");
 656                         n->mode &= ~PKT_ALIAS_LOG;
 657                 } else if (n->mode & PKT_ALIAS_DENY_INCOMING) {
 658                         printf(" deny_in");
 659                         n->mode &= ~PKT_ALIAS_DENY_INCOMING;
 660                 } else if (n->mode & PKT_ALIAS_SAME_PORTS) {
 661                         printf(" same_ports");
 662                         n->mode &= ~PKT_ALIAS_SAME_PORTS;
 663                 } else if (n->mode & PKT_ALIAS_SKIP_GLOBAL) {
 664                         printf(" skip_global");
 665                         n->mode &= ~PKT_ALIAS_SKIP_GLOBAL;
 666                 } else if (n->mode & PKT_ALIAS_UNREGISTERED_ONLY) {
 667                         printf(" unreg_only");
 668                         n->mode &= ~PKT_ALIAS_UNREGISTERED_ONLY;
 669                 } else if (n->mode & PKT_ALIAS_UNREGISTERED_CGN) {
 670                         printf(" unreg_cgn");
 671                         n->mode &= ~PKT_ALIAS_UNREGISTERED_CGN;
 672                 } else if (n->mode & PKT_ALIAS_RESET_ON_ADDR_CHANGE) {
 673                         printf(" reset");
 674                         n->mode &= ~PKT_ALIAS_RESET_ON_ADDR_CHANGE;
 675                 } else if (n->mode & PKT_ALIAS_REVERSE) {
 676                         printf(" reverse");
 677                         n->mode &= ~PKT_ALIAS_REVERSE;
 678                 } else if (n->mode & PKT_ALIAS_PROXY_ONLY) {
 679                         printf(" proxy_only");
 680                         n->mode &= ~PKT_ALIAS_PROXY_ONLY;
 681                 }
 682         }
 683         /* Print all the redirect's data configuration. */
 684         for (cnt = 0; cnt < n->redir_cnt; cnt++) {
 685                 t = (struct nat44_cfg_redir *)&buf[off];
 686                 off += sizeof(struct nat44_cfg_redir);
 687                 switch (t->mode) {
 688                 case REDIR_ADDR:
 689                         printf(" redirect_addr");
 690                         if (t->spool_cnt == 0)
 691                                 printf(" %s", inet_ntoa(t->laddr));
 692                         else
 693                                 for (i = 0; i < t->spool_cnt; i++) {
 694                                         s = (struct nat44_cfg_spool *)&buf[off];
 695                                         if (i)
 696                                                 printf(",");
 697                                         else
 698                                                 printf(" ");
 699                                         printf("%s", inet_ntoa(s->addr));
 700                                         off += sizeof(struct nat44_cfg_spool);
 701                                 }
 702                         printf(" %s", inet_ntoa(t->paddr));
 703                         break;
 704                 case REDIR_PORT:
 705                         p = getprotobynumber(t->proto);
 706                         printf(" redirect_port %s ", p->p_name);
 707                         if (!t->spool_cnt) {
 708                                 printf("%s:%u", inet_ntoa(t->laddr), t->lport);
 709                                 if (t->pport_cnt > 1)
 710                                         printf("-%u", t->lport +
 711                                             t->pport_cnt - 1);
 712                         } else
 713                                 for (i=0; i < t->spool_cnt; i++) {
 714                                         s = (struct nat44_cfg_spool *)&buf[off];
 715                                         if (i)
 716                                                 printf(",");
 717                                         printf("%s:%u", inet_ntoa(s->addr),
 718                                             s->port);
 719                                         off += sizeof(struct nat44_cfg_spool);
 720                                 }
 721
 722                         printf(" ");
 723                         if (t->paddr.s_addr)
 724                                 printf("%s:", inet_ntoa(t->paddr));
 725                         printf("%u", t->pport);
 726                         if (!t->spool_cnt && t->pport_cnt > 1)
 727                                 printf("-%u", t->pport + t->pport_cnt - 1);
 728
 729                         if (t->raddr.s_addr) {
 730                                 printf(" %s", inet_ntoa(t->raddr));
 731                                 if (t->rport) {
 732                                         printf(":%u", t->rport);
 733                                         if (!t->spool_cnt && t->rport_cnt > 1)
 734                                                 printf("-%u", t->rport +
 735                                                     t->rport_cnt - 1);
 736                                 }
 737                         }
 738                         break;
 739                 case REDIR_PROTO:
 740                         p = getprotobynumber(t->proto);
 741                         printf(" redirect_proto %s %s", p->p_name,
 742                             inet_ntoa(t->laddr));
 743                         if (t->paddr.s_addr != 0) {
 744                                 printf(" %s", inet_ntoa(t->paddr));
 745                                 if (t->raddr.s_addr)
 746                                         printf(" %s", inet_ntoa(t->raddr));
 747                         }
 748                         break;
 749                 default:
 750                         errx(EX_DATAERR, "unknown redir mode");
 751                         break;
 752                 }
 753         }
 754         printf("\n");
 755 }
 756
 757 static int
 758 nat_port_alias_parse(char *str, u_short *lpout, u_short *hpout) {
 759         long lp, hp;
 760         char *ptr;
 761         /* Lower port parsing */
 762         lp = (long) strtol(str, &ptr, 10);
 763         if (lp < 1024 || lp > 65535)
 764                 return 0;
 765         if (!ptr || *ptr != '-')
 766                 return 0;
 767         /* Upper port parsing */
 768         hp = (long) strtol(ptr, &ptr, 10);
 769         if (hp < 1024 || hp > 65535)
 770                 return 0;
 771         if (ptr)
 772                 return 0;
 773
 774         *lpout = (u_short) lp;
 775         *hpout = (u_short) hp;
 776         return 1;
 777 }
 778
 779 void
 780 ipfw_config_nat(int ac, char **av)
 781 {
 782         ipfw_obj_header *oh;
 783         struct nat44_cfg_nat *n;                /* Nat instance configuration. */
 784         int i, off, tok, ac1;
 785         u_short lp, hp;
 786         char *id, *buf, **av1, *end;
 787         size_t len;
 788
 789         av++;
 790         ac--;
 791         /* Nat id. */
 792         if (ac == 0)
 793                 errx(EX_DATAERR, "missing nat id");
 794         id = *av;
 795         i = (int)strtol(id, &end, 0);
 796         if (i <= 0 || *end != '\0')
 797                 errx(EX_DATAERR, "illegal nat id: %s", id);
 798         av++;
 799         ac--;
 800         if (ac == 0)
 801                 errx(EX_DATAERR, "missing option");
 802
 803         len = sizeof(*oh) + sizeof(*n);
 804         ac1 = ac;
 805         av1 = av;
 806         while (ac1 > 0) {
 807                 tok = match_token(nat_params, *av1);
 808                 ac1--;
 809                 av1++;
 810                 switch (tok) {
 811                 case TOK_IP:
 812                 case TOK_IF:
 813                 case TOK_PORT_ALIAS:
 814                         ac1--;
 815                         av1++;
 816                         break;
 817                 case TOK_ALOG:
 818                 case TOK_DENY_INC:
 819                 case TOK_SAME_PORTS:
 820                 case TOK_SKIP_GLOBAL:
 821                 case TOK_UNREG_ONLY:
 822                 case TOK_UNREG_CGN:
 823                 case TOK_RESET_ADDR:
 824                 case TOK_ALIAS_REV:
 825                 case TOK_PROXY_ONLY:
 826                         break;
 827                 case TOK_REDIR_ADDR:
 828                         if (ac1 < 2)
 829                                 errx(EX_DATAERR, "redirect_addr: "
 830                                     "not enough arguments");
 831                         len += estimate_redir_addr(&ac1, &av1);
 832                         av1 += 2;
 833                         ac1 -= 2;
 834                         break;
 835                 case TOK_REDIR_PORT:
 836                         if (ac1 < 3)
 837                                 errx(EX_DATAERR, "redirect_port: "
 838                                     "not enough arguments");
 839                         av1++;
 840                         ac1--;
 841                         len += estimate_redir_port(&ac1, &av1);
 842                         av1 += 2;
 843                         ac1 -= 2;
 844                         /* Skip optional remoteIP/port */
 845                         if (ac1 != 0 && isdigit(**av1)) {
 846                                 av1++;
 847                                 ac1--;
 848                         }
 849                         break;
 850                 case TOK_REDIR_PROTO:
 851                         if (ac1 < 2)
 852                                 errx(EX_DATAERR, "redirect_proto: "
 853                                     "not enough arguments");
 854                         len += sizeof(struct nat44_cfg_redir);
 855                         av1 += 2;
 856                         ac1 -= 2;
 857                         /* Skip optional remoteIP/port */
 858                         if (ac1 != 0 && isdigit(**av1)) {
 859                                 av1++;
 860                                 ac1--;
 861                         }
 862                         if (ac1 != 0 && isdigit(**av1)) {
 863                                 av1++;
 864                                 ac1--;
 865                         }
 866                         break;
 867                 default:
 868                         errx(EX_DATAERR, "unrecognised option ``%s''", av1[-1]);
 869                 }
 870         }
 871
 872         if ((buf = malloc(len)) == NULL)
 873                 errx(EX_OSERR, "malloc failed");
 874
 875         /* Offset in buf: save space for header at the beginning. */
 876         off = sizeof(*oh) + sizeof(*n);
 877         memset(buf, 0, len);
 878         oh = (ipfw_obj_header *)buf;
 879         n = (struct nat44_cfg_nat *)(oh + 1);
 880         oh->ntlv.head.length = sizeof(oh->ntlv);
 881         snprintf(oh->ntlv.name, sizeof(oh->ntlv.name), "%d", i);
 882         snprintf(n->name, sizeof(n->name), "%d", i);
 883
 884         while (ac > 0) {
 885                 tok = match_token(nat_params, *av);
 886                 ac--;
 887                 av++;
 888                 switch (tok) {
 889                 case TOK_IP:
 890                         if (ac == 0)
 891                                 errx(EX_DATAERR, "missing option");
 892                         if (!inet_aton(av[0], &(n->ip)))
 893                                 errx(EX_DATAERR, "bad ip address ``%s''",
 894                                     av[0]);
 895                         ac--;
 896                         av++;
 897                         break;
 898                 case TOK_IF:
 899                         if (ac == 0)
 900                                 errx(EX_DATAERR, "missing option");
 901                         set_addr_dynamic(av[0], n);
 902                         ac--;
 903                         av++;
 904                         break;
 905                 case TOK_ALOG:
 906                         n->mode |= PKT_ALIAS_LOG;
 907                         break;
 908                 case TOK_DENY_INC:
 909                         n->mode |= PKT_ALIAS_DENY_INCOMING;
 910                         break;
 911                 case TOK_SAME_PORTS:
 912                         n->mode |= PKT_ALIAS_SAME_PORTS;
 913                         break;
 914                 case TOK_UNREG_ONLY:
 915                         n->mode |= PKT_ALIAS_UNREGISTERED_ONLY;
 916                         break;
 917                 case TOK_UNREG_CGN:
 918                         n->mode |= PKT_ALIAS_UNREGISTERED_CGN;
 919                         break;
 920                 case TOK_SKIP_GLOBAL:
 921                         n->mode |= PKT_ALIAS_SKIP_GLOBAL;
 922                         break;
 923                 case TOK_RESET_ADDR:
 924                         n->mode |= PKT_ALIAS_RESET_ON_ADDR_CHANGE;
 925                         break;
 926                 case TOK_ALIAS_REV:
 927                         n->mode |= PKT_ALIAS_REVERSE;
 928                         break;
 929                 case TOK_PROXY_ONLY:
 930                         n->mode |= PKT_ALIAS_PROXY_ONLY;
 931                         break;
 932                         /*
 933                          * All the setup_redir_* functions work directly in
 934                          * the final buffer, see above for details.
 935                          */
 936                 case TOK_REDIR_ADDR:
 937                 case TOK_REDIR_PORT:
 938                 case TOK_REDIR_PROTO:
 939                         switch (tok) {
 940                         case TOK_REDIR_ADDR:
 941                                 i = setup_redir_addr(&buf[off], &ac, &av);
 942                                 break;
 943                         case TOK_REDIR_PORT:
 944                                 i = setup_redir_port(&buf[off], &ac, &av);
 945                                 break;
 946                         case TOK_REDIR_PROTO:
 947                                 i = setup_redir_proto(&buf[off], &ac, &av);
 948                                 break;
 949                         }
 950                         n->redir_cnt++;
 951                         off += i;
 952                         break;
 953                 case TOK_PORT_ALIAS:
 954                         if (ac == 0)
 955                                 errx(EX_DATAERR, "missing option");
 956                         if (!nat_port_alias_parse(av[0], &lp, &hp))
 957                                 errx(EX_DATAERR,
 958                                     "You need a range of port(s) from 1024 <= x < 65536");
 959                         if (lp >= hp)
 960                                 errx(EX_DATAERR,
 961                                     "Upper port has to be greater than lower port");
 962                         n->alias_port_lo = lp;
 963                         n->alias_port_hi = hp;
 964                         ac--;
 965                         av++;
 966                         break;
 967                 }
 968         }
 969         if (n->mode & PKT_ALIAS_SAME_PORTS && n->alias_port_lo)
 970                 errx(EX_DATAERR, "same_ports and port_range cannot both be selected");
 971
 972         i = do_set3(IP_FW_NAT44_XCONFIG, &oh->opheader, len);
 973         if (i != 0)
 974                 err(1, "setsockopt(%s)", "IP_FW_NAT44_XCONFIG");
 975
 976         if (!g_co.do_quiet) {
 977                 /* After every modification, we show the resultant rule. */
 978                 int _ac = 3;
 979                 const char *_av[] = {"show", "config", id};
 980                 ipfw_show_nat(_ac, (char **)(void *)_av);
 981         }
 982 }
 983
 984 static void
 985 nat_fill_ntlv(ipfw_obj_ntlv *ntlv, int i)
 986 {
 987
 988         ntlv->head.type = IPFW_TLV_EACTION_NAME(1); /* it doesn't matter */
 989         ntlv->head.length = sizeof(ipfw_obj_ntlv);
 990         ntlv->idx = 1;
 991         ntlv->set = 0; /* not yet */
 992         snprintf(ntlv->name, sizeof(ntlv->name), "%d", i);
 993 }
 994
 995 int
 996 ipfw_delete_nat(int i)
 997 {
 998         ipfw_obj_header oh;
 999         int ret;
1000
1001         memset(&oh, 0, sizeof(oh));
1002         nat_fill_ntlv(&oh.ntlv, i);
1003         ret = do_set3(IP_FW_NAT44_DESTROY, &oh.opheader, sizeof(oh));
1004         if (ret == -1) {
1005                 if (!g_co.do_quiet)
1006                         warn("nat %u not available", i);
1007                 return (EX_UNAVAILABLE);
1008         }
1009         return (EX_OK);
1010 }
1011
1012 struct nat_list_arg {
1013         uint16_t        cmd;
1014         int             is_all;
1015 };
1016
1017 static int
1018 nat_show_data(struct nat44_cfg_nat *cfg, void *arg)
1019 {
1020         struct nat_list_arg *nla;
1021         ipfw_obj_header *oh;
1022
1023         nla = (struct nat_list_arg *)arg;
1024
1025         switch (nla->cmd) {
1026         case IP_FW_NAT44_XGETCONFIG:
1027                 if (nat_get_cmd(cfg->name, nla->cmd, &oh) != 0) {
1028                         warnx("Error getting nat instance %s info", cfg->name);
1029                         break;
1030                 }
1031                 nat_show_cfg((struct nat44_cfg_nat *)(oh + 1), NULL);
1032                 free(oh);
1033                 break;
1034         case IP_FW_NAT44_XGETLOG:
1035                 if (nat_get_cmd(cfg->name, nla->cmd, &oh) == 0) {
1036                         nat_show_log((struct nat44_cfg_nat *)(oh + 1), NULL);
1037                         free(oh);
1038                         break;
1039                 }
1040                 /* Handle error */
1041                 if (nla->is_all != 0 && errno == ENOENT)
1042                         break;
1043                 warn("Error getting nat instance %s info", cfg->name);
1044                 break;
1045         }
1046
1047         return (0);
1048 }
1049
1050 /*
1051  * Compare nat names.
1052  * Honor number comparison.
1053  */
1054 static int
1055 natname_cmp(const void *a, const void *b)
1056 {
1057         const struct nat44_cfg_nat *ia, *ib;
1058
1059         ia = (const struct nat44_cfg_nat *)a;
1060         ib = (const struct nat44_cfg_nat *)b;
1061
1062         return (stringnum_cmp(ia->name, ib->name));
1063 }
1064
1065 /*
1066  * Retrieves nat list from kernel,
1067  * optionally sorts it and calls requested function for each table.
1068  * Returns 0 on success.
1069  */
1070 static int
1071 nat_foreach(nat_cb_t *f, void *arg, int sort)
1072 {
1073         ipfw_obj_lheader *olh;
1074         struct nat44_cfg_nat *cfg;
1075         size_t sz;
1076         uint32_t i;
1077         int error;
1078
1079         /* Start with reasonable default */
1080         sz = sizeof(*olh) + 16 * sizeof(struct nat44_cfg_nat);
1081
1082         for (;;) {
1083                 if ((olh = calloc(1, sz)) == NULL)
1084                         return (ENOMEM);
1085
1086                 olh->size = sz;
1087                 if (do_get3(IP_FW_NAT44_LIST_NAT, &olh->opheader, &sz) != 0) {
1088                         sz = olh->size;
1089                         free(olh);
1090                         if (errno == ENOMEM)
1091                                 continue;
1092                         return (errno);
1093                 }
1094
1095                 if (sort != 0)
1096                         qsort(olh + 1, olh->count, olh->objsize, natname_cmp);
1097
1098                 cfg = (struct nat44_cfg_nat*)(olh + 1);
1099                 for (i = 0; i < olh->count; i++) {
1100                         error = f(cfg, arg); /* Ignore errors for now */
1101                         cfg = (struct nat44_cfg_nat *)((caddr_t)cfg +
1102                             olh->objsize);
1103                 }
1104
1105                 free(olh);
1106                 break;
1107         }
1108
1109         return (0);
1110 }
1111
1112 static int
1113 nat_get_cmd(char *name, uint16_t cmd, ipfw_obj_header **ooh)
1114 {
1115         ipfw_obj_header *oh;
1116         struct nat44_cfg_nat *cfg;
1117         size_t sz;
1118
1119         /* Start with reasonable default */
1120         sz = sizeof(*oh) + sizeof(*cfg) + 128;
1121
1122         for (;;) {
1123                 if ((oh = calloc(1, sz)) == NULL)
1124                         return (ENOMEM);
1125                 cfg = (struct nat44_cfg_nat *)(oh + 1);
1126                 oh->ntlv.head.length = sizeof(oh->ntlv);
1127                 strlcpy(oh->ntlv.name, name, sizeof(oh->ntlv.name));
1128                 strlcpy(cfg->name, name, sizeof(cfg->name));
1129
1130                 if (do_get3(cmd, &oh->opheader, &sz) != 0) {
1131                         sz = cfg->size;
1132                         free(oh);
1133                         if (errno == ENOMEM)
1134                                 continue;
1135                         return (errno);
1136                 }
1137
1138                 *ooh = oh;
1139                 break;
1140         }
1141
1142         return (0);
1143 }
1144
1145 void
1146 ipfw_show_nat(int ac, char **av)
1147 {
1148         ipfw_obj_header *oh;
1149         char *name;
1150         int cmd;
1151         struct nat_list_arg nla;
1152
1153         ac--;
1154         av++;
1155
1156         if (g_co.test_only)
1157                 return;
1158
1159         /* Parse parameters. */
1160         cmd = 0; /* XXX: Change to IP_FW_NAT44_XGETLOG @ MFC */
1161         name = NULL;
1162         for ( ; ac != 0; ac--, av++) {
1163                 if (!strncmp(av[0], "config", strlen(av[0]))) {
1164                         cmd = IP_FW_NAT44_XGETCONFIG;
1165                         continue;
1166                 }
1167                 if (strcmp(av[0], "log") == 0) {
1168                         cmd = IP_FW_NAT44_XGETLOG;
1169                         continue;
1170                 }
1171                 if (name != NULL)
1172                         err(EX_USAGE,"only one instance name may be specified");
1173                 name = av[0];
1174         }
1175
1176         if (cmd == 0)
1177                 errx(EX_USAGE, "Please specify action. Available: config,log");
1178
1179         if (name == NULL) {
1180                 memset(&nla, 0, sizeof(nla));
1181                 nla.cmd = cmd;
1182                 nla.is_all = 1;
1183                 nat_foreach(nat_show_data, &nla, 1);
1184         } else {
1185                 if (nat_get_cmd(name, cmd, &oh) != 0)
1186                         err(EX_OSERR, "Error getting nat %s instance info", name);
1187                 nat_show_cfg((struct nat44_cfg_nat *)(oh + 1), NULL);
1188                 free(oh);
1189         }
1190 }
1191