Allow to read descriptors via raw upiu. This in fact was forbidden just
as a precaution, as ufs-bsg actually enforces which functionality is
supported.

Signed-off-by: Avri Altman <avri.alt...@wdc.com>
Reviewed-by: Evan Green <evgr...@chromium.org>
Reviewed-by: Bean Huo <bean...@micron.com>
---
 drivers/scsi/ufs/ufshcd.c | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/drivers/scsi/ufs/ufshcd.c b/drivers/scsi/ufs/ufshcd.c
index 5ca4581..7b5b601 100644
--- a/drivers/scsi/ufs/ufshcd.c
+++ b/drivers/scsi/ufs/ufshcd.c
@@ -5780,6 +5780,20 @@ static int ufshcd_issue_devman_upiu_cmd(struct ufs_hba 
*hba,
 
        /* just copy the upiu response as it is */
        memcpy(rsp_upiu, lrbp->ucd_rsp_ptr, sizeof(*rsp_upiu));
+       if (desc_buff && desc_op == UPIU_QUERY_OPCODE_READ_DESC) {
+               u8 *descp = (u8 *)lrbp->ucd_rsp_ptr + sizeof(*rsp_upiu);
+               u16 resp_len = be32_to_cpu(lrbp->ucd_rsp_ptr->header.dword_2) &
+                              MASK_QUERY_DATA_SEG_LEN;
+
+               if (*buff_len >= resp_len) {
+                       memcpy(desc_buff, descp, resp_len);
+                       *buff_len = resp_len;
+               } else {
+                       dev_warn(hba->dev, "rsp size is bigger than buffer");
+                       *buff_len = 0;
+                       err = -EINVAL;
+               }
+       }
 
        ufshcd_put_dev_cmd_tag(hba, tag);
        wake_up(&hba->dev_cmd.tag_wq);
@@ -5815,11 +5829,6 @@ int ufshcd_exec_raw_upiu_cmd(struct ufs_hba *hba,
        int ocs_value;
        u8 tm_f = be32_to_cpu(req_upiu->header.dword_1) >> 16 & MASK_TM_FUNC;
 
-       if (desc_buff && desc_op != UPIU_QUERY_OPCODE_WRITE_DESC) {
-               err = -ENOTSUPP;
-               goto out;
-       }
-
        switch (msgcode) {
        case UPIU_TRANSACTION_NOP_OUT:
                cmd_type = DEV_CMD_TYPE_NOP;
@@ -5860,7 +5869,6 @@ int ufshcd_exec_raw_upiu_cmd(struct ufs_hba *hba,
                break;
        }
 
-out:
        return err;
 }
 
-- 
1.9.1

Reply via email to