Home | History | Annotate | Line # | Download | only in npfctl
npfctl.c revision 1.54.2.1
      1  1.54.2.1  pgoyette /*	$NetBSD: npfctl.c,v 1.54.2.1 2018/04/16 02:00:10 pgoyette Exp $	*/
      2       1.1     rmind 
      3       1.1     rmind /*-
      4      1.41     rmind  * Copyright (c) 2009-2014 The NetBSD Foundation, Inc.
      5       1.1     rmind  * All rights reserved.
      6       1.1     rmind  *
      7       1.1     rmind  * This material is based upon work partially supported by The
      8       1.1     rmind  * NetBSD Foundation under a contract with Mindaugas Rasiukevicius.
      9       1.1     rmind  *
     10       1.1     rmind  * Redistribution and use in source and binary forms, with or without
     11       1.1     rmind  * modification, are permitted provided that the following conditions
     12       1.1     rmind  * are met:
     13       1.1     rmind  * 1. Redistributions of source code must retain the above copyright
     14       1.1     rmind  *    notice, this list of conditions and the following disclaimer.
     15       1.1     rmind  * 2. Redistributions in binary form must reproduce the above copyright
     16       1.1     rmind  *    notice, this list of conditions and the following disclaimer in the
     17       1.1     rmind  *    documentation and/or other materials provided with the distribution.
     18       1.1     rmind  *
     19       1.1     rmind  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     20       1.1     rmind  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     21       1.1     rmind  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     22       1.1     rmind  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     23       1.1     rmind  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     24       1.1     rmind  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     25       1.1     rmind  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     26       1.1     rmind  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     27       1.1     rmind  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     28       1.1     rmind  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     29       1.1     rmind  * POSSIBILITY OF SUCH DAMAGE.
     30       1.1     rmind  */
     31       1.1     rmind 
     32       1.2     rmind #include <sys/cdefs.h>
     33  1.54.2.1  pgoyette __RCSID("$NetBSD: npfctl.c,v 1.54.2.1 2018/04/16 02:00:10 pgoyette Exp $");
     34       1.2     rmind 
     35       1.1     rmind #include <sys/stat.h>
     36       1.1     rmind #include <sys/types.h>
     37      1.48  christos #include <sys/mman.h>
     38      1.48  christos #ifdef __NetBSD__
     39      1.48  christos #include <sha1.h>
     40      1.48  christos #include <sys/ioctl.h>
     41      1.44     rmind #include <sys/module.h>
     42      1.49  christos #define SHA_DIGEST_LENGTH SHA1_DIGEST_LENGTH
     43      1.49  christos #else
     44      1.49  christos #include <openssl/sha.h>
     45      1.48  christos #endif
     46       1.1     rmind 
     47       1.1     rmind #include <stdio.h>
     48       1.1     rmind #include <stdlib.h>
     49       1.1     rmind #include <string.h>
     50       1.1     rmind #include <err.h>
     51       1.1     rmind #include <fcntl.h>
     52       1.1     rmind #include <unistd.h>
     53      1.15     rmind #include <errno.h>
     54      1.48  christos 
     55      1.48  christos #include <arpa/inet.h>
     56      1.29     rmind 
     57       1.1     rmind #include "npfctl.h"
     58       1.1     rmind 
     59      1.29     rmind extern void		npf_yyparse_string(const char *);
     60       1.8     rmind 
     61      1.11     rmind enum {
     62      1.11     rmind 	NPFCTL_START,
     63      1.11     rmind 	NPFCTL_STOP,
     64      1.11     rmind 	NPFCTL_RELOAD,
     65      1.11     rmind 	NPFCTL_SHOWCONF,
     66      1.11     rmind 	NPFCTL_FLUSH,
     67      1.25     rmind 	NPFCTL_VALIDATE,
     68      1.11     rmind 	NPFCTL_TABLE,
     69      1.29     rmind 	NPFCTL_RULE,
     70      1.11     rmind 	NPFCTL_STATS,
     71      1.41     rmind 	NPFCTL_SAVE,
     72      1.41     rmind 	NPFCTL_LOAD,
     73      1.52     rmind 	NPFCTL_DEBUG,
     74      1.50  christos 	NPFCTL_CONN_LIST,
     75      1.11     rmind };
     76       1.1     rmind 
     77      1.15     rmind static const struct operations_s {
     78       1.1     rmind 	const char *		cmd;
     79       1.1     rmind 	int			action;
     80       1.1     rmind } operations[] = {
     81       1.1     rmind 	/* Start, stop, reload */
     82      1.41     rmind 	{	"start",	NPFCTL_START		},
     83      1.41     rmind 	{	"stop",		NPFCTL_STOP		},
     84      1.41     rmind 	{	"reload",	NPFCTL_RELOAD		},
     85      1.41     rmind 	{	"show",		NPFCTL_SHOWCONF,	},
     86      1.41     rmind 	{	"flush",	NPFCTL_FLUSH		},
     87       1.1     rmind 	/* Table */
     88      1.41     rmind 	{	"table",	NPFCTL_TABLE		},
     89      1.29     rmind 	/* Rule */
     90      1.41     rmind 	{	"rule",		NPFCTL_RULE		},
     91       1.3     rmind 	/* Stats */
     92      1.41     rmind 	{	"stats",	NPFCTL_STATS		},
     93      1.41     rmind 	/* Full state save/load */
     94      1.41     rmind 	{	"save",		NPFCTL_SAVE		},
     95      1.41     rmind 	{	"load",		NPFCTL_LOAD		},
     96      1.50  christos 	{	"list",		NPFCTL_CONN_LIST	},
     97      1.52     rmind 	/* Misc. */
     98      1.52     rmind 	{	"valid",	NPFCTL_VALIDATE		},
     99      1.52     rmind 	{	"debug",	NPFCTL_DEBUG		},
    100       1.1     rmind 	/* --- */
    101      1.41     rmind 	{	NULL,		0			}
    102       1.1     rmind };
    103       1.1     rmind 
    104      1.38     rmind bool
    105      1.38     rmind join(char *buf, size_t buflen, int count, char **args, const char *sep)
    106      1.29     rmind {
    107      1.38     rmind 	const u_int seplen = strlen(sep);
    108      1.29     rmind 	char *s = buf, *p = NULL;
    109      1.29     rmind 
    110      1.29     rmind 	for (int i = 0; i < count; i++) {
    111      1.29     rmind 		size_t len;
    112      1.29     rmind 
    113      1.29     rmind 		p = stpncpy(s, args[i], buflen);
    114      1.38     rmind 		len = p - s + seplen;
    115      1.29     rmind 		if (len >= buflen) {
    116      1.29     rmind 			return false;
    117      1.29     rmind 		}
    118      1.29     rmind 		buflen -= len;
    119      1.38     rmind 		strcpy(p, sep);
    120      1.38     rmind 		s = p + seplen;
    121      1.29     rmind 	}
    122      1.29     rmind 	*p = '\0';
    123      1.29     rmind 	return true;
    124      1.29     rmind }
    125      1.29     rmind 
    126       1.6     joerg __dead static void
    127       1.1     rmind usage(void)
    128       1.1     rmind {
    129       1.1     rmind 	const char *progname = getprogname();
    130       1.1     rmind 
    131       1.1     rmind 	fprintf(stderr,
    132      1.37     rmind 	    "Usage:\t%s start | stop | flush | show | stats\n",
    133      1.33  christos 	    progname);
    134      1.33  christos 	fprintf(stderr,
    135      1.33  christos 	    "\t%s validate | reload [<rule-file>]\n",
    136       1.3     rmind 	    progname);
    137       1.3     rmind 	fprintf(stderr,
    138      1.29     rmind 	    "\t%s rule \"rule-name\" { add | rem } <rule-syntax>\n",
    139      1.29     rmind 	    progname);
    140      1.29     rmind 	fprintf(stderr,
    141      1.29     rmind 	    "\t%s rule \"rule-name\" rem-id <rule-id>\n",
    142       1.1     rmind 	    progname);
    143       1.1     rmind 	fprintf(stderr,
    144      1.31     rmind 	    "\t%s rule \"rule-name\" { list | flush }\n",
    145      1.31     rmind 	    progname);
    146      1.31     rmind 	fprintf(stderr,
    147      1.21     rmind 	    "\t%s table <tid> { add | rem | test } <address/mask>\n",
    148       1.1     rmind 	    progname);
    149       1.1     rmind 	fprintf(stderr,
    150      1.21     rmind 	    "\t%s table <tid> { list | flush }\n",
    151       1.1     rmind 	    progname);
    152      1.37     rmind 	fprintf(stderr,
    153      1.41     rmind 	    "\t%s save | load\n",
    154      1.37     rmind 	    progname);
    155      1.50  christos 	fprintf(stderr,
    156      1.51       wiz 	    "\t%s list [-46hNnw] [-i <ifname>]\n",
    157      1.50  christos 	    progname);
    158  1.54.2.1  pgoyette 	fprintf(stderr,
    159  1.54.2.1  pgoyette 	    "\t%s debug [<rule-file>] [<raw-output>]\n",
    160  1.54.2.1  pgoyette 	    progname);
    161       1.1     rmind 	exit(EXIT_FAILURE);
    162       1.1     rmind }
    163       1.1     rmind 
    164       1.3     rmind static int
    165       1.3     rmind npfctl_print_stats(int fd)
    166       1.3     rmind {
    167      1.17     rmind 	static const struct stats_s {
    168      1.17     rmind 		/* Note: -1 indicates a new section. */
    169      1.17     rmind 		int		index;
    170      1.17     rmind 		const char *	name;
    171      1.17     rmind 	} stats[] = {
    172      1.17     rmind 		{ -1, "Packets passed"					},
    173      1.17     rmind 		{ NPF_STAT_PASS_DEFAULT,	"default pass"		},
    174      1.17     rmind 		{ NPF_STAT_PASS_RULESET,	"ruleset pass"		},
    175      1.41     rmind 		{ NPF_STAT_PASS_CONN,		"state pass"		},
    176      1.17     rmind 
    177      1.17     rmind 		{ -1, "Packets blocked"					},
    178      1.17     rmind 		{ NPF_STAT_BLOCK_DEFAULT,	"default block"		},
    179      1.17     rmind 		{ NPF_STAT_BLOCK_RULESET,	"ruleset block"		},
    180      1.17     rmind 
    181      1.41     rmind 		{ -1, "State and NAT entries"				},
    182      1.41     rmind 		{ NPF_STAT_CONN_CREATE,		"state allocations"},
    183      1.41     rmind 		{ NPF_STAT_CONN_DESTROY,	"state destructions"},
    184      1.17     rmind 		{ NPF_STAT_NAT_CREATE,		"NAT entry allocations"	},
    185      1.17     rmind 		{ NPF_STAT_NAT_DESTROY,		"NAT entry destructions"},
    186      1.17     rmind 
    187      1.27     rmind 		{ -1, "Network buffers"					},
    188      1.27     rmind 		{ NPF_STAT_NBUF_NONCONTIG,	"non-contiguous cases"	},
    189      1.27     rmind 		{ NPF_STAT_NBUF_CONTIG_FAIL,	"contig alloc failures"	},
    190      1.27     rmind 
    191      1.17     rmind 		{ -1, "Invalid packet state cases"			},
    192      1.17     rmind 		{ NPF_STAT_INVALID_STATE,	"cases in total"	},
    193      1.17     rmind 		{ NPF_STAT_INVALID_STATE_TCP1,	"TCP case I"		},
    194      1.17     rmind 		{ NPF_STAT_INVALID_STATE_TCP2,	"TCP case II"		},
    195      1.17     rmind 		{ NPF_STAT_INVALID_STATE_TCP3,	"TCP case III"		},
    196      1.17     rmind 
    197      1.17     rmind 		{ -1, "Packet race cases"				},
    198      1.17     rmind 		{ NPF_STAT_RACE_NAT,		"NAT association race"	},
    199      1.41     rmind 		{ NPF_STAT_RACE_CONN,		"duplicate state race"	},
    200      1.17     rmind 
    201      1.17     rmind 		{ -1, "Fragmentation"					},
    202      1.17     rmind 		{ NPF_STAT_FRAGMENTS,		"fragments"		},
    203      1.17     rmind 		{ NPF_STAT_REASSEMBLY,		"reassembled"		},
    204      1.17     rmind 		{ NPF_STAT_REASSFAIL,		"failed reassembly"	},
    205      1.17     rmind 
    206      1.17     rmind 		{ -1, "Other"						},
    207      1.17     rmind 		{ NPF_STAT_ERROR,		"unexpected errors"	},
    208      1.17     rmind 	};
    209      1.24     rmind 	uint64_t *st = ecalloc(1, NPF_STATS_SIZE);
    210       1.3     rmind 
    211       1.3     rmind 	if (ioctl(fd, IOC_NPF_STATS, &st) != 0) {
    212       1.3     rmind 		err(EXIT_FAILURE, "ioctl(IOC_NPF_STATS)");
    213       1.3     rmind 	}
    214       1.3     rmind 
    215      1.17     rmind 	for (unsigned i = 0; i < __arraycount(stats); i++) {
    216      1.17     rmind 		const char *sname = stats[i].name;
    217      1.17     rmind 		int sidx = stats[i].index;
    218      1.17     rmind 
    219      1.17     rmind 		if (sidx == -1) {
    220      1.17     rmind 			printf("%s:\n", sname);
    221      1.17     rmind 		} else {
    222      1.17     rmind 			printf("\t%"PRIu64" %s\n", st[sidx], sname);
    223      1.17     rmind 		}
    224      1.17     rmind 	}
    225       1.4     rmind 
    226       1.3     rmind 	free(st);
    227       1.3     rmind 	return 0;
    228       1.3     rmind }
    229       1.3     rmind 
    230      1.10     rmind void
    231      1.48  christos npfctl_print_error(const npf_error_t *ne)
    232      1.10     rmind {
    233      1.48  christos 	const char *srcfile = ne->source_file;
    234      1.10     rmind 
    235      1.10     rmind 	if (srcfile) {
    236      1.48  christos 		warnx("source %s line %d", srcfile, ne->source_line);
    237      1.10     rmind 	}
    238      1.48  christos 	if (ne->id) {
    239      1.48  christos 		warnx("object: %" PRIi64, ne->id);
    240      1.10     rmind 	}
    241      1.10     rmind }
    242      1.10     rmind 
    243      1.21     rmind char *
    244      1.50  christos npfctl_print_addrmask(int alen, const char *fmt, const npf_addr_t *addr,
    245      1.50  christos     npf_netmask_t mask)
    246      1.21     rmind {
    247      1.50  christos 	const unsigned buflen = 256;
    248      1.48  christos 	char *buf = ecalloc(1, buflen);
    249      1.21     rmind 	struct sockaddr_storage ss;
    250      1.48  christos 
    251      1.48  christos 	memset(&ss, 0, sizeof(ss));
    252      1.21     rmind 
    253      1.21     rmind 	switch (alen) {
    254      1.21     rmind 	case 4: {
    255      1.21     rmind 		struct sockaddr_in *sin = (void *)&ss;
    256      1.21     rmind 		sin->sin_family = AF_INET;
    257      1.21     rmind 		memcpy(&sin->sin_addr, addr, sizeof(sin->sin_addr));
    258      1.21     rmind 		break;
    259      1.21     rmind 	}
    260      1.21     rmind 	case 16: {
    261      1.21     rmind 		struct sockaddr_in6 *sin6 = (void *)&ss;
    262      1.21     rmind 		sin6->sin6_family = AF_INET6;
    263      1.21     rmind 		memcpy(&sin6->sin6_addr, addr, sizeof(sin6->sin6_addr));
    264      1.21     rmind 		break;
    265      1.21     rmind 	}
    266      1.21     rmind 	default:
    267      1.21     rmind 		assert(false);
    268      1.21     rmind 	}
    269      1.50  christos 	sockaddr_snprintf(buf, buflen, fmt, (const void *)&ss);
    270      1.38     rmind 	if (mask && mask != NPF_NO_NETMASK) {
    271      1.48  christos 		const unsigned len = strlen(buf);
    272      1.48  christos 		snprintf(&buf[len], buflen - len, "/%u", mask);
    273      1.21     rmind 	}
    274      1.21     rmind 	return buf;
    275      1.21     rmind }
    276      1.21     rmind 
    277      1.16     joerg __dead static void
    278      1.21     rmind npfctl_table(int fd, int argc, char **argv)
    279      1.15     rmind {
    280      1.15     rmind 	static const struct tblops_s {
    281      1.15     rmind 		const char *	cmd;
    282      1.15     rmind 		int		action;
    283      1.15     rmind 	} tblops[] = {
    284      1.29     rmind 		{ "add",	NPF_CMD_TABLE_ADD		},
    285      1.29     rmind 		{ "rem",	NPF_CMD_TABLE_REMOVE		},
    286      1.29     rmind 		{ "del",	NPF_CMD_TABLE_REMOVE		},
    287      1.29     rmind 		{ "test",	NPF_CMD_TABLE_LOOKUP		},
    288      1.29     rmind 		{ "list",	NPF_CMD_TABLE_LIST		},
    289      1.37     rmind 		{ "flush",	NPF_CMD_TABLE_FLUSH		},
    290      1.21     rmind 		{ NULL,		0				}
    291      1.15     rmind 	};
    292      1.15     rmind 	npf_ioctl_table_t nct;
    293      1.15     rmind 	fam_addr_mask_t fam;
    294      1.21     rmind 	size_t buflen = 512;
    295      1.40     rmind 	char *cmd, *arg;
    296      1.15     rmind 	int n, alen;
    297      1.15     rmind 
    298      1.21     rmind 	/* Default action is list. */
    299      1.15     rmind 	memset(&nct, 0, sizeof(npf_ioctl_table_t));
    300      1.40     rmind 	nct.nct_name = argv[0];
    301      1.21     rmind 	cmd = argv[1];
    302      1.15     rmind 
    303      1.15     rmind 	for (n = 0; tblops[n].cmd != NULL; n++) {
    304      1.21     rmind 		if (strcmp(cmd, tblops[n].cmd) != 0) {
    305      1.21     rmind 			continue;
    306      1.21     rmind 		}
    307      1.29     rmind 		nct.nct_cmd = tblops[n].action;
    308      1.21     rmind 		break;
    309      1.21     rmind 	}
    310      1.21     rmind 	if (tblops[n].cmd == NULL) {
    311      1.21     rmind 		errx(EXIT_FAILURE, "invalid command '%s'", cmd);
    312      1.21     rmind 	}
    313      1.37     rmind 
    314      1.37     rmind 	switch (nct.nct_cmd) {
    315      1.37     rmind 	case NPF_CMD_TABLE_LIST:
    316      1.37     rmind 	case NPF_CMD_TABLE_FLUSH:
    317      1.40     rmind 		arg = NULL;
    318      1.37     rmind 		break;
    319      1.37     rmind 	default:
    320      1.21     rmind 		if (argc < 3) {
    321      1.21     rmind 			usage();
    322      1.15     rmind 		}
    323      1.21     rmind 		arg = argv[2];
    324      1.15     rmind 	}
    325      1.37     rmind 
    326      1.21     rmind again:
    327      1.37     rmind 	switch (nct.nct_cmd) {
    328      1.37     rmind 	case NPF_CMD_TABLE_LIST:
    329      1.24     rmind 		nct.nct_data.buf.buf = ecalloc(1, buflen);
    330      1.21     rmind 		nct.nct_data.buf.len = buflen;
    331      1.37     rmind 		break;
    332      1.37     rmind 	case NPF_CMD_TABLE_FLUSH:
    333      1.37     rmind 		break;
    334      1.37     rmind 	default:
    335      1.21     rmind 		if (!npfctl_parse_cidr(arg, &fam, &alen)) {
    336      1.21     rmind 			errx(EXIT_FAILURE, "invalid CIDR '%s'", arg);
    337      1.21     rmind 		}
    338      1.21     rmind 		nct.nct_data.ent.alen = alen;
    339      1.26     rmind 		memcpy(&nct.nct_data.ent.addr, &fam.fam_addr, alen);
    340      1.21     rmind 		nct.nct_data.ent.mask = fam.fam_mask;
    341      1.15     rmind 	}
    342      1.15     rmind 
    343      1.15     rmind 	if (ioctl(fd, IOC_NPF_TABLE, &nct) != -1) {
    344      1.15     rmind 		errno = 0;
    345      1.15     rmind 	}
    346      1.15     rmind 	switch (errno) {
    347      1.21     rmind 	case 0:
    348      1.21     rmind 		break;
    349      1.15     rmind 	case EEXIST:
    350      1.21     rmind 		errx(EXIT_FAILURE, "entry already exists or is conflicting");
    351      1.15     rmind 	case ENOENT:
    352      1.21     rmind 		errx(EXIT_FAILURE, "no matching entry was not found");
    353      1.15     rmind 	case EINVAL:
    354      1.21     rmind 		errx(EXIT_FAILURE, "invalid address, mask or table ID");
    355      1.21     rmind 	case ENOMEM:
    356      1.29     rmind 		if (nct.nct_cmd == NPF_CMD_TABLE_LIST) {
    357      1.21     rmind 			/* XXX */
    358      1.21     rmind 			free(nct.nct_data.buf.buf);
    359      1.21     rmind 			buflen <<= 1;
    360      1.21     rmind 			goto again;
    361      1.21     rmind 		}
    362      1.21     rmind 		/* FALLTHROUGH */
    363      1.21     rmind 	default:
    364      1.32  christos 		err(EXIT_FAILURE, "ioctl(IOC_NPF_TABLE)");
    365      1.21     rmind 	}
    366      1.21     rmind 
    367      1.29     rmind 	if (nct.nct_cmd == NPF_CMD_TABLE_LIST) {
    368      1.21     rmind 		npf_ioctl_ent_t *ent = nct.nct_data.buf.buf;
    369      1.21     rmind 		char *buf;
    370      1.21     rmind 
    371      1.21     rmind 		while (nct.nct_data.buf.len--) {
    372      1.21     rmind 			if (!ent->alen)
    373      1.21     rmind 				break;
    374      1.50  christos 			buf = npfctl_print_addrmask(ent->alen, "%a",
    375      1.21     rmind 			    &ent->addr, ent->mask);
    376      1.21     rmind 			puts(buf);
    377      1.21     rmind 			ent++;
    378      1.21     rmind 		}
    379      1.21     rmind 		free(nct.nct_data.buf.buf);
    380      1.21     rmind 	} else {
    381      1.21     rmind 		printf("%s: %s\n", getprogname(),
    382      1.29     rmind 		    nct.nct_cmd == NPF_CMD_TABLE_LOOKUP ?
    383      1.15     rmind 		    "matching entry found" : "success");
    384      1.15     rmind 	}
    385      1.21     rmind 	exit(EXIT_SUCCESS);
    386      1.15     rmind }
    387      1.15     rmind 
    388      1.29     rmind static nl_rule_t *
    389      1.29     rmind npfctl_parse_rule(int argc, char **argv)
    390      1.29     rmind {
    391      1.29     rmind 	char rule_string[1024];
    392      1.29     rmind 	nl_rule_t *rl;
    393      1.29     rmind 
    394      1.29     rmind 	/* Get the rule string and parse it. */
    395      1.38     rmind 	if (!join(rule_string, sizeof(rule_string), argc, argv, " ")) {
    396      1.29     rmind 		errx(EXIT_FAILURE, "command too long");
    397      1.29     rmind 	}
    398      1.29     rmind 	npfctl_parse_string(rule_string);
    399      1.29     rmind 	if ((rl = npfctl_rule_ref()) == NULL) {
    400      1.29     rmind 		errx(EXIT_FAILURE, "could not parse the rule");
    401      1.29     rmind 	}
    402      1.29     rmind 	return rl;
    403      1.29     rmind }
    404      1.29     rmind 
    405      1.48  christos #ifdef __NetBSD__
    406      1.49  christos static unsigned char *
    407      1.49  christos SHA1(const unsigned char *d, size_t l, unsigned char *md)
    408      1.48  christos {
    409      1.48  christos 	SHA1_CTX c;
    410      1.48  christos 
    411      1.48  christos 	SHA1Init(&c);
    412      1.48  christos 	SHA1Update(&c, d, l);
    413      1.48  christos 	SHA1Final(md, &c);
    414      1.48  christos 	return md;
    415      1.47  christos }
    416      1.48  christos #endif
    417      1.47  christos 
    418      1.47  christos static void
    419      1.29     rmind npfctl_generate_key(nl_rule_t *rl, void *key)
    420      1.29     rmind {
    421      1.29     rmind 	void *meta;
    422      1.29     rmind 	size_t len;
    423      1.29     rmind 
    424      1.29     rmind 	if ((meta = npf_rule_export(rl, &len)) == NULL) {
    425      1.29     rmind 		errx(EXIT_FAILURE, "error generating rule key");
    426      1.29     rmind 	}
    427      1.48  christos 	__CTASSERT(NPF_RULE_MAXKEYLEN >= SHA_DIGEST_LENGTH);
    428      1.29     rmind 	memset(key, 0, NPF_RULE_MAXKEYLEN);
    429      1.48  christos 	SHA1(meta, len, key);
    430      1.29     rmind 	free(meta);
    431      1.29     rmind }
    432      1.29     rmind 
    433      1.29     rmind __dead static void
    434      1.29     rmind npfctl_rule(int fd, int argc, char **argv)
    435      1.29     rmind {
    436      1.29     rmind 	static const struct ruleops_s {
    437      1.29     rmind 		const char *	cmd;
    438      1.29     rmind 		int		action;
    439      1.36     rmind 		bool		extra_arg;
    440      1.29     rmind 	} ruleops[] = {
    441      1.36     rmind 		{ "add",	NPF_CMD_RULE_ADD,	true	},
    442      1.36     rmind 		{ "rem",	NPF_CMD_RULE_REMKEY,	true	},
    443      1.36     rmind 		{ "del",	NPF_CMD_RULE_REMKEY,	true	},
    444      1.36     rmind 		{ "rem-id",	NPF_CMD_RULE_REMOVE,	true	},
    445      1.36     rmind 		{ "list",	NPF_CMD_RULE_LIST,	false	},
    446      1.36     rmind 		{ "flush",	NPF_CMD_RULE_FLUSH,	false	},
    447      1.36     rmind 		{ NULL,		0,			0	}
    448      1.29     rmind 	};
    449      1.29     rmind 	uint8_t key[NPF_RULE_MAXKEYLEN];
    450      1.29     rmind 	const char *ruleset_name = argv[0];
    451      1.29     rmind 	const char *cmd = argv[1];
    452      1.29     rmind 	int error, action = 0;
    453      1.31     rmind 	uint64_t rule_id;
    454      1.36     rmind 	bool extra_arg;
    455      1.29     rmind 	nl_rule_t *rl;
    456      1.29     rmind 
    457      1.29     rmind 	for (int n = 0; ruleops[n].cmd != NULL; n++) {
    458      1.29     rmind 		if (strcmp(cmd, ruleops[n].cmd) == 0) {
    459      1.29     rmind 			action = ruleops[n].action;
    460      1.36     rmind 			extra_arg = ruleops[n].extra_arg;
    461      1.29     rmind 			break;
    462      1.29     rmind 		}
    463      1.29     rmind 	}
    464      1.36     rmind 	argc -= 2;
    465      1.36     rmind 	argv += 2;
    466      1.29     rmind 
    467      1.36     rmind 	if (!action || (extra_arg && argc == 0)) {
    468      1.29     rmind 		usage();
    469      1.29     rmind 	}
    470      1.29     rmind 
    471      1.29     rmind 	switch (action) {
    472      1.29     rmind 	case NPF_CMD_RULE_ADD:
    473      1.29     rmind 		rl = npfctl_parse_rule(argc, argv);
    474      1.29     rmind 		npfctl_generate_key(rl, key);
    475      1.29     rmind 		npf_rule_setkey(rl, key, sizeof(key));
    476      1.29     rmind 		error = npf_ruleset_add(fd, ruleset_name, rl, &rule_id);
    477      1.29     rmind 		break;
    478      1.29     rmind 	case NPF_CMD_RULE_REMKEY:
    479      1.29     rmind 		rl = npfctl_parse_rule(argc, argv);
    480      1.29     rmind 		npfctl_generate_key(rl, key);
    481      1.29     rmind 		error = npf_ruleset_remkey(fd, ruleset_name, key, sizeof(key));
    482      1.29     rmind 		break;
    483      1.29     rmind 	case NPF_CMD_RULE_REMOVE:
    484      1.31     rmind 		rule_id = strtoull(argv[0], NULL, 16);
    485      1.29     rmind 		error = npf_ruleset_remove(fd, ruleset_name, rule_id);
    486      1.29     rmind 		break;
    487      1.30     rmind 	case NPF_CMD_RULE_LIST:
    488      1.30     rmind 		error = npfctl_ruleset_show(fd, ruleset_name);
    489      1.30     rmind 		break;
    490      1.30     rmind 	case NPF_CMD_RULE_FLUSH:
    491      1.30     rmind 		error = npf_ruleset_flush(fd, ruleset_name);
    492      1.30     rmind 		break;
    493      1.29     rmind 	default:
    494      1.48  christos 		abort();
    495      1.29     rmind 	}
    496      1.29     rmind 
    497      1.29     rmind 	switch (error) {
    498      1.29     rmind 	case 0:
    499      1.29     rmind 		/* Success. */
    500      1.29     rmind 		break;
    501      1.31     rmind 	case ESRCH:
    502      1.31     rmind 		errx(EXIT_FAILURE, "ruleset \"%s\" not found", ruleset_name);
    503      1.29     rmind 	case ENOENT:
    504      1.31     rmind 		errx(EXIT_FAILURE, "rule was not found");
    505      1.29     rmind 	default:
    506      1.29     rmind 		errx(EXIT_FAILURE, "rule operation: %s", strerror(error));
    507      1.29     rmind 	}
    508      1.29     rmind 	if (action == NPF_CMD_RULE_ADD) {
    509      1.31     rmind 		printf("OK %" PRIx64 "\n", rule_id);
    510      1.29     rmind 	}
    511      1.29     rmind 	exit(EXIT_SUCCESS);
    512      1.29     rmind }
    513      1.29     rmind 
    514      1.45  christos static bool bpfjit = true;
    515      1.45  christos 
    516      1.45  christos void
    517      1.45  christos npfctl_bpfjit(bool onoff)
    518      1.45  christos {
    519      1.45  christos 	bpfjit = onoff;
    520      1.45  christos }
    521      1.45  christos 
    522      1.44     rmind static void
    523      1.44     rmind npfctl_preload_bpfjit(void)
    524      1.44     rmind {
    525      1.48  christos #ifdef __NetBSD__
    526      1.44     rmind 	modctl_load_t args = {
    527      1.44     rmind 		.ml_filename = "bpfjit",
    528      1.44     rmind 		.ml_flags = MODCTL_NO_PROP,
    529      1.44     rmind 		.ml_props = NULL,
    530      1.44     rmind 		.ml_propslen = 0
    531      1.44     rmind 	};
    532      1.44     rmind 
    533      1.45  christos 	if (!bpfjit)
    534      1.45  christos 		return;
    535      1.45  christos 
    536      1.44     rmind 	if (modctl(MODCTL_LOAD, &args) != 0 && errno != EEXIST) {
    537      1.45  christos 		static const char *p = "; performance will be degraded";
    538      1.45  christos 		if (errno == ENOENT)
    539      1.45  christos 			warnx("the bpfjit module seems to be missing%s", p);
    540      1.45  christos 		else
    541      1.45  christos 			warn("error loading the bpfjit module%s", p);
    542      1.45  christos 		warnx("To disable this warning `set bpf.jit off' in "
    543      1.45  christos 		    "/etc/npf.conf");
    544      1.44     rmind 	}
    545      1.48  christos #endif
    546      1.44     rmind }
    547      1.44     rmind 
    548      1.41     rmind static int
    549      1.48  christos npfctl_load(int fd)
    550      1.41     rmind {
    551      1.41     rmind 	nl_config_t *ncf;
    552      1.48  christos 	npf_error_t errinfo;
    553      1.48  christos 	struct stat sb;
    554      1.48  christos 	size_t blen;
    555      1.48  christos 	void *blob;
    556      1.41     rmind 	int error;
    557      1.41     rmind 
    558      1.48  christos 	/*
    559      1.48  christos 	 * The file may change while reading - we are not handling this,
    560      1.48  christos 	 * leaving this responsibility for the caller.
    561      1.48  christos 	 */
    562      1.48  christos 	if (stat(NPF_DB_PATH, &sb) == -1) {
    563      1.48  christos 		err(EXIT_FAILURE, "stat");
    564      1.48  christos 	}
    565      1.48  christos 	if ((blen = sb.st_size) == 0) {
    566      1.48  christos 		err(EXIT_FAILURE, "saved configuration file is empty");
    567      1.48  christos 	}
    568      1.48  christos 	if ((blob = mmap(NULL, blen, PROT_READ,
    569      1.48  christos 	    MAP_FILE | MAP_PRIVATE, fd, 0)) == MAP_FAILED) {
    570      1.48  christos 		err(EXIT_FAILURE, "mmap");
    571      1.48  christos 	}
    572      1.48  christos 	ncf = npf_config_import(blob, blen);
    573      1.48  christos 	munmap(blob, blen);
    574      1.41     rmind 	if (ncf == NULL) {
    575      1.41     rmind 		return errno;
    576      1.41     rmind 	}
    577      1.41     rmind 
    578      1.48  christos 	/*
    579      1.48  christos 	 * Configuration imported - submit it now.
    580      1.48  christos 	 **/
    581      1.48  christos 	errno = error = npf_config_submit(ncf, fd, &errinfo);
    582      1.43     rmind 	if (error) {
    583      1.48  christos 		npfctl_print_error(&errinfo);
    584      1.43     rmind 	}
    585      1.41     rmind 	npf_config_destroy(ncf);
    586      1.41     rmind 	return error;
    587      1.41     rmind }
    588      1.41     rmind 
    589      1.50  christos struct npf_conn_filter {
    590      1.50  christos 	uint16_t alen;
    591      1.50  christos 	const char *ifname;
    592      1.50  christos 	bool nat;
    593      1.50  christos 	bool wide;
    594      1.50  christos 	bool name;
    595      1.50  christos 	int width;
    596      1.50  christos 	FILE *fp;
    597      1.50  christos };
    598      1.50  christos 
    599      1.50  christos static int
    600      1.52     rmind npfctl_conn_print(unsigned alen, const npf_addr_t *a, const in_port_t *p,
    601      1.50  christos     const char *ifname, void *v)
    602      1.50  christos {
    603      1.50  christos 	struct npf_conn_filter *fil = v;
    604      1.50  christos 	FILE *fp = fil->fp;
    605      1.50  christos 	char *src, *dst;
    606      1.50  christos 
    607      1.50  christos 	if (fil->ifname && strcmp(ifname, fil->ifname) != 0)
    608      1.50  christos 		return 0;
    609      1.50  christos 	if (fil->alen && alen != fil->alen)
    610      1.50  christos 		return 0;
    611      1.50  christos 	if (fil->nat && !p[2])
    612      1.50  christos 		return 0;
    613      1.50  christos 
    614      1.50  christos 	int w = fil->width;
    615      1.50  christos 	const char *fmt = fil->name ? "%A" :
    616      1.50  christos 	    (alen == sizeof(struct in_addr) ? "%a" : "[%a]");
    617      1.50  christos 	src = npfctl_print_addrmask(alen, fmt, &a[0], NPF_NO_NETMASK);
    618      1.50  christos 	dst = npfctl_print_addrmask(alen, fmt, &a[1], NPF_NO_NETMASK);
    619      1.50  christos 	if (fil->wide)
    620      1.50  christos 		fprintf(fp, "%s:%d %s:%d", src, p[0], dst, p[1]);
    621      1.50  christos 	else
    622      1.50  christos 		fprintf(fp, "%*.*s:%-5d %*.*s:%-5d", w, w, src, p[0],
    623      1.50  christos 		    w, w, dst, p[1]);
    624      1.50  christos 	free(src);
    625      1.50  christos 	free(dst);
    626      1.50  christos 	if (!p[2]) {
    627      1.50  christos 		fputc('\n', fp);
    628      1.50  christos 		return 1;
    629      1.50  christos 	}
    630      1.54     ozaki 	fprintf(fp, " via %s:%d\n", ifname, p[2]);
    631      1.50  christos 	return 1;
    632      1.50  christos }
    633      1.50  christos 
    634      1.50  christos 
    635      1.50  christos static int
    636      1.50  christos npfctl_conn_list(int fd, int argc, char **argv)
    637      1.50  christos {
    638      1.50  christos 	struct npf_conn_filter f;
    639      1.50  christos 	int c;
    640      1.50  christos 	int header = true;
    641      1.50  christos 	memset(&f, 0, sizeof(f));
    642      1.50  christos 
    643      1.50  christos 	argc--;
    644      1.50  christos 	argv++;
    645      1.50  christos 
    646      1.50  christos 	while ((c = getopt(argc, argv, "46hi:nNw")) != -1) {
    647      1.50  christos 		switch (c) {
    648      1.50  christos 		case '4':
    649      1.50  christos 			f.alen = sizeof(struct in_addr);
    650      1.50  christos 			break;
    651      1.50  christos 		case '6':
    652      1.50  christos 			f.alen = sizeof(struct in6_addr);
    653      1.50  christos 			break;
    654      1.50  christos 		case 'h':
    655      1.50  christos 			header = false;
    656      1.50  christos 		case 'i':
    657      1.50  christos 			f.ifname = optarg;
    658      1.50  christos 			break;
    659      1.50  christos 		case 'n':
    660      1.50  christos 			f.nat = true;
    661      1.50  christos 			break;
    662      1.50  christos 		case 'N':
    663      1.50  christos 			f.name = true;
    664      1.50  christos 			break;
    665      1.50  christos 		case 'w':
    666      1.50  christos 			f.wide = true;
    667      1.50  christos 			break;
    668      1.50  christos 		default:
    669      1.50  christos 			fprintf(stderr,
    670      1.50  christos 			    "Usage: %s list [-46hnNw] [-i <ifname>]\n",
    671      1.50  christos 			    getprogname());
    672      1.50  christos 			exit(EXIT_FAILURE);
    673      1.50  christos 		}
    674      1.50  christos 	}
    675      1.50  christos 	f.width = f.alen == sizeof(struct in_addr) ? 25 : 41;
    676      1.50  christos 	int w = f.width + 6;
    677      1.50  christos 	f.fp = stdout;
    678      1.50  christos 	if (header)
    679      1.50  christos 		fprintf(f.fp, "%*.*s %*.*s\n",
    680      1.50  christos 		    w, w, "From address:port ", w, w, "To address:port ");
    681      1.52     rmind 
    682      1.50  christos 	npf_conn_list(fd, npfctl_conn_print, &f);
    683      1.50  christos 	return 0;
    684      1.50  christos }
    685      1.50  christos 
    686      1.52     rmind static int
    687      1.52     rmind npfctl_open_dev(const char *path)
    688       1.1     rmind {
    689      1.52     rmind 	int fd, ver;
    690       1.1     rmind 
    691      1.52     rmind 	fd = open(path, O_RDONLY);
    692       1.1     rmind 	if (fd == -1) {
    693      1.52     rmind 		err(EXIT_FAILURE, "cannot open '%s'", path);
    694       1.1     rmind 	}
    695      1.29     rmind 	if (ioctl(fd, IOC_NPF_VERSION, &ver) == -1) {
    696      1.32  christos 		err(EXIT_FAILURE, "ioctl(IOC_NPF_VERSION)");
    697      1.15     rmind 	}
    698       1.1     rmind 	if (ver != NPF_VERSION) {
    699       1.4     rmind 		errx(EXIT_FAILURE,
    700      1.14     rmind 		    "incompatible NPF interface version (%d, kernel %d)\n"
    701      1.53  christos 		    "Hint: update %s?", NPF_VERSION, ver,
    702      1.53  christos 		    NPF_VERSION > ver ? "userland" : "kernel");
    703       1.1     rmind 	}
    704      1.52     rmind 	return fd;
    705      1.52     rmind }
    706      1.52     rmind 
    707      1.52     rmind static void
    708      1.52     rmind npfctl(int action, int argc, char **argv)
    709      1.52     rmind {
    710      1.52     rmind 	int fd, boolval, ret = 0;
    711      1.52     rmind 	const char *fun = "";
    712      1.52     rmind 	nl_config_t *ncf;
    713      1.52     rmind 
    714      1.52     rmind 	switch (action) {
    715      1.52     rmind 	case NPFCTL_VALIDATE:
    716      1.52     rmind 	case NPFCTL_DEBUG:
    717      1.52     rmind 		fd = 0;
    718      1.52     rmind 		break;
    719      1.52     rmind 	default:
    720      1.52     rmind 		fd = npfctl_open_dev(NPF_DEV_PATH);
    721      1.52     rmind 	}
    722      1.29     rmind 
    723       1.1     rmind 	switch (action) {
    724       1.1     rmind 	case NPFCTL_START:
    725       1.1     rmind 		boolval = true;
    726       1.1     rmind 		ret = ioctl(fd, IOC_NPF_SWITCH, &boolval);
    727      1.32  christos 		fun = "ioctl(IOC_NPF_SWITCH)";
    728       1.1     rmind 		break;
    729       1.1     rmind 	case NPFCTL_STOP:
    730       1.1     rmind 		boolval = false;
    731       1.1     rmind 		ret = ioctl(fd, IOC_NPF_SWITCH, &boolval);
    732      1.32  christos 		fun = "ioctl(IOC_NPF_SWITCH)";
    733       1.1     rmind 		break;
    734       1.1     rmind 	case NPFCTL_RELOAD:
    735       1.8     rmind 		npfctl_config_init(false);
    736      1.29     rmind 		npfctl_parse_file(argc < 3 ? NPF_CONF_PATH : argv[2]);
    737      1.46  christos 		npfctl_preload_bpfjit();
    738      1.32  christos 		errno = ret = npfctl_config_send(fd, NULL);
    739      1.32  christos 		fun = "npfctl_config_send";
    740       1.1     rmind 		break;
    741      1.11     rmind 	case NPFCTL_SHOWCONF:
    742      1.11     rmind 		ret = npfctl_config_show(fd);
    743      1.32  christos 		fun = "npfctl_config_show";
    744      1.11     rmind 		break;
    745       1.1     rmind 	case NPFCTL_FLUSH:
    746       1.9     rmind 		ret = npf_config_flush(fd);
    747      1.32  christos 		fun = "npf_config_flush";
    748       1.1     rmind 		break;
    749       1.1     rmind 	case NPFCTL_TABLE:
    750      1.21     rmind 		if ((argc -= 2) < 2) {
    751       1.1     rmind 			usage();
    752       1.1     rmind 		}
    753      1.21     rmind 		argv += 2;
    754      1.21     rmind 		npfctl_table(fd, argc, argv);
    755       1.1     rmind 		break;
    756      1.29     rmind 	case NPFCTL_RULE:
    757      1.29     rmind 		if ((argc -= 2) < 2) {
    758      1.29     rmind 			usage();
    759      1.29     rmind 		}
    760      1.29     rmind 		argv += 2;
    761      1.29     rmind 		npfctl_rule(fd, argc, argv);
    762      1.29     rmind 		break;
    763      1.41     rmind 	case NPFCTL_LOAD:
    764      1.44     rmind 		npfctl_preload_bpfjit();
    765      1.41     rmind 		ret = npfctl_load(fd);
    766      1.41     rmind 		fun = "npfctl_config_load";
    767      1.41     rmind 		break;
    768      1.41     rmind 	case NPFCTL_SAVE:
    769      1.48  christos 		ncf = npf_config_retrieve(fd);
    770      1.48  christos 		if (ncf) {
    771      1.48  christos 			npfctl_config_save(ncf, NPF_DB_PATH);
    772      1.48  christos 			npf_config_destroy(ncf);
    773      1.48  christos 		} else {
    774      1.48  christos 			ret = errno;
    775      1.48  christos 		}
    776      1.41     rmind 		fun = "npfctl_config_save";
    777      1.41     rmind 		break;
    778       1.3     rmind 	case NPFCTL_STATS:
    779       1.3     rmind 		ret = npfctl_print_stats(fd);
    780      1.32  christos 		fun = "npfctl_print_stats";
    781       1.3     rmind 		break;
    782      1.50  christos 	case NPFCTL_CONN_LIST:
    783      1.50  christos 		ret = npfctl_conn_list(fd, argc, argv);
    784      1.50  christos 		fun = "npfctl_conn_list";
    785      1.50  christos 		break;
    786      1.52     rmind 	case NPFCTL_VALIDATE:
    787      1.52     rmind 		npfctl_config_init(false);
    788      1.52     rmind 		npfctl_parse_file(argc > 2 ? argv[2] : NPF_CONF_PATH);
    789      1.52     rmind 		ret = npfctl_config_show(0);
    790      1.52     rmind 		fun = "npfctl_config_show";
    791      1.52     rmind 		break;
    792      1.52     rmind 	case NPFCTL_DEBUG:
    793      1.52     rmind 		npfctl_config_init(true);
    794      1.52     rmind 		npfctl_parse_file(argc > 2 ? argv[2] : NPF_CONF_PATH);
    795      1.52     rmind 		npfctl_config_send(0, argc > 3 ? argv[3] : "/tmp/npf.plist");
    796      1.52     rmind 		break;
    797       1.1     rmind 	}
    798       1.7    zoltan 	if (ret) {
    799      1.32  christos 		err(EXIT_FAILURE, "%s", fun);
    800       1.1     rmind 	}
    801      1.52     rmind 	if (fd) {
    802      1.52     rmind 		close(fd);
    803      1.52     rmind 	}
    804       1.1     rmind }
    805       1.1     rmind 
    806       1.1     rmind int
    807       1.1     rmind main(int argc, char **argv)
    808       1.1     rmind {
    809       1.1     rmind 	char *cmd;
    810       1.1     rmind 
    811       1.1     rmind 	if (argc < 2) {
    812       1.1     rmind 		usage();
    813       1.1     rmind 	}
    814      1.48  christos 	npfctl_show_init();
    815       1.1     rmind 	cmd = argv[1];
    816       1.1     rmind 
    817      1.12     rmind 	/* Find and call the subroutine. */
    818       1.8     rmind 	for (int n = 0; operations[n].cmd != NULL; n++) {
    819      1.25     rmind 		const char *opcmd = operations[n].cmd;
    820      1.25     rmind 		if (strncmp(cmd, opcmd, strlen(opcmd)) != 0)
    821       1.1     rmind 			continue;
    822       1.1     rmind 		npfctl(operations[n].action, argc, argv);
    823       1.8     rmind 		return EXIT_SUCCESS;
    824       1.1     rmind 	}
    825       1.3     rmind 	usage();
    826       1.1     rmind }
    827