/* $NetBSD: mpls_interface.c,v 1.13 2013/08/02 07:29:56 kefren Exp $ */

/*-
* Copyright (c) 2010 The NetBSD Foundation, Inc.
* All rights reserved.
*
* This code is derived from software contributed to The NetBSD Foundation
* by Mihai Chelaru <[email protected]>
*
* 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.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. 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 FOUNDATION 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 <arpa/inet.h>
#include <netinet/in.h>
#include <netmpls/mpls.h>
#include <net/route.h>

#include <sys/param.h>

#include <assert.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>

#include "ldp.h"
#include "ldp_peer.h"
#include "ldp_errors.h"
#include "tlv_stack.h"
#include "label.h"
#include "mpls_interface.h"
#include "mpls_routes.h"

extern int no_default_route;

int
mpls_add_label(struct label *lab)
{
       char            p_str[20];
       union sockunion *so_dest, *so_nexthop, *so_tag, so_ifa;
       uint8_t prefixlen;
       uint32_t oldbinding;
       struct peer_map *pm;

       assert(lab != NULL);

       oldbinding = lab->binding;
       prefixlen = from_union_to_cidr(&lab->so_pref);

       strlcpy(p_str, satos(lab->p->address), sizeof(p_str));
       warnp("Trying to add %s/%d as label %d (oldlocal %d) to peer %s\n",
           satos(&lab->so_dest.sa), prefixlen, lab->label, lab->binding,p_str);

       /* Check if we should accept default route */
       if (prefixlen == 0 && no_default_route != 0)
               return LDP_E_BAD_AF;

       /* double check if there is a label mapping for this */
       if ((pm = ldp_test_mapping(&lab->so_dest.sa, prefixlen,
           &lab->so_gate.sa)) == NULL || pm->peer != lab->p) {
               if (pm != NULL)
                       free(pm);
               return LDP_E_NOENT;
       }
       free(pm);

       if (lab->so_gate.sa.sa_family != AF_INET &&
           lab->so_gate.sa.sa_family != AF_INET6) {
               warnp("mpls_add_label: so_gate is not IP or IPv6\n");
               return LDP_E_BAD_AF;
       }

       /* Check if the address is bounded to the peer */
       if (check_ifaddr(lab->p, &lab->so_gate.sa) == NULL) {
               warnp("Failed at next-hop check\n");
               return LDP_E_ROUTE_ERROR;
       }

       /* if binding is implicit null we need to generate a new one */
       if (lab->binding == MPLS_LABEL_IMPLNULL) {
               lab->binding = get_free_local_label();
               if (!lab->binding) {
                       fatalp("Label pool depleted\n");
                       return LDP_E_TOO_MANY_LABELS;
               }
               announce_label_change(lab);
       }

       warnp("[mpls_add_label] Adding %s/%d as local binding %d (%d), label %d"
           " to peer %s\n", satos(&lab->so_dest.sa), prefixlen, lab->binding,
           oldbinding, lab->label, p_str);

       /* Add switching route */
       if ((so_dest = make_mpls_union(lab->binding)) == NULL)
               return LDP_E_MEMORY;
       if ((so_tag = make_mpls_union(lab->label)) == NULL) {
               free(so_dest);
               fatalp("Out of memory\n");
               return LDP_E_MEMORY;
       }
       if ((so_nexthop = malloc(lab->so_gate.sa.sa_len)) == NULL) {
               free(so_dest);
               free(so_tag);
               fatalp("Out of memory\n");
               return LDP_E_MEMORY;
       }
       memcpy(so_nexthop, &lab->so_gate, lab->so_gate.sa.sa_len);

       if (add_route(so_dest, NULL, so_nexthop, NULL, so_tag, FREESO,
           oldbinding == MPLS_LABEL_IMPLNULL ? RTM_ADD : RTM_CHANGE)
           != LDP_E_OK) {
               fatalp("[mpls_add_label] MPLS route error\n");
               return LDP_E_ROUTE_ERROR;
       }

       /* Now, let's add tag to IP route and point it to mpls interface */

       if (getsockname(lab->p->socket, &so_ifa.sa,
           & (socklen_t) { sizeof(union sockunion) } )) {
               fatalp("[mpls_add_label]: getsockname\n");
               return LDP_E_ROUTE_ERROR;
       }

       if ((so_tag = make_mpls_union(lab->label)) == NULL) {
               fatalp("Out of memory\n");
               return LDP_E_MEMORY;
       }

       if (add_route(&lab->so_dest, lab->host ? NULL : &lab->so_pref,
           &lab->so_gate, &so_ifa, so_tag, NO_FREESO, RTM_CHANGE) != LDP_E_OK){
               free(so_tag);
               fatalp("[mpls_add_label]: INET route failure\n");
               return LDP_E_ROUTE_ERROR;
       }
       free(so_tag);

       warnp("[mpls_add_label]: SUCCESS\n");

       return LDP_E_OK;
}

int
mpls_add_ldp_peer(const struct ldp_peer * p)
{
       return LDP_E_OK;
}

int
mpls_delete_ldp_peer(const struct ldp_peer * p)
{

       /* Reput all the routes also to IPv4 */
       label_reattach_all_peer_labels(p, REATT_INET_CHANGE);

       return LDP_E_OK;
}

int
mpls_start_ldp()
{
       ldp_peer_init();
       label_init();

       return LDP_E_OK;
}