// SPDX-License-Identifier: GPL-2.0-or-later

/*

 * Zebra Layer-2 interface handling code

 * Copyright (C) 2016, 2017 Cumulus Networks, Inc.

 */

#include <zebra.h>

#include "if.h"

#include "prefix.h"

#include "table.h"

#include "memory.h"

#include "log.h"

#include "linklist.h"

#include "stream.h"

#include "hash.h"

#include "jhash.h"

#include "zebra/rib.h"

#include "zebra/rt.h"

#include "zebra/zebra_ns.h"

#include "zebra/zserv.h"

#include "zebra/debug.h"

#include "zebra/interface.h"

#include "zebra/zebra_vrf.h"

#include "zebra/rt_netlink.h"

#include "zebra/interface.h"

#include "zebra/zebra_l2.h"

#include "zebra/zebra_l2_bridge_if.h"

#include "zebra/zebra_vxlan.h"

#include "zebra/zebra_vxlan_if.h"

#include "zebra/zebra_evpn_mh.h"

/* definitions */

/* static function declarations */

/* Private functions */

static void map_slaves_to_bridge(struct interface *br_if, int link,

         bool update_slave, uint8_t chgflags)

{

  struct vrf *vrf;

  struct interface *ifp;

  struct zebra_vrf *zvrf;

  struct zebra_ns *zns;

  zvrf = br_if->vrf->info;

  assert(zvrf);

  zns = zvrf->zns;

  assert(zns);

  RB_FOREACH (vrf, vrf_name_head, &vrfs_by_name) {

    FOR_ALL_INTERFACES (vrf, ifp) {

      struct zebra_if *zif;

      struct zebra_l2info_brslave *br_slave;

      if (ifp->ifindex == IFINDEX_INTERNAL || !ifp->info)

        continue;

      if (!IS_ZEBRA_IF_BRIDGE_SLAVE(ifp))

        continue;

      /* NOTE: This assumes 'zebra_l2info_brslave' is the

       * first field

       * for any L2 interface.

       */

      zif = (struct zebra_if *)ifp->info;

      br_slave = &zif->brslave_info;

      if (link) {

        if (br_slave->bridge_ifindex == br_if->ifindex

            && br_slave->ns_id == zns->ns_id) {

          br_slave->br_if = br_if;

          if (update_slave) {

            zebra_l2if_update_bridge_slave(

              ifp,

              br_slave->bridge_ifindex,

              br_slave->ns_id,

              chgflags);

          }

        }

      } else {

        if (br_slave->br_if == br_if)

          br_slave->br_if = NULL;

      }

    }

  }

}

/* Public functions */

void zebra_l2_map_slave_to_bridge(struct zebra_l2info_brslave *br_slave,

          struct zebra_ns *zns)

{

  struct interface *br_if;

  /* TODO: Handle change of master */

  assert(zns);

  br_if = if_lookup_by_index_per_ns(zebra_ns_lookup(zns->ns_id),

            br_slave->bridge_ifindex);

  if (br_if)

    br_slave->br_if = br_if;

}

void zebra_l2_unmap_slave_from_bridge(struct zebra_l2info_brslave *br_slave)

{

  br_slave->br_if = NULL;

}

/* If any of the bond members are in bypass state the bond is placed

 * in bypass state

 */

static void zebra_l2_bond_lacp_bypass_eval(struct zebra_if *bond_zif)

{

  struct listnode *node;

  struct zebra_if *bond_mbr;

  bool old_bypass = !!(bond_zif->flags & ZIF_FLAG_LACP_BYPASS);

  bool new_bypass = false;

  if (bond_zif->bond_info.mbr_zifs) {

    for (ALL_LIST_ELEMENTS_RO(bond_zif->bond_info.mbr_zifs, node,

            bond_mbr)) {

      if (bond_mbr->flags & ZIF_FLAG_LACP_BYPASS) {

        new_bypass = true;

        break;

      }

    }

  }

  if (old_bypass == new_bypass)

    return;

  if (IS_ZEBRA_DEBUG_EVPN_MH_ES || IS_ZEBRA_DEBUG_EVENT)

    zlog_debug("bond %s lacp bypass changed to %s",

         bond_zif->ifp->name, new_bypass ? "on" : "off");

  if (new_bypass)

    bond_zif->flags |= ZIF_FLAG_LACP_BYPASS;

  else

    bond_zif->flags &= ~ZIF_FLAG_LACP_BYPASS;

  if (bond_zif->es_info.es)

    zebra_evpn_es_bypass_update(bond_zif->es_info.es, bond_zif->ifp,

              new_bypass);

}

/* Returns true if member was newly linked to bond */

void zebra_l2_map_slave_to_bond(struct zebra_if *zif, vrf_id_t vrf_id)

{

  struct interface *bond_if;

  struct zebra_if *bond_zif;

  struct zebra_l2info_bondslave *bond_slave = &zif->bondslave_info;

  bond_if = if_lookup_by_index(bond_slave->bond_ifindex, vrf_id);

  if (bond_if == bond_slave->bond_if)

    return;

  /* unlink the slave from the old master */

  zebra_l2_unmap_slave_from_bond(zif);

  /* If the bond is present and ready link the bond-member

   * to it

   */

  if (bond_if && (bond_zif = bond_if->info)) {

    if (bond_zif->bond_info.mbr_zifs) {

      if (IS_ZEBRA_DEBUG_EVPN_MH_ES || IS_ZEBRA_DEBUG_EVENT)

        zlog_debug("bond mbr %s linked to %s",

             zif->ifp->name, bond_if->name);

      bond_slave->bond_if = bond_if;

      /* link the slave to the new bond master */

      listnode_add(bond_zif->bond_info.mbr_zifs, zif);

      /* inherit protodown flags from the es-bond */

      if (zebra_evpn_is_es_bond(bond_if))

        zebra_evpn_mh_update_protodown_bond_mbr(

          zif, false /*clear*/, __func__);

      zebra_l2_bond_lacp_bypass_eval(bond_zif);

    }

  } else {

    if (IS_ZEBRA_DEBUG_EVPN_MH_ES || IS_ZEBRA_DEBUG_EVENT)

      zlog_debug("bond mbr %s link to bond skipped",

           zif->ifp->name);

  }

}

void zebra_l2_unmap_slave_from_bond(struct zebra_if *zif)

{

  struct zebra_l2info_bondslave *bond_slave = &zif->bondslave_info;

  struct zebra_if *bond_zif;

  if (!bond_slave->bond_if) {

    if (IS_ZEBRA_DEBUG_EVPN_MH_ES || IS_ZEBRA_DEBUG_EVENT)

      zlog_debug("bond mbr %s unlink from bond skipped",

           zif->ifp->name);

    return;

  }

  if (IS_ZEBRA_DEBUG_EVPN_MH_ES || IS_ZEBRA_DEBUG_EVENT)

    zlog_debug("bond mbr %s un-linked from %s", zif->ifp->name,

         bond_slave->bond_if->name);

  /* unlink the slave from the bond master */

  bond_zif = bond_slave->bond_if->info;

  /* clear protodown flags */

  if (zebra_evpn_is_es_bond(bond_zif->ifp))

    zebra_evpn_mh_update_protodown_bond_mbr(zif, true /*clear*/,

              __func__);

  listnode_delete(bond_zif->bond_info.mbr_zifs, zif);

  bond_slave->bond_if = NULL;

  zebra_l2_bond_lacp_bypass_eval(bond_zif);

}

void zebra_l2if_update_bond(struct interface *ifp, bool add)

{

  struct zebra_if *zif;

  struct zebra_l2info_bond *bond;

  zif = ifp->info;

  assert(zif);

  bond = &zif->bond_info;

  if (add) {

    if (!bond->mbr_zifs) {

      if (IS_ZEBRA_DEBUG_EVPN_MH_ES || IS_ZEBRA_DEBUG_EVENT)

        zlog_debug("bond %s mbr list create",

             ifp->name);

      bond->mbr_zifs = list_new();

    }

  } else {

    struct listnode *node;

    struct listnode *nnode;

    struct zebra_if *bond_mbr;

    if (!bond->mbr_zifs)

      return;

    if (IS_ZEBRA_DEBUG_EVPN_MH_ES || IS_ZEBRA_DEBUG_EVENT)

      zlog_debug("bond %s mbr list delete", ifp->name);

    for (ALL_LIST_ELEMENTS(bond->mbr_zifs, node, nnode, bond_mbr))

      zebra_l2_unmap_slave_from_bond(bond_mbr);

    list_delete(&bond->mbr_zifs);

  }

}

/*

 * Handle Bridge interface add or update. Update relevant info,

 * map slaves (if any) to the bridge.

 */

void zebra_l2_bridge_add_update(struct interface *ifp,

        struct zebra_l2info_bridge *bridge_info,

        int add)

{

  struct zebra_if *zif;

  struct zebra_l2_bridge_if *br;

  zif = ifp->info;

  assert(zif);

  br = BRIDGE_FROM_ZEBRA_IF(zif);

  br->vlan_aware = bridge_info->bridge.vlan_aware;

  zebra_l2_bridge_if_add(ifp);

  /* Link all slaves to this bridge */

  map_slaves_to_bridge(ifp, 1, false, ZEBRA_BRIDGE_NO_ACTION);

}

/*

 * Handle Bridge interface delete.

 */

void zebra_l2_bridge_del(struct interface *ifp)

{

  zebra_l2_bridge_if_del(ifp);

  /* Unlink all slaves to this bridge */

  map_slaves_to_bridge(ifp, 0, false, ZEBRA_BRIDGE_NO_ACTION);

}

void zebra_l2if_update_bridge(struct interface *ifp, uint8_t chgflags)

{

  if (!chgflags)

    return;

  map_slaves_to_bridge(ifp, 1, true, chgflags);

}

/*

 * Update L2 info for a VLAN interface. Only relevant parameter is the

 * VLAN Id and this cannot change.

 */

void zebra_l2_vlanif_update(struct interface *ifp,

          struct zebra_l2info_vlan *vlan_info)

{

  struct zebra_if *zif;

  zif = ifp->info;

  assert(zif);

  /* Copy over the L2 information. */

  memcpy(&zif->l2info.vl, vlan_info, sizeof(*vlan_info));

}

/*

 * Update L2 info for a GRE interface. This is called upon interface

 * addition as well as update. Upon add/update, need to inform

 * clients about GRE information.

 */

void zebra_l2_greif_add_update(struct interface *ifp,

             struct zebra_l2info_gre *gre_info, int add)

{

  struct zebra_if *zif;

  struct in_addr old_vtep_ip;

  zif = ifp->info;

  assert(zif);

  if (add) {

    memcpy(&zif->l2info.gre, gre_info, sizeof(*gre_info));

    return;

  }

  old_vtep_ip = zif->l2info.gre.vtep_ip;

  if (IPV4_ADDR_SAME(&old_vtep_ip, &gre_info->vtep_ip))

    return;

  zif->l2info.gre.vtep_ip = gre_info->vtep_ip;

}

/*

 * Update L2 info for a VxLAN interface. This is called upon interface

 * addition as well as update. Upon add, need to invoke the VNI create

 * function. Upon update, the params of interest are the local tunnel

 * IP and VLAN mapping, but the latter is handled separately.

 */

void zebra_l2_vxlanif_add_update(struct interface *ifp,

         struct zebra_l2info_vxlan *vxlan_info, int add)

{

  struct zebra_if *zif;

  uint16_t chgflags = 0;

  struct zebra_vxlan_if_update_ctx ctx;

  zif = ifp->info;

  assert(zif);

  if (add) {

    memcpy(&zif->l2info.vxl, vxlan_info, sizeof(*vxlan_info));

    zebra_vxlan_if_add(ifp);

    return;

  }

  memset(&ctx, 0, sizeof(ctx));

  ctx.old_vtep_ip = zif->l2info.vxl.vtep_ip;

  if (!IPV4_ADDR_SAME(&ctx.old_vtep_ip, &vxlan_info->vtep_ip)) {

    chgflags |= ZEBRA_VXLIF_LOCAL_IP_CHANGE;

    zif->l2info.vxl.vtep_ip = vxlan_info->vtep_ip;

  }

  if (IS_ZEBRA_VXLAN_IF_VNI(zif)) {

    ctx.old_vni = vxlan_info->vni_info.vni;

    if (!IPV4_ADDR_SAME(&zif->l2info.vxl.vni_info.vni.mcast_grp,

            &vxlan_info->vni_info.vni.mcast_grp)) {

      chgflags |= ZEBRA_VXLIF_MCAST_GRP_CHANGE;

      zif->l2info.vxl.vni_info.vni.mcast_grp =

        vxlan_info->vni_info.vni.mcast_grp;

    }

  }

  if (chgflags) {

    ctx.chgflags = chgflags;

    zebra_vxlan_if_update(ifp, &ctx);

  }

}

/*

 * Handle change to VLAN to VNI mapping.

 */

void zebra_l2_vxlanif_update_access_vlan(struct interface *ifp,

           vlanid_t access_vlan)

{

  struct zebra_if *zif;

  vlanid_t old_access_vlan;

  struct zebra_vxlan_vni *vni;

  struct zebra_vxlan_if_update_ctx ctx;

  zif = ifp->info;

  assert(zif);

  /* This would be called only in non svd case */

  assert(IS_ZEBRA_VXLAN_IF_VNI(zif));

  old_access_vlan = zif->l2info.vxl.vni_info.vni.access_vlan;

  ;

  if (old_access_vlan == access_vlan)

    return;

  memset(&ctx, 0, sizeof(ctx));

  vni = zebra_vxlan_if_vni_find(zif, 0);

  ctx.old_vni = *vni;

  ctx.chgflags = ZEBRA_VXLIF_VLAN_CHANGE;

  vni->access_vlan = access_vlan;

  zebra_evpn_vl_vxl_deref(old_access_vlan, vni->vni, zif);

  zebra_evpn_vl_vxl_ref(access_vlan, vni->vni, zif);

  zebra_vxlan_if_update(ifp, &ctx);

}

/*

 * Handle VxLAN interface delete.

 */

void zebra_l2_vxlanif_del(struct interface *ifp)

{

  struct zebra_if *zif;

  zif = ifp->info;

  assert(zif);

  zebra_vxlan_if_del(ifp);

}

/*

 * Map or unmap interface from bridge.

 * NOTE: It is currently assumped that an interface has to be unmapped

 * from a bridge before it can be mapped to another bridge.

 */

void zebra_l2if_update_bridge_slave(struct interface *ifp,

            ifindex_t bridge_ifindex, ns_id_t ns_id,

            uint8_t chgflags)

{

  struct zebra_if *zif;

  ifindex_t old_bridge_ifindex;

  ns_id_t old_ns_id;

  struct zebra_vrf *zvrf;

  struct zebra_vxlan_if_update_ctx ctx;

  memset(&ctx, 0, sizeof(ctx));

  zif = ifp->info;

  assert(zif);

  zvrf = ifp->vrf->info;

  if (!zvrf)

    return;

  if (zif->zif_type == ZEBRA_IF_VXLAN

      && chgflags != ZEBRA_BRIDGE_NO_ACTION) {

    if (chgflags & ZEBRA_BRIDGE_MASTER_MAC_CHANGE) {

      ctx.chgflags = ZEBRA_VXLIF_MASTER_MAC_CHANGE;

      zebra_vxlan_if_update(ifp, &ctx);

    }

    if (chgflags & ZEBRA_BRIDGE_MASTER_UP) {

      ctx.chgflags = ZEBRA_VXLIF_MASTER_CHANGE;

      zebra_vxlan_if_update(ifp, &ctx);

    }

  }

  old_bridge_ifindex = zif->brslave_info.bridge_ifindex;

  old_ns_id = zif->brslave_info.ns_id;

  if (old_bridge_ifindex == bridge_ifindex &&

      old_ns_id == zif->brslave_info.ns_id)

    return;

  ctx.chgflags = ZEBRA_VXLIF_MASTER_CHANGE;

  zif->brslave_info.ns_id = ns_id;

  zif->brslave_info.bridge_ifindex = bridge_ifindex;

  /* Set up or remove link with master */

  if (bridge_ifindex != IFINDEX_INTERNAL) {

    zebra_l2_map_slave_to_bridge(&zif->brslave_info, zvrf->zns);

    /* In the case of VxLAN, invoke the handler for EVPN. */

    if (zif->zif_type == ZEBRA_IF_VXLAN)

      zebra_vxlan_if_update(ifp, &ctx);

    if (zif->es_info.es)

      zebra_evpn_es_local_br_port_update(zif);

  } else if (old_bridge_ifindex != IFINDEX_INTERNAL) {

    /*

     * In the case of VxLAN, invoke the handler for EVPN.

     * Note that this should be done *prior*

     * to unmapping the interface from the bridge.

     */

    if (zif->zif_type == ZEBRA_IF_VXLAN)

      zebra_vxlan_if_update(ifp, &ctx);

    if (zif->es_info.es)

      zebra_evpn_es_local_br_port_update(zif);

    zebra_l2_unmap_slave_from_bridge(&zif->brslave_info);

  }

}

void zebra_l2if_update_bond_slave(struct interface *ifp, ifindex_t bond_ifindex,

          bool new_bypass)

{

  struct zebra_if *zif;

  ifindex_t old_bond_ifindex;

  bool old_bypass;

  struct zebra_l2info_bondslave *bond_mbr;

  zif = ifp->info;

  assert(zif);

  old_bypass = !!(zif->flags & ZIF_FLAG_LACP_BYPASS);

  if (old_bypass != new_bypass) {

    if (IS_ZEBRA_DEBUG_EVPN_MH_ES || IS_ZEBRA_DEBUG_EVENT)

      zlog_debug("bond-mbr %s lacp bypass changed to %s",

           zif->ifp->name, new_bypass ? "on" : "off");

    if (new_bypass)

      zif->flags |= ZIF_FLAG_LACP_BYPASS;

    else

      zif->flags &= ~ZIF_FLAG_LACP_BYPASS;

    bond_mbr = &zif->bondslave_info;

    if (bond_mbr->bond_if) {

      struct zebra_if *bond_zif = bond_mbr->bond_if->info;

      zebra_l2_bond_lacp_bypass_eval(bond_zif);

    }

  }

  old_bond_ifindex = zif->bondslave_info.bond_ifindex;

  if (old_bond_ifindex == bond_ifindex)

    return;

  zif->bondslave_info.bond_ifindex = bond_ifindex;

  /* Set up or remove link with master */

  if (bond_ifindex != IFINDEX_INTERNAL)

    zebra_l2_map_slave_to_bond(zif, ifp->vrf->vrf_id);

  else if (old_bond_ifindex != IFINDEX_INTERNAL)

    zebra_l2_unmap_slave_from_bond(zif);

}

void zebra_vlan_bitmap_compute(struct interface *ifp,

    uint32_t vid_start, uint16_t vid_end)

{

  uint32_t vid;

  struct zebra_if *zif;

  zif = (struct zebra_if *)ifp->info;

  assert(zif);

  for (vid = vid_start; vid <= vid_end; ++vid)

    bf_set_bit(zif->vlan_bitmap, vid);

}

void zebra_vlan_mbr_re_eval(struct interface *ifp, bitfield_t old_vlan_bitmap)

{

  uint32_t vid;

  struct zebra_if *zif;

  zif = (struct zebra_if *)ifp->info;

  assert(zif);

  if (!bf_cmp(zif->vlan_bitmap, old_vlan_bitmap))

    /* no change */

    return;

  bf_for_each_set_bit(zif->vlan_bitmap, vid, IF_VLAN_BITMAP_MAX) {

    /* if not already set create new reference */

    if (!bf_test_index(old_vlan_bitmap, vid))

      zebra_evpn_vl_mbr_ref(vid, zif);

    /* also clear from the old vlan bitmap */

    bf_release_index(old_vlan_bitmap, vid);

  }

  /* any bits remaining in the old vlan bitmap are stale references */

  bf_for_each_set_bit(old_vlan_bitmap, vid, IF_VLAN_BITMAP_MAX) {

    zebra_evpn_vl_mbr_deref(vid, zif);

  }

}