// SPDX-License-Identifier: BSD-3-Clause OR GPL-2.0
/* Copyright (c) 2017-2018 Mellanox Technologies. All rights reserved */

#include <net/ip_tunnels.h>
#include <net/ip6_tunnel.h>
#include <net/inet_ecn.h>

#include "spectrum_ipip.h"
#include "reg.h"

struct ip_tunnel_parm_kern
mlxsw_sp_ipip_netdev_parms4(const struct net_device *ol_dev)
{
 struct ip_tunnel *tun = netdev_priv(ol_dev);

 return tun->parms;
}

struct __ip6_tnl_parm
mlxsw_sp_ipip_netdev_parms6(const struct net_device *ol_dev)
{
 struct ip6_tnl *tun = netdev_priv(ol_dev);

 return tun->parms;
}

static bool
mlxsw_sp_ipip_parms4_has_ikey(const struct ip_tunnel_parm_kern *parms)
{
 return test_bit(IP_TUNNEL_KEY_BIT, parms->i_flags);
}

static bool mlxsw_sp_ipip_parms6_has_ikey(const struct __ip6_tnl_parm *parms)
{
 return test_bit(IP_TUNNEL_KEY_BIT, parms->i_flags);
}

static bool
mlxsw_sp_ipip_parms4_has_okey(const struct ip_tunnel_parm_kern *parms)
{
 return test_bit(IP_TUNNEL_KEY_BIT, parms->o_flags);
}

static bool mlxsw_sp_ipip_parms6_has_okey(const struct __ip6_tnl_parm *parms)
{
 return test_bit(IP_TUNNEL_KEY_BIT, parms->o_flags);
}

static u32 mlxsw_sp_ipip_parms4_ikey(const struct ip_tunnel_parm_kern *parms)
{
 return mlxsw_sp_ipip_parms4_has_ikey(parms) ?
  be32_to_cpu(parms->i_key) : 0;
}

static u32 mlxsw_sp_ipip_parms6_ikey(const struct __ip6_tnl_parm *parms)
{
 return mlxsw_sp_ipip_parms6_has_ikey(parms) ?
  be32_to_cpu(parms->i_key) : 0;
}

static u32 mlxsw_sp_ipip_parms4_okey(const struct ip_tunnel_parm_kern *parms)
{
 return mlxsw_sp_ipip_parms4_has_okey(parms) ?
  be32_to_cpu(parms->o_key) : 0;
}

static u32 mlxsw_sp_ipip_parms6_okey(const struct __ip6_tnl_parm *parms)
{
 return mlxsw_sp_ipip_parms6_has_okey(parms) ?
  be32_to_cpu(parms->o_key) : 0;
}

static union mlxsw_sp_l3addr
mlxsw_sp_ipip_parms4_saddr(const struct ip_tunnel_parm_kern *parms)
{
 return (union mlxsw_sp_l3addr) { .addr4 = parms->iph.saddr };
}

static union mlxsw_sp_l3addr
mlxsw_sp_ipip_parms6_saddr(const struct __ip6_tnl_parm *parms)
{
 return (union mlxsw_sp_l3addr) { .addr6 = parms->laddr };
}

static union mlxsw_sp_l3addr
mlxsw_sp_ipip_parms4_daddr(const struct ip_tunnel_parm_kern *parms)
{
 return (union mlxsw_sp_l3addr) { .addr4 = parms->iph.daddr };
}

static union mlxsw_sp_l3addr
mlxsw_sp_ipip_parms6_daddr(const struct __ip6_tnl_parm *parms)
{
 return (union mlxsw_sp_l3addr) { .addr6 = parms->raddr };
}

union mlxsw_sp_l3addr
mlxsw_sp_ipip_netdev_saddr(enum mlxsw_sp_l3proto proto,
      const struct net_device *ol_dev)
{
 struct ip_tunnel_parm_kern parms4;
 struct __ip6_tnl_parm parms6;

 switch (proto) {
 case MLXSW_SP_L3_PROTO_IPV4:
  parms4 = mlxsw_sp_ipip_netdev_parms4(ol_dev);
  return mlxsw_sp_ipip_parms4_saddr(&parms4);
 case MLXSW_SP_L3_PROTO_IPV6:
  parms6 = mlxsw_sp_ipip_netdev_parms6(ol_dev);
  return mlxsw_sp_ipip_parms6_saddr(&parms6);
 }

 WARN_ON(1);
 return (union mlxsw_sp_l3addr) {0};
}

static __be32 mlxsw_sp_ipip_netdev_daddr4(const struct net_device *ol_dev)
{

 struct ip_tunnel_parm_kern parms4;

 parms4 = mlxsw_sp_ipip_netdev_parms4(ol_dev);

 return mlxsw_sp_ipip_parms4_daddr(&parms4).addr4;
}

static union mlxsw_sp_l3addr
mlxsw_sp_ipip_netdev_daddr(enum mlxsw_sp_l3proto proto,
      const struct net_device *ol_dev)
{
 struct ip_tunnel_parm_kern parms4;
 struct __ip6_tnl_parm parms6;

 switch (proto) {
 case MLXSW_SP_L3_PROTO_IPV4:
  parms4 = mlxsw_sp_ipip_netdev_parms4(ol_dev);
  return mlxsw_sp_ipip_parms4_daddr(&parms4);
 case MLXSW_SP_L3_PROTO_IPV6:
  parms6 = mlxsw_sp_ipip_netdev_parms6(ol_dev);
  return mlxsw_sp_ipip_parms6_daddr(&parms6);
 }

 WARN_ON(1);
 return (union mlxsw_sp_l3addr) {0};
}

bool mlxsw_sp_l3addr_is_zero(union mlxsw_sp_l3addr addr)
{
 union mlxsw_sp_l3addr naddr = {0};

 return !memcmp(&addr, &naddr, sizeof(naddr));
}

static struct mlxsw_sp_ipip_parms
mlxsw_sp_ipip_netdev_parms_init_gre4(const struct net_device *ol_dev)
{
 struct ip_tunnel_parm_kern parms = mlxsw_sp_ipip_netdev_parms4(ol_dev);

 return (struct mlxsw_sp_ipip_parms) {
  .proto = MLXSW_SP_L3_PROTO_IPV4,
  .saddr = mlxsw_sp_ipip_parms4_saddr(&parms),
  .daddr = mlxsw_sp_ipip_parms4_daddr(&parms),
  .link = parms.link,
  .ikey = mlxsw_sp_ipip_parms4_ikey(&parms),
  .okey = mlxsw_sp_ipip_parms4_okey(&parms),
 };
}

static int
mlxsw_sp_ipip_nexthop_update_gre4(struct mlxsw_sp *mlxsw_sp, u32 adj_index,
      struct mlxsw_sp_ipip_entry *ipip_entry,
      bool force, char *ratr_pl)
{
 u16 rif_index = mlxsw_sp_ipip_lb_rif_index(ipip_entry->ol_lb);
 __be32 daddr4 = mlxsw_sp_ipip_netdev_daddr4(ipip_entry->ol_dev);
 enum mlxsw_reg_ratr_op op;

 op = force ? MLXSW_REG_RATR_OP_WRITE_WRITE_ENTRY :
       MLXSW_REG_RATR_OP_WRITE_WRITE_ENTRY_ON_ACTIVITY;
 mlxsw_reg_ratr_pack(ratr_pl, op, true, MLXSW_REG_RATR_TYPE_IPIP,
       adj_index, rif_index);
 mlxsw_reg_ratr_ipip4_entry_pack(ratr_pl, be32_to_cpu(daddr4));

 return mlxsw_reg_write(mlxsw_sp->core, MLXSW_REG(ratr), ratr_pl);
}

static int
mlxsw_sp_ipip_decap_config_gre4(struct mlxsw_sp *mlxsw_sp,
    struct mlxsw_sp_ipip_entry *ipip_entry,
    u32 tunnel_index)
{
 u16 rif_index = mlxsw_sp_ipip_lb_rif_index(ipip_entry->ol_lb);
 u16 ul_rif_id = mlxsw_sp_ipip_lb_ul_rif_id(ipip_entry->ol_lb);
 struct ip_tunnel_parm_kern parms;
 char rtdp_pl[MLXSW_REG_RTDP_LEN];
 unsigned int type_check;
 bool has_ikey;
 u32 daddr4;
 u32 ikey;

 parms = mlxsw_sp_ipip_netdev_parms4(ipip_entry->ol_dev);
 has_ikey = mlxsw_sp_ipip_parms4_has_ikey(&parms);
 ikey = mlxsw_sp_ipip_parms4_ikey(&parms);

 mlxsw_reg_rtdp_pack(rtdp_pl, MLXSW_REG_RTDP_TYPE_IPIP, tunnel_index);
 mlxsw_reg_rtdp_egress_router_interface_set(rtdp_pl, ul_rif_id);

 type_check = has_ikey ?
  MLXSW_REG_RTDP_IPIP_TYPE_CHECK_ALLOW_GRE_KEY :
  MLXSW_REG_RTDP_IPIP_TYPE_CHECK_ALLOW_GRE;

 /* Linux demuxes tunnels based on packet SIP (which must match tunnel
  * remote IP). Thus configure decap so that it filters out packets that
  * are not IPv4 or have the wrong SIP. IPIP_DECAP_ERROR trap is
  * generated for packets that fail this criterion. Linux then handles
  * such packets in slow path and generates ICMP destination unreachable.
 */
 daddr4 = be32_to_cpu(mlxsw_sp_ipip_netdev_daddr4(ipip_entry->ol_dev));
 mlxsw_reg_rtdp_ipip4_pack(rtdp_pl, rif_index,
      MLXSW_REG_RTDP_IPIP_SIP_CHECK_FILTER_IPV4,
      type_check, has_ikey, daddr4, ikey);

 return mlxsw_reg_write(mlxsw_sp->core, MLXSW_REG(rtdp), rtdp_pl);
}

static bool mlxsw_sp_ipip_tunnel_complete(enum mlxsw_sp_l3proto proto,
       const struct net_device *ol_dev)
{
 union mlxsw_sp_l3addr saddr = mlxsw_sp_ipip_netdev_saddr(proto, ol_dev);
 union mlxsw_sp_l3addr daddr = mlxsw_sp_ipip_netdev_daddr(proto, ol_dev);

 /* Tunnels with unset local or remote address are valid in Linux and
  * used for lightweight tunnels (LWT) and Non-Broadcast Multi-Access
  * (NBMA) tunnels. In principle these can be offloaded, but the driver
  * currently doesn't support this. So punt.
 */
 return !mlxsw_sp_l3addr_is_zero(saddr) &&
        !mlxsw_sp_l3addr_is_zero(daddr);
}

static bool mlxsw_sp_ipip_can_offload_gre4(const struct mlxsw_sp *mlxsw_sp,
        const struct net_device *ol_dev)
{
 struct ip_tunnel *tunnel = netdev_priv(ol_dev);
 bool inherit_ttl = tunnel->parms.iph.ttl == 0;
 bool inherit_tos = tunnel->parms.iph.tos & 0x1;
 IP_TUNNEL_DECLARE_FLAGS(okflags) = { };

 /* We can't offload any other features. */
 __set_bit(IP_TUNNEL_KEY_BIT, okflags);

 return ip_tunnel_flags_subset(tunnel->parms.i_flags, okflags) &&
        ip_tunnel_flags_subset(tunnel->parms.o_flags, okflags) &&
        inherit_ttl && inherit_tos &&
        mlxsw_sp_ipip_tunnel_complete(MLXSW_SP_L3_PROTO_IPV4, ol_dev);
}

static struct mlxsw_sp_rif_ipip_lb_config
mlxsw_sp_ipip_ol_loopback_config_gre4(struct mlxsw_sp *mlxsw_sp,
          const struct net_device *ol_dev)
{
 struct ip_tunnel_parm_kern parms = mlxsw_sp_ipip_netdev_parms4(ol_dev);
 enum mlxsw_reg_ritr_loopback_ipip_type lb_ipipt;

 lb_ipipt = mlxsw_sp_ipip_parms4_has_okey(&parms) ?
  MLXSW_REG_RITR_LOOPBACK_IPIP_TYPE_IP_IN_GRE_KEY_IN_IP :
  MLXSW_REG_RITR_LOOPBACK_IPIP_TYPE_IP_IN_GRE_IN_IP;
 return (struct mlxsw_sp_rif_ipip_lb_config){
  .lb_ipipt = lb_ipipt,
  .okey = mlxsw_sp_ipip_parms4_okey(&parms),
  .ul_protocol = MLXSW_SP_L3_PROTO_IPV4,
  .saddr = mlxsw_sp_ipip_netdev_saddr(MLXSW_SP_L3_PROTO_IPV4,
          ol_dev),
 };
}

static int
mlxsw_sp_ipip_ol_netdev_change_gre(struct mlxsw_sp *mlxsw_sp,
       struct mlxsw_sp_ipip_entry *ipip_entry,
       const struct mlxsw_sp_ipip_parms *new_parms,
       struct netlink_ext_ack *extack)
{
 const struct mlxsw_sp_ipip_parms *old_parms = &ipip_entry->parms;
 bool update_tunnel = false;
 bool update_decap = false;
 bool update_nhs = false;
 int err = 0;

 if (!mlxsw_sp_l3addr_eq(&new_parms->saddr, &old_parms->saddr)) {
  u16 ul_tb_id = mlxsw_sp_ipip_dev_ul_tb_id(ipip_entry->ol_dev);

  /* Since the local address has changed, if there is another
   * tunnel with a matching saddr, both need to be demoted.
 */
  if (mlxsw_sp_ipip_demote_tunnel_by_saddr(mlxsw_sp,
        new_parms->proto,
        new_parms->saddr,
        ul_tb_id,
        ipip_entry)) {
   mlxsw_sp_ipip_entry_demote_tunnel(mlxsw_sp, ipip_entry);
   return 0;
  }

  update_tunnel = true;
 } else if (old_parms->okey != new_parms->okey ||
     old_parms->link != new_parms->link) {
  update_tunnel = true;
 } else if (!mlxsw_sp_l3addr_eq(&new_parms->daddr, &old_parms->daddr)) {
  update_nhs = true;
 } else if (old_parms->ikey != new_parms->ikey) {
  update_decap = true;
 }

 if (update_tunnel)
  err = __mlxsw_sp_ipip_entry_update_tunnel(mlxsw_sp, ipip_entry,
         true, true, true,
         extack);
 else if (update_nhs)
  err = __mlxsw_sp_ipip_entry_update_tunnel(mlxsw_sp, ipip_entry,
         false, false, true,
         extack);
 else if (update_decap)
  err = __mlxsw_sp_ipip_entry_update_tunnel(mlxsw_sp, ipip_entry,
         false, false, false,
         extack);
 if (err)
  return err;

 ipip_entry->parms = *new_parms;
 return 0;
}

static int
mlxsw_sp_ipip_ol_netdev_change_gre4(struct mlxsw_sp *mlxsw_sp,
        struct mlxsw_sp_ipip_entry *ipip_entry,
        struct netlink_ext_ack *extack)
{
 struct mlxsw_sp_ipip_parms new_parms;

 new_parms = mlxsw_sp_ipip_netdev_parms_init_gre4(ipip_entry->ol_dev);
 return mlxsw_sp_ipip_ol_netdev_change_gre(mlxsw_sp, ipip_entry,
        &new_parms, extack);
}

static int
mlxsw_sp_ipip_rem_addr_set_gre4(struct mlxsw_sp *mlxsw_sp,
    struct mlxsw_sp_ipip_entry *ipip_entry)
{
 return 0;
}

static void
mlxsw_sp_ipip_rem_addr_unset_gre4(struct mlxsw_sp *mlxsw_sp,
      const struct mlxsw_sp_ipip_entry *ipip_entry)
{
}

static const struct mlxsw_sp_ipip_ops mlxsw_sp_ipip_gre4_ops = {
 .dev_type = ARPHRD_IPGRE,
 .ul_proto = MLXSW_SP_L3_PROTO_IPV4,
 .inc_parsing_depth = false,
 .parms_init = mlxsw_sp_ipip_netdev_parms_init_gre4,
 .nexthop_update = mlxsw_sp_ipip_nexthop_update_gre4,
 .decap_config = mlxsw_sp_ipip_decap_config_gre4,
 .can_offload = mlxsw_sp_ipip_can_offload_gre4,
 .ol_loopback_config = mlxsw_sp_ipip_ol_loopback_config_gre4,
 .ol_netdev_change = mlxsw_sp_ipip_ol_netdev_change_gre4,
 .rem_ip_addr_set = mlxsw_sp_ipip_rem_addr_set_gre4,
 .rem_ip_addr_unset = mlxsw_sp_ipip_rem_addr_unset_gre4,
};

static struct mlxsw_sp_ipip_parms
mlxsw_sp_ipip_netdev_parms_init_gre6(const struct net_device *ol_dev)
{
 struct __ip6_tnl_parm parms = mlxsw_sp_ipip_netdev_parms6(ol_dev);

 return (struct mlxsw_sp_ipip_parms) {
  .proto = MLXSW_SP_L3_PROTO_IPV6,
  .saddr = mlxsw_sp_ipip_parms6_saddr(&parms),
  .daddr = mlxsw_sp_ipip_parms6_daddr(&parms),
  .link = parms.link,
  .ikey = mlxsw_sp_ipip_parms6_ikey(&parms),
  .okey = mlxsw_sp_ipip_parms6_okey(&parms),
 };
}

static int
mlxsw_sp_ipip_nexthop_update_gre6(struct mlxsw_sp *mlxsw_sp, u32 adj_index,
      struct mlxsw_sp_ipip_entry *ipip_entry,
      bool force, char *ratr_pl)
{
 u16 rif_index = mlxsw_sp_ipip_lb_rif_index(ipip_entry->ol_lb);
 enum mlxsw_reg_ratr_op op;

 op = force ? MLXSW_REG_RATR_OP_WRITE_WRITE_ENTRY :
       MLXSW_REG_RATR_OP_WRITE_WRITE_ENTRY_ON_ACTIVITY;
 mlxsw_reg_ratr_pack(ratr_pl, op, true, MLXSW_REG_RATR_TYPE_IPIP,
       adj_index, rif_index);
 mlxsw_reg_ratr_ipip6_entry_pack(ratr_pl,
     ipip_entry->dip_kvdl_index);

 return mlxsw_reg_write(mlxsw_sp->core, MLXSW_REG(ratr), ratr_pl);
}

static int
mlxsw_sp_ipip_decap_config_gre6(struct mlxsw_sp *mlxsw_sp,
    struct mlxsw_sp_ipip_entry *ipip_entry,
    u32 tunnel_index)
{
 u16 rif_index = mlxsw_sp_ipip_lb_rif_index(ipip_entry->ol_lb);
 u16 ul_rif_id = mlxsw_sp_ipip_lb_ul_rif_id(ipip_entry->ol_lb);
 char rtdp_pl[MLXSW_REG_RTDP_LEN];
 struct __ip6_tnl_parm parms;
 unsigned int type_check;
 bool has_ikey;
 u32 ikey;

 parms = mlxsw_sp_ipip_netdev_parms6(ipip_entry->ol_dev);
 has_ikey = mlxsw_sp_ipip_parms6_has_ikey(&parms);
 ikey = mlxsw_sp_ipip_parms6_ikey(&parms);

 mlxsw_reg_rtdp_pack(rtdp_pl, MLXSW_REG_RTDP_TYPE_IPIP, tunnel_index);
 mlxsw_reg_rtdp_egress_router_interface_set(rtdp_pl, ul_rif_id);

 type_check = has_ikey ?
  MLXSW_REG_RTDP_IPIP_TYPE_CHECK_ALLOW_GRE_KEY :
  MLXSW_REG_RTDP_IPIP_TYPE_CHECK_ALLOW_GRE;

 /* Linux demuxes tunnels based on packet SIP (which must match tunnel
  * remote IP). Thus configure decap so that it filters out packets that
  * are not IPv6 or have the wrong SIP. IPIP_DECAP_ERROR trap is
  * generated for packets that fail this criterion. Linux then handles
  * such packets in slow path and generates ICMP destination unreachable.
 */
 mlxsw_reg_rtdp_ipip6_pack(rtdp_pl, rif_index,
      MLXSW_REG_RTDP_IPIP_SIP_CHECK_FILTER_IPV6,
      type_check, has_ikey,
      ipip_entry->dip_kvdl_index, ikey);

 return mlxsw_reg_write(mlxsw_sp->core, MLXSW_REG(rtdp), rtdp_pl);
}

static bool mlxsw_sp_ipip_can_offload_gre6(const struct mlxsw_sp *mlxsw_sp,
        const struct net_device *ol_dev)
{
 struct __ip6_tnl_parm tparm = mlxsw_sp_ipip_netdev_parms6(ol_dev);
 bool inherit_tos = tparm.flags & IP6_TNL_F_USE_ORIG_TCLASS;
 bool inherit_ttl = tparm.hop_limit == 0;
 IP_TUNNEL_DECLARE_FLAGS(okflags) = { };

 /* We can't offload any other features. */
 __set_bit(IP_TUNNEL_KEY_BIT, okflags);

 return ip_tunnel_flags_subset(tparm.i_flags, okflags) &&
        ip_tunnel_flags_subset(tparm.o_flags, okflags) &&
        inherit_ttl && inherit_tos &&
        mlxsw_sp_ipip_tunnel_complete(MLXSW_SP_L3_PROTO_IPV6, ol_dev);
}

static struct mlxsw_sp_rif_ipip_lb_config
mlxsw_sp_ipip_ol_loopback_config_gre6(struct mlxsw_sp *mlxsw_sp,
          const struct net_device *ol_dev)
{
 struct __ip6_tnl_parm parms = mlxsw_sp_ipip_netdev_parms6(ol_dev);
 enum mlxsw_reg_ritr_loopback_ipip_type lb_ipipt;

 lb_ipipt = mlxsw_sp_ipip_parms6_has_okey(&parms) ?
  MLXSW_REG_RITR_LOOPBACK_IPIP_TYPE_IP_IN_GRE_KEY_IN_IP :
  MLXSW_REG_RITR_LOOPBACK_IPIP_TYPE_IP_IN_GRE_IN_IP;
 return (struct mlxsw_sp_rif_ipip_lb_config){
  .lb_ipipt = lb_ipipt,
  .okey = mlxsw_sp_ipip_parms6_okey(&parms),
  .ul_protocol = MLXSW_SP_L3_PROTO_IPV6,
  .saddr = mlxsw_sp_ipip_netdev_saddr(MLXSW_SP_L3_PROTO_IPV6,
          ol_dev),
 };
}

static int
mlxsw_sp_ipip_ol_netdev_change_gre6(struct mlxsw_sp *mlxsw_sp,
        struct mlxsw_sp_ipip_entry *ipip_entry,
        struct netlink_ext_ack *extack)
{
 u32 new_kvdl_index, old_kvdl_index = ipip_entry->dip_kvdl_index;
 struct in6_addr old_addr6 = ipip_entry->parms.daddr.addr6;
 struct mlxsw_sp_ipip_parms new_parms;
 int err;

 new_parms = mlxsw_sp_ipip_netdev_parms_init_gre6(ipip_entry->ol_dev);

 err = mlxsw_sp_ipv6_addr_kvdl_index_get(mlxsw_sp,
      &new_parms.daddr.addr6,
      &new_kvdl_index);
 if (err)
  return err;
 ipip_entry->dip_kvdl_index = new_kvdl_index;

 err = mlxsw_sp_ipip_ol_netdev_change_gre(mlxsw_sp, ipip_entry,
       &new_parms, extack);
 if (err)
  goto err_change_gre;

 mlxsw_sp_ipv6_addr_put(mlxsw_sp, &old_addr6);

 return 0;

err_change_gre:
 ipip_entry->dip_kvdl_index = old_kvdl_index;
 mlxsw_sp_ipv6_addr_put(mlxsw_sp, &new_parms.daddr.addr6);
 return err;
}

static int
mlxsw_sp_ipip_rem_addr_set_gre6(struct mlxsw_sp *mlxsw_sp,
    struct mlxsw_sp_ipip_entry *ipip_entry)
{
 return mlxsw_sp_ipv6_addr_kvdl_index_get(mlxsw_sp,
       &ipip_entry->parms.daddr.addr6,
       &ipip_entry->dip_kvdl_index);
}

static void
mlxsw_sp_ipip_rem_addr_unset_gre6(struct mlxsw_sp *mlxsw_sp,
      const struct mlxsw_sp_ipip_entry *ipip_entry)
{
 mlxsw_sp_ipv6_addr_put(mlxsw_sp, &ipip_entry->parms.daddr.addr6);
}

static const struct mlxsw_sp_ipip_ops mlxsw_sp1_ipip_gre6_ops = {
 .dev_type = ARPHRD_IP6GRE,
 .ul_proto = MLXSW_SP_L3_PROTO_IPV6,
 .inc_parsing_depth = true,
 .double_rif_entry = true,
 .parms_init = mlxsw_sp_ipip_netdev_parms_init_gre6,
 .nexthop_update = mlxsw_sp_ipip_nexthop_update_gre6,
 .decap_config = mlxsw_sp_ipip_decap_config_gre6,
 .can_offload = mlxsw_sp_ipip_can_offload_gre6,
 .ol_loopback_config = mlxsw_sp_ipip_ol_loopback_config_gre6,
 .ol_netdev_change = mlxsw_sp_ipip_ol_netdev_change_gre6,
 .rem_ip_addr_set = mlxsw_sp_ipip_rem_addr_set_gre6,
 .rem_ip_addr_unset = mlxsw_sp_ipip_rem_addr_unset_gre6,
};

const struct mlxsw_sp_ipip_ops *mlxsw_sp1_ipip_ops_arr[] = {
 [MLXSW_SP_IPIP_TYPE_GRE4] = &mlxsw_sp_ipip_gre4_ops,
 [MLXSW_SP_IPIP_TYPE_GRE6] = &mlxsw_sp1_ipip_gre6_ops,
};

static const struct mlxsw_sp_ipip_ops mlxsw_sp2_ipip_gre6_ops = {
 .dev_type = ARPHRD_IP6GRE,
 .ul_proto = MLXSW_SP_L3_PROTO_IPV6,
 .inc_parsing_depth = true,
 .parms_init = mlxsw_sp_ipip_netdev_parms_init_gre6,
 .nexthop_update = mlxsw_sp_ipip_nexthop_update_gre6,
 .decap_config = mlxsw_sp_ipip_decap_config_gre6,
 .can_offload = mlxsw_sp_ipip_can_offload_gre6,
 .ol_loopback_config = mlxsw_sp_ipip_ol_loopback_config_gre6,
 .ol_netdev_change = mlxsw_sp_ipip_ol_netdev_change_gre6,
 .rem_ip_addr_set = mlxsw_sp_ipip_rem_addr_set_gre6,
 .rem_ip_addr_unset = mlxsw_sp_ipip_rem_addr_unset_gre6,
};

const struct mlxsw_sp_ipip_ops *mlxsw_sp2_ipip_ops_arr[] = {
 [MLXSW_SP_IPIP_TYPE_GRE4] = &mlxsw_sp_ipip_gre4_ops,
 [MLXSW_SP_IPIP_TYPE_GRE6] = &mlxsw_sp2_ipip_gre6_ops,
};

static int mlxsw_sp_ipip_ecn_encap_init_one(struct mlxsw_sp *mlxsw_sp,
         u8 inner_ecn, u8 outer_ecn)
{
 char tieem_pl[MLXSW_REG_TIEEM_LEN];

 mlxsw_reg_tieem_pack(tieem_pl, inner_ecn, outer_ecn);
 return mlxsw_reg_write(mlxsw_sp->core, MLXSW_REG(tieem), tieem_pl);
}

int mlxsw_sp_ipip_ecn_encap_init(struct mlxsw_sp *mlxsw_sp)
{
 int i;

 /* Iterate over inner ECN values */
 for (i = INET_ECN_NOT_ECT; i <= INET_ECN_CE; i++) {
  u8 outer_ecn = INET_ECN_encapsulate(0, i);
  int err;

  err = mlxsw_sp_ipip_ecn_encap_init_one(mlxsw_sp, i, outer_ecn);
  if (err)
   return err;
 }

 return 0;
}

static int mlxsw_sp_ipip_ecn_decap_init_one(struct mlxsw_sp *mlxsw_sp,
         u8 inner_ecn, u8 outer_ecn)
{
 char tidem_pl[MLXSW_REG_TIDEM_LEN];
 u8 new_inner_ecn;
 bool trap_en;

 new_inner_ecn = mlxsw_sp_tunnel_ecn_decap(outer_ecn, inner_ecn,
        &trap_en);
 mlxsw_reg_tidem_pack(tidem_pl, outer_ecn, inner_ecn, new_inner_ecn,
        trap_en, trap_en ? MLXSW_TRAP_ID_DECAP_ECN0 : 0);
 return mlxsw_reg_write(mlxsw_sp->core, MLXSW_REG(tidem), tidem_pl);
}

int mlxsw_sp_ipip_ecn_decap_init(struct mlxsw_sp *mlxsw_sp)
{
 int i, j, err;

 /* Iterate over inner ECN values */
 for (i = INET_ECN_NOT_ECT; i <= INET_ECN_CE; i++) {
  /* Iterate over outer ECN values */
  for (j = INET_ECN_NOT_ECT; j <= INET_ECN_CE; j++) {
   err = mlxsw_sp_ipip_ecn_decap_init_one(mlxsw_sp, i, j);
   if (err)
    return err;
  }
 }

 return 0;
}

struct net_device *
mlxsw_sp_ipip_netdev_ul_dev_get(const struct net_device *ol_dev)
{
 struct net *net = dev_net(ol_dev);
 struct ip_tunnel *tun4;
 struct ip6_tnl *tun6;

 switch (ol_dev->type) {
 case ARPHRD_IPGRE:
  tun4 = netdev_priv(ol_dev);
  return dev_get_by_index_rcu(net, tun4->parms.link);
 case ARPHRD_IP6GRE:
  tun6 = netdev_priv(ol_dev);
  return dev_get_by_index_rcu(net, tun6->parms.link);
 default:
  return NULL;
 }
}