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(sdata->local->hw.wiphy,
 337                                         &uc, 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 int ieee80211_tdls_oper(struct wiphy *wiphy, struct net_device *dev,
1255                         const u8 *peer, enum nl80211_tdls_operation oper)
1256 {
1257         struct sta_info *sta;
1258         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1259         struct ieee80211_local *local = sdata->local;
1260         int ret;
1261
1262         if (!(wiphy->flags & WIPHY_FLAG_SUPPORTS_TDLS))
1263                 return -ENOTSUPP;
1264
1265         if (sdata->vif.type != NL80211_IFTYPE_STATION)
1266                 return -EINVAL;
1267
1268         switch (oper) {
1269         case NL80211_TDLS_ENABLE_LINK:
1270         case NL80211_TDLS_DISABLE_LINK:
1271                 break;
1272         case NL80211_TDLS_TEARDOWN:
1273         case NL80211_TDLS_SETUP:
1274         case NL80211_TDLS_DISCOVERY_REQ:
1275                 /* We don't support in-driver setup/teardown/discovery */
1276                 return -ENOTSUPP;
1277         }
1278
1279         mutex_lock(&local->mtx);
1280         tdls_dbg(sdata, "TDLS oper %d peer %pM\n", oper, peer);
1281
1282         switch (oper) {
1283         case NL80211_TDLS_ENABLE_LINK:
1284                 if (sdata->vif.csa_active) {
1285                         tdls_dbg(sdata, "TDLS: disallow link during CSA\n");
1286                         ret = -EBUSY;
1287                         break;
1288                 }
1289
1290                 iee80211_tdls_recalc_chanctx(sdata);
1291
1292                 rcu_read_lock();
1293                 sta = sta_info_get(sdata, peer);
1294                 if (!sta) {
1295                         rcu_read_unlock();
1296                         ret = -ENOLINK;
1297                         break;
1298                 }
1299
1300                 set_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH);
1301                 rcu_read_unlock();
1302
1303                 WARN_ON_ONCE(is_zero_ether_addr(sdata->u.mgd.tdls_peer) ||
1304                              !ether_addr_equal(sdata->u.mgd.tdls_peer, peer));
1305                 ret = 0;
1306                 break;
1307         case NL80211_TDLS_DISABLE_LINK:
1308                 /*
1309                  * The teardown message in ieee80211_tdls_mgmt_teardown() was
1310                  * created while the queues were stopped, so it might still be
1311                  * pending. Before flushing the queues we need to be sure the
1312                  * message is handled by the tasklet handling pending messages,
1313                  * otherwise we might start destroying the station before
1314                  * sending the teardown packet.
1315                  * Note that this only forces the tasklet to flush pendings -
1316                  * not to stop the tasklet from rescheduling itself.
1317                  */
1318                 tasklet_kill(&local->tx_pending_tasklet);
1319                 /* flush a potentially queued teardown packet */
1320                 ieee80211_flush_queues(local, sdata, false);
1321
1322                 ret = sta_info_destroy_addr(sdata, peer);
1323                 iee80211_tdls_recalc_chanctx(sdata);
1324                 break;
1325         default:
1326                 ret = -ENOTSUPP;
1327                 break;
1328         }
1329
1330         if (ret == 0 && ether_addr_equal(sdata->u.mgd.tdls_peer, peer)) {
1331                 cancel_delayed_work(&sdata->u.mgd.tdls_peer_del_work);
1332                 eth_zero_addr(sdata->u.mgd.tdls_peer);
1333         }
1334
1335         if (ret == 0)
1336                 ieee80211_queue_work(&sdata->local->hw,
1337                                      &sdata->u.mgd.request_smps_work);
1338
1339         mutex_unlock(&local->mtx);
1340         return ret;
1341 }
1342
1343 void ieee80211_tdls_oper_request(struct ieee80211_vif *vif, const u8 *peer,
1344                                  enum nl80211_tdls_operation oper,
1345                                  u16 reason_code, gfp_t gfp)
1346 {
1347         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1348
1349         if (vif->type != NL80211_IFTYPE_STATION || !vif->bss_conf.assoc) {
1350                 sdata_err(sdata, "Discarding TDLS oper %d - not STA or disconnected\n",
1351                           oper);
1352                 return;
1353         }
1354
1355         cfg80211_tdls_oper_request(sdata->dev, peer, oper, reason_code, gfp);
1356 }
1357 EXPORT_SYMBOL(ieee80211_tdls_oper_request);
1358
1359 static void
1360 iee80211_tdls_add_ch_switch_timing(u8 *buf, u16 switch_time, u16 switch_timeout)
1361 {
1362         struct ieee80211_ch_switch_timing *ch_sw;
1363
1364         *buf++ = WLAN_EID_CHAN_SWITCH_TIMING;
1365         *buf++ = sizeof(struct ieee80211_ch_switch_timing);
1366
1367         ch_sw = (void *)buf;
1368         ch_sw->switch_time = cpu_to_le16(switch_time);
1369         ch_sw->switch_timeout = cpu_to_le16(switch_timeout);
1370 }
1371
1372 /* find switch timing IE in SKB ready for Tx */
1373 static const u8 *ieee80211_tdls_find_sw_timing_ie(struct sk_buff *skb)
1374 {
1375         struct ieee80211_tdls_data *tf;
1376         const u8 *ie_start;
1377
1378         /*
1379          * Get the offset for the new location of the switch timing IE.
1380          * The SKB network header will now point to the "payload_type"
1381          * element of the TDLS data frame struct.
1382          */
1383         tf = container_of(skb->data + skb_network_offset(skb),
1384                           struct ieee80211_tdls_data, payload_type);
1385         ie_start = tf->u.chan_switch_req.variable;
1386         return cfg80211_find_ie(WLAN_EID_CHAN_SWITCH_TIMING, ie_start,
1387                                 skb->len - (ie_start - skb->data));
1388 }
1389
1390 static struct sk_buff *
1391 ieee80211_tdls_ch_sw_tmpl_get(struct sta_info *sta, u8 oper_class,
1392                               struct cfg80211_chan_def *chandef,
1393                               u32 *ch_sw_tm_ie_offset)
1394 {
1395         struct ieee80211_sub_if_data *sdata = sta->sdata;
1396         u8 extra_ies[2 + sizeof(struct ieee80211_sec_chan_offs_ie) +
1397                      2 + sizeof(struct ieee80211_ch_switch_timing)];
1398         int extra_ies_len = 2 + sizeof(struct ieee80211_ch_switch_timing);
1399         u8 *pos = extra_ies;
1400         struct sk_buff *skb;
1401
1402         /*
1403          * if chandef points to a wide channel add a Secondary-Channel
1404          * Offset information element
1405          */
1406         if (chandef->width == NL80211_CHAN_WIDTH_40) {
1407                 struct ieee80211_sec_chan_offs_ie *sec_chan_ie;
1408                 bool ht40plus;
1409
1410                 *pos++ = WLAN_EID_SECONDARY_CHANNEL_OFFSET;
1411                 *pos++ = sizeof(*sec_chan_ie);
1412                 sec_chan_ie = (void *)pos;
1413
1414                 ht40plus = cfg80211_get_chandef_type(chandef) ==
1415                                                         NL80211_CHAN_HT40PLUS;
1416                 sec_chan_ie->sec_chan_offs = ht40plus ?
1417                                              IEEE80211_HT_PARAM_CHA_SEC_ABOVE :
1418                                              IEEE80211_HT_PARAM_CHA_SEC_BELOW;
1419                 pos += sizeof(*sec_chan_ie);
1420
1421                 extra_ies_len += 2 + sizeof(struct ieee80211_sec_chan_offs_ie);
1422         }
1423
1424         /* just set the values to 0, this is a template */
1425         iee80211_tdls_add_ch_switch_timing(pos, 0, 0);
1426
1427         skb = ieee80211_tdls_build_mgmt_packet_data(sdata, sta->sta.addr,
1428                                               WLAN_TDLS_CHANNEL_SWITCH_REQUEST,
1429                                               0, 0, !sta->sta.tdls_initiator,
1430                                               extra_ies, extra_ies_len,
1431                                               oper_class, chandef);
1432         if (!skb)
1433                 return NULL;
1434
1435         skb = ieee80211_build_data_template(sdata, skb, 0);
1436         if (IS_ERR(skb)) {
1437                 tdls_dbg(sdata, "Failed building TDLS channel switch frame\n");
1438                 return NULL;
1439         }
1440
1441         if (ch_sw_tm_ie_offset) {
1442                 const u8 *tm_ie = ieee80211_tdls_find_sw_timing_ie(skb);
1443
1444                 if (!tm_ie) {
1445                         tdls_dbg(sdata, "No switch timing IE in TDLS switch\n");
1446                         dev_kfree_skb_any(skb);
1447                         return NULL;
1448                 }
1449
1450                 *ch_sw_tm_ie_offset = tm_ie - skb->data;
1451         }
1452
1453         tdls_dbg(sdata,
1454                  "TDLS channel switch request template for %pM ch %d width %d\n",
1455                  sta->sta.addr, chandef->chan->center_freq, chandef->width);
1456         return skb;
1457 }
1458
1459 int
1460 ieee80211_tdls_channel_switch(struct wiphy *wiphy, struct net_device *dev,
1461                               const u8 *addr, u8 oper_class,
1462                               struct cfg80211_chan_def *chandef)
1463 {
1464         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1465         struct ieee80211_local *local = sdata->local;
1466         struct sta_info *sta;
1467         struct sk_buff *skb = NULL;
1468         u32 ch_sw_tm_ie;
1469         int ret;
1470
1471         mutex_lock(&local->sta_mtx);
1472         sta = sta_info_get(sdata, addr);
1473         if (!sta) {
1474                 tdls_dbg(sdata,
1475                          "Invalid TDLS peer %pM for channel switch request\n",
1476                          addr);
1477                 ret = -ENOENT;
1478                 goto out;
1479         }
1480
1481         if (!test_sta_flag(sta, WLAN_STA_TDLS_CHAN_SWITCH)) {
1482                 tdls_dbg(sdata, "TDLS channel switch unsupported by %pM\n",
1483                          addr);
1484                 ret = -ENOTSUPP;
1485                 goto out;
1486         }
1487
1488         skb = ieee80211_tdls_ch_sw_tmpl_get(sta, oper_class, chandef,
1489                                             &ch_sw_tm_ie);
1490         if (!skb) {
1491                 ret = -ENOENT;
1492                 goto out;
1493         }
1494
1495         ret = drv_tdls_channel_switch(local, sdata, &sta->sta, oper_class,
1496                                       chandef, skb, ch_sw_tm_ie);
1497         if (!ret)
1498                 set_sta_flag(sta, WLAN_STA_TDLS_OFF_CHANNEL);
1499
1500 out:
1501         mutex_unlock(&local->sta_mtx);
1502         dev_kfree_skb_any(skb);
1503         return ret;
1504 }
1505
1506 void
1507 ieee80211_tdls_cancel_channel_switch(struct wiphy *wiphy,
1508                                      struct net_device *dev,
1509                                      const u8 *addr)
1510 {
1511         struct ieee80211_sub_if_data *sdata = IEEE80211_DEV_TO_SUB_IF(dev);
1512         struct ieee80211_local *local = sdata->local;
1513         struct sta_info *sta;
1514
1515         mutex_lock(&local->sta_mtx);
1516         sta = sta_info_get(sdata, addr);
1517         if (!sta) {
1518                 tdls_dbg(sdata,
1519                          "Invalid TDLS peer %pM for channel switch cancel\n",
1520                          addr);
1521                 goto out;
1522         }
1523
1524         if (!test_sta_flag(sta, WLAN_STA_TDLS_OFF_CHANNEL)) {
1525                 tdls_dbg(sdata, "TDLS channel switch not initiated by %pM\n",
1526                          addr);
1527                 goto out;
1528         }
1529
1530         drv_tdls_cancel_channel_switch(local, sdata, &sta->sta);
1531         clear_sta_flag(sta, WLAN_STA_TDLS_OFF_CHANNEL);
1532
1533 out:
1534         mutex_unlock(&local->sta_mtx);
1535 }
1536
1537 static struct sk_buff *
1538 ieee80211_tdls_ch_sw_resp_tmpl_get(struct sta_info *sta,
1539                                    u32 *ch_sw_tm_ie_offset)
1540 {
1541         struct ieee80211_sub_if_data *sdata = sta->sdata;
1542         struct sk_buff *skb;
1543         u8 extra_ies[2 + sizeof(struct ieee80211_ch_switch_timing)];
1544
1545         /* initial timing are always zero in the template */
1546         iee80211_tdls_add_ch_switch_timing(extra_ies, 0, 0);
1547
1548         skb = ieee80211_tdls_build_mgmt_packet_data(sdata, sta->sta.addr,
1549                                         WLAN_TDLS_CHANNEL_SWITCH_RESPONSE,
1550                                         0, 0, !sta->sta.tdls_initiator,
1551                                         extra_ies, sizeof(extra_ies), 0, NULL);
1552         if (!skb)
1553                 return NULL;
1554
1555         skb = ieee80211_build_data_template(sdata, skb, 0);
1556         if (IS_ERR(skb)) {
1557                 tdls_dbg(sdata,
1558                          "Failed building TDLS channel switch resp frame\n");
1559                 return NULL;
1560         }
1561
1562         if (ch_sw_tm_ie_offset) {
1563                 const u8 *tm_ie = ieee80211_tdls_find_sw_timing_ie(skb);
1564
1565                 if (!tm_ie) {
1566                         tdls_dbg(sdata,
1567                                  "No switch timing IE in TDLS switch resp\n");
1568                         dev_kfree_skb_any(skb);
1569                         return NULL;
1570                 }
1571
1572                 *ch_sw_tm_ie_offset = tm_ie - skb->data;
1573         }
1574
1575         tdls_dbg(sdata, "TDLS get channel switch response template for %pM\n",
1576                  sta->sta.addr);
1577         return skb;
1578 }
1579
1580 static int
1581 ieee80211_process_tdls_channel_switch_resp(struct ieee80211_sub_if_data *sdata,
1582                                            struct sk_buff *skb)
1583 {
1584         struct ieee80211_local *local = sdata->local;
1585         struct ieee802_11_elems elems;
1586         struct sta_info *sta;
1587         struct ieee80211_tdls_data *tf = (void *)skb->data;
1588         bool local_initiator;
1589         struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1590         int baselen = offsetof(typeof(*tf), u.chan_switch_resp.variable);
1591         struct ieee80211_tdls_ch_sw_params params = {};
1592         int ret;
1593
1594         params.action_code = WLAN_TDLS_CHANNEL_SWITCH_RESPONSE;
1595         params.timestamp = rx_status->device_timestamp;
1596
1597         if (skb->len < baselen) {
1598                 tdls_dbg(sdata, "TDLS channel switch resp too short: %d\n",
1599                          skb->len);
1600                 return -EINVAL;
1601         }
1602
1603         mutex_lock(&local->sta_mtx);
1604         sta = sta_info_get(sdata, tf->sa);
1605         if (!sta || !test_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH)) {
1606                 tdls_dbg(sdata, "TDLS chan switch from non-peer sta %pM\n",
1607                          tf->sa);
1608                 ret = -EINVAL;
1609                 goto out;
1610         }
1611
1612         params.sta = &sta->sta;
1613         params.status = le16_to_cpu(tf->u.chan_switch_resp.status_code);
1614         if (params.status != 0) {
1615                 ret = 0;
1616                 goto call_drv;
1617         }
1618
1619         ieee802_11_parse_elems(tf->u.chan_switch_resp.variable,
1620                                skb->len - baselen, false, &elems);
1621         if (elems.parse_error) {
1622                 tdls_dbg(sdata, "Invalid IEs in TDLS channel switch resp\n");
1623                 ret = -EINVAL;
1624                 goto out;
1625         }
1626
1627         if (!elems.ch_sw_timing || !elems.lnk_id) {
1628                 tdls_dbg(sdata, "TDLS channel switch resp - missing IEs\n");
1629                 ret = -EINVAL;
1630                 goto out;
1631         }
1632
1633         /* validate the initiator is set correctly */
1634         local_initiator =
1635                 !memcmp(elems.lnk_id->init_sta, sdata->vif.addr, ETH_ALEN);
1636         if (local_initiator == sta->sta.tdls_initiator) {
1637                 tdls_dbg(sdata, "TDLS chan switch invalid lnk-id initiator\n");
1638                 ret = -EINVAL;
1639                 goto out;
1640         }
1641
1642         params.switch_time = le16_to_cpu(elems.ch_sw_timing->switch_time);
1643         params.switch_timeout = le16_to_cpu(elems.ch_sw_timing->switch_timeout);
1644
1645         params.tmpl_skb =
1646                 ieee80211_tdls_ch_sw_resp_tmpl_get(sta, &params.ch_sw_tm_ie);
1647         if (!params.tmpl_skb) {
1648                 ret = -ENOENT;
1649                 goto out;
1650         }
1651
1652 call_drv:
1653         drv_tdls_recv_channel_switch(sdata->local, sdata, &params);
1654
1655         tdls_dbg(sdata,
1656                  "TDLS channel switch response received from %pM status %d\n",
1657                  tf->sa, params.status);
1658
1659 out:
1660         mutex_unlock(&local->sta_mtx);
1661         dev_kfree_skb_any(params.tmpl_skb);
1662         return ret;
1663 }
1664
1665 static int
1666 ieee80211_process_tdls_channel_switch_req(struct ieee80211_sub_if_data *sdata,
1667                                           struct sk_buff *skb)
1668 {
1669         struct ieee80211_local *local = sdata->local;
1670         struct ieee802_11_elems elems;
1671         struct cfg80211_chan_def chandef;
1672         struct ieee80211_channel *chan;
1673         enum nl80211_channel_type chan_type;
1674         int freq;
1675         u8 target_channel, oper_class;
1676         bool local_initiator;
1677         struct sta_info *sta;
1678         enum ieee80211_band band;
1679         struct ieee80211_tdls_data *tf = (void *)skb->data;
1680         struct ieee80211_rx_status *rx_status = IEEE80211_SKB_RXCB(skb);
1681         int baselen = offsetof(typeof(*tf), u.chan_switch_req.variable);
1682         struct ieee80211_tdls_ch_sw_params params = {};
1683         int ret = 0;
1684
1685         params.action_code = WLAN_TDLS_CHANNEL_SWITCH_REQUEST;
1686         params.timestamp = rx_status->device_timestamp;
1687
1688         if (skb->len < baselen) {
1689                 tdls_dbg(sdata, "TDLS channel switch req too short: %d\n",
1690                          skb->len);
1691                 return -EINVAL;
1692         }
1693
1694         target_channel = tf->u.chan_switch_req.target_channel;
1695         oper_class = tf->u.chan_switch_req.oper_class;
1696
1697         /*
1698          * We can't easily infer the channel band. The operating class is
1699          * ambiguous - there are multiple tables (US/Europe/JP/Global). The
1700          * solution here is to treat channels with number >14 as 5GHz ones,
1701          * and specifically check for the (oper_class, channel) combinations
1702          * where this doesn't hold. These are thankfully unique according to
1703          * IEEE802.11-2012.
1704          * We consider only the 2GHz and 5GHz bands and 20MHz+ channels as
1705          * valid here.
1706          */
1707         if ((oper_class == 112 || oper_class == 2 || oper_class == 3 ||
1708              oper_class == 4 || oper_class == 5 || oper_class == 6) &&
1709              target_channel < 14)
1710                 band = IEEE80211_BAND_5GHZ;
1711         else
1712                 band = target_channel < 14 ? IEEE80211_BAND_2GHZ :
1713                                              IEEE80211_BAND_5GHZ;
1714
1715         freq = ieee80211_channel_to_frequency(target_channel, band);
1716         if (freq == 0) {
1717                 tdls_dbg(sdata, "Invalid channel in TDLS chan switch: %d\n",
1718                          target_channel);
1719                 return -EINVAL;
1720         }
1721
1722         chan = ieee80211_get_channel(sdata->local->hw.wiphy, freq);
1723         if (!chan) {
1724                 tdls_dbg(sdata,
1725                          "Unsupported channel for TDLS chan switch: %d\n",
1726                          target_channel);
1727                 return -EINVAL;
1728         }
1729
1730         ieee802_11_parse_elems(tf->u.chan_switch_req.variable,
1731                                skb->len - baselen, false, &elems);
1732         if (elems.parse_error) {
1733                 tdls_dbg(sdata, "Invalid IEs in TDLS channel switch req\n");
1734                 return -EINVAL;
1735         }
1736
1737         if (!elems.ch_sw_timing || !elems.lnk_id) {
1738                 tdls_dbg(sdata, "TDLS channel switch req - missing IEs\n");
1739                 return -EINVAL;
1740         }
1741
1742         if (!elems.sec_chan_offs) {
1743                 chan_type = NL80211_CHAN_HT20;
1744         } else {
1745                 switch (elems.sec_chan_offs->sec_chan_offs) {
1746                 case IEEE80211_HT_PARAM_CHA_SEC_ABOVE:
1747                         chan_type = NL80211_CHAN_HT40PLUS;
1748                         break;
1749                 case IEEE80211_HT_PARAM_CHA_SEC_BELOW:
1750                         chan_type = NL80211_CHAN_HT40MINUS;
1751                         break;
1752                 default:
1753                         chan_type = NL80211_CHAN_HT20;
1754                         break;
1755                 }
1756         }
1757
1758         cfg80211_chandef_create(&chandef, chan, chan_type);
1759
1760         /* we will be active on the TDLS link */
1761         if (!cfg80211_reg_can_beacon_relax(sdata->local->hw.wiphy, &chandef,
1762                                            sdata->wdev.iftype)) {
1763                 tdls_dbg(sdata, "TDLS chan switch to forbidden channel\n");
1764                 return -EINVAL;
1765         }
1766
1767         mutex_lock(&local->sta_mtx);
1768         sta = sta_info_get(sdata, tf->sa);
1769         if (!sta || !test_sta_flag(sta, WLAN_STA_TDLS_PEER_AUTH)) {
1770                 tdls_dbg(sdata, "TDLS chan switch from non-peer sta %pM\n",
1771                          tf->sa);
1772                 ret = -EINVAL;
1773                 goto out;
1774         }
1775
1776         params.sta = &sta->sta;
1777
1778         /* validate the initiator is set correctly */
1779         local_initiator =
1780                 !memcmp(elems.lnk_id->init_sta, sdata->vif.addr, ETH_ALEN);
1781         if (local_initiator == sta->sta.tdls_initiator) {
1782                 tdls_dbg(sdata, "TDLS chan switch invalid lnk-id initiator\n");
1783                 ret = -EINVAL;
1784                 goto out;
1785         }
1786
1787         /* peer should have known better */
1788         if (!sta->sta.ht_cap.ht_supported && elems.sec_chan_offs &&
1789             elems.sec_chan_offs->sec_chan_offs) {
1790                 tdls_dbg(sdata, "TDLS chan switch - wide chan unsupported\n");
1791                 ret = -ENOTSUPP;
1792                 goto out;
1793         }
1794
1795         params.chandef = &chandef;
1796         params.switch_time = le16_to_cpu(elems.ch_sw_timing->switch_time);
1797         params.switch_timeout = le16_to_cpu(elems.ch_sw_timing->switch_timeout);
1798
1799         params.tmpl_skb =
1800                 ieee80211_tdls_ch_sw_resp_tmpl_get(sta,
1801                                                    &params.ch_sw_tm_ie);
1802         if (!params.tmpl_skb) {
1803                 ret = -ENOENT;
1804                 goto out;
1805         }
1806
1807         drv_tdls_recv_channel_switch(sdata->local, sdata, &params);
1808
1809         tdls_dbg(sdata,
1810                  "TDLS ch switch request received from %pM ch %d width %d\n",
1811                  tf->sa, params.chandef->chan->center_freq,
1812                  params.chandef->width);
1813 out:
1814         mutex_unlock(&local->sta_mtx);
1815         dev_kfree_skb_any(params.tmpl_skb);
1816         return ret;
1817 }
1818
1819 static void
1820 ieee80211_process_tdls_channel_switch(struct ieee80211_sub_if_data *sdata,
1821                                       struct sk_buff *skb)
1822 {
1823         struct ieee80211_tdls_data *tf = (void *)skb->data;
1824         struct wiphy *wiphy = sdata->local->hw.wiphy;
1825
1826         ASSERT_RTNL();
1827
1828         /* make sure the driver supports it */
1829         if (!(wiphy->features & NL80211_FEATURE_TDLS_CHANNEL_SWITCH))
1830                 return;
1831
1832         /* we want to access the entire packet */
1833         if (skb_linearize(skb))
1834                 return;
1835         /*
1836          * The packet/size was already validated by mac80211 Rx path, only look
1837          * at the action type.
1838          */
1839         switch (tf->action_code) {
1840         case WLAN_TDLS_CHANNEL_SWITCH_REQUEST:
1841                 ieee80211_process_tdls_channel_switch_req(sdata, skb);
1842                 break;
1843         case WLAN_TDLS_CHANNEL_SWITCH_RESPONSE:
1844                 ieee80211_process_tdls_channel_switch_resp(sdata, skb);
1845                 break;
1846         default:
1847                 WARN_ON_ONCE(1);
1848                 return;
1849         }
1850 }
1851
1852 void ieee80211_teardown_tdls_peers(struct ieee80211_sub_if_data *sdata)
1853 {
1854         struct sta_info *sta;
1855         u16 reason = WLAN_REASON_TDLS_TEARDOWN_UNSPECIFIED;
1856
1857         rcu_read_lock();
1858         list_for_each_entry_rcu(sta, &sdata->local->sta_list, list) {
1859                 if (!sta->sta.tdls || sta->sdata != sdata || !sta->uploaded ||
1860                     !test_sta_flag(sta, WLAN_STA_AUTHORIZED))
1861                         continue;
1862
1863                 ieee80211_tdls_oper_request(&sdata->vif, sta->sta.addr,
1864                                             NL80211_TDLS_TEARDOWN, reason,
1865                                             GFP_ATOMIC);
1866         }
1867         rcu_read_unlock();
1868 }
1869
1870 void ieee80211_tdls_chsw_work(struct work_struct *wk)
1871 {
1872         struct ieee80211_local *local =
1873                 container_of(wk, struct ieee80211_local, tdls_chsw_work);
1874         struct ieee80211_sub_if_data *sdata;
1875         struct sk_buff *skb;
1876         struct ieee80211_tdls_data *tf;
1877
1878         rtnl_lock();
1879         while ((skb = skb_dequeue(&local->skb_queue_tdls_chsw))) {
1880                 tf = (struct ieee80211_tdls_data *)skb->data;
1881                 list_for_each_entry(sdata, &local->interfaces, list) {
1882                         if (!ieee80211_sdata_running(sdata) ||
1883                             sdata->vif.type != NL80211_IFTYPE_STATION ||
1884                             !ether_addr_equal(tf->da, sdata->vif.addr))
1885                                 continue;
1886
1887                         ieee80211_process_tdls_channel_switch(sdata, skb);
1888                         break;
1889                 }
1890
1891                 kfree_skb(skb);
1892         }
1893         rtnl_unlock();
1894 }