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