From: Yue Hu <huy...@yulong.com>

There are similar code implemetentions for WB configurations in
ufshcd_wb_{ctrl, toggle_flush_during_h8, toggle_flush}. We can
extract the part to create a new helper with a flag parameter to
reduce code duplication.

Meanwhile, change ufshcd_wb_ctrl() -> ufshcd_wb_toggle() for better
readability.

And remove unnecessary log messages from ufshcd_wb_config() since
relevant toggle function will spew log respectively. Also change
ufshcd_wb_toggle_flush{__during_h8} to void type accordingly.

Signed-off-by: Yue Hu <huy...@yulong.com>
---
 drivers/scsi/ufs/ufs-sysfs.c |  2 +-
 drivers/scsi/ufs/ufshcd.c    | 99 +++++++++++++++++++-------------------------
 drivers/scsi/ufs/ufshcd.h    |  2 +-
 3 files changed, 44 insertions(+), 59 deletions(-)

diff --git a/drivers/scsi/ufs/ufs-sysfs.c b/drivers/scsi/ufs/ufs-sysfs.c
index acc54f5..d7c3cff 100644
--- a/drivers/scsi/ufs/ufs-sysfs.c
+++ b/drivers/scsi/ufs/ufs-sysfs.c
@@ -246,7 +246,7 @@ static ssize_t wb_on_store(struct device *dev, struct 
device_attribute *attr,
        }
 
        pm_runtime_get_sync(hba->dev);
-       res = ufshcd_wb_ctrl(hba, wb_enable);
+       res = ufshcd_wb_toggle(hba, wb_enable);
        pm_runtime_put_sync(hba->dev);
 out:
        up(&hba->host_sem);
diff --git a/drivers/scsi/ufs/ufshcd.c b/drivers/scsi/ufs/ufshcd.c
index 7716175..1368f9e 100644
--- a/drivers/scsi/ufs/ufshcd.c
+++ b/drivers/scsi/ufs/ufshcd.c
@@ -247,8 +247,8 @@ static int ufshcd_change_power_mode(struct ufs_hba *hba,
 static inline int ufshcd_config_vreg_hpm(struct ufs_hba *hba,
                                         struct ufs_vreg *vreg);
 static int ufshcd_try_to_abort_task(struct ufs_hba *hba, int tag);
-static int ufshcd_wb_toggle_flush_during_h8(struct ufs_hba *hba, bool set);
-static inline int ufshcd_wb_toggle_flush(struct ufs_hba *hba, bool enable);
+static void ufshcd_wb_toggle_flush_during_h8(struct ufs_hba *hba, bool set);
+static inline void ufshcd_wb_toggle_flush(struct ufs_hba *hba, bool enable);
 static void ufshcd_hba_vreg_set_lpm(struct ufs_hba *hba);
 static void ufshcd_hba_vreg_set_hpm(struct ufs_hba *hba);
 
@@ -275,20 +275,12 @@ static inline void ufshcd_disable_irq(struct ufs_hba *hba)
 
 static inline void ufshcd_wb_config(struct ufs_hba *hba)
 {
-       int ret;
-
        if (!ufshcd_is_wb_allowed(hba))
                return;
 
-       ret = ufshcd_wb_ctrl(hba, true);
-       if (ret)
-               dev_err(hba->dev, "%s: Enable WB failed: %d\n", __func__, ret);
-       else
-               dev_info(hba->dev, "%s: Write Booster Configured\n", __func__);
-       ret = ufshcd_wb_toggle_flush_during_h8(hba, true);
-       if (ret)
-               dev_err(hba->dev, "%s: En WB flush during H8: failed: %d\n",
-                       __func__, ret);
+       ufshcd_wb_toggle(hba, true);
+
+       ufshcd_wb_toggle_flush_during_h8(hba, true);
        if (!(hba->quirks & UFSHCI_QUIRK_SKIP_MANUAL_WB_FLUSH_CTRL))
                ufshcd_wb_toggle_flush(hba, true);
 }
@@ -1268,7 +1260,7 @@ static int ufshcd_devfreq_scale(struct ufs_hba *hba, bool 
scale_up)
        /* Enable Write Booster if we have scaled up else disable it */
        downgrade_write(&hba->clk_scaling_lock);
        is_writelock = false;
-       ufshcd_wb_ctrl(hba, scale_up);
+       ufshcd_wb_toggle(hba, scale_up);
 
 out_unprepare:
        ufshcd_clock_scaling_unprepare(hba, is_writelock);
@@ -5432,85 +5424,78 @@ static void ufshcd_bkops_exception_event_handler(struct 
ufs_hba *hba)
                                __func__, err);
 }
 
-int ufshcd_wb_ctrl(struct ufs_hba *hba, bool enable)
+static int __ufshcd_wb_toggle(struct ufs_hba *hba, bool set, enum flag_idn idn)
 {
-       int ret;
+       int val;
        u8 index;
-       enum query_opcode opcode;
+
+       if (set)
+               val = UPIU_QUERY_OPCODE_SET_FLAG;
+       else
+               val = UPIU_QUERY_OPCODE_CLEAR_FLAG;
+
+       index = ufshcd_wb_get_query_index(hba);
+       return ufshcd_query_flag_retry(hba, val, idn, index, NULL);
+}
+
+int ufshcd_wb_toggle(struct ufs_hba *hba, bool enable)
+{
+       int ret;
 
        if (!ufshcd_is_wb_allowed(hba))
                return 0;
 
        if (!(enable ^ hba->dev_info.wb_enabled))
                return 0;
-       if (enable)
-               opcode = UPIU_QUERY_OPCODE_SET_FLAG;
-       else
-               opcode = UPIU_QUERY_OPCODE_CLEAR_FLAG;
 
-       index = ufshcd_wb_get_query_index(hba);
-       ret = ufshcd_query_flag_retry(hba, opcode,
-                                     QUERY_FLAG_IDN_WB_EN, index, NULL);
+       ret = __ufshcd_wb_toggle(hba, enable, QUERY_FLAG_IDN_WB_EN);
        if (ret) {
-               dev_err(hba->dev, "%s write booster %s failed %d\n",
+               dev_err(hba->dev, "%s Write Booster %s failed %d\n",
                        __func__, enable ? "enable" : "disable", ret);
                return ret;
        }
 
        hba->dev_info.wb_enabled = enable;
-       dev_dbg(hba->dev, "%s write booster %s %d\n",
-                       __func__, enable ? "enable" : "disable", ret);
+       dev_info(hba->dev, "%s Write Booster %s\n",
+                       __func__, enable ? "enabled" : "disabled");
 
        return ret;
 }
 
-static int ufshcd_wb_toggle_flush_during_h8(struct ufs_hba *hba, bool set)
+static void ufshcd_wb_toggle_flush_during_h8(struct ufs_hba *hba, bool set)
 {
-       int val;
-       u8 index;
-
-       if (set)
-               val =  UPIU_QUERY_OPCODE_SET_FLAG;
-       else
-               val = UPIU_QUERY_OPCODE_CLEAR_FLAG;
+       int ret;
 
-       index = ufshcd_wb_get_query_index(hba);
-       return ufshcd_query_flag_retry(hba, val,
-                               QUERY_FLAG_IDN_WB_BUFF_FLUSH_DURING_HIBERN8,
-                               index, NULL);
+       ret = __ufshcd_wb_toggle(hba, set,
+                       QUERY_FLAG_IDN_WB_BUFF_FLUSH_DURING_HIBERN8);
+       if (ret) {
+               dev_err(hba->dev, "%s: WB-Buf Flush during H8 %s failed: %d\n",
+                       __func__, set ? "enable" : "disable", ret);
+               return;
+       }
+       dev_dbg(hba->dev, "%s WB-Buf Flush during H8 %s\n",
+                       __func__, set ? "enabled" : "disabled");
 }
 
-static inline int ufshcd_wb_toggle_flush(struct ufs_hba *hba, bool enable)
+static inline void ufshcd_wb_toggle_flush(struct ufs_hba *hba, bool enable)
 {
        int ret;
-       u8 index;
-       enum query_opcode opcode;
 
        if (!ufshcd_is_wb_allowed(hba) ||
            hba->dev_info.wb_buf_flush_enabled == enable)
-               return 0;
-
-       if (enable)
-               opcode = UPIU_QUERY_OPCODE_SET_FLAG;
-       else
-               opcode = UPIU_QUERY_OPCODE_CLEAR_FLAG;
+               return;
 
-       index = ufshcd_wb_get_query_index(hba);
-       ret = ufshcd_query_flag_retry(hba, opcode,
-                                     QUERY_FLAG_IDN_WB_BUFF_FLUSH_EN, index,
-                                     NULL);
+       ret = __ufshcd_wb_toggle(hba, enable, QUERY_FLAG_IDN_WB_BUFF_FLUSH_EN);
        if (ret) {
                dev_err(hba->dev, "%s WB-Buf Flush %s failed %d\n", __func__,
                        enable ? "enable" : "disable", ret);
-               goto out;
+               return;
        }
 
        hba->dev_info.wb_buf_flush_enabled = enable;
 
-       dev_dbg(hba->dev, "WB-Buf Flush %s\n", enable ? "enabled" : "disabled");
-out:
-       return ret;
-
+       dev_dbg(hba->dev, "%s WB-Buf Flush %s\n",
+                       __func__, enable ? "enabled" : "disabled");
 }
 
 static bool ufshcd_wb_presrv_usrspc_keep_vcc_on(struct ufs_hba *hba,
diff --git a/drivers/scsi/ufs/ufshcd.h b/drivers/scsi/ufs/ufshcd.h
index 18e56c1..eddc819 100644
--- a/drivers/scsi/ufs/ufshcd.h
+++ b/drivers/scsi/ufs/ufshcd.h
@@ -1099,7 +1099,7 @@ int ufshcd_exec_raw_upiu_cmd(struct ufs_hba *hba,
                             u8 *desc_buff, int *buff_len,
                             enum query_opcode desc_op);
 
-int ufshcd_wb_ctrl(struct ufs_hba *hba, bool enable);
+int ufshcd_wb_toggle(struct ufs_hba *hba, bool enable);
 
 /* Wrapper functions for safely calling variant operations */
 static inline const char *ufshcd_get_var_name(struct ufs_hba *hba)
-- 
1.9.1

Reply via email to