/* Driver for Realtek PCI-Express card reader
   *
   * Copyright(c) 2009-2013 Realtek Semiconductor Corp. All rights reserved.
   *
   * This program is free software; you can redistribute it and/or modify it
   * under the terms of the GNU General Public License as published by the
   * Free Software Foundation; either version 2, or (at your option) any
   * later version.
   *
   * This program is distributed in the hope that it will be useful, but
   * WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   * General Public License for more details.
   *
   * You should have received a copy of the GNU General Public License along
   * with this program; if not, see <http://www.gnu.org/licenses/>.
   *
   * Author:
   *   Wei WANG (wei_wang@realsil.com.cn)
   *   Micky Ching (micky_ching@realsil.com.cn)
   */
  
  #include <linux/blkdev.h>
  #include <linux/kthread.h>
  #include <linux/sched.h>
  #include <linux/vmalloc.h>
  
  #include "rtsx.h"

  #include "ms.h"
  
  static inline void ms_set_err_code(struct rtsx_chip *chip, u8 err_code)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  
  	ms_card->err_code = err_code;
  }
  
  static inline int ms_check_err_code(struct rtsx_chip *chip, u8 err_code)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  
  	return (ms_card->err_code == err_code);
  }
  
  static int ms_parse_err_code(struct rtsx_chip *chip)
  {

  	rtsx_trace(chip);
  	return STATUS_FAIL;

  }
  
  static int ms_transfer_tpc(struct rtsx_chip *chip, u8 trans_mode,

  			   u8 tpc, u8 cnt, u8 cfg)

  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;
  	u8 *ptr;

  	dev_dbg(rtsx_dev(chip), "%s: tpc = 0x%x
  ", __func__, tpc);

  
  	rtsx_init_cmd(chip);
  
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC, 0xFF, tpc);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_BYTE_CNT, 0xFF, cnt);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG, 0xFF, cfg);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_DATA_SOURCE,

  		     0x01, PINGPONG_BUFFER);

  
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANSFER,

  		     0xFF, MS_TRANSFER_START | trans_mode);

  	rtsx_add_cmd(chip, CHECK_REG_CMD, MS_TRANSFER,

  		     MS_TRANSFER_END, MS_TRANSFER_END);

  
  	rtsx_add_cmd(chip, READ_REG_CMD, MS_TRANS_CFG, 0, 0);
  
  	retval = rtsx_send_cmd(chip, MS_CARD, 5000);
  	if (retval < 0) {
  		rtsx_clear_ms_error(chip);
  		ms_set_err_code(chip, MS_TO_ERROR);

  		rtsx_trace(chip);
  		return ms_parse_err_code(chip);

  	}
  
  	ptr = rtsx_get_cmd_data(chip) + 1;
  
  	if (!(tpc & 0x08)) {		/* Read Packet */
  		if (*ptr & MS_CRC16_ERR) {
  			ms_set_err_code(chip, MS_CRC16_ERROR);

  			rtsx_trace(chip);
  			return ms_parse_err_code(chip);

  		}
  	} else {			/* Write Packet */
  		if (CHK_MSPRO(ms_card) && !(*ptr & 0x80)) {
  			if (*ptr & (MS_INT_ERR | MS_INT_CMDNK)) {
  				ms_set_err_code(chip, MS_CMD_NK);

  				rtsx_trace(chip);
  				return ms_parse_err_code(chip);

  			}
  		}
  	}
  
  	if (*ptr & MS_RDY_TIMEOUT) {
  		rtsx_clear_ms_error(chip);
  		ms_set_err_code(chip, MS_TO_ERROR);

  		rtsx_trace(chip);
  		return ms_parse_err_code(chip);

  	}
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_transfer_data(struct rtsx_chip *chip, u8 trans_mode,

  			    u8 tpc, u16 sec_cnt, u8 cfg, bool mode_2k,
  			    int use_sg, void *buf, int buf_len)

  {
  	int retval;
  	u8 val, err_code = 0;
  	enum dma_data_direction dir;

  	if (!buf || !buf_len) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (trans_mode == MS_TM_AUTO_READ) {
  		dir = DMA_FROM_DEVICE;
  		err_code = MS_FLASH_READ_ERROR;
  	} else if (trans_mode == MS_TM_AUTO_WRITE) {
  		dir = DMA_TO_DEVICE;
  		err_code = MS_FLASH_WRITE_ERROR;
  	} else {

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	rtsx_init_cmd(chip);
  
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC, 0xFF, tpc);
  	rtsx_add_cmd(chip, WRITE_REG_CMD,
  		     MS_SECTOR_CNT_H, 0xFF, (u8)(sec_cnt >> 8));
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_SECTOR_CNT_L, 0xFF, (u8)sec_cnt);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG, 0xFF, cfg);
  
  	if (mode_2k) {
  		rtsx_add_cmd(chip, WRITE_REG_CMD,
  			     MS_CFG, MS_2K_SECTOR_MODE, MS_2K_SECTOR_MODE);
  	} else {
  		rtsx_add_cmd(chip, WRITE_REG_CMD, MS_CFG, MS_2K_SECTOR_MODE, 0);
  	}
  
  	trans_dma_enable(dir, chip, sec_cnt * 512, DMA_512);
  
  	rtsx_add_cmd(chip, WRITE_REG_CMD,
  		     MS_TRANSFER, 0xFF, MS_TRANSFER_START | trans_mode);
  	rtsx_add_cmd(chip, CHECK_REG_CMD,
  		     MS_TRANSFER, MS_TRANSFER_END, MS_TRANSFER_END);
  
  	rtsx_send_cmd_no_wait(chip);
  
  	retval = rtsx_transfer_data(chip, MS_CARD, buf, buf_len,
  				    use_sg, dir, chip->mspro_timeout);
  	if (retval < 0) {
  		ms_set_err_code(chip, err_code);
  		if (retval == -ETIMEDOUT)
  			retval = STATUS_TIMEDOUT;
  		else
  			retval = STATUS_FAIL;

  		rtsx_trace(chip);
  		return retval;

  	}

  	retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  	if (val & (MS_INT_CMDNK | MS_INT_ERR | MS_CRC16_ERR | MS_RDY_TIMEOUT)) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }
  
  static int ms_write_bytes(struct rtsx_chip *chip,
  			  u8 tpc, u8 cnt, u8 cfg, u8 *data, int data_len)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i;

  	if (!data || (data_len < cnt)) {
  		rtsx_trace(chip);
  		return STATUS_ERROR;
  	}

  
  	rtsx_init_cmd(chip);
  
  	for (i = 0; i < cnt; i++) {
  		rtsx_add_cmd(chip, WRITE_REG_CMD,
  			     PPBUF_BASE2 + i, 0xFF, data[i]);
  	}
  	if (cnt % 2)
  		rtsx_add_cmd(chip, WRITE_REG_CMD, PPBUF_BASE2 + i, 0xFF, 0xFF);
  
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC, 0xFF, tpc);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_BYTE_CNT, 0xFF, cnt);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG, 0xFF, cfg);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_DATA_SOURCE,

  		     0x01, PINGPONG_BUFFER);

  
  	rtsx_add_cmd(chip, WRITE_REG_CMD,
  		     MS_TRANSFER, 0xFF, MS_TRANSFER_START | MS_TM_WRITE_BYTES);
  	rtsx_add_cmd(chip, CHECK_REG_CMD,
  		     MS_TRANSFER, MS_TRANSFER_END, MS_TRANSFER_END);
  
  	retval = rtsx_send_cmd(chip, MS_CARD, 5000);
  	if (retval < 0) {
  		u8 val = 0;
  
  		rtsx_read_register(chip, MS_TRANS_CFG, &val);

  		dev_dbg(rtsx_dev(chip), "MS_TRANS_CFG: 0x%02x
  ", val);

  
  		rtsx_clear_ms_error(chip);
  
  		if (!(tpc & 0x08)) {
  			if (val & MS_CRC16_ERR) {
  				ms_set_err_code(chip, MS_CRC16_ERROR);

  				rtsx_trace(chip);
  				return ms_parse_err_code(chip);

  			}
  		} else {
  			if (CHK_MSPRO(ms_card) && !(val & 0x80)) {
  				if (val & (MS_INT_ERR | MS_INT_CMDNK)) {
  					ms_set_err_code(chip, MS_CMD_NK);

  					rtsx_trace(chip);
  					return ms_parse_err_code(chip);

  				}
  			}
  		}
  
  		if (val & MS_RDY_TIMEOUT) {
  			ms_set_err_code(chip, MS_TO_ERROR);

  			rtsx_trace(chip);
  			return ms_parse_err_code(chip);

  		}
  
  		ms_set_err_code(chip, MS_TO_ERROR);

  		rtsx_trace(chip);
  		return ms_parse_err_code(chip);

  	}
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_read_bytes(struct rtsx_chip *chip,

  			 u8 tpc, u8 cnt, u8 cfg, u8 *data, int data_len)

  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i;
  	u8 *ptr;

  	if (!data) {
  		rtsx_trace(chip);
  		return STATUS_ERROR;
  	}

  
  	rtsx_init_cmd(chip);
  
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TPC, 0xFF, tpc);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_BYTE_CNT, 0xFF, cnt);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANS_CFG, 0xFF, cfg);
  	rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_DATA_SOURCE,

  		     0x01, PINGPONG_BUFFER);

  
  	rtsx_add_cmd(chip, WRITE_REG_CMD, MS_TRANSFER, 0xFF,

  		     MS_TRANSFER_START | MS_TM_READ_BYTES);

  	rtsx_add_cmd(chip, CHECK_REG_CMD, MS_TRANSFER,

  		     MS_TRANSFER_END, MS_TRANSFER_END);

  
  	for (i = 0; i < data_len - 1; i++)

  		rtsx_add_cmd(chip, READ_REG_CMD, PPBUF_BASE2 + i, 0, 0);

  
  	if (data_len % 2)
  		rtsx_add_cmd(chip, READ_REG_CMD, PPBUF_BASE2 + data_len, 0, 0);
  	else
  		rtsx_add_cmd(chip, READ_REG_CMD, PPBUF_BASE2 + data_len - 1,

  			     0, 0);

  
  	retval = rtsx_send_cmd(chip, MS_CARD, 5000);
  	if (retval < 0) {
  		u8 val = 0;
  
  		rtsx_read_register(chip, MS_TRANS_CFG, &val);
  		rtsx_clear_ms_error(chip);
  
  		if (!(tpc & 0x08)) {
  			if (val & MS_CRC16_ERR) {
  				ms_set_err_code(chip, MS_CRC16_ERROR);

  				rtsx_trace(chip);
  				return ms_parse_err_code(chip);

  			}
  		} else {
  			if (CHK_MSPRO(ms_card) && !(val & 0x80)) {
  				if (val & (MS_INT_ERR | MS_INT_CMDNK)) {
  					ms_set_err_code(chip, MS_CMD_NK);

  					rtsx_trace(chip);
  					return ms_parse_err_code(chip);

  				}
  			}
  		}
  
  		if (val & MS_RDY_TIMEOUT) {
  			ms_set_err_code(chip, MS_TO_ERROR);

  			rtsx_trace(chip);
  			return ms_parse_err_code(chip);

  		}
  
  		ms_set_err_code(chip, MS_TO_ERROR);

  		rtsx_trace(chip);
  		return ms_parse_err_code(chip);

  	}
  
  	ptr = rtsx_get_cmd_data(chip) + 1;
  
  	for (i = 0; i < data_len; i++)
  		data[i] = ptr[i];
  
  	if ((tpc == PRO_READ_SHORT_DATA) && (data_len == 8)) {

  		dev_dbg(rtsx_dev(chip), "Read format progress:
  ");

  		print_hex_dump_bytes(KBUILD_MODNAME ": ", DUMP_PREFIX_NONE, ptr,
  				     cnt);

  	}
  
  	return STATUS_SUCCESS;
  }

  static int ms_set_rw_reg_addr(struct rtsx_chip *chip, u8 read_start,
  			      u8 read_cnt, u8 write_start, u8 write_cnt)

  {
  	int retval, i;
  	u8 data[4];
  
  	data[0] = read_start;
  	data[1] = read_cnt;
  	data[2] = write_start;
  	data[3] = write_cnt;
  
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_write_bytes(chip, SET_RW_REG_ADRS, 4,
  					NO_WAIT_INT, data, 4);
  		if (retval == STATUS_SUCCESS)
  			return STATUS_SUCCESS;
  		rtsx_clear_ms_error(chip);
  	}

  	rtsx_trace(chip);
  	return STATUS_FAIL;

  }
  
  static int ms_send_cmd(struct rtsx_chip *chip, u8 cmd, u8 cfg)
  {
  	u8 data[2];
  
  	data[0] = cmd;
  	data[1] = 0;
  
  	return ms_write_bytes(chip, PRO_SET_CMD, 1, cfg, data, 1);
  }
  
  static int ms_set_init_para(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;
  
  	if (CHK_HG8BIT(ms_card)) {
  		if (chip->asic_code)
  			ms_card->ms_clock = chip->asic_ms_hg_clk;
  		else
  			ms_card->ms_clock = chip->fpga_ms_hg_clk;
  
  	} else if (CHK_MSPRO(ms_card) || CHK_MS4BIT(ms_card)) {
  		if (chip->asic_code)
  			ms_card->ms_clock = chip->asic_ms_4bit_clk;
  		else
  			ms_card->ms_clock = chip->fpga_ms_4bit_clk;
  
  	} else {
  		if (chip->asic_code)
  			ms_card->ms_clock = chip->asic_ms_1bit_clk;
  		else
  			ms_card->ms_clock = chip->fpga_ms_1bit_clk;
  	}
  
  	retval = switch_clock(chip, ms_card->ms_clock);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = select_card(chip, MS_CARD);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }
  
  static int ms_switch_clock(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;
  
  	retval = select_card(chip, MS_CARD);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = switch_clock(chip, ms_card->ms_clock);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }
  
  static int ms_pull_ctl_disable(struct rtsx_chip *chip)
  {

  	int retval;

  	if (CHECK_PID(chip, 0x5208)) {

  		retval = rtsx_write_register(chip, CARD_PULL_CTL1, 0xFF,
  					     MS_D1_PD | MS_D2_PD | MS_CLK_PD | MS_D6_PD);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}
  		retval = rtsx_write_register(chip, CARD_PULL_CTL2, 0xFF,
  					     MS_D3_PD | MS_D0_PD | MS_BS_PD | XD_D4_PD);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}
  		retval = rtsx_write_register(chip, CARD_PULL_CTL3, 0xFF,
  					     MS_D7_PD | XD_CE_PD | XD_CLE_PD | XD_CD_PU);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}
  		retval = rtsx_write_register(chip, CARD_PULL_CTL4, 0xFF,
  					     XD_RDY_PD | SD_D3_PD | SD_D2_PD | XD_ALE_PD);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}
  		retval = rtsx_write_register(chip, CARD_PULL_CTL5, 0xFF,
  					     MS_INS_PU | SD_WP_PD | SD_CD_PU | SD_CMD_PD);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}
  		retval = rtsx_write_register(chip, CARD_PULL_CTL6, 0xFF,
  					     MS_D5_PD | MS_D4_PD);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}

  	} else if (CHECK_PID(chip, 0x5288)) {
  		if (CHECK_BARO_PKG(chip, QFN)) {

  			retval = rtsx_write_register(chip, CARD_PULL_CTL1,
  						     0xFF, 0x55);
  			if (retval) {
  				rtsx_trace(chip);
  				return retval;
  			}
  			retval = rtsx_write_register(chip, CARD_PULL_CTL2,
  						     0xFF, 0x55);
  			if (retval) {
  				rtsx_trace(chip);
  				return retval;
  			}
  			retval = rtsx_write_register(chip, CARD_PULL_CTL3,
  						     0xFF, 0x4B);
  			if (retval) {
  				rtsx_trace(chip);
  				return retval;
  			}
  			retval = rtsx_write_register(chip, CARD_PULL_CTL4,
  						     0xFF, 0x69);
  			if (retval) {
  				rtsx_trace(chip);
  				return retval;
  			}

  		}
  	}
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_pull_ctl_enable(struct rtsx_chip *chip)
  {
  	int retval;
  
  	rtsx_init_cmd(chip);
  
  	if (CHECK_PID(chip, 0x5208)) {
  		rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL1, 0xFF,
  			MS_D1_PD | MS_D2_PD | MS_CLK_NP | MS_D6_PD);
  		rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL2, 0xFF,
  			MS_D3_PD | MS_D0_PD | MS_BS_NP | XD_D4_PD);
  		rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL3, 0xFF,
  			MS_D7_PD | XD_CE_PD | XD_CLE_PD | XD_CD_PU);
  		rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL4, 0xFF,
  			XD_RDY_PD | SD_D3_PD | SD_D2_PD | XD_ALE_PD);
  		rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL5, 0xFF,
  			MS_INS_PU | SD_WP_PD | SD_CD_PU | SD_CMD_PD);
  		rtsx_add_cmd(chip, WRITE_REG_CMD, CARD_PULL_CTL6, 0xFF,
  			MS_D5_PD | MS_D4_PD);
  	} else if (CHECK_PID(chip, 0x5288)) {
  		if (CHECK_BARO_PKG(chip, QFN)) {
  			rtsx_add_cmd(chip, WRITE_REG_CMD,
  				     CARD_PULL_CTL1, 0xFF, 0x55);
  			rtsx_add_cmd(chip, WRITE_REG_CMD,
  				     CARD_PULL_CTL2, 0xFF, 0x45);
  			rtsx_add_cmd(chip, WRITE_REG_CMD,
  				     CARD_PULL_CTL3, 0xFF, 0x4B);
  			rtsx_add_cmd(chip, WRITE_REG_CMD,
  				     CARD_PULL_CTL4, 0xFF, 0x29);
  		}
  	}
  
  	retval = rtsx_send_cmd(chip, MS_CARD, 100);

  	if (retval < 0) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }
  
  static int ms_prepare_reset(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;
  	u8 oc_mask = 0;
  
  	ms_card->ms_type = 0;
  	ms_card->check_ms_flow = 0;
  	ms_card->switch_8bit_fail = 0;
  	ms_card->delay_write.delay_write_flag = 0;
  
  	ms_card->pro_under_formatting = 0;
  
  	retval = ms_power_off_card3v3(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (!chip->ft2_fast_mode)
  		wait_timeout(250);
  
  	retval = enable_card_clock(chip, MS_CARD);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (chip->asic_code) {
  		retval = ms_pull_ctl_enable(chip);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  	} else {

  		retval = rtsx_write_register(chip, FPGA_PULL_CTL,
  					     FPGA_MS_PULL_CTL_BIT | 0x20, 0);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}

  	}
  
  	if (!chip->ft2_fast_mode) {
  		retval = card_power_on(chip, MS_CARD);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		wait_timeout(150);
  
  #ifdef SUPPORT_OCP
  		if (CHECK_LUN_MODE(chip, SD_MS_2LUN))
  			oc_mask = MS_OC_NOW | MS_OC_EVER;
  		else
  			oc_mask = SD_OC_NOW | SD_OC_EVER;
  
  		if (chip->ocp_stat & oc_mask) {

  			dev_dbg(rtsx_dev(chip), "Over current, OCPSTAT is 0x%x
  ",
  				chip->ocp_stat);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  #endif
  	}

  	retval = rtsx_write_register(chip, CARD_OE, MS_OUTPUT_EN,
  				     MS_OUTPUT_EN);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  
  	if (chip->asic_code) {

  		retval = rtsx_write_register(chip, MS_CFG, 0xFF,
  					     SAMPLE_TIME_RISING | PUSH_TIME_DEFAULT | NO_EXTEND_TOGGLE | MS_BUS_WIDTH_1);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}

  	} else {

  		retval = rtsx_write_register(chip, MS_CFG, 0xFF,
  					     SAMPLE_TIME_FALLING | PUSH_TIME_DEFAULT | NO_EXTEND_TOGGLE | MS_BUS_WIDTH_1);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}
  	}
  	retval = rtsx_write_register(chip, MS_TRANS_CFG, 0xFF,
  				     NO_WAIT_INT | NO_AUTO_READ_INT_REG);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}
  	retval = rtsx_write_register(chip, CARD_STOP, MS_STOP | MS_CLR_ERR,
  				     MS_STOP | MS_CLR_ERR);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;

  	}

  
  	retval = ms_set_init_para(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }
  
  static int ms_identify_media_type(struct rtsx_chip *chip, int switch_8bit_bus)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i;
  	u8 val;
  
  	retval = ms_set_rw_reg_addr(chip, Pro_StatusReg, 6, SystemParm, 1);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_transfer_tpc(chip, MS_TM_READ_BYTES, READ_REG,
  					6, NO_WAIT_INT);
  		if (retval == STATUS_SUCCESS)
  			break;
  	}

  	if (i == MS_MAX_RETRY_COUNT) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  	retval = rtsx_read_register(chip, PPBUF_BASE2 + 2, &val);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  	dev_dbg(rtsx_dev(chip), "Type register: 0x%x
  ", val);

  	if (val != 0x01) {
  		if (val != 0x02)
  			ms_card->check_ms_flow = 1;

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}

  	retval = rtsx_read_register(chip, PPBUF_BASE2 + 4, &val);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  	dev_dbg(rtsx_dev(chip), "Category register: 0x%x
  ", val);

  	if (val != 0) {
  		ms_card->check_ms_flow = 1;

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}

  	retval = rtsx_read_register(chip, PPBUF_BASE2 + 5, &val);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  	dev_dbg(rtsx_dev(chip), "Class register: 0x%x
  ", val);

  	if (val == 0) {

  		retval = rtsx_read_register(chip, PPBUF_BASE2, &val);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}

  		if (val & WRT_PRTCT)
  			chip->card_wp |= MS_CARD;
  		else
  			chip->card_wp &= ~MS_CARD;
  
  	} else if ((val == 0x01) || (val == 0x02) || (val == 0x03)) {
  		chip->card_wp |= MS_CARD;
  	} else {
  		ms_card->check_ms_flow = 1;

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	ms_card->ms_type |= TYPE_MSPRO;

  	retval = rtsx_read_register(chip, PPBUF_BASE2 + 3, &val);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  	dev_dbg(rtsx_dev(chip), "IF Mode register: 0x%x
  ", val);

  	if (val == 0) {
  		ms_card->ms_type &= 0x0F;
  	} else if (val == 7) {
  		if (switch_8bit_bus)
  			ms_card->ms_type |= MS_HG;
  		else
  			ms_card->ms_type &= 0x0F;
  
  	} else {

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_confirm_cpu_startup(struct rtsx_chip *chip)
  {
  	int retval, i, k;
  	u8 val;
  
  	/* Confirm CPU StartUp */
  	k = 0;
  	do {
  		if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
  			ms_set_err_code(chip, MS_NO_CARD);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  
  		for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  			retval = ms_read_bytes(chip, GET_INT, 1,
  					NO_WAIT_INT, &val, 1);
  			if (retval == STATUS_SUCCESS)
  				break;
  		}

  		if (i == MS_MAX_RETRY_COUNT) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  		if (k > 100) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		k++;
  		wait_timeout(100);
  	} while (!(val & INT_REG_CED));
  
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);
  		if (retval == STATUS_SUCCESS)
  			break;
  	}

  	if (i == MS_MAX_RETRY_COUNT) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (val & INT_REG_ERR) {
  		if (val & INT_REG_CMDNK)
  			chip->card_wp |= (MS_CARD);

  		else {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  	}
  	/* --  end confirm CPU startup */
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_switch_parallel_bus(struct rtsx_chip *chip)
  {
  	int retval, i;
  	u8 data[2];
  
  	data[0] = PARALLEL_4BIT_IF;
  	data[1] = 0;
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_write_bytes(chip, WRITE_REG, 1, NO_WAIT_INT,
  					data, 2);
  		if (retval == STATUS_SUCCESS)
  			break;
  	}

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }
  
  static int ms_switch_8bit_bus(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i;
  	u8 data[2];
  
  	data[0] = PARALLEL_8BIT_IF;
  	data[1] = 0;
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_write_bytes(chip, WRITE_REG, 1,
  					NO_WAIT_INT, data, 2);
  		if (retval == STATUS_SUCCESS)
  			break;
  	}

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  	retval = rtsx_write_register(chip, MS_CFG, 0x98,
  				     MS_BUS_WIDTH_8 | SAMPLE_TIME_FALLING);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  	ms_card->ms_type |= MS_8BIT;
  	retval = ms_set_init_para(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_transfer_tpc(chip, MS_TM_READ_BYTES, GET_INT,
  					1, NO_WAIT_INT);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  	}
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_pro_reset_flow(struct rtsx_chip *chip, int switch_8bit_bus)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i;
  
  	for (i = 0; i < 3; i++) {
  		retval = ms_prepare_reset(chip);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		retval = ms_identify_media_type(chip, switch_8bit_bus);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		retval = ms_confirm_cpu_startup(chip);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		retval = ms_switch_parallel_bus(chip);
  		if (retval != STATUS_SUCCESS) {
  			if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
  				ms_set_err_code(chip, MS_NO_CARD);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  			continue;
  		} else {
  			break;
  		}
  	}

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	/* Switch MS-PRO into Parallel mode */

  	retval = rtsx_write_register(chip, MS_CFG, 0x18, MS_BUS_WIDTH_4);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}
  	retval = rtsx_write_register(chip, MS_CFG, PUSH_TIME_ODD,
  				     PUSH_TIME_ODD);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  
  	retval = ms_set_init_para(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	/* If MSPro HG Card, We shall try to switch to 8-bit bus */
  	if (CHK_MSHG(ms_card) && chip->support_ms_8bit && switch_8bit_bus) {
  		retval = ms_switch_8bit_bus(chip);
  		if (retval != STATUS_SUCCESS) {
  			ms_card->switch_8bit_fail = 1;

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  	}
  
  	return STATUS_SUCCESS;
  }
  
  #ifdef XC_POWERCLASS
  static int msxc_change_power(struct rtsx_chip *chip, u8 mode)
  {
  	int retval;
  	u8 buf[6];
  
  	ms_cleanup_work(chip);
  
  	retval = ms_set_rw_reg_addr(chip, 0, 0, Pro_DataCount1, 6);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	buf[0] = 0;
  	buf[1] = mode;
  	buf[2] = 0;
  	buf[3] = 0;
  	buf[4] = 0;
  	buf[5] = 0;

  	retval = ms_write_bytes(chip, PRO_WRITE_REG, 6, NO_WAIT_INT, buf, 6);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_send_cmd(chip, XC_CHG_POWER, WAIT_INT);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  	retval = rtsx_read_register(chip, MS_TRANS_CFG, buf);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  	if (buf[0] & (MS_INT_CMDNK | MS_INT_ERR)) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }
  #endif
  
  static int ms_read_attribute_info(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i;
  	u8 val, *buf, class_code, device_type, sub_class, data[16];
  	u16 total_blk = 0, blk_size = 0;
  #ifdef SUPPORT_MSXC
  	u32 xc_total_blk = 0, xc_blk_size = 0;
  #endif
  	u32 sys_info_addr = 0, sys_info_size;
  #ifdef SUPPORT_PCGL_1P18
  	u32 model_name_addr = 0, model_name_size;
  	int found_sys_info = 0, found_model_name = 0;
  #endif
  
  	retval = ms_set_rw_reg_addr(chip, Pro_IntReg, 2, Pro_SystemParm, 7);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (CHK_MS8BIT(ms_card))
  		data[0] = PARALLEL_8BIT_IF;
  	else
  		data[0] = PARALLEL_4BIT_IF;
  
  	data[1] = 0;
  
  	data[2] = 0x40;
  	data[3] = 0;
  	data[4] = 0;
  	data[5] = 0;
  	data[6] = 0;
  	data[7] = 0;
  
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_write_bytes(chip, PRO_WRITE_REG, 7, NO_WAIT_INT,
  					data, 8);
  		if (retval == STATUS_SUCCESS)
  			break;
  	}

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	buf = kmalloc(64 * 512, GFP_KERNEL);

  	if (!buf) {

  		rtsx_trace(chip);
  		return STATUS_ERROR;
  	}

  
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_send_cmd(chip, PRO_READ_ATRB, WAIT_INT);
  		if (retval != STATUS_SUCCESS)
  			continue;
  
  		retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
  		if (retval != STATUS_SUCCESS) {
  			kfree(buf);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  		if (!(val & MS_INT_BREQ)) {
  			kfree(buf);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  		retval = ms_transfer_data(chip, MS_TM_AUTO_READ,
  					PRO_READ_LONG_DATA, 0x40, WAIT_INT,
  					0, 0, buf, 64 * 512);
  		if (retval == STATUS_SUCCESS)
  			break;

  
  		rtsx_clear_ms_error(chip);

  	}
  	if (retval != STATUS_SUCCESS) {
  		kfree(buf);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	i = 0;
  	do {
  		retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
  		if (retval != STATUS_SUCCESS) {
  			kfree(buf);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  
  		if ((val & MS_INT_CED) || !(val & MS_INT_BREQ))
  			break;
  
  		retval = ms_transfer_tpc(chip, MS_TM_NORMAL_READ,
  					PRO_READ_LONG_DATA, 0, WAIT_INT);
  		if (retval != STATUS_SUCCESS) {
  			kfree(buf);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  
  		i++;
  	} while (i < 1024);
  
  	if (retval != STATUS_SUCCESS) {
  		kfree(buf);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	if ((buf[0] != 0xa5) && (buf[1] != 0xc3)) {
  		/* Signature code is wrong */
  		kfree(buf);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	if ((buf[4] < 1) || (buf[4] > 12)) {
  		kfree(buf);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	for (i = 0; i < buf[4]; i++) {
  		int cur_addr_off = 16 + i * 12;
  
  #ifdef SUPPORT_MSXC
  		if ((buf[cur_addr_off + 8] == 0x10) ||

  			(buf[cur_addr_off + 8] == 0x13)) {

  #else

  		if (buf[cur_addr_off + 8] == 0x10) {

  #endif

  			sys_info_addr = ((u32)buf[cur_addr_off + 0] << 24) |
  				((u32)buf[cur_addr_off + 1] << 16) |
  				((u32)buf[cur_addr_off + 2] << 8) |
  				buf[cur_addr_off + 3];
  			sys_info_size = ((u32)buf[cur_addr_off + 4] << 24) |
  				((u32)buf[cur_addr_off + 5] << 16) |
  				((u32)buf[cur_addr_off + 6] << 8) |
  				buf[cur_addr_off + 7];

  			dev_dbg(rtsx_dev(chip), "sys_info_addr = 0x%x, sys_info_size = 0x%x
  ",

  				sys_info_addr, sys_info_size);
  			if (sys_info_size != 96)  {
  				kfree(buf);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  			if (sys_info_addr < 0x1A0) {
  				kfree(buf);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  			if ((sys_info_size + sys_info_addr) > 0x8000) {
  				kfree(buf);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  
  #ifdef SUPPORT_MSXC
  			if (buf[cur_addr_off + 8] == 0x13)
  				ms_card->ms_type |= MS_XC;
  #endif
  #ifdef SUPPORT_PCGL_1P18
  			found_sys_info = 1;
  #else
  			break;
  #endif
  		}
  #ifdef SUPPORT_PCGL_1P18
  		if (buf[cur_addr_off + 8] == 0x15) {
  			model_name_addr = ((u32)buf[cur_addr_off + 0] << 24) |
  				((u32)buf[cur_addr_off + 1] << 16) |
  				((u32)buf[cur_addr_off + 2] << 8) |
  				buf[cur_addr_off + 3];
  			model_name_size = ((u32)buf[cur_addr_off + 4] << 24) |
  				((u32)buf[cur_addr_off + 5] << 16) |
  				((u32)buf[cur_addr_off + 6] << 8) |
  				buf[cur_addr_off + 7];

  			dev_dbg(rtsx_dev(chip), "model_name_addr = 0x%x, model_name_size = 0x%x
  ",

  				model_name_addr, model_name_size);
  			if (model_name_size != 48)  {
  				kfree(buf);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  			if (model_name_addr < 0x1A0) {
  				kfree(buf);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  			if ((model_name_size + model_name_addr) > 0x8000) {
  				kfree(buf);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  
  			found_model_name = 1;
  		}
  
  		if (found_sys_info && found_model_name)
  			break;
  #endif
  	}
  
  	if (i == buf[4]) {
  		kfree(buf);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	class_code =  buf[sys_info_addr + 0];
  	device_type = buf[sys_info_addr + 56];
  	sub_class = buf[sys_info_addr + 46];
  #ifdef SUPPORT_MSXC
  	if (CHK_MSXC(ms_card)) {
  		xc_total_blk = ((u32)buf[sys_info_addr + 6] << 24) |
  				((u32)buf[sys_info_addr + 7] << 16) |
  				((u32)buf[sys_info_addr + 8] << 8) |
  				buf[sys_info_addr + 9];
  		xc_blk_size = ((u32)buf[sys_info_addr + 32] << 24) |
  				((u32)buf[sys_info_addr + 33] << 16) |
  				((u32)buf[sys_info_addr + 34] << 8) |
  				buf[sys_info_addr + 35];

  		dev_dbg(rtsx_dev(chip), "xc_total_blk = 0x%x, xc_blk_size = 0x%x
  ",

  			xc_total_blk, xc_blk_size);
  	} else {
  		total_blk = ((u16)buf[sys_info_addr + 6] << 8) |
  			buf[sys_info_addr + 7];
  		blk_size = ((u16)buf[sys_info_addr + 2] << 8) |
  			buf[sys_info_addr + 3];

  		dev_dbg(rtsx_dev(chip), "total_blk = 0x%x, blk_size = 0x%x
  ",

  			total_blk, blk_size);
  	}
  #else
  	total_blk = ((u16)buf[sys_info_addr + 6] << 8) | buf[sys_info_addr + 7];
  	blk_size = ((u16)buf[sys_info_addr + 2] << 8) | buf[sys_info_addr + 3];

  	dev_dbg(rtsx_dev(chip), "total_blk = 0x%x, blk_size = 0x%x
  ",
  		total_blk, blk_size);

  #endif

  	dev_dbg(rtsx_dev(chip), "class_code = 0x%x, device_type = 0x%x, sub_class = 0x%x
  ",
  		class_code, device_type, sub_class);

  
  	memcpy(ms_card->raw_sys_info, buf + sys_info_addr, 96);
  #ifdef SUPPORT_PCGL_1P18
  	memcpy(ms_card->raw_model_name, buf + model_name_addr, 48);
  #endif
  
  	kfree(buf);
  
  #ifdef SUPPORT_MSXC
  	if (CHK_MSXC(ms_card)) {

  		if (class_code != 0x03) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  	} else {

  		if (class_code != 0x02) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  	}
  #else

  	if (class_code != 0x02) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  #endif
  
  	if (device_type != 0x00) {
  		if ((device_type == 0x01) || (device_type == 0x02) ||
  				(device_type == 0x03)) {
  			chip->card_wp |= MS_CARD;
  		} else {

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  	}

  	if (sub_class & 0xC0) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  	dev_dbg(rtsx_dev(chip), "class_code: 0x%x, device_type: 0x%x, sub_class: 0x%x
  ",

  		class_code, device_type, sub_class);
  
  #ifdef SUPPORT_MSXC
  	if (CHK_MSXC(ms_card)) {
  		chip->capacity[chip->card2lun[MS_CARD]] =
  			ms_card->capacity = xc_total_blk * xc_blk_size;
  	} else {
  		chip->capacity[chip->card2lun[MS_CARD]] =
  			ms_card->capacity = total_blk * blk_size;
  	}
  #else
  	ms_card->capacity = total_blk * blk_size;
  	chip->capacity[chip->card2lun[MS_CARD]] = ms_card->capacity;
  #endif
  
  	return STATUS_SUCCESS;
  }
  
  #ifdef SUPPORT_MAGIC_GATE
  static int mg_set_tpc_para_sub(struct rtsx_chip *chip,
  			int type, u8 mg_entry_num);
  #endif
  
  static int reset_ms_pro(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;
  #ifdef XC_POWERCLASS
  	u8 change_power_class;
  
  	if (chip->ms_power_class_en & 0x02)
  		change_power_class = 2;
  	else if (chip->ms_power_class_en & 0x01)
  		change_power_class = 1;
  	else
  		change_power_class = 0;
  #endif
  
  #ifdef XC_POWERCLASS
  Retry:
  #endif
  	retval = ms_pro_reset_flow(chip, 1);
  	if (retval != STATUS_SUCCESS) {
  		if (ms_card->switch_8bit_fail) {
  			retval = ms_pro_reset_flow(chip, 0);

  			if (retval != STATUS_SUCCESS) {
  				rtsx_trace(chip);
  				return STATUS_FAIL;
  			}

  		} else {

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  	}
  
  	retval = ms_read_attribute_info(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  #ifdef XC_POWERCLASS
  	if (CHK_HG8BIT(ms_card))
  		change_power_class = 0;
  
  	if (change_power_class && CHK_MSXC(ms_card)) {
  		u8 power_class_en = chip->ms_power_class_en;

  		dev_dbg(rtsx_dev(chip), "power_class_en = 0x%x
  ",
  			power_class_en);
  		dev_dbg(rtsx_dev(chip), "change_power_class = %d
  ",
  			change_power_class);

  
  		if (change_power_class)
  			power_class_en &= (1 << (change_power_class - 1));
  		else
  			power_class_en = 0;
  
  		if (power_class_en) {
  			u8 power_class_mode =
  				(ms_card->raw_sys_info[46] & 0x18) >> 3;

  			dev_dbg(rtsx_dev(chip), "power_class_mode = 0x%x",

  				power_class_mode);
  			if (change_power_class > power_class_mode)
  				change_power_class = power_class_mode;
  			if (change_power_class) {
  				retval = msxc_change_power(chip,
  							change_power_class);
  				if (retval != STATUS_SUCCESS) {
  					change_power_class--;
  					goto Retry;
  				}
  			}
  		}
  	}
  #endif
  
  #ifdef SUPPORT_MAGIC_GATE
  	retval = mg_set_tpc_para_sub(chip, 0, 0);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  #endif
  
  	if (CHK_HG8BIT(ms_card))
  		chip->card_bus_width[chip->card2lun[MS_CARD]] = 8;
  	else
  		chip->card_bus_width[chip->card2lun[MS_CARD]] = 4;
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_read_status_reg(struct rtsx_chip *chip)
  {
  	int retval;
  	u8 val[2];
  
  	retval = ms_set_rw_reg_addr(chip, StatusReg0, 2, 0, 0);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_read_bytes(chip, READ_REG, 2, NO_WAIT_INT, val, 2);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (val[1] & (STS_UCDT | STS_UCEX | STS_UCFG)) {
  		ms_set_err_code(chip, MS_FLASH_READ_ERROR);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	return STATUS_SUCCESS;
  }

  static int ms_read_extra_data(struct rtsx_chip *chip,
  		u16 block_addr, u8 page_num, u8 *buf, int buf_len)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i;
  	u8 val, data[10];
  
  	retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
  				SystemParm, 6);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (CHK_MS4BIT(ms_card)) {
  		/* Parallel interface */
  		data[0] = 0x88;
  	} else {
  		/* Serial interface */
  		data[0] = 0x80;
  	}
  	data[1] = 0;
  	data[2] = (u8)(block_addr >> 8);
  	data[3] = (u8)block_addr;
  	data[4] = 0x40;
  	data[5] = page_num;
  
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_write_bytes(chip, WRITE_REG, 6, NO_WAIT_INT,
  					data, 6);
  		if (retval == STATUS_SUCCESS)
  			break;
  	}

  	if (i == MS_MAX_RETRY_COUNT) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_set_err_code(chip, MS_NO_ERROR);
  
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_send_cmd(chip, BLOCK_READ, WAIT_INT);
  		if (retval == STATUS_SUCCESS)
  			break;
  	}

  	if (i == MS_MAX_RETRY_COUNT) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_set_err_code(chip, MS_NO_ERROR);
  	retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (val & INT_REG_CMDNK) {
  		ms_set_err_code(chip, MS_CMD_NK);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  	if (val & INT_REG_CED) {
  		if (val & INT_REG_ERR) {
  			retval = ms_read_status_reg(chip);

  			if (retval != STATUS_SUCCESS) {
  				rtsx_trace(chip);
  				return STATUS_FAIL;
  			}

  
  			retval = ms_set_rw_reg_addr(chip, OverwriteFlag,
  						MS_EXTRA_SIZE, SystemParm, 6);

  			if (retval != STATUS_SUCCESS) {
  				rtsx_trace(chip);
  				return STATUS_FAIL;
  			}

  		}
  	}
  
  	retval = ms_read_bytes(chip, READ_REG, MS_EXTRA_SIZE, NO_WAIT_INT,
  			data, MS_EXTRA_SIZE);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (buf && buf_len) {
  		if (buf_len > MS_EXTRA_SIZE)
  			buf_len = MS_EXTRA_SIZE;
  		memcpy(buf, data, buf_len);
  	}
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_write_extra_data(struct rtsx_chip *chip,
  		u16 block_addr, u8 page_num, u8 *buf, int buf_len)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i;
  	u8 val, data[16];

  	if (!buf || (buf_len < MS_EXTRA_SIZE)) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
  				SystemParm, 6 + MS_EXTRA_SIZE);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (CHK_MS4BIT(ms_card))
  		data[0] = 0x88;
  	else
  		data[0] = 0x80;
  
  	data[1] = 0;
  	data[2] = (u8)(block_addr >> 8);
  	data[3] = (u8)block_addr;
  	data[4] = 0x40;
  	data[5] = page_num;
  
  	for (i = 6; i < MS_EXTRA_SIZE + 6; i++)
  		data[i] = buf[i - 6];

  	retval = ms_write_bytes(chip, WRITE_REG, (6 + MS_EXTRA_SIZE),

  				NO_WAIT_INT, data, 16);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_set_err_code(chip, MS_NO_ERROR);
  	retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (val & INT_REG_CMDNK) {
  		ms_set_err_code(chip, MS_CMD_NK);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  	if (val & INT_REG_CED) {
  		if (val & INT_REG_ERR) {
  			ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  	}
  
  	return STATUS_SUCCESS;
  }

  static int ms_read_page(struct rtsx_chip *chip, u16 block_addr, u8 page_num)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;
  	u8 val, data[6];
  
  	retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
  				SystemParm, 6);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (CHK_MS4BIT(ms_card))
  		data[0] = 0x88;
  	else
  		data[0] = 0x80;
  
  	data[1] = 0;
  	data[2] = (u8)(block_addr >> 8);
  	data[3] = (u8)block_addr;
  	data[4] = 0x20;
  	data[5] = page_num;

  	retval = ms_write_bytes(chip, WRITE_REG, 6, NO_WAIT_INT, data, 6);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_send_cmd(chip, BLOCK_READ, WAIT_INT);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_set_err_code(chip, MS_NO_ERROR);
  	retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (val & INT_REG_CMDNK) {
  		ms_set_err_code(chip, MS_CMD_NK);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	if (val & INT_REG_CED) {
  		if (val & INT_REG_ERR) {
  			if (!(val & INT_REG_BREQ)) {
  				ms_set_err_code(chip,  MS_FLASH_READ_ERROR);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  			retval = ms_read_status_reg(chip);
  			if (retval != STATUS_SUCCESS)
  				ms_set_err_code(chip,  MS_FLASH_WRITE_ERROR);
  
  		} else {
  			if (!(val & INT_REG_BREQ)) {
  				ms_set_err_code(chip, MS_BREQ_ERROR);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  		}
  	}
  
  	retval = ms_transfer_tpc(chip, MS_TM_NORMAL_READ, READ_PAGE_DATA,
  				0, NO_WAIT_INT);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  	if (ms_check_err_code(chip, MS_FLASH_WRITE_ERROR)) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }

  static int ms_set_bad_block(struct rtsx_chip *chip, u16 phy_blk)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;
  	u8 val, data[8], extra[MS_EXTRA_SIZE];
  
  	retval = ms_read_extra_data(chip, phy_blk, 0, extra, MS_EXTRA_SIZE);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
  				SystemParm, 7);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_set_err_code(chip, MS_NO_ERROR);
  
  	if (CHK_MS4BIT(ms_card))
  		data[0] = 0x88;
  	else
  		data[0] = 0x80;
  
  	data[1] = 0;
  	data[2] = (u8)(phy_blk >> 8);
  	data[3] = (u8)phy_blk;
  	data[4] = 0x80;
  	data[5] = 0;
  	data[6] = extra[0] & 0x7F;
  	data[7] = 0xFF;
  
  	retval = ms_write_bytes(chip, WRITE_REG, 7, NO_WAIT_INT, data, 7);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_set_err_code(chip, MS_NO_ERROR);
  	retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (val & INT_REG_CMDNK) {
  		ms_set_err_code(chip, MS_CMD_NK);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	if (val & INT_REG_CED) {
  		if (val & INT_REG_ERR) {
  			ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  	}
  
  	return STATUS_SUCCESS;
  }

  static int ms_erase_block(struct rtsx_chip *chip, u16 phy_blk)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i = 0;
  	u8 val, data[6];
  
  	retval = ms_set_rw_reg_addr(chip, OverwriteFlag, MS_EXTRA_SIZE,
  				SystemParm, 6);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_set_err_code(chip, MS_NO_ERROR);
  
  	if (CHK_MS4BIT(ms_card))
  		data[0] = 0x88;
  	else
  		data[0] = 0x80;
  
  	data[1] = 0;
  	data[2] = (u8)(phy_blk >> 8);
  	data[3] = (u8)phy_blk;
  	data[4] = 0;
  	data[5] = 0;
  
  	retval = ms_write_bytes(chip, WRITE_REG, 6, NO_WAIT_INT, data, 6);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  ERASE_RTY:
  	retval = ms_send_cmd(chip, BLOCK_ERASE, WAIT_INT);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_set_err_code(chip, MS_NO_ERROR);
  	retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (val & INT_REG_CMDNK) {
  		if (i < 3) {
  			i++;
  			goto ERASE_RTY;
  		}
  
  		ms_set_err_code(chip, MS_CMD_NK);
  		ms_set_bad_block(chip, phy_blk);

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	if (val & INT_REG_CED) {
  		if (val & INT_REG_ERR) {
  			ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  	}
  
  	return STATUS_SUCCESS;
  }

  static void ms_set_page_status(u16 log_blk, u8 type, u8 *extra, int extra_len)
  {
  	if (!extra || (extra_len < MS_EXTRA_SIZE))
  		return;
  
  	memset(extra, 0xFF, MS_EXTRA_SIZE);
  
  	if (type == setPS_NG) {
  		/* set page status as 1:NG,and block status keep 1:OK */
  		extra[0] = 0xB8;
  	} else {
  		/* set page status as 0:Data Error,and block status keep 1:OK */
  		extra[0] = 0x98;
  	}
  
  	extra[2] = (u8)(log_blk >> 8);
  	extra[3] = (u8)log_blk;
  }
  
  static int ms_init_page(struct rtsx_chip *chip, u16 phy_blk, u16 log_blk,
  			u8 start_page, u8 end_page)
  {
  	int retval;
  	u8 extra[MS_EXTRA_SIZE], i;
  
  	memset(extra, 0xff, MS_EXTRA_SIZE);
  
  	extra[0] = 0xf8;	/* Block, page OK, data erased */
  	extra[1] = 0xff;
  	extra[2] = (u8)(log_blk >> 8);
  	extra[3] = (u8)log_blk;
  
  	for (i = start_page; i < end_page; i++) {
  		if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
  			ms_set_err_code(chip, MS_NO_CARD);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  
  		retval = ms_write_extra_data(chip, phy_blk, i,
  					extra, MS_EXTRA_SIZE);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  	}
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_copy_page(struct rtsx_chip *chip, u16 old_blk, u16 new_blk,
  		u16 log_blk, u8 start_page, u8 end_page)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	bool uncorrect_flag = false;
  	int retval, rty_cnt;

  	u8 extra[MS_EXTRA_SIZE], val, i, j, data[16];

  	dev_dbg(rtsx_dev(chip), "Copy page from 0x%x to 0x%x, logical block is 0x%x
  ",

  		old_blk, new_blk, log_blk);

  	dev_dbg(rtsx_dev(chip), "start_page = %d, end_page = %d
  ",
  		start_page, end_page);

  
  	retval = ms_read_extra_data(chip, new_blk, 0, extra, MS_EXTRA_SIZE);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_read_status_reg(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  	retval = rtsx_read_register(chip, PPBUF_BASE2, &val);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  
  	if (val & BUF_FULL) {
  		retval = ms_send_cmd(chip, CLEAR_BUF, WAIT_INT);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		if (!(val & INT_REG_CED)) {
  			ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  	}
  
  	for (i = start_page; i < end_page; i++) {
  		if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
  			ms_set_err_code(chip, MS_NO_CARD);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  
  		ms_read_extra_data(chip, old_blk, i, extra, MS_EXTRA_SIZE);
  
  		retval = ms_set_rw_reg_addr(chip, OverwriteFlag,
  					MS_EXTRA_SIZE, SystemParm, 6);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		ms_set_err_code(chip, MS_NO_ERROR);
  
  		if (CHK_MS4BIT(ms_card))
  			data[0] = 0x88;
  		else
  			data[0] = 0x80;
  
  		data[1] = 0;
  		data[2] = (u8)(old_blk >> 8);
  		data[3] = (u8)old_blk;
  		data[4] = 0x20;
  		data[5] = i;

  		retval = ms_write_bytes(chip, WRITE_REG, 6, NO_WAIT_INT,

  					data, 6);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		retval = ms_send_cmd(chip, BLOCK_READ, WAIT_INT);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		ms_set_err_code(chip, MS_NO_ERROR);
  		retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		if (val & INT_REG_CMDNK) {
  			ms_set_err_code(chip, MS_CMD_NK);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  
  		if (val & INT_REG_CED) {
  			if (val & INT_REG_ERR) {
  				retval = ms_read_status_reg(chip);
  				if (retval != STATUS_SUCCESS) {

  					uncorrect_flag = true;

  					dev_dbg(rtsx_dev(chip), "Uncorrectable error
  ");

  				} else {

  					uncorrect_flag = false;

  				}
  
  				retval = ms_transfer_tpc(chip,
  							MS_TM_NORMAL_READ,
  							READ_PAGE_DATA,
  							0, NO_WAIT_INT);

  				if (retval != STATUS_SUCCESS) {
  					rtsx_trace(chip);
  					return STATUS_FAIL;
  				}

  
  				if (uncorrect_flag) {
  					ms_set_page_status(log_blk, setPS_NG,
  							extra, MS_EXTRA_SIZE);
  					if (i == 0)
  						extra[0] &= 0xEF;
  
  					ms_write_extra_data(chip, old_blk, i,
  							extra, MS_EXTRA_SIZE);

  					dev_dbg(rtsx_dev(chip), "page %d : extra[0] = 0x%x
  ",
  						i, extra[0]);

  					MS_SET_BAD_BLOCK_FLG(ms_card);
  
  					ms_set_page_status(log_blk, setPS_Error,
  							extra, MS_EXTRA_SIZE);
  					ms_write_extra_data(chip, new_blk, i,
  							extra, MS_EXTRA_SIZE);
  					continue;
  				}
  
  				for (rty_cnt = 0; rty_cnt < MS_MAX_RETRY_COUNT;
  				     rty_cnt++) {
  					retval = ms_transfer_tpc(
  						chip,
  						MS_TM_NORMAL_WRITE,
  						WRITE_PAGE_DATA,
  						0, NO_WAIT_INT);
  					if (retval == STATUS_SUCCESS)
  						break;
  				}

  				if (rty_cnt == MS_MAX_RETRY_COUNT) {
  					rtsx_trace(chip);
  					return STATUS_FAIL;
  				}

  			}
  
  			if (!(val & INT_REG_BREQ)) {
  				ms_set_err_code(chip, MS_BREQ_ERROR);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  		}
  
  		retval = ms_set_rw_reg_addr(chip, OverwriteFlag,

  				MS_EXTRA_SIZE, SystemParm, (6 + MS_EXTRA_SIZE));

  
  		ms_set_err_code(chip, MS_NO_ERROR);
  
  		if (CHK_MS4BIT(ms_card))
  			data[0] = 0x88;
  		else
  			data[0] = 0x80;
  
  		data[1] = 0;
  		data[2] = (u8)(new_blk >> 8);
  		data[3] = (u8)new_blk;
  		data[4] = 0x20;
  		data[5] = i;
  
  		if ((extra[0] & 0x60) != 0x60)
  			data[6] = extra[0];
  		else
  			data[6] = 0xF8;
  
  		data[6 + 1] = 0xFF;
  		data[6 + 2] = (u8)(log_blk >> 8);
  		data[6 + 3] = (u8)log_blk;
  
  		for (j = 4; j <= MS_EXTRA_SIZE; j++)
  			data[6 + j] = 0xFF;
  
  		retval = ms_write_bytes(chip, WRITE_REG, (6 + MS_EXTRA_SIZE),
  					NO_WAIT_INT, data, 16);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		ms_set_err_code(chip, MS_NO_ERROR);
  		retval = ms_read_bytes(chip, GET_INT, 1, NO_WAIT_INT, &val, 1);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  
  		if (val & INT_REG_CMDNK) {
  			ms_set_err_code(chip, MS_CMD_NK);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  
  		if (val & INT_REG_CED) {
  			if (val & INT_REG_ERR) {
  				ms_set_err_code(chip, MS_FLASH_WRITE_ERROR);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  		}
  
  		if (i == 0) {
  			retval = ms_set_rw_reg_addr(chip, OverwriteFlag,
  						MS_EXTRA_SIZE, SystemParm, 7);

  			if (retval != STATUS_SUCCESS) {
  				rtsx_trace(chip);
  				return STATUS_FAIL;
  			}

  
  			ms_set_err_code(chip, MS_NO_ERROR);
  
  			if (CHK_MS4BIT(ms_card))
  				data[0] = 0x88;
  			else
  				data[0] = 0x80;
  
  			data[1] = 0;
  			data[2] = (u8)(old_blk >> 8);
  			data[3] = (u8)old_blk;
  			data[4] = 0x80;
  			data[5] = 0;
  			data[6] = 0xEF;
  			data[7] = 0xFF;
  
  			retval = ms_write_bytes(chip, WRITE_REG, 7,
  						NO_WAIT_INT, data, 8);

  			if (retval != STATUS_SUCCESS) {
  				rtsx_trace(chip);
  				return STATUS_FAIL;
  			}

  
  			retval = ms_send_cmd(chip, BLOCK_WRITE, WAIT_INT);

  			if (retval != STATUS_SUCCESS) {
  				rtsx_trace(chip);
  				return STATUS_FAIL;
  			}

  
  			ms_set_err_code(chip, MS_NO_ERROR);
  			retval = ms_read_bytes(chip, GET_INT, 1,
  					NO_WAIT_INT, &val, 1);

  			if (retval != STATUS_SUCCESS) {
  				rtsx_trace(chip);
  				return STATUS_FAIL;
  			}

  
  			if (val & INT_REG_CMDNK) {
  				ms_set_err_code(chip, MS_CMD_NK);

  				rtsx_trace(chip);
  				return STATUS_FAIL;

  			}
  
  			if (val & INT_REG_CED) {
  				if (val & INT_REG_ERR) {
  					ms_set_err_code(chip,
  							MS_FLASH_WRITE_ERROR);

  					rtsx_trace(chip);
  					return STATUS_FAIL;

  				}
  			}
  		}
  	}
  
  	return STATUS_SUCCESS;
  }

  static int reset_ms(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;
  	u16 i, reg_addr, block_size;
  	u8 val, extra[MS_EXTRA_SIZE], j, *ptr;
  #ifndef SUPPORT_MAGIC_GATE
  	u16 eblock_cnt;
  #endif
  
  	retval = ms_prepare_reset(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_card->ms_type |= TYPE_MS;
  
  	retval = ms_send_cmd(chip, MS_RESET, NO_WAIT_INT);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_read_status_reg(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  	retval = rtsx_read_register(chip, PPBUF_BASE2, &val);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  	if (val & WRT_PRTCT)
  		chip->card_wp |= MS_CARD;
  	else
  		chip->card_wp &= ~MS_CARD;
  
  	i = 0;
  
  RE_SEARCH:
  	/* Search Boot Block */
  	while (i < (MAX_DEFECTIVE_BLOCK + 2)) {
  		if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
  			ms_set_err_code(chip, MS_NO_CARD);

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  
  		retval = ms_read_extra_data(chip, i, 0, extra, MS_EXTRA_SIZE);
  		if (retval != STATUS_SUCCESS) {
  			i++;
  			continue;
  		}
  
  		if (extra[0] & BLOCK_OK) {
  			if (!(extra[1] & NOT_BOOT_BLOCK)) {
  				ms_card->boot_block = i;
  				break;
  			}
  		}
  		i++;
  	}
  
  	if (i == (MAX_DEFECTIVE_BLOCK + 2)) {

  		dev_dbg(rtsx_dev(chip), "No boot block found!");

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	for (j = 0; j < 3; j++) {
  		retval = ms_read_page(chip, ms_card->boot_block, j);
  		if (retval != STATUS_SUCCESS) {
  			if (ms_check_err_code(chip, MS_FLASH_WRITE_ERROR)) {
  				i = ms_card->boot_block + 1;
  				ms_set_err_code(chip, MS_NO_ERROR);
  				goto RE_SEARCH;
  			}
  		}
  	}
  
  	retval = ms_read_page(chip, ms_card->boot_block, 0);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	/* Read MS system information as sys_info */
  	rtsx_init_cmd(chip);
  
  	for (i = 0; i < 96; i++)
  		rtsx_add_cmd(chip, READ_REG_CMD, PPBUF_BASE2 + 0x1A0 + i, 0, 0);
  
  	retval = rtsx_send_cmd(chip, MS_CARD, 100);

  	if (retval < 0) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ptr = rtsx_get_cmd_data(chip);
  	memcpy(ms_card->raw_sys_info, ptr, 96);
  
  	/* Read useful block contents */
  	rtsx_init_cmd(chip);
  
  	rtsx_add_cmd(chip, READ_REG_CMD, HEADER_ID0, 0, 0);
  	rtsx_add_cmd(chip, READ_REG_CMD, HEADER_ID1, 0, 0);
  
  	for (reg_addr = DISABLED_BLOCK0; reg_addr <= DISABLED_BLOCK3;
  	     reg_addr++)
  		rtsx_add_cmd(chip, READ_REG_CMD, reg_addr, 0, 0);
  
  	for (reg_addr = BLOCK_SIZE_0; reg_addr <= PAGE_SIZE_1; reg_addr++)
  		rtsx_add_cmd(chip, READ_REG_CMD, reg_addr, 0, 0);
  
  	rtsx_add_cmd(chip, READ_REG_CMD, MS_Device_Type, 0, 0);
  	rtsx_add_cmd(chip, READ_REG_CMD, MS_4bit_Support, 0, 0);
  
  	retval = rtsx_send_cmd(chip, MS_CARD, 100);

  	if (retval < 0) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ptr = rtsx_get_cmd_data(chip);

  	dev_dbg(rtsx_dev(chip), "Boot block data:
  ");

  	dev_dbg(rtsx_dev(chip), "%*ph
  ", 16, ptr);

  
  	/* Block ID error
  	 * HEADER_ID0, HEADER_ID1
  	 */
  	if (ptr[0] != 0x00 || ptr[1] != 0x01) {
  		i = ms_card->boot_block + 1;
  		goto RE_SEARCH;
  	}
  
  	/* Page size error
  	 * PAGE_SIZE_0, PAGE_SIZE_1
  	 */
  	if (ptr[12] != 0x02 || ptr[13] != 0x00) {
  		i = ms_card->boot_block + 1;
  		goto RE_SEARCH;
  	}
  
  	if ((ptr[14] == 1) || (ptr[14] == 3))
  		chip->card_wp |= MS_CARD;
  
  	/* BLOCK_SIZE_0, BLOCK_SIZE_1 */
  	block_size = ((u16)ptr[6] << 8) | ptr[7];
  	if (block_size == 0x0010) {
  		/* Block size 16KB */
  		ms_card->block_shift = 5;
  		ms_card->page_off = 0x1F;
  	} else if (block_size == 0x0008) {
  		/* Block size 8KB */
  		ms_card->block_shift = 4;
  		ms_card->page_off = 0x0F;
  	}
  
  	/* BLOCK_COUNT_0, BLOCK_COUNT_1 */
  	ms_card->total_block = ((u16)ptr[8] << 8) | ptr[9];
  
  #ifdef SUPPORT_MAGIC_GATE
  	j = ptr[10];
  
  	if (ms_card->block_shift == 4)  { /* 4MB or 8MB */
  		if (j < 2)  { /* Effective block for 4MB: 0x1F0 */
  			ms_card->capacity = 0x1EE0;
  		} else { /* Effective block for 8MB: 0x3E0 */
  			ms_card->capacity = 0x3DE0;
  		}
  	} else  { /* 16MB, 32MB, 64MB or 128MB */
  		if (j < 5)  { /* Effective block for 16MB: 0x3E0 */
  			ms_card->capacity = 0x7BC0;
  		} else if (j < 0xA) { /* Effective block for 32MB: 0x7C0 */
  			ms_card->capacity = 0xF7C0;
  		} else if (j < 0x11) { /* Effective block for 64MB: 0xF80 */
  			ms_card->capacity = 0x1EF80;
  		} else { /* Effective block for 128MB: 0x1F00 */
  			ms_card->capacity = 0x3DF00;
  		}
  	}
  #else
  	/* EBLOCK_COUNT_0, EBLOCK_COUNT_1 */
  	eblock_cnt = ((u16)ptr[10] << 8) | ptr[11];
  
  	ms_card->capacity = ((u32)eblock_cnt - 2) << ms_card->block_shift;
  #endif
  
  	chip->capacity[chip->card2lun[MS_CARD]] = ms_card->capacity;
  
  	/* Switch I/F Mode */
  	if (ptr[15]) {
  		retval = ms_set_rw_reg_addr(chip, 0, 0, SystemParm, 1);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  		retval = rtsx_write_register(chip, PPBUF_BASE2, 0xFF, 0x88);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}
  		retval = rtsx_write_register(chip, PPBUF_BASE2 + 1, 0xFF, 0);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}

  		retval = ms_transfer_tpc(chip, MS_TM_WRITE_BYTES, WRITE_REG, 1,

  					NO_WAIT_INT);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  		retval = rtsx_write_register(chip, MS_CFG,
  					     0x58 | MS_NO_CHECK_INT,
  					     MS_BUS_WIDTH_4 | PUSH_TIME_ODD | MS_NO_CHECK_INT);
  		if (retval) {
  			rtsx_trace(chip);
  			return retval;
  		}

  
  		ms_card->ms_type |= MS_4BIT;
  	}
  
  	if (CHK_MS4BIT(ms_card))
  		chip->card_bus_width[chip->card2lun[MS_CARD]] = 4;
  	else
  		chip->card_bus_width[chip->card2lun[MS_CARD]] = 1;
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_init_l2p_tbl(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int size, i, seg_no, retval;
  	u16 defect_block, reg_addr;
  	u8 val1, val2;
  
  	ms_card->segment_cnt = ms_card->total_block >> 9;

  	dev_dbg(rtsx_dev(chip), "ms_card->segment_cnt = %d
  ",
  		ms_card->segment_cnt);

  
  	size = ms_card->segment_cnt * sizeof(struct zone_entry);
  	ms_card->segment = vzalloc(size);

  	if (!ms_card->segment) {

  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = ms_read_page(chip, ms_card->boot_block, 1);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		goto INIT_FAIL;
  	}

  
  	reg_addr = PPBUF_BASE2;
  	for (i = 0; i < (((ms_card->total_block >> 9) * 10) + 1); i++) {

  		int block_no;

  		retval = rtsx_read_register(chip, reg_addr++, &val1);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			goto INIT_FAIL;
  		}

  
  		retval = rtsx_read_register(chip, reg_addr++, &val2);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			goto INIT_FAIL;
  		}

  
  		defect_block = ((u16)val1 << 8) | val2;
  		if (defect_block == 0xFFFF)
  			break;
  
  		seg_no = defect_block / 512;

  
  		block_no = ms_card->segment[seg_no].disable_count++;
  		ms_card->segment[seg_no].defect_list[block_no] = defect_block;

  	}
  
  	for (i = 0; i < ms_card->segment_cnt; i++) {
  		ms_card->segment[i].build_flag = 0;
  		ms_card->segment[i].l2p_table = NULL;
  		ms_card->segment[i].free_table = NULL;
  		ms_card->segment[i].get_index = 0;
  		ms_card->segment[i].set_index = 0;
  		ms_card->segment[i].unused_blk_cnt = 0;

  		dev_dbg(rtsx_dev(chip), "defective block count of segment %d is %d
  ",
  			i, ms_card->segment[i].disable_count);

  	}
  
  	return STATUS_SUCCESS;
  
  INIT_FAIL:

  	vfree(ms_card->segment);
  	ms_card->segment = NULL;

  
  	return STATUS_FAIL;
  }
  
  static u16 ms_get_l2p_tbl(struct rtsx_chip *chip, int seg_no, u16 log_off)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	struct zone_entry *segment;

  	if (!ms_card->segment)

  		return 0xFFFF;

  	segment = &ms_card->segment[seg_no];

  
  	if (segment->l2p_table)
  		return segment->l2p_table[log_off];
  
  	return 0xFFFF;
  }
  
  static void ms_set_l2p_tbl(struct rtsx_chip *chip,
  			int seg_no, u16 log_off, u16 phy_blk)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	struct zone_entry *segment;

  	if (!ms_card->segment)

  		return;

  	segment = &ms_card->segment[seg_no];

  	if (segment->l2p_table)
  		segment->l2p_table[log_off] = phy_blk;
  }
  
  static void ms_set_unused_block(struct rtsx_chip *chip, u16 phy_blk)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	struct zone_entry *segment;
  	int seg_no;
  
  	seg_no = (int)phy_blk >> 9;

  	segment = &ms_card->segment[seg_no];

  
  	segment->free_table[segment->set_index++] = phy_blk;
  	if (segment->set_index >= MS_FREE_TABLE_CNT)
  		segment->set_index = 0;
  
  	segment->unused_blk_cnt++;
  }
  
  static u16 ms_get_unused_block(struct rtsx_chip *chip, int seg_no)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	struct zone_entry *segment;
  	u16 phy_blk;

  	segment = &ms_card->segment[seg_no];

  
  	if (segment->unused_blk_cnt <= 0)
  		return 0xFFFF;
  
  	phy_blk = segment->free_table[segment->get_index];
  	segment->free_table[segment->get_index++] = 0xFFFF;
  	if (segment->get_index >= MS_FREE_TABLE_CNT)
  		segment->get_index = 0;
  
  	segment->unused_blk_cnt--;
  
  	return phy_blk;
  }
  
  static const unsigned short ms_start_idx[] = {0, 494, 990, 1486, 1982, 2478,
  					      2974, 3470, 3966, 4462, 4958,
  					      5454, 5950, 6446, 6942, 7438,
  					      7934};
  
  static int ms_arbitrate_l2p(struct rtsx_chip *chip, u16 phy_blk,
  			u16 log_off, u8 us1, u8 us2)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	struct zone_entry *segment;
  	int seg_no;
  	u16 tmp_blk;
  
  	seg_no = (int)phy_blk >> 9;

  	segment = &ms_card->segment[seg_no];

  	tmp_blk = segment->l2p_table[log_off];
  
  	if (us1 != us2) {
  		if (us1 == 0) {
  			if (!(chip->card_wp & MS_CARD))
  				ms_erase_block(chip, tmp_blk);
  
  			ms_set_unused_block(chip, tmp_blk);
  			segment->l2p_table[log_off] = phy_blk;
  		} else {
  			if (!(chip->card_wp & MS_CARD))
  				ms_erase_block(chip, phy_blk);
  
  			ms_set_unused_block(chip, phy_blk);
  		}
  	} else {
  		if (phy_blk < tmp_blk) {
  			if (!(chip->card_wp & MS_CARD))
  				ms_erase_block(chip, phy_blk);
  
  			ms_set_unused_block(chip, phy_blk);
  		} else {
  			if (!(chip->card_wp & MS_CARD))
  				ms_erase_block(chip, tmp_blk);
  
  			ms_set_unused_block(chip, tmp_blk);
  			segment->l2p_table[log_off] = phy_blk;
  		}
  	}
  
  	return STATUS_SUCCESS;
  }
  
  static int ms_build_l2p_tbl(struct rtsx_chip *chip, int seg_no)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	struct zone_entry *segment;

  	bool defect_flag;
  	int retval, table_size, disable_cnt, i;

  	u16 start, end, phy_blk, log_blk, tmp_blk, idx;

  	u8 extra[MS_EXTRA_SIZE], us1, us2;

  	dev_dbg(rtsx_dev(chip), "ms_build_l2p_tbl: %d
  ", seg_no);

  	if (!ms_card->segment) {

  		retval = ms_init_l2p_tbl(chip);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return retval;
  		}

  	}
  
  	if (ms_card->segment[seg_no].build_flag) {

  		dev_dbg(rtsx_dev(chip), "l2p table of segment %d has been built
  ",
  			seg_no);

  		return STATUS_SUCCESS;
  	}
  
  	if (seg_no == 0)
  		table_size = 494;
  	else
  		table_size = 496;

  	segment = &ms_card->segment[seg_no];

  	if (!segment->l2p_table) {

  		segment->l2p_table = vmalloc(table_size * 2);

  		if (!segment->l2p_table) {

  			rtsx_trace(chip);
  			goto BUILD_FAIL;
  		}

  	}
  	memset((u8 *)(segment->l2p_table), 0xff, table_size * 2);

  	if (!segment->free_table) {

  		segment->free_table = vmalloc(MS_FREE_TABLE_CNT * 2);

  		if (!segment->free_table) {

  			rtsx_trace(chip);
  			goto BUILD_FAIL;
  		}

  	}
  	memset((u8 *)(segment->free_table), 0xff, MS_FREE_TABLE_CNT * 2);
  
  	start = (u16)seg_no << 9;
  	end = (u16)(seg_no + 1) << 9;
  
  	disable_cnt = segment->disable_count;
  
  	segment->get_index = segment->set_index = 0;
  	segment->unused_blk_cnt = 0;
  
  	for (phy_blk = start; phy_blk < end; phy_blk++) {
  		if (disable_cnt) {

  			defect_flag = false;

  			for (i = 0; i < segment->disable_count; i++) {
  				if (phy_blk == segment->defect_list[i]) {

  					defect_flag = true;

  					break;
  				}
  			}
  			if (defect_flag) {
  				disable_cnt--;
  				continue;
  			}
  		}
  
  		retval = ms_read_extra_data(chip, phy_blk, 0,
  					extra, MS_EXTRA_SIZE);
  		if (retval != STATUS_SUCCESS) {

  			dev_dbg(rtsx_dev(chip), "read extra data fail
  ");

  			ms_set_bad_block(chip, phy_blk);
  			continue;
  		}
  
  		if (seg_no == ms_card->segment_cnt - 1) {
  			if (!(extra[1] & NOT_TRANSLATION_TABLE)) {
  				if (!(chip->card_wp & MS_CARD)) {
  					retval = ms_erase_block(chip, phy_blk);
  					if (retval != STATUS_SUCCESS)
  						continue;
  					extra[2] = 0xff;
  					extra[3] = 0xff;
  				}
  			}
  		}
  
  		if (!(extra[0] & BLOCK_OK))
  			continue;
  		if (!(extra[1] & NOT_BOOT_BLOCK))
  			continue;
  		if ((extra[0] & PAGE_OK) != PAGE_OK)
  			continue;
  
  		log_blk = ((u16)extra[2] << 8) | extra[3];
  
  		if (log_blk == 0xFFFF) {
  			if (!(chip->card_wp & MS_CARD)) {
  				retval = ms_erase_block(chip, phy_blk);
  				if (retval != STATUS_SUCCESS)
  					continue;
  			}
  			ms_set_unused_block(chip, phy_blk);
  			continue;
  		}
  
  		if ((log_blk < ms_start_idx[seg_no]) ||

  				(log_blk >= ms_start_idx[seg_no + 1])) {

  			if (!(chip->card_wp & MS_CARD)) {
  				retval = ms_erase_block(chip, phy_blk);
  				if (retval != STATUS_SUCCESS)
  					continue;
  			}
  			ms_set_unused_block(chip, phy_blk);
  			continue;
  		}

  		idx = log_blk - ms_start_idx[seg_no];
  
  		if (segment->l2p_table[idx] == 0xFFFF) {
  			segment->l2p_table[idx] = phy_blk;

  			continue;
  		}
  
  		us1 = extra[0] & 0x10;

  		tmp_blk = segment->l2p_table[idx];

  		retval = ms_read_extra_data(chip, tmp_blk, 0,
  					extra, MS_EXTRA_SIZE);
  		if (retval != STATUS_SUCCESS)
  			continue;
  		us2 = extra[0] & 0x10;
  
  		(void)ms_arbitrate_l2p(chip, phy_blk,

  				log_blk - ms_start_idx[seg_no], us1, us2);

  		continue;
  	}
  
  	segment->build_flag = 1;

  	dev_dbg(rtsx_dev(chip), "unused block count: %d
  ",
  		segment->unused_blk_cnt);

  
  	/* Logical Address Confirmation Process */
  	if (seg_no == ms_card->segment_cnt - 1) {
  		if (segment->unused_blk_cnt < 2)
  			chip->card_wp |= MS_CARD;
  	} else {
  		if (segment->unused_blk_cnt < 1)
  			chip->card_wp |= MS_CARD;
  	}
  
  	if (chip->card_wp & MS_CARD)
  		return STATUS_SUCCESS;
  
  	for (log_blk = ms_start_idx[seg_no];
  	     log_blk < ms_start_idx[seg_no + 1]; log_blk++) {

  		idx = log_blk - ms_start_idx[seg_no];
  		if (segment->l2p_table[idx] == 0xFFFF) {

  			phy_blk = ms_get_unused_block(chip, seg_no);
  			if (phy_blk == 0xFFFF) {
  				chip->card_wp |= MS_CARD;
  				return STATUS_SUCCESS;
  			}
  			retval = ms_init_page(chip, phy_blk, log_blk, 0, 1);

  			if (retval != STATUS_SUCCESS) {
  				rtsx_trace(chip);
  				goto BUILD_FAIL;
  			}

  			segment->l2p_table[idx] = phy_blk;

  			if (seg_no == ms_card->segment_cnt - 1) {
  				if (segment->unused_blk_cnt < 2) {
  					chip->card_wp |= MS_CARD;
  					return STATUS_SUCCESS;
  				}
  			} else {
  				if (segment->unused_blk_cnt < 1) {
  					chip->card_wp |= MS_CARD;
  					return STATUS_SUCCESS;
  				}
  			}
  		}
  	}
  
  	/* Make boot block be the first normal block */
  	if (seg_no == 0) {
  		for (log_blk = 0; log_blk < 494; log_blk++) {
  			tmp_blk = segment->l2p_table[log_blk];
  			if (tmp_blk < ms_card->boot_block) {

  				dev_dbg(rtsx_dev(chip), "Boot block is not the first normal block.
  ");

  
  				if (chip->card_wp & MS_CARD)
  					break;
  
  				phy_blk = ms_get_unused_block(chip, 0);
  				retval = ms_copy_page(chip, tmp_blk, phy_blk,
  						log_blk, 0, ms_card->page_off + 1);

  				if (retval != STATUS_SUCCESS) {
  					rtsx_trace(chip);
  					return STATUS_FAIL;
  				}

  
  				segment->l2p_table[log_blk] = phy_blk;
  
  				retval = ms_set_bad_block(chip, tmp_blk);

  				if (retval != STATUS_SUCCESS) {
  					rtsx_trace(chip);
  					return STATUS_FAIL;
  				}

  			}
  		}
  	}
  
  	return STATUS_SUCCESS;
  
  BUILD_FAIL:
  	segment->build_flag = 0;

  	vfree(segment->l2p_table);
  	segment->l2p_table = NULL;
  	vfree(segment->free_table);
  	segment->free_table = NULL;

  
  	return STATUS_FAIL;
  }

  int reset_ms_card(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;
  
  	memset(ms_card, 0, sizeof(struct ms_info));
  
  	retval = enable_card_clock(chip, MS_CARD);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	retval = select_card(chip, MS_CARD);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	ms_card->ms_type = 0;
  
  	retval = reset_ms_pro(chip);
  	if (retval != STATUS_SUCCESS) {
  		if (ms_card->check_ms_flow) {
  			retval = reset_ms(chip);

  			if (retval != STATUS_SUCCESS) {
  				rtsx_trace(chip);
  				return STATUS_FAIL;
  			}

  		} else {

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  	}
  
  	retval = ms_set_init_para(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (!CHK_MSPRO(ms_card)) {
  		/* Build table for the last segment,
  		 * to check if L2P table block exists, erasing it
  		 */
  		retval = ms_build_l2p_tbl(chip, ms_card->total_block / 512 - 1);

  		if (retval != STATUS_SUCCESS) {
  			rtsx_trace(chip);
  			return STATUS_FAIL;
  		}

  	}

  	dev_dbg(rtsx_dev(chip), "ms_card->ms_type = 0x%x
  ", ms_card->ms_type);

  
  	return STATUS_SUCCESS;
  }
  
  static int mspro_set_rw_cmd(struct rtsx_chip *chip,
  			u32 start_sec, u16 sec_cnt, u8 cmd)
  {
  	int retval, i;
  	u8 data[8];
  
  	data[0] = cmd;
  	data[1] = (u8)(sec_cnt >> 8);
  	data[2] = (u8)sec_cnt;
  	data[3] = (u8)(start_sec >> 24);
  	data[4] = (u8)(start_sec >> 16);
  	data[5] = (u8)(start_sec >> 8);
  	data[6] = (u8)start_sec;
  	data[7] = 0;
  
  	for (i = 0; i < MS_MAX_RETRY_COUNT; i++) {
  		retval = ms_write_bytes(chip, PRO_EX_SET_CMD, 7,
  					WAIT_INT, data, 8);
  		if (retval == STATUS_SUCCESS)
  			break;
  	}

  	if (i == MS_MAX_RETRY_COUNT) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }

  void mspro_stop_seq_mode(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;

  	if (ms_card->seq_mode) {
  		retval = ms_switch_clock(chip);
  		if (retval != STATUS_SUCCESS)
  			return;
  
  		ms_card->seq_mode = 0;
  		ms_card->total_sec_cnt = 0;
  		ms_send_cmd(chip, PRO_STOP, WAIT_INT);
  
  		rtsx_write_register(chip, RBCTL, RB_FLUSH, RB_FLUSH);
  	}
  }
  
  static inline int ms_auto_tune_clock(struct rtsx_chip *chip)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval;

  	if (chip->asic_code) {
  		if (ms_card->ms_clock > 30)
  			ms_card->ms_clock -= 20;
  	} else {
  		if (ms_card->ms_clock == CLK_80)
  			ms_card->ms_clock = CLK_60;
  		else if (ms_card->ms_clock == CLK_60)
  			ms_card->ms_clock = CLK_40;
  	}
  
  	retval = ms_switch_clock(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	return STATUS_SUCCESS;
  }
  
  static int mspro_rw_multi_sector(struct scsi_cmnd *srb,
  				struct rtsx_chip *chip, u32 start_sector,
  				u16 sector_cnt)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	bool mode_2k = false;
  	int retval;

  	u16 count;
  	u8 val, trans_mode, rw_tpc, rw_cmd;
  
  	ms_set_err_code(chip, MS_NO_ERROR);
  
  	ms_card->cleanup_counter = 0;
  
  	if (CHK_MSHG(ms_card)) {
  		if ((start_sector % 4) || (sector_cnt % 4)) {
  			if (srb->sc_data_direction == DMA_FROM_DEVICE) {
  				rw_tpc = PRO_READ_LONG_DATA;
  				rw_cmd = PRO_READ_DATA;
  			} else {
  				rw_tpc = PRO_WRITE_LONG_DATA;
  				rw_cmd = PRO_WRITE_DATA;
  			}
  		} else {
  			if (srb->sc_data_direction == DMA_FROM_DEVICE) {
  				rw_tpc = PRO_READ_QUAD_DATA;
  				rw_cmd = PRO_READ_2K_DATA;
  			} else {
  				rw_tpc = PRO_WRITE_QUAD_DATA;
  				rw_cmd = PRO_WRITE_2K_DATA;
  			}

  			mode_2k = true;

  		}
  	} else {
  		if (srb->sc_data_direction == DMA_FROM_DEVICE) {
  			rw_tpc = PRO_READ_LONG_DATA;
  			rw_cmd = PRO_READ_DATA;
  		} else {
  			rw_tpc = PRO_WRITE_LONG_DATA;
  			rw_cmd = PRO_WRITE_DATA;
  		}
  	}
  
  	retval = ms_switch_clock(chip);

  	if (retval != STATUS_SUCCESS) {
  		rtsx_trace(chip);
  		return STATUS_FAIL;
  	}

  
  	if (srb->sc_data_direction == DMA_FROM_DEVICE)
  		trans_mode = MS_TM_AUTO_READ;
  	else
  		trans_mode = MS_TM_AUTO_WRITE;

  	retval = rtsx_read_register(chip, MS_TRANS_CFG, &val);
  	if (retval) {
  		rtsx_trace(chip);
  		return retval;
  	}

  
  	if (ms_card->seq_mode) {
  		if ((ms_card->pre_dir != srb->sc_data_direction)
  				|| ((ms_card->pre_sec_addr + ms_card->pre_sec_cnt) != start_sector)
  				|| (mode_2k && (ms_card->seq_mode & MODE_512_SEQ))
  				|| (!mode_2k && (ms_card->seq_mode & MODE_2K_SEQ))
  				|| !(val & MS_INT_BREQ)
  				|| ((ms_card->total_sec_cnt + sector_cnt) > 0xFE00)) {
  			ms_card->seq_mode = 0;
  			ms_card->total_sec_cnt = 0;
  			if (val & MS_INT_BREQ) {
  				retval = ms_send_cmd(chip, PRO_STOP, WAIT_INT);

  				if (retval != STATUS_SUCCESS) {
  					rtsx_trace(chip);
  					return STATUS_FAIL;
  				}

  
  				rtsx_write_register(chip, RBCTL, RB_FLUSH, RB_FLUSH);
  			}
  		}
  	}
  
  	if (!ms_card->seq_mode) {
  		ms_card->total_sec_cnt = 0;
  		if (sector_cnt >= SEQ_START_CRITERIA) {
  			if ((ms_card->capacity - start_sector) > 0xFE00)
  				count = 0xFE00;
  			else
  				count = (u16)(ms_card->capacity - start_sector);
  
  			if (count > sector_cnt) {
  				if (mode_2k)

  					ms_card->seq_mode = MODE_2K_SEQ;

  				else

  					ms_card->seq_mode = MODE_512_SEQ;

  			}
  		} else {
  			count = sector_cnt;
  		}
  		retval = mspro_set_rw_cmd(chip, start_sector, count, rw_cmd);
  		if (retval != STATUS_SUCCESS) {
  			ms_card->seq_mode = 0;

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  	}
  
  	retval = ms_transfer_data(chip, trans_mode, rw_tpc, sector_cnt,
  				WAIT_INT, mode_2k, scsi_sg_count(srb),
  				scsi_sglist(srb), scsi_bufflen(srb));
  	if (retval != STATUS_SUCCESS) {
  		ms_card->seq_mode = 0;
  		rtsx_read_register(chip, MS_TRANS_CFG, &val);
  		rtsx_clear_ms_error(chip);
  
  		if (detect_card_cd(chip, MS_CARD) != STATUS_SUCCESS) {
  			chip->rw_need_retry = 0;

  			dev_dbg(rtsx_dev(chip), "No card exist, exit mspro_rw_multi_sector
  ");

  			rtsx_trace(chip);
  			return STATUS_FAIL;

  		}
  
  		if (val & MS_INT_BREQ)
  			ms_send_cmd(chip, PRO_STOP, WAIT_INT);
  
  		if (val & (MS_CRC16_ERR | MS_RDY_TIMEOUT)) {

  			dev_dbg(rtsx_dev(chip), "MSPro CRC error, tune clock!
  ");

  			chip->rw_need_retry = 1;
  			ms_auto_tune_clock(chip);
  		}

  		rtsx_trace(chip);
  		return retval;

  	}
  
  	if (ms_card->seq_mode) {
  		ms_card->pre_sec_addr = start_sector;
  		ms_card->pre_sec_cnt = sector_cnt;
  		ms_card->pre_dir = srb->sc_data_direction;
  		ms_card->total_sec_cnt += sector_cnt;
  	}
  
  	return STATUS_SUCCESS;
  }
  
  static int mspro_read_format_progress(struct rtsx_chip *chip,
  				const int short_data_len)
  {

  	struct ms_info *ms_card = &chip->ms_card;

  	int retval, i;
  	u32 total_progress, cur_progress;
  	u8 cnt, tmp;
  	u8 data[8];

  	dev_dbg(rtsx_dev(chip), "mspro_read_format_progress, short_data_len = %d
  ",

  		short_data_len);
  
  	retval = ms_switch_clock(chip);
  	if (retval != STATUS_SUCCESS) {
  		ms_card->format_status = FORMAT_FAIL;

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	retval = rtsx_read_register(chip, MS_TRANS_CFG, &tmp);
  	if (retval != STATUS_SUCCESS) {
  		ms_card->format_status = FORMAT_FAIL;

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	if (!(tmp & MS_INT_BREQ)) {
  		if ((tmp & (MS_INT_CED | MS_INT_BREQ | MS_INT_CMDNK | MS_INT_ERR)) == MS_INT_CED) {
  			ms_card->format_status = FORMAT_SUCCESS;
  			return STATUS_SUCCESS;
  		}
  		ms_card->format_status = FORMAT_FAIL;

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	if (short_data_len >= 256)
  		cnt = 0;
  	else
  		cnt = (u8)short_data_len;
  
  	retval = rtsx_write_register(chip, MS_CFG, MS_NO_CHECK_INT,
  				MS_NO_CHECK_INT);
  	if (retval != STATUS_SUCCESS) {
  		ms_card->format_status = FORMAT_FAIL;

  		rtsx_trace(chip);
  		return STATUS_FAIL;

  	}
  
  	retval = ms_read_bytes(chip, PRO_READ_SHORT_DATA, cnt, WAIT_INT,
  			data, 8);
  	if (retval != STATUS_SUCCESS) {
  		ms_card->format_status = FORMAT_FAIL;