modify set_vf_rx_vlan function to handle the i40e PMD.

Signed-off-by: Bernard Iremonger <bernard.iremon...@intel.com>
---
 app/test-pmd/config.c | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/app/test-pmd/config.c b/app/test-pmd/config.c
index 36c47ab..0368dc6 100644
--- a/app/test-pmd/config.c
+++ b/app/test-pmd/config.c
@@ -92,6 +92,9 @@
 #include <rte_ethdev.h>
 #include <rte_string_fns.h>
 #include <rte_cycles.h>
+#ifdef RTE_LIBRTE_I40E_PMD
+#include <rte_pmd_i40e.h>
+#endif
 
 #include "testpmd.h"
 
@@ -2349,12 +2352,24 @@ struct igb_ring_desc_16_bytes {
 set_vf_rx_vlan(portid_t port_id, uint16_t vlan_id, uint64_t vf_mask, uint8_t 
on)
 {
        int diag;
+       struct rte_eth_dev_info dev_info;
 
        if (port_id_is_invalid(port_id, ENABLED_WARN))
                return;
        if (vlan_id_is_invalid(vlan_id))
                return;
-       diag = rte_eth_dev_set_vf_vlan_filter(port_id, vlan_id, vf_mask, on);
+
+       rte_eth_dev_info_get(port_id, &dev_info);
+
+#ifdef RTE_LIBRTE_I40E_PMD
+       if (strstr(dev_info.driver_name, "i40e") != NULL)
+               diag = rte_pmd_i40e_set_vf_vlan_filter(port_id, vlan_id,
+                                                      vf_mask, on);
+       else
+#endif
+               diag = rte_eth_dev_set_vf_vlan_filter(port_id, vlan_id,
+                                                     vf_mask, on);
+
        if (diag == 0)
                return;
        printf("rte_eth_dev_set_vf_vlan_filter for port_id=%d failed "
-- 
1.9.3

Reply via email to