rockchip: iommu: skip rk312x vop mmu stall operation,mask rk312x vop mmu interrupt
[firefly-linux-kernel-4.4.55.git] / drivers / iommu / rockchip-iommu.c
1 /*
2  * This program is free software; you can redistribute it and/or modify
3  * it under the terms of the GNU General Public License version 2 as
4  * published by the Free Software Foundation.
5  */
6
7 #ifdef CONFIG_ROCKCHIP_IOMMU_DEBUG
8 #define DEBUG
9 #endif
10
11 #include <linux/io.h>
12 #include <linux/interrupt.h>
13 #include <linux/slab.h>
14 #include <linux/clk.h>
15 #include <linux/err.h>
16 #include <linux/mm.h>
17 #include <linux/errno.h>
18 #include <linux/memblock.h>
19 #include <linux/export.h>
20 #include <linux/module.h>
21
22 #include <asm/cacheflush.h>
23 #include <asm/pgtable.h>
24 #include <linux/of.h>
25 #include <linux/rockchip-iovmm.h>
26 #include <linux/rockchip/grf.h>
27 #include <linux/rockchip/cpu.h>
28 #include <linux/rockchip/iomap.h>
29 #include <linux/device.h>
30 #include "rockchip-iommu.h"
31
32 /* We does not consider super section mapping (16MB) */
33 #define SPAGE_ORDER 12
34 #define SPAGE_SIZE (1 << SPAGE_ORDER)
35 #define SPAGE_MASK (~(SPAGE_SIZE - 1))
36
37 static void __iomem *rk312x_vop_mmu_base;
38
39 enum iommu_entry_flags {
40         IOMMU_FLAGS_PRESENT = 0x01,
41         IOMMU_FLAGS_READ_PERMISSION = 0x02,
42         IOMMU_FLAGS_WRITE_PERMISSION = 0x04,
43         IOMMU_FLAGS_OVERRIDE_CACHE = 0x8,
44         IOMMU_FLAGS_WRITE_CACHEABLE = 0x10,
45         IOMMU_FLAGS_WRITE_ALLOCATE = 0x20,
46         IOMMU_FLAGS_WRITE_BUFFERABLE = 0x40,
47         IOMMU_FLAGS_READ_CACHEABLE = 0x80,
48         IOMMU_FLAGS_READ_ALLOCATE = 0x100,
49         IOMMU_FLAGS_MASK = 0x1FF,
50 };
51
52 #define rockchip_lv1ent_fault(sent) ((*(sent) & IOMMU_FLAGS_PRESENT) == 0)
53 #define rockchip_lv1ent_page(sent) ((*(sent) & IOMMU_FLAGS_PRESENT) == 1)
54 #define rockchip_lv2ent_fault(pent) ((*(pent) & IOMMU_FLAGS_PRESENT) == 0)
55 #define rockchip_spage_phys(pent) (*(pent) & SPAGE_MASK)
56 #define rockchip_spage_offs(iova) ((iova) & 0x0FFF)
57
58 #define rockchip_lv1ent_offset(iova) (((iova)>>22) & 0x03FF)
59 #define rockchip_lv2ent_offset(iova) (((iova)>>12) & 0x03FF)
60
61 #define NUM_LV1ENTRIES 1024
62 #define NUM_LV2ENTRIES 1024
63
64 #define LV2TABLE_SIZE (NUM_LV2ENTRIES * sizeof(long))
65
66 #define rockchip_lv2table_base(sent) (*(sent) & 0xFFFFFFFE)
67
68 #define rockchip_mk_lv1ent_page(pa) ((pa) | IOMMU_FLAGS_PRESENT)
69 /*write and read permission for level2 page default*/
70 #define rockchip_mk_lv2ent_spage(pa) ((pa) | IOMMU_FLAGS_PRESENT | \
71                              IOMMU_FLAGS_READ_PERMISSION | \
72                              IOMMU_FLAGS_WRITE_PERMISSION)
73
74 #define IOMMU_REG_POLL_COUNT_FAST 1000
75
76 /**
77  * MMU register numbers
78  * Used in the register read/write routines.
79  * See the hardware documentation for more information about each register
80  */
81 enum iommu_register {
82         /**< Current Page Directory Pointer */
83         IOMMU_REGISTER_DTE_ADDR = 0x0000,
84         /**< Status of the MMU */
85         IOMMU_REGISTER_STATUS = 0x0004,
86         /**< Command register, used to control the MMU */
87         IOMMU_REGISTER_COMMAND = 0x0008,
88         /**< Logical address of the last page fault */
89         IOMMU_REGISTER_PAGE_FAULT_ADDR = 0x000C,
90         /**< Used to invalidate the mapping of a single page from the MMU */
91         IOMMU_REGISTER_ZAP_ONE_LINE = 0x010,
92         /**< Raw interrupt status, all interrupts visible */
93         IOMMU_REGISTER_INT_RAWSTAT = 0x0014,
94         /**< Indicate to the MMU that the interrupt has been received */
95         IOMMU_REGISTER_INT_CLEAR = 0x0018,
96         /**< Enable/disable types of interrupts */
97         IOMMU_REGISTER_INT_MASK = 0x001C,
98         /**< Interrupt status based on the mask */
99         IOMMU_REGISTER_INT_STATUS = 0x0020,
100         IOMMU_REGISTER_AUTO_GATING = 0x0024
101 };
102
103 enum iommu_command {
104         /**< Enable paging (memory translation) */
105         IOMMU_COMMAND_ENABLE_PAGING = 0x00,
106         /**< Disable paging (memory translation) */
107         IOMMU_COMMAND_DISABLE_PAGING = 0x01,
108         /**<  Enable stall on page fault */
109         IOMMU_COMMAND_ENABLE_STALL = 0x02,
110         /**< Disable stall on page fault */
111         IOMMU_COMMAND_DISABLE_STALL = 0x03,
112         /**< Zap the entire page table cache */
113         IOMMU_COMMAND_ZAP_CACHE = 0x04,
114         /**< Page fault processed */
115         IOMMU_COMMAND_PAGE_FAULT_DONE = 0x05,
116         /**< Reset the MMU back to power-on settings */
117         IOMMU_COMMAND_HARD_RESET = 0x06
118 };
119
120 /**
121  * MMU interrupt register bits
122  * Each cause of the interrupt is reported
123  * through the (raw) interrupt status registers.
124  * Multiple interrupts can be pending, so multiple bits
125  * can be set at once.
126  */
127 enum iommu_interrupt {
128         IOMMU_INTERRUPT_PAGE_FAULT = 0x01, /**< A page fault occured */
129         IOMMU_INTERRUPT_READ_BUS_ERROR = 0x02 /**< A bus read error occured */
130 };
131
132 enum iommu_status_bits {
133         IOMMU_STATUS_BIT_PAGING_ENABLED      = 1 << 0,
134         IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE   = 1 << 1,
135         IOMMU_STATUS_BIT_STALL_ACTIVE        = 1 << 2,
136         IOMMU_STATUS_BIT_IDLE                = 1 << 3,
137         IOMMU_STATUS_BIT_REPLAY_BUFFER_EMPTY = 1 << 4,
138         IOMMU_STATUS_BIT_PAGE_FAULT_IS_WRITE = 1 << 5,
139         IOMMU_STATUS_BIT_STALL_NOT_ACTIVE    = 1 << 31,
140 };
141
142 /**
143  * Size of an MMU page in bytes
144  */
145 #define IOMMU_PAGE_SIZE 0x1000
146
147 /*
148  * Size of the address space referenced by a page table page
149  */
150 #define IOMMU_VIRTUAL_PAGE_SIZE 0x400000 /* 4 MiB */
151
152 /**
153  * Page directory index from address
154  * Calculates the page directory index from the given address
155  */
156 #define IOMMU_PDE_ENTRY(address) (((address)>>22) & 0x03FF)
157
158 /**
159  * Page table index from address
160  * Calculates the page table index from the given address
161  */
162 #define IOMMU_PTE_ENTRY(address) (((address)>>12) & 0x03FF)
163
164 /**
165  * Extract the memory address from an PDE/PTE entry
166  */
167 #define IOMMU_ENTRY_ADDRESS(value) ((value) & 0xFFFFFC00)
168
169 #define INVALID_PAGE ((u32)(~0))
170
171 static struct kmem_cache *lv2table_kmem_cache;
172
173 static unsigned long *rockchip_section_entry(unsigned long *pgtable, unsigned long iova)
174 {
175         return pgtable + rockchip_lv1ent_offset(iova);
176 }
177
178 static unsigned long *rockchip_page_entry(unsigned long *sent, unsigned long iova)
179 {
180         return (unsigned long *)__va(rockchip_lv2table_base(sent)) +
181                 rockchip_lv2ent_offset(iova);
182 }
183
184 struct rk_iommu_domain {
185         struct list_head clients; /* list of iommu_drvdata.node */
186         unsigned long *pgtable; /* lv1 page table, 4KB */
187         short *lv2entcnt; /* free lv2 entry counter for each section */
188         spinlock_t lock; /* lock for this structure */
189         spinlock_t pgtablelock; /* lock for modifying page table @ pgtable */
190 };
191
192 static bool rockchip_set_iommu_active(struct iommu_drvdata *data)
193 {
194         /* return true if the IOMMU was not active previously
195            and it needs to be initialized */
196         return ++data->activations == 1;
197 }
198
199 static bool rockchip_set_iommu_inactive(struct iommu_drvdata *data)
200 {
201         /* return true if the IOMMU is needed to be disabled */
202         BUG_ON(data->activations < 1);
203         return --data->activations == 0;
204 }
205
206 static bool rockchip_is_iommu_active(struct iommu_drvdata *data)
207 {
208         return data->activations > 0;
209 }
210
211 static void rockchip_iommu_disable_stall(void __iomem *base)
212 {
213         int i;
214         u32 mmu_status;
215
216         if (base != rk312x_vop_mmu_base) {
217                 mmu_status = __raw_readl(base + IOMMU_REGISTER_STATUS);
218         } else {
219                 goto skip_vop_mmu_disable;
220         }
221
222         if (0 == (mmu_status & IOMMU_STATUS_BIT_PAGING_ENABLED)) {
223                 return;
224         }
225
226         if (mmu_status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE) {
227                 pr_info("Aborting MMU disable stall request since it is in pagefault state.\n");
228                 return;
229         }
230
231         if (!(mmu_status & IOMMU_STATUS_BIT_STALL_ACTIVE)) {
232                 return;
233         }
234
235         __raw_writel(IOMMU_COMMAND_DISABLE_STALL, base + IOMMU_REGISTER_COMMAND);
236
237         skip_vop_mmu_disable:
238
239         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
240                 u32 status;
241                 
242                 if (base != rk312x_vop_mmu_base) {
243                         status = __raw_readl(base + IOMMU_REGISTER_STATUS);
244                 } else {
245                         int j;
246                         while (j < 5)
247                                 j++;
248                         return; 
249                 }
250
251                 if (0 == (status & IOMMU_STATUS_BIT_STALL_ACTIVE))
252                         break;
253
254                 if (status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE)
255                         break;
256
257                 if (0 == (mmu_status & IOMMU_STATUS_BIT_PAGING_ENABLED))
258                         break;
259         }
260
261         if (IOMMU_REG_POLL_COUNT_FAST == i) {
262                 pr_info("Disable stall request failed, MMU status is 0x%08X\n",
263                       __raw_readl(base + IOMMU_REGISTER_STATUS));
264         }
265 }
266
267 static bool rockchip_iommu_enable_stall(void __iomem *base)
268 {
269         int i;
270
271         u32 mmu_status;
272         
273         if (base != rk312x_vop_mmu_base) {
274                 mmu_status = __raw_readl(base + IOMMU_REGISTER_STATUS);
275         } else {
276                 goto skip_vop_mmu_enable;
277         }
278
279         if (0 == (mmu_status & IOMMU_STATUS_BIT_PAGING_ENABLED)) {
280                 return true;
281         }
282
283         if (mmu_status & IOMMU_STATUS_BIT_STALL_ACTIVE){
284                 pr_info("MMU stall already enabled\n");
285                 return true;
286         }
287
288         if (mmu_status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE) {
289                 pr_info("Aborting MMU stall request since it is in pagefault state. mmu status is 0x%08x\n",
290                         mmu_status);
291                 return false;
292         }
293
294         __raw_writel(IOMMU_COMMAND_ENABLE_STALL, base + IOMMU_REGISTER_COMMAND);
295
296         skip_vop_mmu_enable:
297
298         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
299                 if (base != rk312x_vop_mmu_base) {
300                         mmu_status = __raw_readl(base + IOMMU_REGISTER_STATUS);
301                 } else {
302                         int j;
303                         while (j < 5)
304                                 j++;
305                         return true;
306                 }
307
308                 if (mmu_status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE)
309                         break;
310
311                 if ((mmu_status & IOMMU_STATUS_BIT_STALL_ACTIVE) &&
312                     (0 == (mmu_status & IOMMU_STATUS_BIT_STALL_NOT_ACTIVE)))
313                         break;
314
315                 if (0 == (mmu_status & (IOMMU_STATUS_BIT_PAGING_ENABLED)))
316                         break;
317         }
318
319         if (IOMMU_REG_POLL_COUNT_FAST == i) {
320                 pr_info("Enable stall request failed, MMU status is 0x%08X\n",
321                        __raw_readl(base + IOMMU_REGISTER_STATUS));
322                 return false;
323         }
324
325         if (mmu_status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE) {
326                 pr_info("Aborting MMU stall request since it has a pagefault.\n");
327                 return false;
328         }
329
330         return true;
331 }
332
333 static bool rockchip_iommu_enable_paging(void __iomem *base)
334 {
335         int i;
336
337         __raw_writel(IOMMU_COMMAND_ENABLE_PAGING,
338                      base + IOMMU_REGISTER_COMMAND);
339
340         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
341                 if (base != rk312x_vop_mmu_base) {
342                         if (__raw_readl(base + IOMMU_REGISTER_STATUS) &
343                                 IOMMU_STATUS_BIT_PAGING_ENABLED)
344                         break;
345                 } else {
346                         int j;
347                         while (j < 5)
348                                 j++;
349                         return true;
350                 }
351         }
352
353         if (IOMMU_REG_POLL_COUNT_FAST == i) {
354                 pr_info("Enable paging request failed, MMU status is 0x%08X\n",
355                        __raw_readl(base + IOMMU_REGISTER_STATUS));
356                 return false;
357         }
358
359         return true;
360 }
361
362 static bool rockchip_iommu_disable_paging(void __iomem *base)
363 {
364         int i;
365
366         __raw_writel(IOMMU_COMMAND_DISABLE_PAGING,
367                      base + IOMMU_REGISTER_COMMAND);
368
369         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
370                 if (base != rk312x_vop_mmu_base) {
371                         if (!(__raw_readl(base + IOMMU_REGISTER_STATUS) &
372                                   IOMMU_STATUS_BIT_PAGING_ENABLED))
373                                 break;
374                 } else {
375                         int j;
376                         while (j < 5)
377                                 j++;
378                         return true;
379                 }
380         }
381
382         if (IOMMU_REG_POLL_COUNT_FAST == i) {
383                 pr_info("Disable paging request failed, MMU status is 0x%08X\n",
384                        __raw_readl(base + IOMMU_REGISTER_STATUS));
385                 return false;
386         }
387
388         return true;
389 }
390
391 static void rockchip_iommu_page_fault_done(void __iomem *base, const char *dbgname)
392 {
393         pr_info("MMU: %s: Leaving page fault mode\n",
394                 dbgname);
395         __raw_writel(IOMMU_COMMAND_PAGE_FAULT_DONE,
396                      base + IOMMU_REGISTER_COMMAND);
397 }
398
399 static int rockchip_iommu_zap_tlb_without_stall (void __iomem *base)
400 {
401         __raw_writel(IOMMU_COMMAND_ZAP_CACHE, base + IOMMU_REGISTER_COMMAND);
402
403         return 0;
404 }
405
406 static int rockchip_iommu_zap_tlb(void __iomem *base)
407 {
408         if (!rockchip_iommu_enable_stall(base)) {
409                 pr_err("%s failed\n", __func__);
410                 return -1;
411         }
412
413         __raw_writel(IOMMU_COMMAND_ZAP_CACHE, base + IOMMU_REGISTER_COMMAND);
414
415         rockchip_iommu_disable_stall(base);
416
417         return 0;
418 }
419
420 static inline bool rockchip_iommu_raw_reset(void __iomem *base)
421 {
422         int i;
423         unsigned int ret;
424         unsigned int grf_value;
425
426         __raw_writel(0xCAFEBABE, base + IOMMU_REGISTER_DTE_ADDR);
427
428         if (base != rk312x_vop_mmu_base) {
429                 ret = __raw_readl(base + IOMMU_REGISTER_DTE_ADDR);
430                 if (!(0xCAFEB000 == ret)) {
431                         grf_value = readl_relaxed(RK_GRF_VIRT + RK3036_GRF_SOC_CON1);
432                         pr_info("error when %s. grf = 0x%08x\n", __func__, grf_value);
433                         return false;
434                 }
435         }
436         __raw_writel(IOMMU_COMMAND_HARD_RESET,
437                      base + IOMMU_REGISTER_COMMAND);
438
439         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
440                 if (base != rk312x_vop_mmu_base) {
441                         if (__raw_readl(base + IOMMU_REGISTER_DTE_ADDR) == 0)
442                                 break;
443                 } else {
444                         int j;
445                         while (j < 5)
446                                 j++;
447                         return true;
448                 }
449         }
450
451         if (IOMMU_REG_POLL_COUNT_FAST == i) {
452                 pr_info("%s,Reset request failed, MMU status is 0x%08X\n",
453                        __func__, __raw_readl(base + IOMMU_REGISTER_DTE_ADDR));
454                 return false;
455         }
456         return true;
457 }
458
459 static void rockchip_iommu_set_ptbase(void __iomem *base, unsigned long pgd)
460 {
461         __raw_writel(pgd, base + IOMMU_REGISTER_DTE_ADDR);
462 }
463
464 static bool rockchip_iommu_reset(void __iomem *base, const char *dbgname)
465 {
466         bool ret = true;
467
468         ret = rockchip_iommu_raw_reset(base);
469         if (!ret) {
470                 pr_info("(%s), %s failed\n", dbgname, __func__);
471                 return ret;
472         }
473
474         if (base != rk312x_vop_mmu_base)
475                 __raw_writel(IOMMU_INTERRUPT_PAGE_FAULT |
476                              IOMMU_INTERRUPT_READ_BUS_ERROR,
477                              base + IOMMU_REGISTER_INT_MASK);
478         else
479                 __raw_writel(0x00, base + IOMMU_REGISTER_INT_MASK);
480
481         return ret;
482 }
483
484 static inline void rockchip_pgtable_flush(void *vastart, void *vaend)
485 {
486         dmac_flush_range(vastart, vaend);
487         outer_flush_range(virt_to_phys(vastart), virt_to_phys(vaend));
488 }
489
490 static void dump_pagetbl(dma_addr_t fault_address, u32 addr_dte)
491 {
492         u32 dte_index, pte_index, page_offset;
493         u32 mmu_dte_addr;
494         phys_addr_t mmu_dte_addr_phys, dte_addr_phys;
495         u32 *dte_addr;
496         u32 dte;
497         phys_addr_t pte_addr_phys = 0;
498         u32 *pte_addr = NULL;
499         u32 pte = 0;
500         phys_addr_t page_addr_phys = 0;
501         u32 page_flags = 0;
502
503         dte_index = rockchip_lv1ent_offset(fault_address);
504         pte_index = rockchip_lv2ent_offset(fault_address);
505         page_offset = (u32)(fault_address & 0x00000fff);
506
507         mmu_dte_addr = addr_dte;
508         mmu_dte_addr_phys = (phys_addr_t)mmu_dte_addr;
509
510         dte_addr_phys = mmu_dte_addr_phys + (4 * dte_index);
511         dte_addr = phys_to_virt(dte_addr_phys);
512         dte = *dte_addr;
513
514         if (!(IOMMU_FLAGS_PRESENT & dte))
515                 goto print_it;
516
517         pte_addr_phys = ((phys_addr_t)dte & 0xfffff000) + (pte_index * 4);
518         pte_addr = phys_to_virt(pte_addr_phys);
519         pte = *pte_addr;
520
521         if (!(IOMMU_FLAGS_PRESENT & pte))
522                 goto print_it;
523
524         page_addr_phys = ((phys_addr_t)pte & 0xfffff000) + page_offset;
525         page_flags = pte & 0x000001fe;
526
527 print_it:
528         pr_err("iova = %pad: dte_index: 0x%03x pte_index: 0x%03x page_offset: 0x%03x\n",
529                 &fault_address, dte_index, pte_index, page_offset);
530         pr_err("mmu_dte_addr: %pa dte@%pa: %#08x valid: %u pte@%pa: %#08x valid: %u page@%pa flags: %#03x\n",
531                 &mmu_dte_addr_phys, &dte_addr_phys, dte,
532                 (dte & IOMMU_FLAGS_PRESENT), &pte_addr_phys, pte,
533                 (pte & IOMMU_FLAGS_PRESENT), &page_addr_phys, page_flags);
534 }
535
536 static irqreturn_t rockchip_iommu_irq(int irq, void *dev_id)
537 {
538         /* SYSMMU is in blocked when interrupt occurred. */
539         struct iommu_drvdata *data = dev_id;
540         u32 status;
541         u32 rawstat;
542         dma_addr_t fault_address;
543         int i;
544         unsigned long flags;
545         int ret;
546         u32 reg_status;
547
548         spin_lock_irqsave(&data->data_lock, flags);
549
550         if (!rockchip_is_iommu_active(data)) {
551                 spin_unlock_irqrestore(&data->data_lock, flags);
552                 return IRQ_HANDLED;
553         }
554
555         for (i = 0; i < data->num_res_mem; i++) {
556                 status = __raw_readl(data->res_bases[i] +
557                                      IOMMU_REGISTER_INT_STATUS);
558                 if (status == 0)
559                         continue;
560
561                 rawstat = __raw_readl(data->res_bases[i] +
562                                       IOMMU_REGISTER_INT_RAWSTAT);
563
564                 reg_status = __raw_readl(data->res_bases[i] +
565                                          IOMMU_REGISTER_STATUS);
566
567                 dev_info(data->iommu, "1.rawstat = 0x%08x,status = 0x%08x,reg_status = 0x%08x\n",
568                          rawstat, status, reg_status);
569
570                 if (rawstat & IOMMU_INTERRUPT_PAGE_FAULT) {
571                         u32 dte;
572                         int flags;
573
574                         fault_address = __raw_readl(data->res_bases[i] +
575                                             IOMMU_REGISTER_PAGE_FAULT_ADDR);
576
577                         dte = __raw_readl(data->res_bases[i] +
578                                           IOMMU_REGISTER_DTE_ADDR);
579
580                         flags = (status & 32) ? 1 : 0;
581
582                         dev_err(data->iommu, "Page fault detected at %pad from bus id %d of type %s on %s\n",
583                                 &fault_address, (status >> 6) & 0x1F,
584                                 (flags == 1) ? "write" : "read", data->dbgname);
585
586                         dump_pagetbl(fault_address, dte);
587
588                         if (data->domain)
589                                 report_iommu_fault(data->domain, data->iommu,
590                                                    fault_address, flags);
591
592                         rockchip_iommu_page_fault_done(data->res_bases[i],
593                                                        data->dbgname);
594                 }
595
596                 if (rawstat & IOMMU_INTERRUPT_READ_BUS_ERROR) {
597                         dev_err(data->iommu, "bus error occured at %pad\n",
598                                 &fault_address);
599                 }
600
601                 if (rawstat & ~(IOMMU_INTERRUPT_READ_BUS_ERROR |
602                     IOMMU_INTERRUPT_PAGE_FAULT)) {
603                         dev_err(data->iommu, "unexpected int_status: %#08x\n\n",
604                                 rawstat);
605                 }
606
607                 __raw_writel(rawstat, data->res_bases[i] +
608                              IOMMU_REGISTER_INT_CLEAR);
609
610                 status = __raw_readl(data->res_bases[i] +
611                                      IOMMU_REGISTER_INT_STATUS);
612
613                 rawstat = __raw_readl(data->res_bases[i] +
614                                       IOMMU_REGISTER_INT_RAWSTAT);
615
616                 reg_status = __raw_readl(data->res_bases[i] +
617                                          IOMMU_REGISTER_STATUS);
618
619                 dev_info(data->iommu, "2.rawstat = 0x%08x,status = 0x%08x,reg_status = 0x%08x\n",
620                          rawstat, status, reg_status);
621
622                 ret = rockchip_iommu_zap_tlb_without_stall(data->res_bases[i]);
623                 if (ret)
624                         dev_err(data->iommu, "(%s) %s failed\n", data->dbgname,
625                                 __func__);
626         }
627
628         spin_unlock_irqrestore(&data->data_lock, flags);
629         return IRQ_HANDLED;
630 }
631
632 static bool rockchip_iommu_disable(struct iommu_drvdata *data)
633 {
634         unsigned long flags;
635         int i;
636         bool ret = false;
637
638         spin_lock_irqsave(&data->data_lock, flags);
639
640         if (!rockchip_set_iommu_inactive(data)) {
641                 spin_unlock_irqrestore(&data->data_lock, flags);
642                 dev_info(data->iommu,"(%s) %d times left to be disabled\n",
643                          data->dbgname, data->activations);
644                 return ret;
645         }
646
647         for (i = 0; i < data->num_res_mem; i++) {
648                 ret = rockchip_iommu_enable_stall(data->res_bases[i]);
649                 if (!ret) {
650                         dev_info(data->iommu, "(%s), %s failed\n",
651                                  data->dbgname, __func__);
652                         spin_unlock_irqrestore(&data->data_lock, flags);
653                         return false;
654                 }
655
656                 __raw_writel(0, data->res_bases[i] + IOMMU_REGISTER_INT_MASK);
657
658                 ret = rockchip_iommu_disable_paging(data->res_bases[i]);
659                 if (!ret) {
660                         rockchip_iommu_disable_stall(data->res_bases[i]);
661                         spin_unlock_irqrestore(&data->data_lock, flags);
662                         dev_info(data->iommu, "%s error\n", __func__);
663                         return ret;
664                 }
665                 rockchip_iommu_disable_stall(data->res_bases[i]);
666         }
667
668         data->pgtable = 0;
669
670         spin_unlock_irqrestore(&data->data_lock, flags);
671
672         dev_dbg(data->iommu,"(%s) Disabled\n", data->dbgname);
673
674         return ret;
675 }
676
677 /* __rk_sysmmu_enable: Enables System MMU
678  *
679  * returns -error if an error occurred and System MMU is not enabled,
680  * 0 if the System MMU has been just enabled and 1 if System MMU was already
681  * enabled before.
682  */
683 static int rockchip_iommu_enable(struct iommu_drvdata *data, unsigned long pgtable)
684 {
685         int i, ret = 0;
686         unsigned long flags;
687
688         spin_lock_irqsave(&data->data_lock, flags);
689
690         if (!rockchip_set_iommu_active(data)) {
691                 if (WARN_ON(pgtable != data->pgtable)) {
692                         ret = -EBUSY;
693                         rockchip_set_iommu_inactive(data);
694                 } else {
695                         ret = 1;
696                 }
697
698                 spin_unlock_irqrestore(&data->data_lock, flags);
699                 dev_info(data->iommu, "(%s) Already enabled\n", data->dbgname);
700
701                 return ret;
702         }
703
704         for (i = 0; i < data->num_res_mem; i++) {
705                 ret = rockchip_iommu_enable_stall(data->res_bases[i]);
706                 if (!ret) {
707                         dev_info(data->iommu, "(%s), %s failed\n",
708                                  data->dbgname, __func__);
709                         spin_unlock_irqrestore(&data->data_lock, flags);
710                         return -EBUSY;
711                 }
712
713                 if (!strstr(data->dbgname, "isp")) {
714                         if (!rockchip_iommu_reset(data->res_bases[i],
715                              data->dbgname)) {
716                                 spin_unlock_irqrestore(&data->data_lock, flags);
717                                 return -ENOENT;
718                         }
719                 }
720
721                 rockchip_iommu_set_ptbase(data->res_bases[i], pgtable);
722
723                 __raw_writel(IOMMU_COMMAND_ZAP_CACHE, data->res_bases[i] +
724                              IOMMU_REGISTER_COMMAND);
725
726                 if (strstr(data->dbgname, "isp")) {
727                         __raw_writel(IOMMU_INTERRUPT_PAGE_FAULT |
728                                 IOMMU_INTERRUPT_READ_BUS_ERROR,
729                              data->res_bases[i] + IOMMU_REGISTER_INT_MASK);
730                 }
731
732                 ret = rockchip_iommu_enable_paging(data->res_bases[i]);
733                 if (!ret) {
734                         spin_unlock_irqrestore(&data->data_lock, flags);
735                         dev_info(data->iommu, "(%s), %s failed\n",
736                                  data->dbgname, __func__);
737                         return -EBUSY;
738                 }
739
740                 rockchip_iommu_disable_stall(data->res_bases[i]);
741         }
742
743         data->pgtable = pgtable;
744
745         dev_dbg(data->iommu,"(%s) Enabled\n", data->dbgname);
746
747         spin_unlock_irqrestore(&data->data_lock, flags);
748
749         return 0;
750 }
751
752 int rockchip_iommu_tlb_invalidate(struct device *dev)
753 {
754         unsigned long flags;
755         struct iommu_drvdata *data = dev_get_drvdata(dev->archdata.iommu);
756
757         spin_lock_irqsave(&data->data_lock, flags);
758
759         if (rockchip_is_iommu_active(data)) {
760                 int i;
761                 int ret;
762
763                 for (i = 0; i < data->num_res_mem; i++) {
764                         ret = rockchip_iommu_zap_tlb(data->res_bases[i]);
765                         if (ret) {
766                                 dev_err(dev->archdata.iommu, "(%s) %s failed\n",
767                                         data->dbgname, __func__);
768                                 spin_unlock_irqrestore(&data->data_lock, flags);
769                                 return ret;
770                         }
771                                 
772                 }
773         } else {
774                 dev_dbg(dev->archdata.iommu, "(%s) Disabled. Skipping invalidating TLB.\n",
775                         data->dbgname);
776         }
777
778         spin_unlock_irqrestore(&data->data_lock, flags);
779
780         return 0;
781 }
782
783 static phys_addr_t rockchip_iommu_iova_to_phys(struct iommu_domain *domain,
784                                                dma_addr_t iova)
785 {
786         struct rk_iommu_domain *priv = domain->priv;
787         unsigned long *entry;
788         unsigned long flags;
789         phys_addr_t phys = 0;
790
791         spin_lock_irqsave(&priv->pgtablelock, flags);
792
793         entry = rockchip_section_entry(priv->pgtable, iova);
794         entry = rockchip_page_entry(entry, iova);
795         phys = rockchip_spage_phys(entry) + rockchip_spage_offs(iova);
796
797         spin_unlock_irqrestore(&priv->pgtablelock, flags);
798
799         return phys;
800 }
801
802 static int rockchip_lv2set_page(unsigned long *pent, phys_addr_t paddr,
803                        size_t size, short *pgcnt)
804 {
805         if (!rockchip_lv2ent_fault(pent))
806                 return -EADDRINUSE;
807
808         *pent = rockchip_mk_lv2ent_spage(paddr);
809         rockchip_pgtable_flush(pent, pent + 1);
810         *pgcnt -= 1;
811         return 0;
812 }
813
814 static unsigned long *rockchip_alloc_lv2entry(unsigned long *sent,
815                                      unsigned long iova, short *pgcounter)
816 {
817         if (rockchip_lv1ent_fault(sent)) {
818                 unsigned long *pent;
819
820                 pent = kmem_cache_zalloc(lv2table_kmem_cache, GFP_ATOMIC);
821                 BUG_ON((unsigned long)pent & (LV2TABLE_SIZE - 1));
822                 if (!pent)
823                         return NULL;
824
825                 *sent = rockchip_mk_lv1ent_page(__pa(pent));
826                 kmemleak_ignore(pent);
827                 *pgcounter = NUM_LV2ENTRIES;
828                 rockchip_pgtable_flush(pent, pent + NUM_LV2ENTRIES);
829                 rockchip_pgtable_flush(sent, sent + 1);
830         }
831         return rockchip_page_entry(sent, iova);
832 }
833
834 static size_t rockchip_iommu_unmap(struct iommu_domain *domain,
835                                    unsigned long iova, size_t size)
836 {
837         struct rk_iommu_domain *priv = domain->priv;
838         unsigned long flags;
839         unsigned long *ent;
840
841         BUG_ON(priv->pgtable == NULL);
842
843         spin_lock_irqsave(&priv->pgtablelock, flags);
844
845         ent = rockchip_section_entry(priv->pgtable, iova);
846
847         if (unlikely(rockchip_lv1ent_fault(ent))) {
848                 if (size > SPAGE_SIZE)
849                         size = SPAGE_SIZE;
850                 goto done;
851         }
852
853         /* lv1ent_page(sent) == true here */
854
855         ent = rockchip_page_entry(ent, iova);
856
857         if (unlikely(rockchip_lv2ent_fault(ent))) {
858                 size = SPAGE_SIZE;
859                 goto done;
860         }
861
862         *ent = 0;
863         size = SPAGE_SIZE;
864         priv->lv2entcnt[rockchip_lv1ent_offset(iova)] += 1;
865         goto done;
866
867 done:
868         #if 0
869         pr_info("%s:unmap iova 0x%lx/0x%x bytes\n",
870                   __func__, iova,size);
871         #endif
872         spin_unlock_irqrestore(&priv->pgtablelock, flags);
873
874         return size;
875 }
876
877 static int rockchip_iommu_map(struct iommu_domain *domain, unsigned long iova,
878                               phys_addr_t paddr, size_t size, int prot)
879 {
880         struct rk_iommu_domain *priv = domain->priv;
881         unsigned long *entry;
882         unsigned long flags;
883         int ret = -ENOMEM;
884         unsigned long *pent;
885
886         BUG_ON(priv->pgtable == NULL);
887
888         spin_lock_irqsave(&priv->pgtablelock, flags);
889
890         entry = rockchip_section_entry(priv->pgtable, iova);
891
892         pent = rockchip_alloc_lv2entry(entry, iova,
893                               &priv->lv2entcnt[rockchip_lv1ent_offset(iova)]);
894         if (!pent)
895                 ret = -ENOMEM;
896         else
897                 ret = rockchip_lv2set_page(pent, paddr, size,
898                                 &priv->lv2entcnt[rockchip_lv1ent_offset(iova)]);
899
900         if (ret) {
901                 pr_info("%s: Failed to map iova 0x%lx/0x%x bytes\n", __func__,
902                        iova, size);
903         }
904         spin_unlock_irqrestore(&priv->pgtablelock, flags);
905
906         return ret;
907 }
908
909 static void rockchip_iommu_detach_device(struct iommu_domain *domain, struct device *dev)
910 {
911         struct iommu_drvdata *data = dev_get_drvdata(dev->archdata.iommu);
912         struct rk_iommu_domain *priv = domain->priv;
913         struct list_head *pos;
914         unsigned long flags;
915         bool found = false;
916
917         spin_lock_irqsave(&priv->lock, flags);
918
919         list_for_each(pos, &priv->clients) {
920                 if (list_entry(pos, struct iommu_drvdata, node) == data) {
921                         found = true;
922                         break;
923                 }
924         }
925
926         if (!found) {
927                 spin_unlock_irqrestore(&priv->lock, flags);
928                 return;
929         }
930
931         if (rockchip_iommu_disable(data)) {
932                 dev_dbg(dev->archdata.iommu,"%s: Detached IOMMU with pgtable %#lx\n",
933                         __func__, __pa(priv->pgtable));
934                 data->domain = NULL;
935                 list_del_init(&data->node);
936
937         } else
938                 dev_err(dev->archdata.iommu,"%s: Detaching IOMMU with pgtable %#lx delayed",
939                         __func__, __pa(priv->pgtable));
940
941         spin_unlock_irqrestore(&priv->lock, flags);
942 }
943
944 static int rockchip_iommu_attach_device(struct iommu_domain *domain, struct device *dev)
945 {
946         struct iommu_drvdata *data = dev_get_drvdata(dev->archdata.iommu);
947         struct rk_iommu_domain *priv = domain->priv;
948         unsigned long flags;
949         int ret;
950
951         spin_lock_irqsave(&priv->lock, flags);
952
953         ret = rockchip_iommu_enable(data, __pa(priv->pgtable));
954
955         if (ret == 0) {
956                 /* 'data->node' must not be appeared in priv->clients */
957                 BUG_ON(!list_empty(&data->node));
958                 list_add_tail(&data->node, &priv->clients);
959                 data->domain = domain;
960         }
961
962         spin_unlock_irqrestore(&priv->lock, flags);
963
964         if (ret < 0) {
965                 dev_err(dev->archdata.iommu,"%s: Failed to attach IOMMU with pgtable %#lx\n",
966                        __func__, __pa(priv->pgtable));
967         } else if (ret > 0) {
968                 dev_dbg(dev->archdata.iommu,"%s: IOMMU with pgtable 0x%lx already attached\n",
969                         __func__, __pa(priv->pgtable));
970         } else {
971                 dev_dbg(dev->archdata.iommu,"%s: Attached new IOMMU with pgtable 0x%lx\n",
972                         __func__, __pa(priv->pgtable));
973         }
974
975         return ret;
976 }
977
978 static void rockchip_iommu_domain_destroy(struct iommu_domain *domain)
979 {
980         struct rk_iommu_domain *priv = domain->priv;
981         int i;
982
983         WARN_ON(!list_empty(&priv->clients));
984
985         for (i = 0; i < NUM_LV1ENTRIES; i++)
986                 if (rockchip_lv1ent_page(priv->pgtable + i))
987                         kmem_cache_free(lv2table_kmem_cache,
988                                         __va(rockchip_lv2table_base(priv->pgtable + i)));
989
990         free_pages((unsigned long)priv->pgtable, 0);
991         free_pages((unsigned long)priv->lv2entcnt, 0);
992         kfree(domain->priv);
993         domain->priv = NULL;
994 }
995
996 static int rockchip_iommu_domain_init(struct iommu_domain *domain)
997 {
998         struct rk_iommu_domain *priv;
999
1000         priv = kzalloc(sizeof(*priv), GFP_KERNEL);
1001         if (!priv)
1002                 return -ENOMEM;
1003
1004 /*rk32xx iommu use 2 level pagetable,
1005    level1 and leve2 both have 1024 entries,each entry  occupy 4 bytes,
1006    so alloc a page size for each page table
1007 */
1008         priv->pgtable = (unsigned long *)__get_free_pages(GFP_KERNEL |
1009                                                           __GFP_ZERO, 0);
1010         if (!priv->pgtable)
1011                 goto err_pgtable;
1012
1013         priv->lv2entcnt = (short *)__get_free_pages(GFP_KERNEL |
1014                                                     __GFP_ZERO, 0);
1015         if (!priv->lv2entcnt)
1016                 goto err_counter;
1017
1018         rockchip_pgtable_flush(priv->pgtable, priv->pgtable + NUM_LV1ENTRIES);
1019
1020         spin_lock_init(&priv->lock);
1021         spin_lock_init(&priv->pgtablelock);
1022         INIT_LIST_HEAD(&priv->clients);
1023
1024         domain->priv = priv;
1025         return 0;
1026
1027 err_counter:
1028         free_pages((unsigned long)priv->pgtable, 0);
1029 err_pgtable:
1030         kfree(priv);
1031         return -ENOMEM;
1032 }
1033
1034 static struct iommu_ops rk_iommu_ops = {
1035         .domain_init = &rockchip_iommu_domain_init,
1036         .domain_destroy = &rockchip_iommu_domain_destroy,
1037         .attach_dev = &rockchip_iommu_attach_device,
1038         .detach_dev = &rockchip_iommu_detach_device,
1039         .map = &rockchip_iommu_map,
1040         .unmap = &rockchip_iommu_unmap,
1041         .iova_to_phys = &rockchip_iommu_iova_to_phys,
1042         .pgsize_bitmap = SPAGE_SIZE,
1043 };
1044
1045 static int  rockchip_get_iommu_resource_num(struct platform_device *pdev,
1046                                              unsigned int type)
1047 {
1048         int num = 0;
1049         int i;
1050
1051         for (i = 0; i < pdev->num_resources; i++) {
1052                 struct resource *r = &pdev->resource[i];
1053                 if (type == resource_type(r))
1054                         num++;
1055         }
1056
1057         return num;
1058 }
1059
1060 static int rockchip_iommu_probe(struct platform_device *pdev)
1061 {
1062         int i, ret;
1063         struct device *dev;
1064         struct iommu_drvdata *data;
1065         
1066         dev = &pdev->dev;
1067
1068         data = devm_kzalloc(dev, sizeof(*data), GFP_KERNEL);
1069         if (!data) {
1070                 dev_dbg(dev, "Not enough memory\n");
1071                 return -ENOMEM;
1072         }
1073
1074         dev_set_drvdata(dev, data);
1075
1076         if (pdev->dev.of_node)
1077                 of_property_read_string(pdev->dev.of_node, "dbgname",
1078                                         &(data->dbgname));
1079         else
1080                 dev_dbg(dev, "dbgname not assigned in device tree or device node not exist\r\n");
1081
1082         dev_info(dev,"(%s) Enter\n", data->dbgname);
1083         
1084         data->num_res_mem = rockchip_get_iommu_resource_num(pdev,
1085                                 IORESOURCE_MEM);
1086         if (0 == data->num_res_mem) {
1087                 dev_err(dev,"can't find iommu memory resource \r\n");
1088                 return -ENOMEM;
1089         }
1090         dev_dbg(dev,"data->num_res_mem=%d\n", data->num_res_mem);
1091
1092         data->num_res_irq = rockchip_get_iommu_resource_num(pdev,
1093                                 IORESOURCE_IRQ);
1094         if (0 == data->num_res_irq) {
1095                 dev_err(dev,"can't find iommu irq resource \r\n");
1096                 return -ENOMEM;
1097         }
1098         dev_dbg(dev,"data->num_res_irq=%d\n", data->num_res_irq);
1099
1100         data->res_bases = devm_kmalloc_array(dev, data->num_res_mem,
1101                                 sizeof(*data->res_bases), GFP_KERNEL);
1102         if (data->res_bases == NULL) {
1103                 dev_err(dev, "Not enough memory\n");
1104                 return -ENOMEM;
1105         }
1106
1107         for (i = 0; i < data->num_res_mem; i++) {
1108                 struct resource *res;
1109
1110                 res = platform_get_resource(pdev, IORESOURCE_MEM, i);
1111                 if (!res) {
1112                         dev_err(dev,"Unable to find IOMEM region\n");
1113                         return -ENOENT;
1114                 }
1115
1116                 data->res_bases[i] = devm_ioremap(dev,res->start,
1117                                                   resource_size(res));
1118                 if (!data->res_bases[i]) {
1119                         dev_err(dev, "Unable to map IOMEM @ PA:%#x\n",
1120                                 res->start);
1121                         return -ENOMEM;
1122                 }
1123
1124                 dev_dbg(dev,"res->start = 0x%08x ioremap to data->res_bases[%d] = 0x%08x\n",
1125                         res->start, i, (unsigned int)data->res_bases[i]);
1126
1127                 if (strstr(data->dbgname, "vop") &&
1128                     (soc_is_rk3128() || soc_is_rk3126())) {
1129                         rk312x_vop_mmu_base = data->res_bases[0];
1130                         dev_dbg(dev, "rk312x_vop_mmu_base = 0x%08x\n",
1131                                 (unsigned int)rk312x_vop_mmu_base);
1132                 }
1133         }
1134
1135         for (i = 0; i < data->num_res_irq; i++) {
1136                 if ((soc_is_rk3128() || soc_is_rk3126()) &&
1137                     strstr(data->dbgname, "vop")) {
1138                         dev_info(dev, "skip request vop mmu irq\n");
1139                         continue;
1140                 }
1141
1142                 ret = platform_get_irq(pdev, i);
1143                 if (ret <= 0) {
1144                         dev_err(dev,"Unable to find IRQ resource\n");
1145                         return -ENOENT;
1146                 }
1147
1148                 ret = devm_request_irq(dev, ret, rockchip_iommu_irq,
1149                                   IRQF_SHARED, dev_name(dev), data);
1150                 if (ret) {
1151                         dev_err(dev, "Unabled to register interrupt handler\n");
1152                         return -ENOENT;
1153                 }
1154         }
1155
1156         ret = rockchip_init_iovmm(dev, &data->vmm);
1157         if (ret)
1158                 return ret;
1159
1160         data->iommu = dev;
1161         spin_lock_init(&data->data_lock);
1162         INIT_LIST_HEAD(&data->node);
1163
1164         dev_info(dev,"(%s) Initialized\n", data->dbgname);
1165
1166         return 0;
1167 }
1168
1169 #ifdef CONFIG_OF
1170 static const struct of_device_id iommu_dt_ids[] = {
1171         { .compatible = IEP_IOMMU_COMPATIBLE_NAME},
1172         { .compatible = VIP_IOMMU_COMPATIBLE_NAME},
1173         { .compatible = VOPB_IOMMU_COMPATIBLE_NAME},
1174         { .compatible = VOPL_IOMMU_COMPATIBLE_NAME},
1175         { .compatible = HEVC_IOMMU_COMPATIBLE_NAME},
1176         { .compatible = VPU_IOMMU_COMPATIBLE_NAME},
1177         { .compatible = ISP_IOMMU_COMPATIBLE_NAME},
1178         { .compatible = VOP_IOMMU_COMPATIBLE_NAME},
1179         { /* end */ }
1180 };
1181
1182 MODULE_DEVICE_TABLE(of, iommu_dt_ids);
1183 #endif
1184
1185 static struct platform_driver rk_iommu_driver = {
1186         .probe = rockchip_iommu_probe,
1187         .remove = NULL,
1188         .driver = {
1189                    .name = "rk_iommu",
1190                    .owner = THIS_MODULE,
1191                    .of_match_table = of_match_ptr(iommu_dt_ids),
1192         },
1193 };
1194
1195 static int __init rockchip_iommu_init_driver(void)
1196 {
1197         int ret;
1198
1199         lv2table_kmem_cache = kmem_cache_create("rk-iommu-lv2table",
1200                                                 LV2TABLE_SIZE, LV2TABLE_SIZE,
1201                                                 0, NULL);
1202         if (!lv2table_kmem_cache) {
1203                 pr_info("%s: failed to create kmem cache\n", __func__);
1204                 return -ENOMEM;
1205         }
1206
1207         ret = bus_set_iommu(&platform_bus_type, &rk_iommu_ops);
1208         if (ret)
1209                 return ret;
1210
1211         return platform_driver_register(&rk_iommu_driver);
1212 }
1213
1214 core_initcall(rockchip_iommu_init_driver);