iwlwifi: mvm: Add tx power condition to bss_info_changed_ap_ibss
[cascardo/linux.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 <net/mac80211.h>
73 #include <net/ieee80211_radiotap.h>
74 #include <net/tcp.h>
75
76 #include "iwl-op-mode.h"
77 #include "iwl-io.h"
78 #include "mvm.h"
79 #include "sta.h"
80 #include "time-event.h"
81 #include "iwl-eeprom-parse.h"
82 #include "fw-api-scan.h"
83 #include "iwl-phy-db.h"
84 #include "testmode.h"
85 #include "iwl-fw-error-dump.h"
86 #include "iwl-prph.h"
87
88 static const struct ieee80211_iface_limit iwl_mvm_limits[] = {
89         {
90                 .max = 1,
91                 .types = BIT(NL80211_IFTYPE_STATION),
92         },
93         {
94                 .max = 1,
95                 .types = BIT(NL80211_IFTYPE_AP) |
96                         BIT(NL80211_IFTYPE_P2P_CLIENT) |
97                         BIT(NL80211_IFTYPE_P2P_GO),
98         },
99         {
100                 .max = 1,
101                 .types = BIT(NL80211_IFTYPE_P2P_DEVICE),
102         },
103 };
104
105 static const struct ieee80211_iface_combination iwl_mvm_iface_combinations[] = {
106         {
107                 .num_different_channels = 1,
108                 .max_interfaces = 3,
109                 .limits = iwl_mvm_limits,
110                 .n_limits = ARRAY_SIZE(iwl_mvm_limits),
111         },
112 };
113
114 #ifdef CONFIG_PM_SLEEP
115 static const struct nl80211_wowlan_tcp_data_token_feature
116 iwl_mvm_wowlan_tcp_token_feature = {
117         .min_len = 0,
118         .max_len = 255,
119         .bufsize = IWL_WOWLAN_REMOTE_WAKE_MAX_TOKENS,
120 };
121
122 static const struct wiphy_wowlan_tcp_support iwl_mvm_wowlan_tcp_support = {
123         .tok = &iwl_mvm_wowlan_tcp_token_feature,
124         .data_payload_max = IWL_WOWLAN_TCP_MAX_PACKET_LEN -
125                             sizeof(struct ethhdr) -
126                             sizeof(struct iphdr) -
127                             sizeof(struct tcphdr),
128         .data_interval_max = 65535, /* __le16 in API */
129         .wake_payload_max = IWL_WOWLAN_REMOTE_WAKE_MAX_PACKET_LEN -
130                             sizeof(struct ethhdr) -
131                             sizeof(struct iphdr) -
132                             sizeof(struct tcphdr),
133         .seq = true,
134 };
135 #endif
136
137 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
138 /*
139  * Use the reserved field to indicate magic values.
140  * these values will only be used internally by the driver,
141  * and won't make it to the fw (reserved will be 0).
142  * BC_FILTER_MAGIC_IP - configure the val of this attribute to
143  *      be the vif's ip address. in case there is not a single
144  *      ip address (0, or more than 1), this attribute will
145  *      be skipped.
146  * BC_FILTER_MAGIC_MAC - set the val of this attribute to
147  *      the LSB bytes of the vif's mac address
148  */
149 enum {
150         BC_FILTER_MAGIC_NONE = 0,
151         BC_FILTER_MAGIC_IP,
152         BC_FILTER_MAGIC_MAC,
153 };
154
155 static const struct iwl_fw_bcast_filter iwl_mvm_default_bcast_filters[] = {
156         {
157                 /* arp */
158                 .discard = 0,
159                 .frame_type = BCAST_FILTER_FRAME_TYPE_ALL,
160                 .attrs = {
161                         {
162                                 /* frame type - arp, hw type - ethernet */
163                                 .offset_type =
164                                         BCAST_FILTER_OFFSET_PAYLOAD_START,
165                                 .offset = sizeof(rfc1042_header),
166                                 .val = cpu_to_be32(0x08060001),
167                                 .mask = cpu_to_be32(0xffffffff),
168                         },
169                         {
170                                 /* arp dest ip */
171                                 .offset_type =
172                                         BCAST_FILTER_OFFSET_PAYLOAD_START,
173                                 .offset = sizeof(rfc1042_header) + 2 +
174                                           sizeof(struct arphdr) +
175                                           ETH_ALEN + sizeof(__be32) +
176                                           ETH_ALEN,
177                                 .mask = cpu_to_be32(0xffffffff),
178                                 /* mark it as special field */
179                                 .reserved1 = cpu_to_le16(BC_FILTER_MAGIC_IP),
180                         },
181                 },
182         },
183         {
184                 /* dhcp offer bcast */
185                 .discard = 0,
186                 .frame_type = BCAST_FILTER_FRAME_TYPE_IPV4,
187                 .attrs = {
188                         {
189                                 /* udp dest port - 68 (bootp client)*/
190                                 .offset_type = BCAST_FILTER_OFFSET_IP_END,
191                                 .offset = offsetof(struct udphdr, dest),
192                                 .val = cpu_to_be32(0x00440000),
193                                 .mask = cpu_to_be32(0xffff0000),
194                         },
195                         {
196                                 /* dhcp - lsb bytes of client hw address */
197                                 .offset_type = BCAST_FILTER_OFFSET_IP_END,
198                                 .offset = 38,
199                                 .mask = cpu_to_be32(0xffffffff),
200                                 /* mark it as special field */
201                                 .reserved1 = cpu_to_le16(BC_FILTER_MAGIC_MAC),
202                         },
203                 },
204         },
205         /* last filter must be empty */
206         {},
207 };
208 #endif
209
210 void iwl_mvm_ref(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
211 {
212         if (!iwl_mvm_is_d0i3_supported(mvm))
213                 return;
214
215         IWL_DEBUG_RPM(mvm, "Take mvm reference - type %d\n", ref_type);
216         spin_lock_bh(&mvm->refs_lock);
217         mvm->refs[ref_type]++;
218         spin_unlock_bh(&mvm->refs_lock);
219         iwl_trans_ref(mvm->trans);
220 }
221
222 void iwl_mvm_unref(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
223 {
224         if (!iwl_mvm_is_d0i3_supported(mvm))
225                 return;
226
227         IWL_DEBUG_RPM(mvm, "Leave mvm reference - type %d\n", ref_type);
228         spin_lock_bh(&mvm->refs_lock);
229         WARN_ON(!mvm->refs[ref_type]--);
230         spin_unlock_bh(&mvm->refs_lock);
231         iwl_trans_unref(mvm->trans);
232 }
233
234 static void iwl_mvm_unref_all_except(struct iwl_mvm *mvm,
235                                      enum iwl_mvm_ref_type except_ref)
236 {
237         int i, j;
238
239         if (!iwl_mvm_is_d0i3_supported(mvm))
240                 return;
241
242         spin_lock_bh(&mvm->refs_lock);
243         for (i = 0; i < IWL_MVM_REF_COUNT; i++) {
244                 if (except_ref == i || !mvm->refs[i])
245                         continue;
246
247                 IWL_DEBUG_RPM(mvm, "Cleanup: remove mvm ref type %d (%d)\n",
248                               i, mvm->refs[i]);
249                 for (j = 0; j < mvm->refs[i]; j++)
250                         iwl_trans_unref(mvm->trans);
251                 mvm->refs[i] = 0;
252         }
253         spin_unlock_bh(&mvm->refs_lock);
254 }
255
256 int iwl_mvm_ref_sync(struct iwl_mvm *mvm, enum iwl_mvm_ref_type ref_type)
257 {
258         iwl_mvm_ref(mvm, ref_type);
259
260         if (!wait_event_timeout(mvm->d0i3_exit_waitq,
261                                 !test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status),
262                                 HZ)) {
263                 WARN_ON_ONCE(1);
264                 iwl_mvm_unref(mvm, ref_type);
265                 return -EIO;
266         }
267
268         return 0;
269 }
270
271 static void iwl_mvm_reset_phy_ctxts(struct iwl_mvm *mvm)
272 {
273         int i;
274
275         memset(mvm->phy_ctxts, 0, sizeof(mvm->phy_ctxts));
276         for (i = 0; i < NUM_PHY_CTX; i++) {
277                 mvm->phy_ctxts[i].id = i;
278                 mvm->phy_ctxts[i].ref = 0;
279         }
280 }
281
282 int iwl_mvm_mac_setup_register(struct iwl_mvm *mvm)
283 {
284         struct ieee80211_hw *hw = mvm->hw;
285         int num_mac, ret, i;
286
287         /* Tell mac80211 our characteristics */
288         hw->flags = IEEE80211_HW_SIGNAL_DBM |
289                     IEEE80211_HW_SPECTRUM_MGMT |
290                     IEEE80211_HW_REPORTS_TX_ACK_STATUS |
291                     IEEE80211_HW_QUEUE_CONTROL |
292                     IEEE80211_HW_WANT_MONITOR_VIF |
293                     IEEE80211_HW_SUPPORTS_PS |
294                     IEEE80211_HW_SUPPORTS_DYNAMIC_PS |
295                     IEEE80211_HW_AMPDU_AGGREGATION |
296                     IEEE80211_HW_TIMING_BEACON_ONLY |
297                     IEEE80211_HW_CONNECTION_MONITOR |
298                     IEEE80211_HW_CHANCTX_STA_CSA |
299                     IEEE80211_HW_SUPPORTS_CLONED_SKBS;
300
301         hw->queues = mvm->first_agg_queue;
302         hw->offchannel_tx_hw_queue = IWL_MVM_OFFCHANNEL_QUEUE;
303         hw->radiotap_mcs_details |= IEEE80211_RADIOTAP_MCS_HAVE_FEC |
304                                     IEEE80211_RADIOTAP_MCS_HAVE_STBC;
305         hw->radiotap_vht_details |= IEEE80211_RADIOTAP_VHT_KNOWN_STBC;
306         hw->rate_control_algorithm = "iwl-mvm-rs";
307
308         /*
309          * Enable 11w if advertised by firmware and software crypto
310          * is not enabled (as the firmware will interpret some mgmt
311          * packets, so enabling it with software crypto isn't safe)
312          */
313         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_MFP &&
314             !iwlwifi_mod_params.sw_crypto)
315                 hw->flags |= IEEE80211_HW_MFP_CAPABLE;
316
317         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_UAPSD_SUPPORT &&
318             IWL_UCODE_API(mvm->fw->ucode_ver) >= 9 &&
319             !iwlwifi_mod_params.uapsd_disable) {
320                 hw->flags |= IEEE80211_HW_SUPPORTS_UAPSD;
321                 hw->uapsd_queues = IWL_MVM_UAPSD_QUEUES;
322                 hw->uapsd_max_sp_len = IWL_UAPSD_MAX_SP;
323         }
324
325         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)
326                 hw->flags |= IEEE80211_SINGLE_HW_SCAN_ON_ALL_BANDS;
327
328         hw->sta_data_size = sizeof(struct iwl_mvm_sta);
329         hw->vif_data_size = sizeof(struct iwl_mvm_vif);
330         hw->chanctx_data_size = sizeof(u16);
331
332         hw->wiphy->interface_modes = BIT(NL80211_IFTYPE_STATION) |
333                 BIT(NL80211_IFTYPE_P2P_CLIENT) |
334                 BIT(NL80211_IFTYPE_AP) |
335                 BIT(NL80211_IFTYPE_P2P_GO) |
336                 BIT(NL80211_IFTYPE_P2P_DEVICE) |
337                 BIT(NL80211_IFTYPE_ADHOC);
338
339         hw->wiphy->flags |= WIPHY_FLAG_IBSS_RSN;
340         hw->wiphy->regulatory_flags |= REGULATORY_CUSTOM_REG |
341                                        REGULATORY_DISABLE_BEACON_HINTS;
342
343         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_GO_UAPSD)
344                 hw->wiphy->flags |= WIPHY_FLAG_AP_UAPSD;
345
346         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_CSA_FLOW)
347                 hw->wiphy->flags |= WIPHY_FLAG_HAS_CHANNEL_SWITCH;
348
349         hw->wiphy->iface_combinations = iwl_mvm_iface_combinations;
350         hw->wiphy->n_iface_combinations =
351                 ARRAY_SIZE(iwl_mvm_iface_combinations);
352
353         hw->wiphy->max_remain_on_channel_duration = 10000;
354         hw->max_listen_interval = IWL_CONN_MAX_LISTEN_INTERVAL;
355
356         /* Extract MAC address */
357         memcpy(mvm->addresses[0].addr, mvm->nvm_data->hw_addr, ETH_ALEN);
358         hw->wiphy->addresses = mvm->addresses;
359         hw->wiphy->n_addresses = 1;
360
361         /* Extract additional MAC addresses if available */
362         num_mac = (mvm->nvm_data->n_hw_addrs > 1) ?
363                 min(IWL_MVM_MAX_ADDRESSES, mvm->nvm_data->n_hw_addrs) : 1;
364
365         for (i = 1; i < num_mac; i++) {
366                 memcpy(mvm->addresses[i].addr, mvm->addresses[i-1].addr,
367                        ETH_ALEN);
368                 mvm->addresses[i].addr[5]++;
369                 hw->wiphy->n_addresses++;
370         }
371
372         iwl_mvm_reset_phy_ctxts(mvm);
373
374         hw->wiphy->max_scan_ie_len = iwl_mvm_max_scan_ie_len(mvm, false);
375
376         hw->wiphy->max_scan_ssids = PROBE_OPTION_MAX;
377
378         if (mvm->nvm_data->bands[IEEE80211_BAND_2GHZ].n_channels)
379                 hw->wiphy->bands[IEEE80211_BAND_2GHZ] =
380                         &mvm->nvm_data->bands[IEEE80211_BAND_2GHZ];
381         if (mvm->nvm_data->bands[IEEE80211_BAND_5GHZ].n_channels)
382                 hw->wiphy->bands[IEEE80211_BAND_5GHZ] =
383                         &mvm->nvm_data->bands[IEEE80211_BAND_5GHZ];
384
385         hw->wiphy->hw_version = mvm->trans->hw_id;
386
387         if (iwlmvm_mod_params.power_scheme != IWL_POWER_SCHEME_CAM)
388                 hw->wiphy->flags |= WIPHY_FLAG_PS_ON_BY_DEFAULT;
389         else
390                 hw->wiphy->flags &= ~WIPHY_FLAG_PS_ON_BY_DEFAULT;
391
392         if (IWL_UCODE_API(mvm->fw->ucode_ver) >= 10) {
393                 hw->wiphy->flags |= WIPHY_FLAG_SUPPORTS_SCHED_SCAN;
394                 hw->wiphy->max_sched_scan_ssids = PROBE_OPTION_MAX;
395                 hw->wiphy->max_match_sets = IWL_SCAN_MAX_PROFILES;
396                 /* we create the 802.11 header and zero length SSID IE. */
397                 hw->wiphy->max_sched_scan_ie_len =
398                         SCAN_OFFLOAD_PROBE_REQ_SIZE - 24 - 2;
399         }
400
401         hw->wiphy->features |= NL80211_FEATURE_P2P_GO_CTWIN |
402                                NL80211_FEATURE_LOW_PRIORITY_SCAN |
403                                NL80211_FEATURE_P2P_GO_OPPPS |
404                                NL80211_FEATURE_DYNAMIC_SMPS |
405                                NL80211_FEATURE_STATIC_SMPS;
406
407         if (mvm->fw->ucode_capa.capa[0] &
408             IWL_UCODE_TLV_CAPA_TXPOWER_INSERTION_SUPPORT)
409                 hw->wiphy->features |= NL80211_FEATURE_TX_POWER_INSERTION;
410         if (mvm->fw->ucode_capa.capa[0] &
411             IWL_UCODE_TLV_CAPA_QUIET_PERIOD_SUPPORT)
412                 hw->wiphy->features |= NL80211_FEATURE_QUIET;
413
414         if (mvm->fw->ucode_capa.capa[0] &
415             IWL_UCODE_TLV_CAPA_DS_PARAM_SET_IE_SUPPORT)
416                 hw->wiphy->features |=
417                         NL80211_FEATURE_DS_PARAM_SET_IE_IN_PROBES;
418
419         if (mvm->fw->ucode_capa.capa[0] &
420             IWL_UCODE_TLV_CAPA_WFA_TPC_REP_IE_SUPPORT)
421                 hw->wiphy->features |= NL80211_FEATURE_WFA_TPC_IE_IN_PROBES;
422
423         mvm->rts_threshold = IEEE80211_MAX_RTS_THRESHOLD;
424
425         /* currently FW API supports only one optional cipher scheme */
426         if (mvm->fw->cs[0].cipher) {
427                 mvm->hw->n_cipher_schemes = 1;
428                 mvm->hw->cipher_schemes = &mvm->fw->cs[0];
429         }
430
431 #ifdef CONFIG_PM_SLEEP
432         if (iwl_mvm_is_d0i3_supported(mvm) &&
433             device_can_wakeup(mvm->trans->dev)) {
434                 mvm->wowlan.flags = WIPHY_WOWLAN_ANY;
435                 hw->wiphy->wowlan = &mvm->wowlan;
436         } else if (mvm->fw->img[IWL_UCODE_WOWLAN].sec[0].len &&
437             mvm->trans->ops->d3_suspend &&
438             mvm->trans->ops->d3_resume &&
439             device_can_wakeup(mvm->trans->dev)) {
440                 mvm->wowlan.flags = WIPHY_WOWLAN_MAGIC_PKT |
441                                     WIPHY_WOWLAN_DISCONNECT |
442                                     WIPHY_WOWLAN_EAP_IDENTITY_REQ |
443                                     WIPHY_WOWLAN_RFKILL_RELEASE;
444                 if (!iwlwifi_mod_params.sw_crypto)
445                         mvm->wowlan.flags |= WIPHY_WOWLAN_SUPPORTS_GTK_REKEY |
446                                              WIPHY_WOWLAN_GTK_REKEY_FAILURE |
447                                              WIPHY_WOWLAN_4WAY_HANDSHAKE;
448
449                 mvm->wowlan.n_patterns = IWL_WOWLAN_MAX_PATTERNS;
450                 mvm->wowlan.pattern_min_len = IWL_WOWLAN_MIN_PATTERN_LEN;
451                 mvm->wowlan.pattern_max_len = IWL_WOWLAN_MAX_PATTERN_LEN;
452                 mvm->wowlan.tcp = &iwl_mvm_wowlan_tcp_support;
453                 hw->wiphy->wowlan = &mvm->wowlan;
454         }
455 #endif
456
457 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
458         /* assign default bcast filtering configuration */
459         mvm->bcast_filters = iwl_mvm_default_bcast_filters;
460 #endif
461
462         ret = iwl_mvm_leds_init(mvm);
463         if (ret)
464                 return ret;
465
466         ret = ieee80211_register_hw(mvm->hw);
467         if (ret)
468                 iwl_mvm_leds_exit(mvm);
469
470         return ret;
471 }
472
473 static bool iwl_mvm_defer_tx(struct iwl_mvm *mvm,
474                              struct ieee80211_sta *sta,
475                              struct sk_buff *skb)
476 {
477         struct iwl_mvm_sta *mvmsta;
478         bool defer = false;
479
480         /*
481          * double check the IN_D0I3 flag both before and after
482          * taking the spinlock, in order to prevent taking
483          * the spinlock when not needed.
484          */
485         if (likely(!test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status)))
486                 return false;
487
488         spin_lock(&mvm->d0i3_tx_lock);
489         /*
490          * testing the flag again ensures the skb dequeue
491          * loop (on d0i3 exit) hasn't run yet.
492          */
493         if (!test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status))
494                 goto out;
495
496         mvmsta = iwl_mvm_sta_from_mac80211(sta);
497         if (mvmsta->sta_id == IWL_MVM_STATION_COUNT ||
498             mvmsta->sta_id != mvm->d0i3_ap_sta_id)
499                 goto out;
500
501         __skb_queue_tail(&mvm->d0i3_tx, skb);
502         ieee80211_stop_queues(mvm->hw);
503
504         /* trigger wakeup */
505         iwl_mvm_ref(mvm, IWL_MVM_REF_TX);
506         iwl_mvm_unref(mvm, IWL_MVM_REF_TX);
507
508         defer = true;
509 out:
510         spin_unlock(&mvm->d0i3_tx_lock);
511         return defer;
512 }
513
514 static void iwl_mvm_mac_tx(struct ieee80211_hw *hw,
515                            struct ieee80211_tx_control *control,
516                            struct sk_buff *skb)
517 {
518         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
519         struct ieee80211_sta *sta = control->sta;
520         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
521         struct ieee80211_hdr *hdr = (void *)skb->data;
522
523         if (iwl_mvm_is_radio_killed(mvm)) {
524                 IWL_DEBUG_DROP(mvm, "Dropping - RF/CT KILL\n");
525                 goto drop;
526         }
527
528         if (IEEE80211_SKB_CB(skb)->hw_queue == IWL_MVM_OFFCHANNEL_QUEUE &&
529             !test_bit(IWL_MVM_STATUS_ROC_RUNNING, &mvm->status))
530                 goto drop;
531
532         /* treat non-bufferable MMPDUs as broadcast if sta is sleeping */
533         if (unlikely(info->flags & IEEE80211_TX_CTL_NO_PS_BUFFER &&
534                      ieee80211_is_mgmt(hdr->frame_control) &&
535                      !ieee80211_is_deauth(hdr->frame_control) &&
536                      !ieee80211_is_disassoc(hdr->frame_control) &&
537                      !ieee80211_is_action(hdr->frame_control)))
538                 sta = NULL;
539
540         if (sta) {
541                 if (iwl_mvm_defer_tx(mvm, sta, skb))
542                         return;
543                 if (iwl_mvm_tx_skb(mvm, skb, sta))
544                         goto drop;
545                 return;
546         }
547
548         if (iwl_mvm_tx_skb_non_sta(mvm, skb))
549                 goto drop;
550         return;
551  drop:
552         ieee80211_free_txskb(hw, skb);
553 }
554
555 static inline bool iwl_enable_rx_ampdu(const struct iwl_cfg *cfg)
556 {
557         if (iwlwifi_mod_params.disable_11n & IWL_DISABLE_HT_RXAGG)
558                 return false;
559         return true;
560 }
561
562 static inline bool iwl_enable_tx_ampdu(const struct iwl_cfg *cfg)
563 {
564         if (iwlwifi_mod_params.disable_11n & IWL_DISABLE_HT_TXAGG)
565                 return false;
566         if (iwlwifi_mod_params.disable_11n & IWL_ENABLE_HT_TXAGG)
567                 return true;
568
569         /* enabled by default */
570         return true;
571 }
572
573 static int iwl_mvm_mac_ampdu_action(struct ieee80211_hw *hw,
574                                     struct ieee80211_vif *vif,
575                                     enum ieee80211_ampdu_mlme_action action,
576                                     struct ieee80211_sta *sta, u16 tid,
577                                     u16 *ssn, u8 buf_size)
578 {
579         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
580         int ret;
581         bool tx_agg_ref = false;
582
583         IWL_DEBUG_HT(mvm, "A-MPDU action on addr %pM tid %d: action %d\n",
584                      sta->addr, tid, action);
585
586         if (!(mvm->nvm_data->sku_cap_11n_enable))
587                 return -EACCES;
588
589         /* return from D0i3 before starting a new Tx aggregation */
590         switch (action) {
591         case IEEE80211_AMPDU_TX_START:
592         case IEEE80211_AMPDU_TX_STOP_CONT:
593         case IEEE80211_AMPDU_TX_STOP_FLUSH:
594         case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
595         case IEEE80211_AMPDU_TX_OPERATIONAL:
596                 /*
597                  * for tx start, wait synchronously until D0i3 exit to
598                  * get the correct sequence number for the tid.
599                  * additionally, some other ampdu actions use direct
600                  * target access, which is not handled automatically
601                  * by the trans layer (unlike commands), so wait for
602                  * d0i3 exit in these cases as well.
603                  */
604                 ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_TX_AGG);
605                 if (ret)
606                         return ret;
607
608                 tx_agg_ref = true;
609                 break;
610         default:
611                 break;
612         }
613
614         mutex_lock(&mvm->mutex);
615
616         switch (action) {
617         case IEEE80211_AMPDU_RX_START:
618                 if (!iwl_enable_rx_ampdu(mvm->cfg)) {
619                         ret = -EINVAL;
620                         break;
621                 }
622                 ret = iwl_mvm_sta_rx_agg(mvm, sta, tid, *ssn, true);
623                 break;
624         case IEEE80211_AMPDU_RX_STOP:
625                 ret = iwl_mvm_sta_rx_agg(mvm, sta, tid, 0, false);
626                 break;
627         case IEEE80211_AMPDU_TX_START:
628                 if (!iwl_enable_tx_ampdu(mvm->cfg)) {
629                         ret = -EINVAL;
630                         break;
631                 }
632                 ret = iwl_mvm_sta_tx_agg_start(mvm, vif, sta, tid, ssn);
633                 break;
634         case IEEE80211_AMPDU_TX_STOP_CONT:
635                 ret = iwl_mvm_sta_tx_agg_stop(mvm, vif, sta, tid);
636                 break;
637         case IEEE80211_AMPDU_TX_STOP_FLUSH:
638         case IEEE80211_AMPDU_TX_STOP_FLUSH_CONT:
639                 ret = iwl_mvm_sta_tx_agg_flush(mvm, vif, sta, tid);
640                 break;
641         case IEEE80211_AMPDU_TX_OPERATIONAL:
642                 ret = iwl_mvm_sta_tx_agg_oper(mvm, vif, sta, tid, buf_size);
643                 break;
644         default:
645                 WARN_ON_ONCE(1);
646                 ret = -EINVAL;
647                 break;
648         }
649         mutex_unlock(&mvm->mutex);
650
651         /*
652          * If the tid is marked as started, we won't use it for offloaded
653          * traffic on the next D0i3 entry. It's safe to unref.
654          */
655         if (tx_agg_ref)
656                 iwl_mvm_unref(mvm, IWL_MVM_REF_TX_AGG);
657
658         return ret;
659 }
660
661 static void iwl_mvm_cleanup_iterator(void *data, u8 *mac,
662                                      struct ieee80211_vif *vif)
663 {
664         struct iwl_mvm *mvm = data;
665         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
666
667         mvmvif->uploaded = false;
668         mvmvif->ap_sta_id = IWL_MVM_STATION_COUNT;
669
670         /* does this make sense at all? */
671         mvmvif->color++;
672
673         spin_lock_bh(&mvm->time_event_lock);
674         iwl_mvm_te_clear_data(mvm, &mvmvif->time_event_data);
675         spin_unlock_bh(&mvm->time_event_lock);
676
677         mvmvif->phy_ctxt = NULL;
678         memset(&mvmvif->bf_data, 0, sizeof(mvmvif->bf_data));
679 }
680
681 #ifdef CONFIG_IWLWIFI_DEBUGFS
682 void iwl_mvm_fw_error_dump(struct iwl_mvm *mvm)
683 {
684         static char *env[] = { "DRIVER=iwlwifi", "EVENT=error_dump", NULL };
685         struct iwl_fw_error_dump_file *dump_file;
686         struct iwl_fw_error_dump_data *dump_data;
687         struct iwl_fw_error_dump_info *dump_info;
688         struct iwl_mvm_dump_ptrs *fw_error_dump;
689         const struct fw_img *img;
690         u32 sram_len, sram_ofs;
691         u32 file_len, rxf_len;
692         unsigned long flags;
693         int reg_val;
694
695         lockdep_assert_held(&mvm->mutex);
696
697         if (mvm->fw_error_dump)
698                 return;
699
700         fw_error_dump = kzalloc(sizeof(*mvm->fw_error_dump), GFP_KERNEL);
701         if (!fw_error_dump)
702                 return;
703
704         img = &mvm->fw->img[mvm->cur_ucode];
705         sram_ofs = img->sec[IWL_UCODE_SECTION_DATA].offset;
706         sram_len = img->sec[IWL_UCODE_SECTION_DATA].len;
707
708         /* reading buffer size */
709         reg_val = iwl_trans_read_prph(mvm->trans, RXF_SIZE_ADDR);
710         rxf_len = (reg_val & RXF_SIZE_BYTE_CNT_MSK) >> RXF_SIZE_BYTE_CND_POS;
711
712         /* the register holds the value divided by 128 */
713         rxf_len = rxf_len << 7;
714
715         file_len = sizeof(*dump_file) +
716                    sizeof(*dump_data) * 3 +
717                    sram_len +
718                    rxf_len +
719                    sizeof(*dump_info);
720
721         dump_file = vzalloc(file_len);
722         if (!dump_file) {
723                 kfree(fw_error_dump);
724                 return;
725         }
726
727         fw_error_dump->op_mode_ptr = dump_file;
728
729         dump_file->barker = cpu_to_le32(IWL_FW_ERROR_DUMP_BARKER);
730         dump_data = (void *)dump_file->data;
731
732         dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_DEV_FW_INFO);
733         dump_data->len = cpu_to_le32(sizeof(*dump_info));
734         dump_info = (void *) dump_data->data;
735         dump_info->device_family =
736                 mvm->cfg->device_family == IWL_DEVICE_FAMILY_7000 ?
737                         cpu_to_le32(IWL_FW_ERROR_DUMP_FAMILY_7) :
738                         cpu_to_le32(IWL_FW_ERROR_DUMP_FAMILY_8);
739         memcpy(dump_info->fw_human_readable, mvm->fw->human_readable,
740                sizeof(dump_info->fw_human_readable));
741         strncpy(dump_info->dev_human_readable, mvm->cfg->name,
742                 sizeof(dump_info->dev_human_readable));
743         strncpy(dump_info->bus_human_readable, mvm->dev->bus->name,
744                 sizeof(dump_info->bus_human_readable));
745
746         dump_data = iwl_fw_error_next_data(dump_data);
747         dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_RXF);
748         dump_data->len = cpu_to_le32(rxf_len);
749
750         if (iwl_trans_grab_nic_access(mvm->trans, false, &flags)) {
751                 u32 *rxf = (void *)dump_data->data;
752                 int i;
753
754                 for (i = 0; i < (rxf_len / sizeof(u32)); i++) {
755                         iwl_trans_write_prph(mvm->trans,
756                                              RXF_LD_FENCE_OFFSET_ADDR,
757                                              i * sizeof(u32));
758                         rxf[i] = iwl_trans_read_prph(mvm->trans,
759                                                      RXF_FIFO_RD_FENCE_ADDR);
760                 }
761                 iwl_trans_release_nic_access(mvm->trans, &flags);
762         }
763
764         dump_data = iwl_fw_error_next_data(dump_data);
765         dump_data->type = cpu_to_le32(IWL_FW_ERROR_DUMP_SRAM);
766         dump_data->len = cpu_to_le32(sram_len);
767         iwl_trans_read_mem_bytes(mvm->trans, sram_ofs, dump_data->data,
768                                  sram_len);
769
770         fw_error_dump->trans_ptr = iwl_trans_dump_data(mvm->trans);
771         fw_error_dump->op_mode_len = file_len;
772         if (fw_error_dump->trans_ptr)
773                 file_len += fw_error_dump->trans_ptr->len;
774         dump_file->file_len = cpu_to_le32(file_len);
775         mvm->fw_error_dump = fw_error_dump;
776
777         /* notify the userspace about the error we had */
778         kobject_uevent_env(&mvm->hw->wiphy->dev.kobj, KOBJ_CHANGE, env);
779 }
780 #endif
781
782 static void iwl_mvm_restart_cleanup(struct iwl_mvm *mvm)
783 {
784         iwl_mvm_fw_error_dump(mvm);
785
786         iwl_trans_stop_device(mvm->trans);
787
788         mvm->scan_status = IWL_MVM_SCAN_NONE;
789         mvm->ps_disabled = false;
790
791         /* just in case one was running */
792         ieee80211_remain_on_channel_expired(mvm->hw);
793
794         ieee80211_iterate_active_interfaces_atomic(
795                 mvm->hw, IEEE80211_IFACE_ITER_RESUME_ALL,
796                 iwl_mvm_cleanup_iterator, mvm);
797
798         mvm->p2p_device_vif = NULL;
799         mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
800
801         iwl_mvm_reset_phy_ctxts(mvm);
802         memset(mvm->fw_key_table, 0, sizeof(mvm->fw_key_table));
803         memset(mvm->sta_drained, 0, sizeof(mvm->sta_drained));
804         memset(&mvm->last_bt_notif, 0, sizeof(mvm->last_bt_notif));
805         memset(&mvm->last_bt_notif_old, 0, sizeof(mvm->last_bt_notif_old));
806         memset(&mvm->last_bt_ci_cmd, 0, sizeof(mvm->last_bt_ci_cmd));
807         memset(&mvm->last_bt_ci_cmd_old, 0, sizeof(mvm->last_bt_ci_cmd_old));
808         memset(&mvm->bt_ack_kill_msk, 0, sizeof(mvm->bt_ack_kill_msk));
809         memset(&mvm->bt_cts_kill_msk, 0, sizeof(mvm->bt_cts_kill_msk));
810
811         ieee80211_wake_queues(mvm->hw);
812
813         /* cleanup all stale references (scan, roc), but keep the
814          * ucode_down ref until reconfig is complete */
815         iwl_mvm_unref_all_except(mvm, IWL_MVM_REF_UCODE_DOWN);
816
817         /* clear any stale d0i3 state */
818         clear_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status);
819
820         mvm->vif_count = 0;
821         mvm->rx_ba_sessions = 0;
822 }
823
824 int __iwl_mvm_mac_start(struct iwl_mvm *mvm)
825 {
826         int ret;
827
828         lockdep_assert_held(&mvm->mutex);
829
830         /* Clean up some internal and mac80211 state on restart */
831         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
832                 iwl_mvm_restart_cleanup(mvm);
833
834         ret = iwl_mvm_up(mvm);
835
836         if (ret && test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
837                 /* Something went wrong - we need to finish some cleanup
838                  * that normally iwl_mvm_mac_restart_complete() below
839                  * would do.
840                  */
841                 clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status);
842                 iwl_mvm_d0i3_enable_tx(mvm, NULL);
843         }
844
845         return ret;
846 }
847
848 static int iwl_mvm_mac_start(struct ieee80211_hw *hw)
849 {
850         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
851         int ret;
852
853         mutex_lock(&mvm->mutex);
854         ret = __iwl_mvm_mac_start(mvm);
855         mutex_unlock(&mvm->mutex);
856
857         return ret;
858 }
859
860 static void iwl_mvm_mac_restart_complete(struct ieee80211_hw *hw)
861 {
862         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
863         int ret;
864
865         mutex_lock(&mvm->mutex);
866
867         clear_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status);
868         iwl_mvm_d0i3_enable_tx(mvm, NULL);
869         ret = iwl_mvm_update_quotas(mvm, NULL);
870         if (ret)
871                 IWL_ERR(mvm, "Failed to update quotas after restart (%d)\n",
872                         ret);
873
874         /* allow transport/FW low power modes */
875         iwl_mvm_unref(mvm, IWL_MVM_REF_UCODE_DOWN);
876
877         mutex_unlock(&mvm->mutex);
878 }
879
880 void __iwl_mvm_mac_stop(struct iwl_mvm *mvm)
881 {
882         lockdep_assert_held(&mvm->mutex);
883
884         /* disallow low power states when the FW is down */
885         iwl_mvm_ref(mvm, IWL_MVM_REF_UCODE_DOWN);
886
887         /* async_handlers_wk is now blocked */
888
889         /*
890          * The work item could be running or queued if the
891          * ROC time event stops just as we get here.
892          */
893         cancel_work_sync(&mvm->roc_done_wk);
894
895         iwl_trans_stop_device(mvm->trans);
896
897         iwl_mvm_async_handlers_purge(mvm);
898         /* async_handlers_list is empty and will stay empty: HW is stopped */
899
900         /* the fw is stopped, the aux sta is dead: clean up driver state */
901         iwl_mvm_del_aux_sta(mvm);
902
903         mvm->ucode_loaded = false;
904 }
905
906 static void iwl_mvm_mac_stop(struct ieee80211_hw *hw)
907 {
908         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
909
910         flush_work(&mvm->d0i3_exit_work);
911         flush_work(&mvm->async_handlers_wk);
912         flush_work(&mvm->fw_error_dump_wk);
913
914         mutex_lock(&mvm->mutex);
915         __iwl_mvm_mac_stop(mvm);
916         mutex_unlock(&mvm->mutex);
917
918         /*
919          * The worker might have been waiting for the mutex, let it run and
920          * discover that its list is now empty.
921          */
922         cancel_work_sync(&mvm->async_handlers_wk);
923 }
924
925 static struct iwl_mvm_phy_ctxt *iwl_mvm_get_free_phy_ctxt(struct iwl_mvm *mvm)
926 {
927         u16 i;
928
929         lockdep_assert_held(&mvm->mutex);
930
931         for (i = 0; i < NUM_PHY_CTX; i++)
932                 if (!mvm->phy_ctxts[i].ref)
933                         return &mvm->phy_ctxts[i];
934
935         IWL_ERR(mvm, "No available PHY context\n");
936         return NULL;
937 }
938
939 static int iwl_mvm_set_tx_power(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
940                                 s8 tx_power)
941 {
942         /* FW is in charge of regulatory enforcement */
943         struct iwl_reduce_tx_power_cmd reduce_txpwr_cmd = {
944                 .mac_context_id = iwl_mvm_vif_from_mac80211(vif)->id,
945                 .pwr_restriction = cpu_to_le16(tx_power),
946         };
947
948         return iwl_mvm_send_cmd_pdu(mvm, REDUCE_TX_POWER_CMD, 0,
949                                     sizeof(reduce_txpwr_cmd),
950                                     &reduce_txpwr_cmd);
951 }
952
953 static int iwl_mvm_mac_add_interface(struct ieee80211_hw *hw,
954                                      struct ieee80211_vif *vif)
955 {
956         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
957         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
958         int ret;
959
960         /*
961          * make sure D0i3 exit is completed, otherwise a target access
962          * during tx queue configuration could be done when still in
963          * D0i3 state.
964          */
965         ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_ADD_IF);
966         if (ret)
967                 return ret;
968
969         /*
970          * Not much to do here. The stack will not allow interface
971          * types or combinations that we didn't advertise, so we
972          * don't really have to check the types.
973          */
974
975         mutex_lock(&mvm->mutex);
976
977         /* Allocate resources for the MAC context, and add it to the fw  */
978         ret = iwl_mvm_mac_ctxt_init(mvm, vif);
979         if (ret)
980                 goto out_unlock;
981
982         /* Counting number of interfaces is needed for legacy PM */
983         if (vif->type != NL80211_IFTYPE_P2P_DEVICE)
984                 mvm->vif_count++;
985
986         /*
987          * The AP binding flow can be done only after the beacon
988          * template is configured (which happens only in the mac80211
989          * start_ap() flow), and adding the broadcast station can happen
990          * only after the binding.
991          * In addition, since modifying the MAC before adding a bcast
992          * station is not allowed by the FW, delay the adding of MAC context to
993          * the point where we can also add the bcast station.
994          * In short: there's not much we can do at this point, other than
995          * allocating resources :)
996          */
997         if (vif->type == NL80211_IFTYPE_AP ||
998             vif->type == NL80211_IFTYPE_ADHOC) {
999                 ret = iwl_mvm_alloc_bcast_sta(mvm, vif);
1000                 if (ret) {
1001                         IWL_ERR(mvm, "Failed to allocate bcast sta\n");
1002                         goto out_release;
1003                 }
1004
1005                 iwl_mvm_vif_dbgfs_register(mvm, vif);
1006                 goto out_unlock;
1007         }
1008
1009         ret = iwl_mvm_mac_ctxt_add(mvm, vif);
1010         if (ret)
1011                 goto out_release;
1012
1013         ret = iwl_mvm_power_update_mac(mvm);
1014         if (ret)
1015                 goto out_release;
1016
1017         /* beacon filtering */
1018         ret = iwl_mvm_disable_beacon_filter(mvm, vif, 0);
1019         if (ret)
1020                 goto out_remove_mac;
1021
1022         if (!mvm->bf_allowed_vif &&
1023             vif->type == NL80211_IFTYPE_STATION && !vif->p2p) {
1024                 mvm->bf_allowed_vif = mvmvif;
1025                 vif->driver_flags |= IEEE80211_VIF_BEACON_FILTER |
1026                                      IEEE80211_VIF_SUPPORTS_CQM_RSSI;
1027         }
1028
1029         /*
1030          * P2P_DEVICE interface does not have a channel context assigned to it,
1031          * so a dedicated PHY context is allocated to it and the corresponding
1032          * MAC context is bound to it at this stage.
1033          */
1034         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1035
1036                 mvmvif->phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
1037                 if (!mvmvif->phy_ctxt) {
1038                         ret = -ENOSPC;
1039                         goto out_free_bf;
1040                 }
1041
1042                 iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
1043                 ret = iwl_mvm_binding_add_vif(mvm, vif);
1044                 if (ret)
1045                         goto out_unref_phy;
1046
1047                 ret = iwl_mvm_add_bcast_sta(mvm, vif);
1048                 if (ret)
1049                         goto out_unbind;
1050
1051                 /* Save a pointer to p2p device vif, so it can later be used to
1052                  * update the p2p device MAC when a GO is started/stopped */
1053                 mvm->p2p_device_vif = vif;
1054         }
1055
1056         iwl_mvm_vif_dbgfs_register(mvm, vif);
1057         goto out_unlock;
1058
1059  out_unbind:
1060         iwl_mvm_binding_remove_vif(mvm, vif);
1061  out_unref_phy:
1062         iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
1063  out_free_bf:
1064         if (mvm->bf_allowed_vif == mvmvif) {
1065                 mvm->bf_allowed_vif = NULL;
1066                 vif->driver_flags &= ~(IEEE80211_VIF_BEACON_FILTER |
1067                                        IEEE80211_VIF_SUPPORTS_CQM_RSSI);
1068         }
1069  out_remove_mac:
1070         mvmvif->phy_ctxt = NULL;
1071         iwl_mvm_mac_ctxt_remove(mvm, vif);
1072  out_release:
1073         if (vif->type != NL80211_IFTYPE_P2P_DEVICE)
1074                 mvm->vif_count--;
1075
1076         iwl_mvm_mac_ctxt_release(mvm, vif);
1077  out_unlock:
1078         mutex_unlock(&mvm->mutex);
1079
1080         iwl_mvm_unref(mvm, IWL_MVM_REF_ADD_IF);
1081
1082         return ret;
1083 }
1084
1085 static void iwl_mvm_prepare_mac_removal(struct iwl_mvm *mvm,
1086                                         struct ieee80211_vif *vif)
1087 {
1088         u32 tfd_msk = iwl_mvm_mac_get_queues_mask(mvm, vif);
1089
1090         if (tfd_msk) {
1091                 mutex_lock(&mvm->mutex);
1092                 iwl_mvm_flush_tx_path(mvm, tfd_msk, true);
1093                 mutex_unlock(&mvm->mutex);
1094         }
1095
1096         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1097                 /*
1098                  * Flush the ROC worker which will flush the OFFCHANNEL queue.
1099                  * We assume here that all the packets sent to the OFFCHANNEL
1100                  * queue are sent in ROC session.
1101                  */
1102                 flush_work(&mvm->roc_done_wk);
1103         } else {
1104                 /*
1105                  * By now, all the AC queues are empty. The AGG queues are
1106                  * empty too. We already got all the Tx responses for all the
1107                  * packets in the queues. The drain work can have been
1108                  * triggered. Flush it.
1109                  */
1110                 flush_work(&mvm->sta_drained_wk);
1111         }
1112 }
1113
1114 static void iwl_mvm_mac_remove_interface(struct ieee80211_hw *hw,
1115                                          struct ieee80211_vif *vif)
1116 {
1117         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1118         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1119
1120         iwl_mvm_prepare_mac_removal(mvm, vif);
1121
1122         mutex_lock(&mvm->mutex);
1123
1124         if (mvm->bf_allowed_vif == mvmvif) {
1125                 mvm->bf_allowed_vif = NULL;
1126                 vif->driver_flags &= ~(IEEE80211_VIF_BEACON_FILTER |
1127                                        IEEE80211_VIF_SUPPORTS_CQM_RSSI);
1128         }
1129
1130         iwl_mvm_vif_dbgfs_clean(mvm, vif);
1131
1132         /*
1133          * For AP/GO interface, the tear down of the resources allocated to the
1134          * interface is be handled as part of the stop_ap flow.
1135          */
1136         if (vif->type == NL80211_IFTYPE_AP ||
1137             vif->type == NL80211_IFTYPE_ADHOC) {
1138 #ifdef CONFIG_NL80211_TESTMODE
1139                 if (vif == mvm->noa_vif) {
1140                         mvm->noa_vif = NULL;
1141                         mvm->noa_duration = 0;
1142                 }
1143 #endif
1144                 iwl_mvm_dealloc_bcast_sta(mvm, vif);
1145                 goto out_release;
1146         }
1147
1148         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1149                 mvm->p2p_device_vif = NULL;
1150                 iwl_mvm_rm_bcast_sta(mvm, vif);
1151                 iwl_mvm_binding_remove_vif(mvm, vif);
1152                 iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
1153                 mvmvif->phy_ctxt = NULL;
1154         }
1155
1156         if (mvm->vif_count && vif->type != NL80211_IFTYPE_P2P_DEVICE)
1157                 mvm->vif_count--;
1158
1159         iwl_mvm_power_update_mac(mvm);
1160         iwl_mvm_mac_ctxt_remove(mvm, vif);
1161
1162 out_release:
1163         iwl_mvm_mac_ctxt_release(mvm, vif);
1164         mutex_unlock(&mvm->mutex);
1165 }
1166
1167 static int iwl_mvm_mac_config(struct ieee80211_hw *hw, u32 changed)
1168 {
1169         return 0;
1170 }
1171
1172 struct iwl_mvm_mc_iter_data {
1173         struct iwl_mvm *mvm;
1174         int port_id;
1175 };
1176
1177 static void iwl_mvm_mc_iface_iterator(void *_data, u8 *mac,
1178                                       struct ieee80211_vif *vif)
1179 {
1180         struct iwl_mvm_mc_iter_data *data = _data;
1181         struct iwl_mvm *mvm = data->mvm;
1182         struct iwl_mcast_filter_cmd *cmd = mvm->mcast_filter_cmd;
1183         int ret, len;
1184
1185         /* if we don't have free ports, mcast frames will be dropped */
1186         if (WARN_ON_ONCE(data->port_id >= MAX_PORT_ID_NUM))
1187                 return;
1188
1189         if (vif->type != NL80211_IFTYPE_STATION ||
1190             !vif->bss_conf.assoc)
1191                 return;
1192
1193         cmd->port_id = data->port_id++;
1194         memcpy(cmd->bssid, vif->bss_conf.bssid, ETH_ALEN);
1195         len = roundup(sizeof(*cmd) + cmd->count * ETH_ALEN, 4);
1196
1197         ret = iwl_mvm_send_cmd_pdu(mvm, MCAST_FILTER_CMD, CMD_ASYNC, len, cmd);
1198         if (ret)
1199                 IWL_ERR(mvm, "mcast filter cmd error. ret=%d\n", ret);
1200 }
1201
1202 static void iwl_mvm_recalc_multicast(struct iwl_mvm *mvm)
1203 {
1204         struct iwl_mvm_mc_iter_data iter_data = {
1205                 .mvm = mvm,
1206         };
1207
1208         lockdep_assert_held(&mvm->mutex);
1209
1210         if (WARN_ON_ONCE(!mvm->mcast_filter_cmd))
1211                 return;
1212
1213         ieee80211_iterate_active_interfaces_atomic(
1214                 mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
1215                 iwl_mvm_mc_iface_iterator, &iter_data);
1216 }
1217
1218 static u64 iwl_mvm_prepare_multicast(struct ieee80211_hw *hw,
1219                                      struct netdev_hw_addr_list *mc_list)
1220 {
1221         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1222         struct iwl_mcast_filter_cmd *cmd;
1223         struct netdev_hw_addr *addr;
1224         int addr_count;
1225         bool pass_all;
1226         int len;
1227
1228         addr_count = netdev_hw_addr_list_count(mc_list);
1229         pass_all = addr_count > MAX_MCAST_FILTERING_ADDRESSES ||
1230                    IWL_MVM_FW_MCAST_FILTER_PASS_ALL;
1231         if (pass_all)
1232                 addr_count = 0;
1233
1234         len = roundup(sizeof(*cmd) + addr_count * ETH_ALEN, 4);
1235         cmd = kzalloc(len, GFP_ATOMIC);
1236         if (!cmd)
1237                 return 0;
1238
1239         if (pass_all) {
1240                 cmd->pass_all = 1;
1241                 return (u64)(unsigned long)cmd;
1242         }
1243
1244         netdev_hw_addr_list_for_each(addr, mc_list) {
1245                 IWL_DEBUG_MAC80211(mvm, "mcast addr (%d): %pM\n",
1246                                    cmd->count, addr->addr);
1247                 memcpy(&cmd->addr_list[cmd->count * ETH_ALEN],
1248                        addr->addr, ETH_ALEN);
1249                 cmd->count++;
1250         }
1251
1252         return (u64)(unsigned long)cmd;
1253 }
1254
1255 static void iwl_mvm_configure_filter(struct ieee80211_hw *hw,
1256                                      unsigned int changed_flags,
1257                                      unsigned int *total_flags,
1258                                      u64 multicast)
1259 {
1260         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1261         struct iwl_mcast_filter_cmd *cmd = (void *)(unsigned long)multicast;
1262
1263         mutex_lock(&mvm->mutex);
1264
1265         /* replace previous configuration */
1266         kfree(mvm->mcast_filter_cmd);
1267         mvm->mcast_filter_cmd = cmd;
1268
1269         if (!cmd)
1270                 goto out;
1271
1272         iwl_mvm_recalc_multicast(mvm);
1273 out:
1274         mutex_unlock(&mvm->mutex);
1275         *total_flags = 0;
1276 }
1277
1278 #ifdef CONFIG_IWLWIFI_BCAST_FILTERING
1279 struct iwl_bcast_iter_data {
1280         struct iwl_mvm *mvm;
1281         struct iwl_bcast_filter_cmd *cmd;
1282         u8 current_filter;
1283 };
1284
1285 static void
1286 iwl_mvm_set_bcast_filter(struct ieee80211_vif *vif,
1287                          const struct iwl_fw_bcast_filter *in_filter,
1288                          struct iwl_fw_bcast_filter *out_filter)
1289 {
1290         struct iwl_fw_bcast_filter_attr *attr;
1291         int i;
1292
1293         memcpy(out_filter, in_filter, sizeof(*out_filter));
1294
1295         for (i = 0; i < ARRAY_SIZE(out_filter->attrs); i++) {
1296                 attr = &out_filter->attrs[i];
1297
1298                 if (!attr->mask)
1299                         break;
1300
1301                 switch (attr->reserved1) {
1302                 case cpu_to_le16(BC_FILTER_MAGIC_IP):
1303                         if (vif->bss_conf.arp_addr_cnt != 1) {
1304                                 attr->mask = 0;
1305                                 continue;
1306                         }
1307
1308                         attr->val = vif->bss_conf.arp_addr_list[0];
1309                         break;
1310                 case cpu_to_le16(BC_FILTER_MAGIC_MAC):
1311                         attr->val = *(__be32 *)&vif->addr[2];
1312                         break;
1313                 default:
1314                         break;
1315                 }
1316                 attr->reserved1 = 0;
1317                 out_filter->num_attrs++;
1318         }
1319 }
1320
1321 static void iwl_mvm_bcast_filter_iterator(void *_data, u8 *mac,
1322                                           struct ieee80211_vif *vif)
1323 {
1324         struct iwl_bcast_iter_data *data = _data;
1325         struct iwl_mvm *mvm = data->mvm;
1326         struct iwl_bcast_filter_cmd *cmd = data->cmd;
1327         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1328         struct iwl_fw_bcast_mac *bcast_mac;
1329         int i;
1330
1331         if (WARN_ON(mvmvif->id >= ARRAY_SIZE(cmd->macs)))
1332                 return;
1333
1334         bcast_mac = &cmd->macs[mvmvif->id];
1335
1336         /*
1337          * enable filtering only for associated stations, but not for P2P
1338          * Clients
1339          */
1340         if (vif->type != NL80211_IFTYPE_STATION || vif->p2p ||
1341             !vif->bss_conf.assoc)
1342                 return;
1343
1344         bcast_mac->default_discard = 1;
1345
1346         /* copy all configured filters */
1347         for (i = 0; mvm->bcast_filters[i].attrs[0].mask; i++) {
1348                 /*
1349                  * Make sure we don't exceed our filters limit.
1350                  * if there is still a valid filter to be configured,
1351                  * be on the safe side and just allow bcast for this mac.
1352                  */
1353                 if (WARN_ON_ONCE(data->current_filter >=
1354                                  ARRAY_SIZE(cmd->filters))) {
1355                         bcast_mac->default_discard = 0;
1356                         bcast_mac->attached_filters = 0;
1357                         break;
1358                 }
1359
1360                 iwl_mvm_set_bcast_filter(vif,
1361                                          &mvm->bcast_filters[i],
1362                                          &cmd->filters[data->current_filter]);
1363
1364                 /* skip current filter if it contains no attributes */
1365                 if (!cmd->filters[data->current_filter].num_attrs)
1366                         continue;
1367
1368                 /* attach the filter to current mac */
1369                 bcast_mac->attached_filters |=
1370                                 cpu_to_le16(BIT(data->current_filter));
1371
1372                 data->current_filter++;
1373         }
1374 }
1375
1376 bool iwl_mvm_bcast_filter_build_cmd(struct iwl_mvm *mvm,
1377                                     struct iwl_bcast_filter_cmd *cmd)
1378 {
1379         struct iwl_bcast_iter_data iter_data = {
1380                 .mvm = mvm,
1381                 .cmd = cmd,
1382         };
1383
1384         memset(cmd, 0, sizeof(*cmd));
1385         cmd->max_bcast_filters = ARRAY_SIZE(cmd->filters);
1386         cmd->max_macs = ARRAY_SIZE(cmd->macs);
1387
1388 #ifdef CONFIG_IWLWIFI_DEBUGFS
1389         /* use debugfs filters/macs if override is configured */
1390         if (mvm->dbgfs_bcast_filtering.override) {
1391                 memcpy(cmd->filters, &mvm->dbgfs_bcast_filtering.cmd.filters,
1392                        sizeof(cmd->filters));
1393                 memcpy(cmd->macs, &mvm->dbgfs_bcast_filtering.cmd.macs,
1394                        sizeof(cmd->macs));
1395                 return true;
1396         }
1397 #endif
1398
1399         /* if no filters are configured, do nothing */
1400         if (!mvm->bcast_filters)
1401                 return false;
1402
1403         /* configure and attach these filters for each associated sta vif */
1404         ieee80211_iterate_active_interfaces(
1405                 mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
1406                 iwl_mvm_bcast_filter_iterator, &iter_data);
1407
1408         return true;
1409 }
1410 static int iwl_mvm_configure_bcast_filter(struct iwl_mvm *mvm,
1411                                           struct ieee80211_vif *vif)
1412 {
1413         struct iwl_bcast_filter_cmd cmd;
1414
1415         if (!(mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_BCAST_FILTERING))
1416                 return 0;
1417
1418         if (!iwl_mvm_bcast_filter_build_cmd(mvm, &cmd))
1419                 return 0;
1420
1421         return iwl_mvm_send_cmd_pdu(mvm, BCAST_FILTER_CMD, 0,
1422                                     sizeof(cmd), &cmd);
1423 }
1424 #else
1425 static inline int iwl_mvm_configure_bcast_filter(struct iwl_mvm *mvm,
1426                                                  struct ieee80211_vif *vif)
1427 {
1428         return 0;
1429 }
1430 #endif
1431
1432 static void iwl_mvm_bss_info_changed_station(struct iwl_mvm *mvm,
1433                                              struct ieee80211_vif *vif,
1434                                              struct ieee80211_bss_conf *bss_conf,
1435                                              u32 changes)
1436 {
1437         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1438         int ret;
1439
1440         /*
1441          * Re-calculate the tsf id, as the master-slave relations depend on the
1442          * beacon interval, which was not known when the station interface was
1443          * added.
1444          */
1445         if (changes & BSS_CHANGED_ASSOC && bss_conf->assoc)
1446                 iwl_mvm_mac_ctxt_recalc_tsf_id(mvm, vif);
1447
1448         /*
1449          * If we're not associated yet, take the (new) BSSID before associating
1450          * so the firmware knows. If we're already associated, then use the old
1451          * BSSID here, and we'll send a cleared one later in the CHANGED_ASSOC
1452          * branch for disassociation below.
1453          */
1454         if (changes & BSS_CHANGED_BSSID && !mvmvif->associated)
1455                 memcpy(mvmvif->bssid, bss_conf->bssid, ETH_ALEN);
1456
1457         ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, mvmvif->bssid);
1458         if (ret)
1459                 IWL_ERR(mvm, "failed to update MAC %pM\n", vif->addr);
1460
1461         /* after sending it once, adopt mac80211 data */
1462         memcpy(mvmvif->bssid, bss_conf->bssid, ETH_ALEN);
1463         mvmvif->associated = bss_conf->assoc;
1464
1465         if (changes & BSS_CHANGED_ASSOC) {
1466                 if (bss_conf->assoc) {
1467                         /* add quota for this interface */
1468                         ret = iwl_mvm_update_quotas(mvm, NULL);
1469                         if (ret) {
1470                                 IWL_ERR(mvm, "failed to update quotas\n");
1471                                 return;
1472                         }
1473
1474                         if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART,
1475                                      &mvm->status)) {
1476                                 /*
1477                                  * If we're restarting then the firmware will
1478                                  * obviously have lost synchronisation with
1479                                  * the AP. It will attempt to synchronise by
1480                                  * itself, but we can make it more reliable by
1481                                  * scheduling a session protection time event.
1482                                  *
1483                                  * The firmware needs to receive a beacon to
1484                                  * catch up with synchronisation, use 110% of
1485                                  * the beacon interval.
1486                                  *
1487                                  * Set a large maximum delay to allow for more
1488                                  * than a single interface.
1489                                  */
1490                                 u32 dur = (11 * vif->bss_conf.beacon_int) / 10;
1491                                 iwl_mvm_protect_session(mvm, vif, dur, dur,
1492                                                         5 * dur, false);
1493                         }
1494
1495                         iwl_mvm_sf_update(mvm, vif, false);
1496                         iwl_mvm_power_vif_assoc(mvm, vif);
1497                         if (vif->p2p) {
1498                                 iwl_mvm_ref(mvm, IWL_MVM_REF_P2P_CLIENT);
1499                                 iwl_mvm_update_smps(mvm, vif,
1500                                                     IWL_MVM_SMPS_REQ_PROT,
1501                                                     IEEE80211_SMPS_DYNAMIC);
1502                         }
1503                 } else if (mvmvif->ap_sta_id != IWL_MVM_STATION_COUNT) {
1504                         /*
1505                          * If update fails - SF might be running in associated
1506                          * mode while disassociated - which is forbidden.
1507                          */
1508                         WARN_ONCE(iwl_mvm_sf_update(mvm, vif, false),
1509                                   "Failed to update SF upon disassociation\n");
1510
1511                         /* remove AP station now that the MAC is unassoc */
1512                         ret = iwl_mvm_rm_sta_id(mvm, vif, mvmvif->ap_sta_id);
1513                         if (ret)
1514                                 IWL_ERR(mvm, "failed to remove AP station\n");
1515
1516                         if (mvm->d0i3_ap_sta_id == mvmvif->ap_sta_id)
1517                                 mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
1518                         mvmvif->ap_sta_id = IWL_MVM_STATION_COUNT;
1519                         /* remove quota for this interface */
1520                         ret = iwl_mvm_update_quotas(mvm, NULL);
1521                         if (ret)
1522                                 IWL_ERR(mvm, "failed to update quotas\n");
1523
1524                         if (vif->p2p)
1525                                 iwl_mvm_unref(mvm, IWL_MVM_REF_P2P_CLIENT);
1526
1527                         /* this will take the cleared BSSID from bss_conf */
1528                         ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
1529                         if (ret)
1530                                 IWL_ERR(mvm,
1531                                         "failed to update MAC %pM (clear after unassoc)\n",
1532                                         vif->addr);
1533                 }
1534
1535                 iwl_mvm_recalc_multicast(mvm);
1536                 iwl_mvm_configure_bcast_filter(mvm, vif);
1537
1538                 /* reset rssi values */
1539                 mvmvif->bf_data.ave_beacon_signal = 0;
1540
1541                 iwl_mvm_bt_coex_vif_change(mvm);
1542                 iwl_mvm_update_smps(mvm, vif, IWL_MVM_SMPS_REQ_TT,
1543                                     IEEE80211_SMPS_AUTOMATIC);
1544         } else if (changes & BSS_CHANGED_BEACON_INFO) {
1545                 /*
1546                  * We received a beacon _after_ association so
1547                  * remove the session protection.
1548                  */
1549                 iwl_mvm_remove_time_event(mvm, mvmvif,
1550                                           &mvmvif->time_event_data);
1551         }
1552
1553         if (changes & BSS_CHANGED_BEACON_INFO) {
1554                 iwl_mvm_sf_update(mvm, vif, false);
1555                 WARN_ON(iwl_mvm_enable_beacon_filter(mvm, vif, 0));
1556         }
1557
1558         if (changes & (BSS_CHANGED_PS | BSS_CHANGED_P2P_PS | BSS_CHANGED_QOS)) {
1559                 ret = iwl_mvm_power_update_mac(mvm);
1560                 if (ret)
1561                         IWL_ERR(mvm, "failed to update power mode\n");
1562         }
1563
1564         if (changes & BSS_CHANGED_TXPOWER) {
1565                 IWL_DEBUG_CALIB(mvm, "Changing TX Power to %d\n",
1566                                 bss_conf->txpower);
1567                 iwl_mvm_set_tx_power(mvm, vif, bss_conf->txpower);
1568         }
1569
1570         if (changes & BSS_CHANGED_CQM) {
1571                 IWL_DEBUG_MAC80211(mvm, "cqm info_changed\n");
1572                 /* reset cqm events tracking */
1573                 mvmvif->bf_data.last_cqm_event = 0;
1574                 if (mvmvif->bf_data.bf_enabled) {
1575                         ret = iwl_mvm_enable_beacon_filter(mvm, vif, 0);
1576                         if (ret)
1577                                 IWL_ERR(mvm,
1578                                         "failed to update CQM thresholds\n");
1579                 }
1580         }
1581
1582         if (changes & BSS_CHANGED_ARP_FILTER) {
1583                 IWL_DEBUG_MAC80211(mvm, "arp filter changed\n");
1584                 iwl_mvm_configure_bcast_filter(mvm, vif);
1585         }
1586 }
1587
1588 static int iwl_mvm_start_ap_ibss(struct ieee80211_hw *hw,
1589                                  struct ieee80211_vif *vif)
1590 {
1591         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1592         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1593         int ret;
1594
1595         /*
1596          * iwl_mvm_mac_ctxt_add() might read directly from the device
1597          * (the system time), so make sure it is available.
1598          */
1599         ret = iwl_mvm_ref_sync(mvm, IWL_MVM_REF_START_AP);
1600         if (ret)
1601                 return ret;
1602
1603         mutex_lock(&mvm->mutex);
1604
1605         /* Send the beacon template */
1606         ret = iwl_mvm_mac_ctxt_beacon_changed(mvm, vif);
1607         if (ret)
1608                 goto out_unlock;
1609
1610         /*
1611          * Re-calculate the tsf id, as the master-slave relations depend on the
1612          * beacon interval, which was not known when the AP interface was added.
1613          */
1614         if (vif->type == NL80211_IFTYPE_AP)
1615                 iwl_mvm_mac_ctxt_recalc_tsf_id(mvm, vif);
1616
1617         /* Add the mac context */
1618         ret = iwl_mvm_mac_ctxt_add(mvm, vif);
1619         if (ret)
1620                 goto out_unlock;
1621
1622         /* Perform the binding */
1623         ret = iwl_mvm_binding_add_vif(mvm, vif);
1624         if (ret)
1625                 goto out_remove;
1626
1627         /* Send the bcast station. At this stage the TBTT and DTIM time events
1628          * are added and applied to the scheduler */
1629         ret = iwl_mvm_send_add_bcast_sta(mvm, vif);
1630         if (ret)
1631                 goto out_unbind;
1632
1633         /* must be set before quota calculations */
1634         mvmvif->ap_ibss_active = true;
1635
1636         /* power updated needs to be done before quotas */
1637         iwl_mvm_power_update_mac(mvm);
1638
1639         ret = iwl_mvm_update_quotas(mvm, NULL);
1640         if (ret)
1641                 goto out_quota_failed;
1642
1643         /* Need to update the P2P Device MAC (only GO, IBSS is single vif) */
1644         if (vif->p2p && mvm->p2p_device_vif)
1645                 iwl_mvm_mac_ctxt_changed(mvm, mvm->p2p_device_vif, false, NULL);
1646
1647         iwl_mvm_ref(mvm, IWL_MVM_REF_AP_IBSS);
1648
1649         iwl_mvm_bt_coex_vif_change(mvm);
1650
1651         /* we don't support TDLS during DCM */
1652         if (iwl_mvm_phy_ctx_count(mvm) > 1)
1653                 iwl_mvm_teardown_tdls_peers(mvm);
1654
1655         mutex_unlock(&mvm->mutex);
1656         return 0;
1657
1658 out_quota_failed:
1659         iwl_mvm_power_update_mac(mvm);
1660         mvmvif->ap_ibss_active = false;
1661         iwl_mvm_send_rm_bcast_sta(mvm, vif);
1662 out_unbind:
1663         iwl_mvm_binding_remove_vif(mvm, vif);
1664 out_remove:
1665         iwl_mvm_mac_ctxt_remove(mvm, vif);
1666 out_unlock:
1667         mutex_unlock(&mvm->mutex);
1668         iwl_mvm_unref(mvm, IWL_MVM_REF_START_AP);
1669         return ret;
1670 }
1671
1672 static void iwl_mvm_stop_ap_ibss(struct ieee80211_hw *hw,
1673                                  struct ieee80211_vif *vif)
1674 {
1675         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1676         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1677
1678         iwl_mvm_prepare_mac_removal(mvm, vif);
1679
1680         mutex_lock(&mvm->mutex);
1681
1682         /* Handle AP stop while in CSA */
1683         if (rcu_access_pointer(mvm->csa_vif) == vif) {
1684                 iwl_mvm_remove_time_event(mvm, mvmvif,
1685                                           &mvmvif->time_event_data);
1686                 RCU_INIT_POINTER(mvm->csa_vif, NULL);
1687         }
1688
1689         if (rcu_access_pointer(mvm->csa_tx_blocked_vif) == vif) {
1690                 RCU_INIT_POINTER(mvm->csa_tx_blocked_vif, NULL);
1691                 mvm->csa_tx_block_bcn_timeout = 0;
1692         }
1693
1694         mvmvif->ap_ibss_active = false;
1695         mvm->ap_last_beacon_gp2 = 0;
1696
1697         iwl_mvm_bt_coex_vif_change(mvm);
1698
1699         iwl_mvm_unref(mvm, IWL_MVM_REF_AP_IBSS);
1700
1701         /* Need to update the P2P Device MAC (only GO, IBSS is single vif) */
1702         if (vif->p2p && mvm->p2p_device_vif)
1703                 iwl_mvm_mac_ctxt_changed(mvm, mvm->p2p_device_vif, false, NULL);
1704
1705         iwl_mvm_update_quotas(mvm, NULL);
1706         iwl_mvm_send_rm_bcast_sta(mvm, vif);
1707         iwl_mvm_binding_remove_vif(mvm, vif);
1708
1709         iwl_mvm_power_update_mac(mvm);
1710
1711         iwl_mvm_mac_ctxt_remove(mvm, vif);
1712
1713         mutex_unlock(&mvm->mutex);
1714 }
1715
1716 static void
1717 iwl_mvm_bss_info_changed_ap_ibss(struct iwl_mvm *mvm,
1718                                  struct ieee80211_vif *vif,
1719                                  struct ieee80211_bss_conf *bss_conf,
1720                                  u32 changes)
1721 {
1722         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1723
1724         /* Changes will be applied when the AP/IBSS is started */
1725         if (!mvmvif->ap_ibss_active)
1726                 return;
1727
1728         if (changes & (BSS_CHANGED_ERP_CTS_PROT | BSS_CHANGED_HT |
1729                        BSS_CHANGED_BANDWIDTH | BSS_CHANGED_QOS) &&
1730             iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL))
1731                 IWL_ERR(mvm, "failed to update MAC %pM\n", vif->addr);
1732
1733         /* Need to send a new beacon template to the FW */
1734         if (changes & BSS_CHANGED_BEACON &&
1735             iwl_mvm_mac_ctxt_beacon_changed(mvm, vif))
1736                 IWL_WARN(mvm, "Failed updating beacon data\n");
1737
1738         if (changes & BSS_CHANGED_TXPOWER) {
1739                 IWL_DEBUG_CALIB(mvm, "Changing TX Power to %d\n",
1740                                 bss_conf->txpower);
1741                 iwl_mvm_set_tx_power(mvm, vif, bss_conf->txpower);
1742         }
1743
1744 }
1745
1746 static void iwl_mvm_bss_info_changed(struct ieee80211_hw *hw,
1747                                      struct ieee80211_vif *vif,
1748                                      struct ieee80211_bss_conf *bss_conf,
1749                                      u32 changes)
1750 {
1751         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1752
1753         /*
1754          * iwl_mvm_bss_info_changed_station() might call
1755          * iwl_mvm_protect_session(), which reads directly from
1756          * the device (the system time), so make sure it is available.
1757          */
1758         if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_BSS_CHANGED))
1759                 return;
1760
1761         mutex_lock(&mvm->mutex);
1762
1763         if (changes & BSS_CHANGED_IDLE && !bss_conf->idle)
1764                 iwl_mvm_scan_offload_stop(mvm, true);
1765
1766         switch (vif->type) {
1767         case NL80211_IFTYPE_STATION:
1768                 iwl_mvm_bss_info_changed_station(mvm, vif, bss_conf, changes);
1769                 break;
1770         case NL80211_IFTYPE_AP:
1771         case NL80211_IFTYPE_ADHOC:
1772                 iwl_mvm_bss_info_changed_ap_ibss(mvm, vif, bss_conf, changes);
1773                 break;
1774         default:
1775                 /* shouldn't happen */
1776                 WARN_ON_ONCE(1);
1777         }
1778
1779         mutex_unlock(&mvm->mutex);
1780         iwl_mvm_unref(mvm, IWL_MVM_REF_BSS_CHANGED);
1781 }
1782
1783 static int iwl_mvm_cancel_scan_wait_notif(struct iwl_mvm *mvm,
1784                                           enum iwl_scan_status scan_type)
1785 {
1786         int ret;
1787         bool wait_for_handlers = false;
1788
1789         mutex_lock(&mvm->mutex);
1790
1791         if (mvm->scan_status != scan_type) {
1792                 ret = 0;
1793                 /* make sure there are no pending notifications */
1794                 wait_for_handlers = true;
1795                 goto out;
1796         }
1797
1798         switch (scan_type) {
1799         case IWL_MVM_SCAN_SCHED:
1800                 ret = iwl_mvm_scan_offload_stop(mvm, true);
1801                 break;
1802         case IWL_MVM_SCAN_OS:
1803                 ret = iwl_mvm_cancel_scan(mvm);
1804                 break;
1805         case IWL_MVM_SCAN_NONE:
1806         default:
1807                 WARN_ON_ONCE(1);
1808                 ret = -EINVAL;
1809                 break;
1810         }
1811         if (ret)
1812                 goto out;
1813
1814         wait_for_handlers = true;
1815 out:
1816         mutex_unlock(&mvm->mutex);
1817
1818         /* make sure we consume the completion notification */
1819         if (wait_for_handlers)
1820                 iwl_mvm_wait_for_async_handlers(mvm);
1821
1822         return ret;
1823 }
1824 static int iwl_mvm_mac_hw_scan(struct ieee80211_hw *hw,
1825                                struct ieee80211_vif *vif,
1826                                struct ieee80211_scan_request *hw_req)
1827 {
1828         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1829         struct cfg80211_scan_request *req = &hw_req->req;
1830         int ret;
1831
1832         if (req->n_channels == 0 ||
1833             req->n_channels > mvm->fw->ucode_capa.n_scan_channels)
1834                 return -EINVAL;
1835
1836         ret = iwl_mvm_cancel_scan_wait_notif(mvm, IWL_MVM_SCAN_SCHED);
1837         if (ret)
1838                 return ret;
1839
1840         mutex_lock(&mvm->mutex);
1841
1842         if (mvm->scan_status != IWL_MVM_SCAN_NONE) {
1843                 ret = -EBUSY;
1844                 goto out;
1845         }
1846
1847         iwl_mvm_ref(mvm, IWL_MVM_REF_SCAN);
1848
1849         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)
1850                 ret = iwl_mvm_unified_scan_lmac(mvm, vif, hw_req);
1851         else
1852                 ret = iwl_mvm_scan_request(mvm, vif, req);
1853
1854         if (ret)
1855                 iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
1856 out:
1857         mutex_unlock(&mvm->mutex);
1858         return ret;
1859 }
1860
1861 static void iwl_mvm_mac_cancel_hw_scan(struct ieee80211_hw *hw,
1862                                        struct ieee80211_vif *vif)
1863 {
1864         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1865
1866         mutex_lock(&mvm->mutex);
1867
1868         iwl_mvm_cancel_scan(mvm);
1869
1870         mutex_unlock(&mvm->mutex);
1871 }
1872
1873 static void
1874 iwl_mvm_mac_allow_buffered_frames(struct ieee80211_hw *hw,
1875                                   struct ieee80211_sta *sta, u16 tids,
1876                                   int num_frames,
1877                                   enum ieee80211_frame_release_type reason,
1878                                   bool more_data)
1879 {
1880         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1881
1882         /* Called when we need to transmit (a) frame(s) from mac80211 */
1883
1884         iwl_mvm_sta_modify_sleep_tx_count(mvm, sta, reason, num_frames,
1885                                           tids, more_data, false);
1886 }
1887
1888 static void
1889 iwl_mvm_mac_release_buffered_frames(struct ieee80211_hw *hw,
1890                                     struct ieee80211_sta *sta, u16 tids,
1891                                     int num_frames,
1892                                     enum ieee80211_frame_release_type reason,
1893                                     bool more_data)
1894 {
1895         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1896
1897         /* Called when we need to transmit (a) frame(s) from agg queue */
1898
1899         iwl_mvm_sta_modify_sleep_tx_count(mvm, sta, reason, num_frames,
1900                                           tids, more_data, true);
1901 }
1902
1903 static void iwl_mvm_mac_sta_notify(struct ieee80211_hw *hw,
1904                                    struct ieee80211_vif *vif,
1905                                    enum sta_notify_cmd cmd,
1906                                    struct ieee80211_sta *sta)
1907 {
1908         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1909         struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1910         int tid;
1911
1912         switch (cmd) {
1913         case STA_NOTIFY_SLEEP:
1914                 if (atomic_read(&mvm->pending_frames[mvmsta->sta_id]) > 0)
1915                         ieee80211_sta_block_awake(hw, sta, true);
1916                 spin_lock_bh(&mvmsta->lock);
1917                 for (tid = 0; tid < IWL_MAX_TID_COUNT; tid++) {
1918                         struct iwl_mvm_tid_data *tid_data;
1919
1920                         tid_data = &mvmsta->tid_data[tid];
1921                         if (tid_data->state != IWL_AGG_ON &&
1922                             tid_data->state != IWL_EMPTYING_HW_QUEUE_DELBA)
1923                                 continue;
1924                         if (iwl_mvm_tid_queued(tid_data) == 0)
1925                                 continue;
1926                         ieee80211_sta_set_buffered(sta, tid, true);
1927                 }
1928                 spin_unlock_bh(&mvmsta->lock);
1929                 /*
1930                  * The fw updates the STA to be asleep. Tx packets on the Tx
1931                  * queues to this station will not be transmitted. The fw will
1932                  * send a Tx response with TX_STATUS_FAIL_DEST_PS.
1933                  */
1934                 break;
1935         case STA_NOTIFY_AWAKE:
1936                 if (WARN_ON(mvmsta->sta_id == IWL_MVM_STATION_COUNT))
1937                         break;
1938                 iwl_mvm_sta_modify_ps_wake(mvm, sta);
1939                 break;
1940         default:
1941                 break;
1942         }
1943 }
1944
1945 static void iwl_mvm_sta_pre_rcu_remove(struct ieee80211_hw *hw,
1946                                        struct ieee80211_vif *vif,
1947                                        struct ieee80211_sta *sta)
1948 {
1949         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1950         struct iwl_mvm_sta *mvm_sta = (void *)sta->drv_priv;
1951
1952         /*
1953          * This is called before mac80211 does RCU synchronisation,
1954          * so here we already invalidate our internal RCU-protected
1955          * station pointer. The rest of the code will thus no longer
1956          * be able to find the station this way, and we don't rely
1957          * on further RCU synchronisation after the sta_state()
1958          * callback deleted the station.
1959          */
1960         mutex_lock(&mvm->mutex);
1961         if (sta == rcu_access_pointer(mvm->fw_id_to_mac_id[mvm_sta->sta_id]))
1962                 rcu_assign_pointer(mvm->fw_id_to_mac_id[mvm_sta->sta_id],
1963                                    ERR_PTR(-ENOENT));
1964         mutex_unlock(&mvm->mutex);
1965 }
1966
1967 static int iwl_mvm_mac_sta_state(struct ieee80211_hw *hw,
1968                                  struct ieee80211_vif *vif,
1969                                  struct ieee80211_sta *sta,
1970                                  enum ieee80211_sta_state old_state,
1971                                  enum ieee80211_sta_state new_state)
1972 {
1973         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
1974         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1975         int ret;
1976
1977         IWL_DEBUG_MAC80211(mvm, "station %pM state change %d->%d\n",
1978                            sta->addr, old_state, new_state);
1979
1980         /* this would be a mac80211 bug ... but don't crash */
1981         if (WARN_ON_ONCE(!mvmvif->phy_ctxt))
1982                 return -EINVAL;
1983
1984         /* if a STA is being removed, reuse its ID */
1985         flush_work(&mvm->sta_drained_wk);
1986
1987         mutex_lock(&mvm->mutex);
1988         if (old_state == IEEE80211_STA_NOTEXIST &&
1989             new_state == IEEE80211_STA_NONE) {
1990                 /*
1991                  * Firmware bug - it'll crash if the beacon interval is less
1992                  * than 16. We can't avoid connecting at all, so refuse the
1993                  * station state change, this will cause mac80211 to abandon
1994                  * attempts to connect to this AP, and eventually wpa_s will
1995                  * blacklist the AP...
1996                  */
1997                 if (vif->type == NL80211_IFTYPE_STATION &&
1998                     vif->bss_conf.beacon_int < 16) {
1999                         IWL_ERR(mvm,
2000                                 "AP %pM beacon interval is %d, refusing due to firmware bug!\n",
2001                                 sta->addr, vif->bss_conf.beacon_int);
2002                         ret = -EINVAL;
2003                         goto out_unlock;
2004                 }
2005
2006                 if (sta->tdls &&
2007                     (vif->p2p ||
2008                      iwl_mvm_tdls_sta_count(mvm, NULL) ==
2009                                                 IWL_MVM_TDLS_STA_COUNT ||
2010                      iwl_mvm_phy_ctx_count(mvm) > 1)) {
2011                         IWL_DEBUG_MAC80211(mvm, "refusing TDLS sta\n");
2012                         ret = -EBUSY;
2013                         goto out_unlock;
2014                 }
2015
2016                 ret = iwl_mvm_add_sta(mvm, vif, sta);
2017                 if (sta->tdls && ret == 0)
2018                         iwl_mvm_recalc_tdls_state(mvm, vif, true);
2019         } else if (old_state == IEEE80211_STA_NONE &&
2020                    new_state == IEEE80211_STA_AUTH) {
2021                 /*
2022                  * EBS may be disabled due to previous failures reported by FW.
2023                  * Reset EBS status here assuming environment has been changed.
2024                  */
2025                 mvm->last_ebs_successful = true;
2026                 ret = 0;
2027         } else if (old_state == IEEE80211_STA_AUTH &&
2028                    new_state == IEEE80211_STA_ASSOC) {
2029                 ret = iwl_mvm_update_sta(mvm, vif, sta);
2030                 if (ret == 0)
2031                         iwl_mvm_rs_rate_init(mvm, sta,
2032                                              mvmvif->phy_ctxt->channel->band,
2033                                              true);
2034         } else if (old_state == IEEE80211_STA_ASSOC &&
2035                    new_state == IEEE80211_STA_AUTHORIZED) {
2036
2037                 /* we don't support TDLS during DCM */
2038                 if (iwl_mvm_phy_ctx_count(mvm) > 1)
2039                         iwl_mvm_teardown_tdls_peers(mvm);
2040
2041                 /* enable beacon filtering */
2042                 WARN_ON(iwl_mvm_enable_beacon_filter(mvm, vif, 0));
2043                 ret = 0;
2044         } else if (old_state == IEEE80211_STA_AUTHORIZED &&
2045                    new_state == IEEE80211_STA_ASSOC) {
2046                 /* disable beacon filtering */
2047                 WARN_ON(iwl_mvm_disable_beacon_filter(mvm, vif, 0));
2048                 ret = 0;
2049         } else if (old_state == IEEE80211_STA_ASSOC &&
2050                    new_state == IEEE80211_STA_AUTH) {
2051                 ret = 0;
2052         } else if (old_state == IEEE80211_STA_AUTH &&
2053                    new_state == IEEE80211_STA_NONE) {
2054                 ret = 0;
2055         } else if (old_state == IEEE80211_STA_NONE &&
2056                    new_state == IEEE80211_STA_NOTEXIST) {
2057                 ret = iwl_mvm_rm_sta(mvm, vif, sta);
2058                 if (sta->tdls)
2059                         iwl_mvm_recalc_tdls_state(mvm, vif, false);
2060         } else {
2061                 ret = -EIO;
2062         }
2063  out_unlock:
2064         mutex_unlock(&mvm->mutex);
2065
2066         return ret;
2067 }
2068
2069 static int iwl_mvm_mac_set_rts_threshold(struct ieee80211_hw *hw, u32 value)
2070 {
2071         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2072
2073         mvm->rts_threshold = value;
2074
2075         return 0;
2076 }
2077
2078 static void iwl_mvm_sta_rc_update(struct ieee80211_hw *hw,
2079                                   struct ieee80211_vif *vif,
2080                                   struct ieee80211_sta *sta, u32 changed)
2081 {
2082         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2083
2084         if (vif->type == NL80211_IFTYPE_STATION &&
2085             changed & IEEE80211_RC_NSS_CHANGED)
2086                 iwl_mvm_sf_update(mvm, vif, false);
2087 }
2088
2089 static int iwl_mvm_mac_conf_tx(struct ieee80211_hw *hw,
2090                                struct ieee80211_vif *vif, u16 ac,
2091                                const struct ieee80211_tx_queue_params *params)
2092 {
2093         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2094         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2095
2096         mvmvif->queue_params[ac] = *params;
2097
2098         /*
2099          * No need to update right away, we'll get BSS_CHANGED_QOS
2100          * The exception is P2P_DEVICE interface which needs immediate update.
2101          */
2102         if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
2103                 int ret;
2104
2105                 mutex_lock(&mvm->mutex);
2106                 ret = iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
2107                 mutex_unlock(&mvm->mutex);
2108                 return ret;
2109         }
2110         return 0;
2111 }
2112
2113 static void iwl_mvm_mac_mgd_prepare_tx(struct ieee80211_hw *hw,
2114                                       struct ieee80211_vif *vif)
2115 {
2116         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2117         u32 duration = min(IWL_MVM_TE_SESSION_PROTECTION_MAX_TIME_MS,
2118                            200 + vif->bss_conf.beacon_int);
2119         u32 min_duration = min(IWL_MVM_TE_SESSION_PROTECTION_MIN_TIME_MS,
2120                                100 + vif->bss_conf.beacon_int);
2121
2122         if (WARN_ON_ONCE(vif->bss_conf.assoc))
2123                 return;
2124
2125         /*
2126          * iwl_mvm_protect_session() reads directly from the device
2127          * (the system time), so make sure it is available.
2128          */
2129         if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_PREPARE_TX))
2130                 return;
2131
2132         mutex_lock(&mvm->mutex);
2133         /* Try really hard to protect the session and hear a beacon */
2134         iwl_mvm_protect_session(mvm, vif, duration, min_duration, 500, false);
2135         mutex_unlock(&mvm->mutex);
2136
2137         iwl_mvm_unref(mvm, IWL_MVM_REF_PREPARE_TX);
2138 }
2139
2140 static int iwl_mvm_mac_sched_scan_start(struct ieee80211_hw *hw,
2141                                         struct ieee80211_vif *vif,
2142                                         struct cfg80211_sched_scan_request *req,
2143                                         struct ieee80211_scan_ies *ies)
2144 {
2145         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2146         int ret;
2147
2148         ret = iwl_mvm_cancel_scan_wait_notif(mvm, IWL_MVM_SCAN_OS);
2149         if (ret)
2150                 return ret;
2151
2152         mutex_lock(&mvm->mutex);
2153
2154         /* Newest FW fixes sched scan while connected on another interface */
2155         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN) {
2156                 if (!vif->bss_conf.idle) {
2157                         ret = -EBUSY;
2158                         goto out;
2159                 }
2160         } else if (!iwl_mvm_is_idle(mvm)) {
2161                 ret = -EBUSY;
2162                 goto out;
2163         }
2164
2165         if (mvm->scan_status != IWL_MVM_SCAN_NONE) {
2166                 ret = -EBUSY;
2167                 goto out;
2168         }
2169
2170         mvm->scan_status = IWL_MVM_SCAN_SCHED;
2171
2172         if (!(mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)) {
2173                 ret = iwl_mvm_config_sched_scan(mvm, vif, req, ies);
2174                 if (ret)
2175                         goto err;
2176         }
2177
2178         ret = iwl_mvm_config_sched_scan_profiles(mvm, req);
2179         if (ret)
2180                 goto err;
2181
2182         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)
2183                 ret = iwl_mvm_unified_sched_scan_lmac(mvm, vif, req, ies);
2184         else
2185                 ret = iwl_mvm_sched_scan_start(mvm, req);
2186
2187         if (!ret)
2188                 goto out;
2189 err:
2190         mvm->scan_status = IWL_MVM_SCAN_NONE;
2191 out:
2192         mutex_unlock(&mvm->mutex);
2193         return ret;
2194 }
2195
2196 static int iwl_mvm_mac_sched_scan_stop(struct ieee80211_hw *hw,
2197                                        struct ieee80211_vif *vif)
2198 {
2199         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2200         int ret;
2201
2202         mutex_lock(&mvm->mutex);
2203         ret = iwl_mvm_scan_offload_stop(mvm, false);
2204         mutex_unlock(&mvm->mutex);
2205         iwl_mvm_wait_for_async_handlers(mvm);
2206
2207         return ret;
2208 }
2209
2210 static int iwl_mvm_mac_set_key(struct ieee80211_hw *hw,
2211                                enum set_key_cmd cmd,
2212                                struct ieee80211_vif *vif,
2213                                struct ieee80211_sta *sta,
2214                                struct ieee80211_key_conf *key)
2215 {
2216         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2217         int ret;
2218
2219         if (iwlwifi_mod_params.sw_crypto) {
2220                 IWL_DEBUG_MAC80211(mvm, "leave - hwcrypto disabled\n");
2221                 return -EOPNOTSUPP;
2222         }
2223
2224         switch (key->cipher) {
2225         case WLAN_CIPHER_SUITE_TKIP:
2226                 key->flags |= IEEE80211_KEY_FLAG_GENERATE_MMIC;
2227                 /* fall-through */
2228         case WLAN_CIPHER_SUITE_CCMP:
2229                 key->flags |= IEEE80211_KEY_FLAG_GENERATE_IV;
2230                 break;
2231         case WLAN_CIPHER_SUITE_AES_CMAC:
2232                 WARN_ON_ONCE(!(hw->flags & IEEE80211_HW_MFP_CAPABLE));
2233                 break;
2234         case WLAN_CIPHER_SUITE_WEP40:
2235         case WLAN_CIPHER_SUITE_WEP104:
2236                 /*
2237                  * Support for TX only, at least for now, so accept
2238                  * the key and do nothing else. Then mac80211 will
2239                  * pass it for TX but we don't have to use it for RX.
2240                  */
2241                 return 0;
2242         default:
2243                 /* currently FW supports only one optional cipher scheme */
2244                 if (hw->n_cipher_schemes &&
2245                     hw->cipher_schemes->cipher == key->cipher)
2246                         key->flags |= IEEE80211_KEY_FLAG_PUT_IV_SPACE;
2247                 else
2248                         return -EOPNOTSUPP;
2249         }
2250
2251         mutex_lock(&mvm->mutex);
2252
2253         switch (cmd) {
2254         case SET_KEY:
2255                 if ((vif->type == NL80211_IFTYPE_ADHOC ||
2256                      vif->type == NL80211_IFTYPE_AP) && !sta) {
2257                         /*
2258                          * GTK on AP interface is a TX-only key, return 0;
2259                          * on IBSS they're per-station and because we're lazy
2260                          * we don't support them for RX, so do the same.
2261                          */
2262                         ret = 0;
2263                         key->hw_key_idx = STA_KEY_IDX_INVALID;
2264                         break;
2265                 }
2266
2267                 IWL_DEBUG_MAC80211(mvm, "set hwcrypto key\n");
2268                 ret = iwl_mvm_set_sta_key(mvm, vif, sta, key, false);
2269                 if (ret) {
2270                         IWL_WARN(mvm, "set key failed\n");
2271                         /*
2272                          * can't add key for RX, but we don't need it
2273                          * in the device for TX so still return 0
2274                          */
2275                         key->hw_key_idx = STA_KEY_IDX_INVALID;
2276                         ret = 0;
2277                 }
2278
2279                 break;
2280         case DISABLE_KEY:
2281                 if (key->hw_key_idx == STA_KEY_IDX_INVALID) {
2282                         ret = 0;
2283                         break;
2284                 }
2285
2286                 IWL_DEBUG_MAC80211(mvm, "disable hwcrypto key\n");
2287                 ret = iwl_mvm_remove_sta_key(mvm, vif, sta, key);
2288                 break;
2289         default:
2290                 ret = -EINVAL;
2291         }
2292
2293         mutex_unlock(&mvm->mutex);
2294         return ret;
2295 }
2296
2297 static void iwl_mvm_mac_update_tkip_key(struct ieee80211_hw *hw,
2298                                         struct ieee80211_vif *vif,
2299                                         struct ieee80211_key_conf *keyconf,
2300                                         struct ieee80211_sta *sta,
2301                                         u32 iv32, u16 *phase1key)
2302 {
2303         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2304
2305         if (keyconf->hw_key_idx == STA_KEY_IDX_INVALID)
2306                 return;
2307
2308         iwl_mvm_update_tkip_key(mvm, vif, keyconf, sta, iv32, phase1key);
2309 }
2310
2311
2312 static bool iwl_mvm_rx_aux_roc(struct iwl_notif_wait_data *notif_wait,
2313                                struct iwl_rx_packet *pkt, void *data)
2314 {
2315         struct iwl_mvm *mvm =
2316                 container_of(notif_wait, struct iwl_mvm, notif_wait);
2317         struct iwl_hs20_roc_res *resp;
2318         int resp_len = iwl_rx_packet_payload_len(pkt);
2319         struct iwl_mvm_time_event_data *te_data = data;
2320
2321         if (WARN_ON(pkt->hdr.cmd != HOT_SPOT_CMD))
2322                 return true;
2323
2324         if (WARN_ON_ONCE(resp_len != sizeof(*resp))) {
2325                 IWL_ERR(mvm, "Invalid HOT_SPOT_CMD response\n");
2326                 return true;
2327         }
2328
2329         resp = (void *)pkt->data;
2330
2331         IWL_DEBUG_TE(mvm,
2332                      "Aux ROC: Recieved response from ucode: status=%d uid=%d\n",
2333                      resp->status, resp->event_unique_id);
2334
2335         te_data->uid = le32_to_cpu(resp->event_unique_id);
2336         IWL_DEBUG_TE(mvm, "TIME_EVENT_CMD response - UID = 0x%x\n",
2337                      te_data->uid);
2338
2339         spin_lock_bh(&mvm->time_event_lock);
2340         list_add_tail(&te_data->list, &mvm->aux_roc_te_list);
2341         spin_unlock_bh(&mvm->time_event_lock);
2342
2343         return true;
2344 }
2345
2346 #define AUX_ROC_MAX_DELAY_ON_CHANNEL 5000
2347 static int iwl_mvm_send_aux_roc_cmd(struct iwl_mvm *mvm,
2348                                     struct ieee80211_channel *channel,
2349                                     struct ieee80211_vif *vif,
2350                                     int duration)
2351 {
2352         int res, time_reg = DEVICE_SYSTEM_TIME_REG;
2353         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2354         struct iwl_mvm_time_event_data *te_data = &mvmvif->hs_time_event_data;
2355         static const u8 time_event_response[] = { HOT_SPOT_CMD };
2356         struct iwl_notification_wait wait_time_event;
2357         struct iwl_hs20_roc_req aux_roc_req = {
2358                 .action = cpu_to_le32(FW_CTXT_ACTION_ADD),
2359                 .id_and_color =
2360                         cpu_to_le32(FW_CMD_ID_AND_COLOR(MAC_INDEX_AUX, 0)),
2361                 .sta_id_and_color = cpu_to_le32(mvm->aux_sta.sta_id),
2362                 /* Set the channel info data */
2363                 .channel_info.band = (channel->band == IEEE80211_BAND_2GHZ) ?
2364                         PHY_BAND_24 : PHY_BAND_5,
2365                 .channel_info.channel = channel->hw_value,
2366                 .channel_info.width = PHY_VHT_CHANNEL_MODE20,
2367                 /* Set the time and duration */
2368                 .apply_time = cpu_to_le32(iwl_read_prph(mvm->trans, time_reg)),
2369                 .apply_time_max_delay =
2370                         cpu_to_le32(MSEC_TO_TU(AUX_ROC_MAX_DELAY_ON_CHANNEL)),
2371                 .duration = cpu_to_le32(MSEC_TO_TU(duration)),
2372          };
2373
2374         /* Set the node address */
2375         memcpy(aux_roc_req.node_addr, vif->addr, ETH_ALEN);
2376
2377         te_data->vif = vif;
2378         te_data->duration = duration;
2379         te_data->id = HOT_SPOT_CMD;
2380
2381         lockdep_assert_held(&mvm->mutex);
2382
2383         spin_lock_bh(&mvm->time_event_lock);
2384         list_add_tail(&te_data->list, &mvm->time_event_list);
2385         spin_unlock_bh(&mvm->time_event_lock);
2386
2387         /*
2388          * Use a notification wait, which really just processes the
2389          * command response and doesn't wait for anything, in order
2390          * to be able to process the response and get the UID inside
2391          * the RX path. Using CMD_WANT_SKB doesn't work because it
2392          * stores the buffer and then wakes up this thread, by which
2393          * time another notification (that the time event started)
2394          * might already be processed unsuccessfully.
2395          */
2396         iwl_init_notification_wait(&mvm->notif_wait, &wait_time_event,
2397                                    time_event_response,
2398                                    ARRAY_SIZE(time_event_response),
2399                                    iwl_mvm_rx_aux_roc, te_data);
2400
2401         res = iwl_mvm_send_cmd_pdu(mvm, HOT_SPOT_CMD, 0, sizeof(aux_roc_req),
2402                                    &aux_roc_req);
2403
2404         if (res) {
2405                 IWL_ERR(mvm, "Couldn't send HOT_SPOT_CMD: %d\n", res);
2406                 iwl_remove_notification(&mvm->notif_wait, &wait_time_event);
2407                 goto out_clear_te;
2408         }
2409
2410         /* No need to wait for anything, so just pass 1 (0 isn't valid) */
2411         res = iwl_wait_notification(&mvm->notif_wait, &wait_time_event, 1);
2412         /* should never fail */
2413         WARN_ON_ONCE(res);
2414
2415         if (res) {
2416  out_clear_te:
2417                 spin_lock_bh(&mvm->time_event_lock);
2418                 iwl_mvm_te_clear_data(mvm, te_data);
2419                 spin_unlock_bh(&mvm->time_event_lock);
2420         }
2421
2422         return res;
2423 }
2424
2425 static int iwl_mvm_roc(struct ieee80211_hw *hw,
2426                        struct ieee80211_vif *vif,
2427                        struct ieee80211_channel *channel,
2428                        int duration,
2429                        enum ieee80211_roc_type type)
2430 {
2431         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2432         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2433         struct cfg80211_chan_def chandef;
2434         struct iwl_mvm_phy_ctxt *phy_ctxt;
2435         int ret, i;
2436
2437         IWL_DEBUG_MAC80211(mvm, "enter (%d, %d, %d)\n", channel->hw_value,
2438                            duration, type);
2439
2440         switch (vif->type) {
2441         case NL80211_IFTYPE_STATION:
2442                 /* Use aux roc framework (HS20) */
2443                 ret = iwl_mvm_send_aux_roc_cmd(mvm, channel,
2444                                                vif, duration);
2445                 return ret;
2446         case NL80211_IFTYPE_P2P_DEVICE:
2447                 /* handle below */
2448                 break;
2449         default:
2450                 IWL_ERR(mvm, "vif isn't P2P_DEVICE: %d\n", vif->type);
2451                 return -EINVAL;
2452         }
2453
2454         mutex_lock(&mvm->mutex);
2455
2456         for (i = 0; i < NUM_PHY_CTX; i++) {
2457                 phy_ctxt = &mvm->phy_ctxts[i];
2458                 if (phy_ctxt->ref == 0 || mvmvif->phy_ctxt == phy_ctxt)
2459                         continue;
2460
2461                 if (phy_ctxt->ref && channel == phy_ctxt->channel) {
2462                         /*
2463                          * Unbind the P2P_DEVICE from the current PHY context,
2464                          * and if the PHY context is not used remove it.
2465                          */
2466                         ret = iwl_mvm_binding_remove_vif(mvm, vif);
2467                         if (WARN(ret, "Failed unbinding P2P_DEVICE\n"))
2468                                 goto out_unlock;
2469
2470                         iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
2471
2472                         /* Bind the P2P_DEVICE to the current PHY Context */
2473                         mvmvif->phy_ctxt = phy_ctxt;
2474
2475                         ret = iwl_mvm_binding_add_vif(mvm, vif);
2476                         if (WARN(ret, "Failed binding P2P_DEVICE\n"))
2477                                 goto out_unlock;
2478
2479                         iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
2480                         goto schedule_time_event;
2481                 }
2482         }
2483
2484         /* Need to update the PHY context only if the ROC channel changed */
2485         if (channel == mvmvif->phy_ctxt->channel)
2486                 goto schedule_time_event;
2487
2488         cfg80211_chandef_create(&chandef, channel, NL80211_CHAN_NO_HT);
2489
2490         /*
2491          * Change the PHY context configuration as it is currently referenced
2492          * only by the P2P Device MAC
2493          */
2494         if (mvmvif->phy_ctxt->ref == 1) {
2495                 ret = iwl_mvm_phy_ctxt_changed(mvm, mvmvif->phy_ctxt,
2496                                                &chandef, 1, 1);
2497                 if (ret)
2498                         goto out_unlock;
2499         } else {
2500                 /*
2501                  * The PHY context is shared with other MACs. Need to remove the
2502                  * P2P Device from the binding, allocate an new PHY context and
2503                  * create a new binding
2504                  */
2505                 phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
2506                 if (!phy_ctxt) {
2507                         ret = -ENOSPC;
2508                         goto out_unlock;
2509                 }
2510
2511                 ret = iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &chandef,
2512                                                1, 1);
2513                 if (ret) {
2514                         IWL_ERR(mvm, "Failed to change PHY context\n");
2515                         goto out_unlock;
2516                 }
2517
2518                 /* Unbind the P2P_DEVICE from the current PHY context */
2519                 ret = iwl_mvm_binding_remove_vif(mvm, vif);
2520                 if (WARN(ret, "Failed unbinding P2P_DEVICE\n"))
2521                         goto out_unlock;
2522
2523                 iwl_mvm_phy_ctxt_unref(mvm, mvmvif->phy_ctxt);
2524
2525                 /* Bind the P2P_DEVICE to the new allocated PHY context */
2526                 mvmvif->phy_ctxt = phy_ctxt;
2527
2528                 ret = iwl_mvm_binding_add_vif(mvm, vif);
2529                 if (WARN(ret, "Failed binding P2P_DEVICE\n"))
2530                         goto out_unlock;
2531
2532                 iwl_mvm_phy_ctxt_ref(mvm, mvmvif->phy_ctxt);
2533         }
2534
2535 schedule_time_event:
2536         /* Schedule the time events */
2537         ret = iwl_mvm_start_p2p_roc(mvm, vif, duration, type);
2538
2539 out_unlock:
2540         mutex_unlock(&mvm->mutex);
2541         IWL_DEBUG_MAC80211(mvm, "leave\n");
2542         return ret;
2543 }
2544
2545 static int iwl_mvm_cancel_roc(struct ieee80211_hw *hw)
2546 {
2547         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2548
2549         IWL_DEBUG_MAC80211(mvm, "enter\n");
2550
2551         mutex_lock(&mvm->mutex);
2552         iwl_mvm_stop_p2p_roc(mvm);
2553         mutex_unlock(&mvm->mutex);
2554
2555         IWL_DEBUG_MAC80211(mvm, "leave\n");
2556         return 0;
2557 }
2558
2559 static int __iwl_mvm_add_chanctx(struct iwl_mvm *mvm,
2560                                  struct ieee80211_chanctx_conf *ctx)
2561 {
2562         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
2563         struct iwl_mvm_phy_ctxt *phy_ctxt;
2564         int ret;
2565
2566         lockdep_assert_held(&mvm->mutex);
2567
2568         IWL_DEBUG_MAC80211(mvm, "Add channel context\n");
2569
2570         phy_ctxt = iwl_mvm_get_free_phy_ctxt(mvm);
2571         if (!phy_ctxt) {
2572                 ret = -ENOSPC;
2573                 goto out;
2574         }
2575
2576         ret = iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &ctx->min_def,
2577                                        ctx->rx_chains_static,
2578                                        ctx->rx_chains_dynamic);
2579         if (ret) {
2580                 IWL_ERR(mvm, "Failed to add PHY context\n");
2581                 goto out;
2582         }
2583
2584         iwl_mvm_phy_ctxt_ref(mvm, phy_ctxt);
2585         *phy_ctxt_id = phy_ctxt->id;
2586 out:
2587         return ret;
2588 }
2589
2590 static int iwl_mvm_add_chanctx(struct ieee80211_hw *hw,
2591                                struct ieee80211_chanctx_conf *ctx)
2592 {
2593         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2594         int ret;
2595
2596         mutex_lock(&mvm->mutex);
2597         ret = __iwl_mvm_add_chanctx(mvm, ctx);
2598         mutex_unlock(&mvm->mutex);
2599
2600         return ret;
2601 }
2602
2603 static void __iwl_mvm_remove_chanctx(struct iwl_mvm *mvm,
2604                                      struct ieee80211_chanctx_conf *ctx)
2605 {
2606         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
2607         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
2608
2609         lockdep_assert_held(&mvm->mutex);
2610
2611         iwl_mvm_phy_ctxt_unref(mvm, phy_ctxt);
2612 }
2613
2614 static void iwl_mvm_remove_chanctx(struct ieee80211_hw *hw,
2615                                    struct ieee80211_chanctx_conf *ctx)
2616 {
2617         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2618
2619         mutex_lock(&mvm->mutex);
2620         __iwl_mvm_remove_chanctx(mvm, ctx);
2621         mutex_unlock(&mvm->mutex);
2622 }
2623
2624 static void iwl_mvm_change_chanctx(struct ieee80211_hw *hw,
2625                                    struct ieee80211_chanctx_conf *ctx,
2626                                    u32 changed)
2627 {
2628         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2629         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
2630         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
2631
2632         if (WARN_ONCE((phy_ctxt->ref > 1) &&
2633                       (changed & ~(IEEE80211_CHANCTX_CHANGE_WIDTH |
2634                                    IEEE80211_CHANCTX_CHANGE_RX_CHAINS |
2635                                    IEEE80211_CHANCTX_CHANGE_RADAR |
2636                                    IEEE80211_CHANCTX_CHANGE_MIN_WIDTH)),
2637                       "Cannot change PHY. Ref=%d, changed=0x%X\n",
2638                       phy_ctxt->ref, changed))
2639                 return;
2640
2641         mutex_lock(&mvm->mutex);
2642         iwl_mvm_bt_coex_vif_change(mvm);
2643         iwl_mvm_phy_ctxt_changed(mvm, phy_ctxt, &ctx->min_def,
2644                                  ctx->rx_chains_static,
2645                                  ctx->rx_chains_dynamic);
2646         mutex_unlock(&mvm->mutex);
2647 }
2648
2649 static int __iwl_mvm_assign_vif_chanctx(struct iwl_mvm *mvm,
2650                                         struct ieee80211_vif *vif,
2651                                         struct ieee80211_chanctx_conf *ctx,
2652                                         bool switching_chanctx)
2653 {
2654         u16 *phy_ctxt_id = (u16 *)ctx->drv_priv;
2655         struct iwl_mvm_phy_ctxt *phy_ctxt = &mvm->phy_ctxts[*phy_ctxt_id];
2656         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2657         int ret;
2658
2659         lockdep_assert_held(&mvm->mutex);
2660
2661         mvmvif->phy_ctxt = phy_ctxt;
2662
2663         switch (vif->type) {
2664         case NL80211_IFTYPE_AP:
2665                 /* Unless it's a CSA flow we have nothing to do here */
2666                 if (vif->csa_active) {
2667                         mvmvif->ap_ibss_active = true;
2668                         break;
2669                 }
2670         case NL80211_IFTYPE_ADHOC:
2671                 /*
2672                  * The AP binding flow is handled as part of the start_ap flow
2673                  * (in bss_info_changed), similarly for IBSS.
2674                  */
2675                 ret = 0;
2676                 goto out;
2677         case NL80211_IFTYPE_STATION:
2678                 break;
2679         case NL80211_IFTYPE_MONITOR:
2680                 /* always disable PS when a monitor interface is active */
2681                 mvmvif->ps_disabled = true;
2682                 break;
2683         default:
2684                 ret = -EINVAL;
2685                 goto out;
2686         }
2687
2688         ret = iwl_mvm_binding_add_vif(mvm, vif);
2689         if (ret)
2690                 goto out;
2691
2692         /*
2693          * Power state must be updated before quotas,
2694          * otherwise fw will complain.
2695          */
2696         iwl_mvm_power_update_mac(mvm);
2697
2698         /* Setting the quota at this stage is only required for monitor
2699          * interfaces. For the other types, the bss_info changed flow
2700          * will handle quota settings.
2701          */
2702         if (vif->type == NL80211_IFTYPE_MONITOR) {
2703                 mvmvif->monitor_active = true;
2704                 ret = iwl_mvm_update_quotas(mvm, NULL);
2705                 if (ret)
2706                         goto out_remove_binding;
2707         }
2708
2709         /* Handle binding during CSA */
2710         if ((vif->type == NL80211_IFTYPE_AP) ||
2711             (switching_chanctx && (vif->type == NL80211_IFTYPE_STATION))) {
2712                 iwl_mvm_update_quotas(mvm, NULL);
2713                 iwl_mvm_mac_ctxt_changed(mvm, vif, false, NULL);
2714         }
2715
2716         if (vif->csa_active && vif->type == NL80211_IFTYPE_STATION) {
2717                 struct iwl_mvm_sta *mvmsta;
2718
2719                 mvmsta = iwl_mvm_sta_from_staid_protected(mvm,
2720                                                           mvmvif->ap_sta_id);
2721
2722                 if (WARN_ON(!mvmsta))
2723                         goto out;
2724
2725                 /* TODO: only re-enable after the first beacon */
2726                 iwl_mvm_sta_modify_disable_tx(mvm, mvmsta, false);
2727         }
2728
2729         goto out;
2730
2731 out_remove_binding:
2732         iwl_mvm_binding_remove_vif(mvm, vif);
2733         iwl_mvm_power_update_mac(mvm);
2734 out:
2735         if (ret)
2736                 mvmvif->phy_ctxt = NULL;
2737         return ret;
2738 }
2739 static int iwl_mvm_assign_vif_chanctx(struct ieee80211_hw *hw,
2740                                       struct ieee80211_vif *vif,
2741                                       struct ieee80211_chanctx_conf *ctx)
2742 {
2743         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2744         int ret;
2745
2746         mutex_lock(&mvm->mutex);
2747         ret = __iwl_mvm_assign_vif_chanctx(mvm, vif, ctx, false);
2748         mutex_unlock(&mvm->mutex);
2749
2750         return ret;
2751 }
2752
2753 static void __iwl_mvm_unassign_vif_chanctx(struct iwl_mvm *mvm,
2754                                            struct ieee80211_vif *vif,
2755                                            struct ieee80211_chanctx_conf *ctx,
2756                                            bool switching_chanctx)
2757 {
2758         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2759         struct ieee80211_vif *disabled_vif = NULL;
2760         struct iwl_mvm_sta *mvmsta;
2761
2762         lockdep_assert_held(&mvm->mutex);
2763
2764         iwl_mvm_remove_time_event(mvm, mvmvif, &mvmvif->time_event_data);
2765
2766         switch (vif->type) {
2767         case NL80211_IFTYPE_ADHOC:
2768                 goto out;
2769         case NL80211_IFTYPE_MONITOR:
2770                 mvmvif->monitor_active = false;
2771                 mvmvif->ps_disabled = false;
2772                 break;
2773         case NL80211_IFTYPE_AP:
2774                 /* This part is triggered only during CSA */
2775                 if (!vif->csa_active || !mvmvif->ap_ibss_active)
2776                         goto out;
2777
2778                 /* Set CS bit on all the stations */
2779                 iwl_mvm_modify_all_sta_disable_tx(mvm, mvmvif, true);
2780
2781                 /* Save blocked iface, the timeout is set on the next beacon */
2782                 rcu_assign_pointer(mvm->csa_tx_blocked_vif, vif);
2783
2784                 mvmvif->ap_ibss_active = false;
2785                 break;
2786         case NL80211_IFTYPE_STATION:
2787                 if (!switching_chanctx)
2788                         break;
2789
2790                 disabled_vif = vif;
2791
2792                 mvmsta = iwl_mvm_sta_from_staid_protected(mvm,
2793                                                           mvmvif->ap_sta_id);
2794
2795                 if (!WARN_ON(!mvmsta))
2796                         iwl_mvm_sta_modify_disable_tx(mvm, mvmsta, true);
2797
2798                 iwl_mvm_mac_ctxt_changed(mvm, vif, true, NULL);
2799                 break;
2800         default:
2801                 break;
2802         }
2803
2804         iwl_mvm_update_quotas(mvm, disabled_vif);
2805         iwl_mvm_binding_remove_vif(mvm, vif);
2806
2807 out:
2808         mvmvif->phy_ctxt = NULL;
2809         iwl_mvm_power_update_mac(mvm);
2810 }
2811
2812 static void iwl_mvm_unassign_vif_chanctx(struct ieee80211_hw *hw,
2813                                          struct ieee80211_vif *vif,
2814                                          struct ieee80211_chanctx_conf *ctx)
2815 {
2816         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2817
2818         mutex_lock(&mvm->mutex);
2819         __iwl_mvm_unassign_vif_chanctx(mvm, vif, ctx, false);
2820         mutex_unlock(&mvm->mutex);
2821 }
2822
2823 static int iwl_mvm_switch_vif_chanctx(struct ieee80211_hw *hw,
2824                                       struct ieee80211_vif_chanctx_switch *vifs,
2825                                       int n_vifs,
2826                                       enum ieee80211_chanctx_switch_mode mode)
2827 {
2828         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2829         int ret;
2830
2831         /* we only support SWAP_CONTEXTS and with a single-vif right now */
2832         if (mode != CHANCTX_SWMODE_SWAP_CONTEXTS || n_vifs > 1)
2833                 return -EOPNOTSUPP;
2834
2835         mutex_lock(&mvm->mutex);
2836         __iwl_mvm_unassign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx, true);
2837         __iwl_mvm_remove_chanctx(mvm, vifs[0].old_ctx);
2838
2839         ret = __iwl_mvm_add_chanctx(mvm, vifs[0].new_ctx);
2840         if (ret) {
2841                 IWL_ERR(mvm, "failed to add new_ctx during channel switch\n");
2842                 goto out_reassign;
2843         }
2844
2845         ret = __iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].new_ctx,
2846                                            true);
2847         if (ret) {
2848                 IWL_ERR(mvm,
2849                         "failed to assign new_ctx during channel switch\n");
2850                 goto out_remove;
2851         }
2852
2853         /* we don't support TDLS during DCM - can be caused by channel switch */
2854         if (iwl_mvm_phy_ctx_count(mvm) > 1)
2855                 iwl_mvm_teardown_tdls_peers(mvm);
2856
2857         goto out;
2858
2859 out_remove:
2860         __iwl_mvm_remove_chanctx(mvm, vifs[0].new_ctx);
2861
2862 out_reassign:
2863         ret = __iwl_mvm_add_chanctx(mvm, vifs[0].old_ctx);
2864         if (ret) {
2865                 IWL_ERR(mvm, "failed to add old_ctx back after failure.\n");
2866                 goto out_restart;
2867         }
2868
2869         ret = __iwl_mvm_assign_vif_chanctx(mvm, vifs[0].vif, vifs[0].old_ctx,
2870                                            true);
2871         if (ret) {
2872                 IWL_ERR(mvm, "failed to reassign old_ctx after failure.\n");
2873                 goto out_restart;
2874         }
2875
2876         goto out;
2877
2878 out_restart:
2879         /* things keep failing, better restart the hw */
2880         iwl_mvm_nic_restart(mvm, false);
2881
2882 out:
2883         mutex_unlock(&mvm->mutex);
2884         return ret;
2885 }
2886
2887 static int iwl_mvm_set_tim(struct ieee80211_hw *hw,
2888                            struct ieee80211_sta *sta,
2889                            bool set)
2890 {
2891         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2892         struct iwl_mvm_sta *mvm_sta = (void *)sta->drv_priv;
2893
2894         if (!mvm_sta || !mvm_sta->vif) {
2895                 IWL_ERR(mvm, "Station is not associated to a vif\n");
2896                 return -EINVAL;
2897         }
2898
2899         return iwl_mvm_mac_ctxt_beacon_changed(mvm, mvm_sta->vif);
2900 }
2901
2902 #ifdef CONFIG_NL80211_TESTMODE
2903 static const struct nla_policy iwl_mvm_tm_policy[IWL_MVM_TM_ATTR_MAX + 1] = {
2904         [IWL_MVM_TM_ATTR_CMD] = { .type = NLA_U32 },
2905         [IWL_MVM_TM_ATTR_NOA_DURATION] = { .type = NLA_U32 },
2906         [IWL_MVM_TM_ATTR_BEACON_FILTER_STATE] = { .type = NLA_U32 },
2907 };
2908
2909 static int __iwl_mvm_mac_testmode_cmd(struct iwl_mvm *mvm,
2910                                       struct ieee80211_vif *vif,
2911                                       void *data, int len)
2912 {
2913         struct nlattr *tb[IWL_MVM_TM_ATTR_MAX + 1];
2914         int err;
2915         u32 noa_duration;
2916
2917         err = nla_parse(tb, IWL_MVM_TM_ATTR_MAX, data, len, iwl_mvm_tm_policy);
2918         if (err)
2919                 return err;
2920
2921         if (!tb[IWL_MVM_TM_ATTR_CMD])
2922                 return -EINVAL;
2923
2924         switch (nla_get_u32(tb[IWL_MVM_TM_ATTR_CMD])) {
2925         case IWL_MVM_TM_CMD_SET_NOA:
2926                 if (!vif || vif->type != NL80211_IFTYPE_AP || !vif->p2p ||
2927                     !vif->bss_conf.enable_beacon ||
2928                     !tb[IWL_MVM_TM_ATTR_NOA_DURATION])
2929                         return -EINVAL;
2930
2931                 noa_duration = nla_get_u32(tb[IWL_MVM_TM_ATTR_NOA_DURATION]);
2932                 if (noa_duration >= vif->bss_conf.beacon_int)
2933                         return -EINVAL;
2934
2935                 mvm->noa_duration = noa_duration;
2936                 mvm->noa_vif = vif;
2937
2938                 return iwl_mvm_update_quotas(mvm, NULL);
2939         case IWL_MVM_TM_CMD_SET_BEACON_FILTER:
2940                 /* must be associated client vif - ignore authorized */
2941                 if (!vif || vif->type != NL80211_IFTYPE_STATION ||
2942                     !vif->bss_conf.assoc || !vif->bss_conf.dtim_period ||
2943                     !tb[IWL_MVM_TM_ATTR_BEACON_FILTER_STATE])
2944                         return -EINVAL;
2945
2946                 if (nla_get_u32(tb[IWL_MVM_TM_ATTR_BEACON_FILTER_STATE]))
2947                         return iwl_mvm_enable_beacon_filter(mvm, vif, 0);
2948                 return iwl_mvm_disable_beacon_filter(mvm, vif, 0);
2949         }
2950
2951         return -EOPNOTSUPP;
2952 }
2953
2954 static int iwl_mvm_mac_testmode_cmd(struct ieee80211_hw *hw,
2955                                     struct ieee80211_vif *vif,
2956                                     void *data, int len)
2957 {
2958         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2959         int err;
2960
2961         mutex_lock(&mvm->mutex);
2962         err = __iwl_mvm_mac_testmode_cmd(mvm, vif, data, len);
2963         mutex_unlock(&mvm->mutex);
2964
2965         return err;
2966 }
2967 #endif
2968
2969 static void iwl_mvm_channel_switch_beacon(struct ieee80211_hw *hw,
2970                                           struct ieee80211_vif *vif,
2971                                           struct cfg80211_chan_def *chandef)
2972 {
2973         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2974         struct ieee80211_vif *csa_vif;
2975
2976         mutex_lock(&mvm->mutex);
2977
2978         csa_vif = rcu_dereference_protected(mvm->csa_vif,
2979                                             lockdep_is_held(&mvm->mutex));
2980         if (WARN(csa_vif && csa_vif->csa_active,
2981                  "Another CSA is already in progress"))
2982                 goto out_unlock;
2983
2984         IWL_DEBUG_MAC80211(mvm, "CSA started to freq %d\n",
2985                            chandef->center_freq1);
2986         rcu_assign_pointer(mvm->csa_vif, vif);
2987
2988 out_unlock:
2989         mutex_unlock(&mvm->mutex);
2990 }
2991
2992 static void iwl_mvm_mac_flush(struct ieee80211_hw *hw,
2993                               struct ieee80211_vif *vif, u32 queues, bool drop)
2994 {
2995         struct iwl_mvm *mvm = IWL_MAC80211_GET_MVM(hw);
2996         struct iwl_mvm_vif *mvmvif;
2997         struct iwl_mvm_sta *mvmsta;
2998
2999         if (!vif || vif->type != NL80211_IFTYPE_STATION)
3000                 return;
3001
3002         mutex_lock(&mvm->mutex);
3003         mvmvif = iwl_mvm_vif_from_mac80211(vif);
3004         mvmsta = iwl_mvm_sta_from_staid_protected(mvm, mvmvif->ap_sta_id);
3005
3006         if (WARN_ON_ONCE(!mvmsta))
3007                 goto done;
3008
3009         if (drop) {
3010                 if (iwl_mvm_flush_tx_path(mvm, mvmsta->tfd_queue_msk, true))
3011                         IWL_ERR(mvm, "flush request fail\n");
3012         } else {
3013                 iwl_trans_wait_tx_queue_empty(mvm->trans,
3014                                               mvmsta->tfd_queue_msk);
3015         }
3016 done:
3017         mutex_unlock(&mvm->mutex);
3018 }
3019
3020 const struct ieee80211_ops iwl_mvm_hw_ops = {
3021         .tx = iwl_mvm_mac_tx,
3022         .ampdu_action = iwl_mvm_mac_ampdu_action,
3023         .start = iwl_mvm_mac_start,
3024         .restart_complete = iwl_mvm_mac_restart_complete,
3025         .stop = iwl_mvm_mac_stop,
3026         .add_interface = iwl_mvm_mac_add_interface,
3027         .remove_interface = iwl_mvm_mac_remove_interface,
3028         .config = iwl_mvm_mac_config,
3029         .prepare_multicast = iwl_mvm_prepare_multicast,
3030         .configure_filter = iwl_mvm_configure_filter,
3031         .bss_info_changed = iwl_mvm_bss_info_changed,
3032         .hw_scan = iwl_mvm_mac_hw_scan,
3033         .cancel_hw_scan = iwl_mvm_mac_cancel_hw_scan,
3034         .sta_pre_rcu_remove = iwl_mvm_sta_pre_rcu_remove,
3035         .sta_state = iwl_mvm_mac_sta_state,
3036         .sta_notify = iwl_mvm_mac_sta_notify,
3037         .allow_buffered_frames = iwl_mvm_mac_allow_buffered_frames,
3038         .release_buffered_frames = iwl_mvm_mac_release_buffered_frames,
3039         .set_rts_threshold = iwl_mvm_mac_set_rts_threshold,
3040         .sta_rc_update = iwl_mvm_sta_rc_update,
3041         .conf_tx = iwl_mvm_mac_conf_tx,
3042         .mgd_prepare_tx = iwl_mvm_mac_mgd_prepare_tx,
3043         .mgd_protect_tdls_discover = iwl_mvm_mac_mgd_protect_tdls_discover,
3044         .flush = iwl_mvm_mac_flush,
3045         .sched_scan_start = iwl_mvm_mac_sched_scan_start,
3046         .sched_scan_stop = iwl_mvm_mac_sched_scan_stop,
3047         .set_key = iwl_mvm_mac_set_key,
3048         .update_tkip_key = iwl_mvm_mac_update_tkip_key,
3049         .remain_on_channel = iwl_mvm_roc,
3050         .cancel_remain_on_channel = iwl_mvm_cancel_roc,
3051         .add_chanctx = iwl_mvm_add_chanctx,
3052         .remove_chanctx = iwl_mvm_remove_chanctx,
3053         .change_chanctx = iwl_mvm_change_chanctx,
3054         .assign_vif_chanctx = iwl_mvm_assign_vif_chanctx,
3055         .unassign_vif_chanctx = iwl_mvm_unassign_vif_chanctx,
3056         .switch_vif_chanctx = iwl_mvm_switch_vif_chanctx,
3057
3058         .start_ap = iwl_mvm_start_ap_ibss,
3059         .stop_ap = iwl_mvm_stop_ap_ibss,
3060         .join_ibss = iwl_mvm_start_ap_ibss,
3061         .leave_ibss = iwl_mvm_stop_ap_ibss,
3062
3063         .set_tim = iwl_mvm_set_tim,
3064
3065         .channel_switch_beacon = iwl_mvm_channel_switch_beacon,
3066
3067         CFG80211_TESTMODE_CMD(iwl_mvm_mac_testmode_cmd)
3068
3069 #ifdef CONFIG_PM_SLEEP
3070         /* look at d3.c */
3071         .suspend = iwl_mvm_suspend,
3072         .resume = iwl_mvm_resume,
3073         .set_wakeup = iwl_mvm_set_wakeup,
3074         .set_rekey_data = iwl_mvm_set_rekey_data,
3075 #if IS_ENABLED(CONFIG_IPV6)
3076         .ipv6_addr_change = iwl_mvm_ipv6_addr_change,
3077 #endif
3078         .set_default_unicast_key = iwl_mvm_set_default_unicast_key,
3079 #endif
3080 };