1 // SPDX-License-Identifier: GPL-2.0-only
2 /* Realtek USB SD/MMC Card Interface driver
3  *
4  * Copyright(c) 2009-2013 Realtek Semiconductor Corp. All rights reserved.
5  *
6  * Author:
7  *   Roger Tseng <rogerable@realtek.com>
8  */
9 
10 #include <linux/module.h>
11 #include <linux/slab.h>
12 #include <linux/delay.h>
13 #include <linux/platform_device.h>
14 #include <linux/usb.h>
15 #include <linux/mmc/host.h>
16 #include <linux/mmc/mmc.h>
17 #include <linux/mmc/sd.h>
18 #include <linux/mmc/card.h>
19 #include <linux/scatterlist.h>
20 #include <linux/pm.h>
21 #include <linux/pm_runtime.h>
22 
23 #include <linux/rtsx_usb.h>
24 #include <asm/unaligned.h>
25 
26 #if defined(CONFIG_LEDS_CLASS) || (defined(CONFIG_LEDS_CLASS_MODULE) && \
27 		defined(CONFIG_MMC_REALTEK_USB_MODULE))
28 #include <linux/leds.h>
29 #include <linux/workqueue.h>
30 #define RTSX_USB_USE_LEDS_CLASS
31 #endif
32 
33 struct rtsx_usb_sdmmc {
34 	struct platform_device	*pdev;
35 	struct rtsx_ucr	*ucr;
36 	struct mmc_host		*mmc;
37 	struct mmc_request	*mrq;
38 
39 	struct mutex		host_mutex;
40 
41 	u8			ssc_depth;
42 	unsigned int		clock;
43 	bool			vpclk;
44 	bool			double_clk;
45 	bool			host_removal;
46 	bool			card_exist;
47 	bool			initial_mode;
48 	bool			ddr_mode;
49 
50 	unsigned char		power_mode;
51 
52 #ifdef RTSX_USB_USE_LEDS_CLASS
53 	struct led_classdev	led;
54 	char			led_name[32];
55 	struct work_struct	led_work;
56 #endif
57 };
58 
59 static inline struct device *sdmmc_dev(struct rtsx_usb_sdmmc *host)
60 {
61 	return &(host->pdev->dev);
62 }
63 
64 static inline void sd_clear_error(struct rtsx_usb_sdmmc *host)
65 {
66 	struct rtsx_ucr *ucr = host->ucr;
67 	rtsx_usb_ep0_write_register(ucr, CARD_STOP,
68 				  SD_STOP | SD_CLR_ERR,
69 				  SD_STOP | SD_CLR_ERR);
70 
71 	rtsx_usb_clear_dma_err(ucr);
72 	rtsx_usb_clear_fsm_err(ucr);
73 }
74 
75 #ifdef DEBUG
76 static void sd_print_debug_regs(struct rtsx_usb_sdmmc *host)
77 {
78 	struct rtsx_ucr *ucr = host->ucr;
79 	u8 val = 0;
80 
81 	rtsx_usb_ep0_read_register(ucr, SD_STAT1, &val);
82 	dev_dbg(sdmmc_dev(host), "SD_STAT1: 0x%x\n", val);
83 	rtsx_usb_ep0_read_register(ucr, SD_STAT2, &val);
84 	dev_dbg(sdmmc_dev(host), "SD_STAT2: 0x%x\n", val);
85 	rtsx_usb_ep0_read_register(ucr, SD_BUS_STAT, &val);
86 	dev_dbg(sdmmc_dev(host), "SD_BUS_STAT: 0x%x\n", val);
87 }
88 #else
89 #define sd_print_debug_regs(host)
90 #endif /* DEBUG */
91 
92 static int sd_read_data(struct rtsx_usb_sdmmc *host, struct mmc_command *cmd,
93 	       u16 byte_cnt, u8 *buf, int buf_len, int timeout)
94 {
95 	struct rtsx_ucr *ucr = host->ucr;
96 	int err;
97 	u8 trans_mode;
98 
99 	if (!buf)
100 		buf_len = 0;
101 
102 	rtsx_usb_init_cmd(ucr);
103 	if (cmd != NULL) {
104 		dev_dbg(sdmmc_dev(host), "%s: SD/MMC CMD%d\n", __func__
105 				, cmd->opcode);
106 		if (cmd->opcode == MMC_SEND_TUNING_BLOCK)
107 			trans_mode = SD_TM_AUTO_TUNING;
108 		else
109 			trans_mode = SD_TM_NORMAL_READ;
110 
111 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
112 				SD_CMD0, 0xFF, (u8)(cmd->opcode) | 0x40);
113 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
114 				SD_CMD1, 0xFF, (u8)(cmd->arg >> 24));
115 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
116 				SD_CMD2, 0xFF, (u8)(cmd->arg >> 16));
117 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
118 				SD_CMD3, 0xFF, (u8)(cmd->arg >> 8));
119 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
120 				SD_CMD4, 0xFF, (u8)cmd->arg);
121 	} else {
122 		trans_mode = SD_TM_AUTO_READ_3;
123 	}
124 
125 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BYTE_CNT_L, 0xFF, (u8)byte_cnt);
126 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BYTE_CNT_H,
127 			0xFF, (u8)(byte_cnt >> 8));
128 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BLOCK_CNT_L, 0xFF, 1);
129 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BLOCK_CNT_H, 0xFF, 0);
130 
131 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CFG2, 0xFF,
132 			SD_CALCULATE_CRC7 | SD_CHECK_CRC16 |
133 			SD_NO_WAIT_BUSY_END | SD_CHECK_CRC7 | SD_RSP_LEN_6);
134 	if (trans_mode != SD_TM_AUTO_TUNING)
135 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
136 				CARD_DATA_SOURCE, 0x01, PINGPONG_BUFFER);
137 
138 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_TRANSFER,
139 			0xFF, trans_mode | SD_TRANSFER_START);
140 	rtsx_usb_add_cmd(ucr, CHECK_REG_CMD, SD_TRANSFER,
141 			SD_TRANSFER_END, SD_TRANSFER_END);
142 
143 	if (cmd != NULL) {
144 		rtsx_usb_add_cmd(ucr, READ_REG_CMD, SD_CMD1, 0, 0);
145 		rtsx_usb_add_cmd(ucr, READ_REG_CMD, SD_CMD2, 0, 0);
146 		rtsx_usb_add_cmd(ucr, READ_REG_CMD, SD_CMD3, 0, 0);
147 		rtsx_usb_add_cmd(ucr, READ_REG_CMD, SD_CMD4, 0, 0);
148 	}
149 
150 	err = rtsx_usb_send_cmd(ucr, MODE_CR, timeout);
151 	if (err) {
152 		dev_dbg(sdmmc_dev(host),
153 			"rtsx_usb_send_cmd failed (err = %d)\n", err);
154 		return err;
155 	}
156 
157 	err = rtsx_usb_get_rsp(ucr, !cmd ? 1 : 5, timeout);
158 	if (err || (ucr->rsp_buf[0] & SD_TRANSFER_ERR)) {
159 		sd_print_debug_regs(host);
160 
161 		if (!err) {
162 			dev_dbg(sdmmc_dev(host),
163 				"Transfer failed (SD_TRANSFER = %02x)\n",
164 				ucr->rsp_buf[0]);
165 			err = -EIO;
166 		} else {
167 			dev_dbg(sdmmc_dev(host),
168 				"rtsx_usb_get_rsp failed (err = %d)\n", err);
169 		}
170 
171 		return err;
172 	}
173 
174 	if (cmd != NULL) {
175 		cmd->resp[0] = get_unaligned_be32(ucr->rsp_buf + 1);
176 		dev_dbg(sdmmc_dev(host), "cmd->resp[0] = 0x%08x\n",
177 				cmd->resp[0]);
178 	}
179 
180 	if (buf && buf_len) {
181 		/* 2-byte aligned part */
182 		err = rtsx_usb_read_ppbuf(ucr, buf, byte_cnt - (byte_cnt % 2));
183 		if (err) {
184 			dev_dbg(sdmmc_dev(host),
185 				"rtsx_usb_read_ppbuf failed (err = %d)\n", err);
186 			return err;
187 		}
188 
189 		/* unaligned byte */
190 		if (byte_cnt % 2)
191 			return rtsx_usb_read_register(ucr,
192 					PPBUF_BASE2 + byte_cnt,
193 					buf + byte_cnt - 1);
194 	}
195 
196 	return 0;
197 }
198 
199 static int sd_write_data(struct rtsx_usb_sdmmc *host, struct mmc_command *cmd,
200 		u16 byte_cnt, u8 *buf, int buf_len, int timeout)
201 {
202 	struct rtsx_ucr *ucr = host->ucr;
203 	int err;
204 	u8 trans_mode;
205 
206 	if (!buf)
207 		buf_len = 0;
208 
209 	if (buf && buf_len) {
210 		err = rtsx_usb_write_ppbuf(ucr, buf, buf_len);
211 		if (err) {
212 			dev_dbg(sdmmc_dev(host),
213 				"rtsx_usb_write_ppbuf failed (err = %d)\n",
214 				err);
215 			return err;
216 		}
217 	}
218 
219 	trans_mode = (cmd != NULL) ? SD_TM_AUTO_WRITE_2 : SD_TM_AUTO_WRITE_3;
220 	rtsx_usb_init_cmd(ucr);
221 
222 	if (cmd != NULL) {
223 		dev_dbg(sdmmc_dev(host), "%s: SD/MMC CMD%d\n", __func__,
224 				cmd->opcode);
225 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
226 				SD_CMD0, 0xFF, (u8)(cmd->opcode) | 0x40);
227 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
228 				SD_CMD1, 0xFF, (u8)(cmd->arg >> 24));
229 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
230 				SD_CMD2, 0xFF, (u8)(cmd->arg >> 16));
231 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
232 				SD_CMD3, 0xFF, (u8)(cmd->arg >> 8));
233 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
234 				SD_CMD4, 0xFF, (u8)cmd->arg);
235 	}
236 
237 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BYTE_CNT_L, 0xFF, (u8)byte_cnt);
238 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BYTE_CNT_H,
239 			0xFF, (u8)(byte_cnt >> 8));
240 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BLOCK_CNT_L, 0xFF, 1);
241 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BLOCK_CNT_H, 0xFF, 0);
242 
243 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CFG2, 0xFF,
244 		SD_CALCULATE_CRC7 | SD_CHECK_CRC16 |
245 		SD_NO_WAIT_BUSY_END | SD_CHECK_CRC7 | SD_RSP_LEN_6);
246 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
247 			CARD_DATA_SOURCE, 0x01, PINGPONG_BUFFER);
248 
249 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_TRANSFER, 0xFF,
250 			trans_mode | SD_TRANSFER_START);
251 	rtsx_usb_add_cmd(ucr, CHECK_REG_CMD, SD_TRANSFER,
252 			SD_TRANSFER_END, SD_TRANSFER_END);
253 
254 	if (cmd != NULL) {
255 		rtsx_usb_add_cmd(ucr, READ_REG_CMD, SD_CMD1, 0, 0);
256 		rtsx_usb_add_cmd(ucr, READ_REG_CMD, SD_CMD2, 0, 0);
257 		rtsx_usb_add_cmd(ucr, READ_REG_CMD, SD_CMD3, 0, 0);
258 		rtsx_usb_add_cmd(ucr, READ_REG_CMD, SD_CMD4, 0, 0);
259 	}
260 
261 	err = rtsx_usb_send_cmd(ucr, MODE_CR, timeout);
262 	if (err) {
263 		dev_dbg(sdmmc_dev(host),
264 			"rtsx_usb_send_cmd failed (err = %d)\n", err);
265 		return err;
266 	}
267 
268 	err = rtsx_usb_get_rsp(ucr, !cmd ? 1 : 5, timeout);
269 	if (err) {
270 		sd_print_debug_regs(host);
271 		dev_dbg(sdmmc_dev(host),
272 			"rtsx_usb_get_rsp failed (err = %d)\n", err);
273 		return err;
274 	}
275 
276 	if (cmd != NULL) {
277 		cmd->resp[0] = get_unaligned_be32(ucr->rsp_buf + 1);
278 		dev_dbg(sdmmc_dev(host), "cmd->resp[0] = 0x%08x\n",
279 				cmd->resp[0]);
280 	}
281 
282 	return 0;
283 }
284 
285 static void sd_send_cmd_get_rsp(struct rtsx_usb_sdmmc *host,
286 		struct mmc_command *cmd)
287 {
288 	struct rtsx_ucr *ucr = host->ucr;
289 	u8 cmd_idx = (u8)cmd->opcode;
290 	u32 arg = cmd->arg;
291 	int err = 0;
292 	int timeout = 100;
293 	int i;
294 	u8 *ptr;
295 	int stat_idx = 0;
296 	int len = 2;
297 	u8 rsp_type;
298 
299 	dev_dbg(sdmmc_dev(host), "%s: SD/MMC CMD %d, arg = 0x%08x\n",
300 			__func__, cmd_idx, arg);
301 
302 	/* Response type:
303 	 * R0
304 	 * R1, R5, R6, R7
305 	 * R1b
306 	 * R2
307 	 * R3, R4
308 	 */
309 	switch (mmc_resp_type(cmd)) {
310 	case MMC_RSP_NONE:
311 		rsp_type = SD_RSP_TYPE_R0;
312 		break;
313 	case MMC_RSP_R1:
314 		rsp_type = SD_RSP_TYPE_R1;
315 		break;
316 	case MMC_RSP_R1_NO_CRC:
317 		rsp_type = SD_RSP_TYPE_R1 | SD_NO_CHECK_CRC7;
318 		break;
319 	case MMC_RSP_R1B:
320 		rsp_type = SD_RSP_TYPE_R1b;
321 		break;
322 	case MMC_RSP_R2:
323 		rsp_type = SD_RSP_TYPE_R2;
324 		break;
325 	case MMC_RSP_R3:
326 		rsp_type = SD_RSP_TYPE_R3;
327 		break;
328 	default:
329 		dev_dbg(sdmmc_dev(host), "cmd->flag is not valid\n");
330 		err = -EINVAL;
331 		goto out;
332 	}
333 
334 	if (rsp_type == SD_RSP_TYPE_R1b)
335 		timeout = cmd->busy_timeout ? cmd->busy_timeout : 3000;
336 
337 	if (cmd->opcode == SD_SWITCH_VOLTAGE) {
338 		err = rtsx_usb_write_register(ucr, SD_BUS_STAT,
339 				SD_CLK_TOGGLE_EN | SD_CLK_FORCE_STOP,
340 				SD_CLK_TOGGLE_EN);
341 		if (err)
342 			goto out;
343 	}
344 
345 	rtsx_usb_init_cmd(ucr);
346 
347 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CMD0, 0xFF, 0x40 | cmd_idx);
348 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CMD1, 0xFF, (u8)(arg >> 24));
349 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CMD2, 0xFF, (u8)(arg >> 16));
350 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CMD3, 0xFF, (u8)(arg >> 8));
351 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CMD4, 0xFF, (u8)arg);
352 
353 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CFG2, 0xFF, rsp_type);
354 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_DATA_SOURCE,
355 			0x01, PINGPONG_BUFFER);
356 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_TRANSFER,
357 			0xFF, SD_TM_CMD_RSP | SD_TRANSFER_START);
358 	rtsx_usb_add_cmd(ucr, CHECK_REG_CMD, SD_TRANSFER,
359 		     SD_TRANSFER_END | SD_STAT_IDLE,
360 		     SD_TRANSFER_END | SD_STAT_IDLE);
361 
362 	if (rsp_type == SD_RSP_TYPE_R2) {
363 		/* Read data from ping-pong buffer */
364 		for (i = PPBUF_BASE2; i < PPBUF_BASE2 + 16; i++)
365 			rtsx_usb_add_cmd(ucr, READ_REG_CMD, (u16)i, 0, 0);
366 		stat_idx = 16;
367 	} else if (rsp_type != SD_RSP_TYPE_R0) {
368 		/* Read data from SD_CMDx registers */
369 		for (i = SD_CMD0; i <= SD_CMD4; i++)
370 			rtsx_usb_add_cmd(ucr, READ_REG_CMD, (u16)i, 0, 0);
371 		stat_idx = 5;
372 	}
373 	len += stat_idx;
374 
375 	rtsx_usb_add_cmd(ucr, READ_REG_CMD, SD_STAT1, 0, 0);
376 
377 	err = rtsx_usb_send_cmd(ucr, MODE_CR, 100);
378 	if (err) {
379 		dev_dbg(sdmmc_dev(host),
380 			"rtsx_usb_send_cmd error (err = %d)\n", err);
381 		goto out;
382 	}
383 
384 	err = rtsx_usb_get_rsp(ucr, len, timeout);
385 	if (err || (ucr->rsp_buf[0] & SD_TRANSFER_ERR)) {
386 		sd_print_debug_regs(host);
387 		sd_clear_error(host);
388 
389 		if (!err) {
390 			dev_dbg(sdmmc_dev(host),
391 				"Transfer failed (SD_TRANSFER = %02x)\n",
392 					ucr->rsp_buf[0]);
393 			err = -EIO;
394 		} else {
395 			dev_dbg(sdmmc_dev(host),
396 				"rtsx_usb_get_rsp failed (err = %d)\n", err);
397 		}
398 
399 		goto out;
400 	}
401 
402 	if (rsp_type == SD_RSP_TYPE_R0) {
403 		err = 0;
404 		goto out;
405 	}
406 
407 	/* Skip result of CHECK_REG_CMD */
408 	ptr = ucr->rsp_buf + 1;
409 
410 	/* Check (Start,Transmission) bit of Response */
411 	if ((ptr[0] & 0xC0) != 0) {
412 		err = -EILSEQ;
413 		dev_dbg(sdmmc_dev(host), "Invalid response bit\n");
414 		goto out;
415 	}
416 
417 	/* Check CRC7 */
418 	if (!(rsp_type & SD_NO_CHECK_CRC7)) {
419 		if (ptr[stat_idx] & SD_CRC7_ERR) {
420 			err = -EILSEQ;
421 			dev_dbg(sdmmc_dev(host), "CRC7 error\n");
422 			goto out;
423 		}
424 	}
425 
426 	if (rsp_type == SD_RSP_TYPE_R2) {
427 		/*
428 		 * The controller offloads the last byte {CRC-7, end bit 1'b1}
429 		 * of response type R2. Assign dummy CRC, 0, and end bit to the
430 		 * byte(ptr[16], goes into the LSB of resp[3] later).
431 		 */
432 		ptr[16] = 1;
433 
434 		for (i = 0; i < 4; i++) {
435 			cmd->resp[i] = get_unaligned_be32(ptr + 1 + i * 4);
436 			dev_dbg(sdmmc_dev(host), "cmd->resp[%d] = 0x%08x\n",
437 					i, cmd->resp[i]);
438 		}
439 	} else {
440 		cmd->resp[0] = get_unaligned_be32(ptr + 1);
441 		dev_dbg(sdmmc_dev(host), "cmd->resp[0] = 0x%08x\n",
442 				cmd->resp[0]);
443 	}
444 
445 out:
446 	cmd->error = err;
447 }
448 
449 static int sd_rw_multi(struct rtsx_usb_sdmmc *host, struct mmc_request *mrq)
450 {
451 	struct rtsx_ucr *ucr = host->ucr;
452 	struct mmc_data *data = mrq->data;
453 	int read = (data->flags & MMC_DATA_READ) ? 1 : 0;
454 	u8 cfg2, trans_mode;
455 	int err;
456 	u8 flag;
457 	size_t data_len = data->blksz * data->blocks;
458 	unsigned int pipe;
459 
460 	if (read) {
461 		dev_dbg(sdmmc_dev(host), "%s: read %zu bytes\n",
462 				__func__, data_len);
463 		cfg2 = SD_CALCULATE_CRC7 | SD_CHECK_CRC16 |
464 			SD_NO_WAIT_BUSY_END | SD_CHECK_CRC7 | SD_RSP_LEN_0;
465 		trans_mode = SD_TM_AUTO_READ_3;
466 	} else {
467 		dev_dbg(sdmmc_dev(host), "%s: write %zu bytes\n",
468 				__func__, data_len);
469 		cfg2 = SD_NO_CALCULATE_CRC7 | SD_CHECK_CRC16 |
470 			SD_NO_WAIT_BUSY_END | SD_NO_CHECK_CRC7 | SD_RSP_LEN_0;
471 		trans_mode = SD_TM_AUTO_WRITE_3;
472 	}
473 
474 	rtsx_usb_init_cmd(ucr);
475 
476 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BYTE_CNT_L, 0xFF, 0x00);
477 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BYTE_CNT_H, 0xFF, 0x02);
478 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BLOCK_CNT_L,
479 			0xFF, (u8)data->blocks);
480 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BLOCK_CNT_H,
481 			0xFF, (u8)(data->blocks >> 8));
482 
483 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_DATA_SOURCE,
484 			0x01, RING_BUFFER);
485 
486 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, MC_DMA_TC3,
487 			0xFF, (u8)(data_len >> 24));
488 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, MC_DMA_TC2,
489 			0xFF, (u8)(data_len >> 16));
490 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, MC_DMA_TC1,
491 			0xFF, (u8)(data_len >> 8));
492 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, MC_DMA_TC0,
493 			0xFF, (u8)data_len);
494 	if (read) {
495 		flag = MODE_CDIR;
496 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, MC_DMA_CTL,
497 				0x03 | DMA_PACK_SIZE_MASK,
498 				DMA_DIR_FROM_CARD | DMA_EN | DMA_512);
499 	} else {
500 		flag = MODE_CDOR;
501 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, MC_DMA_CTL,
502 				0x03 | DMA_PACK_SIZE_MASK,
503 				DMA_DIR_TO_CARD | DMA_EN | DMA_512);
504 	}
505 
506 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CFG2, 0xFF, cfg2);
507 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_TRANSFER, 0xFF,
508 			trans_mode | SD_TRANSFER_START);
509 	rtsx_usb_add_cmd(ucr, CHECK_REG_CMD, SD_TRANSFER,
510 			SD_TRANSFER_END, SD_TRANSFER_END);
511 
512 	err = rtsx_usb_send_cmd(ucr, flag, 100);
513 	if (err)
514 		return err;
515 
516 	if (read)
517 		pipe = usb_rcvbulkpipe(ucr->pusb_dev, EP_BULK_IN);
518 	else
519 		pipe = usb_sndbulkpipe(ucr->pusb_dev, EP_BULK_OUT);
520 
521 	err = rtsx_usb_transfer_data(ucr, pipe, data->sg, data_len,
522 			data->sg_len,  NULL, 10000);
523 	if (err) {
524 		dev_dbg(sdmmc_dev(host), "rtsx_usb_transfer_data error %d\n"
525 				, err);
526 		sd_clear_error(host);
527 		return err;
528 	}
529 
530 	return rtsx_usb_get_rsp(ucr, 1, 2000);
531 }
532 
533 static inline void sd_enable_initial_mode(struct rtsx_usb_sdmmc *host)
534 {
535 	rtsx_usb_write_register(host->ucr, SD_CFG1,
536 			SD_CLK_DIVIDE_MASK, SD_CLK_DIVIDE_128);
537 }
538 
539 static inline void sd_disable_initial_mode(struct rtsx_usb_sdmmc *host)
540 {
541 	rtsx_usb_write_register(host->ucr, SD_CFG1,
542 			SD_CLK_DIVIDE_MASK, SD_CLK_DIVIDE_0);
543 }
544 
545 static void sd_normal_rw(struct rtsx_usb_sdmmc *host,
546 		struct mmc_request *mrq)
547 {
548 	struct mmc_command *cmd = mrq->cmd;
549 	struct mmc_data *data = mrq->data;
550 	u8 *buf;
551 
552 	buf = kzalloc(data->blksz, GFP_NOIO);
553 	if (!buf) {
554 		cmd->error = -ENOMEM;
555 		return;
556 	}
557 
558 	if (data->flags & MMC_DATA_READ) {
559 		if (host->initial_mode)
560 			sd_disable_initial_mode(host);
561 
562 		cmd->error = sd_read_data(host, cmd, (u16)data->blksz, buf,
563 				data->blksz, 200);
564 
565 		if (host->initial_mode)
566 			sd_enable_initial_mode(host);
567 
568 		sg_copy_from_buffer(data->sg, data->sg_len, buf, data->blksz);
569 	} else {
570 		sg_copy_to_buffer(data->sg, data->sg_len, buf, data->blksz);
571 
572 		cmd->error = sd_write_data(host, cmd, (u16)data->blksz, buf,
573 				data->blksz, 200);
574 	}
575 
576 	kfree(buf);
577 }
578 
579 static int sd_change_phase(struct rtsx_usb_sdmmc *host, u8 sample_point, int tx)
580 {
581 	struct rtsx_ucr *ucr = host->ucr;
582 	int err;
583 
584 	dev_dbg(sdmmc_dev(host), "%s: %s sample_point = %d\n",
585 			__func__, tx ? "TX" : "RX", sample_point);
586 
587 	rtsx_usb_init_cmd(ucr);
588 
589 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CLK_DIV, CLK_CHANGE, CLK_CHANGE);
590 
591 	if (tx)
592 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_VPCLK0_CTL,
593 				0x0F, sample_point);
594 	else
595 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_VPCLK1_CTL,
596 				0x0F, sample_point);
597 
598 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_VPCLK0_CTL, PHASE_NOT_RESET, 0);
599 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_VPCLK0_CTL,
600 			PHASE_NOT_RESET, PHASE_NOT_RESET);
601 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CLK_DIV, CLK_CHANGE, 0);
602 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CFG1, SD_ASYNC_FIFO_RST, 0);
603 
604 	err = rtsx_usb_send_cmd(ucr, MODE_C, 100);
605 	if (err)
606 		return err;
607 
608 	return 0;
609 }
610 
611 static inline u32 get_phase_point(u32 phase_map, unsigned int idx)
612 {
613 	idx &= MAX_PHASE;
614 	return phase_map & (1 << idx);
615 }
616 
617 static int get_phase_len(u32 phase_map, unsigned int idx)
618 {
619 	int i;
620 
621 	for (i = 0; i < MAX_PHASE + 1; i++) {
622 		if (get_phase_point(phase_map, idx + i) == 0)
623 			return i;
624 	}
625 	return MAX_PHASE + 1;
626 }
627 
628 static u8 sd_search_final_phase(struct rtsx_usb_sdmmc *host, u32 phase_map)
629 {
630 	int start = 0, len = 0;
631 	int start_final = 0, len_final = 0;
632 	u8 final_phase = 0xFF;
633 
634 	if (phase_map == 0) {
635 		dev_dbg(sdmmc_dev(host), "Phase: [map:%x]\n", phase_map);
636 		return final_phase;
637 	}
638 
639 	while (start < MAX_PHASE + 1) {
640 		len = get_phase_len(phase_map, start);
641 		if (len_final < len) {
642 			start_final = start;
643 			len_final = len;
644 		}
645 		start += len ? len : 1;
646 	}
647 
648 	final_phase = (start_final + len_final / 2) & MAX_PHASE;
649 	dev_dbg(sdmmc_dev(host), "Phase: [map:%x] [maxlen:%d] [final:%d]\n",
650 		phase_map, len_final, final_phase);
651 
652 	return final_phase;
653 }
654 
655 static void sd_wait_data_idle(struct rtsx_usb_sdmmc *host)
656 {
657 	int err, i;
658 	u8 val = 0;
659 
660 	for (i = 0; i < 100; i++) {
661 		err = rtsx_usb_ep0_read_register(host->ucr,
662 				SD_DATA_STATE, &val);
663 		if (val & SD_DATA_IDLE)
664 			return;
665 
666 		usleep_range(100, 1000);
667 	}
668 }
669 
670 static int sd_tuning_rx_cmd(struct rtsx_usb_sdmmc *host,
671 		u8 opcode, u8 sample_point)
672 {
673 	int err;
674 	struct mmc_command cmd = {};
675 
676 	err = sd_change_phase(host, sample_point, 0);
677 	if (err)
678 		return err;
679 
680 	cmd.opcode = MMC_SEND_TUNING_BLOCK;
681 	err = sd_read_data(host, &cmd, 0x40, NULL, 0, 100);
682 	if (err) {
683 		/* Wait till SD DATA IDLE */
684 		sd_wait_data_idle(host);
685 		sd_clear_error(host);
686 		return err;
687 	}
688 
689 	return 0;
690 }
691 
692 static void sd_tuning_phase(struct rtsx_usb_sdmmc *host,
693 		u8 opcode, u16 *phase_map)
694 {
695 	int err, i;
696 	u16 raw_phase_map = 0;
697 
698 	for (i = MAX_PHASE; i >= 0; i--) {
699 		err = sd_tuning_rx_cmd(host, opcode, (u8)i);
700 		if (!err)
701 			raw_phase_map |= 1 << i;
702 	}
703 
704 	if (phase_map)
705 		*phase_map = raw_phase_map;
706 }
707 
708 static int sd_tuning_rx(struct rtsx_usb_sdmmc *host, u8 opcode)
709 {
710 	int err, i;
711 	u16 raw_phase_map[RX_TUNING_CNT] = {0}, phase_map;
712 	u8 final_phase;
713 
714 	/* setting fixed default TX phase */
715 	err = sd_change_phase(host, 0x01, 1);
716 	if (err) {
717 		dev_dbg(sdmmc_dev(host), "TX phase setting failed\n");
718 		return err;
719 	}
720 
721 	/* tuning RX phase */
722 	for (i = 0; i < RX_TUNING_CNT; i++) {
723 		sd_tuning_phase(host, opcode, &(raw_phase_map[i]));
724 
725 		if (raw_phase_map[i] == 0)
726 			break;
727 	}
728 
729 	phase_map = 0xFFFF;
730 	for (i = 0; i < RX_TUNING_CNT; i++) {
731 		dev_dbg(sdmmc_dev(host), "RX raw_phase_map[%d] = 0x%04x\n",
732 				i, raw_phase_map[i]);
733 		phase_map &= raw_phase_map[i];
734 	}
735 	dev_dbg(sdmmc_dev(host), "RX phase_map = 0x%04x\n", phase_map);
736 
737 	if (phase_map) {
738 		final_phase = sd_search_final_phase(host, phase_map);
739 		if (final_phase == 0xFF)
740 			return -EINVAL;
741 
742 		err = sd_change_phase(host, final_phase, 0);
743 		if (err)
744 			return err;
745 	} else {
746 		return -EINVAL;
747 	}
748 
749 	return 0;
750 }
751 
752 static int sdmmc_get_ro(struct mmc_host *mmc)
753 {
754 	struct rtsx_usb_sdmmc *host = mmc_priv(mmc);
755 	struct rtsx_ucr *ucr = host->ucr;
756 	int err;
757 	u16 val;
758 
759 	if (host->host_removal)
760 		return -ENOMEDIUM;
761 
762 	mutex_lock(&ucr->dev_mutex);
763 
764 	/* Check SD card detect */
765 	err = rtsx_usb_get_card_status(ucr, &val);
766 
767 	mutex_unlock(&ucr->dev_mutex);
768 
769 
770 	/* Treat failed detection as non-ro */
771 	if (err)
772 		return 0;
773 
774 	if (val & SD_WP)
775 		return 1;
776 
777 	return 0;
778 }
779 
780 static int sdmmc_get_cd(struct mmc_host *mmc)
781 {
782 	struct rtsx_usb_sdmmc *host = mmc_priv(mmc);
783 	struct rtsx_ucr *ucr = host->ucr;
784 	int err;
785 	u16 val;
786 
787 	if (host->host_removal)
788 		return -ENOMEDIUM;
789 
790 	mutex_lock(&ucr->dev_mutex);
791 
792 	/* Check SD card detect */
793 	err = rtsx_usb_get_card_status(ucr, &val);
794 
795 	mutex_unlock(&ucr->dev_mutex);
796 
797 	/* Treat failed detection as non-exist */
798 	if (err)
799 		goto no_card;
800 
801 	if (val & SD_CD) {
802 		host->card_exist = true;
803 		return 1;
804 	}
805 
806 no_card:
807 	host->card_exist = false;
808 	return 0;
809 }
810 
811 static void sdmmc_request(struct mmc_host *mmc, struct mmc_request *mrq)
812 {
813 	struct rtsx_usb_sdmmc *host = mmc_priv(mmc);
814 	struct rtsx_ucr *ucr = host->ucr;
815 	struct mmc_command *cmd = mrq->cmd;
816 	struct mmc_data *data = mrq->data;
817 	unsigned int data_size = 0;
818 
819 	dev_dbg(sdmmc_dev(host), "%s\n", __func__);
820 
821 	if (host->host_removal) {
822 		cmd->error = -ENOMEDIUM;
823 		goto finish;
824 	}
825 
826 	if ((!host->card_exist)) {
827 		cmd->error = -ENOMEDIUM;
828 		goto finish_detect_card;
829 	}
830 
831 	mutex_lock(&ucr->dev_mutex);
832 
833 	mutex_lock(&host->host_mutex);
834 	host->mrq = mrq;
835 	mutex_unlock(&host->host_mutex);
836 
837 	if (mrq->data)
838 		data_size = data->blocks * data->blksz;
839 
840 	if (!data_size) {
841 		sd_send_cmd_get_rsp(host, cmd);
842 	} else if ((!(data_size % 512) && cmd->opcode != MMC_SEND_EXT_CSD) ||
843 		   mmc_op_multi(cmd->opcode)) {
844 		sd_send_cmd_get_rsp(host, cmd);
845 
846 		if (!cmd->error) {
847 			sd_rw_multi(host, mrq);
848 
849 			if (mmc_op_multi(cmd->opcode) && mrq->stop) {
850 				sd_send_cmd_get_rsp(host, mrq->stop);
851 				rtsx_usb_write_register(ucr, MC_FIFO_CTL,
852 						FIFO_FLUSH, FIFO_FLUSH);
853 			}
854 		}
855 	} else {
856 		sd_normal_rw(host, mrq);
857 	}
858 
859 	if (mrq->data) {
860 		if (cmd->error || data->error)
861 			data->bytes_xfered = 0;
862 		else
863 			data->bytes_xfered = data->blocks * data->blksz;
864 	}
865 
866 	mutex_unlock(&ucr->dev_mutex);
867 
868 finish_detect_card:
869 	if (cmd->error) {
870 		/*
871 		 * detect card when fail to update card existence state and
872 		 * speed up card removal when retry
873 		 */
874 		sdmmc_get_cd(mmc);
875 		dev_dbg(sdmmc_dev(host), "cmd->error = %d\n", cmd->error);
876 	}
877 
878 finish:
879 	mutex_lock(&host->host_mutex);
880 	host->mrq = NULL;
881 	mutex_unlock(&host->host_mutex);
882 
883 	mmc_request_done(mmc, mrq);
884 }
885 
886 static int sd_set_bus_width(struct rtsx_usb_sdmmc *host,
887 		unsigned char bus_width)
888 {
889 	int err = 0;
890 	static const u8 width[] = {
891 		[MMC_BUS_WIDTH_1] = SD_BUS_WIDTH_1BIT,
892 		[MMC_BUS_WIDTH_4] = SD_BUS_WIDTH_4BIT,
893 		[MMC_BUS_WIDTH_8] = SD_BUS_WIDTH_8BIT,
894 	};
895 
896 	if (bus_width <= MMC_BUS_WIDTH_8)
897 		err = rtsx_usb_write_register(host->ucr, SD_CFG1,
898 				0x03, width[bus_width]);
899 
900 	return err;
901 }
902 
903 static int sd_pull_ctl_disable_lqfp48(struct rtsx_ucr *ucr)
904 {
905 	rtsx_usb_init_cmd(ucr);
906 
907 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL1, 0xFF, 0x55);
908 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL2, 0xFF, 0x55);
909 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL3, 0xFF, 0x95);
910 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL4, 0xFF, 0x55);
911 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL5, 0xFF, 0x55);
912 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL6, 0xFF, 0xA5);
913 
914 	return rtsx_usb_send_cmd(ucr, MODE_C, 100);
915 }
916 
917 static int sd_pull_ctl_disable_qfn24(struct rtsx_ucr *ucr)
918 {
919 	rtsx_usb_init_cmd(ucr);
920 
921 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL1, 0xFF, 0x65);
922 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL2, 0xFF, 0x55);
923 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL3, 0xFF, 0x95);
924 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL4, 0xFF, 0x55);
925 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL5, 0xFF, 0x56);
926 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL6, 0xFF, 0x59);
927 
928 	return rtsx_usb_send_cmd(ucr, MODE_C, 100);
929 }
930 
931 static int sd_pull_ctl_enable_lqfp48(struct rtsx_ucr *ucr)
932 {
933 	rtsx_usb_init_cmd(ucr);
934 
935 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL1, 0xFF, 0xAA);
936 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL2, 0xFF, 0xAA);
937 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL3, 0xFF, 0xA9);
938 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL4, 0xFF, 0x55);
939 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL5, 0xFF, 0x55);
940 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL6, 0xFF, 0xA5);
941 
942 	return rtsx_usb_send_cmd(ucr, MODE_C, 100);
943 }
944 
945 static int sd_pull_ctl_enable_qfn24(struct rtsx_ucr *ucr)
946 {
947 	rtsx_usb_init_cmd(ucr);
948 
949 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL1, 0xFF, 0xA5);
950 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL2, 0xFF, 0x9A);
951 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL3, 0xFF, 0xA5);
952 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL4, 0xFF, 0x9A);
953 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL5, 0xFF, 0x65);
954 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PULL_CTL6, 0xFF, 0x5A);
955 
956 	return rtsx_usb_send_cmd(ucr, MODE_C, 100);
957 }
958 
959 static int sd_power_on(struct rtsx_usb_sdmmc *host)
960 {
961 	struct rtsx_ucr *ucr = host->ucr;
962 	int err;
963 
964 	dev_dbg(sdmmc_dev(host), "%s\n", __func__);
965 	rtsx_usb_init_cmd(ucr);
966 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_SELECT, 0x07, SD_MOD_SEL);
967 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_SHARE_MODE,
968 			CARD_SHARE_MASK, CARD_SHARE_SD);
969 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_CLK_EN,
970 			SD_CLK_EN, SD_CLK_EN);
971 	err = rtsx_usb_send_cmd(ucr, MODE_C, 100);
972 	if (err)
973 		return err;
974 
975 	if (CHECK_PKG(ucr, LQFP48))
976 		err = sd_pull_ctl_enable_lqfp48(ucr);
977 	else
978 		err = sd_pull_ctl_enable_qfn24(ucr);
979 	if (err)
980 		return err;
981 
982 	err = rtsx_usb_write_register(ucr, CARD_PWR_CTL,
983 			POWER_MASK, PARTIAL_POWER_ON);
984 	if (err)
985 		return err;
986 
987 	usleep_range(800, 1000);
988 
989 	rtsx_usb_init_cmd(ucr);
990 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PWR_CTL,
991 			POWER_MASK|LDO3318_PWR_MASK, POWER_ON|LDO_ON);
992 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_OE,
993 			SD_OUTPUT_EN, SD_OUTPUT_EN);
994 
995 	return rtsx_usb_send_cmd(ucr, MODE_C, 100);
996 }
997 
998 static int sd_power_off(struct rtsx_usb_sdmmc *host)
999 {
1000 	struct rtsx_ucr *ucr = host->ucr;
1001 	int err;
1002 
1003 	dev_dbg(sdmmc_dev(host), "%s\n", __func__);
1004 	rtsx_usb_init_cmd(ucr);
1005 
1006 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_CLK_EN, SD_CLK_EN, 0);
1007 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_OE, SD_OUTPUT_EN, 0);
1008 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PWR_CTL,
1009 			POWER_MASK, POWER_OFF);
1010 	rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_PWR_CTL,
1011 			POWER_MASK|LDO3318_PWR_MASK, POWER_OFF|LDO_SUSPEND);
1012 
1013 	err = rtsx_usb_send_cmd(ucr, MODE_C, 100);
1014 	if (err)
1015 		return err;
1016 
1017 	if (CHECK_PKG(ucr, LQFP48))
1018 			return sd_pull_ctl_disable_lqfp48(ucr);
1019 	return sd_pull_ctl_disable_qfn24(ucr);
1020 }
1021 
1022 static int sd_set_power_mode(struct rtsx_usb_sdmmc *host,
1023 		unsigned char power_mode)
1024 {
1025 	int err;
1026 
1027 	if (power_mode != MMC_POWER_OFF)
1028 		power_mode = MMC_POWER_ON;
1029 
1030 	if (power_mode == host->power_mode)
1031 		return 0;
1032 
1033 	if (power_mode == MMC_POWER_OFF) {
1034 		err = sd_power_off(host);
1035 		pm_runtime_put_noidle(sdmmc_dev(host));
1036 	} else {
1037 		pm_runtime_get_noresume(sdmmc_dev(host));
1038 		err = sd_power_on(host);
1039 	}
1040 
1041 	if (!err)
1042 		host->power_mode = power_mode;
1043 
1044 	return err;
1045 }
1046 
1047 static int sd_set_timing(struct rtsx_usb_sdmmc *host,
1048 		unsigned char timing, bool *ddr_mode)
1049 {
1050 	struct rtsx_ucr *ucr = host->ucr;
1051 	int err;
1052 
1053 	*ddr_mode = false;
1054 
1055 	rtsx_usb_init_cmd(ucr);
1056 
1057 	switch (timing) {
1058 	case MMC_TIMING_UHS_SDR104:
1059 	case MMC_TIMING_UHS_SDR50:
1060 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CFG1,
1061 				0x0C | SD_ASYNC_FIFO_RST,
1062 				SD_30_MODE | SD_ASYNC_FIFO_RST);
1063 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_CLK_SOURCE, 0xFF,
1064 				CRC_VAR_CLK0 | SD30_FIX_CLK | SAMPLE_VAR_CLK1);
1065 		break;
1066 
1067 	case MMC_TIMING_UHS_DDR50:
1068 		*ddr_mode = true;
1069 
1070 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CFG1,
1071 				0x0C | SD_ASYNC_FIFO_RST,
1072 				SD_DDR_MODE | SD_ASYNC_FIFO_RST);
1073 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_CLK_SOURCE, 0xFF,
1074 				CRC_VAR_CLK0 | SD30_FIX_CLK | SAMPLE_VAR_CLK1);
1075 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_PUSH_POINT_CTL,
1076 				DDR_VAR_TX_CMD_DAT, DDR_VAR_TX_CMD_DAT);
1077 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_SAMPLE_POINT_CTL,
1078 				DDR_VAR_RX_DAT | DDR_VAR_RX_CMD,
1079 				DDR_VAR_RX_DAT | DDR_VAR_RX_CMD);
1080 		break;
1081 
1082 	case MMC_TIMING_MMC_HS:
1083 	case MMC_TIMING_SD_HS:
1084 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_CFG1,
1085 				0x0C, SD_20_MODE);
1086 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_CLK_SOURCE, 0xFF,
1087 				CRC_FIX_CLK | SD30_VAR_CLK0 | SAMPLE_VAR_CLK1);
1088 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_PUSH_POINT_CTL,
1089 				SD20_TX_SEL_MASK, SD20_TX_14_AHEAD);
1090 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_SAMPLE_POINT_CTL,
1091 				SD20_RX_SEL_MASK, SD20_RX_14_DELAY);
1092 		break;
1093 
1094 	default:
1095 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
1096 				SD_CFG1, 0x0C, SD_20_MODE);
1097 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, CARD_CLK_SOURCE, 0xFF,
1098 				CRC_FIX_CLK | SD30_VAR_CLK0 | SAMPLE_VAR_CLK1);
1099 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD,
1100 				SD_PUSH_POINT_CTL, 0xFF, 0);
1101 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_SAMPLE_POINT_CTL,
1102 				SD20_RX_SEL_MASK, SD20_RX_POS_EDGE);
1103 		break;
1104 	}
1105 
1106 	err = rtsx_usb_send_cmd(ucr, MODE_C, 100);
1107 
1108 	return err;
1109 }
1110 
1111 static void sdmmc_set_ios(struct mmc_host *mmc, struct mmc_ios *ios)
1112 {
1113 	struct rtsx_usb_sdmmc *host = mmc_priv(mmc);
1114 	struct rtsx_ucr *ucr = host->ucr;
1115 
1116 	dev_dbg(sdmmc_dev(host), "%s\n", __func__);
1117 	mutex_lock(&ucr->dev_mutex);
1118 
1119 	sd_set_power_mode(host, ios->power_mode);
1120 	sd_set_bus_width(host, ios->bus_width);
1121 	sd_set_timing(host, ios->timing, &host->ddr_mode);
1122 
1123 	host->vpclk = false;
1124 	host->double_clk = true;
1125 
1126 	switch (ios->timing) {
1127 	case MMC_TIMING_UHS_SDR104:
1128 	case MMC_TIMING_UHS_SDR50:
1129 		host->ssc_depth = SSC_DEPTH_2M;
1130 		host->vpclk = true;
1131 		host->double_clk = false;
1132 		break;
1133 	case MMC_TIMING_UHS_DDR50:
1134 	case MMC_TIMING_UHS_SDR25:
1135 		host->ssc_depth = SSC_DEPTH_1M;
1136 		break;
1137 	default:
1138 		host->ssc_depth = SSC_DEPTH_512K;
1139 		break;
1140 	}
1141 
1142 	host->initial_mode = (ios->clock <= 1000000) ? true : false;
1143 	host->clock = ios->clock;
1144 
1145 	rtsx_usb_switch_clock(host->ucr, host->clock, host->ssc_depth,
1146 			host->initial_mode, host->double_clk, host->vpclk);
1147 
1148 	mutex_unlock(&ucr->dev_mutex);
1149 	dev_dbg(sdmmc_dev(host), "%s end\n", __func__);
1150 }
1151 
1152 static int sdmmc_switch_voltage(struct mmc_host *mmc, struct mmc_ios *ios)
1153 {
1154 	struct rtsx_usb_sdmmc *host = mmc_priv(mmc);
1155 	struct rtsx_ucr *ucr = host->ucr;
1156 	int err = 0;
1157 
1158 	dev_dbg(sdmmc_dev(host), "%s: signal_voltage = %d\n",
1159 			__func__, ios->signal_voltage);
1160 
1161 	if (host->host_removal)
1162 		return -ENOMEDIUM;
1163 
1164 	if (ios->signal_voltage == MMC_SIGNAL_VOLTAGE_120)
1165 		return -EPERM;
1166 
1167 	mutex_lock(&ucr->dev_mutex);
1168 
1169 	err = rtsx_usb_card_exclusive_check(ucr, RTSX_USB_SD_CARD);
1170 	if (err) {
1171 		mutex_unlock(&ucr->dev_mutex);
1172 		return err;
1173 	}
1174 
1175 	/* Let mmc core do the busy checking, simply stop the forced-toggle
1176 	 * clock(while issuing CMD11) and switch voltage.
1177 	 */
1178 	rtsx_usb_init_cmd(ucr);
1179 
1180 	if (ios->signal_voltage == MMC_SIGNAL_VOLTAGE_330) {
1181 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_PAD_CTL,
1182 				SD_IO_USING_1V8, SD_IO_USING_3V3);
1183 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, LDO_POWER_CFG,
1184 				TUNE_SD18_MASK, TUNE_SD18_3V3);
1185 	} else {
1186 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_BUS_STAT,
1187 				SD_CLK_TOGGLE_EN | SD_CLK_FORCE_STOP,
1188 				SD_CLK_FORCE_STOP);
1189 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, SD_PAD_CTL,
1190 				SD_IO_USING_1V8, SD_IO_USING_1V8);
1191 		rtsx_usb_add_cmd(ucr, WRITE_REG_CMD, LDO_POWER_CFG,
1192 				TUNE_SD18_MASK, TUNE_SD18_1V8);
1193 	}
1194 
1195 	err = rtsx_usb_send_cmd(ucr, MODE_C, 100);
1196 	mutex_unlock(&ucr->dev_mutex);
1197 
1198 	return err;
1199 }
1200 
1201 static int sdmmc_card_busy(struct mmc_host *mmc)
1202 {
1203 	struct rtsx_usb_sdmmc *host = mmc_priv(mmc);
1204 	struct rtsx_ucr *ucr = host->ucr;
1205 	int err;
1206 	u8 stat;
1207 	u8 mask = SD_DAT3_STATUS | SD_DAT2_STATUS | SD_DAT1_STATUS
1208 		| SD_DAT0_STATUS;
1209 
1210 	dev_dbg(sdmmc_dev(host), "%s\n", __func__);
1211 
1212 	mutex_lock(&ucr->dev_mutex);
1213 
1214 	err = rtsx_usb_write_register(ucr, SD_BUS_STAT,
1215 			SD_CLK_TOGGLE_EN | SD_CLK_FORCE_STOP,
1216 			SD_CLK_TOGGLE_EN);
1217 	if (err)
1218 		goto out;
1219 
1220 	mdelay(1);
1221 
1222 	err = rtsx_usb_read_register(ucr, SD_BUS_STAT, &stat);
1223 	if (err)
1224 		goto out;
1225 
1226 	err = rtsx_usb_write_register(ucr, SD_BUS_STAT,
1227 			SD_CLK_TOGGLE_EN | SD_CLK_FORCE_STOP, 0);
1228 out:
1229 	mutex_unlock(&ucr->dev_mutex);
1230 
1231 	if (err)
1232 		return err;
1233 
1234 	/* check if any pin between dat[0:3] is low */
1235 	if ((stat & mask) != mask)
1236 		return 1;
1237 	else
1238 		return 0;
1239 }
1240 
1241 static int sdmmc_execute_tuning(struct mmc_host *mmc, u32 opcode)
1242 {
1243 	struct rtsx_usb_sdmmc *host = mmc_priv(mmc);
1244 	struct rtsx_ucr *ucr = host->ucr;
1245 	int err = 0;
1246 
1247 	if (host->host_removal)
1248 		return -ENOMEDIUM;
1249 
1250 	mutex_lock(&ucr->dev_mutex);
1251 
1252 	if (!host->ddr_mode)
1253 		err = sd_tuning_rx(host, MMC_SEND_TUNING_BLOCK);
1254 
1255 	mutex_unlock(&ucr->dev_mutex);
1256 
1257 	return err;
1258 }
1259 
1260 static const struct mmc_host_ops rtsx_usb_sdmmc_ops = {
1261 	.request = sdmmc_request,
1262 	.set_ios = sdmmc_set_ios,
1263 	.get_ro = sdmmc_get_ro,
1264 	.get_cd = sdmmc_get_cd,
1265 	.start_signal_voltage_switch = sdmmc_switch_voltage,
1266 	.card_busy = sdmmc_card_busy,
1267 	.execute_tuning = sdmmc_execute_tuning,
1268 };
1269 
1270 #ifdef RTSX_USB_USE_LEDS_CLASS
1271 static void rtsx_usb_led_control(struct led_classdev *led,
1272 	enum led_brightness brightness)
1273 {
1274 	struct rtsx_usb_sdmmc *host = container_of(led,
1275 			struct rtsx_usb_sdmmc, led);
1276 
1277 	if (host->host_removal)
1278 		return;
1279 
1280 	host->led.brightness = brightness;
1281 	schedule_work(&host->led_work);
1282 }
1283 
1284 static void rtsx_usb_update_led(struct work_struct *work)
1285 {
1286 	struct rtsx_usb_sdmmc *host =
1287 		container_of(work, struct rtsx_usb_sdmmc, led_work);
1288 	struct rtsx_ucr *ucr = host->ucr;
1289 
1290 	pm_runtime_get_noresume(sdmmc_dev(host));
1291 	mutex_lock(&ucr->dev_mutex);
1292 
1293 	if (host->power_mode == MMC_POWER_OFF)
1294 		goto out;
1295 
1296 	if (host->led.brightness == LED_OFF)
1297 		rtsx_usb_turn_off_led(ucr);
1298 	else
1299 		rtsx_usb_turn_on_led(ucr);
1300 
1301 out:
1302 	mutex_unlock(&ucr->dev_mutex);
1303 	pm_runtime_put_sync_suspend(sdmmc_dev(host));
1304 }
1305 #endif
1306 
1307 static void rtsx_usb_init_host(struct rtsx_usb_sdmmc *host)
1308 {
1309 	struct mmc_host *mmc = host->mmc;
1310 
1311 	mmc->f_min = 250000;
1312 	mmc->f_max = 208000000;
1313 	mmc->ocr_avail = MMC_VDD_32_33 | MMC_VDD_33_34 | MMC_VDD_165_195;
1314 	mmc->caps = MMC_CAP_4_BIT_DATA | MMC_CAP_SD_HIGHSPEED |
1315 		MMC_CAP_MMC_HIGHSPEED | MMC_CAP_BUS_WIDTH_TEST |
1316 		MMC_CAP_UHS_SDR12 | MMC_CAP_UHS_SDR25 | MMC_CAP_UHS_SDR50 |
1317 		MMC_CAP_ERASE | MMC_CAP_SYNC_RUNTIME_PM;
1318 	mmc->caps2 = MMC_CAP2_NO_PRESCAN_POWERUP | MMC_CAP2_FULL_PWR_CYCLE |
1319 		MMC_CAP2_NO_SDIO;
1320 
1321 	mmc->max_current_330 = 400;
1322 	mmc->max_current_180 = 800;
1323 	mmc->ops = &rtsx_usb_sdmmc_ops;
1324 	mmc->max_segs = 256;
1325 	mmc->max_seg_size = 65536;
1326 	mmc->max_blk_size = 512;
1327 	mmc->max_blk_count = 65535;
1328 	mmc->max_req_size = 524288;
1329 
1330 	host->power_mode = MMC_POWER_OFF;
1331 }
1332 
1333 static int rtsx_usb_sdmmc_drv_probe(struct platform_device *pdev)
1334 {
1335 	struct mmc_host *mmc;
1336 	struct rtsx_usb_sdmmc *host;
1337 	struct rtsx_ucr *ucr;
1338 #ifdef RTSX_USB_USE_LEDS_CLASS
1339 	int err;
1340 #endif
1341 
1342 	ucr = usb_get_intfdata(to_usb_interface(pdev->dev.parent));
1343 	if (!ucr)
1344 		return -ENXIO;
1345 
1346 	dev_dbg(&(pdev->dev), ": Realtek USB SD/MMC controller found\n");
1347 
1348 	mmc = mmc_alloc_host(sizeof(*host), &pdev->dev);
1349 	if (!mmc)
1350 		return -ENOMEM;
1351 
1352 	host = mmc_priv(mmc);
1353 	host->ucr = ucr;
1354 	host->mmc = mmc;
1355 	host->pdev = pdev;
1356 	platform_set_drvdata(pdev, host);
1357 
1358 	mutex_init(&host->host_mutex);
1359 	rtsx_usb_init_host(host);
1360 	pm_runtime_enable(&pdev->dev);
1361 
1362 #ifdef RTSX_USB_USE_LEDS_CLASS
1363 	snprintf(host->led_name, sizeof(host->led_name),
1364 		"%s::", mmc_hostname(mmc));
1365 	host->led.name = host->led_name;
1366 	host->led.brightness = LED_OFF;
1367 	host->led.default_trigger = mmc_hostname(mmc);
1368 	host->led.brightness_set = rtsx_usb_led_control;
1369 
1370 	err = led_classdev_register(mmc_dev(mmc), &host->led);
1371 	if (err)
1372 		dev_err(&(pdev->dev),
1373 				"Failed to register LED device: %d\n", err);
1374 	INIT_WORK(&host->led_work, rtsx_usb_update_led);
1375 
1376 #endif
1377 	mmc_add_host(mmc);
1378 
1379 	return 0;
1380 }
1381 
1382 static int rtsx_usb_sdmmc_drv_remove(struct platform_device *pdev)
1383 {
1384 	struct rtsx_usb_sdmmc *host = platform_get_drvdata(pdev);
1385 	struct mmc_host *mmc;
1386 
1387 	if (!host)
1388 		return 0;
1389 
1390 	mmc = host->mmc;
1391 	host->host_removal = true;
1392 
1393 	mutex_lock(&host->host_mutex);
1394 	if (host->mrq) {
1395 		dev_dbg(&(pdev->dev),
1396 			"%s: Controller removed during transfer\n",
1397 			mmc_hostname(mmc));
1398 		host->mrq->cmd->error = -ENOMEDIUM;
1399 		if (host->mrq->stop)
1400 			host->mrq->stop->error = -ENOMEDIUM;
1401 		mmc_request_done(mmc, host->mrq);
1402 	}
1403 	mutex_unlock(&host->host_mutex);
1404 
1405 	mmc_remove_host(mmc);
1406 
1407 #ifdef RTSX_USB_USE_LEDS_CLASS
1408 	cancel_work_sync(&host->led_work);
1409 	led_classdev_unregister(&host->led);
1410 #endif
1411 
1412 	mmc_free_host(mmc);
1413 	pm_runtime_disable(&pdev->dev);
1414 	platform_set_drvdata(pdev, NULL);
1415 
1416 	dev_dbg(&(pdev->dev),
1417 		": Realtek USB SD/MMC module has been removed\n");
1418 
1419 	return 0;
1420 }
1421 
1422 #ifdef CONFIG_PM
1423 static int rtsx_usb_sdmmc_runtime_suspend(struct device *dev)
1424 {
1425 	struct rtsx_usb_sdmmc *host = dev_get_drvdata(dev);
1426 
1427 	host->mmc->caps &= ~MMC_CAP_NEEDS_POLL;
1428 	return 0;
1429 }
1430 
1431 static int rtsx_usb_sdmmc_runtime_resume(struct device *dev)
1432 {
1433 	struct rtsx_usb_sdmmc *host = dev_get_drvdata(dev);
1434 
1435 	host->mmc->caps |= MMC_CAP_NEEDS_POLL;
1436 	if (sdmmc_get_cd(host->mmc) == 1)
1437 		mmc_detect_change(host->mmc, 0);
1438 	return 0;
1439 }
1440 #endif
1441 
1442 static const struct dev_pm_ops rtsx_usb_sdmmc_dev_pm_ops = {
1443 	SET_RUNTIME_PM_OPS(rtsx_usb_sdmmc_runtime_suspend,
1444 			   rtsx_usb_sdmmc_runtime_resume, NULL)
1445 };
1446 
1447 static const struct platform_device_id rtsx_usb_sdmmc_ids[] = {
1448 	{
1449 		.name = "rtsx_usb_sdmmc",
1450 	}, {
1451 		/* sentinel */
1452 	}
1453 };
1454 MODULE_DEVICE_TABLE(platform, rtsx_usb_sdmmc_ids);
1455 
1456 static struct platform_driver rtsx_usb_sdmmc_driver = {
1457 	.probe		= rtsx_usb_sdmmc_drv_probe,
1458 	.remove		= rtsx_usb_sdmmc_drv_remove,
1459 	.id_table       = rtsx_usb_sdmmc_ids,
1460 	.driver		= {
1461 		.name	= "rtsx_usb_sdmmc",
1462 		.pm	= &rtsx_usb_sdmmc_dev_pm_ops,
1463 	},
1464 };
1465 module_platform_driver(rtsx_usb_sdmmc_driver);
1466 
1467 MODULE_LICENSE("GPL v2");
1468 MODULE_AUTHOR("Roger Tseng <rogerable@realtek.com>");
1469 MODULE_DESCRIPTION("Realtek USB SD/MMC Card Host Driver");
1470