#include <linux/aer.h>
 #include <linux/if_bridge.h>
 #include <net/busy_poll.h>
+#include <net/vxlan.h>
 
 MODULE_VERSION(DRV_VER);
 MODULE_DEVICE_TABLE(pci, be_dev_ids);
        return vlan_tag;
 }
 
+/* Used only for IP tunnel packets */
+static u16 skb_inner_ip_proto(struct sk_buff *skb)
+{
+       return (inner_ip_hdr(skb)->version == 4) ?
+               inner_ip_hdr(skb)->protocol : inner_ipv6_hdr(skb)->nexthdr;
+}
+
+static u16 skb_ip_proto(struct sk_buff *skb)
+{
+       return (ip_hdr(skb)->version == 4) ?
+               ip_hdr(skb)->protocol : ipv6_hdr(skb)->nexthdr;
+}
+
 static void wrb_fill_hdr(struct be_adapter *adapter, struct be_eth_hdr_wrb *hdr,
                struct sk_buff *skb, u32 wrb_cnt, u32 len, bool skip_hw_vlan)
 {
-       u16 vlan_tag;
+       u16 vlan_tag, proto;
 
        memset(hdr, 0, sizeof(*hdr));
 
                if (skb_is_gso_v6(skb) && !lancer_chip(adapter))
                        AMAP_SET_BITS(struct amap_eth_hdr_wrb, lso6, hdr, 1);
        } else if (skb->ip_summed == CHECKSUM_PARTIAL) {
-               if (is_tcp_pkt(skb))
+               if (skb->encapsulation) {
+                       AMAP_SET_BITS(struct amap_eth_hdr_wrb, ipcs, hdr, 1);
+                       proto = skb_inner_ip_proto(skb);
+               } else {
+                       proto = skb_ip_proto(skb);
+               }
+               if (proto == IPPROTO_TCP)
                        AMAP_SET_BITS(struct amap_eth_hdr_wrb, tcpcs, hdr, 1);
-               else if (is_udp_pkt(skb))
+               else if (proto == IPPROTO_UDP)
                        AMAP_SET_BITS(struct amap_eth_hdr_wrb, udpcs, hdr, 1);
        }
 
 static inline bool csum_passed(struct be_rx_compl_info *rxcp)
 {
        /* L4 checksum is not reliable for non TCP/UDP packets.
-        * Also ignore ipcksm for ipv6 pkts */
+        * Also ignore ipcksm for ipv6 pkts
+        */
        return (rxcp->tcpf || rxcp->udpf) && rxcp->l4_csum &&
-                               (rxcp->ip_csum || rxcp->ipv6);
+               (rxcp->ip_csum || rxcp->ipv6) && !rxcp->err;
 }
 
 static struct be_rx_page_info *get_rx_page_info(struct be_rx_obj *rxo)
        skb_record_rx_queue(skb, rxo - &adapter->rx_obj[0]);
        if (netdev->features & NETIF_F_RXHASH)
                skb_set_hash(skb, rxcp->rss_hash, PKT_HASH_TYPE_L3);
+
+       skb->encapsulation = rxcp->tunneled;
        skb_mark_napi_id(skb, napi);
 
        if (rxcp->vlanf)
        skb_record_rx_queue(skb, rxo - &adapter->rx_obj[0]);
        if (adapter->netdev->features & NETIF_F_RXHASH)
                skb_set_hash(skb, rxcp->rss_hash, PKT_HASH_TYPE_L3);
+
+       skb->encapsulation = rxcp->tunneled;
        skb_mark_napi_id(skb, napi);
 
        if (rxcp->vlanf)
                                               compl);
        }
        rxcp->port = AMAP_GET_BITS(struct amap_eth_rx_compl_v1, port, compl);
+       rxcp->tunneled =
+               AMAP_GET_BITS(struct amap_eth_rx_compl_v1, tunneled, compl);
 }
 
 static void be_parse_rx_compl_v0(struct be_eth_rx_compl *compl,
 
        netif_tx_start_all_queues(netdev);
        be_roce_dev_open(adapter);
+
+       if (skyhawk_chip(adapter))
+               vxlan_get_rx_port(netdev);
        return 0;
 err:
        be_close(adapter->netdev);
        }
 }
 
+static void be_disable_vxlan_offloads(struct be_adapter *adapter)
+{
+       if (adapter->flags & BE_FLAGS_VXLAN_OFFLOADS)
+               be_cmd_manage_iface(adapter, adapter->if_handle,
+                                   OP_CONVERT_TUNNEL_TO_NORMAL);
+
+       if (adapter->vxlan_port)
+               be_cmd_set_vxlan_port(adapter, 0);
+
+       adapter->flags &= ~BE_FLAGS_VXLAN_OFFLOADS;
+       adapter->vxlan_port = 0;
+}
+
 static int be_clear(struct be_adapter *adapter)
 {
        be_cancel_worker(adapter);
        if (sriov_enabled(adapter))
                be_vf_clear(adapter);
 
+       be_disable_vxlan_offloads(adapter);
+
        /* delete the primary mac along with the uc-mac list */
        be_mac_clear(adapter);
 
                                       BRIDGE_MODE_VEPA : BRIDGE_MODE_VEB);
 }
 
+static void be_add_vxlan_port(struct net_device *netdev, sa_family_t sa_family,
+                             __be16 port)
+{
+       struct be_adapter *adapter = netdev_priv(netdev);
+       struct device *dev = &adapter->pdev->dev;
+       int status;
+
+       if (lancer_chip(adapter) || BEx_chip(adapter))
+               return;
+
+       if (adapter->flags & BE_FLAGS_VXLAN_OFFLOADS) {
+               dev_warn(dev, "Cannot add UDP port %d for VxLAN offloads\n",
+                        be16_to_cpu(port));
+               dev_info(dev,
+                        "Only one UDP port supported for VxLAN offloads\n");
+               return;
+       }
+
+       status = be_cmd_manage_iface(adapter, adapter->if_handle,
+                                    OP_CONVERT_NORMAL_TO_TUNNEL);
+       if (status) {
+               dev_warn(dev, "Failed to convert normal interface to tunnel\n");
+               goto err;
+       }
+
+       status = be_cmd_set_vxlan_port(adapter, port);
+       if (status) {
+               dev_warn(dev, "Failed to add VxLAN port\n");
+               goto err;
+       }
+       adapter->flags |= BE_FLAGS_VXLAN_OFFLOADS;
+       adapter->vxlan_port = port;
+
+       dev_info(dev, "Enabled VxLAN offloads for UDP port %d\n",
+                be16_to_cpu(port));
+       return;
+err:
+       be_disable_vxlan_offloads(adapter);
+       return;
+}
+
+static void be_del_vxlan_port(struct net_device *netdev, sa_family_t sa_family,
+                             __be16 port)
+{
+       struct be_adapter *adapter = netdev_priv(netdev);
+
+       if (lancer_chip(adapter) || BEx_chip(adapter))
+               return;
+
+       if (adapter->vxlan_port != port)
+               return;
+
+       be_disable_vxlan_offloads(adapter);
+
+       dev_info(&adapter->pdev->dev,
+                "Disabled VxLAN offloads for UDP port %d\n",
+                be16_to_cpu(port));
+}
+
 static const struct net_device_ops be_netdev_ops = {
        .ndo_open               = be_open,
        .ndo_stop               = be_close,
        .ndo_bridge_setlink     = be_ndo_bridge_setlink,
        .ndo_bridge_getlink     = be_ndo_bridge_getlink,
 #ifdef CONFIG_NET_RX_BUSY_POLL
-       .ndo_busy_poll          = be_busy_poll
+       .ndo_busy_poll          = be_busy_poll,
 #endif
+       .ndo_add_vxlan_port     = be_add_vxlan_port,
+       .ndo_del_vxlan_port     = be_del_vxlan_port,
 };
 
 static void be_netdev_init(struct net_device *netdev)
 {
        struct be_adapter *adapter = netdev_priv(netdev);
 
+       if (skyhawk_chip(adapter)) {
+               netdev->hw_enc_features |= NETIF_F_IP_CSUM | NETIF_F_IPV6_CSUM |
+                                          NETIF_F_TSO | NETIF_F_TSO6 |
+                                          NETIF_F_GSO_UDP_TUNNEL;
+               netdev->hw_features |= NETIF_F_GSO_UDP_TUNNEL;
+       }
        netdev->hw_features |= NETIF_F_SG | NETIF_F_TSO | NETIF_F_TSO6 |
                NETIF_F_IP_CSUM | NETIF_F_IPV6_CSUM | NETIF_F_RXCSUM |
                NETIF_F_HW_VLAN_CTAG_TX;