/*
 * Copyright (c) 2003, 2004 Jeffrey Hsu.
 * 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. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *	This product includes software developed by Jeffrey M. Hsu.
 * 4. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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.
 *
 * $DragonFly: src/sys/kern/uipc_msg.c,v 1.4 2004/03/19 17:00:04 dillon Exp $
 */

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/msgport.h>
#include <sys/protosw.h>
#include <sys/socket.h>
#include <sys/socketvar.h>
#include <sys/socketops.h>
#include <sys/thread.h>
#include <sys/thread2.h>
#include <sys/msgport2.h>

#include <net/netisr.h>
#include <net/netmsg.h>

static int netmsg_pru_dispatcher(struct netmsg *msg);

int
so_pru_abort(struct socket *so)
{
	int error;
	struct netmsg_pru_abort msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_abort)(so));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_ABORT);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_abort;
	msg.nm_so = so;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_accept(struct socket *so, struct sockaddr **nam)
{
	int error;
	struct netmsg_pru_accept msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_accept)(so, nam));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_ACCEPT);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_accept;
	msg.nm_so = so;
	msg.nm_nam = nam;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_attach(struct socket *so, int proto, struct pru_attach_info *ai)
{
	int error;
	struct netmsg_pru_attach msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_attach)(so, proto, ai));

	port = so->so_proto->pr_mport(NULL, NULL);

	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_ATTACH);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_attach;
	msg.nm_so = so;
	msg.nm_proto = proto;
	msg.nm_ai = ai;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_bind(struct socket *so, struct sockaddr *nam, struct thread *td)
{
	int error;
	struct netmsg_pru_bind msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_bind)(so, nam, td));

	/* Send mesg to thread for new address. */
	port = so->so_proto->pr_mport(NULL, nam);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_BIND);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_bind;
	msg.nm_so = so;
	msg.nm_nam = nam;
	msg.nm_td = td;		/* used only for prison_ip() XXX JH */
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_connect(struct socket *so, struct sockaddr *nam, struct thread *td)
{
	int error;
	struct netmsg_pru_connect msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_connect)(so, nam, td));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_CONNECT);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_connect;
	msg.nm_so = so;
	msg.nm_nam = nam;
	msg.nm_td = td;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_connect2(struct socket *so1, struct socket *so2)
{
	int error;
	struct netmsg_pru_connect2 msg;
	lwkt_port_t port;

	if (!so1->so_proto->pr_mport)
		return ((*so1->so_proto->pr_usrreqs->pru_connect2)(so1, so2));

	/*
	 * Actually, connect2() is only called for Unix domain sockets
	 * and we currently short-circuit that above, so the following
	 * code is never reached.
	 */
	panic("connect2 on socket type %d", so1->so_type);
	port = so1->so_proto->pr_mport(so1, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_CONNECT2);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so1->so_proto->pr_usrreqs->pru_connect2;
	msg.nm_so1 = so1;
	msg.nm_so2 = so2;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_control(struct socket *so, u_long cmd, caddr_t data, struct ifnet *ifp,
    struct thread *td)
{
	int error;
	struct netmsg_pru_control msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_control)(so, cmd, data,
		    ifp, td));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_CONTROL);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_control;
	msg.nm_so = so;
	msg.nm_cmd = cmd;
	msg.nm_data = data;
	msg.nm_ifp = ifp;
	msg.nm_td = td;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_detach(struct socket *so)
{
	int error;
	struct netmsg_pru_detach msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_detach)(so));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_DETACH);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_detach;
	msg.nm_so = so;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_disconnect(struct socket *so)
{
	int error;
	struct netmsg_pru_disconnect msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_disconnect)(so));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_DISCONNECT);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_disconnect;
	msg.nm_so = so;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_listen(struct socket *so, struct thread *td)
{
	int error;
	struct netmsg_pru_listen msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_listen)(so, td));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_LISTEN);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_listen;
	msg.nm_so = so;
	msg.nm_td = td;		/* used only for prison_ip() XXX JH */
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_peeraddr(struct socket *so, struct sockaddr **nam)
{
	int error;
	struct netmsg_pru_peeraddr msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_peeraddr)(so, nam));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_PEERADDR);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_peeraddr;
	msg.nm_so = so;
	msg.nm_nam = nam;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_rcvd(struct socket *so, int flags)
{
	int error;
	struct netmsg_pru_rcvd msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_rcvd)(so, flags));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_RCVD);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_rcvd;
	msg.nm_so = so;
	msg.nm_flags = flags;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_rcvoob(struct socket *so, struct mbuf *m, int flags)
{
	int error;
	struct netmsg_pru_rcvoob msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_rcvoob)(so, m, flags));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_RCVOOB);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_rcvoob;
	msg.nm_so = so;
	msg.nm_m = m;
	msg.nm_flags = flags;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_send(struct socket *so, int flags, struct mbuf *m, struct sockaddr *addr,
    struct mbuf *control, struct thread *td)
{
	int error;
	struct netmsg_pru_send msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_send)(so, flags, m,
		    addr, control, td));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_SEND);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_send;
	msg.nm_so = so;
	msg.nm_flags = flags;
	msg.nm_m = m;
	msg.nm_addr = addr;
	msg.nm_control = control;
	msg.nm_td = td;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_sense(struct socket *so, struct stat *sb)
{
	int error;
	struct netmsg_pru_sense msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_sense)(so, sb));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_SENSE);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_sense;
	msg.nm_so = so;
	msg.nm_stat = sb;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_shutdown(struct socket *so)
{
	int error;
	struct netmsg_pru_shutdown msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_shutdown)(so));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_SHUTDOWN);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_shutdown;
	msg.nm_so = so;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_sockaddr(struct socket *so, struct sockaddr **nam)
{
	int error;
	struct netmsg_pru_sockaddr msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_sockaddr)(so, nam));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_SOCKADDR);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_sockaddr;
	msg.nm_so = so;
	msg.nm_nam = nam;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pru_sopoll(struct socket *so, int events, struct ucred *cred,
    struct thread *td)
{
	int error;
	struct netmsg_pru_sopoll msg;
	lwkt_port_t port;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_usrreqs->pru_sopoll)(so, events,
		    cred, td));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PRU_SOPOLL);
	msg.nm_handler = netmsg_pru_dispatcher;
	msg.nm_prufn = so->so_proto->pr_usrreqs->pru_sopoll;
	msg.nm_so = so;
	msg.nm_events = events;
	msg.nm_cred = cred;
	msg.nm_td = td;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
}

int
so_pr_ctloutput(struct socket *so, struct sockopt *sopt)
{
	return ((*so->so_proto->pr_ctloutput)(so, sopt));
#ifdef gag	/* does copyin and copyout deep inside stack XXX JH */
	struct netmsg_pr_ctloutput msg;
	lwkt_port_t port;
	int error;

	if (!so->so_proto->pr_mport)
		return ((*so->so_proto->pr_ctloutput)(so, sopt));

	port = so->so_proto->pr_mport(so, NULL);
	lwkt_initmsg(&msg.nm_lmsg, CMD_NETMSG_PR_CTLOUTPUT);
	msg.nm_handler = netmsg_pr_dispatcher;
	msg.nm_prfn = so->so_proto->pr_ctloutput;
	msg.nm_so = so;
	msg.nm_sopt = sopt;
	error = lwkt_domsg(port, &msg.nm_lmsg);
	return (error);
#endif
}

/*
 * If we convert all the pru_usrreq functions for all the protocols
 * to take a message directly, this layer can go away.
 */
static int
netmsg_pru_dispatcher(struct netmsg *msg)
{
	int error;

	switch (msg->nm_lmsg.ms_cmd) {
	case CMD_NETMSG_PRU_ABORT:
	{
		struct netmsg_pru_abort *nm = (struct netmsg_pru_abort *)msg;

		error = nm->nm_prufn(nm->nm_so);
		break;
	}
	case CMD_NETMSG_PRU_ACCEPT:
	{
		struct netmsg_pru_accept *nm = (struct netmsg_pru_accept *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_nam);
		break;
	}
	case CMD_NETMSG_PRU_ATTACH:
	{
		struct netmsg_pru_attach *nm = (struct netmsg_pru_attach *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_proto, nm->nm_ai);
		break;
	}
	case CMD_NETMSG_PRU_BIND:
	{
		struct netmsg_pru_bind *nm = (struct netmsg_pru_bind *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_nam, nm->nm_td);
		break;
	}
	case CMD_NETMSG_PRU_CONNECT:
	{
		struct netmsg_pru_connect *nm =
		    (struct netmsg_pru_connect *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_nam, nm->nm_td);
		break;
	}
	case CMD_NETMSG_PRU_CONNECT2:
	{
		struct netmsg_pru_connect2 *nm =
		    (struct netmsg_pru_connect2 *)msg;

		error = nm->nm_prufn(nm->nm_so1, nm->nm_so2);
		break;
	}
	case CMD_NETMSG_PRU_CONTROL:
	{
		struct netmsg_pru_control *nm =
		    (struct netmsg_pru_control *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_cmd, nm->nm_data,
		    nm->nm_ifp, nm->nm_td);
		break;
	}
	case CMD_NETMSG_PRU_DETACH:
	{
		struct netmsg_pru_detach *nm = (struct netmsg_pru_detach *)msg;

		error = nm->nm_prufn(nm->nm_so);
		break;
	}
	case CMD_NETMSG_PRU_DISCONNECT:
	{
		struct netmsg_pru_disconnect *nm =
		    (struct netmsg_pru_disconnect *)msg;

		error = nm->nm_prufn(nm->nm_so);
		break;
	}
	case CMD_NETMSG_PRU_LISTEN:
	{
		struct netmsg_pru_listen *nm = (struct netmsg_pru_listen *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_td);
		break;
	}
	case CMD_NETMSG_PRU_PEERADDR:
	{
		struct netmsg_pru_peeraddr *nm =
		    (struct netmsg_pru_peeraddr *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_nam);
		break;
	}
	case CMD_NETMSG_PRU_RCVD:
	{
		struct netmsg_pru_rcvd *nm = (struct netmsg_pru_rcvd *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_flags);
		break;
	}
	case CMD_NETMSG_PRU_RCVOOB:
	{
		struct netmsg_pru_rcvoob *nm = (struct netmsg_pru_rcvoob *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_m, nm->nm_flags);
		break;
	}
	case CMD_NETMSG_PRU_SEND:
	{
		struct netmsg_pru_send *nm = (struct netmsg_pru_send *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_flags, nm->nm_m,
		    nm->nm_addr, nm->nm_control, nm->nm_td);
		break;
	}
	case CMD_NETMSG_PRU_SENSE:
	{
		struct netmsg_pru_sense *nm = (struct netmsg_pru_sense *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_stat);
		break;
	}
	case CMD_NETMSG_PRU_SHUTDOWN:
	{
		struct netmsg_pru_shutdown *nm =
		    (struct netmsg_pru_shutdown *)msg;

		error = nm->nm_prufn(nm->nm_so);
		break;
	}
	case CMD_NETMSG_PRU_SOCKADDR:
	{
		struct netmsg_pru_sockaddr *nm =
		    (struct netmsg_pru_sockaddr *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_nam);
		break;
	}
	case CMD_NETMSG_PRU_SOPOLL:
	{
		struct netmsg_pru_sopoll *nm =
		    (struct netmsg_pru_sopoll *)msg;

		error = nm->nm_prufn(nm->nm_so, nm->nm_events, nm->nm_cred,
		    nm->nm_td);
		break;
	}
	default:
		panic("unknown netmsg %d", msg->nm_lmsg.ms_cmd);
		break;
	}
	return(error);
}

/*
 * If we convert all the protosw pr_ functions for all the protocols
 * to take a message directly, this layer can go away.
 */
int
netmsg_pr_dispatcher(struct netmsg *msg)
{
	int error = 0;

	switch (msg->nm_lmsg.ms_cmd) {
	case CMD_NETMSG_PR_CTLOUTPUT:
	{
		struct netmsg_pr_ctloutput *nm =
		    (struct netmsg_pr_ctloutput *)msg;

		error = nm->nm_prfn(nm->nm_so, nm->nm_sopt);
		break;
	}
	case CMD_NETMSG_PR_TIMEOUT:
	{
		struct netmsg_pr_timeout *nm = (struct netmsg_pr_timeout *)msg;

		nm->nm_prfn();
		break;
	}
	default:
		panic("unknown netmsg %d", msg->nm_lmsg.ms_cmd);
		break;
	}
	return(error);
}