* All rights reserved
  * www.brocade.com
  */
+#include <linux/bitops.h>
 #include <linux/netdevice.h>
 #include <linux/skbuff.h>
 #include <linux/etherdevice.h>
 #include <linux/if_ether.h>
 #include <linux/ip.h>
 #include <linux/prefetch.h>
+#include <linux/if_vlan.h>
 
 #include "bnad.h"
 #include "bna.h"
                rcb->rxq->rx_bytes += skb->len;
                skb->protocol = eth_type_trans(skb, bnad->netdev);
 
-               if (bnad->vlan_grp && (flags & BNA_CQ_EF_VLAN)) {
-                       struct bnad_rx_ctrl *rx_ctrl =
-                               (struct bnad_rx_ctrl *)ccb->ctrl;
-                       if (skb->ip_summed == CHECKSUM_UNNECESSARY)
-                               vlan_gro_receive(&rx_ctrl->napi, bnad->vlan_grp,
-                                               ntohs(cmpl->vlan_tag), skb);
-                       else
-                               vlan_hwaccel_receive_skb(skb,
-                                                        bnad->vlan_grp,
-                                                        ntohs(cmpl->vlan_tag));
-
-               } else { /* Not VLAN tagged/stripped */
-                       struct bnad_rx_ctrl *rx_ctrl =
-                               (struct bnad_rx_ctrl *)ccb->ctrl;
-                       if (skb->ip_summed == CHECKSUM_UNNECESSARY)
-                               napi_gro_receive(&rx_ctrl->napi, skb);
-                       else
-                               netif_receive_skb(skb);
+               if (flags & BNA_CQ_EF_VLAN)
+                       __vlan_hwaccel_put_tag(skb, ntohs(cmpl->vlan_tag));
+
+               if (skb->ip_summed == CHECKSUM_UNNECESSARY) {
+                       struct bnad_rx_ctrl *rx_ctrl;
+
+                       rx_ctrl = (struct bnad_rx_ctrl *) ccb->ctrl;
+                       napi_gro_receive(&rx_ctrl->napi, skb);
+               } else {
+                       netif_receive_skb(skb);
                }
 
 next:
 static void
 bnad_restore_vlans(struct bnad *bnad, u32 rx_id)
 {
-       u16 vlan_id;
+       u16 vid;
        unsigned long flags;
 
-       if (!bnad->vlan_grp)
-               return;
-
        BUG_ON(!(VLAN_N_VID == (BFI_MAX_VLAN + 1)));
 
-       for (vlan_id = 0; vlan_id < VLAN_N_VID; vlan_id++) {
-               if (!vlan_group_get_device(bnad->vlan_grp, vlan_id))
-                       continue;
+       for_each_set_bit(vid, bnad->active_vlans, VLAN_N_VID) {
                spin_lock_irqsave(&bnad->bna_lock, flags);
-               bna_rx_vlan_add(bnad->rx_info[rx_id].rx, vlan_id);
+               bna_rx_vlan_add(bnad->rx_info[rx_id].rx, vid);
                spin_unlock_irqrestore(&bnad->bna_lock, flags);
        }
 }
        return err;
 }
 
-static void
-bnad_vlan_rx_register(struct net_device *netdev,
-                                 struct vlan_group *vlan_grp)
-{
-       struct bnad *bnad = netdev_priv(netdev);
-
-       mutex_lock(&bnad->conf_mutex);
-       bnad->vlan_grp = vlan_grp;
-       mutex_unlock(&bnad->conf_mutex);
-}
-
 static void
 bnad_vlan_rx_add_vid(struct net_device *netdev,
                                 unsigned short vid)
 
        spin_lock_irqsave(&bnad->bna_lock, flags);
        bna_rx_vlan_add(bnad->rx_info[0].rx, vid);
+       set_bit(vid, bnad->active_vlans);
        spin_unlock_irqrestore(&bnad->bna_lock, flags);
 
        mutex_unlock(&bnad->conf_mutex);
        mutex_lock(&bnad->conf_mutex);
 
        spin_lock_irqsave(&bnad->bna_lock, flags);
+       clear_bit(vid, bnad->active_vlans);
        bna_rx_vlan_del(bnad->rx_info[0].rx, vid);
        spin_unlock_irqrestore(&bnad->bna_lock, flags);
 
        .ndo_validate_addr      = eth_validate_addr,
        .ndo_set_mac_address    = bnad_set_mac_address,
        .ndo_change_mtu         = bnad_change_mtu,
-       .ndo_vlan_rx_register   = bnad_vlan_rx_register,
        .ndo_vlan_rx_add_vid    = bnad_vlan_rx_add_vid,
        .ndo_vlan_rx_kill_vid   = bnad_vlan_rx_kill_vid,
 #ifdef CONFIG_NET_POLL_CONTROLLER