*
  * Portions of this file
  * Copyright(c) 2015 - 2016 Intel Deutschland GmbH
+ * Copyright (C) 2018 Intel Corporation
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 2 as
        memcpy(&vht_cap->vht_mcs, &vht_cap_ie->supp_mcs,
               sizeof(struct ieee80211_vht_mcs_info));
 
+       /* copy EXT_NSS_BW Support value or remove the capability */
+       if (ieee80211_hw_check(&sdata->local->hw, SUPPORTS_VHT_EXT_NSS_BW))
+               vht_cap->cap |= (cap_info & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK);
+       else
+               vht_cap->vht_mcs.tx_highest &=
+                       ~cpu_to_le16(IEEE80211_VHT_EXT_NSS_BW_CAPABLE);
+
        /* but also restrict MCSes */
        for (i = 0; i < 8; i++) {
                u16 own_rx, own_tx, peer_rx, peer_tx;
                break;
        default:
                sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_80;
+
+               if (!(vht_cap->vht_mcs.tx_highest &
+                               cpu_to_le16(IEEE80211_VHT_EXT_NSS_BW_CAPABLE)))
+                       break;
+
+               /*
+                * If this is non-zero, then it does support 160 MHz after all,
+                * in one form or the other. We don't distinguish here (or even
+                * above) between 160 and 80+80 yet.
+                */
+               if (cap_info & IEEE80211_VHT_CAP_EXT_NSS_BW_MASK)
+                       sta->cur_max_bandwidth = IEEE80211_STA_RX_BW_160;
        }
 
        sta->sta.bandwidth = ieee80211_sta_cur_vht_bw(sta);