perf/core: Replace zero-length array with flexible-array
[linux-2.6-microblaze.git] / drivers / misc / cardreader / rtsx_pcr.c
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /* Driver for Realtek PCI-Express card reader
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/pci.h>
11 #include <linux/module.h>
12 #include <linux/slab.h>
13 #include <linux/dma-mapping.h>
14 #include <linux/highmem.h>
15 #include <linux/interrupt.h>
16 #include <linux/delay.h>
17 #include <linux/idr.h>
18 #include <linux/platform_device.h>
19 #include <linux/mfd/core.h>
20 #include <linux/rtsx_pci.h>
21 #include <linux/mmc/card.h>
22 #include <asm/unaligned.h>
23
24 #include "rtsx_pcr.h"
25 #include "rts5261.h"
26
27 static bool msi_en = true;
28 module_param(msi_en, bool, S_IRUGO | S_IWUSR);
29 MODULE_PARM_DESC(msi_en, "Enable MSI");
30
31 static DEFINE_IDR(rtsx_pci_idr);
32 static DEFINE_SPINLOCK(rtsx_pci_lock);
33
34 static struct mfd_cell rtsx_pcr_cells[] = {
35         [RTSX_SD_CARD] = {
36                 .name = DRV_NAME_RTSX_PCI_SDMMC,
37         },
38 };
39
40 static const struct pci_device_id rtsx_pci_ids[] = {
41         { PCI_DEVICE(0x10EC, 0x5209), PCI_CLASS_OTHERS << 16, 0xFF0000 },
42         { PCI_DEVICE(0x10EC, 0x5229), PCI_CLASS_OTHERS << 16, 0xFF0000 },
43         { PCI_DEVICE(0x10EC, 0x5289), PCI_CLASS_OTHERS << 16, 0xFF0000 },
44         { PCI_DEVICE(0x10EC, 0x5227), PCI_CLASS_OTHERS << 16, 0xFF0000 },
45         { PCI_DEVICE(0x10EC, 0x522A), PCI_CLASS_OTHERS << 16, 0xFF0000 },
46         { PCI_DEVICE(0x10EC, 0x5249), PCI_CLASS_OTHERS << 16, 0xFF0000 },
47         { PCI_DEVICE(0x10EC, 0x5287), PCI_CLASS_OTHERS << 16, 0xFF0000 },
48         { PCI_DEVICE(0x10EC, 0x5286), PCI_CLASS_OTHERS << 16, 0xFF0000 },
49         { PCI_DEVICE(0x10EC, 0x524A), PCI_CLASS_OTHERS << 16, 0xFF0000 },
50         { PCI_DEVICE(0x10EC, 0x525A), PCI_CLASS_OTHERS << 16, 0xFF0000 },
51         { PCI_DEVICE(0x10EC, 0x5260), PCI_CLASS_OTHERS << 16, 0xFF0000 },
52         { PCI_DEVICE(0x10EC, 0x5261), PCI_CLASS_OTHERS << 16, 0xFF0000 },
53         { 0, }
54 };
55
56 MODULE_DEVICE_TABLE(pci, rtsx_pci_ids);
57
58 static inline void rtsx_pci_enable_aspm(struct rtsx_pcr *pcr)
59 {
60         rtsx_pci_update_cfg_byte(pcr, pcr->pcie_cap + PCI_EXP_LNKCTL,
61                 0xFC, pcr->aspm_en);
62 }
63
64 static inline void rtsx_pci_disable_aspm(struct rtsx_pcr *pcr)
65 {
66         rtsx_pci_update_cfg_byte(pcr, pcr->pcie_cap + PCI_EXP_LNKCTL,
67                 0xFC, 0);
68 }
69
70 static int rtsx_comm_set_ltr_latency(struct rtsx_pcr *pcr, u32 latency)
71 {
72         rtsx_pci_write_register(pcr, MSGTXDATA0,
73                                 MASK_8_BIT_DEF, (u8) (latency & 0xFF));
74         rtsx_pci_write_register(pcr, MSGTXDATA1,
75                                 MASK_8_BIT_DEF, (u8)((latency >> 8) & 0xFF));
76         rtsx_pci_write_register(pcr, MSGTXDATA2,
77                                 MASK_8_BIT_DEF, (u8)((latency >> 16) & 0xFF));
78         rtsx_pci_write_register(pcr, MSGTXDATA3,
79                                 MASK_8_BIT_DEF, (u8)((latency >> 24) & 0xFF));
80         rtsx_pci_write_register(pcr, LTR_CTL, LTR_TX_EN_MASK |
81                 LTR_LATENCY_MODE_MASK, LTR_TX_EN_1 | LTR_LATENCY_MODE_SW);
82
83         return 0;
84 }
85
86 int rtsx_set_ltr_latency(struct rtsx_pcr *pcr, u32 latency)
87 {
88         if (pcr->ops->set_ltr_latency)
89                 return pcr->ops->set_ltr_latency(pcr, latency);
90         else
91                 return rtsx_comm_set_ltr_latency(pcr, latency);
92 }
93
94 static void rtsx_comm_set_aspm(struct rtsx_pcr *pcr, bool enable)
95 {
96         struct rtsx_cr_option *option = &pcr->option;
97
98         if (pcr->aspm_enabled == enable)
99                 return;
100
101         if (option->dev_aspm_mode == DEV_ASPM_DYNAMIC) {
102                 if (enable)
103                         rtsx_pci_enable_aspm(pcr);
104                 else
105                         rtsx_pci_disable_aspm(pcr);
106         } else if (option->dev_aspm_mode == DEV_ASPM_BACKDOOR) {
107                 u8 mask = FORCE_ASPM_VAL_MASK;
108                 u8 val = 0;
109
110                 if (enable)
111                         val = pcr->aspm_en;
112                 rtsx_pci_write_register(pcr, ASPM_FORCE_CTL,  mask, val);
113         }
114
115         pcr->aspm_enabled = enable;
116 }
117
118 static void rtsx_disable_aspm(struct rtsx_pcr *pcr)
119 {
120         if (pcr->ops->set_aspm)
121                 pcr->ops->set_aspm(pcr, false);
122         else
123                 rtsx_comm_set_aspm(pcr, false);
124 }
125
126 int rtsx_set_l1off_sub(struct rtsx_pcr *pcr, u8 val)
127 {
128         rtsx_pci_write_register(pcr, L1SUB_CONFIG3, 0xFF, val);
129
130         return 0;
131 }
132
133 static void rtsx_set_l1off_sub_cfg_d0(struct rtsx_pcr *pcr, int active)
134 {
135         if (pcr->ops->set_l1off_cfg_sub_d0)
136                 pcr->ops->set_l1off_cfg_sub_d0(pcr, active);
137 }
138
139 static void rtsx_comm_pm_full_on(struct rtsx_pcr *pcr)
140 {
141         struct rtsx_cr_option *option = &pcr->option;
142
143         rtsx_disable_aspm(pcr);
144
145         if (option->ltr_enabled)
146                 rtsx_set_ltr_latency(pcr, option->ltr_active_latency);
147
148         if (rtsx_check_dev_flag(pcr, LTR_L1SS_PWR_GATE_EN))
149                 rtsx_set_l1off_sub_cfg_d0(pcr, 1);
150 }
151
152 static void rtsx_pm_full_on(struct rtsx_pcr *pcr)
153 {
154         if (pcr->ops->full_on)
155                 pcr->ops->full_on(pcr);
156         else
157                 rtsx_comm_pm_full_on(pcr);
158 }
159
160 void rtsx_pci_start_run(struct rtsx_pcr *pcr)
161 {
162         /* If pci device removed, don't queue idle work any more */
163         if (pcr->remove_pci)
164                 return;
165
166         if (pcr->state != PDEV_STAT_RUN) {
167                 pcr->state = PDEV_STAT_RUN;
168                 if (pcr->ops->enable_auto_blink)
169                         pcr->ops->enable_auto_blink(pcr);
170                 rtsx_pm_full_on(pcr);
171         }
172
173         mod_delayed_work(system_wq, &pcr->idle_work, msecs_to_jiffies(200));
174 }
175 EXPORT_SYMBOL_GPL(rtsx_pci_start_run);
176
177 int rtsx_pci_write_register(struct rtsx_pcr *pcr, u16 addr, u8 mask, u8 data)
178 {
179         int i;
180         u32 val = HAIMR_WRITE_START;
181
182         val |= (u32)(addr & 0x3FFF) << 16;
183         val |= (u32)mask << 8;
184         val |= (u32)data;
185
186         rtsx_pci_writel(pcr, RTSX_HAIMR, val);
187
188         for (i = 0; i < MAX_RW_REG_CNT; i++) {
189                 val = rtsx_pci_readl(pcr, RTSX_HAIMR);
190                 if ((val & HAIMR_TRANS_END) == 0) {
191                         if (data != (u8)val)
192                                 return -EIO;
193                         return 0;
194                 }
195         }
196
197         return -ETIMEDOUT;
198 }
199 EXPORT_SYMBOL_GPL(rtsx_pci_write_register);
200
201 int rtsx_pci_read_register(struct rtsx_pcr *pcr, u16 addr, u8 *data)
202 {
203         u32 val = HAIMR_READ_START;
204         int i;
205
206         val |= (u32)(addr & 0x3FFF) << 16;
207         rtsx_pci_writel(pcr, RTSX_HAIMR, val);
208
209         for (i = 0; i < MAX_RW_REG_CNT; i++) {
210                 val = rtsx_pci_readl(pcr, RTSX_HAIMR);
211                 if ((val & HAIMR_TRANS_END) == 0)
212                         break;
213         }
214
215         if (i >= MAX_RW_REG_CNT)
216                 return -ETIMEDOUT;
217
218         if (data)
219                 *data = (u8)(val & 0xFF);
220
221         return 0;
222 }
223 EXPORT_SYMBOL_GPL(rtsx_pci_read_register);
224
225 int __rtsx_pci_write_phy_register(struct rtsx_pcr *pcr, u8 addr, u16 val)
226 {
227         int err, i, finished = 0;
228         u8 tmp;
229
230         rtsx_pci_init_cmd(pcr);
231
232         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, PHYDATA0, 0xFF, (u8)val);
233         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, PHYDATA1, 0xFF, (u8)(val >> 8));
234         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, PHYADDR, 0xFF, addr);
235         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, PHYRWCTL, 0xFF, 0x81);
236
237         err = rtsx_pci_send_cmd(pcr, 100);
238         if (err < 0)
239                 return err;
240
241         for (i = 0; i < 100000; i++) {
242                 err = rtsx_pci_read_register(pcr, PHYRWCTL, &tmp);
243                 if (err < 0)
244                         return err;
245
246                 if (!(tmp & 0x80)) {
247                         finished = 1;
248                         break;
249                 }
250         }
251
252         if (!finished)
253                 return -ETIMEDOUT;
254
255         return 0;
256 }
257
258 int rtsx_pci_write_phy_register(struct rtsx_pcr *pcr, u8 addr, u16 val)
259 {
260         if (pcr->ops->write_phy)
261                 return pcr->ops->write_phy(pcr, addr, val);
262
263         return __rtsx_pci_write_phy_register(pcr, addr, val);
264 }
265 EXPORT_SYMBOL_GPL(rtsx_pci_write_phy_register);
266
267 int __rtsx_pci_read_phy_register(struct rtsx_pcr *pcr, u8 addr, u16 *val)
268 {
269         int err, i, finished = 0;
270         u16 data;
271         u8 *ptr, tmp;
272
273         rtsx_pci_init_cmd(pcr);
274
275         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, PHYADDR, 0xFF, addr);
276         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, PHYRWCTL, 0xFF, 0x80);
277
278         err = rtsx_pci_send_cmd(pcr, 100);
279         if (err < 0)
280                 return err;
281
282         for (i = 0; i < 100000; i++) {
283                 err = rtsx_pci_read_register(pcr, PHYRWCTL, &tmp);
284                 if (err < 0)
285                         return err;
286
287                 if (!(tmp & 0x80)) {
288                         finished = 1;
289                         break;
290                 }
291         }
292
293         if (!finished)
294                 return -ETIMEDOUT;
295
296         rtsx_pci_init_cmd(pcr);
297
298         rtsx_pci_add_cmd(pcr, READ_REG_CMD, PHYDATA0, 0, 0);
299         rtsx_pci_add_cmd(pcr, READ_REG_CMD, PHYDATA1, 0, 0);
300
301         err = rtsx_pci_send_cmd(pcr, 100);
302         if (err < 0)
303                 return err;
304
305         ptr = rtsx_pci_get_cmd_data(pcr);
306         data = ((u16)ptr[1] << 8) | ptr[0];
307
308         if (val)
309                 *val = data;
310
311         return 0;
312 }
313
314 int rtsx_pci_read_phy_register(struct rtsx_pcr *pcr, u8 addr, u16 *val)
315 {
316         if (pcr->ops->read_phy)
317                 return pcr->ops->read_phy(pcr, addr, val);
318
319         return __rtsx_pci_read_phy_register(pcr, addr, val);
320 }
321 EXPORT_SYMBOL_GPL(rtsx_pci_read_phy_register);
322
323 void rtsx_pci_stop_cmd(struct rtsx_pcr *pcr)
324 {
325         if (pcr->ops->stop_cmd)
326                 return pcr->ops->stop_cmd(pcr);
327
328         rtsx_pci_writel(pcr, RTSX_HCBCTLR, STOP_CMD);
329         rtsx_pci_writel(pcr, RTSX_HDBCTLR, STOP_DMA);
330
331         rtsx_pci_write_register(pcr, DMACTL, 0x80, 0x80);
332         rtsx_pci_write_register(pcr, RBCTL, 0x80, 0x80);
333 }
334 EXPORT_SYMBOL_GPL(rtsx_pci_stop_cmd);
335
336 void rtsx_pci_add_cmd(struct rtsx_pcr *pcr,
337                 u8 cmd_type, u16 reg_addr, u8 mask, u8 data)
338 {
339         unsigned long flags;
340         u32 val = 0;
341         u32 *ptr = (u32 *)(pcr->host_cmds_ptr);
342
343         val |= (u32)(cmd_type & 0x03) << 30;
344         val |= (u32)(reg_addr & 0x3FFF) << 16;
345         val |= (u32)mask << 8;
346         val |= (u32)data;
347
348         spin_lock_irqsave(&pcr->lock, flags);
349         ptr += pcr->ci;
350         if (pcr->ci < (HOST_CMDS_BUF_LEN / 4)) {
351                 put_unaligned_le32(val, ptr);
352                 ptr++;
353                 pcr->ci++;
354         }
355         spin_unlock_irqrestore(&pcr->lock, flags);
356 }
357 EXPORT_SYMBOL_GPL(rtsx_pci_add_cmd);
358
359 void rtsx_pci_send_cmd_no_wait(struct rtsx_pcr *pcr)
360 {
361         u32 val = 1 << 31;
362
363         rtsx_pci_writel(pcr, RTSX_HCBAR, pcr->host_cmds_addr);
364
365         val |= (u32)(pcr->ci * 4) & 0x00FFFFFF;
366         /* Hardware Auto Response */
367         val |= 0x40000000;
368         rtsx_pci_writel(pcr, RTSX_HCBCTLR, val);
369 }
370 EXPORT_SYMBOL_GPL(rtsx_pci_send_cmd_no_wait);
371
372 int rtsx_pci_send_cmd(struct rtsx_pcr *pcr, int timeout)
373 {
374         struct completion trans_done;
375         u32 val = 1 << 31;
376         long timeleft;
377         unsigned long flags;
378         int err = 0;
379
380         spin_lock_irqsave(&pcr->lock, flags);
381
382         /* set up data structures for the wakeup system */
383         pcr->done = &trans_done;
384         pcr->trans_result = TRANS_NOT_READY;
385         init_completion(&trans_done);
386
387         rtsx_pci_writel(pcr, RTSX_HCBAR, pcr->host_cmds_addr);
388
389         val |= (u32)(pcr->ci * 4) & 0x00FFFFFF;
390         /* Hardware Auto Response */
391         val |= 0x40000000;
392         rtsx_pci_writel(pcr, RTSX_HCBCTLR, val);
393
394         spin_unlock_irqrestore(&pcr->lock, flags);
395
396         /* Wait for TRANS_OK_INT */
397         timeleft = wait_for_completion_interruptible_timeout(
398                         &trans_done, msecs_to_jiffies(timeout));
399         if (timeleft <= 0) {
400                 pcr_dbg(pcr, "Timeout (%s %d)\n", __func__, __LINE__);
401                 err = -ETIMEDOUT;
402                 goto finish_send_cmd;
403         }
404
405         spin_lock_irqsave(&pcr->lock, flags);
406         if (pcr->trans_result == TRANS_RESULT_FAIL)
407                 err = -EINVAL;
408         else if (pcr->trans_result == TRANS_RESULT_OK)
409                 err = 0;
410         else if (pcr->trans_result == TRANS_NO_DEVICE)
411                 err = -ENODEV;
412         spin_unlock_irqrestore(&pcr->lock, flags);
413
414 finish_send_cmd:
415         spin_lock_irqsave(&pcr->lock, flags);
416         pcr->done = NULL;
417         spin_unlock_irqrestore(&pcr->lock, flags);
418
419         if ((err < 0) && (err != -ENODEV))
420                 rtsx_pci_stop_cmd(pcr);
421
422         if (pcr->finish_me)
423                 complete(pcr->finish_me);
424
425         return err;
426 }
427 EXPORT_SYMBOL_GPL(rtsx_pci_send_cmd);
428
429 static void rtsx_pci_add_sg_tbl(struct rtsx_pcr *pcr,
430                 dma_addr_t addr, unsigned int len, int end)
431 {
432         u64 *ptr = (u64 *)(pcr->host_sg_tbl_ptr) + pcr->sgi;
433         u64 val;
434         u8 option = RTSX_SG_VALID | RTSX_SG_TRANS_DATA;
435
436         pcr_dbg(pcr, "DMA addr: 0x%x, Len: 0x%x\n", (unsigned int)addr, len);
437
438         if (end)
439                 option |= RTSX_SG_END;
440
441         if (PCI_PID(pcr) == PID_5261) {
442                 if (len > 0xFFFF)
443                         val = ((u64)addr << 32) | (((u64)len & 0xFFFF) << 16)
444                                 | (((u64)len >> 16) << 6) | option;
445                 else
446                         val = ((u64)addr << 32) | ((u64)len << 16) | option;
447         } else {
448                 val = ((u64)addr << 32) | ((u64)len << 12) | option;
449         }
450         put_unaligned_le64(val, ptr);
451         pcr->sgi++;
452 }
453
454 int rtsx_pci_transfer_data(struct rtsx_pcr *pcr, struct scatterlist *sglist,
455                 int num_sg, bool read, int timeout)
456 {
457         int err = 0, count;
458
459         pcr_dbg(pcr, "--> %s: num_sg = %d\n", __func__, num_sg);
460         count = rtsx_pci_dma_map_sg(pcr, sglist, num_sg, read);
461         if (count < 1)
462                 return -EINVAL;
463         pcr_dbg(pcr, "DMA mapping count: %d\n", count);
464
465         err = rtsx_pci_dma_transfer(pcr, sglist, count, read, timeout);
466
467         rtsx_pci_dma_unmap_sg(pcr, sglist, num_sg, read);
468
469         return err;
470 }
471 EXPORT_SYMBOL_GPL(rtsx_pci_transfer_data);
472
473 int rtsx_pci_dma_map_sg(struct rtsx_pcr *pcr, struct scatterlist *sglist,
474                 int num_sg, bool read)
475 {
476         enum dma_data_direction dir = read ? DMA_FROM_DEVICE : DMA_TO_DEVICE;
477
478         if (pcr->remove_pci)
479                 return -EINVAL;
480
481         if ((sglist == NULL) || (num_sg <= 0))
482                 return -EINVAL;
483
484         return dma_map_sg(&(pcr->pci->dev), sglist, num_sg, dir);
485 }
486 EXPORT_SYMBOL_GPL(rtsx_pci_dma_map_sg);
487
488 void rtsx_pci_dma_unmap_sg(struct rtsx_pcr *pcr, struct scatterlist *sglist,
489                 int num_sg, bool read)
490 {
491         enum dma_data_direction dir = read ? DMA_FROM_DEVICE : DMA_TO_DEVICE;
492
493         dma_unmap_sg(&(pcr->pci->dev), sglist, num_sg, dir);
494 }
495 EXPORT_SYMBOL_GPL(rtsx_pci_dma_unmap_sg);
496
497 int rtsx_pci_dma_transfer(struct rtsx_pcr *pcr, struct scatterlist *sglist,
498                 int count, bool read, int timeout)
499 {
500         struct completion trans_done;
501         struct scatterlist *sg;
502         dma_addr_t addr;
503         long timeleft;
504         unsigned long flags;
505         unsigned int len;
506         int i, err = 0;
507         u32 val;
508         u8 dir = read ? DEVICE_TO_HOST : HOST_TO_DEVICE;
509
510         if (pcr->remove_pci)
511                 return -ENODEV;
512
513         if ((sglist == NULL) || (count < 1))
514                 return -EINVAL;
515
516         val = ((u32)(dir & 0x01) << 29) | TRIG_DMA | ADMA_MODE;
517         pcr->sgi = 0;
518         for_each_sg(sglist, sg, count, i) {
519                 addr = sg_dma_address(sg);
520                 len = sg_dma_len(sg);
521                 rtsx_pci_add_sg_tbl(pcr, addr, len, i == count - 1);
522         }
523
524         spin_lock_irqsave(&pcr->lock, flags);
525
526         pcr->done = &trans_done;
527         pcr->trans_result = TRANS_NOT_READY;
528         init_completion(&trans_done);
529         rtsx_pci_writel(pcr, RTSX_HDBAR, pcr->host_sg_tbl_addr);
530         rtsx_pci_writel(pcr, RTSX_HDBCTLR, val);
531
532         spin_unlock_irqrestore(&pcr->lock, flags);
533
534         timeleft = wait_for_completion_interruptible_timeout(
535                         &trans_done, msecs_to_jiffies(timeout));
536         if (timeleft <= 0) {
537                 pcr_dbg(pcr, "Timeout (%s %d)\n", __func__, __LINE__);
538                 err = -ETIMEDOUT;
539                 goto out;
540         }
541
542         spin_lock_irqsave(&pcr->lock, flags);
543         if (pcr->trans_result == TRANS_RESULT_FAIL) {
544                 err = -EILSEQ;
545                 if (pcr->dma_error_count < RTS_MAX_TIMES_FREQ_REDUCTION)
546                         pcr->dma_error_count++;
547         }
548
549         else if (pcr->trans_result == TRANS_NO_DEVICE)
550                 err = -ENODEV;
551         spin_unlock_irqrestore(&pcr->lock, flags);
552
553 out:
554         spin_lock_irqsave(&pcr->lock, flags);
555         pcr->done = NULL;
556         spin_unlock_irqrestore(&pcr->lock, flags);
557
558         if ((err < 0) && (err != -ENODEV))
559                 rtsx_pci_stop_cmd(pcr);
560
561         if (pcr->finish_me)
562                 complete(pcr->finish_me);
563
564         return err;
565 }
566 EXPORT_SYMBOL_GPL(rtsx_pci_dma_transfer);
567
568 int rtsx_pci_read_ppbuf(struct rtsx_pcr *pcr, u8 *buf, int buf_len)
569 {
570         int err;
571         int i, j;
572         u16 reg;
573         u8 *ptr;
574
575         if (buf_len > 512)
576                 buf_len = 512;
577
578         ptr = buf;
579         reg = PPBUF_BASE2;
580         for (i = 0; i < buf_len / 256; i++) {
581                 rtsx_pci_init_cmd(pcr);
582
583                 for (j = 0; j < 256; j++)
584                         rtsx_pci_add_cmd(pcr, READ_REG_CMD, reg++, 0, 0);
585
586                 err = rtsx_pci_send_cmd(pcr, 250);
587                 if (err < 0)
588                         return err;
589
590                 memcpy(ptr, rtsx_pci_get_cmd_data(pcr), 256);
591                 ptr += 256;
592         }
593
594         if (buf_len % 256) {
595                 rtsx_pci_init_cmd(pcr);
596
597                 for (j = 0; j < buf_len % 256; j++)
598                         rtsx_pci_add_cmd(pcr, READ_REG_CMD, reg++, 0, 0);
599
600                 err = rtsx_pci_send_cmd(pcr, 250);
601                 if (err < 0)
602                         return err;
603         }
604
605         memcpy(ptr, rtsx_pci_get_cmd_data(pcr), buf_len % 256);
606
607         return 0;
608 }
609 EXPORT_SYMBOL_GPL(rtsx_pci_read_ppbuf);
610
611 int rtsx_pci_write_ppbuf(struct rtsx_pcr *pcr, u8 *buf, int buf_len)
612 {
613         int err;
614         int i, j;
615         u16 reg;
616         u8 *ptr;
617
618         if (buf_len > 512)
619                 buf_len = 512;
620
621         ptr = buf;
622         reg = PPBUF_BASE2;
623         for (i = 0; i < buf_len / 256; i++) {
624                 rtsx_pci_init_cmd(pcr);
625
626                 for (j = 0; j < 256; j++) {
627                         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD,
628                                         reg++, 0xFF, *ptr);
629                         ptr++;
630                 }
631
632                 err = rtsx_pci_send_cmd(pcr, 250);
633                 if (err < 0)
634                         return err;
635         }
636
637         if (buf_len % 256) {
638                 rtsx_pci_init_cmd(pcr);
639
640                 for (j = 0; j < buf_len % 256; j++) {
641                         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD,
642                                         reg++, 0xFF, *ptr);
643                         ptr++;
644                 }
645
646                 err = rtsx_pci_send_cmd(pcr, 250);
647                 if (err < 0)
648                         return err;
649         }
650
651         return 0;
652 }
653 EXPORT_SYMBOL_GPL(rtsx_pci_write_ppbuf);
654
655 static int rtsx_pci_set_pull_ctl(struct rtsx_pcr *pcr, const u32 *tbl)
656 {
657         rtsx_pci_init_cmd(pcr);
658
659         while (*tbl & 0xFFFF0000) {
660                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD,
661                                 (u16)(*tbl >> 16), 0xFF, (u8)(*tbl));
662                 tbl++;
663         }
664
665         return rtsx_pci_send_cmd(pcr, 100);
666 }
667
668 int rtsx_pci_card_pull_ctl_enable(struct rtsx_pcr *pcr, int card)
669 {
670         const u32 *tbl;
671
672         if (card == RTSX_SD_CARD)
673                 tbl = pcr->sd_pull_ctl_enable_tbl;
674         else if (card == RTSX_MS_CARD)
675                 tbl = pcr->ms_pull_ctl_enable_tbl;
676         else
677                 return -EINVAL;
678
679         return rtsx_pci_set_pull_ctl(pcr, tbl);
680 }
681 EXPORT_SYMBOL_GPL(rtsx_pci_card_pull_ctl_enable);
682
683 int rtsx_pci_card_pull_ctl_disable(struct rtsx_pcr *pcr, int card)
684 {
685         const u32 *tbl;
686
687         if (card == RTSX_SD_CARD)
688                 tbl = pcr->sd_pull_ctl_disable_tbl;
689         else if (card == RTSX_MS_CARD)
690                 tbl = pcr->ms_pull_ctl_disable_tbl;
691         else
692                 return -EINVAL;
693
694         return rtsx_pci_set_pull_ctl(pcr, tbl);
695 }
696 EXPORT_SYMBOL_GPL(rtsx_pci_card_pull_ctl_disable);
697
698 static void rtsx_pci_enable_bus_int(struct rtsx_pcr *pcr)
699 {
700         struct rtsx_hw_param *hw_param = &pcr->hw_param;
701
702         pcr->bier = TRANS_OK_INT_EN | TRANS_FAIL_INT_EN | SD_INT_EN
703                 | hw_param->interrupt_en;
704
705         if (pcr->num_slots > 1)
706                 pcr->bier |= MS_INT_EN;
707
708         /* Enable Bus Interrupt */
709         rtsx_pci_writel(pcr, RTSX_BIER, pcr->bier);
710
711         pcr_dbg(pcr, "RTSX_BIER: 0x%08x\n", pcr->bier);
712 }
713
714 static inline u8 double_ssc_depth(u8 depth)
715 {
716         return ((depth > 1) ? (depth - 1) : depth);
717 }
718
719 static u8 revise_ssc_depth(u8 ssc_depth, u8 div)
720 {
721         if (div > CLK_DIV_1) {
722                 if (ssc_depth > (div - 1))
723                         ssc_depth -= (div - 1);
724                 else
725                         ssc_depth = SSC_DEPTH_4M;
726         }
727
728         return ssc_depth;
729 }
730
731 int rtsx_pci_switch_clock(struct rtsx_pcr *pcr, unsigned int card_clock,
732                 u8 ssc_depth, bool initial_mode, bool double_clk, bool vpclk)
733 {
734         int err, clk;
735         u8 n, clk_divider, mcu_cnt, div;
736         static const u8 depth[] = {
737                 [RTSX_SSC_DEPTH_4M] = SSC_DEPTH_4M,
738                 [RTSX_SSC_DEPTH_2M] = SSC_DEPTH_2M,
739                 [RTSX_SSC_DEPTH_1M] = SSC_DEPTH_1M,
740                 [RTSX_SSC_DEPTH_500K] = SSC_DEPTH_500K,
741                 [RTSX_SSC_DEPTH_250K] = SSC_DEPTH_250K,
742         };
743
744         if (PCI_PID(pcr) == PID_5261)
745                 return rts5261_pci_switch_clock(pcr, card_clock,
746                                 ssc_depth, initial_mode, double_clk, vpclk);
747
748         if (initial_mode) {
749                 /* We use 250k(around) here, in initial stage */
750                 clk_divider = SD_CLK_DIVIDE_128;
751                 card_clock = 30000000;
752         } else {
753                 clk_divider = SD_CLK_DIVIDE_0;
754         }
755         err = rtsx_pci_write_register(pcr, SD_CFG1,
756                         SD_CLK_DIVIDE_MASK, clk_divider);
757         if (err < 0)
758                 return err;
759
760         /* Reduce card clock by 20MHz each time a DMA transfer error occurs */
761         if (card_clock == UHS_SDR104_MAX_DTR &&
762             pcr->dma_error_count &&
763             PCI_PID(pcr) == RTS5227_DEVICE_ID)
764                 card_clock = UHS_SDR104_MAX_DTR -
765                         (pcr->dma_error_count * 20000000);
766
767         card_clock /= 1000000;
768         pcr_dbg(pcr, "Switch card clock to %dMHz\n", card_clock);
769
770         clk = card_clock;
771         if (!initial_mode && double_clk)
772                 clk = card_clock * 2;
773         pcr_dbg(pcr, "Internal SSC clock: %dMHz (cur_clock = %d)\n",
774                 clk, pcr->cur_clock);
775
776         if (clk == pcr->cur_clock)
777                 return 0;
778
779         if (pcr->ops->conv_clk_and_div_n)
780                 n = (u8)pcr->ops->conv_clk_and_div_n(clk, CLK_TO_DIV_N);
781         else
782                 n = (u8)(clk - 2);
783         if ((clk <= 2) || (n > MAX_DIV_N_PCR))
784                 return -EINVAL;
785
786         mcu_cnt = (u8)(125/clk + 3);
787         if (mcu_cnt > 15)
788                 mcu_cnt = 15;
789
790         /* Make sure that the SSC clock div_n is not less than MIN_DIV_N_PCR */
791         div = CLK_DIV_1;
792         while ((n < MIN_DIV_N_PCR) && (div < CLK_DIV_8)) {
793                 if (pcr->ops->conv_clk_and_div_n) {
794                         int dbl_clk = pcr->ops->conv_clk_and_div_n(n,
795                                         DIV_N_TO_CLK) * 2;
796                         n = (u8)pcr->ops->conv_clk_and_div_n(dbl_clk,
797                                         CLK_TO_DIV_N);
798                 } else {
799                         n = (n + 2) * 2 - 2;
800                 }
801                 div++;
802         }
803         pcr_dbg(pcr, "n = %d, div = %d\n", n, div);
804
805         ssc_depth = depth[ssc_depth];
806         if (double_clk)
807                 ssc_depth = double_ssc_depth(ssc_depth);
808
809         ssc_depth = revise_ssc_depth(ssc_depth, div);
810         pcr_dbg(pcr, "ssc_depth = %d\n", ssc_depth);
811
812         rtsx_pci_init_cmd(pcr);
813         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_CTL,
814                         CLK_LOW_FREQ, CLK_LOW_FREQ);
815         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_DIV,
816                         0xFF, (div << 4) | mcu_cnt);
817         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SSC_CTL1, SSC_RSTB, 0);
818         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SSC_CTL2,
819                         SSC_DEPTH_MASK, ssc_depth);
820         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SSC_DIV_N_0, 0xFF, n);
821         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SSC_CTL1, SSC_RSTB, SSC_RSTB);
822         if (vpclk) {
823                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_VPCLK0_CTL,
824                                 PHASE_NOT_RESET, 0);
825                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SD_VPCLK0_CTL,
826                                 PHASE_NOT_RESET, PHASE_NOT_RESET);
827         }
828
829         err = rtsx_pci_send_cmd(pcr, 2000);
830         if (err < 0)
831                 return err;
832
833         /* Wait SSC clock stable */
834         udelay(SSC_CLOCK_STABLE_WAIT);
835         err = rtsx_pci_write_register(pcr, CLK_CTL, CLK_LOW_FREQ, 0);
836         if (err < 0)
837                 return err;
838
839         pcr->cur_clock = clk;
840         return 0;
841 }
842 EXPORT_SYMBOL_GPL(rtsx_pci_switch_clock);
843
844 int rtsx_pci_card_power_on(struct rtsx_pcr *pcr, int card)
845 {
846         if (pcr->ops->card_power_on)
847                 return pcr->ops->card_power_on(pcr, card);
848
849         return 0;
850 }
851 EXPORT_SYMBOL_GPL(rtsx_pci_card_power_on);
852
853 int rtsx_pci_card_power_off(struct rtsx_pcr *pcr, int card)
854 {
855         if (pcr->ops->card_power_off)
856                 return pcr->ops->card_power_off(pcr, card);
857
858         return 0;
859 }
860 EXPORT_SYMBOL_GPL(rtsx_pci_card_power_off);
861
862 int rtsx_pci_card_exclusive_check(struct rtsx_pcr *pcr, int card)
863 {
864         static const unsigned int cd_mask[] = {
865                 [RTSX_SD_CARD] = SD_EXIST,
866                 [RTSX_MS_CARD] = MS_EXIST
867         };
868
869         if (!(pcr->flags & PCR_MS_PMOS)) {
870                 /* When using single PMOS, accessing card is not permitted
871                  * if the existing card is not the designated one.
872                  */
873                 if (pcr->card_exist & (~cd_mask[card]))
874                         return -EIO;
875         }
876
877         return 0;
878 }
879 EXPORT_SYMBOL_GPL(rtsx_pci_card_exclusive_check);
880
881 int rtsx_pci_switch_output_voltage(struct rtsx_pcr *pcr, u8 voltage)
882 {
883         if (pcr->ops->switch_output_voltage)
884                 return pcr->ops->switch_output_voltage(pcr, voltage);
885
886         return 0;
887 }
888 EXPORT_SYMBOL_GPL(rtsx_pci_switch_output_voltage);
889
890 unsigned int rtsx_pci_card_exist(struct rtsx_pcr *pcr)
891 {
892         unsigned int val;
893
894         val = rtsx_pci_readl(pcr, RTSX_BIPR);
895         if (pcr->ops->cd_deglitch)
896                 val = pcr->ops->cd_deglitch(pcr);
897
898         return val;
899 }
900 EXPORT_SYMBOL_GPL(rtsx_pci_card_exist);
901
902 void rtsx_pci_complete_unfinished_transfer(struct rtsx_pcr *pcr)
903 {
904         struct completion finish;
905
906         pcr->finish_me = &finish;
907         init_completion(&finish);
908
909         if (pcr->done)
910                 complete(pcr->done);
911
912         if (!pcr->remove_pci)
913                 rtsx_pci_stop_cmd(pcr);
914
915         wait_for_completion_interruptible_timeout(&finish,
916                         msecs_to_jiffies(2));
917         pcr->finish_me = NULL;
918 }
919 EXPORT_SYMBOL_GPL(rtsx_pci_complete_unfinished_transfer);
920
921 static void rtsx_pci_card_detect(struct work_struct *work)
922 {
923         struct delayed_work *dwork;
924         struct rtsx_pcr *pcr;
925         unsigned long flags;
926         unsigned int card_detect = 0, card_inserted, card_removed;
927         u32 irq_status;
928
929         dwork = to_delayed_work(work);
930         pcr = container_of(dwork, struct rtsx_pcr, carddet_work);
931
932         pcr_dbg(pcr, "--> %s\n", __func__);
933
934         mutex_lock(&pcr->pcr_mutex);
935         spin_lock_irqsave(&pcr->lock, flags);
936
937         irq_status = rtsx_pci_readl(pcr, RTSX_BIPR);
938         pcr_dbg(pcr, "irq_status: 0x%08x\n", irq_status);
939
940         irq_status &= CARD_EXIST;
941         card_inserted = pcr->card_inserted & irq_status;
942         card_removed = pcr->card_removed;
943         pcr->card_inserted = 0;
944         pcr->card_removed = 0;
945
946         spin_unlock_irqrestore(&pcr->lock, flags);
947
948         if (card_inserted || card_removed) {
949                 pcr_dbg(pcr, "card_inserted: 0x%x, card_removed: 0x%x\n",
950                         card_inserted, card_removed);
951
952                 if (pcr->ops->cd_deglitch)
953                         card_inserted = pcr->ops->cd_deglitch(pcr);
954
955                 card_detect = card_inserted | card_removed;
956
957                 pcr->card_exist |= card_inserted;
958                 pcr->card_exist &= ~card_removed;
959         }
960
961         mutex_unlock(&pcr->pcr_mutex);
962
963         if ((card_detect & SD_EXIST) && pcr->slots[RTSX_SD_CARD].card_event)
964                 pcr->slots[RTSX_SD_CARD].card_event(
965                                 pcr->slots[RTSX_SD_CARD].p_dev);
966         if ((card_detect & MS_EXIST) && pcr->slots[RTSX_MS_CARD].card_event)
967                 pcr->slots[RTSX_MS_CARD].card_event(
968                                 pcr->slots[RTSX_MS_CARD].p_dev);
969 }
970
971 static void rtsx_pci_process_ocp(struct rtsx_pcr *pcr)
972 {
973         if (pcr->ops->process_ocp) {
974                 pcr->ops->process_ocp(pcr);
975         } else {
976                 if (!pcr->option.ocp_en)
977                         return;
978                 rtsx_pci_get_ocpstat(pcr, &pcr->ocp_stat);
979                 if (pcr->ocp_stat & (SD_OC_NOW | SD_OC_EVER)) {
980                         rtsx_pci_card_power_off(pcr, RTSX_SD_CARD);
981                         rtsx_pci_write_register(pcr, CARD_OE, SD_OUTPUT_EN, 0);
982                         rtsx_pci_clear_ocpstat(pcr);
983                         pcr->ocp_stat = 0;
984                 }
985         }
986 }
987
988 static int rtsx_pci_process_ocp_interrupt(struct rtsx_pcr *pcr)
989 {
990         if (pcr->option.ocp_en)
991                 rtsx_pci_process_ocp(pcr);
992
993         return 0;
994 }
995
996 static irqreturn_t rtsx_pci_isr(int irq, void *dev_id)
997 {
998         struct rtsx_pcr *pcr = dev_id;
999         u32 int_reg;
1000
1001         if (!pcr)
1002                 return IRQ_NONE;
1003
1004         spin_lock(&pcr->lock);
1005
1006         int_reg = rtsx_pci_readl(pcr, RTSX_BIPR);
1007         /* Clear interrupt flag */
1008         rtsx_pci_writel(pcr, RTSX_BIPR, int_reg);
1009         if ((int_reg & pcr->bier) == 0) {
1010                 spin_unlock(&pcr->lock);
1011                 return IRQ_NONE;
1012         }
1013         if (int_reg == 0xFFFFFFFF) {
1014                 spin_unlock(&pcr->lock);
1015                 return IRQ_HANDLED;
1016         }
1017
1018         int_reg &= (pcr->bier | 0x7FFFFF);
1019
1020         if (int_reg & SD_OC_INT)
1021                 rtsx_pci_process_ocp_interrupt(pcr);
1022
1023         if (int_reg & SD_INT) {
1024                 if (int_reg & SD_EXIST) {
1025                         pcr->card_inserted |= SD_EXIST;
1026                 } else {
1027                         pcr->card_removed |= SD_EXIST;
1028                         pcr->card_inserted &= ~SD_EXIST;
1029                 }
1030                 pcr->dma_error_count = 0;
1031         }
1032
1033         if (int_reg & MS_INT) {
1034                 if (int_reg & MS_EXIST) {
1035                         pcr->card_inserted |= MS_EXIST;
1036                 } else {
1037                         pcr->card_removed |= MS_EXIST;
1038                         pcr->card_inserted &= ~MS_EXIST;
1039                 }
1040         }
1041
1042         if (int_reg & (NEED_COMPLETE_INT | DELINK_INT)) {
1043                 if (int_reg & (TRANS_FAIL_INT | DELINK_INT)) {
1044                         pcr->trans_result = TRANS_RESULT_FAIL;
1045                         if (pcr->done)
1046                                 complete(pcr->done);
1047                 } else if (int_reg & TRANS_OK_INT) {
1048                         pcr->trans_result = TRANS_RESULT_OK;
1049                         if (pcr->done)
1050                                 complete(pcr->done);
1051                 }
1052         }
1053
1054         if ((pcr->card_inserted || pcr->card_removed) && !(int_reg & SD_OC_INT))
1055                 schedule_delayed_work(&pcr->carddet_work,
1056                                 msecs_to_jiffies(200));
1057
1058         spin_unlock(&pcr->lock);
1059         return IRQ_HANDLED;
1060 }
1061
1062 static int rtsx_pci_acquire_irq(struct rtsx_pcr *pcr)
1063 {
1064         pcr_dbg(pcr, "%s: pcr->msi_en = %d, pci->irq = %d\n",
1065                         __func__, pcr->msi_en, pcr->pci->irq);
1066
1067         if (request_irq(pcr->pci->irq, rtsx_pci_isr,
1068                         pcr->msi_en ? 0 : IRQF_SHARED,
1069                         DRV_NAME_RTSX_PCI, pcr)) {
1070                 dev_err(&(pcr->pci->dev),
1071                         "rtsx_sdmmc: unable to grab IRQ %d, disabling device\n",
1072                         pcr->pci->irq);
1073                 return -1;
1074         }
1075
1076         pcr->irq = pcr->pci->irq;
1077         pci_intx(pcr->pci, !pcr->msi_en);
1078
1079         return 0;
1080 }
1081
1082 static void rtsx_enable_aspm(struct rtsx_pcr *pcr)
1083 {
1084         if (pcr->ops->set_aspm)
1085                 pcr->ops->set_aspm(pcr, true);
1086         else
1087                 rtsx_comm_set_aspm(pcr, true);
1088 }
1089
1090 static void rtsx_comm_pm_power_saving(struct rtsx_pcr *pcr)
1091 {
1092         struct rtsx_cr_option *option = &pcr->option;
1093
1094         if (option->ltr_enabled) {
1095                 u32 latency = option->ltr_l1off_latency;
1096
1097                 if (rtsx_check_dev_flag(pcr, L1_SNOOZE_TEST_EN))
1098                         mdelay(option->l1_snooze_delay);
1099
1100                 rtsx_set_ltr_latency(pcr, latency);
1101         }
1102
1103         if (rtsx_check_dev_flag(pcr, LTR_L1SS_PWR_GATE_EN))
1104                 rtsx_set_l1off_sub_cfg_d0(pcr, 0);
1105
1106         rtsx_enable_aspm(pcr);
1107 }
1108
1109 static void rtsx_pm_power_saving(struct rtsx_pcr *pcr)
1110 {
1111         if (pcr->ops->power_saving)
1112                 pcr->ops->power_saving(pcr);
1113         else
1114                 rtsx_comm_pm_power_saving(pcr);
1115 }
1116
1117 static void rtsx_pci_idle_work(struct work_struct *work)
1118 {
1119         struct delayed_work *dwork = to_delayed_work(work);
1120         struct rtsx_pcr *pcr = container_of(dwork, struct rtsx_pcr, idle_work);
1121
1122         pcr_dbg(pcr, "--> %s\n", __func__);
1123
1124         mutex_lock(&pcr->pcr_mutex);
1125
1126         pcr->state = PDEV_STAT_IDLE;
1127
1128         if (pcr->ops->disable_auto_blink)
1129                 pcr->ops->disable_auto_blink(pcr);
1130         if (pcr->ops->turn_off_led)
1131                 pcr->ops->turn_off_led(pcr);
1132
1133         rtsx_pm_power_saving(pcr);
1134
1135         mutex_unlock(&pcr->pcr_mutex);
1136 }
1137
1138 #ifdef CONFIG_PM
1139 static void rtsx_pci_power_off(struct rtsx_pcr *pcr, u8 pm_state)
1140 {
1141         if (pcr->ops->turn_off_led)
1142                 pcr->ops->turn_off_led(pcr);
1143
1144         rtsx_pci_writel(pcr, RTSX_BIER, 0);
1145         pcr->bier = 0;
1146
1147         rtsx_pci_write_register(pcr, PETXCFG, 0x08, 0x08);
1148         rtsx_pci_write_register(pcr, HOST_SLEEP_STATE, 0x03, pm_state);
1149
1150         if (pcr->ops->force_power_down)
1151                 pcr->ops->force_power_down(pcr, pm_state);
1152 }
1153 #endif
1154
1155 void rtsx_pci_enable_ocp(struct rtsx_pcr *pcr)
1156 {
1157         u8 val = SD_OCP_INT_EN | SD_DETECT_EN;
1158
1159         if (pcr->ops->enable_ocp) {
1160                 pcr->ops->enable_ocp(pcr);
1161         } else {
1162                 rtsx_pci_write_register(pcr, FPDCTL, OC_POWER_DOWN, 0);
1163                 rtsx_pci_write_register(pcr, REG_OCPCTL, 0xFF, val);
1164         }
1165
1166 }
1167
1168 void rtsx_pci_disable_ocp(struct rtsx_pcr *pcr)
1169 {
1170         u8 mask = SD_OCP_INT_EN | SD_DETECT_EN;
1171
1172         if (pcr->ops->disable_ocp) {
1173                 pcr->ops->disable_ocp(pcr);
1174         } else {
1175                 rtsx_pci_write_register(pcr, REG_OCPCTL, mask, 0);
1176                 rtsx_pci_write_register(pcr, FPDCTL, OC_POWER_DOWN,
1177                                 OC_POWER_DOWN);
1178         }
1179 }
1180
1181 void rtsx_pci_init_ocp(struct rtsx_pcr *pcr)
1182 {
1183         if (pcr->ops->init_ocp) {
1184                 pcr->ops->init_ocp(pcr);
1185         } else {
1186                 struct rtsx_cr_option *option = &(pcr->option);
1187
1188                 if (option->ocp_en) {
1189                         u8 val = option->sd_800mA_ocp_thd;
1190
1191                         rtsx_pci_write_register(pcr, FPDCTL, OC_POWER_DOWN, 0);
1192                         rtsx_pci_write_register(pcr, REG_OCPPARA1,
1193                                 SD_OCP_TIME_MASK, SD_OCP_TIME_800);
1194                         rtsx_pci_write_register(pcr, REG_OCPPARA2,
1195                                 SD_OCP_THD_MASK, val);
1196                         rtsx_pci_write_register(pcr, REG_OCPGLITCH,
1197                                 SD_OCP_GLITCH_MASK, pcr->hw_param.ocp_glitch);
1198                         rtsx_pci_enable_ocp(pcr);
1199                 } else {
1200                         /* OC power down */
1201                         rtsx_pci_write_register(pcr, FPDCTL, OC_POWER_DOWN,
1202                                 OC_POWER_DOWN);
1203                 }
1204         }
1205 }
1206
1207 int rtsx_pci_get_ocpstat(struct rtsx_pcr *pcr, u8 *val)
1208 {
1209         if (pcr->ops->get_ocpstat)
1210                 return pcr->ops->get_ocpstat(pcr, val);
1211         else
1212                 return rtsx_pci_read_register(pcr, REG_OCPSTAT, val);
1213 }
1214
1215 void rtsx_pci_clear_ocpstat(struct rtsx_pcr *pcr)
1216 {
1217         if (pcr->ops->clear_ocpstat) {
1218                 pcr->ops->clear_ocpstat(pcr);
1219         } else {
1220                 u8 mask = SD_OCP_INT_CLR | SD_OC_CLR;
1221                 u8 val = SD_OCP_INT_CLR | SD_OC_CLR;
1222
1223                 rtsx_pci_write_register(pcr, REG_OCPCTL, mask, val);
1224                 udelay(100);
1225                 rtsx_pci_write_register(pcr, REG_OCPCTL, mask, 0);
1226         }
1227 }
1228
1229 int rtsx_sd_power_off_card3v3(struct rtsx_pcr *pcr)
1230 {
1231         rtsx_pci_write_register(pcr, CARD_CLK_EN, SD_CLK_EN |
1232                 MS_CLK_EN | SD40_CLK_EN, 0);
1233         rtsx_pci_write_register(pcr, CARD_OE, SD_OUTPUT_EN, 0);
1234         rtsx_pci_card_power_off(pcr, RTSX_SD_CARD);
1235
1236         msleep(50);
1237
1238         rtsx_pci_card_pull_ctl_disable(pcr, RTSX_SD_CARD);
1239
1240         return 0;
1241 }
1242
1243 int rtsx_ms_power_off_card3v3(struct rtsx_pcr *pcr)
1244 {
1245         rtsx_pci_write_register(pcr, CARD_CLK_EN, SD_CLK_EN |
1246                 MS_CLK_EN | SD40_CLK_EN, 0);
1247
1248         rtsx_pci_card_pull_ctl_disable(pcr, RTSX_MS_CARD);
1249
1250         rtsx_pci_write_register(pcr, CARD_OE, MS_OUTPUT_EN, 0);
1251         rtsx_pci_card_power_off(pcr, RTSX_MS_CARD);
1252
1253         return 0;
1254 }
1255
1256 static int rtsx_pci_init_hw(struct rtsx_pcr *pcr)
1257 {
1258         int err;
1259
1260         pcr->pcie_cap = pci_find_capability(pcr->pci, PCI_CAP_ID_EXP);
1261         rtsx_pci_writel(pcr, RTSX_HCBAR, pcr->host_cmds_addr);
1262
1263         rtsx_pci_enable_bus_int(pcr);
1264
1265         /* Power on SSC */
1266         if (PCI_PID(pcr) == PID_5261) {
1267                 /* Gating real mcu clock */
1268                 err = rtsx_pci_write_register(pcr, RTS5261_FW_CFG1,
1269                         RTS5261_MCU_CLOCK_GATING, 0);
1270                 err = rtsx_pci_write_register(pcr, RTS5261_REG_FPDCTL,
1271                         SSC_POWER_DOWN, 0);
1272         } else {
1273                 err = rtsx_pci_write_register(pcr, FPDCTL, SSC_POWER_DOWN, 0);
1274         }
1275         if (err < 0)
1276                 return err;
1277
1278         /* Wait SSC power stable */
1279         udelay(200);
1280
1281         rtsx_pci_disable_aspm(pcr);
1282         if (pcr->ops->optimize_phy) {
1283                 err = pcr->ops->optimize_phy(pcr);
1284                 if (err < 0)
1285                         return err;
1286         }
1287
1288         rtsx_pci_init_cmd(pcr);
1289
1290         /* Set mcu_cnt to 7 to ensure data can be sampled properly */
1291         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CLK_DIV, 0x07, 0x07);
1292
1293         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, HOST_SLEEP_STATE, 0x03, 0x00);
1294         /* Disable card clock */
1295         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_CLK_EN, 0x1E, 0);
1296         /* Reset delink mode */
1297         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CHANGE_LINK_STATE, 0x0A, 0);
1298         /* Card driving select */
1299         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CARD_DRIVE_SEL,
1300                         0xFF, pcr->card_drive_sel);
1301         /* Enable SSC Clock */
1302         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SSC_CTL1,
1303                         0xFF, SSC_8X_EN | SSC_SEL_4M);
1304         if (PCI_PID(pcr) == PID_5261)
1305                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SSC_CTL2, 0xFF,
1306                         RTS5261_SSC_DEPTH_2M);
1307         else
1308                 rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, SSC_CTL2, 0xFF, 0x12);
1309
1310         /* Disable cd_pwr_save */
1311         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, CHANGE_LINK_STATE, 0x16, 0x10);
1312         /* Clear Link Ready Interrupt */
1313         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, IRQSTAT0,
1314                         LINK_RDY_INT, LINK_RDY_INT);
1315         /* Enlarge the estimation window of PERST# glitch
1316          * to reduce the chance of invalid card interrupt
1317          */
1318         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, PERST_GLITCH_WIDTH, 0xFF, 0x80);
1319         /* Update RC oscillator to 400k
1320          * bit[0] F_HIGH: for RC oscillator, Rst_value is 1'b1
1321          *                1: 2M  0: 400k
1322          */
1323         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, RCCTL, 0x01, 0x00);
1324         /* Set interrupt write clear
1325          * bit 1: U_elbi_if_rd_clr_en
1326          *      1: Enable ELBI interrupt[31:22] & [7:0] flag read clear
1327          *      0: ELBI interrupt flag[31:22] & [7:0] only can be write clear
1328          */
1329         rtsx_pci_add_cmd(pcr, WRITE_REG_CMD, NFTS_TX_CTRL, 0x02, 0);
1330
1331         err = rtsx_pci_send_cmd(pcr, 100);
1332         if (err < 0)
1333                 return err;
1334
1335         switch (PCI_PID(pcr)) {
1336         case PID_5250:
1337         case PID_524A:
1338         case PID_525A:
1339         case PID_5260:
1340         case PID_5261:
1341                 rtsx_pci_write_register(pcr, PM_CLK_FORCE_CTL, 1, 1);
1342                 break;
1343         default:
1344                 break;
1345         }
1346
1347         /*init ocp*/
1348         rtsx_pci_init_ocp(pcr);
1349
1350         /* Enable clk_request_n to enable clock power management */
1351         rtsx_pci_write_config_byte(pcr, pcr->pcie_cap + PCI_EXP_LNKCTL + 1, 1);
1352         /* Enter L1 when host tx idle */
1353         rtsx_pci_write_config_byte(pcr, 0x70F, 0x5B);
1354
1355         if (pcr->ops->extra_init_hw) {
1356                 err = pcr->ops->extra_init_hw(pcr);
1357                 if (err < 0)
1358                         return err;
1359         }
1360
1361         /* No CD interrupt if probing driver with card inserted.
1362          * So we need to initialize pcr->card_exist here.
1363          */
1364         if (pcr->ops->cd_deglitch)
1365                 pcr->card_exist = pcr->ops->cd_deglitch(pcr);
1366         else
1367                 pcr->card_exist = rtsx_pci_readl(pcr, RTSX_BIPR) & CARD_EXIST;
1368
1369         return 0;
1370 }
1371
1372 static int rtsx_pci_init_chip(struct rtsx_pcr *pcr)
1373 {
1374         int err;
1375
1376         spin_lock_init(&pcr->lock);
1377         mutex_init(&pcr->pcr_mutex);
1378
1379         switch (PCI_PID(pcr)) {
1380         default:
1381         case 0x5209:
1382                 rts5209_init_params(pcr);
1383                 break;
1384
1385         case 0x5229:
1386                 rts5229_init_params(pcr);
1387                 break;
1388
1389         case 0x5289:
1390                 rtl8411_init_params(pcr);
1391                 break;
1392
1393         case 0x5227:
1394                 rts5227_init_params(pcr);
1395                 break;
1396
1397         case 0x522A:
1398                 rts522a_init_params(pcr);
1399                 break;
1400
1401         case 0x5249:
1402                 rts5249_init_params(pcr);
1403                 break;
1404
1405         case 0x524A:
1406                 rts524a_init_params(pcr);
1407                 break;
1408
1409         case 0x525A:
1410                 rts525a_init_params(pcr);
1411                 break;
1412
1413         case 0x5287:
1414                 rtl8411b_init_params(pcr);
1415                 break;
1416
1417         case 0x5286:
1418                 rtl8402_init_params(pcr);
1419                 break;
1420
1421         case 0x5260:
1422                 rts5260_init_params(pcr);
1423                 break;
1424
1425         case 0x5261:
1426                 rts5261_init_params(pcr);
1427                 break;
1428         }
1429
1430         pcr_dbg(pcr, "PID: 0x%04x, IC version: 0x%02x\n",
1431                         PCI_PID(pcr), pcr->ic_version);
1432
1433         pcr->slots = kcalloc(pcr->num_slots, sizeof(struct rtsx_slot),
1434                         GFP_KERNEL);
1435         if (!pcr->slots)
1436                 return -ENOMEM;
1437
1438         if (pcr->ops->fetch_vendor_settings)
1439                 pcr->ops->fetch_vendor_settings(pcr);
1440
1441         pcr_dbg(pcr, "pcr->aspm_en = 0x%x\n", pcr->aspm_en);
1442         pcr_dbg(pcr, "pcr->sd30_drive_sel_1v8 = 0x%x\n",
1443                         pcr->sd30_drive_sel_1v8);
1444         pcr_dbg(pcr, "pcr->sd30_drive_sel_3v3 = 0x%x\n",
1445                         pcr->sd30_drive_sel_3v3);
1446         pcr_dbg(pcr, "pcr->card_drive_sel = 0x%x\n",
1447                         pcr->card_drive_sel);
1448         pcr_dbg(pcr, "pcr->flags = 0x%x\n", pcr->flags);
1449
1450         pcr->state = PDEV_STAT_IDLE;
1451         err = rtsx_pci_init_hw(pcr);
1452         if (err < 0) {
1453                 kfree(pcr->slots);
1454                 return err;
1455         }
1456
1457         return 0;
1458 }
1459
1460 static int rtsx_pci_probe(struct pci_dev *pcidev,
1461                           const struct pci_device_id *id)
1462 {
1463         struct rtsx_pcr *pcr;
1464         struct pcr_handle *handle;
1465         u32 base, len;
1466         int ret, i, bar = 0;
1467
1468         dev_dbg(&(pcidev->dev),
1469                 ": Realtek PCI-E Card Reader found at %s [%04x:%04x] (rev %x)\n",
1470                 pci_name(pcidev), (int)pcidev->vendor, (int)pcidev->device,
1471                 (int)pcidev->revision);
1472
1473         ret = pci_set_dma_mask(pcidev, DMA_BIT_MASK(32));
1474         if (ret < 0)
1475                 return ret;
1476
1477         ret = pci_enable_device(pcidev);
1478         if (ret)
1479                 return ret;
1480
1481         ret = pci_request_regions(pcidev, DRV_NAME_RTSX_PCI);
1482         if (ret)
1483                 goto disable;
1484
1485         pcr = kzalloc(sizeof(*pcr), GFP_KERNEL);
1486         if (!pcr) {
1487                 ret = -ENOMEM;
1488                 goto release_pci;
1489         }
1490
1491         handle = kzalloc(sizeof(*handle), GFP_KERNEL);
1492         if (!handle) {
1493                 ret = -ENOMEM;
1494                 goto free_pcr;
1495         }
1496         handle->pcr = pcr;
1497
1498         idr_preload(GFP_KERNEL);
1499         spin_lock(&rtsx_pci_lock);
1500         ret = idr_alloc(&rtsx_pci_idr, pcr, 0, 0, GFP_NOWAIT);
1501         if (ret >= 0)
1502                 pcr->id = ret;
1503         spin_unlock(&rtsx_pci_lock);
1504         idr_preload_end();
1505         if (ret < 0)
1506                 goto free_handle;
1507
1508         pcr->pci = pcidev;
1509         dev_set_drvdata(&pcidev->dev, handle);
1510
1511         if (CHK_PCI_PID(pcr, 0x525A))
1512                 bar = 1;
1513         len = pci_resource_len(pcidev, bar);
1514         base = pci_resource_start(pcidev, bar);
1515         pcr->remap_addr = ioremap(base, len);
1516         if (!pcr->remap_addr) {
1517                 ret = -ENOMEM;
1518                 goto free_handle;
1519         }
1520
1521         pcr->rtsx_resv_buf = dma_alloc_coherent(&(pcidev->dev),
1522                         RTSX_RESV_BUF_LEN, &(pcr->rtsx_resv_buf_addr),
1523                         GFP_KERNEL);
1524         if (pcr->rtsx_resv_buf == NULL) {
1525                 ret = -ENXIO;
1526                 goto unmap;
1527         }
1528         pcr->host_cmds_ptr = pcr->rtsx_resv_buf;
1529         pcr->host_cmds_addr = pcr->rtsx_resv_buf_addr;
1530         pcr->host_sg_tbl_ptr = pcr->rtsx_resv_buf + HOST_CMDS_BUF_LEN;
1531         pcr->host_sg_tbl_addr = pcr->rtsx_resv_buf_addr + HOST_CMDS_BUF_LEN;
1532
1533         pcr->card_inserted = 0;
1534         pcr->card_removed = 0;
1535         INIT_DELAYED_WORK(&pcr->carddet_work, rtsx_pci_card_detect);
1536         INIT_DELAYED_WORK(&pcr->idle_work, rtsx_pci_idle_work);
1537
1538         pcr->msi_en = msi_en;
1539         if (pcr->msi_en) {
1540                 ret = pci_enable_msi(pcidev);
1541                 if (ret)
1542                         pcr->msi_en = false;
1543         }
1544
1545         ret = rtsx_pci_acquire_irq(pcr);
1546         if (ret < 0)
1547                 goto disable_msi;
1548
1549         pci_set_master(pcidev);
1550         synchronize_irq(pcr->irq);
1551
1552         ret = rtsx_pci_init_chip(pcr);
1553         if (ret < 0)
1554                 goto disable_irq;
1555
1556         for (i = 0; i < ARRAY_SIZE(rtsx_pcr_cells); i++) {
1557                 rtsx_pcr_cells[i].platform_data = handle;
1558                 rtsx_pcr_cells[i].pdata_size = sizeof(*handle);
1559         }
1560         ret = mfd_add_devices(&pcidev->dev, pcr->id, rtsx_pcr_cells,
1561                         ARRAY_SIZE(rtsx_pcr_cells), NULL, 0, NULL);
1562         if (ret < 0)
1563                 goto disable_irq;
1564
1565         schedule_delayed_work(&pcr->idle_work, msecs_to_jiffies(200));
1566
1567         return 0;
1568
1569 disable_irq:
1570         free_irq(pcr->irq, (void *)pcr);
1571 disable_msi:
1572         if (pcr->msi_en)
1573                 pci_disable_msi(pcr->pci);
1574         dma_free_coherent(&(pcr->pci->dev), RTSX_RESV_BUF_LEN,
1575                         pcr->rtsx_resv_buf, pcr->rtsx_resv_buf_addr);
1576 unmap:
1577         iounmap(pcr->remap_addr);
1578 free_handle:
1579         kfree(handle);
1580 free_pcr:
1581         kfree(pcr);
1582 release_pci:
1583         pci_release_regions(pcidev);
1584 disable:
1585         pci_disable_device(pcidev);
1586
1587         return ret;
1588 }
1589
1590 static void rtsx_pci_remove(struct pci_dev *pcidev)
1591 {
1592         struct pcr_handle *handle = pci_get_drvdata(pcidev);
1593         struct rtsx_pcr *pcr = handle->pcr;
1594
1595         pcr->remove_pci = true;
1596
1597         /* Disable interrupts at the pcr level */
1598         spin_lock_irq(&pcr->lock);
1599         rtsx_pci_writel(pcr, RTSX_BIER, 0);
1600         pcr->bier = 0;
1601         spin_unlock_irq(&pcr->lock);
1602
1603         cancel_delayed_work_sync(&pcr->carddet_work);
1604         cancel_delayed_work_sync(&pcr->idle_work);
1605
1606         mfd_remove_devices(&pcidev->dev);
1607
1608         dma_free_coherent(&(pcr->pci->dev), RTSX_RESV_BUF_LEN,
1609                         pcr->rtsx_resv_buf, pcr->rtsx_resv_buf_addr);
1610         free_irq(pcr->irq, (void *)pcr);
1611         if (pcr->msi_en)
1612                 pci_disable_msi(pcr->pci);
1613         iounmap(pcr->remap_addr);
1614
1615         pci_release_regions(pcidev);
1616         pci_disable_device(pcidev);
1617
1618         spin_lock(&rtsx_pci_lock);
1619         idr_remove(&rtsx_pci_idr, pcr->id);
1620         spin_unlock(&rtsx_pci_lock);
1621
1622         kfree(pcr->slots);
1623         kfree(pcr);
1624         kfree(handle);
1625
1626         dev_dbg(&(pcidev->dev),
1627                 ": Realtek PCI-E Card Reader at %s [%04x:%04x] has been removed\n",
1628                 pci_name(pcidev), (int)pcidev->vendor, (int)pcidev->device);
1629 }
1630
1631 #ifdef CONFIG_PM
1632
1633 static int rtsx_pci_suspend(struct pci_dev *pcidev, pm_message_t state)
1634 {
1635         struct pcr_handle *handle;
1636         struct rtsx_pcr *pcr;
1637
1638         dev_dbg(&(pcidev->dev), "--> %s\n", __func__);
1639
1640         handle = pci_get_drvdata(pcidev);
1641         pcr = handle->pcr;
1642
1643         cancel_delayed_work(&pcr->carddet_work);
1644         cancel_delayed_work(&pcr->idle_work);
1645
1646         mutex_lock(&pcr->pcr_mutex);
1647
1648         rtsx_pci_power_off(pcr, HOST_ENTER_S3);
1649
1650         pci_save_state(pcidev);
1651         pci_enable_wake(pcidev, pci_choose_state(pcidev, state), 0);
1652         pci_disable_device(pcidev);
1653         pci_set_power_state(pcidev, pci_choose_state(pcidev, state));
1654
1655         mutex_unlock(&pcr->pcr_mutex);
1656         return 0;
1657 }
1658
1659 static int rtsx_pci_resume(struct pci_dev *pcidev)
1660 {
1661         struct pcr_handle *handle;
1662         struct rtsx_pcr *pcr;
1663         int ret = 0;
1664
1665         dev_dbg(&(pcidev->dev), "--> %s\n", __func__);
1666
1667         handle = pci_get_drvdata(pcidev);
1668         pcr = handle->pcr;
1669
1670         mutex_lock(&pcr->pcr_mutex);
1671
1672         pci_set_power_state(pcidev, PCI_D0);
1673         pci_restore_state(pcidev);
1674         ret = pci_enable_device(pcidev);
1675         if (ret)
1676                 goto out;
1677         pci_set_master(pcidev);
1678
1679         ret = rtsx_pci_write_register(pcr, HOST_SLEEP_STATE, 0x03, 0x00);
1680         if (ret)
1681                 goto out;
1682
1683         ret = rtsx_pci_init_hw(pcr);
1684         if (ret)
1685                 goto out;
1686
1687         schedule_delayed_work(&pcr->idle_work, msecs_to_jiffies(200));
1688
1689 out:
1690         mutex_unlock(&pcr->pcr_mutex);
1691         return ret;
1692 }
1693
1694 static void rtsx_pci_shutdown(struct pci_dev *pcidev)
1695 {
1696         struct pcr_handle *handle;
1697         struct rtsx_pcr *pcr;
1698
1699         dev_dbg(&(pcidev->dev), "--> %s\n", __func__);
1700
1701         handle = pci_get_drvdata(pcidev);
1702         pcr = handle->pcr;
1703         rtsx_pci_power_off(pcr, HOST_ENTER_S1);
1704
1705         pci_disable_device(pcidev);
1706         free_irq(pcr->irq, (void *)pcr);
1707         if (pcr->msi_en)
1708                 pci_disable_msi(pcr->pci);
1709 }
1710
1711 #else /* CONFIG_PM */
1712
1713 #define rtsx_pci_suspend NULL
1714 #define rtsx_pci_resume NULL
1715 #define rtsx_pci_shutdown NULL
1716
1717 #endif /* CONFIG_PM */
1718
1719 static struct pci_driver rtsx_pci_driver = {
1720         .name = DRV_NAME_RTSX_PCI,
1721         .id_table = rtsx_pci_ids,
1722         .probe = rtsx_pci_probe,
1723         .remove = rtsx_pci_remove,
1724         .suspend = rtsx_pci_suspend,
1725         .resume = rtsx_pci_resume,
1726         .shutdown = rtsx_pci_shutdown,
1727 };
1728 module_pci_driver(rtsx_pci_driver);
1729
1730 MODULE_LICENSE("GPL");
1731 MODULE_AUTHOR("Wei WANG <wei_wang@realsil.com.cn>");
1732 MODULE_DESCRIPTION("Realtek PCI-E Card Reader Driver");