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