/*
 * Copyright (c) 1982, 1986, 1988, 1993
 *	The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 *
 *	@(#)raw_ip.c	8.7 (Berkeley) 5/15/95
 * $FreeBSD: src/sys/netinet/raw_ip.c,v 1.64.2.16 2003/08/24 08:24:38 hsu Exp $
 */

#include "opt_inet6.h"
#include "opt_carp.h"

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/jail.h>
#include <sys/malloc.h>
#include <sys/mbuf.h>
#include <sys/proc.h>
#include <sys/caps.h>
#include <sys/protosw.h>
#include <sys/socket.h>
#include <sys/socketvar.h>
#include <sys/sysctl.h>

#include <sys/socketvar2.h>
#include <sys/msgport2.h>

#include <machine/stdarg.h>

#include <net/if.h>
#ifdef CARP
#include <net/if_types.h>
#endif
#include <net/route.h>

#define _IP_VHL
#include <netinet/in.h>
#include <netinet/in_systm.h>
#include <netinet/ip.h>
#include <netinet/in_pcb.h>
#include <netinet/in_var.h>
#include <netinet/ip_var.h>

#include <net/ip_mroute/ip_mroute.h>
#include <net/ipfw/ip_fw.h>
#include <net/ipfw3/ip_fw.h>
#include <net/dummynet/ip_dummynet.h>
#include <net/dummynet3/ip_dummynet3.h>

struct	inpcbinfo ripcbinfo;
struct	inpcbportinfo ripcbportinfo;

/*
 * hooks for multicast routing. They all default to NULL,
 * so leave them not initialized and rely on BSS being set to 0.
 */

/* The socket used to communicate with the multicast routing daemon.  */
struct socket  *ip_mrouter;

/* The various mrouter and rsvp functions */
int (*ip_mrouter_set)(struct socket *, struct sockopt *);
int (*ip_mrouter_get)(struct socket *, struct sockopt *);
int (*ip_mrouter_done)(void);
int (*ip_mforward)(struct ip *, struct ifnet *, struct mbuf *,
		struct ip_moptions *);
int (*mrt_ioctl)(u_long, caddr_t);
int (*legal_vif_num)(int);
u_long (*ip_mcast_src)(int);

int (*rsvp_input_p)(struct mbuf **, int *, int);
int (*ip_rsvp_vif)(struct socket *, struct sockopt *);
void (*ip_rsvp_force_done)(struct socket *);

/*
 * Nominal space allocated to a raw ip socket.
 */
#define	RIPSNDQ		8192
#define	RIPRCVQ		8192

/*
 * Raw interface to IP protocol.
 */

/*
 * Initialize raw connection block queue.
 */
void
rip_init(void)
{
	in_pcbinfo_init(&ripcbinfo, 0, FALSE);
	in_pcbportinfo_init(&ripcbportinfo, 1, 0);
	/*
	 * XXX We don't use the hash list for raw IP, but it's easier
	 * to allocate a one entry hash list than it is to check all
	 * over the place for hashbase == NULL.
	 */
	ripcbinfo.hashbase = hashinit(1, M_PCB, &ripcbinfo.hashmask);
	in_pcbportinfo_set(&ripcbinfo, &ripcbportinfo, 1);
	ripcbinfo.wildcardhashbase = hashinit(1, M_PCB,
					      &ripcbinfo.wildcardhashmask);
	ripcbinfo.ipi_size = sizeof(struct inpcb);
}

/*
 * Setup generic address and protocol structures
 * for raw_input routine, then pass them along with
 * mbuf chain.
 */
int
rip_input(struct mbuf **mp, int *offp, int proto)
{
	struct sockaddr_in ripsrc = { sizeof ripsrc, AF_INET };
	struct mbuf *m = *mp;
	struct ip *ip = mtod(m, struct ip *);
	struct inpcb *inp;
	struct inpcb *last = NULL;
	struct mbuf *opts = NULL;

	ASSERT_NETISR0;

	*mp = NULL;

	ripsrc.sin_addr = ip->ip_src;
	LIST_FOREACH(inp, &ripcbinfo.pcblisthead, inp_list) {
		if (inp->inp_flags & INP_PLACEMARKER)
			continue;
#ifdef INET6
		if (!INP_ISIPV4(inp))
			continue;
#endif
		if (inp->inp_ip_p && inp->inp_ip_p != proto)
			continue;
		if (inp->inp_laddr.s_addr != INADDR_ANY &&
		    inp->inp_laddr.s_addr != ip->ip_dst.s_addr)
			continue;
		if (inp->inp_faddr.s_addr != INADDR_ANY &&
		    inp->inp_faddr.s_addr != ip->ip_src.s_addr)
			continue;
		if (last) {
			struct mbuf *n = m_copypacket(m, M_NOWAIT);

			if (n) {
				lwkt_gettoken(&last->inp_socket->so_rcv.ssb_token);
				if (last->inp_flags & INP_CONTROLOPTS ||
				    last->inp_socket->so_options & SO_TIMESTAMP)
				    ip_savecontrol(last, &opts, ip, n);
				if (ssb_appendaddr(&last->inp_socket->so_rcv,
					    (struct sockaddr *)&ripsrc, n,
					    opts) == 0) {
					m_freem(n);
					if (opts)
					    m_freem(opts);
					soroverflow(last->inp_socket);
				} else {
					sorwakeup(last->inp_socket);
				}
				lwkt_reltoken(&last->inp_socket->so_rcv.ssb_token);
				opts = NULL;
			}
		}
		last = inp;
	}
	/* Check the minimum TTL for socket. */
	if (last && ip->ip_ttl < last->inp_ip_minttl) {
		m_freem(opts);
		ipstat.ips_delivered--;
	} else if (last) {
		if (last->inp_flags & INP_CONTROLOPTS ||
		    last->inp_socket->so_options & SO_TIMESTAMP)
			ip_savecontrol(last, &opts, ip, m);
		lwkt_gettoken(&last->inp_socket->so_rcv.ssb_token);
		if (ssb_appendaddr(&last->inp_socket->so_rcv,
		    (struct sockaddr *)&ripsrc, m, opts) == 0) {
			m_freem(m);
			if (opts)
			    m_freem(opts);
			soroverflow(last->inp_socket);
		} else {
			sorwakeup(last->inp_socket);
		}
		lwkt_reltoken(&last->inp_socket->so_rcv.ssb_token);
	} else {
		m_freem(m);
		ipstat.ips_noproto++;
		ipstat.ips_delivered--;
	}
	return(IPPROTO_DONE);
}

/*
 * Generate IP header and pass packet to ip_output.
 * Tack on options user may have setup with control call.
 */
int
rip_output(struct mbuf *m, struct socket *so, ...)
{
	struct ip *ip;
	struct inpcb *inp = so->so_pcb;
	__va_list ap;
	int flags = (so->so_options & SO_DONTROUTE) | IP_ALLOWBROADCAST;
	u_long dst;

	ASSERT_NETISR0;

	__va_start(ap, so);
	dst = __va_arg(ap, u_long);
	__va_end(ap);

	/*
	 * If the user handed us a complete IP packet, use it.
	 * Otherwise, allocate an mbuf for a header and fill it in.
	 */
	if ((inp->inp_flags & INP_HDRINCL) == 0) {
		if (m->m_pkthdr.len + sizeof(struct ip) > IP_MAXPACKET) {
			m_freem(m);
			return(EMSGSIZE);
		}
		M_PREPEND(m, sizeof(struct ip), M_WAITOK);
		ip = mtod(m, struct ip *);
		ip->ip_tos = inp->inp_ip_tos;
		ip->ip_off = 0;
		ip->ip_p = inp->inp_ip_p;
		ip->ip_len = htons(m->m_pkthdr.len); /* incls header now */
		ip->ip_src = inp->inp_laddr;
		ip->ip_dst.s_addr = dst;
		ip->ip_ttl = inp->inp_ip_ttl;
	} else {
		int hlen;
		int ip_len;

		if (m->m_pkthdr.len > IP_MAXPACKET) {
			m_freem(m);
			return(EMSGSIZE);
		}
		if (m->m_len < sizeof(struct ip)) {
			m = m_pullup(m, sizeof(struct ip));
			if (m == NULL)
				return ENOBUFS;
		}
		ip = mtod(m, struct ip *);
		hlen = IP_VHL_HL(ip->ip_vhl) << 2;
		ip_len = ntohs(ip->ip_len);	/* includes header */

		/*
		 * Don't allow both user specified and setsockopt options.
		 * Don't allow packet length sizes that will crash.
		 */
		if (hlen < sizeof(struct ip) ||
		    (hlen != sizeof(struct ip) && inp->inp_options) ||
		    ip_len > m->m_pkthdr.len ||
		    ip_len < hlen)
		{
			m_freem(m);
			return EINVAL;
		}

		/*
		 * System supplies ip_id if passed as 0.
		 */
		if (ip->ip_id == 0)
			ip->ip_id = ip_newid();

		/* Prevent ip_output from overwriting header fields */
		flags |= IP_RAWOUTPUT;
		ipstat.ips_rawout++;
	}

	return ip_output(m, inp->inp_options, &inp->inp_route, flags,
			 inp->inp_moptions, inp);
}

/*
 * Raw IP socket option processing.
 */
void
rip_ctloutput(netmsg_t msg)
{
	struct socket *so = msg->base.nm_so;
	struct sockopt *sopt = msg->ctloutput.nm_sopt;
	struct	inpcb *inp = so->so_pcb;
	int	error, optval;

	ASSERT_NETISR0;

	error = 0;

	/* Get socket's owner cpuid hint */
	if (sopt->sopt_level == SOL_SOCKET &&
	    sopt->sopt_dir == SOPT_GET &&
	    sopt->sopt_name == SO_CPUHINT) {
		optval = mycpuid;
		soopt_from_kbuf(sopt, &optval, sizeof(optval));
		goto done;
	}

	if (sopt->sopt_level != IPPROTO_IP) {
		error = EINVAL;
		goto done;
	}

	switch (sopt->sopt_dir) {
	case SOPT_GET:
		switch (sopt->sopt_name) {
		case IP_HDRINCL:
			optval = inp->inp_flags & INP_HDRINCL;
			soopt_from_kbuf(sopt, &optval, sizeof optval);
			break;

		case IP_FW_X:
			error = ip_fw3_sockopt(sopt);
			break;

		case IP_FW_ADD: /* ADD actually returns the body... */
		case IP_FW_GET:
		case IP_FW_TBL_GET:
		case IP_FW_TBL_EXPIRE: /* returns # of expired addresses */
			error = ip_fw_sockopt(sopt);
			break;

		case IP_DUMMYNET_GET:
			error = ip_dn_sockopt(sopt);
			break ;

		case MRT_INIT:
		case MRT_DONE:
		case MRT_ADD_VIF:
		case MRT_DEL_VIF:
		case MRT_ADD_MFC:
		case MRT_DEL_MFC:
		case MRT_VERSION:
		case MRT_ASSERT:
		case MRT_API_SUPPORT:
		case MRT_API_CONFIG:
		case MRT_ADD_BW_UPCALL:
		case MRT_DEL_BW_UPCALL:
			error = ip_mrouter_get ? ip_mrouter_get(so, sopt) :
				EOPNOTSUPP;
			break;

		default:
			ip_ctloutput(msg);
			/* msg invalid now */
			return;
		}
		break;

	case SOPT_SET:
		switch (sopt->sopt_name) {
		case IP_HDRINCL:
			error = soopt_to_kbuf(sopt, &optval, sizeof optval,
					      sizeof optval);
			if (error)
				break;
			if (optval)
				inp->inp_flags |= INP_HDRINCL;
			else
				inp->inp_flags &= ~INP_HDRINCL;
			break;

		case IP_FW_X:
			error = ip_fw3_sockopt(sopt);
			break;

		case IP_FW_ADD:
		case IP_FW_DEL:
		case IP_FW_FLUSH:
		case IP_FW_ZERO:
		case IP_FW_RESETLOG:
		case IP_FW_TBL_CREATE:
		case IP_FW_TBL_DESTROY:
		case IP_FW_TBL_ADD:
		case IP_FW_TBL_DEL:
		case IP_FW_TBL_FLUSH:
		case IP_FW_TBL_ZERO:
		case IP_FW_TBL_EXPIRE:
			error = ip_fw_sockopt(sopt);
			break;

		case IP_DUMMYNET_CONFIGURE:
		case IP_DUMMYNET_DEL:
		case IP_DUMMYNET_FLUSH:
			error = ip_dn_sockopt(sopt);
			break;

		case IP_RSVP_ON:
			error = ip_rsvp_init(so);
			break;

		case IP_RSVP_OFF:
			error = ip_rsvp_done();
			break;

		case IP_RSVP_VIF_ON:
		case IP_RSVP_VIF_OFF:
			error = ip_rsvp_vif ?
				ip_rsvp_vif(so, sopt) : EINVAL;
			break;

		case MRT_INIT:
		case MRT_DONE:
		case MRT_ADD_VIF:
		case MRT_DEL_VIF:
		case MRT_ADD_MFC:
		case MRT_DEL_MFC:
		case MRT_VERSION:
		case MRT_ASSERT:
		case MRT_API_SUPPORT:
		case MRT_API_CONFIG:
		case MRT_ADD_BW_UPCALL:
		case MRT_DEL_BW_UPCALL:
			error = ip_mrouter_set ? ip_mrouter_set(so, sopt) :
					EOPNOTSUPP;
			break;

		default:
			ip_ctloutput(msg);
			/* msg invalid now */
			return;
		}
		break;
	}
done:
	lwkt_replymsg(&msg->lmsg, error);
}

u_long	rip_sendspace = RIPSNDQ;
u_long	rip_recvspace = RIPRCVQ;

SYSCTL_INT(_net_inet_raw, OID_AUTO, maxdgram, CTLFLAG_RW,
    &rip_sendspace, 0, "Maximum outgoing raw IP datagram size");
SYSCTL_INT(_net_inet_raw, OID_AUTO, recvspace, CTLFLAG_RW,
    &rip_recvspace, 0, "Maximum incoming raw IP datagram size");

static void
rip_attach(netmsg_t msg)
{
	struct socket *so = msg->base.nm_so;
	int proto = msg->attach.nm_proto;
	struct pru_attach_info *ai = msg->attach.nm_ai;
	struct inpcb *inp;
	int error;

	ASSERT_NETISR0;

	inp = so->so_pcb;
	if (inp)
		panic("rip_attach");
	error = caps_priv_check(ai->p_ucred, SYSCAP_NONET_RAW |
					     __SYSCAP_NULLCRED);
	if (error)
		goto done;

	error = soreserve(so, rip_sendspace, rip_recvspace, ai->sb_rlimit);
	if (error)
		goto done;

	error = in_pcballoc(so, &ripcbinfo);
	if (error == 0) {
		inp = (struct inpcb *)so->so_pcb;
		inp->inp_ip_p = proto;
		inp->inp_ip_ttl = ip_defttl;
	}
done:
	lwkt_replymsg(&msg->lmsg, error);
}

static void
rip_detach(netmsg_t msg)
{
	struct socket *so = msg->base.nm_so;
	struct inpcb *inp;

	ASSERT_NETISR0;

	inp = so->so_pcb;
	if (inp == NULL)
		panic("rip_detach");
	if (so == ip_mrouter && ip_mrouter_done)
		ip_mrouter_done();
	if (ip_rsvp_force_done)
		ip_rsvp_force_done(so);
	if (so == ip_rsvpd)
		ip_rsvp_done();
	in_pcbdetach(inp);
	lwkt_replymsg(&msg->lmsg, 0);
}

static void
rip_abort(netmsg_t msg)
{
	/*
	 * Raw socket does not support listen(2),
	 * so this should never be called.
	 */
	panic("rip_abort is called");
}

static void
rip_disconnect(netmsg_t msg)
{
	struct socket *so = msg->base.nm_so;
	int error;

	ASSERT_NETISR0;

	if (so->so_state & SS_ISCONNECTED) {
		soisdisconnected(so);
		error = 0;
	} else {
		error = ENOTCONN;
	}
	lwkt_replymsg(&msg->lmsg, error);
}

static void
rip_bind(netmsg_t msg)
{
	struct socket *so = msg->base.nm_so;
	struct sockaddr *nam = msg->bind.nm_nam;
	struct inpcb *inp = so->so_pcb;
	struct sockaddr_in *addr = (struct sockaddr_in *)nam;
	int error;

	ASSERT_NETISR0;

	if (nam->sa_len == sizeof(*addr)) {
		if (ifnet_array_isempty() ||
		    ((addr->sin_family != AF_INET) &&
		     (addr->sin_family != AF_IMPLINK)) ||
		    (addr->sin_addr.s_addr != INADDR_ANY &&
		     ifa_ifwithaddr((struct sockaddr *)addr) == 0)) {
			error = EADDRNOTAVAIL;
		} else {
			inp->inp_laddr = addr->sin_addr;
			error = 0;
		}
	} else {
		error = EINVAL;
	}
	lwkt_replymsg(&msg->lmsg, error);
}

static void
rip_connect(netmsg_t msg)
{
	struct socket *so = msg->base.nm_so;
	struct sockaddr *nam = msg->connect.nm_nam;
	struct inpcb *inp = so->so_pcb;
	struct sockaddr_in *addr = (struct sockaddr_in *)nam;
	int error;

	ASSERT_NETISR0;

	if (nam->sa_len != sizeof(*addr)) {
		error = EINVAL;
	} else if (ifnet_array_isempty()) {
		error = EADDRNOTAVAIL;
	} else {
		if ((addr->sin_family != AF_INET) &&
		    (addr->sin_family != AF_IMPLINK)) {
			error = EAFNOSUPPORT;
		} else {
			inp->inp_faddr = addr->sin_addr;
			soisconnected(so);
			error = 0;
		}
	}
	lwkt_replymsg(&msg->lmsg, error);
}

static void
rip_shutdown(netmsg_t msg)
{
	ASSERT_NETISR0;

	socantsendmore(msg->base.nm_so);
	lwkt_replymsg(&msg->lmsg, 0);
}

static void
rip_send(netmsg_t msg)
{
	struct socket *so = msg->base.nm_so;
	struct mbuf *m = msg->send.nm_m;
	/*struct mbuf *control = msg->send.nm_control;*/
	struct sockaddr *nam = msg->send.nm_addr;
	/*int flags = msg->send.nm_flags;*/
	struct inpcb *inp = so->so_pcb;
	u_long dst;
	int error;

	ASSERT_NETISR0;

	if (so->so_state & SS_ISCONNECTED) {
		if (nam) {
			m_freem(m);
			error = EISCONN;
		} else {
			dst = inp->inp_faddr.s_addr;
			error = rip_output(m, so, dst);
		}
	} else {
		if (nam == NULL) {
			m_freem(m);
			error = ENOTCONN;
		} else {
			dst = ((struct sockaddr_in *)nam)->sin_addr.s_addr;
			error = rip_output(m, so, dst);
		}
	}
	lwkt_replymsg(&msg->lmsg, error);
}

SYSCTL_PROC(_net_inet_raw, OID_AUTO/*XXX*/, pcblist, CTLFLAG_RD, &ripcbinfo, 1,
	    in_pcblist_range, "S,xinpcb", "List of active raw IP sockets");

struct pr_usrreqs rip_usrreqs = {
	.pru_abort = rip_abort,
	.pru_accept = pr_generic_notsupp,
	.pru_attach = rip_attach,
	.pru_bind = rip_bind,
	.pru_connect = rip_connect,
	.pru_connect2 = pr_generic_notsupp,
	.pru_control = in_control_dispatch,
	.pru_detach = rip_detach,
	.pru_disconnect = rip_disconnect,
	.pru_listen = pr_generic_notsupp,
	.pru_peeraddr = in_setpeeraddr_dispatch,
	.pru_rcvd = pr_generic_notsupp,
	.pru_rcvoob = pr_generic_notsupp,
	.pru_send = rip_send,
	.pru_sense = pru_sense_null,
	.pru_shutdown = rip_shutdown,
	.pru_sockaddr = in_setsockaddr_dispatch,
	.pru_sosend = sosend,
	.pru_soreceive = soreceive
};