1/*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1982, 1986, 1990, 1993
5 *	The Regents of the University of California.
6 * Copyright (c) 2010-2011 Juniper Networks, Inc.
7 * All rights reserved.
8 *
9 * Portions of this software were developed by Robert N. M. Watson under
10 * contract to Juniper Networks, Inc.
11 *
12 * Redistribution and use in source and binary forms, with or without
13 * modification, are permitted provided that the following conditions
14 * are met:
15 * 1. Redistributions of source code must retain the above copyright
16 *    notice, this list of conditions and the following disclaimer.
17 * 2. Redistributions in binary form must reproduce the above copyright
18 *    notice, this list of conditions and the following disclaimer in the
19 *    documentation and/or other materials provided with the distribution.
20 * 3. Neither the name of the University nor the names of its contributors
21 *    may be used to endorse or promote products derived from this software
22 *    without specific prior written permission.
23 *
24 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 * SUCH DAMAGE.
35 */
36
37#ifndef _NETINET_IN_PCB_H_
38#define _NETINET_IN_PCB_H_
39
40#include <sys/queue.h>
41#include <sys/epoch.h>
42#include <sys/_lock.h>
43#include <sys/_mutex.h>
44#include <sys/_rwlock.h>
45#include <sys/_smr.h>
46#include <net/route.h>
47
48#ifdef _KERNEL
49#include <sys/lock.h>
50#include <sys/proc.h>
51#include <sys/rwlock.h>
52#include <sys/sysctl.h>
53#include <net/vnet.h>
54#include <vm/uma.h>
55#endif
56#include <sys/ck.h>
57
58/*
59 * struct inpcb is the common protocol control block structure used in most
60 * IP transport protocols.
61 *
62 * Pointers to local and foreign host table entries, local and foreign socket
63 * numbers, and pointers up (to a socket structure) and down (to a
64 * protocol-specific control block) are stored here.
65 */
66CK_LIST_HEAD(inpcbhead, inpcb);
67CK_LIST_HEAD(inpcbporthead, inpcbport);
68CK_LIST_HEAD(inpcblbgrouphead, inpcblbgroup);
69typedef	uint64_t	inp_gen_t;
70
71/*
72 * PCB with AF_INET6 null bind'ed laddr can receive AF_INET input packet.
73 * So, AF_INET6 null laddr is also used as AF_INET null laddr, by utilizing
74 * the following structure.  This requires padding always be zeroed out,
75 * which is done right after inpcb allocation and stays through its lifetime.
76 */
77struct in_addr_4in6 {
78	u_int32_t	ia46_pad32[3];
79	struct	in_addr	ia46_addr4;
80};
81
82union in_dependaddr {
83	struct in_addr_4in6 id46_addr;
84	struct in6_addr	id6_addr;
85};
86
87/*
88 * NOTE: ipv6 addrs should be 64-bit aligned, per RFC 2553.  in_conninfo has
89 * some extra padding to accomplish this.
90 * NOTE 2: tcp_syncache.c uses first 5 32-bit words, which identify fport,
91 * lport, faddr to generate hash, so these fields shouldn't be moved.
92 */
93struct in_endpoints {
94	u_int16_t	ie_fport;		/* foreign port */
95	u_int16_t	ie_lport;		/* local port */
96	/* protocol dependent part, local and foreign addr */
97	union in_dependaddr ie_dependfaddr;	/* foreign host table entry */
98	union in_dependaddr ie_dependladdr;	/* local host table entry */
99#define	ie_faddr	ie_dependfaddr.id46_addr.ia46_addr4
100#define	ie_laddr	ie_dependladdr.id46_addr.ia46_addr4
101#define	ie6_faddr	ie_dependfaddr.id6_addr
102#define	ie6_laddr	ie_dependladdr.id6_addr
103	u_int32_t	ie6_zoneid;		/* scope zone id */
104};
105
106/*
107 * XXX The defines for inc_* are hacks and should be changed to direct
108 * references.
109 */
110struct in_conninfo {
111	u_int8_t	inc_flags;
112	u_int8_t	inc_len;
113	u_int16_t	inc_fibnum;	/* XXX was pad, 16 bits is plenty */
114	/* protocol dependent part */
115	struct	in_endpoints inc_ie;
116};
117
118/*
119 * Flags for inc_flags.
120 */
121#define	INC_ISIPV6	0x01
122#define	INC_IPV6MINMTU	0x02
123
124#define	inc_fport	inc_ie.ie_fport
125#define	inc_lport	inc_ie.ie_lport
126#define	inc_faddr	inc_ie.ie_faddr
127#define	inc_laddr	inc_ie.ie_laddr
128#define	inc6_faddr	inc_ie.ie6_faddr
129#define	inc6_laddr	inc_ie.ie6_laddr
130#define	inc6_zoneid	inc_ie.ie6_zoneid
131
132#if defined(_KERNEL) || defined(_WANT_INPCB)
133/*
134 * struct inpcb captures the network layer state for TCP, UDP, and raw IPv4 and
135 * IPv6 sockets.  In the case of TCP and UDP, further per-connection state is
136 * located in a larger protocol specific structure that embeds inpcb in it.
137 * Almost all fields of struct inpcb are static after creation or protected by
138 * a per-inpcb rwlock, inp_lock.
139 *
140 * A inpcb database is indexed by addresses/ports hash as well as list of
141 * all pcbs that belong to a certain proto. Database lookups or list traversals
142 * are be performed inside SMR section. Once desired PCB is found its own
143 * lock is to be obtained and SMR section exited.
144 *
145 * Key:
146 * (c) - Constant after initialization
147 * (e) - Protected by the SMR section
148 * (i) - Protected by the inpcb lock
149 * (p) - Protected by the pcbinfo lock for the inpcb
150 * (h) - Protected by the pcbhash lock for the inpcb
151 * (s) - Protected by another subsystem's locks
152 * (x) - Undefined locking
153 *
154 * A few other notes:
155 *
156 * When a read lock is held, stability of the field is guaranteed; to write
157 * to a field, a write lock must generally be held.
158 *
159 * netinet/netinet6-layer code should not assume that the inp_socket pointer
160 * is safe to dereference without inp_lock being held, there may be
161 * close(2)-related races.
162 *
163 * The inp_vflag field is overloaded, and would otherwise ideally be (c).
164 */
165struct icmp6_filter;
166struct inpcbpolicy;
167struct m_snd_tag;
168struct inpcb {
169	/* Cache line #1 (amd64) */
170	CK_LIST_ENTRY(inpcb) inp_hash_exact;	/* hash table linkage */
171	CK_LIST_ENTRY(inpcb) inp_hash_wild;	/* hash table linkage */
172	struct rwlock	inp_lock;
173	/* Cache line #2 (amd64) */
174#define	inp_start_zero	inp_refcount
175#define	inp_zero_size	(sizeof(struct inpcb) - \
176			    offsetof(struct inpcb, inp_start_zero))
177	u_int	inp_refcount;		/* (i) refcount */
178	int	inp_flags;		/* (i) generic IP/datagram flags */
179	int	inp_flags2;		/* (i) generic IP/datagram flags #2*/
180	uint8_t inp_numa_domain;	/* numa domain */
181	struct	socket *inp_socket;	/* (i) back pointer to socket */
182	struct	inpcbinfo *inp_pcbinfo;	/* (c) PCB list info */
183	struct	ucred	*inp_cred;	/* (c) cache of socket cred */
184	u_int32_t inp_flow;		/* (i) IPv6 flow information */
185	u_char	inp_vflag;		/* (i) IP version flag (v4/v6) */
186	u_char	inp_ip_ttl;		/* (i) time to live proto */
187	u_char	inp_ip_p;		/* (c) protocol proto */
188	u_char	inp_ip_minttl;		/* (i) minimum TTL or drop */
189	uint32_t inp_flowid;		/* (x) flow id / queue id */
190	smr_seq_t inp_smr;		/* (i) sequence number at disconnect */
191	struct m_snd_tag *inp_snd_tag;	/* (i) send tag for outgoing mbufs */
192	uint32_t inp_flowtype;		/* (x) M_HASHTYPE value */
193
194	/* Local and foreign ports, local and foreign addr. */
195	struct	in_conninfo inp_inc;	/* (i,h) list for PCB's local port */
196
197	/* MAC and IPSEC policy information. */
198	struct	label *inp_label;	/* (i) MAC label */
199	struct	inpcbpolicy *inp_sp;    /* (s) for IPSEC */
200
201	/* Protocol-dependent part; options. */
202	struct {
203		u_char	inp_ip_tos;		/* (i) type of service proto */
204		struct mbuf		*inp_options;	/* (i) IP options */
205		struct ip_moptions	*inp_moptions;	/* (i) mcast options */
206	};
207	struct {
208		/* (i) IP options */
209		struct mbuf		*in6p_options;
210		/* (i) IP6 options for outgoing packets */
211		struct ip6_pktopts	*in6p_outputopts;
212		/* (i) IP multicast options */
213		struct ip6_moptions	*in6p_moptions;
214		/* (i) ICMPv6 code type filter */
215		struct icmp6_filter	*in6p_icmp6filt;
216		/* (i) IPV6_CHECKSUM setsockopt */
217		int	in6p_cksum;
218		short	in6p_hops;
219	};
220	CK_LIST_ENTRY(inpcb) inp_portlist;	/* (r:e/w:h) port list */
221	struct	inpcbport *inp_phd;	/* (r:e/w:h) head of this list */
222	inp_gen_t	inp_gencnt;	/* (c) generation count */
223	void		*spare_ptr;	/* Spare pointer. */
224	rt_gen_t	inp_rt_cookie;	/* generation for route entry */
225	union {				/* cached L3 information */
226		struct route inp_route;
227		struct route_in6 inp_route6;
228	};
229	CK_LIST_ENTRY(inpcb) inp_list;	/* (r:e/w:p) all PCBs for proto */
230};
231#endif	/* _KERNEL */
232
233#define	inp_fport	inp_inc.inc_fport
234#define	inp_lport	inp_inc.inc_lport
235#define	inp_faddr	inp_inc.inc_faddr
236#define	inp_laddr	inp_inc.inc_laddr
237
238#define	in6p_faddr	inp_inc.inc6_faddr
239#define	in6p_laddr	inp_inc.inc6_laddr
240#define	in6p_zoneid	inp_inc.inc6_zoneid
241
242#define	inp_vnet	inp_pcbinfo->ipi_vnet
243
244/*
245 * The range of the generation count, as used in this implementation, is 9e19.
246 * We would have to create 300 billion connections per second for this number
247 * to roll over in a year.  This seems sufficiently unlikely that we simply
248 * don't concern ourselves with that possibility.
249 */
250
251/*
252 * Interface exported to userland by various protocols which use inpcbs.  Hack
253 * alert -- only define if struct xsocket is in scope.
254 * Fields prefixed with "xi_" are unique to this structure, and the rest
255 * match fields in the struct inpcb, to ease coding and porting.
256 *
257 * Legend:
258 * (s) - used by userland utilities in src
259 * (p) - used by utilities in ports
260 * (3) - is known to be used by third party software not in ports
261 * (n) - no known usage
262 */
263#ifdef _SYS_SOCKETVAR_H_
264struct xinpcb {
265	ksize_t		xi_len;			/* length of this structure */
266	struct xsocket	xi_socket;		/* (s,p) */
267	struct in_conninfo inp_inc;		/* (s,p) */
268	uint64_t	inp_gencnt;		/* (s,p) */
269	int64_t		inp_spare64[5];
270	uint32_t	inp_flow;		/* (s) */
271	uint32_t	inp_flowid;		/* (s) */
272	uint32_t	inp_flowtype;		/* (s) */
273	int32_t		inp_flags;		/* (s,p) */
274	int32_t		inp_flags2;		/* (s) */
275	uint32_t	inp_unused;
276	int32_t		in6p_cksum;		/* (n) */
277	int32_t		inp_spare32[4];
278	uint16_t	in6p_hops;		/* (n) */
279	uint8_t		inp_ip_tos;		/* (n) */
280	int8_t		pad8;
281	uint8_t		inp_vflag;		/* (s,p) */
282	uint8_t		inp_ip_ttl;		/* (n) */
283	uint8_t		inp_ip_p;		/* (n) */
284	uint8_t		inp_ip_minttl;		/* (n) */
285	int8_t		inp_spare8[4];
286} __aligned(8);
287
288struct xinpgen {
289	ksize_t	xig_len;	/* length of this structure */
290	u_int		xig_count;	/* number of PCBs at this time */
291	uint32_t	_xig_spare32;
292	inp_gen_t	xig_gen;	/* generation count at this time */
293	so_gen_t	xig_sogen;	/* socket generation count this time */
294	uint64_t	_xig_spare64[4];
295} __aligned(8);
296
297struct sockopt_parameters {
298	struct in_conninfo sop_inc;
299	uint64_t sop_id;
300	int sop_level;
301	int sop_optname;
302	char sop_optval[];
303};
304
305#ifdef	_KERNEL
306int	sysctl_setsockopt(SYSCTL_HANDLER_ARGS, struct inpcbinfo *pcbinfo,
307	    int (*ctloutput_set)(struct inpcb *, struct sockopt *));
308void	in_pcbtoxinpcb(const struct inpcb *, struct xinpcb *);
309#endif
310#endif /* _SYS_SOCKETVAR_H_ */
311
312#ifdef _KERNEL
313/*
314 * Per-VNET pcb database for each high-level protocol (UDP, TCP, ...) in both
315 * IPv4 and IPv6.
316 *
317 * The pcbs are protected with SMR section and thus all lists in inpcbinfo
318 * are CK-lists.  Locking is required to insert a pcb into database. Two
319 * locks are provided: one for the hash and one for the global list of pcbs,
320 * as well as overall count and generation count.
321 *
322 * Locking key:
323 *
324 * (c) Constant or nearly constant after initialisation
325 * (e) Protected by SMR section
326 * (g) Locked by ipi_lock
327 * (h) Locked by ipi_hash_lock
328 */
329struct inpcbinfo {
330	/*
331	 * Global lock protecting inpcb list modification
332	 */
333	struct mtx		 ipi_lock;
334	struct inpcbhead	 ipi_listhead;		/* (r:e/w:g) */
335	u_int			 ipi_count;		/* (g) */
336
337	/*
338	 * Generation count -- incremented each time a connection is allocated
339	 * or freed.
340	 */
341	u_quad_t		 ipi_gencnt;		/* (g) */
342
343	/*
344	 * Fields associated with port lookup and allocation.
345	 */
346	u_short			 ipi_lastport;		/* (h) */
347	u_short			 ipi_lastlow;		/* (h) */
348	u_short			 ipi_lasthi;		/* (h) */
349
350	/*
351	 * UMA zone from which inpcbs are allocated for this protocol.
352	 */
353	uma_zone_t		 ipi_zone;		/* (c) */
354	uma_zone_t		 ipi_portzone;		/* (c) */
355	smr_t			 ipi_smr;		/* (c) */
356
357	/*
358	 * Global hash of inpcbs, hashed by local and foreign addresses and
359	 * port numbers.  The "exact" hash holds PCBs connected to a foreign
360	 * address, and "wild" holds the rest.
361	 */
362	struct mtx		 ipi_hash_lock;
363	struct inpcbhead 	*ipi_hash_exact;	/* (r:e/w:h) */
364	struct inpcbhead 	*ipi_hash_wild;		/* (r:e/w:h) */
365	u_long			 ipi_hashmask;		/* (c) */
366
367	/*
368	 * Global hash of inpcbs, hashed by only local port number.
369	 */
370	struct inpcbporthead	*ipi_porthashbase;	/* (h) */
371	u_long			 ipi_porthashmask;	/* (h) */
372
373	/*
374	 * Load balance groups used for the SO_REUSEPORT_LB option,
375	 * hashed by local port.
376	 */
377	struct	inpcblbgrouphead *ipi_lbgrouphashbase;	/* (r:e/w:h) */
378	u_long			 ipi_lbgrouphashmask;	/* (h) */
379
380	/*
381	 * Pointer to network stack instance
382	 */
383	struct vnet		*ipi_vnet;		/* (c) */
384};
385
386/*
387 * Global allocation storage for each high-level protocol (UDP, TCP, ...).
388 * Each corresponding per-VNET inpcbinfo points into this one.
389 */
390struct inpcbstorage {
391	uma_zone_t	ips_zone;
392	uma_zone_t	ips_portzone;
393	uma_init	ips_pcbinit;
394	size_t		ips_size;
395	const char *	ips_zone_name;
396	const char *	ips_portzone_name;
397	const char *	ips_infolock_name;
398	const char *	ips_hashlock_name;
399};
400
401#define INPCBSTORAGE_DEFINE(prot, ppcb, lname, zname, iname, hname)	\
402static int								\
403prot##_inpcb_init(void *mem, int size __unused, int flags __unused)	\
404{									\
405	struct inpcb *inp = mem;					\
406									\
407	rw_init_flags(&inp->inp_lock, lname, RW_RECURSE | RW_DUPOK);	\
408	return (0);							\
409}									\
410static struct inpcbstorage prot = {					\
411	.ips_size = sizeof(struct ppcb),				\
412	.ips_pcbinit = prot##_inpcb_init,				\
413	.ips_zone_name = zname,						\
414	.ips_portzone_name = zname " ports",				\
415	.ips_infolock_name = iname,					\
416	.ips_hashlock_name = hname,					\
417};									\
418SYSINIT(prot##_inpcbstorage_init, SI_SUB_PROTO_DOMAIN,			\
419    SI_ORDER_SECOND, in_pcbstorage_init, &prot);			\
420SYSUNINIT(prot##_inpcbstorage_uninit, SI_SUB_PROTO_DOMAIN,		\
421    SI_ORDER_SECOND, in_pcbstorage_destroy, &prot)
422
423/*
424 * Load balance groups used for the SO_REUSEPORT_LB socket option. Each group
425 * (or unique address:port combination) can be re-used at most
426 * INPCBLBGROUP_SIZMAX (256) times. The inpcbs are stored in il_inp which
427 * is dynamically resized as processes bind/unbind to that specific group.
428 */
429struct inpcblbgroup {
430	CK_LIST_ENTRY(inpcblbgroup) il_list;
431	struct epoch_context il_epoch_ctx;
432	struct ucred	*il_cred;
433	uint16_t	il_lport;			/* (c) */
434	u_char		il_vflag;			/* (c) */
435	uint8_t		il_numa_domain;
436	uint32_t	il_pad2;
437	union in_dependaddr il_dependladdr;		/* (c) */
438#define	il_laddr	il_dependladdr.id46_addr.ia46_addr4
439#define	il6_laddr	il_dependladdr.id6_addr
440	uint32_t	il_inpsiz; /* max count in il_inp[] (h) */
441	uint32_t	il_inpcnt; /* cur count in il_inp[] (h) */
442	struct inpcb	*il_inp[];			/* (h) */
443};
444
445#define INP_LOCK_DESTROY(inp)	rw_destroy(&(inp)->inp_lock)
446#define INP_RLOCK(inp)		rw_rlock(&(inp)->inp_lock)
447#define INP_WLOCK(inp)		rw_wlock(&(inp)->inp_lock)
448#define INP_TRY_RLOCK(inp)	rw_try_rlock(&(inp)->inp_lock)
449#define INP_TRY_WLOCK(inp)	rw_try_wlock(&(inp)->inp_lock)
450#define INP_RUNLOCK(inp)	rw_runlock(&(inp)->inp_lock)
451#define INP_WUNLOCK(inp)	rw_wunlock(&(inp)->inp_lock)
452#define INP_UNLOCK(inp)		rw_unlock(&(inp)->inp_lock)
453#define	INP_TRY_UPGRADE(inp)	rw_try_upgrade(&(inp)->inp_lock)
454#define	INP_DOWNGRADE(inp)	rw_downgrade(&(inp)->inp_lock)
455#define	INP_WLOCKED(inp)	rw_wowned(&(inp)->inp_lock)
456#define	INP_LOCK_ASSERT(inp)	rw_assert(&(inp)->inp_lock, RA_LOCKED)
457#define	INP_RLOCK_ASSERT(inp)	rw_assert(&(inp)->inp_lock, RA_RLOCKED)
458#define	INP_WLOCK_ASSERT(inp)	rw_assert(&(inp)->inp_lock, RA_WLOCKED)
459#define	INP_UNLOCK_ASSERT(inp)	rw_assert(&(inp)->inp_lock, RA_UNLOCKED)
460
461/*
462 * These locking functions are for inpcb consumers outside of sys/netinet,
463 * more specifically, they were added for the benefit of TOE drivers. The
464 * macros are reserved for use by the stack.
465 */
466void inp_wlock(struct inpcb *);
467void inp_wunlock(struct inpcb *);
468void inp_rlock(struct inpcb *);
469void inp_runlock(struct inpcb *);
470
471#ifdef INVARIANT_SUPPORT
472void inp_lock_assert(struct inpcb *);
473void inp_unlock_assert(struct inpcb *);
474#else
475#define	inp_lock_assert(inp)	do {} while (0)
476#define	inp_unlock_assert(inp)	do {} while (0)
477#endif
478
479void	inp_apply_all(struct inpcbinfo *, void (*func)(struct inpcb *, void *),
480	    void *arg);
481struct socket *
482	inp_inpcbtosocket(struct inpcb *inp);
483void 	inp_4tuple_get(struct inpcb *inp, uint32_t *laddr, uint16_t *lp,
484		uint32_t *faddr, uint16_t *fp);
485
486#endif /* _KERNEL */
487
488#define INP_INFO_WLOCK(ipi)	mtx_lock(&(ipi)->ipi_lock)
489#define INP_INFO_WLOCKED(ipi)	mtx_owned(&(ipi)->ipi_lock)
490#define INP_INFO_WUNLOCK(ipi)	mtx_unlock(&(ipi)->ipi_lock)
491#define	INP_INFO_LOCK_ASSERT(ipi)	MPASS(SMR_ENTERED((ipi)->ipi_smr) || \
492					mtx_owned(&(ipi)->ipi_lock))
493#define INP_INFO_WLOCK_ASSERT(ipi)	mtx_assert(&(ipi)->ipi_lock, MA_OWNED)
494#define INP_INFO_WUNLOCK_ASSERT(ipi)	\
495				mtx_assert(&(ipi)->ipi_lock, MA_NOTOWNED)
496
497#define	INP_HASH_WLOCK(ipi)		mtx_lock(&(ipi)->ipi_hash_lock)
498#define	INP_HASH_WUNLOCK(ipi)		mtx_unlock(&(ipi)->ipi_hash_lock)
499#define	INP_HASH_LOCK_ASSERT(ipi)	MPASS(SMR_ENTERED((ipi)->ipi_smr) || \
500					mtx_owned(&(ipi)->ipi_hash_lock))
501#define	INP_HASH_WLOCK_ASSERT(ipi)	mtx_assert(&(ipi)->ipi_hash_lock, \
502					MA_OWNED)
503
504/*
505 * Wildcard matching hash is not just a microoptimisation!  The hash for
506 * wildcard IPv4 and wildcard IPv6 must be the same, otherwise AF_INET6
507 * wildcard bound pcb won't be able to receive AF_INET connections, while:
508 * jenkins_hash(&zeroes, 1, s) != jenkins_hash(&zeroes, 4, s)
509 * See also comment above struct in_addr_4in6.
510 */
511#define	IN_ADDR_JHASH32(addr)						\
512	((addr)->s_addr == INADDR_ANY ? V_in_pcbhashseed :		\
513	    jenkins_hash32((&(addr)->s_addr), 1, V_in_pcbhashseed))
514#define	IN6_ADDR_JHASH32(addr)						\
515	(memcmp((addr), &in6addr_any, sizeof(in6addr_any)) == 0 ?	\
516	    V_in_pcbhashseed :						\
517	    jenkins_hash32((addr)->__u6_addr.__u6_addr32,		\
518	    nitems((addr)->__u6_addr.__u6_addr32), V_in_pcbhashseed))
519
520#define INP_PCBHASH(faddr, lport, fport, mask)				\
521	((IN_ADDR_JHASH32(faddr) ^ ntohs((lport) ^ (fport))) & (mask))
522#define	INP6_PCBHASH(faddr, lport, fport, mask)				\
523	((IN6_ADDR_JHASH32(faddr) ^ ntohs((lport) ^ (fport))) & (mask))
524
525#define	INP_PCBHASH_WILD(lport, mask)					\
526	((V_in_pcbhashseed ^ ntohs(lport)) & (mask))
527
528#define	INP_PCBLBGROUP_PKTHASH(faddr, lport, fport)			\
529	(IN_ADDR_JHASH32(faddr) ^ ntohs((lport) ^ (fport)))
530#define	INP6_PCBLBGROUP_PKTHASH(faddr, lport, fport)			\
531	(IN6_ADDR_JHASH32(faddr) ^ ntohs((lport) ^ (fport)))
532
533#define INP_PCBPORTHASH(lport, mask)	(ntohs((lport)) & (mask))
534
535/*
536 * Flags for inp_vflags -- historically version flags only
537 */
538#define	INP_IPV4	0x1
539#define	INP_IPV6	0x2
540#define	INP_IPV6PROTO	0x4		/* opened under IPv6 protocol */
541
542/*
543 * Flags for inp_flags.
544 */
545#define	INP_RECVOPTS		0x00000001 /* receive incoming IP options */
546#define	INP_RECVRETOPTS		0x00000002 /* receive IP options for reply */
547#define	INP_RECVDSTADDR		0x00000004 /* receive IP dst address */
548#define	INP_HDRINCL		0x00000008 /* user supplies entire IP header */
549#define	INP_HIGHPORT		0x00000010 /* user wants "high" port binding */
550#define	INP_LOWPORT		0x00000020 /* user wants "low" port binding */
551#define	INP_ANONPORT		0x00000040 /* read by netstat(1) */
552#define	INP_RECVIF		0x00000080 /* receive incoming interface */
553#define	INP_MTUDISC		0x00000100 /* user can do MTU discovery */
554/*	INP_FREED		0x00000200 private to in_pcb.c */
555#define	INP_RECVTTL		0x00000400 /* receive incoming IP TTL */
556#define	INP_DONTFRAG		0x00000800 /* don't fragment packet */
557#define	INP_BINDANY		0x00001000 /* allow bind to any address */
558#define	INP_INHASHLIST		0x00002000 /* in_pcbinshash() has been called */
559#define	INP_RECVTOS		0x00004000 /* receive incoming IP TOS */
560#define	IN6P_IPV6_V6ONLY	0x00008000 /* restrict AF_INET6 socket for v6 */
561#define	IN6P_PKTINFO		0x00010000 /* receive IP6 dst and I/F */
562#define	IN6P_HOPLIMIT		0x00020000 /* receive hoplimit */
563#define	IN6P_HOPOPTS		0x00040000 /* receive hop-by-hop options */
564#define	IN6P_DSTOPTS		0x00080000 /* receive dst options after rthdr */
565#define	IN6P_RTHDR		0x00100000 /* receive routing header */
566#define	IN6P_RTHDRDSTOPTS	0x00200000 /* receive dstoptions before rthdr */
567#define	IN6P_TCLASS		0x00400000 /* receive traffic class value */
568#define	IN6P_AUTOFLOWLABEL	0x00800000 /* attach flowlabel automatically */
569/*	INP_INLBGROUP		0x01000000 private to in_pcb.c */
570#define	INP_ONESBCAST		0x02000000 /* send all-ones broadcast */
571#define	INP_DROPPED		0x04000000 /* protocol drop flag */
572#define	INP_SOCKREF		0x08000000 /* strong socket reference */
573#define	INP_RESERVED_0          0x10000000 /* reserved field */
574#define	INP_RESERVED_1          0x20000000 /* reserved field */
575#define	IN6P_RFC2292		0x40000000 /* used RFC2292 API on the socket */
576#define	IN6P_MTU		0x80000000 /* receive path MTU */
577
578#define	INP_CONTROLOPTS		(INP_RECVOPTS|INP_RECVRETOPTS|INP_RECVDSTADDR|\
579				 INP_RECVIF|INP_RECVTTL|INP_RECVTOS|\
580				 IN6P_PKTINFO|IN6P_HOPLIMIT|IN6P_HOPOPTS|\
581				 IN6P_DSTOPTS|IN6P_RTHDR|IN6P_RTHDRDSTOPTS|\
582				 IN6P_TCLASS|IN6P_AUTOFLOWLABEL|IN6P_RFC2292|\
583				 IN6P_MTU)
584
585/*
586 * Flags for inp_flags2.
587 */
588/*				0x00000001 */
589/*				0x00000002 */
590/*				0x00000004 */
591/*				0x00000008 */
592/*				0x00000010 */
593/*				0x00000020 */
594/*				0x00000040 */
595/*				0x00000080 */
596#define	INP_RECVFLOWID		0x00000100 /* populate recv datagram with flow info */
597#define	INP_RECVRSSBUCKETID	0x00000200 /* populate recv datagram with bucket id */
598#define	INP_RATE_LIMIT_CHANGED	0x00000400 /* rate limit needs attention */
599#define	INP_ORIGDSTADDR		0x00000800 /* receive IP dst address/port */
600/*				0x00001000 */
601/*				0x00002000 */
602/*				0x00004000 */
603/*				0x00008000 */
604/*				0x00010000 */
605#define INP_2PCP_SET		0x00020000 /* If the Eth PCP should be set explicitly */
606#define INP_2PCP_BIT0		0x00040000 /* Eth PCP Bit 0 */
607#define INP_2PCP_BIT1		0x00080000 /* Eth PCP Bit 1 */
608#define INP_2PCP_BIT2		0x00100000 /* Eth PCP Bit 2 */
609#define INP_2PCP_BASE	INP_2PCP_BIT0
610#define INP_2PCP_MASK	(INP_2PCP_BIT0 | INP_2PCP_BIT1 | INP_2PCP_BIT2)
611#define INP_2PCP_SHIFT		18         /* shift PCP field in/out of inp_flags2 */
612
613/*
614 * Flags passed to in_pcblookup*(), inp_smr_lock() and inp_next().
615 */
616typedef	enum {
617	INPLOOKUP_WILDCARD = 0x00000001,	/* Allow wildcard sockets. */
618	INPLOOKUP_RLOCKPCB = 0x00000002,	/* Return inpcb read-locked. */
619	INPLOOKUP_WLOCKPCB = 0x00000004,	/* Return inpcb write-locked. */
620} inp_lookup_t;
621
622#define	INPLOOKUP_MASK	(INPLOOKUP_WILDCARD | INPLOOKUP_RLOCKPCB | \
623	    INPLOOKUP_WLOCKPCB)
624#define	INPLOOKUP_LOCKMASK	(INPLOOKUP_RLOCKPCB | INPLOOKUP_WLOCKPCB)
625
626#define	sotoinpcb(so)	((struct inpcb *)(so)->so_pcb)
627
628#define	INP_SOCKAF(so) so->so_proto->pr_domain->dom_family
629
630#define	INP_CHECK_SOCKAF(so, af)	(INP_SOCKAF(so) == af)
631
632#ifdef _KERNEL
633VNET_DECLARE(int, ipport_reservedhigh);
634VNET_DECLARE(int, ipport_reservedlow);
635VNET_DECLARE(int, ipport_lowfirstauto);
636VNET_DECLARE(int, ipport_lowlastauto);
637VNET_DECLARE(int, ipport_firstauto);
638VNET_DECLARE(int, ipport_lastauto);
639VNET_DECLARE(int, ipport_hifirstauto);
640VNET_DECLARE(int, ipport_hilastauto);
641VNET_DECLARE(int, ipport_randomized);
642
643#define	V_ipport_reservedhigh	VNET(ipport_reservedhigh)
644#define	V_ipport_reservedlow	VNET(ipport_reservedlow)
645#define	V_ipport_lowfirstauto	VNET(ipport_lowfirstauto)
646#define	V_ipport_lowlastauto	VNET(ipport_lowlastauto)
647#define	V_ipport_firstauto	VNET(ipport_firstauto)
648#define	V_ipport_lastauto	VNET(ipport_lastauto)
649#define	V_ipport_hifirstauto	VNET(ipport_hifirstauto)
650#define	V_ipport_hilastauto	VNET(ipport_hilastauto)
651#define	V_ipport_randomized	VNET(ipport_randomized)
652
653void	in_pcbinfo_init(struct inpcbinfo *, struct inpcbstorage *,
654	    u_int, u_int);
655void	in_pcbinfo_destroy(struct inpcbinfo *);
656void	in_pcbstorage_init(void *);
657void	in_pcbstorage_destroy(void *);
658
659void	in_pcbpurgeif0(struct inpcbinfo *, struct ifnet *);
660int	in_pcballoc(struct socket *, struct inpcbinfo *);
661int	in_pcbbind(struct inpcb *, struct sockaddr_in *, struct ucred *);
662int	in_pcbbind_setup(struct inpcb *, struct sockaddr_in *, in_addr_t *,
663	    u_short *, struct ucred *);
664int	in_pcbconnect(struct inpcb *, struct sockaddr_in *, struct ucred *,
665	    bool);
666int	in_pcbconnect_setup(struct inpcb *, struct sockaddr_in *, in_addr_t *,
667	    u_short *, in_addr_t *, u_short *, struct ucred *);
668void	in_pcbdisconnect(struct inpcb *);
669void	in_pcbdrop(struct inpcb *);
670void	in_pcbfree(struct inpcb *);
671int	in_pcbinshash(struct inpcb *);
672int	in_pcbladdr(struct inpcb *, struct in_addr *, struct in_addr *,
673	    struct ucred *);
674int	in_pcblbgroup_numa(struct inpcb *, int arg);
675struct inpcb *
676	in_pcblookup(struct inpcbinfo *, struct in_addr, u_int,
677	    struct in_addr, u_int, int, struct ifnet *);
678struct inpcb *
679	in_pcblookup_mbuf(struct inpcbinfo *, struct in_addr, u_int,
680	    struct in_addr, u_int, int, struct ifnet *, struct mbuf *);
681void	in_pcbref(struct inpcb *);
682void	in_pcbrehash(struct inpcb *);
683void	in_pcbremhash_locked(struct inpcb *);
684bool	in_pcbrele(struct inpcb *, inp_lookup_t);
685bool	in_pcbrele_rlocked(struct inpcb *);
686bool	in_pcbrele_wlocked(struct inpcb *);
687
688typedef bool inp_match_t(const struct inpcb *, void *);
689struct inpcb_iterator {
690	const struct inpcbinfo	*ipi;
691	struct inpcb		*inp;
692	inp_match_t		*match;
693	void			*ctx;
694	int			hash;
695#define	INP_ALL_LIST		-1
696	const inp_lookup_t	lock;
697};
698
699/* Note: sparse initializers guarantee .inp = NULL. */
700#define	INP_ITERATOR(_ipi, _lock, _match, _ctx)		\
701	{						\
702		.ipi = (_ipi),				\
703		.lock = (_lock),			\
704		.hash = INP_ALL_LIST,			\
705		.match = (_match),			\
706		.ctx = (_ctx),				\
707	}
708#define	INP_ALL_ITERATOR(_ipi, _lock)			\
709	{						\
710		.ipi = (_ipi),				\
711		.lock = (_lock),			\
712		.hash = INP_ALL_LIST,			\
713	}
714
715struct inpcb *inp_next(struct inpcb_iterator *);
716void	in_losing(struct inpcb *);
717void	in_pcbsetsolabel(struct socket *so);
718int	in_getpeeraddr(struct socket *, struct sockaddr *sa);
719int	in_getsockaddr(struct socket *, struct sockaddr *sa);
720void	in_pcbsosetlabel(struct socket *so);
721#ifdef RATELIMIT
722int
723in_pcboutput_txrtlmt_locked(struct inpcb *, struct ifnet *,
724	    struct mbuf *, uint32_t);
725int	in_pcbattach_txrtlmt(struct inpcb *, struct ifnet *, uint32_t, uint32_t,
726	    uint32_t, struct m_snd_tag **);
727void	in_pcbdetach_txrtlmt(struct inpcb *);
728void    in_pcbdetach_tag(struct m_snd_tag *);
729int	in_pcbmodify_txrtlmt(struct inpcb *, uint32_t);
730int	in_pcbquery_txrtlmt(struct inpcb *, uint32_t *);
731int	in_pcbquery_txrlevel(struct inpcb *, uint32_t *);
732void	in_pcboutput_txrtlmt(struct inpcb *, struct ifnet *, struct mbuf *);
733void	in_pcboutput_eagain(struct inpcb *);
734#endif
735#endif /* _KERNEL */
736
737#endif /* !_NETINET_IN_PCB_H_ */
738