root/drivers/staging/rts5208/ms.c

/* [<][>][^][v][top][bottom][index][help] */

DEFINITIONS

This source file includes following definitions.
  1. ms_set_err_code
  2. ms_check_err_code
  3. ms_parse_err_code
  4. ms_transfer_tpc
  5. ms_transfer_data
  6. ms_write_bytes
  7. ms_read_bytes
  8. ms_set_rw_reg_addr
  9. ms_send_cmd
  10. ms_set_init_para
  11. ms_switch_clock
  12. ms_pull_ctl_disable
  13. ms_pull_ctl_enable
  14. ms_prepare_reset
  15. ms_identify_media_type
  16. ms_confirm_cpu_startup
  17. ms_switch_parallel_bus
  18. ms_switch_8bit_bus
  19. ms_pro_reset_flow
  20. msxc_change_power
  21. ms_read_attribute_info
  22. reset_ms_pro
  23. ms_read_status_reg
  24. ms_read_extra_data
  25. ms_write_extra_data
  26. ms_read_page
  27. ms_set_bad_block
  28. ms_erase_block
  29. ms_set_page_status
  30. ms_init_page
  31. ms_copy_page
  32. reset_ms
  33. ms_init_l2p_tbl
  34. ms_get_l2p_tbl
  35. ms_set_l2p_tbl
  36. ms_set_unused_block
  37. ms_get_unused_block
  38. ms_arbitrate_l2p
  39. ms_build_l2p_tbl
  40. reset_ms_card
  41. mspro_set_rw_cmd
  42. mspro_stop_seq_mode
  43. ms_auto_tune_clock
  44. mspro_rw_multi_sector
  45. mspro_read_format_progress
  46. mspro_polling_format_status
  47. mspro_format
  48. ms_read_multiple_pages
  49. ms_write_multiple_pages
  50. ms_finish_write
  51. ms_prepare_write
  52. ms_delay_write
  53. ms_rw_fail
  54. ms_rw_multi_sector
  55. ms_rw
  56. ms_free_l2p_tbl
  57. ms_poll_int
  58. check_ms_err
  59. check_ms_err
  60. mg_send_ex_cmd
  61. mg_set_tpc_para_sub
  62. mg_set_leaf_id
  63. mg_get_local_EKB
  64. mg_chg
  65. mg_get_rsp_chg
  66. mg_rsp
  67. mg_get_ICV
  68. mg_set_ICV
  69. ms_cleanup_work
  70. ms_power_off_card3v3
  71. release_ms_card

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Driver for Realtek PCI-Express card reader
   4  *
   5  * Copyright(c) 2009-2013 Realtek Semiconductor Corp. All rights reserved.
   6  *
   7  * Author:
   8  *   Wei WANG (wei_wang@realsil.com.cn)
   9  *   Micky Ching (micky_ching@realsil.com.cn)
  10  */
  11 
  12 #include <linux/blkdev.h>
  13 #include <linux/kthread.h>
  14 #include <linux/sched.h>
  15 #include <linux/vmalloc.h>
  16 
  17 #include "rtsx.h"
  18 #include "ms.h"
  19 
  20 static inline void ms_set_err_code(struct rtsx_chip *chip, u8 err_code)
  21 {
  22         struct ms_info *ms_card = &chip->ms_card;
  23 
  24         ms_card->err_code = err_code;
  25 }
  26 
  27 static inline int ms_check_err_code(struct rtsx_chip *chip, u8 err_code)
  28 {
  29         struct ms_info *ms_card = &chip->ms_card;
  30 
  31         return (ms_card->err_code == err_code);
  32 }
  33 
  34 static int ms_parse_err_code(struct rtsx_chip *chip)
  35 {
  36         return STATUS_FAIL;
  37 }
  38 
  39 static int ms_transfer_tpc(struct rtsx_chip *chip, u8 trans_mode,
  40                            u8 tpc, u8 cnt, u8 cfg)
  41 {
  42         struct ms_info *ms_card = &chip->ms_card;
  43         int retval;
  44         u8 *ptr;
  45 
  46         dev_dbg(rtsx_dev(chip), "%s: tpc = 0x%x\n", __func__, tpc);
  47 
  48         rtsx_init_cmd(chip);
  49 
  50         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC, 0xFF, tpc);
  51         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_BYTE_CNT, 0xFF, cnt);
  52         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG, 0xFF, cfg);
  53         rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_DATA_SOURCE,
  54                      0x01, PINGPONG_BUFFER);
  55 
  56         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANSFER,
  57                      0xFF, MS_TRANSFER_START | trans_mode);
  58         rtsx_add_cmd(chip, CHECK_REG_CMD, MS_TRANSFER,
  59                      MS_TRANSFER_END, MS_TRANSFER_END);
  60 
  61         rtsx_add_cmd(chip, READ_REG_CMD, MS_TRANS_CFG, 0, 0);
  62 
  63         retval = rtsx_send_cmd(chip, MS_CARD, 5000);
  64         if (retval < 0) {
  65                 rtsx_clear_ms_error(chip);
  66                 ms_set_err_code(chip, MS_TO_ERROR);
  67                 return ms_parse_err_code(chip);
  68         }
  69 
  70         ptr = rtsx_get_cmd_data(chip) + 1;
  71 
  72         if (!(tpc & 0x08)) {            /* Read Packet */
  73                 if (*ptr & MS_CRC16_ERR) {
  74                         ms_set_err_code(chip, MS_CRC16_ERROR);
  75                         return ms_parse_err_code(chip);
  76                 }
  77         } else {                        /* Write Packet */
  78                 if (CHK_MSPRO(ms_card) && !(*ptr & 0x80)) {
  79                         if (*ptr & (MS_INT_ERR | MS_INT_CMDNK)) {
  80                                 ms_set_err_code(chip, MS_CMD_NK);
  81                                 return ms_parse_err_code(chip);
  82                         }
  83                 }
  84         }
  85 
  86         if (*ptr & MS_RDY_TIMEOUT) {
  87                 rtsx_clear_ms_error(chip);
  88                 ms_set_err_code(chip, MS_TO_ERROR);
  89                 return ms_parse_err_code(chip);
  90         }
  91 
  92         return STATUS_SUCCESS;
  93 }
  94 
  95 static int ms_transfer_data(struct rtsx_chip *chip, u8 trans_mode,
  96                             u8 tpc, u16 sec_cnt, u8 cfg, bool mode_2k,
  97                             int use_sg, void *buf, int buf_len)
  98 {
  99         int retval;
 100         u8 val, err_code = 0;
 101         enum dma_data_direction dir;
 102 
 103         if (!buf || !buf_len)
 104                 return STATUS_FAIL;
 105 
 106         if (trans_mode == MS_TM_AUTO_READ) {
 107                 dir = DMA_FROM_DEVICE;
 108                 err_code = MS_FLASH_READ_ERROR;
 109         } else if (trans_mode == MS_TM_AUTO_WRITE) {
 110                 dir = DMA_TO_DEVICE;
 111                 err_code = MS_FLASH_WRITE_ERROR;
 112         } else {
 113                 return STATUS_FAIL;
 114         }
 115 
 116         rtsx_init_cmd(chip);
 117 
 118         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC, 0xFF, tpc);
 119         rtsx_add_cmd(chip, WRITE_REG_CMD,
 120                      MS_SECTOR_CNT_H, 0xFF, (u8)(sec_cnt >> 8));
 121         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_SECTOR_CNT_L, 0xFF, (u8)sec_cnt);
 122         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG, 0xFF, cfg);
 123 
 124         if (mode_2k) {
 125                 rtsx_add_cmd(chip, WRITE_REG_CMD,
 126                              MS_CFG, MS_2K_SECTOR_MODE, MS_2K_SECTOR_MODE);
 127         } else {
 128                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_CFG, MS_2K_SECTOR_MODE, 0);
 129         }
 130 
 131         trans_dma_enable(dir, chip, sec_cnt * 512, DMA_512);
 132 
 133         rtsx_add_cmd(chip, WRITE_REG_CMD,
 134                      MS_TRANSFER, 0xFF, MS_TRANSFER_START | trans_mode);
 135         rtsx_add_cmd(chip, CHECK_REG_CMD,
 136                      MS_TRANSFER, MS_TRANSFER_END, MS_TRANSFER_END);
 137 
 138         rtsx_send_cmd_no_wait(chip);
 139 
 140         retval = rtsx_transfer_data(chip, MS_CARD, buf, buf_len,
 141                                     use_sg, dir, chip->mspro_timeout);
 142         if (retval < 0) {
 143                 ms_set_err_code(chip, err_code);
 144                 if (retval == -ETIMEDOUT)
 145                         retval = STATUS_TIMEDOUT;
 146                 else
 147                         retval = STATUS_FAIL;
 148 
 149                 return retval;
 150         }
 151 
 152         retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
 153         if (retval)
 154                 return retval;
 155 
 156         if (val & (MS_INT_CMDNK | MS_INT_ERR | MS_CRC16_ERR | MS_RDY_TIMEOUT))
 157                 return STATUS_FAIL;
 158 
 159         return STATUS_SUCCESS;
 160 }
 161 
 162 static int ms_write_bytes(struct rtsx_chip *chip,
 163                           u8 tpc, u8 cnt, u8 cfg, u8 *data, int data_len)
 164 {
 165         struct ms_info *ms_card = &chip->ms_card;
 166         int retval, i;
 167 
 168         if (!data || (data_len < cnt))
 169                 return STATUS_ERROR;
 170 
 171         rtsx_init_cmd(chip);
 172 
 173         for (i = 0; i < cnt; i++) {
 174                 rtsx_add_cmd(chip, WRITE_REG_CMD,
 175                              PPBUF_BASE2 + i, 0xFF, data[i]);
 176         }
 177         if (cnt % 2)
 178                 rtsx_add_cmd(chip, WRITE_REG_CMD, PPBUF_BASE2 + i, 0xFF, 0xFF);
 179 
 180         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC, 0xFF, tpc);
 181         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_BYTE_CNT, 0xFF, cnt);
 182         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG, 0xFF, cfg);
 183         rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_DATA_SOURCE,
 184                      0x01, PINGPONG_BUFFER);
 185 
 186         rtsx_add_cmd(chip, WRITE_REG_CMD,
 187                      MS_TRANSFER, 0xFF, MS_TRANSFER_START | MS_TM_WRITE_BYTES);
 188         rtsx_add_cmd(chip, CHECK_REG_CMD,
 189                      MS_TRANSFER, MS_TRANSFER_END, MS_TRANSFER_END);
 190 
 191         retval = rtsx_send_cmd(chip, MS_CARD, 5000);
 192         if (retval < 0) {
 193                 u8 val = 0;
 194 
 195                 rtsx_read_register(chip, MS_TRANS_CFG, &val);
 196                 dev_dbg(rtsx_dev(chip), "MS_TRANS_CFG: 0x%02x\n", val);
 197 
 198                 rtsx_clear_ms_error(chip);
 199 
 200                 if (!(tpc & 0x08)) {
 201                         if (val & MS_CRC16_ERR) {
 202                                 ms_set_err_code(chip, MS_CRC16_ERROR);
 203                                 return ms_parse_err_code(chip);
 204                         }
 205                 } else {
 206                         if (CHK_MSPRO(ms_card) && !(val & 0x80)) {
 207                                 if (val & (MS_INT_ERR | MS_INT_CMDNK)) {
 208                                         ms_set_err_code(chip, MS_CMD_NK);
 209                                         return ms_parse_err_code(chip);
 210                                 }
 211                         }
 212                 }
 213 
 214                 if (val & MS_RDY_TIMEOUT) {
 215                         ms_set_err_code(chip, MS_TO_ERROR);
 216                         return ms_parse_err_code(chip);
 217                 }
 218 
 219                 ms_set_err_code(chip, MS_TO_ERROR);
 220                 return ms_parse_err_code(chip);
 221         }
 222 
 223         return STATUS_SUCCESS;
 224 }
 225 
 226 static int ms_read_bytes(struct rtsx_chip *chip,
 227                          u8 tpc, u8 cnt, u8 cfg, u8 *data, int data_len)
 228 {
 229         struct ms_info *ms_card = &chip->ms_card;
 230         int retval, i;
 231         u8 *ptr;
 232 
 233         if (!data)
 234                 return STATUS_ERROR;
 235 
 236         rtsx_init_cmd(chip);
 237 
 238         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC, 0xFF, tpc);
 239         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_BYTE_CNT, 0xFF, cnt);
 240         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG, 0xFF, cfg);
 241         rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_DATA_SOURCE,
 242                      0x01, PINGPONG_BUFFER);
 243 
 244         rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANSFER, 0xFF,
 245                      MS_TRANSFER_START | MS_TM_READ_BYTES);
 246         rtsx_add_cmd(chip, CHECK_REG_CMD, MS_TRANSFER,
 247                      MS_TRANSFER_END, MS_TRANSFER_END);
 248 
 249         for (i = 0; i < data_len - 1; i++)
 250                 rtsx_add_cmd(chip, READ_REG_CMD, PPBUF_BASE2 + i, 0, 0);
 251 
 252         if (data_len % 2)
 253                 rtsx_add_cmd(chip, READ_REG_CMD, PPBUF_BASE2 + data_len, 0, 0);
 254         else
 255                 rtsx_add_cmd(chip, READ_REG_CMD, PPBUF_BASE2 + data_len - 1,
 256                              0, 0);
 257 
 258         retval = rtsx_send_cmd(chip, MS_CARD, 5000);
 259         if (retval < 0) {
 260                 u8 val = 0;
 261 
 262                 rtsx_read_register(chip, MS_TRANS_CFG, &val);
 263                 rtsx_clear_ms_error(chip);
 264 
 265                 if (!(tpc & 0x08)) {
 266                         if (val & MS_CRC16_ERR) {
 267                                 ms_set_err_code(chip, MS_CRC16_ERROR);
 268                                 return ms_parse_err_code(chip);
 269                         }
 270                 } else {
 271                         if (CHK_MSPRO(ms_card) && !(val & 0x80)) {
 272                                 if (val & (MS_INT_ERR | MS_INT_CMDNK)) {
 273                                         ms_set_err_code(chip, MS_CMD_NK);
 274                                         return ms_parse_err_code(chip);
 275                                 }
 276                         }
 277                 }
 278 
 279                 if (val & MS_RDY_TIMEOUT) {
 280                         ms_set_err_code(chip, MS_TO_ERROR);
 281                         return ms_parse_err_code(chip);
 282                 }
 283 
 284                 ms_set_err_code(chip, MS_TO_ERROR);
 285                 return ms_parse_err_code(chip);
 286         }
 287 
 288         ptr = rtsx_get_cmd_data(chip) + 1;
 289 
 290         for (i = 0; i < data_len; i++)
 291                 data[i] = ptr[i];
 292 
 293         if ((tpc == PRO_READ_SHORT_DATA) && (data_len == 8)) {
 294                 dev_dbg(rtsx_dev(chip), "Read format progress:\n");
 295                 print_hex_dump_bytes(KBUILD_MODNAME ": ", DUMP_PREFIX_NONE, ptr,
 296                                      cnt);
 297         }
 298 
 299         return STATUS_SUCCESS;
 300 }
 301 
 302 static int ms_set_rw_reg_addr(struct rtsx_chip *chip, u8 read_start,
 303                               u8 read_cnt, u8 write_start, u8 write_cnt)
 304 {
 305         int retval, i;
 306         u8 data[4];
 307 
 308         data[0] = read_start;
 309         data[1] = read_cnt;
 310         data[2] = write_start;
 311         data[3] = write_cnt;
 312 
 313         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
 314                 retval = ms_write_bytes(chip, SET_RW_REG_ADRS, 4,
 315                                         NO_WAIT_INT, data, 4);
 316                 if (retval == STATUS_SUCCESS)
 317                         return STATUS_SUCCESS;
 318                 rtsx_clear_ms_error(chip);
 319         }
 320 
 321         return STATUS_FAIL;
 322 }
 323 
 324 static int ms_send_cmd(struct rtsx_chip *chip, u8 cmd, u8 cfg)
 325 {
 326         u8 data[2];
 327 
 328         data[0] = cmd;
 329         data[1] = 0;
 330 
 331         return ms_write_bytes(chip, PRO_SET_CMD, 1, cfg, data, 1);
 332 }
 333 
 334 static int ms_set_init_para(struct rtsx_chip *chip)
 335 {
 336         struct ms_info *ms_card = &chip->ms_card;
 337         int retval;
 338 
 339         if (CHK_HG8BIT(ms_card)) {
 340                 if (chip->asic_code)
 341                         ms_card->ms_clock = chip->asic_ms_hg_clk;
 342                 else
 343                         ms_card->ms_clock = chip->fpga_ms_hg_clk;
 344 
 345         } else if (CHK_MSPRO(ms_card) || CHK_MS4BIT(ms_card)) {
 346                 if (chip->asic_code)
 347                         ms_card->ms_clock = chip->asic_ms_4bit_clk;
 348                 else
 349                         ms_card->ms_clock = chip->fpga_ms_4bit_clk;
 350 
 351         } else {
 352                 if (chip->asic_code)
 353                         ms_card->ms_clock = chip->asic_ms_1bit_clk;
 354                 else
 355                         ms_card->ms_clock = chip->fpga_ms_1bit_clk;
 356         }
 357 
 358         retval = switch_clock(chip, ms_card->ms_clock);
 359         if (retval != STATUS_SUCCESS)
 360                 return STATUS_FAIL;
 361 
 362         retval = select_card(chip, MS_CARD);
 363         if (retval != STATUS_SUCCESS)
 364                 return STATUS_FAIL;
 365 
 366         return STATUS_SUCCESS;
 367 }
 368 
 369 static int ms_switch_clock(struct rtsx_chip *chip)
 370 {
 371         struct ms_info *ms_card = &chip->ms_card;
 372         int retval;
 373 
 374         retval = select_card(chip, MS_CARD);
 375         if (retval != STATUS_SUCCESS)
 376                 return STATUS_FAIL;
 377 
 378         retval = switch_clock(chip, ms_card->ms_clock);
 379         if (retval != STATUS_SUCCESS)
 380                 return STATUS_FAIL;
 381 
 382         return STATUS_SUCCESS;
 383 }
 384 
 385 static int ms_pull_ctl_disable(struct rtsx_chip *chip)
 386 {
 387         int retval;
 388 
 389         if (CHECK_PID(chip, 0x5208)) {
 390                 retval = rtsx_write_register(chip, CARD_PULL_CTL1, 0xFF,
 391                                              MS_D1_PD | MS_D2_PD | MS_CLK_PD |
 392                                              MS_D6_PD);
 393                 if (retval)
 394                         return retval;
 395 
 396                 retval = rtsx_write_register(chip, CARD_PULL_CTL2, 0xFF,
 397                                              MS_D3_PD | MS_D0_PD | MS_BS_PD |
 398                                              XD_D4_PD);
 399                 if (retval)
 400                         return retval;
 401 
 402                 retval = rtsx_write_register(chip, CARD_PULL_CTL3, 0xFF,
 403                                              MS_D7_PD | XD_CE_PD | XD_CLE_PD |
 404                                              XD_CD_PU);
 405                 if (retval)
 406                         return retval;
 407 
 408                 retval = rtsx_write_register(chip, CARD_PULL_CTL4, 0xFF,
 409                                              XD_RDY_PD | SD_D3_PD | SD_D2_PD |
 410                                              XD_ALE_PD);
 411                 if (retval)
 412                         return retval;
 413 
 414                 retval = rtsx_write_register(chip, CARD_PULL_CTL5, 0xFF,
 415                                              MS_INS_PU | SD_WP_PD | SD_CD_PU |
 416                                              SD_CMD_PD);
 417                 if (retval)
 418                         return retval;
 419 
 420                 retval = rtsx_write_register(chip, CARD_PULL_CTL6, 0xFF,
 421                                              MS_D5_PD | MS_D4_PD);
 422                 if (retval)
 423                         return retval;
 424 
 425         } else if (CHECK_PID(chip, 0x5288)) {
 426                 if (CHECK_BARO_PKG(chip, QFN)) {
 427                         retval = rtsx_write_register(chip, CARD_PULL_CTL1,
 428                                                      0xFF, 0x55);
 429                         if (retval)
 430                                 return retval;
 431 
 432                         retval = rtsx_write_register(chip, CARD_PULL_CTL2,
 433                                                      0xFF, 0x55);
 434                         if (retval)
 435                                 return retval;
 436 
 437                         retval = rtsx_write_register(chip, CARD_PULL_CTL3,
 438                                                      0xFF, 0x4B);
 439                         if (retval)
 440                                 return retval;
 441 
 442                         retval = rtsx_write_register(chip, CARD_PULL_CTL4,
 443                                                      0xFF, 0x69);
 444                         if (retval)
 445                                 return retval;
 446                 }
 447         }
 448 
 449         return STATUS_SUCCESS;
 450 }
 451 
 452 static int ms_pull_ctl_enable(struct rtsx_chip *chip)
 453 {
 454         int retval;
 455 
 456         rtsx_init_cmd(chip);
 457 
 458         if (CHECK_PID(chip, 0x5208)) {
 459                 rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL1, 0xFF,
 460                              MS_D1_PD | MS_D2_PD | MS_CLK_NP | MS_D6_PD);
 461                 rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL2, 0xFF,
 462                              MS_D3_PD | MS_D0_PD | MS_BS_NP | XD_D4_PD);
 463                 rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL3, 0xFF,
 464                              MS_D7_PD | XD_CE_PD | XD_CLE_PD | XD_CD_PU);
 465                 rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL4, 0xFF,
 466                              XD_RDY_PD | SD_D3_PD | SD_D2_PD | XD_ALE_PD);
 467                 rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL5, 0xFF,
 468                              MS_INS_PU | SD_WP_PD | SD_CD_PU | SD_CMD_PD);
 469                 rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL6, 0xFF,
 470                              MS_D5_PD | MS_D4_PD);
 471         } else if (CHECK_PID(chip, 0x5288)) {
 472                 if (CHECK_BARO_PKG(chip, QFN)) {
 473                         rtsx_add_cmd(chip, WRITE_REG_CMD,
 474                                      CARD_PULL_CTL1, 0xFF, 0x55);
 475                         rtsx_add_cmd(chip, WRITE_REG_CMD,
 476                                      CARD_PULL_CTL2, 0xFF, 0x45);
 477                         rtsx_add_cmd(chip, WRITE_REG_CMD,
 478                                      CARD_PULL_CTL3, 0xFF, 0x4B);
 479                         rtsx_add_cmd(chip, WRITE_REG_CMD,
 480                                      CARD_PULL_CTL4, 0xFF, 0x29);
 481                 }
 482         }
 483 
 484         retval = rtsx_send_cmd(chip, MS_CARD, 100);
 485         if (retval < 0)
 486                 return STATUS_FAIL;
 487 
 488         return STATUS_SUCCESS;
 489 }
 490 
 491 static int ms_prepare_reset(struct rtsx_chip *chip)
 492 {
 493         struct ms_info *ms_card = &chip->ms_card;
 494         int retval;
 495         u8 oc_mask = 0;
 496 
 497         ms_card->ms_type = 0;
 498         ms_card->check_ms_flow = 0;
 499         ms_card->switch_8bit_fail = 0;
 500         ms_card->delay_write.delay_write_flag = 0;
 501 
 502         ms_card->pro_under_formatting = 0;
 503 
 504         retval = ms_power_off_card3v3(chip);
 505         if (retval != STATUS_SUCCESS)
 506                 return STATUS_FAIL;
 507 
 508         if (!chip->ft2_fast_mode)
 509                 wait_timeout(250);
 510 
 511         retval = enable_card_clock(chip, MS_CARD);
 512         if (retval != STATUS_SUCCESS)
 513                 return STATUS_FAIL;
 514 
 515         if (chip->asic_code) {
 516                 retval = ms_pull_ctl_enable(chip);
 517                 if (retval != STATUS_SUCCESS)
 518                         return STATUS_FAIL;
 519         } else {
 520                 retval = rtsx_write_register(chip, FPGA_PULL_CTL,
 521                                              FPGA_MS_PULL_CTL_BIT | 0x20, 0);
 522                 if (retval)
 523                         return retval;
 524         }
 525 
 526         if (!chip->ft2_fast_mode) {
 527                 retval = card_power_on(chip, MS_CARD);
 528                 if (retval != STATUS_SUCCESS)
 529                         return STATUS_FAIL;
 530 
 531                 wait_timeout(150);
 532 
 533 #ifdef SUPPORT_OCP
 534                 if (CHECK_LUN_MODE(chip, SD_MS_2LUN))
 535                         oc_mask = MS_OC_NOW | MS_OC_EVER;
 536                 else
 537                         oc_mask = SD_OC_NOW | SD_OC_EVER;
 538 
 539                 if (chip->ocp_stat & oc_mask) {
 540                         dev_dbg(rtsx_dev(chip), "Over current, OCPSTAT is 0x%x\n",
 541                                 chip->ocp_stat);
 542                         return STATUS_FAIL;
 543                 }
 544 #endif
 545         }
 546 
 547         retval = rtsx_write_register(chip, CARD_OE, MS_OUTPUT_EN,
 548                                      MS_OUTPUT_EN);
 549         if (retval)
 550                 return retval;
 551 
 552         if (chip->asic_code) {
 553                 retval = rtsx_write_register(chip, MS_CFG, 0xFF,
 554                                              SAMPLE_TIME_RISING |
 555                                              PUSH_TIME_DEFAULT |
 556                                              NO_EXTEND_TOGGLE |
 557                                              MS_BUS_WIDTH_1);
 558                 if (retval)
 559                         return retval;
 560 
 561         } else {
 562                 retval = rtsx_write_register(chip, MS_CFG, 0xFF,
 563                                              SAMPLE_TIME_FALLING |
 564                                              PUSH_TIME_DEFAULT |
 565                                              NO_EXTEND_TOGGLE |
 566                                              MS_BUS_WIDTH_1);
 567                 if (retval)
 568                         return retval;
 569         }
 570         retval = rtsx_write_register(chip, MS_TRANS_CFG, 0xFF,
 571                                      NO_WAIT_INT | NO_AUTO_READ_INT_REG);
 572         if (retval)
 573                 return retval;
 574 
 575         retval = rtsx_write_register(chip, CARD_STOP, MS_STOP | MS_CLR_ERR,
 576                                      MS_STOP | MS_CLR_ERR);
 577         if (retval)
 578                 return retval;
 579 
 580         retval = ms_set_init_para(chip);
 581         if (retval != STATUS_SUCCESS)
 582                 return STATUS_FAIL;
 583 
 584         return STATUS_SUCCESS;
 585 }
 586 
 587 static int ms_identify_media_type(struct rtsx_chip *chip, int switch_8bit_bus)
 588 {
 589         struct ms_info *ms_card = &chip->ms_card;
 590         int retval, i;
 591         u8 val;
 592 
 593         retval = ms_set_rw_reg_addr(chip, Pro_StatusReg, 6, SystemParm, 1);
 594         if (retval != STATUS_SUCCESS)
 595                 return STATUS_FAIL;
 596 
 597         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
 598                 retval = ms_transfer_tpc(chip, MS_TM_READ_BYTES, READ_REG,
 599                                          6, NO_WAIT_INT);
 600                 if (retval == STATUS_SUCCESS)
 601                         break;
 602         }
 603         if (i == MS_MAX_RETRY_COUNT)
 604                 return STATUS_FAIL;
 605 
 606         retval = rtsx_read_register(chip, PPBUF_BASE2 + 2, &val);
 607         if (retval)
 608                 return retval;
 609 
 610         dev_dbg(rtsx_dev(chip), "Type register: 0x%x\n", val);
 611         if (val != 0x01) {
 612                 if (val != 0x02)
 613                         ms_card->check_ms_flow = 1;
 614 
 615                 return STATUS_FAIL;
 616         }
 617 
 618         retval = rtsx_read_register(chip, PPBUF_BASE2 + 4, &val);
 619         if (retval)
 620                 return retval;
 621 
 622         dev_dbg(rtsx_dev(chip), "Category register: 0x%x\n", val);
 623         if (val != 0) {
 624                 ms_card->check_ms_flow = 1;
 625                 return STATUS_FAIL;
 626         }
 627 
 628         retval = rtsx_read_register(chip, PPBUF_BASE2 + 5, &val);
 629         if (retval)
 630                 return retval;
 631 
 632         dev_dbg(rtsx_dev(chip), "Class register: 0x%x\n", val);
 633         if (val == 0) {
 634                 retval = rtsx_read_register(chip, PPBUF_BASE2, &val);
 635                 if (retval)
 636                         return retval;
 637 
 638                 if (val & WRT_PRTCT)
 639                         chip->card_wp |= MS_CARD;
 640                 else
 641                         chip->card_wp &= ~MS_CARD;
 642 
 643         } else if ((val == 0x01) || (val == 0x02) || (val == 0x03)) {
 644                 chip->card_wp |= MS_CARD;
 645         } else {
 646                 ms_card->check_ms_flow = 1;
 647                 return STATUS_FAIL;
 648         }
 649 
 650         ms_card->ms_type |= TYPE_MSPRO;
 651 
 652         retval = rtsx_read_register(chip, PPBUF_BASE2 + 3, &val);
 653         if (retval)
 654                 return retval;
 655 
 656         dev_dbg(rtsx_dev(chip), "IF Mode register: 0x%x\n", val);
 657         if (val == 0) {
 658                 ms_card->ms_type &= 0x0F;
 659         } else if (val == 7) {
 660                 if (switch_8bit_bus)
 661                         ms_card->ms_type |= MS_HG;
 662                 else
 663                         ms_card->ms_type &= 0x0F;
 664 
 665         } else {
 666                 return STATUS_FAIL;
 667         }
 668 
 669         return STATUS_SUCCESS;
 670 }
 671 
 672 static int ms_confirm_cpu_startup(struct rtsx_chip *chip)
 673 {
 674         int retval, i, k;
 675         u8 val;
 676 
 677         /* Confirm CPU StartUp */
 678         k = 0;
 679         do {
 680                 if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
 681                         ms_set_err_code(chip, MS_NO_CARD);
 682                         return STATUS_FAIL;
 683                 }
 684 
 685                 for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
 686                         retval = ms_read_bytes(chip, GET_INT, 1,
 687                                                NO_WAIT_INT, &val, 1);
 688                         if (retval == STATUS_SUCCESS)
 689                                 break;
 690                 }
 691                 if (i == MS_MAX_RETRY_COUNT)
 692                         return STATUS_FAIL;
 693 
 694                 if (k > 100)
 695                         return STATUS_FAIL;
 696 
 697                 k++;
 698                 wait_timeout(100);
 699         } while (!(val & INT_REG_CED));
 700 
 701         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
 702                 retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
 703                 if (retval == STATUS_SUCCESS)
 704                         break;
 705         }
 706         if (i == MS_MAX_RETRY_COUNT)
 707                 return STATUS_FAIL;
 708 
 709         if (val & INT_REG_ERR) {
 710                 if (val & INT_REG_CMDNK)
 711                         chip->card_wp |= (MS_CARD);
 712                 else
 713                         return STATUS_FAIL;
 714         }
 715         /* --  end confirm CPU startup */
 716 
 717         return STATUS_SUCCESS;
 718 }
 719 
 720 static int ms_switch_parallel_bus(struct rtsx_chip *chip)
 721 {
 722         int retval, i;
 723         u8 data[2];
 724 
 725         data[0] = PARALLEL_4BIT_IF;
 726         data[1] = 0;
 727         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
 728                 retval = ms_write_bytes(chip, WRITE_REG, 1, NO_WAIT_INT,
 729                                         data, 2);
 730                 if (retval == STATUS_SUCCESS)
 731                         break;
 732         }
 733         if (retval != STATUS_SUCCESS)
 734                 return STATUS_FAIL;
 735 
 736         return STATUS_SUCCESS;
 737 }
 738 
 739 static int ms_switch_8bit_bus(struct rtsx_chip *chip)
 740 {
 741         struct ms_info *ms_card = &chip->ms_card;
 742         int retval, i;
 743         u8 data[2];
 744 
 745         data[0] = PARALLEL_8BIT_IF;
 746         data[1] = 0;
 747         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
 748                 retval = ms_write_bytes(chip, WRITE_REG, 1,
 749                                         NO_WAIT_INT, data, 2);
 750                 if (retval == STATUS_SUCCESS)
 751                         break;
 752         }
 753         if (retval != STATUS_SUCCESS)
 754                 return STATUS_FAIL;
 755 
 756         retval = rtsx_write_register(chip, MS_CFG, 0x98,
 757                                      MS_BUS_WIDTH_8 | SAMPLE_TIME_FALLING);
 758         if (retval)
 759                 return retval;
 760 
 761         ms_card->ms_type |= MS_8BIT;
 762         retval = ms_set_init_para(chip);
 763         if (retval != STATUS_SUCCESS)
 764                 return STATUS_FAIL;
 765 
 766         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
 767                 retval = ms_transfer_tpc(chip, MS_TM_READ_BYTES, GET_INT,
 768                                          1, NO_WAIT_INT);
 769                 if (retval != STATUS_SUCCESS)
 770                         return STATUS_FAIL;
 771         }
 772 
 773         return STATUS_SUCCESS;
 774 }
 775 
 776 static int ms_pro_reset_flow(struct rtsx_chip *chip, int switch_8bit_bus)
 777 {
 778         struct ms_info *ms_card = &chip->ms_card;
 779         int retval, i;
 780 
 781         for (i = 0; i < 3; i++) {
 782                 retval = ms_prepare_reset(chip);
 783                 if (retval != STATUS_SUCCESS)
 784                         return STATUS_FAIL;
 785 
 786                 retval = ms_identify_media_type(chip, switch_8bit_bus);
 787                 if (retval != STATUS_SUCCESS)
 788                         return STATUS_FAIL;
 789 
 790                 retval = ms_confirm_cpu_startup(chip);
 791                 if (retval != STATUS_SUCCESS)
 792                         return STATUS_FAIL;
 793 
 794                 retval = ms_switch_parallel_bus(chip);
 795                 if (retval != STATUS_SUCCESS) {
 796                         if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
 797                                 ms_set_err_code(chip, MS_NO_CARD);
 798                                 return STATUS_FAIL;
 799                         }
 800                         continue;
 801                 } else {
 802                         break;
 803                 }
 804         }
 805 
 806         if (retval != STATUS_SUCCESS)
 807                 return STATUS_FAIL;
 808 
 809         /* Switch MS-PRO into Parallel mode */
 810         retval = rtsx_write_register(chip, MS_CFG, 0x18, MS_BUS_WIDTH_4);
 811         if (retval)
 812                 return retval;
 813 
 814         retval = rtsx_write_register(chip, MS_CFG, PUSH_TIME_ODD,
 815                                      PUSH_TIME_ODD);
 816         if (retval)
 817                 return retval;
 818 
 819         retval = ms_set_init_para(chip);
 820         if (retval != STATUS_SUCCESS)
 821                 return STATUS_FAIL;
 822 
 823         /* If MSPro HG Card, We shall try to switch to 8-bit bus */
 824         if (CHK_MSHG(ms_card) && chip->support_ms_8bit && switch_8bit_bus) {
 825                 retval = ms_switch_8bit_bus(chip);
 826                 if (retval != STATUS_SUCCESS) {
 827                         ms_card->switch_8bit_fail = 1;
 828                         return STATUS_FAIL;
 829                 }
 830         }
 831 
 832         return STATUS_SUCCESS;
 833 }
 834 
 835 #ifdef XC_POWERCLASS
 836 static int msxc_change_power(struct rtsx_chip *chip, u8 mode)
 837 {
 838         int retval;
 839         u8 buf[6];
 840 
 841         ms_cleanup_work(chip);
 842 
 843         retval = ms_set_rw_reg_addr(chip, 0, 0, Pro_DataCount1, 6);
 844         if (retval != STATUS_SUCCESS)
 845                 return STATUS_FAIL;
 846 
 847         buf[0] = 0;
 848         buf[1] = mode;
 849         buf[2] = 0;
 850         buf[3] = 0;
 851         buf[4] = 0;
 852         buf[5] = 0;
 853 
 854         retval = ms_write_bytes(chip, PRO_WRITE_REG, 6, NO_WAIT_INT, buf, 6);
 855         if (retval != STATUS_SUCCESS)
 856                 return STATUS_FAIL;
 857 
 858         retval = ms_send_cmd(chip, XC_CHG_POWER, WAIT_INT);
 859         if (retval != STATUS_SUCCESS)
 860                 return STATUS_FAIL;
 861 
 862         retval = rtsx_read_register(chip, MS_TRANS_CFG, buf);
 863         if (retval)
 864                 return retval;
 865 
 866         if (buf[0] & (MS_INT_CMDNK | MS_INT_ERR))
 867                 return STATUS_FAIL;
 868 
 869         return STATUS_SUCCESS;
 870 }
 871 #endif
 872 
 873 static int ms_read_attribute_info(struct rtsx_chip *chip)
 874 {
 875         struct ms_info *ms_card = &chip->ms_card;
 876         int retval, i;
 877         u8 val, *buf, class_code, device_type, sub_class, data[16];
 878         u16 total_blk = 0, blk_size = 0;
 879 #ifdef SUPPORT_MSXC
 880         u32 xc_total_blk = 0, xc_blk_size = 0;
 881 #endif
 882         u32 sys_info_addr = 0, sys_info_size;
 883 #ifdef SUPPORT_PCGL_1P18
 884         u32 model_name_addr = 0, model_name_size;
 885         int found_sys_info = 0, found_model_name = 0;
 886 #endif
 887 
 888         retval = ms_set_rw_reg_addr(chip, Pro_IntReg, 2, Pro_SystemParm, 7);
 889         if (retval != STATUS_SUCCESS)
 890                 return STATUS_FAIL;
 891 
 892         if (CHK_MS8BIT(ms_card))
 893                 data[0] = PARALLEL_8BIT_IF;
 894         else
 895                 data[0] = PARALLEL_4BIT_IF;
 896 
 897         data[1] = 0;
 898 
 899         data[2] = 0x40;
 900         data[3] = 0;
 901         data[4] = 0;
 902         data[5] = 0;
 903         data[6] = 0;
 904         data[7] = 0;
 905 
 906         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
 907                 retval = ms_write_bytes(chip, PRO_WRITE_REG, 7, NO_WAIT_INT,
 908                                         data, 8);
 909                 if (retval == STATUS_SUCCESS)
 910                         break;
 911         }
 912         if (retval != STATUS_SUCCESS)
 913                 return STATUS_FAIL;
 914 
 915         buf = kmalloc(64 * 512, GFP_KERNEL);
 916         if (!buf)
 917                 return STATUS_ERROR;
 918 
 919         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
 920                 retval = ms_send_cmd(chip, PRO_READ_ATRB, WAIT_INT);
 921                 if (retval != STATUS_SUCCESS)
 922                         continue;
 923 
 924                 retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
 925                 if (retval != STATUS_SUCCESS) {
 926                         kfree(buf);
 927                         return STATUS_FAIL;
 928                 }
 929                 if (!(val & MS_INT_BREQ)) {
 930                         kfree(buf);
 931                         return STATUS_FAIL;
 932                 }
 933                 retval = ms_transfer_data(chip, MS_TM_AUTO_READ,
 934                                           PRO_READ_LONG_DATA, 0x40, WAIT_INT,
 935                                           0, 0, buf, 64 * 512);
 936                 if (retval == STATUS_SUCCESS)
 937                         break;
 938 
 939                 rtsx_clear_ms_error(chip);
 940         }
 941         if (retval != STATUS_SUCCESS) {
 942                 kfree(buf);
 943                 return STATUS_FAIL;
 944         }
 945 
 946         i = 0;
 947         do {
 948                 retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
 949                 if (retval != STATUS_SUCCESS) {
 950                         kfree(buf);
 951                         return STATUS_FAIL;
 952                 }
 953 
 954                 if ((val & MS_INT_CED) || !(val & MS_INT_BREQ))
 955                         break;
 956 
 957                 retval = ms_transfer_tpc(chip, MS_TM_NORMAL_READ,
 958                                          PRO_READ_LONG_DATA, 0, WAIT_INT);
 959                 if (retval != STATUS_SUCCESS) {
 960                         kfree(buf);
 961                         return STATUS_FAIL;
 962                 }
 963 
 964                 i++;
 965         } while (i < 1024);
 966 
 967         if ((buf[0] != 0xa5) && (buf[1] != 0xc3)) {
 968                 /* Signature code is wrong */
 969                 kfree(buf);
 970                 return STATUS_FAIL;
 971         }
 972 
 973         if ((buf[4] < 1) || (buf[4] > 12)) {
 974                 kfree(buf);
 975                 return STATUS_FAIL;
 976         }
 977 
 978         for (i = 0; i < buf[4]; i++) {
 979                 int cur_addr_off = 16 + i * 12;
 980 
 981 #ifdef SUPPORT_MSXC
 982                 if ((buf[cur_addr_off + 8] == 0x10) ||
 983                     (buf[cur_addr_off + 8] == 0x13)) {
 984 #else
 985                 if (buf[cur_addr_off + 8] == 0x10) {
 986 #endif
 987                         sys_info_addr = ((u32)buf[cur_addr_off + 0] << 24) |
 988                                 ((u32)buf[cur_addr_off + 1] << 16) |
 989                                 ((u32)buf[cur_addr_off + 2] << 8) |
 990                                 buf[cur_addr_off + 3];
 991                         sys_info_size = ((u32)buf[cur_addr_off + 4] << 24) |
 992                                 ((u32)buf[cur_addr_off + 5] << 16) |
 993                                 ((u32)buf[cur_addr_off + 6] << 8) |
 994                                 buf[cur_addr_off + 7];
 995                         dev_dbg(rtsx_dev(chip), "sys_info_addr = 0x%x, sys_info_size = 0x%x\n",
 996                                 sys_info_addr, sys_info_size);
 997                         if (sys_info_size != 96)  {
 998                                 kfree(buf);
 999                                 return STATUS_FAIL;
1000                         }
1001                         if (sys_info_addr < 0x1A0) {
1002                                 kfree(buf);
1003                                 return STATUS_FAIL;
1004                         }
1005                         if ((sys_info_size + sys_info_addr) > 0x8000) {
1006                                 kfree(buf);
1007                                 return STATUS_FAIL;
1008                         }
1009 
1010 #ifdef SUPPORT_MSXC
1011                         if (buf[cur_addr_off + 8] == 0x13)
1012                                 ms_card->ms_type |= MS_XC;
1013 #endif
1014 #ifdef SUPPORT_PCGL_1P18
1015                         found_sys_info = 1;
1016 #else
1017                         break;
1018 #endif
1019                 }
1020 #ifdef SUPPORT_PCGL_1P18
1021                 if (buf[cur_addr_off + 8] == 0x15) {
1022                         model_name_addr = ((u32)buf[cur_addr_off + 0] << 24) |
1023                                 ((u32)buf[cur_addr_off + 1] << 16) |
1024                                 ((u32)buf[cur_addr_off + 2] << 8) |
1025                                 buf[cur_addr_off + 3];
1026                         model_name_size = ((u32)buf[cur_addr_off + 4] << 24) |
1027                                 ((u32)buf[cur_addr_off + 5] << 16) |
1028                                 ((u32)buf[cur_addr_off + 6] << 8) |
1029                                 buf[cur_addr_off + 7];
1030                         dev_dbg(rtsx_dev(chip), "model_name_addr = 0x%x, model_name_size = 0x%x\n",
1031                                 model_name_addr, model_name_size);
1032                         if (model_name_size != 48)  {
1033                                 kfree(buf);
1034                                 return STATUS_FAIL;
1035                         }
1036                         if (model_name_addr < 0x1A0) {
1037                                 kfree(buf);
1038                                 return STATUS_FAIL;
1039                         }
1040                         if ((model_name_size + model_name_addr) > 0x8000) {
1041                                 kfree(buf);
1042                                 return STATUS_FAIL;
1043                         }
1044 
1045                         found_model_name = 1;
1046                 }
1047 
1048                 if (found_sys_info && found_model_name)
1049                         break;
1050 #endif
1051         }
1052 
1053         if (i == buf[4]) {
1054                 kfree(buf);
1055                 return STATUS_FAIL;
1056         }
1057 
1058         class_code =  buf[sys_info_addr + 0];
1059         device_type = buf[sys_info_addr + 56];
1060         sub_class = buf[sys_info_addr + 46];
1061 #ifdef SUPPORT_MSXC
1062         if (CHK_MSXC(ms_card)) {
1063                 xc_total_blk = ((u32)buf[sys_info_addr + 6] << 24) |
1064                                 ((u32)buf[sys_info_addr + 7] << 16) |
1065                                 ((u32)buf[sys_info_addr + 8] << 8) |
1066                                 buf[sys_info_addr + 9];
1067                 xc_blk_size = ((u32)buf[sys_info_addr + 32] << 24) |
1068                                 ((u32)buf[sys_info_addr + 33] << 16) |
1069                                 ((u32)buf[sys_info_addr + 34] << 8) |
1070                                 buf[sys_info_addr + 35];
1071                 dev_dbg(rtsx_dev(chip), "xc_total_blk = 0x%x, xc_blk_size = 0x%x\n",
1072                         xc_total_blk, xc_blk_size);
1073         } else {
1074                 total_blk = ((u16)buf[sys_info_addr + 6] << 8) |
1075                         buf[sys_info_addr + 7];
1076                 blk_size = ((u16)buf[sys_info_addr + 2] << 8) |
1077                         buf[sys_info_addr + 3];
1078                 dev_dbg(rtsx_dev(chip), "total_blk = 0x%x, blk_size = 0x%x\n",
1079                         total_blk, blk_size);
1080         }
1081 #else
1082         total_blk = ((u16)buf[sys_info_addr + 6] << 8) | buf[sys_info_addr + 7];
1083         blk_size = ((u16)buf[sys_info_addr + 2] << 8) | buf[sys_info_addr + 3];
1084         dev_dbg(rtsx_dev(chip), "total_blk = 0x%x, blk_size = 0x%x\n",
1085                 total_blk, blk_size);
1086 #endif
1087 
1088         dev_dbg(rtsx_dev(chip), "class_code = 0x%x, device_type = 0x%x, sub_class = 0x%x\n",
1089                 class_code, device_type, sub_class);
1090 
1091         memcpy(ms_card->raw_sys_info, buf + sys_info_addr, 96);
1092 #ifdef SUPPORT_PCGL_1P18
1093         memcpy(ms_card->raw_model_name, buf + model_name_addr, 48);
1094 #endif
1095 
1096         kfree(buf);
1097 
1098 #ifdef SUPPORT_MSXC
1099         if (CHK_MSXC(ms_card)) {
1100                 if (class_code != 0x03)
1101                         return STATUS_FAIL;
1102         } else {
1103                 if (class_code != 0x02)
1104                         return STATUS_FAIL;
1105         }
1106 #else
1107         if (class_code != 0x02)
1108                 return STATUS_FAIL;
1109 #endif
1110 
1111         if (device_type != 0x00) {
1112                 if ((device_type == 0x01) || (device_type == 0x02) ||
1113                     (device_type == 0x03)) {
1114                         chip->card_wp |= MS_CARD;
1115                 } else {
1116                         return STATUS_FAIL;
1117                 }
1118         }
1119 
1120         if (sub_class & 0xC0)
1121                 return STATUS_FAIL;
1122 
1123         dev_dbg(rtsx_dev(chip), "class_code: 0x%x, device_type: 0x%x, sub_class: 0x%x\n",
1124                 class_code, device_type, sub_class);
1125 
1126 #ifdef SUPPORT_MSXC
1127         if (CHK_MSXC(ms_card)) {
1128                 chip->capacity[chip->card2lun[MS_CARD]] =
1129                         ms_card->capacity = xc_total_blk * xc_blk_size;
1130         } else {
1131                 chip->capacity[chip->card2lun[MS_CARD]] =
1132                         ms_card->capacity = total_blk * blk_size;
1133         }
1134 #else
1135         ms_card->capacity = total_blk * blk_size;
1136         chip->capacity[chip->card2lun[MS_CARD]] = ms_card->capacity;
1137 #endif
1138 
1139         return STATUS_SUCCESS;
1140 }
1141 
1142 #ifdef SUPPORT_MAGIC_GATE
1143 static int mg_set_tpc_para_sub(struct rtsx_chip *chip,
1144                                int type, u8 mg_entry_num);
1145 #endif
1146 
1147 static int reset_ms_pro(struct rtsx_chip *chip)
1148 {
1149         struct ms_info *ms_card = &chip->ms_card;
1150         int retval;
1151 #ifdef XC_POWERCLASS
1152         u8 change_power_class;
1153 
1154         if (chip->ms_power_class_en & 0x02)
1155                 change_power_class = 2;
1156         else if (chip->ms_power_class_en & 0x01)
1157                 change_power_class = 1;
1158         else
1159                 change_power_class = 0;
1160 #endif
1161 
1162 #ifdef XC_POWERCLASS
1163 retry:
1164 #endif
1165         retval = ms_pro_reset_flow(chip, 1);
1166         if (retval != STATUS_SUCCESS) {
1167                 if (ms_card->switch_8bit_fail) {
1168                         retval = ms_pro_reset_flow(chip, 0);
1169                         if (retval != STATUS_SUCCESS)
1170                                 return STATUS_FAIL;
1171                 } else {
1172                         return STATUS_FAIL;
1173                 }
1174         }
1175 
1176         retval = ms_read_attribute_info(chip);
1177         if (retval != STATUS_SUCCESS)
1178                 return STATUS_FAIL;
1179 
1180 #ifdef XC_POWERCLASS
1181         if (CHK_HG8BIT(ms_card))
1182                 change_power_class = 0;
1183 
1184         if (change_power_class && CHK_MSXC(ms_card)) {
1185                 u8 power_class_en = chip->ms_power_class_en;
1186 
1187                 dev_dbg(rtsx_dev(chip), "power_class_en = 0x%x\n",
1188                         power_class_en);
1189                 dev_dbg(rtsx_dev(chip), "change_power_class = %d\n",
1190                         change_power_class);
1191 
1192                 if (change_power_class)
1193                         power_class_en &= (1 << (change_power_class - 1));
1194                 else
1195                         power_class_en = 0;
1196 
1197                 if (power_class_en) {
1198                         u8 power_class_mode =
1199                                 (ms_card->raw_sys_info[46] & 0x18) >> 3;
1200                         dev_dbg(rtsx_dev(chip), "power_class_mode = 0x%x",
1201                                 power_class_mode);
1202                         if (change_power_class > power_class_mode)
1203                                 change_power_class = power_class_mode;
1204                         if (change_power_class) {
1205                                 retval = msxc_change_power(chip,
1206                                                            change_power_class);
1207                                 if (retval != STATUS_SUCCESS) {
1208                                         change_power_class--;
1209                                         goto retry;
1210                                 }
1211                         }
1212                 }
1213         }
1214 #endif
1215 
1216 #ifdef SUPPORT_MAGIC_GATE
1217         retval = mg_set_tpc_para_sub(chip, 0, 0);
1218         if (retval != STATUS_SUCCESS)
1219                 return STATUS_FAIL;
1220 #endif
1221 
1222         if (CHK_HG8BIT(ms_card))
1223                 chip->card_bus_width[chip->card2lun[MS_CARD]] = 8;
1224         else
1225                 chip->card_bus_width[chip->card2lun[MS_CARD]] = 4;
1226 
1227         return STATUS_SUCCESS;
1228 }
1229 
1230 static int ms_read_status_reg(struct rtsx_chip *chip)
1231 {
1232         int retval;
1233         u8 val[2];
1234 
1235         retval = ms_set_rw_reg_addr(chip, StatusReg0, 2, 0, 0);
1236         if (retval != STATUS_SUCCESS)
1237                 return STATUS_FAIL;
1238 
1239         retval = ms_read_bytes(chip, READ_REG, 2, NO_WAIT_INT, val, 2);
1240         if (retval != STATUS_SUCCESS)
1241                 return STATUS_FAIL;
1242 
1243         if (val[1] & (STS_UCDT | STS_UCEX | STS_UCFG)) {
1244                 ms_set_err_code(chip, MS_FLASH_READ_ERROR);
1245                 return STATUS_FAIL;
1246         }
1247 
1248         return STATUS_SUCCESS;
1249 }
1250 
1251 static int ms_read_extra_data(struct rtsx_chip *chip,
1252                               u16 block_addr, u8 page_num, u8 *buf, int buf_len)
1253 {
1254         struct ms_info *ms_card = &chip->ms_card;
1255         int retval, i;
1256         u8 val, data[10];
1257 
1258         retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
1259                                     SystemParm, 6);
1260         if (retval != STATUS_SUCCESS)
1261                 return STATUS_FAIL;
1262 
1263         if (CHK_MS4BIT(ms_card)) {
1264                 /* Parallel interface */
1265                 data[0] = 0x88;
1266         } else {
1267                 /* Serial interface */
1268                 data[0] = 0x80;
1269         }
1270         data[1] = 0;
1271         data[2] = (u8)(block_addr >> 8);
1272         data[3] = (u8)block_addr;
1273         data[4] = 0x40;
1274         data[5] = page_num;
1275 
1276         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
1277                 retval = ms_write_bytes(chip, WRITE_REG, 6, NO_WAIT_INT,
1278                                         data, 6);
1279                 if (retval == STATUS_SUCCESS)
1280                         break;
1281         }
1282         if (i == MS_MAX_RETRY_COUNT)
1283                 return STATUS_FAIL;
1284 
1285         ms_set_err_code(chip, MS_NO_ERROR);
1286 
1287         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
1288                 retval = ms_send_cmd(chip, BLOCK_READ, WAIT_INT);
1289                 if (retval == STATUS_SUCCESS)
1290                         break;
1291         }
1292         if (i == MS_MAX_RETRY_COUNT)
1293                 return STATUS_FAIL;
1294 
1295         ms_set_err_code(chip, MS_NO_ERROR);
1296         retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
1297         if (retval != STATUS_SUCCESS)
1298                 return STATUS_FAIL;
1299 
1300         if (val & INT_REG_CMDNK) {
1301                 ms_set_err_code(chip, MS_CMD_NK);
1302                 return STATUS_FAIL;
1303         }
1304         if (val & INT_REG_CED) {
1305                 if (val & INT_REG_ERR) {
1306                         retval = ms_read_status_reg(chip);
1307                         if (retval != STATUS_SUCCESS)
1308                                 return STATUS_FAIL;
1309 
1310                         retval = ms_set_rw_reg_addr(chip, OverwriteFlag,
1311                                                     MS_EXTRA_SIZE, SystemParm,
1312                                                     6);
1313                         if (retval != STATUS_SUCCESS)
1314                                 return STATUS_FAIL;
1315                 }
1316         }
1317 
1318         retval = ms_read_bytes(chip, READ_REG, MS_EXTRA_SIZE, NO_WAIT_INT,
1319                                data, MS_EXTRA_SIZE);
1320         if (retval != STATUS_SUCCESS)
1321                 return STATUS_FAIL;
1322 
1323         if (buf && buf_len) {
1324                 if (buf_len > MS_EXTRA_SIZE)
1325                         buf_len = MS_EXTRA_SIZE;
1326                 memcpy(buf, data, buf_len);
1327         }
1328 
1329         return STATUS_SUCCESS;
1330 }
1331 
1332 static int ms_write_extra_data(struct rtsx_chip *chip, u16 block_addr,
1333                                u8 page_num, u8 *buf, int buf_len)
1334 {
1335         struct ms_info *ms_card = &chip->ms_card;
1336         int retval, i;
1337         u8 val, data[16];
1338 
1339         if (!buf || (buf_len < MS_EXTRA_SIZE))
1340                 return STATUS_FAIL;
1341 
1342         retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
1343                                     SystemParm, 6 + MS_EXTRA_SIZE);
1344         if (retval != STATUS_SUCCESS)
1345                 return STATUS_FAIL;
1346 
1347         if (CHK_MS4BIT(ms_card))
1348                 data[0] = 0x88;
1349         else
1350                 data[0] = 0x80;
1351 
1352         data[1] = 0;
1353         data[2] = (u8)(block_addr >> 8);
1354         data[3] = (u8)block_addr;
1355         data[4] = 0x40;
1356         data[5] = page_num;
1357 
1358         for (i = 6; i < MS_EXTRA_SIZE + 6; i++)
1359                 data[i] = buf[i - 6];
1360 
1361         retval = ms_write_bytes(chip, WRITE_REG, (6 + MS_EXTRA_SIZE),
1362                                 NO_WAIT_INT, data, 16);
1363         if (retval != STATUS_SUCCESS)
1364                 return STATUS_FAIL;
1365 
1366         retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);
1367         if (retval != STATUS_SUCCESS)
1368                 return STATUS_FAIL;
1369 
1370         ms_set_err_code(chip, MS_NO_ERROR);
1371         retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
1372         if (retval != STATUS_SUCCESS)
1373                 return STATUS_FAIL;
1374 
1375         if (val & INT_REG_CMDNK) {
1376                 ms_set_err_code(chip, MS_CMD_NK);
1377                 return STATUS_FAIL;
1378         }
1379         if (val & INT_REG_CED) {
1380                 if (val & INT_REG_ERR) {
1381                         ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);
1382                         return STATUS_FAIL;
1383                 }
1384         }
1385 
1386         return STATUS_SUCCESS;
1387 }
1388 
1389 static int ms_read_page(struct rtsx_chip *chip, u16 block_addr, u8 page_num)
1390 {
1391         struct ms_info *ms_card = &chip->ms_card;
1392         int retval;
1393         u8 val, data[6];
1394 
1395         retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
1396                                     SystemParm, 6);
1397         if (retval != STATUS_SUCCESS)
1398                 return STATUS_FAIL;
1399 
1400         if (CHK_MS4BIT(ms_card))
1401                 data[0] = 0x88;
1402         else
1403                 data[0] = 0x80;
1404 
1405         data[1] = 0;
1406         data[2] = (u8)(block_addr >> 8);
1407         data[3] = (u8)block_addr;
1408         data[4] = 0x20;
1409         data[5] = page_num;
1410 
1411         retval = ms_write_bytes(chip, WRITE_REG, 6, NO_WAIT_INT, data, 6);
1412         if (retval != STATUS_SUCCESS)
1413                 return STATUS_FAIL;
1414 
1415         retval = ms_send_cmd(chip, BLOCK_READ, WAIT_INT);
1416         if (retval != STATUS_SUCCESS)
1417                 return STATUS_FAIL;
1418 
1419         ms_set_err_code(chip, MS_NO_ERROR);
1420         retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
1421         if (retval != STATUS_SUCCESS)
1422                 return STATUS_FAIL;
1423 
1424         if (val & INT_REG_CMDNK) {
1425                 ms_set_err_code(chip, MS_CMD_NK);
1426                 return STATUS_FAIL;
1427         }
1428 
1429         if (val & INT_REG_CED) {
1430                 if (val & INT_REG_ERR) {
1431                         if (!(val & INT_REG_BREQ)) {
1432                                 ms_set_err_code(chip,  MS_FLASH_READ_ERROR);
1433                                 return STATUS_FAIL;
1434                         }
1435                         retval = ms_read_status_reg(chip);
1436                         if (retval != STATUS_SUCCESS)
1437                                 ms_set_err_code(chip,  MS_FLASH_WRITE_ERROR);
1438 
1439                 } else {
1440                         if (!(val & INT_REG_BREQ)) {
1441                                 ms_set_err_code(chip, MS_BREQ_ERROR);
1442                                 return STATUS_FAIL;
1443                         }
1444                 }
1445         }
1446 
1447         retval = ms_transfer_tpc(chip, MS_TM_NORMAL_READ, READ_PAGE_DATA,
1448                                  0, NO_WAIT_INT);
1449         if (retval != STATUS_SUCCESS)
1450                 return STATUS_FAIL;
1451 
1452         if (ms_check_err_code(chip, MS_FLASH_WRITE_ERROR))
1453                 return STATUS_FAIL;
1454 
1455         return STATUS_SUCCESS;
1456 }
1457 
1458 static int ms_set_bad_block(struct rtsx_chip *chip, u16 phy_blk)
1459 {
1460         struct ms_info *ms_card = &chip->ms_card;
1461         int retval;
1462         u8 val, data[8], extra[MS_EXTRA_SIZE];
1463 
1464         retval = ms_read_extra_data(chip, phy_blk, 0, extra, MS_EXTRA_SIZE);
1465         if (retval != STATUS_SUCCESS)
1466                 return STATUS_FAIL;
1467 
1468         retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
1469                                     SystemParm, 7);
1470         if (retval != STATUS_SUCCESS)
1471                 return STATUS_FAIL;
1472 
1473         ms_set_err_code(chip, MS_NO_ERROR);
1474 
1475         if (CHK_MS4BIT(ms_card))
1476                 data[0] = 0x88;
1477         else
1478                 data[0] = 0x80;
1479 
1480         data[1] = 0;
1481         data[2] = (u8)(phy_blk >> 8);
1482         data[3] = (u8)phy_blk;
1483         data[4] = 0x80;
1484         data[5] = 0;
1485         data[6] = extra[0] & 0x7F;
1486         data[7] = 0xFF;
1487 
1488         retval = ms_write_bytes(chip, WRITE_REG, 7, NO_WAIT_INT, data, 7);
1489         if (retval != STATUS_SUCCESS)
1490                 return STATUS_FAIL;
1491 
1492         retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);
1493         if (retval != STATUS_SUCCESS)
1494                 return STATUS_FAIL;
1495 
1496         ms_set_err_code(chip, MS_NO_ERROR);
1497         retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
1498         if (retval != STATUS_SUCCESS)
1499                 return STATUS_FAIL;
1500 
1501         if (val & INT_REG_CMDNK) {
1502                 ms_set_err_code(chip, MS_CMD_NK);
1503                 return STATUS_FAIL;
1504         }
1505 
1506         if (val & INT_REG_CED) {
1507                 if (val & INT_REG_ERR) {
1508                         ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);
1509                         return STATUS_FAIL;
1510                 }
1511         }
1512 
1513         return STATUS_SUCCESS;
1514 }
1515 
1516 static int ms_erase_block(struct rtsx_chip *chip, u16 phy_blk)
1517 {
1518         struct ms_info *ms_card = &chip->ms_card;
1519         int retval, i = 0;
1520         u8 val, data[6];
1521 
1522         retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
1523                                     SystemParm, 6);
1524         if (retval != STATUS_SUCCESS)
1525                 return STATUS_FAIL;
1526 
1527         ms_set_err_code(chip, MS_NO_ERROR);
1528 
1529         if (CHK_MS4BIT(ms_card))
1530                 data[0] = 0x88;
1531         else
1532                 data[0] = 0x80;
1533 
1534         data[1] = 0;
1535         data[2] = (u8)(phy_blk >> 8);
1536         data[3] = (u8)phy_blk;
1537         data[4] = 0;
1538         data[5] = 0;
1539 
1540         retval = ms_write_bytes(chip, WRITE_REG, 6, NO_WAIT_INT, data, 6);
1541         if (retval != STATUS_SUCCESS)
1542                 return STATUS_FAIL;
1543 
1544 ERASE_RTY:
1545         retval = ms_send_cmd(chip, BLOCK_ERASE, WAIT_INT);
1546         if (retval != STATUS_SUCCESS)
1547                 return STATUS_FAIL;
1548 
1549         ms_set_err_code(chip, MS_NO_ERROR);
1550         retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
1551         if (retval != STATUS_SUCCESS)
1552                 return STATUS_FAIL;
1553 
1554         if (val & INT_REG_CMDNK) {
1555                 if (i < 3) {
1556                         i++;
1557                         goto ERASE_RTY;
1558                 }
1559 
1560                 ms_set_err_code(chip, MS_CMD_NK);
1561                 ms_set_bad_block(chip, phy_blk);
1562                 return STATUS_FAIL;
1563         }
1564 
1565         if (val & INT_REG_CED) {
1566                 if (val & INT_REG_ERR) {
1567                         ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);
1568                         return STATUS_FAIL;
1569                 }
1570         }
1571 
1572         return STATUS_SUCCESS;
1573 }
1574 
1575 static void ms_set_page_status(u16 log_blk, u8 type, u8 *extra, int extra_len)
1576 {
1577         if (!extra || (extra_len < MS_EXTRA_SIZE))
1578                 return;
1579 
1580         memset(extra, 0xFF, MS_EXTRA_SIZE);
1581 
1582         if (type == setPS_NG) {
1583                 /* set page status as 1:NG,and block status keep 1:OK */
1584                 extra[0] = 0xB8;
1585         } else {
1586                 /* set page status as 0:Data Error,and block status keep 1:OK */
1587                 extra[0] = 0x98;
1588         }
1589 
1590         extra[2] = (u8)(log_blk >> 8);
1591         extra[3] = (u8)log_blk;
1592 }
1593 
1594 static int ms_init_page(struct rtsx_chip *chip, u16 phy_blk, u16 log_blk,
1595                         u8 start_page, u8 end_page)
1596 {
1597         int retval;
1598         u8 extra[MS_EXTRA_SIZE], i;
1599 
1600         memset(extra, 0xff, MS_EXTRA_SIZE);
1601 
1602         extra[0] = 0xf8;        /* Block, page OK, data erased */
1603         extra[1] = 0xff;
1604         extra[2] = (u8)(log_blk >> 8);
1605         extra[3] = (u8)log_blk;
1606 
1607         for (i = start_page; i < end_page; i++) {
1608                 if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
1609                         ms_set_err_code(chip, MS_NO_CARD);
1610                         return STATUS_FAIL;
1611                 }
1612 
1613                 retval = ms_write_extra_data(chip, phy_blk, i,
1614                                              extra, MS_EXTRA_SIZE);
1615                 if (retval != STATUS_SUCCESS)
1616                         return STATUS_FAIL;
1617         }
1618 
1619         return STATUS_SUCCESS;
1620 }
1621 
1622 static int ms_copy_page(struct rtsx_chip *chip, u16 old_blk, u16 new_blk,
1623                         u16 log_blk, u8 start_page, u8 end_page)
1624 {
1625         struct ms_info *ms_card = &chip->ms_card;
1626         bool uncorrect_flag = false;
1627         int retval, rty_cnt;
1628         u8 extra[MS_EXTRA_SIZE], val, i, j, data[16];
1629 
1630         dev_dbg(rtsx_dev(chip), "Copy page from 0x%x to 0x%x, logical block is 0x%x\n",
1631                 old_blk, new_blk, log_blk);
1632         dev_dbg(rtsx_dev(chip), "start_page = %d, end_page = %d\n",
1633                 start_page, end_page);
1634 
1635         retval = ms_read_extra_data(chip, new_blk, 0, extra, MS_EXTRA_SIZE);
1636         if (retval != STATUS_SUCCESS)
1637                 return STATUS_FAIL;
1638 
1639         retval = ms_read_status_reg(chip);
1640         if (retval != STATUS_SUCCESS)
1641                 return STATUS_FAIL;
1642 
1643         retval = rtsx_read_register(chip, PPBUF_BASE2, &val);
1644         if (retval)
1645                 return retval;
1646 
1647         if (val & BUF_FULL) {
1648                 retval = ms_send_cmd(chip, CLEAR_BUF, WAIT_INT);
1649                 if (retval != STATUS_SUCCESS)
1650                         return STATUS_FAIL;
1651 
1652                 retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
1653                 if (retval != STATUS_SUCCESS)
1654                         return STATUS_FAIL;
1655 
1656                 if (!(val & INT_REG_CED)) {
1657                         ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);
1658                         return STATUS_FAIL;
1659                 }
1660         }
1661 
1662         for (i = start_page; i < end_page; i++) {
1663                 if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
1664                         ms_set_err_code(chip, MS_NO_CARD);
1665                         return STATUS_FAIL;
1666                 }
1667 
1668                 retval = ms_read_extra_data(chip, old_blk, i, extra,
1669                                             MS_EXTRA_SIZE);
1670                 if (retval != STATUS_SUCCESS)
1671                         return STATUS_FAIL;
1672 
1673                 retval = ms_set_rw_reg_addr(chip, OverwriteFlag,
1674                                             MS_EXTRA_SIZE, SystemParm, 6);
1675                 if (retval != STATUS_SUCCESS)
1676                         return STATUS_FAIL;
1677 
1678                 ms_set_err_code(chip, MS_NO_ERROR);
1679 
1680                 if (CHK_MS4BIT(ms_card))
1681                         data[0] = 0x88;
1682                 else
1683                         data[0] = 0x80;
1684 
1685                 data[1] = 0;
1686                 data[2] = (u8)(old_blk >> 8);
1687                 data[3] = (u8)old_blk;
1688                 data[4] = 0x20;
1689                 data[5] = i;
1690 
1691                 retval = ms_write_bytes(chip, WRITE_REG, 6, NO_WAIT_INT,
1692                                         data, 6);
1693                 if (retval != STATUS_SUCCESS)
1694                         return STATUS_FAIL;
1695 
1696                 retval = ms_send_cmd(chip, BLOCK_READ, WAIT_INT);
1697                 if (retval != STATUS_SUCCESS)
1698                         return STATUS_FAIL;
1699 
1700                 ms_set_err_code(chip, MS_NO_ERROR);
1701                 retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
1702                 if (retval != STATUS_SUCCESS)
1703                         return STATUS_FAIL;
1704 
1705                 if (val & INT_REG_CMDNK) {
1706                         ms_set_err_code(chip, MS_CMD_NK);
1707                         return STATUS_FAIL;
1708                 }
1709 
1710                 if (val & INT_REG_CED) {
1711                         if (val & INT_REG_ERR) {
1712                                 retval = ms_read_status_reg(chip);
1713                                 if (retval != STATUS_SUCCESS) {
1714                                         uncorrect_flag = true;
1715                                         dev_dbg(rtsx_dev(chip), "Uncorrectable error\n");
1716                                 } else {
1717                                         uncorrect_flag = false;
1718                                 }
1719 
1720                                 retval = ms_transfer_tpc(chip,
1721                                                          MS_TM_NORMAL_READ,
1722                                                          READ_PAGE_DATA,
1723                                                          0, NO_WAIT_INT);
1724                                 if (retval != STATUS_SUCCESS)
1725                                         return STATUS_FAIL;
1726 
1727                                 if (uncorrect_flag) {
1728                                         ms_set_page_status(log_blk, setPS_NG,
1729                                                            extra,
1730                                                            MS_EXTRA_SIZE);
1731                                         if (i == 0)
1732                                                 extra[0] &= 0xEF;
1733 
1734                                         ms_write_extra_data(chip, old_blk, i,
1735                                                             extra,
1736                                                             MS_EXTRA_SIZE);
1737                                         dev_dbg(rtsx_dev(chip), "page %d : extra[0] = 0x%x\n",
1738                                                 i, extra[0]);
1739                                         MS_SET_BAD_BLOCK_FLG(ms_card);
1740 
1741                                         ms_set_page_status(log_blk, setPS_Error,
1742                                                            extra,
1743                                                            MS_EXTRA_SIZE);
1744                                         ms_write_extra_data(chip, new_blk, i,
1745                                                             extra,
1746                                                             MS_EXTRA_SIZE);
1747                                         continue;
1748                                 }
1749 
1750                                 for (rty_cnt = 0; rty_cnt < MS_MAX_RETRY_COUNT;
1751                                      rty_cnt++) {
1752                                         retval = ms_transfer_tpc(
1753                                                 chip,
1754                                                 MS_TM_NORMAL_WRITE,
1755                                                 WRITE_PAGE_DATA,
1756                                                 0, NO_WAIT_INT);
1757                                         if (retval == STATUS_SUCCESS)
1758                                                 break;
1759                                 }
1760                                 if (rty_cnt == MS_MAX_RETRY_COUNT)
1761                                         return STATUS_FAIL;
1762                         }
1763 
1764                         if (!(val & INT_REG_BREQ)) {
1765                                 ms_set_err_code(chip, MS_BREQ_ERROR);
1766                                 return STATUS_FAIL;
1767                         }
1768                 }
1769 
1770                 retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
1771                                             SystemParm, (6 + MS_EXTRA_SIZE));
1772 
1773                 ms_set_err_code(chip, MS_NO_ERROR);
1774 
1775                 if (CHK_MS4BIT(ms_card))
1776                         data[0] = 0x88;
1777                 else
1778                         data[0] = 0x80;
1779 
1780                 data[1] = 0;
1781                 data[2] = (u8)(new_blk >> 8);
1782                 data[3] = (u8)new_blk;
1783                 data[4] = 0x20;
1784                 data[5] = i;
1785 
1786                 if ((extra[0] & 0x60) != 0x60)
1787                         data[6] = extra[0];
1788                 else
1789                         data[6] = 0xF8;
1790 
1791                 data[6 + 1] = 0xFF;
1792                 data[6 + 2] = (u8)(log_blk >> 8);
1793                 data[6 + 3] = (u8)log_blk;
1794 
1795                 for (j = 4; j <= MS_EXTRA_SIZE; j++)
1796                         data[6 + j] = 0xFF;
1797 
1798                 retval = ms_write_bytes(chip, WRITE_REG, (6 + MS_EXTRA_SIZE),
1799                                         NO_WAIT_INT, data, 16);
1800                 if (retval != STATUS_SUCCESS)
1801                         return STATUS_FAIL;
1802 
1803                 retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);
1804                 if (retval != STATUS_SUCCESS)
1805                         return STATUS_FAIL;
1806 
1807                 ms_set_err_code(chip, MS_NO_ERROR);
1808                 retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
1809                 if (retval != STATUS_SUCCESS)
1810                         return STATUS_FAIL;
1811 
1812                 if (val & INT_REG_CMDNK) {
1813                         ms_set_err_code(chip, MS_CMD_NK);
1814                         return STATUS_FAIL;
1815                 }
1816 
1817                 if (val & INT_REG_CED) {
1818                         if (val & INT_REG_ERR) {
1819                                 ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);
1820                                 return STATUS_FAIL;
1821                         }
1822                 }
1823 
1824                 if (i == 0) {
1825                         retval = ms_set_rw_reg_addr(chip, OverwriteFlag,
1826                                                     MS_EXTRA_SIZE, SystemParm,
1827                                                     7);
1828                         if (retval != STATUS_SUCCESS)
1829                                 return STATUS_FAIL;
1830 
1831                         ms_set_err_code(chip, MS_NO_ERROR);
1832 
1833                         if (CHK_MS4BIT(ms_card))
1834                                 data[0] = 0x88;
1835                         else
1836                                 data[0] = 0x80;
1837 
1838                         data[1] = 0;
1839                         data[2] = (u8)(old_blk >> 8);
1840                         data[3] = (u8)old_blk;
1841                         data[4] = 0x80;
1842                         data[5] = 0;
1843                         data[6] = 0xEF;
1844                         data[7] = 0xFF;
1845 
1846                         retval = ms_write_bytes(chip, WRITE_REG, 7,
1847                                                 NO_WAIT_INT, data, 8);
1848                         if (retval != STATUS_SUCCESS)
1849                                 return STATUS_FAIL;
1850 
1851                         retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);
1852                         if (retval != STATUS_SUCCESS)
1853                                 return STATUS_FAIL;
1854 
1855                         ms_set_err_code(chip, MS_NO_ERROR);
1856                         retval = ms_read_bytes(chip, GET_INT, 1,
1857                                                NO_WAIT_INT, &val, 1);
1858                         if (retval != STATUS_SUCCESS)
1859                                 return STATUS_FAIL;
1860 
1861                         if (val & INT_REG_CMDNK) {
1862                                 ms_set_err_code(chip, MS_CMD_NK);
1863                                 return STATUS_FAIL;
1864                         }
1865 
1866                         if (val & INT_REG_CED) {
1867                                 if (val & INT_REG_ERR) {
1868                                         ms_set_err_code(chip,
1869                                                         MS_FLASH_WRITE_ERROR);
1870                                         return STATUS_FAIL;
1871                                 }
1872                         }
1873                 }
1874         }
1875 
1876         return STATUS_SUCCESS;
1877 }
1878 
1879 static int reset_ms(struct rtsx_chip *chip)
1880 {
1881         struct ms_info *ms_card = &chip->ms_card;
1882         int retval;
1883         u16 i, reg_addr, block_size;
1884         u8 val, extra[MS_EXTRA_SIZE], j, *ptr;
1885 #ifndef SUPPORT_MAGIC_GATE
1886         u16 eblock_cnt;
1887 #endif
1888 
1889         retval = ms_prepare_reset(chip);
1890         if (retval != STATUS_SUCCESS)
1891                 return STATUS_FAIL;
1892 
1893         ms_card->ms_type |= TYPE_MS;
1894 
1895         retval = ms_send_cmd(chip, MS_RESET, NO_WAIT_INT);
1896         if (retval != STATUS_SUCCESS)
1897                 return STATUS_FAIL;
1898 
1899         retval = ms_read_status_reg(chip);
1900         if (retval != STATUS_SUCCESS)
1901                 return STATUS_FAIL;
1902 
1903         retval = rtsx_read_register(chip, PPBUF_BASE2, &val);
1904         if (retval)
1905                 return retval;
1906 
1907         if (val & WRT_PRTCT)
1908                 chip->card_wp |= MS_CARD;
1909         else
1910                 chip->card_wp &= ~MS_CARD;
1911 
1912         i = 0;
1913 
1914 RE_SEARCH:
1915         /* Search Boot Block */
1916         while (i < (MAX_DEFECTIVE_BLOCK + 2)) {
1917                 if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
1918                         ms_set_err_code(chip, MS_NO_CARD);
1919                         return STATUS_FAIL;
1920                 }
1921 
1922                 retval = ms_read_extra_data(chip, i, 0, extra, MS_EXTRA_SIZE);
1923                 if (retval != STATUS_SUCCESS) {
1924                         i++;
1925                         continue;
1926                 }
1927 
1928                 if (extra[0] & BLOCK_OK) {
1929                         if (!(extra[1] & NOT_BOOT_BLOCK)) {
1930                                 ms_card->boot_block = i;
1931                                 break;
1932                         }
1933                 }
1934                 i++;
1935         }
1936 
1937         if (i == (MAX_DEFECTIVE_BLOCK + 2)) {
1938                 dev_dbg(rtsx_dev(chip), "No boot block found!");
1939                 return STATUS_FAIL;
1940         }
1941 
1942         for (j = 0; j < 3; j++) {
1943                 retval = ms_read_page(chip, ms_card->boot_block, j);
1944                 if (retval != STATUS_SUCCESS) {
1945                         if (ms_check_err_code(chip, MS_FLASH_WRITE_ERROR)) {
1946                                 i = ms_card->boot_block + 1;
1947                                 ms_set_err_code(chip, MS_NO_ERROR);
1948                                 goto RE_SEARCH;
1949                         }
1950                 }
1951         }
1952 
1953         retval = ms_read_page(chip, ms_card->boot_block, 0);
1954         if (retval != STATUS_SUCCESS)
1955                 return STATUS_FAIL;
1956 
1957         /* Read MS system information as sys_info */
1958         rtsx_init_cmd(chip);
1959 
1960         for (i = 0; i < 96; i++)
1961                 rtsx_add_cmd(chip, READ_REG_CMD, PPBUF_BASE2 + 0x1A0 + i, 0, 0);
1962 
1963         retval = rtsx_send_cmd(chip, MS_CARD, 100);
1964         if (retval < 0)
1965                 return STATUS_FAIL;
1966 
1967         ptr = rtsx_get_cmd_data(chip);
1968         memcpy(ms_card->raw_sys_info, ptr, 96);
1969 
1970         /* Read useful block contents */
1971         rtsx_init_cmd(chip);
1972 
1973         rtsx_add_cmd(chip, READ_REG_CMD, HEADER_ID0, 0, 0);
1974         rtsx_add_cmd(chip, READ_REG_CMD, HEADER_ID1, 0, 0);
1975 
1976         for (reg_addr = DISABLED_BLOCK0; reg_addr <= DISABLED_BLOCK3;
1977              reg_addr++)
1978                 rtsx_add_cmd(chip, READ_REG_CMD, reg_addr, 0, 0);
1979 
1980         for (reg_addr = BLOCK_SIZE_0; reg_addr <= PAGE_SIZE_1; reg_addr++)
1981                 rtsx_add_cmd(chip, READ_REG_CMD, reg_addr, 0, 0);
1982 
1983         rtsx_add_cmd(chip, READ_REG_CMD, MS_Device_Type, 0, 0);
1984         rtsx_add_cmd(chip, READ_REG_CMD, MS_4bit_Support, 0, 0);
1985 
1986         retval = rtsx_send_cmd(chip, MS_CARD, 100);
1987         if (retval < 0)
1988                 return STATUS_FAIL;
1989 
1990         ptr = rtsx_get_cmd_data(chip);
1991 
1992         dev_dbg(rtsx_dev(chip), "Boot block data:\n");
1993         dev_dbg(rtsx_dev(chip), "%*ph\n", 16, ptr);
1994 
1995         /* Block ID error
1996          * HEADER_ID0, HEADER_ID1
1997          */
1998         if (ptr[0] != 0x00 || ptr[1] != 0x01) {
1999                 i = ms_card->boot_block + 1;
2000                 goto RE_SEARCH;
2001         }
2002 
2003         /* Page size error
2004          * PAGE_SIZE_0, PAGE_SIZE_1
2005          */
2006         if (ptr[12] != 0x02 || ptr[13] != 0x00) {
2007                 i = ms_card->boot_block + 1;
2008                 goto RE_SEARCH;
2009         }
2010 
2011         if ((ptr[14] == 1) || (ptr[14] == 3))
2012                 chip->card_wp |= MS_CARD;
2013 
2014         /* BLOCK_SIZE_0, BLOCK_SIZE_1 */
2015         block_size = ((u16)ptr[6] << 8) | ptr[7];
2016         if (block_size == 0x0010) {
2017                 /* Block size 16KB */
2018                 ms_card->block_shift = 5;
2019                 ms_card->page_off = 0x1F;
2020         } else if (block_size == 0x0008) {
2021                 /* Block size 8KB */
2022                 ms_card->block_shift = 4;
2023                 ms_card->page_off = 0x0F;
2024         }
2025 
2026         /* BLOCK_COUNT_0, BLOCK_COUNT_1 */
2027         ms_card->total_block = ((u16)ptr[8] << 8) | ptr[9];
2028 
2029 #ifdef SUPPORT_MAGIC_GATE
2030         j = ptr[10];
2031 
2032         if (ms_card->block_shift == 4)  { /* 4MB or 8MB */
2033                 if (j < 2)  { /* Effective block for 4MB: 0x1F0 */
2034                         ms_card->capacity = 0x1EE0;
2035                 } else { /* Effective block for 8MB: 0x3E0 */
2036                         ms_card->capacity = 0x3DE0;
2037                 }
2038         } else  { /* 16MB, 32MB, 64MB or 128MB */
2039                 if (j < 5)  { /* Effective block for 16MB: 0x3E0 */
2040                         ms_card->capacity = 0x7BC0;
2041                 } else if (j < 0xA) { /* Effective block for 32MB: 0x7C0 */
2042                         ms_card->capacity = 0xF7C0;
2043                 } else if (j < 0x11) { /* Effective block for 64MB: 0xF80 */
2044                         ms_card->capacity = 0x1EF80;
2045                 } else { /* Effective block for 128MB: 0x1F00 */
2046                         ms_card->capacity = 0x3DF00;
2047                 }
2048         }
2049 #else
2050         /* EBLOCK_COUNT_0, EBLOCK_COUNT_1 */
2051         eblock_cnt = ((u16)ptr[10] << 8) | ptr[11];
2052 
2053         ms_card->capacity = ((u32)eblock_cnt - 2) << ms_card->block_shift;
2054 #endif
2055 
2056         chip->capacity[chip->card2lun[MS_CARD]] = ms_card->capacity;
2057 
2058         /* Switch I/F Mode */
2059         if (ptr[15]) {
2060                 retval = ms_set_rw_reg_addr(chip, 0, 0, SystemParm, 1);
2061                 if (retval != STATUS_SUCCESS)
2062                         return STATUS_FAIL;
2063 
2064                 retval = rtsx_write_register(chip, PPBUF_BASE2, 0xFF, 0x88);
2065                 if (retval)
2066                         return retval;
2067 
2068                 retval = rtsx_write_register(chip, PPBUF_BASE2 + 1, 0xFF, 0);
2069                 if (retval)
2070                         return retval;
2071 
2072                 retval = ms_transfer_tpc(chip, MS_TM_WRITE_BYTES, WRITE_REG, 1,
2073                                          NO_WAIT_INT);
2074                 if (retval != STATUS_SUCCESS)
2075                         return STATUS_FAIL;
2076 
2077                 retval = rtsx_write_register(chip, MS_CFG,
2078                                              0x58 | MS_NO_CHECK_INT,
2079                                              MS_BUS_WIDTH_4 |
2080                                              PUSH_TIME_ODD |
2081                                              MS_NO_CHECK_INT);
2082                 if (retval)
2083                         return retval;
2084 
2085                 ms_card->ms_type |= MS_4BIT;
2086         }
2087 
2088         if (CHK_MS4BIT(ms_card))
2089                 chip->card_bus_width[chip->card2lun[MS_CARD]] = 4;
2090         else
2091                 chip->card_bus_width[chip->card2lun[MS_CARD]] = 1;
2092 
2093         return STATUS_SUCCESS;
2094 }
2095 
2096 static int ms_init_l2p_tbl(struct rtsx_chip *chip)
2097 {
2098         struct ms_info *ms_card = &chip->ms_card;
2099         int size, i, seg_no, retval;
2100         u16 defect_block, reg_addr;
2101         u8 val1, val2;
2102 
2103         ms_card->segment_cnt = ms_card->total_block >> 9;
2104         dev_dbg(rtsx_dev(chip), "ms_card->segment_cnt = %d\n",
2105                 ms_card->segment_cnt);
2106 
2107         size = ms_card->segment_cnt * sizeof(struct zone_entry);
2108         ms_card->segment = vzalloc(size);
2109         if (!ms_card->segment)
2110                 return STATUS_FAIL;
2111 
2112         retval = ms_read_page(chip, ms_card->boot_block, 1);
2113         if (retval != STATUS_SUCCESS)
2114                 goto INIT_FAIL;
2115 
2116         reg_addr = PPBUF_BASE2;
2117         for (i = 0; i < (((ms_card->total_block >> 9) * 10) + 1); i++) {
2118                 int block_no;
2119 
2120                 retval = rtsx_read_register(chip, reg_addr++, &val1);
2121                 if (retval != STATUS_SUCCESS)
2122                         goto INIT_FAIL;
2123 
2124                 retval = rtsx_read_register(chip, reg_addr++, &val2);
2125                 if (retval != STATUS_SUCCESS)
2126                         goto INIT_FAIL;
2127 
2128                 defect_block = ((u16)val1 << 8) | val2;
2129                 if (defect_block == 0xFFFF)
2130                         break;
2131 
2132                 seg_no = defect_block / 512;
2133 
2134                 block_no = ms_card->segment[seg_no].disable_count++;
2135                 ms_card->segment[seg_no].defect_list[block_no] = defect_block;
2136         }
2137 
2138         for (i = 0; i < ms_card->segment_cnt; i++) {
2139                 ms_card->segment[i].build_flag = 0;
2140                 ms_card->segment[i].l2p_table = NULL;
2141                 ms_card->segment[i].free_table = NULL;
2142                 ms_card->segment[i].get_index = 0;
2143                 ms_card->segment[i].set_index = 0;
2144                 ms_card->segment[i].unused_blk_cnt = 0;
2145 
2146                 dev_dbg(rtsx_dev(chip), "defective block count of segment %d is %d\n",
2147                         i, ms_card->segment[i].disable_count);
2148         }
2149 
2150         return STATUS_SUCCESS;
2151 
2152 INIT_FAIL:
2153         vfree(ms_card->segment);
2154         ms_card->segment = NULL;
2155 
2156         return STATUS_FAIL;
2157 }
2158 
2159 static u16 ms_get_l2p_tbl(struct rtsx_chip *chip, int seg_no, u16 log_off)
2160 {
2161         struct ms_info *ms_card = &chip->ms_card;
2162         struct zone_entry *segment;
2163 
2164         if (!ms_card->segment)
2165                 return 0xFFFF;
2166 
2167         segment = &ms_card->segment[seg_no];
2168 
2169         if (segment->l2p_table)
2170                 return segment->l2p_table[log_off];
2171 
2172         return 0xFFFF;
2173 }
2174 
2175 static void ms_set_l2p_tbl(struct rtsx_chip *chip,
2176                            int seg_no, u16 log_off, u16 phy_blk)
2177 {
2178         struct ms_info *ms_card = &chip->ms_card;
2179         struct zone_entry *segment;
2180 
2181         if (!ms_card->segment)
2182                 return;
2183 
2184         segment = &ms_card->segment[seg_no];
2185         if (segment->l2p_table)
2186                 segment->l2p_table[log_off] = phy_blk;
2187 }
2188 
2189 static void ms_set_unused_block(struct rtsx_chip *chip, u16 phy_blk)
2190 {
2191         struct ms_info *ms_card = &chip->ms_card;
2192         struct zone_entry *segment;
2193         int seg_no;
2194 
2195         seg_no = (int)phy_blk >> 9;
2196         segment = &ms_card->segment[seg_no];
2197 
2198         segment->free_table[segment->set_index++] = phy_blk;
2199         if (segment->set_index >= MS_FREE_TABLE_CNT)
2200                 segment->set_index = 0;
2201 
2202         segment->unused_blk_cnt++;
2203 }
2204 
2205 static u16 ms_get_unused_block(struct rtsx_chip *chip, int seg_no)
2206 {
2207         struct ms_info *ms_card = &chip->ms_card;
2208         struct zone_entry *segment;
2209         u16 phy_blk;
2210 
2211         segment = &ms_card->segment[seg_no];
2212 
2213         if (segment->unused_blk_cnt <= 0)
2214                 return 0xFFFF;
2215 
2216         phy_blk = segment->free_table[segment->get_index];
2217         segment->free_table[segment->get_index++] = 0xFFFF;
2218         if (segment->get_index >= MS_FREE_TABLE_CNT)
2219                 segment->get_index = 0;
2220 
2221         segment->unused_blk_cnt--;
2222 
2223         return phy_blk;
2224 }
2225 
2226 static const unsigned short ms_start_idx[] = {0, 494, 990, 1486, 1982, 2478,
2227                                               2974, 3470, 3966, 4462, 4958,
2228                                               5454, 5950, 6446, 6942, 7438,
2229                                               7934};
2230 
2231 static int ms_arbitrate_l2p(struct rtsx_chip *chip, u16 phy_blk,
2232                             u16 log_off, u8 us1, u8 us2)
2233 {
2234         struct ms_info *ms_card = &chip->ms_card;
2235         struct zone_entry *segment;
2236         int seg_no;
2237         u16 tmp_blk;
2238 
2239         seg_no = (int)phy_blk >> 9;
2240         segment = &ms_card->segment[seg_no];
2241         tmp_blk = segment->l2p_table[log_off];
2242 
2243         if (us1 != us2) {
2244                 if (us1 == 0) {
2245                         if (!(chip->card_wp & MS_CARD))
2246                                 ms_erase_block(chip, tmp_blk);
2247 
2248                         ms_set_unused_block(chip, tmp_blk);
2249                         segment->l2p_table[log_off] = phy_blk;
2250                 } else {
2251                         if (!(chip->card_wp & MS_CARD))
2252                                 ms_erase_block(chip, phy_blk);
2253 
2254                         ms_set_unused_block(chip, phy_blk);
2255                 }
2256         } else {
2257                 if (phy_blk < tmp_blk) {
2258                         if (!(chip->card_wp & MS_CARD))
2259                                 ms_erase_block(chip, phy_blk);
2260 
2261                         ms_set_unused_block(chip, phy_blk);
2262                 } else {
2263                         if (!(chip->card_wp & MS_CARD))
2264                                 ms_erase_block(chip, tmp_blk);
2265 
2266                         ms_set_unused_block(chip, tmp_blk);
2267                         segment->l2p_table[log_off] = phy_blk;
2268                 }
2269         }
2270 
2271         return STATUS_SUCCESS;
2272 }
2273 
2274 static int ms_build_l2p_tbl(struct rtsx_chip *chip, int seg_no)
2275 {
2276         struct ms_info *ms_card = &chip->ms_card;
2277         struct zone_entry *segment;
2278         bool defect_flag;
2279         int retval, table_size, disable_cnt, i;
2280         u16 start, end, phy_blk, log_blk, tmp_blk, idx;
2281         u8 extra[MS_EXTRA_SIZE], us1, us2;
2282 
2283         dev_dbg(rtsx_dev(chip), "%s: %d\n", __func__, seg_no);
2284 
2285         if (!ms_card->segment) {
2286                 retval = ms_init_l2p_tbl(chip);
2287                 if (retval != STATUS_SUCCESS)
2288                         return retval;
2289         }
2290 
2291         if (ms_card->segment[seg_no].build_flag) {
2292                 dev_dbg(rtsx_dev(chip), "l2p table of segment %d has been built\n",
2293                         seg_no);
2294                 return STATUS_SUCCESS;
2295         }
2296 
2297         if (seg_no == 0)
2298                 table_size = 494;
2299         else
2300                 table_size = 496;
2301 
2302         segment = &ms_card->segment[seg_no];
2303 
2304         if (!segment->l2p_table) {
2305                 segment->l2p_table = vmalloc(array_size(table_size, 2));
2306                 if (!segment->l2p_table)
2307                         goto BUILD_FAIL;
2308         }
2309         memset((u8 *)(segment->l2p_table), 0xff, table_size * 2);
2310 
2311         if (!segment->free_table) {
2312                 segment->free_table = vmalloc(MS_FREE_TABLE_CNT * 2);
2313                 if (!segment->free_table)
2314                         goto BUILD_FAIL;
2315         }
2316         memset((u8 *)(segment->free_table), 0xff, MS_FREE_TABLE_CNT * 2);
2317 
2318         start = (u16)seg_no << 9;
2319         end = (u16)(seg_no + 1) << 9;
2320 
2321         disable_cnt = segment->disable_count;
2322 
2323         segment->get_index = 0;
2324         segment->set_index = 0;
2325         segment->unused_blk_cnt = 0;
2326 
2327         for (phy_blk = start; phy_blk < end; phy_blk++) {
2328                 if (disable_cnt) {
2329                         defect_flag = false;
2330                         for (i = 0; i < segment->disable_count; i++) {
2331                                 if (phy_blk == segment->defect_list[i]) {
2332                                         defect_flag = true;
2333                                         break;
2334                                 }
2335                         }
2336                         if (defect_flag) {
2337                                 disable_cnt--;
2338                                 continue;
2339                         }
2340                 }
2341 
2342                 retval = ms_read_extra_data(chip, phy_blk, 0,
2343                                             extra, MS_EXTRA_SIZE);
2344                 if (retval != STATUS_SUCCESS) {
2345                         dev_dbg(rtsx_dev(chip), "read extra data fail\n");
2346                         ms_set_bad_block(chip, phy_blk);
2347                         continue;
2348                 }
2349 
2350                 if (seg_no == ms_card->segment_cnt - 1) {
2351                         if (!(extra[1] & NOT_TRANSLATION_TABLE)) {
2352                                 if (!(chip->card_wp & MS_CARD)) {
2353                                         retval = ms_erase_block(chip, phy_blk);
2354                                         if (retval != STATUS_SUCCESS)
2355                                                 continue;
2356                                         extra[2] = 0xff;
2357                                         extra[3] = 0xff;
2358                                 }
2359                         }
2360                 }
2361 
2362                 if (!(extra[0] & BLOCK_OK))
2363                         continue;
2364                 if (!(extra[1] & NOT_BOOT_BLOCK))
2365                         continue;
2366                 if ((extra[0] & PAGE_OK) != PAGE_OK)
2367                         continue;
2368 
2369                 log_blk = ((u16)extra[2] << 8) | extra[3];
2370 
2371                 if (log_blk == 0xFFFF) {
2372                         if (!(chip->card_wp & MS_CARD)) {
2373                                 retval = ms_erase_block(chip, phy_blk);
2374                                 if (retval != STATUS_SUCCESS)
2375                                         continue;
2376                         }
2377                         ms_set_unused_block(chip, phy_blk);
2378                         continue;
2379                 }
2380 
2381                 if ((log_blk < ms_start_idx[seg_no]) ||
2382                     (log_blk >= ms_start_idx[seg_no + 1])) {
2383                         if (!(chip->card_wp & MS_CARD)) {
2384                                 retval = ms_erase_block(chip, phy_blk);
2385                                 if (retval != STATUS_SUCCESS)
2386                                         continue;
2387                         }
2388                         ms_set_unused_block(chip, phy_blk);
2389                         continue;
2390                 }
2391 
2392                 idx = log_blk - ms_start_idx[seg_no];
2393 
2394                 if (segment->l2p_table[idx] == 0xFFFF) {
2395                         segment->l2p_table[idx] = phy_blk;
2396                         continue;
2397                 }
2398 
2399                 us1 = extra[0] & 0x10;
2400                 tmp_blk = segment->l2p_table[idx];
2401                 retval = ms_read_extra_data(chip, tmp_blk, 0,
2402                                             extra, MS_EXTRA_SIZE);
2403                 if (retval != STATUS_SUCCESS)
2404                         continue;
2405                 us2 = extra[0] & 0x10;
2406 
2407                 (void)ms_arbitrate_l2p(chip, phy_blk,
2408                                 log_blk - ms_start_idx[seg_no], us1, us2);
2409                 continue;
2410         }
2411 
2412         segment->build_flag = 1;
2413 
2414         dev_dbg(rtsx_dev(chip), "unused block count: %d\n",
2415                 segment->unused_blk_cnt);
2416 
2417         /* Logical Address Confirmation Process */
2418         if (seg_no == ms_card->segment_cnt - 1) {
2419                 if (segment->unused_blk_cnt < 2)
2420                         chip->card_wp |= MS_CARD;
2421         } else {
2422                 if (segment->unused_blk_cnt < 1)
2423                         chip->card_wp |= MS_CARD;
2424         }
2425 
2426         if (chip->card_wp & MS_CARD)
2427                 return STATUS_SUCCESS;
2428 
2429         for (log_blk = ms_start_idx[seg_no];
2430              log_blk < ms_start_idx[seg_no + 1]; log_blk++) {
2431                 idx = log_blk - ms_start_idx[seg_no];
2432                 if (segment->l2p_table[idx] == 0xFFFF) {
2433                         phy_blk = ms_get_unused_block(chip, seg_no);
2434                         if (phy_blk == 0xFFFF) {
2435                                 chip->card_wp |= MS_CARD;
2436                                 return STATUS_SUCCESS;
2437                         }
2438                         retval = ms_init_page(chip, phy_blk, log_blk, 0, 1);
2439                         if (retval != STATUS_SUCCESS)
2440                                 goto BUILD_FAIL;
2441 
2442                         segment->l2p_table[idx] = phy_blk;
2443                         if (seg_no == ms_card->segment_cnt - 1) {
2444                                 if (segment->unused_blk_cnt < 2) {
2445                                         chip->card_wp |= MS_CARD;
2446                                         return STATUS_SUCCESS;
2447                                 }
2448                         } else {
2449                                 if (segment->unused_blk_cnt < 1) {
2450                                         chip->card_wp |= MS_CARD;
2451                                         return STATUS_SUCCESS;
2452                                 }
2453                         }
2454                 }
2455         }
2456 
2457         /* Make boot block be the first normal block */
2458         if (seg_no == 0) {
2459                 for (log_blk = 0; log_blk < 494; log_blk++) {
2460                         tmp_blk = segment->l2p_table[log_blk];
2461                         if (tmp_blk < ms_card->boot_block) {
2462                                 dev_dbg(rtsx_dev(chip), "Boot block is not the first normal block.\n");
2463 
2464                                 if (chip->card_wp & MS_CARD)
2465                                         break;
2466 
2467                                 phy_blk = ms_get_unused_block(chip, 0);
2468                                 retval = ms_copy_page(chip, tmp_blk, phy_blk,
2469                                                       log_blk, 0,
2470                                                       ms_card->page_off + 1);
2471                                 if (retval != STATUS_SUCCESS)
2472                                         return STATUS_FAIL;
2473 
2474                                 segment->l2p_table[log_blk] = phy_blk;
2475 
2476                                 retval = ms_set_bad_block(chip, tmp_blk);
2477                                 if (retval != STATUS_SUCCESS)
2478                                         return STATUS_FAIL;
2479                         }
2480                 }
2481         }
2482 
2483         return STATUS_SUCCESS;
2484 
2485 BUILD_FAIL:
2486         segment->build_flag = 0;
2487         vfree(segment->l2p_table);
2488         segment->l2p_table = NULL;
2489         vfree(segment->free_table);
2490         segment->free_table = NULL;
2491 
2492         return STATUS_FAIL;
2493 }
2494 
2495 int reset_ms_card(struct rtsx_chip *chip)
2496 {
2497         struct ms_info *ms_card = &chip->ms_card;
2498         int seg_no = ms_card->total_block / 512 - 1;
2499         int retval;
2500 
2501         memset(ms_card, 0, sizeof(struct ms_info));
2502 
2503         retval = enable_card_clock(chip, MS_CARD);
2504         if (retval != STATUS_SUCCESS)
2505                 return STATUS_FAIL;
2506 
2507         retval = select_card(chip, MS_CARD);
2508         if (retval != STATUS_SUCCESS)
2509                 return STATUS_FAIL;
2510 
2511         ms_card->ms_type = 0;
2512 
2513         retval = reset_ms_pro(chip);
2514         if (retval != STATUS_SUCCESS) {
2515                 if (ms_card->check_ms_flow) {
2516                         retval = reset_ms(chip);
2517                         if (retval != STATUS_SUCCESS)
2518                                 return STATUS_FAIL;
2519                 } else {
2520                         return STATUS_FAIL;
2521                 }
2522         }
2523 
2524         retval = ms_set_init_para(chip);
2525         if (retval != STATUS_SUCCESS)
2526                 return STATUS_FAIL;
2527 
2528         if (!CHK_MSPRO(ms_card)) {
2529                 /* Build table for the last segment,
2530                  * to check if L2P table block exists, erasing it
2531                  */
2532                 retval = ms_build_l2p_tbl(chip, seg_no);
2533                 if (retval != STATUS_SUCCESS)
2534                         return STATUS_FAIL;
2535         }
2536 
2537         dev_dbg(rtsx_dev(chip), "ms_card->ms_type = 0x%x\n", ms_card->ms_type);
2538 
2539         return STATUS_SUCCESS;
2540 }
2541 
2542 static int mspro_set_rw_cmd(struct rtsx_chip *chip,
2543                             u32 start_sec, u16 sec_cnt, u8 cmd)
2544 {
2545         int retval, i;
2546         u8 data[8];
2547 
2548         data[0] = cmd;
2549         data[1] = (u8)(sec_cnt >> 8);
2550         data[2] = (u8)sec_cnt;
2551         data[3] = (u8)(start_sec >> 24);
2552         data[4] = (u8)(start_sec >> 16);
2553         data[5] = (u8)(start_sec >> 8);
2554         data[6] = (u8)start_sec;
2555         data[7] = 0;
2556 
2557         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
2558                 retval = ms_write_bytes(chip, PRO_EX_SET_CMD, 7,
2559                                         WAIT_INT, data, 8);
2560                 if (retval == STATUS_SUCCESS)
2561                         break;
2562         }
2563         if (i == MS_MAX_RETRY_COUNT)
2564                 return STATUS_FAIL;
2565 
2566         return STATUS_SUCCESS;
2567 }
2568 
2569 void mspro_stop_seq_mode(struct rtsx_chip *chip)
2570 {
2571         struct ms_info *ms_card = &chip->ms_card;
2572         int retval;
2573 
2574         if (ms_card->seq_mode) {
2575                 retval = ms_switch_clock(chip);
2576                 if (retval != STATUS_SUCCESS)
2577                         return;
2578 
2579                 ms_card->seq_mode = 0;
2580                 ms_card->total_sec_cnt = 0;
2581                 ms_send_cmd(chip, PRO_STOP, WAIT_INT);
2582 
2583                 rtsx_write_register(chip, RBCTL, RB_FLUSH, RB_FLUSH);
2584         }
2585 }
2586 
2587 static inline int ms_auto_tune_clock(struct rtsx_chip *chip)
2588 {
2589         struct ms_info *ms_card = &chip->ms_card;
2590         int retval;
2591 
2592         if (chip->asic_code) {
2593                 if (ms_card->ms_clock > 30)
2594                         ms_card->ms_clock -= 20;
2595         } else {
2596                 if (ms_card->ms_clock == CLK_80)
2597                         ms_card->ms_clock = CLK_60;
2598                 else if (ms_card->ms_clock == CLK_60)
2599                         ms_card->ms_clock = CLK_40;
2600         }
2601 
2602         retval = ms_switch_clock(chip);
2603         if (retval != STATUS_SUCCESS)
2604                 return STATUS_FAIL;
2605 
2606         return STATUS_SUCCESS;
2607 }
2608 
2609 static int mspro_rw_multi_sector(struct scsi_cmnd *srb,
2610                                  struct rtsx_chip *chip, u32 start_sector,
2611                                  u16 sector_cnt)
2612 {
2613         struct ms_info *ms_card = &chip->ms_card;
2614         bool mode_2k = false;
2615         int retval;
2616         u16 count;
2617         u8 val, trans_mode, rw_tpc, rw_cmd;
2618 
2619         ms_set_err_code(chip, MS_NO_ERROR);
2620 
2621         ms_card->cleanup_counter = 0;
2622 
2623         if (CHK_MSHG(ms_card)) {
2624                 if ((start_sector % 4) || (sector_cnt % 4)) {
2625                         if (srb->sc_data_direction == DMA_FROM_DEVICE) {
2626                                 rw_tpc = PRO_READ_LONG_DATA;
2627                                 rw_cmd = PRO_READ_DATA;
2628                         } else {
2629                                 rw_tpc = PRO_WRITE_LONG_DATA;
2630                                 rw_cmd = PRO_WRITE_DATA;
2631                         }
2632                 } else {
2633                         if (srb->sc_data_direction == DMA_FROM_DEVICE) {
2634                                 rw_tpc = PRO_READ_QUAD_DATA;
2635                                 rw_cmd = PRO_READ_2K_DATA;
2636                         } else {
2637                                 rw_tpc = PRO_WRITE_QUAD_DATA;
2638                                 rw_cmd = PRO_WRITE_2K_DATA;
2639                         }
2640                         mode_2k = true;
2641                 }
2642         } else {
2643                 if (srb->sc_data_direction == DMA_FROM_DEVICE) {
2644                         rw_tpc = PRO_READ_LONG_DATA;
2645                         rw_cmd = PRO_READ_DATA;
2646                 } else {
2647                         rw_tpc = PRO_WRITE_LONG_DATA;
2648                         rw_cmd = PRO_WRITE_DATA;
2649                 }
2650         }
2651 
2652         retval = ms_switch_clock(chip);
2653         if (retval != STATUS_SUCCESS)
2654                 return STATUS_FAIL;
2655 
2656         if (srb->sc_data_direction == DMA_FROM_DEVICE)
2657                 trans_mode = MS_TM_AUTO_READ;
2658         else
2659                 trans_mode = MS_TM_AUTO_WRITE;
2660 
2661         retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
2662         if (retval)
2663                 return retval;
2664 
2665         if (ms_card->seq_mode) {
2666                 if ((ms_card->pre_dir != srb->sc_data_direction) ||
2667                     ((ms_card->pre_sec_addr + ms_card->pre_sec_cnt) !=
2668                      start_sector) ||
2669                     (mode_2k && (ms_card->seq_mode & MODE_512_SEQ)) ||
2670                     (!mode_2k && (ms_card->seq_mode & MODE_2K_SEQ)) ||
2671                     !(val & MS_INT_BREQ) ||
2672                     ((ms_card->total_sec_cnt + sector_cnt) > 0xFE00)) {
2673                         ms_card->seq_mode = 0;
2674                         ms_card->total_sec_cnt = 0;
2675                         if (val & MS_INT_BREQ) {
2676                                 retval = ms_send_cmd(chip, PRO_STOP, WAIT_INT);
2677                                 if (retval != STATUS_SUCCESS)
2678                                         return STATUS_FAIL;
2679 
2680                                 rtsx_write_register(chip, RBCTL, RB_FLUSH,
2681                                                     RB_FLUSH);
2682                         }
2683                 }
2684         }
2685 
2686         if (!ms_card->seq_mode) {
2687                 ms_card->total_sec_cnt = 0;
2688                 if (sector_cnt >= SEQ_START_CRITERIA) {
2689                         if ((ms_card->capacity - start_sector) > 0xFE00)
2690                                 count = 0xFE00;
2691                         else
2692                                 count = (u16)(ms_card->capacity - start_sector);
2693 
2694                         if (count > sector_cnt) {
2695                                 if (mode_2k)
2696                                         ms_card->seq_mode = MODE_2K_SEQ;
2697                                 else
2698                                         ms_card->seq_mode = MODE_512_SEQ;
2699                         }
2700                 } else {
2701                         count = sector_cnt;
2702                 }
2703                 retval = mspro_set_rw_cmd(chip, start_sector, count, rw_cmd);
2704                 if (retval != STATUS_SUCCESS) {
2705                         ms_card->seq_mode = 0;
2706                         return STATUS_FAIL;
2707                 }
2708         }
2709 
2710         retval = ms_transfer_data(chip, trans_mode, rw_tpc, sector_cnt,
2711                                   WAIT_INT, mode_2k, scsi_sg_count(srb),
2712                                   scsi_sglist(srb), scsi_bufflen(srb));
2713         if (retval != STATUS_SUCCESS) {
2714                 ms_card->seq_mode = 0;
2715                 rtsx_read_register(chip, MS_TRANS_CFG, &val);
2716                 rtsx_clear_ms_error(chip);
2717 
2718                 if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
2719                         chip->rw_need_retry = 0;
2720                         dev_dbg(rtsx_dev(chip), "No card exist, exit %s\n",
2721                                 __func__);
2722                         return STATUS_FAIL;
2723                 }
2724 
2725                 if (val & MS_INT_BREQ)
2726                         ms_send_cmd(chip, PRO_STOP, WAIT_INT);
2727 
2728                 if (val & (MS_CRC16_ERR | MS_RDY_TIMEOUT)) {
2729                         dev_dbg(rtsx_dev(chip), "MSPro CRC error, tune clock!\n");
2730                         chip->rw_need_retry = 1;
2731                         ms_auto_tune_clock(chip);
2732                 }
2733 
2734                 return retval;
2735         }
2736 
2737         if (ms_card->seq_mode) {
2738                 ms_card->pre_sec_addr = start_sector;
2739                 ms_card->pre_sec_cnt = sector_cnt;
2740                 ms_card->pre_dir = srb->sc_data_direction;
2741                 ms_card->total_sec_cnt += sector_cnt;
2742         }
2743 
2744         return STATUS_SUCCESS;
2745 }
2746 
2747 static int mspro_read_format_progress(struct rtsx_chip *chip,
2748                                       const int short_data_len)
2749 {
2750         struct ms_info *ms_card = &chip->ms_card;
2751         int retval, i;
2752         u32 total_progress, cur_progress;
2753         u8 cnt, tmp;
2754         u8 data[8];
2755 
2756         dev_dbg(rtsx_dev(chip), "%s, short_data_len = %d\n", __func__,
2757                 short_data_len);
2758 
2759         retval = ms_switch_clock(chip);
2760         if (retval != STATUS_SUCCESS) {
2761                 ms_card->format_status = FORMAT_FAIL;
2762                 return STATUS_FAIL;
2763         }
2764 
2765         retval = rtsx_read_register(chip, MS_TRANS_CFG, &tmp);
2766         if (retval != STATUS_SUCCESS) {
2767                 ms_card->format_status = FORMAT_FAIL;
2768                 return STATUS_FAIL;
2769         }
2770 
2771         if (!(tmp & MS_INT_BREQ)) {
2772                 if ((tmp & (MS_INT_CED | MS_INT_BREQ | MS_INT_CMDNK |
2773                             MS_INT_ERR)) == MS_INT_CED) {
2774                         ms_card->format_status = FORMAT_SUCCESS;
2775                         return STATUS_SUCCESS;
2776                 }
2777                 ms_card->format_status = FORMAT_FAIL;
2778                 return STATUS_FAIL;
2779         }
2780 
2781         if (short_data_len >= 256)
2782                 cnt = 0;
2783         else
2784                 cnt = (u8)short_data_len;
2785 
2786         retval = rtsx_write_register(chip, MS_CFG, MS_NO_CHECK_INT,
2787                                      MS_NO_CHECK_INT);
2788         if (retval != STATUS_SUCCESS) {
2789                 ms_card->format_status = FORMAT_FAIL;
2790                 return STATUS_FAIL;
2791         }
2792 
2793         retval = ms_read_bytes(chip, PRO_READ_SHORT_DATA, cnt, WAIT_INT,
2794                                data, 8);
2795         if (retval != STATUS_SUCCESS) {
2796                 ms_card->format_status = FORMAT_FAIL;
2797                 return STATUS_FAIL;
2798         }
2799 
2800         total_progress = (data[0] << 24) | (data[1] << 16) |
2801                 (data[2] << 8) | data[3];
2802         cur_progress = (data[4] << 24) | (data[5] << 16) |
2803                 (data[6] << 8) | data[7];
2804 
2805         dev_dbg(rtsx_dev(chip), "total_progress = %d, cur_progress = %d\n",
2806                 total_progress, cur_progress);
2807 
2808         if (total_progress == 0) {
2809                 ms_card->progress = 0;
2810         } else {
2811                 u64 ulltmp = (u64)cur_progress * (u64)65535;
2812 
2813                 do_div(ulltmp, total_progress);
2814                 ms_card->progress = (u16)ulltmp;
2815         }
2816         dev_dbg(rtsx_dev(chip), "progress = %d\n", ms_card->progress);
2817 
2818         for (i = 0; i < 5000; i++) {
2819                 retval = rtsx_read_register(chip, MS_TRANS_CFG, &tmp);
2820                 if (retval != STATUS_SUCCESS) {
2821                         ms_card->format_status = FORMAT_FAIL;
2822                         return STATUS_FAIL;
2823                 }
2824                 if (tmp & (MS_INT_CED | MS_INT_CMDNK |
2825                                 MS_INT_BREQ | MS_INT_ERR))
2826                         break;
2827 
2828                 wait_timeout(1);
2829         }
2830 
2831         retval = rtsx_write_register(chip, MS_CFG, MS_NO_CHECK_INT, 0);
2832         if (retval != STATUS_SUCCESS) {
2833                 ms_card->format_status = FORMAT_FAIL;
2834                 return STATUS_FAIL;
2835         }
2836 
2837         if (i == 5000) {
2838                 ms_card->format_status = FORMAT_FAIL;
2839                 return STATUS_FAIL;
2840         }
2841 
2842         if (tmp & (MS_INT_CMDNK | MS_INT_ERR)) {
2843                 ms_card->format_status = FORMAT_FAIL;
2844                 return STATUS_FAIL;
2845         }
2846 
2847         if (tmp & MS_INT_CED) {
2848                 ms_card->format_status = FORMAT_SUCCESS;
2849                 ms_card->pro_under_formatting = 0;
2850         } else if (tmp & MS_INT_BREQ) {
2851                 ms_card->format_status = FORMAT_IN_PROGRESS;
2852         } else {
2853                 ms_card->format_status = FORMAT_FAIL;
2854                 ms_card->pro_under_formatting = 0;
2855                 return STATUS_FAIL;
2856         }
2857 
2858         return STATUS_SUCCESS;
2859 }
2860 
2861 void mspro_polling_format_status(struct rtsx_chip *chip)
2862 {
2863         struct ms_info *ms_card = &chip->ms_card;
2864         int i;
2865 
2866         if (ms_card->pro_under_formatting &&
2867             (rtsx_get_stat(chip) != RTSX_STAT_SS)) {
2868                 rtsx_set_stat(chip, RTSX_STAT_RUN);
2869 
2870                 for (i = 0; i < 65535; i++) {
2871                         mspro_read_format_progress(chip, MS_SHORT_DATA_LEN);
2872                         if (ms_card->format_status != FORMAT_IN_PROGRESS)
2873                                 break;
2874                 }
2875         }
2876 }
2877 
2878 int mspro_format(struct scsi_cmnd *srb, struct rtsx_chip *chip,
2879                  int short_data_len, bool quick_format)
2880 {
2881         struct ms_info *ms_card = &chip->ms_card;
2882         int retval, i;
2883         u8 buf[8], tmp;
2884         u16 para;
2885 
2886         retval = ms_switch_clock(chip);
2887         if (retval != STATUS_SUCCESS)
2888                 return STATUS_FAIL;
2889 
2890         retval = ms_set_rw_reg_addr(chip, 0x00, 0x00, Pro_TPCParm, 0x01);
2891         if (retval != STATUS_SUCCESS)
2892                 return STATUS_FAIL;
2893 
2894         memset(buf, 0, 2);
2895         switch (short_data_len) {
2896         case 32:
2897                 buf[0] = 0;
2898                 break;
2899         case 64:
2900                 buf[0] = 1;
2901                 break;
2902         case 128:
2903                 buf[0] = 2;
2904                 break;
2905         case 256:
2906         default:
2907                 buf[0] = 3;
2908                 break;
2909         }
2910 
2911         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
2912                 retval = ms_write_bytes(chip, PRO_WRITE_REG, 1,
2913                                         NO_WAIT_INT, buf, 2);
2914                 if (retval == STATUS_SUCCESS)
2915                         break;
2916         }
2917         if (i == MS_MAX_RETRY_COUNT)
2918                 return STATUS_FAIL;
2919 
2920         if (quick_format)
2921                 para = 0x0000;
2922         else
2923                 para = 0x0001;
2924 
2925         retval = mspro_set_rw_cmd(chip, 0, para, PRO_FORMAT);
2926         if (retval != STATUS_SUCCESS)
2927                 return STATUS_FAIL;
2928 
2929         retval = rtsx_read_register(chip, MS_TRANS_CFG, &tmp);
2930         if (retval)
2931                 return retval;
2932 
2933         if (tmp & (MS_INT_CMDNK | MS_INT_ERR))
2934                 return STATUS_FAIL;
2935 
2936         if ((tmp & (MS_INT_BREQ | MS_INT_CED)) == MS_INT_BREQ) {
2937                 ms_card->pro_under_formatting = 1;
2938                 ms_card->progress = 0;
2939                 ms_card->format_status = FORMAT_IN_PROGRESS;
2940                 return STATUS_SUCCESS;
2941         }
2942 
2943         if (tmp & MS_INT_CED) {
2944                 ms_card->pro_under_formatting = 0;
2945                 ms_card->progress = 0;
2946                 ms_card->format_status = FORMAT_SUCCESS;
2947                 set_sense_type(chip, SCSI_LUN(srb), SENSE_TYPE_NO_SENSE);
2948                 return STATUS_SUCCESS;
2949         }
2950 
2951         return STATUS_FAIL;
2952 }
2953 
2954 static int ms_read_multiple_pages(struct rtsx_chip *chip, u16 phy_blk,
2955                                   u16 log_blk, u8 start_page, u8 end_page,
2956                                   u8 *buf, unsigned int *index,
2957                                   unsigned int *offset)
2958 {
2959         struct ms_info *ms_card = &chip->ms_card;
2960         int retval, i;
2961         u8 extra[MS_EXTRA_SIZE], page_addr, val, trans_cfg, data[6];
2962         u8 *ptr;
2963 
2964         retval = ms_read_extra_data(chip, phy_blk, start_page,
2965                                     extra, MS_EXTRA_SIZE);
2966         if (retval == STATUS_SUCCESS) {
2967                 if ((extra[1] & 0x30) != 0x30) {
2968                         ms_set_err_code(chip, MS_FLASH_READ_ERROR);
2969                         return STATUS_FAIL;
2970                 }
2971         }
2972 
2973         retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
2974                                     SystemParm, 6);
2975         if (retval != STATUS_SUCCESS)
2976                 return STATUS_FAIL;
2977 
2978         if (CHK_MS4BIT(ms_card))
2979                 data[0] = 0x88;
2980         else
2981                 data[0] = 0x80;
2982 
2983         data[1] = 0;
2984         data[2] = (u8)(phy_blk >> 8);
2985         data[3] = (u8)phy_blk;
2986         data[4] = 0;
2987         data[5] = start_page;
2988 
2989         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
2990                 retval = ms_write_bytes(chip, WRITE_REG, 6, NO_WAIT_INT,
2991                                         data, 6);
2992                 if (retval == STATUS_SUCCESS)
2993                         break;
2994         }
2995         if (i == MS_MAX_RETRY_COUNT)
2996                 return STATUS_FAIL;
2997 
2998         ms_set_err_code(chip, MS_NO_ERROR);
2999 
3000         retval = ms_send_cmd(chip, BLOCK_READ, WAIT_INT);
3001         if (retval != STATUS_SUCCESS)
3002                 return STATUS_FAIL;
3003 
3004         ptr = buf;
3005 
3006         for (page_addr = start_page; page_addr < end_page; page_addr++) {
3007                 ms_set_err_code(chip, MS_NO_ERROR);
3008 
3009                 if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
3010                         ms_set_err_code(chip, MS_NO_CARD);
3011                         return STATUS_FAIL;
3012                 }
3013 
3014                 retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
3015                 if (retval != STATUS_SUCCESS)
3016                         return STATUS_FAIL;
3017 
3018                 if (val & INT_REG_CMDNK) {
3019                         ms_set_err_code(chip, MS_CMD_NK);
3020                         return STATUS_FAIL;
3021                 }
3022                 if (val & INT_REG_ERR) {
3023                         if (val & INT_REG_BREQ) {
3024                                 retval = ms_read_status_reg(chip);
3025                                 if (retval != STATUS_SUCCESS) {
3026                                         if (!(chip->card_wp & MS_CARD)) {
3027                                                 reset_ms(chip);
3028                                                 ms_set_page_status
3029                                                         (log_blk, setPS_NG,
3030                                                          extra,
3031                                                          MS_EXTRA_SIZE);
3032                                                 ms_write_extra_data
3033                                                         (chip, phy_blk,
3034                                                          page_addr, extra,
3035                                                          MS_EXTRA_SIZE);
3036                                         }
3037                                         ms_set_err_code(chip,
3038                                                         MS_FLASH_READ_ERROR);
3039                                         return STATUS_FAIL;
3040                                 }
3041                         } else {
3042                                 ms_set_err_code(chip, MS_FLASH_READ_ERROR);
3043                                 return STATUS_FAIL;
3044                         }
3045                 } else {
3046                         if (!(val & INT_REG_BREQ)) {
3047                                 ms_set_err_code(chip, MS_BREQ_ERROR);
3048                                 return STATUS_FAIL;
3049                         }
3050                 }
3051 
3052                 if (page_addr == (end_page - 1)) {
3053                         if (!(val & INT_REG_CED)) {
3054                                 retval = ms_send_cmd(chip, BLOCK_END, WAIT_INT);
3055                                 if (retval != STATUS_SUCCESS)
3056                                         return STATUS_FAIL;
3057                         }
3058 
3059                         retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT,
3060                                                &val, 1);
3061                         if (retval != STATUS_SUCCESS)
3062                                 return STATUS_FAIL;
3063 
3064                         if (!(val & INT_REG_CED)) {
3065                                 ms_set_err_code(chip, MS_FLASH_READ_ERROR);
3066                                 return STATUS_FAIL;
3067                         }
3068 
3069                         trans_cfg = NO_WAIT_INT;
3070                 } else {
3071                         trans_cfg = WAIT_INT;
3072                 }
3073 
3074                 rtsx_init_cmd(chip);
3075 
3076                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC, 0xFF, READ_PAGE_DATA);
3077                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG,
3078                              0xFF, trans_cfg);
3079                 rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_DATA_SOURCE,
3080                              0x01, RING_BUFFER);
3081 
3082                 trans_dma_enable(DMA_FROM_DEVICE, chip, 512, DMA_512);
3083 
3084                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANSFER, 0xFF,
3085                              MS_TRANSFER_START |  MS_TM_NORMAL_READ);
3086                 rtsx_add_cmd(chip, CHECK_REG_CMD, MS_TRANSFER,
3087                              MS_TRANSFER_END, MS_TRANSFER_END);
3088 
3089                 rtsx_send_cmd_no_wait(chip);
3090 
3091                 retval = rtsx_transfer_data_partial(chip, MS_CARD, ptr, 512,
3092                                                     scsi_sg_count(chip->srb),
3093                                                     index, offset,
3094                                                     DMA_FROM_DEVICE,
3095                                                     chip->ms_timeout);
3096                 if (retval < 0) {
3097                         if (retval == -ETIMEDOUT) {
3098                                 ms_set_err_code(chip, MS_TO_ERROR);
3099                                 rtsx_clear_ms_error(chip);
3100                                 return STATUS_TIMEDOUT;
3101                         }
3102 
3103                         retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
3104                         if (retval != STATUS_SUCCESS) {
3105                                 ms_set_err_code(chip, MS_TO_ERROR);
3106                                 rtsx_clear_ms_error(chip);
3107                                 return STATUS_TIMEDOUT;
3108                         }
3109                         if (val & (MS_CRC16_ERR | MS_RDY_TIMEOUT)) {
3110                                 ms_set_err_code(chip, MS_CRC16_ERROR);
3111                                 rtsx_clear_ms_error(chip);
3112                                 return STATUS_FAIL;
3113                         }
3114                 }
3115 
3116                 if (scsi_sg_count(chip->srb) == 0)
3117                         ptr += 512;
3118         }
3119 
3120         return STATUS_SUCCESS;
3121 }
3122 
3123 static int ms_write_multiple_pages(struct rtsx_chip *chip, u16 old_blk,
3124                                    u16 new_blk, u16 log_blk, u8 start_page,
3125                                 u8 end_page, u8 *buf, unsigned int *index,
3126                                 unsigned int *offset)
3127 {
3128         struct ms_info *ms_card = &chip->ms_card;
3129         int retval, i;
3130         u8 page_addr, val, data[16];
3131         u8 *ptr;
3132 
3133         if (!start_page) {
3134                 retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
3135                                             SystemParm, 7);
3136                 if (retval != STATUS_SUCCESS)
3137                         return STATUS_FAIL;
3138 
3139                 if (CHK_MS4BIT(ms_card))
3140                         data[0] = 0x88;
3141                 else
3142                         data[0] = 0x80;
3143 
3144                 data[1] = 0;
3145                 data[2] = (u8)(old_blk >> 8);
3146                 data[3] = (u8)old_blk;
3147                 data[4] = 0x80;
3148                 data[5] = 0;
3149                 data[6] = 0xEF;
3150                 data[7] = 0xFF;
3151 
3152                 retval = ms_write_bytes(chip, WRITE_REG, 7, NO_WAIT_INT,
3153                                         data, 8);
3154                 if (retval != STATUS_SUCCESS)
3155                         return STATUS_FAIL;
3156 
3157                 retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);
3158                 if (retval != STATUS_SUCCESS)
3159                         return STATUS_FAIL;
3160 
3161                 ms_set_err_code(chip, MS_NO_ERROR);
3162                 retval = ms_transfer_tpc(chip, MS_TM_READ_BYTES, GET_INT, 1,
3163                                          NO_WAIT_INT);
3164                 if (retval != STATUS_SUCCESS)
3165                         return STATUS_FAIL;
3166         }
3167 
3168         retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
3169                                     SystemParm, (6 + MS_EXTRA_SIZE));
3170         if (retval != STATUS_SUCCESS)
3171                 return STATUS_FAIL;
3172 
3173         ms_set_err_code(chip, MS_NO_ERROR);
3174 
3175         if (CHK_MS4BIT(ms_card))
3176                 data[0] = 0x88;
3177         else
3178                 data[0] = 0x80;
3179 
3180         data[1] = 0;
3181         data[2] = (u8)(new_blk >> 8);
3182         data[3] = (u8)new_blk;
3183         if ((end_page - start_page) == 1)
3184                 data[4] = 0x20;
3185         else
3186                 data[4] = 0;
3187 
3188         data[5] = start_page;
3189         data[6] = 0xF8;
3190         data[7] = 0xFF;
3191         data[8] = (u8)(log_blk >> 8);
3192         data[9] = (u8)log_blk;
3193 
3194         for (i = 0x0A; i < 0x10; i++)
3195                 data[i] = 0xFF;
3196 
3197         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
3198                 retval = ms_write_bytes(chip, WRITE_REG, 6 + MS_EXTRA_SIZE,
3199                                         NO_WAIT_INT, data, 16);
3200                 if (retval == STATUS_SUCCESS)
3201                         break;
3202         }
3203         if (i == MS_MAX_RETRY_COUNT)
3204                 return STATUS_FAIL;
3205 
3206         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
3207                 retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);
3208                 if (retval == STATUS_SUCCESS)
3209                         break;
3210         }
3211         if (i == MS_MAX_RETRY_COUNT)
3212                 return STATUS_FAIL;
3213 
3214         retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
3215         if (retval != STATUS_SUCCESS)
3216                 return STATUS_FAIL;
3217 
3218         ptr = buf;
3219         for (page_addr = start_page; page_addr < end_page; page_addr++) {
3220                 ms_set_err_code(chip, MS_NO_ERROR);
3221 
3222                 if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
3223                         ms_set_err_code(chip, MS_NO_CARD);
3224                         return STATUS_FAIL;
3225                 }
3226 
3227                 if (val & INT_REG_CMDNK) {
3228                         ms_set_err_code(chip, MS_CMD_NK);
3229                         return STATUS_FAIL;
3230                 }
3231                 if (val & INT_REG_ERR) {
3232                         ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);
3233                         return STATUS_FAIL;
3234                 }
3235                 if (!(val & INT_REG_BREQ)) {
3236                         ms_set_err_code(chip, MS_BREQ_ERROR);
3237                         return STATUS_FAIL;
3238                 }
3239 
3240                 udelay(30);
3241 
3242                 rtsx_init_cmd(chip);
3243 
3244                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC,
3245                              0xFF, WRITE_PAGE_DATA);
3246                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG,
3247                              0xFF, WAIT_INT);
3248                 rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_DATA_SOURCE,
3249                              0x01, RING_BUFFER);
3250 
3251                 trans_dma_enable(DMA_TO_DEVICE, chip, 512, DMA_512);
3252 
3253                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANSFER, 0xFF,
3254                              MS_TRANSFER_START |  MS_TM_NORMAL_WRITE);
3255                 rtsx_add_cmd(chip, CHECK_REG_CMD, MS_TRANSFER,
3256                              MS_TRANSFER_END, MS_TRANSFER_END);
3257 
3258                 rtsx_send_cmd_no_wait(chip);
3259 
3260                 retval = rtsx_transfer_data_partial(chip, MS_CARD, ptr, 512,
3261                                                     scsi_sg_count(chip->srb),
3262                                                     index, offset,
3263                                                     DMA_TO_DEVICE,
3264                                                     chip->ms_timeout);
3265                 if (retval < 0) {
3266                         ms_set_err_code(chip, MS_TO_ERROR);
3267                         rtsx_clear_ms_error(chip);
3268 
3269                         if (retval == -ETIMEDOUT)
3270                                 return STATUS_TIMEDOUT;
3271                         return STATUS_FAIL;
3272                 }
3273 
3274                 retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
3275                 if (retval != STATUS_SUCCESS)
3276                         return STATUS_FAIL;
3277 
3278                 if ((end_page - start_page) == 1) {
3279                         if (!(val & INT_REG_CED)) {
3280                                 ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);
3281                                 return STATUS_FAIL;
3282                         }
3283                 } else {
3284                         if (page_addr == (end_page - 1)) {
3285                                 if (!(val & INT_REG_CED)) {
3286                                         retval = ms_send_cmd(chip, BLOCK_END,
3287                                                              WAIT_INT);
3288                                         if (retval != STATUS_SUCCESS)
3289                                                 return STATUS_FAIL;
3290                                 }
3291 
3292                                 retval = ms_read_bytes(chip, GET_INT, 1,
3293                                                        NO_WAIT_INT, &val, 1);
3294                                 if (retval != STATUS_SUCCESS)
3295                                         return STATUS_FAIL;
3296                         }
3297 
3298                         if ((page_addr == (end_page - 1)) ||
3299                             (page_addr == ms_card->page_off)) {
3300                                 if (!(val & INT_REG_CED)) {
3301                                         ms_set_err_code(chip,
3302                                                         MS_FLASH_WRITE_ERROR);
3303                                         return STATUS_FAIL;
3304                                 }
3305                         }
3306                 }
3307 
3308                 if (scsi_sg_count(chip->srb) == 0)
3309                         ptr += 512;
3310         }
3311 
3312         return STATUS_SUCCESS;
3313 }
3314 
3315 static int ms_finish_write(struct rtsx_chip *chip, u16 old_blk, u16 new_blk,
3316                            u16 log_blk, u8 page_off)
3317 {
3318         struct ms_info *ms_card = &chip->ms_card;
3319         int retval, seg_no;
3320 
3321         retval = ms_copy_page(chip, old_blk, new_blk, log_blk,
3322                               page_off, ms_card->page_off + 1);
3323         if (retval != STATUS_SUCCESS)
3324                 return STATUS_FAIL;
3325 
3326         seg_no = old_blk >> 9;
3327 
3328         if (MS_TST_BAD_BLOCK_FLG(ms_card)) {
3329                 MS_CLR_BAD_BLOCK_FLG(ms_card);
3330                 ms_set_bad_block(chip, old_blk);
3331         } else {
3332                 retval = ms_erase_block(chip, old_blk);
3333                 if (retval == STATUS_SUCCESS)
3334                         ms_set_unused_block(chip, old_blk);
3335         }
3336 
3337         ms_set_l2p_tbl(chip, seg_no, log_blk - ms_start_idx[seg_no], new_blk);
3338 
3339         return STATUS_SUCCESS;
3340 }
3341 
3342 static int ms_prepare_write(struct rtsx_chip *chip, u16 old_blk, u16 new_blk,
3343                             u16 log_blk, u8 start_page)
3344 {
3345         int retval;
3346 
3347         if (start_page) {
3348                 retval = ms_copy_page(chip, old_blk, new_blk, log_blk,
3349                                       0, start_page);
3350                 if (retval != STATUS_SUCCESS)
3351                         return STATUS_FAIL;
3352         }
3353 
3354         return STATUS_SUCCESS;
3355 }
3356 
3357 #ifdef MS_DELAY_WRITE
3358 int ms_delay_write(struct rtsx_chip *chip)
3359 {
3360         struct ms_info *ms_card = &chip->ms_card;
3361         struct ms_delay_write_tag *delay_write = &ms_card->delay_write;
3362         int retval;
3363 
3364         if (delay_write->delay_write_flag) {
3365                 retval = ms_set_init_para(chip);
3366                 if (retval != STATUS_SUCCESS)
3367                         return STATUS_FAIL;
3368 
3369                 delay_write->delay_write_flag = 0;
3370                 retval = ms_finish_write(chip,
3371                                          delay_write->old_phyblock,
3372                                         delay_write->new_phyblock,
3373                                         delay_write->logblock,
3374                                         delay_write->pageoff);
3375                 if (retval != STATUS_SUCCESS)
3376                         return STATUS_FAIL;
3377         }
3378 
3379         return STATUS_SUCCESS;
3380 }
3381 #endif
3382 
3383 static inline void ms_rw_fail(struct scsi_cmnd *srb, struct rtsx_chip *chip)
3384 {
3385         if (srb->sc_data_direction == DMA_FROM_DEVICE)
3386                 set_sense_type(chip, SCSI_LUN(srb),
3387                                SENSE_TYPE_MEDIA_UNRECOVER_READ_ERR);
3388         else
3389                 set_sense_type(chip, SCSI_LUN(srb), SENSE_TYPE_MEDIA_WRITE_ERR);
3390 }
3391 
3392 static int ms_rw_multi_sector(struct scsi_cmnd *srb, struct rtsx_chip *chip,
3393                               u32 start_sector, u16 sector_cnt)
3394 {
3395         struct ms_info *ms_card = &chip->ms_card;
3396         unsigned int lun = SCSI_LUN(srb);
3397         int retval, seg_no;
3398         unsigned int index = 0, offset = 0;
3399         u16 old_blk = 0, new_blk = 0, log_blk, total_sec_cnt = sector_cnt;
3400         u8 start_page, end_page = 0, page_cnt;
3401         u8 *ptr;
3402 #ifdef MS_DELAY_WRITE
3403         struct ms_delay_write_tag *delay_write = &ms_card->delay_write;
3404 #endif
3405 
3406         ms_set_err_code(chip, MS_NO_ERROR);
3407 
3408         ms_card->cleanup_counter = 0;
3409 
3410         ptr = (u8 *)scsi_sglist(srb);
3411 
3412         retval = ms_switch_clock(chip);
3413         if (retval != STATUS_SUCCESS) {
3414                 ms_rw_fail(srb, chip);
3415                 return STATUS_FAIL;
3416         }
3417 
3418         log_blk = (u16)(start_sector >> ms_card->block_shift);
3419         start_page = (u8)(start_sector & ms_card->page_off);
3420 
3421         for (seg_no = 0; seg_no < ARRAY_SIZE(ms_start_idx) - 1; seg_no++) {
3422                 if (log_blk < ms_start_idx[seg_no + 1])
3423                         break;
3424         }
3425 
3426         if (ms_card->segment[seg_no].build_flag == 0) {
3427                 retval = ms_build_l2p_tbl(chip, seg_no);
3428                 if (retval != STATUS_SUCCESS) {
3429                         chip->card_fail |= MS_CARD;
3430                         set_sense_type(chip, lun, SENSE_TYPE_MEDIA_NOT_PRESENT);
3431                         return STATUS_FAIL;
3432                 }
3433         }
3434 
3435         if (srb->sc_data_direction == DMA_TO_DEVICE) {
3436 #ifdef MS_DELAY_WRITE
3437                 if (delay_write->delay_write_flag &&
3438                     (delay_write->logblock == log_blk) &&
3439                     (start_page > delay_write->pageoff)) {
3440                         delay_write->delay_write_flag = 0;
3441                         retval = ms_copy_page(chip,
3442                                               delay_write->old_phyblock,
3443                                               delay_write->new_phyblock,
3444                                               log_blk,
3445                                               delay_write->pageoff, start_page);
3446                         if (retval != STATUS_SUCCESS) {
3447                                 set_sense_type(chip, lun,
3448                                                SENSE_TYPE_MEDIA_WRITE_ERR);
3449                                 return STATUS_FAIL;
3450                         }
3451                         old_blk = delay_write->old_phyblock;
3452                         new_blk = delay_write->new_phyblock;
3453                 } else if (delay_write->delay_write_flag &&
3454                                 (delay_write->logblock == log_blk) &&
3455                                 (start_page == delay_write->pageoff)) {
3456                         delay_write->delay_write_flag = 0;
3457                         old_blk = delay_write->old_phyblock;
3458                         new_blk = delay_write->new_phyblock;
3459                 } else {
3460                         retval = ms_delay_write(chip);
3461                         if (retval != STATUS_SUCCESS) {
3462                                 set_sense_type(chip, lun,
3463                                                SENSE_TYPE_MEDIA_WRITE_ERR);
3464                                 return STATUS_FAIL;
3465                         }
3466 #endif
3467                         old_blk = ms_get_l2p_tbl
3468                                         (chip, seg_no,
3469                                          log_blk - ms_start_idx[seg_no]);
3470                         new_blk  = ms_get_unused_block(chip, seg_no);
3471                         if ((old_blk == 0xFFFF) || (new_blk == 0xFFFF)) {
3472                                 set_sense_type(chip, lun,
3473                                                SENSE_TYPE_MEDIA_WRITE_ERR);
3474                                 return STATUS_FAIL;
3475                         }
3476 
3477                         retval = ms_prepare_write(chip, old_blk, new_blk,
3478                                                   log_blk, start_page);
3479                         if (retval != STATUS_SUCCESS) {
3480                                 if (detect_card_cd(chip, MS_CARD) !=
3481                                     STATUS_SUCCESS) {
3482                                         set_sense_type
3483                                                 (chip, lun,
3484                                                 SENSE_TYPE_MEDIA_NOT_PRESENT);
3485                                         return STATUS_FAIL;
3486                                 }
3487                                 set_sense_type(chip, lun,
3488                                                SENSE_TYPE_MEDIA_WRITE_ERR);
3489                                 return STATUS_FAIL;
3490                         }
3491 #ifdef MS_DELAY_WRITE
3492                 }
3493 #endif
3494         } else {
3495 #ifdef MS_DELAY_WRITE
3496                 retval = ms_delay_write(chip);
3497                 if (retval != STATUS_SUCCESS) {
3498                         if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
3499                                 set_sense_type(chip, lun,
3500                                                SENSE_TYPE_MEDIA_NOT_PRESENT);
3501                                 return STATUS_FAIL;
3502                         }
3503                         set_sense_type(chip, lun,
3504                                        SENSE_TYPE_MEDIA_UNRECOVER_READ_ERR);
3505                         return STATUS_FAIL;
3506                 }
3507 #endif
3508                 old_blk = ms_get_l2p_tbl(chip, seg_no,
3509                                          log_blk - ms_start_idx[seg_no]);
3510                 if (old_blk == 0xFFFF) {
3511                         set_sense_type(chip, lun,
3512                                        SENSE_TYPE_MEDIA_UNRECOVER_READ_ERR);
3513                         return STATUS_FAIL;
3514                 }
3515         }
3516 
3517         dev_dbg(rtsx_dev(chip), "seg_no = %d, old_blk = 0x%x, new_blk = 0x%x\n",
3518                 seg_no, old_blk, new_blk);
3519 
3520         while (total_sec_cnt) {
3521                 if ((start_page + total_sec_cnt) > (ms_card->page_off + 1))
3522                         end_page = ms_card->page_off + 1;
3523                 else
3524                         end_page = start_page + (u8)total_sec_cnt;
3525 
3526                 page_cnt = end_page - start_page;
3527 
3528                 dev_dbg(rtsx_dev(chip), "start_page = %d, end_page = %d, page_cnt = %d\n",
3529                         start_page, end_page, page_cnt);
3530 
3531                 if (srb->sc_data_direction == DMA_FROM_DEVICE) {
3532                         retval = ms_read_multiple_pages(chip,
3533                                                         old_blk, log_blk,
3534                                                         start_page, end_page,
3535                                                         ptr, &index, &offset);
3536                 } else {
3537                         retval = ms_write_multiple_pages(chip, old_blk, new_blk,
3538                                                          log_blk, start_page,
3539                                                          end_page, ptr, &index,
3540                                                          &offset);
3541                 }
3542 
3543                 if (retval != STATUS_SUCCESS) {
3544                         toggle_gpio(chip, 1);
3545                         if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
3546                                 set_sense_type(chip, lun,
3547                                                SENSE_TYPE_MEDIA_NOT_PRESENT);
3548                                 return STATUS_FAIL;
3549                         }
3550                         ms_rw_fail(srb, chip);
3551                         return STATUS_FAIL;
3552                 }
3553 
3554                 if (srb->sc_data_direction == DMA_TO_DEVICE) {
3555                         if (end_page == (ms_card->page_off + 1)) {
3556                                 retval = ms_erase_block(chip, old_blk);
3557                                 if (retval == STATUS_SUCCESS)
3558                                         ms_set_unused_block(chip, old_blk);
3559 
3560                                 ms_set_l2p_tbl(chip, seg_no,
3561                                                log_blk - ms_start_idx[seg_no],
3562                                                new_blk);
3563                         }
3564                 }
3565 
3566                 total_sec_cnt -= page_cnt;
3567                 if (scsi_sg_count(srb) == 0)
3568                         ptr += page_cnt * 512;
3569 
3570                 if (total_sec_cnt == 0)
3571                         break;
3572 
3573                 log_blk++;
3574 
3575                 for (seg_no = 0; seg_no < ARRAY_SIZE(ms_start_idx) - 1;
3576                                 seg_no++) {
3577                         if (log_blk < ms_start_idx[seg_no + 1])
3578                                 break;
3579                 }
3580 
3581                 if (ms_card->segment[seg_no].build_flag == 0) {
3582                         retval = ms_build_l2p_tbl(chip, seg_no);
3583                         if (retval != STATUS_SUCCESS) {
3584                                 chip->card_fail |= MS_CARD;
3585                                 set_sense_type(chip, lun,
3586                                                SENSE_TYPE_MEDIA_NOT_PRESENT);
3587                                 return STATUS_FAIL;
3588                         }
3589                 }
3590 
3591                 old_blk = ms_get_l2p_tbl(chip, seg_no,
3592                                          log_blk - ms_start_idx[seg_no]);
3593                 if (old_blk == 0xFFFF) {
3594                         ms_rw_fail(srb, chip);
3595                         return STATUS_FAIL;
3596                 }
3597 
3598                 if (srb->sc_data_direction == DMA_TO_DEVICE) {
3599                         new_blk = ms_get_unused_block(chip, seg_no);
3600                         if (new_blk == 0xFFFF) {
3601                                 ms_rw_fail(srb, chip);
3602                                 return STATUS_FAIL;
3603                         }
3604                 }
3605 
3606                 dev_dbg(rtsx_dev(chip), "seg_no = %d, old_blk = 0x%x, new_blk = 0x%x\n",
3607                         seg_no, old_blk, new_blk);
3608 
3609                 start_page = 0;
3610         }
3611 
3612         if (srb->sc_data_direction == DMA_TO_DEVICE) {
3613                 if (end_page < (ms_card->page_off + 1)) {
3614 #ifdef MS_DELAY_WRITE
3615                         delay_write->delay_write_flag = 1;
3616                         delay_write->old_phyblock = old_blk;
3617                         delay_write->new_phyblock = new_blk;
3618                         delay_write->logblock = log_blk;
3619                         delay_write->pageoff = end_page;
3620 #else
3621                         retval = ms_finish_write(chip, old_blk, new_blk,
3622                                                  log_blk, end_page);
3623                         if (retval != STATUS_SUCCESS) {
3624                                 if (detect_card_cd(chip, MS_CARD) !=
3625                                     STATUS_SUCCESS) {
3626                                         set_sense_type
3627                                                 (chip, lun,
3628                                                 SENSE_TYPE_MEDIA_NOT_PRESENT);
3629                                         return STATUS_FAIL;
3630                                 }
3631 
3632                                 ms_rw_fail(srb, chip);
3633                                 return STATUS_FAIL;
3634                         }
3635 #endif
3636                 }
3637         }
3638 
3639         scsi_set_resid(srb, 0);
3640 
3641         return STATUS_SUCCESS;
3642 }
3643 
3644 int ms_rw(struct scsi_cmnd *srb, struct rtsx_chip *chip,
3645           u32 start_sector, u16 sector_cnt)
3646 {
3647         struct ms_info *ms_card = &chip->ms_card;
3648         int retval;
3649 
3650         if (CHK_MSPRO(ms_card))
3651                 retval = mspro_rw_multi_sector(srb, chip, start_sector,
3652                                                sector_cnt);
3653         else
3654                 retval = ms_rw_multi_sector(srb, chip, start_sector,
3655                                             sector_cnt);
3656 
3657         return retval;
3658 }
3659 
3660 void ms_free_l2p_tbl(struct rtsx_chip *chip)
3661 {
3662         struct ms_info *ms_card = &chip->ms_card;
3663         int i = 0;
3664 
3665         if (ms_card->segment) {
3666                 for (i = 0; i < ms_card->segment_cnt; i++) {
3667                         vfree(ms_card->segment[i].l2p_table);
3668                         ms_card->segment[i].l2p_table = NULL;
3669                         vfree(ms_card->segment[i].free_table);
3670                         ms_card->segment[i].free_table = NULL;
3671                 }
3672                 vfree(ms_card->segment);
3673                 ms_card->segment = NULL;
3674         }
3675 }
3676 
3677 #ifdef SUPPORT_MAGIC_GATE
3678 
3679 #ifdef READ_BYTES_WAIT_INT
3680 static int ms_poll_int(struct rtsx_chip *chip)
3681 {
3682         int retval;
3683         u8 val;
3684 
3685         rtsx_init_cmd(chip);
3686 
3687         rtsx_add_cmd(chip, CHECK_REG_CMD, MS_TRANS_CFG, MS_INT_CED, MS_INT_CED);
3688 
3689         retval = rtsx_send_cmd(chip, MS_CARD, 5000);
3690         if (retval != STATUS_SUCCESS)
3691                 return STATUS_FAIL;
3692 
3693         val = *rtsx_get_cmd_data(chip);
3694         if (val & MS_INT_ERR)
3695                 return STATUS_FAIL;
3696 
3697         return STATUS_SUCCESS;
3698 }
3699 #endif
3700 
3701 #ifdef MS_SAMPLE_INT_ERR
3702 static int check_ms_err(struct rtsx_chip *chip)
3703 {
3704         int retval;
3705         u8 val;
3706 
3707         retval = rtsx_read_register(chip, MS_TRANSFER, &val);
3708         if (retval != STATUS_SUCCESS)
3709                 return 1;
3710         if (val & MS_TRANSFER_ERR)
3711                 return 1;
3712 
3713         retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
3714         if (retval != STATUS_SUCCESS)
3715                 return 1;
3716 
3717         if (val & (MS_INT_ERR | MS_INT_CMDNK))
3718                 return 1;
3719 
3720         return 0;
3721 }
3722 #else
3723 static int check_ms_err(struct rtsx_chip *chip)
3724 {
3725         int retval;
3726         u8 val;
3727 
3728         retval = rtsx_read_register(chip, MS_TRANSFER, &val);
3729         if (retval != STATUS_SUCCESS)
3730                 return 1;
3731         if (val & MS_TRANSFER_ERR)
3732                 return 1;
3733 
3734         return 0;
3735 }
3736 #endif
3737 
3738 static int mg_send_ex_cmd(struct rtsx_chip *chip, u8 cmd, u8 entry_num)
3739 {
3740         int retval, i;
3741         u8 data[8];
3742 
3743         data[0] = cmd;
3744         data[1] = 0;
3745         data[2] = 0;
3746         data[3] = 0;
3747         data[4] = 0;
3748         data[5] = 0;
3749         data[6] = entry_num;
3750         data[7] = 0;
3751 
3752         for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
3753                 retval = ms_write_bytes(chip, PRO_EX_SET_CMD, 7, WAIT_INT,
3754                                         data, 8);
3755                 if (retval == STATUS_SUCCESS)
3756                         break;
3757         }
3758         if (i == MS_MAX_RETRY_COUNT)
3759                 return STATUS_FAIL;
3760 
3761         if (check_ms_err(chip)) {
3762                 rtsx_clear_ms_error(chip);
3763                 return STATUS_FAIL;
3764         }
3765 
3766         return STATUS_SUCCESS;
3767 }
3768 
3769 static int mg_set_tpc_para_sub(struct rtsx_chip *chip, int type,
3770                                u8 mg_entry_num)
3771 {
3772         int retval;
3773         u8 buf[6];
3774 
3775         if (type == 0)
3776                 retval = ms_set_rw_reg_addr(chip, 0, 0, Pro_TPCParm, 1);
3777         else
3778                 retval = ms_set_rw_reg_addr(chip, 0, 0, Pro_DataCount1, 6);
3779 
3780         if (retval != STATUS_SUCCESS)
3781                 return STATUS_FAIL;
3782 
3783         buf[0] = 0;
3784         buf[1] = 0;
3785         if (type == 1) {
3786                 buf[2] = 0;
3787                 buf[3] = 0;
3788                 buf[4] = 0;
3789                 buf[5] = mg_entry_num;
3790         }
3791         retval = ms_write_bytes(chip, PRO_WRITE_REG, (type == 0) ? 1 : 6,
3792                                 NO_WAIT_INT, buf, 6);
3793         if (retval != STATUS_SUCCESS)
3794                 return STATUS_FAIL;
3795 
3796         return STATUS_SUCCESS;
3797 }
3798 
3799 int mg_set_leaf_id(struct scsi_cmnd *srb, struct rtsx_chip *chip)
3800 {
3801         int retval;
3802         int i;
3803         unsigned int lun = SCSI_LUN(srb);
3804         u8 buf1[32], buf2[12];
3805 
3806         if (scsi_bufflen(srb) < 12) {
3807                 set_sense_type(chip, lun, SENSE_TYPE_MEDIA_INVALID_CMD_FIELD);
3808                 return STATUS_FAIL;
3809         }
3810 
3811         ms_cleanup_work(chip);
3812 
3813         retval = ms_switch_clock(chip);
3814         if (retval != STATUS_SUCCESS)
3815                 return STATUS_FAIL;
3816 
3817         retval = mg_send_ex_cmd(chip, MG_SET_LID, 0);
3818         if (retval != STATUS_SUCCESS) {
3819                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_ESTAB);
3820                 return STATUS_FAIL;
3821         }
3822 
3823         memset(buf1, 0, 32);
3824         rtsx_stor_get_xfer_buf(buf2, min_t(int, 12, scsi_bufflen(srb)), srb);
3825         for (i = 0; i < 8; i++)
3826                 buf1[8 + i] = buf2[4 + i];
3827 
3828         retval = ms_write_bytes(chip, PRO_WRITE_SHORT_DATA, 32, WAIT_INT,
3829                                 buf1, 32);
3830         if (retval != STATUS_SUCCESS) {
3831                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_ESTAB);
3832                 return STATUS_FAIL;
3833         }
3834         if (check_ms_err(chip)) {
3835                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_ESTAB);
3836                 rtsx_clear_ms_error(chip);
3837                 return STATUS_FAIL;
3838         }
3839 
3840         return STATUS_SUCCESS;
3841 }
3842 
3843 int mg_get_local_EKB(struct scsi_cmnd *srb, struct rtsx_chip *chip)
3844 {
3845         int retval;
3846         int bufflen;
3847         unsigned int lun = SCSI_LUN(srb);
3848         u8 *buf = NULL;
3849 
3850         ms_cleanup_work(chip);
3851 
3852         retval = ms_switch_clock(chip);
3853         if (retval != STATUS_SUCCESS)
3854                 return STATUS_FAIL;
3855 
3856         buf = kmalloc(1540, GFP_KERNEL);
3857         if (!buf)
3858                 return STATUS_ERROR;
3859 
3860         buf[0] = 0x04;
3861         buf[1] = 0x1A;
3862         buf[2] = 0x00;
3863         buf[3] = 0x00;
3864 
3865         retval = mg_send_ex_cmd(chip, MG_GET_LEKB, 0);
3866         if (retval != STATUS_SUCCESS) {
3867                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
3868                 goto free_buffer;
3869         }
3870 
3871         retval = ms_transfer_data(chip, MS_TM_AUTO_READ, PRO_READ_LONG_DATA,
3872                                   3, WAIT_INT, 0, 0, buf + 4, 1536);
3873         if (retval != STATUS_SUCCESS) {
3874                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
3875                 rtsx_clear_ms_error(chip);
3876                 goto free_buffer;
3877         }
3878         if (check_ms_err(chip)) {
3879                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
3880                 rtsx_clear_ms_error(chip);
3881                 retval = STATUS_FAIL;
3882                 goto free_buffer;
3883         }
3884 
3885         bufflen = min_t(int, 1052, scsi_bufflen(srb));
3886         rtsx_stor_set_xfer_buf(buf, bufflen, srb);
3887 
3888 free_buffer:
3889         kfree(buf);
3890         return retval;
3891 }
3892 
3893 int mg_chg(struct scsi_cmnd *srb, struct rtsx_chip *chip)
3894 {
3895         struct ms_info *ms_card = &chip->ms_card;
3896         int retval;
3897         int bufflen;
3898         int i;
3899         unsigned int lun = SCSI_LUN(srb);
3900         u8 buf[32];
3901 
3902         ms_cleanup_work(chip);
3903 
3904         retval = ms_switch_clock(chip);
3905         if (retval != STATUS_SUCCESS)
3906                 return STATUS_FAIL;
3907 
3908         retval = mg_send_ex_cmd(chip, MG_GET_ID, 0);
3909         if (retval != STATUS_SUCCESS) {
3910                 set_sense_type(chip, lun, SENSE_TYPE_MG_INCOMPATIBLE_MEDIUM);
3911                 return STATUS_FAIL;
3912         }
3913 
3914         retval = ms_read_bytes(chip, PRO_READ_SHORT_DATA, 32, WAIT_INT,
3915                                buf, 32);
3916         if (retval != STATUS_SUCCESS) {
3917                 set_sense_type(chip, lun, SENSE_TYPE_MG_INCOMPATIBLE_MEDIUM);
3918                 return STATUS_FAIL;
3919         }
3920         if (check_ms_err(chip)) {
3921                 set_sense_type(chip, lun, SENSE_TYPE_MG_INCOMPATIBLE_MEDIUM);
3922                 rtsx_clear_ms_error(chip);
3923                 return STATUS_FAIL;
3924         }
3925 
3926         memcpy(ms_card->magic_gate_id, buf, 16);
3927 
3928 #ifdef READ_BYTES_WAIT_INT
3929         retval = ms_poll_int(chip);
3930         if (retval != STATUS_SUCCESS) {
3931                 set_sense_type(chip, lun, SENSE_TYPE_MG_INCOMPATIBLE_MEDIUM);
3932                 return STATUS_FAIL;
3933         }
3934 #endif
3935 
3936         retval = mg_send_ex_cmd(chip, MG_SET_RD, 0);
3937         if (retval != STATUS_SUCCESS) {
3938                 set_sense_type(chip, lun, SENSE_TYPE_MG_INCOMPATIBLE_MEDIUM);
3939                 return STATUS_FAIL;
3940         }
3941 
3942         bufflen = min_t(int, 12, scsi_bufflen(srb));
3943         rtsx_stor_get_xfer_buf(buf, bufflen, srb);
3944 
3945         for (i = 0; i < 8; i++)
3946                 buf[i] = buf[4 + i];
3947 
3948         for (i = 0; i < 24; i++)
3949                 buf[8 + i] = 0;
3950 
3951         retval = ms_write_bytes(chip, PRO_WRITE_SHORT_DATA,
3952                                 32, WAIT_INT, buf, 32);
3953         if (retval != STATUS_SUCCESS) {
3954                 set_sense_type(chip, lun, SENSE_TYPE_MG_INCOMPATIBLE_MEDIUM);
3955                 return STATUS_FAIL;
3956         }
3957         if (check_ms_err(chip)) {
3958                 set_sense_type(chip, lun, SENSE_TYPE_MG_INCOMPATIBLE_MEDIUM);
3959                 rtsx_clear_ms_error(chip);
3960                 return STATUS_FAIL;
3961         }
3962 
3963         ms_card->mg_auth = 0;
3964 
3965         return STATUS_SUCCESS;
3966 }
3967 
3968 int mg_get_rsp_chg(struct scsi_cmnd *srb, struct rtsx_chip *chip)
3969 {
3970         struct ms_info *ms_card = &chip->ms_card;
3971         int retval;
3972         int bufflen;
3973         unsigned int lun = SCSI_LUN(srb);
3974         u8 buf1[32], buf2[36];
3975 
3976         ms_cleanup_work(chip);
3977 
3978         retval = ms_switch_clock(chip);
3979         if (retval != STATUS_SUCCESS)
3980                 return STATUS_FAIL;
3981 
3982         retval = mg_send_ex_cmd(chip, MG_MAKE_RMS, 0);
3983         if (retval != STATUS_SUCCESS) {
3984                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
3985                 return STATUS_FAIL;
3986         }
3987 
3988         retval = ms_read_bytes(chip, PRO_READ_SHORT_DATA, 32, WAIT_INT,
3989                                buf1, 32);
3990         if (retval != STATUS_SUCCESS) {
3991                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
3992                 return STATUS_FAIL;
3993         }
3994         if (check_ms_err(chip)) {
3995                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
3996                 rtsx_clear_ms_error(chip);
3997                 return STATUS_FAIL;
3998         }
3999 
4000         buf2[0] = 0x00;
4001         buf2[1] = 0x22;
4002         buf2[2] = 0x00;
4003         buf2[3] = 0x00;
4004 
4005         memcpy(buf2 + 4, ms_card->magic_gate_id, 16);
4006         memcpy(buf2 + 20, buf1, 16);
4007 
4008         bufflen = min_t(int, 36, scsi_bufflen(srb));
4009         rtsx_stor_set_xfer_buf(buf2, bufflen, srb);
4010 
4011 #ifdef READ_BYTES_WAIT_INT
4012         retval = ms_poll_int(chip);
4013         if (retval != STATUS_SUCCESS) {
4014                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
4015                 return STATUS_FAIL;
4016         }
4017 #endif
4018 
4019         return STATUS_SUCCESS;
4020 }
4021 
4022 int mg_rsp(struct scsi_cmnd *srb, struct rtsx_chip *chip)
4023 {
4024         struct ms_info *ms_card = &chip->ms_card;
4025         int retval;
4026         int i;
4027         int bufflen;
4028         unsigned int lun = SCSI_LUN(srb);
4029         u8 buf[32];
4030 
4031         ms_cleanup_work(chip);
4032 
4033         retval = ms_switch_clock(chip);
4034         if (retval != STATUS_SUCCESS)
4035                 return STATUS_FAIL;
4036 
4037         retval = mg_send_ex_cmd(chip, MG_MAKE_KSE, 0);
4038         if (retval != STATUS_SUCCESS) {
4039                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
4040                 return STATUS_FAIL;
4041         }
4042 
4043         bufflen = min_t(int, 12, scsi_bufflen(srb));
4044         rtsx_stor_get_xfer_buf(buf, bufflen, srb);
4045 
4046         for (i = 0; i < 8; i++)
4047                 buf[i] = buf[4 + i];
4048 
4049         for (i = 0; i < 24; i++)
4050                 buf[8 + i] = 0;
4051 
4052         retval = ms_write_bytes(chip, PRO_WRITE_SHORT_DATA, 32, WAIT_INT,
4053                                 buf, 32);
4054         if (retval != STATUS_SUCCESS) {
4055                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
4056                 return STATUS_FAIL;
4057         }
4058         if (check_ms_err(chip)) {
4059                 set_sense_type(chip, lun, SENSE_TYPE_MG_KEY_FAIL_NOT_AUTHEN);
4060                 rtsx_clear_ms_error(chip);
4061                 return STATUS_FAIL;
4062         }
4063 
4064         ms_card->mg_auth = 1;
4065 
4066         return STATUS_SUCCESS;
4067 }
4068 
4069 int mg_get_ICV(struct scsi_cmnd *srb, struct rtsx_chip *chip)
4070 {
4071         struct ms_info *ms_card = &chip->ms_card;
4072         int retval;
4073         int bufflen;
4074         unsigned int lun = SCSI_LUN(srb);
4075         u8 *buf = NULL;
4076 
4077         ms_cleanup_work(chip);
4078 
4079         retval = ms_switch_clock(chip);
4080         if (retval != STATUS_SUCCESS)
4081                 return STATUS_FAIL;
4082 
4083         buf = kmalloc(1028, GFP_KERNEL);
4084         if (!buf)
4085                 return STATUS_ERROR;
4086 
4087         buf[0] = 0x04;
4088         buf[1] = 0x02;
4089         buf[2] = 0x00;
4090         buf[3] = 0x00;
4091 
4092         retval = mg_send_ex_cmd(chip, MG_GET_IBD, ms_card->mg_entry_num);
4093         if (retval != STATUS_SUCCESS) {
4094                 set_sense_type(chip, lun, SENSE_TYPE_MEDIA_UNRECOVER_READ_ERR);
4095                 goto free_buffer;
4096         }
4097 
4098         retval = ms_transfer_data(chip, MS_TM_AUTO_READ, PRO_READ_LONG_DATA,
4099                                   2, WAIT_INT, 0, 0, buf + 4, 1024);
4100         if (retval != STATUS_SUCCESS) {
4101                 set_sense_type(chip, lun, SENSE_TYPE_MEDIA_UNRECOVER_READ_ERR);
4102                 rtsx_clear_ms_error(chip);
4103                 goto free_buffer;
4104         }
4105         if (check_ms_err(chip)) {
4106                 set_sense_type(chip, lun, SENSE_TYPE_MEDIA_UNRECOVER_READ_ERR);
4107                 rtsx_clear_ms_error(chip);
4108                 retval = STATUS_FAIL;
4109                 goto free_buffer;
4110         }
4111 
4112         bufflen = min_t(int, 1028, scsi_bufflen(srb));
4113         rtsx_stor_set_xfer_buf(buf, bufflen, srb);
4114 
4115 free_buffer:
4116         kfree(buf);
4117         return retval;
4118 }
4119 
4120 int mg_set_ICV(struct scsi_cmnd *srb, struct rtsx_chip *chip)
4121 {
4122         struct ms_info *ms_card = &chip->ms_card;
4123         int retval;
4124         int bufflen;
4125 #ifdef MG_SET_ICV_SLOW
4126         int i;
4127 #endif
4128         unsigned int lun = SCSI_LUN(srb);
4129         u8 *buf = NULL;
4130 
4131         ms_cleanup_work(chip);
4132 
4133         retval = ms_switch_clock(chip);
4134         if (retval != STATUS_SUCCESS)
4135                 return STATUS_FAIL;
4136 
4137         buf = kmalloc(1028, GFP_KERNEL);
4138         if (!buf)
4139                 return STATUS_ERROR;
4140 
4141         bufflen = min_t(int, 1028, scsi_bufflen(srb));
4142         rtsx_stor_get_xfer_buf(buf, bufflen, srb);
4143 
4144         retval = mg_send_ex_cmd(chip, MG_SET_IBD, ms_card->mg_entry_num);
4145         if (retval != STATUS_SUCCESS) {
4146                 if (ms_card->mg_auth == 0) {
4147                         if ((buf[5] & 0xC0) != 0)
4148                                 set_sense_type
4149                                         (chip, lun,
4150                                         SENSE_TYPE_MG_KEY_FAIL_NOT_ESTAB);
4151                         else
4152                                 set_sense_type(chip, lun,
4153                                                SENSE_TYPE_MG_WRITE_ERR);
4154                 } else {
4155                         set_sense_type(chip, lun, SENSE_TYPE_MG_WRITE_ERR);
4156                 }
4157                 goto SetICVFinish;
4158         }
4159 
4160 #ifdef MG_SET_ICV_SLOW
4161         for (i = 0; i < 2; i++) {
4162                 udelay(50);
4163 
4164                 rtsx_init_cmd(chip);
4165 
4166                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC,
4167                              0xFF, PRO_WRITE_LONG_DATA);
4168                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG, 0xFF, WAIT_INT);
4169                 rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_DATA_SOURCE,
4170                              0x01, RING_BUFFER);
4171 
4172                 trans_dma_enable(DMA_TO_DEVICE, chip, 512, DMA_512);
4173 
4174                 rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANSFER, 0xFF,
4175                              MS_TRANSFER_START |  MS_TM_NORMAL_WRITE);
4176                 rtsx_add_cmd(chip, CHECK_REG_CMD, MS_TRANSFER,
4177                              MS_TRANSFER_END, MS_TRANSFER_END);
4178 
4179                 rtsx_send_cmd_no_wait(chip);
4180 
4181                 retval = rtsx_transfer_data(chip, MS_CARD, buf + 4 + i * 512,
4182                                             512, 0, DMA_TO_DEVICE, 3000);
4183                 if ((retval < 0) || check_ms_err(chip)) {
4184                         rtsx_clear_ms_error(chip);
4185                         if (ms_card->mg_auth == 0) {
4186                                 if ((buf[5] & 0xC0) != 0)
4187                                         set_sense_type
4188                                             (chip, lun,
4189                                              SENSE_TYPE_MG_KEY_FAIL_NOT_ESTAB);
4190                                 else
4191                                         set_sense_type(chip, lun,
4192                                                        SENSE_TYPE_MG_WRITE_ERR);
4193                         } else {
4194                                 set_sense_type(chip, lun,
4195                                                SENSE_TYPE_MG_WRITE_ERR);
4196                         }
4197                         retval = STATUS_FAIL;
4198                         goto SetICVFinish;
4199                 }
4200         }
4201 #else
4202         retval = ms_transfer_data(chip, MS_TM_AUTO_WRITE, PRO_WRITE_LONG_DATA,
4203                                   2, WAIT_INT, 0, 0, buf + 4, 1024);
4204         if ((retval != STATUS_SUCCESS) || check_ms_err(chip)) {
4205                 rtsx_clear_ms_error(chip);
4206                 if (ms_card->mg_auth == 0) {
4207                         if ((buf[5] & 0xC0) != 0)
4208                                 set_sense_type
4209                                     (chip, lun,
4210                                      SENSE_TYPE_MG_KEY_FAIL_NOT_ESTAB);
4211                         else
4212                                 set_sense_type(chip, lun,
4213                                                SENSE_TYPE_MG_WRITE_ERR);
4214                 } else {
4215                         set_sense_type(chip, lun, SENSE_TYPE_MG_WRITE_ERR);
4216                 }
4217                 goto SetICVFinish;
4218         }
4219 #endif
4220 
4221 SetICVFinish:
4222         kfree(buf);
4223         return retval;
4224 }
4225 
4226 #endif /* SUPPORT_MAGIC_GATE */
4227 
4228 void ms_cleanup_work(struct rtsx_chip *chip)
4229 {
4230         struct ms_info *ms_card = &chip->ms_card;
4231 
4232         if (CHK_MSPRO(ms_card)) {
4233                 if (ms_card->seq_mode) {
4234                         dev_dbg(rtsx_dev(chip), "MS Pro: stop transmission\n");
4235                         mspro_stop_seq_mode(chip);
4236                         ms_card->cleanup_counter = 0;
4237                 }
4238                 if (CHK_MSHG(ms_card)) {
4239                         rtsx_write_register(chip, MS_CFG,
4240                                             MS_2K_SECTOR_MODE, 0x00);
4241                 }
4242         }
4243 #ifdef MS_DELAY_WRITE
4244         else if ((!CHK_MSPRO(ms_card)) &&
4245                  ms_card->delay_write.delay_write_flag) {
4246                 dev_dbg(rtsx_dev(chip), "MS: delay write\n");
4247                 ms_delay_write(chip);
4248                 ms_card->cleanup_counter = 0;
4249         }
4250 #endif
4251 }
4252 
4253 int ms_power_off_card3v3(struct rtsx_chip *chip)
4254 {
4255         int retval;
4256 
4257         retval = disable_card_clock(chip, MS_CARD);
4258         if (retval != STATUS_SUCCESS)
4259                 return STATUS_FAIL;
4260 
4261         if (chip->asic_code) {
4262                 retval = ms_pull_ctl_disable(chip);
4263                 if (retval != STATUS_SUCCESS)
4264                         return STATUS_FAIL;
4265         } else {
4266                 retval = rtsx_write_register(chip, FPGA_PULL_CTL,
4267                                              FPGA_MS_PULL_CTL_BIT | 0x20,
4268                                              FPGA_MS_PULL_CTL_BIT);
4269                 if (retval)
4270                         return retval;
4271         }
4272         retval = rtsx_write_register(chip, CARD_OE, MS_OUTPUT_EN, 0);
4273         if (retval)
4274                 return retval;
4275 
4276         if (!chip->ft2_fast_mode) {
4277                 retval = card_power_off(chip, MS_CARD);
4278                 if (retval != STATUS_SUCCESS)
4279                         return STATUS_FAIL;
4280         }
4281 
4282         return STATUS_SUCCESS;
4283 }
4284 
4285 int release_ms_card(struct rtsx_chip *chip)
4286 {
4287         struct ms_info *ms_card = &chip->ms_card;
4288         int retval;
4289 
4290 #ifdef MS_DELAY_WRITE
4291         ms_card->delay_write.delay_write_flag = 0;
4292 #endif
4293         ms_card->pro_under_formatting = 0;
4294 
4295         chip->card_ready &= ~MS_CARD;
4296         chip->card_fail &= ~MS_CARD;
4297         chip->card_wp &= ~MS_CARD;
4298 
4299         ms_free_l2p_tbl(chip);
4300 
4301         memset(ms_card->raw_sys_info, 0, 96);
4302 #ifdef SUPPORT_PCGL_1P18
4303         memset(ms_card->raw_model_name, 0, 48);
4304 #endif
4305 
4306         retval = ms_power_off_card3v3(chip);
4307         if (retval != STATUS_SUCCESS)
4308                 return STATUS_FAIL;
4309 
4310         return STATUS_SUCCESS;
4311 }

/* [<][>][^][v][top][bottom][index][help] */