/*
 * Copyright (c) 1980, 1986, 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_usrreq.c        8.1 (Berkeley) 6/10/93
 * $FreeBSD: src/sys/net/raw_usrreq.c,v 1.18 1999/08/28 00:48:28 peter Exp $
 */

#include <sys/param.h>
#include <sys/systm.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/socketvar2.h>
#include <sys/msgport2.h>

#include <net/raw_cb.h>
#include <net/netisr2.h>

/*
 * Except from the raw_init(), rest of interfaces must be called
 * from netisr0.
 */

/*
 * Initialize raw connection block q.
 */
void
raw_init(void)
{
        LIST_INIT(&rawcb_list);
}

/************************************************************************
 *                       RAW PROTOCOL INTERFACE                         *
 ************************************************************************/

/*
 * Raw protocol input routine.  Find the socket associated with the packet(s)
 * and move them over.  If nothing exists for this packet, drop it.  This
 * routine is indirect called via rts_input() and will be serialized on
 * cpu 0.
 *
 * Most other raw protocol interface functions are also serialized XXX.
 */
void
raw_input(struct mbuf *m0, const struct sockproto *proto,
          const struct sockaddr *src, const struct sockaddr *dst,
          const struct rawcb *skip)
{
        struct rawcb *rp;
        struct mbuf *m = m0;
        struct socket *last;

        ASSERT_NETISR0;

        last = NULL;

        LIST_FOREACH(rp, &rawcb_list, list) {
                if (rp == skip)
                        continue;
                if (rp->rcb_proto.sp_family != proto->sp_family)
                        continue;
                if (rp->rcb_proto.sp_protocol  &&
                    rp->rcb_proto.sp_protocol != proto->sp_protocol)
                        continue;
                /*
                 * We assume the lower level routines have
                 * placed the address in a canonical format
                 * suitable for a structure comparison.
                 *
                 * Note that if the lengths are not the same
                 * the comparison will fail at the first byte.
                 */
                if (rp->rcb_laddr && !sa_equal(rp->rcb_laddr, dst))
                        continue;
                if (rp->rcb_faddr && !sa_equal(rp->rcb_faddr, src))
                        continue;
                /* Run any filtering that may have been installed. */
                if (rp->rcb_filter != NULL && rp->rcb_filter(m, proto, rp) != 0)
                        continue;
                if (last) {
                        struct mbuf *n;

                        n = m_copypacket(m, M_NOWAIT);
                        if (n != NULL) {
                                lwkt_gettoken(&last->so_rcv.ssb_token);
                                if (ssb_appendaddr(&last->so_rcv, src, n,
                                                 NULL) == 0) {
                                        m_freem(n);
                                        soroverflow(last);
                                } else {
                                        sorwakeup(last);
                                }
                                lwkt_reltoken(&last->so_rcv.ssb_token);
                        }
                }
                last = rp->rcb_socket;
        }
        if (last) {
                lwkt_gettoken(&last->so_rcv.ssb_token);
                if (ssb_appendaddr(&last->so_rcv, src, m, NULL) == 0) {
                        m_freem(m);
                        soroverflow(last);
                } else
                        sorwakeup(last);
                lwkt_reltoken(&last->so_rcv.ssb_token);
        } else {
                m_freem(m);
        }
}

/*
 * nm_cmd, nm_arg, nm_extra
 */
void
raw_ctlinput(netmsg_t msg)
{
        int error = 0;

        ASSERT_NETISR0;

        if (msg->ctlinput.nm_cmd < 0 || msg->ctlinput.nm_cmd > PRC_NCMDS) {
                ; /* no-op */
        }
        lwkt_replymsg(&msg->lmsg, error);
}

/*
 * NOTE: (so) is referenced from soabort*() and netmsg_pru_abort()
 *       will sofree() it when we return.
 */
static void
raw_uabort(netmsg_t msg)
{
        struct rawcb *rp = sotorawcb(msg->base.nm_so);
        int error;

        ASSERT_NETISR0;

        if (rp) {
                raw_disconnect(rp);
                soisdisconnected(msg->base.nm_so);
                error = 0;
        } else {
                error = EINVAL;
        }
        lwkt_replymsg(&msg->lmsg, error);
}

/* pru_accept is EOPNOTSUPP */

static void
raw_uattach(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 rawcb *rp;
        int error;

        ASSERT_NETISR0;

        rp = sotorawcb(so);
        if (rp) {
                error = caps_priv_check(ai->p_ucred, SYSCAP_RESTRICTEDROOT |
                                                     __SYSCAP_NULLCRED);
                if (error == 0)
                        error = raw_attach(so, proto, ai->sb_rlimit);
        } else {
                error = EINVAL;
        }
        lwkt_replymsg(&msg->lmsg, error);
}

static void
raw_ubind(netmsg_t msg)
{
        ASSERT_NETISR0;
        lwkt_replymsg(&msg->lmsg, EINVAL);
}

static void
raw_uconnect(netmsg_t msg)
{
        ASSERT_NETISR0;
        lwkt_replymsg(&msg->lmsg, EINVAL);
}

/* pru_connect2 is EOPNOTSUPP */
/* pru_control is EOPNOTSUPP */

static void
raw_udetach(netmsg_t msg)
{
        struct rawcb *rp = sotorawcb(msg->base.nm_so);
        int error;

        ASSERT_NETISR0;

        if (rp) {
                raw_detach(rp);
                error = 0;
        } else {
                error = EINVAL;
        }
        lwkt_replymsg(&msg->lmsg, error);
}

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

        ASSERT_NETISR0;

        rp = sotorawcb(so);
        if (rp == NULL) {
                error = EINVAL;
        } else if (rp->rcb_faddr == NULL) {
                error = ENOTCONN;
        } else {
                soreference(so);
                raw_disconnect(rp);
                soisdisconnected(so);
                sofree(so);
                error = 0;
        }
        lwkt_replymsg(&msg->lmsg, error);
}

/* pru_listen is EOPNOTSUPP */

static void
raw_upeeraddr(netmsg_t msg)
{
        struct rawcb *rp = sotorawcb(msg->base.nm_so);
        int error;

        ASSERT_NETISR0;

        if (rp == NULL) {
                error = EINVAL;
        } else if (rp->rcb_faddr == NULL) {
                error = ENOTCONN;
        } else {
                *msg->peeraddr.nm_nam = dup_sockaddr(rp->rcb_faddr);
                error = 0;
        }
        lwkt_replymsg(&msg->lmsg, error);
}

/* pru_rcvd is EOPNOTSUPP */
/* pru_rcvoob is EOPNOTSUPP */

static void
raw_usend(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 rawcb *rp = sotorawcb(so);
        struct pr_output_info oi;
        int flags = msg->send.nm_flags;
        int error;

        ASSERT_NETISR0;

        if (rp == NULL) {
                error = EINVAL;
                goto release;
        }

        if (flags & PRUS_OOB) {
                error = EOPNOTSUPP;
                goto release;
        }

        if (control && control->m_len) {
                error = EOPNOTSUPP;
                goto release;
        }
        if (msg->send.nm_addr) {
                if (rp->rcb_faddr) {
                        error = EISCONN;
                        goto release;
                }
                rp->rcb_faddr = msg->send.nm_addr;
        } else if (rp->rcb_faddr == NULL) {
                error = ENOTCONN;
                goto release;
        }
        oi.p_pid = msg->send.nm_td->td_proc->p_pid;
        error = (*so->so_proto->pr_output)(m, so, &oi);
        m = NULL;
        if (msg->send.nm_addr)
                rp->rcb_faddr = NULL;
release:
        if (m != NULL)
                m_freem(m);
        lwkt_replymsg(&msg->lmsg, error);
}

/* pru_sense is null */

static void
raw_ushutdown(netmsg_t msg)
{
        struct rawcb *rp = sotorawcb(msg->base.nm_so);
        int error;

        ASSERT_NETISR0;

        if (rp) {
                socantsendmore(msg->base.nm_so);
                error = 0;
        } else {
                error = EINVAL;
        }
        lwkt_replymsg(&msg->lmsg, error);
}

static void
raw_usockaddr(netmsg_t msg)
{
        struct rawcb *rp = sotorawcb(msg->base.nm_so);
        int error;

        ASSERT_NETISR0;

        if (rp == NULL) {
                error = EINVAL;
        } else if (rp->rcb_laddr == NULL) {
                error = EINVAL;
        } else {
                *msg->sockaddr.nm_nam = dup_sockaddr(rp->rcb_laddr);
                error = 0;
        }
        lwkt_replymsg(&msg->lmsg, error);
}

struct pr_usrreqs raw_usrreqs = {
        .pru_abort = raw_uabort,
        .pru_accept = pr_generic_notsupp,
        .pru_attach = raw_uattach,
        .pru_bind = raw_ubind,
        .pru_connect = raw_uconnect,
        .pru_connect2 = pr_generic_notsupp,
        .pru_control = pr_generic_notsupp,
        .pru_detach = raw_udetach,
        .pru_disconnect = raw_udisconnect,
        .pru_listen = pr_generic_notsupp,
        .pru_peeraddr = raw_upeeraddr,
        .pru_rcvd = pr_generic_notsupp,
        .pru_rcvoob = pr_generic_notsupp,
        .pru_send = raw_usend,
        .pru_sense = pru_sense_null,
        .pru_shutdown = raw_ushutdown,
        .pru_sockaddr = raw_usockaddr,
        .pru_sosend = sosend,
        .pru_soreceive = soreceive
};