Home | History | Annotate | Line # | Download | only in npfctl
npfctl.c revision 1.57
      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.57     rmind __RCSID("$NetBSD: npfctl.c,v 1.57 2019/01/19 21:19:32 rmind 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.57     rmind 	fprintf(stderr,
    157  1.57     rmind 	    "\t%s debug [<rule-file>] [<raw-output>]\n",
    158  1.57     rmind 	    progname);
    159   1.1     rmind 	exit(EXIT_FAILURE);
    160   1.1     rmind }
    161   1.1     rmind 
    162   1.3     rmind static int
    163   1.3     rmind npfctl_print_stats(int fd)
    164   1.3     rmind {
    165  1.17     rmind 	static const struct stats_s {
    166  1.17     rmind 		/* Note: -1 indicates a new section. */
    167  1.17     rmind 		int		index;
    168  1.17     rmind 		const char *	name;
    169  1.17     rmind 	} stats[] = {
    170  1.17     rmind 		{ -1, "Packets passed"					},
    171  1.17     rmind 		{ NPF_STAT_PASS_DEFAULT,	"default pass"		},
    172  1.17     rmind 		{ NPF_STAT_PASS_RULESET,	"ruleset pass"		},
    173  1.41     rmind 		{ NPF_STAT_PASS_CONN,		"state pass"		},
    174  1.17     rmind 
    175  1.17     rmind 		{ -1, "Packets blocked"					},
    176  1.17     rmind 		{ NPF_STAT_BLOCK_DEFAULT,	"default block"		},
    177  1.17     rmind 		{ NPF_STAT_BLOCK_RULESET,	"ruleset block"		},
    178  1.17     rmind 
    179  1.41     rmind 		{ -1, "State and NAT entries"				},
    180  1.41     rmind 		{ NPF_STAT_CONN_CREATE,		"state allocations"},
    181  1.41     rmind 		{ NPF_STAT_CONN_DESTROY,	"state destructions"},
    182  1.17     rmind 		{ NPF_STAT_NAT_CREATE,		"NAT entry allocations"	},
    183  1.17     rmind 		{ NPF_STAT_NAT_DESTROY,		"NAT entry destructions"},
    184  1.17     rmind 
    185  1.27     rmind 		{ -1, "Network buffers"					},
    186  1.27     rmind 		{ NPF_STAT_NBUF_NONCONTIG,	"non-contiguous cases"	},
    187  1.27     rmind 		{ NPF_STAT_NBUF_CONTIG_FAIL,	"contig alloc failures"	},
    188  1.27     rmind 
    189  1.17     rmind 		{ -1, "Invalid packet state cases"			},
    190  1.17     rmind 		{ NPF_STAT_INVALID_STATE,	"cases in total"	},
    191  1.17     rmind 		{ NPF_STAT_INVALID_STATE_TCP1,	"TCP case I"		},
    192  1.17     rmind 		{ NPF_STAT_INVALID_STATE_TCP2,	"TCP case II"		},
    193  1.17     rmind 		{ NPF_STAT_INVALID_STATE_TCP3,	"TCP case III"		},
    194  1.17     rmind 
    195  1.17     rmind 		{ -1, "Packet race cases"				},
    196  1.17     rmind 		{ NPF_STAT_RACE_NAT,		"NAT association race"	},
    197  1.41     rmind 		{ NPF_STAT_RACE_CONN,		"duplicate state race"	},
    198  1.17     rmind 
    199  1.17     rmind 		{ -1, "Fragmentation"					},
    200  1.17     rmind 		{ NPF_STAT_FRAGMENTS,		"fragments"		},
    201  1.17     rmind 		{ NPF_STAT_REASSEMBLY,		"reassembled"		},
    202  1.17     rmind 		{ NPF_STAT_REASSFAIL,		"failed reassembly"	},
    203  1.17     rmind 
    204  1.17     rmind 		{ -1, "Other"						},
    205  1.17     rmind 		{ NPF_STAT_ERROR,		"unexpected errors"	},
    206  1.17     rmind 	};
    207  1.24     rmind 	uint64_t *st = ecalloc(1, NPF_STATS_SIZE);
    208   1.3     rmind 
    209   1.3     rmind 	if (ioctl(fd, IOC_NPF_STATS, &st) != 0) {
    210   1.3     rmind 		err(EXIT_FAILURE, "ioctl(IOC_NPF_STATS)");
    211   1.3     rmind 	}
    212   1.3     rmind 
    213  1.17     rmind 	for (unsigned i = 0; i < __arraycount(stats); i++) {
    214  1.17     rmind 		const char *sname = stats[i].name;
    215  1.17     rmind 		int sidx = stats[i].index;
    216  1.17     rmind 
    217  1.17     rmind 		if (sidx == -1) {
    218  1.17     rmind 			printf("%s:\n", sname);
    219  1.17     rmind 		} else {
    220  1.17     rmind 			printf("\t%"PRIu64" %s\n", st[sidx], sname);
    221  1.17     rmind 		}
    222  1.17     rmind 	}
    223   1.4     rmind 
    224   1.3     rmind 	free(st);
    225   1.3     rmind 	return 0;
    226   1.3     rmind }
    227   1.3     rmind 
    228  1.10     rmind void
    229  1.48  christos npfctl_print_error(const npf_error_t *ne)
    230  1.10     rmind {
    231  1.48  christos 	const char *srcfile = ne->source_file;
    232  1.10     rmind 
    233  1.10     rmind 	if (srcfile) {
    234  1.48  christos 		warnx("source %s line %d", srcfile, ne->source_line);
    235  1.10     rmind 	}
    236  1.48  christos 	if (ne->id) {
    237  1.48  christos 		warnx("object: %" PRIi64, ne->id);
    238  1.10     rmind 	}
    239  1.10     rmind }
    240  1.10     rmind 
    241  1.21     rmind char *
    242  1.50  christos npfctl_print_addrmask(int alen, const char *fmt, const npf_addr_t *addr,
    243  1.50  christos     npf_netmask_t mask)
    244  1.21     rmind {
    245  1.50  christos 	const unsigned buflen = 256;
    246  1.48  christos 	char *buf = ecalloc(1, buflen);
    247  1.21     rmind 	struct sockaddr_storage ss;
    248  1.48  christos 
    249  1.48  christos 	memset(&ss, 0, sizeof(ss));
    250  1.21     rmind 
    251  1.21     rmind 	switch (alen) {
    252  1.21     rmind 	case 4: {
    253  1.21     rmind 		struct sockaddr_in *sin = (void *)&ss;
    254  1.21     rmind 		sin->sin_family = AF_INET;
    255  1.21     rmind 		memcpy(&sin->sin_addr, addr, sizeof(sin->sin_addr));
    256  1.21     rmind 		break;
    257  1.21     rmind 	}
    258  1.21     rmind 	case 16: {
    259  1.21     rmind 		struct sockaddr_in6 *sin6 = (void *)&ss;
    260  1.21     rmind 		sin6->sin6_family = AF_INET6;
    261  1.21     rmind 		memcpy(&sin6->sin6_addr, addr, sizeof(sin6->sin6_addr));
    262  1.21     rmind 		break;
    263  1.21     rmind 	}
    264  1.21     rmind 	default:
    265  1.21     rmind 		assert(false);
    266  1.21     rmind 	}
    267  1.50  christos 	sockaddr_snprintf(buf, buflen, fmt, (const void *)&ss);
    268  1.38     rmind 	if (mask && mask != NPF_NO_NETMASK) {
    269  1.48  christos 		const unsigned len = strlen(buf);
    270  1.48  christos 		snprintf(&buf[len], buflen - len, "/%u", mask);
    271  1.21     rmind 	}
    272  1.21     rmind 	return buf;
    273  1.21     rmind }
    274  1.21     rmind 
    275  1.16     joerg __dead static void
    276  1.21     rmind npfctl_table(int fd, int argc, char **argv)
    277  1.15     rmind {
    278  1.15     rmind 	static const struct tblops_s {
    279  1.15     rmind 		const char *	cmd;
    280  1.15     rmind 		int		action;
    281  1.15     rmind 	} tblops[] = {
    282  1.29     rmind 		{ "add",	NPF_CMD_TABLE_ADD		},
    283  1.29     rmind 		{ "rem",	NPF_CMD_TABLE_REMOVE		},
    284  1.29     rmind 		{ "del",	NPF_CMD_TABLE_REMOVE		},
    285  1.29     rmind 		{ "test",	NPF_CMD_TABLE_LOOKUP		},
    286  1.29     rmind 		{ "list",	NPF_CMD_TABLE_LIST		},
    287  1.37     rmind 		{ "flush",	NPF_CMD_TABLE_FLUSH		},
    288  1.21     rmind 		{ NULL,		0				}
    289  1.15     rmind 	};
    290  1.15     rmind 	npf_ioctl_table_t nct;
    291  1.15     rmind 	fam_addr_mask_t fam;
    292  1.21     rmind 	size_t buflen = 512;
    293  1.40     rmind 	char *cmd, *arg;
    294  1.15     rmind 	int n, alen;
    295  1.15     rmind 
    296  1.21     rmind 	/* Default action is list. */
    297  1.15     rmind 	memset(&nct, 0, sizeof(npf_ioctl_table_t));
    298  1.40     rmind 	nct.nct_name = argv[0];
    299  1.21     rmind 	cmd = argv[1];
    300  1.15     rmind 
    301  1.15     rmind 	for (n = 0; tblops[n].cmd != NULL; n++) {
    302  1.21     rmind 		if (strcmp(cmd, tblops[n].cmd) != 0) {
    303  1.21     rmind 			continue;
    304  1.21     rmind 		}
    305  1.29     rmind 		nct.nct_cmd = tblops[n].action;
    306  1.21     rmind 		break;
    307  1.21     rmind 	}
    308  1.21     rmind 	if (tblops[n].cmd == NULL) {
    309  1.21     rmind 		errx(EXIT_FAILURE, "invalid command '%s'", cmd);
    310  1.21     rmind 	}
    311  1.37     rmind 
    312  1.37     rmind 	switch (nct.nct_cmd) {
    313  1.37     rmind 	case NPF_CMD_TABLE_LIST:
    314  1.37     rmind 	case NPF_CMD_TABLE_FLUSH:
    315  1.40     rmind 		arg = NULL;
    316  1.37     rmind 		break;
    317  1.37     rmind 	default:
    318  1.21     rmind 		if (argc < 3) {
    319  1.21     rmind 			usage();
    320  1.15     rmind 		}
    321  1.21     rmind 		arg = argv[2];
    322  1.15     rmind 	}
    323  1.37     rmind 
    324  1.21     rmind again:
    325  1.37     rmind 	switch (nct.nct_cmd) {
    326  1.37     rmind 	case NPF_CMD_TABLE_LIST:
    327  1.24     rmind 		nct.nct_data.buf.buf = ecalloc(1, buflen);
    328  1.21     rmind 		nct.nct_data.buf.len = buflen;
    329  1.37     rmind 		break;
    330  1.37     rmind 	case NPF_CMD_TABLE_FLUSH:
    331  1.37     rmind 		break;
    332  1.37     rmind 	default:
    333  1.21     rmind 		if (!npfctl_parse_cidr(arg, &fam, &alen)) {
    334  1.21     rmind 			errx(EXIT_FAILURE, "invalid CIDR '%s'", arg);
    335  1.21     rmind 		}
    336  1.21     rmind 		nct.nct_data.ent.alen = alen;
    337  1.26     rmind 		memcpy(&nct.nct_data.ent.addr, &fam.fam_addr, alen);
    338  1.21     rmind 		nct.nct_data.ent.mask = fam.fam_mask;
    339  1.15     rmind 	}
    340  1.15     rmind 
    341  1.15     rmind 	if (ioctl(fd, IOC_NPF_TABLE, &nct) != -1) {
    342  1.15     rmind 		errno = 0;
    343  1.15     rmind 	}
    344  1.15     rmind 	switch (errno) {
    345  1.21     rmind 	case 0:
    346  1.21     rmind 		break;
    347  1.15     rmind 	case EEXIST:
    348  1.21     rmind 		errx(EXIT_FAILURE, "entry already exists or is conflicting");
    349  1.15     rmind 	case ENOENT:
    350  1.57     rmind 		errx(EXIT_FAILURE, "not found");
    351  1.15     rmind 	case EINVAL:
    352  1.21     rmind 		errx(EXIT_FAILURE, "invalid address, mask or table ID");
    353  1.21     rmind 	case ENOMEM:
    354  1.29     rmind 		if (nct.nct_cmd == NPF_CMD_TABLE_LIST) {
    355  1.21     rmind 			/* XXX */
    356  1.21     rmind 			free(nct.nct_data.buf.buf);
    357  1.21     rmind 			buflen <<= 1;
    358  1.21     rmind 			goto again;
    359  1.21     rmind 		}
    360  1.21     rmind 		/* FALLTHROUGH */
    361  1.21     rmind 	default:
    362  1.32  christos 		err(EXIT_FAILURE, "ioctl(IOC_NPF_TABLE)");
    363  1.21     rmind 	}
    364  1.21     rmind 
    365  1.29     rmind 	if (nct.nct_cmd == NPF_CMD_TABLE_LIST) {
    366  1.21     rmind 		npf_ioctl_ent_t *ent = nct.nct_data.buf.buf;
    367  1.21     rmind 		char *buf;
    368  1.21     rmind 
    369  1.21     rmind 		while (nct.nct_data.buf.len--) {
    370  1.21     rmind 			if (!ent->alen)
    371  1.21     rmind 				break;
    372  1.50  christos 			buf = npfctl_print_addrmask(ent->alen, "%a",
    373  1.21     rmind 			    &ent->addr, ent->mask);
    374  1.21     rmind 			puts(buf);
    375  1.21     rmind 			ent++;
    376  1.21     rmind 		}
    377  1.21     rmind 		free(nct.nct_data.buf.buf);
    378  1.21     rmind 	} else {
    379  1.21     rmind 		printf("%s: %s\n", getprogname(),
    380  1.29     rmind 		    nct.nct_cmd == NPF_CMD_TABLE_LOOKUP ?
    381  1.57     rmind 		    "match" : "success");
    382  1.15     rmind 	}
    383  1.21     rmind 	exit(EXIT_SUCCESS);
    384  1.15     rmind }
    385  1.15     rmind 
    386  1.29     rmind static nl_rule_t *
    387  1.29     rmind npfctl_parse_rule(int argc, char **argv)
    388  1.29     rmind {
    389  1.29     rmind 	char rule_string[1024];
    390  1.29     rmind 	nl_rule_t *rl;
    391  1.29     rmind 
    392  1.29     rmind 	/* Get the rule string and parse it. */
    393  1.38     rmind 	if (!join(rule_string, sizeof(rule_string), argc, argv, " ")) {
    394  1.29     rmind 		errx(EXIT_FAILURE, "command too long");
    395  1.29     rmind 	}
    396  1.29     rmind 	npfctl_parse_string(rule_string);
    397  1.29     rmind 	if ((rl = npfctl_rule_ref()) == NULL) {
    398  1.29     rmind 		errx(EXIT_FAILURE, "could not parse the rule");
    399  1.29     rmind 	}
    400  1.29     rmind 	return rl;
    401  1.29     rmind }
    402  1.29     rmind 
    403  1.48  christos #ifdef __NetBSD__
    404  1.49  christos static unsigned char *
    405  1.49  christos SHA1(const unsigned char *d, size_t l, unsigned char *md)
    406  1.48  christos {
    407  1.48  christos 	SHA1_CTX c;
    408  1.48  christos 
    409  1.48  christos 	SHA1Init(&c);
    410  1.48  christos 	SHA1Update(&c, d, l);
    411  1.48  christos 	SHA1Final(md, &c);
    412  1.48  christos 	return md;
    413  1.47  christos }
    414  1.48  christos #endif
    415  1.47  christos 
    416  1.47  christos static void
    417  1.29     rmind npfctl_generate_key(nl_rule_t *rl, void *key)
    418  1.29     rmind {
    419  1.29     rmind 	void *meta;
    420  1.29     rmind 	size_t len;
    421  1.29     rmind 
    422  1.29     rmind 	if ((meta = npf_rule_export(rl, &len)) == NULL) {
    423  1.29     rmind 		errx(EXIT_FAILURE, "error generating rule key");
    424  1.29     rmind 	}
    425  1.48  christos 	__CTASSERT(NPF_RULE_MAXKEYLEN >= SHA_DIGEST_LENGTH);
    426  1.29     rmind 	memset(key, 0, NPF_RULE_MAXKEYLEN);
    427  1.48  christos 	SHA1(meta, len, key);
    428  1.29     rmind 	free(meta);
    429  1.29     rmind }
    430  1.29     rmind 
    431  1.29     rmind __dead static void
    432  1.29     rmind npfctl_rule(int fd, int argc, char **argv)
    433  1.29     rmind {
    434  1.29     rmind 	static const struct ruleops_s {
    435  1.29     rmind 		const char *	cmd;
    436  1.29     rmind 		int		action;
    437  1.36     rmind 		bool		extra_arg;
    438  1.29     rmind 	} ruleops[] = {
    439  1.36     rmind 		{ "add",	NPF_CMD_RULE_ADD,	true	},
    440  1.36     rmind 		{ "rem",	NPF_CMD_RULE_REMKEY,	true	},
    441  1.36     rmind 		{ "del",	NPF_CMD_RULE_REMKEY,	true	},
    442  1.36     rmind 		{ "rem-id",	NPF_CMD_RULE_REMOVE,	true	},
    443  1.36     rmind 		{ "list",	NPF_CMD_RULE_LIST,	false	},
    444  1.36     rmind 		{ "flush",	NPF_CMD_RULE_FLUSH,	false	},
    445  1.36     rmind 		{ NULL,		0,			0	}
    446  1.29     rmind 	};
    447  1.29     rmind 	uint8_t key[NPF_RULE_MAXKEYLEN];
    448  1.29     rmind 	const char *ruleset_name = argv[0];
    449  1.29     rmind 	const char *cmd = argv[1];
    450  1.29     rmind 	int error, action = 0;
    451  1.31     rmind 	uint64_t rule_id;
    452  1.36     rmind 	bool extra_arg;
    453  1.29     rmind 	nl_rule_t *rl;
    454  1.29     rmind 
    455  1.29     rmind 	for (int n = 0; ruleops[n].cmd != NULL; n++) {
    456  1.29     rmind 		if (strcmp(cmd, ruleops[n].cmd) == 0) {
    457  1.29     rmind 			action = ruleops[n].action;
    458  1.36     rmind 			extra_arg = ruleops[n].extra_arg;
    459  1.29     rmind 			break;
    460  1.29     rmind 		}
    461  1.29     rmind 	}
    462  1.36     rmind 	argc -= 2;
    463  1.36     rmind 	argv += 2;
    464  1.29     rmind 
    465  1.36     rmind 	if (!action || (extra_arg && argc == 0)) {
    466  1.29     rmind 		usage();
    467  1.29     rmind 	}
    468  1.29     rmind 
    469  1.29     rmind 	switch (action) {
    470  1.29     rmind 	case NPF_CMD_RULE_ADD:
    471  1.29     rmind 		rl = npfctl_parse_rule(argc, argv);
    472  1.29     rmind 		npfctl_generate_key(rl, key);
    473  1.29     rmind 		npf_rule_setkey(rl, key, sizeof(key));
    474  1.29     rmind 		error = npf_ruleset_add(fd, ruleset_name, rl, &rule_id);
    475  1.29     rmind 		break;
    476  1.29     rmind 	case NPF_CMD_RULE_REMKEY:
    477  1.29     rmind 		rl = npfctl_parse_rule(argc, argv);
    478  1.29     rmind 		npfctl_generate_key(rl, key);
    479  1.29     rmind 		error = npf_ruleset_remkey(fd, ruleset_name, key, sizeof(key));
    480  1.29     rmind 		break;
    481  1.29     rmind 	case NPF_CMD_RULE_REMOVE:
    482  1.31     rmind 		rule_id = strtoull(argv[0], NULL, 16);
    483  1.29     rmind 		error = npf_ruleset_remove(fd, ruleset_name, rule_id);
    484  1.29     rmind 		break;
    485  1.30     rmind 	case NPF_CMD_RULE_LIST:
    486  1.30     rmind 		error = npfctl_ruleset_show(fd, ruleset_name);
    487  1.30     rmind 		break;
    488  1.30     rmind 	case NPF_CMD_RULE_FLUSH:
    489  1.30     rmind 		error = npf_ruleset_flush(fd, ruleset_name);
    490  1.30     rmind 		break;
    491  1.29     rmind 	default:
    492  1.48  christos 		abort();
    493  1.29     rmind 	}
    494  1.29     rmind 
    495  1.29     rmind 	switch (error) {
    496  1.29     rmind 	case 0:
    497  1.29     rmind 		/* Success. */
    498  1.29     rmind 		break;
    499  1.31     rmind 	case ESRCH:
    500  1.31     rmind 		errx(EXIT_FAILURE, "ruleset \"%s\" not found", ruleset_name);
    501  1.29     rmind 	case ENOENT:
    502  1.31     rmind 		errx(EXIT_FAILURE, "rule was not found");
    503  1.29     rmind 	default:
    504  1.29     rmind 		errx(EXIT_FAILURE, "rule operation: %s", strerror(error));
    505  1.29     rmind 	}
    506  1.29     rmind 	if (action == NPF_CMD_RULE_ADD) {
    507  1.31     rmind 		printf("OK %" PRIx64 "\n", rule_id);
    508  1.29     rmind 	}
    509  1.29     rmind 	exit(EXIT_SUCCESS);
    510  1.29     rmind }
    511  1.29     rmind 
    512  1.45  christos static bool bpfjit = true;
    513  1.45  christos 
    514  1.45  christos void
    515  1.45  christos npfctl_bpfjit(bool onoff)
    516  1.45  christos {
    517  1.45  christos 	bpfjit = onoff;
    518  1.45  christos }
    519  1.45  christos 
    520  1.44     rmind static void
    521  1.44     rmind npfctl_preload_bpfjit(void)
    522  1.44     rmind {
    523  1.48  christos #ifdef __NetBSD__
    524  1.44     rmind 	modctl_load_t args = {
    525  1.44     rmind 		.ml_filename = "bpfjit",
    526  1.44     rmind 		.ml_flags = MODCTL_NO_PROP,
    527  1.44     rmind 		.ml_props = NULL,
    528  1.44     rmind 		.ml_propslen = 0
    529  1.44     rmind 	};
    530  1.44     rmind 
    531  1.45  christos 	if (!bpfjit)
    532  1.45  christos 		return;
    533  1.45  christos 
    534  1.44     rmind 	if (modctl(MODCTL_LOAD, &args) != 0 && errno != EEXIST) {
    535  1.45  christos 		static const char *p = "; performance will be degraded";
    536  1.45  christos 		if (errno == ENOENT)
    537  1.45  christos 			warnx("the bpfjit module seems to be missing%s", p);
    538  1.45  christos 		else
    539  1.45  christos 			warn("error loading the bpfjit module%s", p);
    540  1.45  christos 		warnx("To disable this warning `set bpf.jit off' in "
    541  1.45  christos 		    "/etc/npf.conf");
    542  1.44     rmind 	}
    543  1.48  christos #endif
    544  1.44     rmind }
    545  1.44     rmind 
    546  1.41     rmind static int
    547  1.48  christos npfctl_load(int fd)
    548  1.41     rmind {
    549  1.41     rmind 	nl_config_t *ncf;
    550  1.48  christos 	npf_error_t errinfo;
    551  1.48  christos 	struct stat sb;
    552  1.48  christos 	size_t blen;
    553  1.48  christos 	void *blob;
    554  1.41     rmind 	int error;
    555  1.41     rmind 
    556  1.48  christos 	/*
    557  1.48  christos 	 * The file may change while reading - we are not handling this,
    558  1.48  christos 	 * leaving this responsibility for the caller.
    559  1.48  christos 	 */
    560  1.48  christos 	if (stat(NPF_DB_PATH, &sb) == -1) {
    561  1.48  christos 		err(EXIT_FAILURE, "stat");
    562  1.48  christos 	}
    563  1.48  christos 	if ((blen = sb.st_size) == 0) {
    564  1.48  christos 		err(EXIT_FAILURE, "saved configuration file is empty");
    565  1.48  christos 	}
    566  1.48  christos 	if ((blob = mmap(NULL, blen, PROT_READ,
    567  1.48  christos 	    MAP_FILE | MAP_PRIVATE, fd, 0)) == MAP_FAILED) {
    568  1.48  christos 		err(EXIT_FAILURE, "mmap");
    569  1.48  christos 	}
    570  1.48  christos 	ncf = npf_config_import(blob, blen);
    571  1.48  christos 	munmap(blob, blen);
    572  1.41     rmind 	if (ncf == NULL) {
    573  1.41     rmind 		return errno;
    574  1.41     rmind 	}
    575  1.41     rmind 
    576  1.48  christos 	/*
    577  1.48  christos 	 * Configuration imported - submit it now.
    578  1.48  christos 	 **/
    579  1.48  christos 	errno = error = npf_config_submit(ncf, fd, &errinfo);
    580  1.43     rmind 	if (error) {
    581  1.48  christos 		npfctl_print_error(&errinfo);
    582  1.43     rmind 	}
    583  1.41     rmind 	npf_config_destroy(ncf);
    584  1.41     rmind 	return error;
    585  1.41     rmind }
    586  1.41     rmind 
    587  1.50  christos struct npf_conn_filter {
    588  1.50  christos 	uint16_t alen;
    589  1.50  christos 	const char *ifname;
    590  1.50  christos 	bool nat;
    591  1.50  christos 	bool wide;
    592  1.50  christos 	bool name;
    593  1.50  christos 	int width;
    594  1.50  christos 	FILE *fp;
    595  1.50  christos };
    596  1.50  christos 
    597  1.50  christos static int
    598  1.52     rmind npfctl_conn_print(unsigned alen, const npf_addr_t *a, const in_port_t *p,
    599  1.50  christos     const char *ifname, void *v)
    600  1.50  christos {
    601  1.50  christos 	struct npf_conn_filter *fil = v;
    602  1.50  christos 	FILE *fp = fil->fp;
    603  1.50  christos 	char *src, *dst;
    604  1.50  christos 
    605  1.50  christos 	if (fil->ifname && strcmp(ifname, fil->ifname) != 0)
    606  1.50  christos 		return 0;
    607  1.50  christos 	if (fil->alen && alen != fil->alen)
    608  1.50  christos 		return 0;
    609  1.50  christos 	if (fil->nat && !p[2])
    610  1.50  christos 		return 0;
    611  1.50  christos 
    612  1.50  christos 	int w = fil->width;
    613  1.50  christos 	const char *fmt = fil->name ? "%A" :
    614  1.50  christos 	    (alen == sizeof(struct in_addr) ? "%a" : "[%a]");
    615  1.50  christos 	src = npfctl_print_addrmask(alen, fmt, &a[0], NPF_NO_NETMASK);
    616  1.50  christos 	dst = npfctl_print_addrmask(alen, fmt, &a[1], NPF_NO_NETMASK);
    617  1.50  christos 	if (fil->wide)
    618  1.50  christos 		fprintf(fp, "%s:%d %s:%d", src, p[0], dst, p[1]);
    619  1.50  christos 	else
    620  1.50  christos 		fprintf(fp, "%*.*s:%-5d %*.*s:%-5d", w, w, src, p[0],
    621  1.50  christos 		    w, w, dst, p[1]);
    622  1.50  christos 	free(src);
    623  1.50  christos 	free(dst);
    624  1.50  christos 	if (!p[2]) {
    625  1.50  christos 		fputc('\n', fp);
    626  1.50  christos 		return 1;
    627  1.50  christos 	}
    628  1.54     ozaki 	fprintf(fp, " via %s:%d\n", ifname, p[2]);
    629  1.50  christos 	return 1;
    630  1.50  christos }
    631  1.50  christos 
    632  1.50  christos 
    633  1.50  christos static int
    634  1.50  christos npfctl_conn_list(int fd, int argc, char **argv)
    635  1.50  christos {
    636  1.50  christos 	struct npf_conn_filter f;
    637  1.50  christos 	int c;
    638  1.50  christos 	int header = true;
    639  1.50  christos 	memset(&f, 0, sizeof(f));
    640  1.50  christos 
    641  1.50  christos 	argc--;
    642  1.50  christos 	argv++;
    643  1.50  christos 
    644  1.50  christos 	while ((c = getopt(argc, argv, "46hi:nNw")) != -1) {
    645  1.50  christos 		switch (c) {
    646  1.50  christos 		case '4':
    647  1.50  christos 			f.alen = sizeof(struct in_addr);
    648  1.50  christos 			break;
    649  1.50  christos 		case '6':
    650  1.50  christos 			f.alen = sizeof(struct in6_addr);
    651  1.50  christos 			break;
    652  1.50  christos 		case 'h':
    653  1.50  christos 			header = false;
    654  1.50  christos 		case 'i':
    655  1.50  christos 			f.ifname = optarg;
    656  1.50  christos 			break;
    657  1.50  christos 		case 'n':
    658  1.50  christos 			f.nat = true;
    659  1.50  christos 			break;
    660  1.50  christos 		case 'N':
    661  1.50  christos 			f.name = true;
    662  1.50  christos 			break;
    663  1.50  christos 		case 'w':
    664  1.50  christos 			f.wide = true;
    665  1.50  christos 			break;
    666  1.50  christos 		default:
    667  1.50  christos 			fprintf(stderr,
    668  1.50  christos 			    "Usage: %s list [-46hnNw] [-i <ifname>]\n",
    669  1.50  christos 			    getprogname());
    670  1.50  christos 			exit(EXIT_FAILURE);
    671  1.50  christos 		}
    672  1.50  christos 	}
    673  1.50  christos 	f.width = f.alen == sizeof(struct in_addr) ? 25 : 41;
    674  1.50  christos 	int w = f.width + 6;
    675  1.50  christos 	f.fp = stdout;
    676  1.50  christos 	if (header)
    677  1.50  christos 		fprintf(f.fp, "%*.*s %*.*s\n",
    678  1.50  christos 		    w, w, "From address:port ", w, w, "To address:port ");
    679  1.52     rmind 
    680  1.50  christos 	npf_conn_list(fd, npfctl_conn_print, &f);
    681  1.50  christos 	return 0;
    682  1.50  christos }
    683  1.50  christos 
    684  1.52     rmind static int
    685  1.52     rmind npfctl_open_dev(const char *path)
    686   1.1     rmind {
    687  1.56     rmind 	int fd, kernver;
    688   1.1     rmind 
    689  1.52     rmind 	fd = open(path, O_RDONLY);
    690   1.1     rmind 	if (fd == -1) {
    691  1.52     rmind 		err(EXIT_FAILURE, "cannot open '%s'", path);
    692   1.1     rmind 	}
    693  1.56     rmind 	if (ioctl(fd, IOC_NPF_VERSION, &kernver) == -1) {
    694  1.32  christos 		err(EXIT_FAILURE, "ioctl(IOC_NPF_VERSION)");
    695  1.15     rmind 	}
    696  1.56     rmind 	if (kernver != NPF_VERSION) {
    697   1.4     rmind 		errx(EXIT_FAILURE,
    698  1.14     rmind 		    "incompatible NPF interface version (%d, kernel %d)\n"
    699  1.56     rmind 		    "Hint: update %s?", NPF_VERSION, kernver,
    700  1.56     rmind 		    kernver > NPF_VERSION ? "userland" : "kernel");
    701   1.1     rmind 	}
    702  1.52     rmind 	return fd;
    703  1.52     rmind }
    704  1.52     rmind 
    705  1.52     rmind static void
    706  1.52     rmind npfctl(int action, int argc, char **argv)
    707  1.52     rmind {
    708  1.52     rmind 	int fd, boolval, ret = 0;
    709  1.52     rmind 	const char *fun = "";
    710  1.52     rmind 	nl_config_t *ncf;
    711  1.52     rmind 
    712  1.52     rmind 	switch (action) {
    713  1.52     rmind 	case NPFCTL_VALIDATE:
    714  1.52     rmind 	case NPFCTL_DEBUG:
    715  1.52     rmind 		fd = 0;
    716  1.52     rmind 		break;
    717  1.52     rmind 	default:
    718  1.52     rmind 		fd = npfctl_open_dev(NPF_DEV_PATH);
    719  1.52     rmind 	}
    720  1.29     rmind 
    721   1.1     rmind 	switch (action) {
    722   1.1     rmind 	case NPFCTL_START:
    723   1.1     rmind 		boolval = true;
    724   1.1     rmind 		ret = ioctl(fd, IOC_NPF_SWITCH, &boolval);
    725  1.32  christos 		fun = "ioctl(IOC_NPF_SWITCH)";
    726   1.1     rmind 		break;
    727   1.1     rmind 	case NPFCTL_STOP:
    728   1.1     rmind 		boolval = false;
    729   1.1     rmind 		ret = ioctl(fd, IOC_NPF_SWITCH, &boolval);
    730  1.32  christos 		fun = "ioctl(IOC_NPF_SWITCH)";
    731   1.1     rmind 		break;
    732   1.1     rmind 	case NPFCTL_RELOAD:
    733   1.8     rmind 		npfctl_config_init(false);
    734  1.29     rmind 		npfctl_parse_file(argc < 3 ? NPF_CONF_PATH : argv[2]);
    735  1.46  christos 		npfctl_preload_bpfjit();
    736  1.56     rmind 		errno = ret = npfctl_config_send(fd);
    737  1.32  christos 		fun = "npfctl_config_send";
    738   1.1     rmind 		break;
    739  1.11     rmind 	case NPFCTL_SHOWCONF:
    740  1.11     rmind 		ret = npfctl_config_show(fd);
    741  1.32  christos 		fun = "npfctl_config_show";
    742  1.11     rmind 		break;
    743   1.1     rmind 	case NPFCTL_FLUSH:
    744   1.9     rmind 		ret = npf_config_flush(fd);
    745  1.32  christos 		fun = "npf_config_flush";
    746   1.1     rmind 		break;
    747   1.1     rmind 	case NPFCTL_TABLE:
    748  1.21     rmind 		if ((argc -= 2) < 2) {
    749   1.1     rmind 			usage();
    750   1.1     rmind 		}
    751  1.21     rmind 		argv += 2;
    752  1.21     rmind 		npfctl_table(fd, argc, argv);
    753   1.1     rmind 		break;
    754  1.29     rmind 	case NPFCTL_RULE:
    755  1.29     rmind 		if ((argc -= 2) < 2) {
    756  1.29     rmind 			usage();
    757  1.29     rmind 		}
    758  1.29     rmind 		argv += 2;
    759  1.29     rmind 		npfctl_rule(fd, argc, argv);
    760  1.29     rmind 		break;
    761  1.41     rmind 	case NPFCTL_LOAD:
    762  1.44     rmind 		npfctl_preload_bpfjit();
    763  1.41     rmind 		ret = npfctl_load(fd);
    764  1.41     rmind 		fun = "npfctl_config_load";
    765  1.41     rmind 		break;
    766  1.41     rmind 	case NPFCTL_SAVE:
    767  1.48  christos 		ncf = npf_config_retrieve(fd);
    768  1.48  christos 		if (ncf) {
    769  1.48  christos 			npfctl_config_save(ncf, NPF_DB_PATH);
    770  1.48  christos 			npf_config_destroy(ncf);
    771  1.48  christos 		} else {
    772  1.48  christos 			ret = errno;
    773  1.48  christos 		}
    774  1.41     rmind 		fun = "npfctl_config_save";
    775  1.41     rmind 		break;
    776   1.3     rmind 	case NPFCTL_STATS:
    777   1.3     rmind 		ret = npfctl_print_stats(fd);
    778  1.32  christos 		fun = "npfctl_print_stats";
    779   1.3     rmind 		break;
    780  1.50  christos 	case NPFCTL_CONN_LIST:
    781  1.50  christos 		ret = npfctl_conn_list(fd, argc, argv);
    782  1.50  christos 		fun = "npfctl_conn_list";
    783  1.50  christos 		break;
    784  1.52     rmind 	case NPFCTL_VALIDATE:
    785  1.52     rmind 		npfctl_config_init(false);
    786  1.52     rmind 		npfctl_parse_file(argc > 2 ? argv[2] : NPF_CONF_PATH);
    787  1.52     rmind 		ret = npfctl_config_show(0);
    788  1.52     rmind 		fun = "npfctl_config_show";
    789  1.52     rmind 		break;
    790  1.52     rmind 	case NPFCTL_DEBUG:
    791  1.52     rmind 		npfctl_config_init(true);
    792  1.52     rmind 		npfctl_parse_file(argc > 2 ? argv[2] : NPF_CONF_PATH);
    793  1.56     rmind 		npfctl_config_debug(argc > 3 ? argv[3] : "/tmp/npf.nvlist");
    794  1.52     rmind 		break;
    795   1.1     rmind 	}
    796   1.7    zoltan 	if (ret) {
    797  1.32  christos 		err(EXIT_FAILURE, "%s", fun);
    798   1.1     rmind 	}
    799  1.52     rmind 	if (fd) {
    800  1.52     rmind 		close(fd);
    801  1.52     rmind 	}
    802   1.1     rmind }
    803   1.1     rmind 
    804   1.1     rmind int
    805   1.1     rmind main(int argc, char **argv)
    806   1.1     rmind {
    807   1.1     rmind 	char *cmd;
    808   1.1     rmind 
    809   1.1     rmind 	if (argc < 2) {
    810   1.1     rmind 		usage();
    811   1.1     rmind 	}
    812  1.48  christos 	npfctl_show_init();
    813   1.1     rmind 	cmd = argv[1];
    814   1.1     rmind 
    815  1.12     rmind 	/* Find and call the subroutine. */
    816   1.8     rmind 	for (int n = 0; operations[n].cmd != NULL; n++) {
    817  1.25     rmind 		const char *opcmd = operations[n].cmd;
    818  1.25     rmind 		if (strncmp(cmd, opcmd, strlen(opcmd)) != 0)
    819   1.1     rmind 			continue;
    820   1.1     rmind 		npfctl(operations[n].action, argc, argv);
    821   1.8     rmind 		return EXIT_SUCCESS;
    822   1.1     rmind 	}
    823   1.3     rmind 	usage();
    824   1.1     rmind }
    825