from vpp_interface import VppInterface
from vpp_papi import VppEnum


INDEX_INVALID = 0xffffffff
DEFAULT_PORT = 4790
UNDEFINED_PORT = 0


def find_vxlan_gpe_tunnel(test, src, dst, s_port, d_port, vni):
    ts = test.vapi.vxlan_gpe_tunnel_v2_dump(INDEX_INVALID)

    src_port = DEFAULT_PORT
    if s_port != UNDEFINED_PORT:
        src_port = s_port

    dst_port = DEFAULT_PORT
    if d_port != UNDEFINED_PORT:
        dst_port = d_port

    for t in ts:
        if src == str(t.local) and \
           dst == str(t.remote) and \
           src_port == t.local_port and \
           dst_port == t.remote_port and \
           t.vni == vni:
            return t.sw_if_index
    return INDEX_INVALID


class VppVxlanGpeTunnel(VppInterface):
    """
    VPP VXLAN GPE interface
    """

    def __init__(self, test, src_addr, dst_addr, vni,
                 src_port=UNDEFINED_PORT, dst_port=UNDEFINED_PORT,
                 mcast_sw_if_index=INDEX_INVALID,
                 encap_vrf_id=None,
                 decap_vrf_id=None, protocol=3):
        """ Create VXLAN GPE Tunnel interface """
        super(VppVxlanGpeTunnel, self).__init__(test)
        self.src = src_addr
        self.dst = dst_addr
        self.vni = vni
        self.src_port = src_port
        self.dst_port = dst_port
        self.mcast_sw_if_index = mcast_sw_if_index
        self.encap_vrf_id = encap_vrf_id
        self.decap_vrf_id = decap_vrf_id
        self.protocol = 3

    def add_vpp_config(self):
        reply = self.test.vapi.vxlan_gpe_add_del_tunnel_v2(
            is_add=1, local=self.src, remote=self.dst, vni=self.vni,
            local_port=self.src_port, remote_port=self.dst_port,
            mcast_sw_if_index=self.mcast_sw_if_index,
            encap_vrf_id=self.encap_vrf_id,
            decap_vrf_id=self.decap_vrf_id,
            protocol=self.protocol)
        self.set_sw_if_index(reply.sw_if_index)
        self._test.registry.register(self, self._test.logger)

    def remove_vpp_config(self):
        self.test.vapi.vxlan_gpe_add_del_tunnel_v2(
            is_add=0, local=self.src, remote=self.dst, vni=self.vni,
            local_port=self.src_port, remote_port=self.dst_port,
            mcast_sw_if_index=self.mcast_sw_if_index,
            encap_vrf_id=self.encap_vrf_id,
            decap_vrf_id=self.decap_vrf_id,
            protocol=self.protocol)

    def query_vpp_config(self):
        return (INDEX_INVALID != find_vxlan_gpe_tunnel(self.
@@ -283,7 +283,7 @@ dpdk_create_vhost_user_if_internal (u32 * hw_if_index, u32 if_id, u8 * hwaddr)
     {
       // vui was not retrieved from inactive ifaces - create new
       vec_add2_aligned (dm->devices, xd, 1, CLIB_CACHE_LINE_BYTES);
-      xd->dev_type = VNET_DPDK_DEV_VHOST_USER;
+      xd->flags |= DPDK_DEVICE_FLAG_VHOST_USER;
       xd->rx_q_used = num_qpairs;
       xd->tx_q_used = num_qpairs;
       xd->vu_vhost_dev.virt_qp_nb = num_qpairs;
@@ -1661,7 +1661,7 @@ dpdk_vhost_user_dump_ifs (vnet_main_t * vnm, vlib_main_t * vm,
 
   vec_foreach (xd, dm->devices)
   {
-    if (xd->dev_type == VNET_DPDK_DEV_VHOST_USER && xd->vu_intf->active)
+    if ((xd->flags & DPDK_DEVICE_FLAG_VHOST_USER) && xd->vu_intf->active)
       vec_add1 (hw_if_indices, xd->vlib_hw_if_index);
   }
 
@@ -1988,7 +1988,7 @@ show_dpdk_vhost_user_command_fn (vlib_main_t * vm,
     {
       vec_foreach (xd, dm->devices)
       {
-	if (xd->dev_type == VNET_DPDK_DEV_VHOST_USER && xd->vu_intf->active)
+	if ((xd->flags DPDK_DEVICE_FLAG_VHOST_USER) && xd->vu_intf->active)
 	  vec_add1 (hw_if_indices, xd->vlib_hw_if_index);
       }
     }