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