[unix-history] / usr / src / sys / net / rtsock.c

/*
 * Copyright (c) 1988 Regents of the University of California.
 * All rights reserved.
 *
 * %sccs.include.redist.c%
 *
 *	@(#)rtsock.c	7.13 (Berkeley) %G%
 */

#include "param.h"
#include "mbuf.h"
#include "user.h"
#include "proc.h"
#include "socket.h"
#include "socketvar.h"
#include "domain.h"
#include "protosw.h"
#include "errno.h"

#include "af.h"
#include "if.h"
#include "route.h"
#include "raw_cb.h"

#include "machine/mtpr.h"

struct sockaddr route_dst = { 2, PF_ROUTE, };
struct sockaddr route_src = { 2, PF_ROUTE, };
struct sockproto route_proto = { PF_ROUTE, };

/*ARGSUSED*/
route_usrreq(so, req, m, nam, control)
	register struct socket *so;
	int req;
	struct mbuf *m, *nam, *control;
{
	register int error = 0;
	register struct rawcb *rp = sotorawcb(so);
	int s;
	if (req == PRU_ATTACH) {
		MALLOC(rp, struct rawcb *, sizeof(*rp), M_PCB, M_WAITOK);
		if (so->so_pcb = (caddr_t)rp)
			bzero(so->so_pcb, sizeof(*rp));

	}
	if (req == PRU_DETACH && rp) {
		int af = rp->rcb_proto.sp_protocol;
		if (af == AF_INET)
			route_cb.ip_count--;
		else if (af == AF_NS)
			route_cb.ns_count--;
		else if (af == AF_ISO)
			route_cb.iso_count--;
		route_cb.any_count--;
	}
	s = splnet();
	error = raw_usrreq(so, req, m, nam, control);
	rp = sotorawcb(so);
	if (req == PRU_ATTACH && rp) {
		int af = rp->rcb_proto.sp_protocol;
		if (error) {
			free((caddr_t)rp, M_PCB);
			splx(s);
			return (error);
		}
		if (af == AF_INET)
			route_cb.ip_count++;
		else if (af == AF_NS)
			route_cb.ns_count++;
		else if (af == AF_ISO)
			route_cb.iso_count++;
		rp->rcb_faddr = &route_src;
		route_cb.any_count++;
		soisconnected(so);
		so->so_options |= SO_USELOOPBACK;
	}
	splx(s);
	return (error);
}
#define ROUNDUP(a) (1 + (((a) - 1) | (sizeof(long) - 1)))

/*ARGSUSED*/
route_output(m, so)
	register struct mbuf *m;
	struct socket *so;
{
	register struct rt_msghdr *rtm = 0;
	register struct rtentry *rt = 0;
	struct rtentry *saved_nrt = 0;
	struct sockaddr *dst = 0, *gate = 0, *netmask = 0, *genmask = 0;
	struct sockaddr *ifpaddr = 0;
	caddr_t cp, lim;
	int len, error = 0;
	struct ifnet *ifp = 0;
	struct	ifaddr *ifa;
	extern struct ifaddr *ifaof_ifpforaddr(), *ifa_ifwithroute();

#define senderr(e) { error = e; goto flush;}
	if (m == 0 || m->m_len < sizeof(long))
		return (ENOBUFS);
	if ((m = m_pullup(m, sizeof(long))) == 0)
		return (ENOBUFS);
	if ((m->m_flags & M_PKTHDR) == 0)
		panic("route_output");
	len = m->m_pkthdr.len;
	if (len < sizeof(*rtm) ||
	    len != mtod(m, struct rt_msghdr *)->rtm_msglen)
		senderr(EINVAL);
	R_Malloc(rtm, struct rt_msghdr *, len);
	if (rtm == 0)
		senderr(ENOBUFS);
	m_copydata(m, 0, len, (caddr_t)rtm);
	if (rtm->rtm_version != RTM_VERSION)
		senderr(EPROTONOSUPPORT);
	rtm->rtm_pid = u.u_procp->p_pid;
	lim = len + (caddr_t) rtm;
	cp = (caddr_t) (rtm + 1);
	if (rtm->rtm_addrs & RTA_DST) {
		dst = (struct sockaddr *)cp;
		cp += ROUNDUP(dst->sa_len);
	} else
		senderr(EINVAL);
	if ((rtm->rtm_addrs & RTA_GATEWAY) && cp < lim)  {
		gate = (struct sockaddr *)cp;
		cp += ROUNDUP(gate->sa_len);
	}
	if ((rtm->rtm_addrs & RTA_NETMASK) && cp < lim)  {
		netmask = (struct sockaddr *)cp;
		if (*cp)
			cp += ROUNDUP(netmask->sa_len);
		else
			cp += sizeof(long);

	}
	if ((rtm->rtm_addrs & RTA_GENMASK) && cp < lim)  {
		struct radix_node *t, *rn_addmask();
		genmask = (struct sockaddr *)cp;
		if (*cp)
			cp += ROUNDUP(netmask->sa_len);
		else
			cp += sizeof(long);
		t = rn_addmask(genmask, 1, 2);
		if (t && Bcmp(genmask, t->rn_key, *(u_char *)genmask) == 0)
			genmask = (struct sockaddr *)(t->rn_key);
		else
			senderr(ENOBUFS);
	}
	if ((rtm->rtm_addrs & RTA_IFP) && cp < lim)  {
		ifpaddr = (struct sockaddr *)cp;
	}
	switch (rtm->rtm_type) {
	case RTM_ADD:
		if (gate == 0)
			senderr(EINVAL);
		error = rtrequest(RTM_ADD, dst, gate, netmask,
					rtm->rtm_flags, &saved_nrt);
		if (error == 0 && saved_nrt) {
			rt_setmetrics(rtm->rtm_inits,
				&rtm->rtm_rmx, &saved_nrt->rt_rmx);
			saved_nrt->rt_refcnt--;
			saved_nrt->rt_genmask = genmask;
		}
		break;

	case RTM_DELETE:
		error = rtrequest(RTM_DELETE, dst, gate, netmask,
				rtm->rtm_flags, (struct rtentry **)0);
		break;

	case RTM_GET:
	case RTM_CHANGE:
	case RTM_LOCK:
		rt = rtalloc1(dst, 0);
		if (rt == 0)
			senderr(ESRCH);
		switch(rtm->rtm_type) {
			 struct	sockaddr *outmask;

		case RTM_GET:
			netmask = rt_mask(rt);
			len = sizeof(*rtm) + ROUNDUP(rt_key(rt)->sa_len);
			rtm->rtm_addrs = RTA_DST;
			if (rt->rt_gateway) {
				len += ROUNDUP(rt->rt_gateway->sa_len);
				rtm->rtm_addrs |= RTA_GATEWAY;
			}
			if (netmask) {
				len += netmask->sa_len;
				rtm->rtm_addrs |= RTA_NETMASK;
			}
			if (len > rtm->rtm_msglen) {
				struct rt_msghdr *new_rtm;
				R_Malloc(new_rtm, struct rt_msghdr *, len);
				if (new_rtm == 0)
					senderr(ENOBUFS);
				Bcopy(rtm, new_rtm, rtm->rtm_msglen);
				Free(rtm); rtm = new_rtm;
				gate = (struct sockaddr *)
				    (ROUNDUP(rt->rt_gateway->sa_len)
								+ (char *)dst);
				Bcopy(&rt->rt_gateway, gate,
						rt->rt_gateway->sa_len);
				rtm->rtm_flags = rt->rt_flags;
				if (netmask) {
				    outmask = (struct sockaddr *)
				       (ROUNDUP(netmask->sa_len)+(char *)gate);
				    Bcopy(netmask, outmask, netmask->sa_len);
				}
			}
			break;

		case RTM_CHANGE:
			if (gate == 0 || netmask != 0)
				senderr(EINVAL);
			if (gate->sa_len > (len = rt->rt_gateway->sa_len))
				senderr(EDQUOT);
			if (rt->rt_ifa && rt->rt_ifa->ifa_rtrequest)
				rt->rt_ifa->ifa_rtrequest(RTM_DELETE, rt, gate);
			/* new gateway could require new ifaddr, ifp;
			   flags may also be different; ifp may be specified
			   by ll sockaddr when protocol address is ambiguous */
			if (ifpaddr &&
			    (ifa = ifa_ifwithnet(ifpaddr)) &&
			    (ifp = ifa->ifa_ifp) &&
			    (ifa = ifaof_ifpforaddr(gate, ifp))) {
				     /* We got it */
			} else {
			    ifa = 0; ifp = 0;
			}
			Bcopy(gate, rt->rt_gateway, len);
			rt->rt_gateway->sa_len = len;
			rt_setmetrics(rtm->rtm_inits,
				&rtm->rtm_rmx, &rt->rt_rmx);
			if (ifa == 0)
			    ifa = ifa_ifwithroute(rt->rt_flags, rt_key(rt),
						gate);
			if (ifa) {
				if (rt->rt_ifa != ifa) {
				    rt->rt_ifa = ifa;
				    rt->rt_ifp = ifa->ifa_ifp;
				}
			}
			if (rt->rt_ifa && rt->rt_ifa->ifa_rtrequest)
			       rt->rt_ifa->ifa_rtrequest(RTM_ADD, rt, gate);
			if (genmask)
				rt->rt_genmask = genmask;
			/*
			 * Fall into
			 */
		case RTM_LOCK:
			rt->rt_rmx.rmx_locks |=
				(rtm->rtm_inits & rtm->rtm_rmx.rmx_locks);
			rt->rt_rmx.rmx_locks &= ~(rtm->rtm_inits);
			break;
		}
		goto cleanup;

	default:
		senderr(EOPNOTSUPP);
	}

flush:
	if (rtm) {
		if (error)
			rtm->rtm_errno = error;
		else 
			rtm->rtm_flags |= RTF_DONE;
	}
cleanup:
	if (rt)
		rtfree(rt);
    {
	register struct rawcb *rp = 0;
	/*
	 * Check to see if we don't want our own messages.
	 */
	if ((so->so_options & SO_USELOOPBACK) == 0) {
		if (route_cb.any_count <= 1) {
			if (rtm)
				Free(rtm);
			m_freem(m);
			return (error);
		}
		/* There is another listener, so construct message */
		rp = sotorawcb(so);
	}
	if (cp = (caddr_t)rtm) {
		m_copyback(m, 0, len, cp);
		Free(rtm);
	}
	if (rp)
		rp->rcb_proto.sp_family = 0; /* Avoid us */
	route_proto.sp_protocol = dst->sa_family;
	raw_input(m, &route_proto, &route_src, &route_dst);
	if (rp)
		rp->rcb_proto.sp_family = PF_ROUTE;
    }
	return (error);
}

rt_setmetrics(which, in, out)
	u_long which;
	register struct rt_metrics *in, *out;
{
#define metric(f, e) if (which & (f)) out->e = in->e;
	metric(RTV_RPIPE, rmx_recvpipe);
	metric(RTV_SPIPE, rmx_sendpipe);
	metric(RTV_SSTHRESH, rmx_ssthresh);
	metric(RTV_RTT, rmx_rtt);
	metric(RTV_RTTVAR, rmx_rttvar);
	metric(RTV_HOPCOUNT, rmx_hopcount);
	metric(RTV_MTU, rmx_mtu);
#undef metric
}

/*
 * Copy data from a buffer back into the indicated mbuf chain,
 * starting "off" bytes from the beginning, extending the mbuf
 * chain if necessary.
 */
m_copyback(m0, off, len, cp)
	struct	mbuf *m0;
	register int off;
	register int len;
	caddr_t cp;

{
	register int mlen;
	register struct mbuf *m = m0, *n;
	int totlen = 0;

	if (m0 == 0)
		return;
	while (off >= (mlen = m->m_len)) {
		off -= mlen;
		totlen += mlen;
		if (m->m_next == 0) {
			n = m_getclr(M_DONTWAIT, m->m_type);
			if (n == 0)
				goto out;
			n->m_len = min(MLEN, len + off);
			m->m_next = n;
		}
		m = m->m_next;
	}
	while (len > 0) {
		mlen = min (m->m_len - off, len);
		bcopy(cp, off + mtod(m, caddr_t), (unsigned)mlen);
		cp += mlen;
		len -= mlen;
		mlen += off;
		off = 0;
		totlen += mlen;
		if (len == 0)
			break;
		if (m->m_next == 0) {
			n = m_get(M_DONTWAIT, m->m_type);
			if (n == 0)
				break;
			n->m_len = min(MLEN, len);
			m->m_next = n;
		}
		m = m->m_next;
	}
out:	if (((m = m0)->m_flags & M_PKTHDR) && (m->m_pkthdr.len < totlen))
		m->m_pkthdr.len = totlen;
}

/* 
 * The miss message and losing message are very similar.
 */

rt_missmsg(type, dst, gate, mask, src, flags, error)
register struct sockaddr *dst;
struct sockaddr *gate, *mask, *src;
{
	register struct rt_msghdr *rtm;
	register struct mbuf *m;
	int dlen = ROUNDUP(dst->sa_len);
	int len = dlen + sizeof(*rtm);

	if (route_cb.any_count == 0)
		return;
	m = m_gethdr(M_DONTWAIT, MT_DATA);
	if (m == 0)
		return;
	m->m_pkthdr.len = m->m_len = min(len, MHLEN);
	m->m_pkthdr.rcvif = 0;
	rtm = mtod(m, struct rt_msghdr *);
	bzero((caddr_t)rtm, sizeof(*rtm)); /*XXX assumes sizeof(*rtm) < MHLEN*/
	rtm->rtm_flags = RTF_DONE | flags;
	rtm->rtm_msglen = len;
	rtm->rtm_version = RTM_VERSION;
	rtm->rtm_type = type;
	rtm->rtm_addrs = RTA_DST;
	if (type == RTM_OLDADD || type == RTM_OLDDEL) {
		rtm->rtm_pid = u.u_procp->p_pid;
	}
	m_copyback(m, sizeof (*rtm), dlen, (caddr_t)dst);
	if (gate) {
		dlen = ROUNDUP(gate->sa_len);
		m_copyback(m, len ,  dlen, (caddr_t)gate);
		len += dlen;
		rtm->rtm_addrs |= RTA_GATEWAY;
	}
	if (mask) {
		if (mask->sa_len)
			dlen = ROUNDUP(mask->sa_len);
		else
			dlen = sizeof(long);
		m_copyback(m, len ,  dlen, (caddr_t)mask);
		len += dlen;
		rtm->rtm_addrs |= RTA_NETMASK;
	}
	if (src) {
		dlen = ROUNDUP(src->sa_len);
		m_copyback(m, len ,  dlen, (caddr_t)src);
		len += dlen;
		rtm->rtm_addrs |= RTA_AUTHOR;
	}
	if (m->m_pkthdr.len != len) {
		m_freem(m);
		return;
	}
	rtm->rtm_errno = error;
	rtm->rtm_msglen = len;
	route_proto.sp_protocol = dst->sa_family;
	raw_input(m, &route_proto, &route_src, &route_dst);
}

#include "kinfo.h"
struct walkarg {
	int	w_op, w_arg;
	int	w_given, w_needed;
	caddr_t	w_where;
	struct	{
		struct rt_msghdr m_rtm;
		char	m_sabuf[128];
	} w_m;
#define w_rtm w_m.m_rtm
};
/*
 * This is used in dumping the kernel table via getkinfo().
 */
rt_dumpentry(rn, w)
	struct radix_node *rn;
	register struct walkarg *w;
{
	register struct sockaddr *sa;
	int n, error;

    for (; rn; rn = rn->rn_dupedkey) {
	int count = 0, size = sizeof(w->w_rtm);
	register struct rtentry *rt = (struct rtentry *)rn;

	if (rn->rn_flags & RNF_ROOT)
		continue;
	if (w->w_op == KINFO_RT_FLAGS && !(rt->rt_flags & w->w_arg))
		continue;
#define next(a, l) {size += (l); w->w_rtm.rtm_addrs |= (a); }
	w->w_rtm.rtm_addrs = 0;
	if (sa = rt_key(rt))
		next(RTA_DST, ROUNDUP(sa->sa_len));
	if (sa = rt->rt_gateway)
		next(RTA_GATEWAY, ROUNDUP(sa->sa_len));
	if (sa = rt_mask(rt))
		next(RTA_NETMASK,
			sa->sa_len ? ROUNDUP(sa->sa_len) : sizeof(long));
	if (sa = rt->rt_genmask)
		next(RTA_GENMASK, ROUNDUP(sa->sa_len));
	w->w_needed += size;
	if (w->w_where == NULL || w->w_needed > 0)
		continue;
	w->w_rtm.rtm_msglen = size;
	w->w_rtm.rtm_flags = rt->rt_flags;
	w->w_rtm.rtm_use = rt->rt_use;
	w->w_rtm.rtm_rmx = rt->rt_rmx;
	w->w_rtm.rtm_index = rt->rt_ifp->if_index;
#undef next
#define next(l) {n = (l); Bcopy(sa, cp, n); cp += n;}
	if (size <= sizeof(w->w_m)) {
		register caddr_t cp = (caddr_t)(w->w_m.m_sabuf);
		if (sa = rt_key(rt))
			next(ROUNDUP(sa->sa_len));
		if (sa = rt->rt_gateway)
			next(ROUNDUP(sa->sa_len));
		if (sa = rt_mask(rt))
			next(sa->sa_len ? ROUNDUP(sa->sa_len) : sizeof(long));
		if (sa = rt->rt_genmask)
			next(ROUNDUP(sa->sa_len));
#undef next
#define next(s, l) {n = (l); \
    if (error = copyout((caddr_t)(s), w->w_where, n)) return (error); \
    w->w_where += n;}

		next(&w->w_m, size); /* Copy rtmsg and sockaddrs back */
		continue;
	}
	next(&w->w_rtm, sizeof(w->w_rtm));
	if (sa = rt_key(rt))
		next(sa, ROUNDUP(sa->sa_len));
	if (sa = rt->rt_gateway)
		next(sa, ROUNDUP(sa->sa_len));
	if (sa = rt_mask(rt))
		next(sa, sa->sa_len ? ROUNDUP(sa->sa_len) : sizeof(long));
	if (sa = rt->rt_genmask)
		next(sa, ROUNDUP(sa->sa_len));
    }
	return (0);
#undef next
}

kinfo_rtable(op, where, given, arg, needed)
	int	op, arg;
	caddr_t	where;
	int	*given, *needed;
{
	register struct radix_node_head *rnh;
	int	s, error = 0;
	u_char  af = ki_af(op);
	struct	walkarg w;

	op &= 0xffff;
	if (op != KINFO_RT_DUMP && op != KINFO_RT_FLAGS)
		return (EINVAL);

	Bzero(&w, sizeof(w));
	if ((w.w_where = where) && given)
		w.w_given = *given;
	w.w_needed = 0 - w.w_given;
	w.w_arg = arg;
	w.w_op = op;
	w.w_rtm.rtm_version = RTM_VERSION;
	w.w_rtm.rtm_type = RTM_GET;

	s = splnet();
	for (rnh = radix_node_head; rnh; rnh = rnh->rnh_next) {
		if (rnh->rnh_af == 0)
			continue;
		if (af && af != rnh->rnh_af)
			continue;
		error = rt_walk(rnh->rnh_treetop, rt_dumpentry, &w);
		if (error)
			break;
	}
	w.w_needed += w.w_given;
	if (where && given)
		*given = w.w_where - where;
	else
		w.w_needed = (11 * w.w_needed) / 10;
	*needed = w.w_needed;
	splx(s);
	return (error);
}

rt_walk(rn, f, w)
	register struct radix_node *rn;
	register int (*f)();
	struct walkarg *w;
{
	int error;
	for (;;) {
		while (rn->rn_b >= 0)
			rn = rn->rn_l;	/* First time through node, go left */
		if (error = (*f)(rn, w))
			return (error);	/* Process Leaf */
		while (rn->rn_p->rn_r == rn) {	/* if coming back from right */
			rn = rn->rn_p;		/* go back up */
			if (rn->rn_flags & RNF_ROOT)
				return 0;
		}
		rn = rn->rn_p->rn_r;		/* otherwise, go right*/
	}
}

/*
 * Definitions of protocols supported in the ROUTE domain.
 */

int	raw_init(),raw_usrreq(),raw_input(),raw_ctlinput();
extern	struct domain routedomain;		/* or at least forward */

struct protosw routesw[] = {
{ SOCK_RAW,	&routedomain,	0,		PR_ATOMIC|PR_ADDR,
  raw_input,	route_output,	raw_ctlinput,	0,
  route_usrreq,
  raw_init,	0,		0,		0,
}
};

int	unp_externalize(), unp_dispose();

struct domain routedomain =
    { PF_ROUTE, "route", 0, 0, 0,
      routesw, &routesw[sizeof(routesw)/sizeof(routesw[0])] };
Commit	Line	Data
e831d540 KS	1	/*
	2	* Copyright (c) 1988 Regents of the University of California.
	3	* All rights reserved.
	4	*
dbf0c423	5	* %sccs.include.redist.c%
e831d540	6	*
67336957	7	* @(#)rtsock.c 7.13 (Berkeley) %G%
e831d540 KS	8	*/
e831d540 KS	9
e831d540 KS	10	#include "param.h"
e831d540 KS	11	#include "mbuf.h"
e831d540 KS	12	#include "user.h"
	13	#include "proc.h"
	14	#include "socket.h"
	15	#include "socketvar.h"
	16	#include "domain.h"
	17	#include "protosw.h"
	18	#include "errno.h"
	19
	20	#include "af.h"
a0d75f26	21	#include "if.h"
e831d540 KS	22	#include "route.h"
	23	#include "raw_cb.h"
	24
d301d150	25	#include "machine/mtpr.h"
e831d540	26
1d888e3b KS	27	struct sockaddr route_dst = { 2, PF_ROUTE, };
1d888e3b KS	28	struct sockaddr route_src = { 2, PF_ROUTE, };
b72a6efb KS	29	struct sockproto route_proto = { PF_ROUTE, };
b72a6efb KS	30
e831d540	31	/ARGSUSED/
a0d75f26	32	route_usrreq(so, req, m, nam, control)
e831d540 KS	33	register struct socket *so;
e831d540 KS	34	int req;
a0d75f26	35	struct mbuf m, nam, *control;
e831d540 KS	36	{
	37	register int error = 0;
	38	register struct rawcb *rp = sotorawcb(so);
67336957	39	int s;
b72a6efb KS	40	if (req == PRU_ATTACH) {
	41	MALLOC(rp, struct rawcb , sizeof(rp), M_PCB, M_WAITOK);
	42	if (so->so_pcb = (caddr_t)rp)
	43	bzero(so->so_pcb, sizeof(*rp));
	44
	45	}
e831d540 KS	46	if (req == PRU_DETACH && rp) {
	47	int af = rp->rcb_proto.sp_protocol;
	48	if (af == AF_INET)
	49	route_cb.ip_count--;
	50	else if (af == AF_NS)
	51	route_cb.ns_count--;
e831d540 KS	52	else if (af == AF_ISO)
e831d540 KS	53	route_cb.iso_count--;
e831d540 KS	54	route_cb.any_count--;
e831d540 KS	55	}
67336957	56	s = splnet();
a0d75f26	57	error = raw_usrreq(so, req, m, nam, control);
e831d540	58	rp = sotorawcb(so);
b72a6efb	59	if (req == PRU_ATTACH && rp) {
e831d540	60	int af = rp->rcb_proto.sp_protocol;
b72a6efb KS	61	if (error) {
b72a6efb KS	62	free((caddr_t)rp, M_PCB);
67336957	63	splx(s);
b72a6efb KS	64	return (error);
b72a6efb KS	65	}
e831d540 KS	66	if (af == AF_INET)
	67	route_cb.ip_count++;
	68	else if (af == AF_NS)
	69	route_cb.ns_count++;
e831d540 KS	70	else if (af == AF_ISO)
e831d540 KS	71	route_cb.iso_count++;
b72a6efb	72	rp->rcb_faddr = &route_src;
e831d540 KS	73	route_cb.any_count++;
e831d540 KS	74	soisconnected(so);
2e5be88b	75	so->so_options \|= SO_USELOOPBACK;
e831d540	76	}
67336957	77	splx(s);
e831d540 KS	78	return (error);
	79	}
	80	#define ROUNDUP(a) (1 + (((a) - 1) \| (sizeof(long) - 1)))
	81
b72a6efb	82	/ARGSUSED/
e831d540 KS	83	route_output(m, so)
	84	register struct mbuf *m;
	85	struct socket *so;
	86	{
67336957	87	register struct rt_msghdr *rtm = 0;
e831d540	88	register struct rtentry *rt = 0;
a0d75f26	89	struct rtentry *saved_nrt = 0;
79e0ea5d	90	struct sockaddr dst = 0, gate = 0, netmask = 0, genmask = 0;
d6bc9b28	91	struct sockaddr *ifpaddr = 0;
79e0ea5d	92	caddr_t cp, lim;
b72a6efb	93	int len, error = 0;
d6bc9b28 KS	94	struct ifnet *ifp = 0;
	95	struct ifaddr *ifa;
	96	extern struct ifaddr ifaof_ifpforaddr(), ifa_ifwithroute();
e831d540	97
a0d75f26	98	#define senderr(e) { error = e; goto flush;}
79e0ea5d KS	99	if (m == 0 \|\| m->m_len < sizeof(long))
	100	return (ENOBUFS);
	101	if ((m = m_pullup(m, sizeof(long))) == 0)
32dfe9ab	102	return (ENOBUFS);
e831d540	103	if ((m->m_flags & M_PKTHDR) == 0)
32dfe9ab	104	panic("route_output");
e831d540	105	len = m->m_pkthdr.len;
67336957 KS	106	if (len < sizeof(*rtm) \|\|
67336957 KS	107	len != mtod(m, struct rt_msghdr *)->rtm_msglen)
a0d75f26	108	senderr(EINVAL);
e831d540 KS	109	R_Malloc(rtm, struct rt_msghdr *, len);
e831d540 KS	110	if (rtm == 0)
a0d75f26	111	senderr(ENOBUFS);
e831d540	112	m_copydata(m, 0, len, (caddr_t)rtm);
79e0ea5d	113	if (rtm->rtm_version != RTM_VERSION)
a0d75f26	114	senderr(EPROTONOSUPPORT);
e831d540	115	rtm->rtm_pid = u.u_procp->p_pid;
79e0ea5d	116	lim = len + (caddr_t) rtm;
e831d540	117	cp = (caddr_t) (rtm + 1);
79e0ea5d	118	if (rtm->rtm_addrs & RTA_DST) {
e831d540 KS	119	dst = (struct sockaddr *)cp;
e831d540 KS	120	cp += ROUNDUP(dst->sa_len);
79e0ea5d KS	121	} else
	122	senderr(EINVAL);
	123	if ((rtm->rtm_addrs & RTA_GATEWAY) && cp < lim) {
e831d540 KS	124	gate = (struct sockaddr *)cp;
	125	cp += ROUNDUP(gate->sa_len);
	126	}
79e0ea5d	127	if ((rtm->rtm_addrs & RTA_NETMASK) && cp < lim) {
e831d540	128	netmask = (struct sockaddr *)cp;
b72a6efb KS	129	if (*cp)
	130	cp += ROUNDUP(netmask->sa_len);
	131	else
	132	cp += sizeof(long);
	133
e831d540	134	}
79e0ea5d	135	if ((rtm->rtm_addrs & RTA_GENMASK) && cp < lim) {
d6bc9b28	136	struct radix_node t, rn_addmask();
79e0ea5d	137	genmask = (struct sockaddr *)cp;
d6bc9b28 KS	138	if (*cp)
	139	cp += ROUNDUP(netmask->sa_len);
	140	else
	141	cp += sizeof(long);
	142	t = rn_addmask(genmask, 1, 2);
	143	if (t && Bcmp(genmask, t->rn_key, (u_char )genmask) == 0)
	144	genmask = (struct sockaddr *)(t->rn_key);
	145	else
	146	senderr(ENOBUFS);
	147	}
	148	if ((rtm->rtm_addrs & RTA_IFP) && cp < lim) {
	149	ifpaddr = (struct sockaddr *)cp;
e831d540 KS	150	}
	151	switch (rtm->rtm_type) {
	152	case RTM_ADD:
79e0ea5d KS	153	if (gate == 0)
79e0ea5d KS	154	senderr(EINVAL);
e831d540 KS	155	error = rtrequest(RTM_ADD, dst, gate, netmask,
e831d540 KS	156	rtm->rtm_flags, &saved_nrt);
79e0ea5d KS	157	if (error == 0 && saved_nrt) {
	158	rt_setmetrics(rtm->rtm_inits,
	159	&rtm->rtm_rmx, &saved_nrt->rt_rmx);
a0d75f26	160	saved_nrt->rt_refcnt--;
d6bc9b28	161	saved_nrt->rt_genmask = genmask;
79e0ea5d	162	}
e831d540 KS	163	break;
	164
	165	case RTM_DELETE:
	166	error = rtrequest(RTM_DELETE, dst, gate, netmask,
b72a6efb	167	rtm->rtm_flags, (struct rtentry **)0);
e831d540 KS	168	break;
	169
	170	case RTM_GET:
	171	case RTM_CHANGE:
	172	case RTM_LOCK:
	173	rt = rtalloc1(dst, 0);
	174	if (rt == 0)
a0d75f26	175	senderr(ESRCH);
e831d540 KS	176	switch(rtm->rtm_type) {
	177	struct sockaddr *outmask;
	178
	179	case RTM_GET:
	180	netmask = rt_mask(rt);
79e0ea5d KS	181	len = sizeof(*rtm) + ROUNDUP(rt_key(rt)->sa_len);
	182	rtm->rtm_addrs = RTA_DST;
	183	if (rt->rt_gateway) {
	184	len += ROUNDUP(rt->rt_gateway->sa_len);
	185	rtm->rtm_addrs \|= RTA_GATEWAY;
	186	}
	187	if (netmask) {
b72a6efb	188	len += netmask->sa_len;
79e0ea5d KS	189	rtm->rtm_addrs \|= RTA_NETMASK;
79e0ea5d KS	190	}
e831d540 KS	191	if (len > rtm->rtm_msglen) {
	192	struct rt_msghdr *new_rtm;
	193	R_Malloc(new_rtm, struct rt_msghdr *, len);
	194	if (new_rtm == 0)
a0d75f26	195	senderr(ENOBUFS);
e831d540 KS	196	Bcopy(rtm, new_rtm, rtm->rtm_msglen);
	197	Free(rtm); rtm = new_rtm;
	198	gate = (struct sockaddr *)
	199	(ROUNDUP(rt->rt_gateway->sa_len)
	200	+ (char *)dst);
	201	Bcopy(&rt->rt_gateway, gate,
	202	rt->rt_gateway->sa_len);
	203	rtm->rtm_flags = rt->rt_flags;
e831d540 KS	204	if (netmask) {
	205	outmask = (struct sockaddr *)
	206	(ROUNDUP(netmask->sa_len)+(char *)gate);
	207	Bcopy(netmask, outmask, netmask->sa_len);
e831d540 KS	208	}
	209	}
	210	break;
	211
	212	case RTM_CHANGE:
d6bc9b28	213	if (gate == 0 \|\| netmask != 0)
79e0ea5d	214	senderr(EINVAL);
e831d540	215	if (gate->sa_len > (len = rt->rt_gateway->sa_len))
a0d75f26 KS	216	senderr(EDQUOT);
a0d75f26 KS	217	if (rt->rt_ifa && rt->rt_ifa->ifa_rtrequest)
d6bc9b28 KS	218	rt->rt_ifa->ifa_rtrequest(RTM_DELETE, rt, gate);
	219	/* new gateway could require new ifaddr, ifp;
	220	flags may also be different; ifp may be specified
	221	by ll sockaddr when protocol address is ambiguous */
	222	if (ifpaddr &&
	223	(ifa = ifa_ifwithnet(ifpaddr)) &&
	224	(ifp = ifa->ifa_ifp) &&
	225	(ifa = ifaof_ifpforaddr(gate, ifp))) {
	226	/* We got it */
	227	} else {
	228	ifa = 0; ifp = 0;
	229	}
e831d540 KS	230	Bcopy(gate, rt->rt_gateway, len);
e831d540 KS	231	rt->rt_gateway->sa_len = len;
79e0ea5d KS	232	rt_setmetrics(rtm->rtm_inits,
79e0ea5d KS	233	&rtm->rtm_rmx, &rt->rt_rmx);
d6bc9b28 KS	234	if (ifa == 0)
	235	ifa = ifa_ifwithroute(rt->rt_flags, rt_key(rt),
	236	gate);
	237	if (ifa) {
	238	if (rt->rt_ifa != ifa) {
	239	rt->rt_ifa = ifa;
	240	rt->rt_ifp = ifa->ifa_ifp;
	241	}
	242	}
	243	if (rt->rt_ifa && rt->rt_ifa->ifa_rtrequest)
	244	rt->rt_ifa->ifa_rtrequest(RTM_ADD, rt, gate);
	245	if (genmask)
	246	rt->rt_genmask = genmask;
e831d540 KS	247	/*
	248	* Fall into
	249	*/
79e0ea5d KS	250	case RTM_LOCK:
	251	rt->rt_rmx.rmx_locks \|=
	252	(rtm->rtm_inits & rtm->rtm_rmx.rmx_locks);
	253	rt->rt_rmx.rmx_locks &= ~(rtm->rtm_inits);
e831d540	254	break;
e831d540 KS	255	}
	256	goto cleanup;
	257
	258	default:
a0d75f26	259	senderr(EOPNOTSUPP);
e831d540 KS	260	}
	261
	262	flush:
	263	if (rtm) {
	264	if (error)
	265	rtm->rtm_errno = error;
	266	else
	267	rtm->rtm_flags \|= RTF_DONE;
	268	}
	269	cleanup:
	270	if (rt)
	271	rtfree(rt);
2e5be88b KS	272	{
	273	register struct rawcb *rp = 0;
	274	/*
	275	* Check to see if we don't want our own messages.
	276	*/
	277	if ((so->so_options & SO_USELOOPBACK) == 0) {
	278	if (route_cb.any_count <= 1) {
	279	if (rtm)
	280	Free(rtm);
	281	m_freem(m);
	282	return (error);
	283	}
	284	/* There is another listener, so construct message */
	285	rp = sotorawcb(so);
	286	}
32dfe9ab KS	287	if (cp = (caddr_t)rtm) {
	288	m_copyback(m, 0, len, cp);
	289	Free(rtm);
	290	}
2e5be88b KS	291	if (rp)
2e5be88b KS	292	rp->rcb_proto.sp_family = 0; /* Avoid us */
e831d540	293	route_proto.sp_protocol = dst->sa_family;
32dfe9ab	294	raw_input(m, &route_proto, &route_src, &route_dst);
2e5be88b KS	295	if (rp)
	296	rp->rcb_proto.sp_family = PF_ROUTE;
	297	}
e831d540 KS	298	return (error);
	299	}
	300
d6bc9b28	301	rt_setmetrics(which, in, out)
79e0ea5d KS	302	u_long which;
	303	register struct rt_metrics in, out;
	304	{
	305	#define metric(f, e) if (which & (f)) out->e = in->e;
	306	metric(RTV_RPIPE, rmx_recvpipe);
	307	metric(RTV_SPIPE, rmx_sendpipe);
	308	metric(RTV_SSTHRESH, rmx_ssthresh);
	309	metric(RTV_RTT, rmx_rtt);
	310	metric(RTV_RTTVAR, rmx_rttvar);
	311	metric(RTV_HOPCOUNT, rmx_hopcount);
	312	metric(RTV_MTU, rmx_mtu);
	313	#undef metric
	314	}
	315
e831d540 KS	316	/*
	317	* Copy data from a buffer back into the indicated mbuf chain,
	318	* starting "off" bytes from the beginning, extending the mbuf
	319	* chain if necessary.
	320	*/
	321	m_copyback(m0, off, len, cp)
	322	struct mbuf *m0;
	323	register int off;
	324	register int len;
	325	caddr_t cp;
	326
	327	{
	328	register int mlen;
	329	register struct mbuf m = m0, n;
	330	int totlen = 0;
	331
	332	if (m0 == 0)
	333	return;
	334	while (off >= (mlen = m->m_len)) {
	335	off -= mlen;
	336	totlen += mlen;
	337	if (m->m_next == 0) {
	338	n = m_getclr(M_DONTWAIT, m->m_type);
	339	if (n == 0)
	340	goto out;
	341	n->m_len = min(MLEN, len + off);
	342	m->m_next = n;
	343	}
	344	m = m->m_next;
	345	}
	346	while (len > 0) {
	347	mlen = min (m->m_len - off, len);
b72a6efb	348	bcopy(cp, off + mtod(m, caddr_t), (unsigned)mlen);
e831d540 KS	349	cp += mlen;
	350	len -= mlen;
	351	mlen += off;
	352	off = 0;
	353	totlen += mlen;
	354	if (len == 0)
	355	break;
	356	if (m->m_next == 0) {
	357	n = m_get(M_DONTWAIT, m->m_type);
	358	if (n == 0)
	359	break;
	360	n->m_len = min(MLEN, len);
	361	m->m_next = n;
	362	}
	363	m = m->m_next;
	364	}
	365	out: if (((m = m0)->m_flags & M_PKTHDR) && (m->m_pkthdr.len < totlen))
	366	m->m_pkthdr.len = totlen;
	367	}
	368
	369	/*
	370	* The miss message and losing message are very similar.
	371	*/
	372
b72a6efb	373	rt_missmsg(type, dst, gate, mask, src, flags, error)
e831d540 KS	374	register struct sockaddr *dst;
	375	struct sockaddr gate, mask, *src;
	376	{
	377	register struct rt_msghdr *rtm;
	378	register struct mbuf *m;
	379	int dlen = ROUNDUP(dst->sa_len);
	380	int len = dlen + sizeof(*rtm);
e831d540 KS	381
	382	if (route_cb.any_count == 0)
	383	return;
	384	m = m_gethdr(M_DONTWAIT, MT_DATA);
	385	if (m == 0)
	386	return;
	387	m->m_pkthdr.len = m->m_len = min(len, MHLEN);
	388	m->m_pkthdr.rcvif = 0;
	389	rtm = mtod(m, struct rt_msghdr *);
	390	bzero((caddr_t)rtm, sizeof(rtm)); /XXX assumes sizeof(rtm) < MHLEN/
	391	rtm->rtm_flags = RTF_DONE \| flags;
	392	rtm->rtm_msglen = len;
79e0ea5d	393	rtm->rtm_version = RTM_VERSION;
e831d540	394	rtm->rtm_type = type;
79e0ea5d	395	rtm->rtm_addrs = RTA_DST;
e831d540	396	if (type == RTM_OLDADD \|\| type == RTM_OLDDEL) {
e831d540	397	rtm->rtm_pid = u.u_procp->p_pid;
e831d540 KS	398	}
	399	m_copyback(m, sizeof (*rtm), dlen, (caddr_t)dst);
	400	if (gate) {
	401	dlen = ROUNDUP(gate->sa_len);
	402	m_copyback(m, len , dlen, (caddr_t)gate);
	403	len += dlen;
79e0ea5d	404	rtm->rtm_addrs \|= RTA_GATEWAY;
e831d540 KS	405	}
e831d540 KS	406	if (mask) {
b72a6efb KS	407	if (mask->sa_len)
	408	dlen = ROUNDUP(mask->sa_len);
	409	else
	410	dlen = sizeof(long);
e831d540 KS	411	m_copyback(m, len , dlen, (caddr_t)mask);
e831d540 KS	412	len += dlen;
79e0ea5d	413	rtm->rtm_addrs \|= RTA_NETMASK;
e831d540 KS	414	}
	415	if (src) {
	416	dlen = ROUNDUP(src->sa_len);
	417	m_copyback(m, len , dlen, (caddr_t)src);
	418	len += dlen;
79e0ea5d	419	rtm->rtm_addrs \|= RTA_AUTHOR;
e831d540 KS	420	}
	421	if (m->m_pkthdr.len != len) {
	422	m_freem(m);
	423	return;
	424	}
b72a6efb	425	rtm->rtm_errno = error;
e831d540 KS	426	rtm->rtm_msglen = len;
	427	route_proto.sp_protocol = dst->sa_family;
	428	raw_input(m, &route_proto, &route_src, &route_dst);
	429	}
	430
a0d75f26	431	#include "kinfo.h"
a0d75f26 KS	432	struct walkarg {
	433	int w_op, w_arg;
	434	int w_given, w_needed;
	435	caddr_t w_where;
	436	struct {
	437	struct rt_msghdr m_rtm;
	438	char m_sabuf[128];
	439	} w_m;
	440	#define w_rtm w_m.m_rtm
	441	};
	442	/*
	443	* This is used in dumping the kernel table via getkinfo().
	444	*/
	445	rt_dumpentry(rn, w)
	446	struct radix_node *rn;
	447	register struct walkarg *w;
	448	{
	449	register struct sockaddr *sa;
	450	int n, error;
	451
2e5be88b	452	for (; rn; rn = rn->rn_dupedkey) {
a0d75f26 KS	453	int count = 0, size = sizeof(w->w_rtm);
	454	register struct rtentry rt = (struct rtentry )rn;
	455
2e5be88b KS	456	if (rn->rn_flags & RNF_ROOT)
2e5be88b KS	457	continue;
a0d75f26 KS	458	if (w->w_op == KINFO_RT_FLAGS && !(rt->rt_flags & w->w_arg))
a0d75f26 KS	459	continue;
79e0ea5d KS	460	#define next(a, l) {size += (l); w->w_rtm.rtm_addrs \|= (a); }
79e0ea5d KS	461	w->w_rtm.rtm_addrs = 0;
a0d75f26	462	if (sa = rt_key(rt))
79e0ea5d	463	next(RTA_DST, ROUNDUP(sa->sa_len));
a0d75f26	464	if (sa = rt->rt_gateway)
79e0ea5d KS	465	next(RTA_GATEWAY, ROUNDUP(sa->sa_len));
	466	if (sa = rt_mask(rt))
	467	next(RTA_NETMASK,
	468	sa->sa_len ? ROUNDUP(sa->sa_len) : sizeof(long));
a0d75f26	469	if (sa = rt->rt_genmask)
79e0ea5d	470	next(RTA_GENMASK, ROUNDUP(sa->sa_len));
a0d75f26 KS	471	w->w_needed += size;
	472	if (w->w_where == NULL \|\| w->w_needed > 0)
	473	continue;
	474	w->w_rtm.rtm_msglen = size;
a0d75f26	475	w->w_rtm.rtm_flags = rt->rt_flags;
79e0ea5d KS	476	w->w_rtm.rtm_use = rt->rt_use;
	477	w->w_rtm.rtm_rmx = rt->rt_rmx;
	478	w->w_rtm.rtm_index = rt->rt_ifp->if_index;
	479	#undef next
	480	#define next(l) {n = (l); Bcopy(sa, cp, n); cp += n;}
a0d75f26 KS	481	if (size <= sizeof(w->w_m)) {
a0d75f26 KS	482	register caddr_t cp = (caddr_t)(w->w_m.m_sabuf);
a0d75f26	483	if (sa = rt_key(rt))
79e0ea5d	484	next(ROUNDUP(sa->sa_len));
a0d75f26	485	if (sa = rt->rt_gateway)
79e0ea5d KS	486	next(ROUNDUP(sa->sa_len));
	487	if (sa = rt_mask(rt))
	488	next(sa->sa_len ? ROUNDUP(sa->sa_len) : sizeof(long));
a0d75f26	489	if (sa = rt->rt_genmask)
79e0ea5d	490	next(ROUNDUP(sa->sa_len));
a0d75f26	491	#undef next
79e0ea5d KS	492	#define next(s, l) {n = (l); \
	493	if (error = copyout((caddr_t)(s), w->w_where, n)) return (error); \
	494	w->w_where += n;}
a0d75f26 KS	495
	496	next(&w->w_m, size); /* Copy rtmsg and sockaddrs back */
	497	continue;
	498	}
	499	next(&w->w_rtm, sizeof(w->w_rtm));
	500	if (sa = rt_key(rt))
	501	next(sa, ROUNDUP(sa->sa_len));
	502	if (sa = rt->rt_gateway)
	503	next(sa, ROUNDUP(sa->sa_len));
79e0ea5d KS	504	if (sa = rt_mask(rt))
79e0ea5d KS	505	next(sa, sa->sa_len ? ROUNDUP(sa->sa_len) : sizeof(long));
a0d75f26 KS	506	if (sa = rt->rt_genmask)
	507	next(sa, ROUNDUP(sa->sa_len));
	508	}
	509	return (0);
	510	#undef next
	511	}
	512
	513	kinfo_rtable(op, where, given, arg, needed)
	514	int op, arg;
	515	caddr_t where;
	516	int given, needed;
	517	{
	518	register struct radix_node_head *rnh;
	519	int s, error = 0;
	520	u_char af = ki_af(op);
	521	struct walkarg w;
	522
	523	op &= 0xffff;
	524	if (op != KINFO_RT_DUMP && op != KINFO_RT_FLAGS)
	525	return (EINVAL);
	526
	527	Bzero(&w, sizeof(w));
	528	if ((w.w_where = where) && given)
	529	w.w_given = *given;
	530	w.w_needed = 0 - w.w_given;
	531	w.w_arg = arg;
	532	w.w_op = op;
79e0ea5d	533	w.w_rtm.rtm_version = RTM_VERSION;
a0d75f26 KS	534	w.w_rtm.rtm_type = RTM_GET;
	535
	536	s = splnet();
	537	for (rnh = radix_node_head; rnh; rnh = rnh->rnh_next) {
	538	if (rnh->rnh_af == 0)
	539	continue;
	540	if (af && af != rnh->rnh_af)
	541	continue;
	542	error = rt_walk(rnh->rnh_treetop, rt_dumpentry, &w);
	543	if (error)
	544	break;
	545	}
	546	w.w_needed += w.w_given;
	547	if (where && given)
	548	*given = w.w_where - where;
	549	else
	550	w.w_needed = (11 * w.w_needed) / 10;
	551	*needed = w.w_needed;
	552	splx(s);
	553	return (error);
	554	}
	555
	556	rt_walk(rn, f, w)
	557	register struct radix_node *rn;
	558	register int (*f)();
	559	struct walkarg *w;
	560	{
	561	int error;
	562	for (;;) {
	563	while (rn->rn_b >= 0)
	564	rn = rn->rn_l; /* First time through node, go left */
	565	if (error = (*f)(rn, w))
	566	return (error); /* Process Leaf */
	567	while (rn->rn_p->rn_r == rn) { /* if coming back from right */
	568	rn = rn->rn_p; /* go back up */
	569	if (rn->rn_flags & RNF_ROOT)
	570	return 0;
	571	}
	572	rn = rn->rn_p->rn_r; /* otherwise, go right*/
	573	}
	574	}
	575
e831d540 KS	576	/*
	577	* Definitions of protocols supported in the ROUTE domain.
	578	*/
	579
e831d540 KS	580	int raw_init(),raw_usrreq(),raw_input(),raw_ctlinput();
	581	extern struct domain routedomain; /* or at least forward */
	582
	583	struct protosw routesw[] = {
	584	{ SOCK_RAW, &routedomain, 0, PR_ATOMIC\|PR_ADDR,
	585	raw_input, route_output, raw_ctlinput, 0,
	586	route_usrreq,
	587	raw_init, 0, 0, 0,
e831d540 KS	588	}
	589	};
	590
	591	int unp_externalize(), unp_dispose();
	592
	593	struct domain routedomain =
	594	{ PF_ROUTE, "route", 0, 0, 0,
	595	routesw, &routesw[sizeof(routesw)/sizeof(routesw[0])] };