net/mac80211/tdls.c

   1 /*
   2  * mac80211 TDLS handling code
   3  *
   4  * Copyright 2006-2010  Johannes Berg <johannes@sipsolutions.net>
   5  * Copyright 2014, Intel Corporation
   6  * Copyright 2014  Intel Mobile Communications GmbH
   7  * Copyright 2015  Intel Deutschland GmbH
   8  *
   9  * This file is GPLv2 as found in COPYING.
  10  */
  11
  12 #include <linux/ieee80211.h>
  13 #include <linux/log2.h>
  14 #include <net/cfg80211.h>
  15 #include <linux/rtnetlink.h>
  16 #include "ieee80211_i.h"
  17 #include "driver-ops.h"
  18
  19 /* give usermode some time for retries in setting up the TDLS session */
  20 #define TDLS_PEER_SETUP_TIMEOUT (15 * HZ)
  21
  22 void ieee80211_tdls_peer_del_work(struct work_struct *wk)
  23 {
  24         struct ieee80211_sub_if_data *sdata;
  25         struct ieee80211_local *local;
  26
  27         sdata = container_of(wk, struct ieee80211_sub_if_data,
  28                              u.mgd.tdls_peer_del_work.work);
  29         local = sdata->local;
  30
  31         mutex_lock(&local->mtx);
  32         if (!is_zero_ether_addr(sdata->u.mgd.tdls_peer)) {
  33                 tdls_dbg(sdata, "TDLS del peer %pM\n", sdata->u.mgd.tdls_peer);
  34                 sta_info_destroy_addr(sdata, sdata->u.mgd.tdls_peer);
  35                 eth_zero_addr(sdata->u.mgd.tdls_peer);
  36         }
  37         mutex_unlock(&local->mtx);
  38 }
  39
  40 static void ieee80211_tdls_add_ext_capab(struct ieee80211_sub_if_data *sdata,
  41                                          struct sk_buff *skb)
  42 {
  43         struct ieee80211_local *local = sdata->local;
  44         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
  45         bool chan_switch = local->hw.wiphy->features &
  46                            NL80211_FEATURE_TDLS_CHANNEL_SWITCH;
  47         bool wider_band = ieee80211_hw_check(&local->hw, TDLS_WIDER_BW) &&
  48                           !ifmgd->tdls_wider_bw_prohibited;
  49         enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
  50         struct ieee80211_supported_band *sband = local->hw.wiphy->bands[band];
  51         bool vht = sband && sband->vht_cap.vht_supported;
  52         u8 *pos = (void *)skb_put(skb, 10);
  53
  54         *pos++ = WLAN_EID_EXT_CAPABILITY;
  55         *pos++ = 8; /* len */
  56         *pos++ = 0x0;
  57         *pos++ = 0x0;
  58         *pos++ = 0x0;
  59         *pos++ = chan_switch ? WLAN_EXT_CAPA4_TDLS_CHAN_SWITCH : 0;
  60         *pos++ = WLAN_EXT_CAPA5_TDLS_ENABLED;
  61         *pos++ = 0;
  62         *pos++ = 0;
  63         *pos++ = (vht && wider_band) ? WLAN_EXT_CAPA8_TDLS_WIDE_BW_ENABLED : 0;
  64 }
  65
  66 static u8
  67 ieee80211_tdls_add_subband(struct ieee80211_sub_if_data *sdata,
  68                            struct sk_buff *skb, u16 start, u16 end,
  69                            u16 spacing)
  70 {
  71         u8 subband_cnt = 0, ch_cnt = 0;
  72         struct ieee80211_channel *ch;
  73         struct cfg80211_chan_def chandef;
  74         int i, subband_start;
  75         struct wiphy *wiphy = sdata->local->hw.wiphy;
  76
  77         for (i = start; i <= end; i += spacing) {
  78                 if (!ch_cnt)
  79                         subband_start = i;
  80
  81                 ch = ieee80211_get_channel(sdata->local->hw.wiphy, i);
  82                 if (ch) {
  83                         /* we will be active on the channel */
  84                         cfg80211_chandef_create(&chandef, ch,
  85                                                 NL80211_CHAN_NO_HT);
  86                         if (cfg80211_reg_can_beacon_relax(wiphy, &chandef,
  87                                                           sdata->wdev.iftype)) {
  88                                 ch_cnt++;
  89                                 /*
  90                                  * check if the next channel is also part of
  91                                  * this allowed range
  92                                  */
  93                                 continue;
  94                         }
  95                 }
  96
  97                 /*
  98                  * we've reached the end of a range, with allowed channels
  99                  * found
 100                  */
 101                 if (ch_cnt) {
 102                         u8 *pos = skb_put(skb, 2);
 103                         *pos++ = ieee80211_frequency_to_channel(subband_start);
 104                         *pos++ = ch_cnt;
 105
 106                         subband_cnt++;
 107                         ch_cnt = 0;
 108                 }
 109         }
 110
 111         /* all channels in the requested range are allowed - add them here */
 112         if (ch_cnt) {
 113                 u8 *pos = skb_put(skb, 2);
 114                 *pos++ = ieee80211_frequency_to_channel(subband_start);
 115                 *pos++ = ch_cnt;
 116
 117                 subband_cnt++;
 118         }
 119
 120         return subband_cnt;
 121 }
 122
 123 static void
 124 ieee80211_tdls_add_supp_channels(struct ieee80211_sub_if_data *sdata,
 125                                  struct sk_buff *skb)
 126 {
 127         /*
 128          * Add possible channels for TDLS. These are channels that are allowed
 129          * to be active.
 130          */
 131         u8 subband_cnt;
 132         u8 *pos = skb_put(skb, 2);
 133
 134         *pos++ = WLAN_EID_SUPPORTED_CHANNELS;
 135
 136         /*
 137          * 5GHz and 2GHz channels numbers can overlap. Ignore this for now, as
 138          * this doesn't happen in real world scenarios.
 139          */
 140
 141         /* 2GHz, with 5MHz spacing */
 142         subband_cnt = ieee80211_tdls_add_subband(sdata, skb, 2412, 2472, 5);
 143
 144         /* 5GHz, with 20MHz spacing */
 145         subband_cnt += ieee80211_tdls_add_subband(sdata, skb, 5000, 5825, 20);
 146
 147         /* length */
 148         *pos = 2 * subband_cnt;
 149 }
 150
 151 static void ieee80211_tdls_add_oper_classes(struct ieee80211_sub_if_data *sdata,
 152                                             struct sk_buff *skb)
 153 {
 154         u8 *pos;
 155         u8 op_class;
 156
 157         if (!ieee80211_chandef_to_operating_class(&sdata->vif.bss_conf.chandef,
 158                                                   &op_class))
 159                 return;
 160
 161         pos = skb_put(skb, 4);
 162         *pos++ = WLAN_EID_SUPPORTED_REGULATORY_CLASSES;
 163         *pos++ = 2; /* len */
 164
 165         *pos++ = op_class;
 166         *pos++ = op_class; /* give current operating class as alternate too */
 167 }
 168
 169 static void ieee80211_tdls_add_bss_coex_ie(struct sk_buff *skb)
 170 {
 171         u8 *pos = (void *)skb_put(skb, 3);
 172
 173         *pos++ = WLAN_EID_BSS_COEX_2040;
 174         *pos++ = 1; /* len */
 175
 176         *pos++ = WLAN_BSS_COEX_INFORMATION_REQUEST;
 177 }
 178
 179 static u16 ieee80211_get_tdls_sta_capab(struct ieee80211_sub_if_data *sdata,
 180                                         u16 status_code)
 181 {
 182         /* The capability will be 0 when sending a failure code */
 183         if (status_code != 0)
 184                 return 0;
 185
 186         if (ieee80211_get_sdata_band(sdata) == IEEE80211_BAND_2GHZ) {
 187                 return WLAN_CAPABILITY_SHORT_SLOT_TIME |
 188                        WLAN_CAPABILITY_SHORT_PREAMBLE;
 189         }
 190
 191         return 0;
 192 }
 193
 194 static void ieee80211_tdls_add_link_ie(struct ieee80211_sub_if_data *sdata,
 195                                        struct sk_buff *skb, const u8 *peer,
 196                                        bool initiator)
 197 {
 198         struct ieee80211_tdls_lnkie *lnkid;
 199         const u8 *init_addr, *rsp_addr;
 200
 201         if (initiator) {
 202                 init_addr = sdata->vif.addr;
 203                 rsp_addr = peer;
 204         } else {
 205                 init_addr = peer;
 206                 rsp_addr = sdata->vif.addr;
 207         }
 208
 209         lnkid = (void *)skb_put(skb, sizeof(struct ieee80211_tdls_lnkie));
 210
 211         lnkid->ie_type = WLAN_EID_LINK_ID;
 212         lnkid->ie_len = sizeof(struct ieee80211_tdls_lnkie) - 2;
 213
 214         memcpy(lnkid->bssid, sdata->u.mgd.bssid, ETH_ALEN);
 215         memcpy(lnkid->init_sta, init_addr, ETH_ALEN);
 216         memcpy(lnkid->resp_sta, rsp_addr, ETH_ALEN);
 217 }
 218
 219 static void
 220 ieee80211_tdls_add_aid(struct ieee80211_sub_if_data *sdata, struct sk_buff *skb)
 221 {
 222         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 223         u8 *pos = (void *)skb_put(skb, 4);
 224
 225         *pos++ = WLAN_EID_AID;
 226         *pos++ = 2; /* len */
 227         put_unaligned_le16(ifmgd->aid, pos);
 228 }
 229
 230 /* translate numbering in the WMM parameter IE to the mac80211 notation */
 231 static enum ieee80211_ac_numbers ieee80211_ac_from_wmm(int ac)
 232 {
 233         switch (ac) {
 234         default:
 235                 WARN_ON_ONCE(1);
 236         case 0:
 237                 return IEEE80211_AC_BE;
 238         case 1:
 239                 return IEEE80211_AC_BK;
 240         case 2:
 241                 return IEEE80211_AC_VI;
 242         case 3:
 243                 return IEEE80211_AC_VO;
 244         }
 245 }
 246
 247 static u8 ieee80211_wmm_aci_aifsn(int aifsn, bool acm, int aci)
 248 {
 249         u8 ret;
 250
 251         ret = aifsn & 0x0f;
 252         if (acm)
 253                 ret |= 0x10;
 254         ret |= (aci << 5) & 0x60;
 255         return ret;
 256 }
 257
 258 static u8 ieee80211_wmm_ecw(u16 cw_min, u16 cw_max)
 259 {
 260         return ((ilog2(cw_min + 1) << 0x0) & 0x0f) |
 261                ((ilog2(cw_max + 1) << 0x4) & 0xf0);
 262 }
 263
 264 static void ieee80211_tdls_add_wmm_param_ie(struct ieee80211_sub_if_data *sdata,
 265                                             struct sk_buff *skb)
 266 {
 267         struct ieee80211_wmm_param_ie *wmm;
 268         struct ieee80211_tx_queue_params *txq;
 269         int i;
 270
 271         wmm = (void *)skb_put(skb, sizeof(*wmm));
 272         memset(wmm, 0, sizeof(*wmm));
 273
 274         wmm->element_id = WLAN_EID_VENDOR_SPECIFIC;
 275         wmm->len = sizeof(*wmm) - 2;
 276
 277         wmm->oui[0] = 0x00; /* Microsoft OUI 00:50:F2 */
 278         wmm->oui[1] = 0x50;
 279         wmm->oui[2] = 0xf2;
 280         wmm->oui_type = 2; /* WME */
 281         wmm->oui_subtype = 1; /* WME param */
 282         wmm->version = 1; /* WME ver */
 283         wmm->qos_info = 0; /* U-APSD not in use */
 284
 285         /*
 286          * Use the EDCA parameters defined for the BSS, or default if the AP
 287          * doesn't support it, as mandated by 802.11-2012 section 10.22.4
 288          */
 289         for (i = 0; i < IEEE80211_NUM_ACS; i++) {
 290                 txq = &sdata->tx_conf[ieee80211_ac_from_wmm(i)];
 291                 wmm->ac[i].aci_aifsn = ieee80211_wmm_aci_aifsn(txq->aifs,
 292                                                                txq->acm, i);
 293                 wmm->ac[i].cw = ieee80211_wmm_ecw(txq->cw_min, txq->cw_max);
 294                 wmm->ac[i].txop_limit = cpu_to_le16(txq->txop);
 295         }
 296 }
 297
 298 static void
 299 ieee80211_tdls_chandef_vht_upgrade(struct ieee80211_sub_if_data *sdata,
 300                                    struct sta_info *sta)
 301 {
 302         /* IEEE802.11ac-2013 Table E-4 */
 303         u16 centers_80mhz[] = { 5210, 5290, 5530, 5610, 5690, 5775 };
 304         struct cfg80211_chan_def uc = sta->tdls_chandef;
 305         enum nl80211_chan_width max_width = ieee80211_get_sta_bw(&sta->sta);
 306         int i;
 307
 308         /* only support upgrading non-narrow channels up to 80Mhz */
 309         if (max_width == NL80211_CHAN_WIDTH_5 ||
 310             max_width == NL80211_CHAN_WIDTH_10)
 311                 return;
 312
 313         if (max_width > NL80211_CHAN_WIDTH_80)
 314                 max_width = NL80211_CHAN_WIDTH_80;
 315
 316         if (uc.width == max_width)
 317                 return;
 318         /*
 319          * Channel usage constrains in the IEEE802.11ac-2013 specification only
 320          * allow expanding a 20MHz channel to 80MHz in a single way. In
 321          * addition, there are no 40MHz allowed channels that are not part of
 322          * the allowed 80MHz range in the 5GHz spectrum (the relevant one here).
 323          */
 324         for (i = 0; i < ARRAY_SIZE(centers_80mhz); i++)
 325                 if (abs(uc.chan->center_freq - centers_80mhz[i]) <= 30) {
 326                         uc.center_freq1 = centers_80mhz[i];
 327                         uc.width = NL80211_CHAN_WIDTH_80;
 328                         break;
 329                 }
 330
 331         if (!uc.center_freq1)
 332                 return;
 333
 334         /* proceed to downgrade the chandef until usable or the same */
 335         while (uc.width > max_width &&
 336                !cfg80211_reg_can_beacon_relax(sdata->local->hw.wiphy, &uc,
 337                                               sdata->wdev.iftype))
 338                 ieee80211_chandef_downgrade(&uc);
 339
 340         if (!cfg80211_chandef_identical(&uc, &sta->tdls_chandef)) {
 341                 tdls_dbg(sdata, "TDLS ch width upgraded %d -> %d\n",
 342                          sta->tdls_chandef.width, uc.width);
 343
 344                 /*
 345                  * the station is not yet authorized when BW upgrade is done,
 346                  * locking is not required
 347                  */
 348                 sta->tdls_chandef = uc;
 349         }
 350 }
 351
 352 static void
 353 ieee80211_tdls_add_setup_start_ies(struct ieee80211_sub_if_data *sdata,
 354                                    struct sk_buff *skb, const u8 *peer,
 355                                    u8 action_code, bool initiator,
 356                                    const u8 *extra_ies, size_t extra_ies_len)
 357 {
 358         enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
 359         struct ieee80211_local *local = sdata->local;
 360         struct ieee80211_supported_band *sband;
 361         struct ieee80211_sta_ht_cap ht_cap;
 362         struct ieee80211_sta_vht_cap vht_cap;
 363         struct sta_info *sta = NULL;
 364         size_t offset = 0, noffset;
 365         u8 *pos;
 366
 367         ieee80211_add_srates_ie(sdata, skb, false, band);
 368         ieee80211_add_ext_srates_ie(sdata, skb, false, band);
 369         ieee80211_tdls_add_supp_channels(sdata, skb);
 370
 371         /* add any custom IEs that go before Extended Capabilities */
 372         if (extra_ies_len) {
 373                 static const u8 before_ext_cap[] = {
 374                         WLAN_EID_SUPP_RATES,
 375                         WLAN_EID_COUNTRY,
 376                         WLAN_EID_EXT_SUPP_RATES,
 377                         WLAN_EID_SUPPORTED_CHANNELS,
 378                         WLAN_EID_RSN,
 379                 };
 380                 noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
 381                                              before_ext_cap,
 382                                              ARRAY_SIZE(before_ext_cap),
 383                                              offset);
 384                 pos = skb_put(skb, noffset - offset);
 385                 memcpy(pos, extra_ies + offset, noffset - offset);
 386                 offset = noffset;
 387         }
 388
 389         ieee80211_tdls_add_ext_capab(sdata, skb);
 390
 391         /* add the QoS element if we support it */
 392         if (local->hw.queues >= IEEE80211_NUM_ACS &&
 393             action_code != WLAN_PUB_ACTION_TDLS_DISCOVER_RES)
 394                 ieee80211_add_wmm_info_ie(skb_put(skb, 9), 0); /* no U-APSD */
 395
 396         /* add any custom IEs that go before HT capabilities */
 397         if (extra_ies_len) {
 398                 static const u8 before_ht_cap[] = {
 399                         WLAN_EID_SUPP_RATES,
 400                         WLAN_EID_COUNTRY,
 401                         WLAN_EID_EXT_SUPP_RATES,
 402                         WLAN_EID_SUPPORTED_CHANNELS,
 403                         WLAN_EID_RSN,
 404                         WLAN_EID_EXT_CAPABILITY,
 405                         WLAN_EID_QOS_CAPA,
 406                         WLAN_EID_FAST_BSS_TRANSITION,
 407                         WLAN_EID_TIMEOUT_INTERVAL,
 408                         WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
 409                 };
 410                 noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
 411                                              before_ht_cap,
 412                                              ARRAY_SIZE(before_ht_cap),
 413                                              offset);
 414                 pos = skb_put(skb, noffset - offset);
 415                 memcpy(pos, extra_ies + offset, noffset - offset);
 416                 offset = noffset;
 417         }
 418
 419         mutex_lock(&local->sta_mtx);
 420
 421         /* we should have the peer STA if we're already responding */
 422         if (action_code == WLAN_TDLS_SETUP_RESPONSE) {
 423                 sta = sta_info_get(sdata, peer);
 424                 if (WARN_ON_ONCE(!sta)) {
 425                         mutex_unlock(&local->sta_mtx);
 426                         return;
 427                 }
 428
 429                 sta->tdls_chandef = sdata->vif.bss_conf.chandef;
 430         }
 431
 432         ieee80211_tdls_add_oper_classes(sdata, skb);
 433
 434         /*
 435          * with TDLS we can switch channels, and HT-caps are not necessarily
 436          * the same on all bands. The specification limits the setup to a
 437          * single HT-cap, so use the current band for now.
 438          */
 439         sband = local->hw.wiphy->bands[band];
 440         memcpy(&ht_cap, &sband->ht_cap, sizeof(ht_cap));
 441
 442         if ((action_code == WLAN_TDLS_SETUP_REQUEST ||
 443              action_code == WLAN_PUB_ACTION_TDLS_DISCOVER_RES) &&
 444             ht_cap.ht_supported) {
 445                 ieee80211_apply_htcap_overrides(sdata, &ht_cap);
 446
 447                 /* disable SMPS in TDLS initiator */
 448                 ht_cap.cap |= WLAN_HT_CAP_SM_PS_DISABLED
 449                                 << IEEE80211_HT_CAP_SM_PS_SHIFT;
 450
 451                 pos = skb_put(skb, sizeof(struct ieee80211_ht_cap) + 2);
 452                 ieee80211_ie_build_ht_cap(pos, &ht_cap, ht_cap.cap);
 453         } else if (action_code == WLAN_TDLS_SETUP_RESPONSE &&
 454                    ht_cap.ht_supported && sta->sta.ht_cap.ht_supported) {
 455                 /* the peer caps are already intersected with our own */
 456                 memcpy(&ht_cap, &sta->sta.ht_cap, sizeof(ht_cap));
 457
 458                 pos = skb_put(skb, sizeof(struct ieee80211_ht_cap) + 2);
 459                 ieee80211_ie_build_ht_cap(pos, &ht_cap, ht_cap.cap);
 460         }
 461
 462         if (ht_cap.ht_supported &&
 463             (ht_cap.cap & IEEE80211_HT_CAP_SUP_WIDTH_20_40))
 464                 ieee80211_tdls_add_bss_coex_ie(skb);
 465
 466         ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
 467
 468         /* add any custom IEs that go before VHT capabilities */
 469         if (extra_ies_len) {
 470                 static const u8 before_vht_cap[] = {
 471                         WLAN_EID_SUPP_RATES,
 472                         WLAN_EID_COUNTRY,
 473                         WLAN_EID_EXT_SUPP_RATES,
 474                         WLAN_EID_SUPPORTED_CHANNELS,
 475                         WLAN_EID_RSN,
 476                         WLAN_EID_EXT_CAPABILITY,
 477                         WLAN_EID_QOS_CAPA,
 478                         WLAN_EID_FAST_BSS_TRANSITION,
 479                         WLAN_EID_TIMEOUT_INTERVAL,
 480                         WLAN_EID_SUPPORTED_REGULATORY_CLASSES,
 481                         WLAN_EID_MULTI_BAND,
 482                 };
 483                 noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
 484                                              before_vht_cap,
 485                                              ARRAY_SIZE(before_vht_cap),
 486                                              offset);
 487                 pos = skb_put(skb, noffset - offset);
 488                 memcpy(pos, extra_ies + offset, noffset - offset);
 489                 offset = noffset;
 490         }
 491
 492         /* build the VHT-cap similarly to the HT-cap */
 493         memcpy(&vht_cap, &sband->vht_cap, sizeof(vht_cap));
 494         if ((action_code == WLAN_TDLS_SETUP_REQUEST ||
 495              action_code == WLAN_PUB_ACTION_TDLS_DISCOVER_RES) &&
 496             vht_cap.vht_supported) {
 497                 ieee80211_apply_vhtcap_overrides(sdata, &vht_cap);
 498
 499                 /* the AID is present only when VHT is implemented */
 500                 if (action_code == WLAN_TDLS_SETUP_REQUEST)
 501                         ieee80211_tdls_add_aid(sdata, skb);
 502
 503                 pos = skb_put(skb, sizeof(struct ieee80211_vht_cap) + 2);
 504                 ieee80211_ie_build_vht_cap(pos, &vht_cap, vht_cap.cap);
 505         } else if (action_code == WLAN_TDLS_SETUP_RESPONSE &&
 506                    vht_cap.vht_supported && sta->sta.vht_cap.vht_supported) {
 507                 /* the peer caps are already intersected with our own */
 508                 memcpy(&vht_cap, &sta->sta.vht_cap, sizeof(vht_cap));
 509
 510                 /* the AID is present only when VHT is implemented */
 511                 ieee80211_tdls_add_aid(sdata, skb);
 512
 513                 pos = skb_put(skb, sizeof(struct ieee80211_vht_cap) + 2);
 514                 ieee80211_ie_build_vht_cap(pos, &vht_cap, vht_cap.cap);
 515
 516                 /*
 517                  * if both peers support WIDER_BW, we can expand the chandef to
 518                  * a wider compatible one, up to 80MHz
 519                  */
 520                 if (test_sta_flag(sta, WLAN_STA_TDLS_WIDER_BW))
 521                         ieee80211_tdls_chandef_vht_upgrade(sdata, sta);
 522         }
 523
 524         mutex_unlock(&local->sta_mtx);
 525
 526         /* add any remaining IEs */
 527         if (extra_ies_len) {
 528                 noffset = extra_ies_len;
 529                 pos = skb_put(skb, noffset - offset);
 530                 memcpy(pos, extra_ies + offset, noffset - offset);
 531         }
 532
 533 }
 534
 535 static void
 536 ieee80211_tdls_add_setup_cfm_ies(struct ieee80211_sub_if_data *sdata,
 537                                  struct sk_buff *skb, const u8 *peer,
 538                                  bool initiator, const u8 *extra_ies,
 539                                  size_t extra_ies_len)
 540 {
 541         struct ieee80211_local *local = sdata->local;
 542         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
 543         size_t offset = 0, noffset;
 544         struct sta_info *sta, *ap_sta;
 545         enum ieee80211_band band = ieee80211_get_sdata_band(sdata);
 546         u8 *pos;
 547
 548         mutex_lock(&local->sta_mtx);
 549
 550         sta = sta_info_get(sdata, peer);
 551         ap_sta = sta_info_get(sdata, ifmgd->bssid);
 552         if (WARN_ON_ONCE(!sta || !ap_sta)) {
 553                 mutex_unlock(&local->sta_mtx);
 554                 return;
 555         }
 556
 557         sta->tdls_chandef = sdata->vif.bss_conf.chandef;
 558
 559         /* add any custom IEs that go before the QoS IE */
 560         if (extra_ies_len) {
 561                 static const u8 before_qos[] = {
 562                         WLAN_EID_RSN,
 563                 };
 564                 noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
 565                                              before_qos,
 566                                              ARRAY_SIZE(before_qos),
 567                                              offset);
 568                 pos = skb_put(skb, noffset - offset);
 569                 memcpy(pos, extra_ies + offset, noffset - offset);
 570                 offset = noffset;
 571         }
 572
 573         /* add the QoS param IE if both the peer and we support it */
 574         if (local->hw.queues >= IEEE80211_NUM_ACS && sta->sta.wme)
 575                 ieee80211_tdls_add_wmm_param_ie(sdata, skb);
 576
 577         /* add any custom IEs that go before HT operation */
 578         if (extra_ies_len) {
 579                 static const u8 before_ht_op[] = {
 580                         WLAN_EID_RSN,
 581                         WLAN_EID_QOS_CAPA,
 582                         WLAN_EID_FAST_BSS_TRANSITION,
 583                         WLAN_EID_TIMEOUT_INTERVAL,
 584                 };
 585                 noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
 586                                              before_ht_op,
 587                                              ARRAY_SIZE(before_ht_op),
 588                                              offset);
 589                 pos = skb_put(skb, noffset - offset);
 590                 memcpy(pos, extra_ies + offset, noffset - offset);
 591                 offset = noffset;
 592         }
 593
 594         /* if HT support is only added in TDLS, we need an HT-operation IE */
 595         if (!ap_sta->sta.ht_cap.ht_supported && sta->sta.ht_cap.ht_supported) {
 596                 pos = skb_put(skb, 2 + sizeof(struct ieee80211_ht_operation));
 597                 /* send an empty HT operation IE */
 598                 ieee80211_ie_build_ht_oper(pos, &sta->sta.ht_cap,
 599                                            &sdata->vif.bss_conf.chandef, 0);
 600         }
 601
 602         ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
 603
 604         /* only include VHT-operation if not on the 2.4GHz band */
 605         if (band != IEEE80211_BAND_2GHZ && sta->sta.vht_cap.vht_supported) {
 606                 /*
 607                  * if both peers support WIDER_BW, we can expand the chandef to
 608                  * a wider compatible one, up to 80MHz
 609                  */
 610                 if (test_sta_flag(sta, WLAN_STA_TDLS_WIDER_BW))
 611                         ieee80211_tdls_chandef_vht_upgrade(sdata, sta);
 612
 613                 pos = skb_put(skb, 2 + sizeof(struct ieee80211_vht_operation));
 614                 ieee80211_ie_build_vht_oper(pos, &sta->sta.vht_cap,
 615                                             &sta->tdls_chandef);
 616         }
 617
 618         mutex_unlock(&local->sta_mtx);
 619
 620         /* add any remaining IEs */
 621         if (extra_ies_len) {
 622                 noffset = extra_ies_len;
 623                 pos = skb_put(skb, noffset - offset);
 624                 memcpy(pos, extra_ies + offset, noffset - offset);
 625         }
 626 }
 627
 628 static void
 629 ieee80211_tdls_add_chan_switch_req_ies(struct ieee80211_sub_if_data *sdata,
 630                                        struct sk_buff *skb, const u8 *peer,
 631                                        bool initiator, const u8 *extra_ies,
 632                                        size_t extra_ies_len, u8 oper_class,
 633                                        struct cfg80211_chan_def *chandef)
 634 {
 635         struct ieee80211_tdls_data *tf;
 636         size_t offset = 0, noffset;
 637         u8 *pos;
 638
 639         if (WARN_ON_ONCE(!chandef))
 640                 return;
 641
 642         tf = (void *)skb->data;
 643         tf->u.chan_switch_req.target_channel =
 644                 ieee80211_frequency_to_channel(chandef->chan->center_freq);
 645         tf->u.chan_switch_req.oper_class = oper_class;
 646
 647         if (extra_ies_len) {
 648                 static const u8 before_lnkie[] = {
 649                         WLAN_EID_SECONDARY_CHANNEL_OFFSET,
 650                 };
 651                 noffset = ieee80211_ie_split(extra_ies, extra_ies_len,
 652                                              before_lnkie,
 653                                              ARRAY_SIZE(before_lnkie),
 654                                              offset);
 655                 pos = skb_put(skb, noffset - offset);
 656                 memcpy(pos, extra_ies + offset, noffset - offset);
 657                 offset = noffset;
 658         }
 659
 660         ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
 661
 662         /* add any remaining IEs */
 663         if (extra_ies_len) {
 664                 noffset = extra_ies_len;
 665                 pos = skb_put(skb, noffset - offset);
 666                 memcpy(pos, extra_ies + offset, noffset - offset);
 667         }
 668 }
 669
 670 static void
 671 ieee80211_tdls_add_chan_switch_resp_ies(struct ieee80211_sub_if_data *sdata,
 672                                         struct sk_buff *skb, const u8 *peer,
 673                                         u16 status_code, bool initiator,
 674                                         const u8 *extra_ies,
 675                                         size_t extra_ies_len)
 676 {
 677         if (status_code == 0)
 678                 ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
 679
 680         if (extra_ies_len)
 681                 memcpy(skb_put(skb, extra_ies_len), extra_ies, extra_ies_len);
 682 }
 683
 684 static void ieee80211_tdls_add_ies(struct ieee80211_sub_if_data *sdata,
 685                                    struct sk_buff *skb, const u8 *peer,
 686                                    u8 action_code, u16 status_code,
 687                                    bool initiator, const u8 *extra_ies,
 688                                    size_t extra_ies_len, u8 oper_class,
 689                                    struct cfg80211_chan_def *chandef)
 690 {
 691         switch (action_code) {
 692         case WLAN_TDLS_SETUP_REQUEST:
 693         case WLAN_TDLS_SETUP_RESPONSE:
 694         case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
 695                 if (status_code == 0)
 696                         ieee80211_tdls_add_setup_start_ies(sdata, skb, peer,
 697                                                            action_code,
 698                                                            initiator,
 699                                                            extra_ies,
 700                                                            extra_ies_len);
 701                 break;
 702         case WLAN_TDLS_SETUP_CONFIRM:
 703                 if (status_code == 0)
 704                         ieee80211_tdls_add_setup_cfm_ies(sdata, skb, peer,
 705                                                          initiator, extra_ies,
 706                                                          extra_ies_len);
 707                 break;
 708         case WLAN_TDLS_TEARDOWN:
 709         case WLAN_TDLS_DISCOVERY_REQUEST:
 710                 if (extra_ies_len)
 711                         memcpy(skb_put(skb, extra_ies_len), extra_ies,
 712                                extra_ies_len);
 713                 if (status_code == 0 || action_code == WLAN_TDLS_TEARDOWN)
 714                         ieee80211_tdls_add_link_ie(sdata, skb, peer, initiator);
 715                 break;
 716         case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
 717                 ieee80211_tdls_add_chan_switch_req_ies(sdata, skb, peer,
 718                                                        initiator, extra_ies,
 719                                                        extra_ies_len,
 720                                                        oper_class, chandef);
 721                 break;
 722         case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
 723                 ieee80211_tdls_add_chan_switch_resp_ies(sdata, skb, peer,
 724                                                         status_code,
 725                                                         initiator, extra_ies,
 726                                                         extra_ies_len);
 727                 break;
 728         }
 729
 730 }
 731
 732 static int
 733 ieee80211_prep_tdls_encap_data(struct wiphy *wiphy, struct net_device *dev,
 734                                const u8 *peer, u8 action_code, u8 dialog_token,
 735                                u16 status_code, struct sk_buff *skb)
 736 {
 737         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
 738         struct ieee80211_tdls_data *tf;
 739
 740         tf = (void *)skb_put(skb, offsetof(struct ieee80211_tdls_data, u));
 741
 742         memcpy(tf->da, peer, ETH_ALEN);
 743         memcpy(tf->sa, sdata->vif.addr, ETH_ALEN);
 744         tf->ether_type = cpu_to_be16(ETH_P_TDLS);
 745         tf->payload_type = WLAN_TDLS_SNAP_RFTYPE;
 746
 747         /* network header is after the ethernet header */
 748         skb_set_network_header(skb, ETH_HLEN);
 749
 750         switch (action_code) {
 751         case WLAN_TDLS_SETUP_REQUEST:
 752                 tf->category = WLAN_CATEGORY_TDLS;
 753                 tf->action_code = WLAN_TDLS_SETUP_REQUEST;
 754
 755                 skb_put(skb, sizeof(tf->u.setup_req));
 756                 tf->u.setup_req.dialog_token = dialog_token;
 757                 tf->u.setup_req.capability =
 758                         cpu_to_le16(ieee80211_get_tdls_sta_capab(sdata,
 759                                                                  status_code));
 760                 break;
 761         case WLAN_TDLS_SETUP_RESPONSE:
 762                 tf->category = WLAN_CATEGORY_TDLS;
 763                 tf->action_code = WLAN_TDLS_SETUP_RESPONSE;
 764
 765                 skb_put(skb, sizeof(tf->u.setup_resp));
 766                 tf->u.setup_resp.status_code = cpu_to_le16(status_code);
 767                 tf->u.setup_resp.dialog_token = dialog_token;
 768                 tf->u.setup_resp.capability =
 769                         cpu_to_le16(ieee80211_get_tdls_sta_capab(sdata,
 770                                                                  status_code));
 771                 break;
 772         case WLAN_TDLS_SETUP_CONFIRM:
 773                 tf->category = WLAN_CATEGORY_TDLS;
 774                 tf->action_code = WLAN_TDLS_SETUP_CONFIRM;
 775
 776                 skb_put(skb, sizeof(tf->u.setup_cfm));
 777                 tf->u.setup_cfm.status_code = cpu_to_le16(status_code);
 778                 tf->u.setup_cfm.dialog_token = dialog_token;
 779                 break;
 780         case WLAN_TDLS_TEARDOWN:
 781                 tf->category = WLAN_CATEGORY_TDLS;
 782                 tf->action_code = WLAN_TDLS_TEARDOWN;
 783
 784                 skb_put(skb, sizeof(tf->u.teardown));
 785                 tf->u.teardown.reason_code = cpu_to_le16(status_code);
 786                 break;
 787         case WLAN_TDLS_DISCOVERY_REQUEST:
 788                 tf->category = WLAN_CATEGORY_TDLS;
 789                 tf->action_code = WLAN_TDLS_DISCOVERY_REQUEST;
 790
 791                 skb_put(skb, sizeof(tf->u.discover_req));
 792                 tf->u.discover_req.dialog_token = dialog_token;
 793                 break;
 794         case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
 795                 tf->category = WLAN_CATEGORY_TDLS;
 796                 tf->action_code = WLAN_TDLS_CHANNEL_SWITCH_REQUEST;
 797
 798                 skb_put(skb, sizeof(tf->u.chan_switch_req));
 799                 break;
 800         case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
 801                 tf->category = WLAN_CATEGORY_TDLS;
 802                 tf->action_code = WLAN_TDLS_CHANNEL_SWITCH_RESPONSE;
 803
 804                 skb_put(skb, sizeof(tf->u.chan_switch_resp));
 805                 tf->u.chan_switch_resp.status_code = cpu_to_le16(status_code);
 806                 break;
 807         default:
 808                 return -EINVAL;
 809         }
 810
 811         return 0;
 812 }
 813
 814 static int
 815 ieee80211_prep_tdls_direct(struct wiphy *wiphy, struct net_device *dev,
 816                            const u8 *peer, u8 action_code, u8 dialog_token,
 817                            u16 status_code, struct sk_buff *skb)
 818 {
 819         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
 820         struct ieee80211_mgmt *mgmt;
 821
 822         mgmt = (void *)skb_put(skb, 24);
 823         memset(mgmt, 0, 24);
 824         memcpy(mgmt->da, peer, ETH_ALEN);
 825         memcpy(mgmt->sa, sdata->vif.addr, ETH_ALEN);
 826         memcpy(mgmt->bssid, sdata->u.mgd.bssid, ETH_ALEN);
 827
 828         mgmt->frame_control = cpu_to_le16(IEEE80211_FTYPE_MGMT |
 829                                           IEEE80211_STYPE_ACTION);
 830
 831         switch (action_code) {
 832         case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
 833                 skb_put(skb, 1 + sizeof(mgmt->u.action.u.tdls_discover_resp));
 834                 mgmt->u.action.category = WLAN_CATEGORY_PUBLIC;
 835                 mgmt->u.action.u.tdls_discover_resp.action_code =
 836                         WLAN_PUB_ACTION_TDLS_DISCOVER_RES;
 837                 mgmt->u.action.u.tdls_discover_resp.dialog_token =
 838                         dialog_token;
 839                 mgmt->u.action.u.tdls_discover_resp.capability =
 840                         cpu_to_le16(ieee80211_get_tdls_sta_capab(sdata,
 841                                                                  status_code));
 842                 break;
 843         default:
 844                 return -EINVAL;
 845         }
 846
 847         return 0;
 848 }
 849
 850 static struct sk_buff *
 851 ieee80211_tdls_build_mgmt_packet_data(struct ieee80211_sub_if_data *sdata,
 852                                       const u8 *peer, u8 action_code,
 853                                       u8 dialog_token, u16 status_code,
 854                                       bool initiator, const u8 *extra_ies,
 855                                       size_t extra_ies_len, u8 oper_class,
 856                                       struct cfg80211_chan_def *chandef)
 857 {
 858         struct ieee80211_local *local = sdata->local;
 859         struct sk_buff *skb;
 860         int ret;
 861
 862         skb = netdev_alloc_skb(sdata->dev,
 863                                local->hw.extra_tx_headroom +
 864                                max(sizeof(struct ieee80211_mgmt),
 865                                    sizeof(struct ieee80211_tdls_data)) +
 866                                50 + /* supported rates */
 867                                10 + /* ext capab */
 868                                26 + /* max(WMM-info, WMM-param) */
 869                                2 + max(sizeof(struct ieee80211_ht_cap),
 870                                        sizeof(struct ieee80211_ht_operation)) +
 871                                2 + max(sizeof(struct ieee80211_vht_cap),
 872                                        sizeof(struct ieee80211_vht_operation)) +
 873                                50 + /* supported channels */
 874                                3 + /* 40/20 BSS coex */
 875                                4 + /* AID */
 876                                4 + /* oper classes */
 877                                extra_ies_len +
 878                                sizeof(struct ieee80211_tdls_lnkie));
 879         if (!skb)
 880                 return NULL;
 881
 882         skb_reserve(skb, local->hw.extra_tx_headroom);
 883
 884         switch (action_code) {
 885         case WLAN_TDLS_SETUP_REQUEST:
 886         case WLAN_TDLS_SETUP_RESPONSE:
 887         case WLAN_TDLS_SETUP_CONFIRM:
 888         case WLAN_TDLS_TEARDOWN:
 889         case WLAN_TDLS_DISCOVERY_REQUEST:
 890         case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
 891         case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
 892                 ret = ieee80211_prep_tdls_encap_data(local->hw.wiphy,
 893                                                      sdata->dev, peer,
 894                                                      action_code, dialog_token,
 895                                                      status_code, skb);
 896                 break;
 897         case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
 898                 ret = ieee80211_prep_tdls_direct(local->hw.wiphy, sdata->dev,
 899                                                  peer, action_code,
 900                                                  dialog_token, status_code,
 901                                                  skb);
 902                 break;
 903         default:
 904                 ret = -ENOTSUPP;
 905                 break;
 906         }
 907
 908         if (ret < 0)
 909                 goto fail;
 910
 911         ieee80211_tdls_add_ies(sdata, skb, peer, action_code, status_code,
 912                                initiator, extra_ies, extra_ies_len, oper_class,
 913                                chandef);
 914         return skb;
 915
 916 fail:
 917         dev_kfree_skb(skb);
 918         return NULL;
 919 }
 920
 921 static int
 922 ieee80211_tdls_prep_mgmt_packet(struct wiphy *wiphy, struct net_device *dev,
 923                                 const u8 *peer, u8 action_code, u8 dialog_token,
 924                                 u16 status_code, u32 peer_capability,
 925                                 bool initiator, const u8 *extra_ies,
 926                                 size_t extra_ies_len, u8 oper_class,
 927                                 struct cfg80211_chan_def *chandef)
 928 {
 929         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
 930         struct sk_buff *skb = NULL;
 931         struct sta_info *sta;
 932         u32 flags = 0;
 933         int ret = 0;
 934
 935         rcu_read_lock();
 936         sta = sta_info_get(sdata, peer);
 937
 938         /* infer the initiator if we can, to support old userspace */
 939         switch (action_code) {
 940         case WLAN_TDLS_SETUP_REQUEST:
 941                 if (sta) {
 942                         set_sta_flag(sta, WLAN_STA_TDLS_INITIATOR);
 943                         sta->sta.tdls_initiator = false;
 944                 }
 945                 /* fall-through */
 946         case WLAN_TDLS_SETUP_CONFIRM:
 947         case WLAN_TDLS_DISCOVERY_REQUEST:
 948                 initiator = true;
 949                 break;
 950         case WLAN_TDLS_SETUP_RESPONSE:
 951                 /*
 952                  * In some testing scenarios, we send a request and response.
 953                  * Make the last packet sent take effect for the initiator
 954                  * value.
 955                  */
 956                 if (sta) {
 957                         clear_sta_flag(sta, WLAN_STA_TDLS_INITIATOR);
 958                         sta->sta.tdls_initiator = true;
 959                 }
 960                 /* fall-through */
 961         case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
 962                 initiator = false;
 963                 break;
 964         case WLAN_TDLS_TEARDOWN:
 965         case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
 966         case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
 967                 /* any value is ok */
 968                 break;
 969         default:
 970                 ret = -ENOTSUPP;
 971                 break;
 972         }
 973
 974         if (sta && test_sta_flag(sta, WLAN_STA_TDLS_INITIATOR))
 975                 initiator = true;
 976
 977         rcu_read_unlock();
 978         if (ret < 0)
 979                 goto fail;
 980
 981         skb = ieee80211_tdls_build_mgmt_packet_data(sdata, peer, action_code,
 982                                                     dialog_token, status_code,
 983                                                     initiator, extra_ies,
 984                                                     extra_ies_len, oper_class,
 985                                                     chandef);
 986         if (!skb) {
 987                 ret = -EINVAL;
 988                 goto fail;
 989         }
 990
 991         if (action_code == WLAN_PUB_ACTION_TDLS_DISCOVER_RES) {
 992                 ieee80211_tx_skb(sdata, skb);
 993                 return 0;
 994         }
 995
 996         /*
 997          * According to 802.11z: Setup req/resp are sent in AC_BK, otherwise
 998          * we should default to AC_VI.
 999          */
1000         switch (action_code) {
1001         case WLAN_TDLS_SETUP_REQUEST:
1002         case WLAN_TDLS_SETUP_RESPONSE:
1003                 skb_set_queue_mapping(skb, IEEE80211_AC_BK);
1004                 skb->priority = 2;
1005                 break;
1006         default:
1007                 skb_set_queue_mapping(skb, IEEE80211_AC_VI);
1008                 skb->priority = 5;
1009                 break;
1010         }
1011
1012         /*
1013          * Set the WLAN_TDLS_TEARDOWN flag to indicate a teardown in progress.
1014          * Later, if no ACK is returned from peer, we will re-send the teardown
1015          * packet through the AP.
1016          */
1017         if ((action_code == WLAN_TDLS_TEARDOWN) &&
1018             ieee80211_hw_check(&sdata->local->hw, REPORTS_TX_ACK_STATUS)) {
1019                 bool try_resend; /* Should we keep skb for possible resend */
1020
1021                 /* If not sending directly to peer - no point in keeping skb */
1022                 rcu_read_lock();
1023                 sta = sta_info_get(sdata, peer);
1024                 try_resend = sta && test_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH);
1025                 rcu_read_unlock();
1026
1027                 spin_lock_bh(&sdata->u.mgd.teardown_lock);
1028                 if (try_resend && !sdata->u.mgd.teardown_skb) {
1029                         /* Mark it as requiring TX status callback  */
1030                         flags |= IEEE80211_TX_CTL_REQ_TX_STATUS |
1031                                  IEEE80211_TX_INTFL_MLME_CONN_TX;
1032
1033                         /*
1034                          * skb is copied since mac80211 will later set
1035                          * properties that might not be the same as the AP,
1036                          * such as encryption, QoS, addresses, etc.
1037                          *
1038                          * No problem if skb_copy() fails, so no need to check.
1039                          */
1040                         sdata->u.mgd.teardown_skb = skb_copy(skb, GFP_ATOMIC);
1041                         sdata->u.mgd.orig_teardown_skb = skb;
1042                 }
1043                 spin_unlock_bh(&sdata->u.mgd.teardown_lock);
1044         }
1045
1046         /* disable bottom halves when entering the Tx path */
1047         local_bh_disable();
1048         __ieee80211_subif_start_xmit(skb, dev, flags);
1049         local_bh_enable();
1050
1051         return ret;
1052
1053 fail:
1054         dev_kfree_skb(skb);
1055         return ret;
1056 }
1057
1058 static int
1059 ieee80211_tdls_mgmt_setup(struct wiphy *wiphy, struct net_device *dev,
1060                           const u8 *peer, u8 action_code, u8 dialog_token,
1061                           u16 status_code, u32 peer_capability, bool initiator,
1062                           const u8 *extra_ies, size_t extra_ies_len)
1063 {
1064         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1065         struct ieee80211_local *local = sdata->local;
1066         enum ieee80211_smps_mode smps_mode = sdata->u.mgd.driver_smps_mode;
1067         int ret;
1068
1069         /* don't support setup with forced SMPS mode that's not off */
1070         if (smps_mode != IEEE80211_SMPS_AUTOMATIC &&
1071             smps_mode != IEEE80211_SMPS_OFF) {
1072                 tdls_dbg(sdata, "Aborting TDLS setup due to SMPS mode %d\n",
1073                          smps_mode);
1074                 return -ENOTSUPP;
1075         }
1076
1077         mutex_lock(&local->mtx);
1078
1079         /* we don't support concurrent TDLS peer setups */
1080         if (!is_zero_ether_addr(sdata->u.mgd.tdls_peer) &&
1081             !ether_addr_equal(sdata->u.mgd.tdls_peer, peer)) {
1082                 ret = -EBUSY;
1083                 goto out_unlock;
1084         }
1085
1086         /*
1087          * make sure we have a STA representing the peer so we drop or buffer
1088          * non-TDLS-setup frames to the peer. We can't send other packets
1089          * during setup through the AP path.
1090          * Allow error packets to be sent - sometimes we don't even add a STA
1091          * before failing the setup.
1092          */
1093         if (status_code == 0) {
1094                 rcu_read_lock();
1095                 if (!sta_info_get(sdata, peer)) {
1096                         rcu_read_unlock();
1097                         ret = -ENOLINK;
1098                         goto out_unlock;
1099                 }
1100                 rcu_read_unlock();
1101         }
1102
1103         ieee80211_flush_queues(local, sdata, false);
1104         memcpy(sdata->u.mgd.tdls_peer, peer, ETH_ALEN);
1105         mutex_unlock(&local->mtx);
1106
1107         /* we cannot take the mutex while preparing the setup packet */
1108         ret = ieee80211_tdls_prep_mgmt_packet(wiphy, dev, peer, action_code,
1109                                               dialog_token, status_code,
1110                                               peer_capability, initiator,
1111                                               extra_ies, extra_ies_len, 0,
1112                                               NULL);
1113         if (ret < 0) {
1114                 mutex_lock(&local->mtx);
1115                 eth_zero_addr(sdata->u.mgd.tdls_peer);
1116                 mutex_unlock(&local->mtx);
1117                 return ret;
1118         }
1119
1120         ieee80211_queue_delayed_work(&sdata->local->hw,
1121                                      &sdata->u.mgd.tdls_peer_del_work,
1122                                      TDLS_PEER_SETUP_TIMEOUT);
1123         return 0;
1124
1125 out_unlock:
1126         mutex_unlock(&local->mtx);
1127         return ret;
1128 }
1129
1130 static int
1131 ieee80211_tdls_mgmt_teardown(struct wiphy *wiphy, struct net_device *dev,
1132                              const u8 *peer, u8 action_code, u8 dialog_token,
1133                              u16 status_code, u32 peer_capability,
1134                              bool initiator, const u8 *extra_ies,
1135                              size_t extra_ies_len)
1136 {
1137         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1138         struct ieee80211_local *local = sdata->local;
1139         struct sta_info *sta;
1140         int ret;
1141
1142         /*
1143          * No packets can be transmitted to the peer via the AP during setup -
1144          * the STA is set as a TDLS peer, but is not authorized.
1145          * During teardown, we prevent direct transmissions by stopping the
1146          * queues and flushing all direct packets.
1147          */
1148         ieee80211_stop_vif_queues(local, sdata,
1149                                   IEEE80211_QUEUE_STOP_REASON_TDLS_TEARDOWN);
1150         ieee80211_flush_queues(local, sdata, false);
1151
1152         ret = ieee80211_tdls_prep_mgmt_packet(wiphy, dev, peer, action_code,
1153                                               dialog_token, status_code,
1154                                               peer_capability, initiator,
1155                                               extra_ies, extra_ies_len, 0,
1156                                               NULL);
1157         if (ret < 0)
1158                 sdata_err(sdata, "Failed sending TDLS teardown packet %d\n",
1159                           ret);
1160
1161         /*
1162          * Remove the STA AUTH flag to force further traffic through the AP. If
1163          * the STA was unreachable, it was already removed.
1164          */
1165         rcu_read_lock();
1166         sta = sta_info_get(sdata, peer);
1167         if (sta)
1168                 clear_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH);
1169         rcu_read_unlock();
1170
1171         ieee80211_wake_vif_queues(local, sdata,
1172                                   IEEE80211_QUEUE_STOP_REASON_TDLS_TEARDOWN);
1173
1174         return 0;
1175 }
1176
1177 int ieee80211_tdls_mgmt(struct wiphy *wiphy, struct net_device *dev,
1178                         const u8 *peer, u8 action_code, u8 dialog_token,
1179                         u16 status_code, u32 peer_capability,
1180                         bool initiator, const u8 *extra_ies,
1181                         size_t extra_ies_len)
1182 {
1183         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1184         int ret;
1185
1186         if (!(wiphy->flags & WIPHY_FLAG_SUPPORTS_TDLS))
1187                 return -ENOTSUPP;
1188
1189         /* make sure we are in managed mode, and associated */
1190         if (sdata->vif.type != NL80211_IFTYPE_STATION ||
1191             !sdata->u.mgd.associated)
1192                 return -EINVAL;
1193
1194         switch (action_code) {
1195         case WLAN_TDLS_SETUP_REQUEST:
1196         case WLAN_TDLS_SETUP_RESPONSE:
1197                 ret = ieee80211_tdls_mgmt_setup(wiphy, dev, peer, action_code,
1198                                                 dialog_token, status_code,
1199                                                 peer_capability, initiator,
1200                                                 extra_ies, extra_ies_len);
1201                 break;
1202         case WLAN_TDLS_TEARDOWN:
1203                 ret = ieee80211_tdls_mgmt_teardown(wiphy, dev, peer,
1204                                                    action_code, dialog_token,
1205                                                    status_code,
1206                                                    peer_capability, initiator,
1207                                                    extra_ies, extra_ies_len);
1208                 break;
1209         case WLAN_TDLS_DISCOVERY_REQUEST:
1210                 /*
1211                  * Protect the discovery so we can hear the TDLS discovery
1212                  * response frame. It is transmitted directly and not buffered
1213                  * by the AP.
1214                  */
1215                 drv_mgd_protect_tdls_discover(sdata->local, sdata);
1216                 /* fall-through */
1217         case WLAN_TDLS_SETUP_CONFIRM:
1218         case WLAN_PUB_ACTION_TDLS_DISCOVER_RES:
1219                 /* no special handling */
1220                 ret = ieee80211_tdls_prep_mgmt_packet(wiphy, dev, peer,
1221                                                       action_code,
1222                                                       dialog_token,
1223                                                       status_code,
1224                                                       peer_capability,
1225                                                       initiator, extra_ies,
1226                                                       extra_ies_len, 0, NULL);
1227                 break;
1228         default:
1229                 ret = -EOPNOTSUPP;
1230                 break;
1231         }
1232
1233         tdls_dbg(sdata, "TDLS mgmt action %d peer %pM status %d\n",
1234                  action_code, peer, ret);
1235         return ret;
1236 }
1237
1238 static void iee80211_tdls_recalc_chanctx(struct ieee80211_sub_if_data *sdata)
1239 {
1240         struct ieee80211_local *local = sdata->local;
1241         struct ieee80211_chanctx_conf *conf;
1242         struct ieee80211_chanctx *ctx;
1243
1244         mutex_lock(&local->chanctx_mtx);
1245         conf = rcu_dereference_protected(sdata->vif.chanctx_conf,
1246                                          lockdep_is_held(&local->chanctx_mtx));
1247         if (conf) {
1248                 ctx = container_of(conf, struct ieee80211_chanctx, conf);
1249                 ieee80211_recalc_chanctx_chantype(local, ctx);
1250         }
1251         mutex_unlock(&local->chanctx_mtx);
1252 }
1253
1254 static int iee80211_tdls_have_ht_peers(struct ieee80211_sub_if_data *sdata)
1255 {
1256         struct sta_info *sta;
1257         bool result = false;
1258
1259         rcu_read_lock();
1260         list_for_each_entry_rcu(sta, &sdata->local->sta_list, list) {
1261                 if (!sta->sta.tdls || sta->sdata != sdata || !sta->uploaded ||
1262                     !test_sta_flag(sta, WLAN_STA_AUTHORIZED) ||
1263                     !test_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH) ||
1264                     !sta->sta.ht_cap.ht_supported)
1265                         continue;
1266                 result = true;
1267                 break;
1268         }
1269         rcu_read_unlock();
1270
1271         return result;
1272 }
1273
1274 static void
1275 iee80211_tdls_recalc_ht_protection(struct ieee80211_sub_if_data *sdata,
1276                                    struct sta_info *sta)
1277 {
1278         struct ieee80211_if_managed *ifmgd = &sdata->u.mgd;
1279         bool tdls_ht;
1280         u16 protection = IEEE80211_HT_OP_MODE_PROTECTION_NONHT_MIXED |
1281                          IEEE80211_HT_OP_MODE_NON_GF_STA_PRSNT |
1282                          IEEE80211_HT_OP_MODE_NON_HT_STA_PRSNT;
1283         u16 opmode;
1284
1285         /* Nothing to do if the BSS connection uses HT */
1286         if (!(ifmgd->flags & IEEE80211_STA_DISABLE_HT))
1287                 return;
1288
1289         tdls_ht = (sta && sta->sta.ht_cap.ht_supported) ||
1290                   iee80211_tdls_have_ht_peers(sdata);
1291
1292         opmode = sdata->vif.bss_conf.ht_operation_mode;
1293
1294         if (tdls_ht)
1295                 opmode |= protection;
1296         else
1297                 opmode &= ~protection;
1298
1299         if (opmode == sdata->vif.bss_conf.ht_operation_mode)
1300                 return;
1301
1302         sdata->vif.bss_conf.ht_operation_mode = opmode;
1303         ieee80211_bss_info_change_notify(sdata, BSS_CHANGED_HT);
1304 }
1305
1306 int ieee80211_tdls_oper(struct wiphy *wiphy, struct net_device *dev,
1307                         const u8 *peer, enum nl80211_tdls_operation oper)
1308 {
1309         struct sta_info *sta;
1310         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1311         struct ieee80211_local *local = sdata->local;
1312         int ret;
1313
1314         if (!(wiphy->flags & WIPHY_FLAG_SUPPORTS_TDLS))
1315                 return -ENOTSUPP;
1316
1317         if (sdata->vif.type != NL80211_IFTYPE_STATION)
1318                 return -EINVAL;
1319
1320         switch (oper) {
1321         case NL80211_TDLS_ENABLE_LINK:
1322         case NL80211_TDLS_DISABLE_LINK:
1323                 break;
1324         case NL80211_TDLS_TEARDOWN:
1325         case NL80211_TDLS_SETUP:
1326         case NL80211_TDLS_DISCOVERY_REQ:
1327                 /* We don't support in-driver setup/teardown/discovery */
1328                 return -ENOTSUPP;
1329         }
1330
1331         /* protect possible bss_conf changes and avoid concurrency in
1332          * ieee80211_bss_info_change_notify()
1333          */
1334         sdata_lock(sdata);
1335         mutex_lock(&local->mtx);
1336         tdls_dbg(sdata, "TDLS oper %d peer %pM\n", oper, peer);
1337
1338         switch (oper) {
1339         case NL80211_TDLS_ENABLE_LINK:
1340                 if (sdata->vif.csa_active) {
1341                         tdls_dbg(sdata, "TDLS: disallow link during CSA\n");
1342                         ret = -EBUSY;
1343                         break;
1344                 }
1345
1346                 iee80211_tdls_recalc_chanctx(sdata);
1347
1348                 mutex_lock(&local->sta_mtx);
1349                 sta = sta_info_get(sdata, peer);
1350                 if (!sta) {
1351                         mutex_unlock(&local->sta_mtx);
1352                         ret = -ENOLINK;
1353                         break;
1354                 }
1355
1356                 iee80211_tdls_recalc_ht_protection(sdata, sta);
1357
1358                 set_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH);
1359                 mutex_unlock(&local->sta_mtx);
1360
1361                 WARN_ON_ONCE(is_zero_ether_addr(sdata->u.mgd.tdls_peer) ||
1362                              !ether_addr_equal(sdata->u.mgd.tdls_peer, peer));
1363                 ret = 0;
1364                 break;
1365         case NL80211_TDLS_DISABLE_LINK:
1366                 /*
1367                  * The teardown message in ieee80211_tdls_mgmt_teardown() was
1368                  * created while the queues were stopped, so it might still be
1369                  * pending. Before flushing the queues we need to be sure the
1370                  * message is handled by the tasklet handling pending messages,
1371                  * otherwise we might start destroying the station before
1372                  * sending the teardown packet.
1373                  * Note that this only forces the tasklet to flush pendings -
1374                  * not to stop the tasklet from rescheduling itself.
1375                  */
1376                 tasklet_kill(&local->tx_pending_tasklet);
1377                 /* flush a potentially queued teardown packet */
1378                 ieee80211_flush_queues(local, sdata, false);
1379
1380                 ret = sta_info_destroy_addr(sdata, peer);
1381
1382                 mutex_lock(&local->sta_mtx);
1383                 iee80211_tdls_recalc_ht_protection(sdata, NULL);
1384                 mutex_unlock(&local->sta_mtx);
1385
1386                 iee80211_tdls_recalc_chanctx(sdata);
1387                 break;
1388         default:
1389                 ret = -ENOTSUPP;
1390                 break;
1391         }
1392
1393         if (ret == 0 && ether_addr_equal(sdata->u.mgd.tdls_peer, peer)) {
1394                 cancel_delayed_work(&sdata->u.mgd.tdls_peer_del_work);
1395                 eth_zero_addr(sdata->u.mgd.tdls_peer);
1396         }
1397
1398         if (ret == 0)
1399                 ieee80211_queue_work(&sdata->local->hw,
1400                                      &sdata->u.mgd.request_smps_work);
1401
1402         mutex_unlock(&local->mtx);
1403         sdata_unlock(sdata);
1404         return ret;
1405 }
1406
1407 void ieee80211_tdls_oper_request(struct ieee80211_vif *vif, const u8 *peer,
1408                                  enum nl80211_tdls_operation oper,
1409                                  u16 reason_code, gfp_t gfp)
1410 {
1411         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1412
1413         if (vif->type != NL80211_IFTYPE_STATION || !vif->bss_conf.assoc) {
1414                 sdata_err(sdata, "Discarding TDLS oper %d - not STA or disconnected\n",
1415                           oper);
1416                 return;
1417         }
1418
1419         cfg80211_tdls_oper_request(sdata->dev, peer, oper, reason_code, gfp);
1420 }
1421 EXPORT_SYMBOL(ieee80211_tdls_oper_request);
1422
1423 static void
1424 iee80211_tdls_add_ch_switch_timing(u8 *buf, u16 switch_time, u16 switch_timeout)
1425 {
1426         struct ieee80211_ch_switch_timing *ch_sw;
1427
1428         *buf++ = WLAN_EID_CHAN_SWITCH_TIMING;
1429         *buf++ = sizeof(struct ieee80211_ch_switch_timing);
1430
1431         ch_sw = (void *)buf;
1432         ch_sw->switch_time = cpu_to_le16(switch_time);
1433         ch_sw->switch_timeout = cpu_to_le16(switch_timeout);
1434 }
1435
1436 /* find switch timing IE in SKB ready for Tx */
1437 static const u8 *ieee80211_tdls_find_sw_timing_ie(struct sk_buff *skb)
1438 {
1439         struct ieee80211_tdls_data *tf;
1440         const u8 *ie_start;
1441
1442         /*
1443          * Get the offset for the new location of the switch timing IE.
1444          * The SKB network header will now point to the "payload_type"
1445          * element of the TDLS data frame struct.
1446          */
1447         tf = container_of(skb->data + skb_network_offset(skb),
1448                           struct ieee80211_tdls_data, payload_type);
1449         ie_start = tf->u.chan_switch_req.variable;
1450         return cfg80211_find_ie(WLAN_EID_CHAN_SWITCH_TIMING, ie_start,
1451                                 skb->len - (ie_start - skb->data));
1452 }
1453
1454 static struct sk_buff *
1455 ieee80211_tdls_ch_sw_tmpl_get(struct sta_info *sta, u8 oper_class,
1456                               struct cfg80211_chan_def *chandef,
1457                               u32 *ch_sw_tm_ie_offset)
1458 {
1459         struct ieee80211_sub_if_data *sdata = sta->sdata;
1460         u8 extra_ies[2 + sizeof(struct ieee80211_sec_chan_offs_ie) +
1461                      2 + sizeof(struct ieee80211_ch_switch_timing)];
1462         int extra_ies_len = 2 + sizeof(struct ieee80211_ch_switch_timing);
1463         u8 *pos = extra_ies;
1464         struct sk_buff *skb;
1465
1466         /*
1467          * if chandef points to a wide channel add a Secondary-Channel
1468          * Offset information element
1469          */
1470         if (chandef->width == NL80211_CHAN_WIDTH_40) {
1471                 struct ieee80211_sec_chan_offs_ie *sec_chan_ie;
1472                 bool ht40plus;
1473
1474                 *pos++ = WLAN_EID_SECONDARY_CHANNEL_OFFSET;
1475                 *pos++ = sizeof(*sec_chan_ie);
1476                 sec_chan_ie = (void *)pos;
1477
1478                 ht40plus = cfg80211_get_chandef_type(chandef) ==
1479                                                         NL80211_CHAN_HT40PLUS;
1480                 sec_chan_ie->sec_chan_offs = ht40plus ?
1481                                              IEEE80211_HT_PARAM_CHA_SEC_ABOVE :
1482                                              IEEE80211_HT_PARAM_CHA_SEC_BELOW;
1483                 pos += sizeof(*sec_chan_ie);
1484
1485                 extra_ies_len += 2 + sizeof(struct ieee80211_sec_chan_offs_ie);
1486         }
1487
1488         /* just set the values to 0, this is a template */
1489         iee80211_tdls_add_ch_switch_timing(pos, 0, 0);
1490
1491         skb = ieee80211_tdls_build_mgmt_packet_data(sdata, sta->sta.addr,
1492                                               WLAN_TDLS_CHANNEL_SWITCH_REQUEST,
1493                                               0, 0, !sta->sta.tdls_initiator,
1494                                               extra_ies, extra_ies_len,
1495                                               oper_class, chandef);
1496         if (!skb)
1497                 return NULL;
1498
1499         skb = ieee80211_build_data_template(sdata, skb, 0);
1500         if (IS_ERR(skb)) {
1501                 tdls_dbg(sdata, "Failed building TDLS channel switch frame\n");
1502                 return NULL;
1503         }
1504
1505         if (ch_sw_tm_ie_offset) {
1506                 const u8 *tm_ie = ieee80211_tdls_find_sw_timing_ie(skb);
1507
1508                 if (!tm_ie) {
1509                         tdls_dbg(sdata, "No switch timing IE in TDLS switch\n");
1510                         dev_kfree_skb_any(skb);
1511                         return NULL;
1512                 }
1513
1514                 *ch_sw_tm_ie_offset = tm_ie - skb->data;
1515         }
1516
1517         tdls_dbg(sdata,
1518                  "TDLS channel switch request template for %pM ch %d width %d\n",
1519                  sta->sta.addr, chandef->chan->center_freq, chandef->width);
1520         return skb;
1521 }
1522
1523 int
1524 ieee80211_tdls_channel_switch(struct wiphy *wiphy, struct net_device *dev,
1525                               const u8 *addr, u8 oper_class,
1526                               struct cfg80211_chan_def *chandef)
1527 {
1528         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1529         struct ieee80211_local *local = sdata->local;
1530         struct sta_info *sta;
1531         struct sk_buff *skb = NULL;
1532         u32 ch_sw_tm_ie;
1533         int ret;
1534
1535         mutex_lock(&local->sta_mtx);
1536         sta = sta_info_get(sdata, addr);
1537         if (!sta) {
1538                 tdls_dbg(sdata,
1539                          "Invalid TDLS peer %pM for channel switch request\n",
1540                          addr);
1541                 ret = -ENOENT;
1542                 goto out;
1543         }
1544
1545         if (!test_sta_flag(sta, WLAN_STA_TDLS_CHAN_SWITCH)) {
1546                 tdls_dbg(sdata, "TDLS channel switch unsupported by %pM\n",
1547                          addr);
1548                 ret = -ENOTSUPP;
1549                 goto out;
1550         }
1551
1552         skb = ieee80211_tdls_ch_sw_tmpl_get(sta, oper_class, chandef,
1553                                             &ch_sw_tm_ie);
1554         if (!skb) {
1555                 ret = -ENOENT;
1556                 goto out;
1557         }
1558
1559         ret = drv_tdls_channel_switch(local, sdata, &sta->sta, oper_class,
1560                                       chandef, skb, ch_sw_tm_ie);
1561         if (!ret)
1562                 set_sta_flag(sta, WLAN_STA_TDLS_OFF_CHANNEL);
1563
1564 out:
1565         mutex_unlock(&local->sta_mtx);
1566         dev_kfree_skb_any(skb);
1567         return ret;
1568 }
1569
1570 void
1571 ieee80211_tdls_cancel_channel_switch(struct wiphy *wiphy,
1572                                      struct net_device *dev,
1573                                      const u8 *addr)
1574 {
1575         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1576         struct ieee80211_local *local = sdata->local;
1577         struct sta_info *sta;
1578
1579         mutex_lock(&local->sta_mtx);
1580         sta = sta_info_get(sdata, addr);
1581         if (!sta) {
1582                 tdls_dbg(sdata,
1583                          "Invalid TDLS peer %pM for channel switch cancel\n",
1584                          addr);
1585                 goto out;
1586         }
1587
1588         if (!test_sta_flag(sta, WLAN_STA_TDLS_OFF_CHANNEL)) {
1589                 tdls_dbg(sdata, "TDLS channel switch not initiated by %pM\n",
1590                          addr);
1591                 goto out;
1592         }
1593
1594         drv_tdls_cancel_channel_switch(local, sdata, &sta->sta);
1595         clear_sta_flag(sta, WLAN_STA_TDLS_OFF_CHANNEL);
1596
1597 out:
1598         mutex_unlock(&local->sta_mtx);
1599 }
1600
1601 static struct sk_buff *
1602 ieee80211_tdls_ch_sw_resp_tmpl_get(struct sta_info *sta,
1603                                    u32 *ch_sw_tm_ie_offset)
1604 {
1605         struct ieee80211_sub_if_data *sdata = sta->sdata;
1606         struct sk_buff *skb;
1607         u8 extra_ies[2 + sizeof(struct ieee80211_ch_switch_timing)];
1608
1609         /* initial timing are always zero in the template */
1610         iee80211_tdls_add_ch_switch_timing(extra_ies, 0, 0);
1611
1612         skb = ieee80211_tdls_build_mgmt_packet_data(sdata, sta->sta.addr,
1613                                         WLAN_TDLS_CHANNEL_SWITCH_RESPONSE,
1614                                         0, 0, !sta->sta.tdls_initiator,
1615                                         extra_ies, sizeof(extra_ies), 0, NULL);
1616         if (!skb)
1617                 return NULL;
1618
1619         skb = ieee80211_build_data_template(sdata, skb, 0);
1620         if (IS_ERR(skb)) {
1621                 tdls_dbg(sdata,
1622                          "Failed building TDLS channel switch resp frame\n");
1623                 return NULL;
1624         }
1625
1626         if (ch_sw_tm_ie_offset) {
1627                 const u8 *tm_ie = ieee80211_tdls_find_sw_timing_ie(skb);
1628
1629                 if (!tm_ie) {
1630                         tdls_dbg(sdata,
1631                                  "No switch timing IE in TDLS switch resp\n");
1632                         dev_kfree_skb_any(skb);
1633                         return NULL;
1634                 }
1635
1636                 *ch_sw_tm_ie_offset = tm_ie - skb->data;
1637         }
1638
1639         tdls_dbg(sdata, "TDLS get channel switch response template for %pM\n",
1640                  sta->sta.addr);
1641         return skb;
1642 }
1643
1644 static int
1645 ieee80211_process_tdls_channel_switch_resp(struct ieee80211_sub_if_data *sdata,
1646                                            struct sk_buff *skb)
1647 {
1648         struct ieee80211_local *local = sdata->local;
1649         struct ieee802_11_elems elems;
1650         struct sta_info *sta;
1651         struct ieee80211_tdls_data *tf = (void *)skb->data;
1652         bool local_initiator;
1653         struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1654         int baselen = offsetof(typeof(*tf), u.chan_switch_resp.variable);
1655         struct ieee80211_tdls_ch_sw_params params = {};
1656         int ret;
1657
1658         params.action_code = WLAN_TDLS_CHANNEL_SWITCH_RESPONSE;
1659         params.timestamp = rx_status->device_timestamp;
1660
1661         if (skb->len < baselen) {
1662                 tdls_dbg(sdata, "TDLS channel switch resp too short: %d\n",
1663                          skb->len);
1664                 return -EINVAL;
1665         }
1666
1667         mutex_lock(&local->sta_mtx);
1668         sta = sta_info_get(sdata, tf->sa);
1669         if (!sta || !test_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH)) {
1670                 tdls_dbg(sdata, "TDLS chan switch from non-peer sta %pM\n",
1671                          tf->sa);
1672                 ret = -EINVAL;
1673                 goto out;
1674         }
1675
1676         params.sta = &sta->sta;
1677         params.status = le16_to_cpu(tf->u.chan_switch_resp.status_code);
1678         if (params.status != 0) {
1679                 ret = 0;
1680                 goto call_drv;
1681         }
1682
1683         ieee802_11_parse_elems(tf->u.chan_switch_resp.variable,
1684                                skb->len - baselen, false, &elems);
1685         if (elems.parse_error) {
1686                 tdls_dbg(sdata, "Invalid IEs in TDLS channel switch resp\n");
1687                 ret = -EINVAL;
1688                 goto out;
1689         }
1690
1691         if (!elems.ch_sw_timing || !elems.lnk_id) {
1692                 tdls_dbg(sdata, "TDLS channel switch resp - missing IEs\n");
1693                 ret = -EINVAL;
1694                 goto out;
1695         }
1696
1697         /* validate the initiator is set correctly */
1698         local_initiator =
1699                 !memcmp(elems.lnk_id->init_sta, sdata->vif.addr, ETH_ALEN);
1700         if (local_initiator == sta->sta.tdls_initiator) {
1701                 tdls_dbg(sdata, "TDLS chan switch invalid lnk-id initiator\n");
1702                 ret = -EINVAL;
1703                 goto out;
1704         }
1705
1706         params.switch_time = le16_to_cpu(elems.ch_sw_timing->switch_time);
1707         params.switch_timeout = le16_to_cpu(elems.ch_sw_timing->switch_timeout);
1708
1709         params.tmpl_skb =
1710                 ieee80211_tdls_ch_sw_resp_tmpl_get(sta, &params.ch_sw_tm_ie);
1711         if (!params.tmpl_skb) {
1712                 ret = -ENOENT;
1713                 goto out;
1714         }
1715
1716 call_drv:
1717         drv_tdls_recv_channel_switch(sdata->local, sdata, &params);
1718
1719         tdls_dbg(sdata,
1720                  "TDLS channel switch response received from %pM status %d\n",
1721                  tf->sa, params.status);
1722
1723 out:
1724         mutex_unlock(&local->sta_mtx);
1725         dev_kfree_skb_any(params.tmpl_skb);
1726         return ret;
1727 }
1728
1729 static int
1730 ieee80211_process_tdls_channel_switch_req(struct ieee80211_sub_if_data *sdata,
1731                                           struct sk_buff *skb)
1732 {
1733         struct ieee80211_local *local = sdata->local;
1734         struct ieee802_11_elems elems;
1735         struct cfg80211_chan_def chandef;
1736         struct ieee80211_channel *chan;
1737         enum nl80211_channel_type chan_type;
1738         int freq;
1739         u8 target_channel, oper_class;
1740         bool local_initiator;
1741         struct sta_info *sta;
1742         enum ieee80211_band band;
1743         struct ieee80211_tdls_data *tf = (void *)skb->data;
1744         struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1745         int baselen = offsetof(typeof(*tf), u.chan_switch_req.variable);
1746         struct ieee80211_tdls_ch_sw_params params = {};
1747         int ret = 0;
1748
1749         params.action_code = WLAN_TDLS_CHANNEL_SWITCH_REQUEST;
1750         params.timestamp = rx_status->device_timestamp;
1751
1752         if (skb->len < baselen) {
1753                 tdls_dbg(sdata, "TDLS channel switch req too short: %d\n",
1754                          skb->len);
1755                 return -EINVAL;
1756         }
1757
1758         target_channel = tf->u.chan_switch_req.target_channel;
1759         oper_class = tf->u.chan_switch_req.oper_class;
1760
1761         /*
1762          * We can't easily infer the channel band. The operating class is
1763          * ambiguous - there are multiple tables (US/Europe/JP/Global). The
1764          * solution here is to treat channels with number >14 as 5GHz ones,
1765          * and specifically check for the (oper_class, channel) combinations
1766          * where this doesn't hold. These are thankfully unique according to
1767          * IEEE802.11-2012.
1768          * We consider only the 2GHz and 5GHz bands and 20MHz+ channels as
1769          * valid here.
1770          */
1771         if ((oper_class == 112 || oper_class == 2 || oper_class == 3 ||
1772              oper_class == 4 || oper_class == 5 || oper_class == 6) &&
1773              target_channel < 14)
1774                 band = IEEE80211_BAND_5GHZ;
1775         else
1776                 band = target_channel < 14 ? IEEE80211_BAND_2GHZ :
1777                                              IEEE80211_BAND_5GHZ;
1778
1779         freq = ieee80211_channel_to_frequency(target_channel, band);
1780         if (freq == 0) {
1781                 tdls_dbg(sdata, "Invalid channel in TDLS chan switch: %d\n",
1782                          target_channel);
1783                 return -EINVAL;
1784         }
1785
1786         chan = ieee80211_get_channel(sdata->local->hw.wiphy, freq);
1787         if (!chan) {
1788                 tdls_dbg(sdata,
1789                          "Unsupported channel for TDLS chan switch: %d\n",
1790                          target_channel);
1791                 return -EINVAL;
1792         }
1793
1794         ieee802_11_parse_elems(tf->u.chan_switch_req.variable,
1795                                skb->len - baselen, false, &elems);
1796         if (elems.parse_error) {
1797                 tdls_dbg(sdata, "Invalid IEs in TDLS channel switch req\n");
1798                 return -EINVAL;
1799         }
1800
1801         if (!elems.ch_sw_timing || !elems.lnk_id) {
1802                 tdls_dbg(sdata, "TDLS channel switch req - missing IEs\n");
1803                 return -EINVAL;
1804         }
1805
1806         if (!elems.sec_chan_offs) {
1807                 chan_type = NL80211_CHAN_HT20;
1808         } else {
1809                 switch (elems.sec_chan_offs->sec_chan_offs) {
1810                 case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
1811                         chan_type = NL80211_CHAN_HT40PLUS;
1812                         break;
1813                 case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
1814                         chan_type = NL80211_CHAN_HT40MINUS;
1815                         break;
1816                 default:
1817                         chan_type = NL80211_CHAN_HT20;
1818                         break;
1819                 }
1820         }
1821
1822         cfg80211_chandef_create(&chandef, chan, chan_type);
1823
1824         /* we will be active on the TDLS link */
1825         if (!cfg80211_reg_can_beacon_relax(sdata->local->hw.wiphy, &chandef,
1826                                            sdata->wdev.iftype)) {
1827                 tdls_dbg(sdata, "TDLS chan switch to forbidden channel\n");
1828                 return -EINVAL;
1829         }
1830
1831         mutex_lock(&local->sta_mtx);
1832         sta = sta_info_get(sdata, tf->sa);
1833         if (!sta || !test_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH)) {
1834                 tdls_dbg(sdata, "TDLS chan switch from non-peer sta %pM\n",
1835                          tf->sa);
1836                 ret = -EINVAL;
1837                 goto out;
1838         }
1839
1840         params.sta = &sta->sta;
1841
1842         /* validate the initiator is set correctly */
1843         local_initiator =
1844                 !memcmp(elems.lnk_id->init_sta, sdata->vif.addr, ETH_ALEN);
1845         if (local_initiator == sta->sta.tdls_initiator) {
1846                 tdls_dbg(sdata, "TDLS chan switch invalid lnk-id initiator\n");
1847                 ret = -EINVAL;
1848                 goto out;
1849         }
1850
1851         /* peer should have known better */
1852         if (!sta->sta.ht_cap.ht_supported && elems.sec_chan_offs &&
1853             elems.sec_chan_offs->sec_chan_offs) {
1854                 tdls_dbg(sdata, "TDLS chan switch - wide chan unsupported\n");
1855                 ret = -ENOTSUPP;
1856                 goto out;
1857         }
1858
1859         params.chandef = &chandef;
1860         params.switch_time = le16_to_cpu(elems.ch_sw_timing->switch_time);
1861         params.switch_timeout = le16_to_cpu(elems.ch_sw_timing->switch_timeout);
1862
1863         params.tmpl_skb =
1864                 ieee80211_tdls_ch_sw_resp_tmpl_get(sta,
1865                                                    &params.ch_sw_tm_ie);
1866         if (!params.tmpl_skb) {
1867                 ret = -ENOENT;
1868                 goto out;
1869         }
1870
1871         drv_tdls_recv_channel_switch(sdata->local, sdata, &params);
1872
1873         tdls_dbg(sdata,
1874                  "TDLS ch switch request received from %pM ch %d width %d\n",
1875                  tf->sa, params.chandef->chan->center_freq,
1876                  params.chandef->width);
1877 out:
1878         mutex_unlock(&local->sta_mtx);
1879         dev_kfree_skb_any(params.tmpl_skb);
1880         return ret;
1881 }
1882
1883 static void
1884 ieee80211_process_tdls_channel_switch(struct ieee80211_sub_if_data *sdata,
1885                                       struct sk_buff *skb)
1886 {
1887         struct ieee80211_tdls_data *tf = (void *)skb->data;
1888         struct wiphy *wiphy = sdata->local->hw.wiphy;
1889
1890         ASSERT_RTNL();
1891
1892         /* make sure the driver supports it */
1893         if (!(wiphy->features & NL80211_FEATURE_TDLS_CHANNEL_SWITCH))
1894                 return;
1895
1896         /* we want to access the entire packet */
1897         if (skb_linearize(skb))
1898                 return;
1899         /*
1900          * The packet/size was already validated by mac80211 Rx path, only look
1901          * at the action type.
1902          */
1903         switch (tf->action_code) {
1904         case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
1905                 ieee80211_process_tdls_channel_switch_req(sdata, skb);
1906                 break;
1907         case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
1908                 ieee80211_process_tdls_channel_switch_resp(sdata, skb);
1909                 break;
1910         default:
1911                 WARN_ON_ONCE(1);
1912                 return;
1913         }
1914 }
1915
1916 void ieee80211_teardown_tdls_peers(struct ieee80211_sub_if_data *sdata)
1917 {
1918         struct sta_info *sta;
1919         u16 reason = WLAN_REASON_TDLS_TEARDOWN_UNSPECIFIED;
1920
1921         rcu_read_lock();
1922         list_for_each_entry_rcu(sta, &sdata->local->sta_list, list) {
1923                 if (!sta->sta.tdls || sta->sdata != sdata || !sta->uploaded ||
1924                     !test_sta_flag(sta, WLAN_STA_AUTHORIZED))
1925                         continue;
1926
1927                 ieee80211_tdls_oper_request(&sdata->vif, sta->sta.addr,
1928                                             NL80211_TDLS_TEARDOWN, reason,
1929                                             GFP_ATOMIC);
1930         }
1931         rcu_read_unlock();
1932 }
1933
1934 void ieee80211_tdls_chsw_work(struct work_struct *wk)
1935 {
1936         struct ieee80211_local *local =
1937                 container_of(wk, struct ieee80211_local, tdls_chsw_work);
1938         struct ieee80211_sub_if_data *sdata;
1939         struct sk_buff *skb;
1940         struct ieee80211_tdls_data *tf;
1941
1942         rtnl_lock();
1943         while ((skb = skb_dequeue(&local->skb_queue_tdls_chsw))) {
1944                 tf = (struct ieee80211_tdls_data *)skb->data;
1945                 list_for_each_entry(sdata, &local->interfaces, list) {
1946                         if (!ieee80211_sdata_running(sdata) ||
1947                             sdata->vif.type != NL80211_IFTYPE_STATION ||
1948                             !ether_addr_equal(tf->da, sdata->vif.addr))
1949                                 continue;
1950
1951                         ieee80211_process_tdls_channel_switch(sdata, skb);
1952                         break;
1953                 }
1954
1955                 kfree_skb(skb);
1956         }
1957         rtnl_unlock();
1958 }