[unix-history] / usr / src / sys / netinet / udp_usrreq.c

/*	udp_usrreq.c	4.45	83/02/16	*/

#include "../h/param.h"
#include "../h/dir.h"
#include "../h/user.h"
#include "../h/mbuf.h"
#include "../h/protosw.h"
#include "../h/socket.h"
#include "../h/socketvar.h"
#include "../h/errno.h"

#include "../net/if.h"
#include "../net/route.h"

#include "../netinet/in.h"
#include "../netinet/in_pcb.h"
#include "../netinet/in_systm.h"
#include "../netinet/ip.h"
#include "../netinet/ip_var.h"
#include "../netinet/ip_icmp.h"
#include "../netinet/udp.h"
#include "../netinet/udp_var.h"

/*
 * UDP protocol implementation.
 * Per RFC 768, August, 1980.
 */
udp_init()
{

	udb.inp_next = udb.inp_prev = &udb;
}

int	udpcksum;
struct	sockaddr_in udp_in = { AF_INET };

udp_input(m0)
	struct mbuf *m0;
{
	register struct udpiphdr *ui;
	register struct inpcb *inp;
	register struct mbuf *m;
	int len;

	/*
	 * Get IP and UDP header together in first mbuf.
	 */
	m = m0;
	if ((m->m_off > MMAXOFF || m->m_len < sizeof (struct udpiphdr)) &&
	    (m = m_pullup(m, sizeof (struct udpiphdr))) == 0) {
		udpstat.udps_hdrops++;
		return;
	}
	ui = mtod(m, struct udpiphdr *);
	if (((struct ip *)ui)->ip_hl > (sizeof (struct ip) >> 2))
		ip_stripoptions((struct ip *)ui, (struct mbuf *)0);

	/*
	 * Make mbuf data length reflect UDP length.
	 * If not enough data to reflect UDP length, drop.
	 */
	len = ntohs((u_short)ui->ui_ulen);
	if (((struct ip *)ui)->ip_len != len) {
		if (len > ((struct ip *)ui)->ip_len) {
			udpstat.udps_badlen++;
			goto bad;
		}
		m_adj(m, ((struct ip *)ui)->ip_len - len);
		/* (struct ip *)ui->ip_len = len; */
	}

	/*
	 * Checksum extended UDP header and data.
	 */
	if (udpcksum) {
		ui->ui_next = ui->ui_prev = 0;
		ui->ui_x1 = 0;
		ui->ui_len = htons((u_short)len);
		if (ui->ui_sum = in_cksum(m, len + sizeof (struct ip))) {
			udpstat.udps_badsum++;
			printf("udp cksum %x\n", ui->ui_sum);
			m_freem(m);
			return;
		}
	}

	/*
	 * Locate pcb for datagram.
	 */
	inp = in_pcblookup(&udb,
	    ui->ui_src, ui->ui_sport, ui->ui_dst, ui->ui_dport,
		INPLOOKUP_WILDCARD);
	if (inp == 0) {
		/* don't send ICMP response for broadcast packet */
		if (in_lnaof(ui->ui_dst) == INADDR_ANY)
			goto bad;
		icmp_error((struct ip *)ui, ICMP_UNREACH, ICMP_UNREACH_PORT);
		return;
	}

	/*
	 * Construct sockaddr format source address.
	 * Stuff source address and datagram in user buffer.
	 */
	udp_in.sin_port = ui->ui_sport;
	udp_in.sin_addr = ui->ui_src;
	m->m_len -= sizeof (struct udpiphdr);
	m->m_off += sizeof (struct udpiphdr);
	if (sbappendaddr(&inp->inp_socket->so_rcv, (struct sockaddr *)&udp_in, m) == 0)
		goto bad;
	sorwakeup(inp->inp_socket);
	return;
bad:
	m_freem(m);
}

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

	in_pcbdisconnect(inp);
	soisdisconnected(so);
}

udp_ctlinput(cmd, arg)
	int cmd;
	caddr_t arg;
{
	struct in_addr *sin;
	extern u_char inetctlerrmap[];

	if (cmd < 0 || cmd > PRC_NCMDS)
		return;
	switch (cmd) {

	case PRC_ROUTEDEAD:
		break;

	case PRC_QUENCH:
		break;

	/* these are handled by ip */
	case PRC_IFDOWN:
	case PRC_HOSTDEAD:
	case PRC_HOSTUNREACH:
		break;

	default:
		sin = &((struct icmp *)arg)->icmp_ip.ip_dst;
		in_pcbnotify(&udb, sin, (int)inetctlerrmap[cmd], udp_abort);
	}
}

udp_output(inp, m0)
	struct inpcb *inp;
	struct mbuf *m0;
{
	register struct mbuf *m;
	register struct udpiphdr *ui;
	register struct socket *so;
	register int len = 0;

	/*
	 * Calculate data length and get a mbuf
	 * for UDP and IP headers.
	 */
	for (m = m0; m; m = m->m_next)
		len += m->m_len;
	m = m_get(M_DONTWAIT, MT_HEADER);
	if (m == 0) {
		m_freem(m0);
		return (ENOBUFS);
	}

	/*
	 * Fill in mbuf with extended UDP header
	 * and addresses and length put into network format.
	 */
	m->m_off = MMAXOFF - sizeof (struct udpiphdr);
	m->m_len = sizeof (struct udpiphdr);
	m->m_next = m0;
	ui = mtod(m, struct udpiphdr *);
	ui->ui_next = ui->ui_prev = 0;
	ui->ui_x1 = 0;
	ui->ui_pr = IPPROTO_UDP;
	ui->ui_len = len + sizeof (struct udphdr);
	ui->ui_src = inp->inp_laddr;
	ui->ui_dst = inp->inp_faddr;
	ui->ui_sport = inp->inp_lport;
	ui->ui_dport = inp->inp_fport;
	ui->ui_ulen = htons((u_short)ui->ui_len);

	/*
	 * Stuff checksum and output datagram.
	 */
	ui->ui_sum = 0;
	ui->ui_sum = in_cksum(m, sizeof (struct udpiphdr) + len);
	((struct ip *)ui)->ip_len = sizeof (struct udpiphdr) + len;
	((struct ip *)ui)->ip_ttl = MAXTTL;
	so = inp->inp_socket;
	return (ip_output(m, (struct mbuf *)0,
	    (so->so_options & SO_DONTROUTE) ? &routetoif : (struct route *)0,
	    so->so_state & SS_PRIV));
}

/*ARGSUSED*/
udp_usrreq(so, req, m, nam)
	struct socket *so;
	int req;
	struct mbuf *m, *nam;
{
	struct inpcb *inp = sotoinpcb(so);
	int error = 0;

	if (inp == NULL && req != PRU_ATTACH) {
		error = EINVAL;
		goto release;
	}
	switch (req) {

	case PRU_ATTACH:
		if (inp != NULL) {
			error = EINVAL;
			break;
		}
		error = in_pcballoc(so, &udb);
		if (error)
			break;
		error = soreserve(so, 2048, 2048);
		if (error)
			break;
		break;

	case PRU_DETACH:
		if (inp == NULL) {
			error = ENOTCONN;
			break;
		}
		in_pcbdetach(inp);
		break;

	case PRU_BIND:
		error = in_pcbbind(inp, nam);
		break;

	case PRU_LISTEN:
		error = EOPNOTSUPP;
		break;

	case PRU_CONNECT:
		if (inp->inp_faddr.s_addr != INADDR_ANY) {
			error = EISCONN;
			break;
		}
		error = in_pcbconnect(inp, nam);
		if (error == 0)
			soisconnected(so);
		break;

	case PRU_ACCEPT:
		error = EOPNOTSUPP;
		break;

	case PRU_DISCONNECT:
		if (inp->inp_faddr.s_addr == INADDR_ANY) {
			error = ENOTCONN;
			break;
		}
		in_pcbdisconnect(inp);
		soisdisconnected(so);
		break;

	case PRU_SHUTDOWN:
		socantsendmore(so);
		break;

	case PRU_SEND: {
		struct in_addr laddr;

		if (nam) {
			laddr = inp->inp_laddr;
			if (inp->inp_faddr.s_addr != INADDR_ANY) {
				error = EISCONN;
				break;
			}
			error = in_pcbconnect(inp, nam);
			if (error)
				break;
		} else {
			if (inp->inp_faddr.s_addr == INADDR_ANY) {
				error = ENOTCONN;
				break;
			}
		}
		error = udp_output(inp, m);
		m = NULL;
		if (nam) {
			in_pcbdisconnect(inp);
			inp->inp_laddr = laddr;
		}
		}
		break;

	case PRU_ABORT:
		in_pcbdetach(inp);
		sofree(so);
		soisdisconnected(so);
		break;

	case PRU_CONTROL:
		error =  EOPNOTSUPP;
		break;

	case PRU_SOCKADDR:
		in_setsockaddr(inp, nam);
		break;

	default:
		panic("udp_usrreq");
	}
release:
	if (m != NULL)
		m_freem(m);
	return (error);
}
Commit	Line	Data
a3fa431a	1	/* udp_usrreq.c 4.45 83/02/16 */
1bd53abe BJ	2
1bd53abe BJ	3	#include "../h/param.h"
53a5409e BJ	4	#include "../h/dir.h"
53a5409e BJ	5	#include "../h/user.h"
1bd53abe	6	#include "../h/mbuf.h"
d52566dd	7	#include "../h/protosw.h"
53a5409e BJ	8	#include "../h/socket.h"
53a5409e BJ	9	#include "../h/socketvar.h"
f4d55810 SL	10	#include "../h/errno.h"
f4d55810 SL	11
72e4f44e	12	#include "../net/if.h"
c124e997	13	#include "../net/route.h"
f4d55810 SL	14
f4d55810 SL	15	#include "../netinet/in.h"
bf6ab6ff BJ	16	#include "../netinet/in_pcb.h"
	17	#include "../netinet/in_systm.h"
	18	#include "../netinet/ip.h"
	19	#include "../netinet/ip_var.h"
	20	#include "../netinet/ip_icmp.h"
	21	#include "../netinet/udp.h"
	22	#include "../netinet/udp_var.h"
d52566dd	23
2b4b57cd BJ	24	/*
	25	* UDP protocol implementation.
	26	* Per RFC 768, August, 1980.
	27	*/
d52566dd BJ	28	udp_init()
	29	{
	30
eb44bfb2	31	udb.inp_next = udb.inp_prev = &udb;
d52566dd	32	}
1bd53abe	33
eb44bfb2	34	int udpcksum;
2b4b57cd	35	struct sockaddr_in udp_in = { AF_INET };
eb44bfb2	36
2b4b57cd BJ	37	udp_input(m0)
2b4b57cd BJ	38	struct mbuf *m0;
1bd53abe	39	{
eb44bfb2	40	register struct udpiphdr *ui;
53a5409e	41	register struct inpcb *inp;
2b4b57cd	42	register struct mbuf *m;
e199a8a5	43	int len;
eb44bfb2	44
2b4b57cd	45	/*
4aed14e3	46	* Get IP and UDP header together in first mbuf.
2b4b57cd BJ	47	*/
2b4b57cd BJ	48	m = m0;
89925e89 BJ	49	if ((m->m_off > MMAXOFF \|\| m->m_len < sizeof (struct udpiphdr)) &&
89925e89 BJ	50	(m = m_pullup(m, sizeof (struct udpiphdr))) == 0) {
2b4b57cd	51	udpstat.udps_hdrops++;
89925e89	52	return;
2b4b57cd	53	}
4e60622a	54	ui = mtod(m, struct udpiphdr *);
4aed14e3	55	if (((struct ip *)ui)->ip_hl > (sizeof (struct ip) >> 2))
cca1853f	56	ip_stripoptions((struct ip )ui, (struct mbuf )0);
2b4b57cd BJ	57
2b4b57cd BJ	58	/*
4aed14e3 BJ	59	* Make mbuf data length reflect UDP length.
4aed14e3 BJ	60	* If not enough data to reflect UDP length, drop.
2b4b57cd	61	*/
e199a8a5	62	len = ntohs((u_short)ui->ui_ulen);
2b4b57cd BJ	63	if (((struct ip *)ui)->ip_len != len) {
	64	if (len > ((struct ip *)ui)->ip_len) {
	65	udpstat.udps_badlen++;
	66	goto bad;
	67	}
	68	m_adj(m, ((struct ip *)ui)->ip_len - len);
	69	/* (struct ip )ui->ip_len = len; /
	70	}
	71
	72	/*
4aed14e3	73	* Checksum extended UDP header and data.
2b4b57cd	74	*/
eb44bfb2	75	if (udpcksum) {
2b4b57cd BJ	76	ui->ui_next = ui->ui_prev = 0;
2b4b57cd BJ	77	ui->ui_x1 = 0;
e199a8a5 BJ	78	ui->ui_len = htons((u_short)len);
e199a8a5 BJ	79	if (ui->ui_sum = in_cksum(m, len + sizeof (struct ip))) {
2b4b57cd	80	udpstat.udps_badsum++;
eb44bfb2 BJ	81	printf("udp cksum %x\n", ui->ui_sum);
	82	m_freem(m);
	83	return;
	84	}
	85	}
2b4b57cd BJ	86
2b4b57cd BJ	87	/*
e199a8a5	88	* Locate pcb for datagram.
2b4b57cd	89	*/
2b4b57cd	90	inp = in_pcblookup(&udb,
ebcadd38 BJ	91	ui->ui_src, ui->ui_sport, ui->ui_dst, ui->ui_dport,
ebcadd38 BJ	92	INPLOOKUP_WILDCARD);
72e4f44e	93	if (inp == 0) {
5cdc4d65 SL	94	/* don't send ICMP response for broadcast packet */
5cdc4d65 SL	95	if (in_lnaof(ui->ui_dst) == INADDR_ANY)
72e4f44e SL	96	goto bad;
	97	icmp_error((struct ip *)ui, ICMP_UNREACH, ICMP_UNREACH_PORT);
	98	return;
	99	}
2b4b57cd BJ	100
	101	/*
	102	* Construct sockaddr format source address.
	103	* Stuff source address and datagram in user buffer.
	104	*/
	105	udp_in.sin_port = ui->ui_sport;
	106	udp_in.sin_addr = ui->ui_src;
4e60622a BJ	107	m->m_len -= sizeof (struct udpiphdr);
	108	m->m_off += sizeof (struct udpiphdr);
	109	if (sbappendaddr(&inp->inp_socket->so_rcv, (struct sockaddr *)&udp_in, m) == 0)
2b4b57cd	110	goto bad;
4e60622a	111	sorwakeup(inp->inp_socket);
53a5409e	112	return;
2b4b57cd	113	bad:
53a5409e	114	m_freem(m);
1bd53abe BJ	115	}
1bd53abe BJ	116
72e4f44e SL	117	udp_abort(inp)
	118	struct inpcb *inp;
	119	{
	120	struct socket *so = inp->inp_socket;
	121
	122	in_pcbdisconnect(inp);
	123	soisdisconnected(so);
1bd53abe BJ	124	}
1bd53abe BJ	125
39674d5f SL	126	udp_ctlinput(cmd, arg)
	127	int cmd;
	128	caddr_t arg;
	129	{
	130	struct in_addr *sin;
	131	extern u_char inetctlerrmap[];
39674d5f SL	132
	133	if (cmd < 0 \|\| cmd > PRC_NCMDS)
	134	return;
	135	switch (cmd) {
	136
	137	case PRC_ROUTEDEAD:
	138	break;
	139
	140	case PRC_QUENCH:
	141	break;
	142
	143	/* these are handled by ip */
	144	case PRC_IFDOWN:
	145	case PRC_HOSTDEAD:
	146	case PRC_HOSTUNREACH:
	147	break;
	148
	149	default:
	150	sin = &((struct icmp *)arg)->icmp_ip.ip_dst;
a1edc12b	151	in_pcbnotify(&udb, sin, (int)inetctlerrmap[cmd], udp_abort);
39674d5f SL	152	}
	153	}
	154
4ad99bae	155	udp_output(inp, m0)
2b4b57cd	156	struct inpcb *inp;
2b4b57cd	157	struct mbuf *m0;
1bd53abe	158	{
2b4b57cd BJ	159	register struct mbuf *m;
2b4b57cd BJ	160	register struct udpiphdr *ui;
a13c006d	161	register struct socket *so;
2b4b57cd BJ	162	register int len = 0;
	163
	164	/*
	165	* Calculate data length and get a mbuf
4aed14e3	166	* for UDP and IP headers.
2b4b57cd BJ	167	*/
	168	for (m = m0; m; m = m->m_next)
	169	len += m->m_len;
cce93e4b	170	m = m_get(M_DONTWAIT, MT_HEADER);
8a2f82db SL	171	if (m == 0) {
	172	m_freem(m0);
	173	return (ENOBUFS);
	174	}
2b4b57cd BJ	175
2b4b57cd BJ	176	/*
4aed14e3	177	* Fill in mbuf with extended UDP header
2b4b57cd BJ	178	* and addresses and length put into network format.
	179	*/
	180	m->m_off = MMAXOFF - sizeof (struct udpiphdr);
	181	m->m_len = sizeof (struct udpiphdr);
	182	m->m_next = m0;
	183	ui = mtod(m, struct udpiphdr *);
	184	ui->ui_next = ui->ui_prev = 0;
	185	ui->ui_x1 = 0;
	186	ui->ui_pr = IPPROTO_UDP;
e199a8a5	187	ui->ui_len = len + sizeof (struct udphdr);
4e60622a BJ	188	ui->ui_src = inp->inp_laddr;
	189	ui->ui_dst = inp->inp_faddr;
	190	ui->ui_sport = inp->inp_lport;
	191	ui->ui_dport = inp->inp_fport;
e199a8a5	192	ui->ui_ulen = htons((u_short)ui->ui_len);
2b4b57cd BJ	193
	194	/*
	195	* Stuff checksum and output datagram.
	196	*/
	197	ui->ui_sum = 0;
ce605ce8	198	ui->ui_sum = in_cksum(m, sizeof (struct udpiphdr) + len);
4e60622a BJ	199	((struct ip *)ui)->ip_len = sizeof (struct udpiphdr) + len;
4e60622a BJ	200	((struct ip *)ui)->ip_ttl = MAXTTL;
a13c006d BJ	201	so = inp->inp_socket;
	202	return (ip_output(m, (struct mbuf *)0,
	203	(so->so_options & SO_DONTROUTE) ? &routetoif : (struct route *)0,
	204	so->so_state & SS_PRIV));
1bd53abe BJ	205	}
1bd53abe BJ	206
a8d3bf7f	207	/ARGSUSED/
bc031e39	208	udp_usrreq(so, req, m, nam)
53a5409e	209	struct socket *so;
1bd53abe	210	int req;
8075bb0e	211	struct mbuf m, nam;
1bd53abe	212	{
53a5409e	213	struct inpcb *inp = sotoinpcb(so);
8a2f82db	214	int error = 0;
1bd53abe	215
a3fa431a SL	216	if (inp == NULL && req != PRU_ATTACH) {
	217	error = EINVAL;
	218	goto release;
	219	}
1bd53abe BJ	220	switch (req) {
	221
	222	case PRU_ATTACH:
a3fa431a SL	223	if (inp != NULL) {
	224	error = EINVAL;
	225	break;
	226	}
8075bb0e BJ	227	error = in_pcballoc(so, &udb);
	228	if (error)
	229	break;
59965020	230	error = soreserve(so, 2048, 2048);
8075bb0e BJ	231	if (error)
8075bb0e BJ	232	break;
53a5409e	233	break;
1bd53abe BJ	234
1bd53abe BJ	235	case PRU_DETACH:
a3fa431a SL	236	if (inp == NULL) {
	237	error = ENOTCONN;
	238	break;
	239	}
405c9168	240	in_pcbdetach(inp);
53a5409e	241	break;
1bd53abe	242
8075bb0e BJ	243	case PRU_BIND:
	244	error = in_pcbbind(inp, nam);
	245	break;
	246
	247	case PRU_LISTEN:
	248	error = EOPNOTSUPP;
	249	break;
	250
1bd53abe	251	case PRU_CONNECT:
a3fa431a SL	252	if (inp->inp_faddr.s_addr != INADDR_ANY) {
	253	error = EISCONN;
	254	break;
	255	}
8075bb0e	256	error = in_pcbconnect(inp, nam);
8a2f82db SL	257	if (error == 0)
8a2f82db SL	258	soisconnected(so);
1bd53abe BJ	259	break;
1bd53abe BJ	260
2b4b57cd	261	case PRU_ACCEPT:
a3fa431a SL	262	error = EOPNOTSUPP;
a3fa431a SL	263	break;
2b4b57cd	264
53a5409e	265	case PRU_DISCONNECT:
a3fa431a SL	266	if (inp->inp_faddr.s_addr == INADDR_ANY) {
	267	error = ENOTCONN;
	268	break;
	269	}
405c9168	270	in_pcbdisconnect(inp);
53a5409e	271	soisdisconnected(so);
1bd53abe BJ	272	break;
1bd53abe BJ	273
cdad2eb1 BJ	274	case PRU_SHUTDOWN:
	275	socantsendmore(so);
	276	break;
	277
1aa87517 BJ	278	case PRU_SEND: {
	279	struct in_addr laddr;
	280
8075bb0e	281	if (nam) {
1aa87517	282	laddr = inp->inp_laddr;
a3fa431a SL	283	if (inp->inp_faddr.s_addr != INADDR_ANY) {
	284	error = EISCONN;
	285	break;
	286	}
8075bb0e	287	error = in_pcbconnect(inp, nam);
02bcdb4e	288	if (error)
8a2f82db	289	break;
4ad99bae	290	} else {
a3fa431a SL	291	if (inp->inp_faddr.s_addr == INADDR_ANY) {
	292	error = ENOTCONN;
	293	break;
	294	}
4ad99bae	295	}
8a2f82db	296	error = udp_output(inp, m);
a3fa431a	297	m = NULL;
8075bb0e	298	if (nam) {
405c9168	299	in_pcbdisconnect(inp);
1aa87517 BJ	300	inp->inp_laddr = laddr;
	301	}
	302	}
1bd53abe BJ	303	break;
	304
	305	case PRU_ABORT:
405c9168	306	in_pcbdetach(inp);
53a5409e BJ	307	sofree(so);
53a5409e BJ	308	soisdisconnected(so);
1bd53abe BJ	309	break;
	310
	311	case PRU_CONTROL:
a3fa431a SL	312	error = EOPNOTSUPP;
a3fa431a SL	313	break;
1bd53abe	314
126472ab	315	case PRU_SOCKADDR:
8075bb0e	316	in_setsockaddr(inp, nam);
126472ab SL	317	break;
126472ab SL	318
1bd53abe BJ	319	default:
1bd53abe BJ	320	panic("udp_usrreq");
d52566dd	321	}
a3fa431a SL	322	release:
	323	if (m != NULL)
	324	m_freem(m);
8a2f82db	325	return (error);
d52566dd	326	}