usr/src/sys/netinet/in_pcb.c

/*	in_pcb.c	4.22	82/03/28	*/

#include "../h/param.h"
#include "../h/systm.h"
#include "../h/dir.h"
#include "../h/user.h"
#include "../h/mbuf.h"
#include "../h/socket.h"
#include "../h/socketvar.h"
#include "../net/in.h"
#include "../net/in_systm.h"
#include "../net/if.h"
#include "../net/in_pcb.h"
#include "../h/protosw.h"

/*
 * Routines to manage internet protocol control blocks.
 *
 * At PRU_ATTACH time a protocol control block is allocated in
 * in_pcballoc() and inserted on a doubly-linked list of such blocks
 * for the protocol.  A port address is either requested (and verified
 * to not be in use) or assigned at this time.  We also allocate
 * space in the socket sockbuf structures here, although this is
 * not a clearly correct place to put this function.
 *
 * A connectionless protocol will have its protocol control block
 * removed at PRU_DETACH time, when the socket will be freed (freeing
 * the space reserved) and the block will be removed from the list of
 * blocks for its protocol.
 *
 * A connection-based protocol may be connected to a remote peer at
 * PRU_CONNECT time through the routine in_pcbconnect().  In the normal
 * case a PRU_DISCONNECT occurs causing a in_pcbdisconnect().
 * It is also possible that higher-level routines will opt out of the
 * relationship with the connection before the connection shut down
 * is complete.  This often occurs in protocols like TCP where we must
 * hold on to the protocol control block for a unreasonably long time
 * after the connection is used up to avoid races in later connection
 * establishment.  To handle this we allow higher-level routines to
 * disassociate themselves from the socket, marking it SS_USERGONE while
 * the disconnect is in progress.  We notice that this has happened
 * when the disconnect is complete, and perform the PRU_DETACH operation,
 * freeing the socket.
 *
 * TODO:
 *	use hashing
 */
struct	in_addr zeroin_addr;

/*
 * Allocate a protocol control block, space
 * for send and receive data, and local host information.
 * Return error.  If no error make socket point at pcb.
 */
in_pcbattach(so, head, sndcc, rcvcc, sin)
	struct socket *so;
	struct inpcb *head;
	int sndcc, rcvcc;
	struct sockaddr_in *sin;
{
	struct mbuf *m;
	register struct inpcb *inp;
	u_short lport = 0;

COUNT(IN_PCBATTACH);
	if (ifnet == 0)
		return (EADDRNOTAVAIL);
	if (sin) {
		if (sin->sin_family != AF_INET)
			return (EAFNOSUPPORT);
		if (sin->sin_addr.s_addr &&
		    if_ifwithaddr((struct sockaddr *)sin) == 0)
			return (EADDRNOTAVAIL);
		lport = sin->sin_port;
		if (lport) {
			u_short aport = lport;
			int wild = 0;
#if vax
			aport = htons(aport);
#endif
			/* GROSS */
			if (aport < IPPORT_RESERVED && u.u_uid != 0)
				return (EPERM);
			if ((so->so_proto->pr_flags & PR_CONNREQUIRED) == 0 ||
			    (so->so_options & SO_ACCEPTCONN) == 0)
				wild = INPLOOKUP_WILDCARD;
			if (in_pcblookup(head,
			    zeroin_addr, 0, sin->sin_addr, lport, wild))
				return (EADDRINUSE);
		}
	}
	m = m_getclr(M_DONTWAIT);
	if (m == 0)
		return (ENOBUFS);
	if (sbreserve(&so->so_snd, sndcc) == 0)
		goto bad;
	if (sbreserve(&so->so_rcv, rcvcc) == 0)
		goto bad2;
	inp = mtod(m, struct inpcb *);
	inp->inp_head = head;
	if (sin)
		inp->inp_laddr = sin->sin_addr;
	if (lport == 0)
		do {
			if (head->inp_lport++ < IPPORT_RESERVED)
				head->inp_lport = IPPORT_RESERVED;
			lport = htons(head->inp_lport);
		} while (in_pcblookup(head,
			    zeroin_addr, 0, inp->inp_laddr, lport, 0));
	inp->inp_lport = lport;
	inp->inp_socket = so;
	insque(inp, head);
	so->so_pcb = (caddr_t)inp;
	in_setsockaddr(inp);
	return (0);
bad2:
	sbrelease(&so->so_snd);
bad:
	(void) m_free(m);
	return (ENOBUFS);
}

/*
 * Connect from a socket to a specified address.
 * Both address and port must be specified in argument sin.
 * If don't have a local address for this socket yet,
 * then pick one.
 */
in_pcbconnect(inp, sin)
	struct inpcb *inp;
	struct sockaddr_in *sin;
{
	struct ifnet *ifp;
	struct sockaddr_in *ifaddr;

COUNT(IN_PCBCONNECT);
	if (sin->sin_family != AF_INET)
		return (EAFNOSUPPORT);
	if (sin->sin_addr.s_addr == 0 || sin->sin_port == 0)
		return (EADDRNOTAVAIL);
	if (inp->inp_laddr.s_addr == 0) {
		ifp = if_ifonnetof(sin->sin_addr.s_net);
		if (ifp == 0) {
			ifp = if_ifwithaf(AF_INET);
			if (ifp == 0)
				return (EADDRNOTAVAIL);		/* XXX */
		}
		ifaddr = (struct sockaddr_in *)&ifp->if_addr;
	}
	if (in_pcblookup(inp->inp_head,
	    sin->sin_addr,
	    sin->sin_port,
	    inp->inp_laddr.s_addr ? inp->inp_laddr : ifaddr->sin_addr,
	    inp->inp_lport,
	    0))
		return (EADDRINUSE);
	if (inp->inp_laddr.s_addr == 0) {
		struct sockaddr_in *sin2 =
		    (struct sockaddr_in *)&inp->inp_socket->so_addr;

		inp->inp_laddr = ifaddr->sin_addr;
		sin2->sin_addr = inp->inp_laddr;
	}
	inp->inp_faddr = sin->sin_addr;
	inp->inp_fport = sin->sin_port;
	return (0);
}

in_setsockaddr(inp)
	struct inpcb *inp;
{
	register struct sockaddr_in *sin =
	    (struct sockaddr_in *)&inp->inp_socket->so_addr;

	sin->sin_family = AF_INET;
	sin->sin_addr = inp->inp_laddr;
	sin->sin_port = inp->inp_lport;
}

in_pcbdisconnect(inp)
	struct inpcb *inp;
{

COUNT(IN_PCBDISCONNECT);
	inp->inp_faddr.s_addr = 0;
	inp->inp_fport = 0;
	if (inp->inp_socket->so_state & SS_USERGONE)
		in_pcbdetach(inp);
}

in_pcbdetach(inp)
	struct inpcb *inp;
{
	struct socket *so = inp->inp_socket;

	so->so_pcb = 0;
	sofree(so);
	remque(inp);
	(void) m_free(dtom(inp));
}

/*
 * SHOULD ALLOW MATCH ON MULTI-HOMING ONLY
 */
struct inpcb *
in_pcblookup(head, faddr, fport, laddr, lport, flags)
	struct inpcb *head;
	struct in_addr faddr, laddr;
	u_short fport, lport;
	int flags;
{
	register struct inpcb *inp, *match = 0;
	int matchwild = 3, wildcard;

	for (inp = head->inp_next; inp != head; inp = inp->inp_next) {
		if (inp->inp_lport != lport)
			continue;
		wildcard = 0;
		if (inp->inp_laddr.s_addr != 0) {
			if (laddr.s_addr == 0)
				wildcard++;
			else if (inp->inp_laddr.s_addr != laddr.s_addr)
				continue;
		} else {
			if (laddr.s_addr != 0)
				wildcard++;
		}
		if (inp->inp_faddr.s_addr != 0) {
			if (faddr.s_addr == 0)
				wildcard++;
			else if (inp->inp_faddr.s_addr != faddr.s_addr ||
			    inp->inp_fport != fport)
				continue;
		} else {
			if (faddr.s_addr != 0)
				wildcard++;
		}
		if (wildcard && (flags & INPLOOKUP_WILDCARD) == 0)
			continue;
		if (wildcard < matchwild) {
			match = inp;
			matchwild = wildcard;
			if (matchwild == 0)
				break;
		}
	}
	return (match);
}
Commit	Line	Data
	1	/* in_pcb.c 4.22 82/03/28 */
	2
	3	#include "../h/param.h"
	4	#include "../h/systm.h"
	5	#include "../h/dir.h"
	6	#include "../h/user.h"
	7	#include "../h/mbuf.h"
	8	#include "../h/socket.h"
	9	#include "../h/socketvar.h"
	10	#include "../net/in.h"
	11	#include "../net/in_systm.h"
	12	#include "../net/if.h"
	13	#include "../net/in_pcb.h"
	14	#include "../h/protosw.h"
	15
	16	/*
	17	* Routines to manage internet protocol control blocks.
	18	*
	19	* At PRU_ATTACH time a protocol control block is allocated in
	20	* in_pcballoc() and inserted on a doubly-linked list of such blocks
	21	* for the protocol. A port address is either requested (and verified
	22	* to not be in use) or assigned at this time. We also allocate
	23	* space in the socket sockbuf structures here, although this is
	24	* not a clearly correct place to put this function.
	25	*
	26	* A connectionless protocol will have its protocol control block
	27	* removed at PRU_DETACH time, when the socket will be freed (freeing
	28	* the space reserved) and the block will be removed from the list of
	29	* blocks for its protocol.
	30	*
	31	* A connection-based protocol may be connected to a remote peer at
	32	* PRU_CONNECT time through the routine in_pcbconnect(). In the normal
	33	* case a PRU_DISCONNECT occurs causing a in_pcbdisconnect().
	34	* It is also possible that higher-level routines will opt out of the
	35	* relationship with the connection before the connection shut down
	36	* is complete. This often occurs in protocols like TCP where we must
	37	* hold on to the protocol control block for a unreasonably long time
	38	* after the connection is used up to avoid races in later connection
	39	* establishment. To handle this we allow higher-level routines to
	40	* disassociate themselves from the socket, marking it SS_USERGONE while
	41	* the disconnect is in progress. We notice that this has happened
	42	* when the disconnect is complete, and perform the PRU_DETACH operation,
	43	* freeing the socket.
	44	*
	45	* TODO:
	46	* use hashing
	47	*/
	48	struct in_addr zeroin_addr;
	49
	50	/*
	51	* Allocate a protocol control block, space
	52	* for send and receive data, and local host information.
	53	* Return error. If no error make socket point at pcb.
	54	*/
	55	in_pcbattach(so, head, sndcc, rcvcc, sin)
	56	struct socket *so;
	57	struct inpcb *head;
	58	int sndcc, rcvcc;
	59	struct sockaddr_in *sin;
	60	{
	61	struct mbuf *m;
	62	register struct inpcb *inp;
	63	u_short lport = 0;
	64
	65	COUNT(IN_PCBATTACH);
	66	if (ifnet == 0)
	67	return (EADDRNOTAVAIL);
	68	if (sin) {
	69	if (sin->sin_family != AF_INET)
	70	return (EAFNOSUPPORT);
	71	if (sin->sin_addr.s_addr &&
	72	if_ifwithaddr((struct sockaddr *)sin) == 0)
	73	return (EADDRNOTAVAIL);
	74	lport = sin->sin_port;
	75	if (lport) {
	76	u_short aport = lport;
	77	int wild = 0;
	78	#if vax
	79	aport = htons(aport);
	80	#endif
	81	/* GROSS */
	82	if (aport < IPPORT_RESERVED && u.u_uid != 0)
	83	return (EPERM);
	84	if ((so->so_proto->pr_flags & PR_CONNREQUIRED) == 0 \|\|
	85	(so->so_options & SO_ACCEPTCONN) == 0)
	86	wild = INPLOOKUP_WILDCARD;
	87	if (in_pcblookup(head,
	88	zeroin_addr, 0, sin->sin_addr, lport, wild))
	89	return (EADDRINUSE);
	90	}
	91	}
	92	m = m_getclr(M_DONTWAIT);
	93	if (m == 0)
	94	return (ENOBUFS);
	95	if (sbreserve(&so->so_snd, sndcc) == 0)
	96	goto bad;
	97	if (sbreserve(&so->so_rcv, rcvcc) == 0)
	98	goto bad2;
	99	inp = mtod(m, struct inpcb *);
	100	inp->inp_head = head;
	101	if (sin)
	102	inp->inp_laddr = sin->sin_addr;
	103	if (lport == 0)
	104	do {
	105	if (head->inp_lport++ < IPPORT_RESERVED)
	106	head->inp_lport = IPPORT_RESERVED;
	107	lport = htons(head->inp_lport);
	108	} while (in_pcblookup(head,
	109	zeroin_addr, 0, inp->inp_laddr, lport, 0));
	110	inp->inp_lport = lport;
	111	inp->inp_socket = so;
	112	insque(inp, head);
	113	so->so_pcb = (caddr_t)inp;
	114	in_setsockaddr(inp);
	115	return (0);
	116	bad2:
	117	sbrelease(&so->so_snd);
	118	bad:
	119	(void) m_free(m);
	120	return (ENOBUFS);
	121	}
	122
	123	/*
	124	* Connect from a socket to a specified address.
	125	* Both address and port must be specified in argument sin.
	126	* If don't have a local address for this socket yet,
	127	* then pick one.
	128	*/
	129	in_pcbconnect(inp, sin)
	130	struct inpcb *inp;
	131	struct sockaddr_in *sin;
	132	{
	133	struct ifnet *ifp;
	134	struct sockaddr_in *ifaddr;
	135
	136	COUNT(IN_PCBCONNECT);
	137	if (sin->sin_family != AF_INET)
	138	return (EAFNOSUPPORT);
	139	if (sin->sin_addr.s_addr == 0 \|\| sin->sin_port == 0)
	140	return (EADDRNOTAVAIL);
	141	if (inp->inp_laddr.s_addr == 0) {
	142	ifp = if_ifonnetof(sin->sin_addr.s_net);
	143	if (ifp == 0) {
	144	ifp = if_ifwithaf(AF_INET);
	145	if (ifp == 0)
	146	return (EADDRNOTAVAIL); /* XXX */
	147	}
	148	ifaddr = (struct sockaddr_in *)&ifp->if_addr;
	149	}
	150	if (in_pcblookup(inp->inp_head,
	151	sin->sin_addr,
	152	sin->sin_port,
	153	inp->inp_laddr.s_addr ? inp->inp_laddr : ifaddr->sin_addr,
	154	inp->inp_lport,
	155	0))
	156	return (EADDRINUSE);
	157	if (inp->inp_laddr.s_addr == 0) {
	158	struct sockaddr_in *sin2 =
	159	(struct sockaddr_in *)&inp->inp_socket->so_addr;
	160
	161	inp->inp_laddr = ifaddr->sin_addr;
	162	sin2->sin_addr = inp->inp_laddr;
	163	}
	164	inp->inp_faddr = sin->sin_addr;
	165	inp->inp_fport = sin->sin_port;
	166	return (0);
	167	}
	168
	169	in_setsockaddr(inp)
	170	struct inpcb *inp;
	171	{
	172	register struct sockaddr_in *sin =
	173	(struct sockaddr_in *)&inp->inp_socket->so_addr;
	174
	175	sin->sin_family = AF_INET;
	176	sin->sin_addr = inp->inp_laddr;
	177	sin->sin_port = inp->inp_lport;
	178	}
	179
	180	in_pcbdisconnect(inp)
	181	struct inpcb *inp;
	182	{
	183
	184	COUNT(IN_PCBDISCONNECT);
	185	inp->inp_faddr.s_addr = 0;
	186	inp->inp_fport = 0;
	187	if (inp->inp_socket->so_state & SS_USERGONE)
	188	in_pcbdetach(inp);
	189	}
	190
	191	in_pcbdetach(inp)
	192	struct inpcb *inp;
	193	{
	194	struct socket *so = inp->inp_socket;
	195
	196	so->so_pcb = 0;
	197	sofree(so);
	198	remque(inp);
	199	(void) m_free(dtom(inp));
	200	}
	201
	202	/*
	203	* SHOULD ALLOW MATCH ON MULTI-HOMING ONLY
	204	*/
	205	struct inpcb *
	206	in_pcblookup(head, faddr, fport, laddr, lport, flags)
	207	struct inpcb *head;
	208	struct in_addr faddr, laddr;
	209	u_short fport, lport;
	210	int flags;
	211	{
	212	register struct inpcb inp, match = 0;
	213	int matchwild = 3, wildcard;
	214
	215	for (inp = head->inp_next; inp != head; inp = inp->inp_next) {
	216	if (inp->inp_lport != lport)
	217	continue;
	218	wildcard = 0;
	219	if (inp->inp_laddr.s_addr != 0) {
	220	if (laddr.s_addr == 0)
	221	wildcard++;
	222	else if (inp->inp_laddr.s_addr != laddr.s_addr)
	223	continue;
	224	} else {
	225	if (laddr.s_addr != 0)
	226	wildcard++;
	227	}
	228	if (inp->inp_faddr.s_addr != 0) {
	229	if (faddr.s_addr == 0)
	230	wildcard++;
	231	else if (inp->inp_faddr.s_addr != faddr.s_addr \|\|
	232	inp->inp_fport != fport)
	233	continue;
	234	} else {
	235	if (faddr.s_addr != 0)
	236	wildcard++;
	237	}
	238	if (wildcard && (flags & INPLOOKUP_WILDCARD) == 0)
	239	continue;
	240	if (wildcard < matchwild) {
	241	match = inp;
	242	matchwild = wildcard;
	243	if (matchwild == 0)
	244	break;
	245	}
	246	}
	247	return (match);
	248	}