Merge tag 'net-next-5.15' of git://git.kernel.org/pub/scm/linux/kernel/git/netdev...
[linux-2.6-microblaze.git] / drivers / mmc / host / rtsx_pci_sdmmc.c
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /* Realtek PCI-Express SD/MMC Card Interface driver
3  *
4  * Copyright(c) 2009-2013 Realtek Semiconductor Corp. All rights reserved.
5  *
6  * Author:
7  *   Wei WANG <wei_wang@realsil.com.cn>
8  */
9
10 #include <linux/module.h>
11 #include <linux/slab.h>
12 #include <linux/highmem.h>
13 #include <linux/delay.h>
14 #include <linux/platform_device.h>
15 #include <linux/workqueue.h>
16 #include <linux/mmc/host.h>
17 #include <linux/mmc/mmc.h>
18 #include <linux/mmc/sd.h>
19 #include <linux/mmc/sdio.h>
20 #include <linux/mmc/card.h>
21 #include <linux/rtsx_pci.h>
22 #include <asm/unaligned.h>
23 #include <linux/pm_runtime.h>
24
25 struct realtek_pci_sdmmc {
26         struct platform_device  *pdev;
27         struct rtsx_pcr         *pcr;
28         struct mmc_host         *mmc;
29         struct mmc_request      *mrq;
30 #define SDMMC_WORKQ_NAME        "rtsx_pci_sdmmc_workq"
31
32         struct work_struct      work;
33         struct mutex            host_mutex;
34
35         u8                      ssc_depth;
36         unsigned int            clock;
37         bool                    vpclk;
38         bool                    double_clk;
39         bool                    eject;
40         bool                    initial_mode;
41         int                     power_state;
42 #define SDMMC_POWER_ON          1
43 #define SDMMC_POWER_OFF         0
44
45         int                     sg_count;
46         s32                     cookie;
47         int                     cookie_sg_count;
48         bool                    using_cookie;
49 };
50
51 static int sdmmc_init_sd_express(struct mmc_host *mmc, struct mmc_ios *ios);
52
53 static inline struct device *sdmmc_dev(struct realtek_pci_sdmmc *host)
54 {
55         return &(host->pdev->dev);
56 }
57
58 static inline void sd_clear_error(struct realtek_pci_sdmmc *host)
59 {
60         rtsx_pci_write_register(host->pcr, CARD_STOP,
61                         SD_STOP | SD_CLR_ERR, SD_STOP | SD_CLR_ERR);
62 }
63
64 #ifdef DEBUG
65 static void dump_reg_range(struct realtek_pci_sdmmc *host, u16 start, u16 end)
66 {
67         u16 len = end - start + 1;
68         int i;
69         u8 data[8];
70
71         for (i = 0; i < len; i += 8) {
72                 int j;
73                 int n = min(8, len - i);
74
75                 memset(&data, 0, sizeof(data));
76                 for (j = 0; j < n; j++)
77                         rtsx_pci_read_register(host->pcr, start + i + j,
78                                 data + j);
79                 dev_dbg(sdmmc_dev(host), "0x%04X(%d): %8ph\n",
80                         start + i, n, data);
81         }
82 }
83
84 static void sd_print_debug_regs(struct realtek_pci_sdmmc *host)
85 {
86         dump_reg_range(host, 0xFDA0, 0xFDB3);
87         dump_reg_range(host, 0xFD52, 0xFD69);
88 }
89 #else
90 #define sd_print_debug_regs(host)
91 #endif /* DEBUG */
92
93 static inline int sd_get_cd_int(struct realtek_pci_sdmmc *host)
94 {
95         return rtsx_pci_readl(host->pcr, RTSX_BIPR) & SD_EXIST;
96 }
97
98 static void sd_cmd_set_sd_cmd(struct rtsx_pcr *pcr, struct mmc_command *cmd)
99 {
100         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_CMD0, 0xFF,
101                 SD_CMD_START | cmd->opcode);
102         rtsx_pci_write_be32(pcr, SD_CMD1, cmd->arg);
103 }
104
105 static void sd_cmd_set_data_len(struct rtsx_pcr *pcr, u16 blocks, u16 blksz)
106 {
107         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_BLOCK_CNT_L, 0xFF, blocks);
108         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_BLOCK_CNT_H, 0xFF, blocks >> 8);
109         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_BYTE_CNT_L, 0xFF, blksz);
110         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_BYTE_CNT_H, 0xFF, blksz >> 8);
111 }
112
113 static int sd_response_type(struct mmc_command *cmd)
114 {
115         switch (mmc_resp_type(cmd)) {
116         case MMC_RSP_NONE:
117                 return SD_RSP_TYPE_R0;
118         case MMC_RSP_R1:
119                 return SD_RSP_TYPE_R1;
120         case MMC_RSP_R1_NO_CRC:
121                 return SD_RSP_TYPE_R1 | SD_NO_CHECK_CRC7;
122         case MMC_RSP_R1B:
123                 return SD_RSP_TYPE_R1b;
124         case MMC_RSP_R2:
125                 return SD_RSP_TYPE_R2;
126         case MMC_RSP_R3:
127                 return SD_RSP_TYPE_R3;
128         default:
129                 return -EINVAL;
130         }
131 }
132
133 static int sd_status_index(int resp_type)
134 {
135         if (resp_type == SD_RSP_TYPE_R0)
136                 return 0;
137         else if (resp_type == SD_RSP_TYPE_R2)
138                 return 16;
139
140         return 5;
141 }
142 /*
143  * sd_pre_dma_transfer - do dma_map_sg() or using cookie
144  *
145  * @pre: if called in pre_req()
146  * return:
147  *      0 - do dma_map_sg()
148  *      1 - using cookie
149  */
150 static int sd_pre_dma_transfer(struct realtek_pci_sdmmc *host,
151                 struct mmc_data *data, bool pre)
152 {
153         struct rtsx_pcr *pcr = host->pcr;
154         int read = data->flags & MMC_DATA_READ;
155         int count = 0;
156         int using_cookie = 0;
157
158         if (!pre && data->host_cookie && data->host_cookie != host->cookie) {
159                 dev_err(sdmmc_dev(host),
160                         "error: data->host_cookie = %d, host->cookie = %d\n",
161                         data->host_cookie, host->cookie);
162                 data->host_cookie = 0;
163         }
164
165         if (pre || data->host_cookie != host->cookie) {
166                 count = rtsx_pci_dma_map_sg(pcr, data->sg, data->sg_len, read);
167         } else {
168                 count = host->cookie_sg_count;
169                 using_cookie = 1;
170         }
171
172         if (pre) {
173                 host->cookie_sg_count = count;
174                 if (++host->cookie < 0)
175                         host->cookie = 1;
176                 data->host_cookie = host->cookie;
177         } else {
178                 host->sg_count = count;
179         }
180
181         return using_cookie;
182 }
183
184 static void sdmmc_pre_req(struct mmc_host *mmc, struct mmc_request *mrq)
185 {
186         struct realtek_pci_sdmmc *host = mmc_priv(mmc);
187         struct mmc_data *data = mrq->data;
188
189         if (data->host_cookie) {
190                 dev_err(sdmmc_dev(host),
191                         "error: reset data->host_cookie = %d\n",
192                         data->host_cookie);
193                 data->host_cookie = 0;
194         }
195
196         sd_pre_dma_transfer(host, data, true);
197         dev_dbg(sdmmc_dev(host), "pre dma sg: %d\n", host->cookie_sg_count);
198 }
199
200 static void sdmmc_post_req(struct mmc_host *mmc, struct mmc_request *mrq,
201                 int err)
202 {
203         struct realtek_pci_sdmmc *host = mmc_priv(mmc);
204         struct rtsx_pcr *pcr = host->pcr;
205         struct mmc_data *data = mrq->data;
206         int read = data->flags & MMC_DATA_READ;
207
208         rtsx_pci_dma_unmap_sg(pcr, data->sg, data->sg_len, read);
209         data->host_cookie = 0;
210 }
211
212 static void sd_send_cmd_get_rsp(struct realtek_pci_sdmmc *host,
213                 struct mmc_command *cmd)
214 {
215         struct rtsx_pcr *pcr = host->pcr;
216         u8 cmd_idx = (u8)cmd->opcode;
217         u32 arg = cmd->arg;
218         int err = 0;
219         int timeout = 100;
220         int i;
221         u8 *ptr;
222         int rsp_type;
223         int stat_idx;
224         bool clock_toggled = false;
225
226         dev_dbg(sdmmc_dev(host), "%s: SD/MMC CMD %d, arg = 0x%08x\n",
227                         __func__, cmd_idx, arg);
228
229         rsp_type = sd_response_type(cmd);
230         if (rsp_type < 0)
231                 goto out;
232
233         stat_idx = sd_status_index(rsp_type);
234
235         if (rsp_type == SD_RSP_TYPE_R1b)
236                 timeout = cmd->busy_timeout ? cmd->busy_timeout : 3000;
237
238         if (cmd->opcode == SD_SWITCH_VOLTAGE) {
239                 err = rtsx_pci_write_register(pcr, SD_BUS_STAT,
240                                 0xFF, SD_CLK_TOGGLE_EN);
241                 if (err < 0)
242                         goto out;
243
244                 clock_toggled = true;
245         }
246
247         rtsx_pci_init_cmd(pcr);
248         sd_cmd_set_sd_cmd(pcr, cmd);
249         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_CFG2, 0xFF, rsp_type);
250         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_DATA_SOURCE,
251                         0x01, PINGPONG_BUFFER);
252         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_TRANSFER,
253                         0xFF, SD_TM_CMD_RSP | SD_TRANSFER_START);
254         rtsx_pci_add_cmd(pcr, CHECK_REG_CMD, SD_TRANSFER,
255                      SD_TRANSFER_END | SD_STAT_IDLE,
256                      SD_TRANSFER_END | SD_STAT_IDLE);
257
258         if (rsp_type == SD_RSP_TYPE_R2) {
259                 /* Read data from ping-pong buffer */
260                 for (i = PPBUF_BASE2; i < PPBUF_BASE2 + 16; i++)
261                         rtsx_pci_add_cmd(pcr, READ_REG_CMD, (u16)i, 0, 0);
262         } else if (rsp_type != SD_RSP_TYPE_R0) {
263                 /* Read data from SD_CMDx registers */
264                 for (i = SD_CMD0; i <= SD_CMD4; i++)
265                         rtsx_pci_add_cmd(pcr, READ_REG_CMD, (u16)i, 0, 0);
266         }
267
268         rtsx_pci_add_cmd(pcr, READ_REG_CMD, SD_STAT1, 0, 0);
269
270         err = rtsx_pci_send_cmd(pcr, timeout);
271         if (err < 0) {
272                 sd_print_debug_regs(host);
273                 sd_clear_error(host);
274                 dev_dbg(sdmmc_dev(host),
275                         "rtsx_pci_send_cmd error (err = %d)\n", err);
276                 goto out;
277         }
278
279         if (rsp_type == SD_RSP_TYPE_R0) {
280                 err = 0;
281                 goto out;
282         }
283
284         /* Eliminate returned value of CHECK_REG_CMD */
285         ptr = rtsx_pci_get_cmd_data(pcr) + 1;
286
287         /* Check (Start,Transmission) bit of Response */
288         if ((ptr[0] & 0xC0) != 0) {
289                 err = -EILSEQ;
290                 dev_dbg(sdmmc_dev(host), "Invalid response bit\n");
291                 goto out;
292         }
293
294         /* Check CRC7 */
295         if (!(rsp_type & SD_NO_CHECK_CRC7)) {
296                 if (ptr[stat_idx] & SD_CRC7_ERR) {
297                         err = -EILSEQ;
298                         dev_dbg(sdmmc_dev(host), "CRC7 error\n");
299                         goto out;
300                 }
301         }
302
303         if (rsp_type == SD_RSP_TYPE_R2) {
304                 /*
305                  * The controller offloads the last byte {CRC-7, end bit 1'b1}
306                  * of response type R2. Assign dummy CRC, 0, and end bit to the
307                  * byte(ptr[16], goes into the LSB of resp[3] later).
308                  */
309                 ptr[16] = 1;
310
311                 for (i = 0; i < 4; i++) {
312                         cmd->resp[i] = get_unaligned_be32(ptr + 1 + i * 4);
313                         dev_dbg(sdmmc_dev(host), "cmd->resp[%d] = 0x%08x\n",
314                                         i, cmd->resp[i]);
315                 }
316         } else {
317                 cmd->resp[0] = get_unaligned_be32(ptr + 1);
318                 dev_dbg(sdmmc_dev(host), "cmd->resp[0] = 0x%08x\n",
319                                 cmd->resp[0]);
320         }
321
322 out:
323         cmd->error = err;
324
325         if (err && clock_toggled)
326                 rtsx_pci_write_register(pcr, SD_BUS_STAT,
327                                 SD_CLK_TOGGLE_EN | SD_CLK_FORCE_STOP, 0);
328 }
329
330 static int sd_read_data(struct realtek_pci_sdmmc *host, struct mmc_command *cmd,
331         u16 byte_cnt, u8 *buf, int buf_len, int timeout)
332 {
333         struct rtsx_pcr *pcr = host->pcr;
334         int err;
335         u8 trans_mode;
336
337         dev_dbg(sdmmc_dev(host), "%s: SD/MMC CMD %d, arg = 0x%08x\n",
338                 __func__, cmd->opcode, cmd->arg);
339
340         if (!buf)
341                 buf_len = 0;
342
343         if (cmd->opcode == MMC_SEND_TUNING_BLOCK)
344                 trans_mode = SD_TM_AUTO_TUNING;
345         else
346                 trans_mode = SD_TM_NORMAL_READ;
347
348         rtsx_pci_init_cmd(pcr);
349         sd_cmd_set_sd_cmd(pcr, cmd);
350         sd_cmd_set_data_len(pcr, 1, byte_cnt);
351         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_CFG2, 0xFF,
352                         SD_CALCULATE_CRC7 | SD_CHECK_CRC16 |
353                         SD_NO_WAIT_BUSY_END | SD_CHECK_CRC7 | SD_RSP_LEN_6);
354         if (trans_mode != SD_TM_AUTO_TUNING)
355                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD,
356                                 CARD_DATA_SOURCE, 0x01, PINGPONG_BUFFER);
357
358         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_TRANSFER,
359                         0xFF, trans_mode | SD_TRANSFER_START);
360         rtsx_pci_add_cmd(pcr, CHECK_REG_CMD, SD_TRANSFER,
361                         SD_TRANSFER_END, SD_TRANSFER_END);
362
363         err = rtsx_pci_send_cmd(pcr, timeout);
364         if (err < 0) {
365                 sd_print_debug_regs(host);
366                 dev_dbg(sdmmc_dev(host),
367                         "rtsx_pci_send_cmd fail (err = %d)\n", err);
368                 return err;
369         }
370
371         if (buf && buf_len) {
372                 err = rtsx_pci_read_ppbuf(pcr, buf, buf_len);
373                 if (err < 0) {
374                         dev_dbg(sdmmc_dev(host),
375                                 "rtsx_pci_read_ppbuf fail (err = %d)\n", err);
376                         return err;
377                 }
378         }
379
380         return 0;
381 }
382
383 static int sd_write_data(struct realtek_pci_sdmmc *host,
384         struct mmc_command *cmd, u16 byte_cnt, u8 *buf, int buf_len,
385         int timeout)
386 {
387         struct rtsx_pcr *pcr = host->pcr;
388         int err;
389
390         dev_dbg(sdmmc_dev(host), "%s: SD/MMC CMD %d, arg = 0x%08x\n",
391                 __func__, cmd->opcode, cmd->arg);
392
393         if (!buf)
394                 buf_len = 0;
395
396         sd_send_cmd_get_rsp(host, cmd);
397         if (cmd->error)
398                 return cmd->error;
399
400         if (buf && buf_len) {
401                 err = rtsx_pci_write_ppbuf(pcr, buf, buf_len);
402                 if (err < 0) {
403                         dev_dbg(sdmmc_dev(host),
404                                 "rtsx_pci_write_ppbuf fail (err = %d)\n", err);
405                         return err;
406                 }
407         }
408
409         rtsx_pci_init_cmd(pcr);
410         sd_cmd_set_data_len(pcr, 1, byte_cnt);
411         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_CFG2, 0xFF,
412                 SD_CALCULATE_CRC7 | SD_CHECK_CRC16 |
413                 SD_NO_WAIT_BUSY_END | SD_CHECK_CRC7 | SD_RSP_LEN_0);
414         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_TRANSFER, 0xFF,
415                         SD_TRANSFER_START | SD_TM_AUTO_WRITE_3);
416         rtsx_pci_add_cmd(pcr, CHECK_REG_CMD, SD_TRANSFER,
417                         SD_TRANSFER_END, SD_TRANSFER_END);
418
419         err = rtsx_pci_send_cmd(pcr, timeout);
420         if (err < 0) {
421                 sd_print_debug_regs(host);
422                 dev_dbg(sdmmc_dev(host),
423                         "rtsx_pci_send_cmd fail (err = %d)\n", err);
424                 return err;
425         }
426
427         return 0;
428 }
429
430 static int sd_read_long_data(struct realtek_pci_sdmmc *host,
431         struct mmc_request *mrq)
432 {
433         struct rtsx_pcr *pcr = host->pcr;
434         struct mmc_host *mmc = host->mmc;
435         struct mmc_card *card = mmc->card;
436         struct mmc_command *cmd = mrq->cmd;
437         struct mmc_data *data = mrq->data;
438         int uhs = mmc_card_uhs(card);
439         u8 cfg2 = 0;
440         int err;
441         int resp_type;
442         size_t data_len = data->blksz * data->blocks;
443
444         dev_dbg(sdmmc_dev(host), "%s: SD/MMC CMD %d, arg = 0x%08x\n",
445                 __func__, cmd->opcode, cmd->arg);
446
447         resp_type = sd_response_type(cmd);
448         if (resp_type < 0)
449                 return resp_type;
450
451         if (!uhs)
452                 cfg2 |= SD_NO_CHECK_WAIT_CRC_TO;
453
454         rtsx_pci_init_cmd(pcr);
455         sd_cmd_set_sd_cmd(pcr, cmd);
456         sd_cmd_set_data_len(pcr, data->blocks, data->blksz);
457         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, IRQSTAT0,
458                         DMA_DONE_INT, DMA_DONE_INT);
459         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMATC3,
460                 0xFF, (u8)(data_len >> 24));
461         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMATC2,
462                 0xFF, (u8)(data_len >> 16));
463         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMATC1,
464                 0xFF, (u8)(data_len >> 8));
465         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMATC0, 0xFF, (u8)data_len);
466         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMACTL,
467                 0x03 | DMA_PACK_SIZE_MASK,
468                 DMA_DIR_FROM_CARD | DMA_EN | DMA_512);
469         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_DATA_SOURCE,
470                         0x01, RING_BUFFER);
471         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_CFG2, 0xFF, cfg2 | resp_type);
472         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_TRANSFER, 0xFF,
473                         SD_TRANSFER_START | SD_TM_AUTO_READ_2);
474         rtsx_pci_add_cmd(pcr, CHECK_REG_CMD, SD_TRANSFER,
475                         SD_TRANSFER_END, SD_TRANSFER_END);
476         rtsx_pci_send_cmd_no_wait(pcr);
477
478         err = rtsx_pci_dma_transfer(pcr, data->sg, host->sg_count, 1, 10000);
479         if (err < 0) {
480                 sd_print_debug_regs(host);
481                 sd_clear_error(host);
482                 return err;
483         }
484
485         return 0;
486 }
487
488 static int sd_write_long_data(struct realtek_pci_sdmmc *host,
489         struct mmc_request *mrq)
490 {
491         struct rtsx_pcr *pcr = host->pcr;
492         struct mmc_host *mmc = host->mmc;
493         struct mmc_card *card = mmc->card;
494         struct mmc_command *cmd = mrq->cmd;
495         struct mmc_data *data = mrq->data;
496         int uhs = mmc_card_uhs(card);
497         u8 cfg2;
498         int err;
499         size_t data_len = data->blksz * data->blocks;
500
501         sd_send_cmd_get_rsp(host, cmd);
502         if (cmd->error)
503                 return cmd->error;
504
505         dev_dbg(sdmmc_dev(host), "%s: SD/MMC CMD %d, arg = 0x%08x\n",
506                 __func__, cmd->opcode, cmd->arg);
507
508         cfg2 = SD_NO_CALCULATE_CRC7 | SD_CHECK_CRC16 |
509                 SD_NO_WAIT_BUSY_END | SD_NO_CHECK_CRC7 | SD_RSP_LEN_0;
510
511         if (!uhs)
512                 cfg2 |= SD_NO_CHECK_WAIT_CRC_TO;
513
514         rtsx_pci_init_cmd(pcr);
515         sd_cmd_set_data_len(pcr, data->blocks, data->blksz);
516         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, IRQSTAT0,
517                         DMA_DONE_INT, DMA_DONE_INT);
518         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMATC3,
519                 0xFF, (u8)(data_len >> 24));
520         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMATC2,
521                 0xFF, (u8)(data_len >> 16));
522         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMATC1,
523                 0xFF, (u8)(data_len >> 8));
524         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMATC0, 0xFF, (u8)data_len);
525         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, DMACTL,
526                 0x03 | DMA_PACK_SIZE_MASK,
527                 DMA_DIR_TO_CARD | DMA_EN | DMA_512);
528         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_DATA_SOURCE,
529                         0x01, RING_BUFFER);
530         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_CFG2, 0xFF, cfg2);
531         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_TRANSFER, 0xFF,
532                         SD_TRANSFER_START | SD_TM_AUTO_WRITE_3);
533         rtsx_pci_add_cmd(pcr, CHECK_REG_CMD, SD_TRANSFER,
534                         SD_TRANSFER_END, SD_TRANSFER_END);
535         rtsx_pci_send_cmd_no_wait(pcr);
536         err = rtsx_pci_dma_transfer(pcr, data->sg, host->sg_count, 0, 10000);
537         if (err < 0) {
538                 sd_clear_error(host);
539                 return err;
540         }
541
542         return 0;
543 }
544
545 static inline void sd_enable_initial_mode(struct realtek_pci_sdmmc *host)
546 {
547         rtsx_pci_write_register(host->pcr, SD_CFG1,
548                         SD_CLK_DIVIDE_MASK, SD_CLK_DIVIDE_128);
549 }
550
551 static inline void sd_disable_initial_mode(struct realtek_pci_sdmmc *host)
552 {
553         rtsx_pci_write_register(host->pcr, SD_CFG1,
554                         SD_CLK_DIVIDE_MASK, SD_CLK_DIVIDE_0);
555 }
556
557 static int sd_rw_multi(struct realtek_pci_sdmmc *host, struct mmc_request *mrq)
558 {
559         struct mmc_data *data = mrq->data;
560         int err;
561
562         if (host->sg_count < 0) {
563                 data->error = host->sg_count;
564                 dev_dbg(sdmmc_dev(host), "%s: sg_count = %d is invalid\n",
565                         __func__, host->sg_count);
566                 return data->error;
567         }
568
569         if (data->flags & MMC_DATA_READ) {
570                 if (host->initial_mode)
571                         sd_disable_initial_mode(host);
572
573                 err = sd_read_long_data(host, mrq);
574
575                 if (host->initial_mode)
576                         sd_enable_initial_mode(host);
577
578                 return err;
579         }
580
581         return sd_write_long_data(host, mrq);
582 }
583
584 static void sd_normal_rw(struct realtek_pci_sdmmc *host,
585                 struct mmc_request *mrq)
586 {
587         struct mmc_command *cmd = mrq->cmd;
588         struct mmc_data *data = mrq->data;
589         u8 *buf;
590
591         buf = kzalloc(data->blksz, GFP_NOIO);
592         if (!buf) {
593                 cmd->error = -ENOMEM;
594                 return;
595         }
596
597         if (data->flags & MMC_DATA_READ) {
598                 if (host->initial_mode)
599                         sd_disable_initial_mode(host);
600
601                 cmd->error = sd_read_data(host, cmd, (u16)data->blksz, buf,
602                                 data->blksz, 200);
603
604                 if (host->initial_mode)
605                         sd_enable_initial_mode(host);
606
607                 sg_copy_from_buffer(data->sg, data->sg_len, buf, data->blksz);
608         } else {
609                 sg_copy_to_buffer(data->sg, data->sg_len, buf, data->blksz);
610
611                 cmd->error = sd_write_data(host, cmd, (u16)data->blksz, buf,
612                                 data->blksz, 200);
613         }
614
615         kfree(buf);
616 }
617
618 static int sd_change_phase(struct realtek_pci_sdmmc *host,
619                 u8 sample_point, bool rx)
620 {
621         struct rtsx_pcr *pcr = host->pcr;
622         u16 SD_VP_CTL = 0;
623         dev_dbg(sdmmc_dev(host), "%s(%s): sample_point = %d\n",
624                         __func__, rx ? "RX" : "TX", sample_point);
625
626         rtsx_pci_write_register(pcr, CLK_CTL, CHANGE_CLK, CHANGE_CLK);
627         if (rx) {
628                 SD_VP_CTL = SD_VPRX_CTL;
629                 rtsx_pci_write_register(pcr, SD_VPRX_CTL,
630                         PHASE_SELECT_MASK, sample_point);
631         } else {
632                 SD_VP_CTL = SD_VPTX_CTL;
633                 rtsx_pci_write_register(pcr, SD_VPTX_CTL,
634                         PHASE_SELECT_MASK, sample_point);
635         }
636         rtsx_pci_write_register(pcr, SD_VP_CTL, PHASE_NOT_RESET, 0);
637         rtsx_pci_write_register(pcr, SD_VP_CTL, PHASE_NOT_RESET,
638                                 PHASE_NOT_RESET);
639         rtsx_pci_write_register(pcr, CLK_CTL, CHANGE_CLK, 0);
640         rtsx_pci_write_register(pcr, SD_CFG1, SD_ASYNC_FIFO_NOT_RST, 0);
641
642         return 0;
643 }
644
645 static inline u32 test_phase_bit(u32 phase_map, unsigned int bit)
646 {
647         bit %= RTSX_PHASE_MAX;
648         return phase_map & (1 << bit);
649 }
650
651 static int sd_get_phase_len(u32 phase_map, unsigned int start_bit)
652 {
653         int i;
654
655         for (i = 0; i < RTSX_PHASE_MAX; i++) {
656                 if (test_phase_bit(phase_map, start_bit + i) == 0)
657                         return i;
658         }
659         return RTSX_PHASE_MAX;
660 }
661
662 static u8 sd_search_final_phase(struct realtek_pci_sdmmc *host, u32 phase_map)
663 {
664         int start = 0, len = 0;
665         int start_final = 0, len_final = 0;
666         u8 final_phase = 0xFF;
667
668         if (phase_map == 0) {
669                 dev_err(sdmmc_dev(host), "phase error: [map:%x]\n", phase_map);
670                 return final_phase;
671         }
672
673         while (start < RTSX_PHASE_MAX) {
674                 len = sd_get_phase_len(phase_map, start);
675                 if (len_final < len) {
676                         start_final = start;
677                         len_final = len;
678                 }
679                 start += len ? len : 1;
680         }
681
682         final_phase = (start_final + len_final / 2) % RTSX_PHASE_MAX;
683         dev_dbg(sdmmc_dev(host), "phase: [map:%x] [maxlen:%d] [final:%d]\n",
684                 phase_map, len_final, final_phase);
685
686         return final_phase;
687 }
688
689 static void sd_wait_data_idle(struct realtek_pci_sdmmc *host)
690 {
691         int i;
692         u8 val = 0;
693
694         for (i = 0; i < 100; i++) {
695                 rtsx_pci_read_register(host->pcr, SD_DATA_STATE, &val);
696                 if (val & SD_DATA_IDLE)
697                         return;
698
699                 udelay(100);
700         }
701 }
702
703 static int sd_tuning_rx_cmd(struct realtek_pci_sdmmc *host,
704                 u8 opcode, u8 sample_point)
705 {
706         int err;
707         struct mmc_command cmd = {};
708         struct rtsx_pcr *pcr = host->pcr;
709
710         sd_change_phase(host, sample_point, true);
711
712         rtsx_pci_write_register(pcr, SD_CFG3, SD_RSP_80CLK_TIMEOUT_EN,
713                 SD_RSP_80CLK_TIMEOUT_EN);
714
715         cmd.opcode = opcode;
716         err = sd_read_data(host, &cmd, 0x40, NULL, 0, 100);
717         if (err < 0) {
718                 /* Wait till SD DATA IDLE */
719                 sd_wait_data_idle(host);
720                 sd_clear_error(host);
721                 rtsx_pci_write_register(pcr, SD_CFG3,
722                         SD_RSP_80CLK_TIMEOUT_EN, 0);
723                 return err;
724         }
725
726         rtsx_pci_write_register(pcr, SD_CFG3, SD_RSP_80CLK_TIMEOUT_EN, 0);
727         return 0;
728 }
729
730 static int sd_tuning_phase(struct realtek_pci_sdmmc *host,
731                 u8 opcode, u32 *phase_map)
732 {
733         int err, i;
734         u32 raw_phase_map = 0;
735
736         for (i = 0; i < RTSX_PHASE_MAX; i++) {
737                 err = sd_tuning_rx_cmd(host, opcode, (u8)i);
738                 if (err == 0)
739                         raw_phase_map |= 1 << i;
740         }
741
742         if (phase_map)
743                 *phase_map = raw_phase_map;
744
745         return 0;
746 }
747
748 static int sd_tuning_rx(struct realtek_pci_sdmmc *host, u8 opcode)
749 {
750         int err, i;
751         u32 raw_phase_map[RX_TUNING_CNT] = {0}, phase_map;
752         u8 final_phase;
753
754         for (i = 0; i < RX_TUNING_CNT; i++) {
755                 err = sd_tuning_phase(host, opcode, &(raw_phase_map[i]));
756                 if (err < 0)
757                         return err;
758
759                 if (raw_phase_map[i] == 0)
760                         break;
761         }
762
763         phase_map = 0xFFFFFFFF;
764         for (i = 0; i < RX_TUNING_CNT; i++) {
765                 dev_dbg(sdmmc_dev(host), "RX raw_phase_map[%d] = 0x%08x\n",
766                                 i, raw_phase_map[i]);
767                 phase_map &= raw_phase_map[i];
768         }
769         dev_dbg(sdmmc_dev(host), "RX phase_map = 0x%08x\n", phase_map);
770
771         if (phase_map) {
772                 final_phase = sd_search_final_phase(host, phase_map);
773                 if (final_phase == 0xFF)
774                         return -EINVAL;
775
776                 err = sd_change_phase(host, final_phase, true);
777                 if (err < 0)
778                         return err;
779         } else {
780                 return -EINVAL;
781         }
782
783         return 0;
784 }
785
786 static inline int sdio_extblock_cmd(struct mmc_command *cmd,
787         struct mmc_data *data)
788 {
789         return (cmd->opcode == SD_IO_RW_EXTENDED) && (data->blksz == 512);
790 }
791
792 static inline int sd_rw_cmd(struct mmc_command *cmd)
793 {
794         return mmc_op_multi(cmd->opcode) ||
795                 (cmd->opcode == MMC_READ_SINGLE_BLOCK) ||
796                 (cmd->opcode == MMC_WRITE_BLOCK);
797 }
798
799 static void sd_request(struct work_struct *work)
800 {
801         struct realtek_pci_sdmmc *host = container_of(work,
802                         struct realtek_pci_sdmmc, work);
803         struct rtsx_pcr *pcr = host->pcr;
804
805         struct mmc_host *mmc = host->mmc;
806         struct mmc_request *mrq = host->mrq;
807         struct mmc_command *cmd = mrq->cmd;
808         struct mmc_data *data = mrq->data;
809
810         unsigned int data_size = 0;
811         int err;
812
813         if (host->eject || !sd_get_cd_int(host)) {
814                 cmd->error = -ENOMEDIUM;
815                 goto finish;
816         }
817
818         err = rtsx_pci_card_exclusive_check(host->pcr, RTSX_SD_CARD);
819         if (err) {
820                 cmd->error = err;
821                 goto finish;
822         }
823
824         mutex_lock(&pcr->pcr_mutex);
825
826         rtsx_pci_start_run(pcr);
827
828         rtsx_pci_switch_clock(pcr, host->clock, host->ssc_depth,
829                         host->initial_mode, host->double_clk, host->vpclk);
830         rtsx_pci_write_register(pcr, CARD_SELECT, 0x07, SD_MOD_SEL);
831         rtsx_pci_write_register(pcr, CARD_SHARE_MODE,
832                         CARD_SHARE_MASK, CARD_SHARE_48_SD);
833
834         mutex_lock(&host->host_mutex);
835         host->mrq = mrq;
836         mutex_unlock(&host->host_mutex);
837
838         if (mrq->data)
839                 data_size = data->blocks * data->blksz;
840
841         if (!data_size) {
842                 sd_send_cmd_get_rsp(host, cmd);
843         } else if (sd_rw_cmd(cmd) || sdio_extblock_cmd(cmd, data)) {
844                 cmd->error = sd_rw_multi(host, mrq);
845                 if (!host->using_cookie)
846                         sdmmc_post_req(host->mmc, host->mrq, 0);
847
848                 if (mmc_op_multi(cmd->opcode) && mrq->stop)
849                         sd_send_cmd_get_rsp(host, mrq->stop);
850         } else {
851                 sd_normal_rw(host, mrq);
852         }
853
854         if (mrq->data) {
855                 if (cmd->error || data->error)
856                         data->bytes_xfered = 0;
857                 else
858                         data->bytes_xfered = data->blocks * data->blksz;
859         }
860
861         mutex_unlock(&pcr->pcr_mutex);
862
863 finish:
864         if (cmd->error) {
865                 dev_dbg(sdmmc_dev(host), "CMD %d 0x%08x error(%d)\n",
866                         cmd->opcode, cmd->arg, cmd->error);
867         }
868
869         mutex_lock(&host->host_mutex);
870         host->mrq = NULL;
871         mutex_unlock(&host->host_mutex);
872
873         mmc_request_done(mmc, mrq);
874 }
875
876 static void sdmmc_request(struct mmc_host *mmc, struct mmc_request *mrq)
877 {
878         struct realtek_pci_sdmmc *host = mmc_priv(mmc);
879         struct mmc_data *data = mrq->data;
880
881         mutex_lock(&host->host_mutex);
882         host->mrq = mrq;
883         mutex_unlock(&host->host_mutex);
884
885         if (sd_rw_cmd(mrq->cmd) || sdio_extblock_cmd(mrq->cmd, data))
886                 host->using_cookie = sd_pre_dma_transfer(host, data, false);
887
888         schedule_work(&host->work);
889 }
890
891 static int sd_set_bus_width(struct realtek_pci_sdmmc *host,
892                 unsigned char bus_width)
893 {
894         int err = 0;
895         u8 width[] = {
896                 [MMC_BUS_WIDTH_1] = SD_BUS_WIDTH_1BIT,
897                 [MMC_BUS_WIDTH_4] = SD_BUS_WIDTH_4BIT,
898                 [MMC_BUS_WIDTH_8] = SD_BUS_WIDTH_8BIT,
899         };
900
901         if (bus_width <= MMC_BUS_WIDTH_8)
902                 err = rtsx_pci_write_register(host->pcr, SD_CFG1,
903                                 0x03, width[bus_width]);
904
905         return err;
906 }
907
908 static int sd_power_on(struct realtek_pci_sdmmc *host)
909 {
910         struct rtsx_pcr *pcr = host->pcr;
911         struct mmc_host *mmc = host->mmc;
912         int err;
913         u32 val;
914         u8 test_mode;
915
916         if (host->power_state == SDMMC_POWER_ON)
917                 return 0;
918
919         msleep(100);
920
921         rtsx_pci_init_cmd(pcr);
922         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_SELECT, 0x07, SD_MOD_SEL);
923         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_SHARE_MODE,
924                         CARD_SHARE_MASK, CARD_SHARE_48_SD);
925         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_CLK_EN,
926                         SD_CLK_EN, SD_CLK_EN);
927         err = rtsx_pci_send_cmd(pcr, 100);
928         if (err < 0)
929                 return err;
930
931         err = rtsx_pci_card_pull_ctl_enable(pcr, RTSX_SD_CARD);
932         if (err < 0)
933                 return err;
934
935         err = rtsx_pci_card_power_on(pcr, RTSX_SD_CARD);
936         if (err < 0)
937                 return err;
938
939         err = rtsx_pci_write_register(pcr, CARD_OE, SD_OUTPUT_EN, SD_OUTPUT_EN);
940         if (err < 0)
941                 return err;
942
943         if (PCI_PID(pcr) == PID_5261) {
944                 /*
945                  * If test mode is set switch to SD Express mandatorily,
946                  * this is only for factory testing.
947                  */
948                 rtsx_pci_read_register(pcr, RTS5261_FW_CFG_INFO0, &test_mode);
949                 if (test_mode & RTS5261_FW_EXPRESS_TEST_MASK) {
950                         sdmmc_init_sd_express(mmc, NULL);
951                         return 0;
952                 }
953                 if (pcr->extra_caps & EXTRA_CAPS_SD_EXPRESS)
954                         mmc->caps2 |= MMC_CAP2_SD_EXP | MMC_CAP2_SD_EXP_1_2V;
955                 /*
956                  * HW read wp status when resuming from S3/S4,
957                  * and then picks SD legacy interface if it's set
958                  * in read-only mode.
959                  */
960                 val = rtsx_pci_readl(pcr, RTSX_BIPR);
961                 if (val & SD_WRITE_PROTECT) {
962                         pcr->extra_caps &= ~EXTRA_CAPS_SD_EXPRESS;
963                         mmc->caps2 &= ~(MMC_CAP2_SD_EXP | MMC_CAP2_SD_EXP_1_2V);
964                 }
965         }
966
967         host->power_state = SDMMC_POWER_ON;
968         return 0;
969 }
970
971 static int sd_power_off(struct realtek_pci_sdmmc *host)
972 {
973         struct rtsx_pcr *pcr = host->pcr;
974         int err;
975
976         host->power_state = SDMMC_POWER_OFF;
977
978         rtsx_pci_init_cmd(pcr);
979
980         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_CLK_EN, SD_CLK_EN, 0);
981         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_OE, SD_OUTPUT_EN, 0);
982
983         err = rtsx_pci_send_cmd(pcr, 100);
984         if (err < 0)
985                 return err;
986
987         err = rtsx_pci_card_power_off(pcr, RTSX_SD_CARD);
988         if (err < 0)
989                 return err;
990
991         return rtsx_pci_card_pull_ctl_disable(pcr, RTSX_SD_CARD);
992 }
993
994 static int sd_set_power_mode(struct realtek_pci_sdmmc *host,
995                 unsigned char power_mode)
996 {
997         int err;
998
999         if (power_mode == MMC_POWER_OFF)
1000                 err = sd_power_off(host);
1001         else
1002                 err = sd_power_on(host);
1003
1004         return err;
1005 }
1006
1007 static int sd_set_timing(struct realtek_pci_sdmmc *host, unsigned char timing)
1008 {
1009         struct rtsx_pcr *pcr = host->pcr;
1010         int err = 0;
1011
1012         rtsx_pci_init_cmd(pcr);
1013
1014         switch (timing) {
1015         case MMC_TIMING_UHS_SDR104:
1016         case MMC_TIMING_UHS_SDR50:
1017                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_CFG1,
1018                                 0x0C | SD_ASYNC_FIFO_NOT_RST,
1019                                 SD_30_MODE | SD_ASYNC_FIFO_NOT_RST);
1020                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_CTL,
1021                                 CLK_LOW_FREQ, CLK_LOW_FREQ);
1022                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_CLK_SOURCE, 0xFF,
1023                                 CRC_VAR_CLK0 | SD30_FIX_CLK | SAMPLE_VAR_CLK1);
1024                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_CTL, CLK_LOW_FREQ, 0);
1025                 break;
1026
1027         case MMC_TIMING_MMC_DDR52:
1028         case MMC_TIMING_UHS_DDR50:
1029                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_CFG1,
1030                                 0x0C | SD_ASYNC_FIFO_NOT_RST,
1031                                 SD_DDR_MODE | SD_ASYNC_FIFO_NOT_RST);
1032                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_CTL,
1033                                 CLK_LOW_FREQ, CLK_LOW_FREQ);
1034                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_CLK_SOURCE, 0xFF,
1035                                 CRC_VAR_CLK0 | SD30_FIX_CLK | SAMPLE_VAR_CLK1);
1036                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_CTL, CLK_LOW_FREQ, 0);
1037                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_PUSH_POINT_CTL,
1038                                 DDR_VAR_TX_CMD_DAT, DDR_VAR_TX_CMD_DAT);
1039                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_SAMPLE_POINT_CTL,
1040                                 DDR_VAR_RX_DAT | DDR_VAR_RX_CMD,
1041                                 DDR_VAR_RX_DAT | DDR_VAR_RX_CMD);
1042                 break;
1043
1044         case MMC_TIMING_MMC_HS:
1045         case MMC_TIMING_SD_HS:
1046                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_CFG1,
1047                                 0x0C, SD_20_MODE);
1048                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_CTL,
1049                                 CLK_LOW_FREQ, CLK_LOW_FREQ);
1050                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_CLK_SOURCE, 0xFF,
1051                                 CRC_FIX_CLK | SD30_VAR_CLK0 | SAMPLE_VAR_CLK1);
1052                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_CTL, CLK_LOW_FREQ, 0);
1053                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_PUSH_POINT_CTL,
1054                                 SD20_TX_SEL_MASK, SD20_TX_14_AHEAD);
1055                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_SAMPLE_POINT_CTL,
1056                                 SD20_RX_SEL_MASK, SD20_RX_14_DELAY);
1057                 break;
1058
1059         default:
1060                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD,
1061                                 SD_CFG1, 0x0C, SD_20_MODE);
1062                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_CTL,
1063                                 CLK_LOW_FREQ, CLK_LOW_FREQ);
1064                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_CLK_SOURCE, 0xFF,
1065                                 CRC_FIX_CLK | SD30_VAR_CLK0 | SAMPLE_VAR_CLK1);
1066                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_CTL, CLK_LOW_FREQ, 0);
1067                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD,
1068                                 SD_PUSH_POINT_CTL, 0xFF, 0);
1069                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_SAMPLE_POINT_CTL,
1070                                 SD20_RX_SEL_MASK, SD20_RX_POS_EDGE);
1071                 break;
1072         }
1073
1074         err = rtsx_pci_send_cmd(pcr, 100);
1075
1076         return err;
1077 }
1078
1079 static void sdmmc_set_ios(struct mmc_host *mmc, struct mmc_ios *ios)
1080 {
1081         struct realtek_pci_sdmmc *host = mmc_priv(mmc);
1082         struct rtsx_pcr *pcr = host->pcr;
1083
1084         if (host->eject)
1085                 return;
1086
1087         if (rtsx_pci_card_exclusive_check(host->pcr, RTSX_SD_CARD))
1088                 return;
1089
1090         mutex_lock(&pcr->pcr_mutex);
1091
1092         rtsx_pci_start_run(pcr);
1093
1094         sd_set_bus_width(host, ios->bus_width);
1095         sd_set_power_mode(host, ios->power_mode);
1096         sd_set_timing(host, ios->timing);
1097
1098         host->vpclk = false;
1099         host->double_clk = true;
1100
1101         switch (ios->timing) {
1102         case MMC_TIMING_UHS_SDR104:
1103         case MMC_TIMING_UHS_SDR50:
1104                 host->ssc_depth = RTSX_SSC_DEPTH_2M;
1105                 host->vpclk = true;
1106                 host->double_clk = false;
1107                 break;
1108         case MMC_TIMING_MMC_DDR52:
1109         case MMC_TIMING_UHS_DDR50:
1110         case MMC_TIMING_UHS_SDR25:
1111                 host->ssc_depth = RTSX_SSC_DEPTH_1M;
1112                 break;
1113         default:
1114                 host->ssc_depth = RTSX_SSC_DEPTH_500K;
1115                 break;
1116         }
1117
1118         host->initial_mode = (ios->clock <= 1000000) ? true : false;
1119
1120         host->clock = ios->clock;
1121         rtsx_pci_switch_clock(pcr, ios->clock, host->ssc_depth,
1122                         host->initial_mode, host->double_clk, host->vpclk);
1123
1124         mutex_unlock(&pcr->pcr_mutex);
1125 }
1126
1127 static int sdmmc_get_ro(struct mmc_host *mmc)
1128 {
1129         struct realtek_pci_sdmmc *host = mmc_priv(mmc);
1130         struct rtsx_pcr *pcr = host->pcr;
1131         int ro = 0;
1132         u32 val;
1133
1134         if (host->eject)
1135                 return -ENOMEDIUM;
1136
1137         mutex_lock(&pcr->pcr_mutex);
1138
1139         rtsx_pci_start_run(pcr);
1140
1141         /* Check SD mechanical write-protect switch */
1142         val = rtsx_pci_readl(pcr, RTSX_BIPR);
1143         dev_dbg(sdmmc_dev(host), "%s: RTSX_BIPR = 0x%08x\n", __func__, val);
1144         if (val & SD_WRITE_PROTECT)
1145                 ro = 1;
1146
1147         mutex_unlock(&pcr->pcr_mutex);
1148
1149         return ro;
1150 }
1151
1152 static int sdmmc_get_cd(struct mmc_host *mmc)
1153 {
1154         struct realtek_pci_sdmmc *host = mmc_priv(mmc);
1155         struct rtsx_pcr *pcr = host->pcr;
1156         int cd = 0;
1157         u32 val;
1158
1159         if (host->eject)
1160                 return cd;
1161
1162         mutex_lock(&pcr->pcr_mutex);
1163
1164         rtsx_pci_start_run(pcr);
1165
1166         /* Check SD card detect */
1167         val = rtsx_pci_card_exist(pcr);
1168         dev_dbg(sdmmc_dev(host), "%s: RTSX_BIPR = 0x%08x\n", __func__, val);
1169         if (val & SD_EXIST)
1170                 cd = 1;
1171
1172         mutex_unlock(&pcr->pcr_mutex);
1173
1174         return cd;
1175 }
1176
1177 static int sd_wait_voltage_stable_1(struct realtek_pci_sdmmc *host)
1178 {
1179         struct rtsx_pcr *pcr = host->pcr;
1180         int err;
1181         u8 stat;
1182
1183         /* Reference to Signal Voltage Switch Sequence in SD spec.
1184          * Wait for a period of time so that the card can drive SD_CMD and
1185          * SD_DAT[3:0] to low after sending back CMD11 response.
1186          */
1187         mdelay(1);
1188
1189         /* SD_CMD, SD_DAT[3:0] should be driven to low by card;
1190          * If either one of SD_CMD,SD_DAT[3:0] is not low,
1191          * abort the voltage switch sequence;
1192          */
1193         err = rtsx_pci_read_register(pcr, SD_BUS_STAT, &stat);
1194         if (err < 0)
1195                 return err;
1196
1197         if (stat & (SD_CMD_STATUS | SD_DAT3_STATUS | SD_DAT2_STATUS |
1198                                 SD_DAT1_STATUS | SD_DAT0_STATUS))
1199                 return -EINVAL;
1200
1201         /* Stop toggle SD clock */
1202         err = rtsx_pci_write_register(pcr, SD_BUS_STAT,
1203                         0xFF, SD_CLK_FORCE_STOP);
1204         if (err < 0)
1205                 return err;
1206
1207         return 0;
1208 }
1209
1210 static int sd_wait_voltage_stable_2(struct realtek_pci_sdmmc *host)
1211 {
1212         struct rtsx_pcr *pcr = host->pcr;
1213         int err;
1214         u8 stat, mask, val;
1215
1216         /* Wait 1.8V output of voltage regulator in card stable */
1217         msleep(50);
1218
1219         /* Toggle SD clock again */
1220         err = rtsx_pci_write_register(pcr, SD_BUS_STAT, 0xFF, SD_CLK_TOGGLE_EN);
1221         if (err < 0)
1222                 return err;
1223
1224         /* Wait for a period of time so that the card can drive
1225          * SD_DAT[3:0] to high at 1.8V
1226          */
1227         msleep(20);
1228
1229         /* SD_CMD, SD_DAT[3:0] should be pulled high by host */
1230         err = rtsx_pci_read_register(pcr, SD_BUS_STAT, &stat);
1231         if (err < 0)
1232                 return err;
1233
1234         mask = SD_CMD_STATUS | SD_DAT3_STATUS | SD_DAT2_STATUS |
1235                 SD_DAT1_STATUS | SD_DAT0_STATUS;
1236         val = SD_CMD_STATUS | SD_DAT3_STATUS | SD_DAT2_STATUS |
1237                 SD_DAT1_STATUS | SD_DAT0_STATUS;
1238         if ((stat & mask) != val) {
1239                 dev_dbg(sdmmc_dev(host),
1240                         "%s: SD_BUS_STAT = 0x%x\n", __func__, stat);
1241                 rtsx_pci_write_register(pcr, SD_BUS_STAT,
1242                                 SD_CLK_TOGGLE_EN | SD_CLK_FORCE_STOP, 0);
1243                 rtsx_pci_write_register(pcr, CARD_CLK_EN, 0xFF, 0);
1244                 return -EINVAL;
1245         }
1246
1247         return 0;
1248 }
1249
1250 static int sdmmc_switch_voltage(struct mmc_host *mmc, struct mmc_ios *ios)
1251 {
1252         struct realtek_pci_sdmmc *host = mmc_priv(mmc);
1253         struct rtsx_pcr *pcr = host->pcr;
1254         int err = 0;
1255         u8 voltage;
1256
1257         dev_dbg(sdmmc_dev(host), "%s: signal_voltage = %d\n",
1258                         __func__, ios->signal_voltage);
1259
1260         if (host->eject)
1261                 return -ENOMEDIUM;
1262
1263         err = rtsx_pci_card_exclusive_check(host->pcr, RTSX_SD_CARD);
1264         if (err)
1265                 return err;
1266
1267         mutex_lock(&pcr->pcr_mutex);
1268
1269         rtsx_pci_start_run(pcr);
1270
1271         if (ios->signal_voltage == MMC_SIGNAL_VOLTAGE_330)
1272                 voltage = OUTPUT_3V3;
1273         else
1274                 voltage = OUTPUT_1V8;
1275
1276         if (voltage == OUTPUT_1V8) {
1277                 err = sd_wait_voltage_stable_1(host);
1278                 if (err < 0)
1279                         goto out;
1280         }
1281
1282         err = rtsx_pci_switch_output_voltage(pcr, voltage);
1283         if (err < 0)
1284                 goto out;
1285
1286         if (voltage == OUTPUT_1V8) {
1287                 err = sd_wait_voltage_stable_2(host);
1288                 if (err < 0)
1289                         goto out;
1290         }
1291
1292 out:
1293         /* Stop toggle SD clock in idle */
1294         err = rtsx_pci_write_register(pcr, SD_BUS_STAT,
1295                         SD_CLK_TOGGLE_EN | SD_CLK_FORCE_STOP, 0);
1296
1297         mutex_unlock(&pcr->pcr_mutex);
1298
1299         return err;
1300 }
1301
1302 static int sdmmc_execute_tuning(struct mmc_host *mmc, u32 opcode)
1303 {
1304         struct realtek_pci_sdmmc *host = mmc_priv(mmc);
1305         struct rtsx_pcr *pcr = host->pcr;
1306         int err = 0;
1307
1308         if (host->eject)
1309                 return -ENOMEDIUM;
1310
1311         err = rtsx_pci_card_exclusive_check(host->pcr, RTSX_SD_CARD);
1312         if (err)
1313                 return err;
1314
1315         mutex_lock(&pcr->pcr_mutex);
1316
1317         rtsx_pci_start_run(pcr);
1318
1319         /* Set initial TX phase */
1320         switch (mmc->ios.timing) {
1321         case MMC_TIMING_UHS_SDR104:
1322                 err = sd_change_phase(host, SDR104_TX_PHASE(pcr), false);
1323                 break;
1324
1325         case MMC_TIMING_UHS_SDR50:
1326                 err = sd_change_phase(host, SDR50_TX_PHASE(pcr), false);
1327                 break;
1328
1329         case MMC_TIMING_UHS_DDR50:
1330                 err = sd_change_phase(host, DDR50_TX_PHASE(pcr), false);
1331                 break;
1332
1333         default:
1334                 err = 0;
1335         }
1336
1337         if (err)
1338                 goto out;
1339
1340         /* Tuning RX phase */
1341         if ((mmc->ios.timing == MMC_TIMING_UHS_SDR104) ||
1342                         (mmc->ios.timing == MMC_TIMING_UHS_SDR50))
1343                 err = sd_tuning_rx(host, opcode);
1344         else if (mmc->ios.timing == MMC_TIMING_UHS_DDR50)
1345                 err = sd_change_phase(host, DDR50_RX_PHASE(pcr), true);
1346
1347 out:
1348         mutex_unlock(&pcr->pcr_mutex);
1349
1350         return err;
1351 }
1352
1353 static int sdmmc_init_sd_express(struct mmc_host *mmc, struct mmc_ios *ios)
1354 {
1355         u32 relink_time;
1356         struct realtek_pci_sdmmc *host = mmc_priv(mmc);
1357         struct rtsx_pcr *pcr = host->pcr;
1358
1359         /* Set relink_time for changing to PCIe card */
1360         relink_time = 0x8FFF;
1361
1362         rtsx_pci_write_register(pcr, 0xFF01, 0xFF, relink_time);
1363         rtsx_pci_write_register(pcr, 0xFF02, 0xFF, relink_time >> 8);
1364         rtsx_pci_write_register(pcr, 0xFF03, 0x01, relink_time >> 16);
1365
1366         rtsx_pci_write_register(pcr, PETXCFG, 0x80, 0x80);
1367         rtsx_pci_write_register(pcr, LDO_VCC_CFG0,
1368                 RTS5261_LDO1_OCP_THD_MASK,
1369                 pcr->option.sd_800mA_ocp_thd);
1370
1371         if (pcr->ops->disable_auto_blink)
1372                 pcr->ops->disable_auto_blink(pcr);
1373
1374         /* For PCIe/NVMe mode can't enter delink issue */
1375         pcr->hw_param.interrupt_en &= ~(SD_INT_EN);
1376         rtsx_pci_writel(pcr, RTSX_BIER, pcr->hw_param.interrupt_en);
1377
1378         rtsx_pci_write_register(pcr, RTS5260_AUTOLOAD_CFG4,
1379                 RTS5261_AUX_CLK_16M_EN, RTS5261_AUX_CLK_16M_EN);
1380         rtsx_pci_write_register(pcr, RTS5261_FW_CFG0,
1381                 RTS5261_FW_ENTER_EXPRESS, RTS5261_FW_ENTER_EXPRESS);
1382         rtsx_pci_write_register(pcr, RTS5261_FW_CFG1,
1383                 RTS5261_MCU_CLOCK_GATING, RTS5261_MCU_CLOCK_GATING);
1384         rtsx_pci_write_register(pcr, RTS5261_FW_CFG1,
1385                 RTS5261_MCU_BUS_SEL_MASK | RTS5261_MCU_CLOCK_SEL_MASK
1386                 | RTS5261_DRIVER_ENABLE_FW,
1387                 RTS5261_MCU_CLOCK_SEL_16M | RTS5261_DRIVER_ENABLE_FW);
1388         host->eject = true;
1389         return 0;
1390 }
1391
1392 static const struct mmc_host_ops realtek_pci_sdmmc_ops = {
1393         .pre_req = sdmmc_pre_req,
1394         .post_req = sdmmc_post_req,
1395         .request = sdmmc_request,
1396         .set_ios = sdmmc_set_ios,
1397         .get_ro = sdmmc_get_ro,
1398         .get_cd = sdmmc_get_cd,
1399         .start_signal_voltage_switch = sdmmc_switch_voltage,
1400         .execute_tuning = sdmmc_execute_tuning,
1401         .init_sd_express = sdmmc_init_sd_express,
1402 };
1403
1404 static void init_extra_caps(struct realtek_pci_sdmmc *host)
1405 {
1406         struct mmc_host *mmc = host->mmc;
1407         struct rtsx_pcr *pcr = host->pcr;
1408
1409         dev_dbg(sdmmc_dev(host), "pcr->extra_caps = 0x%x\n", pcr->extra_caps);
1410
1411         if (pcr->extra_caps & EXTRA_CAPS_SD_SDR50)
1412                 mmc->caps |= MMC_CAP_UHS_SDR50;
1413         if (pcr->extra_caps & EXTRA_CAPS_SD_SDR104)
1414                 mmc->caps |= MMC_CAP_UHS_SDR104;
1415         if (pcr->extra_caps & EXTRA_CAPS_SD_DDR50)
1416                 mmc->caps |= MMC_CAP_UHS_DDR50;
1417         if (pcr->extra_caps & EXTRA_CAPS_MMC_HSDDR)
1418                 mmc->caps |= MMC_CAP_1_8V_DDR;
1419         if (pcr->extra_caps & EXTRA_CAPS_MMC_8BIT)
1420                 mmc->caps |= MMC_CAP_8_BIT_DATA;
1421         if (pcr->extra_caps & EXTRA_CAPS_NO_MMC)
1422                 mmc->caps2 |= MMC_CAP2_NO_MMC;
1423         if (pcr->extra_caps & EXTRA_CAPS_SD_EXPRESS)
1424                 mmc->caps2 |= MMC_CAP2_SD_EXP | MMC_CAP2_SD_EXP_1_2V;
1425 }
1426
1427 static void realtek_init_host(struct realtek_pci_sdmmc *host)
1428 {
1429         struct mmc_host *mmc = host->mmc;
1430         struct rtsx_pcr *pcr = host->pcr;
1431
1432         mmc->f_min = 250000;
1433         mmc->f_max = 208000000;
1434         mmc->ocr_avail = MMC_VDD_32_33 | MMC_VDD_33_34 | MMC_VDD_165_195;
1435         mmc->caps = MMC_CAP_4_BIT_DATA | MMC_CAP_SD_HIGHSPEED |
1436                 MMC_CAP_MMC_HIGHSPEED | MMC_CAP_BUS_WIDTH_TEST |
1437                 MMC_CAP_UHS_SDR12 | MMC_CAP_UHS_SDR25;
1438         if (pcr->rtd3_en)
1439                 mmc->caps = mmc->caps | MMC_CAP_AGGRESSIVE_PM;
1440         mmc->caps2 = MMC_CAP2_NO_PRESCAN_POWERUP | MMC_CAP2_FULL_PWR_CYCLE |
1441                 MMC_CAP2_NO_SDIO;
1442         mmc->max_current_330 = 400;
1443         mmc->max_current_180 = 800;
1444         mmc->ops = &realtek_pci_sdmmc_ops;
1445
1446         init_extra_caps(host);
1447
1448         mmc->max_segs = 256;
1449         mmc->max_seg_size = 65536;
1450         mmc->max_blk_size = 512;
1451         mmc->max_blk_count = 65535;
1452         mmc->max_req_size = 524288;
1453 }
1454
1455 static void rtsx_pci_sdmmc_card_event(struct platform_device *pdev)
1456 {
1457         struct realtek_pci_sdmmc *host = platform_get_drvdata(pdev);
1458
1459         host->cookie = -1;
1460         mmc_detect_change(host->mmc, 0);
1461 }
1462
1463 static int rtsx_pci_sdmmc_drv_probe(struct platform_device *pdev)
1464 {
1465         struct mmc_host *mmc;
1466         struct realtek_pci_sdmmc *host;
1467         struct rtsx_pcr *pcr;
1468         struct pcr_handle *handle = pdev->dev.platform_data;
1469
1470         if (!handle)
1471                 return -ENXIO;
1472
1473         pcr = handle->pcr;
1474         if (!pcr)
1475                 return -ENXIO;
1476
1477         dev_dbg(&(pdev->dev), ": Realtek PCI-E SDMMC controller found\n");
1478
1479         mmc = mmc_alloc_host(sizeof(*host), &pdev->dev);
1480         if (!mmc)
1481                 return -ENOMEM;
1482
1483         host = mmc_priv(mmc);
1484         host->pcr = pcr;
1485         host->mmc = mmc;
1486         host->pdev = pdev;
1487         host->cookie = -1;
1488         host->power_state = SDMMC_POWER_OFF;
1489         INIT_WORK(&host->work, sd_request);
1490         platform_set_drvdata(pdev, host);
1491         pcr->slots[RTSX_SD_CARD].p_dev = pdev;
1492         pcr->slots[RTSX_SD_CARD].card_event = rtsx_pci_sdmmc_card_event;
1493
1494         mutex_init(&host->host_mutex);
1495
1496         realtek_init_host(host);
1497
1498         if (pcr->rtd3_en) {
1499                 pm_runtime_set_autosuspend_delay(&pdev->dev, 5000);
1500                 pm_runtime_use_autosuspend(&pdev->dev);
1501                 pm_runtime_enable(&pdev->dev);
1502         }
1503
1504
1505         mmc_add_host(mmc);
1506
1507         return 0;
1508 }
1509
1510 static int rtsx_pci_sdmmc_drv_remove(struct platform_device *pdev)
1511 {
1512         struct realtek_pci_sdmmc *host = platform_get_drvdata(pdev);
1513         struct rtsx_pcr *pcr;
1514         struct mmc_host *mmc;
1515
1516         if (!host)
1517                 return 0;
1518
1519         pcr = host->pcr;
1520         pcr->slots[RTSX_SD_CARD].p_dev = NULL;
1521         pcr->slots[RTSX_SD_CARD].card_event = NULL;
1522         mmc = host->mmc;
1523
1524         if (pcr->rtd3_en) {
1525                 pm_runtime_dont_use_autosuspend(&pdev->dev);
1526                 pm_runtime_disable(&pdev->dev);
1527         }
1528
1529         cancel_work_sync(&host->work);
1530
1531         mutex_lock(&host->host_mutex);
1532         if (host->mrq) {
1533                 dev_dbg(&(pdev->dev),
1534                         "%s: Controller removed during transfer\n",
1535                         mmc_hostname(mmc));
1536
1537                 rtsx_pci_complete_unfinished_transfer(pcr);
1538
1539                 host->mrq->cmd->error = -ENOMEDIUM;
1540                 if (host->mrq->stop)
1541                         host->mrq->stop->error = -ENOMEDIUM;
1542                 mmc_request_done(mmc, host->mrq);
1543         }
1544         mutex_unlock(&host->host_mutex);
1545
1546         mmc_remove_host(mmc);
1547         host->eject = true;
1548
1549         flush_work(&host->work);
1550
1551         mmc_free_host(mmc);
1552
1553         dev_dbg(&(pdev->dev),
1554                 ": Realtek PCI-E SDMMC controller has been removed\n");
1555
1556         return 0;
1557 }
1558
1559 static const struct platform_device_id rtsx_pci_sdmmc_ids[] = {
1560         {
1561                 .name = DRV_NAME_RTSX_PCI_SDMMC,
1562         }, {
1563                 /* sentinel */
1564         }
1565 };
1566 MODULE_DEVICE_TABLE(platform, rtsx_pci_sdmmc_ids);
1567
1568 static struct platform_driver rtsx_pci_sdmmc_driver = {
1569         .probe          = rtsx_pci_sdmmc_drv_probe,
1570         .remove         = rtsx_pci_sdmmc_drv_remove,
1571         .id_table       = rtsx_pci_sdmmc_ids,
1572         .driver         = {
1573                 .name   = DRV_NAME_RTSX_PCI_SDMMC,
1574                 .probe_type = PROBE_PREFER_ASYNCHRONOUS,
1575         },
1576 };
1577 module_platform_driver(rtsx_pci_sdmmc_driver);
1578
1579 MODULE_LICENSE("GPL");
1580 MODULE_AUTHOR("Wei WANG <wei_wang@realsil.com.cn>");
1581 MODULE_DESCRIPTION("Realtek PCI-E SD/MMC Card Host Driver");