iwlwifi: mvm: take scan ref only on success

In some cases, scan reference was taken, but wasn't
released even though scan command wasn't actually issued.

Change the current code to simply take the reference
only on success.

Signed-off-by: Eliad Peller <eliadx.peller@intel.com>
Signed-off-by: Emmanuel Grumbach <emmanuel.grumbach@intel.com>

1 files changed, 6 insertions, 9 deletions

diff --git a/drivers/net/wireless/iwlwifi/mvm/scan.c b/drivers/net/wireless/iwlwifi/mvm/scan.c
index 0e9d28c23023..d6e0c1b5c20c 100644
--- a/drivers/net/wireless/iwlwifi/mvm/scan.c
+++ b/drivers/net/wireless/iwlwifi/mvm/scan.c
@@ -1190,8 +1190,6 @@ int iwl_mvm_reg_scan_start(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
         if (ret)
                 return ret;
 
-        iwl_mvm_ref(mvm, IWL_MVM_REF_SCAN);
-
         /* we should have failed registration if scan_cmd was NULL */
         if (WARN_ON(!mvm->scan_cmd))
                 return -ENOMEM;
@@ -1232,21 +1230,20 @@ int iwl_mvm_reg_scan_start(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
                 return ret;
 
         ret = iwl_mvm_send_cmd(mvm, &hcmd);
-        if (!ret) {
-                IWL_DEBUG_SCAN(mvm, "Scan request was sent successfully\n");
-                mvm->scan_status |= IWL_MVM_SCAN_REGULAR;
-        } else {
+        if (ret) {
                 /* If the scan failed, it usually means that the FW was unable
                  * to allocate the time events. Warn on it, but maybe we
                  * should try to send the command again with different params.
                  */
                 IWL_ERR(mvm, "Scan failed! ret %d\n", ret);
+                return ret;
         }
 
-        if (ret)
-                iwl_mvm_unref(mvm, IWL_MVM_REF_SCAN);
+        IWL_DEBUG_SCAN(mvm, "Scan request was sent successfully\n");
+        mvm->scan_status |= IWL_MVM_SCAN_REGULAR;
+        iwl_mvm_ref(mvm, IWL_MVM_REF_SCAN);
 
-        return ret;
+        return 0;
 }
 
 int iwl_mvm_sched_scan_start(struct iwl_mvm *mvm,