For WriteBooster feature related attributes, the index used by
query shall be LUN ID if LU Dedicated buffer mode is enabled.

Signed-off-by: Stanley Chu <stanley....@mediatek.com>
---
 drivers/scsi/ufs/ufs-sysfs.c | 13 +++++++++++--
 drivers/scsi/ufs/ufshcd.c    | 16 ++++++++++------
 drivers/scsi/ufs/ufshcd.h    |  2 +-
 3 files changed, 22 insertions(+), 9 deletions(-)

diff --git a/drivers/scsi/ufs/ufs-sysfs.c b/drivers/scsi/ufs/ufs-sysfs.c
index a0b3763e1dc2..2d71d232a69d 100644
--- a/drivers/scsi/ufs/ufs-sysfs.c
+++ b/drivers/scsi/ufs/ufs-sysfs.c
@@ -637,7 +637,7 @@ static ssize_t _name##_show(struct device *dev,             
                \
        int ret;                                                        \
        struct ufs_hba *hba = dev_get_drvdata(dev);                     \
        if (ufshcd_is_wb_flags(QUERY_FLAG_IDN##_uname))                 \
-               index = ufshcd_wb_get_flag_index(hba);                  \
+               index = ufshcd_wb_get_query_index(hba);                 \
        pm_runtime_get_sync(hba->dev);                                  \
        ret = ufshcd_query_flag(hba, UPIU_QUERY_OPCODE_READ_FLAG,       \
                QUERY_FLAG_IDN##_uname, index, &flag);                  \
@@ -680,6 +680,12 @@ static const struct attribute_group ufs_sysfs_flags_group 
= {
        .attrs = ufs_sysfs_device_flags,
 };
 
+static inline bool ufshcd_is_wb_attrs(enum attr_idn idn)
+{
+       return ((idn >= QUERY_ATTR_IDN_WB_FLUSH_STATUS) &&
+               (idn <= QUERY_ATTR_IDN_CURR_WB_BUFF_SIZE));
+}
+
 #define UFS_ATTRIBUTE(_name, _uname)                                   \
 static ssize_t _name##_show(struct device *dev,                                
\
        struct device_attribute *attr, char *buf)                       \
@@ -687,9 +693,12 @@ static ssize_t _name##_show(struct device *dev,            
                \
        struct ufs_hba *hba = dev_get_drvdata(dev);                     \
        u32 value;                                                      \
        int ret;                                                        \
+       u8 index = 0;                                                   \
+       if (ufshcd_is_wb_attrs(QUERY_ATTR_IDN##_uname))                 \
+               index = ufshcd_wb_get_query_index(hba);                 \
        pm_runtime_get_sync(hba->dev);                                  \
        ret = ufshcd_query_attr(hba, UPIU_QUERY_OPCODE_READ_ATTR,       \
-               QUERY_ATTR_IDN##_uname, 0, 0, &value);                  \
+               QUERY_ATTR_IDN##_uname, index, 0, &value);              \
        pm_runtime_put_sync(hba->dev);                                  \
        if (ret)                                                        \
                return -EINVAL;                                         \
diff --git a/drivers/scsi/ufs/ufshcd.c b/drivers/scsi/ufs/ufshcd.c
index f7fa571020da..c7c2bd7860b8 100644
--- a/drivers/scsi/ufs/ufshcd.c
+++ b/drivers/scsi/ufs/ufshcd.c
@@ -5195,7 +5195,7 @@ static int ufshcd_wb_ctrl(struct ufs_hba *hba, bool 
enable)
        else
                opcode = UPIU_QUERY_OPCODE_CLEAR_FLAG;
 
-       index = ufshcd_wb_get_flag_index(hba);
+       index = ufshcd_wb_get_query_index(hba);
        ret = ufshcd_query_flag_retry(hba, opcode,
                                      QUERY_FLAG_IDN_WB_EN, index, NULL);
        if (ret) {
@@ -5221,7 +5221,7 @@ static int ufshcd_wb_toggle_flush_during_h8(struct 
ufs_hba *hba, bool set)
        else
                val = UPIU_QUERY_OPCODE_CLEAR_FLAG;
 
-       index = ufshcd_wb_get_flag_index(hba);
+       index = ufshcd_wb_get_query_index(hba);
        return ufshcd_query_flag_retry(hba, val,
                                QUERY_FLAG_IDN_WB_BUFF_FLUSH_DURING_HIBERN8,
                                index, NULL);
@@ -5244,7 +5244,7 @@ static int ufshcd_wb_buf_flush_enable(struct ufs_hba *hba)
        if (!ufshcd_is_wb_allowed(hba) || hba->wb_buf_flush_enabled)
                return 0;
 
-       index = ufshcd_wb_get_flag_index(hba);
+       index = ufshcd_wb_get_query_index(hba);
        ret = ufshcd_query_flag_retry(hba, UPIU_QUERY_OPCODE_SET_FLAG,
                                      QUERY_FLAG_IDN_WB_BUFF_FLUSH_EN,
                                      index, NULL);
@@ -5266,7 +5266,7 @@ static int ufshcd_wb_buf_flush_disable(struct ufs_hba 
*hba)
        if (!ufshcd_is_wb_allowed(hba) || !hba->wb_buf_flush_enabled)
                return 0;
 
-       index = ufshcd_wb_get_flag_index(hba);
+       index = ufshcd_wb_get_query_index(hba);
        ret = ufshcd_query_flag_retry(hba, UPIU_QUERY_OPCODE_CLEAR_FLAG,
                                      QUERY_FLAG_IDN_WB_BUFF_FLUSH_EN,
                                      index, NULL);
@@ -5286,10 +5286,12 @@ static bool ufshcd_wb_presrv_usrspc_keep_vcc_on(struct 
ufs_hba *hba,
 {
        u32 cur_buf;
        int ret;
+       u8 index;
 
+       index = ufshcd_wb_get_query_index(hba);
        ret = ufshcd_query_attr_retry(hba, UPIU_QUERY_OPCODE_READ_ATTR,
                                              QUERY_ATTR_IDN_CURR_WB_BUFF_SIZE,
-                                             0, 0, &cur_buf);
+                                             index, 0, &cur_buf);
        if (ret) {
                dev_err(hba->dev, "%s dCurWriteBoosterBufferSize read failed 
%d\n",
                        __func__, ret);
@@ -5312,6 +5314,7 @@ static bool ufshcd_wb_keep_vcc_on(struct ufs_hba *hba)
 {
        int ret;
        u32 avail_buf;
+       u8 index;
 
        if (!ufshcd_is_wb_allowed(hba))
                return false;
@@ -5326,9 +5329,10 @@ static bool ufshcd_wb_keep_vcc_on(struct ufs_hba *hba)
         * buffer (dCurrentWriteBoosterBufferSize). There's no point in
         * keeping vcc on when current buffer is empty.
         */
+       index = ufshcd_wb_get_query_index(hba);
        ret = ufshcd_query_attr_retry(hba, UPIU_QUERY_OPCODE_READ_ATTR,
                                      QUERY_ATTR_IDN_AVAIL_WB_BUFF_SIZE,
-                                     0, 0, &avail_buf);
+                                     index, 0, &avail_buf);
        if (ret) {
                dev_warn(hba->dev, "%s dAvailableWriteBoosterBufferSize read 
failed %d\n",
                         __func__, ret);
diff --git a/drivers/scsi/ufs/ufshcd.h b/drivers/scsi/ufs/ufshcd.h
index e3dfb48e669e..8db7a6101892 100644
--- a/drivers/scsi/ufs/ufshcd.h
+++ b/drivers/scsi/ufs/ufshcd.h
@@ -868,7 +868,7 @@ static inline bool 
ufshcd_keep_autobkops_enabled_except_suspend(
        return hba->caps & UFSHCD_CAP_KEEP_AUTO_BKOPS_ENABLED_EXCEPT_SUSPEND;
 }
 
-static inline u8 ufshcd_wb_get_flag_index(struct ufs_hba *hba)
+static inline u8 ufshcd_wb_get_query_index(struct ufs_hba *hba)
 {
        if (hba->dev_info.b_wb_buffer_type == WB_BUF_MODE_LU_DEDICATED)
                return hba->dev_info.wb_dedicated_lu;
-- 
2.18.0

Reply via email to