537a15719d45f1035b5cec2c98ff06939f82c2fe
[firefly-linux-kernel-4.4.55.git] / drivers / net / wireless / iwlwifi / mvm / mac80211.c
1 /******************************************************************************
2  *
3  * This file is provided under a dual BSD/GPLv2 license.  When using or
4  * redistributing this file, you may do so under either license.
5  *
6  * GPL LICENSE SUMMARY
7  *
8  * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
9  * Copyright(c) 2013 - 2014 Intel Mobile Communications GmbH
10  *
11  * This program is free software; you can redistribute it and/or modify
12  * it under the terms of version 2 of the GNU General Public License as
13  * published by the Free Software Foundation.
14  *
15  * This program is distributed in the hope that it will be useful, but
16  * WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  * General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program; if not, write to the Free Software
22  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110,
23  * USA
24  *
25  * The full GNU General Public License is included in this distribution
26  * in the file called COPYING.
27  *
28  * Contact Information:
29  *  Intel Linux Wireless <ilw@linux.intel.com>
30  * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
31  *
32  * BSD LICENSE
33  *
34  * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
35  * Copyright(c) 2013 - 2014 Intel Mobile Communications GmbH
36  * All rights reserved.
37  *
38  * Redistribution and use in source and binary forms, with or without
39  * modification, are permitted provided that the following conditions
40  * are met:
41  *
42  *  * Redistributions of source code must retain the above copyright
43  *    notice, this list of conditions and the following disclaimer.
44  *  * Redistributions in binary form must reproduce the above copyright
45  *    notice, this list of conditions and the following disclaimer in
46  *    the documentation and/or other materials provided with the
47  *    distribution.
48  *  * Neither the name Intel Corporation nor the names of its
49  *    contributors may be used to endorse or promote products derived
50  *    from this software without specific prior written permission.
51  *
52  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
53  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
54  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
55  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
56  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
57  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
58  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
59  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
60  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
61  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
62  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
63  *
64  *****************************************************************************/
65 #include <linux/kernel.h>
66 #include <linux/slab.h>
67 #include <linux/skbuff.h>
68 #include <linux/netdevice.h>
69 #include <linux/etherdevice.h>
70 #include <linux/ip.h>
71 #include <linux/if_arp.h>
72 #include <linux/devcoredump.h>
73 #include <net/mac80211.h>
74 #include <net/ieee80211_radiotap.h>
75 #include <net/tcp.h>
76
77 #include "iwl-op-mode.h"
78 #include "iwl-io.h"
79 #include "mvm.h"
80 #include "sta.h"
81 #include "time-event.h"
82 #include "iwl-eeprom-parse.h"
83 #include "iwl-phy-db.h"
84 #include "testmode.h"
85 #include "iwl-fw-error-dump.h"
86 #include "iwl-prph.h"
87 #include "iwl-csr.h"
88 #include "iwl-nvm-parse.h"
89
90 static const struct ieee80211_iface_limit iwl_mvm_limits[] = {
91         {
92                 .max = 1,
93                 .types = BIT(NL80211_IFTYPE_STATION),
94         },
95         {
96                 .max = 1,
97                 .types = BIT(NL80211_IFTYPE_AP) |
98                         BIT(NL80211_IFTYPE_P2P_CLIENT) |
99                         BIT(NL80211_IFTYPE_P2P_GO),
100         },
101         {
102                 .max = 1,
103                 .types = BIT(NL80211_IFTYPE_P2P_DEVICE),
104         },
105 };
106
107 static const struct ieee80211_iface_combination iwl_mvm_iface_combinations[] = {
108         {
109                 .num_different_channels = 2,
110                 .max_interfaces = 3,
111                 .limits = iwl_mvm_limits,
112                 .n_limits = ARRAY_SIZE(iwl_mvm_limits),
113         },
114 };
115
116 #ifdef CONFIG_PM_SLEEP
117 static const struct nl80211_wowlan_tcp_data_token_feature
118 iwl_mvm_wowlan_tcp_token_feature = {
119         .min_len = 0,
120         .max_len = 255,
121         .bufsize = IWL_WOWLAN_REMOTE_WAKE_MAX_TOKENS,
122 };
123
124 static const struct wiphy_wowlan_tcp_support iwl_mvm_wowlan_tcp_support = {
125         .tok = &iwl_mvm_wowlan_tcp_token_feature,
126         .data_payload_max = IWL_WOWLAN_TCP_MAX_PACKET_LEN -
127                             sizeof(struct ethhdr) -
128                             sizeof(struct iphdr) -
129                             sizeof(struct tcphdr),
130         .data_interval_max = 65535, /* __le16 in API */
131         .wake_payload_max = IWL_WOWLAN_REMOTE_WAKE_MAX_PACKET_LEN -
132                             sizeof(struct ethhdr) -
133                             sizeof(struct iphdr) -
134                             sizeof(struct tcphdr),
135         .seq = true,
136 };
137 #endif
138
139 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
140 /*
141  * Use the reserved field to indicate magic values.
142  * these values will only be used internally by the driver,
143  * and won't make it to the fw (reserved will be 0).
144  * BC_FILTER_MAGIC_IP - configure the val of this attribute to
145  *      be the vif's ip address. in case there is not a single
146  *      ip address (0, or more than 1), this attribute will
147  *      be skipped.
148  * BC_FILTER_MAGIC_MAC - set the val of this attribute to
149  *      the LSB bytes of the vif's mac address
150  */
151 enum {
152         BC_FILTER_MAGIC_NONE = 0,
153         BC_FILTER_MAGIC_IP,
154         BC_FILTER_MAGIC_MAC,
155 };
156
157 static const struct iwl_fw_bcast_filter iwl_mvm_default_bcast_filters[] = {
158         {
159                 /* arp */
160                 .discard = 0,
161                 .frame_type = BCAST_FILTER_FRAME_TYPE_ALL,
162                 .attrs = {
163                         {
164                                 /* frame type - arp, hw type - ethernet */
165                                 .offset_type =
166                                         BCAST_FILTER_OFFSET_PAYLOAD_START,
167                                 .offset = sizeof(rfc1042_header),
168                                 .val = cpu_to_be32(0x08060001),
169                                 .mask = cpu_to_be32(0xffffffff),
170                         },
171                         {
172                                 /* arp dest ip */
173                                 .offset_type =
174                                         BCAST_FILTER_OFFSET_PAYLOAD_START,
175                                 .offset = sizeof(rfc1042_header) + 2 +
176                                           sizeof(struct arphdr) +
177                                           ETH_ALEN + sizeof(__be32) +
178                                           ETH_ALEN,
179                                 .mask = cpu_to_be32(0xffffffff),
180                                 /* mark it as special field */
181                                 .reserved1 = cpu_to_le16(BC_FILTER_MAGIC_IP),
182                         },
183                 },
184         },
185         {
186                 /* dhcp offer bcast */
187                 .discard = 0,
188                 .frame_type = BCAST_FILTER_FRAME_TYPE_IPV4,
189                 .attrs = {
190                         {
191                                 /* udp dest port - 68 (bootp client)*/
192                                 .offset_type = BCAST_FILTER_OFFSET_IP_END,
193                                 .offset = offsetof(struct udphdr, dest),
194                                 .val = cpu_to_be32(0x00440000),
195                                 .mask = cpu_to_be32(0xffff0000),
196                         },
197                         {
198                                 /* dhcp - lsb bytes of client hw address */
199                                 .offset_type = BCAST_FILTER_OFFSET_IP_END,
200                                 .offset = 38,
201                                 .mask = cpu_to_be32(0xffffffff),
202                                 /* mark it as special field */
203                                 .reserved1 = cpu_to_le16(BC_FILTER_MAGIC_MAC),
204                         },
205                 },
206         },
207         /* last filter must be empty */
208         {},
209 };
210 #endif
211
212 void iwl_mvm_ref(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
213 {
214         if (!iwl_mvm_is_d0i3_supported(mvm))
215                 return;
216
217         IWL_DEBUG_RPM(mvm, "Take mvm reference - type %d\n", ref_type);
218         spin_lock_bh(&mvm->refs_lock);
219         mvm->refs[ref_type]++;
220         spin_unlock_bh(&mvm->refs_lock);
221         iwl_trans_ref(mvm->trans);
222 }
223
224 void iwl_mvm_unref(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
225 {
226         if (!iwl_mvm_is_d0i3_supported(mvm))
227                 return;
228
229         IWL_DEBUG_RPM(mvm, "Leave mvm reference - type %d\n", ref_type);
230         spin_lock_bh(&mvm->refs_lock);
231         WARN_ON(!mvm->refs[ref_type]--);
232         spin_unlock_bh(&mvm->refs_lock);
233         iwl_trans_unref(mvm->trans);
234 }
235
236 static void iwl_mvm_unref_all_except(struct iwl_mvm *mvm,
237                                      enum iwl_mvm_ref_type except_ref)
238 {
239         int i, j;
240
241         if (!iwl_mvm_is_d0i3_supported(mvm))
242                 return;
243
244         spin_lock_bh(&mvm->refs_lock);
245         for (i = 0; i < IWL_MVM_REF_COUNT; i++) {
246                 if (except_ref == i || !mvm->refs[i])
247                         continue;
248
249                 IWL_DEBUG_RPM(mvm, "Cleanup: remove mvm ref type %d (%d)\n",
250                               i, mvm->refs[i]);
251                 for (j = 0; j < mvm->refs[i]; j++)
252                         iwl_trans_unref(mvm->trans);
253                 mvm->refs[i] = 0;
254         }
255         spin_unlock_bh(&mvm->refs_lock);
256 }
257
258 bool iwl_mvm_ref_taken(struct iwl_mvm *mvm)
259 {
260         int i;
261         bool taken = false;
262
263         if (!iwl_mvm_is_d0i3_supported(mvm))
264                 return true;
265
266         spin_lock_bh(&mvm->refs_lock);
267         for (i = 0; i < IWL_MVM_REF_COUNT; i++) {
268                 if (mvm->refs[i]) {
269                         taken = true;
270                         break;
271                 }
272         }
273         spin_unlock_bh(&mvm->refs_lock);
274
275         return taken;
276 }
277
278 int iwl_mvm_ref_sync(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
279 {
280         iwl_mvm_ref(mvm, ref_type);
281
282         if (!wait_event_timeout(mvm->d0i3_exit_waitq,
283                                 !test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status),
284                                 HZ)) {
285                 WARN_ON_ONCE(1);
286                 iwl_mvm_unref(mvm, ref_type);
287                 return -EIO;
288         }
289
290         return 0;
291 }
292
293 static void iwl_mvm_reset_phy_ctxts(struct iwl_mvm *mvm)
294 {
295         int i;
296
297         memset(mvm->phy_ctxts, 0, sizeof(mvm->phy_ctxts));
298         for (i = 0; i < NUM_PHY_CTX; i++) {
299                 mvm->phy_ctxts[i].id = i;
300                 mvm->phy_ctxts[i].ref = 0;
301         }
302 }
303
304 struct ieee80211_regdomain *iwl_mvm_get_regdomain(struct wiphy *wiphy,
305                                                   const char *alpha2,
306                                                   enum iwl_mcc_source src_id,
307                                                   bool *changed)
308 {
309         struct ieee80211_regdomain *regd = NULL;
310         struct ieee80211_hw *hw = wiphy_to_ieee80211_hw(wiphy);
311         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
312         struct iwl_mcc_update_resp *resp;
313
314         IWL_DEBUG_LAR(mvm, "Getting regdomain data for %s from FW\n", alpha2);
315
316         lockdep_assert_held(&mvm->mutex);
317
318         resp = iwl_mvm_update_mcc(mvm, alpha2, src_id);
319         if (IS_ERR_OR_NULL(resp)) {
320                 IWL_DEBUG_LAR(mvm, "Could not get update from FW %d\n",
321                               PTR_ERR_OR_ZERO(resp));
322                 goto out;
323         }
324
325         if (changed)
326                 *changed = (resp->status == MCC_RESP_NEW_CHAN_PROFILE);
327
328         regd = iwl_parse_nvm_mcc_info(mvm->trans->dev, mvm->cfg,
329                                       __le32_to_cpu(resp->n_channels),
330                                       resp->channels,
331                                       __le16_to_cpu(resp->mcc));
332         /* Store the return source id */
333         src_id = resp->source_id;
334         kfree(resp);
335         if (IS_ERR_OR_NULL(regd)) {
336                 IWL_DEBUG_LAR(mvm, "Could not get parse update from FW %d\n",
337                               PTR_ERR_OR_ZERO(regd));
338                 goto out;
339         }
340
341         IWL_DEBUG_LAR(mvm, "setting alpha2 from FW to %s (0x%x, 0x%x) src=%d\n",
342                       regd->alpha2, regd->alpha2[0], regd->alpha2[1], src_id);
343         mvm->lar_regdom_set = true;
344         mvm->mcc_src = src_id;
345
346 out:
347         return regd;
348 }
349
350 void iwl_mvm_update_changed_regdom(struct iwl_mvm *mvm)
351 {
352         bool changed;
353         struct ieee80211_regdomain *regd;
354
355         if (!iwl_mvm_is_lar_supported(mvm))
356                 return;
357
358         regd = iwl_mvm_get_current_regdomain(mvm, &changed);
359         if (!IS_ERR_OR_NULL(regd)) {
360                 /* only update the regulatory core if changed */
361                 if (changed)
362                         regulatory_set_wiphy_regd(mvm->hw->wiphy, regd);
363
364                 kfree(regd);
365         }
366 }
367
368 struct ieee80211_regdomain *iwl_mvm_get_current_regdomain(struct iwl_mvm *mvm,
369                                                           bool *changed)
370 {
371         return iwl_mvm_get_regdomain(mvm->hw->wiphy, "ZZ",
372                                      iwl_mvm_is_wifi_mcc_supported(mvm) ?
373                                      MCC_SOURCE_GET_CURRENT :
374                                      MCC_SOURCE_OLD_FW, changed);
375 }
376
377 int iwl_mvm_init_fw_regd(struct iwl_mvm *mvm)
378 {
379         enum iwl_mcc_source used_src;
380         struct ieee80211_regdomain *regd;
381         int ret;
382         bool changed;
383         const struct ieee80211_regdomain *r =
384                         rtnl_dereference(mvm->hw->wiphy->regd);
385
386         if (!r)
387                 return -ENOENT;
388
389         /* save the last source in case we overwrite it below */
390         used_src = mvm->mcc_src;
391         if (iwl_mvm_is_wifi_mcc_supported(mvm)) {
392                 /* Notify the firmware we support wifi location updates */
393                 regd = iwl_mvm_get_current_regdomain(mvm, NULL);
394                 if (!IS_ERR_OR_NULL(regd))
395                         kfree(regd);
396         }
397
398         /* Now set our last stored MCC and source */
399         regd = iwl_mvm_get_regdomain(mvm->hw->wiphy, r->alpha2, used_src,
400                                      &changed);
401         if (IS_ERR_OR_NULL(regd))
402                 return -EIO;
403
404         /* update cfg80211 if the regdomain was changed */
405         if (changed)
406                 ret = regulatory_set_wiphy_regd_sync_rtnl(mvm->hw->wiphy, regd);
407         else
408                 ret = 0;
409
410         kfree(regd);
411         return ret;
412 }
413
414 int iwl_mvm_mac_setup_register(struct iwl_mvm *mvm)
415 {
416         struct ieee80211_hw *hw = mvm->hw;
417         int num_mac, ret, i;
418         static const u32 mvm_ciphers[] = {
419                 WLAN_CIPHER_SUITE_WEP40,
420                 WLAN_CIPHER_SUITE_WEP104,
421                 WLAN_CIPHER_SUITE_TKIP,
422                 WLAN_CIPHER_SUITE_CCMP,
423         };
424
425         /* Tell mac80211 our characteristics */
426         ieee80211_hw_set(hw, SIGNAL_DBM);
427         ieee80211_hw_set(hw, SPECTRUM_MGMT);
428         ieee80211_hw_set(hw, REPORTS_TX_ACK_STATUS);
429         ieee80211_hw_set(hw, QUEUE_CONTROL);
430         ieee80211_hw_set(hw, WANT_MONITOR_VIF);
431         ieee80211_hw_set(hw, SUPPORTS_PS);
432         ieee80211_hw_set(hw, SUPPORTS_DYNAMIC_PS);
433         ieee80211_hw_set(hw, AMPDU_AGGREGATION);
434         ieee80211_hw_set(hw, TIMING_BEACON_ONLY);
435         ieee80211_hw_set(hw, CONNECTION_MONITOR);
436         ieee80211_hw_set(hw, CHANCTX_STA_CSA);
437         ieee80211_hw_set(hw, SUPPORT_FAST_XMIT);
438         ieee80211_hw_set(hw, SUPPORTS_CLONED_SKBS);
439
440         hw->queues = mvm->first_agg_queue;
441         hw->offchannel_tx_hw_queue = IWL_MVM_OFFCHANNEL_QUEUE;
442         hw->radiotap_mcs_details |= IEEE80211_RADIOTAP_MCS_HAVE_FEC |
443                                     IEEE80211_RADIOTAP_MCS_HAVE_STBC;
444         hw->radiotap_vht_details |= IEEE80211_RADIOTAP_VHT_KNOWN_STBC |
445                 IEEE80211_RADIOTAP_VHT_KNOWN_BEAMFORMED;
446         hw->rate_control_algorithm = "iwl-mvm-rs";
447         hw->uapsd_queues = IWL_MVM_UAPSD_QUEUES;
448         hw->uapsd_max_sp_len = IWL_UAPSD_MAX_SP;
449
450         BUILD_BUG_ON(ARRAY_SIZE(mvm->ciphers) < ARRAY_SIZE(mvm_ciphers) + 2);
451         memcpy(mvm->ciphers, mvm_ciphers, sizeof(mvm_ciphers));
452         hw->wiphy->n_cipher_suites = ARRAY_SIZE(mvm_ciphers);
453         hw->wiphy->cipher_suites = mvm->ciphers;
454
455         /*
456          * Enable 11w if advertised by firmware and software crypto
457          * is not enabled (as the firmware will interpret some mgmt
458          * packets, so enabling it with software crypto isn't safe)
459          */
460         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_MFP &&
461             !iwlwifi_mod_params.sw_crypto) {
462                 ieee80211_hw_set(hw, MFP_CAPABLE);
463                 mvm->ciphers[hw->wiphy->n_cipher_suites] =
464                         WLAN_CIPHER_SUITE_AES_CMAC;
465                 hw->wiphy->n_cipher_suites++;
466         }
467
468         /* currently FW API supports only one optional cipher scheme */
469         if (mvm->fw->cs[0].cipher) {
470                 mvm->hw->n_cipher_schemes = 1;
471                 mvm->hw->cipher_schemes = &mvm->fw->cs[0];
472                 mvm->ciphers[hw->wiphy->n_cipher_suites] =
473                         mvm->fw->cs[0].cipher;
474                 hw->wiphy->n_cipher_suites++;
475         }
476
477         ieee80211_hw_set(hw, SINGLE_SCAN_ON_ALL_BANDS);
478         hw->wiphy->features |=
479                 NL80211_FEATURE_SCHED_SCAN_RANDOM_MAC_ADDR |
480                 NL80211_FEATURE_SCAN_RANDOM_MAC_ADDR |
481                 NL80211_FEATURE_ND_RANDOM_MAC_ADDR;
482
483         hw->sta_data_size = sizeof(struct iwl_mvm_sta);
484         hw->vif_data_size = sizeof(struct iwl_mvm_vif);
485         hw->chanctx_data_size = sizeof(u16);
486
487         hw->wiphy->interface_modes = BIT(NL80211_IFTYPE_STATION) |
488                 BIT(NL80211_IFTYPE_P2P_CLIENT) |
489                 BIT(NL80211_IFTYPE_AP) |
490                 BIT(NL80211_IFTYPE_P2P_GO) |
491                 BIT(NL80211_IFTYPE_P2P_DEVICE) |
492                 BIT(NL80211_IFTYPE_ADHOC);
493
494         hw->wiphy->flags |= WIPHY_FLAG_IBSS_RSN;
495         hw->wiphy->regulatory_flags |= REGULATORY_ENABLE_RELAX_NO_IR;
496         if (iwl_mvm_is_lar_supported(mvm))
497                 hw->wiphy->regulatory_flags |= REGULATORY_WIPHY_SELF_MANAGED;
498         else
499                 hw->wiphy->regulatory_flags |= REGULATORY_CUSTOM_REG |
500                                                REGULATORY_DISABLE_BEACON_HINTS;
501
502         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_GO_UAPSD)
503                 hw->wiphy->flags |= WIPHY_FLAG_AP_UAPSD;
504
505         hw->wiphy->flags |= WIPHY_FLAG_HAS_CHANNEL_SWITCH;
506
507         hw->wiphy->iface_combinations = iwl_mvm_iface_combinations;
508         hw->wiphy->n_iface_combinations =
509                 ARRAY_SIZE(iwl_mvm_iface_combinations);
510
511         hw->wiphy->max_remain_on_channel_duration = 10000;
512         hw->max_listen_interval = IWL_CONN_MAX_LISTEN_INTERVAL;
513         /* we can compensate an offset of up to 3 channels = 15 MHz */
514         hw->wiphy->max_adj_channel_rssi_comp = 3 * 5;
515
516         /* Extract MAC address */
517         memcpy(mvm->addresses[0].addr, mvm->nvm_data->hw_addr, ETH_ALEN);
518         hw->wiphy->addresses = mvm->addresses;
519         hw->wiphy->n_addresses = 1;
520
521         /* Extract additional MAC addresses if available */
522         num_mac = (mvm->nvm_data->n_hw_addrs > 1) ?
523                 min(IWL_MVM_MAX_ADDRESSES, mvm->nvm_data->n_hw_addrs) : 1;
524
525         for (i = 1; i < num_mac; i++) {
526                 memcpy(mvm->addresses[i].addr, mvm->addresses[i-1].addr,
527                        ETH_ALEN);
528                 mvm->addresses[i].addr[5]++;
529                 hw->wiphy->n_addresses++;
530         }
531
532         iwl_mvm_reset_phy_ctxts(mvm);
533
534         hw->wiphy->max_scan_ie_len = iwl_mvm_max_scan_ie_len(mvm);
535
536         hw->wiphy->max_scan_ssids = PROBE_OPTION_MAX;
537
538         BUILD_BUG_ON(IWL_MVM_SCAN_STOPPING_MASK & IWL_MVM_SCAN_MASK);
539         BUILD_BUG_ON(IWL_MVM_MAX_UMAC_SCANS > HWEIGHT32(IWL_MVM_SCAN_MASK) ||
540                      IWL_MVM_MAX_LMAC_SCANS > HWEIGHT32(IWL_MVM_SCAN_MASK));
541
542         if (fw_has_capa(&mvm->fw->ucode_capa, IWL_UCODE_TLV_CAPA_UMAC_SCAN))
543                 mvm->max_scans = IWL_MVM_MAX_UMAC_SCANS;
544         else
545                 mvm->max_scans = IWL_MVM_MAX_LMAC_SCANS;
546
547         if (mvm->nvm_data->bands[IEEE80211_BAND_2GHZ].n_channels)
548                 hw->wiphy->bands[IEEE80211_BAND_2GHZ] =
549                         &mvm->nvm_data->bands[IEEE80211_BAND_2GHZ];
550         if (mvm->nvm_data->bands[IEEE80211_BAND_5GHZ].n_channels) {
551                 hw->wiphy->bands[IEEE80211_BAND_5GHZ] =
552                         &mvm->nvm_data->bands[IEEE80211_BAND_5GHZ];
553
554                 if (fw_has_capa(&mvm->fw->ucode_capa,
555                                 IWL_UCODE_TLV_CAPA_BEAMFORMER) &&
556                     fw_has_api(&mvm->fw->ucode_capa,
557                                IWL_UCODE_TLV_API_LQ_SS_PARAMS))
558                         hw->wiphy->bands[IEEE80211_BAND_5GHZ]->vht_cap.cap |=
559                                 IEEE80211_VHT_CAP_SU_BEAMFORMER_CAPABLE;
560         }
561
562         hw->wiphy->hw_version = mvm->trans->hw_id;
563
564         if (iwlmvm_mod_params.power_scheme != IWL_POWER_SCHEME_CAM)
565                 hw->wiphy->flags |= WIPHY_FLAG_PS_ON_BY_DEFAULT;
566         else
567                 hw->wiphy->flags &= ~WIPHY_FLAG_PS_ON_BY_DEFAULT;
568
569         hw->wiphy->flags |= WIPHY_FLAG_SUPPORTS_SCHED_SCAN;
570         hw->wiphy->max_sched_scan_ssids = PROBE_OPTION_MAX;
571         hw->wiphy->max_match_sets = IWL_SCAN_MAX_PROFILES;
572         /* we create the 802.11 header and zero length SSID IE. */
573         hw->wiphy->max_sched_scan_ie_len =
574                 SCAN_OFFLOAD_PROBE_REQ_SIZE - 24 - 2;
575
576         hw->wiphy->features |= NL80211_FEATURE_P2P_GO_CTWIN |
577                                NL80211_FEATURE_LOW_PRIORITY_SCAN |
578                                NL80211_FEATURE_P2P_GO_OPPPS |
579                                NL80211_FEATURE_DYNAMIC_SMPS |
580                                NL80211_FEATURE_STATIC_SMPS |
581                                NL80211_FEATURE_SUPPORTS_WMM_ADMISSION;
582
583         if (fw_has_capa(&mvm->fw->ucode_capa,
584                         IWL_UCODE_TLV_CAPA_TXPOWER_INSERTION_SUPPORT))
585                 hw->wiphy->features |= NL80211_FEATURE_TX_POWER_INSERTION;
586         if (fw_has_capa(&mvm->fw->ucode_capa,
587                         IWL_UCODE_TLV_CAPA_QUIET_PERIOD_SUPPORT))
588                 hw->wiphy->features |= NL80211_FEATURE_QUIET;
589
590         if (fw_has_capa(&mvm->fw->ucode_capa,
591                         IWL_UCODE_TLV_CAPA_DS_PARAM_SET_IE_SUPPORT))
592                 hw->wiphy->features |=
593                         NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES;
594
595         if (fw_has_capa(&mvm->fw->ucode_capa,
596                         IWL_UCODE_TLV_CAPA_WFA_TPC_REP_IE_SUPPORT))
597                 hw->wiphy->features |= NL80211_FEATURE_WFA_TPC_IE_IN_PROBES;
598
599         mvm->rts_threshold = IEEE80211_MAX_RTS_THRESHOLD;
600
601 #ifdef CONFIG_PM_SLEEP
602         if (iwl_mvm_is_d0i3_supported(mvm) &&
603             device_can_wakeup(mvm->trans->dev)) {
604                 mvm->wowlan.flags = WIPHY_WOWLAN_ANY;
605                 hw->wiphy->wowlan = &mvm->wowlan;
606         }
607
608         if (mvm->fw->img[IWL_UCODE_WOWLAN].sec[0].len &&
609             mvm->trans->ops->d3_suspend &&
610             mvm->trans->ops->d3_resume &&
611             device_can_wakeup(mvm->trans->dev)) {
612                 mvm->wowlan.flags |= WIPHY_WOWLAN_MAGIC_PKT |
613                                      WIPHY_WOWLAN_DISCONNECT |
614                                      WIPHY_WOWLAN_EAP_IDENTITY_REQ |
615                                      WIPHY_WOWLAN_RFKILL_RELEASE |
616                                      WIPHY_WOWLAN_NET_DETECT;
617                 if (!iwlwifi_mod_params.sw_crypto)
618                         mvm->wowlan.flags |= WIPHY_WOWLAN_SUPPORTS_GTK_REKEY |
619                                              WIPHY_WOWLAN_GTK_REKEY_FAILURE |
620                                              WIPHY_WOWLAN_4WAY_HANDSHAKE;
621
622                 mvm->wowlan.n_patterns = IWL_WOWLAN_MAX_PATTERNS;
623                 mvm->wowlan.pattern_min_len = IWL_WOWLAN_MIN_PATTERN_LEN;
624                 mvm->wowlan.pattern_max_len = IWL_WOWLAN_MAX_PATTERN_LEN;
625                 mvm->wowlan.max_nd_match_sets = IWL_SCAN_MAX_PROFILES;
626                 mvm->wowlan.tcp = &iwl_mvm_wowlan_tcp_support;
627                 hw->wiphy->wowlan = &mvm->wowlan;
628         }
629 #endif
630
631 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
632         /* assign default bcast filtering configuration */
633         mvm->bcast_filters = iwl_mvm_default_bcast_filters;
634 #endif
635
636         ret = iwl_mvm_leds_init(mvm);
637         if (ret)
638                 return ret;
639
640         if (fw_has_capa(&mvm->fw->ucode_capa,
641                         IWL_UCODE_TLV_CAPA_TDLS_SUPPORT)) {
642                 IWL_DEBUG_TDLS(mvm, "TDLS supported\n");
643                 hw->wiphy->flags |= WIPHY_FLAG_SUPPORTS_TDLS;
644                 ieee80211_hw_set(hw, TDLS_WIDER_BW);
645         }
646
647         if (fw_has_capa(&mvm->fw->ucode_capa,
648                         IWL_UCODE_TLV_CAPA_TDLS_CHANNEL_SWITCH)) {
649                 IWL_DEBUG_TDLS(mvm, "TDLS channel switch supported\n");
650                 hw->wiphy->features |= NL80211_FEATURE_TDLS_CHANNEL_SWITCH;
651         }
652
653         hw->netdev_features |= mvm->cfg->features;
654         if (!iwl_mvm_is_csum_supported(mvm))
655                 hw->netdev_features &= ~NETIF_F_RXCSUM;
656
657         ret = ieee80211_register_hw(mvm->hw);
658         if (ret)
659                 iwl_mvm_leds_exit(mvm);
660
661         return ret;
662 }
663
664 static bool iwl_mvm_defer_tx(struct iwl_mvm *mvm,
665                              struct ieee80211_sta *sta,
666                              struct sk_buff *skb)
667 {
668         struct iwl_mvm_sta *mvmsta;
669         bool defer = false;
670
671         /*
672          * double check the IN_D0I3 flag both before and after
673          * taking the spinlock, in order to prevent taking
674          * the spinlock when not needed.
675          */
676         if (likely(!test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status)))
677                 return false;
678
679         spin_lock(&mvm->d0i3_tx_lock);
680         /*
681          * testing the flag again ensures the skb dequeue
682          * loop (on d0i3 exit) hasn't run yet.
683          */
684         if (!test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status))
685                 goto out;
686
687         mvmsta = iwl_mvm_sta_from_mac80211(sta);
688         if (mvmsta->sta_id == IWL_MVM_STATION_COUNT ||
689             mvmsta->sta_id != mvm->d0i3_ap_sta_id)
690                 goto out;
691
692         __skb_queue_tail(&mvm->d0i3_tx, skb);
693         ieee80211_stop_queues(mvm->hw);
694
695         /* trigger wakeup */
696         iwl_mvm_ref(mvm, IWL_MVM_REF_TX);
697         iwl_mvm_unref(mvm, IWL_MVM_REF_TX);
698
699         defer = true;
700 out:
701         spin_unlock(&mvm->d0i3_tx_lock);
702         return defer;
703 }
704
705 static void iwl_mvm_mac_tx(struct ieee80211_hw *hw,
706                            struct ieee80211_tx_control *control,
707                            struct sk_buff *skb)
708 {
709         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
710         struct ieee80211_sta *sta = control->sta;
711         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
712         struct ieee80211_hdr *hdr = (void *)skb->data;
713
714         if (iwl_mvm_is_radio_killed(mvm)) {
715                 IWL_DEBUG_DROP(mvm, "Dropping - RF/CT KILL\n");
716                 goto drop;
717         }
718
719         if (IEEE80211_SKB_CB(skb)->hw_queue == IWL_MVM_OFFCHANNEL_QUEUE &&
720             !test_bit(IWL_MVM_STATUS_ROC_RUNNING, &mvm->status) &&
721             !test_bit(IWL_MVM_STATUS_ROC_AUX_RUNNING, &mvm->status))
722                 goto drop;
723
724         /* treat non-bufferable MMPDUs as broadcast if sta is sleeping */
725         if (unlikely(info->flags & IEEE80211_TX_CTL_NO_PS_BUFFER &&
726                      ieee80211_is_mgmt(hdr->frame_control) &&
727                      !ieee80211_is_deauth(hdr->frame_control) &&
728                      !ieee80211_is_disassoc(hdr->frame_control) &&
729                      !ieee80211_is_action(hdr->frame_control)))
730                 sta = NULL;
731
732         if (sta) {
733                 if (iwl_mvm_defer_tx(mvm, sta, skb))
734                         return;
735                 if (iwl_mvm_tx_skb(mvm, skb, sta))
736                         goto drop;
737                 return;
738         }
739
740         if (iwl_mvm_tx_skb_non_sta(mvm, skb))
741                 goto drop;
742         return;
743  drop:
744         ieee80211_free_txskb(hw, skb);
745 }
746
747 static inline bool iwl_enable_rx_ampdu(const struct iwl_cfg *cfg)
748 {
749         if (iwlwifi_mod_params.disable_11n & IWL_DISABLE_HT_RXAGG)
750                 return false;
751         return true;
752 }
753
754 static inline bool iwl_enable_tx_ampdu(const struct iwl_cfg *cfg)
755 {
756         if (iwlwifi_mod_params.disable_11n & IWL_DISABLE_HT_TXAGG)
757                 return false;
758         if (iwlwifi_mod_params.disable_11n & IWL_ENABLE_HT_TXAGG)
759                 return true;
760
761         /* enabled by default */
762         return true;
763 }
764
765 #define CHECK_BA_TRIGGER(_mvm, _trig, _tid_bm, _tid, _fmt...)   \
766         do {                                                    \
767                 if (!(le16_to_cpu(_tid_bm) & BIT(_tid)))        \
768                         break;                                  \
769                 iwl_mvm_fw_dbg_collect_trig(_mvm, _trig, _fmt); \
770         } while (0)
771
772 static void
773 iwl_mvm_ampdu_check_trigger(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
774                             struct ieee80211_sta *sta, u16 tid, u16 rx_ba_ssn,
775                             enum ieee80211_ampdu_mlme_action action)
776 {
777         struct iwl_fw_dbg_trigger_tlv *trig;
778         struct iwl_fw_dbg_trigger_ba *ba_trig;
779
780         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_BA))
781                 return;
782
783         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_BA);
784         ba_trig = (void *)trig->data;
785
786         if (!iwl_fw_dbg_trigger_check_stop(mvm, vif, trig))
787                 return;
788
789         switch (action) {
790         case IEEE80211_AMPDU_TX_OPERATIONAL: {
791                 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
792                 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
793
794                 CHECK_BA_TRIGGER(mvm, trig, ba_trig->tx_ba_start, tid,
795                                  "TX AGG START: MAC %pM tid %d ssn %d\n",
796                                  sta->addr, tid, tid_data->ssn);
797                 break;
798                 }
799         case IEEE80211_AMPDU_TX_STOP_CONT:
800                 CHECK_BA_TRIGGER(mvm, trig, ba_trig->tx_ba_stop, tid,
801                                  "TX AGG STOP: MAC %pM tid %d\n",
802                                  sta->addr, tid);
803                 break;
804         case IEEE80211_AMPDU_RX_START:
805                 CHECK_BA_TRIGGER(mvm, trig, ba_trig->rx_ba_start, tid,
806                                  "RX AGG START: MAC %pM tid %d ssn %d\n",
807                                  sta->addr, tid, rx_ba_ssn);
808                 break;
809         case IEEE80211_AMPDU_RX_STOP:
810                 CHECK_BA_TRIGGER(mvm, trig, ba_trig->rx_ba_stop, tid,
811                                  "RX AGG STOP: MAC %pM tid %d\n",
812                                  sta->addr, tid);
813                 break;
814         default:
815                 break;
816         }
817 }
818
819 static int iwl_mvm_mac_ampdu_action(struct ieee80211_hw *hw,
820                                     struct ieee80211_vif *vif,
821                                     enum ieee80211_ampdu_mlme_action action,
822                                     struct ieee80211_sta *sta, u16 tid,
823                                     u16 *ssn, u8 buf_size)
824 {
825         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
826         int ret;
827         bool tx_agg_ref = false;
828
829         IWL_DEBUG_HT(mvm, "A-MPDU action on addr %pM tid %d: action %d\n",
830                      sta->addr, tid, action);
831
832         if (!(mvm->nvm_data->sku_cap_11n_enable))
833                 return -EACCES;
834
835         /* return from D0i3 before starting a new Tx aggregation */
836         switch (action) {
837         case IEEE80211_AMPDU_TX_START:
838         case IEEE80211_AMPDU_TX_STOP_CONT:
839         case IEEE80211_AMPDU_TX_STOP_FLUSH:
840         case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
841         case IEEE80211_AMPDU_TX_OPERATIONAL:
842                 /*
843                  * for tx start, wait synchronously until D0i3 exit to
844                  * get the correct sequence number for the tid.
845                  * additionally, some other ampdu actions use direct
846                  * target access, which is not handled automatically
847                  * by the trans layer (unlike commands), so wait for
848                  * d0i3 exit in these cases as well.
849                  */
850                 ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_TX_AGG);
851                 if (ret)
852                         return ret;
853
854                 tx_agg_ref = true;
855                 break;
856         default:
857                 break;
858         }
859
860         mutex_lock(&mvm->mutex);
861
862         switch (action) {
863         case IEEE80211_AMPDU_RX_START:
864                 if (!iwl_enable_rx_ampdu(mvm->cfg)) {
865                         ret = -EINVAL;
866                         break;
867                 }
868                 ret = iwl_mvm_sta_rx_agg(mvm, sta, tid, *ssn, true);
869                 break;
870         case IEEE80211_AMPDU_RX_STOP:
871                 ret = iwl_mvm_sta_rx_agg(mvm, sta, tid, 0, false);
872                 break;
873         case IEEE80211_AMPDU_TX_START:
874                 if (!iwl_enable_tx_ampdu(mvm->cfg)) {
875                         ret = -EINVAL;
876                         break;
877                 }
878                 ret = iwl_mvm_sta_tx_agg_start(mvm, vif, sta, tid, ssn);
879                 break;
880         case IEEE80211_AMPDU_TX_STOP_CONT:
881                 ret = iwl_mvm_sta_tx_agg_stop(mvm, vif, sta, tid);
882                 break;
883         case IEEE80211_AMPDU_TX_STOP_FLUSH:
884         case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
885                 ret = iwl_mvm_sta_tx_agg_flush(mvm, vif, sta, tid);
886                 break;
887         case IEEE80211_AMPDU_TX_OPERATIONAL:
888                 ret = iwl_mvm_sta_tx_agg_oper(mvm, vif, sta, tid, buf_size);
889                 break;
890         default:
891                 WARN_ON_ONCE(1);
892                 ret = -EINVAL;
893                 break;
894         }
895
896         if (!ret) {
897                 u16 rx_ba_ssn = 0;
898
899                 if (action == IEEE80211_AMPDU_RX_START)
900                         rx_ba_ssn = *ssn;
901
902                 iwl_mvm_ampdu_check_trigger(mvm, vif, sta, tid,
903                                             rx_ba_ssn, action);
904         }
905         mutex_unlock(&mvm->mutex);
906
907         /*
908          * If the tid is marked as started, we won't use it for offloaded
909          * traffic on the next D0i3 entry. It's safe to unref.
910          */
911         if (tx_agg_ref)
912                 iwl_mvm_unref(mvm, IWL_MVM_REF_TX_AGG);
913
914         return ret;
915 }
916
917 static void iwl_mvm_cleanup_iterator(void *data, u8 *mac,
918                                      struct ieee80211_vif *vif)
919 {
920         struct iwl_mvm *mvm = data;
921         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
922
923         mvmvif->uploaded = false;
924         mvmvif->ap_sta_id = IWL_MVM_STATION_COUNT;
925
926         spin_lock_bh(&mvm->time_event_lock);
927         iwl_mvm_te_clear_data(mvm, &mvmvif->time_event_data);
928         spin_unlock_bh(&mvm->time_event_lock);
929
930         mvmvif->phy_ctxt = NULL;
931         memset(&mvmvif->bf_data, 0, sizeof(mvmvif->bf_data));
932 }
933
934 static ssize_t iwl_mvm_read_coredump(char *buffer, loff_t offset, size_t count,
935                                      const void *data, size_t datalen)
936 {
937         const struct iwl_mvm_dump_ptrs *dump_ptrs = data;
938         ssize_t bytes_read;
939         ssize_t bytes_read_trans;
940
941         if (offset < dump_ptrs->op_mode_len) {
942                 bytes_read = min_t(ssize_t, count,
943                                    dump_ptrs->op_mode_len - offset);
944                 memcpy(buffer, (u8 *)dump_ptrs->op_mode_ptr + offset,
945                        bytes_read);
946                 offset += bytes_read;
947                 count -= bytes_read;
948
949                 if (count == 0)
950                         return bytes_read;
951         } else {
952                 bytes_read = 0;
953         }
954
955         if (!dump_ptrs->trans_ptr)
956                 return bytes_read;
957
958         offset -= dump_ptrs->op_mode_len;
959         bytes_read_trans = min_t(ssize_t, count,
960                                  dump_ptrs->trans_ptr->len - offset);
961         memcpy(buffer + bytes_read,
962                (u8 *)dump_ptrs->trans_ptr->data + offset,
963                bytes_read_trans);
964
965         return bytes_read + bytes_read_trans;
966 }
967
968 static void iwl_mvm_free_coredump(const void *data)
969 {
970         const struct iwl_mvm_dump_ptrs *fw_error_dump = data;
971
972         vfree(fw_error_dump->op_mode_ptr);
973         vfree(fw_error_dump->trans_ptr);
974         kfree(fw_error_dump);
975 }
976
977 static void iwl_mvm_dump_fifos(struct iwl_mvm *mvm,
978                                struct iwl_fw_error_dump_data **dump_data)
979 {
980         struct iwl_fw_error_dump_fifo *fifo_hdr;
981         u32 *fifo_data;
982         u32 fifo_len;
983         unsigned long flags;
984         int i, j;
985
986         if (!iwl_trans_grab_nic_access(mvm->trans, false, &flags))
987                 return;
988
989         /* Pull RXF data from all RXFs */
990         for (i = 0; i < ARRAY_SIZE(mvm->shared_mem_cfg.rxfifo_size); i++) {
991                 /*
992                  * Keep aside the additional offset that might be needed for
993                  * next RXF
994                  */
995                 u32 offset_diff = RXF_DIFF_FROM_PREV * i;
996
997                 fifo_hdr = (void *)(*dump_data)->data;
998                 fifo_data = (void *)fifo_hdr->data;
999                 fifo_len = mvm->shared_mem_cfg.rxfifo_size[i];
1000
1001                 /* No need to try to read the data if the length is 0 */
1002                 if (fifo_len == 0)
1003                         continue;
1004
1005                 /* Add a TLV for the RXF */
1006                 (*dump_data)->type = cpu_to_le32(IWL_FW_ERROR_DUMP_RXF);
1007                 (*dump_data)->len = cpu_to_le32(fifo_len + sizeof(*fifo_hdr));
1008
1009                 fifo_hdr->fifo_num = cpu_to_le32(i);
1010                 fifo_hdr->available_bytes =
1011                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1012                                                         RXF_RD_D_SPACE +
1013                                                         offset_diff));
1014                 fifo_hdr->wr_ptr =
1015                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1016                                                         RXF_RD_WR_PTR +
1017                                                         offset_diff));
1018                 fifo_hdr->rd_ptr =
1019                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1020                                                         RXF_RD_RD_PTR +
1021                                                         offset_diff));
1022                 fifo_hdr->fence_ptr =
1023                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1024                                                         RXF_RD_FENCE_PTR +
1025                                                         offset_diff));
1026                 fifo_hdr->fence_mode =
1027                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1028                                                         RXF_SET_FENCE_MODE +
1029                                                         offset_diff));
1030
1031                 /* Lock fence */
1032                 iwl_trans_write_prph(mvm->trans,
1033                                      RXF_SET_FENCE_MODE + offset_diff, 0x1);
1034                 /* Set fence pointer to the same place like WR pointer */
1035                 iwl_trans_write_prph(mvm->trans,
1036                                      RXF_LD_WR2FENCE + offset_diff, 0x1);
1037                 /* Set fence offset */
1038                 iwl_trans_write_prph(mvm->trans,
1039                                      RXF_LD_FENCE_OFFSET_ADDR + offset_diff,
1040                                      0x0);
1041
1042                 /* Read FIFO */
1043                 fifo_len /= sizeof(u32); /* Size in DWORDS */
1044                 for (j = 0; j < fifo_len; j++)
1045                         fifo_data[j] = iwl_trans_read_prph(mvm->trans,
1046                                                          RXF_FIFO_RD_FENCE_INC +
1047                                                          offset_diff);
1048                 *dump_data = iwl_fw_error_next_data(*dump_data);
1049         }
1050
1051         /* Pull TXF data from all TXFs */
1052         for (i = 0; i < ARRAY_SIZE(mvm->shared_mem_cfg.txfifo_size); i++) {
1053                 /* Mark the number of TXF we're pulling now */
1054                 iwl_trans_write_prph(mvm->trans, TXF_LARC_NUM, i);
1055
1056                 fifo_hdr = (void *)(*dump_data)->data;
1057                 fifo_data = (void *)fifo_hdr->data;
1058                 fifo_len = mvm->shared_mem_cfg.txfifo_size[i];
1059
1060                 /* No need to try to read the data if the length is 0 */
1061                 if (fifo_len == 0)
1062                         continue;
1063
1064                 /* Add a TLV for the FIFO */
1065                 (*dump_data)->type = cpu_to_le32(IWL_FW_ERROR_DUMP_TXF);
1066                 (*dump_data)->len = cpu_to_le32(fifo_len + sizeof(*fifo_hdr));
1067
1068                 fifo_hdr->fifo_num = cpu_to_le32(i);
1069                 fifo_hdr->available_bytes =
1070                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1071                                                         TXF_FIFO_ITEM_CNT));
1072                 fifo_hdr->wr_ptr =
1073                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1074                                                         TXF_WR_PTR));
1075                 fifo_hdr->rd_ptr =
1076                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1077                                                         TXF_RD_PTR));
1078                 fifo_hdr->fence_ptr =
1079                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1080                                                         TXF_FENCE_PTR));
1081                 fifo_hdr->fence_mode =
1082                         cpu_to_le32(iwl_trans_read_prph(mvm->trans,
1083                                                         TXF_LOCK_FENCE));
1084
1085                 /* Set the TXF_READ_MODIFY_ADDR to TXF_WR_PTR */
1086                 iwl_trans_write_prph(mvm->trans, TXF_READ_MODIFY_ADDR,
1087                                      TXF_WR_PTR);
1088
1089                 /* Dummy-read to advance the read pointer to the head */
1090                 iwl_trans_read_prph(mvm->trans, TXF_READ_MODIFY_DATA);
1091
1092                 /* Read FIFO */
1093                 fifo_len /= sizeof(u32); /* Size in DWORDS */
1094                 for (j = 0; j < fifo_len; j++)
1095                         fifo_data[j] = iwl_trans_read_prph(mvm->trans,
1096                                                           TXF_READ_MODIFY_DATA);
1097                 *dump_data = iwl_fw_error_next_data(*dump_data);
1098         }
1099
1100         iwl_trans_release_nic_access(mvm->trans, &flags);
1101 }
1102
1103 void iwl_mvm_free_fw_dump_desc(struct iwl_mvm *mvm)
1104 {
1105         if (mvm->fw_dump_desc == &iwl_mvm_dump_desc_assert ||
1106             !mvm->fw_dump_desc)
1107                 return;
1108
1109         kfree(mvm->fw_dump_desc);
1110         mvm->fw_dump_desc = NULL;
1111 }
1112
1113 #define IWL8260_ICCM_OFFSET             0x44000 /* Only for B-step */
1114 #define IWL8260_ICCM_LEN                0xC000 /* Only for B-step */
1115
1116 void iwl_mvm_fw_error_dump(struct iwl_mvm *mvm)
1117 {
1118         struct iwl_fw_error_dump_file *dump_file;
1119         struct iwl_fw_error_dump_data *dump_data;
1120         struct iwl_fw_error_dump_info *dump_info;
1121         struct iwl_fw_error_dump_mem *dump_mem;
1122         struct iwl_fw_error_dump_trigger_desc *dump_trig;
1123         struct iwl_mvm_dump_ptrs *fw_error_dump;
1124         u32 sram_len, sram_ofs;
1125         u32 file_len, fifo_data_len = 0;
1126         u32 smem_len = mvm->cfg->smem_len;
1127         u32 sram2_len = mvm->cfg->dccm2_len;
1128         bool monitor_dump_only = false;
1129
1130         lockdep_assert_held(&mvm->mutex);
1131
1132         if (mvm->fw_dump_trig &&
1133             mvm->fw_dump_trig->mode & IWL_FW_DBG_TRIGGER_MONITOR_ONLY)
1134                 monitor_dump_only = true;
1135
1136         fw_error_dump = kzalloc(sizeof(*fw_error_dump), GFP_KERNEL);
1137         if (!fw_error_dump)
1138                 return;
1139
1140         /* SRAM - include stack CCM if driver knows the values for it */
1141         if (!mvm->cfg->dccm_offset || !mvm->cfg->dccm_len) {
1142                 const struct fw_img *img;
1143
1144                 img = &mvm->fw->img[mvm->cur_ucode];
1145                 sram_ofs = img->sec[IWL_UCODE_SECTION_DATA].offset;
1146                 sram_len = img->sec[IWL_UCODE_SECTION_DATA].len;
1147         } else {
1148                 sram_ofs = mvm->cfg->dccm_offset;
1149                 sram_len = mvm->cfg->dccm_len;
1150         }
1151
1152         /* reading RXF/TXF sizes */
1153         if (test_bit(STATUS_FW_ERROR, &mvm->trans->status)) {
1154                 struct iwl_mvm_shared_mem_cfg *mem_cfg = &mvm->shared_mem_cfg;
1155                 int i;
1156
1157                 fifo_data_len = 0;
1158
1159                 /* Count RXF size */
1160                 for (i = 0; i < ARRAY_SIZE(mem_cfg->rxfifo_size); i++) {
1161                         if (!mem_cfg->rxfifo_size[i])
1162                                 continue;
1163
1164                         /* Add header info */
1165                         fifo_data_len += mem_cfg->rxfifo_size[i] +
1166                                          sizeof(*dump_data) +
1167                                          sizeof(struct iwl_fw_error_dump_fifo);
1168                 }
1169
1170                 for (i = 0; i < ARRAY_SIZE(mem_cfg->txfifo_size); i++) {
1171                         if (!mem_cfg->txfifo_size[i])
1172                                 continue;
1173
1174                         /* Add header info */
1175                         fifo_data_len += mem_cfg->txfifo_size[i] +
1176                                          sizeof(*dump_data) +
1177                                          sizeof(struct iwl_fw_error_dump_fifo);
1178                 }
1179         }
1180
1181         file_len = sizeof(*dump_file) +
1182                    sizeof(*dump_data) * 2 +
1183                    sram_len + sizeof(*dump_mem) +
1184                    fifo_data_len +
1185                    sizeof(*dump_info);
1186
1187         /* Make room for the SMEM, if it exists */
1188         if (smem_len)
1189                 file_len += sizeof(*dump_data) + sizeof(*dump_mem) + smem_len;
1190
1191         /* Make room for the secondary SRAM, if it exists */
1192         if (sram2_len)
1193                 file_len += sizeof(*dump_data) + sizeof(*dump_mem) + sram2_len;
1194
1195         /* If we only want a monitor dump, reset the file length */
1196         if (monitor_dump_only) {
1197                 file_len = sizeof(*dump_file) + sizeof(*dump_data) +
1198                            sizeof(*dump_info);
1199         }
1200
1201         /*
1202          * In 8000 HW family B-step include the ICCM (which resides separately)
1203          */
1204         if (mvm->cfg->device_family == IWL_DEVICE_FAMILY_8000 &&
1205             CSR_HW_REV_STEP(mvm->trans->hw_rev) == SILICON_B_STEP)
1206                 file_len += sizeof(*dump_data) + sizeof(*dump_mem) +
1207                             IWL8260_ICCM_LEN;
1208
1209         if (mvm->fw_dump_desc)
1210                 file_len += sizeof(*dump_data) + sizeof(*dump_trig) +
1211                             mvm->fw_dump_desc->len;
1212
1213         dump_file = vzalloc(file_len);
1214         if (!dump_file) {
1215                 kfree(fw_error_dump);
1216                 iwl_mvm_free_fw_dump_desc(mvm);
1217                 return;
1218         }
1219
1220         fw_error_dump->op_mode_ptr = dump_file;
1221
1222         dump_file->barker = cpu_to_le32(IWL_FW_ERROR_DUMP_BARKER);
1223         dump_data = (void *)dump_file->data;
1224
1225         dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_DEV_FW_INFO);
1226         dump_data->len = cpu_to_le32(sizeof(*dump_info));
1227         dump_info = (void *) dump_data->data;
1228         dump_info->device_family =
1229                 mvm->cfg->device_family == IWL_DEVICE_FAMILY_7000 ?
1230                         cpu_to_le32(IWL_FW_ERROR_DUMP_FAMILY_7) :
1231                         cpu_to_le32(IWL_FW_ERROR_DUMP_FAMILY_8);
1232         dump_info->hw_step = cpu_to_le32(CSR_HW_REV_STEP(mvm->trans->hw_rev));
1233         memcpy(dump_info->fw_human_readable, mvm->fw->human_readable,
1234                sizeof(dump_info->fw_human_readable));
1235         strncpy(dump_info->dev_human_readable, mvm->cfg->name,
1236                 sizeof(dump_info->dev_human_readable));
1237         strncpy(dump_info->bus_human_readable, mvm->dev->bus->name,
1238                 sizeof(dump_info->bus_human_readable));
1239
1240         dump_data = iwl_fw_error_next_data(dump_data);
1241         /* We only dump the FIFOs if the FW is in error state */
1242         if (test_bit(STATUS_FW_ERROR, &mvm->trans->status))
1243                 iwl_mvm_dump_fifos(mvm, &dump_data);
1244
1245         if (mvm->fw_dump_desc) {
1246                 dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_ERROR_INFO);
1247                 dump_data->len = cpu_to_le32(sizeof(*dump_trig) +
1248                                              mvm->fw_dump_desc->len);
1249                 dump_trig = (void *)dump_data->data;
1250                 memcpy(dump_trig, &mvm->fw_dump_desc->trig_desc,
1251                        sizeof(*dump_trig) + mvm->fw_dump_desc->len);
1252
1253                 /* now we can free this copy */
1254                 iwl_mvm_free_fw_dump_desc(mvm);
1255                 dump_data = iwl_fw_error_next_data(dump_data);
1256         }
1257
1258         /* In case we only want monitor dump, skip to dump trasport data */
1259         if (monitor_dump_only)
1260                 goto dump_trans_data;
1261
1262         dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_MEM);
1263         dump_data->len = cpu_to_le32(sram_len + sizeof(*dump_mem));
1264         dump_mem = (void *)dump_data->data;
1265         dump_mem->type = cpu_to_le32(IWL_FW_ERROR_DUMP_MEM_SRAM);
1266         dump_mem->offset = cpu_to_le32(sram_ofs);
1267         iwl_trans_read_mem_bytes(mvm->trans, sram_ofs, dump_mem->data,
1268                                  sram_len);
1269
1270         if (smem_len) {
1271                 dump_data = iwl_fw_error_next_data(dump_data);
1272                 dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_MEM);
1273                 dump_data->len = cpu_to_le32(smem_len + sizeof(*dump_mem));
1274                 dump_mem = (void *)dump_data->data;
1275                 dump_mem->type = cpu_to_le32(IWL_FW_ERROR_DUMP_MEM_SMEM);
1276                 dump_mem->offset = cpu_to_le32(mvm->cfg->smem_offset);
1277                 iwl_trans_read_mem_bytes(mvm->trans, mvm->cfg->smem_offset,
1278                                          dump_mem->data, smem_len);
1279         }
1280
1281         if (sram2_len) {
1282                 dump_data = iwl_fw_error_next_data(dump_data);
1283                 dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_MEM);
1284                 dump_data->len = cpu_to_le32(sram2_len + sizeof(*dump_mem));
1285                 dump_mem = (void *)dump_data->data;
1286                 dump_mem->type = cpu_to_le32(IWL_FW_ERROR_DUMP_MEM_SRAM);
1287                 dump_mem->offset = cpu_to_le32(mvm->cfg->dccm2_offset);
1288                 iwl_trans_read_mem_bytes(mvm->trans, mvm->cfg->dccm2_offset,
1289                                          dump_mem->data, sram2_len);
1290         }
1291
1292         if (mvm->cfg->device_family == IWL_DEVICE_FAMILY_8000 &&
1293             CSR_HW_REV_STEP(mvm->trans->hw_rev) == SILICON_B_STEP) {
1294                 dump_data = iwl_fw_error_next_data(dump_data);
1295                 dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_MEM);
1296                 dump_data->len = cpu_to_le32(IWL8260_ICCM_LEN +
1297                                              sizeof(*dump_mem));
1298                 dump_mem = (void *)dump_data->data;
1299                 dump_mem->type = cpu_to_le32(IWL_FW_ERROR_DUMP_MEM_SRAM);
1300                 dump_mem->offset = cpu_to_le32(IWL8260_ICCM_OFFSET);
1301                 iwl_trans_read_mem_bytes(mvm->trans, IWL8260_ICCM_OFFSET,
1302                                          dump_mem->data, IWL8260_ICCM_LEN);
1303         }
1304
1305 dump_trans_data:
1306         fw_error_dump->trans_ptr = iwl_trans_dump_data(mvm->trans,
1307                                                        mvm->fw_dump_trig);
1308         fw_error_dump->op_mode_len = file_len;
1309         if (fw_error_dump->trans_ptr)
1310                 file_len += fw_error_dump->trans_ptr->len;
1311         dump_file->file_len = cpu_to_le32(file_len);
1312
1313         dev_coredumpm(mvm->trans->dev, THIS_MODULE, fw_error_dump, 0,
1314                       GFP_KERNEL, iwl_mvm_read_coredump, iwl_mvm_free_coredump);
1315
1316         mvm->fw_dump_trig = NULL;
1317         clear_bit(IWL_MVM_STATUS_DUMPING_FW_LOG, &mvm->status);
1318 }
1319
1320 struct iwl_mvm_dump_desc iwl_mvm_dump_desc_assert = {
1321         .trig_desc = {
1322                 .type = cpu_to_le32(FW_DBG_TRIGGER_FW_ASSERT),
1323         },
1324 };
1325
1326 static void iwl_mvm_restart_cleanup(struct iwl_mvm *mvm)
1327 {
1328         /* clear the D3 reconfig, we only need it to avoid dumping a
1329          * firmware coredump on reconfiguration, we shouldn't do that
1330          * on D3->D0 transition
1331          */
1332         if (!test_and_clear_bit(IWL_MVM_STATUS_D3_RECONFIG, &mvm->status)) {
1333                 mvm->fw_dump_desc = &iwl_mvm_dump_desc_assert;
1334                 iwl_mvm_fw_error_dump(mvm);
1335         }
1336
1337         /* cleanup all stale references (scan, roc), but keep the
1338          * ucode_down ref until reconfig is complete
1339          */
1340         iwl_mvm_unref_all_except(mvm, IWL_MVM_REF_UCODE_DOWN);
1341
1342         iwl_trans_stop_device(mvm->trans);
1343
1344         mvm->scan_status = 0;
1345         mvm->ps_disabled = false;
1346         mvm->calibrating = false;
1347
1348         /* just in case one was running */
1349         ieee80211_remain_on_channel_expired(mvm->hw);
1350
1351         /*
1352          * cleanup all interfaces, even inactive ones, as some might have
1353          * gone down during the HW restart
1354          */
1355         ieee80211_iterate_interfaces(mvm->hw, 0, iwl_mvm_cleanup_iterator, mvm);
1356
1357         mvm->p2p_device_vif = NULL;
1358         mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
1359
1360         iwl_mvm_reset_phy_ctxts(mvm);
1361         memset(mvm->sta_drained, 0, sizeof(mvm->sta_drained));
1362         memset(mvm->tfd_drained, 0, sizeof(mvm->tfd_drained));
1363         memset(&mvm->last_bt_notif, 0, sizeof(mvm->last_bt_notif));
1364         memset(&mvm->last_bt_notif_old, 0, sizeof(mvm->last_bt_notif_old));
1365         memset(&mvm->last_bt_ci_cmd, 0, sizeof(mvm->last_bt_ci_cmd));
1366         memset(&mvm->last_bt_ci_cmd_old, 0, sizeof(mvm->last_bt_ci_cmd_old));
1367         memset(&mvm->bt_ack_kill_msk, 0, sizeof(mvm->bt_ack_kill_msk));
1368         memset(&mvm->bt_cts_kill_msk, 0, sizeof(mvm->bt_cts_kill_msk));
1369
1370         ieee80211_wake_queues(mvm->hw);
1371
1372         /* clear any stale d0i3 state */
1373         clear_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status);
1374
1375         mvm->vif_count = 0;
1376         mvm->rx_ba_sessions = 0;
1377         mvm->fw_dbg_conf = FW_DBG_INVALID;
1378
1379         /* keep statistics ticking */
1380         iwl_mvm_accu_radio_stats(mvm);
1381 }
1382
1383 int __iwl_mvm_mac_start(struct iwl_mvm *mvm)
1384 {
1385         int ret;
1386
1387         lockdep_assert_held(&mvm->mutex);
1388
1389         /* Clean up some internal and mac80211 state on restart */
1390         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
1391                 iwl_mvm_restart_cleanup(mvm);
1392
1393         ret = iwl_mvm_up(mvm);
1394
1395         if (ret && test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
1396                 /* Something went wrong - we need to finish some cleanup
1397                  * that normally iwl_mvm_mac_restart_complete() below
1398                  * would do.
1399                  */
1400                 clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status);
1401                 iwl_mvm_d0i3_enable_tx(mvm, NULL);
1402         }
1403
1404         return ret;
1405 }
1406
1407 static int iwl_mvm_mac_start(struct ieee80211_hw *hw)
1408 {
1409         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1410         int ret;
1411
1412         /* Some hw restart cleanups must not hold the mutex */
1413         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
1414                 /*
1415                  * Make sure we are out of d0i3. This is needed
1416                  * to make sure the reference accounting is correct
1417                  * (and there is no stale d0i3_exit_work).
1418                  */
1419                 wait_event_timeout(mvm->d0i3_exit_waitq,
1420                                    !test_bit(IWL_MVM_STATUS_IN_D0I3,
1421                                              &mvm->status),
1422                                    HZ);
1423         }
1424
1425         mutex_lock(&mvm->mutex);
1426         ret = __iwl_mvm_mac_start(mvm);
1427         mutex_unlock(&mvm->mutex);
1428
1429         return ret;
1430 }
1431
1432 static void iwl_mvm_restart_complete(struct iwl_mvm *mvm)
1433 {
1434         int ret;
1435
1436         mutex_lock(&mvm->mutex);
1437
1438         clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status);
1439         iwl_mvm_d0i3_enable_tx(mvm, NULL);
1440         ret = iwl_mvm_update_quotas(mvm, true, NULL);
1441         if (ret)
1442                 IWL_ERR(mvm, "Failed to update quotas after restart (%d)\n",
1443                         ret);
1444
1445         /* allow transport/FW low power modes */
1446         iwl_mvm_unref(mvm, IWL_MVM_REF_UCODE_DOWN);
1447
1448         /*
1449          * If we have TDLS peers, remove them. We don't know the last seqno/PN
1450          * of packets the FW sent out, so we must reconnect.
1451          */
1452         iwl_mvm_teardown_tdls_peers(mvm);
1453
1454         mutex_unlock(&mvm->mutex);
1455 }
1456
1457 static void iwl_mvm_resume_complete(struct iwl_mvm *mvm)
1458 {
1459         if (!iwl_mvm_is_d0i3_supported(mvm))
1460                 return;
1461
1462         if (mvm->trans->d0i3_mode == IWL_D0I3_MODE_ON_SUSPEND)
1463                 if (!wait_event_timeout(mvm->d0i3_exit_waitq,
1464                                         !test_bit(IWL_MVM_STATUS_IN_D0I3,
1465                                                   &mvm->status),
1466                                         HZ))
1467                         WARN_ONCE(1, "D0i3 exit on resume timed out\n");
1468 }
1469
1470 static void
1471 iwl_mvm_mac_reconfig_complete(struct ieee80211_hw *hw,
1472                               enum ieee80211_reconfig_type reconfig_type)
1473 {
1474         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1475
1476         switch (reconfig_type) {
1477         case IEEE80211_RECONFIG_TYPE_RESTART:
1478                 iwl_mvm_restart_complete(mvm);
1479                 break;
1480         case IEEE80211_RECONFIG_TYPE_SUSPEND:
1481                 iwl_mvm_resume_complete(mvm);
1482                 break;
1483         }
1484 }
1485
1486 void __iwl_mvm_mac_stop(struct iwl_mvm *mvm)
1487 {
1488         lockdep_assert_held(&mvm->mutex);
1489
1490         /* firmware counters are obviously reset now, but we shouldn't
1491          * partially track so also clear the fw_reset_accu counters.
1492          */
1493         memset(&mvm->accu_radio_stats, 0, sizeof(mvm->accu_radio_stats));
1494
1495         /*
1496          * Disallow low power states when the FW is down by taking
1497          * the UCODE_DOWN ref. in case of ongoing hw restart the
1498          * ref is already taken, so don't take it again.
1499          */
1500         if (!test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
1501                 iwl_mvm_ref(mvm, IWL_MVM_REF_UCODE_DOWN);
1502
1503         /* async_handlers_wk is now blocked */
1504
1505         /*
1506          * The work item could be running or queued if the
1507          * ROC time event stops just as we get here.
1508          */
1509         flush_work(&mvm->roc_done_wk);
1510
1511         iwl_trans_stop_device(mvm->trans);
1512
1513         iwl_mvm_async_handlers_purge(mvm);
1514         /* async_handlers_list is empty and will stay empty: HW is stopped */
1515
1516         /* the fw is stopped, the aux sta is dead: clean up driver state */
1517         iwl_mvm_del_aux_sta(mvm);
1518
1519         /*
1520          * Clear IN_HW_RESTART flag when stopping the hw (as restart_complete()
1521          * won't be called in this case).
1522          * But make sure to cleanup interfaces that have gone down before/during
1523          * HW restart was requested.
1524          */
1525         if (test_and_clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
1526                 ieee80211_iterate_interfaces(mvm->hw, 0,
1527                                              iwl_mvm_cleanup_iterator, mvm);
1528
1529         /* We shouldn't have any UIDs still set.  Loop over all the UIDs to
1530          * make sure there's nothing left there and warn if any is found.
1531          */
1532         if (fw_has_capa(&mvm->fw->ucode_capa, IWL_UCODE_TLV_CAPA_UMAC_SCAN)) {
1533                 int i;
1534
1535                 for (i = 0; i < mvm->max_scans; i++) {
1536                         if (WARN_ONCE(mvm->scan_uid_status[i],
1537                                       "UMAC scan UID %d status was not cleaned\n",
1538                                       i))
1539                                 mvm->scan_uid_status[i] = 0;
1540                 }
1541         }
1542
1543         mvm->ucode_loaded = false;
1544 }
1545
1546 static void iwl_mvm_mac_stop(struct ieee80211_hw *hw)
1547 {
1548         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1549
1550         flush_work(&mvm->d0i3_exit_work);
1551         flush_work(&mvm->async_handlers_wk);
1552         cancel_delayed_work_sync(&mvm->fw_dump_wk);
1553         iwl_mvm_free_fw_dump_desc(mvm);
1554
1555         mutex_lock(&mvm->mutex);
1556         __iwl_mvm_mac_stop(mvm);
1557         mutex_unlock(&mvm->mutex);
1558
1559         /*
1560          * The worker might have been waiting for the mutex, let it run and
1561          * discover that its list is now empty.
1562          */
1563         cancel_work_sync(&mvm->async_handlers_wk);
1564 }
1565
1566 static struct iwl_mvm_phy_ctxt *iwl_mvm_get_free_phy_ctxt(struct iwl_mvm *mvm)
1567 {
1568         u16 i;
1569
1570         lockdep_assert_held(&mvm->mutex);
1571
1572         for (i = 0; i < NUM_PHY_CTX; i++)
1573                 if (!mvm->phy_ctxts[i].ref)
1574                         return &mvm->phy_ctxts[i];
1575
1576         IWL_ERR(mvm, "No available PHY context\n");
1577         return NULL;
1578 }
1579
1580 static int iwl_mvm_set_tx_power_old(struct iwl_mvm *mvm,
1581                                     struct ieee80211_vif *vif, s8 tx_power)
1582 {
1583         /* FW is in charge of regulatory enforcement */
1584         struct iwl_reduce_tx_power_cmd reduce_txpwr_cmd = {
1585                 .mac_context_id = iwl_mvm_vif_from_mac80211(vif)->id,
1586                 .pwr_restriction = cpu_to_le16(tx_power),
1587         };
1588
1589         return iwl_mvm_send_cmd_pdu(mvm, REDUCE_TX_POWER_CMD, 0,
1590                                     sizeof(reduce_txpwr_cmd),
1591                                     &reduce_txpwr_cmd);
1592 }
1593
1594 static int iwl_mvm_set_tx_power(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
1595                                 s16 tx_power)
1596 {
1597         struct iwl_dev_tx_power_cmd cmd = {
1598                 .set_mode = 0,
1599                 .mac_context_id =
1600                         cpu_to_le32(iwl_mvm_vif_from_mac80211(vif)->id),
1601                 .pwr_restriction = cpu_to_le16(8 * tx_power),
1602         };
1603
1604         if (!fw_has_api(&mvm->fw->ucode_capa, IWL_UCODE_TLV_API_TX_POWER_DEV))
1605                 return iwl_mvm_set_tx_power_old(mvm, vif, tx_power);
1606
1607         if (tx_power == IWL_DEFAULT_MAX_TX_POWER)
1608                 cmd.pwr_restriction = cpu_to_le16(IWL_DEV_MAX_TX_POWER);
1609
1610         return iwl_mvm_send_cmd_pdu(mvm, REDUCE_TX_POWER_CMD, 0,
1611                                     sizeof(cmd), &cmd);
1612 }
1613
1614 static int iwl_mvm_mac_add_interface(struct ieee80211_hw *hw,
1615                                      struct ieee80211_vif *vif)
1616 {
1617         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1618         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1619         int ret;
1620
1621         mvmvif->mvm = mvm;
1622
1623         /*
1624          * make sure D0i3 exit is completed, otherwise a target access
1625          * during tx queue configuration could be done when still in
1626          * D0i3 state.
1627          */
1628         ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_ADD_IF);
1629         if (ret)
1630                 return ret;
1631
1632         /*
1633          * Not much to do here. The stack will not allow interface
1634          * types or combinations that we didn't advertise, so we
1635          * don't really have to check the types.
1636          */
1637
1638         mutex_lock(&mvm->mutex);
1639
1640         /* make sure that beacon statistics don't go backwards with FW reset */
1641         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
1642                 mvmvif->beacon_stats.accu_num_beacons +=
1643                         mvmvif->beacon_stats.num_beacons;
1644
1645         /* Allocate resources for the MAC context, and add it to the fw  */
1646         ret = iwl_mvm_mac_ctxt_init(mvm, vif);
1647         if (ret)
1648                 goto out_unlock;
1649
1650         /* Counting number of interfaces is needed for legacy PM */
1651         if (vif->type != NL80211_IFTYPE_P2P_DEVICE)
1652                 mvm->vif_count++;
1653
1654         /*
1655          * The AP binding flow can be done only after the beacon
1656          * template is configured (which happens only in the mac80211
1657          * start_ap() flow), and adding the broadcast station can happen
1658          * only after the binding.
1659          * In addition, since modifying the MAC before adding a bcast
1660          * station is not allowed by the FW, delay the adding of MAC context to
1661          * the point where we can also add the bcast station.
1662          * In short: there's not much we can do at this point, other than
1663          * allocating resources :)
1664          */
1665         if (vif->type == NL80211_IFTYPE_AP ||
1666             vif->type == NL80211_IFTYPE_ADHOC) {
1667                 ret = iwl_mvm_alloc_bcast_sta(mvm, vif);
1668                 if (ret) {
1669                         IWL_ERR(mvm, "Failed to allocate bcast sta\n");
1670                         goto out_release;
1671                 }
1672
1673                 iwl_mvm_vif_dbgfs_register(mvm, vif);
1674                 goto out_unlock;
1675         }
1676
1677         mvmvif->features |= hw->netdev_features;
1678
1679         ret = iwl_mvm_mac_ctxt_add(mvm, vif);
1680         if (ret)
1681                 goto out_release;
1682
1683         ret = iwl_mvm_power_update_mac(mvm);
1684         if (ret)
1685                 goto out_remove_mac;
1686
1687         /* beacon filtering */
1688         ret = iwl_mvm_disable_beacon_filter(mvm, vif, 0);
1689         if (ret)
1690                 goto out_remove_mac;
1691
1692         if (!mvm->bf_allowed_vif &&
1693             vif->type == NL80211_IFTYPE_STATION && !vif->p2p) {
1694                 mvm->bf_allowed_vif = mvmvif;
1695                 vif->driver_flags |= IEEE80211_VIF_BEACON_FILTER |
1696                                      IEEE80211_VIF_SUPPORTS_CQM_RSSI;
1697         }
1698
1699         /*
1700          * P2P_DEVICE interface does not have a channel context assigned to it,
1701          * so a dedicated PHY context is allocated to it and the corresponding
1702          * MAC context is bound to it at this stage.
1703          */
1704         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1705
1706                 mvmvif->phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
1707                 if (!mvmvif->phy_ctxt) {
1708                         ret = -ENOSPC;
1709                         goto out_free_bf;
1710                 }
1711
1712                 iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
1713                 ret = iwl_mvm_binding_add_vif(mvm, vif);
1714                 if (ret)
1715                         goto out_unref_phy;
1716
1717                 ret = iwl_mvm_add_bcast_sta(mvm, vif);
1718                 if (ret)
1719                         goto out_unbind;
1720
1721                 /* Save a pointer to p2p device vif, so it can later be used to
1722                  * update the p2p device MAC when a GO is started/stopped */
1723                 mvm->p2p_device_vif = vif;
1724         }
1725
1726         iwl_mvm_vif_dbgfs_register(mvm, vif);
1727         goto out_unlock;
1728
1729  out_unbind:
1730         iwl_mvm_binding_remove_vif(mvm, vif);
1731  out_unref_phy:
1732         iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
1733  out_free_bf:
1734         if (mvm->bf_allowed_vif == mvmvif) {
1735                 mvm->bf_allowed_vif = NULL;
1736                 vif->driver_flags &= ~(IEEE80211_VIF_BEACON_FILTER |
1737                                        IEEE80211_VIF_SUPPORTS_CQM_RSSI);
1738         }
1739  out_remove_mac:
1740         mvmvif->phy_ctxt = NULL;
1741         iwl_mvm_mac_ctxt_remove(mvm, vif);
1742  out_release:
1743         if (vif->type != NL80211_IFTYPE_P2P_DEVICE)
1744                 mvm->vif_count--;
1745
1746         iwl_mvm_mac_ctxt_release(mvm, vif);
1747  out_unlock:
1748         mutex_unlock(&mvm->mutex);
1749
1750         iwl_mvm_unref(mvm, IWL_MVM_REF_ADD_IF);
1751
1752         return ret;
1753 }
1754
1755 static void iwl_mvm_prepare_mac_removal(struct iwl_mvm *mvm,
1756                                         struct ieee80211_vif *vif)
1757 {
1758         u32 tfd_msk = iwl_mvm_mac_get_queues_mask(vif);
1759
1760         if (tfd_msk) {
1761                 /*
1762                  * mac80211 first removes all the stations of the vif and
1763                  * then removes the vif. When it removes a station it also
1764                  * flushes the AMPDU session. So by now, all the AMPDU sessions
1765                  * of all the stations of this vif are closed, and the queues
1766                  * of these AMPDU sessions are properly closed.
1767                  * We still need to take care of the shared queues of the vif.
1768                  * Flush them here.
1769                  */
1770                 mutex_lock(&mvm->mutex);
1771                 iwl_mvm_flush_tx_path(mvm, tfd_msk, true);
1772                 mutex_unlock(&mvm->mutex);
1773
1774                 /*
1775                  * There are transports that buffer a few frames in the host.
1776                  * For these, the flush above isn't enough since while we were
1777                  * flushing, the transport might have sent more frames to the
1778                  * device. To solve this, wait here until the transport is
1779                  * empty. Technically, this could have replaced the flush
1780                  * above, but flush is much faster than draining. So flush
1781                  * first, and drain to make sure we have no frames in the
1782                  * transport anymore.
1783                  * If a station still had frames on the shared queues, it is
1784                  * already marked as draining, so to complete the draining, we
1785                  * just need to wait until the transport is empty.
1786                  */
1787                 iwl_trans_wait_tx_queue_empty(mvm->trans, tfd_msk);
1788         }
1789
1790         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1791                 /*
1792                  * Flush the ROC worker which will flush the OFFCHANNEL queue.
1793                  * We assume here that all the packets sent to the OFFCHANNEL
1794                  * queue are sent in ROC session.
1795                  */
1796                 flush_work(&mvm->roc_done_wk);
1797         } else {
1798                 /*
1799                  * By now, all the AC queues are empty. The AGG queues are
1800                  * empty too. We already got all the Tx responses for all the
1801                  * packets in the queues. The drain work can have been
1802                  * triggered. Flush it.
1803                  */
1804                 flush_work(&mvm->sta_drained_wk);
1805         }
1806 }
1807
1808 static void iwl_mvm_mac_remove_interface(struct ieee80211_hw *hw,
1809                                          struct ieee80211_vif *vif)
1810 {
1811         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1812         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1813
1814         iwl_mvm_prepare_mac_removal(mvm, vif);
1815
1816         mutex_lock(&mvm->mutex);
1817
1818         if (mvm->bf_allowed_vif == mvmvif) {
1819                 mvm->bf_allowed_vif = NULL;
1820                 vif->driver_flags &= ~(IEEE80211_VIF_BEACON_FILTER |
1821                                        IEEE80211_VIF_SUPPORTS_CQM_RSSI);
1822         }
1823
1824         iwl_mvm_vif_dbgfs_clean(mvm, vif);
1825
1826         /*
1827          * For AP/GO interface, the tear down of the resources allocated to the
1828          * interface is be handled as part of the stop_ap flow.
1829          */
1830         if (vif->type == NL80211_IFTYPE_AP ||
1831             vif->type == NL80211_IFTYPE_ADHOC) {
1832 #ifdef CONFIG_NL80211_TESTMODE
1833                 if (vif == mvm->noa_vif) {
1834                         mvm->noa_vif = NULL;
1835                         mvm->noa_duration = 0;
1836                 }
1837 #endif
1838                 iwl_mvm_dealloc_bcast_sta(mvm, vif);
1839                 goto out_release;
1840         }
1841
1842         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1843                 mvm->p2p_device_vif = NULL;
1844                 iwl_mvm_rm_bcast_sta(mvm, vif);
1845                 iwl_mvm_binding_remove_vif(mvm, vif);
1846                 iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
1847                 mvmvif->phy_ctxt = NULL;
1848         }
1849
1850         if (mvm->vif_count && vif->type != NL80211_IFTYPE_P2P_DEVICE)
1851                 mvm->vif_count--;
1852
1853         iwl_mvm_power_update_mac(mvm);
1854         iwl_mvm_mac_ctxt_remove(mvm, vif);
1855
1856 out_release:
1857         iwl_mvm_mac_ctxt_release(mvm, vif);
1858         mutex_unlock(&mvm->mutex);
1859 }
1860
1861 static int iwl_mvm_mac_config(struct ieee80211_hw *hw, u32 changed)
1862 {
1863         return 0;
1864 }
1865
1866 struct iwl_mvm_mc_iter_data {
1867         struct iwl_mvm *mvm;
1868         int port_id;
1869 };
1870
1871 static void iwl_mvm_mc_iface_iterator(void *_data, u8 *mac,
1872                                       struct ieee80211_vif *vif)
1873 {
1874         struct iwl_mvm_mc_iter_data *data = _data;
1875         struct iwl_mvm *mvm = data->mvm;
1876         struct iwl_mcast_filter_cmd *cmd = mvm->mcast_filter_cmd;
1877         int ret, len;
1878
1879         /* if we don't have free ports, mcast frames will be dropped */
1880         if (WARN_ON_ONCE(data->port_id >= MAX_PORT_ID_NUM))
1881                 return;
1882
1883         if (vif->type != NL80211_IFTYPE_STATION ||
1884             !vif->bss_conf.assoc)
1885                 return;
1886
1887         cmd->port_id = data->port_id++;
1888         memcpy(cmd->bssid, vif->bss_conf.bssid, ETH_ALEN);
1889         len = roundup(sizeof(*cmd) + cmd->count * ETH_ALEN, 4);
1890
1891         ret = iwl_mvm_send_cmd_pdu(mvm, MCAST_FILTER_CMD, CMD_ASYNC, len, cmd);
1892         if (ret)
1893                 IWL_ERR(mvm, "mcast filter cmd error. ret=%d\n", ret);
1894 }
1895
1896 static void iwl_mvm_recalc_multicast(struct iwl_mvm *mvm)
1897 {
1898         struct iwl_mvm_mc_iter_data iter_data = {
1899                 .mvm = mvm,
1900         };
1901
1902         lockdep_assert_held(&mvm->mutex);
1903
1904         if (WARN_ON_ONCE(!mvm->mcast_filter_cmd))
1905                 return;
1906
1907         ieee80211_iterate_active_interfaces_atomic(
1908                 mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
1909                 iwl_mvm_mc_iface_iterator, &iter_data);
1910 }
1911
1912 static u64 iwl_mvm_prepare_multicast(struct ieee80211_hw *hw,
1913                                      struct netdev_hw_addr_list *mc_list)
1914 {
1915         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1916         struct iwl_mcast_filter_cmd *cmd;
1917         struct netdev_hw_addr *addr;
1918         int addr_count;
1919         bool pass_all;
1920         int len;
1921
1922         addr_count = netdev_hw_addr_list_count(mc_list);
1923         pass_all = addr_count > MAX_MCAST_FILTERING_ADDRESSES ||
1924                    IWL_MVM_FW_MCAST_FILTER_PASS_ALL;
1925         if (pass_all)
1926                 addr_count = 0;
1927
1928         len = roundup(sizeof(*cmd) + addr_count * ETH_ALEN, 4);
1929         cmd = kzalloc(len, GFP_ATOMIC);
1930         if (!cmd)
1931                 return 0;
1932
1933         if (pass_all) {
1934                 cmd->pass_all = 1;
1935                 return (u64)(unsigned long)cmd;
1936         }
1937
1938         netdev_hw_addr_list_for_each(addr, mc_list) {
1939                 IWL_DEBUG_MAC80211(mvm, "mcast addr (%d): %pM\n",
1940                                    cmd->count, addr->addr);
1941                 memcpy(&cmd->addr_list[cmd->count * ETH_ALEN],
1942                        addr->addr, ETH_ALEN);
1943                 cmd->count++;
1944         }
1945
1946         return (u64)(unsigned long)cmd;
1947 }
1948
1949 static void iwl_mvm_configure_filter(struct ieee80211_hw *hw,
1950                                      unsigned int changed_flags,
1951                                      unsigned int *total_flags,
1952                                      u64 multicast)
1953 {
1954         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1955         struct iwl_mcast_filter_cmd *cmd = (void *)(unsigned long)multicast;
1956
1957         mutex_lock(&mvm->mutex);
1958
1959         /* replace previous configuration */
1960         kfree(mvm->mcast_filter_cmd);
1961         mvm->mcast_filter_cmd = cmd;
1962
1963         if (!cmd)
1964                 goto out;
1965
1966         iwl_mvm_recalc_multicast(mvm);
1967 out:
1968         mutex_unlock(&mvm->mutex);
1969         *total_flags = 0;
1970 }
1971
1972 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
1973 struct iwl_bcast_iter_data {
1974         struct iwl_mvm *mvm;
1975         struct iwl_bcast_filter_cmd *cmd;
1976         u8 current_filter;
1977 };
1978
1979 static void
1980 iwl_mvm_set_bcast_filter(struct ieee80211_vif *vif,
1981                          const struct iwl_fw_bcast_filter *in_filter,
1982                          struct iwl_fw_bcast_filter *out_filter)
1983 {
1984         struct iwl_fw_bcast_filter_attr *attr;
1985         int i;
1986
1987         memcpy(out_filter, in_filter, sizeof(*out_filter));
1988
1989         for (i = 0; i < ARRAY_SIZE(out_filter->attrs); i++) {
1990                 attr = &out_filter->attrs[i];
1991
1992                 if (!attr->mask)
1993                         break;
1994
1995                 switch (attr->reserved1) {
1996                 case cpu_to_le16(BC_FILTER_MAGIC_IP):
1997                         if (vif->bss_conf.arp_addr_cnt != 1) {
1998                                 attr->mask = 0;
1999                                 continue;
2000                         }
2001
2002                         attr->val = vif->bss_conf.arp_addr_list[0];
2003                         break;
2004                 case cpu_to_le16(BC_FILTER_MAGIC_MAC):
2005                         attr->val = *(__be32 *)&vif->addr[2];
2006                         break;
2007                 default:
2008                         break;
2009                 }
2010                 attr->reserved1 = 0;
2011                 out_filter->num_attrs++;
2012         }
2013 }
2014
2015 static void iwl_mvm_bcast_filter_iterator(void *_data, u8 *mac,
2016                                           struct ieee80211_vif *vif)
2017 {
2018         struct iwl_bcast_iter_data *data = _data;
2019         struct iwl_mvm *mvm = data->mvm;
2020         struct iwl_bcast_filter_cmd *cmd = data->cmd;
2021         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2022         struct iwl_fw_bcast_mac *bcast_mac;
2023         int i;
2024
2025         if (WARN_ON(mvmvif->id >= ARRAY_SIZE(cmd->macs)))
2026                 return;
2027
2028         bcast_mac = &cmd->macs[mvmvif->id];
2029
2030         /*
2031          * enable filtering only for associated stations, but not for P2P
2032          * Clients
2033          */
2034         if (vif->type != NL80211_IFTYPE_STATION || vif->p2p ||
2035             !vif->bss_conf.assoc)
2036                 return;
2037
2038         bcast_mac->default_discard = 1;
2039
2040         /* copy all configured filters */
2041         for (i = 0; mvm->bcast_filters[i].attrs[0].mask; i++) {
2042                 /*
2043                  * Make sure we don't exceed our filters limit.
2044                  * if there is still a valid filter to be configured,
2045                  * be on the safe side and just allow bcast for this mac.
2046                  */
2047                 if (WARN_ON_ONCE(data->current_filter >=
2048                                  ARRAY_SIZE(cmd->filters))) {
2049                         bcast_mac->default_discard = 0;
2050                         bcast_mac->attached_filters = 0;
2051                         break;
2052                 }
2053
2054                 iwl_mvm_set_bcast_filter(vif,
2055                                          &mvm->bcast_filters[i],
2056                                          &cmd->filters[data->current_filter]);
2057
2058                 /* skip current filter if it contains no attributes */
2059                 if (!cmd->filters[data->current_filter].num_attrs)
2060                         continue;
2061
2062                 /* attach the filter to current mac */
2063                 bcast_mac->attached_filters |=
2064                                 cpu_to_le16(BIT(data->current_filter));
2065
2066                 data->current_filter++;
2067         }
2068 }
2069
2070 bool iwl_mvm_bcast_filter_build_cmd(struct iwl_mvm *mvm,
2071                                     struct iwl_bcast_filter_cmd *cmd)
2072 {
2073         struct iwl_bcast_iter_data iter_data = {
2074                 .mvm = mvm,
2075                 .cmd = cmd,
2076         };
2077
2078         if (IWL_MVM_FW_BCAST_FILTER_PASS_ALL)
2079                 return false;
2080
2081         memset(cmd, 0, sizeof(*cmd));
2082         cmd->max_bcast_filters = ARRAY_SIZE(cmd->filters);
2083         cmd->max_macs = ARRAY_SIZE(cmd->macs);
2084
2085 #ifdef CONFIG_IWLWIFI_DEBUGFS
2086         /* use debugfs filters/macs if override is configured */
2087         if (mvm->dbgfs_bcast_filtering.override) {
2088                 memcpy(cmd->filters, &mvm->dbgfs_bcast_filtering.cmd.filters,
2089                        sizeof(cmd->filters));
2090                 memcpy(cmd->macs, &mvm->dbgfs_bcast_filtering.cmd.macs,
2091                        sizeof(cmd->macs));
2092                 return true;
2093         }
2094 #endif
2095
2096         /* if no filters are configured, do nothing */
2097         if (!mvm->bcast_filters)
2098                 return false;
2099
2100         /* configure and attach these filters for each associated sta vif */
2101         ieee80211_iterate_active_interfaces(
2102                 mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
2103                 iwl_mvm_bcast_filter_iterator, &iter_data);
2104
2105         return true;
2106 }
2107 static int iwl_mvm_configure_bcast_filter(struct iwl_mvm *mvm,
2108                                           struct ieee80211_vif *vif)
2109 {
2110         struct iwl_bcast_filter_cmd cmd;
2111
2112         if (!(mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_BCAST_FILTERING))
2113                 return 0;
2114
2115         if (!iwl_mvm_bcast_filter_build_cmd(mvm, &cmd))
2116                 return 0;
2117
2118         return iwl_mvm_send_cmd_pdu(mvm, BCAST_FILTER_CMD, 0,
2119                                     sizeof(cmd), &cmd);
2120 }
2121 #else
2122 static inline int iwl_mvm_configure_bcast_filter(struct iwl_mvm *mvm,
2123                                                  struct ieee80211_vif *vif)
2124 {
2125         return 0;
2126 }
2127 #endif
2128
2129 static void iwl_mvm_bss_info_changed_station(struct iwl_mvm *mvm,
2130                                              struct ieee80211_vif *vif,
2131                                              struct ieee80211_bss_conf *bss_conf,
2132                                              u32 changes)
2133 {
2134         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2135         int ret;
2136
2137         /*
2138          * Re-calculate the tsf id, as the master-slave relations depend on the
2139          * beacon interval, which was not known when the station interface was
2140          * added.
2141          */
2142         if (changes & BSS_CHANGED_ASSOC && bss_conf->assoc)
2143                 iwl_mvm_mac_ctxt_recalc_tsf_id(mvm, vif);
2144
2145         /*
2146          * If we're not associated yet, take the (new) BSSID before associating
2147          * so the firmware knows. If we're already associated, then use the old
2148          * BSSID here, and we'll send a cleared one later in the CHANGED_ASSOC
2149          * branch for disassociation below.
2150          */
2151         if (changes & BSS_CHANGED_BSSID && !mvmvif->associated)
2152                 memcpy(mvmvif->bssid, bss_conf->bssid, ETH_ALEN);
2153
2154         ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, mvmvif->bssid);
2155         if (ret)
2156                 IWL_ERR(mvm, "failed to update MAC %pM\n", vif->addr);
2157
2158         /* after sending it once, adopt mac80211 data */
2159         memcpy(mvmvif->bssid, bss_conf->bssid, ETH_ALEN);
2160         mvmvif->associated = bss_conf->assoc;
2161
2162         if (changes & BSS_CHANGED_ASSOC) {
2163                 if (bss_conf->assoc) {
2164                         /* clear statistics to get clean beacon counter */
2165                         iwl_mvm_request_statistics(mvm, true);
2166                         memset(&mvmvif->beacon_stats, 0,
2167                                sizeof(mvmvif->beacon_stats));
2168
2169                         /* add quota for this interface */
2170                         ret = iwl_mvm_update_quotas(mvm, true, NULL);
2171                         if (ret) {
2172                                 IWL_ERR(mvm, "failed to update quotas\n");
2173                                 return;
2174                         }
2175
2176                         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART,
2177                                      &mvm->status)) {
2178                                 /*
2179                                  * If we're restarting then the firmware will
2180                                  * obviously have lost synchronisation with
2181                                  * the AP. It will attempt to synchronise by
2182                                  * itself, but we can make it more reliable by
2183                                  * scheduling a session protection time event.
2184                                  *
2185                                  * The firmware needs to receive a beacon to
2186                                  * catch up with synchronisation, use 110% of
2187                                  * the beacon interval.
2188                                  *
2189                                  * Set a large maximum delay to allow for more
2190                                  * than a single interface.
2191                                  */
2192                                 u32 dur = (11 * vif->bss_conf.beacon_int) / 10;
2193                                 iwl_mvm_protect_session(mvm, vif, dur, dur,
2194                                                         5 * dur, false);
2195                         }
2196
2197                         iwl_mvm_sf_update(mvm, vif, false);
2198                         iwl_mvm_power_vif_assoc(mvm, vif);
2199                         if (vif->p2p) {
2200                                 iwl_mvm_ref(mvm, IWL_MVM_REF_P2P_CLIENT);
2201                                 iwl_mvm_update_smps(mvm, vif,
2202                                                     IWL_MVM_SMPS_REQ_PROT,
2203                                                     IEEE80211_SMPS_DYNAMIC);
2204                         }
2205                 } else if (mvmvif->ap_sta_id != IWL_MVM_STATION_COUNT) {
2206                         /*
2207                          * If update fails - SF might be running in associated
2208                          * mode while disassociated - which is forbidden.
2209                          */
2210                         WARN_ONCE(iwl_mvm_sf_update(mvm, vif, false),
2211                                   "Failed to update SF upon disassociation\n");
2212
2213                         /* remove AP station now that the MAC is unassoc */
2214                         ret = iwl_mvm_rm_sta_id(mvm, vif, mvmvif->ap_sta_id);
2215                         if (ret)
2216                                 IWL_ERR(mvm, "failed to remove AP station\n");
2217
2218                         if (mvm->d0i3_ap_sta_id == mvmvif->ap_sta_id)
2219                                 mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
2220                         mvmvif->ap_sta_id = IWL_MVM_STATION_COUNT;
2221                         /* remove quota for this interface */
2222                         ret = iwl_mvm_update_quotas(mvm, false, NULL);
2223                         if (ret)
2224                                 IWL_ERR(mvm, "failed to update quotas\n");
2225
2226                         if (vif->p2p)
2227                                 iwl_mvm_unref(mvm, IWL_MVM_REF_P2P_CLIENT);
2228
2229                         /* this will take the cleared BSSID from bss_conf */
2230                         ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
2231                         if (ret)
2232                                 IWL_ERR(mvm,
2233                                         "failed to update MAC %pM (clear after unassoc)\n",
2234                                         vif->addr);
2235                 }
2236
2237                 iwl_mvm_recalc_multicast(mvm);
2238                 iwl_mvm_configure_bcast_filter(mvm, vif);
2239
2240                 /* reset rssi values */
2241                 mvmvif->bf_data.ave_beacon_signal = 0;
2242
2243                 iwl_mvm_bt_coex_vif_change(mvm);
2244                 iwl_mvm_update_smps(mvm, vif, IWL_MVM_SMPS_REQ_TT,
2245                                     IEEE80211_SMPS_AUTOMATIC);
2246         } else if (changes & BSS_CHANGED_BEACON_INFO) {
2247                 /*
2248                  * We received a beacon _after_ association so
2249                  * remove the session protection.
2250                  */
2251                 iwl_mvm_remove_time_event(mvm, mvmvif,
2252                                           &mvmvif->time_event_data);
2253         }
2254
2255         if (changes & BSS_CHANGED_BEACON_INFO) {
2256                 iwl_mvm_sf_update(mvm, vif, false);
2257                 WARN_ON(iwl_mvm_enable_beacon_filter(mvm, vif, 0));
2258         }
2259
2260         if (changes & (BSS_CHANGED_PS | BSS_CHANGED_P2P_PS | BSS_CHANGED_QOS)) {
2261                 ret = iwl_mvm_power_update_mac(mvm);
2262                 if (ret)
2263                         IWL_ERR(mvm, "failed to update power mode\n");
2264         }
2265
2266         if (changes & BSS_CHANGED_TXPOWER) {
2267                 IWL_DEBUG_CALIB(mvm, "Changing TX Power to %d\n",
2268                                 bss_conf->txpower);
2269                 iwl_mvm_set_tx_power(mvm, vif, bss_conf->txpower);
2270         }
2271
2272         if (changes & BSS_CHANGED_CQM) {
2273                 IWL_DEBUG_MAC80211(mvm, "cqm info_changed\n");
2274                 /* reset cqm events tracking */
2275                 mvmvif->bf_data.last_cqm_event = 0;
2276                 if (mvmvif->bf_data.bf_enabled) {
2277                         ret = iwl_mvm_enable_beacon_filter(mvm, vif, 0);
2278                         if (ret)
2279                                 IWL_ERR(mvm,
2280                                         "failed to update CQM thresholds\n");
2281                 }
2282         }
2283
2284         if (changes & BSS_CHANGED_ARP_FILTER) {
2285                 IWL_DEBUG_MAC80211(mvm, "arp filter changed\n");
2286                 iwl_mvm_configure_bcast_filter(mvm, vif);
2287         }
2288 }
2289
2290 static int iwl_mvm_start_ap_ibss(struct ieee80211_hw *hw,
2291                                  struct ieee80211_vif *vif)
2292 {
2293         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2294         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2295         int ret;
2296
2297         /*
2298          * iwl_mvm_mac_ctxt_add() might read directly from the device
2299          * (the system time), so make sure it is available.
2300          */
2301         ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_START_AP);
2302         if (ret)
2303                 return ret;
2304
2305         mutex_lock(&mvm->mutex);
2306
2307         /* Send the beacon template */
2308         ret = iwl_mvm_mac_ctxt_beacon_changed(mvm, vif);
2309         if (ret)
2310                 goto out_unlock;
2311
2312         /*
2313          * Re-calculate the tsf id, as the master-slave relations depend on the
2314          * beacon interval, which was not known when the AP interface was added.
2315          */
2316         if (vif->type == NL80211_IFTYPE_AP)
2317                 iwl_mvm_mac_ctxt_recalc_tsf_id(mvm, vif);
2318
2319         /* Add the mac context */
2320         ret = iwl_mvm_mac_ctxt_add(mvm, vif);
2321         if (ret)
2322                 goto out_unlock;
2323
2324         /* Perform the binding */
2325         ret = iwl_mvm_binding_add_vif(mvm, vif);
2326         if (ret)
2327                 goto out_remove;
2328
2329         /* Send the bcast station. At this stage the TBTT and DTIM time events
2330          * are added and applied to the scheduler */
2331         ret = iwl_mvm_send_add_bcast_sta(mvm, vif);
2332         if (ret)
2333                 goto out_unbind;
2334
2335         /* must be set before quota calculations */
2336         mvmvif->ap_ibss_active = true;
2337
2338         /* power updated needs to be done before quotas */
2339         iwl_mvm_power_update_mac(mvm);
2340
2341         ret = iwl_mvm_update_quotas(mvm, false, NULL);
2342         if (ret)
2343                 goto out_quota_failed;
2344
2345         /* Need to update the P2P Device MAC (only GO, IBSS is single vif) */
2346         if (vif->p2p && mvm->p2p_device_vif)
2347                 iwl_mvm_mac_ctxt_changed(mvm, mvm->p2p_device_vif, false, NULL);
2348
2349         iwl_mvm_ref(mvm, IWL_MVM_REF_AP_IBSS);
2350
2351         iwl_mvm_bt_coex_vif_change(mvm);
2352
2353         /* we don't support TDLS during DCM */
2354         if (iwl_mvm_phy_ctx_count(mvm) > 1)
2355                 iwl_mvm_teardown_tdls_peers(mvm);
2356
2357         goto out_unlock;
2358
2359 out_quota_failed:
2360         iwl_mvm_power_update_mac(mvm);
2361         mvmvif->ap_ibss_active = false;
2362         iwl_mvm_send_rm_bcast_sta(mvm, vif);
2363 out_unbind:
2364         iwl_mvm_binding_remove_vif(mvm, vif);
2365 out_remove:
2366         iwl_mvm_mac_ctxt_remove(mvm, vif);
2367 out_unlock:
2368         mutex_unlock(&mvm->mutex);
2369         iwl_mvm_unref(mvm, IWL_MVM_REF_START_AP);
2370         return ret;
2371 }
2372
2373 static void iwl_mvm_stop_ap_ibss(struct ieee80211_hw *hw,
2374                                  struct ieee80211_vif *vif)
2375 {
2376         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2377         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2378
2379         iwl_mvm_prepare_mac_removal(mvm, vif);
2380
2381         mutex_lock(&mvm->mutex);
2382
2383         /* Handle AP stop while in CSA */
2384         if (rcu_access_pointer(mvm->csa_vif) == vif) {
2385                 iwl_mvm_remove_time_event(mvm, mvmvif,
2386                                           &mvmvif->time_event_data);
2387                 RCU_INIT_POINTER(mvm->csa_vif, NULL);
2388         }
2389
2390         if (rcu_access_pointer(mvm->csa_tx_blocked_vif) == vif) {
2391                 RCU_INIT_POINTER(mvm->csa_tx_blocked_vif, NULL);
2392                 mvm->csa_tx_block_bcn_timeout = 0;
2393         }
2394
2395         mvmvif->ap_ibss_active = false;
2396         mvm->ap_last_beacon_gp2 = 0;
2397
2398         iwl_mvm_bt_coex_vif_change(mvm);
2399
2400         iwl_mvm_unref(mvm, IWL_MVM_REF_AP_IBSS);
2401
2402         /* Need to update the P2P Device MAC (only GO, IBSS is single vif) */
2403         if (vif->p2p && mvm->p2p_device_vif)
2404                 iwl_mvm_mac_ctxt_changed(mvm, mvm->p2p_device_vif, false, NULL);
2405
2406         iwl_mvm_update_quotas(mvm, false, NULL);
2407         iwl_mvm_send_rm_bcast_sta(mvm, vif);
2408         iwl_mvm_binding_remove_vif(mvm, vif);
2409
2410         iwl_mvm_power_update_mac(mvm);
2411
2412         iwl_mvm_mac_ctxt_remove(mvm, vif);
2413
2414         mutex_unlock(&mvm->mutex);
2415 }
2416
2417 static void
2418 iwl_mvm_bss_info_changed_ap_ibss(struct iwl_mvm *mvm,
2419                                  struct ieee80211_vif *vif,
2420                                  struct ieee80211_bss_conf *bss_conf,
2421                                  u32 changes)
2422 {
2423         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2424
2425         /* Changes will be applied when the AP/IBSS is started */
2426         if (!mvmvif->ap_ibss_active)
2427                 return;
2428
2429         if (changes & (BSS_CHANGED_ERP_CTS_PROT | BSS_CHANGED_HT |
2430                        BSS_CHANGED_BANDWIDTH | BSS_CHANGED_QOS) &&
2431             iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL))
2432                 IWL_ERR(mvm, "failed to update MAC %pM\n", vif->addr);
2433
2434         /* Need to send a new beacon template to the FW */
2435         if (changes & BSS_CHANGED_BEACON &&
2436             iwl_mvm_mac_ctxt_beacon_changed(mvm, vif))
2437                 IWL_WARN(mvm, "Failed updating beacon data\n");
2438
2439         if (changes & BSS_CHANGED_TXPOWER) {
2440                 IWL_DEBUG_CALIB(mvm, "Changing TX Power to %d\n",
2441                                 bss_conf->txpower);
2442                 iwl_mvm_set_tx_power(mvm, vif, bss_conf->txpower);
2443         }
2444
2445 }
2446
2447 static void iwl_mvm_bss_info_changed(struct ieee80211_hw *hw,
2448                                      struct ieee80211_vif *vif,
2449                                      struct ieee80211_bss_conf *bss_conf,
2450                                      u32 changes)
2451 {
2452         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2453
2454         /*
2455          * iwl_mvm_bss_info_changed_station() might call
2456          * iwl_mvm_protect_session(), which reads directly from
2457          * the device (the system time), so make sure it is available.
2458          */
2459         if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_BSS_CHANGED))
2460                 return;
2461
2462         mutex_lock(&mvm->mutex);
2463
2464         if (changes & BSS_CHANGED_IDLE && !bss_conf->idle)
2465                 iwl_mvm_scan_stop(mvm, IWL_MVM_SCAN_SCHED, true);
2466
2467         switch (vif->type) {
2468         case NL80211_IFTYPE_STATION:
2469                 iwl_mvm_bss_info_changed_station(mvm, vif, bss_conf, changes);
2470                 break;
2471         case NL80211_IFTYPE_AP:
2472         case NL80211_IFTYPE_ADHOC:
2473                 iwl_mvm_bss_info_changed_ap_ibss(mvm, vif, bss_conf, changes);
2474                 break;
2475         default:
2476                 /* shouldn't happen */
2477                 WARN_ON_ONCE(1);
2478         }
2479
2480         mutex_unlock(&mvm->mutex);
2481         iwl_mvm_unref(mvm, IWL_MVM_REF_BSS_CHANGED);
2482 }
2483
2484 static int iwl_mvm_mac_hw_scan(struct ieee80211_hw *hw,
2485                                struct ieee80211_vif *vif,
2486                                struct ieee80211_scan_request *hw_req)
2487 {
2488         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2489         int ret;
2490
2491         if (hw_req->req.n_channels == 0 ||
2492             hw_req->req.n_channels > mvm->fw->ucode_capa.n_scan_channels)
2493                 return -EINVAL;
2494
2495         mutex_lock(&mvm->mutex);
2496         ret = iwl_mvm_reg_scan_start(mvm, vif, &hw_req->req, &hw_req->ies);
2497         mutex_unlock(&mvm->mutex);
2498
2499         return ret;
2500 }
2501
2502 static void iwl_mvm_mac_cancel_hw_scan(struct ieee80211_hw *hw,
2503                                        struct ieee80211_vif *vif)
2504 {
2505         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2506
2507         mutex_lock(&mvm->mutex);
2508
2509         /* Due to a race condition, it's possible that mac80211 asks
2510          * us to stop a hw_scan when it's already stopped.  This can
2511          * happen, for instance, if we stopped the scan ourselves,
2512          * called ieee80211_scan_completed() and the userspace called
2513          * cancel scan scan before ieee80211_scan_work() could run.
2514          * To handle that, simply return if the scan is not running.
2515         */
2516         if (mvm->scan_status & IWL_MVM_SCAN_REGULAR)
2517                 iwl_mvm_scan_stop(mvm, IWL_MVM_SCAN_REGULAR, true);
2518
2519         mutex_unlock(&mvm->mutex);
2520 }
2521
2522 static void
2523 iwl_mvm_mac_allow_buffered_frames(struct ieee80211_hw *hw,
2524                                   struct ieee80211_sta *sta, u16 tids,
2525                                   int num_frames,
2526                                   enum ieee80211_frame_release_type reason,
2527                                   bool more_data)
2528 {
2529         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2530
2531         /* Called when we need to transmit (a) frame(s) from mac80211 */
2532
2533         iwl_mvm_sta_modify_sleep_tx_count(mvm, sta, reason, num_frames,
2534                                           tids, more_data, false);
2535 }
2536
2537 static void
2538 iwl_mvm_mac_release_buffered_frames(struct ieee80211_hw *hw,
2539                                     struct ieee80211_sta *sta, u16 tids,
2540                                     int num_frames,
2541                                     enum ieee80211_frame_release_type reason,
2542                                     bool more_data)
2543 {
2544         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2545
2546         /* Called when we need to transmit (a) frame(s) from agg queue */
2547
2548         iwl_mvm_sta_modify_sleep_tx_count(mvm, sta, reason, num_frames,
2549                                           tids, more_data, true);
2550 }
2551
2552 static void iwl_mvm_mac_sta_notify(struct ieee80211_hw *hw,
2553                                    struct ieee80211_vif *vif,
2554                                    enum sta_notify_cmd cmd,
2555                                    struct ieee80211_sta *sta)
2556 {
2557         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2558         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
2559         unsigned long txqs = 0, tids = 0;
2560         int tid;
2561
2562         spin_lock_bh(&mvmsta->lock);
2563         for (tid = 0; tid < IWL_MAX_TID_COUNT; tid++) {
2564                 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
2565
2566                 if (tid_data->state != IWL_AGG_ON &&
2567                     tid_data->state != IWL_EMPTYING_HW_QUEUE_DELBA)
2568                         continue;
2569
2570                 __set_bit(tid_data->txq_id, &txqs);
2571
2572                 if (iwl_mvm_tid_queued(tid_data) == 0)
2573                         continue;
2574
2575                 __set_bit(tid, &tids);
2576         }
2577
2578         switch (cmd) {
2579         case STA_NOTIFY_SLEEP:
2580                 if (atomic_read(&mvm->pending_frames[mvmsta->sta_id]) > 0)
2581                         ieee80211_sta_block_awake(hw, sta, true);
2582
2583                 for_each_set_bit(tid, &tids, IWL_MAX_TID_COUNT)
2584                         ieee80211_sta_set_buffered(sta, tid, true);
2585
2586                 if (txqs)
2587                         iwl_trans_freeze_txq_timer(mvm->trans, txqs, true);
2588                 /*
2589                  * The fw updates the STA to be asleep. Tx packets on the Tx
2590                  * queues to this station will not be transmitted. The fw will
2591                  * send a Tx response with TX_STATUS_FAIL_DEST_PS.
2592                  */
2593                 break;
2594         case STA_NOTIFY_AWAKE:
2595                 if (WARN_ON(mvmsta->sta_id == IWL_MVM_STATION_COUNT))
2596                         break;
2597
2598                 if (txqs)
2599                         iwl_trans_freeze_txq_timer(mvm->trans, txqs, false);
2600                 iwl_mvm_sta_modify_ps_wake(mvm, sta);
2601                 break;
2602         default:
2603                 break;
2604         }
2605         spin_unlock_bh(&mvmsta->lock);
2606 }
2607
2608 static void iwl_mvm_sta_pre_rcu_remove(struct ieee80211_hw *hw,
2609                                        struct ieee80211_vif *vif,
2610                                        struct ieee80211_sta *sta)
2611 {
2612         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2613         struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
2614
2615         /*
2616          * This is called before mac80211 does RCU synchronisation,
2617          * so here we already invalidate our internal RCU-protected
2618          * station pointer. The rest of the code will thus no longer
2619          * be able to find the station this way, and we don't rely
2620          * on further RCU synchronisation after the sta_state()
2621          * callback deleted the station.
2622          */
2623         mutex_lock(&mvm->mutex);
2624         if (sta == rcu_access_pointer(mvm->fw_id_to_mac_id[mvm_sta->sta_id]))
2625                 rcu_assign_pointer(mvm->fw_id_to_mac_id[mvm_sta->sta_id],
2626                                    ERR_PTR(-ENOENT));
2627         mutex_unlock(&mvm->mutex);
2628 }
2629
2630 static void iwl_mvm_check_uapsd(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
2631                                 const u8 *bssid)
2632 {
2633         if (!(mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_UAPSD_SUPPORT))
2634                 return;
2635
2636         if (iwlwifi_mod_params.uapsd_disable) {
2637                 vif->driver_flags &= ~IEEE80211_VIF_SUPPORTS_UAPSD;
2638                 return;
2639         }
2640
2641         vif->driver_flags |= IEEE80211_VIF_SUPPORTS_UAPSD;
2642 }
2643
2644 static int iwl_mvm_mac_sta_state(struct ieee80211_hw *hw,
2645                                  struct ieee80211_vif *vif,
2646                                  struct ieee80211_sta *sta,
2647                                  enum ieee80211_sta_state old_state,
2648                                  enum ieee80211_sta_state new_state)
2649 {
2650         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2651         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2652         int ret;
2653
2654         IWL_DEBUG_MAC80211(mvm, "station %pM state change %d->%d\n",
2655                            sta->addr, old_state, new_state);
2656
2657         /* this would be a mac80211 bug ... but don't crash */
2658         if (WARN_ON_ONCE(!mvmvif->phy_ctxt))
2659                 return -EINVAL;
2660
2661         /* if a STA is being removed, reuse its ID */
2662         flush_work(&mvm->sta_drained_wk);
2663
2664         mutex_lock(&mvm->mutex);
2665         if (old_state == IEEE80211_STA_NOTEXIST &&
2666             new_state == IEEE80211_STA_NONE) {
2667                 /*
2668                  * Firmware bug - it'll crash if the beacon interval is less
2669                  * than 16. We can't avoid connecting at all, so refuse the
2670                  * station state change, this will cause mac80211 to abandon
2671                  * attempts to connect to this AP, and eventually wpa_s will
2672                  * blacklist the AP...
2673                  */
2674                 if (vif->type == NL80211_IFTYPE_STATION &&
2675                     vif->bss_conf.beacon_int < 16) {
2676                         IWL_ERR(mvm,
2677                                 "AP %pM beacon interval is %d, refusing due to firmware bug!\n",
2678                                 sta->addr, vif->bss_conf.beacon_int);
2679                         ret = -EINVAL;
2680                         goto out_unlock;
2681                 }
2682
2683                 if (sta->tdls &&
2684                     (vif->p2p ||
2685                      iwl_mvm_tdls_sta_count(mvm, NULL) ==
2686                                                 IWL_MVM_TDLS_STA_COUNT ||
2687                      iwl_mvm_phy_ctx_count(mvm) > 1)) {
2688                         IWL_DEBUG_MAC80211(mvm, "refusing TDLS sta\n");
2689                         ret = -EBUSY;
2690                         goto out_unlock;
2691                 }
2692
2693                 ret = iwl_mvm_add_sta(mvm, vif, sta);
2694                 if (sta->tdls && ret == 0)
2695                         iwl_mvm_recalc_tdls_state(mvm, vif, true);
2696         } else if (old_state == IEEE80211_STA_NONE &&
2697                    new_state == IEEE80211_STA_AUTH) {
2698                 /*
2699                  * EBS may be disabled due to previous failures reported by FW.
2700                  * Reset EBS status here assuming environment has been changed.
2701                  */
2702                 mvm->last_ebs_successful = true;
2703                 iwl_mvm_check_uapsd(mvm, vif, sta->addr);
2704                 ret = 0;
2705         } else if (old_state == IEEE80211_STA_AUTH &&
2706                    new_state == IEEE80211_STA_ASSOC) {
2707                 ret = iwl_mvm_update_sta(mvm, vif, sta);
2708                 if (ret == 0)
2709                         iwl_mvm_rs_rate_init(mvm, sta,
2710                                              mvmvif->phy_ctxt->channel->band,
2711                                              true);
2712         } else if (old_state == IEEE80211_STA_ASSOC &&
2713                    new_state == IEEE80211_STA_AUTHORIZED) {
2714
2715                 /* we don't support TDLS during DCM */
2716                 if (iwl_mvm_phy_ctx_count(mvm) > 1)
2717                         iwl_mvm_teardown_tdls_peers(mvm);
2718
2719                 /* enable beacon filtering */
2720                 WARN_ON(iwl_mvm_enable_beacon_filter(mvm, vif, 0));
2721                 ret = 0;
2722         } else if (old_state == IEEE80211_STA_AUTHORIZED &&
2723                    new_state == IEEE80211_STA_ASSOC) {
2724                 /* disable beacon filtering */
2725                 WARN_ON(iwl_mvm_disable_beacon_filter(mvm, vif, 0));
2726                 ret = 0;
2727         } else if (old_state == IEEE80211_STA_ASSOC &&
2728                    new_state == IEEE80211_STA_AUTH) {
2729                 ret = 0;
2730         } else if (old_state == IEEE80211_STA_AUTH &&
2731                    new_state == IEEE80211_STA_NONE) {
2732                 ret = 0;
2733         } else if (old_state == IEEE80211_STA_NONE &&
2734                    new_state == IEEE80211_STA_NOTEXIST) {
2735                 ret = iwl_mvm_rm_sta(mvm, vif, sta);
2736                 if (sta->tdls)
2737                         iwl_mvm_recalc_tdls_state(mvm, vif, false);
2738         } else {
2739                 ret = -EIO;
2740         }
2741  out_unlock:
2742         mutex_unlock(&mvm->mutex);
2743
2744         if (sta->tdls && ret == 0) {
2745                 if (old_state == IEEE80211_STA_NOTEXIST &&
2746                     new_state == IEEE80211_STA_NONE)
2747                         ieee80211_reserve_tid(sta, IWL_MVM_TDLS_FW_TID);
2748                 else if (old_state == IEEE80211_STA_NONE &&
2749                          new_state == IEEE80211_STA_NOTEXIST)
2750                         ieee80211_unreserve_tid(sta, IWL_MVM_TDLS_FW_TID);
2751         }
2752
2753         return ret;
2754 }
2755
2756 static int iwl_mvm_mac_set_rts_threshold(struct ieee80211_hw *hw, u32 value)
2757 {
2758         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2759
2760         mvm->rts_threshold = value;
2761
2762         return 0;
2763 }
2764
2765 static void iwl_mvm_sta_rc_update(struct ieee80211_hw *hw,
2766                                   struct ieee80211_vif *vif,
2767                                   struct ieee80211_sta *sta, u32 changed)
2768 {
2769         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2770
2771         if (vif->type == NL80211_IFTYPE_STATION &&
2772             changed & IEEE80211_RC_NSS_CHANGED)
2773                 iwl_mvm_sf_update(mvm, vif, false);
2774 }
2775
2776 static int iwl_mvm_mac_conf_tx(struct ieee80211_hw *hw,
2777                                struct ieee80211_vif *vif, u16 ac,
2778                                const struct ieee80211_tx_queue_params *params)
2779 {
2780         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2781         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2782
2783         mvmvif->queue_params[ac] = *params;
2784
2785         /*
2786          * No need to update right away, we'll get BSS_CHANGED_QOS
2787          * The exception is P2P_DEVICE interface which needs immediate update.
2788          */
2789         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
2790                 int ret;
2791
2792                 mutex_lock(&mvm->mutex);
2793                 ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
2794                 mutex_unlock(&mvm->mutex);
2795                 return ret;
2796         }
2797         return 0;
2798 }
2799
2800 static void iwl_mvm_mac_mgd_prepare_tx(struct ieee80211_hw *hw,
2801                                       struct ieee80211_vif *vif)
2802 {
2803         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2804         u32 duration = min(IWL_MVM_TE_SESSION_PROTECTION_MAX_TIME_MS,
2805                            200 + vif->bss_conf.beacon_int);
2806         u32 min_duration = min(IWL_MVM_TE_SESSION_PROTECTION_MIN_TIME_MS,
2807                                100 + vif->bss_conf.beacon_int);
2808
2809         if (WARN_ON_ONCE(vif->bss_conf.assoc))
2810                 return;
2811
2812         /*
2813          * iwl_mvm_protect_session() reads directly from the device
2814          * (the system time), so make sure it is available.
2815          */
2816         if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_PREPARE_TX))
2817                 return;
2818
2819         mutex_lock(&mvm->mutex);
2820         /* Try really hard to protect the session and hear a beacon */
2821         iwl_mvm_protect_session(mvm, vif, duration, min_duration, 500, false);
2822         mutex_unlock(&mvm->mutex);
2823
2824         iwl_mvm_unref(mvm, IWL_MVM_REF_PREPARE_TX);
2825 }
2826
2827 static int iwl_mvm_mac_sched_scan_start(struct ieee80211_hw *hw,
2828                                         struct ieee80211_vif *vif,
2829                                         struct cfg80211_sched_scan_request *req,
2830                                         struct ieee80211_scan_ies *ies)
2831 {
2832         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2833
2834         int ret;
2835
2836         mutex_lock(&mvm->mutex);
2837
2838         if (!vif->bss_conf.idle) {
2839                 ret = -EBUSY;
2840                 goto out;
2841         }
2842
2843         ret = iwl_mvm_sched_scan_start(mvm, vif, req, ies, IWL_MVM_SCAN_SCHED);
2844
2845 out:
2846         mutex_unlock(&mvm->mutex);
2847         return ret;
2848 }
2849
2850 static int iwl_mvm_mac_sched_scan_stop(struct ieee80211_hw *hw,
2851                                        struct ieee80211_vif *vif)
2852 {
2853         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2854         int ret;
2855
2856         mutex_lock(&mvm->mutex);
2857
2858         /* Due to a race condition, it's possible that mac80211 asks
2859          * us to stop a sched_scan when it's already stopped.  This
2860          * can happen, for instance, if we stopped the scan ourselves,
2861          * called ieee80211_sched_scan_stopped() and the userspace called
2862          * stop sched scan scan before ieee80211_sched_scan_stopped_work()
2863          * could run.  To handle this, simply return if the scan is
2864          * not running.
2865         */
2866         if (!(mvm->scan_status & IWL_MVM_SCAN_SCHED)) {
2867                 mutex_unlock(&mvm->mutex);
2868                 return 0;
2869         }
2870
2871         ret = iwl_mvm_scan_stop(mvm, IWL_MVM_SCAN_SCHED, false);
2872         mutex_unlock(&mvm->mutex);
2873         iwl_mvm_wait_for_async_handlers(mvm);
2874
2875         return ret;
2876 }
2877
2878 static int iwl_mvm_mac_set_key(struct ieee80211_hw *hw,
2879                                enum set_key_cmd cmd,
2880                                struct ieee80211_vif *vif,
2881                                struct ieee80211_sta *sta,
2882                                struct ieee80211_key_conf *key)
2883 {
2884         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2885         int ret;
2886
2887         if (iwlwifi_mod_params.sw_crypto) {
2888                 IWL_DEBUG_MAC80211(mvm, "leave - hwcrypto disabled\n");
2889                 return -EOPNOTSUPP;
2890         }
2891
2892         switch (key->cipher) {
2893         case WLAN_CIPHER_SUITE_TKIP:
2894                 key->flags |= IEEE80211_KEY_FLAG_GENERATE_MMIC;
2895                 key->flags |= IEEE80211_KEY_FLAG_GENERATE_IV;
2896                 break;
2897         case WLAN_CIPHER_SUITE_CCMP:
2898                 key->flags |= IEEE80211_KEY_FLAG_PUT_IV_SPACE;
2899                 break;
2900         case WLAN_CIPHER_SUITE_AES_CMAC:
2901                 WARN_ON_ONCE(!ieee80211_hw_check(hw, MFP_CAPABLE));
2902                 break;
2903         case WLAN_CIPHER_SUITE_WEP40:
2904         case WLAN_CIPHER_SUITE_WEP104:
2905                 /* For non-client mode, only use WEP keys for TX as we probably
2906                  * don't have a station yet anyway and would then have to keep
2907                  * track of the keys, linking them to each of the clients/peers
2908                  * as they appear. For now, don't do that, for performance WEP
2909                  * offload doesn't really matter much, but we need it for some
2910                  * other offload features in client mode.
2911                  */
2912                 if (vif->type != NL80211_IFTYPE_STATION)
2913                         return 0;
2914                 break;
2915         default:
2916                 /* currently FW supports only one optional cipher scheme */
2917                 if (hw->n_cipher_schemes &&
2918                     hw->cipher_schemes->cipher == key->cipher)
2919                         key->flags |= IEEE80211_KEY_FLAG_PUT_IV_SPACE;
2920                 else
2921                         return -EOPNOTSUPP;
2922         }
2923
2924         mutex_lock(&mvm->mutex);
2925
2926         switch (cmd) {
2927         case SET_KEY:
2928                 if ((vif->type == NL80211_IFTYPE_ADHOC ||
2929                      vif->type == NL80211_IFTYPE_AP) && !sta) {
2930                         /*
2931                          * GTK on AP interface is a TX-only key, return 0;
2932                          * on IBSS they're per-station and because we're lazy
2933                          * we don't support them for RX, so do the same.
2934                          */
2935                         ret = 0;
2936                         key->hw_key_idx = STA_KEY_IDX_INVALID;
2937                         break;
2938                 }
2939
2940                 /* During FW restart, in order to restore the state as it was,
2941                  * don't try to reprogram keys we previously failed for.
2942                  */
2943                 if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status) &&
2944                     key->hw_key_idx == STA_KEY_IDX_INVALID) {
2945                         IWL_DEBUG_MAC80211(mvm,
2946                                            "skip invalid idx key programming during restart\n");
2947                         ret = 0;
2948                         break;
2949                 }
2950
2951                 IWL_DEBUG_MAC80211(mvm, "set hwcrypto key\n");
2952                 ret = iwl_mvm_set_sta_key(mvm, vif, sta, key,
2953                                           test_bit(IWL_MVM_STATUS_IN_HW_RESTART,
2954                                                    &mvm->status));
2955                 if (ret) {
2956                         IWL_WARN(mvm, "set key failed\n");
2957                         /*
2958                          * can't add key for RX, but we don't need it
2959                          * in the device for TX so still return 0
2960                          */
2961                         key->hw_key_idx = STA_KEY_IDX_INVALID;
2962                         ret = 0;
2963                 }
2964
2965                 break;
2966         case DISABLE_KEY:
2967                 if (key->hw_key_idx == STA_KEY_IDX_INVALID) {
2968                         ret = 0;
2969                         break;
2970                 }
2971
2972                 IWL_DEBUG_MAC80211(mvm, "disable hwcrypto key\n");
2973                 ret = iwl_mvm_remove_sta_key(mvm, vif, sta, key);
2974                 break;
2975         default:
2976                 ret = -EINVAL;
2977         }
2978
2979         mutex_unlock(&mvm->mutex);
2980         return ret;
2981 }
2982
2983 static void iwl_mvm_mac_update_tkip_key(struct ieee80211_hw *hw,
2984                                         struct ieee80211_vif *vif,
2985                                         struct ieee80211_key_conf *keyconf,
2986                                         struct ieee80211_sta *sta,
2987                                         u32 iv32, u16 *phase1key)
2988 {
2989         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2990
2991         if (keyconf->hw_key_idx == STA_KEY_IDX_INVALID)
2992                 return;
2993
2994         iwl_mvm_update_tkip_key(mvm, vif, keyconf, sta, iv32, phase1key);
2995 }
2996
2997
2998 static bool iwl_mvm_rx_aux_roc(struct iwl_notif_wait_data *notif_wait,
2999                                struct iwl_rx_packet *pkt, void *data)
3000 {
3001         struct iwl_mvm *mvm =
3002                 container_of(notif_wait, struct iwl_mvm, notif_wait);
3003         struct iwl_hs20_roc_res *resp;
3004         int resp_len = iwl_rx_packet_payload_len(pkt);
3005         struct iwl_mvm_time_event_data *te_data = data;
3006
3007         if (WARN_ON(pkt->hdr.cmd != HOT_SPOT_CMD))
3008                 return true;
3009
3010         if (WARN_ON_ONCE(resp_len != sizeof(*resp))) {
3011                 IWL_ERR(mvm, "Invalid HOT_SPOT_CMD response\n");
3012                 return true;
3013         }
3014
3015         resp = (void *)pkt->data;
3016
3017         IWL_DEBUG_TE(mvm,
3018                      "Aux ROC: Recieved response from ucode: status=%d uid=%d\n",
3019                      resp->status, resp->event_unique_id);
3020
3021         te_data->uid = le32_to_cpu(resp->event_unique_id);
3022         IWL_DEBUG_TE(mvm, "TIME_EVENT_CMD response - UID = 0x%x\n",
3023                      te_data->uid);
3024
3025         spin_lock_bh(&mvm->time_event_lock);
3026         list_add_tail(&te_data->list, &mvm->aux_roc_te_list);
3027         spin_unlock_bh(&mvm->time_event_lock);
3028
3029         return true;
3030 }
3031
3032 #define AUX_ROC_MAX_DELAY_ON_CHANNEL 200
3033 static int iwl_mvm_send_aux_roc_cmd(struct iwl_mvm *mvm,
3034                                     struct ieee80211_channel *channel,
3035                                     struct ieee80211_vif *vif,
3036                                     int duration)
3037 {
3038         int res, time_reg = DEVICE_SYSTEM_TIME_REG;
3039         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3040         struct iwl_mvm_time_event_data *te_data = &mvmvif->hs_time_event_data;
3041         static const u16 time_event_response[] = { HOT_SPOT_CMD };
3042         struct iwl_notification_wait wait_time_event;
3043         struct iwl_hs20_roc_req aux_roc_req = {
3044                 .action = cpu_to_le32(FW_CTXT_ACTION_ADD),
3045                 .id_and_color =
3046                         cpu_to_le32(FW_CMD_ID_AND_COLOR(MAC_INDEX_AUX, 0)),
3047                 .sta_id_and_color = cpu_to_le32(mvm->aux_sta.sta_id),
3048                 /* Set the channel info data */
3049                 .channel_info.band = (channel->band == IEEE80211_BAND_2GHZ) ?
3050                         PHY_BAND_24 : PHY_BAND_5,
3051                 .channel_info.channel = channel->hw_value,
3052                 .channel_info.width = PHY_VHT_CHANNEL_MODE20,
3053                 /* Set the time and duration */
3054                 .apply_time = cpu_to_le32(iwl_read_prph(mvm->trans, time_reg)),
3055                 .apply_time_max_delay =
3056                         cpu_to_le32(MSEC_TO_TU(AUX_ROC_MAX_DELAY_ON_CHANNEL)),
3057                 .duration = cpu_to_le32(MSEC_TO_TU(duration)),
3058          };
3059
3060         /* Set the node address */
3061         memcpy(aux_roc_req.node_addr, vif->addr, ETH_ALEN);
3062
3063         lockdep_assert_held(&mvm->mutex);
3064
3065         spin_lock_bh(&mvm->time_event_lock);
3066
3067         if (WARN_ON(te_data->id == HOT_SPOT_CMD)) {
3068                 spin_unlock_bh(&mvm->time_event_lock);
3069                 return -EIO;
3070         }
3071
3072         te_data->vif = vif;
3073         te_data->duration = duration;
3074         te_data->id = HOT_SPOT_CMD;
3075
3076         spin_unlock_bh(&mvm->time_event_lock);
3077
3078         /*
3079          * Use a notification wait, which really just processes the
3080          * command response and doesn't wait for anything, in order
3081          * to be able to process the response and get the UID inside
3082          * the RX path. Using CMD_WANT_SKB doesn't work because it
3083          * stores the buffer and then wakes up this thread, by which
3084          * time another notification (that the time event started)
3085          * might already be processed unsuccessfully.
3086          */
3087         iwl_init_notification_wait(&mvm->notif_wait, &wait_time_event,
3088                                    time_event_response,
3089                                    ARRAY_SIZE(time_event_response),
3090                                    iwl_mvm_rx_aux_roc, te_data);
3091
3092         res = iwl_mvm_send_cmd_pdu(mvm, HOT_SPOT_CMD, 0, sizeof(aux_roc_req),
3093                                    &aux_roc_req);
3094
3095         if (res) {
3096                 IWL_ERR(mvm, "Couldn't send HOT_SPOT_CMD: %d\n", res);
3097                 iwl_remove_notification(&mvm->notif_wait, &wait_time_event);
3098                 goto out_clear_te;
3099         }
3100
3101         /* No need to wait for anything, so just pass 1 (0 isn't valid) */
3102         res = iwl_wait_notification(&mvm->notif_wait, &wait_time_event, 1);
3103         /* should never fail */
3104         WARN_ON_ONCE(res);
3105
3106         if (res) {
3107  out_clear_te:
3108                 spin_lock_bh(&mvm->time_event_lock);
3109                 iwl_mvm_te_clear_data(mvm, te_data);
3110                 spin_unlock_bh(&mvm->time_event_lock);
3111         }
3112
3113         return res;
3114 }
3115
3116 static int iwl_mvm_roc(struct ieee80211_hw *hw,
3117                        struct ieee80211_vif *vif,
3118                        struct ieee80211_channel *channel,
3119                        int duration,
3120                        enum ieee80211_roc_type type)
3121 {
3122         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3123         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3124         struct cfg80211_chan_def chandef;
3125         struct iwl_mvm_phy_ctxt *phy_ctxt;
3126         int ret, i;
3127
3128         IWL_DEBUG_MAC80211(mvm, "enter (%d, %d, %d)\n", channel->hw_value,
3129                            duration, type);
3130
3131         flush_work(&mvm->roc_done_wk);
3132
3133         mutex_lock(&mvm->mutex);
3134
3135         switch (vif->type) {
3136         case NL80211_IFTYPE_STATION:
3137                 if (fw_has_capa(&mvm->fw->ucode_capa,
3138                                 IWL_UCODE_TLV_CAPA_HOTSPOT_SUPPORT)) {
3139                         /* Use aux roc framework (HS20) */
3140                         ret = iwl_mvm_send_aux_roc_cmd(mvm, channel,
3141                                                        vif, duration);
3142                         goto out_unlock;
3143                 }
3144                 IWL_ERR(mvm, "hotspot not supported\n");
3145                 ret = -EINVAL;
3146                 goto out_unlock;
3147         case NL80211_IFTYPE_P2P_DEVICE:
3148                 /* handle below */
3149                 break;
3150         default:
3151                 IWL_ERR(mvm, "vif isn't P2P_DEVICE: %d\n", vif->type);
3152                 ret = -EINVAL;
3153                 goto out_unlock;
3154         }
3155
3156         for (i = 0; i < NUM_PHY_CTX; i++) {
3157                 phy_ctxt = &mvm->phy_ctxts[i];
3158                 if (phy_ctxt->ref == 0 || mvmvif->phy_ctxt == phy_ctxt)
3159                         continue;
3160
3161                 if (phy_ctxt->ref && channel == phy_ctxt->channel) {
3162                         /*
3163                          * Unbind the P2P_DEVICE from the current PHY context,
3164                          * and if the PHY context is not used remove it.
3165                          */
3166                         ret = iwl_mvm_binding_remove_vif(mvm, vif);
3167                         if (WARN(ret, "Failed unbinding P2P_DEVICE\n"))
3168                                 goto out_unlock;
3169
3170                         iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
3171
3172                         /* Bind the P2P_DEVICE to the current PHY Context */
3173                         mvmvif->phy_ctxt = phy_ctxt;
3174
3175                         ret = iwl_mvm_binding_add_vif(mvm, vif);
3176                         if (WARN(ret, "Failed binding P2P_DEVICE\n"))
3177                                 goto out_unlock;
3178
3179                         iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
3180                         goto schedule_time_event;
3181                 }
3182         }
3183
3184         /* Need to update the PHY context only if the ROC channel changed */
3185         if (channel == mvmvif->phy_ctxt->channel)
3186                 goto schedule_time_event;
3187
3188         cfg80211_chandef_create(&chandef, channel, NL80211_CHAN_NO_HT);
3189
3190         /*
3191          * Change the PHY context configuration as it is currently referenced
3192          * only by the P2P Device MAC
3193          */
3194         if (mvmvif->phy_ctxt->ref == 1) {
3195                 ret = iwl_mvm_phy_ctxt_changed(mvm, mvmvif->phy_ctxt,
3196                                                &chandef, 1, 1);
3197                 if (ret)
3198                         goto out_unlock;
3199         } else {
3200                 /*
3201                  * The PHY context is shared with other MACs. Need to remove the
3202                  * P2P Device from the binding, allocate an new PHY context and
3203                  * create a new binding
3204                  */
3205                 phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
3206                 if (!phy_ctxt) {
3207                         ret = -ENOSPC;
3208                         goto out_unlock;
3209                 }
3210
3211                 ret = iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &chandef,
3212                                                1, 1);
3213                 if (ret) {
3214                         IWL_ERR(mvm, "Failed to change PHY context\n");
3215                         goto out_unlock;
3216                 }
3217
3218                 /* Unbind the P2P_DEVICE from the current PHY context */
3219                 ret = iwl_mvm_binding_remove_vif(mvm, vif);
3220                 if (WARN(ret, "Failed unbinding P2P_DEVICE\n"))
3221                         goto out_unlock;
3222
3223                 iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
3224
3225                 /* Bind the P2P_DEVICE to the new allocated PHY context */
3226                 mvmvif->phy_ctxt = phy_ctxt;
3227
3228                 ret = iwl_mvm_binding_add_vif(mvm, vif);
3229                 if (WARN(ret, "Failed binding P2P_DEVICE\n"))
3230                         goto out_unlock;
3231
3232                 iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
3233         }
3234
3235 schedule_time_event:
3236         /* Schedule the time events */
3237         ret = iwl_mvm_start_p2p_roc(mvm, vif, duration, type);
3238
3239 out_unlock:
3240         mutex_unlock(&mvm->mutex);
3241         IWL_DEBUG_MAC80211(mvm, "leave\n");
3242         return ret;
3243 }
3244
3245 static int iwl_mvm_cancel_roc(struct ieee80211_hw *hw)
3246 {
3247         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3248
3249         IWL_DEBUG_MAC80211(mvm, "enter\n");
3250
3251         mutex_lock(&mvm->mutex);
3252         iwl_mvm_stop_roc(mvm);
3253         mutex_unlock(&mvm->mutex);
3254
3255         IWL_DEBUG_MAC80211(mvm, "leave\n");
3256         return 0;
3257 }
3258
3259 static int __iwl_mvm_add_chanctx(struct iwl_mvm *mvm,
3260                                  struct ieee80211_chanctx_conf *ctx)
3261 {
3262         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
3263         struct iwl_mvm_phy_ctxt *phy_ctxt;
3264         int ret;
3265
3266         lockdep_assert_held(&mvm->mutex);
3267
3268         IWL_DEBUG_MAC80211(mvm, "Add channel context\n");
3269
3270         phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
3271         if (!phy_ctxt) {
3272                 ret = -ENOSPC;
3273                 goto out;
3274         }
3275
3276         ret = iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &ctx->min_def,
3277                                        ctx->rx_chains_static,
3278                                        ctx->rx_chains_dynamic);
3279         if (ret) {
3280                 IWL_ERR(mvm, "Failed to add PHY context\n");
3281                 goto out;
3282         }
3283
3284         iwl_mvm_phy_ctxt_ref(mvm, phy_ctxt);
3285         *phy_ctxt_id = phy_ctxt->id;
3286 out:
3287         return ret;
3288 }
3289
3290 static int iwl_mvm_add_chanctx(struct ieee80211_hw *hw,
3291                                struct ieee80211_chanctx_conf *ctx)
3292 {
3293         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3294         int ret;
3295
3296         mutex_lock(&mvm->mutex);
3297         ret = __iwl_mvm_add_chanctx(mvm, ctx);
3298         mutex_unlock(&mvm->mutex);
3299
3300         return ret;
3301 }
3302
3303 static void __iwl_mvm_remove_chanctx(struct iwl_mvm *mvm,
3304                                      struct ieee80211_chanctx_conf *ctx)
3305 {
3306         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
3307         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
3308
3309         lockdep_assert_held(&mvm->mutex);
3310
3311         iwl_mvm_phy_ctxt_unref(mvm, phy_ctxt);
3312 }
3313
3314 static void iwl_mvm_remove_chanctx(struct ieee80211_hw *hw,
3315                                    struct ieee80211_chanctx_conf *ctx)
3316 {
3317         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3318
3319         mutex_lock(&mvm->mutex);
3320         __iwl_mvm_remove_chanctx(mvm, ctx);
3321         mutex_unlock(&mvm->mutex);
3322 }
3323
3324 static void iwl_mvm_change_chanctx(struct ieee80211_hw *hw,
3325                                    struct ieee80211_chanctx_conf *ctx,
3326                                    u32 changed)
3327 {
3328         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3329         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
3330         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
3331
3332         if (WARN_ONCE((phy_ctxt->ref > 1) &&
3333                       (changed & ~(IEEE80211_CHANCTX_CHANGE_WIDTH |
3334                                    IEEE80211_CHANCTX_CHANGE_RX_CHAINS |
3335                                    IEEE80211_CHANCTX_CHANGE_RADAR |
3336                                    IEEE80211_CHANCTX_CHANGE_MIN_WIDTH)),
3337                       "Cannot change PHY. Ref=%d, changed=0x%X\n",
3338                       phy_ctxt->ref, changed))
3339                 return;
3340
3341         mutex_lock(&mvm->mutex);
3342         iwl_mvm_bt_coex_vif_change(mvm);
3343         iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &ctx->min_def,
3344                                  ctx->rx_chains_static,
3345                                  ctx->rx_chains_dynamic);
3346         mutex_unlock(&mvm->mutex);
3347 }
3348
3349 static int __iwl_mvm_assign_vif_chanctx(struct iwl_mvm *mvm,
3350                                         struct ieee80211_vif *vif,
3351                                         struct ieee80211_chanctx_conf *ctx,
3352                                         bool switching_chanctx)
3353 {
3354         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
3355         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
3356         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3357         int ret;
3358
3359         lockdep_assert_held(&mvm->mutex);
3360
3361         mvmvif->phy_ctxt = phy_ctxt;
3362
3363         switch (vif->type) {
3364         case NL80211_IFTYPE_AP:
3365                 /* only needed if we're switching chanctx (i.e. during CSA) */
3366                 if (switching_chanctx) {
3367                         mvmvif->ap_ibss_active = true;
3368                         break;
3369                 }
3370         case NL80211_IFTYPE_ADHOC:
3371                 /*
3372                  * The AP binding flow is handled as part of the start_ap flow
3373                  * (in bss_info_changed), similarly for IBSS.
3374                  */
3375                 ret = 0;
3376                 goto out;
3377         case NL80211_IFTYPE_STATION:
3378                 break;
3379         case NL80211_IFTYPE_MONITOR:
3380                 /* always disable PS when a monitor interface is active */
3381                 mvmvif->ps_disabled = true;
3382                 break;
3383         default:
3384                 ret = -EINVAL;
3385                 goto out;
3386         }
3387
3388         ret = iwl_mvm_binding_add_vif(mvm, vif);
3389         if (ret)
3390                 goto out;
3391
3392         /*
3393          * Power state must be updated before quotas,
3394          * otherwise fw will complain.
3395          */
3396         iwl_mvm_power_update_mac(mvm);
3397
3398         /* Setting the quota at this stage is only required for monitor
3399          * interfaces. For the other types, the bss_info changed flow
3400          * will handle quota settings.
3401          */
3402         if (vif->type == NL80211_IFTYPE_MONITOR) {
3403                 mvmvif->monitor_active = true;
3404                 ret = iwl_mvm_update_quotas(mvm, false, NULL);
3405                 if (ret)
3406                         goto out_remove_binding;
3407         }
3408
3409         /* Handle binding during CSA */
3410         if (vif->type == NL80211_IFTYPE_AP) {
3411                 iwl_mvm_update_quotas(mvm, false, NULL);
3412                 iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
3413         }
3414
3415         if (switching_chanctx && vif->type == NL80211_IFTYPE_STATION) {
3416                 u32 duration = 2 * vif->bss_conf.beacon_int;
3417
3418                 /* iwl_mvm_protect_session() reads directly from the
3419                  * device (the system time), so make sure it is
3420                  * available.
3421                  */
3422                 ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_PROTECT_CSA);
3423                 if (ret)
3424                         goto out_remove_binding;
3425
3426                 /* Protect the session to make sure we hear the first
3427                  * beacon on the new channel.
3428                  */
3429                 iwl_mvm_protect_session(mvm, vif, duration, duration,
3430                                         vif->bss_conf.beacon_int / 2,
3431                                         true);
3432
3433                 iwl_mvm_unref(mvm, IWL_MVM_REF_PROTECT_CSA);
3434
3435                 iwl_mvm_update_quotas(mvm, false, NULL);
3436         }
3437
3438         goto out;
3439
3440 out_remove_binding:
3441         iwl_mvm_binding_remove_vif(mvm, vif);
3442         iwl_mvm_power_update_mac(mvm);
3443 out:
3444         if (ret)
3445                 mvmvif->phy_ctxt = NULL;
3446         return ret;
3447 }
3448 static int iwl_mvm_assign_vif_chanctx(struct ieee80211_hw *hw,
3449                                       struct ieee80211_vif *vif,
3450                                       struct ieee80211_chanctx_conf *ctx)
3451 {
3452         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3453         int ret;
3454
3455         mutex_lock(&mvm->mutex);
3456         ret = __iwl_mvm_assign_vif_chanctx(mvm, vif, ctx, false);
3457         mutex_unlock(&mvm->mutex);
3458
3459         return ret;
3460 }
3461
3462 static void __iwl_mvm_unassign_vif_chanctx(struct iwl_mvm *mvm,
3463                                            struct ieee80211_vif *vif,
3464                                            struct ieee80211_chanctx_conf *ctx,
3465                                            bool switching_chanctx)
3466 {
3467         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3468         struct ieee80211_vif *disabled_vif = NULL;
3469
3470         lockdep_assert_held(&mvm->mutex);
3471
3472         iwl_mvm_remove_time_event(mvm, mvmvif, &mvmvif->time_event_data);
3473
3474         switch (vif->type) {
3475         case NL80211_IFTYPE_ADHOC:
3476                 goto out;
3477         case NL80211_IFTYPE_MONITOR:
3478                 mvmvif->monitor_active = false;
3479                 mvmvif->ps_disabled = false;
3480                 break;
3481         case NL80211_IFTYPE_AP:
3482                 /* This part is triggered only during CSA */
3483                 if (!switching_chanctx || !mvmvif->ap_ibss_active)
3484                         goto out;
3485
3486                 mvmvif->csa_countdown = false;
3487
3488                 /* Set CS bit on all the stations */
3489                 iwl_mvm_modify_all_sta_disable_tx(mvm, mvmvif, true);
3490
3491                 /* Save blocked iface, the timeout is set on the next beacon */
3492                 rcu_assign_pointer(mvm->csa_tx_blocked_vif, vif);
3493
3494                 mvmvif->ap_ibss_active = false;
3495                 break;
3496         case NL80211_IFTYPE_STATION:
3497                 if (!switching_chanctx)
3498                         break;
3499
3500                 disabled_vif = vif;
3501
3502                 iwl_mvm_mac_ctxt_changed(mvm, vif, true, NULL);
3503                 break;
3504         default:
3505                 break;
3506         }
3507
3508         iwl_mvm_update_quotas(mvm, false, disabled_vif);
3509         iwl_mvm_binding_remove_vif(mvm, vif);
3510
3511 out:
3512         mvmvif->phy_ctxt = NULL;
3513         iwl_mvm_power_update_mac(mvm);
3514 }
3515
3516 static void iwl_mvm_unassign_vif_chanctx(struct ieee80211_hw *hw,
3517                                          struct ieee80211_vif *vif,
3518                                          struct ieee80211_chanctx_conf *ctx)
3519 {
3520         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3521
3522         mutex_lock(&mvm->mutex);
3523         __iwl_mvm_unassign_vif_chanctx(mvm, vif, ctx, false);
3524         mutex_unlock(&mvm->mutex);
3525 }
3526
3527 static int
3528 iwl_mvm_switch_vif_chanctx_swap(struct iwl_mvm *mvm,
3529                                 struct ieee80211_vif_chanctx_switch *vifs)
3530 {
3531         int ret;
3532
3533         mutex_lock(&mvm->mutex);
3534         __iwl_mvm_unassign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx, true);
3535         __iwl_mvm_remove_chanctx(mvm, vifs[0].old_ctx);
3536
3537         ret = __iwl_mvm_add_chanctx(mvm, vifs[0].new_ctx);
3538         if (ret) {
3539                 IWL_ERR(mvm, "failed to add new_ctx during channel switch\n");
3540                 goto out_reassign;
3541         }
3542
3543         ret = __iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].new_ctx,
3544                                            true);
3545         if (ret) {
3546                 IWL_ERR(mvm,
3547                         "failed to assign new_ctx during channel switch\n");
3548                 goto out_remove;
3549         }
3550
3551         /* we don't support TDLS during DCM - can be caused by channel switch */
3552         if (iwl_mvm_phy_ctx_count(mvm) > 1)
3553                 iwl_mvm_teardown_tdls_peers(mvm);
3554
3555         goto out;
3556
3557 out_remove:
3558         __iwl_mvm_remove_chanctx(mvm, vifs[0].new_ctx);
3559
3560 out_reassign:
3561         if (__iwl_mvm_add_chanctx(mvm, vifs[0].old_ctx)) {
3562                 IWL_ERR(mvm, "failed to add old_ctx back after failure.\n");
3563                 goto out_restart;
3564         }
3565
3566         if (__iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx,
3567                                          true)) {
3568                 IWL_ERR(mvm, "failed to reassign old_ctx after failure.\n");
3569                 goto out_restart;
3570         }
3571
3572         goto out;
3573
3574 out_restart:
3575         /* things keep failing, better restart the hw */
3576         iwl_mvm_nic_restart(mvm, false);
3577
3578 out:
3579         mutex_unlock(&mvm->mutex);
3580
3581         return ret;
3582 }
3583
3584 static int
3585 iwl_mvm_switch_vif_chanctx_reassign(struct iwl_mvm *mvm,
3586                                     struct ieee80211_vif_chanctx_switch *vifs)
3587 {
3588         int ret;
3589
3590         mutex_lock(&mvm->mutex);
3591         __iwl_mvm_unassign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx, true);
3592
3593         ret = __iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].new_ctx,
3594                                            true);
3595         if (ret) {
3596                 IWL_ERR(mvm,
3597                         "failed to assign new_ctx during channel switch\n");
3598                 goto out_reassign;
3599         }
3600
3601         goto out;
3602
3603 out_reassign:
3604         if (__iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx,
3605                                          true)) {
3606                 IWL_ERR(mvm, "failed to reassign old_ctx after failure.\n");
3607                 goto out_restart;
3608         }
3609
3610         goto out;
3611
3612 out_restart:
3613         /* things keep failing, better restart the hw */
3614         iwl_mvm_nic_restart(mvm, false);
3615
3616 out:
3617         mutex_unlock(&mvm->mutex);
3618
3619         return ret;
3620 }
3621
3622 static int iwl_mvm_switch_vif_chanctx(struct ieee80211_hw *hw,
3623                                       struct ieee80211_vif_chanctx_switch *vifs,
3624                                       int n_vifs,
3625                                       enum ieee80211_chanctx_switch_mode mode)
3626 {
3627         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3628         int ret;
3629
3630         /* we only support a single-vif right now */
3631         if (n_vifs > 1)
3632                 return -EOPNOTSUPP;
3633
3634         switch (mode) {
3635         case CHANCTX_SWMODE_SWAP_CONTEXTS:
3636                 ret = iwl_mvm_switch_vif_chanctx_swap(mvm, vifs);
3637                 break;
3638         case CHANCTX_SWMODE_REASSIGN_VIF:
3639                 ret = iwl_mvm_switch_vif_chanctx_reassign(mvm, vifs);
3640                 break;
3641         default:
3642                 ret = -EOPNOTSUPP;
3643                 break;
3644         }
3645
3646         return ret;
3647 }
3648
3649 static int iwl_mvm_set_tim(struct ieee80211_hw *hw,
3650                            struct ieee80211_sta *sta,
3651                            bool set)
3652 {
3653         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3654         struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
3655
3656         if (!mvm_sta || !mvm_sta->vif) {
3657                 IWL_ERR(mvm, "Station is not associated to a vif\n");
3658                 return -EINVAL;
3659         }
3660
3661         return iwl_mvm_mac_ctxt_beacon_changed(mvm, mvm_sta->vif);
3662 }
3663
3664 #ifdef CONFIG_NL80211_TESTMODE
3665 static const struct nla_policy iwl_mvm_tm_policy[IWL_MVM_TM_ATTR_MAX + 1] = {
3666         [IWL_MVM_TM_ATTR_CMD] = { .type = NLA_U32 },
3667         [IWL_MVM_TM_ATTR_NOA_DURATION] = { .type = NLA_U32 },
3668         [IWL_MVM_TM_ATTR_BEACON_FILTER_STATE] = { .type = NLA_U32 },
3669 };
3670
3671 static int __iwl_mvm_mac_testmode_cmd(struct iwl_mvm *mvm,
3672                                       struct ieee80211_vif *vif,
3673                                       void *data, int len)
3674 {
3675         struct nlattr *tb[IWL_MVM_TM_ATTR_MAX + 1];
3676         int err;
3677         u32 noa_duration;
3678
3679         err = nla_parse(tb, IWL_MVM_TM_ATTR_MAX, data, len, iwl_mvm_tm_policy);
3680         if (err)
3681                 return err;
3682
3683         if (!tb[IWL_MVM_TM_ATTR_CMD])
3684                 return -EINVAL;
3685
3686         switch (nla_get_u32(tb[IWL_MVM_TM_ATTR_CMD])) {
3687         case IWL_MVM_TM_CMD_SET_NOA:
3688                 if (!vif || vif->type != NL80211_IFTYPE_AP || !vif->p2p ||
3689                     !vif->bss_conf.enable_beacon ||
3690                     !tb[IWL_MVM_TM_ATTR_NOA_DURATION])
3691                         return -EINVAL;
3692
3693                 noa_duration = nla_get_u32(tb[IWL_MVM_TM_ATTR_NOA_DURATION]);
3694                 if (noa_duration >= vif->bss_conf.beacon_int)
3695                         return -EINVAL;
3696
3697                 mvm->noa_duration = noa_duration;
3698                 mvm->noa_vif = vif;
3699
3700                 return iwl_mvm_update_quotas(mvm, false, NULL);
3701         case IWL_MVM_TM_CMD_SET_BEACON_FILTER:
3702                 /* must be associated client vif - ignore authorized */
3703                 if (!vif || vif->type != NL80211_IFTYPE_STATION ||
3704                     !vif->bss_conf.assoc || !vif->bss_conf.dtim_period ||
3705                     !tb[IWL_MVM_TM_ATTR_BEACON_FILTER_STATE])
3706                         return -EINVAL;
3707
3708                 if (nla_get_u32(tb[IWL_MVM_TM_ATTR_BEACON_FILTER_STATE]))
3709                         return iwl_mvm_enable_beacon_filter(mvm, vif, 0);
3710                 return iwl_mvm_disable_beacon_filter(mvm, vif, 0);
3711         }
3712
3713         return -EOPNOTSUPP;
3714 }
3715
3716 static int iwl_mvm_mac_testmode_cmd(struct ieee80211_hw *hw,
3717                                     struct ieee80211_vif *vif,
3718                                     void *data, int len)
3719 {
3720         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3721         int err;
3722
3723         mutex_lock(&mvm->mutex);
3724         err = __iwl_mvm_mac_testmode_cmd(mvm, vif, data, len);
3725         mutex_unlock(&mvm->mutex);
3726
3727         return err;
3728 }
3729 #endif
3730
3731 static void iwl_mvm_channel_switch(struct ieee80211_hw *hw,
3732                                    struct ieee80211_vif *vif,
3733                                    struct ieee80211_channel_switch *chsw)
3734 {
3735         /* By implementing this operation, we prevent mac80211 from
3736          * starting its own channel switch timer, so that we can call
3737          * ieee80211_chswitch_done() ourselves at the right time
3738          * (which is when the absence time event starts).
3739          */
3740
3741         IWL_DEBUG_MAC80211(IWL_MAC80211_GET_MVM(hw),
3742                            "dummy channel switch op\n");
3743 }
3744
3745 static int iwl_mvm_pre_channel_switch(struct ieee80211_hw *hw,
3746                                       struct ieee80211_vif *vif,
3747                                       struct ieee80211_channel_switch *chsw)
3748 {
3749         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3750         struct ieee80211_vif *csa_vif;
3751         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3752         u32 apply_time;
3753         int ret;
3754
3755         mutex_lock(&mvm->mutex);
3756
3757         mvmvif->csa_failed = false;
3758
3759         IWL_DEBUG_MAC80211(mvm, "pre CSA to freq %d\n",
3760                            chsw->chandef.center_freq1);
3761
3762         iwl_fw_dbg_trigger_simple_stop(mvm, vif, FW_DBG_TRIGGER_CHANNEL_SWITCH);
3763
3764         switch (vif->type) {
3765         case NL80211_IFTYPE_AP:
3766                 csa_vif =
3767                         rcu_dereference_protected(mvm->csa_vif,
3768                                                   lockdep_is_held(&mvm->mutex));
3769                 if (WARN_ONCE(csa_vif && csa_vif->csa_active,
3770                               "Another CSA is already in progress")) {
3771                         ret = -EBUSY;
3772                         goto out_unlock;
3773                 }
3774
3775                 rcu_assign_pointer(mvm->csa_vif, vif);
3776
3777                 if (WARN_ONCE(mvmvif->csa_countdown,
3778                               "Previous CSA countdown didn't complete")) {
3779                         ret = -EBUSY;
3780                         goto out_unlock;
3781                 }
3782
3783                 break;
3784         case NL80211_IFTYPE_STATION:
3785                 /* Schedule the time event to a bit before beacon 1,
3786                  * to make sure we're in the new channel when the
3787                  * GO/AP arrives.
3788                  */
3789                 apply_time = chsw->device_timestamp +
3790                         ((vif->bss_conf.beacon_int * (chsw->count - 1) -
3791                           IWL_MVM_CHANNEL_SWITCH_TIME_CLIENT) * 1024);
3792
3793                 if (chsw->block_tx)
3794                         iwl_mvm_csa_client_absent(mvm, vif);
3795
3796                 iwl_mvm_schedule_csa_period(mvm, vif, vif->bss_conf.beacon_int,
3797                                             apply_time);
3798                 if (mvmvif->bf_data.bf_enabled) {
3799                         ret = iwl_mvm_disable_beacon_filter(mvm, vif, 0);
3800                         if (ret)
3801                                 goto out_unlock;
3802                 }
3803
3804                 break;
3805         default:
3806                 break;
3807         }
3808
3809         mvmvif->ps_disabled = true;
3810
3811         ret = iwl_mvm_power_update_ps(mvm);
3812         if (ret)
3813                 goto out_unlock;
3814
3815         /* we won't be on this channel any longer */
3816         iwl_mvm_teardown_tdls_peers(mvm);
3817
3818 out_unlock:
3819         mutex_unlock(&mvm->mutex);
3820
3821         return ret;
3822 }
3823
3824 static int iwl_mvm_post_channel_switch(struct ieee80211_hw *hw,
3825                                        struct ieee80211_vif *vif)
3826 {
3827         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3828         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3829         int ret;
3830
3831         mutex_lock(&mvm->mutex);
3832
3833         if (mvmvif->csa_failed) {
3834                 mvmvif->csa_failed = false;
3835                 ret = -EIO;
3836                 goto out_unlock;
3837         }
3838
3839         if (vif->type == NL80211_IFTYPE_STATION) {
3840                 struct iwl_mvm_sta *mvmsta;
3841
3842                 mvmsta = iwl_mvm_sta_from_staid_protected(mvm,
3843                                                           mvmvif->ap_sta_id);
3844
3845                 if (WARN_ON(!mvmsta)) {
3846                         ret = -EIO;
3847                         goto out_unlock;
3848                 }
3849
3850                 iwl_mvm_sta_modify_disable_tx(mvm, mvmsta, false);
3851
3852                 iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
3853
3854                 ret = iwl_mvm_enable_beacon_filter(mvm, vif, 0);
3855                 if (ret)
3856                         goto out_unlock;
3857
3858                 iwl_mvm_stop_session_protection(mvm, vif);
3859         }
3860
3861         mvmvif->ps_disabled = false;
3862
3863         ret = iwl_mvm_power_update_ps(mvm);
3864
3865 out_unlock:
3866         mutex_unlock(&mvm->mutex);
3867
3868         return ret;
3869 }
3870
3871 static void iwl_mvm_mac_flush(struct ieee80211_hw *hw,
3872                               struct ieee80211_vif *vif, u32 queues, bool drop)
3873 {
3874         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3875         struct iwl_mvm_vif *mvmvif;
3876         struct iwl_mvm_sta *mvmsta;
3877         struct ieee80211_sta *sta;
3878         int i;
3879         u32 msk = 0;
3880
3881         if (!vif || vif->type != NL80211_IFTYPE_STATION)
3882                 return;
3883
3884         mutex_lock(&mvm->mutex);
3885         mvmvif = iwl_mvm_vif_from_mac80211(vif);
3886
3887         /* flush the AP-station and all TDLS peers */
3888         for (i = 0; i < IWL_MVM_STATION_COUNT; i++) {
3889                 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[i],
3890                                                 lockdep_is_held(&mvm->mutex));
3891                 if (IS_ERR_OR_NULL(sta))
3892                         continue;
3893
3894                 mvmsta = iwl_mvm_sta_from_mac80211(sta);
3895                 if (mvmsta->vif != vif)
3896                         continue;
3897
3898                 /* make sure only TDLS peers or the AP are flushed */
3899                 WARN_ON(i != mvmvif->ap_sta_id && !sta->tdls);
3900
3901                 msk |= mvmsta->tfd_queue_msk;
3902         }
3903
3904         if (drop) {
3905                 if (iwl_mvm_flush_tx_path(mvm, msk, true))
3906                         IWL_ERR(mvm, "flush request fail\n");
3907                 mutex_unlock(&mvm->mutex);
3908         } else {
3909                 mutex_unlock(&mvm->mutex);
3910
3911                 /* this can take a while, and we may need/want other operations
3912                  * to succeed while doing this, so do it without the mutex held
3913                  */
3914                 iwl_trans_wait_tx_queue_empty(mvm->trans, msk);
3915         }
3916 }
3917
3918 static int iwl_mvm_mac_get_survey(struct ieee80211_hw *hw, int idx,
3919                                   struct survey_info *survey)
3920 {
3921         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3922         int ret;
3923
3924         memset(survey, 0, sizeof(*survey));
3925
3926         /* only support global statistics right now */
3927         if (idx != 0)
3928                 return -ENOENT;
3929
3930         if (fw_has_capa(&mvm->fw->ucode_capa,
3931                         IWL_UCODE_TLV_CAPA_RADIO_BEACON_STATS))
3932                 return -ENOENT;
3933
3934         mutex_lock(&mvm->mutex);
3935
3936         if (mvm->ucode_loaded) {
3937                 ret = iwl_mvm_request_statistics(mvm, false);
3938                 if (ret)
3939                         goto out;
3940         }
3941
3942         survey->filled = SURVEY_INFO_TIME |
3943                          SURVEY_INFO_TIME_RX |
3944                          SURVEY_INFO_TIME_TX |
3945                          SURVEY_INFO_TIME_SCAN;
3946         survey->time = mvm->accu_radio_stats.on_time_rf +
3947                        mvm->radio_stats.on_time_rf;
3948         do_div(survey->time, USEC_PER_MSEC);
3949
3950         survey->time_rx = mvm->accu_radio_stats.rx_time +
3951                           mvm->radio_stats.rx_time;
3952         do_div(survey->time_rx, USEC_PER_MSEC);
3953
3954         survey->time_tx = mvm->accu_radio_stats.tx_time +
3955                           mvm->radio_stats.tx_time;
3956         do_div(survey->time_tx, USEC_PER_MSEC);
3957
3958         survey->time_scan = mvm->accu_radio_stats.on_time_scan +
3959                             mvm->radio_stats.on_time_scan;
3960         do_div(survey->time_scan, USEC_PER_MSEC);
3961
3962         ret = 0;
3963  out:
3964         mutex_unlock(&mvm->mutex);
3965         return ret;
3966 }
3967
3968 static void iwl_mvm_mac_sta_statistics(struct ieee80211_hw *hw,
3969                                        struct ieee80211_vif *vif,
3970                                        struct ieee80211_sta *sta,
3971                                        struct station_info *sinfo)
3972 {
3973         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
3974         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3975         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
3976
3977         if (fw_has_capa(&mvm->fw->ucode_capa,
3978                         IWL_UCODE_TLV_CAPA_RADIO_BEACON_STATS))
3979                 return;
3980
3981         /* if beacon filtering isn't on mac80211 does it anyway */
3982         if (!(vif->driver_flags & IEEE80211_VIF_BEACON_FILTER))
3983                 return;
3984
3985         if (!vif->bss_conf.assoc)
3986                 return;
3987
3988         mutex_lock(&mvm->mutex);
3989
3990         if (mvmvif->ap_sta_id != mvmsta->sta_id)
3991                 goto unlock;
3992
3993         if (iwl_mvm_request_statistics(mvm, false))
3994                 goto unlock;
3995
3996         sinfo->rx_beacon = mvmvif->beacon_stats.num_beacons +
3997                            mvmvif->beacon_stats.accu_num_beacons;
3998         sinfo->filled |= BIT(NL80211_STA_INFO_BEACON_RX);
3999         if (mvmvif->beacon_stats.avg_signal) {
4000                 /* firmware only reports a value after RXing a few beacons */
4001                 sinfo->rx_beacon_signal_avg = mvmvif->beacon_stats.avg_signal;
4002                 sinfo->filled |= BIT(NL80211_STA_INFO_BEACON_SIGNAL_AVG);
4003         }
4004  unlock:
4005         mutex_unlock(&mvm->mutex);
4006 }
4007
4008 static void iwl_mvm_event_mlme_callback(struct iwl_mvm *mvm,
4009                                         struct ieee80211_vif *vif,
4010                                         const struct ieee80211_event *event)
4011 {
4012 #define CHECK_MLME_TRIGGER(_mvm, _trig, _buf, _cnt, _fmt...)    \
4013         do {                                                    \
4014                 if ((_cnt) && --(_cnt))                         \
4015                         break;                                  \
4016                 iwl_mvm_fw_dbg_collect_trig(_mvm, _trig, _fmt);\
4017         } while (0)
4018
4019         struct iwl_fw_dbg_trigger_tlv *trig;
4020         struct iwl_fw_dbg_trigger_mlme *trig_mlme;
4021
4022         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_MLME))
4023                 return;
4024
4025         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_MLME);
4026         trig_mlme = (void *)trig->data;
4027         if (!iwl_fw_dbg_trigger_check_stop(mvm, vif, trig))
4028                 return;
4029
4030         if (event->u.mlme.data == ASSOC_EVENT) {
4031                 if (event->u.mlme.status == MLME_DENIED)
4032                         CHECK_MLME_TRIGGER(mvm, trig, buf,
4033                                            trig_mlme->stop_assoc_denied,
4034                                            "DENIED ASSOC: reason %d",
4035                                             event->u.mlme.reason);
4036                 else if (event->u.mlme.status == MLME_TIMEOUT)
4037                         CHECK_MLME_TRIGGER(mvm, trig, buf,
4038                                            trig_mlme->stop_assoc_timeout,
4039                                            "ASSOC TIMEOUT");
4040         } else if (event->u.mlme.data == AUTH_EVENT) {
4041                 if (event->u.mlme.status == MLME_DENIED)
4042                         CHECK_MLME_TRIGGER(mvm, trig, buf,
4043                                            trig_mlme->stop_auth_denied,
4044                                            "DENIED AUTH: reason %d",
4045                                            event->u.mlme.reason);
4046                 else if (event->u.mlme.status == MLME_TIMEOUT)
4047                         CHECK_MLME_TRIGGER(mvm, trig, buf,
4048                                            trig_mlme->stop_auth_timeout,
4049                                            "AUTH TIMEOUT");
4050         } else if (event->u.mlme.data == DEAUTH_RX_EVENT) {
4051                 CHECK_MLME_TRIGGER(mvm, trig, buf,
4052                                    trig_mlme->stop_rx_deauth,
4053                                    "DEAUTH RX %d", event->u.mlme.reason);
4054         } else if (event->u.mlme.data == DEAUTH_TX_EVENT) {
4055                 CHECK_MLME_TRIGGER(mvm, trig, buf,
4056                                    trig_mlme->stop_tx_deauth,
4057                                    "DEAUTH TX %d", event->u.mlme.reason);
4058         }
4059 #undef CHECK_MLME_TRIGGER
4060 }
4061
4062 static void iwl_mvm_event_bar_rx_callback(struct iwl_mvm *mvm,
4063                                           struct ieee80211_vif *vif,
4064                                           const struct ieee80211_event *event)
4065 {
4066         struct iwl_fw_dbg_trigger_tlv *trig;
4067         struct iwl_fw_dbg_trigger_ba *ba_trig;
4068
4069         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_BA))
4070                 return;
4071
4072         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_BA);
4073         ba_trig = (void *)trig->data;
4074         if (!iwl_fw_dbg_trigger_check_stop(mvm, vif, trig))
4075                 return;
4076
4077         if (!(le16_to_cpu(ba_trig->rx_bar) & BIT(event->u.ba.tid)))
4078                 return;
4079
4080         iwl_mvm_fw_dbg_collect_trig(mvm, trig,
4081                                     "BAR received from %pM, tid %d, ssn %d",
4082                                     event->u.ba.sta->addr, event->u.ba.tid,
4083                                     event->u.ba.ssn);
4084 }
4085
4086 static void
4087 iwl_mvm_event_frame_timeout_callback(struct iwl_mvm *mvm,
4088                                      struct ieee80211_vif *vif,
4089                                      const struct ieee80211_event *event)
4090 {
4091         struct iwl_fw_dbg_trigger_tlv *trig;
4092         struct iwl_fw_dbg_trigger_ba *ba_trig;
4093
4094         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_BA))
4095                 return;
4096
4097         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_BA);
4098         ba_trig = (void *)trig->data;
4099         if (!iwl_fw_dbg_trigger_check_stop(mvm, vif, trig))
4100                 return;
4101
4102         if (!(le16_to_cpu(ba_trig->frame_timeout) & BIT(event->u.ba.tid)))
4103                 return;
4104
4105         iwl_mvm_fw_dbg_collect_trig(mvm, trig,
4106                                     "Frame from %pM timed out, tid %d",
4107                                     event->u.ba.sta->addr, event->u.ba.tid);
4108 }
4109
4110 static void iwl_mvm_mac_event_callback(struct ieee80211_hw *hw,
4111                                        struct ieee80211_vif *vif,
4112                                        const struct ieee80211_event *event)
4113 {
4114         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
4115
4116         switch (event->type) {
4117         case MLME_EVENT:
4118                 iwl_mvm_event_mlme_callback(mvm, vif, event);
4119                 break;
4120         case BAR_RX_EVENT:
4121                 iwl_mvm_event_bar_rx_callback(mvm, vif, event);
4122                 break;
4123         case BA_FRAME_TIMEOUT:
4124                 iwl_mvm_event_frame_timeout_callback(mvm, vif, event);
4125                 break;
4126         default:
4127                 break;
4128         }
4129 }
4130
4131 const struct ieee80211_ops iwl_mvm_hw_ops = {
4132         .tx = iwl_mvm_mac_tx,
4133         .ampdu_action = iwl_mvm_mac_ampdu_action,
4134         .start = iwl_mvm_mac_start,
4135         .reconfig_complete = iwl_mvm_mac_reconfig_complete,
4136         .stop = iwl_mvm_mac_stop,
4137         .add_interface = iwl_mvm_mac_add_interface,
4138         .remove_interface = iwl_mvm_mac_remove_interface,
4139         .config = iwl_mvm_mac_config,
4140         .prepare_multicast = iwl_mvm_prepare_multicast,
4141         .configure_filter = iwl_mvm_configure_filter,
4142         .bss_info_changed = iwl_mvm_bss_info_changed,
4143         .hw_scan = iwl_mvm_mac_hw_scan,
4144         .cancel_hw_scan = iwl_mvm_mac_cancel_hw_scan,
4145         .sta_pre_rcu_remove = iwl_mvm_sta_pre_rcu_remove,
4146         .sta_state = iwl_mvm_mac_sta_state,
4147         .sta_notify = iwl_mvm_mac_sta_notify,
4148         .allow_buffered_frames = iwl_mvm_mac_allow_buffered_frames,
4149         .release_buffered_frames = iwl_mvm_mac_release_buffered_frames,
4150         .set_rts_threshold = iwl_mvm_mac_set_rts_threshold,
4151         .sta_rc_update = iwl_mvm_sta_rc_update,
4152         .conf_tx = iwl_mvm_mac_conf_tx,
4153         .mgd_prepare_tx = iwl_mvm_mac_mgd_prepare_tx,
4154         .mgd_protect_tdls_discover = iwl_mvm_mac_mgd_protect_tdls_discover,
4155         .flush = iwl_mvm_mac_flush,
4156         .sched_scan_start = iwl_mvm_mac_sched_scan_start,
4157         .sched_scan_stop = iwl_mvm_mac_sched_scan_stop,
4158         .set_key = iwl_mvm_mac_set_key,
4159         .update_tkip_key = iwl_mvm_mac_update_tkip_key,
4160         .remain_on_channel = iwl_mvm_roc,
4161         .cancel_remain_on_channel = iwl_mvm_cancel_roc,
4162         .add_chanctx = iwl_mvm_add_chanctx,
4163         .remove_chanctx = iwl_mvm_remove_chanctx,
4164         .change_chanctx = iwl_mvm_change_chanctx,
4165         .assign_vif_chanctx = iwl_mvm_assign_vif_chanctx,
4166         .unassign_vif_chanctx = iwl_mvm_unassign_vif_chanctx,
4167         .switch_vif_chanctx = iwl_mvm_switch_vif_chanctx,
4168
4169         .start_ap = iwl_mvm_start_ap_ibss,
4170         .stop_ap = iwl_mvm_stop_ap_ibss,
4171         .join_ibss = iwl_mvm_start_ap_ibss,
4172         .leave_ibss = iwl_mvm_stop_ap_ibss,
4173
4174         .set_tim = iwl_mvm_set_tim,
4175
4176         .channel_switch = iwl_mvm_channel_switch,
4177         .pre_channel_switch = iwl_mvm_pre_channel_switch,
4178         .post_channel_switch = iwl_mvm_post_channel_switch,
4179
4180         .tdls_channel_switch = iwl_mvm_tdls_channel_switch,
4181         .tdls_cancel_channel_switch = iwl_mvm_tdls_cancel_channel_switch,
4182         .tdls_recv_channel_switch = iwl_mvm_tdls_recv_channel_switch,
4183
4184         .event_callback = iwl_mvm_mac_event_callback,
4185
4186         CFG80211_TESTMODE_CMD(iwl_mvm_mac_testmode_cmd)
4187
4188 #ifdef CONFIG_PM_SLEEP
4189         /* look at d3.c */
4190         .suspend = iwl_mvm_suspend,
4191         .resume = iwl_mvm_resume,
4192         .set_wakeup = iwl_mvm_set_wakeup,
4193         .set_rekey_data = iwl_mvm_set_rekey_data,
4194 #if IS_ENABLED(CONFIG_IPV6)
4195         .ipv6_addr_change = iwl_mvm_ipv6_addr_change,
4196 #endif
4197         .set_default_unicast_key = iwl_mvm_set_default_unicast_key,
4198 #endif
4199         .get_survey = iwl_mvm_mac_get_survey,
4200         .sta_statistics = iwl_mvm_mac_sta_statistics,
4201 };