drivers/net/wireless/iwlwifi/mvm/scan.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
  66 #include <linux/etherdevice.h>
  67 #include <net/mac80211.h>
  68
  69 #include "mvm.h"
  70 #include "iwl-eeprom-parse.h"
  71 #include "fw-api-scan.h"
  72
  73 #define IWL_PLCP_QUIET_THRESH 1
  74 #define IWL_ACTIVE_QUIET_TIME 10
  75
  76 struct iwl_mvm_scan_params {
  77         u32 max_out_time;
  78         u32 suspend_time;
  79         bool passive_fragmented;
  80         struct _dwell {
  81                 u16 passive;
  82                 u16 active;
  83         } dwell[IEEE80211_NUM_BANDS];
  84 };
  85
  86 enum iwl_umac_scan_uid_type {
  87         IWL_UMAC_SCAN_UID_REG_SCAN      = BIT(0),
  88         IWL_UMAC_SCAN_UID_SCHED_SCAN    = BIT(1),
  89         IWL_UMAC_SCAN_UID_ALL           = IWL_UMAC_SCAN_UID_REG_SCAN |
  90                                           IWL_UMAC_SCAN_UID_SCHED_SCAN,
  91 };
  92
  93 static int iwl_umac_scan_stop(struct iwl_mvm *mvm,
  94                               enum iwl_umac_scan_uid_type type, bool notify);
  95
  96 static u8 iwl_mvm_scan_rx_ant(struct iwl_mvm *mvm)
  97 {
  98         if (mvm->scan_rx_ant != ANT_NONE)
  99                 return mvm->scan_rx_ant;
 100         return mvm->fw->valid_rx_ant;
 101 }
 102
 103 static inline __le16 iwl_mvm_scan_rx_chain(struct iwl_mvm *mvm)
 104 {
 105         u16 rx_chain;
 106         u8 rx_ant;
 107
 108         rx_ant = iwl_mvm_scan_rx_ant(mvm);
 109         rx_chain = rx_ant << PHY_RX_CHAIN_VALID_POS;
 110         rx_chain |= rx_ant << PHY_RX_CHAIN_FORCE_MIMO_SEL_POS;
 111         rx_chain |= rx_ant << PHY_RX_CHAIN_FORCE_SEL_POS;
 112         rx_chain |= 0x1 << PHY_RX_CHAIN_DRIVER_FORCE_POS;
 113         return cpu_to_le16(rx_chain);
 114 }
 115
 116 static __le32 iwl_mvm_scan_rxon_flags(enum ieee80211_band band)
 117 {
 118         if (band == IEEE80211_BAND_2GHZ)
 119                 return cpu_to_le32(PHY_BAND_24);
 120         else
 121                 return cpu_to_le32(PHY_BAND_5);
 122 }
 123
 124 static inline __le32
 125 iwl_mvm_scan_rate_n_flags(struct iwl_mvm *mvm, enum ieee80211_band band,
 126                           bool no_cck)
 127 {
 128         u32 tx_ant;
 129
 130         mvm->scan_last_antenna_idx =
 131                 iwl_mvm_next_antenna(mvm, mvm->fw->valid_tx_ant,
 132                                      mvm->scan_last_antenna_idx);
 133         tx_ant = BIT(mvm->scan_last_antenna_idx) << RATE_MCS_ANT_POS;
 134
 135         if (band == IEEE80211_BAND_2GHZ && !no_cck)
 136                 return cpu_to_le32(IWL_RATE_1M_PLCP | RATE_MCS_CCK_MSK |
 137                                    tx_ant);
 138         else
 139                 return cpu_to_le32(IWL_RATE_6M_PLCP | tx_ant);
 140 }
 141
 142 /*
 143  * We insert the SSIDs in an inverted order, because the FW will
 144  * invert it back. The most prioritized SSID, which is first in the
 145  * request list, is not copied here, but inserted directly to the probe
 146  * request.
 147  */
 148 static void iwl_mvm_scan_fill_ssids(struct iwl_ssid_ie *cmd_ssid,
 149                                     struct cfg80211_ssid *ssids,
 150                                     int n_ssids, int first)
 151 {
 152         int fw_idx, req_idx;
 153
 154         for (req_idx = n_ssids - 1, fw_idx = 0; req_idx >= first;
 155              req_idx--, fw_idx++) {
 156                 cmd_ssid[fw_idx].id = WLAN_EID_SSID;
 157                 cmd_ssid[fw_idx].len = ssids[req_idx].ssid_len;
 158                 memcpy(cmd_ssid[fw_idx].ssid,
 159                        ssids[req_idx].ssid,
 160                        ssids[req_idx].ssid_len);
 161         }
 162 }
 163
 164 /*
 165  * If req->n_ssids > 0, it means we should do an active scan.
 166  * In case of active scan w/o directed scan, we receive a zero-length SSID
 167  * just to notify that this scan is active and not passive.
 168  * In order to notify the FW of the number of SSIDs we wish to scan (including
 169  * the zero-length one), we need to set the corresponding bits in chan->type,
 170  * one for each SSID, and set the active bit (first). If the first SSID is
 171  * already included in the probe template, so we need to set only
 172  * req->n_ssids - 1 bits in addition to the first bit.
 173  */
 174 static u16 iwl_mvm_get_active_dwell(enum ieee80211_band band, int n_ssids)
 175 {
 176         if (band == IEEE80211_BAND_2GHZ)
 177                 return 20  + 3 * (n_ssids + 1);
 178         return 10  + 2 * (n_ssids + 1);
 179 }
 180
 181 static u16 iwl_mvm_get_passive_dwell(enum ieee80211_band band)
 182 {
 183         return band == IEEE80211_BAND_2GHZ ? 100 + 20 : 100 + 10;
 184 }
 185
 186 static void iwl_mvm_scan_fill_channels(struct iwl_scan_cmd *cmd,
 187                                        struct cfg80211_scan_request *req,
 188                                        bool basic_ssid,
 189                                        struct iwl_mvm_scan_params *params)
 190 {
 191         struct iwl_scan_channel *chan = (struct iwl_scan_channel *)
 192                 (cmd->data + le16_to_cpu(cmd->tx_cmd.len));
 193         int i;
 194         int type = BIT(req->n_ssids) - 1;
 195         enum ieee80211_band band = req->channels[0]->band;
 196
 197         if (!basic_ssid)
 198                 type |= BIT(req->n_ssids);
 199
 200         for (i = 0; i < cmd->channel_count; i++) {
 201                 chan->channel = cpu_to_le16(req->channels[i]->hw_value);
 202                 chan->type = cpu_to_le32(type);
 203                 if (req->channels[i]->flags & IEEE80211_CHAN_NO_IR)
 204                         chan->type &= cpu_to_le32(~SCAN_CHANNEL_TYPE_ACTIVE);
 205                 chan->active_dwell = cpu_to_le16(params->dwell[band].active);
 206                 chan->passive_dwell = cpu_to_le16(params->dwell[band].passive);
 207                 chan->iteration_count = cpu_to_le16(1);
 208                 chan++;
 209         }
 210 }
 211
 212 /*
 213  * Fill in probe request with the following parameters:
 214  * TA is our vif HW address, which mac80211 ensures we have.
 215  * Packet is broadcasted, so this is both SA and DA.
 216  * The probe request IE is made out of two: first comes the most prioritized
 217  * SSID if a directed scan is requested. Second comes whatever extra
 218  * information was given to us as the scan request IE.
 219  */
 220 static u16 iwl_mvm_fill_probe_req(struct ieee80211_mgmt *frame, const u8 *ta,
 221                                   int n_ssids, const u8 *ssid, int ssid_len,
 222                                   const u8 *band_ie, int band_ie_len,
 223                                   const u8 *common_ie, int common_ie_len,
 224                                   int left)
 225 {
 226         int len = 0;
 227         u8 *pos = NULL;
 228
 229         /* Make sure there is enough space for the probe request,
 230          * two mandatory IEs and the data */
 231         left -= 24;
 232         if (left < 0)
 233                 return 0;
 234
 235         frame->frame_control = cpu_to_le16(IEEE80211_STYPE_PROBE_REQ);
 236         eth_broadcast_addr(frame->da);
 237         memcpy(frame->sa, ta, ETH_ALEN);
 238         eth_broadcast_addr(frame->bssid);
 239         frame->seq_ctrl = 0;
 240
 241         len += 24;
 242
 243         /* for passive scans, no need to fill anything */
 244         if (n_ssids == 0)
 245                 return (u16)len;
 246
 247         /* points to the payload of the request */
 248         pos = &frame->u.probe_req.variable[0];
 249
 250         /* fill in our SSID IE */
 251         left -= ssid_len + 2;
 252         if (left < 0)
 253                 return 0;
 254         *pos++ = WLAN_EID_SSID;
 255         *pos++ = ssid_len;
 256         if (ssid && ssid_len) { /* ssid_len may be == 0 even if ssid is valid */
 257                 memcpy(pos, ssid, ssid_len);
 258                 pos += ssid_len;
 259         }
 260
 261         len += ssid_len + 2;
 262
 263         if (WARN_ON(left < band_ie_len + common_ie_len))
 264                 return len;
 265
 266         if (band_ie && band_ie_len) {
 267                 memcpy(pos, band_ie, band_ie_len);
 268                 pos += band_ie_len;
 269                 len += band_ie_len;
 270         }
 271
 272         if (common_ie && common_ie_len) {
 273                 memcpy(pos, common_ie, common_ie_len);
 274                 pos += common_ie_len;
 275                 len += common_ie_len;
 276         }
 277
 278         return (u16)len;
 279 }
 280
 281 static void iwl_mvm_scan_condition_iterator(void *data, u8 *mac,
 282                                             struct ieee80211_vif *vif)
 283 {
 284         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
 285         bool *global_bound = data;
 286
 287         if (vif->type != NL80211_IFTYPE_P2P_DEVICE && mvmvif->phy_ctxt &&
 288             mvmvif->phy_ctxt->id < MAX_PHYS)
 289                 *global_bound = true;
 290 }
 291
 292 static void iwl_mvm_scan_calc_params(struct iwl_mvm *mvm,
 293                                      struct ieee80211_vif *vif,
 294                                      int n_ssids, u32 flags,
 295                                      struct iwl_mvm_scan_params *params)
 296 {
 297         bool global_bound = false;
 298         enum ieee80211_band band;
 299         u8 frag_passive_dwell = 0;
 300
 301         ieee80211_iterate_active_interfaces_atomic(mvm->hw,
 302                                             IEEE80211_IFACE_ITER_NORMAL,
 303                                             iwl_mvm_scan_condition_iterator,
 304                                             &global_bound);
 305
 306         if (!global_bound)
 307                 goto not_bound;
 308
 309         params->suspend_time = 30;
 310         params->max_out_time = 170;
 311
 312         if (iwl_mvm_low_latency(mvm)) {
 313                 if (mvm->fw->ucode_capa.api[0] &
 314                     IWL_UCODE_TLV_API_FRAGMENTED_SCAN) {
 315                         params->suspend_time = 105;
 316                         params->max_out_time = 70;
 317                         frag_passive_dwell = 20;
 318                 } else {
 319                         params->suspend_time = 120;
 320                         params->max_out_time = 120;
 321                 }
 322         }
 323
 324         if (frag_passive_dwell && (mvm->fw->ucode_capa.api[0] &
 325                                    IWL_UCODE_TLV_API_FRAGMENTED_SCAN)) {
 326                 /*
 327                  * P2P device scan should not be fragmented to avoid negative
 328                  * impact on P2P device discovery. Configure max_out_time to be
 329                  * equal to dwell time on passive channel. Take a longest
 330                  * possible value, one that corresponds to 2GHz band
 331                  */
 332                 if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
 333                         u32 passive_dwell =
 334                                 iwl_mvm_get_passive_dwell(IEEE80211_BAND_2GHZ);
 335                         params->max_out_time = passive_dwell;
 336                 } else {
 337                         params->passive_fragmented = true;
 338                 }
 339         }
 340
 341         if (flags & NL80211_SCAN_FLAG_LOW_PRIORITY)
 342                 params->max_out_time = 200;
 343
 344 not_bound:
 345
 346         for (band = IEEE80211_BAND_2GHZ; band < IEEE80211_NUM_BANDS; band++) {
 347                 if (params->passive_fragmented)
 348                         params->dwell[band].passive = frag_passive_dwell;
 349                 else
 350                         params->dwell[band].passive =
 351                                 iwl_mvm_get_passive_dwell(band);
 352                 params->dwell[band].active = iwl_mvm_get_active_dwell(band,
 353                                                                       n_ssids);
 354         }
 355 }
 356
 357 static inline bool iwl_mvm_rrm_scan_needed(struct iwl_mvm *mvm)
 358 {
 359         /* require rrm scan whenever the fw supports it */
 360         return mvm->fw->ucode_capa.capa[0] &
 361                IWL_UCODE_TLV_CAPA_DS_PARAM_SET_IE_SUPPORT;
 362 }
 363
 364 static int iwl_mvm_max_scan_ie_fw_cmd_room(struct iwl_mvm *mvm,
 365                                            bool is_sched_scan)
 366 {
 367         int max_probe_len;
 368
 369         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)
 370                 max_probe_len = SCAN_OFFLOAD_PROBE_REQ_SIZE;
 371         else
 372                 max_probe_len = mvm->fw->ucode_capa.max_probe_length;
 373
 374         /* we create the 802.11 header and SSID element */
 375         max_probe_len -= 24 + 2;
 376
 377         /* basic ssid is added only for hw_scan with and old api */
 378         if (!(mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_NO_BASIC_SSID) &&
 379             !(mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN) &&
 380             !is_sched_scan)
 381                 max_probe_len -= 32;
 382
 383         /* DS parameter set element is added on 2.4GHZ band if required */
 384         if (iwl_mvm_rrm_scan_needed(mvm))
 385                 max_probe_len -= 3;
 386
 387         return max_probe_len;
 388 }
 389
 390 int iwl_mvm_max_scan_ie_len(struct iwl_mvm *mvm, bool is_sched_scan)
 391 {
 392         int max_ie_len = iwl_mvm_max_scan_ie_fw_cmd_room(mvm, is_sched_scan);
 393
 394         if (!(mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN))
 395                 return max_ie_len;
 396
 397         /* TODO: [BUG] This function should return the maximum allowed size of
 398          * scan IEs, however the LMAC scan api contains both 2GHZ and 5GHZ IEs
 399          * in the same command. So the correct implementation of this function
 400          * is just iwl_mvm_max_scan_ie_fw_cmd_room() / 2. Currently the scan
 401          * command has only 512 bytes and it would leave us with about 240
 402          * bytes for scan IEs, which is clearly not enough. So meanwhile
 403          * we will report an incorrect value. This may result in a failure to
 404          * issue a scan in unified_scan_lmac and unified_sched_scan_lmac
 405          * functions with -ENOBUFS, if a large enough probe will be provided.
 406          */
 407         return max_ie_len;
 408 }
 409
 410 int iwl_mvm_scan_request(struct iwl_mvm *mvm,
 411                          struct ieee80211_vif *vif,
 412                          struct cfg80211_scan_request *req)
 413 {
 414         struct iwl_host_cmd hcmd = {
 415                 .id = SCAN_REQUEST_CMD,
 416                 .len = { 0, },
 417                 .data = { mvm->scan_cmd, },
 418                 .dataflags = { IWL_HCMD_DFL_NOCOPY, },
 419         };
 420         struct iwl_scan_cmd *cmd = mvm->scan_cmd;
 421         int ret;
 422         u32 status;
 423         int ssid_len = 0;
 424         u8 *ssid = NULL;
 425         bool basic_ssid = !(mvm->fw->ucode_capa.flags &
 426                            IWL_UCODE_TLV_FLAGS_NO_BASIC_SSID);
 427         struct iwl_mvm_scan_params params = {};
 428
 429         lockdep_assert_held(&mvm->mutex);
 430
 431         /* we should have failed registration if scan_cmd was NULL */
 432         if (WARN_ON(mvm->scan_cmd == NULL))
 433                 return -ENOMEM;
 434
 435         IWL_DEBUG_SCAN(mvm, "Handling mac80211 scan request\n");
 436         mvm->scan_status = IWL_MVM_SCAN_OS;
 437         memset(cmd, 0, ksize(cmd));
 438
 439         cmd->channel_count = (u8)req->n_channels;
 440         cmd->quiet_time = cpu_to_le16(IWL_ACTIVE_QUIET_TIME);
 441         cmd->quiet_plcp_th = cpu_to_le16(IWL_PLCP_QUIET_THRESH);
 442         cmd->rxchain_sel_flags = iwl_mvm_scan_rx_chain(mvm);
 443
 444         iwl_mvm_scan_calc_params(mvm, vif, req->n_ssids, req->flags, &params);
 445         cmd->max_out_time = cpu_to_le32(params.max_out_time);
 446         cmd->suspend_time = cpu_to_le32(params.suspend_time);
 447         if (params.passive_fragmented)
 448                 cmd->scan_flags |= SCAN_FLAGS_FRAGMENTED_SCAN;
 449
 450         cmd->rxon_flags = iwl_mvm_scan_rxon_flags(req->channels[0]->band);
 451         cmd->filter_flags = cpu_to_le32(MAC_FILTER_ACCEPT_GRP |
 452                                         MAC_FILTER_IN_BEACON);
 453
 454         if (vif->type == NL80211_IFTYPE_P2P_DEVICE)
 455                 cmd->type = cpu_to_le32(SCAN_TYPE_DISCOVERY_FORCED);
 456         else
 457                 cmd->type = cpu_to_le32(SCAN_TYPE_FORCED);
 458
 459         cmd->repeats = cpu_to_le32(1);
 460
 461         /*
 462          * If the user asked for passive scan, don't change to active scan if
 463          * you see any activity on the channel - remain passive.
 464          */
 465         if (req->n_ssids > 0) {
 466                 cmd->passive2active = cpu_to_le16(1);
 467                 cmd->scan_flags |= SCAN_FLAGS_PASSIVE2ACTIVE;
 468                 if (basic_ssid) {
 469                         ssid = req->ssids[0].ssid;
 470                         ssid_len = req->ssids[0].ssid_len;
 471                 }
 472         } else {
 473                 cmd->passive2active = 0;
 474                 cmd->scan_flags &= ~SCAN_FLAGS_PASSIVE2ACTIVE;
 475         }
 476
 477         iwl_mvm_scan_fill_ssids(cmd->direct_scan, req->ssids, req->n_ssids,
 478                                 basic_ssid ? 1 : 0);
 479
 480         cmd->tx_cmd.tx_flags = cpu_to_le32(TX_CMD_FLG_SEQ_CTL |
 481                                            3 << TX_CMD_FLG_BT_PRIO_POS);
 482
 483         cmd->tx_cmd.sta_id = mvm->aux_sta.sta_id;
 484         cmd->tx_cmd.life_time = cpu_to_le32(TX_CMD_LIFE_TIME_INFINITE);
 485         cmd->tx_cmd.rate_n_flags =
 486                         iwl_mvm_scan_rate_n_flags(mvm, req->channels[0]->band,
 487                                                   req->no_cck);
 488
 489         cmd->tx_cmd.len =
 490                 cpu_to_le16(iwl_mvm_fill_probe_req(
 491                             (struct ieee80211_mgmt *)cmd->data,
 492                             vif->addr,
 493                             req->n_ssids, ssid, ssid_len,
 494                             req->ie, req->ie_len, NULL, 0,
 495                             mvm->fw->ucode_capa.max_probe_length));
 496
 497         iwl_mvm_scan_fill_channels(cmd, req, basic_ssid, &params);
 498
 499         cmd->len = cpu_to_le16(sizeof(struct iwl_scan_cmd) +
 500                 le16_to_cpu(cmd->tx_cmd.len) +
 501                 (cmd->channel_count * sizeof(struct iwl_scan_channel)));
 502         hcmd.len[0] = le16_to_cpu(cmd->len);
 503
 504         status = SCAN_RESPONSE_OK;
 505         ret = iwl_mvm_send_cmd_status(mvm, &hcmd, &status);
 506         if (!ret && status == SCAN_RESPONSE_OK) {
 507                 IWL_DEBUG_SCAN(mvm, "Scan request was sent successfully\n");
 508         } else {
 509                 /*
 510                  * If the scan failed, it usually means that the FW was unable
 511                  * to allocate the time events. Warn on it, but maybe we
 512                  * should try to send the command again with different params.
 513                  */
 514                 IWL_ERR(mvm, "Scan failed! status 0x%x ret %d\n",
 515                         status, ret);
 516                 mvm->scan_status = IWL_MVM_SCAN_NONE;
 517                 ret = -EIO;
 518         }
 519         return ret;
 520 }
 521
 522 int iwl_mvm_rx_scan_response(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb,
 523                           struct iwl_device_cmd *cmd)
 524 {
 525         struct iwl_rx_packet *pkt = rxb_addr(rxb);
 526         struct iwl_cmd_response *resp = (void *)pkt->data;
 527
 528         IWL_DEBUG_SCAN(mvm, "Scan response received. status 0x%x\n",
 529                        le32_to_cpu(resp->status));
 530         return 0;
 531 }
 532
 533 int iwl_mvm_rx_scan_complete(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb,
 534                           struct iwl_device_cmd *cmd)
 535 {
 536         struct iwl_rx_packet *pkt = rxb_addr(rxb);
 537         struct iwl_scan_complete_notif *notif = (void *)pkt->data;
 538
 539         lockdep_assert_held(&mvm->mutex);
 540
 541         IWL_DEBUG_SCAN(mvm, "Scan complete: status=0x%x scanned channels=%d\n",
 542                        notif->status, notif->scanned_channels);
 543
 544         if (mvm->scan_status == IWL_MVM_SCAN_OS)
 545                 mvm->scan_status = IWL_MVM_SCAN_NONE;
 546         ieee80211_scan_completed(mvm->hw, notif->status != SCAN_COMP_STATUS_OK);
 547
 548         iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
 549
 550         return 0;
 551 }
 552
 553 int iwl_mvm_rx_scan_offload_results(struct iwl_mvm *mvm,
 554                                     struct iwl_rx_cmd_buffer *rxb,
 555                                     struct iwl_device_cmd *cmd)
 556 {
 557         struct iwl_rx_packet *pkt = rxb_addr(rxb);
 558
 559         if (!(mvm->fw->ucode_capa.capa[0] & IWL_UCODE_TLV_CAPA_UMAC_SCAN) &&
 560             !(mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)) {
 561                 struct iwl_sched_scan_results *notif = (void *)pkt->data;
 562
 563                 if (!(notif->client_bitmap & SCAN_CLIENT_SCHED_SCAN))
 564                         return 0;
 565         }
 566
 567         IWL_DEBUG_SCAN(mvm, "Scheduled scan results\n");
 568         ieee80211_sched_scan_results(mvm->hw);
 569
 570         return 0;
 571 }
 572
 573 static bool iwl_mvm_scan_abort_notif(struct iwl_notif_wait_data *notif_wait,
 574                                      struct iwl_rx_packet *pkt, void *data)
 575 {
 576         struct iwl_mvm *mvm =
 577                 container_of(notif_wait, struct iwl_mvm, notif_wait);
 578         struct iwl_scan_complete_notif *notif;
 579         u32 *resp;
 580
 581         switch (pkt->hdr.cmd) {
 582         case SCAN_ABORT_CMD:
 583                 resp = (void *)pkt->data;
 584                 if (*resp == CAN_ABORT_STATUS) {
 585                         IWL_DEBUG_SCAN(mvm,
 586                                        "Scan can be aborted, wait until completion\n");
 587                         return false;
 588                 }
 589
 590                 /*
 591                  * If scan cannot be aborted, it means that we had a
 592                  * SCAN_COMPLETE_NOTIFICATION in the pipe and it called
 593                  * ieee80211_scan_completed already.
 594                  */
 595                 IWL_DEBUG_SCAN(mvm, "Scan cannot be aborted, exit now: %d\n",
 596                                *resp);
 597                 return true;
 598
 599         case SCAN_COMPLETE_NOTIFICATION:
 600                 notif = (void *)pkt->data;
 601                 IWL_DEBUG_SCAN(mvm, "Scan aborted: status 0x%x\n",
 602                                notif->status);
 603                 return true;
 604
 605         default:
 606                 WARN_ON(1);
 607                 return false;
 608         };
 609 }
 610
 611 static int iwl_mvm_cancel_regular_scan(struct iwl_mvm *mvm)
 612 {
 613         struct iwl_notification_wait wait_scan_abort;
 614         static const u8 scan_abort_notif[] = { SCAN_ABORT_CMD,
 615                                                SCAN_COMPLETE_NOTIFICATION };
 616         int ret;
 617
 618         iwl_init_notification_wait(&mvm->notif_wait, &wait_scan_abort,
 619                                    scan_abort_notif,
 620                                    ARRAY_SIZE(scan_abort_notif),
 621                                    iwl_mvm_scan_abort_notif, NULL);
 622
 623         ret = iwl_mvm_send_cmd_pdu(mvm, SCAN_ABORT_CMD, 0, 0, NULL);
 624         if (ret) {
 625                 IWL_ERR(mvm, "Couldn't send SCAN_ABORT_CMD: %d\n", ret);
 626                 /* mac80211's state will be cleaned in the nic_restart flow */
 627                 goto out_remove_notif;
 628         }
 629
 630         return iwl_wait_notification(&mvm->notif_wait, &wait_scan_abort, HZ);
 631
 632 out_remove_notif:
 633         iwl_remove_notification(&mvm->notif_wait, &wait_scan_abort);
 634         return ret;
 635 }
 636
 637 int iwl_mvm_rx_scan_offload_complete_notif(struct iwl_mvm *mvm,
 638                                            struct iwl_rx_cmd_buffer *rxb,
 639                                            struct iwl_device_cmd *cmd)
 640 {
 641         struct iwl_rx_packet *pkt = rxb_addr(rxb);
 642         u8 status, ebs_status;
 643
 644         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN) {
 645                 struct iwl_periodic_scan_complete *scan_notif;
 646
 647                 scan_notif = (void *)pkt->data;
 648                 status = scan_notif->status;
 649                 ebs_status = scan_notif->ebs_status;
 650         } else  {
 651                 struct iwl_scan_offload_complete *scan_notif;
 652
 653                 scan_notif = (void *)pkt->data;
 654                 status = scan_notif->status;
 655                 ebs_status = scan_notif->ebs_status;
 656         }
 657         /* scan status must be locked for proper checking */
 658         lockdep_assert_held(&mvm->mutex);
 659
 660         IWL_DEBUG_SCAN(mvm,
 661                        "%s completed, status %s, EBS status %s\n",
 662                        mvm->scan_status == IWL_MVM_SCAN_SCHED ?
 663                                 "Scheduled scan" : "Scan",
 664                        status == IWL_SCAN_OFFLOAD_COMPLETED ?
 665                                 "completed" : "aborted",
 666                        ebs_status == IWL_SCAN_EBS_SUCCESS ?
 667                                 "success" : "failed");
 668
 669
 670         /* only call mac80211 completion if the stop was initiated by FW */
 671         if (mvm->scan_status == IWL_MVM_SCAN_SCHED) {
 672                 mvm->scan_status = IWL_MVM_SCAN_NONE;
 673                 ieee80211_sched_scan_stopped(mvm->hw);
 674         } else if (mvm->scan_status == IWL_MVM_SCAN_OS) {
 675                 mvm->scan_status = IWL_MVM_SCAN_NONE;
 676                 ieee80211_scan_completed(mvm->hw,
 677                                          status == IWL_SCAN_OFFLOAD_ABORTED);
 678                 iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
 679         }
 680
 681         mvm->last_ebs_successful = !ebs_status;
 682
 683         return 0;
 684 }
 685
 686 static void iwl_scan_offload_build_tx_cmd(struct iwl_mvm *mvm,
 687                                           struct ieee80211_vif *vif,
 688                                           struct ieee80211_scan_ies *ies,
 689                                           enum ieee80211_band band,
 690                                           struct iwl_tx_cmd *cmd,
 691                                           u8 *data)
 692 {
 693         u16 cmd_len;
 694
 695         cmd->tx_flags = cpu_to_le32(TX_CMD_FLG_SEQ_CTL);
 696         cmd->life_time = cpu_to_le32(TX_CMD_LIFE_TIME_INFINITE);
 697         cmd->sta_id = mvm->aux_sta.sta_id;
 698
 699         cmd->rate_n_flags = iwl_mvm_scan_rate_n_flags(mvm, band, false);
 700
 701         cmd_len = iwl_mvm_fill_probe_req((struct ieee80211_mgmt *)data,
 702                                          vif->addr,
 703                                          1, NULL, 0,
 704                                          ies->ies[band], ies->len[band],
 705                                          ies->common_ies, ies->common_ie_len,
 706                                          SCAN_OFFLOAD_PROBE_REQ_SIZE);
 707         cmd->len = cpu_to_le16(cmd_len);
 708 }
 709
 710 static void iwl_build_scan_cmd(struct iwl_mvm *mvm,
 711                                struct ieee80211_vif *vif,
 712                                struct cfg80211_sched_scan_request *req,
 713                                struct iwl_scan_offload_cmd *scan,
 714                                struct iwl_mvm_scan_params *params)
 715 {
 716         scan->channel_count = req->n_channels;
 717         scan->quiet_time = cpu_to_le16(IWL_ACTIVE_QUIET_TIME);
 718         scan->quiet_plcp_th = cpu_to_le16(IWL_PLCP_QUIET_THRESH);
 719         scan->good_CRC_th = IWL_GOOD_CRC_TH_DEFAULT;
 720         scan->rx_chain = iwl_mvm_scan_rx_chain(mvm);
 721
 722         scan->max_out_time = cpu_to_le32(params->max_out_time);
 723         scan->suspend_time = cpu_to_le32(params->suspend_time);
 724
 725         scan->filter_flags |= cpu_to_le32(MAC_FILTER_ACCEPT_GRP |
 726                                           MAC_FILTER_IN_BEACON);
 727         scan->scan_type = cpu_to_le32(SCAN_TYPE_BACKGROUND);
 728         scan->rep_count = cpu_to_le32(1);
 729
 730         if (params->passive_fragmented)
 731                 scan->scan_flags |= SCAN_FLAGS_FRAGMENTED_SCAN;
 732 }
 733
 734 static int iwl_ssid_exist(u8 *ssid, u8 ssid_len, struct iwl_ssid_ie *ssid_list)
 735 {
 736         int i;
 737
 738         for (i = 0; i < PROBE_OPTION_MAX; i++) {
 739                 if (!ssid_list[i].len)
 740                         break;
 741                 if (ssid_list[i].len == ssid_len &&
 742                     !memcmp(ssid_list->ssid, ssid, ssid_len))
 743                         return i;
 744         }
 745         return -1;
 746 }
 747
 748 static void iwl_scan_offload_build_ssid(struct cfg80211_sched_scan_request *req,
 749                                         struct iwl_ssid_ie *direct_scan,
 750                                         u32 *ssid_bitmap, bool basic_ssid)
 751 {
 752         int i, j;
 753         int index;
 754
 755         /*
 756          * copy SSIDs from match list.
 757          * iwl_config_sched_scan_profiles() uses the order of these ssids to
 758          * config match list.
 759          */
 760         for (i = 0; i < req->n_match_sets && i < PROBE_OPTION_MAX; i++) {
 761                 /* skip empty SSID matchsets */
 762                 if (!req->match_sets[i].ssid.ssid_len)
 763                         continue;
 764                 direct_scan[i].id = WLAN_EID_SSID;
 765                 direct_scan[i].len = req->match_sets[i].ssid.ssid_len;
 766                 memcpy(direct_scan[i].ssid, req->match_sets[i].ssid.ssid,
 767                        direct_scan[i].len);
 768         }
 769
 770         /* add SSIDs from scan SSID list */
 771         *ssid_bitmap = 0;
 772         for (j = 0; j < req->n_ssids && i < PROBE_OPTION_MAX; j++) {
 773                 index = iwl_ssid_exist(req->ssids[j].ssid,
 774                                        req->ssids[j].ssid_len,
 775                                        direct_scan);
 776                 if (index < 0) {
 777                         if (!req->ssids[j].ssid_len && basic_ssid)
 778                                 continue;
 779                         direct_scan[i].id = WLAN_EID_SSID;
 780                         direct_scan[i].len = req->ssids[j].ssid_len;
 781                         memcpy(direct_scan[i].ssid, req->ssids[j].ssid,
 782                                direct_scan[i].len);
 783                         *ssid_bitmap |= BIT(i + 1);
 784                         i++;
 785                 } else {
 786                         *ssid_bitmap |= BIT(index + 1);
 787                 }
 788         }
 789 }
 790
 791 static void iwl_build_channel_cfg(struct iwl_mvm *mvm,
 792                                   struct cfg80211_sched_scan_request *req,
 793                                   u8 *channels_buffer,
 794                                   enum ieee80211_band band,
 795                                   int *head,
 796                                   u32 ssid_bitmap,
 797                                   struct iwl_mvm_scan_params *params)
 798 {
 799         u32 n_channels = mvm->fw->ucode_capa.n_scan_channels;
 800         __le32 *type = (__le32 *)channels_buffer;
 801         __le16 *channel_number = (__le16 *)(type + n_channels);
 802         __le16 *iter_count = channel_number + n_channels;
 803         __le32 *iter_interval = (__le32 *)(iter_count + n_channels);
 804         u8 *active_dwell = (u8 *)(iter_interval + n_channels);
 805         u8 *passive_dwell = active_dwell + n_channels;
 806         int i, index = 0;
 807
 808         for (i = 0; i < req->n_channels; i++) {
 809                 struct ieee80211_channel *chan = req->channels[i];
 810
 811                 if (chan->band != band)
 812                         continue;
 813
 814                 index = *head;
 815                 (*head)++;
 816
 817                 channel_number[index] = cpu_to_le16(chan->hw_value);
 818                 active_dwell[index] = params->dwell[band].active;
 819                 passive_dwell[index] = params->dwell[band].passive;
 820
 821                 iter_count[index] = cpu_to_le16(1);
 822                 iter_interval[index] = 0;
 823
 824                 if (!(chan->flags & IEEE80211_CHAN_NO_IR))
 825                         type[index] |=
 826                                 cpu_to_le32(IWL_SCAN_OFFLOAD_CHANNEL_ACTIVE);
 827
 828                 type[index] |= cpu_to_le32(IWL_SCAN_OFFLOAD_CHANNEL_FULL |
 829                                            IWL_SCAN_OFFLOAD_CHANNEL_PARTIAL);
 830
 831                 if (chan->flags & IEEE80211_CHAN_NO_HT40)
 832                         type[index] |=
 833                                 cpu_to_le32(IWL_SCAN_OFFLOAD_CHANNEL_NARROW);
 834
 835                 /* scan for all SSIDs from req->ssids */
 836                 type[index] |= cpu_to_le32(ssid_bitmap);
 837         }
 838 }
 839
 840 int iwl_mvm_config_sched_scan(struct iwl_mvm *mvm,
 841                               struct ieee80211_vif *vif,
 842                               struct cfg80211_sched_scan_request *req,
 843                               struct ieee80211_scan_ies *ies)
 844 {
 845         int band_2ghz = mvm->nvm_data->bands[IEEE80211_BAND_2GHZ].n_channels;
 846         int band_5ghz = mvm->nvm_data->bands[IEEE80211_BAND_5GHZ].n_channels;
 847         int head = 0;
 848         u32 ssid_bitmap;
 849         int cmd_len;
 850         int ret;
 851         u8 *probes;
 852         bool basic_ssid = !(mvm->fw->ucode_capa.flags &
 853                             IWL_UCODE_TLV_FLAGS_NO_BASIC_SSID);
 854
 855         struct iwl_scan_offload_cfg *scan_cfg;
 856         struct iwl_host_cmd cmd = {
 857                 .id = SCAN_OFFLOAD_CONFIG_CMD,
 858         };
 859         struct iwl_mvm_scan_params params = {};
 860
 861         lockdep_assert_held(&mvm->mutex);
 862
 863         cmd_len = sizeof(struct iwl_scan_offload_cfg) +
 864                   mvm->fw->ucode_capa.n_scan_channels * IWL_SCAN_CHAN_SIZE +
 865                   2 * SCAN_OFFLOAD_PROBE_REQ_SIZE;
 866
 867         scan_cfg = kzalloc(cmd_len, GFP_KERNEL);
 868         if (!scan_cfg)
 869                 return -ENOMEM;
 870
 871         probes = scan_cfg->data +
 872                 mvm->fw->ucode_capa.n_scan_channels * IWL_SCAN_CHAN_SIZE;
 873
 874         iwl_mvm_scan_calc_params(mvm, vif, req->n_ssids, 0, &params);
 875         iwl_build_scan_cmd(mvm, vif, req, &scan_cfg->scan_cmd, &params);
 876         scan_cfg->scan_cmd.len = cpu_to_le16(cmd_len);
 877
 878         iwl_scan_offload_build_ssid(req, scan_cfg->scan_cmd.direct_scan,
 879                                     &ssid_bitmap, basic_ssid);
 880         /* build tx frames for supported bands */
 881         if (band_2ghz) {
 882                 iwl_scan_offload_build_tx_cmd(mvm, vif, ies,
 883                                               IEEE80211_BAND_2GHZ,
 884                                               &scan_cfg->scan_cmd.tx_cmd[0],
 885                                               probes);
 886                 iwl_build_channel_cfg(mvm, req, scan_cfg->data,
 887                                       IEEE80211_BAND_2GHZ, &head,
 888                                       ssid_bitmap, &params);
 889         }
 890         if (band_5ghz) {
 891                 iwl_scan_offload_build_tx_cmd(mvm, vif, ies,
 892                                               IEEE80211_BAND_5GHZ,
 893                                               &scan_cfg->scan_cmd.tx_cmd[1],
 894                                               probes +
 895                                                 SCAN_OFFLOAD_PROBE_REQ_SIZE);
 896                 iwl_build_channel_cfg(mvm, req, scan_cfg->data,
 897                                       IEEE80211_BAND_5GHZ, &head,
 898                                       ssid_bitmap, &params);
 899         }
 900
 901         cmd.data[0] = scan_cfg;
 902         cmd.len[0] = cmd_len;
 903         cmd.dataflags[0] = IWL_HCMD_DFL_NOCOPY;
 904
 905         IWL_DEBUG_SCAN(mvm, "Sending scheduled scan config\n");
 906
 907         ret = iwl_mvm_send_cmd(mvm, &cmd);
 908         kfree(scan_cfg);
 909         return ret;
 910 }
 911
 912 int iwl_mvm_config_sched_scan_profiles(struct iwl_mvm *mvm,
 913                                        struct cfg80211_sched_scan_request *req)
 914 {
 915         struct iwl_scan_offload_profile *profile;
 916         struct iwl_scan_offload_profile_cfg *profile_cfg;
 917         struct iwl_scan_offload_blacklist *blacklist;
 918         struct iwl_host_cmd cmd = {
 919                 .id = SCAN_OFFLOAD_UPDATE_PROFILES_CMD,
 920                 .len[1] = sizeof(*profile_cfg),
 921                 .dataflags[0] = IWL_HCMD_DFL_NOCOPY,
 922                 .dataflags[1] = IWL_HCMD_DFL_NOCOPY,
 923         };
 924         int blacklist_len;
 925         int i;
 926         int ret;
 927
 928         if (WARN_ON(req->n_match_sets > IWL_SCAN_MAX_PROFILES))
 929                         return -EIO;
 930
 931         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_SHORT_BL)
 932                 blacklist_len = IWL_SCAN_SHORT_BLACKLIST_LEN;
 933         else
 934                 blacklist_len = IWL_SCAN_MAX_BLACKLIST_LEN;
 935
 936         blacklist = kzalloc(sizeof(*blacklist) * blacklist_len, GFP_KERNEL);
 937         if (!blacklist)
 938                 return -ENOMEM;
 939
 940         profile_cfg = kzalloc(sizeof(*profile_cfg), GFP_KERNEL);
 941         if (!profile_cfg) {
 942                 ret = -ENOMEM;
 943                 goto free_blacklist;
 944         }
 945
 946         cmd.data[0] = blacklist;
 947         cmd.len[0] = sizeof(*blacklist) * blacklist_len;
 948         cmd.data[1] = profile_cfg;
 949
 950         /* No blacklist configuration */
 951
 952         profile_cfg->num_profiles = req->n_match_sets;
 953         profile_cfg->active_clients = SCAN_CLIENT_SCHED_SCAN;
 954         profile_cfg->pass_match = SCAN_CLIENT_SCHED_SCAN;
 955         profile_cfg->match_notify = SCAN_CLIENT_SCHED_SCAN;
 956         if (!req->n_match_sets || !req->match_sets[0].ssid.ssid_len)
 957                 profile_cfg->any_beacon_notify = SCAN_CLIENT_SCHED_SCAN;
 958
 959         for (i = 0; i < req->n_match_sets; i++) {
 960                 profile = &profile_cfg->profiles[i];
 961                 profile->ssid_index = i;
 962                 /* Support any cipher and auth algorithm */
 963                 profile->unicast_cipher = 0xff;
 964                 profile->auth_alg = 0xff;
 965                 profile->network_type = IWL_NETWORK_TYPE_ANY;
 966                 profile->band_selection = IWL_SCAN_OFFLOAD_SELECT_ANY;
 967                 profile->client_bitmap = SCAN_CLIENT_SCHED_SCAN;
 968         }
 969
 970         IWL_DEBUG_SCAN(mvm, "Sending scheduled scan profile config\n");
 971
 972         ret = iwl_mvm_send_cmd(mvm, &cmd);
 973         kfree(profile_cfg);
 974 free_blacklist:
 975         kfree(blacklist);
 976
 977         return ret;
 978 }
 979
 980 static bool iwl_mvm_scan_pass_all(struct iwl_mvm *mvm,
 981                                   struct cfg80211_sched_scan_request *req)
 982 {
 983         if (req->n_match_sets && req->match_sets[0].ssid.ssid_len) {
 984                 IWL_DEBUG_SCAN(mvm,
 985                                "Sending scheduled scan with filtering, n_match_sets %d\n",
 986                                req->n_match_sets);
 987                 return false;
 988         }
 989
 990         IWL_DEBUG_SCAN(mvm, "Sending Scheduled scan without filtering\n");
 991         return true;
 992 }
 993
 994 int iwl_mvm_sched_scan_start(struct iwl_mvm *mvm,
 995                              struct cfg80211_sched_scan_request *req)
 996 {
 997         struct iwl_scan_offload_req scan_req = {
 998                 .watchdog = IWL_SCHED_SCAN_WATCHDOG,
 999
1000                 .schedule_line[0].iterations = IWL_FAST_SCHED_SCAN_ITERATIONS,
1001                 .schedule_line[0].delay = cpu_to_le16(req->interval / 1000),
1002                 .schedule_line[0].full_scan_mul = 1,
1003
1004                 .schedule_line[1].iterations = 0xff,
1005                 .schedule_line[1].delay = cpu_to_le16(req->interval / 1000),
1006                 .schedule_line[1].full_scan_mul = IWL_FULL_SCAN_MULTIPLIER,
1007         };
1008
1009         if (iwl_mvm_scan_pass_all(mvm, req))
1010                 scan_req.flags |= cpu_to_le16(IWL_SCAN_OFFLOAD_FLAG_PASS_ALL);
1011
1012         if (mvm->last_ebs_successful &&
1013             mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_EBS_SUPPORT)
1014                 scan_req.flags |=
1015                         cpu_to_le16(IWL_SCAN_OFFLOAD_FLAG_EBS_ACCURATE_MODE);
1016
1017         return iwl_mvm_send_cmd_pdu(mvm, SCAN_OFFLOAD_REQUEST_CMD, 0,
1018                                     sizeof(scan_req), &scan_req);
1019 }
1020
1021 int iwl_mvm_scan_offload_start(struct iwl_mvm *mvm,
1022                                struct ieee80211_vif *vif,
1023                                struct cfg80211_sched_scan_request *req,
1024                                struct ieee80211_scan_ies *ies)
1025 {
1026         int ret;
1027
1028         if (mvm->fw->ucode_capa.capa[0] & IWL_UCODE_TLV_CAPA_UMAC_SCAN) {
1029                 ret = iwl_mvm_config_sched_scan_profiles(mvm, req);
1030                 if (ret)
1031                         return ret;
1032                 ret = iwl_mvm_sched_scan_umac(mvm, vif, req, ies);
1033         } else if ((mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)) {
1034                 mvm->scan_status = IWL_MVM_SCAN_SCHED;
1035                 ret = iwl_mvm_config_sched_scan_profiles(mvm, req);
1036                 if (ret)
1037                         return ret;
1038                 ret = iwl_mvm_unified_sched_scan_lmac(mvm, vif, req, ies);
1039         } else {
1040                 mvm->scan_status = IWL_MVM_SCAN_SCHED;
1041                 ret = iwl_mvm_config_sched_scan(mvm, vif, req, ies);
1042                 if (ret)
1043                         return ret;
1044                 ret = iwl_mvm_config_sched_scan_profiles(mvm, req);
1045                 if (ret)
1046                         return ret;
1047                 ret = iwl_mvm_sched_scan_start(mvm, req);
1048         }
1049
1050         return ret;
1051 }
1052
1053 static int iwl_mvm_send_scan_offload_abort(struct iwl_mvm *mvm)
1054 {
1055         int ret;
1056         struct iwl_host_cmd cmd = {
1057                 .id = SCAN_OFFLOAD_ABORT_CMD,
1058         };
1059         u32 status;
1060
1061         /* Exit instantly with error when device is not ready
1062          * to receive scan abort command or it does not perform
1063          * scheduled scan currently */
1064         if (mvm->scan_status != IWL_MVM_SCAN_SCHED &&
1065             (!(mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN) ||
1066              mvm->scan_status != IWL_MVM_SCAN_OS))
1067                 return -EIO;
1068
1069         ret = iwl_mvm_send_cmd_status(mvm, &cmd, &status);
1070         if (ret)
1071                 return ret;
1072
1073         if (status != CAN_ABORT_STATUS) {
1074                 /*
1075                  * The scan abort will return 1 for success or
1076                  * 2 for "failure".  A failure condition can be
1077                  * due to simply not being in an active scan which
1078                  * can occur if we send the scan abort before the
1079                  * microcode has notified us that a scan is completed.
1080                  */
1081                 IWL_DEBUG_SCAN(mvm, "SCAN OFFLOAD ABORT ret %d.\n", status);
1082                 ret = -ENOENT;
1083         }
1084
1085         return ret;
1086 }
1087
1088 int iwl_mvm_scan_offload_stop(struct iwl_mvm *mvm, bool notify)
1089 {
1090         int ret;
1091         struct iwl_notification_wait wait_scan_done;
1092         static const u8 scan_done_notif[] = { SCAN_OFFLOAD_COMPLETE, };
1093         bool sched = mvm->scan_status == IWL_MVM_SCAN_SCHED;
1094
1095         lockdep_assert_held(&mvm->mutex);
1096
1097         if (mvm->fw->ucode_capa.capa[0] & IWL_UCODE_TLV_CAPA_UMAC_SCAN)
1098                 return iwl_umac_scan_stop(mvm, IWL_UMAC_SCAN_UID_SCHED_SCAN,
1099                                           notify);
1100
1101         if (mvm->scan_status != IWL_MVM_SCAN_SCHED &&
1102             (!(mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN) ||
1103              mvm->scan_status != IWL_MVM_SCAN_OS)) {
1104                 IWL_DEBUG_SCAN(mvm, "No scan to stop\n");
1105                 return 0;
1106         }
1107
1108         iwl_init_notification_wait(&mvm->notif_wait, &wait_scan_done,
1109                                    scan_done_notif,
1110                                    ARRAY_SIZE(scan_done_notif),
1111                                    NULL, NULL);
1112
1113         ret = iwl_mvm_send_scan_offload_abort(mvm);
1114         if (ret) {
1115                 IWL_DEBUG_SCAN(mvm, "Send stop %sscan failed %d\n",
1116                                sched ? "offloaded " : "", ret);
1117                 iwl_remove_notification(&mvm->notif_wait, &wait_scan_done);
1118                 return ret;
1119         }
1120
1121         IWL_DEBUG_SCAN(mvm, "Successfully sent stop %sscan\n",
1122                        sched ? "offloaded " : "");
1123
1124         ret = iwl_wait_notification(&mvm->notif_wait, &wait_scan_done, 1 * HZ);
1125         if (ret)
1126                 return ret;
1127
1128         /*
1129          * Clear the scan status so the next scan requests will succeed. This
1130          * also ensures the Rx handler doesn't do anything, as the scan was
1131          * stopped from above. Since the rx handler won't do anything now,
1132          * we have to release the scan reference here.
1133          */
1134         if (mvm->scan_status == IWL_MVM_SCAN_OS)
1135                 iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
1136
1137         mvm->scan_status = IWL_MVM_SCAN_NONE;
1138
1139         if (notify) {
1140                 if (sched)
1141                         ieee80211_sched_scan_stopped(mvm->hw);
1142                 else
1143                         ieee80211_scan_completed(mvm->hw, true);
1144         }
1145
1146         return 0;
1147 }
1148
1149 static void iwl_mvm_unified_scan_fill_tx_cmd(struct iwl_mvm *mvm,
1150                                              struct iwl_scan_req_tx_cmd *tx_cmd,
1151                                              bool no_cck)
1152 {
1153         tx_cmd[0].tx_flags = cpu_to_le32(TX_CMD_FLG_SEQ_CTL |
1154                                          TX_CMD_FLG_BT_DIS);
1155         tx_cmd[0].rate_n_flags = iwl_mvm_scan_rate_n_flags(mvm,
1156                                                            IEEE80211_BAND_2GHZ,
1157                                                            no_cck);
1158         tx_cmd[0].sta_id = mvm->aux_sta.sta_id;
1159
1160         tx_cmd[1].tx_flags = cpu_to_le32(TX_CMD_FLG_SEQ_CTL |
1161                                          TX_CMD_FLG_BT_DIS);
1162         tx_cmd[1].rate_n_flags = iwl_mvm_scan_rate_n_flags(mvm,
1163                                                            IEEE80211_BAND_5GHZ,
1164                                                            no_cck);
1165         tx_cmd[1].sta_id = mvm->aux_sta.sta_id;
1166 }
1167
1168 static void
1169 iwl_mvm_lmac_scan_cfg_channels(struct iwl_mvm *mvm,
1170                                struct ieee80211_channel **channels,
1171                                int n_channels, u32 ssid_bitmap,
1172                                struct iwl_scan_req_unified_lmac *cmd)
1173 {
1174         struct iwl_scan_channel_cfg_lmac *channel_cfg = (void *)&cmd->data;
1175         int i;
1176
1177         for (i = 0; i < n_channels; i++) {
1178                 channel_cfg[i].channel_num =
1179                         cpu_to_le16(channels[i]->hw_value);
1180                 channel_cfg[i].iter_count = cpu_to_le16(1);
1181                 channel_cfg[i].iter_interval = 0;
1182                 channel_cfg[i].flags =
1183                         cpu_to_le32(IWL_UNIFIED_SCAN_CHANNEL_PARTIAL |
1184                                     ssid_bitmap);
1185         }
1186 }
1187
1188 static u8 *iwl_mvm_copy_and_insert_ds_elem(struct iwl_mvm *mvm, const u8 *ies,
1189                                            size_t len, u8 *const pos)
1190 {
1191         static const u8 before_ds_params[] = {
1192                         WLAN_EID_SSID,
1193                         WLAN_EID_SUPP_RATES,
1194                         WLAN_EID_REQUEST,
1195                         WLAN_EID_EXT_SUPP_RATES,
1196         };
1197         size_t offs;
1198         u8 *newpos = pos;
1199
1200         if (!iwl_mvm_rrm_scan_needed(mvm)) {
1201                 memcpy(newpos, ies, len);
1202                 return newpos + len;
1203         }
1204
1205         offs = ieee80211_ie_split(ies, len,
1206                                   before_ds_params,
1207                                   ARRAY_SIZE(before_ds_params),
1208                                   0);
1209
1210         memcpy(newpos, ies, offs);
1211         newpos += offs;
1212
1213         /* Add a placeholder for DS Parameter Set element */
1214         *newpos++ = WLAN_EID_DS_PARAMS;
1215         *newpos++ = 1;
1216         *newpos++ = 0;
1217
1218         memcpy(newpos, ies + offs, len - offs);
1219         newpos += len - offs;
1220
1221         return newpos;
1222 }
1223
1224 static void
1225 iwl_mvm_build_unified_scan_probe(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
1226                                  struct ieee80211_scan_ies *ies,
1227                                  struct iwl_scan_probe_req *preq,
1228                                  const u8 *mac_addr, const u8 *mac_addr_mask)
1229 {
1230         struct ieee80211_mgmt *frame = (struct ieee80211_mgmt *)preq->buf;
1231         u8 *pos, *newpos;
1232
1233         /*
1234          * Unfortunately, right now the offload scan doesn't support randomising
1235          * within the firmware, so until the firmware API is ready we implement
1236          * it in the driver. This means that the scan iterations won't really be
1237          * random, only when it's restarted, but at least that helps a bit.
1238          */
1239         if (mac_addr)
1240                 get_random_mask_addr(frame->sa, mac_addr, mac_addr_mask);
1241         else
1242                 memcpy(frame->sa, vif->addr, ETH_ALEN);
1243
1244         frame->frame_control = cpu_to_le16(IEEE80211_STYPE_PROBE_REQ);
1245         eth_broadcast_addr(frame->da);
1246         eth_broadcast_addr(frame->bssid);
1247         frame->seq_ctrl = 0;
1248
1249         pos = frame->u.probe_req.variable;
1250         *pos++ = WLAN_EID_SSID;
1251         *pos++ = 0;
1252
1253         preq->mac_header.offset = 0;
1254         preq->mac_header.len = cpu_to_le16(24 + 2);
1255
1256         /* Insert ds parameter set element on 2.4 GHz band */
1257         newpos = iwl_mvm_copy_and_insert_ds_elem(mvm,
1258                                                  ies->ies[IEEE80211_BAND_2GHZ],
1259                                                  ies->len[IEEE80211_BAND_2GHZ],
1260                                                  pos);
1261         preq->band_data[0].offset = cpu_to_le16(pos - preq->buf);
1262         preq->band_data[0].len = cpu_to_le16(newpos - pos);
1263         pos = newpos;
1264
1265         memcpy(pos, ies->ies[IEEE80211_BAND_5GHZ],
1266                ies->len[IEEE80211_BAND_5GHZ]);
1267         preq->band_data[1].offset = cpu_to_le16(pos - preq->buf);
1268         preq->band_data[1].len = cpu_to_le16(ies->len[IEEE80211_BAND_5GHZ]);
1269         pos += ies->len[IEEE80211_BAND_5GHZ];
1270
1271         memcpy(pos, ies->common_ies, ies->common_ie_len);
1272         preq->common_data.offset = cpu_to_le16(pos - preq->buf);
1273         preq->common_data.len = cpu_to_le16(ies->common_ie_len);
1274 }
1275
1276 static void
1277 iwl_mvm_build_generic_unified_scan_cmd(struct iwl_mvm *mvm,
1278                                        struct iwl_scan_req_unified_lmac *cmd,
1279                                        struct iwl_mvm_scan_params *params)
1280 {
1281         memset(cmd, 0, ksize(cmd));
1282         cmd->active_dwell = params->dwell[IEEE80211_BAND_2GHZ].active;
1283         cmd->passive_dwell = params->dwell[IEEE80211_BAND_2GHZ].passive;
1284         if (params->passive_fragmented)
1285                 cmd->fragmented_dwell =
1286                                 params->dwell[IEEE80211_BAND_2GHZ].passive;
1287         cmd->rx_chain_select = iwl_mvm_scan_rx_chain(mvm);
1288         cmd->max_out_time = cpu_to_le32(params->max_out_time);
1289         cmd->suspend_time = cpu_to_le32(params->suspend_time);
1290         cmd->scan_prio = cpu_to_le32(IWL_SCAN_PRIORITY_HIGH);
1291         cmd->iter_num = cpu_to_le32(1);
1292
1293         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_EBS_SUPPORT &&
1294             mvm->last_ebs_successful) {
1295                 cmd->channel_opt[0].flags =
1296                         cpu_to_le16(IWL_SCAN_CHANNEL_FLAG_EBS |
1297                                     IWL_SCAN_CHANNEL_FLAG_EBS_ACCURATE |
1298                                     IWL_SCAN_CHANNEL_FLAG_CACHE_ADD);
1299                 cmd->channel_opt[1].flags =
1300                         cpu_to_le16(IWL_SCAN_CHANNEL_FLAG_EBS |
1301                                     IWL_SCAN_CHANNEL_FLAG_EBS_ACCURATE |
1302                                     IWL_SCAN_CHANNEL_FLAG_CACHE_ADD);
1303         }
1304
1305         if (iwl_mvm_rrm_scan_needed(mvm))
1306                 cmd->scan_flags |=
1307                         cpu_to_le32(IWL_MVM_LMAC_SCAN_FLAGS_RRM_ENABLED);
1308 }
1309
1310 int iwl_mvm_unified_scan_lmac(struct iwl_mvm *mvm,
1311                               struct ieee80211_vif *vif,
1312                               struct ieee80211_scan_request *req)
1313 {
1314         struct iwl_host_cmd hcmd = {
1315                 .id = SCAN_OFFLOAD_REQUEST_CMD,
1316                 .len = { sizeof(struct iwl_scan_req_unified_lmac) +
1317                          sizeof(struct iwl_scan_channel_cfg_lmac) *
1318                                 mvm->fw->ucode_capa.n_scan_channels +
1319                          sizeof(struct iwl_scan_probe_req), },
1320                 .data = { mvm->scan_cmd, },
1321                 .dataflags = { IWL_HCMD_DFL_NOCOPY, },
1322         };
1323         struct iwl_scan_req_unified_lmac *cmd = mvm->scan_cmd;
1324         struct iwl_scan_probe_req *preq;
1325         struct iwl_mvm_scan_params params = {};
1326         u32 flags;
1327         u32 ssid_bitmap = 0;
1328         int ret, i;
1329
1330         lockdep_assert_held(&mvm->mutex);
1331
1332         /* we should have failed registration if scan_cmd was NULL */
1333         if (WARN_ON(mvm->scan_cmd == NULL))
1334                 return -ENOMEM;
1335
1336         if (req->req.n_ssids > PROBE_OPTION_MAX ||
1337             req->ies.common_ie_len + req->ies.len[NL80211_BAND_2GHZ] +
1338             req->ies.len[NL80211_BAND_5GHZ] >
1339                 iwl_mvm_max_scan_ie_fw_cmd_room(mvm, false) ||
1340             req->req.n_channels > mvm->fw->ucode_capa.n_scan_channels)
1341                 return -ENOBUFS;
1342
1343         mvm->scan_status = IWL_MVM_SCAN_OS;
1344
1345         iwl_mvm_scan_calc_params(mvm, vif, req->req.n_ssids, req->req.flags,
1346                                  &params);
1347
1348         iwl_mvm_build_generic_unified_scan_cmd(mvm, cmd, &params);
1349
1350         cmd->n_channels = (u8)req->req.n_channels;
1351
1352         flags = IWL_MVM_LMAC_SCAN_FLAG_PASS_ALL;
1353
1354         if (req->req.n_ssids == 1 && req->req.ssids[0].ssid_len != 0)
1355                 flags |= IWL_MVM_LMAC_SCAN_FLAG_PRE_CONNECTION;
1356
1357         if (params.passive_fragmented)
1358                 flags |= IWL_MVM_LMAC_SCAN_FLAG_FRAGMENTED;
1359
1360         if (req->req.n_ssids == 0)
1361                 flags |= IWL_MVM_LMAC_SCAN_FLAG_PASSIVE;
1362
1363         cmd->scan_flags |= cpu_to_le32(flags);
1364
1365         cmd->flags = iwl_mvm_scan_rxon_flags(req->req.channels[0]->band);
1366         cmd->filter_flags = cpu_to_le32(MAC_FILTER_ACCEPT_GRP |
1367                                         MAC_FILTER_IN_BEACON);
1368         iwl_mvm_unified_scan_fill_tx_cmd(mvm, cmd->tx_cmd, req->req.no_cck);
1369         iwl_mvm_scan_fill_ssids(cmd->direct_scan, req->req.ssids,
1370                                 req->req.n_ssids, 0);
1371
1372         cmd->schedule[0].delay = 0;
1373         cmd->schedule[0].iterations = 1;
1374         cmd->schedule[0].full_scan_mul = 0;
1375         cmd->schedule[1].delay = 0;
1376         cmd->schedule[1].iterations = 0;
1377         cmd->schedule[1].full_scan_mul = 0;
1378
1379         for (i = 1; i <= req->req.n_ssids; i++)
1380                 ssid_bitmap |= BIT(i);
1381
1382         iwl_mvm_lmac_scan_cfg_channels(mvm, req->req.channels,
1383                                        req->req.n_channels, ssid_bitmap,
1384                                        cmd);
1385
1386         preq = (void *)(cmd->data + sizeof(struct iwl_scan_channel_cfg_lmac) *
1387                         mvm->fw->ucode_capa.n_scan_channels);
1388
1389         iwl_mvm_build_unified_scan_probe(mvm, vif, &req->ies, preq,
1390                 req->req.flags & NL80211_SCAN_FLAG_RANDOM_ADDR ?
1391                         req->req.mac_addr : NULL,
1392                 req->req.mac_addr_mask);
1393
1394         ret = iwl_mvm_send_cmd(mvm, &hcmd);
1395         if (!ret) {
1396                 IWL_DEBUG_SCAN(mvm, "Scan request was sent successfully\n");
1397         } else {
1398                 /*
1399                  * If the scan failed, it usually means that the FW was unable
1400                  * to allocate the time events. Warn on it, but maybe we
1401                  * should try to send the command again with different params.
1402                  */
1403                 IWL_ERR(mvm, "Scan failed! ret %d\n", ret);
1404                 mvm->scan_status = IWL_MVM_SCAN_NONE;
1405                 ret = -EIO;
1406         }
1407         return ret;
1408 }
1409
1410 int iwl_mvm_unified_sched_scan_lmac(struct iwl_mvm *mvm,
1411                                     struct ieee80211_vif *vif,
1412                                     struct cfg80211_sched_scan_request *req,
1413                                     struct ieee80211_scan_ies *ies)
1414 {
1415         struct iwl_host_cmd hcmd = {
1416                 .id = SCAN_OFFLOAD_REQUEST_CMD,
1417                 .len = { sizeof(struct iwl_scan_req_unified_lmac) +
1418                          sizeof(struct iwl_scan_channel_cfg_lmac) *
1419                                 mvm->fw->ucode_capa.n_scan_channels +
1420                          sizeof(struct iwl_scan_probe_req), },
1421                 .data = { mvm->scan_cmd, },
1422                 .dataflags = { IWL_HCMD_DFL_NOCOPY, },
1423         };
1424         struct iwl_scan_req_unified_lmac *cmd = mvm->scan_cmd;
1425         struct iwl_scan_probe_req *preq;
1426         struct iwl_mvm_scan_params params = {};
1427         int ret;
1428         u32 flags = 0, ssid_bitmap = 0;
1429
1430         lockdep_assert_held(&mvm->mutex);
1431
1432         /* we should have failed registration if scan_cmd was NULL */
1433         if (WARN_ON(mvm->scan_cmd == NULL))
1434                 return -ENOMEM;
1435
1436         if (req->n_ssids > PROBE_OPTION_MAX ||
1437             ies->common_ie_len + ies->len[NL80211_BAND_2GHZ] +
1438             ies->len[NL80211_BAND_5GHZ] >
1439                 iwl_mvm_max_scan_ie_fw_cmd_room(mvm, true) ||
1440             req->n_channels > mvm->fw->ucode_capa.n_scan_channels)
1441                 return -ENOBUFS;
1442
1443         iwl_mvm_scan_calc_params(mvm, vif, req->n_ssids, 0, &params);
1444
1445         iwl_mvm_build_generic_unified_scan_cmd(mvm, cmd, &params);
1446
1447         cmd->n_channels = (u8)req->n_channels;
1448
1449         if (iwl_mvm_scan_pass_all(mvm, req))
1450                 flags |= IWL_MVM_LMAC_SCAN_FLAG_PASS_ALL;
1451
1452         if (req->n_ssids == 1 && req->ssids[0].ssid_len != 0)
1453                 flags |= IWL_MVM_LMAC_SCAN_FLAG_PRE_CONNECTION;
1454
1455         if (params.passive_fragmented)
1456                 flags |= IWL_MVM_LMAC_SCAN_FLAG_FRAGMENTED;
1457
1458         if (req->n_ssids == 0)
1459                 flags |= IWL_MVM_LMAC_SCAN_FLAG_PASSIVE;
1460
1461         cmd->scan_flags |= cpu_to_le32(flags);
1462
1463         cmd->flags = iwl_mvm_scan_rxon_flags(req->channels[0]->band);
1464         cmd->filter_flags = cpu_to_le32(MAC_FILTER_ACCEPT_GRP |
1465                                         MAC_FILTER_IN_BEACON);
1466         iwl_mvm_unified_scan_fill_tx_cmd(mvm, cmd->tx_cmd, false);
1467         iwl_scan_offload_build_ssid(req, cmd->direct_scan, &ssid_bitmap, false);
1468
1469         cmd->schedule[0].delay = cpu_to_le16(req->interval / MSEC_PER_SEC);
1470         cmd->schedule[0].iterations = IWL_FAST_SCHED_SCAN_ITERATIONS;
1471         cmd->schedule[0].full_scan_mul = 1;
1472
1473         cmd->schedule[1].delay = cpu_to_le16(req->interval / MSEC_PER_SEC);
1474         cmd->schedule[1].iterations = 0xff;
1475         cmd->schedule[1].full_scan_mul = IWL_FULL_SCAN_MULTIPLIER;
1476
1477         iwl_mvm_lmac_scan_cfg_channels(mvm, req->channels, req->n_channels,
1478                                        ssid_bitmap, cmd);
1479
1480         preq = (void *)(cmd->data + sizeof(struct iwl_scan_channel_cfg_lmac) *
1481                         mvm->fw->ucode_capa.n_scan_channels);
1482
1483         iwl_mvm_build_unified_scan_probe(mvm, vif, ies, preq,
1484                 req->flags & NL80211_SCAN_FLAG_RANDOM_ADDR ?
1485                         req->mac_addr : NULL,
1486                 req->mac_addr_mask);
1487
1488         ret = iwl_mvm_send_cmd(mvm, &hcmd);
1489         if (!ret) {
1490                 IWL_DEBUG_SCAN(mvm,
1491                                "Sched scan request was sent successfully\n");
1492         } else {
1493                 /*
1494                  * If the scan failed, it usually means that the FW was unable
1495                  * to allocate the time events. Warn on it, but maybe we
1496                  * should try to send the command again with different params.
1497                  */
1498                 IWL_ERR(mvm, "Sched scan failed! ret %d\n", ret);
1499                 mvm->scan_status = IWL_MVM_SCAN_NONE;
1500                 ret = -EIO;
1501         }
1502         return ret;
1503 }
1504
1505
1506 int iwl_mvm_cancel_scan(struct iwl_mvm *mvm)
1507 {
1508         if (mvm->fw->ucode_capa.capa[0] & IWL_UCODE_TLV_CAPA_UMAC_SCAN)
1509                 return iwl_umac_scan_stop(mvm, IWL_UMAC_SCAN_UID_REG_SCAN,
1510                                           true);
1511
1512         if (mvm->scan_status == IWL_MVM_SCAN_NONE)
1513                 return 0;
1514
1515         if (iwl_mvm_is_radio_killed(mvm)) {
1516                 ieee80211_scan_completed(mvm->hw, true);
1517                 iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
1518                 mvm->scan_status = IWL_MVM_SCAN_NONE;
1519                 return 0;
1520         }
1521
1522         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)
1523                 return iwl_mvm_scan_offload_stop(mvm, true);
1524         return iwl_mvm_cancel_regular_scan(mvm);
1525 }
1526
1527 /* UMAC scan API */
1528
1529 struct iwl_umac_scan_done {
1530         struct iwl_mvm *mvm;
1531         enum iwl_umac_scan_uid_type type;
1532 };
1533
1534 static int rate_to_scan_rate_flag(unsigned int rate)
1535 {
1536         static const int rate_to_scan_rate[IWL_RATE_COUNT] = {
1537                 [IWL_RATE_1M_INDEX]     = SCAN_CONFIG_RATE_1M,
1538                 [IWL_RATE_2M_INDEX]     = SCAN_CONFIG_RATE_2M,
1539                 [IWL_RATE_5M_INDEX]     = SCAN_CONFIG_RATE_5M,
1540                 [IWL_RATE_11M_INDEX]    = SCAN_CONFIG_RATE_11M,
1541                 [IWL_RATE_6M_INDEX]     = SCAN_CONFIG_RATE_6M,
1542                 [IWL_RATE_9M_INDEX]     = SCAN_CONFIG_RATE_9M,
1543                 [IWL_RATE_12M_INDEX]    = SCAN_CONFIG_RATE_12M,
1544                 [IWL_RATE_18M_INDEX]    = SCAN_CONFIG_RATE_18M,
1545                 [IWL_RATE_24M_INDEX]    = SCAN_CONFIG_RATE_24M,
1546                 [IWL_RATE_36M_INDEX]    = SCAN_CONFIG_RATE_36M,
1547                 [IWL_RATE_48M_INDEX]    = SCAN_CONFIG_RATE_48M,
1548                 [IWL_RATE_54M_INDEX]    = SCAN_CONFIG_RATE_54M,
1549         };
1550
1551         return rate_to_scan_rate[rate];
1552 }
1553
1554 static __le32 iwl_mvm_scan_config_rates(struct iwl_mvm *mvm)
1555 {
1556         struct ieee80211_supported_band *band;
1557         unsigned int rates = 0;
1558         int i;
1559
1560         band = &mvm->nvm_data->bands[IEEE80211_BAND_2GHZ];
1561         for (i = 0; i < band->n_bitrates; i++)
1562                 rates |= rate_to_scan_rate_flag(band->bitrates[i].hw_value);
1563         band = &mvm->nvm_data->bands[IEEE80211_BAND_5GHZ];
1564         for (i = 0; i < band->n_bitrates; i++)
1565                 rates |= rate_to_scan_rate_flag(band->bitrates[i].hw_value);
1566
1567         /* Set both basic rates and supported rates */
1568         rates |= SCAN_CONFIG_SUPPORTED_RATE(rates);
1569
1570         return cpu_to_le32(rates);
1571 }
1572
1573 int iwl_mvm_config_scan(struct iwl_mvm *mvm)
1574 {
1575
1576         struct iwl_scan_config *scan_config;
1577         struct ieee80211_supported_band *band;
1578         int num_channels =
1579                 mvm->nvm_data->bands[IEEE80211_BAND_2GHZ].n_channels +
1580                 mvm->nvm_data->bands[IEEE80211_BAND_5GHZ].n_channels;
1581         int ret, i, j = 0, cmd_size, data_size;
1582         struct iwl_host_cmd cmd = {
1583                 .id = SCAN_CFG_CMD,
1584         };
1585
1586         if (WARN_ON(num_channels > mvm->fw->ucode_capa.n_scan_channels))
1587                 return -ENOBUFS;
1588
1589         cmd_size = sizeof(*scan_config) + mvm->fw->ucode_capa.n_scan_channels;
1590
1591         scan_config = kzalloc(cmd_size, GFP_KERNEL);
1592         if (!scan_config)
1593                 return -ENOMEM;
1594
1595         data_size = cmd_size - sizeof(struct iwl_mvm_umac_cmd_hdr);
1596         scan_config->hdr.size = cpu_to_le16(data_size);
1597         scan_config->flags = cpu_to_le32(SCAN_CONFIG_FLAG_ACTIVATE |
1598                                          SCAN_CONFIG_FLAG_ALLOW_CHUB_REQS |
1599                                          SCAN_CONFIG_FLAG_SET_TX_CHAINS |
1600                                          SCAN_CONFIG_FLAG_SET_RX_CHAINS |
1601                                          SCAN_CONFIG_FLAG_SET_ALL_TIMES |
1602                                          SCAN_CONFIG_FLAG_SET_LEGACY_RATES |
1603                                          SCAN_CONFIG_FLAG_SET_MAC_ADDR |
1604                                          SCAN_CONFIG_FLAG_SET_CHANNEL_FLAGS|
1605                                          SCAN_CONFIG_N_CHANNELS(num_channels));
1606         scan_config->tx_chains = cpu_to_le32(mvm->fw->valid_tx_ant);
1607         scan_config->rx_chains = cpu_to_le32(iwl_mvm_scan_rx_ant(mvm));
1608         scan_config->legacy_rates = iwl_mvm_scan_config_rates(mvm);
1609         scan_config->out_of_channel_time = cpu_to_le32(170);
1610         scan_config->suspend_time = cpu_to_le32(30);
1611         scan_config->dwell_active = 20;
1612         scan_config->dwell_passive = 110;
1613         scan_config->dwell_fragmented = 20;
1614
1615         memcpy(&scan_config->mac_addr, &mvm->addresses[0].addr, ETH_ALEN);
1616
1617         scan_config->bcast_sta_id = mvm->aux_sta.sta_id;
1618         scan_config->channel_flags = IWL_CHANNEL_FLAG_EBS |
1619                                      IWL_CHANNEL_FLAG_ACCURATE_EBS |
1620                                      IWL_CHANNEL_FLAG_EBS_ADD |
1621                                      IWL_CHANNEL_FLAG_PRE_SCAN_PASSIVE2ACTIVE;
1622
1623         band = &mvm->nvm_data->bands[IEEE80211_BAND_2GHZ];
1624         for (i = 0; i < band->n_channels; i++, j++)
1625                 scan_config->channel_array[j] = band->channels[i].center_freq;
1626         band = &mvm->nvm_data->bands[IEEE80211_BAND_5GHZ];
1627         for (i = 0; i < band->n_channels; i++, j++)
1628                 scan_config->channel_array[j] = band->channels[i].center_freq;
1629
1630         cmd.data[0] = scan_config;
1631         cmd.len[0] = cmd_size;
1632         cmd.dataflags[0] = IWL_HCMD_DFL_NOCOPY;
1633
1634         IWL_DEBUG_SCAN(mvm, "Sending UMAC scan config\n");
1635
1636         ret = iwl_mvm_send_cmd(mvm, &cmd);
1637
1638         kfree(scan_config);
1639         return ret;
1640 }
1641
1642 static int iwl_mvm_find_scan_uid(struct iwl_mvm *mvm, u32 uid)
1643 {
1644         int i;
1645
1646         for (i = 0; i < IWL_MVM_MAX_SIMULTANEOUS_SCANS; i++)
1647                 if (mvm->scan_uid[i] == uid)
1648                         return i;
1649
1650         return i;
1651 }
1652
1653 static int iwl_mvm_find_free_scan_uid(struct iwl_mvm *mvm)
1654 {
1655         return iwl_mvm_find_scan_uid(mvm, 0);
1656 }
1657
1658 static bool iwl_mvm_find_scan_type(struct iwl_mvm *mvm,
1659                                    enum iwl_umac_scan_uid_type type)
1660 {
1661         int i;
1662
1663         for (i = 0; i < IWL_MVM_MAX_SIMULTANEOUS_SCANS; i++)
1664                 if (mvm->scan_uid[i] & type)
1665                         return true;
1666
1667         return false;
1668 }
1669
1670 static u32 iwl_generate_scan_uid(struct iwl_mvm *mvm,
1671                                  enum iwl_umac_scan_uid_type type)
1672 {
1673         u32 uid;
1674
1675         /* make sure exactly one bit is on in scan type */
1676         WARN_ON(hweight8(type) != 1);
1677
1678         /*
1679          * Make sure scan uids are unique. If one scan lasts long time while
1680          * others are completing frequently, the seq number will wrap up and
1681          * we may have more than one scan with the same uid.
1682          */
1683         do {
1684                 uid = type | (mvm->scan_seq_num <<
1685                               IWL_UMAC_SCAN_UID_SEQ_OFFSET);
1686                 mvm->scan_seq_num++;
1687         } while (iwl_mvm_find_scan_uid(mvm, uid) <
1688                  IWL_MVM_MAX_SIMULTANEOUS_SCANS);
1689
1690         IWL_DEBUG_SCAN(mvm, "Generated scan UID %u\n", uid);
1691
1692         return uid;
1693 }
1694
1695 static void
1696 iwl_mvm_build_generic_umac_scan_cmd(struct iwl_mvm *mvm,
1697                                     struct iwl_scan_req_umac *cmd,
1698                                     struct iwl_mvm_scan_params *params)
1699 {
1700         memset(cmd, 0, ksize(cmd));
1701         cmd->hdr.size = cpu_to_le16(iwl_mvm_scan_size(mvm) -
1702                                     sizeof(struct iwl_mvm_umac_cmd_hdr));
1703         cmd->active_dwell = params->dwell[IEEE80211_BAND_2GHZ].active;
1704         cmd->passive_dwell = params->dwell[IEEE80211_BAND_2GHZ].passive;
1705         if (params->passive_fragmented)
1706                 cmd->fragmented_dwell =
1707                                 params->dwell[IEEE80211_BAND_2GHZ].passive;
1708         cmd->max_out_time = cpu_to_le32(params->max_out_time);
1709         cmd->suspend_time = cpu_to_le32(params->suspend_time);
1710         cmd->scan_priority = cpu_to_le32(IWL_SCAN_PRIORITY_HIGH);
1711 }
1712
1713 static void
1714 iwl_mvm_umac_scan_cfg_channels(struct iwl_mvm *mvm,
1715                                struct ieee80211_channel **channels,
1716                                int n_channels, u32 ssid_bitmap,
1717                                struct iwl_scan_req_umac *cmd)
1718 {
1719         struct iwl_scan_channel_cfg_umac *channel_cfg = (void *)&cmd->data;
1720         int i;
1721
1722         for (i = 0; i < n_channels; i++) {
1723                 channel_cfg[i].flags = cpu_to_le32(ssid_bitmap);
1724                 channel_cfg[i].channel_num = channels[i]->hw_value;
1725                 channel_cfg[i].iter_count = 1;
1726                 channel_cfg[i].iter_interval = 0;
1727         }
1728 }
1729
1730 static u32 iwl_mvm_scan_umac_common_flags(struct iwl_mvm *mvm, int n_ssids,
1731                                           struct cfg80211_ssid *ssids,
1732                                           int fragmented)
1733 {
1734         int flags = 0;
1735
1736         if (n_ssids == 0)
1737                 flags = IWL_UMAC_SCAN_GEN_FLAGS_PASSIVE;
1738
1739         if (n_ssids == 1 && ssids[0].ssid_len != 0)
1740                 flags |= IWL_UMAC_SCAN_GEN_FLAGS_PRE_CONNECT;
1741
1742         if (fragmented)
1743                 flags |= IWL_UMAC_SCAN_GEN_FLAGS_FRAGMENTED;
1744
1745         if (iwl_mvm_rrm_scan_needed(mvm))
1746                 flags |= IWL_UMAC_SCAN_GEN_FLAGS_RRM_ENABLED;
1747
1748         return flags;
1749 }
1750
1751 int iwl_mvm_scan_umac(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
1752                       struct ieee80211_scan_request *req)
1753 {
1754         struct iwl_host_cmd hcmd = {
1755                 .id = SCAN_REQ_UMAC,
1756                 .len = { iwl_mvm_scan_size(mvm), },
1757                 .data = { mvm->scan_cmd, },
1758                 .dataflags = { IWL_HCMD_DFL_NOCOPY, },
1759         };
1760         struct iwl_scan_req_umac *cmd = mvm->scan_cmd;
1761         struct iwl_scan_req_umac_tail *sec_part = (void *)&cmd->data +
1762                 sizeof(struct iwl_scan_channel_cfg_umac) *
1763                         mvm->fw->ucode_capa.n_scan_channels;
1764         struct iwl_mvm_scan_params params = {};
1765         u32 uid, flags;
1766         u32 ssid_bitmap = 0;
1767         int ret, i, uid_idx;
1768
1769         lockdep_assert_held(&mvm->mutex);
1770
1771         uid_idx = iwl_mvm_find_free_scan_uid(mvm);
1772         if (uid_idx >= IWL_MVM_MAX_SIMULTANEOUS_SCANS)
1773                 return -EBUSY;
1774
1775         /* we should have failed registration if scan_cmd was NULL */
1776         if (WARN_ON(mvm->scan_cmd == NULL))
1777                 return -ENOMEM;
1778
1779         if (WARN_ON(req->req.n_ssids > PROBE_OPTION_MAX ||
1780                     req->ies.common_ie_len +
1781                     req->ies.len[NL80211_BAND_2GHZ] +
1782                     req->ies.len[NL80211_BAND_5GHZ] + 24 + 2 >
1783                     SCAN_OFFLOAD_PROBE_REQ_SIZE || req->req.n_channels >
1784                     mvm->fw->ucode_capa.n_scan_channels))
1785                 return -ENOBUFS;
1786
1787         iwl_mvm_scan_calc_params(mvm, vif, req->req.n_ssids, req->req.flags,
1788                                  &params);
1789
1790         iwl_mvm_build_generic_umac_scan_cmd(mvm, cmd, &params);
1791
1792         uid = iwl_generate_scan_uid(mvm, IWL_UMAC_SCAN_UID_REG_SCAN);
1793         mvm->scan_uid[uid_idx] = uid;
1794         cmd->uid = cpu_to_le32(uid);
1795
1796         cmd->ooc_priority = cpu_to_le32(IWL_SCAN_PRIORITY_HIGH);
1797
1798         flags = iwl_mvm_scan_umac_common_flags(mvm, req->req.n_ssids,
1799                                                req->req.ssids,
1800                                                params.passive_fragmented);
1801
1802         flags |= IWL_UMAC_SCAN_GEN_FLAGS_PASS_ALL;
1803
1804         cmd->general_flags = cpu_to_le32(flags);
1805         cmd->n_channels = req->req.n_channels;
1806
1807         for (i = 0; i < req->req.n_ssids; i++)
1808                 ssid_bitmap |= BIT(i);
1809
1810         iwl_mvm_umac_scan_cfg_channels(mvm, req->req.channels,
1811                                        req->req.n_channels, ssid_bitmap, cmd);
1812
1813         sec_part->schedule[0].iter_count = 1;
1814         sec_part->delay = 0;
1815
1816         iwl_mvm_build_unified_scan_probe(mvm, vif, &req->ies, &sec_part->preq,
1817                 req->req.flags & NL80211_SCAN_FLAG_RANDOM_ADDR ?
1818                         req->req.mac_addr : NULL,
1819                 req->req.mac_addr_mask);
1820
1821         iwl_mvm_scan_fill_ssids(sec_part->direct_scan, req->req.ssids,
1822                                 req->req.n_ssids, 0);
1823
1824         ret = iwl_mvm_send_cmd(mvm, &hcmd);
1825         if (!ret) {
1826                 IWL_DEBUG_SCAN(mvm,
1827                                "Scan request was sent successfully\n");
1828         } else {
1829                 /*
1830                  * If the scan failed, it usually means that the FW was unable
1831                  * to allocate the time events. Warn on it, but maybe we
1832                  * should try to send the command again with different params.
1833                  */
1834                 IWL_ERR(mvm, "Scan failed! ret %d\n", ret);
1835         }
1836         return ret;
1837 }
1838
1839 int iwl_mvm_sched_scan_umac(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
1840                             struct cfg80211_sched_scan_request *req,
1841                             struct ieee80211_scan_ies *ies)
1842 {
1843
1844         struct iwl_host_cmd hcmd = {
1845                 .id = SCAN_REQ_UMAC,
1846                 .len = { iwl_mvm_scan_size(mvm), },
1847                 .data = { mvm->scan_cmd, },
1848                 .dataflags = { IWL_HCMD_DFL_NOCOPY, },
1849         };
1850         struct iwl_scan_req_umac *cmd = mvm->scan_cmd;
1851         struct iwl_scan_req_umac_tail *sec_part = (void *)&cmd->data +
1852                 sizeof(struct iwl_scan_channel_cfg_umac) *
1853                         mvm->fw->ucode_capa.n_scan_channels;
1854         struct iwl_mvm_scan_params params = {};
1855         u32 uid, flags;
1856         u32 ssid_bitmap = 0;
1857         int ret, uid_idx;
1858
1859         lockdep_assert_held(&mvm->mutex);
1860
1861         uid_idx = iwl_mvm_find_free_scan_uid(mvm);
1862         if (uid_idx >= IWL_MVM_MAX_SIMULTANEOUS_SCANS)
1863                 return -EBUSY;
1864
1865         /* we should have failed registration if scan_cmd was NULL */
1866         if (WARN_ON(mvm->scan_cmd == NULL))
1867                 return -ENOMEM;
1868
1869         if (WARN_ON(req->n_ssids > PROBE_OPTION_MAX ||
1870                     ies->common_ie_len + ies->len[NL80211_BAND_2GHZ] +
1871                     ies->len[NL80211_BAND_5GHZ] + 24 + 2 >
1872                     SCAN_OFFLOAD_PROBE_REQ_SIZE || req->n_channels >
1873                     mvm->fw->ucode_capa.n_scan_channels))
1874                 return -ENOBUFS;
1875
1876         iwl_mvm_scan_calc_params(mvm, vif, req->n_ssids, req->flags,
1877                                          &params);
1878
1879         iwl_mvm_build_generic_umac_scan_cmd(mvm, cmd, &params);
1880
1881         cmd->flags = cpu_to_le32(IWL_UMAC_SCAN_FLAG_PREEMPTIVE);
1882
1883         uid = iwl_generate_scan_uid(mvm, IWL_UMAC_SCAN_UID_SCHED_SCAN);
1884         mvm->scan_uid[uid_idx] = uid;
1885         cmd->uid = cpu_to_le32(uid);
1886
1887         cmd->ooc_priority = cpu_to_le32(IWL_SCAN_PRIORITY_LOW);
1888
1889         flags = iwl_mvm_scan_umac_common_flags(mvm, req->n_ssids, req->ssids,
1890                                                params.passive_fragmented);
1891
1892         flags |= IWL_UMAC_SCAN_GEN_FLAGS_PERIODIC;
1893
1894         if (iwl_mvm_scan_pass_all(mvm, req))
1895                 flags |= IWL_UMAC_SCAN_GEN_FLAGS_PASS_ALL;
1896         else
1897                 flags |= IWL_UMAC_SCAN_GEN_FLAGS_MATCH;
1898
1899         cmd->general_flags = cpu_to_le32(flags);
1900
1901         if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_EBS_SUPPORT &&
1902             mvm->last_ebs_successful)
1903                 cmd->channel_flags = IWL_SCAN_CHANNEL_FLAG_EBS |
1904                                      IWL_SCAN_CHANNEL_FLAG_EBS_ACCURATE |
1905                                      IWL_SCAN_CHANNEL_FLAG_CACHE_ADD;
1906
1907         cmd->n_channels = req->n_channels;
1908
1909         iwl_scan_offload_build_ssid(req, sec_part->direct_scan, &ssid_bitmap,
1910                                     false);
1911
1912         /* This API uses bits 0-19 instead of 1-20. */
1913         ssid_bitmap = ssid_bitmap >> 1;
1914
1915         iwl_mvm_umac_scan_cfg_channels(mvm, req->channels, req->n_channels,
1916                                        ssid_bitmap, cmd);
1917
1918         sec_part->schedule[0].interval =
1919                                 cpu_to_le16(req->interval / MSEC_PER_SEC);
1920         sec_part->schedule[0].iter_count = 0xff;
1921
1922         sec_part->delay = 0;
1923
1924         iwl_mvm_build_unified_scan_probe(mvm, vif, ies, &sec_part->preq,
1925                 req->flags & NL80211_SCAN_FLAG_RANDOM_ADDR ?
1926                         req->mac_addr : NULL,
1927                 req->mac_addr_mask);
1928
1929         ret = iwl_mvm_send_cmd(mvm, &hcmd);
1930         if (!ret) {
1931                 IWL_DEBUG_SCAN(mvm,
1932                                "Sched scan request was sent successfully\n");
1933         } else {
1934                 /*
1935                  * If the scan failed, it usually means that the FW was unable
1936                  * to allocate the time events. Warn on it, but maybe we
1937                  * should try to send the command again with different params.
1938                  */
1939                 IWL_ERR(mvm, "Sched scan failed! ret %d\n", ret);
1940         }
1941         return ret;
1942 }
1943
1944 int iwl_mvm_rx_umac_scan_complete_notif(struct iwl_mvm *mvm,
1945                                         struct iwl_rx_cmd_buffer *rxb,
1946                                         struct iwl_device_cmd *cmd)
1947 {
1948         struct iwl_rx_packet *pkt = rxb_addr(rxb);
1949         struct iwl_umac_scan_complete *notif = (void *)pkt->data;
1950         u32 uid = __le32_to_cpu(notif->uid);
1951         bool sched = !!(uid & IWL_UMAC_SCAN_UID_SCHED_SCAN);
1952         int uid_idx = iwl_mvm_find_scan_uid(mvm, uid);
1953
1954         /*
1955          * Scan uid may be set to zero in case of scan abort request from above.
1956          */
1957         if (uid_idx >= IWL_MVM_MAX_SIMULTANEOUS_SCANS)
1958                 return 0;
1959
1960         IWL_DEBUG_SCAN(mvm,
1961                        "Scan completed, uid %u type %s, status %s, EBS status %s\n",
1962                        uid, sched ? "sched" : "regular",
1963                        notif->status == IWL_SCAN_OFFLOAD_COMPLETED ?
1964                                 "completed" : "aborted",
1965                        notif->ebs_status == IWL_SCAN_EBS_SUCCESS ?
1966                                 "success" : "failed");
1967
1968         mvm->last_ebs_successful = !notif->ebs_status;
1969         mvm->scan_uid[uid_idx] = 0;
1970
1971         if (!sched) {
1972                 ieee80211_scan_completed(mvm->hw,
1973                                          notif->status ==
1974                                                 IWL_SCAN_OFFLOAD_ABORTED);
1975                 iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
1976         } else if (!iwl_mvm_find_scan_type(mvm, IWL_UMAC_SCAN_UID_SCHED_SCAN)) {
1977                 ieee80211_sched_scan_stopped(mvm->hw);
1978         } else {
1979                 IWL_DEBUG_SCAN(mvm, "Another sched scan is running\n");
1980         }
1981
1982         return 0;
1983 }
1984
1985 static bool iwl_scan_umac_done_check(struct iwl_notif_wait_data *notif_wait,
1986                                      struct iwl_rx_packet *pkt, void *data)
1987 {
1988         struct iwl_umac_scan_done *scan_done = data;
1989         struct iwl_umac_scan_complete *notif = (void *)pkt->data;
1990         u32 uid = __le32_to_cpu(notif->uid);
1991         int uid_idx = iwl_mvm_find_scan_uid(scan_done->mvm, uid);
1992
1993         if (WARN_ON(pkt->hdr.cmd != SCAN_COMPLETE_UMAC))
1994                 return false;
1995
1996         if (uid_idx >= IWL_MVM_MAX_SIMULTANEOUS_SCANS)
1997                 return false;
1998
1999         /*
2000          * Clear scan uid of scans that was aborted from above and completed
2001          * in FW so the RX handler does nothing.
2002          */
2003         scan_done->mvm->scan_uid[uid_idx] = 0;
2004
2005         return !iwl_mvm_find_scan_type(scan_done->mvm, scan_done->type);
2006 }
2007
2008 static int iwl_umac_scan_abort_one(struct iwl_mvm *mvm, u32 uid)
2009 {
2010         struct iwl_umac_scan_abort cmd = {
2011                 .hdr.size = cpu_to_le16(sizeof(struct iwl_umac_scan_abort) -
2012                                         sizeof(struct iwl_mvm_umac_cmd_hdr)),
2013                 .uid = cpu_to_le32(uid),
2014         };
2015
2016         lockdep_assert_held(&mvm->mutex);
2017
2018         IWL_DEBUG_SCAN(mvm, "Sending scan abort, uid %u\n", uid);
2019
2020         return iwl_mvm_send_cmd_pdu(mvm, SCAN_ABORT_UMAC, 0, sizeof(cmd), &cmd);
2021 }
2022
2023 static int iwl_umac_scan_stop(struct iwl_mvm *mvm,
2024                               enum iwl_umac_scan_uid_type type, bool notify)
2025 {
2026         struct iwl_notification_wait wait_scan_done;
2027         static const u8 scan_done_notif[] = { SCAN_COMPLETE_UMAC, };
2028         struct iwl_umac_scan_done scan_done = {
2029                 .mvm = mvm,
2030                 .type = type,
2031         };
2032         int i, ret = -EIO;
2033
2034         iwl_init_notification_wait(&mvm->notif_wait, &wait_scan_done,
2035                                    scan_done_notif,
2036                                    ARRAY_SIZE(scan_done_notif),
2037                                    iwl_scan_umac_done_check, &scan_done);
2038
2039         IWL_DEBUG_SCAN(mvm, "Preparing to stop scan, type %x\n", type);
2040
2041         for (i = 0; i < IWL_MVM_MAX_SIMULTANEOUS_SCANS; i++) {
2042                 if (mvm->scan_uid[i] & type) {
2043                         int err;
2044
2045                         if (iwl_mvm_is_radio_killed(mvm) &&
2046                             (type & IWL_UMAC_SCAN_UID_REG_SCAN)) {
2047                                 ieee80211_scan_completed(mvm->hw, true);
2048                                 iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
2049                                 break;
2050                         }
2051
2052                         err = iwl_umac_scan_abort_one(mvm, mvm->scan_uid[i]);
2053                         if (!err)
2054                                 ret = 0;
2055                 }
2056         }
2057
2058         if (ret) {
2059                 IWL_DEBUG_SCAN(mvm, "Couldn't stop scan\n");
2060                 iwl_remove_notification(&mvm->notif_wait, &wait_scan_done);
2061                 return ret;
2062         }
2063
2064         ret = iwl_wait_notification(&mvm->notif_wait, &wait_scan_done, 1 * HZ);
2065         if (ret)
2066                 return ret;
2067
2068         if (notify) {
2069                 if (type & IWL_UMAC_SCAN_UID_SCHED_SCAN)
2070                         ieee80211_sched_scan_stopped(mvm->hw);
2071                 if (type & IWL_UMAC_SCAN_UID_REG_SCAN) {
2072                         ieee80211_scan_completed(mvm->hw, true);
2073                         iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
2074                 }
2075         }
2076
2077         return ret;
2078 }
2079
2080 int iwl_mvm_scan_size(struct iwl_mvm *mvm)
2081 {
2082         if (mvm->fw->ucode_capa.capa[0] & IWL_UCODE_TLV_CAPA_UMAC_SCAN)
2083                 return sizeof(struct iwl_scan_req_umac) +
2084                         sizeof(struct iwl_scan_channel_cfg_umac) *
2085                                 mvm->fw->ucode_capa.n_scan_channels +
2086                         sizeof(struct iwl_scan_req_umac_tail);
2087
2088         if (mvm->fw->ucode_capa.api[0] & IWL_UCODE_TLV_API_LMAC_SCAN)
2089                 return sizeof(struct iwl_scan_req_unified_lmac) +
2090                         sizeof(struct iwl_scan_channel_cfg_lmac) *
2091                                 mvm->fw->ucode_capa.n_scan_channels +
2092                         sizeof(struct iwl_scan_probe_req);
2093
2094         return sizeof(struct iwl_scan_cmd) +
2095                 mvm->fw->ucode_capa.max_probe_length +
2096                         mvm->fw->ucode_capa.n_scan_channels *
2097                 sizeof(struct iwl_scan_channel);
2098 }