/*
 * Copyright (c) 2007 The DragonFly Project.  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 DragonFly Project 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 COPYRIGHT HOLDERS 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
 * COPYRIGHT HOLDERS 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.
 */

#include <sys/globaldata.h>
#include <sys/kernel.h>
#include <sys/mbuf.h>
#include <sys/param.h>
#include <sys/sysctl.h>
#include <sys/systm.h>

#include <net/if_var.h>
#include <net/netisr.h>
#include <net/route.h>

#include <sys/mplock2.h>

#include <netproto/mpls/mpls.h>
#include <netproto/mpls/mpls_var.h>

struct mpls_stats       mplsstats_percpu[MAXCPU];
struct route            mplsforward_rt[MAXCPU];

int mplsforwarding = 1;
/*
SYSCTL_INT(_net_mpls, OID_AUTO, forwarding, CTLFLAG_RW,
    &mplsforwarding, 0, "Enable MPLS forwarding between interfaces");
*/

static void     mpls_input_handler(netmsg_t);
static void     mpls_forward(struct mbuf *);

void
mpls_init(void)
{
        int cpu;

        /*
         * Initialize MPLS statistics counters for each CPU.
         *
         */
        for (cpu = 0; cpu < ncpus; ++cpu) {
                bzero(&mplsstats_percpu[cpu], sizeof(struct mpls_stats));
        }

        netisr_register(NETISR_MPLS, mpls_input_handler, mpls_hashfn);
}

static void
mpls_input_handler(netmsg_t msg)
{
        struct mbuf *m = msg->packet.nm_packet;

        get_mplock();
        mpls_input(m);
        rel_mplock();
        /* do not reply, msg embedded in mbuf */
}

void
mpls_input(struct mbuf *m)
{
        struct mpls *mpls = NULL;
        mpls_label_t label;

        M_ASSERTPKTHDR(m);

        mplsstat.mplss_total++;

        /* length checks already performed at mpls_demux() */
        KASSERT(m->m_pkthdr.len >= sizeof(struct mpls),
            ("mpls_input: mpls header too small"));
        
again:
        if (m->m_len < sizeof(struct mpls)) {
                m = m_pullup(m, sizeof(struct mpls));
                if (m == NULL) {
                        mplsstat.mplss_toosmall++;
                        return;
                }
        }

        mpls = mtod(m, struct mpls*);
        label = MPLS_LABEL(ntohl(mpls->mpls_shim));
        switch (label) {
        case 0:
                /* 
                 * Label 0: represents "IPv4 Explicit NULL Label".
                 */
                if (MPLS_STACK(ntohl(mpls->mpls_shim))) {
                        /* Decapsulate the ip datagram from the mpls frame. */
                        m_adj(m, sizeof(struct mpls));
                        netisr_queue(NETISR_IP, m);
                        return;
                }
                goto again; /* If not the bottom label, per RFC4182. */

        case 1:
                /*
                 * Label 1: represents "Router Alert Label" and is valid 
                 * anywhere except at the bottom of the stack.
                 */
                break;

        case 2:
                /* 
                 * Label 2: represents "IPv6 Explicit NULL Label".
                 */
                if (MPLS_STACK(ntohl(mpls->mpls_shim))) {
                        /* Decapsulate the ip datagram from the mpls frame. */
                        m_adj(m, sizeof(struct mpls));
                        netisr_queue(NETISR_IPV6, m);
                        return;
                }
                goto again; /* If not the bottom label, per RFC4182. */

        case 3:
                /*
                 * Label 3: represents the "Implicit NULL Label" and must not 
                 * appear on the wire.
                 */
                break;
        default:
                /*
                 * Labels 4 - 15: reserved, drop them.
                 */
                if (label <= 15) {
                        mplsstat.mplss_reserved++;
                        m_freem(m);
                        return;
                }
                if (mplsforwarding) {
                        mpls_forward(m);
                        return;
                } else {
                        mplsstat.mplss_cantforward++;
                        m_freem(m);
                        return;
                }
        }

        mplsstat.mplss_invalid++;
        m_freem(m);
}

static void
mpls_forward(struct mbuf *m)
{
        struct sockaddr_mpls *smpls;
        struct mpls *mpls;
        struct route *cache_rt = &mplsforward_rt[mycpuid];
        mpls_label_t label;
        struct ifnet *ifp;
        struct sockaddr *dst;
        int error;

        KASSERT(m->m_len >= sizeof(struct mpls),
            ("mpls_forward: mpls header not in one mbuf"));

        mpls = mtod(m, struct mpls *);
        label = MPLS_LABEL(ntohl(mpls->mpls_shim));

        smpls = (struct sockaddr_mpls *) &cache_rt->ro_dst;
        if (cache_rt->ro_rt == NULL || smpls->smpls_label != label) {
                if (cache_rt->ro_rt != NULL) {
                        RTFREE(cache_rt->ro_rt);
                        cache_rt->ro_rt = NULL;
                }
                smpls->smpls_family = AF_MPLS;
                smpls->smpls_len = sizeof(struct sockaddr_mpls);
                smpls->smpls_label = htonl(label);
                rtalloc(cache_rt);
                if (cache_rt->ro_rt == NULL) {
                        /* route not found */
                        return;
                }
        }

        ifp = cache_rt->ro_rt->rt_ifp;
        dst = cache_rt->ro_rt->rt_gateway;
        error = mpls_output(m, cache_rt->ro_rt);
        if (error)
                goto bad;
        error = (*ifp->if_output)(ifp, m, dst, cache_rt->ro_rt);
        if (error)
                goto bad;
        mplsstat.mplss_forwarded++;

        return;
bad:
        m_freem(m);
}