};
 
 struct ufs_dev_info {
-       bool f_power_on_wp_en;
+       bool    f_power_on_wp_en;
        /* Keeps information if any of the LU is power on write protected */
-       bool is_lu_power_on_wp;
+       bool    is_lu_power_on_wp;
        /* Maximum number of general LU supported by the UFS device */
-       u8 max_lu_supported;
-       u8 wb_dedicated_lu;
-       u16 wmanufacturerid;
+       u8      max_lu_supported;
+       u16     wmanufacturerid;
        /*UFS device Product Name */
-       u8 *model;
-       u16 wspecversion;
-       u32 clk_gating_wait_us;
-       u8 b_wb_buffer_type;
-       bool b_rpm_dev_flush_capable;
-       u8 b_presrv_uspc_en;
+       u8      *model;
+       u16     wspecversion;
+       u32     clk_gating_wait_us;
+
+       /* UFS WB related flags */
+       bool    wb_enabled;
+       bool    wb_buf_flush_enabled;
+       u8      wb_dedicated_lu;
+       u8      wb_buffer_type;
+
+       bool    b_rpm_dev_flush_capable;
+       u8      b_presrv_uspc_en;
 };
 
 /*
 
        if (!err) {
                ufshcd_set_ufs_dev_active(hba);
                if (ufshcd_is_wb_allowed(hba)) {
-                       hba->wb_enabled = false;
-                       hba->wb_buf_flush_enabled = false;
+                       hba->dev_info.wb_enabled = false;
+                       hba->dev_info.wb_buf_flush_enabled = false;
                }
        }
        if (err != -EOPNOTSUPP)
        if (!ufshcd_is_wb_allowed(hba))
                return 0;
 
-       if (!(enable ^ hba->wb_enabled))
+       if (!(enable ^ hba->dev_info.wb_enabled))
                return 0;
        if (enable)
                opcode = UPIU_QUERY_OPCODE_SET_FLAG;
                return ret;
        }
 
-       hba->wb_enabled = enable;
+       hba->dev_info.wb_enabled = enable;
        dev_dbg(hba->dev, "%s write booster %s %d\n",
                        __func__, enable ? "enable" : "disable", ret);
 
        int ret;
        u8 index;
 
-       if (!ufshcd_is_wb_allowed(hba) || hba->wb_buf_flush_enabled)
+       if (!ufshcd_is_wb_allowed(hba) || hba->dev_info.wb_buf_flush_enabled)
                return 0;
 
        index = ufshcd_wb_get_query_index(hba);
                dev_err(hba->dev, "%s WB - buf flush enable failed %d\n",
                        __func__, ret);
        else
-               hba->wb_buf_flush_enabled = true;
+               hba->dev_info.wb_buf_flush_enabled = true;
 
        dev_dbg(hba->dev, "WB - Flush enabled: %d\n", ret);
        return ret;
        int ret;
        u8 index;
 
-       if (!ufshcd_is_wb_allowed(hba) || !hba->wb_buf_flush_enabled)
+       if (!ufshcd_is_wb_allowed(hba) || !hba->dev_info.wb_buf_flush_enabled)
                return 0;
 
        index = ufshcd_wb_get_query_index(hba);
                dev_warn(hba->dev, "%s: WB - buf flush disable failed %d\n",
                         __func__, ret);
        } else {
-               hba->wb_buf_flush_enabled = false;
+               hba->dev_info.wb_buf_flush_enabled = false;
                dev_dbg(hba->dev, "WB - Flush disabled: %d\n", ret);
        }
 
         * says, in dedicated wb buffer mode, a max of 1 lun would have wb
         * buffer configured.
         */
-       dev_info->b_wb_buffer_type =
-               desc_buf[DEVICE_DESC_PARAM_WB_TYPE];
+       dev_info->wb_buffer_type = desc_buf[DEVICE_DESC_PARAM_WB_TYPE];
 
        dev_info->b_presrv_uspc_en =
                desc_buf[DEVICE_DESC_PARAM_WB_PRESRV_USRSPC_EN];
 
-       if (dev_info->b_wb_buffer_type == WB_BUF_MODE_SHARED) {
+       if (dev_info->wb_buffer_type == WB_BUF_MODE_SHARED) {
                if (!get_unaligned_be32(desc_buf +
                                   DEVICE_DESC_PARAM_WB_SHARED_ALLOC_UNITS))
                        goto wb_disabled;
 
 
        struct device           bsg_dev;
        struct request_queue    *bsg_queue;
-       bool wb_buf_flush_enabled;
-       bool wb_enabled;
        struct delayed_work rpm_dev_flush_recheck_work;
 
 #ifdef CONFIG_SCSI_UFS_CRYPTO
 
 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)
+       if (hba->dev_info.wb_buffer_type == WB_BUF_MODE_LU_DEDICATED)
                return hba->dev_info.wb_dedicated_lu;
        return 0;
 }