drivers/iommu/rk-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 "rk-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 int *rockchip_section_entry(unsigned int *pgtable, unsigned long iova)
 174 {
 175         return pgtable + rockchip_lv1ent_offset(iova);
 176 }
 177
 178 static unsigned int *rockchip_page_entry(unsigned int *sent, unsigned long iova)
 179 {
 180         return (unsigned int *)phys_to_virt(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 int *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         struct iommu_domain domain;
 191 };
 192
 193 static struct rk_iommu_domain *to_rk_domain(struct iommu_domain *dom)
 194 {
 195         return container_of(dom, struct rk_iommu_domain, domain);
 196 }
 197
 198 static bool rockchip_set_iommu_active(struct iommu_drvdata *data)
 199 {
 200         /* return true if the IOMMU was not active previously
 201            and it needs to be initialized */
 202         return ++data->activations == 1;
 203 }
 204
 205 static bool rockchip_set_iommu_inactive(struct iommu_drvdata *data)
 206 {
 207         /* return true if the IOMMU is needed to be disabled */
 208         BUG_ON(data->activations < 1);
 209         return --data->activations == 0;
 210 }
 211
 212 static bool rockchip_is_iommu_active(struct iommu_drvdata *data)
 213 {
 214         return data->activations > 0;
 215 }
 216
 217 static void rockchip_iommu_disable_stall(void __iomem *base)
 218 {
 219         int i;
 220         u32 mmu_status;
 221
 222         if (base != rk312x_vop_mmu_base) {
 223                 mmu_status = __raw_readl(base + IOMMU_REGISTER_STATUS);
 224         } else {
 225                 goto skip_vop_mmu_disable;
 226         }
 227
 228         if (0 == (mmu_status & IOMMU_STATUS_BIT_PAGING_ENABLED)) {
 229                 return;
 230         }
 231
 232         if (mmu_status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE) {
 233                 pr_info("Aborting MMU disable stall request since it is in pagefault state.\n");
 234                 return;
 235         }
 236
 237         if (!(mmu_status & IOMMU_STATUS_BIT_STALL_ACTIVE)) {
 238                 return;
 239         }
 240
 241         __raw_writel(IOMMU_COMMAND_DISABLE_STALL, base + IOMMU_REGISTER_COMMAND);
 242
 243         skip_vop_mmu_disable:
 244
 245         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
 246                 u32 status;
 247
 248                 if (base != rk312x_vop_mmu_base) {
 249                         status = __raw_readl(base + IOMMU_REGISTER_STATUS);
 250                 } else {
 251                         int j;
 252                         while (j < 5)
 253                                 j++;
 254                         return;
 255                 }
 256
 257                 if (0 == (status & IOMMU_STATUS_BIT_STALL_ACTIVE))
 258                         break;
 259
 260                 if (status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE)
 261                         break;
 262
 263                 if (0 == (mmu_status & IOMMU_STATUS_BIT_PAGING_ENABLED))
 264                         break;
 265         }
 266
 267         if (IOMMU_REG_POLL_COUNT_FAST == i) {
 268                 pr_info("Disable stall request failed, MMU status is 0x%08X\n",
 269                       __raw_readl(base + IOMMU_REGISTER_STATUS));
 270         }
 271 }
 272
 273 static bool rockchip_iommu_enable_stall(void __iomem *base)
 274 {
 275         int i;
 276
 277         u32 mmu_status;
 278
 279         if (base != rk312x_vop_mmu_base) {
 280                 mmu_status = __raw_readl(base + IOMMU_REGISTER_STATUS);
 281         } else {
 282                 goto skip_vop_mmu_enable;
 283         }
 284
 285         if (0 == (mmu_status & IOMMU_STATUS_BIT_PAGING_ENABLED)) {
 286                 return true;
 287         }
 288
 289         if (mmu_status & IOMMU_STATUS_BIT_STALL_ACTIVE){
 290                 pr_info("MMU stall already enabled\n");
 291                 return true;
 292         }
 293
 294         if (mmu_status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE) {
 295                 pr_info("Aborting MMU stall request since it is in pagefault state. mmu status is 0x%08x\n",
 296                         mmu_status);
 297                 return false;
 298         }
 299
 300         __raw_writel(IOMMU_COMMAND_ENABLE_STALL, base + IOMMU_REGISTER_COMMAND);
 301
 302         skip_vop_mmu_enable:
 303
 304         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
 305                 if (base != rk312x_vop_mmu_base) {
 306                         mmu_status = __raw_readl(base + IOMMU_REGISTER_STATUS);
 307                 } else {
 308                         int j;
 309                         while (j < 5)
 310                                 j++;
 311                         return true;
 312                 }
 313
 314                 if (mmu_status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE)
 315                         break;
 316
 317                 if ((mmu_status & IOMMU_STATUS_BIT_STALL_ACTIVE) &&
 318                     (0 == (mmu_status & IOMMU_STATUS_BIT_STALL_NOT_ACTIVE)))
 319                         break;
 320
 321                 if (0 == (mmu_status & (IOMMU_STATUS_BIT_PAGING_ENABLED)))
 322                         break;
 323         }
 324
 325         if (IOMMU_REG_POLL_COUNT_FAST == i) {
 326                 pr_info("Enable stall request failed, MMU status is 0x%08X\n",
 327                        __raw_readl(base + IOMMU_REGISTER_STATUS));
 328                 return false;
 329         }
 330
 331         if (mmu_status & IOMMU_STATUS_BIT_PAGE_FAULT_ACTIVE) {
 332                 pr_info("Aborting MMU stall request since it has a pagefault.\n");
 333                 return false;
 334         }
 335
 336         return true;
 337 }
 338
 339 static bool rockchip_iommu_enable_paging(void __iomem *base)
 340 {
 341         int i;
 342
 343         __raw_writel(IOMMU_COMMAND_ENABLE_PAGING,
 344                      base + IOMMU_REGISTER_COMMAND);
 345
 346         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
 347                 if (base != rk312x_vop_mmu_base) {
 348                         if (__raw_readl(base + IOMMU_REGISTER_STATUS) &
 349                                 IOMMU_STATUS_BIT_PAGING_ENABLED)
 350                         break;
 351                 } else {
 352                         int j;
 353                         while (j < 5)
 354                                 j++;
 355                         return true;
 356                 }
 357         }
 358
 359         if (IOMMU_REG_POLL_COUNT_FAST == i) {
 360                 pr_info("Enable paging request failed, MMU status is 0x%08X\n",
 361                        __raw_readl(base + IOMMU_REGISTER_STATUS));
 362                 return false;
 363         }
 364
 365         return true;
 366 }
 367
 368 static bool rockchip_iommu_disable_paging(void __iomem *base)
 369 {
 370         int i;
 371
 372         __raw_writel(IOMMU_COMMAND_DISABLE_PAGING,
 373                      base + IOMMU_REGISTER_COMMAND);
 374
 375         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
 376                 if (base != rk312x_vop_mmu_base) {
 377                         if (!(__raw_readl(base + IOMMU_REGISTER_STATUS) &
 378                                   IOMMU_STATUS_BIT_PAGING_ENABLED))
 379                                 break;
 380                 } else {
 381                         int j;
 382                         while (j < 5)
 383                                 j++;
 384                         return true;
 385                 }
 386         }
 387
 388         if (IOMMU_REG_POLL_COUNT_FAST == i) {
 389                 pr_info("Disable paging request failed, MMU status is 0x%08X\n",
 390                        __raw_readl(base + IOMMU_REGISTER_STATUS));
 391                 return false;
 392         }
 393
 394         return true;
 395 }
 396
 397 static void rockchip_iommu_page_fault_done(void __iomem *base, const char *dbgname)
 398 {
 399         pr_info("MMU: %s: Leaving page fault mode\n",
 400                 dbgname);
 401         __raw_writel(IOMMU_COMMAND_PAGE_FAULT_DONE,
 402                      base + IOMMU_REGISTER_COMMAND);
 403 }
 404
 405 static int rockchip_iommu_zap_tlb_without_stall (void __iomem *base)
 406 {
 407         __raw_writel(IOMMU_COMMAND_ZAP_CACHE, base + IOMMU_REGISTER_COMMAND);
 408
 409         return 0;
 410 }
 411
 412 static int rockchip_iommu_zap_tlb(void __iomem *base)
 413 {
 414         if (!rockchip_iommu_enable_stall(base)) {
 415                 pr_err("%s failed\n", __func__);
 416                 return -1;
 417         }
 418
 419         __raw_writel(IOMMU_COMMAND_ZAP_CACHE, base + IOMMU_REGISTER_COMMAND);
 420
 421         rockchip_iommu_disable_stall(base);
 422
 423         return 0;
 424 }
 425
 426 static inline bool rockchip_iommu_raw_reset(void __iomem *base)
 427 {
 428         int i;
 429         unsigned int ret;
 430         unsigned int grf_value;
 431
 432         __raw_writel(0xCAFEBABE, base + IOMMU_REGISTER_DTE_ADDR);
 433
 434         if (base != rk312x_vop_mmu_base) {
 435                 ret = __raw_readl(base + IOMMU_REGISTER_DTE_ADDR);
 436                 if (!(0xCAFEB000 == ret)) {
 437                         grf_value = readl_relaxed(RK_GRF_VIRT + RK3036_GRF_SOC_CON1);
 438                         pr_info("error when %s. grf = 0x%08x\n", __func__, grf_value);
 439                         return false;
 440                 }
 441         }
 442         __raw_writel(IOMMU_COMMAND_HARD_RESET,
 443                      base + IOMMU_REGISTER_COMMAND);
 444
 445         for (i = 0; i < IOMMU_REG_POLL_COUNT_FAST; ++i) {
 446                 if (base != rk312x_vop_mmu_base) {
 447                         if (__raw_readl(base + IOMMU_REGISTER_DTE_ADDR) == 0)
 448                                 break;
 449                 } else {
 450                         int j;
 451                         while (j < 5)
 452                                 j++;
 453                         return true;
 454                 }
 455         }
 456
 457         if (IOMMU_REG_POLL_COUNT_FAST == i) {
 458                 pr_info("%s,Reset request failed, MMU status is 0x%08X\n",
 459                        __func__, __raw_readl(base + IOMMU_REGISTER_DTE_ADDR));
 460                 return false;
 461         }
 462         return true;
 463 }
 464
 465 static void rockchip_iommu_set_ptbase(void __iomem *base, unsigned int pgd)
 466 {
 467         __raw_writel(pgd, base + IOMMU_REGISTER_DTE_ADDR);
 468 }
 469
 470 static bool rockchip_iommu_reset(void __iomem *base, const char *dbgname)
 471 {
 472         bool ret = true;
 473
 474         ret = rockchip_iommu_raw_reset(base);
 475         if (!ret) {
 476                 pr_info("(%s), %s failed\n", dbgname, __func__);
 477                 return ret;
 478         }
 479
 480         if (base != rk312x_vop_mmu_base)
 481                 __raw_writel(IOMMU_INTERRUPT_PAGE_FAULT |
 482                              IOMMU_INTERRUPT_READ_BUS_ERROR,
 483                              base + IOMMU_REGISTER_INT_MASK);
 484         else
 485                 __raw_writel(0x00, base + IOMMU_REGISTER_INT_MASK);
 486
 487         return ret;
 488 }
 489
 490 static inline void rockchip_pgtable_flush(void *vastart, void *vaend)
 491 {
 492 #ifdef CONFIG_ARM
 493         dmac_flush_range(vastart, vaend);
 494         outer_flush_range(virt_to_phys(vastart), virt_to_phys(vaend));
 495 #elif defined(CONFIG_ARM64)
 496         __dma_flush_range(vastart, vaend);
 497         //flush_cache_all();
 498 #endif
 499 }
 500
 501 static void dump_pagetbl(dma_addr_t fault_address, u32 addr_dte)
 502 {
 503         u32 dte_index, pte_index, page_offset;
 504         u32 mmu_dte_addr;
 505         phys_addr_t mmu_dte_addr_phys, dte_addr_phys;
 506         u32 *dte_addr;
 507         u32 dte;
 508         phys_addr_t pte_addr_phys = 0;
 509         u32 *pte_addr = NULL;
 510         u32 pte = 0;
 511         phys_addr_t page_addr_phys = 0;
 512         u32 page_flags = 0;
 513
 514         dte_index = rockchip_lv1ent_offset(fault_address);
 515         pte_index = rockchip_lv2ent_offset(fault_address);
 516         page_offset = (u32)(fault_address & 0x00000fff);
 517
 518         mmu_dte_addr = addr_dte;
 519         mmu_dte_addr_phys = (phys_addr_t)mmu_dte_addr;
 520
 521         dte_addr_phys = mmu_dte_addr_phys + (4 * dte_index);
 522         dte_addr = phys_to_virt(dte_addr_phys);
 523         dte = *dte_addr;
 524
 525         if (!(IOMMU_FLAGS_PRESENT & dte))
 526                 goto print_it;
 527
 528         pte_addr_phys = ((phys_addr_t)dte & 0xfffff000) + (pte_index * 4);
 529         pte_addr = phys_to_virt(pte_addr_phys);
 530         pte = *pte_addr;
 531
 532         if (!(IOMMU_FLAGS_PRESENT & pte))
 533                 goto print_it;
 534
 535         page_addr_phys = ((phys_addr_t)pte & 0xfffff000) + page_offset;
 536         page_flags = pte & 0x000001fe;
 537
 538 print_it:
 539         pr_err("iova = %pad: dte_index: 0x%03x pte_index: 0x%03x page_offset: 0x%03x\n",
 540                 &fault_address, dte_index, pte_index, page_offset);
 541         pr_err("mmu_dte_addr: %pa dte@%pa: %#08x valid: %u pte@%pa: %#08x valid: %u page@%pa flags: %#03x\n",
 542                 &mmu_dte_addr_phys, &dte_addr_phys, dte,
 543                 (dte & IOMMU_FLAGS_PRESENT), &pte_addr_phys, pte,
 544                 (pte & IOMMU_FLAGS_PRESENT), &page_addr_phys, page_flags);
 545 }
 546
 547 static irqreturn_t rockchip_iommu_irq(int irq, void *dev_id)
 548 {
 549         /* SYSMMU is in blocked when interrupt occurred. */
 550         struct iommu_drvdata *data = dev_id;
 551         u32 status;
 552         u32 rawstat;
 553         dma_addr_t fault_address;
 554         int i;
 555         unsigned long flags;
 556         int ret;
 557         u32 reg_status;
 558
 559         spin_lock_irqsave(&data->data_lock, flags);
 560
 561         if (!rockchip_is_iommu_active(data)) {
 562                 spin_unlock_irqrestore(&data->data_lock, flags);
 563                 return IRQ_HANDLED;
 564         }
 565
 566         for (i = 0; i < data->num_res_mem; i++) {
 567                 status = __raw_readl(data->res_bases[i] +
 568                                      IOMMU_REGISTER_INT_STATUS);
 569                 if (status == 0)
 570                         continue;
 571
 572                 rawstat = __raw_readl(data->res_bases[i] +
 573                                       IOMMU_REGISTER_INT_RAWSTAT);
 574
 575                 reg_status = __raw_readl(data->res_bases[i] +
 576                                          IOMMU_REGISTER_STATUS);
 577
 578                 dev_info(data->iommu, "1.rawstat = 0x%08x,status = 0x%08x,reg_status = 0x%08x\n",
 579                          rawstat, status, reg_status);
 580
 581                 if (rawstat & IOMMU_INTERRUPT_PAGE_FAULT) {
 582                         u32 dte;
 583                         int flags;
 584
 585                         fault_address = __raw_readl(data->res_bases[i] +
 586                                             IOMMU_REGISTER_PAGE_FAULT_ADDR);
 587
 588                         dte = __raw_readl(data->res_bases[i] +
 589                                           IOMMU_REGISTER_DTE_ADDR);
 590
 591                         flags = (status & 32) ? 1 : 0;
 592
 593                         dev_err(data->iommu, "Page fault detected at %pad from bus id %d of type %s on %s\n",
 594                                 &fault_address, (status >> 6) & 0x1F,
 595                                 (flags == 1) ? "write" : "read", data->dbgname);
 596
 597                         dump_pagetbl(fault_address, dte);
 598
 599                         if (data->domain)
 600                                 report_iommu_fault(data->domain, data->iommu,
 601                                                    fault_address, flags);
 602                         if (data->fault_handler)
 603                                 data->fault_handler(data->master, IOMMU_PAGEFAULT, dte, fault_address, 1);
 604
 605                         rockchip_iommu_page_fault_done(data->res_bases[i],
 606                                                        data->dbgname);
 607                 }
 608
 609                 if (rawstat & IOMMU_INTERRUPT_READ_BUS_ERROR) {
 610                         dev_err(data->iommu, "bus error occured at %pad\n",
 611                                 &fault_address);
 612                 }
 613
 614                 if (rawstat & ~(IOMMU_INTERRUPT_READ_BUS_ERROR |
 615                     IOMMU_INTERRUPT_PAGE_FAULT)) {
 616                         dev_err(data->iommu, "unexpected int_status: %#08x\n\n",
 617                                 rawstat);
 618                 }
 619
 620                 __raw_writel(rawstat, data->res_bases[i] +
 621                              IOMMU_REGISTER_INT_CLEAR);
 622
 623                 status = __raw_readl(data->res_bases[i] +
 624                                      IOMMU_REGISTER_INT_STATUS);
 625
 626                 rawstat = __raw_readl(data->res_bases[i] +
 627                                       IOMMU_REGISTER_INT_RAWSTAT);
 628
 629                 reg_status = __raw_readl(data->res_bases[i] +
 630                                          IOMMU_REGISTER_STATUS);
 631
 632                 dev_info(data->iommu, "2.rawstat = 0x%08x,status = 0x%08x,reg_status = 0x%08x\n",
 633                          rawstat, status, reg_status);
 634
 635                 ret = rockchip_iommu_zap_tlb_without_stall(data->res_bases[i]);
 636                 if (ret)
 637                         dev_err(data->iommu, "(%s) %s failed\n", data->dbgname,
 638                                 __func__);
 639         }
 640
 641         spin_unlock_irqrestore(&data->data_lock, flags);
 642         return IRQ_HANDLED;
 643 }
 644
 645 static bool rockchip_iommu_disable(struct iommu_drvdata *data)
 646 {
 647         unsigned long flags;
 648         int i;
 649         bool ret = false;
 650
 651         spin_lock_irqsave(&data->data_lock, flags);
 652
 653         if (!rockchip_set_iommu_inactive(data)) {
 654                 spin_unlock_irqrestore(&data->data_lock, flags);
 655                 dev_info(data->iommu,"(%s) %d times left to be disabled\n",
 656                          data->dbgname, data->activations);
 657                 return ret;
 658         }
 659
 660         for (i = 0; i < data->num_res_mem; i++) {
 661                 ret = rockchip_iommu_enable_stall(data->res_bases[i]);
 662                 if (!ret) {
 663                         dev_info(data->iommu, "(%s), %s failed\n",
 664                                  data->dbgname, __func__);
 665                         spin_unlock_irqrestore(&data->data_lock, flags);
 666                         return false;
 667                 }
 668
 669                 __raw_writel(0, data->res_bases[i] + IOMMU_REGISTER_INT_MASK);
 670
 671                 ret = rockchip_iommu_disable_paging(data->res_bases[i]);
 672                 if (!ret) {
 673                         rockchip_iommu_disable_stall(data->res_bases[i]);
 674                         spin_unlock_irqrestore(&data->data_lock, flags);
 675                         dev_info(data->iommu, "%s error\n", __func__);
 676                         return ret;
 677                 }
 678                 rockchip_iommu_disable_stall(data->res_bases[i]);
 679         }
 680
 681         data->pgtable = 0;
 682
 683         spin_unlock_irqrestore(&data->data_lock, flags);
 684
 685         dev_dbg(data->iommu,"(%s) Disabled\n", data->dbgname);
 686
 687         return ret;
 688 }
 689
 690 /* __rk_sysmmu_enable: Enables System MMU
 691  *
 692  * returns -error if an error occurred and System MMU is not enabled,
 693  * 0 if the System MMU has been just enabled and 1 if System MMU was already
 694  * enabled before.
 695  */
 696 static int rockchip_iommu_enable(struct iommu_drvdata *data, unsigned int pgtable)
 697 {
 698         int i, ret = 0;
 699         unsigned long flags;
 700
 701         spin_lock_irqsave(&data->data_lock, flags);
 702
 703         if (!rockchip_set_iommu_active(data)) {
 704                 if (WARN_ON(pgtable != data->pgtable))
 705                         ret = -EBUSY;
 706                 else
 707                         ret = 1;
 708
 709                 dev_info(data->iommu, "(%s) Already enabled\n", data->dbgname);
 710
 711                 goto enable_out;
 712         }
 713
 714         for (i = 0; i < data->num_res_mem; i++) {
 715                 ret = rockchip_iommu_enable_stall(data->res_bases[i]);
 716                 if (!ret) {
 717                         dev_info(data->iommu, "(%s), %s failed\n",
 718                                  data->dbgname, __func__);
 719                         ret = -EBUSY;
 720                         goto enable_out;
 721                 }
 722
 723                 if (!strstr(data->dbgname, "isp")) {
 724                         if (!rockchip_iommu_reset(data->res_bases[i],
 725                              data->dbgname)) {
 726                                 rockchip_iommu_disable_stall(data->res_bases[i]);
 727                                 ret = -ENOENT;
 728                                 goto enable_out;
 729                         }
 730                 }
 731
 732                 rockchip_iommu_set_ptbase(data->res_bases[i], pgtable);
 733
 734                 __raw_writel(IOMMU_COMMAND_ZAP_CACHE, data->res_bases[i] +
 735                              IOMMU_REGISTER_COMMAND);
 736
 737                 if (strstr(data->dbgname, "isp")) {
 738                         __raw_writel(IOMMU_INTERRUPT_PAGE_FAULT |
 739                                 IOMMU_INTERRUPT_READ_BUS_ERROR,
 740                              data->res_bases[i] + IOMMU_REGISTER_INT_MASK);
 741                 }
 742
 743                 ret = rockchip_iommu_enable_paging(data->res_bases[i]);
 744                 if (!ret) {
 745                         dev_info(data->iommu, "(%s), %s failed\n",
 746                                  data->dbgname, __func__);
 747                         rockchip_iommu_disable_stall(data->res_bases[i]);
 748                         ret = -EBUSY;
 749                         goto enable_out;
 750                 }
 751
 752                 rockchip_iommu_disable_stall(data->res_bases[i]);
 753         }
 754
 755         data->pgtable = pgtable;
 756         spin_unlock_irqrestore(&data->data_lock, flags);
 757
 758         dev_dbg(data->iommu,"(%s) Enabled\n", data->dbgname);
 759
 760         return 0;
 761
 762 enable_out:
 763         rockchip_set_iommu_inactive(data);
 764         spin_unlock_irqrestore(&data->data_lock, flags);
 765
 766         return ret;
 767 }
 768
 769 int rockchip_iommu_tlb_invalidate_global(struct device *dev)
 770 {
 771         unsigned long flags;
 772         struct iommu_drvdata *data = dev_get_drvdata(dev->archdata.iommu);
 773         int ret = 0;
 774
 775         spin_lock_irqsave(&data->data_lock, flags);
 776
 777         if (rockchip_is_iommu_active(data)) {
 778                 int i;
 779
 780                 for (i = 0; i < data->num_res_mem; i++) {
 781                         ret = rockchip_iommu_zap_tlb(data->res_bases[i]);
 782                         if (ret)
 783                                 dev_err(dev->archdata.iommu, "(%s) %s failed\n",
 784                                         data->dbgname, __func__);
 785                 }
 786         } else {
 787                 dev_dbg(dev->archdata.iommu, "(%s) Disabled. Skipping invalidating TLB.\n",
 788                         data->dbgname);
 789                 ret = -1;
 790         }
 791
 792         spin_unlock_irqrestore(&data->data_lock, flags);
 793
 794         return ret;
 795 }
 796
 797 int rockchip_iommu_tlb_invalidate(struct device *dev)
 798 {
 799         unsigned long flags;
 800         struct iommu_drvdata *data = dev_get_drvdata(dev->archdata.iommu);
 801
 802         if (strstr(data->dbgname, "vpu") || strstr(data->dbgname, "hevc"))
 803                         return 0;
 804
 805         spin_lock_irqsave(&data->data_lock, flags);
 806
 807         if (rockchip_is_iommu_active(data)) {
 808                 int i;
 809                 int ret;
 810
 811                 for (i = 0; i < data->num_res_mem; i++) {
 812                         ret = rockchip_iommu_zap_tlb(data->res_bases[i]);
 813                         if (ret) {
 814                                 dev_err(dev->archdata.iommu, "(%s) %s failed\n",
 815                                         data->dbgname, __func__);
 816                                 spin_unlock_irqrestore(&data->data_lock, flags);
 817                                 return ret;
 818                         }
 819
 820                 }
 821         } else {
 822                 dev_dbg(dev->archdata.iommu, "(%s) Disabled. Skipping invalidating TLB.\n",
 823                         data->dbgname);
 824         }
 825
 826         spin_unlock_irqrestore(&data->data_lock, flags);
 827
 828         return 0;
 829 }
 830
 831 static phys_addr_t rockchip_iommu_iova_to_phys(struct iommu_domain *domain,
 832                                                dma_addr_t iova)
 833 {
 834         struct rk_iommu_domain *priv = to_rk_domain(domain);
 835         unsigned int *entry;
 836         unsigned long flags;
 837         phys_addr_t phys = 0;
 838
 839         spin_lock_irqsave(&priv->pgtablelock, flags);
 840
 841         entry = rockchip_section_entry(priv->pgtable, iova);
 842         entry = rockchip_page_entry(entry, iova);
 843         phys = rockchip_spage_phys(entry) + rockchip_spage_offs(iova);
 844
 845         spin_unlock_irqrestore(&priv->pgtablelock, flags);
 846
 847         return phys;
 848 }
 849
 850 static int rockchip_lv2set_page(unsigned int *pent, phys_addr_t paddr,
 851                        size_t size, short *pgcnt)
 852 {
 853         if (!rockchip_lv2ent_fault(pent))
 854                 return -EADDRINUSE;
 855
 856         *pent = rockchip_mk_lv2ent_spage(paddr);
 857         rockchip_pgtable_flush(pent, pent + 1);
 858         *pgcnt -= 1;
 859         return 0;
 860 }
 861
 862 static unsigned int *rockchip_alloc_lv2entry(unsigned int *sent,
 863                                      unsigned long iova, short *pgcounter)
 864 {
 865         if (rockchip_lv1ent_fault(sent)) {
 866                 unsigned int *pent;
 867
 868                 pent = kmem_cache_zalloc(lv2table_kmem_cache, GFP_ATOMIC);
 869                 BUG_ON((unsigned long)pent & (LV2TABLE_SIZE - 1));
 870                 if (!pent)
 871                         return NULL;
 872
 873                 *sent = rockchip_mk_lv1ent_page(virt_to_phys(pent));
 874                 kmemleak_ignore(pent);
 875                 *pgcounter = NUM_LV2ENTRIES;
 876                 rockchip_pgtable_flush(pent, pent + NUM_LV2ENTRIES);
 877                 rockchip_pgtable_flush(sent, sent + 1);
 878         }
 879         return rockchip_page_entry(sent, iova);
 880 }
 881
 882 static size_t rockchip_iommu_unmap(struct iommu_domain *domain,
 883                                    unsigned long iova, size_t size)
 884 {
 885         struct rk_iommu_domain *priv = to_rk_domain(domain);
 886         unsigned long flags;
 887         unsigned int *ent;
 888
 889         BUG_ON(priv->pgtable == NULL);
 890
 891         spin_lock_irqsave(&priv->pgtablelock, flags);
 892
 893         ent = rockchip_section_entry(priv->pgtable, iova);
 894
 895         if (unlikely(rockchip_lv1ent_fault(ent))) {
 896                 if (size > SPAGE_SIZE)
 897                         size = SPAGE_SIZE;
 898                 goto done;
 899         }
 900
 901         /* lv1ent_page(sent) == true here */
 902
 903         ent = rockchip_page_entry(ent, iova);
 904
 905         if (unlikely(rockchip_lv2ent_fault(ent))) {
 906                 size = SPAGE_SIZE;
 907                 goto done;
 908         }
 909
 910         *ent = 0;
 911         size = SPAGE_SIZE;
 912         priv->lv2entcnt[rockchip_lv1ent_offset(iova)] += 1;
 913         goto done;
 914
 915 done:
 916         pr_debug("%s:unmap iova 0x%lx/%zx bytes\n",
 917                   __func__, iova,size);
 918         spin_unlock_irqrestore(&priv->pgtablelock, flags);
 919
 920         return size;
 921 }
 922
 923 static int rockchip_iommu_map(struct iommu_domain *domain, unsigned long iova,
 924                               phys_addr_t paddr, size_t size, int prot)
 925 {
 926         struct rk_iommu_domain *priv = to_rk_domain(domain);
 927         unsigned int *entry;
 928         unsigned long flags;
 929         int ret = -ENOMEM;
 930         unsigned int *pent;
 931
 932         BUG_ON(priv->pgtable == NULL);
 933
 934         spin_lock_irqsave(&priv->pgtablelock, flags);
 935
 936         entry = rockchip_section_entry(priv->pgtable, iova);
 937
 938         pent = rockchip_alloc_lv2entry(entry, iova,
 939                               &priv->lv2entcnt[rockchip_lv1ent_offset(iova)]);
 940         if (!pent)
 941                 ret = -ENOMEM;
 942         else
 943                 ret = rockchip_lv2set_page(pent, paddr, size,
 944                                 &priv->lv2entcnt[rockchip_lv1ent_offset(iova)]);
 945
 946         if (ret) {
 947                 pr_info("%s: Failed to map iova 0x%lx/%zx bytes\n", __func__,
 948                        iova, size);
 949         }
 950         spin_unlock_irqrestore(&priv->pgtablelock, flags);
 951
 952         return ret;
 953 }
 954
 955 static void rockchip_iommu_detach_device(struct iommu_domain *domain, struct device *dev)
 956 {
 957         struct iommu_drvdata *data = dev_get_drvdata(dev->archdata.iommu);
 958         struct rk_iommu_domain *priv = to_rk_domain(domain);
 959         struct list_head *pos;
 960         unsigned long flags;
 961         bool found = false;
 962
 963         spin_lock_irqsave(&priv->lock, flags);
 964
 965         list_for_each(pos, &priv->clients) {
 966                 if (list_entry(pos, struct iommu_drvdata, node) == data) {
 967                         found = true;
 968                         break;
 969                 }
 970         }
 971
 972         if (!found) {
 973                 spin_unlock_irqrestore(&priv->lock, flags);
 974                 return;
 975         }
 976
 977         if (rockchip_iommu_disable(data)) {
 978                 if (!(strstr(data->dbgname, "vpu") || strstr(data->dbgname, "hevc")))
 979                         dev_dbg(dev->archdata.iommu,"%s: Detached IOMMU with pgtable %08lx\n",
 980                                 __func__, (unsigned long)virt_to_phys(priv->pgtable));
 981                 data->domain = NULL;
 982                 list_del_init(&data->node);
 983
 984         } else
 985                 dev_err(dev->archdata.iommu,"%s: Detaching IOMMU with pgtable %08lx delayed",
 986                         __func__, (unsigned long)virt_to_phys(priv->pgtable));
 987
 988         spin_unlock_irqrestore(&priv->lock, flags);
 989 }
 990
 991 static int rockchip_iommu_attach_device(struct iommu_domain *domain, struct device *dev)
 992 {
 993         struct iommu_drvdata *data = dev_get_drvdata(dev->archdata.iommu);
 994         struct rk_iommu_domain *priv = to_rk_domain(domain);
 995         unsigned long flags;
 996         int ret;
 997
 998         spin_lock_irqsave(&priv->lock, flags);
 999
1000         ret = rockchip_iommu_enable(data, virt_to_phys(priv->pgtable));
1001
1002         if (ret == 0) {
1003                 /* 'data->node' must not be appeared in priv->clients */
1004                 BUG_ON(!list_empty(&data->node));
1005                 list_add_tail(&data->node, &priv->clients);
1006                 data->domain = domain;
1007                 data->master = dev;
1008         }
1009
1010         spin_unlock_irqrestore(&priv->lock, flags);
1011
1012         if (ret < 0) {
1013                 dev_err(dev->archdata.iommu,"%s: Failed to attach IOMMU with pgtable %x\n",
1014                        __func__, (unsigned int)virt_to_phys(priv->pgtable));
1015         } else if (ret > 0) {
1016                 dev_dbg(dev->archdata.iommu,"%s: IOMMU with pgtable 0x%x already attached\n",
1017                         __func__, (unsigned int)virt_to_phys(priv->pgtable));
1018         } else {
1019                 if (!(strstr(data->dbgname, "vpu") ||
1020                       strstr(data->dbgname, "hevc") ||
1021                       strstr(data->dbgname, "vdec")))
1022                         dev_info(dev->archdata.iommu,"%s: Attached new IOMMU with pgtable 0x%x\n",
1023                                 __func__, (unsigned int)virt_to_phys(priv->pgtable));
1024         }
1025
1026         return ret;
1027 }
1028
1029 static void rockchip_iommu_domain_free(struct iommu_domain *domain)
1030 {
1031         struct rk_iommu_domain *priv = to_rk_domain(domain);
1032         int i;
1033
1034         WARN_ON(!list_empty(&priv->clients));
1035
1036         for (i = 0; i < NUM_LV1ENTRIES; i++)
1037                 if (rockchip_lv1ent_page(priv->pgtable + i))
1038                         kmem_cache_free(lv2table_kmem_cache,
1039                                         phys_to_virt(rockchip_lv2table_base(priv->pgtable + i)));
1040
1041         free_pages((unsigned long)priv->pgtable, 0);
1042         free_pages((unsigned long)priv->lv2entcnt, 0);
1043         kfree(priv);
1044 }
1045
1046 static struct iommu_domain *rockchip_iommu_domain_alloc(unsigned type)
1047 {
1048         struct rk_iommu_domain *priv;
1049
1050         if (type != IOMMU_DOMAIN_UNMANAGED)
1051                 return NULL;
1052
1053         priv = kzalloc(sizeof(*priv), GFP_KERNEL);
1054         if (!priv)
1055                 return NULL;
1056
1057 /*rk32xx iommu use 2 level pagetable,
1058    level1 and leve2 both have 1024 entries,each entry  occupy 4 bytes,
1059    so alloc a page size for each page table
1060 */
1061         priv->pgtable = (unsigned int *)__get_free_pages(GFP_KERNEL |
1062                                                           __GFP_ZERO, 0);
1063         if (!priv->pgtable)
1064                 goto err_pgtable;
1065
1066         priv->lv2entcnt = (short *)__get_free_pages(GFP_KERNEL |
1067                                                     __GFP_ZERO, 0);
1068         if (!priv->lv2entcnt)
1069                 goto err_counter;
1070
1071         rockchip_pgtable_flush(priv->pgtable, priv->pgtable + NUM_LV1ENTRIES);
1072
1073         spin_lock_init(&priv->lock);
1074         spin_lock_init(&priv->pgtablelock);
1075         INIT_LIST_HEAD(&priv->clients);
1076
1077         return &priv->domain;
1078
1079 err_counter:
1080         free_pages((unsigned long)priv->pgtable, 0);
1081 err_pgtable:
1082         kfree(priv);
1083         return NULL;
1084 }
1085
1086 static struct iommu_ops rk_iommu_ops = {
1087         .domain_alloc = rockchip_iommu_domain_alloc,
1088         .domain_free = rockchip_iommu_domain_free,
1089         .attach_dev = rockchip_iommu_attach_device,
1090         .detach_dev = rockchip_iommu_detach_device,
1091         .map = rockchip_iommu_map,
1092         .unmap = rockchip_iommu_unmap,
1093         .iova_to_phys = rockchip_iommu_iova_to_phys,
1094         .pgsize_bitmap = SPAGE_SIZE,
1095 };
1096
1097 static int  rockchip_get_iommu_resource_num(struct platform_device *pdev,
1098                                              unsigned int type)
1099 {
1100         int num = 0;
1101         int i;
1102
1103         for (i = 0; i < pdev->num_resources; i++) {
1104                 struct resource *r = &pdev->resource[i];
1105                 if (type == resource_type(r))
1106                         num++;
1107         }
1108
1109         return num;
1110 }
1111
1112 static int rockchip_iommu_probe(struct platform_device *pdev)
1113 {
1114         int i, ret;
1115         struct device *dev;
1116         struct iommu_drvdata *data;
1117
1118         dev = &pdev->dev;
1119
1120         data = devm_kzalloc(dev, sizeof(*data), GFP_KERNEL);
1121         if (!data) {
1122                 dev_dbg(dev, "Not enough memory\n");
1123                 return -ENOMEM;
1124         }
1125
1126         dev_set_drvdata(dev, data);
1127
1128         if (pdev->dev.of_node)
1129                 of_property_read_string(pdev->dev.of_node, "dbgname",
1130                                         &(data->dbgname));
1131         else
1132                 dev_dbg(dev, "dbgname not assigned in device tree or device node not exist\r\n");
1133
1134         dev_info(dev,"(%s) Enter\n", data->dbgname);
1135
1136         data->num_res_mem = rockchip_get_iommu_resource_num(pdev,
1137                                 IORESOURCE_MEM);
1138         if (0 == data->num_res_mem) {
1139                 dev_err(dev,"can't find iommu memory resource \r\n");
1140                 return -ENOMEM;
1141         }
1142         dev_dbg(dev,"data->num_res_mem=%d\n", data->num_res_mem);
1143
1144         data->num_res_irq = rockchip_get_iommu_resource_num(pdev,
1145                                 IORESOURCE_IRQ);
1146         if (0 == data->num_res_irq) {
1147                 dev_err(dev,"can't find iommu irq resource \r\n");
1148                 return -ENOMEM;
1149         }
1150         dev_dbg(dev,"data->num_res_irq=%d\n", data->num_res_irq);
1151
1152         data->res_bases = devm_kmalloc_array(dev, data->num_res_mem,
1153                                 sizeof(*data->res_bases), GFP_KERNEL);
1154         if (data->res_bases == NULL) {
1155                 dev_err(dev, "Not enough memory\n");
1156                 return -ENOMEM;
1157         }
1158
1159         for (i = 0; i < data->num_res_mem; i++) {
1160                 struct resource *res;
1161
1162                 res = platform_get_resource(pdev, IORESOURCE_MEM, i);
1163                 if (!res) {
1164                         dev_err(dev,"Unable to find IOMEM region\n");
1165                         return -ENOENT;
1166                 }
1167
1168                 data->res_bases[i] = devm_ioremap(dev,res->start,
1169                                                   resource_size(res));
1170                 if (!data->res_bases[i]) {
1171                         dev_err(dev, "Unable to map IOMEM @ PA:%pa\n",
1172                                 &res->start);
1173                         return -ENOMEM;
1174                 }
1175
1176                 dev_dbg(dev,"res->start = 0x%pa ioremap to data->res_bases[%d] = %p\n",
1177                         &res->start, i, data->res_bases[i]);
1178
1179                 if (strstr(data->dbgname, "vop") &&
1180                     (soc_is_rk3128() || soc_is_rk3126())) {
1181                         rk312x_vop_mmu_base = data->res_bases[0];
1182                         dev_dbg(dev, "rk312x_vop_mmu_base = %p\n",
1183                                 rk312x_vop_mmu_base);
1184                 }
1185         }
1186
1187         for (i = 0; i < data->num_res_irq; i++) {
1188                 if ((soc_is_rk3128() || soc_is_rk3126()) &&
1189                     strstr(data->dbgname, "vop")) {
1190                         dev_info(dev, "skip request vop mmu irq\n");
1191                         continue;
1192                 }
1193
1194                 ret = platform_get_irq(pdev, i);
1195                 if (ret <= 0) {
1196                         dev_err(dev,"Unable to find IRQ resource\n");
1197                         return -ENOENT;
1198                 }
1199
1200                 ret = devm_request_irq(dev, ret, rockchip_iommu_irq,
1201                                   IRQF_SHARED, dev_name(dev), data);
1202                 if (ret) {
1203                         dev_err(dev, "Unabled to register interrupt handler\n");
1204                         return -ENOENT;
1205                 }
1206         }
1207
1208         ret = rockchip_init_iovmm(dev, &data->vmm);
1209         if (ret)
1210                 return ret;
1211
1212         data->iommu = dev;
1213         spin_lock_init(&data->data_lock);
1214         INIT_LIST_HEAD(&data->node);
1215
1216         dev_info(dev,"(%s) Initialized\n", data->dbgname);
1217
1218         return 0;
1219 }
1220
1221 #ifdef CONFIG_OF
1222 static const struct of_device_id iommu_dt_ids[] = {
1223         { .compatible = IEP_IOMMU_COMPATIBLE_NAME},
1224         { .compatible = VIP_IOMMU_COMPATIBLE_NAME},
1225         { .compatible = VOPB_IOMMU_COMPATIBLE_NAME},
1226         { .compatible = VOPL_IOMMU_COMPATIBLE_NAME},
1227         { .compatible = HEVC_IOMMU_COMPATIBLE_NAME},
1228         { .compatible = VPU_IOMMU_COMPATIBLE_NAME},
1229         { .compatible = ISP_IOMMU_COMPATIBLE_NAME},
1230         { .compatible = ISP0_IOMMU_COMPATIBLE_NAME},
1231         { .compatible = ISP1_IOMMU_COMPATIBLE_NAME},
1232         { .compatible = VOP_IOMMU_COMPATIBLE_NAME},
1233         { .compatible = VDEC_IOMMU_COMPATIBLE_NAME},
1234         { /* end */ }
1235 };
1236
1237 MODULE_DEVICE_TABLE(of, iommu_dt_ids);
1238 #endif
1239
1240 static struct platform_driver rk_iommu_driver = {
1241         .probe = rockchip_iommu_probe,
1242         .remove = NULL,
1243         .driver = {
1244                    .name = "rk_iommu",
1245                    .owner = THIS_MODULE,
1246                    .of_match_table = of_match_ptr(iommu_dt_ids),
1247         },
1248 };
1249
1250 static int __init rockchip_iommu_init_driver(void)
1251 {
1252         struct device_node *np;
1253         int ret;
1254
1255         np = of_find_matching_node(NULL, iommu_dt_ids);
1256         if (!np) {
1257                 pr_err("Failed to find legacy iommu devices\n");
1258                 return -ENODEV;
1259         }
1260
1261         lv2table_kmem_cache = kmem_cache_create("rk-iommu-lv2table",
1262                                                 LV2TABLE_SIZE, LV2TABLE_SIZE,
1263                                                 0, NULL);
1264         if (!lv2table_kmem_cache) {
1265                 pr_info("%s: failed to create kmem cache\n", __func__);
1266                 return -ENOMEM;
1267         }
1268
1269         ret = bus_set_iommu(&platform_bus_type, &rk_iommu_ops);
1270         if (ret)
1271                 return ret;
1272
1273         return platform_driver_register(&rk_iommu_driver);
1274 }
1275
1276 core_initcall(rockchip_iommu_init_driver);