authpf.c revision 127024
1211088Smjacob/*	$FreeBSD: head/contrib/pf/authpf/authpf.c 127024 2004-03-15 13:41:17Z mlaier $	*/
2211088Smjacob/*	$OpenBSD: authpf.c,v 1.68 2003/08/21 19:13:23 frantzen Exp $	*/
3211088Smjacob
4211088Smjacob/*
5211088Smjacob * Copyright (C) 1998 - 2002 Bob Beck (beck@openbsd.org).
6211088Smjacob *
7211088Smjacob * Redistribution and use in source and binary forms, with or without
8211088Smjacob * modification, are permitted provided that the following conditions
9211088Smjacob * are met:
10211088Smjacob * 1. Redistributions of source code must retain the above copyright
11211088Smjacob *    notice, this list of conditions and the following disclaimer.
12211088Smjacob * 2. Redistributions in binary form must reproduce the above copyright
13211088Smjacob *    notice, this list of conditions and the following disclaimer in the
14211088Smjacob *    documentation and/or other materials provided with the distribution.
15211088Smjacob *
16211088Smjacob * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17211088Smjacob * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18211088Smjacob * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19211088Smjacob * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20211088Smjacob * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21211088Smjacob * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22211088Smjacob * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23211088Smjacob * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24211088Smjacob * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25211088Smjacob * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26211088Smjacob * SUCH DAMAGE.
27211088Smjacob */
28211088Smjacob
29211088Smjacob#include <sys/param.h>
30211088Smjacob#include <sys/file.h>
31211088Smjacob#include <sys/ioctl.h>
32#include <sys/socket.h>
33#include <sys/time.h>
34
35#include <net/if.h>
36#include <net/pfvar.h>
37#include <arpa/inet.h>
38
39#include <err.h>
40#include <errno.h>
41#include <pwd.h>
42#include <signal.h>
43#include <stdio.h>
44#include <stdlib.h>
45#include <string.h>
46#include <syslog.h>
47#include <unistd.h>
48
49#include <pfctl_parser.h>
50
51#include "pathnames.h"
52
53extern int	symset(const char *, const char *, int);
54
55static int	read_config(FILE *);
56static void	print_message(char *);
57static int	allowed_luser(char *);
58static int	check_luser(char *, char *);
59static int	remove_stale_rulesets(void);
60static int	change_filter(int, const char *, const char *);
61static void	authpf_kill_states(void);
62
63int	dev;			/* pf device */
64char	anchorname[PF_ANCHOR_NAME_SIZE] = "authpf";
65char	rulesetname[PF_RULESET_NAME_SIZE];
66
67FILE	*pidfp;
68char	*infile;		/* file name printed by yyerror() in parse.y */
69char	 luser[MAXLOGNAME];	/* username */
70char	 ipsrc[256];		/* ip as a string */
71char	 pidfile[MAXPATHLEN];	/* we save pid in this file. */
72
73struct timeval	Tstart, Tend;	/* start and end times of session */
74
75volatile sig_atomic_t	want_death;
76static void		need_death(int signo);
77#ifdef __FreeBSD__
78static __dead2 void	do_death(int);
79#else
80static __dead void	do_death(int);
81#endif
82
83/*
84 * User shell for authenticating gateways. Sole purpose is to allow
85 * a user to ssh to a gateway, and have the gateway modify packet
86 * filters to allow access, then remove access when the user finishes
87 * up. Meant to be used only from ssh(1) connections.
88 */
89int
90main(int argc, char *argv[])
91{
92	int		 lockcnt = 0, n, pidfd;
93	FILE		*config;
94	struct in_addr	 ina;
95	struct passwd	*pw;
96	char		*cp;
97	uid_t		 uid;
98
99	if ((n = snprintf(rulesetname, sizeof(rulesetname), "%ld",
100	    (long)getpid())) < 0 || n >= sizeof(rulesetname)) {
101		syslog(LOG_ERR, "pid too large for ruleset name");
102		exit(1);
103	}
104
105	config = fopen(PATH_CONFFILE, "r");
106
107	if ((cp = getenv("SSH_TTY")) == NULL) {
108		syslog(LOG_ERR, "non-interactive session connection for authpf");
109		exit(1);
110	}
111
112	if ((cp = getenv("SSH_CLIENT")) == NULL) {
113		syslog(LOG_ERR, "cannot determine connection source");
114		exit(1);
115	}
116
117	if (strlcpy(ipsrc, cp, sizeof(ipsrc)) >= sizeof(ipsrc)) {
118		syslog(LOG_ERR, "SSH_CLIENT variable too long");
119		exit(1);
120	}
121	cp = strchr(ipsrc, ' ');
122	if (!cp) {
123		syslog(LOG_ERR, "corrupt SSH_CLIENT variable %s", ipsrc);
124		exit(1);
125	}
126	*cp = '\0';
127	if (inet_pton(AF_INET, ipsrc, &ina) != 1) {
128		syslog(LOG_ERR,
129		    "cannot determine IP from SSH_CLIENT %s", ipsrc);
130		exit(1);
131	}
132
133	/* open the pf device */
134	dev = open(PATH_DEVFILE, O_RDWR);
135	if (dev == -1) {
136		syslog(LOG_ERR, "cannot open packet filter device (%m)");
137		goto die;
138	}
139
140	uid = getuid();
141	pw = getpwuid(uid);
142	if (pw == NULL) {
143		syslog(LOG_ERR, "cannot find user for uid %u", uid);
144		goto die;
145	}
146	if (strcmp(pw->pw_shell, PATH_AUTHPF_SHELL)) {
147		syslog(LOG_ERR, "wrong shell for user %s, uid %u",
148		    pw->pw_name, pw->pw_uid);
149		goto die;
150	}
151
152	/*
153	 * Paranoia, but this data _does_ come from outside authpf, and
154	 * truncation would be bad.
155	 */
156	if (strlcpy(luser, pw->pw_name, sizeof(luser)) >= sizeof(luser)) {
157		syslog(LOG_ERR, "username too long: %s", pw->pw_name);
158		goto die;
159	}
160
161	/* Make our entry in /var/authpf as /var/authpf/ipaddr */
162	n = snprintf(pidfile, sizeof(pidfile), "%s/%s", PATH_PIDFILE, ipsrc);
163	if (n < 0 || (u_int)n >= sizeof(pidfile)) {
164		syslog(LOG_ERR, "path to pidfile too long");
165		goto die;
166	}
167
168	/*
169	 * If someone else is already using this ip, then this person
170	 * wants to switch users - so kill the old process and exit
171	 * as well.
172	 *
173	 * Note, we could print a message and tell them to log out, but the
174	 * usual case of this is that someone has left themselves logged in,
175	 * with the authenticated connection iconized and someone else walks
176	 * up to use and automatically logs in before using. If this just
177	 * gets rid of the old one silently, the new user never knows they
178	 * could have used someone else's old authentication. If we
179	 * tell them to log out before switching users it is an invitation
180	 * for abuse.
181	 */
182
183	do {
184		int	save_errno, otherpid = -1;
185		char	otherluser[MAXLOGNAME];
186
187		if ((pidfd = open(pidfile, O_RDWR|O_CREAT, 0644)) == -1 ||
188		    (pidfp = fdopen(pidfd, "r+")) == NULL) {
189			if (pidfd != -1)
190				close(pidfd);
191			syslog(LOG_ERR, "cannot open or create %s: %s", pidfile,
192			    strerror(errno));
193			goto die;
194		}
195
196		if (flock(fileno(pidfp), LOCK_EX|LOCK_NB) == 0)
197			break;
198		save_errno = errno;
199
200		/* Mark our pid, and username to our file. */
201
202		rewind(pidfp);
203		/* 31 == MAXLOGNAME - 1 */
204		if (fscanf(pidfp, "%d\n%31s\n", &otherpid, otherluser) != 2)
205			otherpid = -1;
206		syslog(LOG_DEBUG, "tried to lock %s, in use by pid %d: %s",
207		    pidfile, otherpid, strerror(save_errno));
208
209		if (otherpid > 0) {
210			syslog(LOG_INFO,
211			    "killing prior auth (pid %d) of %s by user %s",
212			    otherpid, ipsrc, otherluser);
213			if (kill((pid_t) otherpid, SIGTERM) == -1) {
214				syslog(LOG_INFO,
215				    "could not kill process %d: (%m)",
216				    otherpid);
217			}
218		}
219
220		/*
221		 * we try to kill the previous process and acquire the lock
222		 * for 10 seconds, trying once a second. if we can't after
223		 * 10 attempts we log an error and give up
224		 */
225		if (++lockcnt > 10) {
226			syslog(LOG_ERR, "cannot kill previous authpf (pid %d)",
227			    otherpid);
228			goto dogdeath;
229		}
230		sleep(1);
231
232		/* re-open, and try again. The previous authpf process
233		 * we killed above should unlink the file and release
234		 * it's lock, giving us a chance to get it now
235		 */
236		fclose(pidfp);
237	} while (1);
238
239	/* revoke privs */
240	seteuid(getuid());
241	setuid(getuid());
242
243	if (!check_luser(PATH_BAN_DIR, luser) || !allowed_luser(luser))
244		do_death(0);
245
246	openlog("authpf", LOG_PID | LOG_NDELAY, LOG_DAEMON);
247	if (config == NULL || read_config(config))
248		do_death(0);
249
250	if (remove_stale_rulesets())
251		do_death(0);
252
253	/* We appear to be making headway, so actually mark our pid */
254	rewind(pidfp);
255	fprintf(pidfp, "%ld\n%s\n", (long)getpid(), luser);
256	fflush(pidfp);
257	(void) ftruncate(fileno(pidfp), ftell(pidfp));
258
259	if (change_filter(1, luser, ipsrc) == -1) {
260		printf("Unable to modify filters\r\n");
261		do_death(1);
262	}
263
264	signal(SIGTERM, need_death);
265	signal(SIGINT, need_death);
266	signal(SIGALRM, need_death);
267	signal(SIGPIPE, need_death);
268	signal(SIGHUP, need_death);
269	signal(SIGSTOP, need_death);
270	signal(SIGTSTP, need_death);
271	while (1) {
272		printf("\r\nHello %s, ", luser);
273		printf("You are authenticated from host \"%s\"\r\n", ipsrc);
274		setproctitle("%s@%s", luser, ipsrc);
275		print_message(PATH_MESSAGE);
276		while (1) {
277			sleep(10);
278			if (want_death)
279				do_death(1);
280		}
281	}
282
283	/* NOTREACHED */
284dogdeath:
285	printf("\r\n\r\nSorry, this service is currently unavailable due to ");
286	printf("technical difficulties\r\n\r\n");
287	print_message(PATH_PROBLEM);
288	printf("\r\nYour authentication process (pid %ld) was unable to run\n",
289	    (long)getpid());
290	sleep(180); /* them lusers read reaaaaal slow */
291die:
292	do_death(0);
293
294	/* NOTREACHED */
295}
296
297/*
298 * reads config file in PATH_CONFFILE to set optional behaviours up
299 */
300static int
301read_config(FILE *f)
302{
303	char	buf[1024];
304	int	i = 0;
305
306	do {
307		char	**ap;
308		char	 *pair[4], *cp, *tp;
309		int	  len;
310
311		if (fgets(buf, sizeof(buf), f) == NULL) {
312			fclose(f);
313			return (0);
314		}
315		i++;
316		len = strlen(buf);
317		if (buf[len - 1] != '\n' && !feof(f)) {
318			syslog(LOG_ERR, "line %d too long in %s", i,
319			    PATH_CONFFILE);
320			return (1);
321		}
322		buf[len - 1] = '\0';
323
324		for (cp = buf; *cp == ' ' || *cp == '\t'; cp++)
325			; /* nothing */
326
327		if (!*cp || *cp == '#' || *cp == '\n')
328			continue;
329
330		for (ap = pair; ap < &pair[3] &&
331		    (*ap = strsep(&cp, "=")) != NULL; ) {
332			if (**ap != '\0')
333				ap++;
334		}
335		if (ap != &pair[2])
336			goto parse_error;
337
338		tp = pair[1] + strlen(pair[1]);
339		while ((*tp == ' ' || *tp == '\t') && tp >= pair[1])
340			*tp-- = '\0';
341
342		if (strcasecmp(pair[0], "anchor") == 0) {
343			if (!pair[1][0] || strlcpy(anchorname, pair[1],
344			    sizeof(anchorname)) >= sizeof(anchorname))
345				goto parse_error;
346		}
347	} while (!feof(f) && !ferror(f));
348	fclose(f);
349	return (0);
350
351parse_error:
352	fclose(f);
353	syslog(LOG_ERR, "parse error, line %d of %s", i, PATH_CONFFILE);
354	return (1);
355}
356
357
358/*
359 * splatter a file to stdout - max line length of 1024,
360 * used for spitting message files at users to tell them
361 * they've been bad or we're unavailable.
362 */
363static void
364print_message(char *filename)
365{
366	char	 buf[1024];
367	FILE	*f;
368
369	if ((f = fopen(filename, "r")) == NULL)
370		return; /* fail silently, we don't care if it isn't there */
371
372	do {
373		if (fgets(buf, sizeof(buf), f) == NULL) {
374			fflush(stdout);
375			fclose(f);
376			return;
377		}
378	} while (fputs(buf, stdout) != EOF && !feof(f));
379	fflush(stdout);
380	fclose(f);
381}
382
383/*
384 * allowed_luser checks to see if user "luser" is allowed to
385 * use this gateway by virtue of being listed in an allowed
386 * users file, namely /etc/authpf/authpf.allow .
387 *
388 * If /etc/authpf/authpf.allow does not exist, then we assume that
389 * all users who are allowed in by sshd(8) are permitted to
390 * use this gateway. If /etc/authpf/authpf.allow does exist, then a
391 * user must be listed if the connection is to continue, else
392 * the session terminates in the same manner as being banned.
393 */
394static int
395allowed_luser(char *luser)
396{
397	char	*buf, *lbuf;
398	int	 matched;
399	size_t	 len;
400	FILE	*f;
401
402	if ((f = fopen(PATH_ALLOWFILE, "r")) == NULL) {
403		if (errno == ENOENT) {
404			/*
405			 * allowfile doesn't exist, thus this gateway
406			 * isn't restricted to certain users...
407			 */
408			return (1);
409		}
410
411		/*
412		 * luser may in fact be allowed, but we can't open
413		 * the file even though it's there. probably a config
414		 * problem.
415		 */
416		syslog(LOG_ERR, "cannot open allowed users file %s (%s)",
417		    PATH_ALLOWFILE, strerror(errno));
418		return (0);
419	} else {
420		/*
421		 * /etc/authpf/authpf.allow exists, thus we do a linear
422		 * search to see if they are allowed.
423		 * also, if username "*" exists, then this is a
424		 * "public" gateway, such as it is, so let
425		 * everyone use it.
426		 */
427		lbuf = NULL;
428		while ((buf = fgetln(f, &len))) {
429			if (buf[len - 1] == '\n')
430				buf[len - 1] = '\0';
431			else {
432				if ((lbuf = (char *)malloc(len + 1)) == NULL)
433					err(1, NULL);
434				memcpy(lbuf, buf, len);
435				lbuf[len] = '\0';
436				buf = lbuf;
437			}
438
439			matched = strcmp(luser, buf) == 0 || strcmp("*", buf) == 0;
440
441			if (lbuf != NULL) {
442				free(lbuf);
443				lbuf = NULL;
444			}
445
446			if (matched)
447				return (1); /* matched an allowed username */
448		}
449		syslog(LOG_INFO, "denied access to %s: not listed in %s",
450		    luser, PATH_ALLOWFILE);
451
452		/* reuse buf */
453		buf = "\n\nSorry, you are not allowed to use this facility!\n";
454		fputs(buf, stdout);
455	}
456	fflush(stdout);
457	return (0);
458}
459
460/*
461 * check_luser checks to see if user "luser" has been banned
462 * from using us by virtue of having an file of the same name
463 * in the "luserdir" directory.
464 *
465 * If the user has been banned, we copy the contents of the file
466 * to the user's screen. (useful for telling the user what to
467 * do to get un-banned, or just to tell them they aren't
468 * going to be un-banned.)
469 */
470static int
471check_luser(char *luserdir, char *luser)
472{
473	FILE	*f;
474	int	 n;
475	char	 tmp[MAXPATHLEN];
476
477	n = snprintf(tmp, sizeof(tmp), "%s/%s", luserdir, luser);
478	if (n < 0 || (u_int)n >= sizeof(tmp)) {
479		syslog(LOG_ERR, "provided banned directory line too long (%s)",
480		    luserdir);
481		return (0);
482	}
483	if ((f = fopen(tmp, "r")) == NULL) {
484		if (errno == ENOENT) {
485			/*
486			 * file or dir doesn't exist, so therefore
487			 * this luser isn't banned..  all is well
488			 */
489			return (1);
490		} else {
491			/*
492			 * luser may in fact be banned, but we can't open the
493			 * file even though it's there. probably a config
494			 * problem.
495			 */
496			syslog(LOG_ERR, "cannot open banned file %s (%s)",
497			    tmp, strerror(errno));
498			return (0);
499		}
500	} else {
501		/*
502		 * luser is banned - spit the file at them to
503		 * tell what they can do and where they can go.
504		 */
505		syslog(LOG_INFO, "denied access to %s: %s exists",
506		    luser, tmp);
507
508		/* reuse tmp */
509		strlcpy(tmp, "\n\n-**- Sorry, you have been banned! -**-\n\n",
510		    sizeof(tmp));
511		while (fputs(tmp, stdout) != EOF && !feof(f)) {
512			if (fgets(tmp, sizeof(tmp), f) == NULL) {
513				fflush(stdout);
514				return (0);
515			}
516		}
517	}
518	fflush(stdout);
519	return (0);
520}
521
522/*
523 * Search for rulesets left by other authpf processes (either because they
524 * died ungracefully or were terminated) and remove them.
525 */
526static int
527remove_stale_rulesets(void)
528{
529	struct pfioc_ruleset	 prs;
530	const int		 action[PF_RULESET_MAX] = { PF_SCRUB,
531				    PF_PASS, PF_NAT, PF_BINAT, PF_RDR };
532	u_int32_t		 nr, mnr;
533
534	memset(&prs, 0, sizeof(prs));
535	strlcpy(prs.anchor, anchorname, sizeof(prs.anchor));
536	if (ioctl(dev, DIOCGETRULESETS, &prs)) {
537		if (errno == EINVAL)
538			return (0);
539		else
540			return (1);
541	}
542
543	mnr = prs.nr;
544	nr = 0;
545	while (nr < mnr) {
546		char	*s;
547		pid_t	 pid;
548
549		prs.nr = nr;
550		if (ioctl(dev, DIOCGETRULESET, &prs))
551			return (1);
552		errno = 0;
553		pid = strtoul(prs.name, &s, 10);
554		if (!prs.name[0] || errno || *s)
555			return (1);
556		if (kill(pid, 0) && errno != EPERM) {
557			int i;
558
559			for (i = 0; i < PF_RULESET_MAX; ++i) {
560				struct pfioc_rule pr;
561
562				memset(&pr, 0, sizeof(pr));
563				memcpy(pr.anchor, prs.anchor, sizeof(pr.anchor));
564				memcpy(pr.ruleset, prs.name, sizeof(pr.ruleset));
565				pr.rule.action = action[i];
566				if ((ioctl(dev, DIOCBEGINRULES, &pr) ||
567				    ioctl(dev, DIOCCOMMITRULES, &pr)) &&
568				    errno != EINVAL)
569					return (1);
570			}
571			mnr--;
572		} else
573			nr++;
574	}
575	return (0);
576}
577
578/*
579 * Add/remove filter entries for user "luser" from ip "ipsrc"
580 */
581static int
582change_filter(int add, const char *luser, const char *ipsrc)
583{
584	char			 fn[MAXPATHLEN];
585	FILE			*f = NULL;
586	const int		 action[PF_RULESET_MAX] = { PF_SCRUB,
587				    PF_PASS, PF_NAT, PF_BINAT, PF_RDR };
588	struct pfctl		 pf;
589	struct pfioc_rule	 pr[PF_RULESET_MAX];
590	int			 i;
591
592	if (luser == NULL || !luser[0] || strlen(luser) >=
593	    PF_RULESET_NAME_SIZE || ipsrc == NULL || !ipsrc[0]) {
594		syslog(LOG_ERR, "invalid luser/ipsrc");
595		goto error;
596	}
597
598	if (add) {
599		if ((i = snprintf(fn, sizeof(fn), "%s/%s/authpf.rules",
600		    PATH_USER_DIR, luser)) < 0 || i >= sizeof(fn)) {
601			syslog(LOG_ERR, "user rule path too long");
602			goto error;
603		}
604		if ((f = fopen(fn, "r")) == NULL && errno != ENOENT) {
605			syslog(LOG_ERR, "cannot open %s (%m)", fn);
606			goto error;
607		}
608		if (f == NULL) {
609			if (strlcpy(fn, PATH_PFRULES, sizeof(fn)) >=
610			    sizeof(fn)) {
611				syslog(LOG_ERR, "rule path too long");
612				goto error;
613			}
614			if ((f = fopen(fn, "r")) == NULL) {
615				syslog(LOG_ERR, "cannot open %s (%m)", fn);
616				goto error;
617			}
618		}
619	}
620
621	if (pfctl_load_fingerprints(dev, 0)) {
622		syslog(LOG_ERR, "unable to load kernel's OS fingerprints");
623		goto error;
624	}
625
626	memset(&pf, 0, sizeof(pf));
627	for (i = 0; i < PF_RULESET_MAX; ++i) {
628		memset(&pr[i], 0, sizeof(pr[i]));
629		pr[i].rule.action = action[i];
630		strlcpy(pr[i].anchor, anchorname, sizeof(pr[i].anchor));
631		strlcpy(pr[i].ruleset, rulesetname, sizeof(pr[i].ruleset));
632		if (ioctl(dev, DIOCBEGINRULES, &pr[i])) {
633			syslog(LOG_ERR, "DIOCBEGINRULES %m");
634			goto error;
635		}
636		pf.prule[i] = &pr[i];
637	}
638
639	if (add) {
640		if (symset("user_ip", ipsrc, 0) ||
641		    symset("user_id", luser, 0)) {
642			syslog(LOG_ERR, "symset");
643			goto error;
644		}
645
646		pf.dev = dev;
647		infile = fn;
648		if (parse_rules(f, &pf) < 0) {
649			syslog(LOG_ERR, "syntax error in rule file: "
650			    "authpf rules not loaded");
651			goto error;
652		}
653
654		infile = NULL;
655		fclose(f);
656		f = NULL;
657	}
658
659	for (i = 0; i < PF_RULESET_MAX; ++i)
660		/*
661		 * ignore EINVAL on removal, it means the anchor was
662		 * already automatically removed by the kernel.
663		 */
664		if (ioctl(dev, DIOCCOMMITRULES, &pr[i]) &&
665		    (add || errno != EINVAL)) {
666			syslog(LOG_ERR, "DIOCCOMMITRULES %m");
667			goto error;
668		}
669
670	if (add) {
671		gettimeofday(&Tstart, NULL);
672		syslog(LOG_INFO, "allowing %s, user %s", ipsrc, luser);
673	} else {
674		gettimeofday(&Tend, NULL);
675		syslog(LOG_INFO, "removed %s, user %s - duration %ld seconds",
676		    ipsrc, luser, Tend.tv_sec - Tstart.tv_sec);
677	}
678	return (0);
679
680error:
681	if (f != NULL)
682		fclose(f);
683
684	infile = NULL;
685	return (-1);
686}
687
688/*
689 * This is to kill off states that would otherwise be left behind stateful
690 * rules. This means we don't need to allow in more traffic than we really
691 * want to, since we don't have to worry about any luser sessions lasting
692 * longer than their ssh session. This function is based on
693 * pfctl_kill_states from pfctl.
694 */
695static void
696authpf_kill_states(void)
697{
698	struct pfioc_state_kill	psk;
699	struct in_addr		target;
700
701	memset(&psk, 0, sizeof(psk));
702	psk.psk_af = AF_INET;
703
704	inet_pton(AF_INET, ipsrc, &target);
705
706	/* Kill all states from ipsrc */
707	psk.psk_src.addr.v.a.addr.v4 = target;
708	memset(&psk.psk_src.addr.v.a.mask, 0xff,
709	    sizeof(psk.psk_src.addr.v.a.mask));
710	if (ioctl(dev, DIOCKILLSTATES, &psk))
711		syslog(LOG_ERR, "DIOCKILLSTATES failed (%m)");
712
713	/* Kill all states to ipsrc */
714	psk.psk_af = AF_INET;
715	memset(&psk.psk_src, 0, sizeof(psk.psk_src));
716	psk.psk_dst.addr.v.a.addr.v4 = target;
717	memset(&psk.psk_dst.addr.v.a.mask, 0xff,
718	    sizeof(psk.psk_dst.addr.v.a.mask));
719	if (ioctl(dev, DIOCKILLSTATES, &psk))
720		syslog(LOG_ERR, "DIOCKILLSTATES failed (%m)");
721}
722
723/* signal handler that makes us go away properly */
724static void
725need_death(int signo)
726{
727	want_death = 1;
728}
729
730/*
731 * function that removes our stuff when we go away.
732 */
733#ifdef __FreeBSD__
734static __dead2 void
735#else
736static __dead void
737#endif
738do_death(int active)
739{
740	int	ret = 0;
741
742	if (active) {
743		change_filter(0, luser, ipsrc);
744		authpf_kill_states();
745		remove_stale_rulesets();
746	}
747	if (pidfp)
748		ftruncate(fileno(pidfp), 0);
749	if (pidfile[0])
750		if (unlink(pidfile) == -1)
751			syslog(LOG_ERR, "cannot unlink %s (%m)", pidfile);
752	exit(ret);
753}
754
755/*
756 * callbacks for parse_rules(void)
757 */
758
759int
760pfctl_add_rule(struct pfctl *pf, struct pf_rule *r)
761{
762	struct pfioc_rule	*pr;
763
764	switch (r->action) {
765	case PF_PASS:
766	case PF_DROP:
767		pr = pf->prule[PF_RULESET_FILTER];
768		break;
769	case PF_SCRUB:
770		pr = pf->prule[PF_RULESET_SCRUB];
771		break;
772	case PF_NAT:
773	case PF_NONAT:
774		pr = pf->prule[PF_RULESET_NAT];
775		break;
776	case PF_RDR:
777	case PF_NORDR:
778		pr = pf->prule[PF_RULESET_RDR];
779		break;
780	case PF_BINAT:
781	case PF_NOBINAT:
782		pr = pf->prule[PF_RULESET_BINAT];
783		break;
784	default:
785		syslog(LOG_ERR, "invalid rule action %d", r->action);
786		return (1);
787	}
788	if (pfctl_add_pool(pf, &r->rpool, r->af))
789		return (1);
790	pr->pool_ticket = pf->paddr.ticket;
791	memcpy(&pr->rule, r, sizeof(pr->rule));
792	if (ioctl(pf->dev, DIOCADDRULE, pr)) {
793		syslog(LOG_ERR, "DIOCADDRULE %m");
794		return (1);
795	}
796	pfctl_clear_pool(&r->rpool);
797	return (0);
798}
799
800int
801pfctl_add_pool(struct pfctl *pf, struct pf_pool *p, sa_family_t af)
802{
803	struct pf_pooladdr	*pa;
804
805	if (ioctl(pf->dev, DIOCBEGINADDRS, &pf->paddr)) {
806		syslog(LOG_ERR, "DIOCBEGINADDRS %m");
807		return (1);
808	}
809	pf->paddr.af = af;
810	TAILQ_FOREACH(pa, &p->list, entries) {
811		memcpy(&pf->paddr.addr, pa, sizeof(struct pf_pooladdr));
812		if (ioctl(pf->dev, DIOCADDADDR, &pf->paddr)) {
813			syslog(LOG_ERR, "DIOCADDADDR %m");
814			return (1);
815		}
816	}
817	return (0);
818}
819
820void
821pfctl_clear_pool(struct pf_pool *pool)
822{
823	struct pf_pooladdr	*pa;
824
825	while ((pa = TAILQ_FIRST(&pool->list)) != NULL) {
826		TAILQ_REMOVE(&pool->list, pa, entries);
827		free(pa);
828	}
829}
830
831int
832pfctl_add_altq(struct pfctl *pf, struct pf_altq *a)
833{
834	fprintf(stderr, "altq rules not supported in authpf\n");
835	return (1);
836}
837
838int
839pfctl_set_optimization(struct pfctl *pf, const char *opt)
840{
841	fprintf(stderr, "set optimization not supported in authpf\n");
842	return (1);
843}
844
845int
846pfctl_set_logif(struct pfctl *pf, char *ifname)
847{
848	fprintf(stderr, "set loginterface not supported in authpf\n");
849	return (1);
850}
851
852int
853pfctl_set_timeout(struct pfctl *pf, const char *opt, int seconds, int quiet)
854{
855	fprintf(stderr, "set timeout not supported in authpf\n");
856	return (1);
857}
858
859int
860pfctl_set_limit(struct pfctl *pf, const char *opt, unsigned int limit)
861{
862	fprintf(stderr, "set limit not supported in authpf\n");
863	return (1);
864}
865
866int
867pfctl_define_table(char *name, int flags, int addrs, const char *anchor,
868    const char *ruleset, struct pfr_buffer *ab, u_int32_t ticket)
869{
870	fprintf(stderr, "table definitions not yet supported in authpf\n");
871	return (1);
872}
873
874int
875pfctl_rules(int dev, char *filename, int opts, char *anchorname,
876    char *rulesetname)
877{
878	/* never called, no anchors inside anchors, but we need the stub */
879	fprintf(stderr, "load anchor not supported from authpf\n");
880	return (1);
881}
882
883