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