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