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