/******************************************************************************
   *
   * This file is provided under a dual BSD/GPLv2 license.  When using or
   * redistributing this file, you may do so under either license.
   *
   * GPL LICENSE SUMMARY
   *
   * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
   * Copyright(c) 2013 - 2015 Intel Mobile Communications GmbH
   *
   * This program is free software; you can redistribute it and/or modify
   * it under the terms of version 2 of the GNU General Public License as
   * published by the Free Software Foundation.
   *
   * This program is distributed in the hope that it will be useful, but
   * WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   * General Public License for more details.
   *
   * You should have received a copy of the GNU General Public License
   * along with this program; if not, write to the Free Software
   * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110,
   * USA
   *
   * The full GNU General Public License is included in this distribution
   * in the file called COPYING.
   *
   * Contact Information:
   *  Intel Linux Wireless <ilw@linux.intel.com>
   * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
   *
   * BSD LICENSE
   *
   * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
   * Copyright(c) 2013 - 2015 Intel Mobile Communications GmbH
   * All rights reserved.
   *
   * Redistribution and use in source and binary forms, with or without
   * modification, are permitted provided that the following conditions
   * are met:
   *
   *  * Redistributions of source code must retain the above copyright
   *    notice, this list of conditions and the following disclaimer.
   *  * Redistributions in binary form must reproduce the above copyright
   *    notice, this list of conditions and the following disclaimer in
   *    the documentation and/or other materials provided with the
   *    distribution.
   *  * Neither the name Intel Corporation nor the names of its
   *    contributors may be used to endorse or promote products derived
   *    from this software without specific prior written permission.
   *
   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
   * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
   * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
   * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
   * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
   * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
   * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
   * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
   * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
   * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
   * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
   *
   *****************************************************************************/
  #include <linux/module.h>
  #include <linux/vmalloc.h>
  #include <net/mac80211.h>
  
  #include "iwl-notif-wait.h"
  #include "iwl-trans.h"
  #include "iwl-op-mode.h"
  #include "iwl-fw.h"
  #include "iwl-debug.h"
  #include "iwl-drv.h"
  #include "iwl-modparams.h"
  #include "mvm.h"
  #include "iwl-phy-db.h"
  #include "iwl-eeprom-parse.h"
  #include "iwl-csr.h"
  #include "iwl-io.h"
  #include "iwl-prph.h"
  #include "rs.h"
  #include "fw-api-scan.h"
  #include "time-event.h"
  
  #define DRV_DESCRIPTION	"The new Intel(R) wireless AGN driver for Linux"
  MODULE_DESCRIPTION(DRV_DESCRIPTION);
  MODULE_AUTHOR(DRV_COPYRIGHT " " DRV_AUTHOR);
  MODULE_LICENSE("GPL");
  
  static const struct iwl_op_mode_ops iwl_mvm_ops;
  static const struct iwl_op_mode_ops iwl_mvm_ops_mq;
  
  struct iwl_mvm_mod_params iwlmvm_mod_params = {
  	.power_scheme = IWL_POWER_SCHEME_BPS,
  	.tfd_q_hang_detect = true
  	/* rest of fields are 0 by default */
  };
  
  module_param_named(init_dbg, iwlmvm_mod_params.init_dbg, bool, S_IRUGO);
  MODULE_PARM_DESC(init_dbg,
  		 "set to true to debug an ASSERT in INIT fw (default: false");
  module_param_named(power_scheme, iwlmvm_mod_params.power_scheme, int, S_IRUGO);
  MODULE_PARM_DESC(power_scheme,
  		 "power management scheme: 1-active, 2-balanced, 3-low power, default: 2");
  module_param_named(tfd_q_hang_detect, iwlmvm_mod_params.tfd_q_hang_detect,
  		   bool, S_IRUGO);
  MODULE_PARM_DESC(tfd_q_hang_detect,
  		 "TFD queues hang detection (default: true");
  
  /*
   * module init and exit functions
   */
  static int __init iwl_mvm_init(void)
  {
  	int ret;
  
  	ret = iwl_mvm_rate_control_register();
  	if (ret) {
  		pr_err("Unable to register rate control algorithm: %d
  ", ret);
  		return ret;
  	}
  
  	ret = iwl_opmode_register("iwlmvm", &iwl_mvm_ops);
  
  	if (ret) {
  		pr_err("Unable to register MVM op_mode: %d
  ", ret);
  		iwl_mvm_rate_control_unregister();
  	}
  
  	return ret;
  }
  module_init(iwl_mvm_init);
  
  static void __exit iwl_mvm_exit(void)
  {
  	iwl_opmode_deregister("iwlmvm");
  	iwl_mvm_rate_control_unregister();
  }
  module_exit(iwl_mvm_exit);
  
  static void iwl_mvm_nic_config(struct iwl_op_mode *op_mode)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  	u8 radio_cfg_type, radio_cfg_step, radio_cfg_dash;
  	u32 reg_val = 0;
  	u32 phy_config = iwl_mvm_get_phy_config(mvm);
  
  	radio_cfg_type = (phy_config & FW_PHY_CFG_RADIO_TYPE) >>
  			 FW_PHY_CFG_RADIO_TYPE_POS;
  	radio_cfg_step = (phy_config & FW_PHY_CFG_RADIO_STEP) >>
  			 FW_PHY_CFG_RADIO_STEP_POS;
  	radio_cfg_dash = (phy_config & FW_PHY_CFG_RADIO_DASH) >>
  			 FW_PHY_CFG_RADIO_DASH_POS;
  
  	/* SKU control */
  	reg_val |= CSR_HW_REV_STEP(mvm->trans->hw_rev) <<
  				CSR_HW_IF_CONFIG_REG_POS_MAC_STEP;
  	reg_val |= CSR_HW_REV_DASH(mvm->trans->hw_rev) <<
  				CSR_HW_IF_CONFIG_REG_POS_MAC_DASH;
  
  	/* radio configuration */
  	reg_val |= radio_cfg_type << CSR_HW_IF_CONFIG_REG_POS_PHY_TYPE;
  	reg_val |= radio_cfg_step << CSR_HW_IF_CONFIG_REG_POS_PHY_STEP;
  	reg_val |= radio_cfg_dash << CSR_HW_IF_CONFIG_REG_POS_PHY_DASH;
  
  	WARN_ON((radio_cfg_type << CSR_HW_IF_CONFIG_REG_POS_PHY_TYPE) &
  		 ~CSR_HW_IF_CONFIG_REG_MSK_PHY_TYPE);
  
  	/*
  	 * TODO: Bits 7-8 of CSR in 8000 HW family set the ADC sampling, and
  	 * shouldn't be set to any non-zero value. The same is supposed to be
  	 * true of the other HW, but unsetting them (such as the 7260) causes
  	 * automatic tests to fail on seemingly unrelated errors. Need to
  	 * further investigate this, but for now we'll separate cases.
  	 */
  	if (mvm->trans->cfg->device_family != IWL_DEVICE_FAMILY_8000)
  		reg_val |= CSR_HW_IF_CONFIG_REG_BIT_RADIO_SI;
  
  	iwl_trans_set_bits_mask(mvm->trans, CSR_HW_IF_CONFIG_REG,
  				CSR_HW_IF_CONFIG_REG_MSK_MAC_DASH |
  				CSR_HW_IF_CONFIG_REG_MSK_MAC_STEP |
  				CSR_HW_IF_CONFIG_REG_MSK_PHY_TYPE |
  				CSR_HW_IF_CONFIG_REG_MSK_PHY_STEP |
  				CSR_HW_IF_CONFIG_REG_MSK_PHY_DASH |
  				CSR_HW_IF_CONFIG_REG_BIT_RADIO_SI |
  				CSR_HW_IF_CONFIG_REG_BIT_MAC_SI,
  				reg_val);
  
  	IWL_DEBUG_INFO(mvm, "Radio type=0x%x-0x%x-0x%x
  ", radio_cfg_type,
  		       radio_cfg_step, radio_cfg_dash);
  
  	/*
  	 * W/A : NIC is stuck in a reset state after Early PCIe power off
  	 * (PCIe power is lost before PERST# is asserted), causing ME FW
  	 * to lose ownership and not being able to obtain it back.
  	 */
  	if (!mvm->trans->cfg->apmg_not_supported)
  		iwl_set_bits_mask_prph(mvm->trans, APMG_PS_CTRL_REG,
  				       APMG_PS_CTRL_EARLY_PWR_OFF_RESET_DIS,
  				       ~APMG_PS_CTRL_EARLY_PWR_OFF_RESET_DIS);
  }
  
  struct iwl_rx_handlers {
  	u16 cmd_id;
  	bool async;
  	void (*fn)(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb);
  };
  
  #define RX_HANDLER(_cmd_id, _fn, _async)	\
  	{ .cmd_id = _cmd_id , .fn = _fn , .async = _async }
  #define RX_HANDLER_GRP(_grp, _cmd, _fn, _async)	\
  	{ .cmd_id = WIDE_ID(_grp, _cmd), .fn = _fn, .async = _async }
  
  /*
   * Handlers for fw notifications
   * Convention: RX_HANDLER(CMD_NAME, iwl_mvm_rx_CMD_NAME
   * This list should be in order of frequency for performance purposes.
   *
   * The handler can be SYNC - this means that it will be called in the Rx path
   * which can't acquire mvm->mutex. If the handler needs to hold mvm->mutex (and
   * only in this case!), it should be set as ASYNC. In that case, it will be
   * called from a worker with mvm->mutex held.
   */
  static const struct iwl_rx_handlers iwl_mvm_rx_handlers[] = {
  	RX_HANDLER(TX_CMD, iwl_mvm_rx_tx_cmd, false),
  	RX_HANDLER(BA_NOTIF, iwl_mvm_rx_ba_notif, false),
  
  	RX_HANDLER(BT_PROFILE_NOTIFICATION, iwl_mvm_rx_bt_coex_notif, true),
  	RX_HANDLER(BEACON_NOTIFICATION, iwl_mvm_rx_beacon_notif, true),
  	RX_HANDLER(STATISTICS_NOTIFICATION, iwl_mvm_rx_statistics, true),
  	RX_HANDLER(ANTENNA_COUPLING_NOTIFICATION,
  		   iwl_mvm_rx_ant_coupling_notif, true),
  
  	RX_HANDLER(TIME_EVENT_NOTIFICATION, iwl_mvm_rx_time_event_notif, false),
  	RX_HANDLER(MCC_CHUB_UPDATE_CMD, iwl_mvm_rx_chub_update_mcc, true),
  
  	RX_HANDLER(EOSP_NOTIFICATION, iwl_mvm_rx_eosp_notif, false),
  
  	RX_HANDLER(SCAN_ITERATION_COMPLETE,
  		   iwl_mvm_rx_lmac_scan_iter_complete_notif, false),
  	RX_HANDLER(SCAN_OFFLOAD_COMPLETE,
  		   iwl_mvm_rx_lmac_scan_complete_notif, true),
  	RX_HANDLER(MATCH_FOUND_NOTIFICATION, iwl_mvm_rx_scan_match_found,
  		   false),
  	RX_HANDLER(SCAN_COMPLETE_UMAC, iwl_mvm_rx_umac_scan_complete_notif,
  		   true),
  	RX_HANDLER(SCAN_ITERATION_COMPLETE_UMAC,
  		   iwl_mvm_rx_umac_scan_iter_complete_notif, false),
  
  	RX_HANDLER(CARD_STATE_NOTIFICATION, iwl_mvm_rx_card_state_notif, false),
  
  	RX_HANDLER(MISSED_BEACONS_NOTIFICATION, iwl_mvm_rx_missed_beacons_notif,
  		   false),
  
  	RX_HANDLER(REPLY_ERROR, iwl_mvm_rx_fw_error, false),
  	RX_HANDLER(PSM_UAPSD_AP_MISBEHAVING_NOTIFICATION,
  		   iwl_mvm_power_uapsd_misbehaving_ap_notif, false),
  	RX_HANDLER(DTS_MEASUREMENT_NOTIFICATION, iwl_mvm_temp_notif, true),
  	RX_HANDLER_GRP(PHY_OPS_GROUP, DTS_MEASUREMENT_NOTIF_WIDE,
  		       iwl_mvm_temp_notif, true),
  
  	RX_HANDLER(TDLS_CHANNEL_SWITCH_NOTIFICATION, iwl_mvm_rx_tdls_notif,
  		   true),
  	RX_HANDLER(MFUART_LOAD_NOTIFICATION, iwl_mvm_rx_mfuart_notif, false),
  	RX_HANDLER(TOF_NOTIFICATION, iwl_mvm_tof_resp_handler, true),
  
  };
  #undef RX_HANDLER
  #undef RX_HANDLER_GRP
  #define CMD(x) [x] = #x
  
  static const char *const iwl_mvm_cmd_strings[REPLY_MAX + 1] = {
  	CMD(MVM_ALIVE),
  	CMD(REPLY_ERROR),
  	CMD(ECHO_CMD),
  	CMD(INIT_COMPLETE_NOTIF),
  	CMD(PHY_CONTEXT_CMD),
  	CMD(MGMT_MCAST_KEY),
  	CMD(TX_CMD),
  	CMD(TXPATH_FLUSH),
  	CMD(SHARED_MEM_CFG),
  	CMD(MAC_CONTEXT_CMD),
  	CMD(TIME_EVENT_CMD),
  	CMD(TIME_EVENT_NOTIFICATION),
  	CMD(BINDING_CONTEXT_CMD),
  	CMD(TIME_QUOTA_CMD),
  	CMD(NON_QOS_TX_COUNTER_CMD),
  	CMD(DC2DC_CONFIG_CMD),
  	CMD(NVM_ACCESS_CMD),
  	CMD(PHY_CONFIGURATION_CMD),
  	CMD(CALIB_RES_NOTIF_PHY_DB),
  	CMD(SET_CALIB_DEFAULT_CMD),
  	CMD(FW_PAGING_BLOCK_CMD),
  	CMD(ADD_STA_KEY),
  	CMD(ADD_STA),
  	CMD(FW_GET_ITEM_CMD),
  	CMD(REMOVE_STA),
  	CMD(LQ_CMD),
  	CMD(SCAN_OFFLOAD_CONFIG_CMD),
  	CMD(MATCH_FOUND_NOTIFICATION),
  	CMD(SCAN_OFFLOAD_REQUEST_CMD),
  	CMD(SCAN_OFFLOAD_ABORT_CMD),
  	CMD(HOT_SPOT_CMD),
  	CMD(SCAN_OFFLOAD_COMPLETE),
  	CMD(SCAN_OFFLOAD_UPDATE_PROFILES_CMD),
  	CMD(SCAN_ITERATION_COMPLETE),
  	CMD(POWER_TABLE_CMD),
  	CMD(WEP_KEY),
  	CMD(REPLY_RX_PHY_CMD),
  	CMD(REPLY_RX_MPDU_CMD),
  	CMD(BEACON_NOTIFICATION),
  	CMD(BEACON_TEMPLATE_CMD),
  	CMD(STATISTICS_CMD),
  	CMD(STATISTICS_NOTIFICATION),
  	CMD(EOSP_NOTIFICATION),
  	CMD(REDUCE_TX_POWER_CMD),
  	CMD(TX_ANT_CONFIGURATION_CMD),
  	CMD(D3_CONFIG_CMD),
  	CMD(D0I3_END_CMD),
  	CMD(PROT_OFFLOAD_CONFIG_CMD),
  	CMD(OFFLOADS_QUERY_CMD),
  	CMD(REMOTE_WAKE_CONFIG_CMD),
  	CMD(WOWLAN_PATTERNS),
  	CMD(WOWLAN_CONFIGURATION),
  	CMD(WOWLAN_TSC_RSC_PARAM),
  	CMD(WOWLAN_TKIP_PARAM),
  	CMD(WOWLAN_KEK_KCK_MATERIAL),
  	CMD(WOWLAN_GET_STATUSES),
  	CMD(WOWLAN_TX_POWER_PER_DB),
  	CMD(SCAN_OFFLOAD_PROFILES_QUERY_CMD),
  	CMD(SCAN_OFFLOAD_HOTSPOTS_CONFIG_CMD),
  	CMD(SCAN_OFFLOAD_HOTSPOTS_QUERY_CMD),
  	CMD(CARD_STATE_NOTIFICATION),
  	CMD(MISSED_BEACONS_NOTIFICATION),
  	CMD(BT_COEX_PRIO_TABLE),
  	CMD(BT_COEX_PROT_ENV),
  	CMD(BT_PROFILE_NOTIFICATION),
  	CMD(BT_CONFIG),
  	CMD(MCAST_FILTER_CMD),
  	CMD(BCAST_FILTER_CMD),
  	CMD(REPLY_SF_CFG_CMD),
  	CMD(REPLY_BEACON_FILTERING_CMD),
  	CMD(CMD_DTS_MEASUREMENT_TRIGGER),
  	CMD(DTS_MEASUREMENT_NOTIFICATION),
  	CMD(REPLY_THERMAL_MNG_BACKOFF),
  	CMD(MAC_PM_POWER_TABLE),
  	CMD(LTR_CONFIG),
  	CMD(BT_COEX_CI),
  	CMD(BT_COEX_UPDATE_SW_BOOST),
  	CMD(BT_COEX_UPDATE_CORUN_LUT),
  	CMD(BT_COEX_UPDATE_REDUCED_TXP),
  	CMD(PSM_UAPSD_AP_MISBEHAVING_NOTIFICATION),
  	CMD(ANTENNA_COUPLING_NOTIFICATION),
  	CMD(SCD_QUEUE_CFG),
  	CMD(SCAN_CFG_CMD),
  	CMD(SCAN_REQ_UMAC),
  	CMD(SCAN_ABORT_UMAC),
  	CMD(SCAN_COMPLETE_UMAC),
  	CMD(TDLS_CHANNEL_SWITCH_CMD),
  	CMD(TDLS_CHANNEL_SWITCH_NOTIFICATION),
  	CMD(TDLS_CONFIG_CMD),
  	CMD(MCC_UPDATE_CMD),
  	CMD(SCAN_ITERATION_COMPLETE_UMAC),
  };
  #undef CMD
  
  /* this forward declaration can avoid to export the function */
  static void iwl_mvm_async_handlers_wk(struct work_struct *wk);
  static void iwl_mvm_d0i3_exit_work(struct work_struct *wk);
  
  static u32 calc_min_backoff(struct iwl_trans *trans, const struct iwl_cfg *cfg)
  {
  	const struct iwl_pwr_tx_backoff *pwr_tx_backoff = cfg->pwr_tx_backoffs;
  
  	if (!pwr_tx_backoff)
  		return 0;
  
  	while (pwr_tx_backoff->pwr) {
  		if (trans->dflt_pwr_limit >= pwr_tx_backoff->pwr)
  			return pwr_tx_backoff->backoff;
  
  		pwr_tx_backoff++;
  	}
  
  	return 0;
  }
  
  static void iwl_mvm_fw_error_dump_wk(struct work_struct *work);
  
  static struct iwl_op_mode *
  iwl_op_mode_mvm_start(struct iwl_trans *trans, const struct iwl_cfg *cfg,
  		      const struct iwl_fw *fw, struct dentry *dbgfs_dir)
  {
  	struct ieee80211_hw *hw;
  	struct iwl_op_mode *op_mode;
  	struct iwl_mvm *mvm;
  	struct iwl_trans_config trans_cfg = {};
  	static const u8 no_reclaim_cmds[] = {
  		TX_CMD,
  	};
  	int err, scan_size;
  	u32 min_backoff;
  
  	/*
  	 * We use IWL_MVM_STATION_COUNT to check the validity of the station
  	 * index all over the driver - check that its value corresponds to the
  	 * array size.
  	 */
  	BUILD_BUG_ON(ARRAY_SIZE(mvm->fw_id_to_mac_id) != IWL_MVM_STATION_COUNT);
  
  	/********************************
  	 * 1. Allocating and configuring HW data
  	 ********************************/
  	hw = ieee80211_alloc_hw(sizeof(struct iwl_op_mode) +
  				sizeof(struct iwl_mvm),
  				&iwl_mvm_hw_ops);
  	if (!hw)
  		return NULL;
  
  	if (cfg->max_rx_agg_size)
  		hw->max_rx_aggregation_subframes = cfg->max_rx_agg_size;
  
  	if (cfg->max_tx_agg_size)
  		hw->max_tx_aggregation_subframes = cfg->max_tx_agg_size;
  
  	op_mode = hw->priv;
  
  	mvm = IWL_OP_MODE_GET_MVM(op_mode);
  	mvm->dev = trans->dev;
  	mvm->trans = trans;
  	mvm->cfg = cfg;
  	mvm->fw = fw;
  	mvm->hw = hw;
  
  	if (iwl_mvm_has_new_rx_api(mvm)) {
  		op_mode->ops = &iwl_mvm_ops_mq;
  	} else {
  		op_mode->ops = &iwl_mvm_ops;
  
  		if (WARN_ON(trans->num_rx_queues > 1))
  			goto out_free;
  	}
  
  	mvm->restart_fw = iwlwifi_mod_params.restart_fw ? -1 : 0;
  
  	mvm->aux_queue = 15;
  	mvm->first_agg_queue = 16;
  	mvm->last_agg_queue = mvm->cfg->base_params->num_of_queues - 1;
  	if (mvm->cfg->base_params->num_of_queues == 16) {
  		mvm->aux_queue = 11;
  		mvm->first_agg_queue = 12;
  	}
  	mvm->sf_state = SF_UNINIT;
  	mvm->low_latency_agg_frame_limit = 6;
  	mvm->cur_ucode = IWL_UCODE_INIT;
  
  	mutex_init(&mvm->mutex);
  	mutex_init(&mvm->d0i3_suspend_mutex);
  	spin_lock_init(&mvm->async_handlers_lock);
  	INIT_LIST_HEAD(&mvm->time_event_list);
  	INIT_LIST_HEAD(&mvm->aux_roc_te_list);
  	INIT_LIST_HEAD(&mvm->async_handlers_list);
  	spin_lock_init(&mvm->time_event_lock);
  	spin_lock_init(&mvm->queue_info_lock);
  
  	INIT_WORK(&mvm->async_handlers_wk, iwl_mvm_async_handlers_wk);
  	INIT_WORK(&mvm->roc_done_wk, iwl_mvm_roc_done_wk);
  	INIT_WORK(&mvm->sta_drained_wk, iwl_mvm_sta_drained_wk);
  	INIT_WORK(&mvm->d0i3_exit_work, iwl_mvm_d0i3_exit_work);
  	INIT_DELAYED_WORK(&mvm->fw_dump_wk, iwl_mvm_fw_error_dump_wk);
  	INIT_DELAYED_WORK(&mvm->tdls_cs.dwork, iwl_mvm_tdls_ch_switch_work);
  
  	spin_lock_init(&mvm->d0i3_tx_lock);
  	spin_lock_init(&mvm->refs_lock);
  	skb_queue_head_init(&mvm->d0i3_tx);
  	init_waitqueue_head(&mvm->d0i3_exit_waitq);
  
  	SET_IEEE80211_DEV(mvm->hw, mvm->trans->dev);
  
  	/*
  	 * Populate the state variables that the transport layer needs
  	 * to know about.
  	 */
  	trans_cfg.op_mode = op_mode;
  	trans_cfg.no_reclaim_cmds = no_reclaim_cmds;
  	trans_cfg.n_no_reclaim_cmds = ARRAY_SIZE(no_reclaim_cmds);
  	trans_cfg.rx_buf_size_8k = iwlwifi_mod_params.amsdu_size_8K;
  	trans_cfg.wide_cmd_header = fw_has_api(&mvm->fw->ucode_capa,
  					       IWL_UCODE_TLV_API_WIDE_CMD_HDR);
  
  	if (mvm->fw->ucode_capa.flags & IWL_UCODE_TLV_FLAGS_DW_BC_TABLE)
  		trans_cfg.bc_table_dword = true;
  
  	trans_cfg.command_names = iwl_mvm_cmd_strings;
  
  	trans_cfg.cmd_queue = IWL_MVM_CMD_QUEUE;
  	trans_cfg.cmd_fifo = IWL_MVM_TX_FIFO_CMD;
  	trans_cfg.scd_set_active = true;
  
  	trans_cfg.sdio_adma_addr = fw->sdio_adma_addr;
  
  	/* Set a short watchdog for the command queue */
  	trans_cfg.cmd_q_wdg_timeout =
  		iwl_mvm_get_wd_timeout(mvm, NULL, false, true);
  
  	snprintf(mvm->hw->wiphy->fw_version,
  		 sizeof(mvm->hw->wiphy->fw_version),
  		 "%s", fw->fw_version);
  
  	/* Configure transport layer */
  	iwl_trans_configure(mvm->trans, &trans_cfg);
  
  	trans->rx_mpdu_cmd = REPLY_RX_MPDU_CMD;
  	trans->rx_mpdu_cmd_hdr_size = sizeof(struct iwl_rx_mpdu_res_start);
  	trans->dbg_dest_tlv = mvm->fw->dbg_dest_tlv;
  	trans->dbg_dest_reg_num = mvm->fw->dbg_dest_reg_num;
  	memcpy(trans->dbg_conf_tlv, mvm->fw->dbg_conf_tlv,
  	       sizeof(trans->dbg_conf_tlv));
  	trans->dbg_trigger_tlv = mvm->fw->dbg_trigger_tlv;
  
  	/* set up notification wait support */
  	iwl_notification_wait_init(&mvm->notif_wait);
  
  	/* Init phy db */
  	mvm->phy_db = iwl_phy_db_init(trans);
  	if (!mvm->phy_db) {
  		IWL_ERR(mvm, "Cannot init phy_db
  ");
  		goto out_free;
  	}
  
  	IWL_INFO(mvm, "Detected %s, REV=0x%X
  ",
  		 mvm->cfg->name, mvm->trans->hw_rev);
  
  	min_backoff = calc_min_backoff(trans, cfg);
  	iwl_mvm_tt_initialize(mvm, min_backoff);
  
  	if (iwlwifi_mod_params.nvm_file)
  		mvm->nvm_file_name = iwlwifi_mod_params.nvm_file;
  	else
  		IWL_DEBUG_EEPROM(mvm->trans->dev,
  				 "working without external nvm file
  ");
  
  	if (WARN(cfg->no_power_up_nic_in_init && !mvm->nvm_file_name,
  		 "not allowing power-up and not having nvm_file
  "))
  		goto out_free;
  
  	/*
  	 * Even if nvm exists in the nvm_file driver should read again the nvm
  	 * from the nic because there might be entries that exist in the OTP
  	 * and not in the file.
  	 * for nics with no_power_up_nic_in_init: rely completley on nvm_file
  	 */
  	if (cfg->no_power_up_nic_in_init && mvm->nvm_file_name) {
  		err = iwl_nvm_init(mvm, false);
  		if (err)
  			goto out_free;
  	} else {
  		err = iwl_trans_start_hw(mvm->trans);
  		if (err)
  			goto out_free;
  
  		mutex_lock(&mvm->mutex);
  		err = iwl_run_init_mvm_ucode(mvm, true);
  		if (!err || !iwlmvm_mod_params.init_dbg)
  			iwl_trans_stop_device(trans);
  		mutex_unlock(&mvm->mutex);
  		/* returns 0 if successful, 1 if success but in rfkill */
  		if (err < 0 && !iwlmvm_mod_params.init_dbg) {
  			IWL_ERR(mvm, "Failed to run INIT ucode: %d
  ", err);
  			goto out_free;
  		}
  	}
  
  	scan_size = iwl_mvm_scan_size(mvm);
  
  	mvm->scan_cmd = kmalloc(scan_size, GFP_KERNEL);
  	if (!mvm->scan_cmd)
  		goto out_free;
  
  	/* Set EBS as successful as long as not stated otherwise by the FW. */
  	mvm->last_ebs_successful = true;
  
  	err = iwl_mvm_mac_setup_register(mvm);
  	if (err)
  		goto out_free;
  
  	err = iwl_mvm_dbgfs_register(mvm, dbgfs_dir);
  	if (err)
  		goto out_unregister;
  
  	memset(&mvm->rx_stats, 0, sizeof(struct mvm_statistics_rx));
  
  	/* rpm starts with a taken ref. only set the appropriate bit here. */
  	mvm->refs[IWL_MVM_REF_UCODE_DOWN] = 1;
  
  	iwl_mvm_tof_init(mvm);
  
  	return op_mode;
  
   out_unregister:
  	ieee80211_unregister_hw(mvm->hw);
  	iwl_mvm_leds_exit(mvm);
   out_free:
  	flush_delayed_work(&mvm->fw_dump_wk);
  	iwl_phy_db_free(mvm->phy_db);
  	kfree(mvm->scan_cmd);
  	if (!cfg->no_power_up_nic_in_init || !mvm->nvm_file_name)
  		iwl_trans_op_mode_leave(trans);
  	ieee80211_free_hw(mvm->hw);
  	return NULL;
  }
  
  static void iwl_op_mode_mvm_stop(struct iwl_op_mode *op_mode)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  	int i;
  
  	iwl_mvm_leds_exit(mvm);
  
  	iwl_mvm_tt_exit(mvm);
  
  	ieee80211_unregister_hw(mvm->hw);
  
  	kfree(mvm->scan_cmd);
  	kfree(mvm->mcast_filter_cmd);
  	mvm->mcast_filter_cmd = NULL;
  
  #if defined(CONFIG_PM_SLEEP) && defined(CONFIG_IWLWIFI_DEBUGFS)
  	kfree(mvm->d3_resume_sram);
  	if (mvm->nd_config) {
  		kfree(mvm->nd_config->match_sets);
  		kfree(mvm->nd_config->scan_plans);
  		kfree(mvm->nd_config);
  		mvm->nd_config = NULL;
  	}
  #endif
  
  	iwl_trans_op_mode_leave(mvm->trans);
  
  	iwl_phy_db_free(mvm->phy_db);
  	mvm->phy_db = NULL;
  
  	iwl_free_nvm_data(mvm->nvm_data);
  	for (i = 0; i < NVM_MAX_NUM_SECTIONS; i++)
  		kfree(mvm->nvm_sections[i].data);
  
  	iwl_mvm_tof_clean(mvm);
  
  	ieee80211_free_hw(mvm->hw);
  }
  
  struct iwl_async_handler_entry {
  	struct list_head list;
  	struct iwl_rx_cmd_buffer rxb;
  	void (*fn)(struct iwl_mvm *mvm, struct iwl_rx_cmd_buffer *rxb);
  };
  
  void iwl_mvm_async_handlers_purge(struct iwl_mvm *mvm)
  {
  	struct iwl_async_handler_entry *entry, *tmp;
  
  	spin_lock_bh(&mvm->async_handlers_lock);
  	list_for_each_entry_safe(entry, tmp, &mvm->async_handlers_list, list) {
  		iwl_free_rxb(&entry->rxb);
  		list_del(&entry->list);
  		kfree(entry);
  	}
  	spin_unlock_bh(&mvm->async_handlers_lock);
  }
  
  static void iwl_mvm_async_handlers_wk(struct work_struct *wk)
  {
  	struct iwl_mvm *mvm =
  		container_of(wk, struct iwl_mvm, async_handlers_wk);
  	struct iwl_async_handler_entry *entry, *tmp;
  	struct list_head local_list;
  
  	INIT_LIST_HEAD(&local_list);
  
  	/* Ensure that we are not in stop flow (check iwl_mvm_mac_stop) */
  	mutex_lock(&mvm->mutex);
  
  	/*
  	 * Sync with Rx path with a lock. Remove all the entries from this list,
  	 * add them to a local one (lock free), and then handle them.
  	 */
  	spin_lock_bh(&mvm->async_handlers_lock);
  	list_splice_init(&mvm->async_handlers_list, &local_list);
  	spin_unlock_bh(&mvm->async_handlers_lock);
  
  	list_for_each_entry_safe(entry, tmp, &local_list, list) {
  		entry->fn(mvm, &entry->rxb);
  		iwl_free_rxb(&entry->rxb);
  		list_del(&entry->list);
  		kfree(entry);
  	}
  	mutex_unlock(&mvm->mutex);
  }
  
  static inline void iwl_mvm_rx_check_trigger(struct iwl_mvm *mvm,
  					    struct iwl_rx_packet *pkt)
  {
  	struct iwl_fw_dbg_trigger_tlv *trig;
  	struct iwl_fw_dbg_trigger_cmd *cmds_trig;
  	int i;
  
  	if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_FW_NOTIF))
  		return;
  
  	trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_FW_NOTIF);
  	cmds_trig = (void *)trig->data;
  
  	if (!iwl_fw_dbg_trigger_check_stop(mvm, NULL, trig))
  		return;
  
  	for (i = 0; i < ARRAY_SIZE(cmds_trig->cmds); i++) {
  		/* don't collect on CMD 0 */
  		if (!cmds_trig->cmds[i].cmd_id)
  			break;
  
  		if (cmds_trig->cmds[i].cmd_id != pkt->hdr.cmd ||
  		    cmds_trig->cmds[i].group_id != pkt->hdr.group_id)
  			continue;
  
  		iwl_mvm_fw_dbg_collect_trig(mvm, trig,
  					    "CMD 0x%02x.%02x received",
  					    pkt->hdr.group_id, pkt->hdr.cmd);
  		break;
  	}
  }
  
  static void iwl_mvm_rx_common(struct iwl_mvm *mvm,
  			      struct iwl_rx_cmd_buffer *rxb,
  			      struct iwl_rx_packet *pkt)
  {
  	int i;
  
  	iwl_mvm_rx_check_trigger(mvm, pkt);
  
  	/*
  	 * Do the notification wait before RX handlers so
  	 * even if the RX handler consumes the RXB we have
  	 * access to it in the notification wait entry.
  	 */
  	iwl_notification_wait_notify(&mvm->notif_wait, pkt);
  
  	for (i = 0; i < ARRAY_SIZE(iwl_mvm_rx_handlers); i++) {
  		const struct iwl_rx_handlers *rx_h = &iwl_mvm_rx_handlers[i];
  		struct iwl_async_handler_entry *entry;
  
  		if (rx_h->cmd_id != WIDE_ID(pkt->hdr.group_id, pkt->hdr.cmd))
  			continue;
  
  		if (!rx_h->async) {
  			rx_h->fn(mvm, rxb);
  			return;
  		}
  
  		entry = kzalloc(sizeof(*entry), GFP_ATOMIC);
  		/* we can't do much... */
  		if (!entry)
  			return;
  
  		entry->rxb._page = rxb_steal_page(rxb);
  		entry->rxb._offset = rxb->_offset;
  		entry->rxb._rx_page_order = rxb->_rx_page_order;
  		entry->fn = rx_h->fn;
  		spin_lock(&mvm->async_handlers_lock);
  		list_add_tail(&entry->list, &mvm->async_handlers_list);
  		spin_unlock(&mvm->async_handlers_lock);
  		schedule_work(&mvm->async_handlers_wk);
  		break;
  	}
  }
  
  static void iwl_mvm_rx(struct iwl_op_mode *op_mode,
  		       struct napi_struct *napi,
  		       struct iwl_rx_cmd_buffer *rxb)
  {
  	struct iwl_rx_packet *pkt = rxb_addr(rxb);
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  
  	if (likely(pkt->hdr.cmd == REPLY_RX_MPDU_CMD))
  		iwl_mvm_rx_rx_mpdu(mvm, napi, rxb);
  	else if (pkt->hdr.cmd == REPLY_RX_PHY_CMD)
  		iwl_mvm_rx_rx_phy_cmd(mvm, rxb);
  	else
  		iwl_mvm_rx_common(mvm, rxb, pkt);
  }
  
  static void iwl_mvm_rx_mq(struct iwl_op_mode *op_mode,
  			  struct napi_struct *napi,
  			  struct iwl_rx_cmd_buffer *rxb)
  {
  	struct iwl_rx_packet *pkt = rxb_addr(rxb);
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  
  	if (likely(pkt->hdr.cmd == REPLY_RX_MPDU_CMD))
  		iwl_mvm_rx_rx_mpdu(mvm, napi, rxb);
  	else if (pkt->hdr.cmd == REPLY_RX_PHY_CMD)
  		iwl_mvm_rx_rx_phy_cmd(mvm, rxb);
  	else
  		iwl_mvm_rx_common(mvm, rxb, pkt);
  }
  
  static void iwl_mvm_stop_sw_queue(struct iwl_op_mode *op_mode, int queue)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  	unsigned long mq;
  	int q;
  
  	spin_lock_bh(&mvm->queue_info_lock);
  	mq = mvm->queue_info[queue].hw_queue_to_mac80211;
  	spin_unlock_bh(&mvm->queue_info_lock);
  
  	if (WARN_ON_ONCE(!mq))
  		return;
  
  	for_each_set_bit(q, &mq, IEEE80211_MAX_QUEUES) {
  		if (atomic_inc_return(&mvm->mac80211_queue_stop_count[q]) > 1) {
  			IWL_DEBUG_TX_QUEUES(mvm,
  					    "queue %d (mac80211 %d) already stopped
  ",
  					    queue, q);
  			continue;
  		}
  
  		ieee80211_stop_queue(mvm->hw, q);
  	}
  }
  
  static void iwl_mvm_wake_sw_queue(struct iwl_op_mode *op_mode, int queue)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  	unsigned long mq;
  	int q;
  
  	spin_lock_bh(&mvm->queue_info_lock);
  	mq = mvm->queue_info[queue].hw_queue_to_mac80211;
  	spin_unlock_bh(&mvm->queue_info_lock);
  
  	if (WARN_ON_ONCE(!mq))
  		return;
  
  	for_each_set_bit(q, &mq, IEEE80211_MAX_QUEUES) {
  		if (atomic_dec_return(&mvm->mac80211_queue_stop_count[q]) > 0) {
  			IWL_DEBUG_TX_QUEUES(mvm,
  					    "queue %d (mac80211 %d) still stopped
  ",
  					    queue, q);
  			continue;
  		}
  
  		ieee80211_wake_queue(mvm->hw, q);
  	}
  }
  
  void iwl_mvm_set_hw_ctkill_state(struct iwl_mvm *mvm, bool state)
  {
  	if (state)
  		set_bit(IWL_MVM_STATUS_HW_CTKILL, &mvm->status);
  	else
  		clear_bit(IWL_MVM_STATUS_HW_CTKILL, &mvm->status);
  
  	wiphy_rfkill_set_hw_state(mvm->hw->wiphy, iwl_mvm_is_radio_killed(mvm));
  }
  
  static bool iwl_mvm_set_hw_rfkill_state(struct iwl_op_mode *op_mode, bool state)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  	bool calibrating = ACCESS_ONCE(mvm->calibrating);
  
  	if (state)
  		set_bit(IWL_MVM_STATUS_HW_RFKILL, &mvm->status);
  	else
  		clear_bit(IWL_MVM_STATUS_HW_RFKILL, &mvm->status);
  
  	wiphy_rfkill_set_hw_state(mvm->hw->wiphy, iwl_mvm_is_radio_killed(mvm));
  
  	/* iwl_run_init_mvm_ucode is waiting for results, abort it */
  	if (calibrating)
  		iwl_abort_notification_waits(&mvm->notif_wait);
  
  	/*
  	 * Stop the device if we run OPERATIONAL firmware or if we are in the
  	 * middle of the calibrations.
  	 */
  	return state && (mvm->cur_ucode != IWL_UCODE_INIT || calibrating);
  }
  
  static void iwl_mvm_free_skb(struct iwl_op_mode *op_mode, struct sk_buff *skb)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  	struct ieee80211_tx_info *info;
  
  	info = IEEE80211_SKB_CB(skb);
  	iwl_trans_free_tx_cmd(mvm->trans, info->driver_data[1]);
  	ieee80211_free_txskb(mvm->hw, skb);
  }
  
  struct iwl_mvm_reprobe {
  	struct device *dev;
  	struct work_struct work;
  };
  
  static void iwl_mvm_reprobe_wk(struct work_struct *wk)
  {
  	struct iwl_mvm_reprobe *reprobe;
  
  	reprobe = container_of(wk, struct iwl_mvm_reprobe, work);
  	if (device_reprobe(reprobe->dev))
  		dev_err(reprobe->dev, "reprobe failed!
  ");
  	kfree(reprobe);
  	module_put(THIS_MODULE);
  }
  
  static void iwl_mvm_fw_error_dump_wk(struct work_struct *work)
  {
  	struct iwl_mvm *mvm =
  		container_of(work, struct iwl_mvm, fw_dump_wk.work);
  
  	if (iwl_mvm_ref_sync(mvm, IWL_MVM_REF_FW_DBG_COLLECT))
  		return;
  
  	mutex_lock(&mvm->mutex);
  
  	/* stop recording */
  	if (mvm->cfg->device_family == IWL_DEVICE_FAMILY_7000) {
  		iwl_set_bits_prph(mvm->trans, MON_BUFF_SAMPLE_CTL, 0x100);
  	} else {
  		iwl_write_prph(mvm->trans, DBGC_IN_SAMPLE, 0);
  		/* wait before we collect the data till the DBGC stop */
  		udelay(100);
  	}
  
  	iwl_mvm_fw_error_dump(mvm);
  
  	/* start recording again if the firmware is not crashed */
  	WARN_ON_ONCE((!test_bit(STATUS_FW_ERROR, &mvm->trans->status)) &&
  		     mvm->fw->dbg_dest_tlv &&
  		     iwl_mvm_start_fw_dbg_conf(mvm, mvm->fw_dbg_conf));
  
  	mutex_unlock(&mvm->mutex);
  
  	iwl_mvm_unref(mvm, IWL_MVM_REF_FW_DBG_COLLECT);
  }
  
  void iwl_mvm_nic_restart(struct iwl_mvm *mvm, bool fw_error)
  {
  	iwl_abort_notification_waits(&mvm->notif_wait);
  
  	/*
  	 * This is a bit racy, but worst case we tell mac80211 about
  	 * a stopped/aborted scan when that was already done which
  	 * is not a problem. It is necessary to abort any os scan
  	 * here because mac80211 requires having the scan cleared
  	 * before restarting.
  	 * We'll reset the scan_status to NONE in restart cleanup in
  	 * the next start() call from mac80211. If restart isn't called
  	 * (no fw restart) scan status will stay busy.
  	 */
  	iwl_mvm_report_scan_aborted(mvm);
  
  	/*
  	 * If we're restarting already, don't cycle restarts.
  	 * If INIT fw asserted, it will likely fail again.
  	 * If WoWLAN fw asserted, don't restart either, mac80211
  	 * can't recover this since we're already half suspended.
  	 */
  	if (!mvm->restart_fw && fw_error) {
  		iwl_mvm_fw_dbg_collect_desc(mvm, &iwl_mvm_dump_desc_assert,
  					    NULL);
  	} else if (test_and_set_bit(IWL_MVM_STATUS_IN_HW_RESTART,
  				    &mvm->status)) {
  		struct iwl_mvm_reprobe *reprobe;
  
  		IWL_ERR(mvm,
  			"Firmware error during reconfiguration - reprobe!
  ");
  
  		/*
  		 * get a module reference to avoid doing this while unloading
  		 * anyway and to avoid scheduling a work with code that's
  		 * being removed.
  		 */
  		if (!try_module_get(THIS_MODULE)) {
  			IWL_ERR(mvm, "Module is being unloaded - abort
  ");
  			return;
  		}
  
  		reprobe = kzalloc(sizeof(*reprobe), GFP_ATOMIC);
  		if (!reprobe) {
  			module_put(THIS_MODULE);
  			return;
  		}
  		reprobe->dev = mvm->trans->dev;
  		INIT_WORK(&reprobe->work, iwl_mvm_reprobe_wk);
  		schedule_work(&reprobe->work);
  	} else if (mvm->cur_ucode == IWL_UCODE_REGULAR) {
  		/* don't let the transport/FW power down */
  		iwl_mvm_ref(mvm, IWL_MVM_REF_UCODE_DOWN);
  
  		if (fw_error && mvm->restart_fw > 0)
  			mvm->restart_fw--;
  		ieee80211_restart_hw(mvm->hw);
  	}
  }
  
  static void iwl_mvm_nic_error(struct iwl_op_mode *op_mode)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  
  	iwl_mvm_dump_nic_error_log(mvm);
  
  	iwl_mvm_nic_restart(mvm, true);
  }
  
  static void iwl_mvm_cmd_queue_full(struct iwl_op_mode *op_mode)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  
  	WARN_ON(1);
  	iwl_mvm_nic_restart(mvm, true);
  }
  
  struct iwl_d0i3_iter_data {
  	struct iwl_mvm *mvm;
  	u8 ap_sta_id;
  	u8 vif_count;
  	u8 offloading_tid;
  	bool disable_offloading;
  };
  
  static bool iwl_mvm_disallow_offloading(struct iwl_mvm *mvm,
  					struct ieee80211_vif *vif,
  					struct iwl_d0i3_iter_data *iter_data)
  {
  	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
  	struct ieee80211_sta *ap_sta;
  	struct iwl_mvm_sta *mvmsta;
  	u32 available_tids = 0;
  	u8 tid;
  
  	if (WARN_ON(vif->type != NL80211_IFTYPE_STATION ||
  		    mvmvif->ap_sta_id == IWL_MVM_STATION_COUNT))
  		return false;
  
  	ap_sta = rcu_dereference(mvm->fw_id_to_mac_id[mvmvif->ap_sta_id]);
  	if (IS_ERR_OR_NULL(ap_sta))
  		return false;
  
  	mvmsta = iwl_mvm_sta_from_mac80211(ap_sta);
  	spin_lock_bh(&mvmsta->lock);
  	for (tid = 0; tid < IWL_MAX_TID_COUNT; tid++) {
  		struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
  
  		/*
  		 * in case of pending tx packets, don't use this tid
  		 * for offloading in order to prevent reuse of the same
  		 * qos seq counters.
  		 */
  		if (iwl_mvm_tid_queued(tid_data))
  			continue;
  
  		if (tid_data->state != IWL_AGG_OFF)
  			continue;
  
  		available_tids |= BIT(tid);
  	}
  	spin_unlock_bh(&mvmsta->lock);
  
  	/*
  	 * disallow protocol offloading if we have no available tid
  	 * (with no pending frames and no active aggregation,
  	 * as we don't handle "holes" properly - the scheduler needs the
  	 * frame's seq number and TFD index to match)
  	 */
  	if (!available_tids)
  		return true;
  
  	/* for simplicity, just use the first available tid */
  	iter_data->offloading_tid = ffs(available_tids) - 1;
  	return false;
  }
  
  static void iwl_mvm_enter_d0i3_iterator(void *_data, u8 *mac,
  					struct ieee80211_vif *vif)
  {
  	struct iwl_d0i3_iter_data *data = _data;
  	struct iwl_mvm *mvm = data->mvm;
  	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
  	u32 flags = CMD_ASYNC | CMD_HIGH_PRIO | CMD_SEND_IN_IDLE;
  
  	IWL_DEBUG_RPM(mvm, "entering D0i3 - vif %pM
  ", vif->addr);
  	if (vif->type != NL80211_IFTYPE_STATION ||
  	    !vif->bss_conf.assoc)
  		return;
  
  	/*
  	 * in case of pending tx packets or active aggregations,
  	 * avoid offloading features in order to prevent reuse of
  	 * the same qos seq counters.
  	 */
  	if (iwl_mvm_disallow_offloading(mvm, vif, data))
  		data->disable_offloading = true;
  
  	iwl_mvm_update_d0i3_power_mode(mvm, vif, true, flags);
  	iwl_mvm_send_proto_offload(mvm, vif, data->disable_offloading, flags);
  
  	/*
  	 * on init/association, mvm already configures POWER_TABLE_CMD
  	 * and REPLY_MCAST_FILTER_CMD, so currently don't
  	 * reconfigure them (we might want to use different
  	 * params later on, though).
  	 */
  	data->ap_sta_id = mvmvif->ap_sta_id;
  	data->vif_count++;
  }
  
  static void iwl_mvm_set_wowlan_data(struct iwl_mvm *mvm,
  				    struct iwl_wowlan_config_cmd *cmd,
  				    struct iwl_d0i3_iter_data *iter_data)
  {
  	struct ieee80211_sta *ap_sta;
  	struct iwl_mvm_sta *mvm_ap_sta;
  
  	if (iter_data->ap_sta_id == IWL_MVM_STATION_COUNT)
  		return;
  
  	rcu_read_lock();
  
  	ap_sta = rcu_dereference(mvm->fw_id_to_mac_id[iter_data->ap_sta_id]);
  	if (IS_ERR_OR_NULL(ap_sta))
  		goto out;
  
  	mvm_ap_sta = iwl_mvm_sta_from_mac80211(ap_sta);
  	cmd->is_11n_connection = ap_sta->ht_cap.ht_supported;
  	cmd->offloading_tid = iter_data->offloading_tid;
  
  	/*
  	 * The d0i3 uCode takes care of the nonqos counters,
  	 * so configure only the qos seq ones.
  	 */
  	iwl_mvm_set_wowlan_qos_seq(mvm_ap_sta, cmd);
  out:
  	rcu_read_unlock();
  }
  
  int iwl_mvm_enter_d0i3(struct iwl_op_mode *op_mode)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  	u32 flags = CMD_ASYNC | CMD_HIGH_PRIO | CMD_SEND_IN_IDLE;
  	int ret;
  	struct iwl_d0i3_iter_data d0i3_iter_data = {
  		.mvm = mvm,
  	};
  	struct iwl_wowlan_config_cmd wowlan_config_cmd = {
  		.wakeup_filter = cpu_to_le32(IWL_WOWLAN_WAKEUP_RX_FRAME |
  					     IWL_WOWLAN_WAKEUP_BEACON_MISS |
  					     IWL_WOWLAN_WAKEUP_LINK_CHANGE |
  					     IWL_WOWLAN_WAKEUP_BCN_FILTERING),
  	};
  	struct iwl_d3_manager_config d3_cfg_cmd = {
  		.min_sleep_time = cpu_to_le32(1000),
  		.wakeup_flags = cpu_to_le32(IWL_WAKEUP_D3_CONFIG_FW_ERROR),
  	};
  
  	IWL_DEBUG_RPM(mvm, "MVM entering D0i3
  ");
  
  	set_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status);
  
  	/*
  	 * iwl_mvm_ref_sync takes a reference before checking the flag.
  	 * so by checking there is no held reference we prevent a state
  	 * in which iwl_mvm_ref_sync continues successfully while we
  	 * configure the firmware to enter d0i3
  	 */
  	if (iwl_mvm_ref_taken(mvm)) {
  		IWL_DEBUG_RPM(mvm->trans, "abort d0i3 due to taken ref
  ");
  		clear_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status);
  		wake_up(&mvm->d0i3_exit_waitq);
  		return 1;
  	}
  
  	ieee80211_iterate_active_interfaces_atomic(mvm->hw,
  						   IEEE80211_IFACE_ITER_NORMAL,
  						   iwl_mvm_enter_d0i3_iterator,
  						   &d0i3_iter_data);
  	if (d0i3_iter_data.vif_count == 1) {
  		mvm->d0i3_ap_sta_id = d0i3_iter_data.ap_sta_id;
  		mvm->d0i3_offloading = !d0i3_iter_data.disable_offloading;
  	} else {
  		WARN_ON_ONCE(d0i3_iter_data.vif_count > 1);
  		mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
  		mvm->d0i3_offloading = false;
  	}
  
  	/* make sure we have no running tx while configuring the seqno */
  	synchronize_net();
  
  	/* configure wowlan configuration only if needed */
  	if (mvm->d0i3_ap_sta_id != IWL_MVM_STATION_COUNT) {
  		iwl_mvm_set_wowlan_data(mvm, &wowlan_config_cmd,
  					&d0i3_iter_data);
  
  		ret = iwl_mvm_send_cmd_pdu(mvm, WOWLAN_CONFIGURATION, flags,
  					   sizeof(wowlan_config_cmd),
  					   &wowlan_config_cmd);
  		if (ret)
  			return ret;
  	}
  
  	return iwl_mvm_send_cmd_pdu(mvm, D3_CONFIG_CMD,
  				    flags | CMD_MAKE_TRANS_IDLE,
  				    sizeof(d3_cfg_cmd), &d3_cfg_cmd);
  }
  
  static void iwl_mvm_exit_d0i3_iterator(void *_data, u8 *mac,
  				       struct ieee80211_vif *vif)
  {
  	struct iwl_mvm *mvm = _data;
  	u32 flags = CMD_ASYNC | CMD_HIGH_PRIO;
  
  	IWL_DEBUG_RPM(mvm, "exiting D0i3 - vif %pM
  ", vif->addr);
  	if (vif->type != NL80211_IFTYPE_STATION ||
  	    !vif->bss_conf.assoc)
  		return;
  
  	iwl_mvm_update_d0i3_power_mode(mvm, vif, false, flags);
  }
  
  struct iwl_mvm_wakeup_reason_iter_data {
  	struct iwl_mvm *mvm;
  	u32 wakeup_reasons;
  };
  
  static void iwl_mvm_d0i3_wakeup_reason_iter(void *_data, u8 *mac,
  					    struct ieee80211_vif *vif)
  {
  	struct iwl_mvm_wakeup_reason_iter_data *data = _data;
  	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
  
  	if (vif->type == NL80211_IFTYPE_STATION && vif->bss_conf.assoc &&
  	    data->mvm->d0i3_ap_sta_id == mvmvif->ap_sta_id) {
  		if (data->wakeup_reasons &
  		    IWL_WOWLAN_WAKEUP_BY_DISCONNECTION_ON_DEAUTH)
  			iwl_mvm_connection_loss(data->mvm, vif, "D0i3");
  		else
  			ieee80211_beacon_loss(vif);
  	}
  }
  
  void iwl_mvm_d0i3_enable_tx(struct iwl_mvm *mvm, __le16 *qos_seq)
  {
  	struct ieee80211_sta *sta = NULL;
  	struct iwl_mvm_sta *mvm_ap_sta;
  	int i;
  	bool wake_queues = false;
  
  	lockdep_assert_held(&mvm->mutex);
  
  	spin_lock_bh(&mvm->d0i3_tx_lock);
  
  	if (mvm->d0i3_ap_sta_id == IWL_MVM_STATION_COUNT)
  		goto out;
  
  	IWL_DEBUG_RPM(mvm, "re-enqueue packets
  ");
  
  	/* get the sta in order to update seq numbers and re-enqueue skbs */
  	sta = rcu_dereference_protected(
  			mvm->fw_id_to_mac_id[mvm->d0i3_ap_sta_id],
  			lockdep_is_held(&mvm->mutex));
  
  	if (IS_ERR_OR_NULL(sta)) {
  		sta = NULL;
  		goto out;
  	}
  
  	if (mvm->d0i3_offloading && qos_seq) {
  		/* update qos seq numbers if offloading was enabled */
  		mvm_ap_sta = iwl_mvm_sta_from_mac80211(sta);
  		for (i = 0; i < IWL_MAX_TID_COUNT; i++) {
  			u16 seq = le16_to_cpu(qos_seq[i]);
  			/* firmware stores last-used one, we store next one */
  			seq += 0x10;
  			mvm_ap_sta->tid_data[i].seq_number = seq;
  		}
  	}
  out:
  	/* re-enqueue (or drop) all packets */
  	while (!skb_queue_empty(&mvm->d0i3_tx)) {
  		struct sk_buff *skb = __skb_dequeue(&mvm->d0i3_tx);
  
  		if (!sta || iwl_mvm_tx_skb(mvm, skb, sta))
  			ieee80211_free_txskb(mvm->hw, skb);
  
  		/* if the skb_queue is not empty, we need to wake queues */
  		wake_queues = true;
  	}
  	clear_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status);
  	wake_up(&mvm->d0i3_exit_waitq);
  	mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
  	if (wake_queues)
  		ieee80211_wake_queues(mvm->hw);
  
  	spin_unlock_bh(&mvm->d0i3_tx_lock);
  }
  
  static void iwl_mvm_d0i3_exit_work(struct work_struct *wk)
  {
  	struct iwl_mvm *mvm = container_of(wk, struct iwl_mvm, d0i3_exit_work);
  	struct iwl_host_cmd get_status_cmd = {
  		.id = WOWLAN_GET_STATUSES,
  		.flags = CMD_HIGH_PRIO | CMD_WANT_SKB,
  	};
  	struct iwl_wowlan_status *status;
  	int ret;
  	u32 handled_reasons, wakeup_reasons = 0;
  	__le16 *qos_seq = NULL;
  
  	mutex_lock(&mvm->mutex);
  	ret = iwl_mvm_send_cmd(mvm, &get_status_cmd);
  	if (ret)
  		goto out;
  
  	if (!get_status_cmd.resp_pkt)
  		goto out;
  
  	status = (void *)get_status_cmd.resp_pkt->data;
  	wakeup_reasons = le32_to_cpu(status->wakeup_reasons);
  	qos_seq = status->qos_seq_ctr;
  
  	IWL_DEBUG_RPM(mvm, "wakeup reasons: 0x%x
  ", wakeup_reasons);
  
  	handled_reasons = IWL_WOWLAN_WAKEUP_BY_DISCONNECTION_ON_MISSED_BEACON |
  				IWL_WOWLAN_WAKEUP_BY_DISCONNECTION_ON_DEAUTH;
  	if (wakeup_reasons & handled_reasons) {
  		struct iwl_mvm_wakeup_reason_iter_data data = {
  			.mvm = mvm,
  			.wakeup_reasons = wakeup_reasons,
  		};
  
  		ieee80211_iterate_active_interfaces(
  			mvm->hw, IEEE80211_IFACE_ITER_NORMAL,
  			iwl_mvm_d0i3_wakeup_reason_iter, &data);
  	}
  out:
  	iwl_mvm_d0i3_enable_tx(mvm, qos_seq);
  
  	IWL_DEBUG_INFO(mvm, "d0i3 exit completed (wakeup reasons: 0x%x)
  ",
  		       wakeup_reasons);
  
  	/* qos_seq might point inside resp_pkt, so free it only now */
  	if (get_status_cmd.resp_pkt)
  		iwl_free_resp(&get_status_cmd);
  
  	/* the FW might have updated the regdomain */
  	iwl_mvm_update_changed_regdom(mvm);
  
  	iwl_mvm_unref(mvm, IWL_MVM_REF_EXIT_WORK);
  	mutex_unlock(&mvm->mutex);
  }
  
  int _iwl_mvm_exit_d0i3(struct iwl_mvm *mvm)
  {
  	u32 flags = CMD_ASYNC | CMD_HIGH_PRIO | CMD_SEND_IN_IDLE |
  		    CMD_WAKE_UP_TRANS;
  	int ret;
  
  	IWL_DEBUG_RPM(mvm, "MVM exiting D0i3
  ");
  
  	mutex_lock(&mvm->d0i3_suspend_mutex);
  	if (test_bit(D0I3_DEFER_WAKEUP, &mvm->d0i3_suspend_flags)) {
  		IWL_DEBUG_RPM(mvm, "Deferring d0i3 exit until resume
  ");
  		__set_bit(D0I3_PENDING_WAKEUP, &mvm->d0i3_suspend_flags);
  		mutex_unlock(&mvm->d0i3_suspend_mutex);
  		return 0;
  	}
  	mutex_unlock(&mvm->d0i3_suspend_mutex);
  
  	ret = iwl_mvm_send_cmd_pdu(mvm, D0I3_END_CMD, flags, 0, NULL);
  	if (ret)
  		goto out;
  
  	ieee80211_iterate_active_interfaces_atomic(mvm->hw,
  						   IEEE80211_IFACE_ITER_NORMAL,
  						   iwl_mvm_exit_d0i3_iterator,
  						   mvm);
  out:
  	schedule_work(&mvm->d0i3_exit_work);
  	return ret;
  }
  
  int iwl_mvm_exit_d0i3(struct iwl_op_mode *op_mode)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  
  	iwl_mvm_ref(mvm, IWL_MVM_REF_EXIT_WORK);
  	return _iwl_mvm_exit_d0i3(mvm);
  }
  
  #define IWL_MVM_COMMON_OPS					\
  	/* these could be differentiated */			\
  	.queue_full = iwl_mvm_stop_sw_queue,			\
  	.queue_not_full = iwl_mvm_wake_sw_queue,		\
  	.hw_rf_kill = iwl_mvm_set_hw_rfkill_state,		\
  	.free_skb = iwl_mvm_free_skb,				\
  	.nic_error = iwl_mvm_nic_error,				\
  	.cmd_queue_full = iwl_mvm_cmd_queue_full,		\
  	.nic_config = iwl_mvm_nic_config,			\
  	.enter_d0i3 = iwl_mvm_enter_d0i3,			\
  	.exit_d0i3 = iwl_mvm_exit_d0i3,				\
  	/* as we only register one, these MUST be common! */	\
  	.start = iwl_op_mode_mvm_start,				\
  	.stop = iwl_op_mode_mvm_stop
  
  static const struct iwl_op_mode_ops iwl_mvm_ops = {
  	IWL_MVM_COMMON_OPS,
  	.rx = iwl_mvm_rx,
  };
  
  static void iwl_mvm_rx_mq_rss(struct iwl_op_mode *op_mode,
  			      struct napi_struct *napi,
  			      struct iwl_rx_cmd_buffer *rxb,
  			      unsigned int queue)
  {
  	struct iwl_mvm *mvm = IWL_OP_MODE_GET_MVM(op_mode);
  
  	iwl_mvm_rx_rx_mpdu(mvm, napi, rxb);
  }
  
  static const struct iwl_op_mode_ops iwl_mvm_ops_mq = {
  	IWL_MVM_COMMON_OPS,
  	.rx = iwl_mvm_rx_mq,
  	.rx_rss = iwl_mvm_rx_mq_rss,
  };