pf.c revision 171929
1171168Smlaier/*	$OpenBSD: pf.c,v 1.527 2007/02/22 15:23:23 pyr Exp $ */
2126258Smlaier
3126258Smlaier/*
4126258Smlaier * Copyright (c) 2001 Daniel Hartmeier
5130613Smlaier * Copyright (c) 2002,2003 Henning Brauer
6126258Smlaier * All rights reserved.
7126258Smlaier *
8126258Smlaier * Redistribution and use in source and binary forms, with or without
9126258Smlaier * modification, are permitted provided that the following conditions
10126258Smlaier * are met:
11126258Smlaier *
12126258Smlaier *    - Redistributions of source code must retain the above copyright
13126258Smlaier *      notice, this list of conditions and the following disclaimer.
14126258Smlaier *    - Redistributions in binary form must reproduce the above
15126258Smlaier *      copyright notice, this list of conditions and the following
16126258Smlaier *      disclaimer in the documentation and/or other materials provided
17126258Smlaier *      with the distribution.
18126258Smlaier *
19126258Smlaier * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20126258Smlaier * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21126258Smlaier * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
22126258Smlaier * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
23126258Smlaier * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
24126258Smlaier * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
25126258Smlaier * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
26126258Smlaier * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
27126258Smlaier * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28126258Smlaier * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
29126258Smlaier * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30126258Smlaier * POSSIBILITY OF SUCH DAMAGE.
31126258Smlaier *
32126258Smlaier * Effort sponsored in part by the Defense Advanced Research Projects
33126258Smlaier * Agency (DARPA) and Air Force Research Laboratory, Air Force
34126258Smlaier * Materiel Command, USAF, under agreement number F30602-01-2-0537.
35126258Smlaier *
36126258Smlaier */
37126258Smlaier
38127145Smlaier#ifdef __FreeBSD__
39126261Smlaier#include "opt_inet.h"
40126261Smlaier#include "opt_inet6.h"
41171168Smlaier
42171168Smlaier#include <sys/cdefs.h>
43171168Smlaier__FBSDID("$FreeBSD: head/sys/contrib/pf/net/pf.c 171929 2007-08-23 09:30:58Z dhartmei $");
44126261Smlaier#endif
45126261Smlaier
46127145Smlaier#ifdef __FreeBSD__
47162238Scsjp#include "opt_mac.h"
48126261Smlaier#include "opt_bpf.h"
49126261Smlaier#include "opt_pf.h"
50153110Sru
51153110Sru#ifdef DEV_BPF
52127145Smlaier#define	NBPFILTER	DEV_BPF
53153110Sru#else
54153110Sru#define	NBPFILTER	0
55153110Sru#endif
56153110Sru
57153110Sru#ifdef DEV_PFLOG
58127145Smlaier#define	NPFLOG		DEV_PFLOG
59153110Sru#else
60153110Sru#define	NPFLOG		0
61153110Sru#endif
62153110Sru
63153110Sru#ifdef DEV_PFSYNC
64127145Smlaier#define	NPFSYNC		DEV_PFSYNC
65126261Smlaier#else
66153110Sru#define	NPFSYNC		0
67153110Sru#endif
68153110Sru
69153110Sru#else
70126258Smlaier#include "bpfilter.h"
71126258Smlaier#include "pflog.h"
72126258Smlaier#include "pfsync.h"
73126261Smlaier#endif
74126258Smlaier
75126258Smlaier#include <sys/param.h>
76126258Smlaier#include <sys/systm.h>
77126258Smlaier#include <sys/mbuf.h>
78126258Smlaier#include <sys/filio.h>
79126258Smlaier#include <sys/socket.h>
80126258Smlaier#include <sys/socketvar.h>
81126258Smlaier#include <sys/kernel.h>
82126258Smlaier#include <sys/time.h>
83127145Smlaier#ifdef __FreeBSD__
84126261Smlaier#include <sys/sysctl.h>
85130613Smlaier#include <sys/endian.h>
86126261Smlaier#else
87126258Smlaier#include <sys/pool.h>
88126261Smlaier#endif
89171168Smlaier#include <sys/proc.h>
90171168Smlaier#ifdef __FreeBSD__
91171168Smlaier#include <sys/kthread.h>
92171168Smlaier#include <sys/lock.h>
93171168Smlaier#include <sys/sx.h>
94171168Smlaier#else
95171168Smlaier#include <sys/rwlock.h>
96171168Smlaier#endif
97126258Smlaier
98126258Smlaier#include <net/if.h>
99126258Smlaier#include <net/if_types.h>
100126258Smlaier#include <net/bpf.h>
101126258Smlaier#include <net/route.h>
102171168Smlaier#ifndef __FreeBSD__
103171168Smlaier#include <net/radix_mpath.h>
104171168Smlaier#endif
105126258Smlaier
106126258Smlaier#include <netinet/in.h>
107126258Smlaier#include <netinet/in_var.h>
108126258Smlaier#include <netinet/in_systm.h>
109126258Smlaier#include <netinet/ip.h>
110126258Smlaier#include <netinet/ip_var.h>
111126258Smlaier#include <netinet/tcp.h>
112126258Smlaier#include <netinet/tcp_seq.h>
113126258Smlaier#include <netinet/udp.h>
114126258Smlaier#include <netinet/ip_icmp.h>
115126258Smlaier#include <netinet/in_pcb.h>
116126258Smlaier#include <netinet/tcp_timer.h>
117126258Smlaier#include <netinet/tcp_var.h>
118126258Smlaier#include <netinet/udp_var.h>
119126258Smlaier#include <netinet/icmp_var.h>
120145836Smlaier#include <netinet/if_ether.h>
121126258Smlaier
122127145Smlaier#ifndef __FreeBSD__
123126258Smlaier#include <dev/rndvar.h>
124126261Smlaier#endif
125126258Smlaier#include <net/pfvar.h>
126126258Smlaier#include <net/if_pflog.h>
127130613Smlaier
128130613Smlaier#if NPFSYNC > 0
129126258Smlaier#include <net/if_pfsync.h>
130130613Smlaier#endif /* NPFSYNC > 0 */
131126258Smlaier
132126258Smlaier#ifdef INET6
133126258Smlaier#include <netinet/ip6.h>
134126258Smlaier#include <netinet/in_pcb.h>
135126258Smlaier#include <netinet/icmp6.h>
136126258Smlaier#include <netinet6/nd6.h>
137127145Smlaier#ifdef __FreeBSD__
138126261Smlaier#include <netinet6/ip6_var.h>
139126261Smlaier#include <netinet6/in6_pcb.h>
140126261Smlaier#endif
141126258Smlaier#endif /* INET6 */
142126258Smlaier
143127145Smlaier#ifdef __FreeBSD__
144126261Smlaier#include <machine/in_cksum.h>
145126261Smlaier#include <sys/limits.h>
146126261Smlaier#include <sys/ucred.h>
147163606Srwatson#include <security/mac/mac_framework.h>
148126258Smlaier
149126261Smlaierextern int ip_optcopy(struct ip *, struct ip *);
150171168Smlaierextern int debug_pfugidhack;
151126261Smlaier#endif
152126261Smlaier
153126258Smlaier#define DPFPRINTF(n, x)	if (pf_status.debug >= (n)) printf x
154126258Smlaier
155126258Smlaier/*
156126258Smlaier * Global variables
157126258Smlaier */
158126258Smlaier
159126258Smlaierstruct pf_altqqueue	 pf_altqs[2];
160126258Smlaierstruct pf_palist	 pf_pabuf;
161126258Smlaierstruct pf_altqqueue	*pf_altqs_active;
162126258Smlaierstruct pf_altqqueue	*pf_altqs_inactive;
163126258Smlaierstruct pf_status	 pf_status;
164126258Smlaier
165126258Smlaieru_int32_t		 ticket_altqs_active;
166126258Smlaieru_int32_t		 ticket_altqs_inactive;
167130613Smlaierint			 altqs_inactive_open;
168126258Smlaieru_int32_t		 ticket_pabuf;
169126258Smlaier
170145836Smlaierstruct pf_anchor_stackframe {
171145836Smlaier	struct pf_ruleset			*rs;
172145836Smlaier	struct pf_rule				*r;
173145836Smlaier	struct pf_anchor_node			*parent;
174145836Smlaier	struct pf_anchor			*child;
175145836Smlaier} pf_anchor_stack[64];
176126261Smlaier
177127145Smlaier#ifdef __FreeBSD__
178130613Smlaieruma_zone_t		 pf_src_tree_pl, pf_rule_pl;
179126261Smlaieruma_zone_t		 pf_state_pl, pf_altq_pl, pf_pooladdr_pl;
180126261Smlaier#else
181130613Smlaierstruct pool		 pf_src_tree_pl, pf_rule_pl;
182126258Smlaierstruct pool		 pf_state_pl, pf_altq_pl, pf_pooladdr_pl;
183126261Smlaier#endif
184126258Smlaier
185126258Smlaiervoid			 pf_print_host(struct pf_addr *, u_int16_t, u_int8_t);
186126258Smlaier
187145836Smlaiervoid			 pf_init_threshold(struct pf_threshold *, u_int32_t,
188145836Smlaier			    u_int32_t);
189145836Smlaiervoid			 pf_add_threshold(struct pf_threshold *);
190145836Smlaierint			 pf_check_threshold(struct pf_threshold *);
191145836Smlaier
192126258Smlaiervoid			 pf_change_ap(struct pf_addr *, u_int16_t *,
193126258Smlaier			    u_int16_t *, u_int16_t *, struct pf_addr *,
194126258Smlaier			    u_int16_t, u_int8_t, sa_family_t);
195171168Smlaierint			 pf_modulate_sack(struct mbuf *, int, struct pf_pdesc *,
196171168Smlaier			    struct tcphdr *, struct pf_state_peer *);
197126258Smlaier#ifdef INET6
198126258Smlaiervoid			 pf_change_a6(struct pf_addr *, u_int16_t *,
199126258Smlaier			    struct pf_addr *, u_int8_t);
200126258Smlaier#endif /* INET6 */
201126258Smlaiervoid			 pf_change_icmp(struct pf_addr *, u_int16_t *,
202126258Smlaier			    struct pf_addr *, struct pf_addr *, u_int16_t,
203126258Smlaier			    u_int16_t *, u_int16_t *, u_int16_t *,
204126258Smlaier			    u_int16_t *, u_int8_t, sa_family_t);
205162238Scsjp#ifdef __FreeBSD__
206162238Scsjpvoid			 pf_send_tcp(struct mbuf *,
207162238Scsjp			    const struct pf_rule *, sa_family_t,
208162238Scsjp#else
209126258Smlaiervoid			 pf_send_tcp(const struct pf_rule *, sa_family_t,
210162238Scsjp#endif
211126258Smlaier			    const struct pf_addr *, const struct pf_addr *,
212126258Smlaier			    u_int16_t, u_int16_t, u_int32_t, u_int32_t,
213145836Smlaier			    u_int8_t, u_int16_t, u_int16_t, u_int8_t, int,
214171168Smlaier			    u_int16_t, struct ether_header *, struct ifnet *);
215126258Smlaiervoid			 pf_send_icmp(struct mbuf *, u_int8_t, u_int8_t,
216126258Smlaier			    sa_family_t, struct pf_rule *);
217126258Smlaierstruct pf_rule		*pf_match_translation(struct pf_pdesc *, struct mbuf *,
218130613Smlaier			    int, int, struct pfi_kif *,
219126258Smlaier			    struct pf_addr *, u_int16_t, struct pf_addr *,
220126258Smlaier			    u_int16_t, int);
221126258Smlaierstruct pf_rule		*pf_get_translation(struct pf_pdesc *, struct mbuf *,
222130613Smlaier			    int, int, struct pfi_kif *, struct pf_src_node **,
223126258Smlaier			    struct pf_addr *, u_int16_t,
224126258Smlaier			    struct pf_addr *, u_int16_t,
225126258Smlaier			    struct pf_addr *, u_int16_t *);
226126258Smlaierint			 pf_test_tcp(struct pf_rule **, struct pf_state **,
227130613Smlaier			    int, struct pfi_kif *, struct mbuf *, int,
228126258Smlaier			    void *, struct pf_pdesc *, struct pf_rule **,
229135920Smlaier#ifdef __FreeBSD__
230145836Smlaier			    struct pf_ruleset **, struct ifqueue *,
231145836Smlaier			    struct inpcb *);
232135920Smlaier#else
233145836Smlaier			    struct pf_ruleset **, struct ifqueue *);
234135920Smlaier#endif
235126258Smlaierint			 pf_test_udp(struct pf_rule **, struct pf_state **,
236130613Smlaier			    int, struct pfi_kif *, struct mbuf *, int,
237126258Smlaier			    void *, struct pf_pdesc *, struct pf_rule **,
238135920Smlaier#ifdef __FreeBSD__
239145836Smlaier			    struct pf_ruleset **, struct ifqueue *,
240145836Smlaier			    struct inpcb *);
241135920Smlaier#else
242145836Smlaier			    struct pf_ruleset **, struct ifqueue *);
243135920Smlaier#endif
244126258Smlaierint			 pf_test_icmp(struct pf_rule **, struct pf_state **,
245130613Smlaier			    int, struct pfi_kif *, struct mbuf *, int,
246126258Smlaier			    void *, struct pf_pdesc *, struct pf_rule **,
247145836Smlaier			    struct pf_ruleset **, struct ifqueue *);
248126258Smlaierint			 pf_test_other(struct pf_rule **, struct pf_state **,
249130613Smlaier			    int, struct pfi_kif *, struct mbuf *, int, void *,
250126258Smlaier			    struct pf_pdesc *, struct pf_rule **,
251145836Smlaier			    struct pf_ruleset **, struct ifqueue *);
252126258Smlaierint			 pf_test_fragment(struct pf_rule **, int,
253130613Smlaier			    struct pfi_kif *, struct mbuf *, void *,
254126258Smlaier			    struct pf_pdesc *, struct pf_rule **,
255126258Smlaier			    struct pf_ruleset **);
256126258Smlaierint			 pf_test_state_tcp(struct pf_state **, int,
257130613Smlaier			    struct pfi_kif *, struct mbuf *, int,
258126258Smlaier			    void *, struct pf_pdesc *, u_short *);
259126258Smlaierint			 pf_test_state_udp(struct pf_state **, int,
260130613Smlaier			    struct pfi_kif *, struct mbuf *, int,
261126258Smlaier			    void *, struct pf_pdesc *);
262126258Smlaierint			 pf_test_state_icmp(struct pf_state **, int,
263130613Smlaier			    struct pfi_kif *, struct mbuf *, int,
264145836Smlaier			    void *, struct pf_pdesc *, u_short *);
265126258Smlaierint			 pf_test_state_other(struct pf_state **, int,
266130613Smlaier			    struct pfi_kif *, struct pf_pdesc *);
267126258Smlaierint			 pf_match_tag(struct mbuf *, struct pf_rule *,
268171168Smlaier			     struct pf_mtag *, int *);
269171168Smlaierint			 pf_step_out_of_anchor(int *, struct pf_ruleset **,
270171168Smlaier			     int, struct pf_rule **, struct pf_rule **,
271171168Smlaier			     int *);
272126258Smlaiervoid			 pf_hash(struct pf_addr *, struct pf_addr *,
273126258Smlaier			    struct pf_poolhashkey *, sa_family_t);
274130613Smlaierint			 pf_map_addr(u_int8_t, struct pf_rule *,
275126258Smlaier			    struct pf_addr *, struct pf_addr *,
276130613Smlaier			    struct pf_addr *, struct pf_src_node **);
277130613Smlaierint			 pf_get_sport(sa_family_t, u_int8_t, struct pf_rule *,
278126258Smlaier			    struct pf_addr *, struct pf_addr *, u_int16_t,
279130613Smlaier			    struct pf_addr *, u_int16_t*, u_int16_t, u_int16_t,
280130613Smlaier			    struct pf_src_node **);
281126258Smlaiervoid			 pf_route(struct mbuf **, struct pf_rule *, int,
282171168Smlaier			    struct ifnet *, struct pf_state *,
283171168Smlaier			    struct pf_pdesc *);
284126258Smlaiervoid			 pf_route6(struct mbuf **, struct pf_rule *, int,
285171168Smlaier			    struct ifnet *, struct pf_state *,
286171168Smlaier			    struct pf_pdesc *);
287135920Smlaier#ifdef __FreeBSD__
288171168Smlaier/* XXX: import */
289135920Smlaier#else
290171168Smlaierint			 pf_socket_lookup(int, struct pf_pdesc *);
291135920Smlaier#endif
292126258Smlaieru_int8_t		 pf_get_wscale(struct mbuf *, int, u_int16_t,
293126258Smlaier			    sa_family_t);
294126258Smlaieru_int16_t		 pf_get_mss(struct mbuf *, int, u_int16_t,
295126258Smlaier			    sa_family_t);
296126258Smlaieru_int16_t		 pf_calc_mss(struct pf_addr *, sa_family_t,
297126258Smlaier				u_int16_t);
298126258Smlaiervoid			 pf_set_rt_ifp(struct pf_state *,
299126258Smlaier			    struct pf_addr *);
300126258Smlaierint			 pf_check_proto_cksum(struct mbuf *, int, int,
301126258Smlaier			    u_int8_t, sa_family_t);
302126258Smlaierint			 pf_addr_wrap_neq(struct pf_addr_wrap *,
303126258Smlaier			    struct pf_addr_wrap *);
304130613Smlaierstruct pf_state		*pf_find_state_recurse(struct pfi_kif *,
305171168Smlaier			    struct pf_state_cmp *, u_int8_t);
306145836Smlaierint			 pf_src_connlimit(struct pf_state **);
307145836Smlaierint			 pf_check_congestion(struct ifqueue *);
308126258Smlaier
309127145Smlaier#ifdef __FreeBSD__
310126261Smlaierint in4_cksum(struct mbuf *m, u_int8_t nxt, int off, int len);
311126258Smlaier
312171168Smlaierextern int pf_end_threads;
313171168Smlaier
314126261Smlaierstruct pf_pool_limit pf_pool_limits[PF_LIMIT_MAX];
315171168Smlaier#else
316171168Smlaierextern struct pool pfr_ktable_pl;
317171168Smlaierextern struct pool pfr_kentry_pl;
318145836Smlaier
319130613Smlaierstruct pf_pool_limit pf_pool_limits[PF_LIMIT_MAX] = {
320130613Smlaier	{ &pf_state_pl, PFSTATE_HIWAT },
321130613Smlaier	{ &pf_src_tree_pl, PFSNODE_HIWAT },
322171168Smlaier	{ &pf_frent_pl, PFFRAG_FRENT_HIWAT },
323171168Smlaier	{ &pfr_ktable_pl, PFR_KTABLE_HIWAT },
324171168Smlaier	{ &pfr_kentry_pl, PFR_KENTRY_HIWAT }
325130613Smlaier};
326126261Smlaier#endif
327126258Smlaier
328126258Smlaier#define STATE_LOOKUP()							\
329126258Smlaier	do {								\
330126258Smlaier		if (direction == PF_IN)					\
331145836Smlaier			*state = pf_find_state_recurse(			\
332130613Smlaier			    kif, &key, PF_EXT_GWY);			\
333126258Smlaier		else							\
334145836Smlaier			*state = pf_find_state_recurse(			\
335130613Smlaier			    kif, &key, PF_LAN_EXT);			\
336145836Smlaier		if (*state == NULL || (*state)->timeout == PFTM_PURGE)	\
337126258Smlaier			return (PF_DROP);				\
338126258Smlaier		if (direction == PF_OUT &&				\
339126258Smlaier		    (((*state)->rule.ptr->rt == PF_ROUTETO &&		\
340126258Smlaier		    (*state)->rule.ptr->direction == PF_OUT) ||		\
341126258Smlaier		    ((*state)->rule.ptr->rt == PF_REPLYTO &&		\
342126258Smlaier		    (*state)->rule.ptr->direction == PF_IN)) &&		\
343130613Smlaier		    (*state)->rt_kif != NULL &&				\
344130613Smlaier		    (*state)->rt_kif != kif)				\
345126258Smlaier			return (PF_PASS);				\
346126258Smlaier	} while (0)
347126258Smlaier
348126258Smlaier#define	STATE_TRANSLATE(s) \
349126258Smlaier	(s)->lan.addr.addr32[0] != (s)->gwy.addr.addr32[0] || \
350126258Smlaier	((s)->af == AF_INET6 && \
351126258Smlaier	((s)->lan.addr.addr32[1] != (s)->gwy.addr.addr32[1] || \
352126258Smlaier	(s)->lan.addr.addr32[2] != (s)->gwy.addr.addr32[2] || \
353126258Smlaier	(s)->lan.addr.addr32[3] != (s)->gwy.addr.addr32[3])) || \
354126258Smlaier	(s)->lan.port != (s)->gwy.port
355126258Smlaier
356171168Smlaier#define BOUND_IFACE(r, k) \
357171168Smlaier	((r)->rule_flag & PFRULE_IFBOUND) ? (k) : pfi_all
358126258Smlaier
359145836Smlaier#define STATE_INC_COUNTERS(s)				\
360145836Smlaier	do {						\
361145836Smlaier		s->rule.ptr->states++;			\
362145836Smlaier		if (s->anchor.ptr != NULL)		\
363145836Smlaier			s->anchor.ptr->states++;	\
364145836Smlaier		if (s->nat_rule.ptr != NULL)		\
365145836Smlaier			s->nat_rule.ptr->states++;	\
366145836Smlaier	} while (0)
367145836Smlaier
368145836Smlaier#define STATE_DEC_COUNTERS(s)				\
369145836Smlaier	do {						\
370145836Smlaier		if (s->nat_rule.ptr != NULL)		\
371145836Smlaier			s->nat_rule.ptr->states--;	\
372145836Smlaier		if (s->anchor.ptr != NULL)		\
373145836Smlaier			s->anchor.ptr->states--;	\
374145836Smlaier		s->rule.ptr->states--;			\
375145836Smlaier	} while (0)
376145836Smlaier
377130613Smlaierstruct pf_src_tree tree_src_tracking;
378130613Smlaier
379130613Smlaierstruct pf_state_tree_id tree_id;
380171168Smlaierstruct pf_state_queue state_list;
381130613Smlaier
382171168Smlaier#ifdef __FreeBSD__
383171168Smlaierstatic int pf_src_compare(struct pf_src_node *, struct pf_src_node *);
384171168Smlaierstatic int pf_state_compare_lan_ext(struct pf_state *, struct pf_state *);
385171168Smlaierstatic int pf_state_compare_ext_gwy(struct pf_state *, struct pf_state *);
386171168Smlaierstatic int pf_state_compare_id(struct pf_state *, struct pf_state *);
387171168Smlaier#endif
388171168Smlaier
389130613SmlaierRB_GENERATE(pf_src_tree, pf_src_node, entry, pf_src_compare);
390130613SmlaierRB_GENERATE(pf_state_tree_lan_ext, pf_state,
391130613Smlaier    u.s.entry_lan_ext, pf_state_compare_lan_ext);
392130613SmlaierRB_GENERATE(pf_state_tree_ext_gwy, pf_state,
393130613Smlaier    u.s.entry_ext_gwy, pf_state_compare_ext_gwy);
394130613SmlaierRB_GENERATE(pf_state_tree_id, pf_state,
395130613Smlaier    u.s.entry_id, pf_state_compare_id);
396130613Smlaier
397127145Smlaier#ifdef __FreeBSD__
398126409Smlaierstatic int
399126409Smlaier#else
400126258Smlaierstatic __inline int
401126409Smlaier#endif
402130613Smlaierpf_src_compare(struct pf_src_node *a, struct pf_src_node *b)
403126258Smlaier{
404126258Smlaier	int	diff;
405126258Smlaier
406130613Smlaier	if (a->rule.ptr > b->rule.ptr)
407130613Smlaier		return (1);
408130613Smlaier	if (a->rule.ptr < b->rule.ptr)
409130613Smlaier		return (-1);
410130613Smlaier	if ((diff = a->af - b->af) != 0)
411130613Smlaier		return (diff);
412130613Smlaier	switch (a->af) {
413130613Smlaier#ifdef INET
414130613Smlaier	case AF_INET:
415130613Smlaier		if (a->addr.addr32[0] > b->addr.addr32[0])
416130613Smlaier			return (1);
417130613Smlaier		if (a->addr.addr32[0] < b->addr.addr32[0])
418130613Smlaier			return (-1);
419130613Smlaier		break;
420130613Smlaier#endif /* INET */
421130613Smlaier#ifdef INET6
422130613Smlaier	case AF_INET6:
423130613Smlaier		if (a->addr.addr32[3] > b->addr.addr32[3])
424130613Smlaier			return (1);
425130613Smlaier		if (a->addr.addr32[3] < b->addr.addr32[3])
426130613Smlaier			return (-1);
427130613Smlaier		if (a->addr.addr32[2] > b->addr.addr32[2])
428130613Smlaier			return (1);
429130613Smlaier		if (a->addr.addr32[2] < b->addr.addr32[2])
430130613Smlaier			return (-1);
431130613Smlaier		if (a->addr.addr32[1] > b->addr.addr32[1])
432130613Smlaier			return (1);
433130613Smlaier		if (a->addr.addr32[1] < b->addr.addr32[1])
434130613Smlaier			return (-1);
435130613Smlaier		if (a->addr.addr32[0] > b->addr.addr32[0])
436130613Smlaier			return (1);
437130613Smlaier		if (a->addr.addr32[0] < b->addr.addr32[0])
438130613Smlaier			return (-1);
439130613Smlaier		break;
440130613Smlaier#endif /* INET6 */
441130613Smlaier	}
442130613Smlaier	return (0);
443130613Smlaier}
444130613Smlaier
445130613Smlaier#ifdef __FreeBSD__
446130613Smlaierstatic int
447130613Smlaier#else
448130613Smlaierstatic __inline int
449130613Smlaier#endif
450130613Smlaierpf_state_compare_lan_ext(struct pf_state *a, struct pf_state *b)
451130613Smlaier{
452130613Smlaier	int	diff;
453130613Smlaier
454126258Smlaier	if ((diff = a->proto - b->proto) != 0)
455126258Smlaier		return (diff);
456126258Smlaier	if ((diff = a->af - b->af) != 0)
457126258Smlaier		return (diff);
458126258Smlaier	switch (a->af) {
459126258Smlaier#ifdef INET
460126258Smlaier	case AF_INET:
461130613Smlaier		if (a->lan.addr.addr32[0] > b->lan.addr.addr32[0])
462126258Smlaier			return (1);
463130613Smlaier		if (a->lan.addr.addr32[0] < b->lan.addr.addr32[0])
464126258Smlaier			return (-1);
465130613Smlaier		if (a->ext.addr.addr32[0] > b->ext.addr.addr32[0])
466126258Smlaier			return (1);
467130613Smlaier		if (a->ext.addr.addr32[0] < b->ext.addr.addr32[0])
468126258Smlaier			return (-1);
469126258Smlaier		break;
470126258Smlaier#endif /* INET */
471126258Smlaier#ifdef INET6
472126258Smlaier	case AF_INET6:
473130613Smlaier		if (a->lan.addr.addr32[3] > b->lan.addr.addr32[3])
474126258Smlaier			return (1);
475130613Smlaier		if (a->lan.addr.addr32[3] < b->lan.addr.addr32[3])
476126258Smlaier			return (-1);
477130613Smlaier		if (a->ext.addr.addr32[3] > b->ext.addr.addr32[3])
478126258Smlaier			return (1);
479130613Smlaier		if (a->ext.addr.addr32[3] < b->ext.addr.addr32[3])
480126258Smlaier			return (-1);
481130613Smlaier		if (a->lan.addr.addr32[2] > b->lan.addr.addr32[2])
482126258Smlaier			return (1);
483130613Smlaier		if (a->lan.addr.addr32[2] < b->lan.addr.addr32[2])
484126258Smlaier			return (-1);
485130613Smlaier		if (a->ext.addr.addr32[2] > b->ext.addr.addr32[2])
486126258Smlaier			return (1);
487130613Smlaier		if (a->ext.addr.addr32[2] < b->ext.addr.addr32[2])
488126258Smlaier			return (-1);
489130613Smlaier		if (a->lan.addr.addr32[1] > b->lan.addr.addr32[1])
490126258Smlaier			return (1);
491130613Smlaier		if (a->lan.addr.addr32[1] < b->lan.addr.addr32[1])
492126258Smlaier			return (-1);
493130613Smlaier		if (a->ext.addr.addr32[1] > b->ext.addr.addr32[1])
494126258Smlaier			return (1);
495130613Smlaier		if (a->ext.addr.addr32[1] < b->ext.addr.addr32[1])
496126258Smlaier			return (-1);
497130613Smlaier		if (a->lan.addr.addr32[0] > b->lan.addr.addr32[0])
498126258Smlaier			return (1);
499130613Smlaier		if (a->lan.addr.addr32[0] < b->lan.addr.addr32[0])
500126258Smlaier			return (-1);
501130613Smlaier		if (a->ext.addr.addr32[0] > b->ext.addr.addr32[0])
502126258Smlaier			return (1);
503130613Smlaier		if (a->ext.addr.addr32[0] < b->ext.addr.addr32[0])
504126258Smlaier			return (-1);
505126258Smlaier		break;
506126258Smlaier#endif /* INET6 */
507126258Smlaier	}
508126258Smlaier
509130613Smlaier	if ((diff = a->lan.port - b->lan.port) != 0)
510126258Smlaier		return (diff);
511130613Smlaier	if ((diff = a->ext.port - b->ext.port) != 0)
512126258Smlaier		return (diff);
513126258Smlaier
514126258Smlaier	return (0);
515126258Smlaier}
516126258Smlaier
517130613Smlaier#ifdef __FreeBSD__
518130613Smlaierstatic int
519130613Smlaier#else
520130613Smlaierstatic __inline int
521130613Smlaier#endif
522130613Smlaierpf_state_compare_ext_gwy(struct pf_state *a, struct pf_state *b)
523130613Smlaier{
524130613Smlaier	int	diff;
525130613Smlaier
526130613Smlaier	if ((diff = a->proto - b->proto) != 0)
527130613Smlaier		return (diff);
528130613Smlaier	if ((diff = a->af - b->af) != 0)
529130613Smlaier		return (diff);
530130613Smlaier	switch (a->af) {
531130613Smlaier#ifdef INET
532130613Smlaier	case AF_INET:
533130613Smlaier		if (a->ext.addr.addr32[0] > b->ext.addr.addr32[0])
534130613Smlaier			return (1);
535130613Smlaier		if (a->ext.addr.addr32[0] < b->ext.addr.addr32[0])
536130613Smlaier			return (-1);
537130613Smlaier		if (a->gwy.addr.addr32[0] > b->gwy.addr.addr32[0])
538130613Smlaier			return (1);
539130613Smlaier		if (a->gwy.addr.addr32[0] < b->gwy.addr.addr32[0])
540130613Smlaier			return (-1);
541130613Smlaier		break;
542130613Smlaier#endif /* INET */
543126258Smlaier#ifdef INET6
544130613Smlaier	case AF_INET6:
545130613Smlaier		if (a->ext.addr.addr32[3] > b->ext.addr.addr32[3])
546130613Smlaier			return (1);
547130613Smlaier		if (a->ext.addr.addr32[3] < b->ext.addr.addr32[3])
548130613Smlaier			return (-1);
549130613Smlaier		if (a->gwy.addr.addr32[3] > b->gwy.addr.addr32[3])
550130613Smlaier			return (1);
551130613Smlaier		if (a->gwy.addr.addr32[3] < b->gwy.addr.addr32[3])
552130613Smlaier			return (-1);
553130613Smlaier		if (a->ext.addr.addr32[2] > b->ext.addr.addr32[2])
554130613Smlaier			return (1);
555130613Smlaier		if (a->ext.addr.addr32[2] < b->ext.addr.addr32[2])
556130613Smlaier			return (-1);
557130613Smlaier		if (a->gwy.addr.addr32[2] > b->gwy.addr.addr32[2])
558130613Smlaier			return (1);
559130613Smlaier		if (a->gwy.addr.addr32[2] < b->gwy.addr.addr32[2])
560130613Smlaier			return (-1);
561130613Smlaier		if (a->ext.addr.addr32[1] > b->ext.addr.addr32[1])
562130613Smlaier			return (1);
563130613Smlaier		if (a->ext.addr.addr32[1] < b->ext.addr.addr32[1])
564130613Smlaier			return (-1);
565130613Smlaier		if (a->gwy.addr.addr32[1] > b->gwy.addr.addr32[1])
566130613Smlaier			return (1);
567130613Smlaier		if (a->gwy.addr.addr32[1] < b->gwy.addr.addr32[1])
568130613Smlaier			return (-1);
569130613Smlaier		if (a->ext.addr.addr32[0] > b->ext.addr.addr32[0])
570130613Smlaier			return (1);
571130613Smlaier		if (a->ext.addr.addr32[0] < b->ext.addr.addr32[0])
572130613Smlaier			return (-1);
573130613Smlaier		if (a->gwy.addr.addr32[0] > b->gwy.addr.addr32[0])
574130613Smlaier			return (1);
575130613Smlaier		if (a->gwy.addr.addr32[0] < b->gwy.addr.addr32[0])
576130613Smlaier			return (-1);
577130613Smlaier		break;
578130613Smlaier#endif /* INET6 */
579130613Smlaier	}
580130613Smlaier
581130613Smlaier	if ((diff = a->ext.port - b->ext.port) != 0)
582130613Smlaier		return (diff);
583130613Smlaier	if ((diff = a->gwy.port - b->gwy.port) != 0)
584130613Smlaier		return (diff);
585130613Smlaier
586130613Smlaier	return (0);
587130613Smlaier}
588130613Smlaier
589130613Smlaier#ifdef __FreeBSD__
590130613Smlaierstatic int
591130613Smlaier#else
592130613Smlaierstatic __inline int
593130613Smlaier#endif
594130613Smlaierpf_state_compare_id(struct pf_state *a, struct pf_state *b)
595130613Smlaier{
596130613Smlaier	if (a->id > b->id)
597130613Smlaier		return (1);
598130613Smlaier	if (a->id < b->id)
599130613Smlaier		return (-1);
600130613Smlaier	if (a->creatorid > b->creatorid)
601130613Smlaier		return (1);
602130613Smlaier	if (a->creatorid < b->creatorid)
603130613Smlaier		return (-1);
604130613Smlaier
605130613Smlaier	return (0);
606130613Smlaier}
607130613Smlaier
608130613Smlaier#ifdef INET6
609126258Smlaiervoid
610126258Smlaierpf_addrcpy(struct pf_addr *dst, struct pf_addr *src, sa_family_t af)
611126258Smlaier{
612126258Smlaier	switch (af) {
613126258Smlaier#ifdef INET
614126258Smlaier	case AF_INET:
615126258Smlaier		dst->addr32[0] = src->addr32[0];
616126258Smlaier		break;
617126258Smlaier#endif /* INET */
618126258Smlaier	case AF_INET6:
619126258Smlaier		dst->addr32[0] = src->addr32[0];
620126258Smlaier		dst->addr32[1] = src->addr32[1];
621126258Smlaier		dst->addr32[2] = src->addr32[2];
622126258Smlaier		dst->addr32[3] = src->addr32[3];
623126258Smlaier		break;
624126258Smlaier	}
625126258Smlaier}
626145836Smlaier#endif /* INET6 */
627126258Smlaier
628126258Smlaierstruct pf_state *
629171168Smlaierpf_find_state_byid(struct pf_state_cmp *key)
630126258Smlaier{
631130613Smlaier	pf_status.fcounters[FCNT_STATE_SEARCH]++;
632171168Smlaier	return (RB_FIND(pf_state_tree_id, &tree_id, (struct pf_state *)key));
633130613Smlaier}
634126258Smlaier
635130613Smlaierstruct pf_state *
636171168Smlaierpf_find_state_recurse(struct pfi_kif *kif, struct pf_state_cmp *key, u_int8_t tree)
637130613Smlaier{
638130613Smlaier	struct pf_state *s;
639130613Smlaier
640126258Smlaier	pf_status.fcounters[FCNT_STATE_SEARCH]++;
641130613Smlaier
642130613Smlaier	switch (tree) {
643130613Smlaier	case PF_LAN_EXT:
644171168Smlaier		if ((s = RB_FIND(pf_state_tree_lan_ext, &kif->pfik_lan_ext,
645171168Smlaier		    (struct pf_state *)key)) != NULL)
646171168Smlaier			return (s);
647171168Smlaier		if ((s = RB_FIND(pf_state_tree_lan_ext, &pfi_all->pfik_lan_ext,
648171168Smlaier		    (struct pf_state *)key)) != NULL)
649171168Smlaier			return (s);
650126258Smlaier		return (NULL);
651130613Smlaier	case PF_EXT_GWY:
652171168Smlaier		if ((s = RB_FIND(pf_state_tree_ext_gwy, &kif->pfik_ext_gwy,
653171168Smlaier		    (struct pf_state *)key)) != NULL)
654171168Smlaier			return (s);
655171168Smlaier		if ((s = RB_FIND(pf_state_tree_ext_gwy, &pfi_all->pfik_ext_gwy,
656171168Smlaier		    (struct pf_state *)key)) != NULL)
657171168Smlaier			return (s);
658130613Smlaier		return (NULL);
659130613Smlaier	default:
660130613Smlaier		panic("pf_find_state_recurse");
661130613Smlaier	}
662126258Smlaier}
663126258Smlaier
664130613Smlaierstruct pf_state *
665171168Smlaierpf_find_state_all(struct pf_state_cmp *key, u_int8_t tree, int *more)
666130613Smlaier{
667130613Smlaier	struct pf_state *s, *ss = NULL;
668130613Smlaier	struct pfi_kif	*kif;
669130613Smlaier
670130613Smlaier	pf_status.fcounters[FCNT_STATE_SEARCH]++;
671130613Smlaier
672130613Smlaier	switch (tree) {
673130613Smlaier	case PF_LAN_EXT:
674130613Smlaier		TAILQ_FOREACH(kif, &pfi_statehead, pfik_w_states) {
675130613Smlaier			s = RB_FIND(pf_state_tree_lan_ext,
676171168Smlaier			    &kif->pfik_lan_ext, (struct pf_state *)key);
677130613Smlaier			if (s == NULL)
678130613Smlaier				continue;
679130613Smlaier			if (more == NULL)
680130613Smlaier				return (s);
681130613Smlaier			ss = s;
682130613Smlaier			(*more)++;
683130613Smlaier		}
684130613Smlaier		return (ss);
685130613Smlaier	case PF_EXT_GWY:
686130613Smlaier		TAILQ_FOREACH(kif, &pfi_statehead, pfik_w_states) {
687130613Smlaier			s = RB_FIND(pf_state_tree_ext_gwy,
688171168Smlaier			    &kif->pfik_ext_gwy, (struct pf_state *)key);
689130613Smlaier			if (s == NULL)
690130613Smlaier				continue;
691130613Smlaier			if (more == NULL)
692130613Smlaier				return (s);
693130613Smlaier			ss = s;
694130613Smlaier			(*more)++;
695130613Smlaier		}
696130613Smlaier		return (ss);
697130613Smlaier	default:
698130613Smlaier		panic("pf_find_state_all");
699130613Smlaier	}
700130613Smlaier}
701130613Smlaier
702145836Smlaiervoid
703145836Smlaierpf_init_threshold(struct pf_threshold *threshold,
704145836Smlaier    u_int32_t limit, u_int32_t seconds)
705145836Smlaier{
706145836Smlaier	threshold->limit = limit * PF_THRESHOLD_MULT;
707145836Smlaier	threshold->seconds = seconds;
708145836Smlaier	threshold->count = 0;
709145836Smlaier	threshold->last = time_second;
710145836Smlaier}
711145836Smlaier
712145836Smlaiervoid
713145836Smlaierpf_add_threshold(struct pf_threshold *threshold)
714145836Smlaier{
715145836Smlaier	u_int32_t t = time_second, diff = t - threshold->last;
716145836Smlaier
717145836Smlaier	if (diff >= threshold->seconds)
718145836Smlaier		threshold->count = 0;
719145836Smlaier	else
720145836Smlaier		threshold->count -= threshold->count * diff /
721145836Smlaier		    threshold->seconds;
722145836Smlaier	threshold->count += PF_THRESHOLD_MULT;
723145836Smlaier	threshold->last = t;
724145836Smlaier}
725145836Smlaier
726126258Smlaierint
727145836Smlaierpf_check_threshold(struct pf_threshold *threshold)
728145836Smlaier{
729145836Smlaier	return (threshold->count > threshold->limit);
730145836Smlaier}
731145836Smlaier
732145836Smlaierint
733145836Smlaierpf_src_connlimit(struct pf_state **state)
734145836Smlaier{
735145836Smlaier	struct pf_state	*s;
736145836Smlaier	int bad = 0;
737145836Smlaier
738145836Smlaier	(*state)->src_node->conn++;
739171168Smlaier	(*state)->src.tcp_est = 1;
740145836Smlaier	pf_add_threshold(&(*state)->src_node->conn_rate);
741145836Smlaier
742145836Smlaier	if ((*state)->rule.ptr->max_src_conn &&
743145836Smlaier	    (*state)->rule.ptr->max_src_conn <
744145836Smlaier	    (*state)->src_node->conn) {
745145836Smlaier		pf_status.lcounters[LCNT_SRCCONN]++;
746145836Smlaier		bad++;
747145836Smlaier	}
748145836Smlaier
749145836Smlaier	if ((*state)->rule.ptr->max_src_conn_rate.limit &&
750145836Smlaier	    pf_check_threshold(&(*state)->src_node->conn_rate)) {
751145836Smlaier		pf_status.lcounters[LCNT_SRCCONNRATE]++;
752145836Smlaier		bad++;
753145836Smlaier	}
754145836Smlaier
755145836Smlaier	if (!bad)
756145836Smlaier		return (0);
757145836Smlaier
758145836Smlaier	if ((*state)->rule.ptr->overload_tbl) {
759145836Smlaier		struct pfr_addr p;
760145836Smlaier		u_int32_t	killed = 0;
761145836Smlaier
762145836Smlaier		pf_status.lcounters[LCNT_OVERLOAD_TABLE]++;
763145836Smlaier		if (pf_status.debug >= PF_DEBUG_MISC) {
764145836Smlaier			printf("pf_src_connlimit: blocking address ");
765145836Smlaier			pf_print_host(&(*state)->src_node->addr, 0,
766145836Smlaier			    (*state)->af);
767145836Smlaier		}
768145836Smlaier
769145836Smlaier		bzero(&p, sizeof(p));
770145836Smlaier		p.pfra_af = (*state)->af;
771145836Smlaier		switch ((*state)->af) {
772145836Smlaier#ifdef INET
773145836Smlaier		case AF_INET:
774145836Smlaier			p.pfra_net = 32;
775145836Smlaier			p.pfra_ip4addr = (*state)->src_node->addr.v4;
776145836Smlaier			break;
777145836Smlaier#endif /* INET */
778145836Smlaier#ifdef INET6
779145836Smlaier		case AF_INET6:
780145836Smlaier			p.pfra_net = 128;
781145836Smlaier			p.pfra_ip6addr = (*state)->src_node->addr.v6;
782145836Smlaier			break;
783145836Smlaier#endif /* INET6 */
784145836Smlaier		}
785145836Smlaier
786145836Smlaier		pfr_insert_kentry((*state)->rule.ptr->overload_tbl,
787145836Smlaier		    &p, time_second);
788145836Smlaier
789145836Smlaier		/* kill existing states if that's required. */
790145836Smlaier		if ((*state)->rule.ptr->flush) {
791145836Smlaier			pf_status.lcounters[LCNT_OVERLOAD_FLUSH]++;
792145836Smlaier
793145836Smlaier			RB_FOREACH(s, pf_state_tree_id, &tree_id) {
794145836Smlaier				/*
795145836Smlaier				 * Kill states from this source.  (Only those
796145836Smlaier				 * from the same rule if PF_FLUSH_GLOBAL is not
797145836Smlaier				 * set)
798145836Smlaier				 */
799145836Smlaier				if (s->af == (*state)->af &&
800145836Smlaier				    (((*state)->direction == PF_OUT &&
801145836Smlaier				    PF_AEQ(&(*state)->src_node->addr,
802145836Smlaier				    &s->lan.addr, s->af)) ||
803145836Smlaier				    ((*state)->direction == PF_IN &&
804145836Smlaier				    PF_AEQ(&(*state)->src_node->addr,
805145836Smlaier				    &s->ext.addr, s->af))) &&
806145836Smlaier				    ((*state)->rule.ptr->flush &
807145836Smlaier				    PF_FLUSH_GLOBAL ||
808145836Smlaier				    (*state)->rule.ptr == s->rule.ptr)) {
809145836Smlaier					s->timeout = PFTM_PURGE;
810145836Smlaier					s->src.state = s->dst.state =
811145836Smlaier					    TCPS_CLOSED;
812145836Smlaier					killed++;
813145836Smlaier				}
814145836Smlaier			}
815145836Smlaier			if (pf_status.debug >= PF_DEBUG_MISC)
816145836Smlaier				printf(", %u states killed", killed);
817145836Smlaier		}
818145836Smlaier		if (pf_status.debug >= PF_DEBUG_MISC)
819145836Smlaier			printf("\n");
820145836Smlaier	}
821145836Smlaier
822145836Smlaier	/* kill this state */
823145836Smlaier	(*state)->timeout = PFTM_PURGE;
824145836Smlaier	(*state)->src.state = (*state)->dst.state = TCPS_CLOSED;
825145836Smlaier	return (1);
826145836Smlaier}
827145836Smlaier
828145836Smlaierint
829130613Smlaierpf_insert_src_node(struct pf_src_node **sn, struct pf_rule *rule,
830130613Smlaier    struct pf_addr *src, sa_family_t af)
831126258Smlaier{
832130613Smlaier	struct pf_src_node	k;
833126258Smlaier
834130613Smlaier	if (*sn == NULL) {
835130613Smlaier		k.af = af;
836130613Smlaier		PF_ACPY(&k.addr, src, af);
837130613Smlaier		if (rule->rule_flag & PFRULE_RULESRCTRACK ||
838130613Smlaier		    rule->rpool.opts & PF_POOL_STICKYADDR)
839130613Smlaier			k.rule.ptr = rule;
840130613Smlaier		else
841130613Smlaier			k.rule.ptr = NULL;
842130613Smlaier		pf_status.scounters[SCNT_SRC_NODE_SEARCH]++;
843130613Smlaier		*sn = RB_FIND(pf_src_tree, &tree_src_tracking, &k);
844130613Smlaier	}
845130613Smlaier	if (*sn == NULL) {
846130613Smlaier		if (!rule->max_src_nodes ||
847130613Smlaier		    rule->src_nodes < rule->max_src_nodes)
848130613Smlaier			(*sn) = pool_get(&pf_src_tree_pl, PR_NOWAIT);
849145836Smlaier		else
850145836Smlaier			pf_status.lcounters[LCNT_SRCNODES]++;
851130613Smlaier		if ((*sn) == NULL)
852130613Smlaier			return (-1);
853130613Smlaier		bzero(*sn, sizeof(struct pf_src_node));
854145836Smlaier
855145836Smlaier		pf_init_threshold(&(*sn)->conn_rate,
856145836Smlaier		    rule->max_src_conn_rate.limit,
857145836Smlaier		    rule->max_src_conn_rate.seconds);
858145836Smlaier
859130613Smlaier		(*sn)->af = af;
860130613Smlaier		if (rule->rule_flag & PFRULE_RULESRCTRACK ||
861130613Smlaier		    rule->rpool.opts & PF_POOL_STICKYADDR)
862130613Smlaier			(*sn)->rule.ptr = rule;
863130613Smlaier		else
864130613Smlaier			(*sn)->rule.ptr = NULL;
865130613Smlaier		PF_ACPY(&(*sn)->addr, src, af);
866130613Smlaier		if (RB_INSERT(pf_src_tree,
867130613Smlaier		    &tree_src_tracking, *sn) != NULL) {
868130613Smlaier			if (pf_status.debug >= PF_DEBUG_MISC) {
869130613Smlaier				printf("pf: src_tree insert failed: ");
870130613Smlaier				pf_print_host(&(*sn)->addr, 0, af);
871130613Smlaier				printf("\n");
872130613Smlaier			}
873130613Smlaier			pool_put(&pf_src_tree_pl, *sn);
874130613Smlaier			return (-1);
875130613Smlaier		}
876130613Smlaier		(*sn)->creation = time_second;
877130613Smlaier		(*sn)->ruletype = rule->action;
878130613Smlaier		if ((*sn)->rule.ptr != NULL)
879130613Smlaier			(*sn)->rule.ptr->src_nodes++;
880130613Smlaier		pf_status.scounters[SCNT_SRC_NODE_INSERT]++;
881130613Smlaier		pf_status.src_nodes++;
882130613Smlaier	} else {
883130613Smlaier		if (rule->max_src_states &&
884145836Smlaier		    (*sn)->states >= rule->max_src_states) {
885145836Smlaier			pf_status.lcounters[LCNT_SRCSTATES]++;
886130613Smlaier			return (-1);
887145836Smlaier		}
888130613Smlaier	}
889130613Smlaier	return (0);
890130613Smlaier}
891126258Smlaier
892130613Smlaierint
893130613Smlaierpf_insert_state(struct pfi_kif *kif, struct pf_state *state)
894130613Smlaier{
895126258Smlaier	/* Thou MUST NOT insert multiple duplicate keys */
896130613Smlaier	state->u.s.kif = kif;
897130613Smlaier	if (RB_INSERT(pf_state_tree_lan_ext, &kif->pfik_lan_ext, state)) {
898126258Smlaier		if (pf_status.debug >= PF_DEBUG_MISC) {
899126258Smlaier			printf("pf: state insert failed: tree_lan_ext");
900126258Smlaier			printf(" lan: ");
901126258Smlaier			pf_print_host(&state->lan.addr, state->lan.port,
902126258Smlaier			    state->af);
903126258Smlaier			printf(" gwy: ");
904126258Smlaier			pf_print_host(&state->gwy.addr, state->gwy.port,
905126258Smlaier			    state->af);
906126258Smlaier			printf(" ext: ");
907126258Smlaier			pf_print_host(&state->ext.addr, state->ext.port,
908126258Smlaier			    state->af);
909130613Smlaier			if (state->sync_flags & PFSTATE_FROMSYNC)
910130613Smlaier				printf(" (from sync)");
911126258Smlaier			printf("\n");
912126258Smlaier		}
913126258Smlaier		return (-1);
914126258Smlaier	}
915126258Smlaier
916130613Smlaier	if (RB_INSERT(pf_state_tree_ext_gwy, &kif->pfik_ext_gwy, state)) {
917126258Smlaier		if (pf_status.debug >= PF_DEBUG_MISC) {
918126258Smlaier			printf("pf: state insert failed: tree_ext_gwy");
919126258Smlaier			printf(" lan: ");
920126258Smlaier			pf_print_host(&state->lan.addr, state->lan.port,
921126258Smlaier			    state->af);
922126258Smlaier			printf(" gwy: ");
923126258Smlaier			pf_print_host(&state->gwy.addr, state->gwy.port,
924126258Smlaier			    state->af);
925126258Smlaier			printf(" ext: ");
926126258Smlaier			pf_print_host(&state->ext.addr, state->ext.port,
927126258Smlaier			    state->af);
928130613Smlaier			if (state->sync_flags & PFSTATE_FROMSYNC)
929130613Smlaier				printf(" (from sync)");
930126258Smlaier			printf("\n");
931126258Smlaier		}
932130613Smlaier		RB_REMOVE(pf_state_tree_lan_ext, &kif->pfik_lan_ext, state);
933126258Smlaier		return (-1);
934126258Smlaier	}
935126258Smlaier
936130613Smlaier	if (state->id == 0 && state->creatorid == 0) {
937130613Smlaier		state->id = htobe64(pf_status.stateid++);
938130613Smlaier		state->creatorid = pf_status.hostid;
939130613Smlaier	}
940130613Smlaier	if (RB_INSERT(pf_state_tree_id, &tree_id, state) != NULL) {
941130613Smlaier		if (pf_status.debug >= PF_DEBUG_MISC) {
942130613Smlaier#ifdef __FreeBSD__
943130613Smlaier			printf("pf: state insert failed: "
944130613Smlaier			    "id: %016llx creatorid: %08x",
945130613Smlaier			    (long long)be64toh(state->id),
946130613Smlaier			    ntohl(state->creatorid));
947130613Smlaier#else
948130613Smlaier			printf("pf: state insert failed: "
949130613Smlaier			    "id: %016llx creatorid: %08x",
950130613Smlaier			    betoh64(state->id), ntohl(state->creatorid));
951130613Smlaier#endif
952130613Smlaier			if (state->sync_flags & PFSTATE_FROMSYNC)
953130613Smlaier				printf(" (from sync)");
954130613Smlaier			printf("\n");
955130613Smlaier		}
956130613Smlaier		RB_REMOVE(pf_state_tree_lan_ext, &kif->pfik_lan_ext, state);
957130613Smlaier		RB_REMOVE(pf_state_tree_ext_gwy, &kif->pfik_ext_gwy, state);
958130613Smlaier		return (-1);
959130613Smlaier	}
960171168Smlaier	TAILQ_INSERT_TAIL(&state_list, state, u.s.entry_list);
961126258Smlaier	pf_status.fcounters[FCNT_STATE_INSERT]++;
962126258Smlaier	pf_status.states++;
963171168Smlaier	pfi_kif_ref(kif, PFI_KIF_REF_STATE);
964126258Smlaier#if NPFSYNC
965126258Smlaier	pfsync_insert_state(state);
966126258Smlaier#endif
967126258Smlaier	return (0);
968126258Smlaier}
969126258Smlaier
970126258Smlaiervoid
971171168Smlaierpf_purge_thread(void *v)
972126258Smlaier{
973171168Smlaier	int nloops = 0, s;
974171168Smlaier
975171168Smlaier	for (;;) {
976171168Smlaier		tsleep(pf_purge_thread, PWAIT, "pftm", 1 * hz);
977171168Smlaier
978127145Smlaier#ifdef __FreeBSD__
979171168Smlaier		sx_slock(&pf_consistency_lock);
980171168Smlaier		PF_LOCK();
981126258Smlaier
982171168Smlaier		if (pf_end_threads) {
983171168Smlaier			pf_purge_expired_states(pf_status.states);
984171168Smlaier			pf_purge_expired_fragments();
985171168Smlaier			pf_purge_expired_src_nodes(0);
986171168Smlaier			pf_end_threads++;
987171168Smlaier
988171168Smlaier			sx_sunlock(&pf_consistency_lock);
989171168Smlaier			PF_UNLOCK();
990171168Smlaier			wakeup(pf_purge_thread);
991171168Smlaier			kthread_exit(0);
992171168Smlaier		}
993126261Smlaier#endif
994171168Smlaier		s = splsoftnet();
995126258Smlaier
996171168Smlaier		/* process a fraction of the state table every second */
997171168Smlaier		pf_purge_expired_states(1 + (pf_status.states
998171168Smlaier		    / pf_default_rule.timeout[PFTM_INTERVAL]));
999171168Smlaier
1000171168Smlaier		/* purge other expired types every PFTM_INTERVAL seconds */
1001171168Smlaier		if (++nloops >= pf_default_rule.timeout[PFTM_INTERVAL]) {
1002171168Smlaier			pf_purge_expired_fragments();
1003171168Smlaier			pf_purge_expired_src_nodes(0);
1004171168Smlaier			nloops = 0;
1005171168Smlaier		}
1006171168Smlaier
1007171168Smlaier		splx(s);
1008127145Smlaier#ifdef __FreeBSD__
1009171168Smlaier		PF_UNLOCK();
1010171168Smlaier		sx_sunlock(&pf_consistency_lock);
1011126261Smlaier#endif
1012171168Smlaier	}
1013126258Smlaier}
1014126258Smlaier
1015126258Smlaieru_int32_t
1016126258Smlaierpf_state_expires(const struct pf_state *state)
1017126258Smlaier{
1018126258Smlaier	u_int32_t	timeout;
1019126258Smlaier	u_int32_t	start;
1020126258Smlaier	u_int32_t	end;
1021126258Smlaier	u_int32_t	states;
1022126258Smlaier
1023126258Smlaier	/* handle all PFTM_* > PFTM_MAX here */
1024126258Smlaier	if (state->timeout == PFTM_PURGE)
1025126261Smlaier		return (time_second);
1026126258Smlaier	if (state->timeout == PFTM_UNTIL_PACKET)
1027126258Smlaier		return (0);
1028127145Smlaier#ifdef __FreeBSD__
1029171168Smlaier	KASSERT(state->timeout != PFTM_UNLINKED,
1030171168Smlaier	    ("pf_state_expires: timeout == PFTM_UNLINKED"));
1031126261Smlaier	KASSERT((state->timeout < PFTM_MAX),
1032126261Smlaier	    ("pf_state_expires: timeout > PFTM_MAX"));
1033126261Smlaier#else
1034171168Smlaier	KASSERT(state->timeout != PFTM_UNLINKED);
1035126258Smlaier	KASSERT(state->timeout < PFTM_MAX);
1036126261Smlaier#endif
1037126258Smlaier	timeout = state->rule.ptr->timeout[state->timeout];
1038126258Smlaier	if (!timeout)
1039126258Smlaier		timeout = pf_default_rule.timeout[state->timeout];
1040126258Smlaier	start = state->rule.ptr->timeout[PFTM_ADAPTIVE_START];
1041126258Smlaier	if (start) {
1042126258Smlaier		end = state->rule.ptr->timeout[PFTM_ADAPTIVE_END];
1043126258Smlaier		states = state->rule.ptr->states;
1044126258Smlaier	} else {
1045126258Smlaier		start = pf_default_rule.timeout[PFTM_ADAPTIVE_START];
1046126258Smlaier		end = pf_default_rule.timeout[PFTM_ADAPTIVE_END];
1047126258Smlaier		states = pf_status.states;
1048126258Smlaier	}
1049126258Smlaier	if (end && states > start && start < end) {
1050126258Smlaier		if (states < end)
1051126258Smlaier			return (state->expire + timeout * (end - states) /
1052126258Smlaier			    (end - start));
1053126258Smlaier		else
1054126261Smlaier			return (time_second);
1055126258Smlaier	}
1056126258Smlaier	return (state->expire + timeout);
1057126258Smlaier}
1058126258Smlaier
1059126258Smlaiervoid
1060171168Smlaierpf_purge_expired_src_nodes(int waslocked)
1061126258Smlaier{
1062130613Smlaier	 struct pf_src_node		*cur, *next;
1063171168Smlaier	 int				 locked = waslocked;
1064126258Smlaier
1065130613Smlaier	 for (cur = RB_MIN(pf_src_tree, &tree_src_tracking); cur; cur = next) {
1066130613Smlaier		 next = RB_NEXT(pf_src_tree, &tree_src_tracking, cur);
1067126258Smlaier
1068130613Smlaier		 if (cur->states <= 0 && cur->expire <= time_second) {
1069171168Smlaier			 if (! locked) {
1070171168Smlaier#ifdef __FreeBSD__
1071171168Smlaier				 if (!sx_try_upgrade(&pf_consistency_lock)) {
1072171168Smlaier					 PF_UNLOCK();
1073171168Smlaier					 sx_sunlock(&pf_consistency_lock);
1074171168Smlaier					 sx_xlock(&pf_consistency_lock);
1075171168Smlaier					 PF_LOCK();
1076171168Smlaier				 }
1077171168Smlaier#else
1078171168Smlaier				 rw_enter_write(&pf_consistency_lock);
1079171168Smlaier#endif
1080171168Smlaier			 	 next = RB_NEXT(pf_src_tree,
1081171168Smlaier				     &tree_src_tracking, cur);
1082171168Smlaier				 locked = 1;
1083171168Smlaier			 }
1084130613Smlaier			 if (cur->rule.ptr != NULL) {
1085130613Smlaier				 cur->rule.ptr->src_nodes--;
1086130613Smlaier				 if (cur->rule.ptr->states <= 0 &&
1087130613Smlaier				     cur->rule.ptr->max_src_nodes <= 0)
1088130613Smlaier					 pf_rm_rule(NULL, cur->rule.ptr);
1089130613Smlaier			 }
1090130613Smlaier			 RB_REMOVE(pf_src_tree, &tree_src_tracking, cur);
1091130613Smlaier			 pf_status.scounters[SCNT_SRC_NODE_REMOVALS]++;
1092130613Smlaier			 pf_status.src_nodes--;
1093130613Smlaier			 pool_put(&pf_src_tree_pl, cur);
1094130613Smlaier		 }
1095130613Smlaier	 }
1096171168Smlaier
1097171168Smlaier	 if (locked && !waslocked)
1098171168Smlaier#ifdef __FreeBSD__
1099171168Smlaier		sx_downgrade(&pf_consistency_lock);
1100171168Smlaier#else
1101171168Smlaier		rw_exit_write(&pf_consistency_lock);
1102171168Smlaier#endif
1103130613Smlaier}
1104126258Smlaier
1105130613Smlaiervoid
1106130613Smlaierpf_src_tree_remove_state(struct pf_state *s)
1107130613Smlaier{
1108130613Smlaier	u_int32_t timeout;
1109126258Smlaier
1110130613Smlaier	if (s->src_node != NULL) {
1111145836Smlaier		if (s->proto == IPPROTO_TCP) {
1112171168Smlaier			if (s->src.tcp_est)
1113145836Smlaier				--s->src_node->conn;
1114145836Smlaier		}
1115130613Smlaier		if (--s->src_node->states <= 0) {
1116130613Smlaier			timeout = s->rule.ptr->timeout[PFTM_SRC_NODE];
1117130613Smlaier			if (!timeout)
1118130613Smlaier				timeout =
1119130613Smlaier				    pf_default_rule.timeout[PFTM_SRC_NODE];
1120130613Smlaier			s->src_node->expire = time_second + timeout;
1121130613Smlaier		}
1122130613Smlaier	}
1123130613Smlaier	if (s->nat_src_node != s->src_node && s->nat_src_node != NULL) {
1124130613Smlaier		if (--s->nat_src_node->states <= 0) {
1125130613Smlaier			timeout = s->rule.ptr->timeout[PFTM_SRC_NODE];
1126130613Smlaier			if (!timeout)
1127130613Smlaier				timeout =
1128130613Smlaier				    pf_default_rule.timeout[PFTM_SRC_NODE];
1129130613Smlaier			s->nat_src_node->expire = time_second + timeout;
1130130613Smlaier		}
1131130613Smlaier	}
1132130613Smlaier	s->src_node = s->nat_src_node = NULL;
1133130613Smlaier}
1134126258Smlaier
1135171168Smlaier/* callers should be at splsoftnet */
1136130613Smlaiervoid
1137171168Smlaierpf_unlink_state(struct pf_state *cur)
1138145836Smlaier{
1139148196Smlaier#ifdef __FreeBSD__
1140153545Smlaier	if (cur->local_flags & PFSTATE_EXPIRING)
1141148196Smlaier		return;
1142153545Smlaier	cur->local_flags |= PFSTATE_EXPIRING;
1143148196Smlaier#endif
1144171168Smlaier	if (cur->src.state == PF_TCPS_PROXY_DST) {
1145162238Scsjp#ifdef __FreeBSD__
1146162238Scsjp		pf_send_tcp(NULL, cur->rule.ptr, cur->af,
1147162238Scsjp#else
1148145836Smlaier		pf_send_tcp(cur->rule.ptr, cur->af,
1149162238Scsjp#endif
1150145836Smlaier		    &cur->ext.addr, &cur->lan.addr,
1151145836Smlaier		    cur->ext.port, cur->lan.port,
1152145836Smlaier		    cur->src.seqhi, cur->src.seqlo + 1,
1153171168Smlaier		    TH_RST|TH_ACK, 0, 0, 0, 1, cur->tag, NULL, NULL);
1154171168Smlaier	}
1155145836Smlaier	RB_REMOVE(pf_state_tree_ext_gwy,
1156145836Smlaier	    &cur->u.s.kif->pfik_ext_gwy, cur);
1157145836Smlaier	RB_REMOVE(pf_state_tree_lan_ext,
1158145836Smlaier	    &cur->u.s.kif->pfik_lan_ext, cur);
1159145836Smlaier	RB_REMOVE(pf_state_tree_id, &tree_id, cur);
1160145836Smlaier#if NPFSYNC
1161171168Smlaier	if (cur->creatorid == pf_status.hostid)
1162171168Smlaier		pfsync_delete_state(cur);
1163145836Smlaier#endif
1164171168Smlaier	cur->timeout = PFTM_UNLINKED;
1165145836Smlaier	pf_src_tree_remove_state(cur);
1166171168Smlaier}
1167171168Smlaier
1168171168Smlaier/* callers should be at splsoftnet and hold the
1169171168Smlaier * write_lock on pf_consistency_lock */
1170171168Smlaiervoid
1171171168Smlaierpf_free_state(struct pf_state *cur)
1172171168Smlaier{
1173171168Smlaier#if NPFSYNC
1174171168Smlaier	if (pfsyncif != NULL &&
1175171168Smlaier	    (pfsyncif->sc_bulk_send_next == cur ||
1176171168Smlaier	    pfsyncif->sc_bulk_terminator == cur))
1177171168Smlaier		return;
1178171168Smlaier#endif
1179171168Smlaier#ifdef __FreeBSD__
1180171168Smlaier	KASSERT(cur->timeout == PFTM_UNLINKED,
1181171168Smlaier	    ("pf_free_state: cur->timeout != PFTM_UNLINKED"));
1182171168Smlaier#else
1183171168Smlaier	KASSERT(cur->timeout == PFTM_UNLINKED);
1184171168Smlaier#endif
1185145836Smlaier	if (--cur->rule.ptr->states <= 0 &&
1186145836Smlaier	    cur->rule.ptr->src_nodes <= 0)
1187145836Smlaier		pf_rm_rule(NULL, cur->rule.ptr);
1188145836Smlaier	if (cur->nat_rule.ptr != NULL)
1189145836Smlaier		if (--cur->nat_rule.ptr->states <= 0 &&
1190145836Smlaier			cur->nat_rule.ptr->src_nodes <= 0)
1191145836Smlaier			pf_rm_rule(NULL, cur->nat_rule.ptr);
1192145836Smlaier	if (cur->anchor.ptr != NULL)
1193145836Smlaier		if (--cur->anchor.ptr->states <= 0)
1194145836Smlaier			pf_rm_rule(NULL, cur->anchor.ptr);
1195145836Smlaier	pf_normalize_tcp_cleanup(cur);
1196171168Smlaier	pfi_kif_unref(cur->u.s.kif, PFI_KIF_REF_STATE);
1197171168Smlaier	TAILQ_REMOVE(&state_list, cur, u.s.entry_list);
1198145836Smlaier	if (cur->tag)
1199145836Smlaier		pf_tag_unref(cur->tag);
1200145836Smlaier	pool_put(&pf_state_pl, cur);
1201145836Smlaier	pf_status.fcounters[FCNT_STATE_REMOVALS]++;
1202145836Smlaier	pf_status.states--;
1203145836Smlaier}
1204145836Smlaier
1205145836Smlaiervoid
1206171168Smlaierpf_purge_expired_states(u_int32_t maxcheck)
1207130613Smlaier{
1208171168Smlaier	static struct pf_state	*cur = NULL;
1209171168Smlaier	struct pf_state		*next;
1210171168Smlaier	int 			 locked = 0;
1211130613Smlaier
1212171168Smlaier	while (maxcheck--) {
1213171168Smlaier		/* wrap to start of list when we hit the end */
1214171168Smlaier		if (cur == NULL) {
1215171168Smlaier			cur = TAILQ_FIRST(&state_list);
1216171168Smlaier			if (cur == NULL)
1217171168Smlaier				break;	/* list empty */
1218171168Smlaier		}
1219171168Smlaier
1220171168Smlaier		/* get next state, as cur may get deleted */
1221171168Smlaier		next = TAILQ_NEXT(cur, u.s.entry_list);
1222171168Smlaier
1223171168Smlaier		if (cur->timeout == PFTM_UNLINKED) {
1224171168Smlaier			/* free unlinked state */
1225171168Smlaier			if (! locked) {
1226171168Smlaier#ifdef __FreeBSD__
1227171168Smlaier				 if (!sx_try_upgrade(&pf_consistency_lock)) {
1228171168Smlaier					 PF_UNLOCK();
1229171168Smlaier					 sx_sunlock(&pf_consistency_lock);
1230171168Smlaier					 sx_xlock(&pf_consistency_lock);
1231171168Smlaier					 PF_LOCK();
1232171168Smlaier				 }
1233171168Smlaier#else
1234171168Smlaier				rw_enter_write(&pf_consistency_lock);
1235171168Smlaier#endif
1236171168Smlaier				locked = 1;
1237171168Smlaier			}
1238171168Smlaier			pf_free_state(cur);
1239171168Smlaier		} else if (pf_state_expires(cur) <= time_second) {
1240171168Smlaier			/* unlink and free expired state */
1241171168Smlaier			pf_unlink_state(cur);
1242171168Smlaier			if (! locked) {
1243171168Smlaier#ifdef __FreeBSD__
1244171168Smlaier				 if (!sx_try_upgrade(&pf_consistency_lock)) {
1245171168Smlaier					 PF_UNLOCK();
1246171168Smlaier					 sx_sunlock(&pf_consistency_lock);
1247171168Smlaier					 sx_xlock(&pf_consistency_lock);
1248171168Smlaier					 PF_LOCK();
1249171168Smlaier				 }
1250171168Smlaier#else
1251171168Smlaier				rw_enter_write(&pf_consistency_lock);
1252171168Smlaier#endif
1253171168Smlaier				locked = 1;
1254171168Smlaier			}
1255171168Smlaier			pf_free_state(cur);
1256171168Smlaier		}
1257171168Smlaier		cur = next;
1258126258Smlaier	}
1259171168Smlaier
1260171168Smlaier	if (locked)
1261171168Smlaier#ifdef __FreeBSD__
1262171168Smlaier		sx_downgrade(&pf_consistency_lock);
1263171168Smlaier#else
1264171168Smlaier		rw_exit_write(&pf_consistency_lock);
1265171168Smlaier#endif
1266126258Smlaier}
1267126258Smlaier
1268126258Smlaierint
1269126258Smlaierpf_tbladdr_setup(struct pf_ruleset *rs, struct pf_addr_wrap *aw)
1270126258Smlaier{
1271126258Smlaier	if (aw->type != PF_ADDR_TABLE)
1272126258Smlaier		return (0);
1273126258Smlaier	if ((aw->p.tbl = pfr_attach_table(rs, aw->v.tblname)) == NULL)
1274126258Smlaier		return (1);
1275126258Smlaier	return (0);
1276126258Smlaier}
1277126258Smlaier
1278126258Smlaiervoid
1279126258Smlaierpf_tbladdr_remove(struct pf_addr_wrap *aw)
1280126258Smlaier{
1281126258Smlaier	if (aw->type != PF_ADDR_TABLE || aw->p.tbl == NULL)
1282126258Smlaier		return;
1283126258Smlaier	pfr_detach_table(aw->p.tbl);
1284126258Smlaier	aw->p.tbl = NULL;
1285126258Smlaier}
1286126258Smlaier
1287126258Smlaiervoid
1288126258Smlaierpf_tbladdr_copyout(struct pf_addr_wrap *aw)
1289126258Smlaier{
1290126258Smlaier	struct pfr_ktable *kt = aw->p.tbl;
1291126258Smlaier
1292126258Smlaier	if (aw->type != PF_ADDR_TABLE || kt == NULL)
1293126258Smlaier		return;
1294126258Smlaier	if (!(kt->pfrkt_flags & PFR_TFLAG_ACTIVE) && kt->pfrkt_root != NULL)
1295126258Smlaier		kt = kt->pfrkt_root;
1296126258Smlaier	aw->p.tbl = NULL;
1297126258Smlaier	aw->p.tblcnt = (kt->pfrkt_flags & PFR_TFLAG_ACTIVE) ?
1298126258Smlaier		kt->pfrkt_cnt : -1;
1299126258Smlaier}
1300126258Smlaier
1301126258Smlaiervoid
1302126258Smlaierpf_print_host(struct pf_addr *addr, u_int16_t p, sa_family_t af)
1303126258Smlaier{
1304126258Smlaier	switch (af) {
1305126258Smlaier#ifdef INET
1306126258Smlaier	case AF_INET: {
1307126258Smlaier		u_int32_t a = ntohl(addr->addr32[0]);
1308126258Smlaier		printf("%u.%u.%u.%u", (a>>24)&255, (a>>16)&255,
1309126258Smlaier		    (a>>8)&255, a&255);
1310126258Smlaier		if (p) {
1311126258Smlaier			p = ntohs(p);
1312126258Smlaier			printf(":%u", p);
1313126258Smlaier		}
1314126258Smlaier		break;
1315126258Smlaier	}
1316126258Smlaier#endif /* INET */
1317126258Smlaier#ifdef INET6
1318126258Smlaier	case AF_INET6: {
1319126258Smlaier		u_int16_t b;
1320126258Smlaier		u_int8_t i, curstart = 255, curend = 0,
1321126258Smlaier		    maxstart = 0, maxend = 0;
1322126258Smlaier		for (i = 0; i < 8; i++) {
1323126258Smlaier			if (!addr->addr16[i]) {
1324126258Smlaier				if (curstart == 255)
1325126258Smlaier					curstart = i;
1326126258Smlaier				else
1327126258Smlaier					curend = i;
1328126258Smlaier			} else {
1329126258Smlaier				if (curstart) {
1330126258Smlaier					if ((curend - curstart) >
1331126258Smlaier					    (maxend - maxstart)) {
1332126258Smlaier						maxstart = curstart;
1333126258Smlaier						maxend = curend;
1334126258Smlaier						curstart = 255;
1335126258Smlaier					}
1336126258Smlaier				}
1337126258Smlaier			}
1338126258Smlaier		}
1339126258Smlaier		for (i = 0; i < 8; i++) {
1340126258Smlaier			if (i >= maxstart && i <= maxend) {
1341126258Smlaier				if (maxend != 7) {
1342126258Smlaier					if (i == maxstart)
1343126258Smlaier						printf(":");
1344126258Smlaier				} else {
1345126258Smlaier					if (i == maxend)
1346126258Smlaier						printf(":");
1347126258Smlaier				}
1348126258Smlaier			} else {
1349126258Smlaier				b = ntohs(addr->addr16[i]);
1350126258Smlaier				printf("%x", b);
1351126258Smlaier				if (i < 7)
1352126258Smlaier					printf(":");
1353126258Smlaier			}
1354126258Smlaier		}
1355126258Smlaier		if (p) {
1356126258Smlaier			p = ntohs(p);
1357126258Smlaier			printf("[%u]", p);
1358126258Smlaier		}
1359126258Smlaier		break;
1360126258Smlaier	}
1361126258Smlaier#endif /* INET6 */
1362126258Smlaier	}
1363126258Smlaier}
1364126258Smlaier
1365126258Smlaiervoid
1366126258Smlaierpf_print_state(struct pf_state *s)
1367126258Smlaier{
1368126258Smlaier	switch (s->proto) {
1369126258Smlaier	case IPPROTO_TCP:
1370126258Smlaier		printf("TCP ");
1371126258Smlaier		break;
1372126258Smlaier	case IPPROTO_UDP:
1373126258Smlaier		printf("UDP ");
1374126258Smlaier		break;
1375126258Smlaier	case IPPROTO_ICMP:
1376126258Smlaier		printf("ICMP ");
1377126258Smlaier		break;
1378126258Smlaier	case IPPROTO_ICMPV6:
1379126258Smlaier		printf("ICMPV6 ");
1380126258Smlaier		break;
1381126258Smlaier	default:
1382126258Smlaier		printf("%u ", s->proto);
1383126258Smlaier		break;
1384126258Smlaier	}
1385126258Smlaier	pf_print_host(&s->lan.addr, s->lan.port, s->af);
1386126258Smlaier	printf(" ");
1387126258Smlaier	pf_print_host(&s->gwy.addr, s->gwy.port, s->af);
1388126258Smlaier	printf(" ");
1389126258Smlaier	pf_print_host(&s->ext.addr, s->ext.port, s->af);
1390126258Smlaier	printf(" [lo=%u high=%u win=%u modulator=%u", s->src.seqlo,
1391126258Smlaier	    s->src.seqhi, s->src.max_win, s->src.seqdiff);
1392126258Smlaier	if (s->src.wscale && s->dst.wscale)
1393126258Smlaier		printf(" wscale=%u", s->src.wscale & PF_WSCALE_MASK);
1394126258Smlaier	printf("]");
1395126258Smlaier	printf(" [lo=%u high=%u win=%u modulator=%u", s->dst.seqlo,
1396126258Smlaier	    s->dst.seqhi, s->dst.max_win, s->dst.seqdiff);
1397126258Smlaier	if (s->src.wscale && s->dst.wscale)
1398126258Smlaier		printf(" wscale=%u", s->dst.wscale & PF_WSCALE_MASK);
1399126258Smlaier	printf("]");
1400126258Smlaier	printf(" %u:%u", s->src.state, s->dst.state);
1401126258Smlaier}
1402126258Smlaier
1403126258Smlaiervoid
1404126258Smlaierpf_print_flags(u_int8_t f)
1405126258Smlaier{
1406126258Smlaier	if (f)
1407126258Smlaier		printf(" ");
1408126258Smlaier	if (f & TH_FIN)
1409126258Smlaier		printf("F");
1410126258Smlaier	if (f & TH_SYN)
1411126258Smlaier		printf("S");
1412126258Smlaier	if (f & TH_RST)
1413126258Smlaier		printf("R");
1414126258Smlaier	if (f & TH_PUSH)
1415126258Smlaier		printf("P");
1416126258Smlaier	if (f & TH_ACK)
1417126258Smlaier		printf("A");
1418126258Smlaier	if (f & TH_URG)
1419126258Smlaier		printf("U");
1420126258Smlaier	if (f & TH_ECE)
1421126258Smlaier		printf("E");
1422126258Smlaier	if (f & TH_CWR)
1423126258Smlaier		printf("W");
1424126258Smlaier}
1425126258Smlaier
1426126258Smlaier#define	PF_SET_SKIP_STEPS(i)					\
1427126258Smlaier	do {							\
1428126258Smlaier		while (head[i] != cur) {			\
1429126258Smlaier			head[i]->skip[i].ptr = cur;		\
1430126258Smlaier			head[i] = TAILQ_NEXT(head[i], entries);	\
1431126258Smlaier		}						\
1432126258Smlaier	} while (0)
1433126258Smlaier
1434126258Smlaiervoid
1435126258Smlaierpf_calc_skip_steps(struct pf_rulequeue *rules)
1436126258Smlaier{
1437126258Smlaier	struct pf_rule *cur, *prev, *head[PF_SKIP_COUNT];
1438126258Smlaier	int i;
1439126258Smlaier
1440126258Smlaier	cur = TAILQ_FIRST(rules);
1441126258Smlaier	prev = cur;
1442126258Smlaier	for (i = 0; i < PF_SKIP_COUNT; ++i)
1443126258Smlaier		head[i] = cur;
1444126258Smlaier	while (cur != NULL) {
1445126258Smlaier
1446130613Smlaier		if (cur->kif != prev->kif || cur->ifnot != prev->ifnot)
1447126258Smlaier			PF_SET_SKIP_STEPS(PF_SKIP_IFP);
1448126258Smlaier		if (cur->direction != prev->direction)
1449126258Smlaier			PF_SET_SKIP_STEPS(PF_SKIP_DIR);
1450126258Smlaier		if (cur->af != prev->af)
1451126258Smlaier			PF_SET_SKIP_STEPS(PF_SKIP_AF);
1452126258Smlaier		if (cur->proto != prev->proto)
1453126258Smlaier			PF_SET_SKIP_STEPS(PF_SKIP_PROTO);
1454145836Smlaier		if (cur->src.neg != prev->src.neg ||
1455126258Smlaier		    pf_addr_wrap_neq(&cur->src.addr, &prev->src.addr))
1456126258Smlaier			PF_SET_SKIP_STEPS(PF_SKIP_SRC_ADDR);
1457126258Smlaier		if (cur->src.port[0] != prev->src.port[0] ||
1458126258Smlaier		    cur->src.port[1] != prev->src.port[1] ||
1459126258Smlaier		    cur->src.port_op != prev->src.port_op)
1460126258Smlaier			PF_SET_SKIP_STEPS(PF_SKIP_SRC_PORT);
1461145836Smlaier		if (cur->dst.neg != prev->dst.neg ||
1462126258Smlaier		    pf_addr_wrap_neq(&cur->dst.addr, &prev->dst.addr))
1463126258Smlaier			PF_SET_SKIP_STEPS(PF_SKIP_DST_ADDR);
1464126258Smlaier		if (cur->dst.port[0] != prev->dst.port[0] ||
1465126258Smlaier		    cur->dst.port[1] != prev->dst.port[1] ||
1466126258Smlaier		    cur->dst.port_op != prev->dst.port_op)
1467126258Smlaier			PF_SET_SKIP_STEPS(PF_SKIP_DST_PORT);
1468126258Smlaier
1469126258Smlaier		prev = cur;
1470126258Smlaier		cur = TAILQ_NEXT(cur, entries);
1471126258Smlaier	}
1472126258Smlaier	for (i = 0; i < PF_SKIP_COUNT; ++i)
1473126258Smlaier		PF_SET_SKIP_STEPS(i);
1474126258Smlaier}
1475126258Smlaier
1476126258Smlaierint
1477126258Smlaierpf_addr_wrap_neq(struct pf_addr_wrap *aw1, struct pf_addr_wrap *aw2)
1478126258Smlaier{
1479126258Smlaier	if (aw1->type != aw2->type)
1480126258Smlaier		return (1);
1481126258Smlaier	switch (aw1->type) {
1482126258Smlaier	case PF_ADDR_ADDRMASK:
1483126258Smlaier		if (PF_ANEQ(&aw1->v.a.addr, &aw2->v.a.addr, 0))
1484126258Smlaier			return (1);
1485126258Smlaier		if (PF_ANEQ(&aw1->v.a.mask, &aw2->v.a.mask, 0))
1486126258Smlaier			return (1);
1487126258Smlaier		return (0);
1488126258Smlaier	case PF_ADDR_DYNIFTL:
1489130613Smlaier		return (aw1->p.dyn->pfid_kt != aw2->p.dyn->pfid_kt);
1490126258Smlaier	case PF_ADDR_NOROUTE:
1491171168Smlaier	case PF_ADDR_URPFFAILED:
1492126258Smlaier		return (0);
1493126258Smlaier	case PF_ADDR_TABLE:
1494126258Smlaier		return (aw1->p.tbl != aw2->p.tbl);
1495171168Smlaier	case PF_ADDR_RTLABEL:
1496171168Smlaier		return (aw1->v.rtlabel != aw2->v.rtlabel);
1497126258Smlaier	default:
1498126258Smlaier		printf("invalid address type: %d\n", aw1->type);
1499126258Smlaier		return (1);
1500126258Smlaier	}
1501126258Smlaier}
1502126258Smlaier
1503126258Smlaieru_int16_t
1504126258Smlaierpf_cksum_fixup(u_int16_t cksum, u_int16_t old, u_int16_t new, u_int8_t udp)
1505126258Smlaier{
1506126258Smlaier	u_int32_t	l;
1507126258Smlaier
1508126258Smlaier	if (udp && !cksum)
1509126258Smlaier		return (0x0000);
1510126258Smlaier	l = cksum + old - new;
1511126258Smlaier	l = (l >> 16) + (l & 65535);
1512126258Smlaier	l = l & 65535;
1513126258Smlaier	if (udp && !l)
1514126258Smlaier		return (0xFFFF);
1515126258Smlaier	return (l);
1516126258Smlaier}
1517126258Smlaier
1518126258Smlaiervoid
1519126258Smlaierpf_change_ap(struct pf_addr *a, u_int16_t *p, u_int16_t *ic, u_int16_t *pc,
1520126258Smlaier    struct pf_addr *an, u_int16_t pn, u_int8_t u, sa_family_t af)
1521126258Smlaier{
1522126258Smlaier	struct pf_addr	ao;
1523126258Smlaier	u_int16_t	po = *p;
1524126258Smlaier
1525126258Smlaier	PF_ACPY(&ao, a, af);
1526126258Smlaier	PF_ACPY(a, an, af);
1527126258Smlaier
1528126258Smlaier	*p = pn;
1529126258Smlaier
1530126258Smlaier	switch (af) {
1531126258Smlaier#ifdef INET
1532126258Smlaier	case AF_INET:
1533126258Smlaier		*ic = pf_cksum_fixup(pf_cksum_fixup(*ic,
1534126258Smlaier		    ao.addr16[0], an->addr16[0], 0),
1535126258Smlaier		    ao.addr16[1], an->addr16[1], 0);
1536126258Smlaier		*p = pn;
1537126258Smlaier		*pc = pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(*pc,
1538126258Smlaier		    ao.addr16[0], an->addr16[0], u),
1539126258Smlaier		    ao.addr16[1], an->addr16[1], u),
1540126258Smlaier		    po, pn, u);
1541126258Smlaier		break;
1542126258Smlaier#endif /* INET */
1543126258Smlaier#ifdef INET6
1544126258Smlaier	case AF_INET6:
1545126258Smlaier		*pc = pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(
1546126258Smlaier		    pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(
1547126258Smlaier		    pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(*pc,
1548126258Smlaier		    ao.addr16[0], an->addr16[0], u),
1549126258Smlaier		    ao.addr16[1], an->addr16[1], u),
1550126258Smlaier		    ao.addr16[2], an->addr16[2], u),
1551126258Smlaier		    ao.addr16[3], an->addr16[3], u),
1552126258Smlaier		    ao.addr16[4], an->addr16[4], u),
1553126258Smlaier		    ao.addr16[5], an->addr16[5], u),
1554126258Smlaier		    ao.addr16[6], an->addr16[6], u),
1555126258Smlaier		    ao.addr16[7], an->addr16[7], u),
1556126258Smlaier		    po, pn, u);
1557126258Smlaier		break;
1558126258Smlaier#endif /* INET6 */
1559126258Smlaier	}
1560126258Smlaier}
1561126258Smlaier
1562126258Smlaier
1563126258Smlaier/* Changes a u_int32_t.  Uses a void * so there are no align restrictions */
1564126258Smlaiervoid
1565126258Smlaierpf_change_a(void *a, u_int16_t *c, u_int32_t an, u_int8_t u)
1566126258Smlaier{
1567126258Smlaier	u_int32_t	ao;
1568126258Smlaier
1569126258Smlaier	memcpy(&ao, a, sizeof(ao));
1570126258Smlaier	memcpy(a, &an, sizeof(u_int32_t));
1571126258Smlaier	*c = pf_cksum_fixup(pf_cksum_fixup(*c, ao / 65536, an / 65536, u),
1572126258Smlaier	    ao % 65536, an % 65536, u);
1573126258Smlaier}
1574126258Smlaier
1575126258Smlaier#ifdef INET6
1576126258Smlaiervoid
1577126258Smlaierpf_change_a6(struct pf_addr *a, u_int16_t *c, struct pf_addr *an, u_int8_t u)
1578126258Smlaier{
1579126258Smlaier	struct pf_addr	ao;
1580126258Smlaier
1581126258Smlaier	PF_ACPY(&ao, a, AF_INET6);
1582126258Smlaier	PF_ACPY(a, an, AF_INET6);
1583126258Smlaier
1584126258Smlaier	*c = pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(
1585126258Smlaier	    pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(
1586126258Smlaier	    pf_cksum_fixup(pf_cksum_fixup(*c,
1587126258Smlaier	    ao.addr16[0], an->addr16[0], u),
1588126258Smlaier	    ao.addr16[1], an->addr16[1], u),
1589126258Smlaier	    ao.addr16[2], an->addr16[2], u),
1590126258Smlaier	    ao.addr16[3], an->addr16[3], u),
1591126258Smlaier	    ao.addr16[4], an->addr16[4], u),
1592126258Smlaier	    ao.addr16[5], an->addr16[5], u),
1593126258Smlaier	    ao.addr16[6], an->addr16[6], u),
1594126258Smlaier	    ao.addr16[7], an->addr16[7], u);
1595126258Smlaier}
1596126258Smlaier#endif /* INET6 */
1597126258Smlaier
1598126258Smlaiervoid
1599126258Smlaierpf_change_icmp(struct pf_addr *ia, u_int16_t *ip, struct pf_addr *oa,
1600126258Smlaier    struct pf_addr *na, u_int16_t np, u_int16_t *pc, u_int16_t *h2c,
1601126258Smlaier    u_int16_t *ic, u_int16_t *hc, u_int8_t u, sa_family_t af)
1602126258Smlaier{
1603126258Smlaier	struct pf_addr	oia, ooa;
1604126258Smlaier
1605126258Smlaier	PF_ACPY(&oia, ia, af);
1606126258Smlaier	PF_ACPY(&ooa, oa, af);
1607126258Smlaier
1608126258Smlaier	/* Change inner protocol port, fix inner protocol checksum. */
1609126258Smlaier	if (ip != NULL) {
1610126258Smlaier		u_int16_t	oip = *ip;
1611127629Smlaier		u_int32_t	opc = 0;	/* make the compiler happy */
1612126258Smlaier
1613126258Smlaier		if (pc != NULL)
1614126258Smlaier			opc = *pc;
1615126258Smlaier		*ip = np;
1616126258Smlaier		if (pc != NULL)
1617126258Smlaier			*pc = pf_cksum_fixup(*pc, oip, *ip, u);
1618126258Smlaier		*ic = pf_cksum_fixup(*ic, oip, *ip, 0);
1619126258Smlaier		if (pc != NULL)
1620126258Smlaier			*ic = pf_cksum_fixup(*ic, opc, *pc, 0);
1621126258Smlaier	}
1622126258Smlaier	/* Change inner ip address, fix inner ip and icmp checksums. */
1623126258Smlaier	PF_ACPY(ia, na, af);
1624126258Smlaier	switch (af) {
1625126258Smlaier#ifdef INET
1626126258Smlaier	case AF_INET: {
1627126258Smlaier		u_int32_t	 oh2c = *h2c;
1628126258Smlaier
1629126258Smlaier		*h2c = pf_cksum_fixup(pf_cksum_fixup(*h2c,
1630126258Smlaier		    oia.addr16[0], ia->addr16[0], 0),
1631126258Smlaier		    oia.addr16[1], ia->addr16[1], 0);
1632126258Smlaier		*ic = pf_cksum_fixup(pf_cksum_fixup(*ic,
1633126258Smlaier		    oia.addr16[0], ia->addr16[0], 0),
1634126258Smlaier		    oia.addr16[1], ia->addr16[1], 0);
1635126258Smlaier		*ic = pf_cksum_fixup(*ic, oh2c, *h2c, 0);
1636126258Smlaier		break;
1637126258Smlaier	}
1638126258Smlaier#endif /* INET */
1639126258Smlaier#ifdef INET6
1640126258Smlaier	case AF_INET6:
1641126258Smlaier		*ic = pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(
1642126258Smlaier		    pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(
1643126258Smlaier		    pf_cksum_fixup(pf_cksum_fixup(*ic,
1644126258Smlaier		    oia.addr16[0], ia->addr16[0], u),
1645126258Smlaier		    oia.addr16[1], ia->addr16[1], u),
1646126258Smlaier		    oia.addr16[2], ia->addr16[2], u),
1647126258Smlaier		    oia.addr16[3], ia->addr16[3], u),
1648126258Smlaier		    oia.addr16[4], ia->addr16[4], u),
1649126258Smlaier		    oia.addr16[5], ia->addr16[5], u),
1650126258Smlaier		    oia.addr16[6], ia->addr16[6], u),
1651126258Smlaier		    oia.addr16[7], ia->addr16[7], u);
1652126258Smlaier		break;
1653126258Smlaier#endif /* INET6 */
1654126258Smlaier	}
1655126258Smlaier	/* Change outer ip address, fix outer ip or icmpv6 checksum. */
1656126258Smlaier	PF_ACPY(oa, na, af);
1657126258Smlaier	switch (af) {
1658126258Smlaier#ifdef INET
1659126258Smlaier	case AF_INET:
1660126258Smlaier		*hc = pf_cksum_fixup(pf_cksum_fixup(*hc,
1661126258Smlaier		    ooa.addr16[0], oa->addr16[0], 0),
1662126258Smlaier		    ooa.addr16[1], oa->addr16[1], 0);
1663126258Smlaier		break;
1664126258Smlaier#endif /* INET */
1665126258Smlaier#ifdef INET6
1666126258Smlaier	case AF_INET6:
1667126258Smlaier		*ic = pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(
1668126258Smlaier		    pf_cksum_fixup(pf_cksum_fixup(pf_cksum_fixup(
1669126258Smlaier		    pf_cksum_fixup(pf_cksum_fixup(*ic,
1670126258Smlaier		    ooa.addr16[0], oa->addr16[0], u),
1671126258Smlaier		    ooa.addr16[1], oa->addr16[1], u),
1672126258Smlaier		    ooa.addr16[2], oa->addr16[2], u),
1673126258Smlaier		    ooa.addr16[3], oa->addr16[3], u),
1674126258Smlaier		    ooa.addr16[4], oa->addr16[4], u),
1675126258Smlaier		    ooa.addr16[5], oa->addr16[5], u),
1676126258Smlaier		    ooa.addr16[6], oa->addr16[6], u),
1677126258Smlaier		    ooa.addr16[7], oa->addr16[7], u);
1678126258Smlaier		break;
1679126258Smlaier#endif /* INET6 */
1680126258Smlaier	}
1681126258Smlaier}
1682126258Smlaier
1683171168Smlaier
1684171168Smlaier/*
1685171168Smlaier * Need to modulate the sequence numbers in the TCP SACK option
1686171168Smlaier * (credits to Krzysztof Pfaff for report and patch)
1687171168Smlaier */
1688171168Smlaierint
1689171168Smlaierpf_modulate_sack(struct mbuf *m, int off, struct pf_pdesc *pd,
1690171168Smlaier    struct tcphdr *th, struct pf_state_peer *dst)
1691171168Smlaier{
1692171168Smlaier	int hlen = (th->th_off << 2) - sizeof(*th), thoptlen = hlen;
1693171168Smlaier#ifdef __FreeBSD__
1694171168Smlaier	u_int8_t opts[TCP_MAXOLEN], *opt = opts;
1695171168Smlaier#else
1696171168Smlaier	u_int8_t opts[MAX_TCPOPTLEN], *opt = opts;
1697171168Smlaier#endif
1698171168Smlaier	int copyback = 0, i, olen;
1699171168Smlaier	struct sackblk sack;
1700171168Smlaier
1701171168Smlaier#define TCPOLEN_SACKLEN	(TCPOLEN_SACK + 2)
1702171168Smlaier	if (hlen < TCPOLEN_SACKLEN ||
1703171168Smlaier	    !pf_pull_hdr(m, off + sizeof(*th), opts, hlen, NULL, NULL, pd->af))
1704171168Smlaier		return 0;
1705171168Smlaier
1706171168Smlaier	while (hlen >= TCPOLEN_SACKLEN) {
1707171168Smlaier		olen = opt[1];
1708171168Smlaier		switch (*opt) {
1709171168Smlaier		case TCPOPT_EOL:	/* FALLTHROUGH */
1710171168Smlaier		case TCPOPT_NOP:
1711171168Smlaier			opt++;
1712171168Smlaier			hlen--;
1713171168Smlaier			break;
1714171168Smlaier		case TCPOPT_SACK:
1715171168Smlaier			if (olen > hlen)
1716171168Smlaier				olen = hlen;
1717171168Smlaier			if (olen >= TCPOLEN_SACKLEN) {
1718171168Smlaier				for (i = 2; i + TCPOLEN_SACK <= olen;
1719171168Smlaier				    i += TCPOLEN_SACK) {
1720171168Smlaier					memcpy(&sack, &opt[i], sizeof(sack));
1721171168Smlaier					pf_change_a(&sack.start, &th->th_sum,
1722171168Smlaier					    htonl(ntohl(sack.start) -
1723171168Smlaier					    dst->seqdiff), 0);
1724171168Smlaier					pf_change_a(&sack.end, &th->th_sum,
1725171168Smlaier					    htonl(ntohl(sack.end) -
1726171168Smlaier					    dst->seqdiff), 0);
1727171168Smlaier					memcpy(&opt[i], &sack, sizeof(sack));
1728171168Smlaier				}
1729171168Smlaier				copyback = 1;
1730171168Smlaier			}
1731171168Smlaier			/* FALLTHROUGH */
1732171168Smlaier		default:
1733171168Smlaier			if (olen < 2)
1734171168Smlaier				olen = 2;
1735171168Smlaier			hlen -= olen;
1736171168Smlaier			opt += olen;
1737171168Smlaier		}
1738171168Smlaier	}
1739171168Smlaier
1740171168Smlaier	if (copyback)
1741171168Smlaier#ifdef __FreeBSD__
1742171168Smlaier		m_copyback(m, off + sizeof(*th), thoptlen, (caddr_t)opts);
1743171168Smlaier#else
1744171168Smlaier		m_copyback(m, off + sizeof(*th), thoptlen, opts);
1745171168Smlaier#endif
1746171168Smlaier	return (copyback);
1747171168Smlaier}
1748171168Smlaier
1749126258Smlaiervoid
1750162238Scsjp#ifdef __FreeBSD__
1751162238Scsjppf_send_tcp(struct mbuf *replyto, const struct pf_rule *r, sa_family_t af,
1752162238Scsjp#else
1753126258Smlaierpf_send_tcp(const struct pf_rule *r, sa_family_t af,
1754162238Scsjp#endif
1755126258Smlaier    const struct pf_addr *saddr, const struct pf_addr *daddr,
1756126258Smlaier    u_int16_t sport, u_int16_t dport, u_int32_t seq, u_int32_t ack,
1757145836Smlaier    u_int8_t flags, u_int16_t win, u_int16_t mss, u_int8_t ttl, int tag,
1758171168Smlaier    u_int16_t rtag, struct ether_header *eh, struct ifnet *ifp)
1759126258Smlaier{
1760126258Smlaier	struct mbuf	*m;
1761171168Smlaier	int		 len, tlen;
1762126258Smlaier#ifdef INET
1763171168Smlaier	struct ip	*h;
1764126258Smlaier#endif /* INET */
1765126258Smlaier#ifdef INET6
1766171168Smlaier	struct ip6_hdr	*h6;
1767126258Smlaier#endif /* INET6 */
1768171168Smlaier	struct tcphdr	*th;
1769171168Smlaier	char		*opt;
1770171168Smlaier	struct pf_mtag	*pf_mtag;
1771126258Smlaier
1772171168Smlaier#ifdef __FreeBSD__
1773171168Smlaier	KASSERT(
1774171168Smlaier#ifdef INET
1775171168Smlaier	    af == AF_INET
1776171168Smlaier#else
1777171168Smlaier	    0
1778171168Smlaier#endif
1779171168Smlaier	    ||
1780171168Smlaier#ifdef INET6
1781171168Smlaier	    af == AF_INET6
1782171168Smlaier#else
1783171168Smlaier	    0
1784171168Smlaier#endif
1785171168Smlaier	    , ("Unsupported AF %d", af));
1786171168Smlaier	len = 0;
1787171168Smlaier	th = NULL;
1788171168Smlaier#ifdef INET
1789171168Smlaier	h = NULL;
1790171168Smlaier#endif
1791171168Smlaier#ifdef INET6
1792171168Smlaier	h6 = NULL;
1793171168Smlaier#endif
1794171168Smlaier#endif
1795171168Smlaier
1796126258Smlaier	/* maximum segment size tcp option */
1797126258Smlaier	tlen = sizeof(struct tcphdr);
1798126258Smlaier	if (mss)
1799126258Smlaier		tlen += 4;
1800126258Smlaier
1801126258Smlaier	switch (af) {
1802126258Smlaier#ifdef INET
1803126258Smlaier	case AF_INET:
1804126258Smlaier		len = sizeof(struct ip) + tlen;
1805126258Smlaier		break;
1806126258Smlaier#endif /* INET */
1807126258Smlaier#ifdef INET6
1808126258Smlaier	case AF_INET6:
1809126258Smlaier		len = sizeof(struct ip6_hdr) + tlen;
1810126258Smlaier		break;
1811126258Smlaier#endif /* INET6 */
1812126258Smlaier	}
1813126258Smlaier
1814126258Smlaier	/* create outgoing mbuf */
1815132280Smlaier	m = m_gethdr(M_DONTWAIT, MT_HEADER);
1816132280Smlaier	if (m == NULL)
1817132280Smlaier		return;
1818162238Scsjp#ifdef __FreeBSD__
1819162238Scsjp#ifdef MAC
1820162238Scsjp	if (replyto)
1821162238Scsjp		mac_create_mbuf_netlayer(replyto, m);
1822162238Scsjp	else
1823162238Scsjp		mac_create_mbuf_from_firewall(m);
1824162238Scsjp#else
1825162238Scsjp	(void)replyto;
1826162238Scsjp#endif
1827162238Scsjp#endif
1828171168Smlaier	if ((pf_mtag = pf_get_mtag(m)) == NULL) {
1829171168Smlaier		m_freem(m);
1830171168Smlaier		return;
1831171168Smlaier	}
1832171168Smlaier	if (tag)
1833145836Smlaier#ifdef __FreeBSD__
1834145836Smlaier		m->m_flags |= M_SKIP_FIREWALL;
1835132280Smlaier#else
1836171168Smlaier		pf_mtag->flags |= PF_TAG_GENERATED;
1837171168Smlaier#endif
1838145836Smlaier
1839171168Smlaier	pf_mtag->tag = rtag;
1840171168Smlaier
1841171168Smlaier	if (r != NULL && r->rtableid >= 0)
1842171168Smlaier		pf_mtag->rtableid = r->rtableid;
1843126258Smlaier#ifdef ALTQ
1844126258Smlaier	if (r != NULL && r->qid) {
1845171168Smlaier		pf_mtag->qid = r->qid;
1846171168Smlaier		/* add hints for ecn */
1847171168Smlaier		pf_mtag->af = af;
1848171168Smlaier		pf_mtag->hdr = mtod(m, struct ip *);
1849126258Smlaier	}
1850145836Smlaier#endif /* ALTQ */
1851126258Smlaier	m->m_data += max_linkhdr;
1852126258Smlaier	m->m_pkthdr.len = m->m_len = len;
1853126258Smlaier	m->m_pkthdr.rcvif = NULL;
1854126258Smlaier	bzero(m->m_data, len);
1855126258Smlaier	switch (af) {
1856126258Smlaier#ifdef INET
1857126258Smlaier	case AF_INET:
1858126258Smlaier		h = mtod(m, struct ip *);
1859126258Smlaier
1860126258Smlaier		/* IP header fields included in the TCP checksum */
1861126258Smlaier		h->ip_p = IPPROTO_TCP;
1862126258Smlaier		h->ip_len = htons(tlen);
1863126258Smlaier		h->ip_src.s_addr = saddr->v4.s_addr;
1864126258Smlaier		h->ip_dst.s_addr = daddr->v4.s_addr;
1865126258Smlaier
1866126258Smlaier		th = (struct tcphdr *)((caddr_t)h + sizeof(struct ip));
1867126258Smlaier		break;
1868126258Smlaier#endif /* INET */
1869126258Smlaier#ifdef INET6
1870126258Smlaier	case AF_INET6:
1871126258Smlaier		h6 = mtod(m, struct ip6_hdr *);
1872126258Smlaier
1873126258Smlaier		/* IP header fields included in the TCP checksum */
1874126258Smlaier		h6->ip6_nxt = IPPROTO_TCP;
1875126258Smlaier		h6->ip6_plen = htons(tlen);
1876126258Smlaier		memcpy(&h6->ip6_src, &saddr->v6, sizeof(struct in6_addr));
1877126258Smlaier		memcpy(&h6->ip6_dst, &daddr->v6, sizeof(struct in6_addr));
1878126258Smlaier
1879126258Smlaier		th = (struct tcphdr *)((caddr_t)h6 + sizeof(struct ip6_hdr));
1880126258Smlaier		break;
1881126258Smlaier#endif /* INET6 */
1882126258Smlaier	}
1883126258Smlaier
1884126258Smlaier	/* TCP header */
1885126258Smlaier	th->th_sport = sport;
1886126258Smlaier	th->th_dport = dport;
1887126258Smlaier	th->th_seq = htonl(seq);
1888126258Smlaier	th->th_ack = htonl(ack);
1889126258Smlaier	th->th_off = tlen >> 2;
1890126258Smlaier	th->th_flags = flags;
1891126258Smlaier	th->th_win = htons(win);
1892126258Smlaier
1893126258Smlaier	if (mss) {
1894126258Smlaier		opt = (char *)(th + 1);
1895126258Smlaier		opt[0] = TCPOPT_MAXSEG;
1896126258Smlaier		opt[1] = 4;
1897126258Smlaier		HTONS(mss);
1898126258Smlaier		bcopy((caddr_t)&mss, (caddr_t)(opt + 2), 2);
1899126258Smlaier	}
1900126258Smlaier
1901126258Smlaier	switch (af) {
1902126258Smlaier#ifdef INET
1903126258Smlaier	case AF_INET:
1904126258Smlaier		/* TCP checksum */
1905126258Smlaier		th->th_sum = in_cksum(m, len);
1906126258Smlaier
1907126258Smlaier		/* Finish the IP header */
1908126258Smlaier		h->ip_v = 4;
1909126258Smlaier		h->ip_hl = sizeof(*h) >> 2;
1910126258Smlaier		h->ip_tos = IPTOS_LOWDELAY;
1911127145Smlaier#ifdef __FreeBSD__
1912130613Smlaier		h->ip_off = path_mtu_discovery ? IP_DF : 0;
1913130613Smlaier		h->ip_len = len;
1914126261Smlaier#else
1915126261Smlaier		h->ip_off = htons(ip_mtudisc ? IP_DF : 0);
1916130613Smlaier		h->ip_len = htons(len);
1917126261Smlaier#endif
1918126258Smlaier		h->ip_ttl = ttl ? ttl : ip_defttl;
1919126258Smlaier		h->ip_sum = 0;
1920145836Smlaier		if (eh == NULL) {
1921127145Smlaier#ifdef __FreeBSD__
1922145836Smlaier			PF_UNLOCK();
1923145836Smlaier			ip_output(m, (void *)NULL, (void *)NULL, 0,
1924145836Smlaier			    (void *)NULL, (void *)NULL);
1925145836Smlaier			PF_LOCK();
1926126261Smlaier#else /* ! __FreeBSD__ */
1927145836Smlaier			ip_output(m, (void *)NULL, (void *)NULL, 0,
1928145836Smlaier			    (void *)NULL, (void *)NULL);
1929126261Smlaier#endif
1930145836Smlaier		} else {
1931145836Smlaier			struct route		 ro;
1932145836Smlaier			struct rtentry		 rt;
1933145836Smlaier			struct ether_header	*e = (void *)ro.ro_dst.sa_data;
1934145836Smlaier
1935145836Smlaier			if (ifp == NULL) {
1936145836Smlaier				m_freem(m);
1937145836Smlaier				return;
1938145836Smlaier			}
1939145836Smlaier			rt.rt_ifp = ifp;
1940145836Smlaier			ro.ro_rt = &rt;
1941145836Smlaier			ro.ro_dst.sa_len = sizeof(ro.ro_dst);
1942145836Smlaier			ro.ro_dst.sa_family = pseudo_AF_HDRCMPLT;
1943145836Smlaier			bcopy(eh->ether_dhost, e->ether_shost, ETHER_ADDR_LEN);
1944145836Smlaier			bcopy(eh->ether_shost, e->ether_dhost, ETHER_ADDR_LEN);
1945145836Smlaier			e->ether_type = eh->ether_type;
1946145836Smlaier#ifdef __FreeBSD__
1947145836Smlaier			PF_UNLOCK();
1948145836Smlaier			/* XXX_IMPORT: later */
1949145836Smlaier			ip_output(m, (void *)NULL, &ro, 0,
1950145836Smlaier			    (void *)NULL, (void *)NULL);
1951145836Smlaier			PF_LOCK();
1952145836Smlaier#else /* ! __FreeBSD__ */
1953145836Smlaier			ip_output(m, (void *)NULL, &ro, IP_ROUTETOETHER,
1954145836Smlaier			    (void *)NULL, (void *)NULL);
1955145836Smlaier#endif
1956145836Smlaier		}
1957126258Smlaier		break;
1958126258Smlaier#endif /* INET */
1959126258Smlaier#ifdef INET6
1960126258Smlaier	case AF_INET6:
1961126258Smlaier		/* TCP checksum */
1962126258Smlaier		th->th_sum = in6_cksum(m, IPPROTO_TCP,
1963126258Smlaier		    sizeof(struct ip6_hdr), tlen);
1964126258Smlaier
1965126258Smlaier		h6->ip6_vfc |= IPV6_VERSION;
1966126258Smlaier		h6->ip6_hlim = IPV6_DEFHLIM;
1967126258Smlaier
1968127145Smlaier#ifdef __FreeBSD__
1969126261Smlaier		PF_UNLOCK();
1970126261Smlaier		ip6_output(m, NULL, NULL, 0, NULL, NULL, NULL);
1971126261Smlaier		PF_LOCK();
1972126261Smlaier#else
1973126258Smlaier		ip6_output(m, NULL, NULL, 0, NULL, NULL);
1974126261Smlaier#endif
1975126258Smlaier		break;
1976126258Smlaier#endif /* INET6 */
1977126258Smlaier	}
1978126258Smlaier}
1979126258Smlaier
1980126258Smlaiervoid
1981126258Smlaierpf_send_icmp(struct mbuf *m, u_int8_t type, u_int8_t code, sa_family_t af,
1982126258Smlaier    struct pf_rule *r)
1983126258Smlaier{
1984171168Smlaier	struct pf_mtag	*pf_mtag;
1985126258Smlaier	struct mbuf	*m0;
1986127145Smlaier#ifdef __FreeBSD__
1987126261Smlaier	struct ip *ip;
1988126261Smlaier#endif
1989126258Smlaier
1990132280Smlaier#ifdef __FreeBSD__
1991132280Smlaier	m0 = m_copypacket(m, M_DONTWAIT);
1992132280Smlaier	if (m0 == NULL)
1993132280Smlaier		return;
1994132280Smlaier#else
1995126258Smlaier	m0 = m_copy(m, 0, M_COPYALL);
1996171168Smlaier#endif
1997171168Smlaier	if ((pf_mtag = pf_get_mtag(m0)) == NULL)
1998126258Smlaier		return;
1999171168Smlaier#ifdef __FreeBSD__
2000171168Smlaier	/* XXX: revisit */
2001171168Smlaier	m0->m_flags |= M_SKIP_FIREWALL;
2002171168Smlaier#else
2003171168Smlaier	pf_mtag->flags |= PF_TAG_GENERATED;
2004132280Smlaier#endif
2005126258Smlaier
2006171168Smlaier	if (r->rtableid >= 0)
2007171168Smlaier		pf_mtag->rtableid = r->rtableid;
2008171168Smlaier
2009126258Smlaier#ifdef ALTQ
2010126258Smlaier	if (r->qid) {
2011171168Smlaier		pf_mtag->qid = r->qid;
2012171168Smlaier		/* add hints for ecn */
2013171168Smlaier		pf_mtag->af = af;
2014171168Smlaier		pf_mtag->hdr = mtod(m0, struct ip *);
2015126258Smlaier	}
2016145836Smlaier#endif /* ALTQ */
2017126258Smlaier
2018126258Smlaier	switch (af) {
2019126258Smlaier#ifdef INET
2020126258Smlaier	case AF_INET:
2021127145Smlaier#ifdef __FreeBSD__
2022126261Smlaier		/* icmp_error() expects host byte ordering */
2023126261Smlaier		ip = mtod(m0, struct ip *);
2024126261Smlaier		NTOHS(ip->ip_len);
2025126261Smlaier		NTOHS(ip->ip_off);
2026126261Smlaier		PF_UNLOCK();
2027145863Sandre		icmp_error(m0, type, code, 0, 0);
2028126261Smlaier		PF_LOCK();
2029145874Smlaier#else
2030171168Smlaier		icmp_error(m0, type, code, 0, 0);
2031126261Smlaier#endif
2032126258Smlaier		break;
2033126258Smlaier#endif /* INET */
2034126258Smlaier#ifdef INET6
2035126258Smlaier	case AF_INET6:
2036127145Smlaier#ifdef __FreeBSD__
2037126261Smlaier		PF_UNLOCK();
2038126261Smlaier#endif
2039126258Smlaier		icmp6_error(m0, type, code, 0);
2040127145Smlaier#ifdef __FreeBSD__
2041126261Smlaier		PF_LOCK();
2042126261Smlaier#endif
2043126258Smlaier		break;
2044126258Smlaier#endif /* INET6 */
2045126258Smlaier	}
2046126258Smlaier}
2047126258Smlaier
2048126258Smlaier/*
2049126258Smlaier * Return 1 if the addresses a and b match (with mask m), otherwise return 0.
2050126258Smlaier * If n is 0, they match if they are equal. If n is != 0, they match if they
2051126258Smlaier * are different.
2052126258Smlaier */
2053126258Smlaierint
2054126258Smlaierpf_match_addr(u_int8_t n, struct pf_addr *a, struct pf_addr *m,
2055126258Smlaier    struct pf_addr *b, sa_family_t af)
2056126258Smlaier{
2057126258Smlaier	int	match = 0;
2058126258Smlaier
2059126258Smlaier	switch (af) {
2060126258Smlaier#ifdef INET
2061126258Smlaier	case AF_INET:
2062126258Smlaier		if ((a->addr32[0] & m->addr32[0]) ==
2063126258Smlaier		    (b->addr32[0] & m->addr32[0]))
2064126258Smlaier			match++;
2065126258Smlaier		break;
2066126258Smlaier#endif /* INET */
2067126258Smlaier#ifdef INET6
2068126258Smlaier	case AF_INET6:
2069126258Smlaier		if (((a->addr32[0] & m->addr32[0]) ==
2070126258Smlaier		     (b->addr32[0] & m->addr32[0])) &&
2071126258Smlaier		    ((a->addr32[1] & m->addr32[1]) ==
2072126258Smlaier		     (b->addr32[1] & m->addr32[1])) &&
2073126258Smlaier		    ((a->addr32[2] & m->addr32[2]) ==
2074126258Smlaier		     (b->addr32[2] & m->addr32[2])) &&
2075126258Smlaier		    ((a->addr32[3] & m->addr32[3]) ==
2076126258Smlaier		     (b->addr32[3] & m->addr32[3])))
2077126258Smlaier			match++;
2078126258Smlaier		break;
2079126258Smlaier#endif /* INET6 */
2080126258Smlaier	}
2081126258Smlaier	if (match) {
2082126258Smlaier		if (n)
2083126258Smlaier			return (0);
2084126258Smlaier		else
2085126258Smlaier			return (1);
2086126258Smlaier	} else {
2087126258Smlaier		if (n)
2088126258Smlaier			return (1);
2089126258Smlaier		else
2090126258Smlaier			return (0);
2091126258Smlaier	}
2092126258Smlaier}
2093126258Smlaier
2094126258Smlaierint
2095126258Smlaierpf_match(u_int8_t op, u_int32_t a1, u_int32_t a2, u_int32_t p)
2096126258Smlaier{
2097126258Smlaier	switch (op) {
2098126258Smlaier	case PF_OP_IRG:
2099126258Smlaier		return ((p > a1) && (p < a2));
2100126258Smlaier	case PF_OP_XRG:
2101126258Smlaier		return ((p < a1) || (p > a2));
2102126258Smlaier	case PF_OP_RRG:
2103126258Smlaier		return ((p >= a1) && (p <= a2));
2104126258Smlaier	case PF_OP_EQ:
2105126258Smlaier		return (p == a1);
2106126258Smlaier	case PF_OP_NE:
2107126258Smlaier		return (p != a1);
2108126258Smlaier	case PF_OP_LT:
2109126258Smlaier		return (p < a1);
2110126258Smlaier	case PF_OP_LE:
2111126258Smlaier		return (p <= a1);
2112126258Smlaier	case PF_OP_GT:
2113126258Smlaier		return (p > a1);
2114126258Smlaier	case PF_OP_GE:
2115126258Smlaier		return (p >= a1);
2116126258Smlaier	}
2117126258Smlaier	return (0); /* never reached */
2118126258Smlaier}
2119126258Smlaier
2120126258Smlaierint
2121126258Smlaierpf_match_port(u_int8_t op, u_int16_t a1, u_int16_t a2, u_int16_t p)
2122126258Smlaier{
2123126258Smlaier	NTOHS(a1);
2124126258Smlaier	NTOHS(a2);
2125126258Smlaier	NTOHS(p);
2126126258Smlaier	return (pf_match(op, a1, a2, p));
2127126258Smlaier}
2128126258Smlaier
2129126258Smlaierint
2130126258Smlaierpf_match_uid(u_int8_t op, uid_t a1, uid_t a2, uid_t u)
2131126258Smlaier{
2132126258Smlaier	if (u == UID_MAX && op != PF_OP_EQ && op != PF_OP_NE)
2133126258Smlaier		return (0);
2134126258Smlaier	return (pf_match(op, a1, a2, u));
2135126258Smlaier}
2136126258Smlaier
2137126258Smlaierint
2138126258Smlaierpf_match_gid(u_int8_t op, gid_t a1, gid_t a2, gid_t g)
2139126258Smlaier{
2140126258Smlaier	if (g == GID_MAX && op != PF_OP_EQ && op != PF_OP_NE)
2141126258Smlaier		return (0);
2142126258Smlaier	return (pf_match(op, a1, a2, g));
2143126258Smlaier}
2144126258Smlaier
2145171168Smlaier#ifndef __FreeBSD__
2146171168Smlaierstruct pf_mtag *
2147171168Smlaierpf_find_mtag(struct mbuf *m)
2148126258Smlaier{
2149126258Smlaier	struct m_tag	*mtag;
2150126258Smlaier
2151171168Smlaier	if ((mtag = m_tag_find(m, PACKET_TAG_PF, NULL)) == NULL)
2152126258Smlaier		return (NULL);
2153171168Smlaier
2154171168Smlaier	return ((struct pf_mtag *)(mtag + 1));
2155126258Smlaier}
2156126258Smlaier
2157171168Smlaierstruct pf_mtag *
2158171168Smlaierpf_get_mtag(struct mbuf *m)
2159126258Smlaier{
2160171168Smlaier	struct m_tag	*mtag;
2161171168Smlaier
2162171168Smlaier	if ((mtag = m_tag_find(m, PACKET_TAG_PF, NULL)) == NULL) {
2163171168Smlaier		mtag = m_tag_get(PACKET_TAG_PF, sizeof(struct pf_mtag),
2164171168Smlaier		    M_NOWAIT);
2165171168Smlaier		if (mtag == NULL)
2166171168Smlaier			return (NULL);
2167171168Smlaier		bzero(mtag + 1, sizeof(struct pf_mtag));
2168171168Smlaier		m_tag_prepend(m, mtag);
2169126258Smlaier	}
2170126258Smlaier
2171171168Smlaier	return ((struct pf_mtag *)(mtag + 1));
2172171168Smlaier}
2173171168Smlaier#endif
2174171168Smlaier
2175171168Smlaierint
2176171168Smlaierpf_match_tag(struct mbuf *m, struct pf_rule *r, struct pf_mtag *pf_mtag,
2177171168Smlaier    int *tag)
2178171168Smlaier{
2179171168Smlaier	if (*tag == -1)
2180171168Smlaier		*tag = pf_mtag->tag;
2181171168Smlaier
2182126258Smlaier	return ((!r->match_tag_not && r->match_tag == *tag) ||
2183126258Smlaier	    (r->match_tag_not && r->match_tag != *tag));
2184126258Smlaier}
2185126258Smlaier
2186126258Smlaierint
2187171168Smlaierpf_tag_packet(struct mbuf *m, struct pf_mtag *pf_mtag, int tag, int rtableid)
2188126258Smlaier{
2189171168Smlaier	if (tag <= 0 && rtableid < 0)
2190126258Smlaier		return (0);
2191126258Smlaier
2192171168Smlaier	if (pf_mtag == NULL)
2193171168Smlaier		if ((pf_mtag = pf_get_mtag(m)) == NULL)
2194126258Smlaier			return (1);
2195171168Smlaier	if (tag > 0)
2196171168Smlaier		pf_mtag->tag = tag;
2197171168Smlaier	if (rtableid >= 0)
2198171168Smlaier		pf_mtag->rtableid = rtableid;
2199126258Smlaier
2200126258Smlaier	return (0);
2201126258Smlaier}
2202126258Smlaier
2203145836Smlaierstatic void
2204145836Smlaierpf_step_into_anchor(int *depth, struct pf_ruleset **rs, int n,
2205171168Smlaier    struct pf_rule **r, struct pf_rule **a,  int *match)
2206145836Smlaier{
2207145836Smlaier	struct pf_anchor_stackframe	*f;
2208126258Smlaier
2209171168Smlaier	(*r)->anchor->match = 0;
2210171168Smlaier	if (match)
2211171168Smlaier		*match = 0;
2212145836Smlaier	if (*depth >= sizeof(pf_anchor_stack) /
2213145836Smlaier	    sizeof(pf_anchor_stack[0])) {
2214145836Smlaier		printf("pf_step_into_anchor: stack overflow\n");
2215145836Smlaier		*r = TAILQ_NEXT(*r, entries);
2216145836Smlaier		return;
2217145836Smlaier	} else if (*depth == 0 && a != NULL)
2218145836Smlaier		*a = *r;
2219145836Smlaier	f = pf_anchor_stack + (*depth)++;
2220145836Smlaier	f->rs = *rs;
2221145836Smlaier	f->r = *r;
2222145836Smlaier	if ((*r)->anchor_wildcard) {
2223145836Smlaier		f->parent = &(*r)->anchor->children;
2224145836Smlaier		if ((f->child = RB_MIN(pf_anchor_node, f->parent)) ==
2225145836Smlaier		    NULL) {
2226145836Smlaier			*r = NULL;
2227145836Smlaier			return;
2228145836Smlaier		}
2229145836Smlaier		*rs = &f->child->ruleset;
2230145836Smlaier	} else {
2231145836Smlaier		f->parent = NULL;
2232145836Smlaier		f->child = NULL;
2233145836Smlaier		*rs = &(*r)->anchor->ruleset;
2234145836Smlaier	}
2235145836Smlaier	*r = TAILQ_FIRST((*rs)->rules[n].active.ptr);
2236145836Smlaier}
2237126258Smlaier
2238171168Smlaierint
2239145836Smlaierpf_step_out_of_anchor(int *depth, struct pf_ruleset **rs, int n,
2240171168Smlaier    struct pf_rule **r, struct pf_rule **a, int *match)
2241145836Smlaier{
2242145836Smlaier	struct pf_anchor_stackframe	*f;
2243171168Smlaier	int quick = 0;
2244145836Smlaier
2245145836Smlaier	do {
2246145836Smlaier		if (*depth <= 0)
2247145836Smlaier			break;
2248145836Smlaier		f = pf_anchor_stack + *depth - 1;
2249145836Smlaier		if (f->parent != NULL && f->child != NULL) {
2250171168Smlaier			if (f->child->match ||
2251171168Smlaier			    (match != NULL && *match)) {
2252171168Smlaier				f->r->anchor->match = 1;
2253171168Smlaier				*match = 0;
2254171168Smlaier			}
2255145836Smlaier			f->child = RB_NEXT(pf_anchor_node, f->parent, f->child);
2256145836Smlaier			if (f->child != NULL) {
2257145836Smlaier				*rs = &f->child->ruleset;
2258145836Smlaier				*r = TAILQ_FIRST((*rs)->rules[n].active.ptr);
2259145836Smlaier				if (*r == NULL)
2260145836Smlaier					continue;
2261145836Smlaier				else
2262145836Smlaier					break;
2263145836Smlaier			}
2264145836Smlaier		}
2265145836Smlaier		(*depth)--;
2266145836Smlaier		if (*depth == 0 && a != NULL)
2267145836Smlaier			*a = NULL;
2268145836Smlaier		*rs = f->rs;
2269171168Smlaier		if (f->r->anchor->match || (match  != NULL && *match))
2270171168Smlaier			quick = f->r->quick;
2271145836Smlaier		*r = TAILQ_NEXT(f->r, entries);
2272145836Smlaier	} while (*r == NULL);
2273171168Smlaier
2274171168Smlaier	return (quick);
2275145836Smlaier}
2276145836Smlaier
2277126258Smlaier#ifdef INET6
2278126258Smlaiervoid
2279126258Smlaierpf_poolmask(struct pf_addr *naddr, struct pf_addr *raddr,
2280126258Smlaier    struct pf_addr *rmask, struct pf_addr *saddr, sa_family_t af)
2281126258Smlaier{
2282126258Smlaier	switch (af) {
2283126258Smlaier#ifdef INET
2284126258Smlaier	case AF_INET:
2285126258Smlaier		naddr->addr32[0] = (raddr->addr32[0] & rmask->addr32[0]) |
2286126258Smlaier		((rmask->addr32[0] ^ 0xffffffff ) & saddr->addr32[0]);
2287126258Smlaier		break;
2288126258Smlaier#endif /* INET */
2289126258Smlaier	case AF_INET6:
2290126258Smlaier		naddr->addr32[0] = (raddr->addr32[0] & rmask->addr32[0]) |
2291126258Smlaier		((rmask->addr32[0] ^ 0xffffffff ) & saddr->addr32[0]);
2292126258Smlaier		naddr->addr32[1] = (raddr->addr32[1] & rmask->addr32[1]) |
2293126258Smlaier		((rmask->addr32[1] ^ 0xffffffff ) & saddr->addr32[1]);
2294126258Smlaier		naddr->addr32[2] = (raddr->addr32[2] & rmask->addr32[2]) |
2295126258Smlaier		((rmask->addr32[2] ^ 0xffffffff ) & saddr->addr32[2]);
2296126258Smlaier		naddr->addr32[3] = (raddr->addr32[3] & rmask->addr32[3]) |
2297126258Smlaier		((rmask->addr32[3] ^ 0xffffffff ) & saddr->addr32[3]);
2298126258Smlaier		break;
2299126258Smlaier	}
2300126258Smlaier}
2301126258Smlaier
2302126258Smlaiervoid
2303130613Smlaierpf_addr_inc(struct pf_addr *addr, sa_family_t af)
2304126258Smlaier{
2305126258Smlaier	switch (af) {
2306126258Smlaier#ifdef INET
2307126258Smlaier	case AF_INET:
2308126258Smlaier		addr->addr32[0] = htonl(ntohl(addr->addr32[0]) + 1);
2309126258Smlaier		break;
2310126258Smlaier#endif /* INET */
2311126258Smlaier	case AF_INET6:
2312126258Smlaier		if (addr->addr32[3] == 0xffffffff) {
2313126258Smlaier			addr->addr32[3] = 0;
2314126258Smlaier			if (addr->addr32[2] == 0xffffffff) {
2315126258Smlaier				addr->addr32[2] = 0;
2316126258Smlaier				if (addr->addr32[1] == 0xffffffff) {
2317126258Smlaier					addr->addr32[1] = 0;
2318126258Smlaier					addr->addr32[0] =
2319126258Smlaier					    htonl(ntohl(addr->addr32[0]) + 1);
2320126258Smlaier				} else
2321126258Smlaier					addr->addr32[1] =
2322126258Smlaier					    htonl(ntohl(addr->addr32[1]) + 1);
2323126258Smlaier			} else
2324126258Smlaier				addr->addr32[2] =
2325126258Smlaier				    htonl(ntohl(addr->addr32[2]) + 1);
2326126258Smlaier		} else
2327126258Smlaier			addr->addr32[3] =
2328126258Smlaier			    htonl(ntohl(addr->addr32[3]) + 1);
2329126258Smlaier		break;
2330126258Smlaier	}
2331126258Smlaier}
2332126258Smlaier#endif /* INET6 */
2333126258Smlaier
2334126258Smlaier#define mix(a,b,c) \
2335126258Smlaier	do {					\
2336126258Smlaier		a -= b; a -= c; a ^= (c >> 13);	\
2337126258Smlaier		b -= c; b -= a; b ^= (a << 8);	\
2338126258Smlaier		c -= a; c -= b; c ^= (b >> 13);	\
2339126258Smlaier		a -= b; a -= c; a ^= (c >> 12);	\
2340126258Smlaier		b -= c; b -= a; b ^= (a << 16);	\
2341126258Smlaier		c -= a; c -= b; c ^= (b >> 5);	\
2342126258Smlaier		a -= b; a -= c; a ^= (c >> 3);	\
2343126258Smlaier		b -= c; b -= a; b ^= (a << 10);	\
2344126258Smlaier		c -= a; c -= b; c ^= (b >> 15);	\
2345126258Smlaier	} while (0)
2346126258Smlaier
2347126258Smlaier/*
2348126258Smlaier * hash function based on bridge_hash in if_bridge.c
2349126258Smlaier */
2350126258Smlaiervoid
2351126258Smlaierpf_hash(struct pf_addr *inaddr, struct pf_addr *hash,
2352126258Smlaier    struct pf_poolhashkey *key, sa_family_t af)
2353126258Smlaier{
2354126258Smlaier	u_int32_t	a = 0x9e3779b9, b = 0x9e3779b9, c = key->key32[0];
2355126258Smlaier
2356126258Smlaier	switch (af) {
2357126258Smlaier#ifdef INET
2358126258Smlaier	case AF_INET:
2359126258Smlaier		a += inaddr->addr32[0];
2360126258Smlaier		b += key->key32[1];
2361126258Smlaier		mix(a, b, c);
2362126258Smlaier		hash->addr32[0] = c + key->key32[2];
2363126258Smlaier		break;
2364126258Smlaier#endif /* INET */
2365126258Smlaier#ifdef INET6
2366126258Smlaier	case AF_INET6:
2367126258Smlaier		a += inaddr->addr32[0];
2368126258Smlaier		b += inaddr->addr32[2];
2369126258Smlaier		mix(a, b, c);
2370126258Smlaier		hash->addr32[0] = c;
2371126258Smlaier		a += inaddr->addr32[1];
2372126258Smlaier		b += inaddr->addr32[3];
2373126258Smlaier		c += key->key32[1];
2374126258Smlaier		mix(a, b, c);
2375126258Smlaier		hash->addr32[1] = c;
2376126258Smlaier		a += inaddr->addr32[2];
2377126258Smlaier		b += inaddr->addr32[1];
2378126258Smlaier		c += key->key32[2];
2379126258Smlaier		mix(a, b, c);
2380126258Smlaier		hash->addr32[2] = c;
2381126258Smlaier		a += inaddr->addr32[3];
2382126258Smlaier		b += inaddr->addr32[0];
2383126258Smlaier		c += key->key32[3];
2384126258Smlaier		mix(a, b, c);
2385126258Smlaier		hash->addr32[3] = c;
2386126258Smlaier		break;
2387126258Smlaier#endif /* INET6 */
2388126258Smlaier	}
2389126258Smlaier}
2390126258Smlaier
2391126258Smlaierint
2392130613Smlaierpf_map_addr(sa_family_t af, struct pf_rule *r, struct pf_addr *saddr,
2393130613Smlaier    struct pf_addr *naddr, struct pf_addr *init_addr, struct pf_src_node **sn)
2394126258Smlaier{
2395126258Smlaier	unsigned char		 hash[16];
2396130613Smlaier	struct pf_pool		*rpool = &r->rpool;
2397130613Smlaier	struct pf_addr		*raddr = &rpool->cur->addr.v.a.addr;
2398130613Smlaier	struct pf_addr		*rmask = &rpool->cur->addr.v.a.mask;
2399126258Smlaier	struct pf_pooladdr	*acur = rpool->cur;
2400130613Smlaier	struct pf_src_node	 k;
2401126258Smlaier
2402130613Smlaier	if (*sn == NULL && r->rpool.opts & PF_POOL_STICKYADDR &&
2403130613Smlaier	    (r->rpool.opts & PF_POOL_TYPEMASK) != PF_POOL_NONE) {
2404130613Smlaier		k.af = af;
2405130613Smlaier		PF_ACPY(&k.addr, saddr, af);
2406130613Smlaier		if (r->rule_flag & PFRULE_RULESRCTRACK ||
2407130613Smlaier		    r->rpool.opts & PF_POOL_STICKYADDR)
2408130613Smlaier			k.rule.ptr = r;
2409130613Smlaier		else
2410130613Smlaier			k.rule.ptr = NULL;
2411130613Smlaier		pf_status.scounters[SCNT_SRC_NODE_SEARCH]++;
2412130613Smlaier		*sn = RB_FIND(pf_src_tree, &tree_src_tracking, &k);
2413130613Smlaier		if (*sn != NULL && !PF_AZERO(&(*sn)->raddr, af)) {
2414130613Smlaier			PF_ACPY(naddr, &(*sn)->raddr, af);
2415130613Smlaier			if (pf_status.debug >= PF_DEBUG_MISC) {
2416130613Smlaier				printf("pf_map_addr: src tracking maps ");
2417130613Smlaier				pf_print_host(&k.addr, 0, af);
2418130613Smlaier				printf(" to ");
2419130613Smlaier				pf_print_host(naddr, 0, af);
2420130613Smlaier				printf("\n");
2421130613Smlaier			}
2422130613Smlaier			return (0);
2423130613Smlaier		}
2424130613Smlaier	}
2425130613Smlaier
2426126258Smlaier	if (rpool->cur->addr.type == PF_ADDR_NOROUTE)
2427126258Smlaier		return (1);
2428130613Smlaier	if (rpool->cur->addr.type == PF_ADDR_DYNIFTL) {
2429145836Smlaier		switch (af) {
2430145836Smlaier#ifdef INET
2431145836Smlaier		case AF_INET:
2432130613Smlaier			if (rpool->cur->addr.p.dyn->pfid_acnt4 < 1 &&
2433130613Smlaier			    (rpool->opts & PF_POOL_TYPEMASK) !=
2434130613Smlaier			    PF_POOL_ROUNDROBIN)
2435130613Smlaier				return (1);
2436130613Smlaier			 raddr = &rpool->cur->addr.p.dyn->pfid_addr4;
2437130613Smlaier			 rmask = &rpool->cur->addr.p.dyn->pfid_mask4;
2438145836Smlaier			break;
2439145836Smlaier#endif /* INET */
2440145836Smlaier#ifdef INET6
2441145836Smlaier		case AF_INET6:
2442130613Smlaier			if (rpool->cur->addr.p.dyn->pfid_acnt6 < 1 &&
2443130613Smlaier			    (rpool->opts & PF_POOL_TYPEMASK) !=
2444130613Smlaier			    PF_POOL_ROUNDROBIN)
2445130613Smlaier				return (1);
2446130613Smlaier			raddr = &rpool->cur->addr.p.dyn->pfid_addr6;
2447130613Smlaier			rmask = &rpool->cur->addr.p.dyn->pfid_mask6;
2448145836Smlaier			break;
2449145836Smlaier#endif /* INET6 */
2450130613Smlaier		}
2451130613Smlaier	} else if (rpool->cur->addr.type == PF_ADDR_TABLE) {
2452126258Smlaier		if ((rpool->opts & PF_POOL_TYPEMASK) != PF_POOL_ROUNDROBIN)
2453126258Smlaier			return (1); /* unsupported */
2454126258Smlaier	} else {
2455126258Smlaier		raddr = &rpool->cur->addr.v.a.addr;
2456126258Smlaier		rmask = &rpool->cur->addr.v.a.mask;
2457126258Smlaier	}
2458126258Smlaier
2459126258Smlaier	switch (rpool->opts & PF_POOL_TYPEMASK) {
2460126258Smlaier	case PF_POOL_NONE:
2461126258Smlaier		PF_ACPY(naddr, raddr, af);
2462126258Smlaier		break;
2463126258Smlaier	case PF_POOL_BITMASK:
2464126258Smlaier		PF_POOLMASK(naddr, raddr, rmask, saddr, af);
2465126258Smlaier		break;
2466126258Smlaier	case PF_POOL_RANDOM:
2467126258Smlaier		if (init_addr != NULL && PF_AZERO(init_addr, af)) {
2468126258Smlaier			switch (af) {
2469126258Smlaier#ifdef INET
2470126258Smlaier			case AF_INET:
2471145836Smlaier				rpool->counter.addr32[0] = htonl(arc4random());
2472126258Smlaier				break;
2473126258Smlaier#endif /* INET */
2474126258Smlaier#ifdef INET6
2475126258Smlaier			case AF_INET6:
2476126258Smlaier				if (rmask->addr32[3] != 0xffffffff)
2477145836Smlaier					rpool->counter.addr32[3] =
2478145836Smlaier					    htonl(arc4random());
2479126258Smlaier				else
2480126258Smlaier					break;
2481126258Smlaier				if (rmask->addr32[2] != 0xffffffff)
2482145836Smlaier					rpool->counter.addr32[2] =
2483145836Smlaier					    htonl(arc4random());
2484126258Smlaier				else
2485126258Smlaier					break;
2486126258Smlaier				if (rmask->addr32[1] != 0xffffffff)
2487145836Smlaier					rpool->counter.addr32[1] =
2488145836Smlaier					    htonl(arc4random());
2489126258Smlaier				else
2490126258Smlaier					break;
2491126258Smlaier				if (rmask->addr32[0] != 0xffffffff)
2492145836Smlaier					rpool->counter.addr32[0] =
2493145836Smlaier					    htonl(arc4random());
2494126258Smlaier				break;
2495126258Smlaier#endif /* INET6 */
2496126258Smlaier			}
2497126258Smlaier			PF_POOLMASK(naddr, raddr, rmask, &rpool->counter, af);
2498126258Smlaier			PF_ACPY(init_addr, naddr, af);
2499126258Smlaier
2500126258Smlaier		} else {
2501126258Smlaier			PF_AINC(&rpool->counter, af);
2502126258Smlaier			PF_POOLMASK(naddr, raddr, rmask, &rpool->counter, af);
2503126258Smlaier		}
2504126258Smlaier		break;
2505126258Smlaier	case PF_POOL_SRCHASH:
2506126258Smlaier		pf_hash(saddr, (struct pf_addr *)&hash, &rpool->key, af);
2507126258Smlaier		PF_POOLMASK(naddr, raddr, rmask, (struct pf_addr *)&hash, af);
2508126258Smlaier		break;
2509126258Smlaier	case PF_POOL_ROUNDROBIN:
2510126258Smlaier		if (rpool->cur->addr.type == PF_ADDR_TABLE) {
2511126258Smlaier			if (!pfr_pool_get(rpool->cur->addr.p.tbl,
2512126258Smlaier			    &rpool->tblidx, &rpool->counter,
2513126258Smlaier			    &raddr, &rmask, af))
2514126258Smlaier				goto get_addr;
2515130613Smlaier		} else if (rpool->cur->addr.type == PF_ADDR_DYNIFTL) {
2516130613Smlaier			if (!pfr_pool_get(rpool->cur->addr.p.dyn->pfid_kt,
2517130613Smlaier			    &rpool->tblidx, &rpool->counter,
2518130613Smlaier			    &raddr, &rmask, af))
2519130613Smlaier				goto get_addr;
2520126258Smlaier		} else if (pf_match_addr(0, raddr, rmask, &rpool->counter, af))
2521126258Smlaier			goto get_addr;
2522126258Smlaier
2523126258Smlaier	try_next:
2524126258Smlaier		if ((rpool->cur = TAILQ_NEXT(rpool->cur, entries)) == NULL)
2525126258Smlaier			rpool->cur = TAILQ_FIRST(&rpool->list);
2526126258Smlaier		if (rpool->cur->addr.type == PF_ADDR_TABLE) {
2527126258Smlaier			rpool->tblidx = -1;
2528126258Smlaier			if (pfr_pool_get(rpool->cur->addr.p.tbl,
2529126258Smlaier			    &rpool->tblidx, &rpool->counter,
2530126258Smlaier			    &raddr, &rmask, af)) {
2531130613Smlaier				/* table contains no address of type 'af' */
2532126258Smlaier				if (rpool->cur != acur)
2533126258Smlaier					goto try_next;
2534126258Smlaier				return (1);
2535126258Smlaier			}
2536130613Smlaier		} else if (rpool->cur->addr.type == PF_ADDR_DYNIFTL) {
2537130613Smlaier			rpool->tblidx = -1;
2538130613Smlaier			if (pfr_pool_get(rpool->cur->addr.p.dyn->pfid_kt,
2539130613Smlaier			    &rpool->tblidx, &rpool->counter,
2540130613Smlaier			    &raddr, &rmask, af)) {
2541130613Smlaier				/* table contains no address of type 'af' */
2542130613Smlaier				if (rpool->cur != acur)
2543130613Smlaier					goto try_next;
2544130613Smlaier				return (1);
2545130613Smlaier			}
2546126258Smlaier		} else {
2547126258Smlaier			raddr = &rpool->cur->addr.v.a.addr;
2548126258Smlaier			rmask = &rpool->cur->addr.v.a.mask;
2549126258Smlaier			PF_ACPY(&rpool->counter, raddr, af);
2550126258Smlaier		}
2551126258Smlaier
2552126258Smlaier	get_addr:
2553126258Smlaier		PF_ACPY(naddr, &rpool->counter, af);
2554139045Sdhartmei		if (init_addr != NULL && PF_AZERO(init_addr, af))
2555139045Sdhartmei			PF_ACPY(init_addr, naddr, af);
2556126258Smlaier		PF_AINC(&rpool->counter, af);
2557126258Smlaier		break;
2558126258Smlaier	}
2559130613Smlaier	if (*sn != NULL)
2560130613Smlaier		PF_ACPY(&(*sn)->raddr, naddr, af);
2561126258Smlaier
2562126258Smlaier	if (pf_status.debug >= PF_DEBUG_MISC &&
2563126258Smlaier	    (rpool->opts & PF_POOL_TYPEMASK) != PF_POOL_NONE) {
2564130613Smlaier		printf("pf_map_addr: selected address ");
2565126258Smlaier		pf_print_host(naddr, 0, af);
2566126258Smlaier		printf("\n");
2567126258Smlaier	}
2568126258Smlaier
2569126258Smlaier	return (0);
2570126258Smlaier}
2571126258Smlaier
2572126258Smlaierint
2573130613Smlaierpf_get_sport(sa_family_t af, u_int8_t proto, struct pf_rule *r,
2574126258Smlaier    struct pf_addr *saddr, struct pf_addr *daddr, u_int16_t dport,
2575130613Smlaier    struct pf_addr *naddr, u_int16_t *nport, u_int16_t low, u_int16_t high,
2576130613Smlaier    struct pf_src_node **sn)
2577126258Smlaier{
2578171168Smlaier	struct pf_state_cmp	key;
2579126258Smlaier	struct pf_addr		init_addr;
2580126258Smlaier	u_int16_t		cut;
2581126258Smlaier
2582126258Smlaier	bzero(&init_addr, sizeof(init_addr));
2583130613Smlaier	if (pf_map_addr(af, r, saddr, naddr, &init_addr, sn))
2584126258Smlaier		return (1);
2585126258Smlaier
2586149884Smlaier	if (proto == IPPROTO_ICMP) {
2587149884Smlaier		low = 1;
2588149884Smlaier		high = 65535;
2589149884Smlaier	}
2590149884Smlaier
2591126258Smlaier	do {
2592126258Smlaier		key.af = af;
2593126258Smlaier		key.proto = proto;
2594130613Smlaier		PF_ACPY(&key.ext.addr, daddr, key.af);
2595130613Smlaier		PF_ACPY(&key.gwy.addr, naddr, key.af);
2596130613Smlaier		key.ext.port = dport;
2597126258Smlaier
2598126258Smlaier		/*
2599126258Smlaier		 * port search; start random, step;
2600126258Smlaier		 * similar 2 portloop in in_pcbbind
2601126258Smlaier		 */
2602149884Smlaier		if (!(proto == IPPROTO_TCP || proto == IPPROTO_UDP ||
2603149884Smlaier		    proto == IPPROTO_ICMP)) {
2604139045Sdhartmei			key.gwy.port = dport;
2605130613Smlaier			if (pf_find_state_all(&key, PF_EXT_GWY, NULL) == NULL)
2606126258Smlaier				return (0);
2607126258Smlaier		} else if (low == 0 && high == 0) {
2608130613Smlaier			key.gwy.port = *nport;
2609130613Smlaier			if (pf_find_state_all(&key, PF_EXT_GWY, NULL) == NULL)
2610126258Smlaier				return (0);
2611126258Smlaier		} else if (low == high) {
2612130613Smlaier			key.gwy.port = htons(low);
2613130613Smlaier			if (pf_find_state_all(&key, PF_EXT_GWY, NULL) == NULL) {
2614126258Smlaier				*nport = htons(low);
2615126258Smlaier				return (0);
2616126258Smlaier			}
2617126258Smlaier		} else {
2618126258Smlaier			u_int16_t tmp;
2619126258Smlaier
2620126258Smlaier			if (low > high) {
2621126258Smlaier				tmp = low;
2622126258Smlaier				low = high;
2623126258Smlaier				high = tmp;
2624126258Smlaier			}
2625126258Smlaier			/* low < high */
2626145836Smlaier			cut = htonl(arc4random()) % (1 + high - low) + low;
2627126258Smlaier			/* low <= cut <= high */
2628126258Smlaier			for (tmp = cut; tmp <= high; ++(tmp)) {
2629130613Smlaier				key.gwy.port = htons(tmp);
2630130613Smlaier				if (pf_find_state_all(&key, PF_EXT_GWY, NULL) ==
2631126258Smlaier				    NULL) {
2632126258Smlaier					*nport = htons(tmp);
2633126258Smlaier					return (0);
2634126258Smlaier				}
2635126258Smlaier			}
2636126258Smlaier			for (tmp = cut - 1; tmp >= low; --(tmp)) {
2637130613Smlaier				key.gwy.port = htons(tmp);
2638130613Smlaier				if (pf_find_state_all(&key, PF_EXT_GWY, NULL) ==
2639126258Smlaier				    NULL) {
2640126258Smlaier					*nport = htons(tmp);
2641126258Smlaier					return (0);
2642126258Smlaier				}
2643126258Smlaier			}
2644126258Smlaier		}
2645126258Smlaier
2646130613Smlaier		switch (r->rpool.opts & PF_POOL_TYPEMASK) {
2647126258Smlaier		case PF_POOL_RANDOM:
2648126258Smlaier		case PF_POOL_ROUNDROBIN:
2649130613Smlaier			if (pf_map_addr(af, r, saddr, naddr, &init_addr, sn))
2650126258Smlaier				return (1);
2651126258Smlaier			break;
2652126258Smlaier		case PF_POOL_NONE:
2653126258Smlaier		case PF_POOL_SRCHASH:
2654126258Smlaier		case PF_POOL_BITMASK:
2655126258Smlaier		default:
2656126258Smlaier			return (1);
2657126258Smlaier		}
2658126258Smlaier	} while (! PF_AEQ(&init_addr, naddr, af) );
2659126258Smlaier
2660126258Smlaier	return (1);					/* none available */
2661126258Smlaier}
2662126258Smlaier
2663126258Smlaierstruct pf_rule *
2664126258Smlaierpf_match_translation(struct pf_pdesc *pd, struct mbuf *m, int off,
2665130613Smlaier    int direction, struct pfi_kif *kif, struct pf_addr *saddr, u_int16_t sport,
2666126258Smlaier    struct pf_addr *daddr, u_int16_t dport, int rs_num)
2667126258Smlaier{
2668145836Smlaier	struct pf_rule		*r, *rm = NULL;
2669126258Smlaier	struct pf_ruleset	*ruleset = NULL;
2670145836Smlaier	int			 tag = -1;
2671171168Smlaier	int			 rtableid = -1;
2672145836Smlaier	int			 asd = 0;
2673126258Smlaier
2674126258Smlaier	r = TAILQ_FIRST(pf_main_ruleset.rules[rs_num].active.ptr);
2675126258Smlaier	while (r && rm == NULL) {
2676126258Smlaier		struct pf_rule_addr	*src = NULL, *dst = NULL;
2677126258Smlaier		struct pf_addr_wrap	*xdst = NULL;
2678126258Smlaier
2679126258Smlaier		if (r->action == PF_BINAT && direction == PF_IN) {
2680126258Smlaier			src = &r->dst;
2681126258Smlaier			if (r->rpool.cur != NULL)
2682126258Smlaier				xdst = &r->rpool.cur->addr;
2683126258Smlaier		} else {
2684126258Smlaier			src = &r->src;
2685126258Smlaier			dst = &r->dst;
2686126258Smlaier		}
2687126258Smlaier
2688126258Smlaier		r->evaluations++;
2689171168Smlaier		if (pfi_kif_match(r->kif, kif) == r->ifnot)
2690126258Smlaier			r = r->skip[PF_SKIP_IFP].ptr;
2691126258Smlaier		else if (r->direction && r->direction != direction)
2692126258Smlaier			r = r->skip[PF_SKIP_DIR].ptr;
2693126258Smlaier		else if (r->af && r->af != pd->af)
2694126258Smlaier			r = r->skip[PF_SKIP_AF].ptr;
2695126258Smlaier		else if (r->proto && r->proto != pd->proto)
2696126258Smlaier			r = r->skip[PF_SKIP_PROTO].ptr;
2697171168Smlaier		else if (PF_MISMATCHAW(&src->addr, saddr, pd->af,
2698171168Smlaier		    src->neg, kif))
2699126258Smlaier			r = r->skip[src == &r->src ? PF_SKIP_SRC_ADDR :
2700126258Smlaier			    PF_SKIP_DST_ADDR].ptr;
2701126258Smlaier		else if (src->port_op && !pf_match_port(src->port_op,
2702126258Smlaier		    src->port[0], src->port[1], sport))
2703126258Smlaier			r = r->skip[src == &r->src ? PF_SKIP_SRC_PORT :
2704126258Smlaier			    PF_SKIP_DST_PORT].ptr;
2705126258Smlaier		else if (dst != NULL &&
2706171168Smlaier		    PF_MISMATCHAW(&dst->addr, daddr, pd->af, dst->neg, NULL))
2707126258Smlaier			r = r->skip[PF_SKIP_DST_ADDR].ptr;
2708171168Smlaier		else if (xdst != NULL && PF_MISMATCHAW(xdst, daddr, pd->af,
2709171168Smlaier		    0, NULL))
2710126258Smlaier			r = TAILQ_NEXT(r, entries);
2711126258Smlaier		else if (dst != NULL && dst->port_op &&
2712126258Smlaier		    !pf_match_port(dst->port_op, dst->port[0],
2713126258Smlaier		    dst->port[1], dport))
2714126258Smlaier			r = r->skip[PF_SKIP_DST_PORT].ptr;
2715171168Smlaier		else if (r->match_tag && !pf_match_tag(m, r, pd->pf_mtag, &tag))
2716145836Smlaier			r = TAILQ_NEXT(r, entries);
2717126258Smlaier		else if (r->os_fingerprint != PF_OSFP_ANY && (pd->proto !=
2718126258Smlaier		    IPPROTO_TCP || !pf_osfp_match(pf_osfp_fingerprint(pd, m,
2719126258Smlaier		    off, pd->hdr.tcp), r->os_fingerprint)))
2720126258Smlaier			r = TAILQ_NEXT(r, entries);
2721145836Smlaier		else {
2722145836Smlaier			if (r->tag)
2723145836Smlaier				tag = r->tag;
2724171168Smlaier			if (r->rtableid >= 0)
2725171168Smlaier				rtableid = r->rtableid;
2726145836Smlaier			if (r->anchor == NULL) {
2727126258Smlaier				rm = r;
2728145836Smlaier			} else
2729171168Smlaier				pf_step_into_anchor(&asd, &ruleset, rs_num,
2730171168Smlaier				    &r, NULL, NULL);
2731145836Smlaier		}
2732145836Smlaier		if (r == NULL)
2733171168Smlaier			pf_step_out_of_anchor(&asd, &ruleset, rs_num, &r,
2734171168Smlaier			    NULL, NULL);
2735126258Smlaier	}
2736171168Smlaier	if (pf_tag_packet(m, pd->pf_mtag, tag, rtableid))
2737145836Smlaier		return (NULL);
2738126258Smlaier	if (rm != NULL && (rm->action == PF_NONAT ||
2739126258Smlaier	    rm->action == PF_NORDR || rm->action == PF_NOBINAT))
2740126258Smlaier		return (NULL);
2741126258Smlaier	return (rm);
2742126258Smlaier}
2743126258Smlaier
2744126258Smlaierstruct pf_rule *
2745126258Smlaierpf_get_translation(struct pf_pdesc *pd, struct mbuf *m, int off, int direction,
2746130613Smlaier    struct pfi_kif *kif, struct pf_src_node **sn,
2747126258Smlaier    struct pf_addr *saddr, u_int16_t sport,
2748126258Smlaier    struct pf_addr *daddr, u_int16_t dport,
2749126258Smlaier    struct pf_addr *naddr, u_int16_t *nport)
2750126258Smlaier{
2751126258Smlaier	struct pf_rule	*r = NULL;
2752126258Smlaier
2753126258Smlaier	if (direction == PF_OUT) {
2754130613Smlaier		r = pf_match_translation(pd, m, off, direction, kif, saddr,
2755126258Smlaier		    sport, daddr, dport, PF_RULESET_BINAT);
2756126258Smlaier		if (r == NULL)
2757130613Smlaier			r = pf_match_translation(pd, m, off, direction, kif,
2758126258Smlaier			    saddr, sport, daddr, dport, PF_RULESET_NAT);
2759126258Smlaier	} else {
2760130613Smlaier		r = pf_match_translation(pd, m, off, direction, kif, saddr,
2761126258Smlaier		    sport, daddr, dport, PF_RULESET_RDR);
2762126258Smlaier		if (r == NULL)
2763130613Smlaier			r = pf_match_translation(pd, m, off, direction, kif,
2764126258Smlaier			    saddr, sport, daddr, dport, PF_RULESET_BINAT);
2765126258Smlaier	}
2766126258Smlaier
2767126258Smlaier	if (r != NULL) {
2768126258Smlaier		switch (r->action) {
2769126258Smlaier		case PF_NONAT:
2770126258Smlaier		case PF_NOBINAT:
2771126258Smlaier		case PF_NORDR:
2772126258Smlaier			return (NULL);
2773126258Smlaier		case PF_NAT:
2774130613Smlaier			if (pf_get_sport(pd->af, pd->proto, r, saddr,
2775126258Smlaier			    daddr, dport, naddr, nport, r->rpool.proxy_port[0],
2776130613Smlaier			    r->rpool.proxy_port[1], sn)) {
2777126258Smlaier				DPFPRINTF(PF_DEBUG_MISC,
2778126258Smlaier				    ("pf: NAT proxy port allocation "
2779126258Smlaier				    "(%u-%u) failed\n",
2780126258Smlaier				    r->rpool.proxy_port[0],
2781126258Smlaier				    r->rpool.proxy_port[1]));
2782126258Smlaier				return (NULL);
2783126258Smlaier			}
2784126258Smlaier			break;
2785126258Smlaier		case PF_BINAT:
2786126258Smlaier			switch (direction) {
2787126258Smlaier			case PF_OUT:
2788130613Smlaier				if (r->rpool.cur->addr.type == PF_ADDR_DYNIFTL){
2789145836Smlaier					switch (pd->af) {
2790145836Smlaier#ifdef INET
2791145836Smlaier					case AF_INET:
2792130613Smlaier						if (r->rpool.cur->addr.p.dyn->
2793130613Smlaier						    pfid_acnt4 < 1)
2794130613Smlaier							return (NULL);
2795130613Smlaier						PF_POOLMASK(naddr,
2796130613Smlaier						    &r->rpool.cur->addr.p.dyn->
2797130613Smlaier						    pfid_addr4,
2798130613Smlaier						    &r->rpool.cur->addr.p.dyn->
2799130613Smlaier						    pfid_mask4,
2800130613Smlaier						    saddr, AF_INET);
2801145836Smlaier						break;
2802145836Smlaier#endif /* INET */
2803145836Smlaier#ifdef INET6
2804145836Smlaier					case AF_INET6:
2805130613Smlaier						if (r->rpool.cur->addr.p.dyn->
2806130613Smlaier						    pfid_acnt6 < 1)
2807130613Smlaier							return (NULL);
2808130613Smlaier						PF_POOLMASK(naddr,
2809130613Smlaier						    &r->rpool.cur->addr.p.dyn->
2810130613Smlaier						    pfid_addr6,
2811130613Smlaier						    &r->rpool.cur->addr.p.dyn->
2812130613Smlaier						    pfid_mask6,
2813130613Smlaier						    saddr, AF_INET6);
2814145836Smlaier						break;
2815145836Smlaier#endif /* INET6 */
2816130613Smlaier					}
2817130613Smlaier				} else
2818126258Smlaier					PF_POOLMASK(naddr,
2819126258Smlaier					    &r->rpool.cur->addr.v.a.addr,
2820126258Smlaier					    &r->rpool.cur->addr.v.a.mask,
2821126258Smlaier					    saddr, pd->af);
2822126258Smlaier				break;
2823126258Smlaier			case PF_IN:
2824138041Sdhartmei				if (r->src.addr.type == PF_ADDR_DYNIFTL) {
2825145836Smlaier					switch (pd->af) {
2826145836Smlaier#ifdef INET
2827145836Smlaier					case AF_INET:
2828130613Smlaier						if (r->src.addr.p.dyn->
2829130613Smlaier						    pfid_acnt4 < 1)
2830130613Smlaier							return (NULL);
2831130613Smlaier						PF_POOLMASK(naddr,
2832130613Smlaier						    &r->src.addr.p.dyn->
2833130613Smlaier						    pfid_addr4,
2834130613Smlaier						    &r->src.addr.p.dyn->
2835130613Smlaier						    pfid_mask4,
2836130613Smlaier						    daddr, AF_INET);
2837145836Smlaier						break;
2838145836Smlaier#endif /* INET */
2839145836Smlaier#ifdef INET6
2840145836Smlaier					case AF_INET6:
2841130613Smlaier						if (r->src.addr.p.dyn->
2842130613Smlaier						    pfid_acnt6 < 1)
2843130613Smlaier							return (NULL);
2844130613Smlaier						PF_POOLMASK(naddr,
2845130613Smlaier						    &r->src.addr.p.dyn->
2846130613Smlaier						    pfid_addr6,
2847130613Smlaier						    &r->src.addr.p.dyn->
2848130613Smlaier						    pfid_mask6,
2849130613Smlaier						    daddr, AF_INET6);
2850145836Smlaier						break;
2851145836Smlaier#endif /* INET6 */
2852130613Smlaier					}
2853130613Smlaier				} else
2854126258Smlaier					PF_POOLMASK(naddr,
2855126258Smlaier					    &r->src.addr.v.a.addr,
2856126261Smlaier					    &r->src.addr.v.a.mask, daddr,
2857126258Smlaier					    pd->af);
2858126258Smlaier				break;
2859126258Smlaier			}
2860126258Smlaier			break;
2861126258Smlaier		case PF_RDR: {
2862140518Sdhartmei			if (pf_map_addr(pd->af, r, saddr, naddr, NULL, sn))
2863126258Smlaier				return (NULL);
2864149884Smlaier			if ((r->rpool.opts & PF_POOL_TYPEMASK) ==
2865149884Smlaier			    PF_POOL_BITMASK)
2866149884Smlaier				PF_POOLMASK(naddr, naddr,
2867149884Smlaier				    &r->rpool.cur->addr.v.a.mask, daddr,
2868149884Smlaier				    pd->af);
2869126258Smlaier
2870126258Smlaier			if (r->rpool.proxy_port[1]) {
2871126258Smlaier				u_int32_t	tmp_nport;
2872126258Smlaier
2873126258Smlaier				tmp_nport = ((ntohs(dport) -
2874126258Smlaier				    ntohs(r->dst.port[0])) %
2875126258Smlaier				    (r->rpool.proxy_port[1] -
2876126258Smlaier				    r->rpool.proxy_port[0] + 1)) +
2877126258Smlaier				    r->rpool.proxy_port[0];
2878126258Smlaier
2879126258Smlaier				/* wrap around if necessary */
2880126258Smlaier				if (tmp_nport > 65535)
2881126258Smlaier					tmp_nport -= 65535;
2882126258Smlaier				*nport = htons((u_int16_t)tmp_nport);
2883126258Smlaier			} else if (r->rpool.proxy_port[0])
2884126258Smlaier				*nport = htons(r->rpool.proxy_port[0]);
2885126258Smlaier			break;
2886126258Smlaier		}
2887126258Smlaier		default:
2888126258Smlaier			return (NULL);
2889126258Smlaier		}
2890126258Smlaier	}
2891126258Smlaier
2892126258Smlaier	return (r);
2893126258Smlaier}
2894126258Smlaier
2895126258Smlaierint
2896135920Smlaier#ifdef __FreeBSD__
2897171168Smlaierpf_socket_lookup(int direction, struct pf_pdesc *pd, struct inpcb *inp_arg)
2898135920Smlaier#else
2899171168Smlaierpf_socket_lookup(int direction, struct pf_pdesc *pd)
2900135920Smlaier#endif
2901126258Smlaier{
2902126258Smlaier	struct pf_addr		*saddr, *daddr;
2903126258Smlaier	u_int16_t		 sport, dport;
2904127145Smlaier#ifdef __FreeBSD__
2905126261Smlaier	struct inpcbinfo	*pi;
2906126261Smlaier#else
2907126258Smlaier	struct inpcbtable	*tb;
2908126261Smlaier#endif
2909126258Smlaier	struct inpcb		*inp;
2910126258Smlaier
2911171168Smlaier	if (pd == NULL)
2912171168Smlaier		return (-1);
2913171168Smlaier	pd->lookup.uid = UID_MAX;
2914171168Smlaier	pd->lookup.gid = GID_MAX;
2915171168Smlaier	pd->lookup.pid = NO_PID;		/* XXX: revisit */
2916135920Smlaier#ifdef __FreeBSD__
2917135920Smlaier	if (inp_arg != NULL) {
2918135920Smlaier		INP_LOCK_ASSERT(inp_arg);
2919135920Smlaier		if (inp_arg->inp_socket) {
2920171168Smlaier			pd->lookup.uid = inp_arg->inp_socket->so_cred->cr_uid;
2921171168Smlaier			pd->lookup.gid =
2922171168Smlaier			    inp_arg->inp_socket->so_cred->cr_groups[0];
2923135920Smlaier			return (1);
2924135920Smlaier		} else
2925171168Smlaier			return (-1);
2926135920Smlaier	}
2927135920Smlaier#endif
2928130613Smlaier	switch (pd->proto) {
2929126258Smlaier	case IPPROTO_TCP:
2930171168Smlaier		if (pd->hdr.tcp == NULL)
2931171168Smlaier			return (-1);
2932126258Smlaier		sport = pd->hdr.tcp->th_sport;
2933126258Smlaier		dport = pd->hdr.tcp->th_dport;
2934127145Smlaier#ifdef __FreeBSD__
2935126261Smlaier		pi = &tcbinfo;
2936126261Smlaier#else
2937126258Smlaier		tb = &tcbtable;
2938126261Smlaier#endif
2939126258Smlaier		break;
2940126258Smlaier	case IPPROTO_UDP:
2941171168Smlaier		if (pd->hdr.udp == NULL)
2942171168Smlaier			return (-1);
2943126258Smlaier		sport = pd->hdr.udp->uh_sport;
2944126258Smlaier		dport = pd->hdr.udp->uh_dport;
2945127145Smlaier#ifdef __FreeBSD__
2946126261Smlaier		pi = &udbinfo;
2947126261Smlaier#else
2948126258Smlaier		tb = &udbtable;
2949126261Smlaier#endif
2950126258Smlaier		break;
2951126258Smlaier	default:
2952171168Smlaier		return (-1);
2953126258Smlaier	}
2954126258Smlaier	if (direction == PF_IN) {
2955126258Smlaier		saddr = pd->src;
2956126258Smlaier		daddr = pd->dst;
2957126258Smlaier	} else {
2958126258Smlaier		u_int16_t	p;
2959126258Smlaier
2960126258Smlaier		p = sport;
2961126258Smlaier		sport = dport;
2962126258Smlaier		dport = p;
2963126258Smlaier		saddr = pd->dst;
2964126258Smlaier		daddr = pd->src;
2965126258Smlaier	}
2966130613Smlaier	switch (pd->af) {
2967145836Smlaier#ifdef INET
2968126258Smlaier	case AF_INET:
2969127145Smlaier#ifdef __FreeBSD__
2970126261Smlaier		INP_INFO_RLOCK(pi);	/* XXX LOR */
2971126261Smlaier		inp = in_pcblookup_hash(pi, saddr->v4, sport, daddr->v4,
2972126261Smlaier			dport, 0, NULL);
2973126261Smlaier		if (inp == NULL) {
2974126261Smlaier			inp = in_pcblookup_hash(pi, saddr->v4, sport,
2975126261Smlaier			   daddr->v4, dport, INPLOOKUP_WILDCARD, NULL);
2976126261Smlaier			if(inp == NULL) {
2977126261Smlaier				INP_INFO_RUNLOCK(pi);
2978171168Smlaier				return (-1);
2979126261Smlaier			}
2980126261Smlaier		}
2981126261Smlaier#else
2982126258Smlaier		inp = in_pcbhashlookup(tb, saddr->v4, sport, daddr->v4, dport);
2983126258Smlaier		if (inp == NULL) {
2984130613Smlaier			inp = in_pcblookup_listen(tb, daddr->v4, dport, 0);
2985126258Smlaier			if (inp == NULL)
2986171168Smlaier				return (-1);
2987126258Smlaier		}
2988126261Smlaier#endif
2989126258Smlaier		break;
2990145836Smlaier#endif /* INET */
2991126258Smlaier#ifdef INET6
2992126258Smlaier	case AF_INET6:
2993127145Smlaier#ifdef __FreeBSD__
2994126261Smlaier		INP_INFO_RLOCK(pi);
2995126261Smlaier		inp = in6_pcblookup_hash(pi, &saddr->v6, sport,
2996126261Smlaier			&daddr->v6, dport, 0, NULL);
2997126261Smlaier		if (inp == NULL) {
2998126261Smlaier			inp = in6_pcblookup_hash(pi, &saddr->v6, sport,
2999126261Smlaier			&daddr->v6, dport, INPLOOKUP_WILDCARD, NULL);
3000126261Smlaier			if (inp == NULL) {
3001126261Smlaier				INP_INFO_RUNLOCK(pi);
3002171168Smlaier				return (-1);
3003126261Smlaier			}
3004126261Smlaier		}
3005126261Smlaier#else
3006126258Smlaier		inp = in6_pcbhashlookup(tb, &saddr->v6, sport, &daddr->v6,
3007126258Smlaier		    dport);
3008126258Smlaier		if (inp == NULL) {
3009130613Smlaier			inp = in6_pcblookup_listen(tb, &daddr->v6, dport, 0);
3010126258Smlaier			if (inp == NULL)
3011171168Smlaier				return (-1);
3012126258Smlaier		}
3013126261Smlaier#endif
3014126258Smlaier		break;
3015126258Smlaier#endif /* INET6 */
3016126258Smlaier
3017126258Smlaier	default:
3018171168Smlaier		return (-1);
3019126258Smlaier	}
3020127145Smlaier#ifdef __FreeBSD__
3021126261Smlaier	INP_LOCK(inp);
3022136925Smlaier	if ((inp->inp_socket == NULL) || (inp->inp_socket->so_cred == NULL)) {
3023136925Smlaier		INP_UNLOCK(inp);
3024136925Smlaier		INP_INFO_RUNLOCK(pi);
3025171168Smlaier		return (-1);
3026136925Smlaier	}
3027171168Smlaier	pd->lookup.uid = inp->inp_socket->so_cred->cr_uid;
3028171168Smlaier	pd->lookup.gid = inp->inp_socket->so_cred->cr_groups[0];
3029126261Smlaier	INP_UNLOCK(inp);
3030126261Smlaier	INP_INFO_RUNLOCK(pi);
3031126261Smlaier#else
3032171168Smlaier	pd->lookup.uid = inp->inp_socket->so_euid;
3033171168Smlaier	pd->lookup.gid = inp->inp_socket->so_egid;
3034171168Smlaier	pd->lookup.pid = inp->inp_socket->so_cpid;
3035126261Smlaier#endif
3036126258Smlaier	return (1);
3037126258Smlaier}
3038126258Smlaier
3039126258Smlaieru_int8_t
3040126258Smlaierpf_get_wscale(struct mbuf *m, int off, u_int16_t th_off, sa_family_t af)
3041126258Smlaier{
3042126258Smlaier	int		 hlen;
3043126258Smlaier	u_int8_t	 hdr[60];
3044126258Smlaier	u_int8_t	*opt, optlen;
3045126258Smlaier	u_int8_t	 wscale = 0;
3046126258Smlaier
3047126258Smlaier	hlen = th_off << 2;		/* hlen <= sizeof(hdr) */
3048126258Smlaier	if (hlen <= sizeof(struct tcphdr))
3049126258Smlaier		return (0);
3050126258Smlaier	if (!pf_pull_hdr(m, off, hdr, hlen, NULL, NULL, af))
3051126258Smlaier		return (0);
3052126258Smlaier	opt = hdr + sizeof(struct tcphdr);
3053126258Smlaier	hlen -= sizeof(struct tcphdr);
3054126258Smlaier	while (hlen >= 3) {
3055126258Smlaier		switch (*opt) {
3056126258Smlaier		case TCPOPT_EOL:
3057126258Smlaier		case TCPOPT_NOP:
3058126258Smlaier			++opt;
3059126258Smlaier			--hlen;
3060126258Smlaier			break;
3061126258Smlaier		case TCPOPT_WINDOW:
3062126258Smlaier			wscale = opt[2];
3063126258Smlaier			if (wscale > TCP_MAX_WINSHIFT)
3064126258Smlaier				wscale = TCP_MAX_WINSHIFT;
3065126258Smlaier			wscale |= PF_WSCALE_FLAG;
3066130613Smlaier			/* FALLTHROUGH */
3067126258Smlaier		default:
3068126258Smlaier			optlen = opt[1];
3069126258Smlaier			if (optlen < 2)
3070126258Smlaier				optlen = 2;
3071126258Smlaier			hlen -= optlen;
3072126258Smlaier			opt += optlen;
3073130613Smlaier			break;
3074126258Smlaier		}
3075126258Smlaier	}
3076126258Smlaier	return (wscale);
3077126258Smlaier}
3078126258Smlaier
3079126258Smlaieru_int16_t
3080126258Smlaierpf_get_mss(struct mbuf *m, int off, u_int16_t th_off, sa_family_t af)
3081126258Smlaier{
3082126258Smlaier	int		 hlen;
3083126258Smlaier	u_int8_t	 hdr[60];
3084126258Smlaier	u_int8_t	*opt, optlen;
3085126258Smlaier	u_int16_t	 mss = tcp_mssdflt;
3086126258Smlaier
3087126258Smlaier	hlen = th_off << 2;	/* hlen <= sizeof(hdr) */
3088126258Smlaier	if (hlen <= sizeof(struct tcphdr))
3089126258Smlaier		return (0);
3090126258Smlaier	if (!pf_pull_hdr(m, off, hdr, hlen, NULL, NULL, af))
3091126258Smlaier		return (0);
3092126258Smlaier	opt = hdr + sizeof(struct tcphdr);
3093126258Smlaier	hlen -= sizeof(struct tcphdr);
3094126258Smlaier	while (hlen >= TCPOLEN_MAXSEG) {
3095126258Smlaier		switch (*opt) {
3096126258Smlaier		case TCPOPT_EOL:
3097126258Smlaier		case TCPOPT_NOP:
3098126258Smlaier			++opt;
3099126258Smlaier			--hlen;
3100126258Smlaier			break;
3101126258Smlaier		case TCPOPT_MAXSEG:
3102126258Smlaier			bcopy((caddr_t)(opt + 2), (caddr_t)&mss, 2);
3103145030Sglebius			NTOHS(mss);
3104130613Smlaier			/* FALLTHROUGH */
3105126258Smlaier		default:
3106126258Smlaier			optlen = opt[1];
3107126258Smlaier			if (optlen < 2)
3108126258Smlaier				optlen = 2;
3109126258Smlaier			hlen -= optlen;
3110126258Smlaier			opt += optlen;
3111130613Smlaier			break;
3112126258Smlaier		}
3113126258Smlaier	}
3114126258Smlaier	return (mss);
3115126258Smlaier}
3116126258Smlaier
3117126258Smlaieru_int16_t
3118126258Smlaierpf_calc_mss(struct pf_addr *addr, sa_family_t af, u_int16_t offer)
3119126258Smlaier{
3120126258Smlaier#ifdef INET
3121126258Smlaier	struct sockaddr_in	*dst;
3122126258Smlaier	struct route		 ro;
3123126258Smlaier#endif /* INET */
3124126258Smlaier#ifdef INET6
3125126258Smlaier	struct sockaddr_in6	*dst6;
3126126258Smlaier	struct route_in6	 ro6;
3127126258Smlaier#endif /* INET6 */
3128126258Smlaier	struct rtentry		*rt = NULL;
3129127629Smlaier	int			 hlen = 0;	/* make the compiler happy */
3130126258Smlaier	u_int16_t		 mss = tcp_mssdflt;
3131126258Smlaier
3132126258Smlaier	switch (af) {
3133126258Smlaier#ifdef INET
3134126258Smlaier	case AF_INET:
3135126258Smlaier		hlen = sizeof(struct ip);
3136126258Smlaier		bzero(&ro, sizeof(ro));
3137126258Smlaier		dst = (struct sockaddr_in *)&ro.ro_dst;
3138126258Smlaier		dst->sin_family = AF_INET;
3139126258Smlaier		dst->sin_len = sizeof(*dst);
3140126258Smlaier		dst->sin_addr = addr->v4;
3141127145Smlaier#ifdef __FreeBSD__
3142126261Smlaier#ifdef RTF_PRCLONING
3143126261Smlaier		rtalloc_ign(&ro, (RTF_CLONING | RTF_PRCLONING));
3144126261Smlaier#else /* !RTF_PRCLONING */
3145126261Smlaier		rtalloc_ign(&ro, RTF_CLONING);
3146126261Smlaier#endif
3147126261Smlaier#else /* ! __FreeBSD__ */
3148126258Smlaier		rtalloc_noclone(&ro, NO_CLONING);
3149126261Smlaier#endif
3150126258Smlaier		rt = ro.ro_rt;
3151126258Smlaier		break;
3152126258Smlaier#endif /* INET */
3153126258Smlaier#ifdef INET6
3154126258Smlaier	case AF_INET6:
3155126258Smlaier		hlen = sizeof(struct ip6_hdr);
3156126258Smlaier		bzero(&ro6, sizeof(ro6));
3157126258Smlaier		dst6 = (struct sockaddr_in6 *)&ro6.ro_dst;
3158126258Smlaier		dst6->sin6_family = AF_INET6;
3159126258Smlaier		dst6->sin6_len = sizeof(*dst6);
3160126258Smlaier		dst6->sin6_addr = addr->v6;
3161127145Smlaier#ifdef __FreeBSD__
3162126261Smlaier#ifdef RTF_PRCLONING
3163126261Smlaier		rtalloc_ign((struct route *)&ro6,
3164126261Smlaier		    (RTF_CLONING | RTF_PRCLONING));
3165126261Smlaier#else /* !RTF_PRCLONING */
3166126261Smlaier		rtalloc_ign((struct route *)&ro6, RTF_CLONING);
3167126261Smlaier#endif
3168126261Smlaier#else /* ! __FreeBSD__ */
3169126258Smlaier		rtalloc_noclone((struct route *)&ro6, NO_CLONING);
3170126261Smlaier#endif
3171126258Smlaier		rt = ro6.ro_rt;
3172126258Smlaier		break;
3173126258Smlaier#endif /* INET6 */
3174126258Smlaier	}
3175126258Smlaier
3176126258Smlaier	if (rt && rt->rt_ifp) {
3177126258Smlaier		mss = rt->rt_ifp->if_mtu - hlen - sizeof(struct tcphdr);
3178126258Smlaier		mss = max(tcp_mssdflt, mss);
3179126258Smlaier		RTFREE(rt);
3180126258Smlaier	}
3181126258Smlaier	mss = min(mss, offer);
3182126258Smlaier	mss = max(mss, 64);		/* sanity - at least max opt space */
3183126258Smlaier	return (mss);
3184126258Smlaier}
3185126258Smlaier
3186126258Smlaiervoid
3187126258Smlaierpf_set_rt_ifp(struct pf_state *s, struct pf_addr *saddr)
3188126258Smlaier{
3189126258Smlaier	struct pf_rule *r = s->rule.ptr;
3190126258Smlaier
3191130613Smlaier	s->rt_kif = NULL;
3192126258Smlaier	if (!r->rt || r->rt == PF_FASTROUTE)
3193126258Smlaier		return;
3194126258Smlaier	switch (s->af) {
3195126258Smlaier#ifdef INET
3196126258Smlaier	case AF_INET:
3197130613Smlaier		pf_map_addr(AF_INET, r, saddr, &s->rt_addr, NULL,
3198130613Smlaier		    &s->nat_src_node);
3199130613Smlaier		s->rt_kif = r->rpool.cur->kif;
3200126258Smlaier		break;
3201126258Smlaier#endif /* INET */
3202126258Smlaier#ifdef INET6
3203126258Smlaier	case AF_INET6:
3204130613Smlaier		pf_map_addr(AF_INET6, r, saddr, &s->rt_addr, NULL,
3205130613Smlaier		    &s->nat_src_node);
3206130613Smlaier		s->rt_kif = r->rpool.cur->kif;
3207126258Smlaier		break;
3208126258Smlaier#endif /* INET6 */
3209126258Smlaier	}
3210126258Smlaier}
3211126258Smlaier
3212126258Smlaierint
3213126258Smlaierpf_test_tcp(struct pf_rule **rm, struct pf_state **sm, int direction,
3214130613Smlaier    struct pfi_kif *kif, struct mbuf *m, int off, void *h,
3215135920Smlaier#ifdef __FreeBSD__
3216135920Smlaier    struct pf_pdesc *pd, struct pf_rule **am, struct pf_ruleset **rsm,
3217145836Smlaier    struct ifqueue *ifq, struct inpcb *inp)
3218135920Smlaier#else
3219145836Smlaier    struct pf_pdesc *pd, struct pf_rule **am, struct pf_ruleset **rsm,
3220145836Smlaier    struct ifqueue *ifq)
3221135920Smlaier#endif
3222126258Smlaier{
3223130613Smlaier	struct pf_rule		*nr = NULL;
3224126258Smlaier	struct pf_addr		*saddr = pd->src, *daddr = pd->dst;
3225126258Smlaier	struct tcphdr		*th = pd->hdr.tcp;
3226126258Smlaier	u_int16_t		 bport, nport = 0;
3227126258Smlaier	sa_family_t		 af = pd->af;
3228126258Smlaier	struct pf_rule		*r, *a = NULL;
3229126258Smlaier	struct pf_ruleset	*ruleset = NULL;
3230130613Smlaier	struct pf_src_node	*nsn = NULL;
3231126258Smlaier	u_short			 reason;
3232126258Smlaier	int			 rewrite = 0;
3233171168Smlaier	int			 tag = -1, rtableid = -1;
3234126258Smlaier	u_int16_t		 mss = tcp_mssdflt;
3235145836Smlaier	int			 asd = 0;
3236171168Smlaier	int			 match = 0;
3237126258Smlaier
3238145836Smlaier	if (pf_check_congestion(ifq)) {
3239145836Smlaier		REASON_SET(&reason, PFRES_CONGEST);
3240145836Smlaier		return (PF_DROP);
3241145836Smlaier	}
3242145836Smlaier
3243171168Smlaier#ifdef __FreeBSD__
3244171168Smlaier	if (inp != NULL)
3245171168Smlaier		pd->lookup.done = pf_socket_lookup(direction, pd, inp);
3246171168Smlaier	else if (debug_pfugidhack) {
3247171168Smlaier		PF_UNLOCK();
3248171168Smlaier		DPFPRINTF(PF_DEBUG_MISC, ("pf: unlocked lookup\n"));
3249171168Smlaier		pd->lookup.done = pf_socket_lookup(direction, pd, inp);
3250171168Smlaier		PF_LOCK();
3251171168Smlaier	}
3252165631Smlaier#endif
3253165631Smlaier
3254126258Smlaier	r = TAILQ_FIRST(pf_main_ruleset.rules[PF_RULESET_FILTER].active.ptr);
3255126258Smlaier
3256126258Smlaier	if (direction == PF_OUT) {
3257126258Smlaier		bport = nport = th->th_sport;
3258126258Smlaier		/* check outgoing packet for BINAT/NAT */
3259130613Smlaier		if ((nr = pf_get_translation(pd, m, off, PF_OUT, kif, &nsn,
3260126258Smlaier		    saddr, th->th_sport, daddr, th->th_dport,
3261130613Smlaier		    &pd->naddr, &nport)) != NULL) {
3262130613Smlaier			PF_ACPY(&pd->baddr, saddr, af);
3263126258Smlaier			pf_change_ap(saddr, &th->th_sport, pd->ip_sum,
3264130613Smlaier			    &th->th_sum, &pd->naddr, nport, 0, af);
3265126258Smlaier			rewrite++;
3266130613Smlaier			if (nr->natpass)
3267126258Smlaier				r = NULL;
3268130613Smlaier			pd->nat_rule = nr;
3269126258Smlaier		}
3270126258Smlaier	} else {
3271126258Smlaier		bport = nport = th->th_dport;
3272126258Smlaier		/* check incoming packet for BINAT/RDR */
3273130613Smlaier		if ((nr = pf_get_translation(pd, m, off, PF_IN, kif, &nsn,
3274130613Smlaier		    saddr, th->th_sport, daddr, th->th_dport,
3275130613Smlaier		    &pd->naddr, &nport)) != NULL) {
3276130613Smlaier			PF_ACPY(&pd->baddr, daddr, af);
3277126258Smlaier			pf_change_ap(daddr, &th->th_dport, pd->ip_sum,
3278130613Smlaier			    &th->th_sum, &pd->naddr, nport, 0, af);
3279126258Smlaier			rewrite++;
3280130613Smlaier			if (nr->natpass)
3281126258Smlaier				r = NULL;
3282130613Smlaier			pd->nat_rule = nr;
3283126258Smlaier		}
3284126258Smlaier	}
3285126258Smlaier
3286126258Smlaier	while (r != NULL) {
3287126258Smlaier		r->evaluations++;
3288171168Smlaier		if (pfi_kif_match(r->kif, kif) == r->ifnot)
3289126258Smlaier			r = r->skip[PF_SKIP_IFP].ptr;
3290126258Smlaier		else if (r->direction && r->direction != direction)
3291126258Smlaier			r = r->skip[PF_SKIP_DIR].ptr;
3292126258Smlaier		else if (r->af && r->af != af)
3293126258Smlaier			r = r->skip[PF_SKIP_AF].ptr;
3294126258Smlaier		else if (r->proto && r->proto != IPPROTO_TCP)
3295126258Smlaier			r = r->skip[PF_SKIP_PROTO].ptr;
3296171168Smlaier		else if (PF_MISMATCHAW(&r->src.addr, saddr, af,
3297171168Smlaier		    r->src.neg, kif))
3298126258Smlaier			r = r->skip[PF_SKIP_SRC_ADDR].ptr;
3299126258Smlaier		else if (r->src.port_op && !pf_match_port(r->src.port_op,
3300126258Smlaier		    r->src.port[0], r->src.port[1], th->th_sport))
3301126258Smlaier			r = r->skip[PF_SKIP_SRC_PORT].ptr;
3302171168Smlaier		else if (PF_MISMATCHAW(&r->dst.addr, daddr, af,
3303171168Smlaier		    r->dst.neg, NULL))
3304126258Smlaier			r = r->skip[PF_SKIP_DST_ADDR].ptr;
3305126258Smlaier		else if (r->dst.port_op && !pf_match_port(r->dst.port_op,
3306126258Smlaier		    r->dst.port[0], r->dst.port[1], th->th_dport))
3307126258Smlaier			r = r->skip[PF_SKIP_DST_PORT].ptr;
3308171168Smlaier		else if (r->tos && !(r->tos == pd->tos))
3309126258Smlaier			r = TAILQ_NEXT(r, entries);
3310126258Smlaier		else if (r->rule_flag & PFRULE_FRAGMENT)
3311126258Smlaier			r = TAILQ_NEXT(r, entries);
3312126258Smlaier		else if ((r->flagset & th->th_flags) != r->flags)
3313126258Smlaier			r = TAILQ_NEXT(r, entries);
3314171168Smlaier		else if (r->uid.op && (pd->lookup.done || (pd->lookup.done =
3315135920Smlaier#ifdef __FreeBSD__
3316171168Smlaier		    pf_socket_lookup(direction, pd, inp), 1)) &&
3317135920Smlaier#else
3318171168Smlaier		    pf_socket_lookup(direction, pd), 1)) &&
3319135920Smlaier#endif
3320126258Smlaier		    !pf_match_uid(r->uid.op, r->uid.uid[0], r->uid.uid[1],
3321171168Smlaier		    pd->lookup.uid))
3322126258Smlaier			r = TAILQ_NEXT(r, entries);
3323171168Smlaier		else if (r->gid.op && (pd->lookup.done || (pd->lookup.done =
3324135920Smlaier#ifdef __FreeBSD__
3325171168Smlaier		    pf_socket_lookup(direction, pd, inp), 1)) &&
3326135920Smlaier#else
3327171168Smlaier		    pf_socket_lookup(direction, pd), 1)) &&
3328135920Smlaier#endif
3329126258Smlaier		    !pf_match_gid(r->gid.op, r->gid.gid[0], r->gid.gid[1],
3330171168Smlaier		    pd->lookup.gid))
3331126258Smlaier			r = TAILQ_NEXT(r, entries);
3332145836Smlaier		else if (r->prob && r->prob <= arc4random())
3333126258Smlaier			r = TAILQ_NEXT(r, entries);
3334171168Smlaier		else if (r->match_tag && !pf_match_tag(m, r, pd->pf_mtag, &tag))
3335126258Smlaier			r = TAILQ_NEXT(r, entries);
3336126258Smlaier		else if (r->os_fingerprint != PF_OSFP_ANY && !pf_osfp_match(
3337126258Smlaier		    pf_osfp_fingerprint(pd, m, off, th), r->os_fingerprint))
3338126258Smlaier			r = TAILQ_NEXT(r, entries);
3339126258Smlaier		else {
3340126258Smlaier			if (r->tag)
3341126258Smlaier				tag = r->tag;
3342171168Smlaier			if (r->rtableid >= 0)
3343171168Smlaier				rtableid = r->rtableid;
3344126258Smlaier			if (r->anchor == NULL) {
3345171168Smlaier				match = 1;
3346126258Smlaier				*rm = r;
3347126258Smlaier				*am = a;
3348126258Smlaier				*rsm = ruleset;
3349126258Smlaier				if ((*rm)->quick)
3350126258Smlaier					break;
3351126258Smlaier				r = TAILQ_NEXT(r, entries);
3352126258Smlaier			} else
3353145836Smlaier				pf_step_into_anchor(&asd, &ruleset,
3354171168Smlaier				    PF_RULESET_FILTER, &r, &a, &match);
3355126258Smlaier		}
3356171168Smlaier		if (r == NULL && pf_step_out_of_anchor(&asd, &ruleset,
3357171168Smlaier		    PF_RULESET_FILTER, &r, &a, &match))
3358171168Smlaier			break;
3359126258Smlaier	}
3360126258Smlaier	r = *rm;
3361126258Smlaier	a = *am;
3362126258Smlaier	ruleset = *rsm;
3363126258Smlaier
3364126258Smlaier	REASON_SET(&reason, PFRES_MATCH);
3365126258Smlaier
3366171168Smlaier	if (r->log || (nr != NULL && nr->natpass && nr->log)) {
3367126258Smlaier		if (rewrite)
3368171168Smlaier#ifdef __FreeBSD__
3369126261Smlaier			m_copyback(m, off, sizeof(*th), (caddr_t)th);
3370171168Smlaier#else
3371171168Smlaier			m_copyback(m, off, sizeof(*th), th);
3372171168Smlaier#endif
3373171168Smlaier		PFLOG_PACKET(kif, h, m, af, direction, reason, r->log ? r : nr,
3374171168Smlaier		    a, ruleset, pd);
3375126258Smlaier	}
3376126258Smlaier
3377126258Smlaier	if ((r->action == PF_DROP) &&
3378126258Smlaier	    ((r->rule_flag & PFRULE_RETURNRST) ||
3379126258Smlaier	    (r->rule_flag & PFRULE_RETURNICMP) ||
3380126258Smlaier	    (r->rule_flag & PFRULE_RETURN))) {
3381126258Smlaier		/* undo NAT changes, if they have taken place */
3382130613Smlaier		if (nr != NULL) {
3383130613Smlaier			if (direction == PF_OUT) {
3384130613Smlaier				pf_change_ap(saddr, &th->th_sport, pd->ip_sum,
3385130613Smlaier				    &th->th_sum, &pd->baddr, bport, 0, af);
3386130613Smlaier				rewrite++;
3387130613Smlaier			} else {
3388130613Smlaier				pf_change_ap(daddr, &th->th_dport, pd->ip_sum,
3389130613Smlaier				    &th->th_sum, &pd->baddr, bport, 0, af);
3390130613Smlaier				rewrite++;
3391130613Smlaier			}
3392126258Smlaier		}
3393126258Smlaier		if (((r->rule_flag & PFRULE_RETURNRST) ||
3394126258Smlaier		    (r->rule_flag & PFRULE_RETURN)) &&
3395126258Smlaier		    !(th->th_flags & TH_RST)) {
3396126258Smlaier			u_int32_t ack = ntohl(th->th_seq) + pd->p_len;
3397126258Smlaier
3398126258Smlaier			if (th->th_flags & TH_SYN)
3399126258Smlaier				ack++;
3400126258Smlaier			if (th->th_flags & TH_FIN)
3401126258Smlaier				ack++;
3402162238Scsjp#ifdef __FreeBSD__
3403162238Scsjp			pf_send_tcp(m, r, af, pd->dst,
3404162238Scsjp#else
3405126258Smlaier			pf_send_tcp(r, af, pd->dst,
3406162238Scsjp#endif
3407126258Smlaier			    pd->src, th->th_dport, th->th_sport,
3408126258Smlaier			    ntohl(th->th_ack), ack, TH_RST|TH_ACK, 0, 0,
3409171168Smlaier			    r->return_ttl, 1, 0, pd->eh, kif->pfik_ifp);
3410126258Smlaier		} else if ((af == AF_INET) && r->return_icmp)
3411126258Smlaier			pf_send_icmp(m, r->return_icmp >> 8,
3412126258Smlaier			    r->return_icmp & 255, af, r);
3413126258Smlaier		else if ((af == AF_INET6) && r->return_icmp6)
3414126258Smlaier			pf_send_icmp(m, r->return_icmp6 >> 8,
3415126258Smlaier			    r->return_icmp6 & 255, af, r);
3416126258Smlaier	}
3417126258Smlaier
3418126258Smlaier	if (r->action == PF_DROP)
3419126258Smlaier		return (PF_DROP);
3420126258Smlaier
3421171168Smlaier	if (pf_tag_packet(m, pd->pf_mtag, tag, rtableid)) {
3422126258Smlaier		REASON_SET(&reason, PFRES_MEMORY);
3423126258Smlaier		return (PF_DROP);
3424126258Smlaier	}
3425126258Smlaier
3426130613Smlaier	if (r->keep_state || nr != NULL ||
3427126258Smlaier	    (pd->flags & PFDESC_TCP_NORM)) {
3428126258Smlaier		/* create new state */
3429126258Smlaier		u_int16_t	 len;
3430126258Smlaier		struct pf_state	*s = NULL;
3431130613Smlaier		struct pf_src_node *sn = NULL;
3432126258Smlaier
3433126258Smlaier		len = pd->tot_len - off - (th->th_off << 2);
3434130613Smlaier
3435130613Smlaier		/* check maximums */
3436145836Smlaier		if (r->max_states && (r->states >= r->max_states)) {
3437145836Smlaier			pf_status.lcounters[LCNT_STATES]++;
3438145836Smlaier			REASON_SET(&reason, PFRES_MAXSTATES);
3439130613Smlaier			goto cleanup;
3440145836Smlaier		}
3441171168Smlaier		/* src node for filter rule */
3442130613Smlaier		if ((r->rule_flag & PFRULE_SRCTRACK ||
3443130613Smlaier		    r->rpool.opts & PF_POOL_STICKYADDR) &&
3444145836Smlaier		    pf_insert_src_node(&sn, r, saddr, af) != 0) {
3445145836Smlaier			REASON_SET(&reason, PFRES_SRCLIMIT);
3446130613Smlaier			goto cleanup;
3447145836Smlaier		}
3448130613Smlaier		/* src node for translation rule */
3449130613Smlaier		if (nr != NULL && (nr->rpool.opts & PF_POOL_STICKYADDR) &&
3450130613Smlaier		    ((direction == PF_OUT &&
3451130613Smlaier		    pf_insert_src_node(&nsn, nr, &pd->baddr, af) != 0) ||
3452145836Smlaier		    (pf_insert_src_node(&nsn, nr, saddr, af) != 0))) {
3453145836Smlaier			REASON_SET(&reason, PFRES_SRCLIMIT);
3454130613Smlaier			goto cleanup;
3455145836Smlaier		}
3456130613Smlaier		s = pool_get(&pf_state_pl, PR_NOWAIT);
3457126258Smlaier		if (s == NULL) {
3458145836Smlaier			REASON_SET(&reason, PFRES_MEMORY);
3459130613Smlaiercleanup:
3460130613Smlaier			if (sn != NULL && sn->states == 0 && sn->expire == 0) {
3461130613Smlaier				RB_REMOVE(pf_src_tree, &tree_src_tracking, sn);
3462130613Smlaier				pf_status.scounters[SCNT_SRC_NODE_REMOVALS]++;
3463130613Smlaier				pf_status.src_nodes--;
3464130613Smlaier				pool_put(&pf_src_tree_pl, sn);
3465130613Smlaier			}
3466130613Smlaier			if (nsn != sn && nsn != NULL && nsn->states == 0 &&
3467130613Smlaier			    nsn->expire == 0) {
3468130613Smlaier				RB_REMOVE(pf_src_tree, &tree_src_tracking, nsn);
3469130613Smlaier				pf_status.scounters[SCNT_SRC_NODE_REMOVALS]++;
3470130613Smlaier				pf_status.src_nodes--;
3471130613Smlaier				pool_put(&pf_src_tree_pl, nsn);
3472130613Smlaier			}
3473126258Smlaier			return (PF_DROP);
3474126258Smlaier		}
3475126258Smlaier		bzero(s, sizeof(*s));
3476126258Smlaier		s->rule.ptr = r;
3477130613Smlaier		s->nat_rule.ptr = nr;
3478126258Smlaier		s->anchor.ptr = a;
3479145836Smlaier		STATE_INC_COUNTERS(s);
3480126258Smlaier		s->allow_opts = r->allow_opts;
3481171168Smlaier		s->log = r->log & PF_LOG_ALL;
3482171168Smlaier		if (nr != NULL)
3483171168Smlaier			s->log |= nr->log & PF_LOG_ALL;
3484126258Smlaier		s->proto = IPPROTO_TCP;
3485126258Smlaier		s->direction = direction;
3486126258Smlaier		s->af = af;
3487126258Smlaier		if (direction == PF_OUT) {
3488126258Smlaier			PF_ACPY(&s->gwy.addr, saddr, af);
3489126258Smlaier			s->gwy.port = th->th_sport;		/* sport */
3490126258Smlaier			PF_ACPY(&s->ext.addr, daddr, af);
3491126258Smlaier			s->ext.port = th->th_dport;
3492130613Smlaier			if (nr != NULL) {
3493130613Smlaier				PF_ACPY(&s->lan.addr, &pd->baddr, af);
3494126258Smlaier				s->lan.port = bport;
3495126258Smlaier			} else {
3496126258Smlaier				PF_ACPY(&s->lan.addr, &s->gwy.addr, af);
3497126258Smlaier				s->lan.port = s->gwy.port;
3498126258Smlaier			}
3499126258Smlaier		} else {
3500126258Smlaier			PF_ACPY(&s->lan.addr, daddr, af);
3501126258Smlaier			s->lan.port = th->th_dport;
3502126258Smlaier			PF_ACPY(&s->ext.addr, saddr, af);
3503126258Smlaier			s->ext.port = th->th_sport;
3504130613Smlaier			if (nr != NULL) {
3505130613Smlaier				PF_ACPY(&s->gwy.addr, &pd->baddr, af);
3506126258Smlaier				s->gwy.port = bport;
3507126258Smlaier			} else {
3508126258Smlaier				PF_ACPY(&s->gwy.addr, &s->lan.addr, af);
3509126258Smlaier				s->gwy.port = s->lan.port;
3510126258Smlaier			}
3511126258Smlaier		}
3512126258Smlaier
3513126258Smlaier		s->src.seqlo = ntohl(th->th_seq);
3514126258Smlaier		s->src.seqhi = s->src.seqlo + len + 1;
3515126258Smlaier		if ((th->th_flags & (TH_SYN|TH_ACK)) == TH_SYN &&
3516126258Smlaier		    r->keep_state == PF_STATE_MODULATE) {
3517126258Smlaier			/* Generate sequence number modulator */
3518171168Smlaier#ifdef __FreeBSD__
3519171168Smlaier			while ((s->src.seqdiff =
3520171168Smlaier			    pf_new_isn(s) - s->src.seqlo) == 0)
3521171168Smlaier				;
3522171168Smlaier#else
3523171168Smlaier			while ((s->src.seqdiff =
3524171168Smlaier			    tcp_rndiss_next() - s->src.seqlo) == 0)
3525126258Smlaier				;
3526171168Smlaier#endif
3527126258Smlaier			pf_change_a(&th->th_seq, &th->th_sum,
3528126258Smlaier			    htonl(s->src.seqlo + s->src.seqdiff), 0);
3529126258Smlaier			rewrite = 1;
3530126258Smlaier		} else
3531126258Smlaier			s->src.seqdiff = 0;
3532126258Smlaier		if (th->th_flags & TH_SYN) {
3533126258Smlaier			s->src.seqhi++;
3534126258Smlaier			s->src.wscale = pf_get_wscale(m, off, th->th_off, af);
3535126258Smlaier		}
3536126258Smlaier		s->src.max_win = MAX(ntohs(th->th_win), 1);
3537126258Smlaier		if (s->src.wscale & PF_WSCALE_MASK) {
3538126258Smlaier			/* Remove scale factor from initial window */
3539126258Smlaier			int win = s->src.max_win;
3540126258Smlaier			win += 1 << (s->src.wscale & PF_WSCALE_MASK);
3541126258Smlaier			s->src.max_win = (win - 1) >>
3542126258Smlaier			    (s->src.wscale & PF_WSCALE_MASK);
3543126258Smlaier		}
3544126258Smlaier		if (th->th_flags & TH_FIN)
3545126258Smlaier			s->src.seqhi++;
3546126258Smlaier		s->dst.seqhi = 1;
3547126258Smlaier		s->dst.max_win = 1;
3548126258Smlaier		s->src.state = TCPS_SYN_SENT;
3549126258Smlaier		s->dst.state = TCPS_CLOSED;
3550126261Smlaier		s->creation = time_second;
3551126261Smlaier		s->expire = time_second;
3552126258Smlaier		s->timeout = PFTM_TCP_FIRST_PACKET;
3553126258Smlaier		pf_set_rt_ifp(s, saddr);
3554130613Smlaier		if (sn != NULL) {
3555130613Smlaier			s->src_node = sn;
3556130613Smlaier			s->src_node->states++;
3557130613Smlaier		}
3558130613Smlaier		if (nsn != NULL) {
3559130613Smlaier			PF_ACPY(&nsn->raddr, &pd->naddr, af);
3560130613Smlaier			s->nat_src_node = nsn;
3561130613Smlaier			s->nat_src_node->states++;
3562130613Smlaier		}
3563126258Smlaier		if ((pd->flags & PFDESC_TCP_NORM) && pf_normalize_tcp_init(m,
3564126258Smlaier		    off, pd, th, &s->src, &s->dst)) {
3565126258Smlaier			REASON_SET(&reason, PFRES_MEMORY);
3566130613Smlaier			pf_src_tree_remove_state(s);
3567145836Smlaier			STATE_DEC_COUNTERS(s);
3568126258Smlaier			pool_put(&pf_state_pl, s);
3569126258Smlaier			return (PF_DROP);
3570126258Smlaier		}
3571126258Smlaier		if ((pd->flags & PFDESC_TCP_NORM) && s->src.scrub &&
3572145836Smlaier		    pf_normalize_tcp_stateful(m, off, pd, &reason, th, s,
3573145836Smlaier		    &s->src, &s->dst, &rewrite)) {
3574145836Smlaier			/* This really shouldn't happen!!! */
3575145836Smlaier			DPFPRINTF(PF_DEBUG_URGENT,
3576145836Smlaier			    ("pf_normalize_tcp_stateful failed on first pkt"));
3577126258Smlaier			pf_normalize_tcp_cleanup(s);
3578130613Smlaier			pf_src_tree_remove_state(s);
3579145836Smlaier			STATE_DEC_COUNTERS(s);
3580126258Smlaier			pool_put(&pf_state_pl, s);
3581126258Smlaier			return (PF_DROP);
3582126258Smlaier		}
3583130613Smlaier		if (pf_insert_state(BOUND_IFACE(r, kif), s)) {
3584126258Smlaier			pf_normalize_tcp_cleanup(s);
3585145836Smlaier			REASON_SET(&reason, PFRES_STATEINS);
3586130613Smlaier			pf_src_tree_remove_state(s);
3587145836Smlaier			STATE_DEC_COUNTERS(s);
3588126258Smlaier			pool_put(&pf_state_pl, s);
3589126258Smlaier			return (PF_DROP);
3590126258Smlaier		} else
3591126258Smlaier			*sm = s;
3592145836Smlaier		if (tag > 0) {
3593145836Smlaier			pf_tag_ref(tag);
3594145836Smlaier			s->tag = tag;
3595145836Smlaier		}
3596126258Smlaier		if ((th->th_flags & (TH_SYN|TH_ACK)) == TH_SYN &&
3597126258Smlaier		    r->keep_state == PF_STATE_SYNPROXY) {
3598126258Smlaier			s->src.state = PF_TCPS_PROXY_SRC;
3599130613Smlaier			if (nr != NULL) {
3600130613Smlaier				if (direction == PF_OUT) {
3601130613Smlaier					pf_change_ap(saddr, &th->th_sport,
3602130613Smlaier					    pd->ip_sum, &th->th_sum, &pd->baddr,
3603130613Smlaier					    bport, 0, af);
3604130613Smlaier				} else {
3605130613Smlaier					pf_change_ap(daddr, &th->th_dport,
3606130613Smlaier					    pd->ip_sum, &th->th_sum, &pd->baddr,
3607130613Smlaier					    bport, 0, af);
3608130613Smlaier				}
3609130613Smlaier			}
3610145836Smlaier			s->src.seqhi = htonl(arc4random());
3611126258Smlaier			/* Find mss option */
3612126258Smlaier			mss = pf_get_mss(m, off, th->th_off, af);
3613126258Smlaier			mss = pf_calc_mss(saddr, af, mss);
3614126258Smlaier			mss = pf_calc_mss(daddr, af, mss);
3615126258Smlaier			s->src.mss = mss;
3616162238Scsjp#ifdef __FreeBSD__
3617162238Scsjp			pf_send_tcp(NULL, r, af, daddr, saddr, th->th_dport,
3618162238Scsjp#else
3619126258Smlaier			pf_send_tcp(r, af, daddr, saddr, th->th_dport,
3620162238Scsjp#endif
3621130613Smlaier			    th->th_sport, s->src.seqhi, ntohl(th->th_seq) + 1,
3622171168Smlaier			    TH_SYN|TH_ACK, 0, s->src.mss, 0, 1, 0, NULL, NULL);
3623145836Smlaier			REASON_SET(&reason, PFRES_SYNPROXY);
3624126258Smlaier			return (PF_SYNPROXY_DROP);
3625126258Smlaier		}
3626126258Smlaier	}
3627126258Smlaier
3628126258Smlaier	/* copy back packet headers if we performed NAT operations */
3629126258Smlaier	if (rewrite)
3630126261Smlaier		m_copyback(m, off, sizeof(*th), (caddr_t)th);
3631126258Smlaier
3632126258Smlaier	return (PF_PASS);
3633126258Smlaier}
3634126258Smlaier
3635126258Smlaierint
3636126258Smlaierpf_test_udp(struct pf_rule **rm, struct pf_state **sm, int direction,
3637130613Smlaier    struct pfi_kif *kif, struct mbuf *m, int off, void *h,
3638135920Smlaier#ifdef __FreeBSD__
3639135920Smlaier    struct pf_pdesc *pd, struct pf_rule **am, struct pf_ruleset **rsm,
3640145836Smlaier    struct ifqueue *ifq, struct inpcb *inp)
3641135920Smlaier#else
3642145836Smlaier    struct pf_pdesc *pd, struct pf_rule **am, struct pf_ruleset **rsm,
3643145836Smlaier    struct ifqueue *ifq)
3644135920Smlaier#endif
3645126258Smlaier{
3646130613Smlaier	struct pf_rule		*nr = NULL;
3647126258Smlaier	struct pf_addr		*saddr = pd->src, *daddr = pd->dst;
3648126258Smlaier	struct udphdr		*uh = pd->hdr.udp;
3649126258Smlaier	u_int16_t		 bport, nport = 0;
3650126258Smlaier	sa_family_t		 af = pd->af;
3651126258Smlaier	struct pf_rule		*r, *a = NULL;
3652126258Smlaier	struct pf_ruleset	*ruleset = NULL;
3653130613Smlaier	struct pf_src_node	*nsn = NULL;
3654126258Smlaier	u_short			 reason;
3655126258Smlaier	int			 rewrite = 0;
3656171168Smlaier	int			 tag = -1, rtableid = -1;
3657145836Smlaier	int			 asd = 0;
3658171168Smlaier	int			 match = 0;
3659126258Smlaier
3660145836Smlaier	if (pf_check_congestion(ifq)) {
3661145836Smlaier		REASON_SET(&reason, PFRES_CONGEST);
3662145836Smlaier		return (PF_DROP);
3663145836Smlaier	}
3664145836Smlaier
3665171168Smlaier#ifdef __FreeBSD__
3666171168Smlaier	if (inp != NULL)
3667171168Smlaier		pd->lookup.done = pf_socket_lookup(direction, pd, inp);
3668171168Smlaier	else if (debug_pfugidhack) {
3669171168Smlaier		PF_UNLOCK();
3670171168Smlaier		DPFPRINTF(PF_DEBUG_MISC, ("pf: unlocked lookup\n"));
3671171168Smlaier		pd->lookup.done = pf_socket_lookup(direction, pd, inp);
3672171168Smlaier		PF_LOCK();
3673171168Smlaier	}
3674165631Smlaier#endif
3675165631Smlaier
3676126258Smlaier	r = TAILQ_FIRST(pf_main_ruleset.rules[PF_RULESET_FILTER].active.ptr);
3677126258Smlaier
3678126258Smlaier	if (direction == PF_OUT) {
3679126258Smlaier		bport = nport = uh->uh_sport;
3680126258Smlaier		/* check outgoing packet for BINAT/NAT */
3681130613Smlaier		if ((nr = pf_get_translation(pd, m, off, PF_OUT, kif, &nsn,
3682126258Smlaier		    saddr, uh->uh_sport, daddr, uh->uh_dport,
3683130613Smlaier		    &pd->naddr, &nport)) != NULL) {
3684130613Smlaier			PF_ACPY(&pd->baddr, saddr, af);
3685126258Smlaier			pf_change_ap(saddr, &uh->uh_sport, pd->ip_sum,
3686130613Smlaier			    &uh->uh_sum, &pd->naddr, nport, 1, af);
3687126258Smlaier			rewrite++;
3688130613Smlaier			if (nr->natpass)
3689126258Smlaier				r = NULL;
3690130613Smlaier			pd->nat_rule = nr;
3691126258Smlaier		}
3692126258Smlaier	} else {
3693126258Smlaier		bport = nport = uh->uh_dport;
3694126258Smlaier		/* check incoming packet for BINAT/RDR */
3695130613Smlaier		if ((nr = pf_get_translation(pd, m, off, PF_IN, kif, &nsn,
3696130613Smlaier		    saddr, uh->uh_sport, daddr, uh->uh_dport, &pd->naddr,
3697130613Smlaier		    &nport)) != NULL) {
3698130613Smlaier			PF_ACPY(&pd->baddr, daddr, af);
3699126258Smlaier			pf_change_ap(daddr, &uh->uh_dport, pd->ip_sum,
3700130613Smlaier			    &uh->uh_sum, &pd->naddr, nport, 1, af);
3701126258Smlaier			rewrite++;
3702130613Smlaier			if (nr->natpass)
3703126258Smlaier				r = NULL;
3704130613Smlaier			pd->nat_rule = nr;
3705126258Smlaier		}
3706126258Smlaier	}
3707126258Smlaier
3708126258Smlaier	while (r != NULL) {
3709126258Smlaier		r->evaluations++;
3710171168Smlaier		if (pfi_kif_match(r->kif, kif) == r->ifnot)
3711126258Smlaier			r = r->skip[PF_SKIP_IFP].ptr;
3712126258Smlaier		else if (r->direction && r->direction != direction)
3713126258Smlaier			r = r->skip[PF_SKIP_DIR].ptr;
3714126258Smlaier		else if (r->af && r->af != af)
3715126258Smlaier			r = r->skip[PF_SKIP_AF].ptr;
3716126258Smlaier		else if (r->proto && r->proto != IPPROTO_UDP)
3717126258Smlaier			r = r->skip[PF_SKIP_PROTO].ptr;
3718171168Smlaier		else if (PF_MISMATCHAW(&r->src.addr, saddr, af,
3719171168Smlaier		    r->src.neg, kif))
3720126258Smlaier			r = r->skip[PF_SKIP_SRC_ADDR].ptr;
3721126258Smlaier		else if (r->src.port_op && !pf_match_port(r->src.port_op,
3722126258Smlaier		    r->src.port[0], r->src.port[1], uh->uh_sport))
3723126258Smlaier			r = r->skip[PF_SKIP_SRC_PORT].ptr;
3724171168Smlaier		else if (PF_MISMATCHAW(&r->dst.addr, daddr, af,
3725171168Smlaier		    r->dst.neg, NULL))
3726126258Smlaier			r = r->skip[PF_SKIP_DST_ADDR].ptr;
3727126258Smlaier		else if (r->dst.port_op && !pf_match_port(r->dst.port_op,
3728126258Smlaier		    r->dst.port[0], r->dst.port[1], uh->uh_dport))
3729126258Smlaier			r = r->skip[PF_SKIP_DST_PORT].ptr;
3730171168Smlaier		else if (r->tos && !(r->tos == pd->tos))
3731126258Smlaier			r = TAILQ_NEXT(r, entries);
3732126258Smlaier		else if (r->rule_flag & PFRULE_FRAGMENT)
3733126258Smlaier			r = TAILQ_NEXT(r, entries);
3734171168Smlaier		else if (r->uid.op && (pd->lookup.done || (pd->lookup.done =
3735135920Smlaier#ifdef __FreeBSD__
3736171168Smlaier		    pf_socket_lookup(direction, pd, inp), 1)) &&
3737135920Smlaier#else
3738171168Smlaier		    pf_socket_lookup(direction, pd), 1)) &&
3739135920Smlaier#endif
3740126258Smlaier		    !pf_match_uid(r->uid.op, r->uid.uid[0], r->uid.uid[1],
3741171168Smlaier		    pd->lookup.uid))
3742126258Smlaier			r = TAILQ_NEXT(r, entries);
3743171168Smlaier		else if (r->gid.op && (pd->lookup.done || (pd->lookup.done =
3744135920Smlaier#ifdef __FreeBSD__
3745171168Smlaier		    pf_socket_lookup(direction, pd, inp), 1)) &&
3746135920Smlaier#else
3747171168Smlaier		    pf_socket_lookup(direction, pd), 1)) &&
3748135920Smlaier#endif
3749126258Smlaier		    !pf_match_gid(r->gid.op, r->gid.gid[0], r->gid.gid[1],
3750171168Smlaier		    pd->lookup.gid))
3751126258Smlaier			r = TAILQ_NEXT(r, entries);
3752145836Smlaier		else if (r->prob && r->prob <= arc4random())
3753126258Smlaier			r = TAILQ_NEXT(r, entries);
3754171168Smlaier		else if (r->match_tag && !pf_match_tag(m, r, pd->pf_mtag, &tag))
3755126258Smlaier			r = TAILQ_NEXT(r, entries);
3756126258Smlaier		else if (r->os_fingerprint != PF_OSFP_ANY)
3757126258Smlaier			r = TAILQ_NEXT(r, entries);
3758126258Smlaier		else {
3759126258Smlaier			if (r->tag)
3760126258Smlaier				tag = r->tag;
3761171168Smlaier			if (r->rtableid >= 0)
3762171168Smlaier				rtableid = r->rtableid;
3763126258Smlaier			if (r->anchor == NULL) {
3764171168Smlaier				match = 1;
3765126258Smlaier				*rm = r;
3766126258Smlaier				*am = a;
3767126258Smlaier				*rsm = ruleset;
3768126258Smlaier				if ((*rm)->quick)
3769126258Smlaier					break;
3770126258Smlaier				r = TAILQ_NEXT(r, entries);
3771126258Smlaier			} else
3772145836Smlaier				pf_step_into_anchor(&asd, &ruleset,
3773171168Smlaier				    PF_RULESET_FILTER, &r, &a, &match);
3774126258Smlaier		}
3775171168Smlaier		if (r == NULL && pf_step_out_of_anchor(&asd, &ruleset,
3776171168Smlaier		    PF_RULESET_FILTER, &r, &a, &match))
3777171168Smlaier			break;
3778126258Smlaier	}
3779126258Smlaier	r = *rm;
3780126258Smlaier	a = *am;
3781126258Smlaier	ruleset = *rsm;
3782126258Smlaier
3783126258Smlaier	REASON_SET(&reason, PFRES_MATCH);
3784126258Smlaier
3785171168Smlaier	if (r->log || (nr != NULL && nr->natpass && nr->log)) {
3786126258Smlaier		if (rewrite)
3787171168Smlaier#ifdef __FreeBSD__
3788126261Smlaier			m_copyback(m, off, sizeof(*uh), (caddr_t)uh);
3789171168Smlaier#else
3790171168Smlaier			m_copyback(m, off, sizeof(*uh), uh);
3791171168Smlaier#endif
3792171168Smlaier		PFLOG_PACKET(kif, h, m, af, direction, reason, r->log ? r : nr,
3793171168Smlaier		    a, ruleset, pd);
3794126258Smlaier	}
3795126258Smlaier
3796126258Smlaier	if ((r->action == PF_DROP) &&
3797126258Smlaier	    ((r->rule_flag & PFRULE_RETURNICMP) ||
3798126258Smlaier	    (r->rule_flag & PFRULE_RETURN))) {
3799126258Smlaier		/* undo NAT changes, if they have taken place */
3800130613Smlaier		if (nr != NULL) {
3801130613Smlaier			if (direction == PF_OUT) {
3802130613Smlaier				pf_change_ap(saddr, &uh->uh_sport, pd->ip_sum,
3803130613Smlaier				    &uh->uh_sum, &pd->baddr, bport, 1, af);
3804130613Smlaier				rewrite++;
3805130613Smlaier			} else {
3806130613Smlaier				pf_change_ap(daddr, &uh->uh_dport, pd->ip_sum,
3807130613Smlaier				    &uh->uh_sum, &pd->baddr, bport, 1, af);
3808130613Smlaier				rewrite++;
3809130613Smlaier			}
3810126258Smlaier		}
3811126258Smlaier		if ((af == AF_INET) && r->return_icmp)
3812126258Smlaier			pf_send_icmp(m, r->return_icmp >> 8,
3813126258Smlaier			    r->return_icmp & 255, af, r);
3814126258Smlaier		else if ((af == AF_INET6) && r->return_icmp6)
3815126258Smlaier			pf_send_icmp(m, r->return_icmp6 >> 8,
3816126258Smlaier			    r->return_icmp6 & 255, af, r);
3817126258Smlaier	}
3818126258Smlaier
3819126258Smlaier	if (r->action == PF_DROP)
3820126258Smlaier		return (PF_DROP);
3821126258Smlaier
3822171168Smlaier	if (pf_tag_packet(m, pd->pf_mtag, tag, rtableid)) {
3823126258Smlaier		REASON_SET(&reason, PFRES_MEMORY);
3824126258Smlaier		return (PF_DROP);
3825126258Smlaier	}
3826126258Smlaier
3827130613Smlaier	if (r->keep_state || nr != NULL) {
3828126258Smlaier		/* create new state */
3829126258Smlaier		struct pf_state	*s = NULL;
3830130613Smlaier		struct pf_src_node *sn = NULL;
3831126258Smlaier
3832130613Smlaier		/* check maximums */
3833145836Smlaier		if (r->max_states && (r->states >= r->max_states)) {
3834145836Smlaier			pf_status.lcounters[LCNT_STATES]++;
3835145836Smlaier			REASON_SET(&reason, PFRES_MAXSTATES);
3836130613Smlaier			goto cleanup;
3837145836Smlaier		}
3838171168Smlaier		/* src node for filter rule */
3839130613Smlaier		if ((r->rule_flag & PFRULE_SRCTRACK ||
3840130613Smlaier		    r->rpool.opts & PF_POOL_STICKYADDR) &&
3841145836Smlaier		    pf_insert_src_node(&sn, r, saddr, af) != 0) {
3842145836Smlaier			REASON_SET(&reason, PFRES_SRCLIMIT);
3843130613Smlaier			goto cleanup;
3844145836Smlaier		}
3845130613Smlaier		/* src node for translation rule */
3846130613Smlaier		if (nr != NULL && (nr->rpool.opts & PF_POOL_STICKYADDR) &&
3847130613Smlaier		    ((direction == PF_OUT &&
3848130613Smlaier		    pf_insert_src_node(&nsn, nr, &pd->baddr, af) != 0) ||
3849145836Smlaier		    (pf_insert_src_node(&nsn, nr, saddr, af) != 0))) {
3850145836Smlaier			REASON_SET(&reason, PFRES_SRCLIMIT);
3851130613Smlaier			goto cleanup;
3852145836Smlaier		}
3853130613Smlaier		s = pool_get(&pf_state_pl, PR_NOWAIT);
3854126258Smlaier		if (s == NULL) {
3855145836Smlaier			REASON_SET(&reason, PFRES_MEMORY);
3856130613Smlaiercleanup:
3857130613Smlaier			if (sn != NULL && sn->states == 0 && sn->expire == 0) {
3858130613Smlaier				RB_REMOVE(pf_src_tree, &tree_src_tracking, sn);
3859130613Smlaier				pf_status.scounters[SCNT_SRC_NODE_REMOVALS]++;
3860130613Smlaier				pf_status.src_nodes--;
3861130613Smlaier				pool_put(&pf_src_tree_pl, sn);
3862130613Smlaier			}
3863130613Smlaier			if (nsn != sn && nsn != NULL && nsn->states == 0 &&
3864130613Smlaier			    nsn->expire == 0) {
3865130613Smlaier				RB_REMOVE(pf_src_tree, &tree_src_tracking, nsn);
3866130613Smlaier				pf_status.scounters[SCNT_SRC_NODE_REMOVALS]++;
3867130613Smlaier				pf_status.src_nodes--;
3868130613Smlaier				pool_put(&pf_src_tree_pl, nsn);
3869130613Smlaier			}
3870126258Smlaier			return (PF_DROP);
3871126258Smlaier		}
3872126258Smlaier		bzero(s, sizeof(*s));
3873126258Smlaier		s->rule.ptr = r;
3874130613Smlaier		s->nat_rule.ptr = nr;
3875126258Smlaier		s->anchor.ptr = a;
3876145836Smlaier		STATE_INC_COUNTERS(s);
3877126258Smlaier		s->allow_opts = r->allow_opts;
3878171168Smlaier		s->log = r->log & PF_LOG_ALL;
3879171168Smlaier		if (nr != NULL)
3880171168Smlaier			s->log |= nr->log & PF_LOG_ALL;
3881126258Smlaier		s->proto = IPPROTO_UDP;
3882126258Smlaier		s->direction = direction;
3883126258Smlaier		s->af = af;
3884126258Smlaier		if (direction == PF_OUT) {
3885126258Smlaier			PF_ACPY(&s->gwy.addr, saddr, af);
3886126258Smlaier			s->gwy.port = uh->uh_sport;
3887126258Smlaier			PF_ACPY(&s->ext.addr, daddr, af);
3888126258Smlaier			s->ext.port = uh->uh_dport;
3889130613Smlaier			if (nr != NULL) {
3890130613Smlaier				PF_ACPY(&s->lan.addr, &pd->baddr, af);
3891126258Smlaier				s->lan.port = bport;
3892126258Smlaier			} else {
3893126258Smlaier				PF_ACPY(&s->lan.addr, &s->gwy.addr, af);
3894126258Smlaier				s->lan.port = s->gwy.port;
3895126258Smlaier			}
3896126258Smlaier		} else {
3897126258Smlaier			PF_ACPY(&s->lan.addr, daddr, af);
3898126258Smlaier			s->lan.port = uh->uh_dport;
3899126258Smlaier			PF_ACPY(&s->ext.addr, saddr, af);
3900126258Smlaier			s->ext.port = uh->uh_sport;
3901130613Smlaier			if (nr != NULL) {
3902130613Smlaier				PF_ACPY(&s->gwy.addr, &pd->baddr, af);
3903126258Smlaier				s->gwy.port = bport;
3904126258Smlaier			} else {
3905126258Smlaier				PF_ACPY(&s->gwy.addr, &s->lan.addr, af);
3906126258Smlaier				s->gwy.port = s->lan.port;
3907126258Smlaier			}
3908126258Smlaier		}
3909126258Smlaier		s->src.state = PFUDPS_SINGLE;
3910126258Smlaier		s->dst.state = PFUDPS_NO_TRAFFIC;
3911126261Smlaier		s->creation = time_second;
3912126261Smlaier		s->expire = time_second;
3913126258Smlaier		s->timeout = PFTM_UDP_FIRST_PACKET;
3914126258Smlaier		pf_set_rt_ifp(s, saddr);
3915130613Smlaier		if (sn != NULL) {
3916130613Smlaier			s->src_node = sn;
3917130613Smlaier			s->src_node->states++;
3918130613Smlaier		}
3919130613Smlaier		if (nsn != NULL) {
3920130613Smlaier			PF_ACPY(&nsn->raddr, &pd->naddr, af);
3921130613Smlaier			s->nat_src_node = nsn;
3922130613Smlaier			s->nat_src_node->states++;
3923130613Smlaier		}
3924130613Smlaier		if (pf_insert_state(BOUND_IFACE(r, kif), s)) {
3925145836Smlaier			REASON_SET(&reason, PFRES_STATEINS);
3926130613Smlaier			pf_src_tree_remove_state(s);
3927145836Smlaier			STATE_DEC_COUNTERS(s);
3928126258Smlaier			pool_put(&pf_state_pl, s);
3929126258Smlaier			return (PF_DROP);
3930126258Smlaier		} else
3931126258Smlaier			*sm = s;
3932145836Smlaier		if (tag > 0) {
3933145836Smlaier			pf_tag_ref(tag);
3934145836Smlaier			s->tag = tag;
3935145836Smlaier		}
3936126258Smlaier	}
3937126258Smlaier
3938126258Smlaier	/* copy back packet headers if we performed NAT operations */
3939126258Smlaier	if (rewrite)
3940126261Smlaier		m_copyback(m, off, sizeof(*uh), (caddr_t)uh);
3941126258Smlaier
3942126258Smlaier	return (PF_PASS);
3943126258Smlaier}
3944126258Smlaier
3945126258Smlaierint
3946126258Smlaierpf_test_icmp(struct pf_rule **rm, struct pf_state **sm, int direction,
3947130613Smlaier    struct pfi_kif *kif, struct mbuf *m, int off, void *h,
3948145836Smlaier    struct pf_pdesc *pd, struct pf_rule **am, struct pf_ruleset **rsm,
3949145836Smlaier    struct ifqueue *ifq)
3950126258Smlaier{
3951130613Smlaier	struct pf_rule		*nr = NULL;
3952126258Smlaier	struct pf_addr		*saddr = pd->src, *daddr = pd->dst;
3953126258Smlaier	struct pf_rule		*r, *a = NULL;
3954126258Smlaier	struct pf_ruleset	*ruleset = NULL;
3955130613Smlaier	struct pf_src_node	*nsn = NULL;
3956126258Smlaier	u_short			 reason;
3957149893Smlaier	u_int16_t		 icmpid = 0, bport, nport = 0;
3958126258Smlaier	sa_family_t		 af = pd->af;
3959127629Smlaier	u_int8_t		 icmptype = 0;	/* make the compiler happy */
3960127629Smlaier	u_int8_t		 icmpcode = 0;	/* make the compiler happy */
3961126258Smlaier	int			 state_icmp = 0;
3962171168Smlaier	int			 tag = -1, rtableid = -1;
3963126258Smlaier#ifdef INET6
3964126258Smlaier	int			 rewrite = 0;
3965126258Smlaier#endif /* INET6 */
3966145836Smlaier	int			 asd = 0;
3967171168Smlaier	int			 match = 0;
3968126258Smlaier
3969145836Smlaier	if (pf_check_congestion(ifq)) {
3970145836Smlaier		REASON_SET(&reason, PFRES_CONGEST);
3971145836Smlaier		return (PF_DROP);
3972145836Smlaier	}
3973145836Smlaier
3974126258Smlaier	switch (pd->proto) {
3975126258Smlaier#ifdef INET
3976126258Smlaier	case IPPROTO_ICMP:
3977126258Smlaier		icmptype = pd->hdr.icmp->icmp_type;
3978126258Smlaier		icmpcode = pd->hdr.icmp->icmp_code;
3979126258Smlaier		icmpid = pd->hdr.icmp->icmp_id;
3980126258Smlaier
3981126258Smlaier		if (icmptype == ICMP_UNREACH ||
3982126258Smlaier		    icmptype == ICMP_SOURCEQUENCH ||
3983126258Smlaier		    icmptype == ICMP_REDIRECT ||
3984126258Smlaier		    icmptype == ICMP_TIMXCEED ||
3985126258Smlaier		    icmptype == ICMP_PARAMPROB)
3986126258Smlaier			state_icmp++;
3987126258Smlaier		break;
3988126258Smlaier#endif /* INET */
3989126258Smlaier#ifdef INET6
3990126258Smlaier	case IPPROTO_ICMPV6:
3991126258Smlaier		icmptype = pd->hdr.icmp6->icmp6_type;
3992126258Smlaier		icmpcode = pd->hdr.icmp6->icmp6_code;
3993126258Smlaier		icmpid = pd->hdr.icmp6->icmp6_id;
3994126258Smlaier
3995126258Smlaier		if (icmptype == ICMP6_DST_UNREACH ||
3996126258Smlaier		    icmptype == ICMP6_PACKET_TOO_BIG ||
3997126258Smlaier		    icmptype == ICMP6_TIME_EXCEEDED ||
3998126258Smlaier		    icmptype == ICMP6_PARAM_PROB)
3999126258Smlaier			state_icmp++;
4000126258Smlaier		break;
4001126258Smlaier#endif /* INET6 */
4002126258Smlaier	}
4003126258Smlaier
4004126258Smlaier	r = TAILQ_FIRST(pf_main_ruleset.rules[PF_RULESET_FILTER].active.ptr);
4005126258Smlaier
4006126258Smlaier	if (direction == PF_OUT) {
4007149884Smlaier		bport = nport = icmpid;
4008126258Smlaier		/* check outgoing packet for BINAT/NAT */
4009130613Smlaier		if ((nr = pf_get_translation(pd, m, off, PF_OUT, kif, &nsn,
4010149884Smlaier		    saddr, icmpid, daddr, icmpid, &pd->naddr, &nport)) !=
4011149884Smlaier		    NULL) {
4012130613Smlaier			PF_ACPY(&pd->baddr, saddr, af);
4013126258Smlaier			switch (af) {
4014126258Smlaier#ifdef INET
4015126258Smlaier			case AF_INET:
4016126258Smlaier				pf_change_a(&saddr->v4.s_addr, pd->ip_sum,
4017130613Smlaier				    pd->naddr.v4.s_addr, 0);
4018149884Smlaier				pd->hdr.icmp->icmp_cksum = pf_cksum_fixup(
4019149884Smlaier				    pd->hdr.icmp->icmp_cksum, icmpid, nport, 0);
4020149884Smlaier				pd->hdr.icmp->icmp_id = nport;
4021149893Smlaier				m_copyback(m, off, ICMP_MINLEN,
4022149893Smlaier				    (caddr_t)pd->hdr.icmp);
4023126258Smlaier				break;
4024126258Smlaier#endif /* INET */
4025126258Smlaier#ifdef INET6
4026126258Smlaier			case AF_INET6:
4027126258Smlaier				pf_change_a6(saddr, &pd->hdr.icmp6->icmp6_cksum,
4028130613Smlaier				    &pd->naddr, 0);
4029126258Smlaier				rewrite++;
4030126258Smlaier				break;
4031126258Smlaier#endif /* INET6 */
4032126258Smlaier			}
4033130613Smlaier			if (nr->natpass)
4034126258Smlaier				r = NULL;
4035130613Smlaier			pd->nat_rule = nr;
4036126258Smlaier		}
4037126258Smlaier	} else {
4038149884Smlaier		bport = nport = icmpid;
4039126258Smlaier		/* check incoming packet for BINAT/RDR */
4040130613Smlaier		if ((nr = pf_get_translation(pd, m, off, PF_IN, kif, &nsn,
4041149884Smlaier		    saddr, icmpid, daddr, icmpid, &pd->naddr, &nport)) !=
4042149884Smlaier		    NULL) {
4043130613Smlaier			PF_ACPY(&pd->baddr, daddr, af);
4044126258Smlaier			switch (af) {
4045126258Smlaier#ifdef INET
4046126258Smlaier			case AF_INET:
4047126258Smlaier				pf_change_a(&daddr->v4.s_addr,
4048130613Smlaier				    pd->ip_sum, pd->naddr.v4.s_addr, 0);
4049126258Smlaier				break;
4050126258Smlaier#endif /* INET */
4051126258Smlaier#ifdef INET6
4052126258Smlaier			case AF_INET6:
4053126258Smlaier				pf_change_a6(daddr, &pd->hdr.icmp6->icmp6_cksum,
4054130613Smlaier				    &pd->naddr, 0);
4055126258Smlaier				rewrite++;
4056126258Smlaier				break;
4057126258Smlaier#endif /* INET6 */
4058126258Smlaier			}
4059130613Smlaier			if (nr->natpass)
4060126258Smlaier				r = NULL;
4061130613Smlaier			pd->nat_rule = nr;
4062126258Smlaier		}
4063126258Smlaier	}
4064126258Smlaier
4065126258Smlaier	while (r != NULL) {
4066126258Smlaier		r->evaluations++;
4067171168Smlaier		if (pfi_kif_match(r->kif, kif) == r->ifnot)
4068126258Smlaier			r = r->skip[PF_SKIP_IFP].ptr;
4069126258Smlaier		else if (r->direction && r->direction != direction)
4070126258Smlaier			r = r->skip[PF_SKIP_DIR].ptr;
4071126258Smlaier		else if (r->af && r->af != af)
4072126258Smlaier			r = r->skip[PF_SKIP_AF].ptr;
4073126258Smlaier		else if (r->proto && r->proto != pd->proto)
4074126258Smlaier			r = r->skip[PF_SKIP_PROTO].ptr;
4075171168Smlaier		else if (PF_MISMATCHAW(&r->src.addr, saddr, af,
4076171168Smlaier		    r->src.neg, kif))
4077126258Smlaier			r = r->skip[PF_SKIP_SRC_ADDR].ptr;
4078171168Smlaier		else if (PF_MISMATCHAW(&r->dst.addr, daddr, af,
4079171168Smlaier		    r->dst.neg, NULL))
4080126258Smlaier			r = r->skip[PF_SKIP_DST_ADDR].ptr;
4081126258Smlaier		else if (r->type && r->type != icmptype + 1)
4082126258Smlaier			r = TAILQ_NEXT(r, entries);
4083126258Smlaier		else if (r->code && r->code != icmpcode + 1)
4084126258Smlaier			r = TAILQ_NEXT(r, entries);
4085171168Smlaier		else if (r->tos && !(r->tos == pd->tos))
4086126258Smlaier			r = TAILQ_NEXT(r, entries);
4087126258Smlaier		else if (r->rule_flag & PFRULE_FRAGMENT)
4088126258Smlaier			r = TAILQ_NEXT(r, entries);
4089145836Smlaier		else if (r->prob && r->prob <= arc4random())
4090126258Smlaier			r = TAILQ_NEXT(r, entries);
4091171168Smlaier		else if (r->match_tag && !pf_match_tag(m, r, pd->pf_mtag, &tag))
4092126258Smlaier			r = TAILQ_NEXT(r, entries);
4093126258Smlaier		else if (r->os_fingerprint != PF_OSFP_ANY)
4094126258Smlaier			r = TAILQ_NEXT(r, entries);
4095126258Smlaier		else {
4096126258Smlaier			if (r->tag)
4097126258Smlaier				tag = r->tag;
4098171168Smlaier			if (r->rtableid >= 0)
4099171168Smlaier				rtableid = r->rtableid;
4100126258Smlaier			if (r->anchor == NULL) {
4101171168Smlaier				match = 1;
4102126258Smlaier				*rm = r;
4103126258Smlaier				*am = a;
4104126258Smlaier				*rsm = ruleset;
4105126258Smlaier				if ((*rm)->quick)
4106126258Smlaier					break;
4107126258Smlaier				r = TAILQ_NEXT(r, entries);
4108126258Smlaier			} else
4109145836Smlaier				pf_step_into_anchor(&asd, &ruleset,
4110171168Smlaier				    PF_RULESET_FILTER, &r, &a, &match);
4111126258Smlaier		}
4112171168Smlaier		if (r == NULL && pf_step_out_of_anchor(&asd, &ruleset,
4113171168Smlaier		    PF_RULESET_FILTER, &r, &a, &match))
4114171168Smlaier			break;
4115126258Smlaier	}
4116126258Smlaier	r = *rm;
4117126258Smlaier	a = *am;
4118126258Smlaier	ruleset = *rsm;
4119126258Smlaier
4120126258Smlaier	REASON_SET(&reason, PFRES_MATCH);
4121126258Smlaier
4122171168Smlaier	if (r->log || (nr != NULL && nr->natpass && nr->log)) {
4123126258Smlaier#ifdef INET6
4124126258Smlaier		if (rewrite)
4125126258Smlaier			m_copyback(m, off, sizeof(struct icmp6_hdr),
4126126261Smlaier			    (caddr_t)pd->hdr.icmp6);
4127126258Smlaier#endif /* INET6 */
4128171168Smlaier		PFLOG_PACKET(kif, h, m, af, direction, reason, r->log ? r : nr,
4129171168Smlaier		    a, ruleset, pd);
4130126258Smlaier	}
4131126258Smlaier
4132126258Smlaier	if (r->action != PF_PASS)
4133126258Smlaier		return (PF_DROP);
4134126258Smlaier
4135171168Smlaier	if (pf_tag_packet(m, pd->pf_mtag, tag, rtableid)) {
4136126258Smlaier		REASON_SET(&reason, PFRES_MEMORY);
4137126258Smlaier		return (PF_DROP);
4138126258Smlaier	}
4139126258Smlaier
4140130613Smlaier	if (!state_icmp && (r->keep_state || nr != NULL)) {
4141126258Smlaier		/* create new state */
4142126258Smlaier		struct pf_state	*s = NULL;
4143130613Smlaier		struct pf_src_node *sn = NULL;
4144126258Smlaier
4145130613Smlaier		/* check maximums */
4146145836Smlaier		if (r->max_states && (r->states >= r->max_states)) {
4147145836Smlaier			pf_status.lcounters[LCNT_STATES]++;
4148145836Smlaier			REASON_SET(&reason, PFRES_MAXSTATES);
4149130613Smlaier			goto cleanup;
4150145836Smlaier		}
4151171168Smlaier		/* src node for filter rule */
4152130613Smlaier		if ((r->rule_flag & PFRULE_SRCTRACK ||
4153130613Smlaier		    r->rpool.opts & PF_POOL_STICKYADDR) &&
4154145836Smlaier		    pf_insert_src_node(&sn, r, saddr, af) != 0) {
4155145836Smlaier			REASON_SET(&reason, PFRES_SRCLIMIT);
4156130613Smlaier			goto cleanup;
4157145836Smlaier		}
4158130613Smlaier		/* src node for translation rule */
4159130613Smlaier		if (nr != NULL && (nr->rpool.opts & PF_POOL_STICKYADDR) &&
4160130613Smlaier		    ((direction == PF_OUT &&
4161130613Smlaier		    pf_insert_src_node(&nsn, nr, &pd->baddr, af) != 0) ||
4162145836Smlaier		    (pf_insert_src_node(&nsn, nr, saddr, af) != 0))) {
4163145836Smlaier			REASON_SET(&reason, PFRES_SRCLIMIT);
4164130613Smlaier			goto cleanup;
4165145836Smlaier		}
4166130613Smlaier		s = pool_get(&pf_state_pl, PR_NOWAIT);
4167126258Smlaier		if (s == NULL) {
4168145836Smlaier			REASON_SET(&reason, PFRES_MEMORY);
4169130613Smlaiercleanup:
4170130613Smlaier			if (sn != NULL && sn->states == 0 && sn->expire == 0) {
4171130613Smlaier				RB_REMOVE(pf_src_tree, &tree_src_tracking, sn);
4172130613Smlaier				pf_status.scounters[SCNT_SRC_NODE_REMOVALS]++;
4173130613Smlaier				pf_status.src_nodes--;
4174130613Smlaier				pool_put(&pf_src_tree_pl, sn);
4175130613Smlaier			}
4176130613Smlaier			if (nsn != sn && nsn != NULL && nsn->states == 0 &&
4177130613Smlaier			    nsn->expire == 0) {
4178130613Smlaier				RB_REMOVE(pf_src_tree, &tree_src_tracking, nsn);
4179130613Smlaier				pf_status.scounters[SCNT_SRC_NODE_REMOVALS]++;
4180130613Smlaier				pf_status.src_nodes--;
4181130613Smlaier				pool_put(&pf_src_tree_pl, nsn);
4182130613Smlaier			}
4183126258Smlaier			return (PF_DROP);
4184126258Smlaier		}
4185126258Smlaier		bzero(s, sizeof(*s));
4186126258Smlaier		s->rule.ptr = r;
4187130613Smlaier		s->nat_rule.ptr = nr;
4188126258Smlaier		s->anchor.ptr = a;
4189145836Smlaier		STATE_INC_COUNTERS(s);
4190126258Smlaier		s->allow_opts = r->allow_opts;
4191171168Smlaier		s->log = r->log & PF_LOG_ALL;
4192171168Smlaier		if (nr != NULL)
4193171168Smlaier			s->log |= nr->log & PF_LOG_ALL;
4194126258Smlaier		s->proto = pd->proto;
4195126258Smlaier		s->direction = direction;
4196126258Smlaier		s->af = af;
4197126258Smlaier		if (direction == PF_OUT) {
4198126258Smlaier			PF_ACPY(&s->gwy.addr, saddr, af);
4199149884Smlaier			s->gwy.port = nport;
4200126258Smlaier			PF_ACPY(&s->ext.addr, daddr, af);
4201149884Smlaier			s->ext.port = 0;
4202149884Smlaier			if (nr != NULL) {
4203130613Smlaier				PF_ACPY(&s->lan.addr, &pd->baddr, af);
4204149884Smlaier				s->lan.port = bport;
4205149884Smlaier			} else {
4206126258Smlaier				PF_ACPY(&s->lan.addr, &s->gwy.addr, af);
4207149884Smlaier				s->lan.port = s->gwy.port;
4208149884Smlaier			}
4209126258Smlaier		} else {
4210126258Smlaier			PF_ACPY(&s->lan.addr, daddr, af);
4211149884Smlaier			s->lan.port = nport;
4212126258Smlaier			PF_ACPY(&s->ext.addr, saddr, af);
4213149884Smlaier			s->ext.port = 0;
4214149884Smlaier			if (nr != NULL) {
4215130613Smlaier				PF_ACPY(&s->gwy.addr, &pd->baddr, af);
4216149884Smlaier				s->gwy.port = bport;
4217149884Smlaier			} else {
4218126258Smlaier				PF_ACPY(&s->gwy.addr, &s->lan.addr, af);
4219149884Smlaier				s->gwy.port = s->lan.port;
4220149884Smlaier			}
4221126258Smlaier		}
4222126261Smlaier		s->creation = time_second;
4223126261Smlaier		s->expire = time_second;
4224126258Smlaier		s->timeout = PFTM_ICMP_FIRST_PACKET;
4225126258Smlaier		pf_set_rt_ifp(s, saddr);
4226130613Smlaier		if (sn != NULL) {
4227130613Smlaier			s->src_node = sn;
4228130613Smlaier			s->src_node->states++;
4229130613Smlaier		}
4230130613Smlaier		if (nsn != NULL) {
4231130613Smlaier			PF_ACPY(&nsn->raddr, &pd->naddr, af);
4232130613Smlaier			s->nat_src_node = nsn;
4233130613Smlaier			s->nat_src_node->states++;
4234130613Smlaier		}
4235130613Smlaier		if (pf_insert_state(BOUND_IFACE(r, kif), s)) {
4236145836Smlaier			REASON_SET(&reason, PFRES_STATEINS);
4237130613Smlaier			pf_src_tree_remove_state(s);
4238145836Smlaier			STATE_DEC_COUNTERS(s);
4239126258Smlaier			pool_put(&pf_state_pl, s);
4240126258Smlaier			return (PF_DROP);
4241126258Smlaier		} else
4242126258Smlaier			*sm = s;
4243145836Smlaier		if (tag > 0) {
4244145836Smlaier			pf_tag_ref(tag);
4245145836Smlaier			s->tag = tag;
4246145836Smlaier		}
4247126258Smlaier	}
4248126258Smlaier
4249126258Smlaier#ifdef INET6
4250126258Smlaier	/* copy back packet headers if we performed IPv6 NAT operations */
4251126258Smlaier	if (rewrite)
4252126258Smlaier		m_copyback(m, off, sizeof(struct icmp6_hdr),
4253126261Smlaier		    (caddr_t)pd->hdr.icmp6);
4254126258Smlaier#endif /* INET6 */
4255126258Smlaier
4256126258Smlaier	return (PF_PASS);
4257126258Smlaier}
4258126258Smlaier
4259126258Smlaierint
4260126258Smlaierpf_test_other(struct pf_rule **rm, struct pf_state **sm, int direction,
4261130613Smlaier    struct pfi_kif *kif, struct mbuf *m, int off, void *h, struct pf_pdesc *pd,
4262145836Smlaier    struct pf_rule **am, struct pf_ruleset **rsm, struct ifqueue *ifq)
4263126258Smlaier{
4264130613Smlaier	struct pf_rule		*nr = NULL;
4265126258Smlaier	struct pf_rule		*r, *a = NULL;
4266126258Smlaier	struct pf_ruleset	*ruleset = NULL;
4267130613Smlaier	struct pf_src_node	*nsn = NULL;
4268126258Smlaier	struct pf_addr		*saddr = pd->src, *daddr = pd->dst;
4269126258Smlaier	sa_family_t		 af = pd->af;
4270126258Smlaier	u_short			 reason;
4271171168Smlaier	int			 tag = -1, rtableid = -1;
4272145836Smlaier	int			 asd = 0;
4273171168Smlaier	int			 match = 0;
4274126258Smlaier
4275145836Smlaier	if (pf_check_congestion(ifq)) {
4276145836Smlaier		REASON_SET(&reason, PFRES_CONGEST);
4277145836Smlaier		return (PF_DROP);
4278145836Smlaier	}
4279145836Smlaier
4280126258Smlaier	r = TAILQ_FIRST(pf_main_ruleset.rules[PF_RULESET_FILTER].active.ptr);
4281126258Smlaier
4282126258Smlaier	if (direction == PF_OUT) {
4283126258Smlaier		/* check outgoing packet for BINAT/NAT */
4284130613Smlaier		if ((nr = pf_get_translation(pd, m, off, PF_OUT, kif, &nsn,
4285130613Smlaier		    saddr, 0, daddr, 0, &pd->naddr, NULL)) != NULL) {
4286130613Smlaier			PF_ACPY(&pd->baddr, saddr, af);
4287126258Smlaier			switch (af) {
4288126258Smlaier#ifdef INET
4289126258Smlaier			case AF_INET:
4290126258Smlaier				pf_change_a(&saddr->v4.s_addr, pd->ip_sum,
4291130613Smlaier				    pd->naddr.v4.s_addr, 0);
4292126258Smlaier				break;
4293126258Smlaier#endif /* INET */
4294126258Smlaier#ifdef INET6
4295126258Smlaier			case AF_INET6:
4296130613Smlaier				PF_ACPY(saddr, &pd->naddr, af);
4297126258Smlaier				break;
4298126258Smlaier#endif /* INET6 */
4299126258Smlaier			}
4300130613Smlaier			if (nr->natpass)
4301126258Smlaier				r = NULL;
4302130613Smlaier			pd->nat_rule = nr;
4303126258Smlaier		}
4304126258Smlaier	} else {
4305126258Smlaier		/* check incoming packet for BINAT/RDR */
4306130613Smlaier		if ((nr = pf_get_translation(pd, m, off, PF_IN, kif, &nsn,
4307130613Smlaier		    saddr, 0, daddr, 0, &pd->naddr, NULL)) != NULL) {
4308130613Smlaier			PF_ACPY(&pd->baddr, daddr, af);
4309126258Smlaier			switch (af) {
4310126258Smlaier#ifdef INET
4311126258Smlaier			case AF_INET:
4312126258Smlaier				pf_change_a(&daddr->v4.s_addr,
4313130613Smlaier				    pd->ip_sum, pd->naddr.v4.s_addr, 0);
4314126258Smlaier				break;
4315126258Smlaier#endif /* INET */
4316126258Smlaier#ifdef INET6
4317126258Smlaier			case AF_INET6:
4318130613Smlaier				PF_ACPY(daddr, &pd->naddr, af);
4319126258Smlaier				break;
4320126258Smlaier#endif /* INET6 */
4321126258Smlaier			}
4322130613Smlaier			if (nr->natpass)
4323126258Smlaier				r = NULL;
4324130613Smlaier			pd->nat_rule = nr;
4325126258Smlaier		}
4326126258Smlaier	}
4327126258Smlaier
4328126258Smlaier	while (r != NULL) {
4329126258Smlaier		r->evaluations++;
4330171168Smlaier		if (pfi_kif_match(r->kif, kif) == r->ifnot)
4331126258Smlaier			r = r->skip[PF_SKIP_IFP].ptr;
4332126258Smlaier		else if (r->direction && r->direction != direction)
4333126258Smlaier			r = r->skip[PF_SKIP_DIR].ptr;
4334126258Smlaier		else if (r->af && r->af != af)
4335126258Smlaier			r = r->skip[PF_SKIP_AF].ptr;
4336126258Smlaier		else if (r->proto && r->proto != pd->proto)
4337126258Smlaier			r = r->skip[PF_SKIP_PROTO].ptr;
4338171168Smlaier		else if (PF_MISMATCHAW(&r->src.addr, pd->src, af,
4339171168Smlaier		    r->src.neg, kif))
4340126258Smlaier			r = r->skip[PF_SKIP_SRC_ADDR].ptr;
4341171168Smlaier		else if (PF_MISMATCHAW(&r->dst.addr, pd->dst, af,
4342171168Smlaier		    r->dst.neg, NULL))
4343126258Smlaier			r = r->skip[PF_SKIP_DST_ADDR].ptr;
4344171168Smlaier		else if (r->tos && !(r->tos == pd->tos))
4345126258Smlaier			r = TAILQ_NEXT(r, entries);
4346126258Smlaier		else if (r->rule_flag & PFRULE_FRAGMENT)
4347126258Smlaier			r = TAILQ_NEXT(r, entries);
4348145836Smlaier		else if (r->prob && r->prob <= arc4random())
4349126258Smlaier			r = TAILQ_NEXT(r, entries);
4350171168Smlaier		else if (r->match_tag && !pf_match_tag(m, r, pd->pf_mtag, &tag))
4351126258Smlaier			r = TAILQ_NEXT(r, entries);
4352126258Smlaier		else if (r->os_fingerprint != PF_OSFP_ANY)
4353126258Smlaier			r = TAILQ_NEXT(r, entries);
4354126258Smlaier		else {
4355126258Smlaier			if (r->tag)
4356126258Smlaier				tag = r->tag;
4357171168Smlaier			if (r->rtableid >= 0)
4358171168Smlaier				rtableid = r->rtableid;
4359126258Smlaier			if (r->anchor == NULL) {
4360171168Smlaier				match = 1;
4361126258Smlaier				*rm = r;
4362126258Smlaier				*am = a;
4363126258Smlaier				*rsm = ruleset;
4364126258Smlaier				if ((*rm)->quick)
4365126258Smlaier					break;
4366126258Smlaier				r = TAILQ_NEXT(r, entries);
4367126258Smlaier			} else
4368145836Smlaier				pf_step_into_anchor(&asd, &ruleset,
4369171168Smlaier				    PF_RULESET_FILTER, &r, &a, &match);
4370126258Smlaier		}
4371171168Smlaier		if (r == NULL && pf_step_out_of_anchor(&asd, &ruleset,
4372171168Smlaier		    PF_RULESET_FILTER, &r, &a, &match))
4373171168Smlaier			break;
4374126258Smlaier	}
4375126258Smlaier	r = *rm;
4376126258Smlaier	a = *am;
4377126258Smlaier	ruleset = *rsm;
4378126258Smlaier
4379126258Smlaier	REASON_SET(&reason, PFRES_MATCH);
4380130613Smlaier
4381171168Smlaier	if (r->log || (nr != NULL && nr->natpass && nr->log))
4382171168Smlaier		PFLOG_PACKET(kif, h, m, af, direction, reason, r->log ? r : nr,
4383171168Smlaier		    a, ruleset, pd);
4384126258Smlaier
4385126258Smlaier	if ((r->action == PF_DROP) &&
4386126258Smlaier	    ((r->rule_flag & PFRULE_RETURNICMP) ||
4387126258Smlaier	    (r->rule_flag & PFRULE_RETURN))) {
4388126258Smlaier		struct pf_addr *a = NULL;
4389126258Smlaier
4390130613Smlaier		if (nr != NULL) {
4391130613Smlaier			if (direction == PF_OUT)
4392130613Smlaier				a = saddr;
4393130613Smlaier			else
4394130613Smlaier				a = daddr;
4395130613Smlaier		}
4396126258Smlaier		if (a != NULL) {
4397126258Smlaier			switch (af) {
4398126258Smlaier#ifdef INET
4399126258Smlaier			case AF_INET:
4400126258Smlaier				pf_change_a(&a->v4.s_addr, pd->ip_sum,
4401130613Smlaier				    pd->baddr.v4.s_addr, 0);
4402126258Smlaier				break;
4403126258Smlaier#endif /* INET */
4404126258Smlaier#ifdef INET6
4405126258Smlaier			case AF_INET6:
4406130613Smlaier				PF_ACPY(a, &pd->baddr, af);
4407126258Smlaier				break;
4408126258Smlaier#endif /* INET6 */
4409126258Smlaier			}
4410126258Smlaier		}
4411126258Smlaier		if ((af == AF_INET) && r->return_icmp)
4412126258Smlaier			pf_send_icmp(m, r->return_icmp >> 8,
4413126258Smlaier			    r->return_icmp & 255, af, r);
4414126258Smlaier		else if ((af == AF_INET6) && r->return_icmp6)
4415126258Smlaier			pf_send_icmp(m, r->return_icmp6 >> 8,
4416126258Smlaier			    r->return_icmp6 & 255, af, r);
4417126258Smlaier	}
4418126258Smlaier
4419126258Smlaier	if (r->action != PF_PASS)
4420126258Smlaier		return (PF_DROP);
4421126258Smlaier
4422171168Smlaier	if (pf_tag_packet(m, pd->pf_mtag, tag, rtableid)) {
4423126258Smlaier		REASON_SET(&reason, PFRES_MEMORY);
4424126258Smlaier		return (PF_DROP);
4425126258Smlaier	}
4426126258Smlaier
4427130613Smlaier	if (r->keep_state || nr != NULL) {
4428126258Smlaier		/* create new state */
4429126258Smlaier		struct pf_state	*s = NULL;
4430130613Smlaier		struct pf_src_node *sn = NULL;
4431126258Smlaier
4432130613Smlaier		/* check maximums */
4433145836Smlaier		if (r->max_states && (r->states >= r->max_states)) {
4434145836Smlaier			pf_status.lcounters[LCNT_STATES]++;
4435145836Smlaier			REASON_SET(&reason, PFRES_MAXSTATES);
4436130613Smlaier			goto cleanup;
4437145836Smlaier		}
4438171168Smlaier		/* src node for filter rule */
4439130613Smlaier		if ((r->rule_flag & PFRULE_SRCTRACK ||
4440130613Smlaier		    r->rpool.opts & PF_POOL_STICKYADDR) &&
4441145836Smlaier		    pf_insert_src_node(&sn, r, saddr, af) != 0) {
4442145836Smlaier			REASON_SET(&reason, PFRES_SRCLIMIT);
4443130613Smlaier			goto cleanup;
4444145836Smlaier		}
4445130613Smlaier		/* src node for translation rule */
4446130613Smlaier		if (nr != NULL && (nr->rpool.opts & PF_POOL_STICKYADDR) &&
4447130613Smlaier		    ((direction == PF_OUT &&
4448130613Smlaier		    pf_insert_src_node(&nsn, nr, &pd->baddr, af) != 0) ||
4449145836Smlaier		    (pf_insert_src_node(&nsn, nr, saddr, af) != 0))) {
4450145836Smlaier			REASON_SET(&reason, PFRES_SRCLIMIT);
4451130613Smlaier			goto cleanup;
4452145836Smlaier		}
4453130613Smlaier		s = pool_get(&pf_state_pl, PR_NOWAIT);
4454126258Smlaier		if (s == NULL) {
4455145836Smlaier			REASON_SET(&reason, PFRES_MEMORY);
4456130613Smlaiercleanup:
4457130613Smlaier			if (sn != NULL && sn->states == 0 && sn->expire == 0) {
4458130613Smlaier				RB_REMOVE(pf_src_tree, &tree_src_tracking, sn);
4459130613Smlaier				pf_status.scounters[SCNT_SRC_NODE_REMOVALS]++;
4460130613Smlaier				pf_status.src_nodes--;
4461130613Smlaier				pool_put(&pf_src_tree_pl, sn);
4462130613Smlaier			}
4463130613Smlaier			if (nsn != sn && nsn != NULL && nsn->states == 0 &&
4464130613Smlaier			    nsn->expire == 0) {
4465130613Smlaier				RB_REMOVE(pf_src_tree, &tree_src_tracking, nsn);
4466130613Smlaier				pf_status.scounters[SCNT_SRC_NODE_REMOVALS]++;
4467130613Smlaier				pf_status.src_nodes--;
4468130613Smlaier				pool_put(&pf_src_tree_pl, nsn);
4469130613Smlaier			}
4470126258Smlaier			return (PF_DROP);
4471126258Smlaier		}
4472126258Smlaier		bzero(s, sizeof(*s));
4473126258Smlaier		s->rule.ptr = r;
4474130613Smlaier		s->nat_rule.ptr = nr;
4475126258Smlaier		s->anchor.ptr = a;
4476145836Smlaier		STATE_INC_COUNTERS(s);
4477126258Smlaier		s->allow_opts = r->allow_opts;
4478171168Smlaier		s->log = r->log & PF_LOG_ALL;
4479171168Smlaier		if (nr != NULL)
4480171168Smlaier			s->log |= nr->log & PF_LOG_ALL;
4481126258Smlaier		s->proto = pd->proto;
4482126258Smlaier		s->direction = direction;
4483126258Smlaier		s->af = af;
4484126258Smlaier		if (direction == PF_OUT) {
4485126258Smlaier			PF_ACPY(&s->gwy.addr, saddr, af);
4486126258Smlaier			PF_ACPY(&s->ext.addr, daddr, af);
4487130613Smlaier			if (nr != NULL)
4488130613Smlaier				PF_ACPY(&s->lan.addr, &pd->baddr, af);
4489126258Smlaier			else
4490126258Smlaier				PF_ACPY(&s->lan.addr, &s->gwy.addr, af);
4491126258Smlaier		} else {
4492126258Smlaier			PF_ACPY(&s->lan.addr, daddr, af);
4493126258Smlaier			PF_ACPY(&s->ext.addr, saddr, af);
4494130613Smlaier			if (nr != NULL)
4495130613Smlaier				PF_ACPY(&s->gwy.addr, &pd->baddr, af);
4496126258Smlaier			else
4497126258Smlaier				PF_ACPY(&s->gwy.addr, &s->lan.addr, af);
4498126258Smlaier		}
4499126258Smlaier		s->src.state = PFOTHERS_SINGLE;
4500126258Smlaier		s->dst.state = PFOTHERS_NO_TRAFFIC;
4501126261Smlaier		s->creation = time_second;
4502126261Smlaier		s->expire = time_second;
4503126258Smlaier		s->timeout = PFTM_OTHER_FIRST_PACKET;
4504126258Smlaier		pf_set_rt_ifp(s, saddr);
4505130613Smlaier		if (sn != NULL) {
4506130613Smlaier			s->src_node = sn;
4507130613Smlaier			s->src_node->states++;
4508130613Smlaier		}
4509130613Smlaier		if (nsn != NULL) {
4510130613Smlaier			PF_ACPY(&nsn->raddr, &pd->naddr, af);
4511130613Smlaier			s->nat_src_node = nsn;
4512130613Smlaier			s->nat_src_node->states++;
4513130613Smlaier		}
4514130613Smlaier		if (pf_insert_state(BOUND_IFACE(r, kif), s)) {
4515145836Smlaier			REASON_SET(&reason, PFRES_STATEINS);
4516130613Smlaier			pf_src_tree_remove_state(s);
4517145836Smlaier			STATE_DEC_COUNTERS(s);
4518126258Smlaier			pool_put(&pf_state_pl, s);
4519126258Smlaier			return (PF_DROP);
4520126258Smlaier		} else
4521126258Smlaier			*sm = s;
4522145836Smlaier		if (tag > 0) {
4523145836Smlaier			pf_tag_ref(tag);
4524145836Smlaier			s->tag = tag;
4525145836Smlaier		}
4526126258Smlaier	}
4527126258Smlaier
4528126258Smlaier	return (PF_PASS);
4529126258Smlaier}
4530126258Smlaier
4531126258Smlaierint
4532130613Smlaierpf_test_fragment(struct pf_rule **rm, int direction, struct pfi_kif *kif,
4533126258Smlaier    struct mbuf *m, void *h, struct pf_pdesc *pd, struct pf_rule **am,
4534126258Smlaier    struct pf_ruleset **rsm)
4535126258Smlaier{
4536126258Smlaier	struct pf_rule		*r, *a = NULL;
4537126258Smlaier	struct pf_ruleset	*ruleset = NULL;
4538126258Smlaier	sa_family_t		 af = pd->af;
4539126258Smlaier	u_short			 reason;
4540126258Smlaier	int			 tag = -1;
4541145836Smlaier	int			 asd = 0;
4542171168Smlaier	int			 match = 0;
4543126258Smlaier
4544126258Smlaier	r = TAILQ_FIRST(pf_main_ruleset.rules[PF_RULESET_FILTER].active.ptr);
4545126258Smlaier	while (r != NULL) {
4546126258Smlaier		r->evaluations++;
4547171168Smlaier		if (pfi_kif_match(r->kif, kif) == r->ifnot)
4548126258Smlaier			r = r->skip[PF_SKIP_IFP].ptr;
4549126258Smlaier		else if (r->direction && r->direction != direction)
4550126258Smlaier			r = r->skip[PF_SKIP_DIR].ptr;
4551126258Smlaier		else if (r->af && r->af != af)
4552126258Smlaier			r = r->skip[PF_SKIP_AF].ptr;
4553126258Smlaier		else if (r->proto && r->proto != pd->proto)
4554126258Smlaier			r = r->skip[PF_SKIP_PROTO].ptr;
4555171168Smlaier		else if (PF_MISMATCHAW(&r->src.addr, pd->src, af,
4556171168Smlaier		    r->src.neg, kif))
4557126258Smlaier			r = r->skip[PF_SKIP_SRC_ADDR].ptr;
4558171168Smlaier		else if (PF_MISMATCHAW(&r->dst.addr, pd->dst, af,
4559171168Smlaier		    r->dst.neg, NULL))
4560126258Smlaier			r = r->skip[PF_SKIP_DST_ADDR].ptr;
4561171168Smlaier		else if (r->tos && !(r->tos == pd->tos))
4562126258Smlaier			r = TAILQ_NEXT(r, entries);
4563126258Smlaier		else if (r->src.port_op || r->dst.port_op ||
4564126258Smlaier		    r->flagset || r->type || r->code ||
4565126258Smlaier		    r->os_fingerprint != PF_OSFP_ANY)
4566126258Smlaier			r = TAILQ_NEXT(r, entries);
4567145836Smlaier		else if (r->prob && r->prob <= arc4random())
4568126258Smlaier			r = TAILQ_NEXT(r, entries);
4569171168Smlaier		else if (r->match_tag && !pf_match_tag(m, r, pd->pf_mtag, &tag))
4570126258Smlaier			r = TAILQ_NEXT(r, entries);
4571126258Smlaier		else {
4572126258Smlaier			if (r->anchor == NULL) {
4573171168Smlaier				match = 1;
4574126258Smlaier				*rm = r;
4575126258Smlaier				*am = a;
4576126258Smlaier				*rsm = ruleset;
4577126258Smlaier				if ((*rm)->quick)
4578126258Smlaier					break;
4579126258Smlaier				r = TAILQ_NEXT(r, entries);
4580126258Smlaier			} else
4581145836Smlaier				pf_step_into_anchor(&asd, &ruleset,
4582171168Smlaier				    PF_RULESET_FILTER, &r, &a, &match);
4583126258Smlaier		}
4584171168Smlaier		if (r == NULL && pf_step_out_of_anchor(&asd, &ruleset,
4585171168Smlaier		    PF_RULESET_FILTER, &r, &a, &match))
4586171168Smlaier			break;
4587126258Smlaier	}
4588126258Smlaier	r = *rm;
4589126258Smlaier	a = *am;
4590126258Smlaier	ruleset = *rsm;
4591126258Smlaier
4592126258Smlaier	REASON_SET(&reason, PFRES_MATCH);
4593130613Smlaier
4594126258Smlaier	if (r->log)
4595171168Smlaier		PFLOG_PACKET(kif, h, m, af, direction, reason, r, a, ruleset,
4596171168Smlaier		    pd);
4597126258Smlaier
4598126258Smlaier	if (r->action != PF_PASS)
4599126258Smlaier		return (PF_DROP);
4600126258Smlaier
4601171168Smlaier	if (pf_tag_packet(m, pd->pf_mtag, tag, -1)) {
4602126258Smlaier		REASON_SET(&reason, PFRES_MEMORY);
4603126258Smlaier		return (PF_DROP);
4604126258Smlaier	}
4605126258Smlaier
4606126258Smlaier	return (PF_PASS);
4607126258Smlaier}
4608126258Smlaier
4609126258Smlaierint
4610130613Smlaierpf_test_state_tcp(struct pf_state **state, int direction, struct pfi_kif *kif,
4611130613Smlaier    struct mbuf *m, int off, void *h, struct pf_pdesc *pd,
4612126258Smlaier    u_short *reason)
4613126258Smlaier{
4614171168Smlaier	struct pf_state_cmp	 key;
4615126258Smlaier	struct tcphdr		*th = pd->hdr.tcp;
4616126258Smlaier	u_int16_t		 win = ntohs(th->th_win);
4617145836Smlaier	u_int32_t		 ack, end, seq, orig_seq;
4618126258Smlaier	u_int8_t		 sws, dws;
4619130613Smlaier	int			 ackskew;
4620126258Smlaier	int			 copyback = 0;
4621126258Smlaier	struct pf_state_peer	*src, *dst;
4622126258Smlaier
4623126258Smlaier	key.af = pd->af;
4624126258Smlaier	key.proto = IPPROTO_TCP;
4625130613Smlaier	if (direction == PF_IN)	{
4626130613Smlaier		PF_ACPY(&key.ext.addr, pd->src, key.af);
4627130613Smlaier		PF_ACPY(&key.gwy.addr, pd->dst, key.af);
4628130613Smlaier		key.ext.port = th->th_sport;
4629130613Smlaier		key.gwy.port = th->th_dport;
4630130613Smlaier	} else {
4631130613Smlaier		PF_ACPY(&key.lan.addr, pd->src, key.af);
4632130613Smlaier		PF_ACPY(&key.ext.addr, pd->dst, key.af);
4633130613Smlaier		key.lan.port = th->th_sport;
4634130613Smlaier		key.ext.port = th->th_dport;
4635130613Smlaier	}
4636126258Smlaier
4637126258Smlaier	STATE_LOOKUP();
4638126258Smlaier
4639126258Smlaier	if (direction == (*state)->direction) {
4640126258Smlaier		src = &(*state)->src;
4641126258Smlaier		dst = &(*state)->dst;
4642126258Smlaier	} else {
4643126258Smlaier		src = &(*state)->dst;
4644126258Smlaier		dst = &(*state)->src;
4645126258Smlaier	}
4646126258Smlaier
4647126258Smlaier	if ((*state)->src.state == PF_TCPS_PROXY_SRC) {
4648145836Smlaier		if (direction != (*state)->direction) {
4649145836Smlaier			REASON_SET(reason, PFRES_SYNPROXY);
4650126258Smlaier			return (PF_SYNPROXY_DROP);
4651145836Smlaier		}
4652126258Smlaier		if (th->th_flags & TH_SYN) {
4653145836Smlaier			if (ntohl(th->th_seq) != (*state)->src.seqlo) {
4654145836Smlaier				REASON_SET(reason, PFRES_SYNPROXY);
4655126258Smlaier				return (PF_DROP);
4656145836Smlaier			}
4657162238Scsjp#ifdef __FreeBSD__
4658162238Scsjp			pf_send_tcp(NULL, (*state)->rule.ptr, pd->af, pd->dst,
4659162238Scsjp#else
4660126258Smlaier			pf_send_tcp((*state)->rule.ptr, pd->af, pd->dst,
4661162238Scsjp#endif
4662126258Smlaier			    pd->src, th->th_dport, th->th_sport,
4663126258Smlaier			    (*state)->src.seqhi, ntohl(th->th_seq) + 1,
4664145836Smlaier			    TH_SYN|TH_ACK, 0, (*state)->src.mss, 0, 1,
4665171168Smlaier			    0, NULL, NULL);
4666145836Smlaier			REASON_SET(reason, PFRES_SYNPROXY);
4667126258Smlaier			return (PF_SYNPROXY_DROP);
4668126258Smlaier		} else if (!(th->th_flags & TH_ACK) ||
4669126258Smlaier		    (ntohl(th->th_ack) != (*state)->src.seqhi + 1) ||
4670145836Smlaier		    (ntohl(th->th_seq) != (*state)->src.seqlo + 1)) {
4671145836Smlaier			REASON_SET(reason, PFRES_SYNPROXY);
4672126258Smlaier			return (PF_DROP);
4673145836Smlaier		} else if ((*state)->src_node != NULL &&
4674145836Smlaier		    pf_src_connlimit(state)) {
4675145836Smlaier			REASON_SET(reason, PFRES_SRCLIMIT);
4676145836Smlaier			return (PF_DROP);
4677145836Smlaier		} else
4678126258Smlaier			(*state)->src.state = PF_TCPS_PROXY_DST;
4679126258Smlaier	}
4680126258Smlaier	if ((*state)->src.state == PF_TCPS_PROXY_DST) {
4681126258Smlaier		struct pf_state_host *src, *dst;
4682126258Smlaier
4683126258Smlaier		if (direction == PF_OUT) {
4684126258Smlaier			src = &(*state)->gwy;
4685126258Smlaier			dst = &(*state)->ext;
4686126258Smlaier		} else {
4687126258Smlaier			src = &(*state)->ext;
4688126258Smlaier			dst = &(*state)->lan;
4689126258Smlaier		}
4690126258Smlaier		if (direction == (*state)->direction) {
4691126258Smlaier			if (((th->th_flags & (TH_SYN|TH_ACK)) != TH_ACK) ||
4692126258Smlaier			    (ntohl(th->th_ack) != (*state)->src.seqhi + 1) ||
4693145836Smlaier			    (ntohl(th->th_seq) != (*state)->src.seqlo + 1)) {
4694145836Smlaier				REASON_SET(reason, PFRES_SYNPROXY);
4695126258Smlaier				return (PF_DROP);
4696145836Smlaier			}
4697126258Smlaier			(*state)->src.max_win = MAX(ntohs(th->th_win), 1);
4698126258Smlaier			if ((*state)->dst.seqhi == 1)
4699145836Smlaier				(*state)->dst.seqhi = htonl(arc4random());
4700162238Scsjp#ifdef __FreeBSD__
4701162238Scsjp			pf_send_tcp(NULL, (*state)->rule.ptr, pd->af,
4702162238Scsjp			    &src->addr,
4703162238Scsjp#else
4704126258Smlaier			pf_send_tcp((*state)->rule.ptr, pd->af, &src->addr,
4705162238Scsjp#endif
4706126258Smlaier			    &dst->addr, src->port, dst->port,
4707130613Smlaier			    (*state)->dst.seqhi, 0, TH_SYN, 0,
4708171168Smlaier			    (*state)->src.mss, 0, 0, (*state)->tag, NULL, NULL);
4709145836Smlaier			REASON_SET(reason, PFRES_SYNPROXY);
4710126258Smlaier			return (PF_SYNPROXY_DROP);
4711126258Smlaier		} else if (((th->th_flags & (TH_SYN|TH_ACK)) !=
4712126258Smlaier		    (TH_SYN|TH_ACK)) ||
4713145836Smlaier		    (ntohl(th->th_ack) != (*state)->dst.seqhi + 1)) {
4714145836Smlaier			REASON_SET(reason, PFRES_SYNPROXY);
4715126258Smlaier			return (PF_DROP);
4716145836Smlaier		} else {
4717126258Smlaier			(*state)->dst.max_win = MAX(ntohs(th->th_win), 1);
4718126258Smlaier			(*state)->dst.seqlo = ntohl(th->th_seq);
4719162238Scsjp#ifdef __FreeBSD__
4720162238Scsjp			pf_send_tcp(NULL, (*state)->rule.ptr, pd->af, pd->dst,
4721162238Scsjp#else
4722126258Smlaier			pf_send_tcp((*state)->rule.ptr, pd->af, pd->dst,
4723162238Scsjp#endif
4724126258Smlaier			    pd->src, th->th_dport, th->th_sport,
4725126258Smlaier			    ntohl(th->th_ack), ntohl(th->th_seq) + 1,
4726145836Smlaier			    TH_ACK, (*state)->src.max_win, 0, 0, 0,
4727171168Smlaier			    (*state)->tag, NULL, NULL);
4728162238Scsjp#ifdef __FreeBSD__
4729162238Scsjp			pf_send_tcp(NULL, (*state)->rule.ptr, pd->af,
4730162238Scsjp			    &src->addr,
4731162238Scsjp#else
4732126258Smlaier			pf_send_tcp((*state)->rule.ptr, pd->af, &src->addr,
4733162238Scsjp#endif
4734126258Smlaier			    &dst->addr, src->port, dst->port,
4735126258Smlaier			    (*state)->src.seqhi + 1, (*state)->src.seqlo + 1,
4736145836Smlaier			    TH_ACK, (*state)->dst.max_win, 0, 0, 1,
4737171168Smlaier			    0, NULL, NULL);
4738126258Smlaier			(*state)->src.seqdiff = (*state)->dst.seqhi -
4739126258Smlaier			    (*state)->src.seqlo;
4740126258Smlaier			(*state)->dst.seqdiff = (*state)->src.seqhi -
4741126258Smlaier			    (*state)->dst.seqlo;
4742126258Smlaier			(*state)->src.seqhi = (*state)->src.seqlo +
4743145875Smlaier			    (*state)->dst.max_win;
4744145875Smlaier			(*state)->dst.seqhi = (*state)->dst.seqlo +
4745145836Smlaier			    (*state)->src.max_win;
4746126258Smlaier			(*state)->src.wscale = (*state)->dst.wscale = 0;
4747126258Smlaier			(*state)->src.state = (*state)->dst.state =
4748126258Smlaier			    TCPS_ESTABLISHED;
4749145836Smlaier			REASON_SET(reason, PFRES_SYNPROXY);
4750126258Smlaier			return (PF_SYNPROXY_DROP);
4751126258Smlaier		}
4752126258Smlaier	}
4753126258Smlaier
4754126258Smlaier	if (src->wscale && dst->wscale && !(th->th_flags & TH_SYN)) {
4755126258Smlaier		sws = src->wscale & PF_WSCALE_MASK;
4756126258Smlaier		dws = dst->wscale & PF_WSCALE_MASK;
4757126258Smlaier	} else
4758126258Smlaier		sws = dws = 0;
4759126258Smlaier
4760126258Smlaier	/*
4761126258Smlaier	 * Sequence tracking algorithm from Guido van Rooij's paper:
4762126258Smlaier	 *   http://www.madison-gurkha.com/publications/tcp_filtering/
4763126258Smlaier	 *	tcp_filtering.ps
4764126258Smlaier	 */
4765126258Smlaier
4766145836Smlaier	orig_seq = seq = ntohl(th->th_seq);
4767126258Smlaier	if (src->seqlo == 0) {
4768126258Smlaier		/* First packet from this end. Set its state */
4769126258Smlaier
4770126258Smlaier		if ((pd->flags & PFDESC_TCP_NORM || dst->scrub) &&
4771126258Smlaier		    src->scrub == NULL) {
4772126258Smlaier			if (pf_normalize_tcp_init(m, off, pd, th, src, dst)) {
4773126258Smlaier				REASON_SET(reason, PFRES_MEMORY);
4774126258Smlaier				return (PF_DROP);
4775126258Smlaier			}
4776126258Smlaier		}
4777126258Smlaier
4778126258Smlaier		/* Deferred generation of sequence number modulator */
4779126258Smlaier		if (dst->seqdiff && !src->seqdiff) {
4780171168Smlaier#ifdef __FreeBSD__
4781171168Smlaier			while ((src->seqdiff = pf_new_isn(*state) - seq) == 0)
4782126258Smlaier				;
4783171168Smlaier#else
4784171168Smlaier			while ((src->seqdiff = tcp_rndiss_next() - seq) == 0)
4785171168Smlaier				;
4786171168Smlaier#endif
4787126258Smlaier			ack = ntohl(th->th_ack) - dst->seqdiff;
4788126258Smlaier			pf_change_a(&th->th_seq, &th->th_sum, htonl(seq +
4789126258Smlaier			    src->seqdiff), 0);
4790126258Smlaier			pf_change_a(&th->th_ack, &th->th_sum, htonl(ack), 0);
4791126258Smlaier			copyback = 1;
4792126258Smlaier		} else {
4793126258Smlaier			ack = ntohl(th->th_ack);
4794126258Smlaier		}
4795126258Smlaier
4796126258Smlaier		end = seq + pd->p_len;
4797126258Smlaier		if (th->th_flags & TH_SYN) {
4798126258Smlaier			end++;
4799126258Smlaier			if (dst->wscale & PF_WSCALE_FLAG) {
4800126258Smlaier				src->wscale = pf_get_wscale(m, off, th->th_off,
4801126258Smlaier				    pd->af);
4802126258Smlaier				if (src->wscale & PF_WSCALE_FLAG) {
4803126258Smlaier					/* Remove scale factor from initial
4804126258Smlaier					 * window */
4805126258Smlaier					sws = src->wscale & PF_WSCALE_MASK;
4806126258Smlaier					win = ((u_int32_t)win + (1 << sws) - 1)
4807126258Smlaier					    >> sws;
4808126258Smlaier					dws = dst->wscale & PF_WSCALE_MASK;
4809126258Smlaier				} else {
4810126258Smlaier					/* fixup other window */
4811126258Smlaier					dst->max_win <<= dst->wscale &
4812126258Smlaier					    PF_WSCALE_MASK;
4813126258Smlaier					/* in case of a retrans SYN|ACK */
4814126258Smlaier					dst->wscale = 0;
4815126258Smlaier				}
4816126258Smlaier			}
4817126258Smlaier		}
4818126258Smlaier		if (th->th_flags & TH_FIN)
4819126258Smlaier			end++;
4820126258Smlaier
4821126258Smlaier		src->seqlo = seq;
4822126258Smlaier		if (src->state < TCPS_SYN_SENT)
4823126258Smlaier			src->state = TCPS_SYN_SENT;
4824126258Smlaier
4825126258Smlaier		/*
4826126258Smlaier		 * May need to slide the window (seqhi may have been set by
4827126258Smlaier		 * the crappy stack check or if we picked up the connection
4828126258Smlaier		 * after establishment)
4829126258Smlaier		 */
4830126258Smlaier		if (src->seqhi == 1 ||
4831126258Smlaier		    SEQ_GEQ(end + MAX(1, dst->max_win << dws), src->seqhi))
4832126258Smlaier			src->seqhi = end + MAX(1, dst->max_win << dws);
4833126258Smlaier		if (win > src->max_win)
4834126258Smlaier			src->max_win = win;
4835126258Smlaier
4836126258Smlaier	} else {
4837126258Smlaier		ack = ntohl(th->th_ack) - dst->seqdiff;
4838126258Smlaier		if (src->seqdiff) {
4839126258Smlaier			/* Modulate sequence numbers */
4840126258Smlaier			pf_change_a(&th->th_seq, &th->th_sum, htonl(seq +
4841126258Smlaier			    src->seqdiff), 0);
4842126258Smlaier			pf_change_a(&th->th_ack, &th->th_sum, htonl(ack), 0);
4843126258Smlaier			copyback = 1;
4844126258Smlaier		}
4845126258Smlaier		end = seq + pd->p_len;
4846126258Smlaier		if (th->th_flags & TH_SYN)
4847126258Smlaier			end++;
4848126258Smlaier		if (th->th_flags & TH_FIN)
4849126258Smlaier			end++;
4850126258Smlaier	}
4851126258Smlaier
4852126258Smlaier	if ((th->th_flags & TH_ACK) == 0) {
4853126258Smlaier		/* Let it pass through the ack skew check */
4854126258Smlaier		ack = dst->seqlo;
4855126258Smlaier	} else if ((ack == 0 &&
4856126258Smlaier	    (th->th_flags & (TH_ACK|TH_RST)) == (TH_ACK|TH_RST)) ||
4857126258Smlaier	    /* broken tcp stacks do not set ack */
4858126258Smlaier	    (dst->state < TCPS_SYN_SENT)) {
4859126258Smlaier		/*
4860126258Smlaier		 * Many stacks (ours included) will set the ACK number in an
4861126258Smlaier		 * FIN|ACK if the SYN times out -- no sequence to ACK.
4862126258Smlaier		 */
4863126258Smlaier		ack = dst->seqlo;
4864126258Smlaier	}
4865126258Smlaier
4866126258Smlaier	if (seq == end) {
4867126258Smlaier		/* Ease sequencing restrictions on no data packets */
4868126258Smlaier		seq = src->seqlo;
4869126258Smlaier		end = seq;
4870126258Smlaier	}
4871126258Smlaier
4872126258Smlaier	ackskew = dst->seqlo - ack;
4873126258Smlaier
4874171168Smlaier
4875171168Smlaier	/*
4876171168Smlaier	 * Need to demodulate the sequence numbers in any TCP SACK options
4877171168Smlaier	 * (Selective ACK). We could optionally validate the SACK values
4878171168Smlaier	 * against the current ACK window, either forwards or backwards, but
4879171168Smlaier	 * I'm not confident that SACK has been implemented properly
4880171168Smlaier	 * everywhere. It wouldn't surprise me if several stacks accidently
4881171168Smlaier	 * SACK too far backwards of previously ACKed data. There really aren't
4882171168Smlaier	 * any security implications of bad SACKing unless the target stack
4883171168Smlaier	 * doesn't validate the option length correctly. Someone trying to
4884171168Smlaier	 * spoof into a TCP connection won't bother blindly sending SACK
4885171168Smlaier	 * options anyway.
4886171168Smlaier	 */
4887171168Smlaier	if (dst->seqdiff && (th->th_off << 2) > sizeof(struct tcphdr)) {
4888171168Smlaier		if (pf_modulate_sack(m, off, pd, th, dst))
4889171168Smlaier			copyback = 1;
4890171168Smlaier	}
4891171168Smlaier
4892171168Smlaier
4893126258Smlaier#define MAXACKWINDOW (0xffff + 1500)	/* 1500 is an arbitrary fudge factor */
4894126258Smlaier	if (SEQ_GEQ(src->seqhi, end) &&
4895126258Smlaier	    /* Last octet inside other's window space */
4896126258Smlaier	    SEQ_GEQ(seq, src->seqlo - (dst->max_win << dws)) &&
4897126258Smlaier	    /* Retrans: not more than one window back */
4898126258Smlaier	    (ackskew >= -MAXACKWINDOW) &&
4899126258Smlaier	    /* Acking not more than one reassembled fragment backwards */
4900145836Smlaier	    (ackskew <= (MAXACKWINDOW << sws)) &&
4901126258Smlaier	    /* Acking not more than one window forward */
4902145836Smlaier	    ((th->th_flags & TH_RST) == 0 || orig_seq == src->seqlo ||
4903171168Smlaier	    (orig_seq == src->seqlo + 1) || (pd->flags & PFDESC_IP_REAS) == 0)) {
4904171168Smlaier	    /* Require an exact/+1 sequence match on resets when possible */
4905126258Smlaier
4906145836Smlaier		if (dst->scrub || src->scrub) {
4907145836Smlaier			if (pf_normalize_tcp_stateful(m, off, pd, reason, th,
4908145836Smlaier			    *state, src, dst, &copyback))
4909145836Smlaier				return (PF_DROP);
4910145836Smlaier		}
4911145836Smlaier
4912126258Smlaier		/* update max window */
4913126258Smlaier		if (src->max_win < win)
4914126258Smlaier			src->max_win = win;
4915126258Smlaier		/* synchronize sequencing */
4916126258Smlaier		if (SEQ_GT(end, src->seqlo))
4917126258Smlaier			src->seqlo = end;
4918126258Smlaier		/* slide the window of what the other end can send */
4919126258Smlaier		if (SEQ_GEQ(ack + (win << sws), dst->seqhi))
4920126258Smlaier			dst->seqhi = ack + MAX((win << sws), 1);
4921126258Smlaier
4922126258Smlaier
4923126258Smlaier		/* update states */
4924126258Smlaier		if (th->th_flags & TH_SYN)
4925126258Smlaier			if (src->state < TCPS_SYN_SENT)
4926126258Smlaier				src->state = TCPS_SYN_SENT;
4927126258Smlaier		if (th->th_flags & TH_FIN)
4928126258Smlaier			if (src->state < TCPS_CLOSING)
4929126258Smlaier				src->state = TCPS_CLOSING;
4930126258Smlaier		if (th->th_flags & TH_ACK) {
4931145836Smlaier			if (dst->state == TCPS_SYN_SENT) {
4932126258Smlaier				dst->state = TCPS_ESTABLISHED;
4933145836Smlaier				if (src->state == TCPS_ESTABLISHED &&
4934145836Smlaier				    (*state)->src_node != NULL &&
4935145836Smlaier				    pf_src_connlimit(state)) {
4936145836Smlaier					REASON_SET(reason, PFRES_SRCLIMIT);
4937145836Smlaier					return (PF_DROP);
4938145836Smlaier				}
4939145836Smlaier			} else if (dst->state == TCPS_CLOSING)
4940126258Smlaier				dst->state = TCPS_FIN_WAIT_2;
4941126258Smlaier		}
4942126258Smlaier		if (th->th_flags & TH_RST)
4943126258Smlaier			src->state = dst->state = TCPS_TIME_WAIT;
4944126258Smlaier
4945126258Smlaier		/* update expire time */
4946126261Smlaier		(*state)->expire = time_second;
4947126258Smlaier		if (src->state >= TCPS_FIN_WAIT_2 &&
4948126258Smlaier		    dst->state >= TCPS_FIN_WAIT_2)
4949126258Smlaier			(*state)->timeout = PFTM_TCP_CLOSED;
4950171168Smlaier		else if (src->state >= TCPS_CLOSING &&
4951171168Smlaier		    dst->state >= TCPS_CLOSING)
4952126258Smlaier			(*state)->timeout = PFTM_TCP_FIN_WAIT;
4953126258Smlaier		else if (src->state < TCPS_ESTABLISHED ||
4954126258Smlaier		    dst->state < TCPS_ESTABLISHED)
4955126258Smlaier			(*state)->timeout = PFTM_TCP_OPENING;
4956126258Smlaier		else if (src->state >= TCPS_CLOSING ||
4957126258Smlaier		    dst->state >= TCPS_CLOSING)
4958126258Smlaier			(*state)->timeout = PFTM_TCP_CLOSING;
4959126258Smlaier		else
4960126258Smlaier			(*state)->timeout = PFTM_TCP_ESTABLISHED;
4961126258Smlaier
4962126258Smlaier		/* Fall through to PASS packet */
4963126258Smlaier
4964126258Smlaier	} else if ((dst->state < TCPS_SYN_SENT ||
4965126258Smlaier		dst->state >= TCPS_FIN_WAIT_2 ||
4966126258Smlaier		src->state >= TCPS_FIN_WAIT_2) &&
4967126258Smlaier	    SEQ_GEQ(src->seqhi + MAXACKWINDOW, end) &&
4968126258Smlaier	    /* Within a window forward of the originating packet */
4969126258Smlaier	    SEQ_GEQ(seq, src->seqlo - MAXACKWINDOW)) {
4970126258Smlaier	    /* Within a window backward of the originating packet */
4971126258Smlaier
4972126258Smlaier		/*
4973126258Smlaier		 * This currently handles three situations:
4974126258Smlaier		 *  1) Stupid stacks will shotgun SYNs before their peer
4975126258Smlaier		 *     replies.
4976126258Smlaier		 *  2) When PF catches an already established stream (the
4977126258Smlaier		 *     firewall rebooted, the state table was flushed, routes
4978126258Smlaier		 *     changed...)
4979126258Smlaier		 *  3) Packets get funky immediately after the connection
4980126258Smlaier		 *     closes (this should catch Solaris spurious ACK|FINs
4981126258Smlaier		 *     that web servers like to spew after a close)
4982126258Smlaier		 *
4983126258Smlaier		 * This must be a little more careful than the above code
4984126258Smlaier		 * since packet floods will also be caught here. We don't
4985126258Smlaier		 * update the TTL here to mitigate the damage of a packet
4986126258Smlaier		 * flood and so the same code can handle awkward establishment
4987126258Smlaier		 * and a loosened connection close.
4988126258Smlaier		 * In the establishment case, a correct peer response will
4989126258Smlaier		 * validate the connection, go through the normal state code
4990126258Smlaier		 * and keep updating the state TTL.
4991126258Smlaier		 */
4992126258Smlaier
4993126258Smlaier		if (pf_status.debug >= PF_DEBUG_MISC) {
4994126258Smlaier			printf("pf: loose state match: ");
4995126258Smlaier			pf_print_state(*state);
4996126258Smlaier			pf_print_flags(th->th_flags);
4997171168Smlaier			printf(" seq=%u (%u) ack=%u len=%u ackskew=%d "
4998171168Smlaier			    "pkts=%llu:%llu\n", seq, orig_seq, ack, pd->p_len,
4999171168Smlaier#ifdef __FreeBSD__
5000171168Smlaier			    ackskew, (unsigned long long)(*state)->packets[0],
5001171168Smlaier			    (unsigned long long)(*state)->packets[1]);
5002171168Smlaier#else
5003171168Smlaier			    ackskew, (*state)->packets[0],
5004171168Smlaier			    (*state)->packets[1]);
5005171168Smlaier#endif
5006126258Smlaier		}
5007126258Smlaier
5008145836Smlaier		if (dst->scrub || src->scrub) {
5009145836Smlaier			if (pf_normalize_tcp_stateful(m, off, pd, reason, th,
5010145836Smlaier			    *state, src, dst, &copyback))
5011145836Smlaier				return (PF_DROP);
5012145836Smlaier		}
5013145836Smlaier
5014126258Smlaier		/* update max window */
5015126258Smlaier		if (src->max_win < win)
5016126258Smlaier			src->max_win = win;
5017126258Smlaier		/* synchronize sequencing */
5018126258Smlaier		if (SEQ_GT(end, src->seqlo))
5019126258Smlaier			src->seqlo = end;
5020126258Smlaier		/* slide the window of what the other end can send */
5021126258Smlaier		if (SEQ_GEQ(ack + (win << sws), dst->seqhi))
5022126258Smlaier			dst->seqhi = ack + MAX((win << sws), 1);
5023126258Smlaier
5024126258Smlaier		/*
5025126258Smlaier		 * Cannot set dst->seqhi here since this could be a shotgunned
5026126258Smlaier		 * SYN and not an already established connection.
5027126258Smlaier		 */
5028126258Smlaier
5029126258Smlaier		if (th->th_flags & TH_FIN)
5030126258Smlaier			if (src->state < TCPS_CLOSING)
5031126258Smlaier				src->state = TCPS_CLOSING;
5032126258Smlaier		if (th->th_flags & TH_RST)
5033126258Smlaier			src->state = dst->state = TCPS_TIME_WAIT;
5034126258Smlaier
5035126258Smlaier		/* Fall through to PASS packet */
5036126258Smlaier
5037126258Smlaier	} else {
5038126258Smlaier		if ((*state)->dst.state == TCPS_SYN_SENT &&
5039126258Smlaier		    (*state)->src.state == TCPS_SYN_SENT) {
5040126258Smlaier			/* Send RST for state mismatches during handshake */
5041145836Smlaier			if (!(th->th_flags & TH_RST))
5042162238Scsjp#ifdef __FreeBSD__
5043162238Scsjp				pf_send_tcp(m, (*state)->rule.ptr, pd->af,
5044162238Scsjp#else
5045126258Smlaier				pf_send_tcp((*state)->rule.ptr, pd->af,
5046162238Scsjp#endif
5047126258Smlaier				    pd->dst, pd->src, th->th_dport,
5048145836Smlaier				    th->th_sport, ntohl(th->th_ack), 0,
5049145836Smlaier				    TH_RST, 0, 0,
5050171168Smlaier				    (*state)->rule.ptr->return_ttl, 1, 0,
5051145836Smlaier				    pd->eh, kif->pfik_ifp);
5052126258Smlaier			src->seqlo = 0;
5053126258Smlaier			src->seqhi = 1;
5054126258Smlaier			src->max_win = 1;
5055126258Smlaier		} else if (pf_status.debug >= PF_DEBUG_MISC) {
5056126258Smlaier			printf("pf: BAD state: ");
5057126258Smlaier			pf_print_state(*state);
5058126258Smlaier			pf_print_flags(th->th_flags);
5059171168Smlaier			printf(" seq=%u (%u) ack=%u len=%u ackskew=%d "
5060171168Smlaier			    "pkts=%llu:%llu dir=%s,%s\n",
5061171168Smlaier			    seq, orig_seq, ack, pd->p_len, ackskew,
5062171168Smlaier#ifdef __FreeBSD__
5063171168Smlaier			    (unsigned long long)(*state)->packets[0],
5064171168Smlaier			    (unsigned long long)(*state)->packets[1],
5065171168Smlaier#else
5066126258Smlaier			    (*state)->packets[0], (*state)->packets[1],
5067171168Smlaier#endif
5068126258Smlaier			    direction == PF_IN ? "in" : "out",
5069126258Smlaier			    direction == (*state)->direction ? "fwd" : "rev");
5070126258Smlaier			printf("pf: State failure on: %c %c %c %c | %c %c\n",
5071126258Smlaier			    SEQ_GEQ(src->seqhi, end) ? ' ' : '1',
5072126258Smlaier			    SEQ_GEQ(seq, src->seqlo - (dst->max_win << dws)) ?
5073126258Smlaier			    ' ': '2',
5074126258Smlaier			    (ackskew >= -MAXACKWINDOW) ? ' ' : '3',
5075126258Smlaier			    (ackskew <= (MAXACKWINDOW << sws)) ? ' ' : '4',
5076126258Smlaier			    SEQ_GEQ(src->seqhi + MAXACKWINDOW, end) ?' ' :'5',
5077126258Smlaier			    SEQ_GEQ(seq, src->seqlo - MAXACKWINDOW) ?' ' :'6');
5078126258Smlaier		}
5079145836Smlaier		REASON_SET(reason, PFRES_BADSTATE);
5080126258Smlaier		return (PF_DROP);
5081126258Smlaier	}
5082126258Smlaier
5083126258Smlaier	/* Any packets which have gotten here are to be passed */
5084126258Smlaier
5085126258Smlaier	/* translate source/destination address, if necessary */
5086126258Smlaier	if (STATE_TRANSLATE(*state)) {
5087126258Smlaier		if (direction == PF_OUT)
5088126258Smlaier			pf_change_ap(pd->src, &th->th_sport, pd->ip_sum,
5089126258Smlaier			    &th->th_sum, &(*state)->gwy.addr,
5090126258Smlaier			    (*state)->gwy.port, 0, pd->af);
5091126258Smlaier		else
5092126258Smlaier			pf_change_ap(pd->dst, &th->th_dport, pd->ip_sum,
5093126258Smlaier			    &th->th_sum, &(*state)->lan.addr,
5094126258Smlaier			    (*state)->lan.port, 0, pd->af);
5095126261Smlaier		m_copyback(m, off, sizeof(*th), (caddr_t)th);
5096126258Smlaier	} else if (copyback) {
5097126258Smlaier		/* Copyback sequence modulation or stateful scrub changes */
5098126261Smlaier		m_copyback(m, off, sizeof(*th), (caddr_t)th);
5099126258Smlaier	}
5100126258Smlaier
5101126258Smlaier	return (PF_PASS);
5102126258Smlaier}
5103126258Smlaier
5104126258Smlaierint
5105130613Smlaierpf_test_state_udp(struct pf_state **state, int direction, struct pfi_kif *kif,
5106130613Smlaier    struct mbuf *m, int off, void *h, struct pf_pdesc *pd)
5107126258Smlaier{
5108126258Smlaier	struct pf_state_peer	*src, *dst;
5109171168Smlaier	struct pf_state_cmp	 key;
5110126258Smlaier	struct udphdr		*uh = pd->hdr.udp;
5111126258Smlaier
5112126258Smlaier	key.af = pd->af;
5113126258Smlaier	key.proto = IPPROTO_UDP;
5114130613Smlaier	if (direction == PF_IN)	{
5115130613Smlaier		PF_ACPY(&key.ext.addr, pd->src, key.af);
5116130613Smlaier		PF_ACPY(&key.gwy.addr, pd->dst, key.af);
5117130613Smlaier		key.ext.port = uh->uh_sport;
5118130613Smlaier		key.gwy.port = uh->uh_dport;
5119130613Smlaier	} else {
5120130613Smlaier		PF_ACPY(&key.lan.addr, pd->src, key.af);
5121130613Smlaier		PF_ACPY(&key.ext.addr, pd->dst, key.af);
5122130613Smlaier		key.lan.port = uh->uh_sport;
5123130613Smlaier		key.ext.port = uh->uh_dport;
5124130613Smlaier	}
5125126258Smlaier
5126126258Smlaier	STATE_LOOKUP();
5127126258Smlaier
5128126258Smlaier	if (direction == (*state)->direction) {
5129126258Smlaier		src = &(*state)->src;
5130126258Smlaier		dst = &(*state)->dst;
5131126258Smlaier	} else {
5132126258Smlaier		src = &(*state)->dst;
5133126258Smlaier		dst = &(*state)->src;
5134126258Smlaier	}
5135126258Smlaier
5136126258Smlaier	/* update states */
5137126258Smlaier	if (src->state < PFUDPS_SINGLE)
5138126258Smlaier		src->state = PFUDPS_SINGLE;
5139126258Smlaier	if (dst->state == PFUDPS_SINGLE)
5140126258Smlaier		dst->state = PFUDPS_MULTIPLE;
5141126258Smlaier
5142126258Smlaier	/* update expire time */
5143126261Smlaier	(*state)->expire = time_second;
5144126258Smlaier	if (src->state == PFUDPS_MULTIPLE && dst->state == PFUDPS_MULTIPLE)
5145126258Smlaier		(*state)->timeout = PFTM_UDP_MULTIPLE;
5146126258Smlaier	else
5147126258Smlaier		(*state)->timeout = PFTM_UDP_SINGLE;
5148126258Smlaier
5149126258Smlaier	/* translate source/destination address, if necessary */
5150126258Smlaier	if (STATE_TRANSLATE(*state)) {
5151126258Smlaier		if (direction == PF_OUT)
5152126258Smlaier			pf_change_ap(pd->src, &uh->uh_sport, pd->ip_sum,
5153126258Smlaier			    &uh->uh_sum, &(*state)->gwy.addr,
5154126258Smlaier			    (*state)->gwy.port, 1, pd->af);
5155126258Smlaier		else
5156126258Smlaier			pf_change_ap(pd->dst, &uh->uh_dport, pd->ip_sum,
5157126258Smlaier			    &uh->uh_sum, &(*state)->lan.addr,
5158126258Smlaier			    (*state)->lan.port, 1, pd->af);
5159126261Smlaier		m_copyback(m, off, sizeof(*uh), (caddr_t)uh);
5160126258Smlaier	}
5161126258Smlaier
5162126258Smlaier	return (PF_PASS);
5163126258Smlaier}
5164126258Smlaier
5165126258Smlaierint
5166130613Smlaierpf_test_state_icmp(struct pf_state **state, int direction, struct pfi_kif *kif,
5167145836Smlaier    struct mbuf *m, int off, void *h, struct pf_pdesc *pd, u_short *reason)
5168126258Smlaier{
5169126258Smlaier	struct pf_addr	*saddr = pd->src, *daddr = pd->dst;
5170127629Smlaier	u_int16_t	 icmpid = 0;		/* make the compiler happy */
5171127629Smlaier	u_int16_t	*icmpsum = NULL;	/* make the compiler happy */
5172127629Smlaier	u_int8_t	 icmptype = 0;		/* make the compiler happy */
5173130613Smlaier	int		 state_icmp = 0;
5174171168Smlaier	struct pf_state_cmp key;
5175126258Smlaier
5176126258Smlaier	switch (pd->proto) {
5177126258Smlaier#ifdef INET
5178126258Smlaier	case IPPROTO_ICMP:
5179126258Smlaier		icmptype = pd->hdr.icmp->icmp_type;
5180126258Smlaier		icmpid = pd->hdr.icmp->icmp_id;
5181126258Smlaier		icmpsum = &pd->hdr.icmp->icmp_cksum;
5182126258Smlaier
5183126258Smlaier		if (icmptype == ICMP_UNREACH ||
5184126258Smlaier		    icmptype == ICMP_SOURCEQUENCH ||
5185126258Smlaier		    icmptype == ICMP_REDIRECT ||
5186126258Smlaier		    icmptype == ICMP_TIMXCEED ||
5187126258Smlaier		    icmptype == ICMP_PARAMPROB)
5188126258Smlaier			state_icmp++;
5189126258Smlaier		break;
5190126258Smlaier#endif /* INET */
5191126258Smlaier#ifdef INET6
5192126258Smlaier	case IPPROTO_ICMPV6:
5193126258Smlaier		icmptype = pd->hdr.icmp6->icmp6_type;
5194126258Smlaier		icmpid = pd->hdr.icmp6->icmp6_id;
5195126258Smlaier		icmpsum = &pd->hdr.icmp6->icmp6_cksum;
5196126258Smlaier
5197126258Smlaier		if (icmptype == ICMP6_DST_UNREACH ||
5198126258Smlaier		    icmptype == ICMP6_PACKET_TOO_BIG ||
5199126258Smlaier		    icmptype == ICMP6_TIME_EXCEEDED ||
5200126258Smlaier		    icmptype == ICMP6_PARAM_PROB)
5201126258Smlaier			state_icmp++;
5202126258Smlaier		break;
5203126258Smlaier#endif /* INET6 */
5204126258Smlaier	}
5205126258Smlaier
5206126258Smlaier	if (!state_icmp) {
5207126258Smlaier
5208126258Smlaier		/*
5209126258Smlaier		 * ICMP query/reply message not related to a TCP/UDP packet.
5210126258Smlaier		 * Search for an ICMP state.
5211126258Smlaier		 */
5212126258Smlaier		key.af = pd->af;
5213126258Smlaier		key.proto = pd->proto;
5214130613Smlaier		if (direction == PF_IN)	{
5215130613Smlaier			PF_ACPY(&key.ext.addr, pd->src, key.af);
5216130613Smlaier			PF_ACPY(&key.gwy.addr, pd->dst, key.af);
5217149884Smlaier			key.ext.port = 0;
5218130613Smlaier			key.gwy.port = icmpid;
5219130613Smlaier		} else {
5220130613Smlaier			PF_ACPY(&key.lan.addr, pd->src, key.af);
5221130613Smlaier			PF_ACPY(&key.ext.addr, pd->dst, key.af);
5222130613Smlaier			key.lan.port = icmpid;
5223149884Smlaier			key.ext.port = 0;
5224130613Smlaier		}
5225126258Smlaier
5226126258Smlaier		STATE_LOOKUP();
5227126258Smlaier
5228126261Smlaier		(*state)->expire = time_second;
5229126258Smlaier		(*state)->timeout = PFTM_ICMP_ERROR_REPLY;
5230126258Smlaier
5231126258Smlaier		/* translate source/destination address, if necessary */
5232149884Smlaier		if (STATE_TRANSLATE(*state)) {
5233126258Smlaier			if (direction == PF_OUT) {
5234126258Smlaier				switch (pd->af) {
5235126258Smlaier#ifdef INET
5236126258Smlaier				case AF_INET:
5237126258Smlaier					pf_change_a(&saddr->v4.s_addr,
5238126258Smlaier					    pd->ip_sum,
5239126258Smlaier					    (*state)->gwy.addr.v4.s_addr, 0);
5240149884Smlaier					pd->hdr.icmp->icmp_cksum =
5241149884Smlaier					    pf_cksum_fixup(
5242149884Smlaier					    pd->hdr.icmp->icmp_cksum, icmpid,
5243149884Smlaier					    (*state)->gwy.port, 0);
5244149884Smlaier					pd->hdr.icmp->icmp_id =
5245149884Smlaier					    (*state)->gwy.port;
5246149884Smlaier					m_copyback(m, off, ICMP_MINLEN,
5247149893Smlaier					    (caddr_t)pd->hdr.icmp);
5248126258Smlaier					break;
5249126258Smlaier#endif /* INET */
5250126258Smlaier#ifdef INET6
5251126258Smlaier				case AF_INET6:
5252126258Smlaier					pf_change_a6(saddr,
5253126258Smlaier					    &pd->hdr.icmp6->icmp6_cksum,
5254126258Smlaier					    &(*state)->gwy.addr, 0);
5255126258Smlaier					m_copyback(m, off,
5256126258Smlaier					    sizeof(struct icmp6_hdr),
5257126261Smlaier					    (caddr_t)pd->hdr.icmp6);
5258126258Smlaier					break;
5259126258Smlaier#endif /* INET6 */
5260126258Smlaier				}
5261126258Smlaier			} else {
5262126258Smlaier				switch (pd->af) {
5263126258Smlaier#ifdef INET
5264126258Smlaier				case AF_INET:
5265126258Smlaier					pf_change_a(&daddr->v4.s_addr,
5266126258Smlaier					    pd->ip_sum,
5267126258Smlaier					    (*state)->lan.addr.v4.s_addr, 0);
5268149884Smlaier					pd->hdr.icmp->icmp_cksum =
5269149884Smlaier					    pf_cksum_fixup(
5270149884Smlaier					    pd->hdr.icmp->icmp_cksum, icmpid,
5271149884Smlaier					    (*state)->lan.port, 0);
5272149884Smlaier					pd->hdr.icmp->icmp_id =
5273149884Smlaier					    (*state)->lan.port;
5274149884Smlaier					m_copyback(m, off, ICMP_MINLEN,
5275149893Smlaier					    (caddr_t)pd->hdr.icmp);
5276126258Smlaier					break;
5277126258Smlaier#endif /* INET */
5278126258Smlaier#ifdef INET6
5279126258Smlaier				case AF_INET6:
5280126258Smlaier					pf_change_a6(daddr,
5281126258Smlaier					    &pd->hdr.icmp6->icmp6_cksum,
5282126258Smlaier					    &(*state)->lan.addr, 0);
5283126258Smlaier					m_copyback(m, off,
5284126258Smlaier					    sizeof(struct icmp6_hdr),
5285126261Smlaier					    (caddr_t)pd->hdr.icmp6);
5286126258Smlaier					break;
5287126258Smlaier#endif /* INET6 */
5288126258Smlaier				}
5289126258Smlaier			}
5290126258Smlaier		}
5291126258Smlaier
5292126258Smlaier		return (PF_PASS);
5293126258Smlaier
5294126258Smlaier	} else {
5295126258Smlaier		/*
5296126258Smlaier		 * ICMP error message in response to a TCP/UDP packet.
5297126258Smlaier		 * Extract the inner TCP/UDP header and search for that state.
5298126258Smlaier		 */
5299126258Smlaier
5300126258Smlaier		struct pf_pdesc	pd2;
5301126258Smlaier#ifdef INET
5302126258Smlaier		struct ip	h2;
5303126258Smlaier#endif /* INET */
5304126258Smlaier#ifdef INET6
5305126258Smlaier		struct ip6_hdr	h2_6;
5306126258Smlaier		int		terminal = 0;
5307126258Smlaier#endif /* INET6 */
5308127629Smlaier		int		ipoff2 = 0;	/* make the compiler happy */
5309127629Smlaier		int		off2 = 0;	/* make the compiler happy */
5310126258Smlaier
5311126258Smlaier		pd2.af = pd->af;
5312126258Smlaier		switch (pd->af) {
5313126258Smlaier#ifdef INET
5314126258Smlaier		case AF_INET:
5315126258Smlaier			/* offset of h2 in mbuf chain */
5316126258Smlaier			ipoff2 = off + ICMP_MINLEN;
5317126258Smlaier
5318126258Smlaier			if (!pf_pull_hdr(m, ipoff2, &h2, sizeof(h2),
5319145836Smlaier			    NULL, reason, pd2.af)) {
5320126258Smlaier				DPFPRINTF(PF_DEBUG_MISC,
5321126258Smlaier				    ("pf: ICMP error message too short "
5322126258Smlaier				    "(ip)\n"));
5323126258Smlaier				return (PF_DROP);
5324126258Smlaier			}
5325126258Smlaier			/*
5326126258Smlaier			 * ICMP error messages don't refer to non-first
5327126258Smlaier			 * fragments
5328126258Smlaier			 */
5329145836Smlaier			if (h2.ip_off & htons(IP_OFFMASK)) {
5330145836Smlaier				REASON_SET(reason, PFRES_FRAG);
5331126258Smlaier				return (PF_DROP);
5332145836Smlaier			}
5333126258Smlaier
5334126258Smlaier			/* offset of protocol header that follows h2 */
5335126258Smlaier			off2 = ipoff2 + (h2.ip_hl << 2);
5336126258Smlaier
5337126258Smlaier			pd2.proto = h2.ip_p;
5338126258Smlaier			pd2.src = (struct pf_addr *)&h2.ip_src;
5339126258Smlaier			pd2.dst = (struct pf_addr *)&h2.ip_dst;
5340126258Smlaier			pd2.ip_sum = &h2.ip_sum;
5341126258Smlaier			break;
5342126258Smlaier#endif /* INET */
5343126258Smlaier#ifdef INET6
5344126258Smlaier		case AF_INET6:
5345126258Smlaier			ipoff2 = off + sizeof(struct icmp6_hdr);
5346126258Smlaier
5347126258Smlaier			if (!pf_pull_hdr(m, ipoff2, &h2_6, sizeof(h2_6),
5348145836Smlaier			    NULL, reason, pd2.af)) {
5349126258Smlaier				DPFPRINTF(PF_DEBUG_MISC,
5350126258Smlaier				    ("pf: ICMP error message too short "
5351126258Smlaier				    "(ip6)\n"));
5352126258Smlaier				return (PF_DROP);
5353126258Smlaier			}
5354126258Smlaier			pd2.proto = h2_6.ip6_nxt;
5355126258Smlaier			pd2.src = (struct pf_addr *)&h2_6.ip6_src;
5356126258Smlaier			pd2.dst = (struct pf_addr *)&h2_6.ip6_dst;
5357126258Smlaier			pd2.ip_sum = NULL;
5358126258Smlaier			off2 = ipoff2 + sizeof(h2_6);
5359126258Smlaier			do {
5360126258Smlaier				switch (pd2.proto) {
5361126258Smlaier				case IPPROTO_FRAGMENT:
5362126258Smlaier					/*
5363126258Smlaier					 * ICMPv6 error messages for
5364126258Smlaier					 * non-first fragments
5365126258Smlaier					 */
5366145836Smlaier					REASON_SET(reason, PFRES_FRAG);
5367126258Smlaier					return (PF_DROP);
5368126258Smlaier				case IPPROTO_AH:
5369126258Smlaier				case IPPROTO_HOPOPTS:
5370126258Smlaier				case IPPROTO_ROUTING:
5371126258Smlaier				case IPPROTO_DSTOPTS: {
5372126258Smlaier					/* get next header and header length */
5373126258Smlaier					struct ip6_ext opt6;
5374126258Smlaier
5375126258Smlaier					if (!pf_pull_hdr(m, off2, &opt6,
5376145836Smlaier					    sizeof(opt6), NULL, reason,
5377145836Smlaier					    pd2.af)) {
5378126258Smlaier						DPFPRINTF(PF_DEBUG_MISC,
5379126258Smlaier						    ("pf: ICMPv6 short opt\n"));
5380126258Smlaier						return (PF_DROP);
5381126258Smlaier					}
5382126258Smlaier					if (pd2.proto == IPPROTO_AH)
5383126258Smlaier						off2 += (opt6.ip6e_len + 2) * 4;
5384126258Smlaier					else
5385126258Smlaier						off2 += (opt6.ip6e_len + 1) * 8;
5386126258Smlaier					pd2.proto = opt6.ip6e_nxt;
5387126258Smlaier					/* goto the next header */
5388126258Smlaier					break;
5389126258Smlaier				}
5390126258Smlaier				default:
5391126258Smlaier					terminal++;
5392126258Smlaier					break;
5393126258Smlaier				}
5394126258Smlaier			} while (!terminal);
5395126258Smlaier			break;
5396126258Smlaier#endif /* INET6 */
5397171168Smlaier#ifdef __FreeBSD__
5398171168Smlaier		default:
5399171168Smlaier			panic("AF not supported: %d", pd->af);
5400171168Smlaier#endif
5401126258Smlaier		}
5402126258Smlaier
5403126258Smlaier		switch (pd2.proto) {
5404126258Smlaier		case IPPROTO_TCP: {
5405126258Smlaier			struct tcphdr		 th;
5406126258Smlaier			u_int32_t		 seq;
5407126258Smlaier			struct pf_state_peer	*src, *dst;
5408126258Smlaier			u_int8_t		 dws;
5409128129Smlaier			int			 copyback = 0;
5410126258Smlaier
5411126258Smlaier			/*
5412126258Smlaier			 * Only the first 8 bytes of the TCP header can be
5413126258Smlaier			 * expected. Don't access any TCP header fields after
5414126258Smlaier			 * th_seq, an ackskew test is not possible.
5415126258Smlaier			 */
5416145836Smlaier			if (!pf_pull_hdr(m, off2, &th, 8, NULL, reason,
5417145836Smlaier			    pd2.af)) {
5418126258Smlaier				DPFPRINTF(PF_DEBUG_MISC,
5419126258Smlaier				    ("pf: ICMP error message too short "
5420126258Smlaier				    "(tcp)\n"));
5421126258Smlaier				return (PF_DROP);
5422126258Smlaier			}
5423126258Smlaier
5424126258Smlaier			key.af = pd2.af;
5425126258Smlaier			key.proto = IPPROTO_TCP;
5426130613Smlaier			if (direction == PF_IN)	{
5427130613Smlaier				PF_ACPY(&key.ext.addr, pd2.dst, key.af);
5428130613Smlaier				PF_ACPY(&key.gwy.addr, pd2.src, key.af);
5429130613Smlaier				key.ext.port = th.th_dport;
5430130613Smlaier				key.gwy.port = th.th_sport;
5431130613Smlaier			} else {
5432130613Smlaier				PF_ACPY(&key.lan.addr, pd2.dst, key.af);
5433130613Smlaier				PF_ACPY(&key.ext.addr, pd2.src, key.af);
5434130613Smlaier				key.lan.port = th.th_dport;
5435130613Smlaier				key.ext.port = th.th_sport;
5436130613Smlaier			}
5437126258Smlaier
5438126258Smlaier			STATE_LOOKUP();
5439126258Smlaier
5440126258Smlaier			if (direction == (*state)->direction) {
5441126258Smlaier				src = &(*state)->dst;
5442126258Smlaier				dst = &(*state)->src;
5443126258Smlaier			} else {
5444126258Smlaier				src = &(*state)->src;
5445126258Smlaier				dst = &(*state)->dst;
5446126258Smlaier			}
5447126258Smlaier
5448171929Sdhartmei			if (src->wscale && dst->wscale)
5449126258Smlaier				dws = dst->wscale & PF_WSCALE_MASK;
5450126258Smlaier			else
5451126258Smlaier				dws = 0;
5452126258Smlaier
5453126258Smlaier			/* Demodulate sequence number */
5454126258Smlaier			seq = ntohl(th.th_seq) - src->seqdiff;
5455128129Smlaier			if (src->seqdiff) {
5456128129Smlaier				pf_change_a(&th.th_seq, icmpsum,
5457126258Smlaier				    htonl(seq), 0);
5458128129Smlaier				copyback = 1;
5459128129Smlaier			}
5460126258Smlaier
5461126258Smlaier			if (!SEQ_GEQ(src->seqhi, seq) ||
5462126258Smlaier			    !SEQ_GEQ(seq, src->seqlo - (dst->max_win << dws))) {
5463126258Smlaier				if (pf_status.debug >= PF_DEBUG_MISC) {
5464126258Smlaier					printf("pf: BAD ICMP %d:%d ",
5465126258Smlaier					    icmptype, pd->hdr.icmp->icmp_code);
5466126258Smlaier					pf_print_host(pd->src, 0, pd->af);
5467126258Smlaier					printf(" -> ");
5468126258Smlaier					pf_print_host(pd->dst, 0, pd->af);
5469126258Smlaier					printf(" state: ");
5470126258Smlaier					pf_print_state(*state);
5471126258Smlaier					printf(" seq=%u\n", seq);
5472126258Smlaier				}
5473145836Smlaier				REASON_SET(reason, PFRES_BADSTATE);
5474126258Smlaier				return (PF_DROP);
5475126258Smlaier			}
5476126258Smlaier
5477126258Smlaier			if (STATE_TRANSLATE(*state)) {
5478126258Smlaier				if (direction == PF_IN) {
5479126258Smlaier					pf_change_icmp(pd2.src, &th.th_sport,
5480128129Smlaier					    daddr, &(*state)->lan.addr,
5481126258Smlaier					    (*state)->lan.port, NULL,
5482126258Smlaier					    pd2.ip_sum, icmpsum,
5483126258Smlaier					    pd->ip_sum, 0, pd2.af);
5484126258Smlaier				} else {
5485126258Smlaier					pf_change_icmp(pd2.dst, &th.th_dport,
5486126258Smlaier					    saddr, &(*state)->gwy.addr,
5487126258Smlaier					    (*state)->gwy.port, NULL,
5488126258Smlaier					    pd2.ip_sum, icmpsum,
5489126258Smlaier					    pd->ip_sum, 0, pd2.af);
5490126258Smlaier				}
5491128129Smlaier				copyback = 1;
5492128129Smlaier			}
5493128129Smlaier
5494128129Smlaier			if (copyback) {
5495126258Smlaier				switch (pd2.af) {
5496126258Smlaier#ifdef INET
5497126258Smlaier				case AF_INET:
5498126258Smlaier					m_copyback(m, off, ICMP_MINLEN,
5499126261Smlaier					    (caddr_t)pd->hdr.icmp);
5500126258Smlaier					m_copyback(m, ipoff2, sizeof(h2),
5501126261Smlaier					    (caddr_t)&h2);
5502126258Smlaier					break;
5503126258Smlaier#endif /* INET */
5504126258Smlaier#ifdef INET6
5505126258Smlaier				case AF_INET6:
5506126258Smlaier					m_copyback(m, off,
5507126258Smlaier					    sizeof(struct icmp6_hdr),
5508126261Smlaier					    (caddr_t)pd->hdr.icmp6);
5509126258Smlaier					m_copyback(m, ipoff2, sizeof(h2_6),
5510126261Smlaier					    (caddr_t)&h2_6);
5511126258Smlaier					break;
5512126258Smlaier#endif /* INET6 */
5513126258Smlaier				}
5514126261Smlaier				m_copyback(m, off2, 8, (caddr_t)&th);
5515126258Smlaier			}
5516126258Smlaier
5517126258Smlaier			return (PF_PASS);
5518126258Smlaier			break;
5519126258Smlaier		}
5520126258Smlaier		case IPPROTO_UDP: {
5521126258Smlaier			struct udphdr		uh;
5522126258Smlaier
5523126258Smlaier			if (!pf_pull_hdr(m, off2, &uh, sizeof(uh),
5524145836Smlaier			    NULL, reason, pd2.af)) {
5525126258Smlaier				DPFPRINTF(PF_DEBUG_MISC,
5526126258Smlaier				    ("pf: ICMP error message too short "
5527126258Smlaier				    "(udp)\n"));
5528126258Smlaier				return (PF_DROP);
5529126258Smlaier			}
5530126258Smlaier
5531126258Smlaier			key.af = pd2.af;
5532126258Smlaier			key.proto = IPPROTO_UDP;
5533130613Smlaier			if (direction == PF_IN)	{
5534130613Smlaier				PF_ACPY(&key.ext.addr, pd2.dst, key.af);
5535130613Smlaier				PF_ACPY(&key.gwy.addr, pd2.src, key.af);
5536130613Smlaier				key.ext.port = uh.uh_dport;
5537130613Smlaier				key.gwy.port = uh.uh_sport;
5538130613Smlaier			} else {
5539130613Smlaier				PF_ACPY(&key.lan.addr, pd2.dst, key.af);
5540130613Smlaier				PF_ACPY(&key.ext.addr, pd2.src, key.af);
5541130613Smlaier				key.lan.port = uh.uh_dport;
5542130613Smlaier				key.ext.port = uh.uh_sport;
5543130613Smlaier			}
5544126258Smlaier
5545126258Smlaier			STATE_LOOKUP();
5546126258Smlaier
5547126258Smlaier			if (STATE_TRANSLATE(*state)) {
5548126258Smlaier				if (direction == PF_IN) {
5549126258Smlaier					pf_change_icmp(pd2.src, &uh.uh_sport,
5550126258Smlaier					    daddr, &(*state)->lan.addr,
5551126258Smlaier					    (*state)->lan.port, &uh.uh_sum,
5552126258Smlaier					    pd2.ip_sum, icmpsum,
5553126258Smlaier					    pd->ip_sum, 1, pd2.af);
5554126258Smlaier				} else {
5555126258Smlaier					pf_change_icmp(pd2.dst, &uh.uh_dport,
5556126258Smlaier					    saddr, &(*state)->gwy.addr,
5557126258Smlaier					    (*state)->gwy.port, &uh.uh_sum,
5558126258Smlaier					    pd2.ip_sum, icmpsum,
5559126258Smlaier					    pd->ip_sum, 1, pd2.af);
5560126258Smlaier				}
5561126258Smlaier				switch (pd2.af) {
5562126258Smlaier#ifdef INET
5563126258Smlaier				case AF_INET:
5564126258Smlaier					m_copyback(m, off, ICMP_MINLEN,
5565126261Smlaier					    (caddr_t)pd->hdr.icmp);
5566126261Smlaier					m_copyback(m, ipoff2, sizeof(h2),
5567126261Smlaier					    (caddr_t)&h2);
5568126258Smlaier					break;
5569126258Smlaier#endif /* INET */
5570126258Smlaier#ifdef INET6
5571126258Smlaier				case AF_INET6:
5572126258Smlaier					m_copyback(m, off,
5573126258Smlaier					    sizeof(struct icmp6_hdr),
5574126261Smlaier					    (caddr_t)pd->hdr.icmp6);
5575126258Smlaier					m_copyback(m, ipoff2, sizeof(h2_6),
5576126261Smlaier					    (caddr_t)&h2_6);
5577126258Smlaier					break;
5578126258Smlaier#endif /* INET6 */
5579126258Smlaier				}
5580126261Smlaier				m_copyback(m, off2, sizeof(uh),
5581126261Smlaier				    (caddr_t)&uh);
5582126258Smlaier			}
5583126258Smlaier
5584126258Smlaier			return (PF_PASS);
5585126258Smlaier			break;
5586126258Smlaier		}
5587126258Smlaier#ifdef INET
5588126258Smlaier		case IPPROTO_ICMP: {
5589126258Smlaier			struct icmp		iih;
5590126258Smlaier
5591126258Smlaier			if (!pf_pull_hdr(m, off2, &iih, ICMP_MINLEN,
5592145836Smlaier			    NULL, reason, pd2.af)) {
5593126258Smlaier				DPFPRINTF(PF_DEBUG_MISC,
5594126258Smlaier				    ("pf: ICMP error message too short i"
5595126258Smlaier				    "(icmp)\n"));
5596126258Smlaier				return (PF_DROP);
5597126258Smlaier			}
5598126258Smlaier
5599126258Smlaier			key.af = pd2.af;
5600126258Smlaier			key.proto = IPPROTO_ICMP;
5601130613Smlaier			if (direction == PF_IN)	{
5602130613Smlaier				PF_ACPY(&key.ext.addr, pd2.dst, key.af);
5603130613Smlaier				PF_ACPY(&key.gwy.addr, pd2.src, key.af);
5604149884Smlaier				key.ext.port = 0;
5605130613Smlaier				key.gwy.port = iih.icmp_id;
5606130613Smlaier			} else {
5607130613Smlaier				PF_ACPY(&key.lan.addr, pd2.dst, key.af);
5608130613Smlaier				PF_ACPY(&key.ext.addr, pd2.src, key.af);
5609130613Smlaier				key.lan.port = iih.icmp_id;
5610149884Smlaier				key.ext.port = 0;
5611130613Smlaier			}
5612126258Smlaier
5613126258Smlaier			STATE_LOOKUP();
5614126258Smlaier
5615126258Smlaier			if (STATE_TRANSLATE(*state)) {
5616126258Smlaier				if (direction == PF_IN) {
5617126258Smlaier					pf_change_icmp(pd2.src, &iih.icmp_id,
5618126258Smlaier					    daddr, &(*state)->lan.addr,
5619126258Smlaier					    (*state)->lan.port, NULL,
5620126258Smlaier					    pd2.ip_sum, icmpsum,
5621126258Smlaier					    pd->ip_sum, 0, AF_INET);
5622126258Smlaier				} else {
5623126258Smlaier					pf_change_icmp(pd2.dst, &iih.icmp_id,
5624126258Smlaier					    saddr, &(*state)->gwy.addr,
5625126258Smlaier					    (*state)->gwy.port, NULL,
5626126258Smlaier					    pd2.ip_sum, icmpsum,
5627126258Smlaier					    pd->ip_sum, 0, AF_INET);
5628126258Smlaier				}
5629126261Smlaier				m_copyback(m, off, ICMP_MINLEN,
5630126261Smlaier				    (caddr_t)pd->hdr.icmp);
5631126261Smlaier				m_copyback(m, ipoff2, sizeof(h2),
5632126261Smlaier				    (caddr_t)&h2);
5633126261Smlaier				m_copyback(m, off2, ICMP_MINLEN,
5634126261Smlaier				    (caddr_t)&iih);
5635126258Smlaier			}
5636126258Smlaier
5637126258Smlaier			return (PF_PASS);
5638126258Smlaier			break;
5639126258Smlaier		}
5640126258Smlaier#endif /* INET */
5641126258Smlaier#ifdef INET6
5642126258Smlaier		case IPPROTO_ICMPV6: {
5643126258Smlaier			struct icmp6_hdr	iih;
5644126258Smlaier
5645126258Smlaier			if (!pf_pull_hdr(m, off2, &iih,
5646145836Smlaier			    sizeof(struct icmp6_hdr), NULL, reason, pd2.af)) {
5647126258Smlaier				DPFPRINTF(PF_DEBUG_MISC,
5648126258Smlaier				    ("pf: ICMP error message too short "
5649126258Smlaier				    "(icmp6)\n"));
5650126258Smlaier				return (PF_DROP);
5651126258Smlaier			}
5652126258Smlaier
5653126258Smlaier			key.af = pd2.af;
5654126258Smlaier			key.proto = IPPROTO_ICMPV6;
5655130613Smlaier			if (direction == PF_IN)	{
5656130613Smlaier				PF_ACPY(&key.ext.addr, pd2.dst, key.af);
5657130613Smlaier				PF_ACPY(&key.gwy.addr, pd2.src, key.af);
5658149884Smlaier				key.ext.port = 0;
5659130613Smlaier				key.gwy.port = iih.icmp6_id;
5660130613Smlaier			} else {
5661130613Smlaier				PF_ACPY(&key.lan.addr, pd2.dst, key.af);
5662130613Smlaier				PF_ACPY(&key.ext.addr, pd2.src, key.af);
5663130613Smlaier				key.lan.port = iih.icmp6_id;
5664149884Smlaier				key.ext.port = 0;
5665130613Smlaier			}
5666126258Smlaier
5667126258Smlaier			STATE_LOOKUP();
5668126258Smlaier
5669126258Smlaier			if (STATE_TRANSLATE(*state)) {
5670126258Smlaier				if (direction == PF_IN) {
5671126258Smlaier					pf_change_icmp(pd2.src, &iih.icmp6_id,
5672126258Smlaier					    daddr, &(*state)->lan.addr,
5673126258Smlaier					    (*state)->lan.port, NULL,
5674126258Smlaier					    pd2.ip_sum, icmpsum,
5675126258Smlaier					    pd->ip_sum, 0, AF_INET6);
5676126258Smlaier				} else {
5677126258Smlaier					pf_change_icmp(pd2.dst, &iih.icmp6_id,
5678126258Smlaier					    saddr, &(*state)->gwy.addr,
5679126258Smlaier					    (*state)->gwy.port, NULL,
5680126258Smlaier					    pd2.ip_sum, icmpsum,
5681126258Smlaier					    pd->ip_sum, 0, AF_INET6);
5682126258Smlaier				}
5683126258Smlaier				m_copyback(m, off, sizeof(struct icmp6_hdr),
5684126261Smlaier				    (caddr_t)pd->hdr.icmp6);
5685126261Smlaier				m_copyback(m, ipoff2, sizeof(h2_6),
5686126261Smlaier				    (caddr_t)&h2_6);
5687126258Smlaier				m_copyback(m, off2, sizeof(struct icmp6_hdr),
5688126261Smlaier				    (caddr_t)&iih);
5689126258Smlaier			}
5690126258Smlaier
5691126258Smlaier			return (PF_PASS);
5692126258Smlaier			break;
5693126258Smlaier		}
5694126258Smlaier#endif /* INET6 */
5695126258Smlaier		default: {
5696126258Smlaier			key.af = pd2.af;
5697126258Smlaier			key.proto = pd2.proto;
5698130613Smlaier			if (direction == PF_IN)	{
5699130613Smlaier				PF_ACPY(&key.ext.addr, pd2.dst, key.af);
5700130613Smlaier				PF_ACPY(&key.gwy.addr, pd2.src, key.af);
5701130613Smlaier				key.ext.port = 0;
5702130613Smlaier				key.gwy.port = 0;
5703130613Smlaier			} else {
5704130613Smlaier				PF_ACPY(&key.lan.addr, pd2.dst, key.af);
5705130613Smlaier				PF_ACPY(&key.ext.addr, pd2.src, key.af);
5706130613Smlaier				key.lan.port = 0;
5707130613Smlaier				key.ext.port = 0;
5708130613Smlaier			}
5709126258Smlaier
5710126258Smlaier			STATE_LOOKUP();
5711126258Smlaier
5712126258Smlaier			if (STATE_TRANSLATE(*state)) {
5713126258Smlaier				if (direction == PF_IN) {
5714126258Smlaier					pf_change_icmp(pd2.src, NULL,
5715126258Smlaier					    daddr, &(*state)->lan.addr,
5716126258Smlaier					    0, NULL,
5717126258Smlaier					    pd2.ip_sum, icmpsum,
5718126258Smlaier					    pd->ip_sum, 0, pd2.af);
5719126258Smlaier				} else {
5720126258Smlaier					pf_change_icmp(pd2.dst, NULL,
5721126258Smlaier					    saddr, &(*state)->gwy.addr,
5722126258Smlaier					    0, NULL,
5723126258Smlaier					    pd2.ip_sum, icmpsum,
5724126258Smlaier					    pd->ip_sum, 0, pd2.af);
5725126258Smlaier				}
5726126258Smlaier				switch (pd2.af) {
5727126258Smlaier#ifdef INET
5728126258Smlaier				case AF_INET:
5729126258Smlaier					m_copyback(m, off, ICMP_MINLEN,
5730126261Smlaier					    (caddr_t)pd->hdr.icmp);
5731126261Smlaier					m_copyback(m, ipoff2, sizeof(h2),
5732126261Smlaier					    (caddr_t)&h2);
5733126258Smlaier					break;
5734126258Smlaier#endif /* INET */
5735126258Smlaier#ifdef INET6
5736126258Smlaier				case AF_INET6:
5737126258Smlaier					m_copyback(m, off,
5738126258Smlaier					    sizeof(struct icmp6_hdr),
5739126261Smlaier					    (caddr_t)pd->hdr.icmp6);
5740126258Smlaier					m_copyback(m, ipoff2, sizeof(h2_6),
5741126261Smlaier					    (caddr_t)&h2_6);
5742126258Smlaier					break;
5743126258Smlaier#endif /* INET6 */
5744126258Smlaier				}
5745126258Smlaier			}
5746126258Smlaier
5747126258Smlaier			return (PF_PASS);
5748126258Smlaier			break;
5749126258Smlaier		}
5750126258Smlaier		}
5751126258Smlaier	}
5752126258Smlaier}
5753126258Smlaier
5754126258Smlaierint
5755130613Smlaierpf_test_state_other(struct pf_state **state, int direction, struct pfi_kif *kif,
5756126258Smlaier    struct pf_pdesc *pd)
5757126258Smlaier{
5758126258Smlaier	struct pf_state_peer	*src, *dst;
5759171168Smlaier	struct pf_state_cmp	 key;
5760126258Smlaier
5761126258Smlaier	key.af = pd->af;
5762126258Smlaier	key.proto = pd->proto;
5763130613Smlaier	if (direction == PF_IN)	{
5764130613Smlaier		PF_ACPY(&key.ext.addr, pd->src, key.af);
5765130613Smlaier		PF_ACPY(&key.gwy.addr, pd->dst, key.af);
5766130613Smlaier		key.ext.port = 0;
5767130613Smlaier		key.gwy.port = 0;
5768130613Smlaier	} else {
5769130613Smlaier		PF_ACPY(&key.lan.addr, pd->src, key.af);
5770130613Smlaier		PF_ACPY(&key.ext.addr, pd->dst, key.af);
5771130613Smlaier		key.lan.port = 0;
5772130613Smlaier		key.ext.port = 0;
5773130613Smlaier	}
5774126258Smlaier
5775126258Smlaier	STATE_LOOKUP();
5776126258Smlaier
5777126258Smlaier	if (direction == (*state)->direction) {
5778126258Smlaier		src = &(*state)->src;
5779126258Smlaier		dst = &(*state)->dst;
5780126258Smlaier	} else {
5781126258Smlaier		src = &(*state)->dst;
5782126258Smlaier		dst = &(*state)->src;
5783126258Smlaier	}
5784126258Smlaier
5785126258Smlaier	/* update states */
5786126258Smlaier	if (src->state < PFOTHERS_SINGLE)
5787126258Smlaier		src->state = PFOTHERS_SINGLE;
5788126258Smlaier	if (dst->state == PFOTHERS_SINGLE)
5789126258Smlaier		dst->state = PFOTHERS_MULTIPLE;
5790126258Smlaier
5791126258Smlaier	/* update expire time */
5792126261Smlaier	(*state)->expire = time_second;
5793126258Smlaier	if (src->state == PFOTHERS_MULTIPLE && dst->state == PFOTHERS_MULTIPLE)
5794126258Smlaier		(*state)->timeout = PFTM_OTHER_MULTIPLE;
5795126258Smlaier	else
5796126258Smlaier		(*state)->timeout = PFTM_OTHER_SINGLE;
5797126258Smlaier
5798126258Smlaier	/* translate source/destination address, if necessary */
5799126258Smlaier	if (STATE_TRANSLATE(*state)) {
5800126258Smlaier		if (direction == PF_OUT)
5801126258Smlaier			switch (pd->af) {
5802126258Smlaier#ifdef INET
5803126258Smlaier			case AF_INET:
5804126258Smlaier				pf_change_a(&pd->src->v4.s_addr,
5805126258Smlaier				    pd->ip_sum, (*state)->gwy.addr.v4.s_addr,
5806126258Smlaier				    0);
5807126258Smlaier				break;
5808126258Smlaier#endif /* INET */
5809126258Smlaier#ifdef INET6
5810126258Smlaier			case AF_INET6:
5811126258Smlaier				PF_ACPY(pd->src, &(*state)->gwy.addr, pd->af);
5812126258Smlaier				break;
5813126258Smlaier#endif /* INET6 */
5814126258Smlaier			}
5815126258Smlaier		else
5816126258Smlaier			switch (pd->af) {
5817126258Smlaier#ifdef INET
5818126258Smlaier			case AF_INET:
5819126258Smlaier				pf_change_a(&pd->dst->v4.s_addr,
5820126258Smlaier				    pd->ip_sum, (*state)->lan.addr.v4.s_addr,
5821126258Smlaier				    0);
5822126258Smlaier				break;
5823126258Smlaier#endif /* INET */
5824126258Smlaier#ifdef INET6
5825126258Smlaier			case AF_INET6:
5826126258Smlaier				PF_ACPY(pd->dst, &(*state)->lan.addr, pd->af);
5827126258Smlaier				break;
5828126258Smlaier#endif /* INET6 */
5829126258Smlaier			}
5830126258Smlaier	}
5831126258Smlaier
5832126258Smlaier	return (PF_PASS);
5833126258Smlaier}
5834126258Smlaier
5835126258Smlaier/*
5836126258Smlaier * ipoff and off are measured from the start of the mbuf chain.
5837126258Smlaier * h must be at "ipoff" on the mbuf chain.
5838126258Smlaier */
5839126258Smlaiervoid *
5840126258Smlaierpf_pull_hdr(struct mbuf *m, int off, void *p, int len,
5841126258Smlaier    u_short *actionp, u_short *reasonp, sa_family_t af)
5842126258Smlaier{
5843126258Smlaier	switch (af) {
5844126258Smlaier#ifdef INET
5845126258Smlaier	case AF_INET: {
5846126258Smlaier		struct ip	*h = mtod(m, struct ip *);
5847126258Smlaier		u_int16_t	 fragoff = (ntohs(h->ip_off) & IP_OFFMASK) << 3;
5848126258Smlaier
5849126258Smlaier		if (fragoff) {
5850126258Smlaier			if (fragoff >= len)
5851126258Smlaier				ACTION_SET(actionp, PF_PASS);
5852126258Smlaier			else {
5853126258Smlaier				ACTION_SET(actionp, PF_DROP);
5854126258Smlaier				REASON_SET(reasonp, PFRES_FRAG);
5855126258Smlaier			}
5856126258Smlaier			return (NULL);
5857126258Smlaier		}
5858130613Smlaier		if (m->m_pkthdr.len < off + len ||
5859130613Smlaier		    ntohs(h->ip_len) < off + len) {
5860126258Smlaier			ACTION_SET(actionp, PF_DROP);
5861126258Smlaier			REASON_SET(reasonp, PFRES_SHORT);
5862126258Smlaier			return (NULL);
5863126258Smlaier		}
5864126258Smlaier		break;
5865126258Smlaier	}
5866126258Smlaier#endif /* INET */
5867126258Smlaier#ifdef INET6
5868126258Smlaier	case AF_INET6: {
5869126258Smlaier		struct ip6_hdr	*h = mtod(m, struct ip6_hdr *);
5870126258Smlaier
5871126258Smlaier		if (m->m_pkthdr.len < off + len ||
5872126258Smlaier		    (ntohs(h->ip6_plen) + sizeof(struct ip6_hdr)) <
5873126258Smlaier		    (unsigned)(off + len)) {
5874126258Smlaier			ACTION_SET(actionp, PF_DROP);
5875126258Smlaier			REASON_SET(reasonp, PFRES_SHORT);
5876126258Smlaier			return (NULL);
5877126258Smlaier		}
5878126258Smlaier		break;
5879126258Smlaier	}
5880126258Smlaier#endif /* INET6 */
5881126258Smlaier	}
5882126258Smlaier	m_copydata(m, off, len, p);
5883126258Smlaier	return (p);
5884126258Smlaier}
5885126258Smlaier
5886126258Smlaierint
5887171168Smlaierpf_routable(struct pf_addr *addr, sa_family_t af, struct pfi_kif *kif)
5888126258Smlaier{
5889126258Smlaier	struct sockaddr_in	*dst;
5890171168Smlaier	int			 ret = 1;
5891171168Smlaier	int			 check_mpath;
5892171168Smlaier#ifndef __FreeBSD__
5893171168Smlaier	extern int		 ipmultipath;
5894171168Smlaier#endif
5895145836Smlaier#ifdef INET6
5896171168Smlaier#ifndef __FreeBSD__
5897171168Smlaier	extern int		 ip6_multipath;
5898171168Smlaier#endif
5899145836Smlaier	struct sockaddr_in6	*dst6;
5900145836Smlaier	struct route_in6	 ro;
5901145836Smlaier#else
5902126258Smlaier	struct route		 ro;
5903145836Smlaier#endif
5904171168Smlaier	struct radix_node	*rn;
5905171168Smlaier	struct rtentry		*rt;
5906171168Smlaier	struct ifnet		*ifp;
5907126258Smlaier
5908171168Smlaier	check_mpath = 0;
5909126258Smlaier	bzero(&ro, sizeof(ro));
5910145836Smlaier	switch (af) {
5911145836Smlaier	case AF_INET:
5912145836Smlaier		dst = satosin(&ro.ro_dst);
5913145836Smlaier		dst->sin_family = AF_INET;
5914145836Smlaier		dst->sin_len = sizeof(*dst);
5915145836Smlaier		dst->sin_addr = addr->v4;
5916171168Smlaier#ifndef __FreeBSD__	/* MULTIPATH_ROUTING */
5917171168Smlaier		if (ipmultipath)
5918171168Smlaier			check_mpath = 1;
5919171168Smlaier#endif
5920145836Smlaier		break;
5921145836Smlaier#ifdef INET6
5922145836Smlaier	case AF_INET6:
5923145836Smlaier		dst6 = (struct sockaddr_in6 *)&ro.ro_dst;
5924145836Smlaier		dst6->sin6_family = AF_INET6;
5925145836Smlaier		dst6->sin6_len = sizeof(*dst6);
5926145836Smlaier		dst6->sin6_addr = addr->v6;
5927171168Smlaier#ifndef __FreeBSD__	/* MULTIPATH_ROUTING */
5928171168Smlaier		if (ip6_multipath)
5929171168Smlaier			check_mpath = 1;
5930171168Smlaier#endif
5931145836Smlaier		break;
5932145836Smlaier#endif /* INET6 */
5933145836Smlaier	default:
5934145836Smlaier		return (0);
5935145836Smlaier	}
5936145836Smlaier
5937171168Smlaier	/* Skip checks for ipsec interfaces */
5938171168Smlaier	if (kif != NULL && kif->pfik_ifp->if_type == IFT_ENC)
5939171168Smlaier		goto out;
5940171168Smlaier
5941127145Smlaier#ifdef __FreeBSD__
5942145836Smlaier	rtalloc_ign((struct route *)&ro, RTF_CLONING);
5943126261Smlaier#else /* ! __FreeBSD__ */
5944145836Smlaier	rtalloc_noclone((struct route *)&ro, NO_CLONING);
5945126261Smlaier#endif
5946126258Smlaier
5947126258Smlaier	if (ro.ro_rt != NULL) {
5948171168Smlaier		/* No interface given, this is a no-route check */
5949171168Smlaier		if (kif == NULL)
5950171168Smlaier			goto out;
5951171168Smlaier
5952171168Smlaier		if (kif->pfik_ifp == NULL) {
5953171168Smlaier			ret = 0;
5954171168Smlaier			goto out;
5955171168Smlaier		}
5956171168Smlaier
5957171168Smlaier		/* Perform uRPF check if passed input interface */
5958171168Smlaier		ret = 0;
5959171168Smlaier		rn = (struct radix_node *)ro.ro_rt;
5960171168Smlaier		do {
5961171168Smlaier			rt = (struct rtentry *)rn;
5962171168Smlaier#ifndef __FreeBSD__ /* CARPDEV */
5963171168Smlaier			if (rt->rt_ifp->if_type == IFT_CARP)
5964171168Smlaier				ifp = rt->rt_ifp->if_carpdev;
5965171168Smlaier			else
5966171168Smlaier#endif
5967171168Smlaier				ifp = rt->rt_ifp;
5968171168Smlaier
5969171168Smlaier			if (kif->pfik_ifp == ifp)
5970171168Smlaier				ret = 1;
5971171168Smlaier#ifdef __FreeBSD__ /* MULTIPATH_ROUTING */
5972171168Smlaier			rn = NULL;
5973171168Smlaier#else
5974171168Smlaier			rn = rn_mpath_next(rn);
5975171168Smlaier#endif
5976171168Smlaier		} while (check_mpath == 1 && rn != NULL && ret == 0);
5977171168Smlaier	} else
5978171168Smlaier		ret = 0;
5979171168Smlaierout:
5980171168Smlaier	if (ro.ro_rt != NULL)
5981126258Smlaier		RTFREE(ro.ro_rt);
5982171168Smlaier	return (ret);
5983145836Smlaier}
5984145836Smlaier
5985145836Smlaierint
5986145836Smlaierpf_rtlabel_match(struct pf_addr *addr, sa_family_t af, struct pf_addr_wrap *aw)
5987145836Smlaier{
5988145836Smlaier	struct sockaddr_in	*dst;
5989145836Smlaier#ifdef INET6
5990145836Smlaier	struct sockaddr_in6	*dst6;
5991145836Smlaier	struct route_in6	 ro;
5992145836Smlaier#else
5993145836Smlaier	struct route		 ro;
5994145836Smlaier#endif
5995145836Smlaier	int			 ret = 0;
5996145836Smlaier
5997145836Smlaier	bzero(&ro, sizeof(ro));
5998145836Smlaier	switch (af) {
5999145836Smlaier	case AF_INET:
6000145836Smlaier		dst = satosin(&ro.ro_dst);
6001145836Smlaier		dst->sin_family = AF_INET;
6002145836Smlaier		dst->sin_len = sizeof(*dst);
6003145836Smlaier		dst->sin_addr = addr->v4;
6004145836Smlaier		break;
6005145836Smlaier#ifdef INET6
6006145836Smlaier	case AF_INET6:
6007145836Smlaier		dst6 = (struct sockaddr_in6 *)&ro.ro_dst;
6008145836Smlaier		dst6->sin6_family = AF_INET6;
6009145836Smlaier		dst6->sin6_len = sizeof(*dst6);
6010145836Smlaier		dst6->sin6_addr = addr->v6;
6011145836Smlaier		break;
6012145836Smlaier#endif /* INET6 */
6013145836Smlaier	default:
6014145836Smlaier		return (0);
6015145836Smlaier	}
6016145836Smlaier
6017145836Smlaier#ifdef __FreeBSD__
6018145836Smlaier# ifdef RTF_PRCLONING
6019145836Smlaier	rtalloc_ign((struct route *)&ro, (RTF_CLONING|RTF_PRCLONING));
6020145836Smlaier# else /* !RTF_PRCLONING */
6021145836Smlaier	rtalloc_ign((struct route *)&ro, RTF_CLONING);
6022145836Smlaier# endif
6023145836Smlaier#else /* ! __FreeBSD__ */
6024145836Smlaier	rtalloc_noclone((struct route *)&ro, NO_CLONING);
6025145836Smlaier#endif
6026145836Smlaier
6027145836Smlaier	if (ro.ro_rt != NULL) {
6028145836Smlaier#ifdef __FreeBSD__
6029145836Smlaier		/* XXX_IMPORT: later */
6030145836Smlaier#else
6031145836Smlaier		if (ro.ro_rt->rt_labelid == aw->v.rtlabel)
6032145836Smlaier			ret = 1;
6033145836Smlaier#endif
6034145836Smlaier		RTFREE(ro.ro_rt);
6035145836Smlaier	}
6036145836Smlaier
6037126258Smlaier	return (ret);
6038126258Smlaier}
6039126258Smlaier
6040126258Smlaier#ifdef INET
6041126261Smlaier
6042126258Smlaiervoid
6043126258Smlaierpf_route(struct mbuf **m, struct pf_rule *r, int dir, struct ifnet *oifp,
6044171168Smlaier    struct pf_state *s, struct pf_pdesc *pd)
6045126258Smlaier{
6046126258Smlaier	struct mbuf		*m0, *m1;
6047126258Smlaier	struct route		 iproute;
6048171168Smlaier	struct route		*ro = NULL;
6049126258Smlaier	struct sockaddr_in	*dst;
6050126258Smlaier	struct ip		*ip;
6051126258Smlaier	struct ifnet		*ifp = NULL;
6052126258Smlaier	struct pf_addr		 naddr;
6053130613Smlaier	struct pf_src_node	*sn = NULL;
6054126258Smlaier	int			 error = 0;
6055127145Smlaier#ifdef __FreeBSD__
6056126261Smlaier	int sw_csum;
6057126261Smlaier#endif
6058171168Smlaier#ifdef IPSEC
6059171168Smlaier	struct m_tag		*mtag;
6060171168Smlaier#endif /* IPSEC */
6061126258Smlaier
6062126258Smlaier	if (m == NULL || *m == NULL || r == NULL ||
6063126258Smlaier	    (dir != PF_IN && dir != PF_OUT) || oifp == NULL)
6064126258Smlaier		panic("pf_route: invalid parameters");
6065126258Smlaier
6066171168Smlaier	if (pd->pf_mtag->routed++ > 3) {
6067171168Smlaier		m0 = *m;
6068171168Smlaier		*m = NULL;
6069171168Smlaier		goto bad;
6070132303Smlaier	}
6071132303Smlaier
6072126258Smlaier	if (r->rt == PF_DUPTO) {
6073127145Smlaier#ifdef __FreeBSD__
6074132303Smlaier		if ((m0 = m_dup(*m, M_DONTWAIT)) == NULL)
6075126261Smlaier#else
6076132303Smlaier		if ((m0 = m_copym2(*m, 0, M_COPYALL, M_NOWAIT)) == NULL)
6077126261Smlaier#endif
6078126258Smlaier			return;
6079126258Smlaier	} else {
6080126258Smlaier		if ((r->rt == PF_REPLYTO) == (r->direction == dir))
6081126258Smlaier			return;
6082126258Smlaier		m0 = *m;
6083126258Smlaier	}
6084126258Smlaier
6085145836Smlaier	if (m0->m_len < sizeof(struct ip)) {
6086145836Smlaier		DPFPRINTF(PF_DEBUG_URGENT,
6087145836Smlaier		    ("pf_route: m0->m_len < sizeof(struct ip)\n"));
6088145836Smlaier		goto bad;
6089145836Smlaier	}
6090145836Smlaier
6091126258Smlaier	ip = mtod(m0, struct ip *);
6092126258Smlaier
6093126258Smlaier	ro = &iproute;
6094126258Smlaier	bzero((caddr_t)ro, sizeof(*ro));
6095126258Smlaier	dst = satosin(&ro->ro_dst);
6096126258Smlaier	dst->sin_family = AF_INET;
6097126258Smlaier	dst->sin_len = sizeof(*dst);
6098126258Smlaier	dst->sin_addr = ip->ip_dst;
6099126258Smlaier
6100126258Smlaier	if (r->rt == PF_FASTROUTE) {
6101126258Smlaier		rtalloc(ro);
6102126258Smlaier		if (ro->ro_rt == 0) {
6103126258Smlaier			ipstat.ips_noroute++;
6104126258Smlaier			goto bad;
6105126258Smlaier		}
6106126258Smlaier
6107126258Smlaier		ifp = ro->ro_rt->rt_ifp;
6108126258Smlaier		ro->ro_rt->rt_use++;
6109126258Smlaier
6110126258Smlaier		if (ro->ro_rt->rt_flags & RTF_GATEWAY)
6111126258Smlaier			dst = satosin(ro->ro_rt->rt_gateway);
6112126258Smlaier	} else {
6113145836Smlaier		if (TAILQ_EMPTY(&r->rpool.list)) {
6114145836Smlaier			DPFPRINTF(PF_DEBUG_URGENT,
6115145836Smlaier			    ("pf_route: TAILQ_EMPTY(&r->rpool.list)\n"));
6116145836Smlaier			goto bad;
6117145836Smlaier		}
6118126258Smlaier		if (s == NULL) {
6119130613Smlaier			pf_map_addr(AF_INET, r, (struct pf_addr *)&ip->ip_src,
6120130613Smlaier			    &naddr, NULL, &sn);
6121126258Smlaier			if (!PF_AZERO(&naddr, AF_INET))
6122126258Smlaier				dst->sin_addr.s_addr = naddr.v4.s_addr;
6123130613Smlaier			ifp = r->rpool.cur->kif ?
6124130613Smlaier			    r->rpool.cur->kif->pfik_ifp : NULL;
6125126258Smlaier		} else {
6126126258Smlaier			if (!PF_AZERO(&s->rt_addr, AF_INET))
6127126258Smlaier				dst->sin_addr.s_addr =
6128126258Smlaier				    s->rt_addr.v4.s_addr;
6129130613Smlaier			ifp = s->rt_kif ? s->rt_kif->pfik_ifp : NULL;
6130126258Smlaier		}
6131126258Smlaier	}
6132126258Smlaier	if (ifp == NULL)
6133126258Smlaier		goto bad;
6134126258Smlaier
6135130639Smlaier	if (oifp != ifp) {
6136127145Smlaier#ifdef __FreeBSD__
6137126261Smlaier		PF_UNLOCK();
6138145836Smlaier		if (pf_test(PF_OUT, ifp, &m0, NULL, NULL) != PF_PASS) {
6139126261Smlaier			PF_LOCK();
6140126261Smlaier			goto bad;
6141126261Smlaier		} else if (m0 == NULL) {
6142126261Smlaier			PF_LOCK();
6143126261Smlaier			goto done;
6144126261Smlaier		}
6145126261Smlaier		PF_LOCK();
6146126261Smlaier#else
6147145836Smlaier		if (pf_test(PF_OUT, ifp, &m0, NULL) != PF_PASS)
6148126258Smlaier			goto bad;
6149126258Smlaier		else if (m0 == NULL)
6150126258Smlaier			goto done;
6151126261Smlaier#endif
6152145836Smlaier		if (m0->m_len < sizeof(struct ip)) {
6153145836Smlaier			DPFPRINTF(PF_DEBUG_URGENT,
6154145836Smlaier			    ("pf_route: m0->m_len < sizeof(struct ip)\n"));
6155145836Smlaier			goto bad;
6156145836Smlaier		}
6157126258Smlaier		ip = mtod(m0, struct ip *);
6158126258Smlaier	}
6159126258Smlaier
6160127145Smlaier#ifdef __FreeBSD__
6161126261Smlaier	/* Copied from FreeBSD 5.1-CURRENT ip_output. */
6162126261Smlaier	m0->m_pkthdr.csum_flags |= CSUM_IP;
6163126261Smlaier	sw_csum = m0->m_pkthdr.csum_flags & ~ifp->if_hwassist;
6164126261Smlaier	if (sw_csum & CSUM_DELAY_DATA) {
6165126261Smlaier		/*
6166126261Smlaier		 * XXX: in_delayed_cksum assumes HBO for ip->ip_len (at least)
6167126261Smlaier		 */
6168126261Smlaier		NTOHS(ip->ip_len);
6169126261Smlaier		NTOHS(ip->ip_off);	 /* XXX: needed? */
6170126261Smlaier		in_delayed_cksum(m0);
6171126261Smlaier		HTONS(ip->ip_len);
6172126261Smlaier		HTONS(ip->ip_off);
6173126261Smlaier		sw_csum &= ~CSUM_DELAY_DATA;
6174126261Smlaier	}
6175126261Smlaier	m0->m_pkthdr.csum_flags &= ifp->if_hwassist;
6176126261Smlaier
6177126261Smlaier	if (ntohs(ip->ip_len) <= ifp->if_mtu ||
6178126261Smlaier	    (ifp->if_hwassist & CSUM_FRAGMENT &&
6179126261Smlaier		((ip->ip_off & htons(IP_DF)) == 0))) {
6180126261Smlaier		/*
6181126261Smlaier		 * ip->ip_len = htons(ip->ip_len);
6182126261Smlaier		 * ip->ip_off = htons(ip->ip_off);
6183126261Smlaier		 */
6184126261Smlaier		ip->ip_sum = 0;
6185126261Smlaier		if (sw_csum & CSUM_DELAY_IP) {
6186126261Smlaier			/* From KAME */
6187126261Smlaier			if (ip->ip_v == IPVERSION &&
6188126261Smlaier			    (ip->ip_hl << 2) == sizeof(*ip)) {
6189126261Smlaier				ip->ip_sum = in_cksum_hdr(ip);
6190126261Smlaier			} else {
6191126261Smlaier				ip->ip_sum = in_cksum(m0, ip->ip_hl << 2);
6192126261Smlaier			}
6193126261Smlaier		}
6194126261Smlaier		PF_UNLOCK();
6195126261Smlaier		error = (*ifp->if_output)(ifp, m0, sintosa(dst), ro->ro_rt);
6196126261Smlaier		PF_LOCK();
6197126261Smlaier		goto done;
6198126261Smlaier	}
6199126261Smlaier
6200126261Smlaier#else
6201126258Smlaier	/* Copied from ip_output. */
6202130613Smlaier#ifdef IPSEC
6203130613Smlaier	/*
6204130613Smlaier	 * If deferred crypto processing is needed, check that the
6205130613Smlaier	 * interface supports it.
6206130613Smlaier	 */
6207130613Smlaier	if ((mtag = m_tag_find(m0, PACKET_TAG_IPSEC_OUT_CRYPTO_NEEDED, NULL))
6208130613Smlaier	    != NULL && (ifp->if_capabilities & IFCAP_IPSEC) == 0) {
6209130613Smlaier		/* Notify IPsec to do its own crypto. */
6210130613Smlaier		ipsp_skipcrypto_unmark((struct tdb_ident *)(mtag + 1));
6211130613Smlaier		goto bad;
6212130613Smlaier	}
6213130613Smlaier#endif /* IPSEC */
6214130613Smlaier
6215130613Smlaier	/* Catch routing changes wrt. hardware checksumming for TCP or UDP. */
6216171168Smlaier	if (m0->m_pkthdr.csum_flags & M_TCPV4_CSUM_OUT) {
6217130613Smlaier		if (!(ifp->if_capabilities & IFCAP_CSUM_TCPv4) ||
6218130613Smlaier		    ifp->if_bridge != NULL) {
6219130613Smlaier			in_delayed_cksum(m0);
6220171168Smlaier			m0->m_pkthdr.csum_flags &= ~M_TCPV4_CSUM_OUT; /* Clear */
6221130613Smlaier		}
6222171168Smlaier	} else if (m0->m_pkthdr.csum_flags & M_UDPV4_CSUM_OUT) {
6223130613Smlaier		if (!(ifp->if_capabilities & IFCAP_CSUM_UDPv4) ||
6224130613Smlaier		    ifp->if_bridge != NULL) {
6225130613Smlaier			in_delayed_cksum(m0);
6226171168Smlaier			m0->m_pkthdr.csum_flags &= ~M_UDPV4_CSUM_OUT; /* Clear */
6227130613Smlaier		}
6228130613Smlaier	}
6229130613Smlaier
6230126258Smlaier	if (ntohs(ip->ip_len) <= ifp->if_mtu) {
6231126258Smlaier		if ((ifp->if_capabilities & IFCAP_CSUM_IPv4) &&
6232126258Smlaier		    ifp->if_bridge == NULL) {
6233171168Smlaier			m0->m_pkthdr.csum_flags |= M_IPV4_CSUM_OUT;
6234126258Smlaier			ipstat.ips_outhwcsum++;
6235126258Smlaier		} else {
6236126258Smlaier			ip->ip_sum = 0;
6237126258Smlaier			ip->ip_sum = in_cksum(m0, ip->ip_hl << 2);
6238126258Smlaier		}
6239126258Smlaier		/* Update relevant hardware checksum stats for TCP/UDP */
6240171168Smlaier		if (m0->m_pkthdr.csum_flags & M_TCPV4_CSUM_OUT)
6241126258Smlaier			tcpstat.tcps_outhwcsum++;
6242171168Smlaier		else if (m0->m_pkthdr.csum_flags & M_UDPV4_CSUM_OUT)
6243126258Smlaier			udpstat.udps_outhwcsum++;
6244126258Smlaier		error = (*ifp->if_output)(ifp, m0, sintosa(dst), NULL);
6245126258Smlaier		goto done;
6246126258Smlaier	}
6247126261Smlaier#endif
6248126258Smlaier	/*
6249126258Smlaier	 * Too large for interface; fragment if possible.
6250126258Smlaier	 * Must be able to put at least 8 bytes per fragment.
6251126258Smlaier	 */
6252126258Smlaier	if (ip->ip_off & htons(IP_DF)) {
6253126258Smlaier		ipstat.ips_cantfrag++;
6254126258Smlaier		if (r->rt != PF_DUPTO) {
6255127145Smlaier#ifdef __FreeBSD__
6256126261Smlaier			/* icmp_error() expects host byte ordering */
6257126261Smlaier			NTOHS(ip->ip_len);
6258126261Smlaier			NTOHS(ip->ip_off);
6259126261Smlaier			PF_UNLOCK();
6260126258Smlaier			icmp_error(m0, ICMP_UNREACH, ICMP_UNREACH_NEEDFRAG, 0,
6261145886Smlaier			    ifp->if_mtu);
6262145874Smlaier			PF_LOCK();
6263145874Smlaier#else
6264145874Smlaier			icmp_error(m0, ICMP_UNREACH, ICMP_UNREACH_NEEDFRAG, 0,
6265171168Smlaier			    ifp->if_mtu);
6266126261Smlaier#endif
6267126258Smlaier			goto done;
6268126258Smlaier		} else
6269126258Smlaier			goto bad;
6270126258Smlaier	}
6271126258Smlaier
6272126258Smlaier	m1 = m0;
6273127145Smlaier#ifdef __FreeBSD__
6274126261Smlaier	/*
6275126261Smlaier	 * XXX: is cheaper + less error prone than own function
6276126261Smlaier	 */
6277126261Smlaier	NTOHS(ip->ip_len);
6278126261Smlaier	NTOHS(ip->ip_off);
6279126261Smlaier	error = ip_fragment(ip, &m0, ifp->if_mtu, ifp->if_hwassist, sw_csum);
6280126261Smlaier#else
6281126258Smlaier	error = ip_fragment(m0, ifp, ifp->if_mtu);
6282126261Smlaier#endif
6283127531Smlaier	if (error) {
6284127531Smlaier#ifndef __FreeBSD__	/* ip_fragment does not do m_freem() on FreeBSD */
6285127531Smlaier		m0 = NULL;
6286126261Smlaier#endif
6287126258Smlaier		goto bad;
6288127531Smlaier	}
6289126258Smlaier
6290126258Smlaier	for (m0 = m1; m0; m0 = m1) {
6291126258Smlaier		m1 = m0->m_nextpkt;
6292126258Smlaier		m0->m_nextpkt = 0;
6293127145Smlaier#ifdef __FreeBSD__
6294126261Smlaier		if (error == 0) {
6295126261Smlaier			PF_UNLOCK();
6296126261Smlaier			error = (*ifp->if_output)(ifp, m0, sintosa(dst),
6297126261Smlaier			    NULL);
6298126261Smlaier			PF_LOCK();
6299126261Smlaier		} else
6300126261Smlaier#else
6301126258Smlaier		if (error == 0)
6302126258Smlaier			error = (*ifp->if_output)(ifp, m0, sintosa(dst),
6303126258Smlaier			    NULL);
6304126258Smlaier		else
6305126261Smlaier#endif
6306126258Smlaier			m_freem(m0);
6307126258Smlaier	}
6308126258Smlaier
6309126258Smlaier	if (error == 0)
6310126258Smlaier		ipstat.ips_fragmented++;
6311126258Smlaier
6312126258Smlaierdone:
6313126258Smlaier	if (r->rt != PF_DUPTO)
6314126258Smlaier		*m = NULL;
6315126258Smlaier	if (ro == &iproute && ro->ro_rt)
6316126258Smlaier		RTFREE(ro->ro_rt);
6317126258Smlaier	return;
6318126258Smlaier
6319126258Smlaierbad:
6320126258Smlaier	m_freem(m0);
6321126258Smlaier	goto done;
6322126258Smlaier}
6323126258Smlaier#endif /* INET */
6324126258Smlaier
6325126258Smlaier#ifdef INET6
6326126258Smlaiervoid
6327126258Smlaierpf_route6(struct mbuf **m, struct pf_rule *r, int dir, struct ifnet *oifp,
6328171168Smlaier    struct pf_state *s, struct pf_pdesc *pd)
6329126258Smlaier{
6330126258Smlaier	struct mbuf		*m0;
6331126258Smlaier	struct route_in6	 ip6route;
6332126258Smlaier	struct route_in6	*ro;
6333126258Smlaier	struct sockaddr_in6	*dst;
6334126258Smlaier	struct ip6_hdr		*ip6;
6335126258Smlaier	struct ifnet		*ifp = NULL;
6336126258Smlaier	struct pf_addr		 naddr;
6337130613Smlaier	struct pf_src_node	*sn = NULL;
6338126258Smlaier	int			 error = 0;
6339126258Smlaier
6340126258Smlaier	if (m == NULL || *m == NULL || r == NULL ||
6341126258Smlaier	    (dir != PF_IN && dir != PF_OUT) || oifp == NULL)
6342126258Smlaier		panic("pf_route6: invalid parameters");
6343126258Smlaier
6344171168Smlaier	if (pd->pf_mtag->routed++ > 3) {
6345171168Smlaier		m0 = *m;
6346171168Smlaier		*m = NULL;
6347171168Smlaier		goto bad;
6348132303Smlaier	}
6349132303Smlaier
6350126258Smlaier	if (r->rt == PF_DUPTO) {
6351127145Smlaier#ifdef __FreeBSD__
6352132303Smlaier		if ((m0 = m_dup(*m, M_DONTWAIT)) == NULL)
6353126261Smlaier#else
6354132303Smlaier		if ((m0 = m_copym2(*m, 0, M_COPYALL, M_NOWAIT)) == NULL)
6355126261Smlaier#endif
6356126258Smlaier			return;
6357126258Smlaier	} else {
6358126258Smlaier		if ((r->rt == PF_REPLYTO) == (r->direction == dir))
6359126258Smlaier			return;
6360126258Smlaier		m0 = *m;
6361126258Smlaier	}
6362126258Smlaier
6363145836Smlaier	if (m0->m_len < sizeof(struct ip6_hdr)) {
6364145836Smlaier		DPFPRINTF(PF_DEBUG_URGENT,
6365145836Smlaier		    ("pf_route6: m0->m_len < sizeof(struct ip6_hdr)\n"));
6366145836Smlaier		goto bad;
6367145836Smlaier	}
6368126258Smlaier	ip6 = mtod(m0, struct ip6_hdr *);
6369126258Smlaier
6370126258Smlaier	ro = &ip6route;
6371126258Smlaier	bzero((caddr_t)ro, sizeof(*ro));
6372126258Smlaier	dst = (struct sockaddr_in6 *)&ro->ro_dst;
6373126258Smlaier	dst->sin6_family = AF_INET6;
6374126258Smlaier	dst->sin6_len = sizeof(*dst);
6375126258Smlaier	dst->sin6_addr = ip6->ip6_dst;
6376126258Smlaier
6377171168Smlaier	/* Cheat. XXX why only in the v6 case??? */
6378126258Smlaier	if (r->rt == PF_FASTROUTE) {
6379127145Smlaier#ifdef __FreeBSD__
6380132280Smlaier		m0->m_flags |= M_SKIP_FIREWALL;
6381126261Smlaier		PF_UNLOCK();
6382126261Smlaier		ip6_output(m0, NULL, NULL, 0, NULL, NULL, NULL);
6383126261Smlaier		PF_LOCK();
6384126261Smlaier#else
6385132280Smlaier		mtag = m_tag_get(PACKET_TAG_PF_GENERATED, 0, M_NOWAIT);
6386132280Smlaier		if (mtag == NULL)
6387132280Smlaier			goto bad;
6388132280Smlaier		m_tag_prepend(m0, mtag);
6389171168Smlaier		pd->pf_mtag->flags |= PF_TAG_GENERATED;
6390126258Smlaier		ip6_output(m0, NULL, NULL, 0, NULL, NULL);
6391126261Smlaier#endif
6392126258Smlaier		return;
6393126258Smlaier	}
6394126258Smlaier
6395145836Smlaier	if (TAILQ_EMPTY(&r->rpool.list)) {
6396145836Smlaier		DPFPRINTF(PF_DEBUG_URGENT,
6397145836Smlaier		    ("pf_route6: TAILQ_EMPTY(&r->rpool.list)\n"));
6398145836Smlaier		goto bad;
6399145836Smlaier	}
6400126258Smlaier	if (s == NULL) {
6401130613Smlaier		pf_map_addr(AF_INET6, r, (struct pf_addr *)&ip6->ip6_src,
6402130613Smlaier		    &naddr, NULL, &sn);
6403126258Smlaier		if (!PF_AZERO(&naddr, AF_INET6))
6404126258Smlaier			PF_ACPY((struct pf_addr *)&dst->sin6_addr,
6405126258Smlaier			    &naddr, AF_INET6);
6406130613Smlaier		ifp = r->rpool.cur->kif ? r->rpool.cur->kif->pfik_ifp : NULL;
6407126258Smlaier	} else {
6408126258Smlaier		if (!PF_AZERO(&s->rt_addr, AF_INET6))
6409126258Smlaier			PF_ACPY((struct pf_addr *)&dst->sin6_addr,
6410126258Smlaier			    &s->rt_addr, AF_INET6);
6411130613Smlaier		ifp = s->rt_kif ? s->rt_kif->pfik_ifp : NULL;
6412126258Smlaier	}
6413126258Smlaier	if (ifp == NULL)
6414126258Smlaier		goto bad;
6415126258Smlaier
6416126258Smlaier	if (oifp != ifp) {
6417127145Smlaier#ifdef __FreeBSD__
6418132303Smlaier		PF_UNLOCK();
6419145836Smlaier		if (pf_test6(PF_OUT, ifp, &m0, NULL, NULL) != PF_PASS) {
6420126261Smlaier			PF_LOCK();
6421132303Smlaier			goto bad;
6422132303Smlaier		} else if (m0 == NULL) {
6423132303Smlaier			PF_LOCK();
6424132303Smlaier			goto done;
6425132303Smlaier		}
6426132303Smlaier		PF_LOCK();
6427126261Smlaier#else
6428145836Smlaier		if (pf_test6(PF_OUT, ifp, &m0, NULL) != PF_PASS)
6429132303Smlaier			goto bad;
6430132303Smlaier		else if (m0 == NULL)
6431132303Smlaier			goto done;
6432126261Smlaier#endif
6433145836Smlaier		if (m0->m_len < sizeof(struct ip6_hdr)) {
6434145836Smlaier			DPFPRINTF(PF_DEBUG_URGENT,
6435145836Smlaier			    ("pf_route6: m0->m_len < sizeof(struct ip6_hdr)\n"));
6436145836Smlaier			goto bad;
6437145836Smlaier		}
6438132303Smlaier		ip6 = mtod(m0, struct ip6_hdr *);
6439126258Smlaier	}
6440126258Smlaier
6441126258Smlaier	/*
6442126258Smlaier	 * If the packet is too large for the outgoing interface,
6443126258Smlaier	 * send back an icmp6 error.
6444126258Smlaier	 */
6445171168Smlaier	if (IN6_IS_SCOPE_EMBED(&dst->sin6_addr))
6446126258Smlaier		dst->sin6_addr.s6_addr16[1] = htons(ifp->if_index);
6447126258Smlaier	if ((u_long)m0->m_pkthdr.len <= ifp->if_mtu) {
6448127145Smlaier#ifdef __FreeBSD__
6449126261Smlaier		PF_UNLOCK();
6450126261Smlaier#endif
6451126258Smlaier		error = nd6_output(ifp, ifp, m0, dst, NULL);
6452127145Smlaier#ifdef __FreeBSD__
6453126261Smlaier		PF_LOCK();
6454126261Smlaier#endif
6455126258Smlaier	} else {
6456126258Smlaier		in6_ifstat_inc(ifp, ifs6_in_toobig);
6457127145Smlaier#ifdef __FreeBSD__
6458126261Smlaier		if (r->rt != PF_DUPTO) {
6459126261Smlaier			PF_UNLOCK();
6460126261Smlaier			icmp6_error(m0, ICMP6_PACKET_TOO_BIG, 0, ifp->if_mtu);
6461126261Smlaier			PF_LOCK();
6462126261Smlaier		 } else
6463126261Smlaier#else
6464126258Smlaier		if (r->rt != PF_DUPTO)
6465126258Smlaier			icmp6_error(m0, ICMP6_PACKET_TOO_BIG, 0, ifp->if_mtu);
6466126258Smlaier		else
6467126261Smlaier#endif
6468126258Smlaier			goto bad;
6469126258Smlaier	}
6470126258Smlaier
6471126258Smlaierdone:
6472126258Smlaier	if (r->rt != PF_DUPTO)
6473126258Smlaier		*m = NULL;
6474126258Smlaier	return;
6475126258Smlaier
6476126258Smlaierbad:
6477126258Smlaier	m_freem(m0);
6478126258Smlaier	goto done;
6479126258Smlaier}
6480126258Smlaier#endif /* INET6 */
6481126258Smlaier
6482126258Smlaier
6483127145Smlaier#ifdef __FreeBSD__
6484126258Smlaier/*
6485132566Smlaier * FreeBSD supports cksum offloads for the following drivers.
6486137413Sru *  em(4), fxp(4), ixgb(4), lge(4), ndis(4), nge(4), re(4),
6487132566Smlaier *   ti(4), txp(4), xl(4)
6488132566Smlaier *
6489132566Smlaier * CSUM_DATA_VALID | CSUM_PSEUDO_HDR :
6490132566Smlaier *  network driver performed cksum including pseudo header, need to verify
6491132566Smlaier *   csum_data
6492132566Smlaier * CSUM_DATA_VALID :
6493132566Smlaier *  network driver performed cksum, needs to additional pseudo header
6494132566Smlaier *  cksum computation with partial csum_data(i.e. lack of H/W support for
6495132566Smlaier *  pseudo header, for instance hme(4), sk(4) and possibly gem(4))
6496132566Smlaier *
6497132566Smlaier * After validating the cksum of packet, set both flag CSUM_DATA_VALID and
6498132566Smlaier * CSUM_PSEUDO_HDR in order to avoid recomputation of the cksum in upper
6499132566Smlaier * TCP/UDP layer.
6500132566Smlaier * Also, set csum_data to 0xffff to force cksum validation.
6501126261Smlaier */
6502126261Smlaierint
6503126261Smlaierpf_check_proto_cksum(struct mbuf *m, int off, int len, u_int8_t p, sa_family_t af)
6504126261Smlaier{
6505126261Smlaier	u_int16_t sum = 0;
6506126261Smlaier	int hw_assist = 0;
6507126261Smlaier	struct ip *ip;
6508126261Smlaier
6509126261Smlaier	if (off < sizeof(struct ip) || len < sizeof(struct udphdr))
6510126261Smlaier		return (1);
6511126261Smlaier	if (m->m_pkthdr.len < off + len)
6512126261Smlaier		return (1);
6513126261Smlaier
6514126261Smlaier	switch (p) {
6515126261Smlaier	case IPPROTO_TCP:
6516126261Smlaier		if (m->m_pkthdr.csum_flags & CSUM_DATA_VALID) {
6517126261Smlaier			if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR) {
6518126261Smlaier				sum = m->m_pkthdr.csum_data;
6519126261Smlaier			} else {
6520126261Smlaier				ip = mtod(m, struct ip *);
6521126261Smlaier				sum = in_pseudo(ip->ip_src.s_addr,
6522135078Smlaier					ip->ip_dst.s_addr, htonl((u_short)len +
6523135078Smlaier					m->m_pkthdr.csum_data + IPPROTO_TCP));
6524126261Smlaier			}
6525126261Smlaier			sum ^= 0xffff;
6526126261Smlaier			++hw_assist;
6527126261Smlaier		}
6528126261Smlaier		break;
6529126261Smlaier	case IPPROTO_UDP:
6530126261Smlaier		if (m->m_pkthdr.csum_flags & CSUM_DATA_VALID) {
6531126261Smlaier			if (m->m_pkthdr.csum_flags & CSUM_PSEUDO_HDR) {
6532126261Smlaier				sum = m->m_pkthdr.csum_data;
6533126261Smlaier			} else {
6534126261Smlaier				ip = mtod(m, struct ip *);
6535126261Smlaier				sum = in_pseudo(ip->ip_src.s_addr,
6536126261Smlaier					ip->ip_dst.s_addr, htonl((u_short)len +
6537126261Smlaier					m->m_pkthdr.csum_data + IPPROTO_UDP));
6538126261Smlaier			}
6539126261Smlaier			sum ^= 0xffff;
6540126261Smlaier			++hw_assist;
6541126261Smlaier                }
6542126261Smlaier		break;
6543126261Smlaier	case IPPROTO_ICMP:
6544126261Smlaier#ifdef INET6
6545126261Smlaier	case IPPROTO_ICMPV6:
6546126261Smlaier#endif /* INET6 */
6547126261Smlaier		break;
6548126261Smlaier	default:
6549126261Smlaier		return (1);
6550126261Smlaier	}
6551126261Smlaier
6552126261Smlaier	if (!hw_assist) {
6553126261Smlaier		switch (af) {
6554126261Smlaier		case AF_INET:
6555126261Smlaier			if (p == IPPROTO_ICMP) {
6556126261Smlaier				if (m->m_len < off)
6557126261Smlaier					return (1);
6558126261Smlaier				m->m_data += off;
6559126261Smlaier				m->m_len -= off;
6560126261Smlaier				sum = in_cksum(m, len);
6561126261Smlaier				m->m_data -= off;
6562126261Smlaier				m->m_len += off;
6563126261Smlaier			} else {
6564126261Smlaier				if (m->m_len < sizeof(struct ip))
6565126261Smlaier					return (1);
6566126261Smlaier				sum = in4_cksum(m, p, off, len);
6567126261Smlaier			}
6568126261Smlaier			break;
6569126261Smlaier#ifdef INET6
6570126261Smlaier		case AF_INET6:
6571126261Smlaier			if (m->m_len < sizeof(struct ip6_hdr))
6572126261Smlaier				return (1);
6573126261Smlaier			sum = in6_cksum(m, p, off, len);
6574126261Smlaier			break;
6575126261Smlaier#endif /* INET6 */
6576126261Smlaier		default:
6577126261Smlaier			return (1);
6578126261Smlaier		}
6579126261Smlaier	}
6580126261Smlaier	if (sum) {
6581126261Smlaier		switch (p) {
6582126261Smlaier		case IPPROTO_TCP:
6583126261Smlaier			tcpstat.tcps_rcvbadsum++;
6584126261Smlaier			break;
6585126261Smlaier		case IPPROTO_UDP:
6586126261Smlaier			udpstat.udps_badsum++;
6587126261Smlaier			break;
6588126261Smlaier		case IPPROTO_ICMP:
6589126261Smlaier			icmpstat.icps_checksum++;
6590126261Smlaier			break;
6591126261Smlaier#ifdef INET6
6592126261Smlaier		case IPPROTO_ICMPV6:
6593126261Smlaier			icmp6stat.icp6s_checksum++;
6594126261Smlaier			break;
6595126261Smlaier#endif /* INET6 */
6596126261Smlaier		}
6597126261Smlaier		return (1);
6598132566Smlaier	} else {
6599132566Smlaier		if (p == IPPROTO_TCP || p == IPPROTO_UDP) {
6600132566Smlaier			m->m_pkthdr.csum_flags |=
6601132566Smlaier			    (CSUM_DATA_VALID | CSUM_PSEUDO_HDR);
6602132566Smlaier			m->m_pkthdr.csum_data = 0xffff;
6603132566Smlaier		}
6604126261Smlaier	}
6605126261Smlaier	return (0);
6606126261Smlaier}
6607171168Smlaier#else /* !__FreeBSD__ */
6608126261Smlaier/*
6609126258Smlaier * check protocol (tcp/udp/icmp/icmp6) checksum and set mbuf flag
6610126258Smlaier *   off is the offset where the protocol header starts
6611126258Smlaier *   len is the total length of protocol header plus payload
6612126258Smlaier * returns 0 when the checksum is valid, otherwise returns 1.
6613126258Smlaier */
6614126258Smlaierint
6615130613Smlaierpf_check_proto_cksum(struct mbuf *m, int off, int len, u_int8_t p,
6616130613Smlaier    sa_family_t af)
6617126258Smlaier{
6618126258Smlaier	u_int16_t flag_ok, flag_bad;
6619126258Smlaier	u_int16_t sum;
6620126258Smlaier
6621126258Smlaier	switch (p) {
6622126258Smlaier	case IPPROTO_TCP:
6623126258Smlaier		flag_ok = M_TCP_CSUM_IN_OK;
6624126258Smlaier		flag_bad = M_TCP_CSUM_IN_BAD;
6625126258Smlaier		break;
6626126258Smlaier	case IPPROTO_UDP:
6627126258Smlaier		flag_ok = M_UDP_CSUM_IN_OK;
6628126258Smlaier		flag_bad = M_UDP_CSUM_IN_BAD;
6629126258Smlaier		break;
6630126258Smlaier	case IPPROTO_ICMP:
6631126258Smlaier#ifdef INET6
6632126258Smlaier	case IPPROTO_ICMPV6:
6633126258Smlaier#endif /* INET6 */
6634126258Smlaier		flag_ok = flag_bad = 0;
6635126258Smlaier		break;
6636126258Smlaier	default:
6637126258Smlaier		return (1);
6638126258Smlaier	}
6639171168Smlaier	if (m->m_pkthdr.csum_flags & flag_ok)
6640126258Smlaier		return (0);
6641171168Smlaier	if (m->m_pkthdr.csum_flags & flag_bad)
6642126258Smlaier		return (1);
6643126258Smlaier	if (off < sizeof(struct ip) || len < sizeof(struct udphdr))
6644126258Smlaier		return (1);
6645126258Smlaier	if (m->m_pkthdr.len < off + len)
6646126258Smlaier		return (1);
6647145836Smlaier	switch (af) {
6648145836Smlaier#ifdef INET
6649126258Smlaier	case AF_INET:
6650126258Smlaier		if (p == IPPROTO_ICMP) {
6651126258Smlaier			if (m->m_len < off)
6652126258Smlaier				return (1);
6653126258Smlaier			m->m_data += off;
6654126258Smlaier			m->m_len -= off;
6655126258Smlaier			sum = in_cksum(m, len);
6656126258Smlaier			m->m_data -= off;
6657126258Smlaier			m->m_len += off;
6658126258Smlaier		} else {
6659126258Smlaier			if (m->m_len < sizeof(struct ip))
6660126258Smlaier				return (1);
6661126258Smlaier			sum = in4_cksum(m, p, off, len);
6662126258Smlaier		}
6663126258Smlaier		break;
6664145836Smlaier#endif /* INET */
6665126258Smlaier#ifdef INET6
6666126258Smlaier	case AF_INET6:
6667126258Smlaier		if (m->m_len < sizeof(struct ip6_hdr))
6668126258Smlaier			return (1);
6669126258Smlaier		sum = in6_cksum(m, p, off, len);
6670126258Smlaier		break;
6671126258Smlaier#endif /* INET6 */
6672126258Smlaier	default:
6673126258Smlaier		return (1);
6674126258Smlaier	}
6675126258Smlaier	if (sum) {
6676171168Smlaier		m->m_pkthdr.csum_flags |= flag_bad;
6677126258Smlaier		switch (p) {
6678126258Smlaier		case IPPROTO_TCP:
6679126258Smlaier			tcpstat.tcps_rcvbadsum++;
6680126258Smlaier			break;
6681126258Smlaier		case IPPROTO_UDP:
6682126258Smlaier			udpstat.udps_badsum++;
6683126258Smlaier			break;
6684126258Smlaier		case IPPROTO_ICMP:
6685126258Smlaier			icmpstat.icps_checksum++;
6686126258Smlaier			break;
6687126258Smlaier#ifdef INET6
6688126258Smlaier		case IPPROTO_ICMPV6:
6689126258Smlaier			icmp6stat.icp6s_checksum++;
6690126258Smlaier			break;
6691126258Smlaier#endif /* INET6 */
6692126258Smlaier		}
6693126258Smlaier		return (1);
6694126258Smlaier	}
6695171168Smlaier	m->m_pkthdr.csum_flags |= flag_ok;
6696126258Smlaier	return (0);
6697126258Smlaier}
6698171168Smlaier#endif /* __FreeBSD__ */
6699126258Smlaier
6700126258Smlaier#ifdef INET
6701126258Smlaierint
6702135920Smlaier#ifdef __FreeBSD__
6703145836Smlaierpf_test(int dir, struct ifnet *ifp, struct mbuf **m0,
6704145836Smlaier    struct ether_header *eh, struct inpcb *inp)
6705135920Smlaier#else
6706145836Smlaierpf_test(int dir, struct ifnet *ifp, struct mbuf **m0,
6707145836Smlaier    struct ether_header *eh)
6708135920Smlaier#endif
6709126258Smlaier{
6710130613Smlaier	struct pfi_kif		*kif;
6711130613Smlaier	u_short			 action, reason = 0, log = 0;
6712130613Smlaier	struct mbuf		*m = *m0;
6713130613Smlaier	struct ip		*h = NULL;	/* make the compiler happy */
6714130613Smlaier	struct pf_rule		*a = NULL, *r = &pf_default_rule, *tr, *nr;
6715130613Smlaier	struct pf_state		*s = NULL;
6716130613Smlaier	struct pf_ruleset	*ruleset = NULL;
6717130613Smlaier	struct pf_pdesc		 pd;
6718130613Smlaier	int			 off, dirndx, pqid = 0;
6719126258Smlaier
6720127145Smlaier#ifdef __FreeBSD__
6721126261Smlaier	PF_LOCK();
6722126261Smlaier#endif
6723171168Smlaier	if (!pf_status.running)
6724127145Smlaier#ifdef __FreeBSD__
6725171168Smlaier	{
6726126261Smlaier		PF_UNLOCK();
6727126261Smlaier#endif
6728171168Smlaier		return (PF_PASS);
6729171168Smlaier#ifdef __FreeBSD__
6730126261Smlaier	}
6731171168Smlaier#endif
6732126258Smlaier
6733171168Smlaier	memset(&pd, 0, sizeof(pd));
6734171168Smlaier	if ((pd.pf_mtag = pf_get_mtag(m)) == NULL) {
6735145836Smlaier#ifdef __FreeBSD__
6736171168Smlaier		PF_UNLOCK();
6737171168Smlaier#endif
6738171168Smlaier		DPFPRINTF(PF_DEBUG_URGENT,
6739171168Smlaier		    ("pf_test: pf_get_mtag returned NULL\n"));
6740171168Smlaier		return (PF_DROP);
6741171168Smlaier	}
6742171168Smlaier#ifdef __FreeBSD__
6743171168Smlaier	if (m->m_flags & M_SKIP_FIREWALL) {
6744171168Smlaier		PF_UNLOCK();
6745171168Smlaier		return (PF_PASS);
6746171168Smlaier	}
6747171168Smlaier#else
6748171168Smlaier	if (pd.pf_mtag->flags & PF_TAG_GENERATED)
6749171168Smlaier		return (PF_PASS);
6750171168Smlaier#endif
6751171168Smlaier
6752171168Smlaier#ifdef __FreeBSD__
6753145836Smlaier	/* XXX_IMPORT: later */
6754145836Smlaier#else
6755145836Smlaier	if (ifp->if_type == IFT_CARP && ifp->if_carpdev)
6756145836Smlaier		ifp = ifp->if_carpdev;
6757145836Smlaier#endif
6758145836Smlaier
6759171168Smlaier	kif = (struct pfi_kif *)ifp->if_pf_kif;
6760130613Smlaier	if (kif == NULL) {
6761130613Smlaier#ifdef __FreeBSD__
6762130613Smlaier		PF_UNLOCK();
6763130613Smlaier#endif
6764145836Smlaier		DPFPRINTF(PF_DEBUG_URGENT,
6765145836Smlaier		    ("pf_test: kif == NULL, if_xname %s\n", ifp->if_xname));
6766130613Smlaier		return (PF_DROP);
6767130613Smlaier	}
6768145836Smlaier	if (kif->pfik_flags & PFI_IFLAG_SKIP) {
6769145836Smlaier#ifdef __FreeBSD__
6770145836Smlaier		PF_UNLOCK();
6771145836Smlaier#endif
6772145836Smlaier		return (PF_PASS);
6773145836Smlaier	}
6774130613Smlaier
6775130613Smlaier#ifdef __FreeBSD__
6776126261Smlaier	M_ASSERTPKTHDR(m);
6777126261Smlaier#else
6778126258Smlaier#ifdef DIAGNOSTIC
6779126258Smlaier	if ((m->m_flags & M_PKTHDR) == 0)
6780126258Smlaier		panic("non-M_PKTHDR is passed to pf_test");
6781145836Smlaier#endif /* DIAGNOSTIC */
6782145836Smlaier#endif /* __FreeBSD__ */
6783126258Smlaier
6784126258Smlaier	if (m->m_pkthdr.len < (int)sizeof(*h)) {
6785126258Smlaier		action = PF_DROP;
6786126258Smlaier		REASON_SET(&reason, PFRES_SHORT);
6787126258Smlaier		log = 1;
6788126258Smlaier		goto done;
6789126258Smlaier	}
6790126258Smlaier
6791126258Smlaier	/* We do IP header normalization and packet reassembly here */
6792145836Smlaier	if (pf_normalize_ip(m0, dir, kif, &reason, &pd) != PF_PASS) {
6793126258Smlaier		action = PF_DROP;
6794126258Smlaier		goto done;
6795126258Smlaier	}
6796126258Smlaier	m = *m0;
6797126258Smlaier	h = mtod(m, struct ip *);
6798126258Smlaier
6799126258Smlaier	off = h->ip_hl << 2;
6800126258Smlaier	if (off < (int)sizeof(*h)) {
6801126258Smlaier		action = PF_DROP;
6802126258Smlaier		REASON_SET(&reason, PFRES_SHORT);
6803126258Smlaier		log = 1;
6804126258Smlaier		goto done;
6805126258Smlaier	}
6806126258Smlaier
6807126258Smlaier	pd.src = (struct pf_addr *)&h->ip_src;
6808126258Smlaier	pd.dst = (struct pf_addr *)&h->ip_dst;
6809130613Smlaier	PF_ACPY(&pd.baddr, dir == PF_OUT ? pd.src : pd.dst, AF_INET);
6810126258Smlaier	pd.ip_sum = &h->ip_sum;
6811126258Smlaier	pd.proto = h->ip_p;
6812126258Smlaier	pd.af = AF_INET;
6813126258Smlaier	pd.tos = h->ip_tos;
6814126258Smlaier	pd.tot_len = ntohs(h->ip_len);
6815145836Smlaier	pd.eh = eh;
6816126258Smlaier
6817126258Smlaier	/* handle fragments that didn't get reassembled by normalization */
6818126258Smlaier	if (h->ip_off & htons(IP_MF | IP_OFFMASK)) {
6819130613Smlaier		action = pf_test_fragment(&r, dir, kif, m, h,
6820126258Smlaier		    &pd, &a, &ruleset);
6821126258Smlaier		goto done;
6822126258Smlaier	}
6823126258Smlaier
6824126258Smlaier	switch (h->ip_p) {
6825126258Smlaier
6826126258Smlaier	case IPPROTO_TCP: {
6827126258Smlaier		struct tcphdr	th;
6828126258Smlaier
6829126258Smlaier		pd.hdr.tcp = &th;
6830126258Smlaier		if (!pf_pull_hdr(m, off, &th, sizeof(th),
6831126258Smlaier		    &action, &reason, AF_INET)) {
6832126258Smlaier			log = action != PF_PASS;
6833126258Smlaier			goto done;
6834126258Smlaier		}
6835126258Smlaier		if (dir == PF_IN && pf_check_proto_cksum(m, off,
6836126258Smlaier		    ntohs(h->ip_len) - off, IPPROTO_TCP, AF_INET)) {
6837171168Smlaier			REASON_SET(&reason, PFRES_PROTCKSUM);
6838126258Smlaier			action = PF_DROP;
6839126258Smlaier			goto done;
6840126258Smlaier		}
6841126258Smlaier		pd.p_len = pd.tot_len - off - (th.th_off << 2);
6842126258Smlaier		if ((th.th_flags & TH_ACK) && pd.p_len == 0)
6843126258Smlaier			pqid = 1;
6844130613Smlaier		action = pf_normalize_tcp(dir, kif, m, 0, off, h, &pd);
6845126258Smlaier		if (action == PF_DROP)
6846130613Smlaier			goto done;
6847130613Smlaier		action = pf_test_state_tcp(&s, dir, kif, m, off, h, &pd,
6848126258Smlaier		    &reason);
6849126258Smlaier		if (action == PF_PASS) {
6850130613Smlaier#if NPFSYNC
6851130613Smlaier			pfsync_update_state(s);
6852145836Smlaier#endif /* NPFSYNC */
6853126258Smlaier			r = s->rule.ptr;
6854130613Smlaier			a = s->anchor.ptr;
6855126258Smlaier			log = s->log;
6856126258Smlaier		} else if (s == NULL)
6857135920Smlaier#ifdef __FreeBSD__
6858130613Smlaier			action = pf_test_tcp(&r, &s, dir, kif,
6859145836Smlaier			    m, off, h, &pd, &a, &ruleset, NULL, inp);
6860135920Smlaier#else
6861135920Smlaier			action = pf_test_tcp(&r, &s, dir, kif,
6862145836Smlaier			    m, off, h, &pd, &a, &ruleset, &ipintrq);
6863135920Smlaier#endif
6864126258Smlaier		break;
6865126258Smlaier	}
6866126258Smlaier
6867126258Smlaier	case IPPROTO_UDP: {
6868126258Smlaier		struct udphdr	uh;
6869126258Smlaier
6870126258Smlaier		pd.hdr.udp = &uh;
6871126258Smlaier		if (!pf_pull_hdr(m, off, &uh, sizeof(uh),
6872126258Smlaier		    &action, &reason, AF_INET)) {
6873126258Smlaier			log = action != PF_PASS;
6874126258Smlaier			goto done;
6875126258Smlaier		}
6876126258Smlaier		if (dir == PF_IN && uh.uh_sum && pf_check_proto_cksum(m,
6877126258Smlaier		    off, ntohs(h->ip_len) - off, IPPROTO_UDP, AF_INET)) {
6878126258Smlaier			action = PF_DROP;
6879171168Smlaier			REASON_SET(&reason, PFRES_PROTCKSUM);
6880126258Smlaier			goto done;
6881126258Smlaier		}
6882130613Smlaier		if (uh.uh_dport == 0 ||
6883130613Smlaier		    ntohs(uh.uh_ulen) > m->m_pkthdr.len - off ||
6884130613Smlaier		    ntohs(uh.uh_ulen) < sizeof(struct udphdr)) {
6885130613Smlaier			action = PF_DROP;
6886171168Smlaier			REASON_SET(&reason, PFRES_SHORT);
6887130613Smlaier			goto done;
6888130613Smlaier		}
6889130613Smlaier		action = pf_test_state_udp(&s, dir, kif, m, off, h, &pd);
6890126258Smlaier		if (action == PF_PASS) {
6891130613Smlaier#if NPFSYNC
6892130613Smlaier			pfsync_update_state(s);
6893145836Smlaier#endif /* NPFSYNC */
6894126258Smlaier			r = s->rule.ptr;
6895126258Smlaier			a = s->anchor.ptr;
6896126258Smlaier			log = s->log;
6897126258Smlaier		} else if (s == NULL)
6898135920Smlaier#ifdef __FreeBSD__
6899130613Smlaier			action = pf_test_udp(&r, &s, dir, kif,
6900145836Smlaier			    m, off, h, &pd, &a, &ruleset, NULL, inp);
6901135920Smlaier#else
6902135920Smlaier			action = pf_test_udp(&r, &s, dir, kif,
6903145836Smlaier			    m, off, h, &pd, &a, &ruleset, &ipintrq);
6904135920Smlaier#endif
6905126258Smlaier		break;
6906126258Smlaier	}
6907126258Smlaier
6908126258Smlaier	case IPPROTO_ICMP: {
6909126258Smlaier		struct icmp	ih;
6910126258Smlaier
6911126258Smlaier		pd.hdr.icmp = &ih;
6912126258Smlaier		if (!pf_pull_hdr(m, off, &ih, ICMP_MINLEN,
6913126258Smlaier		    &action, &reason, AF_INET)) {
6914126258Smlaier			log = action != PF_PASS;
6915126258Smlaier			goto done;
6916126258Smlaier		}
6917126258Smlaier		if (dir == PF_IN && pf_check_proto_cksum(m, off,
6918126258Smlaier		    ntohs(h->ip_len) - off, IPPROTO_ICMP, AF_INET)) {
6919126258Smlaier			action = PF_DROP;
6920171168Smlaier			REASON_SET(&reason, PFRES_PROTCKSUM);
6921126258Smlaier			goto done;
6922126258Smlaier		}
6923145836Smlaier		action = pf_test_state_icmp(&s, dir, kif, m, off, h, &pd,
6924145836Smlaier		    &reason);
6925126258Smlaier		if (action == PF_PASS) {
6926130613Smlaier#if NPFSYNC
6927130613Smlaier			pfsync_update_state(s);
6928145836Smlaier#endif /* NPFSYNC */
6929126258Smlaier			r = s->rule.ptr;
6930126258Smlaier			a = s->anchor.ptr;
6931126258Smlaier			log = s->log;
6932126258Smlaier		} else if (s == NULL)
6933145836Smlaier#ifdef __FreeBSD__
6934130613Smlaier			action = pf_test_icmp(&r, &s, dir, kif,
6935145836Smlaier			    m, off, h, &pd, &a, &ruleset, NULL);
6936145836Smlaier#else
6937145836Smlaier			action = pf_test_icmp(&r, &s, dir, kif,
6938145836Smlaier			    m, off, h, &pd, &a, &ruleset, &ipintrq);
6939145836Smlaier#endif
6940126258Smlaier		break;
6941126258Smlaier	}
6942126258Smlaier
6943126258Smlaier	default:
6944130613Smlaier		action = pf_test_state_other(&s, dir, kif, &pd);
6945126258Smlaier		if (action == PF_PASS) {
6946130613Smlaier#if NPFSYNC
6947130613Smlaier			pfsync_update_state(s);
6948145836Smlaier#endif /* NPFSYNC */
6949126258Smlaier			r = s->rule.ptr;
6950126258Smlaier			a = s->anchor.ptr;
6951126258Smlaier			log = s->log;
6952126258Smlaier		} else if (s == NULL)
6953145836Smlaier#ifdef __FreeBSD__
6954130613Smlaier			action = pf_test_other(&r, &s, dir, kif, m, off, h,
6955145836Smlaier			    &pd, &a, &ruleset, NULL);
6956145836Smlaier#else
6957145836Smlaier			action = pf_test_other(&r, &s, dir, kif, m, off, h,
6958145836Smlaier			    &pd, &a, &ruleset, &ipintrq);
6959145836Smlaier#endif
6960126258Smlaier		break;
6961126258Smlaier	}
6962126258Smlaier
6963126258Smlaierdone:
6964126258Smlaier	if (action == PF_PASS && h->ip_hl > 5 &&
6965126258Smlaier	    !((s && s->allow_opts) || r->allow_opts)) {
6966126258Smlaier		action = PF_DROP;
6967145836Smlaier		REASON_SET(&reason, PFRES_IPOPTIONS);
6968126258Smlaier		log = 1;
6969126258Smlaier		DPFPRINTF(PF_DEBUG_MISC,
6970126258Smlaier		    ("pf: dropping packet with ip options\n"));
6971126258Smlaier	}
6972126258Smlaier
6973171168Smlaier	if ((s && s->tag) || r->rtableid)
6974171168Smlaier		pf_tag_packet(m, pd.pf_mtag, s ? s->tag : 0, r->rtableid);
6975145836Smlaier
6976126258Smlaier#ifdef ALTQ
6977126258Smlaier	if (action == PF_PASS && r->qid) {
6978171168Smlaier		if (pqid || (pd.tos & IPTOS_LOWDELAY))
6979171168Smlaier			pd.pf_mtag->qid = r->pqid;
6980171168Smlaier		else
6981171168Smlaier			pd.pf_mtag->qid = r->qid;
6982171168Smlaier		/* add hints for ecn */
6983171168Smlaier		pd.pf_mtag->af = AF_INET;
6984171168Smlaier		pd.pf_mtag->hdr = h;
6985126258Smlaier	}
6986145836Smlaier#endif /* ALTQ */
6987126258Smlaier
6988130613Smlaier	/*
6989130613Smlaier	 * connections redirected to loopback should not match sockets
6990130613Smlaier	 * bound specifically to loopback due to security implications,
6991130613Smlaier	 * see tcp_input() and in_pcblookup_listen().
6992130613Smlaier	 */
6993130613Smlaier	if (dir == PF_IN && action == PF_PASS && (pd.proto == IPPROTO_TCP ||
6994130613Smlaier	    pd.proto == IPPROTO_UDP) && s != NULL && s->nat_rule.ptr != NULL &&
6995130613Smlaier	    (s->nat_rule.ptr->action == PF_RDR ||
6996130613Smlaier	    s->nat_rule.ptr->action == PF_BINAT) &&
6997171168Smlaier	    (ntohl(pd.dst->v4.s_addr) >> IN_CLASSA_NSHIFT) == IN_LOOPBACKNET)
6998171168Smlaier		pd.pf_mtag->flags |= PF_TAG_TRANSLATE_LOCALHOST;
6999171168Smlaier
7000171168Smlaier	if (log) {
7001171168Smlaier		struct pf_rule *lr;
7002171168Smlaier
7003171168Smlaier		if (s != NULL && s->nat_rule.ptr != NULL &&
7004171168Smlaier		    s->nat_rule.ptr->log & PF_LOG_ALL)
7005171168Smlaier			lr = s->nat_rule.ptr;
7006171168Smlaier		else
7007171168Smlaier			lr = r;
7008171168Smlaier		PFLOG_PACKET(kif, h, m, AF_INET, dir, reason, lr, a, ruleset,
7009171168Smlaier		    &pd);
7010130613Smlaier	}
7011130613Smlaier
7012130613Smlaier	kif->pfik_bytes[0][dir == PF_OUT][action != PF_PASS] += pd.tot_len;
7013130613Smlaier	kif->pfik_packets[0][dir == PF_OUT][action != PF_PASS]++;
7014130613Smlaier
7015130613Smlaier	if (action == PF_PASS || r->action == PF_DROP) {
7016171168Smlaier		dirndx = (dir == PF_OUT);
7017171168Smlaier		r->packets[dirndx]++;
7018171168Smlaier		r->bytes[dirndx] += pd.tot_len;
7019130613Smlaier		if (a != NULL) {
7020171168Smlaier			a->packets[dirndx]++;
7021171168Smlaier			a->bytes[dirndx] += pd.tot_len;
7022130613Smlaier		}
7023130613Smlaier		if (s != NULL) {
7024130613Smlaier			if (s->nat_rule.ptr != NULL) {
7025171168Smlaier				s->nat_rule.ptr->packets[dirndx]++;
7026171168Smlaier				s->nat_rule.ptr->bytes[dirndx] += pd.tot_len;
7027130613Smlaier			}
7028130613Smlaier			if (s->src_node != NULL) {
7029171168Smlaier				s->src_node->packets[dirndx]++;
7030171168Smlaier				s->src_node->bytes[dirndx] += pd.tot_len;
7031130613Smlaier			}
7032130613Smlaier			if (s->nat_src_node != NULL) {
7033171168Smlaier				s->nat_src_node->packets[dirndx]++;
7034171168Smlaier				s->nat_src_node->bytes[dirndx] += pd.tot_len;
7035130613Smlaier			}
7036171168Smlaier			dirndx = (dir == s->direction) ? 0 : 1;
7037171168Smlaier			s->packets[dirndx]++;
7038171168Smlaier			s->bytes[dirndx] += pd.tot_len;
7039130613Smlaier		}
7040130613Smlaier		tr = r;
7041130613Smlaier		nr = (s != NULL) ? s->nat_rule.ptr : pd.nat_rule;
7042130613Smlaier		if (nr != NULL) {
7043130613Smlaier			struct pf_addr *x;
7044130613Smlaier			/*
7045130613Smlaier			 * XXX: we need to make sure that the addresses
7046130613Smlaier			 * passed to pfr_update_stats() are the same than
7047130613Smlaier			 * the addresses used during matching (pfr_match)
7048130613Smlaier			 */
7049130613Smlaier			if (r == &pf_default_rule) {
7050130613Smlaier				tr = nr;
7051130613Smlaier				x = (s == NULL || s->direction == dir) ?
7052130613Smlaier				    &pd.baddr : &pd.naddr;
7053130613Smlaier			} else
7054130613Smlaier				x = (s == NULL || s->direction == dir) ?
7055130613Smlaier				    &pd.naddr : &pd.baddr;
7056130613Smlaier			if (x == &pd.baddr || s == NULL) {
7057130613Smlaier				/* we need to change the address */
7058130613Smlaier				if (dir == PF_OUT)
7059130613Smlaier					pd.src = x;
7060130613Smlaier				else
7061130613Smlaier					pd.dst = x;
7062130613Smlaier			}
7063130613Smlaier		}
7064130613Smlaier		if (tr->src.addr.type == PF_ADDR_TABLE)
7065130613Smlaier			pfr_update_stats(tr->src.addr.p.tbl, (s == NULL ||
7066130613Smlaier			    s->direction == dir) ? pd.src : pd.dst, pd.af,
7067130613Smlaier			    pd.tot_len, dir == PF_OUT, r->action == PF_PASS,
7068145836Smlaier			    tr->src.neg);
7069130613Smlaier		if (tr->dst.addr.type == PF_ADDR_TABLE)
7070130613Smlaier			pfr_update_stats(tr->dst.addr.p.tbl, (s == NULL ||
7071130613Smlaier			    s->direction == dir) ? pd.dst : pd.src, pd.af,
7072130613Smlaier			    pd.tot_len, dir == PF_OUT, r->action == PF_PASS,
7073145836Smlaier			    tr->dst.neg);
7074130613Smlaier	}
7075130613Smlaier
7076130613Smlaier
7077126258Smlaier	if (action == PF_SYNPROXY_DROP) {
7078126258Smlaier		m_freem(*m0);
7079126258Smlaier		*m0 = NULL;
7080126258Smlaier		action = PF_PASS;
7081126258Smlaier	} else if (r->rt)
7082126258Smlaier		/* pf_route can free the mbuf causing *m0 to become NULL */
7083171168Smlaier		pf_route(m0, r, dir, ifp, s, &pd);
7084126258Smlaier
7085127145Smlaier#ifdef __FreeBSD__
7086126261Smlaier	PF_UNLOCK();
7087126261Smlaier#endif
7088126261Smlaier
7089126258Smlaier	return (action);
7090126258Smlaier}
7091126258Smlaier#endif /* INET */
7092126258Smlaier
7093126258Smlaier#ifdef INET6
7094126258Smlaierint
7095135920Smlaier#ifdef __FreeBSD__
7096145836Smlaierpf_test6(int dir, struct ifnet *ifp, struct mbuf **m0,
7097145836Smlaier    struct ether_header *eh, struct inpcb *inp)
7098135920Smlaier#else
7099145836Smlaierpf_test6(int dir, struct ifnet *ifp, struct mbuf **m0,
7100145836Smlaier    struct ether_header *eh)
7101135920Smlaier#endif
7102126258Smlaier{
7103130613Smlaier	struct pfi_kif		*kif;
7104130613Smlaier	u_short			 action, reason = 0, log = 0;
7105171168Smlaier	struct mbuf		*m = *m0, *n = NULL;
7106171168Smlaier	struct ip6_hdr		*h;
7107130613Smlaier	struct pf_rule		*a = NULL, *r = &pf_default_rule, *tr, *nr;
7108130613Smlaier	struct pf_state		*s = NULL;
7109130613Smlaier	struct pf_ruleset	*ruleset = NULL;
7110130613Smlaier	struct pf_pdesc		 pd;
7111169843Sdhartmei	int			 off, terminal = 0, dirndx, rh_cnt = 0;
7112126258Smlaier
7113127145Smlaier#ifdef __FreeBSD__
7114126261Smlaier	PF_LOCK();
7115126261Smlaier#endif
7116126261Smlaier
7117171168Smlaier	if (!pf_status.running)
7118127145Smlaier#ifdef __FreeBSD__
7119171168Smlaier	{
7120126261Smlaier		PF_UNLOCK();
7121126261Smlaier#endif
7122126258Smlaier		return (PF_PASS);
7123171168Smlaier#ifdef __FreeBSD__
7124126261Smlaier	}
7125171168Smlaier#endif
7126126258Smlaier
7127171168Smlaier	memset(&pd, 0, sizeof(pd));
7128171168Smlaier	if ((pd.pf_mtag = pf_get_mtag(m)) == NULL) {
7129145836Smlaier#ifdef __FreeBSD__
7130171168Smlaier		PF_UNLOCK();
7131171168Smlaier#endif
7132171168Smlaier		DPFPRINTF(PF_DEBUG_URGENT,
7133171168Smlaier		    ("pf_test6: pf_get_mtag returned NULL\n"));
7134171168Smlaier		return (PF_DROP);
7135171168Smlaier	}
7136171168Smlaier	if (pd.pf_mtag->flags & PF_TAG_GENERATED)
7137171168Smlaier		return (PF_PASS);
7138171168Smlaier
7139171168Smlaier#ifdef __FreeBSD__
7140145836Smlaier	/* XXX_IMPORT: later */
7141145836Smlaier#else
7142145836Smlaier	if (ifp->if_type == IFT_CARP && ifp->if_carpdev)
7143145836Smlaier		ifp = ifp->if_carpdev;
7144145836Smlaier#endif
7145145836Smlaier
7146171168Smlaier	kif = (struct pfi_kif *)ifp->if_pf_kif;
7147130613Smlaier	if (kif == NULL) {
7148130613Smlaier#ifdef __FreeBSD__
7149130613Smlaier		PF_UNLOCK();
7150130613Smlaier#endif
7151145836Smlaier		DPFPRINTF(PF_DEBUG_URGENT,
7152145836Smlaier		    ("pf_test6: kif == NULL, if_xname %s\n", ifp->if_xname));
7153130613Smlaier		return (PF_DROP);
7154130613Smlaier	}
7155145836Smlaier	if (kif->pfik_flags & PFI_IFLAG_SKIP) {
7156145836Smlaier#ifdef __FreeBSD__
7157145836Smlaier		PF_UNLOCK();
7158145836Smlaier#endif
7159145836Smlaier		return (PF_PASS);
7160145836Smlaier	}
7161130613Smlaier
7162130613Smlaier#ifdef __FreeBSD__
7163126261Smlaier	M_ASSERTPKTHDR(m);
7164126261Smlaier#else
7165126258Smlaier#ifdef DIAGNOSTIC
7166126258Smlaier	if ((m->m_flags & M_PKTHDR) == 0)
7167145836Smlaier		panic("non-M_PKTHDR is passed to pf_test6");
7168145836Smlaier#endif /* DIAGNOSTIC */
7169126258Smlaier#endif
7170126258Smlaier
7171171168Smlaier#ifdef __FreeBSD__
7172171168Smlaier	h = NULL;	/* make the compiler happy */
7173171168Smlaier#endif
7174171168Smlaier
7175126258Smlaier	if (m->m_pkthdr.len < (int)sizeof(*h)) {
7176126258Smlaier		action = PF_DROP;
7177126258Smlaier		REASON_SET(&reason, PFRES_SHORT);
7178126258Smlaier		log = 1;
7179126258Smlaier		goto done;
7180126258Smlaier	}
7181126258Smlaier
7182126258Smlaier	/* We do IP header normalization and packet reassembly here */
7183145836Smlaier	if (pf_normalize_ip6(m0, dir, kif, &reason, &pd) != PF_PASS) {
7184126258Smlaier		action = PF_DROP;
7185126258Smlaier		goto done;
7186126258Smlaier	}
7187126258Smlaier	m = *m0;
7188126258Smlaier	h = mtod(m, struct ip6_hdr *);
7189126258Smlaier
7190169843Sdhartmei#if 1
7191169843Sdhartmei	/*
7192169843Sdhartmei	 * we do not support jumbogram yet.  if we keep going, zero ip6_plen
7193169843Sdhartmei	 * will do something bad, so drop the packet for now.
7194169843Sdhartmei	 */
7195169843Sdhartmei	if (htons(h->ip6_plen) == 0) {
7196169843Sdhartmei		action = PF_DROP;
7197169843Sdhartmei		REASON_SET(&reason, PFRES_NORM);	/*XXX*/
7198169843Sdhartmei		goto done;
7199169843Sdhartmei	}
7200169843Sdhartmei#endif
7201169843Sdhartmei
7202126258Smlaier	pd.src = (struct pf_addr *)&h->ip6_src;
7203126258Smlaier	pd.dst = (struct pf_addr *)&h->ip6_dst;
7204130613Smlaier	PF_ACPY(&pd.baddr, dir == PF_OUT ? pd.src : pd.dst, AF_INET6);
7205126258Smlaier	pd.ip_sum = NULL;
7206126258Smlaier	pd.af = AF_INET6;
7207126258Smlaier	pd.tos = 0;
7208126258Smlaier	pd.tot_len = ntohs(h->ip6_plen) + sizeof(struct ip6_hdr);
7209145836Smlaier	pd.eh = eh;
7210126258Smlaier
7211126258Smlaier	off = ((caddr_t)h - m->m_data) + sizeof(struct ip6_hdr);
7212126258Smlaier	pd.proto = h->ip6_nxt;
7213126258Smlaier	do {
7214126258Smlaier		switch (pd.proto) {
7215126258Smlaier		case IPPROTO_FRAGMENT:
7216130613Smlaier			action = pf_test_fragment(&r, dir, kif, m, h,
7217126258Smlaier			    &pd, &a, &ruleset);
7218126258Smlaier			if (action == PF_DROP)
7219126258Smlaier				REASON_SET(&reason, PFRES_FRAG);
7220126258Smlaier			goto done;
7221169843Sdhartmei		case IPPROTO_ROUTING: {
7222169843Sdhartmei			struct ip6_rthdr rthdr;
7223169843Sdhartmei
7224169843Sdhartmei			if (rh_cnt++) {
7225169843Sdhartmei				DPFPRINTF(PF_DEBUG_MISC,
7226169843Sdhartmei				    ("pf: IPv6 more than one rthdr\n"));
7227169843Sdhartmei				action = PF_DROP;
7228169843Sdhartmei				REASON_SET(&reason, PFRES_IPOPTIONS);
7229169843Sdhartmei				log = 1;
7230169843Sdhartmei				goto done;
7231169843Sdhartmei			}
7232169843Sdhartmei			if (!pf_pull_hdr(m, off, &rthdr, sizeof(rthdr), NULL,
7233169843Sdhartmei			    &reason, pd.af)) {
7234169843Sdhartmei				DPFPRINTF(PF_DEBUG_MISC,
7235169843Sdhartmei				    ("pf: IPv6 short rthdr\n"));
7236169843Sdhartmei				action = PF_DROP;
7237169843Sdhartmei				REASON_SET(&reason, PFRES_SHORT);
7238169843Sdhartmei				log = 1;
7239169843Sdhartmei				goto done;
7240169843Sdhartmei			}
7241169843Sdhartmei			if (rthdr.ip6r_type == IPV6_RTHDR_TYPE_0) {
7242169843Sdhartmei				DPFPRINTF(PF_DEBUG_MISC,
7243169843Sdhartmei				    ("pf: IPv6 rthdr0\n"));
7244169843Sdhartmei				action = PF_DROP;
7245169843Sdhartmei				REASON_SET(&reason, PFRES_IPOPTIONS);
7246169843Sdhartmei				log = 1;
7247169843Sdhartmei				goto done;
7248169843Sdhartmei			}
7249169843Sdhartmei			/* fallthrough */
7250169843Sdhartmei		}
7251126258Smlaier		case IPPROTO_AH:
7252126258Smlaier		case IPPROTO_HOPOPTS:
7253126258Smlaier		case IPPROTO_DSTOPTS: {
7254126258Smlaier			/* get next header and header length */
7255126258Smlaier			struct ip6_ext	opt6;
7256126258Smlaier
7257126258Smlaier			if (!pf_pull_hdr(m, off, &opt6, sizeof(opt6),
7258145836Smlaier			    NULL, &reason, pd.af)) {
7259126258Smlaier				DPFPRINTF(PF_DEBUG_MISC,
7260126258Smlaier				    ("pf: IPv6 short opt\n"));
7261126258Smlaier				action = PF_DROP;
7262126258Smlaier				log = 1;
7263126258Smlaier				goto done;
7264126258Smlaier			}
7265126258Smlaier			if (pd.proto == IPPROTO_AH)
7266126258Smlaier				off += (opt6.ip6e_len + 2) * 4;
7267126258Smlaier			else
7268126258Smlaier				off += (opt6.ip6e_len + 1) * 8;
7269126258Smlaier			pd.proto = opt6.ip6e_nxt;
7270126258Smlaier			/* goto the next header */
7271126258Smlaier			break;
7272126258Smlaier		}
7273126258Smlaier		default:
7274126258Smlaier			terminal++;
7275126258Smlaier			break;
7276126258Smlaier		}
7277126258Smlaier	} while (!terminal);
7278126258Smlaier
7279171168Smlaier	/* if there's no routing header, use unmodified mbuf for checksumming */
7280171168Smlaier	if (!n)
7281171168Smlaier		n = m;
7282171168Smlaier
7283126258Smlaier	switch (pd.proto) {
7284126258Smlaier
7285126258Smlaier	case IPPROTO_TCP: {
7286126258Smlaier		struct tcphdr	th;
7287126258Smlaier
7288126258Smlaier		pd.hdr.tcp = &th;
7289126258Smlaier		if (!pf_pull_hdr(m, off, &th, sizeof(th),
7290126258Smlaier		    &action, &reason, AF_INET6)) {
7291126258Smlaier			log = action != PF_PASS;
7292126258Smlaier			goto done;
7293126258Smlaier		}
7294171168Smlaier		if (dir == PF_IN && pf_check_proto_cksum(n, off,
7295138387Sdhartmei		    ntohs(h->ip6_plen) - (off - sizeof(struct ip6_hdr)),
7296138387Sdhartmei		    IPPROTO_TCP, AF_INET6)) {
7297126258Smlaier			action = PF_DROP;
7298145836Smlaier			REASON_SET(&reason, PFRES_PROTCKSUM);
7299126258Smlaier			goto done;
7300126258Smlaier		}
7301126258Smlaier		pd.p_len = pd.tot_len - off - (th.th_off << 2);
7302130613Smlaier		action = pf_normalize_tcp(dir, kif, m, 0, off, h, &pd);
7303126258Smlaier		if (action == PF_DROP)
7304130613Smlaier			goto done;
7305130613Smlaier		action = pf_test_state_tcp(&s, dir, kif, m, off, h, &pd,
7306126258Smlaier		    &reason);
7307126258Smlaier		if (action == PF_PASS) {
7308130613Smlaier#if NPFSYNC
7309130613Smlaier			pfsync_update_state(s);
7310145836Smlaier#endif /* NPFSYNC */
7311126258Smlaier			r = s->rule.ptr;
7312130613Smlaier			a = s->anchor.ptr;
7313126258Smlaier			log = s->log;
7314126258Smlaier		} else if (s == NULL)
7315135920Smlaier#ifdef __FreeBSD__
7316130613Smlaier			action = pf_test_tcp(&r, &s, dir, kif,
7317145836Smlaier			    m, off, h, &pd, &a, &ruleset, NULL, inp);
7318135920Smlaier#else
7319135920Smlaier			action = pf_test_tcp(&r, &s, dir, kif,
7320145836Smlaier			    m, off, h, &pd, &a, &ruleset, &ip6intrq);
7321135920Smlaier#endif
7322126258Smlaier		break;
7323126258Smlaier	}
7324126258Smlaier
7325126258Smlaier	case IPPROTO_UDP: {
7326126258Smlaier		struct udphdr	uh;
7327126258Smlaier
7328126258Smlaier		pd.hdr.udp = &uh;
7329126258Smlaier		if (!pf_pull_hdr(m, off, &uh, sizeof(uh),
7330126258Smlaier		    &action, &reason, AF_INET6)) {
7331126258Smlaier			log = action != PF_PASS;
7332126258Smlaier			goto done;
7333126258Smlaier		}
7334171168Smlaier		if (dir == PF_IN && uh.uh_sum && pf_check_proto_cksum(n,
7335138387Sdhartmei		    off, ntohs(h->ip6_plen) - (off - sizeof(struct ip6_hdr)),
7336138387Sdhartmei		    IPPROTO_UDP, AF_INET6)) {
7337126258Smlaier			action = PF_DROP;
7338145836Smlaier			REASON_SET(&reason, PFRES_PROTCKSUM);
7339126258Smlaier			goto done;
7340126258Smlaier		}
7341130613Smlaier		if (uh.uh_dport == 0 ||
7342130613Smlaier		    ntohs(uh.uh_ulen) > m->m_pkthdr.len - off ||
7343130613Smlaier		    ntohs(uh.uh_ulen) < sizeof(struct udphdr)) {
7344130613Smlaier			action = PF_DROP;
7345171168Smlaier			REASON_SET(&reason, PFRES_SHORT);
7346130613Smlaier			goto done;
7347130613Smlaier		}
7348130613Smlaier		action = pf_test_state_udp(&s, dir, kif, m, off, h, &pd);
7349126258Smlaier		if (action == PF_PASS) {
7350130613Smlaier#if NPFSYNC
7351130613Smlaier			pfsync_update_state(s);
7352145836Smlaier#endif /* NPFSYNC */
7353126258Smlaier			r = s->rule.ptr;
7354130613Smlaier			a = s->anchor.ptr;
7355126258Smlaier			log = s->log;
7356126258Smlaier		} else if (s == NULL)
7357135920Smlaier#ifdef __FreeBSD__
7358130613Smlaier			action = pf_test_udp(&r, &s, dir, kif,
7359145836Smlaier			    m, off, h, &pd, &a, &ruleset, NULL, inp);
7360135920Smlaier#else
7361135920Smlaier			action = pf_test_udp(&r, &s, dir, kif,
7362145836Smlaier			    m, off, h, &pd, &a, &ruleset, &ip6intrq);
7363135920Smlaier#endif
7364126258Smlaier		break;
7365126258Smlaier	}
7366126258Smlaier
7367126258Smlaier	case IPPROTO_ICMPV6: {
7368126258Smlaier		struct icmp6_hdr	ih;
7369126258Smlaier
7370126258Smlaier		pd.hdr.icmp6 = &ih;
7371126258Smlaier		if (!pf_pull_hdr(m, off, &ih, sizeof(ih),
7372126258Smlaier		    &action, &reason, AF_INET6)) {
7373126258Smlaier			log = action != PF_PASS;
7374126258Smlaier			goto done;
7375126258Smlaier		}
7376171168Smlaier		if (dir == PF_IN && pf_check_proto_cksum(n, off,
7377145836Smlaier		    ntohs(h->ip6_plen) - (off - sizeof(struct ip6_hdr)),
7378138387Sdhartmei		    IPPROTO_ICMPV6, AF_INET6)) {
7379126258Smlaier			action = PF_DROP;
7380145836Smlaier			REASON_SET(&reason, PFRES_PROTCKSUM);
7381126258Smlaier			goto done;
7382126258Smlaier		}
7383130613Smlaier		action = pf_test_state_icmp(&s, dir, kif,
7384145836Smlaier		    m, off, h, &pd, &reason);
7385126258Smlaier		if (action == PF_PASS) {
7386130613Smlaier#if NPFSYNC
7387130613Smlaier			pfsync_update_state(s);
7388145836Smlaier#endif /* NPFSYNC */
7389126258Smlaier			r = s->rule.ptr;
7390130613Smlaier			a = s->anchor.ptr;
7391126258Smlaier			log = s->log;
7392126258Smlaier		} else if (s == NULL)
7393145836Smlaier#ifdef __FreeBSD__
7394130613Smlaier			action = pf_test_icmp(&r, &s, dir, kif,
7395145836Smlaier			    m, off, h, &pd, &a, &ruleset, NULL);
7396145836Smlaier#else
7397145836Smlaier			action = pf_test_icmp(&r, &s, dir, kif,
7398145836Smlaier			    m, off, h, &pd, &a, &ruleset, &ip6intrq);
7399145836Smlaier#endif
7400126258Smlaier		break;
7401126258Smlaier	}
7402126258Smlaier
7403126258Smlaier	default:
7404130613Smlaier		action = pf_test_state_other(&s, dir, kif, &pd);
7405130613Smlaier		if (action == PF_PASS) {
7406145836Smlaier#if NPFSYNC
7407145836Smlaier			pfsync_update_state(s);
7408145836Smlaier#endif /* NPFSYNC */
7409130613Smlaier			r = s->rule.ptr;
7410130613Smlaier			a = s->anchor.ptr;
7411130613Smlaier			log = s->log;
7412130613Smlaier		} else if (s == NULL)
7413145836Smlaier#ifdef __FreeBSD__
7414130613Smlaier			action = pf_test_other(&r, &s, dir, kif, m, off, h,
7415145836Smlaier			    &pd, &a, &ruleset, NULL);
7416145836Smlaier#else
7417145836Smlaier			action = pf_test_other(&r, &s, dir, kif, m, off, h,
7418145836Smlaier			    &pd, &a, &ruleset, &ip6intrq);
7419145836Smlaier#endif
7420126258Smlaier		break;
7421126258Smlaier	}
7422126258Smlaier
7423126258Smlaierdone:
7424169843Sdhartmei	/* handle dangerous IPv6 extension headers. */
7425169843Sdhartmei	if (action == PF_PASS && rh_cnt &&
7426169843Sdhartmei	    !((s && s->allow_opts) || r->allow_opts)) {
7427169843Sdhartmei		action = PF_DROP;
7428169843Sdhartmei		REASON_SET(&reason, PFRES_IPOPTIONS);
7429169843Sdhartmei		log = 1;
7430169843Sdhartmei		DPFPRINTF(PF_DEBUG_MISC,
7431169843Sdhartmei		    ("pf: dropping packet with dangerous v6 headers\n"));
7432169843Sdhartmei	}
7433126258Smlaier
7434171168Smlaier	if ((s && s->tag) || r->rtableid)
7435171168Smlaier		pf_tag_packet(m, pd.pf_mtag, s ? s->tag : 0, r->rtableid);
7436145836Smlaier
7437126258Smlaier#ifdef ALTQ
7438126258Smlaier	if (action == PF_PASS && r->qid) {
7439171168Smlaier		if (pd.tos & IPTOS_LOWDELAY)
7440171168Smlaier			pd.pf_mtag->qid = r->pqid;
7441171168Smlaier		else
7442171168Smlaier			pd.pf_mtag->qid = r->qid;
7443171168Smlaier		/* add hints for ecn */
7444171168Smlaier		pd.pf_mtag->af = AF_INET6;
7445171168Smlaier		pd.pf_mtag->hdr = h;
7446126258Smlaier	}
7447145836Smlaier#endif /* ALTQ */
7448126258Smlaier
7449130613Smlaier	if (dir == PF_IN && action == PF_PASS && (pd.proto == IPPROTO_TCP ||
7450130613Smlaier	    pd.proto == IPPROTO_UDP) && s != NULL && s->nat_rule.ptr != NULL &&
7451130613Smlaier	    (s->nat_rule.ptr->action == PF_RDR ||
7452130613Smlaier	    s->nat_rule.ptr->action == PF_BINAT) &&
7453171168Smlaier	    IN6_IS_ADDR_LOOPBACK(&pd.dst->v6))
7454171168Smlaier		pd.pf_mtag->flags |= PF_TAG_TRANSLATE_LOCALHOST;
7455171168Smlaier
7456171168Smlaier	if (log) {
7457171168Smlaier		struct pf_rule *lr;
7458171168Smlaier
7459171168Smlaier		if (s != NULL && s->nat_rule.ptr != NULL &&
7460171168Smlaier		    s->nat_rule.ptr->log & PF_LOG_ALL)
7461171168Smlaier			lr = s->nat_rule.ptr;
7462171168Smlaier		else
7463171168Smlaier			lr = r;
7464171168Smlaier		PFLOG_PACKET(kif, h, m, AF_INET6, dir, reason, lr, a, ruleset,
7465171168Smlaier		    &pd);
7466130613Smlaier	}
7467130613Smlaier
7468130613Smlaier	kif->pfik_bytes[1][dir == PF_OUT][action != PF_PASS] += pd.tot_len;
7469130613Smlaier	kif->pfik_packets[1][dir == PF_OUT][action != PF_PASS]++;
7470130613Smlaier
7471130613Smlaier	if (action == PF_PASS || r->action == PF_DROP) {
7472171168Smlaier		dirndx = (dir == PF_OUT);
7473171168Smlaier		r->packets[dirndx]++;
7474171168Smlaier		r->bytes[dirndx] += pd.tot_len;
7475130613Smlaier		if (a != NULL) {
7476171168Smlaier			a->packets[dirndx]++;
7477171168Smlaier			a->bytes[dirndx] += pd.tot_len;
7478130613Smlaier		}
7479130613Smlaier		if (s != NULL) {
7480130613Smlaier			if (s->nat_rule.ptr != NULL) {
7481171168Smlaier				s->nat_rule.ptr->packets[dirndx]++;
7482171168Smlaier				s->nat_rule.ptr->bytes[dirndx] += pd.tot_len;
7483130613Smlaier			}
7484130613Smlaier			if (s->src_node != NULL) {
7485171168Smlaier				s->src_node->packets[dirndx]++;
7486171168Smlaier				s->src_node->bytes[dirndx] += pd.tot_len;
7487130613Smlaier			}
7488130613Smlaier			if (s->nat_src_node != NULL) {
7489171168Smlaier				s->nat_src_node->packets[dirndx]++;
7490171168Smlaier				s->nat_src_node->bytes[dirndx] += pd.tot_len;
7491130613Smlaier			}
7492171168Smlaier			dirndx = (dir == s->direction) ? 0 : 1;
7493171168Smlaier			s->packets[dirndx]++;
7494171168Smlaier			s->bytes[dirndx] += pd.tot_len;
7495130613Smlaier		}
7496130613Smlaier		tr = r;
7497130613Smlaier		nr = (s != NULL) ? s->nat_rule.ptr : pd.nat_rule;
7498130613Smlaier		if (nr != NULL) {
7499130613Smlaier			struct pf_addr *x;
7500130613Smlaier			/*
7501130613Smlaier			 * XXX: we need to make sure that the addresses
7502130613Smlaier			 * passed to pfr_update_stats() are the same than
7503130613Smlaier			 * the addresses used during matching (pfr_match)
7504130613Smlaier			 */
7505130613Smlaier			if (r == &pf_default_rule) {
7506130613Smlaier				tr = nr;
7507130613Smlaier				x = (s == NULL || s->direction == dir) ?
7508130613Smlaier				    &pd.baddr : &pd.naddr;
7509130613Smlaier			} else {
7510130613Smlaier				x = (s == NULL || s->direction == dir) ?
7511130613Smlaier				    &pd.naddr : &pd.baddr;
7512130613Smlaier			}
7513130613Smlaier			if (x == &pd.baddr || s == NULL) {
7514130613Smlaier				if (dir == PF_OUT)
7515130613Smlaier					pd.src = x;
7516130613Smlaier				else
7517130613Smlaier					pd.dst = x;
7518130613Smlaier			}
7519130613Smlaier		}
7520130613Smlaier		if (tr->src.addr.type == PF_ADDR_TABLE)
7521130613Smlaier			pfr_update_stats(tr->src.addr.p.tbl, (s == NULL ||
7522130613Smlaier			    s->direction == dir) ? pd.src : pd.dst, pd.af,
7523130613Smlaier			    pd.tot_len, dir == PF_OUT, r->action == PF_PASS,
7524145836Smlaier			    tr->src.neg);
7525130613Smlaier		if (tr->dst.addr.type == PF_ADDR_TABLE)
7526130613Smlaier			pfr_update_stats(tr->dst.addr.p.tbl, (s == NULL ||
7527130613Smlaier			    s->direction == dir) ? pd.dst : pd.src, pd.af,
7528130613Smlaier			    pd.tot_len, dir == PF_OUT, r->action == PF_PASS,
7529145836Smlaier			    tr->dst.neg);
7530130613Smlaier	}
7531130613Smlaier
7532130613Smlaier
7533126258Smlaier	if (action == PF_SYNPROXY_DROP) {
7534126258Smlaier		m_freem(*m0);
7535126258Smlaier		*m0 = NULL;
7536126258Smlaier		action = PF_PASS;
7537126258Smlaier	} else if (r->rt)
7538126258Smlaier		/* pf_route6 can free the mbuf causing *m0 to become NULL */
7539171168Smlaier		pf_route6(m0, r, dir, ifp, s, &pd);
7540126258Smlaier
7541127145Smlaier#ifdef __FreeBSD__
7542126261Smlaier	PF_UNLOCK();
7543126261Smlaier#endif
7544126258Smlaier	return (action);
7545126258Smlaier}
7546126258Smlaier#endif /* INET6 */
7547145836Smlaier
7548145836Smlaierint
7549145836Smlaierpf_check_congestion(struct ifqueue *ifq)
7550145836Smlaier{
7551145836Smlaier#ifdef __FreeBSD__
7552145836Smlaier	/* XXX_IMPORT: later */
7553145836Smlaier	return (0);
7554145836Smlaier#else
7555145836Smlaier	if (ifq->ifq_congestion)
7556145836Smlaier		return (1);
7557145836Smlaier	else
7558145836Smlaier		return (0);
7559145836Smlaier#endif
7560145836Smlaier}
7561